{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.814697265625e-05,
  "eval_steps": 500,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "model_forward_time": 3.610551118850708,
      "step": 0
    },
    {
      "dynamics_loss_avg": 0.22903507947921753,
      "epoch": 0,
      "step": 0
    },
    {
      "action_loss_avg": 0.4171450436115265,
      "epoch": 0,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 8.803433418273926
    },
    {
      "epoch": 0,
      "model_forward_time": 3.1300106048583984,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 9.876571655273438
    },
    {
      "epoch": 0,
      "model_forward_time": 2.423023223876953,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 7.425048828125
    },
    {
      "epoch": 0,
      "model_forward_time": 3.125946044921875,
      "step": 0
    },
    {
      "epoch": 0,
      "step": 0,
      "training_step_time": 10.04902958869934
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 3.5986990928649902,
      "step": 1
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 1,
      "training_step_time": 10.359023809432983
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 2.436136484146118,
      "step": 1
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 1,
      "training_step_time": 7.4653613567352295
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 3.1427717208862305,
      "step": 1
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 1,
      "training_step_time": 9.91479229927063
    },
    {
      "epoch": 3.814697265625e-09,
      "model_forward_time": 3.1447298526763916,
      "step": 1
    },
    {
      "epoch": 3.814697265625e-09,
      "step": 1,
      "training_step_time": 10.027050256729126
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 3.1618285179138184,
      "step": 2
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 2,
      "training_step_time": 9.93385624885559
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 2.7254161834716797,
      "step": 2
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 2,
      "training_step_time": 7.754691123962402
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 2.4391725063323975,
      "step": 2
    },
    {
      "dynamics_loss_avg": 0.20924262702465057,
      "epoch": 7.62939453125e-09,
      "step": 2
    },
    {
      "action_loss_avg": 0.3283465996384621,
      "epoch": 7.62939453125e-09,
      "step": 2
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 2,
      "training_step_time": 7.471409320831299
    },
    {
      "epoch": 7.62939453125e-09,
      "model_forward_time": 3.1404318809509277,
      "step": 2
    },
    {
      "epoch": 7.62939453125e-09,
      "step": 2,
      "training_step_time": 10.021225452423096
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 3.1740899085998535,
      "step": 3
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 3,
      "training_step_time": 9.965710163116455
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 3.135784387588501,
      "step": 3
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 3,
      "training_step_time": 9.90846872329712
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 3.4377236366271973,
      "step": 3
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 3,
      "training_step_time": 10.2044358253479
    },
    {
      "epoch": 1.1444091796875e-08,
      "model_forward_time": 3.128631114959717,
      "step": 3
    },
    {
      "epoch": 1.1444091796875e-08,
      "step": 3,
      "training_step_time": 10.275131940841675
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 3.1736395359039307,
      "step": 4
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 4,
      "training_step_time": 9.94666862487793
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 2.43034291267395,
      "step": 4
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 4,
      "training_step_time": 7.454351425170898
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 2.4280035495758057,
      "step": 4
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 4,
      "training_step_time": 7.469057083129883
    },
    {
      "epoch": 1.52587890625e-08,
      "model_forward_time": 3.1313750743865967,
      "step": 4
    },
    {
      "epoch": 1.52587890625e-08,
      "step": 4,
      "training_step_time": 10.313454866409302
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 3.1519243717193604,
      "step": 5
    },
    {
      "dynamics_loss_avg": 0.22146085724234582,
      "epoch": 1.9073486328125e-08,
      "step": 5
    },
    {
      "action_loss_avg": 0.32112472504377365,
      "epoch": 1.9073486328125e-08,
      "step": 5
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 5,
      "training_step_time": 9.920408964157104
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 2.42899227142334,
      "step": 5
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 5,
      "training_step_time": 7.452818393707275
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 2.428941249847412,
      "step": 5
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 5,
      "training_step_time": 7.453892230987549
    },
    {
      "epoch": 1.9073486328125e-08,
      "model_forward_time": 2.4279162883758545,
      "step": 5
    },
    {
      "epoch": 1.9073486328125e-08,
      "step": 5,
      "training_step_time": 10.32839035987854
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 3.1929852962493896,
      "step": 6
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 6,
      "training_step_time": 9.98660683631897
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 3.491147518157959,
      "step": 6
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 6,
      "training_step_time": 10.27461552619934
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 2.4501121044158936,
      "step": 6
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 6,
      "training_step_time": 7.489748954772949
    },
    {
      "epoch": 2.288818359375e-08,
      "model_forward_time": 3.159827470779419,
      "step": 6
    },
    {
      "epoch": 2.288818359375e-08,
      "step": 6,
      "training_step_time": 10.041177034378052
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 2.512005090713501,
      "step": 7
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 7,
      "training_step_time": 7.574506998062134
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 2.461508274078369,
      "step": 7
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 7,
      "training_step_time": 7.497191667556763
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 3.4724552631378174,
      "step": 7
    },
    {
      "dynamics_loss_avg": 0.20823537260293962,
      "epoch": 2.6702880859375e-08,
      "step": 7
    },
    {
      "action_loss_avg": 0.3039559543132782,
      "epoch": 2.6702880859375e-08,
      "step": 7
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 7,
      "training_step_time": 10.251328945159912
    },
    {
      "epoch": 2.6702880859375e-08,
      "model_forward_time": 3.1593515872955322,
      "step": 7
    },
    {
      "epoch": 2.6702880859375e-08,
      "step": 7,
      "training_step_time": 10.046679973602295
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 2.4930615425109863,
      "step": 8
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 8,
      "training_step_time": 7.53291654586792
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 3.1553711891174316,
      "step": 8
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 8,
      "training_step_time": 9.928830623626709
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 3.1475138664245605,
      "step": 8
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 8,
      "training_step_time": 9.92460823059082
    },
    {
      "epoch": 3.0517578125e-08,
      "model_forward_time": 3.461642265319824,
      "step": 8
    },
    {
      "epoch": 3.0517578125e-08,
      "step": 8,
      "training_step_time": 10.328017950057983
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 2.448723554611206,
      "step": 9
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 9,
      "training_step_time": 7.473684072494507
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 3.1356730461120605,
      "step": 9
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 9,
      "training_step_time": 9.906652450561523
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 2.431495189666748,
      "step": 9
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 9,
      "training_step_time": 7.456798315048218
    },
    {
      "epoch": 3.4332275390625e-08,
      "model_forward_time": 2.42781400680542,
      "step": 9
    },
    {
      "epoch": 3.4332275390625e-08,
      "step": 9,
      "training_step_time": 10.044111490249634
    },
    {
      "epoch": 3.814697265625e-08,
      "grad_norm": 2.1581640243530273,
      "learning_rate": 1e-05,
      "loss": 0.5328,
      "step": 10
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 3.517183780670166,
      "step": 10
    },
    {
      "dynamics_loss_avg": 0.23379224985837938,
      "epoch": 3.814697265625e-08,
      "step": 10
    },
    {
      "action_loss_avg": 0.31032123863697053,
      "epoch": 3.814697265625e-08,
      "step": 10
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 10,
      "training_step_time": 10.30916953086853
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 3.1491527557373047,
      "step": 10
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 10,
      "training_step_time": 9.92830777168274
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 3.150275707244873,
      "step": 10
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 10,
      "training_step_time": 9.928660869598389
    },
    {
      "epoch": 3.814697265625e-08,
      "model_forward_time": 2.43998384475708,
      "step": 10
    },
    {
      "epoch": 3.814697265625e-08,
      "step": 10,
      "training_step_time": 10.299190759658813
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 3.147780418395996,
      "step": 11
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 11,
      "training_step_time": 9.927370309829712
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 2.7194535732269287,
      "step": 11
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 11,
      "training_step_time": 7.75121545791626
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 2.4326863288879395,
      "step": 11
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 11,
      "training_step_time": 7.457464933395386
    },
    {
      "epoch": 4.1961669921875e-08,
      "model_forward_time": 2.4305307865142822,
      "step": 11
    },
    {
      "epoch": 4.1961669921875e-08,
      "step": 11,
      "training_step_time": 10.341280221939087
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 3.177783250808716,
      "step": 12
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 12,
      "training_step_time": 9.946127653121948
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 2.4261481761932373,
      "step": 12
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 12,
      "training_step_time": 7.447435140609741
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 2.4233615398406982,
      "step": 12
    },
    {
      "dynamics_loss_avg": 0.2362516328692436,
      "epoch": 4.57763671875e-08,
      "step": 12
    },
    {
      "action_loss_avg": 0.3181458614766598,
      "epoch": 4.57763671875e-08,
      "step": 12
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 12,
      "training_step_time": 7.446556568145752
    },
    {
      "epoch": 4.57763671875e-08,
      "model_forward_time": 3.4088709354400635,
      "step": 12
    },
    {
      "epoch": 4.57763671875e-08,
      "step": 12,
      "training_step_time": 10.273874998092651
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 3.1400182247161865,
      "step": 13
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 13,
      "training_step_time": 9.915194749832153
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 2.4287610054016113,
      "step": 13
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 13,
      "training_step_time": 7.459072589874268
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 2.4321067333221436,
      "step": 13
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 13,
      "training_step_time": 7.467123508453369
    },
    {
      "epoch": 4.9591064453125e-08,
      "model_forward_time": 3.1320230960845947,
      "step": 13
    },
    {
      "epoch": 4.9591064453125e-08,
      "step": 13,
      "training_step_time": 10.332281589508057
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 3.4590251445770264,
      "step": 14
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 14,
      "training_step_time": 10.235911130905151
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 3.1403679847717285,
      "step": 14
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 14,
      "training_step_time": 9.923080921173096
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 2.433162212371826,
      "step": 14
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 14,
      "training_step_time": 7.4650421142578125
    },
    {
      "epoch": 5.340576171875e-08,
      "model_forward_time": 3.1349587440490723,
      "step": 14
    },
    {
      "epoch": 5.340576171875e-08,
      "step": 14,
      "training_step_time": 10.037785291671753
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 3.176379919052124,
      "step": 15
    },
    {
      "dynamics_loss_avg": 0.22080072164535522,
      "epoch": 5.7220458984375e-08,
      "step": 15
    },
    {
      "action_loss_avg": 0.26949807107448576,
      "epoch": 5.7220458984375e-08,
      "step": 15
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 15,
      "training_step_time": 9.95644736289978
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 3.444908857345581,
      "step": 15
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 15,
      "training_step_time": 10.217509984970093
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 2.4296374320983887,
      "step": 15
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 15,
      "training_step_time": 7.4632813930511475
    },
    {
      "epoch": 5.7220458984375e-08,
      "model_forward_time": 3.1324949264526367,
      "step": 15
    },
    {
      "epoch": 5.7220458984375e-08,
      "step": 15,
      "training_step_time": 10.02153754234314
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 2.4409031867980957,
      "step": 16
    },
    {
      "epoch": 6.103515625e-08,
      "step": 16,
      "training_step_time": 7.4643940925598145
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 3.1266701221466064,
      "step": 16
    },
    {
      "epoch": 6.103515625e-08,
      "step": 16,
      "training_step_time": 9.88867449760437
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 3.127140998840332,
      "step": 16
    },
    {
      "epoch": 6.103515625e-08,
      "step": 16,
      "training_step_time": 10.165846347808838
    },
    {
      "epoch": 6.103515625e-08,
      "model_forward_time": 2.4278032779693604,
      "step": 16
    },
    {
      "epoch": 6.103515625e-08,
      "step": 16,
      "training_step_time": 10.314647912979126
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 2.440990924835205,
      "step": 17
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 17,
      "training_step_time": 7.469157695770264
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 2.42853045463562,
      "step": 17
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 17,
      "training_step_time": 7.4551801681518555
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 2.4286255836486816,
      "step": 17
    },
    {
      "dynamics_loss_avg": 0.2448069855570793,
      "epoch": 6.4849853515625e-08,
      "step": 17
    },
    {
      "action_loss_avg": 0.29944684356451035,
      "epoch": 6.4849853515625e-08,
      "step": 17
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 17,
      "training_step_time": 7.45790433883667
    },
    {
      "epoch": 6.4849853515625e-08,
      "model_forward_time": 3.130129814147949,
      "step": 17
    },
    {
      "epoch": 6.4849853515625e-08,
      "step": 17,
      "training_step_time": 10.35996413230896
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 3.4203081130981445,
      "step": 18
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 18,
      "training_step_time": 10.202938079833984
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 3.1377670764923096,
      "step": 18
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 18,
      "training_step_time": 9.919862985610962
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 3.149660587310791,
      "step": 18
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 18,
      "training_step_time": 9.933400392532349
    },
    {
      "epoch": 6.866455078125e-08,
      "model_forward_time": 3.1396007537841797,
      "step": 18
    },
    {
      "epoch": 6.866455078125e-08,
      "step": 18,
      "training_step_time": 10.042677879333496
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 3.150395393371582,
      "step": 19
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 19,
      "training_step_time": 9.93287992477417
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 3.412019729614258,
      "step": 19
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 19,
      "training_step_time": 10.188277006149292
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 2.4306633472442627,
      "step": 19
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 19,
      "training_step_time": 7.461437225341797
    },
    {
      "epoch": 7.2479248046875e-08,
      "model_forward_time": 2.429857015609741,
      "step": 19
    },
    {
      "epoch": 7.2479248046875e-08,
      "step": 19,
      "training_step_time": 10.03954529762268
    },
    {
      "epoch": 7.62939453125e-08,
      "grad_norm": 2.18648624420166,
      "learning_rate": 1e-05,
      "loss": 0.5189,
      "step": 20
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 2.461637496948242,
      "step": 20
    },
    {
      "dynamics_loss_avg": 0.21553153693675994,
      "epoch": 7.62939453125e-08,
      "step": 20
    },
    {
      "action_loss_avg": 0.24965579472482205,
      "epoch": 7.62939453125e-08,
      "step": 20
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 20,
      "training_step_time": 7.506435871124268
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 3.142763614654541,
      "step": 20
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 20,
      "training_step_time": 9.918967008590698
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 3.4593143463134766,
      "step": 20
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 20,
      "training_step_time": 10.231177568435669
    },
    {
      "epoch": 7.62939453125e-08,
      "model_forward_time": 3.147707462310791,
      "step": 20
    },
    {
      "epoch": 7.62939453125e-08,
      "step": 20,
      "training_step_time": 10.290237665176392
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 2.4601213932037354,
      "step": 21
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 21,
      "training_step_time": 7.498241186141968
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 3.1502532958984375,
      "step": 21
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 21,
      "training_step_time": 9.926616907119751
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 3.1484289169311523,
      "step": 21
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 21,
      "training_step_time": 9.92476773262024
    },
    {
      "epoch": 8.0108642578125e-08,
      "model_forward_time": 2.7448744773864746,
      "step": 21
    },
    {
      "epoch": 8.0108642578125e-08,
      "step": 21,
      "training_step_time": 10.036815166473389
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 3.1733431816101074,
      "step": 22
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 22,
      "training_step_time": 9.943490266799927
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 3.1412088871002197,
      "step": 22
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 22,
      "training_step_time": 9.920169353485107
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 3.1349456310272217,
      "step": 22
    },
    {
      "dynamics_loss_avg": 0.2517733260989189,
      "epoch": 8.392333984375e-08,
      "step": 22
    },
    {
      "action_loss_avg": 0.3019081562757492,
      "epoch": 8.392333984375e-08,
      "step": 22
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 22,
      "training_step_time": 9.913517475128174
    },
    {
      "epoch": 8.392333984375e-08,
      "model_forward_time": 3.146225929260254,
      "step": 22
    },
    {
      "epoch": 8.392333984375e-08,
      "step": 22,
      "training_step_time": 10.341853141784668
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 3.431525945663452,
      "step": 23
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 23,
      "training_step_time": 10.207837581634521
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 3.133427143096924,
      "step": 23
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 23,
      "training_step_time": 9.91476559638977
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 3.138732433319092,
      "step": 23
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 23,
      "training_step_time": 9.919564008712769
    },
    {
      "epoch": 8.7738037109375e-08,
      "model_forward_time": 2.438131809234619,
      "step": 23
    },
    {
      "epoch": 8.7738037109375e-08,
      "step": 23,
      "training_step_time": 10.044329404830933
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 2.4790854454040527,
      "step": 24
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 24,
      "training_step_time": 7.516199350357056
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 3.458341360092163,
      "step": 24
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 24,
      "training_step_time": 10.231858015060425
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 3.1354312896728516,
      "step": 24
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 24,
      "training_step_time": 9.912609577178955
    },
    {
      "epoch": 9.1552734375e-08,
      "model_forward_time": 3.1383016109466553,
      "step": 24
    },
    {
      "epoch": 9.1552734375e-08,
      "step": 24,
      "training_step_time": 10.039297580718994
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 3.163782835006714,
      "step": 25
    },
    {
      "dynamics_loss_avg": 0.20700162574648856,
      "epoch": 9.5367431640625e-08,
      "step": 25
    },
    {
      "action_loss_avg": 0.2344718359410763,
      "epoch": 9.5367431640625e-08,
      "step": 25
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 25,
      "training_step_time": 9.938891887664795
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 3.1359879970550537,
      "step": 25
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 25,
      "training_step_time": 9.908609390258789
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 3.4200167655944824,
      "step": 25
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 25,
      "training_step_time": 10.206464529037476
    },
    {
      "epoch": 9.5367431640625e-08,
      "model_forward_time": 3.1359686851501465,
      "step": 25
    },
    {
      "epoch": 9.5367431640625e-08,
      "step": 25,
      "training_step_time": 10.034545421600342
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 3.149956464767456,
      "step": 26
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 26,
      "training_step_time": 9.932921886444092
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 2.4279990196228027,
      "step": 26
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 26,
      "training_step_time": 7.4618542194366455
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 2.4395740032196045,
      "step": 26
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 26,
      "training_step_time": 7.478419303894043
    },
    {
      "epoch": 9.918212890625e-08,
      "model_forward_time": 3.4621424674987793,
      "step": 26
    },
    {
      "epoch": 9.918212890625e-08,
      "step": 26,
      "training_step_time": 10.341823101043701
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 2.44826602935791,
      "step": 27
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 27,
      "training_step_time": 7.48429012298584
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 3.1338584423065186,
      "step": 27
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 27,
      "training_step_time": 9.911560297012329
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 2.4360508918762207,
      "step": 27
    },
    {
      "dynamics_loss_avg": 0.2322405681014061,
      "epoch": 1.02996826171875e-07,
      "step": 27
    },
    {
      "action_loss_avg": 0.30652413293719294,
      "epoch": 1.02996826171875e-07,
      "step": 27
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 27,
      "training_step_time": 7.474137306213379
    },
    {
      "epoch": 1.02996826171875e-07,
      "model_forward_time": 3.132394313812256,
      "step": 27
    },
    {
      "epoch": 1.02996826171875e-07,
      "step": 27,
      "training_step_time": 10.32923173904419
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 3.4939470291137695,
      "step": 28
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 28,
      "training_step_time": 10.27503228187561
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 3.1317317485809326,
      "step": 28
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 28,
      "training_step_time": 9.909776449203491
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 3.1381711959838867,
      "step": 28
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 28,
      "training_step_time": 9.914607524871826
    },
    {
      "epoch": 1.068115234375e-07,
      "model_forward_time": 3.1373097896575928,
      "step": 28
    },
    {
      "epoch": 1.068115234375e-07,
      "step": 28,
      "training_step_time": 10.303589582443237
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 2.4589388370513916,
      "step": 29
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 29,
      "training_step_time": 7.490837812423706
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 2.429882287979126,
      "step": 29
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 29,
      "training_step_time": 7.46039080619812
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 3.453531265258789,
      "step": 29
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 29,
      "training_step_time": 10.229482889175415
    },
    {
      "epoch": 1.10626220703125e-07,
      "model_forward_time": 3.130387544631958,
      "step": 29
    },
    {
      "epoch": 1.10626220703125e-07,
      "step": 29,
      "training_step_time": 10.04332947731018
    },
    {
      "epoch": 1.1444091796875e-07,
      "grad_norm": 1.2197703123092651,
      "learning_rate": 1e-05,
      "loss": 0.4916,
      "step": 30
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 1.760732889175415,
      "step": 30
    },
    {
      "dynamics_loss_avg": 0.22124346643686293,
      "epoch": 1.1444091796875e-07,
      "step": 30
    },
    {
      "action_loss_avg": 0.2728277489542961,
      "epoch": 1.1444091796875e-07,
      "step": 30
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 30,
      "training_step_time": 5.226933240890503
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 2.437622308731079,
      "step": 30
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 30,
      "training_step_time": 7.469906806945801
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 2.4310359954833984,
      "step": 30
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 30,
      "training_step_time": 7.467575788497925
    },
    {
      "epoch": 1.1444091796875e-07,
      "model_forward_time": 2.741163730621338,
      "step": 30
    },
    {
      "epoch": 1.1444091796875e-07,
      "step": 30,
      "training_step_time": 10.054308652877808
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 2.4654171466827393,
      "step": 31
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 31,
      "training_step_time": 7.5002312660217285
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 3.1370410919189453,
      "step": 31
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 31,
      "training_step_time": 9.919159650802612
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 3.139584541320801,
      "step": 31
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 31,
      "training_step_time": 9.928280591964722
    },
    {
      "epoch": 1.18255615234375e-07,
      "model_forward_time": 3.140996217727661,
      "step": 31
    },
    {
      "epoch": 1.18255615234375e-07,
      "step": 31,
      "training_step_time": 10.273021221160889
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 2.7247211933135986,
      "step": 32
    },
    {
      "epoch": 1.220703125e-07,
      "step": 32,
      "training_step_time": 7.761104345321655
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 3.1373047828674316,
      "step": 32
    },
    {
      "epoch": 1.220703125e-07,
      "step": 32,
      "training_step_time": 9.922017097473145
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 2.434598922729492,
      "step": 32
    },
    {
      "dynamics_loss_avg": 0.24447987377643585,
      "epoch": 1.220703125e-07,
      "step": 32
    },
    {
      "action_loss_avg": 0.2746193841099739,
      "epoch": 1.220703125e-07,
      "step": 32
    },
    {
      "epoch": 1.220703125e-07,
      "step": 32,
      "training_step_time": 7.4846367835998535
    },
    {
      "epoch": 1.220703125e-07,
      "model_forward_time": 3.1384994983673096,
      "step": 32
    },
    {
      "epoch": 1.220703125e-07,
      "step": 32,
      "training_step_time": 10.338974475860596
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 3.150224208831787,
      "step": 33
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 33,
      "training_step_time": 9.929105997085571
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 3.135296106338501,
      "step": 33
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 33,
      "training_step_time": 10.226794242858887
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 3.1302707195281982,
      "step": 33
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 33,
      "training_step_time": 9.904296398162842
    },
    {
      "epoch": 1.25885009765625e-07,
      "model_forward_time": 1.7450079917907715,
      "step": 33
    },
    {
      "epoch": 1.25885009765625e-07,
      "step": 33,
      "training_step_time": 10.325219631195068
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 3.1422884464263916,
      "step": 34
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 34,
      "training_step_time": 9.916692018508911
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 2.431318521499634,
      "step": 34
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 34,
      "training_step_time": 7.4589948654174805
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 3.1356754302978516,
      "step": 34
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 34,
      "training_step_time": 9.909645557403564
    },
    {
      "epoch": 1.2969970703125e-07,
      "model_forward_time": 2.6807377338409424,
      "step": 34
    },
    {
      "epoch": 1.2969970703125e-07,
      "step": 34,
      "training_step_time": 10.049567937850952
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 2.4612786769866943,
      "step": 35
    },
    {
      "dynamics_loss_avg": 0.17783667296171188,
      "epoch": 1.33514404296875e-07,
      "step": 35
    },
    {
      "action_loss_avg": 0.19623069800436496,
      "epoch": 1.33514404296875e-07,
      "step": 35
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 35,
      "training_step_time": 7.498314142227173
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 2.43489408493042,
      "step": 35
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 35,
      "training_step_time": 7.468815803527832
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 3.135560989379883,
      "step": 35
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 35,
      "training_step_time": 9.914499998092651
    },
    {
      "epoch": 1.33514404296875e-07,
      "model_forward_time": 3.1397464275360107,
      "step": 35
    },
    {
      "epoch": 1.33514404296875e-07,
      "step": 35,
      "training_step_time": 10.044018507003784
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 3.6188225746154785,
      "step": 36
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 36,
      "training_step_time": 10.415419101715088
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 3.1597635746002197,
      "step": 36
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 36,
      "training_step_time": 9.947967290878296
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 2.4514224529266357,
      "step": 36
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 36,
      "training_step_time": 7.482989311218262
    },
    {
      "epoch": 1.373291015625e-07,
      "model_forward_time": 2.4433016777038574,
      "step": 36
    },
    {
      "epoch": 1.373291015625e-07,
      "step": 36,
      "training_step_time": 10.03799057006836
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 2.455742120742798,
      "step": 37
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 37,
      "training_step_time": 7.483600378036499
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 2.7739105224609375,
      "step": 37
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 37,
      "training_step_time": 7.80784797668457
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 2.4273335933685303,
      "step": 37
    },
    {
      "dynamics_loss_avg": 0.1990236334502697,
      "epoch": 1.41143798828125e-07,
      "step": 37
    },
    {
      "action_loss_avg": 0.21045810654759406,
      "epoch": 1.41143798828125e-07,
      "step": 37
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 37,
      "training_step_time": 7.458503007888794
    },
    {
      "epoch": 1.41143798828125e-07,
      "model_forward_time": 2.4328863620758057,
      "step": 37
    },
    {
      "epoch": 1.41143798828125e-07,
      "step": 37,
      "training_step_time": 10.030829906463623
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 2.445667028427124,
      "step": 38
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 38,
      "training_step_time": 7.473156213760376
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 2.4319677352905273,
      "step": 38
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 38,
      "training_step_time": 7.46039605140686
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 3.133554220199585,
      "step": 38
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 38,
      "training_step_time": 9.907777786254883
    },
    {
      "epoch": 1.4495849609375e-07,
      "model_forward_time": 2.731987237930298,
      "step": 38
    },
    {
      "epoch": 1.4495849609375e-07,
      "step": 38,
      "training_step_time": 10.340924263000488
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 1.7848327159881592,
      "step": 39
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 39,
      "training_step_time": 5.241760969161987
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 2.431220293045044,
      "step": 39
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 39,
      "training_step_time": 7.457813501358032
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 3.1321046352386475,
      "step": 39
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 39,
      "training_step_time": 9.911064863204956
    },
    {
      "epoch": 1.48773193359375e-07,
      "model_forward_time": 2.436189889907837,
      "step": 39
    },
    {
      "epoch": 1.48773193359375e-07,
      "step": 39,
      "training_step_time": 10.055972337722778
    },
    {
      "epoch": 1.52587890625e-07,
      "grad_norm": 1.7350033521652222,
      "learning_rate": 1e-05,
      "loss": 0.4471,
      "step": 40
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 2.4553062915802,
      "step": 40
    },
    {
      "dynamics_loss_avg": 0.20083781853318214,
      "epoch": 1.52587890625e-07,
      "step": 40
    },
    {
      "action_loss_avg": 0.2256111465394497,
      "epoch": 1.52587890625e-07,
      "step": 40
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 40,
      "training_step_time": 7.496402740478516
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 3.4383902549743652,
      "step": 40
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 40,
      "training_step_time": 10.210975408554077
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 2.438098430633545,
      "step": 40
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 40,
      "training_step_time": 7.471924781799316
    },
    {
      "epoch": 1.52587890625e-07,
      "model_forward_time": 2.432271957397461,
      "step": 40
    },
    {
      "epoch": 1.52587890625e-07,
      "step": 40,
      "training_step_time": 10.049321174621582
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 3.1444759368896484,
      "step": 41
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 41,
      "training_step_time": 9.9232017993927
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 2.4364495277404785,
      "step": 41
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 41,
      "training_step_time": 7.466695547103882
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 2.716932535171509,
      "step": 41
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 41,
      "training_step_time": 7.745768070220947
    },
    {
      "epoch": 1.56402587890625e-07,
      "model_forward_time": 3.1336395740509033,
      "step": 41
    },
    {
      "epoch": 1.56402587890625e-07,
      "step": 41,
      "training_step_time": 10.040921449661255
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 2.477130889892578,
      "step": 42
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 42,
      "training_step_time": 7.5091071128845215
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 2.43388032913208,
      "step": 42
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 42,
      "training_step_time": 7.4645280838012695
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 3.1386301517486572,
      "step": 42
    },
    {
      "dynamics_loss_avg": 0.21338179484009742,
      "epoch": 1.6021728515625e-07,
      "step": 42
    },
    {
      "action_loss_avg": 0.2625331461429596,
      "epoch": 1.6021728515625e-07,
      "step": 42
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 42,
      "training_step_time": 9.922704458236694
    },
    {
      "epoch": 1.6021728515625e-07,
      "model_forward_time": 3.4180257320404053,
      "step": 42
    },
    {
      "epoch": 1.6021728515625e-07,
      "step": 42,
      "training_step_time": 10.300044775009155
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 2.451876640319824,
      "step": 43
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 43,
      "training_step_time": 7.487983465194702
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 2.4397289752960205,
      "step": 43
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 43,
      "training_step_time": 7.476497411727905
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 3.142178773880005,
      "step": 43
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 43,
      "training_step_time": 9.927038669586182
    },
    {
      "epoch": 1.64031982421875e-07,
      "model_forward_time": 2.4456210136413574,
      "step": 43
    },
    {
      "epoch": 1.64031982421875e-07,
      "step": 43,
      "training_step_time": 10.37391448020935
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 3.164247512817383,
      "step": 44
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 44,
      "training_step_time": 9.946375131607056
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 3.4147145748138428,
      "step": 44
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 44,
      "training_step_time": 10.202686309814453
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 3.148918390274048,
      "step": 44
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 44,
      "training_step_time": 9.9425368309021
    },
    {
      "epoch": 1.678466796875e-07,
      "model_forward_time": 2.439815044403076,
      "step": 44
    },
    {
      "epoch": 1.678466796875e-07,
      "step": 44,
      "training_step_time": 10.352099895477295
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 3.1743032932281494,
      "step": 45
    },
    {
      "dynamics_loss_avg": 0.22762020975351333,
      "epoch": 1.71661376953125e-07,
      "step": 45
    },
    {
      "action_loss_avg": 0.23687051609158516,
      "epoch": 1.71661376953125e-07,
      "step": 45
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 45,
      "training_step_time": 9.962738275527954
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 3.1452434062957764,
      "step": 45
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 45,
      "training_step_time": 9.93509292602539
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 3.450561761856079,
      "step": 45
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 45,
      "training_step_time": 10.234141111373901
    },
    {
      "epoch": 1.71661376953125e-07,
      "model_forward_time": 2.4349312782287598,
      "step": 45
    },
    {
      "epoch": 1.71661376953125e-07,
      "step": 45,
      "training_step_time": 10.049264907836914
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 3.1603429317474365,
      "step": 46
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 46,
      "training_step_time": 9.94087839126587
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 3.144296169281006,
      "step": 46
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 46,
      "training_step_time": 9.924448728561401
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 2.4403533935546875,
      "step": 46
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 46,
      "training_step_time": 7.47680926322937
    },
    {
      "epoch": 1.7547607421875e-07,
      "model_forward_time": 3.4257988929748535,
      "step": 46
    },
    {
      "epoch": 1.7547607421875e-07,
      "step": 46,
      "training_step_time": 10.307112455368042
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 3.1605465412139893,
      "step": 47
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 47,
      "training_step_time": 9.93541669845581
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 3.143967628479004,
      "step": 47
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 47,
      "training_step_time": 9.923530340194702
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 2.437196731567383,
      "step": 47
    },
    {
      "dynamics_loss_avg": 0.2037894655019045,
      "epoch": 1.79290771484375e-07,
      "step": 47
    },
    {
      "action_loss_avg": 0.23363168798387052,
      "epoch": 1.79290771484375e-07,
      "step": 47
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 47,
      "training_step_time": 7.471617698669434
    },
    {
      "epoch": 1.79290771484375e-07,
      "model_forward_time": 3.143599033355713,
      "step": 47
    },
    {
      "epoch": 1.79290771484375e-07,
      "step": 47,
      "training_step_time": 10.329203128814697
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 3.477004051208496,
      "step": 48
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 48,
      "training_step_time": 10.25605034828186
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 2.43880033493042,
      "step": 48
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 48,
      "training_step_time": 7.470978498458862
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 3.139824628829956,
      "step": 48
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 48,
      "training_step_time": 9.913930177688599
    },
    {
      "epoch": 1.8310546875e-07,
      "model_forward_time": 3.141634702682495,
      "step": 48
    },
    {
      "epoch": 1.8310546875e-07,
      "step": 48,
      "training_step_time": 10.308823347091675
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 3.1577866077423096,
      "step": 49
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 49,
      "training_step_time": 9.937919855117798
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 2.7209882736206055,
      "step": 49
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 49,
      "training_step_time": 7.74936318397522
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 3.1361336708068848,
      "step": 49
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 49,
      "training_step_time": 9.910929918289185
    },
    {
      "epoch": 1.86920166015625e-07,
      "model_forward_time": 2.4361348152160645,
      "step": 49
    },
    {
      "epoch": 1.86920166015625e-07,
      "step": 49,
      "training_step_time": 10.327344417572021
    },
    {
      "epoch": 1.9073486328125e-07,
      "grad_norm": 0.957172691822052,
      "learning_rate": 1e-05,
      "loss": 0.4603,
      "step": 50
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 3.145453453063965,
      "step": 50
    },
    {
      "dynamics_loss_avg": 0.21909846514463424,
      "epoch": 1.9073486328125e-07,
      "step": 50
    },
    {
      "action_loss_avg": 0.26254491358995435,
      "epoch": 1.9073486328125e-07,
      "step": 50
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 50,
      "training_step_time": 9.910602331161499
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 3.1332106590270996,
      "step": 50
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 50,
      "training_step_time": 9.9026038646698
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 2.681858539581299,
      "step": 50
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 50,
      "training_step_time": 7.705631971359253
    },
    {
      "epoch": 1.9073486328125e-07,
      "model_forward_time": 2.431098222732544,
      "step": 50
    },
    {
      "epoch": 1.9073486328125e-07,
      "step": 50,
      "training_step_time": 10.027663707733154
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 2.4485039710998535,
      "step": 51
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 51,
      "training_step_time": 7.482080459594727
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 3.131605625152588,
      "step": 51
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 51,
      "training_step_time": 9.897231578826904
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 3.1364521980285645,
      "step": 51
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 51,
      "training_step_time": 9.90694785118103
    },
    {
      "epoch": 1.94549560546875e-07,
      "model_forward_time": 2.7110986709594727,
      "step": 51
    },
    {
      "epoch": 1.94549560546875e-07,
      "step": 51,
      "training_step_time": 10.03237247467041
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 3.150153160095215,
      "step": 52
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 52,
      "training_step_time": 9.91957688331604
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 2.436736822128296,
      "step": 52
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 52,
      "training_step_time": 7.469385147094727
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 3.1410322189331055,
      "step": 52
    },
    {
      "dynamics_loss_avg": 0.18396452888846399,
      "epoch": 1.983642578125e-07,
      "step": 52
    },
    {
      "action_loss_avg": 0.2121353939175606,
      "epoch": 1.983642578125e-07,
      "step": 52
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 52,
      "training_step_time": 9.916197299957275
    },
    {
      "epoch": 1.983642578125e-07,
      "model_forward_time": 2.4354968070983887,
      "step": 52
    },
    {
      "epoch": 1.983642578125e-07,
      "step": 52,
      "training_step_time": 10.26194167137146
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 3.148634910583496,
      "step": 53
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 53,
      "training_step_time": 10.23145341873169
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 2.4417669773101807,
      "step": 53
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 53,
      "training_step_time": 7.469677686691284
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 2.4351139068603516,
      "step": 53
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 53,
      "training_step_time": 7.463704586029053
    },
    {
      "epoch": 2.02178955078125e-07,
      "model_forward_time": 3.1355159282684326,
      "step": 53
    },
    {
      "epoch": 2.02178955078125e-07,
      "step": 53,
      "training_step_time": 10.317893028259277
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 3.1593925952911377,
      "step": 54
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 54,
      "training_step_time": 9.9297194480896
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 2.434499502182007,
      "step": 54
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 54,
      "training_step_time": 7.4629223346710205
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 3.4128997325897217,
      "step": 54
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 54,
      "training_step_time": 10.187632083892822
    },
    {
      "epoch": 2.0599365234375e-07,
      "model_forward_time": 2.4395580291748047,
      "step": 54
    },
    {
      "epoch": 2.0599365234375e-07,
      "step": 54,
      "training_step_time": 10.338985919952393
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 2.4629557132720947,
      "step": 55
    },
    {
      "dynamics_loss_avg": 0.2083815410733223,
      "epoch": 2.09808349609375e-07,
      "step": 55
    },
    {
      "action_loss_avg": 0.25150192752480505,
      "epoch": 2.09808349609375e-07,
      "step": 55
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 55,
      "training_step_time": 7.497665882110596
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 3.142448663711548,
      "step": 55
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 55,
      "training_step_time": 9.917343616485596
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 2.439624309539795,
      "step": 55
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 55,
      "training_step_time": 7.47277307510376
    },
    {
      "epoch": 2.09808349609375e-07,
      "model_forward_time": 2.749072551727295,
      "step": 55
    },
    {
      "epoch": 2.09808349609375e-07,
      "step": 55,
      "training_step_time": 10.358345746994019
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 3.157039165496826,
      "step": 56
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 56,
      "training_step_time": 9.92881178855896
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 2.4366676807403564,
      "step": 56
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 56,
      "training_step_time": 7.463369607925415
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 2.432304859161377,
      "step": 56
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 56,
      "training_step_time": 7.475404500961304
    },
    {
      "epoch": 2.13623046875e-07,
      "model_forward_time": 3.1315228939056396,
      "step": 56
    },
    {
      "epoch": 2.13623046875e-07,
      "step": 56,
      "training_step_time": 10.042516946792603
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 3.4254534244537354,
      "step": 57
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 57,
      "training_step_time": 10.196812629699707
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 3.1376309394836426,
      "step": 57
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 57,
      "training_step_time": 9.909664869308472
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 3.1389682292938232,
      "step": 57
    },
    {
      "dynamics_loss_avg": 0.18313016816973687,
      "epoch": 2.17437744140625e-07,
      "step": 57
    },
    {
      "action_loss_avg": 0.20374237596988679,
      "epoch": 2.17437744140625e-07,
      "step": 57
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 57,
      "training_step_time": 9.91386604309082
    },
    {
      "epoch": 2.17437744140625e-07,
      "model_forward_time": 3.1349406242370605,
      "step": 57
    },
    {
      "epoch": 2.17437744140625e-07,
      "step": 57,
      "training_step_time": 10.314894199371338
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 2.444260597229004,
      "step": 58
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 58,
      "training_step_time": 7.476330757141113
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 3.4237582683563232,
      "step": 58
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 58,
      "training_step_time": 10.19640064239502
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 2.43096661567688,
      "step": 58
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 58,
      "training_step_time": 7.461297512054443
    },
    {
      "epoch": 2.2125244140625e-07,
      "model_forward_time": 2.428647994995117,
      "step": 58
    },
    {
      "epoch": 2.2125244140625e-07,
      "step": 58,
      "training_step_time": 10.312876224517822
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 3.197636842727661,
      "step": 59
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 59,
      "training_step_time": 9.993944644927979
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 3.1526999473571777,
      "step": 59
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 59,
      "training_step_time": 9.927683591842651
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 2.43754506111145,
      "step": 59
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 59,
      "training_step_time": 7.468944311141968
    },
    {
      "epoch": 2.25067138671875e-07,
      "model_forward_time": 3.4609286785125732,
      "step": 59
    },
    {
      "epoch": 2.25067138671875e-07,
      "step": 59,
      "training_step_time": 10.337916612625122
    },
    {
      "epoch": 2.288818359375e-07,
      "grad_norm": 0.865514874458313,
      "learning_rate": 1e-05,
      "loss": 0.3994,
      "step": 60
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 2.445075750350952,
      "step": 60
    },
    {
      "dynamics_loss_avg": 0.16538773477077484,
      "epoch": 2.288818359375e-07,
      "step": 60
    },
    {
      "action_loss_avg": 0.1668723702430725,
      "epoch": 2.288818359375e-07,
      "step": 60
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 60,
      "training_step_time": 7.473145961761475
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 2.4344139099121094,
      "step": 60
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 60,
      "training_step_time": 7.459666967391968
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 2.4338254928588867,
      "step": 60
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 60,
      "training_step_time": 7.454775333404541
    },
    {
      "epoch": 2.288818359375e-07,
      "model_forward_time": 3.139650583267212,
      "step": 60
    },
    {
      "epoch": 2.288818359375e-07,
      "step": 60,
      "training_step_time": 10.311625003814697
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 2.7871499061584473,
      "step": 61
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 61,
      "training_step_time": 7.813512802124023
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 3.136413335800171,
      "step": 61
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 61,
      "training_step_time": 9.90621566772461
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 2.4362313747406006,
      "step": 61
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 61,
      "training_step_time": 7.465237617492676
    },
    {
      "epoch": 2.32696533203125e-07,
      "model_forward_time": 2.437804698944092,
      "step": 61
    },
    {
      "epoch": 2.32696533203125e-07,
      "step": 61,
      "training_step_time": 10.065124034881592
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 3.179208755493164,
      "step": 62
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 62,
      "training_step_time": 9.958590269088745
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 3.1531906127929688,
      "step": 62
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 62,
      "training_step_time": 10.233402252197266
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 2.4366605281829834,
      "step": 62
    },
    {
      "dynamics_loss_avg": 0.2145503766834736,
      "epoch": 2.3651123046875e-07,
      "step": 62
    },
    {
      "action_loss_avg": 0.2787995673716068,
      "epoch": 2.3651123046875e-07,
      "step": 62
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 62,
      "training_step_time": 7.46784520149231
    },
    {
      "epoch": 2.3651123046875e-07,
      "model_forward_time": 3.149031639099121,
      "step": 62
    },
    {
      "epoch": 2.3651123046875e-07,
      "step": 62,
      "training_step_time": 10.316696882247925
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 2.49231219291687,
      "step": 63
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 63,
      "training_step_time": 7.534103155136108
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 2.4337475299835205,
      "step": 63
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 63,
      "training_step_time": 7.461963891983032
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 3.153776168823242,
      "step": 63
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 63,
      "training_step_time": 9.925334215164185
    },
    {
      "epoch": 2.40325927734375e-07,
      "model_forward_time": 3.4406626224517822,
      "step": 63
    },
    {
      "epoch": 2.40325927734375e-07,
      "step": 63,
      "training_step_time": 10.308125734329224
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 3.1670331954956055,
      "step": 64
    },
    {
      "epoch": 2.44140625e-07,
      "step": 64,
      "training_step_time": 9.940595149993896
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 2.43503737449646,
      "step": 64
    },
    {
      "epoch": 2.44140625e-07,
      "step": 64,
      "training_step_time": 7.4625794887542725
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 2.436262369155884,
      "step": 64
    },
    {
      "epoch": 2.44140625e-07,
      "step": 64,
      "training_step_time": 7.47142219543457
    },
    {
      "epoch": 2.44140625e-07,
      "model_forward_time": 3.143217086791992,
      "step": 64
    },
    {
      "epoch": 2.44140625e-07,
      "step": 64,
      "training_step_time": 10.04015851020813
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 2.7886950969696045,
      "step": 65
    },
    {
      "dynamics_loss_avg": 0.17958347126841545,
      "epoch": 2.47955322265625e-07,
      "step": 65
    },
    {
      "action_loss_avg": 0.20017789229750632,
      "epoch": 2.47955322265625e-07,
      "step": 65
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 65,
      "training_step_time": 7.819414377212524
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 3.139538526535034,
      "step": 65
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 65,
      "training_step_time": 9.912374496459961
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 2.436854124069214,
      "step": 65
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 65,
      "training_step_time": 7.46891713142395
    },
    {
      "epoch": 2.47955322265625e-07,
      "model_forward_time": 3.148869514465332,
      "step": 65
    },
    {
      "epoch": 2.47955322265625e-07,
      "step": 65,
      "training_step_time": 10.3025643825531
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 2.466787815093994,
      "step": 66
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 66,
      "training_step_time": 7.502827167510986
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 3.445063352584839,
      "step": 66
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 66,
      "training_step_time": 10.216951847076416
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 2.4335639476776123,
      "step": 66
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 66,
      "training_step_time": 7.458642482757568
    },
    {
      "epoch": 2.5177001953125e-07,
      "model_forward_time": 1.740741491317749,
      "step": 66
    },
    {
      "epoch": 2.5177001953125e-07,
      "step": 66,
      "training_step_time": 10.305394649505615
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 2.466352939605713,
      "step": 67
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 67,
      "training_step_time": 7.487763404846191
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 1.737194299697876,
      "step": 67
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 67,
      "training_step_time": 5.184188365936279
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 3.126882553100586,
      "step": 67
    },
    {
      "dynamics_loss_avg": 0.19116596579551698,
      "epoch": 2.55584716796875e-07,
      "step": 67
    },
    {
      "action_loss_avg": 0.2231720507144928,
      "epoch": 2.55584716796875e-07,
      "step": 67
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 67,
      "training_step_time": 9.895791053771973
    },
    {
      "epoch": 2.55584716796875e-07,
      "model_forward_time": 2.7351696491241455,
      "step": 67
    },
    {
      "epoch": 2.55584716796875e-07,
      "step": 67,
      "training_step_time": 10.267893552780151
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 3.174621105194092,
      "step": 68
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 68,
      "training_step_time": 9.943830728530884
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 3.127126932144165,
      "step": 68
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 68,
      "training_step_time": 9.895349264144897
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 3.1314845085144043,
      "step": 68
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 68,
      "training_step_time": 9.899760007858276
    },
    {
      "epoch": 2.593994140625e-07,
      "model_forward_time": 2.4284613132476807,
      "step": 68
    },
    {
      "epoch": 2.593994140625e-07,
      "step": 68,
      "training_step_time": 10.028730154037476
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 2.7473928928375244,
      "step": 69
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 69,
      "training_step_time": 7.7742719650268555
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 2.424581289291382,
      "step": 69
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 69,
      "training_step_time": 7.448877334594727
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 2.4267051219940186,
      "step": 69
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 69,
      "training_step_time": 7.445415496826172
    },
    {
      "epoch": 2.63214111328125e-07,
      "model_forward_time": 2.426344871520996,
      "step": 69
    },
    {
      "epoch": 2.63214111328125e-07,
      "step": 69,
      "training_step_time": 10.022793292999268
    },
    {
      "epoch": 2.6702880859375e-07,
      "grad_norm": 0.9208059906959534,
      "learning_rate": 1e-05,
      "loss": 0.4031,
      "step": 70
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 2.4426653385162354,
      "step": 70
    },
    {
      "dynamics_loss_avg": 0.1831670045852661,
      "epoch": 2.6702880859375e-07,
      "step": 70
    },
    {
      "action_loss_avg": 0.20754778757691383,
      "epoch": 2.6702880859375e-07,
      "step": 70
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 70,
      "training_step_time": 7.463689565658569
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 3.4191651344299316,
      "step": 70
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 70,
      "training_step_time": 10.181771516799927
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 2.425001859664917,
      "step": 70
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 70,
      "training_step_time": 7.446983337402344
    },
    {
      "epoch": 2.6702880859375e-07,
      "model_forward_time": 2.4235870838165283,
      "step": 70
    },
    {
      "epoch": 2.6702880859375e-07,
      "step": 70,
      "training_step_time": 10.03272795677185
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 3.1628098487854004,
      "step": 71
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 71,
      "training_step_time": 9.930585622787476
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 2.4298717975616455,
      "step": 71
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 71,
      "training_step_time": 7.4644999504089355
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 3.1258349418640137,
      "step": 71
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 71,
      "training_step_time": 9.910987615585327
    },
    {
      "epoch": 2.70843505859375e-07,
      "model_forward_time": 3.3720943927764893,
      "step": 71
    },
    {
      "epoch": 2.70843505859375e-07,
      "step": 71,
      "training_step_time": 10.32460355758667
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 3.1449134349823,
      "step": 72
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 72,
      "training_step_time": 9.922470092773438
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 3.1267333030700684,
      "step": 72
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 72,
      "training_step_time": 9.90114974975586
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 2.431279182434082,
      "step": 72
    },
    {
      "dynamics_loss_avg": 0.1609728828072548,
      "epoch": 2.74658203125e-07,
      "step": 72
    },
    {
      "action_loss_avg": 0.16622658148407937,
      "epoch": 2.74658203125e-07,
      "step": 72
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 72,
      "training_step_time": 7.468315124511719
    },
    {
      "epoch": 2.74658203125e-07,
      "model_forward_time": 2.4323019981384277,
      "step": 72
    },
    {
      "epoch": 2.74658203125e-07,
      "step": 72,
      "training_step_time": 10.040077686309814
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 2.0741634368896484,
      "step": 73
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 73,
      "training_step_time": 5.533454418182373
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 2.4290883541107178,
      "step": 73
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 73,
      "training_step_time": 7.463953733444214
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 2.428757429122925,
      "step": 73
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 73,
      "training_step_time": 7.460983991622925
    },
    {
      "epoch": 2.78472900390625e-07,
      "model_forward_time": 3.1365301609039307,
      "step": 73
    },
    {
      "epoch": 2.78472900390625e-07,
      "step": 73,
      "training_step_time": 10.308748006820679
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 2.4504623413085938,
      "step": 74
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 74,
      "training_step_time": 7.500368356704712
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 1.754927396774292,
      "step": 74
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 74,
      "training_step_time": 5.222405195236206
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 2.760861873626709,
      "step": 74
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 74,
      "training_step_time": 7.796726703643799
    },
    {
      "epoch": 2.8228759765625e-07,
      "model_forward_time": 2.432866096496582,
      "step": 74
    },
    {
      "epoch": 2.8228759765625e-07,
      "step": 74,
      "training_step_time": 10.049538612365723
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 3.1677262783050537,
      "step": 75
    },
    {
      "dynamics_loss_avg": 0.187725293636322,
      "epoch": 2.86102294921875e-07,
      "step": 75
    },
    {
      "action_loss_avg": 0.20990324281156064,
      "epoch": 2.86102294921875e-07,
      "step": 75
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 75,
      "training_step_time": 9.961315631866455
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 3.14221453666687,
      "step": 75
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 75,
      "training_step_time": 9.923230409622192
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 3.135904550552368,
      "step": 75
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 75,
      "training_step_time": 9.912460327148438
    },
    {
      "epoch": 2.86102294921875e-07,
      "model_forward_time": 2.753286838531494,
      "step": 75
    },
    {
      "epoch": 2.86102294921875e-07,
      "step": 75,
      "training_step_time": 10.024606227874756
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 1.756211757659912,
      "step": 76
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 76,
      "training_step_time": 5.221060752868652
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 3.140312433242798,
      "step": 76
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 76,
      "training_step_time": 9.923557758331299
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 2.436138868331909,
      "step": 76
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 76,
      "training_step_time": 7.465210199356079
    },
    {
      "epoch": 2.899169921875e-07,
      "model_forward_time": 3.134606122970581,
      "step": 76
    },
    {
      "epoch": 2.899169921875e-07,
      "step": 76,
      "training_step_time": 10.333984136581421
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 3.461168050765991,
      "step": 77
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 77,
      "training_step_time": 10.237694025039673
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 2.4357364177703857,
      "step": 77
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 77,
      "training_step_time": 7.469119071960449
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 3.140901803970337,
      "step": 77
    },
    {
      "dynamics_loss_avg": 0.1805661216378212,
      "epoch": 2.93731689453125e-07,
      "step": 77
    },
    {
      "action_loss_avg": 0.19686571694910526,
      "epoch": 2.93731689453125e-07,
      "step": 77
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 77,
      "training_step_time": 9.9221510887146
    },
    {
      "epoch": 2.93731689453125e-07,
      "model_forward_time": 2.4395129680633545,
      "step": 77
    },
    {
      "epoch": 2.93731689453125e-07,
      "step": 77,
      "training_step_time": 10.04324460029602
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 3.155017137527466,
      "step": 78
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 78,
      "training_step_time": 9.937620401382446
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 2.752716302871704,
      "step": 78
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 78,
      "training_step_time": 7.781921625137329
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 1.744354486465454,
      "step": 78
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 78,
      "training_step_time": 5.203176498413086
    },
    {
      "epoch": 2.9754638671875e-07,
      "model_forward_time": 2.4318668842315674,
      "step": 78
    },
    {
      "epoch": 2.9754638671875e-07,
      "step": 78,
      "training_step_time": 10.270608901977539
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 3.178882122039795,
      "step": 79
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 79,
      "training_step_time": 9.955610275268555
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 3.138944625854492,
      "step": 79
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 79,
      "training_step_time": 9.92102861404419
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 3.141096830368042,
      "step": 79
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 79,
      "training_step_time": 9.920871496200562
    },
    {
      "epoch": 3.01361083984375e-07,
      "model_forward_time": 3.421128511428833,
      "step": 79
    },
    {
      "epoch": 3.01361083984375e-07,
      "step": 79,
      "training_step_time": 10.295100450515747
    },
    {
      "epoch": 3.0517578125e-07,
      "grad_norm": 0.776980459690094,
      "learning_rate": 1e-05,
      "loss": 0.354,
      "step": 80
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 2.455221176147461,
      "step": 80
    },
    {
      "dynamics_loss_avg": 0.19060971289873124,
      "epoch": 3.0517578125e-07,
      "step": 80
    },
    {
      "action_loss_avg": 0.20637424737215043,
      "epoch": 3.0517578125e-07,
      "step": 80
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 80,
      "training_step_time": 7.4888176918029785
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 2.4334001541137695,
      "step": 80
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 80,
      "training_step_time": 7.470742702484131
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 3.1376116275787354,
      "step": 80
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 80,
      "training_step_time": 9.915928602218628
    },
    {
      "epoch": 3.0517578125e-07,
      "model_forward_time": 2.4342126846313477,
      "step": 80
    },
    {
      "epoch": 3.0517578125e-07,
      "step": 80,
      "training_step_time": 10.03346848487854
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 3.434788227081299,
      "step": 81
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 81,
      "training_step_time": 10.208963632583618
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 3.132582664489746,
      "step": 81
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 81,
      "training_step_time": 9.907166242599487
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 3.135005474090576,
      "step": 81
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 81,
      "training_step_time": 9.911060810089111
    },
    {
      "epoch": 3.08990478515625e-07,
      "model_forward_time": 3.1388514041900635,
      "step": 81
    },
    {
      "epoch": 3.08990478515625e-07,
      "step": 81,
      "training_step_time": 10.3260178565979
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 3.1510541439056396,
      "step": 82
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 82,
      "training_step_time": 9.934186935424805
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 2.6991963386535645,
      "step": 82
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 82,
      "training_step_time": 7.732089281082153
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 2.4281201362609863,
      "step": 82
    },
    {
      "dynamics_loss_avg": 0.15520297288894652,
      "epoch": 3.1280517578125e-07,
      "step": 82
    },
    {
      "action_loss_avg": 0.14997239895164965,
      "epoch": 3.1280517578125e-07,
      "step": 82
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 82,
      "training_step_time": 7.462143898010254
    },
    {
      "epoch": 3.1280517578125e-07,
      "model_forward_time": 2.429536819458008,
      "step": 82
    },
    {
      "epoch": 3.1280517578125e-07,
      "step": 82,
      "training_step_time": 10.321485042572021
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 3.1754231452941895,
      "step": 83
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 83,
      "training_step_time": 9.968010187149048
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 3.1405909061431885,
      "step": 83
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 83,
      "training_step_time": 9.917695760726929
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 2.447791337966919,
      "step": 83
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 83,
      "training_step_time": 7.796343803405762
    },
    {
      "epoch": 3.16619873046875e-07,
      "model_forward_time": 2.442857503890991,
      "step": 83
    },
    {
      "epoch": 3.16619873046875e-07,
      "step": 83,
      "training_step_time": 10.265580415725708
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 3.1851963996887207,
      "step": 84
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 84,
      "training_step_time": 9.969816446304321
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 2.4394257068634033,
      "step": 84
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 84,
      "training_step_time": 7.471181631088257
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 2.451401710510254,
      "step": 84
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 84,
      "training_step_time": 7.486430883407593
    },
    {
      "epoch": 3.204345703125e-07,
      "model_forward_time": 2.4384965896606445,
      "step": 84
    },
    {
      "epoch": 3.204345703125e-07,
      "step": 84,
      "training_step_time": 10.037357330322266
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 2.795705556869507,
      "step": 85
    },
    {
      "dynamics_loss_avg": 0.1960256926715374,
      "epoch": 3.24249267578125e-07,
      "step": 85
    },
    {
      "action_loss_avg": 0.1806217685341835,
      "epoch": 3.24249267578125e-07,
      "step": 85
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 85,
      "training_step_time": 7.833876371383667
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 3.151681900024414,
      "step": 85
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 85,
      "training_step_time": 9.930824279785156
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 2.4349687099456787,
      "step": 85
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 85,
      "training_step_time": 7.4641358852386475
    },
    {
      "epoch": 3.24249267578125e-07,
      "model_forward_time": 2.4366860389709473,
      "step": 85
    },
    {
      "epoch": 3.24249267578125e-07,
      "step": 85,
      "training_step_time": 10.027351379394531
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 3.1533079147338867,
      "step": 86
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 86,
      "training_step_time": 9.939438343048096
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 2.743809700012207,
      "step": 86
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 86,
      "training_step_time": 7.7745208740234375
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 2.4311110973358154,
      "step": 86
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 86,
      "training_step_time": 7.469272136688232
    },
    {
      "epoch": 3.2806396484375e-07,
      "model_forward_time": 1.745373010635376,
      "step": 86
    },
    {
      "epoch": 3.2806396484375e-07,
      "step": 86,
      "training_step_time": 10.336503982543945
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 2.4742441177368164,
      "step": 87
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 87,
      "training_step_time": 7.502598762512207
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 2.430040121078491,
      "step": 87
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 87,
      "training_step_time": 7.461716890335083
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 3.1348278522491455,
      "step": 87
    },
    {
      "dynamics_loss_avg": 0.16628067716956138,
      "epoch": 3.31878662109375e-07,
      "step": 87
    },
    {
      "action_loss_avg": 0.14898131974041462,
      "epoch": 3.31878662109375e-07,
      "step": 87
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 87,
      "training_step_time": 9.91299033164978
    },
    {
      "epoch": 3.31878662109375e-07,
      "model_forward_time": 2.7147293090820312,
      "step": 87
    },
    {
      "epoch": 3.31878662109375e-07,
      "step": 87,
      "training_step_time": 10.330840110778809
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 2.4694364070892334,
      "step": 88
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 88,
      "training_step_time": 7.500348329544067
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 3.1312060356140137,
      "step": 88
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 88,
      "training_step_time": 9.903602361679077
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 3.1322007179260254,
      "step": 88
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 88,
      "training_step_time": 9.90624475479126
    },
    {
      "epoch": 3.35693359375e-07,
      "model_forward_time": 3.1352715492248535,
      "step": 88
    },
    {
      "epoch": 3.35693359375e-07,
      "step": 88,
      "training_step_time": 10.378682851791382
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 3.4299182891845703,
      "step": 89
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 89,
      "training_step_time": 10.210269212722778
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 2.432373523712158,
      "step": 89
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 89,
      "training_step_time": 7.466045379638672
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 3.138840436935425,
      "step": 89
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 89,
      "training_step_time": 9.917562484741211
    },
    {
      "epoch": 3.39508056640625e-07,
      "model_forward_time": 3.1413002014160156,
      "step": 89
    },
    {
      "epoch": 3.39508056640625e-07,
      "step": 89,
      "training_step_time": 10.033089399337769
    },
    {
      "epoch": 3.4332275390625e-07,
      "grad_norm": 0.7117172479629517,
      "learning_rate": 1e-05,
      "loss": 0.357,
      "step": 90
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 3.1825571060180664,
      "step": 90
    },
    {
      "dynamics_loss_avg": 0.17546796575188636,
      "epoch": 3.4332275390625e-07,
      "step": 90
    },
    {
      "action_loss_avg": 0.1957143135368824,
      "epoch": 3.4332275390625e-07,
      "step": 90
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 90,
      "training_step_time": 9.970746994018555
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 3.4635560512542725,
      "step": 90
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 90,
      "training_step_time": 10.247600555419922
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 3.1367928981781006,
      "step": 90
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 90,
      "training_step_time": 9.918903112411499
    },
    {
      "epoch": 3.4332275390625e-07,
      "model_forward_time": 3.1388349533081055,
      "step": 90
    },
    {
      "epoch": 3.4332275390625e-07,
      "step": 90,
      "training_step_time": 10.036497592926025
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 2.4671943187713623,
      "step": 91
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 91,
      "training_step_time": 7.5003674030303955
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 3.1281535625457764,
      "step": 91
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 91,
      "training_step_time": 9.89645504951477
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 3.3739748001098633,
      "step": 91
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 91,
      "training_step_time": 10.144757986068726
    },
    {
      "epoch": 3.47137451171875e-07,
      "model_forward_time": 3.129925489425659,
      "step": 91
    },
    {
      "epoch": 3.47137451171875e-07,
      "step": 91,
      "training_step_time": 10.316424369812012
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 2.435356378555298,
      "step": 92
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 92,
      "training_step_time": 7.460088729858398
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 2.424539804458618,
      "step": 92
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 92,
      "training_step_time": 7.452321767807007
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 2.426144599914551,
      "step": 92
    },
    {
      "dynamics_loss_avg": 0.18360812738537788,
      "epoch": 3.509521484375e-07,
      "step": 92
    },
    {
      "action_loss_avg": 0.18848896920681,
      "epoch": 3.509521484375e-07,
      "step": 92
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 92,
      "training_step_time": 7.451840162277222
    },
    {
      "epoch": 3.509521484375e-07,
      "model_forward_time": 3.418630599975586,
      "step": 92
    },
    {
      "epoch": 3.509521484375e-07,
      "step": 92,
      "training_step_time": 10.281717538833618
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 2.4709627628326416,
      "step": 93
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 93,
      "training_step_time": 7.50070595741272
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 2.4267570972442627,
      "step": 93
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 93,
      "training_step_time": 7.458396673202515
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 3.1342644691467285,
      "step": 93
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 93,
      "training_step_time": 9.910022258758545
    },
    {
      "epoch": 3.54766845703125e-07,
      "model_forward_time": 2.428359270095825,
      "step": 93
    },
    {
      "epoch": 3.54766845703125e-07,
      "step": 93,
      "training_step_time": 10.33362340927124
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 3.153313636779785,
      "step": 94
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 94,
      "training_step_time": 9.934503078460693
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 3.400589942932129,
      "step": 94
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 94,
      "training_step_time": 10.17546272277832
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 3.1359667778015137,
      "step": 94
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 94,
      "training_step_time": 9.918085098266602
    },
    {
      "epoch": 3.5858154296875e-07,
      "model_forward_time": 3.1334927082061768,
      "step": 94
    },
    {
      "epoch": 3.5858154296875e-07,
      "step": 94,
      "training_step_time": 10.35533595085144
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 3.1454508304595947,
      "step": 95
    },
    {
      "dynamics_loss_avg": 0.15631940439343453,
      "epoch": 3.62396240234375e-07,
      "step": 95
    },
    {
      "action_loss_avg": 0.13695881329476833,
      "epoch": 3.62396240234375e-07,
      "step": 95
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 95,
      "training_step_time": 9.927170991897583
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 2.427083969116211,
      "step": 95
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 95,
      "training_step_time": 7.455444812774658
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 2.6901371479034424,
      "step": 95
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 95,
      "training_step_time": 7.71808648109436
    },
    {
      "epoch": 3.62396240234375e-07,
      "model_forward_time": 1.7388479709625244,
      "step": 95
    },
    {
      "epoch": 3.62396240234375e-07,
      "step": 95,
      "training_step_time": 10.040098428726196
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 3.1740760803222656,
      "step": 96
    },
    {
      "epoch": 3.662109375e-07,
      "step": 96,
      "training_step_time": 9.950132131576538
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 3.132998466491699,
      "step": 96
    },
    {
      "epoch": 3.662109375e-07,
      "step": 96,
      "training_step_time": 9.913501024246216
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 3.1319682598114014,
      "step": 96
    },
    {
      "epoch": 3.662109375e-07,
      "step": 96,
      "training_step_time": 9.911998510360718
    },
    {
      "epoch": 3.662109375e-07,
      "model_forward_time": 2.433155059814453,
      "step": 96
    },
    {
      "epoch": 3.662109375e-07,
      "step": 96,
      "training_step_time": 10.046152830123901
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 2.4383933544158936,
      "step": 97
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 97,
      "training_step_time": 7.472345590591431
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 3.1299445629119873,
      "step": 97
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 97,
      "training_step_time": 9.911715269088745
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 2.4317245483398438,
      "step": 97
    },
    {
      "dynamics_loss_avg": 0.1938873738050461,
      "epoch": 3.70025634765625e-07,
      "step": 97
    },
    {
      "action_loss_avg": 0.23244875036180018,
      "epoch": 3.70025634765625e-07,
      "step": 97
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 97,
      "training_step_time": 7.468984127044678
    },
    {
      "epoch": 3.70025634765625e-07,
      "model_forward_time": 2.429060220718384,
      "step": 97
    },
    {
      "epoch": 3.70025634765625e-07,
      "step": 97,
      "training_step_time": 10.053194522857666
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 3.1648147106170654,
      "step": 98
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 98,
      "training_step_time": 9.945895671844482
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 2.6771819591522217,
      "step": 98
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 98,
      "training_step_time": 7.712144136428833
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 3.1357202529907227,
      "step": 98
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 98,
      "training_step_time": 9.918118000030518
    },
    {
      "epoch": 3.7384033203125e-07,
      "model_forward_time": 2.4313430786132812,
      "step": 98
    },
    {
      "epoch": 3.7384033203125e-07,
      "step": 98,
      "training_step_time": 10.356349229812622
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 3.1419858932495117,
      "step": 99
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 99,
      "training_step_time": 9.91692566871643
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 2.425403118133545,
      "step": 99
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 99,
      "training_step_time": 7.4547812938690186
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 2.7396411895751953,
      "step": 99
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 99,
      "training_step_time": 7.768052816390991
    },
    {
      "epoch": 3.77655029296875e-07,
      "model_forward_time": 2.4266741275787354,
      "step": 99
    },
    {
      "epoch": 3.77655029296875e-07,
      "step": 99,
      "training_step_time": 10.350880146026611
    },
    {
      "epoch": 3.814697265625e-07,
      "grad_norm": 0.8621710538864136,
      "learning_rate": 1e-05,
      "loss": 0.3511,
      "step": 100
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 3.1372029781341553,
      "step": 100
    },
    {
      "dynamics_loss_avg": 0.16103590205311774,
      "epoch": 3.814697265625e-07,
      "step": 100
    },
    {
      "action_loss_avg": 0.17085252031683923,
      "epoch": 3.814697265625e-07,
      "step": 100
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 100,
      "training_step_time": 9.91651463508606
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 3.1283082962036133,
      "step": 100
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 100,
      "training_step_time": 9.902151584625244
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 2.428968667984009,
      "step": 100
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 100,
      "training_step_time": 7.4573073387146
    },
    {
      "epoch": 3.814697265625e-07,
      "model_forward_time": 2.424852132797241,
      "step": 100
    },
    {
      "epoch": 3.814697265625e-07,
      "step": 100,
      "training_step_time": 10.353065013885498
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 3.176750659942627,
      "step": 101
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 101,
      "training_step_time": 9.95509123802185
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 3.129711627960205,
      "step": 101
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 101,
      "training_step_time": 9.909854173660278
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 3.1394786834716797,
      "step": 101
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 101,
      "training_step_time": 9.915915489196777
    },
    {
      "epoch": 3.85284423828125e-07,
      "model_forward_time": 3.1344330310821533,
      "step": 101
    },
    {
      "epoch": 3.85284423828125e-07,
      "step": 101,
      "training_step_time": 10.031917810440063
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 3.4580743312835693,
      "step": 102
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 102,
      "training_step_time": 10.235118389129639
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 2.4303367137908936,
      "step": 102
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 102,
      "training_step_time": 7.4588329792022705
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 3.131084680557251,
      "step": 102
    },
    {
      "dynamics_loss_avg": 0.1612263672053814,
      "epoch": 3.8909912109375e-07,
      "step": 102
    },
    {
      "action_loss_avg": 0.16020172834396362,
      "epoch": 3.8909912109375e-07,
      "step": 102
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 102,
      "training_step_time": 9.904916286468506
    },
    {
      "epoch": 3.8909912109375e-07,
      "model_forward_time": 3.1330490112304688,
      "step": 102
    },
    {
      "epoch": 3.8909912109375e-07,
      "step": 102,
      "training_step_time": 10.074356079101562
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 3.1752424240112305,
      "step": 103
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 103,
      "training_step_time": 9.947686672210693
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 3.1344118118286133,
      "step": 103
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 103,
      "training_step_time": 10.201589345932007
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 3.133444309234619,
      "step": 103
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 103,
      "training_step_time": 9.908126831054688
    },
    {
      "epoch": 3.92913818359375e-07,
      "model_forward_time": 3.1305229663848877,
      "step": 103
    },
    {
      "epoch": 3.92913818359375e-07,
      "step": 103,
      "training_step_time": 10.283137083053589
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 2.4425811767578125,
      "step": 104
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 104,
      "training_step_time": 7.47077751159668
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 2.4234812259674072,
      "step": 104
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 104,
      "training_step_time": 7.46091628074646
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 2.426931381225586,
      "step": 104
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 104,
      "training_step_time": 7.457223415374756
    },
    {
      "epoch": 3.96728515625e-07,
      "model_forward_time": 3.3676528930664062,
      "step": 104
    },
    {
      "epoch": 3.96728515625e-07,
      "step": 104,
      "training_step_time": 10.317100524902344
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 3.150822401046753,
      "step": 105
    },
    {
      "dynamics_loss_avg": 0.17409392148256303,
      "epoch": 4.00543212890625e-07,
      "step": 105
    },
    {
      "action_loss_avg": 0.16568992361426355,
      "epoch": 4.00543212890625e-07,
      "step": 105
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 105,
      "training_step_time": 9.927908897399902
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 2.429138660430908,
      "step": 105
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 105,
      "training_step_time": 7.45720362663269
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 3.1287825107574463,
      "step": 105
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 105,
      "training_step_time": 9.901841402053833
    },
    {
      "epoch": 4.00543212890625e-07,
      "model_forward_time": 3.140655994415283,
      "step": 105
    },
    {
      "epoch": 4.00543212890625e-07,
      "step": 105,
      "training_step_time": 10.31259274482727
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 2.7783493995666504,
      "step": 106
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 106,
      "training_step_time": 7.809722185134888
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 2.427619695663452,
      "step": 106
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 106,
      "training_step_time": 7.4585676193237305
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 2.4296154975891113,
      "step": 106
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 106,
      "training_step_time": 7.464155912399292
    },
    {
      "epoch": 4.0435791015625e-07,
      "model_forward_time": 2.4278793334960938,
      "step": 106
    },
    {
      "epoch": 4.0435791015625e-07,
      "step": 106,
      "training_step_time": 10.049486875534058
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 2.4412405490875244,
      "step": 107
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 107,
      "training_step_time": 7.474270582199097
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 3.4037487506866455,
      "step": 107
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 107,
      "training_step_time": 10.17663311958313
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 2.4278643131256104,
      "step": 107
    },
    {
      "dynamics_loss_avg": 0.16480540558695794,
      "epoch": 4.08172607421875e-07,
      "step": 107
    },
    {
      "action_loss_avg": 0.13680629432201385,
      "epoch": 4.08172607421875e-07,
      "step": 107
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 107,
      "training_step_time": 7.460858583450317
    },
    {
      "epoch": 4.08172607421875e-07,
      "model_forward_time": 2.4290831089019775,
      "step": 107
    },
    {
      "epoch": 4.08172607421875e-07,
      "step": 107,
      "training_step_time": 10.074981927871704
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 2.4679758548736572,
      "step": 108
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 108,
      "training_step_time": 7.521811008453369
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 3.1402535438537598,
      "step": 108
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 108,
      "training_step_time": 9.917356252670288
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 2.4432625770568848,
      "step": 108
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 108,
      "training_step_time": 7.487454414367676
    },
    {
      "epoch": 4.119873046875e-07,
      "model_forward_time": 2.748110055923462,
      "step": 108
    },
    {
      "epoch": 4.119873046875e-07,
      "step": 108,
      "training_step_time": 10.317296743392944
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 3.180516242980957,
      "step": 109
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 109,
      "training_step_time": 9.956937789916992
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 3.1387505531311035,
      "step": 109
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 109,
      "training_step_time": 9.91619610786438
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 3.143331527709961,
      "step": 109
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 109,
      "training_step_time": 9.920543670654297
    },
    {
      "epoch": 4.15802001953125e-07,
      "model_forward_time": 3.4573028087615967,
      "step": 109
    },
    {
      "epoch": 4.15802001953125e-07,
      "step": 109,
      "training_step_time": 10.328293561935425
    },
    {
      "epoch": 4.1961669921875e-07,
      "grad_norm": 1.1354774236679077,
      "learning_rate": 1e-05,
      "loss": 0.3116,
      "step": 110
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 1.8009107112884521,
      "step": 110
    },
    {
      "dynamics_loss_avg": 0.15924567207694054,
      "epoch": 4.1961669921875e-07,
      "step": 110
    },
    {
      "action_loss_avg": 0.16350408941507338,
      "epoch": 4.1961669921875e-07,
      "step": 110
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 110,
      "training_step_time": 5.273011207580566
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 1.7510855197906494,
      "step": 110
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 110,
      "training_step_time": 5.2139952182769775
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 3.139650344848633,
      "step": 110
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 110,
      "training_step_time": 9.909877300262451
    },
    {
      "epoch": 4.1961669921875e-07,
      "model_forward_time": 2.4379024505615234,
      "step": 110
    },
    {
      "epoch": 4.1961669921875e-07,
      "step": 110,
      "training_step_time": 10.265742540359497
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 2.4608304500579834,
      "step": 111
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 111,
      "training_step_time": 7.502443790435791
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 3.441025733947754,
      "step": 111
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 111,
      "training_step_time": 10.215909242630005
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 2.434079885482788,
      "step": 111
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 111,
      "training_step_time": 7.467111110687256
    },
    {
      "epoch": 4.23431396484375e-07,
      "model_forward_time": 2.4371562004089355,
      "step": 111
    },
    {
      "epoch": 4.23431396484375e-07,
      "step": 111,
      "training_step_time": 10.36445665359497
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 2.4684202671051025,
      "step": 112
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 112,
      "training_step_time": 7.500744819641113
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 2.4350433349609375,
      "step": 112
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 112,
      "training_step_time": 7.46440577507019
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 3.1372315883636475,
      "step": 112
    },
    {
      "dynamics_loss_avg": 0.17838706001639365,
      "epoch": 4.2724609375e-07,
      "step": 112
    },
    {
      "action_loss_avg": 0.16184035316109657,
      "epoch": 4.2724609375e-07,
      "step": 112
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 112,
      "training_step_time": 9.917309284210205
    },
    {
      "epoch": 4.2724609375e-07,
      "model_forward_time": 3.435614585876465,
      "step": 112
    },
    {
      "epoch": 4.2724609375e-07,
      "step": 112,
      "training_step_time": 10.31876516342163
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 3.191230058670044,
      "step": 113
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 113,
      "training_step_time": 9.9614098072052
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 3.14021372795105,
      "step": 113
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 113,
      "training_step_time": 9.909690380096436
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 3.1355130672454834,
      "step": 113
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 113,
      "training_step_time": 9.907455444335938
    },
    {
      "epoch": 4.31060791015625e-07,
      "model_forward_time": 3.139347553253174,
      "step": 113
    },
    {
      "epoch": 4.31060791015625e-07,
      "step": 113,
      "training_step_time": 10.343631744384766
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 3.4261953830718994,
      "step": 114
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 114,
      "training_step_time": 10.2022545337677
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 2.436570644378662,
      "step": 114
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 114,
      "training_step_time": 7.466013431549072
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 3.1368699073791504,
      "step": 114
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 114,
      "training_step_time": 9.907294273376465
    },
    {
      "epoch": 4.3487548828125e-07,
      "model_forward_time": 2.4315311908721924,
      "step": 114
    },
    {
      "epoch": 4.3487548828125e-07,
      "step": 114,
      "training_step_time": 10.31178331375122
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 3.174610137939453,
      "step": 115
    },
    {
      "dynamics_loss_avg": 0.1887528195977211,
      "epoch": 4.38690185546875e-07,
      "step": 115
    },
    {
      "action_loss_avg": 0.20683527663350104,
      "epoch": 4.38690185546875e-07,
      "step": 115
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 115,
      "training_step_time": 9.949831485748291
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 3.4508631229400635,
      "step": 115
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 115,
      "training_step_time": 10.22641634941101
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 3.1337339878082275,
      "step": 115
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 115,
      "training_step_time": 9.90843152999878
    },
    {
      "epoch": 4.38690185546875e-07,
      "model_forward_time": 2.4373764991760254,
      "step": 115
    },
    {
      "epoch": 4.38690185546875e-07,
      "step": 115,
      "training_step_time": 10.281120300292969
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 3.1490554809570312,
      "step": 116
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 116,
      "training_step_time": 9.92317533493042
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 2.4335944652557373,
      "step": 116
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 116,
      "training_step_time": 7.467587947845459
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 2.709254503250122,
      "step": 116
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 116,
      "training_step_time": 7.741010665893555
    },
    {
      "epoch": 4.425048828125e-07,
      "model_forward_time": 3.1362853050231934,
      "step": 116
    },
    {
      "epoch": 4.425048828125e-07,
      "step": 116,
      "training_step_time": 10.293550729751587
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 3.183636426925659,
      "step": 117
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 117,
      "training_step_time": 9.967014789581299
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 3.14839243888855,
      "step": 117
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 117,
      "training_step_time": 9.932403564453125
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 3.1548197269439697,
      "step": 117
    },
    {
      "dynamics_loss_avg": 0.17761856764554979,
      "epoch": 4.46319580078125e-07,
      "step": 117
    },
    {
      "action_loss_avg": 0.1864478573203087,
      "epoch": 4.46319580078125e-07,
      "step": 117
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 117,
      "training_step_time": 9.93688178062439
    },
    {
      "epoch": 4.46319580078125e-07,
      "model_forward_time": 3.4637975692749023,
      "step": 117
    },
    {
      "epoch": 4.46319580078125e-07,
      "step": 117,
      "training_step_time": 10.338021993637085
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 1.7633347511291504,
      "step": 118
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 118,
      "training_step_time": 5.2322680950164795
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 3.1460301876068115,
      "step": 118
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 118,
      "training_step_time": 9.921441555023193
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 2.435525894165039,
      "step": 118
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 118,
      "training_step_time": 7.470287322998047
    },
    {
      "epoch": 4.5013427734375e-07,
      "model_forward_time": 2.4322941303253174,
      "step": 118
    },
    {
      "epoch": 4.5013427734375e-07,
      "step": 118,
      "training_step_time": 10.259217739105225
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 2.448242664337158,
      "step": 119
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 119,
      "training_step_time": 7.475161790847778
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 2.7473244667053223,
      "step": 119
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 119,
      "training_step_time": 7.77678656578064
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 3.137490749359131,
      "step": 119
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 119,
      "training_step_time": 9.910165071487427
    },
    {
      "epoch": 4.53948974609375e-07,
      "model_forward_time": 2.44675350189209,
      "step": 119
    },
    {
      "epoch": 4.53948974609375e-07,
      "step": 119,
      "training_step_time": 10.308573484420776
    },
    {
      "epoch": 4.57763671875e-07,
      "grad_norm": 1.2156368494033813,
      "learning_rate": 1e-05,
      "loss": 0.3182,
      "step": 120
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 2.445922613143921,
      "step": 120
    },
    {
      "dynamics_loss_avg": 0.11034443341195584,
      "epoch": 4.57763671875e-07,
      "step": 120
    },
    {
      "action_loss_avg": 0.0799975283909589,
      "epoch": 4.57763671875e-07,
      "step": 120
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 120,
      "training_step_time": 7.488397598266602
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 3.1326680183410645,
      "step": 120
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 120,
      "training_step_time": 9.910224437713623
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 2.7017884254455566,
      "step": 120
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 120,
      "training_step_time": 7.7372894287109375
    },
    {
      "epoch": 4.57763671875e-07,
      "model_forward_time": 2.4284489154815674,
      "step": 120
    },
    {
      "epoch": 4.57763671875e-07,
      "step": 120,
      "training_step_time": 10.060354232788086
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 3.180711030960083,
      "step": 121
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 121,
      "training_step_time": 9.964476346969604
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 2.432281255722046,
      "step": 121
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 121,
      "training_step_time": 7.464600563049316
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 3.1319053173065186,
      "step": 121
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 121,
      "training_step_time": 9.900697469711304
    },
    {
      "epoch": 4.61578369140625e-07,
      "model_forward_time": 3.4224209785461426,
      "step": 121
    },
    {
      "epoch": 4.61578369140625e-07,
      "step": 121,
      "training_step_time": 10.31957221031189
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 2.4492969512939453,
      "step": 122
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 122,
      "training_step_time": 7.482911825180054
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 2.435239791870117,
      "step": 122
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 122,
      "training_step_time": 7.465594291687012
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 2.434717893600464,
      "step": 122
    },
    {
      "dynamics_loss_avg": 0.17101249024271964,
      "epoch": 4.6539306640625e-07,
      "step": 122
    },
    {
      "action_loss_avg": 0.18186988458037376,
      "epoch": 4.6539306640625e-07,
      "step": 122
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 122,
      "training_step_time": 7.471681356430054
    },
    {
      "epoch": 4.6539306640625e-07,
      "model_forward_time": 3.133305549621582,
      "step": 122
    },
    {
      "epoch": 4.6539306640625e-07,
      "step": 122,
      "training_step_time": 10.369939804077148
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 3.1792519092559814,
      "step": 123
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 123,
      "training_step_time": 9.948607921600342
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 2.7341318130493164,
      "step": 123
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 123,
      "training_step_time": 7.7607197761535645
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 3.1387178897857666,
      "step": 123
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 123,
      "training_step_time": 9.91063928604126
    },
    {
      "epoch": 4.69207763671875e-07,
      "model_forward_time": 2.4325625896453857,
      "step": 123
    },
    {
      "epoch": 4.69207763671875e-07,
      "step": 123,
      "training_step_time": 10.289449691772461
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 2.4776649475097656,
      "step": 124
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 124,
      "training_step_time": 7.505577087402344
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 3.1345207691192627,
      "step": 124
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 124,
      "training_step_time": 9.901101112365723
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 3.3931708335876465,
      "step": 124
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 124,
      "training_step_time": 10.159824848175049
    },
    {
      "epoch": 4.730224609375e-07,
      "model_forward_time": 3.1394922733306885,
      "step": 124
    },
    {
      "epoch": 4.730224609375e-07,
      "step": 124,
      "training_step_time": 10.30919861793518
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 1.7764263153076172,
      "step": 125
    },
    {
      "dynamics_loss_avg": 0.16348480731248854,
      "epoch": 4.76837158203125e-07,
      "step": 125
    },
    {
      "action_loss_avg": 0.17041967399418353,
      "epoch": 4.76837158203125e-07,
      "step": 125
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 125,
      "training_step_time": 5.238940715789795
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 3.1342575550079346,
      "step": 125
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 125,
      "training_step_time": 9.904984474182129
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 3.1374499797821045,
      "step": 125
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 125,
      "training_step_time": 9.90898585319519
    },
    {
      "epoch": 4.76837158203125e-07,
      "model_forward_time": 3.4095990657806396,
      "step": 125
    },
    {
      "epoch": 4.76837158203125e-07,
      "step": 125,
      "training_step_time": 10.307658195495605
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 3.177065372467041,
      "step": 126
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 126,
      "training_step_time": 9.951213121414185
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 3.1350300312042236,
      "step": 126
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 126,
      "training_step_time": 9.907418489456177
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 3.1391286849975586,
      "step": 126
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 126,
      "training_step_time": 9.913537502288818
    },
    {
      "epoch": 4.8065185546875e-07,
      "model_forward_time": 3.1347451210021973,
      "step": 126
    },
    {
      "epoch": 4.8065185546875e-07,
      "step": 126,
      "training_step_time": 10.311411142349243
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 3.45314884185791,
      "step": 127
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 127,
      "training_step_time": 10.224453449249268
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 1.7393295764923096,
      "step": 127
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 127,
      "training_step_time": 5.191680669784546
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 2.4242119789123535,
      "step": 127
    },
    {
      "dynamics_loss_avg": 0.15766530334949494,
      "epoch": 4.84466552734375e-07,
      "step": 127
    },
    {
      "action_loss_avg": 0.15913912169635297,
      "epoch": 4.84466552734375e-07,
      "step": 127
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 127,
      "training_step_time": 7.446926832199097
    },
    {
      "epoch": 4.84466552734375e-07,
      "model_forward_time": 2.4245121479034424,
      "step": 127
    },
    {
      "epoch": 4.84466552734375e-07,
      "step": 127,
      "training_step_time": 10.272032976150513
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 2.469045877456665,
      "step": 128
    },
    {
      "epoch": 4.8828125e-07,
      "step": 128,
      "training_step_time": 7.490689516067505
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 3.1284561157226562,
      "step": 128
    },
    {
      "epoch": 4.8828125e-07,
      "step": 128,
      "training_step_time": 9.893217086791992
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 3.3915748596191406,
      "step": 128
    },
    {
      "epoch": 4.8828125e-07,
      "step": 128,
      "training_step_time": 10.1597580909729
    },
    {
      "epoch": 4.8828125e-07,
      "model_forward_time": 2.4302494525909424,
      "step": 128
    },
    {
      "epoch": 4.8828125e-07,
      "step": 128,
      "training_step_time": 10.318937540054321
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 3.1465463638305664,
      "step": 129
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 129,
      "training_step_time": 9.91643238067627
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 3.130730152130127,
      "step": 129
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 129,
      "training_step_time": 9.90208911895752
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 3.1334497928619385,
      "step": 129
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 129,
      "training_step_time": 9.907880783081055
    },
    {
      "epoch": 4.92095947265625e-07,
      "model_forward_time": 2.7359936237335205,
      "step": 129
    },
    {
      "epoch": 4.92095947265625e-07,
      "step": 129,
      "training_step_time": 10.031369924545288
    },
    {
      "epoch": 4.9591064453125e-07,
      "grad_norm": 1.044456124305725,
      "learning_rate": 1e-05,
      "loss": 0.3003,
      "step": 130
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 3.145137310028076,
      "step": 130
    },
    {
      "dynamics_loss_avg": 0.13186834156513214,
      "epoch": 4.9591064453125e-07,
      "step": 130
    },
    {
      "action_loss_avg": 0.11467230804264546,
      "epoch": 4.9591064453125e-07,
      "step": 130
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 130,
      "training_step_time": 9.916913509368896
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 2.4313111305236816,
      "step": 130
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 130,
      "training_step_time": 7.455106019973755
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 2.429394245147705,
      "step": 130
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 130,
      "training_step_time": 7.449849843978882
    },
    {
      "epoch": 4.9591064453125e-07,
      "model_forward_time": 3.1305410861968994,
      "step": 130
    },
    {
      "epoch": 4.9591064453125e-07,
      "step": 130,
      "training_step_time": 10.307974100112915
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 3.414719581604004,
      "step": 131
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 131,
      "training_step_time": 10.179817914962769
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 1.7421960830688477,
      "step": 131
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 131,
      "training_step_time": 5.196242094039917
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 3.1300787925720215,
      "step": 131
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 131,
      "training_step_time": 9.89898419380188
    },
    {
      "epoch": 4.99725341796875e-07,
      "model_forward_time": 3.1311609745025635,
      "step": 131
    },
    {
      "epoch": 4.99725341796875e-07,
      "step": 131,
      "training_step_time": 10.028833150863647
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 2.473938226699829,
      "step": 132
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 132,
      "training_step_time": 7.503029108047485
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 2.7136313915252686,
      "step": 132
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 132,
      "training_step_time": 7.740511178970337
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 3.131364107131958,
      "step": 132
    },
    {
      "dynamics_loss_avg": 0.14736735858023167,
      "epoch": 5.035400390625e-07,
      "step": 132
    },
    {
      "action_loss_avg": 0.14746766351163387,
      "epoch": 5.035400390625e-07,
      "step": 132
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 132,
      "training_step_time": 9.909422397613525
    },
    {
      "epoch": 5.035400390625e-07,
      "model_forward_time": 3.1339752674102783,
      "step": 132
    },
    {
      "epoch": 5.035400390625e-07,
      "step": 132,
      "training_step_time": 10.320029497146606
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 3.161830425262451,
      "step": 133
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 133,
      "training_step_time": 9.937207698822021
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 2.436403512954712,
      "step": 133
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 133,
      "training_step_time": 7.4671642780303955
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 3.1368210315704346,
      "step": 133
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 133,
      "training_step_time": 10.188371896743774
    },
    {
      "epoch": 5.07354736328125e-07,
      "model_forward_time": 3.1428356170654297,
      "step": 133
    },
    {
      "epoch": 5.07354736328125e-07,
      "step": 133,
      "training_step_time": 10.335291385650635
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 3.1494202613830566,
      "step": 134
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 134,
      "training_step_time": 9.92825436592102
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 2.4342331886291504,
      "step": 134
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 134,
      "training_step_time": 7.469674825668335
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 2.432542085647583,
      "step": 134
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 134,
      "training_step_time": 7.467092990875244
    },
    {
      "epoch": 5.1116943359375e-07,
      "model_forward_time": 3.1337475776672363,
      "step": 134
    },
    {
      "epoch": 5.1116943359375e-07,
      "step": 134,
      "training_step_time": 10.329260349273682
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 3.4877002239227295,
      "step": 135
    },
    {
      "dynamics_loss_avg": 0.15680023543536664,
      "epoch": 5.14984130859375e-07,
      "step": 135
    },
    {
      "action_loss_avg": 0.1898056659847498,
      "epoch": 5.14984130859375e-07,
      "step": 135
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 135,
      "training_step_time": 10.276003360748291
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 3.1416947841644287,
      "step": 135
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 135,
      "training_step_time": 9.921002388000488
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 3.1406002044677734,
      "step": 135
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 135,
      "training_step_time": 9.917224884033203
    },
    {
      "epoch": 5.14984130859375e-07,
      "model_forward_time": 3.16721248626709,
      "step": 135
    },
    {
      "epoch": 5.14984130859375e-07,
      "step": 135,
      "training_step_time": 10.34251880645752
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 3.473109483718872,
      "step": 136
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 136,
      "training_step_time": 10.253405809402466
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 2.436981678009033,
      "step": 136
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 136,
      "training_step_time": 7.471354007720947
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 2.4393250942230225,
      "step": 136
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 136,
      "training_step_time": 7.472859859466553
    },
    {
      "epoch": 5.18798828125e-07,
      "model_forward_time": 2.429107427597046,
      "step": 136
    },
    {
      "epoch": 5.18798828125e-07,
      "step": 136,
      "training_step_time": 10.260363578796387
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 3.1503047943115234,
      "step": 137
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 137,
      "training_step_time": 9.923844337463379
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 3.1338582038879395,
      "step": 137
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 137,
      "training_step_time": 9.90936017036438
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 3.3984076976776123,
      "step": 137
    },
    {
      "dynamics_loss_avg": 0.13538000658154486,
      "epoch": 5.22613525390625e-07,
      "step": 137
    },
    {
      "action_loss_avg": 0.12907137777656316,
      "epoch": 5.22613525390625e-07,
      "step": 137
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 137,
      "training_step_time": 10.174010515213013
    },
    {
      "epoch": 5.22613525390625e-07,
      "model_forward_time": 2.4332051277160645,
      "step": 137
    },
    {
      "epoch": 5.22613525390625e-07,
      "step": 137,
      "training_step_time": 10.343728303909302
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 2.485912322998047,
      "step": 138
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 138,
      "training_step_time": 7.523162364959717
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 3.1531646251678467,
      "step": 138
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 138,
      "training_step_time": 9.934670448303223
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 2.438420057296753,
      "step": 138
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 138,
      "training_step_time": 7.472330570220947
    },
    {
      "epoch": 5.2642822265625e-07,
      "model_forward_time": 3.4284017086029053,
      "step": 138
    },
    {
      "epoch": 5.2642822265625e-07,
      "step": 138,
      "training_step_time": 10.29098391532898
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 3.1586713790893555,
      "step": 139
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 139,
      "training_step_time": 9.926851987838745
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 3.13753342628479,
      "step": 139
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 139,
      "training_step_time": 9.913177013397217
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 3.1387228965759277,
      "step": 139
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 139,
      "training_step_time": 9.90936279296875
    },
    {
      "epoch": 5.30242919921875e-07,
      "model_forward_time": 2.4346933364868164,
      "step": 139
    },
    {
      "epoch": 5.30242919921875e-07,
      "step": 139,
      "training_step_time": 10.326263427734375
    },
    {
      "epoch": 5.340576171875e-07,
      "grad_norm": 0.746449887752533,
      "learning_rate": 1e-05,
      "loss": 0.2734,
      "step": 140
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 3.612576723098755,
      "step": 140
    },
    {
      "dynamics_loss_avg": 0.11850764229893684,
      "epoch": 5.340576171875e-07,
      "step": 140
    },
    {
      "action_loss_avg": 0.0948716502636671,
      "epoch": 5.340576171875e-07,
      "step": 140
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 140,
      "training_step_time": 10.392796039581299
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 2.446545124053955,
      "step": 140
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 140,
      "training_step_time": 7.48537278175354
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 3.1488022804260254,
      "step": 140
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 140,
      "training_step_time": 9.926568984985352
    },
    {
      "epoch": 5.340576171875e-07,
      "model_forward_time": 3.142244338989258,
      "step": 140
    },
    {
      "epoch": 5.340576171875e-07,
      "step": 140,
      "training_step_time": 10.043915271759033
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 2.478126049041748,
      "step": 141
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 141,
      "training_step_time": 7.521047830581665
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 3.4677257537841797,
      "step": 141
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 141,
      "training_step_time": 10.239915609359741
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 2.4369289875030518,
      "step": 141
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 141,
      "training_step_time": 7.463412284851074
    },
    {
      "epoch": 5.37872314453125e-07,
      "model_forward_time": 3.146339178085327,
      "step": 141
    },
    {
      "epoch": 5.37872314453125e-07,
      "step": 141,
      "training_step_time": 10.286756038665771
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 3.1684114933013916,
      "step": 142
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 142,
      "training_step_time": 9.946135997772217
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 3.135366439819336,
      "step": 142
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 142,
      "training_step_time": 9.90165090560913
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 3.434612512588501,
      "step": 142
    },
    {
      "dynamics_loss_avg": 0.13919461965560914,
      "epoch": 5.4168701171875e-07,
      "step": 142
    },
    {
      "action_loss_avg": 0.14347109124064444,
      "epoch": 5.4168701171875e-07,
      "step": 142
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 142,
      "training_step_time": 10.206902503967285
    },
    {
      "epoch": 5.4168701171875e-07,
      "model_forward_time": 3.135627508163452,
      "step": 142
    },
    {
      "epoch": 5.4168701171875e-07,
      "step": 142,
      "training_step_time": 10.038632869720459
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 3.1542611122131348,
      "step": 143
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 143,
      "training_step_time": 9.922709703445435
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 3.1359105110168457,
      "step": 143
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 143,
      "training_step_time": 9.910600185394287
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 2.433760643005371,
      "step": 143
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 143,
      "training_step_time": 7.466831922531128
    },
    {
      "epoch": 5.45501708984375e-07,
      "model_forward_time": 2.4367804527282715,
      "step": 143
    },
    {
      "epoch": 5.45501708984375e-07,
      "step": 143,
      "training_step_time": 10.268431186676025
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 3.141191244125366,
      "step": 144
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 144,
      "training_step_time": 9.910053253173828
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 2.4333457946777344,
      "step": 144
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 144,
      "training_step_time": 7.451470375061035
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 2.427985668182373,
      "step": 144
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 144,
      "training_step_time": 7.449127674102783
    },
    {
      "epoch": 5.4931640625e-07,
      "model_forward_time": 2.428107738494873,
      "step": 144
    },
    {
      "epoch": 5.4931640625e-07,
      "step": 144,
      "training_step_time": 10.346231460571289
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 2.4534828662872314,
      "step": 145
    },
    {
      "dynamics_loss_avg": 0.1538783334195614,
      "epoch": 5.53131103515625e-07,
      "step": 145
    },
    {
      "action_loss_avg": 0.1470617586746812,
      "epoch": 5.53131103515625e-07,
      "step": 145
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 145,
      "training_step_time": 7.482022047042847
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 3.4108049869537354,
      "step": 145
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 145,
      "training_step_time": 10.173114776611328
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 2.4349000453948975,
      "step": 145
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 145,
      "training_step_time": 7.463622808456421
    },
    {
      "epoch": 5.53131103515625e-07,
      "model_forward_time": 2.428133964538574,
      "step": 145
    },
    {
      "epoch": 5.53131103515625e-07,
      "step": 145,
      "training_step_time": 10.049409627914429
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 2.4720540046691895,
      "step": 146
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 146,
      "training_step_time": 7.498442888259888
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 2.427295684814453,
      "step": 146
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 146,
      "training_step_time": 7.45244288444519
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 2.685529947280884,
      "step": 146
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 146,
      "training_step_time": 7.706712484359741
    },
    {
      "epoch": 5.5694580078125e-07,
      "model_forward_time": 3.127467155456543,
      "step": 146
    },
    {
      "epoch": 5.5694580078125e-07,
      "step": 146,
      "training_step_time": 10.309290885925293
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 2.463862895965576,
      "step": 147
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 147,
      "training_step_time": 7.493175506591797
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 3.1376686096191406,
      "step": 147
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 147,
      "training_step_time": 9.906694650650024
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 3.1350674629211426,
      "step": 147
    },
    {
      "dynamics_loss_avg": 0.11525478698313237,
      "epoch": 5.60760498046875e-07,
      "step": 147
    },
    {
      "action_loss_avg": 0.09180264584720135,
      "epoch": 5.60760498046875e-07,
      "step": 147
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 147,
      "training_step_time": 9.90387511253357
    },
    {
      "epoch": 5.60760498046875e-07,
      "model_forward_time": 3.1404545307159424,
      "step": 147
    },
    {
      "epoch": 5.60760498046875e-07,
      "step": 147,
      "training_step_time": 10.280057430267334
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 3.1765658855438232,
      "step": 148
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 148,
      "training_step_time": 9.936935901641846
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 3.1301229000091553,
      "step": 148
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 148,
      "training_step_time": 9.890270233154297
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 2.4281716346740723,
      "step": 148
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 148,
      "training_step_time": 7.447535514831543
    },
    {
      "epoch": 5.645751953125e-07,
      "model_forward_time": 3.133378744125366,
      "step": 148
    },
    {
      "epoch": 5.645751953125e-07,
      "step": 148,
      "training_step_time": 10.319344520568848
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 3.4637513160705566,
      "step": 149
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 149,
      "training_step_time": 10.237289667129517
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 2.430699586868286,
      "step": 149
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 149,
      "training_step_time": 7.452370643615723
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 2.4313228130340576,
      "step": 149
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 149,
      "training_step_time": 7.450683116912842
    },
    {
      "epoch": 5.68389892578125e-07,
      "model_forward_time": 2.427208185195923,
      "step": 149
    },
    {
      "epoch": 5.68389892578125e-07,
      "step": 149,
      "training_step_time": 10.030056476593018
    },
    {
      "epoch": 5.7220458984375e-07,
      "grad_norm": 0.9007231593132019,
      "learning_rate": 1e-05,
      "loss": 0.2715,
      "step": 150
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 2.4722259044647217,
      "step": 150
    },
    {
      "dynamics_loss_avg": 0.1327960528433323,
      "epoch": 5.7220458984375e-07,
      "step": 150
    },
    {
      "action_loss_avg": 0.12448411881923675,
      "epoch": 5.7220458984375e-07,
      "step": 150
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 150,
      "training_step_time": 7.510247230529785
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 3.1424612998962402,
      "step": 150
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 150,
      "training_step_time": 9.914719343185425
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 3.450059652328491,
      "step": 150
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 150,
      "training_step_time": 10.224339723587036
    },
    {
      "epoch": 5.7220458984375e-07,
      "model_forward_time": 3.1455657482147217,
      "step": 150
    },
    {
      "epoch": 5.7220458984375e-07,
      "step": 150,
      "training_step_time": 10.345205545425415
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 3.1524782180786133,
      "step": 151
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 151,
      "training_step_time": 9.931248426437378
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 2.435589551925659,
      "step": 151
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 151,
      "training_step_time": 7.463848114013672
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 3.1420557498931885,
      "step": 151
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 151,
      "training_step_time": 9.911175966262817
    },
    {
      "epoch": 5.76019287109375e-07,
      "model_forward_time": 2.747209310531616,
      "step": 151
    },
    {
      "epoch": 5.76019287109375e-07,
      "step": 151,
      "training_step_time": 10.035316705703735
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 3.1614465713500977,
      "step": 152
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 152,
      "training_step_time": 9.928894519805908
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 3.1289029121398926,
      "step": 152
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 152,
      "training_step_time": 9.894762754440308
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 2.4344217777252197,
      "step": 152
    },
    {
      "dynamics_loss_avg": 0.1473039098083973,
      "epoch": 5.79833984375e-07,
      "step": 152
    },
    {
      "action_loss_avg": 0.16092545241117479,
      "epoch": 5.79833984375e-07,
      "step": 152
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 152,
      "training_step_time": 7.46941065788269
    },
    {
      "epoch": 5.79833984375e-07,
      "model_forward_time": 3.1419622898101807,
      "step": 152
    },
    {
      "epoch": 5.79833984375e-07,
      "step": 152,
      "training_step_time": 10.32704472541809
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 2.4448626041412354,
      "step": 153
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 153,
      "training_step_time": 7.769912481307983
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 2.4343421459198,
      "step": 153
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 153,
      "training_step_time": 7.466123104095459
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 2.441291570663452,
      "step": 153
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 153,
      "training_step_time": 7.468626260757446
    },
    {
      "epoch": 5.83648681640625e-07,
      "model_forward_time": 3.138364791870117,
      "step": 153
    },
    {
      "epoch": 5.83648681640625e-07,
      "step": 153,
      "training_step_time": 10.270525217056274
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 1.783461570739746,
      "step": 154
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 154,
      "training_step_time": 5.243402719497681
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 2.4363863468170166,
      "step": 154
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 154,
      "training_step_time": 7.464711666107178
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 3.4545936584472656,
      "step": 154
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 154,
      "training_step_time": 10.22709846496582
    },
    {
      "epoch": 5.8746337890625e-07,
      "model_forward_time": 2.433878183364868,
      "step": 154
    },
    {
      "epoch": 5.8746337890625e-07,
      "step": 154,
      "training_step_time": 10.315784454345703
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 3.1459269523620605,
      "step": 155
    },
    {
      "dynamics_loss_avg": 0.13011857122182846,
      "epoch": 5.91278076171875e-07,
      "step": 155
    },
    {
      "action_loss_avg": 0.12882242482155562,
      "epoch": 5.91278076171875e-07,
      "step": 155
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 155,
      "training_step_time": 9.917726278305054
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 3.135061264038086,
      "step": 155
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 155,
      "training_step_time": 9.900688171386719
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 1.7411880493164062,
      "step": 155
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 155,
      "training_step_time": 5.194700717926025
    },
    {
      "epoch": 5.91278076171875e-07,
      "model_forward_time": 3.3758955001831055,
      "step": 155
    },
    {
      "epoch": 5.91278076171875e-07,
      "step": 155,
      "training_step_time": 10.325728178024292
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 2.4869160652160645,
      "step": 156
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 156,
      "training_step_time": 7.519580841064453
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 3.138488292694092,
      "step": 156
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 156,
      "training_step_time": 9.90843939781189
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 2.4297139644622803,
      "step": 156
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 156,
      "training_step_time": 7.454785346984863
    },
    {
      "epoch": 5.950927734375e-07,
      "model_forward_time": 3.130598783493042,
      "step": 156
    },
    {
      "epoch": 5.950927734375e-07,
      "step": 156,
      "training_step_time": 10.291486263275146
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 2.4461169242858887,
      "step": 157
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 157,
      "training_step_time": 7.470693349838257
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 3.4421725273132324,
      "step": 157
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 157,
      "training_step_time": 10.211174964904785
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 2.428838014602661,
      "step": 157
    },
    {
      "dynamics_loss_avg": 0.1310998983681202,
      "epoch": 5.98907470703125e-07,
      "step": 157
    },
    {
      "action_loss_avg": 0.14293652921915054,
      "epoch": 5.98907470703125e-07,
      "step": 157
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 157,
      "training_step_time": 7.462388753890991
    },
    {
      "epoch": 5.98907470703125e-07,
      "model_forward_time": 2.4252049922943115,
      "step": 157
    },
    {
      "epoch": 5.98907470703125e-07,
      "step": 157,
      "training_step_time": 10.036511659622192
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 3.180990219116211,
      "step": 158
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 158,
      "training_step_time": 9.945119142532349
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 2.432349681854248,
      "step": 158
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 158,
      "training_step_time": 7.465074300765991
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 2.737687587738037,
      "step": 158
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 158,
      "training_step_time": 7.7593019008636475
    },
    {
      "epoch": 6.0272216796875e-07,
      "model_forward_time": 3.130753517150879,
      "step": 158
    },
    {
      "epoch": 6.0272216796875e-07,
      "step": 158,
      "training_step_time": 10.304256200790405
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 3.1452789306640625,
      "step": 159
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 159,
      "training_step_time": 9.910987138748169
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 2.4290642738342285,
      "step": 159
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 159,
      "training_step_time": 7.454121828079224
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 2.4296395778656006,
      "step": 159
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 159,
      "training_step_time": 7.457020282745361
    },
    {
      "epoch": 6.06536865234375e-07,
      "model_forward_time": 2.7410826683044434,
      "step": 159
    },
    {
      "epoch": 6.06536865234375e-07,
      "step": 159,
      "training_step_time": 10.319638013839722
    },
    {
      "epoch": 6.103515625e-07,
      "grad_norm": 1.415694236755371,
      "learning_rate": 1e-05,
      "loss": 0.2699,
      "step": 160
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 2.472564220428467,
      "step": 160
    },
    {
      "dynamics_loss_avg": 0.11308788768947124,
      "epoch": 6.103515625e-07,
      "step": 160
    },
    {
      "action_loss_avg": 0.09341623056679964,
      "epoch": 6.103515625e-07,
      "step": 160
    },
    {
      "epoch": 6.103515625e-07,
      "step": 160,
      "training_step_time": 7.5036773681640625
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 3.1294467449188232,
      "step": 160
    },
    {
      "epoch": 6.103515625e-07,
      "step": 160,
      "training_step_time": 9.900206089019775
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 3.1339094638824463,
      "step": 160
    },
    {
      "epoch": 6.103515625e-07,
      "step": 160,
      "training_step_time": 9.90722393989563
    },
    {
      "epoch": 6.103515625e-07,
      "model_forward_time": 2.430384397506714,
      "step": 160
    },
    {
      "epoch": 6.103515625e-07,
      "step": 160,
      "training_step_time": 10.298402070999146
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 2.0404603481292725,
      "step": 161
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 161,
      "training_step_time": 5.526968955993652
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 2.448806047439575,
      "step": 161
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 161,
      "training_step_time": 7.486682415008545
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 2.4383788108825684,
      "step": 161
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 161,
      "training_step_time": 7.4699013233184814
    },
    {
      "epoch": 6.14166259765625e-07,
      "model_forward_time": 2.4404520988464355,
      "step": 161
    },
    {
      "epoch": 6.14166259765625e-07,
      "step": 161,
      "training_step_time": 10.308953762054443
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 2.4752182960510254,
      "step": 162
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 162,
      "training_step_time": 7.508134126663208
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 3.1527047157287598,
      "step": 162
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 162,
      "training_step_time": 9.929567337036133
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 3.4764633178710938,
      "step": 162
    },
    {
      "dynamics_loss_avg": 0.11942607834935189,
      "epoch": 6.1798095703125e-07,
      "step": 162
    },
    {
      "action_loss_avg": 0.11800490580499172,
      "epoch": 6.1798095703125e-07,
      "step": 162
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 162,
      "training_step_time": 10.259807348251343
    },
    {
      "epoch": 6.1798095703125e-07,
      "model_forward_time": 2.4455995559692383,
      "step": 162
    },
    {
      "epoch": 6.1798095703125e-07,
      "step": 162,
      "training_step_time": 10.063642501831055
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 2.478522300720215,
      "step": 163
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 163,
      "training_step_time": 7.511051893234253
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 3.1441755294799805,
      "step": 163
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 163,
      "training_step_time": 9.921513319015503
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 3.1435344219207764,
      "step": 163
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 163,
      "training_step_time": 9.916352033615112
    },
    {
      "epoch": 6.21795654296875e-07,
      "model_forward_time": 2.70455002784729,
      "step": 163
    },
    {
      "epoch": 6.21795654296875e-07,
      "step": 163,
      "training_step_time": 10.363917589187622
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 3.1748201847076416,
      "step": 164
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 164,
      "training_step_time": 9.945222854614258
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 3.1477420330047607,
      "step": 164
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 164,
      "training_step_time": 9.921066761016846
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 3.147024154663086,
      "step": 164
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 164,
      "training_step_time": 9.922274351119995
    },
    {
      "epoch": 6.256103515625e-07,
      "model_forward_time": 3.145681619644165,
      "step": 164
    },
    {
      "epoch": 6.256103515625e-07,
      "step": 164,
      "training_step_time": 10.038458108901978
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 3.4795479774475098,
      "step": 165
    },
    {
      "dynamics_loss_avg": 0.1189661156386137,
      "epoch": 6.29425048828125e-07,
      "step": 165
    },
    {
      "action_loss_avg": 0.1333477046340704,
      "epoch": 6.29425048828125e-07,
      "step": 165
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 165,
      "training_step_time": 10.25398325920105
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 2.4387052059173584,
      "step": 165
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 165,
      "training_step_time": 7.474025249481201
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 3.1386146545410156,
      "step": 165
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 165,
      "training_step_time": 9.908419370651245
    },
    {
      "epoch": 6.29425048828125e-07,
      "model_forward_time": 3.142695188522339,
      "step": 165
    },
    {
      "epoch": 6.29425048828125e-07,
      "step": 165,
      "training_step_time": 10.34143590927124
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 2.4791159629821777,
      "step": 166
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 166,
      "training_step_time": 7.506273508071899
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 2.715229034423828,
      "step": 166
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 166,
      "training_step_time": 7.743653297424316
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 3.1413979530334473,
      "step": 166
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 166,
      "training_step_time": 9.911547660827637
    },
    {
      "epoch": 6.3323974609375e-07,
      "model_forward_time": 2.4361345767974854,
      "step": 166
    },
    {
      "epoch": 6.3323974609375e-07,
      "step": 166,
      "training_step_time": 10.314694881439209
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 3.2019848823547363,
      "step": 167
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 167,
      "training_step_time": 9.974735260009766
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 3.142935037612915,
      "step": 167
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 167,
      "training_step_time": 9.913007974624634
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 3.4424054622650146,
      "step": 167
    },
    {
      "dynamics_loss_avg": 0.14890413209795952,
      "epoch": 6.37054443359375e-07,
      "step": 167
    },
    {
      "action_loss_avg": 0.14972046632319688,
      "epoch": 6.37054443359375e-07,
      "step": 167
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 167,
      "training_step_time": 10.21586561203003
    },
    {
      "epoch": 6.37054443359375e-07,
      "model_forward_time": 3.1445438861846924,
      "step": 167
    },
    {
      "epoch": 6.37054443359375e-07,
      "step": 167,
      "training_step_time": 10.038968086242676
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 2.4502451419830322,
      "step": 168
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 168,
      "training_step_time": 7.479108095169067
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 3.141859531402588,
      "step": 168
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 168,
      "training_step_time": 9.912757635116577
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 3.140113115310669,
      "step": 168
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 168,
      "training_step_time": 9.913340330123901
    },
    {
      "epoch": 6.40869140625e-07,
      "model_forward_time": 2.4392292499542236,
      "step": 168
    },
    {
      "epoch": 6.40869140625e-07,
      "step": 168,
      "training_step_time": 10.050678491592407
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 2.752110242843628,
      "step": 169
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 169,
      "training_step_time": 7.781419515609741
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 3.127979040145874,
      "step": 169
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 169,
      "training_step_time": 9.890308856964111
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 2.429288148880005,
      "step": 169
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 169,
      "training_step_time": 7.453763008117676
    },
    {
      "epoch": 6.44683837890625e-07,
      "model_forward_time": 2.4299705028533936,
      "step": 169
    },
    {
      "epoch": 6.44683837890625e-07,
      "step": 169,
      "training_step_time": 10.329559564590454
    },
    {
      "epoch": 6.4849853515625e-07,
      "grad_norm": 1.4877629280090332,
      "learning_rate": 1e-05,
      "loss": 0.2784,
      "step": 170
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 2.462475061416626,
      "step": 170
    },
    {
      "dynamics_loss_avg": 0.1676573358476162,
      "epoch": 6.4849853515625e-07,
      "step": 170
    },
    {
      "action_loss_avg": 0.188273436576128,
      "epoch": 6.4849853515625e-07,
      "step": 170
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 170,
      "training_step_time": 7.504775047302246
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 2.763523817062378,
      "step": 170
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 170,
      "training_step_time": 7.79242730140686
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 3.1504499912261963,
      "step": 170
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 170,
      "training_step_time": 9.927934646606445
    },
    {
      "epoch": 6.4849853515625e-07,
      "model_forward_time": 1.7509570121765137,
      "step": 170
    },
    {
      "epoch": 6.4849853515625e-07,
      "step": 170,
      "training_step_time": 10.277661800384521
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 3.1925923824310303,
      "step": 171
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 171,
      "training_step_time": 9.964558124542236
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 2.437680959701538,
      "step": 171
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 171,
      "training_step_time": 7.465901851654053
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 3.1474246978759766,
      "step": 171
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 171,
      "training_step_time": 10.227784156799316
    },
    {
      "epoch": 6.52313232421875e-07,
      "model_forward_time": 3.1470751762390137,
      "step": 171
    },
    {
      "epoch": 6.52313232421875e-07,
      "step": 171,
      "training_step_time": 10.286169290542603
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 2.4520299434661865,
      "step": 172
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 172,
      "training_step_time": 7.501118421554565
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 2.431267738342285,
      "step": 172
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 172,
      "training_step_time": 7.462655782699585
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 2.428388833999634,
      "step": 172
    },
    {
      "dynamics_loss_avg": 0.11448791101574898,
      "epoch": 6.561279296875e-07,
      "step": 172
    },
    {
      "action_loss_avg": 0.12606592290103436,
      "epoch": 6.561279296875e-07,
      "step": 172
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 172,
      "training_step_time": 7.4610044956207275
    },
    {
      "epoch": 6.561279296875e-07,
      "model_forward_time": 2.4304933547973633,
      "step": 172
    },
    {
      "epoch": 6.561279296875e-07,
      "step": 172,
      "training_step_time": 10.266642570495605
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 3.4644734859466553,
      "step": 173
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 173,
      "training_step_time": 10.232107877731323
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 3.1319916248321533,
      "step": 173
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 173,
      "training_step_time": 9.892975330352783
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 2.4293172359466553,
      "step": 173
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 173,
      "training_step_time": 7.445499658584595
    },
    {
      "epoch": 6.59942626953125e-07,
      "model_forward_time": 3.131333112716675,
      "step": 173
    },
    {
      "epoch": 6.59942626953125e-07,
      "step": 173,
      "training_step_time": 10.32633376121521
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 2.468071222305298,
      "step": 174
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 174,
      "training_step_time": 7.490451335906982
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 3.4071013927459717,
      "step": 174
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 174,
      "training_step_time": 10.175397872924805
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 3.132803440093994,
      "step": 174
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 174,
      "training_step_time": 9.900062799453735
    },
    {
      "epoch": 6.6375732421875e-07,
      "model_forward_time": 3.1356685161590576,
      "step": 174
    },
    {
      "epoch": 6.6375732421875e-07,
      "step": 174,
      "training_step_time": 10.291397333145142
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 2.471147298812866,
      "step": 175
    },
    {
      "dynamics_loss_avg": 0.12745337933301926,
      "epoch": 6.67572021484375e-07,
      "step": 175
    },
    {
      "action_loss_avg": 0.14367863796651364,
      "epoch": 6.67572021484375e-07,
      "step": 175
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 175,
      "training_step_time": 7.52149772644043
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 2.4388370513916016,
      "step": 175
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 175,
      "training_step_time": 7.461106300354004
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 3.440922260284424,
      "step": 175
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 175,
      "training_step_time": 10.208176851272583
    },
    {
      "epoch": 6.67572021484375e-07,
      "model_forward_time": 3.1415412425994873,
      "step": 175
    },
    {
      "epoch": 6.67572021484375e-07,
      "step": 175,
      "training_step_time": 10.028076410293579
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 3.1821935176849365,
      "step": 176
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 176,
      "training_step_time": 9.952593803405762
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 3.134075880050659,
      "step": 176
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 176,
      "training_step_time": 9.899313926696777
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 3.1420907974243164,
      "step": 176
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 176,
      "training_step_time": 9.90977168083191
    },
    {
      "epoch": 6.7138671875e-07,
      "model_forward_time": 3.461232900619507,
      "step": 176
    },
    {
      "epoch": 6.7138671875e-07,
      "step": 176,
      "training_step_time": 10.323305368423462
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 2.464632749557495,
      "step": 177
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 177,
      "training_step_time": 7.490457057952881
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 3.136540412902832,
      "step": 177
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 177,
      "training_step_time": 9.90629267692566
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 3.1316516399383545,
      "step": 177
    },
    {
      "dynamics_loss_avg": 0.11406724713742733,
      "epoch": 6.75201416015625e-07,
      "step": 177
    },
    {
      "action_loss_avg": 0.1317765086889267,
      "epoch": 6.75201416015625e-07,
      "step": 177
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 177,
      "training_step_time": 9.901789426803589
    },
    {
      "epoch": 6.75201416015625e-07,
      "model_forward_time": 2.4305763244628906,
      "step": 177
    },
    {
      "epoch": 6.75201416015625e-07,
      "step": 177,
      "training_step_time": 10.33027982711792
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 3.145095109939575,
      "step": 178
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 178,
      "training_step_time": 10.192175388336182
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 3.1285183429718018,
      "step": 178
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 178,
      "training_step_time": 9.896532535552979
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 2.433112859725952,
      "step": 178
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 178,
      "training_step_time": 7.468280553817749
    },
    {
      "epoch": 6.7901611328125e-07,
      "model_forward_time": 2.4271717071533203,
      "step": 178
    },
    {
      "epoch": 6.7901611328125e-07,
      "step": 178,
      "training_step_time": 10.303219556808472
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 3.137298822402954,
      "step": 179
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 179,
      "training_step_time": 9.90212106704712
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 3.1307332515716553,
      "step": 179
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 179,
      "training_step_time": 10.184218645095825
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 2.4319159984588623,
      "step": 179
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 179,
      "training_step_time": 7.461691379547119
    },
    {
      "epoch": 6.82830810546875e-07,
      "model_forward_time": 2.433213949203491,
      "step": 179
    },
    {
      "epoch": 6.82830810546875e-07,
      "step": 179,
      "training_step_time": 10.317431926727295
    },
    {
      "epoch": 6.866455078125e-07,
      "grad_norm": 1.3370169401168823,
      "learning_rate": 1e-05,
      "loss": 0.2581,
      "step": 180
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 2.484952688217163,
      "step": 180
    },
    {
      "dynamics_loss_avg": 0.133175278455019,
      "epoch": 6.866455078125e-07,
      "step": 180
    },
    {
      "action_loss_avg": 0.15635462440550327,
      "epoch": 6.866455078125e-07,
      "step": 180
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 180,
      "training_step_time": 7.52131199836731
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 3.142315149307251,
      "step": 180
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 180,
      "training_step_time": 9.915210008621216
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 2.4433753490448,
      "step": 180
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 180,
      "training_step_time": 7.476571083068848
    },
    {
      "epoch": 6.866455078125e-07,
      "model_forward_time": 3.4446041584014893,
      "step": 180
    },
    {
      "epoch": 6.866455078125e-07,
      "step": 180,
      "training_step_time": 10.31382703781128
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 3.163576364517212,
      "step": 181
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 181,
      "training_step_time": 9.935396432876587
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 3.1434476375579834,
      "step": 181
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 181,
      "training_step_time": 9.915116786956787
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 3.142012119293213,
      "step": 181
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 181,
      "training_step_time": 9.9138925075531
    },
    {
      "epoch": 6.90460205078125e-07,
      "model_forward_time": 2.4393563270568848,
      "step": 181
    },
    {
      "epoch": 6.90460205078125e-07,
      "step": 181,
      "training_step_time": 10.278450012207031
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 3.4415228366851807,
      "step": 182
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 182,
      "training_step_time": 10.212094068527222
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 3.1431140899658203,
      "step": 182
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 182,
      "training_step_time": 9.91320252418518
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 3.13987398147583,
      "step": 182
    },
    {
      "dynamics_loss_avg": 0.11612455174326897,
      "epoch": 6.9427490234375e-07,
      "step": 182
    },
    {
      "action_loss_avg": 0.1357668239623308,
      "epoch": 6.9427490234375e-07,
      "step": 182
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 182,
      "training_step_time": 9.91011929512024
    },
    {
      "epoch": 6.9427490234375e-07,
      "model_forward_time": 2.443861961364746,
      "step": 182
    },
    {
      "epoch": 6.9427490234375e-07,
      "step": 182,
      "training_step_time": 10.044553995132446
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 2.482869863510132,
      "step": 183
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 183,
      "training_step_time": 7.511308431625366
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 3.445235013961792,
      "step": 183
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 183,
      "training_step_time": 10.216569423675537
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 3.145637035369873,
      "step": 183
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 183,
      "training_step_time": 9.922485828399658
    },
    {
      "epoch": 6.98089599609375e-07,
      "model_forward_time": 3.144413948059082,
      "step": 183
    },
    {
      "epoch": 6.98089599609375e-07,
      "step": 183,
      "training_step_time": 10.29513669013977
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 2.455143451690674,
      "step": 184
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 184,
      "training_step_time": 7.498278856277466
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 3.1659433841705322,
      "step": 184
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 184,
      "training_step_time": 9.935725450515747
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 2.7428598403930664,
      "step": 184
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 184,
      "training_step_time": 7.764681339263916
    },
    {
      "epoch": 7.01904296875e-07,
      "model_forward_time": 2.430838108062744,
      "step": 184
    },
    {
      "epoch": 7.01904296875e-07,
      "step": 184,
      "training_step_time": 10.025612831115723
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 2.471082925796509,
      "step": 185
    },
    {
      "dynamics_loss_avg": 0.13542401492595674,
      "epoch": 7.05718994140625e-07,
      "step": 185
    },
    {
      "action_loss_avg": 0.1284879233688116,
      "epoch": 7.05718994140625e-07,
      "step": 185
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 185,
      "training_step_time": 7.497520685195923
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 2.433378219604492,
      "step": 185
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 185,
      "training_step_time": 7.451906681060791
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 2.431525945663452,
      "step": 185
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 185,
      "training_step_time": 7.45875096321106
    },
    {
      "epoch": 7.05718994140625e-07,
      "model_forward_time": 3.133733034133911,
      "step": 185
    },
    {
      "epoch": 7.05718994140625e-07,
      "step": 185,
      "training_step_time": 10.349334716796875
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 3.481794595718384,
      "step": 186
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 186,
      "training_step_time": 10.25277328491211
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 2.4341557025909424,
      "step": 186
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 186,
      "training_step_time": 7.460600852966309
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 3.138111114501953,
      "step": 186
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 186,
      "training_step_time": 9.906009912490845
    },
    {
      "epoch": 7.0953369140625e-07,
      "model_forward_time": 3.141831159591675,
      "step": 186
    },
    {
      "epoch": 7.0953369140625e-07,
      "step": 186,
      "training_step_time": 10.313084840774536
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 3.1828768253326416,
      "step": 187
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 187,
      "training_step_time": 9.954816579818726
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 2.7385995388031006,
      "step": 187
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 187,
      "training_step_time": 7.763427019119263
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 3.136035203933716,
      "step": 187
    },
    {
      "dynamics_loss_avg": 0.11100022718310357,
      "epoch": 7.13348388671875e-07,
      "step": 187
    },
    {
      "action_loss_avg": 0.11588224470615387,
      "epoch": 7.13348388671875e-07,
      "step": 187
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 187,
      "training_step_time": 9.908065795898438
    },
    {
      "epoch": 7.13348388671875e-07,
      "model_forward_time": 2.4343068599700928,
      "step": 187
    },
    {
      "epoch": 7.13348388671875e-07,
      "step": 187,
      "training_step_time": 10.303738832473755
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 3.1650960445404053,
      "step": 188
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 188,
      "training_step_time": 9.933809995651245
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 3.131528615951538,
      "step": 188
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 188,
      "training_step_time": 9.900076150894165
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 2.7044432163238525,
      "step": 188
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 188,
      "training_step_time": 7.729721546173096
    },
    {
      "epoch": 7.171630859375e-07,
      "model_forward_time": 3.1415586471557617,
      "step": 188
    },
    {
      "epoch": 7.171630859375e-07,
      "step": 188,
      "training_step_time": 10.031906604766846
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 3.194312572479248,
      "step": 189
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 189,
      "training_step_time": 9.98586368560791
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 3.14007568359375,
      "step": 189
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 189,
      "training_step_time": 9.911245584487915
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 3.140994071960449,
      "step": 189
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 189,
      "training_step_time": 9.912967443466187
    },
    {
      "epoch": 7.20977783203125e-07,
      "model_forward_time": 3.4325053691864014,
      "step": 189
    },
    {
      "epoch": 7.20977783203125e-07,
      "step": 189,
      "training_step_time": 10.359873294830322
    },
    {
      "epoch": 7.2479248046875e-07,
      "grad_norm": 1.34317147731781,
      "learning_rate": 1e-05,
      "loss": 0.2578,
      "step": 190
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 2.4727864265441895,
      "step": 190
    },
    {
      "dynamics_loss_avg": 0.13411990478634833,
      "epoch": 7.2479248046875e-07,
      "step": 190
    },
    {
      "action_loss_avg": 0.14252515248954295,
      "epoch": 7.2479248046875e-07,
      "step": 190
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 190,
      "training_step_time": 7.505781650543213
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 3.1385998725891113,
      "step": 190
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 190,
      "training_step_time": 9.907816886901855
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 3.1373860836029053,
      "step": 190
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 190,
      "training_step_time": 9.908565521240234
    },
    {
      "epoch": 7.2479248046875e-07,
      "model_forward_time": 2.435991048812866,
      "step": 190
    },
    {
      "epoch": 7.2479248046875e-07,
      "step": 190,
      "training_step_time": 10.336551427841187
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 2.7683403491973877,
      "step": 191
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 191,
      "training_step_time": 7.792776107788086
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 3.132293939590454,
      "step": 191
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 191,
      "training_step_time": 9.897035121917725
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 3.1307425498962402,
      "step": 191
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 191,
      "training_step_time": 9.891551494598389
    },
    {
      "epoch": 7.28607177734375e-07,
      "model_forward_time": 2.433671474456787,
      "step": 191
    },
    {
      "epoch": 7.28607177734375e-07,
      "step": 191,
      "training_step_time": 10.327907085418701
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 2.443920612335205,
      "step": 192
    },
    {
      "epoch": 7.32421875e-07,
      "step": 192,
      "training_step_time": 7.465779542922974
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 1.7402403354644775,
      "step": 192
    },
    {
      "epoch": 7.32421875e-07,
      "step": 192,
      "training_step_time": 5.193459987640381
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 2.721696138381958,
      "step": 192
    },
    {
      "dynamics_loss_avg": 0.12134328037500382,
      "epoch": 7.32421875e-07,
      "step": 192
    },
    {
      "action_loss_avg": 0.11907710209488868,
      "epoch": 7.32421875e-07,
      "step": 192
    },
    {
      "epoch": 7.32421875e-07,
      "step": 192,
      "training_step_time": 7.747162103652954
    },
    {
      "epoch": 7.32421875e-07,
      "model_forward_time": 2.428884267807007,
      "step": 192
    },
    {
      "epoch": 7.32421875e-07,
      "step": 192,
      "training_step_time": 10.265605926513672
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 2.4540090560913086,
      "step": 193
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 193,
      "training_step_time": 7.484724998474121
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 2.436746597290039,
      "step": 193
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 193,
      "training_step_time": 7.4638755321502686
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 2.4359354972839355,
      "step": 193
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 193,
      "training_step_time": 7.466696262359619
    },
    {
      "epoch": 7.36236572265625e-07,
      "model_forward_time": 3.4144108295440674,
      "step": 193
    },
    {
      "epoch": 7.36236572265625e-07,
      "step": 193,
      "training_step_time": 10.303957223892212
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 2.454245090484619,
      "step": 194
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 194,
      "training_step_time": 7.483436107635498
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 2.435603618621826,
      "step": 194
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 194,
      "training_step_time": 7.456010580062866
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 2.432474136352539,
      "step": 194
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 194,
      "training_step_time": 7.458740472793579
    },
    {
      "epoch": 7.4005126953125e-07,
      "model_forward_time": 2.4357218742370605,
      "step": 194
    },
    {
      "epoch": 7.4005126953125e-07,
      "step": 194,
      "training_step_time": 10.312291860580444
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 2.4500527381896973,
      "step": 195
    },
    {
      "dynamics_loss_avg": 0.10994812920689583,
      "epoch": 7.43865966796875e-07,
      "step": 195
    },
    {
      "action_loss_avg": 0.12025040052831173,
      "epoch": 7.43865966796875e-07,
      "step": 195
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 195,
      "training_step_time": 7.480521202087402
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 3.4311177730560303,
      "step": 195
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 195,
      "training_step_time": 10.201448917388916
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 2.445115327835083,
      "step": 195
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 195,
      "training_step_time": 7.475292205810547
    },
    {
      "epoch": 7.43865966796875e-07,
      "model_forward_time": 2.440436363220215,
      "step": 195
    },
    {
      "epoch": 7.43865966796875e-07,
      "step": 195,
      "training_step_time": 10.345126152038574
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 3.1828184127807617,
      "step": 196
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 196,
      "training_step_time": 9.96393370628357
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 2.4318318367004395,
      "step": 196
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 196,
      "training_step_time": 7.459007740020752
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 3.4163572788238525,
      "step": 196
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 196,
      "training_step_time": 10.18576693534851
    },
    {
      "epoch": 7.476806640625e-07,
      "model_forward_time": 2.433685779571533,
      "step": 196
    },
    {
      "epoch": 7.476806640625e-07,
      "step": 196,
      "training_step_time": 10.31138563156128
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 2.440084457397461,
      "step": 197
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 197,
      "training_step_time": 7.466232061386108
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 2.4260034561157227,
      "step": 197
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 197,
      "training_step_time": 7.446960210800171
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 2.4296538829803467,
      "step": 197
    },
    {
      "dynamics_loss_avg": 0.11054391786456108,
      "epoch": 7.51495361328125e-07,
      "step": 197
    },
    {
      "action_loss_avg": 0.09477881686761976,
      "epoch": 7.51495361328125e-07,
      "step": 197
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 197,
      "training_step_time": 7.45485782623291
    },
    {
      "epoch": 7.51495361328125e-07,
      "model_forward_time": 3.1287295818328857,
      "step": 197
    },
    {
      "epoch": 7.51495361328125e-07,
      "step": 197,
      "training_step_time": 10.326063871383667
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 3.40169095993042,
      "step": 198
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 198,
      "training_step_time": 10.166720390319824
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 2.427109956741333,
      "step": 198
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 198,
      "training_step_time": 7.440802812576294
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 3.124798536300659,
      "step": 198
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 198,
      "training_step_time": 9.882262468338013
    },
    {
      "epoch": 7.5531005859375e-07,
      "model_forward_time": 2.426004648208618,
      "step": 198
    },
    {
      "epoch": 7.5531005859375e-07,
      "step": 198,
      "training_step_time": 10.294995307922363
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 3.1673223972320557,
      "step": 199
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 199,
      "training_step_time": 9.930444955825806
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 3.4307615756988525,
      "step": 199
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 199,
      "training_step_time": 10.192745685577393
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 2.433988094329834,
      "step": 199
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 199,
      "training_step_time": 7.470332860946655
    },
    {
      "epoch": 7.59124755859375e-07,
      "model_forward_time": 3.1338865756988525,
      "step": 199
    },
    {
      "epoch": 7.59124755859375e-07,
      "step": 199,
      "training_step_time": 10.029458999633789
    },
    {
      "epoch": 7.62939453125e-07,
      "grad_norm": 1.3483319282531738,
      "learning_rate": 1e-05,
      "loss": 0.2486,
      "step": 200
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 3.1482112407684326,
      "step": 200
    },
    {
      "dynamics_loss_avg": 0.12594194561243058,
      "epoch": 7.62939453125e-07,
      "step": 200
    },
    {
      "action_loss_avg": 0.15438942164182662,
      "epoch": 7.62939453125e-07,
      "step": 200
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 200,
      "training_step_time": 9.922653675079346
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 2.4289357662200928,
      "step": 200
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 200,
      "training_step_time": 7.458790063858032
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 2.697235584259033,
      "step": 200
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 200,
      "training_step_time": 7.728097677230835
    },
    {
      "epoch": 7.62939453125e-07,
      "model_forward_time": 2.427152395248413,
      "step": 200
    },
    {
      "epoch": 7.62939453125e-07,
      "step": 200,
      "training_step_time": 10.313205003738403
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 2.4537858963012695,
      "step": 201
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 201,
      "training_step_time": 7.491276979446411
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 3.1330037117004395,
      "step": 201
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 201,
      "training_step_time": 9.909163236618042
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 2.434973955154419,
      "step": 201
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 201,
      "training_step_time": 7.472471237182617
    },
    {
      "epoch": 7.66754150390625e-07,
      "model_forward_time": 2.7025036811828613,
      "step": 201
    },
    {
      "epoch": 7.66754150390625e-07,
      "step": 201,
      "training_step_time": 10.05968165397644
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 2.4724841117858887,
      "step": 202
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 202,
      "training_step_time": 7.507108211517334
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 3.1353800296783447,
      "step": 202
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 202,
      "training_step_time": 9.920806884765625
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 2.434025526046753,
      "step": 202
    },
    {
      "dynamics_loss_avg": 0.12613190561532975,
      "epoch": 7.7056884765625e-07,
      "step": 202
    },
    {
      "action_loss_avg": 0.12401646226644517,
      "epoch": 7.7056884765625e-07,
      "step": 202
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 202,
      "training_step_time": 7.468033313751221
    },
    {
      "epoch": 7.7056884765625e-07,
      "model_forward_time": 3.1344244480133057,
      "step": 202
    },
    {
      "epoch": 7.7056884765625e-07,
      "step": 202,
      "training_step_time": 10.325768232345581
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 3.464977741241455,
      "step": 203
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 203,
      "training_step_time": 10.241883993148804
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 3.1329102516174316,
      "step": 203
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 203,
      "training_step_time": 9.901962518692017
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 3.135516881942749,
      "step": 203
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 203,
      "training_step_time": 9.904874563217163
    },
    {
      "epoch": 7.74383544921875e-07,
      "model_forward_time": 2.431760787963867,
      "step": 203
    },
    {
      "epoch": 7.74383544921875e-07,
      "step": 203,
      "training_step_time": 10.245174407958984
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 2.4635496139526367,
      "step": 204
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 204,
      "training_step_time": 7.493962526321411
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 1.7375156879425049,
      "step": 204
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 204,
      "training_step_time": 5.190170049667358
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 2.7014319896698,
      "step": 204
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 204,
      "training_step_time": 7.722145080566406
    },
    {
      "epoch": 7.781982421875e-07,
      "model_forward_time": 3.1280412673950195,
      "step": 204
    },
    {
      "epoch": 7.781982421875e-07,
      "step": 204,
      "training_step_time": 10.258647918701172
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 2.478321075439453,
      "step": 205
    },
    {
      "dynamics_loss_avg": 0.12780600897967814,
      "epoch": 7.82012939453125e-07,
      "step": 205
    },
    {
      "action_loss_avg": 0.11778588127344847,
      "epoch": 7.82012939453125e-07,
      "step": 205
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 205,
      "training_step_time": 7.510535001754761
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 3.1319806575775146,
      "step": 205
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 205,
      "training_step_time": 9.904346704483032
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 2.429389715194702,
      "step": 205
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 205,
      "training_step_time": 7.45499587059021
    },
    {
      "epoch": 7.82012939453125e-07,
      "model_forward_time": 3.3737573623657227,
      "step": 205
    },
    {
      "epoch": 7.82012939453125e-07,
      "step": 205,
      "training_step_time": 10.307915925979614
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 2.4407737255096436,
      "step": 206
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 206,
      "training_step_time": 7.462700128555298
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 2.4239351749420166,
      "step": 206
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 206,
      "training_step_time": 7.447067499160767
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 3.125873565673828,
      "step": 206
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 206,
      "training_step_time": 9.891858339309692
    },
    {
      "epoch": 7.8582763671875e-07,
      "model_forward_time": 2.429704189300537,
      "step": 206
    },
    {
      "epoch": 7.8582763671875e-07,
      "step": 206,
      "training_step_time": 10.333329439163208
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 2.4676923751831055,
      "step": 207
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 207,
      "training_step_time": 7.49453067779541
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 2.7355918884277344,
      "step": 207
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 207,
      "training_step_time": 7.763939142227173
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 3.1280415058135986,
      "step": 207
    },
    {
      "dynamics_loss_avg": 0.11591338384896517,
      "epoch": 7.89642333984375e-07,
      "step": 207
    },
    {
      "action_loss_avg": 0.12636599661782383,
      "epoch": 7.89642333984375e-07,
      "step": 207
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 207,
      "training_step_time": 9.904348611831665
    },
    {
      "epoch": 7.89642333984375e-07,
      "model_forward_time": 2.4321799278259277,
      "step": 207
    },
    {
      "epoch": 7.89642333984375e-07,
      "step": 207,
      "training_step_time": 10.049624919891357
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 3.1726605892181396,
      "step": 208
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 208,
      "training_step_time": 9.950554847717285
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 2.4296953678131104,
      "step": 208
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 208,
      "training_step_time": 7.457521915435791
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 3.3769583702087402,
      "step": 208
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 208,
      "training_step_time": 10.149625539779663
    },
    {
      "epoch": 7.9345703125e-07,
      "model_forward_time": 3.138916492462158,
      "step": 208
    },
    {
      "epoch": 7.9345703125e-07,
      "step": 208,
      "training_step_time": 10.335910558700562
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 2.469134569168091,
      "step": 209
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 209,
      "training_step_time": 7.499088764190674
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 2.4296419620513916,
      "step": 209
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 209,
      "training_step_time": 7.454933166503906
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 3.1318368911743164,
      "step": 209
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 209,
      "training_step_time": 9.899834156036377
    },
    {
      "epoch": 7.97271728515625e-07,
      "model_forward_time": 2.7417473793029785,
      "step": 209
    },
    {
      "epoch": 7.97271728515625e-07,
      "step": 209,
      "training_step_time": 10.240385293960571
    },
    {
      "epoch": 8.0108642578125e-07,
      "grad_norm": 1.2941811084747314,
      "learning_rate": 1e-05,
      "loss": 0.2419,
      "step": 210
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 3.1643757820129395,
      "step": 210
    },
    {
      "dynamics_loss_avg": 0.11689796894788743,
      "epoch": 8.0108642578125e-07,
      "step": 210
    },
    {
      "action_loss_avg": 0.15405940115451813,
      "epoch": 8.0108642578125e-07,
      "step": 210
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 210,
      "training_step_time": 9.944013595581055
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 3.1522343158721924,
      "step": 210
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 210,
      "training_step_time": 9.924776792526245
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 3.141649007797241,
      "step": 210
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 210,
      "training_step_time": 9.907553672790527
    },
    {
      "epoch": 8.0108642578125e-07,
      "model_forward_time": 2.4359705448150635,
      "step": 210
    },
    {
      "epoch": 8.0108642578125e-07,
      "step": 210,
      "training_step_time": 10.03857421875
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 3.15661883354187,
      "step": 211
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 211,
      "training_step_time": 10.242983102798462
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 2.440866708755493,
      "step": 211
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 211,
      "training_step_time": 7.472041130065918
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 2.436007261276245,
      "step": 211
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 211,
      "training_step_time": 7.465378761291504
    },
    {
      "epoch": 8.04901123046875e-07,
      "model_forward_time": 3.141960382461548,
      "step": 211
    },
    {
      "epoch": 8.04901123046875e-07,
      "step": 211,
      "training_step_time": 10.308426856994629
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 3.189629554748535,
      "step": 212
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 212,
      "training_step_time": 9.980854272842407
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 2.4447479248046875,
      "step": 212
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 212,
      "training_step_time": 7.484695911407471
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 2.743884801864624,
      "step": 212
    },
    {
      "dynamics_loss_avg": 0.12235120758414268,
      "epoch": 8.087158203125e-07,
      "step": 212
    },
    {
      "action_loss_avg": 0.13654907941818237,
      "epoch": 8.087158203125e-07,
      "step": 212
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 212,
      "training_step_time": 7.7696757316589355
    },
    {
      "epoch": 8.087158203125e-07,
      "model_forward_time": 3.135148525238037,
      "step": 212
    },
    {
      "epoch": 8.087158203125e-07,
      "step": 212,
      "training_step_time": 10.254817008972168
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 3.1593620777130127,
      "step": 213
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 213,
      "training_step_time": 9.93651533126831
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 3.144123077392578,
      "step": 213
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 213,
      "training_step_time": 9.913292646408081
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 2.441814661026001,
      "step": 213
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 213,
      "training_step_time": 7.476246356964111
    },
    {
      "epoch": 8.12530517578125e-07,
      "model_forward_time": 3.431934356689453,
      "step": 213
    },
    {
      "epoch": 8.12530517578125e-07,
      "step": 213,
      "training_step_time": 10.333714962005615
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 3.158729076385498,
      "step": 214
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 214,
      "training_step_time": 9.933365106582642
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 3.151200294494629,
      "step": 214
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 214,
      "training_step_time": 9.92935037612915
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 3.149117946624756,
      "step": 214
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 214,
      "training_step_time": 9.928220272064209
    },
    {
      "epoch": 8.1634521484375e-07,
      "model_forward_time": 2.4388158321380615,
      "step": 214
    },
    {
      "epoch": 8.1634521484375e-07,
      "step": 214,
      "training_step_time": 10.235976696014404
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 3.564340829849243,
      "step": 215
    },
    {
      "dynamics_loss_avg": 0.11155563667416572,
      "epoch": 8.20159912109375e-07,
      "step": 215
    },
    {
      "action_loss_avg": 0.1353973113000393,
      "epoch": 8.20159912109375e-07,
      "step": 215
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 215,
      "training_step_time": 10.361741781234741
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 2.448115587234497,
      "step": 215
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 215,
      "training_step_time": 7.47980260848999
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 3.14440655708313,
      "step": 215
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 215,
      "training_step_time": 9.913988590240479
    },
    {
      "epoch": 8.20159912109375e-07,
      "model_forward_time": 2.437244176864624,
      "step": 215
    },
    {
      "epoch": 8.20159912109375e-07,
      "step": 215,
      "training_step_time": 10.29442548751831
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 3.222283124923706,
      "step": 216
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 216,
      "training_step_time": 10.006686687469482
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 3.5900843143463135,
      "step": 216
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 216,
      "training_step_time": 10.374218702316284
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 3.149090051651001,
      "step": 216
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 216,
      "training_step_time": 9.92325735092163
    },
    {
      "epoch": 8.23974609375e-07,
      "model_forward_time": 3.1395962238311768,
      "step": 216
    },
    {
      "epoch": 8.23974609375e-07,
      "step": 216,
      "training_step_time": 10.252221584320068
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 2.4555110931396484,
      "step": 217
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 217,
      "training_step_time": 7.494698762893677
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 2.438210964202881,
      "step": 217
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 217,
      "training_step_time": 7.465391635894775
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 3.4818484783172607,
      "step": 217
    },
    {
      "dynamics_loss_avg": 0.10907220020890236,
      "epoch": 8.27789306640625e-07,
      "step": 217
    },
    {
      "action_loss_avg": 0.11438383162021637,
      "epoch": 8.27789306640625e-07,
      "step": 217
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 217,
      "training_step_time": 10.263283967971802
    },
    {
      "epoch": 8.27789306640625e-07,
      "model_forward_time": 2.437711477279663,
      "step": 217
    },
    {
      "epoch": 8.27789306640625e-07,
      "step": 217,
      "training_step_time": 10.033491134643555
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 3.159043073654175,
      "step": 218
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 218,
      "training_step_time": 9.952542543411255
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 3.1539270877838135,
      "step": 218
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 218,
      "training_step_time": 9.938174486160278
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 2.4410648345947266,
      "step": 218
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 218,
      "training_step_time": 7.471784591674805
    },
    {
      "epoch": 8.3160400390625e-07,
      "model_forward_time": 2.4399428367614746,
      "step": 218
    },
    {
      "epoch": 8.3160400390625e-07,
      "step": 218,
      "training_step_time": 10.287920713424683
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 2.467745542526245,
      "step": 219
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 219,
      "training_step_time": 7.514888763427734
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 2.450467348098755,
      "step": 219
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 219,
      "training_step_time": 7.485523223876953
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 3.140852451324463,
      "step": 219
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 219,
      "training_step_time": 9.916538715362549
    },
    {
      "epoch": 8.35418701171875e-07,
      "model_forward_time": 3.1412293910980225,
      "step": 219
    },
    {
      "epoch": 8.35418701171875e-07,
      "step": 219,
      "training_step_time": 10.274407625198364
    },
    {
      "epoch": 8.392333984375e-07,
      "grad_norm": 1.0129762887954712,
      "learning_rate": 1e-05,
      "loss": 0.2211,
      "step": 220
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 2.4858906269073486,
      "step": 220
    },
    {
      "dynamics_loss_avg": 0.09844334833323956,
      "epoch": 8.392333984375e-07,
      "step": 220
    },
    {
      "action_loss_avg": 0.09457581751048565,
      "epoch": 8.392333984375e-07,
      "step": 220
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 220,
      "training_step_time": 7.835625648498535
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 3.149820566177368,
      "step": 220
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 220,
      "training_step_time": 9.928772449493408
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 3.1451022624969482,
      "step": 220
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 220,
      "training_step_time": 9.9198739528656
    },
    {
      "epoch": 8.392333984375e-07,
      "model_forward_time": 3.1453025341033936,
      "step": 220
    },
    {
      "epoch": 8.392333984375e-07,
      "step": 220,
      "training_step_time": 10.294560432434082
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 2.4791412353515625,
      "step": 221
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 221,
      "training_step_time": 7.510761976242065
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 3.1412079334259033,
      "step": 221
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 221,
      "training_step_time": 9.914719343185425
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 2.7477023601531982,
      "step": 221
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 221,
      "training_step_time": 7.778392791748047
    },
    {
      "epoch": 8.43048095703125e-07,
      "model_forward_time": 3.1389193534851074,
      "step": 221
    },
    {
      "epoch": 8.43048095703125e-07,
      "step": 221,
      "training_step_time": 10.039783954620361
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 2.4538919925689697,
      "step": 222
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 222,
      "training_step_time": 7.481247186660767
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 2.430812120437622,
      "step": 222
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 222,
      "training_step_time": 7.459108114242554
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 2.4307422637939453,
      "step": 222
    },
    {
      "dynamics_loss_avg": 0.10992964804172516,
      "epoch": 8.4686279296875e-07,
      "step": 222
    },
    {
      "action_loss_avg": 0.10951777715235948,
      "epoch": 8.4686279296875e-07,
      "step": 222
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 222,
      "training_step_time": 7.452568769454956
    },
    {
      "epoch": 8.4686279296875e-07,
      "model_forward_time": 3.4232521057128906,
      "step": 222
    },
    {
      "epoch": 8.4686279296875e-07,
      "step": 222,
      "training_step_time": 10.334155082702637
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 2.4794864654541016,
      "step": 223
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 223,
      "training_step_time": 7.5028393268585205
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 3.1368050575256348,
      "step": 223
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 223,
      "training_step_time": 9.899828910827637
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 3.133638381958008,
      "step": 223
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 223,
      "training_step_time": 9.897863626480103
    },
    {
      "epoch": 8.50677490234375e-07,
      "model_forward_time": 3.138549327850342,
      "step": 223
    },
    {
      "epoch": 8.50677490234375e-07,
      "step": 223,
      "training_step_time": 10.27608346939087
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 2.7237319946289062,
      "step": 224
    },
    {
      "epoch": 8.544921875e-07,
      "step": 224,
      "training_step_time": 7.752466201782227
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 2.4348437786102295,
      "step": 224
    },
    {
      "epoch": 8.544921875e-07,
      "step": 224,
      "training_step_time": 7.45915961265564
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 3.1342549324035645,
      "step": 224
    },
    {
      "epoch": 8.544921875e-07,
      "step": 224,
      "training_step_time": 9.902959823608398
    },
    {
      "epoch": 8.544921875e-07,
      "model_forward_time": 2.441284656524658,
      "step": 224
    },
    {
      "epoch": 8.544921875e-07,
      "step": 224,
      "training_step_time": 10.300917387008667
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 3.1858150959014893,
      "step": 225
    },
    {
      "dynamics_loss_avg": 0.12043787017464638,
      "epoch": 8.58306884765625e-07,
      "step": 225
    },
    {
      "action_loss_avg": 0.14107042290270327,
      "epoch": 8.58306884765625e-07,
      "step": 225
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 225,
      "training_step_time": 9.954478740692139
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 2.769069194793701,
      "step": 225
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 225,
      "training_step_time": 7.796704292297363
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 3.1417689323425293,
      "step": 225
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 225,
      "training_step_time": 9.914000988006592
    },
    {
      "epoch": 8.58306884765625e-07,
      "model_forward_time": 3.141552209854126,
      "step": 225
    },
    {
      "epoch": 8.58306884765625e-07,
      "step": 225,
      "training_step_time": 10.247400522232056
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 3.159295082092285,
      "step": 226
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 226,
      "training_step_time": 9.931882381439209
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 2.434781312942505,
      "step": 226
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 226,
      "training_step_time": 7.457959413528442
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 2.4411559104919434,
      "step": 226
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 226,
      "training_step_time": 7.775381326675415
    },
    {
      "epoch": 8.6212158203125e-07,
      "model_forward_time": 2.435692071914673,
      "step": 226
    },
    {
      "epoch": 8.6212158203125e-07,
      "step": 226,
      "training_step_time": 10.02274203300476
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 2.481196880340576,
      "step": 227
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 227,
      "training_step_time": 7.501362323760986
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 3.13681960105896,
      "step": 227
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 227,
      "training_step_time": 9.903223514556885
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 3.136655330657959,
      "step": 227
    },
    {
      "dynamics_loss_avg": 0.11601781249046325,
      "epoch": 8.65936279296875e-07,
      "step": 227
    },
    {
      "action_loss_avg": 0.1277348168194294,
      "epoch": 8.65936279296875e-07,
      "step": 227
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 227,
      "training_step_time": 9.917582750320435
    },
    {
      "epoch": 8.65936279296875e-07,
      "model_forward_time": 2.438108444213867,
      "step": 227
    },
    {
      "epoch": 8.65936279296875e-07,
      "step": 227,
      "training_step_time": 10.261596918106079
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 3.5158822536468506,
      "step": 228
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 228,
      "training_step_time": 10.295766830444336
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 2.4349513053894043,
      "step": 228
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 228,
      "training_step_time": 7.468569040298462
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 3.1453561782836914,
      "step": 228
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 228,
      "training_step_time": 9.92320704460144
    },
    {
      "epoch": 8.697509765625e-07,
      "model_forward_time": 3.1406517028808594,
      "step": 228
    },
    {
      "epoch": 8.697509765625e-07,
      "step": 228,
      "training_step_time": 10.347454071044922
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 1.7684471607208252,
      "step": 229
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 229,
      "training_step_time": 5.254616022109985
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 2.751499652862549,
      "step": 229
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 229,
      "training_step_time": 7.77344274520874
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 2.4302639961242676,
      "step": 229
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 229,
      "training_step_time": 7.453996896743774
    },
    {
      "epoch": 8.73565673828125e-07,
      "model_forward_time": 3.1355483531951904,
      "step": 229
    },
    {
      "epoch": 8.73565673828125e-07,
      "step": 229,
      "training_step_time": 10.03636384010315
    },
    {
      "epoch": 8.7738037109375e-07,
      "grad_norm": 0.9321823120117188,
      "learning_rate": 1e-05,
      "loss": 0.2407,
      "step": 230
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 3.1682000160217285,
      "step": 230
    },
    {
      "dynamics_loss_avg": 0.11210364699363709,
      "epoch": 8.7738037109375e-07,
      "step": 230
    },
    {
      "action_loss_avg": 0.1266365934163332,
      "epoch": 8.7738037109375e-07,
      "step": 230
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 230,
      "training_step_time": 9.942091703414917
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 2.4383292198181152,
      "step": 230
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 230,
      "training_step_time": 7.466451644897461
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 3.416994571685791,
      "step": 230
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 230,
      "training_step_time": 10.197972059249878
    },
    {
      "epoch": 8.7738037109375e-07,
      "model_forward_time": 2.4383187294006348,
      "step": 230
    },
    {
      "epoch": 8.7738037109375e-07,
      "step": 230,
      "training_step_time": 10.324500799179077
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 3.1643500328063965,
      "step": 231
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 231,
      "training_step_time": 9.931705713272095
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 2.439070224761963,
      "step": 231
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 231,
      "training_step_time": 7.462337255477905
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 3.131437063217163,
      "step": 231
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 231,
      "training_step_time": 9.895545244216919
    },
    {
      "epoch": 8.81195068359375e-07,
      "model_forward_time": 3.1504430770874023,
      "step": 231
    },
    {
      "epoch": 8.81195068359375e-07,
      "step": 231,
      "training_step_time": 10.329649686813354
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 3.162280797958374,
      "step": 232
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 232,
      "training_step_time": 9.934152364730835
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 3.1415724754333496,
      "step": 232
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 232,
      "training_step_time": 9.915941953659058
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 2.4420955181121826,
      "step": 232
    },
    {
      "dynamics_loss_avg": 0.10299476273357869,
      "epoch": 8.85009765625e-07,
      "step": 232
    },
    {
      "action_loss_avg": 0.10197930913418532,
      "epoch": 8.85009765625e-07,
      "step": 232
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 232,
      "training_step_time": 7.475046396255493
    },
    {
      "epoch": 8.85009765625e-07,
      "model_forward_time": 2.4370577335357666,
      "step": 232
    },
    {
      "epoch": 8.85009765625e-07,
      "step": 232,
      "training_step_time": 10.048039674758911
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 2.4842960834503174,
      "step": 233
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 233,
      "training_step_time": 7.51420783996582
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 3.4104089736938477,
      "step": 233
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 233,
      "training_step_time": 10.184753656387329
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 3.135962724685669,
      "step": 233
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 233,
      "training_step_time": 9.90662956237793
    },
    {
      "epoch": 8.88824462890625e-07,
      "model_forward_time": 3.136888265609741,
      "step": 233
    },
    {
      "epoch": 8.88824462890625e-07,
      "step": 233,
      "training_step_time": 10.351596117019653
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 2.4720265865325928,
      "step": 234
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 234,
      "training_step_time": 7.504792928695679
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 3.135148763656616,
      "step": 234
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 234,
      "training_step_time": 9.899969577789307
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 3.393761157989502,
      "step": 234
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 234,
      "training_step_time": 10.161014318466187
    },
    {
      "epoch": 8.9263916015625e-07,
      "model_forward_time": 3.1373090744018555,
      "step": 234
    },
    {
      "epoch": 8.9263916015625e-07,
      "step": 234,
      "training_step_time": 10.29081416130066
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 3.148944854736328,
      "step": 235
    },
    {
      "dynamics_loss_avg": 0.10054848734289408,
      "epoch": 8.96453857421875e-07,
      "step": 235
    },
    {
      "action_loss_avg": 0.11205135192722082,
      "epoch": 8.96453857421875e-07,
      "step": 235
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 235,
      "training_step_time": 9.917627811431885
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 3.1271798610687256,
      "step": 235
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 235,
      "training_step_time": 9.89537000656128
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 2.430037498474121,
      "step": 235
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 235,
      "training_step_time": 7.45555567741394
    },
    {
      "epoch": 8.96453857421875e-07,
      "model_forward_time": 2.727283239364624,
      "step": 235
    },
    {
      "epoch": 8.96453857421875e-07,
      "step": 235,
      "training_step_time": 10.315392255783081
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 2.4646663665771484,
      "step": 236
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 236,
      "training_step_time": 7.490597248077393
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 3.122293472290039,
      "step": 236
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 236,
      "training_step_time": 9.886403560638428
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 3.1303982734680176,
      "step": 236
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 236,
      "training_step_time": 9.900713205337524
    },
    {
      "epoch": 9.002685546875e-07,
      "model_forward_time": 3.1297998428344727,
      "step": 236
    },
    {
      "epoch": 9.002685546875e-07,
      "step": 236,
      "training_step_time": 10.279571056365967
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 2.75006103515625,
      "step": 237
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 237,
      "training_step_time": 7.7841339111328125
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 3.1325879096984863,
      "step": 237
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 237,
      "training_step_time": 9.903684377670288
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 2.432603597640991,
      "step": 237
    },
    {
      "dynamics_loss_avg": 0.12566730380058289,
      "epoch": 9.04083251953125e-07,
      "step": 237
    },
    {
      "action_loss_avg": 0.12425860352814197,
      "epoch": 9.04083251953125e-07,
      "step": 237
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 237,
      "training_step_time": 7.45710825920105
    },
    {
      "epoch": 9.04083251953125e-07,
      "model_forward_time": 3.1330418586730957,
      "step": 237
    },
    {
      "epoch": 9.04083251953125e-07,
      "step": 237,
      "training_step_time": 10.299906253814697
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 3.1821045875549316,
      "step": 238
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 238,
      "training_step_time": 9.946146011352539
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 2.7398650646209717,
      "step": 238
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 238,
      "training_step_time": 7.762654542922974
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 2.43117618560791,
      "step": 238
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 238,
      "training_step_time": 7.45547080039978
    },
    {
      "epoch": 9.0789794921875e-07,
      "model_forward_time": 3.1344239711761475,
      "step": 238
    },
    {
      "epoch": 9.0789794921875e-07,
      "step": 238,
      "training_step_time": 10.30963683128357
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 2.4496982097625732,
      "step": 239
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 239,
      "training_step_time": 7.478479385375977
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 2.4357051849365234,
      "step": 239
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 239,
      "training_step_time": 7.4626054763793945
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 2.432025909423828,
      "step": 239
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 239,
      "training_step_time": 7.722508668899536
    },
    {
      "epoch": 9.11712646484375e-07,
      "model_forward_time": 3.1340882778167725,
      "step": 239
    },
    {
      "epoch": 9.11712646484375e-07,
      "step": 239,
      "training_step_time": 10.28866982460022
    },
    {
      "epoch": 9.1552734375e-07,
      "grad_norm": 1.0233086347579956,
      "learning_rate": 1e-05,
      "loss": 0.2365,
      "step": 240
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 3.1577157974243164,
      "step": 240
    },
    {
      "dynamics_loss_avg": 0.08883626703172923,
      "epoch": 9.1552734375e-07,
      "step": 240
    },
    {
      "action_loss_avg": 0.0889099545776844,
      "epoch": 9.1552734375e-07,
      "step": 240
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 240,
      "training_step_time": 9.92901062965393
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 2.4350061416625977,
      "step": 240
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 240,
      "training_step_time": 7.45787239074707
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 3.137230634689331,
      "step": 240
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 240,
      "training_step_time": 9.900647163391113
    },
    {
      "epoch": 9.1552734375e-07,
      "model_forward_time": 2.4345781803131104,
      "step": 240
    },
    {
      "epoch": 9.1552734375e-07,
      "step": 240,
      "training_step_time": 10.307577848434448
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 3.1544272899627686,
      "step": 241
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 241,
      "training_step_time": 9.922561407089233
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 3.1402645111083984,
      "step": 241
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 241,
      "training_step_time": 9.909319162368774
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 3.1405720710754395,
      "step": 241
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 241,
      "training_step_time": 9.913917541503906
    },
    {
      "epoch": 9.19342041015625e-07,
      "model_forward_time": 1.746471881866455,
      "step": 241
    },
    {
      "epoch": 9.19342041015625e-07,
      "step": 241,
      "training_step_time": 10.325708866119385
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 2.4578824043273926,
      "step": 242
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 242,
      "training_step_time": 7.48486590385437
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 3.4411213397979736,
      "step": 242
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 242,
      "training_step_time": 10.211947202682495
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 2.4376955032348633,
      "step": 242
    },
    {
      "dynamics_loss_avg": 0.1098219558596611,
      "epoch": 9.2315673828125e-07,
      "step": 242
    },
    {
      "action_loss_avg": 0.11701427362859249,
      "epoch": 9.2315673828125e-07,
      "step": 242
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 242,
      "training_step_time": 7.468066453933716
    },
    {
      "epoch": 9.2315673828125e-07,
      "model_forward_time": 2.437410831451416,
      "step": 242
    },
    {
      "epoch": 9.2315673828125e-07,
      "step": 242,
      "training_step_time": 10.332056283950806
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 3.185896873474121,
      "step": 243
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 243,
      "training_step_time": 9.962231636047363
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 2.4398014545440674,
      "step": 243
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 243,
      "training_step_time": 7.469767808914185
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 3.415086507797241,
      "step": 243
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 243,
      "training_step_time": 10.18834400177002
    },
    {
      "epoch": 9.26971435546875e-07,
      "model_forward_time": 2.438624382019043,
      "step": 243
    },
    {
      "epoch": 9.26971435546875e-07,
      "step": 243,
      "training_step_time": 10.037033319473267
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 3.1560475826263428,
      "step": 244
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 244,
      "training_step_time": 9.929778575897217
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 3.1389381885528564,
      "step": 244
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 244,
      "training_step_time": 9.911616563796997
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 3.1393723487854004,
      "step": 244
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 244,
      "training_step_time": 9.91187858581543
    },
    {
      "epoch": 9.307861328125e-07,
      "model_forward_time": 2.4326558113098145,
      "step": 244
    },
    {
      "epoch": 9.307861328125e-07,
      "step": 244,
      "training_step_time": 10.3570077419281
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 2.769287586212158,
      "step": 245
    },
    {
      "dynamics_loss_avg": 0.11157996430993081,
      "epoch": 9.34600830078125e-07,
      "step": 245
    },
    {
      "action_loss_avg": 0.119741615280509,
      "epoch": 9.34600830078125e-07,
      "step": 245
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 245,
      "training_step_time": 7.800398826599121
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 3.128732919692993,
      "step": 245
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 245,
      "training_step_time": 9.895601987838745
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 3.133718252182007,
      "step": 245
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 245,
      "training_step_time": 9.901642322540283
    },
    {
      "epoch": 9.34600830078125e-07,
      "model_forward_time": 3.191310405731201,
      "step": 245
    },
    {
      "epoch": 9.34600830078125e-07,
      "step": 245,
      "training_step_time": 10.271177053451538
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 3.1564443111419678,
      "step": 246
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 246,
      "training_step_time": 10.194710493087769
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 3.1404030323028564,
      "step": 246
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 246,
      "training_step_time": 9.912191152572632
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 2.4401633739471436,
      "step": 246
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 246,
      "training_step_time": 7.471178293228149
    },
    {
      "epoch": 9.3841552734375e-07,
      "model_forward_time": 3.1388206481933594,
      "step": 246
    },
    {
      "epoch": 9.3841552734375e-07,
      "step": 246,
      "training_step_time": 10.32095217704773
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 2.451564311981201,
      "step": 247
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 247,
      "training_step_time": 7.480669975280762
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 3.1466832160949707,
      "step": 247
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 247,
      "training_step_time": 9.916875123977661
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 3.460702657699585,
      "step": 247
    },
    {
      "dynamics_loss_avg": 0.09940437898039818,
      "epoch": 9.42230224609375e-07,
      "step": 247
    },
    {
      "action_loss_avg": 0.10258647650480271,
      "epoch": 9.42230224609375e-07,
      "step": 247
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 247,
      "training_step_time": 10.237545728683472
    },
    {
      "epoch": 9.42230224609375e-07,
      "model_forward_time": 2.442146062850952,
      "step": 247
    },
    {
      "epoch": 9.42230224609375e-07,
      "step": 247,
      "training_step_time": 10.290173768997192
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 2.449522018432617,
      "step": 248
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 248,
      "training_step_time": 7.473816633224487
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 3.1372694969177246,
      "step": 248
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 248,
      "training_step_time": 9.901190042495728
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 2.4360833168029785,
      "step": 248
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 248,
      "training_step_time": 7.459176063537598
    },
    {
      "epoch": 9.46044921875e-07,
      "model_forward_time": 2.6975536346435547,
      "step": 248
    },
    {
      "epoch": 9.46044921875e-07,
      "step": 248,
      "training_step_time": 10.01733684539795
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 2.5124382972717285,
      "step": 249
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 249,
      "training_step_time": 7.5381019115448
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 3.1383235454559326,
      "step": 249
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 249,
      "training_step_time": 9.906728744506836
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 3.136950731277466,
      "step": 249
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 249,
      "training_step_time": 9.90779185295105
    },
    {
      "epoch": 9.49859619140625e-07,
      "model_forward_time": 3.144186019897461,
      "step": 249
    },
    {
      "epoch": 9.49859619140625e-07,
      "step": 249,
      "training_step_time": 10.330176591873169
    },
    {
      "epoch": 9.5367431640625e-07,
      "grad_norm": 1.0976215600967407,
      "learning_rate": 1e-05,
      "loss": 0.223,
      "step": 250
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 3.422006845474243,
      "step": 250
    },
    {
      "dynamics_loss_avg": 0.10932484194636345,
      "epoch": 9.5367431640625e-07,
      "step": 250
    },
    {
      "action_loss_avg": 0.12573754154145717,
      "epoch": 9.5367431640625e-07,
      "step": 250
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 250,
      "training_step_time": 10.201629877090454
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 3.146658182144165,
      "step": 250
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 250,
      "training_step_time": 9.92428708076477
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 3.1454920768737793,
      "step": 250
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 250,
      "training_step_time": 9.92523717880249
    },
    {
      "epoch": 9.5367431640625e-07,
      "model_forward_time": 3.1479291915893555,
      "step": 250
    },
    {
      "epoch": 9.5367431640625e-07,
      "step": 250,
      "training_step_time": 10.042906284332275
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 3.194437265396118,
      "step": 251
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 251,
      "training_step_time": 10.000951766967773
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 3.4563560485839844,
      "step": 251
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 251,
      "training_step_time": 10.228582620620728
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 3.1458165645599365,
      "step": 251
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 251,
      "training_step_time": 9.918302297592163
    },
    {
      "epoch": 9.57489013671875e-07,
      "model_forward_time": 3.1438448429107666,
      "step": 251
    },
    {
      "epoch": 9.57489013671875e-07,
      "step": 251,
      "training_step_time": 10.309546709060669
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 3.1870834827423096,
      "step": 252
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 252,
      "training_step_time": 9.952269315719604
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 3.128962278366089,
      "step": 252
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 252,
      "training_step_time": 9.890958309173584
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 3.381019353866577,
      "step": 252
    },
    {
      "dynamics_loss_avg": 0.07956105694174767,
      "epoch": 9.613037109375e-07,
      "step": 252
    },
    {
      "action_loss_avg": 0.07522276733070613,
      "epoch": 9.613037109375e-07,
      "step": 252
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 252,
      "training_step_time": 10.150643110275269
    },
    {
      "epoch": 9.613037109375e-07,
      "model_forward_time": 3.1332144737243652,
      "step": 252
    },
    {
      "epoch": 9.613037109375e-07,
      "step": 252,
      "training_step_time": 10.015955686569214
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 3.1766018867492676,
      "step": 253
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 253,
      "training_step_time": 9.946346521377563
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 3.1314690113067627,
      "step": 253
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 253,
      "training_step_time": 9.898078441619873
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 3.1333327293395996,
      "step": 253
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 253,
      "training_step_time": 9.904557228088379
    },
    {
      "epoch": 9.65118408203125e-07,
      "model_forward_time": 3.395282506942749,
      "step": 253
    },
    {
      "epoch": 9.65118408203125e-07,
      "step": 253,
      "training_step_time": 10.306528091430664
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 3.1764745712280273,
      "step": 254
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 254,
      "training_step_time": 9.948682308197021
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 3.1346590518951416,
      "step": 254
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 254,
      "training_step_time": 9.90080714225769
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 3.1353631019592285,
      "step": 254
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 254,
      "training_step_time": 9.901631832122803
    },
    {
      "epoch": 9.6893310546875e-07,
      "model_forward_time": 3.1306040287017822,
      "step": 254
    },
    {
      "epoch": 9.6893310546875e-07,
      "step": 254,
      "training_step_time": 10.280823707580566
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 3.1768147945404053,
      "step": 255
    },
    {
      "dynamics_loss_avg": 0.10320596508681774,
      "epoch": 9.72747802734375e-07,
      "step": 255
    },
    {
      "action_loss_avg": 0.11131301932036877,
      "epoch": 9.72747802734375e-07,
      "step": 255
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 255,
      "training_step_time": 9.947099208831787
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 3.1295318603515625,
      "step": 255
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 255,
      "training_step_time": 9.900115489959717
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 3.1388299465179443,
      "step": 255
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 255,
      "training_step_time": 9.912980079650879
    },
    {
      "epoch": 9.72747802734375e-07,
      "model_forward_time": 3.12973690032959,
      "step": 255
    },
    {
      "epoch": 9.72747802734375e-07,
      "step": 255,
      "training_step_time": 10.325785875320435
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 3.454336404800415,
      "step": 256
    },
    {
      "epoch": 9.765625e-07,
      "step": 256,
      "training_step_time": 10.22894549369812
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 3.1354494094848633,
      "step": 256
    },
    {
      "epoch": 9.765625e-07,
      "step": 256,
      "training_step_time": 9.906033992767334
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 3.136376142501831,
      "step": 256
    },
    {
      "epoch": 9.765625e-07,
      "step": 256,
      "training_step_time": 9.90779423713684
    },
    {
      "epoch": 9.765625e-07,
      "model_forward_time": 3.132031202316284,
      "step": 256
    },
    {
      "epoch": 9.765625e-07,
      "step": 256,
      "training_step_time": 10.021580457687378
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 3.167163372039795,
      "step": 257
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 257,
      "training_step_time": 9.939186573028564
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 3.1351747512817383,
      "step": 257
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 257,
      "training_step_time": 10.18056321144104
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 3.137421131134033,
      "step": 257
    },
    {
      "dynamics_loss_avg": 0.12774462699890138,
      "epoch": 9.80377197265625e-07,
      "step": 257
    },
    {
      "action_loss_avg": 0.14276767559349537,
      "epoch": 9.80377197265625e-07,
      "step": 257
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 257,
      "training_step_time": 9.912813901901245
    },
    {
      "epoch": 9.80377197265625e-07,
      "model_forward_time": 3.134901285171509,
      "step": 257
    },
    {
      "epoch": 9.80377197265625e-07,
      "step": 257,
      "training_step_time": 10.306885004043579
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 3.179746627807617,
      "step": 258
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 258,
      "training_step_time": 9.948437929153442
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 3.132326602935791,
      "step": 258
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 258,
      "training_step_time": 9.901594161987305
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 3.4347338676452637,
      "step": 258
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 258,
      "training_step_time": 10.212431907653809
    },
    {
      "epoch": 9.8419189453125e-07,
      "model_forward_time": 3.1343748569488525,
      "step": 258
    },
    {
      "epoch": 9.8419189453125e-07,
      "step": 258,
      "training_step_time": 10.28904938697815
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 3.1675469875335693,
      "step": 259
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 259,
      "training_step_time": 9.93877625465393
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 3.1300432682037354,
      "step": 259
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 259,
      "training_step_time": 9.900971412658691
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 3.130250930786133,
      "step": 259
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 259,
      "training_step_time": 9.9013032913208
    },
    {
      "epoch": 9.88006591796875e-07,
      "model_forward_time": 3.137606143951416,
      "step": 259
    },
    {
      "epoch": 9.88006591796875e-07,
      "step": 259,
      "training_step_time": 10.336787223815918
    },
    {
      "epoch": 9.918212890625e-07,
      "grad_norm": 0.8729602098464966,
      "learning_rate": 1e-05,
      "loss": 0.2269,
      "step": 260
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 3.1558024883270264,
      "step": 260
    },
    {
      "dynamics_loss_avg": 0.09950373508036137,
      "epoch": 9.918212890625e-07,
      "step": 260
    },
    {
      "action_loss_avg": 0.11287193782627583,
      "epoch": 9.918212890625e-07,
      "step": 260
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 260,
      "training_step_time": 9.931929111480713
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 3.1330959796905518,
      "step": 260
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 260,
      "training_step_time": 9.903366565704346
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 3.133476972579956,
      "step": 260
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 260,
      "training_step_time": 9.906428813934326
    },
    {
      "epoch": 9.918212890625e-07,
      "model_forward_time": 3.133125066757202,
      "step": 260
    },
    {
      "epoch": 9.918212890625e-07,
      "step": 260,
      "training_step_time": 10.29113507270813
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 3.4388065338134766,
      "step": 261
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 261,
      "training_step_time": 10.210031747817993
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 3.133756637573242,
      "step": 261
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 261,
      "training_step_time": 9.907069444656372
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 2.43276047706604,
      "step": 261
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 261,
      "training_step_time": 7.460565090179443
    },
    {
      "epoch": 9.95635986328125e-07,
      "model_forward_time": 3.134568929672241,
      "step": 261
    },
    {
      "epoch": 9.95635986328125e-07,
      "step": 261,
      "training_step_time": 10.032439231872559
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 3.1612465381622314,
      "step": 262
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 262,
      "training_step_time": 9.939120054244995
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 3.1298744678497314,
      "step": 262
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 262,
      "training_step_time": 10.18740177154541
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 3.129065990447998,
      "step": 262
    },
    {
      "dynamics_loss_avg": 0.12393600717186928,
      "epoch": 9.9945068359375e-07,
      "step": 262
    },
    {
      "action_loss_avg": 0.13282366320490838,
      "epoch": 9.9945068359375e-07,
      "step": 262
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 262,
      "training_step_time": 9.897042989730835
    },
    {
      "epoch": 9.9945068359375e-07,
      "model_forward_time": 3.1309704780578613,
      "step": 262
    },
    {
      "epoch": 9.9945068359375e-07,
      "step": 262,
      "training_step_time": 10.32306170463562
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 3.1592445373535156,
      "step": 263
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 263,
      "training_step_time": 9.924676656723022
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 3.127427577972412,
      "step": 263
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 263,
      "training_step_time": 9.893002271652222
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 3.3775923252105713,
      "step": 263
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 263,
      "training_step_time": 10.142126083374023
    },
    {
      "epoch": 1.003265380859375e-06,
      "model_forward_time": 3.1290781497955322,
      "step": 263
    },
    {
      "epoch": 1.003265380859375e-06,
      "step": 263,
      "training_step_time": 10.019377946853638
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 3.156351089477539,
      "step": 264
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 264,
      "training_step_time": 9.92723298072815
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 3.1312077045440674,
      "step": 264
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 264,
      "training_step_time": 9.903696537017822
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 3.1339213848114014,
      "step": 264
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 264,
      "training_step_time": 9.902888298034668
    },
    {
      "epoch": 1.007080078125e-06,
      "model_forward_time": 3.4415833950042725,
      "step": 264
    },
    {
      "epoch": 1.007080078125e-06,
      "step": 264,
      "training_step_time": 10.30636191368103
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 3.184563159942627,
      "step": 265
    },
    {
      "dynamics_loss_avg": 0.11419532224535942,
      "epoch": 1.010894775390625e-06,
      "step": 265
    },
    {
      "action_loss_avg": 0.12921819202601909,
      "epoch": 1.010894775390625e-06,
      "step": 265
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 265,
      "training_step_time": 9.960017919540405
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 3.136798858642578,
      "step": 265
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 265,
      "training_step_time": 9.910329341888428
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 3.1390464305877686,
      "step": 265
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 265,
      "training_step_time": 9.908182621002197
    },
    {
      "epoch": 1.010894775390625e-06,
      "model_forward_time": 3.1416335105895996,
      "step": 265
    },
    {
      "epoch": 1.010894775390625e-06,
      "step": 265,
      "training_step_time": 10.32389760017395
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 3.1698553562164307,
      "step": 266
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 266,
      "training_step_time": 10.227546453475952
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 3.1386287212371826,
      "step": 266
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 266,
      "training_step_time": 9.909400463104248
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 3.1350715160369873,
      "step": 266
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 266,
      "training_step_time": 9.908821821212769
    },
    {
      "epoch": 1.01470947265625e-06,
      "model_forward_time": 3.1397805213928223,
      "step": 266
    },
    {
      "epoch": 1.01470947265625e-06,
      "step": 266,
      "training_step_time": 10.04006290435791
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 3.177119255065918,
      "step": 267
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 267,
      "training_step_time": 9.949718952178955
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 3.389165163040161,
      "step": 267
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 267,
      "training_step_time": 10.16463565826416
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 3.137279987335205,
      "step": 267
    },
    {
      "dynamics_loss_avg": 0.11527749486267566,
      "epoch": 1.018524169921875e-06,
      "step": 267
    },
    {
      "action_loss_avg": 0.13885045815259217,
      "epoch": 1.018524169921875e-06,
      "step": 267
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 267,
      "training_step_time": 9.91434621810913
    },
    {
      "epoch": 1.018524169921875e-06,
      "model_forward_time": 3.1335437297821045,
      "step": 267
    },
    {
      "epoch": 1.018524169921875e-06,
      "step": 267,
      "training_step_time": 10.30516004562378
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 3.168322801589966,
      "step": 268
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 268,
      "training_step_time": 9.946729183197021
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 3.135253429412842,
      "step": 268
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 268,
      "training_step_time": 9.905941247940063
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 3.1295859813690186,
      "step": 268
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 268,
      "training_step_time": 10.165891408920288
    },
    {
      "epoch": 1.0223388671875e-06,
      "model_forward_time": 3.1326956748962402,
      "step": 268
    },
    {
      "epoch": 1.0223388671875e-06,
      "step": 268,
      "training_step_time": 10.268682956695557
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 3.171861171722412,
      "step": 269
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 269,
      "training_step_time": 9.941367626190186
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 3.14228892326355,
      "step": 269
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 269,
      "training_step_time": 9.918673515319824
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 3.146866798400879,
      "step": 269
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 269,
      "training_step_time": 9.925283432006836
    },
    {
      "epoch": 1.026153564453125e-06,
      "model_forward_time": 3.4361772537231445,
      "step": 269
    },
    {
      "epoch": 1.026153564453125e-06,
      "step": 269,
      "training_step_time": 10.30562162399292
    },
    {
      "epoch": 1.02996826171875e-06,
      "grad_norm": 0.8105283379554749,
      "learning_rate": 1e-05,
      "loss": 0.2199,
      "step": 270
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 3.1755504608154297,
      "step": 270
    },
    {
      "dynamics_loss_avg": 0.09986583143472672,
      "epoch": 1.02996826171875e-06,
      "step": 270
    },
    {
      "action_loss_avg": 0.09614768363535404,
      "epoch": 1.02996826171875e-06,
      "step": 270
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 270,
      "training_step_time": 9.969330072402954
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 3.1452910900115967,
      "step": 270
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 270,
      "training_step_time": 9.917880773544312
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 3.142834424972534,
      "step": 270
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 270,
      "training_step_time": 9.908828973770142
    },
    {
      "epoch": 1.02996826171875e-06,
      "model_forward_time": 3.141002655029297,
      "step": 270
    },
    {
      "epoch": 1.02996826171875e-06,
      "step": 270,
      "training_step_time": 10.278491735458374
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 3.1959400177001953,
      "step": 271
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 271,
      "training_step_time": 10.297944068908691
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 3.139786720275879,
      "step": 271
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 271,
      "training_step_time": 9.91472339630127
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 3.137566566467285,
      "step": 271
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 271,
      "training_step_time": 9.90733027458191
    },
    {
      "epoch": 1.033782958984375e-06,
      "model_forward_time": 3.1439523696899414,
      "step": 271
    },
    {
      "epoch": 1.033782958984375e-06,
      "step": 271,
      "training_step_time": 10.028294324874878
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 3.1629109382629395,
      "step": 272
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 272,
      "training_step_time": 9.933777093887329
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 3.4418931007385254,
      "step": 272
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 272,
      "training_step_time": 10.21327805519104
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 3.135772943496704,
      "step": 272
    },
    {
      "dynamics_loss_avg": 0.10681978650391102,
      "epoch": 1.03759765625e-06,
      "step": 272
    },
    {
      "action_loss_avg": 0.11669883206486702,
      "epoch": 1.03759765625e-06,
      "step": 272
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 272,
      "training_step_time": 9.910974025726318
    },
    {
      "epoch": 1.03759765625e-06,
      "model_forward_time": 3.1364758014678955,
      "step": 272
    },
    {
      "epoch": 1.03759765625e-06,
      "step": 272,
      "training_step_time": 10.292250156402588
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 3.2196836471557617,
      "step": 273
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 273,
      "training_step_time": 10.008495092391968
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 3.1602087020874023,
      "step": 273
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 273,
      "training_step_time": 9.938835382461548
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 3.410722255706787,
      "step": 273
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 273,
      "training_step_time": 10.181321620941162
    },
    {
      "epoch": 1.041412353515625e-06,
      "model_forward_time": 3.1455912590026855,
      "step": 273
    },
    {
      "epoch": 1.041412353515625e-06,
      "step": 273,
      "training_step_time": 10.327128887176514
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 3.1935806274414062,
      "step": 274
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 274,
      "training_step_time": 9.963284969329834
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 3.136063814163208,
      "step": 274
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 274,
      "training_step_time": 9.90436577796936
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 3.137958526611328,
      "step": 274
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 274,
      "training_step_time": 9.908608198165894
    },
    {
      "epoch": 1.04522705078125e-06,
      "model_forward_time": 3.4648256301879883,
      "step": 274
    },
    {
      "epoch": 1.04522705078125e-06,
      "step": 274,
      "training_step_time": 10.334842205047607
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 3.1926848888397217,
      "step": 275
    },
    {
      "dynamics_loss_avg": 0.1008624404668808,
      "epoch": 1.049041748046875e-06,
      "step": 275
    },
    {
      "action_loss_avg": 0.11813533194363117,
      "epoch": 1.049041748046875e-06,
      "step": 275
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 275,
      "training_step_time": 10.001172304153442
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 3.1405482292175293,
      "step": 275
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 275,
      "training_step_time": 9.917747974395752
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 3.1387298107147217,
      "step": 275
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 275,
      "training_step_time": 9.914204359054565
    },
    {
      "epoch": 1.049041748046875e-06,
      "model_forward_time": 3.136305332183838,
      "step": 275
    },
    {
      "epoch": 1.049041748046875e-06,
      "step": 275,
      "training_step_time": 10.334108591079712
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 3.492487907409668,
      "step": 276
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 276,
      "training_step_time": 10.274813175201416
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 3.140143394470215,
      "step": 276
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 276,
      "training_step_time": 9.914133071899414
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 3.132438898086548,
      "step": 276
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 276,
      "training_step_time": 9.906219005584717
    },
    {
      "epoch": 1.0528564453125e-06,
      "model_forward_time": 3.13610577583313,
      "step": 276
    },
    {
      "epoch": 1.0528564453125e-06,
      "step": 276,
      "training_step_time": 10.30666446685791
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 3.18662166595459,
      "step": 277
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 277,
      "training_step_time": 9.96807074546814
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 3.457956075668335,
      "step": 277
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 277,
      "training_step_time": 10.230093479156494
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 3.141072988510132,
      "step": 277
    },
    {
      "dynamics_loss_avg": 0.10814804062247277,
      "epoch": 1.056671142578125e-06,
      "step": 277
    },
    {
      "action_loss_avg": 0.12368155661970377,
      "epoch": 1.056671142578125e-06,
      "step": 277
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 277,
      "training_step_time": 9.91523551940918
    },
    {
      "epoch": 1.056671142578125e-06,
      "model_forward_time": 3.1398496627807617,
      "step": 277
    },
    {
      "epoch": 1.056671142578125e-06,
      "step": 277,
      "training_step_time": 10.261924505233765
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 3.173933267593384,
      "step": 278
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 278,
      "training_step_time": 9.944291591644287
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 3.1449084281921387,
      "step": 278
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 278,
      "training_step_time": 9.916167736053467
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 3.4586730003356934,
      "step": 278
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 278,
      "training_step_time": 10.229299545288086
    },
    {
      "epoch": 1.06048583984375e-06,
      "model_forward_time": 3.1456618309020996,
      "step": 278
    },
    {
      "epoch": 1.06048583984375e-06,
      "step": 278,
      "training_step_time": 10.290211915969849
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 3.201875686645508,
      "step": 279
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 279,
      "training_step_time": 9.978546380996704
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 3.1417994499206543,
      "step": 279
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 279,
      "training_step_time": 9.922328233718872
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 3.1395277976989746,
      "step": 279
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 279,
      "training_step_time": 9.908703565597534
    },
    {
      "epoch": 1.064300537109375e-06,
      "model_forward_time": 3.3984322547912598,
      "step": 279
    },
    {
      "epoch": 1.064300537109375e-06,
      "step": 279,
      "training_step_time": 10.302146673202515
    },
    {
      "epoch": 1.068115234375e-06,
      "grad_norm": 0.7040178775787354,
      "learning_rate": 1e-05,
      "loss": 0.2101,
      "step": 280
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 3.189884662628174,
      "step": 280
    },
    {
      "dynamics_loss_avg": 0.1071284219622612,
      "epoch": 1.068115234375e-06,
      "step": 280
    },
    {
      "action_loss_avg": 0.11843370273709297,
      "epoch": 1.068115234375e-06,
      "step": 280
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 280,
      "training_step_time": 9.966368675231934
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 3.14469838142395,
      "step": 280
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 280,
      "training_step_time": 9.91473913192749
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 3.1474387645721436,
      "step": 280
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 280,
      "training_step_time": 9.923001289367676
    },
    {
      "epoch": 1.068115234375e-06,
      "model_forward_time": 3.1464643478393555,
      "step": 280
    },
    {
      "epoch": 1.068115234375e-06,
      "step": 280,
      "training_step_time": 10.325584650039673
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 3.4998555183410645,
      "step": 281
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 281,
      "training_step_time": 10.27507209777832
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 3.1459603309631348,
      "step": 281
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 281,
      "training_step_time": 9.918170690536499
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 3.14994740486145,
      "step": 281
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 281,
      "training_step_time": 9.936921834945679
    },
    {
      "epoch": 1.071929931640625e-06,
      "model_forward_time": 3.1432125568389893,
      "step": 281
    },
    {
      "epoch": 1.071929931640625e-06,
      "step": 281,
      "training_step_time": 10.038718461990356
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 3.2098963260650635,
      "step": 282
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 282,
      "training_step_time": 10.007535457611084
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 3.4369969367980957,
      "step": 282
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 282,
      "training_step_time": 10.20880651473999
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 3.136728286743164,
      "step": 282
    },
    {
      "dynamics_loss_avg": 0.11169235520064831,
      "epoch": 1.07574462890625e-06,
      "step": 282
    },
    {
      "action_loss_avg": 0.13966683372855188,
      "epoch": 1.07574462890625e-06,
      "step": 282
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 282,
      "training_step_time": 9.909878730773926
    },
    {
      "epoch": 1.07574462890625e-06,
      "model_forward_time": 3.1439528465270996,
      "step": 282
    },
    {
      "epoch": 1.07574462890625e-06,
      "step": 282,
      "training_step_time": 10.31495714187622
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 3.17496395111084,
      "step": 283
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 283,
      "training_step_time": 9.95070219039917
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 3.131769895553589,
      "step": 283
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 283,
      "training_step_time": 9.900668859481812
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 3.4260175228118896,
      "step": 283
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 283,
      "training_step_time": 10.190245628356934
    },
    {
      "epoch": 1.079559326171875e-06,
      "model_forward_time": 3.1317286491394043,
      "step": 283
    },
    {
      "epoch": 1.079559326171875e-06,
      "step": 283,
      "training_step_time": 10.251577615737915
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 3.204409599304199,
      "step": 284
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 284,
      "training_step_time": 9.99399995803833
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 3.1376163959503174,
      "step": 284
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 284,
      "training_step_time": 9.906434059143066
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 3.1344714164733887,
      "step": 284
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 284,
      "training_step_time": 9.899151802062988
    },
    {
      "epoch": 1.0833740234375e-06,
      "model_forward_time": 3.4502198696136475,
      "step": 284
    },
    {
      "epoch": 1.0833740234375e-06,
      "step": 284,
      "training_step_time": 10.319641590118408
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 3.181873083114624,
      "step": 285
    },
    {
      "dynamics_loss_avg": 0.11582472324371337,
      "epoch": 1.087188720703125e-06,
      "step": 285
    },
    {
      "action_loss_avg": 0.12188241891562938,
      "epoch": 1.087188720703125e-06,
      "step": 285
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 285,
      "training_step_time": 9.957600116729736
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 3.1324708461761475,
      "step": 285
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 285,
      "training_step_time": 9.899924039840698
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 3.13857102394104,
      "step": 285
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 285,
      "training_step_time": 9.911739826202393
    },
    {
      "epoch": 1.087188720703125e-06,
      "model_forward_time": 3.133435010910034,
      "step": 285
    },
    {
      "epoch": 1.087188720703125e-06,
      "step": 285,
      "training_step_time": 10.35190749168396
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 3.170877456665039,
      "step": 286
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 286,
      "training_step_time": 10.265246391296387
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 3.1323323249816895,
      "step": 286
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 286,
      "training_step_time": 9.908067464828491
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 3.1360349655151367,
      "step": 286
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 286,
      "training_step_time": 9.91358208656311
    },
    {
      "epoch": 1.09100341796875e-06,
      "model_forward_time": 3.1297106742858887,
      "step": 286
    },
    {
      "epoch": 1.09100341796875e-06,
      "step": 286,
      "training_step_time": 10.254388809204102
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 3.180342435836792,
      "step": 287
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 287,
      "training_step_time": 9.956376552581787
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 3.436730146408081,
      "step": 287
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 287,
      "training_step_time": 10.20585298538208
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 3.1413848400115967,
      "step": 287
    },
    {
      "dynamics_loss_avg": 0.09945889748632908,
      "epoch": 1.094818115234375e-06,
      "step": 287
    },
    {
      "action_loss_avg": 0.1217647347599268,
      "epoch": 1.094818115234375e-06,
      "step": 287
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 287,
      "training_step_time": 9.920721054077148
    },
    {
      "epoch": 1.094818115234375e-06,
      "model_forward_time": 3.129988431930542,
      "step": 287
    },
    {
      "epoch": 1.094818115234375e-06,
      "step": 287,
      "training_step_time": 10.243245363235474
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 3.179347276687622,
      "step": 288
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 288,
      "training_step_time": 9.945821046829224
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 3.1362693309783936,
      "step": 288
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 288,
      "training_step_time": 9.904125213623047
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 3.1358399391174316,
      "step": 288
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 288,
      "training_step_time": 10.170374155044556
    },
    {
      "epoch": 1.0986328125e-06,
      "model_forward_time": 3.136483907699585,
      "step": 288
    },
    {
      "epoch": 1.0986328125e-06,
      "step": 288,
      "training_step_time": 10.222341060638428
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 3.173718214035034,
      "step": 289
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 289,
      "training_step_time": 9.949146509170532
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 3.1401636600494385,
      "step": 289
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 289,
      "training_step_time": 9.910415649414062
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 3.1387832164764404,
      "step": 289
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 289,
      "training_step_time": 9.912867307662964
    },
    {
      "epoch": 1.102447509765625e-06,
      "model_forward_time": 3.432129144668579,
      "step": 289
    },
    {
      "epoch": 1.102447509765625e-06,
      "step": 289,
      "training_step_time": 10.317116022109985
    },
    {
      "epoch": 1.10626220703125e-06,
      "grad_norm": 1.1251235008239746,
      "learning_rate": 1e-05,
      "loss": 0.2229,
      "step": 290
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 3.1945464611053467,
      "step": 290
    },
    {
      "dynamics_loss_avg": 0.10249531865119935,
      "epoch": 1.10626220703125e-06,
      "step": 290
    },
    {
      "action_loss_avg": 0.11864289045333862,
      "epoch": 1.10626220703125e-06,
      "step": 290
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 290,
      "training_step_time": 9.978569507598877
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 3.139188528060913,
      "step": 290
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 290,
      "training_step_time": 9.910271167755127
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 3.1376750469207764,
      "step": 290
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 290,
      "training_step_time": 9.911257266998291
    },
    {
      "epoch": 1.10626220703125e-06,
      "model_forward_time": 3.1378588676452637,
      "step": 290
    },
    {
      "epoch": 1.10626220703125e-06,
      "step": 290,
      "training_step_time": 10.318390130996704
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 3.1754491329193115,
      "step": 291
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 291,
      "training_step_time": 10.22084379196167
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 3.1514198780059814,
      "step": 291
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 291,
      "training_step_time": 9.926787853240967
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 3.142827033996582,
      "step": 291
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 291,
      "training_step_time": 9.910954236984253
    },
    {
      "epoch": 1.110076904296875e-06,
      "model_forward_time": 3.140486001968384,
      "step": 291
    },
    {
      "epoch": 1.110076904296875e-06,
      "step": 291,
      "training_step_time": 10.280630826950073
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 3.1639883518218994,
      "step": 292
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 292,
      "training_step_time": 9.936859846115112
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 3.451155662536621,
      "step": 292
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 292,
      "training_step_time": 10.220048666000366
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 3.1394033432006836,
      "step": 292
    },
    {
      "dynamics_loss_avg": 0.12386244758963585,
      "epoch": 1.1138916015625e-06,
      "step": 292
    },
    {
      "action_loss_avg": 0.13292962610721587,
      "epoch": 1.1138916015625e-06,
      "step": 292
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 292,
      "training_step_time": 9.910911798477173
    },
    {
      "epoch": 1.1138916015625e-06,
      "model_forward_time": 3.140934944152832,
      "step": 292
    },
    {
      "epoch": 1.1138916015625e-06,
      "step": 292,
      "training_step_time": 10.32933521270752
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 3.1934919357299805,
      "step": 293
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 293,
      "training_step_time": 9.973612308502197
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 3.1382875442504883,
      "step": 293
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 293,
      "training_step_time": 9.910674810409546
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 3.4280292987823486,
      "step": 293
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 293,
      "training_step_time": 10.19966983795166
    },
    {
      "epoch": 1.117706298828125e-06,
      "model_forward_time": 3.138540506362915,
      "step": 293
    },
    {
      "epoch": 1.117706298828125e-06,
      "step": 293,
      "training_step_time": 10.02770471572876
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 3.1858503818511963,
      "step": 294
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 294,
      "training_step_time": 9.95969009399414
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 3.136017084121704,
      "step": 294
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 294,
      "training_step_time": 9.905305862426758
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 3.1373798847198486,
      "step": 294
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 294,
      "training_step_time": 9.909459352493286
    },
    {
      "epoch": 1.12152099609375e-06,
      "model_forward_time": 3.420902729034424,
      "step": 294
    },
    {
      "epoch": 1.12152099609375e-06,
      "step": 294,
      "training_step_time": 10.32998776435852
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 3.184725284576416,
      "step": 295
    },
    {
      "dynamics_loss_avg": 0.1120143748819828,
      "epoch": 1.125335693359375e-06,
      "step": 295
    },
    {
      "action_loss_avg": 0.11617923229932785,
      "epoch": 1.125335693359375e-06,
      "step": 295
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 295,
      "training_step_time": 9.956924200057983
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 3.135615825653076,
      "step": 295
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 295,
      "training_step_time": 9.905635356903076
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 3.1412134170532227,
      "step": 295
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 295,
      "training_step_time": 9.913079977035522
    },
    {
      "epoch": 1.125335693359375e-06,
      "model_forward_time": 3.138364315032959,
      "step": 295
    },
    {
      "epoch": 1.125335693359375e-06,
      "step": 295,
      "training_step_time": 10.334773540496826
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 3.4938952922821045,
      "step": 296
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 296,
      "training_step_time": 10.263270378112793
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 3.136322259902954,
      "step": 296
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 296,
      "training_step_time": 9.908570051193237
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 3.1390321254730225,
      "step": 296
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 296,
      "training_step_time": 9.91139030456543
    },
    {
      "epoch": 1.129150390625e-06,
      "model_forward_time": 3.1381752490997314,
      "step": 296
    },
    {
      "epoch": 1.129150390625e-06,
      "step": 296,
      "training_step_time": 10.318747520446777
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 3.179717779159546,
      "step": 297
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 297,
      "training_step_time": 9.946823120117188
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 3.388343095779419,
      "step": 297
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 297,
      "training_step_time": 10.156327962875366
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 3.143120527267456,
      "step": 297
    },
    {
      "dynamics_loss_avg": 0.10159889124333858,
      "epoch": 1.132965087890625e-06,
      "step": 297
    },
    {
      "action_loss_avg": 0.11116073681041598,
      "epoch": 1.132965087890625e-06,
      "step": 297
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 297,
      "training_step_time": 9.919136762619019
    },
    {
      "epoch": 1.132965087890625e-06,
      "model_forward_time": 3.1403486728668213,
      "step": 297
    },
    {
      "epoch": 1.132965087890625e-06,
      "step": 297,
      "training_step_time": 10.245158195495605
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 3.1493494510650635,
      "step": 298
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 298,
      "training_step_time": 9.921563148498535
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 3.141953229904175,
      "step": 298
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 298,
      "training_step_time": 9.910848140716553
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 3.4555857181549072,
      "step": 298
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 298,
      "training_step_time": 10.224138259887695
    },
    {
      "epoch": 1.13677978515625e-06,
      "model_forward_time": 3.141824722290039,
      "step": 298
    },
    {
      "epoch": 1.13677978515625e-06,
      "step": 298,
      "training_step_time": 10.033662796020508
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 3.1799240112304688,
      "step": 299
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 299,
      "training_step_time": 9.955740213394165
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 3.134749174118042,
      "step": 299
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 299,
      "training_step_time": 9.900786399841309
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 3.1291115283966064,
      "step": 299
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 299,
      "training_step_time": 9.896234273910522
    },
    {
      "epoch": 1.140594482421875e-06,
      "model_forward_time": 3.396467924118042,
      "step": 299
    },
    {
      "epoch": 1.140594482421875e-06,
      "step": 299,
      "training_step_time": 10.261351585388184
    },
    {
      "epoch": 1.1444091796875e-06,
      "grad_norm": 0.7954627871513367,
      "learning_rate": 1e-05,
      "loss": 0.2037,
      "step": 300
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 3.1663613319396973,
      "step": 300
    },
    {
      "dynamics_loss_avg": 0.10042457319796086,
      "epoch": 1.1444091796875e-06,
      "step": 300
    },
    {
      "action_loss_avg": 0.10883713848888873,
      "epoch": 1.1444091796875e-06,
      "step": 300
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 300,
      "training_step_time": 9.93581247329712
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 3.13619065284729,
      "step": 300
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 300,
      "training_step_time": 9.902778148651123
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 3.128007650375366,
      "step": 300
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 300,
      "training_step_time": 9.891887187957764
    },
    {
      "epoch": 1.1444091796875e-06,
      "model_forward_time": 3.1309807300567627,
      "step": 300
    },
    {
      "epoch": 1.1444091796875e-06,
      "step": 300,
      "training_step_time": 10.296534061431885
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 3.419093370437622,
      "step": 301
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 301,
      "training_step_time": 10.184656620025635
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 3.1327426433563232,
      "step": 301
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 301,
      "training_step_time": 9.896936893463135
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 3.1327357292175293,
      "step": 301
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 301,
      "training_step_time": 9.902747392654419
    },
    {
      "epoch": 1.148223876953125e-06,
      "model_forward_time": 3.1352763175964355,
      "step": 301
    },
    {
      "epoch": 1.148223876953125e-06,
      "step": 301,
      "training_step_time": 10.28609013557434
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 3.1617228984832764,
      "step": 302
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 302,
      "training_step_time": 9.935842037200928
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 3.4510490894317627,
      "step": 302
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 302,
      "training_step_time": 10.228172779083252
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 3.136444568634033,
      "step": 302
    },
    {
      "dynamics_loss_avg": 0.11189179718494416,
      "epoch": 1.15203857421875e-06,
      "step": 302
    },
    {
      "action_loss_avg": 0.10868712589144706,
      "epoch": 1.15203857421875e-06,
      "step": 302
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 302,
      "training_step_time": 9.914666891098022
    },
    {
      "epoch": 1.15203857421875e-06,
      "model_forward_time": 3.1389834880828857,
      "step": 302
    },
    {
      "epoch": 1.15203857421875e-06,
      "step": 302,
      "training_step_time": 10.285552024841309
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 3.1796116828918457,
      "step": 303
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 303,
      "training_step_time": 9.95275592803955
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 3.1326568126678467,
      "step": 303
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 303,
      "training_step_time": 9.902482032775879
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 3.3915231227874756,
      "step": 303
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 303,
      "training_step_time": 10.164794445037842
    },
    {
      "epoch": 1.155853271484375e-06,
      "model_forward_time": 3.1349895000457764,
      "step": 303
    },
    {
      "epoch": 1.155853271484375e-06,
      "step": 303,
      "training_step_time": 10.02951717376709
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 3.1813156604766846,
      "step": 304
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 304,
      "training_step_time": 9.95291543006897
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 3.135850191116333,
      "step": 304
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 304,
      "training_step_time": 9.907626628875732
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 3.1330275535583496,
      "step": 304
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 304,
      "training_step_time": 9.908424139022827
    },
    {
      "epoch": 1.15966796875e-06,
      "model_forward_time": 3.438016891479492,
      "step": 304
    },
    {
      "epoch": 1.15966796875e-06,
      "step": 304,
      "training_step_time": 10.304147720336914
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 3.169131278991699,
      "step": 305
    },
    {
      "dynamics_loss_avg": 0.09521084800362586,
      "epoch": 1.163482666015625e-06,
      "step": 305
    },
    {
      "action_loss_avg": 0.0932080302387476,
      "epoch": 1.163482666015625e-06,
      "step": 305
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 305,
      "training_step_time": 9.93837833404541
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 3.128140449523926,
      "step": 305
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 305,
      "training_step_time": 9.899760484695435
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 3.136387825012207,
      "step": 305
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 305,
      "training_step_time": 9.908548831939697
    },
    {
      "epoch": 1.163482666015625e-06,
      "model_forward_time": 3.130812168121338,
      "step": 305
    },
    {
      "epoch": 1.163482666015625e-06,
      "step": 305,
      "training_step_time": 10.300021886825562
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 3.1786181926727295,
      "step": 306
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 306,
      "training_step_time": 10.232104778289795
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 3.1437530517578125,
      "step": 306
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 306,
      "training_step_time": 9.91874623298645
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 3.1464695930480957,
      "step": 306
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 306,
      "training_step_time": 9.920958280563354
    },
    {
      "epoch": 1.16729736328125e-06,
      "model_forward_time": 3.146439790725708,
      "step": 306
    },
    {
      "epoch": 1.16729736328125e-06,
      "step": 306,
      "training_step_time": 10.327182054519653
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 3.1881487369537354,
      "step": 307
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 307,
      "training_step_time": 9.963877439498901
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 3.414510726928711,
      "step": 307
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 307,
      "training_step_time": 10.187781810760498
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 3.147226333618164,
      "step": 307
    },
    {
      "dynamics_loss_avg": 0.10385874584317208,
      "epoch": 1.171112060546875e-06,
      "step": 307
    },
    {
      "action_loss_avg": 0.10546690113842487,
      "epoch": 1.171112060546875e-06,
      "step": 307
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 307,
      "training_step_time": 9.923418283462524
    },
    {
      "epoch": 1.171112060546875e-06,
      "model_forward_time": 3.141852855682373,
      "step": 307
    },
    {
      "epoch": 1.171112060546875e-06,
      "step": 307,
      "training_step_time": 10.251951694488525
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 3.1930320262908936,
      "step": 308
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 308,
      "training_step_time": 9.963247776031494
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 3.143667221069336,
      "step": 308
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 308,
      "training_step_time": 9.914747714996338
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 3.1429357528686523,
      "step": 308
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 308,
      "training_step_time": 10.22117304801941
    },
    {
      "epoch": 1.1749267578125e-06,
      "model_forward_time": 3.142848014831543,
      "step": 308
    },
    {
      "epoch": 1.1749267578125e-06,
      "step": 308,
      "training_step_time": 10.298773050308228
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 3.1831650733947754,
      "step": 309
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 309,
      "training_step_time": 9.954267263412476
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 3.1388487815856934,
      "step": 309
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 309,
      "training_step_time": 9.904551267623901
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 3.137310028076172,
      "step": 309
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 309,
      "training_step_time": 9.905715227127075
    },
    {
      "epoch": 1.178741455078125e-06,
      "model_forward_time": 3.4078011512756348,
      "step": 309
    },
    {
      "epoch": 1.178741455078125e-06,
      "step": 309,
      "training_step_time": 10.277678966522217
    },
    {
      "epoch": 1.18255615234375e-06,
      "grad_norm": 0.8340678215026855,
      "learning_rate": 1e-05,
      "loss": 0.2078,
      "step": 310
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 3.1845409870147705,
      "step": 310
    },
    {
      "dynamics_loss_avg": 0.09957225993275642,
      "epoch": 1.18255615234375e-06,
      "step": 310
    },
    {
      "action_loss_avg": 0.11695663332939148,
      "epoch": 1.18255615234375e-06,
      "step": 310
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 310,
      "training_step_time": 9.95560884475708
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 3.1413486003875732,
      "step": 310
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 310,
      "training_step_time": 9.908295631408691
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 3.1373953819274902,
      "step": 310
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 310,
      "training_step_time": 9.905239343643188
    },
    {
      "epoch": 1.18255615234375e-06,
      "model_forward_time": 3.1415956020355225,
      "step": 310
    },
    {
      "epoch": 1.18255615234375e-06,
      "step": 310,
      "training_step_time": 10.250083684921265
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 3.1718156337738037,
      "step": 311
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 311,
      "training_step_time": 10.235721588134766
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 3.1428563594818115,
      "step": 311
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 311,
      "training_step_time": 9.908283710479736
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 3.149538993835449,
      "step": 311
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 311,
      "training_step_time": 9.919320583343506
    },
    {
      "epoch": 1.186370849609375e-06,
      "model_forward_time": 3.1392695903778076,
      "step": 311
    },
    {
      "epoch": 1.186370849609375e-06,
      "step": 311,
      "training_step_time": 10.289329051971436
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 3.1810152530670166,
      "step": 312
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 312,
      "training_step_time": 9.952201843261719
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 3.4387879371643066,
      "step": 312
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 312,
      "training_step_time": 10.208863735198975
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 3.141981363296509,
      "step": 312
    },
    {
      "dynamics_loss_avg": 0.10263939388096333,
      "epoch": 1.190185546875e-06,
      "step": 312
    },
    {
      "action_loss_avg": 0.10295884031802416,
      "epoch": 1.190185546875e-06,
      "step": 312
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 312,
      "training_step_time": 9.91495656967163
    },
    {
      "epoch": 1.190185546875e-06,
      "model_forward_time": 3.146713972091675,
      "step": 312
    },
    {
      "epoch": 1.190185546875e-06,
      "step": 312,
      "training_step_time": 10.306936025619507
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 3.1788079738616943,
      "step": 313
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 313,
      "training_step_time": 9.955006122589111
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 3.142362356185913,
      "step": 313
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 313,
      "training_step_time": 9.921474933624268
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 3.406500816345215,
      "step": 313
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 313,
      "training_step_time": 10.17877459526062
    },
    {
      "epoch": 1.194000244140625e-06,
      "model_forward_time": 3.145833969116211,
      "step": 313
    },
    {
      "epoch": 1.194000244140625e-06,
      "step": 313,
      "training_step_time": 10.317177534103394
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 3.1841139793395996,
      "step": 314
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 314,
      "training_step_time": 9.962743282318115
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 3.145737409591675,
      "step": 314
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 314,
      "training_step_time": 9.919448137283325
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 3.1445138454437256,
      "step": 314
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 314,
      "training_step_time": 9.913832426071167
    },
    {
      "epoch": 1.19781494140625e-06,
      "model_forward_time": 3.4505021572113037,
      "step": 314
    },
    {
      "epoch": 1.19781494140625e-06,
      "step": 314,
      "training_step_time": 10.313095569610596
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 3.178532600402832,
      "step": 315
    },
    {
      "dynamics_loss_avg": 0.08550930395722389,
      "epoch": 1.201629638671875e-06,
      "step": 315
    },
    {
      "action_loss_avg": 0.07886431589722634,
      "epoch": 1.201629638671875e-06,
      "step": 315
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 315,
      "training_step_time": 9.951030969619751
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 3.1410274505615234,
      "step": 315
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 315,
      "training_step_time": 9.911310911178589
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 3.1424951553344727,
      "step": 315
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 315,
      "training_step_time": 9.911179065704346
    },
    {
      "epoch": 1.201629638671875e-06,
      "model_forward_time": 3.1401772499084473,
      "step": 315
    },
    {
      "epoch": 1.201629638671875e-06,
      "step": 315,
      "training_step_time": 10.292606353759766
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 3.4381697177886963,
      "step": 316
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 316,
      "training_step_time": 10.208333492279053
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 3.142036199569702,
      "step": 316
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 316,
      "training_step_time": 9.909320831298828
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 3.144355535507202,
      "step": 316
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 316,
      "training_step_time": 9.914561748504639
    },
    {
      "epoch": 1.2054443359375e-06,
      "model_forward_time": 2.436580181121826,
      "step": 316
    },
    {
      "epoch": 1.2054443359375e-06,
      "step": 316,
      "training_step_time": 10.343233346939087
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 3.1785528659820557,
      "step": 317
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 317,
      "training_step_time": 9.95071029663086
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 3.1412465572357178,
      "step": 317
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 317,
      "training_step_time": 10.178866863250732
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 3.1433093547821045,
      "step": 317
    },
    {
      "dynamics_loss_avg": 0.08980630226433277,
      "epoch": 1.209259033203125e-06,
      "step": 317
    },
    {
      "action_loss_avg": 0.10313082188367843,
      "epoch": 1.209259033203125e-06,
      "step": 317
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 317,
      "training_step_time": 9.913542032241821
    },
    {
      "epoch": 1.209259033203125e-06,
      "model_forward_time": 3.1405255794525146,
      "step": 317
    },
    {
      "epoch": 1.209259033203125e-06,
      "step": 317,
      "training_step_time": 10.292534828186035
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 3.17812180519104,
      "step": 318
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 318,
      "training_step_time": 9.94774580001831
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 3.142510414123535,
      "step": 318
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 318,
      "training_step_time": 9.914474964141846
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 3.425299882888794,
      "step": 318
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 318,
      "training_step_time": 10.201089143753052
    },
    {
      "epoch": 1.21307373046875e-06,
      "model_forward_time": 3.139098882675171,
      "step": 318
    },
    {
      "epoch": 1.21307373046875e-06,
      "step": 318,
      "training_step_time": 10.261598110198975
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 3.1792638301849365,
      "step": 319
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 319,
      "training_step_time": 9.946038484573364
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 3.1385483741760254,
      "step": 319
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 319,
      "training_step_time": 9.90703558921814
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 3.140390157699585,
      "step": 319
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 319,
      "training_step_time": 9.909809350967407
    },
    {
      "epoch": 1.216888427734375e-06,
      "model_forward_time": 3.1417183876037598,
      "step": 319
    },
    {
      "epoch": 1.216888427734375e-06,
      "step": 319,
      "training_step_time": 10.309746503829956
    },
    {
      "epoch": 1.220703125e-06,
      "grad_norm": 1.0865411758422852,
      "learning_rate": 1e-05,
      "loss": 0.2035,
      "step": 320
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 3.1482303142547607,
      "step": 320
    },
    {
      "dynamics_loss_avg": 0.08874515295028687,
      "epoch": 1.220703125e-06,
      "step": 320
    },
    {
      "action_loss_avg": 0.07906185649335384,
      "epoch": 1.220703125e-06,
      "step": 320
    },
    {
      "epoch": 1.220703125e-06,
      "step": 320,
      "training_step_time": 9.920400142669678
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 3.146721601486206,
      "step": 320
    },
    {
      "epoch": 1.220703125e-06,
      "step": 320,
      "training_step_time": 9.917548418045044
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 3.1433229446411133,
      "step": 320
    },
    {
      "epoch": 1.220703125e-06,
      "step": 320,
      "training_step_time": 9.910016536712646
    },
    {
      "epoch": 1.220703125e-06,
      "model_forward_time": 3.1443774700164795,
      "step": 320
    },
    {
      "epoch": 1.220703125e-06,
      "step": 320,
      "training_step_time": 10.329267024993896
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 3.423187255859375,
      "step": 321
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 321,
      "training_step_time": 10.191015243530273
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 3.1392829418182373,
      "step": 321
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 321,
      "training_step_time": 9.911413192749023
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 3.137230396270752,
      "step": 321
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 321,
      "training_step_time": 9.904786825180054
    },
    {
      "epoch": 1.224517822265625e-06,
      "model_forward_time": 3.1376795768737793,
      "step": 321
    },
    {
      "epoch": 1.224517822265625e-06,
      "step": 321,
      "training_step_time": 10.314510345458984
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 3.1741108894348145,
      "step": 322
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 322,
      "training_step_time": 9.94291639328003
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 2.4372825622558594,
      "step": 322
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 322,
      "training_step_time": 7.72943639755249
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 3.134601593017578,
      "step": 322
    },
    {
      "dynamics_loss_avg": 0.09197415299713611,
      "epoch": 1.22833251953125e-06,
      "step": 322
    },
    {
      "action_loss_avg": 0.0931112889200449,
      "epoch": 1.22833251953125e-06,
      "step": 322
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 322,
      "training_step_time": 9.903619289398193
    },
    {
      "epoch": 1.22833251953125e-06,
      "model_forward_time": 3.140822172164917,
      "step": 322
    },
    {
      "epoch": 1.22833251953125e-06,
      "step": 322,
      "training_step_time": 10.304206609725952
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 3.1678836345672607,
      "step": 323
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 323,
      "training_step_time": 9.935513734817505
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 3.1303608417510986,
      "step": 323
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 323,
      "training_step_time": 9.89397382736206
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 3.380671262741089,
      "step": 323
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 323,
      "training_step_time": 10.147293090820312
    },
    {
      "epoch": 1.232147216796875e-06,
      "model_forward_time": 3.131371021270752,
      "step": 323
    },
    {
      "epoch": 1.232147216796875e-06,
      "step": 323,
      "training_step_time": 10.019824266433716
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 3.168926239013672,
      "step": 324
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 324,
      "training_step_time": 9.934753656387329
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 3.1254749298095703,
      "step": 324
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 324,
      "training_step_time": 9.887609243392944
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 3.1303727626800537,
      "step": 324
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 324,
      "training_step_time": 9.892988681793213
    },
    {
      "epoch": 1.2359619140625e-06,
      "model_forward_time": 3.453428268432617,
      "step": 324
    },
    {
      "epoch": 1.2359619140625e-06,
      "step": 324,
      "training_step_time": 10.313827991485596
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 3.1681792736053467,
      "step": 325
    },
    {
      "dynamics_loss_avg": 0.09917316101491451,
      "epoch": 1.239776611328125e-06,
      "step": 325
    },
    {
      "action_loss_avg": 0.1117215596139431,
      "epoch": 1.239776611328125e-06,
      "step": 325
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 325,
      "training_step_time": 9.944509983062744
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 3.129875421524048,
      "step": 325
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 325,
      "training_step_time": 9.897897720336914
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 3.1451666355133057,
      "step": 325
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 325,
      "training_step_time": 9.930306673049927
    },
    {
      "epoch": 1.239776611328125e-06,
      "model_forward_time": 3.1327857971191406,
      "step": 325
    },
    {
      "epoch": 1.239776611328125e-06,
      "step": 325,
      "training_step_time": 10.328032493591309
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 3.1800923347473145,
      "step": 326
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 326,
      "training_step_time": 10.248414278030396
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 3.1328330039978027,
      "step": 326
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 326,
      "training_step_time": 9.910074234008789
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 3.1358718872070312,
      "step": 326
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 326,
      "training_step_time": 9.912507772445679
    },
    {
      "epoch": 1.24359130859375e-06,
      "model_forward_time": 3.1340763568878174,
      "step": 326
    },
    {
      "epoch": 1.24359130859375e-06,
      "step": 326,
      "training_step_time": 10.35563039779663
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 3.1542301177978516,
      "step": 327
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 327,
      "training_step_time": 9.9320650100708
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 3.400177478790283,
      "step": 327
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 327,
      "training_step_time": 10.179075479507446
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 3.134293794631958,
      "step": 327
    },
    {
      "dynamics_loss_avg": 0.12067868784070016,
      "epoch": 1.247406005859375e-06,
      "step": 327
    },
    {
      "action_loss_avg": 0.11950858011841774,
      "epoch": 1.247406005859375e-06,
      "step": 327
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 327,
      "training_step_time": 9.913515329360962
    },
    {
      "epoch": 1.247406005859375e-06,
      "model_forward_time": 3.1324446201324463,
      "step": 327
    },
    {
      "epoch": 1.247406005859375e-06,
      "step": 327,
      "training_step_time": 10.250733137130737
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 3.171576499938965,
      "step": 328
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 328,
      "training_step_time": 9.95025110244751
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 3.1368789672851562,
      "step": 328
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 328,
      "training_step_time": 9.915004014968872
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 3.1355667114257812,
      "step": 328
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 328,
      "training_step_time": 10.225691795349121
    },
    {
      "epoch": 1.251220703125e-06,
      "model_forward_time": 3.1376113891601562,
      "step": 328
    },
    {
      "epoch": 1.251220703125e-06,
      "step": 328,
      "training_step_time": 10.311551570892334
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 3.162510871887207,
      "step": 329
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 329,
      "training_step_time": 9.943234205245972
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 3.1357948780059814,
      "step": 329
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 329,
      "training_step_time": 9.913350343704224
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 3.134580373764038,
      "step": 329
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 329,
      "training_step_time": 9.91753363609314
    },
    {
      "epoch": 1.255035400390625e-06,
      "model_forward_time": 3.394479751586914,
      "step": 329
    },
    {
      "epoch": 1.255035400390625e-06,
      "step": 329,
      "training_step_time": 10.26345419883728
    },
    {
      "epoch": 1.25885009765625e-06,
      "grad_norm": 1.0698583126068115,
      "learning_rate": 1e-05,
      "loss": 0.1909,
      "step": 330
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 3.1581366062164307,
      "step": 330
    },
    {
      "dynamics_loss_avg": 0.09196437932550908,
      "epoch": 1.25885009765625e-06,
      "step": 330
    },
    {
      "action_loss_avg": 0.10429333075881005,
      "epoch": 1.25885009765625e-06,
      "step": 330
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 330,
      "training_step_time": 9.933613777160645
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 3.134498119354248,
      "step": 330
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 330,
      "training_step_time": 9.907256126403809
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 3.1280555725097656,
      "step": 330
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 330,
      "training_step_time": 9.902588844299316
    },
    {
      "epoch": 1.25885009765625e-06,
      "model_forward_time": 3.133594512939453,
      "step": 330
    },
    {
      "epoch": 1.25885009765625e-06,
      "step": 330,
      "training_step_time": 10.26529860496521
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 3.167346954345703,
      "step": 331
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 331,
      "training_step_time": 10.217800855636597
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 3.1382415294647217,
      "step": 331
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 331,
      "training_step_time": 9.915008068084717
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 3.1411168575286865,
      "step": 331
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 331,
      "training_step_time": 9.915374994277954
    },
    {
      "epoch": 1.262664794921875e-06,
      "model_forward_time": 3.1402313709259033,
      "step": 331
    },
    {
      "epoch": 1.262664794921875e-06,
      "step": 331,
      "training_step_time": 10.29813027381897
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 3.1914031505584717,
      "step": 332
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 332,
      "training_step_time": 9.966336011886597
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 3.4517781734466553,
      "step": 332
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 332,
      "training_step_time": 10.223695039749146
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 3.1355111598968506,
      "step": 332
    },
    {
      "dynamics_loss_avg": 0.11240063086152077,
      "epoch": 1.2664794921875e-06,
      "step": 332
    },
    {
      "action_loss_avg": 0.13143489249050616,
      "epoch": 1.2664794921875e-06,
      "step": 332
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 332,
      "training_step_time": 9.916836023330688
    },
    {
      "epoch": 1.2664794921875e-06,
      "model_forward_time": 3.1400146484375,
      "step": 332
    },
    {
      "epoch": 1.2664794921875e-06,
      "step": 332,
      "training_step_time": 10.03217339515686
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 3.180633783340454,
      "step": 333
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 333,
      "training_step_time": 9.955312252044678
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 3.133314371109009,
      "step": 333
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 333,
      "training_step_time": 9.905618906021118
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 3.4025583267211914,
      "step": 333
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 333,
      "training_step_time": 10.177773237228394
    },
    {
      "epoch": 1.270294189453125e-06,
      "model_forward_time": 3.139251232147217,
      "step": 333
    },
    {
      "epoch": 1.270294189453125e-06,
      "step": 333,
      "training_step_time": 10.316893577575684
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 3.1748220920562744,
      "step": 334
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 334,
      "training_step_time": 9.951958417892456
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 3.1349358558654785,
      "step": 334
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 334,
      "training_step_time": 9.90984296798706
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 3.140636682510376,
      "step": 334
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 334,
      "training_step_time": 9.918156147003174
    },
    {
      "epoch": 1.27410888671875e-06,
      "model_forward_time": 3.4143805503845215,
      "step": 334
    },
    {
      "epoch": 1.27410888671875e-06,
      "step": 334,
      "training_step_time": 10.288452625274658
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 3.1798267364501953,
      "step": 335
    },
    {
      "dynamics_loss_avg": 0.10678760372102261,
      "epoch": 1.277923583984375e-06,
      "step": 335
    },
    {
      "action_loss_avg": 0.10899749584496021,
      "epoch": 1.277923583984375e-06,
      "step": 335
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 335,
      "training_step_time": 9.961564779281616
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 3.141770362854004,
      "step": 335
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 335,
      "training_step_time": 9.926103830337524
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 3.135934591293335,
      "step": 335
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 335,
      "training_step_time": 9.915683507919312
    },
    {
      "epoch": 1.277923583984375e-06,
      "model_forward_time": 3.137667179107666,
      "step": 335
    },
    {
      "epoch": 1.277923583984375e-06,
      "step": 335,
      "training_step_time": 10.32369351387024
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 3.5040817260742188,
      "step": 336
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 336,
      "training_step_time": 10.283670663833618
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 2.4353253841400146,
      "step": 336
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 336,
      "training_step_time": 7.468580722808838
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 3.13543701171875,
      "step": 336
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 336,
      "training_step_time": 9.91205382347107
    },
    {
      "epoch": 1.28173828125e-06,
      "model_forward_time": 3.1392581462860107,
      "step": 336
    },
    {
      "epoch": 1.28173828125e-06,
      "step": 336,
      "training_step_time": 10.308295726776123
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 3.17445969581604,
      "step": 337
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 337,
      "training_step_time": 9.951570510864258
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 3.1379125118255615,
      "step": 337
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 337,
      "training_step_time": 10.194199800491333
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 3.1385536193847656,
      "step": 337
    },
    {
      "dynamics_loss_avg": 0.10036730356514453,
      "epoch": 1.285552978515625e-06,
      "step": 337
    },
    {
      "action_loss_avg": 0.10412824116647243,
      "epoch": 1.285552978515625e-06,
      "step": 337
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 337,
      "training_step_time": 9.915715456008911
    },
    {
      "epoch": 1.285552978515625e-06,
      "model_forward_time": 3.134107828140259,
      "step": 337
    },
    {
      "epoch": 1.285552978515625e-06,
      "step": 337,
      "training_step_time": 10.349612712860107
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 3.1694071292877197,
      "step": 338
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 338,
      "training_step_time": 9.947537422180176
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 3.134523868560791,
      "step": 338
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 338,
      "training_step_time": 9.911877632141113
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 3.434788703918457,
      "step": 338
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 338,
      "training_step_time": 10.211722135543823
    },
    {
      "epoch": 1.28936767578125e-06,
      "model_forward_time": 3.1357953548431396,
      "step": 338
    },
    {
      "epoch": 1.28936767578125e-06,
      "step": 338,
      "training_step_time": 10.305102825164795
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 3.1638906002044678,
      "step": 339
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 339,
      "training_step_time": 9.943605899810791
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 3.1362411975860596,
      "step": 339
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 339,
      "training_step_time": 9.914053440093994
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 3.1461381912231445,
      "step": 339
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 339,
      "training_step_time": 9.924036741256714
    },
    {
      "epoch": 1.293182373046875e-06,
      "model_forward_time": 3.1317834854125977,
      "step": 339
    },
    {
      "epoch": 1.293182373046875e-06,
      "step": 339,
      "training_step_time": 10.278807878494263
    },
    {
      "epoch": 1.2969970703125e-06,
      "grad_norm": 0.8751269578933716,
      "learning_rate": 1e-05,
      "loss": 0.1944,
      "step": 340
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 3.1740634441375732,
      "step": 340
    },
    {
      "dynamics_loss_avg": 0.0999645572155714,
      "epoch": 1.2969970703125e-06,
      "step": 340
    },
    {
      "action_loss_avg": 0.11047307886183262,
      "epoch": 1.2969970703125e-06,
      "step": 340
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 340,
      "training_step_time": 9.952747583389282
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 3.1303629875183105,
      "step": 340
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 340,
      "training_step_time": 9.903208494186401
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 3.133239984512329,
      "step": 340
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 340,
      "training_step_time": 9.90801477432251
    },
    {
      "epoch": 1.2969970703125e-06,
      "model_forward_time": 3.1333882808685303,
      "step": 340
    },
    {
      "epoch": 1.2969970703125e-06,
      "step": 340,
      "training_step_time": 10.264314889907837
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 3.4231510162353516,
      "step": 341
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 341,
      "training_step_time": 10.199007272720337
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 3.137181282043457,
      "step": 341
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 341,
      "training_step_time": 9.915181875228882
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 3.1332221031188965,
      "step": 341
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 341,
      "training_step_time": 9.909700393676758
    },
    {
      "epoch": 1.300811767578125e-06,
      "model_forward_time": 3.135913848876953,
      "step": 341
    },
    {
      "epoch": 1.300811767578125e-06,
      "step": 341,
      "training_step_time": 10.300786256790161
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 3.1847784519195557,
      "step": 342
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 342,
      "training_step_time": 9.963847160339355
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 3.4500439167022705,
      "step": 342
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 342,
      "training_step_time": 10.274008989334106
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 3.139960289001465,
      "step": 342
    },
    {
      "dynamics_loss_avg": 0.07489962577819824,
      "epoch": 1.30462646484375e-06,
      "step": 342
    },
    {
      "action_loss_avg": 0.07299272045493126,
      "epoch": 1.30462646484375e-06,
      "step": 342
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 342,
      "training_step_time": 9.966356992721558
    },
    {
      "epoch": 1.30462646484375e-06,
      "model_forward_time": 3.1337645053863525,
      "step": 342
    },
    {
      "epoch": 1.30462646484375e-06,
      "step": 342,
      "training_step_time": 10.302653551101685
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 3.1823275089263916,
      "step": 343
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 343,
      "training_step_time": 9.961764574050903
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 3.1339707374572754,
      "step": 343
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 343,
      "training_step_time": 9.909629821777344
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 3.4082794189453125,
      "step": 343
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 343,
      "training_step_time": 10.187084197998047
    },
    {
      "epoch": 1.308441162109375e-06,
      "model_forward_time": 3.1383683681488037,
      "step": 343
    },
    {
      "epoch": 1.308441162109375e-06,
      "step": 343,
      "training_step_time": 10.350368022918701
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 3.169674873352051,
      "step": 344
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 344,
      "training_step_time": 9.941866397857666
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 3.136472702026367,
      "step": 344
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 344,
      "training_step_time": 9.907278060913086
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 3.1367461681365967,
      "step": 344
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 344,
      "training_step_time": 9.911508798599243
    },
    {
      "epoch": 1.312255859375e-06,
      "model_forward_time": 3.419459581375122,
      "step": 344
    },
    {
      "epoch": 1.312255859375e-06,
      "step": 344,
      "training_step_time": 10.287333726882935
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 3.1832242012023926,
      "step": 345
    },
    {
      "dynamics_loss_avg": 0.09330754578113556,
      "epoch": 1.316070556640625e-06,
      "step": 345
    },
    {
      "action_loss_avg": 0.08993633426725864,
      "epoch": 1.316070556640625e-06,
      "step": 345
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 345,
      "training_step_time": 9.966830015182495
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 3.1393346786499023,
      "step": 345
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 345,
      "training_step_time": 9.913509368896484
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 3.139172077178955,
      "step": 345
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 345,
      "training_step_time": 9.913346767425537
    },
    {
      "epoch": 1.316070556640625e-06,
      "model_forward_time": 3.137720823287964,
      "step": 345
    },
    {
      "epoch": 1.316070556640625e-06,
      "step": 345,
      "training_step_time": 10.02682113647461
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 3.1764638423919678,
      "step": 346
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 346,
      "training_step_time": 10.277458190917969
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 3.1365556716918945,
      "step": 346
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 346,
      "training_step_time": 9.914516925811768
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 3.144427537918091,
      "step": 346
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 346,
      "training_step_time": 9.920564413070679
    },
    {
      "epoch": 1.31988525390625e-06,
      "model_forward_time": 3.136244535446167,
      "step": 346
    },
    {
      "epoch": 1.31988525390625e-06,
      "step": 346,
      "training_step_time": 10.023009777069092
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 3.1804068088531494,
      "step": 347
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 347,
      "training_step_time": 9.953622341156006
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 3.3985166549682617,
      "step": 347
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 347,
      "training_step_time": 10.174013137817383
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 3.1401326656341553,
      "step": 347
    },
    {
      "dynamics_loss_avg": 0.09233238659799099,
      "epoch": 1.323699951171875e-06,
      "step": 347
    },
    {
      "action_loss_avg": 0.11189581416547298,
      "epoch": 1.323699951171875e-06,
      "step": 347
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 347,
      "training_step_time": 9.922425031661987
    },
    {
      "epoch": 1.323699951171875e-06,
      "model_forward_time": 3.1418254375457764,
      "step": 347
    },
    {
      "epoch": 1.323699951171875e-06,
      "step": 347,
      "training_step_time": 10.35253620147705
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 3.168830633163452,
      "step": 348
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 348,
      "training_step_time": 9.950371742248535
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 3.136085271835327,
      "step": 348
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 348,
      "training_step_time": 9.914772510528564
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 3.129683017730713,
      "step": 348
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 348,
      "training_step_time": 10.19194221496582
    },
    {
      "epoch": 1.3275146484375e-06,
      "model_forward_time": 3.130362033843994,
      "step": 348
    },
    {
      "epoch": 1.3275146484375e-06,
      "step": 348,
      "training_step_time": 10.300890684127808
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 3.1739895343780518,
      "step": 349
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 349,
      "training_step_time": 9.950652837753296
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 3.148566484451294,
      "step": 349
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 349,
      "training_step_time": 9.928137302398682
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 3.1363918781280518,
      "step": 349
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 349,
      "training_step_time": 9.914176225662231
    },
    {
      "epoch": 1.331329345703125e-06,
      "model_forward_time": 3.4235141277313232,
      "step": 349
    },
    {
      "epoch": 1.331329345703125e-06,
      "step": 349,
      "training_step_time": 10.29538106918335
    },
    {
      "epoch": 1.33514404296875e-06,
      "grad_norm": 0.766143262386322,
      "learning_rate": 1e-05,
      "loss": 0.188,
      "step": 350
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 3.175257682800293,
      "step": 350
    },
    {
      "dynamics_loss_avg": 0.09883583486080169,
      "epoch": 1.33514404296875e-06,
      "step": 350
    },
    {
      "action_loss_avg": 0.11512487530708312,
      "epoch": 1.33514404296875e-06,
      "step": 350
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 350,
      "training_step_time": 9.956175804138184
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 3.14284610748291,
      "step": 350
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 350,
      "training_step_time": 9.920936822891235
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 3.1421258449554443,
      "step": 350
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 350,
      "training_step_time": 9.928858757019043
    },
    {
      "epoch": 1.33514404296875e-06,
      "model_forward_time": 3.1432011127471924,
      "step": 350
    },
    {
      "epoch": 1.33514404296875e-06,
      "step": 350,
      "training_step_time": 10.034869194030762
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 3.1788196563720703,
      "step": 351
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 351,
      "training_step_time": 10.230672121047974
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 3.1424648761749268,
      "step": 351
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 351,
      "training_step_time": 9.923645973205566
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 3.1394243240356445,
      "step": 351
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 351,
      "training_step_time": 9.915113687515259
    },
    {
      "epoch": 1.338958740234375e-06,
      "model_forward_time": 3.144486665725708,
      "step": 351
    },
    {
      "epoch": 1.338958740234375e-06,
      "step": 351,
      "training_step_time": 10.274569034576416
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 3.1808176040649414,
      "step": 352
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 352,
      "training_step_time": 9.969611406326294
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 3.4517130851745605,
      "step": 352
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 352,
      "training_step_time": 10.22945761680603
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 3.1387598514556885,
      "step": 352
    },
    {
      "dynamics_loss_avg": 0.0922430444508791,
      "epoch": 1.3427734375e-06,
      "step": 352
    },
    {
      "action_loss_avg": 0.0819792240858078,
      "epoch": 1.3427734375e-06,
      "step": 352
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 352,
      "training_step_time": 9.917156219482422
    },
    {
      "epoch": 1.3427734375e-06,
      "model_forward_time": 3.1385703086853027,
      "step": 352
    },
    {
      "epoch": 1.3427734375e-06,
      "step": 352,
      "training_step_time": 10.286552667617798
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 3.17382550239563,
      "step": 353
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 353,
      "training_step_time": 9.954480648040771
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 3.134212017059326,
      "step": 353
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 353,
      "training_step_time": 9.91212248802185
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 3.396510124206543,
      "step": 353
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 353,
      "training_step_time": 10.170884847640991
    },
    {
      "epoch": 1.346588134765625e-06,
      "model_forward_time": 3.138849973678589,
      "step": 353
    },
    {
      "epoch": 1.346588134765625e-06,
      "step": 353,
      "training_step_time": 10.30972146987915
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 3.163936138153076,
      "step": 354
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 354,
      "training_step_time": 9.939017534255981
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 3.13725209236145,
      "step": 354
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 354,
      "training_step_time": 9.9121732711792
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 3.1394708156585693,
      "step": 354
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 354,
      "training_step_time": 9.916577577590942
    },
    {
      "epoch": 1.35040283203125e-06,
      "model_forward_time": 3.4152700901031494,
      "step": 354
    },
    {
      "epoch": 1.35040283203125e-06,
      "step": 354,
      "training_step_time": 10.293651103973389
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 3.166557788848877,
      "step": 355
    },
    {
      "dynamics_loss_avg": 0.094264180585742,
      "epoch": 1.354217529296875e-06,
      "step": 355
    },
    {
      "action_loss_avg": 0.11006533466279507,
      "epoch": 1.354217529296875e-06,
      "step": 355
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 355,
      "training_step_time": 9.946541547775269
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 3.140547037124634,
      "step": 355
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 355,
      "training_step_time": 9.920823335647583
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 3.1451830863952637,
      "step": 355
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 355,
      "training_step_time": 9.92274785041809
    },
    {
      "epoch": 1.354217529296875e-06,
      "model_forward_time": 3.1395084857940674,
      "step": 355
    },
    {
      "epoch": 1.354217529296875e-06,
      "step": 355,
      "training_step_time": 10.269896984100342
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 3.472820997238159,
      "step": 356
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 356,
      "training_step_time": 10.256464004516602
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 3.140965700149536,
      "step": 356
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 356,
      "training_step_time": 9.916923999786377
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 3.131279230117798,
      "step": 356
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 356,
      "training_step_time": 9.906197547912598
    },
    {
      "epoch": 1.3580322265625e-06,
      "model_forward_time": 3.1283860206604004,
      "step": 356
    },
    {
      "epoch": 1.3580322265625e-06,
      "step": 356,
      "training_step_time": 10.2980477809906
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 3.156726837158203,
      "step": 357
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 357,
      "training_step_time": 9.929669618606567
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 3.379851818084717,
      "step": 357
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 357,
      "training_step_time": 10.152763605117798
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 3.1346590518951416,
      "step": 357
    },
    {
      "dynamics_loss_avg": 0.08744877800345421,
      "epoch": 1.361846923828125e-06,
      "step": 357
    },
    {
      "action_loss_avg": 0.08032402805984021,
      "epoch": 1.361846923828125e-06,
      "step": 357
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 357,
      "training_step_time": 9.912540912628174
    },
    {
      "epoch": 1.361846923828125e-06,
      "model_forward_time": 3.133754014968872,
      "step": 357
    },
    {
      "epoch": 1.361846923828125e-06,
      "step": 357,
      "training_step_time": 10.324696063995361
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 3.167083263397217,
      "step": 358
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 358,
      "training_step_time": 9.94457745552063
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 3.1376445293426514,
      "step": 358
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 358,
      "training_step_time": 9.91196084022522
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 3.4135918617248535,
      "step": 358
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 358,
      "training_step_time": 10.190802812576294
    },
    {
      "epoch": 1.36566162109375e-06,
      "model_forward_time": 3.1367886066436768,
      "step": 358
    },
    {
      "epoch": 1.36566162109375e-06,
      "step": 358,
      "training_step_time": 10.301729202270508
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 3.1503753662109375,
      "step": 359
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 359,
      "training_step_time": 9.937914609909058
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 3.1351449489593506,
      "step": 359
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 359,
      "training_step_time": 9.918738842010498
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 3.1359364986419678,
      "step": 359
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 359,
      "training_step_time": 9.9148588180542
    },
    {
      "epoch": 1.369476318359375e-06,
      "model_forward_time": 2.4316020011901855,
      "step": 359
    },
    {
      "epoch": 1.369476318359375e-06,
      "step": 359,
      "training_step_time": 10.041271448135376
    },
    {
      "epoch": 1.373291015625e-06,
      "grad_norm": 1.0322798490524292,
      "learning_rate": 1e-05,
      "loss": 0.1889,
      "step": 360
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 3.1668050289154053,
      "step": 360
    },
    {
      "dynamics_loss_avg": 0.10250849947333336,
      "epoch": 1.373291015625e-06,
      "step": 360
    },
    {
      "action_loss_avg": 0.09544294551014901,
      "epoch": 1.373291015625e-06,
      "step": 360
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 360,
      "training_step_time": 9.943338632583618
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 3.1322457790374756,
      "step": 360
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 360,
      "training_step_time": 9.906051874160767
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 3.130906343460083,
      "step": 360
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 360,
      "training_step_time": 9.90770936012268
    },
    {
      "epoch": 1.373291015625e-06,
      "model_forward_time": 3.1301867961883545,
      "step": 360
    },
    {
      "epoch": 1.373291015625e-06,
      "step": 360,
      "training_step_time": 10.274420976638794
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 3.422689914703369,
      "step": 361
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 361,
      "training_step_time": 10.193734407424927
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 3.130002737045288,
      "step": 361
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 361,
      "training_step_time": 9.899953126907349
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 3.128514289855957,
      "step": 361
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 361,
      "training_step_time": 9.89893627166748
    },
    {
      "epoch": 1.377105712890625e-06,
      "model_forward_time": 3.131728172302246,
      "step": 361
    },
    {
      "epoch": 1.377105712890625e-06,
      "step": 361,
      "training_step_time": 10.285094261169434
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 3.190519094467163,
      "step": 362
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 362,
      "training_step_time": 9.965148448944092
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 3.469566822052002,
      "step": 362
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 362,
      "training_step_time": 10.238369464874268
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 3.1359972953796387,
      "step": 362
    },
    {
      "dynamics_loss_avg": 0.08907883763313293,
      "epoch": 1.38092041015625e-06,
      "step": 362
    },
    {
      "action_loss_avg": 0.09553100429475307,
      "epoch": 1.38092041015625e-06,
      "step": 362
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 362,
      "training_step_time": 9.912286281585693
    },
    {
      "epoch": 1.38092041015625e-06,
      "model_forward_time": 3.1358139514923096,
      "step": 362
    },
    {
      "epoch": 1.38092041015625e-06,
      "step": 362,
      "training_step_time": 10.284487247467041
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 3.196787118911743,
      "step": 363
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 363,
      "training_step_time": 9.971550226211548
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 3.1400339603424072,
      "step": 363
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 363,
      "training_step_time": 9.917332172393799
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 3.43180251121521,
      "step": 363
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 363,
      "training_step_time": 10.201974868774414
    },
    {
      "epoch": 1.384735107421875e-06,
      "model_forward_time": 3.140831708908081,
      "step": 363
    },
    {
      "epoch": 1.384735107421875e-06,
      "step": 363,
      "training_step_time": 10.32369327545166
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 3.1945481300354004,
      "step": 364
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 364,
      "training_step_time": 9.982937335968018
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 3.1448330879211426,
      "step": 364
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 364,
      "training_step_time": 9.92694616317749
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 3.1398680210113525,
      "step": 364
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 364,
      "training_step_time": 9.916034936904907
    },
    {
      "epoch": 1.3885498046875e-06,
      "model_forward_time": 3.447925329208374,
      "step": 364
    },
    {
      "epoch": 1.3885498046875e-06,
      "step": 364,
      "training_step_time": 10.321287631988525
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 3.193441867828369,
      "step": 365
    },
    {
      "dynamics_loss_avg": 0.08594290725886822,
      "epoch": 1.392364501953125e-06,
      "step": 365
    },
    {
      "action_loss_avg": 0.08755789026618004,
      "epoch": 1.392364501953125e-06,
      "step": 365
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 365,
      "training_step_time": 9.970026731491089
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 3.1351523399353027,
      "step": 365
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 365,
      "training_step_time": 9.920843839645386
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 3.138842821121216,
      "step": 365
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 365,
      "training_step_time": 9.903896808624268
    },
    {
      "epoch": 1.392364501953125e-06,
      "model_forward_time": 3.138695240020752,
      "step": 365
    },
    {
      "epoch": 1.392364501953125e-06,
      "step": 365,
      "training_step_time": 10.306828260421753
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 3.1952340602874756,
      "step": 366
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 366,
      "training_step_time": 10.286994934082031
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 3.1379952430725098,
      "step": 366
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 366,
      "training_step_time": 9.909785509109497
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 3.1412315368652344,
      "step": 366
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 366,
      "training_step_time": 9.907240867614746
    },
    {
      "epoch": 1.39617919921875e-06,
      "model_forward_time": 3.139035701751709,
      "step": 366
    },
    {
      "epoch": 1.39617919921875e-06,
      "step": 366,
      "training_step_time": 10.315572023391724
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 3.1819653511047363,
      "step": 367
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 367,
      "training_step_time": 9.969969511032104
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 3.44897198677063,
      "step": 367
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 367,
      "training_step_time": 10.232043743133545
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 3.143528699874878,
      "step": 367
    },
    {
      "dynamics_loss_avg": 0.08339159712195396,
      "epoch": 1.399993896484375e-06,
      "step": 367
    },
    {
      "action_loss_avg": 0.07858921252191067,
      "epoch": 1.399993896484375e-06,
      "step": 367
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 367,
      "training_step_time": 9.925541877746582
    },
    {
      "epoch": 1.399993896484375e-06,
      "model_forward_time": 3.1359097957611084,
      "step": 367
    },
    {
      "epoch": 1.399993896484375e-06,
      "step": 367,
      "training_step_time": 10.267119884490967
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 3.1884803771972656,
      "step": 368
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 368,
      "training_step_time": 9.970668077468872
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 3.1373136043548584,
      "step": 368
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 368,
      "training_step_time": 9.91145634651184
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 3.13765811920166,
      "step": 368
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 368,
      "training_step_time": 10.224867820739746
    },
    {
      "epoch": 1.40380859375e-06,
      "model_forward_time": 3.1340386867523193,
      "step": 368
    },
    {
      "epoch": 1.40380859375e-06,
      "step": 368,
      "training_step_time": 10.314635276794434
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 3.1937496662139893,
      "step": 369
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 369,
      "training_step_time": 9.976104021072388
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 3.1405258178710938,
      "step": 369
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 369,
      "training_step_time": 9.912421703338623
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 3.132385015487671,
      "step": 369
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 369,
      "training_step_time": 9.901744842529297
    },
    {
      "epoch": 1.407623291015625e-06,
      "model_forward_time": 3.4249796867370605,
      "step": 369
    },
    {
      "epoch": 1.407623291015625e-06,
      "step": 369,
      "training_step_time": 10.289698839187622
    },
    {
      "epoch": 1.41143798828125e-06,
      "grad_norm": 1.1813008785247803,
      "learning_rate": 1e-05,
      "loss": 0.1853,
      "step": 370
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 3.176854372024536,
      "step": 370
    },
    {
      "dynamics_loss_avg": 0.08122116066515446,
      "epoch": 1.41143798828125e-06,
      "step": 370
    },
    {
      "action_loss_avg": 0.07148684971034527,
      "epoch": 1.41143798828125e-06,
      "step": 370
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 370,
      "training_step_time": 9.955580472946167
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 3.1380834579467773,
      "step": 370
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 370,
      "training_step_time": 9.912274837493896
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 3.1290621757507324,
      "step": 370
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 370,
      "training_step_time": 9.902297973632812
    },
    {
      "epoch": 1.41143798828125e-06,
      "model_forward_time": 3.139899492263794,
      "step": 370
    },
    {
      "epoch": 1.41143798828125e-06,
      "step": 370,
      "training_step_time": 10.280441045761108
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 3.1722402572631836,
      "step": 371
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 371,
      "training_step_time": 10.247529983520508
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 3.134685516357422,
      "step": 371
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 371,
      "training_step_time": 9.90694260597229
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 3.138315439224243,
      "step": 371
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 371,
      "training_step_time": 9.917680740356445
    },
    {
      "epoch": 1.415252685546875e-06,
      "model_forward_time": 3.1390492916107178,
      "step": 371
    },
    {
      "epoch": 1.415252685546875e-06,
      "step": 371,
      "training_step_time": 10.041187286376953
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 3.1703996658325195,
      "step": 372
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 372,
      "training_step_time": 9.94942021369934
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 3.4178543090820312,
      "step": 372
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 372,
      "training_step_time": 10.192783832550049
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 3.1313047409057617,
      "step": 372
    },
    {
      "dynamics_loss_avg": 0.09683531485497951,
      "epoch": 1.4190673828125e-06,
      "step": 372
    },
    {
      "action_loss_avg": 0.07325037345290183,
      "epoch": 1.4190673828125e-06,
      "step": 372
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 372,
      "training_step_time": 9.906687259674072
    },
    {
      "epoch": 1.4190673828125e-06,
      "model_forward_time": 3.131849527359009,
      "step": 372
    },
    {
      "epoch": 1.4190673828125e-06,
      "step": 372,
      "training_step_time": 10.027378559112549
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 3.186962842941284,
      "step": 373
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 373,
      "training_step_time": 9.96768045425415
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 3.135406494140625,
      "step": 373
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 373,
      "training_step_time": 9.907601833343506
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 3.3862853050231934,
      "step": 373
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 373,
      "training_step_time": 10.161084175109863
    },
    {
      "epoch": 1.422882080078125e-06,
      "model_forward_time": 3.13606595993042,
      "step": 373
    },
    {
      "epoch": 1.422882080078125e-06,
      "step": 373,
      "training_step_time": 10.309510946273804
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 3.190387487411499,
      "step": 374
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 374,
      "training_step_time": 9.964476108551025
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 3.1331655979156494,
      "step": 374
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 374,
      "training_step_time": 9.91983675956726
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 3.1370153427124023,
      "step": 374
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 374,
      "training_step_time": 9.91213870048523
    },
    {
      "epoch": 1.42669677734375e-06,
      "model_forward_time": 3.449847936630249,
      "step": 374
    },
    {
      "epoch": 1.42669677734375e-06,
      "step": 374,
      "training_step_time": 10.321072816848755
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 3.1842706203460693,
      "step": 375
    },
    {
      "dynamics_loss_avg": 0.08753666244447231,
      "epoch": 1.430511474609375e-06,
      "step": 375
    },
    {
      "action_loss_avg": 0.08572570658288896,
      "epoch": 1.430511474609375e-06,
      "step": 375
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 375,
      "training_step_time": 9.964441776275635
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 3.1362273693084717,
      "step": 375
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 375,
      "training_step_time": 9.911587476730347
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 3.1382646560668945,
      "step": 375
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 375,
      "training_step_time": 9.913146495819092
    },
    {
      "epoch": 1.430511474609375e-06,
      "model_forward_time": 3.134373188018799,
      "step": 375
    },
    {
      "epoch": 1.430511474609375e-06,
      "step": 375,
      "training_step_time": 10.288225173950195
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 3.444563388824463,
      "step": 376
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 376,
      "training_step_time": 10.223348379135132
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 3.131805896759033,
      "step": 376
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 376,
      "training_step_time": 9.907405138015747
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 3.1330485343933105,
      "step": 376
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 376,
      "training_step_time": 9.903278827667236
    },
    {
      "epoch": 1.434326171875e-06,
      "model_forward_time": 3.132505416870117,
      "step": 376
    },
    {
      "epoch": 1.434326171875e-06,
      "step": 376,
      "training_step_time": 10.31021785736084
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 3.1944143772125244,
      "step": 377
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 377,
      "training_step_time": 9.970000982284546
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 3.4041929244995117,
      "step": 377
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 377,
      "training_step_time": 10.179404020309448
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 3.1355161666870117,
      "step": 377
    },
    {
      "dynamics_loss_avg": 0.0997240848839283,
      "epoch": 1.438140869140625e-06,
      "step": 377
    },
    {
      "action_loss_avg": 0.0950181506574154,
      "epoch": 1.438140869140625e-06,
      "step": 377
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 377,
      "training_step_time": 9.912656784057617
    },
    {
      "epoch": 1.438140869140625e-06,
      "model_forward_time": 3.135497808456421,
      "step": 377
    },
    {
      "epoch": 1.438140869140625e-06,
      "step": 377,
      "training_step_time": 10.281890630722046
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 3.174727201461792,
      "step": 378
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 378,
      "training_step_time": 9.953205823898315
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 2.4322283267974854,
      "step": 378
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 378,
      "training_step_time": 7.46657133102417
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 3.4313957691192627,
      "step": 378
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 378,
      "training_step_time": 10.200219869613647
    },
    {
      "epoch": 1.44195556640625e-06,
      "model_forward_time": 3.131103038787842,
      "step": 378
    },
    {
      "epoch": 1.44195556640625e-06,
      "step": 378,
      "training_step_time": 10.323762655258179
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 3.1761786937713623,
      "step": 379
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 379,
      "training_step_time": 9.943349123001099
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 3.1319100856781006,
      "step": 379
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 379,
      "training_step_time": 9.903321266174316
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 3.1326935291290283,
      "step": 379
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 379,
      "training_step_time": 9.901707649230957
    },
    {
      "epoch": 1.445770263671875e-06,
      "model_forward_time": 3.1313443183898926,
      "step": 379
    },
    {
      "epoch": 1.445770263671875e-06,
      "step": 379,
      "training_step_time": 10.294920206069946
    },
    {
      "epoch": 1.4495849609375e-06,
      "grad_norm": 0.8966787457466125,
      "learning_rate": 1e-05,
      "loss": 0.185,
      "step": 380
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 3.166541576385498,
      "step": 380
    },
    {
      "dynamics_loss_avg": 0.09255019128322602,
      "epoch": 1.4495849609375e-06,
      "step": 380
    },
    {
      "action_loss_avg": 0.09496249668300152,
      "epoch": 1.4495849609375e-06,
      "step": 380
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 380,
      "training_step_time": 9.94586706161499
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 3.138068199157715,
      "step": 380
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 380,
      "training_step_time": 9.906211853027344
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 3.138087034225464,
      "step": 380
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 380,
      "training_step_time": 9.910913705825806
    },
    {
      "epoch": 1.4495849609375e-06,
      "model_forward_time": 3.1377692222595215,
      "step": 380
    },
    {
      "epoch": 1.4495849609375e-06,
      "step": 380,
      "training_step_time": 10.275381565093994
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 3.4601519107818604,
      "step": 381
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 381,
      "training_step_time": 10.234025239944458
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 3.1406171321868896,
      "step": 381
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 381,
      "training_step_time": 9.912075519561768
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 3.144319534301758,
      "step": 381
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 381,
      "training_step_time": 9.92255187034607
    },
    {
      "epoch": 1.453399658203125e-06,
      "model_forward_time": 3.1442391872406006,
      "step": 381
    },
    {
      "epoch": 1.453399658203125e-06,
      "step": 381,
      "training_step_time": 10.035622596740723
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 3.1733274459838867,
      "step": 382
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 382,
      "training_step_time": 9.946354389190674
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 3.424269199371338,
      "step": 382
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 382,
      "training_step_time": 10.195044755935669
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 3.1405069828033447,
      "step": 382
    },
    {
      "dynamics_loss_avg": 0.08648646622896194,
      "epoch": 1.45721435546875e-06,
      "step": 382
    },
    {
      "action_loss_avg": 0.08654331676661968,
      "epoch": 1.45721435546875e-06,
      "step": 382
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 382,
      "training_step_time": 9.918123006820679
    },
    {
      "epoch": 1.45721435546875e-06,
      "model_forward_time": 3.1391637325286865,
      "step": 382
    },
    {
      "epoch": 1.45721435546875e-06,
      "step": 382,
      "training_step_time": 10.258471012115479
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 3.169506788253784,
      "step": 383
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 383,
      "training_step_time": 9.946091651916504
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 3.1416943073272705,
      "step": 383
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 383,
      "training_step_time": 9.919126987457275
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 3.3907079696655273,
      "step": 383
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 383,
      "training_step_time": 10.16335678100586
    },
    {
      "epoch": 1.461029052734375e-06,
      "model_forward_time": 3.142076253890991,
      "step": 383
    },
    {
      "epoch": 1.461029052734375e-06,
      "step": 383,
      "training_step_time": 10.279277086257935
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 3.1657216548919678,
      "step": 384
    },
    {
      "epoch": 1.46484375e-06,
      "step": 384,
      "training_step_time": 9.947017192840576
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 3.131277322769165,
      "step": 384
    },
    {
      "epoch": 1.46484375e-06,
      "step": 384,
      "training_step_time": 9.906667709350586
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 3.1385884284973145,
      "step": 384
    },
    {
      "epoch": 1.46484375e-06,
      "step": 384,
      "training_step_time": 9.91129207611084
    },
    {
      "epoch": 1.46484375e-06,
      "model_forward_time": 3.446739912033081,
      "step": 384
    },
    {
      "epoch": 1.46484375e-06,
      "step": 384,
      "training_step_time": 10.321932792663574
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 3.170020580291748,
      "step": 385
    },
    {
      "dynamics_loss_avg": 0.10144600607454776,
      "epoch": 1.468658447265625e-06,
      "step": 385
    },
    {
      "action_loss_avg": 0.09012399949133396,
      "epoch": 1.468658447265625e-06,
      "step": 385
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 385,
      "training_step_time": 9.948613166809082
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 3.1517550945281982,
      "step": 385
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 385,
      "training_step_time": 9.927356243133545
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 3.13637113571167,
      "step": 385
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 385,
      "training_step_time": 9.912728786468506
    },
    {
      "epoch": 1.468658447265625e-06,
      "model_forward_time": 3.135395050048828,
      "step": 385
    },
    {
      "epoch": 1.468658447265625e-06,
      "step": 385,
      "training_step_time": 10.266660451889038
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 3.1902427673339844,
      "step": 386
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 386,
      "training_step_time": 10.244672060012817
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 3.1350138187408447,
      "step": 386
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 386,
      "training_step_time": 9.902534484863281
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 3.1359355449676514,
      "step": 386
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 386,
      "training_step_time": 9.920269250869751
    },
    {
      "epoch": 1.47247314453125e-06,
      "model_forward_time": 3.1357362270355225,
      "step": 386
    },
    {
      "epoch": 1.47247314453125e-06,
      "step": 386,
      "training_step_time": 10.281795024871826
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 3.1646947860717773,
      "step": 387
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 387,
      "training_step_time": 9.941740036010742
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 3.438356399536133,
      "step": 387
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 387,
      "training_step_time": 10.206719160079956
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 3.1399967670440674,
      "step": 387
    },
    {
      "dynamics_loss_avg": 0.0931927066296339,
      "epoch": 1.476287841796875e-06,
      "step": 387
    },
    {
      "action_loss_avg": 0.08789346665143967,
      "epoch": 1.476287841796875e-06,
      "step": 387
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 387,
      "training_step_time": 9.915009021759033
    },
    {
      "epoch": 1.476287841796875e-06,
      "model_forward_time": 3.1292951107025146,
      "step": 387
    },
    {
      "epoch": 1.476287841796875e-06,
      "step": 387,
      "training_step_time": 10.024160623550415
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 3.1588900089263916,
      "step": 388
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 388,
      "training_step_time": 9.932032108306885
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 3.1344547271728516,
      "step": 388
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 388,
      "training_step_time": 9.902839183807373
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 3.1346585750579834,
      "step": 388
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 388,
      "training_step_time": 10.22181510925293
    },
    {
      "epoch": 1.4801025390625e-06,
      "model_forward_time": 2.433584213256836,
      "step": 388
    },
    {
      "epoch": 1.4801025390625e-06,
      "step": 388,
      "training_step_time": 10.324175834655762
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 3.168307065963745,
      "step": 389
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 389,
      "training_step_time": 9.941917896270752
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 3.14265513420105,
      "step": 389
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 389,
      "training_step_time": 9.91810941696167
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 3.1418509483337402,
      "step": 389
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 389,
      "training_step_time": 9.913362741470337
    },
    {
      "epoch": 1.483917236328125e-06,
      "model_forward_time": 3.4365105628967285,
      "step": 389
    },
    {
      "epoch": 1.483917236328125e-06,
      "step": 389,
      "training_step_time": 10.319205284118652
    },
    {
      "epoch": 1.48773193359375e-06,
      "grad_norm": 0.451398104429245,
      "learning_rate": 1e-05,
      "loss": 0.1801,
      "step": 390
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 3.1758406162261963,
      "step": 390
    },
    {
      "dynamics_loss_avg": 0.1094014160335064,
      "epoch": 1.48773193359375e-06,
      "step": 390
    },
    {
      "action_loss_avg": 0.11690505146980286,
      "epoch": 1.48773193359375e-06,
      "step": 390
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 390,
      "training_step_time": 9.952924489974976
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 3.138132095336914,
      "step": 390
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 390,
      "training_step_time": 9.9078209400177
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 3.135925769805908,
      "step": 390
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 390,
      "training_step_time": 9.912660837173462
    },
    {
      "epoch": 1.48773193359375e-06,
      "model_forward_time": 3.141364812850952,
      "step": 390
    },
    {
      "epoch": 1.48773193359375e-06,
      "step": 390,
      "training_step_time": 10.257987976074219
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 3.189448356628418,
      "step": 391
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 391,
      "training_step_time": 10.246122121810913
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 3.152461290359497,
      "step": 391
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 391,
      "training_step_time": 9.926835060119629
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 3.146758556365967,
      "step": 391
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 391,
      "training_step_time": 9.918813228607178
    },
    {
      "epoch": 1.491546630859375e-06,
      "model_forward_time": 3.139832019805908,
      "step": 391
    },
    {
      "epoch": 1.491546630859375e-06,
      "step": 391,
      "training_step_time": 10.031198024749756
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 3.1523921489715576,
      "step": 392
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 392,
      "training_step_time": 9.927356004714966
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 3.448366165161133,
      "step": 392
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 392,
      "training_step_time": 10.213624238967896
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 2.4360218048095703,
      "step": 392
    },
    {
      "dynamics_loss_avg": 0.0967685017734766,
      "epoch": 1.495361328125e-06,
      "step": 392
    },
    {
      "action_loss_avg": 0.09162798039615154,
      "epoch": 1.495361328125e-06,
      "step": 392
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 392,
      "training_step_time": 7.46662974357605
    },
    {
      "epoch": 1.495361328125e-06,
      "model_forward_time": 3.130669593811035,
      "step": 392
    },
    {
      "epoch": 1.495361328125e-06,
      "step": 392,
      "training_step_time": 10.027462482452393
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 3.1847987174987793,
      "step": 393
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 393,
      "training_step_time": 9.96116304397583
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 3.1331892013549805,
      "step": 393
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 393,
      "training_step_time": 9.905236959457397
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 3.134326696395874,
      "step": 393
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 393,
      "training_step_time": 10.21262502670288
    },
    {
      "epoch": 1.499176025390625e-06,
      "model_forward_time": 3.13264799118042,
      "step": 393
    },
    {
      "epoch": 1.499176025390625e-06,
      "step": 393,
      "training_step_time": 10.294575452804565
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 3.186012029647827,
      "step": 394
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 394,
      "training_step_time": 9.972436428070068
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 3.143810749053955,
      "step": 394
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 394,
      "training_step_time": 9.919506311416626
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 3.141099452972412,
      "step": 394
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 394,
      "training_step_time": 9.910013437271118
    },
    {
      "epoch": 1.50299072265625e-06,
      "model_forward_time": 3.466343402862549,
      "step": 394
    },
    {
      "epoch": 1.50299072265625e-06,
      "step": 394,
      "training_step_time": 10.335144519805908
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 3.188302516937256,
      "step": 395
    },
    {
      "dynamics_loss_avg": 0.09353254176676273,
      "epoch": 1.506805419921875e-06,
      "step": 395
    },
    {
      "action_loss_avg": 0.07986174784600734,
      "epoch": 1.506805419921875e-06,
      "step": 395
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 395,
      "training_step_time": 9.963236331939697
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 3.1347718238830566,
      "step": 395
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 395,
      "training_step_time": 9.909444808959961
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 3.150359869003296,
      "step": 395
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 395,
      "training_step_time": 9.927441120147705
    },
    {
      "epoch": 1.506805419921875e-06,
      "model_forward_time": 3.1420936584472656,
      "step": 395
    },
    {
      "epoch": 1.506805419921875e-06,
      "step": 395,
      "training_step_time": 10.312843084335327
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 3.50447416305542,
      "step": 396
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 396,
      "training_step_time": 10.280310869216919
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 3.1428792476654053,
      "step": 396
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 396,
      "training_step_time": 9.923240661621094
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 3.1448283195495605,
      "step": 396
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 396,
      "training_step_time": 9.919876337051392
    },
    {
      "epoch": 1.5106201171875e-06,
      "model_forward_time": 3.139274835586548,
      "step": 396
    },
    {
      "epoch": 1.5106201171875e-06,
      "step": 396,
      "training_step_time": 10.032696008682251
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 3.20135498046875,
      "step": 397
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 397,
      "training_step_time": 9.99932336807251
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 3.4107019901275635,
      "step": 397
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 397,
      "training_step_time": 10.184754610061646
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 3.1409659385681152,
      "step": 397
    },
    {
      "dynamics_loss_avg": 0.08672653287649154,
      "epoch": 1.514434814453125e-06,
      "step": 397
    },
    {
      "action_loss_avg": 0.07725149802863598,
      "epoch": 1.514434814453125e-06,
      "step": 397
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 397,
      "training_step_time": 9.914618015289307
    },
    {
      "epoch": 1.514434814453125e-06,
      "model_forward_time": 3.141972064971924,
      "step": 397
    },
    {
      "epoch": 1.514434814453125e-06,
      "step": 397,
      "training_step_time": 10.035907745361328
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 3.1726787090301514,
      "step": 398
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 398,
      "training_step_time": 9.94556212425232
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 3.1356639862060547,
      "step": 398
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 398,
      "training_step_time": 9.909127235412598
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 3.4562735557556152,
      "step": 398
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 398,
      "training_step_time": 10.222673416137695
    },
    {
      "epoch": 1.51824951171875e-06,
      "model_forward_time": 3.126056432723999,
      "step": 398
    },
    {
      "epoch": 1.51824951171875e-06,
      "step": 398,
      "training_step_time": 10.232884645462036
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 3.166848659515381,
      "step": 399
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 399,
      "training_step_time": 9.937238216400146
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 3.133406639099121,
      "step": 399
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 399,
      "training_step_time": 9.904398918151855
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 3.130687952041626,
      "step": 399
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 399,
      "training_step_time": 9.902443647384644
    },
    {
      "epoch": 1.522064208984375e-06,
      "model_forward_time": 3.4377264976501465,
      "step": 399
    },
    {
      "epoch": 1.522064208984375e-06,
      "step": 399,
      "training_step_time": 10.310415983200073
    },
    {
      "epoch": 1.52587890625e-06,
      "grad_norm": 0.775789737701416,
      "learning_rate": 1e-05,
      "loss": 0.1802,
      "step": 400
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 3.169477701187134,
      "step": 400
    },
    {
      "dynamics_loss_avg": 0.07631831876933574,
      "epoch": 1.52587890625e-06,
      "step": 400
    },
    {
      "action_loss_avg": 0.07445034105330706,
      "epoch": 1.52587890625e-06,
      "step": 400
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 400,
      "training_step_time": 9.947817325592041
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 3.13394832611084,
      "step": 400
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 400,
      "training_step_time": 9.909244537353516
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 3.12896728515625,
      "step": 400
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 400,
      "training_step_time": 9.903504371643066
    },
    {
      "epoch": 1.52587890625e-06,
      "model_forward_time": 3.1314337253570557,
      "step": 400
    },
    {
      "epoch": 1.52587890625e-06,
      "step": 400,
      "training_step_time": 10.330991983413696
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 3.4531350135803223,
      "step": 401
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 401,
      "training_step_time": 10.230296611785889
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 3.1411867141723633,
      "step": 401
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 401,
      "training_step_time": 9.910067796707153
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 3.1471645832061768,
      "step": 401
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 401,
      "training_step_time": 9.917967557907104
    },
    {
      "epoch": 1.529693603515625e-06,
      "model_forward_time": 3.1387686729431152,
      "step": 401
    },
    {
      "epoch": 1.529693603515625e-06,
      "step": 401,
      "training_step_time": 10.01261043548584
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 3.1714043617248535,
      "step": 402
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 402,
      "training_step_time": 9.945220470428467
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 3.4946675300598145,
      "step": 402
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 402,
      "training_step_time": 10.265438795089722
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 3.1437277793884277,
      "step": 402
    },
    {
      "dynamics_loss_avg": 0.09601548165082932,
      "epoch": 1.53350830078125e-06,
      "step": 402
    },
    {
      "action_loss_avg": 0.08699168786406516,
      "epoch": 1.53350830078125e-06,
      "step": 402
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 402,
      "training_step_time": 9.926204204559326
    },
    {
      "epoch": 1.53350830078125e-06,
      "model_forward_time": 3.1454267501831055,
      "step": 402
    },
    {
      "epoch": 1.53350830078125e-06,
      "step": 402,
      "training_step_time": 10.036166429519653
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 2.465634346008301,
      "step": 403
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 403,
      "training_step_time": 7.50907826423645
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 3.1361536979675293,
      "step": 403
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 403,
      "training_step_time": 9.911470413208008
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 3.398890495300293,
      "step": 403
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 403,
      "training_step_time": 10.172413110733032
    },
    {
      "epoch": 1.537322998046875e-06,
      "model_forward_time": 3.132373094558716,
      "step": 403
    },
    {
      "epoch": 1.537322998046875e-06,
      "step": 403,
      "training_step_time": 10.295223474502563
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 3.165470600128174,
      "step": 404
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 404,
      "training_step_time": 9.933943271636963
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 3.146090030670166,
      "step": 404
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 404,
      "training_step_time": 9.921866178512573
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 3.146726131439209,
      "step": 404
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 404,
      "training_step_time": 9.921951055526733
    },
    {
      "epoch": 1.5411376953125e-06,
      "model_forward_time": 3.1467297077178955,
      "step": 404
    },
    {
      "epoch": 1.5411376953125e-06,
      "step": 404,
      "training_step_time": 10.320481300354004
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 3.18505859375,
      "step": 405
    },
    {
      "dynamics_loss_avg": 0.08334063803777099,
      "epoch": 1.544952392578125e-06,
      "step": 405
    },
    {
      "action_loss_avg": 0.06956121679395437,
      "epoch": 1.544952392578125e-06,
      "step": 405
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 405,
      "training_step_time": 9.966170310974121
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 3.1450140476226807,
      "step": 405
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 405,
      "training_step_time": 9.92367959022522
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 3.1432812213897705,
      "step": 405
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 405,
      "training_step_time": 9.925704717636108
    },
    {
      "epoch": 1.544952392578125e-06,
      "model_forward_time": 3.143481731414795,
      "step": 405
    },
    {
      "epoch": 1.544952392578125e-06,
      "step": 405,
      "training_step_time": 10.287273406982422
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 3.485438585281372,
      "step": 406
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 406,
      "training_step_time": 10.261895656585693
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 3.1409013271331787,
      "step": 406
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 406,
      "training_step_time": 9.910784721374512
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 3.1395580768585205,
      "step": 406
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 406,
      "training_step_time": 9.912934303283691
    },
    {
      "epoch": 1.54876708984375e-06,
      "model_forward_time": 3.136042594909668,
      "step": 406
    },
    {
      "epoch": 1.54876708984375e-06,
      "step": 406,
      "training_step_time": 10.0238196849823
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 3.1594626903533936,
      "step": 407
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 407,
      "training_step_time": 9.935813665390015
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 3.395939588546753,
      "step": 407
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 407,
      "training_step_time": 10.166738748550415
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 3.1373989582061768,
      "step": 407
    },
    {
      "dynamics_loss_avg": 0.07750701643526554,
      "epoch": 1.552581787109375e-06,
      "step": 407
    },
    {
      "action_loss_avg": 0.07351220678538084,
      "epoch": 1.552581787109375e-06,
      "step": 407
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 407,
      "training_step_time": 9.906932830810547
    },
    {
      "epoch": 1.552581787109375e-06,
      "model_forward_time": 3.1381025314331055,
      "step": 407
    },
    {
      "epoch": 1.552581787109375e-06,
      "step": 407,
      "training_step_time": 10.31329607963562
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 3.1905837059020996,
      "step": 408
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 408,
      "training_step_time": 9.98027491569519
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 3.1381969451904297,
      "step": 408
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 408,
      "training_step_time": 9.908126831054688
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 3.4165241718292236,
      "step": 408
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 408,
      "training_step_time": 10.190688371658325
    },
    {
      "epoch": 1.556396484375e-06,
      "model_forward_time": 3.1355934143066406,
      "step": 408
    },
    {
      "epoch": 1.556396484375e-06,
      "step": 408,
      "training_step_time": 10.275071859359741
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 3.1631245613098145,
      "step": 409
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 409,
      "training_step_time": 9.937492370605469
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 3.1319053173065186,
      "step": 409
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 409,
      "training_step_time": 9.907299280166626
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 3.136599540710449,
      "step": 409
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 409,
      "training_step_time": 9.916951179504395
    },
    {
      "epoch": 1.560211181640625e-06,
      "model_forward_time": 3.4272615909576416,
      "step": 409
    },
    {
      "epoch": 1.560211181640625e-06,
      "step": 409,
      "training_step_time": 10.302700757980347
    },
    {
      "epoch": 1.56402587890625e-06,
      "grad_norm": 0.7363119721412659,
      "learning_rate": 1e-05,
      "loss": 0.1724,
      "step": 410
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 3.242698907852173,
      "step": 410
    },
    {
      "dynamics_loss_avg": 0.1092264175415039,
      "epoch": 1.56402587890625e-06,
      "step": 410
    },
    {
      "action_loss_avg": 0.09782932326197624,
      "epoch": 1.56402587890625e-06,
      "step": 410
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 410,
      "training_step_time": 10.026732206344604
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 3.140449047088623,
      "step": 410
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 410,
      "training_step_time": 9.914668560028076
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 3.139561653137207,
      "step": 410
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 410,
      "training_step_time": 9.916531562805176
    },
    {
      "epoch": 1.56402587890625e-06,
      "model_forward_time": 3.1424431800842285,
      "step": 410
    },
    {
      "epoch": 1.56402587890625e-06,
      "step": 410,
      "training_step_time": 10.032357215881348
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 3.427194595336914,
      "step": 411
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 411,
      "training_step_time": 10.20259404182434
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 3.1404576301574707,
      "step": 411
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 411,
      "training_step_time": 9.917780876159668
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 3.1391429901123047,
      "step": 411
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 411,
      "training_step_time": 9.918428897857666
    },
    {
      "epoch": 1.567840576171875e-06,
      "model_forward_time": 3.1470794677734375,
      "step": 411
    },
    {
      "epoch": 1.567840576171875e-06,
      "step": 411,
      "training_step_time": 10.304716110229492
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 3.1849327087402344,
      "step": 412
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 412,
      "training_step_time": 9.960309982299805
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 3.4272220134735107,
      "step": 412
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 412,
      "training_step_time": 10.20063328742981
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 3.131077289581299,
      "step": 412
    },
    {
      "dynamics_loss_avg": 0.11569703519344329,
      "epoch": 1.5716552734375e-06,
      "step": 412
    },
    {
      "action_loss_avg": 0.11344182044267655,
      "epoch": 1.5716552734375e-06,
      "step": 412
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 412,
      "training_step_time": 9.90418529510498
    },
    {
      "epoch": 1.5716552734375e-06,
      "model_forward_time": 3.131359577178955,
      "step": 412
    },
    {
      "epoch": 1.5716552734375e-06,
      "step": 412,
      "training_step_time": 10.027614116668701
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 3.1807878017425537,
      "step": 413
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 413,
      "training_step_time": 9.95417308807373
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 3.1307990550994873,
      "step": 413
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 413,
      "training_step_time": 9.903914213180542
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 2.433757781982422,
      "step": 413
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 413,
      "training_step_time": 7.778043270111084
    },
    {
      "epoch": 1.575469970703125e-06,
      "model_forward_time": 3.1333446502685547,
      "step": 413
    },
    {
      "epoch": 1.575469970703125e-06,
      "step": 413,
      "training_step_time": 10.277987957000732
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 3.192080020904541,
      "step": 414
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 414,
      "training_step_time": 9.981930494308472
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 3.1454880237579346,
      "step": 414
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 414,
      "training_step_time": 9.927753448486328
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 3.155564069747925,
      "step": 414
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 414,
      "training_step_time": 9.942435503005981
    },
    {
      "epoch": 1.57928466796875e-06,
      "model_forward_time": 3.4699718952178955,
      "step": 414
    },
    {
      "epoch": 1.57928466796875e-06,
      "step": 414,
      "training_step_time": 10.337310552597046
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 3.1764297485351562,
      "step": 415
    },
    {
      "dynamics_loss_avg": 0.09551965594291686,
      "epoch": 1.583099365234375e-06,
      "step": 415
    },
    {
      "action_loss_avg": 0.09619720187038183,
      "epoch": 1.583099365234375e-06,
      "step": 415
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 415,
      "training_step_time": 9.957385778427124
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 3.1359589099884033,
      "step": 415
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 415,
      "training_step_time": 9.911869525909424
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 3.138523578643799,
      "step": 415
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 415,
      "training_step_time": 9.921040534973145
    },
    {
      "epoch": 1.583099365234375e-06,
      "model_forward_time": 3.1344475746154785,
      "step": 415
    },
    {
      "epoch": 1.583099365234375e-06,
      "step": 415,
      "training_step_time": 10.322560548782349
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 3.4796977043151855,
      "step": 416
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 416,
      "training_step_time": 10.259742498397827
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 3.1361255645751953,
      "step": 416
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 416,
      "training_step_time": 9.91284704208374
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 3.1329636573791504,
      "step": 416
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 416,
      "training_step_time": 9.904472827911377
    },
    {
      "epoch": 1.5869140625e-06,
      "model_forward_time": 3.136939764022827,
      "step": 416
    },
    {
      "epoch": 1.5869140625e-06,
      "step": 416,
      "training_step_time": 10.302736282348633
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 3.17855167388916,
      "step": 417
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 417,
      "training_step_time": 9.946019172668457
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 3.393305540084839,
      "step": 417
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 417,
      "training_step_time": 10.158016204833984
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 3.131166934967041,
      "step": 417
    },
    {
      "dynamics_loss_avg": 0.10619053468108178,
      "epoch": 1.590728759765625e-06,
      "step": 417
    },
    {
      "action_loss_avg": 0.10804674588143826,
      "epoch": 1.590728759765625e-06,
      "step": 417
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 417,
      "training_step_time": 9.895248651504517
    },
    {
      "epoch": 1.590728759765625e-06,
      "model_forward_time": 3.1312201023101807,
      "step": 417
    },
    {
      "epoch": 1.590728759765625e-06,
      "step": 417,
      "training_step_time": 10.019580125808716
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 3.178664445877075,
      "step": 418
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 418,
      "training_step_time": 9.959018468856812
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 3.1329259872436523,
      "step": 418
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 418,
      "training_step_time": 9.905738115310669
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 3.451256275177002,
      "step": 418
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 418,
      "training_step_time": 10.21976351737976
    },
    {
      "epoch": 1.59454345703125e-06,
      "model_forward_time": 3.135603666305542,
      "step": 418
    },
    {
      "epoch": 1.59454345703125e-06,
      "step": 418,
      "training_step_time": 10.277761459350586
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 3.167678117752075,
      "step": 419
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 419,
      "training_step_time": 9.941102981567383
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 3.140871524810791,
      "step": 419
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 419,
      "training_step_time": 9.921558618545532
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 3.146085500717163,
      "step": 419
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 419,
      "training_step_time": 9.929481506347656
    },
    {
      "epoch": 1.598358154296875e-06,
      "model_forward_time": 3.4305641651153564,
      "step": 419
    },
    {
      "epoch": 1.598358154296875e-06,
      "step": 419,
      "training_step_time": 10.301901817321777
    },
    {
      "epoch": 1.6021728515625e-06,
      "grad_norm": 0.5560957789421082,
      "learning_rate": 1e-05,
      "loss": 0.178,
      "step": 420
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 3.1904067993164062,
      "step": 420
    },
    {
      "dynamics_loss_avg": 0.0985412359237671,
      "epoch": 1.6021728515625e-06,
      "step": 420
    },
    {
      "action_loss_avg": 0.08637117762118578,
      "epoch": 1.6021728515625e-06,
      "step": 420
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 420,
      "training_step_time": 9.97290587425232
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 3.1504523754119873,
      "step": 420
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 420,
      "training_step_time": 9.935417890548706
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 3.1480531692504883,
      "step": 420
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 420,
      "training_step_time": 9.93221640586853
    },
    {
      "epoch": 1.6021728515625e-06,
      "model_forward_time": 3.150923013687134,
      "step": 420
    },
    {
      "epoch": 1.6021728515625e-06,
      "step": 420,
      "training_step_time": 10.306760311126709
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 3.4487507343292236,
      "step": 421
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 421,
      "training_step_time": 10.23099422454834
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 3.1474788188934326,
      "step": 421
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 421,
      "training_step_time": 9.924842357635498
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 3.1468989849090576,
      "step": 421
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 421,
      "training_step_time": 9.926577806472778
    },
    {
      "epoch": 1.605987548828125e-06,
      "model_forward_time": 3.1475026607513428,
      "step": 421
    },
    {
      "epoch": 1.605987548828125e-06,
      "step": 421,
      "training_step_time": 10.339139461517334
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 3.1765024662017822,
      "step": 422
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 422,
      "training_step_time": 9.959266185760498
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 3.4562599658966064,
      "step": 422
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 422,
      "training_step_time": 10.23307204246521
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 3.1382181644439697,
      "step": 422
    },
    {
      "dynamics_loss_avg": 0.08545293658971786,
      "epoch": 1.60980224609375e-06,
      "step": 422
    },
    {
      "action_loss_avg": 0.07245517298579215,
      "epoch": 1.60980224609375e-06,
      "step": 422
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 422,
      "training_step_time": 9.915616750717163
    },
    {
      "epoch": 1.60980224609375e-06,
      "model_forward_time": 3.141197681427002,
      "step": 422
    },
    {
      "epoch": 1.60980224609375e-06,
      "step": 422,
      "training_step_time": 10.32129955291748
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 3.187018632888794,
      "step": 423
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 423,
      "training_step_time": 9.962896823883057
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 3.141183376312256,
      "step": 423
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 423,
      "training_step_time": 9.9193274974823
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 3.4243905544281006,
      "step": 423
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 423,
      "training_step_time": 10.198972702026367
    },
    {
      "epoch": 1.613616943359375e-06,
      "model_forward_time": 3.143819570541382,
      "step": 423
    },
    {
      "epoch": 1.613616943359375e-06,
      "step": 423,
      "training_step_time": 10.280654907226562
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 3.1794581413269043,
      "step": 424
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 424,
      "training_step_time": 9.973004579544067
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 3.1440157890319824,
      "step": 424
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 424,
      "training_step_time": 9.922024965286255
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 3.144899368286133,
      "step": 424
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 424,
      "training_step_time": 9.918460607528687
    },
    {
      "epoch": 1.617431640625e-06,
      "model_forward_time": 3.4779059886932373,
      "step": 424
    },
    {
      "epoch": 1.617431640625e-06,
      "step": 424,
      "training_step_time": 10.357651233673096
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 3.1726531982421875,
      "step": 425
    },
    {
      "dynamics_loss_avg": 0.08835904635488986,
      "epoch": 1.621246337890625e-06,
      "step": 425
    },
    {
      "action_loss_avg": 0.08742318898439408,
      "epoch": 1.621246337890625e-06,
      "step": 425
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 425,
      "training_step_time": 9.957703828811646
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 3.1466758251190186,
      "step": 425
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 425,
      "training_step_time": 9.933140754699707
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 3.150456666946411,
      "step": 425
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 425,
      "training_step_time": 9.935452938079834
    },
    {
      "epoch": 1.621246337890625e-06,
      "model_forward_time": 3.1430251598358154,
      "step": 425
    },
    {
      "epoch": 1.621246337890625e-06,
      "step": 425,
      "training_step_time": 10.291537761688232
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 3.1770832538604736,
      "step": 426
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 426,
      "training_step_time": 10.277375936508179
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 3.1426234245300293,
      "step": 426
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 426,
      "training_step_time": 9.913639783859253
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 3.1397008895874023,
      "step": 426
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 426,
      "training_step_time": 9.908933877944946
    },
    {
      "epoch": 1.62506103515625e-06,
      "model_forward_time": 3.136639356613159,
      "step": 426
    },
    {
      "epoch": 1.62506103515625e-06,
      "step": 426,
      "training_step_time": 10.305692911148071
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 3.1942811012268066,
      "step": 427
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 427,
      "training_step_time": 9.971988677978516
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 3.3972907066345215,
      "step": 427
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 427,
      "training_step_time": 10.16647458076477
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 3.140321731567383,
      "step": 427
    },
    {
      "dynamics_loss_avg": 0.08848658427596093,
      "epoch": 1.628875732421875e-06,
      "step": 427
    },
    {
      "action_loss_avg": 0.09069486297667026,
      "epoch": 1.628875732421875e-06,
      "step": 427
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 427,
      "training_step_time": 9.915173292160034
    },
    {
      "epoch": 1.628875732421875e-06,
      "model_forward_time": 3.1517367362976074,
      "step": 427
    },
    {
      "epoch": 1.628875732421875e-06,
      "step": 427,
      "training_step_time": 10.282512903213501
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 3.1834304332733154,
      "step": 428
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 428,
      "training_step_time": 9.956765174865723
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 3.1424827575683594,
      "step": 428
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 428,
      "training_step_time": 9.912782669067383
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 3.1393747329711914,
      "step": 428
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 428,
      "training_step_time": 10.219228982925415
    },
    {
      "epoch": 1.6326904296875e-06,
      "model_forward_time": 3.139996290206909,
      "step": 428
    },
    {
      "epoch": 1.6326904296875e-06,
      "step": 428,
      "training_step_time": 10.297133684158325
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 3.1911115646362305,
      "step": 429
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 429,
      "training_step_time": 9.969034433364868
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 3.1443426609039307,
      "step": 429
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 429,
      "training_step_time": 9.917514324188232
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 3.1388893127441406,
      "step": 429
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 429,
      "training_step_time": 9.909923076629639
    },
    {
      "epoch": 1.636505126953125e-06,
      "model_forward_time": 3.426018714904785,
      "step": 429
    },
    {
      "epoch": 1.636505126953125e-06,
      "step": 429,
      "training_step_time": 10.295125246047974
    },
    {
      "epoch": 1.64031982421875e-06,
      "grad_norm": 0.8516829609870911,
      "learning_rate": 1e-05,
      "loss": 0.1671,
      "step": 430
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 3.173002004623413,
      "step": 430
    },
    {
      "dynamics_loss_avg": 0.0960109356790781,
      "epoch": 1.64031982421875e-06,
      "step": 430
    },
    {
      "action_loss_avg": 0.08486681319773197,
      "epoch": 1.64031982421875e-06,
      "step": 430
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 430,
      "training_step_time": 9.950170516967773
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 3.1432833671569824,
      "step": 430
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 430,
      "training_step_time": 9.916329145431519
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 3.1425328254699707,
      "step": 430
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 430,
      "training_step_time": 9.915019512176514
    },
    {
      "epoch": 1.64031982421875e-06,
      "model_forward_time": 3.139437198638916,
      "step": 430
    },
    {
      "epoch": 1.64031982421875e-06,
      "step": 430,
      "training_step_time": 10.282411098480225
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 3.1792967319488525,
      "step": 431
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 431,
      "training_step_time": 10.235576391220093
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 3.1380441188812256,
      "step": 431
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 431,
      "training_step_time": 9.912812948226929
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 3.1347713470458984,
      "step": 431
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 431,
      "training_step_time": 9.909290313720703
    },
    {
      "epoch": 1.644134521484375e-06,
      "model_forward_time": 3.1388072967529297,
      "step": 431
    },
    {
      "epoch": 1.644134521484375e-06,
      "step": 431,
      "training_step_time": 10.329917907714844
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 3.1874582767486572,
      "step": 432
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 432,
      "training_step_time": 9.963046789169312
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 3.4392058849334717,
      "step": 432
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 432,
      "training_step_time": 10.209052801132202
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 3.135150671005249,
      "step": 432
    },
    {
      "dynamics_loss_avg": 0.08550763688981533,
      "epoch": 1.64794921875e-06,
      "step": 432
    },
    {
      "action_loss_avg": 0.06995585784316063,
      "epoch": 1.64794921875e-06,
      "step": 432
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 432,
      "training_step_time": 9.911011934280396
    },
    {
      "epoch": 1.64794921875e-06,
      "model_forward_time": 3.1353354454040527,
      "step": 432
    },
    {
      "epoch": 1.64794921875e-06,
      "step": 432,
      "training_step_time": 10.307377815246582
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 3.1809911727905273,
      "step": 433
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 433,
      "training_step_time": 9.956456899642944
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 3.1478142738342285,
      "step": 433
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 433,
      "training_step_time": 9.921554565429688
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 3.418147087097168,
      "step": 433
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 433,
      "training_step_time": 10.196849822998047
    },
    {
      "epoch": 1.651763916015625e-06,
      "model_forward_time": 3.1410205364227295,
      "step": 433
    },
    {
      "epoch": 1.651763916015625e-06,
      "step": 433,
      "training_step_time": 10.282282829284668
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 3.1733922958374023,
      "step": 434
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 434,
      "training_step_time": 9.950921535491943
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 3.137768268585205,
      "step": 434
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 434,
      "training_step_time": 9.910199165344238
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 3.142270088195801,
      "step": 434
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 434,
      "training_step_time": 9.929759502410889
    },
    {
      "epoch": 1.65557861328125e-06,
      "model_forward_time": 3.462475538253784,
      "step": 434
    },
    {
      "epoch": 1.65557861328125e-06,
      "step": 434,
      "training_step_time": 10.335748434066772
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 3.1930720806121826,
      "step": 435
    },
    {
      "dynamics_loss_avg": 0.06568298302590847,
      "epoch": 1.659393310546875e-06,
      "step": 435
    },
    {
      "action_loss_avg": 0.05936561934649944,
      "epoch": 1.659393310546875e-06,
      "step": 435
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 435,
      "training_step_time": 9.9905424118042
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 3.1443471908569336,
      "step": 435
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 435,
      "training_step_time": 9.921216249465942
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 3.1405529975891113,
      "step": 435
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 435,
      "training_step_time": 9.918421983718872
    },
    {
      "epoch": 1.659393310546875e-06,
      "model_forward_time": 3.138389825820923,
      "step": 435
    },
    {
      "epoch": 1.659393310546875e-06,
      "step": 435,
      "training_step_time": 10.016916036605835
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 3.4961166381835938,
      "step": 436
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 436,
      "training_step_time": 10.275075674057007
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 3.1408638954162598,
      "step": 436
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 436,
      "training_step_time": 9.919195175170898
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 3.142177104949951,
      "step": 436
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 436,
      "training_step_time": 9.921608448028564
    },
    {
      "epoch": 1.6632080078125e-06,
      "model_forward_time": 3.1403069496154785,
      "step": 436
    },
    {
      "epoch": 1.6632080078125e-06,
      "step": 436,
      "training_step_time": 10.29414963722229
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 3.1965725421905518,
      "step": 437
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 437,
      "training_step_time": 9.97416639328003
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 3.455274820327759,
      "step": 437
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 437,
      "training_step_time": 10.237180709838867
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 3.1387593746185303,
      "step": 437
    },
    {
      "dynamics_loss_avg": 0.09389382861554622,
      "epoch": 1.667022705078125e-06,
      "step": 437
    },
    {
      "action_loss_avg": 0.07064746208488941,
      "epoch": 1.667022705078125e-06,
      "step": 437
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 437,
      "training_step_time": 9.934721231460571
    },
    {
      "epoch": 1.667022705078125e-06,
      "model_forward_time": 3.134434700012207,
      "step": 437
    },
    {
      "epoch": 1.667022705078125e-06,
      "step": 437,
      "training_step_time": 10.325871229171753
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 3.165205717086792,
      "step": 438
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 438,
      "training_step_time": 9.956816673278809
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 3.139807939529419,
      "step": 438
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 438,
      "training_step_time": 9.921013593673706
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 3.4467954635620117,
      "step": 438
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 438,
      "training_step_time": 10.243853569030762
    },
    {
      "epoch": 1.67083740234375e-06,
      "model_forward_time": 3.13962459564209,
      "step": 438
    },
    {
      "epoch": 1.67083740234375e-06,
      "step": 438,
      "training_step_time": 10.33546781539917
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 3.167463779449463,
      "step": 439
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 439,
      "training_step_time": 9.951562881469727
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 3.1344566345214844,
      "step": 439
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 439,
      "training_step_time": 9.914889574050903
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 3.132913827896118,
      "step": 439
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 439,
      "training_step_time": 9.9160475730896
    },
    {
      "epoch": 1.674652099609375e-06,
      "model_forward_time": 3.429154396057129,
      "step": 439
    },
    {
      "epoch": 1.674652099609375e-06,
      "step": 439,
      "training_step_time": 10.308925151824951
    },
    {
      "epoch": 1.678466796875e-06,
      "grad_norm": 0.8772161602973938,
      "learning_rate": 1e-05,
      "loss": 0.1676,
      "step": 440
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 3.1782913208007812,
      "step": 440
    },
    {
      "dynamics_loss_avg": 0.09656028635799885,
      "epoch": 1.678466796875e-06,
      "step": 440
    },
    {
      "action_loss_avg": 0.09671720322221518,
      "epoch": 1.678466796875e-06,
      "step": 440
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 440,
      "training_step_time": 9.977063417434692
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 3.1438276767730713,
      "step": 440
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 440,
      "training_step_time": 9.933962106704712
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 3.142402410507202,
      "step": 440
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 440,
      "training_step_time": 9.922980546951294
    },
    {
      "epoch": 1.678466796875e-06,
      "model_forward_time": 3.138054370880127,
      "step": 440
    },
    {
      "epoch": 1.678466796875e-06,
      "step": 440,
      "training_step_time": 10.236557245254517
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 3.4653236865997314,
      "step": 441
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 441,
      "training_step_time": 10.247632026672363
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 3.1374526023864746,
      "step": 441
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 441,
      "training_step_time": 9.915595531463623
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 3.131885528564453,
      "step": 441
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 441,
      "training_step_time": 9.913567543029785
    },
    {
      "epoch": 1.682281494140625e-06,
      "model_forward_time": 3.136371612548828,
      "step": 441
    },
    {
      "epoch": 1.682281494140625e-06,
      "step": 441,
      "training_step_time": 10.269165754318237
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 3.1656229496002197,
      "step": 442
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 442,
      "training_step_time": 9.942940711975098
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 3.405965566635132,
      "step": 442
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 442,
      "training_step_time": 10.180437326431274
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 3.1320440769195557,
      "step": 442
    },
    {
      "dynamics_loss_avg": 0.07257813476026058,
      "epoch": 1.68609619140625e-06,
      "step": 442
    },
    {
      "action_loss_avg": 0.05564716421067715,
      "epoch": 1.68609619140625e-06,
      "step": 442
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 442,
      "training_step_time": 9.909189939498901
    },
    {
      "epoch": 1.68609619140625e-06,
      "model_forward_time": 3.125244140625,
      "step": 442
    },
    {
      "epoch": 1.68609619140625e-06,
      "step": 442,
      "training_step_time": 10.318599224090576
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 3.189418077468872,
      "step": 443
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 443,
      "training_step_time": 9.966658353805542
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 3.1321027278900146,
      "step": 443
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 443,
      "training_step_time": 9.908854007720947
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 3.4217779636383057,
      "step": 443
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 443,
      "training_step_time": 10.203074932098389
    },
    {
      "epoch": 1.689910888671875e-06,
      "model_forward_time": 3.1385557651519775,
      "step": 443
    },
    {
      "epoch": 1.689910888671875e-06,
      "step": 443,
      "training_step_time": 10.315380334854126
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 3.182835578918457,
      "step": 444
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 444,
      "training_step_time": 9.964273929595947
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 3.1404993534088135,
      "step": 444
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 444,
      "training_step_time": 9.921648025512695
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 3.1332929134368896,
      "step": 444
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 444,
      "training_step_time": 9.912906408309937
    },
    {
      "epoch": 1.6937255859375e-06,
      "model_forward_time": 3.4063618183135986,
      "step": 444
    },
    {
      "epoch": 1.6937255859375e-06,
      "step": 444,
      "training_step_time": 10.276330947875977
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 3.1863057613372803,
      "step": 445
    },
    {
      "dynamics_loss_avg": 0.08463420532643795,
      "epoch": 1.697540283203125e-06,
      "step": 445
    },
    {
      "action_loss_avg": 0.07129858117550611,
      "epoch": 1.697540283203125e-06,
      "step": 445
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 445,
      "training_step_time": 9.967721939086914
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 3.1277050971984863,
      "step": 445
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 445,
      "training_step_time": 9.907261610031128
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 3.132502317428589,
      "step": 445
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 445,
      "training_step_time": 9.911669492721558
    },
    {
      "epoch": 1.697540283203125e-06,
      "model_forward_time": 3.1355133056640625,
      "step": 445
    },
    {
      "epoch": 1.697540283203125e-06,
      "step": 445,
      "training_step_time": 10.040868759155273
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 3.1549863815307617,
      "step": 446
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 446,
      "training_step_time": 10.251495599746704
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 3.1313884258270264,
      "step": 446
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 446,
      "training_step_time": 9.91398310661316
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 3.1376097202301025,
      "step": 446
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 446,
      "training_step_time": 9.919126510620117
    },
    {
      "epoch": 1.70135498046875e-06,
      "model_forward_time": 3.1333606243133545,
      "step": 446
    },
    {
      "epoch": 1.70135498046875e-06,
      "step": 446,
      "training_step_time": 10.31319284439087
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 3.1474413871765137,
      "step": 447
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 447,
      "training_step_time": 9.925215721130371
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 3.42124605178833,
      "step": 447
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 447,
      "training_step_time": 10.200305938720703
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 3.1431031227111816,
      "step": 447
    },
    {
      "dynamics_loss_avg": 0.097774738073349,
      "epoch": 1.705169677734375e-06,
      "step": 447
    },
    {
      "action_loss_avg": 0.08789874631911516,
      "epoch": 1.705169677734375e-06,
      "step": 447
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 447,
      "training_step_time": 9.926448106765747
    },
    {
      "epoch": 1.705169677734375e-06,
      "model_forward_time": 2.437375783920288,
      "step": 447
    },
    {
      "epoch": 1.705169677734375e-06,
      "step": 447,
      "training_step_time": 10.315305709838867
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 3.1867966651916504,
      "step": 448
    },
    {
      "epoch": 1.708984375e-06,
      "step": 448,
      "training_step_time": 9.96749997138977
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 3.146784782409668,
      "step": 448
    },
    {
      "epoch": 1.708984375e-06,
      "step": 448,
      "training_step_time": 9.926828384399414
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 3.1476619243621826,
      "step": 448
    },
    {
      "epoch": 1.708984375e-06,
      "step": 448,
      "training_step_time": 10.242772579193115
    },
    {
      "epoch": 1.708984375e-06,
      "model_forward_time": 3.145519256591797,
      "step": 448
    },
    {
      "epoch": 1.708984375e-06,
      "step": 448,
      "training_step_time": 10.30550503730774
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 3.181941509246826,
      "step": 449
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 449,
      "training_step_time": 9.972089052200317
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 3.151466131210327,
      "step": 449
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 449,
      "training_step_time": 9.934858322143555
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 3.152355670928955,
      "step": 449
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 449,
      "training_step_time": 9.937612295150757
    },
    {
      "epoch": 1.712799072265625e-06,
      "model_forward_time": 3.4598751068115234,
      "step": 449
    },
    {
      "epoch": 1.712799072265625e-06,
      "step": 449,
      "training_step_time": 10.33494782447815
    },
    {
      "epoch": 1.71661376953125e-06,
      "grad_norm": 0.4439586102962494,
      "learning_rate": 1e-05,
      "loss": 0.1615,
      "step": 450
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 3.171996593475342,
      "step": 450
    },
    {
      "dynamics_loss_avg": 0.07573987655341625,
      "epoch": 1.71661376953125e-06,
      "step": 450
    },
    {
      "action_loss_avg": 0.07210230249911546,
      "epoch": 1.71661376953125e-06,
      "step": 450
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 450,
      "training_step_time": 9.963092803955078
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 3.145669937133789,
      "step": 450
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 450,
      "training_step_time": 9.929829835891724
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 3.1471025943756104,
      "step": 450
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 450,
      "training_step_time": 9.931837797164917
    },
    {
      "epoch": 1.71661376953125e-06,
      "model_forward_time": 3.149798631668091,
      "step": 450
    },
    {
      "epoch": 1.71661376953125e-06,
      "step": 450,
      "training_step_time": 10.304470777511597
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 3.501124858856201,
      "step": 451
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 451,
      "training_step_time": 10.285486459732056
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 3.1371400356292725,
      "step": 451
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 451,
      "training_step_time": 9.925903558731079
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 3.136453628540039,
      "step": 451
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 451,
      "training_step_time": 9.92022156715393
    },
    {
      "epoch": 1.720428466796875e-06,
      "model_forward_time": 3.138202667236328,
      "step": 451
    },
    {
      "epoch": 1.720428466796875e-06,
      "step": 451,
      "training_step_time": 10.31272006034851
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 3.1860694885253906,
      "step": 452
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 452,
      "training_step_time": 9.97099781036377
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 3.4537789821624756,
      "step": 452
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 452,
      "training_step_time": 10.231772422790527
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 3.1383352279663086,
      "step": 452
    },
    {
      "dynamics_loss_avg": 0.0781300738453865,
      "epoch": 1.7242431640625e-06,
      "step": 452
    },
    {
      "action_loss_avg": 0.06159625593572855,
      "epoch": 1.7242431640625e-06,
      "step": 452
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 452,
      "training_step_time": 9.920583009719849
    },
    {
      "epoch": 1.7242431640625e-06,
      "model_forward_time": 3.1484103202819824,
      "step": 452
    },
    {
      "epoch": 1.7242431640625e-06,
      "step": 452,
      "training_step_time": 10.348859548568726
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 3.177131414413452,
      "step": 453
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 453,
      "training_step_time": 9.965241193771362
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 2.4302639961242676,
      "step": 453
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 453,
      "training_step_time": 7.462016344070435
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 3.1385843753814697,
      "step": 453
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 453,
      "training_step_time": 10.19748568534851
    },
    {
      "epoch": 1.728057861328125e-06,
      "model_forward_time": 3.1462924480438232,
      "step": 453
    },
    {
      "epoch": 1.728057861328125e-06,
      "step": 453,
      "training_step_time": 10.341231107711792
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 3.185184955596924,
      "step": 454
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 454,
      "training_step_time": 9.963012218475342
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 3.1436767578125,
      "step": 454
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 454,
      "training_step_time": 9.921048879623413
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 3.1435201168060303,
      "step": 454
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 454,
      "training_step_time": 9.926119089126587
    },
    {
      "epoch": 1.73187255859375e-06,
      "model_forward_time": 3.4645016193389893,
      "step": 454
    },
    {
      "epoch": 1.73187255859375e-06,
      "step": 454,
      "training_step_time": 10.331291437149048
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 3.1862285137176514,
      "step": 455
    },
    {
      "dynamics_loss_avg": 0.09210361130535602,
      "epoch": 1.735687255859375e-06,
      "step": 455
    },
    {
      "action_loss_avg": 0.0859727157279849,
      "epoch": 1.735687255859375e-06,
      "step": 455
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 455,
      "training_step_time": 9.963099718093872
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 3.137852668762207,
      "step": 455
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 455,
      "training_step_time": 9.912901401519775
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 3.142951011657715,
      "step": 455
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 455,
      "training_step_time": 9.920899152755737
    },
    {
      "epoch": 1.735687255859375e-06,
      "model_forward_time": 3.1438605785369873,
      "step": 455
    },
    {
      "epoch": 1.735687255859375e-06,
      "step": 455,
      "training_step_time": 10.284921169281006
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 3.49790096282959,
      "step": 456
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 456,
      "training_step_time": 10.287359476089478
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 3.1347856521606445,
      "step": 456
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 456,
      "training_step_time": 9.91008973121643
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 3.1297831535339355,
      "step": 456
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 456,
      "training_step_time": 9.900917530059814
    },
    {
      "epoch": 1.739501953125e-06,
      "model_forward_time": 3.1354424953460693,
      "step": 456
    },
    {
      "epoch": 1.739501953125e-06,
      "step": 456,
      "training_step_time": 10.0205717086792
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 3.156038761138916,
      "step": 457
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 457,
      "training_step_time": 9.936156034469604
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 3.423651933670044,
      "step": 457
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 457,
      "training_step_time": 10.203993558883667
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 3.136643886566162,
      "step": 457
    },
    {
      "dynamics_loss_avg": 0.07843009792268277,
      "epoch": 1.743316650390625e-06,
      "step": 457
    },
    {
      "action_loss_avg": 0.06655140612274409,
      "epoch": 1.743316650390625e-06,
      "step": 457
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 457,
      "training_step_time": 9.919198274612427
    },
    {
      "epoch": 1.743316650390625e-06,
      "model_forward_time": 3.13797664642334,
      "step": 457
    },
    {
      "epoch": 1.743316650390625e-06,
      "step": 457,
      "training_step_time": 10.280065298080444
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 3.166562795639038,
      "step": 458
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 458,
      "training_step_time": 9.954022645950317
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 3.142538070678711,
      "step": 458
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 458,
      "training_step_time": 9.930083513259888
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 3.41520094871521,
      "step": 458
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 458,
      "training_step_time": 10.199126482009888
    },
    {
      "epoch": 1.74713134765625e-06,
      "model_forward_time": 3.138880968093872,
      "step": 458
    },
    {
      "epoch": 1.74713134765625e-06,
      "step": 458,
      "training_step_time": 10.329689979553223
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 3.1911745071411133,
      "step": 459
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 459,
      "training_step_time": 9.974196910858154
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 3.1374757289886475,
      "step": 459
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 459,
      "training_step_time": 9.916391372680664
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 3.1330149173736572,
      "step": 459
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 459,
      "training_step_time": 9.910526275634766
    },
    {
      "epoch": 1.750946044921875e-06,
      "model_forward_time": 3.430203914642334,
      "step": 459
    },
    {
      "epoch": 1.750946044921875e-06,
      "step": 459,
      "training_step_time": 10.30243968963623
    },
    {
      "epoch": 1.7547607421875e-06,
      "grad_norm": 0.9522196650505066,
      "learning_rate": 1e-05,
      "loss": 0.1557,
      "step": 460
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 3.18725848197937,
      "step": 460
    },
    {
      "dynamics_loss_avg": 0.09192755557596684,
      "epoch": 1.7547607421875e-06,
      "step": 460
    },
    {
      "action_loss_avg": 0.07328061424195767,
      "epoch": 1.7547607421875e-06,
      "step": 460
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 460,
      "training_step_time": 9.965877056121826
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 3.139944076538086,
      "step": 460
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 460,
      "training_step_time": 9.918615818023682
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 3.151233434677124,
      "step": 460
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 460,
      "training_step_time": 9.940072774887085
    },
    {
      "epoch": 1.7547607421875e-06,
      "model_forward_time": 3.143057107925415,
      "step": 460
    },
    {
      "epoch": 1.7547607421875e-06,
      "step": 460,
      "training_step_time": 10.270301342010498
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 3.4971072673797607,
      "step": 461
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 461,
      "training_step_time": 10.281693935394287
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 3.1535582542419434,
      "step": 461
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 461,
      "training_step_time": 9.93229627609253
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 3.145608901977539,
      "step": 461
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 461,
      "training_step_time": 9.926634073257446
    },
    {
      "epoch": 1.758575439453125e-06,
      "model_forward_time": 3.139040470123291,
      "step": 461
    },
    {
      "epoch": 1.758575439453125e-06,
      "step": 461,
      "training_step_time": 10.295710563659668
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 2.4723777770996094,
      "step": 462
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 462,
      "training_step_time": 7.509876728057861
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 3.1435232162475586,
      "step": 462
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 462,
      "training_step_time": 10.2454252243042
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 3.144946813583374,
      "step": 462
    },
    {
      "dynamics_loss_avg": 0.101043751090765,
      "epoch": 1.76239013671875e-06,
      "step": 462
    },
    {
      "action_loss_avg": 0.0984386958181858,
      "epoch": 1.76239013671875e-06,
      "step": 462
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 462,
      "training_step_time": 9.927489519119263
    },
    {
      "epoch": 1.76239013671875e-06,
      "model_forward_time": 3.1464009284973145,
      "step": 462
    },
    {
      "epoch": 1.76239013671875e-06,
      "step": 462,
      "training_step_time": 10.044369220733643
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 3.1977365016937256,
      "step": 463
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 463,
      "training_step_time": 9.980084657669067
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 3.142005681991577,
      "step": 463
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 463,
      "training_step_time": 9.927225351333618
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 3.403407096862793,
      "step": 463
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 463,
      "training_step_time": 10.182058811187744
    },
    {
      "epoch": 1.766204833984375e-06,
      "model_forward_time": 3.1403753757476807,
      "step": 463
    },
    {
      "epoch": 1.766204833984375e-06,
      "step": 463,
      "training_step_time": 10.313843488693237
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 3.1730027198791504,
      "step": 464
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 464,
      "training_step_time": 9.944883584976196
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 3.1337177753448486,
      "step": 464
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 464,
      "training_step_time": 9.90924859046936
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 3.1384620666503906,
      "step": 464
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 464,
      "training_step_time": 9.910046815872192
    },
    {
      "epoch": 1.77001953125e-06,
      "model_forward_time": 3.465545892715454,
      "step": 464
    },
    {
      "epoch": 1.77001953125e-06,
      "step": 464,
      "training_step_time": 10.334518194198608
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 3.1754398345947266,
      "step": 465
    },
    {
      "dynamics_loss_avg": 0.07635417133569718,
      "epoch": 1.773834228515625e-06,
      "step": 465
    },
    {
      "action_loss_avg": 0.07226917557418347,
      "epoch": 1.773834228515625e-06,
      "step": 465
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 465,
      "training_step_time": 9.959099531173706
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 3.1407949924468994,
      "step": 465
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 465,
      "training_step_time": 9.923525333404541
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 3.139983654022217,
      "step": 465
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 465,
      "training_step_time": 9.914644241333008
    },
    {
      "epoch": 1.773834228515625e-06,
      "model_forward_time": 3.1404922008514404,
      "step": 465
    },
    {
      "epoch": 1.773834228515625e-06,
      "step": 465,
      "training_step_time": 10.30141282081604
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 3.165167808532715,
      "step": 466
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 466,
      "training_step_time": 10.253432273864746
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 3.137590169906616,
      "step": 466
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 466,
      "training_step_time": 9.912788152694702
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 3.1418228149414062,
      "step": 466
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 466,
      "training_step_time": 9.918835639953613
    },
    {
      "epoch": 1.77764892578125e-06,
      "model_forward_time": 3.134897470474243,
      "step": 466
    },
    {
      "epoch": 1.77764892578125e-06,
      "step": 466,
      "training_step_time": 10.301721334457397
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 3.176368474960327,
      "step": 467
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 467,
      "training_step_time": 9.954812288284302
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 3.400050401687622,
      "step": 467
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 467,
      "training_step_time": 10.172016382217407
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 3.1369073390960693,
      "step": 467
    },
    {
      "dynamics_loss_avg": 0.07999135032296181,
      "epoch": 1.781463623046875e-06,
      "step": 467
    },
    {
      "action_loss_avg": 0.06666770577430725,
      "epoch": 1.781463623046875e-06,
      "step": 467
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 467,
      "training_step_time": 9.912572145462036
    },
    {
      "epoch": 1.781463623046875e-06,
      "model_forward_time": 3.1351962089538574,
      "step": 467
    },
    {
      "epoch": 1.781463623046875e-06,
      "step": 467,
      "training_step_time": 10.279280185699463
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 3.1729400157928467,
      "step": 468
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 468,
      "training_step_time": 9.953712940216064
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 3.1359081268310547,
      "step": 468
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 468,
      "training_step_time": 9.92009162902832
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 3.1365296840667725,
      "step": 468
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 468,
      "training_step_time": 10.185047626495361
    },
    {
      "epoch": 1.7852783203125e-06,
      "model_forward_time": 3.144482135772705,
      "step": 468
    },
    {
      "epoch": 1.7852783203125e-06,
      "step": 468,
      "training_step_time": 10.33064079284668
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 3.1712188720703125,
      "step": 469
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 469,
      "training_step_time": 9.95210599899292
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 3.1457064151763916,
      "step": 469
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 469,
      "training_step_time": 9.924439191818237
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 3.148005723953247,
      "step": 469
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 469,
      "training_step_time": 9.933804273605347
    },
    {
      "epoch": 1.789093017578125e-06,
      "model_forward_time": 3.445482015609741,
      "step": 469
    },
    {
      "epoch": 1.789093017578125e-06,
      "step": 469,
      "training_step_time": 10.31947636604309
    },
    {
      "epoch": 1.79290771484375e-06,
      "grad_norm": 0.6421778798103333,
      "learning_rate": 1e-05,
      "loss": 0.1594,
      "step": 470
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 3.168919086456299,
      "step": 470
    },
    {
      "dynamics_loss_avg": 0.08500981517136097,
      "epoch": 1.79290771484375e-06,
      "step": 470
    },
    {
      "action_loss_avg": 0.08439320959150791,
      "epoch": 1.79290771484375e-06,
      "step": 470
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 470,
      "training_step_time": 9.947629928588867
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 3.146563768386841,
      "step": 470
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 470,
      "training_step_time": 9.932074069976807
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 3.141706943511963,
      "step": 470
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 470,
      "training_step_time": 9.924745321273804
    },
    {
      "epoch": 1.79290771484375e-06,
      "model_forward_time": 3.146820306777954,
      "step": 470
    },
    {
      "epoch": 1.79290771484375e-06,
      "step": 470,
      "training_step_time": 10.299322605133057
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 3.183931350708008,
      "step": 471
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 471,
      "training_step_time": 10.225339412689209
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 3.1346676349639893,
      "step": 471
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 471,
      "training_step_time": 9.906642198562622
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 3.131669044494629,
      "step": 471
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 471,
      "training_step_time": 9.906911611557007
    },
    {
      "epoch": 1.796722412109375e-06,
      "model_forward_time": 3.130025863647461,
      "step": 471
    },
    {
      "epoch": 1.796722412109375e-06,
      "step": 471,
      "training_step_time": 10.270147562026978
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 3.1593196392059326,
      "step": 472
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 472,
      "training_step_time": 9.931905269622803
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 3.403289556503296,
      "step": 472
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 472,
      "training_step_time": 10.188538551330566
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 3.1312854290008545,
      "step": 472
    },
    {
      "dynamics_loss_avg": 0.07440120540559292,
      "epoch": 1.800537109375e-06,
      "step": 472
    },
    {
      "action_loss_avg": 0.05981746576726436,
      "epoch": 1.800537109375e-06,
      "step": 472
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 472,
      "training_step_time": 9.90446400642395
    },
    {
      "epoch": 1.800537109375e-06,
      "model_forward_time": 3.136481523513794,
      "step": 472
    },
    {
      "epoch": 1.800537109375e-06,
      "step": 472,
      "training_step_time": 10.288677215576172
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 3.1805830001831055,
      "step": 473
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 473,
      "training_step_time": 9.951567649841309
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 3.1333024501800537,
      "step": 473
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 473,
      "training_step_time": 9.904129981994629
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 3.422168731689453,
      "step": 473
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 473,
      "training_step_time": 10.188710689544678
    },
    {
      "epoch": 1.804351806640625e-06,
      "model_forward_time": 3.1326160430908203,
      "step": 473
    },
    {
      "epoch": 1.804351806640625e-06,
      "step": 473,
      "training_step_time": 10.317706823348999
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 3.199577569961548,
      "step": 474
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 474,
      "training_step_time": 9.9696946144104
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 3.1361348628997803,
      "step": 474
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 474,
      "training_step_time": 9.904947519302368
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 3.138627052307129,
      "step": 474
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 474,
      "training_step_time": 9.907062530517578
    },
    {
      "epoch": 1.80816650390625e-06,
      "model_forward_time": 3.451113224029541,
      "step": 474
    },
    {
      "epoch": 1.80816650390625e-06,
      "step": 474,
      "training_step_time": 10.31670594215393
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 3.1615524291992188,
      "step": 475
    },
    {
      "dynamics_loss_avg": 0.09837793782353402,
      "epoch": 1.811981201171875e-06,
      "step": 475
    },
    {
      "action_loss_avg": 0.07976069767028093,
      "epoch": 1.811981201171875e-06,
      "step": 475
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 475,
      "training_step_time": 9.950927257537842
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 3.1325647830963135,
      "step": 475
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 475,
      "training_step_time": 9.898813009262085
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 3.133340835571289,
      "step": 475
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 475,
      "training_step_time": 9.903631210327148
    },
    {
      "epoch": 1.811981201171875e-06,
      "model_forward_time": 3.131434917449951,
      "step": 475
    },
    {
      "epoch": 1.811981201171875e-06,
      "step": 475,
      "training_step_time": 10.27350926399231
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 3.5081212520599365,
      "step": 476
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 476,
      "training_step_time": 10.273733854293823
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 3.1402173042297363,
      "step": 476
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 476,
      "training_step_time": 9.904710531234741
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 3.1463265419006348,
      "step": 476
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 476,
      "training_step_time": 9.918198108673096
    },
    {
      "epoch": 1.8157958984375e-06,
      "model_forward_time": 3.138106107711792,
      "step": 476
    },
    {
      "epoch": 1.8157958984375e-06,
      "step": 476,
      "training_step_time": 10.305546998977661
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 3.193230628967285,
      "step": 477
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 477,
      "training_step_time": 9.970579385757446
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 3.4062483310699463,
      "step": 477
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 477,
      "training_step_time": 10.176007270812988
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 3.140929937362671,
      "step": 477
    },
    {
      "dynamics_loss_avg": 0.08304062485694885,
      "epoch": 1.819610595703125e-06,
      "step": 477
    },
    {
      "action_loss_avg": 0.06776346452534199,
      "epoch": 1.819610595703125e-06,
      "step": 477
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 477,
      "training_step_time": 9.91338038444519
    },
    {
      "epoch": 1.819610595703125e-06,
      "model_forward_time": 3.137413263320923,
      "step": 477
    },
    {
      "epoch": 1.819610595703125e-06,
      "step": 477,
      "training_step_time": 10.261737585067749
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 3.189368486404419,
      "step": 478
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 478,
      "training_step_time": 9.957841634750366
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 3.1400396823883057,
      "step": 478
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 478,
      "training_step_time": 9.904820919036865
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 3.447523593902588,
      "step": 478
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 478,
      "training_step_time": 10.207510471343994
    },
    {
      "epoch": 1.82342529296875e-06,
      "model_forward_time": 3.1351242065429688,
      "step": 478
    },
    {
      "epoch": 1.82342529296875e-06,
      "step": 478,
      "training_step_time": 10.25345516204834
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 3.161494255065918,
      "step": 479
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 479,
      "training_step_time": 9.927377939224243
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 3.136857271194458,
      "step": 479
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 479,
      "training_step_time": 9.90598177909851
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 3.1382861137390137,
      "step": 479
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 479,
      "training_step_time": 9.90695834159851
    },
    {
      "epoch": 1.827239990234375e-06,
      "model_forward_time": 3.4247281551361084,
      "step": 479
    },
    {
      "epoch": 1.827239990234375e-06,
      "step": 479,
      "training_step_time": 10.28597903251648
    },
    {
      "epoch": 1.8310546875e-06,
      "grad_norm": 0.981899082660675,
      "learning_rate": 1e-05,
      "loss": 0.1565,
      "step": 480
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 3.176771402359009,
      "step": 480
    },
    {
      "dynamics_loss_avg": 0.07141793128103018,
      "epoch": 1.8310546875e-06,
      "step": 480
    },
    {
      "action_loss_avg": 0.049048691522330046,
      "epoch": 1.8310546875e-06,
      "step": 480
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 480,
      "training_step_time": 9.954783916473389
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 3.139277458190918,
      "step": 480
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 480,
      "training_step_time": 9.907734155654907
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 3.1346020698547363,
      "step": 480
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 480,
      "training_step_time": 9.901728868484497
    },
    {
      "epoch": 1.8310546875e-06,
      "model_forward_time": 3.14811635017395,
      "step": 480
    },
    {
      "epoch": 1.8310546875e-06,
      "step": 480,
      "training_step_time": 10.264649391174316
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 3.4460179805755615,
      "step": 481
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 481,
      "training_step_time": 10.213523864746094
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 3.1323037147521973,
      "step": 481
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 481,
      "training_step_time": 9.901596546173096
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 3.1335692405700684,
      "step": 481
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 481,
      "training_step_time": 9.908748388290405
    },
    {
      "epoch": 1.834869384765625e-06,
      "model_forward_time": 3.1300830841064453,
      "step": 481
    },
    {
      "epoch": 1.834869384765625e-06,
      "step": 481,
      "training_step_time": 10.281100749969482
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 2.4649760723114014,
      "step": 482
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 482,
      "training_step_time": 7.4971795082092285
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 3.137000799179077,
      "step": 482
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 482,
      "training_step_time": 10.18941879272461
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 3.1372387409210205,
      "step": 482
    },
    {
      "dynamics_loss_avg": 0.08823522236198186,
      "epoch": 1.83868408203125e-06,
      "step": 482
    },
    {
      "action_loss_avg": 0.07559448499232531,
      "epoch": 1.83868408203125e-06,
      "step": 482
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 482,
      "training_step_time": 9.911618709564209
    },
    {
      "epoch": 1.83868408203125e-06,
      "model_forward_time": 3.139552116394043,
      "step": 482
    },
    {
      "epoch": 1.83868408203125e-06,
      "step": 482,
      "training_step_time": 10.236046552658081
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 3.1735565662384033,
      "step": 483
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 483,
      "training_step_time": 9.950571775436401
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 3.1338088512420654,
      "step": 483
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 483,
      "training_step_time": 9.906343698501587
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 3.4215261936187744,
      "step": 483
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 483,
      "training_step_time": 10.192766666412354
    },
    {
      "epoch": 1.842498779296875e-06,
      "model_forward_time": 3.1342031955718994,
      "step": 483
    },
    {
      "epoch": 1.842498779296875e-06,
      "step": 483,
      "training_step_time": 10.316642045974731
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 3.1844258308410645,
      "step": 484
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 484,
      "training_step_time": 9.951722621917725
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 3.131218194961548,
      "step": 484
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 484,
      "training_step_time": 9.894774913787842
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 3.1359000205993652,
      "step": 484
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 484,
      "training_step_time": 9.905982971191406
    },
    {
      "epoch": 1.8463134765625e-06,
      "model_forward_time": 3.4091615676879883,
      "step": 484
    },
    {
      "epoch": 1.8463134765625e-06,
      "step": 484,
      "training_step_time": 10.279602527618408
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 3.1810781955718994,
      "step": 485
    },
    {
      "dynamics_loss_avg": 0.09218080341815948,
      "epoch": 1.850128173828125e-06,
      "step": 485
    },
    {
      "action_loss_avg": 0.08769168332219124,
      "epoch": 1.850128173828125e-06,
      "step": 485
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 485,
      "training_step_time": 9.958261251449585
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 3.133450746536255,
      "step": 485
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 485,
      "training_step_time": 9.902046918869019
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 3.1339638233184814,
      "step": 485
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 485,
      "training_step_time": 9.90588116645813
    },
    {
      "epoch": 1.850128173828125e-06,
      "model_forward_time": 3.13008451461792,
      "step": 485
    },
    {
      "epoch": 1.850128173828125e-06,
      "step": 485,
      "training_step_time": 10.037574052810669
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 3.1591436862945557,
      "step": 486
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 486,
      "training_step_time": 10.240543365478516
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 3.1332602500915527,
      "step": 486
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 486,
      "training_step_time": 9.911850690841675
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 3.1405527591705322,
      "step": 486
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 486,
      "training_step_time": 9.919551372528076
    },
    {
      "epoch": 1.85394287109375e-06,
      "model_forward_time": 3.1372194290161133,
      "step": 486
    },
    {
      "epoch": 1.85394287109375e-06,
      "step": 486,
      "training_step_time": 10.320797443389893
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 3.180356979370117,
      "step": 487
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 487,
      "training_step_time": 9.953242540359497
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 3.4036543369293213,
      "step": 487
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 487,
      "training_step_time": 10.17869520187378
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 3.1415231227874756,
      "step": 487
    },
    {
      "dynamics_loss_avg": 0.09882091768085957,
      "epoch": 1.857757568359375e-06,
      "step": 487
    },
    {
      "action_loss_avg": 0.0935470324009657,
      "epoch": 1.857757568359375e-06,
      "step": 487
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 487,
      "training_step_time": 9.920802593231201
    },
    {
      "epoch": 1.857757568359375e-06,
      "model_forward_time": 3.1439905166625977,
      "step": 487
    },
    {
      "epoch": 1.857757568359375e-06,
      "step": 487,
      "training_step_time": 10.031358480453491
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 3.1729815006256104,
      "step": 488
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 488,
      "training_step_time": 9.95033860206604
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 3.1357526779174805,
      "step": 488
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 488,
      "training_step_time": 9.912141561508179
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 3.1355745792388916,
      "step": 488
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 488,
      "training_step_time": 10.178369283676147
    },
    {
      "epoch": 1.861572265625e-06,
      "model_forward_time": 3.1368701457977295,
      "step": 488
    },
    {
      "epoch": 1.861572265625e-06,
      "step": 488,
      "training_step_time": 10.301818132400513
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 3.1769397258758545,
      "step": 489
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 489,
      "training_step_time": 9.954897165298462
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 3.1458065509796143,
      "step": 489
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 489,
      "training_step_time": 9.92580771446228
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 3.140134572982788,
      "step": 489
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 489,
      "training_step_time": 9.916275978088379
    },
    {
      "epoch": 1.865386962890625e-06,
      "model_forward_time": 3.4339613914489746,
      "step": 489
    },
    {
      "epoch": 1.865386962890625e-06,
      "step": 489,
      "training_step_time": 10.326007604598999
    },
    {
      "epoch": 1.86920166015625e-06,
      "grad_norm": 0.9243882894515991,
      "learning_rate": 1e-05,
      "loss": 0.1634,
      "step": 490
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 3.1756179332733154,
      "step": 490
    },
    {
      "dynamics_loss_avg": 0.0976451389491558,
      "epoch": 1.86920166015625e-06,
      "step": 490
    },
    {
      "action_loss_avg": 0.09643834121525288,
      "epoch": 1.86920166015625e-06,
      "step": 490
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 490,
      "training_step_time": 9.958166122436523
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 3.142457962036133,
      "step": 490
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 490,
      "training_step_time": 9.922369718551636
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 3.1436197757720947,
      "step": 490
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 490,
      "training_step_time": 9.925067901611328
    },
    {
      "epoch": 1.86920166015625e-06,
      "model_forward_time": 3.1457252502441406,
      "step": 490
    },
    {
      "epoch": 1.86920166015625e-06,
      "step": 490,
      "training_step_time": 10.033028364181519
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 3.1683473587036133,
      "step": 491
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 491,
      "training_step_time": 10.2213716506958
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 3.1407158374786377,
      "step": 491
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 491,
      "training_step_time": 9.919179201126099
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 3.138930320739746,
      "step": 491
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 491,
      "training_step_time": 9.919059038162231
    },
    {
      "epoch": 1.873016357421875e-06,
      "model_forward_time": 3.1416711807250977,
      "step": 491
    },
    {
      "epoch": 1.873016357421875e-06,
      "step": 491,
      "training_step_time": 10.315493822097778
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 3.1740105152130127,
      "step": 492
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 492,
      "training_step_time": 9.957874774932861
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 3.412353038787842,
      "step": 492
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 492,
      "training_step_time": 10.18891716003418
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 3.139091730117798,
      "step": 492
    },
    {
      "dynamics_loss_avg": 0.08580904081463814,
      "epoch": 1.8768310546875e-06,
      "step": 492
    },
    {
      "action_loss_avg": 0.07523204050958157,
      "epoch": 1.8768310546875e-06,
      "step": 492
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 492,
      "training_step_time": 9.919246196746826
    },
    {
      "epoch": 1.8768310546875e-06,
      "model_forward_time": 3.1343798637390137,
      "step": 492
    },
    {
      "epoch": 1.8768310546875e-06,
      "step": 492,
      "training_step_time": 10.259848833084106
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 3.166290283203125,
      "step": 493
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 493,
      "training_step_time": 9.941639184951782
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 3.1383216381073,
      "step": 493
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 493,
      "training_step_time": 9.91176962852478
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 3.4360101222991943,
      "step": 493
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 493,
      "training_step_time": 10.210033178329468
    },
    {
      "epoch": 1.880645751953125e-06,
      "model_forward_time": 3.1405155658721924,
      "step": 493
    },
    {
      "epoch": 1.880645751953125e-06,
      "step": 493,
      "training_step_time": 10.263294696807861
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 3.150275707244873,
      "step": 494
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 494,
      "training_step_time": 9.930567264556885
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 3.1390128135681152,
      "step": 494
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 494,
      "training_step_time": 9.921802997589111
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 3.140894651412964,
      "step": 494
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 494,
      "training_step_time": 9.923628807067871
    },
    {
      "epoch": 1.88446044921875e-06,
      "model_forward_time": 2.4369447231292725,
      "step": 494
    },
    {
      "epoch": 1.88446044921875e-06,
      "step": 494,
      "training_step_time": 10.302771806716919
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 3.164149761199951,
      "step": 495
    },
    {
      "dynamics_loss_avg": 0.09150700606405734,
      "epoch": 1.888275146484375e-06,
      "step": 495
    },
    {
      "action_loss_avg": 0.09271156191825866,
      "epoch": 1.888275146484375e-06,
      "step": 495
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 495,
      "training_step_time": 9.951852798461914
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 3.1402385234832764,
      "step": 495
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 495,
      "training_step_time": 9.921931028366089
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 3.1401875019073486,
      "step": 495
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 495,
      "training_step_time": 9.91866421699524
    },
    {
      "epoch": 1.888275146484375e-06,
      "model_forward_time": 3.137239694595337,
      "step": 495
    },
    {
      "epoch": 1.888275146484375e-06,
      "step": 495,
      "training_step_time": 10.01879072189331
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 3.4420440196990967,
      "step": 496
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 496,
      "training_step_time": 10.224562644958496
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 3.1337733268737793,
      "step": 496
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 496,
      "training_step_time": 9.908293962478638
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 3.1326279640197754,
      "step": 496
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 496,
      "training_step_time": 9.905365943908691
    },
    {
      "epoch": 1.89208984375e-06,
      "model_forward_time": 3.1323623657226562,
      "step": 496
    },
    {
      "epoch": 1.89208984375e-06,
      "step": 496,
      "training_step_time": 10.227457284927368
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 3.177642345428467,
      "step": 497
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 497,
      "training_step_time": 9.949875593185425
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 3.1465840339660645,
      "step": 497
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 497,
      "training_step_time": 10.235329627990723
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 3.1408231258392334,
      "step": 497
    },
    {
      "dynamics_loss_avg": 0.089717922732234,
      "epoch": 1.895904541015625e-06,
      "step": 497
    },
    {
      "action_loss_avg": 0.07910416498780251,
      "epoch": 1.895904541015625e-06,
      "step": 497
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 497,
      "training_step_time": 9.919947862625122
    },
    {
      "epoch": 1.895904541015625e-06,
      "model_forward_time": 3.137907028198242,
      "step": 497
    },
    {
      "epoch": 1.895904541015625e-06,
      "step": 497,
      "training_step_time": 10.02137804031372
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 3.182314157485962,
      "step": 498
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 498,
      "training_step_time": 9.958905220031738
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 3.1329309940338135,
      "step": 498
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 498,
      "training_step_time": 9.906150102615356
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 3.4414641857147217,
      "step": 498
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 498,
      "training_step_time": 10.210112571716309
    },
    {
      "epoch": 1.89971923828125e-06,
      "model_forward_time": 3.1337966918945312,
      "step": 498
    },
    {
      "epoch": 1.89971923828125e-06,
      "step": 498,
      "training_step_time": 10.297138214111328
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 3.2127575874328613,
      "step": 499
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 499,
      "training_step_time": 9.989658117294312
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 3.138974666595459,
      "step": 499
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 499,
      "training_step_time": 9.91469430923462
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 3.1352593898773193,
      "step": 499
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 499,
      "training_step_time": 9.909717082977295
    },
    {
      "epoch": 1.903533935546875e-06,
      "model_forward_time": 3.136690616607666,
      "step": 499
    },
    {
      "epoch": 1.903533935546875e-06,
      "step": 499,
      "training_step_time": 10.284488439559937
    },
    {
      "epoch": 1.9073486328125e-06,
      "grad_norm": 0.5334572792053223,
      "learning_rate": 1e-05,
      "loss": 0.154,
      "step": 500
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 3.178035020828247,
      "step": 500
    },
    {
      "dynamics_loss_avg": 0.08366060666739941,
      "epoch": 1.9073486328125e-06,
      "step": 500
    },
    {
      "action_loss_avg": 0.0693917952477932,
      "epoch": 1.9073486328125e-06,
      "step": 500
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 500,
      "training_step_time": 9.93990683555603
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 3.1292500495910645,
      "step": 500
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 500,
      "training_step_time": 9.883908748626709
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 3.1246581077575684,
      "step": 500
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 500,
      "training_step_time": 9.880820989608765
    },
    {
      "epoch": 1.9073486328125e-06,
      "model_forward_time": 5.158097743988037,
      "step": 500
    },
    {
      "epoch": 1.9073486328125e-06,
      "step": 500,
      "training_step_time": 12.033650636672974
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 3.1791844367980957,
      "step": 501
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 501,
      "training_step_time": 9.944398641586304
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 3.1357195377349854,
      "step": 501
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 501,
      "training_step_time": 9.907575130462646
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 3.137643337249756,
      "step": 501
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 501,
      "training_step_time": 9.917892694473267
    },
    {
      "epoch": 1.911163330078125e-06,
      "model_forward_time": 3.138930559158325,
      "step": 501
    },
    {
      "epoch": 1.911163330078125e-06,
      "step": 501,
      "training_step_time": 10.272027969360352
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 3.151381492614746,
      "step": 502
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 502,
      "training_step_time": 10.253279685974121
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 3.1427292823791504,
      "step": 502
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 502,
      "training_step_time": 9.924020051956177
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 3.1390230655670166,
      "step": 502
    },
    {
      "dynamics_loss_avg": 0.09042033068835735,
      "epoch": 1.91497802734375e-06,
      "step": 502
    },
    {
      "action_loss_avg": 0.07750614751130343,
      "epoch": 1.91497802734375e-06,
      "step": 502
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 502,
      "training_step_time": 9.92337965965271
    },
    {
      "epoch": 1.91497802734375e-06,
      "model_forward_time": 2.438424587249756,
      "step": 502
    },
    {
      "epoch": 1.91497802734375e-06,
      "step": 502,
      "training_step_time": 10.027950763702393
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 3.189606189727783,
      "step": 503
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 503,
      "training_step_time": 9.963984966278076
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 3.133741617202759,
      "step": 503
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 503,
      "training_step_time": 10.217759370803833
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 3.133174419403076,
      "step": 503
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 503,
      "training_step_time": 9.90490984916687
    },
    {
      "epoch": 1.918792724609375e-06,
      "model_forward_time": 3.1360650062561035,
      "step": 503
    },
    {
      "epoch": 1.918792724609375e-06,
      "step": 503,
      "training_step_time": 10.281832695007324
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 3.165961980819702,
      "step": 504
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 504,
      "training_step_time": 9.933213233947754
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 3.133260488510132,
      "step": 504
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 504,
      "training_step_time": 9.902986526489258
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 3.4272396564483643,
      "step": 504
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 504,
      "training_step_time": 10.19877314567566
    },
    {
      "epoch": 1.922607421875e-06,
      "model_forward_time": 3.139892101287842,
      "step": 504
    },
    {
      "epoch": 1.922607421875e-06,
      "step": 504,
      "training_step_time": 10.335692405700684
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 3.172664165496826,
      "step": 505
    },
    {
      "dynamics_loss_avg": 0.08792281113564968,
      "epoch": 1.926422119140625e-06,
      "step": 505
    },
    {
      "action_loss_avg": 0.07112548723816872,
      "epoch": 1.926422119140625e-06,
      "step": 505
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 505,
      "training_step_time": 9.951352834701538
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 3.137268304824829,
      "step": 505
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 505,
      "training_step_time": 9.90720009803772
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 3.1378867626190186,
      "step": 505
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 505,
      "training_step_time": 9.909422397613525
    },
    {
      "epoch": 1.926422119140625e-06,
      "model_forward_time": 3.138070583343506,
      "step": 505
    },
    {
      "epoch": 1.926422119140625e-06,
      "step": 505,
      "training_step_time": 10.289814472198486
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 3.175154685974121,
      "step": 506
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 506,
      "training_step_time": 9.955820083618164
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 3.142120599746704,
      "step": 506
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 506,
      "training_step_time": 9.918851137161255
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 3.139004945755005,
      "step": 506
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 506,
      "training_step_time": 9.916220903396606
    },
    {
      "epoch": 1.93023681640625e-06,
      "model_forward_time": 3.1363213062286377,
      "step": 506
    },
    {
      "epoch": 1.93023681640625e-06,
      "step": 506,
      "training_step_time": 10.226663589477539
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 3.4984943866729736,
      "step": 507
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 507,
      "training_step_time": 10.277356386184692
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 3.141561269760132,
      "step": 507
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 507,
      "training_step_time": 9.920970916748047
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 3.142178535461426,
      "step": 507
    },
    {
      "dynamics_loss_avg": 0.111003065854311,
      "epoch": 1.934051513671875e-06,
      "step": 507
    },
    {
      "action_loss_avg": 0.08903357833623886,
      "epoch": 1.934051513671875e-06,
      "step": 507
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 507,
      "training_step_time": 9.92636752128601
    },
    {
      "epoch": 1.934051513671875e-06,
      "model_forward_time": 3.141209602355957,
      "step": 507
    },
    {
      "epoch": 1.934051513671875e-06,
      "step": 507,
      "training_step_time": 10.267384767532349
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 3.1787919998168945,
      "step": 508
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 508,
      "training_step_time": 9.958832263946533
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 3.1444084644317627,
      "step": 508
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 508,
      "training_step_time": 10.191808462142944
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 3.1381421089172363,
      "step": 508
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 508,
      "training_step_time": 9.915070533752441
    },
    {
      "epoch": 1.9378662109375e-06,
      "model_forward_time": 3.136854410171509,
      "step": 508
    },
    {
      "epoch": 1.9378662109375e-06,
      "step": 508,
      "training_step_time": 10.027076005935669
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 3.1739730834960938,
      "step": 509
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 509,
      "training_step_time": 9.947036981582642
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 3.1433472633361816,
      "step": 509
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 509,
      "training_step_time": 9.914780855178833
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 3.432805061340332,
      "step": 509
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 509,
      "training_step_time": 10.21094274520874
    },
    {
      "epoch": 1.941680908203125e-06,
      "model_forward_time": 3.137012481689453,
      "step": 509
    },
    {
      "epoch": 1.941680908203125e-06,
      "step": 509,
      "training_step_time": 10.302088975906372
    },
    {
      "epoch": 1.94549560546875e-06,
      "grad_norm": 1.0223287343978882,
      "learning_rate": 1e-05,
      "loss": 0.1695,
      "step": 510
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 3.1845791339874268,
      "step": 510
    },
    {
      "dynamics_loss_avg": 0.0819399856030941,
      "epoch": 1.94549560546875e-06,
      "step": 510
    },
    {
      "action_loss_avg": 0.07703893929719925,
      "epoch": 1.94549560546875e-06,
      "step": 510
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 510,
      "training_step_time": 9.96567440032959
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 3.144371747970581,
      "step": 510
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 510,
      "training_step_time": 9.922322511672974
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 3.1394946575164795,
      "step": 510
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 510,
      "training_step_time": 9.961910009384155
    },
    {
      "epoch": 1.94549560546875e-06,
      "model_forward_time": 3.433257579803467,
      "step": 510
    },
    {
      "epoch": 1.94549560546875e-06,
      "step": 510,
      "training_step_time": 10.301163911819458
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 3.172264814376831,
      "step": 511
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 511,
      "training_step_time": 9.947259426116943
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 3.1414406299591064,
      "step": 511
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 511,
      "training_step_time": 9.916569232940674
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 3.1384313106536865,
      "step": 511
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 511,
      "training_step_time": 9.911914110183716
    },
    {
      "epoch": 1.949310302734375e-06,
      "model_forward_time": 2.4330618381500244,
      "step": 511
    },
    {
      "epoch": 1.949310302734375e-06,
      "step": 511,
      "training_step_time": 10.24422025680542
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 3.462123155593872,
      "step": 512
    },
    {
      "epoch": 1.953125e-06,
      "step": 512,
      "training_step_time": 10.23263669013977
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 3.142180919647217,
      "step": 512
    },
    {
      "epoch": 1.953125e-06,
      "step": 512,
      "training_step_time": 9.914212703704834
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 3.140639066696167,
      "step": 512
    },
    {
      "dynamics_loss_avg": 0.08745889160782098,
      "epoch": 1.953125e-06,
      "step": 512
    },
    {
      "action_loss_avg": 0.0821992240846157,
      "epoch": 1.953125e-06,
      "step": 512
    },
    {
      "epoch": 1.953125e-06,
      "step": 512,
      "training_step_time": 9.914826154708862
    },
    {
      "epoch": 1.953125e-06,
      "model_forward_time": 3.149489402770996,
      "step": 512
    },
    {
      "epoch": 1.953125e-06,
      "step": 512,
      "training_step_time": 10.273098468780518
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 3.1680784225463867,
      "step": 513
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 513,
      "training_step_time": 9.945585012435913
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 3.139246940612793,
      "step": 513
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 513,
      "training_step_time": 10.209899663925171
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 3.1385481357574463,
      "step": 513
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 513,
      "training_step_time": 9.911667108535767
    },
    {
      "epoch": 1.956939697265625e-06,
      "model_forward_time": 3.139071226119995,
      "step": 513
    },
    {
      "epoch": 1.956939697265625e-06,
      "step": 513,
      "training_step_time": 10.267744064331055
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 3.1801910400390625,
      "step": 514
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 514,
      "training_step_time": 9.956237077713013
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 3.1415514945983887,
      "step": 514
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 514,
      "training_step_time": 9.92302656173706
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 2.4374399185180664,
      "step": 514
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 514,
      "training_step_time": 7.740957975387573
    },
    {
      "epoch": 1.96075439453125e-06,
      "model_forward_time": 3.1450157165527344,
      "step": 514
    },
    {
      "epoch": 1.96075439453125e-06,
      "step": 514,
      "training_step_time": 10.289829015731812
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 3.169419527053833,
      "step": 515
    },
    {
      "dynamics_loss_avg": 0.09943075254559516,
      "epoch": 1.964569091796875e-06,
      "step": 515
    },
    {
      "action_loss_avg": 0.0790569819509983,
      "epoch": 1.964569091796875e-06,
      "step": 515
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 515,
      "training_step_time": 9.960906505584717
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 3.1355087757110596,
      "step": 515
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 515,
      "training_step_time": 9.912400484085083
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 3.137265682220459,
      "step": 515
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 515,
      "training_step_time": 9.91809892654419
    },
    {
      "epoch": 1.964569091796875e-06,
      "model_forward_time": 3.423375368118286,
      "step": 515
    },
    {
      "epoch": 1.964569091796875e-06,
      "step": 515,
      "training_step_time": 10.346627950668335
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 3.177816867828369,
      "step": 516
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 516,
      "training_step_time": 9.955511569976807
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 3.1360342502593994,
      "step": 516
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 516,
      "training_step_time": 9.905827045440674
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 3.1346383094787598,
      "step": 516
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 516,
      "training_step_time": 9.907929420471191
    },
    {
      "epoch": 1.9683837890625e-06,
      "model_forward_time": 3.1318604946136475,
      "step": 516
    },
    {
      "epoch": 1.9683837890625e-06,
      "step": 516,
      "training_step_time": 10.017529726028442
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 3.1673390865325928,
      "step": 517
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 517,
      "training_step_time": 10.23766040802002
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 3.1337640285491943,
      "step": 517
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 517,
      "training_step_time": 9.8989098072052
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 3.1387438774108887,
      "step": 517
    },
    {
      "dynamics_loss_avg": 0.0945329513400793,
      "epoch": 1.972198486328125e-06,
      "step": 517
    },
    {
      "action_loss_avg": 0.08158970102667809,
      "epoch": 1.972198486328125e-06,
      "step": 517
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 517,
      "training_step_time": 9.907941579818726
    },
    {
      "epoch": 1.972198486328125e-06,
      "model_forward_time": 3.1367268562316895,
      "step": 517
    },
    {
      "epoch": 1.972198486328125e-06,
      "step": 517,
      "training_step_time": 10.274648904800415
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 3.171663761138916,
      "step": 518
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 518,
      "training_step_time": 9.941324949264526
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 2.7148752212524414,
      "step": 518
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 518,
      "training_step_time": 7.74354362487793
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 3.1375606060028076,
      "step": 518
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 518,
      "training_step_time": 9.909037113189697
    },
    {
      "epoch": 1.97601318359375e-06,
      "model_forward_time": 3.1357765197753906,
      "step": 518
    },
    {
      "epoch": 1.97601318359375e-06,
      "step": 518,
      "training_step_time": 10.295531511306763
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 3.1787590980529785,
      "step": 519
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 519,
      "training_step_time": 9.95525050163269
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 3.1434383392333984,
      "step": 519
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 519,
      "training_step_time": 9.919179201126099
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 3.1403088569641113,
      "step": 519
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 519,
      "training_step_time": 10.229999542236328
    },
    {
      "epoch": 1.979827880859375e-06,
      "model_forward_time": 3.14703106880188,
      "step": 519
    },
    {
      "epoch": 1.979827880859375e-06,
      "step": 519,
      "training_step_time": 10.027648687362671
    },
    {
      "epoch": 1.983642578125e-06,
      "grad_norm": 0.6666030883789062,
      "learning_rate": 1e-05,
      "loss": 0.1583,
      "step": 520
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 3.177109718322754,
      "step": 520
    },
    {
      "dynamics_loss_avg": 0.09107537865638733,
      "epoch": 1.983642578125e-06,
      "step": 520
    },
    {
      "action_loss_avg": 0.07724033668637276,
      "epoch": 1.983642578125e-06,
      "step": 520
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 520,
      "training_step_time": 9.959463834762573
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 3.14663028717041,
      "step": 520
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 520,
      "training_step_time": 9.923577547073364
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 3.140291929244995,
      "step": 520
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 520,
      "training_step_time": 9.916687965393066
    },
    {
      "epoch": 1.983642578125e-06,
      "model_forward_time": 3.4047927856445312,
      "step": 520
    },
    {
      "epoch": 1.983642578125e-06,
      "step": 520,
      "training_step_time": 10.285401582717896
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 3.164264678955078,
      "step": 521
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 521,
      "training_step_time": 9.936733484268188
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 3.140951633453369,
      "step": 521
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 521,
      "training_step_time": 9.910828113555908
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 3.1373343467712402,
      "step": 521
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 521,
      "training_step_time": 9.909274101257324
    },
    {
      "epoch": 1.987457275390625e-06,
      "model_forward_time": 3.1492724418640137,
      "step": 521
    },
    {
      "epoch": 1.987457275390625e-06,
      "step": 521,
      "training_step_time": 10.299265146255493
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 3.422654390335083,
      "step": 522
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 522,
      "training_step_time": 10.194159984588623
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 3.1397602558135986,
      "step": 522
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 522,
      "training_step_time": 9.910149097442627
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 3.137620687484741,
      "step": 522
    },
    {
      "dynamics_loss_avg": 0.09518205039203168,
      "epoch": 1.99127197265625e-06,
      "step": 522
    },
    {
      "action_loss_avg": 0.07785051763057708,
      "epoch": 1.99127197265625e-06,
      "step": 522
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 522,
      "training_step_time": 9.910590410232544
    },
    {
      "epoch": 1.99127197265625e-06,
      "model_forward_time": 3.138923168182373,
      "step": 522
    },
    {
      "epoch": 1.99127197265625e-06,
      "step": 522,
      "training_step_time": 10.215434551239014
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 3.1833033561706543,
      "step": 523
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 523,
      "training_step_time": 9.952917098999023
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 3.1345839500427246,
      "step": 523
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 523,
      "training_step_time": 10.226367950439453
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 3.1394193172454834,
      "step": 523
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 523,
      "training_step_time": 9.912339448928833
    },
    {
      "epoch": 1.995086669921875e-06,
      "model_forward_time": 3.13887619972229,
      "step": 523
    },
    {
      "epoch": 1.995086669921875e-06,
      "step": 523,
      "training_step_time": 10.304808378219604
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 3.184004068374634,
      "step": 524
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 524,
      "training_step_time": 9.957074165344238
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 2.436631202697754,
      "step": 524
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 524,
      "training_step_time": 7.470183610916138
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 3.1380860805511475,
      "step": 524
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 524,
      "training_step_time": 9.918165922164917
    },
    {
      "epoch": 1.9989013671875e-06,
      "model_forward_time": 3.423011541366577,
      "step": 524
    },
    {
      "epoch": 1.9989013671875e-06,
      "step": 524,
      "training_step_time": 10.295082807540894
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 3.166232109069824,
      "step": 525
    },
    {
      "dynamics_loss_avg": 0.09925191104412079,
      "epoch": 2.002716064453125e-06,
      "step": 525
    },
    {
      "action_loss_avg": 0.08765056282281876,
      "epoch": 2.002716064453125e-06,
      "step": 525
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 525,
      "training_step_time": 9.946190595626831
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 3.1405913829803467,
      "step": 525
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 525,
      "training_step_time": 9.91963505744934
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 3.136874198913574,
      "step": 525
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 525,
      "training_step_time": 9.909289121627808
    },
    {
      "epoch": 2.002716064453125e-06,
      "model_forward_time": 3.137171745300293,
      "step": 525
    },
    {
      "epoch": 2.002716064453125e-06,
      "step": 525,
      "training_step_time": 10.270150661468506
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 3.1763570308685303,
      "step": 526
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 526,
      "training_step_time": 9.949525594711304
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 3.138432025909424,
      "step": 526
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 526,
      "training_step_time": 9.91065239906311
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 3.140714645385742,
      "step": 526
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 526,
      "training_step_time": 9.912813663482666
    },
    {
      "epoch": 2.00653076171875e-06,
      "model_forward_time": 3.13995623588562,
      "step": 526
    },
    {
      "epoch": 2.00653076171875e-06,
      "step": 526,
      "training_step_time": 10.289896965026855
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 3.4744725227355957,
      "step": 527
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 527,
      "training_step_time": 10.248310089111328
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 3.1331632137298584,
      "step": 527
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 527,
      "training_step_time": 9.91144347190857
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 3.1325039863586426,
      "step": 527
    },
    {
      "dynamics_loss_avg": 0.08488352708518505,
      "epoch": 2.010345458984375e-06,
      "step": 527
    },
    {
      "action_loss_avg": 0.07713038697838784,
      "epoch": 2.010345458984375e-06,
      "step": 527
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 527,
      "training_step_time": 9.907802820205688
    },
    {
      "epoch": 2.010345458984375e-06,
      "model_forward_time": 3.13429594039917,
      "step": 527
    },
    {
      "epoch": 2.010345458984375e-06,
      "step": 527,
      "training_step_time": 10.017457246780396
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 3.1647346019744873,
      "step": 528
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 528,
      "training_step_time": 9.93604850769043
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 3.1283745765686035,
      "step": 528
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 528,
      "training_step_time": 10.169708013534546
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 3.1325743198394775,
      "step": 528
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 528,
      "training_step_time": 9.901252269744873
    },
    {
      "epoch": 2.01416015625e-06,
      "model_forward_time": 3.1409401893615723,
      "step": 528
    },
    {
      "epoch": 2.01416015625e-06,
      "step": 528,
      "training_step_time": 10.2935791015625
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 3.1625475883483887,
      "step": 529
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 529,
      "training_step_time": 9.933106422424316
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 3.1315250396728516,
      "step": 529
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 529,
      "training_step_time": 9.895339250564575
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 3.4006545543670654,
      "step": 529
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 529,
      "training_step_time": 10.170884132385254
    },
    {
      "epoch": 2.017974853515625e-06,
      "model_forward_time": 3.129751443862915,
      "step": 529
    },
    {
      "epoch": 2.017974853515625e-06,
      "step": 529,
      "training_step_time": 10.264951944351196
    },
    {
      "epoch": 2.02178955078125e-06,
      "grad_norm": 1.23357355594635,
      "learning_rate": 1e-05,
      "loss": 0.1671,
      "step": 530
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 3.1754415035247803,
      "step": 530
    },
    {
      "dynamics_loss_avg": 0.09894916713237763,
      "epoch": 2.02178955078125e-06,
      "step": 530
    },
    {
      "action_loss_avg": 0.08104245774447918,
      "epoch": 2.02178955078125e-06,
      "step": 530
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 530,
      "training_step_time": 9.949336290359497
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 2.4335734844207764,
      "step": 530
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 530,
      "training_step_time": 7.461797714233398
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 2.425929069519043,
      "step": 530
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 530,
      "training_step_time": 7.451894283294678
    },
    {
      "epoch": 2.02178955078125e-06,
      "model_forward_time": 3.127645492553711,
      "step": 530
    },
    {
      "epoch": 2.02178955078125e-06,
      "step": 530,
      "training_step_time": 10.023061037063599
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 3.4830482006073,
      "step": 531
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 531,
      "training_step_time": 10.26128888130188
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 3.13619327545166,
      "step": 531
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 531,
      "training_step_time": 9.912350177764893
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 3.133268117904663,
      "step": 531
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 531,
      "training_step_time": 9.912023067474365
    },
    {
      "epoch": 2.025604248046875e-06,
      "model_forward_time": 3.1367225646972656,
      "step": 531
    },
    {
      "epoch": 2.025604248046875e-06,
      "step": 531,
      "training_step_time": 10.027880907058716
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 3.434157609939575,
      "step": 532
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 532,
      "training_step_time": 10.210876226425171
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 3.1407079696655273,
      "step": 532
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 532,
      "training_step_time": 9.912449836730957
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 2.4326488971710205,
      "step": 532
    },
    {
      "dynamics_loss_avg": 0.07601286415010691,
      "epoch": 2.0294189453125e-06,
      "step": 532
    },
    {
      "action_loss_avg": 0.07701750742271543,
      "epoch": 2.0294189453125e-06,
      "step": 532
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 532,
      "training_step_time": 7.466103553771973
    },
    {
      "epoch": 2.0294189453125e-06,
      "model_forward_time": 3.1274361610412598,
      "step": 532
    },
    {
      "epoch": 2.0294189453125e-06,
      "step": 532,
      "training_step_time": 10.271567583084106
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 3.173863410949707,
      "step": 533
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 533,
      "training_step_time": 9.948342561721802
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 3.1337473392486572,
      "step": 533
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 533,
      "training_step_time": 9.901164770126343
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 3.372405529022217,
      "step": 533
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 533,
      "training_step_time": 10.150924682617188
    },
    {
      "epoch": 2.033233642578125e-06,
      "model_forward_time": 3.1346306800842285,
      "step": 533
    },
    {
      "epoch": 2.033233642578125e-06,
      "step": 533,
      "training_step_time": 10.252866983413696
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 3.181567430496216,
      "step": 534
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 534,
      "training_step_time": 9.951866865158081
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 3.127647876739502,
      "step": 534
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 534,
      "training_step_time": 9.895906925201416
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 3.129915714263916,
      "step": 534
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 534,
      "training_step_time": 10.206002950668335
    },
    {
      "epoch": 2.03704833984375e-06,
      "model_forward_time": 3.1307878494262695,
      "step": 534
    },
    {
      "epoch": 2.03704833984375e-06,
      "step": 534,
      "training_step_time": 10.260720491409302
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 3.175264835357666,
      "step": 535
    },
    {
      "dynamics_loss_avg": 0.08817163370549679,
      "epoch": 2.040863037109375e-06,
      "step": 535
    },
    {
      "action_loss_avg": 0.07717048935592175,
      "epoch": 2.040863037109375e-06,
      "step": 535
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 535,
      "training_step_time": 9.948430061340332
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 3.127915859222412,
      "step": 535
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 535,
      "training_step_time": 9.891839981079102
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 3.130336284637451,
      "step": 535
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 535,
      "training_step_time": 9.896674633026123
    },
    {
      "epoch": 2.040863037109375e-06,
      "model_forward_time": 2.430952310562134,
      "step": 535
    },
    {
      "epoch": 2.040863037109375e-06,
      "step": 535,
      "training_step_time": 10.017101049423218
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 3.1753056049346924,
      "step": 536
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 536,
      "training_step_time": 9.947946071624756
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 3.131309986114502,
      "step": 536
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 536,
      "training_step_time": 9.908939361572266
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 3.1352906227111816,
      "step": 536
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 536,
      "training_step_time": 9.91493010520935
    },
    {
      "epoch": 2.044677734375e-06,
      "model_forward_time": 3.133782148361206,
      "step": 536
    },
    {
      "epoch": 2.044677734375e-06,
      "step": 536,
      "training_step_time": 10.024009466171265
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 3.167745351791382,
      "step": 537
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 537,
      "training_step_time": 10.214229583740234
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 3.1382358074188232,
      "step": 537
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 537,
      "training_step_time": 9.910216093063354
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 3.1374847888946533,
      "step": 537
    },
    {
      "dynamics_loss_avg": 0.09830252379179001,
      "epoch": 2.048492431640625e-06,
      "step": 537
    },
    {
      "action_loss_avg": 0.06996215116232633,
      "epoch": 2.048492431640625e-06,
      "step": 537
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 537,
      "training_step_time": 9.912303924560547
    },
    {
      "epoch": 2.048492431640625e-06,
      "model_forward_time": 3.133403778076172,
      "step": 537
    },
    {
      "epoch": 2.048492431640625e-06,
      "step": 537,
      "training_step_time": 10.321097135543823
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 3.178245782852173,
      "step": 538
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 538,
      "training_step_time": 9.95264482498169
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 2.4302449226379395,
      "step": 538
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 538,
      "training_step_time": 7.755682945251465
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 3.1309971809387207,
      "step": 538
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 538,
      "training_step_time": 9.903174877166748
    },
    {
      "epoch": 2.05230712890625e-06,
      "model_forward_time": 3.133131980895996,
      "step": 538
    },
    {
      "epoch": 2.05230712890625e-06,
      "step": 538,
      "training_step_time": 10.244570970535278
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 3.1743738651275635,
      "step": 539
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 539,
      "training_step_time": 9.94349455833435
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 3.133094549179077,
      "step": 539
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 539,
      "training_step_time": 9.903252124786377
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 3.1315646171569824,
      "step": 539
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 539,
      "training_step_time": 10.180447578430176
    },
    {
      "epoch": 2.056121826171875e-06,
      "model_forward_time": 3.135673761367798,
      "step": 539
    },
    {
      "epoch": 2.056121826171875e-06,
      "step": 539,
      "training_step_time": 10.287717580795288
    },
    {
      "epoch": 2.0599365234375e-06,
      "grad_norm": 0.7045297026634216,
      "learning_rate": 1e-05,
      "loss": 0.1619,
      "step": 540
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 3.175269842147827,
      "step": 540
    },
    {
      "dynamics_loss_avg": 0.08691137265413999,
      "epoch": 2.0599365234375e-06,
      "step": 540
    },
    {
      "action_loss_avg": 0.07364887222647667,
      "epoch": 2.0599365234375e-06,
      "step": 540
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 540,
      "training_step_time": 9.957456111907959
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 3.1397902965545654,
      "step": 540
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 540,
      "training_step_time": 9.917764663696289
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 3.1373088359832764,
      "step": 540
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 540,
      "training_step_time": 9.910863399505615
    },
    {
      "epoch": 2.0599365234375e-06,
      "model_forward_time": 3.3926279544830322,
      "step": 540
    },
    {
      "epoch": 2.0599365234375e-06,
      "step": 540,
      "training_step_time": 10.259081840515137
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 3.175226926803589,
      "step": 541
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 541,
      "training_step_time": 9.94786024093628
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 3.1369118690490723,
      "step": 541
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 541,
      "training_step_time": 9.912731409072876
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 3.135246992111206,
      "step": 541
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 541,
      "training_step_time": 9.911344528198242
    },
    {
      "epoch": 2.063751220703125e-06,
      "model_forward_time": 3.140718460083008,
      "step": 541
    },
    {
      "epoch": 2.063751220703125e-06,
      "step": 541,
      "training_step_time": 10.257773637771606
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 3.4617199897766113,
      "step": 542
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 542,
      "training_step_time": 10.240543365478516
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 3.1377766132354736,
      "step": 542
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 542,
      "training_step_time": 9.914612293243408
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 3.1317174434661865,
      "step": 542
    },
    {
      "dynamics_loss_avg": 0.09184226989746094,
      "epoch": 2.06756591796875e-06,
      "step": 542
    },
    {
      "action_loss_avg": 0.07243521641939879,
      "epoch": 2.06756591796875e-06,
      "step": 542
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 542,
      "training_step_time": 9.909644842147827
    },
    {
      "epoch": 2.06756591796875e-06,
      "model_forward_time": 3.13220477104187,
      "step": 542
    },
    {
      "epoch": 2.06756591796875e-06,
      "step": 542,
      "training_step_time": 10.240875720977783
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 3.1738765239715576,
      "step": 543
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 543,
      "training_step_time": 9.94474983215332
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 3.1375505924224854,
      "step": 543
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 543,
      "training_step_time": 10.196585416793823
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 3.139253854751587,
      "step": 543
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 543,
      "training_step_time": 9.910213470458984
    },
    {
      "epoch": 2.071380615234375e-06,
      "model_forward_time": 3.1536781787872314,
      "step": 543
    },
    {
      "epoch": 2.071380615234375e-06,
      "step": 543,
      "training_step_time": 10.307440757751465
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 2.477853298187256,
      "step": 544
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 544,
      "training_step_time": 7.511154890060425
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 3.1395928859710693,
      "step": 544
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 544,
      "training_step_time": 9.922537326812744
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 2.4368438720703125,
      "step": 544
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 544,
      "training_step_time": 7.467644929885864
    },
    {
      "epoch": 2.0751953125e-06,
      "model_forward_time": 3.3977909088134766,
      "step": 544
    },
    {
      "epoch": 2.0751953125e-06,
      "step": 544,
      "training_step_time": 10.265024662017822
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 2.460357904434204,
      "step": 545
    },
    {
      "dynamics_loss_avg": 0.10421213731169701,
      "epoch": 2.079010009765625e-06,
      "step": 545
    },
    {
      "action_loss_avg": 0.08184996042400598,
      "epoch": 2.079010009765625e-06,
      "step": 545
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 545,
      "training_step_time": 7.495085000991821
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 3.127741575241089,
      "step": 545
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 545,
      "training_step_time": 9.902377843856812
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 3.1351494789123535,
      "step": 545
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 545,
      "training_step_time": 9.912019968032837
    },
    {
      "epoch": 2.079010009765625e-06,
      "model_forward_time": 3.132859468460083,
      "step": 545
    },
    {
      "epoch": 2.079010009765625e-06,
      "step": 545,
      "training_step_time": 10.025655031204224
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 3.5039491653442383,
      "step": 546
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 546,
      "training_step_time": 10.280224800109863
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 3.135857105255127,
      "step": 546
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 546,
      "training_step_time": 9.90934443473816
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 3.136216640472412,
      "step": 546
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 546,
      "training_step_time": 9.910651922225952
    },
    {
      "epoch": 2.08282470703125e-06,
      "model_forward_time": 3.139399766921997,
      "step": 546
    },
    {
      "epoch": 2.08282470703125e-06,
      "step": 546,
      "training_step_time": 10.321135520935059
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 3.18188738822937,
      "step": 547
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 547,
      "training_step_time": 10.222678661346436
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 2.4327924251556396,
      "step": 547
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 547,
      "training_step_time": 7.464716196060181
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 3.1334176063537598,
      "step": 547
    },
    {
      "dynamics_loss_avg": 0.09173010364174843,
      "epoch": 2.086639404296875e-06,
      "step": 547
    },
    {
      "action_loss_avg": 0.0672373566776514,
      "epoch": 2.086639404296875e-06,
      "step": 547
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 547,
      "training_step_time": 9.911072969436646
    },
    {
      "epoch": 2.086639404296875e-06,
      "model_forward_time": 3.133072853088379,
      "step": 547
    },
    {
      "epoch": 2.086639404296875e-06,
      "step": 547,
      "training_step_time": 10.028021335601807
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 3.18641996383667,
      "step": 548
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 548,
      "training_step_time": 9.972214937210083
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 3.1353461742401123,
      "step": 548
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 548,
      "training_step_time": 10.177174806594849
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 3.134498357772827,
      "step": 548
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 548,
      "training_step_time": 9.912100315093994
    },
    {
      "epoch": 2.0904541015625e-06,
      "model_forward_time": 3.133775472640991,
      "step": 548
    },
    {
      "epoch": 2.0904541015625e-06,
      "step": 548,
      "training_step_time": 10.293594121932983
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 3.171288251876831,
      "step": 549
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 549,
      "training_step_time": 9.94144058227539
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 3.1333491802215576,
      "step": 549
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 549,
      "training_step_time": 9.900716781616211
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 3.407376289367676,
      "step": 549
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 549,
      "training_step_time": 10.175313949584961
    },
    {
      "epoch": 2.094268798828125e-06,
      "model_forward_time": 3.131113052368164,
      "step": 549
    },
    {
      "epoch": 2.094268798828125e-06,
      "step": 549,
      "training_step_time": 10.28222918510437
    },
    {
      "epoch": 2.09808349609375e-06,
      "grad_norm": 0.7191452980041504,
      "learning_rate": 1e-05,
      "loss": 0.1595,
      "step": 550
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 3.154175281524658,
      "step": 550
    },
    {
      "dynamics_loss_avg": 0.11039590835571289,
      "epoch": 2.09808349609375e-06,
      "step": 550
    },
    {
      "action_loss_avg": 0.08169630616903305,
      "epoch": 2.09808349609375e-06,
      "step": 550
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 550,
      "training_step_time": 9.926798343658447
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 3.1334047317504883,
      "step": 550
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 550,
      "training_step_time": 9.904929876327515
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 3.1355490684509277,
      "step": 550
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 550,
      "training_step_time": 9.910823822021484
    },
    {
      "epoch": 2.09808349609375e-06,
      "model_forward_time": 3.4300496578216553,
      "step": 550
    },
    {
      "epoch": 2.09808349609375e-06,
      "step": 550,
      "training_step_time": 10.295178174972534
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 3.1892411708831787,
      "step": 551
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 551,
      "training_step_time": 9.963569164276123
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 3.136085033416748,
      "step": 551
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 551,
      "training_step_time": 9.907334804534912
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 3.132233142852783,
      "step": 551
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 551,
      "training_step_time": 9.90577507019043
    },
    {
      "epoch": 2.101898193359375e-06,
      "model_forward_time": 3.1369359493255615,
      "step": 551
    },
    {
      "epoch": 2.101898193359375e-06,
      "step": 551,
      "training_step_time": 10.259902715682983
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 3.4118239879608154,
      "step": 552
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 552,
      "training_step_time": 10.186304092407227
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 3.1373236179351807,
      "step": 552
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 552,
      "training_step_time": 9.911864519119263
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 3.13474702835083,
      "step": 552
    },
    {
      "dynamics_loss_avg": 0.083335505053401,
      "epoch": 2.105712890625e-06,
      "step": 552
    },
    {
      "action_loss_avg": 0.0773266321979463,
      "epoch": 2.105712890625e-06,
      "step": 552
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 552,
      "training_step_time": 9.915550470352173
    },
    {
      "epoch": 2.105712890625e-06,
      "model_forward_time": 3.1351661682128906,
      "step": 552
    },
    {
      "epoch": 2.105712890625e-06,
      "step": 552,
      "training_step_time": 10.003512859344482
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 3.1643214225769043,
      "step": 553
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 553,
      "training_step_time": 9.935112237930298
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 3.410426378250122,
      "step": 553
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 553,
      "training_step_time": 10.179233074188232
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 2.4349138736724854,
      "step": 553
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 553,
      "training_step_time": 7.461434841156006
    },
    {
      "epoch": 2.109527587890625e-06,
      "model_forward_time": 3.1343350410461426,
      "step": 553
    },
    {
      "epoch": 2.109527587890625e-06,
      "step": 553,
      "training_step_time": 10.291446685791016
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 3.1528480052948,
      "step": 554
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 554,
      "training_step_time": 9.96894645690918
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 3.1334891319274902,
      "step": 554
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 554,
      "training_step_time": 9.905916690826416
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 2.4288744926452637,
      "step": 554
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 554,
      "training_step_time": 7.4641008377075195
    },
    {
      "epoch": 2.11334228515625e-06,
      "model_forward_time": 3.4361021518707275,
      "step": 554
    },
    {
      "epoch": 2.11334228515625e-06,
      "step": 554,
      "training_step_time": 10.30276894569397
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 3.189134120941162,
      "step": 555
    },
    {
      "dynamics_loss_avg": 0.1007946215569973,
      "epoch": 2.117156982421875e-06,
      "step": 555
    },
    {
      "action_loss_avg": 0.07473165690898895,
      "epoch": 2.117156982421875e-06,
      "step": 555
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 555,
      "training_step_time": 9.967215061187744
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 3.135850191116333,
      "step": 555
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 555,
      "training_step_time": 9.91726040840149
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 3.1393160820007324,
      "step": 555
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 555,
      "training_step_time": 9.917831420898438
    },
    {
      "epoch": 2.117156982421875e-06,
      "model_forward_time": 3.4267756938934326,
      "step": 555
    },
    {
      "epoch": 2.117156982421875e-06,
      "step": 555,
      "training_step_time": 10.29566502571106
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 3.1733386516571045,
      "step": 556
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 556,
      "training_step_time": 9.95237135887146
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 3.1369452476501465,
      "step": 556
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 556,
      "training_step_time": 9.913406372070312
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 3.1410815715789795,
      "step": 556
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 556,
      "training_step_time": 9.923170328140259
    },
    {
      "epoch": 2.1209716796875e-06,
      "model_forward_time": 3.1404170989990234,
      "step": 556
    },
    {
      "epoch": 2.1209716796875e-06,
      "step": 556,
      "training_step_time": 10.036619424819946
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 3.1836462020874023,
      "step": 557
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 557,
      "training_step_time": 10.231644630432129
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 3.1398980617523193,
      "step": 557
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 557,
      "training_step_time": 9.929435729980469
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 3.138857364654541,
      "step": 557
    },
    {
      "dynamics_loss_avg": 0.10777874886989594,
      "epoch": 2.124786376953125e-06,
      "step": 557
    },
    {
      "action_loss_avg": 0.0970717690885067,
      "epoch": 2.124786376953125e-06,
      "step": 557
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 557,
      "training_step_time": 9.922393560409546
    },
    {
      "epoch": 2.124786376953125e-06,
      "model_forward_time": 3.1391053199768066,
      "step": 557
    },
    {
      "epoch": 2.124786376953125e-06,
      "step": 557,
      "training_step_time": 10.026209831237793
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 3.166745901107788,
      "step": 558
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 558,
      "training_step_time": 9.950861930847168
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 3.3914990425109863,
      "step": 558
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 558,
      "training_step_time": 10.158209323883057
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 3.1322720050811768,
      "step": 558
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 558,
      "training_step_time": 9.899874925613403
    },
    {
      "epoch": 2.12860107421875e-06,
      "model_forward_time": 3.131561040878296,
      "step": 558
    },
    {
      "epoch": 2.12860107421875e-06,
      "step": 558,
      "training_step_time": 10.256146430969238
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 3.1728265285491943,
      "step": 559
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 559,
      "training_step_time": 9.938679695129395
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 3.1341135501861572,
      "step": 559
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 559,
      "training_step_time": 9.899569272994995
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 3.461980104446411,
      "step": 559
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 559,
      "training_step_time": 10.23885440826416
    },
    {
      "epoch": 2.132415771484375e-06,
      "model_forward_time": 3.138620376586914,
      "step": 559
    },
    {
      "epoch": 2.132415771484375e-06,
      "step": 559,
      "training_step_time": 10.335445880889893
    },
    {
      "epoch": 2.13623046875e-06,
      "grad_norm": 0.6626716256141663,
      "learning_rate": 1e-05,
      "loss": 0.1589,
      "step": 560
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 3.177100419998169,
      "step": 560
    },
    {
      "dynamics_loss_avg": 0.08228504732251167,
      "epoch": 2.13623046875e-06,
      "step": 560
    },
    {
      "action_loss_avg": 0.06389566510915756,
      "epoch": 2.13623046875e-06,
      "step": 560
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 560,
      "training_step_time": 9.953436613082886
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 3.1393938064575195,
      "step": 560
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 560,
      "training_step_time": 9.915507316589355
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 3.1390016078948975,
      "step": 560
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 560,
      "training_step_time": 9.914691686630249
    },
    {
      "epoch": 2.13623046875e-06,
      "model_forward_time": 3.4065439701080322,
      "step": 560
    },
    {
      "epoch": 2.13623046875e-06,
      "step": 560,
      "training_step_time": 10.282547950744629
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 3.161391019821167,
      "step": 561
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 561,
      "training_step_time": 9.931047439575195
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 3.1388351917266846,
      "step": 561
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 561,
      "training_step_time": 9.913353204727173
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 3.1358144283294678,
      "step": 561
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 561,
      "training_step_time": 9.9076509475708
    },
    {
      "epoch": 2.140045166015625e-06,
      "model_forward_time": 3.138700008392334,
      "step": 561
    },
    {
      "epoch": 2.140045166015625e-06,
      "step": 561,
      "training_step_time": 10.032973527908325
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 3.4671013355255127,
      "step": 562
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 562,
      "training_step_time": 10.239513158798218
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 3.13240122795105,
      "step": 562
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 562,
      "training_step_time": 9.900482654571533
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 3.133824348449707,
      "step": 562
    },
    {
      "dynamics_loss_avg": 0.09773895852267742,
      "epoch": 2.14385986328125e-06,
      "step": 562
    },
    {
      "action_loss_avg": 0.06714365240186453,
      "epoch": 2.14385986328125e-06,
      "step": 562
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 562,
      "training_step_time": 9.901415586471558
    },
    {
      "epoch": 2.14385986328125e-06,
      "model_forward_time": 3.132368564605713,
      "step": 562
    },
    {
      "epoch": 2.14385986328125e-06,
      "step": 562,
      "training_step_time": 10.01805305480957
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 2.4630987644195557,
      "step": 563
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 563,
      "training_step_time": 7.484553575515747
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 3.129796266555786,
      "step": 563
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 563,
      "training_step_time": 10.166083335876465
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 3.132586717605591,
      "step": 563
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 563,
      "training_step_time": 9.898017406463623
    },
    {
      "epoch": 2.147674560546875e-06,
      "model_forward_time": 3.133244514465332,
      "step": 563
    },
    {
      "epoch": 2.147674560546875e-06,
      "step": 563,
      "training_step_time": 10.270336627960205
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 3.164569139480591,
      "step": 564
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 564,
      "training_step_time": 9.930235624313354
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 3.1293134689331055,
      "step": 564
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 564,
      "training_step_time": 9.893892288208008
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 3.388462781906128,
      "step": 564
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 564,
      "training_step_time": 10.151278257369995
    },
    {
      "epoch": 2.1514892578125e-06,
      "model_forward_time": 3.1324825286865234,
      "step": 564
    },
    {
      "epoch": 2.1514892578125e-06,
      "step": 564,
      "training_step_time": 10.282521486282349
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 3.1765944957733154,
      "step": 565
    },
    {
      "dynamics_loss_avg": 0.09414612650871276,
      "epoch": 2.155303955078125e-06,
      "step": 565
    },
    {
      "action_loss_avg": 0.08211876526474952,
      "epoch": 2.155303955078125e-06,
      "step": 565
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 565,
      "training_step_time": 9.948136806488037
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 3.1331512928009033,
      "step": 565
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 565,
      "training_step_time": 9.904855489730835
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 3.138326644897461,
      "step": 565
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 565,
      "training_step_time": 9.912678003311157
    },
    {
      "epoch": 2.155303955078125e-06,
      "model_forward_time": 3.140385627746582,
      "step": 565
    },
    {
      "epoch": 2.155303955078125e-06,
      "step": 565,
      "training_step_time": 10.313578605651855
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 3.1966586112976074,
      "step": 566
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 566,
      "training_step_time": 9.971415042877197
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 3.134549617767334,
      "step": 566
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 566,
      "training_step_time": 9.907336711883545
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 3.137984275817871,
      "step": 566
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 566,
      "training_step_time": 9.912841558456421
    },
    {
      "epoch": 2.15911865234375e-06,
      "model_forward_time": 3.140373706817627,
      "step": 566
    },
    {
      "epoch": 2.15911865234375e-06,
      "step": 566,
      "training_step_time": 10.058285474777222
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 3.463125228881836,
      "step": 567
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 567,
      "training_step_time": 10.23223328590393
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 3.1386520862579346,
      "step": 567
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 567,
      "training_step_time": 9.916156768798828
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 3.1315369606018066,
      "step": 567
    },
    {
      "dynamics_loss_avg": 0.08857547119259834,
      "epoch": 2.162933349609375e-06,
      "step": 567
    },
    {
      "action_loss_avg": 0.07488007210195065,
      "epoch": 2.162933349609375e-06,
      "step": 567
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 567,
      "training_step_time": 9.905307054519653
    },
    {
      "epoch": 2.162933349609375e-06,
      "model_forward_time": 3.1379637718200684,
      "step": 567
    },
    {
      "epoch": 2.162933349609375e-06,
      "step": 567,
      "training_step_time": 10.017179250717163
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 3.166329860687256,
      "step": 568
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 568,
      "training_step_time": 9.942598342895508
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 3.1373908519744873,
      "step": 568
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 568,
      "training_step_time": 10.18377161026001
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 3.1370818614959717,
      "step": 568
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 568,
      "training_step_time": 9.909293413162231
    },
    {
      "epoch": 2.166748046875e-06,
      "model_forward_time": 3.1333913803100586,
      "step": 568
    },
    {
      "epoch": 2.166748046875e-06,
      "step": 568,
      "training_step_time": 10.008257865905762
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 3.1763508319854736,
      "step": 569
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 569,
      "training_step_time": 9.944222927093506
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 3.131964921951294,
      "step": 569
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 569,
      "training_step_time": 9.903999328613281
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 3.4580914974212646,
      "step": 569
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 569,
      "training_step_time": 10.232001781463623
    },
    {
      "epoch": 2.170562744140625e-06,
      "model_forward_time": 3.132925271987915,
      "step": 569
    },
    {
      "epoch": 2.170562744140625e-06,
      "step": 569,
      "training_step_time": 10.293091058731079
    },
    {
      "epoch": 2.17437744140625e-06,
      "grad_norm": 0.7523087859153748,
      "learning_rate": 1e-05,
      "loss": 0.1543,
      "step": 570
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 3.171048164367676,
      "step": 570
    },
    {
      "dynamics_loss_avg": 0.07500165812671185,
      "epoch": 2.17437744140625e-06,
      "step": 570
    },
    {
      "action_loss_avg": 0.05307034645229578,
      "epoch": 2.17437744140625e-06,
      "step": 570
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 570,
      "training_step_time": 9.945316553115845
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 3.138303756713867,
      "step": 570
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 570,
      "training_step_time": 9.907611608505249
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 3.1363437175750732,
      "step": 570
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 570,
      "training_step_time": 9.90699052810669
    },
    {
      "epoch": 2.17437744140625e-06,
      "model_forward_time": 3.39631986618042,
      "step": 570
    },
    {
      "epoch": 2.17437744140625e-06,
      "step": 570,
      "training_step_time": 10.338277578353882
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 3.192854642868042,
      "step": 571
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 571,
      "training_step_time": 9.966151714324951
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 3.135344982147217,
      "step": 571
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 571,
      "training_step_time": 9.915390491485596
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 3.132011651992798,
      "step": 571
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 571,
      "training_step_time": 9.901939630508423
    },
    {
      "epoch": 2.178192138671875e-06,
      "model_forward_time": 3.13725209236145,
      "step": 571
    },
    {
      "epoch": 2.178192138671875e-06,
      "step": 571,
      "training_step_time": 10.020308494567871
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 3.4685537815093994,
      "step": 572
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 572,
      "training_step_time": 10.241485595703125
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 3.1413259506225586,
      "step": 572
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 572,
      "training_step_time": 9.91691255569458
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 3.133837938308716,
      "step": 572
    },
    {
      "dynamics_loss_avg": 0.09992915838956833,
      "epoch": 2.1820068359375e-06,
      "step": 572
    },
    {
      "action_loss_avg": 0.08588457927107811,
      "epoch": 2.1820068359375e-06,
      "step": 572
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 572,
      "training_step_time": 9.904110193252563
    },
    {
      "epoch": 2.1820068359375e-06,
      "model_forward_time": 3.1355249881744385,
      "step": 572
    },
    {
      "epoch": 2.1820068359375e-06,
      "step": 572,
      "training_step_time": 10.011266946792603
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 3.1865007877349854,
      "step": 573
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 573,
      "training_step_time": 9.966626167297363
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 3.433506488800049,
      "step": 573
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 573,
      "training_step_time": 10.20033597946167
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 3.138798236846924,
      "step": 573
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 573,
      "training_step_time": 9.90455675125122
    },
    {
      "epoch": 2.185821533203125e-06,
      "model_forward_time": 3.1404144763946533,
      "step": 573
    },
    {
      "epoch": 2.185821533203125e-06,
      "step": 573,
      "training_step_time": 10.325499534606934
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 3.1665477752685547,
      "step": 574
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 574,
      "training_step_time": 9.93105411529541
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 3.1386444568634033,
      "step": 574
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 574,
      "training_step_time": 9.911071300506592
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 3.141566276550293,
      "step": 574
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 574,
      "training_step_time": 10.188841819763184
    },
    {
      "epoch": 2.18963623046875e-06,
      "model_forward_time": 3.1416406631469727,
      "step": 574
    },
    {
      "epoch": 2.18963623046875e-06,
      "step": 574,
      "training_step_time": 10.284044742584229
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 3.1754696369171143,
      "step": 575
    },
    {
      "dynamics_loss_avg": 0.08818879127502441,
      "epoch": 2.193450927734375e-06,
      "step": 575
    },
    {
      "action_loss_avg": 0.07269831690937281,
      "epoch": 2.193450927734375e-06,
      "step": 575
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 575,
      "training_step_time": 9.9488205909729
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 3.1400675773620605,
      "step": 575
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 575,
      "training_step_time": 9.91321873664856
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 3.140777826309204,
      "step": 575
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 575,
      "training_step_time": 9.909080982208252
    },
    {
      "epoch": 2.193450927734375e-06,
      "model_forward_time": 3.454969644546509,
      "step": 575
    },
    {
      "epoch": 2.193450927734375e-06,
      "step": 575,
      "training_step_time": 10.324652433395386
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 3.175668716430664,
      "step": 576
    },
    {
      "epoch": 2.197265625e-06,
      "step": 576,
      "training_step_time": 9.946362018585205
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 3.140375852584839,
      "step": 576
    },
    {
      "epoch": 2.197265625e-06,
      "step": 576,
      "training_step_time": 9.911837100982666
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 3.1450936794281006,
      "step": 576
    },
    {
      "epoch": 2.197265625e-06,
      "step": 576,
      "training_step_time": 9.919997692108154
    },
    {
      "epoch": 2.197265625e-06,
      "model_forward_time": 3.1384568214416504,
      "step": 576
    },
    {
      "epoch": 2.197265625e-06,
      "step": 576,
      "training_step_time": 10.288431406021118
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 3.159400701522827,
      "step": 577
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 577,
      "training_step_time": 10.208552360534668
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 3.1347177028656006,
      "step": 577
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 577,
      "training_step_time": 9.903629541397095
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 3.1368305683135986,
      "step": 577
    },
    {
      "dynamics_loss_avg": 0.08653322868049144,
      "epoch": 2.201080322265625e-06,
      "step": 577
    },
    {
      "action_loss_avg": 0.06874157898128033,
      "epoch": 2.201080322265625e-06,
      "step": 577
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 577,
      "training_step_time": 9.906470775604248
    },
    {
      "epoch": 2.201080322265625e-06,
      "model_forward_time": 3.1338586807250977,
      "step": 577
    },
    {
      "epoch": 2.201080322265625e-06,
      "step": 577,
      "training_step_time": 10.014216661453247
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 3.1706886291503906,
      "step": 578
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 578,
      "training_step_time": 9.935467004776001
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 3.385328769683838,
      "step": 578
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 578,
      "training_step_time": 10.151789665222168
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 3.1325528621673584,
      "step": 578
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 578,
      "training_step_time": 9.899125099182129
    },
    {
      "epoch": 2.20489501953125e-06,
      "model_forward_time": 3.1296942234039307,
      "step": 578
    },
    {
      "epoch": 2.20489501953125e-06,
      "step": 578,
      "training_step_time": 10.30918288230896
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 3.1757781505584717,
      "step": 579
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 579,
      "training_step_time": 9.943056106567383
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 3.13274884223938,
      "step": 579
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 579,
      "training_step_time": 9.898782730102539
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 3.410050392150879,
      "step": 579
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 579,
      "training_step_time": 10.177988052368164
    },
    {
      "epoch": 2.208709716796875e-06,
      "model_forward_time": 3.1347460746765137,
      "step": 579
    },
    {
      "epoch": 2.208709716796875e-06,
      "step": 579,
      "training_step_time": 10.298993587493896
    },
    {
      "epoch": 2.2125244140625e-06,
      "grad_norm": 0.6072705984115601,
      "learning_rate": 1e-05,
      "loss": 0.1528,
      "step": 580
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 3.165786027908325,
      "step": 580
    },
    {
      "dynamics_loss_avg": 0.07747468128800392,
      "epoch": 2.2125244140625e-06,
      "step": 580
    },
    {
      "action_loss_avg": 0.05793101638555527,
      "epoch": 2.2125244140625e-06,
      "step": 580
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 580,
      "training_step_time": 9.935718059539795
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 3.1346030235290527,
      "step": 580
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 580,
      "training_step_time": 9.901073932647705
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 3.1280360221862793,
      "step": 580
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 580,
      "training_step_time": 9.899857521057129
    },
    {
      "epoch": 2.2125244140625e-06,
      "model_forward_time": 3.421225070953369,
      "step": 580
    },
    {
      "epoch": 2.2125244140625e-06,
      "step": 580,
      "training_step_time": 10.323864698410034
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 3.1817171573638916,
      "step": 581
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 581,
      "training_step_time": 9.946879148483276
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 3.129345178604126,
      "step": 581
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 581,
      "training_step_time": 9.892136096954346
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 3.131972312927246,
      "step": 581
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 581,
      "training_step_time": 9.898892879486084
    },
    {
      "epoch": 2.216339111328125e-06,
      "model_forward_time": 3.13289737701416,
      "step": 581
    },
    {
      "epoch": 2.216339111328125e-06,
      "step": 581,
      "training_step_time": 10.301435947418213
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 3.460792303085327,
      "step": 582
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 582,
      "training_step_time": 10.234900951385498
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 3.132221221923828,
      "step": 582
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 582,
      "training_step_time": 9.905859470367432
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 3.127108573913574,
      "step": 582
    },
    {
      "dynamics_loss_avg": 0.09903111048042774,
      "epoch": 2.22015380859375e-06,
      "step": 582
    },
    {
      "action_loss_avg": 0.06937348712235689,
      "epoch": 2.22015380859375e-06,
      "step": 582
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 582,
      "training_step_time": 9.899291753768921
    },
    {
      "epoch": 2.22015380859375e-06,
      "model_forward_time": 3.130401611328125,
      "step": 582
    },
    {
      "epoch": 2.22015380859375e-06,
      "step": 582,
      "training_step_time": 10.271772384643555
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 3.160412549972534,
      "step": 583
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 583,
      "training_step_time": 9.928402662277222
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 3.123669385910034,
      "step": 583
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 583,
      "training_step_time": 10.167855739593506
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 3.130657434463501,
      "step": 583
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 583,
      "training_step_time": 9.895663976669312
    },
    {
      "epoch": 2.223968505859375e-06,
      "model_forward_time": 3.129497528076172,
      "step": 583
    },
    {
      "epoch": 2.223968505859375e-06,
      "step": 583,
      "training_step_time": 10.214635133743286
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 3.1639904975891113,
      "step": 584
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 584,
      "training_step_time": 9.930697917938232
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 3.1305055618286133,
      "step": 584
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 584,
      "training_step_time": 9.897985458374023
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 3.1275694370269775,
      "step": 584
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 584,
      "training_step_time": 10.19521951675415
    },
    {
      "epoch": 2.227783203125e-06,
      "model_forward_time": 3.1312150955200195,
      "step": 584
    },
    {
      "epoch": 2.227783203125e-06,
      "step": 584,
      "training_step_time": 10.30121397972107
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 3.179147243499756,
      "step": 585
    },
    {
      "dynamics_loss_avg": 0.09006941355764866,
      "epoch": 2.231597900390625e-06,
      "step": 585
    },
    {
      "action_loss_avg": 0.06089385412633419,
      "epoch": 2.231597900390625e-06,
      "step": 585
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 585,
      "training_step_time": 9.956473588943481
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 3.141235113143921,
      "step": 585
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 585,
      "training_step_time": 9.905973196029663
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 3.1402862071990967,
      "step": 585
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 585,
      "training_step_time": 9.911727666854858
    },
    {
      "epoch": 2.231597900390625e-06,
      "model_forward_time": 3.141514301300049,
      "step": 585
    },
    {
      "epoch": 2.231597900390625e-06,
      "step": 585,
      "training_step_time": 10.28547477722168
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 3.1853373050689697,
      "step": 586
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 586,
      "training_step_time": 9.957228422164917
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 3.1503450870513916,
      "step": 586
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 586,
      "training_step_time": 9.925738096237183
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 3.1443965435028076,
      "step": 586
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 586,
      "training_step_time": 9.916768074035645
    },
    {
      "epoch": 2.23541259765625e-06,
      "model_forward_time": 3.1427619457244873,
      "step": 586
    },
    {
      "epoch": 2.23541259765625e-06,
      "step": 586,
      "training_step_time": 10.297551155090332
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 3.453091859817505,
      "step": 587
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 587,
      "training_step_time": 10.22664213180542
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 3.145174264907837,
      "step": 587
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 587,
      "training_step_time": 9.916943073272705
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 3.144507884979248,
      "step": 587
    },
    {
      "dynamics_loss_avg": 0.09163373373448849,
      "epoch": 2.239227294921875e-06,
      "step": 587
    },
    {
      "action_loss_avg": 0.07725387699902057,
      "epoch": 2.239227294921875e-06,
      "step": 587
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 587,
      "training_step_time": 9.924669742584229
    },
    {
      "epoch": 2.239227294921875e-06,
      "model_forward_time": 3.1413700580596924,
      "step": 587
    },
    {
      "epoch": 2.239227294921875e-06,
      "step": 587,
      "training_step_time": 10.264020204544067
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 3.173060178756714,
      "step": 588
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 588,
      "training_step_time": 9.945666074752808
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 3.148348569869995,
      "step": 588
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 588,
      "training_step_time": 10.19745683670044
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 3.1426613330841064,
      "step": 588
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 588,
      "training_step_time": 9.910798788070679
    },
    {
      "epoch": 2.2430419921875e-06,
      "model_forward_time": 3.1408841609954834,
      "step": 588
    },
    {
      "epoch": 2.2430419921875e-06,
      "step": 588,
      "training_step_time": 10.008951663970947
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 3.1796505451202393,
      "step": 589
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 589,
      "training_step_time": 9.95376205444336
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 3.143723964691162,
      "step": 589
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 589,
      "training_step_time": 9.909720420837402
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 3.4500222206115723,
      "step": 589
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 589,
      "training_step_time": 10.21897006034851
    },
    {
      "epoch": 2.246856689453125e-06,
      "model_forward_time": 3.138080596923828,
      "step": 589
    },
    {
      "epoch": 2.246856689453125e-06,
      "step": 589,
      "training_step_time": 10.279719591140747
    },
    {
      "epoch": 2.25067138671875e-06,
      "grad_norm": 1.0145312547683716,
      "learning_rate": 1e-05,
      "loss": 0.1549,
      "step": 590
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 3.1565723419189453,
      "step": 590
    },
    {
      "dynamics_loss_avg": 0.08728164732456208,
      "epoch": 2.25067138671875e-06,
      "step": 590
    },
    {
      "action_loss_avg": 0.07715733945369721,
      "epoch": 2.25067138671875e-06,
      "step": 590
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 590,
      "training_step_time": 9.926939249038696
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 3.133941411972046,
      "step": 590
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 590,
      "training_step_time": 9.907423257827759
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 3.1354215145111084,
      "step": 590
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 590,
      "training_step_time": 9.909185886383057
    },
    {
      "epoch": 2.25067138671875e-06,
      "model_forward_time": 3.386949062347412,
      "step": 590
    },
    {
      "epoch": 2.25067138671875e-06,
      "step": 590,
      "training_step_time": 10.259461641311646
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 3.1638200283050537,
      "step": 591
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 591,
      "training_step_time": 9.930709838867188
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 3.1263515949249268,
      "step": 591
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 591,
      "training_step_time": 9.887673377990723
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 3.1264891624450684,
      "step": 591
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 591,
      "training_step_time": 9.886552572250366
    },
    {
      "epoch": 2.254486083984375e-06,
      "model_forward_time": 3.126727819442749,
      "step": 591
    },
    {
      "epoch": 2.254486083984375e-06,
      "step": 591,
      "training_step_time": 10.304524183273315
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 3.4292354583740234,
      "step": 592
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 592,
      "training_step_time": 10.190959453582764
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 3.1318247318267822,
      "step": 592
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 592,
      "training_step_time": 9.898240566253662
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 3.1467270851135254,
      "step": 592
    },
    {
      "dynamics_loss_avg": 0.08535699322819709,
      "epoch": 2.25830078125e-06,
      "step": 592
    },
    {
      "action_loss_avg": 0.0755317635834217,
      "epoch": 2.25830078125e-06,
      "step": 592
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 592,
      "training_step_time": 9.91732907295227
    },
    {
      "epoch": 2.25830078125e-06,
      "model_forward_time": 3.1359050273895264,
      "step": 592
    },
    {
      "epoch": 2.25830078125e-06,
      "step": 592,
      "training_step_time": 10.284648180007935
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 3.1762070655822754,
      "step": 593
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 593,
      "training_step_time": 9.950049877166748
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 3.451510190963745,
      "step": 593
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 593,
      "training_step_time": 10.223926544189453
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 3.1342220306396484,
      "step": 593
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 593,
      "training_step_time": 9.901692390441895
    },
    {
      "epoch": 2.262115478515625e-06,
      "model_forward_time": 3.134221076965332,
      "step": 593
    },
    {
      "epoch": 2.262115478515625e-06,
      "step": 593,
      "training_step_time": 10.007288932800293
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 3.1701271533966064,
      "step": 594
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 594,
      "training_step_time": 9.936752080917358
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 3.132481575012207,
      "step": 594
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 594,
      "training_step_time": 9.899421691894531
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 3.1304709911346436,
      "step": 594
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 594,
      "training_step_time": 10.16920256614685
    },
    {
      "epoch": 2.26593017578125e-06,
      "model_forward_time": 3.1344165802001953,
      "step": 594
    },
    {
      "epoch": 2.26593017578125e-06,
      "step": 594,
      "training_step_time": 10.269435167312622
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 3.1477620601654053,
      "step": 595
    },
    {
      "dynamics_loss_avg": 0.08963300660252571,
      "epoch": 2.269744873046875e-06,
      "step": 595
    },
    {
      "action_loss_avg": 0.0793032132089138,
      "epoch": 2.269744873046875e-06,
      "step": 595
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 595,
      "training_step_time": 9.921852827072144
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 3.129115581512451,
      "step": 595
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 595,
      "training_step_time": 9.896308660507202
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 3.133936882019043,
      "step": 595
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 595,
      "training_step_time": 9.897615194320679
    },
    {
      "epoch": 2.269744873046875e-06,
      "model_forward_time": 2.4317986965179443,
      "step": 595
    },
    {
      "epoch": 2.269744873046875e-06,
      "step": 595,
      "training_step_time": 10.249863386154175
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 3.181281805038452,
      "step": 596
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 596,
      "training_step_time": 9.952517032623291
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 3.133246898651123,
      "step": 596
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 596,
      "training_step_time": 9.905572891235352
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 3.1351282596588135,
      "step": 596
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 596,
      "training_step_time": 9.904579639434814
    },
    {
      "epoch": 2.2735595703125e-06,
      "model_forward_time": 3.131796360015869,
      "step": 596
    },
    {
      "epoch": 2.2735595703125e-06,
      "step": 596,
      "training_step_time": 10.297806024551392
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 3.1614177227020264,
      "step": 597
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 597,
      "training_step_time": 10.196489810943604
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 3.131505250930786,
      "step": 597
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 597,
      "training_step_time": 9.904039144515991
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 3.1339471340179443,
      "step": 597
    },
    {
      "dynamics_loss_avg": 0.07797238118946552,
      "epoch": 2.277374267578125e-06,
      "step": 597
    },
    {
      "action_loss_avg": 0.06726536955684423,
      "epoch": 2.277374267578125e-06,
      "step": 597
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 597,
      "training_step_time": 9.909509658813477
    },
    {
      "epoch": 2.277374267578125e-06,
      "model_forward_time": 3.1327743530273438,
      "step": 597
    },
    {
      "epoch": 2.277374267578125e-06,
      "step": 597,
      "training_step_time": 10.056718826293945
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 3.165630340576172,
      "step": 598
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 598,
      "training_step_time": 9.936295986175537
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 3.4147250652313232,
      "step": 598
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 598,
      "training_step_time": 10.186818361282349
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 3.1368191242218018,
      "step": 598
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 598,
      "training_step_time": 9.909437417984009
    },
    {
      "epoch": 2.28118896484375e-06,
      "model_forward_time": 3.1327950954437256,
      "step": 598
    },
    {
      "epoch": 2.28118896484375e-06,
      "step": 598,
      "training_step_time": 10.012348651885986
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 3.1759090423583984,
      "step": 599
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 599,
      "training_step_time": 9.946239471435547
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 3.1422524452209473,
      "step": 599
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 599,
      "training_step_time": 9.911494970321655
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 3.1385657787323,
      "step": 599
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 599,
      "training_step_time": 10.173937559127808
    },
    {
      "epoch": 2.285003662109375e-06,
      "model_forward_time": 3.1363611221313477,
      "step": 599
    },
    {
      "epoch": 2.285003662109375e-06,
      "step": 599,
      "training_step_time": 10.28853154182434
    },
    {
      "epoch": 2.288818359375e-06,
      "grad_norm": 0.6864903569221497,
      "learning_rate": 1e-05,
      "loss": 0.1532,
      "step": 600
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 2.464118480682373,
      "step": 600
    },
    {
      "dynamics_loss_avg": 0.09324613735079765,
      "epoch": 2.288818359375e-06,
      "step": 600
    },
    {
      "action_loss_avg": 0.07415386028587818,
      "epoch": 2.288818359375e-06,
      "step": 600
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 600,
      "training_step_time": 7.492046356201172
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 3.1368536949157715,
      "step": 600
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 600,
      "training_step_time": 9.903002262115479
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 3.1362321376800537,
      "step": 600
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 600,
      "training_step_time": 9.900981903076172
    },
    {
      "epoch": 2.288818359375e-06,
      "model_forward_time": 3.3932740688323975,
      "step": 600
    },
    {
      "epoch": 2.288818359375e-06,
      "step": 600,
      "training_step_time": 10.270225048065186
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 3.176140308380127,
      "step": 601
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 601,
      "training_step_time": 9.94279432296753
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 3.135672092437744,
      "step": 601
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 601,
      "training_step_time": 9.901586532592773
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 3.1351962089538574,
      "step": 601
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 601,
      "training_step_time": 9.906058311462402
    },
    {
      "epoch": 2.292633056640625e-06,
      "model_forward_time": 3.1414597034454346,
      "step": 601
    },
    {
      "epoch": 2.292633056640625e-06,
      "step": 601,
      "training_step_time": 10.022037267684937
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 3.44497013092041,
      "step": 602
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 602,
      "training_step_time": 10.21412706375122
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 3.137559652328491,
      "step": 602
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 602,
      "training_step_time": 9.903678894042969
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 3.1353821754455566,
      "step": 602
    },
    {
      "dynamics_loss_avg": 0.08549757115542889,
      "epoch": 2.29644775390625e-06,
      "step": 602
    },
    {
      "action_loss_avg": 0.07330627888441085,
      "epoch": 2.29644775390625e-06,
      "step": 602
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 602,
      "training_step_time": 9.89892578125
    },
    {
      "epoch": 2.29644775390625e-06,
      "model_forward_time": 3.132054090499878,
      "step": 602
    },
    {
      "epoch": 2.29644775390625e-06,
      "step": 602,
      "training_step_time": 10.003688335418701
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 3.1749801635742188,
      "step": 603
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 603,
      "training_step_time": 9.945096015930176
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 3.135594606399536,
      "step": 603
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 603,
      "training_step_time": 10.209729194641113
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 3.1337904930114746,
      "step": 603
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 603,
      "training_step_time": 9.897037267684937
    },
    {
      "epoch": 2.300262451171875e-06,
      "model_forward_time": 3.142362594604492,
      "step": 603
    },
    {
      "epoch": 2.300262451171875e-06,
      "step": 603,
      "training_step_time": 10.274809122085571
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 3.1685378551483154,
      "step": 604
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 604,
      "training_step_time": 9.935423374176025
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 3.138181209564209,
      "step": 604
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 604,
      "training_step_time": 9.903007984161377
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 3.148970365524292,
      "step": 604
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 604,
      "training_step_time": 10.19042706489563
    },
    {
      "epoch": 2.3040771484375e-06,
      "model_forward_time": 3.1434426307678223,
      "step": 604
    },
    {
      "epoch": 2.3040771484375e-06,
      "step": 604,
      "training_step_time": 10.311960935592651
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 3.1821670532226562,
      "step": 605
    },
    {
      "dynamics_loss_avg": 0.08255472928285598,
      "epoch": 2.307891845703125e-06,
      "step": 605
    },
    {
      "action_loss_avg": 0.07055536881089211,
      "epoch": 2.307891845703125e-06,
      "step": 605
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 605,
      "training_step_time": 9.9848153591156
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 3.13722562789917,
      "step": 605
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 605,
      "training_step_time": 9.908527851104736
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 3.1428422927856445,
      "step": 605
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 605,
      "training_step_time": 9.911056280136108
    },
    {
      "epoch": 2.307891845703125e-06,
      "model_forward_time": 3.141395092010498,
      "step": 605
    },
    {
      "epoch": 2.307891845703125e-06,
      "step": 605,
      "training_step_time": 10.275044441223145
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 3.2075586318969727,
      "step": 606
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 606,
      "training_step_time": 9.980865955352783
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 3.131915330886841,
      "step": 606
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 606,
      "training_step_time": 9.896759748458862
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 2.4271628856658936,
      "step": 606
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 606,
      "training_step_time": 7.450900316238403
    },
    {
      "epoch": 2.31170654296875e-06,
      "model_forward_time": 3.1258931159973145,
      "step": 606
    },
    {
      "epoch": 2.31170654296875e-06,
      "step": 606,
      "training_step_time": 10.30530047416687
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 3.1359493732452393,
      "step": 607
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 607,
      "training_step_time": 10.209784746170044
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 3.133796215057373,
      "step": 607
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 607,
      "training_step_time": 9.90674114227295
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 2.42968487739563,
      "step": 607
    },
    {
      "dynamics_loss_avg": 0.07470661848783493,
      "epoch": 2.315521240234375e-06,
      "step": 607
    },
    {
      "action_loss_avg": 0.050901352986693384,
      "epoch": 2.315521240234375e-06,
      "step": 607
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 607,
      "training_step_time": 7.456771612167358
    },
    {
      "epoch": 2.315521240234375e-06,
      "model_forward_time": 3.1269028186798096,
      "step": 607
    },
    {
      "epoch": 2.315521240234375e-06,
      "step": 607,
      "training_step_time": 10.020756721496582
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 3.1795339584350586,
      "step": 608
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 608,
      "training_step_time": 9.955029726028442
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 3.1433515548706055,
      "step": 608
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 608,
      "training_step_time": 9.919072151184082
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 3.397261619567871,
      "step": 608
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 608,
      "training_step_time": 10.170182704925537
    },
    {
      "epoch": 2.3193359375e-06,
      "model_forward_time": 3.1344213485717773,
      "step": 608
    },
    {
      "epoch": 2.3193359375e-06,
      "step": 608,
      "training_step_time": 10.261730194091797
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 3.1574594974517822,
      "step": 609
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 609,
      "training_step_time": 9.929422378540039
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 3.13922119140625,
      "step": 609
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 609,
      "training_step_time": 9.908203840255737
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 3.418217897415161,
      "step": 609
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 609,
      "training_step_time": 10.216644048690796
    },
    {
      "epoch": 2.323150634765625e-06,
      "model_forward_time": 3.1343626976013184,
      "step": 609
    },
    {
      "epoch": 2.323150634765625e-06,
      "step": 609,
      "training_step_time": 10.244446992874146
    },
    {
      "epoch": 2.32696533203125e-06,
      "grad_norm": 0.6634290218353271,
      "learning_rate": 1e-05,
      "loss": 0.1566,
      "step": 610
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 3.1846516132354736,
      "step": 610
    },
    {
      "dynamics_loss_avg": 0.09037934057414532,
      "epoch": 2.32696533203125e-06,
      "step": 610
    },
    {
      "action_loss_avg": 0.06890361681580544,
      "epoch": 2.32696533203125e-06,
      "step": 610
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 610,
      "training_step_time": 9.961098432540894
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 3.1398417949676514,
      "step": 610
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 610,
      "training_step_time": 9.904629230499268
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 3.1354475021362305,
      "step": 610
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 610,
      "training_step_time": 9.905324935913086
    },
    {
      "epoch": 2.32696533203125e-06,
      "model_forward_time": 3.137089729309082,
      "step": 610
    },
    {
      "epoch": 2.32696533203125e-06,
      "step": 610,
      "training_step_time": 10.26568055152893
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 3.490332841873169,
      "step": 611
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 611,
      "training_step_time": 10.256898164749146
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 3.14043927192688,
      "step": 611
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 611,
      "training_step_time": 9.908392429351807
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 3.148624897003174,
      "step": 611
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 611,
      "training_step_time": 9.91783618927002
    },
    {
      "epoch": 2.330780029296875e-06,
      "model_forward_time": 3.1428003311157227,
      "step": 611
    },
    {
      "epoch": 2.330780029296875e-06,
      "step": 611,
      "training_step_time": 10.282727003097534
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 3.465632438659668,
      "step": 612
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 612,
      "training_step_time": 10.239466905593872
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 3.147719621658325,
      "step": 612
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 612,
      "training_step_time": 9.92442798614502
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 3.1409568786621094,
      "step": 612
    },
    {
      "dynamics_loss_avg": 0.0977070014923811,
      "epoch": 2.3345947265625e-06,
      "step": 612
    },
    {
      "action_loss_avg": 0.07459602504968643,
      "epoch": 2.3345947265625e-06,
      "step": 612
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 612,
      "training_step_time": 9.916968822479248
    },
    {
      "epoch": 2.3345947265625e-06,
      "model_forward_time": 2.436021566390991,
      "step": 612
    },
    {
      "epoch": 2.3345947265625e-06,
      "step": 612,
      "training_step_time": 10.033135175704956
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 3.1817657947540283,
      "step": 613
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 613,
      "training_step_time": 9.9625563621521
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 3.1299662590026855,
      "step": 613
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 613,
      "training_step_time": 9.90622091293335
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 3.3721261024475098,
      "step": 613
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 613,
      "training_step_time": 10.149908781051636
    },
    {
      "epoch": 2.338409423828125e-06,
      "model_forward_time": 2.4310624599456787,
      "step": 613
    },
    {
      "epoch": 2.338409423828125e-06,
      "step": 613,
      "training_step_time": 10.01898717880249
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 3.1782925128936768,
      "step": 614
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 614,
      "training_step_time": 9.949337482452393
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 3.1315629482269287,
      "step": 614
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 614,
      "training_step_time": 9.9118812084198
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 3.1340551376342773,
      "step": 614
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 614,
      "training_step_time": 9.910231590270996
    },
    {
      "epoch": 2.34222412109375e-06,
      "model_forward_time": 3.4908339977264404,
      "step": 614
    },
    {
      "epoch": 2.34222412109375e-06,
      "step": 614,
      "training_step_time": 10.361259460449219
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 2.4781973361968994,
      "step": 615
    },
    {
      "dynamics_loss_avg": 0.08788688816130161,
      "epoch": 2.346038818359375e-06,
      "step": 615
    },
    {
      "action_loss_avg": 0.07218681173399091,
      "epoch": 2.346038818359375e-06,
      "step": 615
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 615,
      "training_step_time": 7.509576082229614
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 3.1241066455841064,
      "step": 615
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 615,
      "training_step_time": 9.89652967453003
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 3.130506753921509,
      "step": 615
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 615,
      "training_step_time": 9.902735233306885
    },
    {
      "epoch": 2.346038818359375e-06,
      "model_forward_time": 3.134131908416748,
      "step": 615
    },
    {
      "epoch": 2.346038818359375e-06,
      "step": 615,
      "training_step_time": 10.26780366897583
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 3.172224998474121,
      "step": 616
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 616,
      "training_step_time": 9.948525667190552
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 3.130674123764038,
      "step": 616
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 616,
      "training_step_time": 9.90700650215149
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 2.4314115047454834,
      "step": 616
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 616,
      "training_step_time": 7.46286153793335
    },
    {
      "epoch": 2.349853515625e-06,
      "model_forward_time": 3.1286354064941406,
      "step": 616
    },
    {
      "epoch": 2.349853515625e-06,
      "step": 616,
      "training_step_time": 10.016583681106567
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 3.1666388511657715,
      "step": 617
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 617,
      "training_step_time": 10.213627338409424
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 3.1322388648986816,
      "step": 617
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 617,
      "training_step_time": 9.911644697189331
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 3.1365554332733154,
      "step": 617
    },
    {
      "dynamics_loss_avg": 0.09077354110777378,
      "epoch": 2.353668212890625e-06,
      "step": 617
    },
    {
      "action_loss_avg": 0.07137018889188766,
      "epoch": 2.353668212890625e-06,
      "step": 617
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 617,
      "training_step_time": 9.914060831069946
    },
    {
      "epoch": 2.353668212890625e-06,
      "model_forward_time": 3.1369426250457764,
      "step": 617
    },
    {
      "epoch": 2.353668212890625e-06,
      "step": 617,
      "training_step_time": 10.312679529190063
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 3.1409568786621094,
      "step": 618
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 618,
      "training_step_time": 9.918736457824707
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 3.418783187866211,
      "step": 618
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 618,
      "training_step_time": 10.192962408065796
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 2.4284255504608154,
      "step": 618
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 618,
      "training_step_time": 7.458227634429932
    },
    {
      "epoch": 2.35748291015625e-06,
      "model_forward_time": 3.1299543380737305,
      "step": 618
    },
    {
      "epoch": 2.35748291015625e-06,
      "step": 618,
      "training_step_time": 9.999609231948853
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 3.177751302719116,
      "step": 619
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 619,
      "training_step_time": 9.953645706176758
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 3.1354939937591553,
      "step": 619
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 619,
      "training_step_time": 9.913342714309692
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 3.1353845596313477,
      "step": 619
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 619,
      "training_step_time": 10.199505090713501
    },
    {
      "epoch": 2.361297607421875e-06,
      "model_forward_time": 3.136357069015503,
      "step": 619
    },
    {
      "epoch": 2.361297607421875e-06,
      "step": 619,
      "training_step_time": 10.257745265960693
    },
    {
      "epoch": 2.3651123046875e-06,
      "grad_norm": 0.9740402698516846,
      "learning_rate": 1e-05,
      "loss": 0.1535,
      "step": 620
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 3.174769878387451,
      "step": 620
    },
    {
      "dynamics_loss_avg": 0.09653582200407981,
      "epoch": 2.3651123046875e-06,
      "step": 620
    },
    {
      "action_loss_avg": 0.06985524855554104,
      "epoch": 2.3651123046875e-06,
      "step": 620
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 620,
      "training_step_time": 9.95549726486206
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 3.1336703300476074,
      "step": 620
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 620,
      "training_step_time": 9.90867567062378
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 3.1329727172851562,
      "step": 620
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 620,
      "training_step_time": 9.906029462814331
    },
    {
      "epoch": 2.3651123046875e-06,
      "model_forward_time": 2.7024965286254883,
      "step": 620
    },
    {
      "epoch": 2.3651123046875e-06,
      "step": 620,
      "training_step_time": 10.276148557662964
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 3.1747334003448486,
      "step": 621
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 621,
      "training_step_time": 9.94684362411499
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 3.130844831466675,
      "step": 621
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 621,
      "training_step_time": 9.909455299377441
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 3.129361629486084,
      "step": 621
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 621,
      "training_step_time": 9.906557559967041
    },
    {
      "epoch": 2.368927001953125e-06,
      "model_forward_time": 3.134704113006592,
      "step": 621
    },
    {
      "epoch": 2.368927001953125e-06,
      "step": 621,
      "training_step_time": 10.26294755935669
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 3.174238443374634,
      "step": 622
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 622,
      "training_step_time": 10.258589744567871
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 3.1328582763671875,
      "step": 622
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 622,
      "training_step_time": 9.903993368148804
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 3.1243979930877686,
      "step": 622
    },
    {
      "dynamics_loss_avg": 0.07871050722897052,
      "epoch": 2.37274169921875e-06,
      "step": 622
    },
    {
      "action_loss_avg": 0.07096661701798439,
      "epoch": 2.37274169921875e-06,
      "step": 622
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 622,
      "training_step_time": 9.897692918777466
    },
    {
      "epoch": 2.37274169921875e-06,
      "model_forward_time": 2.431903123855591,
      "step": 622
    },
    {
      "epoch": 2.37274169921875e-06,
      "step": 622,
      "training_step_time": 10.0212881565094
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 3.1679885387420654,
      "step": 623
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 623,
      "training_step_time": 9.940359115600586
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 3.1312525272369385,
      "step": 623
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 623,
      "training_step_time": 10.19072151184082
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 3.130744695663452,
      "step": 623
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 623,
      "training_step_time": 9.90445876121521
    },
    {
      "epoch": 2.376556396484375e-06,
      "model_forward_time": 3.1324779987335205,
      "step": 623
    },
    {
      "epoch": 2.376556396484375e-06,
      "step": 623,
      "training_step_time": 10.008777618408203
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 3.1604998111724854,
      "step": 624
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 624,
      "training_step_time": 9.935128927230835
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 3.132031202316284,
      "step": 624
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 624,
      "training_step_time": 9.908600807189941
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 3.388784646987915,
      "step": 624
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 624,
      "training_step_time": 10.162250518798828
    },
    {
      "epoch": 2.38037109375e-06,
      "model_forward_time": 3.1354598999023438,
      "step": 624
    },
    {
      "epoch": 2.38037109375e-06,
      "step": 624,
      "training_step_time": 10.287567138671875
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 3.1705334186553955,
      "step": 625
    },
    {
      "dynamics_loss_avg": 0.07128981612622738,
      "epoch": 2.384185791015625e-06,
      "step": 625
    },
    {
      "action_loss_avg": 0.05317660011351109,
      "epoch": 2.384185791015625e-06,
      "step": 625
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 625,
      "training_step_time": 9.952589750289917
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 3.132777452468872,
      "step": 625
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 625,
      "training_step_time": 9.911370515823364
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 3.1354944705963135,
      "step": 625
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 625,
      "training_step_time": 9.918950319290161
    },
    {
      "epoch": 2.384185791015625e-06,
      "model_forward_time": 3.1313323974609375,
      "step": 625
    },
    {
      "epoch": 2.384185791015625e-06,
      "step": 625,
      "training_step_time": 10.318885087966919
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 3.1832735538482666,
      "step": 626
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 626,
      "training_step_time": 9.963114023208618
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 3.131194591522217,
      "step": 626
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 626,
      "training_step_time": 9.904298543930054
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 3.137443780899048,
      "step": 626
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 626,
      "training_step_time": 9.913565635681152
    },
    {
      "epoch": 2.38800048828125e-06,
      "model_forward_time": 3.134305477142334,
      "step": 626
    },
    {
      "epoch": 2.38800048828125e-06,
      "step": 626,
      "training_step_time": 10.310105085372925
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 3.453812837600708,
      "step": 627
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 627,
      "training_step_time": 10.227871417999268
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 3.138334274291992,
      "step": 627
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 627,
      "training_step_time": 9.914890766143799
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 3.1401052474975586,
      "step": 627
    },
    {
      "dynamics_loss_avg": 0.07308995388448239,
      "epoch": 2.391815185546875e-06,
      "step": 627
    },
    {
      "action_loss_avg": 0.059756027348339555,
      "epoch": 2.391815185546875e-06,
      "step": 627
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 627,
      "training_step_time": 9.925628900527954
    },
    {
      "epoch": 2.391815185546875e-06,
      "model_forward_time": 3.1414973735809326,
      "step": 627
    },
    {
      "epoch": 2.391815185546875e-06,
      "step": 627,
      "training_step_time": 10.265216588973999
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 3.1808595657348633,
      "step": 628
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 628,
      "training_step_time": 9.95906138420105
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 3.1426868438720703,
      "step": 628
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 628,
      "training_step_time": 10.202625274658203
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 3.1444380283355713,
      "step": 628
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 628,
      "training_step_time": 9.91682505607605
    },
    {
      "epoch": 2.3956298828125e-06,
      "model_forward_time": 3.1421732902526855,
      "step": 628
    },
    {
      "epoch": 2.3956298828125e-06,
      "step": 628,
      "training_step_time": 10.0174081325531
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 3.1755921840667725,
      "step": 629
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 629,
      "training_step_time": 9.954171895980835
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 3.1350502967834473,
      "step": 629
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 629,
      "training_step_time": 9.913802862167358
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 3.446389675140381,
      "step": 629
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 629,
      "training_step_time": 10.223371505737305
    },
    {
      "epoch": 2.399444580078125e-06,
      "model_forward_time": 3.1333749294281006,
      "step": 629
    },
    {
      "epoch": 2.399444580078125e-06,
      "step": 629,
      "training_step_time": 10.02001428604126
    },
    {
      "epoch": 2.40325927734375e-06,
      "grad_norm": 0.6116428971290588,
      "learning_rate": 1e-05,
      "loss": 0.1532,
      "step": 630
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 3.1808712482452393,
      "step": 630
    },
    {
      "dynamics_loss_avg": 0.07858083993196488,
      "epoch": 2.40325927734375e-06,
      "step": 630
    },
    {
      "action_loss_avg": 0.06099752970039844,
      "epoch": 2.40325927734375e-06,
      "step": 630
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 630,
      "training_step_time": 9.996525526046753
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 3.130902051925659,
      "step": 630
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 630,
      "training_step_time": 9.935850143432617
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 3.13382625579834,
      "step": 630
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 630,
      "training_step_time": 9.907348155975342
    },
    {
      "epoch": 2.40325927734375e-06,
      "model_forward_time": 3.3895013332366943,
      "step": 630
    },
    {
      "epoch": 2.40325927734375e-06,
      "step": 630,
      "training_step_time": 10.299451351165771
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 3.175023078918457,
      "step": 631
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 631,
      "training_step_time": 9.94901990890503
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 3.132455348968506,
      "step": 631
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 631,
      "training_step_time": 9.90845251083374
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 3.133319854736328,
      "step": 631
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 631,
      "training_step_time": 9.907452821731567
    },
    {
      "epoch": 2.407073974609375e-06,
      "model_forward_time": 3.1390392780303955,
      "step": 631
    },
    {
      "epoch": 2.407073974609375e-06,
      "step": 631,
      "training_step_time": 10.325077056884766
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 2.763137102127075,
      "step": 632
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 632,
      "training_step_time": 7.792443037033081
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 3.1325275897979736,
      "step": 632
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 632,
      "training_step_time": 9.905545473098755
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 3.1334493160247803,
      "step": 632
    },
    {
      "dynamics_loss_avg": 0.09755456782877445,
      "epoch": 2.410888671875e-06,
      "step": 632
    },
    {
      "action_loss_avg": 0.072963547706604,
      "epoch": 2.410888671875e-06,
      "step": 632
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 632,
      "training_step_time": 9.910366296768188
    },
    {
      "epoch": 2.410888671875e-06,
      "model_forward_time": 3.1366043090820312,
      "step": 632
    },
    {
      "epoch": 2.410888671875e-06,
      "step": 632,
      "training_step_time": 10.02237606048584
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 3.1738932132720947,
      "step": 633
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 633,
      "training_step_time": 9.954826593399048
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 2.430997848510742,
      "step": 633
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 633,
      "training_step_time": 7.459666967391968
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 3.400296211242676,
      "step": 633
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 633,
      "training_step_time": 10.173821449279785
    },
    {
      "epoch": 2.414703369140625e-06,
      "model_forward_time": 3.1304898262023926,
      "step": 633
    },
    {
      "epoch": 2.414703369140625e-06,
      "step": 633,
      "training_step_time": 10.003393411636353
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 3.1764919757843018,
      "step": 634
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 634,
      "training_step_time": 9.952353477478027
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 3.132539987564087,
      "step": 634
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 634,
      "training_step_time": 9.907483100891113
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 3.1359522342681885,
      "step": 634
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 634,
      "training_step_time": 10.18677282333374
    },
    {
      "epoch": 2.41851806640625e-06,
      "model_forward_time": 3.138108015060425,
      "step": 634
    },
    {
      "epoch": 2.41851806640625e-06,
      "step": 634,
      "training_step_time": 10.287760496139526
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 3.174078941345215,
      "step": 635
    },
    {
      "dynamics_loss_avg": 0.08903018422424794,
      "epoch": 2.422332763671875e-06,
      "step": 635
    },
    {
      "action_loss_avg": 0.07040941044688225,
      "epoch": 2.422332763671875e-06,
      "step": 635
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 635,
      "training_step_time": 9.95580768585205
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 3.132103681564331,
      "step": 635
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 635,
      "training_step_time": 9.911885499954224
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 3.139420509338379,
      "step": 635
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 635,
      "training_step_time": 9.9160315990448
    },
    {
      "epoch": 2.422332763671875e-06,
      "model_forward_time": 3.4102563858032227,
      "step": 635
    },
    {
      "epoch": 2.422332763671875e-06,
      "step": 635,
      "training_step_time": 10.319992065429688
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 3.1862027645111084,
      "step": 636
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 636,
      "training_step_time": 9.962339401245117
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 3.134005308151245,
      "step": 636
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 636,
      "training_step_time": 9.910238981246948
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 3.1388440132141113,
      "step": 636
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 636,
      "training_step_time": 9.90944528579712
    },
    {
      "epoch": 2.4261474609375e-06,
      "model_forward_time": 3.131483793258667,
      "step": 636
    },
    {
      "epoch": 2.4261474609375e-06,
      "step": 636,
      "training_step_time": 10.267070531845093
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 3.1775786876678467,
      "step": 637
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 637,
      "training_step_time": 10.264625549316406
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 3.1342129707336426,
      "step": 637
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 637,
      "training_step_time": 9.91198182106018
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 3.1353702545166016,
      "step": 637
    },
    {
      "dynamics_loss_avg": 0.101341000944376,
      "epoch": 2.429962158203125e-06,
      "step": 637
    },
    {
      "action_loss_avg": 0.0845991786569357,
      "epoch": 2.429962158203125e-06,
      "step": 637
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 637,
      "training_step_time": 9.916167497634888
    },
    {
      "epoch": 2.429962158203125e-06,
      "model_forward_time": 3.1308257579803467,
      "step": 637
    },
    {
      "epoch": 2.429962158203125e-06,
      "step": 637,
      "training_step_time": 10.03302812576294
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 3.1806516647338867,
      "step": 638
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 638,
      "training_step_time": 9.951728820800781
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 3.3797473907470703,
      "step": 638
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 638,
      "training_step_time": 10.15850281715393
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 3.1325526237487793,
      "step": 638
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 638,
      "training_step_time": 9.904590368270874
    },
    {
      "epoch": 2.43377685546875e-06,
      "model_forward_time": 3.1300175189971924,
      "step": 638
    },
    {
      "epoch": 2.43377685546875e-06,
      "step": 638,
      "training_step_time": 10.231375455856323
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 3.176551342010498,
      "step": 639
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 639,
      "training_step_time": 9.954128503799438
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 3.1355772018432617,
      "step": 639
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 639,
      "training_step_time": 9.911750316619873
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 3.424278497695923,
      "step": 639
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 639,
      "training_step_time": 10.203276634216309
    },
    {
      "epoch": 2.437591552734375e-06,
      "model_forward_time": 3.130746603012085,
      "step": 639
    },
    {
      "epoch": 2.437591552734375e-06,
      "step": 639,
      "training_step_time": 10.022672891616821
    },
    {
      "epoch": 2.44140625e-06,
      "grad_norm": 0.6220146417617798,
      "learning_rate": 1e-05,
      "loss": 0.1577,
      "step": 640
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 3.175571918487549,
      "step": 640
    },
    {
      "dynamics_loss_avg": 0.09605773873627185,
      "epoch": 2.44140625e-06,
      "step": 640
    },
    {
      "action_loss_avg": 0.08564151264727116,
      "epoch": 2.44140625e-06,
      "step": 640
    },
    {
      "epoch": 2.44140625e-06,
      "step": 640,
      "training_step_time": 9.957147121429443
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 3.137582302093506,
      "step": 640
    },
    {
      "epoch": 2.44140625e-06,
      "step": 640,
      "training_step_time": 9.914725542068481
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 3.1392500400543213,
      "step": 640
    },
    {
      "epoch": 2.44140625e-06,
      "step": 640,
      "training_step_time": 9.91572117805481
    },
    {
      "epoch": 2.44140625e-06,
      "model_forward_time": 3.415269374847412,
      "step": 640
    },
    {
      "epoch": 2.44140625e-06,
      "step": 640,
      "training_step_time": 10.290184020996094
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 3.187796115875244,
      "step": 641
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 641,
      "training_step_time": 9.972929954528809
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 3.1432554721832275,
      "step": 641
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 641,
      "training_step_time": 9.921466827392578
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 3.1390981674194336,
      "step": 641
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 641,
      "training_step_time": 9.917422533035278
    },
    {
      "epoch": 2.445220947265625e-06,
      "model_forward_time": 3.1415836811065674,
      "step": 641
    },
    {
      "epoch": 2.445220947265625e-06,
      "step": 641,
      "training_step_time": 10.315026044845581
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 3.4454026222229004,
      "step": 642
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 642,
      "training_step_time": 10.227022647857666
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 3.139008045196533,
      "step": 642
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 642,
      "training_step_time": 9.915907382965088
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 3.1403775215148926,
      "step": 642
    },
    {
      "dynamics_loss_avg": 0.08125521056354046,
      "epoch": 2.44903564453125e-06,
      "step": 642
    },
    {
      "action_loss_avg": 0.06550402697175742,
      "epoch": 2.44903564453125e-06,
      "step": 642
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 642,
      "training_step_time": 9.921920537948608
    },
    {
      "epoch": 2.44903564453125e-06,
      "model_forward_time": 3.1411640644073486,
      "step": 642
    },
    {
      "epoch": 2.44903564453125e-06,
      "step": 642,
      "training_step_time": 10.03543758392334
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 3.173048496246338,
      "step": 643
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 643,
      "training_step_time": 9.950448751449585
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 3.1353068351745605,
      "step": 643
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 643,
      "training_step_time": 10.197712898254395
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 3.134382724761963,
      "step": 643
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 643,
      "training_step_time": 9.910289764404297
    },
    {
      "epoch": 2.452850341796875e-06,
      "model_forward_time": 3.1388113498687744,
      "step": 643
    },
    {
      "epoch": 2.452850341796875e-06,
      "step": 643,
      "training_step_time": 10.014825820922852
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 3.175321102142334,
      "step": 644
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 644,
      "training_step_time": 9.952503204345703
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 3.1340696811676025,
      "step": 644
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 644,
      "training_step_time": 9.913148164749146
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 3.4279956817626953,
      "step": 644
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 644,
      "training_step_time": 10.208064317703247
    },
    {
      "epoch": 2.4566650390625e-06,
      "model_forward_time": 3.141050338745117,
      "step": 644
    },
    {
      "epoch": 2.4566650390625e-06,
      "step": 644,
      "training_step_time": 10.290146827697754
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 3.1642656326293945,
      "step": 645
    },
    {
      "dynamics_loss_avg": 0.10144047737121582,
      "epoch": 2.460479736328125e-06,
      "step": 645
    },
    {
      "action_loss_avg": 0.08145553171634674,
      "epoch": 2.460479736328125e-06,
      "step": 645
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 645,
      "training_step_time": 9.940768957138062
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 3.134368419647217,
      "step": 645
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 645,
      "training_step_time": 9.910902261734009
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 3.1363515853881836,
      "step": 645
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 645,
      "training_step_time": 9.908604621887207
    },
    {
      "epoch": 2.460479736328125e-06,
      "model_forward_time": 3.1353402137756348,
      "step": 645
    },
    {
      "epoch": 2.460479736328125e-06,
      "step": 645,
      "training_step_time": 10.286943197250366
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 3.186506986618042,
      "step": 646
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 646,
      "training_step_time": 9.962421178817749
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 3.1358728408813477,
      "step": 646
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 646,
      "training_step_time": 9.909238338470459
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 3.1433048248291016,
      "step": 646
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 646,
      "training_step_time": 9.920368194580078
    },
    {
      "epoch": 2.46429443359375e-06,
      "model_forward_time": 3.143878936767578,
      "step": 646
    },
    {
      "epoch": 2.46429443359375e-06,
      "step": 646,
      "training_step_time": 10.326939344406128
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 3.455538272857666,
      "step": 647
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 647,
      "training_step_time": 10.231457233428955
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 3.139209032058716,
      "step": 647
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 647,
      "training_step_time": 9.91506576538086
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 3.136639356613159,
      "step": 647
    },
    {
      "dynamics_loss_avg": 0.09428081214427948,
      "epoch": 2.468109130859375e-06,
      "step": 647
    },
    {
      "action_loss_avg": 0.08584567196667195,
      "epoch": 2.468109130859375e-06,
      "step": 647
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 647,
      "training_step_time": 9.915005683898926
    },
    {
      "epoch": 2.468109130859375e-06,
      "model_forward_time": 3.139024496078491,
      "step": 647
    },
    {
      "epoch": 2.468109130859375e-06,
      "step": 647,
      "training_step_time": 10.306940078735352
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 3.169919967651367,
      "step": 648
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 648,
      "training_step_time": 9.952396392822266
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 3.1364266872406006,
      "step": 648
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 648,
      "training_step_time": 10.165050506591797
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 3.1407294273376465,
      "step": 648
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 648,
      "training_step_time": 9.908821105957031
    },
    {
      "epoch": 2.471923828125e-06,
      "model_forward_time": 2.4340643882751465,
      "step": 648
    },
    {
      "epoch": 2.471923828125e-06,
      "step": 648,
      "training_step_time": 10.020205974578857
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 3.1840288639068604,
      "step": 649
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 649,
      "training_step_time": 9.959685325622559
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 3.136765956878662,
      "step": 649
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 649,
      "training_step_time": 9.903695344924927
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 3.443990707397461,
      "step": 649
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 649,
      "training_step_time": 10.213093042373657
    },
    {
      "epoch": 2.475738525390625e-06,
      "model_forward_time": 3.1296799182891846,
      "step": 649
    },
    {
      "epoch": 2.475738525390625e-06,
      "step": 649,
      "training_step_time": 10.010843753814697
    },
    {
      "epoch": 2.47955322265625e-06,
      "grad_norm": 1.1348289251327515,
      "learning_rate": 1e-05,
      "loss": 0.1519,
      "step": 650
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 3.177417755126953,
      "step": 650
    },
    {
      "dynamics_loss_avg": 0.08230208232998848,
      "epoch": 2.47955322265625e-06,
      "step": 650
    },
    {
      "action_loss_avg": 0.06257175989449024,
      "epoch": 2.47955322265625e-06,
      "step": 650
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 650,
      "training_step_time": 9.9496328830719
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 3.1294186115264893,
      "step": 650
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 650,
      "training_step_time": 9.901514530181885
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 3.130572557449341,
      "step": 650
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 650,
      "training_step_time": 9.897620916366577
    },
    {
      "epoch": 2.47955322265625e-06,
      "model_forward_time": 3.130154609680176,
      "step": 650
    },
    {
      "epoch": 2.47955322265625e-06,
      "step": 650,
      "training_step_time": 9.996310234069824
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 3.467772960662842,
      "step": 651
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 651,
      "training_step_time": 10.230863809585571
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 3.1326658725738525,
      "step": 651
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 651,
      "training_step_time": 9.909039497375488
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 3.127833604812622,
      "step": 651
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 651,
      "training_step_time": 9.89661717414856
    },
    {
      "epoch": 2.483367919921875e-06,
      "model_forward_time": 3.1327714920043945,
      "step": 651
    },
    {
      "epoch": 2.483367919921875e-06,
      "step": 651,
      "training_step_time": 10.018359184265137
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 3.459383487701416,
      "step": 652
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 652,
      "training_step_time": 10.232820749282837
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 3.1389293670654297,
      "step": 652
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 652,
      "training_step_time": 9.912428617477417
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 3.1307597160339355,
      "step": 652
    },
    {
      "dynamics_loss_avg": 0.09286714605987072,
      "epoch": 2.4871826171875e-06,
      "step": 652
    },
    {
      "action_loss_avg": 0.07809786461293697,
      "epoch": 2.4871826171875e-06,
      "step": 652
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 652,
      "training_step_time": 9.906273603439331
    },
    {
      "epoch": 2.4871826171875e-06,
      "model_forward_time": 3.134054660797119,
      "step": 652
    },
    {
      "epoch": 2.4871826171875e-06,
      "step": 652,
      "training_step_time": 10.313947439193726
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 3.1783595085144043,
      "step": 653
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 653,
      "training_step_time": 9.941091060638428
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 3.13038969039917,
      "step": 653
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 653,
      "training_step_time": 10.189688205718994
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 3.1302239894866943,
      "step": 653
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 653,
      "training_step_time": 9.897094011306763
    },
    {
      "epoch": 2.490997314453125e-06,
      "model_forward_time": 3.137504816055298,
      "step": 653
    },
    {
      "epoch": 2.490997314453125e-06,
      "step": 653,
      "training_step_time": 10.297138214111328
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 2.4579477310180664,
      "step": 654
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 654,
      "training_step_time": 7.477669715881348
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 3.1259748935699463,
      "step": 654
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 654,
      "training_step_time": 9.892525911331177
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 3.1314687728881836,
      "step": 654
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 654,
      "training_step_time": 10.173628330230713
    },
    {
      "epoch": 2.49481201171875e-06,
      "model_forward_time": 3.129696846008301,
      "step": 654
    },
    {
      "epoch": 2.49481201171875e-06,
      "step": 654,
      "training_step_time": 10.014428853988647
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 3.1743030548095703,
      "step": 655
    },
    {
      "dynamics_loss_avg": 0.10325966626405716,
      "epoch": 2.498626708984375e-06,
      "step": 655
    },
    {
      "action_loss_avg": 0.0651224672794342,
      "epoch": 2.498626708984375e-06,
      "step": 655
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 655,
      "training_step_time": 9.944697856903076
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 3.1325125694274902,
      "step": 655
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 655,
      "training_step_time": 9.902527809143066
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 3.1314480304718018,
      "step": 655
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 655,
      "training_step_time": 9.90213656425476
    },
    {
      "epoch": 2.498626708984375e-06,
      "model_forward_time": 3.4077329635620117,
      "step": 655
    },
    {
      "epoch": 2.498626708984375e-06,
      "step": 655,
      "training_step_time": 10.27542495727539
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 3.175727605819702,
      "step": 656
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 656,
      "training_step_time": 9.952400207519531
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 3.1352434158325195,
      "step": 656
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 656,
      "training_step_time": 9.91303014755249
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 3.149106979370117,
      "step": 656
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 656,
      "training_step_time": 9.930697202682495
    },
    {
      "epoch": 2.50244140625e-06,
      "model_forward_time": 3.139249801635742,
      "step": 656
    },
    {
      "epoch": 2.50244140625e-06,
      "step": 656,
      "training_step_time": 10.034266948699951
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 3.1717822551727295,
      "step": 657
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 657,
      "training_step_time": 10.252189636230469
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 3.1412737369537354,
      "step": 657
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 657,
      "training_step_time": 9.920946836471558
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 3.140913248062134,
      "step": 657
    },
    {
      "dynamics_loss_avg": 0.07937949914485216,
      "epoch": 2.506256103515625e-06,
      "step": 657
    },
    {
      "action_loss_avg": 0.0601165602914989,
      "epoch": 2.506256103515625e-06,
      "step": 657
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 657,
      "training_step_time": 9.924616813659668
    },
    {
      "epoch": 2.506256103515625e-06,
      "model_forward_time": 3.1441614627838135,
      "step": 657
    },
    {
      "epoch": 2.506256103515625e-06,
      "step": 657,
      "training_step_time": 10.311611413955688
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 3.169334888458252,
      "step": 658
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 658,
      "training_step_time": 9.96434235572815
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 3.39754581451416,
      "step": 658
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 658,
      "training_step_time": 10.171003103256226
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 3.1362409591674805,
      "step": 658
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 658,
      "training_step_time": 9.911130666732788
    },
    {
      "epoch": 2.51007080078125e-06,
      "model_forward_time": 3.1297366619110107,
      "step": 658
    },
    {
      "epoch": 2.51007080078125e-06,
      "step": 658,
      "training_step_time": 10.303614854812622
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 3.1669511795043945,
      "step": 659
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 659,
      "training_step_time": 9.937145948410034
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 3.1307382583618164,
      "step": 659
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 659,
      "training_step_time": 9.904884576797485
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 2.4270148277282715,
      "step": 659
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 659,
      "training_step_time": 7.721078157424927
    },
    {
      "epoch": 2.513885498046875e-06,
      "model_forward_time": 3.124478578567505,
      "step": 659
    },
    {
      "epoch": 2.513885498046875e-06,
      "step": 659,
      "training_step_time": 10.263849258422852
    },
    {
      "epoch": 2.5177001953125e-06,
      "grad_norm": 0.9386076331138611,
      "learning_rate": 1e-05,
      "loss": 0.1487,
      "step": 660
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 3.1629021167755127,
      "step": 660
    },
    {
      "dynamics_loss_avg": 0.09293313808739186,
      "epoch": 2.5177001953125e-06,
      "step": 660
    },
    {
      "action_loss_avg": 0.0764224199578166,
      "epoch": 2.5177001953125e-06,
      "step": 660
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 660,
      "training_step_time": 9.934363603591919
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 3.1313223838806152,
      "step": 660
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 660,
      "training_step_time": 9.90035104751587
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 3.1293227672576904,
      "step": 660
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 660,
      "training_step_time": 9.904757022857666
    },
    {
      "epoch": 2.5177001953125e-06,
      "model_forward_time": 3.4250235557556152,
      "step": 660
    },
    {
      "epoch": 2.5177001953125e-06,
      "step": 660,
      "training_step_time": 10.292692184448242
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 2.4578611850738525,
      "step": 661
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 661,
      "training_step_time": 7.488588809967041
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 3.138443946838379,
      "step": 661
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 661,
      "training_step_time": 9.917600154876709
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 3.1356656551361084,
      "step": 661
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 661,
      "training_step_time": 9.913959264755249
    },
    {
      "epoch": 2.521514892578125e-06,
      "model_forward_time": 3.141324996948242,
      "step": 661
    },
    {
      "epoch": 2.521514892578125e-06,
      "step": 661,
      "training_step_time": 10.277630090713501
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 3.1820547580718994,
      "step": 662
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 662,
      "training_step_time": 10.245189905166626
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 3.1448476314544678,
      "step": 662
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 662,
      "training_step_time": 9.92634916305542
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 3.1384048461914062,
      "step": 662
    },
    {
      "dynamics_loss_avg": 0.08068276830017566,
      "epoch": 2.52532958984375e-06,
      "step": 662
    },
    {
      "action_loss_avg": 0.05595023427158594,
      "epoch": 2.52532958984375e-06,
      "step": 662
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 662,
      "training_step_time": 9.92068600654602
    },
    {
      "epoch": 2.52532958984375e-06,
      "model_forward_time": 3.1400063037872314,
      "step": 662
    },
    {
      "epoch": 2.52532958984375e-06,
      "step": 662,
      "training_step_time": 10.317858695983887
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 3.1547977924346924,
      "step": 663
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 663,
      "training_step_time": 9.944830179214478
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 3.1323704719543457,
      "step": 663
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 663,
      "training_step_time": 10.221604108810425
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 3.1439030170440674,
      "step": 663
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 663,
      "training_step_time": 9.922796964645386
    },
    {
      "epoch": 2.529144287109375e-06,
      "model_forward_time": 2.429142475128174,
      "step": 663
    },
    {
      "epoch": 2.529144287109375e-06,
      "step": 663,
      "training_step_time": 10.280848026275635
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 2.4830384254455566,
      "step": 664
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 664,
      "training_step_time": 7.517438173294067
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 3.131499767303467,
      "step": 664
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 664,
      "training_step_time": 9.907835721969604
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 3.1321799755096436,
      "step": 664
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 664,
      "training_step_time": 9.909749269485474
    },
    {
      "epoch": 2.532958984375e-06,
      "model_forward_time": 3.416980743408203,
      "step": 664
    },
    {
      "epoch": 2.532958984375e-06,
      "step": 664,
      "training_step_time": 10.291985988616943
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 3.1858556270599365,
      "step": 665
    },
    {
      "dynamics_loss_avg": 0.06922592390328645,
      "epoch": 2.536773681640625e-06,
      "step": 665
    },
    {
      "action_loss_avg": 0.0489938635379076,
      "epoch": 2.536773681640625e-06,
      "step": 665
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 665,
      "training_step_time": 9.970313549041748
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 3.1388604640960693,
      "step": 665
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 665,
      "training_step_time": 9.919227361679077
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 3.1431801319122314,
      "step": 665
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 665,
      "training_step_time": 9.925679206848145
    },
    {
      "epoch": 2.536773681640625e-06,
      "model_forward_time": 3.1401095390319824,
      "step": 665
    },
    {
      "epoch": 2.536773681640625e-06,
      "step": 665,
      "training_step_time": 10.035357475280762
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 3.5125160217285156,
      "step": 666
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 666,
      "training_step_time": 10.290945053100586
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 3.133229970932007,
      "step": 666
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 666,
      "training_step_time": 9.91399073600769
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 3.142202138900757,
      "step": 666
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 666,
      "training_step_time": 9.916357278823853
    },
    {
      "epoch": 2.54058837890625e-06,
      "model_forward_time": 3.1388590335845947,
      "step": 666
    },
    {
      "epoch": 2.54058837890625e-06,
      "step": 666,
      "training_step_time": 10.016345262527466
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 3.183220148086548,
      "step": 667
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 667,
      "training_step_time": 10.226580381393433
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 3.136854410171509,
      "step": 667
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 667,
      "training_step_time": 9.910508871078491
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 3.1457247734069824,
      "step": 667
    },
    {
      "dynamics_loss_avg": 0.06773452255874872,
      "epoch": 2.544403076171875e-06,
      "step": 667
    },
    {
      "action_loss_avg": 0.04166306396946311,
      "epoch": 2.544403076171875e-06,
      "step": 667
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 667,
      "training_step_time": 9.927067518234253
    },
    {
      "epoch": 2.544403076171875e-06,
      "model_forward_time": 3.143094301223755,
      "step": 667
    },
    {
      "epoch": 2.544403076171875e-06,
      "step": 667,
      "training_step_time": 10.307305335998535
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 3.184446334838867,
      "step": 668
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 668,
      "training_step_time": 9.966526746749878
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 3.135725975036621,
      "step": 668
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 668,
      "training_step_time": 10.231353282928467
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 3.137733221054077,
      "step": 668
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 668,
      "training_step_time": 9.916006803512573
    },
    {
      "epoch": 2.5482177734375e-06,
      "model_forward_time": 3.1373631954193115,
      "step": 668
    },
    {
      "epoch": 2.5482177734375e-06,
      "step": 668,
      "training_step_time": 10.294580936431885
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 3.1822926998138428,
      "step": 669
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 669,
      "training_step_time": 9.967575073242188
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 3.133883237838745,
      "step": 669
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 669,
      "training_step_time": 9.908357858657837
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 3.405930519104004,
      "step": 669
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 669,
      "training_step_time": 10.180428266525269
    },
    {
      "epoch": 2.552032470703125e-06,
      "model_forward_time": 3.131171941757202,
      "step": 669
    },
    {
      "epoch": 2.552032470703125e-06,
      "step": 669,
      "training_step_time": 10.308042287826538
    },
    {
      "epoch": 2.55584716796875e-06,
      "grad_norm": 0.882199764251709,
      "learning_rate": 1e-05,
      "loss": 0.1474,
      "step": 670
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 3.1728405952453613,
      "step": 670
    },
    {
      "dynamics_loss_avg": 0.08886316046118736,
      "epoch": 2.55584716796875e-06,
      "step": 670
    },
    {
      "action_loss_avg": 0.07697389610111713,
      "epoch": 2.55584716796875e-06,
      "step": 670
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 670,
      "training_step_time": 9.949716329574585
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 3.133244276046753,
      "step": 670
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 670,
      "training_step_time": 9.911460876464844
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 3.1387367248535156,
      "step": 670
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 670,
      "training_step_time": 9.919389724731445
    },
    {
      "epoch": 2.55584716796875e-06,
      "model_forward_time": 3.4048213958740234,
      "step": 670
    },
    {
      "epoch": 2.55584716796875e-06,
      "step": 670,
      "training_step_time": 10.27574872970581
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 3.1750967502593994,
      "step": 671
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 671,
      "training_step_time": 9.951587915420532
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 3.1308236122131348,
      "step": 671
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 671,
      "training_step_time": 9.90769100189209
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 3.1324880123138428,
      "step": 671
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 671,
      "training_step_time": 9.909379243850708
    },
    {
      "epoch": 2.559661865234375e-06,
      "model_forward_time": 3.126995801925659,
      "step": 671
    },
    {
      "epoch": 2.559661865234375e-06,
      "step": 671,
      "training_step_time": 10.254241704940796
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 2.762051582336426,
      "step": 672
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 672,
      "training_step_time": 7.792760610580444
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 3.128152370452881,
      "step": 672
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 672,
      "training_step_time": 9.901444435119629
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 3.1278982162475586,
      "step": 672
    },
    {
      "dynamics_loss_avg": 0.07723485082387924,
      "epoch": 2.5634765625e-06,
      "step": 672
    },
    {
      "action_loss_avg": 0.04561297986656428,
      "epoch": 2.5634765625e-06,
      "step": 672
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 672,
      "training_step_time": 9.905178308486938
    },
    {
      "epoch": 2.5634765625e-06,
      "model_forward_time": 3.1316051483154297,
      "step": 672
    },
    {
      "epoch": 2.5634765625e-06,
      "step": 672,
      "training_step_time": 10.284201383590698
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 3.1572723388671875,
      "step": 673
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 673,
      "training_step_time": 9.931427478790283
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 3.127530097961426,
      "step": 673
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 673,
      "training_step_time": 10.185888767242432
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 3.1287436485290527,
      "step": 673
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 673,
      "training_step_time": 9.902393341064453
    },
    {
      "epoch": 2.567291259765625e-06,
      "model_forward_time": 3.1310713291168213,
      "step": 673
    },
    {
      "epoch": 2.567291259765625e-06,
      "step": 673,
      "training_step_time": 10.259343385696411
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 3.179659128189087,
      "step": 674
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 674,
      "training_step_time": 9.955702781677246
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 3.1286091804504395,
      "step": 674
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 674,
      "training_step_time": 9.899864912033081
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 3.1248815059661865,
      "step": 674
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 674,
      "training_step_time": 10.177915334701538
    },
    {
      "epoch": 2.57110595703125e-06,
      "model_forward_time": 3.1294474601745605,
      "step": 674
    },
    {
      "epoch": 2.57110595703125e-06,
      "step": 674,
      "training_step_time": 10.255043983459473
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 3.173689126968384,
      "step": 675
    },
    {
      "dynamics_loss_avg": 0.0786284787580371,
      "epoch": 2.574920654296875e-06,
      "step": 675
    },
    {
      "action_loss_avg": 0.06044270321726799,
      "epoch": 2.574920654296875e-06,
      "step": 675
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 675,
      "training_step_time": 9.947694778442383
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 3.1258978843688965,
      "step": 675
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 675,
      "training_step_time": 9.901533603668213
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 3.1289377212524414,
      "step": 675
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 675,
      "training_step_time": 9.899978876113892
    },
    {
      "epoch": 2.574920654296875e-06,
      "model_forward_time": 3.4188108444213867,
      "step": 675
    },
    {
      "epoch": 2.574920654296875e-06,
      "step": 675,
      "training_step_time": 10.28550934791565
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 3.1788129806518555,
      "step": 676
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 676,
      "training_step_time": 9.95696473121643
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 3.1321616172790527,
      "step": 676
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 676,
      "training_step_time": 9.911079168319702
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 3.133665084838867,
      "step": 676
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 676,
      "training_step_time": 9.906851768493652
    },
    {
      "epoch": 2.5787353515625e-06,
      "model_forward_time": 3.1342480182647705,
      "step": 676
    },
    {
      "epoch": 2.5787353515625e-06,
      "step": 676,
      "training_step_time": 10.015996217727661
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 3.1725590229034424,
      "step": 677
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 677,
      "training_step_time": 10.24960732460022
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 3.1280603408813477,
      "step": 677
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 677,
      "training_step_time": 9.904472589492798
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 3.1341521739959717,
      "step": 677
    },
    {
      "dynamics_loss_avg": 0.10320555381476879,
      "epoch": 2.582550048828125e-06,
      "step": 677
    },
    {
      "action_loss_avg": 0.07125478405505418,
      "epoch": 2.582550048828125e-06,
      "step": 677
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 677,
      "training_step_time": 9.922131299972534
    },
    {
      "epoch": 2.582550048828125e-06,
      "model_forward_time": 2.4305224418640137,
      "step": 677
    },
    {
      "epoch": 2.582550048828125e-06,
      "step": 677,
      "training_step_time": 10.281080722808838
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 3.172675609588623,
      "step": 678
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 678,
      "training_step_time": 9.948363065719604
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 3.379908561706543,
      "step": 678
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 678,
      "training_step_time": 10.153486013412476
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 3.127758026123047,
      "step": 678
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 678,
      "training_step_time": 9.900649070739746
    },
    {
      "epoch": 2.58636474609375e-06,
      "model_forward_time": 3.1320579051971436,
      "step": 678
    },
    {
      "epoch": 2.58636474609375e-06,
      "step": 678,
      "training_step_time": 10.289294242858887
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 3.1615853309631348,
      "step": 679
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 679,
      "training_step_time": 9.936627864837646
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 3.132551670074463,
      "step": 679
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 679,
      "training_step_time": 9.913803339004517
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 3.1327991485595703,
      "step": 679
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 679,
      "training_step_time": 10.17409634590149
    },
    {
      "epoch": 2.590179443359375e-06,
      "model_forward_time": 3.1349470615386963,
      "step": 679
    },
    {
      "epoch": 2.590179443359375e-06,
      "step": 679,
      "training_step_time": 10.332041025161743
    },
    {
      "epoch": 2.593994140625e-06,
      "grad_norm": 0.7461228966712952,
      "learning_rate": 1e-05,
      "loss": 0.1442,
      "step": 680
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 3.167384624481201,
      "step": 680
    },
    {
      "dynamics_loss_avg": 0.09941343367099761,
      "epoch": 2.593994140625e-06,
      "step": 680
    },
    {
      "action_loss_avg": 0.07704295311123133,
      "epoch": 2.593994140625e-06,
      "step": 680
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 680,
      "training_step_time": 9.952084064483643
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 3.145570755004883,
      "step": 680
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 680,
      "training_step_time": 9.932789087295532
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 3.138410806655884,
      "step": 680
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 680,
      "training_step_time": 9.922520637512207
    },
    {
      "epoch": 2.593994140625e-06,
      "model_forward_time": 3.395165205001831,
      "step": 680
    },
    {
      "epoch": 2.593994140625e-06,
      "step": 680,
      "training_step_time": 10.298885583877563
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 3.1867334842681885,
      "step": 681
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 681,
      "training_step_time": 9.96253490447998
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 3.1484148502349854,
      "step": 681
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 681,
      "training_step_time": 9.929777383804321
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 3.144850730895996,
      "step": 681
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 681,
      "training_step_time": 9.933915853500366
    },
    {
      "epoch": 2.597808837890625e-06,
      "model_forward_time": 3.1419310569763184,
      "step": 681
    },
    {
      "epoch": 2.597808837890625e-06,
      "step": 681,
      "training_step_time": 10.021789789199829
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 3.422492742538452,
      "step": 682
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 682,
      "training_step_time": 10.203145980834961
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 3.1341874599456787,
      "step": 682
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 682,
      "training_step_time": 9.911264657974243
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 3.12812876701355,
      "step": 682
    },
    {
      "dynamics_loss_avg": 0.09300690144300461,
      "epoch": 2.60162353515625e-06,
      "step": 682
    },
    {
      "action_loss_avg": 0.05826256386935711,
      "epoch": 2.60162353515625e-06,
      "step": 682
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 682,
      "training_step_time": 9.91061544418335
    },
    {
      "epoch": 2.60162353515625e-06,
      "model_forward_time": 3.132389783859253,
      "step": 682
    },
    {
      "epoch": 2.60162353515625e-06,
      "step": 682,
      "training_step_time": 10.227277278900146
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 3.1600186824798584,
      "step": 683
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 683,
      "training_step_time": 9.944734811782837
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 3.4202213287353516,
      "step": 683
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 683,
      "training_step_time": 10.195546865463257
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 3.131110429763794,
      "step": 683
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 683,
      "training_step_time": 9.906583309173584
    },
    {
      "epoch": 2.605438232421875e-06,
      "model_forward_time": 3.1361584663391113,
      "step": 683
    },
    {
      "epoch": 2.605438232421875e-06,
      "step": 683,
      "training_step_time": 10.267493724822998
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 2.464141607284546,
      "step": 684
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 684,
      "training_step_time": 7.493158578872681
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 3.1213104724884033,
      "step": 684
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 684,
      "training_step_time": 9.89070200920105
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 3.386439561843872,
      "step": 684
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 684,
      "training_step_time": 10.16172480583191
    },
    {
      "epoch": 2.6092529296875e-06,
      "model_forward_time": 3.1298258304595947,
      "step": 684
    },
    {
      "epoch": 2.6092529296875e-06,
      "step": 684,
      "training_step_time": 10.014519214630127
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 3.1609933376312256,
      "step": 685
    },
    {
      "dynamics_loss_avg": 0.08578020893037319,
      "epoch": 2.613067626953125e-06,
      "step": 685
    },
    {
      "action_loss_avg": 0.05406912118196487,
      "epoch": 2.613067626953125e-06,
      "step": 685
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 685,
      "training_step_time": 9.939677238464355
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 3.131527900695801,
      "step": 685
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 685,
      "training_step_time": 9.90835976600647
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 3.130918025970459,
      "step": 685
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 685,
      "training_step_time": 9.906383752822876
    },
    {
      "epoch": 2.613067626953125e-06,
      "model_forward_time": 3.399850845336914,
      "step": 685
    },
    {
      "epoch": 2.613067626953125e-06,
      "step": 685,
      "training_step_time": 10.274281978607178
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 3.1741347312927246,
      "step": 686
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 686,
      "training_step_time": 9.954290866851807
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 3.1289849281311035,
      "step": 686
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 686,
      "training_step_time": 9.907285451889038
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 3.132207155227661,
      "step": 686
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 686,
      "training_step_time": 9.908985376358032
    },
    {
      "epoch": 2.61688232421875e-06,
      "model_forward_time": 3.129715919494629,
      "step": 686
    },
    {
      "epoch": 2.61688232421875e-06,
      "step": 686,
      "training_step_time": 10.303754568099976
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 3.431659698486328,
      "step": 687
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 687,
      "training_step_time": 10.204329490661621
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 3.1329762935638428,
      "step": 687
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 687,
      "training_step_time": 9.910897016525269
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 3.1370322704315186,
      "step": 687
    },
    {
      "dynamics_loss_avg": 0.0665479689836502,
      "epoch": 2.620697021484375e-06,
      "step": 687
    },
    {
      "action_loss_avg": 0.05561033058911562,
      "epoch": 2.620697021484375e-06,
      "step": 687
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 687,
      "training_step_time": 9.91817331314087
    },
    {
      "epoch": 2.620697021484375e-06,
      "model_forward_time": 3.1425490379333496,
      "step": 687
    },
    {
      "epoch": 2.620697021484375e-06,
      "step": 687,
      "training_step_time": 10.242323875427246
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 3.1713294982910156,
      "step": 688
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 688,
      "training_step_time": 9.947091817855835
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 3.4065070152282715,
      "step": 688
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 688,
      "training_step_time": 10.180883884429932
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 3.140589475631714,
      "step": 688
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 688,
      "training_step_time": 9.917398452758789
    },
    {
      "epoch": 2.62451171875e-06,
      "model_forward_time": 3.144456386566162,
      "step": 688
    },
    {
      "epoch": 2.62451171875e-06,
      "step": 688,
      "training_step_time": 10.297824621200562
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 3.169156551361084,
      "step": 689
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 689,
      "training_step_time": 9.938483715057373
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 3.1399717330932617,
      "step": 689
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 689,
      "training_step_time": 9.913979768753052
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 3.4443273544311523,
      "step": 689
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 689,
      "training_step_time": 10.217796087265015
    },
    {
      "epoch": 2.628326416015625e-06,
      "model_forward_time": 3.138411283493042,
      "step": 689
    },
    {
      "epoch": 2.628326416015625e-06,
      "step": 689,
      "training_step_time": 10.016119480133057
    },
    {
      "epoch": 2.63214111328125e-06,
      "grad_norm": 1.0527727603912354,
      "learning_rate": 1e-05,
      "loss": 0.1446,
      "step": 690
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 3.1886069774627686,
      "step": 690
    },
    {
      "dynamics_loss_avg": 0.07844499126076698,
      "epoch": 2.63214111328125e-06,
      "step": 690
    },
    {
      "action_loss_avg": 0.056786900572478774,
      "epoch": 2.63214111328125e-06,
      "step": 690
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 690,
      "training_step_time": 9.971095323562622
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 3.142014503479004,
      "step": 690
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 690,
      "training_step_time": 9.914602279663086
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 3.1399505138397217,
      "step": 690
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 690,
      "training_step_time": 9.921558618545532
    },
    {
      "epoch": 2.63214111328125e-06,
      "model_forward_time": 3.4002017974853516,
      "step": 690
    },
    {
      "epoch": 2.63214111328125e-06,
      "step": 690,
      "training_step_time": 10.269984722137451
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 3.1905808448791504,
      "step": 691
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 691,
      "training_step_time": 9.969357252120972
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 3.140596628189087,
      "step": 691
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 691,
      "training_step_time": 9.917305946350098
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 3.1379904747009277,
      "step": 691
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 691,
      "training_step_time": 9.920026063919067
    },
    {
      "epoch": 2.635955810546875e-06,
      "model_forward_time": 3.1419334411621094,
      "step": 691
    },
    {
      "epoch": 2.635955810546875e-06,
      "step": 691,
      "training_step_time": 10.305197477340698
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 3.4399805068969727,
      "step": 692
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 692,
      "training_step_time": 10.223780870437622
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 3.1423892974853516,
      "step": 692
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 692,
      "training_step_time": 9.91608476638794
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 3.1379947662353516,
      "step": 692
    },
    {
      "dynamics_loss_avg": 0.07801496535539627,
      "epoch": 2.6397705078125e-06,
      "step": 692
    },
    {
      "action_loss_avg": 0.06025743372738361,
      "epoch": 2.6397705078125e-06,
      "step": 692
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 692,
      "training_step_time": 9.916321516036987
    },
    {
      "epoch": 2.6397705078125e-06,
      "model_forward_time": 3.138906240463257,
      "step": 692
    },
    {
      "epoch": 2.6397705078125e-06,
      "step": 692,
      "training_step_time": 10.28651738166809
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 3.1763904094696045,
      "step": 693
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 693,
      "training_step_time": 9.94924807548523
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 3.412193775177002,
      "step": 693
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 693,
      "training_step_time": 10.186568021774292
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 3.133192300796509,
      "step": 693
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 693,
      "training_step_time": 9.910247325897217
    },
    {
      "epoch": 2.643585205078125e-06,
      "model_forward_time": 3.1340136528015137,
      "step": 693
    },
    {
      "epoch": 2.643585205078125e-06,
      "step": 693,
      "training_step_time": 10.268843650817871
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 3.174488067626953,
      "step": 694
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 694,
      "training_step_time": 9.955915451049805
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 3.134165048599243,
      "step": 694
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 694,
      "training_step_time": 9.914923429489136
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 3.136611223220825,
      "step": 694
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 694,
      "training_step_time": 10.201872825622559
    },
    {
      "epoch": 2.64739990234375e-06,
      "model_forward_time": 3.134408473968506,
      "step": 694
    },
    {
      "epoch": 2.64739990234375e-06,
      "step": 694,
      "training_step_time": 10.288394689559937
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 3.1698362827301025,
      "step": 695
    },
    {
      "dynamics_loss_avg": 0.08518849276006221,
      "epoch": 2.651214599609375e-06,
      "step": 695
    },
    {
      "action_loss_avg": 0.06438527181744576,
      "epoch": 2.651214599609375e-06,
      "step": 695
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 695,
      "training_step_time": 9.945703744888306
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 3.131242275238037,
      "step": 695
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 695,
      "training_step_time": 9.90181016921997
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 3.131267786026001,
      "step": 695
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 695,
      "training_step_time": 9.90648102760315
    },
    {
      "epoch": 2.651214599609375e-06,
      "model_forward_time": 3.444675922393799,
      "step": 695
    },
    {
      "epoch": 2.651214599609375e-06,
      "step": 695,
      "training_step_time": 10.313140392303467
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 3.1643898487091064,
      "step": 696
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 696,
      "training_step_time": 9.938286304473877
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 3.137324810028076,
      "step": 696
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 696,
      "training_step_time": 9.908586263656616
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 3.1428568363189697,
      "step": 696
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 696,
      "training_step_time": 9.90813684463501
    },
    {
      "epoch": 2.655029296875e-06,
      "model_forward_time": 3.1366004943847656,
      "step": 696
    },
    {
      "epoch": 2.655029296875e-06,
      "step": 696,
      "training_step_time": 10.268518924713135
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 3.167269229888916,
      "step": 697
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 697,
      "training_step_time": 10.19743037223816
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 2.435302495956421,
      "step": 697
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 697,
      "training_step_time": 7.457866668701172
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 3.1367921829223633,
      "step": 697
    },
    {
      "dynamics_loss_avg": 0.0956252671778202,
      "epoch": 2.658843994140625e-06,
      "step": 697
    },
    {
      "action_loss_avg": 0.07165191750973463,
      "epoch": 2.658843994140625e-06,
      "step": 697
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 697,
      "training_step_time": 9.897936344146729
    },
    {
      "epoch": 2.658843994140625e-06,
      "model_forward_time": 3.1358695030212402,
      "step": 697
    },
    {
      "epoch": 2.658843994140625e-06,
      "step": 697,
      "training_step_time": 10.007218837738037
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 2.47117280960083,
      "step": 698
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 698,
      "training_step_time": 7.492719888687134
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 3.1354639530181885,
      "step": 698
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 698,
      "training_step_time": 10.148090839385986
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 3.145841360092163,
      "step": 698
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 698,
      "training_step_time": 9.912047147750854
    },
    {
      "epoch": 2.66265869140625e-06,
      "model_forward_time": 3.1326510906219482,
      "step": 698
    },
    {
      "epoch": 2.66265869140625e-06,
      "step": 698,
      "training_step_time": 10.297685861587524
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 3.1840035915374756,
      "step": 699
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 699,
      "training_step_time": 9.948652267456055
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 3.1392247676849365,
      "step": 699
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 699,
      "training_step_time": 9.905166149139404
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 3.1388614177703857,
      "step": 699
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 699,
      "training_step_time": 10.17430067062378
    },
    {
      "epoch": 2.666473388671875e-06,
      "model_forward_time": 3.140904664993286,
      "step": 699
    },
    {
      "epoch": 2.666473388671875e-06,
      "step": 699,
      "training_step_time": 10.312726020812988
    },
    {
      "epoch": 2.6702880859375e-06,
      "grad_norm": 1.1937710046768188,
      "learning_rate": 1e-05,
      "loss": 0.1497,
      "step": 700
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 3.191155433654785,
      "step": 700
    },
    {
      "dynamics_loss_avg": 0.07780849859118462,
      "epoch": 2.6702880859375e-06,
      "step": 700
    },
    {
      "action_loss_avg": 0.06671853468287736,
      "epoch": 2.6702880859375e-06,
      "step": 700
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 700,
      "training_step_time": 9.961926221847534
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 3.135291576385498,
      "step": 700
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 700,
      "training_step_time": 9.913282632827759
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 3.1367104053497314,
      "step": 700
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 700,
      "training_step_time": 9.913126707077026
    },
    {
      "epoch": 2.6702880859375e-06,
      "model_forward_time": 3.431715726852417,
      "step": 700
    },
    {
      "epoch": 2.6702880859375e-06,
      "step": 700,
      "training_step_time": 10.297358751296997
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 3.171077013015747,
      "step": 701
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 701,
      "training_step_time": 9.94586706161499
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 3.138422727584839,
      "step": 701
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 701,
      "training_step_time": 9.9125235080719
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 3.138960361480713,
      "step": 701
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 701,
      "training_step_time": 9.908896207809448
    },
    {
      "epoch": 2.674102783203125e-06,
      "model_forward_time": 3.138619899749756,
      "step": 701
    },
    {
      "epoch": 2.674102783203125e-06,
      "step": 701,
      "training_step_time": 10.296870708465576
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 3.413297653198242,
      "step": 702
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 702,
      "training_step_time": 10.18179202079773
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 3.135932683944702,
      "step": 702
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 702,
      "training_step_time": 9.896848678588867
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 3.1335129737854004,
      "step": 702
    },
    {
      "dynamics_loss_avg": 0.08331412561237812,
      "epoch": 2.67791748046875e-06,
      "step": 702
    },
    {
      "action_loss_avg": 0.05935092307627201,
      "epoch": 2.67791748046875e-06,
      "step": 702
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 702,
      "training_step_time": 9.90263032913208
    },
    {
      "epoch": 2.67791748046875e-06,
      "model_forward_time": 3.1343820095062256,
      "step": 702
    },
    {
      "epoch": 2.67791748046875e-06,
      "step": 702,
      "training_step_time": 10.008829355239868
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 3.174076795578003,
      "step": 703
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 703,
      "training_step_time": 9.939857721328735
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 3.131561279296875,
      "step": 703
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 703,
      "training_step_time": 10.176569700241089
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 3.1375339031219482,
      "step": 703
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 703,
      "training_step_time": 9.901731967926025
    },
    {
      "epoch": 2.681732177734375e-06,
      "model_forward_time": 3.1384572982788086,
      "step": 703
    },
    {
      "epoch": 2.681732177734375e-06,
      "step": 703,
      "training_step_time": 10.284624099731445
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 3.1871378421783447,
      "step": 704
    },
    {
      "epoch": 2.685546875e-06,
      "step": 704,
      "training_step_time": 9.955462217330933
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 3.134310483932495,
      "step": 704
    },
    {
      "epoch": 2.685546875e-06,
      "step": 704,
      "training_step_time": 9.901492595672607
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 3.1392440795898438,
      "step": 704
    },
    {
      "epoch": 2.685546875e-06,
      "step": 704,
      "training_step_time": 10.199025869369507
    },
    {
      "epoch": 2.685546875e-06,
      "model_forward_time": 3.140601396560669,
      "step": 704
    },
    {
      "epoch": 2.685546875e-06,
      "step": 704,
      "training_step_time": 10.293036937713623
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 2.4831345081329346,
      "step": 705
    },
    {
      "dynamics_loss_avg": 0.08992312606424094,
      "epoch": 2.689361572265625e-06,
      "step": 705
    },
    {
      "action_loss_avg": 0.06904308795928955,
      "epoch": 2.689361572265625e-06,
      "step": 705
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 705,
      "training_step_time": 7.505573272705078
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 3.1307239532470703,
      "step": 705
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 705,
      "training_step_time": 9.899402141571045
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 3.1376395225524902,
      "step": 705
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 705,
      "training_step_time": 9.909911394119263
    },
    {
      "epoch": 2.689361572265625e-06,
      "model_forward_time": 3.1397736072540283,
      "step": 705
    },
    {
      "epoch": 2.689361572265625e-06,
      "step": 705,
      "training_step_time": 10.274373054504395
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 3.183528184890747,
      "step": 706
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 706,
      "training_step_time": 9.958628416061401
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 3.1420295238494873,
      "step": 706
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 706,
      "training_step_time": 9.9180006980896
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 3.146275043487549,
      "step": 706
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 706,
      "training_step_time": 9.9224271774292
    },
    {
      "epoch": 2.69317626953125e-06,
      "model_forward_time": 3.1412241458892822,
      "step": 706
    },
    {
      "epoch": 2.69317626953125e-06,
      "step": 706,
      "training_step_time": 10.030451774597168
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 3.4443650245666504,
      "step": 707
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 707,
      "training_step_time": 10.216094017028809
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 3.1398825645446777,
      "step": 707
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 707,
      "training_step_time": 9.907261610031128
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 3.140108108520508,
      "step": 707
    },
    {
      "dynamics_loss_avg": 0.1038014680147171,
      "epoch": 2.696990966796875e-06,
      "step": 707
    },
    {
      "action_loss_avg": 0.07165078595280647,
      "epoch": 2.696990966796875e-06,
      "step": 707
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 707,
      "training_step_time": 9.909738779067993
    },
    {
      "epoch": 2.696990966796875e-06,
      "model_forward_time": 3.135983943939209,
      "step": 707
    },
    {
      "epoch": 2.696990966796875e-06,
      "step": 707,
      "training_step_time": 10.32046365737915
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 2.4595675468444824,
      "step": 708
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 708,
      "training_step_time": 7.4839160442352295
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 3.1306841373443604,
      "step": 708
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 708,
      "training_step_time": 10.203892707824707
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 3.1338260173797607,
      "step": 708
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 708,
      "training_step_time": 9.901130676269531
    },
    {
      "epoch": 2.7008056640625e-06,
      "model_forward_time": 3.1308231353759766,
      "step": 708
    },
    {
      "epoch": 2.7008056640625e-06,
      "step": 708,
      "training_step_time": 10.25869369506836
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 3.161356210708618,
      "step": 709
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 709,
      "training_step_time": 9.928165435791016
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 3.132183790206909,
      "step": 709
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 709,
      "training_step_time": 9.900002479553223
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 3.4146151542663574,
      "step": 709
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 709,
      "training_step_time": 10.185303926467896
    },
    {
      "epoch": 2.704620361328125e-06,
      "model_forward_time": 3.1354775428771973,
      "step": 709
    },
    {
      "epoch": 2.704620361328125e-06,
      "step": 709,
      "training_step_time": 10.284301519393921
    },
    {
      "epoch": 2.70843505859375e-06,
      "grad_norm": 0.6807244420051575,
      "learning_rate": 1e-05,
      "loss": 0.1526,
      "step": 710
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 3.156414270401001,
      "step": 710
    },
    {
      "dynamics_loss_avg": 0.10533167608082294,
      "epoch": 2.70843505859375e-06,
      "step": 710
    },
    {
      "action_loss_avg": 0.08994316644966602,
      "epoch": 2.70843505859375e-06,
      "step": 710
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 710,
      "training_step_time": 9.928692817687988
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 3.134002685546875,
      "step": 710
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 710,
      "training_step_time": 9.906420946121216
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 3.1327478885650635,
      "step": 710
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 710,
      "training_step_time": 9.9010009765625
    },
    {
      "epoch": 2.70843505859375e-06,
      "model_forward_time": 3.3891217708587646,
      "step": 710
    },
    {
      "epoch": 2.70843505859375e-06,
      "step": 710,
      "training_step_time": 10.256458520889282
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 2.4626739025115967,
      "step": 711
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 711,
      "training_step_time": 7.493875503540039
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 3.1291069984436035,
      "step": 711
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 711,
      "training_step_time": 9.900663375854492
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 3.1331844329833984,
      "step": 711
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 711,
      "training_step_time": 9.907363653182983
    },
    {
      "epoch": 2.712249755859375e-06,
      "model_forward_time": 3.132376194000244,
      "step": 711
    },
    {
      "epoch": 2.712249755859375e-06,
      "step": 711,
      "training_step_time": 10.026710510253906
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 3.4420218467712402,
      "step": 712
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 712,
      "training_step_time": 10.218875646591187
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 3.147397994995117,
      "step": 712
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 712,
      "training_step_time": 9.926516771316528
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 3.1490285396575928,
      "step": 712
    },
    {
      "dynamics_loss_avg": 0.09459789544343948,
      "epoch": 2.716064453125e-06,
      "step": 712
    },
    {
      "action_loss_avg": 0.062368198111653325,
      "epoch": 2.716064453125e-06,
      "step": 712
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 712,
      "training_step_time": 9.929348230361938
    },
    {
      "epoch": 2.716064453125e-06,
      "model_forward_time": 3.143084764480591,
      "step": 712
    },
    {
      "epoch": 2.716064453125e-06,
      "step": 712,
      "training_step_time": 10.327887535095215
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 3.1960887908935547,
      "step": 713
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 713,
      "training_step_time": 9.98047661781311
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 3.1421639919281006,
      "step": 713
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 713,
      "training_step_time": 10.210787057876587
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 3.144541025161743,
      "step": 713
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 713,
      "training_step_time": 9.92159390449524
    },
    {
      "epoch": 2.719879150390625e-06,
      "model_forward_time": 3.141746759414673,
      "step": 713
    },
    {
      "epoch": 2.719879150390625e-06,
      "step": 713,
      "training_step_time": 10.28035306930542
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 3.187976360321045,
      "step": 714
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 714,
      "training_step_time": 9.958515882492065
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 3.142585277557373,
      "step": 714
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 714,
      "training_step_time": 9.91653823852539
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 3.1470413208007812,
      "step": 714
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 714,
      "training_step_time": 10.20054841041565
    },
    {
      "epoch": 2.72369384765625e-06,
      "model_forward_time": 3.1506922245025635,
      "step": 714
    },
    {
      "epoch": 2.72369384765625e-06,
      "step": 714,
      "training_step_time": 10.28169584274292
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 3.200807571411133,
      "step": 715
    },
    {
      "dynamics_loss_avg": 0.09916669279336929,
      "epoch": 2.727508544921875e-06,
      "step": 715
    },
    {
      "action_loss_avg": 0.08037935309112072,
      "epoch": 2.727508544921875e-06,
      "step": 715
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 715,
      "training_step_time": 9.98697805404663
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 3.1429941654205322,
      "step": 715
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 715,
      "training_step_time": 9.926137685775757
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 3.1479766368865967,
      "step": 715
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 715,
      "training_step_time": 9.926488399505615
    },
    {
      "epoch": 2.727508544921875e-06,
      "model_forward_time": 3.42463755607605,
      "step": 715
    },
    {
      "epoch": 2.727508544921875e-06,
      "step": 715,
      "training_step_time": 10.297865629196167
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 3.175082206726074,
      "step": 716
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 716,
      "training_step_time": 9.951899290084839
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 3.1412041187286377,
      "step": 716
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 716,
      "training_step_time": 9.91299033164978
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 3.137511968612671,
      "step": 716
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 716,
      "training_step_time": 9.907695293426514
    },
    {
      "epoch": 2.7313232421875e-06,
      "model_forward_time": 3.139991521835327,
      "step": 716
    },
    {
      "epoch": 2.7313232421875e-06,
      "step": 716,
      "training_step_time": 10.26873230934143
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 3.167189121246338,
      "step": 717
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 717,
      "training_step_time": 10.233263969421387
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 3.1384031772613525,
      "step": 717
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 717,
      "training_step_time": 9.908096551895142
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 3.139540910720825,
      "step": 717
    },
    {
      "dynamics_loss_avg": 0.09313692525029182,
      "epoch": 2.735137939453125e-06,
      "step": 717
    },
    {
      "action_loss_avg": 0.062239740416407584,
      "epoch": 2.735137939453125e-06,
      "step": 717
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 717,
      "training_step_time": 9.915364742279053
    },
    {
      "epoch": 2.735137939453125e-06,
      "model_forward_time": 3.136289119720459,
      "step": 717
    },
    {
      "epoch": 2.735137939453125e-06,
      "step": 717,
      "training_step_time": 10.306060075759888
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 3.1714816093444824,
      "step": 718
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 718,
      "training_step_time": 9.947589874267578
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 3.4033825397491455,
      "step": 718
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 718,
      "training_step_time": 10.173515319824219
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 3.14367938041687,
      "step": 718
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 718,
      "training_step_time": 9.914849996566772
    },
    {
      "epoch": 2.73895263671875e-06,
      "model_forward_time": 3.1409475803375244,
      "step": 718
    },
    {
      "epoch": 2.73895263671875e-06,
      "step": 718,
      "training_step_time": 10.310677528381348
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 2.4785873889923096,
      "step": 719
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 719,
      "training_step_time": 7.510334491729736
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 3.1392197608947754,
      "step": 719
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 719,
      "training_step_time": 9.914533138275146
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 3.1503701210021973,
      "step": 719
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 719,
      "training_step_time": 10.192444801330566
    },
    {
      "epoch": 2.742767333984375e-06,
      "model_forward_time": 2.4359490871429443,
      "step": 719
    },
    {
      "epoch": 2.742767333984375e-06,
      "step": 719,
      "training_step_time": 10.284265279769897
    },
    {
      "epoch": 2.74658203125e-06,
      "grad_norm": 0.8262678384780884,
      "learning_rate": 1e-05,
      "loss": 0.1426,
      "step": 720
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 3.173942804336548,
      "step": 720
    },
    {
      "dynamics_loss_avg": 0.08414000403136015,
      "epoch": 2.74658203125e-06,
      "step": 720
    },
    {
      "action_loss_avg": 0.059913348592817786,
      "epoch": 2.74658203125e-06,
      "step": 720
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 720,
      "training_step_time": 9.951704025268555
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 3.1425788402557373,
      "step": 720
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 720,
      "training_step_time": 9.9140305519104
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 3.1412479877471924,
      "step": 720
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 720,
      "training_step_time": 9.916937828063965
    },
    {
      "epoch": 2.74658203125e-06,
      "model_forward_time": 3.415823221206665,
      "step": 720
    },
    {
      "epoch": 2.74658203125e-06,
      "step": 720,
      "training_step_time": 10.292906045913696
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 3.1810462474823,
      "step": 721
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 721,
      "training_step_time": 9.954593658447266
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 3.142456531524658,
      "step": 721
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 721,
      "training_step_time": 9.914796829223633
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 3.139488935470581,
      "step": 721
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 721,
      "training_step_time": 9.916478633880615
    },
    {
      "epoch": 2.750396728515625e-06,
      "model_forward_time": 3.1463167667388916,
      "step": 721
    },
    {
      "epoch": 2.750396728515625e-06,
      "step": 721,
      "training_step_time": 10.2416353225708
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 3.18386173248291,
      "step": 722
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 722,
      "training_step_time": 10.252956628799438
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 3.1460516452789307,
      "step": 722
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 722,
      "training_step_time": 9.920104742050171
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 3.1409897804260254,
      "step": 722
    },
    {
      "dynamics_loss_avg": 0.09791750535368919,
      "epoch": 2.75421142578125e-06,
      "step": 722
    },
    {
      "action_loss_avg": 0.06975062601268292,
      "epoch": 2.75421142578125e-06,
      "step": 722
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 722,
      "training_step_time": 9.918615818023682
    },
    {
      "epoch": 2.75421142578125e-06,
      "model_forward_time": 3.1375153064727783,
      "step": 722
    },
    {
      "epoch": 2.75421142578125e-06,
      "step": 722,
      "training_step_time": 10.014318466186523
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 3.186507225036621,
      "step": 723
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 723,
      "training_step_time": 9.961096286773682
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 3.4189648628234863,
      "step": 723
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 723,
      "training_step_time": 10.190850973129272
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 3.1390953063964844,
      "step": 723
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 723,
      "training_step_time": 9.914458274841309
    },
    {
      "epoch": 2.758026123046875e-06,
      "model_forward_time": 3.1418728828430176,
      "step": 723
    },
    {
      "epoch": 2.758026123046875e-06,
      "step": 723,
      "training_step_time": 10.32740330696106
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 3.176717519760132,
      "step": 724
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 724,
      "training_step_time": 9.95151686668396
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 3.1376333236694336,
      "step": 724
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 724,
      "training_step_time": 9.911842584609985
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 3.3952817916870117,
      "step": 724
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 724,
      "training_step_time": 10.16720986366272
    },
    {
      "epoch": 2.7618408203125e-06,
      "model_forward_time": 3.1417672634124756,
      "step": 724
    },
    {
      "epoch": 2.7618408203125e-06,
      "step": 724,
      "training_step_time": 10.288413286209106
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 3.1900217533111572,
      "step": 725
    },
    {
      "dynamics_loss_avg": 0.09064544327557086,
      "epoch": 2.765655517578125e-06,
      "step": 725
    },
    {
      "action_loss_avg": 0.07070836499333381,
      "epoch": 2.765655517578125e-06,
      "step": 725
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 725,
      "training_step_time": 9.968004703521729
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 3.140169620513916,
      "step": 725
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 725,
      "training_step_time": 9.913171291351318
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 3.1439754962921143,
      "step": 725
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 725,
      "training_step_time": 9.918276309967041
    },
    {
      "epoch": 2.765655517578125e-06,
      "model_forward_time": 3.4575092792510986,
      "step": 725
    },
    {
      "epoch": 2.765655517578125e-06,
      "step": 725,
      "training_step_time": 10.320602178573608
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 3.1712300777435303,
      "step": 726
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 726,
      "training_step_time": 9.948097705841064
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 3.1384530067443848,
      "step": 726
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 726,
      "training_step_time": 9.910869598388672
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 3.145085096359253,
      "step": 726
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 726,
      "training_step_time": 9.918987512588501
    },
    {
      "epoch": 2.76947021484375e-06,
      "model_forward_time": 3.143017530441284,
      "step": 726
    },
    {
      "epoch": 2.76947021484375e-06,
      "step": 726,
      "training_step_time": 10.253361225128174
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 3.4520986080169678,
      "step": 727
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 727,
      "training_step_time": 10.23095154762268
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 3.1448166370391846,
      "step": 727
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 727,
      "training_step_time": 9.920166254043579
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 3.1459100246429443,
      "step": 727
    },
    {
      "dynamics_loss_avg": 0.08990479782223701,
      "epoch": 2.773284912109375e-06,
      "step": 727
    },
    {
      "action_loss_avg": 0.05927510685287416,
      "epoch": 2.773284912109375e-06,
      "step": 727
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 727,
      "training_step_time": 9.923900842666626
    },
    {
      "epoch": 2.773284912109375e-06,
      "model_forward_time": 3.1445529460906982,
      "step": 727
    },
    {
      "epoch": 2.773284912109375e-06,
      "step": 727,
      "training_step_time": 10.020751953125
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 3.174288749694824,
      "step": 728
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 728,
      "training_step_time": 9.950673580169678
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 3.4055635929107666,
      "step": 728
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 728,
      "training_step_time": 10.176868438720703
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 3.147625207901001,
      "step": 728
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 728,
      "training_step_time": 9.921915292739868
    },
    {
      "epoch": 2.777099609375e-06,
      "model_forward_time": 3.137788772583008,
      "step": 728
    },
    {
      "epoch": 2.777099609375e-06,
      "step": 728,
      "training_step_time": 10.005545377731323
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 3.1741929054260254,
      "step": 729
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 729,
      "training_step_time": 9.941041946411133
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 3.135796070098877,
      "step": 729
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 729,
      "training_step_time": 9.912872314453125
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 3.4074904918670654,
      "step": 729
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 729,
      "training_step_time": 10.173276424407959
    },
    {
      "epoch": 2.780914306640625e-06,
      "model_forward_time": 3.130539655685425,
      "step": 729
    },
    {
      "epoch": 2.780914306640625e-06,
      "step": 729,
      "training_step_time": 10.285624027252197
    },
    {
      "epoch": 2.78472900390625e-06,
      "grad_norm": 0.6317223310470581,
      "learning_rate": 1e-05,
      "loss": 0.1523,
      "step": 730
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 3.176283121109009,
      "step": 730
    },
    {
      "dynamics_loss_avg": 0.08866557218134404,
      "epoch": 2.78472900390625e-06,
      "step": 730
    },
    {
      "action_loss_avg": 0.05957840420305729,
      "epoch": 2.78472900390625e-06,
      "step": 730
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 730,
      "training_step_time": 9.946890115737915
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 3.1333351135253906,
      "step": 730
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 730,
      "training_step_time": 9.905197143554688
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 3.135678291320801,
      "step": 730
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 730,
      "training_step_time": 9.91117525100708
    },
    {
      "epoch": 2.78472900390625e-06,
      "model_forward_time": 3.436291456222534,
      "step": 730
    },
    {
      "epoch": 2.78472900390625e-06,
      "step": 730,
      "training_step_time": 10.308412313461304
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 3.181093692779541,
      "step": 731
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 731,
      "training_step_time": 9.954393863677979
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 3.138031005859375,
      "step": 731
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 731,
      "training_step_time": 9.912794351577759
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 3.1381587982177734,
      "step": 731
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 731,
      "training_step_time": 9.91300630569458
    },
    {
      "epoch": 2.788543701171875e-06,
      "model_forward_time": 3.1355881690979004,
      "step": 731
    },
    {
      "epoch": 2.788543701171875e-06,
      "step": 731,
      "training_step_time": 10.25002145767212
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 3.4128851890563965,
      "step": 732
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 732,
      "training_step_time": 10.18432354927063
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 3.137652635574341,
      "step": 732
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 732,
      "training_step_time": 9.903037548065186
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 3.1285152435302734,
      "step": 732
    },
    {
      "dynamics_loss_avg": 0.08276371583342552,
      "epoch": 2.7923583984375e-06,
      "step": 732
    },
    {
      "action_loss_avg": 0.07770373411476612,
      "epoch": 2.7923583984375e-06,
      "step": 732
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 732,
      "training_step_time": 9.896731615066528
    },
    {
      "epoch": 2.7923583984375e-06,
      "model_forward_time": 2.423459053039551,
      "step": 732
    },
    {
      "epoch": 2.7923583984375e-06,
      "step": 732,
      "training_step_time": 9.997321605682373
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 3.1691956520080566,
      "step": 733
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 733,
      "training_step_time": 9.935966491699219
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 2.426532745361328,
      "step": 733
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 733,
      "training_step_time": 7.4421467781066895
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 3.3800241947174072,
      "step": 733
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 733,
      "training_step_time": 10.152076959609985
    },
    {
      "epoch": 2.796173095703125e-06,
      "model_forward_time": 3.129615545272827,
      "step": 733
    },
    {
      "epoch": 2.796173095703125e-06,
      "step": 733,
      "training_step_time": 10.24523377418518
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 3.1821210384368896,
      "step": 734
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 734,
      "training_step_time": 9.949622869491577
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 3.1304643154144287,
      "step": 734
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 734,
      "training_step_time": 9.897819519042969
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 3.1345980167388916,
      "step": 734
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 734,
      "training_step_time": 10.218894958496094
    },
    {
      "epoch": 2.79998779296875e-06,
      "model_forward_time": 3.1371281147003174,
      "step": 734
    },
    {
      "epoch": 2.79998779296875e-06,
      "step": 734,
      "training_step_time": 10.3263099193573
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 3.175671339035034,
      "step": 735
    },
    {
      "dynamics_loss_avg": 0.06717264726758003,
      "epoch": 2.803802490234375e-06,
      "step": 735
    },
    {
      "action_loss_avg": 0.045558747742325066,
      "epoch": 2.803802490234375e-06,
      "step": 735
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 735,
      "training_step_time": 9.958040237426758
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 3.139885187149048,
      "step": 735
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 735,
      "training_step_time": 9.919831991195679
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 3.1377041339874268,
      "step": 735
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 735,
      "training_step_time": 9.9206383228302
    },
    {
      "epoch": 2.803802490234375e-06,
      "model_forward_time": 3.422384262084961,
      "step": 735
    },
    {
      "epoch": 2.803802490234375e-06,
      "step": 735,
      "training_step_time": 10.28907561302185
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 3.17958402633667,
      "step": 736
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 736,
      "training_step_time": 9.956248044967651
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 3.1302168369293213,
      "step": 736
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 736,
      "training_step_time": 9.905218124389648
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 3.134535551071167,
      "step": 736
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 736,
      "training_step_time": 9.907516717910767
    },
    {
      "epoch": 2.8076171875e-06,
      "model_forward_time": 3.133664608001709,
      "step": 736
    },
    {
      "epoch": 2.8076171875e-06,
      "step": 736,
      "training_step_time": 10.262497425079346
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 3.180877923965454,
      "step": 737
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 737,
      "training_step_time": 10.223237991333008
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 3.1346073150634766,
      "step": 737
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 737,
      "training_step_time": 9.904624223709106
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 3.1349411010742188,
      "step": 737
    },
    {
      "dynamics_loss_avg": 0.09825568944215775,
      "epoch": 2.811431884765625e-06,
      "step": 737
    },
    {
      "action_loss_avg": 0.06364327371120453,
      "epoch": 2.811431884765625e-06,
      "step": 737
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 737,
      "training_step_time": 9.908820152282715
    },
    {
      "epoch": 2.811431884765625e-06,
      "model_forward_time": 3.139674425125122,
      "step": 737
    },
    {
      "epoch": 2.811431884765625e-06,
      "step": 737,
      "training_step_time": 10.024368047714233
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 3.1652398109436035,
      "step": 738
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 738,
      "training_step_time": 9.94051718711853
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 3.3844990730285645,
      "step": 738
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 738,
      "training_step_time": 10.159873247146606
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 3.1356382369995117,
      "step": 738
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 738,
      "training_step_time": 9.90953779220581
    },
    {
      "epoch": 2.81524658203125e-06,
      "model_forward_time": 3.133575201034546,
      "step": 738
    },
    {
      "epoch": 2.81524658203125e-06,
      "step": 738,
      "training_step_time": 10.297350645065308
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 3.1834418773651123,
      "step": 739
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 739,
      "training_step_time": 9.962478160858154
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 3.1395697593688965,
      "step": 739
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 739,
      "training_step_time": 9.915069103240967
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 3.4572954177856445,
      "step": 739
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 739,
      "training_step_time": 10.231037139892578
    },
    {
      "epoch": 2.819061279296875e-06,
      "model_forward_time": 3.138446092605591,
      "step": 739
    },
    {
      "epoch": 2.819061279296875e-06,
      "step": 739,
      "training_step_time": 10.286386966705322
    },
    {
      "epoch": 2.8228759765625e-06,
      "grad_norm": 0.47905248403549194,
      "learning_rate": 1e-05,
      "loss": 0.1401,
      "step": 740
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 3.1920366287231445,
      "step": 740
    },
    {
      "dynamics_loss_avg": 0.07475603669881821,
      "epoch": 2.8228759765625e-06,
      "step": 740
    },
    {
      "action_loss_avg": 0.04950989969074726,
      "epoch": 2.8228759765625e-06,
      "step": 740
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 740,
      "training_step_time": 9.973376035690308
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 3.142787218093872,
      "step": 740
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 740,
      "training_step_time": 9.918901681900024
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 3.1421806812286377,
      "step": 740
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 740,
      "training_step_time": 9.920081377029419
    },
    {
      "epoch": 2.8228759765625e-06,
      "model_forward_time": 3.3947112560272217,
      "step": 740
    },
    {
      "epoch": 2.8228759765625e-06,
      "step": 740,
      "training_step_time": 10.283093929290771
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 3.1799943447113037,
      "step": 741
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 741,
      "training_step_time": 9.953386545181274
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 3.144132614135742,
      "step": 741
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 741,
      "training_step_time": 9.91993498802185
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 3.1427836418151855,
      "step": 741
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 741,
      "training_step_time": 9.920054912567139
    },
    {
      "epoch": 2.826690673828125e-06,
      "model_forward_time": 3.149768590927124,
      "step": 741
    },
    {
      "epoch": 2.826690673828125e-06,
      "step": 741,
      "training_step_time": 10.271143913269043
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 3.441657781600952,
      "step": 742
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 742,
      "training_step_time": 10.217564821243286
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 3.1487717628479004,
      "step": 742
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 742,
      "training_step_time": 9.926624059677124
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 2.439588785171509,
      "step": 742
    },
    {
      "dynamics_loss_avg": 0.07400593310594558,
      "epoch": 2.83050537109375e-06,
      "step": 742
    },
    {
      "action_loss_avg": 0.049853449268266556,
      "epoch": 2.83050537109375e-06,
      "step": 742
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 742,
      "training_step_time": 7.474742650985718
    },
    {
      "epoch": 2.83050537109375e-06,
      "model_forward_time": 3.140535831451416,
      "step": 742
    },
    {
      "epoch": 2.83050537109375e-06,
      "step": 742,
      "training_step_time": 10.027138710021973
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 3.1832523345947266,
      "step": 743
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 743,
      "training_step_time": 9.96113920211792
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 3.1432738304138184,
      "step": 743
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 743,
      "training_step_time": 10.227123737335205
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 3.1414027214050293,
      "step": 743
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 743,
      "training_step_time": 9.9167640209198
    },
    {
      "epoch": 2.834320068359375e-06,
      "model_forward_time": 3.145051956176758,
      "step": 743
    },
    {
      "epoch": 2.834320068359375e-06,
      "step": 743,
      "training_step_time": 10.284937858581543
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 3.1633799076080322,
      "step": 744
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 744,
      "training_step_time": 9.942232370376587
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 3.143935203552246,
      "step": 744
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 744,
      "training_step_time": 9.922153234481812
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 3.408494234085083,
      "step": 744
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 744,
      "training_step_time": 10.179070234298706
    },
    {
      "epoch": 2.838134765625e-06,
      "model_forward_time": 3.150160789489746,
      "step": 744
    },
    {
      "epoch": 2.838134765625e-06,
      "step": 744,
      "training_step_time": 10.015866041183472
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 3.170013904571533,
      "step": 745
    },
    {
      "dynamics_loss_avg": 0.09470995999872685,
      "epoch": 2.841949462890625e-06,
      "step": 745
    },
    {
      "action_loss_avg": 0.060736559331417084,
      "epoch": 2.841949462890625e-06,
      "step": 745
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 745,
      "training_step_time": 9.948487520217896
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 3.1374599933624268,
      "step": 745
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 745,
      "training_step_time": 9.909264087677002
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 2.4386966228485107,
      "step": 745
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 745,
      "training_step_time": 7.467353820800781
    },
    {
      "epoch": 2.841949462890625e-06,
      "model_forward_time": 3.135138988494873,
      "step": 745
    },
    {
      "epoch": 2.841949462890625e-06,
      "step": 745,
      "training_step_time": 10.32903504371643
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 2.4607884883880615,
      "step": 746
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 746,
      "training_step_time": 7.493553400039673
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 3.1335160732269287,
      "step": 746
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 746,
      "training_step_time": 9.906860589981079
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 3.1395046710968018,
      "step": 746
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 746,
      "training_step_time": 9.910887718200684
    },
    {
      "epoch": 2.84576416015625e-06,
      "model_forward_time": 3.1393826007843018,
      "step": 746
    },
    {
      "epoch": 2.84576416015625e-06,
      "step": 746,
      "training_step_time": 10.266426086425781
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 3.493640422821045,
      "step": 747
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 747,
      "training_step_time": 10.265120506286621
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 3.140281915664673,
      "step": 747
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 747,
      "training_step_time": 9.916063785552979
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 3.1452934741973877,
      "step": 747
    },
    {
      "dynamics_loss_avg": 0.07003799080848694,
      "epoch": 2.849578857421875e-06,
      "step": 747
    },
    {
      "action_loss_avg": 0.05034127393737435,
      "epoch": 2.849578857421875e-06,
      "step": 747
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 747,
      "training_step_time": 9.924798488616943
    },
    {
      "epoch": 2.849578857421875e-06,
      "model_forward_time": 3.1388049125671387,
      "step": 747
    },
    {
      "epoch": 2.849578857421875e-06,
      "step": 747,
      "training_step_time": 10.016299486160278
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 3.184605360031128,
      "step": 748
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 748,
      "training_step_time": 9.958086013793945
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 3.140293836593628,
      "step": 748
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 748,
      "training_step_time": 10.18982481956482
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 3.1431899070739746,
      "step": 748
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 748,
      "training_step_time": 9.923993587493896
    },
    {
      "epoch": 2.8533935546875e-06,
      "model_forward_time": 3.1894209384918213,
      "step": 748
    },
    {
      "epoch": 2.8533935546875e-06,
      "step": 748,
      "training_step_time": 10.068740367889404
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 3.2139620780944824,
      "step": 749
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 749,
      "training_step_time": 9.994361877441406
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 3.132223606109619,
      "step": 749
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 749,
      "training_step_time": 9.90367865562439
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 3.4042022228240967,
      "step": 749
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 749,
      "training_step_time": 10.179718732833862
    },
    {
      "epoch": 2.857208251953125e-06,
      "model_forward_time": 3.1351211071014404,
      "step": 749
    },
    {
      "epoch": 2.857208251953125e-06,
      "step": 749,
      "training_step_time": 10.267512321472168
    },
    {
      "epoch": 2.86102294921875e-06,
      "grad_norm": 0.7259237170219421,
      "learning_rate": 1e-05,
      "loss": 0.1422,
      "step": 750
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 3.1785354614257812,
      "step": 750
    },
    {
      "dynamics_loss_avg": 0.08905906975269318,
      "epoch": 2.86102294921875e-06,
      "step": 750
    },
    {
      "action_loss_avg": 0.061384999006986615,
      "epoch": 2.86102294921875e-06,
      "step": 750
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 750,
      "training_step_time": 9.950718879699707
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 3.1305384635925293,
      "step": 750
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 750,
      "training_step_time": 9.90678071975708
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 3.1312358379364014,
      "step": 750
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 750,
      "training_step_time": 9.90401840209961
    },
    {
      "epoch": 2.86102294921875e-06,
      "model_forward_time": 3.423173189163208,
      "step": 750
    },
    {
      "epoch": 2.86102294921875e-06,
      "step": 750,
      "training_step_time": 10.28770899772644
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 3.1729702949523926,
      "step": 751
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 751,
      "training_step_time": 9.941441774368286
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 3.132418155670166,
      "step": 751
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 751,
      "training_step_time": 9.903409481048584
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 3.134800910949707,
      "step": 751
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 751,
      "training_step_time": 9.905704259872437
    },
    {
      "epoch": 2.864837646484375e-06,
      "model_forward_time": 3.1349003314971924,
      "step": 751
    },
    {
      "epoch": 2.864837646484375e-06,
      "step": 751,
      "training_step_time": 10.287895441055298
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 3.439385175704956,
      "step": 752
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 752,
      "training_step_time": 10.212283611297607
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 3.1339023113250732,
      "step": 752
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 752,
      "training_step_time": 9.9082510471344
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 3.1355655193328857,
      "step": 752
    },
    {
      "dynamics_loss_avg": 0.06594266630709171,
      "epoch": 2.86865234375e-06,
      "step": 752
    },
    {
      "action_loss_avg": 0.057420529425144196,
      "epoch": 2.86865234375e-06,
      "step": 752
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 752,
      "training_step_time": 9.91196584701538
    },
    {
      "epoch": 2.86865234375e-06,
      "model_forward_time": 3.138718366622925,
      "step": 752
    },
    {
      "epoch": 2.86865234375e-06,
      "step": 752,
      "training_step_time": 10.252622842788696
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 3.1918551921844482,
      "step": 753
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 753,
      "training_step_time": 9.966512680053711
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 3.4163765907287598,
      "step": 753
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 753,
      "training_step_time": 10.19184422492981
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 3.1358642578125,
      "step": 753
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 753,
      "training_step_time": 9.913849592208862
    },
    {
      "epoch": 2.872467041015625e-06,
      "model_forward_time": 3.1397132873535156,
      "step": 753
    },
    {
      "epoch": 2.872467041015625e-06,
      "step": 753,
      "training_step_time": 10.257293939590454
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 3.1559629440307617,
      "step": 754
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 754,
      "training_step_time": 9.933944940567017
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 3.1371524333953857,
      "step": 754
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 754,
      "training_step_time": 9.913970470428467
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 2.434126138687134,
      "step": 754
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 754,
      "training_step_time": 7.73710036277771
    },
    {
      "epoch": 2.87628173828125e-06,
      "model_forward_time": 3.139254570007324,
      "step": 754
    },
    {
      "epoch": 2.87628173828125e-06,
      "step": 754,
      "training_step_time": 10.264525175094604
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 3.1699249744415283,
      "step": 755
    },
    {
      "dynamics_loss_avg": 0.07518294528126716,
      "epoch": 2.880096435546875e-06,
      "step": 755
    },
    {
      "action_loss_avg": 0.07056521438062191,
      "epoch": 2.880096435546875e-06,
      "step": 755
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 755,
      "training_step_time": 9.950975894927979
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 3.1336140632629395,
      "step": 755
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 755,
      "training_step_time": 9.899892568588257
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 3.1332316398620605,
      "step": 755
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 755,
      "training_step_time": 9.910839557647705
    },
    {
      "epoch": 2.880096435546875e-06,
      "model_forward_time": 2.7167582511901855,
      "step": 755
    },
    {
      "epoch": 2.880096435546875e-06,
      "step": 755,
      "training_step_time": 10.275442123413086
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 3.1708762645721436,
      "step": 756
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 756,
      "training_step_time": 9.944655418395996
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 3.1301827430725098,
      "step": 756
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 756,
      "training_step_time": 9.898030996322632
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 3.1297318935394287,
      "step": 756
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 756,
      "training_step_time": 9.8956458568573
    },
    {
      "epoch": 2.8839111328125e-06,
      "model_forward_time": 3.1292200088500977,
      "step": 756
    },
    {
      "epoch": 2.8839111328125e-06,
      "step": 756,
      "training_step_time": 10.290220737457275
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 2.4756853580474854,
      "step": 757
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 757,
      "training_step_time": 7.4989752769470215
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 3.392177104949951,
      "step": 757
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 757,
      "training_step_time": 10.167130947113037
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 3.1366944313049316,
      "step": 757
    },
    {
      "dynamics_loss_avg": 0.07996221780776977,
      "epoch": 2.887725830078125e-06,
      "step": 757
    },
    {
      "action_loss_avg": 0.0535312945023179,
      "epoch": 2.887725830078125e-06,
      "step": 757
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 757,
      "training_step_time": 9.910008192062378
    },
    {
      "epoch": 2.887725830078125e-06,
      "model_forward_time": 3.136979103088379,
      "step": 757
    },
    {
      "epoch": 2.887725830078125e-06,
      "step": 757,
      "training_step_time": 10.27712368965149
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 3.1894586086273193,
      "step": 758
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 758,
      "training_step_time": 9.960325002670288
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 2.71195387840271,
      "step": 758
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 758,
      "training_step_time": 7.733198642730713
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 3.1360368728637695,
      "step": 758
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 758,
      "training_step_time": 9.901566982269287
    },
    {
      "epoch": 2.89154052734375e-06,
      "model_forward_time": 3.1364340782165527,
      "step": 758
    },
    {
      "epoch": 2.89154052734375e-06,
      "step": 758,
      "training_step_time": 10.230202913284302
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 3.1741394996643066,
      "step": 759
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 759,
      "training_step_time": 9.947636365890503
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 3.141263008117676,
      "step": 759
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 759,
      "training_step_time": 9.915229320526123
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 3.141540288925171,
      "step": 759
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 759,
      "training_step_time": 10.182220458984375
    },
    {
      "epoch": 2.895355224609375e-06,
      "model_forward_time": 3.143953561782837,
      "step": 759
    },
    {
      "epoch": 2.895355224609375e-06,
      "step": 759,
      "training_step_time": 10.279956340789795
    },
    {
      "epoch": 2.899169921875e-06,
      "grad_norm": 1.024020791053772,
      "learning_rate": 1e-05,
      "loss": 0.1431,
      "step": 760
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 3.172604560852051,
      "step": 760
    },
    {
      "dynamics_loss_avg": 0.06707912683486938,
      "epoch": 2.899169921875e-06,
      "step": 760
    },
    {
      "action_loss_avg": 0.054940033331513406,
      "epoch": 2.899169921875e-06,
      "step": 760
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 760,
      "training_step_time": 9.95411205291748
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 3.1419835090637207,
      "step": 760
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 760,
      "training_step_time": 9.91732144355774
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 3.137136459350586,
      "step": 760
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 760,
      "training_step_time": 9.90727972984314
    },
    {
      "epoch": 2.899169921875e-06,
      "model_forward_time": 3.4010708332061768,
      "step": 760
    },
    {
      "epoch": 2.899169921875e-06,
      "step": 760,
      "training_step_time": 10.266050100326538
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 3.162461757659912,
      "step": 761
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 761,
      "training_step_time": 9.941352367401123
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 3.134875774383545,
      "step": 761
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 761,
      "training_step_time": 9.909351348876953
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 3.130821943283081,
      "step": 761
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 761,
      "training_step_time": 9.9355947971344
    },
    {
      "epoch": 2.902984619140625e-06,
      "model_forward_time": 2.4293291568756104,
      "step": 761
    },
    {
      "epoch": 2.902984619140625e-06,
      "step": 761,
      "training_step_time": 10.268883466720581
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 3.1571431159973145,
      "step": 762
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 762,
      "training_step_time": 10.23721432685852
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 3.1318771839141846,
      "step": 762
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 762,
      "training_step_time": 9.906273603439331
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 3.1345179080963135,
      "step": 762
    },
    {
      "dynamics_loss_avg": 0.06188577637076378,
      "epoch": 2.90679931640625e-06,
      "step": 762
    },
    {
      "action_loss_avg": 0.04840219654142856,
      "epoch": 2.90679931640625e-06,
      "step": 762
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 762,
      "training_step_time": 9.912816762924194
    },
    {
      "epoch": 2.90679931640625e-06,
      "model_forward_time": 3.133260726928711,
      "step": 762
    },
    {
      "epoch": 2.90679931640625e-06,
      "step": 762,
      "training_step_time": 10.01980471611023
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 3.156182050704956,
      "step": 763
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 763,
      "training_step_time": 9.934523344039917
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 3.1315672397613525,
      "step": 763
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 763,
      "training_step_time": 10.197407484054565
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 2.430286407470703,
      "step": 763
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 763,
      "training_step_time": 7.461206436157227
    },
    {
      "epoch": 2.910614013671875e-06,
      "model_forward_time": 3.1307592391967773,
      "step": 763
    },
    {
      "epoch": 2.910614013671875e-06,
      "step": 763,
      "training_step_time": 10.29960012435913
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 3.16347599029541,
      "step": 764
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 764,
      "training_step_time": 9.933398246765137
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 3.1363182067871094,
      "step": 764
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 764,
      "training_step_time": 9.905304431915283
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 3.13858962059021,
      "step": 764
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 764,
      "training_step_time": 9.906375646591187
    },
    {
      "epoch": 2.9144287109375e-06,
      "model_forward_time": 3.448908805847168,
      "step": 764
    },
    {
      "epoch": 2.9144287109375e-06,
      "step": 764,
      "training_step_time": 10.312055587768555
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 3.178131341934204,
      "step": 765
    },
    {
      "dynamics_loss_avg": 0.07237479500472546,
      "epoch": 2.918243408203125e-06,
      "step": 765
    },
    {
      "action_loss_avg": 0.06381096951663494,
      "epoch": 2.918243408203125e-06,
      "step": 765
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 765,
      "training_step_time": 9.955376863479614
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 3.1393728256225586,
      "step": 765
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 765,
      "training_step_time": 9.91406774520874
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 3.1441493034362793,
      "step": 765
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 765,
      "training_step_time": 9.918434143066406
    },
    {
      "epoch": 2.918243408203125e-06,
      "model_forward_time": 2.4370458126068115,
      "step": 765
    },
    {
      "epoch": 2.918243408203125e-06,
      "step": 765,
      "training_step_time": 10.268498420715332
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 3.478148937225342,
      "step": 766
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 766,
      "training_step_time": 10.25490665435791
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 3.1456215381622314,
      "step": 766
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 766,
      "training_step_time": 9.922054052352905
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 3.143848180770874,
      "step": 766
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 766,
      "training_step_time": 9.922148704528809
    },
    {
      "epoch": 2.92205810546875e-06,
      "model_forward_time": 3.1440646648406982,
      "step": 766
    },
    {
      "epoch": 2.92205810546875e-06,
      "step": 766,
      "training_step_time": 10.325236558914185
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 2.452517509460449,
      "step": 767
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 767,
      "training_step_time": 7.767496109008789
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 2.438663959503174,
      "step": 767
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 767,
      "training_step_time": 7.469075679779053
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 2.430208683013916,
      "step": 767
    },
    {
      "dynamics_loss_avg": 0.07610349804162979,
      "epoch": 2.925872802734375e-06,
      "step": 767
    },
    {
      "action_loss_avg": 0.06305648908019065,
      "epoch": 2.925872802734375e-06,
      "step": 767
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 767,
      "training_step_time": 7.461034297943115
    },
    {
      "epoch": 2.925872802734375e-06,
      "model_forward_time": 3.138432741165161,
      "step": 767
    },
    {
      "epoch": 2.925872802734375e-06,
      "step": 767,
      "training_step_time": 10.003964185714722
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 3.1694469451904297,
      "step": 768
    },
    {
      "epoch": 2.9296875e-06,
      "step": 768,
      "training_step_time": 9.942732095718384
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 3.1431477069854736,
      "step": 768
    },
    {
      "epoch": 2.9296875e-06,
      "step": 768,
      "training_step_time": 9.914144515991211
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 3.4284310340881348,
      "step": 768
    },
    {
      "epoch": 2.9296875e-06,
      "step": 768,
      "training_step_time": 10.203737258911133
    },
    {
      "epoch": 2.9296875e-06,
      "model_forward_time": 3.131417989730835,
      "step": 768
    },
    {
      "epoch": 2.9296875e-06,
      "step": 768,
      "training_step_time": 10.264883041381836
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 3.183818817138672,
      "step": 769
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 769,
      "training_step_time": 9.95626974105835
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 2.4262709617614746,
      "step": 769
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 769,
      "training_step_time": 7.451541900634766
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 3.1235780715942383,
      "step": 769
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 769,
      "training_step_time": 9.892767667770386
    },
    {
      "epoch": 2.933502197265625e-06,
      "model_forward_time": 3.378152847290039,
      "step": 769
    },
    {
      "epoch": 2.933502197265625e-06,
      "step": 769,
      "training_step_time": 10.251019477844238
    },
    {
      "epoch": 2.93731689453125e-06,
      "grad_norm": 0.5788516402244568,
      "learning_rate": 1e-05,
      "loss": 0.1364,
      "step": 770
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 3.171776533126831,
      "step": 770
    },
    {
      "dynamics_loss_avg": 0.07726627625524998,
      "epoch": 2.93731689453125e-06,
      "step": 770
    },
    {
      "action_loss_avg": 0.06702022068202496,
      "epoch": 2.93731689453125e-06,
      "step": 770
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 770,
      "training_step_time": 9.94949984550476
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 3.13767409324646,
      "step": 770
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 770,
      "training_step_time": 9.907950401306152
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 3.1355085372924805,
      "step": 770
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 770,
      "training_step_time": 9.903280019760132
    },
    {
      "epoch": 2.93731689453125e-06,
      "model_forward_time": 3.133920192718506,
      "step": 770
    },
    {
      "epoch": 2.93731689453125e-06,
      "step": 770,
      "training_step_time": 10.294393301010132
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 2.7606048583984375,
      "step": 771
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 771,
      "training_step_time": 7.785843849182129
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 3.132441997528076,
      "step": 771
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 771,
      "training_step_time": 9.899328708648682
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 3.1324992179870605,
      "step": 771
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 771,
      "training_step_time": 9.897440910339355
    },
    {
      "epoch": 2.941131591796875e-06,
      "model_forward_time": 3.1357696056365967,
      "step": 771
    },
    {
      "epoch": 2.941131591796875e-06,
      "step": 771,
      "training_step_time": 10.301981449127197
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 3.464418649673462,
      "step": 772
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 772,
      "training_step_time": 10.234981775283813
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 3.137634515762329,
      "step": 772
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 772,
      "training_step_time": 9.909274101257324
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 3.1356282234191895,
      "step": 772
    },
    {
      "dynamics_loss_avg": 0.06744088204577565,
      "epoch": 2.9449462890625e-06,
      "step": 772
    },
    {
      "action_loss_avg": 0.04681805479340255,
      "epoch": 2.9449462890625e-06,
      "step": 772
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 772,
      "training_step_time": 9.905809164047241
    },
    {
      "epoch": 2.9449462890625e-06,
      "model_forward_time": 3.143435001373291,
      "step": 772
    },
    {
      "epoch": 2.9449462890625e-06,
      "step": 772,
      "training_step_time": 10.015034675598145
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 3.1857993602752686,
      "step": 773
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 773,
      "training_step_time": 9.957434892654419
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 3.1382663249969482,
      "step": 773
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 773,
      "training_step_time": 10.186131715774536
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 3.1411473751068115,
      "step": 773
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 773,
      "training_step_time": 9.916046142578125
    },
    {
      "epoch": 2.948760986328125e-06,
      "model_forward_time": 3.145521640777588,
      "step": 773
    },
    {
      "epoch": 2.948760986328125e-06,
      "step": 773,
      "training_step_time": 10.29860520362854
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 3.167832612991333,
      "step": 774
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 774,
      "training_step_time": 9.95134449005127
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 3.1431117057800293,
      "step": 774
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 774,
      "training_step_time": 9.924092292785645
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 3.143733024597168,
      "step": 774
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 774,
      "training_step_time": 10.198383569717407
    },
    {
      "epoch": 2.95257568359375e-06,
      "model_forward_time": 2.441810369491577,
      "step": 774
    },
    {
      "epoch": 2.95257568359375e-06,
      "step": 774,
      "training_step_time": 10.256468057632446
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 3.1880667209625244,
      "step": 775
    },
    {
      "dynamics_loss_avg": 0.0770741555839777,
      "epoch": 2.956390380859375e-06,
      "step": 775
    },
    {
      "action_loss_avg": 0.06030193455517292,
      "epoch": 2.956390380859375e-06,
      "step": 775
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 775,
      "training_step_time": 9.969764232635498
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 3.1426842212677,
      "step": 775
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 775,
      "training_step_time": 9.91807222366333
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 3.141652822494507,
      "step": 775
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 775,
      "training_step_time": 9.919562339782715
    },
    {
      "epoch": 2.956390380859375e-06,
      "model_forward_time": 3.1314151287078857,
      "step": 775
    },
    {
      "epoch": 2.956390380859375e-06,
      "step": 775,
      "training_step_time": 10.286593437194824
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 3.183560609817505,
      "step": 776
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 776,
      "training_step_time": 9.958528518676758
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 3.130072832107544,
      "step": 776
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 776,
      "training_step_time": 9.90099549293518
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 3.1335418224334717,
      "step": 776
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 776,
      "training_step_time": 9.908275127410889
    },
    {
      "epoch": 2.960205078125e-06,
      "model_forward_time": 3.13193941116333,
      "step": 776
    },
    {
      "epoch": 2.960205078125e-06,
      "step": 776,
      "training_step_time": 10.303308725357056
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 3.17437481880188,
      "step": 777
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 777,
      "training_step_time": 10.264252424240112
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 2.430264711380005,
      "step": 777
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 777,
      "training_step_time": 7.461952447891235
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 3.131781578063965,
      "step": 777
    },
    {
      "dynamics_loss_avg": 0.07751701772212982,
      "epoch": 2.964019775390625e-06,
      "step": 777
    },
    {
      "action_loss_avg": 0.057562272250652316,
      "epoch": 2.964019775390625e-06,
      "step": 777
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 777,
      "training_step_time": 9.91014552116394
    },
    {
      "epoch": 2.964019775390625e-06,
      "model_forward_time": 3.138925075531006,
      "step": 777
    },
    {
      "epoch": 2.964019775390625e-06,
      "step": 777,
      "training_step_time": 10.029448509216309
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 3.181058168411255,
      "step": 778
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 778,
      "training_step_time": 9.956276655197144
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 3.383970022201538,
      "step": 778
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 778,
      "training_step_time": 10.153747081756592
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 3.1316933631896973,
      "step": 778
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 778,
      "training_step_time": 9.903723001480103
    },
    {
      "epoch": 2.96783447265625e-06,
      "model_forward_time": 3.128375291824341,
      "step": 778
    },
    {
      "epoch": 2.96783447265625e-06,
      "step": 778,
      "training_step_time": 10.017860412597656
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 3.1660163402557373,
      "step": 779
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 779,
      "training_step_time": 9.93908143043518
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 3.1334755420684814,
      "step": 779
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 779,
      "training_step_time": 9.902396202087402
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 3.1297173500061035,
      "step": 779
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 779,
      "training_step_time": 10.166706323623657
    },
    {
      "epoch": 2.971649169921875e-06,
      "model_forward_time": 3.134176731109619,
      "step": 779
    },
    {
      "epoch": 2.971649169921875e-06,
      "step": 779,
      "training_step_time": 10.311989784240723
    },
    {
      "epoch": 2.9754638671875e-06,
      "grad_norm": 0.6349285840988159,
      "learning_rate": 1e-05,
      "loss": 0.1371,
      "step": 780
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 3.1650938987731934,
      "step": 780
    },
    {
      "dynamics_loss_avg": 0.07043106332421303,
      "epoch": 2.9754638671875e-06,
      "step": 780
    },
    {
      "action_loss_avg": 0.0688019322231412,
      "epoch": 2.9754638671875e-06,
      "step": 780
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 780,
      "training_step_time": 9.94028353691101
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 3.137605667114258,
      "step": 780
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 780,
      "training_step_time": 9.908088684082031
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 3.1289541721343994,
      "step": 780
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 780,
      "training_step_time": 9.906014919281006
    },
    {
      "epoch": 2.9754638671875e-06,
      "model_forward_time": 3.429654359817505,
      "step": 780
    },
    {
      "epoch": 2.9754638671875e-06,
      "step": 780,
      "training_step_time": 10.297448635101318
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 3.1676535606384277,
      "step": 781
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 781,
      "training_step_time": 9.944921493530273
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 2.436077117919922,
      "step": 781
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 781,
      "training_step_time": 7.4674859046936035
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 3.136484146118164,
      "step": 781
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 781,
      "training_step_time": 9.90868878364563
    },
    {
      "epoch": 2.979278564453125e-06,
      "model_forward_time": 3.1436927318573,
      "step": 781
    },
    {
      "epoch": 2.979278564453125e-06,
      "step": 781,
      "training_step_time": 10.273170471191406
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 3.4392378330230713,
      "step": 782
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 782,
      "training_step_time": 10.214400291442871
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 3.144519805908203,
      "step": 782
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 782,
      "training_step_time": 9.917901754379272
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 3.143062114715576,
      "step": 782
    },
    {
      "dynamics_loss_avg": 0.07730870097875595,
      "epoch": 2.98309326171875e-06,
      "step": 782
    },
    {
      "action_loss_avg": 0.07027732767164707,
      "epoch": 2.98309326171875e-06,
      "step": 782
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 782,
      "training_step_time": 9.921481370925903
    },
    {
      "epoch": 2.98309326171875e-06,
      "model_forward_time": 3.138227701187134,
      "step": 782
    },
    {
      "epoch": 2.98309326171875e-06,
      "step": 782,
      "training_step_time": 10.024230718612671
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 3.1769495010375977,
      "step": 783
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 783,
      "training_step_time": 9.945728302001953
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 3.135826349258423,
      "step": 783
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 783,
      "training_step_time": 10.187373399734497
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 3.137629270553589,
      "step": 783
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 783,
      "training_step_time": 9.903387069702148
    },
    {
      "epoch": 2.986907958984375e-06,
      "model_forward_time": 3.1362407207489014,
      "step": 783
    },
    {
      "epoch": 2.986907958984375e-06,
      "step": 783,
      "training_step_time": 10.020710229873657
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 2.4528937339782715,
      "step": 784
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 784,
      "training_step_time": 7.484752416610718
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 3.1350958347320557,
      "step": 784
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 784,
      "training_step_time": 9.904236078262329
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 2.4372429847717285,
      "step": 784
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 784,
      "training_step_time": 7.462237358093262
    },
    {
      "epoch": 2.99072265625e-06,
      "model_forward_time": 2.7441141605377197,
      "step": 784
    },
    {
      "epoch": 2.99072265625e-06,
      "step": 784,
      "training_step_time": 10.298133373260498
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 3.147278070449829,
      "step": 785
    },
    {
      "dynamics_loss_avg": 0.06649270774796605,
      "epoch": 2.994537353515625e-06,
      "step": 785
    },
    {
      "action_loss_avg": 0.05122095139231533,
      "epoch": 2.994537353515625e-06,
      "step": 785
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 785,
      "training_step_time": 9.915674448013306
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 3.1356236934661865,
      "step": 785
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 785,
      "training_step_time": 9.90857982635498
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 2.439401865005493,
      "step": 785
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 785,
      "training_step_time": 7.465423107147217
    },
    {
      "epoch": 2.994537353515625e-06,
      "model_forward_time": 2.4328525066375732,
      "step": 785
    },
    {
      "epoch": 2.994537353515625e-06,
      "step": 785,
      "training_step_time": 10.265554189682007
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 3.4369757175445557,
      "step": 786
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 786,
      "training_step_time": 10.215281009674072
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 3.138937473297119,
      "step": 786
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 786,
      "training_step_time": 9.911080121994019
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 3.139225482940674,
      "step": 786
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 786,
      "training_step_time": 9.911361694335938
    },
    {
      "epoch": 2.99835205078125e-06,
      "model_forward_time": 3.1476831436157227,
      "step": 786
    },
    {
      "epoch": 2.99835205078125e-06,
      "step": 786,
      "training_step_time": 10.291473388671875
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 2.463043212890625,
      "step": 787
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 787,
      "training_step_time": 7.492655038833618
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 3.3853533267974854,
      "step": 787
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 787,
      "training_step_time": 10.15907621383667
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 3.142491340637207,
      "step": 787
    },
    {
      "dynamics_loss_avg": 0.09151091873645782,
      "epoch": 3.002166748046875e-06,
      "step": 787
    },
    {
      "action_loss_avg": 0.0706741075962782,
      "epoch": 3.002166748046875e-06,
      "step": 787
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 787,
      "training_step_time": 9.919212341308594
    },
    {
      "epoch": 3.002166748046875e-06,
      "model_forward_time": 3.1375176906585693,
      "step": 787
    },
    {
      "epoch": 3.002166748046875e-06,
      "step": 787,
      "training_step_time": 10.348155975341797
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 3.1438748836517334,
      "step": 788
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 788,
      "training_step_time": 9.924269199371338
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 3.142522096633911,
      "step": 788
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 788,
      "training_step_time": 9.913036108016968
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 2.7446930408477783,
      "step": 788
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 788,
      "training_step_time": 7.775374412536621
    },
    {
      "epoch": 3.0059814453125e-06,
      "model_forward_time": 3.1337225437164307,
      "step": 788
    },
    {
      "epoch": 3.0059814453125e-06,
      "step": 788,
      "training_step_time": 10.307481050491333
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 2.474893569946289,
      "step": 789
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 789,
      "training_step_time": 7.499831199645996
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 3.13356614112854,
      "step": 789
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 789,
      "training_step_time": 9.900563955307007
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 3.1356358528137207,
      "step": 789
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 789,
      "training_step_time": 9.913103342056274
    },
    {
      "epoch": 3.009796142578125e-06,
      "model_forward_time": 3.400512933731079,
      "step": 789
    },
    {
      "epoch": 3.009796142578125e-06,
      "step": 789,
      "training_step_time": 10.267880916595459
    },
    {
      "epoch": 3.01361083984375e-06,
      "grad_norm": 0.9594849944114685,
      "learning_rate": 1e-05,
      "loss": 0.1409,
      "step": 790
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 3.172698497772217,
      "step": 790
    },
    {
      "dynamics_loss_avg": 0.07818401698023081,
      "epoch": 3.01361083984375e-06,
      "step": 790
    },
    {
      "action_loss_avg": 0.06646889774128795,
      "epoch": 3.01361083984375e-06,
      "step": 790
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 790,
      "training_step_time": 9.952735424041748
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 3.1337976455688477,
      "step": 790
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 790,
      "training_step_time": 9.907771348953247
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 3.1327333450317383,
      "step": 790
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 790,
      "training_step_time": 9.906398296356201
    },
    {
      "epoch": 3.01361083984375e-06,
      "model_forward_time": 3.1329939365386963,
      "step": 790
    },
    {
      "epoch": 3.01361083984375e-06,
      "step": 790,
      "training_step_time": 10.261478900909424
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 3.492427110671997,
      "step": 791
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 791,
      "training_step_time": 10.268988847732544
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 3.1326229572296143,
      "step": 791
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 791,
      "training_step_time": 9.902384996414185
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 3.1284749507904053,
      "step": 791
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 791,
      "training_step_time": 9.895397424697876
    },
    {
      "epoch": 3.017425537109375e-06,
      "model_forward_time": 2.4261655807495117,
      "step": 791
    },
    {
      "epoch": 3.017425537109375e-06,
      "step": 791,
      "training_step_time": 10.305060625076294
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 2.4574832916259766,
      "step": 792
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 792,
      "training_step_time": 7.478047847747803
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 3.131031036376953,
      "step": 792
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 792,
      "training_step_time": 10.169161558151245
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 3.1327738761901855,
      "step": 792
    },
    {
      "dynamics_loss_avg": 0.07176323868334293,
      "epoch": 3.021240234375e-06,
      "step": 792
    },
    {
      "action_loss_avg": 0.05729884915053844,
      "epoch": 3.021240234375e-06,
      "step": 792
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 792,
      "training_step_time": 9.901018619537354
    },
    {
      "epoch": 3.021240234375e-06,
      "model_forward_time": 3.1342411041259766,
      "step": 792
    },
    {
      "epoch": 3.021240234375e-06,
      "step": 792,
      "training_step_time": 10.270116806030273
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 2.474679946899414,
      "step": 793
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 793,
      "training_step_time": 7.4976806640625
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 3.1301655769348145,
      "step": 793
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 793,
      "training_step_time": 9.90218210220337
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 2.432482957839966,
      "step": 793
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 793,
      "training_step_time": 7.455609083175659
    },
    {
      "epoch": 3.025054931640625e-06,
      "model_forward_time": 3.4500911235809326,
      "step": 793
    },
    {
      "epoch": 3.025054931640625e-06,
      "step": 793,
      "training_step_time": 10.312814474105835
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 3.1798534393310547,
      "step": 794
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 794,
      "training_step_time": 9.949454545974731
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 3.1301677227020264,
      "step": 794
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 794,
      "training_step_time": 9.900367498397827
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 3.1333467960357666,
      "step": 794
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 794,
      "training_step_time": 9.907153367996216
    },
    {
      "epoch": 3.02886962890625e-06,
      "model_forward_time": 3.132335901260376,
      "step": 794
    },
    {
      "epoch": 3.02886962890625e-06,
      "step": 794,
      "training_step_time": 10.257314205169678
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 3.165426731109619,
      "step": 795
    },
    {
      "dynamics_loss_avg": 0.07942569497972726,
      "epoch": 3.032684326171875e-06,
      "step": 795
    },
    {
      "action_loss_avg": 0.05328937266021967,
      "epoch": 3.032684326171875e-06,
      "step": 795
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 795,
      "training_step_time": 9.948887348175049
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 3.130722999572754,
      "step": 795
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 795,
      "training_step_time": 9.90151596069336
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 3.1336655616760254,
      "step": 795
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 795,
      "training_step_time": 9.902449607849121
    },
    {
      "epoch": 3.032684326171875e-06,
      "model_forward_time": 3.1321656703948975,
      "step": 795
    },
    {
      "epoch": 3.032684326171875e-06,
      "step": 795,
      "training_step_time": 10.239021301269531
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 3.452646493911743,
      "step": 796
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 796,
      "training_step_time": 10.22385048866272
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 3.1311843395233154,
      "step": 796
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 796,
      "training_step_time": 9.898438930511475
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 3.1350560188293457,
      "step": 796
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 796,
      "training_step_time": 9.910542488098145
    },
    {
      "epoch": 3.0364990234375e-06,
      "model_forward_time": 3.131466865539551,
      "step": 796
    },
    {
      "epoch": 3.0364990234375e-06,
      "step": 796,
      "training_step_time": 10.25334906578064
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 3.1467275619506836,
      "step": 797
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 797,
      "training_step_time": 9.918288707733154
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 3.130321979522705,
      "step": 797
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 797,
      "training_step_time": 10.206757068634033
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 2.4277420043945312,
      "step": 797
    },
    {
      "dynamics_loss_avg": 0.07046563215553761,
      "epoch": 3.040313720703125e-06,
      "step": 797
    },
    {
      "action_loss_avg": 0.058893514797091484,
      "epoch": 3.040313720703125e-06,
      "step": 797
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 797,
      "training_step_time": 7.453709125518799
    },
    {
      "epoch": 3.040313720703125e-06,
      "model_forward_time": 3.1233160495758057,
      "step": 797
    },
    {
      "epoch": 3.040313720703125e-06,
      "step": 797,
      "training_step_time": 10.309031009674072
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 2.4761710166931152,
      "step": 798
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 798,
      "training_step_time": 7.501073122024536
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 3.131673812866211,
      "step": 798
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 798,
      "training_step_time": 9.900974035263062
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 3.1307199001312256,
      "step": 798
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 798,
      "training_step_time": 9.895640134811401
    },
    {
      "epoch": 3.04412841796875e-06,
      "model_forward_time": 3.3788721561431885,
      "step": 798
    },
    {
      "epoch": 3.04412841796875e-06,
      "step": 798,
      "training_step_time": 10.272400856018066
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 3.1542279720306396,
      "step": 799
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 799,
      "training_step_time": 9.926606893539429
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 3.1320183277130127,
      "step": 799
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 799,
      "training_step_time": 9.90773630142212
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 3.134579658508301,
      "step": 799
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 799,
      "training_step_time": 9.90667200088501
    },
    {
      "epoch": 3.047943115234375e-06,
      "model_forward_time": 3.138286590576172,
      "step": 799
    },
    {
      "epoch": 3.047943115234375e-06,
      "step": 799,
      "training_step_time": 10.308149814605713
    },
    {
      "epoch": 3.0517578125e-06,
      "grad_norm": 0.6167337894439697,
      "learning_rate": 1e-05,
      "loss": 0.1419,
      "step": 800
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 3.4818084239959717,
      "step": 800
    },
    {
      "dynamics_loss_avg": 0.08041624464094639,
      "epoch": 3.0517578125e-06,
      "step": 800
    },
    {
      "action_loss_avg": 0.0747726872563362,
      "epoch": 3.0517578125e-06,
      "step": 800
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 800,
      "training_step_time": 10.275764465332031
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 3.135340690612793,
      "step": 800
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 800,
      "training_step_time": 9.911924362182617
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 2.431849956512451,
      "step": 800
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 800,
      "training_step_time": 7.4644341468811035
    },
    {
      "epoch": 3.0517578125e-06,
      "model_forward_time": 3.132723808288574,
      "step": 800
    },
    {
      "epoch": 3.0517578125e-06,
      "step": 800,
      "training_step_time": 10.028229475021362
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 3.4539408683776855,
      "step": 801
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 801,
      "training_step_time": 10.233368158340454
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 3.1335349082946777,
      "step": 801
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 801,
      "training_step_time": 9.912465333938599
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 3.1317286491394043,
      "step": 801
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 801,
      "training_step_time": 9.90626573562622
    },
    {
      "epoch": 3.055572509765625e-06,
      "model_forward_time": 2.4303841590881348,
      "step": 801
    },
    {
      "epoch": 3.055572509765625e-06,
      "step": 801,
      "training_step_time": 10.27297854423523
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 3.1779725551605225,
      "step": 802
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 802,
      "training_step_time": 9.953681230545044
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 3.133286714553833,
      "step": 802
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 802,
      "training_step_time": 9.904691934585571
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 3.38346004486084,
      "step": 802
    },
    {
      "dynamics_loss_avg": 0.08173011802136898,
      "epoch": 3.05938720703125e-06,
      "step": 802
    },
    {
      "action_loss_avg": 0.06978193297982216,
      "epoch": 3.05938720703125e-06,
      "step": 802
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 802,
      "training_step_time": 10.155706882476807
    },
    {
      "epoch": 3.05938720703125e-06,
      "model_forward_time": 3.1294515132904053,
      "step": 802
    },
    {
      "epoch": 3.05938720703125e-06,
      "step": 802,
      "training_step_time": 10.011905670166016
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 3.177304267883301,
      "step": 803
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 803,
      "training_step_time": 9.952084302902222
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 3.127918004989624,
      "step": 803
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 803,
      "training_step_time": 9.900217533111572
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 3.4238874912261963,
      "step": 803
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 803,
      "training_step_time": 10.1958646774292
    },
    {
      "epoch": 3.063201904296875e-06,
      "model_forward_time": 3.130852222442627,
      "step": 803
    },
    {
      "epoch": 3.063201904296875e-06,
      "step": 803,
      "training_step_time": 10.273821592330933
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 3.166609287261963,
      "step": 804
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 804,
      "training_step_time": 9.943675756454468
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 2.432687759399414,
      "step": 804
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 804,
      "training_step_time": 7.463525295257568
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 3.1303913593292236,
      "step": 804
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 804,
      "training_step_time": 9.900257349014282
    },
    {
      "epoch": 3.0670166015625e-06,
      "model_forward_time": 3.134413957595825,
      "step": 804
    },
    {
      "epoch": 3.0670166015625e-06,
      "step": 804,
      "training_step_time": 10.304153442382812
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 3.4477880001068115,
      "step": 805
    },
    {
      "dynamics_loss_avg": 0.07531510107219219,
      "epoch": 3.070831298828125e-06,
      "step": 805
    },
    {
      "action_loss_avg": 0.06652801726013421,
      "epoch": 3.070831298828125e-06,
      "step": 805
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 805,
      "training_step_time": 10.237005472183228
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 3.1458890438079834,
      "step": 805
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 805,
      "training_step_time": 9.927691221237183
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 2.4363911151885986,
      "step": 805
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 805,
      "training_step_time": 7.468192100524902
    },
    {
      "epoch": 3.070831298828125e-06,
      "model_forward_time": 3.1350393295288086,
      "step": 805
    },
    {
      "epoch": 3.070831298828125e-06,
      "step": 805,
      "training_step_time": 10.037397861480713
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 3.1687331199645996,
      "step": 806
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 806,
      "training_step_time": 10.227102994918823
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 3.1404478549957275,
      "step": 806
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 806,
      "training_step_time": 9.936953067779541
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 3.1426141262054443,
      "step": 806
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 806,
      "training_step_time": 9.92174482345581
    },
    {
      "epoch": 3.07464599609375e-06,
      "model_forward_time": 3.140934944152832,
      "step": 806
    },
    {
      "epoch": 3.07464599609375e-06,
      "step": 806,
      "training_step_time": 10.27180290222168
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 3.1822102069854736,
      "step": 807
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 807,
      "training_step_time": 9.9722261428833
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 2.743684768676758,
      "step": 807
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 807,
      "training_step_time": 7.775497198104858
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 3.1355645656585693,
      "step": 807
    },
    {
      "dynamics_loss_avg": 0.07495985347777605,
      "epoch": 3.078460693359375e-06,
      "step": 807
    },
    {
      "action_loss_avg": 0.061648344388231635,
      "epoch": 3.078460693359375e-06,
      "step": 807
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 807,
      "training_step_time": 9.916438102722168
    },
    {
      "epoch": 3.078460693359375e-06,
      "model_forward_time": 3.1686391830444336,
      "step": 807
    },
    {
      "epoch": 3.078460693359375e-06,
      "step": 807,
      "training_step_time": 10.241924047470093
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 2.4832661151885986,
      "step": 808
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 808,
      "training_step_time": 7.5104944705963135
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 3.1320910453796387,
      "step": 808
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 808,
      "training_step_time": 9.902471780776978
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 3.132772207260132,
      "step": 808
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 808,
      "training_step_time": 10.174320697784424
    },
    {
      "epoch": 3.082275390625e-06,
      "model_forward_time": 3.1341745853424072,
      "step": 808
    },
    {
      "epoch": 3.082275390625e-06,
      "step": 808,
      "training_step_time": 10.305203914642334
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 3.1706864833831787,
      "step": 809
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 809,
      "training_step_time": 9.950126647949219
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 2.4691643714904785,
      "step": 809
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 809,
      "training_step_time": 7.50063157081604
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 3.1347672939300537,
      "step": 809
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 809,
      "training_step_time": 9.906591176986694
    },
    {
      "epoch": 3.086090087890625e-06,
      "model_forward_time": 3.393282175064087,
      "step": 809
    },
    {
      "epoch": 3.086090087890625e-06,
      "step": 809,
      "training_step_time": 10.322464227676392
    },
    {
      "epoch": 3.08990478515625e-06,
      "grad_norm": 0.7577072381973267,
      "learning_rate": 1e-05,
      "loss": 0.1387,
      "step": 810
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 3.183194398880005,
      "step": 810
    },
    {
      "dynamics_loss_avg": 0.0811368178576231,
      "epoch": 3.08990478515625e-06,
      "step": 810
    },
    {
      "action_loss_avg": 0.0662008736282587,
      "epoch": 3.08990478515625e-06,
      "step": 810
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 810,
      "training_step_time": 9.961967468261719
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 3.1369903087615967,
      "step": 810
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 810,
      "training_step_time": 9.91099739074707
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 3.134401798248291,
      "step": 810
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 810,
      "training_step_time": 9.91404414176941
    },
    {
      "epoch": 3.08990478515625e-06,
      "model_forward_time": 3.1342291831970215,
      "step": 810
    },
    {
      "epoch": 3.08990478515625e-06,
      "step": 810,
      "training_step_time": 10.242734432220459
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 3.1824190616607666,
      "step": 811
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 811,
      "training_step_time": 10.259770393371582
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 3.1331162452697754,
      "step": 811
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 811,
      "training_step_time": 9.904780864715576
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 3.13138747215271,
      "step": 811
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 811,
      "training_step_time": 9.908004522323608
    },
    {
      "epoch": 3.093719482421875e-06,
      "model_forward_time": 3.1345555782318115,
      "step": 811
    },
    {
      "epoch": 3.093719482421875e-06,
      "step": 811,
      "training_step_time": 10.01793384552002
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 3.1645822525024414,
      "step": 812
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 812,
      "training_step_time": 9.936636924743652
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 3.404764175415039,
      "step": 812
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 812,
      "training_step_time": 10.183326244354248
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 3.131040096282959,
      "step": 812
    },
    {
      "dynamics_loss_avg": 0.06493316069245339,
      "epoch": 3.0975341796875e-06,
      "step": 812
    },
    {
      "action_loss_avg": 0.055397140141576526,
      "epoch": 3.0975341796875e-06,
      "step": 812
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 812,
      "training_step_time": 9.911212682723999
    },
    {
      "epoch": 3.0975341796875e-06,
      "model_forward_time": 3.1379234790802,
      "step": 812
    },
    {
      "epoch": 3.0975341796875e-06,
      "step": 812,
      "training_step_time": 10.254941701889038
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 3.1650683879852295,
      "step": 813
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 813,
      "training_step_time": 9.946161270141602
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 3.132741689682007,
      "step": 813
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 813,
      "training_step_time": 9.915806531906128
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 3.398395299911499,
      "step": 813
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 813,
      "training_step_time": 10.174726009368896
    },
    {
      "epoch": 3.101348876953125e-06,
      "model_forward_time": 3.1345715522766113,
      "step": 813
    },
    {
      "epoch": 3.101348876953125e-06,
      "step": 813,
      "training_step_time": 10.262740135192871
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 3.167792558670044,
      "step": 814
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 814,
      "training_step_time": 9.944642782211304
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 3.139775514602661,
      "step": 814
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 814,
      "training_step_time": 9.923718452453613
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 3.1387758255004883,
      "step": 814
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 814,
      "training_step_time": 9.914363384246826
    },
    {
      "epoch": 3.10516357421875e-06,
      "model_forward_time": 3.4484188556671143,
      "step": 814
    },
    {
      "epoch": 3.10516357421875e-06,
      "step": 814,
      "training_step_time": 10.320433378219604
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 3.1666643619537354,
      "step": 815
    },
    {
      "dynamics_loss_avg": 0.07475672978907824,
      "epoch": 3.108978271484375e-06,
      "step": 815
    },
    {
      "action_loss_avg": 0.06355597730726004,
      "epoch": 3.108978271484375e-06,
      "step": 815
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 815,
      "training_step_time": 9.940677881240845
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 3.136443614959717,
      "step": 815
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 815,
      "training_step_time": 9.907593250274658
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 3.1395184993743896,
      "step": 815
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 815,
      "training_step_time": 9.910589456558228
    },
    {
      "epoch": 3.108978271484375e-06,
      "model_forward_time": 3.138475179672241,
      "step": 815
    },
    {
      "epoch": 3.108978271484375e-06,
      "step": 815,
      "training_step_time": 10.294003248214722
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 3.4576492309570312,
      "step": 816
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 816,
      "training_step_time": 10.240132093429565
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 3.1384060382843018,
      "step": 816
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 816,
      "training_step_time": 9.911341190338135
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 3.1383819580078125,
      "step": 816
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 816,
      "training_step_time": 9.912192583084106
    },
    {
      "epoch": 3.11279296875e-06,
      "model_forward_time": 3.1386284828186035,
      "step": 816
    },
    {
      "epoch": 3.11279296875e-06,
      "step": 816,
      "training_step_time": 10.259094953536987
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 3.172329902648926,
      "step": 817
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 817,
      "training_step_time": 9.947309017181396
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 3.3850107192993164,
      "step": 817
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 817,
      "training_step_time": 10.158487796783447
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 3.1388065814971924,
      "step": 817
    },
    {
      "dynamics_loss_avg": 0.06299410052597523,
      "epoch": 3.116607666015625e-06,
      "step": 817
    },
    {
      "action_loss_avg": 0.04983504619449377,
      "epoch": 3.116607666015625e-06,
      "step": 817
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 817,
      "training_step_time": 9.924173355102539
    },
    {
      "epoch": 3.116607666015625e-06,
      "model_forward_time": 3.1350600719451904,
      "step": 817
    },
    {
      "epoch": 3.116607666015625e-06,
      "step": 817,
      "training_step_time": 10.291450262069702
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 3.17696213722229,
      "step": 818
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 818,
      "training_step_time": 9.95828652381897
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 3.1440200805664062,
      "step": 818
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 818,
      "training_step_time": 9.922315835952759
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 3.4586827754974365,
      "step": 818
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 818,
      "training_step_time": 10.240283250808716
    },
    {
      "epoch": 3.12042236328125e-06,
      "model_forward_time": 3.146101951599121,
      "step": 818
    },
    {
      "epoch": 3.12042236328125e-06,
      "step": 818,
      "training_step_time": 10.298694849014282
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 3.184634208679199,
      "step": 819
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 819,
      "training_step_time": 9.965025663375854
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 2.4408466815948486,
      "step": 819
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 819,
      "training_step_time": 7.473280429840088
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 3.134680986404419,
      "step": 819
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 819,
      "training_step_time": 9.907569885253906
    },
    {
      "epoch": 3.124237060546875e-06,
      "model_forward_time": 3.3874430656433105,
      "step": 819
    },
    {
      "epoch": 3.124237060546875e-06,
      "step": 819,
      "training_step_time": 10.342413187026978
    },
    {
      "epoch": 3.1280517578125e-06,
      "grad_norm": 0.684550404548645,
      "learning_rate": 1e-05,
      "loss": 0.1366,
      "step": 820
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 3.183436155319214,
      "step": 820
    },
    {
      "dynamics_loss_avg": 0.07174173947423697,
      "epoch": 3.1280517578125e-06,
      "step": 820
    },
    {
      "action_loss_avg": 0.06351620592176914,
      "epoch": 3.1280517578125e-06,
      "step": 820
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 820,
      "training_step_time": 9.956322431564331
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 3.1418142318725586,
      "step": 820
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 820,
      "training_step_time": 9.91335654258728
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 3.13523530960083,
      "step": 820
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 820,
      "training_step_time": 9.911635637283325
    },
    {
      "epoch": 3.1280517578125e-06,
      "model_forward_time": 3.1489086151123047,
      "step": 820
    },
    {
      "epoch": 3.1280517578125e-06,
      "step": 820,
      "training_step_time": 10.253592491149902
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 3.4338879585266113,
      "step": 821
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 821,
      "training_step_time": 10.218127489089966
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 3.1434574127197266,
      "step": 821
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 821,
      "training_step_time": 9.920817613601685
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 3.142383337020874,
      "step": 821
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 821,
      "training_step_time": 9.925616264343262
    },
    {
      "epoch": 3.131866455078125e-06,
      "model_forward_time": 3.144117832183838,
      "step": 821
    },
    {
      "epoch": 3.131866455078125e-06,
      "step": 821,
      "training_step_time": 10.290698528289795
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 3.176173210144043,
      "step": 822
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 822,
      "training_step_time": 9.95751428604126
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 3.428830146789551,
      "step": 822
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 822,
      "training_step_time": 10.206063985824585
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 3.1401236057281494,
      "step": 822
    },
    {
      "dynamics_loss_avg": 0.07343056723475457,
      "epoch": 3.13568115234375e-06,
      "step": 822
    },
    {
      "action_loss_avg": 0.0664772717282176,
      "epoch": 3.13568115234375e-06,
      "step": 822
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 822,
      "training_step_time": 9.922370672225952
    },
    {
      "epoch": 3.13568115234375e-06,
      "model_forward_time": 3.14292049407959,
      "step": 822
    },
    {
      "epoch": 3.13568115234375e-06,
      "step": 822,
      "training_step_time": 10.273477554321289
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 3.1802093982696533,
      "step": 823
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 823,
      "training_step_time": 9.966744184494019
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 3.1421337127685547,
      "step": 823
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 823,
      "training_step_time": 9.91236686706543
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 3.4261908531188965,
      "step": 823
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 823,
      "training_step_time": 10.192009449005127
    },
    {
      "epoch": 3.139495849609375e-06,
      "model_forward_time": 3.1437673568725586,
      "step": 823
    },
    {
      "epoch": 3.139495849609375e-06,
      "step": 823,
      "training_step_time": 10.249578952789307
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 2.470273017883301,
      "step": 824
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 824,
      "training_step_time": 7.494130849838257
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 2.427633762359619,
      "step": 824
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 824,
      "training_step_time": 7.453412294387817
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 3.128333568572998,
      "step": 824
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 824,
      "training_step_time": 9.893277168273926
    },
    {
      "epoch": 3.143310546875e-06,
      "model_forward_time": 3.1263363361358643,
      "step": 824
    },
    {
      "epoch": 3.143310546875e-06,
      "step": 824,
      "training_step_time": 10.008904695510864
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 3.4631662368774414,
      "step": 825
    },
    {
      "dynamics_loss_avg": 0.09128062315285206,
      "epoch": 3.147125244140625e-06,
      "step": 825
    },
    {
      "action_loss_avg": 0.06937326118350029,
      "epoch": 3.147125244140625e-06,
      "step": 825
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 825,
      "training_step_time": 10.229027509689331
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 3.1265547275543213,
      "step": 825
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 825,
      "training_step_time": 9.887266397476196
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 3.127718925476074,
      "step": 825
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 825,
      "training_step_time": 9.887141704559326
    },
    {
      "epoch": 3.147125244140625e-06,
      "model_forward_time": 3.121668577194214,
      "step": 825
    },
    {
      "epoch": 3.147125244140625e-06,
      "step": 825,
      "training_step_time": 10.241403102874756
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 3.1617112159729004,
      "step": 826
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 826,
      "training_step_time": 10.208457469940186
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 3.1292877197265625,
      "step": 826
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 826,
      "training_step_time": 9.893553972244263
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 3.130998134613037,
      "step": 826
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 826,
      "training_step_time": 9.89620304107666
    },
    {
      "epoch": 3.15093994140625e-06,
      "model_forward_time": 3.126971960067749,
      "step": 826
    },
    {
      "epoch": 3.15093994140625e-06,
      "step": 826,
      "training_step_time": 10.277208089828491
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 3.166048288345337,
      "step": 827
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 827,
      "training_step_time": 9.935351371765137
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 3.387728452682495,
      "step": 827
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 827,
      "training_step_time": 10.158894538879395
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 3.1353917121887207,
      "step": 827
    },
    {
      "dynamics_loss_avg": 0.04642113968729973,
      "epoch": 3.154754638671875e-06,
      "step": 827
    },
    {
      "action_loss_avg": 0.03511959700845182,
      "epoch": 3.154754638671875e-06,
      "step": 827
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 827,
      "training_step_time": 9.910506248474121
    },
    {
      "epoch": 3.154754638671875e-06,
      "model_forward_time": 3.1320760250091553,
      "step": 827
    },
    {
      "epoch": 3.154754638671875e-06,
      "step": 827,
      "training_step_time": 10.291305541992188
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 3.1510581970214844,
      "step": 828
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 828,
      "training_step_time": 9.928321838378906
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 3.137216567993164,
      "step": 828
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 828,
      "training_step_time": 9.921012878417969
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 2.4335789680480957,
      "step": 828
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 828,
      "training_step_time": 7.764404296875
    },
    {
      "epoch": 3.1585693359375e-06,
      "model_forward_time": 3.132662296295166,
      "step": 828
    },
    {
      "epoch": 3.1585693359375e-06,
      "step": 828,
      "training_step_time": 10.284125089645386
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 3.1556549072265625,
      "step": 829
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 829,
      "training_step_time": 9.935405969619751
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 3.136597156524658,
      "step": 829
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 829,
      "training_step_time": 9.90862226486206
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 2.430656909942627,
      "step": 829
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 829,
      "training_step_time": 7.45935845375061
    },
    {
      "epoch": 3.162384033203125e-06,
      "model_forward_time": 2.4242465496063232,
      "step": 829
    },
    {
      "epoch": 3.162384033203125e-06,
      "step": 829,
      "training_step_time": 10.282674074172974
    },
    {
      "epoch": 3.16619873046875e-06,
      "grad_norm": 0.6843839883804321,
      "learning_rate": 1e-05,
      "loss": 0.1299,
      "step": 830
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 3.1708929538726807,
      "step": 830
    },
    {
      "dynamics_loss_avg": 0.08406151756644249,
      "epoch": 3.16619873046875e-06,
      "step": 830
    },
    {
      "action_loss_avg": 0.06363615375012159,
      "epoch": 3.16619873046875e-06,
      "step": 830
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 830,
      "training_step_time": 9.947570323944092
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 3.1318249702453613,
      "step": 830
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 830,
      "training_step_time": 9.903316974639893
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 3.1336429119110107,
      "step": 830
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 830,
      "training_step_time": 9.901373863220215
    },
    {
      "epoch": 3.16619873046875e-06,
      "model_forward_time": 3.1350107192993164,
      "step": 830
    },
    {
      "epoch": 3.16619873046875e-06,
      "step": 830,
      "training_step_time": 10.249821901321411
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 3.168858528137207,
      "step": 831
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 831,
      "training_step_time": 10.210606098175049
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 3.136697292327881,
      "step": 831
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 831,
      "training_step_time": 9.907855033874512
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 3.1284611225128174,
      "step": 831
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 831,
      "training_step_time": 9.89855146408081
    },
    {
      "epoch": 3.170013427734375e-06,
      "model_forward_time": 3.1342735290527344,
      "step": 831
    },
    {
      "epoch": 3.170013427734375e-06,
      "step": 831,
      "training_step_time": 10.230836391448975
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 2.4584739208221436,
      "step": 832
    },
    {
      "epoch": 3.173828125e-06,
      "step": 832,
      "training_step_time": 7.484388589859009
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 3.4346702098846436,
      "step": 832
    },
    {
      "epoch": 3.173828125e-06,
      "step": 832,
      "training_step_time": 10.200430631637573
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 3.1313602924346924,
      "step": 832
    },
    {
      "dynamics_loss_avg": 0.07515807561576367,
      "epoch": 3.173828125e-06,
      "step": 832
    },
    {
      "action_loss_avg": 0.07235279083251953,
      "epoch": 3.173828125e-06,
      "step": 832
    },
    {
      "epoch": 3.173828125e-06,
      "step": 832,
      "training_step_time": 9.903200626373291
    },
    {
      "epoch": 3.173828125e-06,
      "model_forward_time": 3.135476589202881,
      "step": 832
    },
    {
      "epoch": 3.173828125e-06,
      "step": 832,
      "training_step_time": 10.32419466972351
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 3.1726183891296387,
      "step": 833
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 833,
      "training_step_time": 9.94867205619812
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 3.130887269973755,
      "step": 833
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 833,
      "training_step_time": 9.89658236503601
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 3.132780075073242,
      "step": 833
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 833,
      "training_step_time": 9.898892641067505
    },
    {
      "epoch": 3.177642822265625e-06,
      "model_forward_time": 2.6989858150482178,
      "step": 833
    },
    {
      "epoch": 3.177642822265625e-06,
      "step": 833,
      "training_step_time": 10.290747165679932
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 3.148824691772461,
      "step": 834
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 834,
      "training_step_time": 9.917560577392578
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 3.138505220413208,
      "step": 834
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 834,
      "training_step_time": 9.90916395187378
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 2.4346563816070557,
      "step": 834
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 834,
      "training_step_time": 7.463524341583252
    },
    {
      "epoch": 3.18145751953125e-06,
      "model_forward_time": 2.4289698600769043,
      "step": 834
    },
    {
      "epoch": 3.18145751953125e-06,
      "step": 834,
      "training_step_time": 10.262027978897095
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 3.1756579875946045,
      "step": 835
    },
    {
      "dynamics_loss_avg": 0.06234807837754488,
      "epoch": 3.185272216796875e-06,
      "step": 835
    },
    {
      "action_loss_avg": 0.06747943200170994,
      "epoch": 3.185272216796875e-06,
      "step": 835
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 835,
      "training_step_time": 9.949829339981079
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 2.4314308166503906,
      "step": 835
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 835,
      "training_step_time": 7.459913015365601
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 3.1353278160095215,
      "step": 835
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 835,
      "training_step_time": 9.915852546691895
    },
    {
      "epoch": 3.185272216796875e-06,
      "model_forward_time": 3.1309947967529297,
      "step": 835
    },
    {
      "epoch": 3.185272216796875e-06,
      "step": 835,
      "training_step_time": 10.276331186294556
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 3.1890456676483154,
      "step": 836
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 836,
      "training_step_time": 10.247915267944336
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 3.1336021423339844,
      "step": 836
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 836,
      "training_step_time": 9.909570455551147
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 3.1357808113098145,
      "step": 836
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 836,
      "training_step_time": 9.910327672958374
    },
    {
      "epoch": 3.1890869140625e-06,
      "model_forward_time": 2.4308319091796875,
      "step": 836
    },
    {
      "epoch": 3.1890869140625e-06,
      "step": 836,
      "training_step_time": 10.241132974624634
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 3.1769304275512695,
      "step": 837
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 837,
      "training_step_time": 9.952803134918213
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 3.1332406997680664,
      "step": 837
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 837,
      "training_step_time": 10.196598052978516
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 3.1328227519989014,
      "step": 837
    },
    {
      "dynamics_loss_avg": 0.0697250071913004,
      "epoch": 3.192901611328125e-06,
      "step": 837
    },
    {
      "action_loss_avg": 0.06544937877915799,
      "epoch": 3.192901611328125e-06,
      "step": 837
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 837,
      "training_step_time": 9.90486478805542
    },
    {
      "epoch": 3.192901611328125e-06,
      "model_forward_time": 3.1289114952087402,
      "step": 837
    },
    {
      "epoch": 3.192901611328125e-06,
      "step": 837,
      "training_step_time": 10.286949157714844
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 3.1620874404907227,
      "step": 838
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 838,
      "training_step_time": 9.930508136749268
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 3.1298162937164307,
      "step": 838
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 838,
      "training_step_time": 9.906893014907837
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 3.40972900390625,
      "step": 838
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 838,
      "training_step_time": 10.180405855178833
    },
    {
      "epoch": 3.19671630859375e-06,
      "model_forward_time": 3.134701728820801,
      "step": 838
    },
    {
      "epoch": 3.19671630859375e-06,
      "step": 838,
      "training_step_time": 10.291886568069458
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 3.177870512008667,
      "step": 839
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 839,
      "training_step_time": 9.952631711959839
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 3.1392109394073486,
      "step": 839
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 839,
      "training_step_time": 9.913235902786255
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 3.1348588466644287,
      "step": 839
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 839,
      "training_step_time": 9.91206955909729
    },
    {
      "epoch": 3.200531005859375e-06,
      "model_forward_time": 2.43430495262146,
      "step": 839
    },
    {
      "epoch": 3.200531005859375e-06,
      "step": 839,
      "training_step_time": 10.308353185653687
    },
    {
      "epoch": 3.204345703125e-06,
      "grad_norm": 0.8698252439498901,
      "learning_rate": 1e-05,
      "loss": 0.1298,
      "step": 840
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 3.4333508014678955,
      "step": 840
    },
    {
      "dynamics_loss_avg": 0.054074503108859064,
      "epoch": 3.204345703125e-06,
      "step": 840
    },
    {
      "action_loss_avg": 0.05273332875221968,
      "epoch": 3.204345703125e-06,
      "step": 840
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 840,
      "training_step_time": 10.211808681488037
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 3.13576340675354,
      "step": 840
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 840,
      "training_step_time": 9.909019470214844
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 3.1342339515686035,
      "step": 840
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 840,
      "training_step_time": 9.90347695350647
    },
    {
      "epoch": 3.204345703125e-06,
      "model_forward_time": 2.4289636611938477,
      "step": 840
    },
    {
      "epoch": 3.204345703125e-06,
      "step": 840,
      "training_step_time": 10.265228033065796
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 3.4277405738830566,
      "step": 841
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 841,
      "training_step_time": 10.197117805480957
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 3.141554832458496,
      "step": 841
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 841,
      "training_step_time": 9.918221235275269
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 3.1348352432250977,
      "step": 841
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 841,
      "training_step_time": 9.902099609375
    },
    {
      "epoch": 3.208160400390625e-06,
      "model_forward_time": 3.1366140842437744,
      "step": 841
    },
    {
      "epoch": 3.208160400390625e-06,
      "step": 841,
      "training_step_time": 10.290626287460327
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 3.1642954349517822,
      "step": 842
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 842,
      "training_step_time": 9.940983295440674
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 3.1365795135498047,
      "step": 842
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 842,
      "training_step_time": 10.20466685295105
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 3.1361067295074463,
      "step": 842
    },
    {
      "dynamics_loss_avg": 0.060406264290213584,
      "epoch": 3.21197509765625e-06,
      "step": 842
    },
    {
      "action_loss_avg": 0.04516446571797132,
      "epoch": 3.21197509765625e-06,
      "step": 842
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 842,
      "training_step_time": 9.918776750564575
    },
    {
      "epoch": 3.21197509765625e-06,
      "model_forward_time": 3.141191005706787,
      "step": 842
    },
    {
      "epoch": 3.21197509765625e-06,
      "step": 842,
      "training_step_time": 10.288955926895142
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 3.181305408477783,
      "step": 843
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 843,
      "training_step_time": 9.958743333816528
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 3.138153076171875,
      "step": 843
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 843,
      "training_step_time": 9.91710376739502
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 3.3958966732025146,
      "step": 843
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 843,
      "training_step_time": 10.173124313354492
    },
    {
      "epoch": 3.215789794921875e-06,
      "model_forward_time": 3.141893148422241,
      "step": 843
    },
    {
      "epoch": 3.215789794921875e-06,
      "step": 843,
      "training_step_time": 10.282432079315186
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 3.1786081790924072,
      "step": 844
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 844,
      "training_step_time": 9.952385902404785
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 3.1341843605041504,
      "step": 844
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 844,
      "training_step_time": 9.911619186401367
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 3.1377956867218018,
      "step": 844
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 844,
      "training_step_time": 9.914271593093872
    },
    {
      "epoch": 3.2196044921875e-06,
      "model_forward_time": 3.4121172428131104,
      "step": 844
    },
    {
      "epoch": 3.2196044921875e-06,
      "step": 844,
      "training_step_time": 10.286082983016968
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 3.1646714210510254,
      "step": 845
    },
    {
      "dynamics_loss_avg": 0.07246801555156708,
      "epoch": 3.223419189453125e-06,
      "step": 845
    },
    {
      "action_loss_avg": 0.06612088400870561,
      "epoch": 3.223419189453125e-06,
      "step": 845
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 845,
      "training_step_time": 9.939570665359497
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 3.1360886096954346,
      "step": 845
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 845,
      "training_step_time": 9.914722204208374
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 3.138148069381714,
      "step": 845
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 845,
      "training_step_time": 9.913619995117188
    },
    {
      "epoch": 3.223419189453125e-06,
      "model_forward_time": 3.1373355388641357,
      "step": 845
    },
    {
      "epoch": 3.223419189453125e-06,
      "step": 845,
      "training_step_time": 10.039530515670776
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 3.1815690994262695,
      "step": 846
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 846,
      "training_step_time": 10.245129823684692
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 3.1385083198547363,
      "step": 846
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 846,
      "training_step_time": 9.91866397857666
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 3.1404404640197754,
      "step": 846
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 846,
      "training_step_time": 9.927772521972656
    },
    {
      "epoch": 3.22723388671875e-06,
      "model_forward_time": 3.1440162658691406,
      "step": 846
    },
    {
      "epoch": 3.22723388671875e-06,
      "step": 846,
      "training_step_time": 10.278801918029785
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 3.1707217693328857,
      "step": 847
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 847,
      "training_step_time": 9.945343255996704
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 3.4052863121032715,
      "step": 847
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 847,
      "training_step_time": 10.176445722579956
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 3.1440279483795166,
      "step": 847
    },
    {
      "dynamics_loss_avg": 0.06959487963467836,
      "epoch": 3.231048583984375e-06,
      "step": 847
    },
    {
      "action_loss_avg": 0.05166332251392305,
      "epoch": 3.231048583984375e-06,
      "step": 847
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 847,
      "training_step_time": 9.91828465461731
    },
    {
      "epoch": 3.231048583984375e-06,
      "model_forward_time": 3.141725540161133,
      "step": 847
    },
    {
      "epoch": 3.231048583984375e-06,
      "step": 847,
      "training_step_time": 10.267102003097534
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 3.170034885406494,
      "step": 848
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 848,
      "training_step_time": 9.946926593780518
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 3.1394760608673096,
      "step": 848
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 848,
      "training_step_time": 9.912586450576782
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 3.144896984100342,
      "step": 848
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 848,
      "training_step_time": 10.219226598739624
    },
    {
      "epoch": 3.23486328125e-06,
      "model_forward_time": 3.13774037361145,
      "step": 848
    },
    {
      "epoch": 3.23486328125e-06,
      "step": 848,
      "training_step_time": 10.264469146728516
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 3.1582577228546143,
      "step": 849
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 849,
      "training_step_time": 9.930532455444336
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 3.137428045272827,
      "step": 849
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 849,
      "training_step_time": 9.90876817703247
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 3.137139320373535,
      "step": 849
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 849,
      "training_step_time": 9.914743661880493
    },
    {
      "epoch": 3.238677978515625e-06,
      "model_forward_time": 2.436270236968994,
      "step": 849
    },
    {
      "epoch": 3.238677978515625e-06,
      "step": 849,
      "training_step_time": 10.302180528640747
    },
    {
      "epoch": 3.24249267578125e-06,
      "grad_norm": 0.5386472940444946,
      "learning_rate": 1e-05,
      "loss": 0.136,
      "step": 850
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 3.1765787601470947,
      "step": 850
    },
    {
      "dynamics_loss_avg": 0.09637323692440987,
      "epoch": 3.24249267578125e-06,
      "step": 850
    },
    {
      "action_loss_avg": 0.07427072897553444,
      "epoch": 3.24249267578125e-06,
      "step": 850
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 850,
      "training_step_time": 9.949204444885254
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 3.1430132389068604,
      "step": 850
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 850,
      "training_step_time": 9.91028118133545
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 3.1324679851531982,
      "step": 850
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 850,
      "training_step_time": 9.896984100341797
    },
    {
      "epoch": 3.24249267578125e-06,
      "model_forward_time": 3.1358964443206787,
      "step": 850
    },
    {
      "epoch": 3.24249267578125e-06,
      "step": 850,
      "training_step_time": 10.008665084838867
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 3.144211530685425,
      "step": 851
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 851,
      "training_step_time": 10.188711166381836
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 3.137166738510132,
      "step": 851
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 851,
      "training_step_time": 9.906891822814941
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 2.4328067302703857,
      "step": 851
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 851,
      "training_step_time": 7.4628636837005615
    },
    {
      "epoch": 3.246307373046875e-06,
      "model_forward_time": 3.1372523307800293,
      "step": 851
    },
    {
      "epoch": 3.246307373046875e-06,
      "step": 851,
      "training_step_time": 10.026587009429932
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 2.485624313354492,
      "step": 852
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 852,
      "training_step_time": 7.515066862106323
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 2.435711145401001,
      "step": 852
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 852,
      "training_step_time": 7.47128438949585
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 3.392801284790039,
      "step": 852
    },
    {
      "dynamics_loss_avg": 0.06644020956009626,
      "epoch": 3.2501220703125e-06,
      "step": 852
    },
    {
      "action_loss_avg": 0.051688785292208196,
      "epoch": 3.2501220703125e-06,
      "step": 852
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 852,
      "training_step_time": 10.172636985778809
    },
    {
      "epoch": 3.2501220703125e-06,
      "model_forward_time": 3.1478540897369385,
      "step": 852
    },
    {
      "epoch": 3.2501220703125e-06,
      "step": 852,
      "training_step_time": 10.119757890701294
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 3.194077491760254,
      "step": 853
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 853,
      "training_step_time": 9.968744277954102
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 2.436044931411743,
      "step": 853
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 853,
      "training_step_time": 7.465408802032471
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 3.1382229328155518,
      "step": 853
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 853,
      "training_step_time": 9.914992094039917
    },
    {
      "epoch": 3.253936767578125e-06,
      "model_forward_time": 3.4162778854370117,
      "step": 853
    },
    {
      "epoch": 3.253936767578125e-06,
      "step": 853,
      "training_step_time": 10.295165777206421
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 3.1752822399139404,
      "step": 854
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 854,
      "training_step_time": 9.952315092086792
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 3.1420116424560547,
      "step": 854
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 854,
      "training_step_time": 9.914699077606201
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 3.1420671939849854,
      "step": 854
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 854,
      "training_step_time": 9.912903070449829
    },
    {
      "epoch": 3.25775146484375e-06,
      "model_forward_time": 3.1401920318603516,
      "step": 854
    },
    {
      "epoch": 3.25775146484375e-06,
      "step": 854,
      "training_step_time": 10.299466371536255
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 2.4780313968658447,
      "step": 855
    },
    {
      "dynamics_loss_avg": 0.08073956780135631,
      "epoch": 3.261566162109375e-06,
      "step": 855
    },
    {
      "action_loss_avg": 0.055341472290456294,
      "epoch": 3.261566162109375e-06,
      "step": 855
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 855,
      "training_step_time": 7.50726056098938
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 3.1382579803466797,
      "step": 855
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 855,
      "training_step_time": 9.911195755004883
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 3.1441738605499268,
      "step": 855
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 855,
      "training_step_time": 9.915703535079956
    },
    {
      "epoch": 3.261566162109375e-06,
      "model_forward_time": 3.130871057510376,
      "step": 855
    },
    {
      "epoch": 3.261566162109375e-06,
      "step": 855,
      "training_step_time": 10.251540184020996
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 3.1860883235931396,
      "step": 856
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 856,
      "training_step_time": 10.233481168746948
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 3.130209445953369,
      "step": 856
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 856,
      "training_step_time": 9.9015953540802
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 3.1310763359069824,
      "step": 856
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 856,
      "training_step_time": 9.904330730438232
    },
    {
      "epoch": 3.265380859375e-06,
      "model_forward_time": 3.135133743286133,
      "step": 856
    },
    {
      "epoch": 3.265380859375e-06,
      "step": 856,
      "training_step_time": 10.01044774055481
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 3.1573712825775146,
      "step": 857
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 857,
      "training_step_time": 9.931042909622192
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 3.134605884552002,
      "step": 857
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 857,
      "training_step_time": 10.18062686920166
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 3.1344571113586426,
      "step": 857
    },
    {
      "dynamics_loss_avg": 0.07851723786443472,
      "epoch": 3.269195556640625e-06,
      "step": 857
    },
    {
      "action_loss_avg": 0.05811037421226502,
      "epoch": 3.269195556640625e-06,
      "step": 857
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 857,
      "training_step_time": 9.911990880966187
    },
    {
      "epoch": 3.269195556640625e-06,
      "model_forward_time": 3.1368794441223145,
      "step": 857
    },
    {
      "epoch": 3.269195556640625e-06,
      "step": 857,
      "training_step_time": 10.01810097694397
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 2.4479968547821045,
      "step": 858
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 858,
      "training_step_time": 7.484152793884277
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 3.132267475128174,
      "step": 858
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 858,
      "training_step_time": 9.907830238342285
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 2.4314463138580322,
      "step": 858
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 858,
      "training_step_time": 7.462311267852783
    },
    {
      "epoch": 3.27301025390625e-06,
      "model_forward_time": 2.6931207180023193,
      "step": 858
    },
    {
      "epoch": 3.27301025390625e-06,
      "step": 858,
      "training_step_time": 10.300519227981567
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 3.1678073406219482,
      "step": 859
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 859,
      "training_step_time": 9.943264484405518
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 3.1353495121002197,
      "step": 859
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 859,
      "training_step_time": 9.91215181350708
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 3.136078357696533,
      "step": 859
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 859,
      "training_step_time": 9.91532564163208
    },
    {
      "epoch": 3.276824951171875e-06,
      "model_forward_time": 3.140773296356201,
      "step": 859
    },
    {
      "epoch": 3.276824951171875e-06,
      "step": 859,
      "training_step_time": 10.283586025238037
    },
    {
      "epoch": 3.2806396484375e-06,
      "grad_norm": 0.7311995029449463,
      "learning_rate": 1e-05,
      "loss": 0.1358,
      "step": 860
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 2.807060480117798,
      "step": 860
    },
    {
      "dynamics_loss_avg": 0.07449707761406898,
      "epoch": 3.2806396484375e-06,
      "step": 860
    },
    {
      "action_loss_avg": 0.054059740342199805,
      "epoch": 3.2806396484375e-06,
      "step": 860
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 860,
      "training_step_time": 7.873171091079712
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 3.1339759826660156,
      "step": 860
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 860,
      "training_step_time": 9.907551288604736
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 3.1324286460876465,
      "step": 860
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 860,
      "training_step_time": 9.907718181610107
    },
    {
      "epoch": 3.2806396484375e-06,
      "model_forward_time": 3.135221481323242,
      "step": 860
    },
    {
      "epoch": 3.2806396484375e-06,
      "step": 860,
      "training_step_time": 10.274728775024414
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 3.1744980812072754,
      "step": 861
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 861,
      "training_step_time": 9.93701958656311
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 3.4054203033447266,
      "step": 861
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 861,
      "training_step_time": 10.172524452209473
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 2.4361252784729004,
      "step": 861
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 861,
      "training_step_time": 7.467738389968872
    },
    {
      "epoch": 3.284454345703125e-06,
      "model_forward_time": 3.1342380046844482,
      "step": 861
    },
    {
      "epoch": 3.284454345703125e-06,
      "step": 861,
      "training_step_time": 10.014539241790771
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 2.445392608642578,
      "step": 862
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 862,
      "training_step_time": 7.473852157592773
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 2.4304513931274414,
      "step": 862
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 862,
      "training_step_time": 7.460359573364258
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 3.387036085128784,
      "step": 862
    },
    {
      "dynamics_loss_avg": 0.07793133575469255,
      "epoch": 3.28826904296875e-06,
      "step": 862
    },
    {
      "action_loss_avg": 0.06798314303159714,
      "epoch": 3.28826904296875e-06,
      "step": 862
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 862,
      "training_step_time": 10.162455558776855
    },
    {
      "epoch": 3.28826904296875e-06,
      "model_forward_time": 2.4376933574676514,
      "step": 862
    },
    {
      "epoch": 3.28826904296875e-06,
      "step": 862,
      "training_step_time": 10.294212102890015
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 3.185811758041382,
      "step": 863
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 863,
      "training_step_time": 9.961787462234497
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 3.1358256340026855,
      "step": 863
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 863,
      "training_step_time": 9.912246942520142
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 3.1358230113983154,
      "step": 863
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 863,
      "training_step_time": 9.918586015701294
    },
    {
      "epoch": 3.292083740234375e-06,
      "model_forward_time": 3.439771890640259,
      "step": 863
    },
    {
      "epoch": 3.292083740234375e-06,
      "step": 863,
      "training_step_time": 10.30883240699768
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 3.187675952911377,
      "step": 864
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 864,
      "training_step_time": 9.961964130401611
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 3.1355819702148438,
      "step": 864
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 864,
      "training_step_time": 9.906253814697266
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 3.1375489234924316,
      "step": 864
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 864,
      "training_step_time": 9.908663272857666
    },
    {
      "epoch": 3.2958984375e-06,
      "model_forward_time": 3.1381099224090576,
      "step": 864
    },
    {
      "epoch": 3.2958984375e-06,
      "step": 864,
      "training_step_time": 10.310532808303833
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 3.453860282897949,
      "step": 865
    },
    {
      "dynamics_loss_avg": 0.07171396985650062,
      "epoch": 3.299713134765625e-06,
      "step": 865
    },
    {
      "action_loss_avg": 0.045145788230001926,
      "epoch": 3.299713134765625e-06,
      "step": 865
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 865,
      "training_step_time": 10.237462282180786
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 3.136483669281006,
      "step": 865
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 865,
      "training_step_time": 9.910315752029419
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 3.1355950832366943,
      "step": 865
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 865,
      "training_step_time": 9.906696557998657
    },
    {
      "epoch": 3.299713134765625e-06,
      "model_forward_time": 3.1295111179351807,
      "step": 865
    },
    {
      "epoch": 3.299713134765625e-06,
      "step": 865,
      "training_step_time": 10.302033185958862
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 3.176408052444458,
      "step": 866
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 866,
      "training_step_time": 9.944432735443115
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 3.3744330406188965,
      "step": 866
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 866,
      "training_step_time": 10.141990423202515
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 3.1338281631469727,
      "step": 866
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 866,
      "training_step_time": 9.904392957687378
    },
    {
      "epoch": 3.30352783203125e-06,
      "model_forward_time": 3.130551338195801,
      "step": 866
    },
    {
      "epoch": 3.30352783203125e-06,
      "step": 866,
      "training_step_time": 10.014644622802734
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 3.165898084640503,
      "step": 867
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 867,
      "training_step_time": 9.933499813079834
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 3.132627248764038,
      "step": 867
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 867,
      "training_step_time": 9.896297454833984
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 3.4377076625823975,
      "step": 867
    },
    {
      "dynamics_loss_avg": 0.058405916392803195,
      "epoch": 3.307342529296875e-06,
      "step": 867
    },
    {
      "action_loss_avg": 0.06637498922646046,
      "epoch": 3.307342529296875e-06,
      "step": 867
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 867,
      "training_step_time": 10.20097827911377
    },
    {
      "epoch": 3.307342529296875e-06,
      "model_forward_time": 3.132188081741333,
      "step": 867
    },
    {
      "epoch": 3.307342529296875e-06,
      "step": 867,
      "training_step_time": 10.288209676742554
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 3.169285297393799,
      "step": 868
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 868,
      "training_step_time": 9.936401844024658
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 2.4320785999298096,
      "step": 868
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 868,
      "training_step_time": 7.454563856124878
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 3.128058671951294,
      "step": 868
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 868,
      "training_step_time": 9.896980047225952
    },
    {
      "epoch": 3.3111572265625e-06,
      "model_forward_time": 2.7106165885925293,
      "step": 868
    },
    {
      "epoch": 3.3111572265625e-06,
      "step": 868,
      "training_step_time": 10.245002269744873
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 3.182562828063965,
      "step": 869
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 869,
      "training_step_time": 9.950560092926025
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 3.1416008472442627,
      "step": 869
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 869,
      "training_step_time": 9.910928010940552
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 3.136488676071167,
      "step": 869
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 869,
      "training_step_time": 9.909677028656006
    },
    {
      "epoch": 3.314971923828125e-06,
      "model_forward_time": 3.13811993598938,
      "step": 869
    },
    {
      "epoch": 3.314971923828125e-06,
      "step": 869,
      "training_step_time": 10.315843343734741
    },
    {
      "epoch": 3.31878662109375e-06,
      "grad_norm": 0.7851423025131226,
      "learning_rate": 1e-05,
      "loss": 0.1323,
      "step": 870
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 3.4192872047424316,
      "step": 870
    },
    {
      "dynamics_loss_avg": 0.05964666996151209,
      "epoch": 3.31878662109375e-06,
      "step": 870
    },
    {
      "action_loss_avg": 0.053596246521919966,
      "epoch": 3.31878662109375e-06,
      "step": 870
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 870,
      "training_step_time": 10.192297458648682
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 3.138549566268921,
      "step": 870
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 870,
      "training_step_time": 9.90662956237793
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 3.136451244354248,
      "step": 870
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 870,
      "training_step_time": 9.909328699111938
    },
    {
      "epoch": 3.31878662109375e-06,
      "model_forward_time": 3.137455463409424,
      "step": 870
    },
    {
      "epoch": 3.31878662109375e-06,
      "step": 870,
      "training_step_time": 10.311658143997192
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 2.467585563659668,
      "step": 871
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 871,
      "training_step_time": 7.4927287101745605
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 3.420586109161377,
      "step": 871
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 871,
      "training_step_time": 10.181130647659302
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 3.1333413124084473,
      "step": 871
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 871,
      "training_step_time": 9.89782452583313
    },
    {
      "epoch": 3.322601318359375e-06,
      "model_forward_time": 3.1353931427001953,
      "step": 871
    },
    {
      "epoch": 3.322601318359375e-06,
      "step": 871,
      "training_step_time": 10.010318517684937
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 3.159440755844116,
      "step": 872
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 872,
      "training_step_time": 9.926698207855225
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 3.1305031776428223,
      "step": 872
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 872,
      "training_step_time": 9.893434762954712
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 3.4226629734039307,
      "step": 872
    },
    {
      "dynamics_loss_avg": 0.06668497771024703,
      "epoch": 3.326416015625e-06,
      "step": 872
    },
    {
      "action_loss_avg": 0.055611195601522924,
      "epoch": 3.326416015625e-06,
      "step": 872
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 872,
      "training_step_time": 10.194926738739014
    },
    {
      "epoch": 3.326416015625e-06,
      "model_forward_time": 3.1398396492004395,
      "step": 872
    },
    {
      "epoch": 3.326416015625e-06,
      "step": 872,
      "training_step_time": 10.26342511177063
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 3.1847100257873535,
      "step": 873
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 873,
      "training_step_time": 9.955845355987549
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 3.1358578205108643,
      "step": 873
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 873,
      "training_step_time": 9.906452178955078
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 3.139904260635376,
      "step": 873
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 873,
      "training_step_time": 9.913615942001343
    },
    {
      "epoch": 3.330230712890625e-06,
      "model_forward_time": 3.446964740753174,
      "step": 873
    },
    {
      "epoch": 3.330230712890625e-06,
      "step": 873,
      "training_step_time": 10.322831153869629
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 3.177114963531494,
      "step": 874
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 874,
      "training_step_time": 9.948589086532593
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 3.143183946609497,
      "step": 874
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 874,
      "training_step_time": 9.913150548934937
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 3.1433675289154053,
      "step": 874
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 874,
      "training_step_time": 9.915061473846436
    },
    {
      "epoch": 3.33404541015625e-06,
      "model_forward_time": 3.1431262493133545,
      "step": 874
    },
    {
      "epoch": 3.33404541015625e-06,
      "step": 874,
      "training_step_time": 10.305747985839844
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 3.453226089477539,
      "step": 875
    },
    {
      "dynamics_loss_avg": 0.06366600356996059,
      "epoch": 3.337860107421875e-06,
      "step": 875
    },
    {
      "action_loss_avg": 0.06501233465969562,
      "epoch": 3.337860107421875e-06,
      "step": 875
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 875,
      "training_step_time": 10.229858160018921
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 3.142123222351074,
      "step": 875
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 875,
      "training_step_time": 9.913728475570679
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 2.4373323917388916,
      "step": 875
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 875,
      "training_step_time": 7.467972993850708
    },
    {
      "epoch": 3.337860107421875e-06,
      "model_forward_time": 3.1404078006744385,
      "step": 875
    },
    {
      "epoch": 3.337860107421875e-06,
      "step": 875,
      "training_step_time": 10.03142786026001
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 2.4793331623077393,
      "step": 876
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 876,
      "training_step_time": 7.50633430480957
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 2.443082332611084,
      "step": 876
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 876,
      "training_step_time": 7.473698854446411
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 3.4658243656158447,
      "step": 876
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 876,
      "training_step_time": 10.23850131034851
    },
    {
      "epoch": 3.3416748046875e-06,
      "model_forward_time": 3.1418192386627197,
      "step": 876
    },
    {
      "epoch": 3.3416748046875e-06,
      "step": 876,
      "training_step_time": 10.302144050598145
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 3.177988052368164,
      "step": 877
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 877,
      "training_step_time": 9.948721647262573
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 3.1417171955108643,
      "step": 877
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 877,
      "training_step_time": 9.913622379302979
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 3.1423892974853516,
      "step": 877
    },
    {
      "dynamics_loss_avg": 0.07777012027800083,
      "epoch": 3.345489501953125e-06,
      "step": 877
    },
    {
      "action_loss_avg": 0.0637652862817049,
      "epoch": 3.345489501953125e-06,
      "step": 877
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 877,
      "training_step_time": 9.91953158378601
    },
    {
      "epoch": 3.345489501953125e-06,
      "model_forward_time": 3.4012131690979004,
      "step": 877
    },
    {
      "epoch": 3.345489501953125e-06,
      "step": 877,
      "training_step_time": 10.268749237060547
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 2.4705193042755127,
      "step": 878
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 878,
      "training_step_time": 7.501302242279053
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 3.1431725025177,
      "step": 878
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 878,
      "training_step_time": 9.910286664962769
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 3.137305498123169,
      "step": 878
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 878,
      "training_step_time": 9.90072226524353
    },
    {
      "epoch": 3.34930419921875e-06,
      "model_forward_time": 3.1405253410339355,
      "step": 878
    },
    {
      "epoch": 3.34930419921875e-06,
      "step": 878,
      "training_step_time": 10.260879516601562
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 3.466871738433838,
      "step": 879
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 879,
      "training_step_time": 10.23913836479187
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 3.138761281967163,
      "step": 879
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 879,
      "training_step_time": 9.907501459121704
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 3.1389365196228027,
      "step": 879
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 879,
      "training_step_time": 9.909634113311768
    },
    {
      "epoch": 3.353118896484375e-06,
      "model_forward_time": 3.144932985305786,
      "step": 879
    },
    {
      "epoch": 3.353118896484375e-06,
      "step": 879,
      "training_step_time": 10.282116413116455
    },
    {
      "epoch": 3.35693359375e-06,
      "grad_norm": 0.4356783926486969,
      "learning_rate": 1e-05,
      "loss": 0.1256,
      "step": 880
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 3.197141170501709,
      "step": 880
    },
    {
      "dynamics_loss_avg": 0.07388954758644103,
      "epoch": 3.35693359375e-06,
      "step": 880
    },
    {
      "action_loss_avg": 0.05621569622308016,
      "epoch": 3.35693359375e-06,
      "step": 880
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 880,
      "training_step_time": 10.276237726211548
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 3.146392583847046,
      "step": 880
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 880,
      "training_step_time": 9.920161247253418
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 3.1510250568389893,
      "step": 880
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 880,
      "training_step_time": 9.927215814590454
    },
    {
      "epoch": 3.35693359375e-06,
      "model_forward_time": 3.1475133895874023,
      "step": 880
    },
    {
      "epoch": 3.35693359375e-06,
      "step": 880,
      "training_step_time": 10.02751088142395
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 2.485870122909546,
      "step": 881
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 881,
      "training_step_time": 7.52136492729187
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 3.4324543476104736,
      "step": 881
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 881,
      "training_step_time": 10.204510688781738
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 3.148385524749756,
      "step": 881
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 881,
      "training_step_time": 9.929651975631714
    },
    {
      "epoch": 3.360748291015625e-06,
      "model_forward_time": 2.4419631958007812,
      "step": 881
    },
    {
      "epoch": 3.360748291015625e-06,
      "step": 881,
      "training_step_time": 10.337415218353271
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 3.1921467781066895,
      "step": 882
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 882,
      "training_step_time": 9.964879035949707
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 3.144714117050171,
      "step": 882
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 882,
      "training_step_time": 9.918752670288086
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 3.146725654602051,
      "step": 882
    },
    {
      "dynamics_loss_avg": 0.07378169130533933,
      "epoch": 3.36456298828125e-06,
      "step": 882
    },
    {
      "action_loss_avg": 0.06665974333882332,
      "epoch": 3.36456298828125e-06,
      "step": 882
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 882,
      "training_step_time": 9.922691345214844
    },
    {
      "epoch": 3.36456298828125e-06,
      "model_forward_time": 3.4573099613189697,
      "step": 882
    },
    {
      "epoch": 3.36456298828125e-06,
      "step": 882,
      "training_step_time": 10.323731184005737
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 3.1871066093444824,
      "step": 883
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 883,
      "training_step_time": 9.963101387023926
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 3.1446402072906494,
      "step": 883
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 883,
      "training_step_time": 9.91348123550415
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 3.1457266807556152,
      "step": 883
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 883,
      "training_step_time": 9.916634321212769
    },
    {
      "epoch": 3.368377685546875e-06,
      "model_forward_time": 3.436264753341675,
      "step": 883
    },
    {
      "epoch": 3.368377685546875e-06,
      "step": 883,
      "training_step_time": 10.292770147323608
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 3.177140951156616,
      "step": 884
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 884,
      "training_step_time": 9.940972566604614
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 3.1399900913238525,
      "step": 884
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 884,
      "training_step_time": 9.906453132629395
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 3.1405978202819824,
      "step": 884
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 884,
      "training_step_time": 9.911685228347778
    },
    {
      "epoch": 3.3721923828125e-06,
      "model_forward_time": 3.140928030014038,
      "step": 884
    },
    {
      "epoch": 3.3721923828125e-06,
      "step": 884,
      "training_step_time": 10.273032188415527
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 3.173708915710449,
      "step": 885
    },
    {
      "dynamics_loss_avg": 0.06393726598471403,
      "epoch": 3.376007080078125e-06,
      "step": 885
    },
    {
      "action_loss_avg": 0.057603027066215874,
      "epoch": 3.376007080078125e-06,
      "step": 885
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 885,
      "training_step_time": 10.20977783203125
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 3.1414377689361572,
      "step": 885
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 885,
      "training_step_time": 9.911575078964233
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 3.1416544914245605,
      "step": 885
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 885,
      "training_step_time": 9.909497737884521
    },
    {
      "epoch": 3.376007080078125e-06,
      "model_forward_time": 3.137518882751465,
      "step": 885
    },
    {
      "epoch": 3.376007080078125e-06,
      "step": 885,
      "training_step_time": 10.259270191192627
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 3.1526544094085693,
      "step": 886
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 886,
      "training_step_time": 9.92552924156189
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 3.405878782272339,
      "step": 886
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 886,
      "training_step_time": 10.172961235046387
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 3.1414952278137207,
      "step": 886
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 886,
      "training_step_time": 9.910552024841309
    },
    {
      "epoch": 3.37982177734375e-06,
      "model_forward_time": 2.4429564476013184,
      "step": 886
    },
    {
      "epoch": 3.37982177734375e-06,
      "step": 886,
      "training_step_time": 10.269388914108276
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 3.1852385997772217,
      "step": 887
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 887,
      "training_step_time": 9.963773012161255
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 3.1451642513275146,
      "step": 887
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 887,
      "training_step_time": 9.917592763900757
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 3.4734537601470947,
      "step": 887
    },
    {
      "dynamics_loss_avg": 0.07714751996099949,
      "epoch": 3.383636474609375e-06,
      "step": 887
    },
    {
      "action_loss_avg": 0.06467300616204738,
      "epoch": 3.383636474609375e-06,
      "step": 887
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 887,
      "training_step_time": 10.25196123123169
    },
    {
      "epoch": 3.383636474609375e-06,
      "model_forward_time": 3.145660877227783,
      "step": 887
    },
    {
      "epoch": 3.383636474609375e-06,
      "step": 887,
      "training_step_time": 10.032138586044312
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 3.2031588554382324,
      "step": 888
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 888,
      "training_step_time": 9.980058193206787
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 2.440502405166626,
      "step": 888
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 888,
      "training_step_time": 7.468250751495361
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 3.139812707901001,
      "step": 888
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 888,
      "training_step_time": 9.911959171295166
    },
    {
      "epoch": 3.387451171875e-06,
      "model_forward_time": 3.1498920917510986,
      "step": 888
    },
    {
      "epoch": 3.387451171875e-06,
      "step": 888,
      "training_step_time": 10.301956176757812
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 3.1952121257781982,
      "step": 889
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 889,
      "training_step_time": 9.972062349319458
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 3.1503403186798096,
      "step": 889
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 889,
      "training_step_time": 9.92596697807312
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 3.144967794418335,
      "step": 889
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 889,
      "training_step_time": 9.918050765991211
    },
    {
      "epoch": 3.391265869140625e-06,
      "model_forward_time": 3.1446688175201416,
      "step": 889
    },
    {
      "epoch": 3.391265869140625e-06,
      "step": 889,
      "training_step_time": 10.283500909805298
    },
    {
      "epoch": 3.39508056640625e-06,
      "grad_norm": 0.7840091586112976,
      "learning_rate": 1e-05,
      "loss": 0.1335,
      "step": 890
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 3.454389810562134,
      "step": 890
    },
    {
      "dynamics_loss_avg": 0.08282234258949757,
      "epoch": 3.39508056640625e-06,
      "step": 890
    },
    {
      "action_loss_avg": 0.06650889478623867,
      "epoch": 3.39508056640625e-06,
      "step": 890
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 890,
      "training_step_time": 10.228377342224121
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 3.144559860229492,
      "step": 890
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 890,
      "training_step_time": 9.91690707206726
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 3.139817237854004,
      "step": 890
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 890,
      "training_step_time": 9.914618253707886
    },
    {
      "epoch": 3.39508056640625e-06,
      "model_forward_time": 3.1472973823547363,
      "step": 890
    },
    {
      "epoch": 3.39508056640625e-06,
      "step": 890,
      "training_step_time": 10.27411437034607
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 3.1965701580047607,
      "step": 891
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 891,
      "training_step_time": 9.971006870269775
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 3.1449129581451416,
      "step": 891
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 891,
      "training_step_time": 10.195621013641357
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 3.145263195037842,
      "step": 891
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 891,
      "training_step_time": 9.924975395202637
    },
    {
      "epoch": 3.398895263671875e-06,
      "model_forward_time": 3.145134687423706,
      "step": 891
    },
    {
      "epoch": 3.398895263671875e-06,
      "step": 891,
      "training_step_time": 10.017391204833984
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 3.189394235610962,
      "step": 892
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 892,
      "training_step_time": 9.97298002243042
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 3.1451146602630615,
      "step": 892
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 892,
      "training_step_time": 9.923547506332397
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 3.437849521636963,
      "step": 892
    },
    {
      "dynamics_loss_avg": 0.07321584038436413,
      "epoch": 3.4027099609375e-06,
      "step": 892
    },
    {
      "action_loss_avg": 0.07467499785125256,
      "epoch": 3.4027099609375e-06,
      "step": 892
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 892,
      "training_step_time": 10.22132396697998
    },
    {
      "epoch": 3.4027099609375e-06,
      "model_forward_time": 3.1508629322052,
      "step": 892
    },
    {
      "epoch": 3.4027099609375e-06,
      "step": 892,
      "training_step_time": 10.324402093887329
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 3.177388906478882,
      "step": 893
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 893,
      "training_step_time": 9.957127571105957
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 2.4335556030273438,
      "step": 893
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 893,
      "training_step_time": 7.462648630142212
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 3.1328704357147217,
      "step": 893
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 893,
      "training_step_time": 9.902378797531128
    },
    {
      "epoch": 3.406524658203125e-06,
      "model_forward_time": 3.131549835205078,
      "step": 893
    },
    {
      "epoch": 3.406524658203125e-06,
      "step": 893,
      "training_step_time": 10.26706314086914
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 3.4760050773620605,
      "step": 894
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 894,
      "training_step_time": 10.244468688964844
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 3.1300833225250244,
      "step": 894
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 894,
      "training_step_time": 9.89662766456604
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 3.1312222480773926,
      "step": 894
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 894,
      "training_step_time": 9.898128747940063
    },
    {
      "epoch": 3.41033935546875e-06,
      "model_forward_time": 3.1285183429718018,
      "step": 894
    },
    {
      "epoch": 3.41033935546875e-06,
      "step": 894,
      "training_step_time": 10.24495792388916
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 3.432417154312134,
      "step": 895
    },
    {
      "dynamics_loss_avg": 0.0641944944858551,
      "epoch": 3.414154052734375e-06,
      "step": 895
    },
    {
      "action_loss_avg": 0.04627005215734244,
      "epoch": 3.414154052734375e-06,
      "step": 895
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 895,
      "training_step_time": 10.201478719711304
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 3.130764961242676,
      "step": 895
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 895,
      "training_step_time": 9.898991823196411
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 3.135547161102295,
      "step": 895
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 895,
      "training_step_time": 9.907711744308472
    },
    {
      "epoch": 3.414154052734375e-06,
      "model_forward_time": 3.132899045944214,
      "step": 895
    },
    {
      "epoch": 3.414154052734375e-06,
      "step": 895,
      "training_step_time": 10.300649642944336
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 3.161126136779785,
      "step": 896
    },
    {
      "epoch": 3.41796875e-06,
      "step": 896,
      "training_step_time": 9.9351806640625
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 3.4015989303588867,
      "step": 896
    },
    {
      "epoch": 3.41796875e-06,
      "step": 896,
      "training_step_time": 10.17650818824768
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 3.1355693340301514,
      "step": 896
    },
    {
      "epoch": 3.41796875e-06,
      "step": 896,
      "training_step_time": 9.911696910858154
    },
    {
      "epoch": 3.41796875e-06,
      "model_forward_time": 3.1336264610290527,
      "step": 896
    },
    {
      "epoch": 3.41796875e-06,
      "step": 896,
      "training_step_time": 10.026829242706299
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 3.1924681663513184,
      "step": 897
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 897,
      "training_step_time": 9.974347114562988
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 3.145826816558838,
      "step": 897
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 897,
      "training_step_time": 9.922188758850098
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 3.139275550842285,
      "step": 897
    },
    {
      "dynamics_loss_avg": 0.08198587130755186,
      "epoch": 3.421783447265625e-06,
      "step": 897
    },
    {
      "action_loss_avg": 0.07875844119116664,
      "epoch": 3.421783447265625e-06,
      "step": 897
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 897,
      "training_step_time": 10.232823848724365
    },
    {
      "epoch": 3.421783447265625e-06,
      "model_forward_time": 3.1344499588012695,
      "step": 897
    },
    {
      "epoch": 3.421783447265625e-06,
      "step": 897,
      "training_step_time": 10.26292872428894
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 3.1830129623413086,
      "step": 898
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 898,
      "training_step_time": 9.958276271820068
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 3.1366662979125977,
      "step": 898
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 898,
      "training_step_time": 9.912566423416138
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 2.430241584777832,
      "step": 898
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 898,
      "training_step_time": 7.462278842926025
    },
    {
      "epoch": 3.42559814453125e-06,
      "model_forward_time": 3.133483648300171,
      "step": 898
    },
    {
      "epoch": 3.42559814453125e-06,
      "step": 898,
      "training_step_time": 10.334696769714355
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 3.4418857097625732,
      "step": 899
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 899,
      "training_step_time": 10.211235761642456
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 3.133315086364746,
      "step": 899
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 899,
      "training_step_time": 9.905653476715088
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 3.1328353881835938,
      "step": 899
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 899,
      "training_step_time": 9.905768871307373
    },
    {
      "epoch": 3.429412841796875e-06,
      "model_forward_time": 3.13425874710083,
      "step": 899
    },
    {
      "epoch": 3.429412841796875e-06,
      "step": 899,
      "training_step_time": 10.307752132415771
    },
    {
      "epoch": 3.4332275390625e-06,
      "grad_norm": 0.6403791308403015,
      "learning_rate": 1e-05,
      "loss": 0.1352,
      "step": 900
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 2.470531463623047,
      "step": 900
    },
    {
      "dynamics_loss_avg": 0.08006791211664677,
      "epoch": 3.4332275390625e-06,
      "step": 900
    },
    {
      "action_loss_avg": 0.06372344233095646,
      "epoch": 3.4332275390625e-06,
      "step": 900
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 900,
      "training_step_time": 7.500306606292725
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 2.686774969100952,
      "step": 900
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 900,
      "training_step_time": 7.711669921875
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 3.1298670768737793,
      "step": 900
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 900,
      "training_step_time": 9.894733428955078
    },
    {
      "epoch": 3.4332275390625e-06,
      "model_forward_time": 3.1359283924102783,
      "step": 900
    },
    {
      "epoch": 3.4332275390625e-06,
      "step": 900,
      "training_step_time": 10.307918787002563
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 2.446106433868408,
      "step": 901
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 901,
      "training_step_time": 7.470922946929932
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 2.4261927604675293,
      "step": 901
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 901,
      "training_step_time": 7.450557231903076
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 2.7292873859405518,
      "step": 901
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 901,
      "training_step_time": 7.753804683685303
    },
    {
      "epoch": 3.437042236328125e-06,
      "model_forward_time": 3.131682872772217,
      "step": 901
    },
    {
      "epoch": 3.437042236328125e-06,
      "step": 901,
      "training_step_time": 10.007473230361938
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 3.177837610244751,
      "step": 902
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 902,
      "training_step_time": 9.949807405471802
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 3.135160446166992,
      "step": 902
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 902,
      "training_step_time": 9.903801202774048
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 3.1383111476898193,
      "step": 902
    },
    {
      "dynamics_loss_avg": 0.06705307886004448,
      "epoch": 3.44085693359375e-06,
      "step": 902
    },
    {
      "action_loss_avg": 0.05614190325140953,
      "epoch": 3.44085693359375e-06,
      "step": 902
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 902,
      "training_step_time": 9.916909456253052
    },
    {
      "epoch": 3.44085693359375e-06,
      "model_forward_time": 3.41298246383667,
      "step": 902
    },
    {
      "epoch": 3.44085693359375e-06,
      "step": 902,
      "training_step_time": 10.305555820465088
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 3.1578946113586426,
      "step": 903
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 903,
      "training_step_time": 9.934033632278442
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 3.140638589859009,
      "step": 903
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 903,
      "training_step_time": 9.914444923400879
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 3.138951301574707,
      "step": 903
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 903,
      "training_step_time": 9.908940315246582
    },
    {
      "epoch": 3.444671630859375e-06,
      "model_forward_time": 2.4384868144989014,
      "step": 903
    },
    {
      "epoch": 3.444671630859375e-06,
      "step": 903,
      "training_step_time": 10.283957958221436
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 3.4593567848205566,
      "step": 904
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 904,
      "training_step_time": 10.229304790496826
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 3.138779878616333,
      "step": 904
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 904,
      "training_step_time": 9.914248704910278
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 3.139812469482422,
      "step": 904
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 904,
      "training_step_time": 9.90699028968811
    },
    {
      "epoch": 3.448486328125e-06,
      "model_forward_time": 3.1380252838134766,
      "step": 904
    },
    {
      "epoch": 3.448486328125e-06,
      "step": 904,
      "training_step_time": 10.264179706573486
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 3.169182062149048,
      "step": 905
    },
    {
      "dynamics_loss_avg": 0.0722511786967516,
      "epoch": 3.452301025390625e-06,
      "step": 905
    },
    {
      "action_loss_avg": 0.05606391970068216,
      "epoch": 3.452301025390625e-06,
      "step": 905
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 905,
      "training_step_time": 9.94333553314209
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 3.4333105087280273,
      "step": 905
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 905,
      "training_step_time": 10.200722217559814
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 3.1415908336639404,
      "step": 905
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 905,
      "training_step_time": 9.91287899017334
    },
    {
      "epoch": 3.452301025390625e-06,
      "model_forward_time": 3.1384177207946777,
      "step": 905
    },
    {
      "epoch": 3.452301025390625e-06,
      "step": 905,
      "training_step_time": 10.301687479019165
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 3.1791775226593018,
      "step": 906
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 906,
      "training_step_time": 9.951066017150879
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 3.1401164531707764,
      "step": 906
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 906,
      "training_step_time": 10.173311233520508
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 2.431232213973999,
      "step": 906
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 906,
      "training_step_time": 7.4586169719696045
    },
    {
      "epoch": 3.45611572265625e-06,
      "model_forward_time": 3.1350674629211426,
      "step": 906
    },
    {
      "epoch": 3.45611572265625e-06,
      "step": 906,
      "training_step_time": 10.026038408279419
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 3.1750473976135254,
      "step": 907
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 907,
      "training_step_time": 9.950220584869385
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 3.1365649700164795,
      "step": 907
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 907,
      "training_step_time": 9.905269384384155
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 3.1350035667419434,
      "step": 907
    },
    {
      "dynamics_loss_avg": 0.08009144626557826,
      "epoch": 3.459930419921875e-06,
      "step": 907
    },
    {
      "action_loss_avg": 0.06352209858596325,
      "epoch": 3.459930419921875e-06,
      "step": 907
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 907,
      "training_step_time": 9.905225992202759
    },
    {
      "epoch": 3.459930419921875e-06,
      "model_forward_time": 3.403527021408081,
      "step": 907
    },
    {
      "epoch": 3.459930419921875e-06,
      "step": 907,
      "training_step_time": 10.270384550094604
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 2.4802684783935547,
      "step": 908
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 908,
      "training_step_time": 7.508910417556763
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 3.133688449859619,
      "step": 908
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 908,
      "training_step_time": 9.907008171081543
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 3.141292095184326,
      "step": 908
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 908,
      "training_step_time": 9.915929794311523
    },
    {
      "epoch": 3.4637451171875e-06,
      "model_forward_time": 3.1433563232421875,
      "step": 908
    },
    {
      "epoch": 3.4637451171875e-06,
      "step": 908,
      "training_step_time": 10.31743311882019
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 3.1888272762298584,
      "step": 909
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 909,
      "training_step_time": 9.96220064163208
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 3.1408324241638184,
      "step": 909
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 909,
      "training_step_time": 9.916141510009766
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 3.1433053016662598,
      "step": 909
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 909,
      "training_step_time": 9.917947053909302
    },
    {
      "epoch": 3.467559814453125e-06,
      "model_forward_time": 3.1419434547424316,
      "step": 909
    },
    {
      "epoch": 3.467559814453125e-06,
      "step": 909,
      "training_step_time": 10.021566390991211
    },
    {
      "epoch": 3.47137451171875e-06,
      "grad_norm": 0.5697770118713379,
      "learning_rate": 1e-05,
      "loss": 0.1335,
      "step": 910
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 3.449847459793091,
      "step": 910
    },
    {
      "dynamics_loss_avg": 0.08883221633732319,
      "epoch": 3.47137451171875e-06,
      "step": 910
    },
    {
      "action_loss_avg": 0.07424534950405359,
      "epoch": 3.47137451171875e-06,
      "step": 910
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 910,
      "training_step_time": 10.223406553268433
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 3.137913703918457,
      "step": 910
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 910,
      "training_step_time": 9.900850534439087
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 3.13694167137146,
      "step": 910
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 910,
      "training_step_time": 9.90470004081726
    },
    {
      "epoch": 3.47137451171875e-06,
      "model_forward_time": 3.1374120712280273,
      "step": 910
    },
    {
      "epoch": 3.47137451171875e-06,
      "step": 910,
      "training_step_time": 10.339990139007568
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 3.19175124168396,
      "step": 911
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 911,
      "training_step_time": 9.961249351501465
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 3.13515305519104,
      "step": 911
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 911,
      "training_step_time": 10.197452545166016
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 3.1342153549194336,
      "step": 911
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 911,
      "training_step_time": 9.903875827789307
    },
    {
      "epoch": 3.475189208984375e-06,
      "model_forward_time": 3.142179489135742,
      "step": 911
    },
    {
      "epoch": 3.475189208984375e-06,
      "step": 911,
      "training_step_time": 10.01686692237854
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 3.1740214824676514,
      "step": 912
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 912,
      "training_step_time": 9.945803880691528
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 3.140064239501953,
      "step": 912
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 912,
      "training_step_time": 9.912375926971436
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 3.405580997467041,
      "step": 912
    },
    {
      "dynamics_loss_avg": 0.06372524797916412,
      "epoch": 3.47900390625e-06,
      "step": 912
    },
    {
      "action_loss_avg": 0.05045999772846699,
      "epoch": 3.47900390625e-06,
      "step": 912
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 912,
      "training_step_time": 10.182824611663818
    },
    {
      "epoch": 3.47900390625e-06,
      "model_forward_time": 3.147502899169922,
      "step": 912
    },
    {
      "epoch": 3.47900390625e-06,
      "step": 912,
      "training_step_time": 10.050750970840454
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 3.1836845874786377,
      "step": 913
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 913,
      "training_step_time": 9.958369255065918
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 3.142639398574829,
      "step": 913
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 913,
      "training_step_time": 9.921252727508545
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 3.1432690620422363,
      "step": 913
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 913,
      "training_step_time": 9.919605255126953
    },
    {
      "epoch": 3.482818603515625e-06,
      "model_forward_time": 3.1410248279571533,
      "step": 913
    },
    {
      "epoch": 3.482818603515625e-06,
      "step": 913,
      "training_step_time": 10.31187129020691
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 3.1700820922851562,
      "step": 914
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 914,
      "training_step_time": 9.945533990859985
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 3.1417899131774902,
      "step": 914
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 914,
      "training_step_time": 9.916961669921875
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 3.14243221282959,
      "step": 914
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 914,
      "training_step_time": 9.916833877563477
    },
    {
      "epoch": 3.48663330078125e-06,
      "model_forward_time": 3.13802170753479,
      "step": 914
    },
    {
      "epoch": 3.48663330078125e-06,
      "step": 914,
      "training_step_time": 10.323710918426514
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 3.4566261768341064,
      "step": 915
    },
    {
      "dynamics_loss_avg": 0.08447006233036518,
      "epoch": 3.490447998046875e-06,
      "step": 915
    },
    {
      "action_loss_avg": 0.07449070364236832,
      "epoch": 3.490447998046875e-06,
      "step": 915
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 915,
      "training_step_time": 10.233752012252808
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 3.1379847526550293,
      "step": 915
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 915,
      "training_step_time": 9.915473937988281
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 3.1507444381713867,
      "step": 915
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 915,
      "training_step_time": 9.923638105392456
    },
    {
      "epoch": 3.490447998046875e-06,
      "model_forward_time": 3.1382710933685303,
      "step": 915
    },
    {
      "epoch": 3.490447998046875e-06,
      "step": 915,
      "training_step_time": 10.309611558914185
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 3.190662145614624,
      "step": 916
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 916,
      "training_step_time": 9.963546752929688
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 3.3943967819213867,
      "step": 916
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 916,
      "training_step_time": 10.16224718093872
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 3.1385865211486816,
      "step": 916
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 916,
      "training_step_time": 9.910457134246826
    },
    {
      "epoch": 3.4942626953125e-06,
      "model_forward_time": 3.1456735134124756,
      "step": 916
    },
    {
      "epoch": 3.4942626953125e-06,
      "step": 916,
      "training_step_time": 10.028581142425537
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 2.4821724891662598,
      "step": 917
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 917,
      "training_step_time": 7.514348030090332
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 3.1348769664764404,
      "step": 917
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 917,
      "training_step_time": 9.907806634902954
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 3.406068801879883,
      "step": 917
    },
    {
      "dynamics_loss_avg": 0.05661124736070633,
      "epoch": 3.498077392578125e-06,
      "step": 917
    },
    {
      "action_loss_avg": 0.05250779315829277,
      "epoch": 3.498077392578125e-06,
      "step": 917
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 917,
      "training_step_time": 10.180921077728271
    },
    {
      "epoch": 3.498077392578125e-06,
      "model_forward_time": 3.1340084075927734,
      "step": 917
    },
    {
      "epoch": 3.498077392578125e-06,
      "step": 917,
      "training_step_time": 10.024709224700928
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 2.4571316242218018,
      "step": 918
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 918,
      "training_step_time": 7.484766244888306
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 3.1378815174102783,
      "step": 918
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 918,
      "training_step_time": 9.91143012046814
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 3.139017343521118,
      "step": 918
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 918,
      "training_step_time": 9.911264896392822
    },
    {
      "epoch": 3.50189208984375e-06,
      "model_forward_time": 2.43406081199646,
      "step": 918
    },
    {
      "epoch": 3.50189208984375e-06,
      "step": 918,
      "training_step_time": 10.265964984893799
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 2.79278826713562,
      "step": 919
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 919,
      "training_step_time": 7.818962574005127
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 3.1352744102478027,
      "step": 919
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 919,
      "training_step_time": 9.9012930393219
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 3.1367995738983154,
      "step": 919
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 919,
      "training_step_time": 9.902419567108154
    },
    {
      "epoch": 3.505706787109375e-06,
      "model_forward_time": 3.145732879638672,
      "step": 919
    },
    {
      "epoch": 3.505706787109375e-06,
      "step": 919,
      "training_step_time": 10.273634195327759
    },
    {
      "epoch": 3.509521484375e-06,
      "grad_norm": 0.5206982493400574,
      "learning_rate": 1e-05,
      "loss": 0.1324,
      "step": 920
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 3.1563408374786377,
      "step": 920
    },
    {
      "dynamics_loss_avg": 0.06732271313667297,
      "epoch": 3.509521484375e-06,
      "step": 920
    },
    {
      "action_loss_avg": 0.05742576792836189,
      "epoch": 3.509521484375e-06,
      "step": 920
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 920,
      "training_step_time": 10.215349912643433
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 3.133923053741455,
      "step": 920
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 920,
      "training_step_time": 9.899458408355713
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 2.432748794555664,
      "step": 920
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 920,
      "training_step_time": 7.451984167098999
    },
    {
      "epoch": 3.509521484375e-06,
      "model_forward_time": 2.4262914657592773,
      "step": 920
    },
    {
      "epoch": 3.509521484375e-06,
      "step": 920,
      "training_step_time": 10.314769268035889
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 3.1689438819885254,
      "step": 921
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 921,
      "training_step_time": 9.939170598983765
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 3.1357555389404297,
      "step": 921
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 921,
      "training_step_time": 9.900174856185913
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 3.380727767944336,
      "step": 921
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 921,
      "training_step_time": 10.14831018447876
    },
    {
      "epoch": 3.513336181640625e-06,
      "model_forward_time": 3.142300844192505,
      "step": 921
    },
    {
      "epoch": 3.513336181640625e-06,
      "step": 921,
      "training_step_time": 10.028907060623169
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 2.476351499557495,
      "step": 922
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 922,
      "training_step_time": 7.514486074447632
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 3.1369385719299316,
      "step": 922
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 922,
      "training_step_time": 9.908471584320068
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 3.1471471786499023,
      "step": 922
    },
    {
      "dynamics_loss_avg": 0.06901010796427727,
      "epoch": 3.51715087890625e-06,
      "step": 922
    },
    {
      "action_loss_avg": 0.04551063915714622,
      "epoch": 3.51715087890625e-06,
      "step": 922
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 922,
      "training_step_time": 9.927703857421875
    },
    {
      "epoch": 3.51715087890625e-06,
      "model_forward_time": 2.7056777477264404,
      "step": 922
    },
    {
      "epoch": 3.51715087890625e-06,
      "step": 922,
      "training_step_time": 10.025795221328735
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 3.1570982933044434,
      "step": 923
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 923,
      "training_step_time": 9.935238361358643
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 3.1371829509735107,
      "step": 923
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 923,
      "training_step_time": 9.909116268157959
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 2.431694984436035,
      "step": 923
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 923,
      "training_step_time": 7.4621686935424805
    },
    {
      "epoch": 3.520965576171875e-06,
      "model_forward_time": 3.127399444580078,
      "step": 923
    },
    {
      "epoch": 3.520965576171875e-06,
      "step": 923,
      "training_step_time": 10.274212121963501
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 3.4575791358947754,
      "step": 924
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 924,
      "training_step_time": 10.225512504577637
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 3.1370580196380615,
      "step": 924
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 924,
      "training_step_time": 9.904733896255493
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 2.429731845855713,
      "step": 924
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 924,
      "training_step_time": 7.450946092605591
    },
    {
      "epoch": 3.5247802734375e-06,
      "model_forward_time": 3.129415988922119,
      "step": 924
    },
    {
      "epoch": 3.5247802734375e-06,
      "step": 924,
      "training_step_time": 10.270731210708618
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 3.17915940284729,
      "step": 925
    },
    {
      "dynamics_loss_avg": 0.08130330480635166,
      "epoch": 3.528594970703125e-06,
      "step": 925
    },
    {
      "action_loss_avg": 0.06173315010964871,
      "epoch": 3.528594970703125e-06,
      "step": 925
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 925,
      "training_step_time": 9.95298719406128
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 3.4175643920898438,
      "step": 925
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 925,
      "training_step_time": 10.191473245620728
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 3.143577814102173,
      "step": 925
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 925,
      "training_step_time": 9.918763160705566
    },
    {
      "epoch": 3.528594970703125e-06,
      "model_forward_time": 3.1446926593780518,
      "step": 925
    },
    {
      "epoch": 3.528594970703125e-06,
      "step": 925,
      "training_step_time": 10.315199375152588
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 3.1928534507751465,
      "step": 926
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 926,
      "training_step_time": 9.970646142959595
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 3.144507884979248,
      "step": 926
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 926,
      "training_step_time": 9.917690753936768
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 3.409942626953125,
      "step": 926
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 926,
      "training_step_time": 10.1822829246521
    },
    {
      "epoch": 3.53240966796875e-06,
      "model_forward_time": 3.144925117492676,
      "step": 926
    },
    {
      "epoch": 3.53240966796875e-06,
      "step": 926,
      "training_step_time": 10.268142461776733
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 3.1768062114715576,
      "step": 927
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 927,
      "training_step_time": 9.950730800628662
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 3.1462910175323486,
      "step": 927
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 927,
      "training_step_time": 9.9236159324646
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 3.1393163204193115,
      "step": 927
    },
    {
      "dynamics_loss_avg": 0.05953346509486437,
      "epoch": 3.536224365234375e-06,
      "step": 927
    },
    {
      "action_loss_avg": 0.04479712583124638,
      "epoch": 3.536224365234375e-06,
      "step": 927
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 927,
      "training_step_time": 9.922330379486084
    },
    {
      "epoch": 3.536224365234375e-06,
      "model_forward_time": 3.3827760219573975,
      "step": 927
    },
    {
      "epoch": 3.536224365234375e-06,
      "step": 927,
      "training_step_time": 10.243598699569702
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 3.1834912300109863,
      "step": 928
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 928,
      "training_step_time": 9.960935354232788
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 3.1405391693115234,
      "step": 928
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 928,
      "training_step_time": 9.91084337234497
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 3.1366965770721436,
      "step": 928
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 928,
      "training_step_time": 9.905708312988281
    },
    {
      "epoch": 3.5400390625e-06,
      "model_forward_time": 3.1381611824035645,
      "step": 928
    },
    {
      "epoch": 3.5400390625e-06,
      "step": 928,
      "training_step_time": 10.279097080230713
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 3.160505771636963,
      "step": 929
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 929,
      "training_step_time": 9.92920994758606
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 3.138437509536743,
      "step": 929
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 929,
      "training_step_time": 9.916778087615967
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 3.1435799598693848,
      "step": 929
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 929,
      "training_step_time": 9.917537689208984
    },
    {
      "epoch": 3.543853759765625e-06,
      "model_forward_time": 2.433623790740967,
      "step": 929
    },
    {
      "epoch": 3.543853759765625e-06,
      "step": 929,
      "training_step_time": 10.252631664276123
    },
    {
      "epoch": 3.54766845703125e-06,
      "grad_norm": 0.4607387185096741,
      "learning_rate": 1e-05,
      "loss": 0.1314,
      "step": 930
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 3.1750600337982178,
      "step": 930
    },
    {
      "dynamics_loss_avg": 0.06769370883703232,
      "epoch": 3.54766845703125e-06,
      "step": 930
    },
    {
      "action_loss_avg": 0.044172496907413004,
      "epoch": 3.54766845703125e-06,
      "step": 930
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 930,
      "training_step_time": 9.949290990829468
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 3.423534393310547,
      "step": 930
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 930,
      "training_step_time": 10.196126937866211
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 3.1341233253479004,
      "step": 930
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 930,
      "training_step_time": 9.903441667556763
    },
    {
      "epoch": 3.54766845703125e-06,
      "model_forward_time": 3.1343116760253906,
      "step": 930
    },
    {
      "epoch": 3.54766845703125e-06,
      "step": 930,
      "training_step_time": 10.288578510284424
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 3.1548216342926025,
      "step": 931
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 931,
      "training_step_time": 9.935587882995605
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 2.4375040531158447,
      "step": 931
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 931,
      "training_step_time": 7.742340564727783
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 2.4346539974212646,
      "step": 931
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 931,
      "training_step_time": 7.466046094894409
    },
    {
      "epoch": 3.551483154296875e-06,
      "model_forward_time": 2.434675931930542,
      "step": 931
    },
    {
      "epoch": 3.551483154296875e-06,
      "step": 931,
      "training_step_time": 10.294912576675415
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 3.178877830505371,
      "step": 932
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 932,
      "training_step_time": 9.958234071731567
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 3.1409575939178467,
      "step": 932
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 932,
      "training_step_time": 9.918143033981323
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 3.1426608562469482,
      "step": 932
    },
    {
      "dynamics_loss_avg": 0.07955278269946575,
      "epoch": 3.5552978515625e-06,
      "step": 932
    },
    {
      "action_loss_avg": 0.06327421721071005,
      "epoch": 3.5552978515625e-06,
      "step": 932
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 932,
      "training_step_time": 10.183310270309448
    },
    {
      "epoch": 3.5552978515625e-06,
      "model_forward_time": 3.1447982788085938,
      "step": 932
    },
    {
      "epoch": 3.5552978515625e-06,
      "step": 932,
      "training_step_time": 10.021982669830322
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 2.472431182861328,
      "step": 933
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 933,
      "training_step_time": 7.498142242431641
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 3.133512496948242,
      "step": 933
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 933,
      "training_step_time": 9.905839204788208
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 3.142047882080078,
      "step": 933
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 933,
      "training_step_time": 9.908722162246704
    },
    {
      "epoch": 3.559112548828125e-06,
      "model_forward_time": 3.1397736072540283,
      "step": 933
    },
    {
      "epoch": 3.559112548828125e-06,
      "step": 933,
      "training_step_time": 10.324149131774902
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 3.5080294609069824,
      "step": 934
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 934,
      "training_step_time": 10.27493691444397
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 3.126725196838379,
      "step": 934
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 934,
      "training_step_time": 9.89163088798523
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 3.1279513835906982,
      "step": 934
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 934,
      "training_step_time": 9.895467758178711
    },
    {
      "epoch": 3.56292724609375e-06,
      "model_forward_time": 3.1306066513061523,
      "step": 934
    },
    {
      "epoch": 3.56292724609375e-06,
      "step": 934,
      "training_step_time": 10.269169569015503
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 2.7294039726257324,
      "step": 935
    },
    {
      "dynamics_loss_avg": 0.08108333460986614,
      "epoch": 3.566741943359375e-06,
      "step": 935
    },
    {
      "action_loss_avg": 0.08150936812162399,
      "epoch": 3.566741943359375e-06,
      "step": 935
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 935,
      "training_step_time": 7.758784294128418
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 3.1351306438446045,
      "step": 935
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 935,
      "training_step_time": 9.90012526512146
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 3.1403443813323975,
      "step": 935
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 935,
      "training_step_time": 9.905503273010254
    },
    {
      "epoch": 3.566741943359375e-06,
      "model_forward_time": 3.1345643997192383,
      "step": 935
    },
    {
      "epoch": 3.566741943359375e-06,
      "step": 935,
      "training_step_time": 10.287679195404053
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 3.172574281692505,
      "step": 936
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 936,
      "training_step_time": 9.943148851394653
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 3.1368680000305176,
      "step": 936
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 936,
      "training_step_time": 10.170934438705444
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 3.1438441276550293,
      "step": 936
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 936,
      "training_step_time": 9.917581796646118
    },
    {
      "epoch": 3.570556640625e-06,
      "model_forward_time": 3.1420845985412598,
      "step": 936
    },
    {
      "epoch": 3.570556640625e-06,
      "step": 936,
      "training_step_time": 10.329519271850586
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 3.163172721862793,
      "step": 937
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 937,
      "training_step_time": 9.940013885498047
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 3.144467353820801,
      "step": 937
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 937,
      "training_step_time": 9.924988746643066
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 3.422224998474121,
      "step": 937
    },
    {
      "dynamics_loss_avg": 0.0848317552357912,
      "epoch": 3.574371337890625e-06,
      "step": 937
    },
    {
      "action_loss_avg": 0.06965525634586811,
      "epoch": 3.574371337890625e-06,
      "step": 937
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 937,
      "training_step_time": 10.201925039291382
    },
    {
      "epoch": 3.574371337890625e-06,
      "model_forward_time": 2.44311261177063,
      "step": 937
    },
    {
      "epoch": 3.574371337890625e-06,
      "step": 937,
      "training_step_time": 10.028073787689209
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 3.180083751678467,
      "step": 938
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 938,
      "training_step_time": 9.95341682434082
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 3.1428894996643066,
      "step": 938
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 938,
      "training_step_time": 9.9190092086792
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 3.1415741443634033,
      "step": 938
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 938,
      "training_step_time": 9.917096138000488
    },
    {
      "epoch": 3.57818603515625e-06,
      "model_forward_time": 3.447216510772705,
      "step": 938
    },
    {
      "epoch": 3.57818603515625e-06,
      "step": 938,
      "training_step_time": 10.312842845916748
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 3.1488208770751953,
      "step": 939
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 939,
      "training_step_time": 9.925676345825195
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 3.1434457302093506,
      "step": 939
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 939,
      "training_step_time": 9.920651912689209
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 3.1397194862365723,
      "step": 939
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 939,
      "training_step_time": 9.916294813156128
    },
    {
      "epoch": 3.582000732421875e-06,
      "model_forward_time": 2.4372148513793945,
      "step": 939
    },
    {
      "epoch": 3.582000732421875e-06,
      "step": 939,
      "training_step_time": 10.014727354049683
    },
    {
      "epoch": 3.5858154296875e-06,
      "grad_norm": 0.47896653413772583,
      "learning_rate": 1e-05,
      "loss": 0.1326,
      "step": 940
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 3.1677095890045166,
      "step": 940
    },
    {
      "dynamics_loss_avg": 0.05583719704300165,
      "epoch": 3.5858154296875e-06,
      "step": 940
    },
    {
      "action_loss_avg": 0.0450450723990798,
      "epoch": 3.5858154296875e-06,
      "step": 940
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 940,
      "training_step_time": 10.208547830581665
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 3.135127305984497,
      "step": 940
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 940,
      "training_step_time": 9.907686233520508
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 3.134195327758789,
      "step": 940
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 940,
      "training_step_time": 9.909382343292236
    },
    {
      "epoch": 3.5858154296875e-06,
      "model_forward_time": 3.1392126083374023,
      "step": 940
    },
    {
      "epoch": 3.5858154296875e-06,
      "step": 940,
      "training_step_time": 10.013698101043701
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 3.1850152015686035,
      "step": 941
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 941,
      "training_step_time": 9.9604172706604
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 3.4078965187072754,
      "step": 941
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 941,
      "training_step_time": 10.181077480316162
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 3.131497859954834,
      "step": 941
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 941,
      "training_step_time": 9.907392978668213
    },
    {
      "epoch": 3.589630126953125e-06,
      "model_forward_time": 3.1360831260681152,
      "step": 941
    },
    {
      "epoch": 3.589630126953125e-06,
      "step": 941,
      "training_step_time": 10.291207790374756
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 3.182518482208252,
      "step": 942
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 942,
      "training_step_time": 9.962213516235352
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 3.1362693309783936,
      "step": 942
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 942,
      "training_step_time": 9.909883975982666
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 3.4291441440582275,
      "step": 942
    },
    {
      "dynamics_loss_avg": 0.06362891672179102,
      "epoch": 3.59344482421875e-06,
      "step": 942
    },
    {
      "action_loss_avg": 0.04370445162057877,
      "epoch": 3.59344482421875e-06,
      "step": 942
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 942,
      "training_step_time": 10.208100080490112
    },
    {
      "epoch": 3.59344482421875e-06,
      "model_forward_time": 3.138143539428711,
      "step": 942
    },
    {
      "epoch": 3.59344482421875e-06,
      "step": 942,
      "training_step_time": 10.275454044342041
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 3.182781219482422,
      "step": 943
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 943,
      "training_step_time": 9.963258504867554
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 3.1480541229248047,
      "step": 943
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 943,
      "training_step_time": 9.925633430480957
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 3.1500046253204346,
      "step": 943
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 943,
      "training_step_time": 9.92998480796814
    },
    {
      "epoch": 3.597259521484375e-06,
      "model_forward_time": 3.420999526977539,
      "step": 943
    },
    {
      "epoch": 3.597259521484375e-06,
      "step": 943,
      "training_step_time": 10.292062282562256
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 3.1846961975097656,
      "step": 944
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 944,
      "training_step_time": 9.96430230140686
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 3.1436266899108887,
      "step": 944
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 944,
      "training_step_time": 9.923589944839478
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 3.149444103240967,
      "step": 944
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 944,
      "training_step_time": 9.927430152893066
    },
    {
      "epoch": 3.60107421875e-06,
      "model_forward_time": 3.1451942920684814,
      "step": 944
    },
    {
      "epoch": 3.60107421875e-06,
      "step": 944,
      "training_step_time": 10.321242094039917
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 3.4667506217956543,
      "step": 945
    },
    {
      "dynamics_loss_avg": 0.08086825534701347,
      "epoch": 3.604888916015625e-06,
      "step": 945
    },
    {
      "action_loss_avg": 0.07204306498169899,
      "epoch": 3.604888916015625e-06,
      "step": 945
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 945,
      "training_step_time": 10.245422840118408
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 3.145594835281372,
      "step": 945
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 945,
      "training_step_time": 9.919854879379272
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 3.14272403717041,
      "step": 945
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 945,
      "training_step_time": 9.910055875778198
    },
    {
      "epoch": 3.604888916015625e-06,
      "model_forward_time": 2.4325389862060547,
      "step": 945
    },
    {
      "epoch": 3.604888916015625e-06,
      "step": 945,
      "training_step_time": 10.301041841506958
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 2.4536209106445312,
      "step": 946
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 946,
      "training_step_time": 7.477567672729492
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 3.1357955932617188,
      "step": 946
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 946,
      "training_step_time": 10.183554887771606
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 2.435680866241455,
      "step": 946
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 946,
      "training_step_time": 7.463625431060791
    },
    {
      "epoch": 3.60870361328125e-06,
      "model_forward_time": 3.1333820819854736,
      "step": 946
    },
    {
      "epoch": 3.60870361328125e-06,
      "step": 946,
      "training_step_time": 10.275858402252197
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 3.1469078063964844,
      "step": 947
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 947,
      "training_step_time": 9.916192293167114
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 3.134465217590332,
      "step": 947
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 947,
      "training_step_time": 9.902254104614258
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 2.430271625518799,
      "step": 947
    },
    {
      "dynamics_loss_avg": 0.0786661971360445,
      "epoch": 3.612518310546875e-06,
      "step": 947
    },
    {
      "action_loss_avg": 0.048765717260539534,
      "epoch": 3.612518310546875e-06,
      "step": 947
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 947,
      "training_step_time": 7.454606294631958
    },
    {
      "epoch": 3.612518310546875e-06,
      "model_forward_time": 2.69926381111145,
      "step": 947
    },
    {
      "epoch": 3.612518310546875e-06,
      "step": 947,
      "training_step_time": 10.282057285308838
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 2.4629909992218018,
      "step": 948
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 948,
      "training_step_time": 7.486372709274292
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 2.432523727416992,
      "step": 948
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 948,
      "training_step_time": 7.45467734336853
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 3.1319448947906494,
      "step": 948
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 948,
      "training_step_time": 9.897701740264893
    },
    {
      "epoch": 3.6163330078125e-06,
      "model_forward_time": 3.1383414268493652,
      "step": 948
    },
    {
      "epoch": 3.6163330078125e-06,
      "step": 948,
      "training_step_time": 10.007967710494995
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 3.4609336853027344,
      "step": 949
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 949,
      "training_step_time": 10.240773439407349
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 3.201852560043335,
      "step": 949
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 949,
      "training_step_time": 9.98334264755249
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 3.1415855884552,
      "step": 949
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 949,
      "training_step_time": 9.918801069259644
    },
    {
      "epoch": 3.620147705078125e-06,
      "model_forward_time": 3.14837384223938,
      "step": 949
    },
    {
      "epoch": 3.620147705078125e-06,
      "step": 949,
      "training_step_time": 10.031984806060791
    },
    {
      "epoch": 3.62396240234375e-06,
      "grad_norm": 1.1772176027297974,
      "learning_rate": 1e-05,
      "loss": 0.1307,
      "step": 950
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 3.193066120147705,
      "step": 950
    },
    {
      "dynamics_loss_avg": 0.06594509780406951,
      "epoch": 3.62396240234375e-06,
      "step": 950
    },
    {
      "action_loss_avg": 0.056472569704055786,
      "epoch": 3.62396240234375e-06,
      "step": 950
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 950,
      "training_step_time": 9.972756624221802
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 3.432326555252075,
      "step": 950
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 950,
      "training_step_time": 10.203912734985352
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 3.1401796340942383,
      "step": 950
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 950,
      "training_step_time": 9.918776035308838
    },
    {
      "epoch": 3.62396240234375e-06,
      "model_forward_time": 3.1417076587677,
      "step": 950
    },
    {
      "epoch": 3.62396240234375e-06,
      "step": 950,
      "training_step_time": 10.282717227935791
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 3.1656711101531982,
      "step": 951
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 951,
      "training_step_time": 9.938680410385132
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 3.1349048614501953,
      "step": 951
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 951,
      "training_step_time": 9.910027980804443
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 3.3933298587799072,
      "step": 951
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 951,
      "training_step_time": 10.161810874938965
    },
    {
      "epoch": 3.627777099609375e-06,
      "model_forward_time": 2.435872793197632,
      "step": 951
    },
    {
      "epoch": 3.627777099609375e-06,
      "step": 951,
      "training_step_time": 10.303529500961304
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 3.1778838634490967,
      "step": 952
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 952,
      "training_step_time": 9.94733452796936
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 3.138187885284424,
      "step": 952
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 952,
      "training_step_time": 9.908113479614258
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 3.1425974369049072,
      "step": 952
    },
    {
      "dynamics_loss_avg": 0.06307047493755817,
      "epoch": 3.631591796875e-06,
      "step": 952
    },
    {
      "action_loss_avg": 0.04922897107899189,
      "epoch": 3.631591796875e-06,
      "step": 952
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 952,
      "training_step_time": 9.922296524047852
    },
    {
      "epoch": 3.631591796875e-06,
      "model_forward_time": 3.4688658714294434,
      "step": 952
    },
    {
      "epoch": 3.631591796875e-06,
      "step": 952,
      "training_step_time": 10.338635444641113
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 2.455886125564575,
      "step": 953
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 953,
      "training_step_time": 7.486629486083984
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 3.1388652324676514,
      "step": 953
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 953,
      "training_step_time": 9.91375994682312
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 2.436417579650879,
      "step": 953
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 953,
      "training_step_time": 7.461121082305908
    },
    {
      "epoch": 3.635406494140625e-06,
      "model_forward_time": 3.1340181827545166,
      "step": 953
    },
    {
      "epoch": 3.635406494140625e-06,
      "step": 953,
      "training_step_time": 10.014666557312012
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 3.4390645027160645,
      "step": 954
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 954,
      "training_step_time": 10.209924936294556
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 2.436100721359253,
      "step": 954
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 954,
      "training_step_time": 7.461758613586426
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 3.1345622539520264,
      "step": 954
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 954,
      "training_step_time": 9.905104637145996
    },
    {
      "epoch": 3.63922119140625e-06,
      "model_forward_time": 3.141296863555908,
      "step": 954
    },
    {
      "epoch": 3.63922119140625e-06,
      "step": 954,
      "training_step_time": 10.021555423736572
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 3.194430351257324,
      "step": 955
    },
    {
      "dynamics_loss_avg": 0.06763074602931737,
      "epoch": 3.643035888671875e-06,
      "step": 955
    },
    {
      "action_loss_avg": 0.050947349052876234,
      "epoch": 3.643035888671875e-06,
      "step": 955
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 955,
      "training_step_time": 9.97094464302063
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 3.394914150238037,
      "step": 955
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 955,
      "training_step_time": 10.170236349105835
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 3.1419429779052734,
      "step": 955
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 955,
      "training_step_time": 9.916701316833496
    },
    {
      "epoch": 3.643035888671875e-06,
      "model_forward_time": 3.1403534412384033,
      "step": 955
    },
    {
      "epoch": 3.643035888671875e-06,
      "step": 955,
      "training_step_time": 10.328018188476562
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 3.1590349674224854,
      "step": 956
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 956,
      "training_step_time": 9.934425354003906
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 2.436710834503174,
      "step": 956
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 956,
      "training_step_time": 7.466339588165283
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 3.4527909755706787,
      "step": 956
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 956,
      "training_step_time": 10.228149652481079
    },
    {
      "epoch": 3.6468505859375e-06,
      "model_forward_time": 3.1396656036376953,
      "step": 956
    },
    {
      "epoch": 3.6468505859375e-06,
      "step": 956,
      "training_step_time": 10.283263921737671
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 3.1680638790130615,
      "step": 957
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 957,
      "training_step_time": 9.942257642745972
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 3.137868642807007,
      "step": 957
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 957,
      "training_step_time": 9.90906572341919
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 3.134690999984741,
      "step": 957
    },
    {
      "dynamics_loss_avg": 0.08260552510619164,
      "epoch": 3.650665283203125e-06,
      "step": 957
    },
    {
      "action_loss_avg": 0.0696177420206368,
      "epoch": 3.650665283203125e-06,
      "step": 957
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 957,
      "training_step_time": 9.921024799346924
    },
    {
      "epoch": 3.650665283203125e-06,
      "model_forward_time": 3.384246349334717,
      "step": 957
    },
    {
      "epoch": 3.650665283203125e-06,
      "step": 957,
      "training_step_time": 10.271655321121216
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 3.168262243270874,
      "step": 958
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 958,
      "training_step_time": 9.947174787521362
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 3.1348230838775635,
      "step": 958
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 958,
      "training_step_time": 9.90859055519104
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 3.136216163635254,
      "step": 958
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 958,
      "training_step_time": 9.904948949813843
    },
    {
      "epoch": 3.65447998046875e-06,
      "model_forward_time": 3.135496139526367,
      "step": 958
    },
    {
      "epoch": 3.65447998046875e-06,
      "step": 958,
      "training_step_time": 10.264238119125366
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 3.456294059753418,
      "step": 959
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 959,
      "training_step_time": 10.22630524635315
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 3.132925271987915,
      "step": 959
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 959,
      "training_step_time": 9.899958848953247
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 3.132155418395996,
      "step": 959
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 959,
      "training_step_time": 9.902897596359253
    },
    {
      "epoch": 3.658294677734375e-06,
      "model_forward_time": 3.1377298831939697,
      "step": 959
    },
    {
      "epoch": 3.658294677734375e-06,
      "step": 959,
      "training_step_time": 10.014092206954956
    },
    {
      "epoch": 3.662109375e-06,
      "grad_norm": 0.8332472443580627,
      "learning_rate": 1e-05,
      "loss": 0.1338,
      "step": 960
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 3.173204183578491,
      "step": 960
    },
    {
      "dynamics_loss_avg": 0.05983040053397417,
      "epoch": 3.662109375e-06,
      "step": 960
    },
    {
      "action_loss_avg": 0.06225195769220591,
      "epoch": 3.662109375e-06,
      "step": 960
    },
    {
      "epoch": 3.662109375e-06,
      "step": 960,
      "training_step_time": 9.953113079071045
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 3.439316749572754,
      "step": 960
    },
    {
      "epoch": 3.662109375e-06,
      "step": 960,
      "training_step_time": 10.218037128448486
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 3.133727788925171,
      "step": 960
    },
    {
      "epoch": 3.662109375e-06,
      "step": 960,
      "training_step_time": 9.90924882888794
    },
    {
      "epoch": 3.662109375e-06,
      "model_forward_time": 3.1359527111053467,
      "step": 960
    },
    {
      "epoch": 3.662109375e-06,
      "step": 960,
      "training_step_time": 10.323676109313965
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 3.175271511077881,
      "step": 961
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 961,
      "training_step_time": 9.94729733467102
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 3.1333093643188477,
      "step": 961
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 961,
      "training_step_time": 9.90377402305603
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 3.3762266635894775,
      "step": 961
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 961,
      "training_step_time": 10.157537937164307
    },
    {
      "epoch": 3.665924072265625e-06,
      "model_forward_time": 3.1426799297332764,
      "step": 961
    },
    {
      "epoch": 3.665924072265625e-06,
      "step": 961,
      "training_step_time": 10.319091558456421
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 3.1890299320220947,
      "step": 962
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 962,
      "training_step_time": 9.964324474334717
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 3.1319589614868164,
      "step": 962
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 962,
      "training_step_time": 9.907670497894287
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 3.140078067779541,
      "step": 962
    },
    {
      "dynamics_loss_avg": 0.06160689759999514,
      "epoch": 3.66973876953125e-06,
      "step": 962
    },
    {
      "action_loss_avg": 0.044305622158572075,
      "epoch": 3.66973876953125e-06,
      "step": 962
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 962,
      "training_step_time": 9.914347171783447
    },
    {
      "epoch": 3.66973876953125e-06,
      "model_forward_time": 3.433964490890503,
      "step": 962
    },
    {
      "epoch": 3.66973876953125e-06,
      "step": 962,
      "training_step_time": 10.299863576889038
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 3.1723763942718506,
      "step": 963
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 963,
      "training_step_time": 9.950791120529175
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 3.1317601203918457,
      "step": 963
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 963,
      "training_step_time": 9.90280795097351
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 3.135193109512329,
      "step": 963
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 963,
      "training_step_time": 9.906657934188843
    },
    {
      "epoch": 3.673553466796875e-06,
      "model_forward_time": 3.4300878047943115,
      "step": 963
    },
    {
      "epoch": 3.673553466796875e-06,
      "step": 963,
      "training_step_time": 10.298264265060425
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 3.1705620288848877,
      "step": 964
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 964,
      "training_step_time": 9.946878910064697
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 3.1354639530181885,
      "step": 964
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 964,
      "training_step_time": 9.909794092178345
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 3.1380703449249268,
      "step": 964
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 964,
      "training_step_time": 9.91863203048706
    },
    {
      "epoch": 3.6773681640625e-06,
      "model_forward_time": 3.136723756790161,
      "step": 964
    },
    {
      "epoch": 3.6773681640625e-06,
      "step": 964,
      "training_step_time": 10.03060531616211
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 2.4460957050323486,
      "step": 965
    },
    {
      "dynamics_loss_avg": 0.08257102258503438,
      "epoch": 3.681182861328125e-06,
      "step": 965
    },
    {
      "action_loss_avg": 0.07184232883155346,
      "epoch": 3.681182861328125e-06,
      "step": 965
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 965,
      "training_step_time": 7.4803972244262695
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 3.3789095878601074,
      "step": 965
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 965,
      "training_step_time": 10.158243417739868
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 2.435153007507324,
      "step": 965
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 965,
      "training_step_time": 7.467205286026001
    },
    {
      "epoch": 3.681182861328125e-06,
      "model_forward_time": 3.136854410171509,
      "step": 965
    },
    {
      "epoch": 3.681182861328125e-06,
      "step": 965,
      "training_step_time": 10.324140071868896
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 2.473517417907715,
      "step": 966
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 966,
      "training_step_time": 7.505679368972778
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 3.1406736373901367,
      "step": 966
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 966,
      "training_step_time": 9.914299964904785
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 3.446613073348999,
      "step": 966
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 966,
      "training_step_time": 10.219952821731567
    },
    {
      "epoch": 3.68499755859375e-06,
      "model_forward_time": 3.136925220489502,
      "step": 966
    },
    {
      "epoch": 3.68499755859375e-06,
      "step": 966,
      "training_step_time": 10.325580596923828
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 3.1718826293945312,
      "step": 967
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 967,
      "training_step_time": 9.939913034439087
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 2.429854393005371,
      "step": 967
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 967,
      "training_step_time": 7.455450534820557
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 3.1301698684692383,
      "step": 967
    },
    {
      "dynamics_loss_avg": 0.07286195307970048,
      "epoch": 3.688812255859375e-06,
      "step": 967
    },
    {
      "action_loss_avg": 0.07241843324154615,
      "epoch": 3.688812255859375e-06,
      "step": 967
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 967,
      "training_step_time": 9.89894700050354
    },
    {
      "epoch": 3.688812255859375e-06,
      "model_forward_time": 3.381749391555786,
      "step": 967
    },
    {
      "epoch": 3.688812255859375e-06,
      "step": 967,
      "training_step_time": 10.29233455657959
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 3.172462224960327,
      "step": 968
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 968,
      "training_step_time": 9.941334247589111
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 3.132014036178589,
      "step": 968
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 968,
      "training_step_time": 9.899499893188477
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 3.1324403285980225,
      "step": 968
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 968,
      "training_step_time": 9.899773836135864
    },
    {
      "epoch": 3.692626953125e-06,
      "model_forward_time": 3.1472675800323486,
      "step": 968
    },
    {
      "epoch": 3.692626953125e-06,
      "step": 968,
      "training_step_time": 10.29577350616455
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 3.183213949203491,
      "step": 969
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 969,
      "training_step_time": 9.958301305770874
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 3.1358864307403564,
      "step": 969
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 969,
      "training_step_time": 9.907639265060425
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 3.1326253414154053,
      "step": 969
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 969,
      "training_step_time": 9.905601739883423
    },
    {
      "epoch": 3.696441650390625e-06,
      "model_forward_time": 3.1331448554992676,
      "step": 969
    },
    {
      "epoch": 3.696441650390625e-06,
      "step": 969,
      "training_step_time": 10.028498649597168
    },
    {
      "epoch": 3.70025634765625e-06,
      "grad_norm": 0.9129956960678101,
      "learning_rate": 1e-05,
      "loss": 0.1368,
      "step": 970
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 3.453112840652466,
      "step": 970
    },
    {
      "dynamics_loss_avg": 0.07750101462006569,
      "epoch": 3.70025634765625e-06,
      "step": 970
    },
    {
      "action_loss_avg": 0.06287671737372875,
      "epoch": 3.70025634765625e-06,
      "step": 970
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 970,
      "training_step_time": 10.229796171188354
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 3.1374080181121826,
      "step": 970
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 970,
      "training_step_time": 9.909999370574951
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 3.129987955093384,
      "step": 970
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 970,
      "training_step_time": 9.901688575744629
    },
    {
      "epoch": 3.70025634765625e-06,
      "model_forward_time": 3.1313767433166504,
      "step": 970
    },
    {
      "epoch": 3.70025634765625e-06,
      "step": 970,
      "training_step_time": 10.291411399841309
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 3.165614604949951,
      "step": 971
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 971,
      "training_step_time": 9.937164783477783
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 3.131774663925171,
      "step": 971
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 971,
      "training_step_time": 10.174434423446655
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 3.134028434753418,
      "step": 971
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 971,
      "training_step_time": 9.913462162017822
    },
    {
      "epoch": 3.704071044921875e-06,
      "model_forward_time": 3.1338589191436768,
      "step": 971
    },
    {
      "epoch": 3.704071044921875e-06,
      "step": 971,
      "training_step_time": 10.319335460662842
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 3.1656265258789062,
      "step": 972
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 972,
      "training_step_time": 9.939939975738525
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 2.438976526260376,
      "step": 972
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 972,
      "training_step_time": 7.469479084014893
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 3.4337728023529053,
      "step": 972
    },
    {
      "dynamics_loss_avg": 0.07470154594630003,
      "epoch": 3.7078857421875e-06,
      "step": 972
    },
    {
      "action_loss_avg": 0.04958639666438103,
      "epoch": 3.7078857421875e-06,
      "step": 972
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 972,
      "training_step_time": 10.21157431602478
    },
    {
      "epoch": 3.7078857421875e-06,
      "model_forward_time": 2.4364206790924072,
      "step": 972
    },
    {
      "epoch": 3.7078857421875e-06,
      "step": 972,
      "training_step_time": 10.284822225570679
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 2.4808244705200195,
      "step": 973
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 973,
      "training_step_time": 7.51082706451416
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 3.132707357406616,
      "step": 973
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 973,
      "training_step_time": 9.903060913085938
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 3.13703989982605,
      "step": 973
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 973,
      "training_step_time": 9.907687902450562
    },
    {
      "epoch": 3.711700439453125e-06,
      "model_forward_time": 3.1310577392578125,
      "step": 973
    },
    {
      "epoch": 3.711700439453125e-06,
      "step": 973,
      "training_step_time": 10.021878004074097
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 3.470407247543335,
      "step": 974
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 974,
      "training_step_time": 10.236515760421753
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 3.1336090564727783,
      "step": 974
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 974,
      "training_step_time": 9.901179313659668
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 3.135751724243164,
      "step": 974
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 974,
      "training_step_time": 9.902239799499512
    },
    {
      "epoch": 3.71551513671875e-06,
      "model_forward_time": 2.4335379600524902,
      "step": 974
    },
    {
      "epoch": 3.71551513671875e-06,
      "step": 974,
      "training_step_time": 10.285305500030518
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 3.174625873565674,
      "step": 975
    },
    {
      "dynamics_loss_avg": 0.0735815405845642,
      "epoch": 3.719329833984375e-06,
      "step": 975
    },
    {
      "action_loss_avg": 0.062054386176168916,
      "epoch": 3.719329833984375e-06,
      "step": 975
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 975,
      "training_step_time": 9.947630167007446
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 3.3859643936157227,
      "step": 975
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 975,
      "training_step_time": 10.157296657562256
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 3.1383907794952393,
      "step": 975
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 975,
      "training_step_time": 9.911821126937866
    },
    {
      "epoch": 3.719329833984375e-06,
      "model_forward_time": 3.1356518268585205,
      "step": 975
    },
    {
      "epoch": 3.719329833984375e-06,
      "step": 975,
      "training_step_time": 10.018803596496582
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 3.176018714904785,
      "step": 976
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 976,
      "training_step_time": 9.950370073318481
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 3.1363494396209717,
      "step": 976
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 976,
      "training_step_time": 9.916049718856812
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 3.4449703693389893,
      "step": 976
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 976,
      "training_step_time": 10.219087600708008
    },
    {
      "epoch": 3.72314453125e-06,
      "model_forward_time": 3.134284019470215,
      "step": 976
    },
    {
      "epoch": 3.72314453125e-06,
      "step": 976,
      "training_step_time": 10.298380613327026
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 3.154125690460205,
      "step": 977
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 977,
      "training_step_time": 9.929623365402222
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 2.4299795627593994,
      "step": 977
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 977,
      "training_step_time": 7.458498954772949
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 2.4276955127716064,
      "step": 977
    },
    {
      "dynamics_loss_avg": 0.0719617422670126,
      "epoch": 3.726959228515625e-06,
      "step": 977
    },
    {
      "action_loss_avg": 0.057602853327989575,
      "epoch": 3.726959228515625e-06,
      "step": 977
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 977,
      "training_step_time": 7.454575538635254
    },
    {
      "epoch": 3.726959228515625e-06,
      "model_forward_time": 3.368967294692993,
      "step": 977
    },
    {
      "epoch": 3.726959228515625e-06,
      "step": 977,
      "training_step_time": 10.255386590957642
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 3.1846210956573486,
      "step": 978
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 978,
      "training_step_time": 9.951941728591919
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 3.1305911540985107,
      "step": 978
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 978,
      "training_step_time": 9.898536205291748
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 3.133465528488159,
      "step": 978
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 978,
      "training_step_time": 9.902599811553955
    },
    {
      "epoch": 3.73077392578125e-06,
      "model_forward_time": 3.398846387863159,
      "step": 978
    },
    {
      "epoch": 3.73077392578125e-06,
      "step": 978,
      "training_step_time": 10.259374141693115
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 3.1602535247802734,
      "step": 979
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 979,
      "training_step_time": 9.936527252197266
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 3.1334664821624756,
      "step": 979
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 979,
      "training_step_time": 9.909702777862549
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 3.1347620487213135,
      "step": 979
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 979,
      "training_step_time": 9.91270899772644
    },
    {
      "epoch": 3.734588623046875e-06,
      "model_forward_time": 3.135942220687866,
      "step": 979
    },
    {
      "epoch": 3.734588623046875e-06,
      "step": 979,
      "training_step_time": 10.020561456680298
    },
    {
      "epoch": 3.7384033203125e-06,
      "grad_norm": 0.6422886252403259,
      "learning_rate": 1e-05,
      "loss": 0.1368,
      "step": 980
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 2.45957350730896,
      "step": 980
    },
    {
      "dynamics_loss_avg": 0.08777924589812755,
      "epoch": 3.7384033203125e-06,
      "step": 980
    },
    {
      "action_loss_avg": 0.07806007862091065,
      "epoch": 3.7384033203125e-06,
      "step": 980
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 980,
      "training_step_time": 7.497282028198242
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 3.4263405799865723,
      "step": 980
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 980,
      "training_step_time": 10.201538801193237
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 3.134140968322754,
      "step": 980
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 980,
      "training_step_time": 9.908069610595703
    },
    {
      "epoch": 3.7384033203125e-06,
      "model_forward_time": 3.1325910091400146,
      "step": 980
    },
    {
      "epoch": 3.7384033203125e-06,
      "step": 980,
      "training_step_time": 10.306716442108154
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 3.1585147380828857,
      "step": 981
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 981,
      "training_step_time": 9.935746669769287
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 3.404637098312378,
      "step": 981
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 981,
      "training_step_time": 10.192279577255249
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 3.142152786254883,
      "step": 981
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 981,
      "training_step_time": 9.923051357269287
    },
    {
      "epoch": 3.742218017578125e-06,
      "model_forward_time": 3.1403756141662598,
      "step": 981
    },
    {
      "epoch": 3.742218017578125e-06,
      "step": 981,
      "training_step_time": 10.284200429916382
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 3.1618432998657227,
      "step": 982
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 982,
      "training_step_time": 9.94378113746643
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 3.13885235786438,
      "step": 982
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 982,
      "training_step_time": 9.919185400009155
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 3.1433165073394775,
      "step": 982
    },
    {
      "dynamics_loss_avg": 0.07942323349416255,
      "epoch": 3.74603271484375e-06,
      "step": 982
    },
    {
      "action_loss_avg": 0.06122681684792042,
      "epoch": 3.74603271484375e-06,
      "step": 982
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 982,
      "training_step_time": 10.204284191131592
    },
    {
      "epoch": 3.74603271484375e-06,
      "model_forward_time": 2.43803071975708,
      "step": 982
    },
    {
      "epoch": 3.74603271484375e-06,
      "step": 982,
      "training_step_time": 10.320090532302856
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 2.467691421508789,
      "step": 983
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 983,
      "training_step_time": 7.497140884399414
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 3.1341588497161865,
      "step": 983
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 983,
      "training_step_time": 9.910447597503662
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 3.1430017948150635,
      "step": 983
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 983,
      "training_step_time": 9.918123960494995
    },
    {
      "epoch": 3.749847412109375e-06,
      "model_forward_time": 3.13448166847229,
      "step": 983
    },
    {
      "epoch": 3.749847412109375e-06,
      "step": 983,
      "training_step_time": 10.311212301254272
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 3.1825709342956543,
      "step": 984
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 984,
      "training_step_time": 9.964195489883423
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 3.136329412460327,
      "step": 984
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 984,
      "training_step_time": 9.919189453125
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 3.138627529144287,
      "step": 984
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 984,
      "training_step_time": 9.913283109664917
    },
    {
      "epoch": 3.753662109375e-06,
      "model_forward_time": 3.1419100761413574,
      "step": 984
    },
    {
      "epoch": 3.753662109375e-06,
      "step": 984,
      "training_step_time": 10.29242992401123
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 3.1851141452789307,
      "step": 985
    },
    {
      "dynamics_loss_avg": 0.06780528388917446,
      "epoch": 3.757476806640625e-06,
      "step": 985
    },
    {
      "action_loss_avg": 0.05963738039135933,
      "epoch": 3.757476806640625e-06,
      "step": 985
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 985,
      "training_step_time": 10.242218494415283
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 3.1424245834350586,
      "step": 985
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 985,
      "training_step_time": 9.917346477508545
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 3.1369593143463135,
      "step": 985
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 985,
      "training_step_time": 9.917518854141235
    },
    {
      "epoch": 3.757476806640625e-06,
      "model_forward_time": 3.137291431427002,
      "step": 985
    },
    {
      "epoch": 3.757476806640625e-06,
      "step": 985,
      "training_step_time": 10.259417533874512
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 3.1581077575683594,
      "step": 986
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 986,
      "training_step_time": 9.930381536483765
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 3.426898956298828,
      "step": 986
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 986,
      "training_step_time": 10.20090627670288
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 2.4351727962493896,
      "step": 986
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 986,
      "training_step_time": 7.467727184295654
    },
    {
      "epoch": 3.76129150390625e-06,
      "model_forward_time": 3.1412198543548584,
      "step": 986
    },
    {
      "epoch": 3.76129150390625e-06,
      "step": 986,
      "training_step_time": 10.32531213760376
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 2.4559621810913086,
      "step": 987
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 987,
      "training_step_time": 7.489183664321899
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 3.1419527530670166,
      "step": 987
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 987,
      "training_step_time": 9.914491891860962
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 2.4351487159729004,
      "step": 987
    },
    {
      "dynamics_loss_avg": 0.0641190230846405,
      "epoch": 3.765106201171875e-06,
      "step": 987
    },
    {
      "action_loss_avg": 0.06471935119479895,
      "epoch": 3.765106201171875e-06,
      "step": 987
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 987,
      "training_step_time": 7.467942476272583
    },
    {
      "epoch": 3.765106201171875e-06,
      "model_forward_time": 2.6896185874938965,
      "step": 987
    },
    {
      "epoch": 3.765106201171875e-06,
      "step": 987,
      "training_step_time": 10.302260160446167
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 2.4717812538146973,
      "step": 988
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 988,
      "training_step_time": 7.495466709136963
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 3.1365716457366943,
      "step": 988
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 988,
      "training_step_time": 9.905853271484375
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 3.138532876968384,
      "step": 988
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 988,
      "training_step_time": 9.910065174102783
    },
    {
      "epoch": 3.7689208984375e-06,
      "model_forward_time": 3.1405842304229736,
      "step": 988
    },
    {
      "epoch": 3.7689208984375e-06,
      "step": 988,
      "training_step_time": 10.024590492248535
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 3.4566314220428467,
      "step": 989
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 989,
      "training_step_time": 10.249638557434082
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 3.143609046936035,
      "step": 989
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 989,
      "training_step_time": 9.917486429214478
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 3.140700340270996,
      "step": 989
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 989,
      "training_step_time": 9.919483184814453
    },
    {
      "epoch": 3.772735595703125e-06,
      "model_forward_time": 3.1463727951049805,
      "step": 989
    },
    {
      "epoch": 3.772735595703125e-06,
      "step": 989,
      "training_step_time": 10.241295576095581
    },
    {
      "epoch": 3.77655029296875e-06,
      "grad_norm": 0.9598671793937683,
      "learning_rate": 1e-05,
      "loss": 0.1377,
      "step": 990
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 3.1598448753356934,
      "step": 990
    },
    {
      "dynamics_loss_avg": 0.07926532216370105,
      "epoch": 3.77655029296875e-06,
      "step": 990
    },
    {
      "action_loss_avg": 0.06629691310226918,
      "epoch": 3.77655029296875e-06,
      "step": 990
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 990,
      "training_step_time": 9.937748670578003
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 3.447413921356201,
      "step": 990
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 990,
      "training_step_time": 10.21644902229309
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 3.140381097793579,
      "step": 990
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 990,
      "training_step_time": 9.912642240524292
    },
    {
      "epoch": 3.77655029296875e-06,
      "model_forward_time": 2.4354851245880127,
      "step": 990
    },
    {
      "epoch": 3.77655029296875e-06,
      "step": 990,
      "training_step_time": 10.277520656585693
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 3.1953680515289307,
      "step": 991
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 991,
      "training_step_time": 9.967417478561401
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 3.1377954483032227,
      "step": 991
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 991,
      "training_step_time": 9.916632652282715
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 3.4084041118621826,
      "step": 991
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 991,
      "training_step_time": 10.178789377212524
    },
    {
      "epoch": 3.780364990234375e-06,
      "model_forward_time": 3.1429755687713623,
      "step": 991
    },
    {
      "epoch": 3.780364990234375e-06,
      "step": 991,
      "training_step_time": 10.289286851882935
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 3.170780658721924,
      "step": 992
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 992,
      "training_step_time": 9.94750690460205
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 3.1411099433898926,
      "step": 992
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 992,
      "training_step_time": 9.917678356170654
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 3.1447229385375977,
      "step": 992
    },
    {
      "dynamics_loss_avg": 0.07107761166989804,
      "epoch": 3.7841796875e-06,
      "step": 992
    },
    {
      "action_loss_avg": 0.05747926011681557,
      "epoch": 3.7841796875e-06,
      "step": 992
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 992,
      "training_step_time": 9.921846151351929
    },
    {
      "epoch": 3.7841796875e-06,
      "model_forward_time": 3.4144089221954346,
      "step": 992
    },
    {
      "epoch": 3.7841796875e-06,
      "step": 992,
      "training_step_time": 10.293560028076172
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 3.193742513656616,
      "step": 993
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 993,
      "training_step_time": 9.972108602523804
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 3.1408040523529053,
      "step": 993
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 993,
      "training_step_time": 9.917036056518555
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 3.1463675498962402,
      "step": 993
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 993,
      "training_step_time": 9.925663232803345
    },
    {
      "epoch": 3.787994384765625e-06,
      "model_forward_time": 3.1410698890686035,
      "step": 993
    },
    {
      "epoch": 3.787994384765625e-06,
      "step": 993,
      "training_step_time": 10.019983053207397
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 3.4631998538970947,
      "step": 994
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 994,
      "training_step_time": 10.241763353347778
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 3.14046311378479,
      "step": 994
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 994,
      "training_step_time": 9.914316654205322
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 2.437863826751709,
      "step": 994
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 994,
      "training_step_time": 7.467026472091675
    },
    {
      "epoch": 3.79180908203125e-06,
      "model_forward_time": 3.137312173843384,
      "step": 994
    },
    {
      "epoch": 3.79180908203125e-06,
      "step": 994,
      "training_step_time": 10.28748345375061
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 3.436222553253174,
      "step": 995
    },
    {
      "dynamics_loss_avg": 0.08127466347068549,
      "epoch": 3.795623779296875e-06,
      "step": 995
    },
    {
      "action_loss_avg": 0.06454540491104126,
      "epoch": 3.795623779296875e-06,
      "step": 995
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 995,
      "training_step_time": 10.20700216293335
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 3.142035484313965,
      "step": 995
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 995,
      "training_step_time": 9.911847114562988
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 3.1378417015075684,
      "step": 995
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 995,
      "training_step_time": 9.90462875366211
    },
    {
      "epoch": 3.795623779296875e-06,
      "model_forward_time": 2.441049337387085,
      "step": 995
    },
    {
      "epoch": 3.795623779296875e-06,
      "step": 995,
      "training_step_time": 10.02468466758728
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 3.156994104385376,
      "step": 996
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 996,
      "training_step_time": 9.92993950843811
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 3.138232469558716,
      "step": 996
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 996,
      "training_step_time": 9.906090021133423
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 2.739715337753296,
      "step": 996
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 996,
      "training_step_time": 7.765872240066528
    },
    {
      "epoch": 3.7994384765625e-06,
      "model_forward_time": 3.1351468563079834,
      "step": 996
    },
    {
      "epoch": 3.7994384765625e-06,
      "step": 996,
      "training_step_time": 10.017845869064331
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 3.156019449234009,
      "step": 997
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 997,
      "training_step_time": 9.928462266921997
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 3.137158155441284,
      "step": 997
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 997,
      "training_step_time": 9.908456802368164
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 2.4361326694488525,
      "step": 997
    },
    {
      "dynamics_loss_avg": 0.06316368132829667,
      "epoch": 3.803253173828125e-06,
      "step": 997
    },
    {
      "action_loss_avg": 0.06758919032290578,
      "epoch": 3.803253173828125e-06,
      "step": 997
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 997,
      "training_step_time": 7.464313983917236
    },
    {
      "epoch": 3.803253173828125e-06,
      "model_forward_time": 3.3919341564178467,
      "step": 997
    },
    {
      "epoch": 3.803253173828125e-06,
      "step": 997,
      "training_step_time": 10.32772159576416
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 1.7829058170318604,
      "step": 998
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 998,
      "training_step_time": 5.24957799911499
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 2.4323089122772217,
      "step": 998
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 998,
      "training_step_time": 7.455899477005005
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 3.1383628845214844,
      "step": 998
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 998,
      "training_step_time": 9.908109903335571
    },
    {
      "epoch": 3.80706787109375e-06,
      "model_forward_time": 3.1388635635375977,
      "step": 998
    },
    {
      "epoch": 3.80706787109375e-06,
      "step": 998,
      "training_step_time": 10.030414581298828
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 3.5466809272766113,
      "step": 999
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 999,
      "training_step_time": 10.322415828704834
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 3.1396095752716064,
      "step": 999
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 999,
      "training_step_time": 9.905815362930298
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 3.136626720428467,
      "step": 999
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 999,
      "training_step_time": 9.907804250717163
    },
    {
      "epoch": 3.810882568359375e-06,
      "model_forward_time": 3.140434980392456,
      "step": 999
    },
    {
      "epoch": 3.810882568359375e-06,
      "step": 999,
      "training_step_time": 10.02119255065918
    },
    {
      "epoch": 3.814697265625e-06,
      "grad_norm": 1.098722219467163,
      "learning_rate": 1e-05,
      "loss": 0.1349,
      "step": 1000
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 5.0912370681762695,
      "step": 1000
    },
    {
      "dynamics_loss_avg": 0.06837313007563353,
      "epoch": 3.814697265625e-06,
      "step": 1000
    },
    {
      "action_loss_avg": 0.06053274963051081,
      "epoch": 3.814697265625e-06,
      "step": 1000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 1000,
      "training_step_time": 11.852210760116577
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 3.144165515899658,
      "step": 1000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 1000,
      "training_step_time": 9.887123584747314
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 3.1344661712646484,
      "step": 1000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 1000,
      "training_step_time": 9.886296272277832
    },
    {
      "epoch": 3.814697265625e-06,
      "model_forward_time": 3.133678674697876,
      "step": 1000
    },
    {
      "epoch": 3.814697265625e-06,
      "step": 1000,
      "training_step_time": 10.2978835105896
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 3.1789050102233887,
      "step": 1001
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 1001,
      "training_step_time": 9.941202878952026
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 3.4320435523986816,
      "step": 1001
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 1001,
      "training_step_time": 10.196874618530273
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 3.1446645259857178,
      "step": 1001
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 1001,
      "training_step_time": 9.90635085105896
    },
    {
      "epoch": 3.818511962890625e-06,
      "model_forward_time": 2.434490203857422,
      "step": 1001
    },
    {
      "epoch": 3.818511962890625e-06,
      "step": 1001,
      "training_step_time": 10.272261619567871
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 2.4488062858581543,
      "step": 1002
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 1002,
      "training_step_time": 7.472402811050415
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 3.13407301902771,
      "step": 1002
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 1002,
      "training_step_time": 9.89226222038269
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 2.7102982997894287,
      "step": 1002
    },
    {
      "dynamics_loss_avg": 0.07356749586760998,
      "epoch": 3.82232666015625e-06,
      "step": 1002
    },
    {
      "action_loss_avg": 0.07062516286969185,
      "epoch": 3.82232666015625e-06,
      "step": 1002
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 1002,
      "training_step_time": 7.72992467880249
    },
    {
      "epoch": 3.82232666015625e-06,
      "model_forward_time": 2.430143356323242,
      "step": 1002
    },
    {
      "epoch": 3.82232666015625e-06,
      "step": 1002,
      "training_step_time": 10.30982780456543
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 2.452385902404785,
      "step": 1003
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 1003,
      "training_step_time": 7.463167905807495
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 3.1357581615448,
      "step": 1003
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 1003,
      "training_step_time": 9.889558553695679
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 2.43461012840271,
      "step": 1003
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 1003,
      "training_step_time": 7.448309898376465
    },
    {
      "epoch": 3.826141357421875e-06,
      "model_forward_time": 3.13515567779541,
      "step": 1003
    },
    {
      "epoch": 3.826141357421875e-06,
      "step": 1003,
      "training_step_time": 10.290109157562256
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 3.4880409240722656,
      "step": 1004
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 1004,
      "training_step_time": 10.24863600730896
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 3.138040065765381,
      "step": 1004
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 1004,
      "training_step_time": 9.903896808624268
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 2.435305595397949,
      "step": 1004
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 1004,
      "training_step_time": 7.459346055984497
    },
    {
      "epoch": 3.8299560546875e-06,
      "model_forward_time": 3.1395785808563232,
      "step": 1004
    },
    {
      "epoch": 3.8299560546875e-06,
      "step": 1004,
      "training_step_time": 10.037183284759521
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 2.471721649169922,
      "step": 1005
    },
    {
      "dynamics_loss_avg": 0.07482499331235885,
      "epoch": 3.833770751953125e-06,
      "step": 1005
    },
    {
      "action_loss_avg": 0.06742027252912522,
      "epoch": 3.833770751953125e-06,
      "step": 1005
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 1005,
      "training_step_time": 7.501846790313721
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 3.4435207843780518,
      "step": 1005
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 1005,
      "training_step_time": 10.213263273239136
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 3.1397886276245117,
      "step": 1005
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 1005,
      "training_step_time": 9.909399509429932
    },
    {
      "epoch": 3.833770751953125e-06,
      "model_forward_time": 3.1463804244995117,
      "step": 1005
    },
    {
      "epoch": 3.833770751953125e-06,
      "step": 1005,
      "training_step_time": 10.044901132583618
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 3.155937910079956,
      "step": 1006
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 1006,
      "training_step_time": 9.929351329803467
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 2.4408233165740967,
      "step": 1006
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 1006,
      "training_step_time": 7.467170238494873
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 3.4004123210906982,
      "step": 1006
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 1006,
      "training_step_time": 10.20854663848877
    },
    {
      "epoch": 3.83758544921875e-06,
      "model_forward_time": 2.4382781982421875,
      "step": 1006
    },
    {
      "epoch": 3.83758544921875e-06,
      "step": 1006,
      "training_step_time": 10.31691861152649
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 3.2018074989318848,
      "step": 1007
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 1007,
      "training_step_time": 9.97208571434021
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 3.145993709564209,
      "step": 1007
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 1007,
      "training_step_time": 9.916462182998657
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 3.144770860671997,
      "step": 1007
    },
    {
      "dynamics_loss_avg": 0.08160937651991844,
      "epoch": 3.841400146484375e-06,
      "step": 1007
    },
    {
      "action_loss_avg": 0.058390486426651476,
      "epoch": 3.841400146484375e-06,
      "step": 1007
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 1007,
      "training_step_time": 9.918430805206299
    },
    {
      "epoch": 3.841400146484375e-06,
      "model_forward_time": 3.4153451919555664,
      "step": 1007
    },
    {
      "epoch": 3.841400146484375e-06,
      "step": 1007,
      "training_step_time": 10.280489921569824
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 3.1593008041381836,
      "step": 1008
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 1008,
      "training_step_time": 9.932564735412598
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 3.1361794471740723,
      "step": 1008
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 1008,
      "training_step_time": 9.899779081344604
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 3.148528575897217,
      "step": 1008
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 1008,
      "training_step_time": 9.916349411010742
    },
    {
      "epoch": 3.84521484375e-06,
      "model_forward_time": 2.431234121322632,
      "step": 1008
    },
    {
      "epoch": 3.84521484375e-06,
      "step": 1008,
      "training_step_time": 10.322445631027222
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 2.7796826362609863,
      "step": 1009
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 1009,
      "training_step_time": 7.800451993942261
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 3.1383330821990967,
      "step": 1009
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 1009,
      "training_step_time": 9.90520977973938
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 2.433133363723755,
      "step": 1009
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 1009,
      "training_step_time": 7.452976942062378
    },
    {
      "epoch": 3.849029541015625e-06,
      "model_forward_time": 2.4341626167297363,
      "step": 1009
    },
    {
      "epoch": 3.849029541015625e-06,
      "step": 1009,
      "training_step_time": 10.027680397033691
    },
    {
      "epoch": 3.85284423828125e-06,
      "grad_norm": 0.7411555051803589,
      "learning_rate": 1e-05,
      "loss": 0.1367,
      "step": 1010
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 3.151916742324829,
      "step": 1010
    },
    {
      "dynamics_loss_avg": 0.06763768158853053,
      "epoch": 3.85284423828125e-06,
      "step": 1010
    },
    {
      "action_loss_avg": 0.04582874402403832,
      "epoch": 3.85284423828125e-06,
      "step": 1010
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 1010,
      "training_step_time": 9.919613599777222
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 2.7209036350250244,
      "step": 1010
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 1010,
      "training_step_time": 7.742582559585571
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 3.137885808944702,
      "step": 1010
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 1010,
      "training_step_time": 9.902661800384521
    },
    {
      "epoch": 3.85284423828125e-06,
      "model_forward_time": 2.432530641555786,
      "step": 1010
    },
    {
      "epoch": 3.85284423828125e-06,
      "step": 1010,
      "training_step_time": 10.023423671722412
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 3.1701853275299072,
      "step": 1011
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 1011,
      "training_step_time": 9.936457633972168
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 2.437725305557251,
      "step": 1011
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 1011,
      "training_step_time": 7.466570138931274
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 3.13909912109375,
      "step": 1011
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 1011,
      "training_step_time": 10.18332815170288
    },
    {
      "epoch": 3.856658935546875e-06,
      "model_forward_time": 3.1395702362060547,
      "step": 1011
    },
    {
      "epoch": 3.856658935546875e-06,
      "step": 1011,
      "training_step_time": 10.314669609069824
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 3.1541783809661865,
      "step": 1012
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 1012,
      "training_step_time": 9.92326307296753
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 3.142798900604248,
      "step": 1012
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 1012,
      "training_step_time": 9.915058135986328
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 2.4392647743225098,
      "step": 1012
    },
    {
      "dynamics_loss_avg": 0.06767295487225056,
      "epoch": 3.8604736328125e-06,
      "step": 1012
    },
    {
      "action_loss_avg": 0.06571266502141952,
      "epoch": 3.8604736328125e-06,
      "step": 1012
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 1012,
      "training_step_time": 7.466053247451782
    },
    {
      "epoch": 3.8604736328125e-06,
      "model_forward_time": 2.437657594680786,
      "step": 1012
    },
    {
      "epoch": 3.8604736328125e-06,
      "step": 1012,
      "training_step_time": 10.018707752227783
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 2.738118886947632,
      "step": 1013
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 1013,
      "training_step_time": 7.7599098682403564
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 3.134561538696289,
      "step": 1013
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 1013,
      "training_step_time": 9.89331579208374
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 3.1346421241760254,
      "step": 1013
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 1013,
      "training_step_time": 9.894505262374878
    },
    {
      "epoch": 3.864288330078125e-06,
      "model_forward_time": 2.442063808441162,
      "step": 1013
    },
    {
      "epoch": 3.864288330078125e-06,
      "step": 1013,
      "training_step_time": 10.296940803527832
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 2.485530376434326,
      "step": 1014
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 1014,
      "training_step_time": 7.515502214431763
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 3.4527244567871094,
      "step": 1014
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 1014,
      "training_step_time": 10.215264558792114
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 3.135941505432129,
      "step": 1014
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 1014,
      "training_step_time": 9.897167682647705
    },
    {
      "epoch": 3.86810302734375e-06,
      "model_forward_time": 3.140012741088867,
      "step": 1014
    },
    {
      "epoch": 3.86810302734375e-06,
      "step": 1014,
      "training_step_time": 10.299424171447754
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 2.4510622024536133,
      "step": 1015
    },
    {
      "dynamics_loss_avg": 0.07563806362450123,
      "epoch": 3.871917724609375e-06,
      "step": 1015
    },
    {
      "action_loss_avg": 0.06396246217191219,
      "epoch": 3.871917724609375e-06,
      "step": 1015
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 1015,
      "training_step_time": 7.474910259246826
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 3.1345975399017334,
      "step": 1015
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 1015,
      "training_step_time": 9.895313024520874
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 2.7153570652008057,
      "step": 1015
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 1015,
      "training_step_time": 7.737993240356445
    },
    {
      "epoch": 3.871917724609375e-06,
      "model_forward_time": 2.433375835418701,
      "step": 1015
    },
    {
      "epoch": 3.871917724609375e-06,
      "step": 1015,
      "training_step_time": 10.019030332565308
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 1.7606275081634521,
      "step": 1016
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 1016,
      "training_step_time": 5.2149271965026855
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 3.136122226715088,
      "step": 1016
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 1016,
      "training_step_time": 9.896528244018555
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 2.438006639480591,
      "step": 1016
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 1016,
      "training_step_time": 7.46815037727356
    },
    {
      "epoch": 3.875732421875e-06,
      "model_forward_time": 2.4350509643554688,
      "step": 1016
    },
    {
      "epoch": 3.875732421875e-06,
      "step": 1016,
      "training_step_time": 10.044435024261475
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 2.0764873027801514,
      "step": 1017
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 1017,
      "training_step_time": 5.525487184524536
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 3.133894205093384,
      "step": 1017
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 1017,
      "training_step_time": 9.891414880752563
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 3.1357574462890625,
      "step": 1017
    },
    {
      "dynamics_loss_avg": 0.07693404033780098,
      "epoch": 3.879547119140625e-06,
      "step": 1017
    },
    {
      "action_loss_avg": 0.059582775738090274,
      "epoch": 3.879547119140625e-06,
      "step": 1017
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 1017,
      "training_step_time": 9.904176235198975
    },
    {
      "epoch": 3.879547119140625e-06,
      "model_forward_time": 3.140549898147583,
      "step": 1017
    },
    {
      "epoch": 3.879547119140625e-06,
      "step": 1017,
      "training_step_time": 10.292369842529297
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 3.1489710807800293,
      "step": 1018
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 1018,
      "training_step_time": 9.910912036895752
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 2.7214760780334473,
      "step": 1018
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 1018,
      "training_step_time": 7.746361255645752
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 2.430838108062744,
      "step": 1018
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 1018,
      "training_step_time": 7.448618650436401
    },
    {
      "epoch": 3.88336181640625e-06,
      "model_forward_time": 2.4354360103607178,
      "step": 1018
    },
    {
      "epoch": 3.88336181640625e-06,
      "step": 1018,
      "training_step_time": 10.311567783355713
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 3.1566600799560547,
      "step": 1019
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 1019,
      "training_step_time": 9.915069818496704
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 2.4325180053710938,
      "step": 1019
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 1019,
      "training_step_time": 7.466573715209961
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 3.130307197570801,
      "step": 1019
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 1019,
      "training_step_time": 9.88283371925354
    },
    {
      "epoch": 3.887176513671875e-06,
      "model_forward_time": 2.7137207984924316,
      "step": 1019
    },
    {
      "epoch": 3.887176513671875e-06,
      "step": 1019,
      "training_step_time": 10.294389486312866
    },
    {
      "epoch": 3.8909912109375e-06,
      "grad_norm": 0.629910945892334,
      "learning_rate": 1e-05,
      "loss": 0.1399,
      "step": 1020
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 2.4398744106292725,
      "step": 1020
    },
    {
      "dynamics_loss_avg": 0.0719239866361022,
      "epoch": 3.8909912109375e-06,
      "step": 1020
    },
    {
      "action_loss_avg": 0.055556478444486856,
      "epoch": 3.8909912109375e-06,
      "step": 1020
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 1020,
      "training_step_time": 7.452405214309692
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 2.4213037490844727,
      "step": 1020
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 1020,
      "training_step_time": 7.4335339069366455
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 2.4207303524017334,
      "step": 1020
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 1020,
      "training_step_time": 7.436909437179565
    },
    {
      "epoch": 3.8909912109375e-06,
      "model_forward_time": 3.125591516494751,
      "step": 1020
    },
    {
      "epoch": 3.8909912109375e-06,
      "step": 1020,
      "training_step_time": 10.009245157241821
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 3.4197113513946533,
      "step": 1021
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 1021,
      "training_step_time": 10.185573101043701
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 2.4240403175354004,
      "step": 1021
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 1021,
      "training_step_time": 7.447152376174927
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 2.425243377685547,
      "step": 1021
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 1021,
      "training_step_time": 7.445199012756348
    },
    {
      "epoch": 3.894805908203125e-06,
      "model_forward_time": 3.1225454807281494,
      "step": 1021
    },
    {
      "epoch": 3.894805908203125e-06,
      "step": 1021,
      "training_step_time": 10.01291036605835
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 3.1387829780578613,
      "step": 1022
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 1022,
      "training_step_time": 9.900749444961548
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 3.4237728118896484,
      "step": 1022
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 1022,
      "training_step_time": 10.18637466430664
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 2.427978277206421,
      "step": 1022
    },
    {
      "dynamics_loss_avg": 0.06533104293048382,
      "epoch": 3.89862060546875e-06,
      "step": 1022
    },
    {
      "action_loss_avg": 0.07933156564831734,
      "epoch": 3.89862060546875e-06,
      "step": 1022
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 1022,
      "training_step_time": 7.453643083572388
    },
    {
      "epoch": 3.89862060546875e-06,
      "model_forward_time": 3.1299781799316406,
      "step": 1022
    },
    {
      "epoch": 3.89862060546875e-06,
      "step": 1022,
      "training_step_time": 10.279912233352661
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 2.4651129245758057,
      "step": 1023
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 1023,
      "training_step_time": 7.488598585128784
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 2.424417734146118,
      "step": 1023
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 1023,
      "training_step_time": 7.441620826721191
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 3.1266865730285645,
      "step": 1023
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 1023,
      "training_step_time": 10.197615623474121
    },
    {
      "epoch": 3.902435302734375e-06,
      "model_forward_time": 3.123107671737671,
      "step": 1023
    },
    {
      "epoch": 3.902435302734375e-06,
      "step": 1023,
      "training_step_time": 10.314847946166992
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 2.463408946990967,
      "step": 1024
    },
    {
      "epoch": 3.90625e-06,
      "step": 1024,
      "training_step_time": 7.482328176498413
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 2.4244260787963867,
      "step": 1024
    },
    {
      "epoch": 3.90625e-06,
      "step": 1024,
      "training_step_time": 7.4391186237335205
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 3.128401279449463,
      "step": 1024
    },
    {
      "epoch": 3.90625e-06,
      "step": 1024,
      "training_step_time": 9.888039350509644
    },
    {
      "epoch": 3.90625e-06,
      "model_forward_time": 2.428179979324341,
      "step": 1024
    },
    {
      "epoch": 3.90625e-06,
      "step": 1024,
      "training_step_time": 10.277411699295044
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 2.071732997894287,
      "step": 1025
    },
    {
      "dynamics_loss_avg": 0.08290244434028864,
      "epoch": 3.910064697265625e-06,
      "step": 1025
    },
    {
      "action_loss_avg": 0.06450797747820616,
      "epoch": 3.910064697265625e-06,
      "step": 1025
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 1025,
      "training_step_time": 5.519678354263306
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 3.1234068870544434,
      "step": 1025
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 1025,
      "training_step_time": 9.88223123550415
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 3.1319897174835205,
      "step": 1025
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 1025,
      "training_step_time": 9.891966819763184
    },
    {
      "epoch": 3.910064697265625e-06,
      "model_forward_time": 3.1273815631866455,
      "step": 1025
    },
    {
      "epoch": 3.910064697265625e-06,
      "step": 1025,
      "training_step_time": 10.275846719741821
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 1.7792024612426758,
      "step": 1026
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 1026,
      "training_step_time": 5.226587772369385
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 3.126697063446045,
      "step": 1026
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 1026,
      "training_step_time": 9.880737543106079
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 2.7043495178222656,
      "step": 1026
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 1026,
      "training_step_time": 7.715277433395386
    },
    {
      "epoch": 3.91387939453125e-06,
      "model_forward_time": 3.125981092453003,
      "step": 1026
    },
    {
      "epoch": 3.91387939453125e-06,
      "step": 1026,
      "training_step_time": 10.007984161376953
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 2.4430558681488037,
      "step": 1027
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 1027,
      "training_step_time": 7.454702615737915
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 3.1321940422058105,
      "step": 1027
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 1027,
      "training_step_time": 9.888181209564209
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 2.42836856842041,
      "step": 1027
    },
    {
      "dynamics_loss_avg": 0.08334853500127792,
      "epoch": 3.917694091796875e-06,
      "step": 1027
    },
    {
      "action_loss_avg": 0.0773921974003315,
      "epoch": 3.917694091796875e-06,
      "step": 1027
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 1027,
      "training_step_time": 7.451651096343994
    },
    {
      "epoch": 3.917694091796875e-06,
      "model_forward_time": 2.7252919673919678,
      "step": 1027
    },
    {
      "epoch": 3.917694091796875e-06,
      "step": 1027,
      "training_step_time": 10.247867345809937
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 3.1432833671569824,
      "step": 1028
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 1028,
      "training_step_time": 9.906568765640259
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 3.1339166164398193,
      "step": 1028
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 1028,
      "training_step_time": 9.897910833358765
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 2.431752920150757,
      "step": 1028
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 1028,
      "training_step_time": 7.4538867473602295
    },
    {
      "epoch": 3.9215087890625e-06,
      "model_forward_time": 2.4304909706115723,
      "step": 1028
    },
    {
      "epoch": 3.9215087890625e-06,
      "step": 1028,
      "training_step_time": 10.298128128051758
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 2.7109200954437256,
      "step": 1029
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 1029,
      "training_step_time": 7.733051061630249
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 1.7434275150299072,
      "step": 1029
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 1029,
      "training_step_time": 5.195577383041382
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 2.430663585662842,
      "step": 1029
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 1029,
      "training_step_time": 7.455962896347046
    },
    {
      "epoch": 3.925323486328125e-06,
      "model_forward_time": 3.137795925140381,
      "step": 1029
    },
    {
      "epoch": 3.925323486328125e-06,
      "step": 1029,
      "training_step_time": 10.28952932357788
    },
    {
      "epoch": 3.92913818359375e-06,
      "grad_norm": 0.8532668948173523,
      "learning_rate": 1e-05,
      "loss": 0.1395,
      "step": 1030
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 2.466028928756714,
      "step": 1030
    },
    {
      "dynamics_loss_avg": 0.07629346400499344,
      "epoch": 3.92913818359375e-06,
      "step": 1030
    },
    {
      "action_loss_avg": 0.06581431496888399,
      "epoch": 3.92913818359375e-06,
      "step": 1030
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 1030,
      "training_step_time": 7.491239070892334
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 2.430629253387451,
      "step": 1030
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 1030,
      "training_step_time": 7.450627088546753
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 2.690898895263672,
      "step": 1030
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 1030,
      "training_step_time": 7.7086021900177
    },
    {
      "epoch": 3.92913818359375e-06,
      "model_forward_time": 3.1298439502716064,
      "step": 1030
    },
    {
      "epoch": 3.92913818359375e-06,
      "step": 1030,
      "training_step_time": 10.319489002227783
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 3.145768642425537,
      "step": 1031
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 1031,
      "training_step_time": 9.909141302108765
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 3.1311120986938477,
      "step": 1031
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 1031,
      "training_step_time": 9.895810604095459
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 2.429354667663574,
      "step": 1031
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 1031,
      "training_step_time": 7.45123815536499
    },
    {
      "epoch": 3.932952880859375e-06,
      "model_forward_time": 2.731238603591919,
      "step": 1031
    },
    {
      "epoch": 3.932952880859375e-06,
      "step": 1031,
      "training_step_time": 10.28281021118164
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 3.181288003921509,
      "step": 1032
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 1032,
      "training_step_time": 9.944620132446289
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 3.134601593017578,
      "step": 1032
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 1032,
      "training_step_time": 9.900729417800903
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 2.437910318374634,
      "step": 1032
    },
    {
      "dynamics_loss_avg": 0.09415050484240055,
      "epoch": 3.936767578125e-06,
      "step": 1032
    },
    {
      "action_loss_avg": 0.06827402245253325,
      "epoch": 3.936767578125e-06,
      "step": 1032
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 1032,
      "training_step_time": 7.470617294311523
    },
    {
      "epoch": 3.936767578125e-06,
      "model_forward_time": 3.131990432739258,
      "step": 1032
    },
    {
      "epoch": 3.936767578125e-06,
      "step": 1032,
      "training_step_time": 10.273900032043457
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 3.431750774383545,
      "step": 1033
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 1033,
      "training_step_time": 10.200569868087769
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 2.4322543144226074,
      "step": 1033
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 1033,
      "training_step_time": 7.4545042514801025
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 2.430152654647827,
      "step": 1033
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 1033,
      "training_step_time": 7.453758001327515
    },
    {
      "epoch": 3.940582275390625e-06,
      "model_forward_time": 2.42931866645813,
      "step": 1033
    },
    {
      "epoch": 3.940582275390625e-06,
      "step": 1033,
      "training_step_time": 10.330200672149658
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 3.168200969696045,
      "step": 1034
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 1034,
      "training_step_time": 9.931750059127808
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 2.7042222023010254,
      "step": 1034
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 1034,
      "training_step_time": 7.724119663238525
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 2.427144765853882,
      "step": 1034
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 1034,
      "training_step_time": 7.444300889968872
    },
    {
      "epoch": 3.94439697265625e-06,
      "model_forward_time": 3.128535509109497,
      "step": 1034
    },
    {
      "epoch": 3.94439697265625e-06,
      "step": 1034,
      "training_step_time": 10.010848760604858
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 2.444453716278076,
      "step": 1035
    },
    {
      "dynamics_loss_avg": 0.0901846781373024,
      "epoch": 3.948211669921875e-06,
      "step": 1035
    },
    {
      "action_loss_avg": 0.08194327391684056,
      "epoch": 3.948211669921875e-06,
      "step": 1035
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 1035,
      "training_step_time": 7.471604108810425
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 3.1343607902526855,
      "step": 1035
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 1035,
      "training_step_time": 9.899438858032227
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 1.7418091297149658,
      "step": 1035
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 1035,
      "training_step_time": 5.192217826843262
    },
    {
      "epoch": 3.948211669921875e-06,
      "model_forward_time": 3.398268938064575,
      "step": 1035
    },
    {
      "epoch": 3.948211669921875e-06,
      "step": 1035,
      "training_step_time": 10.28247857093811
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 3.176745891571045,
      "step": 1036
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 1036,
      "training_step_time": 9.954729318618774
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 2.433838367462158,
      "step": 1036
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 1036,
      "training_step_time": 7.457931280136108
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 3.1280269622802734,
      "step": 1036
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 1036,
      "training_step_time": 9.897635698318481
    },
    {
      "epoch": 3.9520263671875e-06,
      "model_forward_time": 3.1363956928253174,
      "step": 1036
    },
    {
      "epoch": 3.9520263671875e-06,
      "step": 1036,
      "training_step_time": 10.286083221435547
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 3.4046730995178223,
      "step": 1037
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 1037,
      "training_step_time": 10.17736029624939
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 2.433135747909546,
      "step": 1037
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 1037,
      "training_step_time": 7.464077472686768
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 3.1356887817382812,
      "step": 1037
    },
    {
      "dynamics_loss_avg": 0.0714671764522791,
      "epoch": 3.955841064453125e-06,
      "step": 1037
    },
    {
      "action_loss_avg": 0.054798648878932,
      "epoch": 3.955841064453125e-06,
      "step": 1037
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 1037,
      "training_step_time": 9.912009477615356
    },
    {
      "epoch": 3.955841064453125e-06,
      "model_forward_time": 2.436556100845337,
      "step": 1037
    },
    {
      "epoch": 3.955841064453125e-06,
      "step": 1037,
      "training_step_time": 10.01705813407898
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 3.158660888671875,
      "step": 1038
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 1038,
      "training_step_time": 9.934228658676147
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 3.136378526687622,
      "step": 1038
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 1038,
      "training_step_time": 9.909003257751465
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 2.719496726989746,
      "step": 1038
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 1038,
      "training_step_time": 7.748798370361328
    },
    {
      "epoch": 3.95965576171875e-06,
      "model_forward_time": 3.138840913772583,
      "step": 1038
    },
    {
      "epoch": 3.95965576171875e-06,
      "step": 1038,
      "training_step_time": 10.290661811828613
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 3.184753656387329,
      "step": 1039
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 1039,
      "training_step_time": 9.958420991897583
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 1.7461159229278564,
      "step": 1039
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 1039,
      "training_step_time": 5.208482265472412
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 3.1353073120117188,
      "step": 1039
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 1039,
      "training_step_time": 9.90239691734314
    },
    {
      "epoch": 3.963470458984375e-06,
      "model_forward_time": 2.7170701026916504,
      "step": 1039
    },
    {
      "epoch": 3.963470458984375e-06,
      "step": 1039,
      "training_step_time": 10.278578519821167
    },
    {
      "epoch": 3.96728515625e-06,
      "grad_norm": 0.8559488654136658,
      "learning_rate": 1e-05,
      "loss": 0.1356,
      "step": 1040
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 2.4450650215148926,
      "step": 1040
    },
    {
      "dynamics_loss_avg": 0.06956034265458584,
      "epoch": 3.96728515625e-06,
      "step": 1040
    },
    {
      "action_loss_avg": 0.05222976040095091,
      "epoch": 3.96728515625e-06,
      "step": 1040
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 1040,
      "training_step_time": 7.470202684402466
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 2.431755304336548,
      "step": 1040
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 1040,
      "training_step_time": 7.451286792755127
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 2.4300742149353027,
      "step": 1040
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 1040,
      "training_step_time": 7.450575590133667
    },
    {
      "epoch": 3.96728515625e-06,
      "model_forward_time": 3.1326053142547607,
      "step": 1040
    },
    {
      "epoch": 3.96728515625e-06,
      "step": 1040,
      "training_step_time": 10.30475926399231
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 3.4431633949279785,
      "step": 1041
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 1041,
      "training_step_time": 10.215195894241333
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 2.4305593967437744,
      "step": 1041
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 1041,
      "training_step_time": 7.449402570724487
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 3.1338565349578857,
      "step": 1041
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 1041,
      "training_step_time": 9.901040077209473
    },
    {
      "epoch": 3.971099853515625e-06,
      "model_forward_time": 2.4325215816497803,
      "step": 1041
    },
    {
      "epoch": 3.971099853515625e-06,
      "step": 1041,
      "training_step_time": 10.32597827911377
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 2.47495174407959,
      "step": 1042
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 1042,
      "training_step_time": 7.498223304748535
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 3.1300933361053467,
      "step": 1042
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 1042,
      "training_step_time": 9.90009593963623
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 3.4122724533081055,
      "step": 1042
    },
    {
      "dynamics_loss_avg": 0.07246612496674061,
      "epoch": 3.97491455078125e-06,
      "step": 1042
    },
    {
      "action_loss_avg": 0.050273002684116365,
      "epoch": 3.97491455078125e-06,
      "step": 1042
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 1042,
      "training_step_time": 10.182472944259644
    },
    {
      "epoch": 3.97491455078125e-06,
      "model_forward_time": 3.127387762069702,
      "step": 1042
    },
    {
      "epoch": 3.97491455078125e-06,
      "step": 1042,
      "training_step_time": 10.038492202758789
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 3.1443843841552734,
      "step": 1043
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 1043,
      "training_step_time": 9.916706562042236
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 2.426624059677124,
      "step": 1043
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 1043,
      "training_step_time": 7.447744131088257
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 2.425382137298584,
      "step": 1043
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 1043,
      "training_step_time": 7.4489030838012695
    },
    {
      "epoch": 3.978729248046875e-06,
      "model_forward_time": 3.387727737426758,
      "step": 1043
    },
    {
      "epoch": 3.978729248046875e-06,
      "step": 1043,
      "training_step_time": 10.296834945678711
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 3.1377882957458496,
      "step": 1044
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 1044,
      "training_step_time": 9.90206527709961
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 2.425844669342041,
      "step": 1044
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 1044,
      "training_step_time": 7.44635534286499
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 2.426443099975586,
      "step": 1044
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 1044,
      "training_step_time": 7.445705890655518
    },
    {
      "epoch": 3.9825439453125e-06,
      "model_forward_time": 2.425429582595825,
      "step": 1044
    },
    {
      "epoch": 3.9825439453125e-06,
      "step": 1044,
      "training_step_time": 10.306378602981567
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 3.4468603134155273,
      "step": 1045
    },
    {
      "dynamics_loss_avg": 0.08156602568924427,
      "epoch": 3.986358642578125e-06,
      "step": 1045
    },
    {
      "action_loss_avg": 0.07572003789246082,
      "epoch": 3.986358642578125e-06,
      "step": 1045
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 1045,
      "training_step_time": 10.220399618148804
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 2.4330356121063232,
      "step": 1045
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 1045,
      "training_step_time": 7.454963207244873
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 3.131072998046875,
      "step": 1045
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 1045,
      "training_step_time": 9.896494388580322
    },
    {
      "epoch": 3.986358642578125e-06,
      "model_forward_time": 3.1333730220794678,
      "step": 1045
    },
    {
      "epoch": 3.986358642578125e-06,
      "step": 1045,
      "training_step_time": 10.024403810501099
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 2.467320203781128,
      "step": 1046
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 1046,
      "training_step_time": 7.490971565246582
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 2.703885078430176,
      "step": 1046
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 1046,
      "training_step_time": 7.725358247756958
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 3.128643751144409,
      "step": 1046
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 1046,
      "training_step_time": 9.890462160110474
    },
    {
      "epoch": 3.99017333984375e-06,
      "model_forward_time": 3.1279308795928955,
      "step": 1046
    },
    {
      "epoch": 3.99017333984375e-06,
      "step": 1046,
      "training_step_time": 10.249348163604736
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 2.451143741607666,
      "step": 1047
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 1047,
      "training_step_time": 7.469691038131714
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 3.128257989883423,
      "step": 1047
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 1047,
      "training_step_time": 9.88733458518982
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 2.4295878410339355,
      "step": 1047
    },
    {
      "dynamics_loss_avg": 0.08870643675327301,
      "epoch": 3.993988037109375e-06,
      "step": 1047
    },
    {
      "action_loss_avg": 0.06118701081722975,
      "epoch": 3.993988037109375e-06,
      "step": 1047
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 1047,
      "training_step_time": 7.45251727104187
    },
    {
      "epoch": 3.993988037109375e-06,
      "model_forward_time": 3.395991802215576,
      "step": 1047
    },
    {
      "epoch": 3.993988037109375e-06,
      "step": 1047,
      "training_step_time": 10.308383226394653
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 2.473322868347168,
      "step": 1048
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 1048,
      "training_step_time": 7.495523452758789
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 3.127264976501465,
      "step": 1048
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 1048,
      "training_step_time": 9.886417627334595
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 3.127960205078125,
      "step": 1048
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 1048,
      "training_step_time": 9.881627798080444
    },
    {
      "epoch": 3.997802734375e-06,
      "model_forward_time": 2.428692102432251,
      "step": 1048
    },
    {
      "epoch": 3.997802734375e-06,
      "step": 1048,
      "training_step_time": 10.011136293411255
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 3.4522509574890137,
      "step": 1049
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 1049,
      "training_step_time": 10.213444709777832
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 3.1356000900268555,
      "step": 1049
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 1049,
      "training_step_time": 9.900628566741943
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 3.135392189025879,
      "step": 1049
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 1049,
      "training_step_time": 9.89704418182373
    },
    {
      "epoch": 4.001617431640625e-06,
      "model_forward_time": 2.433377265930176,
      "step": 1049
    },
    {
      "epoch": 4.001617431640625e-06,
      "step": 1049,
      "training_step_time": 10.25071382522583
    },
    {
      "epoch": 4.00543212890625e-06,
      "grad_norm": 0.5201653838157654,
      "learning_rate": 1e-05,
      "loss": 0.1357,
      "step": 1050
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 3.1537375450134277,
      "step": 1050
    },
    {
      "dynamics_loss_avg": 0.08783002234995366,
      "epoch": 4.00543212890625e-06,
      "step": 1050
    },
    {
      "action_loss_avg": 0.058450534380972384,
      "epoch": 4.00543212890625e-06,
      "step": 1050
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 1050,
      "training_step_time": 9.922749996185303
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 2.7094240188598633,
      "step": 1050
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 1050,
      "training_step_time": 7.731837749481201
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 2.431182622909546,
      "step": 1050
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 1050,
      "training_step_time": 7.451400279998779
    },
    {
      "epoch": 4.00543212890625e-06,
      "model_forward_time": 2.4273841381073,
      "step": 1050
    },
    {
      "epoch": 4.00543212890625e-06,
      "step": 1050,
      "training_step_time": 10.008321762084961
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 3.1453075408935547,
      "step": 1051
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 1051,
      "training_step_time": 9.910186529159546
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 2.427644729614258,
      "step": 1051
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 1051,
      "training_step_time": 7.44633150100708
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 2.426581859588623,
      "step": 1051
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 1051,
      "training_step_time": 7.74019455909729
    },
    {
      "epoch": 4.009246826171875e-06,
      "model_forward_time": 3.1312713623046875,
      "step": 1051
    },
    {
      "epoch": 4.009246826171875e-06,
      "step": 1051,
      "training_step_time": 10.257619142532349
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 2.453831911087036,
      "step": 1052
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 1052,
      "training_step_time": 7.4789557456970215
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 3.1316211223602295,
      "step": 1052
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 1052,
      "training_step_time": 9.897550106048584
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 2.4328694343566895,
      "step": 1052
    },
    {
      "dynamics_loss_avg": 0.07610912285745144,
      "epoch": 4.0130615234375e-06,
      "step": 1052
    },
    {
      "action_loss_avg": 0.07511449884623289,
      "epoch": 4.0130615234375e-06,
      "step": 1052
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 1052,
      "training_step_time": 7.460249900817871
    },
    {
      "epoch": 4.0130615234375e-06,
      "model_forward_time": 2.4305648803710938,
      "step": 1052
    },
    {
      "epoch": 4.0130615234375e-06,
      "step": 1052,
      "training_step_time": 10.292908668518066
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 3.457275390625,
      "step": 1053
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 1053,
      "training_step_time": 10.225892305374146
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 2.4289283752441406,
      "step": 1053
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 1053,
      "training_step_time": 7.453770637512207
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 2.4268805980682373,
      "step": 1053
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 1053,
      "training_step_time": 7.44849967956543
    },
    {
      "epoch": 4.016876220703125e-06,
      "model_forward_time": 2.4283478260040283,
      "step": 1053
    },
    {
      "epoch": 4.016876220703125e-06,
      "step": 1053,
      "training_step_time": 10.007872819900513
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 3.1643528938293457,
      "step": 1054
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 1054,
      "training_step_time": 9.926125288009644
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 2.712841033935547,
      "step": 1054
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 1054,
      "training_step_time": 7.732632875442505
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 2.426811695098877,
      "step": 1054
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 1054,
      "training_step_time": 7.444976568222046
    },
    {
      "epoch": 4.02069091796875e-06,
      "model_forward_time": 3.1246426105499268,
      "step": 1054
    },
    {
      "epoch": 4.02069091796875e-06,
      "step": 1054,
      "training_step_time": 10.278019666671753
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 1.7839338779449463,
      "step": 1055
    },
    {
      "dynamics_loss_avg": 0.0715997938066721,
      "epoch": 4.024505615234375e-06,
      "step": 1055
    },
    {
      "action_loss_avg": 0.0726899086497724,
      "epoch": 4.024505615234375e-06,
      "step": 1055
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 1055,
      "training_step_time": 5.234756946563721
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 2.4347901344299316,
      "step": 1055
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 1055,
      "training_step_time": 7.449705362319946
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 3.133767604827881,
      "step": 1055
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 1055,
      "training_step_time": 9.889688730239868
    },
    {
      "epoch": 4.024505615234375e-06,
      "model_forward_time": 3.4464480876922607,
      "step": 1055
    },
    {
      "epoch": 4.024505615234375e-06,
      "step": 1055,
      "training_step_time": 10.301167488098145
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 2.446652889251709,
      "step": 1056
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 1056,
      "training_step_time": 7.46366024017334
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 3.1311192512512207,
      "step": 1056
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 1056,
      "training_step_time": 9.895205974578857
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 3.137722969055176,
      "step": 1056
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 1056,
      "training_step_time": 9.904387950897217
    },
    {
      "epoch": 4.0283203125e-06,
      "model_forward_time": 2.4346699714660645,
      "step": 1056
    },
    {
      "epoch": 4.0283203125e-06,
      "step": 1056,
      "training_step_time": 10.25359845161438
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 2.7519140243530273,
      "step": 1057
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 1057,
      "training_step_time": 7.776634216308594
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 3.13433575630188,
      "step": 1057
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 1057,
      "training_step_time": 9.898778200149536
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 2.4369566440582275,
      "step": 1057
    },
    {
      "dynamics_loss_avg": 0.06777074746787548,
      "epoch": 4.032135009765625e-06,
      "step": 1057
    },
    {
      "action_loss_avg": 0.06300116684287786,
      "epoch": 4.032135009765625e-06,
      "step": 1057
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 1057,
      "training_step_time": 7.466840028762817
    },
    {
      "epoch": 4.032135009765625e-06,
      "model_forward_time": 2.436509847640991,
      "step": 1057
    },
    {
      "epoch": 4.032135009765625e-06,
      "step": 1057,
      "training_step_time": 10.247875690460205
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 3.175236225128174,
      "step": 1058
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 1058,
      "training_step_time": 9.941710948944092
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 3.4265332221984863,
      "step": 1058
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 1058,
      "training_step_time": 10.184952974319458
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 3.1410813331604004,
      "step": 1058
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 1058,
      "training_step_time": 9.903676986694336
    },
    {
      "epoch": 4.03594970703125e-06,
      "model_forward_time": 2.434890031814575,
      "step": 1058
    },
    {
      "epoch": 4.03594970703125e-06,
      "step": 1058,
      "training_step_time": 10.306732654571533
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 3.167473077774048,
      "step": 1059
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 1059,
      "training_step_time": 9.930205583572388
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 3.13720703125,
      "step": 1059
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 1059,
      "training_step_time": 9.901117086410522
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 3.4351694583892822,
      "step": 1059
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 1059,
      "training_step_time": 10.197368383407593
    },
    {
      "epoch": 4.039764404296875e-06,
      "model_forward_time": 2.4327757358551025,
      "step": 1059
    },
    {
      "epoch": 4.039764404296875e-06,
      "step": 1059,
      "training_step_time": 10.0165433883667
    },
    {
      "epoch": 4.0435791015625e-06,
      "grad_norm": 1.0380113124847412,
      "learning_rate": 1e-05,
      "loss": 0.1336,
      "step": 1060
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 3.1644937992095947,
      "step": 1060
    },
    {
      "dynamics_loss_avg": 0.0719294898211956,
      "epoch": 4.0435791015625e-06,
      "step": 1060
    },
    {
      "action_loss_avg": 0.07270087059587241,
      "epoch": 4.0435791015625e-06,
      "step": 1060
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 1060,
      "training_step_time": 9.936248540878296
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 3.1324331760406494,
      "step": 1060
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 1060,
      "training_step_time": 9.904985427856445
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 3.1314897537231445,
      "step": 1060
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 1060,
      "training_step_time": 9.902443408966064
    },
    {
      "epoch": 4.0435791015625e-06,
      "model_forward_time": 3.3956496715545654,
      "step": 1060
    },
    {
      "epoch": 4.0435791015625e-06,
      "step": 1060,
      "training_step_time": 10.31035590171814
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 2.447692394256592,
      "step": 1061
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 1061,
      "training_step_time": 7.479140520095825
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 2.4310696125030518,
      "step": 1061
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 1061,
      "training_step_time": 7.461735486984253
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 2.4290452003479004,
      "step": 1061
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 1061,
      "training_step_time": 7.458497762680054
    },
    {
      "epoch": 4.047393798828125e-06,
      "model_forward_time": 3.1328229904174805,
      "step": 1061
    },
    {
      "epoch": 4.047393798828125e-06,
      "step": 1061,
      "training_step_time": 10.011267900466919
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 2.452019453048706,
      "step": 1062
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 1062,
      "training_step_time": 7.478132486343384
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 2.7359819412231445,
      "step": 1062
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 1062,
      "training_step_time": 7.7595295906066895
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 1.7345516681671143,
      "step": 1062
    },
    {
      "dynamics_loss_avg": 0.08797182515263557,
      "epoch": 4.05120849609375e-06,
      "step": 1062
    },
    {
      "action_loss_avg": 0.0746942438185215,
      "epoch": 4.05120849609375e-06,
      "step": 1062
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 1062,
      "training_step_time": 5.188105344772339
    },
    {
      "epoch": 4.05120849609375e-06,
      "model_forward_time": 2.4343812465667725,
      "step": 1062
    },
    {
      "epoch": 4.05120849609375e-06,
      "step": 1062,
      "training_step_time": 10.27709698677063
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 3.1810383796691895,
      "step": 1063
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 1063,
      "training_step_time": 9.953843116760254
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 3.140406847000122,
      "step": 1063
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 1063,
      "training_step_time": 9.91366982460022
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 3.1322364807128906,
      "step": 1063
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 1063,
      "training_step_time": 9.902786254882812
    },
    {
      "epoch": 4.055023193359375e-06,
      "model_forward_time": 3.4123666286468506,
      "step": 1063
    },
    {
      "epoch": 4.055023193359375e-06,
      "step": 1063,
      "training_step_time": 10.3167245388031
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 3.173215866088867,
      "step": 1064
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 1064,
      "training_step_time": 9.936984777450562
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 2.43591570854187,
      "step": 1064
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 1064,
      "training_step_time": 7.455538272857666
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 2.432633876800537,
      "step": 1064
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 1064,
      "training_step_time": 7.451486110687256
    },
    {
      "epoch": 4.058837890625e-06,
      "model_forward_time": 3.1371889114379883,
      "step": 1064
    },
    {
      "epoch": 4.058837890625e-06,
      "step": 1064,
      "training_step_time": 10.01689863204956
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 2.0341033935546875,
      "step": 1065
    },
    {
      "dynamics_loss_avg": 0.08099641352891922,
      "epoch": 4.062652587890625e-06,
      "step": 1065
    },
    {
      "action_loss_avg": 0.05941188242286444,
      "epoch": 4.062652587890625e-06,
      "step": 1065
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 1065,
      "training_step_time": 5.497762441635132
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 2.426027774810791,
      "step": 1065
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 1065,
      "training_step_time": 7.445672988891602
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 1.7391502857208252,
      "step": 1065
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 1065,
      "training_step_time": 5.19108247756958
    },
    {
      "epoch": 4.062652587890625e-06,
      "model_forward_time": 2.4242591857910156,
      "step": 1065
    },
    {
      "epoch": 4.062652587890625e-06,
      "step": 1065,
      "training_step_time": 10.325473070144653
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 3.1479580402374268,
      "step": 1066
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 1066,
      "training_step_time": 9.916859865188599
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 3.13024640083313,
      "step": 1066
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 1066,
      "training_step_time": 9.889561414718628
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 2.7263691425323486,
      "step": 1066
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 1066,
      "training_step_time": 7.743382215499878
    },
    {
      "epoch": 4.06646728515625e-06,
      "model_forward_time": 3.121999740600586,
      "step": 1066
    },
    {
      "epoch": 4.06646728515625e-06,
      "step": 1066,
      "training_step_time": 10.003620862960815
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 3.140368938446045,
      "step": 1067
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 1067,
      "training_step_time": 9.914249897003174
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 2.4277546405792236,
      "step": 1067
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 1067,
      "training_step_time": 7.450738191604614
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 2.4268834590911865,
      "step": 1067
    },
    {
      "dynamics_loss_avg": 0.07498536147177219,
      "epoch": 4.070281982421875e-06,
      "step": 1067
    },
    {
      "action_loss_avg": 0.05564967067912221,
      "epoch": 4.070281982421875e-06,
      "step": 1067
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 1067,
      "training_step_time": 7.4470391273498535
    },
    {
      "epoch": 4.070281982421875e-06,
      "model_forward_time": 3.3823060989379883,
      "step": 1067
    },
    {
      "epoch": 4.070281982421875e-06,
      "step": 1067,
      "training_step_time": 10.234405994415283
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 3.1433377265930176,
      "step": 1068
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 1068,
      "training_step_time": 9.902301549911499
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 3.1304876804351807,
      "step": 1068
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 1068,
      "training_step_time": 9.892100095748901
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 2.4305455684661865,
      "step": 1068
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 1068,
      "training_step_time": 7.452002286911011
    },
    {
      "epoch": 4.0740966796875e-06,
      "model_forward_time": 1.7359960079193115,
      "step": 1068
    },
    {
      "epoch": 4.0740966796875e-06,
      "step": 1068,
      "training_step_time": 10.294670104980469
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 3.4071905612945557,
      "step": 1069
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 1069,
      "training_step_time": 10.171189308166504
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 1.7389957904815674,
      "step": 1069
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 1069,
      "training_step_time": 5.18850302696228
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 1.7388503551483154,
      "step": 1069
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 1069,
      "training_step_time": 5.1864824295043945
    },
    {
      "epoch": 4.077911376953125e-06,
      "model_forward_time": 3.1327714920043945,
      "step": 1069
    },
    {
      "epoch": 4.077911376953125e-06,
      "step": 1069,
      "training_step_time": 10.34765362739563
    },
    {
      "epoch": 4.08172607421875e-06,
      "grad_norm": 0.5101640224456787,
      "learning_rate": 1e-05,
      "loss": 0.1356,
      "step": 1070
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 2.4616448879241943,
      "step": 1070
    },
    {
      "dynamics_loss_avg": 0.05531905274838209,
      "epoch": 4.08172607421875e-06,
      "step": 1070
    },
    {
      "action_loss_avg": 0.050034942291677,
      "epoch": 4.08172607421875e-06,
      "step": 1070
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 1070,
      "training_step_time": 7.491514444351196
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 2.425598621368408,
      "step": 1070
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 1070,
      "training_step_time": 7.452692031860352
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 2.706503391265869,
      "step": 1070
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 1070,
      "training_step_time": 7.7318949699401855
    },
    {
      "epoch": 4.08172607421875e-06,
      "model_forward_time": 3.1309759616851807,
      "step": 1070
    },
    {
      "epoch": 4.08172607421875e-06,
      "step": 1070,
      "training_step_time": 10.254924774169922
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 2.4440226554870605,
      "step": 1071
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 1071,
      "training_step_time": 7.467412710189819
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 2.4235990047454834,
      "step": 1071
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 1071,
      "training_step_time": 7.4417266845703125
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 2.4235899448394775,
      "step": 1071
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 1071,
      "training_step_time": 7.442095994949341
    },
    {
      "epoch": 4.085540771484375e-06,
      "model_forward_time": 3.1274685859680176,
      "step": 1071
    },
    {
      "epoch": 4.085540771484375e-06,
      "step": 1071,
      "training_step_time": 10.236205339431763
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 3.496182441711426,
      "step": 1072
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 1072,
      "training_step_time": 10.25449514389038
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 2.4311487674713135,
      "step": 1072
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 1072,
      "training_step_time": 7.442864656448364
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 3.1300182342529297,
      "step": 1072
    },
    {
      "dynamics_loss_avg": 0.06792482007294894,
      "epoch": 4.08935546875e-06,
      "step": 1072
    },
    {
      "action_loss_avg": 0.0822702769190073,
      "epoch": 4.08935546875e-06,
      "step": 1072
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 1072,
      "training_step_time": 9.888622522354126
    },
    {
      "epoch": 4.08935546875e-06,
      "model_forward_time": 3.128324031829834,
      "step": 1072
    },
    {
      "epoch": 4.08935546875e-06,
      "step": 1072,
      "training_step_time": 10.01066541671753
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 3.1699435710906982,
      "step": 1073
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 1073,
      "training_step_time": 9.932973861694336
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 2.6660959720611572,
      "step": 1073
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 1073,
      "training_step_time": 7.683768033981323
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 3.1206533908843994,
      "step": 1073
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 1073,
      "training_step_time": 9.875471591949463
    },
    {
      "epoch": 4.093170166015625e-06,
      "model_forward_time": 3.120656967163086,
      "step": 1073
    },
    {
      "epoch": 4.093170166015625e-06,
      "step": 1073,
      "training_step_time": 10.258722305297852
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 3.1395015716552734,
      "step": 1074
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 1074,
      "training_step_time": 9.90074872970581
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 2.4237735271453857,
      "step": 1074
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 1074,
      "training_step_time": 7.448798894882202
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 3.4378421306610107,
      "step": 1074
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 1074,
      "training_step_time": 10.200051069259644
    },
    {
      "epoch": 4.09698486328125e-06,
      "model_forward_time": 2.4267215728759766,
      "step": 1074
    },
    {
      "epoch": 4.09698486328125e-06,
      "step": 1074,
      "training_step_time": 10.311431169509888
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 1.7516844272613525,
      "step": 1075
    },
    {
      "dynamics_loss_avg": 0.05694104433059692,
      "epoch": 4.100799560546875e-06,
      "step": 1075
    },
    {
      "action_loss_avg": 0.05466614020988345,
      "epoch": 4.100799560546875e-06,
      "step": 1075
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 1075,
      "training_step_time": 5.215050220489502
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 2.423743724822998,
      "step": 1075
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 1075,
      "training_step_time": 7.44045090675354
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 2.427680492401123,
      "step": 1075
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 1075,
      "training_step_time": 7.445486068725586
    },
    {
      "epoch": 4.100799560546875e-06,
      "model_forward_time": 2.4213342666625977,
      "step": 1075
    },
    {
      "epoch": 4.100799560546875e-06,
      "step": 1075,
      "training_step_time": 10.312364339828491
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 2.465959072113037,
      "step": 1076
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 1076,
      "training_step_time": 7.485889434814453
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 3.126352548599243,
      "step": 1076
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 1076,
      "training_step_time": 9.887604475021362
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 3.1238903999328613,
      "step": 1076
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 1076,
      "training_step_time": 9.886502981185913
    },
    {
      "epoch": 4.1046142578125e-06,
      "model_forward_time": 3.1269571781158447,
      "step": 1076
    },
    {
      "epoch": 4.1046142578125e-06,
      "step": 1076,
      "training_step_time": 10.319494724273682
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 3.1350479125976562,
      "step": 1077
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 1077,
      "training_step_time": 9.896318435668945
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 3.403289794921875,
      "step": 1077
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 1077,
      "training_step_time": 10.165825843811035
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 2.4256787300109863,
      "step": 1077
    },
    {
      "dynamics_loss_avg": 0.07972723431885242,
      "epoch": 4.108428955078125e-06,
      "step": 1077
    },
    {
      "action_loss_avg": 0.063567697070539,
      "epoch": 4.108428955078125e-06,
      "step": 1077
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 1077,
      "training_step_time": 7.451759099960327
    },
    {
      "epoch": 4.108428955078125e-06,
      "model_forward_time": 2.4304380416870117,
      "step": 1077
    },
    {
      "epoch": 4.108428955078125e-06,
      "step": 1077,
      "training_step_time": 10.027688980102539
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 3.164334297180176,
      "step": 1078
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 1078,
      "training_step_time": 9.929434299468994
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 3.1276493072509766,
      "step": 1078
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 1078,
      "training_step_time": 9.895098209381104
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 3.4134039878845215,
      "step": 1078
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 1078,
      "training_step_time": 10.176491498947144
    },
    {
      "epoch": 4.11224365234375e-06,
      "model_forward_time": 3.133524179458618,
      "step": 1078
    },
    {
      "epoch": 4.11224365234375e-06,
      "step": 1078,
      "training_step_time": 10.015142917633057
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 2.468141794204712,
      "step": 1079
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 1079,
      "training_step_time": 7.492715358734131
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 3.1247715950012207,
      "step": 1079
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 1079,
      "training_step_time": 9.889188051223755
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 2.4283804893493652,
      "step": 1079
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 1079,
      "training_step_time": 7.467933177947998
    },
    {
      "epoch": 4.116058349609375e-06,
      "model_forward_time": 3.3939085006713867,
      "step": 1079
    },
    {
      "epoch": 4.116058349609375e-06,
      "step": 1079,
      "training_step_time": 10.287365436553955
    },
    {
      "epoch": 4.119873046875e-06,
      "grad_norm": 0.8205209970474243,
      "learning_rate": 1e-05,
      "loss": 0.1371,
      "step": 1080
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 2.4476633071899414,
      "step": 1080
    },
    {
      "dynamics_loss_avg": 0.07408623397350311,
      "epoch": 4.119873046875e-06,
      "step": 1080
    },
    {
      "action_loss_avg": 0.05569957531988621,
      "epoch": 4.119873046875e-06,
      "step": 1080
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 1080,
      "training_step_time": 7.480700969696045
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 3.1325418949127197,
      "step": 1080
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 1080,
      "training_step_time": 9.901726007461548
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 3.13568115234375,
      "step": 1080
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 1080,
      "training_step_time": 9.91004204750061
    },
    {
      "epoch": 4.119873046875e-06,
      "model_forward_time": 2.436284065246582,
      "step": 1080
    },
    {
      "epoch": 4.119873046875e-06,
      "step": 1080,
      "training_step_time": 10.263877391815186
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 2.7262918949127197,
      "step": 1081
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 1081,
      "training_step_time": 7.757092714309692
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 2.433814525604248,
      "step": 1081
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 1081,
      "training_step_time": 7.46400785446167
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 3.1348729133605957,
      "step": 1081
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 1081,
      "training_step_time": 9.908694982528687
    },
    {
      "epoch": 4.123687744140625e-06,
      "model_forward_time": 2.4407479763031006,
      "step": 1081
    },
    {
      "epoch": 4.123687744140625e-06,
      "step": 1081,
      "training_step_time": 10.330143690109253
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 2.4754748344421387,
      "step": 1082
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 1082,
      "training_step_time": 7.498425483703613
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 2.430947780609131,
      "step": 1082
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 1082,
      "training_step_time": 7.760025978088379
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 2.4367380142211914,
      "step": 1082
    },
    {
      "dynamics_loss_avg": 0.07310364153236151,
      "epoch": 4.12750244140625e-06,
      "step": 1082
    },
    {
      "action_loss_avg": 0.07392867859452963,
      "epoch": 4.12750244140625e-06,
      "step": 1082
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 1082,
      "training_step_time": 7.461177110671997
    },
    {
      "epoch": 4.12750244140625e-06,
      "model_forward_time": 3.1223361492156982,
      "step": 1082
    },
    {
      "epoch": 4.12750244140625e-06,
      "step": 1082,
      "training_step_time": 10.029411315917969
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 3.1472294330596924,
      "step": 1083
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 1083,
      "training_step_time": 9.911690950393677
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 2.4253034591674805,
      "step": 1083
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 1083,
      "training_step_time": 7.447087049484253
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 2.4241549968719482,
      "step": 1083
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 1083,
      "training_step_time": 7.442203998565674
    },
    {
      "epoch": 4.131317138671875e-06,
      "model_forward_time": 2.7112412452697754,
      "step": 1083
    },
    {
      "epoch": 4.131317138671875e-06,
      "step": 1083,
      "training_step_time": 10.02193546295166
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 3.173725128173828,
      "step": 1084
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 1084,
      "training_step_time": 9.941496849060059
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 2.4240622520446777,
      "step": 1084
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 1084,
      "training_step_time": 7.444612264633179
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 2.427727222442627,
      "step": 1084
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 1084,
      "training_step_time": 7.448839902877808
    },
    {
      "epoch": 4.1351318359375e-06,
      "model_forward_time": 3.130286693572998,
      "step": 1084
    },
    {
      "epoch": 4.1351318359375e-06,
      "step": 1084,
      "training_step_time": 10.296591758728027
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 2.7423157691955566,
      "step": 1085
    },
    {
      "dynamics_loss_avg": 0.08963280338793993,
      "epoch": 4.138946533203125e-06,
      "step": 1085
    },
    {
      "action_loss_avg": 0.08802275406196713,
      "epoch": 4.138946533203125e-06,
      "step": 1085
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 1085,
      "training_step_time": 7.771087884902954
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 3.1295359134674072,
      "step": 1085
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 1085,
      "training_step_time": 9.89871335029602
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 3.1285204887390137,
      "step": 1085
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 1085,
      "training_step_time": 9.904043674468994
    },
    {
      "epoch": 4.138946533203125e-06,
      "model_forward_time": 3.135374069213867,
      "step": 1085
    },
    {
      "epoch": 4.138946533203125e-06,
      "step": 1085,
      "training_step_time": 10.282343626022339
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 2.4717350006103516,
      "step": 1086
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 1086,
      "training_step_time": 7.504338502883911
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 3.4053192138671875,
      "step": 1086
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 1086,
      "training_step_time": 10.175765991210938
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 2.4302968978881836,
      "step": 1086
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 1086,
      "training_step_time": 7.45698881149292
    },
    {
      "epoch": 4.14276123046875e-06,
      "model_forward_time": 3.1300363540649414,
      "step": 1086
    },
    {
      "epoch": 4.14276123046875e-06,
      "step": 1086,
      "training_step_time": 10.314438104629517
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 2.4734601974487305,
      "step": 1087
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 1087,
      "training_step_time": 7.501787185668945
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 2.4279770851135254,
      "step": 1087
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 1087,
      "training_step_time": 7.455254554748535
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 2.4263253211975098,
      "step": 1087
    },
    {
      "dynamics_loss_avg": 0.0769407443702221,
      "epoch": 4.146575927734375e-06,
      "step": 1087
    },
    {
      "action_loss_avg": 0.06356347613036632,
      "epoch": 4.146575927734375e-06,
      "step": 1087
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 1087,
      "training_step_time": 7.449837684631348
    },
    {
      "epoch": 4.146575927734375e-06,
      "model_forward_time": 3.434333086013794,
      "step": 1087
    },
    {
      "epoch": 4.146575927734375e-06,
      "step": 1087,
      "training_step_time": 10.327358484268188
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 3.1496987342834473,
      "step": 1088
    },
    {
      "epoch": 4.150390625e-06,
      "step": 1088,
      "training_step_time": 9.921508312225342
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 2.428788423538208,
      "step": 1088
    },
    {
      "epoch": 4.150390625e-06,
      "step": 1088,
      "training_step_time": 7.457886695861816
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 2.4299588203430176,
      "step": 1088
    },
    {
      "epoch": 4.150390625e-06,
      "step": 1088,
      "training_step_time": 7.456619739532471
    },
    {
      "epoch": 4.150390625e-06,
      "model_forward_time": 3.128359079360962,
      "step": 1088
    },
    {
      "epoch": 4.150390625e-06,
      "step": 1088,
      "training_step_time": 10.027841091156006
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 3.4741930961608887,
      "step": 1089
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 1089,
      "training_step_time": 10.242451429367065
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 3.135629177093506,
      "step": 1089
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 1089,
      "training_step_time": 9.903650283813477
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 3.1377928256988525,
      "step": 1089
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 1089,
      "training_step_time": 9.908899784088135
    },
    {
      "epoch": 4.154205322265625e-06,
      "model_forward_time": 2.4471895694732666,
      "step": 1089
    },
    {
      "epoch": 4.154205322265625e-06,
      "step": 1089,
      "training_step_time": 10.018536567687988
    },
    {
      "epoch": 4.15802001953125e-06,
      "grad_norm": 0.8912187218666077,
      "learning_rate": 1e-05,
      "loss": 0.1299,
      "step": 1090
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 3.1906726360321045,
      "step": 1090
    },
    {
      "dynamics_loss_avg": 0.07675366178154945,
      "epoch": 4.15802001953125e-06,
      "step": 1090
    },
    {
      "action_loss_avg": 0.0774526685476303,
      "epoch": 4.15802001953125e-06,
      "step": 1090
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 1090,
      "training_step_time": 9.96228313446045
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 2.71272873878479,
      "step": 1090
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 1090,
      "training_step_time": 7.736382961273193
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 3.134159564971924,
      "step": 1090
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 1090,
      "training_step_time": 9.902214288711548
    },
    {
      "epoch": 4.15802001953125e-06,
      "model_forward_time": 3.1364643573760986,
      "step": 1090
    },
    {
      "epoch": 4.15802001953125e-06,
      "step": 1090,
      "training_step_time": 10.279194593429565
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 1.7804436683654785,
      "step": 1091
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 1091,
      "training_step_time": 5.238140821456909
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 1.7413601875305176,
      "step": 1091
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 1091,
      "training_step_time": 5.190717697143555
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 3.1302685737609863,
      "step": 1091
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 1091,
      "training_step_time": 9.891417980194092
    },
    {
      "epoch": 4.161834716796875e-06,
      "model_forward_time": 3.417616844177246,
      "step": 1091
    },
    {
      "epoch": 4.161834716796875e-06,
      "step": 1091,
      "training_step_time": 10.286756753921509
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 2.4610793590545654,
      "step": 1092
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 1092,
      "training_step_time": 7.480695962905884
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 1.7386093139648438,
      "step": 1092
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 1092,
      "training_step_time": 5.189683675765991
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 3.1322121620178223,
      "step": 1092
    },
    {
      "dynamics_loss_avg": 0.07456732913851738,
      "epoch": 4.1656494140625e-06,
      "step": 1092
    },
    {
      "action_loss_avg": 0.06456184163689613,
      "epoch": 4.1656494140625e-06,
      "step": 1092
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 1092,
      "training_step_time": 9.895052433013916
    },
    {
      "epoch": 4.1656494140625e-06,
      "model_forward_time": 3.137619733810425,
      "step": 1092
    },
    {
      "epoch": 4.1656494140625e-06,
      "step": 1092,
      "training_step_time": 10.27027440071106
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 3.4467966556549072,
      "step": 1093
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 1093,
      "training_step_time": 10.217285633087158
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 1.748185157775879,
      "step": 1093
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 1093,
      "training_step_time": 5.20375657081604
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 2.4336109161376953,
      "step": 1093
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 1093,
      "training_step_time": 7.458229064941406
    },
    {
      "epoch": 4.169464111328125e-06,
      "model_forward_time": 1.7468349933624268,
      "step": 1093
    },
    {
      "epoch": 4.169464111328125e-06,
      "step": 1093,
      "training_step_time": 10.023762702941895
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 1.7838716506958008,
      "step": 1094
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 1094,
      "training_step_time": 5.236760139465332
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 3.130319356918335,
      "step": 1094
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 1094,
      "training_step_time": 9.89760684967041
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 2.7018892765045166,
      "step": 1094
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 1094,
      "training_step_time": 7.727236270904541
    },
    {
      "epoch": 4.17327880859375e-06,
      "model_forward_time": 3.1322021484375,
      "step": 1094
    },
    {
      "epoch": 4.17327880859375e-06,
      "step": 1094,
      "training_step_time": 10.022449016571045
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 1.7626948356628418,
      "step": 1095
    },
    {
      "dynamics_loss_avg": 0.08282646611332893,
      "epoch": 4.177093505859375e-06,
      "step": 1095
    },
    {
      "action_loss_avg": 0.0790037114173174,
      "epoch": 4.177093505859375e-06,
      "step": 1095
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 1095,
      "training_step_time": 5.236086130142212
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 2.4304614067077637,
      "step": 1095
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 1095,
      "training_step_time": 7.453117370605469
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 2.4300200939178467,
      "step": 1095
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 1095,
      "training_step_time": 7.456161022186279
    },
    {
      "epoch": 4.177093505859375e-06,
      "model_forward_time": 2.6790783405303955,
      "step": 1095
    },
    {
      "epoch": 4.177093505859375e-06,
      "step": 1095,
      "training_step_time": 10.294177532196045
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 3.1603477001190186,
      "step": 1096
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 1096,
      "training_step_time": 9.920595169067383
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 3.132528066635132,
      "step": 1096
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 1096,
      "training_step_time": 9.893319606781006
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 3.134913206100464,
      "step": 1096
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 1096,
      "training_step_time": 9.896336317062378
    },
    {
      "epoch": 4.180908203125e-06,
      "model_forward_time": 3.131589412689209,
      "step": 1096
    },
    {
      "epoch": 4.180908203125e-06,
      "step": 1096,
      "training_step_time": 10.273651123046875
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 2.447652816772461,
      "step": 1097
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 1097,
      "training_step_time": 7.738437652587891
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 2.42988657951355,
      "step": 1097
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 1097,
      "training_step_time": 7.445235967636108
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 1.7424423694610596,
      "step": 1097
    },
    {
      "dynamics_loss_avg": 0.07729951497167349,
      "epoch": 4.184722900390625e-06,
      "step": 1097
    },
    {
      "action_loss_avg": 0.06733410619199276,
      "epoch": 4.184722900390625e-06,
      "step": 1097
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 1097,
      "training_step_time": 5.196367025375366
    },
    {
      "epoch": 4.184722900390625e-06,
      "model_forward_time": 1.7443230152130127,
      "step": 1097
    },
    {
      "epoch": 4.184722900390625e-06,
      "step": 1097,
      "training_step_time": 10.294795989990234
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 3.1740028858184814,
      "step": 1098
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 1098,
      "training_step_time": 9.933547019958496
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 3.1360795497894287,
      "step": 1098
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 1098,
      "training_step_time": 9.89803409576416
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 3.4406354427337646,
      "step": 1098
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 1098,
      "training_step_time": 10.198383092880249
    },
    {
      "epoch": 4.18853759765625e-06,
      "model_forward_time": 2.431260108947754,
      "step": 1098
    },
    {
      "epoch": 4.18853759765625e-06,
      "step": 1098,
      "training_step_time": 10.013729333877563
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 3.1517279148101807,
      "step": 1099
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 1099,
      "training_step_time": 9.911144733428955
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 2.4313085079193115,
      "step": 1099
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 1099,
      "training_step_time": 7.450984477996826
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 2.427501916885376,
      "step": 1099
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 1099,
      "training_step_time": 7.444692134857178
    },
    {
      "epoch": 4.192352294921875e-06,
      "model_forward_time": 2.431044101715088,
      "step": 1099
    },
    {
      "epoch": 4.192352294921875e-06,
      "step": 1099,
      "training_step_time": 10.012908697128296
    },
    {
      "epoch": 4.1961669921875e-06,
      "grad_norm": 0.5589960813522339,
      "learning_rate": 1e-05,
      "loss": 0.1324,
      "step": 1100
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 2.7453994750976562,
      "step": 1100
    },
    {
      "dynamics_loss_avg": 0.08161212429404259,
      "epoch": 4.1961669921875e-06,
      "step": 1100
    },
    {
      "action_loss_avg": 0.06802606992423535,
      "epoch": 4.1961669921875e-06,
      "step": 1100
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 1100,
      "training_step_time": 7.7617785930633545
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 2.418074369430542,
      "step": 1100
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 1100,
      "training_step_time": 7.429162979125977
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 2.4194557666778564,
      "step": 1100
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 1100,
      "training_step_time": 7.430651426315308
    },
    {
      "epoch": 4.1961669921875e-06,
      "model_forward_time": 3.1194865703582764,
      "step": 1100
    },
    {
      "epoch": 4.1961669921875e-06,
      "step": 1100,
      "training_step_time": 9.99834394454956
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 2.4630680084228516,
      "step": 1101
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 1101,
      "training_step_time": 7.47776198387146
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 3.3990235328674316,
      "step": 1101
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 1101,
      "training_step_time": 10.146071672439575
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 3.1284332275390625,
      "step": 1101
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 1101,
      "training_step_time": 9.878838777542114
    },
    {
      "epoch": 4.199981689453125e-06,
      "model_forward_time": 3.1357967853546143,
      "step": 1101
    },
    {
      "epoch": 4.199981689453125e-06,
      "step": 1101,
      "training_step_time": 10.254822015762329
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 2.443824529647827,
      "step": 1102
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 1102,
      "training_step_time": 7.462166786193848
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 2.4326534271240234,
      "step": 1102
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 1102,
      "training_step_time": 7.448476552963257
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 2.7028579711914062,
      "step": 1102
    },
    {
      "dynamics_loss_avg": 0.0632999811321497,
      "epoch": 4.20379638671875e-06,
      "step": 1102
    },
    {
      "action_loss_avg": 0.057333250902593134,
      "epoch": 4.20379638671875e-06,
      "step": 1102
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 1102,
      "training_step_time": 7.726554870605469
    },
    {
      "epoch": 4.20379638671875e-06,
      "model_forward_time": 2.4279003143310547,
      "step": 1102
    },
    {
      "epoch": 4.20379638671875e-06,
      "step": 1102,
      "training_step_time": 10.265063047409058
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 3.1810975074768066,
      "step": 1103
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 1103,
      "training_step_time": 9.944929838180542
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 3.134246349334717,
      "step": 1103
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 1103,
      "training_step_time": 9.898784875869751
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 3.1353840827941895,
      "step": 1103
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 1103,
      "training_step_time": 9.896896123886108
    },
    {
      "epoch": 4.207611083984375e-06,
      "model_forward_time": 2.4316861629486084,
      "step": 1103
    },
    {
      "epoch": 4.207611083984375e-06,
      "step": 1103,
      "training_step_time": 10.018436670303345
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 3.176727294921875,
      "step": 1104
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 1104,
      "training_step_time": 9.934195518493652
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 2.4257802963256836,
      "step": 1104
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 1104,
      "training_step_time": 7.447085857391357
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 2.429837226867676,
      "step": 1104
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 1104,
      "training_step_time": 7.446280002593994
    },
    {
      "epoch": 4.21142578125e-06,
      "model_forward_time": 3.1253716945648193,
      "step": 1104
    },
    {
      "epoch": 4.21142578125e-06,
      "step": 1104,
      "training_step_time": 10.023059606552124
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 2.4756839275360107,
      "step": 1105
    },
    {
      "dynamics_loss_avg": 0.07450716495513916,
      "epoch": 4.215240478515625e-06,
      "step": 1105
    },
    {
      "action_loss_avg": 0.04873354509472847,
      "epoch": 4.215240478515625e-06,
      "step": 1105
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 1105,
      "training_step_time": 7.497734546661377
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 3.391169786453247,
      "step": 1105
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 1105,
      "training_step_time": 10.151206254959106
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 2.4249696731567383,
      "step": 1105
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 1105,
      "training_step_time": 7.441506385803223
    },
    {
      "epoch": 4.215240478515625e-06,
      "model_forward_time": 3.127040147781372,
      "step": 1105
    },
    {
      "epoch": 4.215240478515625e-06,
      "step": 1105,
      "training_step_time": 10.026280164718628
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 2.44114089012146,
      "step": 1106
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 1106,
      "training_step_time": 7.460859775543213
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 1.7389051914215088,
      "step": 1106
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 1106,
      "training_step_time": 5.187393665313721
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 2.6931450366973877,
      "step": 1106
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 1106,
      "training_step_time": 7.710506439208984
    },
    {
      "epoch": 4.21905517578125e-06,
      "model_forward_time": 3.1523172855377197,
      "step": 1106
    },
    {
      "epoch": 4.21905517578125e-06,
      "step": 1106,
      "training_step_time": 10.335641622543335
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 2.4697656631469727,
      "step": 1107
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 1107,
      "training_step_time": 7.489539861679077
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 2.427664041519165,
      "step": 1107
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 1107,
      "training_step_time": 7.447245359420776
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 3.123311758041382,
      "step": 1107
    },
    {
      "dynamics_loss_avg": 0.06986609026789665,
      "epoch": 4.222869873046875e-06,
      "step": 1107
    },
    {
      "action_loss_avg": 0.051939709950238465,
      "epoch": 4.222869873046875e-06,
      "step": 1107
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 1107,
      "training_step_time": 9.886662006378174
    },
    {
      "epoch": 4.222869873046875e-06,
      "model_forward_time": 3.131624937057495,
      "step": 1107
    },
    {
      "epoch": 4.222869873046875e-06,
      "step": 1107,
      "training_step_time": 10.028274536132812
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 2.7626376152038574,
      "step": 1108
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 1108,
      "training_step_time": 7.787633180618286
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 2.426109552383423,
      "step": 1108
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 1108,
      "training_step_time": 7.4464430809021
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 3.127263307571411,
      "step": 1108
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 1108,
      "training_step_time": 9.88975739479065
    },
    {
      "epoch": 4.2266845703125e-06,
      "model_forward_time": 1.7432105541229248,
      "step": 1108
    },
    {
      "epoch": 4.2266845703125e-06,
      "step": 1108,
      "training_step_time": 10.291832685470581
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 3.1525514125823975,
      "step": 1109
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 1109,
      "training_step_time": 9.917206287384033
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 3.4403908252716064,
      "step": 1109
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 1109,
      "training_step_time": 10.197627782821655
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 2.4294917583465576,
      "step": 1109
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 1109,
      "training_step_time": 7.4444873332977295
    },
    {
      "epoch": 4.230499267578125e-06,
      "model_forward_time": 3.126044750213623,
      "step": 1109
    },
    {
      "epoch": 4.230499267578125e-06,
      "step": 1109,
      "training_step_time": 10.015633344650269
    },
    {
      "epoch": 4.23431396484375e-06,
      "grad_norm": 1.4203922748565674,
      "learning_rate": 1e-05,
      "loss": 0.1316,
      "step": 1110
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 2.4722142219543457,
      "step": 1110
    },
    {
      "dynamics_loss_avg": 0.07136670537292958,
      "epoch": 4.23431396484375e-06,
      "step": 1110
    },
    {
      "action_loss_avg": 0.09012988097965717,
      "epoch": 4.23431396484375e-06,
      "step": 1110
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 1110,
      "training_step_time": 7.491737365722656
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 1.7419822216033936,
      "step": 1110
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 1110,
      "training_step_time": 5.192120552062988
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 3.133517265319824,
      "step": 1110
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 1110,
      "training_step_time": 9.890848159790039
    },
    {
      "epoch": 4.23431396484375e-06,
      "model_forward_time": 3.405688524246216,
      "step": 1110
    },
    {
      "epoch": 4.23431396484375e-06,
      "step": 1110,
      "training_step_time": 10.260442972183228
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 2.4638102054595947,
      "step": 1111
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 1111,
      "training_step_time": 7.481001138687134
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 3.1301565170288086,
      "step": 1111
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 1111,
      "training_step_time": 9.88772463798523
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 3.1352851390838623,
      "step": 1111
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 1111,
      "training_step_time": 9.896223545074463
    },
    {
      "epoch": 4.238128662109375e-06,
      "model_forward_time": 3.1334543228149414,
      "step": 1111
    },
    {
      "epoch": 4.238128662109375e-06,
      "step": 1111,
      "training_step_time": 10.27917218208313
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 1.7588152885437012,
      "step": 1112
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 1112,
      "training_step_time": 5.210885047912598
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 2.4294443130493164,
      "step": 1112
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 1112,
      "training_step_time": 7.447339296340942
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 1.7375407218933105,
      "step": 1112
    },
    {
      "dynamics_loss_avg": 0.06775885857641697,
      "epoch": 4.241943359375e-06,
      "step": 1112
    },
    {
      "action_loss_avg": 0.06493876036256552,
      "epoch": 4.241943359375e-06,
      "step": 1112
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 1112,
      "training_step_time": 5.184314966201782
    },
    {
      "epoch": 4.241943359375e-06,
      "model_forward_time": 3.1246819496154785,
      "step": 1112
    },
    {
      "epoch": 4.241943359375e-06,
      "step": 1112,
      "training_step_time": 10.27990984916687
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 1.792382001876831,
      "step": 1113
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 1113,
      "training_step_time": 5.234693288803101
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 2.4268600940704346,
      "step": 1113
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 1113,
      "training_step_time": 7.439583778381348
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 3.440995693206787,
      "step": 1113
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 1113,
      "training_step_time": 10.196027040481567
    },
    {
      "epoch": 4.245758056640625e-06,
      "model_forward_time": 3.1319196224212646,
      "step": 1113
    },
    {
      "epoch": 4.245758056640625e-06,
      "step": 1113,
      "training_step_time": 10.290001630783081
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 1.7755818367004395,
      "step": 1114
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 1114,
      "training_step_time": 5.225064516067505
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 3.1338536739349365,
      "step": 1114
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 1114,
      "training_step_time": 9.887601375579834
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 3.1327104568481445,
      "step": 1114
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 1114,
      "training_step_time": 9.885840892791748
    },
    {
      "epoch": 4.24957275390625e-06,
      "model_forward_time": 3.3986804485321045,
      "step": 1114
    },
    {
      "epoch": 4.24957275390625e-06,
      "step": 1114,
      "training_step_time": 10.244312286376953
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 2.4509949684143066,
      "step": 1115
    },
    {
      "dynamics_loss_avg": 0.05938109252601862,
      "epoch": 4.253387451171875e-06,
      "step": 1115
    },
    {
      "action_loss_avg": 0.053766800463199614,
      "epoch": 4.253387451171875e-06,
      "step": 1115
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 1115,
      "training_step_time": 7.470574140548706
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 2.4339635372161865,
      "step": 1115
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 1115,
      "training_step_time": 7.446699142456055
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 2.4301156997680664,
      "step": 1115
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 1115,
      "training_step_time": 7.4390549659729
    },
    {
      "epoch": 4.253387451171875e-06,
      "model_forward_time": 3.135687828063965,
      "step": 1115
    },
    {
      "epoch": 4.253387451171875e-06,
      "step": 1115,
      "training_step_time": 10.244548797607422
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 3.428380012512207,
      "step": 1116
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 1116,
      "training_step_time": 10.188580751419067
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 3.1348278522491455,
      "step": 1116
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 1116,
      "training_step_time": 9.891123056411743
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 2.4388270378112793,
      "step": 1116
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 1116,
      "training_step_time": 7.454876899719238
    },
    {
      "epoch": 4.2572021484375e-06,
      "model_forward_time": 3.1321444511413574,
      "step": 1116
    },
    {
      "epoch": 4.2572021484375e-06,
      "step": 1116,
      "training_step_time": 10.253854274749756
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 3.1672420501708984,
      "step": 1117
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 1117,
      "training_step_time": 9.92534613609314
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 3.424649715423584,
      "step": 1117
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 1117,
      "training_step_time": 10.182599067687988
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 3.1346874237060547,
      "step": 1117
    },
    {
      "dynamics_loss_avg": 0.07604971900582314,
      "epoch": 4.261016845703125e-06,
      "step": 1117
    },
    {
      "action_loss_avg": 0.07739566620439291,
      "epoch": 4.261016845703125e-06,
      "step": 1117
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 1117,
      "training_step_time": 9.895909786224365
    },
    {
      "epoch": 4.261016845703125e-06,
      "model_forward_time": 3.1388649940490723,
      "step": 1117
    },
    {
      "epoch": 4.261016845703125e-06,
      "step": 1117,
      "training_step_time": 10.335068225860596
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 3.1431965827941895,
      "step": 1118
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 1118,
      "training_step_time": 9.90137004852295
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 3.132056713104248,
      "step": 1118
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 1118,
      "training_step_time": 9.892261266708374
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 2.717971086502075,
      "step": 1118
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 1118,
      "training_step_time": 7.730955123901367
    },
    {
      "epoch": 4.26483154296875e-06,
      "model_forward_time": 3.1309967041015625,
      "step": 1118
    },
    {
      "epoch": 4.26483154296875e-06,
      "step": 1118,
      "training_step_time": 10.296452760696411
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 3.1471104621887207,
      "step": 1119
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 1119,
      "training_step_time": 9.909324884414673
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 3.128903388977051,
      "step": 1119
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 1119,
      "training_step_time": 9.894189596176147
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 2.4282960891723633,
      "step": 1119
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 1119,
      "training_step_time": 7.447458505630493
    },
    {
      "epoch": 4.268646240234375e-06,
      "model_forward_time": 3.402338743209839,
      "step": 1119
    },
    {
      "epoch": 4.268646240234375e-06,
      "step": 1119,
      "training_step_time": 10.289031028747559
    },
    {
      "epoch": 4.2724609375e-06,
      "grad_norm": 0.6632852554321289,
      "learning_rate": 1e-05,
      "loss": 0.1317,
      "step": 1120
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 2.4412074089050293,
      "step": 1120
    },
    {
      "dynamics_loss_avg": 0.07531239725649357,
      "epoch": 4.2724609375e-06,
      "step": 1120
    },
    {
      "action_loss_avg": 0.061980363167822364,
      "epoch": 4.2724609375e-06,
      "step": 1120
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 1120,
      "training_step_time": 7.46604585647583
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 1.741391897201538,
      "step": 1120
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 1120,
      "training_step_time": 5.194392442703247
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 3.128969192504883,
      "step": 1120
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 1120,
      "training_step_time": 9.891836404800415
    },
    {
      "epoch": 4.2724609375e-06,
      "model_forward_time": 2.431192398071289,
      "step": 1120
    },
    {
      "epoch": 4.2724609375e-06,
      "step": 1120,
      "training_step_time": 10.024981021881104
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 2.4577994346618652,
      "step": 1121
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 1121,
      "training_step_time": 7.478625535964966
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 2.7113616466522217,
      "step": 1121
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 1121,
      "training_step_time": 7.729041814804077
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 3.1309800148010254,
      "step": 1121
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 1121,
      "training_step_time": 9.89107370376587
    },
    {
      "epoch": 4.276275634765625e-06,
      "model_forward_time": 3.1296703815460205,
      "step": 1121
    },
    {
      "epoch": 4.276275634765625e-06,
      "step": 1121,
      "training_step_time": 10.019175052642822
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 3.147134780883789,
      "step": 1122
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 1122,
      "training_step_time": 9.913790941238403
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 3.1320481300354004,
      "step": 1122
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 1122,
      "training_step_time": 9.902858257293701
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 3.4627151489257812,
      "step": 1122
    },
    {
      "dynamics_loss_avg": 0.07340265586972236,
      "epoch": 4.28009033203125e-06,
      "step": 1122
    },
    {
      "action_loss_avg": 0.047299342975020406,
      "epoch": 4.28009033203125e-06,
      "step": 1122
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 1122,
      "training_step_time": 10.23082685470581
    },
    {
      "epoch": 4.28009033203125e-06,
      "model_forward_time": 2.434295177459717,
      "step": 1122
    },
    {
      "epoch": 4.28009033203125e-06,
      "step": 1122,
      "training_step_time": 10.290868997573853
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 3.17121958732605,
      "step": 1123
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 1123,
      "training_step_time": 9.937205076217651
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 3.1320674419403076,
      "step": 1123
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 1123,
      "training_step_time": 9.900856256484985
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 3.1346359252929688,
      "step": 1123
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 1123,
      "training_step_time": 9.899933576583862
    },
    {
      "epoch": 4.283905029296875e-06,
      "model_forward_time": 2.4324586391448975,
      "step": 1123
    },
    {
      "epoch": 4.283905029296875e-06,
      "step": 1123,
      "training_step_time": 10.293451070785522
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 2.723386287689209,
      "step": 1124
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 1124,
      "training_step_time": 7.749752521514893
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 2.428375720977783,
      "step": 1124
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 1124,
      "training_step_time": 7.45475697517395
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 2.426602602005005,
      "step": 1124
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 1124,
      "training_step_time": 7.451459646224976
    },
    {
      "epoch": 4.2877197265625e-06,
      "model_forward_time": 3.12980055809021,
      "step": 1124
    },
    {
      "epoch": 4.2877197265625e-06,
      "step": 1124,
      "training_step_time": 10.30156421661377
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 2.4477479457855225,
      "step": 1125
    },
    {
      "dynamics_loss_avg": 0.06800691820681096,
      "epoch": 4.291534423828125e-06,
      "step": 1125
    },
    {
      "action_loss_avg": 0.06174925696104765,
      "epoch": 4.291534423828125e-06,
      "step": 1125
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 1125,
      "training_step_time": 7.478170394897461
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 2.716557264328003,
      "step": 1125
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 1125,
      "training_step_time": 7.743878126144409
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 2.4329559803009033,
      "step": 1125
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 1125,
      "training_step_time": 7.458201169967651
    },
    {
      "epoch": 4.291534423828125e-06,
      "model_forward_time": 2.430288314819336,
      "step": 1125
    },
    {
      "epoch": 4.291534423828125e-06,
      "step": 1125,
      "training_step_time": 10.024749994277954
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 3.18042254447937,
      "step": 1126
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 1126,
      "training_step_time": 9.950668811798096
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 3.1411705017089844,
      "step": 1126
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 1126,
      "training_step_time": 9.909798622131348
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 3.4309136867523193,
      "step": 1126
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 1126,
      "training_step_time": 10.198569297790527
    },
    {
      "epoch": 4.29534912109375e-06,
      "model_forward_time": 3.1334848403930664,
      "step": 1126
    },
    {
      "epoch": 4.29534912109375e-06,
      "step": 1126,
      "training_step_time": 10.029407501220703
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 3.1825742721557617,
      "step": 1127
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 1127,
      "training_step_time": 9.951652765274048
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 2.429499387741089,
      "step": 1127
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 1127,
      "training_step_time": 7.451638698577881
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 2.4318599700927734,
      "step": 1127
    },
    {
      "dynamics_loss_avg": 0.0768941380083561,
      "epoch": 4.299163818359375e-06,
      "step": 1127
    },
    {
      "action_loss_avg": 0.06992660779505969,
      "epoch": 4.299163818359375e-06,
      "step": 1127
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 1127,
      "training_step_time": 7.457870960235596
    },
    {
      "epoch": 4.299163818359375e-06,
      "model_forward_time": 3.41713547706604,
      "step": 1127
    },
    {
      "epoch": 4.299163818359375e-06,
      "step": 1127,
      "training_step_time": 10.33536672592163
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 3.1511354446411133,
      "step": 1128
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 1128,
      "training_step_time": 9.920766115188599
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 3.1330482959747314,
      "step": 1128
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 1128,
      "training_step_time": 9.897794008255005
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 2.4306933879852295,
      "step": 1128
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 1128,
      "training_step_time": 7.453998327255249
    },
    {
      "epoch": 4.302978515625e-06,
      "model_forward_time": 2.433812379837036,
      "step": 1128
    },
    {
      "epoch": 4.302978515625e-06,
      "step": 1128,
      "training_step_time": 10.290318250656128
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 2.4466280937194824,
      "step": 1129
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 1129,
      "training_step_time": 7.4723875522613525
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 2.7404983043670654,
      "step": 1129
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 1129,
      "training_step_time": 7.75990891456604
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 2.4262852668762207,
      "step": 1129
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 1129,
      "training_step_time": 7.442863941192627
    },
    {
      "epoch": 4.306793212890625e-06,
      "model_forward_time": 2.4288554191589355,
      "step": 1129
    },
    {
      "epoch": 4.306793212890625e-06,
      "step": 1129,
      "training_step_time": 10.271944046020508
    },
    {
      "epoch": 4.31060791015625e-06,
      "grad_norm": 0.9310925602912903,
      "learning_rate": 1e-05,
      "loss": 0.1316,
      "step": 1130
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 3.1779885292053223,
      "step": 1130
    },
    {
      "dynamics_loss_avg": 0.08946855068206787,
      "epoch": 4.31060791015625e-06,
      "step": 1130
    },
    {
      "action_loss_avg": 0.07070496566593647,
      "epoch": 4.31060791015625e-06,
      "step": 1130
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 1130,
      "training_step_time": 9.943952798843384
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 1.7356009483337402,
      "step": 1130
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 1130,
      "training_step_time": 5.185978651046753
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 3.4168918132781982,
      "step": 1130
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 1130,
      "training_step_time": 10.1807701587677
    },
    {
      "epoch": 4.31060791015625e-06,
      "model_forward_time": 3.121838092803955,
      "step": 1130
    },
    {
      "epoch": 4.31060791015625e-06,
      "step": 1130,
      "training_step_time": 10.014161825180054
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 2.4384818077087402,
      "step": 1131
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 1131,
      "training_step_time": 7.460476636886597
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 3.1289169788360596,
      "step": 1131
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 1131,
      "training_step_time": 9.89141321182251
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 2.427027463912964,
      "step": 1131
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 1131,
      "training_step_time": 7.4417805671691895
    },
    {
      "epoch": 4.314422607421875e-06,
      "model_forward_time": 2.423621892929077,
      "step": 1131
    },
    {
      "epoch": 4.314422607421875e-06,
      "step": 1131,
      "training_step_time": 10.260721206665039
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 2.7338192462921143,
      "step": 1132
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 1132,
      "training_step_time": 7.754220485687256
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 3.124083995819092,
      "step": 1132
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 1132,
      "training_step_time": 9.887134313583374
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 2.4265079498291016,
      "step": 1132
    },
    {
      "dynamics_loss_avg": 0.06571183986961841,
      "epoch": 4.3182373046875e-06,
      "step": 1132
    },
    {
      "action_loss_avg": 0.0543631112203002,
      "epoch": 4.3182373046875e-06,
      "step": 1132
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 1132,
      "training_step_time": 7.4527716636657715
    },
    {
      "epoch": 4.3182373046875e-06,
      "model_forward_time": 3.1282663345336914,
      "step": 1132
    },
    {
      "epoch": 4.3182373046875e-06,
      "step": 1132,
      "training_step_time": 10.025684118270874
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 2.4674484729766846,
      "step": 1133
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 1133,
      "training_step_time": 7.488068342208862
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 3.4318478107452393,
      "step": 1133
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 1133,
      "training_step_time": 10.194210290908813
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 3.1276161670684814,
      "step": 1133
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 1133,
      "training_step_time": 9.89476490020752
    },
    {
      "epoch": 4.322052001953125e-06,
      "model_forward_time": 2.4268054962158203,
      "step": 1133
    },
    {
      "epoch": 4.322052001953125e-06,
      "step": 1133,
      "training_step_time": 10.299365282058716
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 1.7802770137786865,
      "step": 1134
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 1134,
      "training_step_time": 5.234225511550903
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 3.1194424629211426,
      "step": 1134
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 1134,
      "training_step_time": 9.877898216247559
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 3.1248154640197754,
      "step": 1134
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 1134,
      "training_step_time": 10.169613122940063
    },
    {
      "epoch": 4.32586669921875e-06,
      "model_forward_time": 2.426971435546875,
      "step": 1134
    },
    {
      "epoch": 4.32586669921875e-06,
      "step": 1134,
      "training_step_time": 10.267205238342285
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 2.4391472339630127,
      "step": 1135
    },
    {
      "dynamics_loss_avg": 0.08375965692102909,
      "epoch": 4.329681396484375e-06,
      "step": 1135
    },
    {
      "action_loss_avg": 0.07131253220140935,
      "epoch": 4.329681396484375e-06,
      "step": 1135
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 1135,
      "training_step_time": 7.46284818649292
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 3.121029853820801,
      "step": 1135
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 1135,
      "training_step_time": 9.874175786972046
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 2.4203269481658936,
      "step": 1135
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 1135,
      "training_step_time": 7.435892105102539
    },
    {
      "epoch": 4.329681396484375e-06,
      "model_forward_time": 2.4236278533935547,
      "step": 1135
    },
    {
      "epoch": 4.329681396484375e-06,
      "step": 1135,
      "training_step_time": 10.26721453666687
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 2.762625217437744,
      "step": 1136
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 1136,
      "training_step_time": 7.781001091003418
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 1.7334787845611572,
      "step": 1136
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 1136,
      "training_step_time": 5.177751064300537
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 3.1276214122772217,
      "step": 1136
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 1136,
      "training_step_time": 9.888784170150757
    },
    {
      "epoch": 4.33349609375e-06,
      "model_forward_time": 2.426412582397461,
      "step": 1136
    },
    {
      "epoch": 4.33349609375e-06,
      "step": 1136,
      "training_step_time": 10.307450294494629
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 3.135921001434326,
      "step": 1137
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 1137,
      "training_step_time": 9.903289079666138
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 3.130246639251709,
      "step": 1137
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 1137,
      "training_step_time": 9.897063255310059
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 2.7196195125579834,
      "step": 1137
    },
    {
      "dynamics_loss_avg": 0.06982035767287016,
      "epoch": 4.337310791015625e-06,
      "step": 1137
    },
    {
      "action_loss_avg": 0.07618273533880711,
      "epoch": 4.337310791015625e-06,
      "step": 1137
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 1137,
      "training_step_time": 7.7496020793914795
    },
    {
      "epoch": 4.337310791015625e-06,
      "model_forward_time": 2.42535400390625,
      "step": 1137
    },
    {
      "epoch": 4.337310791015625e-06,
      "step": 1137,
      "training_step_time": 10.023056983947754
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 2.444056510925293,
      "step": 1138
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 1138,
      "training_step_time": 7.465076923370361
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 3.12943959236145,
      "step": 1138
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 1138,
      "training_step_time": 9.89744234085083
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 2.430824041366577,
      "step": 1138
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 1138,
      "training_step_time": 7.4589855670928955
    },
    {
      "epoch": 4.34112548828125e-06,
      "model_forward_time": 2.6967527866363525,
      "step": 1138
    },
    {
      "epoch": 4.34112548828125e-06,
      "step": 1138,
      "training_step_time": 10.313653945922852
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 2.4436192512512207,
      "step": 1139
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 1139,
      "training_step_time": 7.465880870819092
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 2.426117181777954,
      "step": 1139
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 1139,
      "training_step_time": 7.447359085083008
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 2.4256598949432373,
      "step": 1139
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 1139,
      "training_step_time": 7.447031497955322
    },
    {
      "epoch": 4.344940185546875e-06,
      "model_forward_time": 2.427896738052368,
      "step": 1139
    },
    {
      "epoch": 4.344940185546875e-06,
      "step": 1139,
      "training_step_time": 10.31930947303772
    },
    {
      "epoch": 4.3487548828125e-06,
      "grad_norm": 1.3969039916992188,
      "learning_rate": 1e-05,
      "loss": 0.131,
      "step": 1140
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 2.4432992935180664,
      "step": 1140
    },
    {
      "dynamics_loss_avg": 0.070553267467767,
      "epoch": 4.3487548828125e-06,
      "step": 1140
    },
    {
      "action_loss_avg": 0.05596978706307709,
      "epoch": 4.3487548828125e-06,
      "step": 1140
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 1140,
      "training_step_time": 7.470815896987915
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 2.7353508472442627,
      "step": 1140
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 1140,
      "training_step_time": 7.761078596115112
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 2.423489809036255,
      "step": 1140
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 1140,
      "training_step_time": 7.440704584121704
    },
    {
      "epoch": 4.3487548828125e-06,
      "model_forward_time": 3.127004384994507,
      "step": 1140
    },
    {
      "epoch": 4.3487548828125e-06,
      "step": 1140,
      "training_step_time": 10.337303161621094
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 3.1486384868621826,
      "step": 1141
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 1141,
      "training_step_time": 9.917253971099854
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 2.431246757507324,
      "step": 1141
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 1141,
      "training_step_time": 7.459592342376709
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 2.69207763671875,
      "step": 1141
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 1141,
      "training_step_time": 7.719322204589844
    },
    {
      "epoch": 4.352569580078125e-06,
      "model_forward_time": 3.128612518310547,
      "step": 1141
    },
    {
      "epoch": 4.352569580078125e-06,
      "step": 1141,
      "training_step_time": 10.263489961624146
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 2.4511830806732178,
      "step": 1142
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 1142,
      "training_step_time": 7.476778745651245
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 3.1294455528259277,
      "step": 1142
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 1142,
      "training_step_time": 9.89763355255127
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 1.7456767559051514,
      "step": 1142
    },
    {
      "dynamics_loss_avg": 0.06930369064211846,
      "epoch": 4.35638427734375e-06,
      "step": 1142
    },
    {
      "action_loss_avg": 0.06212929468601942,
      "epoch": 4.35638427734375e-06,
      "step": 1142
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 1142,
      "training_step_time": 5.200941562652588
    },
    {
      "epoch": 4.35638427734375e-06,
      "model_forward_time": 3.429745674133301,
      "step": 1142
    },
    {
      "epoch": 4.35638427734375e-06,
      "step": 1142,
      "training_step_time": 10.297932386398315
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 3.1456124782562256,
      "step": 1143
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 1143,
      "training_step_time": 9.921592473983765
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 2.4293932914733887,
      "step": 1143
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 1143,
      "training_step_time": 7.455655336380005
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 2.4276297092437744,
      "step": 1143
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 1143,
      "training_step_time": 7.452977657318115
    },
    {
      "epoch": 4.360198974609375e-06,
      "model_forward_time": 3.122091293334961,
      "step": 1143
    },
    {
      "epoch": 4.360198974609375e-06,
      "step": 1143,
      "training_step_time": 10.273284435272217
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 3.1670851707458496,
      "step": 1144
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 1144,
      "training_step_time": 9.932553052902222
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 3.399061441421509,
      "step": 1144
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 1144,
      "training_step_time": 10.167405605316162
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 3.1319730281829834,
      "step": 1144
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 1144,
      "training_step_time": 9.903141260147095
    },
    {
      "epoch": 4.364013671875e-06,
      "model_forward_time": 3.1289496421813965,
      "step": 1144
    },
    {
      "epoch": 4.364013671875e-06,
      "step": 1144,
      "training_step_time": 10.312968969345093
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 2.4457614421844482,
      "step": 1145
    },
    {
      "dynamics_loss_avg": 0.06497825626283885,
      "epoch": 4.367828369140625e-06,
      "step": 1145
    },
    {
      "action_loss_avg": 0.05810399688780308,
      "epoch": 4.367828369140625e-06,
      "step": 1145
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 1145,
      "training_step_time": 7.478602409362793
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 3.12955904006958,
      "step": 1145
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 1145,
      "training_step_time": 9.896883249282837
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 2.694279432296753,
      "step": 1145
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 1145,
      "training_step_time": 7.722575664520264
    },
    {
      "epoch": 4.367828369140625e-06,
      "model_forward_time": 2.4278616905212402,
      "step": 1145
    },
    {
      "epoch": 4.367828369140625e-06,
      "step": 1145,
      "training_step_time": 10.286071062088013
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 2.469569206237793,
      "step": 1146
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 1146,
      "training_step_time": 7.503185987472534
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 2.4255294799804688,
      "step": 1146
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 1146,
      "training_step_time": 7.445721864700317
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 2.4232683181762695,
      "step": 1146
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 1146,
      "training_step_time": 7.44323468208313
    },
    {
      "epoch": 4.37164306640625e-06,
      "model_forward_time": 3.406035900115967,
      "step": 1146
    },
    {
      "epoch": 4.37164306640625e-06,
      "step": 1146,
      "training_step_time": 10.279709339141846
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 2.4754254817962646,
      "step": 1147
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 1147,
      "training_step_time": 7.5002570152282715
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 1.7394688129425049,
      "step": 1147
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 1147,
      "training_step_time": 5.19423508644104
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 2.423884868621826,
      "step": 1147
    },
    {
      "dynamics_loss_avg": 0.08593617305159569,
      "epoch": 4.375457763671875e-06,
      "step": 1147
    },
    {
      "action_loss_avg": 0.07575019020587206,
      "epoch": 4.375457763671875e-06,
      "step": 1147
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 1147,
      "training_step_time": 7.444556474685669
    },
    {
      "epoch": 4.375457763671875e-06,
      "model_forward_time": 3.1234543323516846,
      "step": 1147
    },
    {
      "epoch": 4.375457763671875e-06,
      "step": 1147,
      "training_step_time": 10.310631513595581
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 2.7438242435455322,
      "step": 1148
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 1148,
      "training_step_time": 7.765210390090942
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 3.126908540725708,
      "step": 1148
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 1148,
      "training_step_time": 9.890384674072266
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 1.741365909576416,
      "step": 1148
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 1148,
      "training_step_time": 5.196744680404663
    },
    {
      "epoch": 4.3792724609375e-06,
      "model_forward_time": 2.426671028137207,
      "step": 1148
    },
    {
      "epoch": 4.3792724609375e-06,
      "step": 1148,
      "training_step_time": 10.037251949310303
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 3.1674928665161133,
      "step": 1149
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 1149,
      "training_step_time": 9.930322170257568
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 2.426992893218994,
      "step": 1149
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 1149,
      "training_step_time": 7.451170921325684
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 3.3740670680999756,
      "step": 1149
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 1149,
      "training_step_time": 10.13850712776184
    },
    {
      "epoch": 4.383087158203125e-06,
      "model_forward_time": 3.1314775943756104,
      "step": 1149
    },
    {
      "epoch": 4.383087158203125e-06,
      "step": 1149,
      "training_step_time": 10.31923508644104
    },
    {
      "epoch": 4.38690185546875e-06,
      "grad_norm": 0.4615877568721771,
      "learning_rate": 1e-05,
      "loss": 0.1296,
      "step": 1150
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 2.448002815246582,
      "step": 1150
    },
    {
      "dynamics_loss_avg": 0.06877737604081631,
      "epoch": 4.38690185546875e-06,
      "step": 1150
    },
    {
      "action_loss_avg": 0.06165155945345759,
      "epoch": 4.38690185546875e-06,
      "step": 1150
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 1150,
      "training_step_time": 7.478396892547607
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 2.4224867820739746,
      "step": 1150
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 1150,
      "training_step_time": 7.444184303283691
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 1.738276481628418,
      "step": 1150
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 1150,
      "training_step_time": 5.1895751953125
    },
    {
      "epoch": 4.38690185546875e-06,
      "model_forward_time": 2.4235897064208984,
      "step": 1150
    },
    {
      "epoch": 4.38690185546875e-06,
      "step": 1150,
      "training_step_time": 10.306742191314697
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 3.4926226139068604,
      "step": 1151
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 1151,
      "training_step_time": 10.263970136642456
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 3.128262996673584,
      "step": 1151
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 1151,
      "training_step_time": 9.89880895614624
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 2.4275169372558594,
      "step": 1151
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 1151,
      "training_step_time": 7.4543678760528564
    },
    {
      "epoch": 4.390716552734375e-06,
      "model_forward_time": 3.1283411979675293,
      "step": 1151
    },
    {
      "epoch": 4.390716552734375e-06,
      "step": 1151,
      "training_step_time": 10.29139518737793
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 3.169799327850342,
      "step": 1152
    },
    {
      "epoch": 4.39453125e-06,
      "step": 1152,
      "training_step_time": 10.20989441871643
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 3.127448797225952,
      "step": 1152
    },
    {
      "epoch": 4.39453125e-06,
      "step": 1152,
      "training_step_time": 9.89400601387024
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 2.4285240173339844,
      "step": 1152
    },
    {
      "dynamics_loss_avg": 0.07444831766188145,
      "epoch": 4.39453125e-06,
      "step": 1152
    },
    {
      "action_loss_avg": 0.0537126861512661,
      "epoch": 4.39453125e-06,
      "step": 1152
    },
    {
      "epoch": 4.39453125e-06,
      "step": 1152,
      "training_step_time": 7.454902172088623
    },
    {
      "epoch": 4.39453125e-06,
      "model_forward_time": 3.1232128143310547,
      "step": 1152
    },
    {
      "epoch": 4.39453125e-06,
      "step": 1152,
      "training_step_time": 10.28833818435669
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 2.463892936706543,
      "step": 1153
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 1153,
      "training_step_time": 7.485821008682251
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 2.4271717071533203,
      "step": 1153
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 1153,
      "training_step_time": 7.454244136810303
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 3.3952744007110596,
      "step": 1153
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 1153,
      "training_step_time": 10.159794092178345
    },
    {
      "epoch": 4.398345947265625e-06,
      "model_forward_time": 3.128040075302124,
      "step": 1153
    },
    {
      "epoch": 4.398345947265625e-06,
      "step": 1153,
      "training_step_time": 10.01792573928833
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 2.4809327125549316,
      "step": 1154
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 1154,
      "training_step_time": 7.506566047668457
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 2.4250426292419434,
      "step": 1154
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 1154,
      "training_step_time": 7.44822359085083
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 3.128614902496338,
      "step": 1154
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 1154,
      "training_step_time": 9.895040273666382
    },
    {
      "epoch": 4.40216064453125e-06,
      "model_forward_time": 3.409507989883423,
      "step": 1154
    },
    {
      "epoch": 4.40216064453125e-06,
      "step": 1154,
      "training_step_time": 10.265791177749634
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 3.160468578338623,
      "step": 1155
    },
    {
      "dynamics_loss_avg": 0.07219168730080128,
      "epoch": 4.405975341796875e-06,
      "step": 1155
    },
    {
      "action_loss_avg": 0.05625092405825853,
      "epoch": 4.405975341796875e-06,
      "step": 1155
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 1155,
      "training_step_time": 9.938919305801392
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 3.132969856262207,
      "step": 1155
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 1155,
      "training_step_time": 9.901411294937134
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 3.1278741359710693,
      "step": 1155
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 1155,
      "training_step_time": 9.893725156784058
    },
    {
      "epoch": 4.405975341796875e-06,
      "model_forward_time": 3.1362950801849365,
      "step": 1155
    },
    {
      "epoch": 4.405975341796875e-06,
      "step": 1155,
      "training_step_time": 10.2883460521698
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 3.405979871749878,
      "step": 1156
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 1156,
      "training_step_time": 10.172946214675903
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 2.429368257522583,
      "step": 1156
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 1156,
      "training_step_time": 7.4528374671936035
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 3.1256706714630127,
      "step": 1156
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 1156,
      "training_step_time": 9.891639471054077
    },
    {
      "epoch": 4.4097900390625e-06,
      "model_forward_time": 1.7401432991027832,
      "step": 1156
    },
    {
      "epoch": 4.4097900390625e-06,
      "step": 1156,
      "training_step_time": 10.249361038208008
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 2.4376845359802246,
      "step": 1157
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 1157,
      "training_step_time": 7.464062929153442
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 2.42470645904541,
      "step": 1157
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 1157,
      "training_step_time": 7.708597660064697
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 2.4241554737091064,
      "step": 1157
    },
    {
      "dynamics_loss_avg": 0.0758731011301279,
      "epoch": 4.413604736328125e-06,
      "step": 1157
    },
    {
      "action_loss_avg": 0.08360658064484597,
      "epoch": 4.413604736328125e-06,
      "step": 1157
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 1157,
      "training_step_time": 7.441120147705078
    },
    {
      "epoch": 4.413604736328125e-06,
      "model_forward_time": 3.1256635189056396,
      "step": 1157
    },
    {
      "epoch": 4.413604736328125e-06,
      "step": 1157,
      "training_step_time": 10.023449420928955
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 3.1677558422088623,
      "step": 1158
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 1158,
      "training_step_time": 9.93338418006897
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 3.127432346343994,
      "step": 1158
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 1158,
      "training_step_time": 9.891123294830322
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 3.4298534393310547,
      "step": 1158
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 1158,
      "training_step_time": 10.198009729385376
    },
    {
      "epoch": 4.41741943359375e-06,
      "model_forward_time": 2.4250192642211914,
      "step": 1158
    },
    {
      "epoch": 4.41741943359375e-06,
      "step": 1158,
      "training_step_time": 10.021955251693726
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 2.4407360553741455,
      "step": 1159
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 1159,
      "training_step_time": 7.464534282684326
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 2.4240317344665527,
      "step": 1159
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 1159,
      "training_step_time": 7.442216396331787
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 2.4271600246429443,
      "step": 1159
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 1159,
      "training_step_time": 7.44620680809021
    },
    {
      "epoch": 4.421234130859375e-06,
      "model_forward_time": 3.124443292617798,
      "step": 1159
    },
    {
      "epoch": 4.421234130859375e-06,
      "step": 1159,
      "training_step_time": 10.020811796188354
    },
    {
      "epoch": 4.425048828125e-06,
      "grad_norm": 0.6421554684638977,
      "learning_rate": 1e-05,
      "loss": 0.1277,
      "step": 1160
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 3.4584574699401855,
      "step": 1160
    },
    {
      "dynamics_loss_avg": 0.05624879151582718,
      "epoch": 4.425048828125e-06,
      "step": 1160
    },
    {
      "action_loss_avg": 0.04421250284649432,
      "epoch": 4.425048828125e-06,
      "step": 1160
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 1160,
      "training_step_time": 10.224331855773926
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 2.4342095851898193,
      "step": 1160
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 1160,
      "training_step_time": 7.4566121101379395
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 2.429694652557373,
      "step": 1160
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 1160,
      "training_step_time": 7.451702356338501
    },
    {
      "epoch": 4.425048828125e-06,
      "model_forward_time": 3.1328108310699463,
      "step": 1160
    },
    {
      "epoch": 4.425048828125e-06,
      "step": 1160,
      "training_step_time": 10.301738739013672
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 3.1584019660949707,
      "step": 1161
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 1161,
      "training_step_time": 9.926330327987671
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 3.386079788208008,
      "step": 1161
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 1161,
      "training_step_time": 10.157429218292236
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 1.745664119720459,
      "step": 1161
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 1161,
      "training_step_time": 5.199828863143921
    },
    {
      "epoch": 4.428863525390625e-06,
      "model_forward_time": 1.7414202690124512,
      "step": 1161
    },
    {
      "epoch": 4.428863525390625e-06,
      "step": 1161,
      "training_step_time": 10.305330753326416
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 2.445061445236206,
      "step": 1162
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 1162,
      "training_step_time": 7.464818000793457
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 2.4306437969207764,
      "step": 1162
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 1162,
      "training_step_time": 7.450381278991699
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 2.4302525520324707,
      "step": 1162
    },
    {
      "dynamics_loss_avg": 0.07105659246444702,
      "epoch": 4.43267822265625e-06,
      "step": 1162
    },
    {
      "action_loss_avg": 0.06534523479640483,
      "epoch": 4.43267822265625e-06,
      "step": 1162
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 1162,
      "training_step_time": 7.4484593868255615
    },
    {
      "epoch": 4.43267822265625e-06,
      "model_forward_time": 3.430987596511841,
      "step": 1162
    },
    {
      "epoch": 4.43267822265625e-06,
      "step": 1162,
      "training_step_time": 10.288698196411133
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 3.1495866775512695,
      "step": 1163
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 1163,
      "training_step_time": 9.91572093963623
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 1.7394862174987793,
      "step": 1163
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 1163,
      "training_step_time": 5.186193227767944
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 2.4261422157287598,
      "step": 1163
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 1163,
      "training_step_time": 7.436651945114136
    },
    {
      "epoch": 4.436492919921875e-06,
      "model_forward_time": 3.128751516342163,
      "step": 1163
    },
    {
      "epoch": 4.436492919921875e-06,
      "step": 1163,
      "training_step_time": 10.268434762954712
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 1.7880194187164307,
      "step": 1164
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 1164,
      "training_step_time": 5.245539426803589
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 3.403644561767578,
      "step": 1164
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 1164,
      "training_step_time": 10.157866716384888
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 2.42891788482666,
      "step": 1164
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 1164,
      "training_step_time": 7.445636749267578
    },
    {
      "epoch": 4.4403076171875e-06,
      "model_forward_time": 3.130134344100952,
      "step": 1164
    },
    {
      "epoch": 4.4403076171875e-06,
      "step": 1164,
      "training_step_time": 10.021541118621826
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 3.17891263961792,
      "step": 1165
    },
    {
      "dynamics_loss_avg": 0.07871616743505001,
      "epoch": 4.444122314453125e-06,
      "step": 1165
    },
    {
      "action_loss_avg": 0.059309663530439136,
      "epoch": 4.444122314453125e-06,
      "step": 1165
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 1165,
      "training_step_time": 9.943545579910278
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 3.1278388500213623,
      "step": 1165
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 1165,
      "training_step_time": 9.888980627059937
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 3.38352632522583,
      "step": 1165
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 1165,
      "training_step_time": 10.145787000656128
    },
    {
      "epoch": 4.444122314453125e-06,
      "model_forward_time": 3.136082172393799,
      "step": 1165
    },
    {
      "epoch": 4.444122314453125e-06,
      "step": 1165,
      "training_step_time": 10.278635263442993
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 3.174520969390869,
      "step": 1166
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 1166,
      "training_step_time": 9.937167882919312
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 2.429060697555542,
      "step": 1166
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 1166,
      "training_step_time": 7.449981689453125
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 3.129225969314575,
      "step": 1166
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 1166,
      "training_step_time": 9.900503158569336
    },
    {
      "epoch": 4.44793701171875e-06,
      "model_forward_time": 2.7484660148620605,
      "step": 1166
    },
    {
      "epoch": 4.44793701171875e-06,
      "step": 1166,
      "training_step_time": 10.304668188095093
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 3.1818318367004395,
      "step": 1167
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 1167,
      "training_step_time": 9.94667387008667
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 2.4305479526519775,
      "step": 1167
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 1167,
      "training_step_time": 7.449012756347656
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 2.4313132762908936,
      "step": 1167
    },
    {
      "dynamics_loss_avg": 0.07737442404031754,
      "epoch": 4.451751708984375e-06,
      "step": 1167
    },
    {
      "action_loss_avg": 0.05715412907302379,
      "epoch": 4.451751708984375e-06,
      "step": 1167
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 1167,
      "training_step_time": 7.45372462272644
    },
    {
      "epoch": 4.451751708984375e-06,
      "model_forward_time": 3.13268780708313,
      "step": 1167
    },
    {
      "epoch": 4.451751708984375e-06,
      "step": 1167,
      "training_step_time": 10.022945880889893
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 2.0424273014068604,
      "step": 1168
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 1168,
      "training_step_time": 5.494035482406616
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 2.429381847381592,
      "step": 1168
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 1168,
      "training_step_time": 7.448504209518433
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 2.4281814098358154,
      "step": 1168
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 1168,
      "training_step_time": 7.446922779083252
    },
    {
      "epoch": 4.45556640625e-06,
      "model_forward_time": 2.428128957748413,
      "step": 1168
    },
    {
      "epoch": 4.45556640625e-06,
      "step": 1168,
      "training_step_time": 10.27159333229065
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 3.178018569946289,
      "step": 1169
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 1169,
      "training_step_time": 9.93606448173523
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 3.394268035888672,
      "step": 1169
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 1169,
      "training_step_time": 10.147921085357666
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 3.132429361343384,
      "step": 1169
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 1169,
      "training_step_time": 9.88908576965332
    },
    {
      "epoch": 4.459381103515625e-06,
      "model_forward_time": 3.1329545974731445,
      "step": 1169
    },
    {
      "epoch": 4.459381103515625e-06,
      "step": 1169,
      "training_step_time": 10.015748262405396
    },
    {
      "epoch": 4.46319580078125e-06,
      "grad_norm": 0.8007385730743408,
      "learning_rate": 1e-05,
      "loss": 0.127,
      "step": 1170
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 2.4394702911376953,
      "step": 1170
    },
    {
      "dynamics_loss_avg": 0.07050458751618863,
      "epoch": 4.46319580078125e-06,
      "step": 1170
    },
    {
      "action_loss_avg": 0.06187467500567436,
      "epoch": 4.46319580078125e-06,
      "step": 1170
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 1170,
      "training_step_time": 7.459370136260986
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 1.7363619804382324,
      "step": 1170
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 1170,
      "training_step_time": 5.185416221618652
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 1.7384629249572754,
      "step": 1170
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 1170,
      "training_step_time": 5.186759948730469
    },
    {
      "epoch": 4.46319580078125e-06,
      "model_forward_time": 2.43003511428833,
      "step": 1170
    },
    {
      "epoch": 4.46319580078125e-06,
      "step": 1170,
      "training_step_time": 10.303843021392822
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 2.4429264068603516,
      "step": 1171
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 1171,
      "training_step_time": 7.465860366821289
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 3.131721258163452,
      "step": 1171
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 1171,
      "training_step_time": 9.896817922592163
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 2.429320812225342,
      "step": 1171
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 1171,
      "training_step_time": 7.453192949295044
    },
    {
      "epoch": 4.467010498046875e-06,
      "model_forward_time": 2.4300031661987305,
      "step": 1171
    },
    {
      "epoch": 4.467010498046875e-06,
      "step": 1171,
      "training_step_time": 10.311966180801392
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 2.469248056411743,
      "step": 1172
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 1172,
      "training_step_time": 7.49188494682312
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 2.6727142333984375,
      "step": 1172
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 1172,
      "training_step_time": 7.69199800491333
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 3.125019073486328,
      "step": 1172
    },
    {
      "dynamics_loss_avg": 0.08455742597579956,
      "epoch": 4.4708251953125e-06,
      "step": 1172
    },
    {
      "action_loss_avg": 0.05906684305518865,
      "epoch": 4.4708251953125e-06,
      "step": 1172
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 1172,
      "training_step_time": 9.89508867263794
    },
    {
      "epoch": 4.4708251953125e-06,
      "model_forward_time": 3.1318297386169434,
      "step": 1172
    },
    {
      "epoch": 4.4708251953125e-06,
      "step": 1172,
      "training_step_time": 10.008591651916504
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 2.4467732906341553,
      "step": 1173
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 1173,
      "training_step_time": 7.47104287147522
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 2.4286015033721924,
      "step": 1173
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 1173,
      "training_step_time": 7.450943231582642
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 2.702883720397949,
      "step": 1173
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 1173,
      "training_step_time": 7.727538585662842
    },
    {
      "epoch": 4.474639892578125e-06,
      "model_forward_time": 1.7420809268951416,
      "step": 1173
    },
    {
      "epoch": 4.474639892578125e-06,
      "step": 1173,
      "training_step_time": 10.275188446044922
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 2.443937301635742,
      "step": 1174
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 1174,
      "training_step_time": 7.46053409576416
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 3.127476930618286,
      "step": 1174
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 1174,
      "training_step_time": 9.883599758148193
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 3.128612756729126,
      "step": 1174
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 1174,
      "training_step_time": 9.889376640319824
    },
    {
      "epoch": 4.47845458984375e-06,
      "model_forward_time": 2.431061267852783,
      "step": 1174
    },
    {
      "epoch": 4.47845458984375e-06,
      "step": 1174,
      "training_step_time": 10.257988691329956
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 2.1129190921783447,
      "step": 1175
    },
    {
      "dynamics_loss_avg": 0.07601450271904468,
      "epoch": 4.482269287109375e-06,
      "step": 1175
    },
    {
      "action_loss_avg": 0.050300211552530526,
      "epoch": 4.482269287109375e-06,
      "step": 1175
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 1175,
      "training_step_time": 5.561899900436401
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 3.122363805770874,
      "step": 1175
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 1175,
      "training_step_time": 9.880899906158447
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 3.1325607299804688,
      "step": 1175
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 1175,
      "training_step_time": 9.894053936004639
    },
    {
      "epoch": 4.482269287109375e-06,
      "model_forward_time": 3.129642963409424,
      "step": 1175
    },
    {
      "epoch": 4.482269287109375e-06,
      "step": 1175,
      "training_step_time": 10.265816688537598
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 2.4684367179870605,
      "step": 1176
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 1176,
      "training_step_time": 7.494928359985352
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 3.374760389328003,
      "step": 1176
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 1176,
      "training_step_time": 10.137335062026978
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 3.129426956176758,
      "step": 1176
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 1176,
      "training_step_time": 9.8934965133667
    },
    {
      "epoch": 4.486083984375e-06,
      "model_forward_time": 3.1356394290924072,
      "step": 1176
    },
    {
      "epoch": 4.486083984375e-06,
      "step": 1176,
      "training_step_time": 10.261569738388062
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 2.477731943130493,
      "step": 1177
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 1177,
      "training_step_time": 7.492724418640137
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 3.1319491863250732,
      "step": 1177
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 1177,
      "training_step_time": 9.898416996002197
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 3.408869743347168,
      "step": 1177
    },
    {
      "dynamics_loss_avg": 0.07123685292899609,
      "epoch": 4.489898681640625e-06,
      "step": 1177
    },
    {
      "action_loss_avg": 0.05514475367963314,
      "epoch": 4.489898681640625e-06,
      "step": 1177
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 1177,
      "training_step_time": 10.17625093460083
    },
    {
      "epoch": 4.489898681640625e-06,
      "model_forward_time": 3.131171703338623,
      "step": 1177
    },
    {
      "epoch": 4.489898681640625e-06,
      "step": 1177,
      "training_step_time": 10.278611421585083
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 2.477545976638794,
      "step": 1178
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 1178,
      "training_step_time": 7.502451419830322
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 2.4321305751800537,
      "step": 1178
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 1178,
      "training_step_time": 7.452397584915161
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 3.126183032989502,
      "step": 1178
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 1178,
      "training_step_time": 9.891902685165405
    },
    {
      "epoch": 4.49371337890625e-06,
      "model_forward_time": 3.4205055236816406,
      "step": 1178
    },
    {
      "epoch": 4.49371337890625e-06,
      "step": 1178,
      "training_step_time": 10.281386852264404
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 2.448833703994751,
      "step": 1179
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 1179,
      "training_step_time": 7.472140789031982
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 3.133017063140869,
      "step": 1179
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 1179,
      "training_step_time": 9.897221326828003
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 2.431492567062378,
      "step": 1179
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 1179,
      "training_step_time": 7.456566333770752
    },
    {
      "epoch": 4.497528076171875e-06,
      "model_forward_time": 3.1306214332580566,
      "step": 1179
    },
    {
      "epoch": 4.497528076171875e-06,
      "step": 1179,
      "training_step_time": 10.311239957809448
    },
    {
      "epoch": 4.5013427734375e-06,
      "grad_norm": 0.9170838594436646,
      "learning_rate": 1e-05,
      "loss": 0.1328,
      "step": 1180
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 3.1435086727142334,
      "step": 1180
    },
    {
      "dynamics_loss_avg": 0.08321303799748421,
      "epoch": 4.5013427734375e-06,
      "step": 1180
    },
    {
      "action_loss_avg": 0.07113311532884836,
      "epoch": 4.5013427734375e-06,
      "step": 1180
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 1180,
      "training_step_time": 9.907621383666992
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 3.406891107559204,
      "step": 1180
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 1180,
      "training_step_time": 10.168697834014893
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 3.12868595123291,
      "step": 1180
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 1180,
      "training_step_time": 9.892327785491943
    },
    {
      "epoch": 4.5013427734375e-06,
      "model_forward_time": 1.7415616512298584,
      "step": 1180
    },
    {
      "epoch": 4.5013427734375e-06,
      "step": 1180,
      "training_step_time": 10.007439851760864
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 2.4485621452331543,
      "step": 1181
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 1181,
      "training_step_time": 7.468535661697388
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 3.127054452896118,
      "step": 1181
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 1181,
      "training_step_time": 9.887546300888062
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 3.377478837966919,
      "step": 1181
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 1181,
      "training_step_time": 10.14377236366272
    },
    {
      "epoch": 4.505157470703125e-06,
      "model_forward_time": 1.7444219589233398,
      "step": 1181
    },
    {
      "epoch": 4.505157470703125e-06,
      "step": 1181,
      "training_step_time": 10.29667353630066
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 3.1772351264953613,
      "step": 1182
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 1182,
      "training_step_time": 9.944586277008057
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 2.4317495822906494,
      "step": 1182
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 1182,
      "training_step_time": 7.459554433822632
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 3.1290836334228516,
      "step": 1182
    },
    {
      "dynamics_loss_avg": 0.053562499769032004,
      "epoch": 4.50897216796875e-06,
      "step": 1182
    },
    {
      "action_loss_avg": 0.060188906639814375,
      "epoch": 4.50897216796875e-06,
      "step": 1182
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 1182,
      "training_step_time": 9.893248081207275
    },
    {
      "epoch": 4.50897216796875e-06,
      "model_forward_time": 2.6771340370178223,
      "step": 1182
    },
    {
      "epoch": 4.50897216796875e-06,
      "step": 1182,
      "training_step_time": 10.30973768234253
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 2.4541635513305664,
      "step": 1183
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 1183,
      "training_step_time": 7.474133491516113
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 2.4288947582244873,
      "step": 1183
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 1183,
      "training_step_time": 7.444625377655029
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 2.4252495765686035,
      "step": 1183
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 1183,
      "training_step_time": 7.441222667694092
    },
    {
      "epoch": 4.512786865234375e-06,
      "model_forward_time": 3.123271942138672,
      "step": 1183
    },
    {
      "epoch": 4.512786865234375e-06,
      "step": 1183,
      "training_step_time": 10.031834602355957
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 3.171269416809082,
      "step": 1184
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 1184,
      "training_step_time": 9.932777166366577
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 3.3787143230438232,
      "step": 1184
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 1184,
      "training_step_time": 10.142691373825073
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 3.134413242340088,
      "step": 1184
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 1184,
      "training_step_time": 9.90051555633545
    },
    {
      "epoch": 4.5166015625e-06,
      "model_forward_time": 2.4561827182769775,
      "step": 1184
    },
    {
      "epoch": 4.5166015625e-06,
      "step": 1184,
      "training_step_time": 10.274990320205688
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 3.1523191928863525,
      "step": 1185
    },
    {
      "dynamics_loss_avg": 0.047502875328063965,
      "epoch": 4.520416259765625e-06,
      "step": 1185
    },
    {
      "action_loss_avg": 0.046776282414793965,
      "epoch": 4.520416259765625e-06,
      "step": 1185
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 1185,
      "training_step_time": 9.921788454055786
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 3.1321728229522705,
      "step": 1185
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 1185,
      "training_step_time": 9.890845775604248
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 3.424862861633301,
      "step": 1185
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 1185,
      "training_step_time": 10.184079647064209
    },
    {
      "epoch": 4.520416259765625e-06,
      "model_forward_time": 2.428133010864258,
      "step": 1185
    },
    {
      "epoch": 4.520416259765625e-06,
      "step": 1185,
      "training_step_time": 10.015380859375
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 1.7832224369049072,
      "step": 1186
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 1186,
      "training_step_time": 5.238883972167969
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 3.1266324520111084,
      "step": 1186
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 1186,
      "training_step_time": 9.890570640563965
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 3.126753568649292,
      "step": 1186
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 1186,
      "training_step_time": 9.89475154876709
    },
    {
      "epoch": 4.52423095703125e-06,
      "model_forward_time": 2.7057697772979736,
      "step": 1186
    },
    {
      "epoch": 4.52423095703125e-06,
      "step": 1186,
      "training_step_time": 10.019264459609985
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 1.7789263725280762,
      "step": 1187
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 1187,
      "training_step_time": 5.234482049942017
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 2.42380690574646,
      "step": 1187
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 1187,
      "training_step_time": 7.441343307495117
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 3.1204395294189453,
      "step": 1187
    },
    {
      "dynamics_loss_avg": 0.05922940075397491,
      "epoch": 4.528045654296875e-06,
      "step": 1187
    },
    {
      "action_loss_avg": 0.047930471040308474,
      "epoch": 4.528045654296875e-06,
      "step": 1187
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 1187,
      "training_step_time": 9.886839628219604
    },
    {
      "epoch": 4.528045654296875e-06,
      "model_forward_time": 2.427248477935791,
      "step": 1187
    },
    {
      "epoch": 4.528045654296875e-06,
      "step": 1187,
      "training_step_time": 10.311971187591553
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 3.1655783653259277,
      "step": 1188
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 1188,
      "training_step_time": 9.930225133895874
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 1.9912242889404297,
      "step": 1188
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 1188,
      "training_step_time": 5.437873363494873
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 2.424070119857788,
      "step": 1188
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 1188,
      "training_step_time": 7.4457385540008545
    },
    {
      "epoch": 4.5318603515625e-06,
      "model_forward_time": 3.1290619373321533,
      "step": 1188
    },
    {
      "epoch": 4.5318603515625e-06,
      "step": 1188,
      "training_step_time": 10.287841796875
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 3.1715266704559326,
      "step": 1189
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 1189,
      "training_step_time": 9.939249753952026
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 2.4282329082489014,
      "step": 1189
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 1189,
      "training_step_time": 7.454699993133545
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 2.696929693222046,
      "step": 1189
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 1189,
      "training_step_time": 7.718115329742432
    },
    {
      "epoch": 4.535675048828125e-06,
      "model_forward_time": 3.1270081996917725,
      "step": 1189
    },
    {
      "epoch": 4.535675048828125e-06,
      "step": 1189,
      "training_step_time": 10.300177335739136
    },
    {
      "epoch": 4.53948974609375e-06,
      "grad_norm": 0.8726524710655212,
      "learning_rate": 1e-05,
      "loss": 0.1243,
      "step": 1190
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 2.467696189880371,
      "step": 1190
    },
    {
      "dynamics_loss_avg": 0.07414298364892602,
      "epoch": 4.53948974609375e-06,
      "step": 1190
    },
    {
      "action_loss_avg": 0.061290723038837314,
      "epoch": 4.53948974609375e-06,
      "step": 1190
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 1190,
      "training_step_time": 7.494379758834839
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 1.7392957210540771,
      "step": 1190
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 1190,
      "training_step_time": 5.1925177574157715
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 3.1235082149505615,
      "step": 1190
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 1190,
      "training_step_time": 9.876830339431763
    },
    {
      "epoch": 4.53948974609375e-06,
      "model_forward_time": 2.425442695617676,
      "step": 1190
    },
    {
      "epoch": 4.53948974609375e-06,
      "step": 1190,
      "training_step_time": 10.026470422744751
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 2.7153611183166504,
      "step": 1191
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 1191,
      "training_step_time": 7.731216907501221
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 3.1233904361724854,
      "step": 1191
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 1191,
      "training_step_time": 9.87933349609375
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 3.1276400089263916,
      "step": 1191
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 1191,
      "training_step_time": 9.890020370483398
    },
    {
      "epoch": 4.543304443359375e-06,
      "model_forward_time": 3.1274828910827637,
      "step": 1191
    },
    {
      "epoch": 4.543304443359375e-06,
      "step": 1191,
      "training_step_time": 10.02338695526123
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 3.17210054397583,
      "step": 1192
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 1192,
      "training_step_time": 9.942963123321533
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 3.441012144088745,
      "step": 1192
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 1192,
      "training_step_time": 10.20679235458374
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 3.131239652633667,
      "step": 1192
    },
    {
      "dynamics_loss_avg": 0.055468006804585454,
      "epoch": 4.547119140625e-06,
      "step": 1192
    },
    {
      "action_loss_avg": 0.040528439730405805,
      "epoch": 4.547119140625e-06,
      "step": 1192
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 1192,
      "training_step_time": 9.8971107006073
    },
    {
      "epoch": 4.547119140625e-06,
      "model_forward_time": 3.130809783935547,
      "step": 1192
    },
    {
      "epoch": 4.547119140625e-06,
      "step": 1192,
      "training_step_time": 10.295748472213745
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 3.1526682376861572,
      "step": 1193
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 1193,
      "training_step_time": 9.916563272476196
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 1.7410461902618408,
      "step": 1193
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 1193,
      "training_step_time": 5.191493272781372
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 1.7376654148101807,
      "step": 1193
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 1193,
      "training_step_time": 5.19135308265686
    },
    {
      "epoch": 4.550933837890625e-06,
      "model_forward_time": 2.6938438415527344,
      "step": 1193
    },
    {
      "epoch": 4.550933837890625e-06,
      "step": 1193,
      "training_step_time": 10.333855628967285
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 2.442166328430176,
      "step": 1194
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 1194,
      "training_step_time": 7.461889028549194
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 2.423489570617676,
      "step": 1194
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 1194,
      "training_step_time": 7.438823223114014
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 2.424846887588501,
      "step": 1194
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 1194,
      "training_step_time": 7.443997621536255
    },
    {
      "epoch": 4.55474853515625e-06,
      "model_forward_time": 3.126617431640625,
      "step": 1194
    },
    {
      "epoch": 4.55474853515625e-06,
      "step": 1194,
      "training_step_time": 10.264910697937012
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 2.722425937652588,
      "step": 1195
    },
    {
      "dynamics_loss_avg": 0.06822093036025763,
      "epoch": 4.558563232421875e-06,
      "step": 1195
    },
    {
      "action_loss_avg": 0.05394486244767904,
      "epoch": 4.558563232421875e-06,
      "step": 1195
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 1195,
      "training_step_time": 7.743377447128296
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 3.1262054443359375,
      "step": 1195
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 1195,
      "training_step_time": 9.88487458229065
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 3.1248161792755127,
      "step": 1195
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 1195,
      "training_step_time": 9.889129638671875
    },
    {
      "epoch": 4.558563232421875e-06,
      "model_forward_time": 2.4295663833618164,
      "step": 1195
    },
    {
      "epoch": 4.558563232421875e-06,
      "step": 1195,
      "training_step_time": 10.029425144195557
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 2.4538118839263916,
      "step": 1196
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 1196,
      "training_step_time": 7.4791014194488525
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 2.4260411262512207,
      "step": 1196
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 1196,
      "training_step_time": 7.4585466384887695
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 3.3777201175689697,
      "step": 1196
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 1196,
      "training_step_time": 10.144347429275513
    },
    {
      "epoch": 4.5623779296875e-06,
      "model_forward_time": 3.1306827068328857,
      "step": 1196
    },
    {
      "epoch": 4.5623779296875e-06,
      "step": 1196,
      "training_step_time": 10.040322065353394
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 3.1571590900421143,
      "step": 1197
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 1197,
      "training_step_time": 9.924307346343994
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 3.1281447410583496,
      "step": 1197
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 1197,
      "training_step_time": 9.895717144012451
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 1.7404320240020752,
      "step": 1197
    },
    {
      "dynamics_loss_avg": 0.05803997740149498,
      "epoch": 4.566192626953125e-06,
      "step": 1197
    },
    {
      "action_loss_avg": 0.0472641758620739,
      "epoch": 4.566192626953125e-06,
      "step": 1197
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 1197,
      "training_step_time": 5.1976318359375
    },
    {
      "epoch": 4.566192626953125e-06,
      "model_forward_time": 2.71562123298645,
      "step": 1197
    },
    {
      "epoch": 4.566192626953125e-06,
      "step": 1197,
      "training_step_time": 10.016395568847656
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 3.1734094619750977,
      "step": 1198
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 1198,
      "training_step_time": 9.9316565990448
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 2.4293556213378906,
      "step": 1198
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 1198,
      "training_step_time": 7.443991422653198
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 2.4293782711029053,
      "step": 1198
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 1198,
      "training_step_time": 7.443402051925659
    },
    {
      "epoch": 4.57000732421875e-06,
      "model_forward_time": 3.1298933029174805,
      "step": 1198
    },
    {
      "epoch": 4.57000732421875e-06,
      "step": 1198,
      "training_step_time": 10.284692287445068
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 3.441673755645752,
      "step": 1199
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 1199,
      "training_step_time": 10.199676275253296
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 1.7403144836425781,
      "step": 1199
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 1199,
      "training_step_time": 5.185675621032715
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 3.1284584999084473,
      "step": 1199
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 1199,
      "training_step_time": 9.879863500595093
    },
    {
      "epoch": 4.573822021484375e-06,
      "model_forward_time": 3.127227306365967,
      "step": 1199
    },
    {
      "epoch": 4.573822021484375e-06,
      "step": 1199,
      "training_step_time": 10.00515365600586
    },
    {
      "epoch": 4.57763671875e-06,
      "grad_norm": 0.7671394944190979,
      "learning_rate": 1e-05,
      "loss": 0.1206,
      "step": 1200
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 2.4797470569610596,
      "step": 1200
    },
    {
      "dynamics_loss_avg": 0.06760461712256075,
      "epoch": 4.57763671875e-06,
      "step": 1200
    },
    {
      "action_loss_avg": 0.06883605467155576,
      "epoch": 4.57763671875e-06,
      "step": 1200
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 1200,
      "training_step_time": 7.4976067543029785
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 3.1305227279663086,
      "step": 1200
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 1200,
      "training_step_time": 9.882827281951904
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 3.4129345417022705,
      "step": 1200
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 1200,
      "training_step_time": 10.176192283630371
    },
    {
      "epoch": 4.57763671875e-06,
      "model_forward_time": 3.134615421295166,
      "step": 1200
    },
    {
      "epoch": 4.57763671875e-06,
      "step": 1200,
      "training_step_time": 10.239125728607178
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 3.1733169555664062,
      "step": 1201
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 1201,
      "training_step_time": 9.936424970626831
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 3.143498659133911,
      "step": 1201
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 1201,
      "training_step_time": 9.908880949020386
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 3.132967233657837,
      "step": 1201
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 1201,
      "training_step_time": 10.191078186035156
    },
    {
      "epoch": 4.581451416015625e-06,
      "model_forward_time": 3.134282112121582,
      "step": 1201
    },
    {
      "epoch": 4.581451416015625e-06,
      "step": 1201,
      "training_step_time": 10.022000789642334
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 3.175420045852661,
      "step": 1202
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 1202,
      "training_step_time": 9.93996548652649
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 3.13736629486084,
      "step": 1202
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 1202,
      "training_step_time": 9.899333238601685
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 3.139906167984009,
      "step": 1202
    },
    {
      "dynamics_loss_avg": 0.08114126399159431,
      "epoch": 4.58526611328125e-06,
      "step": 1202
    },
    {
      "action_loss_avg": 0.0587464690208435,
      "epoch": 4.58526611328125e-06,
      "step": 1202
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 1202,
      "training_step_time": 9.909119844436646
    },
    {
      "epoch": 4.58526611328125e-06,
      "model_forward_time": 2.437558650970459,
      "step": 1202
    },
    {
      "epoch": 4.58526611328125e-06,
      "step": 1202,
      "training_step_time": 10.247634172439575
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 3.1596338748931885,
      "step": 1203
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 1203,
      "training_step_time": 9.929457187652588
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 3.142263174057007,
      "step": 1203
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 1203,
      "training_step_time": 9.912960290908813
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 3.1354172229766846,
      "step": 1203
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 1203,
      "training_step_time": 9.909087896347046
    },
    {
      "epoch": 4.589080810546875e-06,
      "model_forward_time": 2.4363880157470703,
      "step": 1203
    },
    {
      "epoch": 4.589080810546875e-06,
      "step": 1203,
      "training_step_time": 10.335885286331177
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 3.142200231552124,
      "step": 1204
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 1204,
      "training_step_time": 10.186828374862671
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 3.1332449913024902,
      "step": 1204
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 1204,
      "training_step_time": 9.901352643966675
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 2.4294745922088623,
      "step": 1204
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 1204,
      "training_step_time": 7.455903768539429
    },
    {
      "epoch": 4.5928955078125e-06,
      "model_forward_time": 2.4281909465789795,
      "step": 1204
    },
    {
      "epoch": 4.5928955078125e-06,
      "step": 1204,
      "training_step_time": 10.364513874053955
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 2.4439871311187744,
      "step": 1205
    },
    {
      "dynamics_loss_avg": 0.07207853980362415,
      "epoch": 4.596710205078125e-06,
      "step": 1205
    },
    {
      "action_loss_avg": 0.04141898788511753,
      "epoch": 4.596710205078125e-06,
      "step": 1205
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 1205,
      "training_step_time": 7.4659905433654785
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 2.423387050628662,
      "step": 1205
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 1205,
      "training_step_time": 7.440216064453125
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 3.4101955890655518,
      "step": 1205
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 1205,
      "training_step_time": 10.168831586837769
    },
    {
      "epoch": 4.596710205078125e-06,
      "model_forward_time": 2.4245760440826416,
      "step": 1205
    },
    {
      "epoch": 4.596710205078125e-06,
      "step": 1205,
      "training_step_time": 10.240914344787598
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 3.146465539932251,
      "step": 1206
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 1206,
      "training_step_time": 9.911277532577515
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 2.4290175437927246,
      "step": 1206
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 1206,
      "training_step_time": 7.447722673416138
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 2.422232151031494,
      "step": 1206
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 1206,
      "training_step_time": 7.44443416595459
    },
    {
      "epoch": 4.60052490234375e-06,
      "model_forward_time": 2.6738667488098145,
      "step": 1206
    },
    {
      "epoch": 4.60052490234375e-06,
      "step": 1206,
      "training_step_time": 10.20976972579956
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 2.4485955238342285,
      "step": 1207
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 1207,
      "training_step_time": 7.478171110153198
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 2.42637038230896,
      "step": 1207
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 1207,
      "training_step_time": 7.44939398765564
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 1.7415761947631836,
      "step": 1207
    },
    {
      "dynamics_loss_avg": 0.06081716269254685,
      "epoch": 4.604339599609375e-06,
      "step": 1207
    },
    {
      "action_loss_avg": 0.0599416540004313,
      "epoch": 4.604339599609375e-06,
      "step": 1207
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 1207,
      "training_step_time": 5.196486473083496
    },
    {
      "epoch": 4.604339599609375e-06,
      "model_forward_time": 3.1278066635131836,
      "step": 1207
    },
    {
      "epoch": 4.604339599609375e-06,
      "step": 1207,
      "training_step_time": 10.03038477897644
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 2.445617914199829,
      "step": 1208
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 1208,
      "training_step_time": 7.4693052768707275
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 3.4059031009674072,
      "step": 1208
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 1208,
      "training_step_time": 10.168951511383057
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 2.4287495613098145,
      "step": 1208
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 1208,
      "training_step_time": 7.450531959533691
    },
    {
      "epoch": 4.608154296875e-06,
      "model_forward_time": 2.4290337562561035,
      "step": 1208
    },
    {
      "epoch": 4.608154296875e-06,
      "step": 1208,
      "training_step_time": 10.030329704284668
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 2.4553334712982178,
      "step": 1209
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 1209,
      "training_step_time": 7.483301162719727
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 3.1275343894958496,
      "step": 1209
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 1209,
      "training_step_time": 9.898251295089722
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 3.3980720043182373,
      "step": 1209
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 1209,
      "training_step_time": 10.16475248336792
    },
    {
      "epoch": 4.611968994140625e-06,
      "model_forward_time": 1.745384693145752,
      "step": 1209
    },
    {
      "epoch": 4.611968994140625e-06,
      "step": 1209,
      "training_step_time": 10.330588579177856
    },
    {
      "epoch": 4.61578369140625e-06,
      "grad_norm": 0.5189809203147888,
      "learning_rate": 1e-05,
      "loss": 0.1184,
      "step": 1210
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 3.153141975402832,
      "step": 1210
    },
    {
      "dynamics_loss_avg": 0.05972112976014614,
      "epoch": 4.61578369140625e-06,
      "step": 1210
    },
    {
      "action_loss_avg": 0.045745207369327544,
      "epoch": 4.61578369140625e-06,
      "step": 1210
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 1210,
      "training_step_time": 9.918523073196411
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 3.1290442943573,
      "step": 1210
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 1210,
      "training_step_time": 9.889006614685059
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 1.7428314685821533,
      "step": 1210
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 1210,
      "training_step_time": 5.192468881607056
    },
    {
      "epoch": 4.61578369140625e-06,
      "model_forward_time": 2.4256999492645264,
      "step": 1210
    },
    {
      "epoch": 4.61578369140625e-06,
      "step": 1210,
      "training_step_time": 10.29564094543457
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 2.7107176780700684,
      "step": 1211
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 1211,
      "training_step_time": 7.7216315269470215
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 1.7333014011383057,
      "step": 1211
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 1211,
      "training_step_time": 5.176829814910889
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 2.4205830097198486,
      "step": 1211
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 1211,
      "training_step_time": 7.426856994628906
    },
    {
      "epoch": 4.619598388671875e-06,
      "model_forward_time": 3.1250925064086914,
      "step": 1211
    },
    {
      "epoch": 4.619598388671875e-06,
      "step": 1211,
      "training_step_time": 10.006860971450806
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 2.4766008853912354,
      "step": 1212
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 1212,
      "training_step_time": 7.489325761795044
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 3.130075216293335,
      "step": 1212
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 1212,
      "training_step_time": 9.882254123687744
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 3.396920919418335,
      "step": 1212
    },
    {
      "dynamics_loss_avg": 0.06317014433443546,
      "epoch": 4.6234130859375e-06,
      "step": 1212
    },
    {
      "action_loss_avg": 0.06574880564585328,
      "epoch": 4.6234130859375e-06,
      "step": 1212
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 1212,
      "training_step_time": 10.160531044006348
    },
    {
      "epoch": 4.6234130859375e-06,
      "model_forward_time": 3.1257688999176025,
      "step": 1212
    },
    {
      "epoch": 4.6234130859375e-06,
      "step": 1212,
      "training_step_time": 10.010311126708984
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 3.1441211700439453,
      "step": 1213
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 1213,
      "training_step_time": 9.91246509552002
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 3.134385347366333,
      "step": 1213
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 1213,
      "training_step_time": 9.904959201812744
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 2.431351900100708,
      "step": 1213
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 1213,
      "training_step_time": 7.45599889755249
    },
    {
      "epoch": 4.627227783203125e-06,
      "model_forward_time": 2.004923105239868,
      "step": 1213
    },
    {
      "epoch": 4.627227783203125e-06,
      "step": 1213,
      "training_step_time": 10.022650241851807
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 2.4733927249908447,
      "step": 1214
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 1214,
      "training_step_time": 7.496365785598755
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 3.131169557571411,
      "step": 1214
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 1214,
      "training_step_time": 9.889351844787598
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 2.4311723709106445,
      "step": 1214
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 1214,
      "training_step_time": 7.449084997177124
    },
    {
      "epoch": 4.63104248046875e-06,
      "model_forward_time": 3.1307883262634277,
      "step": 1214
    },
    {
      "epoch": 4.63104248046875e-06,
      "step": 1214,
      "training_step_time": 10.31764006614685
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 3.4487802982330322,
      "step": 1215
    },
    {
      "dynamics_loss_avg": 0.06907507926225662,
      "epoch": 4.634857177734375e-06,
      "step": 1215
    },
    {
      "action_loss_avg": 0.07108589205890894,
      "epoch": 4.634857177734375e-06,
      "step": 1215
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 1215,
      "training_step_time": 10.212255954742432
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 2.4306092262268066,
      "step": 1215
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 1215,
      "training_step_time": 7.446545362472534
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 3.128098487854004,
      "step": 1215
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 1215,
      "training_step_time": 9.891504049301147
    },
    {
      "epoch": 4.634857177734375e-06,
      "model_forward_time": 3.1366963386535645,
      "step": 1215
    },
    {
      "epoch": 4.634857177734375e-06,
      "step": 1215,
      "training_step_time": 10.299174070358276
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 2.4733469486236572,
      "step": 1216
    },
    {
      "epoch": 4.638671875e-06,
      "step": 1216,
      "training_step_time": 7.494321584701538
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 2.4270851612091064,
      "step": 1216
    },
    {
      "epoch": 4.638671875e-06,
      "step": 1216,
      "training_step_time": 7.4455249309539795
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 3.3795416355133057,
      "step": 1216
    },
    {
      "epoch": 4.638671875e-06,
      "step": 1216,
      "training_step_time": 10.138485193252563
    },
    {
      "epoch": 4.638671875e-06,
      "model_forward_time": 3.133488893508911,
      "step": 1216
    },
    {
      "epoch": 4.638671875e-06,
      "step": 1216,
      "training_step_time": 10.005188465118408
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 1.7561044692993164,
      "step": 1217
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 1217,
      "training_step_time": 5.207295656204224
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 3.121025562286377,
      "step": 1217
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 1217,
      "training_step_time": 9.87583065032959
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 1.7359097003936768,
      "step": 1217
    },
    {
      "dynamics_loss_avg": 0.06339764557778835,
      "epoch": 4.642486572265625e-06,
      "step": 1217
    },
    {
      "action_loss_avg": 0.047196164913475516,
      "epoch": 4.642486572265625e-06,
      "step": 1217
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 1217,
      "training_step_time": 5.184097051620483
    },
    {
      "epoch": 4.642486572265625e-06,
      "model_forward_time": 3.3941822052001953,
      "step": 1217
    },
    {
      "epoch": 4.642486572265625e-06,
      "step": 1217,
      "training_step_time": 10.23996114730835
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 2.4372544288635254,
      "step": 1218
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 1218,
      "training_step_time": 7.449633836746216
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 3.1238152980804443,
      "step": 1218
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 1218,
      "training_step_time": 9.881931781768799
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 2.4255831241607666,
      "step": 1218
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 1218,
      "training_step_time": 7.451327323913574
    },
    {
      "epoch": 4.64630126953125e-06,
      "model_forward_time": 2.429553985595703,
      "step": 1218
    },
    {
      "epoch": 4.64630126953125e-06,
      "step": 1218,
      "training_step_time": 10.02231764793396
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 3.428659200668335,
      "step": 1219
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 1219,
      "training_step_time": 10.191235542297363
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 2.427049160003662,
      "step": 1219
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 1219,
      "training_step_time": 7.445513725280762
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 1.7403626441955566,
      "step": 1219
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 1219,
      "training_step_time": 5.190890073776245
    },
    {
      "epoch": 4.650115966796875e-06,
      "model_forward_time": 3.123793363571167,
      "step": 1219
    },
    {
      "epoch": 4.650115966796875e-06,
      "step": 1219,
      "training_step_time": 10.266523361206055
    },
    {
      "epoch": 4.6539306640625e-06,
      "grad_norm": 0.49649009108543396,
      "learning_rate": 1e-05,
      "loss": 0.1225,
      "step": 1220
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 2.440702199935913,
      "step": 1220
    },
    {
      "dynamics_loss_avg": 0.0673874905332923,
      "epoch": 4.6539306640625e-06,
      "step": 1220
    },
    {
      "action_loss_avg": 0.03754705637693405,
      "epoch": 4.6539306640625e-06,
      "step": 1220
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 1220,
      "training_step_time": 7.456476926803589
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 3.12605357170105,
      "step": 1220
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 1220,
      "training_step_time": 9.877912282943726
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 3.417112112045288,
      "step": 1220
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 1220,
      "training_step_time": 10.173411846160889
    },
    {
      "epoch": 4.6539306640625e-06,
      "model_forward_time": 1.7357847690582275,
      "step": 1220
    },
    {
      "epoch": 4.6539306640625e-06,
      "step": 1220,
      "training_step_time": 10.312682867050171
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 3.135876417160034,
      "step": 1221
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 1221,
      "training_step_time": 9.889820337295532
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 2.423877477645874,
      "step": 1221
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 1221,
      "training_step_time": 7.433595895767212
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 2.4224491119384766,
      "step": 1221
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 1221,
      "training_step_time": 7.437139272689819
    },
    {
      "epoch": 4.657745361328125e-06,
      "model_forward_time": 1.7355921268463135,
      "step": 1221
    },
    {
      "epoch": 4.657745361328125e-06,
      "step": 1221,
      "training_step_time": 10.014120101928711
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 2.7729361057281494,
      "step": 1222
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 1222,
      "training_step_time": 7.787886142730713
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 3.1275758743286133,
      "step": 1222
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 1222,
      "training_step_time": 9.88534951210022
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 2.4274604320526123,
      "step": 1222
    },
    {
      "dynamics_loss_avg": 0.0690448958426714,
      "epoch": 4.66156005859375e-06,
      "step": 1222
    },
    {
      "action_loss_avg": 0.056548270303756,
      "epoch": 4.66156005859375e-06,
      "step": 1222
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 1222,
      "training_step_time": 7.4518349170684814
    },
    {
      "epoch": 4.66156005859375e-06,
      "model_forward_time": 3.1272361278533936,
      "step": 1222
    },
    {
      "epoch": 4.66156005859375e-06,
      "step": 1222,
      "training_step_time": 10.035603523254395
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 2.4454543590545654,
      "step": 1223
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 1223,
      "training_step_time": 7.471389532089233
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 3.4069271087646484,
      "step": 1223
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 1223,
      "training_step_time": 10.170527458190918
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 3.1350278854370117,
      "step": 1223
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 1223,
      "training_step_time": 9.902215957641602
    },
    {
      "epoch": 4.665374755859375e-06,
      "model_forward_time": 2.433013916015625,
      "step": 1223
    },
    {
      "epoch": 4.665374755859375e-06,
      "step": 1223,
      "training_step_time": 10.023584604263306
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 2.444746494293213,
      "step": 1224
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 1224,
      "training_step_time": 7.46983528137207
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 3.128328561782837,
      "step": 1224
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 1224,
      "training_step_time": 9.900568008422852
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 3.3925132751464844,
      "step": 1224
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 1224,
      "training_step_time": 10.16784930229187
    },
    {
      "epoch": 4.669189453125e-06,
      "model_forward_time": 2.434954881668091,
      "step": 1224
    },
    {
      "epoch": 4.669189453125e-06,
      "step": 1224,
      "training_step_time": 10.316181898117065
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 3.146517276763916,
      "step": 1225
    },
    {
      "dynamics_loss_avg": 0.0778695710003376,
      "epoch": 4.673004150390625e-06,
      "step": 1225
    },
    {
      "action_loss_avg": 0.06462602838873863,
      "epoch": 4.673004150390625e-06,
      "step": 1225
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 1225,
      "training_step_time": 9.9164137840271
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 2.4327824115753174,
      "step": 1225
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 1225,
      "training_step_time": 7.456658363342285
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 3.135624647140503,
      "step": 1225
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 1225,
      "training_step_time": 9.898427724838257
    },
    {
      "epoch": 4.673004150390625e-06,
      "model_forward_time": 2.4279699325561523,
      "step": 1225
    },
    {
      "epoch": 4.673004150390625e-06,
      "step": 1225,
      "training_step_time": 10.29494833946228
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 3.4809634685516357,
      "step": 1226
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 1226,
      "training_step_time": 10.243354558944702
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 3.1358978748321533,
      "step": 1226
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 1226,
      "training_step_time": 9.899722576141357
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 3.1401333808898926,
      "step": 1226
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 1226,
      "training_step_time": 9.9080650806427
    },
    {
      "epoch": 4.67681884765625e-06,
      "model_forward_time": 3.137305498123169,
      "step": 1226
    },
    {
      "epoch": 4.67681884765625e-06,
      "step": 1226,
      "training_step_time": 10.345723390579224
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 3.1494390964508057,
      "step": 1227
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 1227,
      "training_step_time": 9.920228004455566
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 2.7355895042419434,
      "step": 1227
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 1227,
      "training_step_time": 7.762385129928589
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 2.437052011489868,
      "step": 1227
    },
    {
      "dynamics_loss_avg": 0.05833753459155559,
      "epoch": 4.680633544921875e-06,
      "step": 1227
    },
    {
      "action_loss_avg": 0.05272754598408937,
      "epoch": 4.680633544921875e-06,
      "step": 1227
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 1227,
      "training_step_time": 7.461354494094849
    },
    {
      "epoch": 4.680633544921875e-06,
      "model_forward_time": 3.137144088745117,
      "step": 1227
    },
    {
      "epoch": 4.680633544921875e-06,
      "step": 1227,
      "training_step_time": 10.004535675048828
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 2.4871840476989746,
      "step": 1228
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 1228,
      "training_step_time": 7.509801864624023
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 2.4295830726623535,
      "step": 1228
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 1228,
      "training_step_time": 7.447698593139648
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 3.418213367462158,
      "step": 1228
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 1228,
      "training_step_time": 10.180627584457397
    },
    {
      "epoch": 4.6844482421875e-06,
      "model_forward_time": 3.134643316268921,
      "step": 1228
    },
    {
      "epoch": 4.6844482421875e-06,
      "step": 1228,
      "training_step_time": 10.00997805595398
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 2.4490556716918945,
      "step": 1229
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 1229,
      "training_step_time": 7.4704296588897705
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 2.4343316555023193,
      "step": 1229
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 1229,
      "training_step_time": 7.4537224769592285
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 1.7450029850006104,
      "step": 1229
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 1229,
      "training_step_time": 5.193267583847046
    },
    {
      "epoch": 4.688262939453125e-06,
      "model_forward_time": 2.4343883991241455,
      "step": 1229
    },
    {
      "epoch": 4.688262939453125e-06,
      "step": 1229,
      "training_step_time": 10.278048038482666
    },
    {
      "epoch": 4.69207763671875e-06,
      "grad_norm": 0.5527411699295044,
      "learning_rate": 1e-05,
      "loss": 0.1246,
      "step": 1230
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 2.053391456604004,
      "step": 1230
    },
    {
      "dynamics_loss_avg": 0.06983247585594654,
      "epoch": 4.69207763671875e-06,
      "step": 1230
    },
    {
      "action_loss_avg": 0.06774530168622732,
      "epoch": 4.69207763671875e-06,
      "step": 1230
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 1230,
      "training_step_time": 5.5076916217803955
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 3.134505033493042,
      "step": 1230
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 1230,
      "training_step_time": 9.896850347518921
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 1.7433264255523682,
      "step": 1230
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 1230,
      "training_step_time": 5.1948325634002686
    },
    {
      "epoch": 4.69207763671875e-06,
      "model_forward_time": 2.4323184490203857,
      "step": 1230
    },
    {
      "epoch": 4.69207763671875e-06,
      "step": 1230,
      "training_step_time": 10.276743412017822
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 3.1472716331481934,
      "step": 1231
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 1231,
      "training_step_time": 9.910006284713745
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 2.4324324131011963,
      "step": 1231
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 1231,
      "training_step_time": 7.724279403686523
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 3.13541316986084,
      "step": 1231
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 1231,
      "training_step_time": 9.898443222045898
    },
    {
      "epoch": 4.695892333984375e-06,
      "model_forward_time": 2.4379563331604004,
      "step": 1231
    },
    {
      "epoch": 4.695892333984375e-06,
      "step": 1231,
      "training_step_time": 10.29025912284851
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 2.4497904777526855,
      "step": 1232
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 1232,
      "training_step_time": 7.474967956542969
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 1.7398605346679688,
      "step": 1232
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 1232,
      "training_step_time": 5.190002679824829
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 1.740260362625122,
      "step": 1232
    },
    {
      "dynamics_loss_avg": 0.06323644332587719,
      "epoch": 4.69970703125e-06,
      "step": 1232
    },
    {
      "action_loss_avg": 0.04773163958452642,
      "epoch": 4.69970703125e-06,
      "step": 1232
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 1232,
      "training_step_time": 5.189902305603027
    },
    {
      "epoch": 4.69970703125e-06,
      "model_forward_time": 2.0644659996032715,
      "step": 1232
    },
    {
      "epoch": 4.69970703125e-06,
      "step": 1232,
      "training_step_time": 10.025347471237183
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 3.180323600769043,
      "step": 1233
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 1233,
      "training_step_time": 9.939695596694946
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 2.433318853378296,
      "step": 1233
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 1233,
      "training_step_time": 7.446989059448242
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 2.427412509918213,
      "step": 1233
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 1233,
      "training_step_time": 7.442462921142578
    },
    {
      "epoch": 4.703521728515625e-06,
      "model_forward_time": 3.1308746337890625,
      "step": 1233
    },
    {
      "epoch": 4.703521728515625e-06,
      "step": 1233,
      "training_step_time": 10.017304182052612
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 3.1457855701446533,
      "step": 1234
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 1234,
      "training_step_time": 9.906521320343018
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 2.7274937629699707,
      "step": 1234
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 1234,
      "training_step_time": 7.744112730026245
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 3.1357245445251465,
      "step": 1234
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 1234,
      "training_step_time": 9.890328407287598
    },
    {
      "epoch": 4.70733642578125e-06,
      "model_forward_time": 2.4315221309661865,
      "step": 1234
    },
    {
      "epoch": 4.70733642578125e-06,
      "step": 1234,
      "training_step_time": 10.272552490234375
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 3.168466806411743,
      "step": 1235
    },
    {
      "dynamics_loss_avg": 0.06218680711463094,
      "epoch": 4.711151123046875e-06,
      "step": 1235
    },
    {
      "action_loss_avg": 0.04065850037150085,
      "epoch": 4.711151123046875e-06,
      "step": 1235
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 1235,
      "training_step_time": 9.929354906082153
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 3.1339025497436523,
      "step": 1235
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 1235,
      "training_step_time": 9.888962984085083
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 3.406001091003418,
      "step": 1235
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 1235,
      "training_step_time": 10.16056227684021
    },
    {
      "epoch": 4.711151123046875e-06,
      "model_forward_time": 3.13407301902771,
      "step": 1235
    },
    {
      "epoch": 4.711151123046875e-06,
      "step": 1235,
      "training_step_time": 10.26656699180603
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 3.1704764366149902,
      "step": 1236
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 1236,
      "training_step_time": 9.92806887626648
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 2.4335834980010986,
      "step": 1236
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 1236,
      "training_step_time": 7.454269886016846
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 3.1307313442230225,
      "step": 1236
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 1236,
      "training_step_time": 9.894171237945557
    },
    {
      "epoch": 4.7149658203125e-06,
      "model_forward_time": 3.4254462718963623,
      "step": 1236
    },
    {
      "epoch": 4.7149658203125e-06,
      "step": 1236,
      "training_step_time": 10.32539963722229
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 2.4564201831817627,
      "step": 1237
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 1237,
      "training_step_time": 7.482840061187744
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 3.1462502479553223,
      "step": 1237
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 1237,
      "training_step_time": 9.918205261230469
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 2.437342643737793,
      "step": 1237
    },
    {
      "dynamics_loss_avg": 0.07237791940569878,
      "epoch": 4.718780517578125e-06,
      "step": 1237
    },
    {
      "action_loss_avg": 0.04849719516932964,
      "epoch": 4.718780517578125e-06,
      "step": 1237
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 1237,
      "training_step_time": 7.469151496887207
    },
    {
      "epoch": 4.718780517578125e-06,
      "model_forward_time": 2.4383466243743896,
      "step": 1237
    },
    {
      "epoch": 4.718780517578125e-06,
      "step": 1237,
      "training_step_time": 10.313759565353394
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 2.7427444458007812,
      "step": 1238
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 1238,
      "training_step_time": 7.765231132507324
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 2.4354915618896484,
      "step": 1238
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 1238,
      "training_step_time": 7.4538702964782715
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 3.1375415325164795,
      "step": 1238
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 1238,
      "training_step_time": 9.901054382324219
    },
    {
      "epoch": 4.72259521484375e-06,
      "model_forward_time": 3.1374456882476807,
      "step": 1238
    },
    {
      "epoch": 4.72259521484375e-06,
      "step": 1238,
      "training_step_time": 10.016731262207031
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 2.4728760719299316,
      "step": 1239
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 1239,
      "training_step_time": 7.491472482681274
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 3.4216697216033936,
      "step": 1239
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 1239,
      "training_step_time": 10.180582761764526
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 2.4353928565979004,
      "step": 1239
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 1239,
      "training_step_time": 7.452328205108643
    },
    {
      "epoch": 4.726409912109375e-06,
      "model_forward_time": 3.1372318267822266,
      "step": 1239
    },
    {
      "epoch": 4.726409912109375e-06,
      "step": 1239,
      "training_step_time": 10.291048765182495
    },
    {
      "epoch": 4.730224609375e-06,
      "grad_norm": 0.6280481815338135,
      "learning_rate": 1e-05,
      "loss": 0.1212,
      "step": 1240
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 2.475919246673584,
      "step": 1240
    },
    {
      "dynamics_loss_avg": 0.07849410958588124,
      "epoch": 4.730224609375e-06,
      "step": 1240
    },
    {
      "action_loss_avg": 0.05823127701878548,
      "epoch": 4.730224609375e-06,
      "step": 1240
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 1240,
      "training_step_time": 7.503226041793823
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 2.4229753017425537,
      "step": 1240
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 1240,
      "training_step_time": 7.442317008972168
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 3.1272833347320557,
      "step": 1240
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 1240,
      "training_step_time": 9.890080451965332
    },
    {
      "epoch": 4.730224609375e-06,
      "model_forward_time": 2.697401285171509,
      "step": 1240
    },
    {
      "epoch": 4.730224609375e-06,
      "step": 1240,
      "training_step_time": 10.015192031860352
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 3.165088176727295,
      "step": 1241
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 1241,
      "training_step_time": 9.92839527130127
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 3.126767635345459,
      "step": 1241
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 1241,
      "training_step_time": 9.889939069747925
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 3.133967876434326,
      "step": 1241
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 1241,
      "training_step_time": 9.896329164505005
    },
    {
      "epoch": 4.734039306640625e-06,
      "model_forward_time": 3.126568078994751,
      "step": 1241
    },
    {
      "epoch": 4.734039306640625e-06,
      "step": 1241,
      "training_step_time": 10.278855323791504
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 2.070279598236084,
      "step": 1242
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 1242,
      "training_step_time": 5.523383855819702
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 3.123866081237793,
      "step": 1242
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 1242,
      "training_step_time": 9.887449741363525
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 3.1366031169891357,
      "step": 1242
    },
    {
      "dynamics_loss_avg": 0.06646110657602548,
      "epoch": 4.73785400390625e-06,
      "step": 1242
    },
    {
      "action_loss_avg": 0.044405975844711065,
      "epoch": 4.73785400390625e-06,
      "step": 1242
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 1242,
      "training_step_time": 9.905350923538208
    },
    {
      "epoch": 4.73785400390625e-06,
      "model_forward_time": 1.7430377006530762,
      "step": 1242
    },
    {
      "epoch": 4.73785400390625e-06,
      "step": 1242,
      "training_step_time": 10.261742353439331
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 2.4435086250305176,
      "step": 1243
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 1243,
      "training_step_time": 7.461638450622559
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 2.422278881072998,
      "step": 1243
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 1243,
      "training_step_time": 7.438884258270264
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 3.4027812480926514,
      "step": 1243
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 1243,
      "training_step_time": 10.158737897872925
    },
    {
      "epoch": 4.741668701171875e-06,
      "model_forward_time": 2.4259183406829834,
      "step": 1243
    },
    {
      "epoch": 4.741668701171875e-06,
      "step": 1243,
      "training_step_time": 10.013561725616455
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 2.4407618045806885,
      "step": 1244
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 1244,
      "training_step_time": 7.457063674926758
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 3.1354448795318604,
      "step": 1244
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 1244,
      "training_step_time": 9.8887357711792
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 2.4287960529327393,
      "step": 1244
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 1244,
      "training_step_time": 7.439752817153931
    },
    {
      "epoch": 4.7454833984375e-06,
      "model_forward_time": 2.6845898628234863,
      "step": 1244
    },
    {
      "epoch": 4.7454833984375e-06,
      "step": 1244,
      "training_step_time": 10.001758098602295
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 3.156752109527588,
      "step": 1245
    },
    {
      "dynamics_loss_avg": 0.06779483929276467,
      "epoch": 4.749298095703125e-06,
      "step": 1245
    },
    {
      "action_loss_avg": 0.05374050997197628,
      "epoch": 4.749298095703125e-06,
      "step": 1245
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 1245,
      "training_step_time": 9.91728663444519
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 3.1347527503967285,
      "step": 1245
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 1245,
      "training_step_time": 9.894219160079956
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 2.4322855472564697,
      "step": 1245
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 1245,
      "training_step_time": 7.450615406036377
    },
    {
      "epoch": 4.749298095703125e-06,
      "model_forward_time": 1.7428631782531738,
      "step": 1245
    },
    {
      "epoch": 4.749298095703125e-06,
      "step": 1245,
      "training_step_time": 10.271838903427124
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 3.4041941165924072,
      "step": 1246
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 1246,
      "training_step_time": 10.16143536567688
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 2.4330272674560547,
      "step": 1246
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 1246,
      "training_step_time": 7.454505443572998
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 3.1351449489593506,
      "step": 1246
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 1246,
      "training_step_time": 9.897195100784302
    },
    {
      "epoch": 4.75311279296875e-06,
      "model_forward_time": 2.4327239990234375,
      "step": 1246
    },
    {
      "epoch": 4.75311279296875e-06,
      "step": 1246,
      "training_step_time": 10.315007448196411
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 2.446690559387207,
      "step": 1247
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 1247,
      "training_step_time": 7.466123342514038
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 2.4313178062438965,
      "step": 1247
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 1247,
      "training_step_time": 7.450031518936157
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 2.7432727813720703,
      "step": 1247
    },
    {
      "dynamics_loss_avg": 0.06792859435081482,
      "epoch": 4.756927490234375e-06,
      "step": 1247
    },
    {
      "action_loss_avg": 0.04706329358741641,
      "epoch": 4.756927490234375e-06,
      "step": 1247
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 1247,
      "training_step_time": 7.764781951904297
    },
    {
      "epoch": 4.756927490234375e-06,
      "model_forward_time": 3.136120557785034,
      "step": 1247
    },
    {
      "epoch": 4.756927490234375e-06,
      "step": 1247,
      "training_step_time": 10.340077638626099
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 3.1544313430786133,
      "step": 1248
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 1248,
      "training_step_time": 9.92519211769104
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 3.1398236751556396,
      "step": 1248
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 1248,
      "training_step_time": 9.913113832473755
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 2.4358620643615723,
      "step": 1248
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 1248,
      "training_step_time": 7.465389728546143
    },
    {
      "epoch": 4.7607421875e-06,
      "model_forward_time": 2.4349582195281982,
      "step": 1248
    },
    {
      "epoch": 4.7607421875e-06,
      "step": 1248,
      "training_step_time": 10.308199644088745
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 3.1908328533172607,
      "step": 1249
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 1249,
      "training_step_time": 9.962080001831055
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 3.128361225128174,
      "step": 1249
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 1249,
      "training_step_time": 9.892926216125488
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 3.130863904953003,
      "step": 1249
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 1249,
      "training_step_time": 9.895998001098633
    },
    {
      "epoch": 4.764556884765625e-06,
      "model_forward_time": 2.429955244064331,
      "step": 1249
    },
    {
      "epoch": 4.764556884765625e-06,
      "step": 1249,
      "training_step_time": 10.005913734436035
    },
    {
      "epoch": 4.76837158203125e-06,
      "grad_norm": 0.8789255023002625,
      "learning_rate": 1e-05,
      "loss": 0.1265,
      "step": 1250
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 3.1459641456604004,
      "step": 1250
    },
    {
      "dynamics_loss_avg": 0.07437640465795994,
      "epoch": 4.76837158203125e-06,
      "step": 1250
    },
    {
      "action_loss_avg": 0.05410216245800257,
      "epoch": 4.76837158203125e-06,
      "step": 1250
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 1250,
      "training_step_time": 9.914058685302734
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 3.3671069145202637,
      "step": 1250
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 1250,
      "training_step_time": 10.135427713394165
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 3.1286892890930176,
      "step": 1250
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 1250,
      "training_step_time": 9.898169040679932
    },
    {
      "epoch": 4.76837158203125e-06,
      "model_forward_time": 3.125441312789917,
      "step": 1250
    },
    {
      "epoch": 4.76837158203125e-06,
      "step": 1250,
      "training_step_time": 10.006916761398315
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 3.17630934715271,
      "step": 1251
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 1251,
      "training_step_time": 9.95698094367981
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 3.140861749649048,
      "step": 1251
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 1251,
      "training_step_time": 10.236485004425049
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 2.4346706867218018,
      "step": 1251
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 1251,
      "training_step_time": 7.469367027282715
    },
    {
      "epoch": 4.772186279296875e-06,
      "model_forward_time": 3.1318604946136475,
      "step": 1251
    },
    {
      "epoch": 4.772186279296875e-06,
      "step": 1251,
      "training_step_time": 10.316200494766235
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 2.4771664142608643,
      "step": 1252
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 1252,
      "training_step_time": 7.522017240524292
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 3.1371803283691406,
      "step": 1252
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 1252,
      "training_step_time": 9.914987802505493
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 3.1381213665008545,
      "step": 1252
    },
    {
      "dynamics_loss_avg": 0.08197107762098313,
      "epoch": 4.7760009765625e-06,
      "step": 1252
    },
    {
      "action_loss_avg": 0.05294404439628124,
      "epoch": 4.7760009765625e-06,
      "step": 1252
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 1252,
      "training_step_time": 9.922937870025635
    },
    {
      "epoch": 4.7760009765625e-06,
      "model_forward_time": 3.384451150894165,
      "step": 1252
    },
    {
      "epoch": 4.7760009765625e-06,
      "step": 1252,
      "training_step_time": 10.258341073989868
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 3.1834516525268555,
      "step": 1253
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 1253,
      "training_step_time": 9.96206784248352
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 3.13637638092041,
      "step": 1253
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 1253,
      "training_step_time": 9.909493446350098
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 3.132202625274658,
      "step": 1253
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 1253,
      "training_step_time": 9.910678625106812
    },
    {
      "epoch": 4.779815673828125e-06,
      "model_forward_time": 2.430664300918579,
      "step": 1253
    },
    {
      "epoch": 4.779815673828125e-06,
      "step": 1253,
      "training_step_time": 10.299079895019531
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 3.4767391681671143,
      "step": 1254
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 1254,
      "training_step_time": 10.25171685218811
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 3.1360812187194824,
      "step": 1254
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 1254,
      "training_step_time": 9.911577939987183
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 3.1379802227020264,
      "step": 1254
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 1254,
      "training_step_time": 9.920092582702637
    },
    {
      "epoch": 4.78363037109375e-06,
      "model_forward_time": 3.1444015502929688,
      "step": 1254
    },
    {
      "epoch": 4.78363037109375e-06,
      "step": 1254,
      "training_step_time": 10.281177759170532
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 3.162325143814087,
      "step": 1255
    },
    {
      "dynamics_loss_avg": 0.07286732122302056,
      "epoch": 4.787445068359375e-06,
      "step": 1255
    },
    {
      "action_loss_avg": 0.04562793858349323,
      "epoch": 4.787445068359375e-06,
      "step": 1255
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 1255,
      "training_step_time": 10.229828834533691
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 3.134584903717041,
      "step": 1255
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 1255,
      "training_step_time": 9.911027431488037
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 3.1343882083892822,
      "step": 1255
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 1255,
      "training_step_time": 9.90955638885498
    },
    {
      "epoch": 4.787445068359375e-06,
      "model_forward_time": 3.136209011077881,
      "step": 1255
    },
    {
      "epoch": 4.787445068359375e-06,
      "step": 1255,
      "training_step_time": 10.015023708343506
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 3.1876277923583984,
      "step": 1256
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 1256,
      "training_step_time": 9.97009539604187
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 3.135415554046631,
      "step": 1256
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 1256,
      "training_step_time": 10.183682203292847
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 3.1355302333831787,
      "step": 1256
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 1256,
      "training_step_time": 9.915530443191528
    },
    {
      "epoch": 4.791259765625e-06,
      "model_forward_time": 3.141951084136963,
      "step": 1256
    },
    {
      "epoch": 4.791259765625e-06,
      "step": 1256,
      "training_step_time": 10.292160511016846
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 3.167402505874634,
      "step": 1257
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 1257,
      "training_step_time": 9.951575994491577
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 3.140882968902588,
      "step": 1257
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 1257,
      "training_step_time": 9.92615032196045
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 3.409335136413574,
      "step": 1257
    },
    {
      "dynamics_loss_avg": 0.08112905491143466,
      "epoch": 4.795074462890625e-06,
      "step": 1257
    },
    {
      "action_loss_avg": 0.05829081516712904,
      "epoch": 4.795074462890625e-06,
      "step": 1257
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 1257,
      "training_step_time": 10.201234579086304
    },
    {
      "epoch": 4.795074462890625e-06,
      "model_forward_time": 3.146909713745117,
      "step": 1257
    },
    {
      "epoch": 4.795074462890625e-06,
      "step": 1257,
      "training_step_time": 10.343837976455688
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 2.4845268726348877,
      "step": 1258
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 1258,
      "training_step_time": 7.5166895389556885
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 3.135370969772339,
      "step": 1258
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 1258,
      "training_step_time": 9.917197465896606
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 3.1433522701263428,
      "step": 1258
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 1258,
      "training_step_time": 9.92664909362793
    },
    {
      "epoch": 4.79888916015625e-06,
      "model_forward_time": 3.1387882232666016,
      "step": 1258
    },
    {
      "epoch": 4.79888916015625e-06,
      "step": 1258,
      "training_step_time": 10.314927816390991
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 2.4758613109588623,
      "step": 1259
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 1259,
      "training_step_time": 7.511321544647217
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 3.133988857269287,
      "step": 1259
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 1259,
      "training_step_time": 9.916226863861084
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 3.1383540630340576,
      "step": 1259
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 1259,
      "training_step_time": 9.926515817642212
    },
    {
      "epoch": 4.802703857421875e-06,
      "model_forward_time": 3.1390113830566406,
      "step": 1259
    },
    {
      "epoch": 4.802703857421875e-06,
      "step": 1259,
      "training_step_time": 10.255066871643066
    },
    {
      "epoch": 4.8065185546875e-06,
      "grad_norm": 0.5926593542098999,
      "learning_rate": 1e-05,
      "loss": 0.1272,
      "step": 1260
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 3.1836800575256348,
      "step": 1260
    },
    {
      "dynamics_loss_avg": 0.059574679285287854,
      "epoch": 4.8065185546875e-06,
      "step": 1260
    },
    {
      "action_loss_avg": 0.057218436151742935,
      "epoch": 4.8065185546875e-06,
      "step": 1260
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 1260,
      "training_step_time": 10.242393493652344
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 3.1363654136657715,
      "step": 1260
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 1260,
      "training_step_time": 9.92826533317566
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 3.1316282749176025,
      "step": 1260
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 1260,
      "training_step_time": 9.905207395553589
    },
    {
      "epoch": 4.8065185546875e-06,
      "model_forward_time": 3.1299870014190674,
      "step": 1260
    },
    {
      "epoch": 4.8065185546875e-06,
      "step": 1260,
      "training_step_time": 10.016488313674927
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 2.471792459487915,
      "step": 1261
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 1261,
      "training_step_time": 7.50213885307312
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 3.133415699005127,
      "step": 1261
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 1261,
      "training_step_time": 10.171231508255005
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 3.1310551166534424,
      "step": 1261
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 1261,
      "training_step_time": 9.903799057006836
    },
    {
      "epoch": 4.810333251953125e-06,
      "model_forward_time": 2.432283878326416,
      "step": 1261
    },
    {
      "epoch": 4.810333251953125e-06,
      "step": 1261,
      "training_step_time": 10.299669027328491
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 3.1791086196899414,
      "step": 1262
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 1262,
      "training_step_time": 9.952093362808228
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 3.137352466583252,
      "step": 1262
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 1262,
      "training_step_time": 9.916549682617188
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 3.132190704345703,
      "step": 1262
    },
    {
      "dynamics_loss_avg": 0.07048600092530251,
      "epoch": 4.81414794921875e-06,
      "step": 1262
    },
    {
      "action_loss_avg": 0.051027346216142175,
      "epoch": 4.81414794921875e-06,
      "step": 1262
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 1262,
      "training_step_time": 10.182469367980957
    },
    {
      "epoch": 4.81414794921875e-06,
      "model_forward_time": 3.136425733566284,
      "step": 1262
    },
    {
      "epoch": 4.81414794921875e-06,
      "step": 1262,
      "training_step_time": 10.30752182006836
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 3.1426055431365967,
      "step": 1263
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 1263,
      "training_step_time": 9.921051502227783
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 3.131666421890259,
      "step": 1263
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 1263,
      "training_step_time": 9.910099744796753
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 2.4326376914978027,
      "step": 1263
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 1263,
      "training_step_time": 7.465916872024536
    },
    {
      "epoch": 4.817962646484375e-06,
      "model_forward_time": 3.4146509170532227,
      "step": 1263
    },
    {
      "epoch": 4.817962646484375e-06,
      "step": 1263,
      "training_step_time": 10.296786546707153
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 3.1715497970581055,
      "step": 1264
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 1264,
      "training_step_time": 9.949697732925415
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 3.1359429359436035,
      "step": 1264
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 1264,
      "training_step_time": 9.914458990097046
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 3.127453088760376,
      "step": 1264
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 1264,
      "training_step_time": 9.908829927444458
    },
    {
      "epoch": 4.82177734375e-06,
      "model_forward_time": 3.138172149658203,
      "step": 1264
    },
    {
      "epoch": 4.82177734375e-06,
      "step": 1264,
      "training_step_time": 10.252470970153809
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 3.166377067565918,
      "step": 1265
    },
    {
      "dynamics_loss_avg": 0.06996574923396111,
      "epoch": 4.825592041015625e-06,
      "step": 1265
    },
    {
      "action_loss_avg": 0.05756958834826946,
      "epoch": 4.825592041015625e-06,
      "step": 1265
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 1265,
      "training_step_time": 10.224439144134521
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 3.1367104053497314,
      "step": 1265
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 1265,
      "training_step_time": 9.91050910949707
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 3.129544734954834,
      "step": 1265
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 1265,
      "training_step_time": 9.901361465454102
    },
    {
      "epoch": 4.825592041015625e-06,
      "model_forward_time": 3.1318020820617676,
      "step": 1265
    },
    {
      "epoch": 4.825592041015625e-06,
      "step": 1265,
      "training_step_time": 10.011129140853882
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 3.166872978210449,
      "step": 1266
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 1266,
      "training_step_time": 9.939684867858887
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 3.4183075428009033,
      "step": 1266
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 1266,
      "training_step_time": 10.191068649291992
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 3.132380723953247,
      "step": 1266
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 1266,
      "training_step_time": 9.912404537200928
    },
    {
      "epoch": 4.82940673828125e-06,
      "model_forward_time": 3.13301944732666,
      "step": 1266
    },
    {
      "epoch": 4.82940673828125e-06,
      "step": 1266,
      "training_step_time": 10.019620656967163
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 3.16211199760437,
      "step": 1267
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 1267,
      "training_step_time": 9.943051815032959
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 3.1365978717803955,
      "step": 1267
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 1267,
      "training_step_time": 9.915340423583984
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 3.4162323474884033,
      "step": 1267
    },
    {
      "dynamics_loss_avg": 0.0713335320353508,
      "epoch": 4.833221435546875e-06,
      "step": 1267
    },
    {
      "action_loss_avg": 0.052737926319241527,
      "epoch": 4.833221435546875e-06,
      "step": 1267
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 1267,
      "training_step_time": 10.200500726699829
    },
    {
      "epoch": 4.833221435546875e-06,
      "model_forward_time": 3.1395130157470703,
      "step": 1267
    },
    {
      "epoch": 4.833221435546875e-06,
      "step": 1267,
      "training_step_time": 10.022683382034302
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 3.1746013164520264,
      "step": 1268
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 1268,
      "training_step_time": 9.954496383666992
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 3.1322059631347656,
      "step": 1268
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 1268,
      "training_step_time": 9.910510540008545
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 2.4337215423583984,
      "step": 1268
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 1268,
      "training_step_time": 7.470208168029785
    },
    {
      "epoch": 4.8370361328125e-06,
      "model_forward_time": 3.4151644706726074,
      "step": 1268
    },
    {
      "epoch": 4.8370361328125e-06,
      "step": 1268,
      "training_step_time": 10.32451844215393
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 3.16062068939209,
      "step": 1269
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 1269,
      "training_step_time": 9.940994501113892
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 3.135063648223877,
      "step": 1269
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 1269,
      "training_step_time": 9.915628910064697
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 3.139165163040161,
      "step": 1269
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 1269,
      "training_step_time": 9.919202089309692
    },
    {
      "epoch": 4.840850830078125e-06,
      "model_forward_time": 3.136971950531006,
      "step": 1269
    },
    {
      "epoch": 4.840850830078125e-06,
      "step": 1269,
      "training_step_time": 10.266703128814697
    },
    {
      "epoch": 4.84466552734375e-06,
      "grad_norm": 0.6691280603408813,
      "learning_rate": 1e-05,
      "loss": 0.1275,
      "step": 1270
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 3.4561080932617188,
      "step": 1270
    },
    {
      "dynamics_loss_avg": 0.08025651760399341,
      "epoch": 4.84466552734375e-06,
      "step": 1270
    },
    {
      "action_loss_avg": 0.05880178846418858,
      "epoch": 4.84466552734375e-06,
      "step": 1270
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 1270,
      "training_step_time": 10.239523649215698
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 2.4335615634918213,
      "step": 1270
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 1270,
      "training_step_time": 7.468446731567383
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 3.1346185207366943,
      "step": 1270
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 1270,
      "training_step_time": 9.909547805786133
    },
    {
      "epoch": 4.84466552734375e-06,
      "model_forward_time": 3.125939130783081,
      "step": 1270
    },
    {
      "epoch": 4.84466552734375e-06,
      "step": 1270,
      "training_step_time": 10.288815259933472
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 2.458265542984009,
      "step": 1271
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 1271,
      "training_step_time": 7.485277891159058
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 3.1298508644104004,
      "step": 1271
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 1271,
      "training_step_time": 10.20583200454712
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 3.130831480026245,
      "step": 1271
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 1271,
      "training_step_time": 9.897485733032227
    },
    {
      "epoch": 4.848480224609375e-06,
      "model_forward_time": 3.1302528381347656,
      "step": 1271
    },
    {
      "epoch": 4.848480224609375e-06,
      "step": 1271,
      "training_step_time": 10.007608413696289
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 3.1625497341156006,
      "step": 1272
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 1272,
      "training_step_time": 9.937490940093994
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 2.430565357208252,
      "step": 1272
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 1272,
      "training_step_time": 7.464913368225098
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 3.131202220916748,
      "step": 1272
    },
    {
      "dynamics_loss_avg": 0.07535805087536573,
      "epoch": 4.852294921875e-06,
      "step": 1272
    },
    {
      "action_loss_avg": 0.05581144206225872,
      "epoch": 4.852294921875e-06,
      "step": 1272
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 1272,
      "training_step_time": 10.19199538230896
    },
    {
      "epoch": 4.852294921875e-06,
      "model_forward_time": 3.1354923248291016,
      "step": 1272
    },
    {
      "epoch": 4.852294921875e-06,
      "step": 1272,
      "training_step_time": 10.023967742919922
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 3.187242031097412,
      "step": 1273
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 1273,
      "training_step_time": 9.967151403427124
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 3.148252010345459,
      "step": 1273
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 1273,
      "training_step_time": 9.932713985443115
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 3.149702548980713,
      "step": 1273
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 1273,
      "training_step_time": 9.93667984008789
    },
    {
      "epoch": 4.856109619140625e-06,
      "model_forward_time": 3.153040647506714,
      "step": 1273
    },
    {
      "epoch": 4.856109619140625e-06,
      "step": 1273,
      "training_step_time": 10.317935943603516
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 3.178950071334839,
      "step": 1274
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 1274,
      "training_step_time": 9.968339920043945
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 3.1514205932617188,
      "step": 1274
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 1274,
      "training_step_time": 9.933781385421753
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 3.149441957473755,
      "step": 1274
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 1274,
      "training_step_time": 9.939523458480835
    },
    {
      "epoch": 4.85992431640625e-06,
      "model_forward_time": 3.1534531116485596,
      "step": 1274
    },
    {
      "epoch": 4.85992431640625e-06,
      "step": 1274,
      "training_step_time": 10.280976057052612
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 3.4493472576141357,
      "step": 1275
    },
    {
      "dynamics_loss_avg": 0.07541464194655419,
      "epoch": 4.863739013671875e-06,
      "step": 1275
    },
    {
      "action_loss_avg": 0.05218508653342724,
      "epoch": 4.863739013671875e-06,
      "step": 1275
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 1275,
      "training_step_time": 10.238354206085205
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 3.139049768447876,
      "step": 1275
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 1275,
      "training_step_time": 9.919015169143677
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 3.137688159942627,
      "step": 1275
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 1275,
      "training_step_time": 9.910851955413818
    },
    {
      "epoch": 4.863739013671875e-06,
      "model_forward_time": 3.1390130519866943,
      "step": 1275
    },
    {
      "epoch": 4.863739013671875e-06,
      "step": 1275,
      "training_step_time": 10.30942702293396
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 3.167971134185791,
      "step": 1276
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 1276,
      "training_step_time": 9.947032928466797
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 3.4201154708862305,
      "step": 1276
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 1276,
      "training_step_time": 10.194694519042969
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 3.1324923038482666,
      "step": 1276
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 1276,
      "training_step_time": 9.908406972885132
    },
    {
      "epoch": 4.8675537109375e-06,
      "model_forward_time": 3.1393160820007324,
      "step": 1276
    },
    {
      "epoch": 4.8675537109375e-06,
      "step": 1276,
      "training_step_time": 10.244401454925537
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 3.1616692543029785,
      "step": 1277
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 1277,
      "training_step_time": 9.943304061889648
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 3.1382479667663574,
      "step": 1277
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 1277,
      "training_step_time": 9.92061972618103
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 3.398747444152832,
      "step": 1277
    },
    {
      "dynamics_loss_avg": 0.08383618704974652,
      "epoch": 4.871368408203125e-06,
      "step": 1277
    },
    {
      "action_loss_avg": 0.05884843301028013,
      "epoch": 4.871368408203125e-06,
      "step": 1277
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 1277,
      "training_step_time": 10.18218731880188
    },
    {
      "epoch": 4.871368408203125e-06,
      "model_forward_time": 3.1436352729797363,
      "step": 1277
    },
    {
      "epoch": 4.871368408203125e-06,
      "step": 1277,
      "training_step_time": 10.031464338302612
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 3.166985273361206,
      "step": 1278
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 1278,
      "training_step_time": 9.948891639709473
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 3.139298677444458,
      "step": 1278
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 1278,
      "training_step_time": 9.921067953109741
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 3.1413257122039795,
      "step": 1278
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 1278,
      "training_step_time": 9.924715995788574
    },
    {
      "epoch": 4.87518310546875e-06,
      "model_forward_time": 3.4489054679870605,
      "step": 1278
    },
    {
      "epoch": 4.87518310546875e-06,
      "step": 1278,
      "training_step_time": 10.323920488357544
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 2.463369846343994,
      "step": 1279
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 1279,
      "training_step_time": 7.491875648498535
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 3.134291410446167,
      "step": 1279
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 1279,
      "training_step_time": 9.915810108184814
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 3.140177011489868,
      "step": 1279
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 1279,
      "training_step_time": 9.919472932815552
    },
    {
      "epoch": 4.878997802734375e-06,
      "model_forward_time": 3.1395554542541504,
      "step": 1279
    },
    {
      "epoch": 4.878997802734375e-06,
      "step": 1279,
      "training_step_time": 10.334126710891724
    },
    {
      "epoch": 4.8828125e-06,
      "grad_norm": 0.6002122759819031,
      "learning_rate": 1e-05,
      "loss": 0.1263,
      "step": 1280
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 3.1924071311950684,
      "step": 1280
    },
    {
      "dynamics_loss_avg": 0.07538134232163429,
      "epoch": 4.8828125e-06,
      "step": 1280
    },
    {
      "action_loss_avg": 0.05817571394145489,
      "epoch": 4.8828125e-06,
      "step": 1280
    },
    {
      "epoch": 4.8828125e-06,
      "step": 1280,
      "training_step_time": 10.271899938583374
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 3.142791986465454,
      "step": 1280
    },
    {
      "epoch": 4.8828125e-06,
      "step": 1280,
      "training_step_time": 9.925581216812134
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 3.145648956298828,
      "step": 1280
    },
    {
      "epoch": 4.8828125e-06,
      "step": 1280,
      "training_step_time": 9.92772650718689
    },
    {
      "epoch": 4.8828125e-06,
      "model_forward_time": 3.144193649291992,
      "step": 1280
    },
    {
      "epoch": 4.8828125e-06,
      "step": 1280,
      "training_step_time": 10.286427736282349
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 3.1609034538269043,
      "step": 1281
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 1281,
      "training_step_time": 9.941386222839355
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 3.4002089500427246,
      "step": 1281
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 1281,
      "training_step_time": 10.181599855422974
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 3.1445209980010986,
      "step": 1281
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 1281,
      "training_step_time": 9.92385721206665
    },
    {
      "epoch": 4.886627197265625e-06,
      "model_forward_time": 2.438492774963379,
      "step": 1281
    },
    {
      "epoch": 4.886627197265625e-06,
      "step": 1281,
      "training_step_time": 10.28705096244812
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 3.1722257137298584,
      "step": 1282
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 1282,
      "training_step_time": 9.953534126281738
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 3.1376123428344727,
      "step": 1282
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 1282,
      "training_step_time": 9.913216829299927
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 3.1331398487091064,
      "step": 1282
    },
    {
      "dynamics_loss_avg": 0.08024538569152355,
      "epoch": 4.89044189453125e-06,
      "step": 1282
    },
    {
      "action_loss_avg": 0.05564060658216476,
      "epoch": 4.89044189453125e-06,
      "step": 1282
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 1282,
      "training_step_time": 10.172307014465332
    },
    {
      "epoch": 4.89044189453125e-06,
      "model_forward_time": 3.131873846054077,
      "step": 1282
    },
    {
      "epoch": 4.89044189453125e-06,
      "step": 1282,
      "training_step_time": 10.285423278808594
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 2.471468210220337,
      "step": 1283
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 1283,
      "training_step_time": 7.503661632537842
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 3.1331231594085693,
      "step": 1283
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 1283,
      "training_step_time": 9.900032758712769
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 3.1334075927734375,
      "step": 1283
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 1283,
      "training_step_time": 9.897159576416016
    },
    {
      "epoch": 4.894256591796875e-06,
      "model_forward_time": 3.422407388687134,
      "step": 1283
    },
    {
      "epoch": 4.894256591796875e-06,
      "step": 1283,
      "training_step_time": 10.277585983276367
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 2.472649335861206,
      "step": 1284
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 1284,
      "training_step_time": 7.4905571937561035
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 3.1325533390045166,
      "step": 1284
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 1284,
      "training_step_time": 9.894410133361816
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 2.431936025619507,
      "step": 1284
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 1284,
      "training_step_time": 7.456521272659302
    },
    {
      "epoch": 4.8980712890625e-06,
      "model_forward_time": 3.1356539726257324,
      "step": 1284
    },
    {
      "epoch": 4.8980712890625e-06,
      "step": 1284,
      "training_step_time": 10.278918981552124
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 3.171049118041992,
      "step": 1285
    },
    {
      "dynamics_loss_avg": 0.06942528448998928,
      "epoch": 4.901885986328125e-06,
      "step": 1285
    },
    {
      "action_loss_avg": 0.06772016733884811,
      "epoch": 4.901885986328125e-06,
      "step": 1285
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 1285,
      "training_step_time": 9.949269533157349
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 3.417715072631836,
      "step": 1285
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 1285,
      "training_step_time": 10.184844732284546
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 3.1352696418762207,
      "step": 1285
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 1285,
      "training_step_time": 9.903147459030151
    },
    {
      "epoch": 4.901885986328125e-06,
      "model_forward_time": 3.1420505046844482,
      "step": 1285
    },
    {
      "epoch": 4.901885986328125e-06,
      "step": 1285,
      "training_step_time": 10.282228708267212
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 3.180147409439087,
      "step": 1286
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 1286,
      "training_step_time": 9.949886798858643
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 3.141178846359253,
      "step": 1286
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 1286,
      "training_step_time": 10.174036741256714
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 3.139130115509033,
      "step": 1286
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 1286,
      "training_step_time": 9.904325723648071
    },
    {
      "epoch": 4.90570068359375e-06,
      "model_forward_time": 3.1422834396362305,
      "step": 1286
    },
    {
      "epoch": 4.90570068359375e-06,
      "step": 1286,
      "training_step_time": 10.302429914474487
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 3.1827125549316406,
      "step": 1287
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 1287,
      "training_step_time": 9.951748609542847
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 3.1347920894622803,
      "step": 1287
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 1287,
      "training_step_time": 9.89791202545166
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 3.1385507583618164,
      "step": 1287
    },
    {
      "dynamics_loss_avg": 0.07659001797437667,
      "epoch": 4.909515380859375e-06,
      "step": 1287
    },
    {
      "action_loss_avg": 0.06141771376132965,
      "epoch": 4.909515380859375e-06,
      "step": 1287
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 1287,
      "training_step_time": 10.206753969192505
    },
    {
      "epoch": 4.909515380859375e-06,
      "model_forward_time": 3.144157648086548,
      "step": 1287
    },
    {
      "epoch": 4.909515380859375e-06,
      "step": 1287,
      "training_step_time": 10.248109817504883
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 3.1610617637634277,
      "step": 1288
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 1288,
      "training_step_time": 9.928235530853271
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 3.134909152984619,
      "step": 1288
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 1288,
      "training_step_time": 9.903732299804688
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 3.1390316486358643,
      "step": 1288
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 1288,
      "training_step_time": 9.90242624282837
    },
    {
      "epoch": 4.913330078125e-06,
      "model_forward_time": 3.426896572113037,
      "step": 1288
    },
    {
      "epoch": 4.913330078125e-06,
      "step": 1288,
      "training_step_time": 10.288078308105469
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 3.1669554710388184,
      "step": 1289
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 1289,
      "training_step_time": 9.939105033874512
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 3.139251708984375,
      "step": 1289
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 1289,
      "training_step_time": 9.904979944229126
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 3.141710042953491,
      "step": 1289
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 1289,
      "training_step_time": 9.910137176513672
    },
    {
      "epoch": 4.917144775390625e-06,
      "model_forward_time": 3.142141342163086,
      "step": 1289
    },
    {
      "epoch": 4.917144775390625e-06,
      "step": 1289,
      "training_step_time": 10.011672496795654
    },
    {
      "epoch": 4.92095947265625e-06,
      "grad_norm": 0.5102371573448181,
      "learning_rate": 1e-05,
      "loss": 0.1242,
      "step": 1290
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 3.4551854133605957,
      "step": 1290
    },
    {
      "dynamics_loss_avg": 0.07563164420425891,
      "epoch": 4.92095947265625e-06,
      "step": 1290
    },
    {
      "action_loss_avg": 0.0556735148653388,
      "epoch": 4.92095947265625e-06,
      "step": 1290
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 1290,
      "training_step_time": 10.2244131565094
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 3.138126850128174,
      "step": 1290
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 1290,
      "training_step_time": 9.90805959701538
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 3.141430377960205,
      "step": 1290
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 1290,
      "training_step_time": 9.911460161209106
    },
    {
      "epoch": 4.92095947265625e-06,
      "model_forward_time": 3.1403017044067383,
      "step": 1290
    },
    {
      "epoch": 4.92095947265625e-06,
      "step": 1290,
      "training_step_time": 10.332358598709106
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 3.1747238636016846,
      "step": 1291
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 1291,
      "training_step_time": 9.945552825927734
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 3.424368143081665,
      "step": 1291
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 1291,
      "training_step_time": 10.190936088562012
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 3.143106460571289,
      "step": 1291
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 1291,
      "training_step_time": 9.911741256713867
    },
    {
      "epoch": 4.924774169921875e-06,
      "model_forward_time": 3.1383020877838135,
      "step": 1291
    },
    {
      "epoch": 4.924774169921875e-06,
      "step": 1291,
      "training_step_time": 10.29996132850647
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 3.1853742599487305,
      "step": 1292
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 1292,
      "training_step_time": 9.959235906600952
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 3.1443264484405518,
      "step": 1292
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 1292,
      "training_step_time": 9.911962985992432
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 3.4306888580322266,
      "step": 1292
    },
    {
      "dynamics_loss_avg": 0.07746489644050598,
      "epoch": 4.9285888671875e-06,
      "step": 1292
    },
    {
      "action_loss_avg": 0.06254627835005522,
      "epoch": 4.9285888671875e-06,
      "step": 1292
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 1292,
      "training_step_time": 10.204843044281006
    },
    {
      "epoch": 4.9285888671875e-06,
      "model_forward_time": 3.139859676361084,
      "step": 1292
    },
    {
      "epoch": 4.9285888671875e-06,
      "step": 1292,
      "training_step_time": 10.252717018127441
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 3.1763572692871094,
      "step": 1293
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 1293,
      "training_step_time": 9.949388265609741
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 3.1463990211486816,
      "step": 1293
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 1293,
      "training_step_time": 9.916359424591064
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 3.1440300941467285,
      "step": 1293
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 1293,
      "training_step_time": 9.917369365692139
    },
    {
      "epoch": 4.932403564453125e-06,
      "model_forward_time": 3.40240740776062,
      "step": 1293
    },
    {
      "epoch": 4.932403564453125e-06,
      "step": 1293,
      "training_step_time": 10.273224115371704
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 3.1814217567443848,
      "step": 1294
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 1294,
      "training_step_time": 9.961530685424805
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 3.1477198600769043,
      "step": 1294
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 1294,
      "training_step_time": 9.92413330078125
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 2.4408915042877197,
      "step": 1294
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 1294,
      "training_step_time": 7.476579666137695
    },
    {
      "epoch": 4.93621826171875e-06,
      "model_forward_time": 3.1450374126434326,
      "step": 1294
    },
    {
      "epoch": 4.93621826171875e-06,
      "step": 1294,
      "training_step_time": 10.015913009643555
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 3.472203254699707,
      "step": 1295
    },
    {
      "dynamics_loss_avg": 0.07612301856279373,
      "epoch": 4.940032958984375e-06,
      "step": 1295
    },
    {
      "action_loss_avg": 0.04614935293793678,
      "epoch": 4.940032958984375e-06,
      "step": 1295
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 1295,
      "training_step_time": 10.242105484008789
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 3.144827365875244,
      "step": 1295
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 1295,
      "training_step_time": 9.915471315383911
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 3.141634941101074,
      "step": 1295
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 1295,
      "training_step_time": 9.9105863571167
    },
    {
      "epoch": 4.940032958984375e-06,
      "model_forward_time": 3.143354654312134,
      "step": 1295
    },
    {
      "epoch": 4.940032958984375e-06,
      "step": 1295,
      "training_step_time": 10.259945154190063
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 2.4869637489318848,
      "step": 1296
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 1296,
      "training_step_time": 7.511836290359497
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 3.1377785205841064,
      "step": 1296
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 1296,
      "training_step_time": 9.90478515625
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 3.3903887271881104,
      "step": 1296
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 1296,
      "training_step_time": 10.168414115905762
    },
    {
      "epoch": 4.94384765625e-06,
      "model_forward_time": 3.1460556983947754,
      "step": 1296
    },
    {
      "epoch": 4.94384765625e-06,
      "step": 1296,
      "training_step_time": 10.032991170883179
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 3.196479082107544,
      "step": 1297
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 1297,
      "training_step_time": 9.970202922821045
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 3.1430816650390625,
      "step": 1297
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 1297,
      "training_step_time": 9.91731595993042
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 3.4193522930145264,
      "step": 1297
    },
    {
      "dynamics_loss_avg": 0.09578517079353333,
      "epoch": 4.947662353515625e-06,
      "step": 1297
    },
    {
      "action_loss_avg": 0.047923816740512847,
      "epoch": 4.947662353515625e-06,
      "step": 1297
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 1297,
      "training_step_time": 10.192651748657227
    },
    {
      "epoch": 4.947662353515625e-06,
      "model_forward_time": 3.1467738151550293,
      "step": 1297
    },
    {
      "epoch": 4.947662353515625e-06,
      "step": 1297,
      "training_step_time": 10.275686979293823
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 3.19781494140625,
      "step": 1298
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 1298,
      "training_step_time": 9.973866701126099
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 3.144902467727661,
      "step": 1298
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 1298,
      "training_step_time": 9.922734498977661
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 3.1396377086639404,
      "step": 1298
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 1298,
      "training_step_time": 9.914108991622925
    },
    {
      "epoch": 4.95147705078125e-06,
      "model_forward_time": 3.1400721073150635,
      "step": 1298
    },
    {
      "epoch": 4.95147705078125e-06,
      "step": 1298,
      "training_step_time": 10.303365468978882
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 3.195497989654541,
      "step": 1299
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 1299,
      "training_step_time": 9.962332487106323
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 3.1391453742980957,
      "step": 1299
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 1299,
      "training_step_time": 9.90434741973877
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 3.1415469646453857,
      "step": 1299
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 1299,
      "training_step_time": 9.90692663192749
    },
    {
      "epoch": 4.955291748046875e-06,
      "model_forward_time": 3.1379313468933105,
      "step": 1299
    },
    {
      "epoch": 4.955291748046875e-06,
      "step": 1299,
      "training_step_time": 10.003982782363892
    },
    {
      "epoch": 4.9591064453125e-06,
      "grad_norm": 0.7612583041191101,
      "learning_rate": 1e-05,
      "loss": 0.1259,
      "step": 1300
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 3.4663572311401367,
      "step": 1300
    },
    {
      "dynamics_loss_avg": 0.06972921658307314,
      "epoch": 4.9591064453125e-06,
      "step": 1300
    },
    {
      "action_loss_avg": 0.052862979285418984,
      "epoch": 4.9591064453125e-06,
      "step": 1300
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 1300,
      "training_step_time": 10.237093448638916
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 3.1367228031158447,
      "step": 1300
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 1300,
      "training_step_time": 9.90161395072937
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 3.1353824138641357,
      "step": 1300
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 1300,
      "training_step_time": 9.901967287063599
    },
    {
      "epoch": 4.9591064453125e-06,
      "model_forward_time": 3.1329052448272705,
      "step": 1300
    },
    {
      "epoch": 4.9591064453125e-06,
      "step": 1300,
      "training_step_time": 10.007539987564087
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 2.4754722118377686,
      "step": 1301
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 1301,
      "training_step_time": 7.499887466430664
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 3.4061429500579834,
      "step": 1301
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 1301,
      "training_step_time": 10.170695304870605
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 3.1380326747894287,
      "step": 1301
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 1301,
      "training_step_time": 9.910250663757324
    },
    {
      "epoch": 4.962921142578125e-06,
      "model_forward_time": 3.1332991123199463,
      "step": 1301
    },
    {
      "epoch": 4.962921142578125e-06,
      "step": 1301,
      "training_step_time": 10.307504892349243
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 3.1688051223754883,
      "step": 1302
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 1302,
      "training_step_time": 9.93997836112976
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 3.139655113220215,
      "step": 1302
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 1302,
      "training_step_time": 9.91277289390564
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 3.1384406089782715,
      "step": 1302
    },
    {
      "dynamics_loss_avg": 0.06363991163671016,
      "epoch": 4.96673583984375e-06,
      "step": 1302
    },
    {
      "action_loss_avg": 0.04390417542308569,
      "epoch": 4.96673583984375e-06,
      "step": 1302
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 1302,
      "training_step_time": 10.180303812026978
    },
    {
      "epoch": 4.96673583984375e-06,
      "model_forward_time": 3.136340379714966,
      "step": 1302
    },
    {
      "epoch": 4.96673583984375e-06,
      "step": 1302,
      "training_step_time": 10.270745038986206
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 3.167574167251587,
      "step": 1303
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 1303,
      "training_step_time": 9.94340205192566
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 2.4345335960388184,
      "step": 1303
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 1303,
      "training_step_time": 7.463863849639893
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 3.1344873905181885,
      "step": 1303
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 1303,
      "training_step_time": 9.905164241790771
    },
    {
      "epoch": 4.970550537109375e-06,
      "model_forward_time": 3.1416265964508057,
      "step": 1303
    },
    {
      "epoch": 4.970550537109375e-06,
      "step": 1303,
      "training_step_time": 10.303240776062012
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 3.1723809242248535,
      "step": 1304
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 1304,
      "training_step_time": 9.947091341018677
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 3.140087604522705,
      "step": 1304
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 1304,
      "training_step_time": 9.914872407913208
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 3.1397628784179688,
      "step": 1304
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 1304,
      "training_step_time": 9.91785740852356
    },
    {
      "epoch": 4.974365234375e-06,
      "model_forward_time": 3.1445178985595703,
      "step": 1304
    },
    {
      "epoch": 4.974365234375e-06,
      "step": 1304,
      "training_step_time": 10.027822732925415
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 3.151750087738037,
      "step": 1305
    },
    {
      "dynamics_loss_avg": 0.06321661546826363,
      "epoch": 4.978179931640625e-06,
      "step": 1305
    },
    {
      "action_loss_avg": 0.047693542949855325,
      "epoch": 4.978179931640625e-06,
      "step": 1305
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 1305,
      "training_step_time": 10.20583963394165
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 3.1380558013916016,
      "step": 1305
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 1305,
      "training_step_time": 9.912238121032715
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 2.4358348846435547,
      "step": 1305
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 1305,
      "training_step_time": 7.4644083976745605
    },
    {
      "epoch": 4.978179931640625e-06,
      "model_forward_time": 2.430302381515503,
      "step": 1305
    },
    {
      "epoch": 4.978179931640625e-06,
      "step": 1305,
      "training_step_time": 10.010540008544922
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 3.1737942695617676,
      "step": 1306
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 1306,
      "training_step_time": 9.9444739818573
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 2.4344863891601562,
      "step": 1306
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 1306,
      "training_step_time": 7.456918954849243
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 3.3783955574035645,
      "step": 1306
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 1306,
      "training_step_time": 10.143693923950195
    },
    {
      "epoch": 4.98199462890625e-06,
      "model_forward_time": 3.1369948387145996,
      "step": 1306
    },
    {
      "epoch": 4.98199462890625e-06,
      "step": 1306,
      "training_step_time": 10.269991159439087
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 3.1719229221343994,
      "step": 1307
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 1307,
      "training_step_time": 9.937992811203003
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 3.1326792240142822,
      "step": 1307
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 1307,
      "training_step_time": 9.902102708816528
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 3.1399621963500977,
      "step": 1307
    },
    {
      "dynamics_loss_avg": 0.07438312284648418,
      "epoch": 4.985809326171875e-06,
      "step": 1307
    },
    {
      "action_loss_avg": 0.044337647035717964,
      "epoch": 4.985809326171875e-06,
      "step": 1307
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 1307,
      "training_step_time": 9.91135549545288
    },
    {
      "epoch": 4.985809326171875e-06,
      "model_forward_time": 3.442857265472412,
      "step": 1307
    },
    {
      "epoch": 4.985809326171875e-06,
      "step": 1307,
      "training_step_time": 10.306148529052734
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 2.467883825302124,
      "step": 1308
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 1308,
      "training_step_time": 7.498403787612915
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 3.138216733932495,
      "step": 1308
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 1308,
      "training_step_time": 9.910239696502686
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 3.136967182159424,
      "step": 1308
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 1308,
      "training_step_time": 9.907618284225464
    },
    {
      "epoch": 4.9896240234375e-06,
      "model_forward_time": 3.133838653564453,
      "step": 1308
    },
    {
      "epoch": 4.9896240234375e-06,
      "step": 1308,
      "training_step_time": 10.26970362663269
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 3.149965524673462,
      "step": 1309
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 1309,
      "training_step_time": 9.915306568145752
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 3.1395931243896484,
      "step": 1309
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 1309,
      "training_step_time": 9.906824588775635
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 2.4341580867767334,
      "step": 1309
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 1309,
      "training_step_time": 7.459019422531128
    },
    {
      "epoch": 4.993438720703125e-06,
      "model_forward_time": 3.135650396347046,
      "step": 1309
    },
    {
      "epoch": 4.993438720703125e-06,
      "step": 1309,
      "training_step_time": 10.28070855140686
    },
    {
      "epoch": 4.99725341796875e-06,
      "grad_norm": 0.7325623035430908,
      "learning_rate": 1e-05,
      "loss": 0.1207,
      "step": 1310
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 3.15936279296875,
      "step": 1310
    },
    {
      "dynamics_loss_avg": 0.07048095762729645,
      "epoch": 4.99725341796875e-06,
      "step": 1310
    },
    {
      "action_loss_avg": 0.04839467499405146,
      "epoch": 4.99725341796875e-06,
      "step": 1310
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 1310,
      "training_step_time": 9.941611766815186
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 3.379303216934204,
      "step": 1310
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 1310,
      "training_step_time": 10.141678810119629
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 3.1324222087860107,
      "step": 1310
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 1310,
      "training_step_time": 9.900639772415161
    },
    {
      "epoch": 4.99725341796875e-06,
      "model_forward_time": 3.1743011474609375,
      "step": 1310
    },
    {
      "epoch": 4.99725341796875e-06,
      "step": 1310,
      "training_step_time": 10.042490720748901
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 3.1875503063201904,
      "step": 1311
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 1311,
      "training_step_time": 9.960172176361084
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 3.1339070796966553,
      "step": 1311
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 1311,
      "training_step_time": 10.193734884262085
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 3.1386570930480957,
      "step": 1311
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 1311,
      "training_step_time": 9.91030216217041
    },
    {
      "epoch": 5.001068115234375e-06,
      "model_forward_time": 3.1348912715911865,
      "step": 1311
    },
    {
      "epoch": 5.001068115234375e-06,
      "step": 1311,
      "training_step_time": 10.319191932678223
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 3.1506989002227783,
      "step": 1312
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 1312,
      "training_step_time": 9.926376819610596
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 3.1332414150238037,
      "step": 1312
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 1312,
      "training_step_time": 9.900825023651123
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 3.4341917037963867,
      "step": 1312
    },
    {
      "dynamics_loss_avg": 0.06569433212280273,
      "epoch": 5.0048828125e-06,
      "step": 1312
    },
    {
      "action_loss_avg": 0.05386839583516121,
      "epoch": 5.0048828125e-06,
      "step": 1312
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 1312,
      "training_step_time": 10.20551872253418
    },
    {
      "epoch": 5.0048828125e-06,
      "model_forward_time": 2.451171636581421,
      "step": 1312
    },
    {
      "epoch": 5.0048828125e-06,
      "step": 1312,
      "training_step_time": 10.008777618408203
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 3.1377437114715576,
      "step": 1313
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 1313,
      "training_step_time": 9.899028062820435
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 3.1302783489227295,
      "step": 1313
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 1313,
      "training_step_time": 9.89618968963623
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 3.131706953048706,
      "step": 1313
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 1313,
      "training_step_time": 9.901896476745605
    },
    {
      "epoch": 5.008697509765625e-06,
      "model_forward_time": 2.431589365005493,
      "step": 1313
    },
    {
      "epoch": 5.008697509765625e-06,
      "step": 1313,
      "training_step_time": 10.306240797042847
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 2.791163682937622,
      "step": 1314
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 1314,
      "training_step_time": 7.818275690078735
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 2.4362151622772217,
      "step": 1314
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 1314,
      "training_step_time": 7.46082329750061
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 3.1325466632843018,
      "step": 1314
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 1314,
      "training_step_time": 9.903056859970093
    },
    {
      "epoch": 5.01251220703125e-06,
      "model_forward_time": 3.1407997608184814,
      "step": 1314
    },
    {
      "epoch": 5.01251220703125e-06,
      "step": 1314,
      "training_step_time": 10.22851300239563
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 3.1593832969665527,
      "step": 1315
    },
    {
      "dynamics_loss_avg": 0.09312092289328575,
      "epoch": 5.016326904296875e-06,
      "step": 1315
    },
    {
      "action_loss_avg": 0.06294942402746528,
      "epoch": 5.016326904296875e-06,
      "step": 1315
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 1315,
      "training_step_time": 9.924752235412598
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 3.431370496749878,
      "step": 1315
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 1315,
      "training_step_time": 10.192291975021362
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 3.1291725635528564,
      "step": 1315
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 1315,
      "training_step_time": 9.89966082572937
    },
    {
      "epoch": 5.016326904296875e-06,
      "model_forward_time": 3.141781806945801,
      "step": 1315
    },
    {
      "epoch": 5.016326904296875e-06,
      "step": 1315,
      "training_step_time": 10.308993577957153
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 2.4771480560302734,
      "step": 1316
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 1316,
      "training_step_time": 7.505994081497192
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 3.1366429328918457,
      "step": 1316
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 1316,
      "training_step_time": 9.90409255027771
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 3.3943185806274414,
      "step": 1316
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 1316,
      "training_step_time": 10.162968873977661
    },
    {
      "epoch": 5.0201416015625e-06,
      "model_forward_time": 3.144805669784546,
      "step": 1316
    },
    {
      "epoch": 5.0201416015625e-06,
      "step": 1316,
      "training_step_time": 10.286062002182007
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 3.1818714141845703,
      "step": 1317
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 1317,
      "training_step_time": 9.953702211380005
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 3.1390256881713867,
      "step": 1317
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 1317,
      "training_step_time": 9.911992311477661
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 3.1413910388946533,
      "step": 1317
    },
    {
      "dynamics_loss_avg": 0.07675189860165119,
      "epoch": 5.023956298828125e-06,
      "step": 1317
    },
    {
      "action_loss_avg": 0.06211234237998724,
      "epoch": 5.023956298828125e-06,
      "step": 1317
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 1317,
      "training_step_time": 9.920578241348267
    },
    {
      "epoch": 5.023956298828125e-06,
      "model_forward_time": 3.417480945587158,
      "step": 1317
    },
    {
      "epoch": 5.023956298828125e-06,
      "step": 1317,
      "training_step_time": 10.285744905471802
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 3.1797096729278564,
      "step": 1318
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 1318,
      "training_step_time": 9.95794415473938
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 3.142268419265747,
      "step": 1318
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 1318,
      "training_step_time": 9.920563697814941
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 3.1435413360595703,
      "step": 1318
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 1318,
      "training_step_time": 9.919780015945435
    },
    {
      "epoch": 5.02777099609375e-06,
      "model_forward_time": 3.138948440551758,
      "step": 1318
    },
    {
      "epoch": 5.02777099609375e-06,
      "step": 1318,
      "training_step_time": 10.304174900054932
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 3.5059330463409424,
      "step": 1319
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 1319,
      "training_step_time": 10.277979850769043
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 3.139639139175415,
      "step": 1319
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 1319,
      "training_step_time": 9.909685850143433
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 3.1382195949554443,
      "step": 1319
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 1319,
      "training_step_time": 9.906831502914429
    },
    {
      "epoch": 5.031585693359375e-06,
      "model_forward_time": 3.136878252029419,
      "step": 1319
    },
    {
      "epoch": 5.031585693359375e-06,
      "step": 1319,
      "training_step_time": 10.014864921569824
    },
    {
      "epoch": 5.035400390625e-06,
      "grad_norm": 0.6399853229522705,
      "learning_rate": 1e-05,
      "loss": 0.1289,
      "step": 1320
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 3.1850345134735107,
      "step": 1320
    },
    {
      "dynamics_loss_avg": 0.06017192183062434,
      "epoch": 5.035400390625e-06,
      "step": 1320
    },
    {
      "action_loss_avg": 0.042580130673013625,
      "epoch": 5.035400390625e-06,
      "step": 1320
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 1320,
      "training_step_time": 10.249414920806885
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 3.1333110332489014,
      "step": 1320
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 1320,
      "training_step_time": 9.904092788696289
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 2.4317665100097656,
      "step": 1320
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 1320,
      "training_step_time": 7.461475610733032
    },
    {
      "epoch": 5.035400390625e-06,
      "model_forward_time": 3.1341068744659424,
      "step": 1320
    },
    {
      "epoch": 5.035400390625e-06,
      "step": 1320,
      "training_step_time": 10.309262990951538
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 3.187932252883911,
      "step": 1321
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 1321,
      "training_step_time": 9.970071792602539
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 3.1450839042663574,
      "step": 1321
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 1321,
      "training_step_time": 9.918333292007446
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 3.4124770164489746,
      "step": 1321
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 1321,
      "training_step_time": 10.178808689117432
    },
    {
      "epoch": 5.039215087890625e-06,
      "model_forward_time": 3.129538059234619,
      "step": 1321
    },
    {
      "epoch": 5.039215087890625e-06,
      "step": 1321,
      "training_step_time": 10.010278940200806
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 3.1680517196655273,
      "step": 1322
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 1322,
      "training_step_time": 9.934137344360352
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 3.1305716037750244,
      "step": 1322
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 1322,
      "training_step_time": 9.8926682472229
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 3.1286983489990234,
      "step": 1322
    },
    {
      "dynamics_loss_avg": 0.06447331011295318,
      "epoch": 5.04302978515625e-06,
      "step": 1322
    },
    {
      "action_loss_avg": 0.06042724568396807,
      "epoch": 5.04302978515625e-06,
      "step": 1322
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 1322,
      "training_step_time": 10.184016227722168
    },
    {
      "epoch": 5.04302978515625e-06,
      "model_forward_time": 3.130617141723633,
      "step": 1322
    },
    {
      "epoch": 5.04302978515625e-06,
      "step": 1322,
      "training_step_time": 10.279149770736694
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 3.174947738647461,
      "step": 1323
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 1323,
      "training_step_time": 9.950716495513916
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 2.430874824523926,
      "step": 1323
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 1323,
      "training_step_time": 7.461344242095947
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 3.1304781436920166,
      "step": 1323
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 1323,
      "training_step_time": 9.909234523773193
    },
    {
      "epoch": 5.046844482421875e-06,
      "model_forward_time": 3.4121253490448,
      "step": 1323
    },
    {
      "epoch": 5.046844482421875e-06,
      "step": 1323,
      "training_step_time": 10.289854526519775
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 3.175236463546753,
      "step": 1324
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 1324,
      "training_step_time": 9.957181453704834
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 3.135915517807007,
      "step": 1324
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 1324,
      "training_step_time": 9.9144926071167
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 3.132913589477539,
      "step": 1324
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 1324,
      "training_step_time": 9.907868385314941
    },
    {
      "epoch": 5.0506591796875e-06,
      "model_forward_time": 3.132934093475342,
      "step": 1324
    },
    {
      "epoch": 5.0506591796875e-06,
      "step": 1324,
      "training_step_time": 10.276018381118774
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 3.1850597858428955,
      "step": 1325
    },
    {
      "dynamics_loss_avg": 0.06616059243679047,
      "epoch": 5.054473876953125e-06,
      "step": 1325
    },
    {
      "action_loss_avg": 0.05146691296249628,
      "epoch": 5.054473876953125e-06,
      "step": 1325
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 1325,
      "training_step_time": 10.249327421188354
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 3.139817476272583,
      "step": 1325
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 1325,
      "training_step_time": 9.920856475830078
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 3.139730215072632,
      "step": 1325
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 1325,
      "training_step_time": 9.921287775039673
    },
    {
      "epoch": 5.054473876953125e-06,
      "model_forward_time": 3.1416311264038086,
      "step": 1325
    },
    {
      "epoch": 5.054473876953125e-06,
      "step": 1325,
      "training_step_time": 10.251829862594604
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 3.155263900756836,
      "step": 1326
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 1326,
      "training_step_time": 9.939215660095215
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 3.421328544616699,
      "step": 1326
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 1326,
      "training_step_time": 10.199406623840332
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 2.4386467933654785,
      "step": 1326
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 1326,
      "training_step_time": 7.469195365905762
    },
    {
      "epoch": 5.05828857421875e-06,
      "model_forward_time": 3.138495683670044,
      "step": 1326
    },
    {
      "epoch": 5.05828857421875e-06,
      "step": 1326,
      "training_step_time": 10.259187459945679
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 3.1744327545166016,
      "step": 1327
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 1327,
      "training_step_time": 9.957784652709961
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 3.142028570175171,
      "step": 1327
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 1327,
      "training_step_time": 9.92314100265503
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 3.4461050033569336,
      "step": 1327
    },
    {
      "dynamics_loss_avg": 0.06658003069460391,
      "epoch": 5.062103271484375e-06,
      "step": 1327
    },
    {
      "action_loss_avg": 0.048888104036450386,
      "epoch": 5.062103271484375e-06,
      "step": 1327
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 1327,
      "training_step_time": 10.230021715164185
    },
    {
      "epoch": 5.062103271484375e-06,
      "model_forward_time": 3.1449968814849854,
      "step": 1327
    },
    {
      "epoch": 5.062103271484375e-06,
      "step": 1327,
      "training_step_time": 10.292852878570557
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 2.4487462043762207,
      "step": 1328
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 1328,
      "training_step_time": 7.484166383743286
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 3.1310360431671143,
      "step": 1328
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 1328,
      "training_step_time": 9.906687498092651
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 3.1407690048217773,
      "step": 1328
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 1328,
      "training_step_time": 9.918430805206299
    },
    {
      "epoch": 5.06591796875e-06,
      "model_forward_time": 2.4348044395446777,
      "step": 1328
    },
    {
      "epoch": 5.06591796875e-06,
      "step": 1328,
      "training_step_time": 10.268059968948364
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 3.1834044456481934,
      "step": 1329
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 1329,
      "training_step_time": 9.96709418296814
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 3.135767936706543,
      "step": 1329
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 1329,
      "training_step_time": 9.917410612106323
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 3.138878107070923,
      "step": 1329
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 1329,
      "training_step_time": 9.916544198989868
    },
    {
      "epoch": 5.069732666015625e-06,
      "model_forward_time": 3.1389989852905273,
      "step": 1329
    },
    {
      "epoch": 5.069732666015625e-06,
      "step": 1329,
      "training_step_time": 10.310246467590332
    },
    {
      "epoch": 5.07354736328125e-06,
      "grad_norm": 0.635976254940033,
      "learning_rate": 1e-05,
      "loss": 0.1247,
      "step": 1330
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 3.4751124382019043,
      "step": 1330
    },
    {
      "dynamics_loss_avg": 0.07220943868160248,
      "epoch": 5.07354736328125e-06,
      "step": 1330
    },
    {
      "action_loss_avg": 0.060482002794742584,
      "epoch": 5.07354736328125e-06,
      "step": 1330
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 1330,
      "training_step_time": 10.256667613983154
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 3.136223793029785,
      "step": 1330
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 1330,
      "training_step_time": 9.913217067718506
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 3.140190601348877,
      "step": 1330
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 1330,
      "training_step_time": 9.917147874832153
    },
    {
      "epoch": 5.07354736328125e-06,
      "model_forward_time": 3.1378841400146484,
      "step": 1330
    },
    {
      "epoch": 5.07354736328125e-06,
      "step": 1330,
      "training_step_time": 10.01283574104309
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 3.1818957328796387,
      "step": 1331
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 1331,
      "training_step_time": 9.975263833999634
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 3.136974334716797,
      "step": 1331
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 1331,
      "training_step_time": 10.224488973617554
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 3.1381566524505615,
      "step": 1331
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 1331,
      "training_step_time": 9.91698694229126
    },
    {
      "epoch": 5.077362060546875e-06,
      "model_forward_time": 3.1354713439941406,
      "step": 1331
    },
    {
      "epoch": 5.077362060546875e-06,
      "step": 1331,
      "training_step_time": 10.252830982208252
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 3.173818826675415,
      "step": 1332
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 1332,
      "training_step_time": 9.948276281356812
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 3.1378886699676514,
      "step": 1332
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 1332,
      "training_step_time": 9.91247844696045
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 3.408435106277466,
      "step": 1332
    },
    {
      "dynamics_loss_avg": 0.05947505347430706,
      "epoch": 5.0811767578125e-06,
      "step": 1332
    },
    {
      "action_loss_avg": 0.05344612319022417,
      "epoch": 5.0811767578125e-06,
      "step": 1332
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 1332,
      "training_step_time": 10.189067363739014
    },
    {
      "epoch": 5.0811767578125e-06,
      "model_forward_time": 3.1354284286499023,
      "step": 1332
    },
    {
      "epoch": 5.0811767578125e-06,
      "step": 1332,
      "training_step_time": 10.317075729370117
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 3.169968366622925,
      "step": 1333
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 1333,
      "training_step_time": 9.942759990692139
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 3.134216547012329,
      "step": 1333
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 1333,
      "training_step_time": 9.910890340805054
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 3.132638692855835,
      "step": 1333
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 1333,
      "training_step_time": 9.908267974853516
    },
    {
      "epoch": 5.084991455078125e-06,
      "model_forward_time": 3.1386728286743164,
      "step": 1333
    },
    {
      "epoch": 5.084991455078125e-06,
      "step": 1333,
      "training_step_time": 10.299003839492798
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 3.1679067611694336,
      "step": 1334
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 1334,
      "training_step_time": 9.946224212646484
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 3.1358554363250732,
      "step": 1334
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 1334,
      "training_step_time": 9.910125017166138
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 3.132150173187256,
      "step": 1334
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 1334,
      "training_step_time": 9.90444564819336
    },
    {
      "epoch": 5.08880615234375e-06,
      "model_forward_time": 3.1360087394714355,
      "step": 1334
    },
    {
      "epoch": 5.08880615234375e-06,
      "step": 1334,
      "training_step_time": 10.282454490661621
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 3.443201780319214,
      "step": 1335
    },
    {
      "dynamics_loss_avg": 0.06474857125431299,
      "epoch": 5.092620849609375e-06,
      "step": 1335
    },
    {
      "action_loss_avg": 0.04560841722413898,
      "epoch": 5.092620849609375e-06,
      "step": 1335
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 1335,
      "training_step_time": 10.218692064285278
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 3.136598825454712,
      "step": 1335
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 1335,
      "training_step_time": 9.913634061813354
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 3.134795904159546,
      "step": 1335
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 1335,
      "training_step_time": 9.913034915924072
    },
    {
      "epoch": 5.092620849609375e-06,
      "model_forward_time": 3.137115716934204,
      "step": 1335
    },
    {
      "epoch": 5.092620849609375e-06,
      "step": 1335,
      "training_step_time": 10.02458906173706
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 2.461425542831421,
      "step": 1336
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 1336,
      "training_step_time": 7.493654012680054
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 3.134124994277954,
      "step": 1336
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 1336,
      "training_step_time": 10.185981512069702
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 3.1365885734558105,
      "step": 1336
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 1336,
      "training_step_time": 9.917855739593506
    },
    {
      "epoch": 5.096435546875e-06,
      "model_forward_time": 3.1367170810699463,
      "step": 1336
    },
    {
      "epoch": 5.096435546875e-06,
      "step": 1336,
      "training_step_time": 10.020111083984375
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 3.1827316284179688,
      "step": 1337
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 1337,
      "training_step_time": 9.972270488739014
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 3.132882595062256,
      "step": 1337
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 1337,
      "training_step_time": 9.907715320587158
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 3.3881375789642334,
      "step": 1337
    },
    {
      "dynamics_loss_avg": 0.06228932701051235,
      "epoch": 5.100250244140625e-06,
      "step": 1337
    },
    {
      "action_loss_avg": 0.04014873597770929,
      "epoch": 5.100250244140625e-06,
      "step": 1337
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 1337,
      "training_step_time": 10.165996551513672
    },
    {
      "epoch": 5.100250244140625e-06,
      "model_forward_time": 3.1342389583587646,
      "step": 1337
    },
    {
      "epoch": 5.100250244140625e-06,
      "step": 1337,
      "training_step_time": 10.26228928565979
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 2.475022792816162,
      "step": 1338
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 1338,
      "training_step_time": 7.502588272094727
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 3.1283466815948486,
      "step": 1338
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 1338,
      "training_step_time": 9.902834415435791
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 3.1362924575805664,
      "step": 1338
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 1338,
      "training_step_time": 9.91433835029602
    },
    {
      "epoch": 5.10406494140625e-06,
      "model_forward_time": 3.1337106227874756,
      "step": 1338
    },
    {
      "epoch": 5.10406494140625e-06,
      "step": 1338,
      "training_step_time": 10.313679695129395
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 2.445483684539795,
      "step": 1339
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 1339,
      "training_step_time": 7.476310968399048
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 3.1276566982269287,
      "step": 1339
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 1339,
      "training_step_time": 9.904467344284058
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 2.4332427978515625,
      "step": 1339
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 1339,
      "training_step_time": 7.4694600105285645
    },
    {
      "epoch": 5.107879638671875e-06,
      "model_forward_time": 3.1313860416412354,
      "step": 1339
    },
    {
      "epoch": 5.107879638671875e-06,
      "step": 1339,
      "training_step_time": 10.3056800365448
    },
    {
      "epoch": 5.1116943359375e-06,
      "grad_norm": 0.6346295475959778,
      "learning_rate": 1e-05,
      "loss": 0.1233,
      "step": 1340
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 2.4561331272125244,
      "step": 1340
    },
    {
      "dynamics_loss_avg": 0.06422189120203256,
      "epoch": 5.1116943359375e-06,
      "step": 1340
    },
    {
      "action_loss_avg": 0.04910170566290617,
      "epoch": 5.1116943359375e-06,
      "step": 1340
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 1340,
      "training_step_time": 7.484960556030273
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 3.3713762760162354,
      "step": 1340
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 1340,
      "training_step_time": 10.13912844657898
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 3.1318085193634033,
      "step": 1340
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 1340,
      "training_step_time": 9.903876304626465
    },
    {
      "epoch": 5.1116943359375e-06,
      "model_forward_time": 3.131305456161499,
      "step": 1340
    },
    {
      "epoch": 5.1116943359375e-06,
      "step": 1340,
      "training_step_time": 10.021850109100342
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 3.1665570735931396,
      "step": 1341
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 1341,
      "training_step_time": 9.938910484313965
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 3.13344407081604,
      "step": 1341
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 1341,
      "training_step_time": 9.907252550125122
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 3.4010889530181885,
      "step": 1341
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 1341,
      "training_step_time": 10.180242538452148
    },
    {
      "epoch": 5.115509033203125e-06,
      "model_forward_time": 3.136319875717163,
      "step": 1341
    },
    {
      "epoch": 5.115509033203125e-06,
      "step": 1341,
      "training_step_time": 10.026816368103027
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 3.180271625518799,
      "step": 1342
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 1342,
      "training_step_time": 9.968801498413086
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 3.147454023361206,
      "step": 1342
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 1342,
      "training_step_time": 9.92813754081726
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 3.1319539546966553,
      "step": 1342
    },
    {
      "dynamics_loss_avg": 0.06496405284851789,
      "epoch": 5.11932373046875e-06,
      "step": 1342
    },
    {
      "action_loss_avg": 0.04346797615289688,
      "epoch": 5.11932373046875e-06,
      "step": 1342
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 1342,
      "training_step_time": 10.209184646606445
    },
    {
      "epoch": 5.11932373046875e-06,
      "model_forward_time": 3.129584312438965,
      "step": 1342
    },
    {
      "epoch": 5.11932373046875e-06,
      "step": 1342,
      "training_step_time": 10.301722288131714
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 3.1370444297790527,
      "step": 1343
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 1343,
      "training_step_time": 9.90534496307373
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 2.4237966537475586,
      "step": 1343
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 1343,
      "training_step_time": 7.4487974643707275
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 2.4250388145446777,
      "step": 1343
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 1343,
      "training_step_time": 7.451482772827148
    },
    {
      "epoch": 5.123138427734375e-06,
      "model_forward_time": 3.127014636993408,
      "step": 1343
    },
    {
      "epoch": 5.123138427734375e-06,
      "step": 1343,
      "training_step_time": 10.307972431182861
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 2.469168186187744,
      "step": 1344
    },
    {
      "epoch": 5.126953125e-06,
      "step": 1344,
      "training_step_time": 7.504093647003174
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 3.140303134918213,
      "step": 1344
    },
    {
      "epoch": 5.126953125e-06,
      "step": 1344,
      "training_step_time": 9.907492637634277
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 3.140551805496216,
      "step": 1344
    },
    {
      "epoch": 5.126953125e-06,
      "step": 1344,
      "training_step_time": 9.915876388549805
    },
    {
      "epoch": 5.126953125e-06,
      "model_forward_time": 3.1480321884155273,
      "step": 1344
    },
    {
      "epoch": 5.126953125e-06,
      "step": 1344,
      "training_step_time": 10.29093074798584
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 3.1759066581726074,
      "step": 1345
    },
    {
      "dynamics_loss_avg": 0.0828558187931776,
      "epoch": 5.130767822265625e-06,
      "step": 1345
    },
    {
      "action_loss_avg": 0.05229335585609078,
      "epoch": 5.130767822265625e-06,
      "step": 1345
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 1345,
      "training_step_time": 9.95469331741333
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 3.4123780727386475,
      "step": 1345
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 1345,
      "training_step_time": 10.19708251953125
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 3.146014928817749,
      "step": 1345
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 1345,
      "training_step_time": 9.919170141220093
    },
    {
      "epoch": 5.130767822265625e-06,
      "model_forward_time": 3.149197816848755,
      "step": 1345
    },
    {
      "epoch": 5.130767822265625e-06,
      "step": 1345,
      "training_step_time": 10.26684832572937
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 2.4752185344696045,
      "step": 1346
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 1346,
      "training_step_time": 7.507242202758789
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 3.4256644248962402,
      "step": 1346
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 1346,
      "training_step_time": 10.195387601852417
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 3.1444432735443115,
      "step": 1346
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 1346,
      "training_step_time": 9.919780969619751
    },
    {
      "epoch": 5.13458251953125e-06,
      "model_forward_time": 3.1448142528533936,
      "step": 1346
    },
    {
      "epoch": 5.13458251953125e-06,
      "step": 1346,
      "training_step_time": 10.010792970657349
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 3.1884891986846924,
      "step": 1347
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 1347,
      "training_step_time": 9.958446979522705
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 3.1397347450256348,
      "step": 1347
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 1347,
      "training_step_time": 9.907817840576172
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 3.1456995010375977,
      "step": 1347
    },
    {
      "dynamics_loss_avg": 0.08053974006325007,
      "epoch": 5.138397216796875e-06,
      "step": 1347
    },
    {
      "action_loss_avg": 0.056331484019756316,
      "epoch": 5.138397216796875e-06,
      "step": 1347
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 1347,
      "training_step_time": 9.918880462646484
    },
    {
      "epoch": 5.138397216796875e-06,
      "model_forward_time": 3.4450089931488037,
      "step": 1347
    },
    {
      "epoch": 5.138397216796875e-06,
      "step": 1347,
      "training_step_time": 10.313210725784302
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 3.1902379989624023,
      "step": 1348
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 1348,
      "training_step_time": 9.95994758605957
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 3.143825054168701,
      "step": 1348
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 1348,
      "training_step_time": 9.91454291343689
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 3.1452887058258057,
      "step": 1348
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 1348,
      "training_step_time": 9.918060541152954
    },
    {
      "epoch": 5.1422119140625e-06,
      "model_forward_time": 3.4237582683563232,
      "step": 1348
    },
    {
      "epoch": 5.1422119140625e-06,
      "step": 1348,
      "training_step_time": 10.291393995285034
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 2.449979782104492,
      "step": 1349
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 1349,
      "training_step_time": 7.4786694049835205
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 3.143977403640747,
      "step": 1349
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 1349,
      "training_step_time": 9.917434930801392
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 2.4359447956085205,
      "step": 1349
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 1349,
      "training_step_time": 7.463796138763428
    },
    {
      "epoch": 5.146026611328125e-06,
      "model_forward_time": 3.135061502456665,
      "step": 1349
    },
    {
      "epoch": 5.146026611328125e-06,
      "step": 1349,
      "training_step_time": 10.331268787384033
    },
    {
      "epoch": 5.14984130859375e-06,
      "grad_norm": 0.7088509202003479,
      "learning_rate": 1e-05,
      "loss": 0.1294,
      "step": 1350
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 3.4663009643554688,
      "step": 1350
    },
    {
      "dynamics_loss_avg": 0.08135505765676498,
      "epoch": 5.14984130859375e-06,
      "step": 1350
    },
    {
      "action_loss_avg": 0.058260877057909964,
      "epoch": 5.14984130859375e-06,
      "step": 1350
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 1350,
      "training_step_time": 10.243293762207031
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 3.1387336254119873,
      "step": 1350
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 1350,
      "training_step_time": 9.912831544876099
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 3.140113115310669,
      "step": 1350
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 1350,
      "training_step_time": 9.918352127075195
    },
    {
      "epoch": 5.14984130859375e-06,
      "model_forward_time": 3.1398720741271973,
      "step": 1350
    },
    {
      "epoch": 5.14984130859375e-06,
      "step": 1350,
      "training_step_time": 10.312988996505737
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 3.177274465560913,
      "step": 1351
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 1351,
      "training_step_time": 9.958482503890991
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 3.1419012546539307,
      "step": 1351
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 1351,
      "training_step_time": 10.194746971130371
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 3.145940065383911,
      "step": 1351
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 1351,
      "training_step_time": 9.92292046546936
    },
    {
      "epoch": 5.153656005859375e-06,
      "model_forward_time": 3.1390788555145264,
      "step": 1351
    },
    {
      "epoch": 5.153656005859375e-06,
      "step": 1351,
      "training_step_time": 10.023305892944336
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 3.1651058197021484,
      "step": 1352
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 1352,
      "training_step_time": 9.945430755615234
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 3.1408579349517822,
      "step": 1352
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 1352,
      "training_step_time": 9.917133092880249
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 3.4535155296325684,
      "step": 1352
    },
    {
      "dynamics_loss_avg": 0.0894948523491621,
      "epoch": 5.157470703125e-06,
      "step": 1352
    },
    {
      "action_loss_avg": 0.0602980924770236,
      "epoch": 5.157470703125e-06,
      "step": 1352
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 1352,
      "training_step_time": 10.22762155532837
    },
    {
      "epoch": 5.157470703125e-06,
      "model_forward_time": 3.1386163234710693,
      "step": 1352
    },
    {
      "epoch": 5.157470703125e-06,
      "step": 1352,
      "training_step_time": 10.282737970352173
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 3.1832287311553955,
      "step": 1353
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 1353,
      "training_step_time": 9.963460683822632
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 3.1964173316955566,
      "step": 1353
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 1353,
      "training_step_time": 9.975742101669312
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 3.1405508518218994,
      "step": 1353
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 1353,
      "training_step_time": 9.920346975326538
    },
    {
      "epoch": 5.161285400390625e-06,
      "model_forward_time": 3.1441195011138916,
      "step": 1353
    },
    {
      "epoch": 5.161285400390625e-06,
      "step": 1353,
      "training_step_time": 10.291593074798584
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 3.1780734062194824,
      "step": 1354
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 1354,
      "training_step_time": 9.952868461608887
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 3.1387147903442383,
      "step": 1354
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 1354,
      "training_step_time": 9.91570258140564
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 3.137725591659546,
      "step": 1354
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 1354,
      "training_step_time": 9.913394927978516
    },
    {
      "epoch": 5.16510009765625e-06,
      "model_forward_time": 3.1433238983154297,
      "step": 1354
    },
    {
      "epoch": 5.16510009765625e-06,
      "step": 1354,
      "training_step_time": 10.2955162525177
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 3.46842885017395,
      "step": 1355
    },
    {
      "dynamics_loss_avg": 0.07765733078122139,
      "epoch": 5.168914794921875e-06,
      "step": 1355
    },
    {
      "action_loss_avg": 0.056583512015640734,
      "epoch": 5.168914794921875e-06,
      "step": 1355
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 1355,
      "training_step_time": 10.243141412734985
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 3.1427299976348877,
      "step": 1355
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 1355,
      "training_step_time": 9.917075872421265
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 3.1332805156707764,
      "step": 1355
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 1355,
      "training_step_time": 9.909825325012207
    },
    {
      "epoch": 5.168914794921875e-06,
      "model_forward_time": 3.1445393562316895,
      "step": 1355
    },
    {
      "epoch": 5.168914794921875e-06,
      "step": 1355,
      "training_step_time": 10.261655330657959
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 3.1678378582000732,
      "step": 1356
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 1356,
      "training_step_time": 9.949331521987915
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 3.445950984954834,
      "step": 1356
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 1356,
      "training_step_time": 10.2225980758667
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 3.138230800628662,
      "step": 1356
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 1356,
      "training_step_time": 9.919575929641724
    },
    {
      "epoch": 5.1727294921875e-06,
      "model_forward_time": 3.1453065872192383,
      "step": 1356
    },
    {
      "epoch": 5.1727294921875e-06,
      "step": 1356,
      "training_step_time": 10.030354738235474
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 3.1874568462371826,
      "step": 1357
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 1357,
      "training_step_time": 9.97257661819458
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 2.4365992546081543,
      "step": 1357
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 1357,
      "training_step_time": 7.46931529045105
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 3.42848539352417,
      "step": 1357
    },
    {
      "dynamics_loss_avg": 0.08572853282094002,
      "epoch": 5.176544189453125e-06,
      "step": 1357
    },
    {
      "action_loss_avg": 0.07001642836257815,
      "epoch": 5.176544189453125e-06,
      "step": 1357
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 1357,
      "training_step_time": 10.22280478477478
    },
    {
      "epoch": 5.176544189453125e-06,
      "model_forward_time": 3.1503241062164307,
      "step": 1357
    },
    {
      "epoch": 5.176544189453125e-06,
      "step": 1357,
      "training_step_time": 10.020486116409302
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 3.167689085006714,
      "step": 1358
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 1358,
      "training_step_time": 9.946494579315186
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 3.1415300369262695,
      "step": 1358
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 1358,
      "training_step_time": 9.91248083114624
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 3.136000156402588,
      "step": 1358
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 1358,
      "training_step_time": 9.90529489517212
    },
    {
      "epoch": 5.18035888671875e-06,
      "model_forward_time": 3.1379878520965576,
      "step": 1358
    },
    {
      "epoch": 5.18035888671875e-06,
      "step": 1358,
      "training_step_time": 10.340946435928345
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 3.180772066116333,
      "step": 1359
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 1359,
      "training_step_time": 9.955390453338623
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 3.1361541748046875,
      "step": 1359
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 1359,
      "training_step_time": 9.908005714416504
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 3.142589569091797,
      "step": 1359
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 1359,
      "training_step_time": 9.922977924346924
    },
    {
      "epoch": 5.184173583984375e-06,
      "model_forward_time": 3.140237331390381,
      "step": 1359
    },
    {
      "epoch": 5.184173583984375e-06,
      "step": 1359,
      "training_step_time": 10.283555507659912
    },
    {
      "epoch": 5.18798828125e-06,
      "grad_norm": 0.4120410978794098,
      "learning_rate": 1e-05,
      "loss": 0.1337,
      "step": 1360
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 3.497570514678955,
      "step": 1360
    },
    {
      "dynamics_loss_avg": 0.0775916488841176,
      "epoch": 5.18798828125e-06,
      "step": 1360
    },
    {
      "action_loss_avg": 0.057892752531915906,
      "epoch": 5.18798828125e-06,
      "step": 1360
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 1360,
      "training_step_time": 10.278486013412476
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 3.1405322551727295,
      "step": 1360
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 1360,
      "training_step_time": 9.922885417938232
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 3.1491549015045166,
      "step": 1360
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 1360,
      "training_step_time": 9.924432277679443
    },
    {
      "epoch": 5.18798828125e-06,
      "model_forward_time": 3.142596483230591,
      "step": 1360
    },
    {
      "epoch": 5.18798828125e-06,
      "step": 1360,
      "training_step_time": 10.271864175796509
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 3.159130811691284,
      "step": 1361
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 1361,
      "training_step_time": 9.929135799407959
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 3.4057328701019287,
      "step": 1361
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 1361,
      "training_step_time": 10.172367572784424
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 3.142996311187744,
      "step": 1361
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 1361,
      "training_step_time": 9.913448572158813
    },
    {
      "epoch": 5.191802978515625e-06,
      "model_forward_time": 3.1374142169952393,
      "step": 1361
    },
    {
      "epoch": 5.191802978515625e-06,
      "step": 1361,
      "training_step_time": 10.00305724143982
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 3.176157236099243,
      "step": 1362
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 1362,
      "training_step_time": 9.942439317703247
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 3.1441147327423096,
      "step": 1362
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 1362,
      "training_step_time": 9.912603855133057
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 3.466871976852417,
      "step": 1362
    },
    {
      "dynamics_loss_avg": 0.0752921236678958,
      "epoch": 5.19561767578125e-06,
      "step": 1362
    },
    {
      "action_loss_avg": 0.0539333887398243,
      "epoch": 5.19561767578125e-06,
      "step": 1362
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 1362,
      "training_step_time": 10.23522424697876
    },
    {
      "epoch": 5.19561767578125e-06,
      "model_forward_time": 3.144543409347534,
      "step": 1362
    },
    {
      "epoch": 5.19561767578125e-06,
      "step": 1362,
      "training_step_time": 10.022971630096436
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 3.190330743789673,
      "step": 1363
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 1363,
      "training_step_time": 9.969476699829102
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 3.145120859146118,
      "step": 1363
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 1363,
      "training_step_time": 9.917897939682007
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 3.142987012863159,
      "step": 1363
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 1363,
      "training_step_time": 9.918166160583496
    },
    {
      "epoch": 5.199432373046875e-06,
      "model_forward_time": 3.406834363937378,
      "step": 1363
    },
    {
      "epoch": 5.199432373046875e-06,
      "step": 1363,
      "training_step_time": 10.272244215011597
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 3.178016185760498,
      "step": 1364
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 1364,
      "training_step_time": 9.957340717315674
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 2.4402430057525635,
      "step": 1364
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 1364,
      "training_step_time": 7.471896171569824
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 3.1391162872314453,
      "step": 1364
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 1364,
      "training_step_time": 9.902600526809692
    },
    {
      "epoch": 5.2032470703125e-06,
      "model_forward_time": 3.144040822982788,
      "step": 1364
    },
    {
      "epoch": 5.2032470703125e-06,
      "step": 1364,
      "training_step_time": 10.291407585144043
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 2.4509024620056152,
      "step": 1365
    },
    {
      "dynamics_loss_avg": 0.0820688720792532,
      "epoch": 5.207061767578125e-06,
      "step": 1365
    },
    {
      "action_loss_avg": 0.05630968492478132,
      "epoch": 5.207061767578125e-06,
      "step": 1365
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 1365,
      "training_step_time": 7.476483583450317
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 3.4448513984680176,
      "step": 1365
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 1365,
      "training_step_time": 10.211318016052246
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 2.435542345046997,
      "step": 1365
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 1365,
      "training_step_time": 7.4629786014556885
    },
    {
      "epoch": 5.207061767578125e-06,
      "model_forward_time": 3.137347936630249,
      "step": 1365
    },
    {
      "epoch": 5.207061767578125e-06,
      "step": 1365,
      "training_step_time": 10.01401662826538
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 3.1884524822235107,
      "step": 1366
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 1366,
      "training_step_time": 9.957206726074219
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 3.4164083003997803,
      "step": 1366
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 1366,
      "training_step_time": 10.184985160827637
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 3.1312763690948486,
      "step": 1366
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 1366,
      "training_step_time": 9.902649879455566
    },
    {
      "epoch": 5.21087646484375e-06,
      "model_forward_time": 3.1360795497894287,
      "step": 1366
    },
    {
      "epoch": 5.21087646484375e-06,
      "step": 1366,
      "training_step_time": 10.272035360336304
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 2.459120750427246,
      "step": 1367
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 1367,
      "training_step_time": 7.484192609786987
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 3.128852367401123,
      "step": 1367
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 1367,
      "training_step_time": 9.897199153900146
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 3.137796401977539,
      "step": 1367
    },
    {
      "dynamics_loss_avg": 0.056078726053237916,
      "epoch": 5.214691162109375e-06,
      "step": 1367
    },
    {
      "action_loss_avg": 0.0376394453458488,
      "epoch": 5.214691162109375e-06,
      "step": 1367
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 1367,
      "training_step_time": 9.910244464874268
    },
    {
      "epoch": 5.214691162109375e-06,
      "model_forward_time": 3.4034371376037598,
      "step": 1367
    },
    {
      "epoch": 5.214691162109375e-06,
      "step": 1367,
      "training_step_time": 10.26901364326477
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 3.180032730102539,
      "step": 1368
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 1368,
      "training_step_time": 9.952337503433228
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 3.1311569213867188,
      "step": 1368
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 1368,
      "training_step_time": 9.899181842803955
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 3.1364994049072266,
      "step": 1368
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 1368,
      "training_step_time": 9.911265134811401
    },
    {
      "epoch": 5.218505859375e-06,
      "model_forward_time": 3.1383373737335205,
      "step": 1368
    },
    {
      "epoch": 5.218505859375e-06,
      "step": 1368,
      "training_step_time": 10.332277774810791
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 3.169215679168701,
      "step": 1369
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 1369,
      "training_step_time": 9.944483995437622
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 3.1370673179626465,
      "step": 1369
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 1369,
      "training_step_time": 9.905962467193604
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 3.139638900756836,
      "step": 1369
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 1369,
      "training_step_time": 9.910257577896118
    },
    {
      "epoch": 5.222320556640625e-06,
      "model_forward_time": 3.1375701427459717,
      "step": 1369
    },
    {
      "epoch": 5.222320556640625e-06,
      "step": 1369,
      "training_step_time": 10.274539470672607
    },
    {
      "epoch": 5.22613525390625e-06,
      "grad_norm": 0.5999894142150879,
      "learning_rate": 1e-05,
      "loss": 0.1237,
      "step": 1370
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 3.178131103515625,
      "step": 1370
    },
    {
      "dynamics_loss_avg": 0.0779135663062334,
      "epoch": 5.22613525390625e-06,
      "step": 1370
    },
    {
      "action_loss_avg": 0.04970535598695278,
      "epoch": 5.22613525390625e-06,
      "step": 1370
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 1370,
      "training_step_time": 10.216082572937012
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 3.137129306793213,
      "step": 1370
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 1370,
      "training_step_time": 9.909739017486572
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 3.138570785522461,
      "step": 1370
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 1370,
      "training_step_time": 9.906201124191284
    },
    {
      "epoch": 5.22613525390625e-06,
      "model_forward_time": 3.13507342338562,
      "step": 1370
    },
    {
      "epoch": 5.22613525390625e-06,
      "step": 1370,
      "training_step_time": 10.276569604873657
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 2.442225694656372,
      "step": 1371
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 1371,
      "training_step_time": 7.466604948043823
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 3.1279969215393066,
      "step": 1371
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 1371,
      "training_step_time": 10.156164646148682
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 3.134209156036377,
      "step": 1371
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 1371,
      "training_step_time": 9.899572372436523
    },
    {
      "epoch": 5.229949951171875e-06,
      "model_forward_time": 2.430776834487915,
      "step": 1371
    },
    {
      "epoch": 5.229949951171875e-06,
      "step": 1371,
      "training_step_time": 10.31064248085022
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 3.1656103134155273,
      "step": 1372
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 1372,
      "training_step_time": 9.931589841842651
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 3.1344258785247803,
      "step": 1372
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 1372,
      "training_step_time": 9.897221088409424
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 3.43664288520813,
      "step": 1372
    },
    {
      "dynamics_loss_avg": 0.0715832494199276,
      "epoch": 5.2337646484375e-06,
      "step": 1372
    },
    {
      "action_loss_avg": 0.05869686342775822,
      "epoch": 5.2337646484375e-06,
      "step": 1372
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 1372,
      "training_step_time": 10.197692632675171
    },
    {
      "epoch": 5.2337646484375e-06,
      "model_forward_time": 3.1328964233398438,
      "step": 1372
    },
    {
      "epoch": 5.2337646484375e-06,
      "step": 1372,
      "training_step_time": 10.245146751403809
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 3.180025577545166,
      "step": 1373
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 1373,
      "training_step_time": 9.947169303894043
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 3.1347129344940186,
      "step": 1373
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 1373,
      "training_step_time": 9.904892921447754
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 3.1325881481170654,
      "step": 1373
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 1373,
      "training_step_time": 9.902598857879639
    },
    {
      "epoch": 5.237579345703125e-06,
      "model_forward_time": 3.1385107040405273,
      "step": 1373
    },
    {
      "epoch": 5.237579345703125e-06,
      "step": 1373,
      "training_step_time": 10.288208723068237
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 2.464311361312866,
      "step": 1374
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 1374,
      "training_step_time": 7.496547698974609
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 3.1344423294067383,
      "step": 1374
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 1374,
      "training_step_time": 9.913923025131226
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 3.1424059867858887,
      "step": 1374
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 1374,
      "training_step_time": 9.92020869255066
    },
    {
      "epoch": 5.24139404296875e-06,
      "model_forward_time": 3.140284299850464,
      "step": 1374
    },
    {
      "epoch": 5.24139404296875e-06,
      "step": 1374,
      "training_step_time": 10.319850206375122
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 3.4249660968780518,
      "step": 1375
    },
    {
      "dynamics_loss_avg": 0.07864516470581293,
      "epoch": 5.245208740234375e-06,
      "step": 1375
    },
    {
      "action_loss_avg": 0.05423945132642984,
      "epoch": 5.245208740234375e-06,
      "step": 1375
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 1375,
      "training_step_time": 10.2008957862854
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 3.136704921722412,
      "step": 1375
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 1375,
      "training_step_time": 9.908021450042725
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 3.1338160037994385,
      "step": 1375
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 1375,
      "training_step_time": 9.906275510787964
    },
    {
      "epoch": 5.245208740234375e-06,
      "model_forward_time": 3.1355011463165283,
      "step": 1375
    },
    {
      "epoch": 5.245208740234375e-06,
      "step": 1375,
      "training_step_time": 10.34177541732788
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 2.446795701980591,
      "step": 1376
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 1376,
      "training_step_time": 7.475550413131714
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 3.126776695251465,
      "step": 1376
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 1376,
      "training_step_time": 9.897135972976685
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 2.728550910949707,
      "step": 1376
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 1376,
      "training_step_time": 7.754192113876343
    },
    {
      "epoch": 5.2490234375e-06,
      "model_forward_time": 3.132864475250244,
      "step": 1376
    },
    {
      "epoch": 5.2490234375e-06,
      "step": 1376,
      "training_step_time": 10.028922080993652
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 3.1646783351898193,
      "step": 1377
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 1377,
      "training_step_time": 9.932928562164307
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 3.130009651184082,
      "step": 1377
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 1377,
      "training_step_time": 9.912597417831421
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 3.1377604007720947,
      "step": 1377
    },
    {
      "dynamics_loss_avg": 0.0759549729526043,
      "epoch": 5.252838134765625e-06,
      "step": 1377
    },
    {
      "action_loss_avg": 0.05490999650210142,
      "epoch": 5.252838134765625e-06,
      "step": 1377
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 1377,
      "training_step_time": 10.183796644210815
    },
    {
      "epoch": 5.252838134765625e-06,
      "model_forward_time": 3.1401453018188477,
      "step": 1377
    },
    {
      "epoch": 5.252838134765625e-06,
      "step": 1377,
      "training_step_time": 10.29783320426941
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 3.1907970905303955,
      "step": 1378
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 1378,
      "training_step_time": 9.963582277297974
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 3.1344492435455322,
      "step": 1378
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 1378,
      "training_step_time": 9.905143737792969
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 3.136147975921631,
      "step": 1378
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 1378,
      "training_step_time": 9.91001033782959
    },
    {
      "epoch": 5.25665283203125e-06,
      "model_forward_time": 3.13847279548645,
      "step": 1378
    },
    {
      "epoch": 5.25665283203125e-06,
      "step": 1378,
      "training_step_time": 10.294350862503052
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 2.4579997062683105,
      "step": 1379
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 1379,
      "training_step_time": 7.484802961349487
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 2.4272403717041016,
      "step": 1379
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 1379,
      "training_step_time": 7.4562177658081055
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 2.431950092315674,
      "step": 1379
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 1379,
      "training_step_time": 7.457370758056641
    },
    {
      "epoch": 5.260467529296875e-06,
      "model_forward_time": 3.133596420288086,
      "step": 1379
    },
    {
      "epoch": 5.260467529296875e-06,
      "step": 1379,
      "training_step_time": 10.262161016464233
    },
    {
      "epoch": 5.2642822265625e-06,
      "grad_norm": 0.9989472031593323,
      "learning_rate": 1e-05,
      "loss": 0.124,
      "step": 1380
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 3.222386360168457,
      "step": 1380
    },
    {
      "dynamics_loss_avg": 0.08118661288172006,
      "epoch": 5.2642822265625e-06,
      "step": 1380
    },
    {
      "action_loss_avg": 0.05054710432887077,
      "epoch": 5.2642822265625e-06,
      "step": 1380
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 1380,
      "training_step_time": 10.003498315811157
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 2.7443888187408447,
      "step": 1380
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 1380,
      "training_step_time": 7.77503514289856
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 3.1344404220581055,
      "step": 1380
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 1380,
      "training_step_time": 9.904047966003418
    },
    {
      "epoch": 5.2642822265625e-06,
      "model_forward_time": 3.134242296218872,
      "step": 1380
    },
    {
      "epoch": 5.2642822265625e-06,
      "step": 1380,
      "training_step_time": 10.289377212524414
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 3.143359661102295,
      "step": 1381
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 1381,
      "training_step_time": 9.91367244720459
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 3.1330103874206543,
      "step": 1381
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 1381,
      "training_step_time": 9.906634330749512
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 2.7063212394714355,
      "step": 1381
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 1381,
      "training_step_time": 7.734465599060059
    },
    {
      "epoch": 5.268096923828125e-06,
      "model_forward_time": 3.1277053356170654,
      "step": 1381
    },
    {
      "epoch": 5.268096923828125e-06,
      "step": 1381,
      "training_step_time": 10.320174217224121
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 3.157705307006836,
      "step": 1382
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 1382,
      "training_step_time": 9.932686805725098
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 3.1326024532318115,
      "step": 1382
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 1382,
      "training_step_time": 9.903154373168945
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 3.131744146347046,
      "step": 1382
    },
    {
      "dynamics_loss_avg": 0.07706283889710903,
      "epoch": 5.27191162109375e-06,
      "step": 1382
    },
    {
      "action_loss_avg": 0.06518280040472746,
      "epoch": 5.27191162109375e-06,
      "step": 1382
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 1382,
      "training_step_time": 9.903156518936157
    },
    {
      "epoch": 5.27191162109375e-06,
      "model_forward_time": 3.376676321029663,
      "step": 1382
    },
    {
      "epoch": 5.27191162109375e-06,
      "step": 1382,
      "training_step_time": 10.287087440490723
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 3.166710615158081,
      "step": 1383
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 1383,
      "training_step_time": 9.942882537841797
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 3.135078191757202,
      "step": 1383
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 1383,
      "training_step_time": 9.905717134475708
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 3.1280651092529297,
      "step": 1383
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 1383,
      "training_step_time": 9.896764755249023
    },
    {
      "epoch": 5.275726318359375e-06,
      "model_forward_time": 3.135965585708618,
      "step": 1383
    },
    {
      "epoch": 5.275726318359375e-06,
      "step": 1383,
      "training_step_time": 10.023858308792114
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 3.4803617000579834,
      "step": 1384
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 1384,
      "training_step_time": 10.250596046447754
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 3.1364846229553223,
      "step": 1384
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 1384,
      "training_step_time": 9.911946773529053
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 3.1399343013763428,
      "step": 1384
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 1384,
      "training_step_time": 9.910171270370483
    },
    {
      "epoch": 5.279541015625e-06,
      "model_forward_time": 3.1397650241851807,
      "step": 1384
    },
    {
      "epoch": 5.279541015625e-06,
      "step": 1384,
      "training_step_time": 10.248266696929932
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 3.1757709980010986,
      "step": 1385
    },
    {
      "dynamics_loss_avg": 0.07354549299925565,
      "epoch": 5.283355712890625e-06,
      "step": 1385
    },
    {
      "action_loss_avg": 0.03890540171414614,
      "epoch": 5.283355712890625e-06,
      "step": 1385
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 1385,
      "training_step_time": 9.952162742614746
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 3.4225428104400635,
      "step": 1385
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 1385,
      "training_step_time": 10.193530797958374
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 3.1386799812316895,
      "step": 1385
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 1385,
      "training_step_time": 9.910341262817383
    },
    {
      "epoch": 5.283355712890625e-06,
      "model_forward_time": 3.1504337787628174,
      "step": 1385
    },
    {
      "epoch": 5.283355712890625e-06,
      "step": 1385,
      "training_step_time": 10.309556007385254
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 3.1699352264404297,
      "step": 1386
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 1386,
      "training_step_time": 9.956055164337158
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 3.423389196395874,
      "step": 1386
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 1386,
      "training_step_time": 10.195002794265747
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 3.138228178024292,
      "step": 1386
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 1386,
      "training_step_time": 9.91154146194458
    },
    {
      "epoch": 5.28717041015625e-06,
      "model_forward_time": 2.435512065887451,
      "step": 1386
    },
    {
      "epoch": 5.28717041015625e-06,
      "step": 1386,
      "training_step_time": 10.341437816619873
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 3.1906919479370117,
      "step": 1387
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 1387,
      "training_step_time": 9.968771934509277
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 3.1899566650390625,
      "step": 1387
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 1387,
      "training_step_time": 9.968185663223267
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 3.137622117996216,
      "step": 1387
    },
    {
      "dynamics_loss_avg": 0.0737291194498539,
      "epoch": 5.290985107421875e-06,
      "step": 1387
    },
    {
      "action_loss_avg": 0.046655723825097084,
      "epoch": 5.290985107421875e-06,
      "step": 1387
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 1387,
      "training_step_time": 9.920100450515747
    },
    {
      "epoch": 5.290985107421875e-06,
      "model_forward_time": 3.4476678371429443,
      "step": 1387
    },
    {
      "epoch": 5.290985107421875e-06,
      "step": 1387,
      "training_step_time": 10.316636323928833
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 3.1881768703460693,
      "step": 1388
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 1388,
      "training_step_time": 9.969626426696777
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 3.1418745517730713,
      "step": 1388
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 1388,
      "training_step_time": 9.920438289642334
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 3.1459577083587646,
      "step": 1388
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 1388,
      "training_step_time": 9.92572808265686
    },
    {
      "epoch": 5.2947998046875e-06,
      "model_forward_time": 3.1411292552948,
      "step": 1388
    },
    {
      "epoch": 5.2947998046875e-06,
      "step": 1388,
      "training_step_time": 10.306093454360962
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 3.192018508911133,
      "step": 1389
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 1389,
      "training_step_time": 9.973153352737427
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 2.439155101776123,
      "step": 1389
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 1389,
      "training_step_time": 7.468548774719238
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 3.142406940460205,
      "step": 1389
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 1389,
      "training_step_time": 9.9127779006958
    },
    {
      "epoch": 5.298614501953125e-06,
      "model_forward_time": 3.1378414630889893,
      "step": 1389
    },
    {
      "epoch": 5.298614501953125e-06,
      "step": 1389,
      "training_step_time": 10.015413522720337
    },
    {
      "epoch": 5.30242919921875e-06,
      "grad_norm": 0.555070161819458,
      "learning_rate": 1e-05,
      "loss": 0.1261,
      "step": 1390
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 3.1663105487823486,
      "step": 1390
    },
    {
      "dynamics_loss_avg": 0.0648940946906805,
      "epoch": 5.30242919921875e-06,
      "step": 1390
    },
    {
      "action_loss_avg": 0.04963735938072204,
      "epoch": 5.30242919921875e-06,
      "step": 1390
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 1390,
      "training_step_time": 9.942227125167847
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 3.386286973953247,
      "step": 1390
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 1390,
      "training_step_time": 10.158636093139648
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 3.1438608169555664,
      "step": 1390
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 1390,
      "training_step_time": 9.915183305740356
    },
    {
      "epoch": 5.30242919921875e-06,
      "model_forward_time": 3.1415982246398926,
      "step": 1390
    },
    {
      "epoch": 5.30242919921875e-06,
      "step": 1390,
      "training_step_time": 10.285522222518921
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 3.176370859146118,
      "step": 1391
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 1391,
      "training_step_time": 9.94891357421875
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 3.1401286125183105,
      "step": 1391
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 1391,
      "training_step_time": 10.20783257484436
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 3.140254259109497,
      "step": 1391
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 1391,
      "training_step_time": 9.917474031448364
    },
    {
      "epoch": 5.306243896484375e-06,
      "model_forward_time": 3.138444185256958,
      "step": 1391
    },
    {
      "epoch": 5.306243896484375e-06,
      "step": 1391,
      "training_step_time": 10.303683996200562
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 3.167353868484497,
      "step": 1392
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 1392,
      "training_step_time": 9.935096502304077
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 3.1400303840637207,
      "step": 1392
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 1392,
      "training_step_time": 9.908961772918701
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 3.4261703491210938,
      "step": 1392
    },
    {
      "dynamics_loss_avg": 0.06097187474370003,
      "epoch": 5.31005859375e-06,
      "step": 1392
    },
    {
      "action_loss_avg": 0.049315666779875755,
      "epoch": 5.31005859375e-06,
      "step": 1392
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 1392,
      "training_step_time": 10.196216344833374
    },
    {
      "epoch": 5.31005859375e-06,
      "model_forward_time": 3.138782262802124,
      "step": 1392
    },
    {
      "epoch": 5.31005859375e-06,
      "step": 1392,
      "training_step_time": 10.275270938873291
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 3.1629319190979004,
      "step": 1393
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 1393,
      "training_step_time": 9.938852787017822
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 3.1486570835113525,
      "step": 1393
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 1393,
      "training_step_time": 9.920154094696045
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 2.4348697662353516,
      "step": 1393
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 1393,
      "training_step_time": 7.459468841552734
    },
    {
      "epoch": 5.313873291015625e-06,
      "model_forward_time": 3.1351845264434814,
      "step": 1393
    },
    {
      "epoch": 5.313873291015625e-06,
      "step": 1393,
      "training_step_time": 10.330735206604004
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 3.4358747005462646,
      "step": 1394
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 1394,
      "training_step_time": 10.206780910491943
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 3.1419734954833984,
      "step": 1394
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 1394,
      "training_step_time": 9.91330599784851
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 2.439720392227173,
      "step": 1394
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 1394,
      "training_step_time": 7.467104911804199
    },
    {
      "epoch": 5.31768798828125e-06,
      "model_forward_time": 2.436283588409424,
      "step": 1394
    },
    {
      "epoch": 5.31768798828125e-06,
      "step": 1394,
      "training_step_time": 10.027372121810913
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 3.185422658920288,
      "step": 1395
    },
    {
      "dynamics_loss_avg": 0.07655806597322226,
      "epoch": 5.321502685546875e-06,
      "step": 1395
    },
    {
      "action_loss_avg": 0.05609101057052612,
      "epoch": 5.321502685546875e-06,
      "step": 1395
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 1395,
      "training_step_time": 9.961768865585327
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 2.740356683731079,
      "step": 1395
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 1395,
      "training_step_time": 7.76421856880188
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 3.1435065269470215,
      "step": 1395
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 1395,
      "training_step_time": 9.914509534835815
    },
    {
      "epoch": 5.321502685546875e-06,
      "model_forward_time": 3.138580560684204,
      "step": 1395
    },
    {
      "epoch": 5.321502685546875e-06,
      "step": 1395,
      "training_step_time": 10.271761178970337
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 2.4754581451416016,
      "step": 1396
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 1396,
      "training_step_time": 7.515438556671143
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 3.130384922027588,
      "step": 1396
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 1396,
      "training_step_time": 9.894938468933105
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 3.3785085678100586,
      "step": 1396
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 1396,
      "training_step_time": 10.14255142211914
    },
    {
      "epoch": 5.3253173828125e-06,
      "model_forward_time": 3.136324405670166,
      "step": 1396
    },
    {
      "epoch": 5.3253173828125e-06,
      "step": 1396,
      "training_step_time": 10.031203985214233
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 3.183748722076416,
      "step": 1397
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 1397,
      "training_step_time": 9.95063066482544
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 2.4323394298553467,
      "step": 1397
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 1397,
      "training_step_time": 7.45222544670105
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 3.1307408809661865,
      "step": 1397
    },
    {
      "dynamics_loss_avg": 0.09579540509730577,
      "epoch": 5.329132080078125e-06,
      "step": 1397
    },
    {
      "action_loss_avg": 0.06369205638766288,
      "epoch": 5.329132080078125e-06,
      "step": 1397
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 1397,
      "training_step_time": 9.893743991851807
    },
    {
      "epoch": 5.329132080078125e-06,
      "model_forward_time": 3.433802604675293,
      "step": 1397
    },
    {
      "epoch": 5.329132080078125e-06,
      "step": 1397,
      "training_step_time": 10.293670177459717
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 2.4687559604644775,
      "step": 1398
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 1398,
      "training_step_time": 7.488072156906128
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 3.1303415298461914,
      "step": 1398
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 1398,
      "training_step_time": 9.901148080825806
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 3.1388161182403564,
      "step": 1398
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 1398,
      "training_step_time": 9.90544867515564
    },
    {
      "epoch": 5.33294677734375e-06,
      "model_forward_time": 3.134244680404663,
      "step": 1398
    },
    {
      "epoch": 5.33294677734375e-06,
      "step": 1398,
      "training_step_time": 10.300343990325928
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 3.46794056892395,
      "step": 1399
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 1399,
      "training_step_time": 10.23897647857666
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 3.1364383697509766,
      "step": 1399
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 1399,
      "training_step_time": 9.911694049835205
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 3.13948655128479,
      "step": 1399
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 1399,
      "training_step_time": 9.914459228515625
    },
    {
      "epoch": 5.336761474609375e-06,
      "model_forward_time": 3.1463968753814697,
      "step": 1399
    },
    {
      "epoch": 5.336761474609375e-06,
      "step": 1399,
      "training_step_time": 10.023347616195679
    },
    {
      "epoch": 5.340576171875e-06,
      "grad_norm": 0.8520705699920654,
      "learning_rate": 1e-05,
      "loss": 0.1249,
      "step": 1400
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 2.4537341594696045,
      "step": 1400
    },
    {
      "dynamics_loss_avg": 0.08090417459607124,
      "epoch": 5.340576171875e-06,
      "step": 1400
    },
    {
      "action_loss_avg": 0.06018179170787334,
      "epoch": 5.340576171875e-06,
      "step": 1400
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 1400,
      "training_step_time": 7.484672546386719
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 3.3809800148010254,
      "step": 1400
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 1400,
      "training_step_time": 10.151727676391602
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 3.143341302871704,
      "step": 1400
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 1400,
      "training_step_time": 9.910857200622559
    },
    {
      "epoch": 5.340576171875e-06,
      "model_forward_time": 2.4391677379608154,
      "step": 1400
    },
    {
      "epoch": 5.340576171875e-06,
      "step": 1400,
      "training_step_time": 10.307548999786377
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 3.1512205600738525,
      "step": 1401
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 1401,
      "training_step_time": 9.926616430282593
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 3.141601324081421,
      "step": 1401
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 1401,
      "training_step_time": 9.909301280975342
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 3.414879083633423,
      "step": 1401
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 1401,
      "training_step_time": 10.184635400772095
    },
    {
      "epoch": 5.344390869140625e-06,
      "model_forward_time": 3.1405887603759766,
      "step": 1401
    },
    {
      "epoch": 5.344390869140625e-06,
      "step": 1401,
      "training_step_time": 10.022196292877197
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 3.1651690006256104,
      "step": 1402
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 1402,
      "training_step_time": 9.940567970275879
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 3.143313407897949,
      "step": 1402
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 1402,
      "training_step_time": 9.916020154953003
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 2.4363560676574707,
      "step": 1402
    },
    {
      "dynamics_loss_avg": 0.07768842875957489,
      "epoch": 5.34820556640625e-06,
      "step": 1402
    },
    {
      "action_loss_avg": 0.05470651183277368,
      "epoch": 5.34820556640625e-06,
      "step": 1402
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 1402,
      "training_step_time": 7.47228217124939
    },
    {
      "epoch": 5.34820556640625e-06,
      "model_forward_time": 3.4026126861572266,
      "step": 1402
    },
    {
      "epoch": 5.34820556640625e-06,
      "step": 1402,
      "training_step_time": 10.303367853164673
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 3.175765037536621,
      "step": 1403
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 1403,
      "training_step_time": 9.94544768333435
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 3.1343185901641846,
      "step": 1403
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 1403,
      "training_step_time": 9.911176681518555
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 3.1360414028167725,
      "step": 1403
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 1403,
      "training_step_time": 9.905879497528076
    },
    {
      "epoch": 5.352020263671875e-06,
      "model_forward_time": 3.397779703140259,
      "step": 1403
    },
    {
      "epoch": 5.352020263671875e-06,
      "step": 1403,
      "training_step_time": 10.288952589035034
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 3.1517770290374756,
      "step": 1404
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 1404,
      "training_step_time": 9.921548843383789
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 3.1297757625579834,
      "step": 1404
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 1404,
      "training_step_time": 9.896290063858032
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 3.13521671295166,
      "step": 1404
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 1404,
      "training_step_time": 9.900738954544067
    },
    {
      "epoch": 5.3558349609375e-06,
      "model_forward_time": 2.4336888790130615,
      "step": 1404
    },
    {
      "epoch": 5.3558349609375e-06,
      "step": 1404,
      "training_step_time": 10.296977281570435
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 3.17002272605896,
      "step": 1405
    },
    {
      "dynamics_loss_avg": 0.06435290388762951,
      "epoch": 5.359649658203125e-06,
      "step": 1405
    },
    {
      "action_loss_avg": 0.04300216101109981,
      "epoch": 5.359649658203125e-06,
      "step": 1405
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 1405,
      "training_step_time": 9.93881344795227
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 3.400104284286499,
      "step": 1405
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 1405,
      "training_step_time": 10.164912939071655
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 3.129988193511963,
      "step": 1405
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 1405,
      "training_step_time": 9.896488904953003
    },
    {
      "epoch": 5.359649658203125e-06,
      "model_forward_time": 3.130758047103882,
      "step": 1405
    },
    {
      "epoch": 5.359649658203125e-06,
      "step": 1405,
      "training_step_time": 10.26357126235962
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 3.141096830368042,
      "step": 1406
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 1406,
      "training_step_time": 9.908392906188965
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 3.4395370483398438,
      "step": 1406
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 1406,
      "training_step_time": 10.206907510757446
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 3.131704568862915,
      "step": 1406
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 1406,
      "training_step_time": 9.896979570388794
    },
    {
      "epoch": 5.36346435546875e-06,
      "model_forward_time": 3.131866455078125,
      "step": 1406
    },
    {
      "epoch": 5.36346435546875e-06,
      "step": 1406,
      "training_step_time": 10.011934280395508
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 3.1655564308166504,
      "step": 1407
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 1407,
      "training_step_time": 9.931983232498169
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 3.125999689102173,
      "step": 1407
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 1407,
      "training_step_time": 9.889864921569824
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 3.3909807205200195,
      "step": 1407
    },
    {
      "dynamics_loss_avg": 0.06052759625017643,
      "epoch": 5.367279052734375e-06,
      "step": 1407
    },
    {
      "action_loss_avg": 0.057363124750554564,
      "epoch": 5.367279052734375e-06,
      "step": 1407
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 1407,
      "training_step_time": 10.155633926391602
    },
    {
      "epoch": 5.367279052734375e-06,
      "model_forward_time": 3.1316680908203125,
      "step": 1407
    },
    {
      "epoch": 5.367279052734375e-06,
      "step": 1407,
      "training_step_time": 10.301379680633545
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 3.1674416065216064,
      "step": 1408
    },
    {
      "epoch": 5.37109375e-06,
      "step": 1408,
      "training_step_time": 9.94043517112732
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 3.131359815597534,
      "step": 1408
    },
    {
      "epoch": 5.37109375e-06,
      "step": 1408,
      "training_step_time": 9.899083137512207
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 3.1326138973236084,
      "step": 1408
    },
    {
      "epoch": 5.37109375e-06,
      "step": 1408,
      "training_step_time": 9.901857376098633
    },
    {
      "epoch": 5.37109375e-06,
      "model_forward_time": 3.4037814140319824,
      "step": 1408
    },
    {
      "epoch": 5.37109375e-06,
      "step": 1408,
      "training_step_time": 10.319917440414429
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 3.146425485610962,
      "step": 1409
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 1409,
      "training_step_time": 9.913391828536987
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 3.1363275051116943,
      "step": 1409
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 1409,
      "training_step_time": 9.904513359069824
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 3.1339144706726074,
      "step": 1409
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 1409,
      "training_step_time": 9.907789468765259
    },
    {
      "epoch": 5.374908447265625e-06,
      "model_forward_time": 2.4334678649902344,
      "step": 1409
    },
    {
      "epoch": 5.374908447265625e-06,
      "step": 1409,
      "training_step_time": 10.238980293273926
    },
    {
      "epoch": 5.37872314453125e-06,
      "grad_norm": 0.9896101355552673,
      "learning_rate": 1e-05,
      "loss": 0.1253,
      "step": 1410
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 2.510181427001953,
      "step": 1410
    },
    {
      "dynamics_loss_avg": 0.08789502680301667,
      "epoch": 5.37872314453125e-06,
      "step": 1410
    },
    {
      "action_loss_avg": 0.05594857633113861,
      "epoch": 5.37872314453125e-06,
      "step": 1410
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 1410,
      "training_step_time": 7.539895057678223
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 3.405912399291992,
      "step": 1410
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 1410,
      "training_step_time": 10.172187805175781
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 3.1350088119506836,
      "step": 1410
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 1410,
      "training_step_time": 9.910447597503662
    },
    {
      "epoch": 5.37872314453125e-06,
      "model_forward_time": 3.1344265937805176,
      "step": 1410
    },
    {
      "epoch": 5.37872314453125e-06,
      "step": 1410,
      "training_step_time": 10.241005182266235
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 3.168896198272705,
      "step": 1411
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 1411,
      "training_step_time": 9.939863204956055
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 3.1338181495666504,
      "step": 1411
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 1411,
      "training_step_time": 10.174686908721924
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 3.1361236572265625,
      "step": 1411
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 1411,
      "training_step_time": 9.906777143478394
    },
    {
      "epoch": 5.382537841796875e-06,
      "model_forward_time": 3.1337523460388184,
      "step": 1411
    },
    {
      "epoch": 5.382537841796875e-06,
      "step": 1411,
      "training_step_time": 10.018880605697632
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 3.173964023590088,
      "step": 1412
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 1412,
      "training_step_time": 9.952096939086914
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 3.139695405960083,
      "step": 1412
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 1412,
      "training_step_time": 9.918548107147217
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 3.4155173301696777,
      "step": 1412
    },
    {
      "dynamics_loss_avg": 0.08242305442690849,
      "epoch": 5.3863525390625e-06,
      "step": 1412
    },
    {
      "action_loss_avg": 0.07177776172757148,
      "epoch": 5.3863525390625e-06,
      "step": 1412
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 1412,
      "training_step_time": 10.197723865509033
    },
    {
      "epoch": 5.3863525390625e-06,
      "model_forward_time": 3.141226053237915,
      "step": 1412
    },
    {
      "epoch": 5.3863525390625e-06,
      "step": 1412,
      "training_step_time": 10.2735435962677
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 3.1719765663146973,
      "step": 1413
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 1413,
      "training_step_time": 9.947358846664429
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 3.1370909214019775,
      "step": 1413
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 1413,
      "training_step_time": 9.907233238220215
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 3.135179281234741,
      "step": 1413
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 1413,
      "training_step_time": 9.906527757644653
    },
    {
      "epoch": 5.390167236328125e-06,
      "model_forward_time": 3.1348421573638916,
      "step": 1413
    },
    {
      "epoch": 5.390167236328125e-06,
      "step": 1413,
      "training_step_time": 10.308344602584839
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 3.162006378173828,
      "step": 1414
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 1414,
      "training_step_time": 9.927622556686401
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 3.1295166015625,
      "step": 1414
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 1414,
      "training_step_time": 9.895186185836792
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 3.130523920059204,
      "step": 1414
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 1414,
      "training_step_time": 9.895563840866089
    },
    {
      "epoch": 5.39398193359375e-06,
      "model_forward_time": 3.133625030517578,
      "step": 1414
    },
    {
      "epoch": 5.39398193359375e-06,
      "step": 1414,
      "training_step_time": 10.254977941513062
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 3.4083251953125,
      "step": 1415
    },
    {
      "dynamics_loss_avg": 0.07552349064499139,
      "epoch": 5.397796630859375e-06,
      "step": 1415
    },
    {
      "action_loss_avg": 0.06865391423925757,
      "epoch": 5.397796630859375e-06,
      "step": 1415
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 1415,
      "training_step_time": 10.178096771240234
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 3.135408401489258,
      "step": 1415
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 1415,
      "training_step_time": 9.90466570854187
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 3.130418062210083,
      "step": 1415
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 1415,
      "training_step_time": 9.896711587905884
    },
    {
      "epoch": 5.397796630859375e-06,
      "model_forward_time": 3.133408784866333,
      "step": 1415
    },
    {
      "epoch": 5.397796630859375e-06,
      "step": 1415,
      "training_step_time": 10.266862869262695
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 3.1629931926727295,
      "step": 1416
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 1416,
      "training_step_time": 9.927191019058228
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 3.4404726028442383,
      "step": 1416
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 1416,
      "training_step_time": 10.207491636276245
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 3.139834403991699,
      "step": 1416
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 1416,
      "training_step_time": 9.917147636413574
    },
    {
      "epoch": 5.401611328125e-06,
      "model_forward_time": 3.1367714405059814,
      "step": 1416
    },
    {
      "epoch": 5.401611328125e-06,
      "step": 1416,
      "training_step_time": 10.313368320465088
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 3.1488139629364014,
      "step": 1417
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 1417,
      "training_step_time": 9.920295715332031
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 3.134921073913574,
      "step": 1417
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 1417,
      "training_step_time": 9.91101884841919
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 3.39192271232605,
      "step": 1417
    },
    {
      "dynamics_loss_avg": 0.0796703953295946,
      "epoch": 5.405426025390625e-06,
      "step": 1417
    },
    {
      "action_loss_avg": 0.08125720284879208,
      "epoch": 5.405426025390625e-06,
      "step": 1417
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 1417,
      "training_step_time": 10.166920900344849
    },
    {
      "epoch": 5.405426025390625e-06,
      "model_forward_time": 2.429042339324951,
      "step": 1417
    },
    {
      "epoch": 5.405426025390625e-06,
      "step": 1417,
      "training_step_time": 10.011053085327148
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 3.1853811740875244,
      "step": 1418
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 1418,
      "training_step_time": 9.954877376556396
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 3.1304233074188232,
      "step": 1418
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 1418,
      "training_step_time": 9.901842594146729
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 3.137207269668579,
      "step": 1418
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 1418,
      "training_step_time": 9.91259217262268
    },
    {
      "epoch": 5.40924072265625e-06,
      "model_forward_time": 3.1326956748962402,
      "step": 1418
    },
    {
      "epoch": 5.40924072265625e-06,
      "step": 1418,
      "training_step_time": 10.284855604171753
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 3.1855032444000244,
      "step": 1419
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 1419,
      "training_step_time": 9.959752798080444
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 3.1395280361175537,
      "step": 1419
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 1419,
      "training_step_time": 9.918245077133179
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 3.1351518630981445,
      "step": 1419
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 1419,
      "training_step_time": 9.904290199279785
    },
    {
      "epoch": 5.413055419921875e-06,
      "model_forward_time": 3.1347663402557373,
      "step": 1419
    },
    {
      "epoch": 5.413055419921875e-06,
      "step": 1419,
      "training_step_time": 10.3199782371521
    },
    {
      "epoch": 5.4168701171875e-06,
      "grad_norm": 0.744488000869751,
      "learning_rate": 1e-05,
      "loss": 0.1299,
      "step": 1420
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 3.1412813663482666,
      "step": 1420
    },
    {
      "dynamics_loss_avg": 0.08117818832397461,
      "epoch": 5.4168701171875e-06,
      "step": 1420
    },
    {
      "action_loss_avg": 0.05280002672225237,
      "epoch": 5.4168701171875e-06,
      "step": 1420
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 1420,
      "training_step_time": 10.228150844573975
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 3.135493516921997,
      "step": 1420
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 1420,
      "training_step_time": 9.909242153167725
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 3.138925313949585,
      "step": 1420
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 1420,
      "training_step_time": 9.911121129989624
    },
    {
      "epoch": 5.4168701171875e-06,
      "model_forward_time": 2.4308621883392334,
      "step": 1420
    },
    {
      "epoch": 5.4168701171875e-06,
      "step": 1420,
      "training_step_time": 10.2784583568573
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 3.1523892879486084,
      "step": 1421
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 1421,
      "training_step_time": 9.92220950126648
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 3.1324462890625,
      "step": 1421
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 1421,
      "training_step_time": 9.901034832000732
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 3.4172914028167725,
      "step": 1421
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 1421,
      "training_step_time": 10.184545755386353
    },
    {
      "epoch": 5.420684814453125e-06,
      "model_forward_time": 2.4366698265075684,
      "step": 1421
    },
    {
      "epoch": 5.420684814453125e-06,
      "step": 1421,
      "training_step_time": 10.28495454788208
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 3.1740920543670654,
      "step": 1422
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 1422,
      "training_step_time": 9.952343463897705
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 2.431933641433716,
      "step": 1422
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 1422,
      "training_step_time": 7.452971458435059
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 3.1311118602752686,
      "step": 1422
    },
    {
      "dynamics_loss_avg": 0.0766953656449914,
      "epoch": 5.42449951171875e-06,
      "step": 1422
    },
    {
      "action_loss_avg": 0.05255971411243081,
      "epoch": 5.42449951171875e-06,
      "step": 1422
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 1422,
      "training_step_time": 9.90280270576477
    },
    {
      "epoch": 5.42449951171875e-06,
      "model_forward_time": 2.701321840286255,
      "step": 1422
    },
    {
      "epoch": 5.42449951171875e-06,
      "step": 1422,
      "training_step_time": 10.019117832183838
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 3.1423487663269043,
      "step": 1423
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 1423,
      "training_step_time": 9.909606456756592
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 3.1382455825805664,
      "step": 1423
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 1423,
      "training_step_time": 9.902303695678711
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 3.1387901306152344,
      "step": 1423
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 1423,
      "training_step_time": 9.904549360275269
    },
    {
      "epoch": 5.428314208984375e-06,
      "model_forward_time": 2.4296326637268066,
      "step": 1423
    },
    {
      "epoch": 5.428314208984375e-06,
      "step": 1423,
      "training_step_time": 10.00923490524292
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 3.4531219005584717,
      "step": 1424
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 1424,
      "training_step_time": 10.225379467010498
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 3.1355535984039307,
      "step": 1424
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 1424,
      "training_step_time": 9.898597240447998
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 3.1343021392822266,
      "step": 1424
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 1424,
      "training_step_time": 9.898977041244507
    },
    {
      "epoch": 5.43212890625e-06,
      "model_forward_time": 3.131197214126587,
      "step": 1424
    },
    {
      "epoch": 5.43212890625e-06,
      "step": 1424,
      "training_step_time": 10.002426624298096
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 3.1875531673431396,
      "step": 1425
    },
    {
      "dynamics_loss_avg": 0.07667135195806622,
      "epoch": 5.435943603515625e-06,
      "step": 1425
    },
    {
      "action_loss_avg": 0.05100102312862873,
      "epoch": 5.435943603515625e-06,
      "step": 1425
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 1425,
      "training_step_time": 9.953771829605103
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 2.719569683074951,
      "step": 1425
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 1425,
      "training_step_time": 7.73539137840271
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 3.1336443424224854,
      "step": 1425
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 1425,
      "training_step_time": 9.893513679504395
    },
    {
      "epoch": 5.435943603515625e-06,
      "model_forward_time": 3.139906167984009,
      "step": 1425
    },
    {
      "epoch": 5.435943603515625e-06,
      "step": 1425,
      "training_step_time": 10.307279825210571
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 3.173443078994751,
      "step": 1426
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 1426,
      "training_step_time": 9.93931245803833
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 3.13683819770813,
      "step": 1426
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 1426,
      "training_step_time": 9.911020517349243
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 3.4011330604553223,
      "step": 1426
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 1426,
      "training_step_time": 10.175461530685425
    },
    {
      "epoch": 5.43975830078125e-06,
      "model_forward_time": 3.149087905883789,
      "step": 1426
    },
    {
      "epoch": 5.43975830078125e-06,
      "step": 1426,
      "training_step_time": 10.29241681098938
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 3.203972339630127,
      "step": 1427
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 1427,
      "training_step_time": 9.974090814590454
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 3.1383020877838135,
      "step": 1427
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 1427,
      "training_step_time": 9.910239458084106
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 3.1343674659729004,
      "step": 1427
    },
    {
      "dynamics_loss_avg": 0.07535155322402716,
      "epoch": 5.443572998046875e-06,
      "step": 1427
    },
    {
      "action_loss_avg": 0.05348838157951832,
      "epoch": 5.443572998046875e-06,
      "step": 1427
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 1427,
      "training_step_time": 9.907743692398071
    },
    {
      "epoch": 5.443572998046875e-06,
      "model_forward_time": 3.416975498199463,
      "step": 1427
    },
    {
      "epoch": 5.443572998046875e-06,
      "step": 1427,
      "training_step_time": 10.283655881881714
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 3.1702771186828613,
      "step": 1428
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 1428,
      "training_step_time": 9.939792394638062
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 3.1330373287200928,
      "step": 1428
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 1428,
      "training_step_time": 9.900570154190063
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 3.1360764503479004,
      "step": 1428
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 1428,
      "training_step_time": 9.903018712997437
    },
    {
      "epoch": 5.4473876953125e-06,
      "model_forward_time": 3.135802984237671,
      "step": 1428
    },
    {
      "epoch": 5.4473876953125e-06,
      "step": 1428,
      "training_step_time": 10.305849552154541
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 3.171079397201538,
      "step": 1429
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 1429,
      "training_step_time": 9.94532561302185
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 3.1308796405792236,
      "step": 1429
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 1429,
      "training_step_time": 9.89593243598938
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 3.136354923248291,
      "step": 1429
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 1429,
      "training_step_time": 9.902764558792114
    },
    {
      "epoch": 5.451202392578125e-06,
      "model_forward_time": 3.1334946155548096,
      "step": 1429
    },
    {
      "epoch": 5.451202392578125e-06,
      "step": 1429,
      "training_step_time": 10.277735471725464
    },
    {
      "epoch": 5.45501708984375e-06,
      "grad_norm": 1.0141652822494507,
      "learning_rate": 1e-05,
      "loss": 0.1285,
      "step": 1430
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 3.1586949825286865,
      "step": 1430
    },
    {
      "dynamics_loss_avg": 0.08516656961292028,
      "epoch": 5.45501708984375e-06,
      "step": 1430
    },
    {
      "action_loss_avg": 0.05938493562862277,
      "epoch": 5.45501708984375e-06,
      "step": 1430
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 1430,
      "training_step_time": 10.196957349777222
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 3.13039231300354,
      "step": 1430
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 1430,
      "training_step_time": 9.895135879516602
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 3.1337668895721436,
      "step": 1430
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 1430,
      "training_step_time": 9.899766206741333
    },
    {
      "epoch": 5.45501708984375e-06,
      "model_forward_time": 3.1320416927337646,
      "step": 1430
    },
    {
      "epoch": 5.45501708984375e-06,
      "step": 1430,
      "training_step_time": 10.273268222808838
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 3.156106948852539,
      "step": 1431
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 1431,
      "training_step_time": 9.920878410339355
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 3.1376335620880127,
      "step": 1431
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 1431,
      "training_step_time": 10.191352605819702
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 3.1389524936676025,
      "step": 1431
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 1431,
      "training_step_time": 9.911338567733765
    },
    {
      "epoch": 5.458831787109375e-06,
      "model_forward_time": 3.136138439178467,
      "step": 1431
    },
    {
      "epoch": 5.458831787109375e-06,
      "step": 1431,
      "training_step_time": 10.280279159545898
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 3.186713218688965,
      "step": 1432
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 1432,
      "training_step_time": 9.959115505218506
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 3.139427900314331,
      "step": 1432
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 1432,
      "training_step_time": 9.913304567337036
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 3.4149398803710938,
      "step": 1432
    },
    {
      "dynamics_loss_avg": 0.07154186014086009,
      "epoch": 5.462646484375e-06,
      "step": 1432
    },
    {
      "action_loss_avg": 0.04116011615842581,
      "epoch": 5.462646484375e-06,
      "step": 1432
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 1432,
      "training_step_time": 10.226210117340088
    },
    {
      "epoch": 5.462646484375e-06,
      "model_forward_time": 3.1516270637512207,
      "step": 1432
    },
    {
      "epoch": 5.462646484375e-06,
      "step": 1432,
      "training_step_time": 10.361875057220459
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 3.1712300777435303,
      "step": 1433
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 1433,
      "training_step_time": 9.951621770858765
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 3.1414403915405273,
      "step": 1433
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 1433,
      "training_step_time": 9.919255256652832
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 3.1361517906188965,
      "step": 1433
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 1433,
      "training_step_time": 9.911575078964233
    },
    {
      "epoch": 5.466461181640625e-06,
      "model_forward_time": 2.4350926876068115,
      "step": 1433
    },
    {
      "epoch": 5.466461181640625e-06,
      "step": 1433,
      "training_step_time": 10.3130464553833
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 3.4925003051757812,
      "step": 1434
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 1434,
      "training_step_time": 10.268823146820068
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 3.1375961303710938,
      "step": 1434
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 1434,
      "training_step_time": 9.915347337722778
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 3.1364424228668213,
      "step": 1434
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 1434,
      "training_step_time": 9.920209884643555
    },
    {
      "epoch": 5.47027587890625e-06,
      "model_forward_time": 3.1437299251556396,
      "step": 1434
    },
    {
      "epoch": 5.47027587890625e-06,
      "step": 1434,
      "training_step_time": 10.046695232391357
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 3.4662668704986572,
      "step": 1435
    },
    {
      "dynamics_loss_avg": 0.07528861351311207,
      "epoch": 5.474090576171875e-06,
      "step": 1435
    },
    {
      "action_loss_avg": 0.05182753745466471,
      "epoch": 5.474090576171875e-06,
      "step": 1435
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 1435,
      "training_step_time": 10.24732232093811
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 3.141633987426758,
      "step": 1435
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 1435,
      "training_step_time": 9.91696047782898
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 3.137701988220215,
      "step": 1435
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 1435,
      "training_step_time": 9.916876316070557
    },
    {
      "epoch": 5.474090576171875e-06,
      "model_forward_time": 3.150161027908325,
      "step": 1435
    },
    {
      "epoch": 5.474090576171875e-06,
      "step": 1435,
      "training_step_time": 10.036322832107544
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 3.1662814617156982,
      "step": 1436
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 1436,
      "training_step_time": 9.955543994903564
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 3.47114634513855,
      "step": 1436
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 1436,
      "training_step_time": 10.260412693023682
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 3.14518404006958,
      "step": 1436
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 1436,
      "training_step_time": 9.92796802520752
    },
    {
      "epoch": 5.4779052734375e-06,
      "model_forward_time": 3.146528482437134,
      "step": 1436
    },
    {
      "epoch": 5.4779052734375e-06,
      "step": 1436,
      "training_step_time": 10.294312238693237
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 3.186103582382202,
      "step": 1437
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 1437,
      "training_step_time": 9.973795175552368
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 2.4361157417297363,
      "step": 1437
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 1437,
      "training_step_time": 7.467280626296997
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 3.444322109222412,
      "step": 1437
    },
    {
      "dynamics_loss_avg": 0.08061044737696647,
      "epoch": 5.481719970703125e-06,
      "step": 1437
    },
    {
      "action_loss_avg": 0.06397749409079552,
      "epoch": 5.481719970703125e-06,
      "step": 1437
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 1437,
      "training_step_time": 10.22091031074524
    },
    {
      "epoch": 5.481719970703125e-06,
      "model_forward_time": 3.141080141067505,
      "step": 1437
    },
    {
      "epoch": 5.481719970703125e-06,
      "step": 1437,
      "training_step_time": 10.325493812561035
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 3.1808226108551025,
      "step": 1438
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 1438,
      "training_step_time": 9.953794956207275
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 3.1382598876953125,
      "step": 1438
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 1438,
      "training_step_time": 9.908706903457642
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 3.14239764213562,
      "step": 1438
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 1438,
      "training_step_time": 9.91428828239441
    },
    {
      "epoch": 5.48553466796875e-06,
      "model_forward_time": 3.143474578857422,
      "step": 1438
    },
    {
      "epoch": 5.48553466796875e-06,
      "step": 1438,
      "training_step_time": 10.280780553817749
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 3.197539806365967,
      "step": 1439
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 1439,
      "training_step_time": 9.972094774246216
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 3.145988702774048,
      "step": 1439
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 1439,
      "training_step_time": 9.91887378692627
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 3.142953872680664,
      "step": 1439
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 1439,
      "training_step_time": 9.913766622543335
    },
    {
      "epoch": 5.489349365234375e-06,
      "model_forward_time": 3.1460578441619873,
      "step": 1439
    },
    {
      "epoch": 5.489349365234375e-06,
      "step": 1439,
      "training_step_time": 10.033996343612671
    },
    {
      "epoch": 5.4931640625e-06,
      "grad_norm": 0.6613829135894775,
      "learning_rate": 1e-05,
      "loss": 0.1209,
      "step": 1440
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 3.4815382957458496,
      "step": 1440
    },
    {
      "dynamics_loss_avg": 0.08167225047945977,
      "epoch": 5.4931640625e-06,
      "step": 1440
    },
    {
      "action_loss_avg": 0.0488310806453228,
      "epoch": 5.4931640625e-06,
      "step": 1440
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 1440,
      "training_step_time": 10.255812644958496
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 3.141549825668335,
      "step": 1440
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 1440,
      "training_step_time": 9.92170786857605
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 3.1506001949310303,
      "step": 1440
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 1440,
      "training_step_time": 9.92477297782898
    },
    {
      "epoch": 5.4931640625e-06,
      "model_forward_time": 3.1403586864471436,
      "step": 1440
    },
    {
      "epoch": 5.4931640625e-06,
      "step": 1440,
      "training_step_time": 10.021216630935669
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 3.179013967514038,
      "step": 1441
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 1441,
      "training_step_time": 9.950886964797974
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 3.4036428928375244,
      "step": 1441
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 1441,
      "training_step_time": 10.166805028915405
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 3.1411187648773193,
      "step": 1441
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 1441,
      "training_step_time": 9.904823541641235
    },
    {
      "epoch": 5.496978759765625e-06,
      "model_forward_time": 3.144282102584839,
      "step": 1441
    },
    {
      "epoch": 5.496978759765625e-06,
      "step": 1441,
      "training_step_time": 10.284706115722656
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 3.170459270477295,
      "step": 1442
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 1442,
      "training_step_time": 9.941411018371582
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 3.137702465057373,
      "step": 1442
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 1442,
      "training_step_time": 9.909118890762329
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 3.41403865814209,
      "step": 1442
    },
    {
      "dynamics_loss_avg": 0.09217328764498234,
      "epoch": 5.50079345703125e-06,
      "step": 1442
    },
    {
      "action_loss_avg": 0.07960044723004103,
      "epoch": 5.50079345703125e-06,
      "step": 1442
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 1442,
      "training_step_time": 10.189380407333374
    },
    {
      "epoch": 5.50079345703125e-06,
      "model_forward_time": 3.1388471126556396,
      "step": 1442
    },
    {
      "epoch": 5.50079345703125e-06,
      "step": 1442,
      "training_step_time": 10.271633386611938
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 3.1723272800445557,
      "step": 1443
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 1443,
      "training_step_time": 9.945875883102417
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 3.139535903930664,
      "step": 1443
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 1443,
      "training_step_time": 9.902549266815186
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 3.13411021232605,
      "step": 1443
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 1443,
      "training_step_time": 9.897841453552246
    },
    {
      "epoch": 5.504608154296875e-06,
      "model_forward_time": 3.4288737773895264,
      "step": 1443
    },
    {
      "epoch": 5.504608154296875e-06,
      "step": 1443,
      "training_step_time": 10.290534973144531
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 3.1728336811065674,
      "step": 1444
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 1444,
      "training_step_time": 9.946668863296509
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 3.1401398181915283,
      "step": 1444
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 1444,
      "training_step_time": 9.909003973007202
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 3.1378867626190186,
      "step": 1444
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 1444,
      "training_step_time": 9.90513014793396
    },
    {
      "epoch": 5.5084228515625e-06,
      "model_forward_time": 3.141648530960083,
      "step": 1444
    },
    {
      "epoch": 5.5084228515625e-06,
      "step": 1444,
      "training_step_time": 10.008846282958984
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 3.4301598072052,
      "step": 1445
    },
    {
      "dynamics_loss_avg": 0.06898273341357708,
      "epoch": 5.512237548828125e-06,
      "step": 1445
    },
    {
      "action_loss_avg": 0.05553504973649979,
      "epoch": 5.512237548828125e-06,
      "step": 1445
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 1445,
      "training_step_time": 10.201714515686035
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 3.14068603515625,
      "step": 1445
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 1445,
      "training_step_time": 9.910157918930054
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 3.1440236568450928,
      "step": 1445
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 1445,
      "training_step_time": 9.916539430618286
    },
    {
      "epoch": 5.512237548828125e-06,
      "model_forward_time": 3.1420116424560547,
      "step": 1445
    },
    {
      "epoch": 5.512237548828125e-06,
      "step": 1445,
      "training_step_time": 10.022815942764282
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 3.190873146057129,
      "step": 1446
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 1446,
      "training_step_time": 9.969258785247803
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 3.4279465675354004,
      "step": 1446
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 1446,
      "training_step_time": 10.20201563835144
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 3.137620449066162,
      "step": 1446
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 1446,
      "training_step_time": 9.908205032348633
    },
    {
      "epoch": 5.51605224609375e-06,
      "model_forward_time": 3.1455237865448,
      "step": 1446
    },
    {
      "epoch": 5.51605224609375e-06,
      "step": 1446,
      "training_step_time": 10.286195993423462
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 3.1845624446868896,
      "step": 1447
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 1447,
      "training_step_time": 9.952282190322876
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 3.1370553970336914,
      "step": 1447
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 1447,
      "training_step_time": 9.908362865447998
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 3.45383882522583,
      "step": 1447
    },
    {
      "dynamics_loss_avg": 0.07645603027194739,
      "epoch": 5.519866943359375e-06,
      "step": 1447
    },
    {
      "action_loss_avg": 0.057385275792330506,
      "epoch": 5.519866943359375e-06,
      "step": 1447
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 1447,
      "training_step_time": 10.222317218780518
    },
    {
      "epoch": 5.519866943359375e-06,
      "model_forward_time": 2.4366140365600586,
      "step": 1447
    },
    {
      "epoch": 5.519866943359375e-06,
      "step": 1447,
      "training_step_time": 10.326688289642334
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 3.185678720474243,
      "step": 1448
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 1448,
      "training_step_time": 9.954949617385864
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 2.4335849285125732,
      "step": 1448
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 1448,
      "training_step_time": 7.464349269866943
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 3.1347532272338867,
      "step": 1448
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 1448,
      "training_step_time": 9.900858640670776
    },
    {
      "epoch": 5.523681640625e-06,
      "model_forward_time": 3.1360387802124023,
      "step": 1448
    },
    {
      "epoch": 5.523681640625e-06,
      "step": 1448,
      "training_step_time": 10.310927391052246
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 3.1914329528808594,
      "step": 1449
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 1449,
      "training_step_time": 9.956477165222168
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 3.141481637954712,
      "step": 1449
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 1449,
      "training_step_time": 9.909417629241943
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 3.1450071334838867,
      "step": 1449
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 1449,
      "training_step_time": 9.921160221099854
    },
    {
      "epoch": 5.527496337890625e-06,
      "model_forward_time": 3.145864725112915,
      "step": 1449
    },
    {
      "epoch": 5.527496337890625e-06,
      "step": 1449,
      "training_step_time": 10.271816730499268
    },
    {
      "epoch": 5.53131103515625e-06,
      "grad_norm": 0.8826021552085876,
      "learning_rate": 1e-05,
      "loss": 0.1323,
      "step": 1450
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 3.4703805446624756,
      "step": 1450
    },
    {
      "dynamics_loss_avg": 0.07786243595182896,
      "epoch": 5.53131103515625e-06,
      "step": 1450
    },
    {
      "action_loss_avg": 0.049899305775761606,
      "epoch": 5.53131103515625e-06,
      "step": 1450
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 1450,
      "training_step_time": 10.252253770828247
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 3.144587993621826,
      "step": 1450
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 1450,
      "training_step_time": 9.916582822799683
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 2.437058687210083,
      "step": 1450
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 1450,
      "training_step_time": 7.465138912200928
    },
    {
      "epoch": 5.53131103515625e-06,
      "model_forward_time": 3.134364128112793,
      "step": 1450
    },
    {
      "epoch": 5.53131103515625e-06,
      "step": 1450,
      "training_step_time": 10.00990343093872
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 3.1873104572296143,
      "step": 1451
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 1451,
      "training_step_time": 9.958423137664795
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 3.140066623687744,
      "step": 1451
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 1451,
      "training_step_time": 10.21283483505249
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 3.134596824645996,
      "step": 1451
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 1451,
      "training_step_time": 9.9018075466156
    },
    {
      "epoch": 5.535125732421875e-06,
      "model_forward_time": 3.131531000137329,
      "step": 1451
    },
    {
      "epoch": 5.535125732421875e-06,
      "step": 1451,
      "training_step_time": 10.272461652755737
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 3.1822566986083984,
      "step": 1452
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 1452,
      "training_step_time": 9.946422100067139
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 3.1369686126708984,
      "step": 1452
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 1452,
      "training_step_time": 9.906984806060791
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 3.416651964187622,
      "step": 1452
    },
    {
      "dynamics_loss_avg": 0.07940616197884083,
      "epoch": 5.5389404296875e-06,
      "step": 1452
    },
    {
      "action_loss_avg": 0.04805106353014708,
      "epoch": 5.5389404296875e-06,
      "step": 1452
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 1452,
      "training_step_time": 10.188791036605835
    },
    {
      "epoch": 5.5389404296875e-06,
      "model_forward_time": 3.1396853923797607,
      "step": 1452
    },
    {
      "epoch": 5.5389404296875e-06,
      "step": 1452,
      "training_step_time": 10.321880340576172
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 3.1743273735046387,
      "step": 1453
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 1453,
      "training_step_time": 9.944186210632324
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 3.1395633220672607,
      "step": 1453
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 1453,
      "training_step_time": 9.906933546066284
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 3.137491464614868,
      "step": 1453
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 1453,
      "training_step_time": 9.907672882080078
    },
    {
      "epoch": 5.542755126953125e-06,
      "model_forward_time": 3.436718702316284,
      "step": 1453
    },
    {
      "epoch": 5.542755126953125e-06,
      "step": 1453,
      "training_step_time": 10.301440954208374
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 3.1860156059265137,
      "step": 1454
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 1454,
      "training_step_time": 9.97019338607788
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 2.4380671977996826,
      "step": 1454
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 1454,
      "training_step_time": 7.4674646854400635
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 3.144134759902954,
      "step": 1454
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 1454,
      "training_step_time": 9.922794818878174
    },
    {
      "epoch": 5.54656982421875e-06,
      "model_forward_time": 3.141087532043457,
      "step": 1454
    },
    {
      "epoch": 5.54656982421875e-06,
      "step": 1454,
      "training_step_time": 10.260117769241333
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 3.477153778076172,
      "step": 1455
    },
    {
      "dynamics_loss_avg": 0.08443723060190678,
      "epoch": 5.550384521484375e-06,
      "step": 1455
    },
    {
      "action_loss_avg": 0.0568636815994978,
      "epoch": 5.550384521484375e-06,
      "step": 1455
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 1455,
      "training_step_time": 10.252599239349365
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 3.147876024246216,
      "step": 1455
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 1455,
      "training_step_time": 9.926627397537231
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 3.14207124710083,
      "step": 1455
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 1455,
      "training_step_time": 9.91822075843811
    },
    {
      "epoch": 5.550384521484375e-06,
      "model_forward_time": 3.1489806175231934,
      "step": 1455
    },
    {
      "epoch": 5.550384521484375e-06,
      "step": 1455,
      "training_step_time": 10.026180744171143
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 3.17496919631958,
      "step": 1456
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 1456,
      "training_step_time": 9.948293447494507
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 3.1394104957580566,
      "step": 1456
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 1456,
      "training_step_time": 10.195965051651001
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 3.136662006378174,
      "step": 1456
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 1456,
      "training_step_time": 9.906418561935425
    },
    {
      "epoch": 5.55419921875e-06,
      "model_forward_time": 3.13690447807312,
      "step": 1456
    },
    {
      "epoch": 5.55419921875e-06,
      "step": 1456,
      "training_step_time": 10.00716233253479
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 3.156773567199707,
      "step": 1457
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 1457,
      "training_step_time": 9.93531346321106
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 3.1357369422912598,
      "step": 1457
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 1457,
      "training_step_time": 9.905820846557617
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 3.426905393600464,
      "step": 1457
    },
    {
      "dynamics_loss_avg": 0.08284749314188958,
      "epoch": 5.558013916015625e-06,
      "step": 1457
    },
    {
      "action_loss_avg": 0.05406714230775833,
      "epoch": 5.558013916015625e-06,
      "step": 1457
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 1457,
      "training_step_time": 10.201096534729004
    },
    {
      "epoch": 5.558013916015625e-06,
      "model_forward_time": 3.138171911239624,
      "step": 1457
    },
    {
      "epoch": 5.558013916015625e-06,
      "step": 1457,
      "training_step_time": 10.275079250335693
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 3.1702845096588135,
      "step": 1458
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 1458,
      "training_step_time": 9.942190170288086
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 3.1432156562805176,
      "step": 1458
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 1458,
      "training_step_time": 9.917163133621216
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 3.1469712257385254,
      "step": 1458
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 1458,
      "training_step_time": 9.923619747161865
    },
    {
      "epoch": 5.56182861328125e-06,
      "model_forward_time": 3.434497833251953,
      "step": 1458
    },
    {
      "epoch": 5.56182861328125e-06,
      "step": 1458,
      "training_step_time": 10.304545402526855
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 3.174539566040039,
      "step": 1459
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 1459,
      "training_step_time": 9.947147369384766
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 3.1374027729034424,
      "step": 1459
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 1459,
      "training_step_time": 9.908576488494873
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 3.1391241550445557,
      "step": 1459
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 1459,
      "training_step_time": 9.915892124176025
    },
    {
      "epoch": 5.565643310546875e-06,
      "model_forward_time": 3.1376454830169678,
      "step": 1459
    },
    {
      "epoch": 5.565643310546875e-06,
      "step": 1459,
      "training_step_time": 10.016244649887085
    },
    {
      "epoch": 5.5694580078125e-06,
      "grad_norm": 0.8693417906761169,
      "learning_rate": 1e-05,
      "loss": 0.122,
      "step": 1460
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 3.1712636947631836,
      "step": 1460
    },
    {
      "dynamics_loss_avg": 0.06277834437787533,
      "epoch": 5.5694580078125e-06,
      "step": 1460
    },
    {
      "action_loss_avg": 0.04237244091928005,
      "epoch": 5.5694580078125e-06,
      "step": 1460
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 1460,
      "training_step_time": 10.22846531867981
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 3.1321356296539307,
      "step": 1460
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 1460,
      "training_step_time": 9.903335809707642
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 3.1409311294555664,
      "step": 1460
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 1460,
      "training_step_time": 9.911041975021362
    },
    {
      "epoch": 5.5694580078125e-06,
      "model_forward_time": 3.1372222900390625,
      "step": 1460
    },
    {
      "epoch": 5.5694580078125e-06,
      "step": 1460,
      "training_step_time": 10.308074951171875
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 3.1699514389038086,
      "step": 1461
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 1461,
      "training_step_time": 9.944557428359985
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 3.4243996143341064,
      "step": 1461
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 1461,
      "training_step_time": 10.1959388256073
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 3.1408088207244873,
      "step": 1461
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 1461,
      "training_step_time": 9.92387843132019
    },
    {
      "epoch": 5.573272705078125e-06,
      "model_forward_time": 3.143573045730591,
      "step": 1461
    },
    {
      "epoch": 5.573272705078125e-06,
      "step": 1461,
      "training_step_time": 10.02251410484314
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 2.4710264205932617,
      "step": 1462
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 1462,
      "training_step_time": 7.500880718231201
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 3.139598846435547,
      "step": 1462
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 1462,
      "training_step_time": 9.909759759902954
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 3.1387479305267334,
      "step": 1462
    },
    {
      "dynamics_loss_avg": 0.07640239894390106,
      "epoch": 5.57708740234375e-06,
      "step": 1462
    },
    {
      "action_loss_avg": 0.06228226087987423,
      "epoch": 5.57708740234375e-06,
      "step": 1462
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 1462,
      "training_step_time": 10.177597999572754
    },
    {
      "epoch": 5.57708740234375e-06,
      "model_forward_time": 3.1350414752960205,
      "step": 1462
    },
    {
      "epoch": 5.57708740234375e-06,
      "step": 1462,
      "training_step_time": 10.291107416152954
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 3.142869710922241,
      "step": 1463
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 1463,
      "training_step_time": 9.912492513656616
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 3.1351799964904785,
      "step": 1463
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 1463,
      "training_step_time": 9.902814149856567
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 3.1368982791900635,
      "step": 1463
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 1463,
      "training_step_time": 9.907305240631104
    },
    {
      "epoch": 5.580902099609375e-06,
      "model_forward_time": 2.7189667224884033,
      "step": 1463
    },
    {
      "epoch": 5.580902099609375e-06,
      "step": 1463,
      "training_step_time": 10.28494381904602
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 3.173602342605591,
      "step": 1464
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 1464,
      "training_step_time": 9.94353723526001
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 2.432321786880493,
      "step": 1464
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 1464,
      "training_step_time": 7.458422899246216
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 3.1312742233276367,
      "step": 1464
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 1464,
      "training_step_time": 9.898407697677612
    },
    {
      "epoch": 5.584716796875e-06,
      "model_forward_time": 3.1381499767303467,
      "step": 1464
    },
    {
      "epoch": 5.584716796875e-06,
      "step": 1464,
      "training_step_time": 10.02341103553772
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 3.170229911804199,
      "step": 1465
    },
    {
      "dynamics_loss_avg": 0.0834574993699789,
      "epoch": 5.588531494140625e-06,
      "step": 1465
    },
    {
      "action_loss_avg": 0.06370308324694633,
      "epoch": 5.588531494140625e-06,
      "step": 1465
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 1465,
      "training_step_time": 9.94335389137268
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 3.418280601501465,
      "step": 1465
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 1465,
      "training_step_time": 10.19270920753479
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 3.1416468620300293,
      "step": 1465
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 1465,
      "training_step_time": 9.914597034454346
    },
    {
      "epoch": 5.588531494140625e-06,
      "model_forward_time": 3.147400379180908,
      "step": 1465
    },
    {
      "epoch": 5.588531494140625e-06,
      "step": 1465,
      "training_step_time": 10.019978761672974
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 3.185831308364868,
      "step": 1466
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 1466,
      "training_step_time": 9.963000297546387
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 3.4542694091796875,
      "step": 1466
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 1466,
      "training_step_time": 10.225694417953491
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 3.142981767654419,
      "step": 1466
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 1466,
      "training_step_time": 9.915195941925049
    },
    {
      "epoch": 5.59234619140625e-06,
      "model_forward_time": 3.1510276794433594,
      "step": 1466
    },
    {
      "epoch": 5.59234619140625e-06,
      "step": 1466,
      "training_step_time": 10.018483638763428
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 3.157186508178711,
      "step": 1467
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 1467,
      "training_step_time": 9.932544708251953
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 3.1415209770202637,
      "step": 1467
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 1467,
      "training_step_time": 9.918349981307983
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 3.147940158843994,
      "step": 1467
    },
    {
      "dynamics_loss_avg": 0.06021084189414978,
      "epoch": 5.596160888671875e-06,
      "step": 1467
    },
    {
      "action_loss_avg": 0.05394438337534666,
      "epoch": 5.596160888671875e-06,
      "step": 1467
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 1467,
      "training_step_time": 9.925945043563843
    },
    {
      "epoch": 5.596160888671875e-06,
      "model_forward_time": 2.716014862060547,
      "step": 1467
    },
    {
      "epoch": 5.596160888671875e-06,
      "step": 1467,
      "training_step_time": 10.020687580108643
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 3.187039613723755,
      "step": 1468
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 1468,
      "training_step_time": 9.96040153503418
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 3.1436209678649902,
      "step": 1468
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 1468,
      "training_step_time": 9.920613765716553
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 3.1368613243103027,
      "step": 1468
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 1468,
      "training_step_time": 9.911216020584106
    },
    {
      "epoch": 5.5999755859375e-06,
      "model_forward_time": 3.4090304374694824,
      "step": 1468
    },
    {
      "epoch": 5.5999755859375e-06,
      "step": 1468,
      "training_step_time": 10.307910919189453
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 2.466691017150879,
      "step": 1469
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 1469,
      "training_step_time": 7.4929585456848145
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 3.129570960998535,
      "step": 1469
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 1469,
      "training_step_time": 9.90212082862854
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 3.133357286453247,
      "step": 1469
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 1469,
      "training_step_time": 9.901005744934082
    },
    {
      "epoch": 5.603790283203125e-06,
      "model_forward_time": 3.1309752464294434,
      "step": 1469
    },
    {
      "epoch": 5.603790283203125e-06,
      "step": 1469,
      "training_step_time": 10.284746408462524
    },
    {
      "epoch": 5.60760498046875e-06,
      "grad_norm": 0.7031849026679993,
      "learning_rate": 1e-05,
      "loss": 0.1243,
      "step": 1470
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 3.4774630069732666,
      "step": 1470
    },
    {
      "dynamics_loss_avg": 0.08399158641695977,
      "epoch": 5.60760498046875e-06,
      "step": 1470
    },
    {
      "action_loss_avg": 0.06298774452880025,
      "epoch": 5.60760498046875e-06,
      "step": 1470
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 1470,
      "training_step_time": 10.243444919586182
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 3.1329469680786133,
      "step": 1470
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 1470,
      "training_step_time": 9.90483045578003
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 3.1359989643096924,
      "step": 1470
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 1470,
      "training_step_time": 9.909056901931763
    },
    {
      "epoch": 5.60760498046875e-06,
      "model_forward_time": 3.136739730834961,
      "step": 1470
    },
    {
      "epoch": 5.60760498046875e-06,
      "step": 1470,
      "training_step_time": 10.008509874343872
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 3.1783981323242188,
      "step": 1471
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 1471,
      "training_step_time": 9.948879480361938
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 3.1372697353363037,
      "step": 1471
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 1471,
      "training_step_time": 10.196000099182129
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 3.136420726776123,
      "step": 1471
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 1471,
      "training_step_time": 9.905442714691162
    },
    {
      "epoch": 5.611419677734375e-06,
      "model_forward_time": 3.1332051753997803,
      "step": 1471
    },
    {
      "epoch": 5.611419677734375e-06,
      "step": 1471,
      "training_step_time": 10.016656875610352
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 3.171213150024414,
      "step": 1472
    },
    {
      "epoch": 5.615234375e-06,
      "step": 1472,
      "training_step_time": 9.937349557876587
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 3.136364459991455,
      "step": 1472
    },
    {
      "epoch": 5.615234375e-06,
      "step": 1472,
      "training_step_time": 9.90725064277649
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 3.4238803386688232,
      "step": 1472
    },
    {
      "dynamics_loss_avg": 0.061144710332155225,
      "epoch": 5.615234375e-06,
      "step": 1472
    },
    {
      "action_loss_avg": 0.05049099549651146,
      "epoch": 5.615234375e-06,
      "step": 1472
    },
    {
      "epoch": 5.615234375e-06,
      "step": 1472,
      "training_step_time": 10.199475288391113
    },
    {
      "epoch": 5.615234375e-06,
      "model_forward_time": 3.140230417251587,
      "step": 1472
    },
    {
      "epoch": 5.615234375e-06,
      "step": 1472,
      "training_step_time": 10.030146598815918
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 3.1461331844329834,
      "step": 1473
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 1473,
      "training_step_time": 9.924707412719727
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 2.4345154762268066,
      "step": 1473
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 1473,
      "training_step_time": 7.464035511016846
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 3.136185884475708,
      "step": 1473
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 1473,
      "training_step_time": 9.903900623321533
    },
    {
      "epoch": 5.619049072265625e-06,
      "model_forward_time": 2.4383394718170166,
      "step": 1473
    },
    {
      "epoch": 5.619049072265625e-06,
      "step": 1473,
      "training_step_time": 10.26753568649292
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 3.4249913692474365,
      "step": 1474
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 1474,
      "training_step_time": 10.201681137084961
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 2.434384346008301,
      "step": 1474
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 1474,
      "training_step_time": 7.457258939743042
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 3.1340787410736084,
      "step": 1474
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 1474,
      "training_step_time": 9.898383140563965
    },
    {
      "epoch": 5.62286376953125e-06,
      "model_forward_time": 2.432577133178711,
      "step": 1474
    },
    {
      "epoch": 5.62286376953125e-06,
      "step": 1474,
      "training_step_time": 10.31360650062561
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 3.155029773712158,
      "step": 1475
    },
    {
      "dynamics_loss_avg": 0.07051421105861663,
      "epoch": 5.626678466796875e-06,
      "step": 1475
    },
    {
      "action_loss_avg": 0.04553358089178801,
      "epoch": 5.626678466796875e-06,
      "step": 1475
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 1475,
      "training_step_time": 9.924060821533203
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 3.4482507705688477,
      "step": 1475
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 1475,
      "training_step_time": 10.21902060508728
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 3.1425161361694336,
      "step": 1475
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 1475,
      "training_step_time": 9.917444705963135
    },
    {
      "epoch": 5.626678466796875e-06,
      "model_forward_time": 2.4373021125793457,
      "step": 1475
    },
    {
      "epoch": 5.626678466796875e-06,
      "step": 1475,
      "training_step_time": 10.028051614761353
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 3.1584720611572266,
      "step": 1476
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 1476,
      "training_step_time": 9.927554607391357
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 2.4352242946624756,
      "step": 1476
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 1476,
      "training_step_time": 7.459753513336182
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 2.6922428607940674,
      "step": 1476
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 1476,
      "training_step_time": 7.712955951690674
    },
    {
      "epoch": 5.6304931640625e-06,
      "model_forward_time": 3.1358494758605957,
      "step": 1476
    },
    {
      "epoch": 5.6304931640625e-06,
      "step": 1476,
      "training_step_time": 10.305533647537231
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 3.147765874862671,
      "step": 1477
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 1477,
      "training_step_time": 9.919387340545654
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 3.136446237564087,
      "step": 1477
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 1477,
      "training_step_time": 9.903251647949219
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 3.1410140991210938,
      "step": 1477
    },
    {
      "dynamics_loss_avg": 0.06460719108581543,
      "epoch": 5.634307861328125e-06,
      "step": 1477
    },
    {
      "action_loss_avg": 0.046634520217776296,
      "epoch": 5.634307861328125e-06,
      "step": 1477
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 1477,
      "training_step_time": 9.91910195350647
    },
    {
      "epoch": 5.634307861328125e-06,
      "model_forward_time": 2.689816474914551,
      "step": 1477
    },
    {
      "epoch": 5.634307861328125e-06,
      "step": 1477,
      "training_step_time": 10.009587526321411
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 3.1673030853271484,
      "step": 1478
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 1478,
      "training_step_time": 9.937775373458862
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 3.1396336555480957,
      "step": 1478
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 1478,
      "training_step_time": 9.917318344116211
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 3.1443192958831787,
      "step": 1478
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 1478,
      "training_step_time": 9.919191360473633
    },
    {
      "epoch": 5.63812255859375e-06,
      "model_forward_time": 3.145092487335205,
      "step": 1478
    },
    {
      "epoch": 5.63812255859375e-06,
      "step": 1478,
      "training_step_time": 10.302885293960571
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 3.465519428253174,
      "step": 1479
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 1479,
      "training_step_time": 10.252055644989014
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 2.440917730331421,
      "step": 1479
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 1479,
      "training_step_time": 7.475301504135132
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 2.4348106384277344,
      "step": 1479
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 1479,
      "training_step_time": 7.470783948898315
    },
    {
      "epoch": 5.641937255859375e-06,
      "model_forward_time": 3.1390137672424316,
      "step": 1479
    },
    {
      "epoch": 5.641937255859375e-06,
      "step": 1479,
      "training_step_time": 10.317571878433228
    },
    {
      "epoch": 5.645751953125e-06,
      "grad_norm": 0.8558874130249023,
      "learning_rate": 1e-05,
      "loss": 0.1222,
      "step": 1480
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 3.1836485862731934,
      "step": 1480
    },
    {
      "dynamics_loss_avg": 0.05692038061097264,
      "epoch": 5.645751953125e-06,
      "step": 1480
    },
    {
      "action_loss_avg": 0.0513690774794668,
      "epoch": 5.645751953125e-06,
      "step": 1480
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 1480,
      "training_step_time": 9.958927154541016
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 3.413024425506592,
      "step": 1480
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 1480,
      "training_step_time": 10.184941291809082
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 3.145254373550415,
      "step": 1480
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 1480,
      "training_step_time": 9.919013023376465
    },
    {
      "epoch": 5.645751953125e-06,
      "model_forward_time": 3.141754388809204,
      "step": 1480
    },
    {
      "epoch": 5.645751953125e-06,
      "step": 1480,
      "training_step_time": 10.24187159538269
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 3.180473566055298,
      "step": 1481
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 1481,
      "training_step_time": 9.95948076248169
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 3.141751289367676,
      "step": 1481
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 1481,
      "training_step_time": 9.91295313835144
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 3.140610933303833,
      "step": 1481
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 1481,
      "training_step_time": 10.171925067901611
    },
    {
      "epoch": 5.649566650390625e-06,
      "model_forward_time": 3.1389966011047363,
      "step": 1481
    },
    {
      "epoch": 5.649566650390625e-06,
      "step": 1481,
      "training_step_time": 10.246764898300171
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 3.158466339111328,
      "step": 1482
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 1482,
      "training_step_time": 9.928218841552734
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 3.1407785415649414,
      "step": 1482
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 1482,
      "training_step_time": 9.912337303161621
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 3.1429450511932373,
      "step": 1482
    },
    {
      "dynamics_loss_avg": 0.06888517998158931,
      "epoch": 5.65338134765625e-06,
      "step": 1482
    },
    {
      "action_loss_avg": 0.04924879213795066,
      "epoch": 5.65338134765625e-06,
      "step": 1482
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 1482,
      "training_step_time": 9.916118621826172
    },
    {
      "epoch": 5.65338134765625e-06,
      "model_forward_time": 2.4368984699249268,
      "step": 1482
    },
    {
      "epoch": 5.65338134765625e-06,
      "step": 1482,
      "training_step_time": 10.017663717269897
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 3.1721787452697754,
      "step": 1483
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 1483,
      "training_step_time": 9.948348999023438
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 3.142627716064453,
      "step": 1483
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 1483,
      "training_step_time": 9.91702914237976
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 3.137662172317505,
      "step": 1483
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 1483,
      "training_step_time": 9.910040378570557
    },
    {
      "epoch": 5.657196044921875e-06,
      "model_forward_time": 3.141005754470825,
      "step": 1483
    },
    {
      "epoch": 5.657196044921875e-06,
      "step": 1483,
      "training_step_time": 10.012677431106567
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 3.4690468311309814,
      "step": 1484
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 1484,
      "training_step_time": 10.242778062820435
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 3.137704372406006,
      "step": 1484
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 1484,
      "training_step_time": 9.91100525856018
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 3.1353023052215576,
      "step": 1484
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 1484,
      "training_step_time": 9.90844464302063
    },
    {
      "epoch": 5.6610107421875e-06,
      "model_forward_time": 3.1379787921905518,
      "step": 1484
    },
    {
      "epoch": 5.6610107421875e-06,
      "step": 1484,
      "training_step_time": 10.287936925888062
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 3.1656203269958496,
      "step": 1485
    },
    {
      "dynamics_loss_avg": 0.05927603151649237,
      "epoch": 5.664825439453125e-06,
      "step": 1485
    },
    {
      "action_loss_avg": 0.04380470979958773,
      "epoch": 5.664825439453125e-06,
      "step": 1485
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 1485,
      "training_step_time": 9.94219708442688
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 3.133732318878174,
      "step": 1485
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 1485,
      "training_step_time": 10.175405740737915
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 3.137669563293457,
      "step": 1485
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 1485,
      "training_step_time": 9.908035039901733
    },
    {
      "epoch": 5.664825439453125e-06,
      "model_forward_time": 3.136538505554199,
      "step": 1485
    },
    {
      "epoch": 5.664825439453125e-06,
      "step": 1485,
      "training_step_time": 10.263991117477417
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 3.170142889022827,
      "step": 1486
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 1486,
      "training_step_time": 9.938430070877075
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 3.1319611072540283,
      "step": 1486
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 1486,
      "training_step_time": 9.906042575836182
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 3.435350179672241,
      "step": 1486
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 1486,
      "training_step_time": 10.199340105056763
    },
    {
      "epoch": 5.66864013671875e-06,
      "model_forward_time": 3.138063430786133,
      "step": 1486
    },
    {
      "epoch": 5.66864013671875e-06,
      "step": 1486,
      "training_step_time": 10.012670516967773
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 2.471487045288086,
      "step": 1487
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 1487,
      "training_step_time": 7.495707035064697
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 3.1310057640075684,
      "step": 1487
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 1487,
      "training_step_time": 9.897947549819946
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 3.136430025100708,
      "step": 1487
    },
    {
      "dynamics_loss_avg": 0.06322469636797905,
      "epoch": 5.672454833984375e-06,
      "step": 1487
    },
    {
      "action_loss_avg": 0.04001955427229405,
      "epoch": 5.672454833984375e-06,
      "step": 1487
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 1487,
      "training_step_time": 9.912222862243652
    },
    {
      "epoch": 5.672454833984375e-06,
      "model_forward_time": 3.1341874599456787,
      "step": 1487
    },
    {
      "epoch": 5.672454833984375e-06,
      "step": 1487,
      "training_step_time": 10.013094425201416
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 3.460012674331665,
      "step": 1488
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 1488,
      "training_step_time": 10.233339071273804
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 2.430959463119507,
      "step": 1488
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 1488,
      "training_step_time": 7.46009373664856
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 3.132025957107544,
      "step": 1488
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 1488,
      "training_step_time": 9.90146017074585
    },
    {
      "epoch": 5.67626953125e-06,
      "model_forward_time": 3.1311516761779785,
      "step": 1488
    },
    {
      "epoch": 5.67626953125e-06,
      "step": 1488,
      "training_step_time": 10.250465154647827
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 3.42781662940979,
      "step": 1489
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 1489,
      "training_step_time": 10.195027112960815
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 3.132208824157715,
      "step": 1489
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 1489,
      "training_step_time": 9.896512269973755
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 3.134242296218872,
      "step": 1489
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 1489,
      "training_step_time": 9.903399229049683
    },
    {
      "epoch": 5.680084228515625e-06,
      "model_forward_time": 2.4284651279449463,
      "step": 1489
    },
    {
      "epoch": 5.680084228515625e-06,
      "step": 1489,
      "training_step_time": 10.304095268249512
    },
    {
      "epoch": 5.68389892578125e-06,
      "grad_norm": 0.7378729581832886,
      "learning_rate": 1e-05,
      "loss": 0.1163,
      "step": 1490
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 3.151230812072754,
      "step": 1490
    },
    {
      "dynamics_loss_avg": 0.06622921153903008,
      "epoch": 5.68389892578125e-06,
      "step": 1490
    },
    {
      "action_loss_avg": 0.04880161136388779,
      "epoch": 5.68389892578125e-06,
      "step": 1490
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 1490,
      "training_step_time": 9.918442487716675
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 3.1286933422088623,
      "step": 1490
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 1490,
      "training_step_time": 9.8951735496521
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 3.444385051727295,
      "step": 1490
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 1490,
      "training_step_time": 10.211007356643677
    },
    {
      "epoch": 5.68389892578125e-06,
      "model_forward_time": 3.1330325603485107,
      "step": 1490
    },
    {
      "epoch": 5.68389892578125e-06,
      "step": 1490,
      "training_step_time": 10.310184001922607
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 3.1647098064422607,
      "step": 1491
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 1491,
      "training_step_time": 9.935320854187012
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 3.1341187953948975,
      "step": 1491
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 1491,
      "training_step_time": 9.90058183670044
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 3.1296751499176025,
      "step": 1491
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 1491,
      "training_step_time": 10.173722267150879
    },
    {
      "epoch": 5.687713623046875e-06,
      "model_forward_time": 3.1307802200317383,
      "step": 1491
    },
    {
      "epoch": 5.687713623046875e-06,
      "step": 1491,
      "training_step_time": 10.283073663711548
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 2.4735777378082275,
      "step": 1492
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 1492,
      "training_step_time": 7.493813991546631
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 3.129197359085083,
      "step": 1492
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 1492,
      "training_step_time": 9.8965322971344
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 3.129612684249878,
      "step": 1492
    },
    {
      "dynamics_loss_avg": 0.06655875518918038,
      "epoch": 5.6915283203125e-06,
      "step": 1492
    },
    {
      "action_loss_avg": 0.057362452149391174,
      "epoch": 5.6915283203125e-06,
      "step": 1492
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 1492,
      "training_step_time": 9.906231164932251
    },
    {
      "epoch": 5.6915283203125e-06,
      "model_forward_time": 3.1315956115722656,
      "step": 1492
    },
    {
      "epoch": 5.6915283203125e-06,
      "step": 1492,
      "training_step_time": 10.305297374725342
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 2.7598257064819336,
      "step": 1493
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 1493,
      "training_step_time": 7.788249492645264
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 2.4285407066345215,
      "step": 1493
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 1493,
      "training_step_time": 7.454092741012573
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 3.12644624710083,
      "step": 1493
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 1493,
      "training_step_time": 9.897259950637817
    },
    {
      "epoch": 5.695343017578125e-06,
      "model_forward_time": 2.430769443511963,
      "step": 1493
    },
    {
      "epoch": 5.695343017578125e-06,
      "step": 1493,
      "training_step_time": 10.016908645629883
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 3.168001651763916,
      "step": 1494
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 1494,
      "training_step_time": 9.939062118530273
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 2.712003707885742,
      "step": 1494
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 1494,
      "training_step_time": 7.736052989959717
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 3.127420425415039,
      "step": 1494
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 1494,
      "training_step_time": 9.89265251159668
    },
    {
      "epoch": 5.69915771484375e-06,
      "model_forward_time": 3.133174419403076,
      "step": 1494
    },
    {
      "epoch": 5.69915771484375e-06,
      "step": 1494,
      "training_step_time": 10.263520240783691
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 2.4773855209350586,
      "step": 1495
    },
    {
      "dynamics_loss_avg": 0.07790761888027191,
      "epoch": 5.702972412109375e-06,
      "step": 1495
    },
    {
      "action_loss_avg": 0.05061781760305166,
      "epoch": 5.702972412109375e-06,
      "step": 1495
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 1495,
      "training_step_time": 7.50684118270874
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 3.1280248165130615,
      "step": 1495
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 1495,
      "training_step_time": 9.900935649871826
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 3.381115674972534,
      "step": 1495
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 1495,
      "training_step_time": 10.159192085266113
    },
    {
      "epoch": 5.702972412109375e-06,
      "model_forward_time": 3.136077880859375,
      "step": 1495
    },
    {
      "epoch": 5.702972412109375e-06,
      "step": 1495,
      "training_step_time": 10.30389666557312
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 3.164095878601074,
      "step": 1496
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 1496,
      "training_step_time": 9.94038724899292
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 3.13623046875,
      "step": 1496
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 1496,
      "training_step_time": 9.909982204437256
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 3.1391377449035645,
      "step": 1496
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 1496,
      "training_step_time": 9.91686463356018
    },
    {
      "epoch": 5.706787109375e-06,
      "model_forward_time": 3.420456886291504,
      "step": 1496
    },
    {
      "epoch": 5.706787109375e-06,
      "step": 1496,
      "training_step_time": 10.290685653686523
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 3.1791179180145264,
      "step": 1497
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 1497,
      "training_step_time": 9.957048416137695
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 3.1325809955596924,
      "step": 1497
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 1497,
      "training_step_time": 9.908108949661255
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 3.139244556427002,
      "step": 1497
    },
    {
      "dynamics_loss_avg": 0.0794178694486618,
      "epoch": 5.710601806640625e-06,
      "step": 1497
    },
    {
      "action_loss_avg": 0.051846370846033094,
      "epoch": 5.710601806640625e-06,
      "step": 1497
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 1497,
      "training_step_time": 9.918992042541504
    },
    {
      "epoch": 5.710601806640625e-06,
      "model_forward_time": 3.145395517349243,
      "step": 1497
    },
    {
      "epoch": 5.710601806640625e-06,
      "step": 1497,
      "training_step_time": 10.260764360427856
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 3.486656427383423,
      "step": 1498
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 1498,
      "training_step_time": 10.271093606948853
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 3.134397268295288,
      "step": 1498
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 1498,
      "training_step_time": 9.913526773452759
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 2.430466651916504,
      "step": 1498
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 1498,
      "training_step_time": 7.459791898727417
    },
    {
      "epoch": 5.71441650390625e-06,
      "model_forward_time": 3.133120536804199,
      "step": 1498
    },
    {
      "epoch": 5.71441650390625e-06,
      "step": 1498,
      "training_step_time": 10.327358484268188
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 3.2066564559936523,
      "step": 1499
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 1499,
      "training_step_time": 9.987009048461914
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 3.386157751083374,
      "step": 1499
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 1499,
      "training_step_time": 10.167834281921387
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 3.140502691268921,
      "step": 1499
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 1499,
      "training_step_time": 9.913107872009277
    },
    {
      "epoch": 5.718231201171875e-06,
      "model_forward_time": 3.135141372680664,
      "step": 1499
    },
    {
      "epoch": 5.718231201171875e-06,
      "step": 1499,
      "training_step_time": 10.007258176803589
    },
    {
      "epoch": 5.7220458984375e-06,
      "grad_norm": 0.4544418752193451,
      "learning_rate": 1e-05,
      "loss": 0.1185,
      "step": 1500
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 3.1565988063812256,
      "step": 1500
    },
    {
      "dynamics_loss_avg": 0.06262519732117652,
      "epoch": 5.7220458984375e-06,
      "step": 1500
    },
    {
      "action_loss_avg": 0.045165525563061235,
      "epoch": 5.7220458984375e-06,
      "step": 1500
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 1500,
      "training_step_time": 9.91751480102539
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 4.8318772315979,
      "step": 1500
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 1500,
      "training_step_time": 11.604790449142456
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 3.133610725402832,
      "step": 1500
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 1500,
      "training_step_time": 9.893485069274902
    },
    {
      "epoch": 5.7220458984375e-06,
      "model_forward_time": 3.1325979232788086,
      "step": 1500
    },
    {
      "epoch": 5.7220458984375e-06,
      "step": 1500,
      "training_step_time": 10.304356813430786
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 3.1886768341064453,
      "step": 1501
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 1501,
      "training_step_time": 9.965147972106934
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 3.140040874481201,
      "step": 1501
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 1501,
      "training_step_time": 10.230603694915771
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 3.13952374458313,
      "step": 1501
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 1501,
      "training_step_time": 9.91735577583313
    },
    {
      "epoch": 5.725860595703125e-06,
      "model_forward_time": 3.1374430656433105,
      "step": 1501
    },
    {
      "epoch": 5.725860595703125e-06,
      "step": 1501,
      "training_step_time": 10.321654796600342
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 3.1766035556793213,
      "step": 1502
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 1502,
      "training_step_time": 9.9666109085083
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 3.1442015171051025,
      "step": 1502
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 1502,
      "training_step_time": 9.928044319152832
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 3.143329381942749,
      "step": 1502
    },
    {
      "dynamics_loss_avg": 0.06575882341712713,
      "epoch": 5.72967529296875e-06,
      "step": 1502
    },
    {
      "action_loss_avg": 0.05453638881444931,
      "epoch": 5.72967529296875e-06,
      "step": 1502
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 1502,
      "training_step_time": 10.241371154785156
    },
    {
      "epoch": 5.72967529296875e-06,
      "model_forward_time": 3.144132375717163,
      "step": 1502
    },
    {
      "epoch": 5.72967529296875e-06,
      "step": 1502,
      "training_step_time": 10.283333778381348
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 2.452104091644287,
      "step": 1503
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 1503,
      "training_step_time": 7.492902755737305
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 3.138638973236084,
      "step": 1503
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 1503,
      "training_step_time": 9.916518926620483
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 3.13844895362854,
      "step": 1503
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 1503,
      "training_step_time": 9.915529251098633
    },
    {
      "epoch": 5.733489990234375e-06,
      "model_forward_time": 2.4355955123901367,
      "step": 1503
    },
    {
      "epoch": 5.733489990234375e-06,
      "step": 1503,
      "training_step_time": 10.335655450820923
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 3.1683855056762695,
      "step": 1504
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 1504,
      "training_step_time": 9.947855710983276
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 3.1440749168395996,
      "step": 1504
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 1504,
      "training_step_time": 9.917885541915894
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 3.141601085662842,
      "step": 1504
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 1504,
      "training_step_time": 9.921193361282349
    },
    {
      "epoch": 5.7373046875e-06,
      "model_forward_time": 3.146364212036133,
      "step": 1504
    },
    {
      "epoch": 5.7373046875e-06,
      "step": 1504,
      "training_step_time": 10.029428958892822
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 3.18803071975708,
      "step": 1505
    },
    {
      "dynamics_loss_avg": 0.07507885191589594,
      "epoch": 5.741119384765625e-06,
      "step": 1505
    },
    {
      "action_loss_avg": 0.05465535447001457,
      "epoch": 5.741119384765625e-06,
      "step": 1505
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 1505,
      "training_step_time": 10.248471975326538
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 3.14670467376709,
      "step": 1505
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 1505,
      "training_step_time": 9.929319858551025
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 3.148374319076538,
      "step": 1505
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 1505,
      "training_step_time": 9.93209171295166
    },
    {
      "epoch": 5.741119384765625e-06,
      "model_forward_time": 2.437325954437256,
      "step": 1505
    },
    {
      "epoch": 5.741119384765625e-06,
      "step": 1505,
      "training_step_time": 10.307444095611572
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 3.192533016204834,
      "step": 1506
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 1506,
      "training_step_time": 9.971505165100098
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 2.4368457794189453,
      "step": 1506
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 1506,
      "training_step_time": 7.466876983642578
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 3.4148194789886475,
      "step": 1506
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 1506,
      "training_step_time": 10.188195943832397
    },
    {
      "epoch": 5.74493408203125e-06,
      "model_forward_time": 3.1420950889587402,
      "step": 1506
    },
    {
      "epoch": 5.74493408203125e-06,
      "step": 1506,
      "training_step_time": 10.32005763053894
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 2.4825429916381836,
      "step": 1507
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 1507,
      "training_step_time": 7.506756544113159
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 3.132206916809082,
      "step": 1507
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 1507,
      "training_step_time": 9.896161079406738
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 3.1333847045898438,
      "step": 1507
    },
    {
      "dynamics_loss_avg": 0.07341146171092987,
      "epoch": 5.748748779296875e-06,
      "step": 1507
    },
    {
      "action_loss_avg": 0.0512840498238802,
      "epoch": 5.748748779296875e-06,
      "step": 1507
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 1507,
      "training_step_time": 9.896801233291626
    },
    {
      "epoch": 5.748748779296875e-06,
      "model_forward_time": 3.4132986068725586,
      "step": 1507
    },
    {
      "epoch": 5.748748779296875e-06,
      "step": 1507,
      "training_step_time": 10.271638870239258
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 3.1742441654205322,
      "step": 1508
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 1508,
      "training_step_time": 9.941279649734497
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 3.1348185539245605,
      "step": 1508
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 1508,
      "training_step_time": 9.903321504592896
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 3.1408071517944336,
      "step": 1508
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 1508,
      "training_step_time": 9.914727210998535
    },
    {
      "epoch": 5.7525634765625e-06,
      "model_forward_time": 3.4398863315582275,
      "step": 1508
    },
    {
      "epoch": 5.7525634765625e-06,
      "step": 1508,
      "training_step_time": 10.310269117355347
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 2.446136474609375,
      "step": 1509
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 1509,
      "training_step_time": 7.484870433807373
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 3.1387240886688232,
      "step": 1509
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 1509,
      "training_step_time": 9.920642614364624
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 3.147002935409546,
      "step": 1509
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 1509,
      "training_step_time": 9.929512023925781
    },
    {
      "epoch": 5.756378173828125e-06,
      "model_forward_time": 2.4445910453796387,
      "step": 1509
    },
    {
      "epoch": 5.756378173828125e-06,
      "step": 1509,
      "training_step_time": 10.323180198669434
    },
    {
      "epoch": 5.76019287109375e-06,
      "grad_norm": 0.6058142185211182,
      "learning_rate": 1e-05,
      "loss": 0.1253,
      "step": 1510
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 3.1755685806274414,
      "step": 1510
    },
    {
      "dynamics_loss_avg": 0.09554668739438058,
      "epoch": 5.76019287109375e-06,
      "step": 1510
    },
    {
      "action_loss_avg": 0.05654368065297603,
      "epoch": 5.76019287109375e-06,
      "step": 1510
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 1510,
      "training_step_time": 10.260286092758179
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 3.1406548023223877,
      "step": 1510
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 1510,
      "training_step_time": 9.919858932495117
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 3.1422548294067383,
      "step": 1510
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 1510,
      "training_step_time": 9.919764280319214
    },
    {
      "epoch": 5.76019287109375e-06,
      "model_forward_time": 3.1380691528320312,
      "step": 1510
    },
    {
      "epoch": 5.76019287109375e-06,
      "step": 1510,
      "training_step_time": 10.275933980941772
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 3.1546618938446045,
      "step": 1511
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 1511,
      "training_step_time": 9.937942028045654
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 3.138305902481079,
      "step": 1511
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 1511,
      "training_step_time": 10.180891752243042
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 2.435210704803467,
      "step": 1511
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 1511,
      "training_step_time": 7.466184854507446
    },
    {
      "epoch": 5.764007568359375e-06,
      "model_forward_time": 3.1369338035583496,
      "step": 1511
    },
    {
      "epoch": 5.764007568359375e-06,
      "step": 1511,
      "training_step_time": 10.302223920822144
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 3.146674633026123,
      "step": 1512
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 1512,
      "training_step_time": 9.92397165298462
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 3.1413116455078125,
      "step": 1512
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 1512,
      "training_step_time": 9.91772198677063
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 2.437255859375,
      "step": 1512
    },
    {
      "dynamics_loss_avg": 0.07161498181521893,
      "epoch": 5.767822265625e-06,
      "step": 1512
    },
    {
      "action_loss_avg": 0.041126581374555826,
      "epoch": 5.767822265625e-06,
      "step": 1512
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 1512,
      "training_step_time": 7.7850611209869385
    },
    {
      "epoch": 5.767822265625e-06,
      "model_forward_time": 3.1345906257629395,
      "step": 1512
    },
    {
      "epoch": 5.767822265625e-06,
      "step": 1512,
      "training_step_time": 10.019471168518066
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 3.1894137859344482,
      "step": 1513
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 1513,
      "training_step_time": 9.966104984283447
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 3.1416478157043457,
      "step": 1513
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 1513,
      "training_step_time": 9.919398784637451
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 3.1380114555358887,
      "step": 1513
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 1513,
      "training_step_time": 9.918403625488281
    },
    {
      "epoch": 5.771636962890625e-06,
      "model_forward_time": 3.1404926776885986,
      "step": 1513
    },
    {
      "epoch": 5.771636962890625e-06,
      "step": 1513,
      "training_step_time": 10.287333250045776
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 3.1904287338256836,
      "step": 1514
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 1514,
      "training_step_time": 9.973891735076904
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 3.1423165798187256,
      "step": 1514
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 1514,
      "training_step_time": 9.921841382980347
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 3.134270668029785,
      "step": 1514
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 1514,
      "training_step_time": 9.90662956237793
    },
    {
      "epoch": 5.77545166015625e-06,
      "model_forward_time": 3.1379854679107666,
      "step": 1514
    },
    {
      "epoch": 5.77545166015625e-06,
      "step": 1514,
      "training_step_time": 10.025090217590332
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 2.7524540424346924,
      "step": 1515
    },
    {
      "dynamics_loss_avg": 0.059548578783869745,
      "epoch": 5.779266357421875e-06,
      "step": 1515
    },
    {
      "action_loss_avg": 0.042253130674362184,
      "epoch": 5.779266357421875e-06,
      "step": 1515
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 1515,
      "training_step_time": 7.81652307510376
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 3.1385278701782227,
      "step": 1515
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 1515,
      "training_step_time": 9.914742708206177
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 3.1359193325042725,
      "step": 1515
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 1515,
      "training_step_time": 9.914208173751831
    },
    {
      "epoch": 5.779266357421875e-06,
      "model_forward_time": 3.1369073390960693,
      "step": 1515
    },
    {
      "epoch": 5.779266357421875e-06,
      "step": 1515,
      "training_step_time": 10.298696517944336
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 3.178443670272827,
      "step": 1516
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 1516,
      "training_step_time": 9.959149360656738
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 3.144657611846924,
      "step": 1516
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 1516,
      "training_step_time": 10.225457906723022
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 3.1343727111816406,
      "step": 1516
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 1516,
      "training_step_time": 9.912604808807373
    },
    {
      "epoch": 5.7830810546875e-06,
      "model_forward_time": 3.1394026279449463,
      "step": 1516
    },
    {
      "epoch": 5.7830810546875e-06,
      "step": 1516,
      "training_step_time": 10.326375246047974
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 3.15419602394104,
      "step": 1517
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 1517,
      "training_step_time": 9.930645227432251
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 3.1340243816375732,
      "step": 1517
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 1517,
      "training_step_time": 9.909659385681152
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 3.1376500129699707,
      "step": 1517
    },
    {
      "dynamics_loss_avg": 0.06932154558598995,
      "epoch": 5.786895751953125e-06,
      "step": 1517
    },
    {
      "action_loss_avg": 0.047433438897132876,
      "epoch": 5.786895751953125e-06,
      "step": 1517
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 1517,
      "training_step_time": 10.200380802154541
    },
    {
      "epoch": 5.786895751953125e-06,
      "model_forward_time": 2.4333362579345703,
      "step": 1517
    },
    {
      "epoch": 5.786895751953125e-06,
      "step": 1517,
      "training_step_time": 10.026416540145874
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 3.175584077835083,
      "step": 1518
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 1518,
      "training_step_time": 9.949881553649902
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 3.136725902557373,
      "step": 1518
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 1518,
      "training_step_time": 9.905989646911621
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 3.1365835666656494,
      "step": 1518
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 1518,
      "training_step_time": 9.905746221542358
    },
    {
      "epoch": 5.79071044921875e-06,
      "model_forward_time": 3.1381704807281494,
      "step": 1518
    },
    {
      "epoch": 5.79071044921875e-06,
      "step": 1518,
      "training_step_time": 10.26893162727356
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 3.187906503677368,
      "step": 1519
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 1519,
      "training_step_time": 9.96121859550476
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 3.1353065967559814,
      "step": 1519
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 1519,
      "training_step_time": 9.908203601837158
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 3.1414873600006104,
      "step": 1519
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 1519,
      "training_step_time": 9.918012142181396
    },
    {
      "epoch": 5.794525146484375e-06,
      "model_forward_time": 3.134230852127075,
      "step": 1519
    },
    {
      "epoch": 5.794525146484375e-06,
      "step": 1519,
      "training_step_time": 10.004477500915527
    },
    {
      "epoch": 5.79833984375e-06,
      "grad_norm": 0.5734317898750305,
      "learning_rate": 1e-05,
      "loss": 0.1185,
      "step": 1520
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 3.1642909049987793,
      "step": 1520
    },
    {
      "dynamics_loss_avg": 0.08545246794819832,
      "epoch": 5.79833984375e-06,
      "step": 1520
    },
    {
      "action_loss_avg": 0.0720617026090622,
      "epoch": 5.79833984375e-06,
      "step": 1520
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 1520,
      "training_step_time": 10.21785569190979
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 3.1361286640167236,
      "step": 1520
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 1520,
      "training_step_time": 9.911271095275879
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 3.1401705741882324,
      "step": 1520
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 1520,
      "training_step_time": 9.917054891586304
    },
    {
      "epoch": 5.79833984375e-06,
      "model_forward_time": 3.138662576675415,
      "step": 1520
    },
    {
      "epoch": 5.79833984375e-06,
      "step": 1520,
      "training_step_time": 10.332916498184204
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 2.449500322341919,
      "step": 1521
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 1521,
      "training_step_time": 7.488647937774658
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 3.425644874572754,
      "step": 1521
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 1521,
      "training_step_time": 10.198939085006714
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 3.1414315700531006,
      "step": 1521
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 1521,
      "training_step_time": 9.922407627105713
    },
    {
      "epoch": 5.802154541015625e-06,
      "model_forward_time": 3.1434197425842285,
      "step": 1521
    },
    {
      "epoch": 5.802154541015625e-06,
      "step": 1521,
      "training_step_time": 10.310915231704712
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 3.160064935684204,
      "step": 1522
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 1522,
      "training_step_time": 9.940315961837769
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 2.434565782546997,
      "step": 1522
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 1522,
      "training_step_time": 7.471092224121094
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 2.4327211380004883,
      "step": 1522
    },
    {
      "dynamics_loss_avg": 0.055751895532011986,
      "epoch": 5.80596923828125e-06,
      "step": 1522
    },
    {
      "action_loss_avg": 0.04259059382602572,
      "epoch": 5.80596923828125e-06,
      "step": 1522
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 1522,
      "training_step_time": 7.468991994857788
    },
    {
      "epoch": 5.80596923828125e-06,
      "model_forward_time": 3.3891749382019043,
      "step": 1522
    },
    {
      "epoch": 5.80596923828125e-06,
      "step": 1522,
      "training_step_time": 10.259776830673218
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 2.4503722190856934,
      "step": 1523
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 1523,
      "training_step_time": 7.4840381145477295
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 3.1384310722351074,
      "step": 1523
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 1523,
      "training_step_time": 9.914284706115723
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 2.4349966049194336,
      "step": 1523
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 1523,
      "training_step_time": 7.467271089553833
    },
    {
      "epoch": 5.809783935546875e-06,
      "model_forward_time": 3.1371030807495117,
      "step": 1523
    },
    {
      "epoch": 5.809783935546875e-06,
      "step": 1523,
      "training_step_time": 10.269497632980347
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 3.4405517578125,
      "step": 1524
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 1524,
      "training_step_time": 10.219276189804077
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 3.141096830368042,
      "step": 1524
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 1524,
      "training_step_time": 9.920658588409424
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 3.137125015258789,
      "step": 1524
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 1524,
      "training_step_time": 9.91677188873291
    },
    {
      "epoch": 5.8135986328125e-06,
      "model_forward_time": 3.1414127349853516,
      "step": 1524
    },
    {
      "epoch": 5.8135986328125e-06,
      "step": 1524,
      "training_step_time": 10.02972674369812
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 2.4753174781799316,
      "step": 1525
    },
    {
      "dynamics_loss_avg": 0.06506836991757155,
      "epoch": 5.817413330078125e-06,
      "step": 1525
    },
    {
      "action_loss_avg": 0.0482961508911103,
      "epoch": 5.817413330078125e-06,
      "step": 1525
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 1525,
      "training_step_time": 7.511294603347778
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 2.704017400741577,
      "step": 1525
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 1525,
      "training_step_time": 7.735094308853149
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 3.1298325061798096,
      "step": 1525
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 1525,
      "training_step_time": 9.897673845291138
    },
    {
      "epoch": 5.817413330078125e-06,
      "model_forward_time": 3.1376631259918213,
      "step": 1525
    },
    {
      "epoch": 5.817413330078125e-06,
      "step": 1525,
      "training_step_time": 10.013138771057129
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 3.177460193634033,
      "step": 1526
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 1526,
      "training_step_time": 9.947287321090698
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 3.1323623657226562,
      "step": 1526
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 1526,
      "training_step_time": 9.905150413513184
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 3.416776418685913,
      "step": 1526
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 1526,
      "training_step_time": 10.197749137878418
    },
    {
      "epoch": 5.82122802734375e-06,
      "model_forward_time": 3.147515058517456,
      "step": 1526
    },
    {
      "epoch": 5.82122802734375e-06,
      "step": 1526,
      "training_step_time": 10.249130725860596
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 3.1558244228363037,
      "step": 1527
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 1527,
      "training_step_time": 9.93632173538208
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 2.4366791248321533,
      "step": 1527
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 1527,
      "training_step_time": 7.471205711364746
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 3.135981559753418,
      "step": 1527
    },
    {
      "dynamics_loss_avg": 0.061587605625391006,
      "epoch": 5.825042724609375e-06,
      "step": 1527
    },
    {
      "action_loss_avg": 0.056754890643060205,
      "epoch": 5.825042724609375e-06,
      "step": 1527
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 1527,
      "training_step_time": 9.921772480010986
    },
    {
      "epoch": 5.825042724609375e-06,
      "model_forward_time": 2.7097315788269043,
      "step": 1527
    },
    {
      "epoch": 5.825042724609375e-06,
      "step": 1527,
      "training_step_time": 10.029595613479614
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 3.1897997856140137,
      "step": 1528
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 1528,
      "training_step_time": 9.974860191345215
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 3.127657651901245,
      "step": 1528
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 1528,
      "training_step_time": 9.90633749961853
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 3.132993221282959,
      "step": 1528
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 1528,
      "training_step_time": 9.905054569244385
    },
    {
      "epoch": 5.828857421875e-06,
      "model_forward_time": 3.130127191543579,
      "step": 1528
    },
    {
      "epoch": 5.828857421875e-06,
      "step": 1528,
      "training_step_time": 10.264866352081299
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 3.437138557434082,
      "step": 1529
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 1529,
      "training_step_time": 10.216999769210815
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 2.429926633834839,
      "step": 1529
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 1529,
      "training_step_time": 7.45747971534729
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 3.1308276653289795,
      "step": 1529
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 1529,
      "training_step_time": 9.905499696731567
    },
    {
      "epoch": 5.832672119140625e-06,
      "model_forward_time": 3.130046844482422,
      "step": 1529
    },
    {
      "epoch": 5.832672119140625e-06,
      "step": 1529,
      "training_step_time": 10.019678354263306
    },
    {
      "epoch": 5.83648681640625e-06,
      "grad_norm": 0.807491660118103,
      "learning_rate": 1e-05,
      "loss": 0.1236,
      "step": 1530
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 3.1663942337036133,
      "step": 1530
    },
    {
      "dynamics_loss_avg": 0.07455873303115368,
      "epoch": 5.83648681640625e-06,
      "step": 1530
    },
    {
      "action_loss_avg": 0.054912093095481396,
      "epoch": 5.83648681640625e-06,
      "step": 1530
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 1530,
      "training_step_time": 9.944213151931763
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 3.4104321002960205,
      "step": 1530
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 1530,
      "training_step_time": 10.182020902633667
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 3.135538339614868,
      "step": 1530
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 1530,
      "training_step_time": 9.904712200164795
    },
    {
      "epoch": 5.83648681640625e-06,
      "model_forward_time": 3.131448984146118,
      "step": 1530
    },
    {
      "epoch": 5.83648681640625e-06,
      "step": 1530,
      "training_step_time": 10.010274410247803
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 2.47098970413208,
      "step": 1531
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 1531,
      "training_step_time": 7.498034477233887
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 2.428279399871826,
      "step": 1531
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 1531,
      "training_step_time": 7.451770782470703
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 3.4087839126586914,
      "step": 1531
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 1531,
      "training_step_time": 10.179408550262451
    },
    {
      "epoch": 5.840301513671875e-06,
      "model_forward_time": 3.1326904296875,
      "step": 1531
    },
    {
      "epoch": 5.840301513671875e-06,
      "step": 1531,
      "training_step_time": 10.286873579025269
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 3.1664140224456787,
      "step": 1532
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 1532,
      "training_step_time": 9.943192958831787
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 3.1388697624206543,
      "step": 1532
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 1532,
      "training_step_time": 9.911234617233276
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 3.1310956478118896,
      "step": 1532
    },
    {
      "dynamics_loss_avg": 0.08294253051280975,
      "epoch": 5.8441162109375e-06,
      "step": 1532
    },
    {
      "action_loss_avg": 0.05912484973669052,
      "epoch": 5.8441162109375e-06,
      "step": 1532
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 1532,
      "training_step_time": 9.905033826828003
    },
    {
      "epoch": 5.8441162109375e-06,
      "model_forward_time": 3.3766191005706787,
      "step": 1532
    },
    {
      "epoch": 5.8441162109375e-06,
      "step": 1532,
      "training_step_time": 10.30868673324585
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 3.1683363914489746,
      "step": 1533
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 1533,
      "training_step_time": 9.940222024917603
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 2.4290263652801514,
      "step": 1533
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 1533,
      "training_step_time": 7.452902793884277
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 3.129554510116577,
      "step": 1533
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 1533,
      "training_step_time": 9.899323463439941
    },
    {
      "epoch": 5.847930908203125e-06,
      "model_forward_time": 3.1333189010620117,
      "step": 1533
    },
    {
      "epoch": 5.847930908203125e-06,
      "step": 1533,
      "training_step_time": 10.285571575164795
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 3.414457082748413,
      "step": 1534
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 1534,
      "training_step_time": 10.186182737350464
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 3.1328001022338867,
      "step": 1534
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 1534,
      "training_step_time": 9.898667812347412
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 3.12943172454834,
      "step": 1534
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 1534,
      "training_step_time": 9.906290292739868
    },
    {
      "epoch": 5.85174560546875e-06,
      "model_forward_time": 3.134284257888794,
      "step": 1534
    },
    {
      "epoch": 5.85174560546875e-06,
      "step": 1534,
      "training_step_time": 10.264266014099121
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 3.1814124584198,
      "step": 1535
    },
    {
      "dynamics_loss_avg": 0.06747138444334269,
      "epoch": 5.855560302734375e-06,
      "step": 1535
    },
    {
      "action_loss_avg": 0.04984398297965527,
      "epoch": 5.855560302734375e-06,
      "step": 1535
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 1535,
      "training_step_time": 9.956939458847046
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 3.4440248012542725,
      "step": 1535
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 1535,
      "training_step_time": 10.216073513031006
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 3.13407564163208,
      "step": 1535
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 1535,
      "training_step_time": 9.907283067703247
    },
    {
      "epoch": 5.855560302734375e-06,
      "model_forward_time": 3.1367244720458984,
      "step": 1535
    },
    {
      "epoch": 5.855560302734375e-06,
      "step": 1535,
      "training_step_time": 10.209438562393188
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 3.147958278656006,
      "step": 1536
    },
    {
      "epoch": 5.859375e-06,
      "step": 1536,
      "training_step_time": 9.92703127861023
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 3.1325936317443848,
      "step": 1536
    },
    {
      "epoch": 5.859375e-06,
      "step": 1536,
      "training_step_time": 9.918845176696777
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 3.3840577602386475,
      "step": 1536
    },
    {
      "epoch": 5.859375e-06,
      "step": 1536,
      "training_step_time": 10.161900043487549
    },
    {
      "epoch": 5.859375e-06,
      "model_forward_time": 2.434382915496826,
      "step": 1536
    },
    {
      "epoch": 5.859375e-06,
      "step": 1536,
      "training_step_time": 10.30080246925354
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 3.1512930393218994,
      "step": 1537
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 1537,
      "training_step_time": 9.927913665771484
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 3.1372032165527344,
      "step": 1537
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 1537,
      "training_step_time": 9.91560983657837
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 2.4323630332946777,
      "step": 1537
    },
    {
      "dynamics_loss_avg": 0.07531218230724335,
      "epoch": 5.863189697265625e-06,
      "step": 1537
    },
    {
      "action_loss_avg": 0.046903468063101174,
      "epoch": 5.863189697265625e-06,
      "step": 1537
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 1537,
      "training_step_time": 7.470832586288452
    },
    {
      "epoch": 5.863189697265625e-06,
      "model_forward_time": 3.4111547470092773,
      "step": 1537
    },
    {
      "epoch": 5.863189697265625e-06,
      "step": 1537,
      "training_step_time": 10.31342601776123
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 3.204890012741089,
      "step": 1538
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 1538,
      "training_step_time": 9.990155220031738
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 3.132516622543335,
      "step": 1538
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 1538,
      "training_step_time": 9.910569906234741
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 2.431166648864746,
      "step": 1538
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 1538,
      "training_step_time": 7.464119911193848
    },
    {
      "epoch": 5.86700439453125e-06,
      "model_forward_time": 3.131535530090332,
      "step": 1538
    },
    {
      "epoch": 5.86700439453125e-06,
      "step": 1538,
      "training_step_time": 10.31897759437561
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 2.7831573486328125,
      "step": 1539
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 1539,
      "training_step_time": 7.813520908355713
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 3.131622314453125,
      "step": 1539
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 1539,
      "training_step_time": 9.904025077819824
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 3.1322550773620605,
      "step": 1539
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 1539,
      "training_step_time": 9.901617050170898
    },
    {
      "epoch": 5.870819091796875e-06,
      "model_forward_time": 3.131312847137451,
      "step": 1539
    },
    {
      "epoch": 5.870819091796875e-06,
      "step": 1539,
      "training_step_time": 10.25890302658081
    },
    {
      "epoch": 5.8746337890625e-06,
      "grad_norm": 0.47752776741981506,
      "learning_rate": 1e-05,
      "loss": 0.124,
      "step": 1540
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 3.1670985221862793,
      "step": 1540
    },
    {
      "dynamics_loss_avg": 0.09709373414516449,
      "epoch": 5.8746337890625e-06,
      "step": 1540
    },
    {
      "action_loss_avg": 0.07062925733625888,
      "epoch": 5.8746337890625e-06,
      "step": 1540
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 1540,
      "training_step_time": 9.94325876235962
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 2.698082685470581,
      "step": 1540
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 1540,
      "training_step_time": 7.727294206619263
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 3.138432741165161,
      "step": 1540
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 1540,
      "training_step_time": 9.9162917137146
    },
    {
      "epoch": 5.8746337890625e-06,
      "model_forward_time": 3.136077642440796,
      "step": 1540
    },
    {
      "epoch": 5.8746337890625e-06,
      "step": 1540,
      "training_step_time": 10.025563478469849
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 3.1540894508361816,
      "step": 1541
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 1541,
      "training_step_time": 9.930413484573364
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 3.1383650302886963,
      "step": 1541
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 1541,
      "training_step_time": 9.915215253829956
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 3.408323049545288,
      "step": 1541
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 1541,
      "training_step_time": 10.187052249908447
    },
    {
      "epoch": 5.878448486328125e-06,
      "model_forward_time": 2.4352681636810303,
      "step": 1541
    },
    {
      "epoch": 5.878448486328125e-06,
      "step": 1541,
      "training_step_time": 10.299312829971313
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 3.1538784503936768,
      "step": 1542
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 1542,
      "training_step_time": 9.94057011604309
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 3.135561227798462,
      "step": 1542
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 1542,
      "training_step_time": 9.914084672927856
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 3.136176347732544,
      "step": 1542
    },
    {
      "dynamics_loss_avg": 0.07374848611652851,
      "epoch": 5.88226318359375e-06,
      "step": 1542
    },
    {
      "action_loss_avg": 0.0691944058984518,
      "epoch": 5.88226318359375e-06,
      "step": 1542
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 1542,
      "training_step_time": 9.908820867538452
    },
    {
      "epoch": 5.88226318359375e-06,
      "model_forward_time": 2.727515697479248,
      "step": 1542
    },
    {
      "epoch": 5.88226318359375e-06,
      "step": 1542,
      "training_step_time": 10.289564371109009
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 3.153447151184082,
      "step": 1543
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 1543,
      "training_step_time": 9.927518367767334
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 3.131939172744751,
      "step": 1543
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 1543,
      "training_step_time": 9.903657913208008
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 3.1297993659973145,
      "step": 1543
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 1543,
      "training_step_time": 9.895204067230225
    },
    {
      "epoch": 5.886077880859375e-06,
      "model_forward_time": 3.1310250759124756,
      "step": 1543
    },
    {
      "epoch": 5.886077880859375e-06,
      "step": 1543,
      "training_step_time": 10.002732515335083
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 3.407890558242798,
      "step": 1544
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 1544,
      "training_step_time": 10.174191236495972
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 2.426661968231201,
      "step": 1544
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 1544,
      "training_step_time": 7.447803258895874
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 3.1240313053131104,
      "step": 1544
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 1544,
      "training_step_time": 9.88707160949707
    },
    {
      "epoch": 5.889892578125e-06,
      "model_forward_time": 3.1287944316864014,
      "step": 1544
    },
    {
      "epoch": 5.889892578125e-06,
      "step": 1544,
      "training_step_time": 10.00010347366333
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 3.1698737144470215,
      "step": 1545
    },
    {
      "dynamics_loss_avg": 0.06571574676781892,
      "epoch": 5.893707275390625e-06,
      "step": 1545
    },
    {
      "action_loss_avg": 0.04228256493806839,
      "epoch": 5.893707275390625e-06,
      "step": 1545
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 1545,
      "training_step_time": 9.968501091003418
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 3.4159960746765137,
      "step": 1545
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 1545,
      "training_step_time": 10.187119245529175
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 3.1332576274871826,
      "step": 1545
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 1545,
      "training_step_time": 9.912895917892456
    },
    {
      "epoch": 5.893707275390625e-06,
      "model_forward_time": 3.1350269317626953,
      "step": 1545
    },
    {
      "epoch": 5.893707275390625e-06,
      "step": 1545,
      "training_step_time": 10.027413368225098
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 3.147357940673828,
      "step": 1546
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 1546,
      "training_step_time": 9.927979946136475
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 3.135327100753784,
      "step": 1546
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 1546,
      "training_step_time": 9.915156364440918
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 3.4235405921936035,
      "step": 1546
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 1546,
      "training_step_time": 10.204686641693115
    },
    {
      "epoch": 5.89752197265625e-06,
      "model_forward_time": 2.4333670139312744,
      "step": 1546
    },
    {
      "epoch": 5.89752197265625e-06,
      "step": 1546,
      "training_step_time": 10.266260623931885
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 3.1523046493530273,
      "step": 1547
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 1547,
      "training_step_time": 9.933820962905884
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 3.139317512512207,
      "step": 1547
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 1547,
      "training_step_time": 9.914087295532227
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 2.4343156814575195,
      "step": 1547
    },
    {
      "dynamics_loss_avg": 0.08864647373557091,
      "epoch": 5.901336669921875e-06,
      "step": 1547
    },
    {
      "action_loss_avg": 0.05725246164947748,
      "epoch": 5.901336669921875e-06,
      "step": 1547
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 1547,
      "training_step_time": 7.473472356796265
    },
    {
      "epoch": 5.901336669921875e-06,
      "model_forward_time": 3.4234323501586914,
      "step": 1547
    },
    {
      "epoch": 5.901336669921875e-06,
      "step": 1547,
      "training_step_time": 10.342391729354858
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 3.1761834621429443,
      "step": 1548
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 1548,
      "training_step_time": 9.956089735031128
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 3.1362059116363525,
      "step": 1548
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 1548,
      "training_step_time": 9.911534070968628
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 3.1369638442993164,
      "step": 1548
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 1548,
      "training_step_time": 9.911023139953613
    },
    {
      "epoch": 5.9051513671875e-06,
      "model_forward_time": 3.1350314617156982,
      "step": 1548
    },
    {
      "epoch": 5.9051513671875e-06,
      "step": 1548,
      "training_step_time": 10.283356666564941
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 3.1554079055786133,
      "step": 1549
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 1549,
      "training_step_time": 10.21186351776123
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 3.1334447860717773,
      "step": 1549
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 1549,
      "training_step_time": 9.90842080116272
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 3.138737916946411,
      "step": 1549
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 1549,
      "training_step_time": 9.915286779403687
    },
    {
      "epoch": 5.908966064453125e-06,
      "model_forward_time": 2.435847520828247,
      "step": 1549
    },
    {
      "epoch": 5.908966064453125e-06,
      "step": 1549,
      "training_step_time": 10.01615285873413
    },
    {
      "epoch": 5.91278076171875e-06,
      "grad_norm": 0.7719043493270874,
      "learning_rate": 1e-05,
      "loss": 0.1235,
      "step": 1550
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 3.1822919845581055,
      "step": 1550
    },
    {
      "dynamics_loss_avg": 0.06686983779072761,
      "epoch": 5.91278076171875e-06,
      "step": 1550
    },
    {
      "action_loss_avg": 0.04604947417974472,
      "epoch": 5.91278076171875e-06,
      "step": 1550
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 1550,
      "training_step_time": 9.970542669296265
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 3.1479008197784424,
      "step": 1550
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 1550,
      "training_step_time": 9.92269492149353
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 3.4688069820404053,
      "step": 1550
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 1550,
      "training_step_time": 10.24767541885376
    },
    {
      "epoch": 5.91278076171875e-06,
      "model_forward_time": 3.1438512802124023,
      "step": 1550
    },
    {
      "epoch": 5.91278076171875e-06,
      "step": 1550,
      "training_step_time": 10.034856796264648
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 3.194267988204956,
      "step": 1551
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 1551,
      "training_step_time": 9.974464893341064
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 3.1462931632995605,
      "step": 1551
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 1551,
      "training_step_time": 9.920937299728394
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 3.1403679847717285,
      "step": 1551
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 1551,
      "training_step_time": 10.178724527359009
    },
    {
      "epoch": 5.916595458984375e-06,
      "model_forward_time": 3.1431658267974854,
      "step": 1551
    },
    {
      "epoch": 5.916595458984375e-06,
      "step": 1551,
      "training_step_time": 10.306239128112793
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 3.171973705291748,
      "step": 1552
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 1552,
      "training_step_time": 9.949559211730957
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 3.14167857170105,
      "step": 1552
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 1552,
      "training_step_time": 9.918502807617188
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 3.140831470489502,
      "step": 1552
    },
    {
      "dynamics_loss_avg": 0.06201371774077415,
      "epoch": 5.92041015625e-06,
      "step": 1552
    },
    {
      "action_loss_avg": 0.041014714166522026,
      "epoch": 5.92041015625e-06,
      "step": 1552
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 1552,
      "training_step_time": 9.914629220962524
    },
    {
      "epoch": 5.92041015625e-06,
      "model_forward_time": 3.4042131900787354,
      "step": 1552
    },
    {
      "epoch": 5.92041015625e-06,
      "step": 1552,
      "training_step_time": 10.274911165237427
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 3.165374279022217,
      "step": 1553
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 1553,
      "training_step_time": 9.94115924835205
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 3.1453418731689453,
      "step": 1553
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 1553,
      "training_step_time": 9.926571607589722
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 3.1476001739501953,
      "step": 1553
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 1553,
      "training_step_time": 9.93149209022522
    },
    {
      "epoch": 5.924224853515625e-06,
      "model_forward_time": 3.148923635482788,
      "step": 1553
    },
    {
      "epoch": 5.924224853515625e-06,
      "step": 1553,
      "training_step_time": 10.27575421333313
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 2.4676294326782227,
      "step": 1554
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 1554,
      "training_step_time": 7.796344757080078
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 3.1373701095581055,
      "step": 1554
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 1554,
      "training_step_time": 9.909048318862915
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 3.138800621032715,
      "step": 1554
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 1554,
      "training_step_time": 9.911308526992798
    },
    {
      "epoch": 5.92803955078125e-06,
      "model_forward_time": 3.1399025917053223,
      "step": 1554
    },
    {
      "epoch": 5.92803955078125e-06,
      "step": 1554,
      "training_step_time": 10.028412818908691
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 3.1773593425750732,
      "step": 1555
    },
    {
      "dynamics_loss_avg": 0.07748004756867885,
      "epoch": 5.931854248046875e-06,
      "step": 1555
    },
    {
      "action_loss_avg": 0.054285785742104055,
      "epoch": 5.931854248046875e-06,
      "step": 1555
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 1555,
      "training_step_time": 9.950870275497437
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 3.4182629585266113,
      "step": 1555
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 1555,
      "training_step_time": 10.185420989990234
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 3.1376283168792725,
      "step": 1555
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 1555,
      "training_step_time": 9.909040689468384
    },
    {
      "epoch": 5.931854248046875e-06,
      "model_forward_time": 3.138453245162964,
      "step": 1555
    },
    {
      "epoch": 5.931854248046875e-06,
      "step": 1555,
      "training_step_time": 10.032612323760986
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 3.1527113914489746,
      "step": 1556
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 1556,
      "training_step_time": 9.92970585823059
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 3.1350784301757812,
      "step": 1556
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 1556,
      "training_step_time": 9.90714144706726
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 2.4336163997650146,
      "step": 1556
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 1556,
      "training_step_time": 7.718980073928833
    },
    {
      "epoch": 5.9356689453125e-06,
      "model_forward_time": 3.1298227310180664,
      "step": 1556
    },
    {
      "epoch": 5.9356689453125e-06,
      "step": 1556,
      "training_step_time": 10.011675596237183
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 3.1697890758514404,
      "step": 1557
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 1557,
      "training_step_time": 9.937504768371582
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 3.129725694656372,
      "step": 1557
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 1557,
      "training_step_time": 9.894503355026245
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 3.1324825286865234,
      "step": 1557
    },
    {
      "dynamics_loss_avg": 0.05835735909640789,
      "epoch": 5.939483642578125e-06,
      "step": 1557
    },
    {
      "action_loss_avg": 0.035046125389635564,
      "epoch": 5.939483642578125e-06,
      "step": 1557
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 1557,
      "training_step_time": 9.898913145065308
    },
    {
      "epoch": 5.939483642578125e-06,
      "model_forward_time": 3.4448471069335938,
      "step": 1557
    },
    {
      "epoch": 5.939483642578125e-06,
      "step": 1557,
      "training_step_time": 10.298824071884155
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 2.440378189086914,
      "step": 1558
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 1558,
      "training_step_time": 7.4642462730407715
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 3.13285756111145,
      "step": 1558
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 1558,
      "training_step_time": 9.897640705108643
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 2.4309303760528564,
      "step": 1558
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 1558,
      "training_step_time": 7.45501446723938
    },
    {
      "epoch": 5.94329833984375e-06,
      "model_forward_time": 3.134922742843628,
      "step": 1558
    },
    {
      "epoch": 5.94329833984375e-06,
      "step": 1558,
      "training_step_time": 10.28135061264038
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 3.176882266998291,
      "step": 1559
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 1559,
      "training_step_time": 10.227097988128662
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 3.140287160873413,
      "step": 1559
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 1559,
      "training_step_time": 9.910510301589966
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 3.1350159645080566,
      "step": 1559
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 1559,
      "training_step_time": 9.90183401107788
    },
    {
      "epoch": 5.947113037109375e-06,
      "model_forward_time": 3.135676383972168,
      "step": 1559
    },
    {
      "epoch": 5.947113037109375e-06,
      "step": 1559,
      "training_step_time": 10.24333381652832
    },
    {
      "epoch": 5.950927734375e-06,
      "grad_norm": 0.47026312351226807,
      "learning_rate": 1e-05,
      "loss": 0.1159,
      "step": 1560
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 3.1764609813690186,
      "step": 1560
    },
    {
      "dynamics_loss_avg": 0.06640256717801094,
      "epoch": 5.950927734375e-06,
      "step": 1560
    },
    {
      "action_loss_avg": 0.03855547290295362,
      "epoch": 5.950927734375e-06,
      "step": 1560
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 1560,
      "training_step_time": 9.951122522354126
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 3.1351637840270996,
      "step": 1560
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 1560,
      "training_step_time": 10.176469802856445
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 3.135472536087036,
      "step": 1560
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 1560,
      "training_step_time": 9.900367498397827
    },
    {
      "epoch": 5.950927734375e-06,
      "model_forward_time": 3.138967514038086,
      "step": 1560
    },
    {
      "epoch": 5.950927734375e-06,
      "step": 1560,
      "training_step_time": 10.008555889129639
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 3.1675808429718018,
      "step": 1561
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 1561,
      "training_step_time": 9.939971685409546
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 3.137009620666504,
      "step": 1561
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 1561,
      "training_step_time": 9.905067682266235
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 3.435462236404419,
      "step": 1561
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 1561,
      "training_step_time": 10.201653242111206
    },
    {
      "epoch": 5.954742431640625e-06,
      "model_forward_time": 3.1363518238067627,
      "step": 1561
    },
    {
      "epoch": 5.954742431640625e-06,
      "step": 1561,
      "training_step_time": 10.283695459365845
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 3.149799346923828,
      "step": 1562
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 1562,
      "training_step_time": 9.92092514038086
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 3.129923105239868,
      "step": 1562
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 1562,
      "training_step_time": 9.897057056427002
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 3.1404049396514893,
      "step": 1562
    },
    {
      "dynamics_loss_avg": 0.05224704034626484,
      "epoch": 5.95855712890625e-06,
      "step": 1562
    },
    {
      "action_loss_avg": 0.034771586768329144,
      "epoch": 5.95855712890625e-06,
      "step": 1562
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 1562,
      "training_step_time": 9.906864404678345
    },
    {
      "epoch": 5.95855712890625e-06,
      "model_forward_time": 2.4413788318634033,
      "step": 1562
    },
    {
      "epoch": 5.95855712890625e-06,
      "step": 1562,
      "training_step_time": 10.32070803642273
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 3.4925358295440674,
      "step": 1563
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 1563,
      "training_step_time": 10.259152173995972
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 3.1388213634490967,
      "step": 1563
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 1563,
      "training_step_time": 9.905231237411499
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 3.1345975399017334,
      "step": 1563
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 1563,
      "training_step_time": 9.906419515609741
    },
    {
      "epoch": 5.962371826171875e-06,
      "model_forward_time": 3.130568504333496,
      "step": 1563
    },
    {
      "epoch": 5.962371826171875e-06,
      "step": 1563,
      "training_step_time": 10.28966212272644
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 2.708505868911743,
      "step": 1564
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 1564,
      "training_step_time": 7.733529806137085
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 3.128840684890747,
      "step": 1564
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 1564,
      "training_step_time": 9.89535641670227
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 3.1340184211730957,
      "step": 1564
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 1564,
      "training_step_time": 9.900690078735352
    },
    {
      "epoch": 5.9661865234375e-06,
      "model_forward_time": 3.1336910724639893,
      "step": 1564
    },
    {
      "epoch": 5.9661865234375e-06,
      "step": 1564,
      "training_step_time": 10.260540008544922
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 3.1740548610687256,
      "step": 1565
    },
    {
      "dynamics_loss_avg": 0.07171690165996551,
      "epoch": 5.970001220703125e-06,
      "step": 1565
    },
    {
      "action_loss_avg": 0.052770454436540604,
      "epoch": 5.970001220703125e-06,
      "step": 1565
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 1565,
      "training_step_time": 9.946233987808228
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 3.1323750019073486,
      "step": 1565
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 1565,
      "training_step_time": 10.194875717163086
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 3.130178928375244,
      "step": 1565
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 1565,
      "training_step_time": 9.895999431610107
    },
    {
      "epoch": 5.970001220703125e-06,
      "model_forward_time": 3.1272408962249756,
      "step": 1565
    },
    {
      "epoch": 5.970001220703125e-06,
      "step": 1565,
      "training_step_time": 10.007718801498413
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 3.1458282470703125,
      "step": 1566
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 1566,
      "training_step_time": 9.915583848953247
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 3.131300926208496,
      "step": 1566
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 1566,
      "training_step_time": 9.898544311523438
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 2.431424617767334,
      "step": 1566
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 1566,
      "training_step_time": 7.735285997390747
    },
    {
      "epoch": 5.97381591796875e-06,
      "model_forward_time": 3.130575180053711,
      "step": 1566
    },
    {
      "epoch": 5.97381591796875e-06,
      "step": 1566,
      "training_step_time": 10.292023181915283
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 2.462815523147583,
      "step": 1567
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 1567,
      "training_step_time": 7.485428810119629
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 3.131653308868408,
      "step": 1567
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 1567,
      "training_step_time": 9.897770881652832
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 3.1301472187042236,
      "step": 1567
    },
    {
      "dynamics_loss_avg": 0.055325944907963276,
      "epoch": 5.977630615234375e-06,
      "step": 1567
    },
    {
      "action_loss_avg": 0.033152517210692165,
      "epoch": 5.977630615234375e-06,
      "step": 1567
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 1567,
      "training_step_time": 9.900146484375
    },
    {
      "epoch": 5.977630615234375e-06,
      "model_forward_time": 3.132326126098633,
      "step": 1567
    },
    {
      "epoch": 5.977630615234375e-06,
      "step": 1567,
      "training_step_time": 10.322877168655396
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 3.4176158905029297,
      "step": 1568
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 1568,
      "training_step_time": 10.178792476654053
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 3.131028175354004,
      "step": 1568
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 1568,
      "training_step_time": 9.897330284118652
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 3.129624366760254,
      "step": 1568
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 1568,
      "training_step_time": 9.890012741088867
    },
    {
      "epoch": 5.9814453125e-06,
      "model_forward_time": 2.4288649559020996,
      "step": 1568
    },
    {
      "epoch": 5.9814453125e-06,
      "step": 1568,
      "training_step_time": 10.315311193466187
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 3.141481399536133,
      "step": 1569
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 1569,
      "training_step_time": 9.904850721359253
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 3.4255454540252686,
      "step": 1569
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 1569,
      "training_step_time": 10.186315059661865
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 3.1322574615478516,
      "step": 1569
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 1569,
      "training_step_time": 9.89844298362732
    },
    {
      "epoch": 5.985260009765625e-06,
      "model_forward_time": 2.4317429065704346,
      "step": 1569
    },
    {
      "epoch": 5.985260009765625e-06,
      "step": 1569,
      "training_step_time": 10.2461678981781
    },
    {
      "epoch": 5.98907470703125e-06,
      "grad_norm": 0.7700897455215454,
      "learning_rate": 1e-05,
      "loss": 0.118,
      "step": 1570
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 2.4532806873321533,
      "step": 1570
    },
    {
      "dynamics_loss_avg": 0.07947997637093067,
      "epoch": 5.98907470703125e-06,
      "step": 1570
    },
    {
      "action_loss_avg": 0.050294787622988224,
      "epoch": 5.98907470703125e-06,
      "step": 1570
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 1570,
      "training_step_time": 7.476622104644775
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 3.13087797164917,
      "step": 1570
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 1570,
      "training_step_time": 9.897066593170166
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 3.3841965198516846,
      "step": 1570
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 1570,
      "training_step_time": 10.152245998382568
    },
    {
      "epoch": 5.98907470703125e-06,
      "model_forward_time": 3.137227773666382,
      "step": 1570
    },
    {
      "epoch": 5.98907470703125e-06,
      "step": 1570,
      "training_step_time": 10.282200574874878
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 3.147169351577759,
      "step": 1571
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 1571,
      "training_step_time": 9.920587062835693
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 3.138235569000244,
      "step": 1571
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 1571,
      "training_step_time": 9.90654444694519
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 3.1380834579467773,
      "step": 1571
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 1571,
      "training_step_time": 9.911221027374268
    },
    {
      "epoch": 5.992889404296875e-06,
      "model_forward_time": 3.4101665019989014,
      "step": 1571
    },
    {
      "epoch": 5.992889404296875e-06,
      "step": 1571,
      "training_step_time": 10.279446840286255
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 3.1821303367614746,
      "step": 1572
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 1572,
      "training_step_time": 9.956354141235352
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 3.1370863914489746,
      "step": 1572
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 1572,
      "training_step_time": 9.912219762802124
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 3.1445200443267822,
      "step": 1572
    },
    {
      "dynamics_loss_avg": 0.06875182781368494,
      "epoch": 5.9967041015625e-06,
      "step": 1572
    },
    {
      "action_loss_avg": 0.04134504795074463,
      "epoch": 5.9967041015625e-06,
      "step": 1572
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 1572,
      "training_step_time": 9.923031330108643
    },
    {
      "epoch": 5.9967041015625e-06,
      "model_forward_time": 3.4505972862243652,
      "step": 1572
    },
    {
      "epoch": 5.9967041015625e-06,
      "step": 1572,
      "training_step_time": 10.320261716842651
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 3.1791231632232666,
      "step": 1573
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 1573,
      "training_step_time": 9.95631718635559
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 3.1434226036071777,
      "step": 1573
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 1573,
      "training_step_time": 9.917646646499634
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 3.1449804306030273,
      "step": 1573
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 1573,
      "training_step_time": 9.922781705856323
    },
    {
      "epoch": 6.000518798828125e-06,
      "model_forward_time": 3.1405694484710693,
      "step": 1573
    },
    {
      "epoch": 6.000518798828125e-06,
      "step": 1573,
      "training_step_time": 10.267953395843506
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 3.144188165664673,
      "step": 1574
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 1574,
      "training_step_time": 10.186296939849854
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 3.1372108459472656,
      "step": 1574
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 1574,
      "training_step_time": 9.912509441375732
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 2.4376862049102783,
      "step": 1574
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 1574,
      "training_step_time": 7.4686925411224365
    },
    {
      "epoch": 6.00433349609375e-06,
      "model_forward_time": 3.137530565261841,
      "step": 1574
    },
    {
      "epoch": 6.00433349609375e-06,
      "step": 1574,
      "training_step_time": 10.300310373306274
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 3.1839609146118164,
      "step": 1575
    },
    {
      "dynamics_loss_avg": 0.07651876322925091,
      "epoch": 6.008148193359375e-06,
      "step": 1575
    },
    {
      "action_loss_avg": 0.046276057697832586,
      "epoch": 6.008148193359375e-06,
      "step": 1575
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 1575,
      "training_step_time": 9.96190333366394
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 3.391523599624634,
      "step": 1575
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 1575,
      "training_step_time": 10.162784337997437
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 3.1400625705718994,
      "step": 1575
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 1575,
      "training_step_time": 9.911153793334961
    },
    {
      "epoch": 6.008148193359375e-06,
      "model_forward_time": 3.135666608810425,
      "step": 1575
    },
    {
      "epoch": 6.008148193359375e-06,
      "step": 1575,
      "training_step_time": 10.026915073394775
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 2.4617085456848145,
      "step": 1576
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 1576,
      "training_step_time": 7.490155935287476
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 2.435197114944458,
      "step": 1576
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 1576,
      "training_step_time": 7.463399171829224
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 3.1339616775512695,
      "step": 1576
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 1576,
      "training_step_time": 9.9090895652771
    },
    {
      "epoch": 6.011962890625e-06,
      "model_forward_time": 3.380621910095215,
      "step": 1576
    },
    {
      "epoch": 6.011962890625e-06,
      "step": 1576,
      "training_step_time": 10.246598720550537
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 2.478743314743042,
      "step": 1577
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 1577,
      "training_step_time": 7.506811141967773
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 2.431408405303955,
      "step": 1577
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 1577,
      "training_step_time": 7.4604010581970215
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 3.1328136920928955,
      "step": 1577
    },
    {
      "dynamics_loss_avg": 0.07629295364022255,
      "epoch": 6.015777587890625e-06,
      "step": 1577
    },
    {
      "action_loss_avg": 0.053749530017375945,
      "epoch": 6.015777587890625e-06,
      "step": 1577
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 1577,
      "training_step_time": 9.906413316726685
    },
    {
      "epoch": 6.015777587890625e-06,
      "model_forward_time": 3.1391489505767822,
      "step": 1577
    },
    {
      "epoch": 6.015777587890625e-06,
      "step": 1577,
      "training_step_time": 10.324054956436157
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 3.447518825531006,
      "step": 1578
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 1578,
      "training_step_time": 10.221788167953491
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 2.432565689086914,
      "step": 1578
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 1578,
      "training_step_time": 7.464184284210205
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 3.133524179458618,
      "step": 1578
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 1578,
      "training_step_time": 9.906992197036743
    },
    {
      "epoch": 6.01959228515625e-06,
      "model_forward_time": 3.1353399753570557,
      "step": 1578
    },
    {
      "epoch": 6.01959228515625e-06,
      "step": 1578,
      "training_step_time": 10.0166335105896
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 3.174938440322876,
      "step": 1579
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 1579,
      "training_step_time": 9.946213245391846
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 3.4369852542877197,
      "step": 1579
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 1579,
      "training_step_time": 10.2101411819458
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 3.1378204822540283,
      "step": 1579
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 1579,
      "training_step_time": 9.910966873168945
    },
    {
      "epoch": 6.023406982421875e-06,
      "model_forward_time": 3.1354637145996094,
      "step": 1579
    },
    {
      "epoch": 6.023406982421875e-06,
      "step": 1579,
      "training_step_time": 10.28201675415039
    },
    {
      "epoch": 6.0272216796875e-06,
      "grad_norm": 0.9647389650344849,
      "learning_rate": 1e-05,
      "loss": 0.1159,
      "step": 1580
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 3.144656181335449,
      "step": 1580
    },
    {
      "dynamics_loss_avg": 0.06575259193778038,
      "epoch": 6.0272216796875e-06,
      "step": 1580
    },
    {
      "action_loss_avg": 0.04169958438724279,
      "epoch": 6.0272216796875e-06,
      "step": 1580
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 1580,
      "training_step_time": 9.919344186782837
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 3.129910945892334,
      "step": 1580
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 1580,
      "training_step_time": 9.89766001701355
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 2.7044851779937744,
      "step": 1580
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 1580,
      "training_step_time": 7.72497820854187
    },
    {
      "epoch": 6.0272216796875e-06,
      "model_forward_time": 3.13151216506958,
      "step": 1580
    },
    {
      "epoch": 6.0272216796875e-06,
      "step": 1580,
      "training_step_time": 10.285336017608643
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 3.169175863265991,
      "step": 1581
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 1581,
      "training_step_time": 9.939005851745605
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 3.1315131187438965,
      "step": 1581
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 1581,
      "training_step_time": 9.903488159179688
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 3.13716983795166,
      "step": 1581
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 1581,
      "training_step_time": 9.907740831375122
    },
    {
      "epoch": 6.031036376953125e-06,
      "model_forward_time": 3.401078462600708,
      "step": 1581
    },
    {
      "epoch": 6.031036376953125e-06,
      "step": 1581,
      "training_step_time": 10.266713857650757
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 2.4659252166748047,
      "step": 1582
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 1582,
      "training_step_time": 7.498392343521118
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 3.1306614875793457,
      "step": 1582
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 1582,
      "training_step_time": 9.899135112762451
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 3.1325364112854004,
      "step": 1582
    },
    {
      "dynamics_loss_avg": 0.06920157726854086,
      "epoch": 6.03485107421875e-06,
      "step": 1582
    },
    {
      "action_loss_avg": 0.047877815645188095,
      "epoch": 6.03485107421875e-06,
      "step": 1582
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 1582,
      "training_step_time": 9.901813268661499
    },
    {
      "epoch": 6.03485107421875e-06,
      "model_forward_time": 3.12931752204895,
      "step": 1582
    },
    {
      "epoch": 6.03485107421875e-06,
      "step": 1582,
      "training_step_time": 10.281253099441528
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 3.4542675018310547,
      "step": 1583
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 1583,
      "training_step_time": 10.2218496799469
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 3.1335012912750244,
      "step": 1583
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 1583,
      "training_step_time": 9.903773307800293
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 3.135213851928711,
      "step": 1583
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 1583,
      "training_step_time": 9.908349752426147
    },
    {
      "epoch": 6.038665771484375e-06,
      "model_forward_time": 3.133892297744751,
      "step": 1583
    },
    {
      "epoch": 6.038665771484375e-06,
      "step": 1583,
      "training_step_time": 10.013832330703735
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 3.1464462280273438,
      "step": 1584
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 1584,
      "training_step_time": 10.230015993118286
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 3.135476589202881,
      "step": 1584
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 1584,
      "training_step_time": 9.913319110870361
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 1.7480990886688232,
      "step": 1584
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 1584,
      "training_step_time": 5.208600282669067
    },
    {
      "epoch": 6.04248046875e-06,
      "model_forward_time": 2.4347329139709473,
      "step": 1584
    },
    {
      "epoch": 6.04248046875e-06,
      "step": 1584,
      "training_step_time": 10.304368257522583
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 3.1677346229553223,
      "step": 1585
    },
    {
      "dynamics_loss_avg": 0.06422074176371098,
      "epoch": 6.046295166015625e-06,
      "step": 1585
    },
    {
      "action_loss_avg": 0.051278114505112174,
      "epoch": 6.046295166015625e-06,
      "step": 1585
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 1585,
      "training_step_time": 9.944873809814453
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 3.1395275592803955,
      "step": 1585
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 1585,
      "training_step_time": 9.91322922706604
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 3.4097208976745605,
      "step": 1585
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 1585,
      "training_step_time": 10.187601089477539
    },
    {
      "epoch": 6.046295166015625e-06,
      "model_forward_time": 3.138274669647217,
      "step": 1585
    },
    {
      "epoch": 6.046295166015625e-06,
      "step": 1585,
      "training_step_time": 10.3202543258667
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 3.1725382804870605,
      "step": 1586
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 1586,
      "training_step_time": 9.950603485107422
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 3.139077663421631,
      "step": 1586
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 1586,
      "training_step_time": 9.912701606750488
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 3.1359643936157227,
      "step": 1586
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 1586,
      "training_step_time": 9.909586668014526
    },
    {
      "epoch": 6.05010986328125e-06,
      "model_forward_time": 3.397779941558838,
      "step": 1586
    },
    {
      "epoch": 6.05010986328125e-06,
      "step": 1586,
      "training_step_time": 10.258387565612793
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 3.18544602394104,
      "step": 1587
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 1587,
      "training_step_time": 9.956486940383911
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 3.140692710876465,
      "step": 1587
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 1587,
      "training_step_time": 9.913176536560059
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 3.136201858520508,
      "step": 1587
    },
    {
      "dynamics_loss_avg": 0.07775925174355507,
      "epoch": 6.053924560546875e-06,
      "step": 1587
    },
    {
      "action_loss_avg": 0.05756701473146677,
      "epoch": 6.053924560546875e-06,
      "step": 1587
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 1587,
      "training_step_time": 9.911019563674927
    },
    {
      "epoch": 6.053924560546875e-06,
      "model_forward_time": 3.141751766204834,
      "step": 1587
    },
    {
      "epoch": 6.053924560546875e-06,
      "step": 1587,
      "training_step_time": 10.021588563919067
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 3.488297939300537,
      "step": 1588
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 1588,
      "training_step_time": 10.273404836654663
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 3.135402202606201,
      "step": 1588
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 1588,
      "training_step_time": 9.922611236572266
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 3.138718366622925,
      "step": 1588
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 1588,
      "training_step_time": 9.92164921760559
    },
    {
      "epoch": 6.0577392578125e-06,
      "model_forward_time": 3.1398165225982666,
      "step": 1588
    },
    {
      "epoch": 6.0577392578125e-06,
      "step": 1588,
      "training_step_time": 10.020947694778442
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 3.1766793727874756,
      "step": 1589
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 1589,
      "training_step_time": 9.949517011642456
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 2.7415595054626465,
      "step": 1589
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 1589,
      "training_step_time": 7.777385711669922
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 3.1345741748809814,
      "step": 1589
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 1589,
      "training_step_time": 9.903523445129395
    },
    {
      "epoch": 6.061553955078125e-06,
      "model_forward_time": 3.135754346847534,
      "step": 1589
    },
    {
      "epoch": 6.061553955078125e-06,
      "step": 1589,
      "training_step_time": 10.224175453186035
    },
    {
      "epoch": 6.06536865234375e-06,
      "grad_norm": 1.2952383756637573,
      "learning_rate": 1e-05,
      "loss": 0.1206,
      "step": 1590
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 3.175673484802246,
      "step": 1590
    },
    {
      "dynamics_loss_avg": 0.06936277374625206,
      "epoch": 6.06536865234375e-06,
      "step": 1590
    },
    {
      "action_loss_avg": 0.055135573633015154,
      "epoch": 6.06536865234375e-06,
      "step": 1590
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 1590,
      "training_step_time": 9.955299854278564
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 3.453458786010742,
      "step": 1590
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 1590,
      "training_step_time": 10.218953609466553
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 3.1309287548065186,
      "step": 1590
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 1590,
      "training_step_time": 9.898580551147461
    },
    {
      "epoch": 6.06536865234375e-06,
      "model_forward_time": 3.127605676651001,
      "step": 1590
    },
    {
      "epoch": 6.06536865234375e-06,
      "step": 1590,
      "training_step_time": 10.267399072647095
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 2.4667422771453857,
      "step": 1591
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 1591,
      "training_step_time": 7.48793363571167
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 3.1297924518585205,
      "step": 1591
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 1591,
      "training_step_time": 9.899234771728516
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 3.135425090789795,
      "step": 1591
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 1591,
      "training_step_time": 9.901213645935059
    },
    {
      "epoch": 6.069183349609375e-06,
      "model_forward_time": 3.43033766746521,
      "step": 1591
    },
    {
      "epoch": 6.069183349609375e-06,
      "step": 1591,
      "training_step_time": 10.29892373085022
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 2.4799485206604004,
      "step": 1592
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 1592,
      "training_step_time": 7.5052759647369385
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 2.427346706390381,
      "step": 1592
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 1592,
      "training_step_time": 7.4531049728393555
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 3.1396467685699463,
      "step": 1592
    },
    {
      "dynamics_loss_avg": 0.0829608429223299,
      "epoch": 6.072998046875e-06,
      "step": 1592
    },
    {
      "action_loss_avg": 0.042408426851034166,
      "epoch": 6.072998046875e-06,
      "step": 1592
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 1592,
      "training_step_time": 9.91588020324707
    },
    {
      "epoch": 6.072998046875e-06,
      "model_forward_time": 3.137047529220581,
      "step": 1592
    },
    {
      "epoch": 6.072998046875e-06,
      "step": 1592,
      "training_step_time": 10.316143274307251
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 3.489541530609131,
      "step": 1593
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 1593,
      "training_step_time": 10.26704478263855
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 3.1356008052825928,
      "step": 1593
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 1593,
      "training_step_time": 9.912702798843384
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 3.1368327140808105,
      "step": 1593
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 1593,
      "training_step_time": 9.911383867263794
    },
    {
      "epoch": 6.076812744140625e-06,
      "model_forward_time": 3.142249345779419,
      "step": 1593
    },
    {
      "epoch": 6.076812744140625e-06,
      "step": 1593,
      "training_step_time": 10.031398057937622
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 3.510683059692383,
      "step": 1594
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 1594,
      "training_step_time": 10.286858558654785
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 3.1429343223571777,
      "step": 1594
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 1594,
      "training_step_time": 9.91335654258728
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 3.1391537189483643,
      "step": 1594
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 1594,
      "training_step_time": 9.908570289611816
    },
    {
      "epoch": 6.08062744140625e-06,
      "model_forward_time": 3.1336164474487305,
      "step": 1594
    },
    {
      "epoch": 6.08062744140625e-06,
      "step": 1594,
      "training_step_time": 10.02336859703064
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 3.1816678047180176,
      "step": 1595
    },
    {
      "dynamics_loss_avg": 0.0695865299552679,
      "epoch": 6.084442138671875e-06,
      "step": 1595
    },
    {
      "action_loss_avg": 0.045872296392917636,
      "epoch": 6.084442138671875e-06,
      "step": 1595
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 1595,
      "training_step_time": 9.962934732437134
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 2.4328460693359375,
      "step": 1595
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 1595,
      "training_step_time": 7.721203804016113
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 3.1325809955596924,
      "step": 1595
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 1595,
      "training_step_time": 9.904518842697144
    },
    {
      "epoch": 6.084442138671875e-06,
      "model_forward_time": 2.4319536685943604,
      "step": 1595
    },
    {
      "epoch": 6.084442138671875e-06,
      "step": 1595,
      "training_step_time": 10.248886585235596
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 3.1525065898895264,
      "step": 1596
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 1596,
      "training_step_time": 9.926824569702148
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 3.133387565612793,
      "step": 1596
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 1596,
      "training_step_time": 9.903465270996094
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 2.4318974018096924,
      "step": 1596
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 1596,
      "training_step_time": 7.761207342147827
    },
    {
      "epoch": 6.0882568359375e-06,
      "model_forward_time": 3.126532554626465,
      "step": 1596
    },
    {
      "epoch": 6.0882568359375e-06,
      "step": 1596,
      "training_step_time": 10.014807224273682
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 3.185793399810791,
      "step": 1597
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 1597,
      "training_step_time": 9.957133293151855
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 3.1349899768829346,
      "step": 1597
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 1597,
      "training_step_time": 9.906800031661987
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 3.1323843002319336,
      "step": 1597
    },
    {
      "dynamics_loss_avg": 0.0689319595694542,
      "epoch": 6.092071533203125e-06,
      "step": 1597
    },
    {
      "action_loss_avg": 0.050210755318403244,
      "epoch": 6.092071533203125e-06,
      "step": 1597
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 1597,
      "training_step_time": 9.908927202224731
    },
    {
      "epoch": 6.092071533203125e-06,
      "model_forward_time": 3.1387383937835693,
      "step": 1597
    },
    {
      "epoch": 6.092071533203125e-06,
      "step": 1597,
      "training_step_time": 10.326278924942017
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 2.446930170059204,
      "step": 1598
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 1598,
      "training_step_time": 7.477297782897949
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 3.137753963470459,
      "step": 1598
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 1598,
      "training_step_time": 9.909037828445435
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 2.433201551437378,
      "step": 1598
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 1598,
      "training_step_time": 7.463320970535278
    },
    {
      "epoch": 6.09588623046875e-06,
      "model_forward_time": 3.131193161010742,
      "step": 1598
    },
    {
      "epoch": 6.09588623046875e-06,
      "step": 1598,
      "training_step_time": 10.02330207824707
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 3.168921947479248,
      "step": 1599
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 1599,
      "training_step_time": 10.204237461090088
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 3.1328978538513184,
      "step": 1599
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 1599,
      "training_step_time": 9.905523777008057
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 3.132857084274292,
      "step": 1599
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 1599,
      "training_step_time": 9.906174182891846
    },
    {
      "epoch": 6.099700927734375e-06,
      "model_forward_time": 3.1438353061676025,
      "step": 1599
    },
    {
      "epoch": 6.099700927734375e-06,
      "step": 1599,
      "training_step_time": 10.007504940032959
    },
    {
      "epoch": 6.103515625e-06,
      "grad_norm": 0.7981442809104919,
      "learning_rate": 1e-05,
      "loss": 0.1182,
      "step": 1600
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 3.1591761112213135,
      "step": 1600
    },
    {
      "dynamics_loss_avg": 0.07437893766909838,
      "epoch": 6.103515625e-06,
      "step": 1600
    },
    {
      "action_loss_avg": 0.06164071448147297,
      "epoch": 6.103515625e-06,
      "step": 1600
    },
    {
      "epoch": 6.103515625e-06,
      "step": 1600,
      "training_step_time": 9.93543529510498
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 3.1297788619995117,
      "step": 1600
    },
    {
      "epoch": 6.103515625e-06,
      "step": 1600,
      "training_step_time": 10.19832158088684
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 3.1349897384643555,
      "step": 1600
    },
    {
      "epoch": 6.103515625e-06,
      "step": 1600,
      "training_step_time": 9.906567573547363
    },
    {
      "epoch": 6.103515625e-06,
      "model_forward_time": 3.1329543590545654,
      "step": 1600
    },
    {
      "epoch": 6.103515625e-06,
      "step": 1600,
      "training_step_time": 10.256904125213623
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 3.192964553833008,
      "step": 1601
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 1601,
      "training_step_time": 9.962152481079102
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 2.437290668487549,
      "step": 1601
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 1601,
      "training_step_time": 7.462558746337891
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 3.405733108520508,
      "step": 1601
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 1601,
      "training_step_time": 10.176366329193115
    },
    {
      "epoch": 6.107330322265625e-06,
      "model_forward_time": 2.4375388622283936,
      "step": 1601
    },
    {
      "epoch": 6.107330322265625e-06,
      "step": 1601,
      "training_step_time": 10.029199600219727
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 3.1828272342681885,
      "step": 1602
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 1602,
      "training_step_time": 9.95836091041565
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 3.1394107341766357,
      "step": 1602
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 1602,
      "training_step_time": 9.917048931121826
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 3.146606683731079,
      "step": 1602
    },
    {
      "dynamics_loss_avg": 0.061354519054293635,
      "epoch": 6.11114501953125e-06,
      "step": 1602
    },
    {
      "action_loss_avg": 0.03748312927782536,
      "epoch": 6.11114501953125e-06,
      "step": 1602
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 1602,
      "training_step_time": 9.924050331115723
    },
    {
      "epoch": 6.11114501953125e-06,
      "model_forward_time": 3.141570806503296,
      "step": 1602
    },
    {
      "epoch": 6.11114501953125e-06,
      "step": 1602,
      "training_step_time": 10.311351299285889
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 3.4697532653808594,
      "step": 1603
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 1603,
      "training_step_time": 10.247495889663696
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 3.141105890274048,
      "step": 1603
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 1603,
      "training_step_time": 9.914491653442383
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 3.141451835632324,
      "step": 1603
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 1603,
      "training_step_time": 9.912133932113647
    },
    {
      "epoch": 6.114959716796875e-06,
      "model_forward_time": 3.1369447708129883,
      "step": 1603
    },
    {
      "epoch": 6.114959716796875e-06,
      "step": 1603,
      "training_step_time": 10.29616641998291
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 3.163804769515991,
      "step": 1604
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 1604,
      "training_step_time": 10.24290919303894
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 3.130843162536621,
      "step": 1604
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 1604,
      "training_step_time": 9.897433757781982
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 3.1428301334381104,
      "step": 1604
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 1604,
      "training_step_time": 9.908084392547607
    },
    {
      "epoch": 6.1187744140625e-06,
      "model_forward_time": 3.1324636936187744,
      "step": 1604
    },
    {
      "epoch": 6.1187744140625e-06,
      "step": 1604,
      "training_step_time": 10.276230335235596
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 3.1610453128814697,
      "step": 1605
    },
    {
      "dynamics_loss_avg": 0.05574632240459323,
      "epoch": 6.122589111328125e-06,
      "step": 1605
    },
    {
      "action_loss_avg": 0.03908291291445494,
      "epoch": 6.122589111328125e-06,
      "step": 1605
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 1605,
      "training_step_time": 9.926244020462036
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 3.391313314437866,
      "step": 1605
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 1605,
      "training_step_time": 10.156254291534424
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 3.1288180351257324,
      "step": 1605
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 1605,
      "training_step_time": 9.893740177154541
    },
    {
      "epoch": 6.122589111328125e-06,
      "model_forward_time": 3.1275811195373535,
      "step": 1605
    },
    {
      "epoch": 6.122589111328125e-06,
      "step": 1605,
      "training_step_time": 10.27462124824524
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 3.1365432739257812,
      "step": 1606
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 1606,
      "training_step_time": 9.907950639724731
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 2.428823471069336,
      "step": 1606
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 1606,
      "training_step_time": 7.4570229053497314
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 3.129922866821289,
      "step": 1606
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 1606,
      "training_step_time": 10.166402578353882
    },
    {
      "epoch": 6.12640380859375e-06,
      "model_forward_time": 2.439134120941162,
      "step": 1606
    },
    {
      "epoch": 6.12640380859375e-06,
      "step": 1606,
      "training_step_time": 10.271056413650513
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 3.188577651977539,
      "step": 1607
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 1607,
      "training_step_time": 9.967182636260986
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 3.1397359371185303,
      "step": 1607
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 1607,
      "training_step_time": 9.920830965042114
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 3.1366264820098877,
      "step": 1607
    },
    {
      "dynamics_loss_avg": 0.06481845322996378,
      "epoch": 6.130218505859375e-06,
      "step": 1607
    },
    {
      "action_loss_avg": 0.04133211998268962,
      "epoch": 6.130218505859375e-06,
      "step": 1607
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 1607,
      "training_step_time": 9.91858959197998
    },
    {
      "epoch": 6.130218505859375e-06,
      "model_forward_time": 3.3903889656066895,
      "step": 1607
    },
    {
      "epoch": 6.130218505859375e-06,
      "step": 1607,
      "training_step_time": 10.336341381072998
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 3.173429489135742,
      "step": 1608
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 1608,
      "training_step_time": 9.948217153549194
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 3.13346529006958,
      "step": 1608
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 1608,
      "training_step_time": 9.903440237045288
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 3.134380578994751,
      "step": 1608
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 1608,
      "training_step_time": 9.91367244720459
    },
    {
      "epoch": 6.134033203125e-06,
      "model_forward_time": 3.13631010055542,
      "step": 1608
    },
    {
      "epoch": 6.134033203125e-06,
      "step": 1608,
      "training_step_time": 10.271119594573975
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 3.1756515502929688,
      "step": 1609
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 1609,
      "training_step_time": 10.244019031524658
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 3.138545036315918,
      "step": 1609
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 1609,
      "training_step_time": 9.91539192199707
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 2.4311766624450684,
      "step": 1609
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 1609,
      "training_step_time": 7.4611122608184814
    },
    {
      "epoch": 6.137847900390625e-06,
      "model_forward_time": 3.1355931758880615,
      "step": 1609
    },
    {
      "epoch": 6.137847900390625e-06,
      "step": 1609,
      "training_step_time": 10.030817985534668
    },
    {
      "epoch": 6.14166259765625e-06,
      "grad_norm": 0.7588966488838196,
      "learning_rate": 1e-05,
      "loss": 0.1144,
      "step": 1610
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 3.1667964458465576,
      "step": 1610
    },
    {
      "dynamics_loss_avg": 0.04479257445782423,
      "epoch": 6.14166259765625e-06,
      "step": 1610
    },
    {
      "action_loss_avg": 0.03421516064554453,
      "epoch": 6.14166259765625e-06,
      "step": 1610
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 1610,
      "training_step_time": 9.956886291503906
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 3.420306444168091,
      "step": 1610
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 1610,
      "training_step_time": 10.19901967048645
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 3.1394729614257812,
      "step": 1610
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 1610,
      "training_step_time": 9.91851258277893
    },
    {
      "epoch": 6.14166259765625e-06,
      "model_forward_time": 3.1407523155212402,
      "step": 1610
    },
    {
      "epoch": 6.14166259765625e-06,
      "step": 1610,
      "training_step_time": 10.0277578830719
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 3.1739983558654785,
      "step": 1611
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 1611,
      "training_step_time": 9.951406002044678
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 3.1330344676971436,
      "step": 1611
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 1611,
      "training_step_time": 9.906168460845947
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 3.398378372192383,
      "step": 1611
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 1611,
      "training_step_time": 10.170185804367065
    },
    {
      "epoch": 6.145477294921875e-06,
      "model_forward_time": 3.1301567554473877,
      "step": 1611
    },
    {
      "epoch": 6.145477294921875e-06,
      "step": 1611,
      "training_step_time": 10.322764158248901
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 3.1837077140808105,
      "step": 1612
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 1612,
      "training_step_time": 9.96979808807373
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 3.134690999984741,
      "step": 1612
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 1612,
      "training_step_time": 9.907671928405762
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 3.131235361099243,
      "step": 1612
    },
    {
      "dynamics_loss_avg": 0.0824130728840828,
      "epoch": 6.1492919921875e-06,
      "step": 1612
    },
    {
      "action_loss_avg": 0.0556933119893074,
      "epoch": 6.1492919921875e-06,
      "step": 1612
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 1612,
      "training_step_time": 9.909477472305298
    },
    {
      "epoch": 6.1492919921875e-06,
      "model_forward_time": 3.418632745742798,
      "step": 1612
    },
    {
      "epoch": 6.1492919921875e-06,
      "step": 1612,
      "training_step_time": 10.289125680923462
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 3.1636624336242676,
      "step": 1613
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 1613,
      "training_step_time": 9.945778846740723
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 3.1268198490142822,
      "step": 1613
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 1613,
      "training_step_time": 9.895557403564453
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 3.1334567070007324,
      "step": 1613
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 1613,
      "training_step_time": 9.901442766189575
    },
    {
      "epoch": 6.153106689453125e-06,
      "model_forward_time": 3.1346120834350586,
      "step": 1613
    },
    {
      "epoch": 6.153106689453125e-06,
      "step": 1613,
      "training_step_time": 10.280844449996948
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 3.4639089107513428,
      "step": 1614
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 1614,
      "training_step_time": 10.23612380027771
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 3.1344900131225586,
      "step": 1614
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 1614,
      "training_step_time": 9.909349203109741
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 3.130647897720337,
      "step": 1614
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 1614,
      "training_step_time": 9.901866436004639
    },
    {
      "epoch": 6.15692138671875e-06,
      "model_forward_time": 3.1321542263031006,
      "step": 1614
    },
    {
      "epoch": 6.15692138671875e-06,
      "step": 1614,
      "training_step_time": 10.2817862033844
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 3.1733384132385254,
      "step": 1615
    },
    {
      "dynamics_loss_avg": 0.07982004024088382,
      "epoch": 6.160736083984375e-06,
      "step": 1615
    },
    {
      "action_loss_avg": 0.05556993018835783,
      "epoch": 6.160736083984375e-06,
      "step": 1615
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 1615,
      "training_step_time": 9.949548959732056
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 3.3898534774780273,
      "step": 1615
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 1615,
      "training_step_time": 10.163684129714966
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 3.141498327255249,
      "step": 1615
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 1615,
      "training_step_time": 9.923588037490845
    },
    {
      "epoch": 6.160736083984375e-06,
      "model_forward_time": 3.141712188720703,
      "step": 1615
    },
    {
      "epoch": 6.160736083984375e-06,
      "step": 1615,
      "training_step_time": 10.250450134277344
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 3.1728782653808594,
      "step": 1616
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 1616,
      "training_step_time": 9.948611736297607
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 3.1445913314819336,
      "step": 1616
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 1616,
      "training_step_time": 9.923386812210083
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 3.4226560592651367,
      "step": 1616
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 1616,
      "training_step_time": 10.200875997543335
    },
    {
      "epoch": 6.16455078125e-06,
      "model_forward_time": 3.143683433532715,
      "step": 1616
    },
    {
      "epoch": 6.16455078125e-06,
      "step": 1616,
      "training_step_time": 10.02658462524414
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 3.1633691787719727,
      "step": 1617
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 1617,
      "training_step_time": 9.94509744644165
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 3.1410515308380127,
      "step": 1617
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 1617,
      "training_step_time": 9.915305137634277
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 2.4352657794952393,
      "step": 1617
    },
    {
      "dynamics_loss_avg": 0.07393157687038183,
      "epoch": 6.168365478515625e-06,
      "step": 1617
    },
    {
      "action_loss_avg": 0.04889579089358449,
      "epoch": 6.168365478515625e-06,
      "step": 1617
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 1617,
      "training_step_time": 7.46563458442688
    },
    {
      "epoch": 6.168365478515625e-06,
      "model_forward_time": 2.435220956802368,
      "step": 1617
    },
    {
      "epoch": 6.168365478515625e-06,
      "step": 1617,
      "training_step_time": 10.31087040901184
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 2.1296067237854004,
      "step": 1618
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 1618,
      "training_step_time": 5.588881015777588
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 3.1396219730377197,
      "step": 1618
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 1618,
      "training_step_time": 9.91116714477539
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 3.1367342472076416,
      "step": 1618
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 1618,
      "training_step_time": 9.909703969955444
    },
    {
      "epoch": 6.17218017578125e-06,
      "model_forward_time": 3.1420936584472656,
      "step": 1618
    },
    {
      "epoch": 6.17218017578125e-06,
      "step": 1618,
      "training_step_time": 10.279695749282837
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 3.182112455368042,
      "step": 1619
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 1619,
      "training_step_time": 9.95714521408081
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 3.3965182304382324,
      "step": 1619
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 1619,
      "training_step_time": 10.167861700057983
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 3.1340126991271973,
      "step": 1619
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 1619,
      "training_step_time": 9.907985210418701
    },
    {
      "epoch": 6.175994873046875e-06,
      "model_forward_time": 3.1386091709136963,
      "step": 1619
    },
    {
      "epoch": 6.175994873046875e-06,
      "step": 1619,
      "training_step_time": 10.274043560028076
    },
    {
      "epoch": 6.1798095703125e-06,
      "grad_norm": 0.6198502779006958,
      "learning_rate": 1e-05,
      "loss": 0.1191,
      "step": 1620
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 3.1808416843414307,
      "step": 1620
    },
    {
      "dynamics_loss_avg": 0.06367144547402859,
      "epoch": 6.1798095703125e-06,
      "step": 1620
    },
    {
      "action_loss_avg": 0.05726159233599901,
      "epoch": 6.1798095703125e-06,
      "step": 1620
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 1620,
      "training_step_time": 9.964126348495483
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 3.4139699935913086,
      "step": 1620
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 1620,
      "training_step_time": 10.190736055374146
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 3.134967803955078,
      "step": 1620
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 1620,
      "training_step_time": 9.9174222946167
    },
    {
      "epoch": 6.1798095703125e-06,
      "model_forward_time": 3.1434528827667236,
      "step": 1620
    },
    {
      "epoch": 6.1798095703125e-06,
      "step": 1620,
      "training_step_time": 10.028151988983154
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 3.1816213130950928,
      "step": 1621
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 1621,
      "training_step_time": 9.964446544647217
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 2.4322564601898193,
      "step": 1621
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 1621,
      "training_step_time": 7.463943958282471
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 3.135251522064209,
      "step": 1621
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 1621,
      "training_step_time": 10.21420955657959
    },
    {
      "epoch": 6.183624267578125e-06,
      "model_forward_time": 2.4303297996520996,
      "step": 1621
    },
    {
      "epoch": 6.183624267578125e-06,
      "step": 1621,
      "training_step_time": 10.257822036743164
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 3.1790859699249268,
      "step": 1622
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 1622,
      "training_step_time": 9.963558912277222
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 3.1328940391540527,
      "step": 1622
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 1622,
      "training_step_time": 9.907980918884277
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 3.1416685581207275,
      "step": 1622
    },
    {
      "dynamics_loss_avg": 0.0555383425205946,
      "epoch": 6.18743896484375e-06,
      "step": 1622
    },
    {
      "action_loss_avg": 0.04097722601145506,
      "epoch": 6.18743896484375e-06,
      "step": 1622
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 1622,
      "training_step_time": 9.92137598991394
    },
    {
      "epoch": 6.18743896484375e-06,
      "model_forward_time": 2.4339959621429443,
      "step": 1622
    },
    {
      "epoch": 6.18743896484375e-06,
      "step": 1622,
      "training_step_time": 10.311339139938354
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 2.057980537414551,
      "step": 1623
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 1623,
      "training_step_time": 5.517239809036255
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 3.127434730529785,
      "step": 1623
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 1623,
      "training_step_time": 9.902100324630737
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 3.13086199760437,
      "step": 1623
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 1623,
      "training_step_time": 9.896590232849121
    },
    {
      "epoch": 6.191253662109375e-06,
      "model_forward_time": 2.4250636100769043,
      "step": 1623
    },
    {
      "epoch": 6.191253662109375e-06,
      "step": 1623,
      "training_step_time": 10.279336214065552
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 3.156205892562866,
      "step": 1624
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 1624,
      "training_step_time": 9.922017574310303
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 3.38140606880188,
      "step": 1624
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 1624,
      "training_step_time": 10.1398184299469
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 3.1288208961486816,
      "step": 1624
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 1624,
      "training_step_time": 9.892672538757324
    },
    {
      "epoch": 6.195068359375e-06,
      "model_forward_time": 3.130979061126709,
      "step": 1624
    },
    {
      "epoch": 6.195068359375e-06,
      "step": 1624,
      "training_step_time": 10.243155479431152
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 3.168088912963867,
      "step": 1625
    },
    {
      "dynamics_loss_avg": 0.06596698742359877,
      "epoch": 6.198883056640625e-06,
      "step": 1625
    },
    {
      "action_loss_avg": 0.043323201965540646,
      "epoch": 6.198883056640625e-06,
      "step": 1625
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 1625,
      "training_step_time": 9.942899465560913
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 3.1337642669677734,
      "step": 1625
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 1625,
      "training_step_time": 9.899885177612305
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 3.436274766921997,
      "step": 1625
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 1625,
      "training_step_time": 10.206027030944824
    },
    {
      "epoch": 6.198883056640625e-06,
      "model_forward_time": 3.1332356929779053,
      "step": 1625
    },
    {
      "epoch": 6.198883056640625e-06,
      "step": 1625,
      "training_step_time": 10.320759773254395
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 3.1709253787994385,
      "step": 1626
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 1626,
      "training_step_time": 9.947993993759155
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 2.4322986602783203,
      "step": 1626
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 1626,
      "training_step_time": 7.462174654006958
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 3.131359815597534,
      "step": 1626
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 1626,
      "training_step_time": 9.899644374847412
    },
    {
      "epoch": 6.20269775390625e-06,
      "model_forward_time": 3.394709348678589,
      "step": 1626
    },
    {
      "epoch": 6.20269775390625e-06,
      "step": 1626,
      "training_step_time": 10.260002374649048
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 3.153312921524048,
      "step": 1627
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 1627,
      "training_step_time": 9.927376747131348
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 3.1369106769561768,
      "step": 1627
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 1627,
      "training_step_time": 9.906511306762695
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 3.134040355682373,
      "step": 1627
    },
    {
      "dynamics_loss_avg": 0.05902715669944882,
      "epoch": 6.206512451171875e-06,
      "step": 1627
    },
    {
      "action_loss_avg": 0.03242424046620727,
      "epoch": 6.206512451171875e-06,
      "step": 1627
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 1627,
      "training_step_time": 9.917359113693237
    },
    {
      "epoch": 6.206512451171875e-06,
      "model_forward_time": 3.1370601654052734,
      "step": 1627
    },
    {
      "epoch": 6.206512451171875e-06,
      "step": 1627,
      "training_step_time": 10.257991552352905
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 2.7560272216796875,
      "step": 1628
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 1628,
      "training_step_time": 7.789646863937378
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 3.132997512817383,
      "step": 1628
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 1628,
      "training_step_time": 9.907546043395996
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 3.129626989364624,
      "step": 1628
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 1628,
      "training_step_time": 9.897601842880249
    },
    {
      "epoch": 6.2103271484375e-06,
      "model_forward_time": 3.1326165199279785,
      "step": 1628
    },
    {
      "epoch": 6.2103271484375e-06,
      "step": 1628,
      "training_step_time": 10.294726848602295
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 3.14522123336792,
      "step": 1629
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 1629,
      "training_step_time": 9.914034128189087
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 3.126230478286743,
      "step": 1629
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 1629,
      "training_step_time": 10.17618179321289
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 2.4305968284606934,
      "step": 1629
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 1629,
      "training_step_time": 7.457107067108154
    },
    {
      "epoch": 6.214141845703125e-06,
      "model_forward_time": 3.127674102783203,
      "step": 1629
    },
    {
      "epoch": 6.214141845703125e-06,
      "step": 1629,
      "training_step_time": 10.306611061096191
    },
    {
      "epoch": 6.21795654296875e-06,
      "grad_norm": 0.6450509428977966,
      "learning_rate": 1e-05,
      "loss": 0.1132,
      "step": 1630
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 3.172280788421631,
      "step": 1630
    },
    {
      "dynamics_loss_avg": 0.05727145429700613,
      "epoch": 6.21795654296875e-06,
      "step": 1630
    },
    {
      "action_loss_avg": 0.04487045691348612,
      "epoch": 6.21795654296875e-06,
      "step": 1630
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 1630,
      "training_step_time": 9.954334259033203
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 3.1357083320617676,
      "step": 1630
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 1630,
      "training_step_time": 9.912519931793213
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 3.1377875804901123,
      "step": 1630
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 1630,
      "training_step_time": 10.213760137557983
    },
    {
      "epoch": 6.21795654296875e-06,
      "model_forward_time": 3.1360647678375244,
      "step": 1630
    },
    {
      "epoch": 6.21795654296875e-06,
      "step": 1630,
      "training_step_time": 10.30673360824585
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 3.143465757369995,
      "step": 1631
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 1631,
      "training_step_time": 9.922426223754883
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 3.1350739002227783,
      "step": 1631
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 1631,
      "training_step_time": 9.909937620162964
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 2.431105613708496,
      "step": 1631
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 1631,
      "training_step_time": 7.461639165878296
    },
    {
      "epoch": 6.221771240234375e-06,
      "model_forward_time": 3.405073642730713,
      "step": 1631
    },
    {
      "epoch": 6.221771240234375e-06,
      "step": 1631,
      "training_step_time": 10.275742292404175
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 3.1589043140411377,
      "step": 1632
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 1632,
      "training_step_time": 9.929737567901611
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 3.136457920074463,
      "step": 1632
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 1632,
      "training_step_time": 9.90332841873169
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 3.139592170715332,
      "step": 1632
    },
    {
      "dynamics_loss_avg": 0.0627940209582448,
      "epoch": 6.2255859375e-06,
      "step": 1632
    },
    {
      "action_loss_avg": 0.039216017350554466,
      "epoch": 6.2255859375e-06,
      "step": 1632
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 1632,
      "training_step_time": 9.916111469268799
    },
    {
      "epoch": 6.2255859375e-06,
      "model_forward_time": 2.4362235069274902,
      "step": 1632
    },
    {
      "epoch": 6.2255859375e-06,
      "step": 1632,
      "training_step_time": 10.290366649627686
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 3.14530611038208,
      "step": 1633
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 1633,
      "training_step_time": 9.918227434158325
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 2.731567144393921,
      "step": 1633
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 1633,
      "training_step_time": 7.757662057876587
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 2.4322547912597656,
      "step": 1633
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 1633,
      "training_step_time": 7.457081317901611
    },
    {
      "epoch": 6.229400634765625e-06,
      "model_forward_time": 3.131030797958374,
      "step": 1633
    },
    {
      "epoch": 6.229400634765625e-06,
      "step": 1633,
      "training_step_time": 10.314609289169312
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 3.172506809234619,
      "step": 1634
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 1634,
      "training_step_time": 9.942363977432251
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 2.435044765472412,
      "step": 1634
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 1634,
      "training_step_time": 7.465337038040161
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 3.421450614929199,
      "step": 1634
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 1634,
      "training_step_time": 10.19825267791748
    },
    {
      "epoch": 6.23321533203125e-06,
      "model_forward_time": 3.1422042846679688,
      "step": 1634
    },
    {
      "epoch": 6.23321533203125e-06,
      "step": 1634,
      "training_step_time": 10.29869556427002
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 3.184692621231079,
      "step": 1635
    },
    {
      "dynamics_loss_avg": 0.08005315735936165,
      "epoch": 6.237030029296875e-06,
      "step": 1635
    },
    {
      "action_loss_avg": 0.05447867512702942,
      "epoch": 6.237030029296875e-06,
      "step": 1635
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 1635,
      "training_step_time": 9.964127540588379
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 3.1412830352783203,
      "step": 1635
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 1635,
      "training_step_time": 9.911812543869019
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 3.1386351585388184,
      "step": 1635
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 1635,
      "training_step_time": 9.904195308685303
    },
    {
      "epoch": 6.237030029296875e-06,
      "model_forward_time": 3.3740508556365967,
      "step": 1635
    },
    {
      "epoch": 6.237030029296875e-06,
      "step": 1635,
      "training_step_time": 10.297677040100098
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 3.1814513206481934,
      "step": 1636
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 1636,
      "training_step_time": 9.947151899337769
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 2.428530216217041,
      "step": 1636
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 1636,
      "training_step_time": 7.453835964202881
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 3.1289877891540527,
      "step": 1636
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 1636,
      "training_step_time": 9.89229679107666
    },
    {
      "epoch": 6.2408447265625e-06,
      "model_forward_time": 3.395005702972412,
      "step": 1636
    },
    {
      "epoch": 6.2408447265625e-06,
      "step": 1636,
      "training_step_time": 10.257777214050293
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 3.1639366149902344,
      "step": 1637
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 1637,
      "training_step_time": 9.936644792556763
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 3.1347410678863525,
      "step": 1637
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 1637,
      "training_step_time": 9.904723644256592
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 3.136589288711548,
      "step": 1637
    },
    {
      "dynamics_loss_avg": 0.05433816332370043,
      "epoch": 6.244659423828125e-06,
      "step": 1637
    },
    {
      "action_loss_avg": 0.033833188097923995,
      "epoch": 6.244659423828125e-06,
      "step": 1637
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 1637,
      "training_step_time": 9.904714345932007
    },
    {
      "epoch": 6.244659423828125e-06,
      "model_forward_time": 3.133749485015869,
      "step": 1637
    },
    {
      "epoch": 6.244659423828125e-06,
      "step": 1637,
      "training_step_time": 9.99617624282837
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 3.1764299869537354,
      "step": 1638
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 1638,
      "training_step_time": 10.235763549804688
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 2.433473825454712,
      "step": 1638
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 1638,
      "training_step_time": 7.459083557128906
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 3.1329100131988525,
      "step": 1638
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 1638,
      "training_step_time": 9.898743629455566
    },
    {
      "epoch": 6.24847412109375e-06,
      "model_forward_time": 3.136611223220825,
      "step": 1638
    },
    {
      "epoch": 6.24847412109375e-06,
      "step": 1638,
      "training_step_time": 10.301402568817139
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 3.170628309249878,
      "step": 1639
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 1639,
      "training_step_time": 9.941978693008423
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 3.13771653175354,
      "step": 1639
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 1639,
      "training_step_time": 10.194703817367554
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 3.1366939544677734,
      "step": 1639
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 1639,
      "training_step_time": 9.908008575439453
    },
    {
      "epoch": 6.252288818359375e-06,
      "model_forward_time": 3.148306369781494,
      "step": 1639
    },
    {
      "epoch": 6.252288818359375e-06,
      "step": 1639,
      "training_step_time": 10.284381628036499
    },
    {
      "epoch": 6.256103515625e-06,
      "grad_norm": 0.4141820967197418,
      "learning_rate": 1e-05,
      "loss": 0.118,
      "step": 1640
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 3.189814805984497,
      "step": 1640
    },
    {
      "dynamics_loss_avg": 0.06663777232170105,
      "epoch": 6.256103515625e-06,
      "step": 1640
    },
    {
      "action_loss_avg": 0.040899666771292684,
      "epoch": 6.256103515625e-06,
      "step": 1640
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 1640,
      "training_step_time": 9.968685388565063
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 3.142561674118042,
      "step": 1640
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 1640,
      "training_step_time": 9.914016008377075
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 3.1451220512390137,
      "step": 1640
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 1640,
      "training_step_time": 10.185031175613403
    },
    {
      "epoch": 6.256103515625e-06,
      "model_forward_time": 3.141108751296997,
      "step": 1640
    },
    {
      "epoch": 6.256103515625e-06,
      "step": 1640,
      "training_step_time": 10.242595911026001
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 2.471684694290161,
      "step": 1641
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 1641,
      "training_step_time": 7.498927116394043
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 3.136112689971924,
      "step": 1641
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 1641,
      "training_step_time": 9.90238094329834
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 3.146000385284424,
      "step": 1641
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 1641,
      "training_step_time": 9.91689419746399
    },
    {
      "epoch": 6.259918212890625e-06,
      "model_forward_time": 3.144752025604248,
      "step": 1641
    },
    {
      "epoch": 6.259918212890625e-06,
      "step": 1641,
      "training_step_time": 10.2666015625
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 3.1862635612487793,
      "step": 1642
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 1642,
      "training_step_time": 9.967715978622437
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 3.135822057723999,
      "step": 1642
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 1642,
      "training_step_time": 9.917823553085327
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 3.1354029178619385,
      "step": 1642
    },
    {
      "dynamics_loss_avg": 0.06341724880039692,
      "epoch": 6.26373291015625e-06,
      "step": 1642
    },
    {
      "action_loss_avg": 0.03786719422787428,
      "epoch": 6.26373291015625e-06,
      "step": 1642
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 1642,
      "training_step_time": 9.91372275352478
    },
    {
      "epoch": 6.26373291015625e-06,
      "model_forward_time": 3.134812116622925,
      "step": 1642
    },
    {
      "epoch": 6.26373291015625e-06,
      "step": 1642,
      "training_step_time": 10.021322965621948
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 3.4434032440185547,
      "step": 1643
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 1643,
      "training_step_time": 10.220314025878906
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 3.139840602874756,
      "step": 1643
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 1643,
      "training_step_time": 9.919884204864502
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 3.1426303386688232,
      "step": 1643
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 1643,
      "training_step_time": 9.919187068939209
    },
    {
      "epoch": 6.267547607421875e-06,
      "model_forward_time": 3.1429526805877686,
      "step": 1643
    },
    {
      "epoch": 6.267547607421875e-06,
      "step": 1643,
      "training_step_time": 10.254295587539673
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 3.1522819995880127,
      "step": 1644
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 1644,
      "training_step_time": 9.928913354873657
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 3.4351727962493896,
      "step": 1644
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 1644,
      "training_step_time": 10.207571744918823
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 3.1448147296905518,
      "step": 1644
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 1644,
      "training_step_time": 9.925964593887329
    },
    {
      "epoch": 6.2713623046875e-06,
      "model_forward_time": 2.4346492290496826,
      "step": 1644
    },
    {
      "epoch": 6.2713623046875e-06,
      "step": 1644,
      "training_step_time": 10.287687063217163
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 3.158003568649292,
      "step": 1645
    },
    {
      "dynamics_loss_avg": 0.0663189459592104,
      "epoch": 6.275177001953125e-06,
      "step": 1645
    },
    {
      "action_loss_avg": 0.0496813228353858,
      "epoch": 6.275177001953125e-06,
      "step": 1645
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 1645,
      "training_step_time": 9.94434666633606
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 3.1411378383636475,
      "step": 1645
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 1645,
      "training_step_time": 9.917501449584961
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 3.4301655292510986,
      "step": 1645
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 1645,
      "training_step_time": 10.205104112625122
    },
    {
      "epoch": 6.275177001953125e-06,
      "model_forward_time": 2.438671827316284,
      "step": 1645
    },
    {
      "epoch": 6.275177001953125e-06,
      "step": 1645,
      "training_step_time": 10.260814666748047
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 3.158245801925659,
      "step": 1646
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 1646,
      "training_step_time": 9.934521198272705
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 3.1406257152557373,
      "step": 1646
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 1646,
      "training_step_time": 9.915680408477783
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 3.1395516395568848,
      "step": 1646
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 1646,
      "training_step_time": 9.917481184005737
    },
    {
      "epoch": 6.27899169921875e-06,
      "model_forward_time": 3.138719081878662,
      "step": 1646
    },
    {
      "epoch": 6.27899169921875e-06,
      "step": 1646,
      "training_step_time": 10.289899110794067
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 3.418783187866211,
      "step": 1647
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 1647,
      "training_step_time": 10.191221714019775
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 3.136789560317993,
      "step": 1647
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 1647,
      "training_step_time": 9.90748405456543
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 3.132547378540039,
      "step": 1647
    },
    {
      "dynamics_loss_avg": 0.06900852955877781,
      "epoch": 6.282806396484375e-06,
      "step": 1647
    },
    {
      "action_loss_avg": 0.04225722206756473,
      "epoch": 6.282806396484375e-06,
      "step": 1647
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 1647,
      "training_step_time": 9.90430212020874
    },
    {
      "epoch": 6.282806396484375e-06,
      "model_forward_time": 3.1317315101623535,
      "step": 1647
    },
    {
      "epoch": 6.282806396484375e-06,
      "step": 1647,
      "training_step_time": 10.323788404464722
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 3.4641573429107666,
      "step": 1648
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 1648,
      "training_step_time": 10.243645429611206
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 2.4284322261810303,
      "step": 1648
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 1648,
      "training_step_time": 7.451072692871094
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 2.4318349361419678,
      "step": 1648
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 1648,
      "training_step_time": 7.4544596672058105
    },
    {
      "epoch": 6.28662109375e-06,
      "model_forward_time": 3.1310999393463135,
      "step": 1648
    },
    {
      "epoch": 6.28662109375e-06,
      "step": 1648,
      "training_step_time": 10.260448932647705
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 3.1423070430755615,
      "step": 1649
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 1649,
      "training_step_time": 9.91074252128601
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 3.131030321121216,
      "step": 1649
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 1649,
      "training_step_time": 9.89646029472351
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 2.709963798522949,
      "step": 1649
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 1649,
      "training_step_time": 7.740874290466309
    },
    {
      "epoch": 6.290435791015625e-06,
      "model_forward_time": 3.1353585720062256,
      "step": 1649
    },
    {
      "epoch": 6.290435791015625e-06,
      "step": 1649,
      "training_step_time": 10.24270248413086
    },
    {
      "epoch": 6.29425048828125e-06,
      "grad_norm": 0.6464210152626038,
      "learning_rate": 1e-05,
      "loss": 0.1174,
      "step": 1650
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 3.1860392093658447,
      "step": 1650
    },
    {
      "dynamics_loss_avg": 0.06425282917916775,
      "epoch": 6.29425048828125e-06,
      "step": 1650
    },
    {
      "action_loss_avg": 0.05185344349592924,
      "epoch": 6.29425048828125e-06,
      "step": 1650
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 1650,
      "training_step_time": 9.968621253967285
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 3.1396284103393555,
      "step": 1650
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 1650,
      "training_step_time": 9.917733907699585
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 2.4363954067230225,
      "step": 1650
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 1650,
      "training_step_time": 7.470038414001465
    },
    {
      "epoch": 6.29425048828125e-06,
      "model_forward_time": 3.418853759765625,
      "step": 1650
    },
    {
      "epoch": 6.29425048828125e-06,
      "step": 1650,
      "training_step_time": 10.323492765426636
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 3.165929079055786,
      "step": 1651
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 1651,
      "training_step_time": 9.943879842758179
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 3.132073402404785,
      "step": 1651
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 1651,
      "training_step_time": 9.916494607925415
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 3.138434648513794,
      "step": 1651
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 1651,
      "training_step_time": 9.915315628051758
    },
    {
      "epoch": 6.298065185546875e-06,
      "model_forward_time": 3.136563301086426,
      "step": 1651
    },
    {
      "epoch": 6.298065185546875e-06,
      "step": 1651,
      "training_step_time": 10.293045043945312
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 3.4643473625183105,
      "step": 1652
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 1652,
      "training_step_time": 10.245720386505127
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 3.1384358406066895,
      "step": 1652
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 1652,
      "training_step_time": 9.919152975082397
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 3.142110824584961,
      "step": 1652
    },
    {
      "dynamics_loss_avg": 0.06180557254701853,
      "epoch": 6.3018798828125e-06,
      "step": 1652
    },
    {
      "action_loss_avg": 0.04151281928643584,
      "epoch": 6.3018798828125e-06,
      "step": 1652
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 1652,
      "training_step_time": 9.93131160736084
    },
    {
      "epoch": 6.3018798828125e-06,
      "model_forward_time": 3.141925811767578,
      "step": 1652
    },
    {
      "epoch": 6.3018798828125e-06,
      "step": 1652,
      "training_step_time": 10.017847776412964
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 3.193472146987915,
      "step": 1653
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 1653,
      "training_step_time": 9.97482943534851
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 3.385254144668579,
      "step": 1653
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 1653,
      "training_step_time": 10.172675848007202
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 3.1457743644714355,
      "step": 1653
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 1653,
      "training_step_time": 9.930151224136353
    },
    {
      "epoch": 6.305694580078125e-06,
      "model_forward_time": 3.1413378715515137,
      "step": 1653
    },
    {
      "epoch": 6.305694580078125e-06,
      "step": 1653,
      "training_step_time": 10.272578477859497
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 3.172477960586548,
      "step": 1654
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 1654,
      "training_step_time": 9.953854322433472
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 3.1460416316986084,
      "step": 1654
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 1654,
      "training_step_time": 10.240227699279785
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 3.143892288208008,
      "step": 1654
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 1654,
      "training_step_time": 9.927314043045044
    },
    {
      "epoch": 6.30950927734375e-06,
      "model_forward_time": 3.1424999237060547,
      "step": 1654
    },
    {
      "epoch": 6.30950927734375e-06,
      "step": 1654,
      "training_step_time": 10.030572414398193
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 3.1759448051452637,
      "step": 1655
    },
    {
      "dynamics_loss_avg": 0.062833546474576,
      "epoch": 6.313323974609375e-06,
      "step": 1655
    },
    {
      "action_loss_avg": 0.04101723860949278,
      "epoch": 6.313323974609375e-06,
      "step": 1655
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 1655,
      "training_step_time": 9.96071457862854
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 3.144045352935791,
      "step": 1655
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 1655,
      "training_step_time": 9.924989461898804
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 3.1387038230895996,
      "step": 1655
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 1655,
      "training_step_time": 10.193564653396606
    },
    {
      "epoch": 6.313323974609375e-06,
      "model_forward_time": 3.143115997314453,
      "step": 1655
    },
    {
      "epoch": 6.313323974609375e-06,
      "step": 1655,
      "training_step_time": 10.317090511322021
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 3.168412923812866,
      "step": 1656
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 1656,
      "training_step_time": 9.945190906524658
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 3.137679100036621,
      "step": 1656
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 1656,
      "training_step_time": 9.916998624801636
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 3.13238787651062,
      "step": 1656
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 1656,
      "training_step_time": 9.909348011016846
    },
    {
      "epoch": 6.317138671875e-06,
      "model_forward_time": 3.4287123680114746,
      "step": 1656
    },
    {
      "epoch": 6.317138671875e-06,
      "step": 1656,
      "training_step_time": 10.297125101089478
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 3.1599981784820557,
      "step": 1657
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 1657,
      "training_step_time": 9.933252334594727
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 3.139824867248535,
      "step": 1657
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 1657,
      "training_step_time": 9.910489082336426
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 1.748227596282959,
      "step": 1657
    },
    {
      "dynamics_loss_avg": 0.07638318426907062,
      "epoch": 6.320953369140625e-06,
      "step": 1657
    },
    {
      "action_loss_avg": 0.0568094814196229,
      "epoch": 6.320953369140625e-06,
      "step": 1657
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 1657,
      "training_step_time": 5.204714298248291
    },
    {
      "epoch": 6.320953369140625e-06,
      "model_forward_time": 2.433537244796753,
      "step": 1657
    },
    {
      "epoch": 6.320953369140625e-06,
      "step": 1657,
      "training_step_time": 10.01476526260376
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 3.1661555767059326,
      "step": 1658
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 1658,
      "training_step_time": 9.937945365905762
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 3.4110023975372314,
      "step": 1658
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 1658,
      "training_step_time": 10.181589365005493
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 3.138737916946411,
      "step": 1658
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 1658,
      "training_step_time": 9.900075197219849
    },
    {
      "epoch": 6.32476806640625e-06,
      "model_forward_time": 3.143106698989868,
      "step": 1658
    },
    {
      "epoch": 6.32476806640625e-06,
      "step": 1658,
      "training_step_time": 10.278863430023193
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 3.185953140258789,
      "step": 1659
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 1659,
      "training_step_time": 9.963030576705933
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 3.1446566581726074,
      "step": 1659
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 1659,
      "training_step_time": 9.917270183563232
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 3.3948750495910645,
      "step": 1659
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 1659,
      "training_step_time": 10.171800374984741
    },
    {
      "epoch": 6.328582763671875e-06,
      "model_forward_time": 3.1488397121429443,
      "step": 1659
    },
    {
      "epoch": 6.328582763671875e-06,
      "step": 1659,
      "training_step_time": 10.029577493667603
    },
    {
      "epoch": 6.3323974609375e-06,
      "grad_norm": 0.5945157408714294,
      "learning_rate": 1e-05,
      "loss": 0.119,
      "step": 1660
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 3.1964213848114014,
      "step": 1660
    },
    {
      "dynamics_loss_avg": 0.06255513690412044,
      "epoch": 6.3323974609375e-06,
      "step": 1660
    },
    {
      "action_loss_avg": 0.039884285628795625,
      "epoch": 6.3323974609375e-06,
      "step": 1660
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 1660,
      "training_step_time": 9.975823640823364
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 3.1462225914001465,
      "step": 1660
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 1660,
      "training_step_time": 9.920528888702393
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 3.1470744609832764,
      "step": 1660
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 1660,
      "training_step_time": 10.216273784637451
    },
    {
      "epoch": 6.3323974609375e-06,
      "model_forward_time": 3.1464247703552246,
      "step": 1660
    },
    {
      "epoch": 6.3323974609375e-06,
      "step": 1660,
      "training_step_time": 10.333906412124634
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 3.187293529510498,
      "step": 1661
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 1661,
      "training_step_time": 9.952995777130127
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 3.140942096710205,
      "step": 1661
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 1661,
      "training_step_time": 9.908504009246826
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 2.4319612979888916,
      "step": 1661
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 1661,
      "training_step_time": 7.457207202911377
    },
    {
      "epoch": 6.336212158203125e-06,
      "model_forward_time": 3.128485918045044,
      "step": 1661
    },
    {
      "epoch": 6.336212158203125e-06,
      "step": 1661,
      "training_step_time": 10.25379753112793
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 3.4759485721588135,
      "step": 1662
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 1662,
      "training_step_time": 10.241180658340454
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 3.1297483444213867,
      "step": 1662
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 1662,
      "training_step_time": 9.889219999313354
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 3.13275408744812,
      "step": 1662
    },
    {
      "dynamics_loss_avg": 0.06452971752732992,
      "epoch": 6.34002685546875e-06,
      "step": 1662
    },
    {
      "action_loss_avg": 0.039015933079645036,
      "epoch": 6.34002685546875e-06,
      "step": 1662
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 1662,
      "training_step_time": 9.894201040267944
    },
    {
      "epoch": 6.34002685546875e-06,
      "model_forward_time": 3.129504680633545,
      "step": 1662
    },
    {
      "epoch": 6.34002685546875e-06,
      "step": 1662,
      "training_step_time": 10.00580906867981
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 3.469843864440918,
      "step": 1663
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 1663,
      "training_step_time": 10.226941108703613
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 3.1306254863739014,
      "step": 1663
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 1663,
      "training_step_time": 9.890809059143066
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 3.136340618133545,
      "step": 1663
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 1663,
      "training_step_time": 9.899934530258179
    },
    {
      "epoch": 6.343841552734375e-06,
      "model_forward_time": 3.131857395172119,
      "step": 1663
    },
    {
      "epoch": 6.343841552734375e-06,
      "step": 1663,
      "training_step_time": 10.269045352935791
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 3.1909854412078857,
      "step": 1664
    },
    {
      "epoch": 6.34765625e-06,
      "step": 1664,
      "training_step_time": 9.962875604629517
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 3.398324966430664,
      "step": 1664
    },
    {
      "epoch": 6.34765625e-06,
      "step": 1664,
      "training_step_time": 10.165045022964478
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 3.1366660594940186,
      "step": 1664
    },
    {
      "epoch": 6.34765625e-06,
      "step": 1664,
      "training_step_time": 9.903146028518677
    },
    {
      "epoch": 6.34765625e-06,
      "model_forward_time": 3.1339242458343506,
      "step": 1664
    },
    {
      "epoch": 6.34765625e-06,
      "step": 1664,
      "training_step_time": 10.01841402053833
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 3.184892416000366,
      "step": 1665
    },
    {
      "dynamics_loss_avg": 0.07281026300042867,
      "epoch": 6.351470947265625e-06,
      "step": 1665
    },
    {
      "action_loss_avg": 0.047950323671102524,
      "epoch": 6.351470947265625e-06,
      "step": 1665
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 1665,
      "training_step_time": 9.960204601287842
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 3.141216516494751,
      "step": 1665
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 1665,
      "training_step_time": 9.910578727722168
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 3.1383543014526367,
      "step": 1665
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 1665,
      "training_step_time": 10.185590982437134
    },
    {
      "epoch": 6.351470947265625e-06,
      "model_forward_time": 3.135769844055176,
      "step": 1665
    },
    {
      "epoch": 6.351470947265625e-06,
      "step": 1665,
      "training_step_time": 10.27549123764038
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 3.1634469032287598,
      "step": 1666
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 1666,
      "training_step_time": 9.931544065475464
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 3.137850284576416,
      "step": 1666
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 1666,
      "training_step_time": 9.900684118270874
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 3.1316230297088623,
      "step": 1666
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 1666,
      "training_step_time": 9.893607139587402
    },
    {
      "epoch": 6.35528564453125e-06,
      "model_forward_time": 3.138594627380371,
      "step": 1666
    },
    {
      "epoch": 6.35528564453125e-06,
      "step": 1666,
      "training_step_time": 10.301695823669434
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 3.1697280406951904,
      "step": 1667
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 1667,
      "training_step_time": 9.938515901565552
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 2.4347448348999023,
      "step": 1667
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 1667,
      "training_step_time": 7.457775592803955
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 3.1359331607818604,
      "step": 1667
    },
    {
      "dynamics_loss_avg": 0.06656376868486405,
      "epoch": 6.359100341796875e-06,
      "step": 1667
    },
    {
      "action_loss_avg": 0.043014743644744155,
      "epoch": 6.359100341796875e-06,
      "step": 1667
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 1667,
      "training_step_time": 9.90309476852417
    },
    {
      "epoch": 6.359100341796875e-06,
      "model_forward_time": 3.1391587257385254,
      "step": 1667
    },
    {
      "epoch": 6.359100341796875e-06,
      "step": 1667,
      "training_step_time": 10.282294988632202
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 2.4657819271087646,
      "step": 1668
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 1668,
      "training_step_time": 7.496167421340942
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 3.4238898754119873,
      "step": 1668
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 1668,
      "training_step_time": 10.198482751846313
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 3.133225440979004,
      "step": 1668
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 1668,
      "training_step_time": 9.903255462646484
    },
    {
      "epoch": 6.3629150390625e-06,
      "model_forward_time": 3.1364974975585938,
      "step": 1668
    },
    {
      "epoch": 6.3629150390625e-06,
      "step": 1668,
      "training_step_time": 10.273637533187866
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 3.1519601345062256,
      "step": 1669
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 1669,
      "training_step_time": 9.925740242004395
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 3.1326534748077393,
      "step": 1669
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 1669,
      "training_step_time": 9.899500370025635
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 2.668261766433716,
      "step": 1669
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 1669,
      "training_step_time": 7.689509153366089
    },
    {
      "epoch": 6.366729736328125e-06,
      "model_forward_time": 3.134845733642578,
      "step": 1669
    },
    {
      "epoch": 6.366729736328125e-06,
      "step": 1669,
      "training_step_time": 10.01753544807434
    },
    {
      "epoch": 6.37054443359375e-06,
      "grad_norm": 0.7932475209236145,
      "learning_rate": 1e-05,
      "loss": 0.1163,
      "step": 1670
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 3.161226987838745,
      "step": 1670
    },
    {
      "dynamics_loss_avg": 0.06423645932227373,
      "epoch": 6.37054443359375e-06,
      "step": 1670
    },
    {
      "action_loss_avg": 0.04458716083317995,
      "epoch": 6.37054443359375e-06,
      "step": 1670
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 1670,
      "training_step_time": 9.929733753204346
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 3.1299214363098145,
      "step": 1670
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 1670,
      "training_step_time": 9.89826512336731
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 3.1410343647003174,
      "step": 1670
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 1670,
      "training_step_time": 10.178780555725098
    },
    {
      "epoch": 6.37054443359375e-06,
      "model_forward_time": 3.1449649333953857,
      "step": 1670
    },
    {
      "epoch": 6.37054443359375e-06,
      "step": 1670,
      "training_step_time": 10.288623571395874
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 3.186901092529297,
      "step": 1671
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 1671,
      "training_step_time": 9.960324764251709
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 3.137854814529419,
      "step": 1671
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 1671,
      "training_step_time": 9.90236759185791
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 3.1415798664093018,
      "step": 1671
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 1671,
      "training_step_time": 9.907556772232056
    },
    {
      "epoch": 6.374359130859375e-06,
      "model_forward_time": 3.420020818710327,
      "step": 1671
    },
    {
      "epoch": 6.374359130859375e-06,
      "step": 1671,
      "training_step_time": 10.284044027328491
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 3.18816876411438,
      "step": 1672
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 1672,
      "training_step_time": 9.958768129348755
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 2.4342002868652344,
      "step": 1672
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 1672,
      "training_step_time": 7.460347414016724
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 3.1381030082702637,
      "step": 1672
    },
    {
      "dynamics_loss_avg": 0.07170359939336776,
      "epoch": 6.378173828125e-06,
      "step": 1672
    },
    {
      "action_loss_avg": 0.044022710807621476,
      "epoch": 6.378173828125e-06,
      "step": 1672
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 1672,
      "training_step_time": 9.908239841461182
    },
    {
      "epoch": 6.378173828125e-06,
      "model_forward_time": 3.1356723308563232,
      "step": 1672
    },
    {
      "epoch": 6.378173828125e-06,
      "step": 1672,
      "training_step_time": 10.246435642242432
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 2.4701051712036133,
      "step": 1673
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 1673,
      "training_step_time": 7.4916181564331055
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 3.41839599609375,
      "step": 1673
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 1673,
      "training_step_time": 10.194240093231201
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 2.4361863136291504,
      "step": 1673
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 1673,
      "training_step_time": 7.459427833557129
    },
    {
      "epoch": 6.381988525390625e-06,
      "model_forward_time": 3.1365740299224854,
      "step": 1673
    },
    {
      "epoch": 6.381988525390625e-06,
      "step": 1673,
      "training_step_time": 10.283088207244873
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 3.178847551345825,
      "step": 1674
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 1674,
      "training_step_time": 9.95037293434143
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 3.1412816047668457,
      "step": 1674
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 1674,
      "training_step_time": 10.178772687911987
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 3.140131950378418,
      "step": 1674
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 1674,
      "training_step_time": 9.904858589172363
    },
    {
      "epoch": 6.38580322265625e-06,
      "model_forward_time": 3.133460521697998,
      "step": 1674
    },
    {
      "epoch": 6.38580322265625e-06,
      "step": 1674,
      "training_step_time": 10.000318765640259
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 3.16438627243042,
      "step": 1675
    },
    {
      "dynamics_loss_avg": 0.08598407246172428,
      "epoch": 6.389617919921875e-06,
      "step": 1675
    },
    {
      "action_loss_avg": 0.05559536684304476,
      "epoch": 6.389617919921875e-06,
      "step": 1675
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 1675,
      "training_step_time": 9.935988187789917
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 3.1400389671325684,
      "step": 1675
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 1675,
      "training_step_time": 9.908206462860107
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 3.13604998588562,
      "step": 1675
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 1675,
      "training_step_time": 10.16773796081543
    },
    {
      "epoch": 6.389617919921875e-06,
      "model_forward_time": 3.138808488845825,
      "step": 1675
    },
    {
      "epoch": 6.389617919921875e-06,
      "step": 1675,
      "training_step_time": 10.04910397529602
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 2.4743547439575195,
      "step": 1676
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 1676,
      "training_step_time": 7.500305891036987
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 3.134579658508301,
      "step": 1676
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 1676,
      "training_step_time": 9.903571367263794
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 3.137206792831421,
      "step": 1676
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 1676,
      "training_step_time": 9.912049055099487
    },
    {
      "epoch": 6.3934326171875e-06,
      "model_forward_time": 3.1368274688720703,
      "step": 1676
    },
    {
      "epoch": 6.3934326171875e-06,
      "step": 1676,
      "training_step_time": 10.314653873443604
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 3.178554058074951,
      "step": 1677
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 1677,
      "training_step_time": 9.950891971588135
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 2.4335672855377197,
      "step": 1677
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 1677,
      "training_step_time": 7.464084625244141
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 3.129221200942993,
      "step": 1677
    },
    {
      "dynamics_loss_avg": 0.06678617065772414,
      "epoch": 6.397247314453125e-06,
      "step": 1677
    },
    {
      "action_loss_avg": 0.04183125537820161,
      "epoch": 6.397247314453125e-06,
      "step": 1677
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 1677,
      "training_step_time": 9.912007570266724
    },
    {
      "epoch": 6.397247314453125e-06,
      "model_forward_time": 2.433485746383667,
      "step": 1677
    },
    {
      "epoch": 6.397247314453125e-06,
      "step": 1677,
      "training_step_time": 10.322423458099365
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 3.1412975788116455,
      "step": 1678
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 1678,
      "training_step_time": 9.914491653442383
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 3.412264347076416,
      "step": 1678
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 1678,
      "training_step_time": 10.1828293800354
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 3.134655714035034,
      "step": 1678
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 1678,
      "training_step_time": 9.905811309814453
    },
    {
      "epoch": 6.40106201171875e-06,
      "model_forward_time": 2.432251214981079,
      "step": 1678
    },
    {
      "epoch": 6.40106201171875e-06,
      "step": 1678,
      "training_step_time": 10.28087830543518
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 3.1682276725769043,
      "step": 1679
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 1679,
      "training_step_time": 9.94098973274231
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 3.1311569213867188,
      "step": 1679
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 1679,
      "training_step_time": 9.899704694747925
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 3.4062492847442627,
      "step": 1679
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 1679,
      "training_step_time": 10.179503679275513
    },
    {
      "epoch": 6.404876708984375e-06,
      "model_forward_time": 3.1380112171173096,
      "step": 1679
    },
    {
      "epoch": 6.404876708984375e-06,
      "step": 1679,
      "training_step_time": 10.279383897781372
    },
    {
      "epoch": 6.40869140625e-06,
      "grad_norm": 0.7899755239486694,
      "learning_rate": 1e-05,
      "loss": 0.1158,
      "step": 1680
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 3.1492061614990234,
      "step": 1680
    },
    {
      "dynamics_loss_avg": 0.06952318139374256,
      "epoch": 6.40869140625e-06,
      "step": 1680
    },
    {
      "action_loss_avg": 0.052002336643636224,
      "epoch": 6.40869140625e-06,
      "step": 1680
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 1680,
      "training_step_time": 9.92754316329956
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 3.132660150527954,
      "step": 1680
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 1680,
      "training_step_time": 9.90268850326538
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 3.4006311893463135,
      "step": 1680
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 1680,
      "training_step_time": 10.163375854492188
    },
    {
      "epoch": 6.40869140625e-06,
      "model_forward_time": 2.427342653274536,
      "step": 1680
    },
    {
      "epoch": 6.40869140625e-06,
      "step": 1680,
      "training_step_time": 10.3003511428833
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 3.1684305667877197,
      "step": 1681
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 1681,
      "training_step_time": 9.93066143989563
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 3.130599021911621,
      "step": 1681
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 1681,
      "training_step_time": 9.893836498260498
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 2.4282233715057373,
      "step": 1681
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 1681,
      "training_step_time": 7.4517903327941895
    },
    {
      "epoch": 6.412506103515625e-06,
      "model_forward_time": 3.125817060470581,
      "step": 1681
    },
    {
      "epoch": 6.412506103515625e-06,
      "step": 1681,
      "training_step_time": 10.01871395111084
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 2.794692277908325,
      "step": 1682
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 1682,
      "training_step_time": 7.813664197921753
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 3.1288509368896484,
      "step": 1682
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 1682,
      "training_step_time": 9.90632963180542
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 3.135141611099243,
      "step": 1682
    },
    {
      "dynamics_loss_avg": 0.07903760448098182,
      "epoch": 6.41632080078125e-06,
      "step": 1682
    },
    {
      "action_loss_avg": 0.04714870844036341,
      "epoch": 6.41632080078125e-06,
      "step": 1682
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 1682,
      "training_step_time": 9.904141664505005
    },
    {
      "epoch": 6.41632080078125e-06,
      "model_forward_time": 2.432365655899048,
      "step": 1682
    },
    {
      "epoch": 6.41632080078125e-06,
      "step": 1682,
      "training_step_time": 10.267110586166382
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 3.168722152709961,
      "step": 1683
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 1683,
      "training_step_time": 9.940629959106445
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 3.373027801513672,
      "step": 1683
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 1683,
      "training_step_time": 10.143836259841919
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 3.1344809532165527,
      "step": 1683
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 1683,
      "training_step_time": 9.900191307067871
    },
    {
      "epoch": 6.420135498046875e-06,
      "model_forward_time": 3.134866237640381,
      "step": 1683
    },
    {
      "epoch": 6.420135498046875e-06,
      "step": 1683,
      "training_step_time": 10.26093602180481
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 3.1863110065460205,
      "step": 1684
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 1684,
      "training_step_time": 9.960595607757568
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 2.436345338821411,
      "step": 1684
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 1684,
      "training_step_time": 7.467318058013916
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 3.409546136856079,
      "step": 1684
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 1684,
      "training_step_time": 10.177059650421143
    },
    {
      "epoch": 6.4239501953125e-06,
      "model_forward_time": 3.134474754333496,
      "step": 1684
    },
    {
      "epoch": 6.4239501953125e-06,
      "step": 1684,
      "training_step_time": 10.281472206115723
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 3.1735100746154785,
      "step": 1685
    },
    {
      "dynamics_loss_avg": 0.08329014927148819,
      "epoch": 6.427764892578125e-06,
      "step": 1685
    },
    {
      "action_loss_avg": 0.04807761088013649,
      "epoch": 6.427764892578125e-06,
      "step": 1685
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 1685,
      "training_step_time": 9.943989515304565
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 3.139312744140625,
      "step": 1685
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 1685,
      "training_step_time": 9.910272121429443
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 3.1374568939208984,
      "step": 1685
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 1685,
      "training_step_time": 9.905802249908447
    },
    {
      "epoch": 6.427764892578125e-06,
      "model_forward_time": 3.3849270343780518,
      "step": 1685
    },
    {
      "epoch": 6.427764892578125e-06,
      "step": 1685,
      "training_step_time": 10.252971172332764
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 3.1728806495666504,
      "step": 1686
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 1686,
      "training_step_time": 9.943673610687256
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 2.4348838329315186,
      "step": 1686
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 1686,
      "training_step_time": 7.464890480041504
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 3.1495347023010254,
      "step": 1686
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 1686,
      "training_step_time": 9.923525094985962
    },
    {
      "epoch": 6.43157958984375e-06,
      "model_forward_time": 3.1384730339050293,
      "step": 1686
    },
    {
      "epoch": 6.43157958984375e-06,
      "step": 1686,
      "training_step_time": 10.036436557769775
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 2.7984437942504883,
      "step": 1687
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 1687,
      "training_step_time": 7.826994180679321
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 3.1372170448303223,
      "step": 1687
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 1687,
      "training_step_time": 9.908768892288208
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 3.1350741386413574,
      "step": 1687
    },
    {
      "dynamics_loss_avg": 0.06155865676701069,
      "epoch": 6.435394287109375e-06,
      "step": 1687
    },
    {
      "action_loss_avg": 0.041240304335951807,
      "epoch": 6.435394287109375e-06,
      "step": 1687
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 1687,
      "training_step_time": 9.911453008651733
    },
    {
      "epoch": 6.435394287109375e-06,
      "model_forward_time": 3.1409525871276855,
      "step": 1687
    },
    {
      "epoch": 6.435394287109375e-06,
      "step": 1687,
      "training_step_time": 10.313015937805176
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 3.1829426288604736,
      "step": 1688
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 1688,
      "training_step_time": 9.958208084106445
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 3.421449899673462,
      "step": 1688
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 1688,
      "training_step_time": 10.188372373580933
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 3.1401023864746094,
      "step": 1688
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 1688,
      "training_step_time": 9.905522346496582
    },
    {
      "epoch": 6.439208984375e-06,
      "model_forward_time": 3.1373136043548584,
      "step": 1688
    },
    {
      "epoch": 6.439208984375e-06,
      "step": 1688,
      "training_step_time": 9.999059677124023
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 3.1877572536468506,
      "step": 1689
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 1689,
      "training_step_time": 9.959462881088257
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 3.419675827026367,
      "step": 1689
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 1689,
      "training_step_time": 10.18759822845459
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 3.137204170227051,
      "step": 1689
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 1689,
      "training_step_time": 9.905465364456177
    },
    {
      "epoch": 6.443023681640625e-06,
      "model_forward_time": 3.138331890106201,
      "step": 1689
    },
    {
      "epoch": 6.443023681640625e-06,
      "step": 1689,
      "training_step_time": 10.26757550239563
    },
    {
      "epoch": 6.44683837890625e-06,
      "grad_norm": 0.5806781649589539,
      "learning_rate": 1e-05,
      "loss": 0.1176,
      "step": 1690
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 3.1787359714508057,
      "step": 1690
    },
    {
      "dynamics_loss_avg": 0.06552002113312483,
      "epoch": 6.44683837890625e-06,
      "step": 1690
    },
    {
      "action_loss_avg": 0.033574006194248794,
      "epoch": 6.44683837890625e-06,
      "step": 1690
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 1690,
      "training_step_time": 9.951170682907104
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 3.1378679275512695,
      "step": 1690
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 1690,
      "training_step_time": 9.905873775482178
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 3.1410763263702393,
      "step": 1690
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 1690,
      "training_step_time": 10.182273864746094
    },
    {
      "epoch": 6.44683837890625e-06,
      "model_forward_time": 3.1537156105041504,
      "step": 1690
    },
    {
      "epoch": 6.44683837890625e-06,
      "step": 1690,
      "training_step_time": 10.020993947982788
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 2.4482245445251465,
      "step": 1691
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 1691,
      "training_step_time": 7.479885101318359
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 3.1389670372009277,
      "step": 1691
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 1691,
      "training_step_time": 9.912613868713379
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 3.1457149982452393,
      "step": 1691
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 1691,
      "training_step_time": 9.914846420288086
    },
    {
      "epoch": 6.450653076171875e-06,
      "model_forward_time": 2.439685106277466,
      "step": 1691
    },
    {
      "epoch": 6.450653076171875e-06,
      "step": 1691,
      "training_step_time": 10.265474557876587
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 3.191195487976074,
      "step": 1692
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 1692,
      "training_step_time": 9.965409278869629
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 2.4372763633728027,
      "step": 1692
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 1692,
      "training_step_time": 7.463776111602783
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 3.1365058422088623,
      "step": 1692
    },
    {
      "dynamics_loss_avg": 0.06673171389847994,
      "epoch": 6.4544677734375e-06,
      "step": 1692
    },
    {
      "action_loss_avg": 0.034335680026561025,
      "epoch": 6.4544677734375e-06,
      "step": 1692
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 1692,
      "training_step_time": 9.906508684158325
    },
    {
      "epoch": 6.4544677734375e-06,
      "model_forward_time": 3.137582302093506,
      "step": 1692
    },
    {
      "epoch": 6.4544677734375e-06,
      "step": 1692,
      "training_step_time": 10.235239267349243
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 3.1723711490631104,
      "step": 1693
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 1693,
      "training_step_time": 9.9376962184906
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 3.390442371368408,
      "step": 1693
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 1693,
      "training_step_time": 10.154610395431519
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 2.434617280960083,
      "step": 1693
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 1693,
      "training_step_time": 7.459366083145142
    },
    {
      "epoch": 6.458282470703125e-06,
      "model_forward_time": 3.1298041343688965,
      "step": 1693
    },
    {
      "epoch": 6.458282470703125e-06,
      "step": 1693,
      "training_step_time": 10.295114040374756
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 3.1711080074310303,
      "step": 1694
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 1694,
      "training_step_time": 9.934844732284546
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 3.1372385025024414,
      "step": 1694
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 1694,
      "training_step_time": 10.171581029891968
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 3.1430623531341553,
      "step": 1694
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 1694,
      "training_step_time": 9.913529634475708
    },
    {
      "epoch": 6.46209716796875e-06,
      "model_forward_time": 3.1399173736572266,
      "step": 1694
    },
    {
      "epoch": 6.46209716796875e-06,
      "step": 1694,
      "training_step_time": 10.25914192199707
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 3.1810834407806396,
      "step": 1695
    },
    {
      "dynamics_loss_avg": 0.0673095241189003,
      "epoch": 6.465911865234375e-06,
      "step": 1695
    },
    {
      "action_loss_avg": 0.04174007121473551,
      "epoch": 6.465911865234375e-06,
      "step": 1695
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 1695,
      "training_step_time": 9.953452348709106
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 3.1416878700256348,
      "step": 1695
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 1695,
      "training_step_time": 9.91401743888855
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 3.1454355716705322,
      "step": 1695
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 1695,
      "training_step_time": 10.224929809570312
    },
    {
      "epoch": 6.465911865234375e-06,
      "model_forward_time": 3.145745038986206,
      "step": 1695
    },
    {
      "epoch": 6.465911865234375e-06,
      "step": 1695,
      "training_step_time": 10.318826675415039
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 2.4632678031921387,
      "step": 1696
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 1696,
      "training_step_time": 7.4885478019714355
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 3.142747402191162,
      "step": 1696
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 1696,
      "training_step_time": 9.915465593338013
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 3.1418962478637695,
      "step": 1696
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 1696,
      "training_step_time": 9.910743713378906
    },
    {
      "epoch": 6.4697265625e-06,
      "model_forward_time": 3.1442604064941406,
      "step": 1696
    },
    {
      "epoch": 6.4697265625e-06,
      "step": 1696,
      "training_step_time": 10.286380290985107
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 3.159987211227417,
      "step": 1697
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 1697,
      "training_step_time": 9.933837890625
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 3.1447746753692627,
      "step": 1697
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 1697,
      "training_step_time": 9.919175624847412
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 3.1402506828308105,
      "step": 1697
    },
    {
      "dynamics_loss_avg": 0.05882885754108429,
      "epoch": 6.473541259765625e-06,
      "step": 1697
    },
    {
      "action_loss_avg": 0.042509699054062364,
      "epoch": 6.473541259765625e-06,
      "step": 1697
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 1697,
      "training_step_time": 9.916658639907837
    },
    {
      "epoch": 6.473541259765625e-06,
      "model_forward_time": 2.4372496604919434,
      "step": 1697
    },
    {
      "epoch": 6.473541259765625e-06,
      "step": 1697,
      "training_step_time": 10.284278869628906
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 2.465282678604126,
      "step": 1698
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 1698,
      "training_step_time": 7.492663383483887
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 2.717129707336426,
      "step": 1698
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 1698,
      "training_step_time": 7.742201566696167
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 3.1387791633605957,
      "step": 1698
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 1698,
      "training_step_time": 9.904088020324707
    },
    {
      "epoch": 6.47735595703125e-06,
      "model_forward_time": 3.135010004043579,
      "step": 1698
    },
    {
      "epoch": 6.47735595703125e-06,
      "step": 1698,
      "training_step_time": 10.010525226593018
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 2.473496913909912,
      "step": 1699
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 1699,
      "training_step_time": 7.5031898021698
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 3.135868549346924,
      "step": 1699
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 1699,
      "training_step_time": 9.904475450515747
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 3.41221022605896,
      "step": 1699
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 1699,
      "training_step_time": 10.177694320678711
    },
    {
      "epoch": 6.481170654296875e-06,
      "model_forward_time": 3.1400606632232666,
      "step": 1699
    },
    {
      "epoch": 6.481170654296875e-06,
      "step": 1699,
      "training_step_time": 10.250221014022827
    },
    {
      "epoch": 6.4849853515625e-06,
      "grad_norm": 0.7030178904533386,
      "learning_rate": 1e-05,
      "loss": 0.1133,
      "step": 1700
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 3.17722749710083,
      "step": 1700
    },
    {
      "dynamics_loss_avg": 0.05961447544395924,
      "epoch": 6.4849853515625e-06,
      "step": 1700
    },
    {
      "action_loss_avg": 0.03209217032417655,
      "epoch": 6.4849853515625e-06,
      "step": 1700
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 1700,
      "training_step_time": 9.946630239486694
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 3.1369614601135254,
      "step": 1700
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 1700,
      "training_step_time": 9.902263164520264
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 3.139378547668457,
      "step": 1700
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 1700,
      "training_step_time": 10.169690132141113
    },
    {
      "epoch": 6.4849853515625e-06,
      "model_forward_time": 3.1404361724853516,
      "step": 1700
    },
    {
      "epoch": 6.4849853515625e-06,
      "step": 1700,
      "training_step_time": 10.291125774383545
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 3.1779139041900635,
      "step": 1701
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 1701,
      "training_step_time": 9.941611766815186
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 2.4285943508148193,
      "step": 1701
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 1701,
      "training_step_time": 7.444446802139282
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 3.1214981079101562,
      "step": 1701
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 1701,
      "training_step_time": 9.880063533782959
    },
    {
      "epoch": 6.488800048828125e-06,
      "model_forward_time": 3.124234914779663,
      "step": 1701
    },
    {
      "epoch": 6.488800048828125e-06,
      "step": 1701,
      "training_step_time": 10.277016639709473
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 3.473938226699829,
      "step": 1702
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 1702,
      "training_step_time": 10.23070240020752
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 3.127324342727661,
      "step": 1702
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 1702,
      "training_step_time": 9.884236335754395
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 3.1310088634490967,
      "step": 1702
    },
    {
      "dynamics_loss_avg": 0.056857268698513506,
      "epoch": 6.49261474609375e-06,
      "step": 1702
    },
    {
      "action_loss_avg": 0.029378190636634827,
      "epoch": 6.49261474609375e-06,
      "step": 1702
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 1702,
      "training_step_time": 9.899126529693604
    },
    {
      "epoch": 6.49261474609375e-06,
      "model_forward_time": 3.1279773712158203,
      "step": 1702
    },
    {
      "epoch": 6.49261474609375e-06,
      "step": 1702,
      "training_step_time": 10.249056339263916
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 3.4878733158111572,
      "step": 1703
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 1703,
      "training_step_time": 10.257750034332275
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 3.13690447807312,
      "step": 1703
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 1703,
      "training_step_time": 9.909385204315186
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 3.139437198638916,
      "step": 1703
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 1703,
      "training_step_time": 9.909204959869385
    },
    {
      "epoch": 6.496429443359375e-06,
      "model_forward_time": 3.1363701820373535,
      "step": 1703
    },
    {
      "epoch": 6.496429443359375e-06,
      "step": 1703,
      "training_step_time": 10.019564151763916
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 3.1830787658691406,
      "step": 1704
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 1704,
      "training_step_time": 9.961090803146362
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 3.397264003753662,
      "step": 1704
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 1704,
      "training_step_time": 10.16922402381897
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 3.1363987922668457,
      "step": 1704
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 1704,
      "training_step_time": 9.908307075500488
    },
    {
      "epoch": 6.500244140625e-06,
      "model_forward_time": 3.1363332271575928,
      "step": 1704
    },
    {
      "epoch": 6.500244140625e-06,
      "step": 1704,
      "training_step_time": 10.307138919830322
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 3.165480852127075,
      "step": 1705
    },
    {
      "dynamics_loss_avg": 0.0744224689900875,
      "epoch": 6.504058837890625e-06,
      "step": 1705
    },
    {
      "action_loss_avg": 0.04019313668832183,
      "epoch": 6.504058837890625e-06,
      "step": 1705
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 1705,
      "training_step_time": 9.941091775894165
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 3.1374833583831787,
      "step": 1705
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 1705,
      "training_step_time": 9.908530473709106
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 3.131601572036743,
      "step": 1705
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 1705,
      "training_step_time": 10.178025960922241
    },
    {
      "epoch": 6.504058837890625e-06,
      "model_forward_time": 3.137240171432495,
      "step": 1705
    },
    {
      "epoch": 6.504058837890625e-06,
      "step": 1705,
      "training_step_time": 10.262826681137085
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 3.169498920440674,
      "step": 1706
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 1706,
      "training_step_time": 9.944343328475952
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 3.1356639862060547,
      "step": 1706
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 1706,
      "training_step_time": 9.920943975448608
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 3.1313655376434326,
      "step": 1706
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 1706,
      "training_step_time": 9.90196180343628
    },
    {
      "epoch": 6.50787353515625e-06,
      "model_forward_time": 2.431853771209717,
      "step": 1706
    },
    {
      "epoch": 6.50787353515625e-06,
      "step": 1706,
      "training_step_time": 10.291694164276123
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 3.459886074066162,
      "step": 1707
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 1707,
      "training_step_time": 10.230513095855713
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 3.1359968185424805,
      "step": 1707
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 1707,
      "training_step_time": 9.90276551246643
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 3.1351230144500732,
      "step": 1707
    },
    {
      "dynamics_loss_avg": 0.05833082925528288,
      "epoch": 6.511688232421875e-06,
      "step": 1707
    },
    {
      "action_loss_avg": 0.04130395762622356,
      "epoch": 6.511688232421875e-06,
      "step": 1707
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 1707,
      "training_step_time": 9.909512042999268
    },
    {
      "epoch": 6.511688232421875e-06,
      "model_forward_time": 3.1355206966400146,
      "step": 1707
    },
    {
      "epoch": 6.511688232421875e-06,
      "step": 1707,
      "training_step_time": 10.023636817932129
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 3.144221305847168,
      "step": 1708
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 1708,
      "training_step_time": 10.220177412033081
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 2.432229995727539,
      "step": 1708
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 1708,
      "training_step_time": 7.461799383163452
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 2.426760196685791,
      "step": 1708
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 1708,
      "training_step_time": 7.451827764511108
    },
    {
      "epoch": 6.5155029296875e-06,
      "model_forward_time": 3.126283645629883,
      "step": 1708
    },
    {
      "epoch": 6.5155029296875e-06,
      "step": 1708,
      "training_step_time": 10.024498462677002
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 3.182138204574585,
      "step": 1709
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 1709,
      "training_step_time": 9.9534592628479
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 3.1311845779418945,
      "step": 1709
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 1709,
      "training_step_time": 9.903525590896606
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 3.389280319213867,
      "step": 1709
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 1709,
      "training_step_time": 10.15838885307312
    },
    {
      "epoch": 6.519317626953125e-06,
      "model_forward_time": 3.1382076740264893,
      "step": 1709
    },
    {
      "epoch": 6.519317626953125e-06,
      "step": 1709,
      "training_step_time": 10.25905442237854
    },
    {
      "epoch": 6.52313232421875e-06,
      "grad_norm": 0.7184993624687195,
      "learning_rate": 1e-05,
      "loss": 0.1113,
      "step": 1710
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 3.1833202838897705,
      "step": 1710
    },
    {
      "dynamics_loss_avg": 0.05849610436707735,
      "epoch": 6.52313232421875e-06,
      "step": 1710
    },
    {
      "action_loss_avg": 0.036034838669002055,
      "epoch": 6.52313232421875e-06,
      "step": 1710
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 1710,
      "training_step_time": 9.96183466911316
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 3.1365530490875244,
      "step": 1710
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 1710,
      "training_step_time": 9.913590431213379
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 3.1419506072998047,
      "step": 1710
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 1710,
      "training_step_time": 9.910266876220703
    },
    {
      "epoch": 6.52313232421875e-06,
      "model_forward_time": 3.4045822620391846,
      "step": 1710
    },
    {
      "epoch": 6.52313232421875e-06,
      "step": 1710,
      "training_step_time": 10.265244245529175
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 2.4801783561706543,
      "step": 1711
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 1711,
      "training_step_time": 7.506094217300415
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 3.132941961288452,
      "step": 1711
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 1711,
      "training_step_time": 9.901825428009033
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 3.135496139526367,
      "step": 1711
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 1711,
      "training_step_time": 9.899887084960938
    },
    {
      "epoch": 6.526947021484375e-06,
      "model_forward_time": 3.42867112159729,
      "step": 1711
    },
    {
      "epoch": 6.526947021484375e-06,
      "step": 1711,
      "training_step_time": 10.306204795837402
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 3.178701877593994,
      "step": 1712
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 1712,
      "training_step_time": 9.945644617080688
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 3.1363461017608643,
      "step": 1712
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 1712,
      "training_step_time": 9.903708219528198
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 3.1404736042022705,
      "step": 1712
    },
    {
      "dynamics_loss_avg": 0.07949067316949368,
      "epoch": 6.53076171875e-06,
      "step": 1712
    },
    {
      "action_loss_avg": 0.047882305644452575,
      "epoch": 6.53076171875e-06,
      "step": 1712
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 1712,
      "training_step_time": 9.91294527053833
    },
    {
      "epoch": 6.53076171875e-06,
      "model_forward_time": 3.134779691696167,
      "step": 1712
    },
    {
      "epoch": 6.53076171875e-06,
      "step": 1712,
      "training_step_time": 10.337062358856201
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 2.4416654109954834,
      "step": 1713
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 1713,
      "training_step_time": 7.468981027603149
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 3.409499168395996,
      "step": 1713
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 1713,
      "training_step_time": 10.17952561378479
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 3.148097515106201,
      "step": 1713
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 1713,
      "training_step_time": 9.919586181640625
    },
    {
      "epoch": 6.534576416015625e-06,
      "model_forward_time": 2.442018985748291,
      "step": 1713
    },
    {
      "epoch": 6.534576416015625e-06,
      "step": 1713,
      "training_step_time": 10.315647602081299
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 3.188570261001587,
      "step": 1714
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 1714,
      "training_step_time": 9.963553190231323
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 3.1488471031188965,
      "step": 1714
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 1714,
      "training_step_time": 10.193859815597534
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 3.1423933506011963,
      "step": 1714
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 1714,
      "training_step_time": 9.915326118469238
    },
    {
      "epoch": 6.53839111328125e-06,
      "model_forward_time": 3.142261028289795,
      "step": 1714
    },
    {
      "epoch": 6.53839111328125e-06,
      "step": 1714,
      "training_step_time": 10.301141023635864
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 3.177603244781494,
      "step": 1715
    },
    {
      "dynamics_loss_avg": 0.061147022247314456,
      "epoch": 6.542205810546875e-06,
      "step": 1715
    },
    {
      "action_loss_avg": 0.03759426325559616,
      "epoch": 6.542205810546875e-06,
      "step": 1715
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 1715,
      "training_step_time": 9.945759773254395
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 3.138313055038452,
      "step": 1715
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 1715,
      "training_step_time": 9.911080837249756
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 3.138331413269043,
      "step": 1715
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 1715,
      "training_step_time": 10.185362100601196
    },
    {
      "epoch": 6.542205810546875e-06,
      "model_forward_time": 3.13680100440979,
      "step": 1715
    },
    {
      "epoch": 6.542205810546875e-06,
      "step": 1715,
      "training_step_time": 10.025750875473022
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 3.178649663925171,
      "step": 1716
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 1716,
      "training_step_time": 9.954481601715088
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 3.1405515670776367,
      "step": 1716
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 1716,
      "training_step_time": 9.913249015808105
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 3.139848470687866,
      "step": 1716
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 1716,
      "training_step_time": 9.91715693473816
    },
    {
      "epoch": 6.5460205078125e-06,
      "model_forward_time": 3.4317662715911865,
      "step": 1716
    },
    {
      "epoch": 6.5460205078125e-06,
      "step": 1716,
      "training_step_time": 10.293959140777588
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 3.1720945835113525,
      "step": 1717
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 1717,
      "training_step_time": 9.946157455444336
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 3.141263961791992,
      "step": 1717
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 1717,
      "training_step_time": 9.9151611328125
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 3.1389622688293457,
      "step": 1717
    },
    {
      "dynamics_loss_avg": 0.0777380496263504,
      "epoch": 6.549835205078125e-06,
      "step": 1717
    },
    {
      "action_loss_avg": 0.06528351157903671,
      "epoch": 6.549835205078125e-06,
      "step": 1717
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 1717,
      "training_step_time": 9.914607524871826
    },
    {
      "epoch": 6.549835205078125e-06,
      "model_forward_time": 3.1413724422454834,
      "step": 1717
    },
    {
      "epoch": 6.549835205078125e-06,
      "step": 1717,
      "training_step_time": 10.29093861579895
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 3.4239590167999268,
      "step": 1718
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 1718,
      "training_step_time": 10.19640564918518
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 2.435338020324707,
      "step": 1718
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 1718,
      "training_step_time": 7.46189022064209
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 3.137693405151367,
      "step": 1718
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 1718,
      "training_step_time": 9.907631158828735
    },
    {
      "epoch": 6.55364990234375e-06,
      "model_forward_time": 3.139399528503418,
      "step": 1718
    },
    {
      "epoch": 6.55364990234375e-06,
      "step": 1718,
      "training_step_time": 10.290021896362305
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 3.1769790649414062,
      "step": 1719
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 1719,
      "training_step_time": 9.94374394416809
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 3.134721517562866,
      "step": 1719
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 1719,
      "training_step_time": 10.168984651565552
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 3.1339282989501953,
      "step": 1719
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 1719,
      "training_step_time": 9.897063732147217
    },
    {
      "epoch": 6.557464599609375e-06,
      "model_forward_time": 3.1390633583068848,
      "step": 1719
    },
    {
      "epoch": 6.557464599609375e-06,
      "step": 1719,
      "training_step_time": 10.297029733657837
    },
    {
      "epoch": 6.561279296875e-06,
      "grad_norm": 0.6637210845947266,
      "learning_rate": 1e-05,
      "loss": 0.1151,
      "step": 1720
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 3.1696605682373047,
      "step": 1720
    },
    {
      "dynamics_loss_avg": 0.06086495704948902,
      "epoch": 6.561279296875e-06,
      "step": 1720
    },
    {
      "action_loss_avg": 0.04278061287477612,
      "epoch": 6.561279296875e-06,
      "step": 1720
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 1720,
      "training_step_time": 9.939433336257935
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 3.137538194656372,
      "step": 1720
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 1720,
      "training_step_time": 9.906131982803345
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 3.4333136081695557,
      "step": 1720
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 1720,
      "training_step_time": 10.200299978256226
    },
    {
      "epoch": 6.561279296875e-06,
      "model_forward_time": 3.1400747299194336,
      "step": 1720
    },
    {
      "epoch": 6.561279296875e-06,
      "step": 1720,
      "training_step_time": 10.019607067108154
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 3.19260573387146,
      "step": 1721
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 1721,
      "training_step_time": 9.965107440948486
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 3.139650821685791,
      "step": 1721
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 1721,
      "training_step_time": 9.90645432472229
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 3.143719434738159,
      "step": 1721
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 1721,
      "training_step_time": 9.92047667503357
    },
    {
      "epoch": 6.565093994140625e-06,
      "model_forward_time": 3.145124912261963,
      "step": 1721
    },
    {
      "epoch": 6.565093994140625e-06,
      "step": 1721,
      "training_step_time": 10.31432580947876
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 3.1904640197753906,
      "step": 1722
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 1722,
      "training_step_time": 9.964212894439697
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 3.141331672668457,
      "step": 1722
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 1722,
      "training_step_time": 9.914010286331177
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 3.1466081142425537,
      "step": 1722
    },
    {
      "dynamics_loss_avg": 0.06181070283055305,
      "epoch": 6.56890869140625e-06,
      "step": 1722
    },
    {
      "action_loss_avg": 0.04056569524109364,
      "epoch": 6.56890869140625e-06,
      "step": 1722
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 1722,
      "training_step_time": 9.923184156417847
    },
    {
      "epoch": 6.56890869140625e-06,
      "model_forward_time": 3.1412532329559326,
      "step": 1722
    },
    {
      "epoch": 6.56890869140625e-06,
      "step": 1722,
      "training_step_time": 10.295394897460938
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 3.451998472213745,
      "step": 1723
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 1723,
      "training_step_time": 10.226138353347778
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 3.1362874507904053,
      "step": 1723
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 1723,
      "training_step_time": 9.907255411148071
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 3.1406731605529785,
      "step": 1723
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 1723,
      "training_step_time": 9.906115531921387
    },
    {
      "epoch": 6.572723388671875e-06,
      "model_forward_time": 3.139559507369995,
      "step": 1723
    },
    {
      "epoch": 6.572723388671875e-06,
      "step": 1723,
      "training_step_time": 10.271245241165161
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 3.1486899852752686,
      "step": 1724
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 1724,
      "training_step_time": 9.913983821868896
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 3.4251890182495117,
      "step": 1724
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 1724,
      "training_step_time": 10.193061828613281
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 2.438643455505371,
      "step": 1724
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 1724,
      "training_step_time": 7.4600090980529785
    },
    {
      "epoch": 6.5765380859375e-06,
      "model_forward_time": 3.1321938037872314,
      "step": 1724
    },
    {
      "epoch": 6.5765380859375e-06,
      "step": 1724,
      "training_step_time": 10.003377914428711
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 3.1450018882751465,
      "step": 1725
    },
    {
      "dynamics_loss_avg": 0.07948892936110497,
      "epoch": 6.580352783203125e-06,
      "step": 1725
    },
    {
      "action_loss_avg": 0.04562247712165117,
      "epoch": 6.580352783203125e-06,
      "step": 1725
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 1725,
      "training_step_time": 9.916708946228027
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 3.1389260292053223,
      "step": 1725
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 1725,
      "training_step_time": 9.90579342842102
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 1.7471332550048828,
      "step": 1725
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 1725,
      "training_step_time": 5.201876401901245
    },
    {
      "epoch": 6.580352783203125e-06,
      "model_forward_time": 3.394090414047241,
      "step": 1725
    },
    {
      "epoch": 6.580352783203125e-06,
      "step": 1725,
      "training_step_time": 10.260818481445312
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 3.1756608486175537,
      "step": 1726
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 1726,
      "training_step_time": 9.944327354431152
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 3.142322063446045,
      "step": 1726
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 1726,
      "training_step_time": 9.916658401489258
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 3.143244981765747,
      "step": 1726
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 1726,
      "training_step_time": 9.919236421585083
    },
    {
      "epoch": 6.58416748046875e-06,
      "model_forward_time": 3.40720534324646,
      "step": 1726
    },
    {
      "epoch": 6.58416748046875e-06,
      "step": 1726,
      "training_step_time": 10.276051998138428
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 3.190295696258545,
      "step": 1727
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 1727,
      "training_step_time": 9.964908123016357
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 3.1424241065979004,
      "step": 1727
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 1727,
      "training_step_time": 9.915882587432861
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 3.1430118083953857,
      "step": 1727
    },
    {
      "dynamics_loss_avg": 0.06548929065465928,
      "epoch": 6.587982177734375e-06,
      "step": 1727
    },
    {
      "action_loss_avg": 0.046528136171400544,
      "epoch": 6.587982177734375e-06,
      "step": 1727
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 1727,
      "training_step_time": 9.920229196548462
    },
    {
      "epoch": 6.587982177734375e-06,
      "model_forward_time": 3.1476526260375977,
      "step": 1727
    },
    {
      "epoch": 6.587982177734375e-06,
      "step": 1727,
      "training_step_time": 10.289759159088135
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 3.150883197784424,
      "step": 1728
    },
    {
      "epoch": 6.591796875e-06,
      "step": 1728,
      "training_step_time": 10.210891723632812
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 3.144211530685425,
      "step": 1728
    },
    {
      "epoch": 6.591796875e-06,
      "step": 1728,
      "training_step_time": 9.923677444458008
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 3.140734910964966,
      "step": 1728
    },
    {
      "epoch": 6.591796875e-06,
      "step": 1728,
      "training_step_time": 9.916980981826782
    },
    {
      "epoch": 6.591796875e-06,
      "model_forward_time": 2.4366321563720703,
      "step": 1728
    },
    {
      "epoch": 6.591796875e-06,
      "step": 1728,
      "training_step_time": 10.288280963897705
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 2.4423601627349854,
      "step": 1729
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 1729,
      "training_step_time": 7.468916893005371
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 3.4168481826782227,
      "step": 1729
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 1729,
      "training_step_time": 10.182989597320557
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 3.1324422359466553,
      "step": 1729
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 1729,
      "training_step_time": 9.901919603347778
    },
    {
      "epoch": 6.595611572265625e-06,
      "model_forward_time": 2.427199602127075,
      "step": 1729
    },
    {
      "epoch": 6.595611572265625e-06,
      "step": 1729,
      "training_step_time": 10.239818334579468
    },
    {
      "epoch": 6.59942626953125e-06,
      "grad_norm": 0.890254020690918,
      "learning_rate": 1e-05,
      "loss": 0.1136,
      "step": 1730
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 3.1799914836883545,
      "step": 1730
    },
    {
      "dynamics_loss_avg": 0.05130659993737936,
      "epoch": 6.59942626953125e-06,
      "step": 1730
    },
    {
      "action_loss_avg": 0.02883147485554218,
      "epoch": 6.59942626953125e-06,
      "step": 1730
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 1730,
      "training_step_time": 9.950563907623291
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 3.1327693462371826,
      "step": 1730
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 1730,
      "training_step_time": 9.898797273635864
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 3.1325557231903076,
      "step": 1730
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 1730,
      "training_step_time": 9.903345584869385
    },
    {
      "epoch": 6.59942626953125e-06,
      "model_forward_time": 3.397226333618164,
      "step": 1730
    },
    {
      "epoch": 6.59942626953125e-06,
      "step": 1730,
      "training_step_time": 10.258845567703247
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 2.4821486473083496,
      "step": 1731
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 1731,
      "training_step_time": 7.518658638000488
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 3.142347812652588,
      "step": 1731
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 1731,
      "training_step_time": 9.91958236694336
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 3.149650812149048,
      "step": 1731
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 1731,
      "training_step_time": 9.928277254104614
    },
    {
      "epoch": 6.603240966796875e-06,
      "model_forward_time": 3.144016981124878,
      "step": 1731
    },
    {
      "epoch": 6.603240966796875e-06,
      "step": 1731,
      "training_step_time": 10.312655210494995
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 3.1911416053771973,
      "step": 1732
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 1732,
      "training_step_time": 9.963722467422485
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 2.441718101501465,
      "step": 1732
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 1732,
      "training_step_time": 7.470287084579468
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 3.145897150039673,
      "step": 1732
    },
    {
      "dynamics_loss_avg": 0.08113322518765927,
      "epoch": 6.6070556640625e-06,
      "step": 1732
    },
    {
      "action_loss_avg": 0.04896125821396709,
      "epoch": 6.6070556640625e-06,
      "step": 1732
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 1732,
      "training_step_time": 9.920845031738281
    },
    {
      "epoch": 6.6070556640625e-06,
      "model_forward_time": 3.145336866378784,
      "step": 1732
    },
    {
      "epoch": 6.6070556640625e-06,
      "step": 1732,
      "training_step_time": 10.282611846923828
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 2.4801876544952393,
      "step": 1733
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 1733,
      "training_step_time": 7.509921312332153
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 3.403891086578369,
      "step": 1733
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 1733,
      "training_step_time": 10.17380428314209
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 3.146700620651245,
      "step": 1733
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 1733,
      "training_step_time": 9.919538259506226
    },
    {
      "epoch": 6.610870361328125e-06,
      "model_forward_time": 3.1438887119293213,
      "step": 1733
    },
    {
      "epoch": 6.610870361328125e-06,
      "step": 1733,
      "training_step_time": 10.336404085159302
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 3.1878914833068848,
      "step": 1734
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 1734,
      "training_step_time": 9.964391469955444
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 3.145749568939209,
      "step": 1734
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 1734,
      "training_step_time": 9.92829418182373
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 3.4214670658111572,
      "step": 1734
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 1734,
      "training_step_time": 10.191824674606323
    },
    {
      "epoch": 6.61468505859375e-06,
      "model_forward_time": 3.1418073177337646,
      "step": 1734
    },
    {
      "epoch": 6.61468505859375e-06,
      "step": 1734,
      "training_step_time": 10.301676750183105
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 3.1757395267486572,
      "step": 1735
    },
    {
      "dynamics_loss_avg": 0.07225556336343289,
      "epoch": 6.618499755859375e-06,
      "step": 1735
    },
    {
      "action_loss_avg": 0.04711144152097404,
      "epoch": 6.618499755859375e-06,
      "step": 1735
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 1735,
      "training_step_time": 9.955323934555054
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 3.1435678005218506,
      "step": 1735
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 1735,
      "training_step_time": 9.912466526031494
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 2.4396188259124756,
      "step": 1735
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 1735,
      "training_step_time": 7.468179702758789
    },
    {
      "epoch": 6.618499755859375e-06,
      "model_forward_time": 3.423985719680786,
      "step": 1735
    },
    {
      "epoch": 6.618499755859375e-06,
      "step": 1735,
      "training_step_time": 10.286492109298706
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 3.1839823722839355,
      "step": 1736
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 1736,
      "training_step_time": 9.951951503753662
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 3.1473960876464844,
      "step": 1736
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 1736,
      "training_step_time": 9.920564413070679
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 3.144375801086426,
      "step": 1736
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 1736,
      "training_step_time": 9.918071746826172
    },
    {
      "epoch": 6.622314453125e-06,
      "model_forward_time": 3.14487624168396,
      "step": 1736
    },
    {
      "epoch": 6.622314453125e-06,
      "step": 1736,
      "training_step_time": 10.014961004257202
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 3.4823176860809326,
      "step": 1737
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 1737,
      "training_step_time": 10.259263515472412
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 3.1484575271606445,
      "step": 1737
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 1737,
      "training_step_time": 9.92055606842041
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 3.140023708343506,
      "step": 1737
    },
    {
      "dynamics_loss_avg": 0.0775640368461609,
      "epoch": 6.626129150390625e-06,
      "step": 1737
    },
    {
      "action_loss_avg": 0.04698569569736719,
      "epoch": 6.626129150390625e-06,
      "step": 1737
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 1737,
      "training_step_time": 9.914680004119873
    },
    {
      "epoch": 6.626129150390625e-06,
      "model_forward_time": 3.146047353744507,
      "step": 1737
    },
    {
      "epoch": 6.626129150390625e-06,
      "step": 1737,
      "training_step_time": 10.235561609268188
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 3.448151111602783,
      "step": 1738
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 1738,
      "training_step_time": 10.220289468765259
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 3.144850015640259,
      "step": 1738
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 1738,
      "training_step_time": 9.920504808425903
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 2.4437835216522217,
      "step": 1738
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 1738,
      "training_step_time": 7.475458145141602
    },
    {
      "epoch": 6.62994384765625e-06,
      "model_forward_time": 2.4381227493286133,
      "step": 1738
    },
    {
      "epoch": 6.62994384765625e-06,
      "step": 1738,
      "training_step_time": 10.284815788269043
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 3.1936099529266357,
      "step": 1739
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 1739,
      "training_step_time": 9.973058700561523
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 3.142960786819458,
      "step": 1739
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 1739,
      "training_step_time": 9.92330551147461
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 3.4106006622314453,
      "step": 1739
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 1739,
      "training_step_time": 10.190688610076904
    },
    {
      "epoch": 6.633758544921875e-06,
      "model_forward_time": 3.1487176418304443,
      "step": 1739
    },
    {
      "epoch": 6.633758544921875e-06,
      "step": 1739,
      "training_step_time": 10.273682594299316
    },
    {
      "epoch": 6.6375732421875e-06,
      "grad_norm": 0.6301508545875549,
      "learning_rate": 1e-05,
      "loss": 0.1131,
      "step": 1740
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 3.200866222381592,
      "step": 1740
    },
    {
      "dynamics_loss_avg": 0.07374203763902187,
      "epoch": 6.6375732421875e-06,
      "step": 1740
    },
    {
      "action_loss_avg": 0.05276254266500473,
      "epoch": 6.6375732421875e-06,
      "step": 1740
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 1740,
      "training_step_time": 9.985384702682495
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 2.435581922531128,
      "step": 1740
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 1740,
      "training_step_time": 7.468925476074219
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 3.141566514968872,
      "step": 1740
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 1740,
      "training_step_time": 10.226480722427368
    },
    {
      "epoch": 6.6375732421875e-06,
      "model_forward_time": 3.1469619274139404,
      "step": 1740
    },
    {
      "epoch": 6.6375732421875e-06,
      "step": 1740,
      "training_step_time": 10.292074918746948
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 2.461751937866211,
      "step": 1741
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 1741,
      "training_step_time": 7.493671178817749
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 3.138127326965332,
      "step": 1741
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 1741,
      "training_step_time": 9.919279098510742
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 3.1403980255126953,
      "step": 1741
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 1741,
      "training_step_time": 9.914274215698242
    },
    {
      "epoch": 6.641387939453125e-06,
      "model_forward_time": 3.1383211612701416,
      "step": 1741
    },
    {
      "epoch": 6.641387939453125e-06,
      "step": 1741,
      "training_step_time": 10.311533451080322
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 3.45005202293396,
      "step": 1742
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 1742,
      "training_step_time": 10.2259361743927
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 2.433600664138794,
      "step": 1742
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 1742,
      "training_step_time": 7.462414026260376
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 2.435622215270996,
      "step": 1742
    },
    {
      "dynamics_loss_avg": 0.06407178957015276,
      "epoch": 6.64520263671875e-06,
      "step": 1742
    },
    {
      "action_loss_avg": 0.05001237941905856,
      "epoch": 6.64520263671875e-06,
      "step": 1742
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 1742,
      "training_step_time": 7.472513198852539
    },
    {
      "epoch": 6.64520263671875e-06,
      "model_forward_time": 3.1385538578033447,
      "step": 1742
    },
    {
      "epoch": 6.64520263671875e-06,
      "step": 1742,
      "training_step_time": 10.035913228988647
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 3.167837142944336,
      "step": 1743
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 1743,
      "training_step_time": 9.944940328598022
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 3.3908066749572754,
      "step": 1743
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 1743,
      "training_step_time": 10.159035682678223
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 3.1402463912963867,
      "step": 1743
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 1743,
      "training_step_time": 9.910273790359497
    },
    {
      "epoch": 6.649017333984375e-06,
      "model_forward_time": 3.1366922855377197,
      "step": 1743
    },
    {
      "epoch": 6.649017333984375e-06,
      "step": 1743,
      "training_step_time": 10.342190265655518
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 3.187488317489624,
      "step": 1744
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 1744,
      "training_step_time": 9.963120698928833
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 2.435187816619873,
      "step": 1744
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 1744,
      "training_step_time": 7.46501612663269
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 3.4513375759124756,
      "step": 1744
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 1744,
      "training_step_time": 10.225302696228027
    },
    {
      "epoch": 6.65283203125e-06,
      "model_forward_time": 3.137516498565674,
      "step": 1744
    },
    {
      "epoch": 6.65283203125e-06,
      "step": 1744,
      "training_step_time": 10.32828688621521
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 3.1669647693634033,
      "step": 1745
    },
    {
      "dynamics_loss_avg": 0.06334243416786194,
      "epoch": 6.656646728515625e-06,
      "step": 1745
    },
    {
      "action_loss_avg": 0.0412890917621553,
      "epoch": 6.656646728515625e-06,
      "step": 1745
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 1745,
      "training_step_time": 9.94593596458435
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 3.1390881538391113,
      "step": 1745
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 1745,
      "training_step_time": 9.91532039642334
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 3.138610601425171,
      "step": 1745
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 1745,
      "training_step_time": 9.916959524154663
    },
    {
      "epoch": 6.656646728515625e-06,
      "model_forward_time": 3.3863656520843506,
      "step": 1745
    },
    {
      "epoch": 6.656646728515625e-06,
      "step": 1745,
      "training_step_time": 10.256332874298096
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 2.451500654220581,
      "step": 1746
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 1746,
      "training_step_time": 7.4853222370147705
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 3.137864112854004,
      "step": 1746
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 1746,
      "training_step_time": 9.9130220413208
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 2.4379372596740723,
      "step": 1746
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 1746,
      "training_step_time": 7.472987651824951
    },
    {
      "epoch": 6.66046142578125e-06,
      "model_forward_time": 3.143878936767578,
      "step": 1746
    },
    {
      "epoch": 6.66046142578125e-06,
      "step": 1746,
      "training_step_time": 10.317710399627686
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 3.4529762268066406,
      "step": 1747
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 1747,
      "training_step_time": 10.236750602722168
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 3.150989055633545,
      "step": 1747
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 1747,
      "training_step_time": 9.937257289886475
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 3.1495723724365234,
      "step": 1747
    },
    {
      "dynamics_loss_avg": 0.06922838650643826,
      "epoch": 6.664276123046875e-06,
      "step": 1747
    },
    {
      "action_loss_avg": 0.05136006008833647,
      "epoch": 6.664276123046875e-06,
      "step": 1747
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 1747,
      "training_step_time": 9.935368299484253
    },
    {
      "epoch": 6.664276123046875e-06,
      "model_forward_time": 3.149465322494507,
      "step": 1747
    },
    {
      "epoch": 6.664276123046875e-06,
      "step": 1747,
      "training_step_time": 10.03527569770813
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 2.470383644104004,
      "step": 1748
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 1748,
      "training_step_time": 7.505374431610107
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 3.4597721099853516,
      "step": 1748
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 1748,
      "training_step_time": 10.24073338508606
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 3.1425161361694336,
      "step": 1748
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 1748,
      "training_step_time": 9.920154333114624
    },
    {
      "epoch": 6.6680908203125e-06,
      "model_forward_time": 3.147278308868408,
      "step": 1748
    },
    {
      "epoch": 6.6680908203125e-06,
      "step": 1748,
      "training_step_time": 10.272014617919922
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 3.2308948040008545,
      "step": 1749
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 1749,
      "training_step_time": 10.005484580993652
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 3.1432647705078125,
      "step": 1749
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 1749,
      "training_step_time": 9.921490907669067
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 3.3969249725341797,
      "step": 1749
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 1749,
      "training_step_time": 10.171881437301636
    },
    {
      "epoch": 6.671905517578125e-06,
      "model_forward_time": 3.149527072906494,
      "step": 1749
    },
    {
      "epoch": 6.671905517578125e-06,
      "step": 1749,
      "training_step_time": 10.317990064620972
    },
    {
      "epoch": 6.67572021484375e-06,
      "grad_norm": 0.5776817798614502,
      "learning_rate": 1e-05,
      "loss": 0.1106,
      "step": 1750
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 3.1901726722717285,
      "step": 1750
    },
    {
      "dynamics_loss_avg": 0.06522263269871473,
      "epoch": 6.67572021484375e-06,
      "step": 1750
    },
    {
      "action_loss_avg": 0.03995293090119958,
      "epoch": 6.67572021484375e-06,
      "step": 1750
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 1750,
      "training_step_time": 9.970894575119019
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 3.1442508697509766,
      "step": 1750
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 1750,
      "training_step_time": 9.920467853546143
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 3.1436376571655273,
      "step": 1750
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 1750,
      "training_step_time": 9.926554203033447
    },
    {
      "epoch": 6.67572021484375e-06,
      "model_forward_time": 3.4188923835754395,
      "step": 1750
    },
    {
      "epoch": 6.67572021484375e-06,
      "step": 1750,
      "training_step_time": 10.307961463928223
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 3.189084529876709,
      "step": 1751
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 1751,
      "training_step_time": 9.972513437271118
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 3.1410582065582275,
      "step": 1751
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 1751,
      "training_step_time": 9.917463302612305
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 3.1455237865448,
      "step": 1751
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 1751,
      "training_step_time": 9.926095008850098
    },
    {
      "epoch": 6.679534912109375e-06,
      "model_forward_time": 3.1440489292144775,
      "step": 1751
    },
    {
      "epoch": 6.679534912109375e-06,
      "step": 1751,
      "training_step_time": 10.301891088485718
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 3.1918413639068604,
      "step": 1752
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 1752,
      "training_step_time": 9.976329803466797
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 3.1417107582092285,
      "step": 1752
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 1752,
      "training_step_time": 9.919706583023071
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 3.141803026199341,
      "step": 1752
    },
    {
      "dynamics_loss_avg": 0.05808299742639065,
      "epoch": 6.683349609375e-06,
      "step": 1752
    },
    {
      "action_loss_avg": 0.041382890287786724,
      "epoch": 6.683349609375e-06,
      "step": 1752
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 1752,
      "training_step_time": 9.929060459136963
    },
    {
      "epoch": 6.683349609375e-06,
      "model_forward_time": 3.140571355819702,
      "step": 1752
    },
    {
      "epoch": 6.683349609375e-06,
      "step": 1752,
      "training_step_time": 10.026849746704102
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 3.1685473918914795,
      "step": 1753
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 1753,
      "training_step_time": 10.274598121643066
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 2.427699089050293,
      "step": 1753
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 1753,
      "training_step_time": 7.450934886932373
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 3.1276190280914307,
      "step": 1753
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 1753,
      "training_step_time": 9.893028736114502
    },
    {
      "epoch": 6.687164306640625e-06,
      "model_forward_time": 3.127528429031372,
      "step": 1753
    },
    {
      "epoch": 6.687164306640625e-06,
      "step": 1753,
      "training_step_time": 10.300625562667847
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 2.4681894779205322,
      "step": 1754
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 1754,
      "training_step_time": 7.495049953460693
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 3.128873348236084,
      "step": 1754
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 1754,
      "training_step_time": 9.896285772323608
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 3.3985755443573,
      "step": 1754
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 1754,
      "training_step_time": 10.161556243896484
    },
    {
      "epoch": 6.69097900390625e-06,
      "model_forward_time": 3.128718852996826,
      "step": 1754
    },
    {
      "epoch": 6.69097900390625e-06,
      "step": 1754,
      "training_step_time": 10.336794137954712
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 3.167379856109619,
      "step": 1755
    },
    {
      "dynamics_loss_avg": 0.06811009086668492,
      "epoch": 6.694793701171875e-06,
      "step": 1755
    },
    {
      "action_loss_avg": 0.03836233224719763,
      "epoch": 6.694793701171875e-06,
      "step": 1755
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 1755,
      "training_step_time": 9.93877625465393
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 3.130514144897461,
      "step": 1755
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 1755,
      "training_step_time": 9.899779319763184
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 3.4049150943756104,
      "step": 1755
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 1755,
      "training_step_time": 10.174111604690552
    },
    {
      "epoch": 6.694793701171875e-06,
      "model_forward_time": 3.126150608062744,
      "step": 1755
    },
    {
      "epoch": 6.694793701171875e-06,
      "step": 1755,
      "training_step_time": 10.289743423461914
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 3.1769747734069824,
      "step": 1756
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 1756,
      "training_step_time": 9.948905229568481
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 2.431150436401367,
      "step": 1756
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 1756,
      "training_step_time": 7.46272873878479
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 3.132133722305298,
      "step": 1756
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 1756,
      "training_step_time": 9.906759977340698
    },
    {
      "epoch": 6.6986083984375e-06,
      "model_forward_time": 3.140305995941162,
      "step": 1756
    },
    {
      "epoch": 6.6986083984375e-06,
      "step": 1756,
      "training_step_time": 10.04336953163147
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 3.4964370727539062,
      "step": 1757
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 1757,
      "training_step_time": 10.276015996932983
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 2.4338090419769287,
      "step": 1757
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 1757,
      "training_step_time": 7.46891713142395
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 3.1349759101867676,
      "step": 1757
    },
    {
      "dynamics_loss_avg": 0.07122360840439797,
      "epoch": 6.702423095703125e-06,
      "step": 1757
    },
    {
      "action_loss_avg": 0.04800520204007626,
      "epoch": 6.702423095703125e-06,
      "step": 1757
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 1757,
      "training_step_time": 9.918035984039307
    },
    {
      "epoch": 6.702423095703125e-06,
      "model_forward_time": 3.1403536796569824,
      "step": 1757
    },
    {
      "epoch": 6.702423095703125e-06,
      "step": 1757,
      "training_step_time": 10.043198108673096
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 3.400559663772583,
      "step": 1758
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 1758,
      "training_step_time": 10.184558391571045
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 3.1387574672698975,
      "step": 1758
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 1758,
      "training_step_time": 9.915786266326904
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 3.1366803646087646,
      "step": 1758
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 1758,
      "training_step_time": 9.915964603424072
    },
    {
      "epoch": 6.70623779296875e-06,
      "model_forward_time": 2.4305567741394043,
      "step": 1758
    },
    {
      "epoch": 6.70623779296875e-06,
      "step": 1758,
      "training_step_time": 10.263278007507324
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 3.1666676998138428,
      "step": 1759
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 1759,
      "training_step_time": 9.94091010093689
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 3.130263566970825,
      "step": 1759
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 1759,
      "training_step_time": 9.900797605514526
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 3.391615152359009,
      "step": 1759
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 1759,
      "training_step_time": 10.163157939910889
    },
    {
      "epoch": 6.710052490234375e-06,
      "model_forward_time": 3.1324493885040283,
      "step": 1759
    },
    {
      "epoch": 6.710052490234375e-06,
      "step": 1759,
      "training_step_time": 10.26415205001831
    },
    {
      "epoch": 6.7138671875e-06,
      "grad_norm": 0.676109254360199,
      "learning_rate": 1e-05,
      "loss": 0.1162,
      "step": 1760
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 3.174636125564575,
      "step": 1760
    },
    {
      "dynamics_loss_avg": 0.07459324635565281,
      "epoch": 6.7138671875e-06,
      "step": 1760
    },
    {
      "action_loss_avg": 0.054609532654285434,
      "epoch": 6.7138671875e-06,
      "step": 1760
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 1760,
      "training_step_time": 9.951780080795288
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 3.130754232406616,
      "step": 1760
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 1760,
      "training_step_time": 9.900783061981201
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 3.411651134490967,
      "step": 1760
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 1760,
      "training_step_time": 10.180906057357788
    },
    {
      "epoch": 6.7138671875e-06,
      "model_forward_time": 3.1335325241088867,
      "step": 1760
    },
    {
      "epoch": 6.7138671875e-06,
      "step": 1760,
      "training_step_time": 10.302784442901611
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 3.1623499393463135,
      "step": 1761
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 1761,
      "training_step_time": 9.931760787963867
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 3.129080057144165,
      "step": 1761
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 1761,
      "training_step_time": 9.896007537841797
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 3.1345794200897217,
      "step": 1761
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 1761,
      "training_step_time": 9.907250881195068
    },
    {
      "epoch": 6.717681884765625e-06,
      "model_forward_time": 3.132089614868164,
      "step": 1761
    },
    {
      "epoch": 6.717681884765625e-06,
      "step": 1761,
      "training_step_time": 10.300787925720215
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 2.4446306228637695,
      "step": 1762
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 1762,
      "training_step_time": 7.477163791656494
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 3.1266837120056152,
      "step": 1762
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 1762,
      "training_step_time": 9.896347761154175
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 3.130786418914795,
      "step": 1762
    },
    {
      "dynamics_loss_avg": 0.058927221596240996,
      "epoch": 6.72149658203125e-06,
      "step": 1762
    },
    {
      "action_loss_avg": 0.030848196474835277,
      "epoch": 6.72149658203125e-06,
      "step": 1762
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 1762,
      "training_step_time": 9.901885271072388
    },
    {
      "epoch": 6.72149658203125e-06,
      "model_forward_time": 3.1276192665100098,
      "step": 1762
    },
    {
      "epoch": 6.72149658203125e-06,
      "step": 1762,
      "training_step_time": 10.014122247695923
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 3.1375200748443604,
      "step": 1763
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 1763,
      "training_step_time": 10.23363971710205
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 3.1267096996307373,
      "step": 1763
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 1763,
      "training_step_time": 9.891567468643188
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 3.12727427482605,
      "step": 1763
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 1763,
      "training_step_time": 9.896251916885376
    },
    {
      "epoch": 6.725311279296875e-06,
      "model_forward_time": 2.4280388355255127,
      "step": 1763
    },
    {
      "epoch": 6.725311279296875e-06,
      "step": 1763,
      "training_step_time": 10.259955406188965
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 3.159815549850464,
      "step": 1764
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 1764,
      "training_step_time": 9.925613641738892
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 3.130035877227783,
      "step": 1764
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 1764,
      "training_step_time": 9.898349046707153
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 3.4061295986175537,
      "step": 1764
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 1764,
      "training_step_time": 10.173967599868774
    },
    {
      "epoch": 6.7291259765625e-06,
      "model_forward_time": 3.133760452270508,
      "step": 1764
    },
    {
      "epoch": 6.7291259765625e-06,
      "step": 1764,
      "training_step_time": 10.318260669708252
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 3.1623754501342773,
      "step": 1765
    },
    {
      "dynamics_loss_avg": 0.0720592625439167,
      "epoch": 6.732940673828125e-06,
      "step": 1765
    },
    {
      "action_loss_avg": 0.040015354938805106,
      "epoch": 6.732940673828125e-06,
      "step": 1765
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 1765,
      "training_step_time": 9.941344976425171
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 3.135084629058838,
      "step": 1765
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 1765,
      "training_step_time": 9.908590078353882
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 3.1296308040618896,
      "step": 1765
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 1765,
      "training_step_time": 10.168293476104736
    },
    {
      "epoch": 6.732940673828125e-06,
      "model_forward_time": 3.1313040256500244,
      "step": 1765
    },
    {
      "epoch": 6.732940673828125e-06,
      "step": 1765,
      "training_step_time": 10.295811414718628
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 3.176114559173584,
      "step": 1766
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 1766,
      "training_step_time": 9.946498394012451
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 3.1296186447143555,
      "step": 1766
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 1766,
      "training_step_time": 9.898934364318848
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 3.1260459423065186,
      "step": 1766
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 1766,
      "training_step_time": 9.901301860809326
    },
    {
      "epoch": 6.73675537109375e-06,
      "model_forward_time": 3.379659652709961,
      "step": 1766
    },
    {
      "epoch": 6.73675537109375e-06,
      "step": 1766,
      "training_step_time": 10.270657062530518
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 3.1603569984436035,
      "step": 1767
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 1767,
      "training_step_time": 9.93096399307251
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 3.132122278213501,
      "step": 1767
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 1767,
      "training_step_time": 9.899601936340332
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 3.126509666442871,
      "step": 1767
    },
    {
      "dynamics_loss_avg": 0.06299771312624217,
      "epoch": 6.740570068359375e-06,
      "step": 1767
    },
    {
      "action_loss_avg": 0.04940431360155344,
      "epoch": 6.740570068359375e-06,
      "step": 1767
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 1767,
      "training_step_time": 9.893741607666016
    },
    {
      "epoch": 6.740570068359375e-06,
      "model_forward_time": 3.1278607845306396,
      "step": 1767
    },
    {
      "epoch": 6.740570068359375e-06,
      "step": 1767,
      "training_step_time": 10.01445484161377
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 1.7556450366973877,
      "step": 1768
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 1768,
      "training_step_time": 5.208393096923828
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 3.4247138500213623,
      "step": 1768
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 1768,
      "training_step_time": 10.181886911392212
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 2.4285497665405273,
      "step": 1768
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 1768,
      "training_step_time": 7.4481589794158936
    },
    {
      "epoch": 6.744384765625e-06,
      "model_forward_time": 3.1287076473236084,
      "step": 1768
    },
    {
      "epoch": 6.744384765625e-06,
      "step": 1768,
      "training_step_time": 10.02772307395935
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 3.169119358062744,
      "step": 1769
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 1769,
      "training_step_time": 9.938236951828003
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 3.142569065093994,
      "step": 1769
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 1769,
      "training_step_time": 10.179996967315674
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 3.1311123371124268,
      "step": 1769
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 1769,
      "training_step_time": 9.900847434997559
    },
    {
      "epoch": 6.748199462890625e-06,
      "model_forward_time": 3.137097120285034,
      "step": 1769
    },
    {
      "epoch": 6.748199462890625e-06,
      "step": 1769,
      "training_step_time": 10.30804181098938
    },
    {
      "epoch": 6.75201416015625e-06,
      "grad_norm": 0.7480739951133728,
      "learning_rate": 1e-05,
      "loss": 0.1094,
      "step": 1770
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 3.1755106449127197,
      "step": 1770
    },
    {
      "dynamics_loss_avg": 0.05749672632664442,
      "epoch": 6.75201416015625e-06,
      "step": 1770
    },
    {
      "action_loss_avg": 0.040196698345243934,
      "epoch": 6.75201416015625e-06,
      "step": 1770
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 1770,
      "training_step_time": 9.94757604598999
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 3.129704475402832,
      "step": 1770
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 1770,
      "training_step_time": 9.889086961746216
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 3.3845183849334717,
      "step": 1770
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 1770,
      "training_step_time": 10.14388918876648
    },
    {
      "epoch": 6.75201416015625e-06,
      "model_forward_time": 3.128282308578491,
      "step": 1770
    },
    {
      "epoch": 6.75201416015625e-06,
      "step": 1770,
      "training_step_time": 10.311160564422607
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 3.1833655834198,
      "step": 1771
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 1771,
      "training_step_time": 9.94432020187378
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 3.128161907196045,
      "step": 1771
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 1771,
      "training_step_time": 9.889497756958008
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 3.1304802894592285,
      "step": 1771
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 1771,
      "training_step_time": 9.892374277114868
    },
    {
      "epoch": 6.755828857421875e-06,
      "model_forward_time": 3.130121946334839,
      "step": 1771
    },
    {
      "epoch": 6.755828857421875e-06,
      "step": 1771,
      "training_step_time": 10.252819061279297
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 3.140676498413086,
      "step": 1772
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 1772,
      "training_step_time": 9.901476383209229
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 3.127622365951538,
      "step": 1772
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 1772,
      "training_step_time": 9.886714696884155
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 2.4281342029571533,
      "step": 1772
    },
    {
      "dynamics_loss_avg": 0.05421132557094097,
      "epoch": 6.7596435546875e-06,
      "step": 1772
    },
    {
      "action_loss_avg": 0.03172779530286789,
      "epoch": 6.7596435546875e-06,
      "step": 1772
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 1772,
      "training_step_time": 7.452022314071655
    },
    {
      "epoch": 6.7596435546875e-06,
      "model_forward_time": 2.4284658432006836,
      "step": 1772
    },
    {
      "epoch": 6.7596435546875e-06,
      "step": 1772,
      "training_step_time": 10.284083366394043
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 3.159618377685547,
      "step": 1773
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 1773,
      "training_step_time": 9.924588680267334
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 3.4051363468170166,
      "step": 1773
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 1773,
      "training_step_time": 10.171973466873169
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 3.136992931365967,
      "step": 1773
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 1773,
      "training_step_time": 9.90603756904602
    },
    {
      "epoch": 6.763458251953125e-06,
      "model_forward_time": 3.131352663040161,
      "step": 1773
    },
    {
      "epoch": 6.763458251953125e-06,
      "step": 1773,
      "training_step_time": 10.030393600463867
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 3.1625163555145264,
      "step": 1774
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 1774,
      "training_step_time": 9.936684131622314
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 3.1363489627838135,
      "step": 1774
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 1774,
      "training_step_time": 9.898430347442627
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 3.418318748474121,
      "step": 1774
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 1774,
      "training_step_time": 10.185381650924683
    },
    {
      "epoch": 6.76727294921875e-06,
      "model_forward_time": 3.130772590637207,
      "step": 1774
    },
    {
      "epoch": 6.76727294921875e-06,
      "step": 1774,
      "training_step_time": 10.301854610443115
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 3.169149398803711,
      "step": 1775
    },
    {
      "dynamics_loss_avg": 0.05965270325541496,
      "epoch": 6.771087646484375e-06,
      "step": 1775
    },
    {
      "action_loss_avg": 0.04070360362529755,
      "epoch": 6.771087646484375e-06,
      "step": 1775
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 1775,
      "training_step_time": 9.935216903686523
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 3.1344077587127686,
      "step": 1775
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 1775,
      "training_step_time": 9.898974895477295
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 3.4098589420318604,
      "step": 1775
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 1775,
      "training_step_time": 10.172996997833252
    },
    {
      "epoch": 6.771087646484375e-06,
      "model_forward_time": 3.1310300827026367,
      "step": 1775
    },
    {
      "epoch": 6.771087646484375e-06,
      "step": 1775,
      "training_step_time": 10.323983192443848
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 2.466221809387207,
      "step": 1776
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 1776,
      "training_step_time": 7.48734450340271
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 3.1309006214141846,
      "step": 1776
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 1776,
      "training_step_time": 9.900258541107178
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 3.133765697479248,
      "step": 1776
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 1776,
      "training_step_time": 9.903648138046265
    },
    {
      "epoch": 6.77490234375e-06,
      "model_forward_time": 3.135974168777466,
      "step": 1776
    },
    {
      "epoch": 6.77490234375e-06,
      "step": 1776,
      "training_step_time": 10.293029546737671
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 3.4554569721221924,
      "step": 1777
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 1777,
      "training_step_time": 10.219357252120972
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 3.1319830417633057,
      "step": 1777
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 1777,
      "training_step_time": 9.900364875793457
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 3.1326088905334473,
      "step": 1777
    },
    {
      "dynamics_loss_avg": 0.06874790098518133,
      "epoch": 6.778717041015625e-06,
      "step": 1777
    },
    {
      "action_loss_avg": 0.04901415025815368,
      "epoch": 6.778717041015625e-06,
      "step": 1777
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 1777,
      "training_step_time": 9.902719497680664
    },
    {
      "epoch": 6.778717041015625e-06,
      "model_forward_time": 3.1347031593322754,
      "step": 1777
    },
    {
      "epoch": 6.778717041015625e-06,
      "step": 1777,
      "training_step_time": 10.316934823989868
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 3.4612908363342285,
      "step": 1778
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 1778,
      "training_step_time": 10.228891849517822
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 3.1390223503112793,
      "step": 1778
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 1778,
      "training_step_time": 9.908423662185669
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 3.134510040283203,
      "step": 1778
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 1778,
      "training_step_time": 9.909017086029053
    },
    {
      "epoch": 6.78253173828125e-06,
      "model_forward_time": 3.1426122188568115,
      "step": 1778
    },
    {
      "epoch": 6.78253173828125e-06,
      "step": 1778,
      "training_step_time": 10.043076276779175
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 2.4814555644989014,
      "step": 1779
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 1779,
      "training_step_time": 7.51004695892334
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 3.130981683731079,
      "step": 1779
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 1779,
      "training_step_time": 9.899514198303223
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 3.415825843811035,
      "step": 1779
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 1779,
      "training_step_time": 10.184597492218018
    },
    {
      "epoch": 6.786346435546875e-06,
      "model_forward_time": 3.136580228805542,
      "step": 1779
    },
    {
      "epoch": 6.786346435546875e-06,
      "step": 1779,
      "training_step_time": 10.020524740219116
    },
    {
      "epoch": 6.7901611328125e-06,
      "grad_norm": 0.6476795673370361,
      "learning_rate": 1e-05,
      "loss": 0.114,
      "step": 1780
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 3.1517980098724365,
      "step": 1780
    },
    {
      "dynamics_loss_avg": 0.06304388754069805,
      "epoch": 6.7901611328125e-06,
      "step": 1780
    },
    {
      "action_loss_avg": 0.04068034999072552,
      "epoch": 6.7901611328125e-06,
      "step": 1780
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 1780,
      "training_step_time": 9.921928644180298
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 3.1354634761810303,
      "step": 1780
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 1780,
      "training_step_time": 9.899726152420044
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 2.4355533123016357,
      "step": 1780
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 1780,
      "training_step_time": 7.465955018997192
    },
    {
      "epoch": 6.7901611328125e-06,
      "model_forward_time": 3.4046075344085693,
      "step": 1780
    },
    {
      "epoch": 6.7901611328125e-06,
      "step": 1780,
      "training_step_time": 10.27966570854187
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 3.176967144012451,
      "step": 1781
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 1781,
      "training_step_time": 9.936157703399658
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 1.7436723709106445,
      "step": 1781
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 1781,
      "training_step_time": 5.192217826843262
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 3.1299118995666504,
      "step": 1781
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 1781,
      "training_step_time": 9.886301040649414
    },
    {
      "epoch": 6.793975830078125e-06,
      "model_forward_time": 2.4327855110168457,
      "step": 1781
    },
    {
      "epoch": 6.793975830078125e-06,
      "step": 1781,
      "training_step_time": 10.019447326660156
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 2.772759437561035,
      "step": 1782
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 1782,
      "training_step_time": 7.790063142776489
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 2.4308011531829834,
      "step": 1782
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 1782,
      "training_step_time": 7.44433331489563
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 3.1316118240356445,
      "step": 1782
    },
    {
      "dynamics_loss_avg": 0.061679696664214134,
      "epoch": 6.79779052734375e-06,
      "step": 1782
    },
    {
      "action_loss_avg": 0.03682389780879021,
      "epoch": 6.79779052734375e-06,
      "step": 1782
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 1782,
      "training_step_time": 9.89064335823059
    },
    {
      "epoch": 6.79779052734375e-06,
      "model_forward_time": 3.1325876712799072,
      "step": 1782
    },
    {
      "epoch": 6.79779052734375e-06,
      "step": 1782,
      "training_step_time": 10.011140823364258
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 3.1561334133148193,
      "step": 1783
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 1783,
      "training_step_time": 9.917750120162964
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 3.3863956928253174,
      "step": 1783
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 1783,
      "training_step_time": 10.149313688278198
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 3.137298345565796,
      "step": 1783
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 1783,
      "training_step_time": 9.901799440383911
    },
    {
      "epoch": 6.801605224609375e-06,
      "model_forward_time": 2.4337570667266846,
      "step": 1783
    },
    {
      "epoch": 6.801605224609375e-06,
      "step": 1783,
      "training_step_time": 10.297707319259644
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 3.1675314903259277,
      "step": 1784
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 1784,
      "training_step_time": 9.935900926589966
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 3.136986255645752,
      "step": 1784
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 1784,
      "training_step_time": 9.899223566055298
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 3.4426071643829346,
      "step": 1784
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 1784,
      "training_step_time": 10.202394008636475
    },
    {
      "epoch": 6.805419921875e-06,
      "model_forward_time": 3.136122226715088,
      "step": 1784
    },
    {
      "epoch": 6.805419921875e-06,
      "step": 1784,
      "training_step_time": 10.287906408309937
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 3.1776328086853027,
      "step": 1785
    },
    {
      "dynamics_loss_avg": 0.07620378062129021,
      "epoch": 6.809234619140625e-06,
      "step": 1785
    },
    {
      "action_loss_avg": 0.05812859740108252,
      "epoch": 6.809234619140625e-06,
      "step": 1785
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 1785,
      "training_step_time": 9.942590951919556
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 2.431260824203491,
      "step": 1785
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 1785,
      "training_step_time": 7.448714017868042
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 3.1287894248962402,
      "step": 1785
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 1785,
      "training_step_time": 9.891103029251099
    },
    {
      "epoch": 6.809234619140625e-06,
      "model_forward_time": 3.3943440914154053,
      "step": 1785
    },
    {
      "epoch": 6.809234619140625e-06,
      "step": 1785,
      "training_step_time": 10.29783821105957
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 3.1673803329467773,
      "step": 1786
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 1786,
      "training_step_time": 9.940575122833252
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 3.130552291870117,
      "step": 1786
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 1786,
      "training_step_time": 9.896681308746338
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 3.1298680305480957,
      "step": 1786
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 1786,
      "training_step_time": 9.898512601852417
    },
    {
      "epoch": 6.81304931640625e-06,
      "model_forward_time": 3.131671905517578,
      "step": 1786
    },
    {
      "epoch": 6.81304931640625e-06,
      "step": 1786,
      "training_step_time": 10.277171850204468
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 3.421359062194824,
      "step": 1787
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 1787,
      "training_step_time": 10.188086032867432
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 3.1341514587402344,
      "step": 1787
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 1787,
      "training_step_time": 9.901055335998535
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 3.1333200931549072,
      "step": 1787
    },
    {
      "dynamics_loss_avg": 0.049029132537543776,
      "epoch": 6.816864013671875e-06,
      "step": 1787
    },
    {
      "action_loss_avg": 0.030460081808269025,
      "epoch": 6.816864013671875e-06,
      "step": 1787
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 1787,
      "training_step_time": 9.9097421169281
    },
    {
      "epoch": 6.816864013671875e-06,
      "model_forward_time": 2.4296329021453857,
      "step": 1787
    },
    {
      "epoch": 6.816864013671875e-06,
      "step": 1787,
      "training_step_time": 10.023333311080933
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 3.172846555709839,
      "step": 1788
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 1788,
      "training_step_time": 9.942205905914307
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 3.4444010257720947,
      "step": 1788
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 1788,
      "training_step_time": 10.215876817703247
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 3.1353018283843994,
      "step": 1788
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 1788,
      "training_step_time": 9.910097599029541
    },
    {
      "epoch": 6.8206787109375e-06,
      "model_forward_time": 3.142871856689453,
      "step": 1788
    },
    {
      "epoch": 6.8206787109375e-06,
      "step": 1788,
      "training_step_time": 10.286004781723022
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 3.1796295642852783,
      "step": 1789
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 1789,
      "training_step_time": 9.953718185424805
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 2.437065839767456,
      "step": 1789
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 1789,
      "training_step_time": 7.463732004165649
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 3.3989834785461426,
      "step": 1789
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 1789,
      "training_step_time": 10.162088632583618
    },
    {
      "epoch": 6.824493408203125e-06,
      "model_forward_time": 3.136629819869995,
      "step": 1789
    },
    {
      "epoch": 6.824493408203125e-06,
      "step": 1789,
      "training_step_time": 10.014125108718872
    },
    {
      "epoch": 6.82830810546875e-06,
      "grad_norm": 0.7828715443611145,
      "learning_rate": 1e-05,
      "loss": 0.1118,
      "step": 1790
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 2.461792469024658,
      "step": 1790
    },
    {
      "dynamics_loss_avg": 0.05143330059945583,
      "epoch": 6.82830810546875e-06,
      "step": 1790
    },
    {
      "action_loss_avg": 0.035889053344726564,
      "epoch": 6.82830810546875e-06,
      "step": 1790
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 1790,
      "training_step_time": 7.491392135620117
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 3.1350221633911133,
      "step": 1790
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 1790,
      "training_step_time": 9.908626317977905
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 3.144585371017456,
      "step": 1790
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 1790,
      "training_step_time": 9.918249607086182
    },
    {
      "epoch": 6.82830810546875e-06,
      "model_forward_time": 3.421743869781494,
      "step": 1790
    },
    {
      "epoch": 6.82830810546875e-06,
      "step": 1790,
      "training_step_time": 10.286343097686768
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 3.1688530445098877,
      "step": 1791
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 1791,
      "training_step_time": 9.946572303771973
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 3.1383676528930664,
      "step": 1791
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 1791,
      "training_step_time": 9.912166357040405
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 3.1400086879730225,
      "step": 1791
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 1791,
      "training_step_time": 9.912951707839966
    },
    {
      "epoch": 6.832122802734375e-06,
      "model_forward_time": 1.7502520084381104,
      "step": 1791
    },
    {
      "epoch": 6.832122802734375e-06,
      "step": 1791,
      "training_step_time": 10.299808979034424
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 3.150531768798828,
      "step": 1792
    },
    {
      "epoch": 6.8359375e-06,
      "step": 1792,
      "training_step_time": 9.919399738311768
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 3.3850152492523193,
      "step": 1792
    },
    {
      "epoch": 6.8359375e-06,
      "step": 1792,
      "training_step_time": 10.150157690048218
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 2.4357850551605225,
      "step": 1792
    },
    {
      "dynamics_loss_avg": 0.062261206656694414,
      "epoch": 6.8359375e-06,
      "step": 1792
    },
    {
      "action_loss_avg": 0.04059122642502189,
      "epoch": 6.8359375e-06,
      "step": 1792
    },
    {
      "epoch": 6.8359375e-06,
      "step": 1792,
      "training_step_time": 7.4614951610565186
    },
    {
      "epoch": 6.8359375e-06,
      "model_forward_time": 3.135291337966919,
      "step": 1792
    },
    {
      "epoch": 6.8359375e-06,
      "step": 1792,
      "training_step_time": 10.018031597137451
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 3.1441709995269775,
      "step": 1793
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 1793,
      "training_step_time": 9.914166450500488
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 2.4371895790100098,
      "step": 1793
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 1793,
      "training_step_time": 7.467296838760376
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 3.454440116882324,
      "step": 1793
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 1793,
      "training_step_time": 10.218810558319092
    },
    {
      "epoch": 6.839752197265625e-06,
      "model_forward_time": 2.437241315841675,
      "step": 1793
    },
    {
      "epoch": 6.839752197265625e-06,
      "step": 1793,
      "training_step_time": 10.2561514377594
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 2.4752755165100098,
      "step": 1794
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 1794,
      "training_step_time": 7.498547315597534
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 2.432790756225586,
      "step": 1794
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 1794,
      "training_step_time": 7.453454494476318
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 3.1362991333007812,
      "step": 1794
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 1794,
      "training_step_time": 9.903918504714966
    },
    {
      "epoch": 6.84356689453125e-06,
      "model_forward_time": 3.3915297985076904,
      "step": 1794
    },
    {
      "epoch": 6.84356689453125e-06,
      "step": 1794,
      "training_step_time": 10.248313188552856
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 2.471750497817993,
      "step": 1795
    },
    {
      "dynamics_loss_avg": 0.05067108469083905,
      "epoch": 6.847381591796875e-06,
      "step": 1795
    },
    {
      "action_loss_avg": 0.03550643839407712,
      "epoch": 6.847381591796875e-06,
      "step": 1795
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 1795,
      "training_step_time": 7.495137691497803
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 3.1336305141448975,
      "step": 1795
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 1795,
      "training_step_time": 9.900144100189209
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 3.1370491981506348,
      "step": 1795
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 1795,
      "training_step_time": 9.901490211486816
    },
    {
      "epoch": 6.847381591796875e-06,
      "model_forward_time": 3.1304171085357666,
      "step": 1795
    },
    {
      "epoch": 6.847381591796875e-06,
      "step": 1795,
      "training_step_time": 10.28938341140747
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 3.430166482925415,
      "step": 1796
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 1796,
      "training_step_time": 10.196279525756836
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 3.139962673187256,
      "step": 1796
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 1796,
      "training_step_time": 9.90067195892334
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 3.1336288452148438,
      "step": 1796
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 1796,
      "training_step_time": 9.894765377044678
    },
    {
      "epoch": 6.8511962890625e-06,
      "model_forward_time": 3.135725259780884,
      "step": 1796
    },
    {
      "epoch": 6.8511962890625e-06,
      "step": 1796,
      "training_step_time": 10.262943029403687
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 3.172152280807495,
      "step": 1797
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 1797,
      "training_step_time": 9.937356948852539
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 3.1349236965179443,
      "step": 1797
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 1797,
      "training_step_time": 10.225692987442017
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 3.1367380619049072,
      "step": 1797
    },
    {
      "dynamics_loss_avg": 0.06794573049992322,
      "epoch": 6.855010986328125e-06,
      "step": 1797
    },
    {
      "action_loss_avg": 0.04007739452645183,
      "epoch": 6.855010986328125e-06,
      "step": 1797
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 1797,
      "training_step_time": 9.905248880386353
    },
    {
      "epoch": 6.855010986328125e-06,
      "model_forward_time": 3.1406774520874023,
      "step": 1797
    },
    {
      "epoch": 6.855010986328125e-06,
      "step": 1797,
      "training_step_time": 10.288956642150879
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 3.1976420879364014,
      "step": 1798
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 1798,
      "training_step_time": 9.972479581832886
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 3.135556221008301,
      "step": 1798
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 1798,
      "training_step_time": 9.904115915298462
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 3.1363391876220703,
      "step": 1798
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 1798,
      "training_step_time": 10.183425664901733
    },
    {
      "epoch": 6.85882568359375e-06,
      "model_forward_time": 3.138152837753296,
      "step": 1798
    },
    {
      "epoch": 6.85882568359375e-06,
      "step": 1798,
      "training_step_time": 10.02736759185791
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 3.1551055908203125,
      "step": 1799
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 1799,
      "training_step_time": 9.928768396377563
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 3.1365163326263428,
      "step": 1799
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 1799,
      "training_step_time": 9.909311532974243
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 2.4349358081817627,
      "step": 1799
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 1799,
      "training_step_time": 7.458970069885254
    },
    {
      "epoch": 6.862640380859375e-06,
      "model_forward_time": 3.1361005306243896,
      "step": 1799
    },
    {
      "epoch": 6.862640380859375e-06,
      "step": 1799,
      "training_step_time": 10.309288740158081
    },
    {
      "epoch": 6.866455078125e-06,
      "grad_norm": 0.6760427355766296,
      "learning_rate": 1e-05,
      "loss": 0.113,
      "step": 1800
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 3.454251289367676,
      "step": 1800
    },
    {
      "dynamics_loss_avg": 0.07861147038638591,
      "epoch": 6.866455078125e-06,
      "step": 1800
    },
    {
      "action_loss_avg": 0.05229511652141809,
      "epoch": 6.866455078125e-06,
      "step": 1800
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 1800,
      "training_step_time": 10.228467464447021
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 3.1302032470703125,
      "step": 1800
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 1800,
      "training_step_time": 9.89639401435852
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 2.4307644367218018,
      "step": 1800
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 1800,
      "training_step_time": 7.452592372894287
    },
    {
      "epoch": 6.866455078125e-06,
      "model_forward_time": 3.132845401763916,
      "step": 1800
    },
    {
      "epoch": 6.866455078125e-06,
      "step": 1800,
      "training_step_time": 10.300709962844849
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 3.181060552597046,
      "step": 1801
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 1801,
      "training_step_time": 10.241536378860474
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 3.130237340927124,
      "step": 1801
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 1801,
      "training_step_time": 9.895370960235596
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 3.1386172771453857,
      "step": 1801
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 1801,
      "training_step_time": 9.89976954460144
    },
    {
      "epoch": 6.870269775390625e-06,
      "model_forward_time": 3.126532793045044,
      "step": 1801
    },
    {
      "epoch": 6.870269775390625e-06,
      "step": 1801,
      "training_step_time": 10.308865785598755
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 3.1360228061676025,
      "step": 1802
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 1802,
      "training_step_time": 9.903114080429077
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 2.7031400203704834,
      "step": 1802
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 1802,
      "training_step_time": 7.723240375518799
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 2.425421953201294,
      "step": 1802
    },
    {
      "dynamics_loss_avg": 0.09475100338459015,
      "epoch": 6.87408447265625e-06,
      "step": 1802
    },
    {
      "action_loss_avg": 0.05285684652626514,
      "epoch": 6.87408447265625e-06,
      "step": 1802
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 1802,
      "training_step_time": 7.4477598667144775
    },
    {
      "epoch": 6.87408447265625e-06,
      "model_forward_time": 3.1254544258117676,
      "step": 1802
    },
    {
      "epoch": 6.87408447265625e-06,
      "step": 1802,
      "training_step_time": 10.243002653121948
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 3.1561760902404785,
      "step": 1803
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 1803,
      "training_step_time": 9.922334671020508
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 3.127943277359009,
      "step": 1803
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 1803,
      "training_step_time": 9.890081405639648
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 3.125373601913452,
      "step": 1803
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 1803,
      "training_step_time": 9.887484788894653
    },
    {
      "epoch": 6.877899169921875e-06,
      "model_forward_time": 3.4004344940185547,
      "step": 1803
    },
    {
      "epoch": 6.877899169921875e-06,
      "step": 1803,
      "training_step_time": 10.251911401748657
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 3.1758499145507812,
      "step": 1804
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 1804,
      "training_step_time": 9.939756870269775
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 3.132721185684204,
      "step": 1804
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 1804,
      "training_step_time": 9.90147590637207
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 3.126298189163208,
      "step": 1804
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 1804,
      "training_step_time": 9.892305374145508
    },
    {
      "epoch": 6.8817138671875e-06,
      "model_forward_time": 3.391301155090332,
      "step": 1804
    },
    {
      "epoch": 6.8817138671875e-06,
      "step": 1804,
      "training_step_time": 10.283146858215332
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 3.1510016918182373,
      "step": 1805
    },
    {
      "dynamics_loss_avg": 0.07055516541004181,
      "epoch": 6.885528564453125e-06,
      "step": 1805
    },
    {
      "action_loss_avg": 0.051586083322763446,
      "epoch": 6.885528564453125e-06,
      "step": 1805
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 1805,
      "training_step_time": 9.919612169265747
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 2.4339139461517334,
      "step": 1805
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 1805,
      "training_step_time": 7.459543228149414
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 3.13452410697937,
      "step": 1805
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 1805,
      "training_step_time": 9.90518307685852
    },
    {
      "epoch": 6.885528564453125e-06,
      "model_forward_time": 2.433295488357544,
      "step": 1805
    },
    {
      "epoch": 6.885528564453125e-06,
      "step": 1805,
      "training_step_time": 10.026195287704468
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 3.161463499069214,
      "step": 1806
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 1806,
      "training_step_time": 9.932478189468384
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 3.412992238998413,
      "step": 1806
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 1806,
      "training_step_time": 10.18234133720398
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 3.1340272426605225,
      "step": 1806
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 1806,
      "training_step_time": 9.902620553970337
    },
    {
      "epoch": 6.88934326171875e-06,
      "model_forward_time": 3.1383211612701416,
      "step": 1806
    },
    {
      "epoch": 6.88934326171875e-06,
      "step": 1806,
      "training_step_time": 10.32153582572937
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 3.1527276039123535,
      "step": 1807
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 1807,
      "training_step_time": 9.92033052444458
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 3.1375391483306885,
      "step": 1807
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 1807,
      "training_step_time": 9.904859781265259
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 3.388427257537842,
      "step": 1807
    },
    {
      "dynamics_loss_avg": 0.04976212186738849,
      "epoch": 6.893157958984375e-06,
      "step": 1807
    },
    {
      "action_loss_avg": 0.04635027023032308,
      "epoch": 6.893157958984375e-06,
      "step": 1807
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 1807,
      "training_step_time": 10.155957221984863
    },
    {
      "epoch": 6.893157958984375e-06,
      "model_forward_time": 2.4301159381866455,
      "step": 1807
    },
    {
      "epoch": 6.893157958984375e-06,
      "step": 1807,
      "training_step_time": 10.253656148910522
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 3.1751036643981934,
      "step": 1808
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 1808,
      "training_step_time": 9.937589406967163
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 3.1295554637908936,
      "step": 1808
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 1808,
      "training_step_time": 9.891160726547241
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 3.137211799621582,
      "step": 1808
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 1808,
      "training_step_time": 9.894976139068604
    },
    {
      "epoch": 6.89697265625e-06,
      "model_forward_time": 3.437669038772583,
      "step": 1808
    },
    {
      "epoch": 6.89697265625e-06,
      "step": 1808,
      "training_step_time": 10.287973642349243
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 3.177457332611084,
      "step": 1809
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 1809,
      "training_step_time": 9.943522214889526
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 3.1292147636413574,
      "step": 1809
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 1809,
      "training_step_time": 9.894087791442871
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 3.1333634853363037,
      "step": 1809
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 1809,
      "training_step_time": 9.898438692092896
    },
    {
      "epoch": 6.900787353515625e-06,
      "model_forward_time": 3.4155397415161133,
      "step": 1809
    },
    {
      "epoch": 6.900787353515625e-06,
      "step": 1809,
      "training_step_time": 10.27439546585083
    },
    {
      "epoch": 6.90460205078125e-06,
      "grad_norm": 0.6541144251823425,
      "learning_rate": 1e-05,
      "loss": 0.1076,
      "step": 1810
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 3.17964768409729,
      "step": 1810
    },
    {
      "dynamics_loss_avg": 0.06157484445720911,
      "epoch": 6.90460205078125e-06,
      "step": 1810
    },
    {
      "action_loss_avg": 0.042138238251209256,
      "epoch": 6.90460205078125e-06,
      "step": 1810
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 1810,
      "training_step_time": 9.950488090515137
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 3.1352665424346924,
      "step": 1810
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 1810,
      "training_step_time": 9.9033842086792
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 3.13457989692688,
      "step": 1810
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 1810,
      "training_step_time": 9.899248123168945
    },
    {
      "epoch": 6.90460205078125e-06,
      "model_forward_time": 3.1307551860809326,
      "step": 1810
    },
    {
      "epoch": 6.90460205078125e-06,
      "step": 1810,
      "training_step_time": 10.30759859085083
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 3.4472579956054688,
      "step": 1811
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 1811,
      "training_step_time": 10.212161779403687
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 3.131014585494995,
      "step": 1811
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 1811,
      "training_step_time": 9.896384239196777
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 2.4303691387176514,
      "step": 1811
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 1811,
      "training_step_time": 7.45142936706543
    },
    {
      "epoch": 6.908416748046875e-06,
      "model_forward_time": 3.1268744468688965,
      "step": 1811
    },
    {
      "epoch": 6.908416748046875e-06,
      "step": 1811,
      "training_step_time": 10.295093297958374
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 3.167382001876831,
      "step": 1812
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 1812,
      "training_step_time": 9.93533205986023
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 3.1338298320770264,
      "step": 1812
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 1812,
      "training_step_time": 10.202876806259155
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 3.1351139545440674,
      "step": 1812
    },
    {
      "dynamics_loss_avg": 0.06791175492107868,
      "epoch": 6.9122314453125e-06,
      "step": 1812
    },
    {
      "action_loss_avg": 0.04656924717128277,
      "epoch": 6.9122314453125e-06,
      "step": 1812
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 1812,
      "training_step_time": 9.898531913757324
    },
    {
      "epoch": 6.9122314453125e-06,
      "model_forward_time": 3.1362764835357666,
      "step": 1812
    },
    {
      "epoch": 6.9122314453125e-06,
      "step": 1812,
      "training_step_time": 10.279969692230225
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 3.1859383583068848,
      "step": 1813
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 1813,
      "training_step_time": 9.951701641082764
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 2.4275705814361572,
      "step": 1813
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 1813,
      "training_step_time": 7.447163343429565
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 3.412771701812744,
      "step": 1813
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 1813,
      "training_step_time": 10.174929141998291
    },
    {
      "epoch": 6.916046142578125e-06,
      "model_forward_time": 3.132056474685669,
      "step": 1813
    },
    {
      "epoch": 6.916046142578125e-06,
      "step": 1813,
      "training_step_time": 10.224592924118042
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 3.1559529304504395,
      "step": 1814
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 1814,
      "training_step_time": 9.925852298736572
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 3.13423752784729,
      "step": 1814
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 1814,
      "training_step_time": 9.898186445236206
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 3.1284217834472656,
      "step": 1814
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 1814,
      "training_step_time": 9.888949155807495
    },
    {
      "epoch": 6.91986083984375e-06,
      "model_forward_time": 3.1316921710968018,
      "step": 1814
    },
    {
      "epoch": 6.91986083984375e-06,
      "step": 1814,
      "training_step_time": 10.010883569717407
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 3.4300484657287598,
      "step": 1815
    },
    {
      "dynamics_loss_avg": 0.062130773440003395,
      "epoch": 6.923675537109375e-06,
      "step": 1815
    },
    {
      "action_loss_avg": 0.03923407196998596,
      "epoch": 6.923675537109375e-06,
      "step": 1815
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 1815,
      "training_step_time": 10.198161602020264
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 3.1321609020233154,
      "step": 1815
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 1815,
      "training_step_time": 9.89938759803772
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 3.131279706954956,
      "step": 1815
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 1815,
      "training_step_time": 9.90007472038269
    },
    {
      "epoch": 6.923675537109375e-06,
      "model_forward_time": 3.1368651390075684,
      "step": 1815
    },
    {
      "epoch": 6.923675537109375e-06,
      "step": 1815,
      "training_step_time": 10.287630558013916
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 3.460160255432129,
      "step": 1816
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 1816,
      "training_step_time": 10.233397960662842
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 3.1361942291259766,
      "step": 1816
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 1816,
      "training_step_time": 9.904227018356323
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 3.1392507553100586,
      "step": 1816
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 1816,
      "training_step_time": 9.90715503692627
    },
    {
      "epoch": 6.927490234375e-06,
      "model_forward_time": 2.4291694164276123,
      "step": 1816
    },
    {
      "epoch": 6.927490234375e-06,
      "step": 1816,
      "training_step_time": 10.278860330581665
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 3.188192367553711,
      "step": 1817
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 1817,
      "training_step_time": 9.957212924957275
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 3.129199504852295,
      "step": 1817
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 1817,
      "training_step_time": 9.896130561828613
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 3.413980484008789,
      "step": 1817
    },
    {
      "dynamics_loss_avg": 0.053501987271010874,
      "epoch": 6.931304931640625e-06,
      "step": 1817
    },
    {
      "action_loss_avg": 0.039699928089976314,
      "epoch": 6.931304931640625e-06,
      "step": 1817
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 1817,
      "training_step_time": 10.183323621749878
    },
    {
      "epoch": 6.931304931640625e-06,
      "model_forward_time": 3.137578010559082,
      "step": 1817
    },
    {
      "epoch": 6.931304931640625e-06,
      "step": 1817,
      "training_step_time": 10.291545629501343
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 3.1493172645568848,
      "step": 1818
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 1818,
      "training_step_time": 9.915136098861694
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 3.131246328353882,
      "step": 1818
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 1818,
      "training_step_time": 9.902791976928711
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 2.431234121322632,
      "step": 1818
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 1818,
      "training_step_time": 7.453554630279541
    },
    {
      "epoch": 6.93511962890625e-06,
      "model_forward_time": 3.3955674171447754,
      "step": 1818
    },
    {
      "epoch": 6.93511962890625e-06,
      "step": 1818,
      "training_step_time": 10.255446195602417
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 3.1875696182250977,
      "step": 1819
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 1819,
      "training_step_time": 9.966252088546753
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 3.138685464859009,
      "step": 1819
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 1819,
      "training_step_time": 9.910702228546143
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 3.138073682785034,
      "step": 1819
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 1819,
      "training_step_time": 9.911383628845215
    },
    {
      "epoch": 6.938934326171875e-06,
      "model_forward_time": 3.430755853652954,
      "step": 1819
    },
    {
      "epoch": 6.938934326171875e-06,
      "step": 1819,
      "training_step_time": 10.298400640487671
    },
    {
      "epoch": 6.9427490234375e-06,
      "grad_norm": 0.61176598072052,
      "learning_rate": 1e-05,
      "loss": 0.1136,
      "step": 1820
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 3.180434465408325,
      "step": 1820
    },
    {
      "dynamics_loss_avg": 0.07638876028358936,
      "epoch": 6.9427490234375e-06,
      "step": 1820
    },
    {
      "action_loss_avg": 0.054765264969319105,
      "epoch": 6.9427490234375e-06,
      "step": 1820
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 1820,
      "training_step_time": 9.955535411834717
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 3.141146659851074,
      "step": 1820
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 1820,
      "training_step_time": 9.906733989715576
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 3.1391961574554443,
      "step": 1820
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 1820,
      "training_step_time": 9.910521745681763
    },
    {
      "epoch": 6.9427490234375e-06,
      "model_forward_time": 3.137333393096924,
      "step": 1820
    },
    {
      "epoch": 6.9427490234375e-06,
      "step": 1820,
      "training_step_time": 10.317198276519775
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 2.4717772006988525,
      "step": 1821
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 1821,
      "training_step_time": 7.496258020401001
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 3.425755023956299,
      "step": 1821
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 1821,
      "training_step_time": 10.189042091369629
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 3.143112897872925,
      "step": 1821
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 1821,
      "training_step_time": 9.905906438827515
    },
    {
      "epoch": 6.946563720703125e-06,
      "model_forward_time": 2.4340665340423584,
      "step": 1821
    },
    {
      "epoch": 6.946563720703125e-06,
      "step": 1821,
      "training_step_time": 10.291159391403198
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 3.1683523654937744,
      "step": 1822
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 1822,
      "training_step_time": 9.935896635055542
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 3.3975095748901367,
      "step": 1822
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 1822,
      "training_step_time": 10.163911581039429
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 3.1420023441314697,
      "step": 1822
    },
    {
      "dynamics_loss_avg": 0.06295124478638173,
      "epoch": 6.95037841796875e-06,
      "step": 1822
    },
    {
      "action_loss_avg": 0.049044110160320994,
      "epoch": 6.95037841796875e-06,
      "step": 1822
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 1822,
      "training_step_time": 9.91415810585022
    },
    {
      "epoch": 6.95037841796875e-06,
      "model_forward_time": 3.138864278793335,
      "step": 1822
    },
    {
      "epoch": 6.95037841796875e-06,
      "step": 1822,
      "training_step_time": 10.243799209594727
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 2.4720890522003174,
      "step": 1823
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 1823,
      "training_step_time": 7.500967979431152
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 3.1358559131622314,
      "step": 1823
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 1823,
      "training_step_time": 9.899476051330566
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 3.1364474296569824,
      "step": 1823
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 1823,
      "training_step_time": 9.898924827575684
    },
    {
      "epoch": 6.954193115234375e-06,
      "model_forward_time": 3.3813228607177734,
      "step": 1823
    },
    {
      "epoch": 6.954193115234375e-06,
      "step": 1823,
      "training_step_time": 10.257607698440552
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 3.1692569255828857,
      "step": 1824
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 1824,
      "training_step_time": 9.93533182144165
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 3.1413421630859375,
      "step": 1824
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 1824,
      "training_step_time": 9.908931255340576
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 3.1367151737213135,
      "step": 1824
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 1824,
      "training_step_time": 9.904662847518921
    },
    {
      "epoch": 6.9580078125e-06,
      "model_forward_time": 3.4124345779418945,
      "step": 1824
    },
    {
      "epoch": 6.9580078125e-06,
      "step": 1824,
      "training_step_time": 10.27231502532959
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 3.1590216159820557,
      "step": 1825
    },
    {
      "dynamics_loss_avg": 0.06866758856922388,
      "epoch": 6.961822509765625e-06,
      "step": 1825
    },
    {
      "action_loss_avg": 0.0358698558062315,
      "epoch": 6.961822509765625e-06,
      "step": 1825
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 1825,
      "training_step_time": 9.92983603477478
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 3.1408348083496094,
      "step": 1825
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 1825,
      "training_step_time": 9.915080070495605
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 3.141310691833496,
      "step": 1825
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 1825,
      "training_step_time": 9.9130220413208
    },
    {
      "epoch": 6.961822509765625e-06,
      "model_forward_time": 2.433910608291626,
      "step": 1825
    },
    {
      "epoch": 6.961822509765625e-06,
      "step": 1825,
      "training_step_time": 10.02445673942566
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 3.182802438735962,
      "step": 1826
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 1826,
      "training_step_time": 9.95217776298523
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 3.4402530193328857,
      "step": 1826
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 1826,
      "training_step_time": 10.207436323165894
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 3.138587474822998,
      "step": 1826
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 1826,
      "training_step_time": 9.90551209449768
    },
    {
      "epoch": 6.96563720703125e-06,
      "model_forward_time": 3.143627166748047,
      "step": 1826
    },
    {
      "epoch": 6.96563720703125e-06,
      "step": 1826,
      "training_step_time": 10.29936933517456
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 3.172438383102417,
      "step": 1827
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 1827,
      "training_step_time": 9.940379619598389
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 3.1360292434692383,
      "step": 1827
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 1827,
      "training_step_time": 10.174302577972412
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 3.1382157802581787,
      "step": 1827
    },
    {
      "dynamics_loss_avg": 0.06154061444103718,
      "epoch": 6.969451904296875e-06,
      "step": 1827
    },
    {
      "action_loss_avg": 0.034675643965601924,
      "epoch": 6.969451904296875e-06,
      "step": 1827
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 1827,
      "training_step_time": 9.904856443405151
    },
    {
      "epoch": 6.969451904296875e-06,
      "model_forward_time": 2.4313106536865234,
      "step": 1827
    },
    {
      "epoch": 6.969451904296875e-06,
      "step": 1827,
      "training_step_time": 10.321350574493408
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 3.179377555847168,
      "step": 1828
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 1828,
      "training_step_time": 9.949882745742798
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 3.135075092315674,
      "step": 1828
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 1828,
      "training_step_time": 9.901950120925903
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 3.140646457672119,
      "step": 1828
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 1828,
      "training_step_time": 9.908471822738647
    },
    {
      "epoch": 6.9732666015625e-06,
      "model_forward_time": 3.41878080368042,
      "step": 1828
    },
    {
      "epoch": 6.9732666015625e-06,
      "step": 1828,
      "training_step_time": 10.288763999938965
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 3.1544816493988037,
      "step": 1829
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 1829,
      "training_step_time": 9.932541131973267
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 3.1423563957214355,
      "step": 1829
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 1829,
      "training_step_time": 9.913903951644897
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 2.438011646270752,
      "step": 1829
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 1829,
      "training_step_time": 7.469069242477417
    },
    {
      "epoch": 6.977081298828125e-06,
      "model_forward_time": 3.13421893119812,
      "step": 1829
    },
    {
      "epoch": 6.977081298828125e-06,
      "step": 1829,
      "training_step_time": 10.287847757339478
    },
    {
      "epoch": 6.98089599609375e-06,
      "grad_norm": 0.7566860318183899,
      "learning_rate": 1e-05,
      "loss": 0.1104,
      "step": 1830
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 3.519148588180542,
      "step": 1830
    },
    {
      "dynamics_loss_avg": 0.06784855909645557,
      "epoch": 6.98089599609375e-06,
      "step": 1830
    },
    {
      "action_loss_avg": 0.06094938032329082,
      "epoch": 6.98089599609375e-06,
      "step": 1830
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 1830,
      "training_step_time": 10.2850182056427
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 2.437466859817505,
      "step": 1830
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 1830,
      "training_step_time": 7.461055040359497
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 3.1400020122528076,
      "step": 1830
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 1830,
      "training_step_time": 9.908523082733154
    },
    {
      "epoch": 6.98089599609375e-06,
      "model_forward_time": 3.1404459476470947,
      "step": 1830
    },
    {
      "epoch": 6.98089599609375e-06,
      "step": 1830,
      "training_step_time": 10.02845573425293
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 3.1564760208129883,
      "step": 1831
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 1831,
      "training_step_time": 10.205954313278198
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 3.1364524364471436,
      "step": 1831
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 1831,
      "training_step_time": 9.905352115631104
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 2.4316768646240234,
      "step": 1831
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 1831,
      "training_step_time": 7.452287673950195
    },
    {
      "epoch": 6.984710693359375e-06,
      "model_forward_time": 3.1268486976623535,
      "step": 1831
    },
    {
      "epoch": 6.984710693359375e-06,
      "step": 1831,
      "training_step_time": 10.017401695251465
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 3.140913248062134,
      "step": 1832
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 1832,
      "training_step_time": 9.90522813796997
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 3.1374831199645996,
      "step": 1832
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 1832,
      "training_step_time": 9.902380466461182
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 3.3959503173828125,
      "step": 1832
    },
    {
      "dynamics_loss_avg": 0.05040368475019932,
      "epoch": 6.988525390625e-06,
      "step": 1832
    },
    {
      "action_loss_avg": 0.03256592443212867,
      "epoch": 6.988525390625e-06,
      "step": 1832
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 1832,
      "training_step_time": 10.16424298286438
    },
    {
      "epoch": 6.988525390625e-06,
      "model_forward_time": 2.4284191131591797,
      "step": 1832
    },
    {
      "epoch": 6.988525390625e-06,
      "step": 1832,
      "training_step_time": 10.313281297683716
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 2.476637601852417,
      "step": 1833
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 1833,
      "training_step_time": 7.500234365463257
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 2.4250755310058594,
      "step": 1833
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 1833,
      "training_step_time": 7.446601390838623
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 3.125351905822754,
      "step": 1833
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 1833,
      "training_step_time": 9.89155888557434
    },
    {
      "epoch": 6.992340087890625e-06,
      "model_forward_time": 3.4079315662384033,
      "step": 1833
    },
    {
      "epoch": 6.992340087890625e-06,
      "step": 1833,
      "training_step_time": 10.271238565444946
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 3.136486053466797,
      "step": 1834
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 1834,
      "training_step_time": 9.905420541763306
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 3.1336395740509033,
      "step": 1834
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 1834,
      "training_step_time": 9.896344661712646
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 3.127377510070801,
      "step": 1834
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 1834,
      "training_step_time": 9.893100261688232
    },
    {
      "epoch": 6.99615478515625e-06,
      "model_forward_time": 2.4293558597564697,
      "step": 1834
    },
    {
      "epoch": 6.99615478515625e-06,
      "step": 1834,
      "training_step_time": 10.011355876922607
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 3.4416887760162354,
      "step": 1835
    },
    {
      "dynamics_loss_avg": 0.06990959774702787,
      "epoch": 6.999969482421875e-06,
      "step": 1835
    },
    {
      "action_loss_avg": 0.04463784079998732,
      "epoch": 6.999969482421875e-06,
      "step": 1835
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 1835,
      "training_step_time": 10.213651180267334
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 3.134268283843994,
      "step": 1835
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 1835,
      "training_step_time": 9.90462875366211
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 3.132991313934326,
      "step": 1835
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 1835,
      "training_step_time": 9.906068801879883
    },
    {
      "epoch": 6.999969482421875e-06,
      "model_forward_time": 3.137000560760498,
      "step": 1835
    },
    {
      "epoch": 6.999969482421875e-06,
      "step": 1835,
      "training_step_time": 10.261286735534668
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 3.170419692993164,
      "step": 1836
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 1836,
      "training_step_time": 10.20824408531189
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 3.139869451522827,
      "step": 1836
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 1836,
      "training_step_time": 9.914095401763916
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 3.133168935775757,
      "step": 1836
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 1836,
      "training_step_time": 9.911362886428833
    },
    {
      "epoch": 7.0037841796875e-06,
      "model_forward_time": 3.138366460800171,
      "step": 1836
    },
    {
      "epoch": 7.0037841796875e-06,
      "step": 1836,
      "training_step_time": 10.02746844291687
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 3.190201759338379,
      "step": 1837
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 1837,
      "training_step_time": 9.966129779815674
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 2.433462381362915,
      "step": 1837
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 1837,
      "training_step_time": 7.759890556335449
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 3.135277509689331,
      "step": 1837
    },
    {
      "dynamics_loss_avg": 0.07922720536589622,
      "epoch": 7.007598876953125e-06,
      "step": 1837
    },
    {
      "action_loss_avg": 0.04607285615056753,
      "epoch": 7.007598876953125e-06,
      "step": 1837
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 1837,
      "training_step_time": 9.913260459899902
    },
    {
      "epoch": 7.007598876953125e-06,
      "model_forward_time": 3.1392602920532227,
      "step": 1837
    },
    {
      "epoch": 7.007598876953125e-06,
      "step": 1837,
      "training_step_time": 10.306638240814209
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 3.186058759689331,
      "step": 1838
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 1838,
      "training_step_time": 9.965937376022339
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 3.136657953262329,
      "step": 1838
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 1838,
      "training_step_time": 9.912692070007324
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 3.3987009525299072,
      "step": 1838
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 1838,
      "training_step_time": 10.180893659591675
    },
    {
      "epoch": 7.01141357421875e-06,
      "model_forward_time": 3.140651226043701,
      "step": 1838
    },
    {
      "epoch": 7.01141357421875e-06,
      "step": 1838,
      "training_step_time": 10.320269584655762
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 3.174327850341797,
      "step": 1839
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 1839,
      "training_step_time": 9.951587915420532
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 3.1295711994171143,
      "step": 1839
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 1839,
      "training_step_time": 9.907684326171875
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 3.139070510864258,
      "step": 1839
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 1839,
      "training_step_time": 9.912131309509277
    },
    {
      "epoch": 7.015228271484375e-06,
      "model_forward_time": 3.409348726272583,
      "step": 1839
    },
    {
      "epoch": 7.015228271484375e-06,
      "step": 1839,
      "training_step_time": 10.276841402053833
    },
    {
      "epoch": 7.01904296875e-06,
      "grad_norm": 0.47956299781799316,
      "learning_rate": 1e-05,
      "loss": 0.1108,
      "step": 1840
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 3.173513174057007,
      "step": 1840
    },
    {
      "dynamics_loss_avg": 0.06615793220698833,
      "epoch": 7.01904296875e-06,
      "step": 1840
    },
    {
      "action_loss_avg": 0.04305375777184963,
      "epoch": 7.01904296875e-06,
      "step": 1840
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 1840,
      "training_step_time": 9.955012083053589
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 3.133929491043091,
      "step": 1840
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 1840,
      "training_step_time": 9.913289546966553
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 3.1392982006073,
      "step": 1840
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 1840,
      "training_step_time": 9.919699430465698
    },
    {
      "epoch": 7.01904296875e-06,
      "model_forward_time": 3.143402099609375,
      "step": 1840
    },
    {
      "epoch": 7.01904296875e-06,
      "step": 1840,
      "training_step_time": 10.32705044746399
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 1.797208309173584,
      "step": 1841
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 1841,
      "training_step_time": 5.262103796005249
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 3.407731294631958,
      "step": 1841
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 1841,
      "training_step_time": 10.187151908874512
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 3.1354634761810303,
      "step": 1841
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 1841,
      "training_step_time": 9.90618896484375
    },
    {
      "epoch": 7.022857666015625e-06,
      "model_forward_time": 3.136075019836426,
      "step": 1841
    },
    {
      "epoch": 7.022857666015625e-06,
      "step": 1841,
      "training_step_time": 10.02307391166687
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 2.4716389179229736,
      "step": 1842
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 1842,
      "training_step_time": 7.4997475147247314
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 3.1295506954193115,
      "step": 1842
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 1842,
      "training_step_time": 10.183593511581421
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 3.130844831466675,
      "step": 1842
    },
    {
      "dynamics_loss_avg": 0.08341887257993222,
      "epoch": 7.02667236328125e-06,
      "step": 1842
    },
    {
      "action_loss_avg": 0.05164545774459839,
      "epoch": 7.02667236328125e-06,
      "step": 1842
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 1842,
      "training_step_time": 9.906428098678589
    },
    {
      "epoch": 7.02667236328125e-06,
      "model_forward_time": 3.1346888542175293,
      "step": 1842
    },
    {
      "epoch": 7.02667236328125e-06,
      "step": 1842,
      "training_step_time": 10.28161907196045
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 3.188547134399414,
      "step": 1843
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 1843,
      "training_step_time": 9.964869260787964
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 3.133248805999756,
      "step": 1843
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 1843,
      "training_step_time": 9.909231662750244
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 3.391414165496826,
      "step": 1843
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 1843,
      "training_step_time": 10.167317390441895
    },
    {
      "epoch": 7.030487060546875e-06,
      "model_forward_time": 3.134056329727173,
      "step": 1843
    },
    {
      "epoch": 7.030487060546875e-06,
      "step": 1843,
      "training_step_time": 10.294358730316162
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 3.142211437225342,
      "step": 1844
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 1844,
      "training_step_time": 9.919058799743652
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 2.427802801132202,
      "step": 1844
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 1844,
      "training_step_time": 7.447302579879761
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 3.132297992706299,
      "step": 1844
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 1844,
      "training_step_time": 9.899715900421143
    },
    {
      "epoch": 7.0343017578125e-06,
      "model_forward_time": 2.4299354553222656,
      "step": 1844
    },
    {
      "epoch": 7.0343017578125e-06,
      "step": 1844,
      "training_step_time": 10.017057418823242
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 3.448310136795044,
      "step": 1845
    },
    {
      "dynamics_loss_avg": 0.07479904945939779,
      "epoch": 7.038116455078125e-06,
      "step": 1845
    },
    {
      "action_loss_avg": 0.041296448279172185,
      "epoch": 7.038116455078125e-06,
      "step": 1845
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 1845,
      "training_step_time": 10.220847368240356
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 3.137065887451172,
      "step": 1845
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 1845,
      "training_step_time": 9.909779787063599
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 2.431621789932251,
      "step": 1845
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 1845,
      "training_step_time": 7.464722633361816
    },
    {
      "epoch": 7.038116455078125e-06,
      "model_forward_time": 3.135115146636963,
      "step": 1845
    },
    {
      "epoch": 7.038116455078125e-06,
      "step": 1845,
      "training_step_time": 10.28602910041809
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 3.179716110229492,
      "step": 1846
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 1846,
      "training_step_time": 10.270841121673584
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 2.437964677810669,
      "step": 1846
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 1846,
      "training_step_time": 7.471410036087036
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 3.136854410171509,
      "step": 1846
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 1846,
      "training_step_time": 9.9094877243042
    },
    {
      "epoch": 7.04193115234375e-06,
      "model_forward_time": 3.135924816131592,
      "step": 1846
    },
    {
      "epoch": 7.04193115234375e-06,
      "step": 1846,
      "training_step_time": 10.02113151550293
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 3.1560213565826416,
      "step": 1847
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 1847,
      "training_step_time": 9.925594568252563
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 3.1332573890686035,
      "step": 1847
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 1847,
      "training_step_time": 9.90164065361023
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 2.7013092041015625,
      "step": 1847
    },
    {
      "dynamics_loss_avg": 0.05866771712899208,
      "epoch": 7.045745849609375e-06,
      "step": 1847
    },
    {
      "action_loss_avg": 0.033345235884189604,
      "epoch": 7.045745849609375e-06,
      "step": 1847
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 1847,
      "training_step_time": 7.733031749725342
    },
    {
      "epoch": 7.045745849609375e-06,
      "model_forward_time": 3.1327250003814697,
      "step": 1847
    },
    {
      "epoch": 7.045745849609375e-06,
      "step": 1847,
      "training_step_time": 10.049589395523071
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 3.172124147415161,
      "step": 1848
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 1848,
      "training_step_time": 9.946454048156738
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 3.1394331455230713,
      "step": 1848
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 1848,
      "training_step_time": 9.910762310028076
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 3.4220480918884277,
      "step": 1848
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 1848,
      "training_step_time": 10.196590185165405
    },
    {
      "epoch": 7.049560546875e-06,
      "model_forward_time": 3.136690616607666,
      "step": 1848
    },
    {
      "epoch": 7.049560546875e-06,
      "step": 1848,
      "training_step_time": 10.314846277236938
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 3.172330379486084,
      "step": 1849
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 1849,
      "training_step_time": 9.943555116653442
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 3.1373133659362793,
      "step": 1849
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 1849,
      "training_step_time": 9.905865907669067
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 3.1310713291168213,
      "step": 1849
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 1849,
      "training_step_time": 9.900423765182495
    },
    {
      "epoch": 7.053375244140625e-06,
      "model_forward_time": 3.13598895072937,
      "step": 1849
    },
    {
      "epoch": 7.053375244140625e-06,
      "step": 1849,
      "training_step_time": 10.013546228408813
    },
    {
      "epoch": 7.05718994140625e-06,
      "grad_norm": 0.5104018449783325,
      "learning_rate": 1e-05,
      "loss": 0.1137,
      "step": 1850
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 3.469531774520874,
      "step": 1850
    },
    {
      "dynamics_loss_avg": 0.06087101623415947,
      "epoch": 7.05718994140625e-06,
      "step": 1850
    },
    {
      "action_loss_avg": 0.04031596593558788,
      "epoch": 7.05718994140625e-06,
      "step": 1850
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 1850,
      "training_step_time": 10.243473052978516
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 2.432060956954956,
      "step": 1850
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 1850,
      "training_step_time": 7.456827640533447
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 3.1285130977630615,
      "step": 1850
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 1850,
      "training_step_time": 9.895793437957764
    },
    {
      "epoch": 7.05718994140625e-06,
      "model_forward_time": 3.132908344268799,
      "step": 1850
    },
    {
      "epoch": 7.05718994140625e-06,
      "step": 1850,
      "training_step_time": 10.280810117721558
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 3.42155385017395,
      "step": 1851
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 1851,
      "training_step_time": 10.18955683708191
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 3.13431453704834,
      "step": 1851
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 1851,
      "training_step_time": 9.903444290161133
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 3.1300065517425537,
      "step": 1851
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 1851,
      "training_step_time": 9.896377563476562
    },
    {
      "epoch": 7.061004638671875e-06,
      "model_forward_time": 3.1340057849884033,
      "step": 1851
    },
    {
      "epoch": 7.061004638671875e-06,
      "step": 1851,
      "training_step_time": 10.010582685470581
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 3.1554672718048096,
      "step": 1852
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 1852,
      "training_step_time": 9.925970077514648
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 3.1293275356292725,
      "step": 1852
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 1852,
      "training_step_time": 10.16269588470459
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 2.4309604167938232,
      "step": 1852
    },
    {
      "dynamics_loss_avg": 0.05471693091094494,
      "epoch": 7.0648193359375e-06,
      "step": 1852
    },
    {
      "action_loss_avg": 0.03247193060815334,
      "epoch": 7.0648193359375e-06,
      "step": 1852
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 1852,
      "training_step_time": 7.457745552062988
    },
    {
      "epoch": 7.0648193359375e-06,
      "model_forward_time": 2.4302194118499756,
      "step": 1852
    },
    {
      "epoch": 7.0648193359375e-06,
      "step": 1852,
      "training_step_time": 10.23910903930664
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 3.170680284500122,
      "step": 1853
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 1853,
      "training_step_time": 9.937440633773804
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 3.129845142364502,
      "step": 1853
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 1853,
      "training_step_time": 9.896695613861084
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 3.132375717163086,
      "step": 1853
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 1853,
      "training_step_time": 9.893166065216064
    },
    {
      "epoch": 7.068634033203125e-06,
      "model_forward_time": 3.4330780506134033,
      "step": 1853
    },
    {
      "epoch": 7.068634033203125e-06,
      "step": 1853,
      "training_step_time": 10.289042949676514
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 3.1433019638061523,
      "step": 1854
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 1854,
      "training_step_time": 9.91003704071045
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 3.13061261177063,
      "step": 1854
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 1854,
      "training_step_time": 9.892099380493164
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 3.134097099304199,
      "step": 1854
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 1854,
      "training_step_time": 9.904726266860962
    },
    {
      "epoch": 7.07244873046875e-06,
      "model_forward_time": 2.4352355003356934,
      "step": 1854
    },
    {
      "epoch": 7.07244873046875e-06,
      "step": 1854,
      "training_step_time": 10.321040391921997
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 3.1518380641937256,
      "step": 1855
    },
    {
      "dynamics_loss_avg": 0.07851966824382543,
      "epoch": 7.076263427734375e-06,
      "step": 1855
    },
    {
      "action_loss_avg": 0.05109265483915806,
      "epoch": 7.076263427734375e-06,
      "step": 1855
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 1855,
      "training_step_time": 9.93180227279663
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 3.1268579959869385,
      "step": 1855
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 1855,
      "training_step_time": 9.889930963516235
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 2.428544521331787,
      "step": 1855
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 1855,
      "training_step_time": 7.451563358306885
    },
    {
      "epoch": 7.076263427734375e-06,
      "model_forward_time": 2.4250216484069824,
      "step": 1855
    },
    {
      "epoch": 7.076263427734375e-06,
      "step": 1855,
      "training_step_time": 10.015267372131348
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 3.177830696105957,
      "step": 1856
    },
    {
      "epoch": 7.080078125e-06,
      "step": 1856,
      "training_step_time": 9.947839736938477
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 3.375913143157959,
      "step": 1856
    },
    {
      "epoch": 7.080078125e-06,
      "step": 1856,
      "training_step_time": 10.138716220855713
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 3.1309001445770264,
      "step": 1856
    },
    {
      "epoch": 7.080078125e-06,
      "step": 1856,
      "training_step_time": 9.895285844802856
    },
    {
      "epoch": 7.080078125e-06,
      "model_forward_time": 3.128329277038574,
      "step": 1856
    },
    {
      "epoch": 7.080078125e-06,
      "step": 1856,
      "training_step_time": 10.249752283096313
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 3.1727750301361084,
      "step": 1857
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 1857,
      "training_step_time": 9.941437244415283
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 3.3877248764038086,
      "step": 1857
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 1857,
      "training_step_time": 10.15560245513916
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 3.1413772106170654,
      "step": 1857
    },
    {
      "dynamics_loss_avg": 0.07505393885076046,
      "epoch": 7.083892822265625e-06,
      "step": 1857
    },
    {
      "action_loss_avg": 0.03550966521725059,
      "epoch": 7.083892822265625e-06,
      "step": 1857
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 1857,
      "training_step_time": 9.924419164657593
    },
    {
      "epoch": 7.083892822265625e-06,
      "model_forward_time": 3.1355044841766357,
      "step": 1857
    },
    {
      "epoch": 7.083892822265625e-06,
      "step": 1857,
      "training_step_time": 10.031450271606445
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 3.1719653606414795,
      "step": 1858
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 1858,
      "training_step_time": 9.965126752853394
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 3.135209083557129,
      "step": 1858
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 1858,
      "training_step_time": 9.942172527313232
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 3.1360995769500732,
      "step": 1858
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 1858,
      "training_step_time": 10.200778245925903
    },
    {
      "epoch": 7.08770751953125e-06,
      "model_forward_time": 3.1358985900878906,
      "step": 1858
    },
    {
      "epoch": 7.08770751953125e-06,
      "step": 1858,
      "training_step_time": 10.326259851455688
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 3.179619073867798,
      "step": 1859
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 1859,
      "training_step_time": 9.962433815002441
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 3.1478474140167236,
      "step": 1859
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 1859,
      "training_step_time": 9.922552585601807
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 3.13999342918396,
      "step": 1859
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 1859,
      "training_step_time": 9.910289287567139
    },
    {
      "epoch": 7.091522216796875e-06,
      "model_forward_time": 3.431222915649414,
      "step": 1859
    },
    {
      "epoch": 7.091522216796875e-06,
      "step": 1859,
      "training_step_time": 10.297777652740479
    },
    {
      "epoch": 7.0953369140625e-06,
      "grad_norm": 0.8861154913902283,
      "learning_rate": 1e-05,
      "loss": 0.1157,
      "step": 1860
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 3.166836738586426,
      "step": 1860
    },
    {
      "dynamics_loss_avg": 0.06587112098932266,
      "epoch": 7.0953369140625e-06,
      "step": 1860
    },
    {
      "action_loss_avg": 0.04427282959222793,
      "epoch": 7.0953369140625e-06,
      "step": 1860
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 1860,
      "training_step_time": 9.939974069595337
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 3.1397616863250732,
      "step": 1860
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 1860,
      "training_step_time": 9.908858060836792
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 3.1375396251678467,
      "step": 1860
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 1860,
      "training_step_time": 9.908466100692749
    },
    {
      "epoch": 7.0953369140625e-06,
      "model_forward_time": 3.141274929046631,
      "step": 1860
    },
    {
      "epoch": 7.0953369140625e-06,
      "step": 1860,
      "training_step_time": 10.29905366897583
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 3.4279446601867676,
      "step": 1861
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 1861,
      "training_step_time": 10.202173233032227
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 3.139003276824951,
      "step": 1861
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 1861,
      "training_step_time": 9.909498691558838
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 3.146980047225952,
      "step": 1861
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 1861,
      "training_step_time": 9.916471481323242
    },
    {
      "epoch": 7.099151611328125e-06,
      "model_forward_time": 3.142225742340088,
      "step": 1861
    },
    {
      "epoch": 7.099151611328125e-06,
      "step": 1861,
      "training_step_time": 10.0263512134552
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 3.1751863956451416,
      "step": 1862
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 1862,
      "training_step_time": 9.954651355743408
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 2.7158029079437256,
      "step": 1862
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 1862,
      "training_step_time": 7.744525909423828
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 3.135643243789673,
      "step": 1862
    },
    {
      "dynamics_loss_avg": 0.07625493593513966,
      "epoch": 7.10296630859375e-06,
      "step": 1862
    },
    {
      "action_loss_avg": 0.04928314108401537,
      "epoch": 7.10296630859375e-06,
      "step": 1862
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 1862,
      "training_step_time": 9.911290407180786
    },
    {
      "epoch": 7.10296630859375e-06,
      "model_forward_time": 3.1379001140594482,
      "step": 1862
    },
    {
      "epoch": 7.10296630859375e-06,
      "step": 1862,
      "training_step_time": 10.021910667419434
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 3.154731035232544,
      "step": 1863
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 1863,
      "training_step_time": 9.926277875900269
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 3.138946056365967,
      "step": 1863
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 1863,
      "training_step_time": 9.90920376777649
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 3.42803955078125,
      "step": 1863
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 1863,
      "training_step_time": 10.199602127075195
    },
    {
      "epoch": 7.106781005859375e-06,
      "model_forward_time": 2.440056085586548,
      "step": 1863
    },
    {
      "epoch": 7.106781005859375e-06,
      "step": 1863,
      "training_step_time": 10.25498628616333
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 3.180375576019287,
      "step": 1864
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 1864,
      "training_step_time": 9.958827495574951
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 3.1449759006500244,
      "step": 1864
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 1864,
      "training_step_time": 9.924989223480225
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 3.144165277481079,
      "step": 1864
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 1864,
      "training_step_time": 9.922917366027832
    },
    {
      "epoch": 7.110595703125e-06,
      "model_forward_time": 3.143592357635498,
      "step": 1864
    },
    {
      "epoch": 7.110595703125e-06,
      "step": 1864,
      "training_step_time": 10.307034254074097
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 3.185655117034912,
      "step": 1865
    },
    {
      "dynamics_loss_avg": 0.06746509671211243,
      "epoch": 7.114410400390625e-06,
      "step": 1865
    },
    {
      "action_loss_avg": 0.04833348914980888,
      "epoch": 7.114410400390625e-06,
      "step": 1865
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 1865,
      "training_step_time": 9.973278760910034
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 3.136523723602295,
      "step": 1865
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 1865,
      "training_step_time": 9.90767216682434
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 3.1443774700164795,
      "step": 1865
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 1865,
      "training_step_time": 9.913737535476685
    },
    {
      "epoch": 7.114410400390625e-06,
      "model_forward_time": 3.142585277557373,
      "step": 1865
    },
    {
      "epoch": 7.114410400390625e-06,
      "step": 1865,
      "training_step_time": 10.305885076522827
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 3.4698994159698486,
      "step": 1866
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 1866,
      "training_step_time": 10.240068435668945
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 3.1414401531219482,
      "step": 1866
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 1866,
      "training_step_time": 9.905072450637817
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 3.142470598220825,
      "step": 1866
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 1866,
      "training_step_time": 9.906239032745361
    },
    {
      "epoch": 7.11822509765625e-06,
      "model_forward_time": 3.137904644012451,
      "step": 1866
    },
    {
      "epoch": 7.11822509765625e-06,
      "step": 1866,
      "training_step_time": 10.280880451202393
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 2.444464921951294,
      "step": 1867
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 1867,
      "training_step_time": 7.467258930206299
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 2.431056261062622,
      "step": 1867
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 1867,
      "training_step_time": 7.451516628265381
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 3.4152417182922363,
      "step": 1867
    },
    {
      "dynamics_loss_avg": 0.06132922898977995,
      "epoch": 7.122039794921875e-06,
      "step": 1867
    },
    {
      "action_loss_avg": 0.03895655507221818,
      "epoch": 7.122039794921875e-06,
      "step": 1867
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 1867,
      "training_step_time": 10.175666809082031
    },
    {
      "epoch": 7.122039794921875e-06,
      "model_forward_time": 2.433267831802368,
      "step": 1867
    },
    {
      "epoch": 7.122039794921875e-06,
      "step": 1867,
      "training_step_time": 10.01537299156189
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 3.1646604537963867,
      "step": 1868
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 1868,
      "training_step_time": 9.930883646011353
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 3.1378512382507324,
      "step": 1868
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 1868,
      "training_step_time": 9.89853811264038
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 3.134211301803589,
      "step": 1868
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 1868,
      "training_step_time": 9.895115375518799
    },
    {
      "epoch": 7.1258544921875e-06,
      "model_forward_time": 3.419931650161743,
      "step": 1868
    },
    {
      "epoch": 7.1258544921875e-06,
      "step": 1868,
      "training_step_time": 10.274842739105225
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 3.1778926849365234,
      "step": 1869
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 1869,
      "training_step_time": 9.937729597091675
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 3.1317598819732666,
      "step": 1869
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 1869,
      "training_step_time": 9.889569997787476
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 3.1285061836242676,
      "step": 1869
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 1869,
      "training_step_time": 9.890941858291626
    },
    {
      "epoch": 7.129669189453125e-06,
      "model_forward_time": 3.133845806121826,
      "step": 1869
    },
    {
      "epoch": 7.129669189453125e-06,
      "step": 1869,
      "training_step_time": 10.273084163665771
    },
    {
      "epoch": 7.13348388671875e-06,
      "grad_norm": 0.5643344521522522,
      "learning_rate": 1e-05,
      "loss": 0.1111,
      "step": 1870
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 3.481480360031128,
      "step": 1870
    },
    {
      "dynamics_loss_avg": 0.0681963387876749,
      "epoch": 7.13348388671875e-06,
      "step": 1870
    },
    {
      "action_loss_avg": 0.03996026162058115,
      "epoch": 7.13348388671875e-06,
      "step": 1870
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 1870,
      "training_step_time": 10.25018310546875
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 3.1360554695129395,
      "step": 1870
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 1870,
      "training_step_time": 9.905250787734985
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 3.134775400161743,
      "step": 1870
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 1870,
      "training_step_time": 9.90288496017456
    },
    {
      "epoch": 7.13348388671875e-06,
      "model_forward_time": 3.1362106800079346,
      "step": 1870
    },
    {
      "epoch": 7.13348388671875e-06,
      "step": 1870,
      "training_step_time": 10.270893335342407
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 3.4766957759857178,
      "step": 1871
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 1871,
      "training_step_time": 10.247220039367676
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 3.1379737854003906,
      "step": 1871
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 1871,
      "training_step_time": 9.911728858947754
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 3.1344149112701416,
      "step": 1871
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 1871,
      "training_step_time": 9.907335996627808
    },
    {
      "epoch": 7.137298583984375e-06,
      "model_forward_time": 3.1405863761901855,
      "step": 1871
    },
    {
      "epoch": 7.137298583984375e-06,
      "step": 1871,
      "training_step_time": 10.304203271865845
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 2.4614102840423584,
      "step": 1872
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 1872,
      "training_step_time": 7.485188007354736
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 3.137730598449707,
      "step": 1872
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 1872,
      "training_step_time": 9.908981084823608
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 2.6959292888641357,
      "step": 1872
    },
    {
      "dynamics_loss_avg": 0.058412319608032705,
      "epoch": 7.14111328125e-06,
      "step": 1872
    },
    {
      "action_loss_avg": 0.034829181246459485,
      "epoch": 7.14111328125e-06,
      "step": 1872
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 1872,
      "training_step_time": 7.725384712219238
    },
    {
      "epoch": 7.14111328125e-06,
      "model_forward_time": 3.138935089111328,
      "step": 1872
    },
    {
      "epoch": 7.14111328125e-06,
      "step": 1872,
      "training_step_time": 10.017026901245117
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 3.1723380088806152,
      "step": 1873
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 1873,
      "training_step_time": 9.93843698501587
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 3.1290817260742188,
      "step": 1873
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 1873,
      "training_step_time": 9.893401384353638
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 3.12709379196167,
      "step": 1873
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 1873,
      "training_step_time": 9.888096332550049
    },
    {
      "epoch": 7.144927978515625e-06,
      "model_forward_time": 3.426109790802002,
      "step": 1873
    },
    {
      "epoch": 7.144927978515625e-06,
      "step": 1873,
      "training_step_time": 10.307080268859863
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 3.166876792907715,
      "step": 1874
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 1874,
      "training_step_time": 9.929145812988281
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 3.1268820762634277,
      "step": 1874
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 1874,
      "training_step_time": 9.88683271408081
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 3.1275081634521484,
      "step": 1874
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 1874,
      "training_step_time": 9.887091398239136
    },
    {
      "epoch": 7.14874267578125e-06,
      "model_forward_time": 3.424983024597168,
      "step": 1874
    },
    {
      "epoch": 7.14874267578125e-06,
      "step": 1874,
      "training_step_time": 10.311111688613892
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 3.1801412105560303,
      "step": 1875
    },
    {
      "dynamics_loss_avg": 0.07164738401770591,
      "epoch": 7.152557373046875e-06,
      "step": 1875
    },
    {
      "action_loss_avg": 0.05390114989131689,
      "epoch": 7.152557373046875e-06,
      "step": 1875
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 1875,
      "training_step_time": 9.944276094436646
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 3.1272385120391846,
      "step": 1875
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 1875,
      "training_step_time": 9.88841724395752
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 3.1288602352142334,
      "step": 1875
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 1875,
      "training_step_time": 9.891765832901001
    },
    {
      "epoch": 7.152557373046875e-06,
      "model_forward_time": 3.1265323162078857,
      "step": 1875
    },
    {
      "epoch": 7.152557373046875e-06,
      "step": 1875,
      "training_step_time": 10.006205081939697
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 3.173840284347534,
      "step": 1876
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 1876,
      "training_step_time": 10.210194110870361
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 3.1355032920837402,
      "step": 1876
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 1876,
      "training_step_time": 9.902267694473267
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 3.141458511352539,
      "step": 1876
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 1876,
      "training_step_time": 9.908820152282715
    },
    {
      "epoch": 7.1563720703125e-06,
      "model_forward_time": 3.137295722961426,
      "step": 1876
    },
    {
      "epoch": 7.1563720703125e-06,
      "step": 1876,
      "training_step_time": 10.31472659111023
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 3.1893482208251953,
      "step": 1877
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 1877,
      "training_step_time": 9.957740545272827
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 3.440202474594116,
      "step": 1877
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 1877,
      "training_step_time": 10.20796513557434
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 3.1369822025299072,
      "step": 1877
    },
    {
      "dynamics_loss_avg": 0.059664669632911685,
      "epoch": 7.160186767578125e-06,
      "step": 1877
    },
    {
      "action_loss_avg": 0.03328770529478788,
      "epoch": 7.160186767578125e-06,
      "step": 1877
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 1877,
      "training_step_time": 9.917927026748657
    },
    {
      "epoch": 7.160186767578125e-06,
      "model_forward_time": 3.1381258964538574,
      "step": 1877
    },
    {
      "epoch": 7.160186767578125e-06,
      "step": 1877,
      "training_step_time": 10.271990299224854
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 3.146599769592285,
      "step": 1878
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 1878,
      "training_step_time": 9.923552513122559
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 3.134821891784668,
      "step": 1878
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 1878,
      "training_step_time": 9.905102014541626
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 3.42880916595459,
      "step": 1878
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 1878,
      "training_step_time": 10.1995849609375
    },
    {
      "epoch": 7.16400146484375e-06,
      "model_forward_time": 2.433288812637329,
      "step": 1878
    },
    {
      "epoch": 7.16400146484375e-06,
      "step": 1878,
      "training_step_time": 10.022601127624512
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 3.1813948154449463,
      "step": 1879
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 1879,
      "training_step_time": 9.945640802383423
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 2.4311044216156006,
      "step": 1879
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 1879,
      "training_step_time": 7.450053691864014
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 3.1287240982055664,
      "step": 1879
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 1879,
      "training_step_time": 9.888048648834229
    },
    {
      "epoch": 7.167816162109375e-06,
      "model_forward_time": 3.131762742996216,
      "step": 1879
    },
    {
      "epoch": 7.167816162109375e-06,
      "step": 1879,
      "training_step_time": 10.270534038543701
    },
    {
      "epoch": 7.171630859375e-06,
      "grad_norm": 0.5485338568687439,
      "learning_rate": 1e-05,
      "loss": 0.1085,
      "step": 1880
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 3.183412790298462,
      "step": 1880
    },
    {
      "dynamics_loss_avg": 0.05807233564555645,
      "epoch": 7.171630859375e-06,
      "step": 1880
    },
    {
      "action_loss_avg": 0.035901207476854324,
      "epoch": 7.171630859375e-06,
      "step": 1880
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 1880,
      "training_step_time": 9.964918851852417
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 3.138458251953125,
      "step": 1880
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 1880,
      "training_step_time": 9.90242075920105
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 3.133955955505371,
      "step": 1880
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 1880,
      "training_step_time": 9.90203070640564
    },
    {
      "epoch": 7.171630859375e-06,
      "model_forward_time": 2.42918062210083,
      "step": 1880
    },
    {
      "epoch": 7.171630859375e-06,
      "step": 1880,
      "training_step_time": 10.030662775039673
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 3.1830430030822754,
      "step": 1881
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 1881,
      "training_step_time": 9.949023962020874
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 2.74123215675354,
      "step": 1881
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 1881,
      "training_step_time": 7.76216459274292
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 3.1323556900024414,
      "step": 1881
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 1881,
      "training_step_time": 9.89927864074707
    },
    {
      "epoch": 7.175445556640625e-06,
      "model_forward_time": 3.1374857425689697,
      "step": 1881
    },
    {
      "epoch": 7.175445556640625e-06,
      "step": 1881,
      "training_step_time": 10.307507038116455
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 2.4678633213043213,
      "step": 1882
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 1882,
      "training_step_time": 7.490632772445679
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 3.1310203075408936,
      "step": 1882
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 1882,
      "training_step_time": 9.892933368682861
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 3.376246690750122,
      "step": 1882
    },
    {
      "dynamics_loss_avg": 0.07409236691892147,
      "epoch": 7.17926025390625e-06,
      "step": 1882
    },
    {
      "action_loss_avg": 0.040024266391992566,
      "epoch": 7.17926025390625e-06,
      "step": 1882
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 1882,
      "training_step_time": 10.141242504119873
    },
    {
      "epoch": 7.17926025390625e-06,
      "model_forward_time": 3.1392831802368164,
      "step": 1882
    },
    {
      "epoch": 7.17926025390625e-06,
      "step": 1882,
      "training_step_time": 10.242252588272095
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 3.1737136840820312,
      "step": 1883
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 1883,
      "training_step_time": 9.939460039138794
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 1.7435040473937988,
      "step": 1883
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 1883,
      "training_step_time": 5.197458267211914
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 2.428887128829956,
      "step": 1883
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 1883,
      "training_step_time": 7.445638418197632
    },
    {
      "epoch": 7.183074951171875e-06,
      "model_forward_time": 3.4169464111328125,
      "step": 1883
    },
    {
      "epoch": 7.183074951171875e-06,
      "step": 1883,
      "training_step_time": 10.27824330329895
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 3.1805591583251953,
      "step": 1884
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 1884,
      "training_step_time": 9.945715188980103
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 3.136545419692993,
      "step": 1884
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 1884,
      "training_step_time": 9.899145126342773
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 3.1355106830596924,
      "step": 1884
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 1884,
      "training_step_time": 9.89988088607788
    },
    {
      "epoch": 7.1868896484375e-06,
      "model_forward_time": 3.1375887393951416,
      "step": 1884
    },
    {
      "epoch": 7.1868896484375e-06,
      "step": 1884,
      "training_step_time": 10.268558979034424
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 3.4706344604492188,
      "step": 1885
    },
    {
      "dynamics_loss_avg": 0.05925499219447374,
      "epoch": 7.190704345703125e-06,
      "step": 1885
    },
    {
      "action_loss_avg": 0.03954711072146892,
      "epoch": 7.190704345703125e-06,
      "step": 1885
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 1885,
      "training_step_time": 10.239188194274902
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 3.1307454109191895,
      "step": 1885
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 1885,
      "training_step_time": 9.9007728099823
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 3.135300636291504,
      "step": 1885
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 1885,
      "training_step_time": 9.916555881500244
    },
    {
      "epoch": 7.190704345703125e-06,
      "model_forward_time": 3.1375184059143066,
      "step": 1885
    },
    {
      "epoch": 7.190704345703125e-06,
      "step": 1885,
      "training_step_time": 10.364838123321533
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 3.170405626296997,
      "step": 1886
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 1886,
      "training_step_time": 9.947159051895142
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 3.386436700820923,
      "step": 1886
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 1886,
      "training_step_time": 10.156415939331055
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 3.136780261993408,
      "step": 1886
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 1886,
      "training_step_time": 9.903698682785034
    },
    {
      "epoch": 7.19451904296875e-06,
      "model_forward_time": 2.4330906867980957,
      "step": 1886
    },
    {
      "epoch": 7.19451904296875e-06,
      "step": 1886,
      "training_step_time": 10.32179307937622
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 3.1724464893341064,
      "step": 1887
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 1887,
      "training_step_time": 9.941839694976807
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 3.134981155395508,
      "step": 1887
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 1887,
      "training_step_time": 9.902103424072266
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 3.439415454864502,
      "step": 1887
    },
    {
      "dynamics_loss_avg": 0.06675977371633053,
      "epoch": 7.198333740234375e-06,
      "step": 1887
    },
    {
      "action_loss_avg": 0.0416723663918674,
      "epoch": 7.198333740234375e-06,
      "step": 1887
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 1887,
      "training_step_time": 10.20615267753601
    },
    {
      "epoch": 7.198333740234375e-06,
      "model_forward_time": 3.130016565322876,
      "step": 1887
    },
    {
      "epoch": 7.198333740234375e-06,
      "step": 1887,
      "training_step_time": 10.312962293624878
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 3.165242910385132,
      "step": 1888
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 1888,
      "training_step_time": 9.935095071792603
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 3.1353952884674072,
      "step": 1888
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 1888,
      "training_step_time": 9.907476902008057
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 3.1318559646606445,
      "step": 1888
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 1888,
      "training_step_time": 9.907927989959717
    },
    {
      "epoch": 7.2021484375e-06,
      "model_forward_time": 3.411314010620117,
      "step": 1888
    },
    {
      "epoch": 7.2021484375e-06,
      "step": 1888,
      "training_step_time": 10.286125421524048
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 3.175544261932373,
      "step": 1889
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 1889,
      "training_step_time": 9.947790145874023
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 3.1363654136657715,
      "step": 1889
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 1889,
      "training_step_time": 9.904252052307129
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 3.1355204582214355,
      "step": 1889
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 1889,
      "training_step_time": 9.905577182769775
    },
    {
      "epoch": 7.205963134765625e-06,
      "model_forward_time": 3.134448289871216,
      "step": 1889
    },
    {
      "epoch": 7.205963134765625e-06,
      "step": 1889,
      "training_step_time": 10.014491558074951
    },
    {
      "epoch": 7.20977783203125e-06,
      "grad_norm": 0.5560533404350281,
      "learning_rate": 1e-05,
      "loss": 0.1136,
      "step": 1890
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 3.4403505325317383,
      "step": 1890
    },
    {
      "dynamics_loss_avg": 0.07126791067421437,
      "epoch": 7.20977783203125e-06,
      "step": 1890
    },
    {
      "action_loss_avg": 0.04485194236040115,
      "epoch": 7.20977783203125e-06,
      "step": 1890
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 1890,
      "training_step_time": 10.211305856704712
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 3.137031078338623,
      "step": 1890
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 1890,
      "training_step_time": 9.905739784240723
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 3.1385300159454346,
      "step": 1890
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 1890,
      "training_step_time": 9.911784410476685
    },
    {
      "epoch": 7.20977783203125e-06,
      "model_forward_time": 3.1430134773254395,
      "step": 1890
    },
    {
      "epoch": 7.20977783203125e-06,
      "step": 1890,
      "training_step_time": 10.291583061218262
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 3.1851234436035156,
      "step": 1891
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 1891,
      "training_step_time": 9.969173669815063
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 3.4523346424102783,
      "step": 1891
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 1891,
      "training_step_time": 10.221317768096924
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 3.13061785697937,
      "step": 1891
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 1891,
      "training_step_time": 9.899863481521606
    },
    {
      "epoch": 7.213592529296875e-06,
      "model_forward_time": 3.132256269454956,
      "step": 1891
    },
    {
      "epoch": 7.213592529296875e-06,
      "step": 1891,
      "training_step_time": 10.298648118972778
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 3.1713647842407227,
      "step": 1892
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 1892,
      "training_step_time": 9.932319641113281
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 3.126128673553467,
      "step": 1892
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 1892,
      "training_step_time": 9.885534763336182
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 3.3816726207733154,
      "step": 1892
    },
    {
      "dynamics_loss_avg": 0.0652004038915038,
      "epoch": 7.2174072265625e-06,
      "step": 1892
    },
    {
      "action_loss_avg": 0.0428038252517581,
      "epoch": 7.2174072265625e-06,
      "step": 1892
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 1892,
      "training_step_time": 10.148236513137817
    },
    {
      "epoch": 7.2174072265625e-06,
      "model_forward_time": 3.1352972984313965,
      "step": 1892
    },
    {
      "epoch": 7.2174072265625e-06,
      "step": 1892,
      "training_step_time": 10.282631874084473
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 3.156902551651001,
      "step": 1893
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 1893,
      "training_step_time": 9.91929030418396
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 2.42628812789917,
      "step": 1893
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 1893,
      "training_step_time": 7.4484381675720215
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 3.126459836959839,
      "step": 1893
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 1893,
      "training_step_time": 9.890288352966309
    },
    {
      "epoch": 7.221221923828125e-06,
      "model_forward_time": 3.40069317817688,
      "step": 1893
    },
    {
      "epoch": 7.221221923828125e-06,
      "step": 1893,
      "training_step_time": 10.26290774345398
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 3.184232711791992,
      "step": 1894
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 1894,
      "training_step_time": 9.951401472091675
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 3.13132381439209,
      "step": 1894
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 1894,
      "training_step_time": 9.896726131439209
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 3.137256383895874,
      "step": 1894
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 1894,
      "training_step_time": 9.90700387954712
    },
    {
      "epoch": 7.22503662109375e-06,
      "model_forward_time": 3.1381771564483643,
      "step": 1894
    },
    {
      "epoch": 7.22503662109375e-06,
      "step": 1894,
      "training_step_time": 10.291919231414795
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 3.485591173171997,
      "step": 1895
    },
    {
      "dynamics_loss_avg": 0.05773291466757655,
      "epoch": 7.228851318359375e-06,
      "step": 1895
    },
    {
      "action_loss_avg": 0.03819205288309604,
      "epoch": 7.228851318359375e-06,
      "step": 1895
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 1895,
      "training_step_time": 10.258301019668579
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 3.134727954864502,
      "step": 1895
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 1895,
      "training_step_time": 9.91086459159851
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 3.142237424850464,
      "step": 1895
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 1895,
      "training_step_time": 9.913254976272583
    },
    {
      "epoch": 7.228851318359375e-06,
      "model_forward_time": 3.1393094062805176,
      "step": 1895
    },
    {
      "epoch": 7.228851318359375e-06,
      "step": 1895,
      "training_step_time": 10.313962459564209
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 3.1828534603118896,
      "step": 1896
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 1896,
      "training_step_time": 9.958250284194946
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 3.385171413421631,
      "step": 1896
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 1896,
      "training_step_time": 10.15153431892395
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 3.138195753097534,
      "step": 1896
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 1896,
      "training_step_time": 9.907358169555664
    },
    {
      "epoch": 7.232666015625e-06,
      "model_forward_time": 3.1350674629211426,
      "step": 1896
    },
    {
      "epoch": 7.232666015625e-06,
      "step": 1896,
      "training_step_time": 10.273288488388062
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 3.1815829277038574,
      "step": 1897
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 1897,
      "training_step_time": 9.952484369277954
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 3.137181043624878,
      "step": 1897
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 1897,
      "training_step_time": 9.91314697265625
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 3.427079916000366,
      "step": 1897
    },
    {
      "dynamics_loss_avg": 0.06270789299160243,
      "epoch": 7.236480712890625e-06,
      "step": 1897
    },
    {
      "action_loss_avg": 0.03716250564903021,
      "epoch": 7.236480712890625e-06,
      "step": 1897
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 1897,
      "training_step_time": 10.20706844329834
    },
    {
      "epoch": 7.236480712890625e-06,
      "model_forward_time": 3.1358776092529297,
      "step": 1897
    },
    {
      "epoch": 7.236480712890625e-06,
      "step": 1897,
      "training_step_time": 10.036822319030762
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 3.1594295501708984,
      "step": 1898
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 1898,
      "training_step_time": 9.937387466430664
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 3.136476993560791,
      "step": 1898
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 1898,
      "training_step_time": 9.903788089752197
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 3.128635883331299,
      "step": 1898
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 1898,
      "training_step_time": 9.898573398590088
    },
    {
      "epoch": 7.24029541015625e-06,
      "model_forward_time": 3.4148149490356445,
      "step": 1898
    },
    {
      "epoch": 7.24029541015625e-06,
      "step": 1898,
      "training_step_time": 10.301276922225952
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 3.1407546997070312,
      "step": 1899
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 1899,
      "training_step_time": 9.905705213546753
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 3.133812189102173,
      "step": 1899
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 1899,
      "training_step_time": 9.901144027709961
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 2.4288454055786133,
      "step": 1899
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 1899,
      "training_step_time": 7.461024761199951
    },
    {
      "epoch": 7.244110107421875e-06,
      "model_forward_time": 2.4323313236236572,
      "step": 1899
    },
    {
      "epoch": 7.244110107421875e-06,
      "step": 1899,
      "training_step_time": 10.295881271362305
    },
    {
      "epoch": 7.2479248046875e-06,
      "grad_norm": 0.4007532596588135,
      "learning_rate": 1e-05,
      "loss": 0.1112,
      "step": 1900
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 3.410811424255371,
      "step": 1900
    },
    {
      "dynamics_loss_avg": 0.04388602990657091,
      "epoch": 7.2479248046875e-06,
      "step": 1900
    },
    {
      "action_loss_avg": 0.030372895300388336,
      "epoch": 7.2479248046875e-06,
      "step": 1900
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 1900,
      "training_step_time": 10.180831670761108
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 3.133014440536499,
      "step": 1900
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 1900,
      "training_step_time": 9.896157026290894
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 3.1303799152374268,
      "step": 1900
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 1900,
      "training_step_time": 9.891541957855225
    },
    {
      "epoch": 7.2479248046875e-06,
      "model_forward_time": 3.133246421813965,
      "step": 1900
    },
    {
      "epoch": 7.2479248046875e-06,
      "step": 1900,
      "training_step_time": 10.278034925460815
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 3.1610333919525146,
      "step": 1901
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 1901,
      "training_step_time": 9.92748475074768
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 3.134101152420044,
      "step": 1901
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 1901,
      "training_step_time": 10.195526361465454
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 3.1300461292266846,
      "step": 1901
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 1901,
      "training_step_time": 9.895534992218018
    },
    {
      "epoch": 7.251739501953125e-06,
      "model_forward_time": 3.1376309394836426,
      "step": 1901
    },
    {
      "epoch": 7.251739501953125e-06,
      "step": 1901,
      "training_step_time": 10.293431043624878
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 2.479290723800659,
      "step": 1902
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 1902,
      "training_step_time": 7.5115861892700195
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 3.130655288696289,
      "step": 1902
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 1902,
      "training_step_time": 9.904480695724487
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 3.1363747119903564,
      "step": 1902
    },
    {
      "dynamics_loss_avg": 0.06766159497201443,
      "epoch": 7.25555419921875e-06,
      "step": 1902
    },
    {
      "action_loss_avg": 0.042973912134766576,
      "epoch": 7.25555419921875e-06,
      "step": 1902
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 1902,
      "training_step_time": 10.198169469833374
    },
    {
      "epoch": 7.25555419921875e-06,
      "model_forward_time": 3.135596990585327,
      "step": 1902
    },
    {
      "epoch": 7.25555419921875e-06,
      "step": 1902,
      "training_step_time": 10.029839515686035
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 3.1668272018432617,
      "step": 1903
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 1903,
      "training_step_time": 9.938164472579956
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 3.1327641010284424,
      "step": 1903
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 1903,
      "training_step_time": 9.90317153930664
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 3.135216474533081,
      "step": 1903
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 1903,
      "training_step_time": 9.910651683807373
    },
    {
      "epoch": 7.259368896484375e-06,
      "model_forward_time": 3.411871910095215,
      "step": 1903
    },
    {
      "epoch": 7.259368896484375e-06,
      "step": 1903,
      "training_step_time": 10.28126072883606
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 3.181121587753296,
      "step": 1904
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 1904,
      "training_step_time": 9.955268859863281
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 3.1342666149139404,
      "step": 1904
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 1904,
      "training_step_time": 9.91048550605774
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 3.137676477432251,
      "step": 1904
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 1904,
      "training_step_time": 9.907822608947754
    },
    {
      "epoch": 7.26318359375e-06,
      "model_forward_time": 3.1305079460144043,
      "step": 1904
    },
    {
      "epoch": 7.26318359375e-06,
      "step": 1904,
      "training_step_time": 10.263824939727783
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 3.1590535640716553,
      "step": 1905
    },
    {
      "dynamics_loss_avg": 0.06143185682594776,
      "epoch": 7.266998291015625e-06,
      "step": 1905
    },
    {
      "action_loss_avg": 0.0357708441093564,
      "epoch": 7.266998291015625e-06,
      "step": 1905
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 1905,
      "training_step_time": 10.239651679992676
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 3.1359989643096924,
      "step": 1905
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 1905,
      "training_step_time": 9.904822826385498
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 3.140717029571533,
      "step": 1905
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 1905,
      "training_step_time": 9.913562536239624
    },
    {
      "epoch": 7.266998291015625e-06,
      "model_forward_time": 3.132279396057129,
      "step": 1905
    },
    {
      "epoch": 7.266998291015625e-06,
      "step": 1905,
      "training_step_time": 10.302798986434937
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 2.467508316040039,
      "step": 1906
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 1906,
      "training_step_time": 7.500957250595093
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 3.398188352584839,
      "step": 1906
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 1906,
      "training_step_time": 10.167607069015503
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 3.134922742843628,
      "step": 1906
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 1906,
      "training_step_time": 9.90703797340393
    },
    {
      "epoch": 7.27081298828125e-06,
      "model_forward_time": 3.1320018768310547,
      "step": 1906
    },
    {
      "epoch": 7.27081298828125e-06,
      "step": 1906,
      "training_step_time": 10.274131298065186
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 3.153348684310913,
      "step": 1907
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 1907,
      "training_step_time": 9.922157764434814
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 3.13832426071167,
      "step": 1907
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 1907,
      "training_step_time": 9.909918308258057
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 2.4324584007263184,
      "step": 1907
    },
    {
      "dynamics_loss_avg": 0.06705906745046378,
      "epoch": 7.274627685546875e-06,
      "step": 1907
    },
    {
      "action_loss_avg": 0.039087447337806225,
      "epoch": 7.274627685546875e-06,
      "step": 1907
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 1907,
      "training_step_time": 7.465649127960205
    },
    {
      "epoch": 7.274627685546875e-06,
      "model_forward_time": 3.375849723815918,
      "step": 1907
    },
    {
      "epoch": 7.274627685546875e-06,
      "step": 1907,
      "training_step_time": 10.245794534683228
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 3.14983868598938,
      "step": 1908
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 1908,
      "training_step_time": 9.926519393920898
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 3.142075777053833,
      "step": 1908
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 1908,
      "training_step_time": 9.916285276412964
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 3.1425299644470215,
      "step": 1908
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 1908,
      "training_step_time": 9.915979623794556
    },
    {
      "epoch": 7.2784423828125e-06,
      "model_forward_time": 2.4374969005584717,
      "step": 1908
    },
    {
      "epoch": 7.2784423828125e-06,
      "step": 1908,
      "training_step_time": 10.23663854598999
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 3.1757781505584717,
      "step": 1909
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 1909,
      "training_step_time": 9.942310333251953
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 3.136681318283081,
      "step": 1909
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 1909,
      "training_step_time": 9.900885105133057
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 3.1370346546173096,
      "step": 1909
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 1909,
      "training_step_time": 9.904413223266602
    },
    {
      "epoch": 7.282257080078125e-06,
      "model_forward_time": 3.1438658237457275,
      "step": 1909
    },
    {
      "epoch": 7.282257080078125e-06,
      "step": 1909,
      "training_step_time": 10.284592151641846
    },
    {
      "epoch": 7.28607177734375e-06,
      "grad_norm": 0.5412471294403076,
      "learning_rate": 1e-05,
      "loss": 0.1082,
      "step": 1910
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 3.176424503326416,
      "step": 1910
    },
    {
      "dynamics_loss_avg": 0.06297013834118843,
      "epoch": 7.28607177734375e-06,
      "step": 1910
    },
    {
      "action_loss_avg": 0.03625170756131411,
      "epoch": 7.28607177734375e-06,
      "step": 1910
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 1910,
      "training_step_time": 10.250901937484741
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 3.1418581008911133,
      "step": 1910
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 1910,
      "training_step_time": 9.914503335952759
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 3.1389458179473877,
      "step": 1910
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 1910,
      "training_step_time": 9.9129159450531
    },
    {
      "epoch": 7.28607177734375e-06,
      "model_forward_time": 3.1443004608154297,
      "step": 1910
    },
    {
      "epoch": 7.28607177734375e-06,
      "step": 1910,
      "training_step_time": 10.278596878051758
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 3.1689155101776123,
      "step": 1911
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 1911,
      "training_step_time": 9.948545455932617
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 3.1408166885375977,
      "step": 1911
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 1911,
      "training_step_time": 10.196408987045288
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 3.145536422729492,
      "step": 1911
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 1911,
      "training_step_time": 9.924925565719604
    },
    {
      "epoch": 7.289886474609375e-06,
      "model_forward_time": 3.150369167327881,
      "step": 1911
    },
    {
      "epoch": 7.289886474609375e-06,
      "step": 1911,
      "training_step_time": 10.36558222770691
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 2.4550211429595947,
      "step": 1912
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 1912,
      "training_step_time": 7.490056276321411
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 3.149803400039673,
      "step": 1912
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 1912,
      "training_step_time": 9.928617477416992
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 3.147448778152466,
      "step": 1912
    },
    {
      "dynamics_loss_avg": 0.06226980071514845,
      "epoch": 7.293701171875e-06,
      "step": 1912
    },
    {
      "action_loss_avg": 0.043210283108055594,
      "epoch": 7.293701171875e-06,
      "step": 1912
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 1912,
      "training_step_time": 9.925457239151001
    },
    {
      "epoch": 7.293701171875e-06,
      "model_forward_time": 2.724865198135376,
      "step": 1912
    },
    {
      "epoch": 7.293701171875e-06,
      "step": 1912,
      "training_step_time": 10.340609312057495
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 3.1771466732025146,
      "step": 1913
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 1913,
      "training_step_time": 9.95496916770935
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 3.1411471366882324,
      "step": 1913
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 1913,
      "training_step_time": 9.913825035095215
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 3.140519618988037,
      "step": 1913
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 1913,
      "training_step_time": 9.915340662002563
    },
    {
      "epoch": 7.297515869140625e-06,
      "model_forward_time": 3.1394479274749756,
      "step": 1913
    },
    {
      "epoch": 7.297515869140625e-06,
      "step": 1913,
      "training_step_time": 10.011865854263306
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 3.484004259109497,
      "step": 1914
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 1914,
      "training_step_time": 10.25495982170105
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 3.134713649749756,
      "step": 1914
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 1914,
      "training_step_time": 9.906590700149536
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 2.43534255027771,
      "step": 1914
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 1914,
      "training_step_time": 7.4634082317352295
    },
    {
      "epoch": 7.30133056640625e-06,
      "model_forward_time": 3.136228322982788,
      "step": 1914
    },
    {
      "epoch": 7.30133056640625e-06,
      "step": 1914,
      "training_step_time": 10.01899766921997
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 3.1522979736328125,
      "step": 1915
    },
    {
      "dynamics_loss_avg": 0.054202431067824366,
      "epoch": 7.305145263671875e-06,
      "step": 1915
    },
    {
      "action_loss_avg": 0.03215151522308588,
      "epoch": 7.305145263671875e-06,
      "step": 1915
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 1915,
      "training_step_time": 10.206421852111816
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 3.1365318298339844,
      "step": 1915
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 1915,
      "training_step_time": 9.912925004959106
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 2.4382741451263428,
      "step": 1915
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 1915,
      "training_step_time": 7.471798896789551
    },
    {
      "epoch": 7.305145263671875e-06,
      "model_forward_time": 3.1397314071655273,
      "step": 1915
    },
    {
      "epoch": 7.305145263671875e-06,
      "step": 1915,
      "training_step_time": 10.271276473999023
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 3.1906638145446777,
      "step": 1916
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 1916,
      "training_step_time": 9.976421356201172
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 3.142834186553955,
      "step": 1916
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 1916,
      "training_step_time": 9.914374589920044
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 3.4584357738494873,
      "step": 1916
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 1916,
      "training_step_time": 10.23369288444519
    },
    {
      "epoch": 7.3089599609375e-06,
      "model_forward_time": 3.1465678215026855,
      "step": 1916
    },
    {
      "epoch": 7.3089599609375e-06,
      "step": 1916,
      "training_step_time": 10.297170162200928
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 3.1926212310791016,
      "step": 1917
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 1917,
      "training_step_time": 9.968758821487427
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 3.142411470413208,
      "step": 1917
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 1917,
      "training_step_time": 9.913251161575317
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 3.435507297515869,
      "step": 1917
    },
    {
      "dynamics_loss_avg": 0.07074929289519787,
      "epoch": 7.312774658203125e-06,
      "step": 1917
    },
    {
      "action_loss_avg": 0.05486574675887823,
      "epoch": 7.312774658203125e-06,
      "step": 1917
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 1917,
      "training_step_time": 10.209223985671997
    },
    {
      "epoch": 7.312774658203125e-06,
      "model_forward_time": 3.143369197845459,
      "step": 1917
    },
    {
      "epoch": 7.312774658203125e-06,
      "step": 1917,
      "training_step_time": 10.298432111740112
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 3.176629066467285,
      "step": 1918
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 1918,
      "training_step_time": 9.951613664627075
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 3.142646074295044,
      "step": 1918
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 1918,
      "training_step_time": 9.920265913009644
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 3.138749599456787,
      "step": 1918
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 1918,
      "training_step_time": 9.91335940361023
    },
    {
      "epoch": 7.31658935546875e-06,
      "model_forward_time": 3.1437153816223145,
      "step": 1918
    },
    {
      "epoch": 7.31658935546875e-06,
      "step": 1918,
      "training_step_time": 10.021820545196533
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 3.46243953704834,
      "step": 1919
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 1919,
      "training_step_time": 10.23900032043457
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 2.4394350051879883,
      "step": 1919
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 1919,
      "training_step_time": 7.466189861297607
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 3.1397719383239746,
      "step": 1919
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 1919,
      "training_step_time": 9.913132905960083
    },
    {
      "epoch": 7.320404052734375e-06,
      "model_forward_time": 3.146880865097046,
      "step": 1919
    },
    {
      "epoch": 7.320404052734375e-06,
      "step": 1919,
      "training_step_time": 10.039222002029419
    },
    {
      "epoch": 7.32421875e-06,
      "grad_norm": 0.4362439215183258,
      "learning_rate": 1e-05,
      "loss": 0.1052,
      "step": 1920
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 3.4947659969329834,
      "step": 1920
    },
    {
      "dynamics_loss_avg": 0.0458205315284431,
      "epoch": 7.32421875e-06,
      "step": 1920
    },
    {
      "action_loss_avg": 0.022796647809445857,
      "epoch": 7.32421875e-06,
      "step": 1920
    },
    {
      "epoch": 7.32421875e-06,
      "step": 1920,
      "training_step_time": 10.266976833343506
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 3.1454293727874756,
      "step": 1920
    },
    {
      "epoch": 7.32421875e-06,
      "step": 1920,
      "training_step_time": 9.916343927383423
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 3.1365935802459717,
      "step": 1920
    },
    {
      "epoch": 7.32421875e-06,
      "step": 1920,
      "training_step_time": 9.913370847702026
    },
    {
      "epoch": 7.32421875e-06,
      "model_forward_time": 3.1424148082733154,
      "step": 1920
    },
    {
      "epoch": 7.32421875e-06,
      "step": 1920,
      "training_step_time": 10.293979406356812
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 3.1822352409362793,
      "step": 1921
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 1921,
      "training_step_time": 9.954347133636475
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 3.1367156505584717,
      "step": 1921
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 1921,
      "training_step_time": 10.209697723388672
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 3.137748956680298,
      "step": 1921
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 1921,
      "training_step_time": 9.904344320297241
    },
    {
      "epoch": 7.328033447265625e-06,
      "model_forward_time": 3.140782117843628,
      "step": 1921
    },
    {
      "epoch": 7.328033447265625e-06,
      "step": 1921,
      "training_step_time": 10.27377700805664
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 3.1586549282073975,
      "step": 1922
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 1922,
      "training_step_time": 9.927562236785889
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 3.1365745067596436,
      "step": 1922
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 1922,
      "training_step_time": 9.906147480010986
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 3.1348462104797363,
      "step": 1922
    },
    {
      "dynamics_loss_avg": 0.06797687616199255,
      "epoch": 7.33184814453125e-06,
      "step": 1922
    },
    {
      "action_loss_avg": 0.04904385600239038,
      "epoch": 7.33184814453125e-06,
      "step": 1922
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 1922,
      "training_step_time": 10.18536376953125
    },
    {
      "epoch": 7.33184814453125e-06,
      "model_forward_time": 2.4354162216186523,
      "step": 1922
    },
    {
      "epoch": 7.33184814453125e-06,
      "step": 1922,
      "training_step_time": 10.307352781295776
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 3.1856284141540527,
      "step": 1923
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 1923,
      "training_step_time": 9.955442190170288
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 3.146744966506958,
      "step": 1923
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 1923,
      "training_step_time": 9.916113138198853
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 3.1491546630859375,
      "step": 1923
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 1923,
      "training_step_time": 9.917618036270142
    },
    {
      "epoch": 7.335662841796875e-06,
      "model_forward_time": 3.1359703540802,
      "step": 1923
    },
    {
      "epoch": 7.335662841796875e-06,
      "step": 1923,
      "training_step_time": 10.298036336898804
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 2.473936080932617,
      "step": 1924
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 1924,
      "training_step_time": 7.498544931411743
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 3.133857250213623,
      "step": 1924
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 1924,
      "training_step_time": 9.907564401626587
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 3.1378886699676514,
      "step": 1924
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 1924,
      "training_step_time": 9.906927585601807
    },
    {
      "epoch": 7.3394775390625e-06,
      "model_forward_time": 3.13413667678833,
      "step": 1924
    },
    {
      "epoch": 7.3394775390625e-06,
      "step": 1924,
      "training_step_time": 10.01569390296936
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 3.1401195526123047,
      "step": 1925
    },
    {
      "dynamics_loss_avg": 0.061001515574753286,
      "epoch": 7.343292236328125e-06,
      "step": 1925
    },
    {
      "action_loss_avg": 0.03659581560641527,
      "epoch": 7.343292236328125e-06,
      "step": 1925
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 1925,
      "training_step_time": 10.18669605255127
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 2.4312820434570312,
      "step": 1925
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 1925,
      "training_step_time": 7.457874774932861
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 3.1338257789611816,
      "step": 1925
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 1925,
      "training_step_time": 9.906703233718872
    },
    {
      "epoch": 7.343292236328125e-06,
      "model_forward_time": 2.4326705932617188,
      "step": 1925
    },
    {
      "epoch": 7.343292236328125e-06,
      "step": 1925,
      "training_step_time": 10.03123164176941
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 3.160841464996338,
      "step": 1926
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 1926,
      "training_step_time": 9.930980443954468
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 3.3972535133361816,
      "step": 1926
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 1926,
      "training_step_time": 10.168853521347046
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 3.137998342514038,
      "step": 1926
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 1926,
      "training_step_time": 9.90854549407959
    },
    {
      "epoch": 7.34710693359375e-06,
      "model_forward_time": 3.135998487472534,
      "step": 1926
    },
    {
      "epoch": 7.34710693359375e-06,
      "step": 1926,
      "training_step_time": 10.289516925811768
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 3.1445398330688477,
      "step": 1927
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 1927,
      "training_step_time": 9.923818349838257
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 3.13643217086792,
      "step": 1927
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 1927,
      "training_step_time": 9.913918733596802
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 2.431065797805786,
      "step": 1927
    },
    {
      "dynamics_loss_avg": 0.08212280310690404,
      "epoch": 7.350921630859375e-06,
      "step": 1927
    },
    {
      "action_loss_avg": 0.04501263424754143,
      "epoch": 7.350921630859375e-06,
      "step": 1927
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 1927,
      "training_step_time": 7.469895362854004
    },
    {
      "epoch": 7.350921630859375e-06,
      "model_forward_time": 3.4182960987091064,
      "step": 1927
    },
    {
      "epoch": 7.350921630859375e-06,
      "step": 1927,
      "training_step_time": 10.320336103439331
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 3.1484973430633545,
      "step": 1928
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 1928,
      "training_step_time": 9.931125402450562
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 3.13838791847229,
      "step": 1928
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 1928,
      "training_step_time": 9.91372299194336
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 3.132223606109619,
      "step": 1928
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 1928,
      "training_step_time": 9.905116319656372
    },
    {
      "epoch": 7.354736328125e-06,
      "model_forward_time": 2.4282915592193604,
      "step": 1928
    },
    {
      "epoch": 7.354736328125e-06,
      "step": 1928,
      "training_step_time": 10.019556522369385
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 3.162761926651001,
      "step": 1929
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 1929,
      "training_step_time": 9.931009769439697
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 3.1327407360076904,
      "step": 1929
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 1929,
      "training_step_time": 9.901298761367798
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 3.1314826011657715,
      "step": 1929
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 1929,
      "training_step_time": 9.90254521369934
    },
    {
      "epoch": 7.358551025390625e-06,
      "model_forward_time": 3.132101535797119,
      "step": 1929
    },
    {
      "epoch": 7.358551025390625e-06,
      "step": 1929,
      "training_step_time": 10.054635286331177
    },
    {
      "epoch": 7.36236572265625e-06,
      "grad_norm": 0.5017653107643127,
      "learning_rate": 1e-05,
      "loss": 0.11,
      "step": 1930
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 3.1459529399871826,
      "step": 1930
    },
    {
      "dynamics_loss_avg": 0.06484887870028615,
      "epoch": 7.36236572265625e-06,
      "step": 1930
    },
    {
      "action_loss_avg": 0.03553573181852698,
      "epoch": 7.36236572265625e-06,
      "step": 1930
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 1930,
      "training_step_time": 10.20204210281372
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 2.4342355728149414,
      "step": 1930
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 1930,
      "training_step_time": 7.462357759475708
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 3.1319005489349365,
      "step": 1930
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 1930,
      "training_step_time": 9.908272981643677
    },
    {
      "epoch": 7.36236572265625e-06,
      "model_forward_time": 2.4340555667877197,
      "step": 1930
    },
    {
      "epoch": 7.36236572265625e-06,
      "step": 1930,
      "training_step_time": 10.28879165649414
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 3.1807658672332764,
      "step": 1931
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 1931,
      "training_step_time": 9.953209638595581
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 3.1309337615966797,
      "step": 1931
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 1931,
      "training_step_time": 9.896638631820679
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 3.3763885498046875,
      "step": 1931
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 1931,
      "training_step_time": 10.14931607246399
    },
    {
      "epoch": 7.366180419921875e-06,
      "model_forward_time": 3.1367855072021484,
      "step": 1931
    },
    {
      "epoch": 7.366180419921875e-06,
      "step": 1931,
      "training_step_time": 10.31876802444458
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 2.475419521331787,
      "step": 1932
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 1932,
      "training_step_time": 7.501826763153076
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 3.1279807090759277,
      "step": 1932
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 1932,
      "training_step_time": 9.892049074172974
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 3.1314220428466797,
      "step": 1932
    },
    {
      "dynamics_loss_avg": 0.060895558632910254,
      "epoch": 7.3699951171875e-06,
      "step": 1932
    },
    {
      "action_loss_avg": 0.043578993529081345,
      "epoch": 7.3699951171875e-06,
      "step": 1932
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 1932,
      "training_step_time": 9.897076845169067
    },
    {
      "epoch": 7.3699951171875e-06,
      "model_forward_time": 3.438663959503174,
      "step": 1932
    },
    {
      "epoch": 7.3699951171875e-06,
      "step": 1932,
      "training_step_time": 10.307743549346924
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 3.164109230041504,
      "step": 1933
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 1933,
      "training_step_time": 9.931880235671997
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 3.1335556507110596,
      "step": 1933
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 1933,
      "training_step_time": 9.908543109893799
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 3.135326862335205,
      "step": 1933
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 1933,
      "training_step_time": 9.905882358551025
    },
    {
      "epoch": 7.373809814453125e-06,
      "model_forward_time": 3.4133706092834473,
      "step": 1933
    },
    {
      "epoch": 7.373809814453125e-06,
      "step": 1933,
      "training_step_time": 10.276714086532593
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 3.174868583679199,
      "step": 1934
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 1934,
      "training_step_time": 9.945188045501709
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 3.1340079307556152,
      "step": 1934
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 1934,
      "training_step_time": 9.909596681594849
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 3.139648914337158,
      "step": 1934
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 1934,
      "training_step_time": 9.91614055633545
    },
    {
      "epoch": 7.37762451171875e-06,
      "model_forward_time": 3.141526937484741,
      "step": 1934
    },
    {
      "epoch": 7.37762451171875e-06,
      "step": 1934,
      "training_step_time": 10.026023626327515
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 3.4548592567443848,
      "step": 1935
    },
    {
      "dynamics_loss_avg": 0.06082500275224447,
      "epoch": 7.381439208984375e-06,
      "step": 1935
    },
    {
      "action_loss_avg": 0.04136576578021049,
      "epoch": 7.381439208984375e-06,
      "step": 1935
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 1935,
      "training_step_time": 10.231672286987305
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 2.439236879348755,
      "step": 1935
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 1935,
      "training_step_time": 7.466768503189087
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 3.13862943649292,
      "step": 1935
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 1935,
      "training_step_time": 9.9132080078125
    },
    {
      "epoch": 7.381439208984375e-06,
      "model_forward_time": 3.1422204971313477,
      "step": 1935
    },
    {
      "epoch": 7.381439208984375e-06,
      "step": 1935,
      "training_step_time": 10.029919624328613
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 3.183570623397827,
      "step": 1936
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 1936,
      "training_step_time": 9.96101427078247
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 3.1393396854400635,
      "step": 1936
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 1936,
      "training_step_time": 10.228749752044678
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 3.139214515686035,
      "step": 1936
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 1936,
      "training_step_time": 9.910657405853271
    },
    {
      "epoch": 7.38525390625e-06,
      "model_forward_time": 3.135972023010254,
      "step": 1936
    },
    {
      "epoch": 7.38525390625e-06,
      "step": 1936,
      "training_step_time": 10.232292175292969
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 3.17781400680542,
      "step": 1937
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 1937,
      "training_step_time": 9.94467806816101
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 3.1325221061706543,
      "step": 1937
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 1937,
      "training_step_time": 9.903234958648682
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 3.4132323265075684,
      "step": 1937
    },
    {
      "dynamics_loss_avg": 0.07047995701432228,
      "epoch": 7.389068603515625e-06,
      "step": 1937
    },
    {
      "action_loss_avg": 0.04107131715863943,
      "epoch": 7.389068603515625e-06,
      "step": 1937
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 1937,
      "training_step_time": 10.18728494644165
    },
    {
      "epoch": 7.389068603515625e-06,
      "model_forward_time": 3.1360859870910645,
      "step": 1937
    },
    {
      "epoch": 7.389068603515625e-06,
      "step": 1937,
      "training_step_time": 10.296711921691895
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 3.192326784133911,
      "step": 1938
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 1938,
      "training_step_time": 9.967970132827759
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 3.138643980026245,
      "step": 1938
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 1938,
      "training_step_time": 9.91083812713623
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 3.1378695964813232,
      "step": 1938
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 1938,
      "training_step_time": 9.90984034538269
    },
    {
      "epoch": 7.39288330078125e-06,
      "model_forward_time": 3.4038007259368896,
      "step": 1938
    },
    {
      "epoch": 7.39288330078125e-06,
      "step": 1938,
      "training_step_time": 10.294849872589111
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 2.4797303676605225,
      "step": 1939
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 1939,
      "training_step_time": 7.510163307189941
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 3.135584831237793,
      "step": 1939
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 1939,
      "training_step_time": 9.907262563705444
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 3.1341397762298584,
      "step": 1939
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 1939,
      "training_step_time": 9.90265679359436
    },
    {
      "epoch": 7.396697998046875e-06,
      "model_forward_time": 3.137049436569214,
      "step": 1939
    },
    {
      "epoch": 7.396697998046875e-06,
      "step": 1939,
      "training_step_time": 10.033864974975586
    },
    {
      "epoch": 7.4005126953125e-06,
      "grad_norm": 0.48751264810562134,
      "learning_rate": 1e-05,
      "loss": 0.1075,
      "step": 1940
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 3.4535820484161377,
      "step": 1940
    },
    {
      "dynamics_loss_avg": 0.06462164130061865,
      "epoch": 7.4005126953125e-06,
      "step": 1940
    },
    {
      "action_loss_avg": 0.04435341777279973,
      "epoch": 7.4005126953125e-06,
      "step": 1940
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 1940,
      "training_step_time": 10.230058431625366
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 3.1387689113616943,
      "step": 1940
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 1940,
      "training_step_time": 9.910966157913208
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 3.137291193008423,
      "step": 1940
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 1940,
      "training_step_time": 9.905397415161133
    },
    {
      "epoch": 7.4005126953125e-06,
      "model_forward_time": 3.1379528045654297,
      "step": 1940
    },
    {
      "epoch": 7.4005126953125e-06,
      "step": 1940,
      "training_step_time": 10.023026704788208
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 3.1970622539520264,
      "step": 1941
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 1941,
      "training_step_time": 9.970576047897339
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 3.1340115070343018,
      "step": 1941
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 1941,
      "training_step_time": 10.188538074493408
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 3.132589340209961,
      "step": 1941
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 1941,
      "training_step_time": 9.905273675918579
    },
    {
      "epoch": 7.404327392578125e-06,
      "model_forward_time": 3.139183521270752,
      "step": 1941
    },
    {
      "epoch": 7.404327392578125e-06,
      "step": 1941,
      "training_step_time": 10.277506589889526
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 3.1808104515075684,
      "step": 1942
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 1942,
      "training_step_time": 9.95120882987976
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 3.1307170391082764,
      "step": 1942
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 1942,
      "training_step_time": 9.900465488433838
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 3.1368017196655273,
      "step": 1942
    },
    {
      "dynamics_loss_avg": 0.062335750181227924,
      "epoch": 7.40814208984375e-06,
      "step": 1942
    },
    {
      "action_loss_avg": 0.03383032334968448,
      "epoch": 7.40814208984375e-06,
      "step": 1942
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 1942,
      "training_step_time": 10.18833613395691
    },
    {
      "epoch": 7.40814208984375e-06,
      "model_forward_time": 3.1440651416778564,
      "step": 1942
    },
    {
      "epoch": 7.40814208984375e-06,
      "step": 1942,
      "training_step_time": 10.333717107772827
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 3.175046920776367,
      "step": 1943
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 1943,
      "training_step_time": 9.954512596130371
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 2.4368436336517334,
      "step": 1943
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 1943,
      "training_step_time": 7.467644453048706
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 3.135965585708618,
      "step": 1943
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 1943,
      "training_step_time": 9.907429218292236
    },
    {
      "epoch": 7.411956787109375e-06,
      "model_forward_time": 3.1338937282562256,
      "step": 1943
    },
    {
      "epoch": 7.411956787109375e-06,
      "step": 1943,
      "training_step_time": 10.311094999313354
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 3.1706883907318115,
      "step": 1944
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 1944,
      "training_step_time": 9.955273866653442
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 3.1335606575012207,
      "step": 1944
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 1944,
      "training_step_time": 9.90750002861023
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 3.1389400959014893,
      "step": 1944
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 1944,
      "training_step_time": 9.910335063934326
    },
    {
      "epoch": 7.415771484375e-06,
      "model_forward_time": 3.1347315311431885,
      "step": 1944
    },
    {
      "epoch": 7.415771484375e-06,
      "step": 1944,
      "training_step_time": 10.014984607696533
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 3.178330659866333,
      "step": 1945
    },
    {
      "dynamics_loss_avg": 0.0632871825248003,
      "epoch": 7.419586181640625e-06,
      "step": 1945
    },
    {
      "action_loss_avg": 0.04007587898522615,
      "epoch": 7.419586181640625e-06,
      "step": 1945
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 1945,
      "training_step_time": 10.234118700027466
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 1.7443275451660156,
      "step": 1945
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 1945,
      "training_step_time": 5.195419788360596
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 3.136190891265869,
      "step": 1945
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 1945,
      "training_step_time": 9.900250911712646
    },
    {
      "epoch": 7.419586181640625e-06,
      "model_forward_time": 3.136519193649292,
      "step": 1945
    },
    {
      "epoch": 7.419586181640625e-06,
      "step": 1945,
      "training_step_time": 10.274721622467041
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 2.4472572803497314,
      "step": 1946
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 1946,
      "training_step_time": 7.465606451034546
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 3.133253812789917,
      "step": 1946
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 1946,
      "training_step_time": 9.892425060272217
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 2.682875394821167,
      "step": 1946
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 1946,
      "training_step_time": 7.7044103145599365
    },
    {
      "epoch": 7.42340087890625e-06,
      "model_forward_time": 3.130798101425171,
      "step": 1946
    },
    {
      "epoch": 7.42340087890625e-06,
      "step": 1946,
      "training_step_time": 10.269995927810669
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 3.1835615634918213,
      "step": 1947
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 1947,
      "training_step_time": 9.953209638595581
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 3.136474132537842,
      "step": 1947
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 1947,
      "training_step_time": 9.902211427688599
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 3.1323399543762207,
      "step": 1947
    },
    {
      "dynamics_loss_avg": 0.05648253001272678,
      "epoch": 7.427215576171875e-06,
      "step": 1947
    },
    {
      "action_loss_avg": 0.04123482778668404,
      "epoch": 7.427215576171875e-06,
      "step": 1947
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 1947,
      "training_step_time": 9.897638082504272
    },
    {
      "epoch": 7.427215576171875e-06,
      "model_forward_time": 2.7108778953552246,
      "step": 1947
    },
    {
      "epoch": 7.427215576171875e-06,
      "step": 1947,
      "training_step_time": 10.25170612335205
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 3.1540133953094482,
      "step": 1948
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 1948,
      "training_step_time": 9.911072254180908
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 3.130647897720337,
      "step": 1948
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 1948,
      "training_step_time": 9.890554189682007
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 3.1280760765075684,
      "step": 1948
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 1948,
      "training_step_time": 9.897416114807129
    },
    {
      "epoch": 7.4310302734375e-06,
      "model_forward_time": 3.131822109222412,
      "step": 1948
    },
    {
      "epoch": 7.4310302734375e-06,
      "step": 1948,
      "training_step_time": 10.0167555809021
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 2.7795424461364746,
      "step": 1949
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 1949,
      "training_step_time": 7.8058555126190186
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 2.429819345474243,
      "step": 1949
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 1949,
      "training_step_time": 7.457746505737305
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 3.130812168121338,
      "step": 1949
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 1949,
      "training_step_time": 9.899914026260376
    },
    {
      "epoch": 7.434844970703125e-06,
      "model_forward_time": 3.1359171867370605,
      "step": 1949
    },
    {
      "epoch": 7.434844970703125e-06,
      "step": 1949,
      "training_step_time": 10.272284984588623
    },
    {
      "epoch": 7.43865966796875e-06,
      "grad_norm": 1.0084041357040405,
      "learning_rate": 1e-05,
      "loss": 0.1106,
      "step": 1950
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 3.176862955093384,
      "step": 1950
    },
    {
      "dynamics_loss_avg": 0.07567964531481267,
      "epoch": 7.43865966796875e-06,
      "step": 1950
    },
    {
      "action_loss_avg": 0.04170372374355793,
      "epoch": 7.43865966796875e-06,
      "step": 1950
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 1950,
      "training_step_time": 9.969374418258667
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 3.419607400894165,
      "step": 1950
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 1950,
      "training_step_time": 10.196539402008057
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 3.1424269676208496,
      "step": 1950
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 1950,
      "training_step_time": 9.916188955307007
    },
    {
      "epoch": 7.43865966796875e-06,
      "model_forward_time": 3.148132085800171,
      "step": 1950
    },
    {
      "epoch": 7.43865966796875e-06,
      "step": 1950,
      "training_step_time": 10.02834439277649
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 3.1841647624969482,
      "step": 1951
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 1951,
      "training_step_time": 9.959868431091309
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 3.143738269805908,
      "step": 1951
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 1951,
      "training_step_time": 9.915105819702148
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 3.406393527984619,
      "step": 1951
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 1951,
      "training_step_time": 10.17993450164795
    },
    {
      "epoch": 7.442474365234375e-06,
      "model_forward_time": 3.1414215564727783,
      "step": 1951
    },
    {
      "epoch": 7.442474365234375e-06,
      "step": 1951,
      "training_step_time": 10.284592151641846
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 3.1759445667266846,
      "step": 1952
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 1952,
      "training_step_time": 9.939558744430542
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 3.1349966526031494,
      "step": 1952
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 1952,
      "training_step_time": 9.89337158203125
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 3.1318016052246094,
      "step": 1952
    },
    {
      "dynamics_loss_avg": 0.07662109695374966,
      "epoch": 7.4462890625e-06,
      "step": 1952
    },
    {
      "action_loss_avg": 0.052228010073304174,
      "epoch": 7.4462890625e-06,
      "step": 1952
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 1952,
      "training_step_time": 10.178113222122192
    },
    {
      "epoch": 7.4462890625e-06,
      "model_forward_time": 3.1377692222595215,
      "step": 1952
    },
    {
      "epoch": 7.4462890625e-06,
      "step": 1952,
      "training_step_time": 10.305219411849976
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 3.178229808807373,
      "step": 1953
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 1953,
      "training_step_time": 9.941604614257812
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 3.133467197418213,
      "step": 1953
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 1953,
      "training_step_time": 9.900858163833618
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 3.1374664306640625,
      "step": 1953
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 1953,
      "training_step_time": 9.90475583076477
    },
    {
      "epoch": 7.450103759765625e-06,
      "model_forward_time": 3.1320931911468506,
      "step": 1953
    },
    {
      "epoch": 7.450103759765625e-06,
      "step": 1953,
      "training_step_time": 10.324215412139893
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 3.1829264163970947,
      "step": 1954
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 1954,
      "training_step_time": 9.957984685897827
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 3.142439842224121,
      "step": 1954
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 1954,
      "training_step_time": 9.914297342300415
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 3.133824586868286,
      "step": 1954
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 1954,
      "training_step_time": 9.90213656425476
    },
    {
      "epoch": 7.45391845703125e-06,
      "model_forward_time": 2.4299488067626953,
      "step": 1954
    },
    {
      "epoch": 7.45391845703125e-06,
      "step": 1954,
      "training_step_time": 10.29462194442749
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 3.4396934509277344,
      "step": 1955
    },
    {
      "dynamics_loss_avg": 0.06380844041705132,
      "epoch": 7.457733154296875e-06,
      "step": 1955
    },
    {
      "action_loss_avg": 0.0364142756909132,
      "epoch": 7.457733154296875e-06,
      "step": 1955
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 1955,
      "training_step_time": 10.206205129623413
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 3.12650990486145,
      "step": 1955
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 1955,
      "training_step_time": 9.883793592453003
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 3.1298906803131104,
      "step": 1955
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 1955,
      "training_step_time": 9.892986059188843
    },
    {
      "epoch": 7.457733154296875e-06,
      "model_forward_time": 3.128971815109253,
      "step": 1955
    },
    {
      "epoch": 7.457733154296875e-06,
      "step": 1955,
      "training_step_time": 10.00579023361206
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 3.1643192768096924,
      "step": 1956
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 1956,
      "training_step_time": 9.926011800765991
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 3.1378579139709473,
      "step": 1956
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 1956,
      "training_step_time": 9.899502754211426
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 3.437615156173706,
      "step": 1956
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 1956,
      "training_step_time": 10.203782558441162
    },
    {
      "epoch": 7.4615478515625e-06,
      "model_forward_time": 3.1339404582977295,
      "step": 1956
    },
    {
      "epoch": 7.4615478515625e-06,
      "step": 1956,
      "training_step_time": 10.01078200340271
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 3.1628592014312744,
      "step": 1957
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 1957,
      "training_step_time": 9.929691791534424
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 3.138749837875366,
      "step": 1957
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 1957,
      "training_step_time": 9.911186695098877
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 2.4396796226501465,
      "step": 1957
    },
    {
      "dynamics_loss_avg": 0.06648539453744888,
      "epoch": 7.465362548828125e-06,
      "step": 1957
    },
    {
      "action_loss_avg": 0.041141916811466214,
      "epoch": 7.465362548828125e-06,
      "step": 1957
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 1957,
      "training_step_time": 7.761168956756592
    },
    {
      "epoch": 7.465362548828125e-06,
      "model_forward_time": 3.1442179679870605,
      "step": 1957
    },
    {
      "epoch": 7.465362548828125e-06,
      "step": 1957,
      "training_step_time": 10.312796354293823
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 3.189526081085205,
      "step": 1958
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 1958,
      "training_step_time": 9.96053671836853
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 3.1418569087982178,
      "step": 1958
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 1958,
      "training_step_time": 9.909393548965454
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 3.1359763145446777,
      "step": 1958
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 1958,
      "training_step_time": 9.902149677276611
    },
    {
      "epoch": 7.46917724609375e-06,
      "model_forward_time": 3.40584397315979,
      "step": 1958
    },
    {
      "epoch": 7.46917724609375e-06,
      "step": 1958,
      "training_step_time": 10.306780815124512
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 3.186575412750244,
      "step": 1959
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 1959,
      "training_step_time": 9.954522132873535
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 2.4318971633911133,
      "step": 1959
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 1959,
      "training_step_time": 7.457712888717651
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 2.4281809329986572,
      "step": 1959
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 1959,
      "training_step_time": 7.449954032897949
    },
    {
      "epoch": 7.472991943359375e-06,
      "model_forward_time": 3.12924861907959,
      "step": 1959
    },
    {
      "epoch": 7.472991943359375e-06,
      "step": 1959,
      "training_step_time": 10.036648273468018
    },
    {
      "epoch": 7.476806640625e-06,
      "grad_norm": 0.8627364635467529,
      "learning_rate": 1e-05,
      "loss": 0.1133,
      "step": 1960
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 3.1441140174865723,
      "step": 1960
    },
    {
      "dynamics_loss_avg": 0.07440043091773987,
      "epoch": 7.476806640625e-06,
      "step": 1960
    },
    {
      "action_loss_avg": 0.053526051342487335,
      "epoch": 7.476806640625e-06,
      "step": 1960
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 1960,
      "training_step_time": 9.91722583770752
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 3.420795202255249,
      "step": 1960
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 1960,
      "training_step_time": 10.194833993911743
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 3.1428370475769043,
      "step": 1960
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 1960,
      "training_step_time": 9.915897607803345
    },
    {
      "epoch": 7.476806640625e-06,
      "model_forward_time": 2.4398202896118164,
      "step": 1960
    },
    {
      "epoch": 7.476806640625e-06,
      "step": 1960,
      "training_step_time": 10.287221670150757
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 3.1644287109375,
      "step": 1961
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 1961,
      "training_step_time": 9.93165135383606
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 3.141847610473633,
      "step": 1961
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 1961,
      "training_step_time": 9.914753437042236
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 3.4320881366729736,
      "step": 1961
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 1961,
      "training_step_time": 10.19748854637146
    },
    {
      "epoch": 7.480621337890625e-06,
      "model_forward_time": 2.436140775680542,
      "step": 1961
    },
    {
      "epoch": 7.480621337890625e-06,
      "step": 1961,
      "training_step_time": 10.020319938659668
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 3.17113995552063,
      "step": 1962
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 1962,
      "training_step_time": 9.933797121047974
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 3.1350584030151367,
      "step": 1962
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 1962,
      "training_step_time": 9.896589040756226
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 3.1367087364196777,
      "step": 1962
    },
    {
      "dynamics_loss_avg": 0.07709107249975204,
      "epoch": 7.48443603515625e-06,
      "step": 1962
    },
    {
      "action_loss_avg": 0.030944830924272537,
      "epoch": 7.48443603515625e-06,
      "step": 1962
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 1962,
      "training_step_time": 9.907720804214478
    },
    {
      "epoch": 7.48443603515625e-06,
      "model_forward_time": 2.721479892730713,
      "step": 1962
    },
    {
      "epoch": 7.48443603515625e-06,
      "step": 1962,
      "training_step_time": 10.299036026000977
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 3.1784751415252686,
      "step": 1963
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 1963,
      "training_step_time": 9.948281526565552
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 3.1335110664367676,
      "step": 1963
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 1963,
      "training_step_time": 9.90354585647583
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 3.139343023300171,
      "step": 1963
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 1963,
      "training_step_time": 9.909548044204712
    },
    {
      "epoch": 7.488250732421875e-06,
      "model_forward_time": 3.1387813091278076,
      "step": 1963
    },
    {
      "epoch": 7.488250732421875e-06,
      "step": 1963,
      "training_step_time": 10.318778991699219
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 2.786086082458496,
      "step": 1964
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 1964,
      "training_step_time": 7.815309762954712
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 3.136688709259033,
      "step": 1964
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 1964,
      "training_step_time": 9.90546464920044
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 3.139728307723999,
      "step": 1964
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 1964,
      "training_step_time": 9.908164978027344
    },
    {
      "epoch": 7.4920654296875e-06,
      "model_forward_time": 3.137887477874756,
      "step": 1964
    },
    {
      "epoch": 7.4920654296875e-06,
      "step": 1964,
      "training_step_time": 10.024483442306519
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 3.1794159412384033,
      "step": 1965
    },
    {
      "dynamics_loss_avg": 0.09432502575218678,
      "epoch": 7.495880126953125e-06,
      "step": 1965
    },
    {
      "action_loss_avg": 0.04974943399429321,
      "epoch": 7.495880126953125e-06,
      "step": 1965
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 1965,
      "training_step_time": 9.951653480529785
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 3.414242744445801,
      "step": 1965
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 1965,
      "training_step_time": 10.18262004852295
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 3.14517879486084,
      "step": 1965
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 1965,
      "training_step_time": 9.91559886932373
    },
    {
      "epoch": 7.495880126953125e-06,
      "model_forward_time": 3.13988995552063,
      "step": 1965
    },
    {
      "epoch": 7.495880126953125e-06,
      "step": 1965,
      "training_step_time": 10.324594259262085
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 3.1830949783325195,
      "step": 1966
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 1966,
      "training_step_time": 9.953953266143799
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 3.1403353214263916,
      "step": 1966
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 1966,
      "training_step_time": 10.177384376525879
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 3.143505573272705,
      "step": 1966
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 1966,
      "training_step_time": 9.912603855133057
    },
    {
      "epoch": 7.49969482421875e-06,
      "model_forward_time": 3.138164520263672,
      "step": 1966
    },
    {
      "epoch": 7.49969482421875e-06,
      "step": 1966,
      "training_step_time": 10.024395227432251
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 3.1795921325683594,
      "step": 1967
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 1967,
      "training_step_time": 9.950703382492065
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 3.13187313079834,
      "step": 1967
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 1967,
      "training_step_time": 9.903552055358887
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 3.415266275405884,
      "step": 1967
    },
    {
      "dynamics_loss_avg": 0.06269293464720249,
      "epoch": 7.503509521484375e-06,
      "step": 1967
    },
    {
      "action_loss_avg": 0.042602133471518755,
      "epoch": 7.503509521484375e-06,
      "step": 1967
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 1967,
      "training_step_time": 10.187832832336426
    },
    {
      "epoch": 7.503509521484375e-06,
      "model_forward_time": 3.13415789604187,
      "step": 1967
    },
    {
      "epoch": 7.503509521484375e-06,
      "step": 1967,
      "training_step_time": 10.244468450546265
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 2.467508316040039,
      "step": 1968
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 1968,
      "training_step_time": 7.4968273639678955
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 3.1303324699401855,
      "step": 1968
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 1968,
      "training_step_time": 9.902374505996704
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 3.1343941688537598,
      "step": 1968
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 1968,
      "training_step_time": 9.907746076583862
    },
    {
      "epoch": 7.50732421875e-06,
      "model_forward_time": 3.1377365589141846,
      "step": 1968
    },
    {
      "epoch": 7.50732421875e-06,
      "step": 1968,
      "training_step_time": 10.318053722381592
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 3.1704559326171875,
      "step": 1969
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 1969,
      "training_step_time": 9.940980434417725
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 3.133840560913086,
      "step": 1969
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 1969,
      "training_step_time": 9.90269923210144
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 3.1321873664855957,
      "step": 1969
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 1969,
      "training_step_time": 9.901001930236816
    },
    {
      "epoch": 7.511138916015625e-06,
      "model_forward_time": 3.1398532390594482,
      "step": 1969
    },
    {
      "epoch": 7.511138916015625e-06,
      "step": 1969,
      "training_step_time": 10.010729789733887
    },
    {
      "epoch": 7.51495361328125e-06,
      "grad_norm": 0.6845282316207886,
      "learning_rate": 1e-05,
      "loss": 0.1169,
      "step": 1970
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 3.446416139602661,
      "step": 1970
    },
    {
      "dynamics_loss_avg": 0.08514448683708906,
      "epoch": 7.51495361328125e-06,
      "step": 1970
    },
    {
      "action_loss_avg": 0.04456613166257739,
      "epoch": 7.51495361328125e-06,
      "step": 1970
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 1970,
      "training_step_time": 10.220279693603516
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 3.147113800048828,
      "step": 1970
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 1970,
      "training_step_time": 9.923356533050537
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 2.4351162910461426,
      "step": 1970
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 1970,
      "training_step_time": 7.462503910064697
    },
    {
      "epoch": 7.51495361328125e-06,
      "model_forward_time": 2.4348440170288086,
      "step": 1970
    },
    {
      "epoch": 7.51495361328125e-06,
      "step": 1970,
      "training_step_time": 10.27206563949585
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 3.1773786544799805,
      "step": 1971
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 1971,
      "training_step_time": 9.95684003829956
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 3.1369452476501465,
      "step": 1971
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 1971,
      "training_step_time": 9.908859968185425
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 2.7019383907318115,
      "step": 1971
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 1971,
      "training_step_time": 7.722824335098267
    },
    {
      "epoch": 7.518768310546875e-06,
      "model_forward_time": 3.130770683288574,
      "step": 1971
    },
    {
      "epoch": 7.518768310546875e-06,
      "step": 1971,
      "training_step_time": 10.028492450714111
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 3.1794893741607666,
      "step": 1972
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 1972,
      "training_step_time": 9.946740865707397
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 3.1334779262542725,
      "step": 1972
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 1972,
      "training_step_time": 9.901928663253784
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 3.14158296585083,
      "step": 1972
    },
    {
      "dynamics_loss_avg": 0.06763590574264526,
      "epoch": 7.5225830078125e-06,
      "step": 1972
    },
    {
      "action_loss_avg": 0.03881386872380972,
      "epoch": 7.5225830078125e-06,
      "step": 1972
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 1972,
      "training_step_time": 9.908761978149414
    },
    {
      "epoch": 7.5225830078125e-06,
      "model_forward_time": 3.4609649181365967,
      "step": 1972
    },
    {
      "epoch": 7.5225830078125e-06,
      "step": 1972,
      "training_step_time": 10.322543621063232
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 3.177868127822876,
      "step": 1973
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 1973,
      "training_step_time": 9.950749635696411
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 3.1368484497070312,
      "step": 1973
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 1973,
      "training_step_time": 9.908729553222656
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 3.1410634517669678,
      "step": 1973
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 1973,
      "training_step_time": 9.913477420806885
    },
    {
      "epoch": 7.526397705078125e-06,
      "model_forward_time": 3.1390178203582764,
      "step": 1973
    },
    {
      "epoch": 7.526397705078125e-06,
      "step": 1973,
      "training_step_time": 10.30701231956482
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 3.488677501678467,
      "step": 1974
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 1974,
      "training_step_time": 10.257753372192383
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 3.1370155811309814,
      "step": 1974
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 1974,
      "training_step_time": 9.907389879226685
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 2.4319419860839844,
      "step": 1974
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 1974,
      "training_step_time": 7.459361553192139
    },
    {
      "epoch": 7.53021240234375e-06,
      "model_forward_time": 3.133829355239868,
      "step": 1974
    },
    {
      "epoch": 7.53021240234375e-06,
      "step": 1974,
      "training_step_time": 10.027269124984741
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 3.4620821475982666,
      "step": 1975
    },
    {
      "dynamics_loss_avg": 0.0610213091596961,
      "epoch": 7.534027099609375e-06,
      "step": 1975
    },
    {
      "action_loss_avg": 0.03810706976801157,
      "epoch": 7.534027099609375e-06,
      "step": 1975
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 1975,
      "training_step_time": 10.225197076797485
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 3.1361353397369385,
      "step": 1975
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 1975,
      "training_step_time": 9.901395320892334
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 3.1375741958618164,
      "step": 1975
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 1975,
      "training_step_time": 9.910866498947144
    },
    {
      "epoch": 7.534027099609375e-06,
      "model_forward_time": 3.1362297534942627,
      "step": 1975
    },
    {
      "epoch": 7.534027099609375e-06,
      "step": 1975,
      "training_step_time": 10.029744625091553
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 3.169708490371704,
      "step": 1976
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 1976,
      "training_step_time": 9.939751625061035
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 2.433995485305786,
      "step": 1976
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 1976,
      "training_step_time": 7.459001302719116
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 3.438192844390869,
      "step": 1976
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 1976,
      "training_step_time": 10.209404468536377
    },
    {
      "epoch": 7.537841796875e-06,
      "model_forward_time": 3.137618064880371,
      "step": 1976
    },
    {
      "epoch": 7.537841796875e-06,
      "step": 1976,
      "training_step_time": 10.33202576637268
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 2.4482438564300537,
      "step": 1977
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 1977,
      "training_step_time": 7.4741339683532715
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 2.4303858280181885,
      "step": 1977
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 1977,
      "training_step_time": 7.45142674446106
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 2.4303462505340576,
      "step": 1977
    },
    {
      "dynamics_loss_avg": 0.07710251808166504,
      "epoch": 7.541656494140625e-06,
      "step": 1977
    },
    {
      "action_loss_avg": 0.041910229809582235,
      "epoch": 7.541656494140625e-06,
      "step": 1977
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 1977,
      "training_step_time": 7.453872919082642
    },
    {
      "epoch": 7.541656494140625e-06,
      "model_forward_time": 2.708578586578369,
      "step": 1977
    },
    {
      "epoch": 7.541656494140625e-06,
      "step": 1977,
      "training_step_time": 10.243340969085693
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 3.1791365146636963,
      "step": 1978
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 1978,
      "training_step_time": 9.944473028182983
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 3.134248733520508,
      "step": 1978
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 1978,
      "training_step_time": 9.896197319030762
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 2.432039976119995,
      "step": 1978
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 1978,
      "training_step_time": 7.453466415405273
    },
    {
      "epoch": 7.54547119140625e-06,
      "model_forward_time": 3.135159492492676,
      "step": 1978
    },
    {
      "epoch": 7.54547119140625e-06,
      "step": 1978,
      "training_step_time": 10.305257081985474
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 3.4623641967773438,
      "step": 1979
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 1979,
      "training_step_time": 10.226369619369507
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 3.1374802589416504,
      "step": 1979
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 1979,
      "training_step_time": 9.899143695831299
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 3.1379377841949463,
      "step": 1979
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 1979,
      "training_step_time": 9.91006064414978
    },
    {
      "epoch": 7.549285888671875e-06,
      "model_forward_time": 3.1404192447662354,
      "step": 1979
    },
    {
      "epoch": 7.549285888671875e-06,
      "step": 1979,
      "training_step_time": 10.321382999420166
    },
    {
      "epoch": 7.5531005859375e-06,
      "grad_norm": 0.5005112886428833,
      "learning_rate": 1e-05,
      "loss": 0.1139,
      "step": 1980
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 3.180412769317627,
      "step": 1980
    },
    {
      "dynamics_loss_avg": 0.06062574572861194,
      "epoch": 7.5531005859375e-06,
      "step": 1980
    },
    {
      "action_loss_avg": 0.03090565325692296,
      "epoch": 7.5531005859375e-06,
      "step": 1980
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 1980,
      "training_step_time": 9.956624507904053
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 3.443908214569092,
      "step": 1980
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 1980,
      "training_step_time": 10.21318531036377
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 3.139042854309082,
      "step": 1980
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 1980,
      "training_step_time": 9.909771203994751
    },
    {
      "epoch": 7.5531005859375e-06,
      "model_forward_time": 3.1448073387145996,
      "step": 1980
    },
    {
      "epoch": 7.5531005859375e-06,
      "step": 1980,
      "training_step_time": 10.030648946762085
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 3.186464786529541,
      "step": 1981
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 1981,
      "training_step_time": 9.962834596633911
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 3.137878656387329,
      "step": 1981
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 1981,
      "training_step_time": 9.907387256622314
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 3.3891589641571045,
      "step": 1981
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 1981,
      "training_step_time": 10.160698652267456
    },
    {
      "epoch": 7.556915283203125e-06,
      "model_forward_time": 3.139787435531616,
      "step": 1981
    },
    {
      "epoch": 7.556915283203125e-06,
      "step": 1981,
      "training_step_time": 10.288127183914185
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 3.182279348373413,
      "step": 1982
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 1982,
      "training_step_time": 9.951314449310303
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 2.4315342903137207,
      "step": 1982
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 1982,
      "training_step_time": 7.458590984344482
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 3.1359078884124756,
      "step": 1982
    },
    {
      "dynamics_loss_avg": 0.07075678929686546,
      "epoch": 7.56072998046875e-06,
      "step": 1982
    },
    {
      "action_loss_avg": 0.05584732498973608,
      "epoch": 7.56072998046875e-06,
      "step": 1982
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 1982,
      "training_step_time": 9.906626224517822
    },
    {
      "epoch": 7.56072998046875e-06,
      "model_forward_time": 3.4041295051574707,
      "step": 1982
    },
    {
      "epoch": 7.56072998046875e-06,
      "step": 1982,
      "training_step_time": 10.262659311294556
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 2.469431161880493,
      "step": 1983
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 1983,
      "training_step_time": 7.490690469741821
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 2.4269635677337646,
      "step": 1983
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 1983,
      "training_step_time": 7.445091485977173
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 3.1309103965759277,
      "step": 1983
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 1983,
      "training_step_time": 9.891714096069336
    },
    {
      "epoch": 7.564544677734375e-06,
      "model_forward_time": 2.4373888969421387,
      "step": 1983
    },
    {
      "epoch": 7.564544677734375e-06,
      "step": 1983,
      "training_step_time": 10.299436330795288
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 3.4475831985473633,
      "step": 1984
    },
    {
      "epoch": 7.568359375e-06,
      "step": 1984,
      "training_step_time": 10.205698728561401
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 3.1294779777526855,
      "step": 1984
    },
    {
      "epoch": 7.568359375e-06,
      "step": 1984,
      "training_step_time": 9.890674114227295
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 3.1325223445892334,
      "step": 1984
    },
    {
      "epoch": 7.568359375e-06,
      "step": 1984,
      "training_step_time": 9.890626668930054
    },
    {
      "epoch": 7.568359375e-06,
      "model_forward_time": 3.1323366165161133,
      "step": 1984
    },
    {
      "epoch": 7.568359375e-06,
      "step": 1984,
      "training_step_time": 10.299349308013916
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 3.17504620552063,
      "step": 1985
    },
    {
      "dynamics_loss_avg": 0.06869953647255897,
      "epoch": 7.572174072265625e-06,
      "step": 1985
    },
    {
      "action_loss_avg": 0.03883314225822687,
      "epoch": 7.572174072265625e-06,
      "step": 1985
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 1985,
      "training_step_time": 9.941961288452148
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 2.717294454574585,
      "step": 1985
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 1985,
      "training_step_time": 7.737812757492065
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 2.431006908416748,
      "step": 1985
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 1985,
      "training_step_time": 7.4468607902526855
    },
    {
      "epoch": 7.572174072265625e-06,
      "model_forward_time": 3.129014015197754,
      "step": 1985
    },
    {
      "epoch": 7.572174072265625e-06,
      "step": 1985,
      "training_step_time": 10.02358365058899
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 3.1772921085357666,
      "step": 1986
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 1986,
      "training_step_time": 9.940815925598145
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 3.1338706016540527,
      "step": 1986
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 1986,
      "training_step_time": 9.898123502731323
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 3.412440538406372,
      "step": 1986
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 1986,
      "training_step_time": 10.180239915847778
    },
    {
      "epoch": 7.57598876953125e-06,
      "model_forward_time": 3.137054204940796,
      "step": 1986
    },
    {
      "epoch": 7.57598876953125e-06,
      "step": 1986,
      "training_step_time": 10.015861511230469
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 3.179913282394409,
      "step": 1987
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 1987,
      "training_step_time": 9.949799299240112
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 3.138521432876587,
      "step": 1987
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 1987,
      "training_step_time": 9.908421277999878
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 3.1360905170440674,
      "step": 1987
    },
    {
      "dynamics_loss_avg": 0.07199962511658668,
      "epoch": 7.579803466796875e-06,
      "step": 1987
    },
    {
      "action_loss_avg": 0.03771194163709879,
      "epoch": 7.579803466796875e-06,
      "step": 1987
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 1987,
      "training_step_time": 9.91138768196106
    },
    {
      "epoch": 7.579803466796875e-06,
      "model_forward_time": 3.3947150707244873,
      "step": 1987
    },
    {
      "epoch": 7.579803466796875e-06,
      "step": 1987,
      "training_step_time": 10.323391199111938
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 3.159611940383911,
      "step": 1988
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 1988,
      "training_step_time": 9.932124614715576
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 3.1399385929107666,
      "step": 1988
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 1988,
      "training_step_time": 9.908563137054443
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 3.1407697200775146,
      "step": 1988
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 1988,
      "training_step_time": 9.90965723991394
    },
    {
      "epoch": 7.5836181640625e-06,
      "model_forward_time": 3.1443567276000977,
      "step": 1988
    },
    {
      "epoch": 7.5836181640625e-06,
      "step": 1988,
      "training_step_time": 10.279558420181274
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 3.4599854946136475,
      "step": 1989
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 1989,
      "training_step_time": 10.22550106048584
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 3.1373701095581055,
      "step": 1989
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 1989,
      "training_step_time": 9.90419626235962
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 3.135199785232544,
      "step": 1989
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 1989,
      "training_step_time": 9.903674125671387
    },
    {
      "epoch": 7.587432861328125e-06,
      "model_forward_time": 3.1384410858154297,
      "step": 1989
    },
    {
      "epoch": 7.587432861328125e-06,
      "step": 1989,
      "training_step_time": 10.329461336135864
    },
    {
      "epoch": 7.59124755859375e-06,
      "grad_norm": 0.37940606474876404,
      "learning_rate": 1e-05,
      "loss": 0.113,
      "step": 1990
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 3.188920259475708,
      "step": 1990
    },
    {
      "dynamics_loss_avg": 0.0657243587076664,
      "epoch": 7.59124755859375e-06,
      "step": 1990
    },
    {
      "action_loss_avg": 0.039688313379883766,
      "epoch": 7.59124755859375e-06,
      "step": 1990
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 1990,
      "training_step_time": 9.962812185287476
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 3.408013105392456,
      "step": 1990
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 1990,
      "training_step_time": 10.17821979522705
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 3.137054681777954,
      "step": 1990
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 1990,
      "training_step_time": 9.914253950119019
    },
    {
      "epoch": 7.59124755859375e-06,
      "model_forward_time": 3.1508514881134033,
      "step": 1990
    },
    {
      "epoch": 7.59124755859375e-06,
      "step": 1990,
      "training_step_time": 10.042622804641724
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 3.1744821071624756,
      "step": 1991
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 1991,
      "training_step_time": 9.944498300552368
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 3.1321725845336914,
      "step": 1991
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 1991,
      "training_step_time": 9.897081851959229
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 3.3946478366851807,
      "step": 1991
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 1991,
      "training_step_time": 10.156996488571167
    },
    {
      "epoch": 7.595062255859375e-06,
      "model_forward_time": 2.430213212966919,
      "step": 1991
    },
    {
      "epoch": 7.595062255859375e-06,
      "step": 1991,
      "training_step_time": 10.02188515663147
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 3.1400182247161865,
      "step": 1992
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 1992,
      "training_step_time": 9.906047105789185
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 3.1327896118164062,
      "step": 1992
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 1992,
      "training_step_time": 9.889587163925171
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 3.1350555419921875,
      "step": 1992
    },
    {
      "dynamics_loss_avg": 0.046749755181372164,
      "epoch": 7.598876953125e-06,
      "step": 1992
    },
    {
      "action_loss_avg": 0.030029961187392474,
      "epoch": 7.598876953125e-06,
      "step": 1992
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 1992,
      "training_step_time": 9.901377201080322
    },
    {
      "epoch": 7.598876953125e-06,
      "model_forward_time": 2.715716600418091,
      "step": 1992
    },
    {
      "epoch": 7.598876953125e-06,
      "step": 1992,
      "training_step_time": 10.249820232391357
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 3.1792662143707275,
      "step": 1993
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 1993,
      "training_step_time": 9.944336652755737
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 3.1291825771331787,
      "step": 1993
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 1993,
      "training_step_time": 9.889597654342651
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 3.1339266300201416,
      "step": 1993
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 1993,
      "training_step_time": 9.893706321716309
    },
    {
      "epoch": 7.602691650390625e-06,
      "model_forward_time": 3.1320981979370117,
      "step": 1993
    },
    {
      "epoch": 7.602691650390625e-06,
      "step": 1993,
      "training_step_time": 10.298075914382935
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 3.430927038192749,
      "step": 1994
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 1994,
      "training_step_time": 10.193102359771729
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 3.1298344135284424,
      "step": 1994
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 1994,
      "training_step_time": 9.894649982452393
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 3.1321377754211426,
      "step": 1994
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 1994,
      "training_step_time": 9.8964204788208
    },
    {
      "epoch": 7.60650634765625e-06,
      "model_forward_time": 3.131852865219116,
      "step": 1994
    },
    {
      "epoch": 7.60650634765625e-06,
      "step": 1994,
      "training_step_time": 10.309428453445435
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 3.179779291152954,
      "step": 1995
    },
    {
      "dynamics_loss_avg": 0.05992477014660835,
      "epoch": 7.610321044921875e-06,
      "step": 1995
    },
    {
      "action_loss_avg": 0.0440443170722574,
      "epoch": 7.610321044921875e-06,
      "step": 1995
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 1995,
      "training_step_time": 9.947220087051392
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 3.391263484954834,
      "step": 1995
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 1995,
      "training_step_time": 10.153832197189331
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 3.1332950592041016,
      "step": 1995
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 1995,
      "training_step_time": 9.896060943603516
    },
    {
      "epoch": 7.610321044921875e-06,
      "model_forward_time": 3.1303579807281494,
      "step": 1995
    },
    {
      "epoch": 7.610321044921875e-06,
      "step": 1995,
      "training_step_time": 10.0158531665802
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 3.1771187782287598,
      "step": 1996
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 1996,
      "training_step_time": 9.9473557472229
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 3.1357669830322266,
      "step": 1996
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 1996,
      "training_step_time": 9.90314269065857
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 3.4428606033325195,
      "step": 1996
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 1996,
      "training_step_time": 10.208396673202515
    },
    {
      "epoch": 7.6141357421875e-06,
      "model_forward_time": 3.1309187412261963,
      "step": 1996
    },
    {
      "epoch": 7.6141357421875e-06,
      "step": 1996,
      "training_step_time": 10.01836085319519
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 3.177286148071289,
      "step": 1997
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 1997,
      "training_step_time": 9.938318014144897
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 3.1356594562530518,
      "step": 1997
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 1997,
      "training_step_time": 9.904741287231445
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 3.133023738861084,
      "step": 1997
    },
    {
      "dynamics_loss_avg": 0.06775664612650871,
      "epoch": 7.617950439453125e-06,
      "step": 1997
    },
    {
      "action_loss_avg": 0.03842183202505112,
      "epoch": 7.617950439453125e-06,
      "step": 1997
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 1997,
      "training_step_time": 9.902362823486328
    },
    {
      "epoch": 7.617950439453125e-06,
      "model_forward_time": 3.4018378257751465,
      "step": 1997
    },
    {
      "epoch": 7.617950439453125e-06,
      "step": 1997,
      "training_step_time": 10.26546311378479
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 3.1820833683013916,
      "step": 1998
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 1998,
      "training_step_time": 9.949906587600708
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 3.137188673019409,
      "step": 1998
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 1998,
      "training_step_time": 9.900464296340942
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 2.4350528717041016,
      "step": 1998
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 1998,
      "training_step_time": 7.461202144622803
    },
    {
      "epoch": 7.62176513671875e-06,
      "model_forward_time": 3.1361260414123535,
      "step": 1998
    },
    {
      "epoch": 7.62176513671875e-06,
      "step": 1998,
      "training_step_time": 10.337588548660278
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 3.5465011596679688,
      "step": 1999
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 1999,
      "training_step_time": 10.324020385742188
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 3.1413064002990723,
      "step": 1999
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 1999,
      "training_step_time": 9.905054569244385
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 2.4317526817321777,
      "step": 1999
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 1999,
      "training_step_time": 7.454568862915039
    },
    {
      "epoch": 7.625579833984375e-06,
      "model_forward_time": 3.133528709411621,
      "step": 1999
    },
    {
      "epoch": 7.625579833984375e-06,
      "step": 1999,
      "training_step_time": 10.33756947517395
    },
    {
      "epoch": 7.62939453125e-06,
      "grad_norm": 1.0774054527282715,
      "learning_rate": 1e-05,
      "loss": 0.1092,
      "step": 2000
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 4.804356336593628,
      "step": 2000
    },
    {
      "dynamics_loss_avg": 0.07374848145991564,
      "epoch": 7.62939453125e-06,
      "step": 2000
    },
    {
      "action_loss_avg": 0.045245138742029664,
      "epoch": 7.62939453125e-06,
      "step": 2000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 2000,
      "training_step_time": 11.562549114227295
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 3.1201040744781494,
      "step": 2000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 2000,
      "training_step_time": 9.880467414855957
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 3.1259548664093018,
      "step": 2000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 2000,
      "training_step_time": 9.880257844924927
    },
    {
      "epoch": 7.62939453125e-06,
      "model_forward_time": 3.1342878341674805,
      "step": 2000
    },
    {
      "epoch": 7.62939453125e-06,
      "step": 2000,
      "training_step_time": 9.992728233337402
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 3.1897919178009033,
      "step": 2001
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 2001,
      "training_step_time": 9.961015224456787
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 3.133889675140381,
      "step": 2001
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 2001,
      "training_step_time": 10.204061031341553
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 3.1296985149383545,
      "step": 2001
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 2001,
      "training_step_time": 9.896121740341187
    },
    {
      "epoch": 7.633209228515625e-06,
      "model_forward_time": 3.135756015777588,
      "step": 2001
    },
    {
      "epoch": 7.633209228515625e-06,
      "step": 2001,
      "training_step_time": 10.011169195175171
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 3.1703176498413086,
      "step": 2002
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 2002,
      "training_step_time": 9.943982124328613
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 3.138150215148926,
      "step": 2002
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 2002,
      "training_step_time": 9.908447980880737
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 3.4009597301483154,
      "step": 2002
    },
    {
      "dynamics_loss_avg": 0.07459578439593315,
      "epoch": 7.63702392578125e-06,
      "step": 2002
    },
    {
      "action_loss_avg": 0.04329767134040594,
      "epoch": 7.63702392578125e-06,
      "step": 2002
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 2002,
      "training_step_time": 10.177351713180542
    },
    {
      "epoch": 7.63702392578125e-06,
      "model_forward_time": 3.1408486366271973,
      "step": 2002
    },
    {
      "epoch": 7.63702392578125e-06,
      "step": 2002,
      "training_step_time": 10.020283937454224
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 3.166248083114624,
      "step": 2003
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 2003,
      "training_step_time": 9.936603784561157
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 3.1402034759521484,
      "step": 2003
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 2003,
      "training_step_time": 9.920897483825684
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 3.14559268951416,
      "step": 2003
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 2003,
      "training_step_time": 9.922782182693481
    },
    {
      "epoch": 7.640838623046875e-06,
      "model_forward_time": 3.463955879211426,
      "step": 2003
    },
    {
      "epoch": 7.640838623046875e-06,
      "step": 2003,
      "training_step_time": 10.335165977478027
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 3.1802597045898438,
      "step": 2004
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 2004,
      "training_step_time": 9.957022666931152
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 3.1376776695251465,
      "step": 2004
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 2004,
      "training_step_time": 9.913479804992676
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 3.140817642211914,
      "step": 2004
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 2004,
      "training_step_time": 9.919171333312988
    },
    {
      "epoch": 7.6446533203125e-06,
      "model_forward_time": 3.1380615234375,
      "step": 2004
    },
    {
      "epoch": 7.6446533203125e-06,
      "step": 2004,
      "training_step_time": 10.315906047821045
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 3.1728951930999756,
      "step": 2005
    },
    {
      "dynamics_loss_avg": 0.08104609325528145,
      "epoch": 7.648468017578125e-06,
      "step": 2005
    },
    {
      "action_loss_avg": 0.0682732742279768,
      "epoch": 7.648468017578125e-06,
      "step": 2005
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 2005,
      "training_step_time": 10.236532211303711
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 3.1535451412200928,
      "step": 2005
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 2005,
      "training_step_time": 9.927266836166382
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 3.1407430171966553,
      "step": 2005
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 2005,
      "training_step_time": 9.913836479187012
    },
    {
      "epoch": 7.648468017578125e-06,
      "model_forward_time": 3.1419427394866943,
      "step": 2005
    },
    {
      "epoch": 7.648468017578125e-06,
      "step": 2005,
      "training_step_time": 10.277239561080933
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 3.1612534523010254,
      "step": 2006
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 2006,
      "training_step_time": 9.930219173431396
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 3.4309589862823486,
      "step": 2006
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 2006,
      "training_step_time": 10.198991775512695
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 3.1373608112335205,
      "step": 2006
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 2006,
      "training_step_time": 9.910226821899414
    },
    {
      "epoch": 7.65228271484375e-06,
      "model_forward_time": 3.1343979835510254,
      "step": 2006
    },
    {
      "epoch": 7.65228271484375e-06,
      "step": 2006,
      "training_step_time": 10.008827447891235
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 3.1831095218658447,
      "step": 2007
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 2007,
      "training_step_time": 9.952721357345581
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 3.1378605365753174,
      "step": 2007
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 2007,
      "training_step_time": 9.91056513786316
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 3.134744644165039,
      "step": 2007
    },
    {
      "dynamics_loss_avg": 0.055898102931678295,
      "epoch": 7.656097412109375e-06,
      "step": 2007
    },
    {
      "action_loss_avg": 0.03984706187620759,
      "epoch": 7.656097412109375e-06,
      "step": 2007
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 2007,
      "training_step_time": 10.180231094360352
    },
    {
      "epoch": 7.656097412109375e-06,
      "model_forward_time": 3.136622667312622,
      "step": 2007
    },
    {
      "epoch": 7.656097412109375e-06,
      "step": 2007,
      "training_step_time": 10.29672384262085
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 3.171792507171631,
      "step": 2008
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 2008,
      "training_step_time": 9.942553043365479
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 3.1385531425476074,
      "step": 2008
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 2008,
      "training_step_time": 9.904237508773804
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 3.135848045349121,
      "step": 2008
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 2008,
      "training_step_time": 9.907719612121582
    },
    {
      "epoch": 7.659912109375e-06,
      "model_forward_time": 3.3954598903656006,
      "step": 2008
    },
    {
      "epoch": 7.659912109375e-06,
      "step": 2008,
      "training_step_time": 10.304423093795776
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 2.4674324989318848,
      "step": 2009
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 2009,
      "training_step_time": 7.493751287460327
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 3.1373379230499268,
      "step": 2009
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 2009,
      "training_step_time": 9.908332347869873
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 3.1424455642700195,
      "step": 2009
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 2009,
      "training_step_time": 9.916366338729858
    },
    {
      "epoch": 7.663726806640625e-06,
      "model_forward_time": 3.1413187980651855,
      "step": 2009
    },
    {
      "epoch": 7.663726806640625e-06,
      "step": 2009,
      "training_step_time": 10.312990665435791
    },
    {
      "epoch": 7.66754150390625e-06,
      "grad_norm": 0.6306824684143066,
      "learning_rate": 1e-05,
      "loss": 0.1135,
      "step": 2010
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 3.1707656383514404,
      "step": 2010
    },
    {
      "dynamics_loss_avg": 0.0704189632087946,
      "epoch": 7.66754150390625e-06,
      "step": 2010
    },
    {
      "action_loss_avg": 0.05183864627033472,
      "epoch": 7.66754150390625e-06,
      "step": 2010
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 2010,
      "training_step_time": 9.944327592849731
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 3.441987991333008,
      "step": 2010
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 2010,
      "training_step_time": 10.210596799850464
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 3.143584728240967,
      "step": 2010
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 2010,
      "training_step_time": 9.915385246276855
    },
    {
      "epoch": 7.66754150390625e-06,
      "model_forward_time": 3.141437292098999,
      "step": 2010
    },
    {
      "epoch": 7.66754150390625e-06,
      "step": 2010,
      "training_step_time": 10.304580926895142
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 3.1785225868225098,
      "step": 2011
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 2011,
      "training_step_time": 9.947604179382324
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 3.41737699508667,
      "step": 2011
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 2011,
      "training_step_time": 10.186298847198486
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 3.1336557865142822,
      "step": 2011
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 2011,
      "training_step_time": 9.901190042495728
    },
    {
      "epoch": 7.671356201171875e-06,
      "model_forward_time": 3.136535882949829,
      "step": 2011
    },
    {
      "epoch": 7.671356201171875e-06,
      "step": 2011,
      "training_step_time": 10.247034788131714
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 3.1714327335357666,
      "step": 2012
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 2012,
      "training_step_time": 9.939032077789307
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 3.1335256099700928,
      "step": 2012
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 2012,
      "training_step_time": 9.903045415878296
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 3.390798568725586,
      "step": 2012
    },
    {
      "dynamics_loss_avg": 0.0819537241011858,
      "epoch": 7.6751708984375e-06,
      "step": 2012
    },
    {
      "action_loss_avg": 0.04631649982184172,
      "epoch": 7.6751708984375e-06,
      "step": 2012
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 2012,
      "training_step_time": 10.160881280899048
    },
    {
      "epoch": 7.6751708984375e-06,
      "model_forward_time": 3.137382745742798,
      "step": 2012
    },
    {
      "epoch": 7.6751708984375e-06,
      "step": 2012,
      "training_step_time": 10.281041145324707
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 3.1665496826171875,
      "step": 2013
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 2013,
      "training_step_time": 9.938513994216919
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 3.1362595558166504,
      "step": 2013
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 2013,
      "training_step_time": 9.906673431396484
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 3.134486675262451,
      "step": 2013
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 2013,
      "training_step_time": 9.902617931365967
    },
    {
      "epoch": 7.678985595703125e-06,
      "model_forward_time": 3.406968832015991,
      "step": 2013
    },
    {
      "epoch": 7.678985595703125e-06,
      "step": 2013,
      "training_step_time": 10.268069982528687
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 3.1674187183380127,
      "step": 2014
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 2014,
      "training_step_time": 9.933036804199219
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 3.13175630569458,
      "step": 2014
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 2014,
      "training_step_time": 9.900283336639404
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 3.13508677482605,
      "step": 2014
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 2014,
      "training_step_time": 9.905847787857056
    },
    {
      "epoch": 7.68280029296875e-06,
      "model_forward_time": 3.133882761001587,
      "step": 2014
    },
    {
      "epoch": 7.68280029296875e-06,
      "step": 2014,
      "training_step_time": 10.261409521102905
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 2.748178482055664,
      "step": 2015
    },
    {
      "dynamics_loss_avg": 0.06459303013980389,
      "epoch": 7.686614990234375e-06,
      "step": 2015
    },
    {
      "action_loss_avg": 0.0413825542666018,
      "epoch": 7.686614990234375e-06,
      "step": 2015
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 2015,
      "training_step_time": 7.778879165649414
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 3.1323561668395996,
      "step": 2015
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 2015,
      "training_step_time": 9.903333187103271
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 3.1355466842651367,
      "step": 2015
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 2015,
      "training_step_time": 9.905080318450928
    },
    {
      "epoch": 7.686614990234375e-06,
      "model_forward_time": 3.1357669830322266,
      "step": 2015
    },
    {
      "epoch": 7.686614990234375e-06,
      "step": 2015,
      "training_step_time": 10.291058778762817
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 3.15537691116333,
      "step": 2016
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 2016,
      "training_step_time": 9.9223952293396
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 3.133422374725342,
      "step": 2016
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 2016,
      "training_step_time": 10.170208930969238
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 3.1371114253997803,
      "step": 2016
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 2016,
      "training_step_time": 9.904012680053711
    },
    {
      "epoch": 7.6904296875e-06,
      "model_forward_time": 3.1310482025146484,
      "step": 2016
    },
    {
      "epoch": 7.6904296875e-06,
      "step": 2016,
      "training_step_time": 9.999587297439575
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 3.18135666847229,
      "step": 2017
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 2017,
      "training_step_time": 9.946053981781006
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 3.132200002670288,
      "step": 2017
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 2017,
      "training_step_time": 9.900104284286499
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 3.4469244480133057,
      "step": 2017
    },
    {
      "dynamics_loss_avg": 0.08219624534249306,
      "epoch": 7.694244384765625e-06,
      "step": 2017
    },
    {
      "action_loss_avg": 0.04434583764523268,
      "epoch": 7.694244384765625e-06,
      "step": 2017
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 2017,
      "training_step_time": 10.21210241317749
    },
    {
      "epoch": 7.694244384765625e-06,
      "model_forward_time": 3.1291091442108154,
      "step": 2017
    },
    {
      "epoch": 7.694244384765625e-06,
      "step": 2017,
      "training_step_time": 10.2792489528656
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 3.1585853099823,
      "step": 2018
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 2018,
      "training_step_time": 9.936010360717773
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 3.139019727706909,
      "step": 2018
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 2018,
      "training_step_time": 9.916479587554932
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 3.132995843887329,
      "step": 2018
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 2018,
      "training_step_time": 9.912032127380371
    },
    {
      "epoch": 7.69805908203125e-06,
      "model_forward_time": 3.1351521015167236,
      "step": 2018
    },
    {
      "epoch": 7.69805908203125e-06,
      "step": 2018,
      "training_step_time": 10.289222240447998
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 3.186389207839966,
      "step": 2019
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 2019,
      "training_step_time": 9.965502262115479
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 3.1372530460357666,
      "step": 2019
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 2019,
      "training_step_time": 9.916693687438965
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 3.13690185546875,
      "step": 2019
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 2019,
      "training_step_time": 9.916765689849854
    },
    {
      "epoch": 7.701873779296875e-06,
      "model_forward_time": 3.1346049308776855,
      "step": 2019
    },
    {
      "epoch": 7.701873779296875e-06,
      "step": 2019,
      "training_step_time": 10.015219688415527
    },
    {
      "epoch": 7.7056884765625e-06,
      "grad_norm": 0.5070356130599976,
      "learning_rate": 1e-05,
      "loss": 0.1129,
      "step": 2020
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 3.4233078956604004,
      "step": 2020
    },
    {
      "dynamics_loss_avg": 0.0526619428768754,
      "epoch": 7.7056884765625e-06,
      "step": 2020
    },
    {
      "action_loss_avg": 0.03453182931989431,
      "epoch": 7.7056884765625e-06,
      "step": 2020
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 2020,
      "training_step_time": 10.206104278564453
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 3.136885643005371,
      "step": 2020
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 2020,
      "training_step_time": 9.915730237960815
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 3.1337594985961914,
      "step": 2020
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 2020,
      "training_step_time": 9.913799285888672
    },
    {
      "epoch": 7.7056884765625e-06,
      "model_forward_time": 3.1351778507232666,
      "step": 2020
    },
    {
      "epoch": 7.7056884765625e-06,
      "step": 2020,
      "training_step_time": 10.278549909591675
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 3.188262462615967,
      "step": 2021
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 2021,
      "training_step_time": 9.964214086532593
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 3.4106884002685547,
      "step": 2021
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 2021,
      "training_step_time": 10.180484294891357
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 3.1302812099456787,
      "step": 2021
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 2021,
      "training_step_time": 9.90198826789856
    },
    {
      "epoch": 7.709503173828125e-06,
      "model_forward_time": 3.1322097778320312,
      "step": 2021
    },
    {
      "epoch": 7.709503173828125e-06,
      "step": 2021,
      "training_step_time": 10.291043281555176
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 3.1837425231933594,
      "step": 2022
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 2022,
      "training_step_time": 9.957354307174683
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 3.1310927867889404,
      "step": 2022
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 2022,
      "training_step_time": 9.905016899108887
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 3.382246732711792,
      "step": 2022
    },
    {
      "dynamics_loss_avg": 0.086885716766119,
      "epoch": 7.71331787109375e-06,
      "step": 2022
    },
    {
      "action_loss_avg": 0.058937886729836464,
      "epoch": 7.71331787109375e-06,
      "step": 2022
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 2022,
      "training_step_time": 10.158578634262085
    },
    {
      "epoch": 7.71331787109375e-06,
      "model_forward_time": 3.1358487606048584,
      "step": 2022
    },
    {
      "epoch": 7.71331787109375e-06,
      "step": 2022,
      "training_step_time": 10.30353045463562
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 3.18877911567688,
      "step": 2023
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 2023,
      "training_step_time": 9.964786529541016
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 3.1323394775390625,
      "step": 2023
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 2023,
      "training_step_time": 9.902735233306885
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 3.1354494094848633,
      "step": 2023
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 2023,
      "training_step_time": 9.913411140441895
    },
    {
      "epoch": 7.717132568359376e-06,
      "model_forward_time": 3.4086406230926514,
      "step": 2023
    },
    {
      "epoch": 7.717132568359376e-06,
      "step": 2023,
      "training_step_time": 10.27690052986145
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 3.1749069690704346,
      "step": 2024
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 2024,
      "training_step_time": 9.94812297821045
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 3.1366329193115234,
      "step": 2024
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 2024,
      "training_step_time": 9.910893201828003
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 3.1326465606689453,
      "step": 2024
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 2024,
      "training_step_time": 9.899259090423584
    },
    {
      "epoch": 7.720947265625e-06,
      "model_forward_time": 3.12949538230896,
      "step": 2024
    },
    {
      "epoch": 7.720947265625e-06,
      "step": 2024,
      "training_step_time": 10.272587299346924
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 3.1772141456604004,
      "step": 2025
    },
    {
      "dynamics_loss_avg": 0.05088377837091684,
      "epoch": 7.724761962890625e-06,
      "step": 2025
    },
    {
      "action_loss_avg": 0.03741978649049997,
      "epoch": 7.724761962890625e-06,
      "step": 2025
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 2025,
      "training_step_time": 10.239409685134888
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 3.1386911869049072,
      "step": 2025
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 2025,
      "training_step_time": 9.909214496612549
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 3.1396427154541016,
      "step": 2025
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 2025,
      "training_step_time": 9.911919832229614
    },
    {
      "epoch": 7.724761962890625e-06,
      "model_forward_time": 3.136178731918335,
      "step": 2025
    },
    {
      "epoch": 7.724761962890625e-06,
      "step": 2025,
      "training_step_time": 10.268561363220215
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 3.1720306873321533,
      "step": 2026
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 2026,
      "training_step_time": 9.942241191864014
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 3.4063751697540283,
      "step": 2026
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 2026,
      "training_step_time": 10.181752443313599
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 3.148683547973633,
      "step": 2026
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 2026,
      "training_step_time": 9.930810689926147
    },
    {
      "epoch": 7.72857666015625e-06,
      "model_forward_time": 3.139991521835327,
      "step": 2026
    },
    {
      "epoch": 7.72857666015625e-06,
      "step": 2026,
      "training_step_time": 10.308995246887207
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 3.1849441528320312,
      "step": 2027
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 2027,
      "training_step_time": 9.968050956726074
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 3.1465327739715576,
      "step": 2027
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 2027,
      "training_step_time": 9.923910140991211
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 2.4367170333862305,
      "step": 2027
    },
    {
      "dynamics_loss_avg": 0.07200362384319306,
      "epoch": 7.732391357421876e-06,
      "step": 2027
    },
    {
      "action_loss_avg": 0.05211946070194244,
      "epoch": 7.732391357421876e-06,
      "step": 2027
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 2027,
      "training_step_time": 7.736018180847168
    },
    {
      "epoch": 7.732391357421876e-06,
      "model_forward_time": 3.1398513317108154,
      "step": 2027
    },
    {
      "epoch": 7.732391357421876e-06,
      "step": 2027,
      "training_step_time": 10.287299156188965
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 3.171834707260132,
      "step": 2028
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 2028,
      "training_step_time": 9.954623937606812
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 3.141632556915283,
      "step": 2028
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 2028,
      "training_step_time": 9.907485723495483
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 3.1392688751220703,
      "step": 2028
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 2028,
      "training_step_time": 9.908798217773438
    },
    {
      "epoch": 7.7362060546875e-06,
      "model_forward_time": 3.400698184967041,
      "step": 2028
    },
    {
      "epoch": 7.7362060546875e-06,
      "step": 2028,
      "training_step_time": 10.26205039024353
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 3.154812812805176,
      "step": 2029
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 2029,
      "training_step_time": 9.925940036773682
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 3.14155912399292,
      "step": 2029
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 2029,
      "training_step_time": 9.910484552383423
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 2.4357552528381348,
      "step": 2029
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 2029,
      "training_step_time": 7.460090637207031
    },
    {
      "epoch": 7.740020751953125e-06,
      "model_forward_time": 3.1339728832244873,
      "step": 2029
    },
    {
      "epoch": 7.740020751953125e-06,
      "step": 2029,
      "training_step_time": 10.018007040023804
    },
    {
      "epoch": 7.74383544921875e-06,
      "grad_norm": 0.7926910519599915,
      "learning_rate": 1e-05,
      "loss": 0.117,
      "step": 2030
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 3.172919988632202,
      "step": 2030
    },
    {
      "dynamics_loss_avg": 0.06764244996011257,
      "epoch": 7.74383544921875e-06,
      "step": 2030
    },
    {
      "action_loss_avg": 0.03916089180856943,
      "epoch": 7.74383544921875e-06,
      "step": 2030
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 2030,
      "training_step_time": 10.254738807678223
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 3.139028787612915,
      "step": 2030
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 2030,
      "training_step_time": 9.907555103302002
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 3.140432834625244,
      "step": 2030
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 2030,
      "training_step_time": 9.912647247314453
    },
    {
      "epoch": 7.74383544921875e-06,
      "model_forward_time": 3.1389522552490234,
      "step": 2030
    },
    {
      "epoch": 7.74383544921875e-06,
      "step": 2030,
      "training_step_time": 10.031360626220703
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 3.1841318607330322,
      "step": 2031
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 2031,
      "training_step_time": 9.953864336013794
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 2.7140660285949707,
      "step": 2031
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 2031,
      "training_step_time": 7.7456581592559814
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 3.135171413421631,
      "step": 2031
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 2031,
      "training_step_time": 9.91099762916565
    },
    {
      "epoch": 7.747650146484376e-06,
      "model_forward_time": 3.1421878337860107,
      "step": 2031
    },
    {
      "epoch": 7.747650146484376e-06,
      "step": 2031,
      "training_step_time": 10.315611600875854
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 3.1613404750823975,
      "step": 2032
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 2032,
      "training_step_time": 9.935646057128906
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 3.1350767612457275,
      "step": 2032
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 2032,
      "training_step_time": 9.905887842178345
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 3.1363909244537354,
      "step": 2032
    },
    {
      "dynamics_loss_avg": 0.07145646642893552,
      "epoch": 7.75146484375e-06,
      "step": 2032
    },
    {
      "action_loss_avg": 0.050947395525872706,
      "epoch": 7.75146484375e-06,
      "step": 2032
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 2032,
      "training_step_time": 10.178218126296997
    },
    {
      "epoch": 7.75146484375e-06,
      "model_forward_time": 2.4349570274353027,
      "step": 2032
    },
    {
      "epoch": 7.75146484375e-06,
      "step": 2032,
      "training_step_time": 10.314706563949585
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 3.188267946243286,
      "step": 2033
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 2033,
      "training_step_time": 9.9559645652771
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 3.1310617923736572,
      "step": 2033
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 2033,
      "training_step_time": 9.90407681465149
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 3.1300907135009766,
      "step": 2033
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 2033,
      "training_step_time": 9.894361972808838
    },
    {
      "epoch": 7.755279541015625e-06,
      "model_forward_time": 3.1453166007995605,
      "step": 2033
    },
    {
      "epoch": 7.755279541015625e-06,
      "step": 2033,
      "training_step_time": 10.286009550094604
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 3.1913223266601562,
      "step": 2034
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 2034,
      "training_step_time": 9.95473051071167
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 3.137831449508667,
      "step": 2034
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 2034,
      "training_step_time": 9.89592170715332
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 3.1348116397857666,
      "step": 2034
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 2034,
      "training_step_time": 9.896767377853394
    },
    {
      "epoch": 7.75909423828125e-06,
      "model_forward_time": 3.1357412338256836,
      "step": 2034
    },
    {
      "epoch": 7.75909423828125e-06,
      "step": 2034,
      "training_step_time": 10.282629489898682
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 3.1770424842834473,
      "step": 2035
    },
    {
      "dynamics_loss_avg": 0.06445556096732616,
      "epoch": 7.762908935546876e-06,
      "step": 2035
    },
    {
      "action_loss_avg": 0.03879517931491137,
      "epoch": 7.762908935546876e-06,
      "step": 2035
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 2035,
      "training_step_time": 10.211026191711426
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 3.13879132270813,
      "step": 2035
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 2035,
      "training_step_time": 9.906352758407593
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 3.1439504623413086,
      "step": 2035
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 2035,
      "training_step_time": 9.915101289749146
    },
    {
      "epoch": 7.762908935546876e-06,
      "model_forward_time": 3.137094497680664,
      "step": 2035
    },
    {
      "epoch": 7.762908935546876e-06,
      "step": 2035,
      "training_step_time": 10.264506578445435
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 3.158601760864258,
      "step": 2036
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 2036,
      "training_step_time": 9.933306694030762
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 3.1414921283721924,
      "step": 2036
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 2036,
      "training_step_time": 10.181293964385986
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 3.1445729732513428,
      "step": 2036
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 2036,
      "training_step_time": 9.915327787399292
    },
    {
      "epoch": 7.7667236328125e-06,
      "model_forward_time": 2.4385759830474854,
      "step": 2036
    },
    {
      "epoch": 7.7667236328125e-06,
      "step": 2036,
      "training_step_time": 10.29644513130188
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 3.190044403076172,
      "step": 2037
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 2037,
      "training_step_time": 9.966849565505981
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 3.140512466430664,
      "step": 2037
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 2037,
      "training_step_time": 9.90746784210205
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 3.1365907192230225,
      "step": 2037
    },
    {
      "dynamics_loss_avg": 0.0766977671533823,
      "epoch": 7.770538330078125e-06,
      "step": 2037
    },
    {
      "action_loss_avg": 0.04654411692172289,
      "epoch": 7.770538330078125e-06,
      "step": 2037
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 2037,
      "training_step_time": 10.176016807556152
    },
    {
      "epoch": 7.770538330078125e-06,
      "model_forward_time": 3.137680768966675,
      "step": 2037
    },
    {
      "epoch": 7.770538330078125e-06,
      "step": 2037,
      "training_step_time": 10.27857518196106
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 3.183931589126587,
      "step": 2038
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 2038,
      "training_step_time": 9.95828127861023
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 3.1282191276550293,
      "step": 2038
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 2038,
      "training_step_time": 9.90160584449768
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 3.1331076622009277,
      "step": 2038
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 2038,
      "training_step_time": 9.908055782318115
    },
    {
      "epoch": 7.77435302734375e-06,
      "model_forward_time": 3.1378490924835205,
      "step": 2038
    },
    {
      "epoch": 7.77435302734375e-06,
      "step": 2038,
      "training_step_time": 10.290687799453735
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 3.1614086627960205,
      "step": 2039
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 2039,
      "training_step_time": 9.936402320861816
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 3.138124704360962,
      "step": 2039
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 2039,
      "training_step_time": 9.916111469268799
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 3.1392149925231934,
      "step": 2039
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 2039,
      "training_step_time": 9.920518636703491
    },
    {
      "epoch": 7.778167724609376e-06,
      "model_forward_time": 3.14003849029541,
      "step": 2039
    },
    {
      "epoch": 7.778167724609376e-06,
      "step": 2039,
      "training_step_time": 10.333990573883057
    },
    {
      "epoch": 7.781982421875e-06,
      "grad_norm": 0.6106740236282349,
      "learning_rate": 1e-05,
      "loss": 0.1166,
      "step": 2040
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 3.4175848960876465,
      "step": 2040
    },
    {
      "dynamics_loss_avg": 0.0752085343003273,
      "epoch": 7.781982421875e-06,
      "step": 2040
    },
    {
      "action_loss_avg": 0.04735154379159212,
      "epoch": 7.781982421875e-06,
      "step": 2040
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 2040,
      "training_step_time": 10.197463274002075
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 3.140150308609009,
      "step": 2040
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 2040,
      "training_step_time": 9.92133903503418
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 3.133061170578003,
      "step": 2040
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 2040,
      "training_step_time": 9.9046471118927
    },
    {
      "epoch": 7.781982421875e-06,
      "model_forward_time": 3.1335737705230713,
      "step": 2040
    },
    {
      "epoch": 7.781982421875e-06,
      "step": 2040,
      "training_step_time": 10.004593849182129
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 3.1721880435943604,
      "step": 2041
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 2041,
      "training_step_time": 9.944181680679321
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 3.422253370285034,
      "step": 2041
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 2041,
      "training_step_time": 10.194697141647339
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 3.133596420288086,
      "step": 2041
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 2041,
      "training_step_time": 9.90797758102417
    },
    {
      "epoch": 7.785797119140625e-06,
      "model_forward_time": 3.1378116607666016,
      "step": 2041
    },
    {
      "epoch": 7.785797119140625e-06,
      "step": 2041,
      "training_step_time": 10.272831439971924
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 3.1644225120544434,
      "step": 2042
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 2042,
      "training_step_time": 9.943721532821655
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 3.1339051723480225,
      "step": 2042
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 2042,
      "training_step_time": 9.910154104232788
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 3.406041383743286,
      "step": 2042
    },
    {
      "dynamics_loss_avg": 0.0761172391474247,
      "epoch": 7.78961181640625e-06,
      "step": 2042
    },
    {
      "action_loss_avg": 0.04068127516657114,
      "epoch": 7.78961181640625e-06,
      "step": 2042
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 2042,
      "training_step_time": 10.183458805084229
    },
    {
      "epoch": 7.78961181640625e-06,
      "model_forward_time": 3.1403989791870117,
      "step": 2042
    },
    {
      "epoch": 7.78961181640625e-06,
      "step": 2042,
      "training_step_time": 10.284162759780884
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 3.1741812229156494,
      "step": 2043
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 2043,
      "training_step_time": 9.954246520996094
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 3.1328651905059814,
      "step": 2043
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 2043,
      "training_step_time": 9.912172317504883
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 3.133526086807251,
      "step": 2043
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 2043,
      "training_step_time": 9.91049075126648
    },
    {
      "epoch": 7.793426513671876e-06,
      "model_forward_time": 3.420421600341797,
      "step": 2043
    },
    {
      "epoch": 7.793426513671876e-06,
      "step": 2043,
      "training_step_time": 10.292327404022217
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 3.1778204441070557,
      "step": 2044
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 2044,
      "training_step_time": 9.959559202194214
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 3.1328890323638916,
      "step": 2044
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 2044,
      "training_step_time": 9.91599988937378
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 3.14047908782959,
      "step": 2044
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 2044,
      "training_step_time": 9.925873756408691
    },
    {
      "epoch": 7.7972412109375e-06,
      "model_forward_time": 3.1388115882873535,
      "step": 2044
    },
    {
      "epoch": 7.7972412109375e-06,
      "step": 2044,
      "training_step_time": 10.282960653305054
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 3.184683322906494,
      "step": 2045
    },
    {
      "dynamics_loss_avg": 0.07820675186812878,
      "epoch": 7.801055908203125e-06,
      "step": 2045
    },
    {
      "action_loss_avg": 0.04217525143176317,
      "epoch": 7.801055908203125e-06,
      "step": 2045
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 2045,
      "training_step_time": 10.25307846069336
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 3.136715888977051,
      "step": 2045
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 2045,
      "training_step_time": 9.920321464538574
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 3.1410770416259766,
      "step": 2045
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 2045,
      "training_step_time": 9.920084476470947
    },
    {
      "epoch": 7.801055908203125e-06,
      "model_forward_time": 3.1401054859161377,
      "step": 2045
    },
    {
      "epoch": 7.801055908203125e-06,
      "step": 2045,
      "training_step_time": 10.303797960281372
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 3.1972806453704834,
      "step": 2046
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 2046,
      "training_step_time": 9.98167085647583
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 3.3935272693634033,
      "step": 2046
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 2046,
      "training_step_time": 10.169690132141113
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 3.1383767127990723,
      "step": 2046
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 2046,
      "training_step_time": 9.910583019256592
    },
    {
      "epoch": 7.80487060546875e-06,
      "model_forward_time": 3.132559299468994,
      "step": 2046
    },
    {
      "epoch": 7.80487060546875e-06,
      "step": 2046,
      "training_step_time": 10.012787818908691
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 3.1577725410461426,
      "step": 2047
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 2047,
      "training_step_time": 9.925802230834961
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 3.1350927352905273,
      "step": 2047
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 2047,
      "training_step_time": 9.906596183776855
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 3.1313207149505615,
      "step": 2047
    },
    {
      "dynamics_loss_avg": 0.06129601523280144,
      "epoch": 7.808685302734376e-06,
      "step": 2047
    },
    {
      "action_loss_avg": 0.03566990494728088,
      "epoch": 7.808685302734376e-06,
      "step": 2047
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 2047,
      "training_step_time": 10.174946069717407
    },
    {
      "epoch": 7.808685302734376e-06,
      "model_forward_time": 3.1354126930236816,
      "step": 2047
    },
    {
      "epoch": 7.808685302734376e-06,
      "step": 2047,
      "training_step_time": 10.300387382507324
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 3.1562001705169678,
      "step": 2048
    },
    {
      "epoch": 7.8125e-06,
      "step": 2048,
      "training_step_time": 9.926125288009644
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 3.1328346729278564,
      "step": 2048
    },
    {
      "epoch": 7.8125e-06,
      "step": 2048,
      "training_step_time": 9.899567127227783
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 3.1306488513946533,
      "step": 2048
    },
    {
      "epoch": 7.8125e-06,
      "step": 2048,
      "training_step_time": 9.900566577911377
    },
    {
      "epoch": 7.8125e-06,
      "model_forward_time": 3.4005367755889893,
      "step": 2048
    },
    {
      "epoch": 7.8125e-06,
      "step": 2048,
      "training_step_time": 10.290831804275513
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 3.1693265438079834,
      "step": 2049
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 2049,
      "training_step_time": 9.943551778793335
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 3.1352312564849854,
      "step": 2049
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 2049,
      "training_step_time": 9.91015362739563
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 3.1366920471191406,
      "step": 2049
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 2049,
      "training_step_time": 9.913296699523926
    },
    {
      "epoch": 7.816314697265625e-06,
      "model_forward_time": 3.139953851699829,
      "step": 2049
    },
    {
      "epoch": 7.816314697265625e-06,
      "step": 2049,
      "training_step_time": 10.023423671722412
    },
    {
      "epoch": 7.82012939453125e-06,
      "grad_norm": 0.6004961133003235,
      "learning_rate": 1e-05,
      "loss": 0.1121,
      "step": 2050
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 3.182084560394287,
      "step": 2050
    },
    {
      "dynamics_loss_avg": 0.06357934474945068,
      "epoch": 7.82012939453125e-06,
      "step": 2050
    },
    {
      "action_loss_avg": 0.03366546016186476,
      "epoch": 7.82012939453125e-06,
      "step": 2050
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 2050,
      "training_step_time": 10.228323459625244
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 3.139195442199707,
      "step": 2050
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 2050,
      "training_step_time": 9.916500329971313
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 3.1382689476013184,
      "step": 2050
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 2050,
      "training_step_time": 9.913150072097778
    },
    {
      "epoch": 7.82012939453125e-06,
      "model_forward_time": 3.1436474323272705,
      "step": 2050
    },
    {
      "epoch": 7.82012939453125e-06,
      "step": 2050,
      "training_step_time": 10.290939569473267
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 3.1736228466033936,
      "step": 2051
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 2051,
      "training_step_time": 9.958822250366211
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 3.42342209815979,
      "step": 2051
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 2051,
      "training_step_time": 10.198166370391846
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 3.1388766765594482,
      "step": 2051
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 2051,
      "training_step_time": 9.913302421569824
    },
    {
      "epoch": 7.823944091796876e-06,
      "model_forward_time": 3.1455488204956055,
      "step": 2051
    },
    {
      "epoch": 7.823944091796876e-06,
      "step": 2051,
      "training_step_time": 10.013378620147705
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 3.187870740890503,
      "step": 2052
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 2052,
      "training_step_time": 9.963402032852173
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 3.136998176574707,
      "step": 2052
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 2052,
      "training_step_time": 9.91398286819458
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 3.413299560546875,
      "step": 2052
    },
    {
      "dynamics_loss_avg": 0.08027861565351486,
      "epoch": 7.8277587890625e-06,
      "step": 2052
    },
    {
      "action_loss_avg": 0.05237418171018362,
      "epoch": 7.8277587890625e-06,
      "step": 2052
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 2052,
      "training_step_time": 10.193689346313477
    },
    {
      "epoch": 7.8277587890625e-06,
      "model_forward_time": 3.1457579135894775,
      "step": 2052
    },
    {
      "epoch": 7.8277587890625e-06,
      "step": 2052,
      "training_step_time": 10.025326251983643
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 3.178931951522827,
      "step": 2053
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 2053,
      "training_step_time": 9.95270037651062
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 3.1334481239318848,
      "step": 2053
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 2053,
      "training_step_time": 9.911133766174316
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 3.1361913681030273,
      "step": 2053
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 2053,
      "training_step_time": 9.909083604812622
    },
    {
      "epoch": 7.831573486328125e-06,
      "model_forward_time": 2.435527801513672,
      "step": 2053
    },
    {
      "epoch": 7.831573486328125e-06,
      "step": 2053,
      "training_step_time": 10.301710605621338
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 3.177523612976074,
      "step": 2054
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 2054,
      "training_step_time": 9.949777841567993
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 3.135282039642334,
      "step": 2054
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 2054,
      "training_step_time": 9.907031059265137
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 3.1397645473480225,
      "step": 2054
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 2054,
      "training_step_time": 9.910725593566895
    },
    {
      "epoch": 7.83538818359375e-06,
      "model_forward_time": 3.136831760406494,
      "step": 2054
    },
    {
      "epoch": 7.83538818359375e-06,
      "step": 2054,
      "training_step_time": 10.268757581710815
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 3.1619629859924316,
      "step": 2055
    },
    {
      "dynamics_loss_avg": 0.07384241484105587,
      "epoch": 7.839202880859376e-06,
      "step": 2055
    },
    {
      "action_loss_avg": 0.051469644904136656,
      "epoch": 7.839202880859376e-06,
      "step": 2055
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 2055,
      "training_step_time": 10.211352348327637
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 3.1469452381134033,
      "step": 2055
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 2055,
      "training_step_time": 9.923953533172607
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 3.138780355453491,
      "step": 2055
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 2055,
      "training_step_time": 9.909351587295532
    },
    {
      "epoch": 7.839202880859376e-06,
      "model_forward_time": 3.1372663974761963,
      "step": 2055
    },
    {
      "epoch": 7.839202880859376e-06,
      "step": 2055,
      "training_step_time": 10.259649753570557
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 3.180318832397461,
      "step": 2056
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 2056,
      "training_step_time": 9.951074361801147
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 3.1392982006073,
      "step": 2056
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 2056,
      "training_step_time": 10.195075511932373
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 3.134648323059082,
      "step": 2056
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 2056,
      "training_step_time": 9.905900239944458
    },
    {
      "epoch": 7.843017578125e-06,
      "model_forward_time": 3.1361875534057617,
      "step": 2056
    },
    {
      "epoch": 7.843017578125e-06,
      "step": 2056,
      "training_step_time": 10.015164375305176
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 3.18493914604187,
      "step": 2057
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 2057,
      "training_step_time": 9.970912456512451
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 3.1359612941741943,
      "step": 2057
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 2057,
      "training_step_time": 9.906574249267578
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 3.4303107261657715,
      "step": 2057
    },
    {
      "dynamics_loss_avg": 0.052737264148890975,
      "epoch": 7.846832275390625e-06,
      "step": 2057
    },
    {
      "action_loss_avg": 0.03568986766040325,
      "epoch": 7.846832275390625e-06,
      "step": 2057
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 2057,
      "training_step_time": 10.198567152023315
    },
    {
      "epoch": 7.846832275390625e-06,
      "model_forward_time": 3.1326744556427,
      "step": 2057
    },
    {
      "epoch": 7.846832275390625e-06,
      "step": 2057,
      "training_step_time": 10.04431438446045
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 3.189035654067993,
      "step": 2058
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 2058,
      "training_step_time": 9.95979619026184
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 3.1367595195770264,
      "step": 2058
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 2058,
      "training_step_time": 9.905132532119751
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 3.1319451332092285,
      "step": 2058
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 2058,
      "training_step_time": 9.904843807220459
    },
    {
      "epoch": 7.85064697265625e-06,
      "model_forward_time": 3.1378393173217773,
      "step": 2058
    },
    {
      "epoch": 7.85064697265625e-06,
      "step": 2058,
      "training_step_time": 10.294478178024292
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 3.1378636360168457,
      "step": 2059
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 2059,
      "training_step_time": 9.90885853767395
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 3.1318013668060303,
      "step": 2059
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 2059,
      "training_step_time": 9.900991678237915
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 2.4312543869018555,
      "step": 2059
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 2059,
      "training_step_time": 7.459587812423706
    },
    {
      "epoch": 7.854461669921876e-06,
      "model_forward_time": 3.131422281265259,
      "step": 2059
    },
    {
      "epoch": 7.854461669921876e-06,
      "step": 2059,
      "training_step_time": 10.273484945297241
    },
    {
      "epoch": 7.8582763671875e-06,
      "grad_norm": 0.4982415437698364,
      "learning_rate": 1e-05,
      "loss": 0.1147,
      "step": 2060
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 3.4525249004364014,
      "step": 2060
    },
    {
      "dynamics_loss_avg": 0.06682684868574143,
      "epoch": 7.8582763671875e-06,
      "step": 2060
    },
    {
      "action_loss_avg": 0.04060581903904677,
      "epoch": 7.8582763671875e-06,
      "step": 2060
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 2060,
      "training_step_time": 10.224426031112671
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 3.133864641189575,
      "step": 2060
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 2060,
      "training_step_time": 9.900248765945435
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 3.1329424381256104,
      "step": 2060
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 2060,
      "training_step_time": 9.90150260925293
    },
    {
      "epoch": 7.8582763671875e-06,
      "model_forward_time": 3.1385557651519775,
      "step": 2060
    },
    {
      "epoch": 7.8582763671875e-06,
      "step": 2060,
      "training_step_time": 10.312420845031738
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 3.1863017082214355,
      "step": 2061
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 2061,
      "training_step_time": 9.958507776260376
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 3.1331701278686523,
      "step": 2061
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 2061,
      "training_step_time": 10.178086519241333
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 3.1347720623016357,
      "step": 2061
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 2061,
      "training_step_time": 9.903241157531738
    },
    {
      "epoch": 7.862091064453125e-06,
      "model_forward_time": 3.1359004974365234,
      "step": 2061
    },
    {
      "epoch": 7.862091064453125e-06,
      "step": 2061,
      "training_step_time": 10.00170612335205
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 3.1520586013793945,
      "step": 2062
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 2062,
      "training_step_time": 9.919920921325684
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 3.127532958984375,
      "step": 2062
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 2062,
      "training_step_time": 9.89381718635559
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 2.7222707271575928,
      "step": 2062
    },
    {
      "dynamics_loss_avg": 0.06892496086657048,
      "epoch": 7.86590576171875e-06,
      "step": 2062
    },
    {
      "action_loss_avg": 0.04021057281643152,
      "epoch": 7.86590576171875e-06,
      "step": 2062
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 2062,
      "training_step_time": 7.744360446929932
    },
    {
      "epoch": 7.86590576171875e-06,
      "model_forward_time": 3.131343126296997,
      "step": 2062
    },
    {
      "epoch": 7.86590576171875e-06,
      "step": 2062,
      "training_step_time": 10.005281209945679
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 3.1555421352386475,
      "step": 2063
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 2063,
      "training_step_time": 9.926289558410645
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 2.437779188156128,
      "step": 2063
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 2063,
      "training_step_time": 7.461839914321899
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 3.139838457107544,
      "step": 2063
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 2063,
      "training_step_time": 9.911605596542358
    },
    {
      "epoch": 7.869720458984376e-06,
      "model_forward_time": 3.1438522338867188,
      "step": 2063
    },
    {
      "epoch": 7.869720458984376e-06,
      "step": 2063,
      "training_step_time": 10.311468601226807
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 3.48038911819458,
      "step": 2064
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 2064,
      "training_step_time": 10.259781122207642
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 3.1358392238616943,
      "step": 2064
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 2064,
      "training_step_time": 9.910926342010498
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 3.137831449508667,
      "step": 2064
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 2064,
      "training_step_time": 9.910715341567993
    },
    {
      "epoch": 7.87353515625e-06,
      "model_forward_time": 3.1325342655181885,
      "step": 2064
    },
    {
      "epoch": 7.87353515625e-06,
      "step": 2064,
      "training_step_time": 10.27720594406128
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 3.179489850997925,
      "step": 2065
    },
    {
      "dynamics_loss_avg": 0.06576571054756641,
      "epoch": 7.877349853515625e-06,
      "step": 2065
    },
    {
      "action_loss_avg": 0.03475359734147787,
      "epoch": 7.877349853515625e-06,
      "step": 2065
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 2065,
      "training_step_time": 10.237722635269165
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 3.1316988468170166,
      "step": 2065
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 2065,
      "training_step_time": 9.902867317199707
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 3.131472587585449,
      "step": 2065
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 2065,
      "training_step_time": 9.906224012374878
    },
    {
      "epoch": 7.877349853515625e-06,
      "model_forward_time": 3.1341540813446045,
      "step": 2065
    },
    {
      "epoch": 7.877349853515625e-06,
      "step": 2065,
      "training_step_time": 10.285367965698242
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 3.1710667610168457,
      "step": 2066
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 2066,
      "training_step_time": 9.938395023345947
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 3.386326789855957,
      "step": 2066
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 2066,
      "training_step_time": 10.15412974357605
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 3.134737968444824,
      "step": 2066
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 2066,
      "training_step_time": 9.904017448425293
    },
    {
      "epoch": 7.88116455078125e-06,
      "model_forward_time": 3.1337382793426514,
      "step": 2066
    },
    {
      "epoch": 7.88116455078125e-06,
      "step": 2066,
      "training_step_time": 10.018471956253052
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 3.1636598110198975,
      "step": 2067
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 2067,
      "training_step_time": 9.934608936309814
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 3.137146472930908,
      "step": 2067
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 2067,
      "training_step_time": 9.907166481018066
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 3.1372809410095215,
      "step": 2067
    },
    {
      "dynamics_loss_avg": 0.06746680513024331,
      "epoch": 7.884979248046876e-06,
      "step": 2067
    },
    {
      "action_loss_avg": 0.04230323499068618,
      "epoch": 7.884979248046876e-06,
      "step": 2067
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 2067,
      "training_step_time": 10.179279565811157
    },
    {
      "epoch": 7.884979248046876e-06,
      "model_forward_time": 3.1330316066741943,
      "step": 2067
    },
    {
      "epoch": 7.884979248046876e-06,
      "step": 2067,
      "training_step_time": 10.019408464431763
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 3.18643856048584,
      "step": 2068
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 2068,
      "training_step_time": 9.961477041244507
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 3.138375759124756,
      "step": 2068
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 2068,
      "training_step_time": 9.91364860534668
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 3.134150266647339,
      "step": 2068
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 2068,
      "training_step_time": 9.908085107803345
    },
    {
      "epoch": 7.8887939453125e-06,
      "model_forward_time": 3.3973050117492676,
      "step": 2068
    },
    {
      "epoch": 7.8887939453125e-06,
      "step": 2068,
      "training_step_time": 10.322376251220703
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 3.1684622764587402,
      "step": 2069
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 2069,
      "training_step_time": 9.943615674972534
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 3.1401472091674805,
      "step": 2069
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 2069,
      "training_step_time": 9.91405439376831
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 3.138099193572998,
      "step": 2069
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 2069,
      "training_step_time": 9.91279935836792
    },
    {
      "epoch": 7.892608642578125e-06,
      "model_forward_time": 3.141580104827881,
      "step": 2069
    },
    {
      "epoch": 7.892608642578125e-06,
      "step": 2069,
      "training_step_time": 10.30337142944336
    },
    {
      "epoch": 7.89642333984375e-06,
      "grad_norm": 0.5247408747673035,
      "learning_rate": 1e-05,
      "loss": 0.1079,
      "step": 2070
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 3.1608211994171143,
      "step": 2070
    },
    {
      "dynamics_loss_avg": 0.06333620510995389,
      "epoch": 7.89642333984375e-06,
      "step": 2070
    },
    {
      "action_loss_avg": 0.03788143647834659,
      "epoch": 7.89642333984375e-06,
      "step": 2070
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 2070,
      "training_step_time": 10.207950830459595
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 3.1423425674438477,
      "step": 2070
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 2070,
      "training_step_time": 9.917441606521606
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 3.1373841762542725,
      "step": 2070
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 2070,
      "training_step_time": 9.910093784332275
    },
    {
      "epoch": 7.89642333984375e-06,
      "model_forward_time": 3.138686180114746,
      "step": 2070
    },
    {
      "epoch": 7.89642333984375e-06,
      "step": 2070,
      "training_step_time": 10.306025505065918
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 3.167780876159668,
      "step": 2071
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 2071,
      "training_step_time": 9.945515871047974
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 3.411386251449585,
      "step": 2071
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 2071,
      "training_step_time": 10.183266401290894
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 3.136369466781616,
      "step": 2071
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 2071,
      "training_step_time": 9.906302452087402
    },
    {
      "epoch": 7.900238037109376e-06,
      "model_forward_time": 3.139554738998413,
      "step": 2071
    },
    {
      "epoch": 7.900238037109376e-06,
      "step": 2071,
      "training_step_time": 10.011142015457153
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 3.1935136318206787,
      "step": 2072
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 2072,
      "training_step_time": 9.96007251739502
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 3.1375277042388916,
      "step": 2072
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 2072,
      "training_step_time": 9.905152797698975
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 3.3861801624298096,
      "step": 2072
    },
    {
      "dynamics_loss_avg": 0.061295867431908846,
      "epoch": 7.904052734375e-06,
      "step": 2072
    },
    {
      "action_loss_avg": 0.03405559747479856,
      "epoch": 7.904052734375e-06,
      "step": 2072
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 2072,
      "training_step_time": 10.158704280853271
    },
    {
      "epoch": 7.904052734375e-06,
      "model_forward_time": 3.138096809387207,
      "step": 2072
    },
    {
      "epoch": 7.904052734375e-06,
      "step": 2072,
      "training_step_time": 10.01390266418457
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 3.1696414947509766,
      "step": 2073
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 2073,
      "training_step_time": 9.941428184509277
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 3.136866807937622,
      "step": 2073
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 2073,
      "training_step_time": 9.91259765625
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 3.1415185928344727,
      "step": 2073
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 2073,
      "training_step_time": 9.91028380393982
    },
    {
      "epoch": 7.907867431640625e-06,
      "model_forward_time": 3.4088590145111084,
      "step": 2073
    },
    {
      "epoch": 7.907867431640625e-06,
      "step": 2073,
      "training_step_time": 10.277948379516602
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 3.1704375743865967,
      "step": 2074
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 2074,
      "training_step_time": 9.942097425460815
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 3.1330678462982178,
      "step": 2074
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 2074,
      "training_step_time": 9.900150537490845
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 3.1351161003112793,
      "step": 2074
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 2074,
      "training_step_time": 9.903204679489136
    },
    {
      "epoch": 7.91168212890625e-06,
      "model_forward_time": 3.1351101398468018,
      "step": 2074
    },
    {
      "epoch": 7.91168212890625e-06,
      "step": 2074,
      "training_step_time": 10.278752326965332
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 3.4473912715911865,
      "step": 2075
    },
    {
      "dynamics_loss_avg": 0.05620414782315493,
      "epoch": 7.915496826171876e-06,
      "step": 2075
    },
    {
      "action_loss_avg": 0.028874032525345684,
      "epoch": 7.915496826171876e-06,
      "step": 2075
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 2075,
      "training_step_time": 10.21665334701538
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 3.1340770721435547,
      "step": 2075
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 2075,
      "training_step_time": 9.902321815490723
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 3.136824131011963,
      "step": 2075
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 2075,
      "training_step_time": 9.907104730606079
    },
    {
      "epoch": 7.915496826171876e-06,
      "model_forward_time": 3.1359400749206543,
      "step": 2075
    },
    {
      "epoch": 7.915496826171876e-06,
      "step": 2075,
      "training_step_time": 10.298327922821045
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 3.1716127395629883,
      "step": 2076
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 2076,
      "training_step_time": 9.939893245697021
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 3.39300799369812,
      "step": 2076
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 2076,
      "training_step_time": 10.160753965377808
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 3.141310930252075,
      "step": 2076
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 2076,
      "training_step_time": 9.91240382194519
    },
    {
      "epoch": 7.9193115234375e-06,
      "model_forward_time": 3.1289515495300293,
      "step": 2076
    },
    {
      "epoch": 7.9193115234375e-06,
      "step": 2076,
      "training_step_time": 10.279430866241455
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 3.1798477172851562,
      "step": 2077
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 2077,
      "training_step_time": 9.95456838607788
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 3.136739492416382,
      "step": 2077
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 2077,
      "training_step_time": 9.904027938842773
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 3.421839475631714,
      "step": 2077
    },
    {
      "dynamics_loss_avg": 0.04867506455630064,
      "epoch": 7.923126220703125e-06,
      "step": 2077
    },
    {
      "action_loss_avg": 0.03431215351447463,
      "epoch": 7.923126220703125e-06,
      "step": 2077
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 2077,
      "training_step_time": 10.191162109375
    },
    {
      "epoch": 7.923126220703125e-06,
      "model_forward_time": 3.134514570236206,
      "step": 2077
    },
    {
      "epoch": 7.923126220703125e-06,
      "step": 2077,
      "training_step_time": 10.073265552520752
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 3.1818888187408447,
      "step": 2078
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 2078,
      "training_step_time": 9.952815771102905
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 3.1391429901123047,
      "step": 2078
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 2078,
      "training_step_time": 9.91006588935852
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 3.1375110149383545,
      "step": 2078
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 2078,
      "training_step_time": 9.914506435394287
    },
    {
      "epoch": 7.92694091796875e-06,
      "model_forward_time": 3.3999531269073486,
      "step": 2078
    },
    {
      "epoch": 7.92694091796875e-06,
      "step": 2078,
      "training_step_time": 10.271668910980225
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 3.179435968399048,
      "step": 2079
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 2079,
      "training_step_time": 9.95613169670105
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 3.14005184173584,
      "step": 2079
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 2079,
      "training_step_time": 9.915720224380493
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 3.1405417919158936,
      "step": 2079
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 2079,
      "training_step_time": 9.917279720306396
    },
    {
      "epoch": 7.930755615234376e-06,
      "model_forward_time": 3.1374568939208984,
      "step": 2079
    },
    {
      "epoch": 7.930755615234376e-06,
      "step": 2079,
      "training_step_time": 10.331103086471558
    },
    {
      "epoch": 7.9345703125e-06,
      "grad_norm": 0.7332921624183655,
      "learning_rate": 1e-05,
      "loss": 0.1063,
      "step": 2080
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 3.432529926300049,
      "step": 2080
    },
    {
      "dynamics_loss_avg": 0.059823932498693465,
      "epoch": 7.9345703125e-06,
      "step": 2080
    },
    {
      "action_loss_avg": 0.04257566705346107,
      "epoch": 7.9345703125e-06,
      "step": 2080
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 2080,
      "training_step_time": 10.204898834228516
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 3.136688470840454,
      "step": 2080
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 2080,
      "training_step_time": 9.903529167175293
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 3.129901170730591,
      "step": 2080
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 2080,
      "training_step_time": 9.893261432647705
    },
    {
      "epoch": 7.9345703125e-06,
      "model_forward_time": 3.136195182800293,
      "step": 2080
    },
    {
      "epoch": 7.9345703125e-06,
      "step": 2080,
      "training_step_time": 10.28923225402832
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 3.1752352714538574,
      "step": 2081
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 2081,
      "training_step_time": 9.950361251831055
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 3.4098548889160156,
      "step": 2081
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 2081,
      "training_step_time": 10.172074556350708
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 3.1287310123443604,
      "step": 2081
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 2081,
      "training_step_time": 9.890630006790161
    },
    {
      "epoch": 7.938385009765625e-06,
      "model_forward_time": 3.1327767372131348,
      "step": 2081
    },
    {
      "epoch": 7.938385009765625e-06,
      "step": 2081,
      "training_step_time": 10.276707649230957
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 3.157622814178467,
      "step": 2082
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 2082,
      "training_step_time": 9.927280902862549
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 3.1260929107666016,
      "step": 2082
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 2082,
      "training_step_time": 9.894992589950562
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 3.3752996921539307,
      "step": 2082
    },
    {
      "dynamics_loss_avg": 0.07377989329397679,
      "epoch": 7.94219970703125e-06,
      "step": 2082
    },
    {
      "action_loss_avg": 0.0398975683376193,
      "epoch": 7.94219970703125e-06,
      "step": 2082
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 2082,
      "training_step_time": 10.139739513397217
    },
    {
      "epoch": 7.94219970703125e-06,
      "model_forward_time": 3.1322832107543945,
      "step": 2082
    },
    {
      "epoch": 7.94219970703125e-06,
      "step": 2082,
      "training_step_time": 10.003057718276978
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 3.186074733734131,
      "step": 2083
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 2083,
      "training_step_time": 9.950565338134766
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 3.1324119567871094,
      "step": 2083
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 2083,
      "training_step_time": 9.894335985183716
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 3.136587619781494,
      "step": 2083
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 2083,
      "training_step_time": 9.901346921920776
    },
    {
      "epoch": 7.946014404296876e-06,
      "model_forward_time": 3.40875244140625,
      "step": 2083
    },
    {
      "epoch": 7.946014404296876e-06,
      "step": 2083,
      "training_step_time": 10.273993730545044
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 3.1668431758880615,
      "step": 2084
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 2084,
      "training_step_time": 9.937907695770264
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 3.135068416595459,
      "step": 2084
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 2084,
      "training_step_time": 9.906600713729858
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 3.1374568939208984,
      "step": 2084
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 2084,
      "training_step_time": 9.904744625091553
    },
    {
      "epoch": 7.9498291015625e-06,
      "model_forward_time": 3.1345582008361816,
      "step": 2084
    },
    {
      "epoch": 7.9498291015625e-06,
      "step": 2084,
      "training_step_time": 10.264883279800415
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 3.440859317779541,
      "step": 2085
    },
    {
      "dynamics_loss_avg": 0.07453354224562644,
      "epoch": 7.953643798828125e-06,
      "step": 2085
    },
    {
      "action_loss_avg": 0.0426840839907527,
      "epoch": 7.953643798828125e-06,
      "step": 2085
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 2085,
      "training_step_time": 10.218145847320557
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 3.137127161026001,
      "step": 2085
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 2085,
      "training_step_time": 9.911713600158691
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 3.14054012298584,
      "step": 2085
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 2085,
      "training_step_time": 9.912786722183228
    },
    {
      "epoch": 7.953643798828125e-06,
      "model_forward_time": 3.137199878692627,
      "step": 2085
    },
    {
      "epoch": 7.953643798828125e-06,
      "step": 2085,
      "training_step_time": 10.308367490768433
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 3.1754491329193115,
      "step": 2086
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 2086,
      "training_step_time": 9.94487452507019
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 3.382438898086548,
      "step": 2086
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 2086,
      "training_step_time": 10.151968002319336
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 3.1398305892944336,
      "step": 2086
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 2086,
      "training_step_time": 9.921525239944458
    },
    {
      "epoch": 7.95745849609375e-06,
      "model_forward_time": 3.1406548023223877,
      "step": 2086
    },
    {
      "epoch": 7.95745849609375e-06,
      "step": 2086,
      "training_step_time": 10.277330160140991
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 3.1876349449157715,
      "step": 2087
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 2087,
      "training_step_time": 9.961899995803833
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 3.141144037246704,
      "step": 2087
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 2087,
      "training_step_time": 9.913824319839478
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 2.708329439163208,
      "step": 2087
    },
    {
      "dynamics_loss_avg": 0.06578987408429385,
      "epoch": 7.961273193359376e-06,
      "step": 2087
    },
    {
      "action_loss_avg": 0.039048149809241296,
      "epoch": 7.961273193359376e-06,
      "step": 2087
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 2087,
      "training_step_time": 7.741587400436401
    },
    {
      "epoch": 7.961273193359376e-06,
      "model_forward_time": 3.136981248855591,
      "step": 2087
    },
    {
      "epoch": 7.961273193359376e-06,
      "step": 2087,
      "training_step_time": 10.020069122314453
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 3.1528537273406982,
      "step": 2088
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 2088,
      "training_step_time": 9.928152561187744
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 3.141545534133911,
      "step": 2088
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 2088,
      "training_step_time": 9.916775465011597
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 3.1389005184173584,
      "step": 2088
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 2088,
      "training_step_time": 9.91238784790039
    },
    {
      "epoch": 7.965087890625e-06,
      "model_forward_time": 2.7065417766571045,
      "step": 2088
    },
    {
      "epoch": 7.965087890625e-06,
      "step": 2088,
      "training_step_time": 10.015514135360718
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 3.173543691635132,
      "step": 2089
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 2089,
      "training_step_time": 9.946564435958862
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 3.1387202739715576,
      "step": 2089
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 2089,
      "training_step_time": 9.911723136901855
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 3.1385648250579834,
      "step": 2089
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 2089,
      "training_step_time": 9.909696340560913
    },
    {
      "epoch": 7.968902587890625e-06,
      "model_forward_time": 3.1406524181365967,
      "step": 2089
    },
    {
      "epoch": 7.968902587890625e-06,
      "step": 2089,
      "training_step_time": 10.299684762954712
    },
    {
      "epoch": 7.97271728515625e-06,
      "grad_norm": 0.5992704033851624,
      "learning_rate": 1e-05,
      "loss": 0.1121,
      "step": 2090
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 3.44553542137146,
      "step": 2090
    },
    {
      "dynamics_loss_avg": 0.08238036707043647,
      "epoch": 7.97271728515625e-06,
      "step": 2090
    },
    {
      "action_loss_avg": 0.051193788461387155,
      "epoch": 7.97271728515625e-06,
      "step": 2090
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 2090,
      "training_step_time": 10.221605062484741
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 3.145296335220337,
      "step": 2090
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 2090,
      "training_step_time": 9.918385982513428
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 3.144289970397949,
      "step": 2090
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 2090,
      "training_step_time": 9.920226812362671
    },
    {
      "epoch": 7.97271728515625e-06,
      "model_forward_time": 3.1484909057617188,
      "step": 2090
    },
    {
      "epoch": 7.97271728515625e-06,
      "step": 2090,
      "training_step_time": 10.30326247215271
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 3.1850011348724365,
      "step": 2091
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 2091,
      "training_step_time": 9.965147256851196
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 3.4189071655273438,
      "step": 2091
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 2091,
      "training_step_time": 10.19481348991394
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 3.146585702896118,
      "step": 2091
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 2091,
      "training_step_time": 9.9243004322052
    },
    {
      "epoch": 7.976531982421876e-06,
      "model_forward_time": 3.1485538482666016,
      "step": 2091
    },
    {
      "epoch": 7.976531982421876e-06,
      "step": 2091,
      "training_step_time": 10.276317358016968
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 3.164616823196411,
      "step": 2092
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 2092,
      "training_step_time": 9.943056344985962
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 3.1385021209716797,
      "step": 2092
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 2092,
      "training_step_time": 9.918779850006104
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 2.672078847885132,
      "step": 2092
    },
    {
      "dynamics_loss_avg": 0.0644497625529766,
      "epoch": 7.9803466796875e-06,
      "step": 2092
    },
    {
      "action_loss_avg": 0.037367967702448365,
      "epoch": 7.9803466796875e-06,
      "step": 2092
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 2092,
      "training_step_time": 7.7022035121917725
    },
    {
      "epoch": 7.9803466796875e-06,
      "model_forward_time": 3.1326513290405273,
      "step": 2092
    },
    {
      "epoch": 7.9803466796875e-06,
      "step": 2092,
      "training_step_time": 10.260904788970947
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 3.1787867546081543,
      "step": 2093
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 2093,
      "training_step_time": 9.95272707939148
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 3.1332099437713623,
      "step": 2093
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 2093,
      "training_step_time": 9.904446363449097
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 3.136749267578125,
      "step": 2093
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 2093,
      "training_step_time": 9.914937019348145
    },
    {
      "epoch": 7.984161376953125e-06,
      "model_forward_time": 3.4020802974700928,
      "step": 2093
    },
    {
      "epoch": 7.984161376953125e-06,
      "step": 2093,
      "training_step_time": 10.27311086654663
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 3.17071270942688,
      "step": 2094
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 2094,
      "training_step_time": 9.943381547927856
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 3.1348748207092285,
      "step": 2094
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 2094,
      "training_step_time": 9.905879020690918
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 3.1375732421875,
      "step": 2094
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 2094,
      "training_step_time": 9.906470775604248
    },
    {
      "epoch": 7.98797607421875e-06,
      "model_forward_time": 3.135343551635742,
      "step": 2094
    },
    {
      "epoch": 7.98797607421875e-06,
      "step": 2094,
      "training_step_time": 10.238971710205078
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 3.4443249702453613,
      "step": 2095
    },
    {
      "dynamics_loss_avg": 0.0737327441573143,
      "epoch": 7.991790771484376e-06,
      "step": 2095
    },
    {
      "action_loss_avg": 0.04864320158958435,
      "epoch": 7.991790771484376e-06,
      "step": 2095
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 2095,
      "training_step_time": 10.226119041442871
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 3.1362979412078857,
      "step": 2095
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 2095,
      "training_step_time": 9.911399841308594
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 2.4343621730804443,
      "step": 2095
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 2095,
      "training_step_time": 7.467448711395264
    },
    {
      "epoch": 7.991790771484376e-06,
      "model_forward_time": 3.131721258163452,
      "step": 2095
    },
    {
      "epoch": 7.991790771484376e-06,
      "step": 2095,
      "training_step_time": 10.284754753112793
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 3.1904332637786865,
      "step": 2096
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 2096,
      "training_step_time": 9.963157653808594
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 3.383836269378662,
      "step": 2096
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 2096,
      "training_step_time": 10.156819581985474
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 3.1463420391082764,
      "step": 2096
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 2096,
      "training_step_time": 9.925542116165161
    },
    {
      "epoch": 7.99560546875e-06,
      "model_forward_time": 3.1441006660461426,
      "step": 2096
    },
    {
      "epoch": 7.99560546875e-06,
      "step": 2096,
      "training_step_time": 10.313657999038696
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 3.168348789215088,
      "step": 2097
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 2097,
      "training_step_time": 9.943798542022705
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 3.1396193504333496,
      "step": 2097
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 2097,
      "training_step_time": 10.176650047302246
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 3.1395232677459717,
      "step": 2097
    },
    {
      "dynamics_loss_avg": 0.06454609092324973,
      "epoch": 7.999420166015625e-06,
      "step": 2097
    },
    {
      "action_loss_avg": 0.03558895001187921,
      "epoch": 7.999420166015625e-06,
      "step": 2097
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 2097,
      "training_step_time": 9.916112661361694
    },
    {
      "epoch": 7.999420166015625e-06,
      "model_forward_time": 3.1417758464813232,
      "step": 2097
    },
    {
      "epoch": 7.999420166015625e-06,
      "step": 2097,
      "training_step_time": 10.300612449645996
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 3.184802293777466,
      "step": 2098
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 2098,
      "training_step_time": 9.956443548202515
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 3.137751579284668,
      "step": 2098
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 2098,
      "training_step_time": 9.916696310043335
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 3.1375601291656494,
      "step": 2098
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 2098,
      "training_step_time": 10.186811923980713
    },
    {
      "epoch": 8.00323486328125e-06,
      "model_forward_time": 3.1366353034973145,
      "step": 2098
    },
    {
      "epoch": 8.00323486328125e-06,
      "step": 2098,
      "training_step_time": 10.301183700561523
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 3.1443259716033936,
      "step": 2099
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 2099,
      "training_step_time": 9.917483806610107
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 3.1394553184509277,
      "step": 2099
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 2099,
      "training_step_time": 9.916303634643555
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 3.142328977584839,
      "step": 2099
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 2099,
      "training_step_time": 9.921410083770752
    },
    {
      "epoch": 8.007049560546876e-06,
      "model_forward_time": 2.7361302375793457,
      "step": 2099
    },
    {
      "epoch": 8.007049560546876e-06,
      "step": 2099,
      "training_step_time": 10.27214527130127
    },
    {
      "epoch": 8.0108642578125e-06,
      "grad_norm": 0.4367607831954956,
      "learning_rate": 1e-05,
      "loss": 0.1104,
      "step": 2100
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 3.1916611194610596,
      "step": 2100
    },
    {
      "dynamics_loss_avg": 0.0651170639321208,
      "epoch": 8.0108642578125e-06,
      "step": 2100
    },
    {
      "action_loss_avg": 0.045867695938795805,
      "epoch": 8.0108642578125e-06,
      "step": 2100
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 2100,
      "training_step_time": 9.967540264129639
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 3.1364002227783203,
      "step": 2100
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 2100,
      "training_step_time": 9.909406661987305
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 3.135974645614624,
      "step": 2100
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 2100,
      "training_step_time": 9.912193298339844
    },
    {
      "epoch": 8.0108642578125e-06,
      "model_forward_time": 3.140150547027588,
      "step": 2100
    },
    {
      "epoch": 8.0108642578125e-06,
      "step": 2100,
      "training_step_time": 10.296066284179688
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 3.1780178546905518,
      "step": 2101
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 2101,
      "training_step_time": 10.219773054122925
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 3.1391971111297607,
      "step": 2101
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 2101,
      "training_step_time": 9.91093921661377
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 3.134305715560913,
      "step": 2101
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 2101,
      "training_step_time": 9.908987283706665
    },
    {
      "epoch": 8.014678955078125e-06,
      "model_forward_time": 3.139193296432495,
      "step": 2101
    },
    {
      "epoch": 8.014678955078125e-06,
      "step": 2101,
      "training_step_time": 10.299896001815796
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 3.1889195442199707,
      "step": 2102
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 2102,
      "training_step_time": 9.961642980575562
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 3.131728172302246,
      "step": 2102
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 2102,
      "training_step_time": 10.189381837844849
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 3.1335110664367676,
      "step": 2102
    },
    {
      "dynamics_loss_avg": 0.07597407829016448,
      "epoch": 8.01849365234375e-06,
      "step": 2102
    },
    {
      "action_loss_avg": 0.044186435639858246,
      "epoch": 8.01849365234375e-06,
      "step": 2102
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 2102,
      "training_step_time": 9.901748418807983
    },
    {
      "epoch": 8.01849365234375e-06,
      "model_forward_time": 3.131688117980957,
      "step": 2102
    },
    {
      "epoch": 8.01849365234375e-06,
      "step": 2102,
      "training_step_time": 10.281164407730103
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 3.1773905754089355,
      "step": 2103
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 2103,
      "training_step_time": 9.967148780822754
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 3.13577938079834,
      "step": 2103
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 2103,
      "training_step_time": 9.908004522323608
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 3.4037163257598877,
      "step": 2103
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 2103,
      "training_step_time": 10.179515361785889
    },
    {
      "epoch": 8.022308349609376e-06,
      "model_forward_time": 3.141447067260742,
      "step": 2103
    },
    {
      "epoch": 8.022308349609376e-06,
      "step": 2103,
      "training_step_time": 10.333740472793579
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 3.1707887649536133,
      "step": 2104
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 2104,
      "training_step_time": 9.941946744918823
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 3.1419355869293213,
      "step": 2104
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 2104,
      "training_step_time": 9.91506314277649
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 3.144901752471924,
      "step": 2104
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 2104,
      "training_step_time": 9.921548128128052
    },
    {
      "epoch": 8.026123046875e-06,
      "model_forward_time": 3.1434431076049805,
      "step": 2104
    },
    {
      "epoch": 8.026123046875e-06,
      "step": 2104,
      "training_step_time": 10.290401458740234
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 3.1807947158813477,
      "step": 2105
    },
    {
      "dynamics_loss_avg": 0.061826421320438384,
      "epoch": 8.029937744140625e-06,
      "step": 2105
    },
    {
      "action_loss_avg": 0.0378875644877553,
      "epoch": 8.029937744140625e-06,
      "step": 2105
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 2105,
      "training_step_time": 9.963703155517578
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 3.144392490386963,
      "step": 2105
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 2105,
      "training_step_time": 9.917975902557373
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 3.1485092639923096,
      "step": 2105
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 2105,
      "training_step_time": 9.926029920578003
    },
    {
      "epoch": 8.029937744140625e-06,
      "model_forward_time": 3.1425421237945557,
      "step": 2105
    },
    {
      "epoch": 8.029937744140625e-06,
      "step": 2105,
      "training_step_time": 10.279680967330933
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 3.4613122940063477,
      "step": 2106
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 2106,
      "training_step_time": 10.239999532699585
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 3.1439476013183594,
      "step": 2106
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 2106,
      "training_step_time": 9.920269250869751
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 3.142475128173828,
      "step": 2106
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 2106,
      "training_step_time": 9.915909767150879
    },
    {
      "epoch": 8.03375244140625e-06,
      "model_forward_time": 2.437628746032715,
      "step": 2106
    },
    {
      "epoch": 8.03375244140625e-06,
      "step": 2106,
      "training_step_time": 10.252228021621704
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 3.175281524658203,
      "step": 2107
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 2107,
      "training_step_time": 9.945181846618652
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 3.1406824588775635,
      "step": 2107
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 2107,
      "training_step_time": 9.913587808609009
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 3.4192564487457275,
      "step": 2107
    },
    {
      "dynamics_loss_avg": 0.07733224667608737,
      "epoch": 8.037567138671876e-06,
      "step": 2107
    },
    {
      "action_loss_avg": 0.04587157228961587,
      "epoch": 8.037567138671876e-06,
      "step": 2107
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 2107,
      "training_step_time": 10.195430994033813
    },
    {
      "epoch": 8.037567138671876e-06,
      "model_forward_time": 3.1407086849212646,
      "step": 2107
    },
    {
      "epoch": 8.037567138671876e-06,
      "step": 2107,
      "training_step_time": 10.30045485496521
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 3.1725378036499023,
      "step": 2108
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 2108,
      "training_step_time": 9.943955898284912
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 3.139195680618286,
      "step": 2108
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 2108,
      "training_step_time": 9.91065502166748
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 3.4188992977142334,
      "step": 2108
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 2108,
      "training_step_time": 10.182174682617188
    },
    {
      "epoch": 8.0413818359375e-06,
      "model_forward_time": 3.136780261993408,
      "step": 2108
    },
    {
      "epoch": 8.0413818359375e-06,
      "step": 2108,
      "training_step_time": 10.000831365585327
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 3.177490234375,
      "step": 2109
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 2109,
      "training_step_time": 9.948025703430176
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 3.138484477996826,
      "step": 2109
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 2109,
      "training_step_time": 9.909294366836548
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 3.13786244392395,
      "step": 2109
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 2109,
      "training_step_time": 9.911207675933838
    },
    {
      "epoch": 8.045196533203125e-06,
      "model_forward_time": 3.144578695297241,
      "step": 2109
    },
    {
      "epoch": 8.045196533203125e-06,
      "step": 2109,
      "training_step_time": 10.307283639907837
    },
    {
      "epoch": 8.04901123046875e-06,
      "grad_norm": 0.5247260332107544,
      "learning_rate": 1e-05,
      "loss": 0.112,
      "step": 2110
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 3.461893320083618,
      "step": 2110
    },
    {
      "dynamics_loss_avg": 0.06818240880966187,
      "epoch": 8.04901123046875e-06,
      "step": 2110
    },
    {
      "action_loss_avg": 0.04387010177597404,
      "epoch": 8.04901123046875e-06,
      "step": 2110
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 2110,
      "training_step_time": 10.24021053314209
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 3.14341139793396,
      "step": 2110
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 2110,
      "training_step_time": 9.917598485946655
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 3.1396396160125732,
      "step": 2110
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 2110,
      "training_step_time": 9.913512229919434
    },
    {
      "epoch": 8.04901123046875e-06,
      "model_forward_time": 3.1414828300476074,
      "step": 2110
    },
    {
      "epoch": 8.04901123046875e-06,
      "step": 2110,
      "training_step_time": 10.28567361831665
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 3.4420197010040283,
      "step": 2111
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 2111,
      "training_step_time": 10.213231563568115
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 3.1435084342956543,
      "step": 2111
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 2111,
      "training_step_time": 9.912883520126343
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 2.4364869594573975,
      "step": 2111
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 2111,
      "training_step_time": 7.457794189453125
    },
    {
      "epoch": 8.052825927734376e-06,
      "model_forward_time": 3.1362996101379395,
      "step": 2111
    },
    {
      "epoch": 8.052825927734376e-06,
      "step": 2111,
      "training_step_time": 10.28608775138855
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 3.1782565116882324,
      "step": 2112
    },
    {
      "epoch": 8.056640625e-06,
      "step": 2112,
      "training_step_time": 9.957144021987915
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 3.1351492404937744,
      "step": 2112
    },
    {
      "epoch": 8.056640625e-06,
      "step": 2112,
      "training_step_time": 9.905363082885742
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 3.3820905685424805,
      "step": 2112
    },
    {
      "dynamics_loss_avg": 0.06432087570428849,
      "epoch": 8.056640625e-06,
      "step": 2112
    },
    {
      "action_loss_avg": 0.03933626562356949,
      "epoch": 8.056640625e-06,
      "step": 2112
    },
    {
      "epoch": 8.056640625e-06,
      "step": 2112,
      "training_step_time": 10.154276609420776
    },
    {
      "epoch": 8.056640625e-06,
      "model_forward_time": 3.139071464538574,
      "step": 2112
    },
    {
      "epoch": 8.056640625e-06,
      "step": 2112,
      "training_step_time": 10.311617374420166
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 3.182060480117798,
      "step": 2113
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 2113,
      "training_step_time": 9.956824541091919
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 3.1357476711273193,
      "step": 2113
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 2113,
      "training_step_time": 9.906566143035889
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 3.1355912685394287,
      "step": 2113
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 2113,
      "training_step_time": 10.183715343475342
    },
    {
      "epoch": 8.060455322265625e-06,
      "model_forward_time": 3.135857582092285,
      "step": 2113
    },
    {
      "epoch": 8.060455322265625e-06,
      "step": 2113,
      "training_step_time": 10.231478214263916
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 3.172431707382202,
      "step": 2114
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 2114,
      "training_step_time": 9.941540479660034
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 3.1312849521636963,
      "step": 2114
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 2114,
      "training_step_time": 9.89599895477295
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 3.1358373165130615,
      "step": 2114
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 2114,
      "training_step_time": 9.902854919433594
    },
    {
      "epoch": 8.06427001953125e-06,
      "model_forward_time": 3.4294607639312744,
      "step": 2114
    },
    {
      "epoch": 8.06427001953125e-06,
      "step": 2114,
      "training_step_time": 10.292694568634033
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 3.1724772453308105,
      "step": 2115
    },
    {
      "dynamics_loss_avg": 0.06620943881571292,
      "epoch": 8.068084716796876e-06,
      "step": 2115
    },
    {
      "action_loss_avg": 0.036912073567509654,
      "epoch": 8.068084716796876e-06,
      "step": 2115
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 2115,
      "training_step_time": 9.944431781768799
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 3.134075880050659,
      "step": 2115
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 2115,
      "training_step_time": 9.904092073440552
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 3.1319878101348877,
      "step": 2115
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 2115,
      "training_step_time": 9.899029731750488
    },
    {
      "epoch": 8.068084716796876e-06,
      "model_forward_time": 3.1314713954925537,
      "step": 2115
    },
    {
      "epoch": 8.068084716796876e-06,
      "step": 2115,
      "training_step_time": 10.259654998779297
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 3.173811674118042,
      "step": 2116
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 2116,
      "training_step_time": 10.209885358810425
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 3.1283156871795654,
      "step": 2116
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 2116,
      "training_step_time": 9.894954681396484
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 3.132607936859131,
      "step": 2116
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 2116,
      "training_step_time": 9.90020203590393
    },
    {
      "epoch": 8.0718994140625e-06,
      "model_forward_time": 3.1307735443115234,
      "step": 2116
    },
    {
      "epoch": 8.0718994140625e-06,
      "step": 2116,
      "training_step_time": 10.278038263320923
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 3.172300100326538,
      "step": 2117
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 2117,
      "training_step_time": 9.945865154266357
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 3.4052791595458984,
      "step": 2117
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 2117,
      "training_step_time": 10.173710823059082
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 3.137270450592041,
      "step": 2117
    },
    {
      "dynamics_loss_avg": 0.05740037728101015,
      "epoch": 8.075714111328125e-06,
      "step": 2117
    },
    {
      "action_loss_avg": 0.03677238281816244,
      "epoch": 8.075714111328125e-06,
      "step": 2117
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 2117,
      "training_step_time": 9.908952474594116
    },
    {
      "epoch": 8.075714111328125e-06,
      "model_forward_time": 3.136038303375244,
      "step": 2117
    },
    {
      "epoch": 8.075714111328125e-06,
      "step": 2117,
      "training_step_time": 10.0094633102417
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 3.1735246181488037,
      "step": 2118
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 2118,
      "training_step_time": 9.945296049118042
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 3.137619733810425,
      "step": 2118
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 2118,
      "training_step_time": 9.911725044250488
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 3.431086301803589,
      "step": 2118
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 2118,
      "training_step_time": 10.200830936431885
    },
    {
      "epoch": 8.07952880859375e-06,
      "model_forward_time": 3.1372644901275635,
      "step": 2118
    },
    {
      "epoch": 8.07952880859375e-06,
      "step": 2118,
      "training_step_time": 10.307494878768921
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 3.153106451034546,
      "step": 2119
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 2119,
      "training_step_time": 9.924197673797607
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 3.1359472274780273,
      "step": 2119
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 2119,
      "training_step_time": 9.906871557235718
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 3.1347808837890625,
      "step": 2119
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 2119,
      "training_step_time": 9.90306544303894
    },
    {
      "epoch": 8.083343505859376e-06,
      "model_forward_time": 2.7222306728363037,
      "step": 2119
    },
    {
      "epoch": 8.083343505859376e-06,
      "step": 2119,
      "training_step_time": 10.281383991241455
    },
    {
      "epoch": 8.087158203125e-06,
      "grad_norm": 0.7309942841529846,
      "learning_rate": 1e-05,
      "loss": 0.1118,
      "step": 2120
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 3.165524959564209,
      "step": 2120
    },
    {
      "dynamics_loss_avg": 0.08083512485027314,
      "epoch": 8.087158203125e-06,
      "step": 2120
    },
    {
      "action_loss_avg": 0.06619978621602059,
      "epoch": 8.087158203125e-06,
      "step": 2120
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 2120,
      "training_step_time": 9.936965227127075
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 3.1363837718963623,
      "step": 2120
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 2120,
      "training_step_time": 9.908766031265259
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 3.1360437870025635,
      "step": 2120
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 2120,
      "training_step_time": 9.909775018692017
    },
    {
      "epoch": 8.087158203125e-06,
      "model_forward_time": 3.1398837566375732,
      "step": 2120
    },
    {
      "epoch": 8.087158203125e-06,
      "step": 2120,
      "training_step_time": 10.31436562538147
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 3.160526990890503,
      "step": 2121
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 2121,
      "training_step_time": 10.199051856994629
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 3.1400675773620605,
      "step": 2121
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 2121,
      "training_step_time": 9.910637855529785
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 3.1391761302948,
      "step": 2121
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 2121,
      "training_step_time": 9.911949872970581
    },
    {
      "epoch": 8.090972900390625e-06,
      "model_forward_time": 3.1468684673309326,
      "step": 2121
    },
    {
      "epoch": 8.090972900390625e-06,
      "step": 2121,
      "training_step_time": 10.281808376312256
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 3.1667888164520264,
      "step": 2122
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 2122,
      "training_step_time": 9.943467140197754
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 3.138526439666748,
      "step": 2122
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 2122,
      "training_step_time": 10.201353073120117
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 3.1419923305511475,
      "step": 2122
    },
    {
      "dynamics_loss_avg": 0.07810661941766739,
      "epoch": 8.09478759765625e-06,
      "step": 2122
    },
    {
      "action_loss_avg": 0.04860628489404917,
      "epoch": 8.09478759765625e-06,
      "step": 2122
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 2122,
      "training_step_time": 9.925134181976318
    },
    {
      "epoch": 8.09478759765625e-06,
      "model_forward_time": 3.140129804611206,
      "step": 2122
    },
    {
      "epoch": 8.09478759765625e-06,
      "step": 2122,
      "training_step_time": 10.028379678726196
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 3.180619478225708,
      "step": 2123
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 2123,
      "training_step_time": 9.95829153060913
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 3.137387752532959,
      "step": 2123
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 2123,
      "training_step_time": 9.913377285003662
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 3.3900628089904785,
      "step": 2123
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 2123,
      "training_step_time": 10.163640260696411
    },
    {
      "epoch": 8.098602294921876e-06,
      "model_forward_time": 3.139432430267334,
      "step": 2123
    },
    {
      "epoch": 8.098602294921876e-06,
      "step": 2123,
      "training_step_time": 10.275714874267578
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 3.160036325454712,
      "step": 2124
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 2124,
      "training_step_time": 9.92947268486023
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 3.1301980018615723,
      "step": 2124
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 2124,
      "training_step_time": 9.894903659820557
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 3.1360721588134766,
      "step": 2124
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 2124,
      "training_step_time": 9.902810335159302
    },
    {
      "epoch": 8.1024169921875e-06,
      "model_forward_time": 3.127800703048706,
      "step": 2124
    },
    {
      "epoch": 8.1024169921875e-06,
      "step": 2124,
      "training_step_time": 10.258637189865112
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 3.184777021408081,
      "step": 2125
    },
    {
      "dynamics_loss_avg": 0.08487166445702314,
      "epoch": 8.106231689453125e-06,
      "step": 2125
    },
    {
      "action_loss_avg": 0.044825860764831305,
      "epoch": 8.106231689453125e-06,
      "step": 2125
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 2125,
      "training_step_time": 9.955966472625732
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 3.128685712814331,
      "step": 2125
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 2125,
      "training_step_time": 9.894065141677856
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 3.130910634994507,
      "step": 2125
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 2125,
      "training_step_time": 9.894323110580444
    },
    {
      "epoch": 8.106231689453125e-06,
      "model_forward_time": 3.1314713954925537,
      "step": 2125
    },
    {
      "epoch": 8.106231689453125e-06,
      "step": 2125,
      "training_step_time": 10.29814600944519
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 3.463940143585205,
      "step": 2126
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 2126,
      "training_step_time": 10.231746435165405
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 3.1397881507873535,
      "step": 2126
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 2126,
      "training_step_time": 9.910440921783447
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 3.1473116874694824,
      "step": 2126
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 2126,
      "training_step_time": 9.92758560180664
    },
    {
      "epoch": 8.11004638671875e-06,
      "model_forward_time": 3.1412248611450195,
      "step": 2126
    },
    {
      "epoch": 8.11004638671875e-06,
      "step": 2126,
      "training_step_time": 10.016358375549316
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 3.1683967113494873,
      "step": 2127
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 2127,
      "training_step_time": 9.951099634170532
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 3.1438355445861816,
      "step": 2127
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 2127,
      "training_step_time": 10.22607970237732
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 3.1502773761749268,
      "step": 2127
    },
    {
      "dynamics_loss_avg": 0.06112161725759506,
      "epoch": 8.113861083984374e-06,
      "step": 2127
    },
    {
      "action_loss_avg": 0.03340984536334872,
      "epoch": 8.113861083984374e-06,
      "step": 2127
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 2127,
      "training_step_time": 9.927927494049072
    },
    {
      "epoch": 8.113861083984374e-06,
      "model_forward_time": 3.143862724304199,
      "step": 2127
    },
    {
      "epoch": 8.113861083984374e-06,
      "step": 2127,
      "training_step_time": 10.266520261764526
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 3.1777215003967285,
      "step": 2128
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 2128,
      "training_step_time": 9.953661441802979
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 3.1440269947052,
      "step": 2128
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 2128,
      "training_step_time": 9.918303728103638
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 3.432384490966797,
      "step": 2128
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 2128,
      "training_step_time": 10.212403059005737
    },
    {
      "epoch": 8.11767578125e-06,
      "model_forward_time": 3.148484706878662,
      "step": 2128
    },
    {
      "epoch": 8.11767578125e-06,
      "step": 2128,
      "training_step_time": 10.022218465805054
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 3.191121816635132,
      "step": 2129
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 2129,
      "training_step_time": 9.967304468154907
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 3.1417384147644043,
      "step": 2129
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 2129,
      "training_step_time": 9.915418148040771
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 3.142340898513794,
      "step": 2129
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 2129,
      "training_step_time": 9.91607141494751
    },
    {
      "epoch": 8.121490478515625e-06,
      "model_forward_time": 3.4019644260406494,
      "step": 2129
    },
    {
      "epoch": 8.121490478515625e-06,
      "step": 2129,
      "training_step_time": 10.32256531715393
    },
    {
      "epoch": 8.12530517578125e-06,
      "grad_norm": 1.0250422954559326,
      "learning_rate": 1e-05,
      "loss": 0.1137,
      "step": 2130
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 3.197237014770508,
      "step": 2130
    },
    {
      "dynamics_loss_avg": 0.06194471605122089,
      "epoch": 8.12530517578125e-06,
      "step": 2130
    },
    {
      "action_loss_avg": 0.041291159112006426,
      "epoch": 8.12530517578125e-06,
      "step": 2130
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 2130,
      "training_step_time": 9.982433795928955
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 3.144620418548584,
      "step": 2130
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 2130,
      "training_step_time": 9.916932821273804
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 3.1379735469818115,
      "step": 2130
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 2130,
      "training_step_time": 9.911137580871582
    },
    {
      "epoch": 8.12530517578125e-06,
      "model_forward_time": 3.1435511112213135,
      "step": 2130
    },
    {
      "epoch": 8.12530517578125e-06,
      "step": 2130,
      "training_step_time": 10.297211408615112
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 3.4869320392608643,
      "step": 2131
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 2131,
      "training_step_time": 10.26322627067566
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 3.142947196960449,
      "step": 2131
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 2131,
      "training_step_time": 9.920746326446533
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 3.140756845474243,
      "step": 2131
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 2131,
      "training_step_time": 9.91059160232544
    },
    {
      "epoch": 8.129119873046874e-06,
      "model_forward_time": 3.142556667327881,
      "step": 2131
    },
    {
      "epoch": 8.129119873046874e-06,
      "step": 2131,
      "training_step_time": 10.312572240829468
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 3.1793625354766846,
      "step": 2132
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 2132,
      "training_step_time": 9.950028657913208
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 3.453906774520874,
      "step": 2132
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 2132,
      "training_step_time": 10.228520631790161
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 3.143981456756592,
      "step": 2132
    },
    {
      "dynamics_loss_avg": 0.0742145873606205,
      "epoch": 8.1329345703125e-06,
      "step": 2132
    },
    {
      "action_loss_avg": 0.048260764125734566,
      "epoch": 8.1329345703125e-06,
      "step": 2132
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 2132,
      "training_step_time": 9.914529800415039
    },
    {
      "epoch": 8.1329345703125e-06,
      "model_forward_time": 3.1464688777923584,
      "step": 2132
    },
    {
      "epoch": 8.1329345703125e-06,
      "step": 2132,
      "training_step_time": 10.27866268157959
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 3.172506809234619,
      "step": 2133
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 2133,
      "training_step_time": 9.946659564971924
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 3.139847993850708,
      "step": 2133
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 2133,
      "training_step_time": 9.909500122070312
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 3.1533193588256836,
      "step": 2133
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 2133,
      "training_step_time": 10.204241037368774
    },
    {
      "epoch": 8.136749267578125e-06,
      "model_forward_time": 3.1424367427825928,
      "step": 2133
    },
    {
      "epoch": 8.136749267578125e-06,
      "step": 2133,
      "training_step_time": 10.01890778541565
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 3.1862528324127197,
      "step": 2134
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 2134,
      "training_step_time": 9.964760065078735
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 3.143120288848877,
      "step": 2134
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 2134,
      "training_step_time": 9.915885210037231
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 3.1448168754577637,
      "step": 2134
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 2134,
      "training_step_time": 9.91487431526184
    },
    {
      "epoch": 8.14056396484375e-06,
      "model_forward_time": 3.421952724456787,
      "step": 2134
    },
    {
      "epoch": 8.14056396484375e-06,
      "step": 2134,
      "training_step_time": 10.289432525634766
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 3.1721689701080322,
      "step": 2135
    },
    {
      "dynamics_loss_avg": 0.06707965061068535,
      "epoch": 8.144378662109374e-06,
      "step": 2135
    },
    {
      "action_loss_avg": 0.04225523769855499,
      "epoch": 8.144378662109374e-06,
      "step": 2135
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 2135,
      "training_step_time": 9.946146726608276
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 3.1352131366729736,
      "step": 2135
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 2135,
      "training_step_time": 9.905708312988281
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 3.1419246196746826,
      "step": 2135
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 2135,
      "training_step_time": 9.917953729629517
    },
    {
      "epoch": 8.144378662109374e-06,
      "model_forward_time": 3.140561580657959,
      "step": 2135
    },
    {
      "epoch": 8.144378662109374e-06,
      "step": 2135,
      "training_step_time": 10.245327711105347
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 3.1695010662078857,
      "step": 2136
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 2136,
      "training_step_time": 10.214518308639526
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 3.1344540119171143,
      "step": 2136
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 2136,
      "training_step_time": 9.904561042785645
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 3.135706901550293,
      "step": 2136
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 2136,
      "training_step_time": 9.899099588394165
    },
    {
      "epoch": 8.148193359375e-06,
      "model_forward_time": 3.135791778564453,
      "step": 2136
    },
    {
      "epoch": 8.148193359375e-06,
      "step": 2136,
      "training_step_time": 10.243316173553467
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 3.1869547367095947,
      "step": 2137
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 2137,
      "training_step_time": 9.96283745765686
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 3.406371831893921,
      "step": 2137
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 2137,
      "training_step_time": 10.183897733688354
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 3.1428916454315186,
      "step": 2137
    },
    {
      "dynamics_loss_avg": 0.07104874402284622,
      "epoch": 8.152008056640625e-06,
      "step": 2137
    },
    {
      "action_loss_avg": 0.04291552621871233,
      "epoch": 8.152008056640625e-06,
      "step": 2137
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 2137,
      "training_step_time": 9.922157049179077
    },
    {
      "epoch": 8.152008056640625e-06,
      "model_forward_time": 3.140120267868042,
      "step": 2137
    },
    {
      "epoch": 8.152008056640625e-06,
      "step": 2137,
      "training_step_time": 10.270857334136963
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 3.182121515274048,
      "step": 2138
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 2138,
      "training_step_time": 9.95885968208313
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 3.1439383029937744,
      "step": 2138
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 2138,
      "training_step_time": 9.927998781204224
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 3.4510092735290527,
      "step": 2138
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 2138,
      "training_step_time": 10.22929334640503
    },
    {
      "epoch": 8.15582275390625e-06,
      "model_forward_time": 3.140639305114746,
      "step": 2138
    },
    {
      "epoch": 8.15582275390625e-06,
      "step": 2138,
      "training_step_time": 10.300294160842896
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 3.1950860023498535,
      "step": 2139
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 2139,
      "training_step_time": 9.972437620162964
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 3.1414237022399902,
      "step": 2139
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 2139,
      "training_step_time": 9.921467542648315
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 3.137882947921753,
      "step": 2139
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 2139,
      "training_step_time": 9.912605285644531
    },
    {
      "epoch": 8.159637451171874e-06,
      "model_forward_time": 3.3913795948028564,
      "step": 2139
    },
    {
      "epoch": 8.159637451171874e-06,
      "step": 2139,
      "training_step_time": 10.289108753204346
    },
    {
      "epoch": 8.1634521484375e-06,
      "grad_norm": 0.5478000640869141,
      "learning_rate": 1e-05,
      "loss": 0.1145,
      "step": 2140
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 3.1921470165252686,
      "step": 2140
    },
    {
      "dynamics_loss_avg": 0.08132025357335806,
      "epoch": 8.1634521484375e-06,
      "step": 2140
    },
    {
      "action_loss_avg": 0.05137694142758846,
      "epoch": 8.1634521484375e-06,
      "step": 2140
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 2140,
      "training_step_time": 9.974236726760864
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 3.140117645263672,
      "step": 2140
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 2140,
      "training_step_time": 9.916739225387573
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 3.1374711990356445,
      "step": 2140
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 2140,
      "training_step_time": 9.912342309951782
    },
    {
      "epoch": 8.1634521484375e-06,
      "model_forward_time": 3.1378753185272217,
      "step": 2140
    },
    {
      "epoch": 8.1634521484375e-06,
      "step": 2140,
      "training_step_time": 10.265469789505005
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 3.438032627105713,
      "step": 2141
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 2141,
      "training_step_time": 10.21334719657898
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 3.141714334487915,
      "step": 2141
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 2141,
      "training_step_time": 9.916162014007568
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 3.1351726055145264,
      "step": 2141
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 2141,
      "training_step_time": 9.908554553985596
    },
    {
      "epoch": 8.167266845703125e-06,
      "model_forward_time": 3.1393072605133057,
      "step": 2141
    },
    {
      "epoch": 8.167266845703125e-06,
      "step": 2141,
      "training_step_time": 10.257737159729004
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 3.1769657135009766,
      "step": 2142
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 2142,
      "training_step_time": 9.955750226974487
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 3.1388447284698486,
      "step": 2142
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 2142,
      "training_step_time": 10.199934959411621
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 3.1387217044830322,
      "step": 2142
    },
    {
      "dynamics_loss_avg": 0.061918864212930205,
      "epoch": 8.17108154296875e-06,
      "step": 2142
    },
    {
      "action_loss_avg": 0.045953438989818095,
      "epoch": 8.17108154296875e-06,
      "step": 2142
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 2142,
      "training_step_time": 9.91894006729126
    },
    {
      "epoch": 8.17108154296875e-06,
      "model_forward_time": 3.14330792427063,
      "step": 2142
    },
    {
      "epoch": 8.17108154296875e-06,
      "step": 2142,
      "training_step_time": 10.308959484100342
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 3.176740884780884,
      "step": 2143
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 2143,
      "training_step_time": 9.955029964447021
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 3.140119791030884,
      "step": 2143
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 2143,
      "training_step_time": 9.916422843933105
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 3.4027323722839355,
      "step": 2143
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 2143,
      "training_step_time": 10.178720951080322
    },
    {
      "epoch": 8.174896240234374e-06,
      "model_forward_time": 3.143536329269409,
      "step": 2143
    },
    {
      "epoch": 8.174896240234374e-06,
      "step": 2143,
      "training_step_time": 10.326547384262085
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 3.1834380626678467,
      "step": 2144
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 2144,
      "training_step_time": 9.972095727920532
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 3.1308276653289795,
      "step": 2144
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 2144,
      "training_step_time": 9.908555507659912
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 3.1366446018218994,
      "step": 2144
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 2144,
      "training_step_time": 9.911540985107422
    },
    {
      "epoch": 8.1787109375e-06,
      "model_forward_time": 3.1324124336242676,
      "step": 2144
    },
    {
      "epoch": 8.1787109375e-06,
      "step": 2144,
      "training_step_time": 10.278775215148926
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 3.1560583114624023,
      "step": 2145
    },
    {
      "dynamics_loss_avg": 0.07302389331161976,
      "epoch": 8.182525634765625e-06,
      "step": 2145
    },
    {
      "action_loss_avg": 0.0404413815587759,
      "epoch": 8.182525634765625e-06,
      "step": 2145
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 2145,
      "training_step_time": 9.933817863464355
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 3.1308345794677734,
      "step": 2145
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 2145,
      "training_step_time": 9.906662940979004
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 3.1348791122436523,
      "step": 2145
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 2145,
      "training_step_time": 9.911760807037354
    },
    {
      "epoch": 8.182525634765625e-06,
      "model_forward_time": 3.1325151920318604,
      "step": 2145
    },
    {
      "epoch": 8.182525634765625e-06,
      "step": 2145,
      "training_step_time": 10.2750825881958
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 3.454183340072632,
      "step": 2146
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 2146,
      "training_step_time": 10.225430727005005
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 3.127016305923462,
      "step": 2146
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 2146,
      "training_step_time": 9.898240566253662
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 3.1312272548675537,
      "step": 2146
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 2146,
      "training_step_time": 9.89999771118164
    },
    {
      "epoch": 8.18634033203125e-06,
      "model_forward_time": 3.1282496452331543,
      "step": 2146
    },
    {
      "epoch": 8.18634033203125e-06,
      "step": 2146,
      "training_step_time": 10.278653621673584
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 3.158217191696167,
      "step": 2147
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 2147,
      "training_step_time": 9.933293581008911
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 3.130608320236206,
      "step": 2147
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 2147,
      "training_step_time": 10.200021266937256
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 3.135956287384033,
      "step": 2147
    },
    {
      "dynamics_loss_avg": 0.055102807097136976,
      "epoch": 8.190155029296874e-06,
      "step": 2147
    },
    {
      "action_loss_avg": 0.034844955708831546,
      "epoch": 8.190155029296874e-06,
      "step": 2147
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 2147,
      "training_step_time": 9.920821905136108
    },
    {
      "epoch": 8.190155029296874e-06,
      "model_forward_time": 3.1354095935821533,
      "step": 2147
    },
    {
      "epoch": 8.190155029296874e-06,
      "step": 2147,
      "training_step_time": 10.295951843261719
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 3.1737215518951416,
      "step": 2148
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 2148,
      "training_step_time": 9.956401586532593
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 3.1457927227020264,
      "step": 2148
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 2148,
      "training_step_time": 9.924576997756958
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 3.422106981277466,
      "step": 2148
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 2148,
      "training_step_time": 10.197307586669922
    },
    {
      "epoch": 8.1939697265625e-06,
      "model_forward_time": 3.1355278491973877,
      "step": 2148
    },
    {
      "epoch": 8.1939697265625e-06,
      "step": 2148,
      "training_step_time": 10.016830682754517
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 3.1656742095947266,
      "step": 2149
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 2149,
      "training_step_time": 9.943198680877686
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 3.1327826976776123,
      "step": 2149
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 2149,
      "training_step_time": 9.911023139953613
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 3.131653070449829,
      "step": 2149
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 2149,
      "training_step_time": 9.909111738204956
    },
    {
      "epoch": 8.197784423828125e-06,
      "model_forward_time": 3.390024185180664,
      "step": 2149
    },
    {
      "epoch": 8.197784423828125e-06,
      "step": 2149,
      "training_step_time": 10.259861946105957
    },
    {
      "epoch": 8.20159912109375e-06,
      "grad_norm": 1.0011119842529297,
      "learning_rate": 1e-05,
      "loss": 0.1107,
      "step": 2150
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 3.1677768230438232,
      "step": 2150
    },
    {
      "dynamics_loss_avg": 0.07755673825740814,
      "epoch": 8.20159912109375e-06,
      "step": 2150
    },
    {
      "action_loss_avg": 0.055216458067297934,
      "epoch": 8.20159912109375e-06,
      "step": 2150
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 2150,
      "training_step_time": 9.943432092666626
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 3.1333653926849365,
      "step": 2150
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 2150,
      "training_step_time": 9.907291650772095
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 3.1307015419006348,
      "step": 2150
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 2150,
      "training_step_time": 9.903169870376587
    },
    {
      "epoch": 8.20159912109375e-06,
      "model_forward_time": 3.138765335083008,
      "step": 2150
    },
    {
      "epoch": 8.20159912109375e-06,
      "step": 2150,
      "training_step_time": 10.264659881591797
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 3.422962188720703,
      "step": 2151
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 2151,
      "training_step_time": 10.201871633529663
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 3.1352994441986084,
      "step": 2151
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 2151,
      "training_step_time": 9.909538269042969
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 3.136312961578369,
      "step": 2151
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 2151,
      "training_step_time": 9.910314321517944
    },
    {
      "epoch": 8.205413818359374e-06,
      "model_forward_time": 3.136730909347534,
      "step": 2151
    },
    {
      "epoch": 8.205413818359374e-06,
      "step": 2151,
      "training_step_time": 10.268798589706421
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 3.17851185798645,
      "step": 2152
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 2152,
      "training_step_time": 9.95297908782959
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 3.419095516204834,
      "step": 2152
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 2152,
      "training_step_time": 10.196188926696777
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 3.1308391094207764,
      "step": 2152
    },
    {
      "dynamics_loss_avg": 0.06840731091797352,
      "epoch": 8.209228515625e-06,
      "step": 2152
    },
    {
      "action_loss_avg": 0.03693390050902963,
      "epoch": 8.209228515625e-06,
      "step": 2152
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 2152,
      "training_step_time": 9.911796569824219
    },
    {
      "epoch": 8.209228515625e-06,
      "model_forward_time": 3.137590169906616,
      "step": 2152
    },
    {
      "epoch": 8.209228515625e-06,
      "step": 2152,
      "training_step_time": 10.313251972198486
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 3.181584596633911,
      "step": 2153
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 2153,
      "training_step_time": 9.959660768508911
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 3.1318130493164062,
      "step": 2153
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 2153,
      "training_step_time": 9.90605092048645
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 3.134885311126709,
      "step": 2153
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 2153,
      "training_step_time": 10.1890869140625
    },
    {
      "epoch": 8.213043212890625e-06,
      "model_forward_time": 3.1340548992156982,
      "step": 2153
    },
    {
      "epoch": 8.213043212890625e-06,
      "step": 2153,
      "training_step_time": 10.006625890731812
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 3.174586296081543,
      "step": 2154
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 2154,
      "training_step_time": 9.945979833602905
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 3.140211343765259,
      "step": 2154
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 2154,
      "training_step_time": 9.914773941040039
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 3.147644281387329,
      "step": 2154
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 2154,
      "training_step_time": 9.923348665237427
    },
    {
      "epoch": 8.21685791015625e-06,
      "model_forward_time": 3.471949577331543,
      "step": 2154
    },
    {
      "epoch": 8.21685791015625e-06,
      "step": 2154,
      "training_step_time": 10.344138383865356
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 3.1772141456604004,
      "step": 2155
    },
    {
      "dynamics_loss_avg": 0.07977724783122539,
      "epoch": 8.220672607421874e-06,
      "step": 2155
    },
    {
      "action_loss_avg": 0.053123315796256065,
      "epoch": 8.220672607421874e-06,
      "step": 2155
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 2155,
      "training_step_time": 9.958848237991333
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 3.1427650451660156,
      "step": 2155
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 2155,
      "training_step_time": 9.917432069778442
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 3.1463735103607178,
      "step": 2155
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 2155,
      "training_step_time": 9.917752504348755
    },
    {
      "epoch": 8.220672607421874e-06,
      "model_forward_time": 3.1406238079071045,
      "step": 2155
    },
    {
      "epoch": 8.220672607421874e-06,
      "step": 2155,
      "training_step_time": 10.270890474319458
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 2.4838173389434814,
      "step": 2156
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 2156,
      "training_step_time": 7.513680458068848
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 3.391396999359131,
      "step": 2156
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 2156,
      "training_step_time": 10.166189432144165
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 3.1423840522766113,
      "step": 2156
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 2156,
      "training_step_time": 9.920339107513428
    },
    {
      "epoch": 8.2244873046875e-06,
      "model_forward_time": 3.1419596672058105,
      "step": 2156
    },
    {
      "epoch": 8.2244873046875e-06,
      "step": 2156,
      "training_step_time": 10.323178052902222
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 3.1860995292663574,
      "step": 2157
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 2157,
      "training_step_time": 9.961907625198364
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 3.441480875015259,
      "step": 2157
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 2157,
      "training_step_time": 10.217992782592773
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 3.1438891887664795,
      "step": 2157
    },
    {
      "dynamics_loss_avg": 0.0816668039187789,
      "epoch": 8.228302001953125e-06,
      "step": 2157
    },
    {
      "action_loss_avg": 0.05250594578683376,
      "epoch": 8.228302001953125e-06,
      "step": 2157
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 2157,
      "training_step_time": 9.9265775680542
    },
    {
      "epoch": 8.228302001953125e-06,
      "model_forward_time": 3.1440017223358154,
      "step": 2157
    },
    {
      "epoch": 8.228302001953125e-06,
      "step": 2157,
      "training_step_time": 10.276207685470581
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 3.1840147972106934,
      "step": 2158
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 2158,
      "training_step_time": 9.971378803253174
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 3.1517748832702637,
      "step": 2158
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 2158,
      "training_step_time": 9.93293571472168
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 3.1464266777038574,
      "step": 2158
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 2158,
      "training_step_time": 10.197621583938599
    },
    {
      "epoch": 8.23211669921875e-06,
      "model_forward_time": 3.146167039871216,
      "step": 2158
    },
    {
      "epoch": 8.23211669921875e-06,
      "step": 2158,
      "training_step_time": 10.295429229736328
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 3.167548418045044,
      "step": 2159
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 2159,
      "training_step_time": 9.949480772018433
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 3.14383602142334,
      "step": 2159
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 2159,
      "training_step_time": 9.924680233001709
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 3.1494221687316895,
      "step": 2159
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 2159,
      "training_step_time": 9.928775310516357
    },
    {
      "epoch": 8.235931396484374e-06,
      "model_forward_time": 3.3866801261901855,
      "step": 2159
    },
    {
      "epoch": 8.235931396484374e-06,
      "step": 2159,
      "training_step_time": 10.254549503326416
    },
    {
      "epoch": 8.23974609375e-06,
      "grad_norm": 0.9157759547233582,
      "learning_rate": 1e-05,
      "loss": 0.1124,
      "step": 2160
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 3.1736562252044678,
      "step": 2160
    },
    {
      "dynamics_loss_avg": 0.08125748466700315,
      "epoch": 8.23974609375e-06,
      "step": 2160
    },
    {
      "action_loss_avg": 0.049238013289868834,
      "epoch": 8.23974609375e-06,
      "step": 2160
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 2160,
      "training_step_time": 9.951139688491821
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 3.136448383331299,
      "step": 2160
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 2160,
      "training_step_time": 9.908014297485352
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 3.1384851932525635,
      "step": 2160
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 2160,
      "training_step_time": 9.914373874664307
    },
    {
      "epoch": 8.23974609375e-06,
      "model_forward_time": 3.146709442138672,
      "step": 2160
    },
    {
      "epoch": 8.23974609375e-06,
      "step": 2160,
      "training_step_time": 10.022732257843018
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 3.4364070892333984,
      "step": 2161
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 2161,
      "training_step_time": 10.215389728546143
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 3.1435375213623047,
      "step": 2161
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 2161,
      "training_step_time": 9.922773122787476
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 3.141392230987549,
      "step": 2161
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 2161,
      "training_step_time": 9.924000978469849
    },
    {
      "epoch": 8.243560791015625e-06,
      "model_forward_time": 3.14471697807312,
      "step": 2161
    },
    {
      "epoch": 8.243560791015625e-06,
      "step": 2161,
      "training_step_time": 10.296859979629517
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 3.176326274871826,
      "step": 2162
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 2162,
      "training_step_time": 9.959758520126343
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 3.4128637313842773,
      "step": 2162
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 2162,
      "training_step_time": 10.190783262252808
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 3.139807939529419,
      "step": 2162
    },
    {
      "dynamics_loss_avg": 0.0636377926915884,
      "epoch": 8.24737548828125e-06,
      "step": 2162
    },
    {
      "action_loss_avg": 0.031185501348227264,
      "epoch": 8.24737548828125e-06,
      "step": 2162
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 2162,
      "training_step_time": 9.92009687423706
    },
    {
      "epoch": 8.24737548828125e-06,
      "model_forward_time": 3.139676570892334,
      "step": 2162
    },
    {
      "epoch": 8.24737548828125e-06,
      "step": 2162,
      "training_step_time": 10.303662538528442
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 3.17134428024292,
      "step": 2163
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 2163,
      "training_step_time": 9.951217412948608
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 3.1364526748657227,
      "step": 2163
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 2163,
      "training_step_time": 9.915297746658325
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 3.421342611312866,
      "step": 2163
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 2163,
      "training_step_time": 10.21232533454895
    },
    {
      "epoch": 8.251190185546874e-06,
      "model_forward_time": 3.149168014526367,
      "step": 2163
    },
    {
      "epoch": 8.251190185546874e-06,
      "step": 2163,
      "training_step_time": 10.301661968231201
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 3.176384925842285,
      "step": 2164
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 2164,
      "training_step_time": 9.957109928131104
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 3.141677141189575,
      "step": 2164
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 2164,
      "training_step_time": 9.920821189880371
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 3.1423957347869873,
      "step": 2164
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 2164,
      "training_step_time": 9.919208526611328
    },
    {
      "epoch": 8.2550048828125e-06,
      "model_forward_time": 3.4212749004364014,
      "step": 2164
    },
    {
      "epoch": 8.2550048828125e-06,
      "step": 2164,
      "training_step_time": 10.298553705215454
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 3.175225019454956,
      "step": 2165
    },
    {
      "dynamics_loss_avg": 0.0740026168525219,
      "epoch": 8.258819580078125e-06,
      "step": 2165
    },
    {
      "action_loss_avg": 0.049712717346847055,
      "epoch": 8.258819580078125e-06,
      "step": 2165
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 2165,
      "training_step_time": 9.956928730010986
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 3.1393110752105713,
      "step": 2165
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 2165,
      "training_step_time": 9.915977478027344
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 3.144226551055908,
      "step": 2165
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 2165,
      "training_step_time": 9.922642946243286
    },
    {
      "epoch": 8.258819580078125e-06,
      "model_forward_time": 3.1390650272369385,
      "step": 2165
    },
    {
      "epoch": 8.258819580078125e-06,
      "step": 2165,
      "training_step_time": 10.039241790771484
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 3.4549107551574707,
      "step": 2166
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 2166,
      "training_step_time": 10.236600637435913
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 3.1414871215820312,
      "step": 2166
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 2166,
      "training_step_time": 9.932742595672607
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 3.140364408493042,
      "step": 2166
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 2166,
      "training_step_time": 9.925542116165161
    },
    {
      "epoch": 8.26263427734375e-06,
      "model_forward_time": 3.1430606842041016,
      "step": 2166
    },
    {
      "epoch": 8.26263427734375e-06,
      "step": 2166,
      "training_step_time": 10.32151198387146
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 3.1896860599517822,
      "step": 2167
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 2167,
      "training_step_time": 9.968644618988037
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 3.391608953475952,
      "step": 2167
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 2167,
      "training_step_time": 10.16829776763916
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 3.137014150619507,
      "step": 2167
    },
    {
      "dynamics_loss_avg": 0.05858709421008825,
      "epoch": 8.266448974609374e-06,
      "step": 2167
    },
    {
      "action_loss_avg": 0.033173383213579655,
      "epoch": 8.266448974609374e-06,
      "step": 2167
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 2167,
      "training_step_time": 9.91176986694336
    },
    {
      "epoch": 8.266448974609374e-06,
      "model_forward_time": 3.143867015838623,
      "step": 2167
    },
    {
      "epoch": 8.266448974609374e-06,
      "step": 2167,
      "training_step_time": 10.299880504608154
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 3.1707465648651123,
      "step": 2168
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 2168,
      "training_step_time": 9.943627834320068
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 3.1359565258026123,
      "step": 2168
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 2168,
      "training_step_time": 9.907955169677734
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 3.4182136058807373,
      "step": 2168
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 2168,
      "training_step_time": 10.194257497787476
    },
    {
      "epoch": 8.270263671875e-06,
      "model_forward_time": 3.144183874130249,
      "step": 2168
    },
    {
      "epoch": 8.270263671875e-06,
      "step": 2168,
      "training_step_time": 10.010915279388428
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 3.1979830265045166,
      "step": 2169
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 2169,
      "training_step_time": 9.980584859848022
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 3.1479036808013916,
      "step": 2169
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 2169,
      "training_step_time": 9.928433656692505
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 3.144238233566284,
      "step": 2169
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 2169,
      "training_step_time": 9.924623727798462
    },
    {
      "epoch": 8.274078369140625e-06,
      "model_forward_time": 3.4395930767059326,
      "step": 2169
    },
    {
      "epoch": 8.274078369140625e-06,
      "step": 2169,
      "training_step_time": 10.313862800598145
    },
    {
      "epoch": 8.27789306640625e-06,
      "grad_norm": 0.891907274723053,
      "learning_rate": 1e-05,
      "loss": 0.1078,
      "step": 2170
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 3.1844217777252197,
      "step": 2170
    },
    {
      "dynamics_loss_avg": 0.06497552804648876,
      "epoch": 8.27789306640625e-06,
      "step": 2170
    },
    {
      "action_loss_avg": 0.03677626363933086,
      "epoch": 8.27789306640625e-06,
      "step": 2170
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 2170,
      "training_step_time": 9.971996545791626
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 3.1499733924865723,
      "step": 2170
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 2170,
      "training_step_time": 9.938896894454956
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 3.1488583087921143,
      "step": 2170
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 2170,
      "training_step_time": 9.9414701461792
    },
    {
      "epoch": 8.27789306640625e-06,
      "model_forward_time": 3.1534109115600586,
      "step": 2170
    },
    {
      "epoch": 8.27789306640625e-06,
      "step": 2170,
      "training_step_time": 10.311014175415039
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 3.437410831451416,
      "step": 2171
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 2171,
      "training_step_time": 10.216907024383545
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 3.145761251449585,
      "step": 2171
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 2171,
      "training_step_time": 9.922133445739746
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 3.1469931602478027,
      "step": 2171
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 2171,
      "training_step_time": 9.921530485153198
    },
    {
      "epoch": 8.281707763671874e-06,
      "model_forward_time": 3.1404428482055664,
      "step": 2171
    },
    {
      "epoch": 8.281707763671874e-06,
      "step": 2171,
      "training_step_time": 10.022092580795288
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 3.1761891841888428,
      "step": 2172
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 2172,
      "training_step_time": 9.94948124885559
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 3.4179141521453857,
      "step": 2172
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 2172,
      "training_step_time": 10.188391208648682
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 3.1374542713165283,
      "step": 2172
    },
    {
      "dynamics_loss_avg": 0.06195277944207191,
      "epoch": 8.2855224609375e-06,
      "step": 2172
    },
    {
      "action_loss_avg": 0.03753872336819768,
      "epoch": 8.2855224609375e-06,
      "step": 2172
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 2172,
      "training_step_time": 9.912004470825195
    },
    {
      "epoch": 8.2855224609375e-06,
      "model_forward_time": 3.1436078548431396,
      "step": 2172
    },
    {
      "epoch": 8.2855224609375e-06,
      "step": 2172,
      "training_step_time": 10.273179531097412
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 3.1842453479766846,
      "step": 2173
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 2173,
      "training_step_time": 9.955261468887329
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 2.438645839691162,
      "step": 2173
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 2173,
      "training_step_time": 7.47390604019165
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 3.3995676040649414,
      "step": 2173
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 2173,
      "training_step_time": 10.173702478408813
    },
    {
      "epoch": 8.289337158203125e-06,
      "model_forward_time": 3.145866870880127,
      "step": 2173
    },
    {
      "epoch": 8.289337158203125e-06,
      "step": 2173,
      "training_step_time": 10.016363382339478
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 3.1846959590911865,
      "step": 2174
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 2174,
      "training_step_time": 9.960476398468018
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 3.14180850982666,
      "step": 2174
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 2174,
      "training_step_time": 9.91454005241394
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 3.1442012786865234,
      "step": 2174
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 2174,
      "training_step_time": 9.920014142990112
    },
    {
      "epoch": 8.29315185546875e-06,
      "model_forward_time": 3.433593988418579,
      "step": 2174
    },
    {
      "epoch": 8.29315185546875e-06,
      "step": 2174,
      "training_step_time": 10.316532135009766
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 3.1728570461273193,
      "step": 2175
    },
    {
      "dynamics_loss_avg": 0.059543330781161787,
      "epoch": 8.296966552734374e-06,
      "step": 2175
    },
    {
      "action_loss_avg": 0.03045099610462785,
      "epoch": 8.296966552734374e-06,
      "step": 2175
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 2175,
      "training_step_time": 9.956193208694458
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 3.1305811405181885,
      "step": 2175
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 2175,
      "training_step_time": 9.908417701721191
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 3.138216495513916,
      "step": 2175
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 2175,
      "training_step_time": 9.907545328140259
    },
    {
      "epoch": 8.296966552734374e-06,
      "model_forward_time": 3.1379733085632324,
      "step": 2175
    },
    {
      "epoch": 8.296966552734374e-06,
      "step": 2175,
      "training_step_time": 9.999326467514038
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 3.4634337425231934,
      "step": 2176
    },
    {
      "epoch": 8.30078125e-06,
      "step": 2176,
      "training_step_time": 10.23220181465149
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 3.142474889755249,
      "step": 2176
    },
    {
      "epoch": 8.30078125e-06,
      "step": 2176,
      "training_step_time": 9.913267135620117
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 3.147482395172119,
      "step": 2176
    },
    {
      "epoch": 8.30078125e-06,
      "step": 2176,
      "training_step_time": 9.934354305267334
    },
    {
      "epoch": 8.30078125e-06,
      "model_forward_time": 3.141545057296753,
      "step": 2176
    },
    {
      "epoch": 8.30078125e-06,
      "step": 2176,
      "training_step_time": 10.01252293586731
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 3.1862339973449707,
      "step": 2177
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 2177,
      "training_step_time": 9.964861154556274
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 3.399515151977539,
      "step": 2177
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 2177,
      "training_step_time": 10.174351453781128
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 3.1399712562561035,
      "step": 2177
    },
    {
      "dynamics_loss_avg": 0.06739020347595215,
      "epoch": 8.304595947265625e-06,
      "step": 2177
    },
    {
      "action_loss_avg": 0.04662754163146019,
      "epoch": 8.304595947265625e-06,
      "step": 2177
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 2177,
      "training_step_time": 9.919569253921509
    },
    {
      "epoch": 8.304595947265625e-06,
      "model_forward_time": 3.1384165287017822,
      "step": 2177
    },
    {
      "epoch": 8.304595947265625e-06,
      "step": 2177,
      "training_step_time": 10.261949300765991
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 3.182229518890381,
      "step": 2178
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 2178,
      "training_step_time": 9.957537651062012
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 3.137457847595215,
      "step": 2178
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 2178,
      "training_step_time": 9.912812232971191
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 3.4086780548095703,
      "step": 2178
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 2178,
      "training_step_time": 10.186431169509888
    },
    {
      "epoch": 8.30841064453125e-06,
      "model_forward_time": 3.1374642848968506,
      "step": 2178
    },
    {
      "epoch": 8.30841064453125e-06,
      "step": 2178,
      "training_step_time": 10.013923645019531
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 3.173595905303955,
      "step": 2179
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 2179,
      "training_step_time": 9.955670595169067
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 3.1363863945007324,
      "step": 2179
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 2179,
      "training_step_time": 9.914184331893921
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 3.1327054500579834,
      "step": 2179
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 2179,
      "training_step_time": 9.908148050308228
    },
    {
      "epoch": 8.312225341796874e-06,
      "model_forward_time": 3.379560708999634,
      "step": 2179
    },
    {
      "epoch": 8.312225341796874e-06,
      "step": 2179,
      "training_step_time": 10.253847360610962
    },
    {
      "epoch": 8.3160400390625e-06,
      "grad_norm": 0.5629953742027283,
      "learning_rate": 1e-05,
      "loss": 0.1111,
      "step": 2180
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 3.189034938812256,
      "step": 2180
    },
    {
      "dynamics_loss_avg": 0.06575565971434116,
      "epoch": 8.3160400390625e-06,
      "step": 2180
    },
    {
      "action_loss_avg": 0.03686686838045716,
      "epoch": 8.3160400390625e-06,
      "step": 2180
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 2180,
      "training_step_time": 9.96677565574646
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 3.1380279064178467,
      "step": 2180
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 2180,
      "training_step_time": 9.914734363555908
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 3.141467809677124,
      "step": 2180
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 2180,
      "training_step_time": 9.913647651672363
    },
    {
      "epoch": 8.3160400390625e-06,
      "model_forward_time": 3.144951820373535,
      "step": 2180
    },
    {
      "epoch": 8.3160400390625e-06,
      "step": 2180,
      "training_step_time": 10.258607149124146
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 3.4246084690093994,
      "step": 2181
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 2181,
      "training_step_time": 10.198756694793701
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 3.1419336795806885,
      "step": 2181
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 2181,
      "training_step_time": 9.922418594360352
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 3.1422245502471924,
      "step": 2181
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 2181,
      "training_step_time": 9.918020963668823
    },
    {
      "epoch": 8.319854736328125e-06,
      "model_forward_time": 3.1450183391571045,
      "step": 2181
    },
    {
      "epoch": 8.319854736328125e-06,
      "step": 2181,
      "training_step_time": 10.0264310836792
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 3.1759755611419678,
      "step": 2182
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 2182,
      "training_step_time": 9.95272707939148
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 3.4285953044891357,
      "step": 2182
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 2182,
      "training_step_time": 10.203410387039185
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 3.1392440795898438,
      "step": 2182
    },
    {
      "dynamics_loss_avg": 0.058740238286554815,
      "epoch": 8.32366943359375e-06,
      "step": 2182
    },
    {
      "action_loss_avg": 0.02551974575035274,
      "epoch": 8.32366943359375e-06,
      "step": 2182
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 2182,
      "training_step_time": 9.91482663154602
    },
    {
      "epoch": 8.32366943359375e-06,
      "model_forward_time": 3.14031982421875,
      "step": 2182
    },
    {
      "epoch": 8.32366943359375e-06,
      "step": 2182,
      "training_step_time": 10.018723487854004
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 3.1825966835021973,
      "step": 2183
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 2183,
      "training_step_time": 9.958748579025269
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 3.144824266433716,
      "step": 2183
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 2183,
      "training_step_time": 9.919142007827759
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 3.3992953300476074,
      "step": 2183
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 2183,
      "training_step_time": 10.172335386276245
    },
    {
      "epoch": 8.327484130859374e-06,
      "model_forward_time": 3.1418139934539795,
      "step": 2183
    },
    {
      "epoch": 8.327484130859374e-06,
      "step": 2183,
      "training_step_time": 10.301254510879517
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 3.176802635192871,
      "step": 2184
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 2184,
      "training_step_time": 9.95153546333313
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 3.137462615966797,
      "step": 2184
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 2184,
      "training_step_time": 9.916048765182495
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 3.143218517303467,
      "step": 2184
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 2184,
      "training_step_time": 9.923508167266846
    },
    {
      "epoch": 8.331298828125e-06,
      "model_forward_time": 3.4123151302337646,
      "step": 2184
    },
    {
      "epoch": 8.331298828125e-06,
      "step": 2184,
      "training_step_time": 10.301963806152344
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 3.1822726726531982,
      "step": 2185
    },
    {
      "dynamics_loss_avg": 0.07905937470495701,
      "epoch": 8.335113525390625e-06,
      "step": 2185
    },
    {
      "action_loss_avg": 0.055878729559481144,
      "epoch": 8.335113525390625e-06,
      "step": 2185
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 2185,
      "training_step_time": 9.965739011764526
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 3.142904758453369,
      "step": 2185
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 2185,
      "training_step_time": 9.922993659973145
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 3.142246723175049,
      "step": 2185
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 2185,
      "training_step_time": 9.921014785766602
    },
    {
      "epoch": 8.335113525390625e-06,
      "model_forward_time": 3.1436309814453125,
      "step": 2185
    },
    {
      "epoch": 8.335113525390625e-06,
      "step": 2185,
      "training_step_time": 10.285058975219727
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 3.1669163703918457,
      "step": 2186
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 2186,
      "training_step_time": 9.94706678390503
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 3.141183853149414,
      "step": 2186
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 2186,
      "training_step_time": 9.923772811889648
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 3.143908739089966,
      "step": 2186
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 2186,
      "training_step_time": 9.920289039611816
    },
    {
      "epoch": 8.33892822265625e-06,
      "model_forward_time": 3.139051675796509,
      "step": 2186
    },
    {
      "epoch": 8.33892822265625e-06,
      "step": 2186,
      "training_step_time": 10.286849975585938
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 3.1746420860290527,
      "step": 2187
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 2187,
      "training_step_time": 10.223480463027954
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 3.1398167610168457,
      "step": 2187
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 2187,
      "training_step_time": 9.913241147994995
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 3.141918897628784,
      "step": 2187
    },
    {
      "dynamics_loss_avg": 0.08755248561501502,
      "epoch": 8.342742919921874e-06,
      "step": 2187
    },
    {
      "action_loss_avg": 0.047695356979966165,
      "epoch": 8.342742919921874e-06,
      "step": 2187
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 2187,
      "training_step_time": 9.920620203018188
    },
    {
      "epoch": 8.342742919921874e-06,
      "model_forward_time": 3.1378588676452637,
      "step": 2187
    },
    {
      "epoch": 8.342742919921874e-06,
      "step": 2187,
      "training_step_time": 10.298199892044067
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 3.1842691898345947,
      "step": 2188
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 2188,
      "training_step_time": 9.959855079650879
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 3.3942835330963135,
      "step": 2188
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 2188,
      "training_step_time": 10.163215398788452
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 3.139190673828125,
      "step": 2188
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 2188,
      "training_step_time": 9.913866758346558
    },
    {
      "epoch": 8.3465576171875e-06,
      "model_forward_time": 3.1409811973571777,
      "step": 2188
    },
    {
      "epoch": 8.3465576171875e-06,
      "step": 2188,
      "training_step_time": 10.289483547210693
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 3.1810295581817627,
      "step": 2189
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 2189,
      "training_step_time": 9.959213972091675
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 3.1434364318847656,
      "step": 2189
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 2189,
      "training_step_time": 9.95268201828003
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 3.4198479652404785,
      "step": 2189
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 2189,
      "training_step_time": 10.193503618240356
    },
    {
      "epoch": 8.350372314453125e-06,
      "model_forward_time": 3.137446403503418,
      "step": 2189
    },
    {
      "epoch": 8.350372314453125e-06,
      "step": 2189,
      "training_step_time": 10.020454168319702
    },
    {
      "epoch": 8.35418701171875e-06,
      "grad_norm": 0.5540801882743835,
      "learning_rate": 1e-05,
      "loss": 0.1108,
      "step": 2190
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 3.183180093765259,
      "step": 2190
    },
    {
      "dynamics_loss_avg": 0.08138442300260067,
      "epoch": 8.35418701171875e-06,
      "step": 2190
    },
    {
      "action_loss_avg": 0.05031943190842867,
      "epoch": 8.35418701171875e-06,
      "step": 2190
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 2190,
      "training_step_time": 9.961906909942627
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 3.148117780685425,
      "step": 2190
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 2190,
      "training_step_time": 9.926512718200684
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 3.1416797637939453,
      "step": 2190
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 2190,
      "training_step_time": 9.918986797332764
    },
    {
      "epoch": 8.35418701171875e-06,
      "model_forward_time": 3.1439855098724365,
      "step": 2190
    },
    {
      "epoch": 8.35418701171875e-06,
      "step": 2190,
      "training_step_time": 10.296372413635254
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 3.182715654373169,
      "step": 2191
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 2191,
      "training_step_time": 9.960718154907227
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 3.1454081535339355,
      "step": 2191
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 2191,
      "training_step_time": 9.923750877380371
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 3.139268159866333,
      "step": 2191
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 2191,
      "training_step_time": 9.9156494140625
    },
    {
      "epoch": 8.358001708984374e-06,
      "model_forward_time": 3.142632484436035,
      "step": 2191
    },
    {
      "epoch": 8.358001708984374e-06,
      "step": 2191,
      "training_step_time": 10.252103805541992
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 3.445024251937866,
      "step": 2192
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 2192,
      "training_step_time": 10.21819019317627
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 3.1417860984802246,
      "step": 2192
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 2192,
      "training_step_time": 9.920076608657837
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 3.138296127319336,
      "step": 2192
    },
    {
      "dynamics_loss_avg": 0.05858609564602375,
      "epoch": 8.36181640625e-06,
      "step": 2192
    },
    {
      "action_loss_avg": 0.04081212524324655,
      "epoch": 8.36181640625e-06,
      "step": 2192
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 2192,
      "training_step_time": 9.917092084884644
    },
    {
      "epoch": 8.36181640625e-06,
      "model_forward_time": 3.138810396194458,
      "step": 2192
    },
    {
      "epoch": 8.36181640625e-06,
      "step": 2192,
      "training_step_time": 10.322220802307129
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 3.1966543197631836,
      "step": 2193
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 2193,
      "training_step_time": 9.978418111801147
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 3.135927677154541,
      "step": 2193
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 2193,
      "training_step_time": 10.191096305847168
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 3.134033679962158,
      "step": 2193
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 2193,
      "training_step_time": 9.907270669937134
    },
    {
      "epoch": 8.365631103515625e-06,
      "model_forward_time": 3.136679172515869,
      "step": 2193
    },
    {
      "epoch": 8.365631103515625e-06,
      "step": 2193,
      "training_step_time": 10.273675203323364
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 3.1677212715148926,
      "step": 2194
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 2194,
      "training_step_time": 9.944979667663574
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 3.133937120437622,
      "step": 2194
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 2194,
      "training_step_time": 9.912317514419556
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 3.4038026332855225,
      "step": 2194
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 2194,
      "training_step_time": 10.183370113372803
    },
    {
      "epoch": 8.36944580078125e-06,
      "model_forward_time": 3.1442670822143555,
      "step": 2194
    },
    {
      "epoch": 8.36944580078125e-06,
      "step": 2194,
      "training_step_time": 10.020926475524902
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 3.165724277496338,
      "step": 2195
    },
    {
      "dynamics_loss_avg": 0.07689273469150067,
      "epoch": 8.373260498046874e-06,
      "step": 2195
    },
    {
      "action_loss_avg": 0.04153766445815563,
      "epoch": 8.373260498046874e-06,
      "step": 2195
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 2195,
      "training_step_time": 9.9436936378479
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 3.1354079246520996,
      "step": 2195
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 2195,
      "training_step_time": 9.912278890609741
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 3.1386117935180664,
      "step": 2195
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 2195,
      "training_step_time": 9.916008234024048
    },
    {
      "epoch": 8.373260498046874e-06,
      "model_forward_time": 3.1371710300445557,
      "step": 2195
    },
    {
      "epoch": 8.373260498046874e-06,
      "step": 2195,
      "training_step_time": 10.287962198257446
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 3.181150197982788,
      "step": 2196
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 2196,
      "training_step_time": 9.959481000900269
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 3.1371002197265625,
      "step": 2196
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 2196,
      "training_step_time": 9.913943767547607
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 3.139951229095459,
      "step": 2196
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 2196,
      "training_step_time": 9.91365361213684
    },
    {
      "epoch": 8.3770751953125e-06,
      "model_forward_time": 3.1348464488983154,
      "step": 2196
    },
    {
      "epoch": 8.3770751953125e-06,
      "step": 2196,
      "training_step_time": 10.29539966583252
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 3.449571371078491,
      "step": 2197
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 2197,
      "training_step_time": 10.224271297454834
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 3.133319139480591,
      "step": 2197
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 2197,
      "training_step_time": 9.910290956497192
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 3.139404535293579,
      "step": 2197
    },
    {
      "dynamics_loss_avg": 0.07229305095970631,
      "epoch": 8.380889892578125e-06,
      "step": 2197
    },
    {
      "action_loss_avg": 0.04796709325164557,
      "epoch": 8.380889892578125e-06,
      "step": 2197
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 2197,
      "training_step_time": 9.91941499710083
    },
    {
      "epoch": 8.380889892578125e-06,
      "model_forward_time": 3.1371798515319824,
      "step": 2197
    },
    {
      "epoch": 8.380889892578125e-06,
      "step": 2197,
      "training_step_time": 10.28468656539917
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 3.188554286956787,
      "step": 2198
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 2198,
      "training_step_time": 9.965103387832642
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 3.3878509998321533,
      "step": 2198
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 2198,
      "training_step_time": 10.166565895080566
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 3.143174886703491,
      "step": 2198
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 2198,
      "training_step_time": 9.920944452285767
    },
    {
      "epoch": 8.38470458984375e-06,
      "model_forward_time": 3.137145519256592,
      "step": 2198
    },
    {
      "epoch": 8.38470458984375e-06,
      "step": 2198,
      "training_step_time": 10.27014684677124
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 3.175096273422241,
      "step": 2199
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 2199,
      "training_step_time": 9.952998399734497
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 3.1370747089385986,
      "step": 2199
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 2199,
      "training_step_time": 9.913604021072388
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 3.4137685298919678,
      "step": 2199
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 2199,
      "training_step_time": 10.191429138183594
    },
    {
      "epoch": 8.388519287109374e-06,
      "model_forward_time": 3.141179323196411,
      "step": 2199
    },
    {
      "epoch": 8.388519287109374e-06,
      "step": 2199,
      "training_step_time": 10.277461290359497
    },
    {
      "epoch": 8.392333984375e-06,
      "grad_norm": 0.41520288586616516,
      "learning_rate": 1e-05,
      "loss": 0.1073,
      "step": 2200
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 3.1919667720794678,
      "step": 2200
    },
    {
      "dynamics_loss_avg": 0.07974987849593163,
      "epoch": 8.392333984375e-06,
      "step": 2200
    },
    {
      "action_loss_avg": 0.04678989518433809,
      "epoch": 8.392333984375e-06,
      "step": 2200
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 2200,
      "training_step_time": 9.980777025222778
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 3.1427838802337646,
      "step": 2200
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 2200,
      "training_step_time": 9.922169208526611
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 3.1379501819610596,
      "step": 2200
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 2200,
      "training_step_time": 9.910134077072144
    },
    {
      "epoch": 8.392333984375e-06,
      "model_forward_time": 3.3842999935150146,
      "step": 2200
    },
    {
      "epoch": 8.392333984375e-06,
      "step": 2200,
      "training_step_time": 10.254894018173218
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 3.1623759269714355,
      "step": 2201
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 2201,
      "training_step_time": 9.937236547470093
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 2.432260274887085,
      "step": 2201
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 2201,
      "training_step_time": 7.461284399032593
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 3.1339964866638184,
      "step": 2201
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 2201,
      "training_step_time": 9.907471656799316
    },
    {
      "epoch": 8.396148681640625e-06,
      "model_forward_time": 3.1376092433929443,
      "step": 2201
    },
    {
      "epoch": 8.396148681640625e-06,
      "step": 2201,
      "training_step_time": 10.322190046310425
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 3.444284200668335,
      "step": 2202
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 2202,
      "training_step_time": 10.221364259719849
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 2.4337313175201416,
      "step": 2202
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 2202,
      "training_step_time": 7.46080756187439
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 3.132420301437378,
      "step": 2202
    },
    {
      "dynamics_loss_avg": 0.08127909880131483,
      "epoch": 8.39996337890625e-06,
      "step": 2202
    },
    {
      "action_loss_avg": 0.051441779173910615,
      "epoch": 8.39996337890625e-06,
      "step": 2202
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 2202,
      "training_step_time": 9.904198169708252
    },
    {
      "epoch": 8.39996337890625e-06,
      "model_forward_time": 3.1376450061798096,
      "step": 2202
    },
    {
      "epoch": 8.39996337890625e-06,
      "step": 2202,
      "training_step_time": 10.280547857284546
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 3.1515419483184814,
      "step": 2203
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 2203,
      "training_step_time": 9.928406953811646
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 3.4161391258239746,
      "step": 2203
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 2203,
      "training_step_time": 10.185847520828247
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 2.4378318786621094,
      "step": 2203
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 2203,
      "training_step_time": 7.461914539337158
    },
    {
      "epoch": 8.403778076171874e-06,
      "model_forward_time": 3.133310556411743,
      "step": 2203
    },
    {
      "epoch": 8.403778076171874e-06,
      "step": 2203,
      "training_step_time": 10.015663623809814
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 3.1838130950927734,
      "step": 2204
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 2204,
      "training_step_time": 9.95559811592102
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 3.1339807510375977,
      "step": 2204
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 2204,
      "training_step_time": 9.906390905380249
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 3.1392412185668945,
      "step": 2204
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 2204,
      "training_step_time": 9.91930866241455
    },
    {
      "epoch": 8.4075927734375e-06,
      "model_forward_time": 3.409477472305298,
      "step": 2204
    },
    {
      "epoch": 8.4075927734375e-06,
      "step": 2204,
      "training_step_time": 10.310141324996948
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 3.1870598793029785,
      "step": 2205
    },
    {
      "dynamics_loss_avg": 0.08201497755944728,
      "epoch": 8.411407470703125e-06,
      "step": 2205
    },
    {
      "action_loss_avg": 0.051903379149734974,
      "epoch": 8.411407470703125e-06,
      "step": 2205
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 2205,
      "training_step_time": 9.96813178062439
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 3.1387875080108643,
      "step": 2205
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 2205,
      "training_step_time": 9.917713165283203
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 3.1435015201568604,
      "step": 2205
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 2205,
      "training_step_time": 9.929054975509644
    },
    {
      "epoch": 8.411407470703125e-06,
      "model_forward_time": 3.426490306854248,
      "step": 2205
    },
    {
      "epoch": 8.411407470703125e-06,
      "step": 2205,
      "training_step_time": 10.309049844741821
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 3.19344425201416,
      "step": 2206
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 2206,
      "training_step_time": 9.9760422706604
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 3.1454079151153564,
      "step": 2206
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 2206,
      "training_step_time": 9.924789905548096
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 3.1491498947143555,
      "step": 2206
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 2206,
      "training_step_time": 9.928253173828125
    },
    {
      "epoch": 8.41522216796875e-06,
      "model_forward_time": 3.1442723274230957,
      "step": 2206
    },
    {
      "epoch": 8.41522216796875e-06,
      "step": 2206,
      "training_step_time": 10.02029824256897
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 3.1982195377349854,
      "step": 2207
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 2207,
      "training_step_time": 10.233561038970947
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 3.144320249557495,
      "step": 2207
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 2207,
      "training_step_time": 9.918090105056763
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 3.1419873237609863,
      "step": 2207
    },
    {
      "dynamics_loss_avg": 0.06055802870541811,
      "epoch": 8.419036865234374e-06,
      "step": 2207
    },
    {
      "action_loss_avg": 0.046865079551935196,
      "epoch": 8.419036865234374e-06,
      "step": 2207
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 2207,
      "training_step_time": 9.915380477905273
    },
    {
      "epoch": 8.419036865234374e-06,
      "model_forward_time": 3.1382853984832764,
      "step": 2207
    },
    {
      "epoch": 8.419036865234374e-06,
      "step": 2207,
      "training_step_time": 10.291707754135132
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 3.188936948776245,
      "step": 2208
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 2208,
      "training_step_time": 9.962246417999268
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 3.3996288776397705,
      "step": 2208
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 2208,
      "training_step_time": 10.170629262924194
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 3.144803762435913,
      "step": 2208
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 2208,
      "training_step_time": 9.91652798652649
    },
    {
      "epoch": 8.4228515625e-06,
      "model_forward_time": 2.440310478210449,
      "step": 2208
    },
    {
      "epoch": 8.4228515625e-06,
      "step": 2208,
      "training_step_time": 10.314524412155151
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 3.186246156692505,
      "step": 2209
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 2209,
      "training_step_time": 9.95592737197876
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 3.146301746368408,
      "step": 2209
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 2209,
      "training_step_time": 9.91854190826416
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 3.461700201034546,
      "step": 2209
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 2209,
      "training_step_time": 10.236510515213013
    },
    {
      "epoch": 8.426666259765625e-06,
      "model_forward_time": 3.1462392807006836,
      "step": 2209
    },
    {
      "epoch": 8.426666259765625e-06,
      "step": 2209,
      "training_step_time": 10.295081615447998
    },
    {
      "epoch": 8.43048095703125e-06,
      "grad_norm": 0.5359917283058167,
      "learning_rate": 1e-05,
      "loss": 0.11,
      "step": 2210
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 3.1986756324768066,
      "step": 2210
    },
    {
      "dynamics_loss_avg": 0.053987279534339905,
      "epoch": 8.43048095703125e-06,
      "step": 2210
    },
    {
      "action_loss_avg": 0.03985393280163407,
      "epoch": 8.43048095703125e-06,
      "step": 2210
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 2210,
      "training_step_time": 9.975907564163208
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 3.146975040435791,
      "step": 2210
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 2210,
      "training_step_time": 9.920387029647827
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 3.145430088043213,
      "step": 2210
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 2210,
      "training_step_time": 9.921067714691162
    },
    {
      "epoch": 8.43048095703125e-06,
      "model_forward_time": 3.1495447158813477,
      "step": 2210
    },
    {
      "epoch": 8.43048095703125e-06,
      "step": 2210,
      "training_step_time": 10.30241870880127
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 3.183424472808838,
      "step": 2211
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 2211,
      "training_step_time": 9.956493377685547
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 3.1479320526123047,
      "step": 2211
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 2211,
      "training_step_time": 9.916187047958374
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 3.1438305377960205,
      "step": 2211
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 2211,
      "training_step_time": 9.917041063308716
    },
    {
      "epoch": 8.434295654296874e-06,
      "model_forward_time": 3.1460957527160645,
      "step": 2211
    },
    {
      "epoch": 8.434295654296874e-06,
      "step": 2211,
      "training_step_time": 10.013155221939087
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 3.4463024139404297,
      "step": 2212
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 2212,
      "training_step_time": 10.215364933013916
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 3.1422789096832275,
      "step": 2212
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 2212,
      "training_step_time": 9.906335353851318
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 3.1394729614257812,
      "step": 2212
    },
    {
      "dynamics_loss_avg": 0.07282418198883533,
      "epoch": 8.4381103515625e-06,
      "step": 2212
    },
    {
      "action_loss_avg": 0.045211656391620635,
      "epoch": 8.4381103515625e-06,
      "step": 2212
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 2212,
      "training_step_time": 9.912433862686157
    },
    {
      "epoch": 8.4381103515625e-06,
      "model_forward_time": 3.142474889755249,
      "step": 2212
    },
    {
      "epoch": 8.4381103515625e-06,
      "step": 2212,
      "training_step_time": 10.314487218856812
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 3.193148136138916,
      "step": 2213
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 2213,
      "training_step_time": 9.963072538375854
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 3.1382195949554443,
      "step": 2213
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 2213,
      "training_step_time": 10.178835391998291
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 3.140713930130005,
      "step": 2213
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 2213,
      "training_step_time": 9.909908056259155
    },
    {
      "epoch": 8.441925048828125e-06,
      "model_forward_time": 3.1422629356384277,
      "step": 2213
    },
    {
      "epoch": 8.441925048828125e-06,
      "step": 2213,
      "training_step_time": 10.251490116119385
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 3.1981287002563477,
      "step": 2214
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 2214,
      "training_step_time": 9.970739364624023
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 3.139291763305664,
      "step": 2214
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 2214,
      "training_step_time": 9.907948732376099
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 3.403538227081299,
      "step": 2214
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 2214,
      "training_step_time": 10.169291019439697
    },
    {
      "epoch": 8.44573974609375e-06,
      "model_forward_time": 3.143848419189453,
      "step": 2214
    },
    {
      "epoch": 8.44573974609375e-06,
      "step": 2214,
      "training_step_time": 10.300007104873657
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 3.174722909927368,
      "step": 2215
    },
    {
      "dynamics_loss_avg": 0.07782015129923821,
      "epoch": 8.449554443359374e-06,
      "step": 2215
    },
    {
      "action_loss_avg": 0.05417167618870735,
      "epoch": 8.449554443359374e-06,
      "step": 2215
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 2215,
      "training_step_time": 9.945157527923584
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 3.1407809257507324,
      "step": 2215
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 2215,
      "training_step_time": 9.9091055393219
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 3.1426398754119873,
      "step": 2215
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 2215,
      "training_step_time": 9.913358211517334
    },
    {
      "epoch": 8.449554443359374e-06,
      "model_forward_time": 3.142383575439453,
      "step": 2215
    },
    {
      "epoch": 8.449554443359374e-06,
      "step": 2215,
      "training_step_time": 10.283884286880493
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 3.1807334423065186,
      "step": 2216
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 2216,
      "training_step_time": 9.953075647354126
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 3.139099597930908,
      "step": 2216
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 2216,
      "training_step_time": 9.907985925674438
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 3.1403300762176514,
      "step": 2216
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 2216,
      "training_step_time": 9.910855531692505
    },
    {
      "epoch": 8.453369140625e-06,
      "model_forward_time": 3.142408609390259,
      "step": 2216
    },
    {
      "epoch": 8.453369140625e-06,
      "step": 2216,
      "training_step_time": 10.020917892456055
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 3.4297170639038086,
      "step": 2217
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 2217,
      "training_step_time": 10.217256546020508
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 3.132641553878784,
      "step": 2217
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 2217,
      "training_step_time": 9.898162126541138
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 3.1370291709899902,
      "step": 2217
    },
    {
      "dynamics_loss_avg": 0.07331336867064238,
      "epoch": 8.457183837890625e-06,
      "step": 2217
    },
    {
      "action_loss_avg": 0.04532734863460064,
      "epoch": 8.457183837890625e-06,
      "step": 2217
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 2217,
      "training_step_time": 9.903200149536133
    },
    {
      "epoch": 8.457183837890625e-06,
      "model_forward_time": 3.12969970703125,
      "step": 2217
    },
    {
      "epoch": 8.457183837890625e-06,
      "step": 2217,
      "training_step_time": 10.24606466293335
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 3.1570537090301514,
      "step": 2218
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 2218,
      "training_step_time": 9.923370122909546
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 3.3959121704101562,
      "step": 2218
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 2218,
      "training_step_time": 10.166055679321289
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 3.1351966857910156,
      "step": 2218
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 2218,
      "training_step_time": 9.905645847320557
    },
    {
      "epoch": 8.46099853515625e-06,
      "model_forward_time": 3.1331005096435547,
      "step": 2218
    },
    {
      "epoch": 8.46099853515625e-06,
      "step": 2218,
      "training_step_time": 10.306168794631958
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 3.153045177459717,
      "step": 2219
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 2219,
      "training_step_time": 9.92532992362976
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 3.130707025527954,
      "step": 2219
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 2219,
      "training_step_time": 9.903269529342651
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 3.4015772342681885,
      "step": 2219
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 2219,
      "training_step_time": 10.172948598861694
    },
    {
      "epoch": 8.464813232421874e-06,
      "model_forward_time": 2.4313197135925293,
      "step": 2219
    },
    {
      "epoch": 8.464813232421874e-06,
      "step": 2219,
      "training_step_time": 10.269558668136597
    },
    {
      "epoch": 8.4686279296875e-06,
      "grad_norm": 0.4646880030632019,
      "learning_rate": 1e-05,
      "loss": 0.1092,
      "step": 2220
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 3.1754069328308105,
      "step": 2220
    },
    {
      "dynamics_loss_avg": 0.09179209582507611,
      "epoch": 8.4686279296875e-06,
      "step": 2220
    },
    {
      "action_loss_avg": 0.051429643854498865,
      "epoch": 8.4686279296875e-06,
      "step": 2220
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 2220,
      "training_step_time": 9.949676990509033
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 3.1334660053253174,
      "step": 2220
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 2220,
      "training_step_time": 9.907772302627563
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 3.132545232772827,
      "step": 2220
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 2220,
      "training_step_time": 9.904492378234863
    },
    {
      "epoch": 8.4686279296875e-06,
      "model_forward_time": 3.4037318229675293,
      "step": 2220
    },
    {
      "epoch": 8.4686279296875e-06,
      "step": 2220,
      "training_step_time": 10.294626712799072
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 3.1724979877471924,
      "step": 2221
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 2221,
      "training_step_time": 9.949148178100586
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 3.1379141807556152,
      "step": 2221
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 2221,
      "training_step_time": 9.919500350952148
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 3.1421639919281006,
      "step": 2221
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 2221,
      "training_step_time": 9.925792694091797
    },
    {
      "epoch": 8.472442626953125e-06,
      "model_forward_time": 3.1463170051574707,
      "step": 2221
    },
    {
      "epoch": 8.472442626953125e-06,
      "step": 2221,
      "training_step_time": 10.304426193237305
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 3.4474802017211914,
      "step": 2222
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 2222,
      "training_step_time": 10.222352504730225
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 3.140611171722412,
      "step": 2222
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 2222,
      "training_step_time": 9.921941757202148
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 3.130556106567383,
      "step": 2222
    },
    {
      "dynamics_loss_avg": 0.07802722677588463,
      "epoch": 8.47625732421875e-06,
      "step": 2222
    },
    {
      "action_loss_avg": 0.04698818400502205,
      "epoch": 8.47625732421875e-06,
      "step": 2222
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 2222,
      "training_step_time": 9.90709638595581
    },
    {
      "epoch": 8.47625732421875e-06,
      "model_forward_time": 3.1323952674865723,
      "step": 2222
    },
    {
      "epoch": 8.47625732421875e-06,
      "step": 2222,
      "training_step_time": 9.996999263763428
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 3.183814764022827,
      "step": 2223
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 2223,
      "training_step_time": 9.950919151306152
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 3.43379545211792,
      "step": 2223
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 2223,
      "training_step_time": 10.197853803634644
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 3.1296586990356445,
      "step": 2223
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 2223,
      "training_step_time": 9.895657777786255
    },
    {
      "epoch": 8.480072021484374e-06,
      "model_forward_time": 3.1310160160064697,
      "step": 2223
    },
    {
      "epoch": 8.480072021484374e-06,
      "step": 2223,
      "training_step_time": 10.254984855651855
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 3.1847753524780273,
      "step": 2224
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 2224,
      "training_step_time": 9.9522545337677
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 3.1303601264953613,
      "step": 2224
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 2224,
      "training_step_time": 9.895080327987671
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 3.399749517440796,
      "step": 2224
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 2224,
      "training_step_time": 10.16704511642456
    },
    {
      "epoch": 8.48388671875e-06,
      "model_forward_time": 3.1359593868255615,
      "step": 2224
    },
    {
      "epoch": 8.48388671875e-06,
      "step": 2224,
      "training_step_time": 10.309287548065186
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 3.173103094100952,
      "step": 2225
    },
    {
      "dynamics_loss_avg": 0.06328402925282717,
      "epoch": 8.487701416015625e-06,
      "step": 2225
    },
    {
      "action_loss_avg": 0.04114083331078291,
      "epoch": 8.487701416015625e-06,
      "step": 2225
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 2225,
      "training_step_time": 9.947689533233643
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 3.1332478523254395,
      "step": 2225
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 2225,
      "training_step_time": 9.906728506088257
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 3.136643886566162,
      "step": 2225
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 2225,
      "training_step_time": 9.91307258605957
    },
    {
      "epoch": 8.487701416015625e-06,
      "model_forward_time": 3.4241256713867188,
      "step": 2225
    },
    {
      "epoch": 8.487701416015625e-06,
      "step": 2225,
      "training_step_time": 10.305049419403076
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 3.1784446239471436,
      "step": 2226
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 2226,
      "training_step_time": 9.952995777130127
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 3.134161949157715,
      "step": 2226
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 2226,
      "training_step_time": 9.906249284744263
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 3.135881185531616,
      "step": 2226
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 2226,
      "training_step_time": 9.912288427352905
    },
    {
      "epoch": 8.49151611328125e-06,
      "model_forward_time": 3.138869285583496,
      "step": 2226
    },
    {
      "epoch": 8.49151611328125e-06,
      "step": 2226,
      "training_step_time": 10.012190341949463
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 3.4541523456573486,
      "step": 2227
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 2227,
      "training_step_time": 10.229710578918457
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 3.1388473510742188,
      "step": 2227
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 2227,
      "training_step_time": 9.911189556121826
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 3.1407556533813477,
      "step": 2227
    },
    {
      "dynamics_loss_avg": 0.06551837511360645,
      "epoch": 8.495330810546874e-06,
      "step": 2227
    },
    {
      "action_loss_avg": 0.04060480920597911,
      "epoch": 8.495330810546874e-06,
      "step": 2227
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 2227,
      "training_step_time": 9.918456315994263
    },
    {
      "epoch": 8.495330810546874e-06,
      "model_forward_time": 3.1388444900512695,
      "step": 2227
    },
    {
      "epoch": 8.495330810546874e-06,
      "step": 2227,
      "training_step_time": 10.023571968078613
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 3.1752912998199463,
      "step": 2228
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 2228,
      "training_step_time": 9.950456857681274
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 3.387812852859497,
      "step": 2228
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 2228,
      "training_step_time": 10.16267466545105
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 3.1408615112304688,
      "step": 2228
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 2228,
      "training_step_time": 9.91387128829956
    },
    {
      "epoch": 8.4991455078125e-06,
      "model_forward_time": 3.136556386947632,
      "step": 2228
    },
    {
      "epoch": 8.4991455078125e-06,
      "step": 2228,
      "training_step_time": 10.258723258972168
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 3.1680779457092285,
      "step": 2229
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 2229,
      "training_step_time": 9.943214893341064
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 3.1389882564544678,
      "step": 2229
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 2229,
      "training_step_time": 9.912609100341797
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 3.4144961833953857,
      "step": 2229
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 2229,
      "training_step_time": 10.185668706893921
    },
    {
      "epoch": 8.502960205078125e-06,
      "model_forward_time": 3.130816698074341,
      "step": 2229
    },
    {
      "epoch": 8.502960205078125e-06,
      "step": 2229,
      "training_step_time": 10.284157752990723
    },
    {
      "epoch": 8.50677490234375e-06,
      "grad_norm": 0.6715245842933655,
      "learning_rate": 1e-05,
      "loss": 0.1085,
      "step": 2230
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 3.1813738346099854,
      "step": 2230
    },
    {
      "dynamics_loss_avg": 0.06431585010141135,
      "epoch": 8.50677490234375e-06,
      "step": 2230
    },
    {
      "action_loss_avg": 0.04517533350735903,
      "epoch": 8.50677490234375e-06,
      "step": 2230
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 2230,
      "training_step_time": 9.952774286270142
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 3.1346452236175537,
      "step": 2230
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 2230,
      "training_step_time": 9.903616428375244
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 3.1336545944213867,
      "step": 2230
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 2230,
      "training_step_time": 9.903214454650879
    },
    {
      "epoch": 8.50677490234375e-06,
      "model_forward_time": 3.132568120956421,
      "step": 2230
    },
    {
      "epoch": 8.50677490234375e-06,
      "step": 2230,
      "training_step_time": 10.29352617263794
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 3.176254987716675,
      "step": 2231
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 2231,
      "training_step_time": 9.944063425064087
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 3.133887767791748,
      "step": 2231
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 2231,
      "training_step_time": 9.89822244644165
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 3.128343105316162,
      "step": 2231
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 2231,
      "training_step_time": 9.89640760421753
    },
    {
      "epoch": 8.510589599609375e-06,
      "model_forward_time": 3.1357085704803467,
      "step": 2231
    },
    {
      "epoch": 8.510589599609375e-06,
      "step": 2231,
      "training_step_time": 10.247594356536865
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 3.4260683059692383,
      "step": 2232
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 2232,
      "training_step_time": 10.19687533378601
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 3.1388349533081055,
      "step": 2232
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 2232,
      "training_step_time": 9.910950422286987
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 3.132774829864502,
      "step": 2232
    },
    {
      "dynamics_loss_avg": 0.08385757319629192,
      "epoch": 8.514404296875e-06,
      "step": 2232
    },
    {
      "action_loss_avg": 0.044421862345188856,
      "epoch": 8.514404296875e-06,
      "step": 2232
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 2232,
      "training_step_time": 9.91011095046997
    },
    {
      "epoch": 8.514404296875e-06,
      "model_forward_time": 3.13631010055542,
      "step": 2232
    },
    {
      "epoch": 8.514404296875e-06,
      "step": 2232,
      "training_step_time": 10.251787185668945
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 3.161386489868164,
      "step": 2233
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 2233,
      "training_step_time": 9.936755895614624
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 3.135796070098877,
      "step": 2233
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 2233,
      "training_step_time": 10.184509038925171
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 3.1360740661621094,
      "step": 2233
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 2233,
      "training_step_time": 9.905454635620117
    },
    {
      "epoch": 8.518218994140625e-06,
      "model_forward_time": 3.1354329586029053,
      "step": 2233
    },
    {
      "epoch": 8.518218994140625e-06,
      "step": 2233,
      "training_step_time": 10.275880575180054
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 3.1703720092773438,
      "step": 2234
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 2234,
      "training_step_time": 9.938768863677979
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 2.4314701557159424,
      "step": 2234
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 2234,
      "training_step_time": 7.453327178955078
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 3.3856966495513916,
      "step": 2234
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 2234,
      "training_step_time": 10.148925065994263
    },
    {
      "epoch": 8.52203369140625e-06,
      "model_forward_time": 3.136225938796997,
      "step": 2234
    },
    {
      "epoch": 8.52203369140625e-06,
      "step": 2234,
      "training_step_time": 10.007771492004395
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 3.1706619262695312,
      "step": 2235
    },
    {
      "dynamics_loss_avg": 0.05572921726852655,
      "epoch": 8.525848388671875e-06,
      "step": 2235
    },
    {
      "action_loss_avg": 0.028293372318148612,
      "epoch": 8.525848388671875e-06,
      "step": 2235
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 2235,
      "training_step_time": 9.942413568496704
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 3.134460210800171,
      "step": 2235
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 2235,
      "training_step_time": 9.904350996017456
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 3.1370956897735596,
      "step": 2235
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 2235,
      "training_step_time": 9.906737565994263
    },
    {
      "epoch": 8.525848388671875e-06,
      "model_forward_time": 3.134182929992676,
      "step": 2235
    },
    {
      "epoch": 8.525848388671875e-06,
      "step": 2235,
      "training_step_time": 10.274423122406006
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 3.1826043128967285,
      "step": 2236
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 2236,
      "training_step_time": 9.955327272415161
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 3.1375441551208496,
      "step": 2236
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 2236,
      "training_step_time": 9.916730642318726
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 3.142012357711792,
      "step": 2236
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 2236,
      "training_step_time": 9.919097900390625
    },
    {
      "epoch": 8.5296630859375e-06,
      "model_forward_time": 3.141876459121704,
      "step": 2236
    },
    {
      "epoch": 8.5296630859375e-06,
      "step": 2236,
      "training_step_time": 10.298141956329346
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 3.4501428604125977,
      "step": 2237
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 2237,
      "training_step_time": 10.224910736083984
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 3.141026496887207,
      "step": 2237
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 2237,
      "training_step_time": 9.916667699813843
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 3.1446893215179443,
      "step": 2237
    },
    {
      "dynamics_loss_avg": 0.05056110974401236,
      "epoch": 8.533477783203125e-06,
      "step": 2237
    },
    {
      "action_loss_avg": 0.036204972816631195,
      "epoch": 8.533477783203125e-06,
      "step": 2237
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 2237,
      "training_step_time": 9.922520160675049
    },
    {
      "epoch": 8.533477783203125e-06,
      "model_forward_time": 3.141873836517334,
      "step": 2237
    },
    {
      "epoch": 8.533477783203125e-06,
      "step": 2237,
      "training_step_time": 10.31502890586853
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 3.168623924255371,
      "step": 2238
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 2238,
      "training_step_time": 9.938318014144897
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 3.406336784362793,
      "step": 2238
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 2238,
      "training_step_time": 10.174541473388672
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 3.1384048461914062,
      "step": 2238
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 2238,
      "training_step_time": 9.907934665679932
    },
    {
      "epoch": 8.53729248046875e-06,
      "model_forward_time": 3.1375889778137207,
      "step": 2238
    },
    {
      "epoch": 8.53729248046875e-06,
      "step": 2238,
      "training_step_time": 10.317410469055176
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 3.1711654663085938,
      "step": 2239
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 2239,
      "training_step_time": 9.941138505935669
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 3.138216733932495,
      "step": 2239
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 2239,
      "training_step_time": 9.90914011001587
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 3.4258322715759277,
      "step": 2239
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 2239,
      "training_step_time": 10.19549560546875
    },
    {
      "epoch": 8.541107177734375e-06,
      "model_forward_time": 3.1382508277893066,
      "step": 2239
    },
    {
      "epoch": 8.541107177734375e-06,
      "step": 2239,
      "training_step_time": 10.00359320640564
    },
    {
      "epoch": 8.544921875e-06,
      "grad_norm": 0.6346741318702698,
      "learning_rate": 1e-05,
      "loss": 0.1046,
      "step": 2240
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 3.162670373916626,
      "step": 2240
    },
    {
      "dynamics_loss_avg": 0.06335569228976964,
      "epoch": 8.544921875e-06,
      "step": 2240
    },
    {
      "action_loss_avg": 0.044136438984423876,
      "epoch": 8.544921875e-06,
      "step": 2240
    },
    {
      "epoch": 8.544921875e-06,
      "step": 2240,
      "training_step_time": 9.932645320892334
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 3.135924816131592,
      "step": 2240
    },
    {
      "epoch": 8.544921875e-06,
      "step": 2240,
      "training_step_time": 9.897698163986206
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 3.1339282989501953,
      "step": 2240
    },
    {
      "epoch": 8.544921875e-06,
      "step": 2240,
      "training_step_time": 9.903136968612671
    },
    {
      "epoch": 8.544921875e-06,
      "model_forward_time": 3.395073413848877,
      "step": 2240
    },
    {
      "epoch": 8.544921875e-06,
      "step": 2240,
      "training_step_time": 10.305343866348267
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 2.4767425060272217,
      "step": 2241
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 2241,
      "training_step_time": 7.49988579750061
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 3.133481979370117,
      "step": 2241
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 2241,
      "training_step_time": 9.899680852890015
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 3.133460760116577,
      "step": 2241
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 2241,
      "training_step_time": 9.899546146392822
    },
    {
      "epoch": 8.548736572265625e-06,
      "model_forward_time": 3.1357455253601074,
      "step": 2241
    },
    {
      "epoch": 8.548736572265625e-06,
      "step": 2241,
      "training_step_time": 10.284157514572144
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 3.4701623916625977,
      "step": 2242
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 2242,
      "training_step_time": 10.24474024772644
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 3.1389737129211426,
      "step": 2242
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 2242,
      "training_step_time": 9.911630392074585
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 3.1387674808502197,
      "step": 2242
    },
    {
      "dynamics_loss_avg": 0.06254309229552746,
      "epoch": 8.55255126953125e-06,
      "step": 2242
    },
    {
      "action_loss_avg": 0.04570372924208641,
      "epoch": 8.55255126953125e-06,
      "step": 2242
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 2242,
      "training_step_time": 9.913069248199463
    },
    {
      "epoch": 8.55255126953125e-06,
      "model_forward_time": 3.1429362297058105,
      "step": 2242
    },
    {
      "epoch": 8.55255126953125e-06,
      "step": 2242,
      "training_step_time": 10.259595394134521
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 3.188786745071411,
      "step": 2243
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 2243,
      "training_step_time": 9.964232444763184
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 3.415043830871582,
      "step": 2243
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 2243,
      "training_step_time": 10.188665628433228
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 3.141191005706787,
      "step": 2243
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 2243,
      "training_step_time": 9.916801452636719
    },
    {
      "epoch": 8.556365966796875e-06,
      "model_forward_time": 3.1442184448242188,
      "step": 2243
    },
    {
      "epoch": 8.556365966796875e-06,
      "step": 2243,
      "training_step_time": 10.2942636013031
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 3.188518762588501,
      "step": 2244
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 2244,
      "training_step_time": 9.96365761756897
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 3.1407976150512695,
      "step": 2244
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 2244,
      "training_step_time": 9.917128086090088
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 3.4060142040252686,
      "step": 2244
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 2244,
      "training_step_time": 10.176438093185425
    },
    {
      "epoch": 8.5601806640625e-06,
      "model_forward_time": 3.1420371532440186,
      "step": 2244
    },
    {
      "epoch": 8.5601806640625e-06,
      "step": 2244,
      "training_step_time": 10.010097980499268
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 3.1856110095977783,
      "step": 2245
    },
    {
      "dynamics_loss_avg": 0.054452274739742276,
      "epoch": 8.563995361328125e-06,
      "step": 2245
    },
    {
      "action_loss_avg": 0.037210611160844564,
      "epoch": 8.563995361328125e-06,
      "step": 2245
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 2245,
      "training_step_time": 9.959112167358398
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 3.135895252227783,
      "step": 2245
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 2245,
      "training_step_time": 9.904614686965942
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 3.1343233585357666,
      "step": 2245
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 2245,
      "training_step_time": 9.902969360351562
    },
    {
      "epoch": 8.563995361328125e-06,
      "model_forward_time": 3.408313035964966,
      "step": 2245
    },
    {
      "epoch": 8.563995361328125e-06,
      "step": 2245,
      "training_step_time": 10.305161714553833
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 3.1634554862976074,
      "step": 2246
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 2246,
      "training_step_time": 9.936630487442017
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 3.135474920272827,
      "step": 2246
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 2246,
      "training_step_time": 9.90762448310852
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 3.140439510345459,
      "step": 2246
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 2246,
      "training_step_time": 9.914149045944214
    },
    {
      "epoch": 8.56781005859375e-06,
      "model_forward_time": 3.1397528648376465,
      "step": 2246
    },
    {
      "epoch": 8.56781005859375e-06,
      "step": 2246,
      "training_step_time": 10.269118070602417
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 3.440488338470459,
      "step": 2247
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 2247,
      "training_step_time": 10.213667631149292
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 3.135387897491455,
      "step": 2247
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 2247,
      "training_step_time": 9.905705451965332
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 3.139918327331543,
      "step": 2247
    },
    {
      "dynamics_loss_avg": 0.058041559904813765,
      "epoch": 8.571624755859375e-06,
      "step": 2247
    },
    {
      "action_loss_avg": 0.034759608656167985,
      "epoch": 8.571624755859375e-06,
      "step": 2247
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 2247,
      "training_step_time": 9.913158178329468
    },
    {
      "epoch": 8.571624755859375e-06,
      "model_forward_time": 3.1357858180999756,
      "step": 2247
    },
    {
      "epoch": 8.571624755859375e-06,
      "step": 2247,
      "training_step_time": 10.281783103942871
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 3.1765739917755127,
      "step": 2248
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 2248,
      "training_step_time": 9.945574283599854
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 3.384413003921509,
      "step": 2248
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 2248,
      "training_step_time": 10.153380393981934
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 3.1374006271362305,
      "step": 2248
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 2248,
      "training_step_time": 9.909140825271606
    },
    {
      "epoch": 8.575439453125e-06,
      "model_forward_time": 3.1359362602233887,
      "step": 2248
    },
    {
      "epoch": 8.575439453125e-06,
      "step": 2248,
      "training_step_time": 10.320247173309326
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 3.2172389030456543,
      "step": 2249
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 2249,
      "training_step_time": 9.993630647659302
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 3.139881134033203,
      "step": 2249
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 2249,
      "training_step_time": 9.911450386047363
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 3.415381669998169,
      "step": 2249
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 2249,
      "training_step_time": 10.187126874923706
    },
    {
      "epoch": 8.579254150390625e-06,
      "model_forward_time": 3.1362314224243164,
      "step": 2249
    },
    {
      "epoch": 8.579254150390625e-06,
      "step": 2249,
      "training_step_time": 10.016185283660889
    },
    {
      "epoch": 8.58306884765625e-06,
      "grad_norm": 0.5530171990394592,
      "learning_rate": 1e-05,
      "loss": 0.1049,
      "step": 2250
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 3.1768136024475098,
      "step": 2250
    },
    {
      "dynamics_loss_avg": 0.07326105125248432,
      "epoch": 8.58306884765625e-06,
      "step": 2250
    },
    {
      "action_loss_avg": 0.04211033498868346,
      "epoch": 8.58306884765625e-06,
      "step": 2250
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 2250,
      "training_step_time": 9.949891090393066
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 3.1394076347351074,
      "step": 2250
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 2250,
      "training_step_time": 9.917124271392822
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 3.1364121437072754,
      "step": 2250
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 2250,
      "training_step_time": 9.903144359588623
    },
    {
      "epoch": 8.58306884765625e-06,
      "model_forward_time": 3.135768175125122,
      "step": 2250
    },
    {
      "epoch": 8.58306884765625e-06,
      "step": 2250,
      "training_step_time": 10.268094301223755
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 3.186126470565796,
      "step": 2251
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 2251,
      "training_step_time": 9.955491304397583
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 3.1348061561584473,
      "step": 2251
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 2251,
      "training_step_time": 9.900978565216064
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 3.1352617740631104,
      "step": 2251
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 2251,
      "training_step_time": 9.903555631637573
    },
    {
      "epoch": 8.586883544921875e-06,
      "model_forward_time": 3.141141891479492,
      "step": 2251
    },
    {
      "epoch": 8.586883544921875e-06,
      "step": 2251,
      "training_step_time": 10.299884557723999
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 3.434326410293579,
      "step": 2252
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 2252,
      "training_step_time": 10.207610130310059
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 3.1399123668670654,
      "step": 2252
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 2252,
      "training_step_time": 9.912241458892822
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 3.135577917098999,
      "step": 2252
    },
    {
      "dynamics_loss_avg": 0.05735665485262871,
      "epoch": 8.5906982421875e-06,
      "step": 2252
    },
    {
      "action_loss_avg": 0.046586580015718936,
      "epoch": 8.5906982421875e-06,
      "step": 2252
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 2252,
      "training_step_time": 9.904979228973389
    },
    {
      "epoch": 8.5906982421875e-06,
      "model_forward_time": 3.1390326023101807,
      "step": 2252
    },
    {
      "epoch": 8.5906982421875e-06,
      "step": 2252,
      "training_step_time": 10.271199941635132
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 3.148298501968384,
      "step": 2253
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 2253,
      "training_step_time": 9.92143201828003
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 3.1362130641937256,
      "step": 2253
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 2253,
      "training_step_time": 10.178037881851196
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 2.4331722259521484,
      "step": 2253
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 2253,
      "training_step_time": 7.463761806488037
    },
    {
      "epoch": 8.594512939453125e-06,
      "model_forward_time": 3.128828287124634,
      "step": 2253
    },
    {
      "epoch": 8.594512939453125e-06,
      "step": 2253,
      "training_step_time": 10.254379510879517
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 2.465657949447632,
      "step": 2254
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 2254,
      "training_step_time": 7.494379758834839
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 2.430074453353882,
      "step": 2254
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 2254,
      "training_step_time": 7.45171046257019
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 3.135228395462036,
      "step": 2254
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 2254,
      "training_step_time": 9.908887147903442
    },
    {
      "epoch": 8.59832763671875e-06,
      "model_forward_time": 3.408986806869507,
      "step": 2254
    },
    {
      "epoch": 8.59832763671875e-06,
      "step": 2254,
      "training_step_time": 10.297460079193115
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 3.1544220447540283,
      "step": 2255
    },
    {
      "dynamics_loss_avg": 0.05460189506411552,
      "epoch": 8.602142333984375e-06,
      "step": 2255
    },
    {
      "action_loss_avg": 0.03913858411833644,
      "epoch": 8.602142333984375e-06,
      "step": 2255
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 2255,
      "training_step_time": 9.928654909133911
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 3.1346616744995117,
      "step": 2255
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 2255,
      "training_step_time": 9.91272258758545
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 3.1367177963256836,
      "step": 2255
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 2255,
      "training_step_time": 9.913369417190552
    },
    {
      "epoch": 8.602142333984375e-06,
      "model_forward_time": 2.432565212249756,
      "step": 2255
    },
    {
      "epoch": 8.602142333984375e-06,
      "step": 2255,
      "training_step_time": 10.019892692565918
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 3.488529920578003,
      "step": 2256
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 2256,
      "training_step_time": 10.269137620925903
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 3.136704444885254,
      "step": 2256
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 2256,
      "training_step_time": 9.914928197860718
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 3.138073205947876,
      "step": 2256
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 2256,
      "training_step_time": 9.916116952896118
    },
    {
      "epoch": 8.60595703125e-06,
      "model_forward_time": 3.1388325691223145,
      "step": 2256
    },
    {
      "epoch": 8.60595703125e-06,
      "step": 2256,
      "training_step_time": 10.269012928009033
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 3.175689935684204,
      "step": 2257
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 2257,
      "training_step_time": 10.246296644210815
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 3.138528347015381,
      "step": 2257
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 2257,
      "training_step_time": 9.920282363891602
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 3.1418943405151367,
      "step": 2257
    },
    {
      "dynamics_loss_avg": 0.06547250021249056,
      "epoch": 8.609771728515625e-06,
      "step": 2257
    },
    {
      "action_loss_avg": 0.04632778661325574,
      "epoch": 8.609771728515625e-06,
      "step": 2257
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 2257,
      "training_step_time": 9.926132678985596
    },
    {
      "epoch": 8.609771728515625e-06,
      "model_forward_time": 3.1388683319091797,
      "step": 2257
    },
    {
      "epoch": 8.609771728515625e-06,
      "step": 2257,
      "training_step_time": 10.268939733505249
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 3.1747100353240967,
      "step": 2258
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 2258,
      "training_step_time": 9.953612804412842
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 3.1381428241729736,
      "step": 2258
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 2258,
      "training_step_time": 10.197368621826172
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 3.1390321254730225,
      "step": 2258
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 2258,
      "training_step_time": 9.909228801727295
    },
    {
      "epoch": 8.61358642578125e-06,
      "model_forward_time": 3.135014533996582,
      "step": 2258
    },
    {
      "epoch": 8.61358642578125e-06,
      "step": 2258,
      "training_step_time": 10.30222749710083
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 3.1626269817352295,
      "step": 2259
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 2259,
      "training_step_time": 9.938470125198364
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 3.1330814361572266,
      "step": 2259
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 2259,
      "training_step_time": 9.904402494430542
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 3.414752960205078,
      "step": 2259
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 2259,
      "training_step_time": 10.189063787460327
    },
    {
      "epoch": 8.617401123046875e-06,
      "model_forward_time": 3.1331820487976074,
      "step": 2259
    },
    {
      "epoch": 8.617401123046875e-06,
      "step": 2259,
      "training_step_time": 10.008505582809448
    },
    {
      "epoch": 8.6212158203125e-06,
      "grad_norm": 0.5832422971725464,
      "learning_rate": 1e-05,
      "loss": 0.1055,
      "step": 2260
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 3.1694366931915283,
      "step": 2260
    },
    {
      "dynamics_loss_avg": 0.05708068236708641,
      "epoch": 8.6212158203125e-06,
      "step": 2260
    },
    {
      "action_loss_avg": 0.045292914658784864,
      "epoch": 8.6212158203125e-06,
      "step": 2260
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 2260,
      "training_step_time": 9.945537567138672
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 3.140420913696289,
      "step": 2260
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 2260,
      "training_step_time": 9.91274356842041
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 3.129338502883911,
      "step": 2260
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 2260,
      "training_step_time": 9.89741849899292
    },
    {
      "epoch": 8.6212158203125e-06,
      "model_forward_time": 3.384645462036133,
      "step": 2260
    },
    {
      "epoch": 8.6212158203125e-06,
      "step": 2260,
      "training_step_time": 10.26501178741455
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 3.1658504009246826,
      "step": 2261
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 2261,
      "training_step_time": 9.935515880584717
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 3.1360855102539062,
      "step": 2261
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 2261,
      "training_step_time": 9.904768705368042
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 3.1298422813415527,
      "step": 2261
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 2261,
      "training_step_time": 9.908271074295044
    },
    {
      "epoch": 8.625030517578125e-06,
      "model_forward_time": 3.139392614364624,
      "step": 2261
    },
    {
      "epoch": 8.625030517578125e-06,
      "step": 2261,
      "training_step_time": 10.01752257347107
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 3.398103952407837,
      "step": 2262
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 2262,
      "training_step_time": 10.177492141723633
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 3.137876033782959,
      "step": 2262
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 2262,
      "training_step_time": 9.918407201766968
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 2.4314637184143066,
      "step": 2262
    },
    {
      "dynamics_loss_avg": 0.06328479386866093,
      "epoch": 8.62884521484375e-06,
      "step": 2262
    },
    {
      "action_loss_avg": 0.04857853055000305,
      "epoch": 8.62884521484375e-06,
      "step": 2262
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 2262,
      "training_step_time": 7.462988376617432
    },
    {
      "epoch": 8.62884521484375e-06,
      "model_forward_time": 3.138195276260376,
      "step": 2262
    },
    {
      "epoch": 8.62884521484375e-06,
      "step": 2262,
      "training_step_time": 10.307027339935303
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 3.1759414672851562,
      "step": 2263
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 2263,
      "training_step_time": 9.957845449447632
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 3.4135639667510986,
      "step": 2263
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 2263,
      "training_step_time": 10.194493532180786
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 3.137674331665039,
      "step": 2263
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 2263,
      "training_step_time": 9.923145055770874
    },
    {
      "epoch": 8.632659912109375e-06,
      "model_forward_time": 3.1418113708496094,
      "step": 2263
    },
    {
      "epoch": 8.632659912109375e-06,
      "step": 2263,
      "training_step_time": 10.283758878707886
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 3.176868438720703,
      "step": 2264
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 2264,
      "training_step_time": 9.96117091178894
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 2.4307117462158203,
      "step": 2264
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 2264,
      "training_step_time": 7.464287042617798
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 2.429380178451538,
      "step": 2264
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 2264,
      "training_step_time": 7.461146116256714
    },
    {
      "epoch": 8.636474609375e-06,
      "model_forward_time": 3.3978850841522217,
      "step": 2264
    },
    {
      "epoch": 8.636474609375e-06,
      "step": 2264,
      "training_step_time": 10.331058025360107
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 3.1637139320373535,
      "step": 2265
    },
    {
      "dynamics_loss_avg": 0.05632740780711174,
      "epoch": 8.640289306640625e-06,
      "step": 2265
    },
    {
      "action_loss_avg": 0.049202233646065,
      "epoch": 8.640289306640625e-06,
      "step": 2265
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 2265,
      "training_step_time": 9.943153142929077
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 3.134519338607788,
      "step": 2265
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 2265,
      "training_step_time": 9.908144235610962
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 3.136776924133301,
      "step": 2265
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 2265,
      "training_step_time": 9.913461685180664
    },
    {
      "epoch": 8.640289306640625e-06,
      "model_forward_time": 3.1383821964263916,
      "step": 2265
    },
    {
      "epoch": 8.640289306640625e-06,
      "step": 2265,
      "training_step_time": 10.292131662368774
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 3.1771080493927,
      "step": 2266
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 2266,
      "training_step_time": 9.96180009841919
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 3.1363112926483154,
      "step": 2266
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 2266,
      "training_step_time": 9.912401676177979
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 3.135697364807129,
      "step": 2266
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 2266,
      "training_step_time": 9.91107439994812
    },
    {
      "epoch": 8.64410400390625e-06,
      "model_forward_time": 3.1369810104370117,
      "step": 2266
    },
    {
      "epoch": 8.64410400390625e-06,
      "step": 2266,
      "training_step_time": 10.026043176651001
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 3.1611127853393555,
      "step": 2267
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 2267,
      "training_step_time": 10.20608901977539
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 3.135118007659912,
      "step": 2267
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 2267,
      "training_step_time": 9.909438371658325
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 3.136483669281006,
      "step": 2267
    },
    {
      "dynamics_loss_avg": 0.0611611619591713,
      "epoch": 8.647918701171875e-06,
      "step": 2267
    },
    {
      "action_loss_avg": 0.04714059494435787,
      "epoch": 8.647918701171875e-06,
      "step": 2267
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 2267,
      "training_step_time": 9.91548490524292
    },
    {
      "epoch": 8.647918701171875e-06,
      "model_forward_time": 3.1345696449279785,
      "step": 2267
    },
    {
      "epoch": 8.647918701171875e-06,
      "step": 2267,
      "training_step_time": 10.267026901245117
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 3.1550917625427246,
      "step": 2268
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 2268,
      "training_step_time": 9.934279203414917
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 3.424921989440918,
      "step": 2268
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 2268,
      "training_step_time": 10.199131965637207
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 3.144667625427246,
      "step": 2268
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 2268,
      "training_step_time": 9.924032211303711
    },
    {
      "epoch": 8.6517333984375e-06,
      "model_forward_time": 2.435153007507324,
      "step": 2268
    },
    {
      "epoch": 8.6517333984375e-06,
      "step": 2268,
      "training_step_time": 10.269448518753052
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 3.1849687099456787,
      "step": 2269
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 2269,
      "training_step_time": 9.963938474655151
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 3.136502742767334,
      "step": 2269
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 2269,
      "training_step_time": 9.907951593399048
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 3.412674903869629,
      "step": 2269
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 2269,
      "training_step_time": 10.18485426902771
    },
    {
      "epoch": 8.655548095703125e-06,
      "model_forward_time": 3.1336491107940674,
      "step": 2269
    },
    {
      "epoch": 8.655548095703125e-06,
      "step": 2269,
      "training_step_time": 10.310338973999023
    },
    {
      "epoch": 8.65936279296875e-06,
      "grad_norm": 0.7251645922660828,
      "learning_rate": 1e-05,
      "loss": 0.1094,
      "step": 2270
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 2.4743165969848633,
      "step": 2270
    },
    {
      "dynamics_loss_avg": 0.07063347529619932,
      "epoch": 8.65936279296875e-06,
      "step": 2270
    },
    {
      "action_loss_avg": 0.04244317496195436,
      "epoch": 8.65936279296875e-06,
      "step": 2270
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 2270,
      "training_step_time": 7.5076515674591064
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 2.434054136276245,
      "step": 2270
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 2270,
      "training_step_time": 7.458269834518433
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 3.131617307662964,
      "step": 2270
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 2270,
      "training_step_time": 9.896897315979004
    },
    {
      "epoch": 8.65936279296875e-06,
      "model_forward_time": 3.1371285915374756,
      "step": 2270
    },
    {
      "epoch": 8.65936279296875e-06,
      "step": 2270,
      "training_step_time": 10.325865507125854
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 3.4688198566436768,
      "step": 2271
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 2271,
      "training_step_time": 10.239642143249512
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 3.1359875202178955,
      "step": 2271
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 2271,
      "training_step_time": 9.911666631698608
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 3.1364424228668213,
      "step": 2271
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 2271,
      "training_step_time": 9.908623933792114
    },
    {
      "epoch": 8.663177490234375e-06,
      "model_forward_time": 3.13678240776062,
      "step": 2271
    },
    {
      "epoch": 8.663177490234375e-06,
      "step": 2271,
      "training_step_time": 10.266434907913208
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 3.4421610832214355,
      "step": 2272
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 2272,
      "training_step_time": 10.212884426116943
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 3.1355671882629395,
      "step": 2272
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 2272,
      "training_step_time": 9.904282569885254
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 3.131192445755005,
      "step": 2272
    },
    {
      "dynamics_loss_avg": 0.07200149670243264,
      "epoch": 8.6669921875e-06,
      "step": 2272
    },
    {
      "action_loss_avg": 0.0537457138299942,
      "epoch": 8.6669921875e-06,
      "step": 2272
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 2272,
      "training_step_time": 9.903611421585083
    },
    {
      "epoch": 8.6669921875e-06,
      "model_forward_time": 3.1364922523498535,
      "step": 2272
    },
    {
      "epoch": 8.6669921875e-06,
      "step": 2272,
      "training_step_time": 10.019958257675171
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 3.1785669326782227,
      "step": 2273
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 2273,
      "training_step_time": 9.946726322174072
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 2.4331846237182617,
      "step": 2273
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 2273,
      "training_step_time": 7.463123798370361
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 3.3814632892608643,
      "step": 2273
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 2273,
      "training_step_time": 10.148136615753174
    },
    {
      "epoch": 8.670806884765626e-06,
      "model_forward_time": 3.138986825942993,
      "step": 2273
    },
    {
      "epoch": 8.670806884765626e-06,
      "step": 2273,
      "training_step_time": 10.296368598937988
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 3.1568503379821777,
      "step": 2274
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 2274,
      "training_step_time": 9.934726238250732
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 3.13909649848938,
      "step": 2274
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 2274,
      "training_step_time": 9.911181449890137
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 2.43619704246521,
      "step": 2274
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 2274,
      "training_step_time": 7.733335971832275
    },
    {
      "epoch": 8.67462158203125e-06,
      "model_forward_time": 3.1397175788879395,
      "step": 2274
    },
    {
      "epoch": 8.67462158203125e-06,
      "step": 2274,
      "training_step_time": 10.277388095855713
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 3.152186632156372,
      "step": 2275
    },
    {
      "dynamics_loss_avg": 0.06343303192406893,
      "epoch": 8.678436279296875e-06,
      "step": 2275
    },
    {
      "action_loss_avg": 0.037361941579729316,
      "epoch": 8.678436279296875e-06,
      "step": 2275
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 2275,
      "training_step_time": 9.927573442459106
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 3.14155650138855,
      "step": 2275
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 2275,
      "training_step_time": 9.915233373641968
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 2.4378583431243896,
      "step": 2275
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 2275,
      "training_step_time": 7.468671560287476
    },
    {
      "epoch": 8.678436279296875e-06,
      "model_forward_time": 3.1315298080444336,
      "step": 2275
    },
    {
      "epoch": 8.678436279296875e-06,
      "step": 2275,
      "training_step_time": 10.318269968032837
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 3.4596009254455566,
      "step": 2276
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 2276,
      "training_step_time": 10.234259128570557
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 3.136772394180298,
      "step": 2276
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 2276,
      "training_step_time": 9.905954599380493
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 2.43660569190979,
      "step": 2276
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 2276,
      "training_step_time": 7.458824634552002
    },
    {
      "epoch": 8.6822509765625e-06,
      "model_forward_time": 2.432976722717285,
      "step": 2276
    },
    {
      "epoch": 8.6822509765625e-06,
      "step": 2276,
      "training_step_time": 10.322208166122437
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 3.158858299255371,
      "step": 2277
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 2277,
      "training_step_time": 9.928606986999512
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 3.39152193069458,
      "step": 2277
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 2277,
      "training_step_time": 10.158465147018433
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 2.436030149459839,
      "step": 2277
    },
    {
      "dynamics_loss_avg": 0.07597170397639275,
      "epoch": 8.686065673828126e-06,
      "step": 2277
    },
    {
      "action_loss_avg": 0.047866642475128174,
      "epoch": 8.686065673828126e-06,
      "step": 2277
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 2277,
      "training_step_time": 7.467021465301514
    },
    {
      "epoch": 8.686065673828126e-06,
      "model_forward_time": 2.4363272190093994,
      "step": 2277
    },
    {
      "epoch": 8.686065673828126e-06,
      "step": 2277,
      "training_step_time": 10.020084381103516
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 3.176553249359131,
      "step": 2278
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 2278,
      "training_step_time": 9.948979377746582
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 2.440295934677124,
      "step": 2278
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 2278,
      "training_step_time": 7.470924139022827
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 3.423041820526123,
      "step": 2278
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 2278,
      "training_step_time": 10.196141719818115
    },
    {
      "epoch": 8.68988037109375e-06,
      "model_forward_time": 3.1378369331359863,
      "step": 2278
    },
    {
      "epoch": 8.68988037109375e-06,
      "step": 2278,
      "training_step_time": 10.02725863456726
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 3.1715798377990723,
      "step": 2279
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 2279,
      "training_step_time": 9.948458194732666
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 3.146921157836914,
      "step": 2279
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 2279,
      "training_step_time": 9.92254638671875
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 3.1448137760162354,
      "step": 2279
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 2279,
      "training_step_time": 9.92248797416687
    },
    {
      "epoch": 8.693695068359375e-06,
      "model_forward_time": 3.4062957763671875,
      "step": 2279
    },
    {
      "epoch": 8.693695068359375e-06,
      "step": 2279,
      "training_step_time": 10.306470394134521
    },
    {
      "epoch": 8.697509765625e-06,
      "grad_norm": 0.5567466020584106,
      "learning_rate": 1e-05,
      "loss": 0.1157,
      "step": 2280
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 3.1683342456817627,
      "step": 2280
    },
    {
      "dynamics_loss_avg": 0.07762598134577274,
      "epoch": 8.697509765625e-06,
      "step": 2280
    },
    {
      "action_loss_avg": 0.051777339354157445,
      "epoch": 8.697509765625e-06,
      "step": 2280
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 2280,
      "training_step_time": 9.950500249862671
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 3.1460154056549072,
      "step": 2280
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 2280,
      "training_step_time": 9.925406217575073
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 3.143294095993042,
      "step": 2280
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 2280,
      "training_step_time": 9.921385765075684
    },
    {
      "epoch": 8.697509765625e-06,
      "model_forward_time": 2.4407501220703125,
      "step": 2280
    },
    {
      "epoch": 8.697509765625e-06,
      "step": 2280,
      "training_step_time": 10.285548448562622
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 3.4440207481384277,
      "step": 2281
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 2281,
      "training_step_time": 10.221673965454102
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 3.1456148624420166,
      "step": 2281
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 2281,
      "training_step_time": 9.91803240776062
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 2.438598155975342,
      "step": 2281
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 2281,
      "training_step_time": 7.467148780822754
    },
    {
      "epoch": 8.701324462890626e-06,
      "model_forward_time": 2.4377715587615967,
      "step": 2281
    },
    {
      "epoch": 8.701324462890626e-06,
      "step": 2281,
      "training_step_time": 10.357776403427124
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 3.1794376373291016,
      "step": 2282
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 2282,
      "training_step_time": 9.951799869537354
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 3.429975986480713,
      "step": 2282
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 2282,
      "training_step_time": 10.19997763633728
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 3.1385700702667236,
      "step": 2282
    },
    {
      "dynamics_loss_avg": 0.06666547693312168,
      "epoch": 8.70513916015625e-06,
      "step": 2282
    },
    {
      "action_loss_avg": 0.046532342303544284,
      "epoch": 8.70513916015625e-06,
      "step": 2282
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 2282,
      "training_step_time": 9.913013458251953
    },
    {
      "epoch": 8.70513916015625e-06,
      "model_forward_time": 3.1475675106048584,
      "step": 2282
    },
    {
      "epoch": 8.70513916015625e-06,
      "step": 2282,
      "training_step_time": 10.297158241271973
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 3.1711268424987793,
      "step": 2283
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 2283,
      "training_step_time": 9.946236848831177
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 3.142855167388916,
      "step": 2283
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 2283,
      "training_step_time": 9.913449764251709
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 3.406980276107788,
      "step": 2283
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 2283,
      "training_step_time": 10.178916931152344
    },
    {
      "epoch": 8.708953857421875e-06,
      "model_forward_time": 3.137701988220215,
      "step": 2283
    },
    {
      "epoch": 8.708953857421875e-06,
      "step": 2283,
      "training_step_time": 10.022515296936035
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 3.163381814956665,
      "step": 2284
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 2284,
      "training_step_time": 9.936756134033203
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 3.143480062484741,
      "step": 2284
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 2284,
      "training_step_time": 9.914584398269653
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 3.147984504699707,
      "step": 2284
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 2284,
      "training_step_time": 9.921421527862549
    },
    {
      "epoch": 8.7127685546875e-06,
      "model_forward_time": 2.7395055294036865,
      "step": 2284
    },
    {
      "epoch": 8.7127685546875e-06,
      "step": 2284,
      "training_step_time": 10.246610164642334
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 3.1795027256011963,
      "step": 2285
    },
    {
      "dynamics_loss_avg": 0.06945886202156544,
      "epoch": 8.716583251953126e-06,
      "step": 2285
    },
    {
      "action_loss_avg": 0.04547813571989536,
      "epoch": 8.716583251953126e-06,
      "step": 2285
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 2285,
      "training_step_time": 9.971564769744873
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 2.433993101119995,
      "step": 2285
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 2285,
      "training_step_time": 7.463939428329468
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 3.132668972015381,
      "step": 2285
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 2285,
      "training_step_time": 9.906267404556274
    },
    {
      "epoch": 8.716583251953126e-06,
      "model_forward_time": 3.1335461139678955,
      "step": 2285
    },
    {
      "epoch": 8.716583251953126e-06,
      "step": 2285,
      "training_step_time": 10.300951719284058
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 2.7396152019500732,
      "step": 2286
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 2286,
      "training_step_time": 7.769636392593384
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 3.130964517593384,
      "step": 2286
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 2286,
      "training_step_time": 9.902631282806396
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 2.4323649406433105,
      "step": 2286
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 2286,
      "training_step_time": 7.462897777557373
    },
    {
      "epoch": 8.72039794921875e-06,
      "model_forward_time": 3.13466739654541,
      "step": 2286
    },
    {
      "epoch": 8.72039794921875e-06,
      "step": 2286,
      "training_step_time": 10.30798864364624
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 3.160416841506958,
      "step": 2287
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 2287,
      "training_step_time": 9.93481993675232
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 3.390077590942383,
      "step": 2287
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 2287,
      "training_step_time": 10.162789583206177
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 3.1389570236206055,
      "step": 2287
    },
    {
      "dynamics_loss_avg": 0.070533331297338,
      "epoch": 8.724212646484375e-06,
      "step": 2287
    },
    {
      "action_loss_avg": 0.044217883795499804,
      "epoch": 8.724212646484375e-06,
      "step": 2287
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 2287,
      "training_step_time": 9.91051435470581
    },
    {
      "epoch": 8.724212646484375e-06,
      "model_forward_time": 3.136370897293091,
      "step": 2287
    },
    {
      "epoch": 8.724212646484375e-06,
      "step": 2287,
      "training_step_time": 10.321000814437866
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 3.159064292907715,
      "step": 2288
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 2288,
      "training_step_time": 9.934395551681519
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 3.140256643295288,
      "step": 2288
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 2288,
      "training_step_time": 10.17378282546997
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 2.433434247970581,
      "step": 2288
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 2288,
      "training_step_time": 7.4636547565460205
    },
    {
      "epoch": 8.72802734375e-06,
      "model_forward_time": 3.133547782897949,
      "step": 2288
    },
    {
      "epoch": 8.72802734375e-06,
      "step": 2288,
      "training_step_time": 10.295385599136353
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 3.15913724899292,
      "step": 2289
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 2289,
      "training_step_time": 9.934117555618286
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 3.137291431427002,
      "step": 2289
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 2289,
      "training_step_time": 9.91041898727417
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 3.1384687423706055,
      "step": 2289
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 2289,
      "training_step_time": 9.915792465209961
    },
    {
      "epoch": 8.731842041015626e-06,
      "model_forward_time": 3.3959767818450928,
      "step": 2289
    },
    {
      "epoch": 8.731842041015626e-06,
      "step": 2289,
      "training_step_time": 10.265651226043701
    },
    {
      "epoch": 8.73565673828125e-06,
      "grad_norm": 0.6354585289955139,
      "learning_rate": 1e-05,
      "loss": 0.1118,
      "step": 2290
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 2.4688823223114014,
      "step": 2290
    },
    {
      "dynamics_loss_avg": 0.06918572783470153,
      "epoch": 8.73565673828125e-06,
      "step": 2290
    },
    {
      "action_loss_avg": 0.043682107143104075,
      "epoch": 8.73565673828125e-06,
      "step": 2290
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 2290,
      "training_step_time": 7.502930402755737
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 3.134223461151123,
      "step": 2290
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 2290,
      "training_step_time": 9.90938115119934
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 3.133192300796509,
      "step": 2290
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 2290,
      "training_step_time": 9.902793407440186
    },
    {
      "epoch": 8.73565673828125e-06,
      "model_forward_time": 3.138951301574707,
      "step": 2290
    },
    {
      "epoch": 8.73565673828125e-06,
      "step": 2290,
      "training_step_time": 10.287460803985596
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 3.167696237564087,
      "step": 2291
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 2291,
      "training_step_time": 9.942784309387207
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 2.429729700088501,
      "step": 2291
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 2291,
      "training_step_time": 7.460103511810303
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 3.12508487701416,
      "step": 2291
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 2291,
      "training_step_time": 9.891318559646606
    },
    {
      "epoch": 8.739471435546875e-06,
      "model_forward_time": 3.13092041015625,
      "step": 2291
    },
    {
      "epoch": 8.739471435546875e-06,
      "step": 2291,
      "training_step_time": 10.255058526992798
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 3.164898633956909,
      "step": 2292
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 2292,
      "training_step_time": 9.927411079406738
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 3.3959765434265137,
      "step": 2292
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 2292,
      "training_step_time": 10.16207480430603
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 3.132513999938965,
      "step": 2292
    },
    {
      "dynamics_loss_avg": 0.08106858860701323,
      "epoch": 8.7432861328125e-06,
      "step": 2292
    },
    {
      "action_loss_avg": 0.06516060950234533,
      "epoch": 8.7432861328125e-06,
      "step": 2292
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 2292,
      "training_step_time": 9.901890516281128
    },
    {
      "epoch": 8.7432861328125e-06,
      "model_forward_time": 3.1324405670166016,
      "step": 2292
    },
    {
      "epoch": 8.7432861328125e-06,
      "step": 2292,
      "training_step_time": 10.298126697540283
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 3.1744930744171143,
      "step": 2293
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 2293,
      "training_step_time": 9.946839094161987
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 3.1316516399383545,
      "step": 2293
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 2293,
      "training_step_time": 10.168469429016113
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 3.131284475326538,
      "step": 2293
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 2293,
      "training_step_time": 9.899154663085938
    },
    {
      "epoch": 8.747100830078126e-06,
      "model_forward_time": 3.1424708366394043,
      "step": 2293
    },
    {
      "epoch": 8.747100830078126e-06,
      "step": 2293,
      "training_step_time": 10.293986797332764
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 3.179795503616333,
      "step": 2294
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 2294,
      "training_step_time": 9.948729753494263
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 3.131243944168091,
      "step": 2294
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 2294,
      "training_step_time": 9.89821982383728
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 3.3902809619903564,
      "step": 2294
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 2294,
      "training_step_time": 10.159365177154541
    },
    {
      "epoch": 8.75091552734375e-06,
      "model_forward_time": 3.1365129947662354,
      "step": 2294
    },
    {
      "epoch": 8.75091552734375e-06,
      "step": 2294,
      "training_step_time": 10.043266773223877
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 3.170421600341797,
      "step": 2295
    },
    {
      "dynamics_loss_avg": 0.06206487771123648,
      "epoch": 8.754730224609375e-06,
      "step": 2295
    },
    {
      "action_loss_avg": 0.04384063836187124,
      "epoch": 8.754730224609375e-06,
      "step": 2295
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 2295,
      "training_step_time": 9.948882102966309
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 3.1339292526245117,
      "step": 2295
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 2295,
      "training_step_time": 9.904418706893921
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 3.1401288509368896,
      "step": 2295
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 2295,
      "training_step_time": 9.909218072891235
    },
    {
      "epoch": 8.754730224609375e-06,
      "model_forward_time": 3.137451410293579,
      "step": 2295
    },
    {
      "epoch": 8.754730224609375e-06,
      "step": 2295,
      "training_step_time": 10.254762172698975
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 3.200207471847534,
      "step": 2296
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 2296,
      "training_step_time": 9.977451801300049
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 3.1442105770111084,
      "step": 2296
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 2296,
      "training_step_time": 9.914961576461792
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 3.141108751296997,
      "step": 2296
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 2296,
      "training_step_time": 9.914713144302368
    },
    {
      "epoch": 8.758544921875e-06,
      "model_forward_time": 3.1385295391082764,
      "step": 2296
    },
    {
      "epoch": 8.758544921875e-06,
      "step": 2296,
      "training_step_time": 10.310122728347778
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 2.7093911170959473,
      "step": 2297
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 2297,
      "training_step_time": 7.740574598312378
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 3.1371138095855713,
      "step": 2297
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 2297,
      "training_step_time": 9.910072565078735
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 2.4379332065582275,
      "step": 2297
    },
    {
      "dynamics_loss_avg": 0.07558510098606348,
      "epoch": 8.762359619140626e-06,
      "step": 2297
    },
    {
      "action_loss_avg": 0.049930153787136076,
      "epoch": 8.762359619140626e-06,
      "step": 2297
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 2297,
      "training_step_time": 7.4709601402282715
    },
    {
      "epoch": 8.762359619140626e-06,
      "model_forward_time": 3.1397976875305176,
      "step": 2297
    },
    {
      "epoch": 8.762359619140626e-06,
      "step": 2297,
      "training_step_time": 10.277266263961792
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 3.161205768585205,
      "step": 2298
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 2298,
      "training_step_time": 9.937728643417358
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 2.4436888694763184,
      "step": 2298
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 2298,
      "training_step_time": 7.4722959995269775
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 3.409433603286743,
      "step": 2298
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 2298,
      "training_step_time": 10.18154001235962
    },
    {
      "epoch": 8.76617431640625e-06,
      "model_forward_time": 2.438511848449707,
      "step": 2298
    },
    {
      "epoch": 8.76617431640625e-06,
      "step": 2298,
      "training_step_time": 10.336225032806396
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 3.149162530899048,
      "step": 2299
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 2299,
      "training_step_time": 9.921048402786255
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 3.1455206871032715,
      "step": 2299
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 2299,
      "training_step_time": 9.923119306564331
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 3.14078688621521,
      "step": 2299
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 2299,
      "training_step_time": 10.180743932723999
    },
    {
      "epoch": 8.769989013671875e-06,
      "model_forward_time": 2.440495014190674,
      "step": 2299
    },
    {
      "epoch": 8.769989013671875e-06,
      "step": 2299,
      "training_step_time": 10.071114301681519
    },
    {
      "epoch": 8.7738037109375e-06,
      "grad_norm": 0.8714238405227661,
      "learning_rate": 1e-05,
      "loss": 0.1132,
      "step": 2300
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 3.1843674182891846,
      "step": 2300
    },
    {
      "dynamics_loss_avg": 0.09439227283000946,
      "epoch": 8.7738037109375e-06,
      "step": 2300
    },
    {
      "action_loss_avg": 0.06070112008601427,
      "epoch": 8.7738037109375e-06,
      "step": 2300
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 2300,
      "training_step_time": 9.964519023895264
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 3.1437740325927734,
      "step": 2300
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 2300,
      "training_step_time": 9.919126272201538
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 3.1427764892578125,
      "step": 2300
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 2300,
      "training_step_time": 9.914315938949585
    },
    {
      "epoch": 8.7738037109375e-06,
      "model_forward_time": 3.1459529399871826,
      "step": 2300
    },
    {
      "epoch": 8.7738037109375e-06,
      "step": 2300,
      "training_step_time": 10.291432857513428
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 3.145228147506714,
      "step": 2301
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 2301,
      "training_step_time": 9.926177263259888
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 2.4309842586517334,
      "step": 2301
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 2301,
      "training_step_time": 7.458559989929199
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 3.1320347785949707,
      "step": 2301
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 2301,
      "training_step_time": 9.902867317199707
    },
    {
      "epoch": 8.777618408203126e-06,
      "model_forward_time": 2.4385383129119873,
      "step": 2301
    },
    {
      "epoch": 8.777618408203126e-06,
      "step": 2301,
      "training_step_time": 10.290648698806763
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 2.4691710472106934,
      "step": 2302
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 2302,
      "training_step_time": 7.4959821701049805
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 3.4178946018218994,
      "step": 2302
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 2302,
      "training_step_time": 10.180284023284912
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 3.125493049621582,
      "step": 2302
    },
    {
      "dynamics_loss_avg": 0.07064757719635964,
      "epoch": 8.78143310546875e-06,
      "step": 2302
    },
    {
      "action_loss_avg": 0.041855496168136594,
      "epoch": 8.78143310546875e-06,
      "step": 2302
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 2302,
      "training_step_time": 9.89313268661499
    },
    {
      "epoch": 8.78143310546875e-06,
      "model_forward_time": 3.1316614151000977,
      "step": 2302
    },
    {
      "epoch": 8.78143310546875e-06,
      "step": 2302,
      "training_step_time": 10.280578374862671
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 3.1732921600341797,
      "step": 2303
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 2303,
      "training_step_time": 9.93983793258667
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 3.130239248275757,
      "step": 2303
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 2303,
      "training_step_time": 9.899667978286743
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 3.392784595489502,
      "step": 2303
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 2303,
      "training_step_time": 10.15642499923706
    },
    {
      "epoch": 8.785247802734375e-06,
      "model_forward_time": 3.139059066772461,
      "step": 2303
    },
    {
      "epoch": 8.785247802734375e-06,
      "step": 2303,
      "training_step_time": 10.312164545059204
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 3.1875271797180176,
      "step": 2304
    },
    {
      "epoch": 8.7890625e-06,
      "step": 2304,
      "training_step_time": 9.982242107391357
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 3.1416542530059814,
      "step": 2304
    },
    {
      "epoch": 8.7890625e-06,
      "step": 2304,
      "training_step_time": 9.916217803955078
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 3.142070770263672,
      "step": 2304
    },
    {
      "epoch": 8.7890625e-06,
      "step": 2304,
      "training_step_time": 10.1930570602417
    },
    {
      "epoch": 8.7890625e-06,
      "model_forward_time": 3.1414222717285156,
      "step": 2304
    },
    {
      "epoch": 8.7890625e-06,
      "step": 2304,
      "training_step_time": 10.267948150634766
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 2.4817898273468018,
      "step": 2305
    },
    {
      "dynamics_loss_avg": 0.06053837724030018,
      "epoch": 8.792877197265626e-06,
      "step": 2305
    },
    {
      "action_loss_avg": 0.044806708209216596,
      "epoch": 8.792877197265626e-06,
      "step": 2305
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 2305,
      "training_step_time": 7.511228561401367
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 3.13948392868042,
      "step": 2305
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 2305,
      "training_step_time": 9.907689571380615
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 3.1401658058166504,
      "step": 2305
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 2305,
      "training_step_time": 9.912379264831543
    },
    {
      "epoch": 8.792877197265626e-06,
      "model_forward_time": 2.439556837081909,
      "step": 2305
    },
    {
      "epoch": 8.792877197265626e-06,
      "step": 2305,
      "training_step_time": 10.021520376205444
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 3.4767768383026123,
      "step": 2306
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 2306,
      "training_step_time": 10.247120380401611
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 3.140889883041382,
      "step": 2306
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 2306,
      "training_step_time": 9.916305303573608
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 3.1407899856567383,
      "step": 2306
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 2306,
      "training_step_time": 9.912077903747559
    },
    {
      "epoch": 8.79669189453125e-06,
      "model_forward_time": 3.139631748199463,
      "step": 2306
    },
    {
      "epoch": 8.79669189453125e-06,
      "step": 2306,
      "training_step_time": 10.030001163482666
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 3.1696839332580566,
      "step": 2307
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 2307,
      "training_step_time": 10.214089632034302
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 3.141043186187744,
      "step": 2307
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 2307,
      "training_step_time": 9.910115242004395
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 3.139171838760376,
      "step": 2307
    },
    {
      "dynamics_loss_avg": 0.059102417528629304,
      "epoch": 8.800506591796875e-06,
      "step": 2307
    },
    {
      "action_loss_avg": 0.039350821450352666,
      "epoch": 8.800506591796875e-06,
      "step": 2307
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 2307,
      "training_step_time": 9.911895036697388
    },
    {
      "epoch": 8.800506591796875e-06,
      "model_forward_time": 3.144826889038086,
      "step": 2307
    },
    {
      "epoch": 8.800506591796875e-06,
      "step": 2307,
      "training_step_time": 10.310027360916138
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 3.1605875492095947,
      "step": 2308
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 2308,
      "training_step_time": 9.92884111404419
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 3.3949930667877197,
      "step": 2308
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 2308,
      "training_step_time": 10.161394596099854
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 2.43281626701355,
      "step": 2308
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 2308,
      "training_step_time": 7.45870566368103
    },
    {
      "epoch": 8.8043212890625e-06,
      "model_forward_time": 3.1298301219940186,
      "step": 2308
    },
    {
      "epoch": 8.8043212890625e-06,
      "step": 2308,
      "training_step_time": 10.296147108078003
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 3.180678367614746,
      "step": 2309
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 2309,
      "training_step_time": 9.952552556991577
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 3.1487667560577393,
      "step": 2309
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 2309,
      "training_step_time": 9.942254781723022
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 3.14365553855896,
      "step": 2309
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 2309,
      "training_step_time": 10.19014048576355
    },
    {
      "epoch": 8.808135986328126e-06,
      "model_forward_time": 3.1473090648651123,
      "step": 2309
    },
    {
      "epoch": 8.808135986328126e-06,
      "step": 2309,
      "training_step_time": 10.288618326187134
    },
    {
      "epoch": 8.81195068359375e-06,
      "grad_norm": 0.5531513094902039,
      "learning_rate": 1e-05,
      "loss": 0.1043,
      "step": 2310
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 3.1903512477874756,
      "step": 2310
    },
    {
      "dynamics_loss_avg": 0.07125525176525116,
      "epoch": 8.81195068359375e-06,
      "step": 2310
    },
    {
      "action_loss_avg": 0.04744827169924974,
      "epoch": 8.81195068359375e-06,
      "step": 2310
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 2310,
      "training_step_time": 9.96794056892395
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 3.1498303413391113,
      "step": 2310
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 2310,
      "training_step_time": 9.923522233963013
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 3.146533966064453,
      "step": 2310
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 2310,
      "training_step_time": 9.925678968429565
    },
    {
      "epoch": 8.81195068359375e-06,
      "model_forward_time": 3.1429505348205566,
      "step": 2310
    },
    {
      "epoch": 8.81195068359375e-06,
      "step": 2310,
      "training_step_time": 10.285874128341675
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 3.17930269241333,
      "step": 2311
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 2311,
      "training_step_time": 9.950618028640747
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 3.139415979385376,
      "step": 2311
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 2311,
      "training_step_time": 9.908897161483765
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 3.1411325931549072,
      "step": 2311
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 2311,
      "training_step_time": 9.916565179824829
    },
    {
      "epoch": 8.815765380859375e-06,
      "model_forward_time": 3.1366753578186035,
      "step": 2311
    },
    {
      "epoch": 8.815765380859375e-06,
      "step": 2311,
      "training_step_time": 10.012116193771362
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 3.4627718925476074,
      "step": 2312
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 2312,
      "training_step_time": 10.240177631378174
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 3.1399824619293213,
      "step": 2312
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 2312,
      "training_step_time": 9.913486242294312
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 3.139247179031372,
      "step": 2312
    },
    {
      "dynamics_loss_avg": 0.05828456673771143,
      "epoch": 8.819580078125e-06,
      "step": 2312
    },
    {
      "action_loss_avg": 0.03453604467213154,
      "epoch": 8.819580078125e-06,
      "step": 2312
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 2312,
      "training_step_time": 9.914426565170288
    },
    {
      "epoch": 8.819580078125e-06,
      "model_forward_time": 3.142958402633667,
      "step": 2312
    },
    {
      "epoch": 8.819580078125e-06,
      "step": 2312,
      "training_step_time": 10.266887426376343
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 3.1812496185302734,
      "step": 2313
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 2313,
      "training_step_time": 9.953343391418457
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 3.1367597579956055,
      "step": 2313
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 2313,
      "training_step_time": 10.180399656295776
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 3.1356377601623535,
      "step": 2313
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 2313,
      "training_step_time": 9.905397653579712
    },
    {
      "epoch": 8.823394775390626e-06,
      "model_forward_time": 3.1382033824920654,
      "step": 2313
    },
    {
      "epoch": 8.823394775390626e-06,
      "step": 2313,
      "training_step_time": 10.3351571559906
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 3.184213399887085,
      "step": 2314
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 2314,
      "training_step_time": 9.95597219467163
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 3.135087013244629,
      "step": 2314
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 2314,
      "training_step_time": 9.905781030654907
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 3.3959641456604004,
      "step": 2314
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 2314,
      "training_step_time": 10.166577577590942
    },
    {
      "epoch": 8.82720947265625e-06,
      "model_forward_time": 3.1415131092071533,
      "step": 2314
    },
    {
      "epoch": 8.82720947265625e-06,
      "step": 2314,
      "training_step_time": 10.285614728927612
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 2.479137420654297,
      "step": 2315
    },
    {
      "dynamics_loss_avg": 0.07096467297524214,
      "epoch": 8.831024169921875e-06,
      "step": 2315
    },
    {
      "action_loss_avg": 0.04328466691076756,
      "epoch": 8.831024169921875e-06,
      "step": 2315
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 2315,
      "training_step_time": 7.514455318450928
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 3.133679151535034,
      "step": 2315
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 2315,
      "training_step_time": 9.906346321105957
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 3.141496419906616,
      "step": 2315
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 2315,
      "training_step_time": 9.913515567779541
    },
    {
      "epoch": 8.831024169921875e-06,
      "model_forward_time": 3.140444040298462,
      "step": 2315
    },
    {
      "epoch": 8.831024169921875e-06,
      "step": 2315,
      "training_step_time": 10.021153688430786
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 3.4949216842651367,
      "step": 2316
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 2316,
      "training_step_time": 10.268000841140747
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 3.137730836868286,
      "step": 2316
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 2316,
      "training_step_time": 9.907968282699585
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 3.1403820514678955,
      "step": 2316
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 2316,
      "training_step_time": 9.914077997207642
    },
    {
      "epoch": 8.8348388671875e-06,
      "model_forward_time": 3.1363399028778076,
      "step": 2316
    },
    {
      "epoch": 8.8348388671875e-06,
      "step": 2316,
      "training_step_time": 10.226651906967163
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 3.4783191680908203,
      "step": 2317
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 2317,
      "training_step_time": 10.252268552780151
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 3.140495777130127,
      "step": 2317
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 2317,
      "training_step_time": 9.917445659637451
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 3.1460766792297363,
      "step": 2317
    },
    {
      "dynamics_loss_avg": 0.06333041694015265,
      "epoch": 8.838653564453126e-06,
      "step": 2317
    },
    {
      "action_loss_avg": 0.04202189045026898,
      "epoch": 8.838653564453126e-06,
      "step": 2317
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 2317,
      "training_step_time": 9.920032262802124
    },
    {
      "epoch": 8.838653564453126e-06,
      "model_forward_time": 3.1383118629455566,
      "step": 2317
    },
    {
      "epoch": 8.838653564453126e-06,
      "step": 2317,
      "training_step_time": 10.033889293670654
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 3.174516439437866,
      "step": 2318
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 2318,
      "training_step_time": 9.95057225227356
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 3.4196252822875977,
      "step": 2318
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 2318,
      "training_step_time": 10.189548015594482
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 3.1363284587860107,
      "step": 2318
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 2318,
      "training_step_time": 9.905626058578491
    },
    {
      "epoch": 8.84246826171875e-06,
      "model_forward_time": 3.1361846923828125,
      "step": 2318
    },
    {
      "epoch": 8.84246826171875e-06,
      "step": 2318,
      "training_step_time": 10.279597282409668
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 3.1756367683410645,
      "step": 2319
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 2319,
      "training_step_time": 9.944581985473633
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 3.1384661197662354,
      "step": 2319
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 2319,
      "training_step_time": 9.908663511276245
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 3.135922908782959,
      "step": 2319
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 2319,
      "training_step_time": 10.191931009292603
    },
    {
      "epoch": 8.846282958984375e-06,
      "model_forward_time": 3.140904664993286,
      "step": 2319
    },
    {
      "epoch": 8.846282958984375e-06,
      "step": 2319,
      "training_step_time": 10.305798292160034
    },
    {
      "epoch": 8.85009765625e-06,
      "grad_norm": 0.842884361743927,
      "learning_rate": 1e-05,
      "loss": 0.1094,
      "step": 2320
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 3.1760268211364746,
      "step": 2320
    },
    {
      "dynamics_loss_avg": 0.06172048393636942,
      "epoch": 8.85009765625e-06,
      "step": 2320
    },
    {
      "action_loss_avg": 0.044005297124385834,
      "epoch": 8.85009765625e-06,
      "step": 2320
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 2320,
      "training_step_time": 9.954559326171875
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 3.1372296810150146,
      "step": 2320
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 2320,
      "training_step_time": 9.920793771743774
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 3.1456854343414307,
      "step": 2320
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 2320,
      "training_step_time": 9.925611972808838
    },
    {
      "epoch": 8.85009765625e-06,
      "model_forward_time": 3.15061616897583,
      "step": 2320
    },
    {
      "epoch": 8.85009765625e-06,
      "step": 2320,
      "training_step_time": 10.302683591842651
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 3.169365167617798,
      "step": 2321
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 2321,
      "training_step_time": 9.945121765136719
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 3.1456339359283447,
      "step": 2321
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 2321,
      "training_step_time": 9.921803712844849
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 3.141625165939331,
      "step": 2321
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 2321,
      "training_step_time": 9.92000675201416
    },
    {
      "epoch": 8.853912353515626e-06,
      "model_forward_time": 3.1504738330841064,
      "step": 2321
    },
    {
      "epoch": 8.853912353515626e-06,
      "step": 2321,
      "training_step_time": 10.033100366592407
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 3.1515390872955322,
      "step": 2322
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 2322,
      "training_step_time": 10.211342811584473
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 3.1432745456695557,
      "step": 2322
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 2322,
      "training_step_time": 9.920171022415161
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 3.142047882080078,
      "step": 2322
    },
    {
      "dynamics_loss_avg": 0.07240669224411249,
      "epoch": 8.85772705078125e-06,
      "step": 2322
    },
    {
      "action_loss_avg": 0.046487008593976496,
      "epoch": 8.85772705078125e-06,
      "step": 2322
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 2322,
      "training_step_time": 9.916676998138428
    },
    {
      "epoch": 8.85772705078125e-06,
      "model_forward_time": 2.437195062637329,
      "step": 2322
    },
    {
      "epoch": 8.85772705078125e-06,
      "step": 2322,
      "training_step_time": 10.035205364227295
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 3.1594133377075195,
      "step": 2323
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 2323,
      "training_step_time": 9.932704448699951
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 3.1383893489837646,
      "step": 2323
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 2323,
      "training_step_time": 10.176119327545166
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 2.435295581817627,
      "step": 2323
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 2323,
      "training_step_time": 7.467092514038086
    },
    {
      "epoch": 8.861541748046875e-06,
      "model_forward_time": 3.1388304233551025,
      "step": 2323
    },
    {
      "epoch": 8.861541748046875e-06,
      "step": 2323,
      "training_step_time": 10.2820885181427
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 3.1651790142059326,
      "step": 2324
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 2324,
      "training_step_time": 9.935909986495972
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 3.1377034187316895,
      "step": 2324
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 2324,
      "training_step_time": 9.905552625656128
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 3.1365292072296143,
      "step": 2324
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 2324,
      "training_step_time": 10.16991925239563
    },
    {
      "epoch": 8.8653564453125e-06,
      "model_forward_time": 3.1395161151885986,
      "step": 2324
    },
    {
      "epoch": 8.8653564453125e-06,
      "step": 2324,
      "training_step_time": 10.321506977081299
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 3.179312229156494,
      "step": 2325
    },
    {
      "dynamics_loss_avg": 0.057303947024047376,
      "epoch": 8.869171142578126e-06,
      "step": 2325
    },
    {
      "action_loss_avg": 0.04282810147851705,
      "epoch": 8.869171142578126e-06,
      "step": 2325
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 2325,
      "training_step_time": 9.955024003982544
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 3.1307852268218994,
      "step": 2325
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 2325,
      "training_step_time": 9.898310422897339
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 3.1365628242492676,
      "step": 2325
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 2325,
      "training_step_time": 9.902791976928711
    },
    {
      "epoch": 8.869171142578126e-06,
      "model_forward_time": 3.411721706390381,
      "step": 2325
    },
    {
      "epoch": 8.869171142578126e-06,
      "step": 2325,
      "training_step_time": 10.288052082061768
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 3.186533212661743,
      "step": 2326
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 2326,
      "training_step_time": 9.960075378417969
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 3.1382131576538086,
      "step": 2326
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 2326,
      "training_step_time": 9.911586046218872
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 3.1394340991973877,
      "step": 2326
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 2326,
      "training_step_time": 9.918910264968872
    },
    {
      "epoch": 8.87298583984375e-06,
      "model_forward_time": 3.142879009246826,
      "step": 2326
    },
    {
      "epoch": 8.87298583984375e-06,
      "step": 2326,
      "training_step_time": 10.02838921546936
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 3.426868200302124,
      "step": 2327
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 2327,
      "training_step_time": 10.20056676864624
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 3.136446475982666,
      "step": 2327
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 2327,
      "training_step_time": 9.909737825393677
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 2.435912609100342,
      "step": 2327
    },
    {
      "dynamics_loss_avg": 0.06892575956881046,
      "epoch": 8.876800537109375e-06,
      "step": 2327
    },
    {
      "action_loss_avg": 0.057311244681477545,
      "epoch": 8.876800537109375e-06,
      "step": 2327
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 2327,
      "training_step_time": 7.467444181442261
    },
    {
      "epoch": 8.876800537109375e-06,
      "model_forward_time": 3.131735324859619,
      "step": 2327
    },
    {
      "epoch": 8.876800537109375e-06,
      "step": 2327,
      "training_step_time": 10.014384031295776
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 3.1843137741088867,
      "step": 2328
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 2328,
      "training_step_time": 9.9563889503479
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 2.433605194091797,
      "step": 2328
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 2328,
      "training_step_time": 7.7167580127716064
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 3.13515567779541,
      "step": 2328
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 2328,
      "training_step_time": 9.897480010986328
    },
    {
      "epoch": 8.880615234375e-06,
      "model_forward_time": 3.1371243000030518,
      "step": 2328
    },
    {
      "epoch": 8.880615234375e-06,
      "step": 2328,
      "training_step_time": 10.238953351974487
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 3.1557044982910156,
      "step": 2329
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 2329,
      "training_step_time": 9.92265510559082
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 3.136798858642578,
      "step": 2329
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 2329,
      "training_step_time": 9.903263092041016
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 2.4334728717803955,
      "step": 2329
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 2329,
      "training_step_time": 7.719008445739746
    },
    {
      "epoch": 8.884429931640626e-06,
      "model_forward_time": 3.1364660263061523,
      "step": 2329
    },
    {
      "epoch": 8.884429931640626e-06,
      "step": 2329,
      "training_step_time": 10.280555963516235
    },
    {
      "epoch": 8.88824462890625e-06,
      "grad_norm": 0.4961267113685608,
      "learning_rate": 1e-05,
      "loss": 0.1101,
      "step": 2330
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 2.477217674255371,
      "step": 2330
    },
    {
      "dynamics_loss_avg": 0.056597594544291495,
      "epoch": 8.88824462890625e-06,
      "step": 2330
    },
    {
      "action_loss_avg": 0.03691165400668979,
      "epoch": 8.88824462890625e-06,
      "step": 2330
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 2330,
      "training_step_time": 7.505721092224121
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 2.434718370437622,
      "step": 2330
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 2330,
      "training_step_time": 7.461500644683838
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 3.1338608264923096,
      "step": 2330
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 2330,
      "training_step_time": 9.902939319610596
    },
    {
      "epoch": 8.88824462890625e-06,
      "model_forward_time": 3.141672372817993,
      "step": 2330
    },
    {
      "epoch": 8.88824462890625e-06,
      "step": 2330,
      "training_step_time": 10.286905765533447
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 3.157883644104004,
      "step": 2331
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 2331,
      "training_step_time": 9.929527044296265
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 3.141305446624756,
      "step": 2331
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 2331,
      "training_step_time": 9.910243034362793
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 2.4392502307891846,
      "step": 2331
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 2331,
      "training_step_time": 7.469028949737549
    },
    {
      "epoch": 8.892059326171875e-06,
      "model_forward_time": 3.1385293006896973,
      "step": 2331
    },
    {
      "epoch": 8.892059326171875e-06,
      "step": 2331,
      "training_step_time": 10.043612241744995
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 3.461634874343872,
      "step": 2332
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 2332,
      "training_step_time": 10.237980365753174
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 2.439831495285034,
      "step": 2332
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 2332,
      "training_step_time": 7.470813989639282
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 3.13948655128479,
      "step": 2332
    },
    {
      "dynamics_loss_avg": 0.05638085077516734,
      "epoch": 8.8958740234375e-06,
      "step": 2332
    },
    {
      "action_loss_avg": 0.0379117339849472,
      "epoch": 8.8958740234375e-06,
      "step": 2332
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 2332,
      "training_step_time": 9.917064666748047
    },
    {
      "epoch": 8.8958740234375e-06,
      "model_forward_time": 3.143899440765381,
      "step": 2332
    },
    {
      "epoch": 8.8958740234375e-06,
      "step": 2332,
      "training_step_time": 10.025746822357178
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 3.1930370330810547,
      "step": 2333
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 2333,
      "training_step_time": 9.967074394226074
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 3.1427624225616455,
      "step": 2333
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 2333,
      "training_step_time": 9.91913914680481
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 3.3978264331817627,
      "step": 2333
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 2333,
      "training_step_time": 10.171764373779297
    },
    {
      "epoch": 8.899688720703126e-06,
      "model_forward_time": 3.146852970123291,
      "step": 2333
    },
    {
      "epoch": 8.899688720703126e-06,
      "step": 2333,
      "training_step_time": 10.298457622528076
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 2.4770586490631104,
      "step": 2334
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 2334,
      "training_step_time": 7.504800796508789
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 3.13662052154541,
      "step": 2334
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 2334,
      "training_step_time": 9.906302690505981
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 3.1429169178009033,
      "step": 2334
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 2334,
      "training_step_time": 10.181653261184692
    },
    {
      "epoch": 8.90350341796875e-06,
      "model_forward_time": 3.1434097290039062,
      "step": 2334
    },
    {
      "epoch": 8.90350341796875e-06,
      "step": 2334,
      "training_step_time": 10.031117677688599
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 3.177804470062256,
      "step": 2335
    },
    {
      "dynamics_loss_avg": 0.06432931944727897,
      "epoch": 8.907318115234375e-06,
      "step": 2335
    },
    {
      "action_loss_avg": 0.034275361336767673,
      "epoch": 8.907318115234375e-06,
      "step": 2335
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 2335,
      "training_step_time": 9.947466135025024
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 3.142507791519165,
      "step": 2335
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 2335,
      "training_step_time": 9.91028904914856
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 3.142808437347412,
      "step": 2335
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 2335,
      "training_step_time": 9.912797927856445
    },
    {
      "epoch": 8.907318115234375e-06,
      "model_forward_time": 2.4357526302337646,
      "step": 2335
    },
    {
      "epoch": 8.907318115234375e-06,
      "step": 2335,
      "training_step_time": 10.294642210006714
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 3.44791579246521,
      "step": 2336
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 2336,
      "training_step_time": 10.219034433364868
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 2.4333884716033936,
      "step": 2336
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 2336,
      "training_step_time": 7.4626312255859375
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 2.4339873790740967,
      "step": 2336
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 2336,
      "training_step_time": 7.461502313613892
    },
    {
      "epoch": 8.9111328125e-06,
      "model_forward_time": 3.135195732116699,
      "step": 2336
    },
    {
      "epoch": 8.9111328125e-06,
      "step": 2336,
      "training_step_time": 10.278824090957642
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 3.1706128120422363,
      "step": 2337
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 2337,
      "training_step_time": 9.942978620529175
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 2.7030394077301025,
      "step": 2337
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 2337,
      "training_step_time": 7.727763414382935
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 3.1377429962158203,
      "step": 2337
    },
    {
      "dynamics_loss_avg": 0.05302384085953236,
      "epoch": 8.914947509765626e-06,
      "step": 2337
    },
    {
      "action_loss_avg": 0.039445785246789454,
      "epoch": 8.914947509765626e-06,
      "step": 2337
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 2337,
      "training_step_time": 9.906728506088257
    },
    {
      "epoch": 8.914947509765626e-06,
      "model_forward_time": 3.1343698501586914,
      "step": 2337
    },
    {
      "epoch": 8.914947509765626e-06,
      "step": 2337,
      "training_step_time": 10.280954360961914
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 3.148103713989258,
      "step": 2338
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 2338,
      "training_step_time": 9.916468143463135
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 3.1412155628204346,
      "step": 2338
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 2338,
      "training_step_time": 9.90430736541748
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 2.700991630554199,
      "step": 2338
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 2338,
      "training_step_time": 7.727709531784058
    },
    {
      "epoch": 8.91876220703125e-06,
      "model_forward_time": 3.135943651199341,
      "step": 2338
    },
    {
      "epoch": 8.91876220703125e-06,
      "step": 2338,
      "training_step_time": 10.264434576034546
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 3.178541660308838,
      "step": 2339
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 2339,
      "training_step_time": 9.9476957321167
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 3.139723062515259,
      "step": 2339
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 2339,
      "training_step_time": 9.912106037139893
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 3.140930414199829,
      "step": 2339
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 2339,
      "training_step_time": 10.174421548843384
    },
    {
      "epoch": 8.922576904296875e-06,
      "model_forward_time": 3.1379384994506836,
      "step": 2339
    },
    {
      "epoch": 8.922576904296875e-06,
      "step": 2339,
      "training_step_time": 10.017554998397827
    },
    {
      "epoch": 8.9263916015625e-06,
      "grad_norm": 0.8264027237892151,
      "learning_rate": 1e-05,
      "loss": 0.1087,
      "step": 2340
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 2.4747772216796875,
      "step": 2340
    },
    {
      "dynamics_loss_avg": 0.0562015263363719,
      "epoch": 8.9263916015625e-06,
      "step": 2340
    },
    {
      "action_loss_avg": 0.03751244209706783,
      "epoch": 8.9263916015625e-06,
      "step": 2340
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 2340,
      "training_step_time": 7.51002836227417
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 3.136244535446167,
      "step": 2340
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 2340,
      "training_step_time": 9.90803337097168
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 3.1343884468078613,
      "step": 2340
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 2340,
      "training_step_time": 9.905506134033203
    },
    {
      "epoch": 8.9263916015625e-06,
      "model_forward_time": 3.1413021087646484,
      "step": 2340
    },
    {
      "epoch": 8.9263916015625e-06,
      "step": 2340,
      "training_step_time": 10.272989511489868
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 3.1771697998046875,
      "step": 2341
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 2341,
      "training_step_time": 9.949811220169067
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 3.143862247467041,
      "step": 2341
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 2341,
      "training_step_time": 9.916849851608276
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 3.142383575439453,
      "step": 2341
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 2341,
      "training_step_time": 9.92347002029419
    },
    {
      "epoch": 8.930206298828126e-06,
      "model_forward_time": 3.147753953933716,
      "step": 2341
    },
    {
      "epoch": 8.930206298828126e-06,
      "step": 2341,
      "training_step_time": 10.325280666351318
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 3.1763947010040283,
      "step": 2342
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 2342,
      "training_step_time": 10.239999771118164
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 3.14908504486084,
      "step": 2342
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 2342,
      "training_step_time": 9.934333086013794
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 3.1470911502838135,
      "step": 2342
    },
    {
      "dynamics_loss_avg": 0.06497125755995511,
      "epoch": 8.93402099609375e-06,
      "step": 2342
    },
    {
      "action_loss_avg": 0.059039083682000636,
      "epoch": 8.93402099609375e-06,
      "step": 2342
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 2342,
      "training_step_time": 9.931899785995483
    },
    {
      "epoch": 8.93402099609375e-06,
      "model_forward_time": 3.148658275604248,
      "step": 2342
    },
    {
      "epoch": 8.93402099609375e-06,
      "step": 2342,
      "training_step_time": 10.044819116592407
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 3.183950901031494,
      "step": 2343
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 2343,
      "training_step_time": 9.965043306350708
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 3.427311420440674,
      "step": 2343
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 2343,
      "training_step_time": 10.203077793121338
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 3.1417438983917236,
      "step": 2343
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 2343,
      "training_step_time": 9.915209770202637
    },
    {
      "epoch": 8.937835693359375e-06,
      "model_forward_time": 3.144740104675293,
      "step": 2343
    },
    {
      "epoch": 8.937835693359375e-06,
      "step": 2343,
      "training_step_time": 10.258829832077026
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 3.1662533283233643,
      "step": 2344
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 2344,
      "training_step_time": 9.937114000320435
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 3.1403114795684814,
      "step": 2344
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 2344,
      "training_step_time": 9.914963245391846
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 3.143691301345825,
      "step": 2344
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 2344,
      "training_step_time": 10.214717864990234
    },
    {
      "epoch": 8.941650390625e-06,
      "model_forward_time": 3.1447508335113525,
      "step": 2344
    },
    {
      "epoch": 8.941650390625e-06,
      "step": 2344,
      "training_step_time": 10.015870809555054
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 3.155555486679077,
      "step": 2345
    },
    {
      "dynamics_loss_avg": 0.05602389443665743,
      "epoch": 8.945465087890626e-06,
      "step": 2345
    },
    {
      "action_loss_avg": 0.036597728822380306,
      "epoch": 8.945465087890626e-06,
      "step": 2345
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 2345,
      "training_step_time": 9.93666386604309
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 3.1422784328460693,
      "step": 2345
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 2345,
      "training_step_time": 9.919783592224121
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 2.4373419284820557,
      "step": 2345
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 2345,
      "training_step_time": 7.470362424850464
    },
    {
      "epoch": 8.945465087890626e-06,
      "model_forward_time": 3.4254252910614014,
      "step": 2345
    },
    {
      "epoch": 8.945465087890626e-06,
      "step": 2345,
      "training_step_time": 10.29687237739563
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 3.1645681858062744,
      "step": 2346
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 2346,
      "training_step_time": 9.939429998397827
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 3.1365811824798584,
      "step": 2346
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 2346,
      "training_step_time": 9.908386945724487
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 3.1413676738739014,
      "step": 2346
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 2346,
      "training_step_time": 9.911435604095459
    },
    {
      "epoch": 8.94927978515625e-06,
      "model_forward_time": 2.4375271797180176,
      "step": 2346
    },
    {
      "epoch": 8.94927978515625e-06,
      "step": 2346,
      "training_step_time": 10.2699613571167
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 3.1776225566864014,
      "step": 2347
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 2347,
      "training_step_time": 9.945723533630371
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 3.392367124557495,
      "step": 2347
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 2347,
      "training_step_time": 10.159103631973267
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 3.1392078399658203,
      "step": 2347
    },
    {
      "dynamics_loss_avg": 0.06761638596653938,
      "epoch": 8.953094482421875e-06,
      "step": 2347
    },
    {
      "action_loss_avg": 0.03781689116731286,
      "epoch": 8.953094482421875e-06,
      "step": 2347
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 2347,
      "training_step_time": 9.906409740447998
    },
    {
      "epoch": 8.953094482421875e-06,
      "model_forward_time": 3.1364855766296387,
      "step": 2347
    },
    {
      "epoch": 8.953094482421875e-06,
      "step": 2347,
      "training_step_time": 10.253640174865723
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 3.1942214965820312,
      "step": 2348
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 2348,
      "training_step_time": 9.963566541671753
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 3.3836913108825684,
      "step": 2348
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 2348,
      "training_step_time": 10.154341459274292
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 3.1356678009033203,
      "step": 2348
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 2348,
      "training_step_time": 9.905150413513184
    },
    {
      "epoch": 8.9569091796875e-06,
      "model_forward_time": 3.1328213214874268,
      "step": 2348
    },
    {
      "epoch": 8.9569091796875e-06,
      "step": 2348,
      "training_step_time": 10.012967348098755
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 3.163269281387329,
      "step": 2349
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 2349,
      "training_step_time": 9.928818941116333
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 3.133138656616211,
      "step": 2349
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 2349,
      "training_step_time": 9.896215200424194
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 3.130496025085449,
      "step": 2349
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 2349,
      "training_step_time": 10.158461570739746
    },
    {
      "epoch": 8.960723876953126e-06,
      "model_forward_time": 3.131193161010742,
      "step": 2349
    },
    {
      "epoch": 8.960723876953126e-06,
      "step": 2349,
      "training_step_time": 10.256795883178711
    },
    {
      "epoch": 8.96453857421875e-06,
      "grad_norm": 0.5169376134872437,
      "learning_rate": 1e-05,
      "loss": 0.1102,
      "step": 2350
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 3.163106918334961,
      "step": 2350
    },
    {
      "dynamics_loss_avg": 0.07153446823358536,
      "epoch": 8.96453857421875e-06,
      "step": 2350
    },
    {
      "action_loss_avg": 0.054498824290931225,
      "epoch": 8.96453857421875e-06,
      "step": 2350
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 2350,
      "training_step_time": 9.93422555923462
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 3.1347737312316895,
      "step": 2350
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 2350,
      "training_step_time": 9.907611608505249
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 3.1327123641967773,
      "step": 2350
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 2350,
      "training_step_time": 9.900121927261353
    },
    {
      "epoch": 8.96453857421875e-06,
      "model_forward_time": 3.3821492195129395,
      "step": 2350
    },
    {
      "epoch": 8.96453857421875e-06,
      "step": 2350,
      "training_step_time": 10.252079963684082
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 3.1559247970581055,
      "step": 2351
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 2351,
      "training_step_time": 9.93228816986084
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 3.1362807750701904,
      "step": 2351
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 2351,
      "training_step_time": 9.91422152519226
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 3.13525652885437,
      "step": 2351
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 2351,
      "training_step_time": 9.911919116973877
    },
    {
      "epoch": 8.968353271484375e-06,
      "model_forward_time": 3.135314702987671,
      "step": 2351
    },
    {
      "epoch": 8.968353271484375e-06,
      "step": 2351,
      "training_step_time": 10.292102813720703
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 3.4017016887664795,
      "step": 2352
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 2352,
      "training_step_time": 10.17762541770935
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 2.432060956954956,
      "step": 2352
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 2352,
      "training_step_time": 7.464756488800049
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 2.4325623512268066,
      "step": 2352
    },
    {
      "dynamics_loss_avg": 0.07260063141584397,
      "epoch": 8.97216796875e-06,
      "step": 2352
    },
    {
      "action_loss_avg": 0.0423743849620223,
      "epoch": 8.97216796875e-06,
      "step": 2352
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 2352,
      "training_step_time": 7.468510866165161
    },
    {
      "epoch": 8.97216796875e-06,
      "model_forward_time": 3.1390321254730225,
      "step": 2352
    },
    {
      "epoch": 8.97216796875e-06,
      "step": 2352,
      "training_step_time": 10.289499521255493
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 3.1698243618011475,
      "step": 2353
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 2353,
      "training_step_time": 9.949289321899414
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 3.137167453765869,
      "step": 2353
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 2353,
      "training_step_time": 10.181487798690796
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 3.135010004043579,
      "step": 2353
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 2353,
      "training_step_time": 9.91460394859314
    },
    {
      "epoch": 8.975982666015626e-06,
      "model_forward_time": 3.137547492980957,
      "step": 2353
    },
    {
      "epoch": 8.975982666015626e-06,
      "step": 2353,
      "training_step_time": 10.011996746063232
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 3.182218313217163,
      "step": 2354
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 2354,
      "training_step_time": 9.963459491729736
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 3.132812738418579,
      "step": 2354
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 2354,
      "training_step_time": 9.909932374954224
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 3.3908610343933105,
      "step": 2354
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 2354,
      "training_step_time": 10.171045541763306
    },
    {
      "epoch": 8.97979736328125e-06,
      "model_forward_time": 3.1401431560516357,
      "step": 2354
    },
    {
      "epoch": 8.97979736328125e-06,
      "step": 2354,
      "training_step_time": 10.272182703018188
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 3.189323663711548,
      "step": 2355
    },
    {
      "dynamics_loss_avg": 0.05674273576587439,
      "epoch": 8.983612060546875e-06,
      "step": 2355
    },
    {
      "action_loss_avg": 0.033274589106440544,
      "epoch": 8.983612060546875e-06,
      "step": 2355
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 2355,
      "training_step_time": 9.971156358718872
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 3.1391594409942627,
      "step": 2355
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 2355,
      "training_step_time": 9.918206930160522
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 3.137847661972046,
      "step": 2355
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 2355,
      "training_step_time": 9.917307376861572
    },
    {
      "epoch": 8.983612060546875e-06,
      "model_forward_time": 3.4318182468414307,
      "step": 2355
    },
    {
      "epoch": 8.983612060546875e-06,
      "step": 2355,
      "training_step_time": 10.305806398391724
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 3.176170587539673,
      "step": 2356
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 2356,
      "training_step_time": 9.956460952758789
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 3.131875514984131,
      "step": 2356
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 2356,
      "training_step_time": 9.912884712219238
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 3.1405441761016846,
      "step": 2356
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 2356,
      "training_step_time": 9.918192148208618
    },
    {
      "epoch": 8.9874267578125e-06,
      "model_forward_time": 3.132161855697632,
      "step": 2356
    },
    {
      "epoch": 8.9874267578125e-06,
      "step": 2356,
      "training_step_time": 10.012857437133789
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 3.4719974994659424,
      "step": 2357
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 2357,
      "training_step_time": 10.247910261154175
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 2.4339582920074463,
      "step": 2357
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 2357,
      "training_step_time": 7.463770389556885
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 3.1314914226531982,
      "step": 2357
    },
    {
      "dynamics_loss_avg": 0.06644142819568515,
      "epoch": 8.991241455078126e-06,
      "step": 2357
    },
    {
      "action_loss_avg": 0.04112146354746073,
      "epoch": 8.991241455078126e-06,
      "step": 2357
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 2357,
      "training_step_time": 9.908408164978027
    },
    {
      "epoch": 8.991241455078126e-06,
      "model_forward_time": 3.1347973346710205,
      "step": 2357
    },
    {
      "epoch": 8.991241455078126e-06,
      "step": 2357,
      "training_step_time": 10.268388748168945
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 3.171354055404663,
      "step": 2358
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 2358,
      "training_step_time": 9.954058170318604
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 3.1393587589263916,
      "step": 2358
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 2358,
      "training_step_time": 10.194035768508911
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 3.1402359008789062,
      "step": 2358
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 2358,
      "training_step_time": 9.917418718338013
    },
    {
      "epoch": 8.99505615234375e-06,
      "model_forward_time": 3.1351020336151123,
      "step": 2358
    },
    {
      "epoch": 8.99505615234375e-06,
      "step": 2358,
      "training_step_time": 10.269010543823242
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 3.1810693740844727,
      "step": 2359
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 2359,
      "training_step_time": 9.967650890350342
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 3.1403274536132812,
      "step": 2359
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 2359,
      "training_step_time": 9.9225914478302
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 3.4236373901367188,
      "step": 2359
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 2359,
      "training_step_time": 10.20746636390686
    },
    {
      "epoch": 8.998870849609375e-06,
      "model_forward_time": 3.1402528285980225,
      "step": 2359
    },
    {
      "epoch": 8.998870849609375e-06,
      "step": 2359,
      "training_step_time": 10.282089710235596
    },
    {
      "epoch": 9.002685546875e-06,
      "grad_norm": 0.40559858083724976,
      "learning_rate": 1e-05,
      "loss": 0.1035,
      "step": 2360
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 3.1768038272857666,
      "step": 2360
    },
    {
      "dynamics_loss_avg": 0.05145848263055086,
      "epoch": 9.002685546875e-06,
      "step": 2360
    },
    {
      "action_loss_avg": 0.034361067786812784,
      "epoch": 9.002685546875e-06,
      "step": 2360
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 2360,
      "training_step_time": 9.957939863204956
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 3.142642021179199,
      "step": 2360
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 2360,
      "training_step_time": 9.92270040512085
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 3.136319398880005,
      "step": 2360
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 2360,
      "training_step_time": 9.923198461532593
    },
    {
      "epoch": 9.002685546875e-06,
      "model_forward_time": 2.7117867469787598,
      "step": 2360
    },
    {
      "epoch": 9.002685546875e-06,
      "step": 2360,
      "training_step_time": 10.02771258354187
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 3.182739734649658,
      "step": 2361
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 2361,
      "training_step_time": 9.963911533355713
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 2.4361515045166016,
      "step": 2361
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 2361,
      "training_step_time": 7.470915079116821
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 3.134986639022827,
      "step": 2361
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 2361,
      "training_step_time": 9.91779613494873
    },
    {
      "epoch": 9.006500244140626e-06,
      "model_forward_time": 3.141180992126465,
      "step": 2361
    },
    {
      "epoch": 9.006500244140626e-06,
      "step": 2361,
      "training_step_time": 10.039772033691406
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 3.164299249649048,
      "step": 2362
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 2362,
      "training_step_time": 10.21975588798523
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 3.140155076980591,
      "step": 2362
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 2362,
      "training_step_time": 9.927855491638184
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 3.1443965435028076,
      "step": 2362
    },
    {
      "dynamics_loss_avg": 0.0790887849405408,
      "epoch": 9.01031494140625e-06,
      "step": 2362
    },
    {
      "action_loss_avg": 0.04755557989701629,
      "epoch": 9.01031494140625e-06,
      "step": 2362
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 2362,
      "training_step_time": 9.931419372558594
    },
    {
      "epoch": 9.01031494140625e-06,
      "model_forward_time": 3.145580768585205,
      "step": 2362
    },
    {
      "epoch": 9.01031494140625e-06,
      "step": 2362,
      "training_step_time": 10.314065933227539
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 3.179211378097534,
      "step": 2363
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 2363,
      "training_step_time": 9.96163272857666
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 3.419778823852539,
      "step": 2363
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 2363,
      "training_step_time": 10.198552370071411
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 3.1394593715667725,
      "step": 2363
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 2363,
      "training_step_time": 9.917211055755615
    },
    {
      "epoch": 9.014129638671875e-06,
      "model_forward_time": 3.14029860496521,
      "step": 2363
    },
    {
      "epoch": 9.014129638671875e-06,
      "step": 2363,
      "training_step_time": 10.301851034164429
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 3.1771833896636963,
      "step": 2364
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 2364,
      "training_step_time": 9.955682039260864
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 3.135058879852295,
      "step": 2364
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 2364,
      "training_step_time": 9.910264730453491
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 3.3968088626861572,
      "step": 2364
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 2364,
      "training_step_time": 10.168197870254517
    },
    {
      "epoch": 9.0179443359375e-06,
      "model_forward_time": 2.433906316757202,
      "step": 2364
    },
    {
      "epoch": 9.0179443359375e-06,
      "step": 2364,
      "training_step_time": 10.283520698547363
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 3.1828458309173584,
      "step": 2365
    },
    {
      "dynamics_loss_avg": 0.0637644050642848,
      "epoch": 9.021759033203126e-06,
      "step": 2365
    },
    {
      "action_loss_avg": 0.044302492402493954,
      "epoch": 9.021759033203126e-06,
      "step": 2365
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 2365,
      "training_step_time": 9.961028575897217
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 3.1356961727142334,
      "step": 2365
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 2365,
      "training_step_time": 9.911375284194946
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 3.1398556232452393,
      "step": 2365
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 2365,
      "training_step_time": 9.91826319694519
    },
    {
      "epoch": 9.021759033203126e-06,
      "model_forward_time": 3.4191112518310547,
      "step": 2365
    },
    {
      "epoch": 9.021759033203126e-06,
      "step": 2365,
      "training_step_time": 10.290397882461548
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 3.180081367492676,
      "step": 2366
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 2366,
      "training_step_time": 9.961461544036865
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 3.1401047706604004,
      "step": 2366
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 2366,
      "training_step_time": 9.917697429656982
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 3.1401872634887695,
      "step": 2366
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 2366,
      "training_step_time": 9.924329280853271
    },
    {
      "epoch": 9.02557373046875e-06,
      "model_forward_time": 3.136563539505005,
      "step": 2366
    },
    {
      "epoch": 9.02557373046875e-06,
      "step": 2366,
      "training_step_time": 10.038039922714233
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 3.1639139652252197,
      "step": 2367
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 2367,
      "training_step_time": 10.213132858276367
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 3.139017105102539,
      "step": 2367
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 2367,
      "training_step_time": 9.913655281066895
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 3.1422200202941895,
      "step": 2367
    },
    {
      "dynamics_loss_avg": 0.08145638778805733,
      "epoch": 9.029388427734375e-06,
      "step": 2367
    },
    {
      "action_loss_avg": 0.057246594689786434,
      "epoch": 9.029388427734375e-06,
      "step": 2367
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 2367,
      "training_step_time": 9.918164014816284
    },
    {
      "epoch": 9.029388427734375e-06,
      "model_forward_time": 3.1341583728790283,
      "step": 2367
    },
    {
      "epoch": 9.029388427734375e-06,
      "step": 2367,
      "training_step_time": 10.287962913513184
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 3.191460609436035,
      "step": 2368
    },
    {
      "epoch": 9.033203125e-06,
      "step": 2368,
      "training_step_time": 9.970834016799927
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 3.391792058944702,
      "step": 2368
    },
    {
      "epoch": 9.033203125e-06,
      "step": 2368,
      "training_step_time": 10.159379243850708
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 3.1379547119140625,
      "step": 2368
    },
    {
      "epoch": 9.033203125e-06,
      "step": 2368,
      "training_step_time": 9.912288904190063
    },
    {
      "epoch": 9.033203125e-06,
      "model_forward_time": 3.1356351375579834,
      "step": 2368
    },
    {
      "epoch": 9.033203125e-06,
      "step": 2368,
      "training_step_time": 10.276962518692017
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 3.1774582862854004,
      "step": 2369
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 2369,
      "training_step_time": 9.951627016067505
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 3.1399571895599365,
      "step": 2369
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 2369,
      "training_step_time": 9.917188167572021
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 3.424125909805298,
      "step": 2369
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 2369,
      "training_step_time": 10.1947762966156
    },
    {
      "epoch": 9.037017822265626e-06,
      "model_forward_time": 2.4365227222442627,
      "step": 2369
    },
    {
      "epoch": 9.037017822265626e-06,
      "step": 2369,
      "training_step_time": 10.296524047851562
    },
    {
      "epoch": 9.04083251953125e-06,
      "grad_norm": 0.6158942580223083,
      "learning_rate": 1e-05,
      "loss": 0.1065,
      "step": 2370
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 3.180776834487915,
      "step": 2370
    },
    {
      "dynamics_loss_avg": 0.07714102789759636,
      "epoch": 9.04083251953125e-06,
      "step": 2370
    },
    {
      "action_loss_avg": 0.050053957290947436,
      "epoch": 9.04083251953125e-06,
      "step": 2370
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 2370,
      "training_step_time": 9.953940153121948
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 3.1401453018188477,
      "step": 2370
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 2370,
      "training_step_time": 9.908771991729736
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 3.136514186859131,
      "step": 2370
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 2370,
      "training_step_time": 9.902225017547607
    },
    {
      "epoch": 9.04083251953125e-06,
      "model_forward_time": 3.140960693359375,
      "step": 2370
    },
    {
      "epoch": 9.04083251953125e-06,
      "step": 2370,
      "training_step_time": 10.286596536636353
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 3.1760568618774414,
      "step": 2371
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 2371,
      "training_step_time": 9.947903871536255
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 3.135127067565918,
      "step": 2371
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 2371,
      "training_step_time": 9.902516603469849
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 3.1342012882232666,
      "step": 2371
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 2371,
      "training_step_time": 9.902028560638428
    },
    {
      "epoch": 9.044647216796875e-06,
      "model_forward_time": 3.136765241622925,
      "step": 2371
    },
    {
      "epoch": 9.044647216796875e-06,
      "step": 2371,
      "training_step_time": 10.02217960357666
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 2.4443297386169434,
      "step": 2372
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 2372,
      "training_step_time": 7.730373382568359
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 3.134962797164917,
      "step": 2372
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 2372,
      "training_step_time": 9.898818731307983
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 2.4295594692230225,
      "step": 2372
    },
    {
      "dynamics_loss_avg": 0.07977285012602806,
      "epoch": 9.0484619140625e-06,
      "step": 2372
    },
    {
      "action_loss_avg": 0.04814018029719591,
      "epoch": 9.0484619140625e-06,
      "step": 2372
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 2372,
      "training_step_time": 7.449588775634766
    },
    {
      "epoch": 9.0484619140625e-06,
      "model_forward_time": 2.4265966415405273,
      "step": 2372
    },
    {
      "epoch": 9.0484619140625e-06,
      "step": 2372,
      "training_step_time": 10.013157606124878
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 3.1560609340667725,
      "step": 2373
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 2373,
      "training_step_time": 9.919392585754395
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 3.132244825363159,
      "step": 2373
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 2373,
      "training_step_time": 9.90015435218811
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 3.380251169204712,
      "step": 2373
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 2373,
      "training_step_time": 10.159669160842896
    },
    {
      "epoch": 9.052276611328126e-06,
      "model_forward_time": 2.435199499130249,
      "step": 2373
    },
    {
      "epoch": 9.052276611328126e-06,
      "step": 2373,
      "training_step_time": 10.277069568634033
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 3.1901423931121826,
      "step": 2374
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 2374,
      "training_step_time": 9.962314128875732
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 3.136892080307007,
      "step": 2374
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 2374,
      "training_step_time": 9.910853385925293
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 3.139815092086792,
      "step": 2374
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 2374,
      "training_step_time": 10.185412168502808
    },
    {
      "epoch": 9.05609130859375e-06,
      "model_forward_time": 3.1504690647125244,
      "step": 2374
    },
    {
      "epoch": 9.05609130859375e-06,
      "step": 2374,
      "training_step_time": 10.335354804992676
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 3.1702349185943604,
      "step": 2375
    },
    {
      "dynamics_loss_avg": 0.05622371174395084,
      "epoch": 9.059906005859375e-06,
      "step": 2375
    },
    {
      "action_loss_avg": 0.03979094084352255,
      "epoch": 9.059906005859375e-06,
      "step": 2375
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 2375,
      "training_step_time": 9.95588994026184
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 3.147399663925171,
      "step": 2375
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 2375,
      "training_step_time": 9.929733991622925
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 3.1551003456115723,
      "step": 2375
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 2375,
      "training_step_time": 9.935532569885254
    },
    {
      "epoch": 9.059906005859375e-06,
      "model_forward_time": 2.4436326026916504,
      "step": 2375
    },
    {
      "epoch": 9.059906005859375e-06,
      "step": 2375,
      "training_step_time": 10.049983739852905
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 2.805631160736084,
      "step": 2376
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 2376,
      "training_step_time": 7.835325241088867
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 2.4371743202209473,
      "step": 2376
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 2376,
      "training_step_time": 7.463570833206177
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 3.140779495239258,
      "step": 2376
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 2376,
      "training_step_time": 9.910675287246704
    },
    {
      "epoch": 9.063720703125e-06,
      "model_forward_time": 3.1412508487701416,
      "step": 2376
    },
    {
      "epoch": 9.063720703125e-06,
      "step": 2376,
      "training_step_time": 10.036402702331543
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 3.171372413635254,
      "step": 2377
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 2377,
      "training_step_time": 9.954075336456299
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 3.3967792987823486,
      "step": 2377
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 2377,
      "training_step_time": 10.164210557937622
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 3.1392264366149902,
      "step": 2377
    },
    {
      "dynamics_loss_avg": 0.06716423984616995,
      "epoch": 9.067535400390624e-06,
      "step": 2377
    },
    {
      "action_loss_avg": 0.04172267857939005,
      "epoch": 9.067535400390624e-06,
      "step": 2377
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 2377,
      "training_step_time": 9.909865856170654
    },
    {
      "epoch": 9.067535400390624e-06,
      "model_forward_time": 3.1307566165924072,
      "step": 2377
    },
    {
      "epoch": 9.067535400390624e-06,
      "step": 2377,
      "training_step_time": 10.026196479797363
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 3.1633245944976807,
      "step": 2378
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 2378,
      "training_step_time": 9.931405305862427
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 3.1357266902923584,
      "step": 2378
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 2378,
      "training_step_time": 10.167348861694336
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 3.137096881866455,
      "step": 2378
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 2378,
      "training_step_time": 9.902533054351807
    },
    {
      "epoch": 9.07135009765625e-06,
      "model_forward_time": 3.1312966346740723,
      "step": 2378
    },
    {
      "epoch": 9.07135009765625e-06,
      "step": 2378,
      "training_step_time": 10.011438608169556
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 3.186354875564575,
      "step": 2379
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 2379,
      "training_step_time": 9.963215827941895
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 3.1459097862243652,
      "step": 2379
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 2379,
      "training_step_time": 9.919549942016602
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 3.4390227794647217,
      "step": 2379
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 2379,
      "training_step_time": 10.209734678268433
    },
    {
      "epoch": 9.075164794921875e-06,
      "model_forward_time": 3.1401655673980713,
      "step": 2379
    },
    {
      "epoch": 9.075164794921875e-06,
      "step": 2379,
      "training_step_time": 10.279046297073364
    },
    {
      "epoch": 9.0789794921875e-06,
      "grad_norm": 0.4235542118549347,
      "learning_rate": 1e-05,
      "loss": 0.1047,
      "step": 2380
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 3.1822550296783447,
      "step": 2380
    },
    {
      "dynamics_loss_avg": 0.049601280316710475,
      "epoch": 9.0789794921875e-06,
      "step": 2380
    },
    {
      "action_loss_avg": 0.032595170103013514,
      "epoch": 9.0789794921875e-06,
      "step": 2380
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 2380,
      "training_step_time": 9.949447631835938
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 3.142225742340088,
      "step": 2380
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 2380,
      "training_step_time": 9.911802768707275
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 3.1427910327911377,
      "step": 2380
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 2380,
      "training_step_time": 9.912191152572632
    },
    {
      "epoch": 9.0789794921875e-06,
      "model_forward_time": 3.400799036026001,
      "step": 2380
    },
    {
      "epoch": 9.0789794921875e-06,
      "step": 2380,
      "training_step_time": 10.264501810073853
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 3.1778314113616943,
      "step": 2381
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 2381,
      "training_step_time": 9.949596166610718
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 3.146778106689453,
      "step": 2381
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 2381,
      "training_step_time": 9.916497945785522
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 3.1422362327575684,
      "step": 2381
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 2381,
      "training_step_time": 9.913512945175171
    },
    {
      "epoch": 9.082794189453124e-06,
      "model_forward_time": 3.145873785018921,
      "step": 2381
    },
    {
      "epoch": 9.082794189453124e-06,
      "step": 2381,
      "training_step_time": 10.020429134368896
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 3.4116435050964355,
      "step": 2382
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 2382,
      "training_step_time": 10.197822332382202
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 2.4361958503723145,
      "step": 2382
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 2382,
      "training_step_time": 7.464375019073486
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 2.427070140838623,
      "step": 2382
    },
    {
      "dynamics_loss_avg": 0.055344168562442064,
      "epoch": 9.08660888671875e-06,
      "step": 2382
    },
    {
      "action_loss_avg": 0.03794833482243121,
      "epoch": 9.08660888671875e-06,
      "step": 2382
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 2382,
      "training_step_time": 7.448569059371948
    },
    {
      "epoch": 9.08660888671875e-06,
      "model_forward_time": 3.1322743892669678,
      "step": 2382
    },
    {
      "epoch": 9.08660888671875e-06,
      "step": 2382,
      "training_step_time": 10.025213479995728
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 3.1793196201324463,
      "step": 2383
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 2383,
      "training_step_time": 9.94704818725586
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 3.418978691101074,
      "step": 2383
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 2383,
      "training_step_time": 10.183380365371704
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 2.4334757328033447,
      "step": 2383
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 2383,
      "training_step_time": 7.462822914123535
    },
    {
      "epoch": 9.090423583984375e-06,
      "model_forward_time": 3.1337811946868896,
      "step": 2383
    },
    {
      "epoch": 9.090423583984375e-06,
      "step": 2383,
      "training_step_time": 10.03020453453064
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 3.1733884811401367,
      "step": 2384
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 2384,
      "training_step_time": 9.945964574813843
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 3.135572671890259,
      "step": 2384
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 2384,
      "training_step_time": 9.903508424758911
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 3.1351795196533203,
      "step": 2384
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 2384,
      "training_step_time": 9.901132106781006
    },
    {
      "epoch": 9.09423828125e-06,
      "model_forward_time": 3.3982882499694824,
      "step": 2384
    },
    {
      "epoch": 9.09423828125e-06,
      "step": 2384,
      "training_step_time": 10.34131932258606
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 3.178363800048828,
      "step": 2385
    },
    {
      "dynamics_loss_avg": 0.04904789123684168,
      "epoch": 9.098052978515624e-06,
      "step": 2385
    },
    {
      "action_loss_avg": 0.04200157271698117,
      "epoch": 9.098052978515624e-06,
      "step": 2385
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 2385,
      "training_step_time": 9.947717905044556
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 2.434001922607422,
      "step": 2385
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 2385,
      "training_step_time": 7.4556660652160645
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 3.1329822540283203,
      "step": 2385
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 2385,
      "training_step_time": 9.898278713226318
    },
    {
      "epoch": 9.098052978515624e-06,
      "model_forward_time": 3.4117794036865234,
      "step": 2385
    },
    {
      "epoch": 9.098052978515624e-06,
      "step": 2385,
      "training_step_time": 10.314275741577148
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 2.473388671875,
      "step": 2386
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 2386,
      "training_step_time": 7.500450134277344
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 3.1442432403564453,
      "step": 2386
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 2386,
      "training_step_time": 9.914066791534424
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 2.4370298385620117,
      "step": 2386
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 2386,
      "training_step_time": 7.464243650436401
    },
    {
      "epoch": 9.10186767578125e-06,
      "model_forward_time": 3.137849807739258,
      "step": 2386
    },
    {
      "epoch": 9.10186767578125e-06,
      "step": 2386,
      "training_step_time": 10.036108493804932
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 3.168687105178833,
      "step": 2387
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 2387,
      "training_step_time": 9.935828447341919
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 3.3857364654541016,
      "step": 2387
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 2387,
      "training_step_time": 10.151691913604736
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 3.140120029449463,
      "step": 2387
    },
    {
      "dynamics_loss_avg": 0.06320159584283828,
      "epoch": 9.105682373046875e-06,
      "step": 2387
    },
    {
      "action_loss_avg": 0.036756913061253726,
      "epoch": 9.105682373046875e-06,
      "step": 2387
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 2387,
      "training_step_time": 9.911392211914062
    },
    {
      "epoch": 9.105682373046875e-06,
      "model_forward_time": 3.1396257877349854,
      "step": 2387
    },
    {
      "epoch": 9.105682373046875e-06,
      "step": 2387,
      "training_step_time": 10.019420862197876
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 3.154423236846924,
      "step": 2388
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 2388,
      "training_step_time": 9.92738676071167
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 3.1382346153259277,
      "step": 2388
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 2388,
      "training_step_time": 10.169780492782593
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 2.4324097633361816,
      "step": 2388
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 2388,
      "training_step_time": 7.46331524848938
    },
    {
      "epoch": 9.1094970703125e-06,
      "model_forward_time": 3.132704973220825,
      "step": 2388
    },
    {
      "epoch": 9.1094970703125e-06,
      "step": 2388,
      "training_step_time": 10.042064905166626
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 3.174260377883911,
      "step": 2389
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 2389,
      "training_step_time": 9.94398283958435
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 3.138925552368164,
      "step": 2389
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 2389,
      "training_step_time": 9.910138845443726
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 3.13065767288208,
      "step": 2389
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 2389,
      "training_step_time": 9.898259401321411
    },
    {
      "epoch": 9.113311767578124e-06,
      "model_forward_time": 3.3927292823791504,
      "step": 2389
    },
    {
      "epoch": 9.113311767578124e-06,
      "step": 2389,
      "training_step_time": 10.344388008117676
    },
    {
      "epoch": 9.11712646484375e-06,
      "grad_norm": 0.6745353937149048,
      "learning_rate": 1e-05,
      "loss": 0.1056,
      "step": 2390
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 2.469025135040283,
      "step": 2390
    },
    {
      "dynamics_loss_avg": 0.054664444737136365,
      "epoch": 9.11712646484375e-06,
      "step": 2390
    },
    {
      "action_loss_avg": 0.04344486985355615,
      "epoch": 9.11712646484375e-06,
      "step": 2390
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 2390,
      "training_step_time": 7.495527982711792
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 3.1304209232330322,
      "step": 2390
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 2390,
      "training_step_time": 9.890740394592285
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 3.1372759342193604,
      "step": 2390
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 2390,
      "training_step_time": 9.90575647354126
    },
    {
      "epoch": 9.11712646484375e-06,
      "model_forward_time": 3.1352412700653076,
      "step": 2390
    },
    {
      "epoch": 9.11712646484375e-06,
      "step": 2390,
      "training_step_time": 10.286160945892334
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 3.1811740398406982,
      "step": 2391
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 2391,
      "training_step_time": 9.954269647598267
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 3.1378042697906494,
      "step": 2391
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 2391,
      "training_step_time": 9.902493000030518
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 3.1309540271759033,
      "step": 2391
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 2391,
      "training_step_time": 9.89951229095459
    },
    {
      "epoch": 9.120941162109375e-06,
      "model_forward_time": 3.136234760284424,
      "step": 2391
    },
    {
      "epoch": 9.120941162109375e-06,
      "step": 2391,
      "training_step_time": 10.270312309265137
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 3.4289956092834473,
      "step": 2392
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 2392,
      "training_step_time": 10.19803500175476
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 2.4325551986694336,
      "step": 2392
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 2392,
      "training_step_time": 7.457226037979126
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 3.131758451461792,
      "step": 2392
    },
    {
      "dynamics_loss_avg": 0.062392018362879755,
      "epoch": 9.124755859375e-06,
      "step": 2392
    },
    {
      "action_loss_avg": 0.038771549426019195,
      "epoch": 9.124755859375e-06,
      "step": 2392
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 2392,
      "training_step_time": 9.903791189193726
    },
    {
      "epoch": 9.124755859375e-06,
      "model_forward_time": 2.433797597885132,
      "step": 2392
    },
    {
      "epoch": 9.124755859375e-06,
      "step": 2392,
      "training_step_time": 10.036750555038452
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 3.1480958461761475,
      "step": 2393
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 2393,
      "training_step_time": 9.915471315383911
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 3.137035608291626,
      "step": 2393
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 2393,
      "training_step_time": 9.905725955963135
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 3.377896308898926,
      "step": 2393
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 2393,
      "training_step_time": 10.14577031135559
    },
    {
      "epoch": 9.128570556640624e-06,
      "model_forward_time": 2.4325363636016846,
      "step": 2393
    },
    {
      "epoch": 9.128570556640624e-06,
      "step": 2393,
      "training_step_time": 10.048004627227783
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 3.1830837726593018,
      "step": 2394
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 2394,
      "training_step_time": 9.951932668685913
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 3.1339409351348877,
      "step": 2394
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 2394,
      "training_step_time": 9.904233932495117
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 3.134974718093872,
      "step": 2394
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 2394,
      "training_step_time": 9.9010488986969
    },
    {
      "epoch": 9.13238525390625e-06,
      "model_forward_time": 3.4056754112243652,
      "step": 2394
    },
    {
      "epoch": 9.13238525390625e-06,
      "step": 2394,
      "training_step_time": 10.279964923858643
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 3.17465877532959,
      "step": 2395
    },
    {
      "dynamics_loss_avg": 0.07281227838248014,
      "epoch": 9.136199951171875e-06,
      "step": 2395
    },
    {
      "action_loss_avg": 0.046727341040968894,
      "epoch": 9.136199951171875e-06,
      "step": 2395
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 2395,
      "training_step_time": 9.944971799850464
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 2.4314939975738525,
      "step": 2395
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 2395,
      "training_step_time": 7.453817367553711
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 3.133824348449707,
      "step": 2395
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 2395,
      "training_step_time": 9.90198278427124
    },
    {
      "epoch": 9.136199951171875e-06,
      "model_forward_time": 3.133294105529785,
      "step": 2395
    },
    {
      "epoch": 9.136199951171875e-06,
      "step": 2395,
      "training_step_time": 10.049794435501099
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 3.4676413536071777,
      "step": 2396
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 2396,
      "training_step_time": 10.243871450424194
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 3.132225513458252,
      "step": 2396
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 2396,
      "training_step_time": 9.906486988067627
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 3.1375064849853516,
      "step": 2396
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 2396,
      "training_step_time": 9.906047105789185
    },
    {
      "epoch": 9.1400146484375e-06,
      "model_forward_time": 3.134277582168579,
      "step": 2396
    },
    {
      "epoch": 9.1400146484375e-06,
      "step": 2396,
      "training_step_time": 10.300637006759644
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 3.161930561065674,
      "step": 2397
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 2397,
      "training_step_time": 10.190350770950317
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 3.1286606788635254,
      "step": 2397
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 2397,
      "training_step_time": 9.892935037612915
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 3.131412982940674,
      "step": 2397
    },
    {
      "dynamics_loss_avg": 0.06956415586173534,
      "epoch": 9.143829345703124e-06,
      "step": 2397
    },
    {
      "action_loss_avg": 0.04074772428721189,
      "epoch": 9.143829345703124e-06,
      "step": 2397
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 2397,
      "training_step_time": 9.90562891960144
    },
    {
      "epoch": 9.143829345703124e-06,
      "model_forward_time": 2.4383115768432617,
      "step": 2397
    },
    {
      "epoch": 9.143829345703124e-06,
      "step": 2397,
      "training_step_time": 10.271567583084106
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 3.1858069896698,
      "step": 2398
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 2398,
      "training_step_time": 9.955519199371338
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 2.435542345046997,
      "step": 2398
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 2398,
      "training_step_time": 7.45860743522644
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 3.4177775382995605,
      "step": 2398
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 2398,
      "training_step_time": 10.183661699295044
    },
    {
      "epoch": 9.14764404296875e-06,
      "model_forward_time": 3.1374008655548096,
      "step": 2398
    },
    {
      "epoch": 9.14764404296875e-06,
      "step": 2398,
      "training_step_time": 10.315433263778687
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 3.181349992752075,
      "step": 2399
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 2399,
      "training_step_time": 9.947020292282104
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 3.142408847808838,
      "step": 2399
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 2399,
      "training_step_time": 9.907250881195068
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 3.138277769088745,
      "step": 2399
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 2399,
      "training_step_time": 9.903089046478271
    },
    {
      "epoch": 9.151458740234375e-06,
      "model_forward_time": 3.403312921524048,
      "step": 2399
    },
    {
      "epoch": 9.151458740234375e-06,
      "step": 2399,
      "training_step_time": 10.264191627502441
    },
    {
      "epoch": 9.1552734375e-06,
      "grad_norm": 0.7128163576126099,
      "learning_rate": 1e-05,
      "loss": 0.1035,
      "step": 2400
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 3.180593490600586,
      "step": 2400
    },
    {
      "dynamics_loss_avg": 0.05924373306334019,
      "epoch": 9.1552734375e-06,
      "step": 2400
    },
    {
      "action_loss_avg": 0.03954944172874093,
      "epoch": 9.1552734375e-06,
      "step": 2400
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 2400,
      "training_step_time": 9.94834566116333
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 3.145009994506836,
      "step": 2400
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 2400,
      "training_step_time": 9.916855335235596
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 3.1474082469940186,
      "step": 2400
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 2400,
      "training_step_time": 9.916786193847656
    },
    {
      "epoch": 9.1552734375e-06,
      "model_forward_time": 3.4190242290496826,
      "step": 2400
    },
    {
      "epoch": 9.1552734375e-06,
      "step": 2400,
      "training_step_time": 10.310368299484253
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 3.184154510498047,
      "step": 2401
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 2401,
      "training_step_time": 9.960002660751343
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 3.149730920791626,
      "step": 2401
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 2401,
      "training_step_time": 9.927560567855835
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 3.1485791206359863,
      "step": 2401
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 2401,
      "training_step_time": 9.92687463760376
    },
    {
      "epoch": 9.159088134765624e-06,
      "model_forward_time": 3.152670383453369,
      "step": 2401
    },
    {
      "epoch": 9.159088134765624e-06,
      "step": 2401,
      "training_step_time": 10.322725772857666
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 3.1795566082000732,
      "step": 2402
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 2402,
      "training_step_time": 10.244971752166748
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 3.1503512859344482,
      "step": 2402
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 2402,
      "training_step_time": 9.924206733703613
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 3.1513118743896484,
      "step": 2402
    },
    {
      "dynamics_loss_avg": 0.053808167949318884,
      "epoch": 9.16290283203125e-06,
      "step": 2402
    },
    {
      "action_loss_avg": 0.037259122263640164,
      "epoch": 9.16290283203125e-06,
      "step": 2402
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 2402,
      "training_step_time": 9.925096035003662
    },
    {
      "epoch": 9.16290283203125e-06,
      "model_forward_time": 3.1460859775543213,
      "step": 2402
    },
    {
      "epoch": 9.16290283203125e-06,
      "step": 2402,
      "training_step_time": 10.280471563339233
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 3.1851186752319336,
      "step": 2403
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 2403,
      "training_step_time": 9.959230422973633
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 2.438934803009033,
      "step": 2403
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 2403,
      "training_step_time": 7.727680206298828
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 3.1406495571136475,
      "step": 2403
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 2403,
      "training_step_time": 9.914875507354736
    },
    {
      "epoch": 9.166717529296875e-06,
      "model_forward_time": 3.1440436840057373,
      "step": 2403
    },
    {
      "epoch": 9.166717529296875e-06,
      "step": 2403,
      "training_step_time": 10.029017925262451
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 3.192021608352661,
      "step": 2404
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 2404,
      "training_step_time": 9.96552586555481
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 3.145031452178955,
      "step": 2404
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 2404,
      "training_step_time": 9.915147304534912
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 3.144174337387085,
      "step": 2404
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 2404,
      "training_step_time": 10.17351508140564
    },
    {
      "epoch": 9.1705322265625e-06,
      "model_forward_time": 2.438382625579834,
      "step": 2404
    },
    {
      "epoch": 9.1705322265625e-06,
      "step": 2404,
      "training_step_time": 10.266428709030151
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 3.1868231296539307,
      "step": 2405
    },
    {
      "dynamics_loss_avg": 0.06780767720192671,
      "epoch": 9.174346923828124e-06,
      "step": 2405
    },
    {
      "action_loss_avg": 0.04298311173915863,
      "epoch": 9.174346923828124e-06,
      "step": 2405
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 2405,
      "training_step_time": 9.963139772415161
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 3.1429443359375,
      "step": 2405
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 2405,
      "training_step_time": 9.912131786346436
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 3.136693239212036,
      "step": 2405
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 2405,
      "training_step_time": 9.906258344650269
    },
    {
      "epoch": 9.174346923828124e-06,
      "model_forward_time": 3.4023005962371826,
      "step": 2405
    },
    {
      "epoch": 9.174346923828124e-06,
      "step": 2405,
      "training_step_time": 10.268531322479248
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 3.2005908489227295,
      "step": 2406
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 2406,
      "training_step_time": 9.968995332717896
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 3.1432931423187256,
      "step": 2406
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 2406,
      "training_step_time": 9.909406185150146
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 3.1388278007507324,
      "step": 2406
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 2406,
      "training_step_time": 9.906020879745483
    },
    {
      "epoch": 9.17816162109375e-06,
      "model_forward_time": 3.142972946166992,
      "step": 2406
    },
    {
      "epoch": 9.17816162109375e-06,
      "step": 2406,
      "training_step_time": 10.321462392807007
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 3.182227373123169,
      "step": 2407
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 2407,
      "training_step_time": 10.208295583724976
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 2.439824342727661,
      "step": 2407
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 2407,
      "training_step_time": 7.468886613845825
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 3.1407012939453125,
      "step": 2407
    },
    {
      "dynamics_loss_avg": 0.07134542278945447,
      "epoch": 9.181976318359375e-06,
      "step": 2407
    },
    {
      "action_loss_avg": 0.037045967392623426,
      "epoch": 9.181976318359375e-06,
      "step": 2407
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 2407,
      "training_step_time": 9.918560266494751
    },
    {
      "epoch": 9.181976318359375e-06,
      "model_forward_time": 3.146138906478882,
      "step": 2407
    },
    {
      "epoch": 9.181976318359375e-06,
      "step": 2407,
      "training_step_time": 10.312352657318115
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 3.1785857677459717,
      "step": 2408
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 2408,
      "training_step_time": 9.956880331039429
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 3.142860174179077,
      "step": 2408
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 2408,
      "training_step_time": 10.18965220451355
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 3.1456971168518066,
      "step": 2408
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 2408,
      "training_step_time": 9.918582677841187
    },
    {
      "epoch": 9.185791015625e-06,
      "model_forward_time": 3.144963502883911,
      "step": 2408
    },
    {
      "epoch": 9.185791015625e-06,
      "step": 2408,
      "training_step_time": 10.314102411270142
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 3.169743061065674,
      "step": 2409
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 2409,
      "training_step_time": 9.93586015701294
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 3.138493061065674,
      "step": 2409
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 2409,
      "training_step_time": 9.906740665435791
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 3.4073002338409424,
      "step": 2409
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 2409,
      "training_step_time": 10.172656536102295
    },
    {
      "epoch": 9.189605712890624e-06,
      "model_forward_time": 3.1427927017211914,
      "step": 2409
    },
    {
      "epoch": 9.189605712890624e-06,
      "step": 2409,
      "training_step_time": 10.26130485534668
    },
    {
      "epoch": 9.19342041015625e-06,
      "grad_norm": 0.305222749710083,
      "learning_rate": 1e-05,
      "loss": 0.1018,
      "step": 2410
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 3.1699094772338867,
      "step": 2410
    },
    {
      "dynamics_loss_avg": 0.05941515136510134,
      "epoch": 9.19342041015625e-06,
      "step": 2410
    },
    {
      "action_loss_avg": 0.041756481118500235,
      "epoch": 9.19342041015625e-06,
      "step": 2410
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 2410,
      "training_step_time": 9.939153909683228
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 3.1310572624206543,
      "step": 2410
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 2410,
      "training_step_time": 9.89534854888916
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 3.139955520629883,
      "step": 2410
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 2410,
      "training_step_time": 9.905965566635132
    },
    {
      "epoch": 9.19342041015625e-06,
      "model_forward_time": 2.433689832687378,
      "step": 2410
    },
    {
      "epoch": 9.19342041015625e-06,
      "step": 2410,
      "training_step_time": 10.024103879928589
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 2.4717071056365967,
      "step": 2411
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 2411,
      "training_step_time": 7.501214504241943
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 3.1404738426208496,
      "step": 2411
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 2411,
      "training_step_time": 9.912829637527466
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 3.148197889328003,
      "step": 2411
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 2411,
      "training_step_time": 9.9288969039917
    },
    {
      "epoch": 9.197235107421875e-06,
      "model_forward_time": 3.1465353965759277,
      "step": 2411
    },
    {
      "epoch": 9.197235107421875e-06,
      "step": 2411,
      "training_step_time": 10.311204195022583
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 3.4597949981689453,
      "step": 2412
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 2412,
      "training_step_time": 10.233378648757935
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 3.1411759853363037,
      "step": 2412
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 2412,
      "training_step_time": 9.916423797607422
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 3.141892194747925,
      "step": 2412
    },
    {
      "dynamics_loss_avg": 0.06434163339436054,
      "epoch": 9.2010498046875e-06,
      "step": 2412
    },
    {
      "action_loss_avg": 0.039671322982758286,
      "epoch": 9.2010498046875e-06,
      "step": 2412
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 2412,
      "training_step_time": 9.931785821914673
    },
    {
      "epoch": 9.2010498046875e-06,
      "model_forward_time": 3.140657424926758,
      "step": 2412
    },
    {
      "epoch": 9.2010498046875e-06,
      "step": 2412,
      "training_step_time": 10.33180832862854
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 3.1690168380737305,
      "step": 2413
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 2413,
      "training_step_time": 9.944865703582764
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 3.1457600593566895,
      "step": 2413
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 2413,
      "training_step_time": 10.179856777191162
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 3.1478044986724854,
      "step": 2413
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 2413,
      "training_step_time": 9.922468185424805
    },
    {
      "epoch": 9.204864501953124e-06,
      "model_forward_time": 3.1411991119384766,
      "step": 2413
    },
    {
      "epoch": 9.204864501953124e-06,
      "step": 2413,
      "training_step_time": 10.28775405883789
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 2.4807207584381104,
      "step": 2414
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 2414,
      "training_step_time": 7.510765552520752
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 3.1395328044891357,
      "step": 2414
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 2414,
      "training_step_time": 9.909884691238403
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 3.4272468090057373,
      "step": 2414
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 2414,
      "training_step_time": 10.194481134414673
    },
    {
      "epoch": 9.20867919921875e-06,
      "model_forward_time": 3.13832426071167,
      "step": 2414
    },
    {
      "epoch": 9.20867919921875e-06,
      "step": 2414,
      "training_step_time": 10.291015625
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 3.1874444484710693,
      "step": 2415
    },
    {
      "dynamics_loss_avg": 0.0526728343218565,
      "epoch": 9.212493896484375e-06,
      "step": 2415
    },
    {
      "action_loss_avg": 0.03516724891960621,
      "epoch": 9.212493896484375e-06,
      "step": 2415
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 2415,
      "training_step_time": 9.960319995880127
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 3.1380927562713623,
      "step": 2415
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 2415,
      "training_step_time": 9.907098054885864
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 3.1351499557495117,
      "step": 2415
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 2415,
      "training_step_time": 9.907451391220093
    },
    {
      "epoch": 9.212493896484375e-06,
      "model_forward_time": 2.434687376022339,
      "step": 2415
    },
    {
      "epoch": 9.212493896484375e-06,
      "step": 2415,
      "training_step_time": 10.021105289459229
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 3.505760669708252,
      "step": 2416
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 2416,
      "training_step_time": 10.27497935295105
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 2.433476448059082,
      "step": 2416
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 2416,
      "training_step_time": 7.461436033248901
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 3.1301329135894775,
      "step": 2416
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 2416,
      "training_step_time": 9.897316455841064
    },
    {
      "epoch": 9.21630859375e-06,
      "model_forward_time": 3.138129711151123,
      "step": 2416
    },
    {
      "epoch": 9.21630859375e-06,
      "step": 2416,
      "training_step_time": 10.025360345840454
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 3.425182819366455,
      "step": 2417
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 2417,
      "training_step_time": 10.196232318878174
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 3.1380043029785156,
      "step": 2417
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 2417,
      "training_step_time": 9.905473709106445
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 3.136284112930298,
      "step": 2417
    },
    {
      "dynamics_loss_avg": 0.07891236217692495,
      "epoch": 9.220123291015624e-06,
      "step": 2417
    },
    {
      "action_loss_avg": 0.05183910317718983,
      "epoch": 9.220123291015624e-06,
      "step": 2417
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 2417,
      "training_step_time": 9.90808916091919
    },
    {
      "epoch": 9.220123291015624e-06,
      "model_forward_time": 3.1391713619232178,
      "step": 2417
    },
    {
      "epoch": 9.220123291015624e-06,
      "step": 2417,
      "training_step_time": 10.34280776977539
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 3.156619071960449,
      "step": 2418
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 2418,
      "training_step_time": 9.929548025131226
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 3.1339378356933594,
      "step": 2418
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 2418,
      "training_step_time": 10.178444385528564
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 2.43400239944458,
      "step": 2418
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 2418,
      "training_step_time": 7.465587139129639
    },
    {
      "epoch": 9.22393798828125e-06,
      "model_forward_time": 3.1359870433807373,
      "step": 2418
    },
    {
      "epoch": 9.22393798828125e-06,
      "step": 2418,
      "training_step_time": 10.27531099319458
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 3.1803011894226074,
      "step": 2419
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 2419,
      "training_step_time": 9.951302528381348
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 3.1355140209198,
      "step": 2419
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 2419,
      "training_step_time": 9.9022958278656
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 3.1354479789733887,
      "step": 2419
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 2419,
      "training_step_time": 10.164878845214844
    },
    {
      "epoch": 9.227752685546875e-06,
      "model_forward_time": 3.135718584060669,
      "step": 2419
    },
    {
      "epoch": 9.227752685546875e-06,
      "step": 2419,
      "training_step_time": 10.27821159362793
    },
    {
      "epoch": 9.2315673828125e-06,
      "grad_norm": 0.566956639289856,
      "learning_rate": 1e-05,
      "loss": 0.1016,
      "step": 2420
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 3.1729516983032227,
      "step": 2420
    },
    {
      "dynamics_loss_avg": 0.0515507847070694,
      "epoch": 9.2315673828125e-06,
      "step": 2420
    },
    {
      "action_loss_avg": 0.03719417154788971,
      "epoch": 9.2315673828125e-06,
      "step": 2420
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 2420,
      "training_step_time": 9.943907976150513
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 3.1321818828582764,
      "step": 2420
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 2420,
      "training_step_time": 9.899127960205078
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 3.137502908706665,
      "step": 2420
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 2420,
      "training_step_time": 9.907019138336182
    },
    {
      "epoch": 9.2315673828125e-06,
      "model_forward_time": 3.4168243408203125,
      "step": 2420
    },
    {
      "epoch": 9.2315673828125e-06,
      "step": 2420,
      "training_step_time": 10.283448934555054
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 3.176370859146118,
      "step": 2421
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 2421,
      "training_step_time": 9.951566219329834
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 2.4321908950805664,
      "step": 2421
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 2421,
      "training_step_time": 7.4628143310546875
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 3.1378188133239746,
      "step": 2421
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 2421,
      "training_step_time": 9.908971786499023
    },
    {
      "epoch": 9.235382080078124e-06,
      "model_forward_time": 2.432513952255249,
      "step": 2421
    },
    {
      "epoch": 9.235382080078124e-06,
      "step": 2421,
      "training_step_time": 10.019763708114624
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 2.476104736328125,
      "step": 2422
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 2422,
      "training_step_time": 7.501305103302002
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 3.3762269020080566,
      "step": 2422
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 2422,
      "training_step_time": 10.137871026992798
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 3.134002447128296,
      "step": 2422
    },
    {
      "dynamics_loss_avg": 0.07344245091080666,
      "epoch": 9.23919677734375e-06,
      "step": 2422
    },
    {
      "action_loss_avg": 0.05167346056550741,
      "epoch": 9.23919677734375e-06,
      "step": 2422
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 2422,
      "training_step_time": 9.90319561958313
    },
    {
      "epoch": 9.23919677734375e-06,
      "model_forward_time": 3.1359870433807373,
      "step": 2422
    },
    {
      "epoch": 9.23919677734375e-06,
      "step": 2422,
      "training_step_time": 10.272935390472412
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 3.175821304321289,
      "step": 2423
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 2423,
      "training_step_time": 9.940092086791992
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 3.1392788887023926,
      "step": 2423
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 2423,
      "training_step_time": 9.907504796981812
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 3.4140472412109375,
      "step": 2423
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 2423,
      "training_step_time": 10.184030532836914
    },
    {
      "epoch": 9.243011474609375e-06,
      "model_forward_time": 3.1429238319396973,
      "step": 2423
    },
    {
      "epoch": 9.243011474609375e-06,
      "step": 2423,
      "training_step_time": 10.341319561004639
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 3.191540479660034,
      "step": 2424
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 2424,
      "training_step_time": 9.97132158279419
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 3.1455705165863037,
      "step": 2424
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 2424,
      "training_step_time": 9.920241594314575
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 3.1407063007354736,
      "step": 2424
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 2424,
      "training_step_time": 9.914671421051025
    },
    {
      "epoch": 9.246826171875e-06,
      "model_forward_time": 3.386040210723877,
      "step": 2424
    },
    {
      "epoch": 9.246826171875e-06,
      "step": 2424,
      "training_step_time": 10.299876928329468
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 3.1939613819122314,
      "step": 2425
    },
    {
      "dynamics_loss_avg": 0.0822883564978838,
      "epoch": 9.250640869140624e-06,
      "step": 2425
    },
    {
      "action_loss_avg": 0.04607465285807848,
      "epoch": 9.250640869140624e-06,
      "step": 2425
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 2425,
      "training_step_time": 9.964349508285522
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 3.142225980758667,
      "step": 2425
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 2425,
      "training_step_time": 9.910501480102539
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 3.140191078186035,
      "step": 2425
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 2425,
      "training_step_time": 9.913126468658447
    },
    {
      "epoch": 9.250640869140624e-06,
      "model_forward_time": 3.3985939025878906,
      "step": 2425
    },
    {
      "epoch": 9.250640869140624e-06,
      "step": 2425,
      "training_step_time": 10.264445781707764
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 3.152163505554199,
      "step": 2426
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 2426,
      "training_step_time": 9.924242973327637
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 3.1317148208618164,
      "step": 2426
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 2426,
      "training_step_time": 9.893426895141602
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 3.1302661895751953,
      "step": 2426
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 2426,
      "training_step_time": 9.894326448440552
    },
    {
      "epoch": 9.25445556640625e-06,
      "model_forward_time": 2.424945831298828,
      "step": 2426
    },
    {
      "epoch": 9.25445556640625e-06,
      "step": 2426,
      "training_step_time": 10.007224082946777
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 3.164461374282837,
      "step": 2427
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 2427,
      "training_step_time": 9.926806449890137
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 3.3988418579101562,
      "step": 2427
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 2427,
      "training_step_time": 10.159846067428589
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 3.130997896194458,
      "step": 2427
    },
    {
      "dynamics_loss_avg": 0.0608034323900938,
      "epoch": 9.258270263671875e-06,
      "step": 2427
    },
    {
      "action_loss_avg": 0.04198330417275429,
      "epoch": 9.258270263671875e-06,
      "step": 2427
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 2427,
      "training_step_time": 9.894912958145142
    },
    {
      "epoch": 9.258270263671875e-06,
      "model_forward_time": 3.131009578704834,
      "step": 2427
    },
    {
      "epoch": 9.258270263671875e-06,
      "step": 2427,
      "training_step_time": 10.273038387298584
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 3.154128074645996,
      "step": 2428
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 2428,
      "training_step_time": 9.921324968338013
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 3.1306958198547363,
      "step": 2428
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 2428,
      "training_step_time": 10.201566934585571
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 3.133263349533081,
      "step": 2428
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 2428,
      "training_step_time": 9.902743339538574
    },
    {
      "epoch": 9.2620849609375e-06,
      "model_forward_time": 3.133899211883545,
      "step": 2428
    },
    {
      "epoch": 9.2620849609375e-06,
      "step": 2428,
      "training_step_time": 10.312101602554321
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 3.157073497772217,
      "step": 2429
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 2429,
      "training_step_time": 9.92408013343811
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 2.4305646419525146,
      "step": 2429
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 2429,
      "training_step_time": 7.457354784011841
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 3.1292002201080322,
      "step": 2429
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 2429,
      "training_step_time": 9.89847445487976
    },
    {
      "epoch": 9.265899658203124e-06,
      "model_forward_time": 3.418816328048706,
      "step": 2429
    },
    {
      "epoch": 9.265899658203124e-06,
      "step": 2429,
      "training_step_time": 10.294275999069214
    },
    {
      "epoch": 9.26971435546875e-06,
      "grad_norm": 0.3687472343444824,
      "learning_rate": 1e-05,
      "loss": 0.1086,
      "step": 2430
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 3.17579984664917,
      "step": 2430
    },
    {
      "dynamics_loss_avg": 0.04582926370203495,
      "epoch": 9.26971435546875e-06,
      "step": 2430
    },
    {
      "action_loss_avg": 0.02513576466590166,
      "epoch": 9.26971435546875e-06,
      "step": 2430
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 2430,
      "training_step_time": 9.95456862449646
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 3.146135091781616,
      "step": 2430
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 2430,
      "training_step_time": 9.920796394348145
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 3.1448051929473877,
      "step": 2430
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 2430,
      "training_step_time": 9.922790050506592
    },
    {
      "epoch": 9.26971435546875e-06,
      "model_forward_time": 3.140843152999878,
      "step": 2430
    },
    {
      "epoch": 9.26971435546875e-06,
      "step": 2430,
      "training_step_time": 10.325372457504272
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 3.1784026622772217,
      "step": 2431
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 2431,
      "training_step_time": 9.959117650985718
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 3.138862133026123,
      "step": 2431
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 2431,
      "training_step_time": 9.915883302688599
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 3.1433539390563965,
      "step": 2431
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 2431,
      "training_step_time": 9.921588897705078
    },
    {
      "epoch": 9.273529052734375e-06,
      "model_forward_time": 3.1433370113372803,
      "step": 2431
    },
    {
      "epoch": 9.273529052734375e-06,
      "step": 2431,
      "training_step_time": 10.025673389434814
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 3.458780288696289,
      "step": 2432
    },
    {
      "epoch": 9.27734375e-06,
      "step": 2432,
      "training_step_time": 10.23682713508606
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 3.1430227756500244,
      "step": 2432
    },
    {
      "epoch": 9.27734375e-06,
      "step": 2432,
      "training_step_time": 9.922285079956055
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 3.156536817550659,
      "step": 2432
    },
    {
      "dynamics_loss_avg": 0.057698328606784345,
      "epoch": 9.27734375e-06,
      "step": 2432
    },
    {
      "action_loss_avg": 0.0338827857747674,
      "epoch": 9.27734375e-06,
      "step": 2432
    },
    {
      "epoch": 9.27734375e-06,
      "step": 2432,
      "training_step_time": 9.937956809997559
    },
    {
      "epoch": 9.27734375e-06,
      "model_forward_time": 3.13840389251709,
      "step": 2432
    },
    {
      "epoch": 9.27734375e-06,
      "step": 2432,
      "training_step_time": 10.013258457183838
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 3.2008097171783447,
      "step": 2433
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 2433,
      "training_step_time": 9.981046438217163
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 3.147975206375122,
      "step": 2433
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 2433,
      "training_step_time": 10.19908618927002
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 3.1465022563934326,
      "step": 2433
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 2433,
      "training_step_time": 9.920835494995117
    },
    {
      "epoch": 9.281158447265624e-06,
      "model_forward_time": 3.142390012741089,
      "step": 2433
    },
    {
      "epoch": 9.281158447265624e-06,
      "step": 2433,
      "training_step_time": 10.275708436965942
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 2.47586989402771,
      "step": 2434
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 2434,
      "training_step_time": 7.508458375930786
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 3.13346529006958,
      "step": 2434
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 2434,
      "training_step_time": 9.90681791305542
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 3.4215660095214844,
      "step": 2434
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 2434,
      "training_step_time": 10.197662353515625
    },
    {
      "epoch": 9.28497314453125e-06,
      "model_forward_time": 3.1362791061401367,
      "step": 2434
    },
    {
      "epoch": 9.28497314453125e-06,
      "step": 2434,
      "training_step_time": 10.268133640289307
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 3.183560848236084,
      "step": 2435
    },
    {
      "dynamics_loss_avg": 0.06610588803887367,
      "epoch": 9.288787841796875e-06,
      "step": 2435
    },
    {
      "action_loss_avg": 0.03768421402201057,
      "epoch": 9.288787841796875e-06,
      "step": 2435
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 2435,
      "training_step_time": 9.960230588912964
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 3.1331796646118164,
      "step": 2435
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 2435,
      "training_step_time": 9.912550210952759
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 3.1304545402526855,
      "step": 2435
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 2435,
      "training_step_time": 9.903451442718506
    },
    {
      "epoch": 9.288787841796875e-06,
      "model_forward_time": 2.4335570335388184,
      "step": 2435
    },
    {
      "epoch": 9.288787841796875e-06,
      "step": 2435,
      "training_step_time": 10.278119802474976
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 3.4651715755462646,
      "step": 2436
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 2436,
      "training_step_time": 10.23406457901001
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 3.1345860958099365,
      "step": 2436
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 2436,
      "training_step_time": 9.905022382736206
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 3.1330394744873047,
      "step": 2436
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 2436,
      "training_step_time": 9.911642074584961
    },
    {
      "epoch": 9.2926025390625e-06,
      "model_forward_time": 3.1354827880859375,
      "step": 2436
    },
    {
      "epoch": 9.2926025390625e-06,
      "step": 2436,
      "training_step_time": 10.033522129058838
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 3.4050896167755127,
      "step": 2437
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 2437,
      "training_step_time": 10.183826684951782
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 3.1380412578582764,
      "step": 2437
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 2437,
      "training_step_time": 9.909422874450684
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 2.434730052947998,
      "step": 2437
    },
    {
      "dynamics_loss_avg": 0.07066628150641918,
      "epoch": 9.296417236328124e-06,
      "step": 2437
    },
    {
      "action_loss_avg": 0.06146334540098906,
      "epoch": 9.296417236328124e-06,
      "step": 2437
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 2437,
      "training_step_time": 7.466143369674683
    },
    {
      "epoch": 9.296417236328124e-06,
      "model_forward_time": 2.4304025173187256,
      "step": 2437
    },
    {
      "epoch": 9.296417236328124e-06,
      "step": 2437,
      "training_step_time": 10.031982898712158
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 3.1780474185943604,
      "step": 2438
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 2438,
      "training_step_time": 9.944493532180786
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 3.135756492614746,
      "step": 2438
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 2438,
      "training_step_time": 9.898725032806396
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 3.38411021232605,
      "step": 2438
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 2438,
      "training_step_time": 10.149126291275024
    },
    {
      "epoch": 9.30023193359375e-06,
      "model_forward_time": 3.1381757259368896,
      "step": 2438
    },
    {
      "epoch": 9.30023193359375e-06,
      "step": 2438,
      "training_step_time": 10.268340110778809
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 3.181447982788086,
      "step": 2439
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 2439,
      "training_step_time": 9.947600841522217
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 3.1378934383392334,
      "step": 2439
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 2439,
      "training_step_time": 9.90299391746521
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 3.1421632766723633,
      "step": 2439
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 2439,
      "training_step_time": 9.908304929733276
    },
    {
      "epoch": 9.304046630859375e-06,
      "model_forward_time": 3.406707525253296,
      "step": 2439
    },
    {
      "epoch": 9.304046630859375e-06,
      "step": 2439,
      "training_step_time": 10.288705825805664
    },
    {
      "epoch": 9.307861328125e-06,
      "grad_norm": 0.3516888916492462,
      "learning_rate": 1e-05,
      "loss": 0.1066,
      "step": 2440
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 3.171138286590576,
      "step": 2440
    },
    {
      "dynamics_loss_avg": 0.06200920008122921,
      "epoch": 9.307861328125e-06,
      "step": 2440
    },
    {
      "action_loss_avg": 0.038273153640329835,
      "epoch": 9.307861328125e-06,
      "step": 2440
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 2440,
      "training_step_time": 9.940619230270386
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 3.133129835128784,
      "step": 2440
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 2440,
      "training_step_time": 9.89490032196045
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 3.1370978355407715,
      "step": 2440
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 2440,
      "training_step_time": 9.898360013961792
    },
    {
      "epoch": 9.307861328125e-06,
      "model_forward_time": 3.4144325256347656,
      "step": 2440
    },
    {
      "epoch": 9.307861328125e-06,
      "step": 2440,
      "training_step_time": 10.278491497039795
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 2.4728217124938965,
      "step": 2441
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 2441,
      "training_step_time": 7.49670934677124
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 3.133694648742676,
      "step": 2441
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 2441,
      "training_step_time": 9.896847248077393
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 3.1410069465637207,
      "step": 2441
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 2441,
      "training_step_time": 9.911161184310913
    },
    {
      "epoch": 9.311676025390624e-06,
      "model_forward_time": 3.137559652328491,
      "step": 2441
    },
    {
      "epoch": 9.311676025390624e-06,
      "step": 2441,
      "training_step_time": 10.301106452941895
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 3.177497148513794,
      "step": 2442
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 2442,
      "training_step_time": 9.95551872253418
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 3.418295383453369,
      "step": 2442
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 2442,
      "training_step_time": 10.186225652694702
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 3.141965627670288,
      "step": 2442
    },
    {
      "dynamics_loss_avg": 0.05452702511101961,
      "epoch": 9.31549072265625e-06,
      "step": 2442
    },
    {
      "action_loss_avg": 0.03501485073938966,
      "epoch": 9.31549072265625e-06,
      "step": 2442
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 2442,
      "training_step_time": 9.913355588912964
    },
    {
      "epoch": 9.31549072265625e-06,
      "model_forward_time": 3.141817569732666,
      "step": 2442
    },
    {
      "epoch": 9.31549072265625e-06,
      "step": 2442,
      "training_step_time": 10.028873205184937
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 2.4648804664611816,
      "step": 2443
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 2443,
      "training_step_time": 7.49355673789978
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 3.3983185291290283,
      "step": 2443
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 2443,
      "training_step_time": 10.166096925735474
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 3.1444828510284424,
      "step": 2443
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 2443,
      "training_step_time": 9.92079210281372
    },
    {
      "epoch": 9.319305419921875e-06,
      "model_forward_time": 3.143312692642212,
      "step": 2443
    },
    {
      "epoch": 9.319305419921875e-06,
      "step": 2443,
      "training_step_time": 10.085551023483276
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 3.185471296310425,
      "step": 2444
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 2444,
      "training_step_time": 9.956416130065918
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 3.1408517360687256,
      "step": 2444
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 2444,
      "training_step_time": 9.91031265258789
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 3.136683702468872,
      "step": 2444
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 2444,
      "training_step_time": 10.170276880264282
    },
    {
      "epoch": 9.3231201171875e-06,
      "model_forward_time": 3.13388729095459,
      "step": 2444
    },
    {
      "epoch": 9.3231201171875e-06,
      "step": 2444,
      "training_step_time": 10.30424451828003
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 3.171565055847168,
      "step": 2445
    },
    {
      "dynamics_loss_avg": 0.0727892430499196,
      "epoch": 9.326934814453124e-06,
      "step": 2445
    },
    {
      "action_loss_avg": 0.04131679153069854,
      "epoch": 9.326934814453124e-06,
      "step": 2445
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 2445,
      "training_step_time": 9.944663047790527
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 3.1393682956695557,
      "step": 2445
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 2445,
      "training_step_time": 9.91170048713684
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 3.1397953033447266,
      "step": 2445
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 2445,
      "training_step_time": 9.910595417022705
    },
    {
      "epoch": 9.326934814453124e-06,
      "model_forward_time": 3.3923380374908447,
      "step": 2445
    },
    {
      "epoch": 9.326934814453124e-06,
      "step": 2445,
      "training_step_time": 10.290379047393799
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 2.4694390296936035,
      "step": 2446
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 2446,
      "training_step_time": 7.506381034851074
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 3.132065534591675,
      "step": 2446
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 2446,
      "training_step_time": 9.904879331588745
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 3.1386239528656006,
      "step": 2446
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 2446,
      "training_step_time": 9.908457040786743
    },
    {
      "epoch": 9.33074951171875e-06,
      "model_forward_time": 2.434998035430908,
      "step": 2446
    },
    {
      "epoch": 9.33074951171875e-06,
      "step": 2446,
      "training_step_time": 10.266409635543823
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 3.4143946170806885,
      "step": 2447
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 2447,
      "training_step_time": 10.181654453277588
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 3.1373164653778076,
      "step": 2447
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 2447,
      "training_step_time": 9.90190315246582
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 3.12752103805542,
      "step": 2447
    },
    {
      "dynamics_loss_avg": 0.05925516523420811,
      "epoch": 9.334564208984375e-06,
      "step": 2447
    },
    {
      "action_loss_avg": 0.045683967880904675,
      "epoch": 9.334564208984375e-06,
      "step": 2447
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 2447,
      "training_step_time": 9.896662712097168
    },
    {
      "epoch": 9.334564208984375e-06,
      "model_forward_time": 3.131566286087036,
      "step": 2447
    },
    {
      "epoch": 9.334564208984375e-06,
      "step": 2447,
      "training_step_time": 10.288392305374146
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 3.165861129760742,
      "step": 2448
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 2448,
      "training_step_time": 9.929202795028687
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 3.132432222366333,
      "step": 2448
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 2448,
      "training_step_time": 10.17787790298462
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 3.1323981285095215,
      "step": 2448
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 2448,
      "training_step_time": 9.89824366569519
    },
    {
      "epoch": 9.33837890625e-06,
      "model_forward_time": 3.1407530307769775,
      "step": 2448
    },
    {
      "epoch": 9.33837890625e-06,
      "step": 2448,
      "training_step_time": 10.023093461990356
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 2.4776017665863037,
      "step": 2449
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 2449,
      "training_step_time": 7.506639242172241
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 2.4332523345947266,
      "step": 2449
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 2449,
      "training_step_time": 7.462769508361816
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 3.1356916427612305,
      "step": 2449
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 2449,
      "training_step_time": 9.904173135757446
    },
    {
      "epoch": 9.342193603515624e-06,
      "model_forward_time": 3.4014790058135986,
      "step": 2449
    },
    {
      "epoch": 9.342193603515624e-06,
      "step": 2449,
      "training_step_time": 10.297849893569946
    },
    {
      "epoch": 9.34600830078125e-06,
      "grad_norm": 0.5196924805641174,
      "learning_rate": 1e-05,
      "loss": 0.1003,
      "step": 2450
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 3.1616828441619873,
      "step": 2450
    },
    {
      "dynamics_loss_avg": 0.06395927667617798,
      "epoch": 9.34600830078125e-06,
      "step": 2450
    },
    {
      "action_loss_avg": 0.0370193250477314,
      "epoch": 9.34600830078125e-06,
      "step": 2450
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 2450,
      "training_step_time": 9.935742139816284
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 3.1352410316467285,
      "step": 2450
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 2450,
      "training_step_time": 9.904178857803345
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 3.1360278129577637,
      "step": 2450
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 2450,
      "training_step_time": 9.905120849609375
    },
    {
      "epoch": 9.34600830078125e-06,
      "model_forward_time": 3.137058973312378,
      "step": 2450
    },
    {
      "epoch": 9.34600830078125e-06,
      "step": 2450,
      "training_step_time": 10.264475345611572
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 3.170698642730713,
      "step": 2451
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 2451,
      "training_step_time": 9.933042049407959
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 3.125171422958374,
      "step": 2451
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 2451,
      "training_step_time": 9.886205196380615
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 3.1321394443511963,
      "step": 2451
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 2451,
      "training_step_time": 9.896161794662476
    },
    {
      "epoch": 9.349822998046875e-06,
      "model_forward_time": 3.127372980117798,
      "step": 2451
    },
    {
      "epoch": 9.349822998046875e-06,
      "step": 2451,
      "training_step_time": 10.25294804573059
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 2.7279489040374756,
      "step": 2452
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 2452,
      "training_step_time": 7.755897760391235
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 3.1293935775756836,
      "step": 2452
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 2452,
      "training_step_time": 9.89609980583191
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 2.431401252746582,
      "step": 2452
    },
    {
      "dynamics_loss_avg": 0.06563045652583241,
      "epoch": 9.3536376953125e-06,
      "step": 2452
    },
    {
      "action_loss_avg": 0.04606672348454595,
      "epoch": 9.3536376953125e-06,
      "step": 2452
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 2452,
      "training_step_time": 7.45466423034668
    },
    {
      "epoch": 9.3536376953125e-06,
      "model_forward_time": 3.130791425704956,
      "step": 2452
    },
    {
      "epoch": 9.3536376953125e-06,
      "step": 2452,
      "training_step_time": 10.305605173110962
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 3.170475482940674,
      "step": 2453
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 2453,
      "training_step_time": 9.942658185958862
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 3.134082555770874,
      "step": 2453
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 2453,
      "training_step_time": 9.912428617477417
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 3.406757354736328,
      "step": 2453
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 2453,
      "training_step_time": 10.189707040786743
    },
    {
      "epoch": 9.357452392578124e-06,
      "model_forward_time": 3.135989189147949,
      "step": 2453
    },
    {
      "epoch": 9.357452392578124e-06,
      "step": 2453,
      "training_step_time": 10.020019054412842
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 3.1835246086120605,
      "step": 2454
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 2454,
      "training_step_time": 9.955187320709229
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 2.430399179458618,
      "step": 2454
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 2454,
      "training_step_time": 7.45888614654541
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 3.128739356994629,
      "step": 2454
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 2454,
      "training_step_time": 10.167985439300537
    },
    {
      "epoch": 9.36126708984375e-06,
      "model_forward_time": 3.1373493671417236,
      "step": 2454
    },
    {
      "epoch": 9.36126708984375e-06,
      "step": 2454,
      "training_step_time": 10.298352479934692
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 2.4719343185424805,
      "step": 2455
    },
    {
      "dynamics_loss_avg": 0.06049764333292842,
      "epoch": 9.365081787109375e-06,
      "step": 2455
    },
    {
      "action_loss_avg": 0.035887589957565066,
      "epoch": 9.365081787109375e-06,
      "step": 2455
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 2455,
      "training_step_time": 7.5054731369018555
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 2.430908203125,
      "step": 2455
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 2455,
      "training_step_time": 7.458168983459473
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 3.1241488456726074,
      "step": 2455
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 2455,
      "training_step_time": 9.892869710922241
    },
    {
      "epoch": 9.365081787109375e-06,
      "model_forward_time": 3.1314749717712402,
      "step": 2455
    },
    {
      "epoch": 9.365081787109375e-06,
      "step": 2455,
      "training_step_time": 10.280585050582886
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 3.4568073749542236,
      "step": 2456
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 2456,
      "training_step_time": 10.221763610839844
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 3.1362714767456055,
      "step": 2456
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 2456,
      "training_step_time": 9.898993730545044
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 3.125044584274292,
      "step": 2456
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 2456,
      "training_step_time": 9.884804487228394
    },
    {
      "epoch": 9.368896484375e-06,
      "model_forward_time": 3.1305325031280518,
      "step": 2456
    },
    {
      "epoch": 9.368896484375e-06,
      "step": 2456,
      "training_step_time": 10.286972045898438
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 3.1697208881378174,
      "step": 2457
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 2457,
      "training_step_time": 10.216841220855713
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 3.130905866622925,
      "step": 2457
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 2457,
      "training_step_time": 9.896270990371704
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 3.1315929889678955,
      "step": 2457
    },
    {
      "dynamics_loss_avg": 0.06293513681739568,
      "epoch": 9.372711181640624e-06,
      "step": 2457
    },
    {
      "action_loss_avg": 0.03848661785013974,
      "epoch": 9.372711181640624e-06,
      "step": 2457
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 2457,
      "training_step_time": 9.900320768356323
    },
    {
      "epoch": 9.372711181640624e-06,
      "model_forward_time": 3.1353611946105957,
      "step": 2457
    },
    {
      "epoch": 9.372711181640624e-06,
      "step": 2457,
      "training_step_time": 10.273662090301514
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 3.1629557609558105,
      "step": 2458
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 2458,
      "training_step_time": 9.929208040237427
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 3.430497407913208,
      "step": 2458
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 2458,
      "training_step_time": 10.19864535331726
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 3.130265474319458,
      "step": 2458
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 2458,
      "training_step_time": 9.89918065071106
    },
    {
      "epoch": 9.37652587890625e-06,
      "model_forward_time": 3.1344666481018066,
      "step": 2458
    },
    {
      "epoch": 9.37652587890625e-06,
      "step": 2458,
      "training_step_time": 10.265233755111694
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 3.18564772605896,
      "step": 2459
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 2459,
      "training_step_time": 9.95498776435852
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 2.4302961826324463,
      "step": 2459
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 2459,
      "training_step_time": 7.457252025604248
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 3.1320526599884033,
      "step": 2459
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 2459,
      "training_step_time": 10.172096967697144
    },
    {
      "epoch": 9.380340576171875e-06,
      "model_forward_time": 3.1408777236938477,
      "step": 2459
    },
    {
      "epoch": 9.380340576171875e-06,
      "step": 2459,
      "training_step_time": 10.29911756515503
    },
    {
      "epoch": 9.3841552734375e-06,
      "grad_norm": 0.6108559966087341,
      "learning_rate": 1e-05,
      "loss": 0.1024,
      "step": 2460
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 2.4720401763916016,
      "step": 2460
    },
    {
      "dynamics_loss_avg": 0.07308696135878563,
      "epoch": 9.3841552734375e-06,
      "step": 2460
    },
    {
      "action_loss_avg": 0.03849685806781054,
      "epoch": 9.3841552734375e-06,
      "step": 2460
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 2460,
      "training_step_time": 7.504717826843262
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 3.1343610286712646,
      "step": 2460
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 2460,
      "training_step_time": 9.902124643325806
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 3.1344847679138184,
      "step": 2460
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 2460,
      "training_step_time": 9.902697086334229
    },
    {
      "epoch": 9.3841552734375e-06,
      "model_forward_time": 3.4190049171447754,
      "step": 2460
    },
    {
      "epoch": 9.3841552734375e-06,
      "step": 2460,
      "training_step_time": 10.289266109466553
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 3.1459786891937256,
      "step": 2461
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 2461,
      "training_step_time": 9.921501874923706
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 3.1317079067230225,
      "step": 2461
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 2461,
      "training_step_time": 9.906140089035034
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 3.136315107345581,
      "step": 2461
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 2461,
      "training_step_time": 9.908945322036743
    },
    {
      "epoch": 9.387969970703124e-06,
      "model_forward_time": 2.430433988571167,
      "step": 2461
    },
    {
      "epoch": 9.387969970703124e-06,
      "step": 2461,
      "training_step_time": 10.287867069244385
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 3.171638011932373,
      "step": 2462
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 2462,
      "training_step_time": 9.940074682235718
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 3.382004499435425,
      "step": 2462
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 2462,
      "training_step_time": 10.152084827423096
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 3.1385347843170166,
      "step": 2462
    },
    {
      "dynamics_loss_avg": 0.072713066264987,
      "epoch": 9.39178466796875e-06,
      "step": 2462
    },
    {
      "action_loss_avg": 0.04191872160881758,
      "epoch": 9.39178466796875e-06,
      "step": 2462
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 2462,
      "training_step_time": 9.914628028869629
    },
    {
      "epoch": 9.39178466796875e-06,
      "model_forward_time": 3.137509822845459,
      "step": 2462
    },
    {
      "epoch": 9.39178466796875e-06,
      "step": 2462,
      "training_step_time": 10.288822412490845
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 3.1496732234954834,
      "step": 2463
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 2463,
      "training_step_time": 9.924328804016113
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 3.3982977867126465,
      "step": 2463
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 2463,
      "training_step_time": 10.170835971832275
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 3.143256187438965,
      "step": 2463
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 2463,
      "training_step_time": 9.919877290725708
    },
    {
      "epoch": 9.395599365234375e-06,
      "model_forward_time": 1.7431728839874268,
      "step": 2463
    },
    {
      "epoch": 9.395599365234375e-06,
      "step": 2463,
      "training_step_time": 10.308308839797974
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 3.1906967163085938,
      "step": 2464
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 2464,
      "training_step_time": 9.962158203125
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 3.1382944583892822,
      "step": 2464
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 2464,
      "training_step_time": 9.912131071090698
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 3.1361539363861084,
      "step": 2464
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 2464,
      "training_step_time": 9.906607389450073
    },
    {
      "epoch": 9.3994140625e-06,
      "model_forward_time": 3.3848533630371094,
      "step": 2464
    },
    {
      "epoch": 9.3994140625e-06,
      "step": 2464,
      "training_step_time": 10.311334371566772
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 3.1808583736419678,
      "step": 2465
    },
    {
      "dynamics_loss_avg": 0.06896367948502302,
      "epoch": 9.403228759765624e-06,
      "step": 2465
    },
    {
      "action_loss_avg": 0.03803278915584087,
      "epoch": 9.403228759765624e-06,
      "step": 2465
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 2465,
      "training_step_time": 9.958982467651367
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 2.433534860610962,
      "step": 2465
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 2465,
      "training_step_time": 7.4610276222229
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 3.136380195617676,
      "step": 2465
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 2465,
      "training_step_time": 9.913354635238647
    },
    {
      "epoch": 9.403228759765624e-06,
      "model_forward_time": 3.1395435333251953,
      "step": 2465
    },
    {
      "epoch": 9.403228759765624e-06,
      "step": 2465,
      "training_step_time": 10.245898723602295
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 3.4465367794036865,
      "step": 2466
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 2466,
      "training_step_time": 10.225483894348145
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 3.142374038696289,
      "step": 2466
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 2466,
      "training_step_time": 9.919555187225342
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 3.1364524364471436,
      "step": 2466
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 2466,
      "training_step_time": 9.91304349899292
    },
    {
      "epoch": 9.40704345703125e-06,
      "model_forward_time": 3.139664649963379,
      "step": 2466
    },
    {
      "epoch": 9.40704345703125e-06,
      "step": 2466,
      "training_step_time": 10.306918382644653
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 3.1889185905456543,
      "step": 2467
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 2467,
      "training_step_time": 9.965389966964722
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 3.407848358154297,
      "step": 2467
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 2467,
      "training_step_time": 10.182818174362183
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 3.1437928676605225,
      "step": 2467
    },
    {
      "dynamics_loss_avg": 0.07478175424039364,
      "epoch": 9.410858154296875e-06,
      "step": 2467
    },
    {
      "action_loss_avg": 0.05560173261910677,
      "epoch": 9.410858154296875e-06,
      "step": 2467
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 2467,
      "training_step_time": 9.923234462738037
    },
    {
      "epoch": 9.410858154296875e-06,
      "model_forward_time": 3.140017509460449,
      "step": 2467
    },
    {
      "epoch": 9.410858154296875e-06,
      "step": 2467,
      "training_step_time": 10.033185243606567
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 3.178483486175537,
      "step": 2468
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 2468,
      "training_step_time": 9.95251202583313
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 3.1387436389923096,
      "step": 2468
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 2468,
      "training_step_time": 9.916624307632446
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 3.380397081375122,
      "step": 2468
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 2468,
      "training_step_time": 10.158645629882812
    },
    {
      "epoch": 9.4146728515625e-06,
      "model_forward_time": 3.1366212368011475,
      "step": 2468
    },
    {
      "epoch": 9.4146728515625e-06,
      "step": 2468,
      "training_step_time": 10.303821802139282
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 3.1725926399230957,
      "step": 2469
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 2469,
      "training_step_time": 9.951520442962646
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 3.1347875595092773,
      "step": 2469
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 2469,
      "training_step_time": 9.906359672546387
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 3.138366222381592,
      "step": 2469
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 2469,
      "training_step_time": 10.211309671401978
    },
    {
      "epoch": 9.418487548828124e-06,
      "model_forward_time": 3.1400458812713623,
      "step": 2469
    },
    {
      "epoch": 9.418487548828124e-06,
      "step": 2469,
      "training_step_time": 10.309343338012695
    },
    {
      "epoch": 9.42230224609375e-06,
      "grad_norm": 0.47619763016700745,
      "learning_rate": 1e-05,
      "loss": 0.1036,
      "step": 2470
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 3.171517848968506,
      "step": 2470
    },
    {
      "dynamics_loss_avg": 0.04753180406987667,
      "epoch": 9.42230224609375e-06,
      "step": 2470
    },
    {
      "action_loss_avg": 0.028644352639093994,
      "epoch": 9.42230224609375e-06,
      "step": 2470
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 2470,
      "training_step_time": 9.941745281219482
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 3.142289400100708,
      "step": 2470
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 2470,
      "training_step_time": 9.910154104232788
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 3.1276469230651855,
      "step": 2470
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 2470,
      "training_step_time": 9.892520427703857
    },
    {
      "epoch": 9.42230224609375e-06,
      "model_forward_time": 3.129162549972534,
      "step": 2470
    },
    {
      "epoch": 9.42230224609375e-06,
      "step": 2470,
      "training_step_time": 10.27713656425476
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 3.1616475582122803,
      "step": 2471
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 2471,
      "training_step_time": 9.934544324874878
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 3.133762836456299,
      "step": 2471
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 2471,
      "training_step_time": 9.906445503234863
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 3.1310153007507324,
      "step": 2471
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 2471,
      "training_step_time": 9.907648086547852
    },
    {
      "epoch": 9.426116943359375e-06,
      "model_forward_time": 3.1371796131134033,
      "step": 2471
    },
    {
      "epoch": 9.426116943359375e-06,
      "step": 2471,
      "training_step_time": 10.02473759651184
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 3.4476444721221924,
      "step": 2472
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 2472,
      "training_step_time": 10.219829320907593
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 3.1372623443603516,
      "step": 2472
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 2472,
      "training_step_time": 9.91015362739563
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 3.138742685317993,
      "step": 2472
    },
    {
      "dynamics_loss_avg": 0.07244644574820995,
      "epoch": 9.429931640625e-06,
      "step": 2472
    },
    {
      "action_loss_avg": 0.03890518974512815,
      "epoch": 9.429931640625e-06,
      "step": 2472
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 2472,
      "training_step_time": 9.915560960769653
    },
    {
      "epoch": 9.429931640625e-06,
      "model_forward_time": 3.1366500854492188,
      "step": 2472
    },
    {
      "epoch": 9.429931640625e-06,
      "step": 2472,
      "training_step_time": 10.304795503616333
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 3.146949291229248,
      "step": 2473
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 2473,
      "training_step_time": 9.923028230667114
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 3.394153356552124,
      "step": 2473
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 2473,
      "training_step_time": 10.165363311767578
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 2.4364585876464844,
      "step": 2473
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 2473,
      "training_step_time": 7.468308925628662
    },
    {
      "epoch": 9.433746337890624e-06,
      "model_forward_time": 2.425828695297241,
      "step": 2473
    },
    {
      "epoch": 9.433746337890624e-06,
      "step": 2473,
      "training_step_time": 10.03221869468689
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 2.474107503890991,
      "step": 2474
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 2474,
      "training_step_time": 7.501985549926758
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 3.1306960582733154,
      "step": 2474
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 2474,
      "training_step_time": 9.903400182723999
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 3.136336326599121,
      "step": 2474
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 2474,
      "training_step_time": 9.908401489257812
    },
    {
      "epoch": 9.43756103515625e-06,
      "model_forward_time": 3.3786823749542236,
      "step": 2474
    },
    {
      "epoch": 9.43756103515625e-06,
      "step": 2474,
      "training_step_time": 10.315728902816772
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 3.1943695545196533,
      "step": 2475
    },
    {
      "dynamics_loss_avg": 0.05654405448585749,
      "epoch": 9.441375732421875e-06,
      "step": 2475
    },
    {
      "action_loss_avg": 0.033033368177711964,
      "epoch": 9.441375732421875e-06,
      "step": 2475
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 2475,
      "training_step_time": 9.971138000488281
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 3.144230842590332,
      "step": 2475
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 2475,
      "training_step_time": 9.914273023605347
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 3.1397969722747803,
      "step": 2475
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 2475,
      "training_step_time": 9.9098219871521
    },
    {
      "epoch": 9.441375732421875e-06,
      "model_forward_time": 3.146202564239502,
      "step": 2475
    },
    {
      "epoch": 9.441375732421875e-06,
      "step": 2475,
      "training_step_time": 10.302252292633057
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 3.436903953552246,
      "step": 2476
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 2476,
      "training_step_time": 10.21307635307312
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 3.1361806392669678,
      "step": 2476
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 2476,
      "training_step_time": 9.906576156616211
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 3.136939525604248,
      "step": 2476
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 2476,
      "training_step_time": 9.911738395690918
    },
    {
      "epoch": 9.4451904296875e-06,
      "model_forward_time": 2.434093952178955,
      "step": 2476
    },
    {
      "epoch": 9.4451904296875e-06,
      "step": 2476,
      "training_step_time": 10.076936960220337
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 3.1778454780578613,
      "step": 2477
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 2477,
      "training_step_time": 9.95629596710205
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 3.402480125427246,
      "step": 2477
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 2477,
      "training_step_time": 10.183423519134521
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 3.136357545852661,
      "step": 2477
    },
    {
      "dynamics_loss_avg": 0.05863569024950266,
      "epoch": 9.449005126953124e-06,
      "step": 2477
    },
    {
      "action_loss_avg": 0.03915785923600197,
      "epoch": 9.449005126953124e-06,
      "step": 2477
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 2477,
      "training_step_time": 9.920458316802979
    },
    {
      "epoch": 9.449005126953124e-06,
      "model_forward_time": 3.1407742500305176,
      "step": 2477
    },
    {
      "epoch": 9.449005126953124e-06,
      "step": 2477,
      "training_step_time": 10.296907663345337
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 3.171032190322876,
      "step": 2478
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 2478,
      "training_step_time": 9.951797246932983
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 3.410432815551758,
      "step": 2478
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 2478,
      "training_step_time": 10.189682245254517
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 3.133516788482666,
      "step": 2478
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 2478,
      "training_step_time": 9.910127401351929
    },
    {
      "epoch": 9.45281982421875e-06,
      "model_forward_time": 2.4334628582000732,
      "step": 2478
    },
    {
      "epoch": 9.45281982421875e-06,
      "step": 2478,
      "training_step_time": 10.05791711807251
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 3.163475751876831,
      "step": 2479
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 2479,
      "training_step_time": 9.938928127288818
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 3.1300137042999268,
      "step": 2479
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 2479,
      "training_step_time": 9.902650356292725
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 3.1326680183410645,
      "step": 2479
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 2479,
      "training_step_time": 9.903271436691284
    },
    {
      "epoch": 9.456634521484375e-06,
      "model_forward_time": 3.3975844383239746,
      "step": 2479
    },
    {
      "epoch": 9.456634521484375e-06,
      "step": 2479,
      "training_step_time": 10.293494939804077
    },
    {
      "epoch": 9.46044921875e-06,
      "grad_norm": 0.524253249168396,
      "learning_rate": 1e-05,
      "loss": 0.1069,
      "step": 2480
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 3.1695597171783447,
      "step": 2480
    },
    {
      "dynamics_loss_avg": 0.05555531606078148,
      "epoch": 9.46044921875e-06,
      "step": 2480
    },
    {
      "action_loss_avg": 0.03330036122351885,
      "epoch": 9.46044921875e-06,
      "step": 2480
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 2480,
      "training_step_time": 9.946630001068115
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 3.128312826156616,
      "step": 2480
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 2480,
      "training_step_time": 9.897460222244263
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 3.1323273181915283,
      "step": 2480
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 2480,
      "training_step_time": 9.902959108352661
    },
    {
      "epoch": 9.46044921875e-06,
      "model_forward_time": 2.7051031589508057,
      "step": 2480
    },
    {
      "epoch": 9.46044921875e-06,
      "step": 2480,
      "training_step_time": 10.301300287246704
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 2.4632883071899414,
      "step": 2481
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 2481,
      "training_step_time": 7.4932215213775635
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 3.127540111541748,
      "step": 2481
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 2481,
      "training_step_time": 9.9012451171875
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 3.1303443908691406,
      "step": 2481
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 2481,
      "training_step_time": 9.90603518486023
    },
    {
      "epoch": 9.464263916015625e-06,
      "model_forward_time": 3.1310606002807617,
      "step": 2481
    },
    {
      "epoch": 9.464263916015625e-06,
      "step": 2481,
      "training_step_time": 10.238331079483032
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 3.1665775775909424,
      "step": 2482
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 2482,
      "training_step_time": 9.945527076721191
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 3.3839111328125,
      "step": 2482
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 2482,
      "training_step_time": 10.161815881729126
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 3.14003849029541,
      "step": 2482
    },
    {
      "dynamics_loss_avg": 0.05149209778755903,
      "epoch": 9.46807861328125e-06,
      "step": 2482
    },
    {
      "action_loss_avg": 0.03234142754226923,
      "epoch": 9.46807861328125e-06,
      "step": 2482
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 2482,
      "training_step_time": 9.921157598495483
    },
    {
      "epoch": 9.46807861328125e-06,
      "model_forward_time": 3.1344985961914062,
      "step": 2482
    },
    {
      "epoch": 9.46807861328125e-06,
      "step": 2482,
      "training_step_time": 10.018577098846436
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 3.1630935668945312,
      "step": 2483
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 2483,
      "training_step_time": 9.945616483688354
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 3.400299072265625,
      "step": 2483
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 2483,
      "training_step_time": 10.173943519592285
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 3.132336139678955,
      "step": 2483
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 2483,
      "training_step_time": 9.908649444580078
    },
    {
      "epoch": 9.471893310546875e-06,
      "model_forward_time": 3.1330699920654297,
      "step": 2483
    },
    {
      "epoch": 9.471893310546875e-06,
      "step": 2483,
      "training_step_time": 10.302848815917969
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 2.474871873855591,
      "step": 2484
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 2484,
      "training_step_time": 7.506672620773315
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 3.1352298259735107,
      "step": 2484
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 2484,
      "training_step_time": 9.911699056625366
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 3.1351375579833984,
      "step": 2484
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 2484,
      "training_step_time": 9.908324003219604
    },
    {
      "epoch": 9.4757080078125e-06,
      "model_forward_time": 3.396726608276367,
      "step": 2484
    },
    {
      "epoch": 9.4757080078125e-06,
      "step": 2484,
      "training_step_time": 10.272316932678223
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 3.184800148010254,
      "step": 2485
    },
    {
      "dynamics_loss_avg": 0.0807871276512742,
      "epoch": 9.479522705078125e-06,
      "step": 2485
    },
    {
      "action_loss_avg": 0.06436433419585227,
      "epoch": 9.479522705078125e-06,
      "step": 2485
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 2485,
      "training_step_time": 9.972140550613403
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 3.13981032371521,
      "step": 2485
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 2485,
      "training_step_time": 9.91871166229248
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 3.1379261016845703,
      "step": 2485
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 2485,
      "training_step_time": 9.917310953140259
    },
    {
      "epoch": 9.479522705078125e-06,
      "model_forward_time": 2.4377896785736084,
      "step": 2485
    },
    {
      "epoch": 9.479522705078125e-06,
      "step": 2485,
      "training_step_time": 10.314082384109497
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 3.180002212524414,
      "step": 2486
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 2486,
      "training_step_time": 9.962285041809082
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 2.43001651763916,
      "step": 2486
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 2486,
      "training_step_time": 7.462478160858154
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 3.1282331943511963,
      "step": 2486
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 2486,
      "training_step_time": 9.902442693710327
    },
    {
      "epoch": 9.48333740234375e-06,
      "model_forward_time": 3.137845754623413,
      "step": 2486
    },
    {
      "epoch": 9.48333740234375e-06,
      "step": 2486,
      "training_step_time": 10.280146360397339
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 3.4187567234039307,
      "step": 2487
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 2487,
      "training_step_time": 10.197705030441284
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 3.1375813484191895,
      "step": 2487
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 2487,
      "training_step_time": 9.916046619415283
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 3.133145809173584,
      "step": 2487
    },
    {
      "dynamics_loss_avg": 0.055275974422693254,
      "epoch": 9.487152099609375e-06,
      "step": 2487
    },
    {
      "action_loss_avg": 0.03825796283781528,
      "epoch": 9.487152099609375e-06,
      "step": 2487
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 2487,
      "training_step_time": 9.913086652755737
    },
    {
      "epoch": 9.487152099609375e-06,
      "model_forward_time": 3.13509464263916,
      "step": 2487
    },
    {
      "epoch": 9.487152099609375e-06,
      "step": 2487,
      "training_step_time": 10.027958393096924
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 3.1626832485198975,
      "step": 2488
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 2488,
      "training_step_time": 9.937276601791382
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 3.1322295665740967,
      "step": 2488
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 2488,
      "training_step_time": 10.18578314781189
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 3.1369199752807617,
      "step": 2488
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 2488,
      "training_step_time": 9.9150972366333
    },
    {
      "epoch": 9.490966796875e-06,
      "model_forward_time": 3.137510299682617,
      "step": 2488
    },
    {
      "epoch": 9.490966796875e-06,
      "step": 2488,
      "training_step_time": 10.337251901626587
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 3.1582016944885254,
      "step": 2489
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 2489,
      "training_step_time": 9.936634063720703
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 3.1343095302581787,
      "step": 2489
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 2489,
      "training_step_time": 9.91365361213684
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 2.4330220222473145,
      "step": 2489
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 2489,
      "training_step_time": 7.742876768112183
    },
    {
      "epoch": 9.494781494140625e-06,
      "model_forward_time": 3.138169050216675,
      "step": 2489
    },
    {
      "epoch": 9.494781494140625e-06,
      "step": 2489,
      "training_step_time": 10.032681941986084
    },
    {
      "epoch": 9.49859619140625e-06,
      "grad_norm": 0.8438184261322021,
      "learning_rate": 1e-05,
      "loss": 0.0989,
      "step": 2490
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 3.1698145866394043,
      "step": 2490
    },
    {
      "dynamics_loss_avg": 0.05530715007334948,
      "epoch": 9.49859619140625e-06,
      "step": 2490
    },
    {
      "action_loss_avg": 0.03773726839572191,
      "epoch": 9.49859619140625e-06,
      "step": 2490
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 2490,
      "training_step_time": 9.9671311378479
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 3.1372263431549072,
      "step": 2490
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 2490,
      "training_step_time": 9.921041488647461
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 3.138561964035034,
      "step": 2490
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 2490,
      "training_step_time": 9.92979884147644
    },
    {
      "epoch": 9.49859619140625e-06,
      "model_forward_time": 3.140979051589966,
      "step": 2490
    },
    {
      "epoch": 9.49859619140625e-06,
      "step": 2490,
      "training_step_time": 10.288285970687866
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 3.1529057025909424,
      "step": 2491
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 2491,
      "training_step_time": 9.93931269645691
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 3.135495662689209,
      "step": 2491
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 2491,
      "training_step_time": 9.925248146057129
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 2.4333038330078125,
      "step": 2491
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 2491,
      "training_step_time": 7.474726676940918
    },
    {
      "epoch": 9.502410888671875e-06,
      "model_forward_time": 3.1352243423461914,
      "step": 2491
    },
    {
      "epoch": 9.502410888671875e-06,
      "step": 2491,
      "training_step_time": 10.306791543960571
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 3.4200596809387207,
      "step": 2492
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 2492,
      "training_step_time": 10.20095682144165
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 3.137857437133789,
      "step": 2492
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 2492,
      "training_step_time": 9.916881561279297
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 2.42637300491333,
      "step": 2492
    },
    {
      "dynamics_loss_avg": 0.06529864724725484,
      "epoch": 9.5062255859375e-06,
      "step": 2492
    },
    {
      "action_loss_avg": 0.0396572433412075,
      "epoch": 9.5062255859375e-06,
      "step": 2492
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 2492,
      "training_step_time": 7.462461709976196
    },
    {
      "epoch": 9.5062255859375e-06,
      "model_forward_time": 3.129763603210449,
      "step": 2492
    },
    {
      "epoch": 9.5062255859375e-06,
      "step": 2492,
      "training_step_time": 10.270873308181763
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 2.4444544315338135,
      "step": 2493
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 2493,
      "training_step_time": 7.476914405822754
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 2.428929090499878,
      "step": 2493
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 2493,
      "training_step_time": 7.455073833465576
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 2.6947405338287354,
      "step": 2493
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 2493,
      "training_step_time": 7.719354629516602
    },
    {
      "epoch": 9.510040283203125e-06,
      "model_forward_time": 3.126152276992798,
      "step": 2493
    },
    {
      "epoch": 9.510040283203125e-06,
      "step": 2493,
      "training_step_time": 10.019190549850464
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 3.169482707977295,
      "step": 2494
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 2494,
      "training_step_time": 9.940482139587402
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 3.1317286491394043,
      "step": 2494
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 2494,
      "training_step_time": 9.907190084457397
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 3.1323354244232178,
      "step": 2494
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 2494,
      "training_step_time": 9.911142110824585
    },
    {
      "epoch": 9.51385498046875e-06,
      "model_forward_time": 3.3861210346221924,
      "step": 2494
    },
    {
      "epoch": 9.51385498046875e-06,
      "step": 2494,
      "training_step_time": 10.26844072341919
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 3.1740849018096924,
      "step": 2495
    },
    {
      "dynamics_loss_avg": 0.06495179068297148,
      "epoch": 9.517669677734375e-06,
      "step": 2495
    },
    {
      "action_loss_avg": 0.040669671911746266,
      "epoch": 9.517669677734375e-06,
      "step": 2495
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 2495,
      "training_step_time": 9.954640626907349
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 3.138353109359741,
      "step": 2495
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 2495,
      "training_step_time": 9.919633626937866
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 3.1401352882385254,
      "step": 2495
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 2495,
      "training_step_time": 9.923023223876953
    },
    {
      "epoch": 9.517669677734375e-06,
      "model_forward_time": 3.1429603099823,
      "step": 2495
    },
    {
      "epoch": 9.517669677734375e-06,
      "step": 2495,
      "training_step_time": 10.283970832824707
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 3.455864667892456,
      "step": 2496
    },
    {
      "epoch": 9.521484375e-06,
      "step": 2496,
      "training_step_time": 10.240210771560669
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 3.1366472244262695,
      "step": 2496
    },
    {
      "epoch": 9.521484375e-06,
      "step": 2496,
      "training_step_time": 9.920831441879272
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 3.141282320022583,
      "step": 2496
    },
    {
      "epoch": 9.521484375e-06,
      "step": 2496,
      "training_step_time": 9.924926280975342
    },
    {
      "epoch": 9.521484375e-06,
      "model_forward_time": 3.1420915126800537,
      "step": 2496
    },
    {
      "epoch": 9.521484375e-06,
      "step": 2496,
      "training_step_time": 10.302187442779541
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 3.1905295848846436,
      "step": 2497
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 2497,
      "training_step_time": 9.975863218307495
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 3.4137938022613525,
      "step": 2497
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 2497,
      "training_step_time": 10.193161249160767
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 3.134445905685425,
      "step": 2497
    },
    {
      "dynamics_loss_avg": 0.0686651029624045,
      "epoch": 9.525299072265625e-06,
      "step": 2497
    },
    {
      "action_loss_avg": 0.04807618292979896,
      "epoch": 9.525299072265625e-06,
      "step": 2497
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 2497,
      "training_step_time": 9.912977457046509
    },
    {
      "epoch": 9.525299072265625e-06,
      "model_forward_time": 3.141688108444214,
      "step": 2497
    },
    {
      "epoch": 9.525299072265625e-06,
      "step": 2497,
      "training_step_time": 10.297408819198608
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 2.4732139110565186,
      "step": 2498
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 2498,
      "training_step_time": 7.50912618637085
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 3.1316232681274414,
      "step": 2498
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 2498,
      "training_step_time": 9.907463312149048
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 3.3943941593170166,
      "step": 2498
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 2498,
      "training_step_time": 10.168462038040161
    },
    {
      "epoch": 9.52911376953125e-06,
      "model_forward_time": 3.1373324394226074,
      "step": 2498
    },
    {
      "epoch": 9.52911376953125e-06,
      "step": 2498,
      "training_step_time": 10.268439054489136
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 2.5126800537109375,
      "step": 2499
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 2499,
      "training_step_time": 7.55696702003479
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 3.13183856010437,
      "step": 2499
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 2499,
      "training_step_time": 9.902306318283081
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 3.132406234741211,
      "step": 2499
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 2499,
      "training_step_time": 9.904345989227295
    },
    {
      "epoch": 9.532928466796875e-06,
      "model_forward_time": 3.413296937942505,
      "step": 2499
    },
    {
      "epoch": 9.532928466796875e-06,
      "step": 2499,
      "training_step_time": 10.29764986038208
    },
    {
      "epoch": 9.5367431640625e-06,
      "grad_norm": 0.5778653621673584,
      "learning_rate": 1e-05,
      "loss": 0.1021,
      "step": 2500
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 3.153541088104248,
      "step": 2500
    },
    {
      "dynamics_loss_avg": 0.06430726386606693,
      "epoch": 9.5367431640625e-06,
      "step": 2500
    },
    {
      "action_loss_avg": 0.038495197892189026,
      "epoch": 9.5367431640625e-06,
      "step": 2500
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 2500,
      "training_step_time": 9.9070725440979
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 3.1149773597717285,
      "step": 2500
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 2500,
      "training_step_time": 9.867069959640503
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 3.122023105621338,
      "step": 2500
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 2500,
      "training_step_time": 9.878978967666626
    },
    {
      "epoch": 9.5367431640625e-06,
      "model_forward_time": 5.137107610702515,
      "step": 2500
    },
    {
      "epoch": 9.5367431640625e-06,
      "step": 2500,
      "training_step_time": 11.99990463256836
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 2.490675449371338,
      "step": 2501
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 2501,
      "training_step_time": 7.512776136398315
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 3.1285829544067383,
      "step": 2501
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 2501,
      "training_step_time": 9.892049551010132
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 3.1230509281158447,
      "step": 2501
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 2501,
      "training_step_time": 9.889734029769897
    },
    {
      "epoch": 9.540557861328125e-06,
      "model_forward_time": 3.1331515312194824,
      "step": 2501
    },
    {
      "epoch": 9.540557861328125e-06,
      "step": 2501,
      "training_step_time": 10.319430828094482
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 3.4219517707824707,
      "step": 2502
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 2502,
      "training_step_time": 10.192071676254272
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 3.1347129344940186,
      "step": 2502
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 2502,
      "training_step_time": 9.90273404121399
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 3.132349729537964,
      "step": 2502
    },
    {
      "dynamics_loss_avg": 0.07411953825503588,
      "epoch": 9.54437255859375e-06,
      "step": 2502
    },
    {
      "action_loss_avg": 0.04831063328310847,
      "epoch": 9.54437255859375e-06,
      "step": 2502
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 2502,
      "training_step_time": 9.913516998291016
    },
    {
      "epoch": 9.54437255859375e-06,
      "model_forward_time": 3.1351983547210693,
      "step": 2502
    },
    {
      "epoch": 9.54437255859375e-06,
      "step": 2502,
      "training_step_time": 10.301374435424805
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 3.170376777648926,
      "step": 2503
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 2503,
      "training_step_time": 9.943189144134521
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 2.678164482116699,
      "step": 2503
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 2503,
      "training_step_time": 7.706006050109863
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 3.1358532905578613,
      "step": 2503
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 2503,
      "training_step_time": 9.906185626983643
    },
    {
      "epoch": 9.548187255859375e-06,
      "model_forward_time": 3.1401636600494385,
      "step": 2503
    },
    {
      "epoch": 9.548187255859375e-06,
      "step": 2503,
      "training_step_time": 10.277160167694092
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 3.187359571456909,
      "step": 2504
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 2504,
      "training_step_time": 9.959607124328613
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 3.1396148204803467,
      "step": 2504
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 2504,
      "training_step_time": 9.912126779556274
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 3.388815402984619,
      "step": 2504
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 2504,
      "training_step_time": 10.157525539398193
    },
    {
      "epoch": 9.552001953125e-06,
      "model_forward_time": 3.1408028602600098,
      "step": 2504
    },
    {
      "epoch": 9.552001953125e-06,
      "step": 2504,
      "training_step_time": 10.286122560501099
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 3.165046453475952,
      "step": 2505
    },
    {
      "dynamics_loss_avg": 0.08674026466906071,
      "epoch": 9.555816650390625e-06,
      "step": 2505
    },
    {
      "action_loss_avg": 0.058810293674468994,
      "epoch": 9.555816650390625e-06,
      "step": 2505
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 2505,
      "training_step_time": 9.951403379440308
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 3.1309072971343994,
      "step": 2505
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 2505,
      "training_step_time": 9.901471376419067
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 3.137012481689453,
      "step": 2505
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 2505,
      "training_step_time": 9.912208318710327
    },
    {
      "epoch": 9.555816650390625e-06,
      "model_forward_time": 3.404639720916748,
      "step": 2505
    },
    {
      "epoch": 9.555816650390625e-06,
      "step": 2505,
      "training_step_time": 10.273966312408447
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 3.1768670082092285,
      "step": 2506
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 2506,
      "training_step_time": 9.959263563156128
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 3.134871482849121,
      "step": 2506
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 2506,
      "training_step_time": 9.912898302078247
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 3.141580820083618,
      "step": 2506
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 2506,
      "training_step_time": 9.91861629486084
    },
    {
      "epoch": 9.55963134765625e-06,
      "model_forward_time": 3.4108262062072754,
      "step": 2506
    },
    {
      "epoch": 9.55963134765625e-06,
      "step": 2506,
      "training_step_time": 10.317818403244019
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 2.459085702896118,
      "step": 2507
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 2507,
      "training_step_time": 7.489267826080322
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 3.130661964416504,
      "step": 2507
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 2507,
      "training_step_time": 9.911375284194946
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 3.1427669525146484,
      "step": 2507
    },
    {
      "dynamics_loss_avg": 0.06739146821200848,
      "epoch": 9.563446044921875e-06,
      "step": 2507
    },
    {
      "action_loss_avg": 0.04572556409984827,
      "epoch": 9.563446044921875e-06,
      "step": 2507
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 2507,
      "training_step_time": 9.923670053482056
    },
    {
      "epoch": 9.563446044921875e-06,
      "model_forward_time": 3.1408228874206543,
      "step": 2507
    },
    {
      "epoch": 9.563446044921875e-06,
      "step": 2507,
      "training_step_time": 10.288433313369751
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 3.1845784187316895,
      "step": 2508
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 2508,
      "training_step_time": 9.970532417297363
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 3.4152138233184814,
      "step": 2508
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 2508,
      "training_step_time": 10.183822631835938
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 3.1420397758483887,
      "step": 2508
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 2508,
      "training_step_time": 9.90903902053833
    },
    {
      "epoch": 9.5672607421875e-06,
      "model_forward_time": 3.1416478157043457,
      "step": 2508
    },
    {
      "epoch": 9.5672607421875e-06,
      "step": 2508,
      "training_step_time": 10.293256044387817
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 3.182936191558838,
      "step": 2509
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 2509,
      "training_step_time": 9.955328464508057
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 3.397845983505249,
      "step": 2509
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 2509,
      "training_step_time": 10.163553953170776
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 3.14145565032959,
      "step": 2509
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 2509,
      "training_step_time": 9.909761667251587
    },
    {
      "epoch": 9.571075439453125e-06,
      "model_forward_time": 3.1385040283203125,
      "step": 2509
    },
    {
      "epoch": 9.571075439453125e-06,
      "step": 2509,
      "training_step_time": 10.036169528961182
    },
    {
      "epoch": 9.57489013671875e-06,
      "grad_norm": 0.6447353959083557,
      "learning_rate": 1e-05,
      "loss": 0.1083,
      "step": 2510
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 2.4821689128875732,
      "step": 2510
    },
    {
      "dynamics_loss_avg": 0.0682497814297676,
      "epoch": 9.57489013671875e-06,
      "step": 2510
    },
    {
      "action_loss_avg": 0.051245092786848544,
      "epoch": 9.57489013671875e-06,
      "step": 2510
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 2510,
      "training_step_time": 7.512043714523315
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 2.434962034225464,
      "step": 2510
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 2510,
      "training_step_time": 7.465352535247803
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 3.135528087615967,
      "step": 2510
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 2510,
      "training_step_time": 9.901676893234253
    },
    {
      "epoch": 9.57489013671875e-06,
      "model_forward_time": 3.396601438522339,
      "step": 2510
    },
    {
      "epoch": 9.57489013671875e-06,
      "step": 2510,
      "training_step_time": 10.334304571151733
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 3.1780054569244385,
      "step": 2511
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 2511,
      "training_step_time": 9.947833061218262
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 3.1447484493255615,
      "step": 2511
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 2511,
      "training_step_time": 9.913080215454102
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 3.1400506496429443,
      "step": 2511
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 2511,
      "training_step_time": 9.90736198425293
    },
    {
      "epoch": 9.578704833984375e-06,
      "model_forward_time": 3.1428263187408447,
      "step": 2511
    },
    {
      "epoch": 9.578704833984375e-06,
      "step": 2511,
      "training_step_time": 10.31875205039978
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 3.4651851654052734,
      "step": 2512
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 2512,
      "training_step_time": 10.230722904205322
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 3.134183168411255,
      "step": 2512
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 2512,
      "training_step_time": 9.899986505508423
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 3.142720937728882,
      "step": 2512
    },
    {
      "dynamics_loss_avg": 0.07132834903895854,
      "epoch": 9.58251953125e-06,
      "step": 2512
    },
    {
      "action_loss_avg": 0.03397235032171011,
      "epoch": 9.58251953125e-06,
      "step": 2512
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 2512,
      "training_step_time": 9.909216403961182
    },
    {
      "epoch": 9.58251953125e-06,
      "model_forward_time": 2.4345881938934326,
      "step": 2512
    },
    {
      "epoch": 9.58251953125e-06,
      "step": 2512,
      "training_step_time": 10.262110471725464
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 3.1669702529907227,
      "step": 2513
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 2513,
      "training_step_time": 10.19072413444519
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 3.1404976844787598,
      "step": 2513
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 2513,
      "training_step_time": 9.907034873962402
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 3.136597156524658,
      "step": 2513
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 2513,
      "training_step_time": 9.901268243789673
    },
    {
      "epoch": 9.586334228515625e-06,
      "model_forward_time": 3.1417899131774902,
      "step": 2513
    },
    {
      "epoch": 9.586334228515625e-06,
      "step": 2513,
      "training_step_time": 10.010414361953735
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 3.1635212898254395,
      "step": 2514
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 2514,
      "training_step_time": 9.931331157684326
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 3.1316306591033936,
      "step": 2514
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 2514,
      "training_step_time": 10.167811870574951
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 3.1287317276000977,
      "step": 2514
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 2514,
      "training_step_time": 9.897673845291138
    },
    {
      "epoch": 9.59014892578125e-06,
      "model_forward_time": 3.1332664489746094,
      "step": 2514
    },
    {
      "epoch": 9.59014892578125e-06,
      "step": 2514,
      "training_step_time": 10.013139963150024
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 3.194196939468384,
      "step": 2515
    },
    {
      "dynamics_loss_avg": 0.06385015342384577,
      "epoch": 9.593963623046875e-06,
      "step": 2515
    },
    {
      "action_loss_avg": 0.03876811843365431,
      "epoch": 9.593963623046875e-06,
      "step": 2515
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 2515,
      "training_step_time": 9.964252710342407
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 3.1405158042907715,
      "step": 2515
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 2515,
      "training_step_time": 9.9130220413208
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 3.4012272357940674,
      "step": 2515
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 2515,
      "training_step_time": 10.172104597091675
    },
    {
      "epoch": 9.593963623046875e-06,
      "model_forward_time": 3.149092435836792,
      "step": 2515
    },
    {
      "epoch": 9.593963623046875e-06,
      "step": 2515,
      "training_step_time": 10.297226428985596
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 3.1837267875671387,
      "step": 2516
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 2516,
      "training_step_time": 9.95675015449524
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 3.14087176322937,
      "step": 2516
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 2516,
      "training_step_time": 9.91288685798645
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 3.150421380996704,
      "step": 2516
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 2516,
      "training_step_time": 9.92517352104187
    },
    {
      "epoch": 9.5977783203125e-06,
      "model_forward_time": 3.1531219482421875,
      "step": 2516
    },
    {
      "epoch": 9.5977783203125e-06,
      "step": 2516,
      "training_step_time": 10.27880048751831
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 3.191903829574585,
      "step": 2517
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 2517,
      "training_step_time": 9.964219808578491
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 3.139328956604004,
      "step": 2517
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 2517,
      "training_step_time": 9.90869426727295
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 3.1487274169921875,
      "step": 2517
    },
    {
      "dynamics_loss_avg": 0.08008178025484085,
      "epoch": 9.601593017578125e-06,
      "step": 2517
    },
    {
      "action_loss_avg": 0.0423359178006649,
      "epoch": 9.601593017578125e-06,
      "step": 2517
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 2517,
      "training_step_time": 9.925593852996826
    },
    {
      "epoch": 9.601593017578125e-06,
      "model_forward_time": 3.1428678035736084,
      "step": 2517
    },
    {
      "epoch": 9.601593017578125e-06,
      "step": 2517,
      "training_step_time": 10.306576013565063
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 3.44954252243042,
      "step": 2518
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 2518,
      "training_step_time": 10.225053787231445
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 3.143911838531494,
      "step": 2518
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 2518,
      "training_step_time": 9.914390563964844
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 3.1396644115448,
      "step": 2518
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 2518,
      "training_step_time": 9.905177116394043
    },
    {
      "epoch": 9.60540771484375e-06,
      "model_forward_time": 3.1385254859924316,
      "step": 2518
    },
    {
      "epoch": 9.60540771484375e-06,
      "step": 2518,
      "training_step_time": 10.297561645507812
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 2.483652353286743,
      "step": 2519
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 2519,
      "training_step_time": 7.509631156921387
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 2.432123899459839,
      "step": 2519
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 2519,
      "training_step_time": 7.722185373306274
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 3.141087532043457,
      "step": 2519
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 2519,
      "training_step_time": 9.923220157623291
    },
    {
      "epoch": 9.609222412109375e-06,
      "model_forward_time": 3.1363658905029297,
      "step": 2519
    },
    {
      "epoch": 9.609222412109375e-06,
      "step": 2519,
      "training_step_time": 10.015989780426025
    },
    {
      "epoch": 9.613037109375e-06,
      "grad_norm": 0.4984488785266876,
      "learning_rate": 1e-05,
      "loss": 0.1077,
      "step": 2520
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 3.157219171524048,
      "step": 2520
    },
    {
      "dynamics_loss_avg": 0.07515863291919231,
      "epoch": 9.613037109375e-06,
      "step": 2520
    },
    {
      "action_loss_avg": 0.05302207618951797,
      "epoch": 9.613037109375e-06,
      "step": 2520
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 2520,
      "training_step_time": 9.93394136428833
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 2.428905487060547,
      "step": 2520
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 2520,
      "training_step_time": 7.453277349472046
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 3.126466989517212,
      "step": 2520
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 2520,
      "training_step_time": 10.179191589355469
    },
    {
      "epoch": 9.613037109375e-06,
      "model_forward_time": 3.1312811374664307,
      "step": 2520
    },
    {
      "epoch": 9.613037109375e-06,
      "step": 2520,
      "training_step_time": 10.316895008087158
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 3.140277862548828,
      "step": 2521
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 2521,
      "training_step_time": 9.915672063827515
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 3.132192850112915,
      "step": 2521
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 2521,
      "training_step_time": 9.905595541000366
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 2.4350903034210205,
      "step": 2521
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 2521,
      "training_step_time": 7.467344284057617
    },
    {
      "epoch": 9.616851806640625e-06,
      "model_forward_time": 3.1357767581939697,
      "step": 2521
    },
    {
      "epoch": 9.616851806640625e-06,
      "step": 2521,
      "training_step_time": 10.023926973342896
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 2.7443649768829346,
      "step": 2522
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 2522,
      "training_step_time": 7.7777228355407715
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 3.1364550590515137,
      "step": 2522
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 2522,
      "training_step_time": 9.913278341293335
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 3.1317713260650635,
      "step": 2522
    },
    {
      "dynamics_loss_avg": 0.08962546549737453,
      "epoch": 9.62066650390625e-06,
      "step": 2522
    },
    {
      "action_loss_avg": 0.04979998730123043,
      "epoch": 9.62066650390625e-06,
      "step": 2522
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 2522,
      "training_step_time": 9.911366701126099
    },
    {
      "epoch": 9.62066650390625e-06,
      "model_forward_time": 3.1347646713256836,
      "step": 2522
    },
    {
      "epoch": 9.62066650390625e-06,
      "step": 2522,
      "training_step_time": 10.280249834060669
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 3.162792444229126,
      "step": 2523
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 2523,
      "training_step_time": 9.93997573852539
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 2.6712236404418945,
      "step": 2523
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 2523,
      "training_step_time": 7.704809904098511
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 3.1309144496917725,
      "step": 2523
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 2523,
      "training_step_time": 9.905190467834473
    },
    {
      "epoch": 9.624481201171876e-06,
      "model_forward_time": 3.1372876167297363,
      "step": 2523
    },
    {
      "epoch": 9.624481201171876e-06,
      "step": 2523,
      "training_step_time": 10.304102897644043
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 3.1723387241363525,
      "step": 2524
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 2524,
      "training_step_time": 9.948754787445068
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 3.142923355102539,
      "step": 2524
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 2524,
      "training_step_time": 10.194340944290161
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 3.1406056880950928,
      "step": 2524
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 2524,
      "training_step_time": 9.921785116195679
    },
    {
      "epoch": 9.6282958984375e-06,
      "model_forward_time": 3.148383855819702,
      "step": 2524
    },
    {
      "epoch": 9.6282958984375e-06,
      "step": 2524,
      "training_step_time": 10.270280599594116
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 3.181124448776245,
      "step": 2525
    },
    {
      "dynamics_loss_avg": 0.06751441173255443,
      "epoch": 9.632110595703125e-06,
      "step": 2525
    },
    {
      "action_loss_avg": 0.040873204730451106,
      "epoch": 9.632110595703125e-06,
      "step": 2525
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 2525,
      "training_step_time": 9.969969511032104
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 3.146430015563965,
      "step": 2525
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 2525,
      "training_step_time": 9.927231550216675
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 3.148453712463379,
      "step": 2525
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 2525,
      "training_step_time": 10.233442783355713
    },
    {
      "epoch": 9.632110595703125e-06,
      "model_forward_time": 3.1438374519348145,
      "step": 2525
    },
    {
      "epoch": 9.632110595703125e-06,
      "step": 2525,
      "training_step_time": 10.262159585952759
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 2.4619085788726807,
      "step": 2526
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 2526,
      "training_step_time": 7.497024297714233
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 3.1316213607788086,
      "step": 2526
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 2526,
      "training_step_time": 9.90715765953064
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 2.4328668117523193,
      "step": 2526
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 2526,
      "training_step_time": 7.4700212478637695
    },
    {
      "epoch": 9.63592529296875e-06,
      "model_forward_time": 3.1361703872680664,
      "step": 2526
    },
    {
      "epoch": 9.63592529296875e-06,
      "step": 2526,
      "training_step_time": 10.289782285690308
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 3.466999053955078,
      "step": 2527
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 2527,
      "training_step_time": 10.258907794952393
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 3.137556552886963,
      "step": 2527
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 2527,
      "training_step_time": 9.919782876968384
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 3.1367526054382324,
      "step": 2527
    },
    {
      "dynamics_loss_avg": 0.07304722480475903,
      "epoch": 9.639739990234376e-06,
      "step": 2527
    },
    {
      "action_loss_avg": 0.03560163117945194,
      "epoch": 9.639739990234376e-06,
      "step": 2527
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 2527,
      "training_step_time": 9.916208982467651
    },
    {
      "epoch": 9.639739990234376e-06,
      "model_forward_time": 3.1306345462799072,
      "step": 2527
    },
    {
      "epoch": 9.639739990234376e-06,
      "step": 2527,
      "training_step_time": 10.268535614013672
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 3.1683337688446045,
      "step": 2528
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 2528,
      "training_step_time": 10.227831602096558
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 3.138701915740967,
      "step": 2528
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 2528,
      "training_step_time": 9.911882877349854
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 3.138615131378174,
      "step": 2528
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 2528,
      "training_step_time": 9.917875051498413
    },
    {
      "epoch": 9.6435546875e-06,
      "model_forward_time": 3.141066551208496,
      "step": 2528
    },
    {
      "epoch": 9.6435546875e-06,
      "step": 2528,
      "training_step_time": 10.295040607452393
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 2.446345567703247,
      "step": 2529
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 2529,
      "training_step_time": 7.480249404907227
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 3.3835980892181396,
      "step": 2529
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 2529,
      "training_step_time": 10.158121347427368
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 2.42876935005188,
      "step": 2529
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 2529,
      "training_step_time": 7.460729122161865
    },
    {
      "epoch": 9.647369384765625e-06,
      "model_forward_time": 3.1271286010742188,
      "step": 2529
    },
    {
      "epoch": 9.647369384765625e-06,
      "step": 2529,
      "training_step_time": 10.279084205627441
    },
    {
      "epoch": 9.65118408203125e-06,
      "grad_norm": 0.568315863609314,
      "learning_rate": 1e-05,
      "loss": 0.1031,
      "step": 2530
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 3.135451555252075,
      "step": 2530
    },
    {
      "dynamics_loss_avg": 0.06445727162063122,
      "epoch": 9.65118408203125e-06,
      "step": 2530
    },
    {
      "action_loss_avg": 0.04521518005058169,
      "epoch": 9.65118408203125e-06,
      "step": 2530
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 2530,
      "training_step_time": 9.916321277618408
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 2.4302659034729004,
      "step": 2530
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 2530,
      "training_step_time": 7.463277101516724
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 2.42564058303833,
      "step": 2530
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 2530,
      "training_step_time": 7.457879066467285
    },
    {
      "epoch": 9.65118408203125e-06,
      "model_forward_time": 3.3817129135131836,
      "step": 2530
    },
    {
      "epoch": 9.65118408203125e-06,
      "step": 2530,
      "training_step_time": 10.337094068527222
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 2.466437578201294,
      "step": 2531
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 2531,
      "training_step_time": 7.5107262134552
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 3.1368370056152344,
      "step": 2531
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 2531,
      "training_step_time": 9.910544872283936
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 3.13254976272583,
      "step": 2531
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 2531,
      "training_step_time": 9.905009984970093
    },
    {
      "epoch": 9.654998779296876e-06,
      "model_forward_time": 3.1324784755706787,
      "step": 2531
    },
    {
      "epoch": 9.654998779296876e-06,
      "step": 2531,
      "training_step_time": 10.285297870635986
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 3.4196619987487793,
      "step": 2532
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 2532,
      "training_step_time": 10.191344261169434
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 3.1351845264434814,
      "step": 2532
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 2532,
      "training_step_time": 9.906372785568237
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 3.1333069801330566,
      "step": 2532
    },
    {
      "dynamics_loss_avg": 0.06685806773602962,
      "epoch": 9.6588134765625e-06,
      "step": 2532
    },
    {
      "action_loss_avg": 0.03387813130393624,
      "epoch": 9.6588134765625e-06,
      "step": 2532
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 2532,
      "training_step_time": 9.90702509880066
    },
    {
      "epoch": 9.6588134765625e-06,
      "model_forward_time": 3.136383295059204,
      "step": 2532
    },
    {
      "epoch": 9.6588134765625e-06,
      "step": 2532,
      "training_step_time": 10.276892900466919
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 3.1701810359954834,
      "step": 2533
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 2533,
      "training_step_time": 9.944125413894653
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 3.403226613998413,
      "step": 2533
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 2533,
      "training_step_time": 10.173001766204834
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 2.4321413040161133,
      "step": 2533
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 2533,
      "training_step_time": 7.464848279953003
    },
    {
      "epoch": 9.662628173828125e-06,
      "model_forward_time": 3.132157802581787,
      "step": 2533
    },
    {
      "epoch": 9.662628173828125e-06,
      "step": 2533,
      "training_step_time": 10.006168842315674
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 3.179516315460205,
      "step": 2534
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 2534,
      "training_step_time": 9.951772451400757
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 3.1336770057678223,
      "step": 2534
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 2534,
      "training_step_time": 9.905360698699951
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 2.6967601776123047,
      "step": 2534
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 2534,
      "training_step_time": 7.725088834762573
    },
    {
      "epoch": 9.66644287109375e-06,
      "model_forward_time": 3.1374332904815674,
      "step": 2534
    },
    {
      "epoch": 9.66644287109375e-06,
      "step": 2534,
      "training_step_time": 10.322059154510498
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 3.1936652660369873,
      "step": 2535
    },
    {
      "dynamics_loss_avg": 0.0559880081564188,
      "epoch": 9.670257568359376e-06,
      "step": 2535
    },
    {
      "action_loss_avg": 0.030304369889199732,
      "epoch": 9.670257568359376e-06,
      "step": 2535
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 2535,
      "training_step_time": 9.970287322998047
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 3.1393182277679443,
      "step": 2535
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 2535,
      "training_step_time": 9.910306930541992
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 3.1440255641937256,
      "step": 2535
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 2535,
      "training_step_time": 9.919541597366333
    },
    {
      "epoch": 9.670257568359376e-06,
      "model_forward_time": 3.4108545780181885,
      "step": 2535
    },
    {
      "epoch": 9.670257568359376e-06,
      "step": 2535,
      "training_step_time": 10.305464267730713
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 3.188906669616699,
      "step": 2536
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 2536,
      "training_step_time": 9.966144323348999
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 3.135615348815918,
      "step": 2536
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 2536,
      "training_step_time": 9.91076135635376
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 3.1372923851013184,
      "step": 2536
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 2536,
      "training_step_time": 9.911945104598999
    },
    {
      "epoch": 9.674072265625e-06,
      "model_forward_time": 3.1351356506347656,
      "step": 2536
    },
    {
      "epoch": 9.674072265625e-06,
      "step": 2536,
      "training_step_time": 10.043501377105713
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 3.4577512741088867,
      "step": 2537
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 2537,
      "training_step_time": 10.230292797088623
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 3.138411045074463,
      "step": 2537
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 2537,
      "training_step_time": 9.90934443473816
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 3.1399543285369873,
      "step": 2537
    },
    {
      "dynamics_loss_avg": 0.0709973780438304,
      "epoch": 9.677886962890625e-06,
      "step": 2537
    },
    {
      "action_loss_avg": 0.0305949080735445,
      "epoch": 9.677886962890625e-06,
      "step": 2537
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 2537,
      "training_step_time": 9.913736343383789
    },
    {
      "epoch": 9.677886962890625e-06,
      "model_forward_time": 3.1397616863250732,
      "step": 2537
    },
    {
      "epoch": 9.677886962890625e-06,
      "step": 2537,
      "training_step_time": 10.27658462524414
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 3.172685384750366,
      "step": 2538
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 2538,
      "training_step_time": 9.941692590713501
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 3.4049746990203857,
      "step": 2538
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 2538,
      "training_step_time": 10.17230224609375
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 3.1350812911987305,
      "step": 2538
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 2538,
      "training_step_time": 9.898517847061157
    },
    {
      "epoch": 9.68170166015625e-06,
      "model_forward_time": 3.136674165725708,
      "step": 2538
    },
    {
      "epoch": 9.68170166015625e-06,
      "step": 2538,
      "training_step_time": 10.01383090019226
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 3.1828877925872803,
      "step": 2539
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 2539,
      "training_step_time": 9.961631059646606
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 3.1347367763519287,
      "step": 2539
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 2539,
      "training_step_time": 9.896725177764893
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 2.7100915908813477,
      "step": 2539
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 2539,
      "training_step_time": 7.7301740646362305
    },
    {
      "epoch": 9.685516357421876e-06,
      "model_forward_time": 3.1312310695648193,
      "step": 2539
    },
    {
      "epoch": 9.685516357421876e-06,
      "step": 2539,
      "training_step_time": 10.300152063369751
    },
    {
      "epoch": 9.6893310546875e-06,
      "grad_norm": 0.4310937821865082,
      "learning_rate": 1e-05,
      "loss": 0.1027,
      "step": 2540
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 3.1791975498199463,
      "step": 2540
    },
    {
      "dynamics_loss_avg": 0.07178889345377684,
      "epoch": 9.6893310546875e-06,
      "step": 2540
    },
    {
      "action_loss_avg": 0.038563806237652896,
      "epoch": 9.6893310546875e-06,
      "step": 2540
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 2540,
      "training_step_time": 9.94678783416748
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 3.132570505142212,
      "step": 2540
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 2540,
      "training_step_time": 9.893224954605103
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 3.1410694122314453,
      "step": 2540
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 2540,
      "training_step_time": 10.171673774719238
    },
    {
      "epoch": 9.6893310546875e-06,
      "model_forward_time": 3.130584478378296,
      "step": 2540
    },
    {
      "epoch": 9.6893310546875e-06,
      "step": 2540,
      "training_step_time": 10.025279521942139
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 3.160959005355835,
      "step": 2541
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 2541,
      "training_step_time": 9.932848691940308
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 3.1333515644073486,
      "step": 2541
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 2541,
      "training_step_time": 9.902899026870728
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 3.1362226009368896,
      "step": 2541
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 2541,
      "training_step_time": 9.905387878417969
    },
    {
      "epoch": 9.693145751953125e-06,
      "model_forward_time": 3.3926405906677246,
      "step": 2541
    },
    {
      "epoch": 9.693145751953125e-06,
      "step": 2541,
      "training_step_time": 10.307056903839111
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 3.1739792823791504,
      "step": 2542
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 2542,
      "training_step_time": 9.942600965499878
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 2.43333101272583,
      "step": 2542
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 2542,
      "training_step_time": 7.4605443477630615
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 3.130126476287842,
      "step": 2542
    },
    {
      "dynamics_loss_avg": 0.05802732314914465,
      "epoch": 9.69696044921875e-06,
      "step": 2542
    },
    {
      "action_loss_avg": 0.032172172935679554,
      "epoch": 9.69696044921875e-06,
      "step": 2542
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 2542,
      "training_step_time": 9.90195083618164
    },
    {
      "epoch": 9.69696044921875e-06,
      "model_forward_time": 3.1356611251831055,
      "step": 2542
    },
    {
      "epoch": 9.69696044921875e-06,
      "step": 2542,
      "training_step_time": 10.289294719696045
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 3.184748411178589,
      "step": 2543
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 2543,
      "training_step_time": 10.22290849685669
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 3.140648365020752,
      "step": 2543
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 2543,
      "training_step_time": 9.912078142166138
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 3.1330435276031494,
      "step": 2543
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 2543,
      "training_step_time": 9.895655870437622
    },
    {
      "epoch": 9.700775146484376e-06,
      "model_forward_time": 3.1326019763946533,
      "step": 2543
    },
    {
      "epoch": 9.700775146484376e-06,
      "step": 2543,
      "training_step_time": 10.285744905471802
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 3.1783201694488525,
      "step": 2544
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 2544,
      "training_step_time": 9.945776224136353
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 3.411421537399292,
      "step": 2544
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 2544,
      "training_step_time": 10.176970720291138
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 2.434748649597168,
      "step": 2544
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 2544,
      "training_step_time": 7.457236051559448
    },
    {
      "epoch": 9.70458984375e-06,
      "model_forward_time": 3.1353931427001953,
      "step": 2544
    },
    {
      "epoch": 9.70458984375e-06,
      "step": 2544,
      "training_step_time": 10.010162830352783
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 3.183607816696167,
      "step": 2545
    },
    {
      "dynamics_loss_avg": 0.06803955398499965,
      "epoch": 9.708404541015625e-06,
      "step": 2545
    },
    {
      "action_loss_avg": 0.04526267424225807,
      "epoch": 9.708404541015625e-06,
      "step": 2545
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 2545,
      "training_step_time": 9.964980840682983
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 3.12652850151062,
      "step": 2545
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 2545,
      "training_step_time": 9.892293214797974
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 3.1313276290893555,
      "step": 2545
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 2545,
      "training_step_time": 10.167737245559692
    },
    {
      "epoch": 9.708404541015625e-06,
      "model_forward_time": 3.132765769958496,
      "step": 2545
    },
    {
      "epoch": 9.708404541015625e-06,
      "step": 2545,
      "training_step_time": 10.239759683609009
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 3.1602981090545654,
      "step": 2546
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 2546,
      "training_step_time": 9.931662559509277
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 3.1314849853515625,
      "step": 2546
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 2546,
      "training_step_time": 9.896937847137451
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 3.1335060596466064,
      "step": 2546
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 2546,
      "training_step_time": 9.899458885192871
    },
    {
      "epoch": 9.71221923828125e-06,
      "model_forward_time": 3.4059512615203857,
      "step": 2546
    },
    {
      "epoch": 9.71221923828125e-06,
      "step": 2546,
      "training_step_time": 10.280608892440796
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 3.17330265045166,
      "step": 2547
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 2547,
      "training_step_time": 9.940206289291382
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 3.129009485244751,
      "step": 2547
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 2547,
      "training_step_time": 9.895762205123901
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 3.1330788135528564,
      "step": 2547
    },
    {
      "dynamics_loss_avg": 0.06068914057686925,
      "epoch": 9.716033935546876e-06,
      "step": 2547
    },
    {
      "action_loss_avg": 0.03804466346045956,
      "epoch": 9.716033935546876e-06,
      "step": 2547
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 2547,
      "training_step_time": 9.898988008499146
    },
    {
      "epoch": 9.716033935546876e-06,
      "model_forward_time": 3.130049228668213,
      "step": 2547
    },
    {
      "epoch": 9.716033935546876e-06,
      "step": 2547,
      "training_step_time": 10.283314228057861
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 3.4415535926818848,
      "step": 2548
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 2548,
      "training_step_time": 10.20595383644104
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 3.1311562061309814,
      "step": 2548
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 2548,
      "training_step_time": 9.896923780441284
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 3.1342244148254395,
      "step": 2548
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 2548,
      "training_step_time": 9.90456247329712
    },
    {
      "epoch": 9.7198486328125e-06,
      "model_forward_time": 3.133650302886963,
      "step": 2548
    },
    {
      "epoch": 9.7198486328125e-06,
      "step": 2548,
      "training_step_time": 10.295037984848022
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 3.171337127685547,
      "step": 2549
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 2549,
      "training_step_time": 9.943803310394287
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 3.40579891204834,
      "step": 2549
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 2549,
      "training_step_time": 10.172998428344727
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 3.142737865447998,
      "step": 2549
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 2549,
      "training_step_time": 9.911341905593872
    },
    {
      "epoch": 9.723663330078125e-06,
      "model_forward_time": 3.1355206966400146,
      "step": 2549
    },
    {
      "epoch": 9.723663330078125e-06,
      "step": 2549,
      "training_step_time": 10.26883053779602
    },
    {
      "epoch": 9.72747802734375e-06,
      "grad_norm": 0.5467236638069153,
      "learning_rate": 1e-05,
      "loss": 0.1045,
      "step": 2550
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 2.4765279293060303,
      "step": 2550
    },
    {
      "dynamics_loss_avg": 0.062398819625377654,
      "epoch": 9.72747802734375e-06,
      "step": 2550
    },
    {
      "action_loss_avg": 0.031587587855756286,
      "epoch": 9.72747802734375e-06,
      "step": 2550
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 2550,
      "training_step_time": 7.496920108795166
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 2.4347455501556396,
      "step": 2550
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 2550,
      "training_step_time": 7.452981233596802
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 3.136054754257202,
      "step": 2550
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 2550,
      "training_step_time": 9.901266098022461
    },
    {
      "epoch": 9.72747802734375e-06,
      "model_forward_time": 3.385859727859497,
      "step": 2550
    },
    {
      "epoch": 9.72747802734375e-06,
      "step": 2550,
      "training_step_time": 10.277364492416382
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 3.186094284057617,
      "step": 2551
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 2551,
      "training_step_time": 9.951236963272095
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 3.137208938598633,
      "step": 2551
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 2551,
      "training_step_time": 9.897114515304565
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 3.142017126083374,
      "step": 2551
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 2551,
      "training_step_time": 9.910617589950562
    },
    {
      "epoch": 9.731292724609376e-06,
      "model_forward_time": 3.1452810764312744,
      "step": 2551
    },
    {
      "epoch": 9.731292724609376e-06,
      "step": 2551,
      "training_step_time": 10.285377740859985
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 3.182608127593994,
      "step": 2552
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 2552,
      "training_step_time": 9.954237461090088
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 3.137601375579834,
      "step": 2552
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 2552,
      "training_step_time": 9.906174898147583
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 3.135043144226074,
      "step": 2552
    },
    {
      "dynamics_loss_avg": 0.06927897296845913,
      "epoch": 9.735107421875e-06,
      "step": 2552
    },
    {
      "action_loss_avg": 0.04001306463032961,
      "epoch": 9.735107421875e-06,
      "step": 2552
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 2552,
      "training_step_time": 9.905188798904419
    },
    {
      "epoch": 9.735107421875e-06,
      "model_forward_time": 3.138230562210083,
      "step": 2552
    },
    {
      "epoch": 9.735107421875e-06,
      "step": 2552,
      "training_step_time": 10.315426349639893
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 3.4454801082611084,
      "step": 2553
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 2553,
      "training_step_time": 10.22102427482605
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 3.141294479370117,
      "step": 2553
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 2553,
      "training_step_time": 9.913597583770752
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 3.1414730548858643,
      "step": 2553
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 2553,
      "training_step_time": 9.908933877944946
    },
    {
      "epoch": 9.738922119140625e-06,
      "model_forward_time": 2.4390761852264404,
      "step": 2553
    },
    {
      "epoch": 9.738922119140625e-06,
      "step": 2553,
      "training_step_time": 10.02827000617981
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 3.167978048324585,
      "step": 2554
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 2554,
      "training_step_time": 9.945150136947632
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 3.1386563777923584,
      "step": 2554
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 2554,
      "training_step_time": 9.916119813919067
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 3.4055287837982178,
      "step": 2554
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 2554,
      "training_step_time": 10.173307418823242
    },
    {
      "epoch": 9.74273681640625e-06,
      "model_forward_time": 3.1384682655334473,
      "step": 2554
    },
    {
      "epoch": 9.74273681640625e-06,
      "step": 2554,
      "training_step_time": 10.288923263549805
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 3.178102731704712,
      "step": 2555
    },
    {
      "dynamics_loss_avg": 0.05149848610162735,
      "epoch": 9.746551513671876e-06,
      "step": 2555
    },
    {
      "action_loss_avg": 0.02217604694887996,
      "epoch": 9.746551513671876e-06,
      "step": 2555
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 2555,
      "training_step_time": 9.948292255401611
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 3.135197162628174,
      "step": 2555
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 2555,
      "training_step_time": 9.900261163711548
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 3.408951759338379,
      "step": 2555
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 2555,
      "training_step_time": 10.175609111785889
    },
    {
      "epoch": 9.746551513671876e-06,
      "model_forward_time": 3.1402781009674072,
      "step": 2555
    },
    {
      "epoch": 9.746551513671876e-06,
      "step": 2555,
      "training_step_time": 10.287836790084839
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 3.177027702331543,
      "step": 2556
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 2556,
      "training_step_time": 9.947494745254517
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 3.137148141860962,
      "step": 2556
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 2556,
      "training_step_time": 9.90151858329773
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 3.13836669921875,
      "step": 2556
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 2556,
      "training_step_time": 9.907639026641846
    },
    {
      "epoch": 9.7503662109375e-06,
      "model_forward_time": 3.1367454528808594,
      "step": 2556
    },
    {
      "epoch": 9.7503662109375e-06,
      "step": 2556,
      "training_step_time": 10.256651639938354
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 3.1868324279785156,
      "step": 2557
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 2557,
      "training_step_time": 9.952884674072266
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 3.128009080886841,
      "step": 2557
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 2557,
      "training_step_time": 9.898875713348389
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 3.137887954711914,
      "step": 2557
    },
    {
      "dynamics_loss_avg": 0.06582984421402216,
      "epoch": 9.754180908203125e-06,
      "step": 2557
    },
    {
      "action_loss_avg": 0.030961552960798146,
      "epoch": 9.754180908203125e-06,
      "step": 2557
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 2557,
      "training_step_time": 9.906254529953003
    },
    {
      "epoch": 9.754180908203125e-06,
      "model_forward_time": 3.137845516204834,
      "step": 2557
    },
    {
      "epoch": 9.754180908203125e-06,
      "step": 2557,
      "training_step_time": 10.304768323898315
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 2.738426685333252,
      "step": 2558
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 2558,
      "training_step_time": 7.767820119857788
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 3.1315345764160156,
      "step": 2558
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 2558,
      "training_step_time": 9.898330926895142
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 3.136577844619751,
      "step": 2558
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 2558,
      "training_step_time": 9.909124612808228
    },
    {
      "epoch": 9.75799560546875e-06,
      "model_forward_time": 3.1365199089050293,
      "step": 2558
    },
    {
      "epoch": 9.75799560546875e-06,
      "step": 2558,
      "training_step_time": 10.303598165512085
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 3.1804754734039307,
      "step": 2559
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 2559,
      "training_step_time": 9.953461647033691
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 3.398258686065674,
      "step": 2559
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 2559,
      "training_step_time": 10.167450904846191
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 2.438262701034546,
      "step": 2559
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 2559,
      "training_step_time": 7.463905334472656
    },
    {
      "epoch": 9.761810302734376e-06,
      "model_forward_time": 3.1360838413238525,
      "step": 2559
    },
    {
      "epoch": 9.761810302734376e-06,
      "step": 2559,
      "training_step_time": 10.30408263206482
    },
    {
      "epoch": 9.765625e-06,
      "grad_norm": 0.3641569912433624,
      "learning_rate": 1e-05,
      "loss": 0.1052,
      "step": 2560
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 3.1759426593780518,
      "step": 2560
    },
    {
      "dynamics_loss_avg": 0.06536174677312374,
      "epoch": 9.765625e-06,
      "step": 2560
    },
    {
      "action_loss_avg": 0.03644547965377569,
      "epoch": 9.765625e-06,
      "step": 2560
    },
    {
      "epoch": 9.765625e-06,
      "step": 2560,
      "training_step_time": 9.952003479003906
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 3.137131452560425,
      "step": 2560
    },
    {
      "epoch": 9.765625e-06,
      "step": 2560,
      "training_step_time": 9.904678583145142
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 3.136711835861206,
      "step": 2560
    },
    {
      "epoch": 9.765625e-06,
      "step": 2560,
      "training_step_time": 10.181693315505981
    },
    {
      "epoch": 9.765625e-06,
      "model_forward_time": 3.137267827987671,
      "step": 2560
    },
    {
      "epoch": 9.765625e-06,
      "step": 2560,
      "training_step_time": 10.228456735610962
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 2.467094659805298,
      "step": 2561
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 2561,
      "training_step_time": 7.492088317871094
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 3.132857322692871,
      "step": 2561
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 2561,
      "training_step_time": 9.896028280258179
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 3.1369411945343018,
      "step": 2561
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 2561,
      "training_step_time": 9.900840759277344
    },
    {
      "epoch": 9.769439697265625e-06,
      "model_forward_time": 3.133582353591919,
      "step": 2561
    },
    {
      "epoch": 9.769439697265625e-06,
      "step": 2561,
      "training_step_time": 10.288748502731323
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 3.4385833740234375,
      "step": 2562
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 2562,
      "training_step_time": 10.201855182647705
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 3.1355438232421875,
      "step": 2562
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 2562,
      "training_step_time": 9.89825177192688
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 3.1336827278137207,
      "step": 2562
    },
    {
      "dynamics_loss_avg": 0.05863860491663218,
      "epoch": 9.77325439453125e-06,
      "step": 2562
    },
    {
      "action_loss_avg": 0.03296042559668422,
      "epoch": 9.77325439453125e-06,
      "step": 2562
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 2562,
      "training_step_time": 9.901423931121826
    },
    {
      "epoch": 9.77325439453125e-06,
      "model_forward_time": 3.138198137283325,
      "step": 2562
    },
    {
      "epoch": 9.77325439453125e-06,
      "step": 2562,
      "training_step_time": 10.32101321220398
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 3.169616937637329,
      "step": 2563
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 2563,
      "training_step_time": 10.205446720123291
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 3.140765428543091,
      "step": 2563
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 2563,
      "training_step_time": 9.912766456604004
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 3.1398494243621826,
      "step": 2563
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 2563,
      "training_step_time": 9.912188291549683
    },
    {
      "epoch": 9.777069091796876e-06,
      "model_forward_time": 3.143235683441162,
      "step": 2563
    },
    {
      "epoch": 9.777069091796876e-06,
      "step": 2563,
      "training_step_time": 10.237514019012451
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 3.1819753646850586,
      "step": 2564
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 2564,
      "training_step_time": 9.95590591430664
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 3.4238100051879883,
      "step": 2564
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 2564,
      "training_step_time": 10.195863485336304
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 3.1433544158935547,
      "step": 2564
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 2564,
      "training_step_time": 9.912839651107788
    },
    {
      "epoch": 9.7808837890625e-06,
      "model_forward_time": 2.4392006397247314,
      "step": 2564
    },
    {
      "epoch": 9.7808837890625e-06,
      "step": 2564,
      "training_step_time": 10.285032987594604
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 3.1732981204986572,
      "step": 2565
    },
    {
      "dynamics_loss_avg": 0.06155602373182774,
      "epoch": 9.784698486328125e-06,
      "step": 2565
    },
    {
      "action_loss_avg": 0.0411532424390316,
      "epoch": 9.784698486328125e-06,
      "step": 2565
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 2565,
      "training_step_time": 9.94721007347107
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 3.1328632831573486,
      "step": 2565
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 2565,
      "training_step_time": 9.897971630096436
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 3.1397793292999268,
      "step": 2565
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 2565,
      "training_step_time": 10.175897121429443
    },
    {
      "epoch": 9.784698486328125e-06,
      "model_forward_time": 3.1410422325134277,
      "step": 2565
    },
    {
      "epoch": 9.784698486328125e-06,
      "step": 2565,
      "training_step_time": 10.017577409744263
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 3.16938853263855,
      "step": 2566
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 2566,
      "training_step_time": 9.948126792907715
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 3.1322824954986572,
      "step": 2566
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 2566,
      "training_step_time": 9.9041109085083
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 3.1358718872070312,
      "step": 2566
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 2566,
      "training_step_time": 9.911357402801514
    },
    {
      "epoch": 9.78851318359375e-06,
      "model_forward_time": 3.4064743518829346,
      "step": 2566
    },
    {
      "epoch": 9.78851318359375e-06,
      "step": 2566,
      "training_step_time": 10.275363683700562
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 2.4594571590423584,
      "step": 2567
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 2567,
      "training_step_time": 7.4936745166778564
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 3.1297852993011475,
      "step": 2567
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 2567,
      "training_step_time": 9.902313470840454
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 3.131152629852295,
      "step": 2567
    },
    {
      "dynamics_loss_avg": 0.07822662256658078,
      "epoch": 9.792327880859376e-06,
      "step": 2567
    },
    {
      "action_loss_avg": 0.04783633686602116,
      "epoch": 9.792327880859376e-06,
      "step": 2567
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 2567,
      "training_step_time": 9.906446933746338
    },
    {
      "epoch": 9.792327880859376e-06,
      "model_forward_time": 3.1312217712402344,
      "step": 2567
    },
    {
      "epoch": 9.792327880859376e-06,
      "step": 2567,
      "training_step_time": 10.305242776870728
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 3.1590020656585693,
      "step": 2568
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 2568,
      "training_step_time": 10.206506729125977
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 3.129304885864258,
      "step": 2568
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 2568,
      "training_step_time": 9.898670196533203
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 3.1359527111053467,
      "step": 2568
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 2568,
      "training_step_time": 9.909024477005005
    },
    {
      "epoch": 9.796142578125e-06,
      "model_forward_time": 3.134100914001465,
      "step": 2568
    },
    {
      "epoch": 9.796142578125e-06,
      "step": 2568,
      "training_step_time": 10.351056337356567
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 3.1706762313842773,
      "step": 2569
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 2569,
      "training_step_time": 9.942848920822144
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 3.384617328643799,
      "step": 2569
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 2569,
      "training_step_time": 10.158724069595337
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 3.1354458332061768,
      "step": 2569
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 2569,
      "training_step_time": 9.910914897918701
    },
    {
      "epoch": 9.799957275390625e-06,
      "model_forward_time": 3.1357603073120117,
      "step": 2569
    },
    {
      "epoch": 9.799957275390625e-06,
      "step": 2569,
      "training_step_time": 10.307018518447876
    },
    {
      "epoch": 9.80377197265625e-06,
      "grad_norm": 0.47733741998672485,
      "learning_rate": 1e-05,
      "loss": 0.1035,
      "step": 2570
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 3.1672799587249756,
      "step": 2570
    },
    {
      "dynamics_loss_avg": 0.07512093670666217,
      "epoch": 9.80377197265625e-06,
      "step": 2570
    },
    {
      "action_loss_avg": 0.050451317615807054,
      "epoch": 9.80377197265625e-06,
      "step": 2570
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 2570,
      "training_step_time": 9.94801950454712
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 2.4321794509887695,
      "step": 2570
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 2570,
      "training_step_time": 7.467259645462036
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 2.432098388671875,
      "step": 2570
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 2570,
      "training_step_time": 7.465075254440308
    },
    {
      "epoch": 9.80377197265625e-06,
      "model_forward_time": 3.379725456237793,
      "step": 2570
    },
    {
      "epoch": 9.80377197265625e-06,
      "step": 2570,
      "training_step_time": 10.25396728515625
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 3.175124168395996,
      "step": 2571
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 2571,
      "training_step_time": 9.954949378967285
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 3.136164665222168,
      "step": 2571
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 2571,
      "training_step_time": 9.913212776184082
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 3.1422224044799805,
      "step": 2571
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 2571,
      "training_step_time": 9.931827783584595
    },
    {
      "epoch": 9.807586669921876e-06,
      "model_forward_time": 3.1446430683135986,
      "step": 2571
    },
    {
      "epoch": 9.807586669921876e-06,
      "step": 2571,
      "training_step_time": 10.297311782836914
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 3.1749658584594727,
      "step": 2572
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 2572,
      "training_step_time": 9.953046560287476
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 3.1436755657196045,
      "step": 2572
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 2572,
      "training_step_time": 9.917192459106445
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 3.1383886337280273,
      "step": 2572
    },
    {
      "dynamics_loss_avg": 0.05873370636254549,
      "epoch": 9.8114013671875e-06,
      "step": 2572
    },
    {
      "action_loss_avg": 0.0298823076300323,
      "epoch": 9.8114013671875e-06,
      "step": 2572
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 2572,
      "training_step_time": 9.918341636657715
    },
    {
      "epoch": 9.8114013671875e-06,
      "model_forward_time": 3.140425205230713,
      "step": 2572
    },
    {
      "epoch": 9.8114013671875e-06,
      "step": 2572,
      "training_step_time": 10.294856548309326
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 3.4714455604553223,
      "step": 2573
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 2573,
      "training_step_time": 10.241979598999023
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 3.1436445713043213,
      "step": 2573
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 2573,
      "training_step_time": 9.91336703300476
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 3.139660120010376,
      "step": 2573
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 2573,
      "training_step_time": 9.911571502685547
    },
    {
      "epoch": 9.815216064453125e-06,
      "model_forward_time": 3.1453239917755127,
      "step": 2573
    },
    {
      "epoch": 9.815216064453125e-06,
      "step": 2573,
      "training_step_time": 10.259823083877563
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 3.1955156326293945,
      "step": 2574
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 2574,
      "training_step_time": 9.980045318603516
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 3.1443839073181152,
      "step": 2574
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 2574,
      "training_step_time": 10.196053981781006
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 3.141279697418213,
      "step": 2574
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 2574,
      "training_step_time": 9.920750617980957
    },
    {
      "epoch": 9.81903076171875e-06,
      "model_forward_time": 3.144815444946289,
      "step": 2574
    },
    {
      "epoch": 9.81903076171875e-06,
      "step": 2574,
      "training_step_time": 10.31016993522644
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 3.1842198371887207,
      "step": 2575
    },
    {
      "dynamics_loss_avg": 0.07163831619545817,
      "epoch": 9.822845458984376e-06,
      "step": 2575
    },
    {
      "action_loss_avg": 0.031944615114480254,
      "epoch": 9.822845458984376e-06,
      "step": 2575
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 2575,
      "training_step_time": 9.967004776000977
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 3.145005702972412,
      "step": 2575
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 2575,
      "training_step_time": 9.927398443222046
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 3.41878342628479,
      "step": 2575
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 2575,
      "training_step_time": 10.204551935195923
    },
    {
      "epoch": 9.822845458984376e-06,
      "model_forward_time": 3.1509687900543213,
      "step": 2575
    },
    {
      "epoch": 9.822845458984376e-06,
      "step": 2575,
      "training_step_time": 10.034832239151001
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 2.4761903285980225,
      "step": 2576
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 2576,
      "training_step_time": 7.511476755142212
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 3.1420323848724365,
      "step": 2576
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 2576,
      "training_step_time": 9.921650171279907
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 3.1451385021209717,
      "step": 2576
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 2576,
      "training_step_time": 9.926228761672974
    },
    {
      "epoch": 9.82666015625e-06,
      "model_forward_time": 3.1423935890197754,
      "step": 2576
    },
    {
      "epoch": 9.82666015625e-06,
      "step": 2576,
      "training_step_time": 10.026498794555664
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 3.4981141090393066,
      "step": 2577
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 2577,
      "training_step_time": 10.280239820480347
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 3.143406391143799,
      "step": 2577
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 2577,
      "training_step_time": 9.924778938293457
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 3.148529052734375,
      "step": 2577
    },
    {
      "dynamics_loss_avg": 0.06366172228008508,
      "epoch": 9.830474853515625e-06,
      "step": 2577
    },
    {
      "action_loss_avg": 0.03442533714696765,
      "epoch": 9.830474853515625e-06,
      "step": 2577
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 2577,
      "training_step_time": 9.933501243591309
    },
    {
      "epoch": 9.830474853515625e-06,
      "model_forward_time": 3.1429033279418945,
      "step": 2577
    },
    {
      "epoch": 9.830474853515625e-06,
      "step": 2577,
      "training_step_time": 10.28237795829773
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 3.458123207092285,
      "step": 2578
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 2578,
      "training_step_time": 10.23351764678955
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 3.138223171234131,
      "step": 2578
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 2578,
      "training_step_time": 9.922943830490112
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 3.1432387828826904,
      "step": 2578
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 2578,
      "training_step_time": 9.925304889678955
    },
    {
      "epoch": 9.83428955078125e-06,
      "model_forward_time": 3.1415534019470215,
      "step": 2578
    },
    {
      "epoch": 9.83428955078125e-06,
      "step": 2578,
      "training_step_time": 10.327073574066162
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 3.1833505630493164,
      "step": 2579
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 2579,
      "training_step_time": 9.960482835769653
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 3.396925926208496,
      "step": 2579
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 2579,
      "training_step_time": 10.170746803283691
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 3.138333797454834,
      "step": 2579
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 2579,
      "training_step_time": 9.91016149520874
    },
    {
      "epoch": 9.838104248046876e-06,
      "model_forward_time": 3.1367642879486084,
      "step": 2579
    },
    {
      "epoch": 9.838104248046876e-06,
      "step": 2579,
      "training_step_time": 10.287461996078491
    },
    {
      "epoch": 9.8419189453125e-06,
      "grad_norm": 0.5669876337051392,
      "learning_rate": 1e-05,
      "loss": 0.1022,
      "step": 2580
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 3.1700189113616943,
      "step": 2580
    },
    {
      "dynamics_loss_avg": 0.08052845895290375,
      "epoch": 9.8419189453125e-06,
      "step": 2580
    },
    {
      "action_loss_avg": 0.04790293779224157,
      "epoch": 9.8419189453125e-06,
      "step": 2580
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 2580,
      "training_step_time": 9.947924852371216
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 3.13692307472229,
      "step": 2580
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 2580,
      "training_step_time": 9.908334732055664
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 3.133211851119995,
      "step": 2580
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 2580,
      "training_step_time": 10.183878183364868
    },
    {
      "epoch": 9.8419189453125e-06,
      "model_forward_time": 3.135206937789917,
      "step": 2580
    },
    {
      "epoch": 9.8419189453125e-06,
      "step": 2580,
      "training_step_time": 10.014094829559326
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 3.188552141189575,
      "step": 2581
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 2581,
      "training_step_time": 9.957010269165039
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 3.137561082839966,
      "step": 2581
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 2581,
      "training_step_time": 9.907150506973267
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 3.1281843185424805,
      "step": 2581
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 2581,
      "training_step_time": 9.89707636833191
    },
    {
      "epoch": 9.845733642578125e-06,
      "model_forward_time": 3.136840343475342,
      "step": 2581
    },
    {
      "epoch": 9.845733642578125e-06,
      "step": 2581,
      "training_step_time": 10.276794672012329
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 3.172807455062866,
      "step": 2582
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 2582,
      "training_step_time": 9.946380853652954
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 3.138292074203491,
      "step": 2582
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 2582,
      "training_step_time": 9.907629489898682
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 3.13810658454895,
      "step": 2582
    },
    {
      "dynamics_loss_avg": 0.06743003148585558,
      "epoch": 9.84954833984375e-06,
      "step": 2582
    },
    {
      "action_loss_avg": 0.030962893180549143,
      "epoch": 9.84954833984375e-06,
      "step": 2582
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 2582,
      "training_step_time": 9.913309812545776
    },
    {
      "epoch": 9.84954833984375e-06,
      "model_forward_time": 3.1389071941375732,
      "step": 2582
    },
    {
      "epoch": 9.84954833984375e-06,
      "step": 2582,
      "training_step_time": 10.274061679840088
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 3.1864399909973145,
      "step": 2583
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 2583,
      "training_step_time": 10.240617513656616
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 3.1404144763946533,
      "step": 2583
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 2583,
      "training_step_time": 9.914380073547363
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 3.133897066116333,
      "step": 2583
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 2583,
      "training_step_time": 9.905776500701904
    },
    {
      "epoch": 9.853363037109376e-06,
      "model_forward_time": 3.1407809257507324,
      "step": 2583
    },
    {
      "epoch": 9.853363037109376e-06,
      "step": 2583,
      "training_step_time": 10.312990188598633
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 3.1865882873535156,
      "step": 2584
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 2584,
      "training_step_time": 9.964320659637451
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 3.424569845199585,
      "step": 2584
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 2584,
      "training_step_time": 10.192710399627686
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 3.141474962234497,
      "step": 2584
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 2584,
      "training_step_time": 9.910607099533081
    },
    {
      "epoch": 9.857177734375e-06,
      "model_forward_time": 3.1409218311309814,
      "step": 2584
    },
    {
      "epoch": 9.857177734375e-06,
      "step": 2584,
      "training_step_time": 10.292489528656006
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 3.1833949089050293,
      "step": 2585
    },
    {
      "dynamics_loss_avg": 0.07565408088266849,
      "epoch": 9.860992431640625e-06,
      "step": 2585
    },
    {
      "action_loss_avg": 0.03561512716114521,
      "epoch": 9.860992431640625e-06,
      "step": 2585
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 2585,
      "training_step_time": 9.952210187911987
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 3.144115686416626,
      "step": 2585
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 2585,
      "training_step_time": 9.915042877197266
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 3.142641305923462,
      "step": 2585
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 2585,
      "training_step_time": 10.18162488937378
    },
    {
      "epoch": 9.860992431640625e-06,
      "model_forward_time": 2.4394724369049072,
      "step": 2585
    },
    {
      "epoch": 9.860992431640625e-06,
      "step": 2585,
      "training_step_time": 10.02851414680481
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 3.1689603328704834,
      "step": 2586
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 2586,
      "training_step_time": 9.945692777633667
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 3.1417620182037354,
      "step": 2586
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 2586,
      "training_step_time": 9.91890287399292
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 3.1450247764587402,
      "step": 2586
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 2586,
      "training_step_time": 9.91908574104309
    },
    {
      "epoch": 9.86480712890625e-06,
      "model_forward_time": 3.4237868785858154,
      "step": 2586
    },
    {
      "epoch": 9.86480712890625e-06,
      "step": 2586,
      "training_step_time": 10.295296669006348
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 2.478877544403076,
      "step": 2587
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 2587,
      "training_step_time": 7.509052515029907
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 3.1390492916107178,
      "step": 2587
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 2587,
      "training_step_time": 9.91232442855835
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 3.1454381942749023,
      "step": 2587
    },
    {
      "dynamics_loss_avg": 0.06173949260264635,
      "epoch": 9.868621826171876e-06,
      "step": 2587
    },
    {
      "action_loss_avg": 0.03319378634914756,
      "epoch": 9.868621826171876e-06,
      "step": 2587
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 2587,
      "training_step_time": 9.926499128341675
    },
    {
      "epoch": 9.868621826171876e-06,
      "model_forward_time": 3.142221689224243,
      "step": 2587
    },
    {
      "epoch": 9.868621826171876e-06,
      "step": 2587,
      "training_step_time": 10.014036893844604
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 3.1748924255371094,
      "step": 2588
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 2588,
      "training_step_time": 9.950413227081299
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 3.394644021987915,
      "step": 2588
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 2588,
      "training_step_time": 10.162061214447021
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 3.143876791000366,
      "step": 2588
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 2588,
      "training_step_time": 9.915550947189331
    },
    {
      "epoch": 9.8724365234375e-06,
      "model_forward_time": 2.434058904647827,
      "step": 2588
    },
    {
      "epoch": 9.8724365234375e-06,
      "step": 2588,
      "training_step_time": 10.273396253585815
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 3.180309295654297,
      "step": 2589
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 2589,
      "training_step_time": 9.944880962371826
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 3.1389825344085693,
      "step": 2589
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 2589,
      "training_step_time": 10.196417331695557
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 3.1437013149261475,
      "step": 2589
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 2589,
      "training_step_time": 9.91089677810669
    },
    {
      "epoch": 9.876251220703125e-06,
      "model_forward_time": 3.141653299331665,
      "step": 2589
    },
    {
      "epoch": 9.876251220703125e-06,
      "step": 2589,
      "training_step_time": 10.275743007659912
    },
    {
      "epoch": 9.88006591796875e-06,
      "grad_norm": 0.6467995643615723,
      "learning_rate": 1e-05,
      "loss": 0.1097,
      "step": 2590
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 3.1890170574188232,
      "step": 2590
    },
    {
      "dynamics_loss_avg": 0.08343441709876061,
      "epoch": 9.88006591796875e-06,
      "step": 2590
    },
    {
      "action_loss_avg": 0.050629104115068914,
      "epoch": 9.88006591796875e-06,
      "step": 2590
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 2590,
      "training_step_time": 9.964322805404663
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 3.152506113052368,
      "step": 2590
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 2590,
      "training_step_time": 9.920965671539307
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 3.140376091003418,
      "step": 2590
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 2590,
      "training_step_time": 10.17589521408081
    },
    {
      "epoch": 9.88006591796875e-06,
      "model_forward_time": 3.13750958442688,
      "step": 2590
    },
    {
      "epoch": 9.88006591796875e-06,
      "step": 2590,
      "training_step_time": 10.294815301895142
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 3.1878297328948975,
      "step": 2591
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 2591,
      "training_step_time": 9.96049189567566
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 2.4408836364746094,
      "step": 2591
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 2591,
      "training_step_time": 7.4677894115448
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 3.141472101211548,
      "step": 2591
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 2591,
      "training_step_time": 9.910271167755127
    },
    {
      "epoch": 9.883880615234376e-06,
      "model_forward_time": 2.4401967525482178,
      "step": 2591
    },
    {
      "epoch": 9.883880615234376e-06,
      "step": 2591,
      "training_step_time": 10.011799335479736
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 3.4871203899383545,
      "step": 2592
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 2592,
      "training_step_time": 10.259477138519287
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 3.1582071781158447,
      "step": 2592
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 2592,
      "training_step_time": 9.936148643493652
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 3.1434576511383057,
      "step": 2592
    },
    {
      "dynamics_loss_avg": 0.06795771569013595,
      "epoch": 9.8876953125e-06,
      "step": 2592
    },
    {
      "action_loss_avg": 0.040574651397764686,
      "epoch": 9.8876953125e-06,
      "step": 2592
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 2592,
      "training_step_time": 9.917577743530273
    },
    {
      "epoch": 9.8876953125e-06,
      "model_forward_time": 2.436917304992676,
      "step": 2592
    },
    {
      "epoch": 9.8876953125e-06,
      "step": 2592,
      "training_step_time": 10.322305679321289
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 3.1707847118377686,
      "step": 2593
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 2593,
      "training_step_time": 10.205836057662964
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 3.152890920639038,
      "step": 2593
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 2593,
      "training_step_time": 9.922313690185547
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 3.145291805267334,
      "step": 2593
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 2593,
      "training_step_time": 9.918075561523438
    },
    {
      "epoch": 9.891510009765625e-06,
      "model_forward_time": 3.153447151184082,
      "step": 2593
    },
    {
      "epoch": 9.891510009765625e-06,
      "step": 2593,
      "training_step_time": 10.334059000015259
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 3.193941593170166,
      "step": 2594
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 2594,
      "training_step_time": 9.969951868057251
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 3.144468069076538,
      "step": 2594
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 2594,
      "training_step_time": 10.18718409538269
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 3.142505407333374,
      "step": 2594
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 2594,
      "training_step_time": 9.913129568099976
    },
    {
      "epoch": 9.89532470703125e-06,
      "model_forward_time": 2.44123911857605,
      "step": 2594
    },
    {
      "epoch": 9.89532470703125e-06,
      "step": 2594,
      "training_step_time": 10.287899494171143
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 3.1775293350219727,
      "step": 2595
    },
    {
      "dynamics_loss_avg": 0.06487846374511719,
      "epoch": 9.899139404296876e-06,
      "step": 2595
    },
    {
      "action_loss_avg": 0.03934681173413992,
      "epoch": 9.899139404296876e-06,
      "step": 2595
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 2595,
      "training_step_time": 9.952670812606812
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 3.139878034591675,
      "step": 2595
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 2595,
      "training_step_time": 9.90148663520813
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 3.136190414428711,
      "step": 2595
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 2595,
      "training_step_time": 10.174663782119751
    },
    {
      "epoch": 9.899139404296876e-06,
      "model_forward_time": 3.143179416656494,
      "step": 2595
    },
    {
      "epoch": 9.899139404296876e-06,
      "step": 2595,
      "training_step_time": 10.319646120071411
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 3.185929298400879,
      "step": 2596
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 2596,
      "training_step_time": 9.951355695724487
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 3.1375980377197266,
      "step": 2596
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 2596,
      "training_step_time": 9.904348134994507
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 3.1377639770507812,
      "step": 2596
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 2596,
      "training_step_time": 9.903333187103271
    },
    {
      "epoch": 9.9029541015625e-06,
      "model_forward_time": 3.138552665710449,
      "step": 2596
    },
    {
      "epoch": 9.9029541015625e-06,
      "step": 2596,
      "training_step_time": 10.253969192504883
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 3.1593685150146484,
      "step": 2597
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 2597,
      "training_step_time": 9.934230089187622
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 3.1299238204956055,
      "step": 2597
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 2597,
      "training_step_time": 9.906018495559692
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 3.1287055015563965,
      "step": 2597
    },
    {
      "dynamics_loss_avg": 0.04558881446719169,
      "epoch": 9.906768798828125e-06,
      "step": 2597
    },
    {
      "action_loss_avg": 0.02387572331354022,
      "epoch": 9.906768798828125e-06,
      "step": 2597
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 2597,
      "training_step_time": 9.908777713775635
    },
    {
      "epoch": 9.906768798828125e-06,
      "model_forward_time": 3.1290786266326904,
      "step": 2597
    },
    {
      "epoch": 9.906768798828125e-06,
      "step": 2597,
      "training_step_time": 10.272410154342651
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 3.432934284210205,
      "step": 2598
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 2598,
      "training_step_time": 10.211810111999512
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 3.1445541381835938,
      "step": 2598
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 2598,
      "training_step_time": 9.922141551971436
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 3.150200128555298,
      "step": 2598
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 2598,
      "training_step_time": 9.934828996658325
    },
    {
      "epoch": 9.91058349609375e-06,
      "model_forward_time": 3.1492371559143066,
      "step": 2598
    },
    {
      "epoch": 9.91058349609375e-06,
      "step": 2598,
      "training_step_time": 10.288739919662476
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 3.168144941329956,
      "step": 2599
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 2599,
      "training_step_time": 9.95981740951538
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 3.3946943283081055,
      "step": 2599
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 2599,
      "training_step_time": 10.174369096755981
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 3.1389544010162354,
      "step": 2599
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 2599,
      "training_step_time": 9.915489435195923
    },
    {
      "epoch": 9.914398193359376e-06,
      "model_forward_time": 3.137146472930908,
      "step": 2599
    },
    {
      "epoch": 9.914398193359376e-06,
      "step": 2599,
      "training_step_time": 10.282958507537842
    },
    {
      "epoch": 9.918212890625e-06,
      "grad_norm": 0.44798287749290466,
      "learning_rate": 1e-05,
      "loss": 0.0992,
      "step": 2600
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 2.4752299785614014,
      "step": 2600
    },
    {
      "dynamics_loss_avg": 0.06590171325951814,
      "epoch": 9.918212890625e-06,
      "step": 2600
    },
    {
      "action_loss_avg": 0.029220589064061643,
      "epoch": 9.918212890625e-06,
      "step": 2600
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 2600,
      "training_step_time": 7.509268045425415
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 3.1350364685058594,
      "step": 2600
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 2600,
      "training_step_time": 9.910430908203125
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 2.436593532562256,
      "step": 2600
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 2600,
      "training_step_time": 7.7510199546813965
    },
    {
      "epoch": 9.918212890625e-06,
      "model_forward_time": 3.134599447250366,
      "step": 2600
    },
    {
      "epoch": 9.918212890625e-06,
      "step": 2600,
      "training_step_time": 10.033007621765137
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 3.1763341426849365,
      "step": 2601
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 2601,
      "training_step_time": 9.957242488861084
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 3.1382479667663574,
      "step": 2601
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 2601,
      "training_step_time": 9.919941902160645
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 3.1371421813964844,
      "step": 2601
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 2601,
      "training_step_time": 9.916703701019287
    },
    {
      "epoch": 9.922027587890625e-06,
      "model_forward_time": 2.433708667755127,
      "step": 2601
    },
    {
      "epoch": 9.922027587890625e-06,
      "step": 2601,
      "training_step_time": 10.28881311416626
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 3.433612585067749,
      "step": 2602
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 2602,
      "training_step_time": 10.21352243423462
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 3.1444404125213623,
      "step": 2602
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 2602,
      "training_step_time": 9.920157432556152
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 3.137025833129883,
      "step": 2602
    },
    {
      "dynamics_loss_avg": 0.05108510823920369,
      "epoch": 9.92584228515625e-06,
      "step": 2602
    },
    {
      "action_loss_avg": 0.032284918683581056,
      "epoch": 9.92584228515625e-06,
      "step": 2602
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 2602,
      "training_step_time": 9.917557716369629
    },
    {
      "epoch": 9.92584228515625e-06,
      "model_forward_time": 3.13801908493042,
      "step": 2602
    },
    {
      "epoch": 9.92584228515625e-06,
      "step": 2602,
      "training_step_time": 10.269869089126587
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 3.1709840297698975,
      "step": 2603
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 2603,
      "training_step_time": 10.209931135177612
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 3.1341283321380615,
      "step": 2603
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 2603,
      "training_step_time": 9.90692687034607
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 3.1332356929779053,
      "step": 2603
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 2603,
      "training_step_time": 9.908147096633911
    },
    {
      "epoch": 9.929656982421876e-06,
      "model_forward_time": 3.1368348598480225,
      "step": 2603
    },
    {
      "epoch": 9.929656982421876e-06,
      "step": 2603,
      "training_step_time": 10.228234052658081
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 3.175959825515747,
      "step": 2604
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 2604,
      "training_step_time": 9.953363418579102
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 3.416496515274048,
      "step": 2604
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 2604,
      "training_step_time": 10.184964895248413
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 3.134382486343384,
      "step": 2604
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 2604,
      "training_step_time": 9.902636051177979
    },
    {
      "epoch": 9.9334716796875e-06,
      "model_forward_time": 3.1348049640655518,
      "step": 2604
    },
    {
      "epoch": 9.9334716796875e-06,
      "step": 2604,
      "training_step_time": 10.247514724731445
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 3.170775890350342,
      "step": 2605
    },
    {
      "dynamics_loss_avg": 0.07930483333766461,
      "epoch": 9.937286376953125e-06,
      "step": 2605
    },
    {
      "action_loss_avg": 0.04650107007473707,
      "epoch": 9.937286376953125e-06,
      "step": 2605
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 2605,
      "training_step_time": 9.943146228790283
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 3.133086919784546,
      "step": 2605
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 2605,
      "training_step_time": 9.904260396957397
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 3.4022743701934814,
      "step": 2605
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 2605,
      "training_step_time": 10.178101062774658
    },
    {
      "epoch": 9.937286376953125e-06,
      "model_forward_time": 3.141164541244507,
      "step": 2605
    },
    {
      "epoch": 9.937286376953125e-06,
      "step": 2605,
      "training_step_time": 10.010598182678223
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 3.174546957015991,
      "step": 2606
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 2606,
      "training_step_time": 9.948175191879272
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 3.140430212020874,
      "step": 2606
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 2606,
      "training_step_time": 9.914802312850952
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 3.140415906906128,
      "step": 2606
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 2606,
      "training_step_time": 9.91709280014038
    },
    {
      "epoch": 9.94110107421875e-06,
      "model_forward_time": 2.7170052528381348,
      "step": 2606
    },
    {
      "epoch": 9.94110107421875e-06,
      "step": 2606,
      "training_step_time": 10.258258581161499
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 3.173557758331299,
      "step": 2607
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 2607,
      "training_step_time": 9.954671382904053
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 3.1338672637939453,
      "step": 2607
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 2607,
      "training_step_time": 9.904926300048828
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 3.1383132934570312,
      "step": 2607
    },
    {
      "dynamics_loss_avg": 0.0746742432937026,
      "epoch": 9.944915771484376e-06,
      "step": 2607
    },
    {
      "action_loss_avg": 0.04731205832213163,
      "epoch": 9.944915771484376e-06,
      "step": 2607
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 2607,
      "training_step_time": 9.909781455993652
    },
    {
      "epoch": 9.944915771484376e-06,
      "model_forward_time": 3.1333601474761963,
      "step": 2607
    },
    {
      "epoch": 9.944915771484376e-06,
      "step": 2607,
      "training_step_time": 10.285204887390137
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 2.4573187828063965,
      "step": 2608
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 2608,
      "training_step_time": 7.486542463302612
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 3.38861083984375,
      "step": 2608
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 2608,
      "training_step_time": 10.161134481430054
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 3.136087656021118,
      "step": 2608
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 2608,
      "training_step_time": 9.907411813735962
    },
    {
      "epoch": 9.94873046875e-06,
      "model_forward_time": 3.133857011795044,
      "step": 2608
    },
    {
      "epoch": 9.94873046875e-06,
      "step": 2608,
      "training_step_time": 10.240171670913696
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 2.4695420265197754,
      "step": 2609
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 2609,
      "training_step_time": 7.501142501831055
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 3.1383657455444336,
      "step": 2609
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 2609,
      "training_step_time": 10.199952363967896
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 3.1388421058654785,
      "step": 2609
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 2609,
      "training_step_time": 9.907582521438599
    },
    {
      "epoch": 9.952545166015625e-06,
      "model_forward_time": 3.1358819007873535,
      "step": 2609
    },
    {
      "epoch": 9.952545166015625e-06,
      "step": 2609,
      "training_step_time": 10.016464710235596
    },
    {
      "epoch": 9.95635986328125e-06,
      "grad_norm": 0.5271713137626648,
      "learning_rate": 1e-05,
      "loss": 0.1027,
      "step": 2610
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 3.171980857849121,
      "step": 2610
    },
    {
      "dynamics_loss_avg": 0.06094050873070955,
      "epoch": 9.95635986328125e-06,
      "step": 2610
    },
    {
      "action_loss_avg": 0.03023559171706438,
      "epoch": 9.95635986328125e-06,
      "step": 2610
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 2610,
      "training_step_time": 9.94967532157898
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 3.1320416927337646,
      "step": 2610
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 2610,
      "training_step_time": 9.906986951828003
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 3.135272979736328,
      "step": 2610
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 2610,
      "training_step_time": 10.188478708267212
    },
    {
      "epoch": 9.95635986328125e-06,
      "model_forward_time": 3.135695695877075,
      "step": 2610
    },
    {
      "epoch": 9.95635986328125e-06,
      "step": 2610,
      "training_step_time": 10.312347173690796
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 3.1612548828125,
      "step": 2611
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 2611,
      "training_step_time": 9.93895697593689
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 3.1403493881225586,
      "step": 2611
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 2611,
      "training_step_time": 9.920692443847656
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 3.1358864307403564,
      "step": 2611
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 2611,
      "training_step_time": 9.91382622718811
    },
    {
      "epoch": 9.960174560546876e-06,
      "model_forward_time": 3.1408236026763916,
      "step": 2611
    },
    {
      "epoch": 9.960174560546876e-06,
      "step": 2611,
      "training_step_time": 10.299497604370117
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 3.156883478164673,
      "step": 2612
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 2612,
      "training_step_time": 9.93309497833252
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 3.1371798515319824,
      "step": 2612
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 2612,
      "training_step_time": 9.912285089492798
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 3.1352427005767822,
      "step": 2612
    },
    {
      "dynamics_loss_avg": 0.0660583145916462,
      "epoch": 9.9639892578125e-06,
      "step": 2612
    },
    {
      "action_loss_avg": 0.042755937203764915,
      "epoch": 9.9639892578125e-06,
      "step": 2612
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 2612,
      "training_step_time": 9.912112951278687
    },
    {
      "epoch": 9.9639892578125e-06,
      "model_forward_time": 2.431910514831543,
      "step": 2612
    },
    {
      "epoch": 9.9639892578125e-06,
      "step": 2612,
      "training_step_time": 10.30705714225769
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 3.1607277393341064,
      "step": 2613
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 2613,
      "training_step_time": 9.935933828353882
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 3.415442943572998,
      "step": 2613
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 2613,
      "training_step_time": 10.187124729156494
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 3.1301918029785156,
      "step": 2613
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 2613,
      "training_step_time": 9.901740550994873
    },
    {
      "epoch": 9.967803955078125e-06,
      "model_forward_time": 3.133841037750244,
      "step": 2613
    },
    {
      "epoch": 9.967803955078125e-06,
      "step": 2613,
      "training_step_time": 10.278212785720825
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 3.173809766769409,
      "step": 2614
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 2614,
      "training_step_time": 9.946184396743774
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 2.431497812271118,
      "step": 2614
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 2614,
      "training_step_time": 7.723693132400513
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 3.1315431594848633,
      "step": 2614
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 2614,
      "training_step_time": 9.906903505325317
    },
    {
      "epoch": 9.97161865234375e-06,
      "model_forward_time": 3.139207124710083,
      "step": 2614
    },
    {
      "epoch": 9.97161865234375e-06,
      "step": 2614,
      "training_step_time": 10.02590560913086
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 3.166525363922119,
      "step": 2615
    },
    {
      "dynamics_loss_avg": 0.07403008844703436,
      "epoch": 9.975433349609376e-06,
      "step": 2615
    },
    {
      "action_loss_avg": 0.03290761630050838,
      "epoch": 9.975433349609376e-06,
      "step": 2615
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 2615,
      "training_step_time": 9.938035488128662
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 3.136241912841797,
      "step": 2615
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 2615,
      "training_step_time": 9.90675401687622
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 3.403830051422119,
      "step": 2615
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 2615,
      "training_step_time": 10.174548149108887
    },
    {
      "epoch": 9.975433349609376e-06,
      "model_forward_time": 3.1367251873016357,
      "step": 2615
    },
    {
      "epoch": 9.975433349609376e-06,
      "step": 2615,
      "training_step_time": 10.282196760177612
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 3.149444818496704,
      "step": 2616
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 2616,
      "training_step_time": 9.921201705932617
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 2.4313697814941406,
      "step": 2616
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 2616,
      "training_step_time": 7.45487904548645
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 2.430879592895508,
      "step": 2616
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 2616,
      "training_step_time": 7.456074237823486
    },
    {
      "epoch": 9.979248046875e-06,
      "model_forward_time": 2.4252068996429443,
      "step": 2616
    },
    {
      "epoch": 9.979248046875e-06,
      "step": 2616,
      "training_step_time": 10.276202201843262
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 3.4399421215057373,
      "step": 2617
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 2617,
      "training_step_time": 10.20884919166565
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 3.131439685821533,
      "step": 2617
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 2617,
      "training_step_time": 9.900026559829712
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 3.1338722705841064,
      "step": 2617
    },
    {
      "dynamics_loss_avg": 0.048096480593085286,
      "epoch": 9.983062744140625e-06,
      "step": 2617
    },
    {
      "action_loss_avg": 0.02266939599066973,
      "epoch": 9.983062744140625e-06,
      "step": 2617
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 2617,
      "training_step_time": 9.900850772857666
    },
    {
      "epoch": 9.983062744140625e-06,
      "model_forward_time": 2.4322292804718018,
      "step": 2617
    },
    {
      "epoch": 9.983062744140625e-06,
      "step": 2617,
      "training_step_time": 10.243332147598267
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 3.1701836585998535,
      "step": 2618
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 2618,
      "training_step_time": 9.935929775238037
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 3.3787217140197754,
      "step": 2618
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 2618,
      "training_step_time": 10.139089345932007
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 3.1353986263275146,
      "step": 2618
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 2618,
      "training_step_time": 9.895031690597534
    },
    {
      "epoch": 9.98687744140625e-06,
      "model_forward_time": 3.13447904586792,
      "step": 2618
    },
    {
      "epoch": 9.98687744140625e-06,
      "step": 2618,
      "training_step_time": 10.302920579910278
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 3.1784508228302,
      "step": 2619
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 2619,
      "training_step_time": 9.952497482299805
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 3.1366753578186035,
      "step": 2619
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 2619,
      "training_step_time": 9.907024621963501
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 3.409625291824341,
      "step": 2619
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 2619,
      "training_step_time": 10.181725025177002
    },
    {
      "epoch": 9.990692138671876e-06,
      "model_forward_time": 3.1403839588165283,
      "step": 2619
    },
    {
      "epoch": 9.990692138671876e-06,
      "step": 2619,
      "training_step_time": 10.026596546173096
    },
    {
      "epoch": 9.9945068359375e-06,
      "grad_norm": 0.46521133184432983,
      "learning_rate": 1e-05,
      "loss": 0.1035,
      "step": 2620
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 3.172578811645508,
      "step": 2620
    },
    {
      "dynamics_loss_avg": 0.07119987169280648,
      "epoch": 9.9945068359375e-06,
      "step": 2620
    },
    {
      "action_loss_avg": 0.037236414523795246,
      "epoch": 9.9945068359375e-06,
      "step": 2620
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 2620,
      "training_step_time": 9.951030731201172
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 3.1423890590667725,
      "step": 2620
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 2620,
      "training_step_time": 9.920547485351562
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 3.1417648792266846,
      "step": 2620
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 2620,
      "training_step_time": 10.206420421600342
    },
    {
      "epoch": 9.9945068359375e-06,
      "model_forward_time": 3.1412153244018555,
      "step": 2620
    },
    {
      "epoch": 9.9945068359375e-06,
      "step": 2620,
      "training_step_time": 10.288031578063965
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 3.182872772216797,
      "step": 2621
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 2621,
      "training_step_time": 9.959473609924316
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 3.1424880027770996,
      "step": 2621
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 2621,
      "training_step_time": 9.919130563735962
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 3.137256145477295,
      "step": 2621
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 2621,
      "training_step_time": 9.91323447227478
    },
    {
      "epoch": 9.998321533203125e-06,
      "model_forward_time": 3.1447625160217285,
      "step": 2621
    },
    {
      "epoch": 9.998321533203125e-06,
      "step": 2621,
      "training_step_time": 10.291503667831421
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 2.471879005432129,
      "step": 2622
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 2622,
      "training_step_time": 7.501789808273315
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 3.13670015335083,
      "step": 2622
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 2622,
      "training_step_time": 9.906151533126831
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 3.133643627166748,
      "step": 2622
    },
    {
      "dynamics_loss_avg": 0.06362075991928577,
      "epoch": 1.000213623046875e-05,
      "step": 2622
    },
    {
      "action_loss_avg": 0.04167800210416317,
      "epoch": 1.000213623046875e-05,
      "step": 2622
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 2622,
      "training_step_time": 9.91025710105896
    },
    {
      "epoch": 1.000213623046875e-05,
      "model_forward_time": 3.134800434112549,
      "step": 2622
    },
    {
      "epoch": 1.000213623046875e-05,
      "step": 2622,
      "training_step_time": 10.259051084518433
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 3.16453218460083,
      "step": 2623
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 2623,
      "training_step_time": 9.937862634658813
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 3.402003288269043,
      "step": 2623
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 2623,
      "training_step_time": 10.170788526535034
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 3.1364243030548096,
      "step": 2623
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 2623,
      "training_step_time": 9.907307863235474
    },
    {
      "epoch": 1.0005950927734376e-05,
      "model_forward_time": 3.141113042831421,
      "step": 2623
    },
    {
      "epoch": 1.0005950927734376e-05,
      "step": 2623,
      "training_step_time": 10.279699325561523
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 3.175194501876831,
      "step": 2624
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 2624,
      "training_step_time": 9.951026678085327
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 3.428785800933838,
      "step": 2624
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 2624,
      "training_step_time": 10.20642352104187
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 3.1390349864959717,
      "step": 2624
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 2624,
      "training_step_time": 9.914966583251953
    },
    {
      "epoch": 1.0009765625e-05,
      "model_forward_time": 3.141932964324951,
      "step": 2624
    },
    {
      "epoch": 1.0009765625e-05,
      "step": 2624,
      "training_step_time": 10.252005815505981
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 3.176504373550415,
      "step": 2625
    },
    {
      "dynamics_loss_avg": 0.06653185114264488,
      "epoch": 1.0013580322265625e-05,
      "step": 2625
    },
    {
      "action_loss_avg": 0.04397455044090748,
      "epoch": 1.0013580322265625e-05,
      "step": 2625
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 2625,
      "training_step_time": 9.960029125213623
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 2.4384326934814453,
      "step": 2625
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 2625,
      "training_step_time": 7.468775033950806
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 3.139321804046631,
      "step": 2625
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 2625,
      "training_step_time": 9.917335033416748
    },
    {
      "epoch": 1.0013580322265625e-05,
      "model_forward_time": 3.433300018310547,
      "step": 2625
    },
    {
      "epoch": 1.0013580322265625e-05,
      "step": 2625,
      "training_step_time": 10.310499429702759
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 3.1801295280456543,
      "step": 2626
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 2626,
      "training_step_time": 9.960701704025269
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 3.139591693878174,
      "step": 2626
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 2626,
      "training_step_time": 9.915473222732544
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 3.145193099975586,
      "step": 2626
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 2626,
      "training_step_time": 9.920841932296753
    },
    {
      "epoch": 1.001739501953125e-05,
      "model_forward_time": 3.427318572998047,
      "step": 2626
    },
    {
      "epoch": 1.001739501953125e-05,
      "step": 2626,
      "training_step_time": 10.296669721603394
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 3.156111478805542,
      "step": 2627
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 2627,
      "training_step_time": 9.930822849273682
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 3.1382431983947754,
      "step": 2627
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 2627,
      "training_step_time": 9.91172194480896
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 2.435128927230835,
      "step": 2627
    },
    {
      "dynamics_loss_avg": 0.07886593863368034,
      "epoch": 1.0021209716796874e-05,
      "step": 2627
    },
    {
      "action_loss_avg": 0.03823896329849959,
      "epoch": 1.0021209716796874e-05,
      "step": 2627
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 2627,
      "training_step_time": 7.463497638702393
    },
    {
      "epoch": 1.0021209716796874e-05,
      "model_forward_time": 3.131610631942749,
      "step": 2627
    },
    {
      "epoch": 1.0021209716796874e-05,
      "step": 2627,
      "training_step_time": 10.276683807373047
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 2.4523208141326904,
      "step": 2628
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 2628,
      "training_step_time": 7.479658365249634
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 2.682826519012451,
      "step": 2628
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 2628,
      "training_step_time": 7.711009979248047
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 2.430903911590576,
      "step": 2628
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 2628,
      "training_step_time": 7.460010051727295
    },
    {
      "epoch": 1.00250244140625e-05,
      "model_forward_time": 3.133389711380005,
      "step": 2628
    },
    {
      "epoch": 1.00250244140625e-05,
      "step": 2628,
      "training_step_time": 10.297491788864136
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 3.181210517883301,
      "step": 2629
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 2629,
      "training_step_time": 9.952651739120483
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 3.1388156414031982,
      "step": 2629
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 2629,
      "training_step_time": 9.912169694900513
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 3.4087960720062256,
      "step": 2629
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 2629,
      "training_step_time": 10.18769907951355
    },
    {
      "epoch": 1.0028839111328125e-05,
      "model_forward_time": 3.140004873275757,
      "step": 2629
    },
    {
      "epoch": 1.0028839111328125e-05,
      "step": 2629,
      "training_step_time": 10.293726444244385
    },
    {
      "epoch": 1.003265380859375e-05,
      "grad_norm": 0.6054173111915588,
      "learning_rate": 1e-05,
      "loss": 0.1048,
      "step": 2630
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 3.1887805461883545,
      "step": 2630
    },
    {
      "dynamics_loss_avg": 0.06280944272875785,
      "epoch": 1.003265380859375e-05,
      "step": 2630
    },
    {
      "action_loss_avg": 0.0315978086553514,
      "epoch": 1.003265380859375e-05,
      "step": 2630
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 2630,
      "training_step_time": 9.972711324691772
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 3.1417717933654785,
      "step": 2630
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 2630,
      "training_step_time": 9.918338298797607
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 3.137476921081543,
      "step": 2630
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 2630,
      "training_step_time": 9.916987657546997
    },
    {
      "epoch": 1.003265380859375e-05,
      "model_forward_time": 3.3973004817962646,
      "step": 2630
    },
    {
      "epoch": 1.003265380859375e-05,
      "step": 2630,
      "training_step_time": 10.268709659576416
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 3.1719777584075928,
      "step": 2631
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 2631,
      "training_step_time": 9.947066068649292
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 3.137960433959961,
      "step": 2631
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 2631,
      "training_step_time": 9.914328575134277
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 3.1377406120300293,
      "step": 2631
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 2631,
      "training_step_time": 9.914247512817383
    },
    {
      "epoch": 1.0036468505859374e-05,
      "model_forward_time": 3.140528917312622,
      "step": 2631
    },
    {
      "epoch": 1.0036468505859374e-05,
      "step": 2631,
      "training_step_time": 10.016268491744995
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 3.4397084712982178,
      "step": 2632
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 2632,
      "training_step_time": 10.219320058822632
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 3.1404507160186768,
      "step": 2632
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 2632,
      "training_step_time": 9.914581537246704
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 2.4356610774993896,
      "step": 2632
    },
    {
      "dynamics_loss_avg": 0.05871590618044138,
      "epoch": 1.0040283203125e-05,
      "step": 2632
    },
    {
      "action_loss_avg": 0.03392131319269538,
      "epoch": 1.0040283203125e-05,
      "step": 2632
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 2632,
      "training_step_time": 7.471047401428223
    },
    {
      "epoch": 1.0040283203125e-05,
      "model_forward_time": 3.137040376663208,
      "step": 2632
    },
    {
      "epoch": 1.0040283203125e-05,
      "step": 2632,
      "training_step_time": 10.248172044754028
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 3.1764209270477295,
      "step": 2633
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 2633,
      "training_step_time": 10.220720052719116
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 3.1406500339508057,
      "step": 2633
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 2633,
      "training_step_time": 9.91560435295105
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 3.138028860092163,
      "step": 2633
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 2633,
      "training_step_time": 9.916316986083984
    },
    {
      "epoch": 1.0044097900390625e-05,
      "model_forward_time": 3.1404826641082764,
      "step": 2633
    },
    {
      "epoch": 1.0044097900390625e-05,
      "step": 2633,
      "training_step_time": 10.274637460708618
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 2.477856397628784,
      "step": 2634
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 2634,
      "training_step_time": 7.507695436477661
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 3.133690357208252,
      "step": 2634
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 2634,
      "training_step_time": 10.168545961380005
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 3.137645959854126,
      "step": 2634
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 2634,
      "training_step_time": 9.917263269424438
    },
    {
      "epoch": 1.004791259765625e-05,
      "model_forward_time": 3.138509750366211,
      "step": 2634
    },
    {
      "epoch": 1.004791259765625e-05,
      "step": 2634,
      "training_step_time": 10.30744457244873
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 3.160332679748535,
      "step": 2635
    },
    {
      "dynamics_loss_avg": 0.0711876168847084,
      "epoch": 1.0051727294921874e-05,
      "step": 2635
    },
    {
      "action_loss_avg": 0.04219693373888731,
      "epoch": 1.0051727294921874e-05,
      "step": 2635
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 2635,
      "training_step_time": 9.941753149032593
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 3.137192487716675,
      "step": 2635
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 2635,
      "training_step_time": 9.91076922416687
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 3.3920724391937256,
      "step": 2635
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 2635,
      "training_step_time": 10.167623043060303
    },
    {
      "epoch": 1.0051727294921874e-05,
      "model_forward_time": 2.432276487350464,
      "step": 2635
    },
    {
      "epoch": 1.0051727294921874e-05,
      "step": 2635,
      "training_step_time": 10.277618885040283
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 3.171347141265869,
      "step": 2636
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 2636,
      "training_step_time": 9.942789077758789
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 3.1332573890686035,
      "step": 2636
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 2636,
      "training_step_time": 9.907101392745972
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 3.1324307918548584,
      "step": 2636
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 2636,
      "training_step_time": 9.896642208099365
    },
    {
      "epoch": 1.00555419921875e-05,
      "model_forward_time": 3.131711006164551,
      "step": 2636
    },
    {
      "epoch": 1.00555419921875e-05,
      "step": 2636,
      "training_step_time": 10.259658098220825
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 3.183227062225342,
      "step": 2637
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 2637,
      "training_step_time": 9.96326470375061
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 3.130720853805542,
      "step": 2637
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 2637,
      "training_step_time": 9.906635999679565
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 3.1362533569335938,
      "step": 2637
    },
    {
      "dynamics_loss_avg": 0.0864202905446291,
      "epoch": 1.0059356689453125e-05,
      "step": 2637
    },
    {
      "action_loss_avg": 0.04168228842318058,
      "epoch": 1.0059356689453125e-05,
      "step": 2637
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 2637,
      "training_step_time": 9.913575887680054
    },
    {
      "epoch": 1.0059356689453125e-05,
      "model_forward_time": 3.136699676513672,
      "step": 2637
    },
    {
      "epoch": 1.0059356689453125e-05,
      "step": 2637,
      "training_step_time": 10.008755683898926
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 3.4417929649353027,
      "step": 2638
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 2638,
      "training_step_time": 10.21921157836914
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 3.1381163597106934,
      "step": 2638
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 2638,
      "training_step_time": 9.918034076690674
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 3.1371920108795166,
      "step": 2638
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 2638,
      "training_step_time": 9.9118971824646
    },
    {
      "epoch": 1.006317138671875e-05,
      "model_forward_time": 3.1330108642578125,
      "step": 2638
    },
    {
      "epoch": 1.006317138671875e-05,
      "step": 2638,
      "training_step_time": 10.272050142288208
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 3.192803144454956,
      "step": 2639
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 2639,
      "training_step_time": 9.962383508682251
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 3.3925602436065674,
      "step": 2639
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 2639,
      "training_step_time": 10.15951156616211
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 3.1366782188415527,
      "step": 2639
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 2639,
      "training_step_time": 9.908047676086426
    },
    {
      "epoch": 1.0066986083984374e-05,
      "model_forward_time": 3.1318020820617676,
      "step": 2639
    },
    {
      "epoch": 1.0066986083984374e-05,
      "step": 2639,
      "training_step_time": 10.281203985214233
    },
    {
      "epoch": 1.007080078125e-05,
      "grad_norm": 0.8073552250862122,
      "learning_rate": 1e-05,
      "loss": 0.1028,
      "step": 2640
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 3.1510515213012695,
      "step": 2640
    },
    {
      "dynamics_loss_avg": 0.054168407991528514,
      "epoch": 1.007080078125e-05,
      "step": 2640
    },
    {
      "action_loss_avg": 0.028190185222774743,
      "epoch": 1.007080078125e-05,
      "step": 2640
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 2640,
      "training_step_time": 9.922817707061768
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 3.1325464248657227,
      "step": 2640
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 2640,
      "training_step_time": 9.900724172592163
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 2.4311985969543457,
      "step": 2640
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 2640,
      "training_step_time": 7.748020172119141
    },
    {
      "epoch": 1.007080078125e-05,
      "model_forward_time": 3.1271920204162598,
      "step": 2640
    },
    {
      "epoch": 1.007080078125e-05,
      "step": 2640,
      "training_step_time": 10.302297592163086
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 3.1779842376708984,
      "step": 2641
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 2641,
      "training_step_time": 9.95593786239624
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 3.133338689804077,
      "step": 2641
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 2641,
      "training_step_time": 9.906909465789795
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 3.1318626403808594,
      "step": 2641
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 2641,
      "training_step_time": 9.905982732772827
    },
    {
      "epoch": 1.0074615478515625e-05,
      "model_forward_time": 3.408924102783203,
      "step": 2641
    },
    {
      "epoch": 1.0074615478515625e-05,
      "step": 2641,
      "training_step_time": 10.268644332885742
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 3.1541388034820557,
      "step": 2642
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 2642,
      "training_step_time": 9.929417610168457
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 3.137606143951416,
      "step": 2642
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 2642,
      "training_step_time": 9.910783767700195
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 2.4283053874969482,
      "step": 2642
    },
    {
      "dynamics_loss_avg": 0.06626074872910977,
      "epoch": 1.007843017578125e-05,
      "step": 2642
    },
    {
      "action_loss_avg": 0.035634423792362216,
      "epoch": 1.007843017578125e-05,
      "step": 2642
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 2642,
      "training_step_time": 7.458099365234375
    },
    {
      "epoch": 1.007843017578125e-05,
      "model_forward_time": 3.135100841522217,
      "step": 2642
    },
    {
      "epoch": 1.007843017578125e-05,
      "step": 2642,
      "training_step_time": 10.030460834503174
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 3.180119037628174,
      "step": 2643
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 2643,
      "training_step_time": 10.231895446777344
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 3.135160446166992,
      "step": 2643
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 2643,
      "training_step_time": 9.907111883163452
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 3.132265090942383,
      "step": 2643
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 2643,
      "training_step_time": 9.90327262878418
    },
    {
      "epoch": 1.0082244873046874e-05,
      "model_forward_time": 3.1328158378601074,
      "step": 2643
    },
    {
      "epoch": 1.0082244873046874e-05,
      "step": 2643,
      "training_step_time": 10.313002109527588
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 2.4746296405792236,
      "step": 2644
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 2644,
      "training_step_time": 7.506383180618286
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 2.428586483001709,
      "step": 2644
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 2644,
      "training_step_time": 7.721520185470581
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 3.1284544467926025,
      "step": 2644
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 2644,
      "training_step_time": 9.90025806427002
    },
    {
      "epoch": 1.00860595703125e-05,
      "model_forward_time": 3.1337814331054688,
      "step": 2644
    },
    {
      "epoch": 1.00860595703125e-05,
      "step": 2644,
      "training_step_time": 10.347020626068115
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 3.1864826679229736,
      "step": 2645
    },
    {
      "dynamics_loss_avg": 0.07343824319541455,
      "epoch": 1.0089874267578125e-05,
      "step": 2645
    },
    {
      "action_loss_avg": 0.03450866360217333,
      "epoch": 1.0089874267578125e-05,
      "step": 2645
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 2645,
      "training_step_time": 9.960761547088623
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 3.1298437118530273,
      "step": 2645
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 2645,
      "training_step_time": 9.901552438735962
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 3.1339542865753174,
      "step": 2645
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 2645,
      "training_step_time": 10.17856240272522
    },
    {
      "epoch": 1.0089874267578125e-05,
      "model_forward_time": 3.1293063163757324,
      "step": 2645
    },
    {
      "epoch": 1.0089874267578125e-05,
      "step": 2645,
      "training_step_time": 10.011266231536865
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 3.180248498916626,
      "step": 2646
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 2646,
      "training_step_time": 9.952074766159058
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 3.130178451538086,
      "step": 2646
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 2646,
      "training_step_time": 9.896642208099365
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 3.128215789794922,
      "step": 2646
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 2646,
      "training_step_time": 9.893578290939331
    },
    {
      "epoch": 1.009368896484375e-05,
      "model_forward_time": 3.409426689147949,
      "step": 2646
    },
    {
      "epoch": 1.009368896484375e-05,
      "step": 2646,
      "training_step_time": 10.268937110900879
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 3.16874623298645,
      "step": 2647
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 2647,
      "training_step_time": 9.933292865753174
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 3.132648468017578,
      "step": 2647
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 2647,
      "training_step_time": 9.901145458221436
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 3.1359217166900635,
      "step": 2647
    },
    {
      "dynamics_loss_avg": 0.06494861040264369,
      "epoch": 1.0097503662109374e-05,
      "step": 2647
    },
    {
      "action_loss_avg": 0.04104935908690095,
      "epoch": 1.0097503662109374e-05,
      "step": 2647
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 2647,
      "training_step_time": 9.90737509727478
    },
    {
      "epoch": 1.0097503662109374e-05,
      "model_forward_time": 3.131422758102417,
      "step": 2647
    },
    {
      "epoch": 1.0097503662109374e-05,
      "step": 2647,
      "training_step_time": 10.015734910964966
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 2.46305513381958,
      "step": 2648
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 2648,
      "training_step_time": 7.492676496505737
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 3.368852138519287,
      "step": 2648
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 2648,
      "training_step_time": 10.141553163528442
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 3.1422982215881348,
      "step": 2648
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 2648,
      "training_step_time": 9.915635585784912
    },
    {
      "epoch": 1.0101318359375e-05,
      "model_forward_time": 3.141277313232422,
      "step": 2648
    },
    {
      "epoch": 1.0101318359375e-05,
      "step": 2648,
      "training_step_time": 10.311394214630127
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 3.1869308948516846,
      "step": 2649
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 2649,
      "training_step_time": 9.95781660079956
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 3.3956995010375977,
      "step": 2649
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 2649,
      "training_step_time": 10.168330669403076
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 3.1400046348571777,
      "step": 2649
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 2649,
      "training_step_time": 9.923086881637573
    },
    {
      "epoch": 1.0105133056640625e-05,
      "model_forward_time": 3.141193151473999,
      "step": 2649
    },
    {
      "epoch": 1.0105133056640625e-05,
      "step": 2649,
      "training_step_time": 10.01667046546936
    },
    {
      "epoch": 1.010894775390625e-05,
      "grad_norm": 0.39163488149642944,
      "learning_rate": 1e-05,
      "loss": 0.1024,
      "step": 2650
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 3.17144513130188,
      "step": 2650
    },
    {
      "dynamics_loss_avg": 0.06443145014345646,
      "epoch": 1.010894775390625e-05,
      "step": 2650
    },
    {
      "action_loss_avg": 0.03531414773315191,
      "epoch": 1.010894775390625e-05,
      "step": 2650
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 2650,
      "training_step_time": 9.956477165222168
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 3.133978843688965,
      "step": 2650
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 2650,
      "training_step_time": 9.910200119018555
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 3.13321852684021,
      "step": 2650
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 2650,
      "training_step_time": 10.176070213317871
    },
    {
      "epoch": 1.010894775390625e-05,
      "model_forward_time": 3.137666702270508,
      "step": 2650
    },
    {
      "epoch": 1.010894775390625e-05,
      "step": 2650,
      "training_step_time": 10.279414415359497
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 3.141439199447632,
      "step": 2651
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 2651,
      "training_step_time": 9.92019534111023
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 3.1348278522491455,
      "step": 2651
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 2651,
      "training_step_time": 9.908725023269653
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 3.1300535202026367,
      "step": 2651
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 2651,
      "training_step_time": 9.896897792816162
    },
    {
      "epoch": 1.0112762451171874e-05,
      "model_forward_time": 2.433819055557251,
      "step": 2651
    },
    {
      "epoch": 1.0112762451171874e-05,
      "step": 2651,
      "training_step_time": 10.243999481201172
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 3.150277853012085,
      "step": 2652
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 2652,
      "training_step_time": 9.922065734863281
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 3.1278088092803955,
      "step": 2652
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 2652,
      "training_step_time": 9.901447057723999
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 3.1281509399414062,
      "step": 2652
    },
    {
      "dynamics_loss_avg": 0.05606147199869156,
      "epoch": 1.01165771484375e-05,
      "step": 2652
    },
    {
      "action_loss_avg": 0.03153488067910075,
      "epoch": 1.01165771484375e-05,
      "step": 2652
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 2652,
      "training_step_time": 9.899499416351318
    },
    {
      "epoch": 1.01165771484375e-05,
      "model_forward_time": 3.1329925060272217,
      "step": 2652
    },
    {
      "epoch": 1.01165771484375e-05,
      "step": 2652,
      "training_step_time": 10.240759134292603
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 2.4649651050567627,
      "step": 2653
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 2653,
      "training_step_time": 7.752582550048828
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 3.1291563510894775,
      "step": 2653
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 2653,
      "training_step_time": 9.891169786453247
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 3.126952886581421,
      "step": 2653
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 2653,
      "training_step_time": 9.895141363143921
    },
    {
      "epoch": 1.0120391845703125e-05,
      "model_forward_time": 3.129568338394165,
      "step": 2653
    },
    {
      "epoch": 1.0120391845703125e-05,
      "step": 2653,
      "training_step_time": 10.303046941757202
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 3.1692895889282227,
      "step": 2654
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 2654,
      "training_step_time": 9.937088251113892
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 2.4251513481140137,
      "step": 2654
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 2654,
      "training_step_time": 7.7103986740112305
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 3.1226742267608643,
      "step": 2654
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 2654,
      "training_step_time": 9.888722896575928
    },
    {
      "epoch": 1.012420654296875e-05,
      "model_forward_time": 3.1266720294952393,
      "step": 2654
    },
    {
      "epoch": 1.012420654296875e-05,
      "step": 2654,
      "training_step_time": 10.282806873321533
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 2.4746170043945312,
      "step": 2655
    },
    {
      "dynamics_loss_avg": 0.06351553220301867,
      "epoch": 1.0128021240234374e-05,
      "step": 2655
    },
    {
      "action_loss_avg": 0.03559374073520303,
      "epoch": 1.0128021240234374e-05,
      "step": 2655
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 2655,
      "training_step_time": 7.502687454223633
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 3.1260077953338623,
      "step": 2655
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 2655,
      "training_step_time": 9.891710042953491
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 3.386456251144409,
      "step": 2655
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 2655,
      "training_step_time": 10.152106523513794
    },
    {
      "epoch": 1.0128021240234374e-05,
      "model_forward_time": 3.1342904567718506,
      "step": 2655
    },
    {
      "epoch": 1.0128021240234374e-05,
      "step": 2655,
      "training_step_time": 10.28446626663208
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 3.173985242843628,
      "step": 2656
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 2656,
      "training_step_time": 9.948359727859497
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 3.132026195526123,
      "step": 2656
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 2656,
      "training_step_time": 9.903000116348267
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 3.1360747814178467,
      "step": 2656
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 2656,
      "training_step_time": 9.907338857650757
    },
    {
      "epoch": 1.01318359375e-05,
      "model_forward_time": 3.13703989982605,
      "step": 2656
    },
    {
      "epoch": 1.01318359375e-05,
      "step": 2656,
      "training_step_time": 10.295564413070679
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 2.468492031097412,
      "step": 2657
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 2657,
      "training_step_time": 7.5011022090911865
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 3.1301279067993164,
      "step": 2657
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 2657,
      "training_step_time": 9.903316259384155
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 3.135411500930786,
      "step": 2657
    },
    {
      "dynamics_loss_avg": 0.061351251974701884,
      "epoch": 1.0135650634765625e-05,
      "step": 2657
    },
    {
      "action_loss_avg": 0.03320320993661881,
      "epoch": 1.0135650634765625e-05,
      "step": 2657
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 2657,
      "training_step_time": 9.916618347167969
    },
    {
      "epoch": 1.0135650634765625e-05,
      "model_forward_time": 3.1352665424346924,
      "step": 2657
    },
    {
      "epoch": 1.0135650634765625e-05,
      "step": 2657,
      "training_step_time": 10.273664712905884
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 3.4504683017730713,
      "step": 2658
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 2658,
      "training_step_time": 10.228009462356567
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 3.1358160972595215,
      "step": 2658
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 2658,
      "training_step_time": 9.912231922149658
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 3.136098861694336,
      "step": 2658
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 2658,
      "training_step_time": 9.904319763183594
    },
    {
      "epoch": 1.013946533203125e-05,
      "model_forward_time": 2.4335224628448486,
      "step": 2658
    },
    {
      "epoch": 1.013946533203125e-05,
      "step": 2658,
      "training_step_time": 10.0182466506958
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 3.1628403663635254,
      "step": 2659
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 2659,
      "training_step_time": 9.93286418914795
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 3.134141683578491,
      "step": 2659
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 2659,
      "training_step_time": 10.172701835632324
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 3.1373355388641357,
      "step": 2659
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 2659,
      "training_step_time": 9.908123970031738
    },
    {
      "epoch": 1.0143280029296874e-05,
      "model_forward_time": 3.1302101612091064,
      "step": 2659
    },
    {
      "epoch": 1.0143280029296874e-05,
      "step": 2659,
      "training_step_time": 10.269054174423218
    },
    {
      "epoch": 1.01470947265625e-05,
      "grad_norm": 0.9356478452682495,
      "learning_rate": 1e-05,
      "loss": 0.1012,
      "step": 2660
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 3.1528100967407227,
      "step": 2660
    },
    {
      "dynamics_loss_avg": 0.06989310402423143,
      "epoch": 1.01470947265625e-05,
      "step": 2660
    },
    {
      "action_loss_avg": 0.04750140458345413,
      "epoch": 1.01470947265625e-05,
      "step": 2660
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 2660,
      "training_step_time": 9.92540454864502
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 3.13265323638916,
      "step": 2660
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 2660,
      "training_step_time": 9.899781942367554
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 2.4289016723632812,
      "step": 2660
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 2660,
      "training_step_time": 7.743881940841675
    },
    {
      "epoch": 1.01470947265625e-05,
      "model_forward_time": 3.128143787384033,
      "step": 2660
    },
    {
      "epoch": 1.01470947265625e-05,
      "step": 2660,
      "training_step_time": 10.31725525856018
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 2.470038890838623,
      "step": 2661
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 2661,
      "training_step_time": 7.495585203170776
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 3.13046932220459,
      "step": 2661
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 2661,
      "training_step_time": 9.897176742553711
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 3.1297013759613037,
      "step": 2661
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 2661,
      "training_step_time": 9.901596307754517
    },
    {
      "epoch": 1.0150909423828125e-05,
      "model_forward_time": 3.1360225677490234,
      "step": 2661
    },
    {
      "epoch": 1.0150909423828125e-05,
      "step": 2661,
      "training_step_time": 10.265238046646118
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 3.451392650604248,
      "step": 2662
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 2662,
      "training_step_time": 10.224721670150757
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 3.135176658630371,
      "step": 2662
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 2662,
      "training_step_time": 9.908225297927856
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 3.133042335510254,
      "step": 2662
    },
    {
      "dynamics_loss_avg": 0.060573984496295454,
      "epoch": 1.015472412109375e-05,
      "step": 2662
    },
    {
      "action_loss_avg": 0.029431201983243227,
      "epoch": 1.015472412109375e-05,
      "step": 2662
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 2662,
      "training_step_time": 9.908958911895752
    },
    {
      "epoch": 1.015472412109375e-05,
      "model_forward_time": 2.435753583908081,
      "step": 2662
    },
    {
      "epoch": 1.015472412109375e-05,
      "step": 2662,
      "training_step_time": 10.038458824157715
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 3.184875011444092,
      "step": 2663
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 2663,
      "training_step_time": 9.962188720703125
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 3.404430627822876,
      "step": 2663
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 2663,
      "training_step_time": 10.183129072189331
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 3.136507987976074,
      "step": 2663
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 2663,
      "training_step_time": 9.912920475006104
    },
    {
      "epoch": 1.0158538818359374e-05,
      "model_forward_time": 3.1461453437805176,
      "step": 2663
    },
    {
      "epoch": 1.0158538818359374e-05,
      "step": 2663,
      "training_step_time": 10.038065910339355
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 3.192439556121826,
      "step": 2664
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 2664,
      "training_step_time": 9.96525764465332
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 3.4220407009124756,
      "step": 2664
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 2664,
      "training_step_time": 10.1905517578125
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 3.137047052383423,
      "step": 2664
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 2664,
      "training_step_time": 9.901834487915039
    },
    {
      "epoch": 1.0162353515625e-05,
      "model_forward_time": 3.135507583618164,
      "step": 2664
    },
    {
      "epoch": 1.0162353515625e-05,
      "step": 2664,
      "training_step_time": 10.316530704498291
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 3.155409574508667,
      "step": 2665
    },
    {
      "dynamics_loss_avg": 0.04759279768913984,
      "epoch": 1.0166168212890625e-05,
      "step": 2665
    },
    {
      "action_loss_avg": 0.02180260345339775,
      "epoch": 1.0166168212890625e-05,
      "step": 2665
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 2665,
      "training_step_time": 9.926879644393921
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 3.1346986293792725,
      "step": 2665
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 2665,
      "training_step_time": 9.89732813835144
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 2.438631057739258,
      "step": 2665
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 2665,
      "training_step_time": 7.463655471801758
    },
    {
      "epoch": 1.0166168212890625e-05,
      "model_forward_time": 2.700178623199463,
      "step": 2665
    },
    {
      "epoch": 1.0166168212890625e-05,
      "step": 2665,
      "training_step_time": 10.274662971496582
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 3.145533800125122,
      "step": 2666
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 2666,
      "training_step_time": 9.91019892692566
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 3.133132219314575,
      "step": 2666
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 2666,
      "training_step_time": 9.893935918807983
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 2.4291696548461914,
      "step": 2666
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 2666,
      "training_step_time": 7.4549150466918945
    },
    {
      "epoch": 1.016998291015625e-05,
      "model_forward_time": 2.4315285682678223,
      "step": 2666
    },
    {
      "epoch": 1.016998291015625e-05,
      "step": 2666,
      "training_step_time": 10.227328062057495
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 3.4605681896209717,
      "step": 2667
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 2667,
      "training_step_time": 10.225879907608032
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 3.132146120071411,
      "step": 2667
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 2667,
      "training_step_time": 9.891110181808472
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 3.1324753761291504,
      "step": 2667
    },
    {
      "dynamics_loss_avg": 0.06148078814148903,
      "epoch": 1.0173797607421874e-05,
      "step": 2667
    },
    {
      "action_loss_avg": 0.02898742388933897,
      "epoch": 1.0173797607421874e-05,
      "step": 2667
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 2667,
      "training_step_time": 9.892411947250366
    },
    {
      "epoch": 1.0173797607421874e-05,
      "model_forward_time": 3.1309974193573,
      "step": 2667
    },
    {
      "epoch": 1.0173797607421874e-05,
      "step": 2667,
      "training_step_time": 10.006420373916626
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 3.1520416736602783,
      "step": 2668
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 2668,
      "training_step_time": 9.918589353561401
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 3.379021167755127,
      "step": 2668
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 2668,
      "training_step_time": 10.140613555908203
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 3.1354546546936035,
      "step": 2668
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 2668,
      "training_step_time": 9.893617391586304
    },
    {
      "epoch": 1.01776123046875e-05,
      "model_forward_time": 2.4309651851654053,
      "step": 2668
    },
    {
      "epoch": 1.01776123046875e-05,
      "step": 2668,
      "training_step_time": 10.00272274017334
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 3.1666650772094727,
      "step": 2669
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 2669,
      "training_step_time": 9.92707633972168
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 3.134920358657837,
      "step": 2669
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 2669,
      "training_step_time": 10.176360845565796
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 3.1398494243621826,
      "step": 2669
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 2669,
      "training_step_time": 9.906179666519165
    },
    {
      "epoch": 1.0181427001953125e-05,
      "model_forward_time": 3.138242721557617,
      "step": 2669
    },
    {
      "epoch": 1.0181427001953125e-05,
      "step": 2669,
      "training_step_time": 10.272369623184204
    },
    {
      "epoch": 1.018524169921875e-05,
      "grad_norm": 0.5255791544914246,
      "learning_rate": 1e-05,
      "loss": 0.0984,
      "step": 2670
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 3.1889801025390625,
      "step": 2670
    },
    {
      "dynamics_loss_avg": 0.055789438728243115,
      "epoch": 1.018524169921875e-05,
      "step": 2670
    },
    {
      "action_loss_avg": 0.0265225687995553,
      "epoch": 1.018524169921875e-05,
      "step": 2670
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 2670,
      "training_step_time": 9.962024688720703
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 2.435370683670044,
      "step": 2670
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 2670,
      "training_step_time": 7.461142063140869
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 3.131956100463867,
      "step": 2670
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 2670,
      "training_step_time": 9.90313172340393
    },
    {
      "epoch": 1.018524169921875e-05,
      "model_forward_time": 3.401963233947754,
      "step": 2670
    },
    {
      "epoch": 1.018524169921875e-05,
      "step": 2670,
      "training_step_time": 10.274056434631348
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 3.1773526668548584,
      "step": 2671
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 2671,
      "training_step_time": 9.95093297958374
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 3.1391549110412598,
      "step": 2671
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 2671,
      "training_step_time": 9.909608840942383
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 3.1404008865356445,
      "step": 2671
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 2671,
      "training_step_time": 9.911926984786987
    },
    {
      "epoch": 1.0189056396484374e-05,
      "model_forward_time": 2.434196949005127,
      "step": 2671
    },
    {
      "epoch": 1.0189056396484374e-05,
      "step": 2671,
      "training_step_time": 10.29215145111084
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 3.1695332527160645,
      "step": 2672
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 2672,
      "training_step_time": 9.939207792282104
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 3.13761568069458,
      "step": 2672
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 2672,
      "training_step_time": 9.917143821716309
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 3.138500690460205,
      "step": 2672
    },
    {
      "dynamics_loss_avg": 0.06143603604286909,
      "epoch": 1.019287109375e-05,
      "step": 2672
    },
    {
      "action_loss_avg": 0.02466259435750544,
      "epoch": 1.019287109375e-05,
      "step": 2672
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 2672,
      "training_step_time": 9.915501832962036
    },
    {
      "epoch": 1.019287109375e-05,
      "model_forward_time": 3.143117904663086,
      "step": 2672
    },
    {
      "epoch": 1.019287109375e-05,
      "step": 2672,
      "training_step_time": 10.260663986206055
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 3.443321704864502,
      "step": 2673
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 2673,
      "training_step_time": 10.216390609741211
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 3.1417577266693115,
      "step": 2673
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 2673,
      "training_step_time": 9.915971994400024
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 3.146313428878784,
      "step": 2673
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 2673,
      "training_step_time": 9.920381307601929
    },
    {
      "epoch": 1.0196685791015625e-05,
      "model_forward_time": 3.144951820373535,
      "step": 2673
    },
    {
      "epoch": 1.0196685791015625e-05,
      "step": 2673,
      "training_step_time": 10.033735990524292
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 3.1790103912353516,
      "step": 2674
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 2674,
      "training_step_time": 9.950648307800293
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 3.4202229976654053,
      "step": 2674
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 2674,
      "training_step_time": 10.185575723648071
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 3.1379175186157227,
      "step": 2674
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 2674,
      "training_step_time": 9.905724287033081
    },
    {
      "epoch": 1.020050048828125e-05,
      "model_forward_time": 3.13403582572937,
      "step": 2674
    },
    {
      "epoch": 1.020050048828125e-05,
      "step": 2674,
      "training_step_time": 10.282417058944702
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 3.1767656803131104,
      "step": 2675
    },
    {
      "dynamics_loss_avg": 0.07239589020609856,
      "epoch": 1.0204315185546874e-05,
      "step": 2675
    },
    {
      "action_loss_avg": 0.03526187017560005,
      "epoch": 1.0204315185546874e-05,
      "step": 2675
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 2675,
      "training_step_time": 9.94936990737915
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 3.1342806816101074,
      "step": 2675
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 2675,
      "training_step_time": 9.897239685058594
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 3.3902668952941895,
      "step": 2675
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 2675,
      "training_step_time": 10.156466007232666
    },
    {
      "epoch": 1.0204315185546874e-05,
      "model_forward_time": 3.138789653778076,
      "step": 2675
    },
    {
      "epoch": 1.0204315185546874e-05,
      "step": 2675,
      "training_step_time": 10.327266454696655
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 2.4806909561157227,
      "step": 2676
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 2676,
      "training_step_time": 7.500742197036743
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 3.127601146697998,
      "step": 2676
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 2676,
      "training_step_time": 9.890642642974854
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 3.1392948627471924,
      "step": 2676
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 2676,
      "training_step_time": 9.901936292648315
    },
    {
      "epoch": 1.02081298828125e-05,
      "model_forward_time": 3.133833885192871,
      "step": 2676
    },
    {
      "epoch": 1.02081298828125e-05,
      "step": 2676,
      "training_step_time": 10.259037017822266
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 3.171992301940918,
      "step": 2677
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 2677,
      "training_step_time": 9.939772129058838
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 3.1349475383758545,
      "step": 2677
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 2677,
      "training_step_time": 9.900063276290894
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 3.1337149143218994,
      "step": 2677
    },
    {
      "dynamics_loss_avg": 0.061668021231889726,
      "epoch": 1.0211944580078125e-05,
      "step": 2677
    },
    {
      "action_loss_avg": 0.032070958707481625,
      "epoch": 1.0211944580078125e-05,
      "step": 2677
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 2677,
      "training_step_time": 9.902089834213257
    },
    {
      "epoch": 1.0211944580078125e-05,
      "model_forward_time": 3.1408584117889404,
      "step": 2677
    },
    {
      "epoch": 1.0211944580078125e-05,
      "step": 2677,
      "training_step_time": 10.002264738082886
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 3.4664113521575928,
      "step": 2678
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 2678,
      "training_step_time": 10.230870723724365
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 3.1356077194213867,
      "step": 2678
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 2678,
      "training_step_time": 9.908136129379272
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 3.1409544944763184,
      "step": 2678
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 2678,
      "training_step_time": 9.911349058151245
    },
    {
      "epoch": 1.021575927734375e-05,
      "model_forward_time": 3.140425682067871,
      "step": 2678
    },
    {
      "epoch": 1.021575927734375e-05,
      "step": 2678,
      "training_step_time": 10.021425724029541
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 3.179548501968384,
      "step": 2679
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 2679,
      "training_step_time": 9.953240394592285
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 3.4242660999298096,
      "step": 2679
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 2679,
      "training_step_time": 10.191872119903564
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 3.140599012374878,
      "step": 2679
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 2679,
      "training_step_time": 9.91001582145691
    },
    {
      "epoch": 1.0219573974609374e-05,
      "model_forward_time": 3.132441759109497,
      "step": 2679
    },
    {
      "epoch": 1.0219573974609374e-05,
      "step": 2679,
      "training_step_time": 10.239194631576538
    },
    {
      "epoch": 1.0223388671875e-05,
      "grad_norm": 0.7823753952980042,
      "learning_rate": 1e-05,
      "loss": 0.0959,
      "step": 2680
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 3.1889543533325195,
      "step": 2680
    },
    {
      "dynamics_loss_avg": 0.08205710835754872,
      "epoch": 1.0223388671875e-05,
      "step": 2680
    },
    {
      "action_loss_avg": 0.04066564803943038,
      "epoch": 1.0223388671875e-05,
      "step": 2680
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 2680,
      "training_step_time": 9.957720756530762
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 3.1420092582702637,
      "step": 2680
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 2680,
      "training_step_time": 9.910861492156982
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 3.1359148025512695,
      "step": 2680
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 2680,
      "training_step_time": 10.20291519165039
    },
    {
      "epoch": 1.0223388671875e-05,
      "model_forward_time": 3.139871120452881,
      "step": 2680
    },
    {
      "epoch": 1.0223388671875e-05,
      "step": 2680,
      "training_step_time": 10.309074401855469
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 3.1751456260681152,
      "step": 2681
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 2681,
      "training_step_time": 9.941410779953003
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 3.1434667110443115,
      "step": 2681
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 2681,
      "training_step_time": 9.910718441009521
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 3.1392409801483154,
      "step": 2681
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 2681,
      "training_step_time": 9.913040399551392
    },
    {
      "epoch": 1.0227203369140625e-05,
      "model_forward_time": 3.1435112953186035,
      "step": 2681
    },
    {
      "epoch": 1.0227203369140625e-05,
      "step": 2681,
      "training_step_time": 10.278170347213745
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 3.1802618503570557,
      "step": 2682
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 2682,
      "training_step_time": 9.95340085029602
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 3.138092279434204,
      "step": 2682
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 2682,
      "training_step_time": 9.904118061065674
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 3.133384943008423,
      "step": 2682
    },
    {
      "dynamics_loss_avg": 0.06301420908421278,
      "epoch": 1.023101806640625e-05,
      "step": 2682
    },
    {
      "action_loss_avg": 0.03713901741430163,
      "epoch": 1.023101806640625e-05,
      "step": 2682
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 2682,
      "training_step_time": 9.903167724609375
    },
    {
      "epoch": 1.023101806640625e-05,
      "model_forward_time": 3.1344597339630127,
      "step": 2682
    },
    {
      "epoch": 1.023101806640625e-05,
      "step": 2682,
      "training_step_time": 10.25661563873291
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 3.1712067127227783,
      "step": 2683
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 2683,
      "training_step_time": 10.210288047790527
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 3.1335787773132324,
      "step": 2683
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 2683,
      "training_step_time": 9.89581561088562
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 2.4279253482818604,
      "step": 2683
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 2683,
      "training_step_time": 7.448308944702148
    },
    {
      "epoch": 1.0234832763671874e-05,
      "model_forward_time": 3.1300578117370605,
      "step": 2683
    },
    {
      "epoch": 1.0234832763671874e-05,
      "step": 2683,
      "training_step_time": 10.009279012680054
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 3.185323476791382,
      "step": 2684
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 2684,
      "training_step_time": 9.94586181640625
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 3.1287131309509277,
      "step": 2684
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 2684,
      "training_step_time": 10.153189182281494
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 3.1294944286346436,
      "step": 2684
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 2684,
      "training_step_time": 9.892597675323486
    },
    {
      "epoch": 1.02386474609375e-05,
      "model_forward_time": 3.134211301803589,
      "step": 2684
    },
    {
      "epoch": 1.02386474609375e-05,
      "step": 2684,
      "training_step_time": 10.027318477630615
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 3.1842284202575684,
      "step": 2685
    },
    {
      "dynamics_loss_avg": 0.08671851679682732,
      "epoch": 1.0242462158203125e-05,
      "step": 2685
    },
    {
      "action_loss_avg": 0.04450957402586937,
      "epoch": 1.0242462158203125e-05,
      "step": 2685
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 2685,
      "training_step_time": 9.954480409622192
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 3.1346895694732666,
      "step": 2685
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 2685,
      "training_step_time": 9.903440952301025
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 3.135786533355713,
      "step": 2685
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 2685,
      "training_step_time": 10.168383121490479
    },
    {
      "epoch": 1.0242462158203125e-05,
      "model_forward_time": 3.1377742290496826,
      "step": 2685
    },
    {
      "epoch": 1.0242462158203125e-05,
      "step": 2685,
      "training_step_time": 10.306364059448242
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 3.1771934032440186,
      "step": 2686
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 2686,
      "training_step_time": 9.950800657272339
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 3.1434168815612793,
      "step": 2686
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 2686,
      "training_step_time": 9.915789127349854
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 3.148393392562866,
      "step": 2686
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 2686,
      "training_step_time": 9.92439603805542
    },
    {
      "epoch": 1.024627685546875e-05,
      "model_forward_time": 3.4304752349853516,
      "step": 2686
    },
    {
      "epoch": 1.024627685546875e-05,
      "step": 2686,
      "training_step_time": 10.299815654754639
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 3.168693780899048,
      "step": 2687
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 2687,
      "training_step_time": 9.945027351379395
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 2.4396824836730957,
      "step": 2687
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 2687,
      "training_step_time": 7.468998193740845
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 2.4363250732421875,
      "step": 2687
    },
    {
      "dynamics_loss_avg": 0.06236737184226513,
      "epoch": 1.0250091552734374e-05,
      "step": 2687
    },
    {
      "action_loss_avg": 0.033275147434324026,
      "epoch": 1.0250091552734374e-05,
      "step": 2687
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 2687,
      "training_step_time": 7.465263843536377
    },
    {
      "epoch": 1.0250091552734374e-05,
      "model_forward_time": 2.439011335372925,
      "step": 2687
    },
    {
      "epoch": 1.0250091552734374e-05,
      "step": 2687,
      "training_step_time": 10.251945495605469
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 3.1484546661376953,
      "step": 2688
    },
    {
      "epoch": 1.025390625e-05,
      "step": 2688,
      "training_step_time": 9.919353485107422
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 3.38525128364563,
      "step": 2688
    },
    {
      "epoch": 1.025390625e-05,
      "step": 2688,
      "training_step_time": 10.151363849639893
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 2.432020902633667,
      "step": 2688
    },
    {
      "epoch": 1.025390625e-05,
      "step": 2688,
      "training_step_time": 7.453665494918823
    },
    {
      "epoch": 1.025390625e-05,
      "model_forward_time": 3.1305525302886963,
      "step": 2688
    },
    {
      "epoch": 1.025390625e-05,
      "step": 2688,
      "training_step_time": 10.284394264221191
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 3.173842430114746,
      "step": 2689
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 2689,
      "training_step_time": 9.938776016235352
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 3.140241861343384,
      "step": 2689
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 2689,
      "training_step_time": 9.906989812850952
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 3.414332151412964,
      "step": 2689
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 2689,
      "training_step_time": 10.186599016189575
    },
    {
      "epoch": 1.0257720947265625e-05,
      "model_forward_time": 3.1434569358825684,
      "step": 2689
    },
    {
      "epoch": 1.0257720947265625e-05,
      "step": 2689,
      "training_step_time": 10.239802122116089
    },
    {
      "epoch": 1.026153564453125e-05,
      "grad_norm": 0.5654829144477844,
      "learning_rate": 1e-05,
      "loss": 0.0975,
      "step": 2690
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 3.1713600158691406,
      "step": 2690
    },
    {
      "dynamics_loss_avg": 0.05740269832313061,
      "epoch": 1.026153564453125e-05,
      "step": 2690
    },
    {
      "action_loss_avg": 0.032379026897251605,
      "epoch": 1.026153564453125e-05,
      "step": 2690
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 2690,
      "training_step_time": 9.953936100006104
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 3.15724778175354,
      "step": 2690
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 2690,
      "training_step_time": 9.932430505752563
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 3.1339051723480225,
      "step": 2690
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 2690,
      "training_step_time": 9.904921770095825
    },
    {
      "epoch": 1.026153564453125e-05,
      "model_forward_time": 3.3818984031677246,
      "step": 2690
    },
    {
      "epoch": 1.026153564453125e-05,
      "step": 2690,
      "training_step_time": 10.317071437835693
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 2.4620048999786377,
      "step": 2691
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 2691,
      "training_step_time": 7.4887049198150635
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 3.1358375549316406,
      "step": 2691
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 2691,
      "training_step_time": 9.903862714767456
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 3.1308302879333496,
      "step": 2691
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 2691,
      "training_step_time": 9.899556875228882
    },
    {
      "epoch": 1.0265350341796874e-05,
      "model_forward_time": 3.1348373889923096,
      "step": 2691
    },
    {
      "epoch": 1.0265350341796874e-05,
      "step": 2691,
      "training_step_time": 10.288399457931519
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 3.446251630783081,
      "step": 2692
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 2692,
      "training_step_time": 10.219528436660767
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 3.1442224979400635,
      "step": 2692
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 2692,
      "training_step_time": 9.916187524795532
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 3.138195037841797,
      "step": 2692
    },
    {
      "dynamics_loss_avg": 0.06717156507074833,
      "epoch": 1.02691650390625e-05,
      "step": 2692
    },
    {
      "action_loss_avg": 0.03842817088589072,
      "epoch": 1.02691650390625e-05,
      "step": 2692
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 2692,
      "training_step_time": 9.912198781967163
    },
    {
      "epoch": 1.02691650390625e-05,
      "model_forward_time": 3.147002935409546,
      "step": 2692
    },
    {
      "epoch": 1.02691650390625e-05,
      "step": 2692,
      "training_step_time": 10.260789394378662
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 3.192244052886963,
      "step": 2693
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 2693,
      "training_step_time": 9.965686082839966
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 3.4224853515625,
      "step": 2693
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 2693,
      "training_step_time": 10.192274332046509
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 3.139516592025757,
      "step": 2693
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 2693,
      "training_step_time": 9.907065629959106
    },
    {
      "epoch": 1.0272979736328125e-05,
      "model_forward_time": 3.1415326595306396,
      "step": 2693
    },
    {
      "epoch": 1.0272979736328125e-05,
      "step": 2693,
      "training_step_time": 10.285373210906982
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 3.1728827953338623,
      "step": 2694
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 2694,
      "training_step_time": 9.938511848449707
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 2.435051202774048,
      "step": 2694
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 2694,
      "training_step_time": 7.456202983856201
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 3.3897221088409424,
      "step": 2694
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 2694,
      "training_step_time": 10.161397457122803
    },
    {
      "epoch": 1.027679443359375e-05,
      "model_forward_time": 3.141561508178711,
      "step": 2694
    },
    {
      "epoch": 1.027679443359375e-05,
      "step": 2694,
      "training_step_time": 10.2755708694458
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 3.1856346130371094,
      "step": 2695
    },
    {
      "dynamics_loss_avg": 0.07465599142014981,
      "epoch": 1.0280609130859374e-05,
      "step": 2695
    },
    {
      "action_loss_avg": 0.03442593775689602,
      "epoch": 1.0280609130859374e-05,
      "step": 2695
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 2695,
      "training_step_time": 9.958309888839722
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 3.1368601322174072,
      "step": 2695
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 2695,
      "training_step_time": 9.907167911529541
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 3.1414453983306885,
      "step": 2695
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 2695,
      "training_step_time": 10.18777346611023
    },
    {
      "epoch": 1.0280609130859374e-05,
      "model_forward_time": 3.1408207416534424,
      "step": 2695
    },
    {
      "epoch": 1.0280609130859374e-05,
      "step": 2695,
      "training_step_time": 10.29476547241211
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 3.1776645183563232,
      "step": 2696
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 2696,
      "training_step_time": 9.952152490615845
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 3.13944411277771,
      "step": 2696
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 2696,
      "training_step_time": 9.912377834320068
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 3.138596296310425,
      "step": 2696
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 2696,
      "training_step_time": 9.91146445274353
    },
    {
      "epoch": 1.0284423828125e-05,
      "model_forward_time": 3.138373851776123,
      "step": 2696
    },
    {
      "epoch": 1.0284423828125e-05,
      "step": 2696,
      "training_step_time": 10.317213535308838
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 3.163262367248535,
      "step": 2697
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 2697,
      "training_step_time": 9.94029974937439
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 3.1381306648254395,
      "step": 2697
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 2697,
      "training_step_time": 9.90586233139038
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 3.1369338035583496,
      "step": 2697
    },
    {
      "dynamics_loss_avg": 0.07640900760889054,
      "epoch": 1.0288238525390625e-05,
      "step": 2697
    },
    {
      "action_loss_avg": 0.04163590306416154,
      "epoch": 1.0288238525390625e-05,
      "step": 2697
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 2697,
      "training_step_time": 9.906030893325806
    },
    {
      "epoch": 1.0288238525390625e-05,
      "model_forward_time": 3.1317219734191895,
      "step": 2697
    },
    {
      "epoch": 1.0288238525390625e-05,
      "step": 2697,
      "training_step_time": 10.02785038948059
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 2.7536206245422363,
      "step": 2698
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 2698,
      "training_step_time": 7.777434825897217
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 3.1356863975524902,
      "step": 2698
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 2698,
      "training_step_time": 9.898765802383423
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 3.133859872817993,
      "step": 2698
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 2698,
      "training_step_time": 9.900146007537842
    },
    {
      "epoch": 1.029205322265625e-05,
      "model_forward_time": 3.1338441371917725,
      "step": 2698
    },
    {
      "epoch": 1.029205322265625e-05,
      "step": 2698,
      "training_step_time": 10.236744165420532
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 3.1710045337677,
      "step": 2699
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 2699,
      "training_step_time": 9.943254470825195
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 2.4343652725219727,
      "step": 2699
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 2699,
      "training_step_time": 7.461237907409668
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 3.411515712738037,
      "step": 2699
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 2699,
      "training_step_time": 10.180088758468628
    },
    {
      "epoch": 1.0295867919921874e-05,
      "model_forward_time": 3.1341991424560547,
      "step": 2699
    },
    {
      "epoch": 1.0295867919921874e-05,
      "step": 2699,
      "training_step_time": 10.260993719100952
    },
    {
      "epoch": 1.02996826171875e-05,
      "grad_norm": 0.4034816324710846,
      "learning_rate": 1e-05,
      "loss": 0.103,
      "step": 2700
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 2.4794137477874756,
      "step": 2700
    },
    {
      "dynamics_loss_avg": 0.06306995879858732,
      "epoch": 1.02996826171875e-05,
      "step": 2700
    },
    {
      "action_loss_avg": 0.037888207007199526,
      "epoch": 1.02996826171875e-05,
      "step": 2700
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 2700,
      "training_step_time": 7.524951457977295
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 3.1419146060943604,
      "step": 2700
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 2700,
      "training_step_time": 9.906085014343262
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 3.138751268386841,
      "step": 2700
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 2700,
      "training_step_time": 10.171076536178589
    },
    {
      "epoch": 1.02996826171875e-05,
      "model_forward_time": 3.136309862136841,
      "step": 2700
    },
    {
      "epoch": 1.02996826171875e-05,
      "step": 2700,
      "training_step_time": 10.017060041427612
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 3.1576168537139893,
      "step": 2701
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 2701,
      "training_step_time": 9.934198379516602
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 3.1416897773742676,
      "step": 2701
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 2701,
      "training_step_time": 9.913142442703247
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 2.441265821456909,
      "step": 2701
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 2701,
      "training_step_time": 7.47454309463501
    },
    {
      "epoch": 1.0303497314453125e-05,
      "model_forward_time": 2.4358651638031006,
      "step": 2701
    },
    {
      "epoch": 1.0303497314453125e-05,
      "step": 2701,
      "training_step_time": 10.315124034881592
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 3.4585039615631104,
      "step": 2702
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 2702,
      "training_step_time": 10.231144666671753
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 3.1435816287994385,
      "step": 2702
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 2702,
      "training_step_time": 9.916072368621826
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 3.1391866207122803,
      "step": 2702
    },
    {
      "dynamics_loss_avg": 0.06377825327217579,
      "epoch": 1.030731201171875e-05,
      "step": 2702
    },
    {
      "action_loss_avg": 0.045129315368831156,
      "epoch": 1.030731201171875e-05,
      "step": 2702
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 2702,
      "training_step_time": 9.917143106460571
    },
    {
      "epoch": 1.030731201171875e-05,
      "model_forward_time": 3.1441292762756348,
      "step": 2702
    },
    {
      "epoch": 1.030731201171875e-05,
      "step": 2702,
      "training_step_time": 10.299043893814087
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 3.1890413761138916,
      "step": 2703
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 2703,
      "training_step_time": 10.246670484542847
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 2.4378652572631836,
      "step": 2703
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 2703,
      "training_step_time": 7.470586776733398
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 3.137711524963379,
      "step": 2703
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 2703,
      "training_step_time": 9.911094903945923
    },
    {
      "epoch": 1.0311126708984374e-05,
      "model_forward_time": 2.432960271835327,
      "step": 2703
    },
    {
      "epoch": 1.0311126708984374e-05,
      "step": 2703,
      "training_step_time": 10.018909931182861
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 3.1760284900665283,
      "step": 2704
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 2704,
      "training_step_time": 9.945396423339844
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 3.1360080242156982,
      "step": 2704
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 2704,
      "training_step_time": 9.913666009902954
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 3.400193929672241,
      "step": 2704
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 2704,
      "training_step_time": 10.169895648956299
    },
    {
      "epoch": 1.031494140625e-05,
      "model_forward_time": 3.141024589538574,
      "step": 2704
    },
    {
      "epoch": 1.031494140625e-05,
      "step": 2704,
      "training_step_time": 10.288641452789307
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 3.182762622833252,
      "step": 2705
    },
    {
      "dynamics_loss_avg": 0.06898230975493788,
      "epoch": 1.0318756103515625e-05,
      "step": 2705
    },
    {
      "action_loss_avg": 0.03220749660395086,
      "epoch": 1.0318756103515625e-05,
      "step": 2705
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 2705,
      "training_step_time": 9.961400032043457
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 3.1404805183410645,
      "step": 2705
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 2705,
      "training_step_time": 9.921929121017456
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 3.145212411880493,
      "step": 2705
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 2705,
      "training_step_time": 10.199683904647827
    },
    {
      "epoch": 1.0318756103515625e-05,
      "model_forward_time": 3.1496646404266357,
      "step": 2705
    },
    {
      "epoch": 1.0318756103515625e-05,
      "step": 2705,
      "training_step_time": 10.326129913330078
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 3.1906864643096924,
      "step": 2706
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 2706,
      "training_step_time": 9.976826667785645
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 3.1450371742248535,
      "step": 2706
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 2706,
      "training_step_time": 9.928963422775269
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 3.1467976570129395,
      "step": 2706
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 2706,
      "training_step_time": 9.928570747375488
    },
    {
      "epoch": 1.032257080078125e-05,
      "model_forward_time": 2.7246036529541016,
      "step": 2706
    },
    {
      "epoch": 1.032257080078125e-05,
      "step": 2706,
      "training_step_time": 10.309183597564697
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 2.4699180126190186,
      "step": 2707
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 2707,
      "training_step_time": 7.503194808959961
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 2.4394681453704834,
      "step": 2707
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 2707,
      "training_step_time": 7.471080303192139
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 3.1397738456726074,
      "step": 2707
    },
    {
      "dynamics_loss_avg": 0.0561885055154562,
      "epoch": 1.0326385498046874e-05,
      "step": 2707
    },
    {
      "action_loss_avg": 0.025492695439606906,
      "epoch": 1.0326385498046874e-05,
      "step": 2707
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 2707,
      "training_step_time": 9.913115739822388
    },
    {
      "epoch": 1.0326385498046874e-05,
      "model_forward_time": 3.13319730758667,
      "step": 2707
    },
    {
      "epoch": 1.0326385498046874e-05,
      "step": 2707,
      "training_step_time": 10.02124547958374
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 3.168053150177002,
      "step": 2708
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 2708,
      "training_step_time": 9.941654920578003
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 3.39851713180542,
      "step": 2708
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 2708,
      "training_step_time": 10.164527416229248
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 3.139683723449707,
      "step": 2708
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 2708,
      "training_step_time": 9.908252477645874
    },
    {
      "epoch": 1.03302001953125e-05,
      "model_forward_time": 3.13427734375,
      "step": 2708
    },
    {
      "epoch": 1.03302001953125e-05,
      "step": 2708,
      "training_step_time": 10.297163486480713
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 3.190776824951172,
      "step": 2709
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 2709,
      "training_step_time": 9.96024227142334
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 2.4341533184051514,
      "step": 2709
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 2709,
      "training_step_time": 7.459249496459961
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 3.4048397541046143,
      "step": 2709
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 2709,
      "training_step_time": 10.17177438735962
    },
    {
      "epoch": 1.0334014892578125e-05,
      "model_forward_time": 3.1275742053985596,
      "step": 2709
    },
    {
      "epoch": 1.0334014892578125e-05,
      "step": 2709,
      "training_step_time": 10.242405891418457
    },
    {
      "epoch": 1.033782958984375e-05,
      "grad_norm": 0.8557394742965698,
      "learning_rate": 1e-05,
      "loss": 0.1033,
      "step": 2710
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 3.1667304039001465,
      "step": 2710
    },
    {
      "dynamics_loss_avg": 0.05838957950472832,
      "epoch": 1.033782958984375e-05,
      "step": 2710
    },
    {
      "action_loss_avg": 0.04068978745490313,
      "epoch": 1.033782958984375e-05,
      "step": 2710
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 2710,
      "training_step_time": 9.941174507141113
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 3.1278302669525146,
      "step": 2710
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 2710,
      "training_step_time": 9.898222208023071
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 3.128737211227417,
      "step": 2710
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 2710,
      "training_step_time": 9.90062141418457
    },
    {
      "epoch": 1.033782958984375e-05,
      "model_forward_time": 2.672412633895874,
      "step": 2710
    },
    {
      "epoch": 1.033782958984375e-05,
      "step": 2710,
      "training_step_time": 10.27004075050354
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 3.154261827468872,
      "step": 2711
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 2711,
      "training_step_time": 9.927722692489624
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 3.1312832832336426,
      "step": 2711
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 2711,
      "training_step_time": 9.905510902404785
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 3.129781484603882,
      "step": 2711
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 2711,
      "training_step_time": 9.90624451637268
    },
    {
      "epoch": 1.0341644287109374e-05,
      "model_forward_time": 3.1344900131225586,
      "step": 2711
    },
    {
      "epoch": 1.0341644287109374e-05,
      "step": 2711,
      "training_step_time": 10.274429559707642
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 3.4500832557678223,
      "step": 2712
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 2712,
      "training_step_time": 10.223015308380127
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 3.134392261505127,
      "step": 2712
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 2712,
      "training_step_time": 9.9207181930542
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 3.13692045211792,
      "step": 2712
    },
    {
      "dynamics_loss_avg": 0.05939977392554283,
      "epoch": 1.0345458984375e-05,
      "step": 2712
    },
    {
      "action_loss_avg": 0.030163428746163845,
      "epoch": 1.0345458984375e-05,
      "step": 2712
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 2712,
      "training_step_time": 9.920870542526245
    },
    {
      "epoch": 1.0345458984375e-05,
      "model_forward_time": 3.13651180267334,
      "step": 2712
    },
    {
      "epoch": 1.0345458984375e-05,
      "step": 2712,
      "training_step_time": 10.321941137313843
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 2.473750591278076,
      "step": 2713
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 2713,
      "training_step_time": 7.779005527496338
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 3.1374526023864746,
      "step": 2713
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 2713,
      "training_step_time": 9.908170223236084
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 3.1347031593322754,
      "step": 2713
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 2713,
      "training_step_time": 9.909948587417603
    },
    {
      "epoch": 1.0349273681640625e-05,
      "model_forward_time": 2.4350457191467285,
      "step": 2713
    },
    {
      "epoch": 1.0349273681640625e-05,
      "step": 2713,
      "training_step_time": 10.3278226852417
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 3.1838550567626953,
      "step": 2714
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 2714,
      "training_step_time": 9.960795879364014
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 3.1361331939697266,
      "step": 2714
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 2714,
      "training_step_time": 10.189410924911499
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 3.1390187740325928,
      "step": 2714
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 2714,
      "training_step_time": 9.912614345550537
    },
    {
      "epoch": 1.035308837890625e-05,
      "model_forward_time": 3.134507656097412,
      "step": 2714
    },
    {
      "epoch": 1.035308837890625e-05,
      "step": 2714,
      "training_step_time": 10.315951347351074
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 3.1678011417388916,
      "step": 2715
    },
    {
      "dynamics_loss_avg": 0.054024343751370905,
      "epoch": 1.0356903076171874e-05,
      "step": 2715
    },
    {
      "action_loss_avg": 0.026368970470502972,
      "epoch": 1.0356903076171874e-05,
      "step": 2715
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 2715,
      "training_step_time": 9.934690237045288
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 3.128814458847046,
      "step": 2715
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 2715,
      "training_step_time": 9.889304637908936
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 3.3904240131378174,
      "step": 2715
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 2715,
      "training_step_time": 10.153140544891357
    },
    {
      "epoch": 1.0356903076171874e-05,
      "model_forward_time": 3.135486602783203,
      "step": 2715
    },
    {
      "epoch": 1.0356903076171874e-05,
      "step": 2715,
      "training_step_time": 10.274875164031982
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 3.1704061031341553,
      "step": 2716
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 2716,
      "training_step_time": 9.937469720840454
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 3.130570650100708,
      "step": 2716
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 2716,
      "training_step_time": 9.894380569458008
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 2.4304261207580566,
      "step": 2716
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 2716,
      "training_step_time": 7.454945802688599
    },
    {
      "epoch": 1.03607177734375e-05,
      "model_forward_time": 3.1278510093688965,
      "step": 2716
    },
    {
      "epoch": 1.03607177734375e-05,
      "step": 2716,
      "training_step_time": 10.27979850769043
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 3.174879550933838,
      "step": 2717
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 2717,
      "training_step_time": 9.951406478881836
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 3.136544704437256,
      "step": 2717
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 2717,
      "training_step_time": 9.912891864776611
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 3.139037609100342,
      "step": 2717
    },
    {
      "dynamics_loss_avg": 0.07175306975841522,
      "epoch": 1.0364532470703125e-05,
      "step": 2717
    },
    {
      "action_loss_avg": 0.0372142544016242,
      "epoch": 1.0364532470703125e-05,
      "step": 2717
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 2717,
      "training_step_time": 9.916969299316406
    },
    {
      "epoch": 1.0364532470703125e-05,
      "model_forward_time": 3.137007474899292,
      "step": 2717
    },
    {
      "epoch": 1.0364532470703125e-05,
      "step": 2717,
      "training_step_time": 10.03226637840271
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 3.452770709991455,
      "step": 2718
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 2718,
      "training_step_time": 10.228777885437012
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 3.1377360820770264,
      "step": 2718
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 2718,
      "training_step_time": 9.909746646881104
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 3.136608600616455,
      "step": 2718
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 2718,
      "training_step_time": 9.910622596740723
    },
    {
      "epoch": 1.036834716796875e-05,
      "model_forward_time": 3.131230592727661,
      "step": 2718
    },
    {
      "epoch": 1.036834716796875e-05,
      "step": 2718,
      "training_step_time": 10.253118515014648
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 3.169538974761963,
      "step": 2719
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 2719,
      "training_step_time": 9.941563606262207
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 3.391876220703125,
      "step": 2719
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 2719,
      "training_step_time": 10.16006088256836
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 3.1358673572540283,
      "step": 2719
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 2719,
      "training_step_time": 9.906543493270874
    },
    {
      "epoch": 1.0372161865234374e-05,
      "model_forward_time": 2.43174147605896,
      "step": 2719
    },
    {
      "epoch": 1.0372161865234374e-05,
      "step": 2719,
      "training_step_time": 10.023288488388062
    },
    {
      "epoch": 1.03759765625e-05,
      "grad_norm": 0.4294630289077759,
      "learning_rate": 1e-05,
      "loss": 0.1007,
      "step": 2720
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 3.1703803539276123,
      "step": 2720
    },
    {
      "dynamics_loss_avg": 0.0591392045840621,
      "epoch": 1.03759765625e-05,
      "step": 2720
    },
    {
      "action_loss_avg": 0.025296660792082547,
      "epoch": 1.03759765625e-05,
      "step": 2720
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 2720,
      "training_step_time": 9.94599199295044
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 3.1361100673675537,
      "step": 2720
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 2720,
      "training_step_time": 9.90454363822937
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 3.134387969970703,
      "step": 2720
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 2720,
      "training_step_time": 10.17839765548706
    },
    {
      "epoch": 1.03759765625e-05,
      "model_forward_time": 3.13128662109375,
      "step": 2720
    },
    {
      "epoch": 1.03759765625e-05,
      "step": 2720,
      "training_step_time": 10.28989863395691
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 3.1881628036499023,
      "step": 2721
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 2721,
      "training_step_time": 9.955689907073975
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 3.1388683319091797,
      "step": 2721
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 2721,
      "training_step_time": 9.906954765319824
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 3.1341583728790283,
      "step": 2721
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 2721,
      "training_step_time": 9.901363372802734
    },
    {
      "epoch": 1.0379791259765625e-05,
      "model_forward_time": 3.1366732120513916,
      "step": 2721
    },
    {
      "epoch": 1.0379791259765625e-05,
      "step": 2721,
      "training_step_time": 10.287374496459961
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 3.1703388690948486,
      "step": 2722
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 2722,
      "training_step_time": 9.93932557106018
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 2.43418025970459,
      "step": 2722
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 2722,
      "training_step_time": 7.45833158493042
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 3.131011486053467,
      "step": 2722
    },
    {
      "dynamics_loss_avg": 0.06364039219915867,
      "epoch": 1.038360595703125e-05,
      "step": 2722
    },
    {
      "action_loss_avg": 0.04356779046356678,
      "epoch": 1.038360595703125e-05,
      "step": 2722
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 2722,
      "training_step_time": 9.899477481842041
    },
    {
      "epoch": 1.038360595703125e-05,
      "model_forward_time": 3.1340558528900146,
      "step": 2722
    },
    {
      "epoch": 1.038360595703125e-05,
      "step": 2722,
      "training_step_time": 10.010370254516602
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 3.1684656143188477,
      "step": 2723
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 2723,
      "training_step_time": 10.209302425384521
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 3.137880325317383,
      "step": 2723
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 2723,
      "training_step_time": 9.912060976028442
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 3.1397931575775146,
      "step": 2723
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 2723,
      "training_step_time": 9.913002014160156
    },
    {
      "epoch": 1.0387420654296874e-05,
      "model_forward_time": 2.4368648529052734,
      "step": 2723
    },
    {
      "epoch": 1.0387420654296874e-05,
      "step": 2723,
      "training_step_time": 10.289512157440186
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 3.1739726066589355,
      "step": 2724
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 2724,
      "training_step_time": 9.949808597564697
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 3.422243118286133,
      "step": 2724
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 2724,
      "training_step_time": 10.198719024658203
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 3.1380615234375,
      "step": 2724
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 2724,
      "training_step_time": 9.91577935218811
    },
    {
      "epoch": 1.03912353515625e-05,
      "model_forward_time": 3.1444010734558105,
      "step": 2724
    },
    {
      "epoch": 1.03912353515625e-05,
      "step": 2724,
      "training_step_time": 10.314385175704956
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 3.187790870666504,
      "step": 2725
    },
    {
      "dynamics_loss_avg": 0.06655976083129644,
      "epoch": 1.0395050048828125e-05,
      "step": 2725
    },
    {
      "action_loss_avg": 0.03681974494829774,
      "epoch": 1.0395050048828125e-05,
      "step": 2725
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 2725,
      "training_step_time": 9.967442035675049
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 3.137406587600708,
      "step": 2725
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 2725,
      "training_step_time": 9.913285970687866
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 3.147003650665283,
      "step": 2725
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 2725,
      "training_step_time": 10.187523126602173
    },
    {
      "epoch": 1.0395050048828125e-05,
      "model_forward_time": 3.1440320014953613,
      "step": 2725
    },
    {
      "epoch": 1.0395050048828125e-05,
      "step": 2725,
      "training_step_time": 10.25499176979065
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 3.1820428371429443,
      "step": 2726
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 2726,
      "training_step_time": 9.955415487289429
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 3.1352734565734863,
      "step": 2726
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 2726,
      "training_step_time": 9.903311729431152
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 3.137521743774414,
      "step": 2726
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 2726,
      "training_step_time": 9.913579940795898
    },
    {
      "epoch": 1.039886474609375e-05,
      "model_forward_time": 3.4156150817871094,
      "step": 2726
    },
    {
      "epoch": 1.039886474609375e-05,
      "step": 2726,
      "training_step_time": 10.30296778678894
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 3.1845335960388184,
      "step": 2727
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 2727,
      "training_step_time": 9.955678462982178
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 3.132356643676758,
      "step": 2727
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 2727,
      "training_step_time": 9.900765419006348
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 3.1398518085479736,
      "step": 2727
    },
    {
      "dynamics_loss_avg": 0.0699751403182745,
      "epoch": 1.0402679443359374e-05,
      "step": 2727
    },
    {
      "action_loss_avg": 0.03421547524631023,
      "epoch": 1.0402679443359374e-05,
      "step": 2727
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 2727,
      "training_step_time": 9.90938687324524
    },
    {
      "epoch": 1.0402679443359374e-05,
      "model_forward_time": 3.1334726810455322,
      "step": 2727
    },
    {
      "epoch": 1.0402679443359374e-05,
      "step": 2727,
      "training_step_time": 10.018161535263062
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 2.446357011795044,
      "step": 2728
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 2728,
      "training_step_time": 7.74004340171814
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 3.1351613998413086,
      "step": 2728
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 2728,
      "training_step_time": 9.906617879867554
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 3.1442816257476807,
      "step": 2728
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 2728,
      "training_step_time": 9.916489124298096
    },
    {
      "epoch": 1.0406494140625e-05,
      "model_forward_time": 3.1406872272491455,
      "step": 2728
    },
    {
      "epoch": 1.0406494140625e-05,
      "step": 2728,
      "training_step_time": 10.288520336151123
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 2.456050157546997,
      "step": 2729
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 2729,
      "training_step_time": 7.4843409061431885
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 3.410024881362915,
      "step": 2729
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 2729,
      "training_step_time": 10.178448915481567
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 3.1463332176208496,
      "step": 2729
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 2729,
      "training_step_time": 9.918127298355103
    },
    {
      "epoch": 1.0410308837890625e-05,
      "model_forward_time": 3.1410160064697266,
      "step": 2729
    },
    {
      "epoch": 1.0410308837890625e-05,
      "step": 2729,
      "training_step_time": 10.242472410202026
    },
    {
      "epoch": 1.041412353515625e-05,
      "grad_norm": 0.5936000943183899,
      "learning_rate": 1e-05,
      "loss": 0.0994,
      "step": 2730
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 3.1502623558044434,
      "step": 2730
    },
    {
      "dynamics_loss_avg": 0.06985066868364811,
      "epoch": 1.041412353515625e-05,
      "step": 2730
    },
    {
      "action_loss_avg": 0.04401792362332344,
      "epoch": 1.041412353515625e-05,
      "step": 2730
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 2730,
      "training_step_time": 9.924756050109863
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 3.1431596279144287,
      "step": 2730
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 2730,
      "training_step_time": 9.914198637008667
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 3.136542797088623,
      "step": 2730
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 2730,
      "training_step_time": 10.174390316009521
    },
    {
      "epoch": 1.041412353515625e-05,
      "model_forward_time": 3.143752336502075,
      "step": 2730
    },
    {
      "epoch": 1.041412353515625e-05,
      "step": 2730,
      "training_step_time": 10.314911127090454
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 3.1540606021881104,
      "step": 2731
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 2731,
      "training_step_time": 9.927360773086548
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 3.1418004035949707,
      "step": 2731
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 2731,
      "training_step_time": 9.916159868240356
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 3.141681432723999,
      "step": 2731
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 2731,
      "training_step_time": 9.917160511016846
    },
    {
      "epoch": 1.0417938232421875e-05,
      "model_forward_time": 3.4178977012634277,
      "step": 2731
    },
    {
      "epoch": 1.0417938232421875e-05,
      "step": 2731,
      "training_step_time": 10.302392959594727
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 3.1544601917266846,
      "step": 2732
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 2732,
      "training_step_time": 9.928364038467407
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 3.1427321434020996,
      "step": 2732
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 2732,
      "training_step_time": 9.917998313903809
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 3.1419830322265625,
      "step": 2732
    },
    {
      "dynamics_loss_avg": 0.05749768018722534,
      "epoch": 1.04217529296875e-05,
      "step": 2732
    },
    {
      "action_loss_avg": 0.032059756200760604,
      "epoch": 1.04217529296875e-05,
      "step": 2732
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 2732,
      "training_step_time": 9.920657396316528
    },
    {
      "epoch": 1.04217529296875e-05,
      "model_forward_time": 3.145176410675049,
      "step": 2732
    },
    {
      "epoch": 1.04217529296875e-05,
      "step": 2732,
      "training_step_time": 10.274883270263672
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 2.750782012939453,
      "step": 2733
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 2733,
      "training_step_time": 7.779214143753052
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 3.140300750732422,
      "step": 2733
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 2733,
      "training_step_time": 9.911180257797241
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 3.139481544494629,
      "step": 2733
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 2733,
      "training_step_time": 9.906845092773438
    },
    {
      "epoch": 1.0425567626953125e-05,
      "model_forward_time": 3.1461021900177,
      "step": 2733
    },
    {
      "epoch": 1.0425567626953125e-05,
      "step": 2733,
      "training_step_time": 10.247254133224487
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 2.452030658721924,
      "step": 2734
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 2734,
      "training_step_time": 7.481787919998169
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 3.137695550918579,
      "step": 2734
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 2734,
      "training_step_time": 10.179177045822144
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 3.139883041381836,
      "step": 2734
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 2734,
      "training_step_time": 9.912399291992188
    },
    {
      "epoch": 1.042938232421875e-05,
      "model_forward_time": 3.1418697834014893,
      "step": 2734
    },
    {
      "epoch": 1.042938232421875e-05,
      "step": 2734,
      "training_step_time": 10.252096891403198
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 3.158982992172241,
      "step": 2735
    },
    {
      "dynamics_loss_avg": 0.06767785735428333,
      "epoch": 1.0433197021484375e-05,
      "step": 2735
    },
    {
      "action_loss_avg": 0.03143650009296835,
      "epoch": 1.0433197021484375e-05,
      "step": 2735
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 2735,
      "training_step_time": 9.938632011413574
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 2.443767786026001,
      "step": 2735
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 2735,
      "training_step_time": 7.473593473434448
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 3.402005910873413,
      "step": 2735
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 2735,
      "training_step_time": 10.17311692237854
    },
    {
      "epoch": 1.0433197021484375e-05,
      "model_forward_time": 3.1458873748779297,
      "step": 2735
    },
    {
      "epoch": 1.0433197021484375e-05,
      "step": 2735,
      "training_step_time": 10.345043897628784
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 3.1550872325897217,
      "step": 2736
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 2736,
      "training_step_time": 9.928879261016846
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 3.142000913619995,
      "step": 2736
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 2736,
      "training_step_time": 9.91095232963562
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 2.4390392303466797,
      "step": 2736
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 2736,
      "training_step_time": 7.46931791305542
    },
    {
      "epoch": 1.043701171875e-05,
      "model_forward_time": 3.141021251678467,
      "step": 2736
    },
    {
      "epoch": 1.043701171875e-05,
      "step": 2736,
      "training_step_time": 10.32674241065979
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 2.7670209407806396,
      "step": 2737
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 2737,
      "training_step_time": 7.801701068878174
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 3.138873338699341,
      "step": 2737
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 2737,
      "training_step_time": 9.908735036849976
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 3.139888048171997,
      "step": 2737
    },
    {
      "dynamics_loss_avg": 0.08182382071390748,
      "epoch": 1.0440826416015625e-05,
      "step": 2737
    },
    {
      "action_loss_avg": 0.04226041263900697,
      "epoch": 1.0440826416015625e-05,
      "step": 2737
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 2737,
      "training_step_time": 9.911888122558594
    },
    {
      "epoch": 1.0440826416015625e-05,
      "model_forward_time": 3.1409912109375,
      "step": 2737
    },
    {
      "epoch": 1.0440826416015625e-05,
      "step": 2737,
      "training_step_time": 10.032193422317505
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 3.4322266578674316,
      "step": 2738
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 2738,
      "training_step_time": 10.222121953964233
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 3.13517427444458,
      "step": 2738
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 2738,
      "training_step_time": 9.910356521606445
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 3.1400790214538574,
      "step": 2738
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 2738,
      "training_step_time": 9.915967226028442
    },
    {
      "epoch": 1.044464111328125e-05,
      "model_forward_time": 3.139543056488037,
      "step": 2738
    },
    {
      "epoch": 1.044464111328125e-05,
      "step": 2738,
      "training_step_time": 10.288339138031006
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 3.1425886154174805,
      "step": 2739
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 2739,
      "training_step_time": 9.92033314704895
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 3.137479305267334,
      "step": 2739
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 2739,
      "training_step_time": 9.912426710128784
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 3.403412103652954,
      "step": 2739
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 2739,
      "training_step_time": 10.168592691421509
    },
    {
      "epoch": 1.0448455810546875e-05,
      "model_forward_time": 3.1328396797180176,
      "step": 2739
    },
    {
      "epoch": 1.0448455810546875e-05,
      "step": 2739,
      "training_step_time": 10.008219242095947
    },
    {
      "epoch": 1.04522705078125e-05,
      "grad_norm": 0.5217394828796387,
      "learning_rate": 1e-05,
      "loss": 0.1002,
      "step": 2740
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 3.1386241912841797,
      "step": 2740
    },
    {
      "dynamics_loss_avg": 0.05254545956850052,
      "epoch": 1.04522705078125e-05,
      "step": 2740
    },
    {
      "action_loss_avg": 0.02382979020476341,
      "epoch": 1.04522705078125e-05,
      "step": 2740
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 2740,
      "training_step_time": 9.909942626953125
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 3.131195068359375,
      "step": 2740
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 2740,
      "training_step_time": 9.889649629592896
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 3.4106240272521973,
      "step": 2740
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 2740,
      "training_step_time": 10.179998397827148
    },
    {
      "epoch": 1.04522705078125e-05,
      "model_forward_time": 3.13332462310791,
      "step": 2740
    },
    {
      "epoch": 1.04522705078125e-05,
      "step": 2740,
      "training_step_time": 10.007887840270996
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 3.153724193572998,
      "step": 2741
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 2741,
      "training_step_time": 9.925437688827515
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 3.136718273162842,
      "step": 2741
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 2741,
      "training_step_time": 9.906713962554932
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 2.4334802627563477,
      "step": 2741
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 2741,
      "training_step_time": 7.460882902145386
    },
    {
      "epoch": 1.0456085205078125e-05,
      "model_forward_time": 3.3921287059783936,
      "step": 2741
    },
    {
      "epoch": 1.0456085205078125e-05,
      "step": 2741,
      "training_step_time": 10.278642416000366
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 3.1486878395080566,
      "step": 2742
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 2742,
      "training_step_time": 9.918167114257812
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 2.43404221534729,
      "step": 2742
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 2742,
      "training_step_time": 7.4574058055877686
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 3.1300482749938965,
      "step": 2742
    },
    {
      "dynamics_loss_avg": 0.05584405716508627,
      "epoch": 1.045989990234375e-05,
      "step": 2742
    },
    {
      "action_loss_avg": 0.031218554731458424,
      "epoch": 1.045989990234375e-05,
      "step": 2742
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 2742,
      "training_step_time": 9.898295164108276
    },
    {
      "epoch": 1.045989990234375e-05,
      "model_forward_time": 3.133483648300171,
      "step": 2742
    },
    {
      "epoch": 1.045989990234375e-05,
      "step": 2742,
      "training_step_time": 10.013428926467896
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 3.1440317630767822,
      "step": 2743
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 2743,
      "training_step_time": 10.174001932144165
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 3.133483409881592,
      "step": 2743
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 2743,
      "training_step_time": 9.896677255630493
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 3.133725166320801,
      "step": 2743
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 2743,
      "training_step_time": 9.902925252914429
    },
    {
      "epoch": 1.0463714599609375e-05,
      "model_forward_time": 3.1367321014404297,
      "step": 2743
    },
    {
      "epoch": 1.0463714599609375e-05,
      "step": 2743,
      "training_step_time": 10.256252527236938
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 3.1670494079589844,
      "step": 2744
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 2744,
      "training_step_time": 9.944273471832275
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 3.4301090240478516,
      "step": 2744
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 2744,
      "training_step_time": 10.197850227355957
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 3.1403286457061768,
      "step": 2744
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 2744,
      "training_step_time": 9.90677285194397
    },
    {
      "epoch": 1.0467529296875e-05,
      "model_forward_time": 2.4370384216308594,
      "step": 2744
    },
    {
      "epoch": 1.0467529296875e-05,
      "step": 2744,
      "training_step_time": 10.315273761749268
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 3.1546413898468018,
      "step": 2745
    },
    {
      "dynamics_loss_avg": 0.0682195357978344,
      "epoch": 1.0471343994140625e-05,
      "step": 2745
    },
    {
      "action_loss_avg": 0.03887447537854314,
      "epoch": 1.0471343994140625e-05,
      "step": 2745
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 2745,
      "training_step_time": 9.931269407272339
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 3.133561611175537,
      "step": 2745
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 2745,
      "training_step_time": 9.89672565460205
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 3.1288137435913086,
      "step": 2745
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 2745,
      "training_step_time": 10.202335834503174
    },
    {
      "epoch": 1.0471343994140625e-05,
      "model_forward_time": 3.132514238357544,
      "step": 2745
    },
    {
      "epoch": 1.0471343994140625e-05,
      "step": 2745,
      "training_step_time": 10.017405271530151
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 2.446702480316162,
      "step": 2746
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 2746,
      "training_step_time": 7.473106861114502
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 3.1315715312957764,
      "step": 2746
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 2746,
      "training_step_time": 9.898412942886353
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 3.1403863430023193,
      "step": 2746
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 2746,
      "training_step_time": 9.910404920578003
    },
    {
      "epoch": 1.047515869140625e-05,
      "model_forward_time": 3.419161796569824,
      "step": 2746
    },
    {
      "epoch": 1.047515869140625e-05,
      "step": 2746,
      "training_step_time": 10.287964344024658
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 2.4453327655792236,
      "step": 2747
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 2747,
      "training_step_time": 7.474099159240723
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 3.1386613845825195,
      "step": 2747
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 2747,
      "training_step_time": 9.920053243637085
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 3.138678550720215,
      "step": 2747
    },
    {
      "dynamics_loss_avg": 0.0857459980994463,
      "epoch": 1.0478973388671875e-05,
      "step": 2747
    },
    {
      "action_loss_avg": 0.04798930827528238,
      "epoch": 1.0478973388671875e-05,
      "step": 2747
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 2747,
      "training_step_time": 9.915347337722778
    },
    {
      "epoch": 1.0478973388671875e-05,
      "model_forward_time": 3.141509532928467,
      "step": 2747
    },
    {
      "epoch": 1.0478973388671875e-05,
      "step": 2747,
      "training_step_time": 10.324488401412964
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 2.4439454078674316,
      "step": 2748
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 2748,
      "training_step_time": 7.4737279415130615
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 3.4274933338165283,
      "step": 2748
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 2748,
      "training_step_time": 10.193787813186646
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 3.1402790546417236,
      "step": 2748
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 2748,
      "training_step_time": 9.908679008483887
    },
    {
      "epoch": 1.04827880859375e-05,
      "model_forward_time": 3.137158155441284,
      "step": 2748
    },
    {
      "epoch": 1.04827880859375e-05,
      "step": 2748,
      "training_step_time": 10.309004068374634
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 3.1938228607177734,
      "step": 2749
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 2749,
      "training_step_time": 9.96866512298584
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 3.141277551651001,
      "step": 2749
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 2749,
      "training_step_time": 10.186902284622192
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 3.137503147125244,
      "step": 2749
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 2749,
      "training_step_time": 9.910558223724365
    },
    {
      "epoch": 1.0486602783203125e-05,
      "model_forward_time": 2.432878017425537,
      "step": 2749
    },
    {
      "epoch": 1.0486602783203125e-05,
      "step": 2749,
      "training_step_time": 10.277328968048096
    },
    {
      "epoch": 1.049041748046875e-05,
      "grad_norm": 0.31823205947875977,
      "learning_rate": 1e-05,
      "loss": 0.0998,
      "step": 2750
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 3.152153968811035,
      "step": 2750
    },
    {
      "dynamics_loss_avg": 0.07293282784521579,
      "epoch": 1.049041748046875e-05,
      "step": 2750
    },
    {
      "action_loss_avg": 0.03908609040081501,
      "epoch": 1.049041748046875e-05,
      "step": 2750
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 2750,
      "training_step_time": 9.925334930419922
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 2.4296653270721436,
      "step": 2750
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 2750,
      "training_step_time": 7.4547202587127686
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 3.134904384613037,
      "step": 2750
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 2750,
      "training_step_time": 9.90943431854248
    },
    {
      "epoch": 1.049041748046875e-05,
      "model_forward_time": 3.3830695152282715,
      "step": 2750
    },
    {
      "epoch": 1.049041748046875e-05,
      "step": 2750,
      "training_step_time": 10.273942470550537
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 2.448582649230957,
      "step": 2751
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 2751,
      "training_step_time": 7.490761995315552
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 3.1451187133789062,
      "step": 2751
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 2751,
      "training_step_time": 9.920382261276245
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 3.145993709564209,
      "step": 2751
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 2751,
      "training_step_time": 9.920433044433594
    },
    {
      "epoch": 1.0494232177734375e-05,
      "model_forward_time": 2.441312551498413,
      "step": 2751
    },
    {
      "epoch": 1.0494232177734375e-05,
      "step": 2751,
      "training_step_time": 10.26949143409729
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 3.487030506134033,
      "step": 2752
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 2752,
      "training_step_time": 10.275573253631592
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 3.145845651626587,
      "step": 2752
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 2752,
      "training_step_time": 9.919942140579224
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 3.143052577972412,
      "step": 2752
    },
    {
      "dynamics_loss_avg": 0.04678913280367851,
      "epoch": 1.0498046875e-05,
      "step": 2752
    },
    {
      "action_loss_avg": 0.029461838491261005,
      "epoch": 1.0498046875e-05,
      "step": 2752
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 2752,
      "training_step_time": 9.916123867034912
    },
    {
      "epoch": 1.0498046875e-05,
      "model_forward_time": 3.1406655311584473,
      "step": 2752
    },
    {
      "epoch": 1.0498046875e-05,
      "step": 2752,
      "training_step_time": 10.02261471748352
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 2.446812868118286,
      "step": 2753
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 2753,
      "training_step_time": 7.47271466255188
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 3.412977457046509,
      "step": 2753
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 2753,
      "training_step_time": 10.180775880813599
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 2.42950439453125,
      "step": 2753
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 2753,
      "training_step_time": 7.457720756530762
    },
    {
      "epoch": 1.0501861572265625e-05,
      "model_forward_time": 3.1342554092407227,
      "step": 2753
    },
    {
      "epoch": 1.0501861572265625e-05,
      "step": 2753,
      "training_step_time": 10.26122260093689
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 3.1522305011749268,
      "step": 2754
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 2754,
      "training_step_time": 9.926734924316406
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 3.1370837688446045,
      "step": 2754
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 2754,
      "training_step_time": 9.912887334823608
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 3.429089307785034,
      "step": 2754
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 2754,
      "training_step_time": 10.203934669494629
    },
    {
      "epoch": 1.050567626953125e-05,
      "model_forward_time": 3.13852858543396,
      "step": 2754
    },
    {
      "epoch": 1.050567626953125e-05,
      "step": 2754,
      "training_step_time": 10.288414716720581
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 3.1547510623931885,
      "step": 2755
    },
    {
      "dynamics_loss_avg": 0.06299998592585325,
      "epoch": 1.0509490966796875e-05,
      "step": 2755
    },
    {
      "action_loss_avg": 0.04088871143758297,
      "epoch": 1.0509490966796875e-05,
      "step": 2755
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 2755,
      "training_step_time": 9.932511329650879
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 3.1384918689727783,
      "step": 2755
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 2755,
      "training_step_time": 9.912680387496948
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 3.146324634552002,
      "step": 2755
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 2755,
      "training_step_time": 10.178967952728271
    },
    {
      "epoch": 1.0509490966796875e-05,
      "model_forward_time": 3.142888069152832,
      "step": 2755
    },
    {
      "epoch": 1.0509490966796875e-05,
      "step": 2755,
      "training_step_time": 10.011521100997925
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 3.1492631435394287,
      "step": 2756
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 2756,
      "training_step_time": 9.91815710067749
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 3.1385083198547363,
      "step": 2756
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 2756,
      "training_step_time": 9.917236089706421
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 3.135067939758301,
      "step": 2756
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 2756,
      "training_step_time": 9.905188083648682
    },
    {
      "epoch": 1.05133056640625e-05,
      "model_forward_time": 2.4296607971191406,
      "step": 2756
    },
    {
      "epoch": 1.05133056640625e-05,
      "step": 2756,
      "training_step_time": 10.279285907745361
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 3.4532909393310547,
      "step": 2757
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 2757,
      "training_step_time": 10.223560333251953
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 3.1373789310455322,
      "step": 2757
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 2757,
      "training_step_time": 9.913520574569702
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 3.1391048431396484,
      "step": 2757
    },
    {
      "dynamics_loss_avg": 0.05619660820811987,
      "epoch": 1.0517120361328125e-05,
      "step": 2757
    },
    {
      "action_loss_avg": 0.0338159856852144,
      "epoch": 1.0517120361328125e-05,
      "step": 2757
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 2757,
      "training_step_time": 9.914371252059937
    },
    {
      "epoch": 1.0517120361328125e-05,
      "model_forward_time": 3.1383798122406006,
      "step": 2757
    },
    {
      "epoch": 1.0517120361328125e-05,
      "step": 2757,
      "training_step_time": 10.133155822753906
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 3.4715616703033447,
      "step": 2758
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 2758,
      "training_step_time": 10.24405288696289
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 3.136270046234131,
      "step": 2758
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 2758,
      "training_step_time": 9.905484914779663
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 3.140338659286499,
      "step": 2758
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 2758,
      "training_step_time": 9.911528587341309
    },
    {
      "epoch": 1.052093505859375e-05,
      "model_forward_time": 3.1370065212249756,
      "step": 2758
    },
    {
      "epoch": 1.052093505859375e-05,
      "step": 2758,
      "training_step_time": 10.278038740158081
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 2.440518856048584,
      "step": 2759
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 2759,
      "training_step_time": 7.465330362319946
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 3.135637044906616,
      "step": 2759
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 2759,
      "training_step_time": 10.18201994895935
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 3.1350631713867188,
      "step": 2759
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 2759,
      "training_step_time": 9.898684978485107
    },
    {
      "epoch": 1.0524749755859375e-05,
      "model_forward_time": 3.132781505584717,
      "step": 2759
    },
    {
      "epoch": 1.0524749755859375e-05,
      "step": 2759,
      "training_step_time": 10.013989210128784
    },
    {
      "epoch": 1.0528564453125e-05,
      "grad_norm": 0.44425877928733826,
      "learning_rate": 1e-05,
      "loss": 0.1029,
      "step": 2760
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 3.136045217514038,
      "step": 2760
    },
    {
      "dynamics_loss_avg": 0.07814267659559845,
      "epoch": 1.0528564453125e-05,
      "step": 2760
    },
    {
      "action_loss_avg": 0.03942260667681694,
      "epoch": 1.0528564453125e-05,
      "step": 2760
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 2760,
      "training_step_time": 9.905639410018921
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 3.1377556324005127,
      "step": 2760
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 2760,
      "training_step_time": 9.907256126403809
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 3.43216872215271,
      "step": 2760
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 2760,
      "training_step_time": 10.204673528671265
    },
    {
      "epoch": 1.0528564453125e-05,
      "model_forward_time": 3.1340105533599854,
      "step": 2760
    },
    {
      "epoch": 1.0528564453125e-05,
      "step": 2760,
      "training_step_time": 10.313694715499878
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 3.1434247493743896,
      "step": 2761
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 2761,
      "training_step_time": 9.915873050689697
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 2.433436393737793,
      "step": 2761
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 2761,
      "training_step_time": 7.45904803276062
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 3.135530948638916,
      "step": 2761
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 2761,
      "training_step_time": 9.907588720321655
    },
    {
      "epoch": 1.0532379150390625e-05,
      "model_forward_time": 3.440756320953369,
      "step": 2761
    },
    {
      "epoch": 1.0532379150390625e-05,
      "step": 2761,
      "training_step_time": 10.309863090515137
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 3.1539182662963867,
      "step": 2762
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 2762,
      "training_step_time": 9.937602758407593
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 2.441938877105713,
      "step": 2762
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 2762,
      "training_step_time": 7.47949481010437
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 3.1407761573791504,
      "step": 2762
    },
    {
      "dynamics_loss_avg": 0.05140133798122406,
      "epoch": 1.053619384765625e-05,
      "step": 2762
    },
    {
      "action_loss_avg": 0.03928933870047331,
      "epoch": 1.053619384765625e-05,
      "step": 2762
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 2762,
      "training_step_time": 9.91974949836731
    },
    {
      "epoch": 1.053619384765625e-05,
      "model_forward_time": 3.1457126140594482,
      "step": 2762
    },
    {
      "epoch": 1.053619384765625e-05,
      "step": 2762,
      "training_step_time": 10.031381130218506
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 3.1521434783935547,
      "step": 2763
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 2763,
      "training_step_time": 10.200774192810059
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 3.1462366580963135,
      "step": 2763
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 2763,
      "training_step_time": 9.923337936401367
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 3.1443912982940674,
      "step": 2763
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 2763,
      "training_step_time": 9.920324087142944
    },
    {
      "epoch": 1.0540008544921875e-05,
      "model_forward_time": 3.1419930458068848,
      "step": 2763
    },
    {
      "epoch": 1.0540008544921875e-05,
      "step": 2763,
      "training_step_time": 10.032881259918213
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 2.4458084106445312,
      "step": 2764
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 2764,
      "training_step_time": 7.475031852722168
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 3.4131526947021484,
      "step": 2764
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 2764,
      "training_step_time": 10.178823947906494
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 3.1340324878692627,
      "step": 2764
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 2764,
      "training_step_time": 9.90437650680542
    },
    {
      "epoch": 1.05438232421875e-05,
      "model_forward_time": 3.1425061225891113,
      "step": 2764
    },
    {
      "epoch": 1.05438232421875e-05,
      "step": 2764,
      "training_step_time": 10.288296937942505
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 3.1455440521240234,
      "step": 2765
    },
    {
      "dynamics_loss_avg": 0.06530800890177488,
      "epoch": 1.0547637939453125e-05,
      "step": 2765
    },
    {
      "action_loss_avg": 0.040401583164930345,
      "epoch": 1.0547637939453125e-05,
      "step": 2765
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 2765,
      "training_step_time": 9.917336463928223
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 3.1328110694885254,
      "step": 2765
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 2765,
      "training_step_time": 9.897239923477173
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 3.141902208328247,
      "step": 2765
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 2765,
      "training_step_time": 10.177191257476807
    },
    {
      "epoch": 1.0547637939453125e-05,
      "model_forward_time": 3.142784595489502,
      "step": 2765
    },
    {
      "epoch": 1.0547637939453125e-05,
      "step": 2765,
      "training_step_time": 10.025380849838257
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 3.1541197299957275,
      "step": 2766
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 2766,
      "training_step_time": 9.926523923873901
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 2.444831371307373,
      "step": 2766
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 2766,
      "training_step_time": 7.473981618881226
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 3.145766258239746,
      "step": 2766
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 2766,
      "training_step_time": 9.915170431137085
    },
    {
      "epoch": 1.055145263671875e-05,
      "model_forward_time": 3.4442625045776367,
      "step": 2766
    },
    {
      "epoch": 1.055145263671875e-05,
      "step": 2766,
      "training_step_time": 10.313605785369873
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 3.1531684398651123,
      "step": 2767
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 2767,
      "training_step_time": 9.933493375778198
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 3.134173631668091,
      "step": 2767
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 2767,
      "training_step_time": 9.903894186019897
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 2.4311301708221436,
      "step": 2767
    },
    {
      "dynamics_loss_avg": 0.07649185415357351,
      "epoch": 1.0555267333984375e-05,
      "step": 2767
    },
    {
      "action_loss_avg": 0.045573586225509645,
      "epoch": 1.0555267333984375e-05,
      "step": 2767
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 2767,
      "training_step_time": 7.463664531707764
    },
    {
      "epoch": 1.0555267333984375e-05,
      "model_forward_time": 3.1330068111419678,
      "step": 2767
    },
    {
      "epoch": 1.0555267333984375e-05,
      "step": 2767,
      "training_step_time": 10.011390447616577
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 3.153886079788208,
      "step": 2768
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 2768,
      "training_step_time": 9.929612159729004
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 3.40582275390625,
      "step": 2768
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 2768,
      "training_step_time": 10.182126760482788
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 3.1441590785980225,
      "step": 2768
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 2768,
      "training_step_time": 9.919866800308228
    },
    {
      "epoch": 1.055908203125e-05,
      "model_forward_time": 2.4350128173828125,
      "step": 2768
    },
    {
      "epoch": 1.055908203125e-05,
      "step": 2768,
      "training_step_time": 10.310606956481934
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 3.13779616355896,
      "step": 2769
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 2769,
      "training_step_time": 9.910382509231567
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 3.1321184635162354,
      "step": 2769
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 2769,
      "training_step_time": 10.17007327079773
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 3.13334321975708,
      "step": 2769
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 2769,
      "training_step_time": 9.90071964263916
    },
    {
      "epoch": 1.0562896728515625e-05,
      "model_forward_time": 3.129204034805298,
      "step": 2769
    },
    {
      "epoch": 1.0562896728515625e-05,
      "step": 2769,
      "training_step_time": 10.018653392791748
    },
    {
      "epoch": 1.056671142578125e-05,
      "grad_norm": 0.45652303099632263,
      "learning_rate": 1e-05,
      "loss": 0.1001,
      "step": 2770
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 3.153710126876831,
      "step": 2770
    },
    {
      "dynamics_loss_avg": 0.06583753935992717,
      "epoch": 1.056671142578125e-05,
      "step": 2770
    },
    {
      "action_loss_avg": 0.04508955320343375,
      "epoch": 1.056671142578125e-05,
      "step": 2770
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 2770,
      "training_step_time": 9.925219297409058
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 3.1376445293426514,
      "step": 2770
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 2770,
      "training_step_time": 9.91119384765625
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 3.1384212970733643,
      "step": 2770
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 2770,
      "training_step_time": 10.181520462036133
    },
    {
      "epoch": 1.056671142578125e-05,
      "model_forward_time": 3.14030122756958,
      "step": 2770
    },
    {
      "epoch": 1.056671142578125e-05,
      "step": 2770,
      "training_step_time": 10.311057090759277
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 3.155061960220337,
      "step": 2771
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 2771,
      "training_step_time": 9.931461572647095
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 3.1364519596099854,
      "step": 2771
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 2771,
      "training_step_time": 9.909841299057007
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 3.133169412612915,
      "step": 2771
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 2771,
      "training_step_time": 9.906693458557129
    },
    {
      "epoch": 1.0570526123046875e-05,
      "model_forward_time": 3.397515058517456,
      "step": 2771
    },
    {
      "epoch": 1.0570526123046875e-05,
      "step": 2771,
      "training_step_time": 10.27145266532898
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 3.1437368392944336,
      "step": 2772
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 2772,
      "training_step_time": 9.918991565704346
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 3.1362645626068115,
      "step": 2772
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 2772,
      "training_step_time": 9.902960300445557
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 2.4346933364868164,
      "step": 2772
    },
    {
      "dynamics_loss_avg": 0.06819634884595871,
      "epoch": 1.05743408203125e-05,
      "step": 2772
    },
    {
      "action_loss_avg": 0.049698739778250456,
      "epoch": 1.05743408203125e-05,
      "step": 2772
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 2772,
      "training_step_time": 7.462309122085571
    },
    {
      "epoch": 1.05743408203125e-05,
      "model_forward_time": 3.1319029331207275,
      "step": 2772
    },
    {
      "epoch": 1.05743408203125e-05,
      "step": 2772,
      "training_step_time": 10.312207698822021
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 3.3976542949676514,
      "step": 2773
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 2773,
      "training_step_time": 10.168729782104492
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 3.138058662414551,
      "step": 2773
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 2773,
      "training_step_time": 9.905725002288818
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 3.1318747997283936,
      "step": 2773
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 2773,
      "training_step_time": 9.896998882293701
    },
    {
      "epoch": 1.0578155517578126e-05,
      "model_forward_time": 2.432725667953491,
      "step": 2773
    },
    {
      "epoch": 1.0578155517578126e-05,
      "step": 2773,
      "training_step_time": 10.346535444259644
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 2.443197727203369,
      "step": 2774
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 2774,
      "training_step_time": 7.472010612487793
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 3.1294171810150146,
      "step": 2774
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 2774,
      "training_step_time": 9.898877620697021
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 3.3955469131469727,
      "step": 2774
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 2774,
      "training_step_time": 10.172260284423828
    },
    {
      "epoch": 1.058197021484375e-05,
      "model_forward_time": 3.1366426944732666,
      "step": 2774
    },
    {
      "epoch": 1.058197021484375e-05,
      "step": 2774,
      "training_step_time": 10.007463216781616
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 3.1509318351745605,
      "step": 2775
    },
    {
      "dynamics_loss_avg": 0.07033096365630627,
      "epoch": 1.0585784912109375e-05,
      "step": 2775
    },
    {
      "action_loss_avg": 0.04110460439696908,
      "epoch": 1.0585784912109375e-05,
      "step": 2775
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 2775,
      "training_step_time": 9.930559158325195
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 3.1357154846191406,
      "step": 2775
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 2775,
      "training_step_time": 9.91647481918335
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 3.1433756351470947,
      "step": 2775
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 2775,
      "training_step_time": 10.198521852493286
    },
    {
      "epoch": 1.0585784912109375e-05,
      "model_forward_time": 3.145566463470459,
      "step": 2775
    },
    {
      "epoch": 1.0585784912109375e-05,
      "step": 2775,
      "training_step_time": 10.310336589813232
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 3.1607275009155273,
      "step": 2776
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 2776,
      "training_step_time": 9.94395923614502
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 3.1395750045776367,
      "step": 2776
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 2776,
      "training_step_time": 9.91897201538086
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 2.4360103607177734,
      "step": 2776
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 2776,
      "training_step_time": 7.4724812507629395
    },
    {
      "epoch": 1.0589599609375e-05,
      "model_forward_time": 3.1289947032928467,
      "step": 2776
    },
    {
      "epoch": 1.0589599609375e-05,
      "step": 2776,
      "training_step_time": 10.303257703781128
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 3.4764695167541504,
      "step": 2777
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 2777,
      "training_step_time": 10.248069763183594
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 3.1319992542266846,
      "step": 2777
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 2777,
      "training_step_time": 9.906527996063232
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 3.1383564472198486,
      "step": 2777
    },
    {
      "dynamics_loss_avg": 0.04504080396145582,
      "epoch": 1.0593414306640626e-05,
      "step": 2777
    },
    {
      "action_loss_avg": 0.025457216799259184,
      "epoch": 1.0593414306640626e-05,
      "step": 2777
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 2777,
      "training_step_time": 9.907034158706665
    },
    {
      "epoch": 1.0593414306640626e-05,
      "model_forward_time": 3.135143756866455,
      "step": 2777
    },
    {
      "epoch": 1.0593414306640626e-05,
      "step": 2777,
      "training_step_time": 10.29482889175415
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 3.421680450439453,
      "step": 2778
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 2778,
      "training_step_time": 10.203510046005249
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 3.1308810710906982,
      "step": 2778
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 2778,
      "training_step_time": 9.905352115631104
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 2.432906150817871,
      "step": 2778
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 2778,
      "training_step_time": 7.4607579708099365
    },
    {
      "epoch": 1.059722900390625e-05,
      "model_forward_time": 3.135453462600708,
      "step": 2778
    },
    {
      "epoch": 1.059722900390625e-05,
      "step": 2778,
      "training_step_time": 10.024749994277954
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 3.1430535316467285,
      "step": 2779
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 2779,
      "training_step_time": 9.920893669128418
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 3.1437549591064453,
      "step": 2779
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 2779,
      "training_step_time": 10.18961477279663
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 3.1423099040985107,
      "step": 2779
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 2779,
      "training_step_time": 9.91949462890625
    },
    {
      "epoch": 1.0601043701171875e-05,
      "model_forward_time": 3.140453815460205,
      "step": 2779
    },
    {
      "epoch": 1.0601043701171875e-05,
      "step": 2779,
      "training_step_time": 10.333081007003784
    },
    {
      "epoch": 1.06048583984375e-05,
      "grad_norm": 0.30302947759628296,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 2780
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 2.463221311569214,
      "step": 2780
    },
    {
      "dynamics_loss_avg": 0.06468349639326335,
      "epoch": 1.06048583984375e-05,
      "step": 2780
    },
    {
      "action_loss_avg": 0.0359509278088808,
      "epoch": 1.06048583984375e-05,
      "step": 2780
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 2780,
      "training_step_time": 7.501716613769531
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 3.153477191925049,
      "step": 2780
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 2780,
      "training_step_time": 9.94075608253479
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 3.1398766040802,
      "step": 2780
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 2780,
      "training_step_time": 10.24229621887207
    },
    {
      "epoch": 1.06048583984375e-05,
      "model_forward_time": 3.134347677230835,
      "step": 2780
    },
    {
      "epoch": 1.06048583984375e-05,
      "step": 2780,
      "training_step_time": 10.282049894332886
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 3.1609230041503906,
      "step": 2781
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 2781,
      "training_step_time": 9.929419040679932
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 3.140862464904785,
      "step": 2781
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 2781,
      "training_step_time": 9.91274619102478
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 3.135082244873047,
      "step": 2781
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 2781,
      "training_step_time": 9.906640768051147
    },
    {
      "epoch": 1.0608673095703126e-05,
      "model_forward_time": 3.1398794651031494,
      "step": 2781
    },
    {
      "epoch": 1.0608673095703126e-05,
      "step": 2781,
      "training_step_time": 10.275368213653564
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 3.161407947540283,
      "step": 2782
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 2782,
      "training_step_time": 9.93489408493042
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 3.1356942653656006,
      "step": 2782
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 2782,
      "training_step_time": 9.903362512588501
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 3.1313529014587402,
      "step": 2782
    },
    {
      "dynamics_loss_avg": 0.07786989994347096,
      "epoch": 1.061248779296875e-05,
      "step": 2782
    },
    {
      "action_loss_avg": 0.05916931368410587,
      "epoch": 1.061248779296875e-05,
      "step": 2782
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 2782,
      "training_step_time": 9.903527021408081
    },
    {
      "epoch": 1.061248779296875e-05,
      "model_forward_time": 3.13765025138855,
      "step": 2782
    },
    {
      "epoch": 1.061248779296875e-05,
      "step": 2782,
      "training_step_time": 10.286789894104004
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 3.1462180614471436,
      "step": 2783
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 2783,
      "training_step_time": 10.239218473434448
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 3.1354098320007324,
      "step": 2783
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 2783,
      "training_step_time": 9.907962560653687
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 3.136967182159424,
      "step": 2783
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 2783,
      "training_step_time": 9.91576337814331
    },
    {
      "epoch": 1.0616302490234375e-05,
      "model_forward_time": 3.147219181060791,
      "step": 2783
    },
    {
      "epoch": 1.0616302490234375e-05,
      "step": 2783,
      "training_step_time": 10.014713764190674
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 3.1473991870880127,
      "step": 2784
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 2784,
      "training_step_time": 9.92131233215332
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 3.428696393966675,
      "step": 2784
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 2784,
      "training_step_time": 10.203634023666382
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 3.1371877193450928,
      "step": 2784
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 2784,
      "training_step_time": 9.910173654556274
    },
    {
      "epoch": 1.06201171875e-05,
      "model_forward_time": 3.1424505710601807,
      "step": 2784
    },
    {
      "epoch": 1.06201171875e-05,
      "step": 2784,
      "training_step_time": 10.023815631866455
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 3.1472699642181396,
      "step": 2785
    },
    {
      "dynamics_loss_avg": 0.06566500663757324,
      "epoch": 1.0623931884765626e-05,
      "step": 2785
    },
    {
      "action_loss_avg": 0.04313294254243374,
      "epoch": 1.0623931884765626e-05,
      "step": 2785
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 2785,
      "training_step_time": 9.926011085510254
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 2.435199022293091,
      "step": 2785
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 2785,
      "training_step_time": 7.469594717025757
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 3.1359996795654297,
      "step": 2785
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 2785,
      "training_step_time": 9.911943197250366
    },
    {
      "epoch": 1.0623931884765626e-05,
      "model_forward_time": 3.405362606048584,
      "step": 2785
    },
    {
      "epoch": 1.0623931884765626e-05,
      "step": 2785,
      "training_step_time": 10.281482696533203
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 3.1433019638061523,
      "step": 2786
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 2786,
      "training_step_time": 9.919445514678955
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 3.1385080814361572,
      "step": 2786
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 2786,
      "training_step_time": 9.913881540298462
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 3.1422033309936523,
      "step": 2786
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 2786,
      "training_step_time": 9.916099786758423
    },
    {
      "epoch": 1.062774658203125e-05,
      "model_forward_time": 3.4440503120422363,
      "step": 2786
    },
    {
      "epoch": 1.062774658203125e-05,
      "step": 2786,
      "training_step_time": 10.311464548110962
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 3.151249647140503,
      "step": 2787
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 2787,
      "training_step_time": 9.93041729927063
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 3.1341097354888916,
      "step": 2787
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 2787,
      "training_step_time": 9.915250062942505
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 3.132702589035034,
      "step": 2787
    },
    {
      "dynamics_loss_avg": 0.05045745428651571,
      "epoch": 1.0631561279296875e-05,
      "step": 2787
    },
    {
      "action_loss_avg": 0.03332286588847637,
      "epoch": 1.0631561279296875e-05,
      "step": 2787
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 2787,
      "training_step_time": 9.911595344543457
    },
    {
      "epoch": 1.0631561279296875e-05,
      "model_forward_time": 3.1347734928131104,
      "step": 2787
    },
    {
      "epoch": 1.0631561279296875e-05,
      "step": 2787,
      "training_step_time": 10.321416139602661
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 2.436934232711792,
      "step": 2788
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 2788,
      "training_step_time": 7.46479344367981
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 3.3821566104888916,
      "step": 2788
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 2788,
      "training_step_time": 10.152775287628174
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 3.1340386867523193,
      "step": 2788
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 2788,
      "training_step_time": 9.917206048965454
    },
    {
      "epoch": 1.06353759765625e-05,
      "model_forward_time": 3.1380090713500977,
      "step": 2788
    },
    {
      "epoch": 1.06353759765625e-05,
      "step": 2788,
      "training_step_time": 10.301422834396362
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 2.4493093490600586,
      "step": 2789
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 2789,
      "training_step_time": 7.485212326049805
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 3.137385368347168,
      "step": 2789
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 2789,
      "training_step_time": 10.197839736938477
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 3.139378547668457,
      "step": 2789
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 2789,
      "training_step_time": 9.92210865020752
    },
    {
      "epoch": 1.0639190673828126e-05,
      "model_forward_time": 2.435393810272217,
      "step": 2789
    },
    {
      "epoch": 1.0639190673828126e-05,
      "step": 2789,
      "training_step_time": 10.022966623306274
    },
    {
      "epoch": 1.064300537109375e-05,
      "grad_norm": 0.49817541241645813,
      "learning_rate": 1e-05,
      "loss": 0.1032,
      "step": 2790
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 3.1506102085113525,
      "step": 2790
    },
    {
      "dynamics_loss_avg": 0.06814453303813935,
      "epoch": 1.064300537109375e-05,
      "step": 2790
    },
    {
      "action_loss_avg": 0.032178031466901305,
      "epoch": 1.064300537109375e-05,
      "step": 2790
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 2790,
      "training_step_time": 9.937175989151001
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 3.1390140056610107,
      "step": 2790
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 2790,
      "training_step_time": 9.921126127243042
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 3.142936944961548,
      "step": 2790
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 2790,
      "training_step_time": 9.928521394729614
    },
    {
      "epoch": 1.064300537109375e-05,
      "model_forward_time": 3.3808698654174805,
      "step": 2790
    },
    {
      "epoch": 1.064300537109375e-05,
      "step": 2790,
      "training_step_time": 10.257517337799072
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 3.146437883377075,
      "step": 2791
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 2791,
      "training_step_time": 9.92676830291748
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 3.135408639907837,
      "step": 2791
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 2791,
      "training_step_time": 9.916868686676025
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 3.1374459266662598,
      "step": 2791
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 2791,
      "training_step_time": 9.918787002563477
    },
    {
      "epoch": 1.0646820068359375e-05,
      "model_forward_time": 3.13828182220459,
      "step": 2791
    },
    {
      "epoch": 1.0646820068359375e-05,
      "step": 2791,
      "training_step_time": 10.299364566802979
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 3.146630048751831,
      "step": 2792
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 2792,
      "training_step_time": 9.929713487625122
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 3.148303270339966,
      "step": 2792
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 2792,
      "training_step_time": 9.932981252670288
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 3.1430764198303223,
      "step": 2792
    },
    {
      "dynamics_loss_avg": 0.06567717026919126,
      "epoch": 1.0650634765625e-05,
      "step": 2792
    },
    {
      "action_loss_avg": 0.039761231373995544,
      "epoch": 1.0650634765625e-05,
      "step": 2792
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 2792,
      "training_step_time": 9.927681922912598
    },
    {
      "epoch": 1.0650634765625e-05,
      "model_forward_time": 3.149047374725342,
      "step": 2792
    },
    {
      "epoch": 1.0650634765625e-05,
      "step": 2792,
      "training_step_time": 10.321216583251953
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 3.4605259895324707,
      "step": 2793
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 2793,
      "training_step_time": 10.250685453414917
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 3.1391286849975586,
      "step": 2793
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 2793,
      "training_step_time": 9.922203302383423
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 3.138763427734375,
      "step": 2793
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 2793,
      "training_step_time": 9.922109365463257
    },
    {
      "epoch": 1.0654449462890626e-05,
      "model_forward_time": 3.1398589611053467,
      "step": 2793
    },
    {
      "epoch": 1.0654449462890626e-05,
      "step": 2793,
      "training_step_time": 10.013317584991455
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 2.449490785598755,
      "step": 2794
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 2794,
      "training_step_time": 7.491205215454102
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 3.13146710395813,
      "step": 2794
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 2794,
      "training_step_time": 9.907369375228882
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 3.3775293827056885,
      "step": 2794
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 2794,
      "training_step_time": 10.14883804321289
    },
    {
      "epoch": 1.065826416015625e-05,
      "model_forward_time": 3.137397050857544,
      "step": 2794
    },
    {
      "epoch": 1.065826416015625e-05,
      "step": 2794,
      "training_step_time": 10.011699914932251
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 3.1466567516326904,
      "step": 2795
    },
    {
      "dynamics_loss_avg": 0.068461590167135,
      "epoch": 1.0662078857421875e-05,
      "step": 2795
    },
    {
      "action_loss_avg": 0.04120949781499803,
      "epoch": 1.0662078857421875e-05,
      "step": 2795
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 2795,
      "training_step_time": 9.92519760131836
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 3.1300346851348877,
      "step": 2795
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 2795,
      "training_step_time": 9.900357723236084
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 3.4225940704345703,
      "step": 2795
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 2795,
      "training_step_time": 10.197501182556152
    },
    {
      "epoch": 1.0662078857421875e-05,
      "model_forward_time": 3.142871379852295,
      "step": 2795
    },
    {
      "epoch": 1.0662078857421875e-05,
      "step": 2795,
      "training_step_time": 10.271711587905884
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 3.1435229778289795,
      "step": 2796
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 2796,
      "training_step_time": 9.928727388381958
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 2.4429125785827637,
      "step": 2796
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 2796,
      "training_step_time": 7.4761810302734375
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 3.150630474090576,
      "step": 2796
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 2796,
      "training_step_time": 9.928386449813843
    },
    {
      "epoch": 1.06658935546875e-05,
      "model_forward_time": 3.1403255462646484,
      "step": 2796
    },
    {
      "epoch": 1.06658935546875e-05,
      "step": 2796,
      "training_step_time": 10.314183473587036
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 3.5204532146453857,
      "step": 2797
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 2797,
      "training_step_time": 10.300373792648315
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 3.1372382640838623,
      "step": 2797
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 2797,
      "training_step_time": 9.919520139694214
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 3.138751745223999,
      "step": 2797
    },
    {
      "dynamics_loss_avg": 0.05901372097432613,
      "epoch": 1.0669708251953126e-05,
      "step": 2797
    },
    {
      "action_loss_avg": 0.02945097554475069,
      "epoch": 1.0669708251953126e-05,
      "step": 2797
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 2797,
      "training_step_time": 9.923521757125854
    },
    {
      "epoch": 1.0669708251953126e-05,
      "model_forward_time": 3.1462302207946777,
      "step": 2797
    },
    {
      "epoch": 1.0669708251953126e-05,
      "step": 2797,
      "training_step_time": 10.288015604019165
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 3.4503045082092285,
      "step": 2798
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 2798,
      "training_step_time": 10.231064319610596
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 3.1390328407287598,
      "step": 2798
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 2798,
      "training_step_time": 9.917572259902954
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 3.1397807598114014,
      "step": 2798
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 2798,
      "training_step_time": 9.914485931396484
    },
    {
      "epoch": 1.067352294921875e-05,
      "model_forward_time": 3.139286756515503,
      "step": 2798
    },
    {
      "epoch": 1.067352294921875e-05,
      "step": 2798,
      "training_step_time": 10.03103494644165
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 3.1480014324188232,
      "step": 2799
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 2799,
      "training_step_time": 9.928658485412598
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 3.4001920223236084,
      "step": 2799
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 2799,
      "training_step_time": 10.17078709602356
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 3.1452972888946533,
      "step": 2799
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 2799,
      "training_step_time": 9.916132926940918
    },
    {
      "epoch": 1.0677337646484375e-05,
      "model_forward_time": 3.134493350982666,
      "step": 2799
    },
    {
      "epoch": 1.0677337646484375e-05,
      "step": 2799,
      "training_step_time": 10.006043195724487
    },
    {
      "epoch": 1.068115234375e-05,
      "grad_norm": 0.7353233695030212,
      "learning_rate": 1e-05,
      "loss": 0.1014,
      "step": 2800
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 3.198294162750244,
      "step": 2800
    },
    {
      "dynamics_loss_avg": 0.07008228618651628,
      "epoch": 1.068115234375e-05,
      "step": 2800
    },
    {
      "action_loss_avg": 0.045325849391520026,
      "epoch": 1.068115234375e-05,
      "step": 2800
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 2800,
      "training_step_time": 9.978574991226196
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 3.1417579650878906,
      "step": 2800
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 2800,
      "training_step_time": 9.910403966903687
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 2.44549822807312,
      "step": 2800
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 2800,
      "training_step_time": 7.796947479248047
    },
    {
      "epoch": 1.068115234375e-05,
      "model_forward_time": 3.13826322555542,
      "step": 2800
    },
    {
      "epoch": 1.068115234375e-05,
      "step": 2800,
      "training_step_time": 10.268814086914062
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 2.457399606704712,
      "step": 2801
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 2801,
      "training_step_time": 7.495826244354248
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 3.1421213150024414,
      "step": 2801
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 2801,
      "training_step_time": 9.918333768844604
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 2.4286906719207764,
      "step": 2801
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 2801,
      "training_step_time": 7.453373670578003
    },
    {
      "epoch": 1.0684967041015626e-05,
      "model_forward_time": 3.1379897594451904,
      "step": 2801
    },
    {
      "epoch": 1.0684967041015626e-05,
      "step": 2801,
      "training_step_time": 10.270805835723877
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 3.4944682121276855,
      "step": 2802
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 2802,
      "training_step_time": 10.262296438217163
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 3.136789560317993,
      "step": 2802
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 2802,
      "training_step_time": 9.905481338500977
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 3.1324751377105713,
      "step": 2802
    },
    {
      "dynamics_loss_avg": 0.05413664728403091,
      "epoch": 1.068878173828125e-05,
      "step": 2802
    },
    {
      "action_loss_avg": 0.03205387610942125,
      "epoch": 1.068878173828125e-05,
      "step": 2802
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 2802,
      "training_step_time": 9.907147407531738
    },
    {
      "epoch": 1.068878173828125e-05,
      "model_forward_time": 3.1389880180358887,
      "step": 2802
    },
    {
      "epoch": 1.068878173828125e-05,
      "step": 2802,
      "training_step_time": 10.034775972366333
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 3.1508986949920654,
      "step": 2803
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 2803,
      "training_step_time": 10.191384315490723
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 3.138542413711548,
      "step": 2803
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 2803,
      "training_step_time": 9.908473491668701
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 3.1357262134552,
      "step": 2803
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 2803,
      "training_step_time": 9.8995361328125
    },
    {
      "epoch": 1.0692596435546875e-05,
      "model_forward_time": 2.4288949966430664,
      "step": 2803
    },
    {
      "epoch": 1.0692596435546875e-05,
      "step": 2803,
      "training_step_time": 10.314234495162964
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 3.1582868099212646,
      "step": 2804
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 2804,
      "training_step_time": 9.927871227264404
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 2.428762435913086,
      "step": 2804
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 2804,
      "training_step_time": 7.449277877807617
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 3.408557415008545,
      "step": 2804
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 2804,
      "training_step_time": 10.164088249206543
    },
    {
      "epoch": 1.06964111328125e-05,
      "model_forward_time": 3.132107734680176,
      "step": 2804
    },
    {
      "epoch": 1.06964111328125e-05,
      "step": 2804,
      "training_step_time": 10.276567935943604
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 3.138566493988037,
      "step": 2805
    },
    {
      "dynamics_loss_avg": 0.04818274527788162,
      "epoch": 1.0700225830078126e-05,
      "step": 2805
    },
    {
      "action_loss_avg": 0.03308284496888518,
      "epoch": 1.0700225830078126e-05,
      "step": 2805
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 2805,
      "training_step_time": 9.904175519943237
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 3.128650426864624,
      "step": 2805
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 2805,
      "training_step_time": 9.897031307220459
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 3.1312572956085205,
      "step": 2805
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 2805,
      "training_step_time": 9.895942211151123
    },
    {
      "epoch": 1.0700225830078126e-05,
      "model_forward_time": 3.4005420207977295,
      "step": 2805
    },
    {
      "epoch": 1.0700225830078126e-05,
      "step": 2805,
      "training_step_time": 10.287099599838257
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 3.1527411937713623,
      "step": 2806
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 2806,
      "training_step_time": 9.943809986114502
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 3.141542434692383,
      "step": 2806
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 2806,
      "training_step_time": 9.911621570587158
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 3.142369508743286,
      "step": 2806
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 2806,
      "training_step_time": 9.908760070800781
    },
    {
      "epoch": 1.070404052734375e-05,
      "model_forward_time": 3.414200782775879,
      "step": 2806
    },
    {
      "epoch": 1.070404052734375e-05,
      "step": 2806,
      "training_step_time": 10.290181398391724
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 3.1530654430389404,
      "step": 2807
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 2807,
      "training_step_time": 9.924129724502563
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 3.137031316757202,
      "step": 2807
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 2807,
      "training_step_time": 9.90470576286316
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 3.1401207447052,
      "step": 2807
    },
    {
      "dynamics_loss_avg": 0.07429844103753566,
      "epoch": 1.0707855224609375e-05,
      "step": 2807
    },
    {
      "action_loss_avg": 0.0416030827909708,
      "epoch": 1.0707855224609375e-05,
      "step": 2807
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 2807,
      "training_step_time": 9.913936614990234
    },
    {
      "epoch": 1.0707855224609375e-05,
      "model_forward_time": 3.135329008102417,
      "step": 2807
    },
    {
      "epoch": 1.0707855224609375e-05,
      "step": 2807,
      "training_step_time": 10.327594995498657
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 3.1372292041778564,
      "step": 2808
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 2808,
      "training_step_time": 10.154927968978882
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 3.1384823322296143,
      "step": 2808
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 2808,
      "training_step_time": 9.903796672821045
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 3.144562244415283,
      "step": 2808
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 2808,
      "training_step_time": 9.906700134277344
    },
    {
      "epoch": 1.0711669921875e-05,
      "model_forward_time": 3.1404190063476562,
      "step": 2808
    },
    {
      "epoch": 1.0711669921875e-05,
      "step": 2808,
      "training_step_time": 10.261488676071167
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 3.1529040336608887,
      "step": 2809
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 2809,
      "training_step_time": 9.920111656188965
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 3.425528049468994,
      "step": 2809
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 2809,
      "training_step_time": 10.191067457199097
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 3.1442067623138428,
      "step": 2809
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 2809,
      "training_step_time": 9.917009353637695
    },
    {
      "epoch": 1.0715484619140626e-05,
      "model_forward_time": 3.140191078186035,
      "step": 2809
    },
    {
      "epoch": 1.0715484619140626e-05,
      "step": 2809,
      "training_step_time": 10.264379024505615
    },
    {
      "epoch": 1.071929931640625e-05,
      "grad_norm": 0.7031672596931458,
      "learning_rate": 1e-05,
      "loss": 0.0983,
      "step": 2810
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 3.151759624481201,
      "step": 2810
    },
    {
      "dynamics_loss_avg": 0.05713153164833784,
      "epoch": 1.071929931640625e-05,
      "step": 2810
    },
    {
      "action_loss_avg": 0.03518092501908541,
      "epoch": 1.071929931640625e-05,
      "step": 2810
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 2810,
      "training_step_time": 9.927190065383911
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 3.152498722076416,
      "step": 2810
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 2810,
      "training_step_time": 9.935895681381226
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 3.4372973442077637,
      "step": 2810
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 2810,
      "training_step_time": 10.204181671142578
    },
    {
      "epoch": 1.071929931640625e-05,
      "model_forward_time": 3.1416351795196533,
      "step": 2810
    },
    {
      "epoch": 1.071929931640625e-05,
      "step": 2810,
      "training_step_time": 10.288526773452759
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 3.143610954284668,
      "step": 2811
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 2811,
      "training_step_time": 9.914941549301147
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 3.1395115852355957,
      "step": 2811
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 2811,
      "training_step_time": 9.905757188796997
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 3.1358046531677246,
      "step": 2811
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 2811,
      "training_step_time": 9.89969801902771
    },
    {
      "epoch": 1.0723114013671875e-05,
      "model_forward_time": 3.135634422302246,
      "step": 2811
    },
    {
      "epoch": 1.0723114013671875e-05,
      "step": 2811,
      "training_step_time": 10.310481071472168
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 3.142043113708496,
      "step": 2812
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 2812,
      "training_step_time": 9.914659261703491
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 3.137532949447632,
      "step": 2812
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 2812,
      "training_step_time": 9.903959512710571
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 3.1341586112976074,
      "step": 2812
    },
    {
      "dynamics_loss_avg": 0.06406886279582977,
      "epoch": 1.07269287109375e-05,
      "step": 2812
    },
    {
      "action_loss_avg": 0.04053190164268017,
      "epoch": 1.07269287109375e-05,
      "step": 2812
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 2812,
      "training_step_time": 9.907737493515015
    },
    {
      "epoch": 1.07269287109375e-05,
      "model_forward_time": 3.1419856548309326,
      "step": 2812
    },
    {
      "epoch": 1.07269287109375e-05,
      "step": 2812,
      "training_step_time": 10.29776644706726
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 2.4475722312927246,
      "step": 2813
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 2813,
      "training_step_time": 7.492784023284912
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 2.7303686141967773,
      "step": 2813
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 2813,
      "training_step_time": 7.756475210189819
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 3.139469861984253,
      "step": 2813
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 2813,
      "training_step_time": 9.912009000778198
    },
    {
      "epoch": 1.0730743408203126e-05,
      "model_forward_time": 3.1328632831573486,
      "step": 2813
    },
    {
      "epoch": 1.0730743408203126e-05,
      "step": 2813,
      "training_step_time": 10.289406776428223
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 3.1458418369293213,
      "step": 2814
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 2814,
      "training_step_time": 9.92094111442566
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 3.1322011947631836,
      "step": 2814
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 2814,
      "training_step_time": 10.178679466247559
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 3.131781578063965,
      "step": 2814
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 2814,
      "training_step_time": 9.90053939819336
    },
    {
      "epoch": 1.073455810546875e-05,
      "model_forward_time": 3.135568857192993,
      "step": 2814
    },
    {
      "epoch": 1.073455810546875e-05,
      "step": 2814,
      "training_step_time": 10.306297302246094
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 3.1500167846679688,
      "step": 2815
    },
    {
      "dynamics_loss_avg": 0.03731604311615229,
      "epoch": 1.0738372802734375e-05,
      "step": 2815
    },
    {
      "action_loss_avg": 0.02476849895901978,
      "epoch": 1.0738372802734375e-05,
      "step": 2815
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 2815,
      "training_step_time": 9.921703815460205
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 2.4361636638641357,
      "step": 2815
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 2815,
      "training_step_time": 7.464212656021118
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 3.392521619796753,
      "step": 2815
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 2815,
      "training_step_time": 10.158549547195435
    },
    {
      "epoch": 1.0738372802734375e-05,
      "model_forward_time": 3.137153387069702,
      "step": 2815
    },
    {
      "epoch": 1.0738372802734375e-05,
      "step": 2815,
      "training_step_time": 10.032316207885742
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 3.1507904529571533,
      "step": 2816
    },
    {
      "epoch": 1.07421875e-05,
      "step": 2816,
      "training_step_time": 9.921884775161743
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 2.4323840141296387,
      "step": 2816
    },
    {
      "epoch": 1.07421875e-05,
      "step": 2816,
      "training_step_time": 7.461026430130005
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 3.1350018978118896,
      "step": 2816
    },
    {
      "epoch": 1.07421875e-05,
      "step": 2816,
      "training_step_time": 9.910006523132324
    },
    {
      "epoch": 1.07421875e-05,
      "model_forward_time": 3.1362662315368652,
      "step": 2816
    },
    {
      "epoch": 1.07421875e-05,
      "step": 2816,
      "training_step_time": 10.024658679962158
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 3.4975273609161377,
      "step": 2817
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 2817,
      "training_step_time": 10.268167495727539
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 3.1460158824920654,
      "step": 2817
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 2817,
      "training_step_time": 9.916529893875122
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 3.149893045425415,
      "step": 2817
    },
    {
      "dynamics_loss_avg": 0.06911319568753242,
      "epoch": 1.0746002197265626e-05,
      "step": 2817
    },
    {
      "action_loss_avg": 0.03596662431955337,
      "epoch": 1.0746002197265626e-05,
      "step": 2817
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 2817,
      "training_step_time": 9.92245626449585
    },
    {
      "epoch": 1.0746002197265626e-05,
      "model_forward_time": 3.148453712463379,
      "step": 2817
    },
    {
      "epoch": 1.0746002197265626e-05,
      "step": 2817,
      "training_step_time": 10.312062978744507
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 3.4488203525543213,
      "step": 2818
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 2818,
      "training_step_time": 10.227107286453247
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 3.1454198360443115,
      "step": 2818
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 2818,
      "training_step_time": 9.920092582702637
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 3.146768569946289,
      "step": 2818
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 2818,
      "training_step_time": 9.918763399124146
    },
    {
      "epoch": 1.074981689453125e-05,
      "model_forward_time": 3.1426525115966797,
      "step": 2818
    },
    {
      "epoch": 1.074981689453125e-05,
      "step": 2818,
      "training_step_time": 10.302628517150879
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 3.147108554840088,
      "step": 2819
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 2819,
      "training_step_time": 9.919028282165527
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 3.4055185317993164,
      "step": 2819
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 2819,
      "training_step_time": 10.176132202148438
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 3.1400747299194336,
      "step": 2819
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 2819,
      "training_step_time": 9.912715196609497
    },
    {
      "epoch": 1.0753631591796875e-05,
      "model_forward_time": 3.137730836868286,
      "step": 2819
    },
    {
      "epoch": 1.0753631591796875e-05,
      "step": 2819,
      "training_step_time": 10.244433403015137
    },
    {
      "epoch": 1.07574462890625e-05,
      "grad_norm": 0.48825207352638245,
      "learning_rate": 1e-05,
      "loss": 0.1016,
      "step": 2820
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 3.139801263809204,
      "step": 2820
    },
    {
      "dynamics_loss_avg": 0.07633676584810019,
      "epoch": 1.07574462890625e-05,
      "step": 2820
    },
    {
      "action_loss_avg": 0.05461406540125609,
      "epoch": 1.07574462890625e-05,
      "step": 2820
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 2820,
      "training_step_time": 9.910393238067627
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 3.1408350467681885,
      "step": 2820
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 2820,
      "training_step_time": 9.906490325927734
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 3.1365528106689453,
      "step": 2820
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 2820,
      "training_step_time": 10.183985948562622
    },
    {
      "epoch": 1.07574462890625e-05,
      "model_forward_time": 3.1404600143432617,
      "step": 2820
    },
    {
      "epoch": 1.07574462890625e-05,
      "step": 2820,
      "training_step_time": 10.00957703590393
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 3.1561150550842285,
      "step": 2821
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 2821,
      "training_step_time": 9.929124355316162
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 2.4305360317230225,
      "step": 2821
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 2821,
      "training_step_time": 7.460847616195679
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 3.133366584777832,
      "step": 2821
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 2821,
      "training_step_time": 9.904505252838135
    },
    {
      "epoch": 1.0761260986328126e-05,
      "model_forward_time": 2.4342281818389893,
      "step": 2821
    },
    {
      "epoch": 1.0761260986328126e-05,
      "step": 2821,
      "training_step_time": 10.32252025604248
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 3.4206883907318115,
      "step": 2822
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 2822,
      "training_step_time": 10.195436954498291
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 3.144308567047119,
      "step": 2822
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 2822,
      "training_step_time": 9.914708137512207
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 3.132969856262207,
      "step": 2822
    },
    {
      "dynamics_loss_avg": 0.05194316785782575,
      "epoch": 1.076507568359375e-05,
      "step": 2822
    },
    {
      "action_loss_avg": 0.02802489995956421,
      "epoch": 1.076507568359375e-05,
      "step": 2822
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 2822,
      "training_step_time": 9.911460399627686
    },
    {
      "epoch": 1.076507568359375e-05,
      "model_forward_time": 3.1366701126098633,
      "step": 2822
    },
    {
      "epoch": 1.076507568359375e-05,
      "step": 2822,
      "training_step_time": 10.284797668457031
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 3.1517980098724365,
      "step": 2823
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 2823,
      "training_step_time": 9.922923803329468
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 2.6791038513183594,
      "step": 2823
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 2823,
      "training_step_time": 7.703716516494751
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 3.1286587715148926,
      "step": 2823
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 2823,
      "training_step_time": 9.897631883621216
    },
    {
      "epoch": 1.0768890380859375e-05,
      "model_forward_time": 3.1358225345611572,
      "step": 2823
    },
    {
      "epoch": 1.0768890380859375e-05,
      "step": 2823,
      "training_step_time": 10.25516152381897
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 3.1408512592315674,
      "step": 2824
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 2824,
      "training_step_time": 9.917147159576416
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 3.131958484649658,
      "step": 2824
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 2824,
      "training_step_time": 9.910522937774658
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 2.697568416595459,
      "step": 2824
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 2824,
      "training_step_time": 7.727576732635498
    },
    {
      "epoch": 1.0772705078125e-05,
      "model_forward_time": 3.1334474086761475,
      "step": 2824
    },
    {
      "epoch": 1.0772705078125e-05,
      "step": 2824,
      "training_step_time": 10.305239200592041
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 3.1452832221984863,
      "step": 2825
    },
    {
      "dynamics_loss_avg": 0.05958307385444641,
      "epoch": 1.0776519775390626e-05,
      "step": 2825
    },
    {
      "action_loss_avg": 0.05217093266546726,
      "epoch": 1.0776519775390626e-05,
      "step": 2825
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 2825,
      "training_step_time": 9.935601472854614
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 3.13118052482605,
      "step": 2825
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 2825,
      "training_step_time": 9.907443046569824
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 3.140918731689453,
      "step": 2825
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 2825,
      "training_step_time": 9.925427913665771
    },
    {
      "epoch": 1.0776519775390626e-05,
      "model_forward_time": 2.744213819503784,
      "step": 2825
    },
    {
      "epoch": 1.0776519775390626e-05,
      "step": 2825,
      "training_step_time": 10.308776140213013
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 3.1499454975128174,
      "step": 2826
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 2826,
      "training_step_time": 9.932795763015747
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 3.1323509216308594,
      "step": 2826
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 2826,
      "training_step_time": 9.906456232070923
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 2.432055711746216,
      "step": 2826
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 2826,
      "training_step_time": 7.468812465667725
    },
    {
      "epoch": 1.078033447265625e-05,
      "model_forward_time": 3.1317546367645264,
      "step": 2826
    },
    {
      "epoch": 1.078033447265625e-05,
      "step": 2826,
      "training_step_time": 10.313566207885742
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 3.451223134994507,
      "step": 2827
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 2827,
      "training_step_time": 10.235887050628662
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 3.1327455043792725,
      "step": 2827
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 2827,
      "training_step_time": 9.91457200050354
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 3.140748977661133,
      "step": 2827
    },
    {
      "dynamics_loss_avg": 0.07136006541550159,
      "epoch": 1.0784149169921875e-05,
      "step": 2827
    },
    {
      "action_loss_avg": 0.04997817687690258,
      "epoch": 1.0784149169921875e-05,
      "step": 2827
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 2827,
      "training_step_time": 9.928879499435425
    },
    {
      "epoch": 1.0784149169921875e-05,
      "model_forward_time": 3.1440374851226807,
      "step": 2827
    },
    {
      "epoch": 1.0784149169921875e-05,
      "step": 2827,
      "training_step_time": 10.024523973464966
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 3.1639857292175293,
      "step": 2828
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 2828,
      "training_step_time": 9.945659399032593
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 3.4365952014923096,
      "step": 2828
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 2828,
      "training_step_time": 10.21443247795105
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 3.138895273208618,
      "step": 2828
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 2828,
      "training_step_time": 9.920378923416138
    },
    {
      "epoch": 1.07879638671875e-05,
      "model_forward_time": 3.135754346847534,
      "step": 2828
    },
    {
      "epoch": 1.07879638671875e-05,
      "step": 2828,
      "training_step_time": 10.321648359298706
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 3.1482510566711426,
      "step": 2829
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 2829,
      "training_step_time": 9.924855709075928
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 3.1370654106140137,
      "step": 2829
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 2829,
      "training_step_time": 10.179931163787842
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 3.1341514587402344,
      "step": 2829
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 2829,
      "training_step_time": 9.911448001861572
    },
    {
      "epoch": 1.0791778564453126e-05,
      "model_forward_time": 3.1304984092712402,
      "step": 2829
    },
    {
      "epoch": 1.0791778564453126e-05,
      "step": 2829,
      "training_step_time": 10.019728899002075
    },
    {
      "epoch": 1.079559326171875e-05,
      "grad_norm": 0.44118648767471313,
      "learning_rate": 1e-05,
      "loss": 0.1026,
      "step": 2830
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 3.1488606929779053,
      "step": 2830
    },
    {
      "dynamics_loss_avg": 0.05582756828516722,
      "epoch": 1.079559326171875e-05,
      "step": 2830
    },
    {
      "action_loss_avg": 0.045634226873517036,
      "epoch": 1.079559326171875e-05,
      "step": 2830
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 2830,
      "training_step_time": 9.926523208618164
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 3.132774591445923,
      "step": 2830
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 2830,
      "training_step_time": 9.907200574874878
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 3.411715269088745,
      "step": 2830
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 2830,
      "training_step_time": 10.184812307357788
    },
    {
      "epoch": 1.079559326171875e-05,
      "model_forward_time": 3.130979537963867,
      "step": 2830
    },
    {
      "epoch": 1.079559326171875e-05,
      "step": 2830,
      "training_step_time": 10.318936347961426
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 3.1345391273498535,
      "step": 2831
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 2831,
      "training_step_time": 9.907694578170776
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 2.430535316467285,
      "step": 2831
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 2831,
      "training_step_time": 7.46123194694519
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 3.13494610786438,
      "step": 2831
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 2831,
      "training_step_time": 9.909624814987183
    },
    {
      "epoch": 1.0799407958984375e-05,
      "model_forward_time": 3.137467384338379,
      "step": 2831
    },
    {
      "epoch": 1.0799407958984375e-05,
      "step": 2831,
      "training_step_time": 10.328733682632446
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 3.1446266174316406,
      "step": 2832
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 2832,
      "training_step_time": 9.925278663635254
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 2.4330923557281494,
      "step": 2832
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 2832,
      "training_step_time": 7.468806266784668
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 3.134685516357422,
      "step": 2832
    },
    {
      "dynamics_loss_avg": 0.0685678031295538,
      "epoch": 1.080322265625e-05,
      "step": 2832
    },
    {
      "action_loss_avg": 0.04178057434037328,
      "epoch": 1.080322265625e-05,
      "step": 2832
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 2832,
      "training_step_time": 9.918671131134033
    },
    {
      "epoch": 1.080322265625e-05,
      "model_forward_time": 3.14156436920166,
      "step": 2832
    },
    {
      "epoch": 1.080322265625e-05,
      "step": 2832,
      "training_step_time": 10.274171590805054
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 3.151642084121704,
      "step": 2833
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 2833,
      "training_step_time": 10.197086095809937
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 3.1394248008728027,
      "step": 2833
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 2833,
      "training_step_time": 9.921122550964355
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 3.139880895614624,
      "step": 2833
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 2833,
      "training_step_time": 9.924983024597168
    },
    {
      "epoch": 1.0807037353515626e-05,
      "model_forward_time": 3.1429333686828613,
      "step": 2833
    },
    {
      "epoch": 1.0807037353515626e-05,
      "step": 2833,
      "training_step_time": 10.304769515991211
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 3.158292531967163,
      "step": 2834
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 2834,
      "training_step_time": 9.942220687866211
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 3.1373908519744873,
      "step": 2834
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 2834,
      "training_step_time": 10.194191217422485
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 2.4333364963531494,
      "step": 2834
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 2834,
      "training_step_time": 7.464569091796875
    },
    {
      "epoch": 1.081085205078125e-05,
      "model_forward_time": 3.1369056701660156,
      "step": 2834
    },
    {
      "epoch": 1.081085205078125e-05,
      "step": 2834,
      "training_step_time": 10.316572427749634
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 2.445636510848999,
      "step": 2835
    },
    {
      "dynamics_loss_avg": 0.06691143102943897,
      "epoch": 1.0814666748046875e-05,
      "step": 2835
    },
    {
      "action_loss_avg": 0.03608781397342682,
      "epoch": 1.0814666748046875e-05,
      "step": 2835
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 2835,
      "training_step_time": 7.478442907333374
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 3.133166790008545,
      "step": 2835
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 2835,
      "training_step_time": 9.907622337341309
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 3.13857102394104,
      "step": 2835
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 2835,
      "training_step_time": 9.917763471603394
    },
    {
      "epoch": 1.0814666748046875e-05,
      "model_forward_time": 3.4344704151153564,
      "step": 2835
    },
    {
      "epoch": 1.0814666748046875e-05,
      "step": 2835,
      "training_step_time": 10.307234525680542
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 3.152914047241211,
      "step": 2836
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 2836,
      "training_step_time": 9.931176900863647
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 3.1390700340270996,
      "step": 2836
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 2836,
      "training_step_time": 9.919819831848145
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 3.146582841873169,
      "step": 2836
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 2836,
      "training_step_time": 9.931897163391113
    },
    {
      "epoch": 1.08184814453125e-05,
      "model_forward_time": 2.433757781982422,
      "step": 2836
    },
    {
      "epoch": 1.08184814453125e-05,
      "step": 2836,
      "training_step_time": 10.044946432113647
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 3.4578919410705566,
      "step": 2837
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 2837,
      "training_step_time": 10.241399049758911
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 3.13531756401062,
      "step": 2837
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 2837,
      "training_step_time": 9.913593292236328
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 3.1365678310394287,
      "step": 2837
    },
    {
      "dynamics_loss_avg": 0.06596598271280527,
      "epoch": 1.0822296142578126e-05,
      "step": 2837
    },
    {
      "action_loss_avg": 0.04959312099963427,
      "epoch": 1.0822296142578126e-05,
      "step": 2837
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 2837,
      "training_step_time": 9.916706085205078
    },
    {
      "epoch": 1.0822296142578126e-05,
      "model_forward_time": 3.1317968368530273,
      "step": 2837
    },
    {
      "epoch": 1.0822296142578126e-05,
      "step": 2837,
      "training_step_time": 10.002506256103516
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 3.4254586696624756,
      "step": 2838
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 2838,
      "training_step_time": 10.201430797576904
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 2.4333012104034424,
      "step": 2838
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 2838,
      "training_step_time": 7.461150646209717
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 3.1336846351623535,
      "step": 2838
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 2838,
      "training_step_time": 9.908063173294067
    },
    {
      "epoch": 1.082611083984375e-05,
      "model_forward_time": 3.1286892890930176,
      "step": 2838
    },
    {
      "epoch": 1.082611083984375e-05,
      "step": 2838,
      "training_step_time": 10.016045331954956
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 3.1406490802764893,
      "step": 2839
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 2839,
      "training_step_time": 9.91821026802063
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 3.137894868850708,
      "step": 2839
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 2839,
      "training_step_time": 10.181614398956299
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 3.137688159942627,
      "step": 2839
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 2839,
      "training_step_time": 9.91355299949646
    },
    {
      "epoch": 1.0829925537109375e-05,
      "model_forward_time": 3.139163017272949,
      "step": 2839
    },
    {
      "epoch": 1.0829925537109375e-05,
      "step": 2839,
      "training_step_time": 10.27699875831604
    },
    {
      "epoch": 1.0833740234375e-05,
      "grad_norm": 0.6056073904037476,
      "learning_rate": 1e-05,
      "loss": 0.1034,
      "step": 2840
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 3.1435558795928955,
      "step": 2840
    },
    {
      "dynamics_loss_avg": 0.06926370598375797,
      "epoch": 1.0833740234375e-05,
      "step": 2840
    },
    {
      "action_loss_avg": 0.045249304920434955,
      "epoch": 1.0833740234375e-05,
      "step": 2840
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 2840,
      "training_step_time": 9.933480024337769
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 2.4360404014587402,
      "step": 2840
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 2840,
      "training_step_time": 7.470692873001099
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 3.1404337882995605,
      "step": 2840
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 2840,
      "training_step_time": 10.210137605667114
    },
    {
      "epoch": 1.0833740234375e-05,
      "model_forward_time": 3.1405742168426514,
      "step": 2840
    },
    {
      "epoch": 1.0833740234375e-05,
      "step": 2840,
      "training_step_time": 10.03691554069519
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 3.169044017791748,
      "step": 2841
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 2841,
      "training_step_time": 9.955144166946411
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 3.1458845138549805,
      "step": 2841
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 2841,
      "training_step_time": 9.929388284683228
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 2.4433271884918213,
      "step": 2841
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 2841,
      "training_step_time": 7.4793829917907715
    },
    {
      "epoch": 1.0837554931640626e-05,
      "model_forward_time": 3.148510217666626,
      "step": 2841
    },
    {
      "epoch": 1.0837554931640626e-05,
      "step": 2841,
      "training_step_time": 10.311556100845337
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 3.4502415657043457,
      "step": 2842
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 2842,
      "training_step_time": 10.242558717727661
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 3.152055025100708,
      "step": 2842
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 2842,
      "training_step_time": 9.92947793006897
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 3.145297050476074,
      "step": 2842
    },
    {
      "dynamics_loss_avg": 0.0709869934245944,
      "epoch": 1.084136962890625e-05,
      "step": 2842
    },
    {
      "action_loss_avg": 0.037048806063830854,
      "epoch": 1.084136962890625e-05,
      "step": 2842
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 2842,
      "training_step_time": 9.930513858795166
    },
    {
      "epoch": 1.084136962890625e-05,
      "model_forward_time": 3.1451969146728516,
      "step": 2842
    },
    {
      "epoch": 1.084136962890625e-05,
      "step": 2842,
      "training_step_time": 10.027260065078735
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 3.1704766750335693,
      "step": 2843
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 2843,
      "training_step_time": 10.265172004699707
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 2.441300392150879,
      "step": 2843
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 2843,
      "training_step_time": 7.485016345977783
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 3.1441524028778076,
      "step": 2843
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 2843,
      "training_step_time": 9.927501678466797
    },
    {
      "epoch": 1.0845184326171875e-05,
      "model_forward_time": 3.144223213195801,
      "step": 2843
    },
    {
      "epoch": 1.0845184326171875e-05,
      "step": 2843,
      "training_step_time": 10.241432428359985
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 3.153658151626587,
      "step": 2844
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 2844,
      "training_step_time": 9.93100881576538
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 3.4672422409057617,
      "step": 2844
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 2844,
      "training_step_time": 10.24488377571106
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 3.1394870281219482,
      "step": 2844
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 2844,
      "training_step_time": 9.919819593429565
    },
    {
      "epoch": 1.08489990234375e-05,
      "model_forward_time": 3.1500468254089355,
      "step": 2844
    },
    {
      "epoch": 1.08489990234375e-05,
      "step": 2844,
      "training_step_time": 10.270365715026855
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 3.153857707977295,
      "step": 2845
    },
    {
      "dynamics_loss_avg": 0.05734237898141146,
      "epoch": 1.0852813720703126e-05,
      "step": 2845
    },
    {
      "action_loss_avg": 0.026668108906596898,
      "epoch": 1.0852813720703126e-05,
      "step": 2845
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 2845,
      "training_step_time": 9.939284086227417
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 2.4355103969573975,
      "step": 2845
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 2845,
      "training_step_time": 7.466510534286499
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 3.1343259811401367,
      "step": 2845
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 2845,
      "training_step_time": 9.907989263534546
    },
    {
      "epoch": 1.0852813720703126e-05,
      "model_forward_time": 2.7088944911956787,
      "step": 2845
    },
    {
      "epoch": 1.0852813720703126e-05,
      "step": 2845,
      "training_step_time": 10.01601243019104
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 3.1529901027679443,
      "step": 2846
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 2846,
      "training_step_time": 9.93121337890625
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 3.1358025074005127,
      "step": 2846
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 2846,
      "training_step_time": 9.91024398803711
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 3.142042636871338,
      "step": 2846
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 2846,
      "training_step_time": 9.925788879394531
    },
    {
      "epoch": 1.085662841796875e-05,
      "model_forward_time": 3.4703969955444336,
      "step": 2846
    },
    {
      "epoch": 1.085662841796875e-05,
      "step": 2846,
      "training_step_time": 10.343538284301758
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 3.1592299938201904,
      "step": 2847
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 2847,
      "training_step_time": 9.94559907913208
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 3.1403322219848633,
      "step": 2847
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 2847,
      "training_step_time": 9.913883447647095
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 3.146634817123413,
      "step": 2847
    },
    {
      "dynamics_loss_avg": 0.06520072016865015,
      "epoch": 1.0860443115234375e-05,
      "step": 2847
    },
    {
      "action_loss_avg": 0.03738488107919693,
      "epoch": 1.0860443115234375e-05,
      "step": 2847
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 2847,
      "training_step_time": 9.92878270149231
    },
    {
      "epoch": 1.0860443115234375e-05,
      "model_forward_time": 2.4384684562683105,
      "step": 2847
    },
    {
      "epoch": 1.0860443115234375e-05,
      "step": 2847,
      "training_step_time": 10.35695219039917
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 3.1562674045562744,
      "step": 2848
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 2848,
      "training_step_time": 9.936772584915161
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 3.383202075958252,
      "step": 2848
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 2848,
      "training_step_time": 10.161442041397095
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 3.1439807415008545,
      "step": 2848
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 2848,
      "training_step_time": 9.919418573379517
    },
    {
      "epoch": 1.08642578125e-05,
      "model_forward_time": 3.139885663986206,
      "step": 2848
    },
    {
      "epoch": 1.08642578125e-05,
      "step": 2848,
      "training_step_time": 10.016765117645264
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 3.1669018268585205,
      "step": 2849
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 2849,
      "training_step_time": 9.96124005317688
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 3.3905725479125977,
      "step": 2849
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 2849,
      "training_step_time": 10.176812171936035
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 3.1397435665130615,
      "step": 2849
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 2849,
      "training_step_time": 9.920379638671875
    },
    {
      "epoch": 1.0868072509765626e-05,
      "model_forward_time": 3.1410365104675293,
      "step": 2849
    },
    {
      "epoch": 1.0868072509765626e-05,
      "step": 2849,
      "training_step_time": 10.338269233703613
    },
    {
      "epoch": 1.087188720703125e-05,
      "grad_norm": 0.5201898217201233,
      "learning_rate": 1e-05,
      "loss": 0.105,
      "step": 2850
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 3.1447532176971436,
      "step": 2850
    },
    {
      "dynamics_loss_avg": 0.06229447145015001,
      "epoch": 1.087188720703125e-05,
      "step": 2850
    },
    {
      "action_loss_avg": 0.026941556297242642,
      "epoch": 1.087188720703125e-05,
      "step": 2850
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 2850,
      "training_step_time": 9.928352117538452
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 2.4343326091766357,
      "step": 2850
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 2850,
      "training_step_time": 7.468113899230957
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 3.131425619125366,
      "step": 2850
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 2850,
      "training_step_time": 9.906095743179321
    },
    {
      "epoch": 1.087188720703125e-05,
      "model_forward_time": 3.4095964431762695,
      "step": 2850
    },
    {
      "epoch": 1.087188720703125e-05,
      "step": 2850,
      "training_step_time": 10.274513959884644
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 3.1479153633117676,
      "step": 2851
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 2851,
      "training_step_time": 9.922740459442139
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 3.1305317878723145,
      "step": 2851
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 2851,
      "training_step_time": 9.900737047195435
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 3.1283745765686035,
      "step": 2851
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 2851,
      "training_step_time": 9.900483131408691
    },
    {
      "epoch": 1.0875701904296875e-05,
      "model_forward_time": 3.134917736053467,
      "step": 2851
    },
    {
      "epoch": 1.0875701904296875e-05,
      "step": 2851,
      "training_step_time": 10.293896675109863
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 3.148280143737793,
      "step": 2852
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 2852,
      "training_step_time": 9.921217203140259
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 3.1324002742767334,
      "step": 2852
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 2852,
      "training_step_time": 9.904319524765015
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 3.1341261863708496,
      "step": 2852
    },
    {
      "dynamics_loss_avg": 0.05331238377839327,
      "epoch": 1.08795166015625e-05,
      "step": 2852
    },
    {
      "action_loss_avg": 0.03680834462866187,
      "epoch": 1.08795166015625e-05,
      "step": 2852
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 2852,
      "training_step_time": 9.91158390045166
    },
    {
      "epoch": 1.08795166015625e-05,
      "model_forward_time": 3.13478422164917,
      "step": 2852
    },
    {
      "epoch": 1.08795166015625e-05,
      "step": 2852,
      "training_step_time": 10.296797752380371
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 3.3983983993530273,
      "step": 2853
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 2853,
      "training_step_time": 10.179584741592407
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 3.1398537158966064,
      "step": 2853
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 2853,
      "training_step_time": 9.922120332717896
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 3.1366689205169678,
      "step": 2853
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 2853,
      "training_step_time": 9.914736270904541
    },
    {
      "epoch": 1.0883331298828126e-05,
      "model_forward_time": 3.137998580932617,
      "step": 2853
    },
    {
      "epoch": 1.0883331298828126e-05,
      "step": 2853,
      "training_step_time": 10.024864673614502
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 3.1500651836395264,
      "step": 2854
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 2854,
      "training_step_time": 9.927996397018433
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 3.137526750564575,
      "step": 2854
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 2854,
      "training_step_time": 10.19300651550293
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 3.135732889175415,
      "step": 2854
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 2854,
      "training_step_time": 9.916992425918579
    },
    {
      "epoch": 1.088714599609375e-05,
      "model_forward_time": 3.1372265815734863,
      "step": 2854
    },
    {
      "epoch": 1.088714599609375e-05,
      "step": 2854,
      "training_step_time": 10.345137357711792
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 3.1488983631134033,
      "step": 2855
    },
    {
      "dynamics_loss_avg": 0.053925682231783865,
      "epoch": 1.0890960693359375e-05,
      "step": 2855
    },
    {
      "action_loss_avg": 0.033300137240439655,
      "epoch": 1.0890960693359375e-05,
      "step": 2855
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 2855,
      "training_step_time": 9.927111148834229
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 3.1299431324005127,
      "step": 2855
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 2855,
      "training_step_time": 9.902600288391113
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 2.428262948989868,
      "step": 2855
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 2855,
      "training_step_time": 7.7324817180633545
    },
    {
      "epoch": 1.0890960693359375e-05,
      "model_forward_time": 3.136042833328247,
      "step": 2855
    },
    {
      "epoch": 1.0890960693359375e-05,
      "step": 2855,
      "training_step_time": 10.298918724060059
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 3.1499531269073486,
      "step": 2856
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 2856,
      "training_step_time": 9.927564859390259
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 3.1314103603363037,
      "step": 2856
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 2856,
      "training_step_time": 9.905717134475708
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 3.1357035636901855,
      "step": 2856
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 2856,
      "training_step_time": 9.90807318687439
    },
    {
      "epoch": 1.0894775390625e-05,
      "model_forward_time": 3.1333415508270264,
      "step": 2856
    },
    {
      "epoch": 1.0894775390625e-05,
      "step": 2856,
      "training_step_time": 10.303828001022339
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 3.148627758026123,
      "step": 2857
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 2857,
      "training_step_time": 9.923815965652466
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 3.1354904174804688,
      "step": 2857
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 2857,
      "training_step_time": 9.91470217704773
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 3.1403677463531494,
      "step": 2857
    },
    {
      "dynamics_loss_avg": 0.06321796756237745,
      "epoch": 1.0898590087890626e-05,
      "step": 2857
    },
    {
      "action_loss_avg": 0.0323825977742672,
      "epoch": 1.0898590087890626e-05,
      "step": 2857
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 2857,
      "training_step_time": 9.922770738601685
    },
    {
      "epoch": 1.0898590087890626e-05,
      "model_forward_time": 3.1345255374908447,
      "step": 2857
    },
    {
      "epoch": 1.0898590087890626e-05,
      "step": 2857,
      "training_step_time": 10.270167589187622
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 3.490344524383545,
      "step": 2858
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 2858,
      "training_step_time": 10.267561674118042
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 3.143237352371216,
      "step": 2858
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 2858,
      "training_step_time": 9.920269250869751
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 3.1483376026153564,
      "step": 2858
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 2858,
      "training_step_time": 9.931174755096436
    },
    {
      "epoch": 1.090240478515625e-05,
      "model_forward_time": 3.1440060138702393,
      "step": 2858
    },
    {
      "epoch": 1.090240478515625e-05,
      "step": 2858,
      "training_step_time": 10.2907395362854
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 3.1447911262512207,
      "step": 2859
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 2859,
      "training_step_time": 9.918198108673096
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 3.4439964294433594,
      "step": 2859
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 2859,
      "training_step_time": 10.213279485702515
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 3.1366074085235596,
      "step": 2859
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 2859,
      "training_step_time": 9.91863203048706
    },
    {
      "epoch": 1.0906219482421875e-05,
      "model_forward_time": 3.132362127304077,
      "step": 2859
    },
    {
      "epoch": 1.0906219482421875e-05,
      "step": 2859,
      "training_step_time": 10.255389928817749
    },
    {
      "epoch": 1.09100341796875e-05,
      "grad_norm": 0.2882874011993408,
      "learning_rate": 1e-05,
      "loss": 0.1033,
      "step": 2860
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 3.147711753845215,
      "step": 2860
    },
    {
      "dynamics_loss_avg": 0.07386703714728356,
      "epoch": 1.09100341796875e-05,
      "step": 2860
    },
    {
      "action_loss_avg": 0.041329798847436906,
      "epoch": 1.09100341796875e-05,
      "step": 2860
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 2860,
      "training_step_time": 9.937412738800049
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 3.137500762939453,
      "step": 2860
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 2860,
      "training_step_time": 9.910909414291382
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 3.144758939743042,
      "step": 2860
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 2860,
      "training_step_time": 10.218393564224243
    },
    {
      "epoch": 1.09100341796875e-05,
      "model_forward_time": 3.141303300857544,
      "step": 2860
    },
    {
      "epoch": 1.09100341796875e-05,
      "step": 2860,
      "training_step_time": 10.348165035247803
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 3.1461331844329834,
      "step": 2861
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 2861,
      "training_step_time": 9.925350666046143
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 3.137921094894409,
      "step": 2861
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 2861,
      "training_step_time": 9.91303563117981
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 3.1375980377197266,
      "step": 2861
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 2861,
      "training_step_time": 9.915022373199463
    },
    {
      "epoch": 1.0913848876953126e-05,
      "model_forward_time": 3.1438512802124023,
      "step": 2861
    },
    {
      "epoch": 1.0913848876953126e-05,
      "step": 2861,
      "training_step_time": 10.292781591415405
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 3.1525418758392334,
      "step": 2862
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 2862,
      "training_step_time": 9.926947116851807
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 2.4281437397003174,
      "step": 2862
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 2862,
      "training_step_time": 7.451989650726318
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 3.125481605529785,
      "step": 2862
    },
    {
      "dynamics_loss_avg": 0.06285290457308293,
      "epoch": 1.091766357421875e-05,
      "step": 2862
    },
    {
      "action_loss_avg": 0.032782052829861644,
      "epoch": 1.091766357421875e-05,
      "step": 2862
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 2862,
      "training_step_time": 9.893725395202637
    },
    {
      "epoch": 1.091766357421875e-05,
      "model_forward_time": 3.1364247798919678,
      "step": 2862
    },
    {
      "epoch": 1.091766357421875e-05,
      "step": 2862,
      "training_step_time": 10.318617582321167
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 3.146472692489624,
      "step": 2863
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 2863,
      "training_step_time": 10.223594903945923
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 3.1342759132385254,
      "step": 2863
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 2863,
      "training_step_time": 9.904028177261353
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 2.4341518878936768,
      "step": 2863
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 2863,
      "training_step_time": 7.465641498565674
    },
    {
      "epoch": 1.0921478271484375e-05,
      "model_forward_time": 3.1332926750183105,
      "step": 2863
    },
    {
      "epoch": 1.0921478271484375e-05,
      "step": 2863,
      "training_step_time": 10.266026735305786
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 3.1422271728515625,
      "step": 2864
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 2864,
      "training_step_time": 9.915544509887695
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 3.4260613918304443,
      "step": 2864
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 2864,
      "training_step_time": 10.19861912727356
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 3.1351921558380127,
      "step": 2864
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 2864,
      "training_step_time": 9.907265186309814
    },
    {
      "epoch": 1.092529296875e-05,
      "model_forward_time": 3.138643264770508,
      "step": 2864
    },
    {
      "epoch": 1.092529296875e-05,
      "step": 2864,
      "training_step_time": 10.005015134811401
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 3.1460516452789307,
      "step": 2865
    },
    {
      "dynamics_loss_avg": 0.05934129916131496,
      "epoch": 1.0929107666015626e-05,
      "step": 2865
    },
    {
      "action_loss_avg": 0.042104412242770194,
      "epoch": 1.0929107666015626e-05,
      "step": 2865
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 2865,
      "training_step_time": 9.923711061477661
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 3.1279585361480713,
      "step": 2865
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 2865,
      "training_step_time": 9.905133485794067
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 3.1359894275665283,
      "step": 2865
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 2865,
      "training_step_time": 10.177388906478882
    },
    {
      "epoch": 1.0929107666015626e-05,
      "model_forward_time": 2.435133218765259,
      "step": 2865
    },
    {
      "epoch": 1.0929107666015626e-05,
      "step": 2865,
      "training_step_time": 10.249876499176025
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 3.148224115371704,
      "step": 2866
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 2866,
      "training_step_time": 9.921907663345337
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 3.1336312294006348,
      "step": 2866
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 2866,
      "training_step_time": 9.909584760665894
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 3.134918689727783,
      "step": 2866
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 2866,
      "training_step_time": 9.909082889556885
    },
    {
      "epoch": 1.093292236328125e-05,
      "model_forward_time": 3.431528329849243,
      "step": 2866
    },
    {
      "epoch": 1.093292236328125e-05,
      "step": 2866,
      "training_step_time": 10.301826477050781
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 3.147916316986084,
      "step": 2867
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 2867,
      "training_step_time": 9.92851185798645
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 3.136420249938965,
      "step": 2867
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 2867,
      "training_step_time": 9.913622379302979
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 3.1420466899871826,
      "step": 2867
    },
    {
      "dynamics_loss_avg": 0.05632781693711877,
      "epoch": 1.0936737060546875e-05,
      "step": 2867
    },
    {
      "action_loss_avg": 0.03436037003993988,
      "epoch": 1.0936737060546875e-05,
      "step": 2867
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 2867,
      "training_step_time": 9.923184156417847
    },
    {
      "epoch": 1.0936737060546875e-05,
      "model_forward_time": 2.432961940765381,
      "step": 2867
    },
    {
      "epoch": 1.0936737060546875e-05,
      "step": 2867,
      "training_step_time": 10.311676979064941
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 3.1411826610565186,
      "step": 2868
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 2868,
      "training_step_time": 9.915758848190308
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 3.376598596572876,
      "step": 2868
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 2868,
      "training_step_time": 10.145134210586548
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 3.136672019958496,
      "step": 2868
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 2868,
      "training_step_time": 9.910185813903809
    },
    {
      "epoch": 1.09405517578125e-05,
      "model_forward_time": 3.136610269546509,
      "step": 2868
    },
    {
      "epoch": 1.09405517578125e-05,
      "step": 2868,
      "training_step_time": 10.31843113899231
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 3.1582753658294678,
      "step": 2869
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 2869,
      "training_step_time": 9.934616804122925
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 3.4304420948028564,
      "step": 2869
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 2869,
      "training_step_time": 10.203298568725586
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 2.4325549602508545,
      "step": 2869
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 2869,
      "training_step_time": 7.460471868515015
    },
    {
      "epoch": 1.0944366455078126e-05,
      "model_forward_time": 3.1299970149993896,
      "step": 2869
    },
    {
      "epoch": 1.0944366455078126e-05,
      "step": 2869,
      "training_step_time": 10.315991401672363
    },
    {
      "epoch": 1.094818115234375e-05,
      "grad_norm": 0.7508499622344971,
      "learning_rate": 1e-05,
      "loss": 0.1032,
      "step": 2870
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 2.448978900909424,
      "step": 2870
    },
    {
      "dynamics_loss_avg": 0.06425228305160999,
      "epoch": 1.094818115234375e-05,
      "step": 2870
    },
    {
      "action_loss_avg": 0.04219314455986023,
      "epoch": 1.094818115234375e-05,
      "step": 2870
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 2870,
      "training_step_time": 7.484085321426392
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 3.1396079063415527,
      "step": 2870
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 2870,
      "training_step_time": 9.910754919052124
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 2.436673641204834,
      "step": 2870
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 2870,
      "training_step_time": 7.469143629074097
    },
    {
      "epoch": 1.094818115234375e-05,
      "model_forward_time": 3.4050827026367188,
      "step": 2870
    },
    {
      "epoch": 1.094818115234375e-05,
      "step": 2870,
      "training_step_time": 10.270593404769897
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 3.1522326469421387,
      "step": 2871
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 2871,
      "training_step_time": 9.924817085266113
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 3.1399221420288086,
      "step": 2871
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 2871,
      "training_step_time": 9.914073944091797
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 3.1363353729248047,
      "step": 2871
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 2871,
      "training_step_time": 9.911128282546997
    },
    {
      "epoch": 1.0951995849609375e-05,
      "model_forward_time": 3.1370506286621094,
      "step": 2871
    },
    {
      "epoch": 1.0951995849609375e-05,
      "step": 2871,
      "training_step_time": 10.272594451904297
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 2.7583792209625244,
      "step": 2872
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 2872,
      "training_step_time": 7.792218446731567
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 3.142078161239624,
      "step": 2872
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 2872,
      "training_step_time": 9.92239785194397
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 3.138596773147583,
      "step": 2872
    },
    {
      "dynamics_loss_avg": 0.06283891685307026,
      "epoch": 1.0955810546875e-05,
      "step": 2872
    },
    {
      "action_loss_avg": 0.0375947643071413,
      "epoch": 1.0955810546875e-05,
      "step": 2872
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 2872,
      "training_step_time": 9.920207738876343
    },
    {
      "epoch": 1.0955810546875e-05,
      "model_forward_time": 3.1437346935272217,
      "step": 2872
    },
    {
      "epoch": 1.0955810546875e-05,
      "step": 2872,
      "training_step_time": 10.295897245407104
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 3.156733512878418,
      "step": 2873
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 2873,
      "training_step_time": 9.940292596817017
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 3.4117395877838135,
      "step": 2873
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 2873,
      "training_step_time": 10.190905570983887
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 3.137871026992798,
      "step": 2873
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 2873,
      "training_step_time": 9.916503190994263
    },
    {
      "epoch": 1.0959625244140626e-05,
      "model_forward_time": 3.1409430503845215,
      "step": 2873
    },
    {
      "epoch": 1.0959625244140626e-05,
      "step": 2873,
      "training_step_time": 10.283411502838135
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 3.1438701152801514,
      "step": 2874
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 2874,
      "training_step_time": 9.937306642532349
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 3.148303985595703,
      "step": 2874
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 2874,
      "training_step_time": 10.226850748062134
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 3.1384682655334473,
      "step": 2874
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 2874,
      "training_step_time": 9.917006254196167
    },
    {
      "epoch": 1.096343994140625e-05,
      "model_forward_time": 3.147223949432373,
      "step": 2874
    },
    {
      "epoch": 1.096343994140625e-05,
      "step": 2874,
      "training_step_time": 10.283357620239258
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 3.158332347869873,
      "step": 2875
    },
    {
      "dynamics_loss_avg": 0.062706208974123,
      "epoch": 1.0967254638671875e-05,
      "step": 2875
    },
    {
      "action_loss_avg": 0.040025169402360915,
      "epoch": 1.0967254638671875e-05,
      "step": 2875
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 2875,
      "training_step_time": 9.939221620559692
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 3.1412837505340576,
      "step": 2875
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 2875,
      "training_step_time": 9.914468050003052
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 3.4185266494750977,
      "step": 2875
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 2875,
      "training_step_time": 10.194783449172974
    },
    {
      "epoch": 1.0967254638671875e-05,
      "model_forward_time": 2.43727970123291,
      "step": 2875
    },
    {
      "epoch": 1.0967254638671875e-05,
      "step": 2875,
      "training_step_time": 10.012008905410767
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 3.1552224159240723,
      "step": 2876
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 2876,
      "training_step_time": 9.931660652160645
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 3.1415085792541504,
      "step": 2876
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 2876,
      "training_step_time": 9.917314767837524
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 3.147663116455078,
      "step": 2876
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 2876,
      "training_step_time": 9.924431324005127
    },
    {
      "epoch": 1.09710693359375e-05,
      "model_forward_time": 3.1453585624694824,
      "step": 2876
    },
    {
      "epoch": 1.09710693359375e-05,
      "step": 2876,
      "training_step_time": 10.287277936935425
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 3.1621851921081543,
      "step": 2877
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 2877,
      "training_step_time": 9.954710006713867
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 3.1396796703338623,
      "step": 2877
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 2877,
      "training_step_time": 9.91688847541809
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 3.1410138607025146,
      "step": 2877
    },
    {
      "dynamics_loss_avg": 0.07284118980169296,
      "epoch": 1.0974884033203124e-05,
      "step": 2877
    },
    {
      "action_loss_avg": 0.053347364999353886,
      "epoch": 1.0974884033203124e-05,
      "step": 2877
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 2877,
      "training_step_time": 9.92586898803711
    },
    {
      "epoch": 1.0974884033203124e-05,
      "model_forward_time": 3.140913963317871,
      "step": 2877
    },
    {
      "epoch": 1.0974884033203124e-05,
      "step": 2877,
      "training_step_time": 10.307555198669434
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 3.4362010955810547,
      "step": 2878
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 2878,
      "training_step_time": 10.210358381271362
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 3.1364669799804688,
      "step": 2878
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 2878,
      "training_step_time": 9.917614221572876
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 3.1430349349975586,
      "step": 2878
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 2878,
      "training_step_time": 9.917455196380615
    },
    {
      "epoch": 1.097869873046875e-05,
      "model_forward_time": 3.1421308517456055,
      "step": 2878
    },
    {
      "epoch": 1.097869873046875e-05,
      "step": 2878,
      "training_step_time": 10.016579151153564
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 3.1477909088134766,
      "step": 2879
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 2879,
      "training_step_time": 9.924813747406006
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 2.4330594539642334,
      "step": 2879
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 2879,
      "training_step_time": 7.771086931228638
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 2.4321908950805664,
      "step": 2879
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 2879,
      "training_step_time": 7.456318140029907
    },
    {
      "epoch": 1.0982513427734375e-05,
      "model_forward_time": 3.13287615776062,
      "step": 2879
    },
    {
      "epoch": 1.0982513427734375e-05,
      "step": 2879,
      "training_step_time": 10.028104066848755
    },
    {
      "epoch": 1.0986328125e-05,
      "grad_norm": 0.7356587648391724,
      "learning_rate": 1e-05,
      "loss": 0.1034,
      "step": 2880
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 3.1400322914123535,
      "step": 2880
    },
    {
      "dynamics_loss_avg": 0.06765929237008095,
      "epoch": 1.0986328125e-05,
      "step": 2880
    },
    {
      "action_loss_avg": 0.04259555358439684,
      "epoch": 1.0986328125e-05,
      "step": 2880
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 2880,
      "training_step_time": 9.915894746780396
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 3.1354150772094727,
      "step": 2880
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 2880,
      "training_step_time": 9.905406713485718
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 3.1320900917053223,
      "step": 2880
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 2880,
      "training_step_time": 10.203813076019287
    },
    {
      "epoch": 1.0986328125e-05,
      "model_forward_time": 3.1339950561523438,
      "step": 2880
    },
    {
      "epoch": 1.0986328125e-05,
      "step": 2880,
      "training_step_time": 10.287283897399902
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 3.1507580280303955,
      "step": 2881
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 2881,
      "training_step_time": 9.92501449584961
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 3.1280517578125,
      "step": 2881
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 2881,
      "training_step_time": 9.893947124481201
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 3.1290743350982666,
      "step": 2881
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 2881,
      "training_step_time": 9.893385887145996
    },
    {
      "epoch": 1.0990142822265624e-05,
      "model_forward_time": 3.1301395893096924,
      "step": 2881
    },
    {
      "epoch": 1.0990142822265624e-05,
      "step": 2881,
      "training_step_time": 10.2913498878479
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 3.138418436050415,
      "step": 2882
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 2882,
      "training_step_time": 9.906483888626099
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 3.1289210319519043,
      "step": 2882
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 2882,
      "training_step_time": 9.89260721206665
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 3.1300363540649414,
      "step": 2882
    },
    {
      "dynamics_loss_avg": 0.0708017809316516,
      "epoch": 1.099395751953125e-05,
      "step": 2882
    },
    {
      "action_loss_avg": 0.03601567819714546,
      "epoch": 1.099395751953125e-05,
      "step": 2882
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 2882,
      "training_step_time": 9.901953220367432
    },
    {
      "epoch": 1.099395751953125e-05,
      "model_forward_time": 3.132801055908203,
      "step": 2882
    },
    {
      "epoch": 1.099395751953125e-05,
      "step": 2882,
      "training_step_time": 10.306401014328003
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 3.1478142738342285,
      "step": 2883
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 2883,
      "training_step_time": 10.200697898864746
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 3.125432252883911,
      "step": 2883
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 2883,
      "training_step_time": 9.897041320800781
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 3.1298365592956543,
      "step": 2883
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 2883,
      "training_step_time": 9.90110731124878
    },
    {
      "epoch": 1.0997772216796875e-05,
      "model_forward_time": 3.1359219551086426,
      "step": 2883
    },
    {
      "epoch": 1.0997772216796875e-05,
      "step": 2883,
      "training_step_time": 10.035961627960205
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 3.1519174575805664,
      "step": 2884
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 2884,
      "training_step_time": 9.928489685058594
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 3.415722608566284,
      "step": 2884
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 2884,
      "training_step_time": 10.19128966331482
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 2.4342355728149414,
      "step": 2884
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 2884,
      "training_step_time": 7.4659082889556885
    },
    {
      "epoch": 1.10015869140625e-05,
      "model_forward_time": 3.1359612941741943,
      "step": 2884
    },
    {
      "epoch": 1.10015869140625e-05,
      "step": 2884,
      "training_step_time": 10.042771816253662
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 3.1530921459198,
      "step": 2885
    },
    {
      "dynamics_loss_avg": 0.05736753772944212,
      "epoch": 1.1005401611328124e-05,
      "step": 2885
    },
    {
      "action_loss_avg": 0.028614044655114412,
      "epoch": 1.1005401611328124e-05,
      "step": 2885
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 2885,
      "training_step_time": 9.933945178985596
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 3.1421501636505127,
      "step": 2885
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 2885,
      "training_step_time": 9.930785179138184
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 3.1433987617492676,
      "step": 2885
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 2885,
      "training_step_time": 9.9262056350708
    },
    {
      "epoch": 1.1005401611328124e-05,
      "model_forward_time": 3.43456768989563,
      "step": 2885
    },
    {
      "epoch": 1.1005401611328124e-05,
      "step": 2885,
      "training_step_time": 10.309634447097778
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 2.447303295135498,
      "step": 2886
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 2886,
      "training_step_time": 7.4795098304748535
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 3.131021022796631,
      "step": 2886
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 2886,
      "training_step_time": 9.908197402954102
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 3.1343626976013184,
      "step": 2886
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 2886,
      "training_step_time": 9.90680456161499
    },
    {
      "epoch": 1.100921630859375e-05,
      "model_forward_time": 2.4354817867279053,
      "step": 2886
    },
    {
      "epoch": 1.100921630859375e-05,
      "step": 2886,
      "training_step_time": 10.27542233467102
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 3.4605209827423096,
      "step": 2887
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 2887,
      "training_step_time": 10.244948387145996
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 3.1254427433013916,
      "step": 2887
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 2887,
      "training_step_time": 9.898898601531982
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 3.1402318477630615,
      "step": 2887
    },
    {
      "dynamics_loss_avg": 0.0676951553672552,
      "epoch": 1.1013031005859375e-05,
      "step": 2887
    },
    {
      "action_loss_avg": 0.047139761969447136,
      "epoch": 1.1013031005859375e-05,
      "step": 2887
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 2887,
      "training_step_time": 9.920037984848022
    },
    {
      "epoch": 1.1013031005859375e-05,
      "model_forward_time": 3.133930206298828,
      "step": 2887
    },
    {
      "epoch": 1.1013031005859375e-05,
      "step": 2887,
      "training_step_time": 10.291567087173462
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 3.1394927501678467,
      "step": 2888
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 2888,
      "training_step_time": 10.196668148040771
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 3.12772536277771,
      "step": 2888
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 2888,
      "training_step_time": 9.902785778045654
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 3.1298046112060547,
      "step": 2888
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 2888,
      "training_step_time": 9.897279739379883
    },
    {
      "epoch": 1.1016845703125e-05,
      "model_forward_time": 3.1306233406066895,
      "step": 2888
    },
    {
      "epoch": 1.1016845703125e-05,
      "step": 2888,
      "training_step_time": 10.018895387649536
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 3.1530814170837402,
      "step": 2889
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 2889,
      "training_step_time": 9.92844820022583
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 2.716506242752075,
      "step": 2889
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 2889,
      "training_step_time": 7.742491245269775
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 3.133030652999878,
      "step": 2889
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 2889,
      "training_step_time": 9.904386758804321
    },
    {
      "epoch": 1.1020660400390624e-05,
      "model_forward_time": 3.135931968688965,
      "step": 2889
    },
    {
      "epoch": 1.1020660400390624e-05,
      "step": 2889,
      "training_step_time": 10.342055797576904
    },
    {
      "epoch": 1.102447509765625e-05,
      "grad_norm": 0.5250608921051025,
      "learning_rate": 1e-05,
      "loss": 0.1015,
      "step": 2890
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 3.151723623275757,
      "step": 2890
    },
    {
      "dynamics_loss_avg": 0.05343616493046284,
      "epoch": 1.102447509765625e-05,
      "step": 2890
    },
    {
      "action_loss_avg": 0.03475840054452419,
      "epoch": 1.102447509765625e-05,
      "step": 2890
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 2890,
      "training_step_time": 9.92532205581665
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 3.135106325149536,
      "step": 2890
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 2890,
      "training_step_time": 9.905629873275757
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 3.134996175765991,
      "step": 2890
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 2890,
      "training_step_time": 10.174025297164917
    },
    {
      "epoch": 1.102447509765625e-05,
      "model_forward_time": 3.132293224334717,
      "step": 2890
    },
    {
      "epoch": 1.102447509765625e-05,
      "step": 2890,
      "training_step_time": 10.27894139289856
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 3.1378045082092285,
      "step": 2891
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 2891,
      "training_step_time": 9.910142421722412
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 3.132059335708618,
      "step": 2891
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 2891,
      "training_step_time": 9.899274587631226
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 3.132525682449341,
      "step": 2891
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 2891,
      "training_step_time": 9.895139932632446
    },
    {
      "epoch": 1.1028289794921875e-05,
      "model_forward_time": 3.1309103965759277,
      "step": 2891
    },
    {
      "epoch": 1.1028289794921875e-05,
      "step": 2891,
      "training_step_time": 10.316775560379028
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 3.1538145542144775,
      "step": 2892
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 2892,
      "training_step_time": 9.923182249069214
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 3.1367361545562744,
      "step": 2892
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 2892,
      "training_step_time": 9.906687498092651
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 3.132814884185791,
      "step": 2892
    },
    {
      "dynamics_loss_avg": 0.05423874277621508,
      "epoch": 1.10321044921875e-05,
      "step": 2892
    },
    {
      "action_loss_avg": 0.03863012585788965,
      "epoch": 1.10321044921875e-05,
      "step": 2892
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 2892,
      "training_step_time": 9.90204405784607
    },
    {
      "epoch": 1.10321044921875e-05,
      "model_forward_time": 3.1353509426116943,
      "step": 2892
    },
    {
      "epoch": 1.10321044921875e-05,
      "step": 2892,
      "training_step_time": 10.261035203933716
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 3.4086012840270996,
      "step": 2893
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 2893,
      "training_step_time": 10.177467584609985
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 3.140946388244629,
      "step": 2893
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 2893,
      "training_step_time": 9.907540798187256
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 3.1323721408843994,
      "step": 2893
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 2893,
      "training_step_time": 9.902874231338501
    },
    {
      "epoch": 1.1035919189453124e-05,
      "model_forward_time": 3.133627414703369,
      "step": 2893
    },
    {
      "epoch": 1.1035919189453124e-05,
      "step": 2893,
      "training_step_time": 10.007102727890015
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 3.158845901489258,
      "step": 2894
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 2894,
      "training_step_time": 9.933505773544312
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 3.1299173831939697,
      "step": 2894
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 2894,
      "training_step_time": 10.219342231750488
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 3.133700370788574,
      "step": 2894
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 2894,
      "training_step_time": 9.90847110748291
    },
    {
      "epoch": 1.103973388671875e-05,
      "model_forward_time": 3.1403238773345947,
      "step": 2894
    },
    {
      "epoch": 1.103973388671875e-05,
      "step": 2894,
      "training_step_time": 10.01098346710205
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 3.1534624099731445,
      "step": 2895
    },
    {
      "dynamics_loss_avg": 0.04612298663705587,
      "epoch": 1.1043548583984375e-05,
      "step": 2895
    },
    {
      "action_loss_avg": 0.021473506325855853,
      "epoch": 1.1043548583984375e-05,
      "step": 2895
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 2895,
      "training_step_time": 9.934646606445312
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 3.151893138885498,
      "step": 2895
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 2895,
      "training_step_time": 9.932042837142944
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 3.4076361656188965,
      "step": 2895
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 2895,
      "training_step_time": 10.180622339248657
    },
    {
      "epoch": 1.1043548583984375e-05,
      "model_forward_time": 2.4332926273345947,
      "step": 2895
    },
    {
      "epoch": 1.1043548583984375e-05,
      "step": 2895,
      "training_step_time": 10.28624439239502
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 3.1546754837036133,
      "step": 2896
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 2896,
      "training_step_time": 9.924527645111084
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 2.4365389347076416,
      "step": 2896
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 2896,
      "training_step_time": 7.462287425994873
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 3.137148380279541,
      "step": 2896
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 2896,
      "training_step_time": 9.901847839355469
    },
    {
      "epoch": 1.104736328125e-05,
      "model_forward_time": 2.429565191268921,
      "step": 2896
    },
    {
      "epoch": 1.104736328125e-05,
      "step": 2896,
      "training_step_time": 10.011768102645874
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 3.474407911300659,
      "step": 2897
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 2897,
      "training_step_time": 10.239673614501953
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 3.1365957260131836,
      "step": 2897
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 2897,
      "training_step_time": 9.903909683227539
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 3.1399495601654053,
      "step": 2897
    },
    {
      "dynamics_loss_avg": 0.06064648050814867,
      "epoch": 1.1051177978515624e-05,
      "step": 2897
    },
    {
      "action_loss_avg": 0.03568856376223266,
      "epoch": 1.1051177978515624e-05,
      "step": 2897
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 2897,
      "training_step_time": 9.913619995117188
    },
    {
      "epoch": 1.1051177978515624e-05,
      "model_forward_time": 3.1408863067626953,
      "step": 2897
    },
    {
      "epoch": 1.1051177978515624e-05,
      "step": 2897,
      "training_step_time": 10.303773403167725
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 2.452026128768921,
      "step": 2898
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 2898,
      "training_step_time": 7.480639934539795
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 3.400158643722534,
      "step": 2898
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 2898,
      "training_step_time": 10.170451641082764
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 3.1424856185913086,
      "step": 2898
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 2898,
      "training_step_time": 9.912490844726562
    },
    {
      "epoch": 1.105499267578125e-05,
      "model_forward_time": 3.138741970062256,
      "step": 2898
    },
    {
      "epoch": 1.105499267578125e-05,
      "step": 2898,
      "training_step_time": 10.030196189880371
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 3.143817186355591,
      "step": 2899
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 2899,
      "training_step_time": 9.91279673576355
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 2.4312756061553955,
      "step": 2899
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 2899,
      "training_step_time": 7.458133935928345
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 2.701490640640259,
      "step": 2899
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 2899,
      "training_step_time": 7.7276036739349365
    },
    {
      "epoch": 1.1058807373046875e-05,
      "model_forward_time": 2.4258992671966553,
      "step": 2899
    },
    {
      "epoch": 1.1058807373046875e-05,
      "step": 2899,
      "training_step_time": 10.014997005462646
    },
    {
      "epoch": 1.10626220703125e-05,
      "grad_norm": 0.7155885696411133,
      "learning_rate": 1e-05,
      "loss": 0.0992,
      "step": 2900
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 3.1521248817443848,
      "step": 2900
    },
    {
      "dynamics_loss_avg": 0.06737188454717398,
      "epoch": 1.10626220703125e-05,
      "step": 2900
    },
    {
      "action_loss_avg": 0.04052911098115146,
      "epoch": 1.10626220703125e-05,
      "step": 2900
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 2900,
      "training_step_time": 9.9212646484375
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 2.428457260131836,
      "step": 2900
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 2900,
      "training_step_time": 7.452075958251953
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 2.4285390377044678,
      "step": 2900
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 2900,
      "training_step_time": 7.451185464859009
    },
    {
      "epoch": 1.10626220703125e-05,
      "model_forward_time": 3.4295787811279297,
      "step": 2900
    },
    {
      "epoch": 1.10626220703125e-05,
      "step": 2900,
      "training_step_time": 10.31223201751709
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 3.140825033187866,
      "step": 2901
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 2901,
      "training_step_time": 9.908853769302368
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 3.131196975708008,
      "step": 2901
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 2901,
      "training_step_time": 9.897631883621216
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 3.1322624683380127,
      "step": 2901
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 2901,
      "training_step_time": 9.905324935913086
    },
    {
      "epoch": 1.1066436767578124e-05,
      "model_forward_time": 3.13409161567688,
      "step": 2901
    },
    {
      "epoch": 1.1066436767578124e-05,
      "step": 2901,
      "training_step_time": 10.311078310012817
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 3.415858030319214,
      "step": 2902
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 2902,
      "training_step_time": 10.187624454498291
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 3.134183406829834,
      "step": 2902
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 2902,
      "training_step_time": 9.90582799911499
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 3.133103370666504,
      "step": 2902
    },
    {
      "dynamics_loss_avg": 0.06594114080071449,
      "epoch": 1.107025146484375e-05,
      "step": 2902
    },
    {
      "action_loss_avg": 0.030592985358089207,
      "epoch": 1.107025146484375e-05,
      "step": 2902
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 2902,
      "training_step_time": 9.905513048171997
    },
    {
      "epoch": 1.107025146484375e-05,
      "model_forward_time": 3.1369593143463135,
      "step": 2902
    },
    {
      "epoch": 1.107025146484375e-05,
      "step": 2902,
      "training_step_time": 10.283888101577759
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 3.1460046768188477,
      "step": 2903
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 2903,
      "training_step_time": 9.919599056243896
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 3.129645586013794,
      "step": 2903
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 2903,
      "training_step_time": 10.185338020324707
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 3.13340425491333,
      "step": 2903
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 2903,
      "training_step_time": 9.912248134613037
    },
    {
      "epoch": 1.1074066162109375e-05,
      "model_forward_time": 3.1367688179016113,
      "step": 2903
    },
    {
      "epoch": 1.1074066162109375e-05,
      "step": 2903,
      "training_step_time": 10.293657779693604
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 3.148064613342285,
      "step": 2904
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 2904,
      "training_step_time": 9.927478790283203
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 3.1349077224731445,
      "step": 2904
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 2904,
      "training_step_time": 9.909978151321411
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 3.3957836627960205,
      "step": 2904
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 2904,
      "training_step_time": 10.168728113174438
    },
    {
      "epoch": 1.1077880859375e-05,
      "model_forward_time": 3.13470196723938,
      "step": 2904
    },
    {
      "epoch": 1.1077880859375e-05,
      "step": 2904,
      "training_step_time": 10.016947269439697
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 3.1414079666137695,
      "step": 2905
    },
    {
      "dynamics_loss_avg": 0.05629528108984232,
      "epoch": 1.1081695556640624e-05,
      "step": 2905
    },
    {
      "action_loss_avg": 0.02720695869065821,
      "epoch": 1.1081695556640624e-05,
      "step": 2905
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 2905,
      "training_step_time": 9.924713611602783
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 3.1323108673095703,
      "step": 2905
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 2905,
      "training_step_time": 9.904242038726807
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 3.1315534114837646,
      "step": 2905
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 2905,
      "training_step_time": 9.901554584503174
    },
    {
      "epoch": 1.1081695556640624e-05,
      "model_forward_time": 3.1278462409973145,
      "step": 2905
    },
    {
      "epoch": 1.1081695556640624e-05,
      "step": 2905,
      "training_step_time": 10.28251051902771
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 3.1466915607452393,
      "step": 2906
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 2906,
      "training_step_time": 9.922649383544922
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 3.1313180923461914,
      "step": 2906
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 2906,
      "training_step_time": 9.908029556274414
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 2.4330122470855713,
      "step": 2906
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 2906,
      "training_step_time": 7.467127323150635
    },
    {
      "epoch": 1.108551025390625e-05,
      "model_forward_time": 3.134824514389038,
      "step": 2906
    },
    {
      "epoch": 1.108551025390625e-05,
      "step": 2906,
      "training_step_time": 10.008408546447754
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 3.1454782485961914,
      "step": 2907
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 2907,
      "training_step_time": 10.208889722824097
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 2.435366153717041,
      "step": 2907
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 2907,
      "training_step_time": 7.474636077880859
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 3.135491371154785,
      "step": 2907
    },
    {
      "dynamics_loss_avg": 0.0570964589715004,
      "epoch": 1.1089324951171875e-05,
      "step": 2907
    },
    {
      "action_loss_avg": 0.03418782679364085,
      "epoch": 1.1089324951171875e-05,
      "step": 2907
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 2907,
      "training_step_time": 9.920660495758057
    },
    {
      "epoch": 1.1089324951171875e-05,
      "model_forward_time": 3.1405205726623535,
      "step": 2907
    },
    {
      "epoch": 1.1089324951171875e-05,
      "step": 2907,
      "training_step_time": 10.265318155288696
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 3.1522204875946045,
      "step": 2908
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 2908,
      "training_step_time": 9.93745756149292
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 3.1376583576202393,
      "step": 2908
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 2908,
      "training_step_time": 9.912692070007324
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 3.4303221702575684,
      "step": 2908
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 2908,
      "training_step_time": 10.201336145401001
    },
    {
      "epoch": 1.10931396484375e-05,
      "model_forward_time": 3.1338207721710205,
      "step": 2908
    },
    {
      "epoch": 1.10931396484375e-05,
      "step": 2908,
      "training_step_time": 10.28913950920105
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 3.1557252407073975,
      "step": 2909
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 2909,
      "training_step_time": 9.931257247924805
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 3.1353843212127686,
      "step": 2909
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 2909,
      "training_step_time": 9.910311460494995
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 3.4069628715515137,
      "step": 2909
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 2909,
      "training_step_time": 10.18071985244751
    },
    {
      "epoch": 1.1096954345703124e-05,
      "model_forward_time": 3.134742021560669,
      "step": 2909
    },
    {
      "epoch": 1.1096954345703124e-05,
      "step": 2909,
      "training_step_time": 10.303305387496948
    },
    {
      "epoch": 1.110076904296875e-05,
      "grad_norm": 0.6472618579864502,
      "learning_rate": 1e-05,
      "loss": 0.0954,
      "step": 2910
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 3.151035785675049,
      "step": 2910
    },
    {
      "dynamics_loss_avg": 0.06064935345202684,
      "epoch": 1.110076904296875e-05,
      "step": 2910
    },
    {
      "action_loss_avg": 0.043589742854237556,
      "epoch": 1.110076904296875e-05,
      "step": 2910
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 2910,
      "training_step_time": 9.93299913406372
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 3.1375911235809326,
      "step": 2910
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 2910,
      "training_step_time": 9.913349151611328
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 3.136723518371582,
      "step": 2910
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 2910,
      "training_step_time": 9.918008804321289
    },
    {
      "epoch": 1.110076904296875e-05,
      "model_forward_time": 2.4384467601776123,
      "step": 2910
    },
    {
      "epoch": 1.110076904296875e-05,
      "step": 2910,
      "training_step_time": 10.013032674789429
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 3.436195135116577,
      "step": 2911
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 2911,
      "training_step_time": 10.21416449546814
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 3.134629726409912,
      "step": 2911
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 2911,
      "training_step_time": 9.906862735748291
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 3.129793167114258,
      "step": 2911
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 2911,
      "training_step_time": 9.899321556091309
    },
    {
      "epoch": 1.1104583740234375e-05,
      "model_forward_time": 3.135796070098877,
      "step": 2911
    },
    {
      "epoch": 1.1104583740234375e-05,
      "step": 2911,
      "training_step_time": 10.008117437362671
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 3.4044151306152344,
      "step": 2912
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 2912,
      "training_step_time": 10.17952036857605
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 3.133718967437744,
      "step": 2912
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 2912,
      "training_step_time": 9.903532028198242
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 2.432241916656494,
      "step": 2912
    },
    {
      "dynamics_loss_avg": 0.07263497561216355,
      "epoch": 1.11083984375e-05,
      "step": 2912
    },
    {
      "action_loss_avg": 0.047859096247702834,
      "epoch": 1.11083984375e-05,
      "step": 2912
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 2912,
      "training_step_time": 7.464588642120361
    },
    {
      "epoch": 1.11083984375e-05,
      "model_forward_time": 3.133049488067627,
      "step": 2912
    },
    {
      "epoch": 1.11083984375e-05,
      "step": 2912,
      "training_step_time": 10.020846843719482
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 3.149681568145752,
      "step": 2913
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 2913,
      "training_step_time": 9.92919635772705
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 3.1325249671936035,
      "step": 2913
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 2913,
      "training_step_time": 9.905714988708496
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 3.381011962890625,
      "step": 2913
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 2913,
      "training_step_time": 10.159032583236694
    },
    {
      "epoch": 1.1112213134765624e-05,
      "model_forward_time": 3.1408581733703613,
      "step": 2913
    },
    {
      "epoch": 1.1112213134765624e-05,
      "step": 2913,
      "training_step_time": 10.308441162109375
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 3.151634693145752,
      "step": 2914
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 2914,
      "training_step_time": 9.946958780288696
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 3.1395769119262695,
      "step": 2914
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 2914,
      "training_step_time": 9.92162013053894
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 2.4357736110687256,
      "step": 2914
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 2914,
      "training_step_time": 7.471344709396362
    },
    {
      "epoch": 1.111602783203125e-05,
      "model_forward_time": 3.4495320320129395,
      "step": 2914
    },
    {
      "epoch": 1.111602783203125e-05,
      "step": 2914,
      "training_step_time": 10.322636365890503
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 3.156486988067627,
      "step": 2915
    },
    {
      "dynamics_loss_avg": 0.07639183346182107,
      "epoch": 1.1119842529296875e-05,
      "step": 2915
    },
    {
      "action_loss_avg": 0.04226126447319985,
      "epoch": 1.1119842529296875e-05,
      "step": 2915
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 2915,
      "training_step_time": 9.946852207183838
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 3.136845588684082,
      "step": 2915
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 2915,
      "training_step_time": 9.916213274002075
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 3.1419594287872314,
      "step": 2915
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 2915,
      "training_step_time": 9.916136503219604
    },
    {
      "epoch": 1.1119842529296875e-05,
      "model_forward_time": 3.455626964569092,
      "step": 2915
    },
    {
      "epoch": 1.1119842529296875e-05,
      "step": 2915,
      "training_step_time": 10.325077056884766
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 3.1491644382476807,
      "step": 2916
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 2916,
      "training_step_time": 9.932506322860718
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 3.1386632919311523,
      "step": 2916
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 2916,
      "training_step_time": 9.91016149520874
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 3.1403396129608154,
      "step": 2916
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 2916,
      "training_step_time": 9.912897109985352
    },
    {
      "epoch": 1.11236572265625e-05,
      "model_forward_time": 2.435260534286499,
      "step": 2916
    },
    {
      "epoch": 1.11236572265625e-05,
      "step": 2916,
      "training_step_time": 10.013249397277832
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 3.1496315002441406,
      "step": 2917
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 2917,
      "training_step_time": 9.928031206130981
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 3.430891990661621,
      "step": 2917
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 2917,
      "training_step_time": 10.201982975006104
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 3.1436069011688232,
      "step": 2917
    },
    {
      "dynamics_loss_avg": 0.07028598729521036,
      "epoch": 1.1127471923828124e-05,
      "step": 2917
    },
    {
      "action_loss_avg": 0.040109419263899326,
      "epoch": 1.1127471923828124e-05,
      "step": 2917
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 2917,
      "training_step_time": 9.914427518844604
    },
    {
      "epoch": 1.1127471923828124e-05,
      "model_forward_time": 3.1453731060028076,
      "step": 2917
    },
    {
      "epoch": 1.1127471923828124e-05,
      "step": 2917,
      "training_step_time": 10.318061828613281
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 3.1584370136260986,
      "step": 2918
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 2918,
      "training_step_time": 9.939024448394775
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 2.72078800201416,
      "step": 2918
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 2918,
      "training_step_time": 7.752122402191162
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 3.13971209526062,
      "step": 2918
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 2918,
      "training_step_time": 9.912276029586792
    },
    {
      "epoch": 1.113128662109375e-05,
      "model_forward_time": 3.138512372970581,
      "step": 2918
    },
    {
      "epoch": 1.113128662109375e-05,
      "step": 2918,
      "training_step_time": 10.300901412963867
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 3.1604602336883545,
      "step": 2919
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 2919,
      "training_step_time": 9.939366579055786
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 3.1428616046905518,
      "step": 2919
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 2919,
      "training_step_time": 9.915632486343384
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 3.1420559883117676,
      "step": 2919
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 2919,
      "training_step_time": 10.2147536277771
    },
    {
      "epoch": 1.1135101318359375e-05,
      "model_forward_time": 3.1403965950012207,
      "step": 2919
    },
    {
      "epoch": 1.1135101318359375e-05,
      "step": 2919,
      "training_step_time": 10.291322469711304
    },
    {
      "epoch": 1.1138916015625e-05,
      "grad_norm": 0.5531150698661804,
      "learning_rate": 1e-05,
      "loss": 0.1027,
      "step": 2920
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 3.160200357437134,
      "step": 2920
    },
    {
      "dynamics_loss_avg": 0.06425509825348855,
      "epoch": 1.1138916015625e-05,
      "step": 2920
    },
    {
      "action_loss_avg": 0.043196363002061845,
      "epoch": 1.1138916015625e-05,
      "step": 2920
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 2920,
      "training_step_time": 9.931591510772705
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 2.4354848861694336,
      "step": 2920
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 2920,
      "training_step_time": 7.460418701171875
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 3.1336910724639893,
      "step": 2920
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 2920,
      "training_step_time": 9.906055450439453
    },
    {
      "epoch": 1.1138916015625e-05,
      "model_forward_time": 2.431518793106079,
      "step": 2920
    },
    {
      "epoch": 1.1138916015625e-05,
      "step": 2920,
      "training_step_time": 10.01042127609253
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 3.1519808769226074,
      "step": 2921
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 2921,
      "training_step_time": 9.914169073104858
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 3.135671377182007,
      "step": 2921
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 2921,
      "training_step_time": 9.898358583450317
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 3.1328351497650146,
      "step": 2921
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 2921,
      "training_step_time": 9.892930746078491
    },
    {
      "epoch": 1.1142730712890624e-05,
      "model_forward_time": 3.135319709777832,
      "step": 2921
    },
    {
      "epoch": 1.1142730712890624e-05,
      "step": 2921,
      "training_step_time": 10.005643606185913
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 3.152514934539795,
      "step": 2922
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 2922,
      "training_step_time": 10.221179723739624
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 3.1396336555480957,
      "step": 2922
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 2922,
      "training_step_time": 9.903992176055908
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 3.1348156929016113,
      "step": 2922
    },
    {
      "dynamics_loss_avg": 0.051234662905335425,
      "epoch": 1.114654541015625e-05,
      "step": 2922
    },
    {
      "action_loss_avg": 0.029980878252536057,
      "epoch": 1.114654541015625e-05,
      "step": 2922
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 2922,
      "training_step_time": 9.905761957168579
    },
    {
      "epoch": 1.114654541015625e-05,
      "model_forward_time": 3.14212965965271,
      "step": 2922
    },
    {
      "epoch": 1.114654541015625e-05,
      "step": 2922,
      "training_step_time": 10.300380945205688
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 3.1566107273101807,
      "step": 2923
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 2923,
      "training_step_time": 9.922404527664185
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 3.138075590133667,
      "step": 2923
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 2923,
      "training_step_time": 10.201968431472778
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 3.1432626247406006,
      "step": 2923
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 2923,
      "training_step_time": 9.91651201248169
    },
    {
      "epoch": 1.1150360107421875e-05,
      "model_forward_time": 3.1444027423858643,
      "step": 2923
    },
    {
      "epoch": 1.1150360107421875e-05,
      "step": 2923,
      "training_step_time": 10.341630935668945
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 2.454149007797241,
      "step": 2924
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 2924,
      "training_step_time": 7.483712434768677
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 3.1383845806121826,
      "step": 2924
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 2924,
      "training_step_time": 9.905151128768921
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 3.1451897621154785,
      "step": 2924
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 2924,
      "training_step_time": 9.918063402175903
    },
    {
      "epoch": 1.11541748046875e-05,
      "model_forward_time": 2.7055916786193848,
      "step": 2924
    },
    {
      "epoch": 1.11541748046875e-05,
      "step": 2924,
      "training_step_time": 10.320290565490723
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 3.157930850982666,
      "step": 2925
    },
    {
      "dynamics_loss_avg": 0.07041432037949562,
      "epoch": 1.1157989501953124e-05,
      "step": 2925
    },
    {
      "action_loss_avg": 0.038440412003546955,
      "epoch": 1.1157989501953124e-05,
      "step": 2925
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 2925,
      "training_step_time": 9.934580326080322
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 3.1423897743225098,
      "step": 2925
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 2925,
      "training_step_time": 9.913973569869995
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 3.144547939300537,
      "step": 2925
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 2925,
      "training_step_time": 9.916379690170288
    },
    {
      "epoch": 1.1157989501953124e-05,
      "model_forward_time": 3.146164655685425,
      "step": 2925
    },
    {
      "epoch": 1.1157989501953124e-05,
      "step": 2925,
      "training_step_time": 10.328201055526733
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 2.7434887886047363,
      "step": 2926
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 2926,
      "training_step_time": 7.775670051574707
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 3.1402671337127686,
      "step": 2926
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 2926,
      "training_step_time": 9.914428949356079
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 2.4334659576416016,
      "step": 2926
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 2926,
      "training_step_time": 7.462919235229492
    },
    {
      "epoch": 1.116180419921875e-05,
      "model_forward_time": 3.137512445449829,
      "step": 2926
    },
    {
      "epoch": 1.116180419921875e-05,
      "step": 2926,
      "training_step_time": 10.031290292739868
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 3.1462156772613525,
      "step": 2927
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 2927,
      "training_step_time": 9.91702389717102
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 3.414155960083008,
      "step": 2927
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 2927,
      "training_step_time": 10.181955099105835
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 2.4377222061157227,
      "step": 2927
    },
    {
      "dynamics_loss_avg": 0.06337164407595992,
      "epoch": 1.1165618896484375e-05,
      "step": 2927
    },
    {
      "action_loss_avg": 0.032782961847260594,
      "epoch": 1.1165618896484375e-05,
      "step": 2927
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 2927,
      "training_step_time": 7.4705376625061035
    },
    {
      "epoch": 1.1165618896484375e-05,
      "model_forward_time": 3.1382734775543213,
      "step": 2927
    },
    {
      "epoch": 1.1165618896484375e-05,
      "step": 2927,
      "training_step_time": 10.324640989303589
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 3.1511621475219727,
      "step": 2928
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 2928,
      "training_step_time": 9.924442052841187
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 3.1491127014160156,
      "step": 2928
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 2928,
      "training_step_time": 9.927510023117065
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 3.4266555309295654,
      "step": 2928
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 2928,
      "training_step_time": 10.203649520874023
    },
    {
      "epoch": 1.116943359375e-05,
      "model_forward_time": 2.439934015274048,
      "step": 2928
    },
    {
      "epoch": 1.116943359375e-05,
      "step": 2928,
      "training_step_time": 10.032885074615479
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 3.162442207336426,
      "step": 2929
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 2929,
      "training_step_time": 9.943779945373535
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 3.150468349456787,
      "step": 2929
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 2929,
      "training_step_time": 9.927849769592285
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 3.146188259124756,
      "step": 2929
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 2929,
      "training_step_time": 9.919752597808838
    },
    {
      "epoch": 1.1173248291015624e-05,
      "model_forward_time": 3.391376495361328,
      "step": 2929
    },
    {
      "epoch": 1.1173248291015624e-05,
      "step": 2929,
      "training_step_time": 10.25966215133667
    },
    {
      "epoch": 1.117706298828125e-05,
      "grad_norm": 0.7420591711997986,
      "learning_rate": 1e-05,
      "loss": 0.0974,
      "step": 2930
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 3.1518619060516357,
      "step": 2930
    },
    {
      "dynamics_loss_avg": 0.06421385630965233,
      "epoch": 1.117706298828125e-05,
      "step": 2930
    },
    {
      "action_loss_avg": 0.04569007642567158,
      "epoch": 1.117706298828125e-05,
      "step": 2930
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 2930,
      "training_step_time": 9.927984237670898
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 3.1447956562042236,
      "step": 2930
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 2930,
      "training_step_time": 9.922120809555054
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 3.1423733234405518,
      "step": 2930
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 2930,
      "training_step_time": 9.914307355880737
    },
    {
      "epoch": 1.117706298828125e-05,
      "model_forward_time": 3.1412851810455322,
      "step": 2930
    },
    {
      "epoch": 1.117706298828125e-05,
      "step": 2930,
      "training_step_time": 10.285998344421387
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 3.4222395420074463,
      "step": 2931
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 2931,
      "training_step_time": 10.18328070640564
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 2.4397668838500977,
      "step": 2931
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 2931,
      "training_step_time": 7.458219051361084
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 3.1321616172790527,
      "step": 2931
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 2931,
      "training_step_time": 9.89868712425232
    },
    {
      "epoch": 1.1180877685546875e-05,
      "model_forward_time": 3.1351146697998047,
      "step": 2931
    },
    {
      "epoch": 1.1180877685546875e-05,
      "step": 2931,
      "training_step_time": 10.006116151809692
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 3.4327633380889893,
      "step": 2932
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 2932,
      "training_step_time": 10.19902753829956
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 3.1347451210021973,
      "step": 2932
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 2932,
      "training_step_time": 9.901209354400635
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 3.133324146270752,
      "step": 2932
    },
    {
      "dynamics_loss_avg": 0.054187438637018207,
      "epoch": 1.11846923828125e-05,
      "step": 2932
    },
    {
      "action_loss_avg": 0.03438342418521643,
      "epoch": 1.11846923828125e-05,
      "step": 2932
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 2932,
      "training_step_time": 9.904518127441406
    },
    {
      "epoch": 1.11846923828125e-05,
      "model_forward_time": 3.1374502182006836,
      "step": 2932
    },
    {
      "epoch": 1.11846923828125e-05,
      "step": 2932,
      "training_step_time": 10.015795707702637
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 3.1751699447631836,
      "step": 2933
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 2933,
      "training_step_time": 9.948107481002808
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 3.1402390003204346,
      "step": 2933
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 2933,
      "training_step_time": 10.213806390762329
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 2.4367339611053467,
      "step": 2933
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 2933,
      "training_step_time": 7.464476823806763
    },
    {
      "epoch": 1.1188507080078124e-05,
      "model_forward_time": 3.142728090286255,
      "step": 2933
    },
    {
      "epoch": 1.1188507080078124e-05,
      "step": 2933,
      "training_step_time": 10.319163799285889
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 3.1679978370666504,
      "step": 2934
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 2934,
      "training_step_time": 9.940402746200562
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 3.1415257453918457,
      "step": 2934
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 2934,
      "training_step_time": 9.910131454467773
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 3.1438207626342773,
      "step": 2934
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 2934,
      "training_step_time": 10.225367307662964
    },
    {
      "epoch": 1.119232177734375e-05,
      "model_forward_time": 3.138526201248169,
      "step": 2934
    },
    {
      "epoch": 1.119232177734375e-05,
      "step": 2934,
      "training_step_time": 10.28835940361023
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 3.1474149227142334,
      "step": 2935
    },
    {
      "dynamics_loss_avg": 0.05904736109077931,
      "epoch": 1.1196136474609375e-05,
      "step": 2935
    },
    {
      "action_loss_avg": 0.028880254738032816,
      "epoch": 1.1196136474609375e-05,
      "step": 2935
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 2935,
      "training_step_time": 9.923716068267822
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 3.1342084407806396,
      "step": 2935
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 2935,
      "training_step_time": 9.89456033706665
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 3.1447598934173584,
      "step": 2935
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 2935,
      "training_step_time": 9.919394731521606
    },
    {
      "epoch": 1.1196136474609375e-05,
      "model_forward_time": 3.434169054031372,
      "step": 2935
    },
    {
      "epoch": 1.1196136474609375e-05,
      "step": 2935,
      "training_step_time": 10.301614999771118
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 3.1540560722351074,
      "step": 2936
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 2936,
      "training_step_time": 9.92868709564209
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 3.1325085163116455,
      "step": 2936
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 2936,
      "training_step_time": 9.904541730880737
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 2.4371531009674072,
      "step": 2936
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 2936,
      "training_step_time": 7.465317487716675
    },
    {
      "epoch": 1.1199951171875e-05,
      "model_forward_time": 3.1326043605804443,
      "step": 2936
    },
    {
      "epoch": 1.1199951171875e-05,
      "step": 2936,
      "training_step_time": 10.339526176452637
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 3.1402432918548584,
      "step": 2937
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 2937,
      "training_step_time": 9.913159132003784
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 3.3757808208465576,
      "step": 2937
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 2937,
      "training_step_time": 10.147274732589722
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 3.1401896476745605,
      "step": 2937
    },
    {
      "dynamics_loss_avg": 0.04882647804915905,
      "epoch": 1.1203765869140624e-05,
      "step": 2937
    },
    {
      "action_loss_avg": 0.028328933473676442,
      "epoch": 1.1203765869140624e-05,
      "step": 2937
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 2937,
      "training_step_time": 9.918190002441406
    },
    {
      "epoch": 1.1203765869140624e-05,
      "model_forward_time": 3.1371233463287354,
      "step": 2937
    },
    {
      "epoch": 1.1203765869140624e-05,
      "step": 2937,
      "training_step_time": 10.029950618743896
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 3.1434147357940674,
      "step": 2938
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 2938,
      "training_step_time": 9.919577360153198
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 3.394777297973633,
      "step": 2938
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 2938,
      "training_step_time": 10.167920589447021
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 3.1440773010253906,
      "step": 2938
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 2938,
      "training_step_time": 9.92020297050476
    },
    {
      "epoch": 1.120758056640625e-05,
      "model_forward_time": 3.143503189086914,
      "step": 2938
    },
    {
      "epoch": 1.120758056640625e-05,
      "step": 2938,
      "training_step_time": 10.040313482284546
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 3.1487512588500977,
      "step": 2939
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 2939,
      "training_step_time": 9.926697015762329
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 2.4366424083709717,
      "step": 2939
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 2939,
      "training_step_time": 7.466686725616455
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 3.132699728012085,
      "step": 2939
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 2939,
      "training_step_time": 9.910158157348633
    },
    {
      "epoch": 1.1211395263671875e-05,
      "model_forward_time": 2.7303740978240967,
      "step": 2939
    },
    {
      "epoch": 1.1211395263671875e-05,
      "step": 2939,
      "training_step_time": 10.320820808410645
    },
    {
      "epoch": 1.12152099609375e-05,
      "grad_norm": 0.38278844952583313,
      "learning_rate": 1e-05,
      "loss": 0.0958,
      "step": 2940
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 3.1396079063415527,
      "step": 2940
    },
    {
      "dynamics_loss_avg": 0.05702754883095622,
      "epoch": 1.12152099609375e-05,
      "step": 2940
    },
    {
      "action_loss_avg": 0.026871816534549,
      "epoch": 1.12152099609375e-05,
      "step": 2940
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 2940,
      "training_step_time": 9.917227745056152
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 3.137348175048828,
      "step": 2940
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 2940,
      "training_step_time": 9.913792610168457
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 3.1366970539093018,
      "step": 2940
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 2940,
      "training_step_time": 9.910653829574585
    },
    {
      "epoch": 1.12152099609375e-05,
      "model_forward_time": 3.3913023471832275,
      "step": 2940
    },
    {
      "epoch": 1.12152099609375e-05,
      "step": 2940,
      "training_step_time": 10.264962911605835
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 3.1471846103668213,
      "step": 2941
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 2941,
      "training_step_time": 9.919012546539307
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 3.1340460777282715,
      "step": 2941
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 2941,
      "training_step_time": 9.910302877426147
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 3.1358563899993896,
      "step": 2941
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 2941,
      "training_step_time": 9.914305925369263
    },
    {
      "epoch": 1.1219024658203124e-05,
      "model_forward_time": 2.433795213699341,
      "step": 2941
    },
    {
      "epoch": 1.1219024658203124e-05,
      "step": 2941,
      "training_step_time": 10.28412675857544
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 3.1414785385131836,
      "step": 2942
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 2942,
      "training_step_time": 9.910380363464355
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 3.4036943912506104,
      "step": 2942
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 2942,
      "training_step_time": 10.17451524734497
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 3.1390953063964844,
      "step": 2942
    },
    {
      "dynamics_loss_avg": 0.05550707075744867,
      "epoch": 1.122283935546875e-05,
      "step": 2942
    },
    {
      "action_loss_avg": 0.030099086975678802,
      "epoch": 1.122283935546875e-05,
      "step": 2942
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 2942,
      "training_step_time": 9.917371034622192
    },
    {
      "epoch": 1.122283935546875e-05,
      "model_forward_time": 3.1374690532684326,
      "step": 2942
    },
    {
      "epoch": 1.122283935546875e-05,
      "step": 2942,
      "training_step_time": 10.022104978561401
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 3.1635141372680664,
      "step": 2943
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 2943,
      "training_step_time": 9.94758939743042
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 2.4375057220458984,
      "step": 2943
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 2943,
      "training_step_time": 7.756821870803833
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 3.133509635925293,
      "step": 2943
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 2943,
      "training_step_time": 9.89723014831543
    },
    {
      "epoch": 1.1226654052734375e-05,
      "model_forward_time": 3.1399085521698,
      "step": 2943
    },
    {
      "epoch": 1.1226654052734375e-05,
      "step": 2943,
      "training_step_time": 10.272074460983276
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 3.156174898147583,
      "step": 2944
    },
    {
      "epoch": 1.123046875e-05,
      "step": 2944,
      "training_step_time": 9.931123733520508
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 3.1337740421295166,
      "step": 2944
    },
    {
      "epoch": 1.123046875e-05,
      "step": 2944,
      "training_step_time": 9.898377656936646
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 3.4183473587036133,
      "step": 2944
    },
    {
      "epoch": 1.123046875e-05,
      "step": 2944,
      "training_step_time": 10.181877613067627
    },
    {
      "epoch": 1.123046875e-05,
      "model_forward_time": 2.4259395599365234,
      "step": 2944
    },
    {
      "epoch": 1.123046875e-05,
      "step": 2944,
      "training_step_time": 10.010799407958984
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 3.1485178470611572,
      "step": 2945
    },
    {
      "dynamics_loss_avg": 0.06690078638494015,
      "epoch": 1.1234283447265624e-05,
      "step": 2945
    },
    {
      "action_loss_avg": 0.03774795029312372,
      "epoch": 1.1234283447265624e-05,
      "step": 2945
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 2945,
      "training_step_time": 9.912297487258911
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 3.1299192905426025,
      "step": 2945
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 2945,
      "training_step_time": 9.892396450042725
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 3.1276817321777344,
      "step": 2945
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 2945,
      "training_step_time": 9.89241909980774
    },
    {
      "epoch": 1.1234283447265624e-05,
      "model_forward_time": 3.1303915977478027,
      "step": 2945
    },
    {
      "epoch": 1.1234283447265624e-05,
      "step": 2945,
      "training_step_time": 10.284948110580444
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 3.1456685066223145,
      "step": 2946
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 2946,
      "training_step_time": 9.914099931716919
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 3.128368616104126,
      "step": 2946
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 2946,
      "training_step_time": 9.893835306167603
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 3.129953622817993,
      "step": 2946
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 2946,
      "training_step_time": 9.895369052886963
    },
    {
      "epoch": 1.123809814453125e-05,
      "model_forward_time": 3.1316988468170166,
      "step": 2946
    },
    {
      "epoch": 1.123809814453125e-05,
      "step": 2946,
      "training_step_time": 10.312772274017334
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 3.423288583755493,
      "step": 2947
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 2947,
      "training_step_time": 10.190403461456299
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 3.1331214904785156,
      "step": 2947
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 2947,
      "training_step_time": 9.905977725982666
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 3.1341471672058105,
      "step": 2947
    },
    {
      "dynamics_loss_avg": 0.04966528434306383,
      "epoch": 1.1241912841796875e-05,
      "step": 2947
    },
    {
      "action_loss_avg": 0.03169052628800273,
      "epoch": 1.1241912841796875e-05,
      "step": 2947
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 2947,
      "training_step_time": 9.905425548553467
    },
    {
      "epoch": 1.1241912841796875e-05,
      "model_forward_time": 3.132884979248047,
      "step": 2947
    },
    {
      "epoch": 1.1241912841796875e-05,
      "step": 2947,
      "training_step_time": 10.280348539352417
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 3.1439223289489746,
      "step": 2948
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 2948,
      "training_step_time": 9.913174629211426
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 3.399934768676758,
      "step": 2948
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 2948,
      "training_step_time": 10.164734363555908
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 3.1292788982391357,
      "step": 2948
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 2948,
      "training_step_time": 9.897386074066162
    },
    {
      "epoch": 1.12457275390625e-05,
      "model_forward_time": 3.1337034702301025,
      "step": 2948
    },
    {
      "epoch": 1.12457275390625e-05,
      "step": 2948,
      "training_step_time": 10.012876749038696
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 3.1393356323242188,
      "step": 2949
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 2949,
      "training_step_time": 9.906617164611816
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 3.132843494415283,
      "step": 2949
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 2949,
      "training_step_time": 9.900666236877441
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 3.410876750946045,
      "step": 2949
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 2949,
      "training_step_time": 10.180083990097046
    },
    {
      "epoch": 1.1249542236328124e-05,
      "model_forward_time": 2.4352777004241943,
      "step": 2949
    },
    {
      "epoch": 1.1249542236328124e-05,
      "step": 2949,
      "training_step_time": 10.011881351470947
    },
    {
      "epoch": 1.125335693359375e-05,
      "grad_norm": 0.28752097487449646,
      "learning_rate": 1e-05,
      "loss": 0.0947,
      "step": 2950
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 3.164280652999878,
      "step": 2950
    },
    {
      "dynamics_loss_avg": 0.05819915309548378,
      "epoch": 1.125335693359375e-05,
      "step": 2950
    },
    {
      "action_loss_avg": 0.029656291659921407,
      "epoch": 1.125335693359375e-05,
      "step": 2950
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 2950,
      "training_step_time": 9.943567752838135
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 3.138862133026123,
      "step": 2950
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 2950,
      "training_step_time": 9.905759572982788
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 3.1334128379821777,
      "step": 2950
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 2950,
      "training_step_time": 9.902891874313354
    },
    {
      "epoch": 1.125335693359375e-05,
      "model_forward_time": 3.1352035999298096,
      "step": 2950
    },
    {
      "epoch": 1.125335693359375e-05,
      "step": 2950,
      "training_step_time": 10.30088496208191
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 3.1491568088531494,
      "step": 2951
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 2951,
      "training_step_time": 9.92192554473877
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 3.1347038745880127,
      "step": 2951
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 2951,
      "training_step_time": 9.902220964431763
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 3.1445446014404297,
      "step": 2951
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 2951,
      "training_step_time": 9.921291589736938
    },
    {
      "epoch": 1.1257171630859375e-05,
      "model_forward_time": 3.139415979385376,
      "step": 2951
    },
    {
      "epoch": 1.1257171630859375e-05,
      "step": 2951,
      "training_step_time": 10.239983558654785
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 3.411306619644165,
      "step": 2952
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 2952,
      "training_step_time": 10.183990478515625
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 2.4286141395568848,
      "step": 2952
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 2952,
      "training_step_time": 7.449498176574707
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 3.1279091835021973,
      "step": 2952
    },
    {
      "dynamics_loss_avg": 0.061190034821629526,
      "epoch": 1.1260986328125e-05,
      "step": 2952
    },
    {
      "action_loss_avg": 0.03998452238738537,
      "epoch": 1.1260986328125e-05,
      "step": 2952
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 2952,
      "training_step_time": 9.894271850585938
    },
    {
      "epoch": 1.1260986328125e-05,
      "model_forward_time": 2.4286155700683594,
      "step": 2952
    },
    {
      "epoch": 1.1260986328125e-05,
      "step": 2952,
      "training_step_time": 10.300140857696533
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 3.1535885334014893,
      "step": 2953
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 2953,
      "training_step_time": 9.922061204910278
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 2.4308032989501953,
      "step": 2953
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 2953,
      "training_step_time": 7.455625772476196
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 2.6775641441345215,
      "step": 2953
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 2953,
      "training_step_time": 7.696714878082275
    },
    {
      "epoch": 1.1264801025390624e-05,
      "model_forward_time": 3.131499767303467,
      "step": 2953
    },
    {
      "epoch": 1.1264801025390624e-05,
      "step": 2953,
      "training_step_time": 10.00310492515564
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 3.15167236328125,
      "step": 2954
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 2954,
      "training_step_time": 9.918078422546387
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 3.1350347995758057,
      "step": 2954
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 2954,
      "training_step_time": 9.896687746047974
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 3.1343071460723877,
      "step": 2954
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 2954,
      "training_step_time": 9.897668838500977
    },
    {
      "epoch": 1.126861572265625e-05,
      "model_forward_time": 3.402106285095215,
      "step": 2954
    },
    {
      "epoch": 1.126861572265625e-05,
      "step": 2954,
      "training_step_time": 10.258843421936035
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 3.146949529647827,
      "step": 2955
    },
    {
      "dynamics_loss_avg": 0.07323035262525082,
      "epoch": 1.1272430419921875e-05,
      "step": 2955
    },
    {
      "action_loss_avg": 0.04571073856204748,
      "epoch": 1.1272430419921875e-05,
      "step": 2955
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 2955,
      "training_step_time": 9.916468858718872
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 3.130709409713745,
      "step": 2955
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 2955,
      "training_step_time": 9.904614210128784
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 3.1359620094299316,
      "step": 2955
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 2955,
      "training_step_time": 9.912463188171387
    },
    {
      "epoch": 1.1272430419921875e-05,
      "model_forward_time": 3.137321949005127,
      "step": 2955
    },
    {
      "epoch": 1.1272430419921875e-05,
      "step": 2955,
      "training_step_time": 10.316466808319092
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 3.4618937969207764,
      "step": 2956
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 2956,
      "training_step_time": 10.247853994369507
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 3.1469476222991943,
      "step": 2956
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 2956,
      "training_step_time": 9.930006265640259
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 3.145157814025879,
      "step": 2956
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 2956,
      "training_step_time": 9.93398642539978
    },
    {
      "epoch": 1.12762451171875e-05,
      "model_forward_time": 3.1439242362976074,
      "step": 2956
    },
    {
      "epoch": 1.12762451171875e-05,
      "step": 2956,
      "training_step_time": 10.310309648513794
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 3.1540117263793945,
      "step": 2957
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 2957,
      "training_step_time": 9.943094491958618
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 3.4274401664733887,
      "step": 2957
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 2957,
      "training_step_time": 10.20737099647522
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 3.1403164863586426,
      "step": 2957
    },
    {
      "dynamics_loss_avg": 0.07001547031104564,
      "epoch": 1.1280059814453124e-05,
      "step": 2957
    },
    {
      "action_loss_avg": 0.04167840145528316,
      "epoch": 1.1280059814453124e-05,
      "step": 2957
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 2957,
      "training_step_time": 9.921072244644165
    },
    {
      "epoch": 1.1280059814453124e-05,
      "model_forward_time": 3.1365363597869873,
      "step": 2957
    },
    {
      "epoch": 1.1280059814453124e-05,
      "step": 2957,
      "training_step_time": 10.319575548171997
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 3.158560276031494,
      "step": 2958
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 2958,
      "training_step_time": 9.941056489944458
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 2.7574408054351807,
      "step": 2958
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 2958,
      "training_step_time": 7.793590545654297
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 2.435835838317871,
      "step": 2958
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 2958,
      "training_step_time": 7.473236083984375
    },
    {
      "epoch": 1.128387451171875e-05,
      "model_forward_time": 3.1404266357421875,
      "step": 2958
    },
    {
      "epoch": 1.128387451171875e-05,
      "step": 2958,
      "training_step_time": 10.293140411376953
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 3.1571662425994873,
      "step": 2959
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 2959,
      "training_step_time": 9.93931770324707
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 3.1452696323394775,
      "step": 2959
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 2959,
      "training_step_time": 9.92965054512024
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 3.1476526260375977,
      "step": 2959
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 2959,
      "training_step_time": 9.928328275680542
    },
    {
      "epoch": 1.1287689208984375e-05,
      "model_forward_time": 3.432615280151367,
      "step": 2959
    },
    {
      "epoch": 1.1287689208984375e-05,
      "step": 2959,
      "training_step_time": 10.311278104782104
    },
    {
      "epoch": 1.129150390625e-05,
      "grad_norm": 0.42714378237724304,
      "learning_rate": 1e-05,
      "loss": 0.0972,
      "step": 2960
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 3.1636252403259277,
      "step": 2960
    },
    {
      "dynamics_loss_avg": 0.06590876840054989,
      "epoch": 1.129150390625e-05,
      "step": 2960
    },
    {
      "action_loss_avg": 0.035975085385143754,
      "epoch": 1.129150390625e-05,
      "step": 2960
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 2960,
      "training_step_time": 9.948516607284546
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 3.1418445110321045,
      "step": 2960
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 2960,
      "training_step_time": 9.923415660858154
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 3.141383171081543,
      "step": 2960
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 2960,
      "training_step_time": 9.920750617980957
    },
    {
      "epoch": 1.129150390625e-05,
      "model_forward_time": 2.7164955139160156,
      "step": 2960
    },
    {
      "epoch": 1.129150390625e-05,
      "step": 2960,
      "training_step_time": 10.294906854629517
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 3.1525216102600098,
      "step": 2961
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 2961,
      "training_step_time": 9.934410333633423
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 3.1396279335021973,
      "step": 2961
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 2961,
      "training_step_time": 9.917587757110596
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 3.138411521911621,
      "step": 2961
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 2961,
      "training_step_time": 9.912268161773682
    },
    {
      "epoch": 1.1295318603515624e-05,
      "model_forward_time": 2.435425281524658,
      "step": 2961
    },
    {
      "epoch": 1.1295318603515624e-05,
      "step": 2961,
      "training_step_time": 10.29371428489685
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 3.156904935836792,
      "step": 2962
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 2962,
      "training_step_time": 9.930448293685913
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 3.4421634674072266,
      "step": 2962
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 2962,
      "training_step_time": 10.21185302734375
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 3.130922317504883,
      "step": 2962
    },
    {
      "dynamics_loss_avg": 0.062396423704922196,
      "epoch": 1.129913330078125e-05,
      "step": 2962
    },
    {
      "action_loss_avg": 0.04022564766928553,
      "epoch": 1.129913330078125e-05,
      "step": 2962
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 2962,
      "training_step_time": 9.90488862991333
    },
    {
      "epoch": 1.129913330078125e-05,
      "model_forward_time": 3.13887882232666,
      "step": 2962
    },
    {
      "epoch": 1.129913330078125e-05,
      "step": 2962,
      "training_step_time": 10.33215069770813
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 3.1486475467681885,
      "step": 2963
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 2963,
      "training_step_time": 9.923021793365479
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 2.430955648422241,
      "step": 2963
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 2963,
      "training_step_time": 7.73411226272583
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 3.130176067352295,
      "step": 2963
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 2963,
      "training_step_time": 9.902128458023071
    },
    {
      "epoch": 1.1302947998046875e-05,
      "model_forward_time": 3.1359949111938477,
      "step": 2963
    },
    {
      "epoch": 1.1302947998046875e-05,
      "step": 2963,
      "training_step_time": 10.258431196212769
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 3.149446487426758,
      "step": 2964
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 2964,
      "training_step_time": 9.925322771072388
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 3.1358511447906494,
      "step": 2964
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 2964,
      "training_step_time": 9.91213846206665
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 3.3957841396331787,
      "step": 2964
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 2964,
      "training_step_time": 10.170457363128662
    },
    {
      "epoch": 1.13067626953125e-05,
      "model_forward_time": 3.1434621810913086,
      "step": 2964
    },
    {
      "epoch": 1.13067626953125e-05,
      "step": 2964,
      "training_step_time": 10.017140626907349
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 3.153071403503418,
      "step": 2965
    },
    {
      "dynamics_loss_avg": 0.06273202802985907,
      "epoch": 1.1310577392578124e-05,
      "step": 2965
    },
    {
      "action_loss_avg": 0.03820869717746973,
      "epoch": 1.1310577392578124e-05,
      "step": 2965
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 2965,
      "training_step_time": 9.931849002838135
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 3.1462063789367676,
      "step": 2965
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 2965,
      "training_step_time": 9.923576593399048
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 3.159428834915161,
      "step": 2965
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 2965,
      "training_step_time": 9.937610387802124
    },
    {
      "epoch": 1.1310577392578124e-05,
      "model_forward_time": 3.151789426803589,
      "step": 2965
    },
    {
      "epoch": 1.1310577392578124e-05,
      "step": 2965,
      "training_step_time": 10.340551137924194
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 3.15443754196167,
      "step": 2966
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 2966,
      "training_step_time": 9.936688423156738
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 3.1368510723114014,
      "step": 2966
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 2966,
      "training_step_time": 9.913299322128296
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 3.1398260593414307,
      "step": 2966
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 2966,
      "training_step_time": 9.917434930801392
    },
    {
      "epoch": 1.131439208984375e-05,
      "model_forward_time": 3.149150848388672,
      "step": 2966
    },
    {
      "epoch": 1.131439208984375e-05,
      "step": 2966,
      "training_step_time": 10.024399042129517
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 3.4379165172576904,
      "step": 2967
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 2967,
      "training_step_time": 10.228168725967407
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 3.1368281841278076,
      "step": 2967
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 2967,
      "training_step_time": 9.913062810897827
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 3.1413698196411133,
      "step": 2967
    },
    {
      "dynamics_loss_avg": 0.046849146485328674,
      "epoch": 1.1318206787109375e-05,
      "step": 2967
    },
    {
      "action_loss_avg": 0.02818349516019225,
      "epoch": 1.1318206787109375e-05,
      "step": 2967
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 2967,
      "training_step_time": 9.921181440353394
    },
    {
      "epoch": 1.1318206787109375e-05,
      "model_forward_time": 3.1383838653564453,
      "step": 2967
    },
    {
      "epoch": 1.1318206787109375e-05,
      "step": 2967,
      "training_step_time": 10.318339347839355
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 3.1399483680725098,
      "step": 2968
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 2968,
      "training_step_time": 9.91231393814087
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 3.1450092792510986,
      "step": 2968
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 2968,
      "training_step_time": 10.238053560256958
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 3.1416471004486084,
      "step": 2968
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 2968,
      "training_step_time": 9.911332607269287
    },
    {
      "epoch": 1.1322021484375e-05,
      "model_forward_time": 3.1388537883758545,
      "step": 2968
    },
    {
      "epoch": 1.1322021484375e-05,
      "step": 2968,
      "training_step_time": 10.302805423736572
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 3.1600141525268555,
      "step": 2969
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 2969,
      "training_step_time": 9.936127185821533
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 3.140519857406616,
      "step": 2969
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 2969,
      "training_step_time": 9.905738830566406
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 3.423795223236084,
      "step": 2969
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 2969,
      "training_step_time": 10.188035011291504
    },
    {
      "epoch": 1.1325836181640624e-05,
      "model_forward_time": 2.4393014907836914,
      "step": 2969
    },
    {
      "epoch": 1.1325836181640624e-05,
      "step": 2969,
      "training_step_time": 10.299825668334961
    },
    {
      "epoch": 1.132965087890625e-05,
      "grad_norm": 0.7491216659545898,
      "learning_rate": 1e-05,
      "loss": 0.1,
      "step": 2970
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 3.151158332824707,
      "step": 2970
    },
    {
      "dynamics_loss_avg": 0.05361614041030407,
      "epoch": 1.132965087890625e-05,
      "step": 2970
    },
    {
      "action_loss_avg": 0.03325283974409103,
      "epoch": 1.132965087890625e-05,
      "step": 2970
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 2970,
      "training_step_time": 9.917858839035034
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 3.135517120361328,
      "step": 2970
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 2970,
      "training_step_time": 9.902478456497192
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 3.137481689453125,
      "step": 2970
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 2970,
      "training_step_time": 9.908394813537598
    },
    {
      "epoch": 1.132965087890625e-05,
      "model_forward_time": 3.1428515911102295,
      "step": 2970
    },
    {
      "epoch": 1.132965087890625e-05,
      "step": 2970,
      "training_step_time": 10.271909952163696
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 3.154831886291504,
      "step": 2971
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 2971,
      "training_step_time": 9.935567617416382
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 3.140932083129883,
      "step": 2971
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 2971,
      "training_step_time": 9.912575483322144
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 3.140695095062256,
      "step": 2971
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 2971,
      "training_step_time": 9.911429405212402
    },
    {
      "epoch": 1.1333465576171875e-05,
      "model_forward_time": 3.1446709632873535,
      "step": 2971
    },
    {
      "epoch": 1.1333465576171875e-05,
      "step": 2971,
      "training_step_time": 10.226043224334717
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 2.7558937072753906,
      "step": 2972
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 2972,
      "training_step_time": 7.781442880630493
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 3.1358556747436523,
      "step": 2972
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 2972,
      "training_step_time": 9.906166076660156
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 2.439429998397827,
      "step": 2972
    },
    {
      "dynamics_loss_avg": 0.06940462477505208,
      "epoch": 1.13372802734375e-05,
      "step": 2972
    },
    {
      "action_loss_avg": 0.036864242143929006,
      "epoch": 1.13372802734375e-05,
      "step": 2972
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 2972,
      "training_step_time": 7.471483469009399
    },
    {
      "epoch": 1.13372802734375e-05,
      "model_forward_time": 3.1464450359344482,
      "step": 2972
    },
    {
      "epoch": 1.13372802734375e-05,
      "step": 2972,
      "training_step_time": 10.010700702667236
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 3.154527425765991,
      "step": 2973
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 2973,
      "training_step_time": 9.92453122138977
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 3.1434519290924072,
      "step": 2973
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 2973,
      "training_step_time": 9.91236162185669
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 3.393136501312256,
      "step": 2973
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 2973,
      "training_step_time": 10.1626558303833
    },
    {
      "epoch": 1.1341094970703124e-05,
      "model_forward_time": 3.1400468349456787,
      "step": 2973
    },
    {
      "epoch": 1.1341094970703124e-05,
      "step": 2973,
      "training_step_time": 10.296424627304077
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 3.1484599113464355,
      "step": 2974
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 2974,
      "training_step_time": 9.918954133987427
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 3.138157844543457,
      "step": 2974
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 2974,
      "training_step_time": 9.907996892929077
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 3.1396141052246094,
      "step": 2974
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 2974,
      "training_step_time": 10.186907768249512
    },
    {
      "epoch": 1.134490966796875e-05,
      "model_forward_time": 3.1411006450653076,
      "step": 2974
    },
    {
      "epoch": 1.134490966796875e-05,
      "step": 2974,
      "training_step_time": 10.293256282806396
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 3.142878532409668,
      "step": 2975
    },
    {
      "dynamics_loss_avg": 0.06454079058021307,
      "epoch": 1.1348724365234375e-05,
      "step": 2975
    },
    {
      "action_loss_avg": 0.034032698906958106,
      "epoch": 1.1348724365234375e-05,
      "step": 2975
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 2975,
      "training_step_time": 9.917736530303955
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 3.1296334266662598,
      "step": 2975
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 2975,
      "training_step_time": 9.909537315368652
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 3.132110834121704,
      "step": 2975
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 2975,
      "training_step_time": 9.893821001052856
    },
    {
      "epoch": 1.1348724365234375e-05,
      "model_forward_time": 3.4097704887390137,
      "step": 2975
    },
    {
      "epoch": 1.1348724365234375e-05,
      "step": 2975,
      "training_step_time": 10.266742467880249
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 3.140605926513672,
      "step": 2976
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 2976,
      "training_step_time": 9.906437397003174
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 3.1312522888183594,
      "step": 2976
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 2976,
      "training_step_time": 9.898845672607422
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 3.131809711456299,
      "step": 2976
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 2976,
      "training_step_time": 9.901630640029907
    },
    {
      "epoch": 1.13525390625e-05,
      "model_forward_time": 3.132424831390381,
      "step": 2976
    },
    {
      "epoch": 1.13525390625e-05,
      "step": 2976,
      "training_step_time": 10.024150848388672
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 3.1418025493621826,
      "step": 2977
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 2977,
      "training_step_time": 10.192917346954346
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 2.428140640258789,
      "step": 2977
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 2977,
      "training_step_time": 7.4516987800598145
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 2.4303336143493652,
      "step": 2977
    },
    {
      "dynamics_loss_avg": 0.07030254006385803,
      "epoch": 1.1356353759765624e-05,
      "step": 2977
    },
    {
      "action_loss_avg": 0.03849983746185899,
      "epoch": 1.1356353759765624e-05,
      "step": 2977
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 2977,
      "training_step_time": 7.460134267807007
    },
    {
      "epoch": 1.1356353759765624e-05,
      "model_forward_time": 3.1265573501586914,
      "step": 2977
    },
    {
      "epoch": 1.1356353759765624e-05,
      "step": 2977,
      "training_step_time": 10.026750087738037
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 3.150491237640381,
      "step": 2978
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 2978,
      "training_step_time": 9.923656940460205
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 2.43153715133667,
      "step": 2978
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 2978,
      "training_step_time": 7.456717014312744
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 3.4214446544647217,
      "step": 2978
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 2978,
      "training_step_time": 10.187668561935425
    },
    {
      "epoch": 1.136016845703125e-05,
      "model_forward_time": 3.1302881240844727,
      "step": 2978
    },
    {
      "epoch": 1.136016845703125e-05,
      "step": 2978,
      "training_step_time": 10.02063775062561
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 2.438581943511963,
      "step": 2979
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 2979,
      "training_step_time": 7.4655070304870605
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 3.1301865577697754,
      "step": 2979
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 2979,
      "training_step_time": 9.898168325424194
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 3.134176254272461,
      "step": 2979
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 2979,
      "training_step_time": 9.909296751022339
    },
    {
      "epoch": 1.1363983154296875e-05,
      "model_forward_time": 3.385556221008301,
      "step": 2979
    },
    {
      "epoch": 1.1363983154296875e-05,
      "step": 2979,
      "training_step_time": 10.322270393371582
    },
    {
      "epoch": 1.13677978515625e-05,
      "grad_norm": 0.30124524235725403,
      "learning_rate": 1e-05,
      "loss": 0.1014,
      "step": 2980
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 3.1491262912750244,
      "step": 2980
    },
    {
      "dynamics_loss_avg": 0.0677251409739256,
      "epoch": 1.13677978515625e-05,
      "step": 2980
    },
    {
      "action_loss_avg": 0.03840731605887413,
      "epoch": 1.13677978515625e-05,
      "step": 2980
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 2980,
      "training_step_time": 9.92338252067566
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 3.126241683959961,
      "step": 2980
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 2980,
      "training_step_time": 9.891951560974121
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 3.1296823024749756,
      "step": 2980
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 2980,
      "training_step_time": 9.891604661941528
    },
    {
      "epoch": 1.13677978515625e-05,
      "model_forward_time": 2.422323703765869,
      "step": 2980
    },
    {
      "epoch": 1.13677978515625e-05,
      "step": 2980,
      "training_step_time": 10.005008220672607
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 3.3981287479400635,
      "step": 2981
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 2981,
      "training_step_time": 10.162786722183228
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 3.1209630966186523,
      "step": 2981
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 2981,
      "training_step_time": 9.878694295883179
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 2.422739267349243,
      "step": 2981
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 2981,
      "training_step_time": 7.436919450759888
    },
    {
      "epoch": 1.1371612548828125e-05,
      "model_forward_time": 3.126732349395752,
      "step": 2981
    },
    {
      "epoch": 1.1371612548828125e-05,
      "step": 2981,
      "training_step_time": 10.238455533981323
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 3.143538475036621,
      "step": 2982
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 2982,
      "training_step_time": 9.900522232055664
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 3.4106431007385254,
      "step": 2982
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 2982,
      "training_step_time": 10.171138763427734
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 3.132591962814331,
      "step": 2982
    },
    {
      "dynamics_loss_avg": 0.04844924155622721,
      "epoch": 1.137542724609375e-05,
      "step": 2982
    },
    {
      "action_loss_avg": 0.024044823367148638,
      "epoch": 1.137542724609375e-05,
      "step": 2982
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 2982,
      "training_step_time": 9.894212245941162
    },
    {
      "epoch": 1.137542724609375e-05,
      "model_forward_time": 3.13431715965271,
      "step": 2982
    },
    {
      "epoch": 1.137542724609375e-05,
      "step": 2982,
      "training_step_time": 10.26951789855957
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 3.1463921070098877,
      "step": 2983
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 2983,
      "training_step_time": 9.907124280929565
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 3.1336333751678467,
      "step": 2983
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 2983,
      "training_step_time": 9.896838903427124
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 3.3826706409454346,
      "step": 2983
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 2983,
      "training_step_time": 10.14959454536438
    },
    {
      "epoch": 1.1379241943359375e-05,
      "model_forward_time": 3.1451690196990967,
      "step": 2983
    },
    {
      "epoch": 1.1379241943359375e-05,
      "step": 2983,
      "training_step_time": 10.31553339958191
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 3.146327257156372,
      "step": 2984
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 2984,
      "training_step_time": 9.91603708267212
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 3.1314339637756348,
      "step": 2984
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 2984,
      "training_step_time": 9.903130054473877
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 3.405179262161255,
      "step": 2984
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 2984,
      "training_step_time": 10.17370057106018
    },
    {
      "epoch": 1.1383056640625e-05,
      "model_forward_time": 3.1335864067077637,
      "step": 2984
    },
    {
      "epoch": 1.1383056640625e-05,
      "step": 2984,
      "training_step_time": 10.291198968887329
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 3.143768787384033,
      "step": 2985
    },
    {
      "dynamics_loss_avg": 0.05658228769898414,
      "epoch": 1.1386871337890625e-05,
      "step": 2985
    },
    {
      "action_loss_avg": 0.04173151906579733,
      "epoch": 1.1386871337890625e-05,
      "step": 2985
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 2985,
      "training_step_time": 9.915407419204712
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 3.1389694213867188,
      "step": 2985
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 2985,
      "training_step_time": 9.905015468597412
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 3.1345536708831787,
      "step": 2985
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 2985,
      "training_step_time": 9.900383949279785
    },
    {
      "epoch": 1.1386871337890625e-05,
      "model_forward_time": 3.133554458618164,
      "step": 2985
    },
    {
      "epoch": 1.1386871337890625e-05,
      "step": 2985,
      "training_step_time": 10.30190658569336
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 3.147366762161255,
      "step": 2986
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 2986,
      "training_step_time": 9.916597366333008
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 3.1337296962738037,
      "step": 2986
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 2986,
      "training_step_time": 9.900926351547241
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 3.137084722518921,
      "step": 2986
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 2986,
      "training_step_time": 9.91394829750061
    },
    {
      "epoch": 1.139068603515625e-05,
      "model_forward_time": 3.1364521980285645,
      "step": 2986
    },
    {
      "epoch": 1.139068603515625e-05,
      "step": 2986,
      "training_step_time": 10.27500605583191
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 3.432813882827759,
      "step": 2987
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 2987,
      "training_step_time": 10.207422733306885
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 3.134298801422119,
      "step": 2987
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 2987,
      "training_step_time": 9.913126468658447
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 3.1396267414093018,
      "step": 2987
    },
    {
      "dynamics_loss_avg": 0.061006111837923524,
      "epoch": 1.1394500732421875e-05,
      "step": 2987
    },
    {
      "action_loss_avg": 0.03425455871038139,
      "epoch": 1.1394500732421875e-05,
      "step": 2987
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 2987,
      "training_step_time": 9.920098781585693
    },
    {
      "epoch": 1.1394500732421875e-05,
      "model_forward_time": 3.1373844146728516,
      "step": 2987
    },
    {
      "epoch": 1.1394500732421875e-05,
      "step": 2987,
      "training_step_time": 10.019494533538818
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 3.151986598968506,
      "step": 2988
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 2988,
      "training_step_time": 9.93341326713562
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 3.147003173828125,
      "step": 2988
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 2988,
      "training_step_time": 10.18713665008545
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 2.4418423175811768,
      "step": 2988
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 2988,
      "training_step_time": 7.474803447723389
    },
    {
      "epoch": 1.13983154296875e-05,
      "model_forward_time": 3.1398839950561523,
      "step": 2988
    },
    {
      "epoch": 1.13983154296875e-05,
      "step": 2988,
      "training_step_time": 10.01725459098816
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 3.1597065925598145,
      "step": 2989
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 2989,
      "training_step_time": 9.932680368423462
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 2.435467004776001,
      "step": 2989
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 2989,
      "training_step_time": 7.469389200210571
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 3.1301209926605225,
      "step": 2989
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 2989,
      "training_step_time": 10.171613693237305
    },
    {
      "epoch": 1.1402130126953125e-05,
      "model_forward_time": 3.1319069862365723,
      "step": 2989
    },
    {
      "epoch": 1.1402130126953125e-05,
      "step": 2989,
      "training_step_time": 10.004506349563599
    },
    {
      "epoch": 1.140594482421875e-05,
      "grad_norm": 0.3923461139202118,
      "learning_rate": 1e-05,
      "loss": 0.0968,
      "step": 2990
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 2.4419336318969727,
      "step": 2990
    },
    {
      "dynamics_loss_avg": 0.06232594372704625,
      "epoch": 1.140594482421875e-05,
      "step": 2990
    },
    {
      "action_loss_avg": 0.02874676501378417,
      "epoch": 1.140594482421875e-05,
      "step": 2990
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 2990,
      "training_step_time": 7.470109701156616
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 3.133105993270874,
      "step": 2990
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 2990,
      "training_step_time": 9.905765295028687
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 3.13215970993042,
      "step": 2990
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 2990,
      "training_step_time": 9.903550863265991
    },
    {
      "epoch": 1.140594482421875e-05,
      "model_forward_time": 3.1367597579956055,
      "step": 2990
    },
    {
      "epoch": 1.140594482421875e-05,
      "step": 2990,
      "training_step_time": 10.320873260498047
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 3.419149398803711,
      "step": 2991
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 2991,
      "training_step_time": 10.191312074661255
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 3.137068510055542,
      "step": 2991
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 2991,
      "training_step_time": 9.914376258850098
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 3.1360485553741455,
      "step": 2991
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 2991,
      "training_step_time": 9.914849996566772
    },
    {
      "epoch": 1.1409759521484375e-05,
      "model_forward_time": 3.1422996520996094,
      "step": 2991
    },
    {
      "epoch": 1.1409759521484375e-05,
      "step": 2991,
      "training_step_time": 10.278729438781738
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 3.4052624702453613,
      "step": 2992
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 2992,
      "training_step_time": 10.184690713882446
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 3.1414544582366943,
      "step": 2992
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 2992,
      "training_step_time": 9.91570234298706
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 3.1378254890441895,
      "step": 2992
    },
    {
      "dynamics_loss_avg": 0.0516173480078578,
      "epoch": 1.141357421875e-05,
      "step": 2992
    },
    {
      "action_loss_avg": 0.03668791325762868,
      "epoch": 1.141357421875e-05,
      "step": 2992
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 2992,
      "training_step_time": 9.918529510498047
    },
    {
      "epoch": 1.141357421875e-05,
      "model_forward_time": 3.140099287033081,
      "step": 2992
    },
    {
      "epoch": 1.141357421875e-05,
      "step": 2992,
      "training_step_time": 10.031429052352905
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 3.149994373321533,
      "step": 2993
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 2993,
      "training_step_time": 9.930897951126099
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 3.1371777057647705,
      "step": 2993
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 2993,
      "training_step_time": 10.188058376312256
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 3.138035297393799,
      "step": 2993
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 2993,
      "training_step_time": 9.922530889511108
    },
    {
      "epoch": 1.1417388916015625e-05,
      "model_forward_time": 3.1428346633911133,
      "step": 2993
    },
    {
      "epoch": 1.1417388916015625e-05,
      "step": 2993,
      "training_step_time": 10.282400369644165
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 3.1523237228393555,
      "step": 2994
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 2994,
      "training_step_time": 9.928117752075195
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 3.135582208633423,
      "step": 2994
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 2994,
      "training_step_time": 9.908282995223999
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 3.136219024658203,
      "step": 2994
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 2994,
      "training_step_time": 10.189258813858032
    },
    {
      "epoch": 1.142120361328125e-05,
      "model_forward_time": 3.1445789337158203,
      "step": 2994
    },
    {
      "epoch": 1.142120361328125e-05,
      "step": 2994,
      "training_step_time": 10.018399477005005
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 3.1595075130462646,
      "step": 2995
    },
    {
      "dynamics_loss_avg": 0.06406078450381755,
      "epoch": 1.1425018310546875e-05,
      "step": 2995
    },
    {
      "action_loss_avg": 0.034684638120234014,
      "epoch": 1.1425018310546875e-05,
      "step": 2995
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 2995,
      "training_step_time": 9.942797422409058
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 3.1436867713928223,
      "step": 2995
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 2995,
      "training_step_time": 9.920082569122314
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 3.1458568572998047,
      "step": 2995
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 2995,
      "training_step_time": 9.922441244125366
    },
    {
      "epoch": 1.1425018310546875e-05,
      "model_forward_time": 3.430232286453247,
      "step": 2995
    },
    {
      "epoch": 1.1425018310546875e-05,
      "step": 2995,
      "training_step_time": 10.301313638687134
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 3.162266969680786,
      "step": 2996
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 2996,
      "training_step_time": 9.94114899635315
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 3.1423439979553223,
      "step": 2996
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 2996,
      "training_step_time": 9.92214059829712
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 3.148402214050293,
      "step": 2996
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 2996,
      "training_step_time": 9.931474924087524
    },
    {
      "epoch": 1.14288330078125e-05,
      "model_forward_time": 3.1433522701263428,
      "step": 2996
    },
    {
      "epoch": 1.14288330078125e-05,
      "step": 2996,
      "training_step_time": 10.315863609313965
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 3.1511645317077637,
      "step": 2997
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 2997,
      "training_step_time": 10.20396089553833
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 3.1417109966278076,
      "step": 2997
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 2997,
      "training_step_time": 9.921940803527832
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 3.1434218883514404,
      "step": 2997
    },
    {
      "dynamics_loss_avg": 0.07725302558392286,
      "epoch": 1.1432647705078125e-05,
      "step": 2997
    },
    {
      "action_loss_avg": 0.04646697733551264,
      "epoch": 1.1432647705078125e-05,
      "step": 2997
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 2997,
      "training_step_time": 9.926621437072754
    },
    {
      "epoch": 1.1432647705078125e-05,
      "model_forward_time": 3.143688201904297,
      "step": 2997
    },
    {
      "epoch": 1.1432647705078125e-05,
      "step": 2997,
      "training_step_time": 10.237095594406128
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 3.159803628921509,
      "step": 2998
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 2998,
      "training_step_time": 9.936822414398193
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 2.7087042331695557,
      "step": 2998
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 2998,
      "training_step_time": 7.734888553619385
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 3.1321444511413574,
      "step": 2998
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 2998,
      "training_step_time": 9.905543088912964
    },
    {
      "epoch": 1.143646240234375e-05,
      "model_forward_time": 3.127425193786621,
      "step": 2998
    },
    {
      "epoch": 1.143646240234375e-05,
      "step": 2998,
      "training_step_time": 10.000751495361328
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 3.17645001411438,
      "step": 2999
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 2999,
      "training_step_time": 9.95923399925232
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 2.430919885635376,
      "step": 2999
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 2999,
      "training_step_time": 7.464054107666016
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 2.4267568588256836,
      "step": 2999
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 2999,
      "training_step_time": 7.453303575515747
    },
    {
      "epoch": 1.1440277099609375e-05,
      "model_forward_time": 3.402632236480713,
      "step": 2999
    },
    {
      "epoch": 1.1440277099609375e-05,
      "step": 2999,
      "training_step_time": 10.27617073059082
    },
    {
      "epoch": 1.1444091796875e-05,
      "grad_norm": 0.5677189826965332,
      "learning_rate": 1e-05,
      "loss": 0.0991,
      "step": 3000
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 3.169870376586914,
      "step": 3000
    },
    {
      "dynamics_loss_avg": 0.059525054134428504,
      "epoch": 1.1444091796875e-05,
      "step": 3000
    },
    {
      "action_loss_avg": 0.03398711527697742,
      "epoch": 1.1444091796875e-05,
      "step": 3000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 3000,
      "training_step_time": 9.931098699569702
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 3.1251134872436523,
      "step": 3000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 3000,
      "training_step_time": 9.874154806137085
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 3.1327826976776123,
      "step": 3000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 3000,
      "training_step_time": 9.887189388275146
    },
    {
      "epoch": 1.1444091796875e-05,
      "model_forward_time": 4.687343120574951,
      "step": 3000
    },
    {
      "epoch": 1.1444091796875e-05,
      "step": 3000,
      "training_step_time": 10.313762903213501
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 3.138552188873291,
      "step": 3001
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 3001,
      "training_step_time": 9.905439376831055
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 2.4279873371124268,
      "step": 3001
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 3001,
      "training_step_time": 7.4501378536224365
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 3.128728151321411,
      "step": 3001
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 3001,
      "training_step_time": 9.894932270050049
    },
    {
      "epoch": 1.1447906494140625e-05,
      "model_forward_time": 3.130429744720459,
      "step": 3001
    },
    {
      "epoch": 1.1447906494140625e-05,
      "step": 3001,
      "training_step_time": 10.30104660987854
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 3.4539406299591064,
      "step": 3002
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 3002,
      "training_step_time": 10.232166290283203
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 2.4318392276763916,
      "step": 3002
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 3002,
      "training_step_time": 7.461523056030273
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 3.1323511600494385,
      "step": 3002
    },
    {
      "dynamics_loss_avg": 0.06370178703218699,
      "epoch": 1.145172119140625e-05,
      "step": 3002
    },
    {
      "action_loss_avg": 0.03284069374203682,
      "epoch": 1.145172119140625e-05,
      "step": 3002
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 3002,
      "training_step_time": 9.904930830001831
    },
    {
      "epoch": 1.145172119140625e-05,
      "model_forward_time": 3.128847599029541,
      "step": 3002
    },
    {
      "epoch": 1.145172119140625e-05,
      "step": 3002,
      "training_step_time": 10.268515586853027
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 3.148191452026367,
      "step": 3003
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 3003,
      "training_step_time": 9.925889253616333
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 3.382727861404419,
      "step": 3003
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 3003,
      "training_step_time": 10.158747911453247
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 3.137254238128662,
      "step": 3003
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 3003,
      "training_step_time": 9.920877456665039
    },
    {
      "epoch": 1.1455535888671875e-05,
      "model_forward_time": 3.1371068954467773,
      "step": 3003
    },
    {
      "epoch": 1.1455535888671875e-05,
      "step": 3003,
      "training_step_time": 10.035924911499023
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 3.153630256652832,
      "step": 3004
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 3004,
      "training_step_time": 9.933709859848022
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 3.139556884765625,
      "step": 3004
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 3004,
      "training_step_time": 9.92288064956665
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 3.452601432800293,
      "step": 3004
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 3004,
      "training_step_time": 10.233511924743652
    },
    {
      "epoch": 1.14593505859375e-05,
      "model_forward_time": 3.134397268295288,
      "step": 3004
    },
    {
      "epoch": 1.14593505859375e-05,
      "step": 3004,
      "training_step_time": 10.313480377197266
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 3.1407248973846436,
      "step": 3005
    },
    {
      "dynamics_loss_avg": 0.08486682865768672,
      "epoch": 1.1463165283203125e-05,
      "step": 3005
    },
    {
      "action_loss_avg": 0.04679241317790002,
      "epoch": 1.1463165283203125e-05,
      "step": 3005
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 3005,
      "training_step_time": 9.922766208648682
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 3.133838176727295,
      "step": 3005
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 3005,
      "training_step_time": 9.907686948776245
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 3.130568027496338,
      "step": 3005
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 3005,
      "training_step_time": 9.901619672775269
    },
    {
      "epoch": 1.1463165283203125e-05,
      "model_forward_time": 2.718109130859375,
      "step": 3005
    },
    {
      "epoch": 1.1463165283203125e-05,
      "step": 3005,
      "training_step_time": 10.019516706466675
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 3.1376097202301025,
      "step": 3006
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 3006,
      "training_step_time": 9.911935091018677
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 3.133772611618042,
      "step": 3006
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 3006,
      "training_step_time": 9.9109468460083
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 3.135784387588501,
      "step": 3006
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 3006,
      "training_step_time": 9.911141157150269
    },
    {
      "epoch": 1.146697998046875e-05,
      "model_forward_time": 3.1323883533477783,
      "step": 3006
    },
    {
      "epoch": 1.146697998046875e-05,
      "step": 3006,
      "training_step_time": 10.284743785858154
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 3.1439762115478516,
      "step": 3007
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 3007,
      "training_step_time": 10.189797401428223
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 3.133066415786743,
      "step": 3007
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 3007,
      "training_step_time": 9.904215097427368
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 2.426793098449707,
      "step": 3007
    },
    {
      "dynamics_loss_avg": 0.06575932875275611,
      "epoch": 1.1470794677734375e-05,
      "step": 3007
    },
    {
      "action_loss_avg": 0.04068226516246796,
      "epoch": 1.1470794677734375e-05,
      "step": 3007
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 3007,
      "training_step_time": 7.453725576400757
    },
    {
      "epoch": 1.1470794677734375e-05,
      "model_forward_time": 3.134243965148926,
      "step": 3007
    },
    {
      "epoch": 1.1470794677734375e-05,
      "step": 3007,
      "training_step_time": 10.314764738082886
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 2.440077781677246,
      "step": 3008
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 3008,
      "training_step_time": 7.46846079826355
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 3.1324329376220703,
      "step": 3008
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 3008,
      "training_step_time": 10.223670244216919
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 3.1280436515808105,
      "step": 3008
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 3008,
      "training_step_time": 9.900283575057983
    },
    {
      "epoch": 1.1474609375e-05,
      "model_forward_time": 3.1331281661987305,
      "step": 3008
    },
    {
      "epoch": 1.1474609375e-05,
      "step": 3008,
      "training_step_time": 10.247071743011475
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 3.1456522941589355,
      "step": 3009
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 3009,
      "training_step_time": 9.919947147369385
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 3.1348893642425537,
      "step": 3009
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 3009,
      "training_step_time": 9.906706094741821
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 3.139930248260498,
      "step": 3009
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 3009,
      "training_step_time": 10.18787693977356
    },
    {
      "epoch": 1.1478424072265625e-05,
      "model_forward_time": 2.436678171157837,
      "step": 3009
    },
    {
      "epoch": 1.1478424072265625e-05,
      "step": 3009,
      "training_step_time": 9.996025800704956
    },
    {
      "epoch": 1.148223876953125e-05,
      "grad_norm": 0.46151432394981384,
      "learning_rate": 1e-05,
      "loss": 0.1004,
      "step": 3010
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 3.1588642597198486,
      "step": 3010
    },
    {
      "dynamics_loss_avg": 0.04874413786455989,
      "epoch": 1.148223876953125e-05,
      "step": 3010
    },
    {
      "action_loss_avg": 0.022989092790521682,
      "epoch": 1.148223876953125e-05,
      "step": 3010
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 3010,
      "training_step_time": 9.93604063987732
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 3.1353089809417725,
      "step": 3010
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 3010,
      "training_step_time": 9.901594638824463
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 3.1337015628814697,
      "step": 3010
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 3010,
      "training_step_time": 9.943639039993286
    },
    {
      "epoch": 1.148223876953125e-05,
      "model_forward_time": 3.141522169113159,
      "step": 3010
    },
    {
      "epoch": 1.148223876953125e-05,
      "step": 3010,
      "training_step_time": 10.335956335067749
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 3.152263879776001,
      "step": 3011
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 3011,
      "training_step_time": 9.923181533813477
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 2.4398646354675293,
      "step": 3011
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 3011,
      "training_step_time": 7.470779895782471
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 3.148183584213257,
      "step": 3011
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 3011,
      "training_step_time": 9.921987295150757
    },
    {
      "epoch": 1.1486053466796875e-05,
      "model_forward_time": 3.1362009048461914,
      "step": 3011
    },
    {
      "epoch": 1.1486053466796875e-05,
      "step": 3011,
      "training_step_time": 10.014785528182983
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 3.1595683097839355,
      "step": 3012
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 3012,
      "training_step_time": 9.948630094528198
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 3.425191879272461,
      "step": 3012
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 3012,
      "training_step_time": 10.199796915054321
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 3.1348206996917725,
      "step": 3012
    },
    {
      "dynamics_loss_avg": 0.06630328111350536,
      "epoch": 1.14898681640625e-05,
      "step": 3012
    },
    {
      "action_loss_avg": 0.03768516350537539,
      "epoch": 1.14898681640625e-05,
      "step": 3012
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 3012,
      "training_step_time": 9.908238887786865
    },
    {
      "epoch": 1.14898681640625e-05,
      "model_forward_time": 3.1331446170806885,
      "step": 3012
    },
    {
      "epoch": 1.14898681640625e-05,
      "step": 3012,
      "training_step_time": 10.327661991119385
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 3.148127794265747,
      "step": 3013
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 3013,
      "training_step_time": 9.919867515563965
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 3.141279935836792,
      "step": 3013
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 3013,
      "training_step_time": 10.231696844100952
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 3.1367270946502686,
      "step": 3013
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 3013,
      "training_step_time": 9.908697366714478
    },
    {
      "epoch": 1.1493682861328125e-05,
      "model_forward_time": 3.131040096282959,
      "step": 3013
    },
    {
      "epoch": 1.1493682861328125e-05,
      "step": 3013,
      "training_step_time": 10.005699157714844
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 3.141655445098877,
      "step": 3014
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 3014,
      "training_step_time": 9.913624286651611
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 3.1371729373931885,
      "step": 3014
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 3014,
      "training_step_time": 9.911743640899658
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 3.425814628601074,
      "step": 3014
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 3014,
      "training_step_time": 10.194869756698608
    },
    {
      "epoch": 1.149749755859375e-05,
      "model_forward_time": 3.1328601837158203,
      "step": 3014
    },
    {
      "epoch": 1.149749755859375e-05,
      "step": 3014,
      "training_step_time": 10.002116918563843
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 3.1420938968658447,
      "step": 3015
    },
    {
      "dynamics_loss_avg": 0.05886946711689234,
      "epoch": 1.1501312255859375e-05,
      "step": 3015
    },
    {
      "action_loss_avg": 0.03185971528291702,
      "epoch": 1.1501312255859375e-05,
      "step": 3015
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 3015,
      "training_step_time": 9.917297601699829
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 3.135016679763794,
      "step": 3015
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 3015,
      "training_step_time": 9.904677152633667
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 3.1294167041778564,
      "step": 3015
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 3015,
      "training_step_time": 9.900954484939575
    },
    {
      "epoch": 1.1501312255859375e-05,
      "model_forward_time": 3.1393632888793945,
      "step": 3015
    },
    {
      "epoch": 1.1501312255859375e-05,
      "step": 3015,
      "training_step_time": 10.275672674179077
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 2.444284677505493,
      "step": 3016
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 3016,
      "training_step_time": 7.474581480026245
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 3.1415789127349854,
      "step": 3016
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 3016,
      "training_step_time": 9.92001485824585
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 3.1376702785491943,
      "step": 3016
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 3016,
      "training_step_time": 9.9122154712677
    },
    {
      "epoch": 1.1505126953125e-05,
      "model_forward_time": 3.140204429626465,
      "step": 3016
    },
    {
      "epoch": 1.1505126953125e-05,
      "step": 3016,
      "training_step_time": 10.308699607849121
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 3.4066991806030273,
      "step": 3017
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 3017,
      "training_step_time": 10.184983015060425
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 3.1452038288116455,
      "step": 3017
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 3017,
      "training_step_time": 9.91911244392395
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 3.1405422687530518,
      "step": 3017
    },
    {
      "dynamics_loss_avg": 0.04790560659021139,
      "epoch": 1.1508941650390625e-05,
      "step": 3017
    },
    {
      "action_loss_avg": 0.033599520288407805,
      "epoch": 1.1508941650390625e-05,
      "step": 3017
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 3017,
      "training_step_time": 9.917607069015503
    },
    {
      "epoch": 1.1508941650390625e-05,
      "model_forward_time": 3.140305519104004,
      "step": 3017
    },
    {
      "epoch": 1.1508941650390625e-05,
      "step": 3017,
      "training_step_time": 10.350778818130493
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 2.447599411010742,
      "step": 3018
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 3018,
      "training_step_time": 7.480226039886475
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 2.432199239730835,
      "step": 3018
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 3018,
      "training_step_time": 7.463719367980957
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 3.429733991622925,
      "step": 3018
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 3018,
      "training_step_time": 10.19650650024414
    },
    {
      "epoch": 1.151275634765625e-05,
      "model_forward_time": 3.1399292945861816,
      "step": 3018
    },
    {
      "epoch": 1.151275634765625e-05,
      "step": 3018,
      "training_step_time": 10.023441314697266
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 3.144765853881836,
      "step": 3019
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 3019,
      "training_step_time": 9.922067403793335
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 3.135845184326172,
      "step": 3019
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 3019,
      "training_step_time": 9.9108247756958
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 2.4325788021087646,
      "step": 3019
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 3019,
      "training_step_time": 7.4611122608184814
    },
    {
      "epoch": 1.1516571044921875e-05,
      "model_forward_time": 3.4067509174346924,
      "step": 3019
    },
    {
      "epoch": 1.1516571044921875e-05,
      "step": 3019,
      "training_step_time": 10.289231061935425
    },
    {
      "epoch": 1.15203857421875e-05,
      "grad_norm": 0.41511982679367065,
      "learning_rate": 1e-05,
      "loss": 0.0959,
      "step": 3020
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 3.154358386993408,
      "step": 3020
    },
    {
      "dynamics_loss_avg": 0.07724153138697147,
      "epoch": 1.15203857421875e-05,
      "step": 3020
    },
    {
      "action_loss_avg": 0.045182406343519686,
      "epoch": 1.15203857421875e-05,
      "step": 3020
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 3020,
      "training_step_time": 9.93500304222107
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 3.136930227279663,
      "step": 3020
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 3020,
      "training_step_time": 9.912623167037964
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 3.142712116241455,
      "step": 3020
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 3020,
      "training_step_time": 9.917919397354126
    },
    {
      "epoch": 1.15203857421875e-05,
      "model_forward_time": 2.4368903636932373,
      "step": 3020
    },
    {
      "epoch": 1.15203857421875e-05,
      "step": 3020,
      "training_step_time": 10.369513273239136
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 3.436422109603882,
      "step": 3021
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 3021,
      "training_step_time": 10.207763195037842
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 3.1350300312042236,
      "step": 3021
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 3021,
      "training_step_time": 9.903589248657227
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 3.1392595767974854,
      "step": 3021
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 3021,
      "training_step_time": 9.906812191009521
    },
    {
      "epoch": 1.1524200439453125e-05,
      "model_forward_time": 3.130873918533325,
      "step": 3021
    },
    {
      "epoch": 1.1524200439453125e-05,
      "step": 3021,
      "training_step_time": 10.007749795913696
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 3.140866279602051,
      "step": 3022
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 3022,
      "training_step_time": 9.917125225067139
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 2.694885730743408,
      "step": 3022
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 3022,
      "training_step_time": 7.722494602203369
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 3.132814645767212,
      "step": 3022
    },
    {
      "dynamics_loss_avg": 0.054045718256384136,
      "epoch": 1.152801513671875e-05,
      "step": 3022
    },
    {
      "action_loss_avg": 0.02519820025190711,
      "epoch": 1.152801513671875e-05,
      "step": 3022
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 3022,
      "training_step_time": 9.905914545059204
    },
    {
      "epoch": 1.152801513671875e-05,
      "model_forward_time": 2.433772563934326,
      "step": 3022
    },
    {
      "epoch": 1.152801513671875e-05,
      "step": 3022,
      "training_step_time": 10.020347833633423
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 3.147900104522705,
      "step": 3023
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 3023,
      "training_step_time": 9.920606136322021
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 3.135521411895752,
      "step": 3023
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 3023,
      "training_step_time": 9.907440185546875
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 3.401343584060669,
      "step": 3023
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 3023,
      "training_step_time": 10.172895669937134
    },
    {
      "epoch": 1.1531829833984376e-05,
      "model_forward_time": 3.132702589035034,
      "step": 3023
    },
    {
      "epoch": 1.1531829833984376e-05,
      "step": 3023,
      "training_step_time": 10.271764993667603
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 2.445650100708008,
      "step": 3024
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 3024,
      "training_step_time": 7.472428560256958
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 2.434506416320801,
      "step": 3024
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 3024,
      "training_step_time": 7.458447217941284
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 3.132422685623169,
      "step": 3024
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 3024,
      "training_step_time": 9.900223970413208
    },
    {
      "epoch": 1.153564453125e-05,
      "model_forward_time": 2.7326574325561523,
      "step": 3024
    },
    {
      "epoch": 1.153564453125e-05,
      "step": 3024,
      "training_step_time": 10.011399507522583
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 3.1426851749420166,
      "step": 3025
    },
    {
      "dynamics_loss_avg": 0.05793818957172334,
      "epoch": 1.1539459228515625e-05,
      "step": 3025
    },
    {
      "action_loss_avg": 0.032061701553175226,
      "epoch": 1.1539459228515625e-05,
      "step": 3025
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 3025,
      "training_step_time": 9.915178537368774
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 3.13511061668396,
      "step": 3025
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 3025,
      "training_step_time": 9.900131464004517
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 3.1353976726531982,
      "step": 3025
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 3025,
      "training_step_time": 9.901602268218994
    },
    {
      "epoch": 1.1539459228515625e-05,
      "model_forward_time": 3.1360442638397217,
      "step": 3025
    },
    {
      "epoch": 1.1539459228515625e-05,
      "step": 3025,
      "training_step_time": 10.268310308456421
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 3.435332775115967,
      "step": 3026
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 3026,
      "training_step_time": 10.202125310897827
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 3.136691093444824,
      "step": 3026
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 3026,
      "training_step_time": 9.907552719116211
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 2.4319546222686768,
      "step": 3026
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 3026,
      "training_step_time": 7.458975315093994
    },
    {
      "epoch": 1.154327392578125e-05,
      "model_forward_time": 2.433586835861206,
      "step": 3026
    },
    {
      "epoch": 1.154327392578125e-05,
      "step": 3026,
      "training_step_time": 10.28296685218811
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 3.161478281021118,
      "step": 3027
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 3027,
      "training_step_time": 9.935429573059082
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 3.402503490447998,
      "step": 3027
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 3027,
      "training_step_time": 10.173383951187134
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 3.13669490814209,
      "step": 3027
    },
    {
      "dynamics_loss_avg": 0.06616786606609822,
      "epoch": 1.1547088623046876e-05,
      "step": 3027
    },
    {
      "action_loss_avg": 0.04897218411788344,
      "epoch": 1.1547088623046876e-05,
      "step": 3027
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 3027,
      "training_step_time": 9.91323208808899
    },
    {
      "epoch": 1.1547088623046876e-05,
      "model_forward_time": 3.1376612186431885,
      "step": 3027
    },
    {
      "epoch": 1.1547088623046876e-05,
      "step": 3027,
      "training_step_time": 10.026010990142822
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 3.155406951904297,
      "step": 3028
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 3028,
      "training_step_time": 9.928488492965698
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 3.1282694339752197,
      "step": 3028
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 3028,
      "training_step_time": 9.896817922592163
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 2.701181173324585,
      "step": 3028
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 3028,
      "training_step_time": 7.7253148555755615
    },
    {
      "epoch": 1.15509033203125e-05,
      "model_forward_time": 3.136202812194824,
      "step": 3028
    },
    {
      "epoch": 1.15509033203125e-05,
      "step": 3028,
      "training_step_time": 10.292418479919434
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 3.1451756954193115,
      "step": 3029
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 3029,
      "training_step_time": 9.913110494613647
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 2.42724347114563,
      "step": 3029
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 3029,
      "training_step_time": 7.4532692432403564
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 3.1280007362365723,
      "step": 3029
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 3029,
      "training_step_time": 9.895938634872437
    },
    {
      "epoch": 1.1554718017578125e-05,
      "model_forward_time": 3.4049277305603027,
      "step": 3029
    },
    {
      "epoch": 1.1554718017578125e-05,
      "step": 3029,
      "training_step_time": 10.266813039779663
    },
    {
      "epoch": 1.155853271484375e-05,
      "grad_norm": 0.3960479199886322,
      "learning_rate": 1e-05,
      "loss": 0.101,
      "step": 3030
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 3.146522283554077,
      "step": 3030
    },
    {
      "dynamics_loss_avg": 0.06726428512483835,
      "epoch": 1.155853271484375e-05,
      "step": 3030
    },
    {
      "action_loss_avg": 0.041796785593032834,
      "epoch": 1.155853271484375e-05,
      "step": 3030
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 3030,
      "training_step_time": 9.92077374458313
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 3.1292848587036133,
      "step": 3030
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 3030,
      "training_step_time": 9.902389526367188
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 3.1406779289245605,
      "step": 3030
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 3030,
      "training_step_time": 9.916402578353882
    },
    {
      "epoch": 1.155853271484375e-05,
      "model_forward_time": 3.1371779441833496,
      "step": 3030
    },
    {
      "epoch": 1.155853271484375e-05,
      "step": 3030,
      "training_step_time": 10.275860071182251
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 3.4529316425323486,
      "step": 3031
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 3031,
      "training_step_time": 10.241746425628662
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 3.1417274475097656,
      "step": 3031
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 3031,
      "training_step_time": 9.923850059509277
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 3.14046311378479,
      "step": 3031
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 3031,
      "training_step_time": 9.922015190124512
    },
    {
      "epoch": 1.1562347412109376e-05,
      "model_forward_time": 2.4381675720214844,
      "step": 3031
    },
    {
      "epoch": 1.1562347412109376e-05,
      "step": 3031,
      "training_step_time": 10.300243377685547
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 3.153017282485962,
      "step": 3032
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 3032,
      "training_step_time": 9.930867195129395
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 3.4290099143981934,
      "step": 3032
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 3032,
      "training_step_time": 10.200655937194824
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 3.1351375579833984,
      "step": 3032
    },
    {
      "dynamics_loss_avg": 0.05104042608290911,
      "epoch": 1.1566162109375e-05,
      "step": 3032
    },
    {
      "action_loss_avg": 0.029228236339986325,
      "epoch": 1.1566162109375e-05,
      "step": 3032
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 3032,
      "training_step_time": 9.904384136199951
    },
    {
      "epoch": 1.1566162109375e-05,
      "model_forward_time": 3.1323578357696533,
      "step": 3032
    },
    {
      "epoch": 1.1566162109375e-05,
      "step": 3032,
      "training_step_time": 10.296992063522339
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 3.135578155517578,
      "step": 3033
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 3033,
      "training_step_time": 9.90816593170166
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 3.1313912868499756,
      "step": 3033
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 3033,
      "training_step_time": 9.903289318084717
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 2.6837756633758545,
      "step": 3033
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 3033,
      "training_step_time": 7.707127809524536
    },
    {
      "epoch": 1.1569976806640625e-05,
      "model_forward_time": 3.1275076866149902,
      "step": 3033
    },
    {
      "epoch": 1.1569976806640625e-05,
      "step": 3033,
      "training_step_time": 10.326679944992065
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 3.1611011028289795,
      "step": 3034
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 3034,
      "training_step_time": 9.94997262954712
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 3.14715313911438,
      "step": 3034
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 3034,
      "training_step_time": 9.920373439788818
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 3.141190767288208,
      "step": 3034
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 3034,
      "training_step_time": 9.914576053619385
    },
    {
      "epoch": 1.157379150390625e-05,
      "model_forward_time": 3.4240756034851074,
      "step": 3034
    },
    {
      "epoch": 1.157379150390625e-05,
      "step": 3034,
      "training_step_time": 10.293824434280396
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 3.1543383598327637,
      "step": 3035
    },
    {
      "dynamics_loss_avg": 0.06957651264965534,
      "epoch": 1.1577606201171876e-05,
      "step": 3035
    },
    {
      "action_loss_avg": 0.045522396080195905,
      "epoch": 1.1577606201171876e-05,
      "step": 3035
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 3035,
      "training_step_time": 9.9307701587677
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 3.149524450302124,
      "step": 3035
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 3035,
      "training_step_time": 9.92253589630127
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 3.142787218093872,
      "step": 3035
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 3035,
      "training_step_time": 9.914178133010864
    },
    {
      "epoch": 1.1577606201171876e-05,
      "model_forward_time": 3.1449830532073975,
      "step": 3035
    },
    {
      "epoch": 1.1577606201171876e-05,
      "step": 3035,
      "training_step_time": 10.028326749801636
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 3.439796209335327,
      "step": 3036
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 3036,
      "training_step_time": 10.219426155090332
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 3.145508289337158,
      "step": 3036
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 3036,
      "training_step_time": 9.915435791015625
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 3.1400270462036133,
      "step": 3036
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 3036,
      "training_step_time": 9.910503625869751
    },
    {
      "epoch": 1.15814208984375e-05,
      "model_forward_time": 3.139744758605957,
      "step": 3036
    },
    {
      "epoch": 1.15814208984375e-05,
      "step": 3036,
      "training_step_time": 10.28080940246582
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 3.4375081062316895,
      "step": 3037
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 3037,
      "training_step_time": 10.202126264572144
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 3.1443228721618652,
      "step": 3037
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 3037,
      "training_step_time": 9.918355703353882
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 3.1403961181640625,
      "step": 3037
    },
    {
      "dynamics_loss_avg": 0.07940240390598774,
      "epoch": 1.1585235595703125e-05,
      "step": 3037
    },
    {
      "action_loss_avg": 0.04204819835722447,
      "epoch": 1.1585235595703125e-05,
      "step": 3037
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 3037,
      "training_step_time": 9.912108421325684
    },
    {
      "epoch": 1.1585235595703125e-05,
      "model_forward_time": 3.141796350479126,
      "step": 3037
    },
    {
      "epoch": 1.1585235595703125e-05,
      "step": 3037,
      "training_step_time": 10.311765909194946
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 3.154757261276245,
      "step": 3038
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 3038,
      "training_step_time": 9.926889657974243
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 3.45924973487854,
      "step": 3038
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 3038,
      "training_step_time": 10.2304368019104
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 3.147704839706421,
      "step": 3038
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 3038,
      "training_step_time": 9.92295789718628
    },
    {
      "epoch": 1.158905029296875e-05,
      "model_forward_time": 3.145325183868408,
      "step": 3038
    },
    {
      "epoch": 1.158905029296875e-05,
      "step": 3038,
      "training_step_time": 10.296979665756226
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 3.150522232055664,
      "step": 3039
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 3039,
      "training_step_time": 9.929150819778442
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 2.432600975036621,
      "step": 3039
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 3039,
      "training_step_time": 7.466455698013306
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 3.4053103923797607,
      "step": 3039
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 3039,
      "training_step_time": 10.183976411819458
    },
    {
      "epoch": 1.1592864990234376e-05,
      "model_forward_time": 3.143962860107422,
      "step": 3039
    },
    {
      "epoch": 1.1592864990234376e-05,
      "step": 3039,
      "training_step_time": 10.275035858154297
    },
    {
      "epoch": 1.15966796875e-05,
      "grad_norm": 0.4522646963596344,
      "learning_rate": 1e-05,
      "loss": 0.0985,
      "step": 3040
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 3.1544241905212402,
      "step": 3040
    },
    {
      "dynamics_loss_avg": 0.056222403421998024,
      "epoch": 1.15966796875e-05,
      "step": 3040
    },
    {
      "action_loss_avg": 0.033465572819113734,
      "epoch": 1.15966796875e-05,
      "step": 3040
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 3040,
      "training_step_time": 9.941229820251465
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 3.1380996704101562,
      "step": 3040
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 3040,
      "training_step_time": 9.912476062774658
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 3.141873836517334,
      "step": 3040
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 3040,
      "training_step_time": 9.919748306274414
    },
    {
      "epoch": 1.15966796875e-05,
      "model_forward_time": 2.432213306427002,
      "step": 3040
    },
    {
      "epoch": 1.15966796875e-05,
      "step": 3040,
      "training_step_time": 10.031465530395508
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 3.4461100101470947,
      "step": 3041
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 3041,
      "training_step_time": 10.216660976409912
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 3.139481782913208,
      "step": 3041
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 3041,
      "training_step_time": 9.91110873222351
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 2.434417963027954,
      "step": 3041
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 3041,
      "training_step_time": 7.463806867599487
    },
    {
      "epoch": 1.1600494384765625e-05,
      "model_forward_time": 3.1276168823242188,
      "step": 3041
    },
    {
      "epoch": 1.1600494384765625e-05,
      "step": 3041,
      "training_step_time": 10.014686346054077
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 2.440188407897949,
      "step": 3042
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 3042,
      "training_step_time": 7.46795654296875
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 3.3754584789276123,
      "step": 3042
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 3042,
      "training_step_time": 10.142780542373657
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 3.133558750152588,
      "step": 3042
    },
    {
      "dynamics_loss_avg": 0.07483008094131946,
      "epoch": 1.160430908203125e-05,
      "step": 3042
    },
    {
      "action_loss_avg": 0.04146588072180748,
      "epoch": 1.160430908203125e-05,
      "step": 3042
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 3042,
      "training_step_time": 9.905402183532715
    },
    {
      "epoch": 1.160430908203125e-05,
      "model_forward_time": 3.1286566257476807,
      "step": 3042
    },
    {
      "epoch": 1.160430908203125e-05,
      "step": 3042,
      "training_step_time": 10.281527042388916
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 2.441307306289673,
      "step": 3043
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 3043,
      "training_step_time": 7.467766046524048
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 2.424713373184204,
      "step": 3043
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 3043,
      "training_step_time": 7.449399948120117
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 3.3952741622924805,
      "step": 3043
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 3043,
      "training_step_time": 10.158203601837158
    },
    {
      "epoch": 1.1608123779296876e-05,
      "model_forward_time": 3.127288818359375,
      "step": 3043
    },
    {
      "epoch": 1.1608123779296876e-05,
      "step": 3043,
      "training_step_time": 10.00713849067688
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 3.136556625366211,
      "step": 3044
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 3044,
      "training_step_time": 9.90609860420227
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 3.1308014392852783,
      "step": 3044
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 3044,
      "training_step_time": 9.894971132278442
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 3.1279165744781494,
      "step": 3044
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 3044,
      "training_step_time": 9.893227815628052
    },
    {
      "epoch": 1.16119384765625e-05,
      "model_forward_time": 2.7232022285461426,
      "step": 3044
    },
    {
      "epoch": 1.16119384765625e-05,
      "step": 3044,
      "training_step_time": 10.000604391098022
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 2.4450440406799316,
      "step": 3045
    },
    {
      "dynamics_loss_avg": 0.049004807509481906,
      "epoch": 1.1615753173828125e-05,
      "step": 3045
    },
    {
      "action_loss_avg": 0.032166299642995,
      "epoch": 1.1615753173828125e-05,
      "step": 3045
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 3045,
      "training_step_time": 7.469928979873657
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 3.1370205879211426,
      "step": 3045
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 3045,
      "training_step_time": 9.90256643295288
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 3.133991241455078,
      "step": 3045
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 3045,
      "training_step_time": 9.902776002883911
    },
    {
      "epoch": 1.1615753173828125e-05,
      "model_forward_time": 3.147355079650879,
      "step": 3045
    },
    {
      "epoch": 1.1615753173828125e-05,
      "step": 3045,
      "training_step_time": 10.019597291946411
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 3.390920877456665,
      "step": 3046
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 3046,
      "training_step_time": 10.165980339050293
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 3.138308525085449,
      "step": 3046
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 3046,
      "training_step_time": 9.907017946243286
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 3.133294105529785,
      "step": 3046
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 3046,
      "training_step_time": 9.903286695480347
    },
    {
      "epoch": 1.161956787109375e-05,
      "model_forward_time": 3.138798713684082,
      "step": 3046
    },
    {
      "epoch": 1.161956787109375e-05,
      "step": 3046,
      "training_step_time": 10.307780742645264
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 2.4475808143615723,
      "step": 3047
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 3047,
      "training_step_time": 7.476753234863281
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 3.395573377609253,
      "step": 3047
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 3047,
      "training_step_time": 10.161333560943604
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 2.4290859699249268,
      "step": 3047
    },
    {
      "dynamics_loss_avg": 0.07172912172973156,
      "epoch": 1.1623382568359376e-05,
      "step": 3047
    },
    {
      "action_loss_avg": 0.03916330616921186,
      "epoch": 1.1623382568359376e-05,
      "step": 3047
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 3047,
      "training_step_time": 7.45357608795166
    },
    {
      "epoch": 1.1623382568359376e-05,
      "model_forward_time": 2.428128957748413,
      "step": 3047
    },
    {
      "epoch": 1.1623382568359376e-05,
      "step": 3047,
      "training_step_time": 10.277890920639038
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 3.1409096717834473,
      "step": 3048
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 3048,
      "training_step_time": 9.917017459869385
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 3.126746892929077,
      "step": 3048
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 3048,
      "training_step_time": 9.893102645874023
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 3.3802101612091064,
      "step": 3048
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 3048,
      "training_step_time": 10.14951491355896
    },
    {
      "epoch": 1.1627197265625e-05,
      "model_forward_time": 3.1309361457824707,
      "step": 3048
    },
    {
      "epoch": 1.1627197265625e-05,
      "step": 3048,
      "training_step_time": 10.27712082862854
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 3.140805721282959,
      "step": 3049
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 3049,
      "training_step_time": 9.909755229949951
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 3.128366231918335,
      "step": 3049
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 3049,
      "training_step_time": 9.892992496490479
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 2.4324042797088623,
      "step": 3049
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 3049,
      "training_step_time": 7.45351243019104
    },
    {
      "epoch": 1.1631011962890625e-05,
      "model_forward_time": 2.7023632526397705,
      "step": 3049
    },
    {
      "epoch": 1.1631011962890625e-05,
      "step": 3049,
      "training_step_time": 10.269530057907104
    },
    {
      "epoch": 1.163482666015625e-05,
      "grad_norm": 0.46044009923934937,
      "learning_rate": 1e-05,
      "loss": 0.0987,
      "step": 3050
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 3.142374038696289,
      "step": 3050
    },
    {
      "dynamics_loss_avg": 0.07848734445869923,
      "epoch": 1.163482666015625e-05,
      "step": 3050
    },
    {
      "action_loss_avg": 0.04446150343865156,
      "epoch": 1.163482666015625e-05,
      "step": 3050
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 3050,
      "training_step_time": 9.910249948501587
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 2.4295196533203125,
      "step": 3050
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 3050,
      "training_step_time": 7.449362754821777
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 3.1292309761047363,
      "step": 3050
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 3050,
      "training_step_time": 9.899610757827759
    },
    {
      "epoch": 1.163482666015625e-05,
      "model_forward_time": 3.1302380561828613,
      "step": 3050
    },
    {
      "epoch": 1.163482666015625e-05,
      "step": 3050,
      "training_step_time": 10.226207494735718
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 3.4151828289031982,
      "step": 3051
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 3051,
      "training_step_time": 10.181734800338745
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 3.12642240524292,
      "step": 3051
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 3051,
      "training_step_time": 9.893994331359863
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 3.135971784591675,
      "step": 3051
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 3051,
      "training_step_time": 9.906295537948608
    },
    {
      "epoch": 1.1638641357421876e-05,
      "model_forward_time": 3.135139226913452,
      "step": 3051
    },
    {
      "epoch": 1.1638641357421876e-05,
      "step": 3051,
      "training_step_time": 10.336973905563354
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 3.1499075889587402,
      "step": 3052
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 3052,
      "training_step_time": 9.92260217666626
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 3.392643451690674,
      "step": 3052
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 3052,
      "training_step_time": 10.168004512786865
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 2.432917356491089,
      "step": 3052
    },
    {
      "dynamics_loss_avg": 0.06689454261213541,
      "epoch": 1.16424560546875e-05,
      "step": 3052
    },
    {
      "action_loss_avg": 0.041529189143329856,
      "epoch": 1.16424560546875e-05,
      "step": 3052
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 3052,
      "training_step_time": 7.468750476837158
    },
    {
      "epoch": 1.16424560546875e-05,
      "model_forward_time": 3.1341938972473145,
      "step": 3052
    },
    {
      "epoch": 1.16424560546875e-05,
      "step": 3052,
      "training_step_time": 10.030829429626465
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 3.1492068767547607,
      "step": 3053
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 3053,
      "training_step_time": 9.930862188339233
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 3.1359357833862305,
      "step": 3053
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 3053,
      "training_step_time": 9.91359829902649
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 2.428884267807007,
      "step": 3053
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 3053,
      "training_step_time": 7.7416276931762695
    },
    {
      "epoch": 1.1646270751953125e-05,
      "model_forward_time": 3.128478527069092,
      "step": 3053
    },
    {
      "epoch": 1.1646270751953125e-05,
      "step": 3053,
      "training_step_time": 10.299872875213623
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 3.144221782684326,
      "step": 3054
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 3054,
      "training_step_time": 9.918512344360352
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 2.4298229217529297,
      "step": 3054
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 3054,
      "training_step_time": 7.452588319778442
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 3.123112201690674,
      "step": 3054
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 3054,
      "training_step_time": 9.885642766952515
    },
    {
      "epoch": 1.165008544921875e-05,
      "model_forward_time": 3.129594087600708,
      "step": 3054
    },
    {
      "epoch": 1.165008544921875e-05,
      "step": 3054,
      "training_step_time": 10.324978828430176
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 3.437589406967163,
      "step": 3055
    },
    {
      "dynamics_loss_avg": 0.06843733470886945,
      "epoch": 1.1653900146484376e-05,
      "step": 3055
    },
    {
      "action_loss_avg": 0.03963026637211442,
      "epoch": 1.1653900146484376e-05,
      "step": 3055
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 3055,
      "training_step_time": 10.20618987083435
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 3.1458747386932373,
      "step": 3055
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 3055,
      "training_step_time": 9.916120290756226
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 3.1318366527557373,
      "step": 3055
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 3055,
      "training_step_time": 9.89930534362793
    },
    {
      "epoch": 1.1653900146484376e-05,
      "model_forward_time": 3.1339244842529297,
      "step": 3055
    },
    {
      "epoch": 1.1653900146484376e-05,
      "step": 3055,
      "training_step_time": 10.301925897598267
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 3.1527819633483887,
      "step": 3056
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 3056,
      "training_step_time": 10.202350616455078
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 3.134860038757324,
      "step": 3056
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 3056,
      "training_step_time": 9.905952215194702
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 3.1461782455444336,
      "step": 3056
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 3056,
      "training_step_time": 9.92159914970398
    },
    {
      "epoch": 1.165771484375e-05,
      "model_forward_time": 3.135103464126587,
      "step": 3056
    },
    {
      "epoch": 1.165771484375e-05,
      "step": 3056,
      "training_step_time": 10.01456093788147
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 3.14555025100708,
      "step": 3057
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 3057,
      "training_step_time": 9.922651290893555
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 3.456319808959961,
      "step": 3057
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 3057,
      "training_step_time": 10.229706287384033
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 3.136363983154297,
      "step": 3057
    },
    {
      "dynamics_loss_avg": 0.06672379430383443,
      "epoch": 1.1661529541015625e-05,
      "step": 3057
    },
    {
      "action_loss_avg": 0.040328256972134116,
      "epoch": 1.1661529541015625e-05,
      "step": 3057
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 3057,
      "training_step_time": 9.914640426635742
    },
    {
      "epoch": 1.1661529541015625e-05,
      "model_forward_time": 3.13896107673645,
      "step": 3057
    },
    {
      "epoch": 1.1661529541015625e-05,
      "step": 3057,
      "training_step_time": 10.022496461868286
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 3.157965898513794,
      "step": 3058
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 3058,
      "training_step_time": 9.939770460128784
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 3.1380133628845215,
      "step": 3058
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 3058,
      "training_step_time": 9.91501522064209
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 3.4339592456817627,
      "step": 3058
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 3058,
      "training_step_time": 10.210923433303833
    },
    {
      "epoch": 1.166534423828125e-05,
      "model_forward_time": 3.140855312347412,
      "step": 3058
    },
    {
      "epoch": 1.166534423828125e-05,
      "step": 3058,
      "training_step_time": 10.275199174880981
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 3.1563944816589355,
      "step": 3059
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 3059,
      "training_step_time": 9.953457593917847
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 3.145199775695801,
      "step": 3059
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 3059,
      "training_step_time": 9.91997504234314
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 3.1468019485473633,
      "step": 3059
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 3059,
      "training_step_time": 9.921594381332397
    },
    {
      "epoch": 1.1669158935546876e-05,
      "model_forward_time": 3.443002223968506,
      "step": 3059
    },
    {
      "epoch": 1.1669158935546876e-05,
      "step": 3059,
      "training_step_time": 10.319903135299683
    },
    {
      "epoch": 1.16729736328125e-05,
      "grad_norm": 0.5546677112579346,
      "learning_rate": 1e-05,
      "loss": 0.1023,
      "step": 3060
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 3.1685750484466553,
      "step": 3060
    },
    {
      "dynamics_loss_avg": 0.07201741486787797,
      "epoch": 1.16729736328125e-05,
      "step": 3060
    },
    {
      "action_loss_avg": 0.047331852838397026,
      "epoch": 1.16729736328125e-05,
      "step": 3060
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 3060,
      "training_step_time": 9.956686735153198
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 3.1429927349090576,
      "step": 3060
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 3060,
      "training_step_time": 9.918407917022705
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 3.148315191268921,
      "step": 3060
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 3060,
      "training_step_time": 9.927470207214355
    },
    {
      "epoch": 1.16729736328125e-05,
      "model_forward_time": 3.1492927074432373,
      "step": 3060
    },
    {
      "epoch": 1.16729736328125e-05,
      "step": 3060,
      "training_step_time": 10.027918100357056
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 3.469417095184326,
      "step": 3061
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 3061,
      "training_step_time": 10.25312328338623
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 2.431562900543213,
      "step": 3061
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 3061,
      "training_step_time": 7.4630126953125
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 3.141512155532837,
      "step": 3061
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 3061,
      "training_step_time": 9.912217617034912
    },
    {
      "epoch": 1.1676788330078125e-05,
      "model_forward_time": 3.1301968097686768,
      "step": 3061
    },
    {
      "epoch": 1.1676788330078125e-05,
      "step": 3061,
      "training_step_time": 10.025092363357544
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 3.145587921142578,
      "step": 3062
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 3062,
      "training_step_time": 9.91823935508728
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 3.4304327964782715,
      "step": 3062
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 3062,
      "training_step_time": 10.197577476501465
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 3.13714599609375,
      "step": 3062
    },
    {
      "dynamics_loss_avg": 0.0633515764027834,
      "epoch": 1.168060302734375e-05,
      "step": 3062
    },
    {
      "action_loss_avg": 0.036239027697592976,
      "epoch": 1.168060302734375e-05,
      "step": 3062
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 3062,
      "training_step_time": 9.916936159133911
    },
    {
      "epoch": 1.168060302734375e-05,
      "model_forward_time": 3.1337831020355225,
      "step": 3062
    },
    {
      "epoch": 1.168060302734375e-05,
      "step": 3062,
      "training_step_time": 10.016613245010376
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 2.4467313289642334,
      "step": 3063
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 3063,
      "training_step_time": 7.478630781173706
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 3.132559299468994,
      "step": 3063
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 3063,
      "training_step_time": 9.905045986175537
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 3.131624221801758,
      "step": 3063
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 3063,
      "training_step_time": 9.899794578552246
    },
    {
      "epoch": 1.1684417724609376e-05,
      "model_forward_time": 3.375513792037964,
      "step": 3063
    },
    {
      "epoch": 1.1684417724609376e-05,
      "step": 3063,
      "training_step_time": 10.287678003311157
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 3.138798475265503,
      "step": 3064
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 3064,
      "training_step_time": 9.914787292480469
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 3.134526252746582,
      "step": 3064
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 3064,
      "training_step_time": 9.904964447021484
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 3.1319682598114014,
      "step": 3064
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 3064,
      "training_step_time": 9.904579639434814
    },
    {
      "epoch": 1.1688232421875e-05,
      "model_forward_time": 2.4330716133117676,
      "step": 3064
    },
    {
      "epoch": 1.1688232421875e-05,
      "step": 3064,
      "training_step_time": 10.307581663131714
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 3.141467571258545,
      "step": 3065
    },
    {
      "dynamics_loss_avg": 0.05631502401083708,
      "epoch": 1.1692047119140625e-05,
      "step": 3065
    },
    {
      "action_loss_avg": 0.035402801958844064,
      "epoch": 1.1692047119140625e-05,
      "step": 3065
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 3065,
      "training_step_time": 9.910020589828491
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 3.1289219856262207,
      "step": 3065
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 3065,
      "training_step_time": 9.892882823944092
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 3.124725103378296,
      "step": 3065
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 3065,
      "training_step_time": 9.888136863708496
    },
    {
      "epoch": 1.1692047119140625e-05,
      "model_forward_time": 3.129317045211792,
      "step": 3065
    },
    {
      "epoch": 1.1692047119140625e-05,
      "step": 3065,
      "training_step_time": 10.013712406158447
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 3.392000436782837,
      "step": 3066
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 3066,
      "training_step_time": 10.158047914505005
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 3.128890037536621,
      "step": 3066
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 3066,
      "training_step_time": 9.890508890151978
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 3.131662368774414,
      "step": 3066
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 3066,
      "training_step_time": 9.895817041397095
    },
    {
      "epoch": 1.169586181640625e-05,
      "model_forward_time": 3.1331753730773926,
      "step": 3066
    },
    {
      "epoch": 1.169586181640625e-05,
      "step": 3066,
      "training_step_time": 10.014990091323853
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 3.1388790607452393,
      "step": 3067
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 3067,
      "training_step_time": 9.905281782150269
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 3.409605026245117,
      "step": 3067
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 3067,
      "training_step_time": 10.17461633682251
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 3.1323082447052,
      "step": 3067
    },
    {
      "dynamics_loss_avg": 0.05533694662153721,
      "epoch": 1.1699676513671876e-05,
      "step": 3067
    },
    {
      "action_loss_avg": 0.02828496117144823,
      "epoch": 1.1699676513671876e-05,
      "step": 3067
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 3067,
      "training_step_time": 9.903011322021484
    },
    {
      "epoch": 1.1699676513671876e-05,
      "model_forward_time": 3.1375463008880615,
      "step": 3067
    },
    {
      "epoch": 1.1699676513671876e-05,
      "step": 3067,
      "training_step_time": 10.22141432762146
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 3.148461103439331,
      "step": 3068
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 3068,
      "training_step_time": 9.9192054271698
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 2.4416775703430176,
      "step": 3068
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 3068,
      "training_step_time": 7.472797632217407
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 2.7614963054656982,
      "step": 3068
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 3068,
      "training_step_time": 7.791542291641235
    },
    {
      "epoch": 1.17034912109375e-05,
      "model_forward_time": 3.1422407627105713,
      "step": 3068
    },
    {
      "epoch": 1.17034912109375e-05,
      "step": 3068,
      "training_step_time": 10.325310230255127
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 3.1568892002105713,
      "step": 3069
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 3069,
      "training_step_time": 9.934858083724976
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 3.138503313064575,
      "step": 3069
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 3069,
      "training_step_time": 9.908465147018433
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 3.139017343521118,
      "step": 3069
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 3069,
      "training_step_time": 9.914361715316772
    },
    {
      "epoch": 1.1707305908203125e-05,
      "model_forward_time": 3.141470432281494,
      "step": 3069
    },
    {
      "epoch": 1.1707305908203125e-05,
      "step": 3069,
      "training_step_time": 10.287712097167969
    },
    {
      "epoch": 1.171112060546875e-05,
      "grad_norm": 0.7069768905639648,
      "learning_rate": 1e-05,
      "loss": 0.1014,
      "step": 3070
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 2.464367628097534,
      "step": 3070
    },
    {
      "dynamics_loss_avg": 0.05878625204786658,
      "epoch": 1.171112060546875e-05,
      "step": 3070
    },
    {
      "action_loss_avg": 0.03980722571723163,
      "epoch": 1.171112060546875e-05,
      "step": 3070
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 3070,
      "training_step_time": 7.4971723556518555
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 3.135256052017212,
      "step": 3070
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 3070,
      "training_step_time": 9.9078950881958
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 3.1346144676208496,
      "step": 3070
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 3070,
      "training_step_time": 9.90182375907898
    },
    {
      "epoch": 1.171112060546875e-05,
      "model_forward_time": 2.4361932277679443,
      "step": 3070
    },
    {
      "epoch": 1.171112060546875e-05,
      "step": 3070,
      "training_step_time": 10.31665825843811
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 3.4709441661834717,
      "step": 3071
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 3071,
      "training_step_time": 10.254530191421509
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 3.1345596313476562,
      "step": 3071
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 3071,
      "training_step_time": 9.91072964668274
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 3.1301612854003906,
      "step": 3071
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 3071,
      "training_step_time": 9.900315761566162
    },
    {
      "epoch": 1.1714935302734376e-05,
      "model_forward_time": 3.1291849613189697,
      "step": 3071
    },
    {
      "epoch": 1.1714935302734376e-05,
      "step": 3071,
      "training_step_time": 10.019339799880981
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 3.1413309574127197,
      "step": 3072
    },
    {
      "epoch": 1.171875e-05,
      "step": 3072,
      "training_step_time": 9.913907051086426
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 3.1306729316711426,
      "step": 3072
    },
    {
      "epoch": 1.171875e-05,
      "step": 3072,
      "training_step_time": 9.899806499481201
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 3.4114937782287598,
      "step": 3072
    },
    {
      "dynamics_loss_avg": 0.058150168135762216,
      "epoch": 1.171875e-05,
      "step": 3072
    },
    {
      "action_loss_avg": 0.03395978705957532,
      "epoch": 1.171875e-05,
      "step": 3072
    },
    {
      "epoch": 1.171875e-05,
      "step": 3072,
      "training_step_time": 10.186647653579712
    },
    {
      "epoch": 1.171875e-05,
      "model_forward_time": 3.134432792663574,
      "step": 3072
    },
    {
      "epoch": 1.171875e-05,
      "step": 3072,
      "training_step_time": 10.28886365890503
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 3.1810476779937744,
      "step": 3073
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 3073,
      "training_step_time": 9.95548677444458
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 3.1457066535949707,
      "step": 3073
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 3073,
      "training_step_time": 9.919322729110718
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 2.438020944595337,
      "step": 3073
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 3073,
      "training_step_time": 7.779868841171265
    },
    {
      "epoch": 1.1722564697265625e-05,
      "model_forward_time": 3.1343891620635986,
      "step": 3073
    },
    {
      "epoch": 1.1722564697265625e-05,
      "step": 3073,
      "training_step_time": 10.028755187988281
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 3.1724088191986084,
      "step": 3074
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 3074,
      "training_step_time": 9.946194648742676
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 2.4300832748413086,
      "step": 3074
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 3074,
      "training_step_time": 7.454877138137817
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 3.1285955905914307,
      "step": 3074
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 3074,
      "training_step_time": 9.896713256835938
    },
    {
      "epoch": 1.172637939453125e-05,
      "model_forward_time": 3.132067918777466,
      "step": 3074
    },
    {
      "epoch": 1.172637939453125e-05,
      "step": 3074,
      "training_step_time": 10.282811880111694
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 3.4666216373443604,
      "step": 3075
    },
    {
      "dynamics_loss_avg": 0.07887580879032612,
      "epoch": 1.1730194091796876e-05,
      "step": 3075
    },
    {
      "action_loss_avg": 0.036959486082196236,
      "epoch": 1.1730194091796876e-05,
      "step": 3075
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 3075,
      "training_step_time": 10.246140956878662
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 3.134971857070923,
      "step": 3075
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 3075,
      "training_step_time": 9.910948753356934
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 3.1307222843170166,
      "step": 3075
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 3075,
      "training_step_time": 9.897292613983154
    },
    {
      "epoch": 1.1730194091796876e-05,
      "model_forward_time": 3.1340675354003906,
      "step": 3075
    },
    {
      "epoch": 1.1730194091796876e-05,
      "step": 3075,
      "training_step_time": 10.304352760314941
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 3.14988112449646,
      "step": 3076
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 3076,
      "training_step_time": 10.20929765701294
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 3.1429319381713867,
      "step": 3076
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 3076,
      "training_step_time": 9.92124319076538
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 3.142756700515747,
      "step": 3076
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 3076,
      "training_step_time": 9.922244310379028
    },
    {
      "epoch": 1.17340087890625e-05,
      "model_forward_time": 3.1516523361206055,
      "step": 3076
    },
    {
      "epoch": 1.17340087890625e-05,
      "step": 3076,
      "training_step_time": 10.027702331542969
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 2.4504079818725586,
      "step": 3077
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 3077,
      "training_step_time": 7.486204624176025
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 3.138981819152832,
      "step": 3077
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 3077,
      "training_step_time": 10.213463068008423
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 3.1407933235168457,
      "step": 3077
    },
    {
      "dynamics_loss_avg": 0.06668213605880738,
      "epoch": 1.1737823486328125e-05,
      "step": 3077
    },
    {
      "action_loss_avg": 0.02688629454933107,
      "epoch": 1.1737823486328125e-05,
      "step": 3077
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 3077,
      "training_step_time": 9.912151575088501
    },
    {
      "epoch": 1.1737823486328125e-05,
      "model_forward_time": 2.4376542568206787,
      "step": 3077
    },
    {
      "epoch": 1.1737823486328125e-05,
      "step": 3077,
      "training_step_time": 10.044207334518433
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 3.151886463165283,
      "step": 3078
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 3078,
      "training_step_time": 9.927311897277832
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 3.1438629627227783,
      "step": 3078
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 3078,
      "training_step_time": 9.916970491409302
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 3.1479783058166504,
      "step": 3078
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 3078,
      "training_step_time": 9.92665410041809
    },
    {
      "epoch": 1.174163818359375e-05,
      "model_forward_time": 3.4265265464782715,
      "step": 3078
    },
    {
      "epoch": 1.174163818359375e-05,
      "step": 3078,
      "training_step_time": 10.301512002944946
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 3.158860683441162,
      "step": 3079
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 3079,
      "training_step_time": 9.934393882751465
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 3.139346122741699,
      "step": 3079
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 3079,
      "training_step_time": 9.909410953521729
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 3.145632743835449,
      "step": 3079
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 3079,
      "training_step_time": 9.91521668434143
    },
    {
      "epoch": 1.1745452880859376e-05,
      "model_forward_time": 3.419996738433838,
      "step": 3079
    },
    {
      "epoch": 1.1745452880859376e-05,
      "step": 3079,
      "training_step_time": 10.278473615646362
    },
    {
      "epoch": 1.1749267578125e-05,
      "grad_norm": 0.5373247265815735,
      "learning_rate": 1e-05,
      "loss": 0.1038,
      "step": 3080
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 3.143524646759033,
      "step": 3080
    },
    {
      "dynamics_loss_avg": 0.06019109189510345,
      "epoch": 1.1749267578125e-05,
      "step": 3080
    },
    {
      "action_loss_avg": 0.03595096850767732,
      "epoch": 1.1749267578125e-05,
      "step": 3080
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 3080,
      "training_step_time": 9.915618896484375
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 3.1358306407928467,
      "step": 3080
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 3080,
      "training_step_time": 9.905896425247192
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 3.139068603515625,
      "step": 3080
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 3080,
      "training_step_time": 9.904339075088501
    },
    {
      "epoch": 1.1749267578125e-05,
      "model_forward_time": 3.133855104446411,
      "step": 3080
    },
    {
      "epoch": 1.1749267578125e-05,
      "step": 3080,
      "training_step_time": 10.28659176826477
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 3.145409107208252,
      "step": 3081
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 3081,
      "training_step_time": 10.174323320388794
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 3.13590145111084,
      "step": 3081
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 3081,
      "training_step_time": 9.904077053070068
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 3.1417133808135986,
      "step": 3081
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 3081,
      "training_step_time": 9.904848575592041
    },
    {
      "epoch": 1.1753082275390625e-05,
      "model_forward_time": 3.1363203525543213,
      "step": 3081
    },
    {
      "epoch": 1.1753082275390625e-05,
      "step": 3081,
      "training_step_time": 10.307841300964355
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 3.1554694175720215,
      "step": 3082
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 3082,
      "training_step_time": 9.923073053359985
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 3.4372353553771973,
      "step": 3082
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 3082,
      "training_step_time": 10.20786738395691
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 3.1437737941741943,
      "step": 3082
    },
    {
      "dynamics_loss_avg": 0.06076881587505341,
      "epoch": 1.175689697265625e-05,
      "step": 3082
    },
    {
      "action_loss_avg": 0.033839347027242186,
      "epoch": 1.175689697265625e-05,
      "step": 3082
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 3082,
      "training_step_time": 9.918055057525635
    },
    {
      "epoch": 1.175689697265625e-05,
      "model_forward_time": 2.4407129287719727,
      "step": 3082
    },
    {
      "epoch": 1.175689697265625e-05,
      "step": 3082,
      "training_step_time": 10.026811838150024
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 3.1510555744171143,
      "step": 3083
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 3083,
      "training_step_time": 9.933260202407837
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 3.1448934078216553,
      "step": 3083
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 3083,
      "training_step_time": 9.916532754898071
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 2.4367587566375732,
      "step": 3083
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 3083,
      "training_step_time": 7.460374116897583
    },
    {
      "epoch": 1.1760711669921876e-05,
      "model_forward_time": 3.404493570327759,
      "step": 3083
    },
    {
      "epoch": 1.1760711669921876e-05,
      "step": 3083,
      "training_step_time": 10.268473386764526
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 3.141303539276123,
      "step": 3084
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 3084,
      "training_step_time": 9.906371831893921
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 3.1362836360931396,
      "step": 3084
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 3084,
      "training_step_time": 9.900200366973877
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 3.1393239498138428,
      "step": 3084
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 3084,
      "training_step_time": 9.911790609359741
    },
    {
      "epoch": 1.17645263671875e-05,
      "model_forward_time": 3.139101982116699,
      "step": 3084
    },
    {
      "epoch": 1.17645263671875e-05,
      "step": 3084,
      "training_step_time": 10.28299593925476
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 3.1458470821380615,
      "step": 3085
    },
    {
      "dynamics_loss_avg": 0.061408122442662715,
      "epoch": 1.1768341064453125e-05,
      "step": 3085
    },
    {
      "action_loss_avg": 0.03810504050925374,
      "epoch": 1.1768341064453125e-05,
      "step": 3085
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 3085,
      "training_step_time": 9.920440912246704
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 3.141805648803711,
      "step": 3085
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 3085,
      "training_step_time": 9.910105466842651
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 2.4359841346740723,
      "step": 3085
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 3085,
      "training_step_time": 7.4602015018463135
    },
    {
      "epoch": 1.1768341064453125e-05,
      "model_forward_time": 3.137518882751465,
      "step": 3085
    },
    {
      "epoch": 1.1768341064453125e-05,
      "step": 3085,
      "training_step_time": 10.318589448928833
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 2.442406177520752,
      "step": 3086
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 3086,
      "training_step_time": 7.46299147605896
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 3.4089155197143555,
      "step": 3086
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 3086,
      "training_step_time": 10.172733783721924
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 3.137078285217285,
      "step": 3086
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 3086,
      "training_step_time": 9.908246040344238
    },
    {
      "epoch": 1.177215576171875e-05,
      "model_forward_time": 3.1430752277374268,
      "step": 3086
    },
    {
      "epoch": 1.177215576171875e-05,
      "step": 3086,
      "training_step_time": 10.014107704162598
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 3.1544570922851562,
      "step": 3087
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 3087,
      "training_step_time": 9.932538032531738
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 3.1470344066619873,
      "step": 3087
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 3087,
      "training_step_time": 10.220784664154053
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 3.1383819580078125,
      "step": 3087
    },
    {
      "dynamics_loss_avg": 0.06998527739197016,
      "epoch": 1.1775970458984376e-05,
      "step": 3087
    },
    {
      "action_loss_avg": 0.039358255174010995,
      "epoch": 1.1775970458984376e-05,
      "step": 3087
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 3087,
      "training_step_time": 9.90989637374878
    },
    {
      "epoch": 1.1775970458984376e-05,
      "model_forward_time": 3.1414966583251953,
      "step": 3087
    },
    {
      "epoch": 1.1775970458984376e-05,
      "step": 3087,
      "training_step_time": 10.030312776565552
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 3.1555099487304688,
      "step": 3088
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 3088,
      "training_step_time": 9.92589783668518
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 3.138861656188965,
      "step": 3088
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 3088,
      "training_step_time": 9.90566110610962
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 3.404548168182373,
      "step": 3088
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 3088,
      "training_step_time": 10.169114828109741
    },
    {
      "epoch": 1.177978515625e-05,
      "model_forward_time": 3.140136480331421,
      "step": 3088
    },
    {
      "epoch": 1.177978515625e-05,
      "step": 3088,
      "training_step_time": 10.02518343925476
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 3.147538661956787,
      "step": 3089
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 3089,
      "training_step_time": 9.911553144454956
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 3.13485050201416,
      "step": 3089
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 3089,
      "training_step_time": 9.903206825256348
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 3.140199661254883,
      "step": 3089
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 3089,
      "training_step_time": 9.904970169067383
    },
    {
      "epoch": 1.1783599853515625e-05,
      "model_forward_time": 3.424380302429199,
      "step": 3089
    },
    {
      "epoch": 1.1783599853515625e-05,
      "step": 3089,
      "training_step_time": 10.283615827560425
    },
    {
      "epoch": 1.178741455078125e-05,
      "grad_norm": 0.5609635710716248,
      "learning_rate": 1e-05,
      "loss": 0.1013,
      "step": 3090
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 3.147204875946045,
      "step": 3090
    },
    {
      "dynamics_loss_avg": 0.07208442352712155,
      "epoch": 1.178741455078125e-05,
      "step": 3090
    },
    {
      "action_loss_avg": 0.03644418017938733,
      "epoch": 1.178741455078125e-05,
      "step": 3090
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 3090,
      "training_step_time": 9.915539026260376
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 3.135735511779785,
      "step": 3090
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 3090,
      "training_step_time": 9.896778583526611
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 3.143031597137451,
      "step": 3090
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 3090,
      "training_step_time": 9.910489559173584
    },
    {
      "epoch": 1.178741455078125e-05,
      "model_forward_time": 3.143622875213623,
      "step": 3090
    },
    {
      "epoch": 1.178741455078125e-05,
      "step": 3090,
      "training_step_time": 10.284724950790405
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 3.459012508392334,
      "step": 3091
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 3091,
      "training_step_time": 10.221277713775635
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 2.439676523208618,
      "step": 3091
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 3091,
      "training_step_time": 7.4607908725738525
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 3.137524366378784,
      "step": 3091
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 3091,
      "training_step_time": 9.901087284088135
    },
    {
      "epoch": 1.1791229248046876e-05,
      "model_forward_time": 3.1355857849121094,
      "step": 3091
    },
    {
      "epoch": 1.1791229248046876e-05,
      "step": 3091,
      "training_step_time": 10.338257551193237
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 3.142638921737671,
      "step": 3092
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 3092,
      "training_step_time": 9.91046690940857
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 3.141303539276123,
      "step": 3092
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 3092,
      "training_step_time": 10.18430781364441
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 3.1384389400482178,
      "step": 3092
    },
    {
      "dynamics_loss_avg": 0.07477985043078661,
      "epoch": 1.17950439453125e-05,
      "step": 3092
    },
    {
      "action_loss_avg": 0.03150533246807754,
      "epoch": 1.17950439453125e-05,
      "step": 3092
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 3092,
      "training_step_time": 9.908166885375977
    },
    {
      "epoch": 1.17950439453125e-05,
      "model_forward_time": 3.1410348415374756,
      "step": 3092
    },
    {
      "epoch": 1.17950439453125e-05,
      "step": 3092,
      "training_step_time": 10.302452325820923
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 3.1479685306549072,
      "step": 3093
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 3093,
      "training_step_time": 9.92276930809021
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 3.1395599842071533,
      "step": 3093
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 3093,
      "training_step_time": 9.908665180206299
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 3.4132978916168213,
      "step": 3093
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 3093,
      "training_step_time": 10.184192180633545
    },
    {
      "epoch": 1.1798858642578125e-05,
      "model_forward_time": 3.129258632659912,
      "step": 3093
    },
    {
      "epoch": 1.1798858642578125e-05,
      "step": 3093,
      "training_step_time": 10.022207021713257
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 2.4392731189727783,
      "step": 3094
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 3094,
      "training_step_time": 7.4595582485198975
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 3.133946180343628,
      "step": 3094
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 3094,
      "training_step_time": 9.898141622543335
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 3.123209238052368,
      "step": 3094
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 3094,
      "training_step_time": 9.885997533798218
    },
    {
      "epoch": 1.180267333984375e-05,
      "model_forward_time": 2.429656982421875,
      "step": 3094
    },
    {
      "epoch": 1.180267333984375e-05,
      "step": 3094,
      "training_step_time": 10.261731386184692
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 3.4751462936401367,
      "step": 3095
    },
    {
      "dynamics_loss_avg": 0.0693552928045392,
      "epoch": 1.1806488037109376e-05,
      "step": 3095
    },
    {
      "action_loss_avg": 0.032188774272799495,
      "epoch": 1.1806488037109376e-05,
      "step": 3095
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 3095,
      "training_step_time": 10.243162631988525
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 3.1384284496307373,
      "step": 3095
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 3095,
      "training_step_time": 9.901591062545776
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 3.138300895690918,
      "step": 3095
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 3095,
      "training_step_time": 9.906182765960693
    },
    {
      "epoch": 1.1806488037109376e-05,
      "model_forward_time": 3.1330983638763428,
      "step": 3095
    },
    {
      "epoch": 1.1806488037109376e-05,
      "step": 3095,
      "training_step_time": 10.01198697090149
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 3.4505221843719482,
      "step": 3096
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 3096,
      "training_step_time": 10.219854354858398
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 3.1340856552124023,
      "step": 3096
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 3096,
      "training_step_time": 9.904853820800781
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 3.130842447280884,
      "step": 3096
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 3096,
      "training_step_time": 9.892653942108154
    },
    {
      "epoch": 1.1810302734375e-05,
      "model_forward_time": 3.129315137863159,
      "step": 3096
    },
    {
      "epoch": 1.1810302734375e-05,
      "step": 3096,
      "training_step_time": 10.279870748519897
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 3.1490371227264404,
      "step": 3097
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 3097,
      "training_step_time": 9.923324346542358
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 3.4288289546966553,
      "step": 3097
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 3097,
      "training_step_time": 10.190118789672852
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 3.1306495666503906,
      "step": 3097
    },
    {
      "dynamics_loss_avg": 0.06700712591409683,
      "epoch": 1.1814117431640625e-05,
      "step": 3097
    },
    {
      "action_loss_avg": 0.040359537862241265,
      "epoch": 1.1814117431640625e-05,
      "step": 3097
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 3097,
      "training_step_time": 9.896069765090942
    },
    {
      "epoch": 1.1814117431640625e-05,
      "model_forward_time": 3.133056879043579,
      "step": 3097
    },
    {
      "epoch": 1.1814117431640625e-05,
      "step": 3097,
      "training_step_time": 10.008095026016235
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 3.1507551670074463,
      "step": 3098
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 3098,
      "training_step_time": 9.91796088218689
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 3.134262800216675,
      "step": 3098
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 3098,
      "training_step_time": 9.896960973739624
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 3.4285151958465576,
      "step": 3098
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 3098,
      "training_step_time": 10.191367626190186
    },
    {
      "epoch": 1.181793212890625e-05,
      "model_forward_time": 3.1381263732910156,
      "step": 3098
    },
    {
      "epoch": 1.181793212890625e-05,
      "step": 3098,
      "training_step_time": 10.261951446533203
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 3.148611068725586,
      "step": 3099
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 3099,
      "training_step_time": 9.922833442687988
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 3.13702654838562,
      "step": 3099
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 3099,
      "training_step_time": 9.900946617126465
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 3.136384963989258,
      "step": 3099
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 3099,
      "training_step_time": 9.90721583366394
    },
    {
      "epoch": 1.1821746826171876e-05,
      "model_forward_time": 3.4134416580200195,
      "step": 3099
    },
    {
      "epoch": 1.1821746826171876e-05,
      "step": 3099,
      "training_step_time": 10.300234079360962
    },
    {
      "epoch": 1.18255615234375e-05,
      "grad_norm": 0.45978161692619324,
      "learning_rate": 1e-05,
      "loss": 0.1036,
      "step": 3100
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 3.142612934112549,
      "step": 3100
    },
    {
      "dynamics_loss_avg": 0.06243279697373509,
      "epoch": 1.18255615234375e-05,
      "step": 3100
    },
    {
      "action_loss_avg": 0.030624959943816067,
      "epoch": 1.18255615234375e-05,
      "step": 3100
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 3100,
      "training_step_time": 9.908415794372559
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 3.131638526916504,
      "step": 3100
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 3100,
      "training_step_time": 9.897780656814575
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 2.4319007396698,
      "step": 3100
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 3100,
      "training_step_time": 7.4562084674835205
    },
    {
      "epoch": 1.18255615234375e-05,
      "model_forward_time": 3.132408380508423,
      "step": 3100
    },
    {
      "epoch": 1.18255615234375e-05,
      "step": 3100,
      "training_step_time": 10.019577741622925
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 3.1456432342529297,
      "step": 3101
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 3101,
      "training_step_time": 10.194932460784912
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 3.13836669921875,
      "step": 3101
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 3101,
      "training_step_time": 9.905460834503174
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 2.432915210723877,
      "step": 3101
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 3101,
      "training_step_time": 7.457770586013794
    },
    {
      "epoch": 1.1829376220703125e-05,
      "model_forward_time": 2.4302823543548584,
      "step": 3101
    },
    {
      "epoch": 1.1829376220703125e-05,
      "step": 3101,
      "training_step_time": 10.300451278686523
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 3.1472275257110596,
      "step": 3102
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 3102,
      "training_step_time": 9.917323112487793
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 3.138040065765381,
      "step": 3102
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 3102,
      "training_step_time": 10.1975998878479
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 3.1379659175872803,
      "step": 3102
    },
    {
      "dynamics_loss_avg": 0.07036622501909733,
      "epoch": 1.183319091796875e-05,
      "step": 3102
    },
    {
      "action_loss_avg": 0.03603257331997156,
      "epoch": 1.183319091796875e-05,
      "step": 3102
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 3102,
      "training_step_time": 9.899436712265015
    },
    {
      "epoch": 1.183319091796875e-05,
      "model_forward_time": 3.132619857788086,
      "step": 3102
    },
    {
      "epoch": 1.183319091796875e-05,
      "step": 3102,
      "training_step_time": 10.02257776260376
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 3.153555393218994,
      "step": 3103
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 3103,
      "training_step_time": 9.915705442428589
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 3.143954277038574,
      "step": 3103
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 3103,
      "training_step_time": 9.9177565574646
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 2.437734603881836,
      "step": 3103
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 3103,
      "training_step_time": 7.460212707519531
    },
    {
      "epoch": 1.1837005615234376e-05,
      "model_forward_time": 3.4188120365142822,
      "step": 3103
    },
    {
      "epoch": 1.1837005615234376e-05,
      "step": 3103,
      "training_step_time": 10.276894807815552
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 2.4506707191467285,
      "step": 3104
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 3104,
      "training_step_time": 7.474437236785889
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 3.150272846221924,
      "step": 3104
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 3104,
      "training_step_time": 9.918861150741577
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 3.146899461746216,
      "step": 3104
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 3104,
      "training_step_time": 9.921090841293335
    },
    {
      "epoch": 1.18408203125e-05,
      "model_forward_time": 3.1564249992370605,
      "step": 3104
    },
    {
      "epoch": 1.18408203125e-05,
      "step": 3104,
      "training_step_time": 10.333614826202393
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 3.1659605503082275,
      "step": 3105
    },
    {
      "dynamics_loss_avg": 0.05431142691522837,
      "epoch": 1.1844635009765625e-05,
      "step": 3105
    },
    {
      "action_loss_avg": 0.02910027150064707,
      "epoch": 1.1844635009765625e-05,
      "step": 3105
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 3105,
      "training_step_time": 9.938874959945679
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 3.1501853466033936,
      "step": 3105
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 3105,
      "training_step_time": 9.915725708007812
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 2.444594144821167,
      "step": 3105
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 3105,
      "training_step_time": 7.472339391708374
    },
    {
      "epoch": 1.1844635009765625e-05,
      "model_forward_time": 3.145455837249756,
      "step": 3105
    },
    {
      "epoch": 1.1844635009765625e-05,
      "step": 3105,
      "training_step_time": 10.030264377593994
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 3.417565107345581,
      "step": 3106
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 3106,
      "training_step_time": 10.19184947013855
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 3.152038097381592,
      "step": 3106
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 3106,
      "training_step_time": 9.922889709472656
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 3.1405251026153564,
      "step": 3106
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 3106,
      "training_step_time": 9.901367425918579
    },
    {
      "epoch": 1.184844970703125e-05,
      "model_forward_time": 3.142270803451538,
      "step": 3106
    },
    {
      "epoch": 1.184844970703125e-05,
      "step": 3106,
      "training_step_time": 10.324651718139648
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 3.152209758758545,
      "step": 3107
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 3107,
      "training_step_time": 9.917741060256958
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 2.439208984375,
      "step": 3107
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 3107,
      "training_step_time": 7.464145183563232
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 3.407376766204834,
      "step": 3107
    },
    {
      "dynamics_loss_avg": 0.05555412881076336,
      "epoch": 1.1852264404296876e-05,
      "step": 3107
    },
    {
      "action_loss_avg": 0.03381836283951998,
      "epoch": 1.1852264404296876e-05,
      "step": 3107
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 3107,
      "training_step_time": 10.180180788040161
    },
    {
      "epoch": 1.1852264404296876e-05,
      "model_forward_time": 3.154093027114868,
      "step": 3107
    },
    {
      "epoch": 1.1852264404296876e-05,
      "step": 3107,
      "training_step_time": 10.335756063461304
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 3.1481494903564453,
      "step": 3108
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 3108,
      "training_step_time": 9.92024540901184
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 2.434795379638672,
      "step": 3108
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 3108,
      "training_step_time": 7.457556247711182
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 3.397993803024292,
      "step": 3108
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 3108,
      "training_step_time": 10.161793947219849
    },
    {
      "epoch": 1.18560791015625e-05,
      "model_forward_time": 3.1399192810058594,
      "step": 3108
    },
    {
      "epoch": 1.18560791015625e-05,
      "step": 3108,
      "training_step_time": 10.033589839935303
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 3.1658730506896973,
      "step": 3109
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 3109,
      "training_step_time": 9.936371326446533
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 3.139296054840088,
      "step": 3109
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 3109,
      "training_step_time": 9.904181957244873
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 3.1454973220825195,
      "step": 3109
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 3109,
      "training_step_time": 9.913717031478882
    },
    {
      "epoch": 1.1859893798828125e-05,
      "model_forward_time": 3.1430013179779053,
      "step": 3109
    },
    {
      "epoch": 1.1859893798828125e-05,
      "step": 3109,
      "training_step_time": 10.307225942611694
    },
    {
      "epoch": 1.186370849609375e-05,
      "grad_norm": 0.5718832612037659,
      "learning_rate": 1e-05,
      "loss": 0.0988,
      "step": 3110
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 3.1538069248199463,
      "step": 3110
    },
    {
      "dynamics_loss_avg": 0.05059948097914457,
      "epoch": 1.186370849609375e-05,
      "step": 3110
    },
    {
      "action_loss_avg": 0.03971821255981922,
      "epoch": 1.186370849609375e-05,
      "step": 3110
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 3110,
      "training_step_time": 9.922433376312256
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 3.1381020545959473,
      "step": 3110
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 3110,
      "training_step_time": 9.90851902961731
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 2.4371933937072754,
      "step": 3110
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 3110,
      "training_step_time": 7.466450214385986
    },
    {
      "epoch": 1.186370849609375e-05,
      "model_forward_time": 3.1363935470581055,
      "step": 3110
    },
    {
      "epoch": 1.186370849609375e-05,
      "step": 3110,
      "training_step_time": 10.029401540756226
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 3.4271774291992188,
      "step": 3111
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 3111,
      "training_step_time": 10.20058012008667
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 3.141692876815796,
      "step": 3111
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 3111,
      "training_step_time": 9.910265684127808
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 3.1414947509765625,
      "step": 3111
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 3111,
      "training_step_time": 9.910583019256592
    },
    {
      "epoch": 1.1867523193359376e-05,
      "model_forward_time": 3.1417133808135986,
      "step": 3111
    },
    {
      "epoch": 1.1867523193359376e-05,
      "step": 3111,
      "training_step_time": 10.310282945632935
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 3.1552891731262207,
      "step": 3112
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 3112,
      "training_step_time": 9.935726165771484
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 3.403364658355713,
      "step": 3112
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 3112,
      "training_step_time": 10.17273736000061
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 3.1433825492858887,
      "step": 3112
    },
    {
      "dynamics_loss_avg": 0.054436526447534564,
      "epoch": 1.1871337890625e-05,
      "step": 3112
    },
    {
      "action_loss_avg": 0.02652987465262413,
      "epoch": 1.1871337890625e-05,
      "step": 3112
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 3112,
      "training_step_time": 9.916449308395386
    },
    {
      "epoch": 1.1871337890625e-05,
      "model_forward_time": 3.1417012214660645,
      "step": 3112
    },
    {
      "epoch": 1.1871337890625e-05,
      "step": 3112,
      "training_step_time": 10.313337802886963
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 3.1406030654907227,
      "step": 3113
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 3113,
      "training_step_time": 9.913264274597168
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 3.141767740249634,
      "step": 3113
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 3113,
      "training_step_time": 9.914416551589966
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 3.1343212127685547,
      "step": 3113
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 3113,
      "training_step_time": 10.199178218841553
    },
    {
      "epoch": 1.1875152587890625e-05,
      "model_forward_time": 3.134073495864868,
      "step": 3113
    },
    {
      "epoch": 1.1875152587890625e-05,
      "step": 3113,
      "training_step_time": 10.01407790184021
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 2.451850652694702,
      "step": 3114
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 3114,
      "training_step_time": 7.4781494140625
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 2.431826591491699,
      "step": 3114
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 3114,
      "training_step_time": 7.456149101257324
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 3.135524034500122,
      "step": 3114
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 3114,
      "training_step_time": 9.909132242202759
    },
    {
      "epoch": 1.187896728515625e-05,
      "model_forward_time": 2.434041976928711,
      "step": 3114
    },
    {
      "epoch": 1.187896728515625e-05,
      "step": 3114,
      "training_step_time": 10.02775263786316
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 3.4313583374023438,
      "step": 3115
    },
    {
      "dynamics_loss_avg": 0.06160086300224066,
      "epoch": 1.1882781982421876e-05,
      "step": 3115
    },
    {
      "action_loss_avg": 0.030529920291155577,
      "epoch": 1.1882781982421876e-05,
      "step": 3115
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 3115,
      "training_step_time": 10.199815034866333
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 3.1410980224609375,
      "step": 3115
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 3115,
      "training_step_time": 9.909629583358765
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 3.1398355960845947,
      "step": 3115
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 3115,
      "training_step_time": 9.908418655395508
    },
    {
      "epoch": 1.1882781982421876e-05,
      "model_forward_time": 3.1382312774658203,
      "step": 3115
    },
    {
      "epoch": 1.1882781982421876e-05,
      "step": 3115,
      "training_step_time": 10.019264459609985
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 3.1612277030944824,
      "step": 3116
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 3116,
      "training_step_time": 9.93398928642273
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 3.4011647701263428,
      "step": 3116
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 3116,
      "training_step_time": 10.173673152923584
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 3.1351380348205566,
      "step": 3116
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 3116,
      "training_step_time": 9.908675909042358
    },
    {
      "epoch": 1.18865966796875e-05,
      "model_forward_time": 3.1405651569366455,
      "step": 3116
    },
    {
      "epoch": 1.18865966796875e-05,
      "step": 3116,
      "training_step_time": 10.023270845413208
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 3.1540215015411377,
      "step": 3117
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 3117,
      "training_step_time": 9.932204484939575
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 3.134378433227539,
      "step": 3117
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 3117,
      "training_step_time": 9.919463872909546
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 3.3927223682403564,
      "step": 3117
    },
    {
      "dynamics_loss_avg": 0.06468128394335508,
      "epoch": 1.1890411376953125e-05,
      "step": 3117
    },
    {
      "action_loss_avg": 0.040267480071634054,
      "epoch": 1.1890411376953125e-05,
      "step": 3117
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 3117,
      "training_step_time": 10.181246042251587
    },
    {
      "epoch": 1.1890411376953125e-05,
      "model_forward_time": 3.1449930667877197,
      "step": 3117
    },
    {
      "epoch": 1.1890411376953125e-05,
      "step": 3117,
      "training_step_time": 10.299824953079224
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 3.152412176132202,
      "step": 3118
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 3118,
      "training_step_time": 9.938706874847412
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 3.1401636600494385,
      "step": 3118
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 3118,
      "training_step_time": 9.924550533294678
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 3.140901803970337,
      "step": 3118
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 3118,
      "training_step_time": 9.924412488937378
    },
    {
      "epoch": 1.189422607421875e-05,
      "model_forward_time": 3.4088311195373535,
      "step": 3118
    },
    {
      "epoch": 1.189422607421875e-05,
      "step": 3118,
      "training_step_time": 10.284627437591553
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 3.1538209915161133,
      "step": 3119
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 3119,
      "training_step_time": 9.936374187469482
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 3.1376895904541016,
      "step": 3119
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 3119,
      "training_step_time": 9.915064811706543
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 3.1502928733825684,
      "step": 3119
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 3119,
      "training_step_time": 9.92725157737732
    },
    {
      "epoch": 1.1898040771484376e-05,
      "model_forward_time": 3.1450936794281006,
      "step": 3119
    },
    {
      "epoch": 1.1898040771484376e-05,
      "step": 3119,
      "training_step_time": 10.277631998062134
    },
    {
      "epoch": 1.190185546875e-05,
      "grad_norm": 0.575046181678772,
      "learning_rate": 1e-05,
      "loss": 0.0994,
      "step": 3120
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 3.151595115661621,
      "step": 3120
    },
    {
      "dynamics_loss_avg": 0.05646052416414023,
      "epoch": 1.190185546875e-05,
      "step": 3120
    },
    {
      "action_loss_avg": 0.03827879652380943,
      "epoch": 1.190185546875e-05,
      "step": 3120
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 3120,
      "training_step_time": 9.92828369140625
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 3.1309916973114014,
      "step": 3120
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 3120,
      "training_step_time": 9.901243686676025
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 3.132678270339966,
      "step": 3120
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 3120,
      "training_step_time": 9.897898435592651
    },
    {
      "epoch": 1.190185546875e-05,
      "model_forward_time": 3.134902000427246,
      "step": 3120
    },
    {
      "epoch": 1.190185546875e-05,
      "step": 3120,
      "training_step_time": 10.305328607559204
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 2.437917470932007,
      "step": 3121
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 3121,
      "training_step_time": 7.466267347335815
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 3.373671770095825,
      "step": 3121
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 3121,
      "training_step_time": 10.136089563369751
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 3.1287500858306885,
      "step": 3121
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 3121,
      "training_step_time": 9.894133567810059
    },
    {
      "epoch": 1.1905670166015625e-05,
      "model_forward_time": 3.130526542663574,
      "step": 3121
    },
    {
      "epoch": 1.1905670166015625e-05,
      "step": 3121,
      "training_step_time": 10.269803524017334
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 3.133380889892578,
      "step": 3122
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 3122,
      "training_step_time": 9.900131940841675
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 3.42891263961792,
      "step": 3122
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 3122,
      "training_step_time": 10.193529605865479
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 2.4344043731689453,
      "step": 3122
    },
    {
      "dynamics_loss_avg": 0.06457262961193919,
      "epoch": 1.190948486328125e-05,
      "step": 3122
    },
    {
      "action_loss_avg": 0.034488524729385975,
      "epoch": 1.190948486328125e-05,
      "step": 3122
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 3122,
      "training_step_time": 7.458571910858154
    },
    {
      "epoch": 1.190948486328125e-05,
      "model_forward_time": 3.128108024597168,
      "step": 3122
    },
    {
      "epoch": 1.190948486328125e-05,
      "step": 3122,
      "training_step_time": 10.300526142120361
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 3.146068572998047,
      "step": 3123
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 3123,
      "training_step_time": 9.917420148849487
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 3.1342406272888184,
      "step": 3123
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 3123,
      "training_step_time": 9.898373365402222
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 2.43251895904541,
      "step": 3123
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 3123,
      "training_step_time": 7.461844444274902
    },
    {
      "epoch": 1.1913299560546876e-05,
      "model_forward_time": 3.385871410369873,
      "step": 3123
    },
    {
      "epoch": 1.1913299560546876e-05,
      "step": 3123,
      "training_step_time": 10.249916553497314
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 3.1461639404296875,
      "step": 3124
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 3124,
      "training_step_time": 9.916602373123169
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 3.1342012882232666,
      "step": 3124
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 3124,
      "training_step_time": 9.906137943267822
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 3.1360764503479004,
      "step": 3124
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 3124,
      "training_step_time": 9.909629583358765
    },
    {
      "epoch": 1.19171142578125e-05,
      "model_forward_time": 3.1387994289398193,
      "step": 3124
    },
    {
      "epoch": 1.19171142578125e-05,
      "step": 3124,
      "training_step_time": 10.294202089309692
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 3.4585189819335938,
      "step": 3125
    },
    {
      "dynamics_loss_avg": 0.05855732392519712,
      "epoch": 1.1920928955078125e-05,
      "step": 3125
    },
    {
      "action_loss_avg": 0.03629352655261755,
      "epoch": 1.1920928955078125e-05,
      "step": 3125
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 3125,
      "training_step_time": 10.235729694366455
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 3.140329360961914,
      "step": 3125
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 3125,
      "training_step_time": 9.916926383972168
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 3.1353302001953125,
      "step": 3125
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 3125,
      "training_step_time": 9.911494255065918
    },
    {
      "epoch": 1.1920928955078125e-05,
      "model_forward_time": 3.1444363594055176,
      "step": 3125
    },
    {
      "epoch": 1.1920928955078125e-05,
      "step": 3125,
      "training_step_time": 10.28706407546997
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 3.4120240211486816,
      "step": 3126
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 3126,
      "training_step_time": 10.185232639312744
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 3.137726068496704,
      "step": 3126
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 3126,
      "training_step_time": 9.90695333480835
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 3.128737688064575,
      "step": 3126
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 3126,
      "training_step_time": 9.895829677581787
    },
    {
      "epoch": 1.192474365234375e-05,
      "model_forward_time": 3.136423349380493,
      "step": 3126
    },
    {
      "epoch": 1.192474365234375e-05,
      "step": 3126,
      "training_step_time": 10.272018194198608
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 3.1757471561431885,
      "step": 3127
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 3127,
      "training_step_time": 9.946464538574219
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 3.1349246501922607,
      "step": 3127
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 3127,
      "training_step_time": 10.180670022964478
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 3.1342761516571045,
      "step": 3127
    },
    {
      "dynamics_loss_avg": 0.07162779588252306,
      "epoch": 1.1928558349609374e-05,
      "step": 3127
    },
    {
      "action_loss_avg": 0.046387947537004946,
      "epoch": 1.1928558349609374e-05,
      "step": 3127
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 3127,
      "training_step_time": 9.901586055755615
    },
    {
      "epoch": 1.1928558349609374e-05,
      "model_forward_time": 3.142052412033081,
      "step": 3127
    },
    {
      "epoch": 1.1928558349609374e-05,
      "step": 3127,
      "training_step_time": 10.304450988769531
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 3.151395082473755,
      "step": 3128
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 3128,
      "training_step_time": 9.916603088378906
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 3.141732692718506,
      "step": 3128
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 3128,
      "training_step_time": 9.905573606491089
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 3.441005229949951,
      "step": 3128
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 3128,
      "training_step_time": 10.202935934066772
    },
    {
      "epoch": 1.1932373046875e-05,
      "model_forward_time": 3.1540894508361816,
      "step": 3128
    },
    {
      "epoch": 1.1932373046875e-05,
      "step": 3128,
      "training_step_time": 10.313418626785278
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 2.4514617919921875,
      "step": 3129
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 3129,
      "training_step_time": 7.479798078536987
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 3.1346325874328613,
      "step": 3129
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 3129,
      "training_step_time": 9.89995527267456
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 3.1439666748046875,
      "step": 3129
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 3129,
      "training_step_time": 9.909780502319336
    },
    {
      "epoch": 1.1936187744140625e-05,
      "model_forward_time": 3.1420650482177734,
      "step": 3129
    },
    {
      "epoch": 1.1936187744140625e-05,
      "step": 3129,
      "training_step_time": 10.24794054031372
    },
    {
      "epoch": 1.194000244140625e-05,
      "grad_norm": 0.5352928042411804,
      "learning_rate": 1e-05,
      "loss": 0.1005,
      "step": 3130
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 3.465200901031494,
      "step": 3130
    },
    {
      "dynamics_loss_avg": 0.05380098856985569,
      "epoch": 1.194000244140625e-05,
      "step": 3130
    },
    {
      "action_loss_avg": 0.031266310904175046,
      "epoch": 1.194000244140625e-05,
      "step": 3130
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 3130,
      "training_step_time": 10.25772500038147
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 2.44307017326355,
      "step": 3130
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 3130,
      "training_step_time": 7.469801902770996
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 3.129519462585449,
      "step": 3130
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 3130,
      "training_step_time": 9.8977632522583
    },
    {
      "epoch": 1.194000244140625e-05,
      "model_forward_time": 3.1359519958496094,
      "step": 3130
    },
    {
      "epoch": 1.194000244140625e-05,
      "step": 3130,
      "training_step_time": 10.280741691589355
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 3.467329978942871,
      "step": 3131
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 3131,
      "training_step_time": 10.239014625549316
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 3.134126901626587,
      "step": 3131
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 3131,
      "training_step_time": 9.902385711669922
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 3.147695779800415,
      "step": 3131
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 3131,
      "training_step_time": 9.924355506896973
    },
    {
      "epoch": 1.1943817138671874e-05,
      "model_forward_time": 3.1381454467773438,
      "step": 3131
    },
    {
      "epoch": 1.1943817138671874e-05,
      "step": 3131,
      "training_step_time": 10.02379059791565
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 3.145007848739624,
      "step": 3132
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 3132,
      "training_step_time": 9.916626691818237
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 3.1423099040985107,
      "step": 3132
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 3132,
      "training_step_time": 9.910521030426025
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 3.4033145904541016,
      "step": 3132
    },
    {
      "dynamics_loss_avg": 0.04548653177917004,
      "epoch": 1.19476318359375e-05,
      "step": 3132
    },
    {
      "action_loss_avg": 0.03200397603213787,
      "epoch": 1.19476318359375e-05,
      "step": 3132
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 3132,
      "training_step_time": 10.181939363479614
    },
    {
      "epoch": 1.19476318359375e-05,
      "model_forward_time": 3.1339378356933594,
      "step": 3132
    },
    {
      "epoch": 1.19476318359375e-05,
      "step": 3132,
      "training_step_time": 10.272457599639893
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 3.138092279434204,
      "step": 3133
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 3133,
      "training_step_time": 9.9055335521698
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 2.428340196609497,
      "step": 3133
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 3133,
      "training_step_time": 7.456794023513794
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 3.1373374462127686,
      "step": 3133
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 3133,
      "training_step_time": 10.222328662872314
    },
    {
      "epoch": 1.1951446533203125e-05,
      "model_forward_time": 3.1357972621917725,
      "step": 3133
    },
    {
      "epoch": 1.1951446533203125e-05,
      "step": 3133,
      "training_step_time": 10.255169153213501
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 2.452242612838745,
      "step": 3134
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 3134,
      "training_step_time": 7.480441570281982
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 3.1359519958496094,
      "step": 3134
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 3134,
      "training_step_time": 9.912158489227295
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 2.4307382106781006,
      "step": 3134
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 3134,
      "training_step_time": 7.454498291015625
    },
    {
      "epoch": 1.195526123046875e-05,
      "model_forward_time": 2.4303510189056396,
      "step": 3134
    },
    {
      "epoch": 1.195526123046875e-05,
      "step": 3134,
      "training_step_time": 10.270695209503174
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 3.392138957977295,
      "step": 3135
    },
    {
      "dynamics_loss_avg": 0.06199705023318529,
      "epoch": 1.1959075927734374e-05,
      "step": 3135
    },
    {
      "action_loss_avg": 0.04129836964420974,
      "epoch": 1.1959075927734374e-05,
      "step": 3135
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 3135,
      "training_step_time": 10.15816593170166
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 3.134467840194702,
      "step": 3135
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 3135,
      "training_step_time": 9.902032375335693
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 3.1298537254333496,
      "step": 3135
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 3135,
      "training_step_time": 9.899187088012695
    },
    {
      "epoch": 1.1959075927734374e-05,
      "model_forward_time": 3.1328718662261963,
      "step": 3135
    },
    {
      "epoch": 1.1959075927734374e-05,
      "step": 3135,
      "training_step_time": 10.29323673248291
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 3.137542724609375,
      "step": 3136
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 3136,
      "training_step_time": 9.901804208755493
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 3.399047374725342,
      "step": 3136
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 3136,
      "training_step_time": 10.157440900802612
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 3.1308722496032715,
      "step": 3136
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 3136,
      "training_step_time": 9.893292665481567
    },
    {
      "epoch": 1.1962890625e-05,
      "model_forward_time": 2.4276208877563477,
      "step": 3136
    },
    {
      "epoch": 1.1962890625e-05,
      "step": 3136,
      "training_step_time": 10.271207571029663
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 3.146378517150879,
      "step": 3137
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 3137,
      "training_step_time": 9.911974668502808
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 3.1310954093933105,
      "step": 3137
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 3137,
      "training_step_time": 9.906090497970581
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 3.416431188583374,
      "step": 3137
    },
    {
      "dynamics_loss_avg": 0.08060932829976082,
      "epoch": 1.1966705322265625e-05,
      "step": 3137
    },
    {
      "action_loss_avg": 0.041895521059632304,
      "epoch": 1.1966705322265625e-05,
      "step": 3137
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 3137,
      "training_step_time": 10.18134069442749
    },
    {
      "epoch": 1.1966705322265625e-05,
      "model_forward_time": 3.129349708557129,
      "step": 3137
    },
    {
      "epoch": 1.1966705322265625e-05,
      "step": 3137,
      "training_step_time": 10.291778326034546
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 3.141125202178955,
      "step": 3138
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 3138,
      "training_step_time": 9.91467833518982
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 3.1364142894744873,
      "step": 3138
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 3138,
      "training_step_time": 9.908428192138672
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 3.1357100009918213,
      "step": 3138
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 3138,
      "training_step_time": 9.913987398147583
    },
    {
      "epoch": 1.197052001953125e-05,
      "model_forward_time": 3.444453001022339,
      "step": 3138
    },
    {
      "epoch": 1.197052001953125e-05,
      "step": 3138,
      "training_step_time": 10.317574977874756
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 3.1430892944335938,
      "step": 3139
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 3139,
      "training_step_time": 9.923943281173706
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 3.1333186626434326,
      "step": 3139
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 3139,
      "training_step_time": 9.911743640899658
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 2.4346120357513428,
      "step": 3139
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 3139,
      "training_step_time": 7.466312646865845
    },
    {
      "epoch": 1.1974334716796874e-05,
      "model_forward_time": 3.131082057952881,
      "step": 3139
    },
    {
      "epoch": 1.1974334716796874e-05,
      "step": 3139,
      "training_step_time": 10.32007908821106
    },
    {
      "epoch": 1.19781494140625e-05,
      "grad_norm": 0.36838459968566895,
      "learning_rate": 1e-05,
      "loss": 0.0999,
      "step": 3140
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 3.143420696258545,
      "step": 3140
    },
    {
      "dynamics_loss_avg": 0.06434601619839668,
      "epoch": 1.19781494140625e-05,
      "step": 3140
    },
    {
      "action_loss_avg": 0.03719216464087367,
      "epoch": 1.19781494140625e-05,
      "step": 3140
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 3140,
      "training_step_time": 9.922820091247559
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 3.130542039871216,
      "step": 3140
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 3140,
      "training_step_time": 9.904431819915771
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 3.133230209350586,
      "step": 3140
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 3140,
      "training_step_time": 9.907567501068115
    },
    {
      "epoch": 1.19781494140625e-05,
      "model_forward_time": 3.133805990219116,
      "step": 3140
    },
    {
      "epoch": 1.19781494140625e-05,
      "step": 3140,
      "training_step_time": 10.003915309906006
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 3.146192789077759,
      "step": 3141
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 3141,
      "training_step_time": 10.191047191619873
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 3.1391265392303467,
      "step": 3141
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 3141,
      "training_step_time": 9.915034532546997
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 3.143904209136963,
      "step": 3141
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 3141,
      "training_step_time": 9.918549060821533
    },
    {
      "epoch": 1.1981964111328125e-05,
      "model_forward_time": 3.136920928955078,
      "step": 3141
    },
    {
      "epoch": 1.1981964111328125e-05,
      "step": 3141,
      "training_step_time": 10.244537830352783
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 3.146049976348877,
      "step": 3142
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 3142,
      "training_step_time": 9.922661542892456
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 3.44093656539917,
      "step": 3142
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 3142,
      "training_step_time": 10.218602657318115
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 2.4370391368865967,
      "step": 3142
    },
    {
      "dynamics_loss_avg": 0.053538781404495236,
      "epoch": 1.198577880859375e-05,
      "step": 3142
    },
    {
      "action_loss_avg": 0.035920847300440074,
      "epoch": 1.198577880859375e-05,
      "step": 3142
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 3142,
      "training_step_time": 7.467006206512451
    },
    {
      "epoch": 1.198577880859375e-05,
      "model_forward_time": 3.1436572074890137,
      "step": 3142
    },
    {
      "epoch": 1.198577880859375e-05,
      "step": 3142,
      "training_step_time": 10.305798292160034
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 3.150287389755249,
      "step": 3143
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 3143,
      "training_step_time": 9.926939487457275
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 3.1419341564178467,
      "step": 3143
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 3143,
      "training_step_time": 9.921128511428833
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 3.136953115463257,
      "step": 3143
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 3143,
      "training_step_time": 10.183649778366089
    },
    {
      "epoch": 1.1989593505859374e-05,
      "model_forward_time": 3.136641502380371,
      "step": 3143
    },
    {
      "epoch": 1.1989593505859374e-05,
      "step": 3143,
      "training_step_time": 10.005004405975342
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 3.146099090576172,
      "step": 3144
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 3144,
      "training_step_time": 9.921983242034912
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 3.1447460651397705,
      "step": 3144
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 3144,
      "training_step_time": 9.916212320327759
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 3.138340950012207,
      "step": 3144
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 3144,
      "training_step_time": 9.908296346664429
    },
    {
      "epoch": 1.1993408203125e-05,
      "model_forward_time": 3.1445648670196533,
      "step": 3144
    },
    {
      "epoch": 1.1993408203125e-05,
      "step": 3144,
      "training_step_time": 10.29448413848877
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 3.1572959423065186,
      "step": 3145
    },
    {
      "dynamics_loss_avg": 0.06955231637693941,
      "epoch": 1.1997222900390625e-05,
      "step": 3145
    },
    {
      "action_loss_avg": 0.03810391454026103,
      "epoch": 1.1997222900390625e-05,
      "step": 3145
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 3145,
      "training_step_time": 9.940061569213867
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 3.1458728313446045,
      "step": 3145
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 3145,
      "training_step_time": 9.915224075317383
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 3.139923334121704,
      "step": 3145
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 3145,
      "training_step_time": 9.921518564224243
    },
    {
      "epoch": 1.1997222900390625e-05,
      "model_forward_time": 2.4409899711608887,
      "step": 3145
    },
    {
      "epoch": 1.1997222900390625e-05,
      "step": 3145,
      "training_step_time": 10.292908668518066
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 3.145379066467285,
      "step": 3146
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 3146,
      "training_step_time": 9.912451267242432
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 3.456008195877075,
      "step": 3146
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 3146,
      "training_step_time": 10.221174478530884
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 3.1367850303649902,
      "step": 3146
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 3146,
      "training_step_time": 9.903690338134766
    },
    {
      "epoch": 1.200103759765625e-05,
      "model_forward_time": 2.436567783355713,
      "step": 3146
    },
    {
      "epoch": 1.200103759765625e-05,
      "step": 3146,
      "training_step_time": 10.291346549987793
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 3.1442174911499023,
      "step": 3147
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 3147,
      "training_step_time": 9.914183855056763
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 3.1373119354248047,
      "step": 3147
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 3147,
      "training_step_time": 10.176490068435669
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 2.4357476234436035,
      "step": 3147
    },
    {
      "dynamics_loss_avg": 0.051228121668100354,
      "epoch": 1.2004852294921874e-05,
      "step": 3147
    },
    {
      "action_loss_avg": 0.029709473671391606,
      "epoch": 1.2004852294921874e-05,
      "step": 3147
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 3147,
      "training_step_time": 7.461972713470459
    },
    {
      "epoch": 1.2004852294921874e-05,
      "model_forward_time": 3.1391334533691406,
      "step": 3147
    },
    {
      "epoch": 1.2004852294921874e-05,
      "step": 3147,
      "training_step_time": 10.32047414779663
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 3.1567962169647217,
      "step": 3148
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 3148,
      "training_step_time": 9.92514181137085
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 3.1389386653900146,
      "step": 3148
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 3148,
      "training_step_time": 9.91303014755249
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 3.4385507106781006,
      "step": 3148
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 3148,
      "training_step_time": 10.206241369247437
    },
    {
      "epoch": 1.20086669921875e-05,
      "model_forward_time": 3.1409173011779785,
      "step": 3148
    },
    {
      "epoch": 1.20086669921875e-05,
      "step": 3148,
      "training_step_time": 10.238608837127686
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 3.142969846725464,
      "step": 3149
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 3149,
      "training_step_time": 9.913416862487793
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 3.1266071796417236,
      "step": 3149
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 3149,
      "training_step_time": 9.889617204666138
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 3.1355323791503906,
      "step": 3149
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 3149,
      "training_step_time": 9.90335750579834
    },
    {
      "epoch": 1.2012481689453125e-05,
      "model_forward_time": 3.1371822357177734,
      "step": 3149
    },
    {
      "epoch": 1.2012481689453125e-05,
      "step": 3149,
      "training_step_time": 10.325041770935059
    },
    {
      "epoch": 1.201629638671875e-05,
      "grad_norm": 0.45554542541503906,
      "learning_rate": 1e-05,
      "loss": 0.0987,
      "step": 3150
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 3.1453254222869873,
      "step": 3150
    },
    {
      "dynamics_loss_avg": 0.06066518537700176,
      "epoch": 1.201629638671875e-05,
      "step": 3150
    },
    {
      "action_loss_avg": 0.032381688989698884,
      "epoch": 1.201629638671875e-05,
      "step": 3150
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 3150,
      "training_step_time": 9.918285608291626
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 2.4267947673797607,
      "step": 3150
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 3150,
      "training_step_time": 7.447076797485352
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 3.128765821456909,
      "step": 3150
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 3150,
      "training_step_time": 9.891870021820068
    },
    {
      "epoch": 1.201629638671875e-05,
      "model_forward_time": 3.1275088787078857,
      "step": 3150
    },
    {
      "epoch": 1.201629638671875e-05,
      "step": 3150,
      "training_step_time": 10.008726596832275
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 3.4458377361297607,
      "step": 3151
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 3151,
      "training_step_time": 10.211841344833374
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 3.1315643787384033,
      "step": 3151
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 3151,
      "training_step_time": 9.892263412475586
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 3.1321001052856445,
      "step": 3151
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 3151,
      "training_step_time": 9.894710540771484
    },
    {
      "epoch": 1.2020111083984374e-05,
      "model_forward_time": 3.1309216022491455,
      "step": 3151
    },
    {
      "epoch": 1.2020111083984374e-05,
      "step": 3151,
      "training_step_time": 10.023799657821655
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 3.145315408706665,
      "step": 3152
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 3152,
      "training_step_time": 9.910480976104736
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 3.3970096111297607,
      "step": 3152
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 3152,
      "training_step_time": 10.160812139511108
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 3.134361505508423,
      "step": 3152
    },
    {
      "dynamics_loss_avg": 0.05340338461101055,
      "epoch": 1.202392578125e-05,
      "step": 3152
    },
    {
      "action_loss_avg": 0.0355042977258563,
      "epoch": 1.202392578125e-05,
      "step": 3152
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 3152,
      "training_step_time": 9.909115314483643
    },
    {
      "epoch": 1.202392578125e-05,
      "model_forward_time": 3.1324753761291504,
      "step": 3152
    },
    {
      "epoch": 1.202392578125e-05,
      "step": 3152,
      "training_step_time": 10.289062261581421
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 3.1325337886810303,
      "step": 3153
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 3153,
      "training_step_time": 9.898439884185791
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 3.1316232681274414,
      "step": 3153
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 3153,
      "training_step_time": 9.893548011779785
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 3.1291255950927734,
      "step": 3153
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 3153,
      "training_step_time": 10.172958850860596
    },
    {
      "epoch": 1.2027740478515625e-05,
      "model_forward_time": 3.1298863887786865,
      "step": 3153
    },
    {
      "epoch": 1.2027740478515625e-05,
      "step": 3153,
      "training_step_time": 10.009732961654663
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 3.1438233852386475,
      "step": 3154
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 3154,
      "training_step_time": 9.92217755317688
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 2.430588483810425,
      "step": 3154
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 3154,
      "training_step_time": 7.4548070430755615
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 3.1273908615112305,
      "step": 3154
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 3154,
      "training_step_time": 9.89551305770874
    },
    {
      "epoch": 1.203155517578125e-05,
      "model_forward_time": 3.1357858180999756,
      "step": 3154
    },
    {
      "epoch": 1.203155517578125e-05,
      "step": 3154,
      "training_step_time": 10.011166095733643
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 3.4302031993865967,
      "step": 3155
    },
    {
      "dynamics_loss_avg": 0.050912576448172334,
      "epoch": 1.2035369873046874e-05,
      "step": 3155
    },
    {
      "action_loss_avg": 0.032610284816473725,
      "epoch": 1.2035369873046874e-05,
      "step": 3155
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 3155,
      "training_step_time": 10.200427532196045
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 3.137871742248535,
      "step": 3155
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 3155,
      "training_step_time": 9.905681371688843
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 3.1349639892578125,
      "step": 3155
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 3155,
      "training_step_time": 9.905158042907715
    },
    {
      "epoch": 1.2035369873046874e-05,
      "model_forward_time": 3.1420655250549316,
      "step": 3155
    },
    {
      "epoch": 1.2035369873046874e-05,
      "step": 3155,
      "training_step_time": 10.327261209487915
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 3.1547040939331055,
      "step": 3156
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 3156,
      "training_step_time": 10.2261323928833
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 3.136535882949829,
      "step": 3156
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 3156,
      "training_step_time": 9.90642762184143
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 3.133453369140625,
      "step": 3156
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 3156,
      "training_step_time": 9.899635314941406
    },
    {
      "epoch": 1.20391845703125e-05,
      "model_forward_time": 3.139525890350342,
      "step": 3156
    },
    {
      "epoch": 1.20391845703125e-05,
      "step": 3156,
      "training_step_time": 10.026744365692139
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 3.154520034790039,
      "step": 3157
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 3157,
      "training_step_time": 9.925034999847412
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 3.417083263397217,
      "step": 3157
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 3157,
      "training_step_time": 10.181010007858276
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 2.433758020401001,
      "step": 3157
    },
    {
      "dynamics_loss_avg": 0.046307641454041,
      "epoch": 1.2042999267578125e-05,
      "step": 3157
    },
    {
      "action_loss_avg": 0.0306675442494452,
      "epoch": 1.2042999267578125e-05,
      "step": 3157
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 3157,
      "training_step_time": 7.459271669387817
    },
    {
      "epoch": 1.2042999267578125e-05,
      "model_forward_time": 3.1357109546661377,
      "step": 3157
    },
    {
      "epoch": 1.2042999267578125e-05,
      "step": 3157,
      "training_step_time": 10.007843017578125
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 3.147064685821533,
      "step": 3158
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 3158,
      "training_step_time": 9.914242029190063
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 3.12973952293396,
      "step": 3158
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 3158,
      "training_step_time": 9.891891717910767
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 2.4336917400360107,
      "step": 3158
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 3158,
      "training_step_time": 7.453033447265625
    },
    {
      "epoch": 1.204681396484375e-05,
      "model_forward_time": 3.401423215866089,
      "step": 3158
    },
    {
      "epoch": 1.204681396484375e-05,
      "step": 3158,
      "training_step_time": 10.283376455307007
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 3.147914409637451,
      "step": 3159
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 3159,
      "training_step_time": 9.915623188018799
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 3.1350555419921875,
      "step": 3159
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 3159,
      "training_step_time": 9.905324935913086
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 3.1341283321380615,
      "step": 3159
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 3159,
      "training_step_time": 9.9011812210083
    },
    {
      "epoch": 1.2050628662109374e-05,
      "model_forward_time": 3.1383657455444336,
      "step": 3159
    },
    {
      "epoch": 1.2050628662109374e-05,
      "step": 3159,
      "training_step_time": 10.287186861038208
    },
    {
      "epoch": 1.2054443359375e-05,
      "grad_norm": 0.6546280980110168,
      "learning_rate": 1e-05,
      "loss": 0.0966,
      "step": 3160
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 3.1510393619537354,
      "step": 3160
    },
    {
      "dynamics_loss_avg": 0.0528950147330761,
      "epoch": 1.2054443359375e-05,
      "step": 3160
    },
    {
      "action_loss_avg": 0.038011529669165614,
      "epoch": 1.2054443359375e-05,
      "step": 3160
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 3160,
      "training_step_time": 9.91841435432434
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 3.1342737674713135,
      "step": 3160
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 3160,
      "training_step_time": 9.903610706329346
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 3.139357805252075,
      "step": 3160
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 3160,
      "training_step_time": 9.90949296951294
    },
    {
      "epoch": 1.2054443359375e-05,
      "model_forward_time": 2.439598560333252,
      "step": 3160
    },
    {
      "epoch": 1.2054443359375e-05,
      "step": 3160,
      "training_step_time": 10.353511810302734
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 3.1468989849090576,
      "step": 3161
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 3161,
      "training_step_time": 9.920817613601685
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 3.4011857509613037,
      "step": 3161
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 3161,
      "training_step_time": 10.172039270401001
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 3.136232376098633,
      "step": 3161
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 3161,
      "training_step_time": 9.905278205871582
    },
    {
      "epoch": 1.2058258056640625e-05,
      "model_forward_time": 3.1349563598632812,
      "step": 3161
    },
    {
      "epoch": 1.2058258056640625e-05,
      "step": 3161,
      "training_step_time": 10.303831577301025
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 2.4477076530456543,
      "step": 3162
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 3162,
      "training_step_time": 7.474658727645874
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 3.136465311050415,
      "step": 3162
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 3162,
      "training_step_time": 10.206092834472656
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 3.1405141353607178,
      "step": 3162
    },
    {
      "dynamics_loss_avg": 0.058268822729587555,
      "epoch": 1.206207275390625e-05,
      "step": 3162
    },
    {
      "action_loss_avg": 0.029322945047169924,
      "epoch": 1.206207275390625e-05,
      "step": 3162
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 3162,
      "training_step_time": 9.914822340011597
    },
    {
      "epoch": 1.206207275390625e-05,
      "model_forward_time": 3.134028673171997,
      "step": 3162
    },
    {
      "epoch": 1.206207275390625e-05,
      "step": 3162,
      "training_step_time": 10.297568082809448
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 3.152167558670044,
      "step": 3163
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 3163,
      "training_step_time": 9.92510461807251
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 3.1384894847869873,
      "step": 3163
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 3163,
      "training_step_time": 9.902368545532227
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 3.1392054557800293,
      "step": 3163
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 3163,
      "training_step_time": 10.173815727233887
    },
    {
      "epoch": 1.2065887451171874e-05,
      "model_forward_time": 3.136103391647339,
      "step": 3163
    },
    {
      "epoch": 1.2065887451171874e-05,
      "step": 3163,
      "training_step_time": 10.318423509597778
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 3.1501576900482178,
      "step": 3164
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 3164,
      "training_step_time": 9.914315223693848
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 3.1349735260009766,
      "step": 3164
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 3164,
      "training_step_time": 9.897343397140503
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 3.134255886077881,
      "step": 3164
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 3164,
      "training_step_time": 9.902238368988037
    },
    {
      "epoch": 1.20697021484375e-05,
      "model_forward_time": 3.39490008354187,
      "step": 3164
    },
    {
      "epoch": 1.20697021484375e-05,
      "step": 3164,
      "training_step_time": 10.255920171737671
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 3.142826795578003,
      "step": 3165
    },
    {
      "dynamics_loss_avg": 0.06413824893534184,
      "epoch": 1.2073516845703125e-05,
      "step": 3165
    },
    {
      "action_loss_avg": 0.03336147326044738,
      "epoch": 1.2073516845703125e-05,
      "step": 3165
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 3165,
      "training_step_time": 9.915503978729248
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 3.1463124752044678,
      "step": 3165
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 3165,
      "training_step_time": 9.915377378463745
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 3.1374101638793945,
      "step": 3165
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 3165,
      "training_step_time": 9.906130313873291
    },
    {
      "epoch": 1.2073516845703125e-05,
      "model_forward_time": 3.141087055206299,
      "step": 3165
    },
    {
      "epoch": 1.2073516845703125e-05,
      "step": 3165,
      "training_step_time": 10.028005361557007
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 2.717576265335083,
      "step": 3166
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 3166,
      "training_step_time": 7.746006011962891
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 3.1334331035614014,
      "step": 3166
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 3166,
      "training_step_time": 9.895582914352417
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 3.129267930984497,
      "step": 3166
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 3166,
      "training_step_time": 9.911275386810303
    },
    {
      "epoch": 1.207733154296875e-05,
      "model_forward_time": 3.147385597229004,
      "step": 3166
    },
    {
      "epoch": 1.207733154296875e-05,
      "step": 3166,
      "training_step_time": 10.285956859588623
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 3.144239902496338,
      "step": 3167
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 3167,
      "training_step_time": 9.915310382843018
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 3.136359691619873,
      "step": 3167
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 3167,
      "training_step_time": 10.179019451141357
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 3.1366794109344482,
      "step": 3167
    },
    {
      "dynamics_loss_avg": 0.07974511459469795,
      "epoch": 1.2081146240234374e-05,
      "step": 3167
    },
    {
      "action_loss_avg": 0.0474860617890954,
      "epoch": 1.2081146240234374e-05,
      "step": 3167
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 3167,
      "training_step_time": 9.903526544570923
    },
    {
      "epoch": 1.2081146240234374e-05,
      "model_forward_time": 3.136845827102661,
      "step": 3167
    },
    {
      "epoch": 1.2081146240234374e-05,
      "step": 3167,
      "training_step_time": 9.997023582458496
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 3.180292844772339,
      "step": 3168
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 3168,
      "training_step_time": 9.95076870918274
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 3.131948709487915,
      "step": 3168
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 3168,
      "training_step_time": 9.897076845169067
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 3.43349552154541,
      "step": 3168
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 3168,
      "training_step_time": 10.19846224784851
    },
    {
      "epoch": 1.20849609375e-05,
      "model_forward_time": 3.1471290588378906,
      "step": 3168
    },
    {
      "epoch": 1.20849609375e-05,
      "step": 3168,
      "training_step_time": 10.280926942825317
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 3.15435791015625,
      "step": 3169
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 3169,
      "training_step_time": 9.924537181854248
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 3.135833501815796,
      "step": 3169
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 3169,
      "training_step_time": 9.906417608261108
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 3.139094352722168,
      "step": 3169
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 3169,
      "training_step_time": 9.908107280731201
    },
    {
      "epoch": 1.2088775634765625e-05,
      "model_forward_time": 3.133624315261841,
      "step": 3169
    },
    {
      "epoch": 1.2088775634765625e-05,
      "step": 3169,
      "training_step_time": 10.263800382614136
    },
    {
      "epoch": 1.209259033203125e-05,
      "grad_norm": 0.790313184261322,
      "learning_rate": 1e-05,
      "loss": 0.102,
      "step": 3170
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 3.1606521606445312,
      "step": 3170
    },
    {
      "dynamics_loss_avg": 0.06330097876489163,
      "epoch": 1.209259033203125e-05,
      "step": 3170
    },
    {
      "action_loss_avg": 0.04778683930635452,
      "epoch": 1.209259033203125e-05,
      "step": 3170
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 3170,
      "training_step_time": 9.933824300765991
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 3.138963222503662,
      "step": 3170
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 3170,
      "training_step_time": 9.907918691635132
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 3.1455576419830322,
      "step": 3170
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 3170,
      "training_step_time": 9.912311792373657
    },
    {
      "epoch": 1.209259033203125e-05,
      "model_forward_time": 3.1406781673431396,
      "step": 3170
    },
    {
      "epoch": 1.209259033203125e-05,
      "step": 3170,
      "training_step_time": 10.011286497116089
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 3.4607694149017334,
      "step": 3171
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 3171,
      "training_step_time": 10.227056980133057
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 3.1410791873931885,
      "step": 3171
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 3171,
      "training_step_time": 9.908597707748413
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 3.1521925926208496,
      "step": 3171
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 3171,
      "training_step_time": 9.924174785614014
    },
    {
      "epoch": 1.2096405029296874e-05,
      "model_forward_time": 3.1487462520599365,
      "step": 3171
    },
    {
      "epoch": 1.2096405029296874e-05,
      "step": 3171,
      "training_step_time": 10.298856735229492
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 3.149959087371826,
      "step": 3172
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 3172,
      "training_step_time": 9.926777362823486
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 3.426403284072876,
      "step": 3172
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 3172,
      "training_step_time": 10.201981782913208
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 3.1497802734375,
      "step": 3172
    },
    {
      "dynamics_loss_avg": 0.06775710340589285,
      "epoch": 1.21002197265625e-05,
      "step": 3172
    },
    {
      "action_loss_avg": 0.035356228798627855,
      "epoch": 1.21002197265625e-05,
      "step": 3172
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 3172,
      "training_step_time": 9.926184177398682
    },
    {
      "epoch": 1.21002197265625e-05,
      "model_forward_time": 3.14267635345459,
      "step": 3172
    },
    {
      "epoch": 1.21002197265625e-05,
      "step": 3172,
      "training_step_time": 10.025464057922363
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 3.1498591899871826,
      "step": 3173
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 3173,
      "training_step_time": 9.91909646987915
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 3.1453399658203125,
      "step": 3173
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 3173,
      "training_step_time": 9.915872812271118
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 2.7235758304595947,
      "step": 3173
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 3173,
      "training_step_time": 7.750767946243286
    },
    {
      "epoch": 1.2104034423828125e-05,
      "model_forward_time": 3.1412434577941895,
      "step": 3173
    },
    {
      "epoch": 1.2104034423828125e-05,
      "step": 3173,
      "training_step_time": 10.298991680145264
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 3.162851095199585,
      "step": 3174
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 3174,
      "training_step_time": 9.936481237411499
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 3.142643690109253,
      "step": 3174
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 3174,
      "training_step_time": 9.914878129959106
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 3.1369752883911133,
      "step": 3174
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 3174,
      "training_step_time": 9.903092861175537
    },
    {
      "epoch": 1.210784912109375e-05,
      "model_forward_time": 3.4039108753204346,
      "step": 3174
    },
    {
      "epoch": 1.210784912109375e-05,
      "step": 3174,
      "training_step_time": 10.26148509979248
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 3.161552906036377,
      "step": 3175
    },
    {
      "dynamics_loss_avg": 0.04516894351691007,
      "epoch": 1.2111663818359374e-05,
      "step": 3175
    },
    {
      "action_loss_avg": 0.027397147845476866,
      "epoch": 1.2111663818359374e-05,
      "step": 3175
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 3175,
      "training_step_time": 9.942481517791748
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 3.1439967155456543,
      "step": 3175
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 3175,
      "training_step_time": 9.906405210494995
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 3.141538143157959,
      "step": 3175
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 3175,
      "training_step_time": 9.908916234970093
    },
    {
      "epoch": 1.2111663818359374e-05,
      "model_forward_time": 3.142942190170288,
      "step": 3175
    },
    {
      "epoch": 1.2111663818359374e-05,
      "step": 3175,
      "training_step_time": 10.278905391693115
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 3.445478916168213,
      "step": 3176
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 3176,
      "training_step_time": 10.22306513786316
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 3.1500022411346436,
      "step": 3176
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 3176,
      "training_step_time": 9.93306279182434
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 3.149052381515503,
      "step": 3176
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 3176,
      "training_step_time": 9.926697015762329
    },
    {
      "epoch": 1.2115478515625e-05,
      "model_forward_time": 3.147388219833374,
      "step": 3176
    },
    {
      "epoch": 1.2115478515625e-05,
      "step": 3176,
      "training_step_time": 10.313074350357056
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 3.165799617767334,
      "step": 3177
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 3177,
      "training_step_time": 9.943139553070068
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 3.4418327808380127,
      "step": 3177
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 3177,
      "training_step_time": 10.21517300605774
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 3.1451985836029053,
      "step": 3177
    },
    {
      "dynamics_loss_avg": 0.05348224900662899,
      "epoch": 1.2119293212890625e-05,
      "step": 3177
    },
    {
      "action_loss_avg": 0.028685102937743068,
      "epoch": 1.2119293212890625e-05,
      "step": 3177
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 3177,
      "training_step_time": 9.921526193618774
    },
    {
      "epoch": 1.2119293212890625e-05,
      "model_forward_time": 3.1515088081359863,
      "step": 3177
    },
    {
      "epoch": 1.2119293212890625e-05,
      "step": 3177,
      "training_step_time": 10.03117561340332
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 3.1555657386779785,
      "step": 3178
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 3178,
      "training_step_time": 9.931232690811157
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 3.1430788040161133,
      "step": 3178
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 3178,
      "training_step_time": 9.918658971786499
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 2.4419894218444824,
      "step": 3178
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 3178,
      "training_step_time": 7.787814140319824
    },
    {
      "epoch": 1.212310791015625e-05,
      "model_forward_time": 3.140627861022949,
      "step": 3178
    },
    {
      "epoch": 1.212310791015625e-05,
      "step": 3178,
      "training_step_time": 10.336418628692627
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 2.4415078163146973,
      "step": 3179
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 3179,
      "training_step_time": 7.468793153762817
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 3.1309103965759277,
      "step": 3179
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 3179,
      "training_step_time": 9.89460015296936
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 3.1354763507843018,
      "step": 3179
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 3179,
      "training_step_time": 9.899947881698608
    },
    {
      "epoch": 1.2126922607421874e-05,
      "model_forward_time": 3.455996036529541,
      "step": 3179
    },
    {
      "epoch": 1.2126922607421874e-05,
      "step": 3179,
      "training_step_time": 10.318211555480957
    },
    {
      "epoch": 1.21307373046875e-05,
      "grad_norm": 0.4883093237876892,
      "learning_rate": 1e-05,
      "loss": 0.0969,
      "step": 3180
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 3.1506876945495605,
      "step": 3180
    },
    {
      "dynamics_loss_avg": 0.05932008866220713,
      "epoch": 1.21307373046875e-05,
      "step": 3180
    },
    {
      "action_loss_avg": 0.035828623175621035,
      "epoch": 1.21307373046875e-05,
      "step": 3180
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 3180,
      "training_step_time": 9.918742179870605
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 3.1340208053588867,
      "step": 3180
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 3180,
      "training_step_time": 9.899675846099854
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 3.1357803344726562,
      "step": 3180
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 3180,
      "training_step_time": 9.905359983444214
    },
    {
      "epoch": 1.21307373046875e-05,
      "model_forward_time": 3.1327643394470215,
      "step": 3180
    },
    {
      "epoch": 1.21307373046875e-05,
      "step": 3180,
      "training_step_time": 10.010448455810547
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 2.4403247833251953,
      "step": 3181
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 3181,
      "training_step_time": 7.462715148925781
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 3.4066858291625977,
      "step": 3181
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 3181,
      "training_step_time": 10.166984796524048
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 2.43108868598938,
      "step": 3181
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 3181,
      "training_step_time": 7.452059507369995
    },
    {
      "epoch": 1.2134552001953125e-05,
      "model_forward_time": 3.1266908645629883,
      "step": 3181
    },
    {
      "epoch": 1.2134552001953125e-05,
      "step": 3181,
      "training_step_time": 10.277629852294922
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 3.1368765830993652,
      "step": 3182
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 3182,
      "training_step_time": 9.89845895767212
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 3.1287174224853516,
      "step": 3182
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 3182,
      "training_step_time": 10.163832664489746
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 3.1283986568450928,
      "step": 3182
    },
    {
      "dynamics_loss_avg": 0.07277056202292442,
      "epoch": 1.213836669921875e-05,
      "step": 3182
    },
    {
      "action_loss_avg": 0.033215641230344775,
      "epoch": 1.213836669921875e-05,
      "step": 3182
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 3182,
      "training_step_time": 9.893692970275879
    },
    {
      "epoch": 1.213836669921875e-05,
      "model_forward_time": 3.1317625045776367,
      "step": 3182
    },
    {
      "epoch": 1.213836669921875e-05,
      "step": 3182,
      "training_step_time": 10.008700609207153
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 3.1473193168640137,
      "step": 3183
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 3183,
      "training_step_time": 9.910110235214233
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 3.1318178176879883,
      "step": 3183
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 3183,
      "training_step_time": 9.895724773406982
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 3.13845157623291,
      "step": 3183
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 3183,
      "training_step_time": 10.167227029800415
    },
    {
      "epoch": 1.2142181396484374e-05,
      "model_forward_time": 3.128213405609131,
      "step": 3183
    },
    {
      "epoch": 1.2142181396484374e-05,
      "step": 3183,
      "training_step_time": 10.00552248954773
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 3.13856840133667,
      "step": 3184
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 3184,
      "training_step_time": 9.898289442062378
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 3.1336452960968018,
      "step": 3184
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 3184,
      "training_step_time": 9.894191980361938
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 3.1292152404785156,
      "step": 3184
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 3184,
      "training_step_time": 9.893361806869507
    },
    {
      "epoch": 1.214599609375e-05,
      "model_forward_time": 3.13460636138916,
      "step": 3184
    },
    {
      "epoch": 1.214599609375e-05,
      "step": 3184,
      "training_step_time": 10.356855154037476
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 3.149733781814575,
      "step": 3185
    },
    {
      "dynamics_loss_avg": 0.05686689680442214,
      "epoch": 1.2149810791015625e-05,
      "step": 3185
    },
    {
      "action_loss_avg": 0.03440364385023713,
      "epoch": 1.2149810791015625e-05,
      "step": 3185
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 3185,
      "training_step_time": 9.924215078353882
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 3.136430263519287,
      "step": 3185
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 3185,
      "training_step_time": 9.906540870666504
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 3.1326775550842285,
      "step": 3185
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 3185,
      "training_step_time": 9.906442403793335
    },
    {
      "epoch": 1.2149810791015625e-05,
      "model_forward_time": 3.140428304672241,
      "step": 3185
    },
    {
      "epoch": 1.2149810791015625e-05,
      "step": 3185,
      "training_step_time": 10.02930212020874
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 3.454590082168579,
      "step": 3186
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 3186,
      "training_step_time": 10.231209516525269
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 3.1452393531799316,
      "step": 3186
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 3186,
      "training_step_time": 9.917343854904175
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 3.13481068611145,
      "step": 3186
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 3186,
      "training_step_time": 9.902955055236816
    },
    {
      "epoch": 1.215362548828125e-05,
      "model_forward_time": 3.1369526386260986,
      "step": 3186
    },
    {
      "epoch": 1.215362548828125e-05,
      "step": 3186,
      "training_step_time": 10.30281662940979
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 3.147810220718384,
      "step": 3187
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 3187,
      "training_step_time": 9.916863203048706
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 3.131553888320923,
      "step": 3187
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 3187,
      "training_step_time": 10.168160915374756
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 3.1305105686187744,
      "step": 3187
    },
    {
      "dynamics_loss_avg": 0.06529982704669238,
      "epoch": 1.2157440185546874e-05,
      "step": 3187
    },
    {
      "action_loss_avg": 0.034546493319794536,
      "epoch": 1.2157440185546874e-05,
      "step": 3187
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 3187,
      "training_step_time": 9.899433612823486
    },
    {
      "epoch": 1.2157440185546874e-05,
      "model_forward_time": 3.134974479675293,
      "step": 3187
    },
    {
      "epoch": 1.2157440185546874e-05,
      "step": 3187,
      "training_step_time": 10.298187494277954
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 3.1431498527526855,
      "step": 3188
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 3188,
      "training_step_time": 9.914611101150513
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 3.1321985721588135,
      "step": 3188
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 3188,
      "training_step_time": 9.900711297988892
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 3.3919804096221924,
      "step": 3188
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 3188,
      "training_step_time": 10.154164552688599
    },
    {
      "epoch": 1.21612548828125e-05,
      "model_forward_time": 3.136373996734619,
      "step": 3188
    },
    {
      "epoch": 1.21612548828125e-05,
      "step": 3188,
      "training_step_time": 10.297920942306519
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 3.1492607593536377,
      "step": 3189
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 3189,
      "training_step_time": 9.917126893997192
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 2.426647663116455,
      "step": 3189
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 3189,
      "training_step_time": 7.453023672103882
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 3.131831645965576,
      "step": 3189
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 3189,
      "training_step_time": 9.895751237869263
    },
    {
      "epoch": 1.2165069580078125e-05,
      "model_forward_time": 3.1320416927337646,
      "step": 3189
    },
    {
      "epoch": 1.2165069580078125e-05,
      "step": 3189,
      "training_step_time": 10.28503131866455
    },
    {
      "epoch": 1.216888427734375e-05,
      "grad_norm": 0.6470289826393127,
      "learning_rate": 1e-05,
      "loss": 0.0983,
      "step": 3190
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 3.4794390201568604,
      "step": 3190
    },
    {
      "dynamics_loss_avg": 0.06062921434640885,
      "epoch": 1.216888427734375e-05,
      "step": 3190
    },
    {
      "action_loss_avg": 0.03998795934021473,
      "epoch": 1.216888427734375e-05,
      "step": 3190
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 3190,
      "training_step_time": 10.251587390899658
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 2.4369819164276123,
      "step": 3190
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 3190,
      "training_step_time": 7.459455966949463
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 3.1448659896850586,
      "step": 3190
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 3190,
      "training_step_time": 9.911559104919434
    },
    {
      "epoch": 1.216888427734375e-05,
      "model_forward_time": 3.1398239135742188,
      "step": 3190
    },
    {
      "epoch": 1.216888427734375e-05,
      "step": 3190,
      "training_step_time": 10.017061233520508
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 3.458714246749878,
      "step": 3191
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 3191,
      "training_step_time": 10.229354858398438
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 3.1376962661743164,
      "step": 3191
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 3191,
      "training_step_time": 9.901580810546875
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 3.139343500137329,
      "step": 3191
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 3191,
      "training_step_time": 9.90130615234375
    },
    {
      "epoch": 1.2172698974609374e-05,
      "model_forward_time": 3.135104179382324,
      "step": 3191
    },
    {
      "epoch": 1.2172698974609374e-05,
      "step": 3191,
      "training_step_time": 10.23542594909668
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 3.1416749954223633,
      "step": 3192
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 3192,
      "training_step_time": 9.902937173843384
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 3.1367430686950684,
      "step": 3192
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 3192,
      "training_step_time": 9.897950887680054
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 3.3994386196136475,
      "step": 3192
    },
    {
      "dynamics_loss_avg": 0.06931165456771851,
      "epoch": 1.2176513671875e-05,
      "step": 3192
    },
    {
      "action_loss_avg": 0.03708460833877325,
      "epoch": 1.2176513671875e-05,
      "step": 3192
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 3192,
      "training_step_time": 10.159104108810425
    },
    {
      "epoch": 1.2176513671875e-05,
      "model_forward_time": 3.1341793537139893,
      "step": 3192
    },
    {
      "epoch": 1.2176513671875e-05,
      "step": 3192,
      "training_step_time": 10.263795375823975
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 3.152735471725464,
      "step": 3193
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 3193,
      "training_step_time": 9.915038108825684
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 3.1378703117370605,
      "step": 3193
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 3193,
      "training_step_time": 9.899120092391968
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 3.418053388595581,
      "step": 3193
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 3193,
      "training_step_time": 10.185457468032837
    },
    {
      "epoch": 1.2180328369140625e-05,
      "model_forward_time": 3.138928174972534,
      "step": 3193
    },
    {
      "epoch": 1.2180328369140625e-05,
      "step": 3193,
      "training_step_time": 10.012892961502075
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 3.1514391899108887,
      "step": 3194
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 3194,
      "training_step_time": 9.920331239700317
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 3.1414999961853027,
      "step": 3194
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 3194,
      "training_step_time": 9.906476497650146
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 2.4398014545440674,
      "step": 3194
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 3194,
      "training_step_time": 7.467965126037598
    },
    {
      "epoch": 1.218414306640625e-05,
      "model_forward_time": 3.4021339416503906,
      "step": 3194
    },
    {
      "epoch": 1.218414306640625e-05,
      "step": 3194,
      "training_step_time": 10.317677021026611
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 3.1555421352386475,
      "step": 3195
    },
    {
      "dynamics_loss_avg": 0.06334435818716884,
      "epoch": 1.2187957763671874e-05,
      "step": 3195
    },
    {
      "action_loss_avg": 0.036570321395993234,
      "epoch": 1.2187957763671874e-05,
      "step": 3195
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 3195,
      "training_step_time": 9.923298835754395
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 3.133826732635498,
      "step": 3195
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 3195,
      "training_step_time": 9.894895076751709
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 3.1355068683624268,
      "step": 3195
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 3195,
      "training_step_time": 9.902438402175903
    },
    {
      "epoch": 1.2187957763671874e-05,
      "model_forward_time": 2.4408364295959473,
      "step": 3195
    },
    {
      "epoch": 1.2187957763671874e-05,
      "step": 3195,
      "training_step_time": 10.279809713363647
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 3.1595866680145264,
      "step": 3196
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 3196,
      "training_step_time": 10.202484130859375
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 3.1431334018707275,
      "step": 3196
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 3196,
      "training_step_time": 9.915491580963135
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 3.143698215484619,
      "step": 3196
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 3196,
      "training_step_time": 9.91327714920044
    },
    {
      "epoch": 1.21917724609375e-05,
      "model_forward_time": 3.1456046104431152,
      "step": 3196
    },
    {
      "epoch": 1.21917724609375e-05,
      "step": 3196,
      "training_step_time": 10.268094301223755
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 3.157482385635376,
      "step": 3197
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 3197,
      "training_step_time": 9.931013345718384
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 3.4221856594085693,
      "step": 3197
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 3197,
      "training_step_time": 10.188279390335083
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 2.4411704540252686,
      "step": 3197
    },
    {
      "dynamics_loss_avg": 0.0558864576742053,
      "epoch": 1.2195587158203125e-05,
      "step": 3197
    },
    {
      "action_loss_avg": 0.036952432617545126,
      "epoch": 1.2195587158203125e-05,
      "step": 3197
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 3197,
      "training_step_time": 7.4705970287323
    },
    {
      "epoch": 1.2195587158203125e-05,
      "model_forward_time": 3.137718677520752,
      "step": 3197
    },
    {
      "epoch": 1.2195587158203125e-05,
      "step": 3197,
      "training_step_time": 10.01122760772705
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 3.150198221206665,
      "step": 3198
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 3198,
      "training_step_time": 9.920066356658936
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 3.1427152156829834,
      "step": 3198
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 3198,
      "training_step_time": 9.90721344947815
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 2.4426488876342773,
      "step": 3198
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 3198,
      "training_step_time": 7.731688976287842
    },
    {
      "epoch": 1.219940185546875e-05,
      "model_forward_time": 3.1461243629455566,
      "step": 3198
    },
    {
      "epoch": 1.219940185546875e-05,
      "step": 3198,
      "training_step_time": 10.010781288146973
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 3.155520439147949,
      "step": 3199
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 3199,
      "training_step_time": 9.929795742034912
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 3.1420867443084717,
      "step": 3199
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 3199,
      "training_step_time": 9.907938480377197
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 3.1417617797851562,
      "step": 3199
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 3199,
      "training_step_time": 9.904839038848877
    },
    {
      "epoch": 1.2203216552734374e-05,
      "model_forward_time": 3.4404585361480713,
      "step": 3199
    },
    {
      "epoch": 1.2203216552734374e-05,
      "step": 3199,
      "training_step_time": 10.298949241638184
    },
    {
      "epoch": 1.220703125e-05,
      "grad_norm": 0.4629131555557251,
      "learning_rate": 1e-05,
      "loss": 0.0939,
      "step": 3200
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 2.452504873275757,
      "step": 3200
    },
    {
      "dynamics_loss_avg": 0.057397092692554,
      "epoch": 1.220703125e-05,
      "step": 3200
    },
    {
      "action_loss_avg": 0.03501464538276196,
      "epoch": 1.220703125e-05,
      "step": 3200
    },
    {
      "epoch": 1.220703125e-05,
      "step": 3200,
      "training_step_time": 7.475881814956665
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 3.1384782791137695,
      "step": 3200
    },
    {
      "epoch": 1.220703125e-05,
      "step": 3200,
      "training_step_time": 9.906128168106079
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 2.4400405883789062,
      "step": 3200
    },
    {
      "epoch": 1.220703125e-05,
      "step": 3200,
      "training_step_time": 7.4626383781433105
    },
    {
      "epoch": 1.220703125e-05,
      "model_forward_time": 3.13797926902771,
      "step": 3200
    },
    {
      "epoch": 1.220703125e-05,
      "step": 3200,
      "training_step_time": 10.326907634735107
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 3.148526430130005,
      "step": 3201
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 3201,
      "training_step_time": 9.929954767227173
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 3.4214696884155273,
      "step": 3201
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 3201,
      "training_step_time": 10.191213846206665
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 3.1466548442840576,
      "step": 3201
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 3201,
      "training_step_time": 9.922370672225952
    },
    {
      "epoch": 1.2210845947265625e-05,
      "model_forward_time": 3.1472439765930176,
      "step": 3201
    },
    {
      "epoch": 1.2210845947265625e-05,
      "step": 3201,
      "training_step_time": 10.324501752853394
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 3.1530795097351074,
      "step": 3202
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 3202,
      "training_step_time": 9.925591230392456
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 3.4069769382476807,
      "step": 3202
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 3202,
      "training_step_time": 10.170963287353516
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 3.144097089767456,
      "step": 3202
    },
    {
      "dynamics_loss_avg": 0.051127574406564236,
      "epoch": 1.221466064453125e-05,
      "step": 3202
    },
    {
      "action_loss_avg": 0.028254247410222887,
      "epoch": 1.221466064453125e-05,
      "step": 3202
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 3202,
      "training_step_time": 9.909592866897583
    },
    {
      "epoch": 1.221466064453125e-05,
      "model_forward_time": 3.138091802597046,
      "step": 3202
    },
    {
      "epoch": 1.221466064453125e-05,
      "step": 3202,
      "training_step_time": 10.013803243637085
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 3.140709400177002,
      "step": 3203
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 3203,
      "training_step_time": 9.909482479095459
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 3.142303228378296,
      "step": 3203
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 3203,
      "training_step_time": 9.90906572341919
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 3.1352217197418213,
      "step": 3203
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 3203,
      "training_step_time": 10.160953760147095
    },
    {
      "epoch": 1.2218475341796874e-05,
      "model_forward_time": 3.137631893157959,
      "step": 3203
    },
    {
      "epoch": 1.2218475341796874e-05,
      "step": 3203,
      "training_step_time": 10.29586410522461
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 2.4471592903137207,
      "step": 3204
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 3204,
      "training_step_time": 7.471350908279419
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 3.132547616958618,
      "step": 3204
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 3204,
      "training_step_time": 9.894853830337524
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 2.430122137069702,
      "step": 3204
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 3204,
      "training_step_time": 7.450852870941162
    },
    {
      "epoch": 1.22222900390625e-05,
      "model_forward_time": 3.1331870555877686,
      "step": 3204
    },
    {
      "epoch": 1.22222900390625e-05,
      "step": 3204,
      "training_step_time": 10.249256610870361
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 3.1450138092041016,
      "step": 3205
    },
    {
      "dynamics_loss_avg": 0.06862642839550973,
      "epoch": 1.2226104736328125e-05,
      "step": 3205
    },
    {
      "action_loss_avg": 0.043136192671954635,
      "epoch": 1.2226104736328125e-05,
      "step": 3205
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 3205,
      "training_step_time": 9.907169342041016
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 3.1297526359558105,
      "step": 3205
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 3205,
      "training_step_time": 9.893288135528564
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 3.1293840408325195,
      "step": 3205
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 3205,
      "training_step_time": 9.89300537109375
    },
    {
      "epoch": 1.2226104736328125e-05,
      "model_forward_time": 3.1332244873046875,
      "step": 3205
    },
    {
      "epoch": 1.2226104736328125e-05,
      "step": 3205,
      "training_step_time": 10.025511264801025
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 3.3979902267456055,
      "step": 3206
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 3206,
      "training_step_time": 10.171769380569458
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 3.1350553035736084,
      "step": 3206
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 3206,
      "training_step_time": 9.90257716178894
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 3.1325337886810303,
      "step": 3206
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 3206,
      "training_step_time": 9.8983793258667
    },
    {
      "epoch": 1.222991943359375e-05,
      "model_forward_time": 3.136106014251709,
      "step": 3206
    },
    {
      "epoch": 1.222991943359375e-05,
      "step": 3206,
      "training_step_time": 10.025249242782593
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 3.16226863861084,
      "step": 3207
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 3207,
      "training_step_time": 9.945082664489746
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 2.4347381591796875,
      "step": 3207
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 3207,
      "training_step_time": 7.460510015487671
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 2.699392318725586,
      "step": 3207
    },
    {
      "dynamics_loss_avg": 0.06153781227767467,
      "epoch": 1.2233734130859374e-05,
      "step": 3207
    },
    {
      "action_loss_avg": 0.031452803127467635,
      "epoch": 1.2233734130859374e-05,
      "step": 3207
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 3207,
      "training_step_time": 7.7226481437683105
    },
    {
      "epoch": 1.2233734130859374e-05,
      "model_forward_time": 3.1348626613616943,
      "step": 3207
    },
    {
      "epoch": 1.2233734130859374e-05,
      "step": 3207,
      "training_step_time": 10.02098560333252
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 3.1482603549957275,
      "step": 3208
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 3208,
      "training_step_time": 9.913007974624634
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 2.434015989303589,
      "step": 3208
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 3208,
      "training_step_time": 7.451709270477295
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 3.131828546524048,
      "step": 3208
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 3208,
      "training_step_time": 9.889298677444458
    },
    {
      "epoch": 1.2237548828125e-05,
      "model_forward_time": 2.6988282203674316,
      "step": 3208
    },
    {
      "epoch": 1.2237548828125e-05,
      "step": 3208,
      "training_step_time": 10.300427198410034
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 3.1505088806152344,
      "step": 3209
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 3209,
      "training_step_time": 9.913128137588501
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 3.1327319145202637,
      "step": 3209
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 3209,
      "training_step_time": 9.896432876586914
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 3.1373980045318604,
      "step": 3209
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 3209,
      "training_step_time": 9.899142026901245
    },
    {
      "epoch": 1.2241363525390625e-05,
      "model_forward_time": 3.134521722793579,
      "step": 3209
    },
    {
      "epoch": 1.2241363525390625e-05,
      "step": 3209,
      "training_step_time": 10.337876796722412
    },
    {
      "epoch": 1.224517822265625e-05,
      "grad_norm": 0.5945848822593689,
      "learning_rate": 1e-05,
      "loss": 0.104,
      "step": 3210
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 3.449697494506836,
      "step": 3210
    },
    {
      "dynamics_loss_avg": 0.049646342638880014,
      "epoch": 1.224517822265625e-05,
      "step": 3210
    },
    {
      "action_loss_avg": 0.02315306819509715,
      "epoch": 1.224517822265625e-05,
      "step": 3210
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 3210,
      "training_step_time": 10.219305038452148
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 3.1374740600585938,
      "step": 3210
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 3210,
      "training_step_time": 9.904377460479736
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 3.137082576751709,
      "step": 3210
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 3210,
      "training_step_time": 9.89991283416748
    },
    {
      "epoch": 1.224517822265625e-05,
      "model_forward_time": 3.132509469985962,
      "step": 3210
    },
    {
      "epoch": 1.224517822265625e-05,
      "step": 3210,
      "training_step_time": 10.295688390731812
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 3.4351723194122314,
      "step": 3211
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 3211,
      "training_step_time": 10.203957080841064
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 3.139986991882324,
      "step": 3211
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 3211,
      "training_step_time": 9.904544830322266
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 3.137791872024536,
      "step": 3211
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 3211,
      "training_step_time": 9.913431882858276
    },
    {
      "epoch": 1.2248992919921874e-05,
      "model_forward_time": 3.141432523727417,
      "step": 3211
    },
    {
      "epoch": 1.2248992919921874e-05,
      "step": 3211,
      "training_step_time": 10.023450374603271
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 3.1568074226379395,
      "step": 3212
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 3212,
      "training_step_time": 9.927242517471313
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 3.4138598442077637,
      "step": 3212
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 3212,
      "training_step_time": 10.182691812515259
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 3.1468148231506348,
      "step": 3212
    },
    {
      "dynamics_loss_avg": 0.056223719753324984,
      "epoch": 1.22528076171875e-05,
      "step": 3212
    },
    {
      "action_loss_avg": 0.04393209009431302,
      "epoch": 1.22528076171875e-05,
      "step": 3212
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 3212,
      "training_step_time": 9.920391321182251
    },
    {
      "epoch": 1.22528076171875e-05,
      "model_forward_time": 3.142967700958252,
      "step": 3212
    },
    {
      "epoch": 1.22528076171875e-05,
      "step": 3212,
      "training_step_time": 10.248457431793213
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 3.1615777015686035,
      "step": 3213
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 3213,
      "training_step_time": 9.934667587280273
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 3.1461033821105957,
      "step": 3213
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 3213,
      "training_step_time": 9.914478063583374
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 3.1422064304351807,
      "step": 3213
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 3213,
      "training_step_time": 10.211223602294922
    },
    {
      "epoch": 1.2256622314453125e-05,
      "model_forward_time": 3.146085262298584,
      "step": 3213
    },
    {
      "epoch": 1.2256622314453125e-05,
      "step": 3213,
      "training_step_time": 10.271942377090454
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 3.155099868774414,
      "step": 3214
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 3214,
      "training_step_time": 9.929799318313599
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 3.1406474113464355,
      "step": 3214
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 3214,
      "training_step_time": 9.913121223449707
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 3.142092227935791,
      "step": 3214
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 3214,
      "training_step_time": 9.915756940841675
    },
    {
      "epoch": 1.226043701171875e-05,
      "model_forward_time": 3.136932611465454,
      "step": 3214
    },
    {
      "epoch": 1.226043701171875e-05,
      "step": 3214,
      "training_step_time": 10.299517631530762
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 3.156597375869751,
      "step": 3215
    },
    {
      "dynamics_loss_avg": 0.06243144311010838,
      "epoch": 1.2264251708984374e-05,
      "step": 3215
    },
    {
      "action_loss_avg": 0.034093362651765345,
      "epoch": 1.2264251708984374e-05,
      "step": 3215
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 3215,
      "training_step_time": 9.935453414916992
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 3.1365203857421875,
      "step": 3215
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 3215,
      "training_step_time": 9.902762651443481
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 2.424034595489502,
      "step": 3215
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 3215,
      "training_step_time": 7.442615032196045
    },
    {
      "epoch": 1.2264251708984374e-05,
      "model_forward_time": 3.1309258937835693,
      "step": 3215
    },
    {
      "epoch": 1.2264251708984374e-05,
      "step": 3215,
      "training_step_time": 10.24195647239685
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 3.143047571182251,
      "step": 3216
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 3216,
      "training_step_time": 9.906593799591064
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 2.6711761951446533,
      "step": 3216
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 3216,
      "training_step_time": 7.689328193664551
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 3.1268343925476074,
      "step": 3216
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 3216,
      "training_step_time": 9.8854238986969
    },
    {
      "epoch": 1.226806640625e-05,
      "model_forward_time": 3.1311845779418945,
      "step": 3216
    },
    {
      "epoch": 1.226806640625e-05,
      "step": 3216,
      "training_step_time": 10.316413640975952
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 3.137760877609253,
      "step": 3217
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 3217,
      "training_step_time": 9.897905826568604
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 3.1278433799743652,
      "step": 3217
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 3217,
      "training_step_time": 10.15309453010559
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 3.12630558013916,
      "step": 3217
    },
    {
      "dynamics_loss_avg": 0.06431904984638095,
      "epoch": 1.2271881103515625e-05,
      "step": 3217
    },
    {
      "action_loss_avg": 0.03617225945927203,
      "epoch": 1.2271881103515625e-05,
      "step": 3217
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 3217,
      "training_step_time": 9.8870108127594
    },
    {
      "epoch": 1.2271881103515625e-05,
      "model_forward_time": 3.1328675746917725,
      "step": 3217
    },
    {
      "epoch": 1.2271881103515625e-05,
      "step": 3217,
      "training_step_time": 10.005170345306396
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 3.1366348266601562,
      "step": 3218
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 3218,
      "training_step_time": 9.899568796157837
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 3.1283366680145264,
      "step": 3218
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 3218,
      "training_step_time": 9.889320850372314
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 3.131281614303589,
      "step": 3218
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 3218,
      "training_step_time": 10.172957181930542
    },
    {
      "epoch": 1.227569580078125e-05,
      "model_forward_time": 3.136378049850464,
      "step": 3218
    },
    {
      "epoch": 1.227569580078125e-05,
      "step": 3218,
      "training_step_time": 10.325832605361938
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 3.152631998062134,
      "step": 3219
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 3219,
      "training_step_time": 9.92299771308899
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 3.1296873092651367,
      "step": 3219
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 3219,
      "training_step_time": 9.8943190574646
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 3.134894609451294,
      "step": 3219
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 3219,
      "training_step_time": 9.900665760040283
    },
    {
      "epoch": 1.2279510498046874e-05,
      "model_forward_time": 2.713627815246582,
      "step": 3219
    },
    {
      "epoch": 1.2279510498046874e-05,
      "step": 3219,
      "training_step_time": 10.258813381195068
    },
    {
      "epoch": 1.22833251953125e-05,
      "grad_norm": 0.6799596548080444,
      "learning_rate": 1e-05,
      "loss": 0.0953,
      "step": 3220
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 3.143817901611328,
      "step": 3220
    },
    {
      "dynamics_loss_avg": 0.06227108482271433,
      "epoch": 1.22833251953125e-05,
      "step": 3220
    },
    {
      "action_loss_avg": 0.03426613132469356,
      "epoch": 1.22833251953125e-05,
      "step": 3220
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 3220,
      "training_step_time": 9.916367769241333
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 3.1302127838134766,
      "step": 3220
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 3220,
      "training_step_time": 9.892816305160522
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 3.132037878036499,
      "step": 3220
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 3220,
      "training_step_time": 9.893089294433594
    },
    {
      "epoch": 1.22833251953125e-05,
      "model_forward_time": 3.1263394355773926,
      "step": 3220
    },
    {
      "epoch": 1.22833251953125e-05,
      "step": 3220,
      "training_step_time": 10.290616512298584
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 3.137946844100952,
      "step": 3221
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 3221,
      "training_step_time": 10.163901329040527
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 3.1262550354003906,
      "step": 3221
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 3221,
      "training_step_time": 9.88701868057251
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 3.1346030235290527,
      "step": 3221
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 3221,
      "training_step_time": 9.899622678756714
    },
    {
      "epoch": 1.2287139892578125e-05,
      "model_forward_time": 3.130305051803589,
      "step": 3221
    },
    {
      "epoch": 1.2287139892578125e-05,
      "step": 3221,
      "training_step_time": 10.276286840438843
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 3.1375412940979004,
      "step": 3222
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 3222,
      "training_step_time": 9.90669322013855
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 3.3849904537200928,
      "step": 3222
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 3222,
      "training_step_time": 10.15296983718872
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 3.1380889415740967,
      "step": 3222
    },
    {
      "dynamics_loss_avg": 0.05676374509930611,
      "epoch": 1.229095458984375e-05,
      "step": 3222
    },
    {
      "action_loss_avg": 0.0314744415692985,
      "epoch": 1.229095458984375e-05,
      "step": 3222
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 3222,
      "training_step_time": 9.910829544067383
    },
    {
      "epoch": 1.229095458984375e-05,
      "model_forward_time": 3.1332240104675293,
      "step": 3222
    },
    {
      "epoch": 1.229095458984375e-05,
      "step": 3222,
      "training_step_time": 10.02724814414978
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 3.1528689861297607,
      "step": 3223
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 3223,
      "training_step_time": 9.921591520309448
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 2.4326860904693604,
      "step": 3223
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 3223,
      "training_step_time": 7.455643177032471
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 3.131152868270874,
      "step": 3223
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 3223,
      "training_step_time": 10.161791324615479
    },
    {
      "epoch": 1.2294769287109374e-05,
      "model_forward_time": 3.1345064640045166,
      "step": 3223
    },
    {
      "epoch": 1.2294769287109374e-05,
      "step": 3223,
      "training_step_time": 10.262352466583252
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 3.143122673034668,
      "step": 3224
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 3224,
      "training_step_time": 9.90725064277649
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 3.138753652572632,
      "step": 3224
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 3224,
      "training_step_time": 9.90608525276184
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 3.136657476425171,
      "step": 3224
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 3224,
      "training_step_time": 9.90271258354187
    },
    {
      "epoch": 1.2298583984375e-05,
      "model_forward_time": 3.136662244796753,
      "step": 3224
    },
    {
      "epoch": 1.2298583984375e-05,
      "step": 3224,
      "training_step_time": 10.27007794380188
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 3.1563339233398438,
      "step": 3225
    },
    {
      "dynamics_loss_avg": 0.08216034099459649,
      "epoch": 1.2302398681640625e-05,
      "step": 3225
    },
    {
      "action_loss_avg": 0.05133786424994469,
      "epoch": 1.2302398681640625e-05,
      "step": 3225
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 3225,
      "training_step_time": 9.925697326660156
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 3.137241840362549,
      "step": 3225
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 3225,
      "training_step_time": 9.901630163192749
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 2.4345905780792236,
      "step": 3225
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 3225,
      "training_step_time": 7.458575963973999
    },
    {
      "epoch": 1.2302398681640625e-05,
      "model_forward_time": 3.1413955688476562,
      "step": 3225
    },
    {
      "epoch": 1.2302398681640625e-05,
      "step": 3225,
      "training_step_time": 10.339748620986938
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 2.444288730621338,
      "step": 3226
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 3226,
      "training_step_time": 7.479596138000488
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 3.4117116928100586,
      "step": 3226
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 3226,
      "training_step_time": 10.178064346313477
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 3.1328465938568115,
      "step": 3226
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 3226,
      "training_step_time": 9.900404453277588
    },
    {
      "epoch": 1.230621337890625e-05,
      "model_forward_time": 3.135777711868286,
      "step": 3226
    },
    {
      "epoch": 1.230621337890625e-05,
      "step": 3226,
      "training_step_time": 10.249546766281128
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 3.154942750930786,
      "step": 3227
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 3227,
      "training_step_time": 9.933028221130371
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 3.1289048194885254,
      "step": 3227
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 3227,
      "training_step_time": 10.178078174591064
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 3.133072853088379,
      "step": 3227
    },
    {
      "dynamics_loss_avg": 0.07779541499912739,
      "epoch": 1.2310028076171874e-05,
      "step": 3227
    },
    {
      "action_loss_avg": 0.05093330852687359,
      "epoch": 1.2310028076171874e-05,
      "step": 3227
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 3227,
      "training_step_time": 9.905874252319336
    },
    {
      "epoch": 1.2310028076171874e-05,
      "model_forward_time": 3.1349294185638428,
      "step": 3227
    },
    {
      "epoch": 1.2310028076171874e-05,
      "step": 3227,
      "training_step_time": 10.037070274353027
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 3.145641326904297,
      "step": 3228
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 3228,
      "training_step_time": 9.917067527770996
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 3.1331980228424072,
      "step": 3228
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 3228,
      "training_step_time": 9.903890132904053
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 3.4028847217559814,
      "step": 3228
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 3228,
      "training_step_time": 10.175898313522339
    },
    {
      "epoch": 1.23138427734375e-05,
      "model_forward_time": 3.1367173194885254,
      "step": 3228
    },
    {
      "epoch": 1.23138427734375e-05,
      "step": 3228,
      "training_step_time": 10.285099983215332
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 3.1721692085266113,
      "step": 3229
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 3229,
      "training_step_time": 9.946647644042969
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 3.132930278778076,
      "step": 3229
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 3229,
      "training_step_time": 9.90550446510315
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 3.137514591217041,
      "step": 3229
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 3229,
      "training_step_time": 9.909477949142456
    },
    {
      "epoch": 1.2317657470703125e-05,
      "model_forward_time": 2.4317123889923096,
      "step": 3229
    },
    {
      "epoch": 1.2317657470703125e-05,
      "step": 3229,
      "training_step_time": 10.04737401008606
    },
    {
      "epoch": 1.232147216796875e-05,
      "grad_norm": 0.43165743350982666,
      "learning_rate": 1e-05,
      "loss": 0.1001,
      "step": 3230
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 3.153208017349243,
      "step": 3230
    },
    {
      "dynamics_loss_avg": 0.061517555825412275,
      "epoch": 1.232147216796875e-05,
      "step": 3230
    },
    {
      "action_loss_avg": 0.03547528898343444,
      "epoch": 1.232147216796875e-05,
      "step": 3230
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 3230,
      "training_step_time": 9.931946754455566
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 3.129544973373413,
      "step": 3230
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 3230,
      "training_step_time": 9.899425745010376
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 3.131439208984375,
      "step": 3230
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 3230,
      "training_step_time": 9.899794578552246
    },
    {
      "epoch": 1.232147216796875e-05,
      "model_forward_time": 3.125657081604004,
      "step": 3230
    },
    {
      "epoch": 1.232147216796875e-05,
      "step": 3230,
      "training_step_time": 10.258670330047607
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 3.466048002243042,
      "step": 3231
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 3231,
      "training_step_time": 10.23646068572998
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 3.131619453430176,
      "step": 3231
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 3231,
      "training_step_time": 9.900622367858887
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 3.1354331970214844,
      "step": 3231
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 3231,
      "training_step_time": 9.902296781539917
    },
    {
      "epoch": 1.2325286865234375e-05,
      "model_forward_time": 3.133115291595459,
      "step": 3231
    },
    {
      "epoch": 1.2325286865234375e-05,
      "step": 3231,
      "training_step_time": 10.268019676208496
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 3.1475913524627686,
      "step": 3232
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 3232,
      "training_step_time": 9.917418956756592
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 3.3896331787109375,
      "step": 3232
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 3232,
      "training_step_time": 10.153729438781738
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 3.1299118995666504,
      "step": 3232
    },
    {
      "dynamics_loss_avg": 0.06750759407877922,
      "epoch": 1.23291015625e-05,
      "step": 3232
    },
    {
      "action_loss_avg": 0.04197732973843813,
      "epoch": 1.23291015625e-05,
      "step": 3232
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 3232,
      "training_step_time": 9.90110731124878
    },
    {
      "epoch": 1.23291015625e-05,
      "model_forward_time": 2.427534580230713,
      "step": 3232
    },
    {
      "epoch": 1.23291015625e-05,
      "step": 3232,
      "training_step_time": 10.277897119522095
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 3.138519287109375,
      "step": 3233
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 3233,
      "training_step_time": 9.913280725479126
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 3.1307408809661865,
      "step": 3233
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 3233,
      "training_step_time": 9.899846315383911
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 3.1279475688934326,
      "step": 3233
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 3233,
      "training_step_time": 10.21323037147522
    },
    {
      "epoch": 1.2332916259765625e-05,
      "model_forward_time": 3.1316704750061035,
      "step": 3233
    },
    {
      "epoch": 1.2332916259765625e-05,
      "step": 3233,
      "training_step_time": 10.024844408035278
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 3.1508941650390625,
      "step": 3234
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 3234,
      "training_step_time": 9.935576915740967
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 3.137230634689331,
      "step": 3234
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 3234,
      "training_step_time": 9.908124208450317
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 3.1324644088745117,
      "step": 3234
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 3234,
      "training_step_time": 9.907306909561157
    },
    {
      "epoch": 1.233673095703125e-05,
      "model_forward_time": 3.134864330291748,
      "step": 3234
    },
    {
      "epoch": 1.233673095703125e-05,
      "step": 3234,
      "training_step_time": 10.278283834457397
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 3.1387743949890137,
      "step": 3235
    },
    {
      "dynamics_loss_avg": 0.06513872565701603,
      "epoch": 1.2340545654296875e-05,
      "step": 3235
    },
    {
      "action_loss_avg": 0.0378154571633786,
      "epoch": 1.2340545654296875e-05,
      "step": 3235
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 3235,
      "training_step_time": 9.91216778755188
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 3.132436990737915,
      "step": 3235
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 3235,
      "training_step_time": 9.899782419204712
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 3.130951166152954,
      "step": 3235
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 3235,
      "training_step_time": 9.898286581039429
    },
    {
      "epoch": 1.2340545654296875e-05,
      "model_forward_time": 3.1321659088134766,
      "step": 3235
    },
    {
      "epoch": 1.2340545654296875e-05,
      "step": 3235,
      "training_step_time": 10.335575580596924
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 3.150686502456665,
      "step": 3236
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 3236,
      "training_step_time": 10.193784713745117
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 3.1415257453918457,
      "step": 3236
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 3236,
      "training_step_time": 9.906368494033813
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 3.1371326446533203,
      "step": 3236
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 3236,
      "training_step_time": 9.899714469909668
    },
    {
      "epoch": 1.23443603515625e-05,
      "model_forward_time": 2.433148145675659,
      "step": 3236
    },
    {
      "epoch": 1.23443603515625e-05,
      "step": 3236,
      "training_step_time": 10.282498121261597
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 3.1524765491485596,
      "step": 3237
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 3237,
      "training_step_time": 9.923964977264404
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 2.434023857116699,
      "step": 3237
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 3237,
      "training_step_time": 7.4545464515686035
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 3.379317045211792,
      "step": 3237
    },
    {
      "dynamics_loss_avg": 0.06767885759472847,
      "epoch": 1.2348175048828125e-05,
      "step": 3237
    },
    {
      "action_loss_avg": 0.042622589878737925,
      "epoch": 1.2348175048828125e-05,
      "step": 3237
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 3237,
      "training_step_time": 10.144572019577026
    },
    {
      "epoch": 1.2348175048828125e-05,
      "model_forward_time": 3.141961097717285,
      "step": 3237
    },
    {
      "epoch": 1.2348175048828125e-05,
      "step": 3237,
      "training_step_time": 10.017657995223999
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 3.148444414138794,
      "step": 3238
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 3238,
      "training_step_time": 9.914631843566895
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 3.1348071098327637,
      "step": 3238
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 3238,
      "training_step_time": 9.904677867889404
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 3.1396899223327637,
      "step": 3238
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 3238,
      "training_step_time": 9.91034483909607
    },
    {
      "epoch": 1.235198974609375e-05,
      "model_forward_time": 2.706944465637207,
      "step": 3238
    },
    {
      "epoch": 1.235198974609375e-05,
      "step": 3238,
      "training_step_time": 10.026249885559082
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 3.152283191680908,
      "step": 3239
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 3239,
      "training_step_time": 9.923323631286621
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 3.137458562850952,
      "step": 3239
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 3239,
      "training_step_time": 9.903762578964233
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 3.1385700702667236,
      "step": 3239
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 3239,
      "training_step_time": 9.906529664993286
    },
    {
      "epoch": 1.2355804443359375e-05,
      "model_forward_time": 3.4700539112091064,
      "step": 3239
    },
    {
      "epoch": 1.2355804443359375e-05,
      "step": 3239,
      "training_step_time": 10.34037470817566
    },
    {
      "epoch": 1.2359619140625e-05,
      "grad_norm": 0.5548939108848572,
      "learning_rate": 1e-05,
      "loss": 0.1005,
      "step": 3240
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 3.154233694076538,
      "step": 3240
    },
    {
      "dynamics_loss_avg": 0.06787765976041556,
      "epoch": 1.2359619140625e-05,
      "step": 3240
    },
    {
      "action_loss_avg": 0.034398345602676274,
      "epoch": 1.2359619140625e-05,
      "step": 3240
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 3240,
      "training_step_time": 9.931822061538696
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 3.140571355819702,
      "step": 3240
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 3240,
      "training_step_time": 9.917375802993774
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 3.139862298965454,
      "step": 3240
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 3240,
      "training_step_time": 9.91599702835083
    },
    {
      "epoch": 1.2359619140625e-05,
      "model_forward_time": 2.437713623046875,
      "step": 3240
    },
    {
      "epoch": 1.2359619140625e-05,
      "step": 3240,
      "training_step_time": 10.304500341415405
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 3.142930746078491,
      "step": 3241
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 3241,
      "training_step_time": 9.927025556564331
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 3.386141538619995,
      "step": 3241
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 3241,
      "training_step_time": 10.157570600509644
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 2.437960624694824,
      "step": 3241
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 3241,
      "training_step_time": 7.464735269546509
    },
    {
      "epoch": 1.2363433837890625e-05,
      "model_forward_time": 3.1386632919311523,
      "step": 3241
    },
    {
      "epoch": 1.2363433837890625e-05,
      "step": 3241,
      "training_step_time": 10.297514200210571
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 2.4458706378936768,
      "step": 3242
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 3242,
      "training_step_time": 7.48158073425293
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 3.1370527744293213,
      "step": 3242
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 3242,
      "training_step_time": 9.91654658317566
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 3.4106032848358154,
      "step": 3242
    },
    {
      "dynamics_loss_avg": 0.06517336703836918,
      "epoch": 1.236724853515625e-05,
      "step": 3242
    },
    {
      "action_loss_avg": 0.03880681060254574,
      "epoch": 1.236724853515625e-05,
      "step": 3242
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 3242,
      "training_step_time": 10.20215892791748
    },
    {
      "epoch": 1.236724853515625e-05,
      "model_forward_time": 3.1306536197662354,
      "step": 3242
    },
    {
      "epoch": 1.236724853515625e-05,
      "step": 3242,
      "training_step_time": 10.222813129425049
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 2.4435105323791504,
      "step": 3243
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 3243,
      "training_step_time": 7.4666218757629395
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 3.1348910331726074,
      "step": 3243
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 3243,
      "training_step_time": 9.89819049835205
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 3.130399227142334,
      "step": 3243
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 3243,
      "training_step_time": 9.899209976196289
    },
    {
      "epoch": 1.2371063232421875e-05,
      "model_forward_time": 3.3747613430023193,
      "step": 3243
    },
    {
      "epoch": 1.2371063232421875e-05,
      "step": 3243,
      "training_step_time": 10.28923487663269
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 2.439934492111206,
      "step": 3244
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 3244,
      "training_step_time": 7.465261936187744
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 3.134059429168701,
      "step": 3244
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 3244,
      "training_step_time": 9.897558689117432
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 3.130964756011963,
      "step": 3244
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 3244,
      "training_step_time": 9.894563674926758
    },
    {
      "epoch": 1.23748779296875e-05,
      "model_forward_time": 3.130854845046997,
      "step": 3244
    },
    {
      "epoch": 1.23748779296875e-05,
      "step": 3244,
      "training_step_time": 10.25172233581543
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 2.4502034187316895,
      "step": 3245
    },
    {
      "dynamics_loss_avg": 0.059961695596575736,
      "epoch": 1.2378692626953125e-05,
      "step": 3245
    },
    {
      "action_loss_avg": 0.03739044414833188,
      "epoch": 1.2378692626953125e-05,
      "step": 3245
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 3245,
      "training_step_time": 7.479897737503052
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 3.1377382278442383,
      "step": 3245
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 3245,
      "training_step_time": 9.908411026000977
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 3.135317087173462,
      "step": 3245
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 3245,
      "training_step_time": 9.904380321502686
    },
    {
      "epoch": 1.2378692626953125e-05,
      "model_forward_time": 2.429009437561035,
      "step": 3245
    },
    {
      "epoch": 1.2378692626953125e-05,
      "step": 3245,
      "training_step_time": 10.300022840499878
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 3.148341178894043,
      "step": 3246
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 3246,
      "training_step_time": 9.920983076095581
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 3.4381215572357178,
      "step": 3246
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 3246,
      "training_step_time": 10.207465410232544
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 3.1358907222747803,
      "step": 3246
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 3246,
      "training_step_time": 9.90590786933899
    },
    {
      "epoch": 1.238250732421875e-05,
      "model_forward_time": 3.136967658996582,
      "step": 3246
    },
    {
      "epoch": 1.238250732421875e-05,
      "step": 3246,
      "training_step_time": 10.325733661651611
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 3.1432671546936035,
      "step": 3247
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 3247,
      "training_step_time": 9.915717124938965
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 3.130012035369873,
      "step": 3247
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 3247,
      "training_step_time": 10.168980360031128
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 3.132112979888916,
      "step": 3247
    },
    {
      "dynamics_loss_avg": 0.047297014482319356,
      "epoch": 1.2386322021484375e-05,
      "step": 3247
    },
    {
      "action_loss_avg": 0.028016174957156182,
      "epoch": 1.2386322021484375e-05,
      "step": 3247
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 3247,
      "training_step_time": 9.896312713623047
    },
    {
      "epoch": 1.2386322021484375e-05,
      "model_forward_time": 3.1329715251922607,
      "step": 3247
    },
    {
      "epoch": 1.2386322021484375e-05,
      "step": 3247,
      "training_step_time": 10.00962209701538
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 3.150001287460327,
      "step": 3248
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 3248,
      "training_step_time": 9.919056415557861
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 3.131465196609497,
      "step": 3248
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 3248,
      "training_step_time": 9.90118670463562
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 3.3937013149261475,
      "step": 3248
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 3248,
      "training_step_time": 10.159242153167725
    },
    {
      "epoch": 1.239013671875e-05,
      "model_forward_time": 3.135902166366577,
      "step": 3248
    },
    {
      "epoch": 1.239013671875e-05,
      "step": 3248,
      "training_step_time": 10.31549334526062
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 3.198512554168701,
      "step": 3249
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 3249,
      "training_step_time": 9.974660634994507
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 3.135017156600952,
      "step": 3249
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 3249,
      "training_step_time": 9.914442539215088
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 3.1384143829345703,
      "step": 3249
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 3249,
      "training_step_time": 9.913553476333618
    },
    {
      "epoch": 1.2393951416015625e-05,
      "model_forward_time": 3.1351194381713867,
      "step": 3249
    },
    {
      "epoch": 1.2393951416015625e-05,
      "step": 3249,
      "training_step_time": 10.269864797592163
    },
    {
      "epoch": 1.239776611328125e-05,
      "grad_norm": 0.6567586660385132,
      "learning_rate": 1e-05,
      "loss": 0.0934,
      "step": 3250
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 3.151660203933716,
      "step": 3250
    },
    {
      "dynamics_loss_avg": 0.06494546988978982,
      "epoch": 1.239776611328125e-05,
      "step": 3250
    },
    {
      "action_loss_avg": 0.03392072860151529,
      "epoch": 1.239776611328125e-05,
      "step": 3250
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 3250,
      "training_step_time": 9.929219484329224
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 3.130871534347534,
      "step": 3250
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 3250,
      "training_step_time": 9.905830383300781
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 3.136277437210083,
      "step": 3250
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 3250,
      "training_step_time": 9.913213968276978
    },
    {
      "epoch": 1.239776611328125e-05,
      "model_forward_time": 2.4342527389526367,
      "step": 3250
    },
    {
      "epoch": 1.239776611328125e-05,
      "step": 3250,
      "training_step_time": 10.303638935089111
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 3.578324317932129,
      "step": 3251
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 3251,
      "training_step_time": 10.370089530944824
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 3.149552822113037,
      "step": 3251
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 3251,
      "training_step_time": 9.937795400619507
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 2.434555768966675,
      "step": 3251
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 3251,
      "training_step_time": 7.460484504699707
    },
    {
      "epoch": 1.2401580810546875e-05,
      "model_forward_time": 3.1318249702453613,
      "step": 3251
    },
    {
      "epoch": 1.2401580810546875e-05,
      "step": 3251,
      "training_step_time": 10.281324625015259
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 3.1421592235565186,
      "step": 3252
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 3252,
      "training_step_time": 9.92634129524231
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 3.1388025283813477,
      "step": 3252
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 3252,
      "training_step_time": 10.258013725280762
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 3.143292188644409,
      "step": 3252
    },
    {
      "dynamics_loss_avg": 0.07413740046322345,
      "epoch": 1.24053955078125e-05,
      "step": 3252
    },
    {
      "action_loss_avg": 0.033194043952971694,
      "epoch": 1.24053955078125e-05,
      "step": 3252
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 3252,
      "training_step_time": 9.920138597488403
    },
    {
      "epoch": 1.24053955078125e-05,
      "model_forward_time": 3.1366474628448486,
      "step": 3252
    },
    {
      "epoch": 1.24053955078125e-05,
      "step": 3252,
      "training_step_time": 10.302708148956299
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 2.4441940784454346,
      "step": 3253
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 3253,
      "training_step_time": 7.475869178771973
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 3.1332550048828125,
      "step": 3253
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 3253,
      "training_step_time": 9.91539478302002
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 3.139641284942627,
      "step": 3253
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 3253,
      "training_step_time": 10.24449110031128
    },
    {
      "epoch": 1.2409210205078125e-05,
      "model_forward_time": 3.157461643218994,
      "step": 3253
    },
    {
      "epoch": 1.2409210205078125e-05,
      "step": 3253,
      "training_step_time": 10.031370162963867
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 3.1492738723754883,
      "step": 3254
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 3254,
      "training_step_time": 9.936391115188599
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 3.1470248699188232,
      "step": 3254
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 3254,
      "training_step_time": 9.930760383605957
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 3.141505479812622,
      "step": 3254
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 3254,
      "training_step_time": 9.92703652381897
    },
    {
      "epoch": 1.241302490234375e-05,
      "model_forward_time": 3.147005558013916,
      "step": 3254
    },
    {
      "epoch": 1.241302490234375e-05,
      "step": 3254,
      "training_step_time": 10.344711065292358
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 3.1496188640594482,
      "step": 3255
    },
    {
      "dynamics_loss_avg": 0.08224550765007735,
      "epoch": 1.2416839599609375e-05,
      "step": 3255
    },
    {
      "action_loss_avg": 0.03948286371305585,
      "epoch": 1.2416839599609375e-05,
      "step": 3255
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 3255,
      "training_step_time": 9.943412065505981
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 3.1436731815338135,
      "step": 3255
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 3255,
      "training_step_time": 9.923260927200317
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 3.1386144161224365,
      "step": 3255
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 3255,
      "training_step_time": 9.922951221466064
    },
    {
      "epoch": 1.2416839599609375e-05,
      "model_forward_time": 3.14387845993042,
      "step": 3255
    },
    {
      "epoch": 1.2416839599609375e-05,
      "step": 3255,
      "training_step_time": 10.272292375564575
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 3.1508123874664307,
      "step": 3256
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 3256,
      "training_step_time": 10.252589702606201
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 3.139294147491455,
      "step": 3256
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 3256,
      "training_step_time": 9.914746046066284
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 3.136786937713623,
      "step": 3256
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 3256,
      "training_step_time": 9.919169664382935
    },
    {
      "epoch": 1.2420654296875e-05,
      "model_forward_time": 2.4419784545898438,
      "step": 3256
    },
    {
      "epoch": 1.2420654296875e-05,
      "step": 3256,
      "training_step_time": 10.320125341415405
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 3.15496826171875,
      "step": 3257
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 3257,
      "training_step_time": 9.932129621505737
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 3.1398961544036865,
      "step": 3257
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 3257,
      "training_step_time": 10.176944971084595
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 3.1409780979156494,
      "step": 3257
    },
    {
      "dynamics_loss_avg": 0.06287979017943143,
      "epoch": 1.2424468994140625e-05,
      "step": 3257
    },
    {
      "action_loss_avg": 0.030260634236037733,
      "epoch": 1.2424468994140625e-05,
      "step": 3257
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 3257,
      "training_step_time": 9.917965412139893
    },
    {
      "epoch": 1.2424468994140625e-05,
      "model_forward_time": 3.1472246646881104,
      "step": 3257
    },
    {
      "epoch": 1.2424468994140625e-05,
      "step": 3257,
      "training_step_time": 10.26743197441101
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 3.1618497371673584,
      "step": 3258
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 3258,
      "training_step_time": 9.934827327728271
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 3.138944387435913,
      "step": 3258
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 3258,
      "training_step_time": 9.915039777755737
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 3.143397808074951,
      "step": 3258
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 3258,
      "training_step_time": 10.233887195587158
    },
    {
      "epoch": 1.242828369140625e-05,
      "model_forward_time": 3.1475436687469482,
      "step": 3258
    },
    {
      "epoch": 1.242828369140625e-05,
      "step": 3258,
      "training_step_time": 10.011256456375122
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 3.1545376777648926,
      "step": 3259
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 3259,
      "training_step_time": 9.928295373916626
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 3.1447384357452393,
      "step": 3259
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 3259,
      "training_step_time": 9.920026779174805
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 3.148024320602417,
      "step": 3259
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 3259,
      "training_step_time": 9.92034912109375
    },
    {
      "epoch": 1.2432098388671875e-05,
      "model_forward_time": 3.431441068649292,
      "step": 3259
    },
    {
      "epoch": 1.2432098388671875e-05,
      "step": 3259,
      "training_step_time": 10.29909610748291
    },
    {
      "epoch": 1.24359130859375e-05,
      "grad_norm": 0.4832718074321747,
      "learning_rate": 1e-05,
      "loss": 0.1004,
      "step": 3260
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 3.1635594367980957,
      "step": 3260
    },
    {
      "dynamics_loss_avg": 0.07457627169787884,
      "epoch": 1.24359130859375e-05,
      "step": 3260
    },
    {
      "action_loss_avg": 0.03331169355660677,
      "epoch": 1.24359130859375e-05,
      "step": 3260
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 3260,
      "training_step_time": 9.943979740142822
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 3.1475229263305664,
      "step": 3260
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 3260,
      "training_step_time": 9.921658754348755
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 3.145277976989746,
      "step": 3260
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 3260,
      "training_step_time": 9.923182964324951
    },
    {
      "epoch": 1.24359130859375e-05,
      "model_forward_time": 3.153005361557007,
      "step": 3260
    },
    {
      "epoch": 1.24359130859375e-05,
      "step": 3260,
      "training_step_time": 10.429319381713867
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 3.4281389713287354,
      "step": 3261
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 3261,
      "training_step_time": 10.209403991699219
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 2.444021463394165,
      "step": 3261
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 3261,
      "training_step_time": 7.47669529914856
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 3.1477789878845215,
      "step": 3261
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 3261,
      "training_step_time": 9.924092054367065
    },
    {
      "epoch": 1.2439727783203125e-05,
      "model_forward_time": 3.146615743637085,
      "step": 3261
    },
    {
      "epoch": 1.2439727783203125e-05,
      "step": 3261,
      "training_step_time": 10.268764734268188
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 3.156447649002075,
      "step": 3262
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 3262,
      "training_step_time": 9.937957048416138
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 3.1458306312561035,
      "step": 3262
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 3262,
      "training_step_time": 10.235406398773193
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 3.1420280933380127,
      "step": 3262
    },
    {
      "dynamics_loss_avg": 0.07241471875458956,
      "epoch": 1.244354248046875e-05,
      "step": 3262
    },
    {
      "action_loss_avg": 0.0337939431425184,
      "epoch": 1.244354248046875e-05,
      "step": 3262
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 3262,
      "training_step_time": 9.914641857147217
    },
    {
      "epoch": 1.244354248046875e-05,
      "model_forward_time": 3.1485331058502197,
      "step": 3262
    },
    {
      "epoch": 1.244354248046875e-05,
      "step": 3262,
      "training_step_time": 10.281314611434937
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 3.155771017074585,
      "step": 3263
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 3263,
      "training_step_time": 9.928654193878174
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 3.1328275203704834,
      "step": 3263
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 3263,
      "training_step_time": 9.905272722244263
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 3.4006664752960205,
      "step": 3263
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 3263,
      "training_step_time": 10.1782066822052
    },
    {
      "epoch": 1.2447357177734375e-05,
      "model_forward_time": 3.138768196105957,
      "step": 3263
    },
    {
      "epoch": 1.2447357177734375e-05,
      "step": 3263,
      "training_step_time": 10.262008905410767
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 3.1534206867218018,
      "step": 3264
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 3264,
      "training_step_time": 9.933122634887695
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 3.136626720428467,
      "step": 3264
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 3264,
      "training_step_time": 9.913901805877686
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 3.140120267868042,
      "step": 3264
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 3264,
      "training_step_time": 9.915781736373901
    },
    {
      "epoch": 1.2451171875e-05,
      "model_forward_time": 3.138362407684326,
      "step": 3264
    },
    {
      "epoch": 1.2451171875e-05,
      "step": 3264,
      "training_step_time": 10.288026571273804
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 3.155681610107422,
      "step": 3265
    },
    {
      "dynamics_loss_avg": 0.04884478934109211,
      "epoch": 1.2454986572265625e-05,
      "step": 3265
    },
    {
      "action_loss_avg": 0.019391209213063122,
      "epoch": 1.2454986572265625e-05,
      "step": 3265
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 3265,
      "training_step_time": 9.940442562103271
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 3.139714479446411,
      "step": 3265
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 3265,
      "training_step_time": 9.919931411743164
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 3.138942003250122,
      "step": 3265
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 3265,
      "training_step_time": 9.920338869094849
    },
    {
      "epoch": 1.2454986572265625e-05,
      "model_forward_time": 3.137573480606079,
      "step": 3265
    },
    {
      "epoch": 1.2454986572265625e-05,
      "step": 3265,
      "training_step_time": 10.259772062301636
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 3.424170732498169,
      "step": 3266
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 3266,
      "training_step_time": 10.198243856430054
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 3.1332809925079346,
      "step": 3266
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 3266,
      "training_step_time": 9.911779880523682
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 3.1376218795776367,
      "step": 3266
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 3266,
      "training_step_time": 9.911338090896606
    },
    {
      "epoch": 1.245880126953125e-05,
      "model_forward_time": 3.1339588165283203,
      "step": 3266
    },
    {
      "epoch": 1.245880126953125e-05,
      "step": 3266,
      "training_step_time": 10.281696319580078
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 3.1440651416778564,
      "step": 3267
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 3267,
      "training_step_time": 9.920613765716553
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 3.1429555416107178,
      "step": 3267
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 3267,
      "training_step_time": 10.18889832496643
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 3.1411232948303223,
      "step": 3267
    },
    {
      "dynamics_loss_avg": 0.07306025046855211,
      "epoch": 1.2462615966796875e-05,
      "step": 3267
    },
    {
      "action_loss_avg": 0.03733426569961011,
      "epoch": 1.2462615966796875e-05,
      "step": 3267
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 3267,
      "training_step_time": 9.920267581939697
    },
    {
      "epoch": 1.2462615966796875e-05,
      "model_forward_time": 3.1434707641601562,
      "step": 3267
    },
    {
      "epoch": 1.2462615966796875e-05,
      "step": 3267,
      "training_step_time": 10.349103927612305
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 3.149387836456299,
      "step": 3268
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 3268,
      "training_step_time": 9.932884693145752
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 3.1461169719696045,
      "step": 3268
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 3268,
      "training_step_time": 9.925604820251465
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 3.431138753890991,
      "step": 3268
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 3268,
      "training_step_time": 10.215287923812866
    },
    {
      "epoch": 1.24664306640625e-05,
      "model_forward_time": 3.1404547691345215,
      "step": 3268
    },
    {
      "epoch": 1.24664306640625e-05,
      "step": 3268,
      "training_step_time": 10.31065845489502
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 3.1486358642578125,
      "step": 3269
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 3269,
      "training_step_time": 9.93021535873413
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 3.143831968307495,
      "step": 3269
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 3269,
      "training_step_time": 9.922115564346313
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 3.1435906887054443,
      "step": 3269
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 3269,
      "training_step_time": 9.92521858215332
    },
    {
      "epoch": 1.2470245361328125e-05,
      "model_forward_time": 3.4421157836914062,
      "step": 3269
    },
    {
      "epoch": 1.2470245361328125e-05,
      "step": 3269,
      "training_step_time": 10.314160108566284
    },
    {
      "epoch": 1.247406005859375e-05,
      "grad_norm": 0.4016943871974945,
      "learning_rate": 1e-05,
      "loss": 0.0993,
      "step": 3270
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 3.1494903564453125,
      "step": 3270
    },
    {
      "dynamics_loss_avg": 0.07415962629020215,
      "epoch": 1.247406005859375e-05,
      "step": 3270
    },
    {
      "action_loss_avg": 0.03796817939728499,
      "epoch": 1.247406005859375e-05,
      "step": 3270
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 3270,
      "training_step_time": 9.929699420928955
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 3.1447975635528564,
      "step": 3270
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 3270,
      "training_step_time": 9.928821325302124
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 3.14560866355896,
      "step": 3270
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 3270,
      "training_step_time": 9.92874813079834
    },
    {
      "epoch": 1.247406005859375e-05,
      "model_forward_time": 3.1425118446350098,
      "step": 3270
    },
    {
      "epoch": 1.247406005859375e-05,
      "step": 3270,
      "training_step_time": 10.028338432312012
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 3.4156627655029297,
      "step": 3271
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 3271,
      "training_step_time": 10.192472457885742
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 3.1487412452697754,
      "step": 3271
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 3271,
      "training_step_time": 9.928873538970947
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 3.144484281539917,
      "step": 3271
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 3271,
      "training_step_time": 9.921571969985962
    },
    {
      "epoch": 1.2477874755859375e-05,
      "model_forward_time": 3.147135019302368,
      "step": 3271
    },
    {
      "epoch": 1.2477874755859375e-05,
      "step": 3271,
      "training_step_time": 10.257177591323853
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 3.1567649841308594,
      "step": 3272
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 3272,
      "training_step_time": 9.93935251235962
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 3.4166407585144043,
      "step": 3272
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 3272,
      "training_step_time": 10.190966129302979
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 3.133976459503174,
      "step": 3272
    },
    {
      "dynamics_loss_avg": 0.07611274085938931,
      "epoch": 1.2481689453125e-05,
      "step": 3272
    },
    {
      "action_loss_avg": 0.038641276955604556,
      "epoch": 1.2481689453125e-05,
      "step": 3272
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 3272,
      "training_step_time": 9.913491010665894
    },
    {
      "epoch": 1.2481689453125e-05,
      "model_forward_time": 3.1346752643585205,
      "step": 3272
    },
    {
      "epoch": 1.2481689453125e-05,
      "step": 3272,
      "training_step_time": 10.307927131652832
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 3.144273281097412,
      "step": 3273
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 3273,
      "training_step_time": 9.919268369674683
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 3.1416103839874268,
      "step": 3273
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 3273,
      "training_step_time": 9.923516750335693
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 3.146648645401001,
      "step": 3273
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 3273,
      "training_step_time": 10.203824758529663
    },
    {
      "epoch": 1.2485504150390626e-05,
      "model_forward_time": 3.1494617462158203,
      "step": 3273
    },
    {
      "epoch": 1.2485504150390626e-05,
      "step": 3273,
      "training_step_time": 10.294772386550903
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 3.1560966968536377,
      "step": 3274
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 3274,
      "training_step_time": 9.936119079589844
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 3.141124963760376,
      "step": 3274
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 3274,
      "training_step_time": 9.913927793502808
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 3.1401119232177734,
      "step": 3274
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 3274,
      "training_step_time": 9.914311408996582
    },
    {
      "epoch": 1.248931884765625e-05,
      "model_forward_time": 3.4274771213531494,
      "step": 3274
    },
    {
      "epoch": 1.248931884765625e-05,
      "step": 3274,
      "training_step_time": 10.297751903533936
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 3.1403491497039795,
      "step": 3275
    },
    {
      "dynamics_loss_avg": 0.060394308529794216,
      "epoch": 1.2493133544921875e-05,
      "step": 3275
    },
    {
      "action_loss_avg": 0.03419253500178456,
      "epoch": 1.2493133544921875e-05,
      "step": 3275
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 3275,
      "training_step_time": 9.917515277862549
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 3.1301400661468506,
      "step": 3275
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 3275,
      "training_step_time": 9.90548062324524
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 3.1360013484954834,
      "step": 3275
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 3275,
      "training_step_time": 9.906967878341675
    },
    {
      "epoch": 1.2493133544921875e-05,
      "model_forward_time": 3.1322364807128906,
      "step": 3275
    },
    {
      "epoch": 1.2493133544921875e-05,
      "step": 3275,
      "training_step_time": 10.023025274276733
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 3.138472080230713,
      "step": 3276
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 3276,
      "training_step_time": 10.182432651519775
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 3.132831335067749,
      "step": 3276
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 3276,
      "training_step_time": 9.906902074813843
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 3.1318271160125732,
      "step": 3276
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 3276,
      "training_step_time": 9.907208681106567
    },
    {
      "epoch": 1.24969482421875e-05,
      "model_forward_time": 3.132580041885376,
      "step": 3276
    },
    {
      "epoch": 1.24969482421875e-05,
      "step": 3276,
      "training_step_time": 10.0051589012146
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 3.1436710357666016,
      "step": 3277
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 3277,
      "training_step_time": 9.919137477874756
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 3.3882265090942383,
      "step": 3277
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 3277,
      "training_step_time": 10.161875009536743
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 3.1436634063720703,
      "step": 3277
    },
    {
      "dynamics_loss_avg": 0.0537225142121315,
      "epoch": 1.2500762939453126e-05,
      "step": 3277
    },
    {
      "action_loss_avg": 0.02999397541861981,
      "epoch": 1.2500762939453126e-05,
      "step": 3277
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 3277,
      "training_step_time": 9.93005108833313
    },
    {
      "epoch": 1.2500762939453126e-05,
      "model_forward_time": 2.4346728324890137,
      "step": 3277
    },
    {
      "epoch": 1.2500762939453126e-05,
      "step": 3277,
      "training_step_time": 10.30475401878357
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 3.142037868499756,
      "step": 3278
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 3278,
      "training_step_time": 9.919617652893066
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 3.1347339153289795,
      "step": 3278
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 3278,
      "training_step_time": 9.905561923980713
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 3.1342368125915527,
      "step": 3278
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 3278,
      "training_step_time": 10.187350988388062
    },
    {
      "epoch": 1.250457763671875e-05,
      "model_forward_time": 3.1345436573028564,
      "step": 3278
    },
    {
      "epoch": 1.250457763671875e-05,
      "step": 3278,
      "training_step_time": 10.307327270507812
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 2.4492344856262207,
      "step": 3279
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 3279,
      "training_step_time": 7.48967981338501
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 3.1356067657470703,
      "step": 3279
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 3279,
      "training_step_time": 9.906749248504639
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 3.133521556854248,
      "step": 3279
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 3279,
      "training_step_time": 9.904958963394165
    },
    {
      "epoch": 1.2508392333984375e-05,
      "model_forward_time": 3.429718494415283,
      "step": 3279
    },
    {
      "epoch": 1.2508392333984375e-05,
      "step": 3279,
      "training_step_time": 10.324438333511353
    },
    {
      "epoch": 1.251220703125e-05,
      "grad_norm": 0.37627512216567993,
      "learning_rate": 1e-05,
      "loss": 0.0974,
      "step": 3280
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 3.156649351119995,
      "step": 3280
    },
    {
      "dynamics_loss_avg": 0.054231184720993045,
      "epoch": 1.251220703125e-05,
      "step": 3280
    },
    {
      "action_loss_avg": 0.02445357693359256,
      "epoch": 1.251220703125e-05,
      "step": 3280
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 3280,
      "training_step_time": 9.938056468963623
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 3.1472649574279785,
      "step": 3280
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 3280,
      "training_step_time": 9.929213285446167
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 3.1435787677764893,
      "step": 3280
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 3280,
      "training_step_time": 9.920565366744995
    },
    {
      "epoch": 1.251220703125e-05,
      "model_forward_time": 3.14522385597229,
      "step": 3280
    },
    {
      "epoch": 1.251220703125e-05,
      "step": 3280,
      "training_step_time": 10.030774593353271
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 3.157332181930542,
      "step": 3281
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 3281,
      "training_step_time": 10.215312719345093
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 3.1524105072021484,
      "step": 3281
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 3281,
      "training_step_time": 9.935582637786865
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 3.14689302444458,
      "step": 3281
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 3281,
      "training_step_time": 9.931062698364258
    },
    {
      "epoch": 1.2516021728515626e-05,
      "model_forward_time": 3.1506593227386475,
      "step": 3281
    },
    {
      "epoch": 1.2516021728515626e-05,
      "step": 3281,
      "training_step_time": 10.302814722061157
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 3.1706113815307617,
      "step": 3282
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 3282,
      "training_step_time": 9.957228183746338
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 3.564638376235962,
      "step": 3282
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 3282,
      "training_step_time": 10.35179090499878
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 3.151773691177368,
      "step": 3282
    },
    {
      "dynamics_loss_avg": 0.06515751332044602,
      "epoch": 1.251983642578125e-05,
      "step": 3282
    },
    {
      "action_loss_avg": 0.031376026663929225,
      "epoch": 1.251983642578125e-05,
      "step": 3282
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 3282,
      "training_step_time": 9.941783428192139
    },
    {
      "epoch": 1.251983642578125e-05,
      "model_forward_time": 3.1448259353637695,
      "step": 3282
    },
    {
      "epoch": 1.251983642578125e-05,
      "step": 3282,
      "training_step_time": 10.281436443328857
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 3.156419515609741,
      "step": 3283
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 3283,
      "training_step_time": 9.936517477035522
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 3.1387345790863037,
      "step": 3283
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 3283,
      "training_step_time": 9.908899307250977
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 3.4597175121307373,
      "step": 3283
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 3283,
      "training_step_time": 10.228087425231934
    },
    {
      "epoch": 1.2523651123046875e-05,
      "model_forward_time": 3.137824296951294,
      "step": 3283
    },
    {
      "epoch": 1.2523651123046875e-05,
      "step": 3283,
      "training_step_time": 10.309946298599243
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 3.1558759212493896,
      "step": 3284
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 3284,
      "training_step_time": 9.92615532875061
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 3.1439881324768066,
      "step": 3284
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 3284,
      "training_step_time": 9.916118860244751
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 3.1419320106506348,
      "step": 3284
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 3284,
      "training_step_time": 9.916672945022583
    },
    {
      "epoch": 1.25274658203125e-05,
      "model_forward_time": 3.425858974456787,
      "step": 3284
    },
    {
      "epoch": 1.25274658203125e-05,
      "step": 3284,
      "training_step_time": 10.292312145233154
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 3.1548781394958496,
      "step": 3285
    },
    {
      "dynamics_loss_avg": 0.07181067280471325,
      "epoch": 1.2531280517578126e-05,
      "step": 3285
    },
    {
      "action_loss_avg": 0.04128152970224619,
      "epoch": 1.2531280517578126e-05,
      "step": 3285
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 3285,
      "training_step_time": 9.93543791770935
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 3.133697748184204,
      "step": 3285
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 3285,
      "training_step_time": 9.903473377227783
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 3.1373021602630615,
      "step": 3285
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 3285,
      "training_step_time": 9.908135175704956
    },
    {
      "epoch": 1.2531280517578126e-05,
      "model_forward_time": 3.1352474689483643,
      "step": 3285
    },
    {
      "epoch": 1.2531280517578126e-05,
      "step": 3285,
      "training_step_time": 10.230472803115845
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 3.4289004802703857,
      "step": 3286
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 3286,
      "training_step_time": 10.210859775543213
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 3.1379971504211426,
      "step": 3286
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 3286,
      "training_step_time": 9.911418199539185
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 3.136652946472168,
      "step": 3286
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 3286,
      "training_step_time": 9.910282373428345
    },
    {
      "epoch": 1.253509521484375e-05,
      "model_forward_time": 3.132425546646118,
      "step": 3286
    },
    {
      "epoch": 1.253509521484375e-05,
      "step": 3286,
      "training_step_time": 10.284494400024414
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 3.1680922508239746,
      "step": 3287
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 3287,
      "training_step_time": 9.956258058547974
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 2.4357457160949707,
      "step": 3287
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 3287,
      "training_step_time": 7.777130842208862
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 2.4362964630126953,
      "step": 3287
    },
    {
      "dynamics_loss_avg": 0.0617721302434802,
      "epoch": 1.2538909912109375e-05,
      "step": 3287
    },
    {
      "action_loss_avg": 0.028175786696374418,
      "epoch": 1.2538909912109375e-05,
      "step": 3287
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 3287,
      "training_step_time": 7.463451862335205
    },
    {
      "epoch": 1.2538909912109375e-05,
      "model_forward_time": 3.1301591396331787,
      "step": 3287
    },
    {
      "epoch": 1.2538909912109375e-05,
      "step": 3287,
      "training_step_time": 10.005548477172852
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 3.13862681388855,
      "step": 3288
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 3288,
      "training_step_time": 9.909034967422485
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 2.433763027191162,
      "step": 3288
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 3288,
      "training_step_time": 7.46000599861145
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 3.1328468322753906,
      "step": 3288
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 3288,
      "training_step_time": 9.90317177772522
    },
    {
      "epoch": 1.2542724609375e-05,
      "model_forward_time": 3.398175001144409,
      "step": 3288
    },
    {
      "epoch": 1.2542724609375e-05,
      "step": 3288,
      "training_step_time": 10.314446687698364
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 3.149540901184082,
      "step": 3289
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 3289,
      "training_step_time": 9.921099662780762
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 3.1337339878082275,
      "step": 3289
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 3289,
      "training_step_time": 9.906930208206177
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 3.1345202922821045,
      "step": 3289
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 3289,
      "training_step_time": 9.910916805267334
    },
    {
      "epoch": 1.2546539306640626e-05,
      "model_forward_time": 3.1418797969818115,
      "step": 3289
    },
    {
      "epoch": 1.2546539306640626e-05,
      "step": 3289,
      "training_step_time": 10.280807256698608
    },
    {
      "epoch": 1.255035400390625e-05,
      "grad_norm": 0.3987472951412201,
      "learning_rate": 1e-05,
      "loss": 0.0976,
      "step": 3290
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 3.1539576053619385,
      "step": 3290
    },
    {
      "dynamics_loss_avg": 0.0668006232008338,
      "epoch": 1.255035400390625e-05,
      "step": 3290
    },
    {
      "action_loss_avg": 0.0392747353296727,
      "epoch": 1.255035400390625e-05,
      "step": 3290
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 3290,
      "training_step_time": 9.938915967941284
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 3.1400113105773926,
      "step": 3290
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 3290,
      "training_step_time": 9.91600513458252
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 3.136293649673462,
      "step": 3290
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 3290,
      "training_step_time": 9.913877248764038
    },
    {
      "epoch": 1.255035400390625e-05,
      "model_forward_time": 3.1380345821380615,
      "step": 3290
    },
    {
      "epoch": 1.255035400390625e-05,
      "step": 3290,
      "training_step_time": 10.012287616729736
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 3.4082956314086914,
      "step": 3291
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 3291,
      "training_step_time": 10.182947874069214
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 3.1444828510284424,
      "step": 3291
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 3291,
      "training_step_time": 9.918633937835693
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 3.140902280807495,
      "step": 3291
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 3291,
      "training_step_time": 9.921833276748657
    },
    {
      "epoch": 1.2554168701171875e-05,
      "model_forward_time": 3.146273374557495,
      "step": 3291
    },
    {
      "epoch": 1.2554168701171875e-05,
      "step": 3291,
      "training_step_time": 10.324294328689575
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 3.1514368057250977,
      "step": 3292
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 3292,
      "training_step_time": 9.929635286331177
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 3.438788414001465,
      "step": 3292
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 3292,
      "training_step_time": 10.213794231414795
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 3.1398909091949463,
      "step": 3292
    },
    {
      "dynamics_loss_avg": 0.07181937899440527,
      "epoch": 1.25579833984375e-05,
      "step": 3292
    },
    {
      "action_loss_avg": 0.03062854539602995,
      "epoch": 1.25579833984375e-05,
      "step": 3292
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 3292,
      "training_step_time": 9.917196989059448
    },
    {
      "epoch": 1.25579833984375e-05,
      "model_forward_time": 3.1431097984313965,
      "step": 3292
    },
    {
      "epoch": 1.25579833984375e-05,
      "step": 3292,
      "training_step_time": 10.263503313064575
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 3.1611459255218506,
      "step": 3293
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 3293,
      "training_step_time": 9.942553758621216
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 3.138575792312622,
      "step": 3293
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 3293,
      "training_step_time": 9.91830587387085
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 3.1396524906158447,
      "step": 3293
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 3293,
      "training_step_time": 10.216408967971802
    },
    {
      "epoch": 1.2561798095703126e-05,
      "model_forward_time": 3.136103868484497,
      "step": 3293
    },
    {
      "epoch": 1.2561798095703126e-05,
      "step": 3293,
      "training_step_time": 10.311076402664185
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 2.4488584995269775,
      "step": 3294
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 3294,
      "training_step_time": 7.4881250858306885
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 3.1376824378967285,
      "step": 3294
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 3294,
      "training_step_time": 9.916907548904419
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 3.141441822052002,
      "step": 3294
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 3294,
      "training_step_time": 9.91580581665039
    },
    {
      "epoch": 1.256561279296875e-05,
      "model_forward_time": 3.1381618976593018,
      "step": 3294
    },
    {
      "epoch": 1.256561279296875e-05,
      "step": 3294,
      "training_step_time": 10.008238554000854
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 3.516559600830078,
      "step": 3295
    },
    {
      "dynamics_loss_avg": 0.06407633703202009,
      "epoch": 1.2569427490234375e-05,
      "step": 3295
    },
    {
      "action_loss_avg": 0.028993066493421794,
      "epoch": 1.2569427490234375e-05,
      "step": 3295
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 3295,
      "training_step_time": 10.310800790786743
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 3.135793924331665,
      "step": 3295
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 3295,
      "training_step_time": 9.911644697189331
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 3.1393566131591797,
      "step": 3295
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 3295,
      "training_step_time": 9.913647651672363
    },
    {
      "epoch": 1.2569427490234375e-05,
      "model_forward_time": 3.134953737258911,
      "step": 3295
    },
    {
      "epoch": 1.2569427490234375e-05,
      "step": 3295,
      "training_step_time": 10.010092496871948
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 3.143893003463745,
      "step": 3296
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 3296,
      "training_step_time": 10.20321536064148
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 3.1378333568573,
      "step": 3296
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 3296,
      "training_step_time": 9.913688898086548
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 3.139277219772339,
      "step": 3296
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 3296,
      "training_step_time": 9.922732591629028
    },
    {
      "epoch": 1.25732421875e-05,
      "model_forward_time": 3.144270658493042,
      "step": 3296
    },
    {
      "epoch": 1.25732421875e-05,
      "step": 3296,
      "training_step_time": 10.28013277053833
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 3.1494946479797363,
      "step": 3297
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 3297,
      "training_step_time": 9.92763876914978
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 2.439363718032837,
      "step": 3297
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 3297,
      "training_step_time": 7.79193377494812
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 3.1402838230133057,
      "step": 3297
    },
    {
      "dynamics_loss_avg": 0.07349188663065434,
      "epoch": 1.2577056884765626e-05,
      "step": 3297
    },
    {
      "action_loss_avg": 0.037118575535714626,
      "epoch": 1.2577056884765626e-05,
      "step": 3297
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 3297,
      "training_step_time": 9.920960903167725
    },
    {
      "epoch": 1.2577056884765626e-05,
      "model_forward_time": 3.138226270675659,
      "step": 3297
    },
    {
      "epoch": 1.2577056884765626e-05,
      "step": 3297,
      "training_step_time": 10.257053852081299
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 3.153676748275757,
      "step": 3298
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 3298,
      "training_step_time": 9.937471151351929
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 3.1386008262634277,
      "step": 3298
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 3298,
      "training_step_time": 9.916723489761353
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 3.1359074115753174,
      "step": 3298
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 3298,
      "training_step_time": 10.190920352935791
    },
    {
      "epoch": 1.258087158203125e-05,
      "model_forward_time": 3.136976718902588,
      "step": 3298
    },
    {
      "epoch": 1.258087158203125e-05,
      "step": 3298,
      "training_step_time": 10.30711841583252
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 3.1529691219329834,
      "step": 3299
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 3299,
      "training_step_time": 9.932156801223755
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 2.4322094917297363,
      "step": 3299
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 3299,
      "training_step_time": 7.460259199142456
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 2.4304332733154297,
      "step": 3299
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 3299,
      "training_step_time": 7.461145639419556
    },
    {
      "epoch": 1.2584686279296875e-05,
      "model_forward_time": 3.4266767501831055,
      "step": 3299
    },
    {
      "epoch": 1.2584686279296875e-05,
      "step": 3299,
      "training_step_time": 10.29421877861023
    },
    {
      "epoch": 1.25885009765625e-05,
      "grad_norm": 0.3396669030189514,
      "learning_rate": 1e-05,
      "loss": 0.0947,
      "step": 3300
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 3.145883798599243,
      "step": 3300
    },
    {
      "dynamics_loss_avg": 0.05321716796606779,
      "epoch": 1.25885009765625e-05,
      "step": 3300
    },
    {
      "action_loss_avg": 0.02261065817438066,
      "epoch": 1.25885009765625e-05,
      "step": 3300
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 3300,
      "training_step_time": 9.936612129211426
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 3.140669345855713,
      "step": 3300
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 3300,
      "training_step_time": 9.919232845306396
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 3.137173891067505,
      "step": 3300
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 3300,
      "training_step_time": 9.918362855911255
    },
    {
      "epoch": 1.25885009765625e-05,
      "model_forward_time": 3.1433773040771484,
      "step": 3300
    },
    {
      "epoch": 1.25885009765625e-05,
      "step": 3300,
      "training_step_time": 10.251986026763916
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 3.1508700847625732,
      "step": 3301
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 3301,
      "training_step_time": 10.205046892166138
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 3.141333818435669,
      "step": 3301
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 3301,
      "training_step_time": 9.926069021224976
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 3.1422603130340576,
      "step": 3301
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 3301,
      "training_step_time": 9.925189971923828
    },
    {
      "epoch": 1.2592315673828126e-05,
      "model_forward_time": 3.1459264755249023,
      "step": 3301
    },
    {
      "epoch": 1.2592315673828126e-05,
      "step": 3301,
      "training_step_time": 10.027444124221802
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 3.158080816268921,
      "step": 3302
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 3302,
      "training_step_time": 9.948675870895386
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 3.4209280014038086,
      "step": 3302
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 3302,
      "training_step_time": 10.20583724975586
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 2.442688226699829,
      "step": 3302
    },
    {
      "dynamics_loss_avg": 0.07845109980553389,
      "epoch": 1.259613037109375e-05,
      "step": 3302
    },
    {
      "action_loss_avg": 0.037453463114798066,
      "epoch": 1.259613037109375e-05,
      "step": 3302
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 3302,
      "training_step_time": 7.477200269699097
    },
    {
      "epoch": 1.259613037109375e-05,
      "model_forward_time": 3.143127679824829,
      "step": 3302
    },
    {
      "epoch": 1.259613037109375e-05,
      "step": 3302,
      "training_step_time": 10.29967713356018
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 3.151015043258667,
      "step": 3303
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 3303,
      "training_step_time": 9.929547786712646
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 3.138554096221924,
      "step": 3303
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 3303,
      "training_step_time": 9.907563209533691
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 3.4418442249298096,
      "step": 3303
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 3303,
      "training_step_time": 10.209983587265015
    },
    {
      "epoch": 1.2599945068359375e-05,
      "model_forward_time": 3.1382315158843994,
      "step": 3303
    },
    {
      "epoch": 1.2599945068359375e-05,
      "step": 3303,
      "training_step_time": 10.30408000946045
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 3.148005723953247,
      "step": 3304
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 3304,
      "training_step_time": 9.920706272125244
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 3.138960123062134,
      "step": 3304
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 3304,
      "training_step_time": 9.915077924728394
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 3.1420912742614746,
      "step": 3304
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 3304,
      "training_step_time": 9.914588451385498
    },
    {
      "epoch": 1.2603759765625e-05,
      "model_forward_time": 3.13632869720459,
      "step": 3304
    },
    {
      "epoch": 1.2603759765625e-05,
      "step": 3304,
      "training_step_time": 10.312152624130249
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 2.450324058532715,
      "step": 3305
    },
    {
      "dynamics_loss_avg": 0.05592055432498455,
      "epoch": 1.2607574462890626e-05,
      "step": 3305
    },
    {
      "action_loss_avg": 0.02555751255713403,
      "epoch": 1.2607574462890626e-05,
      "step": 3305
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 3305,
      "training_step_time": 7.492429256439209
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 3.136392116546631,
      "step": 3305
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 3305,
      "training_step_time": 9.916097640991211
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 3.141382932662964,
      "step": 3305
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 3305,
      "training_step_time": 9.928181648254395
    },
    {
      "epoch": 1.2607574462890626e-05,
      "model_forward_time": 3.139456272125244,
      "step": 3305
    },
    {
      "epoch": 1.2607574462890626e-05,
      "step": 3305,
      "training_step_time": 10.066845178604126
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 3.15496563911438,
      "step": 3306
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 3306,
      "training_step_time": 10.251348495483398
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 3.1415462493896484,
      "step": 3306
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 3306,
      "training_step_time": 9.922786235809326
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 3.1446659564971924,
      "step": 3306
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 3306,
      "training_step_time": 9.928702592849731
    },
    {
      "epoch": 1.261138916015625e-05,
      "model_forward_time": 3.1390326023101807,
      "step": 3306
    },
    {
      "epoch": 1.261138916015625e-05,
      "step": 3306,
      "training_step_time": 10.039348125457764
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 3.1510190963745117,
      "step": 3307
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 3307,
      "training_step_time": 9.937122821807861
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 3.1395208835601807,
      "step": 3307
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 3307,
      "training_step_time": 10.229836463928223
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 3.138145923614502,
      "step": 3307
    },
    {
      "dynamics_loss_avg": 0.06056291051208973,
      "epoch": 1.2615203857421875e-05,
      "step": 3307
    },
    {
      "action_loss_avg": 0.03539119753986597,
      "epoch": 1.2615203857421875e-05,
      "step": 3307
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 3307,
      "training_step_time": 9.921454191207886
    },
    {
      "epoch": 1.2615203857421875e-05,
      "model_forward_time": 3.1372230052948,
      "step": 3307
    },
    {
      "epoch": 1.2615203857421875e-05,
      "step": 3307,
      "training_step_time": 10.258379697799683
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 3.1620748043060303,
      "step": 3308
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 3308,
      "training_step_time": 9.957477807998657
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 3.144669771194458,
      "step": 3308
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 3308,
      "training_step_time": 9.924464702606201
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 3.442901611328125,
      "step": 3308
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 3308,
      "training_step_time": 10.228803396224976
    },
    {
      "epoch": 1.26190185546875e-05,
      "model_forward_time": 3.141456365585327,
      "step": 3308
    },
    {
      "epoch": 1.26190185546875e-05,
      "step": 3308,
      "training_step_time": 10.22771692276001
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 3.1426713466644287,
      "step": 3309
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 3309,
      "training_step_time": 9.927019119262695
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 3.141166925430298,
      "step": 3309
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 3309,
      "training_step_time": 9.921828985214233
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 3.1396827697753906,
      "step": 3309
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 3309,
      "training_step_time": 9.920240879058838
    },
    {
      "epoch": 1.2622833251953126e-05,
      "model_forward_time": 3.434649705886841,
      "step": 3309
    },
    {
      "epoch": 1.2622833251953126e-05,
      "step": 3309,
      "training_step_time": 10.317574501037598
    },
    {
      "epoch": 1.262664794921875e-05,
      "grad_norm": 0.49606257677078247,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 3310
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 3.1431727409362793,
      "step": 3310
    },
    {
      "dynamics_loss_avg": 0.06990343295037746,
      "epoch": 1.262664794921875e-05,
      "step": 3310
    },
    {
      "action_loss_avg": 0.028197235148400068,
      "epoch": 1.262664794921875e-05,
      "step": 3310
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 3310,
      "training_step_time": 9.92354702949524
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 3.1412453651428223,
      "step": 3310
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 3310,
      "training_step_time": 9.919533491134644
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 3.1362361907958984,
      "step": 3310
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 3310,
      "training_step_time": 9.916019916534424
    },
    {
      "epoch": 1.262664794921875e-05,
      "model_forward_time": 3.141277551651001,
      "step": 3310
    },
    {
      "epoch": 1.262664794921875e-05,
      "step": 3310,
      "training_step_time": 10.033551216125488
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 3.423302412033081,
      "step": 3311
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 3311,
      "training_step_time": 10.203573226928711
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 3.1452627182006836,
      "step": 3311
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 3311,
      "training_step_time": 9.929315090179443
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 3.138157367706299,
      "step": 3311
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 3311,
      "training_step_time": 9.916501998901367
    },
    {
      "epoch": 1.2630462646484375e-05,
      "model_forward_time": 3.1373727321624756,
      "step": 3311
    },
    {
      "epoch": 1.2630462646484375e-05,
      "step": 3311,
      "training_step_time": 10.019266366958618
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 3.149930953979492,
      "step": 3312
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 3312,
      "training_step_time": 9.926215887069702
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 3.4360761642456055,
      "step": 3312
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 3312,
      "training_step_time": 10.214046239852905
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 3.1354146003723145,
      "step": 3312
    },
    {
      "dynamics_loss_avg": 0.07112163081765174,
      "epoch": 1.263427734375e-05,
      "step": 3312
    },
    {
      "action_loss_avg": 0.03588725160807371,
      "epoch": 1.263427734375e-05,
      "step": 3312
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 3312,
      "training_step_time": 9.914990425109863
    },
    {
      "epoch": 1.263427734375e-05,
      "model_forward_time": 3.1412594318389893,
      "step": 3312
    },
    {
      "epoch": 1.263427734375e-05,
      "step": 3312,
      "training_step_time": 10.264313697814941
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 3.1587085723876953,
      "step": 3313
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 3313,
      "training_step_time": 9.934650659561157
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 3.137134552001953,
      "step": 3313
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 3313,
      "training_step_time": 9.917316198348999
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 3.1397616863250732,
      "step": 3313
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 3313,
      "training_step_time": 10.232705354690552
    },
    {
      "epoch": 1.2638092041015626e-05,
      "model_forward_time": 3.1363344192504883,
      "step": 3313
    },
    {
      "epoch": 1.2638092041015626e-05,
      "step": 3313,
      "training_step_time": 10.020838499069214
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 3.15201735496521,
      "step": 3314
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 3314,
      "training_step_time": 9.933576822280884
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 3.139432907104492,
      "step": 3314
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 3314,
      "training_step_time": 9.921323299407959
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 3.143751621246338,
      "step": 3314
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 3314,
      "training_step_time": 9.922686338424683
    },
    {
      "epoch": 1.264190673828125e-05,
      "model_forward_time": 3.425403118133545,
      "step": 3314
    },
    {
      "epoch": 1.264190673828125e-05,
      "step": 3314,
      "training_step_time": 10.34054183959961
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 3.151425361633301,
      "step": 3315
    },
    {
      "dynamics_loss_avg": 0.07605354525148869,
      "epoch": 1.2645721435546875e-05,
      "step": 3315
    },
    {
      "action_loss_avg": 0.03442020630463958,
      "epoch": 1.2645721435546875e-05,
      "step": 3315
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 3315,
      "training_step_time": 9.933862686157227
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 3.137587308883667,
      "step": 3315
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 3315,
      "training_step_time": 9.917912483215332
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 2.440304756164551,
      "step": 3315
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 3315,
      "training_step_time": 7.477259159088135
    },
    {
      "epoch": 1.2645721435546875e-05,
      "model_forward_time": 3.1392452716827393,
      "step": 3315
    },
    {
      "epoch": 1.2645721435546875e-05,
      "step": 3315,
      "training_step_time": 10.289869785308838
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 3.1407694816589355,
      "step": 3316
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 3316,
      "training_step_time": 9.922926425933838
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 3.378943920135498,
      "step": 3316
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 3316,
      "training_step_time": 10.149760961532593
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 3.1383941173553467,
      "step": 3316
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 3316,
      "training_step_time": 9.911973714828491
    },
    {
      "epoch": 1.26495361328125e-05,
      "model_forward_time": 2.432163715362549,
      "step": 3316
    },
    {
      "epoch": 1.26495361328125e-05,
      "step": 3316,
      "training_step_time": 10.033851861953735
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 3.13967227935791,
      "step": 3317
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 3317,
      "training_step_time": 9.925477504730225
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 2.4335579872131348,
      "step": 3317
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 3317,
      "training_step_time": 7.471594333648682
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 3.4484636783599854,
      "step": 3317
    },
    {
      "dynamics_loss_avg": 0.06091447481885552,
      "epoch": 1.2653350830078126e-05,
      "step": 3317
    },
    {
      "action_loss_avg": 0.02734594764187932,
      "epoch": 1.2653350830078126e-05,
      "step": 3317
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 3317,
      "training_step_time": 10.219155550003052
    },
    {
      "epoch": 1.2653350830078126e-05,
      "model_forward_time": 3.1297357082366943,
      "step": 3317
    },
    {
      "epoch": 1.2653350830078126e-05,
      "step": 3317,
      "training_step_time": 10.017467260360718
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 3.1492278575897217,
      "step": 3318
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 3318,
      "training_step_time": 9.924795866012573
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 3.1376755237579346,
      "step": 3318
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 3318,
      "training_step_time": 9.911663293838501
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 3.1310737133026123,
      "step": 3318
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 3318,
      "training_step_time": 10.187093019485474
    },
    {
      "epoch": 1.265716552734375e-05,
      "model_forward_time": 3.134474039077759,
      "step": 3318
    },
    {
      "epoch": 1.265716552734375e-05,
      "step": 3318,
      "training_step_time": 10.276360988616943
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 3.1386053562164307,
      "step": 3319
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 3319,
      "training_step_time": 9.914345502853394
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 3.1344988346099854,
      "step": 3319
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 3319,
      "training_step_time": 9.909993410110474
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 3.1350204944610596,
      "step": 3319
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 3319,
      "training_step_time": 9.907011985778809
    },
    {
      "epoch": 1.2660980224609375e-05,
      "model_forward_time": 3.4184625148773193,
      "step": 3319
    },
    {
      "epoch": 1.2660980224609375e-05,
      "step": 3319,
      "training_step_time": 10.340157508850098
    },
    {
      "epoch": 1.2664794921875e-05,
      "grad_norm": 0.527317225933075,
      "learning_rate": 1e-05,
      "loss": 0.098,
      "step": 3320
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 3.149125576019287,
      "step": 3320
    },
    {
      "dynamics_loss_avg": 0.06494298744946718,
      "epoch": 1.2664794921875e-05,
      "step": 3320
    },
    {
      "action_loss_avg": 0.03204755699262023,
      "epoch": 1.2664794921875e-05,
      "step": 3320
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 3320,
      "training_step_time": 9.931033611297607
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 3.143012285232544,
      "step": 3320
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 3320,
      "training_step_time": 9.924182653427124
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 3.1392734050750732,
      "step": 3320
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 3320,
      "training_step_time": 9.913810729980469
    },
    {
      "epoch": 1.2664794921875e-05,
      "model_forward_time": 3.1403369903564453,
      "step": 3320
    },
    {
      "epoch": 1.2664794921875e-05,
      "step": 3320,
      "training_step_time": 10.30730938911438
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 3.415858030319214,
      "step": 3321
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 3321,
      "training_step_time": 10.189884185791016
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 3.135479688644409,
      "step": 3321
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 3321,
      "training_step_time": 9.906442880630493
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 3.136284351348877,
      "step": 3321
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 3321,
      "training_step_time": 9.90208101272583
    },
    {
      "epoch": 1.2668609619140626e-05,
      "model_forward_time": 3.1373519897460938,
      "step": 3321
    },
    {
      "epoch": 1.2668609619140626e-05,
      "step": 3321,
      "training_step_time": 10.004410743713379
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 3.1547369956970215,
      "step": 3322
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 3322,
      "training_step_time": 9.925029039382935
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 3.1321191787719727,
      "step": 3322
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 3322,
      "training_step_time": 10.218779563903809
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 3.1381444931030273,
      "step": 3322
    },
    {
      "dynamics_loss_avg": 0.06236463002860546,
      "epoch": 1.267242431640625e-05,
      "step": 3322
    },
    {
      "action_loss_avg": 0.027955334773287176,
      "epoch": 1.267242431640625e-05,
      "step": 3322
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 3322,
      "training_step_time": 9.915175676345825
    },
    {
      "epoch": 1.267242431640625e-05,
      "model_forward_time": 3.1431283950805664,
      "step": 3322
    },
    {
      "epoch": 1.267242431640625e-05,
      "step": 3322,
      "training_step_time": 10.321227073669434
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 3.1515932083129883,
      "step": 3323
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 3323,
      "training_step_time": 9.924576044082642
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 3.135751247406006,
      "step": 3323
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 3323,
      "training_step_time": 9.905627727508545
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 3.3969669342041016,
      "step": 3323
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 3323,
      "training_step_time": 10.166703939437866
    },
    {
      "epoch": 1.2676239013671875e-05,
      "model_forward_time": 3.1390745639801025,
      "step": 3323
    },
    {
      "epoch": 1.2676239013671875e-05,
      "step": 3323,
      "training_step_time": 10.01207971572876
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 3.150953531265259,
      "step": 3324
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 3324,
      "training_step_time": 9.922585725784302
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 3.1330296993255615,
      "step": 3324
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 3324,
      "training_step_time": 9.899648189544678
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 3.1378390789031982,
      "step": 3324
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 3324,
      "training_step_time": 9.904366970062256
    },
    {
      "epoch": 1.26800537109375e-05,
      "model_forward_time": 3.1339218616485596,
      "step": 3324
    },
    {
      "epoch": 1.26800537109375e-05,
      "step": 3324,
      "training_step_time": 10.29881739616394
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 3.143453359603882,
      "step": 3325
    },
    {
      "dynamics_loss_avg": 0.08049869686365127,
      "epoch": 1.2683868408203126e-05,
      "step": 3325
    },
    {
      "action_loss_avg": 0.04785598535090685,
      "epoch": 1.2683868408203126e-05,
      "step": 3325
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 3325,
      "training_step_time": 9.917730569839478
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 3.1308541297912598,
      "step": 3325
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 3325,
      "training_step_time": 9.903947353363037
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 3.145418167114258,
      "step": 3325
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 3325,
      "training_step_time": 9.916471481323242
    },
    {
      "epoch": 1.2683868408203126e-05,
      "model_forward_time": 3.1391119956970215,
      "step": 3325
    },
    {
      "epoch": 1.2683868408203126e-05,
      "step": 3325,
      "training_step_time": 10.29133129119873
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 3.439213991165161,
      "step": 3326
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 3326,
      "training_step_time": 10.21425724029541
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 3.14111590385437,
      "step": 3326
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 3326,
      "training_step_time": 9.91411828994751
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 3.144869565963745,
      "step": 3326
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 3326,
      "training_step_time": 9.915700674057007
    },
    {
      "epoch": 1.268768310546875e-05,
      "model_forward_time": 3.1456353664398193,
      "step": 3326
    },
    {
      "epoch": 1.268768310546875e-05,
      "step": 3326,
      "training_step_time": 10.336501121520996
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 3.1476995944976807,
      "step": 3327
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 3327,
      "training_step_time": 9.921308994293213
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 3.140503406524658,
      "step": 3327
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 3327,
      "training_step_time": 10.177234172821045
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 3.1407530307769775,
      "step": 3327
    },
    {
      "dynamics_loss_avg": 0.068027688190341,
      "epoch": 1.2691497802734375e-05,
      "step": 3327
    },
    {
      "action_loss_avg": 0.03203560840338469,
      "epoch": 1.2691497802734375e-05,
      "step": 3327
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 3327,
      "training_step_time": 9.919734001159668
    },
    {
      "epoch": 1.2691497802734375e-05,
      "model_forward_time": 3.1401658058166504,
      "step": 3327
    },
    {
      "epoch": 1.2691497802734375e-05,
      "step": 3327,
      "training_step_time": 10.300272703170776
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 3.1534650325775146,
      "step": 3328
    },
    {
      "epoch": 1.26953125e-05,
      "step": 3328,
      "training_step_time": 9.931938409805298
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 3.1443746089935303,
      "step": 3328
    },
    {
      "epoch": 1.26953125e-05,
      "step": 3328,
      "training_step_time": 9.919074058532715
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 3.4207074642181396,
      "step": 3328
    },
    {
      "epoch": 1.26953125e-05,
      "step": 3328,
      "training_step_time": 10.191567659378052
    },
    {
      "epoch": 1.26953125e-05,
      "model_forward_time": 3.142301321029663,
      "step": 3328
    },
    {
      "epoch": 1.26953125e-05,
      "step": 3328,
      "training_step_time": 10.022572994232178
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 3.15731143951416,
      "step": 3329
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 3329,
      "training_step_time": 9.932010173797607
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 3.1456406116485596,
      "step": 3329
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 3329,
      "training_step_time": 9.917258262634277
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 3.142108201980591,
      "step": 3329
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 3329,
      "training_step_time": 9.921315670013428
    },
    {
      "epoch": 1.2699127197265626e-05,
      "model_forward_time": 3.4072799682617188,
      "step": 3329
    },
    {
      "epoch": 1.2699127197265626e-05,
      "step": 3329,
      "training_step_time": 10.333048820495605
    },
    {
      "epoch": 1.270294189453125e-05,
      "grad_norm": 0.6972860097885132,
      "learning_rate": 1e-05,
      "loss": 0.0994,
      "step": 3330
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 3.152498483657837,
      "step": 3330
    },
    {
      "dynamics_loss_avg": 0.06319630034267902,
      "epoch": 1.270294189453125e-05,
      "step": 3330
    },
    {
      "action_loss_avg": 0.032689739018678665,
      "epoch": 1.270294189453125e-05,
      "step": 3330
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 3330,
      "training_step_time": 9.93112325668335
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 3.1481528282165527,
      "step": 3330
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 3330,
      "training_step_time": 9.925609350204468
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 3.1426548957824707,
      "step": 3330
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 3330,
      "training_step_time": 9.921101093292236
    },
    {
      "epoch": 1.270294189453125e-05,
      "model_forward_time": 3.151237726211548,
      "step": 3330
    },
    {
      "epoch": 1.270294189453125e-05,
      "step": 3330,
      "training_step_time": 10.325573921203613
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 3.473806619644165,
      "step": 3331
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 3331,
      "training_step_time": 10.243575096130371
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 3.142076015472412,
      "step": 3331
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 3331,
      "training_step_time": 9.913546323776245
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 3.14906907081604,
      "step": 3331
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 3331,
      "training_step_time": 9.923485517501831
    },
    {
      "epoch": 1.2706756591796875e-05,
      "model_forward_time": 3.142508029937744,
      "step": 3331
    },
    {
      "epoch": 1.2706756591796875e-05,
      "step": 3331,
      "training_step_time": 10.01123833656311
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 3.1419262886047363,
      "step": 3332
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 3332,
      "training_step_time": 9.919433355331421
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 3.418466806411743,
      "step": 3332
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 3332,
      "training_step_time": 10.197724342346191
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 3.132824182510376,
      "step": 3332
    },
    {
      "dynamics_loss_avg": 0.07518929969519376,
      "epoch": 1.27105712890625e-05,
      "step": 3332
    },
    {
      "action_loss_avg": 0.038363369088619946,
      "epoch": 1.27105712890625e-05,
      "step": 3332
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 3332,
      "training_step_time": 9.910889387130737
    },
    {
      "epoch": 1.27105712890625e-05,
      "model_forward_time": 3.1360530853271484,
      "step": 3332
    },
    {
      "epoch": 1.27105712890625e-05,
      "step": 3332,
      "training_step_time": 10.303784370422363
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 3.151289463043213,
      "step": 3333
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 3333,
      "training_step_time": 9.929420709609985
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 3.1330771446228027,
      "step": 3333
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 3333,
      "training_step_time": 9.911882400512695
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 3.137413501739502,
      "step": 3333
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 3333,
      "training_step_time": 10.198606491088867
    },
    {
      "epoch": 1.2714385986328126e-05,
      "model_forward_time": 3.1435201168060303,
      "step": 3333
    },
    {
      "epoch": 1.2714385986328126e-05,
      "step": 3333,
      "training_step_time": 10.02325963973999
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 3.158142328262329,
      "step": 3334
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 3334,
      "training_step_time": 9.936480045318604
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 3.142061471939087,
      "step": 3334
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 3334,
      "training_step_time": 9.921012878417969
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 3.1481311321258545,
      "step": 3334
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 3334,
      "training_step_time": 9.924495697021484
    },
    {
      "epoch": 1.271820068359375e-05,
      "model_forward_time": 3.431058406829834,
      "step": 3334
    },
    {
      "epoch": 1.271820068359375e-05,
      "step": 3334,
      "training_step_time": 10.301711559295654
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 3.1630592346191406,
      "step": 3335
    },
    {
      "dynamics_loss_avg": 0.0680924855172634,
      "epoch": 1.2722015380859375e-05,
      "step": 3335
    },
    {
      "action_loss_avg": 0.03558260556310415,
      "epoch": 1.2722015380859375e-05,
      "step": 3335
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 3335,
      "training_step_time": 9.946584463119507
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 3.1455037593841553,
      "step": 3335
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 3335,
      "training_step_time": 9.925641775131226
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 3.149796962738037,
      "step": 3335
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 3335,
      "training_step_time": 9.93111252784729
    },
    {
      "epoch": 1.2722015380859375e-05,
      "model_forward_time": 3.1471757888793945,
      "step": 3335
    },
    {
      "epoch": 1.2722015380859375e-05,
      "step": 3335,
      "training_step_time": 10.28128695487976
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 3.148494243621826,
      "step": 3336
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 3336,
      "training_step_time": 10.207053184509277
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 3.137158155441284,
      "step": 3336
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 3336,
      "training_step_time": 9.919105529785156
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 3.1422555446624756,
      "step": 3336
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 3336,
      "training_step_time": 9.933681726455688
    },
    {
      "epoch": 1.2725830078125e-05,
      "model_forward_time": 3.1402428150177,
      "step": 3336
    },
    {
      "epoch": 1.2725830078125e-05,
      "step": 3336,
      "training_step_time": 10.309837102890015
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 3.147348642349243,
      "step": 3337
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 3337,
      "training_step_time": 9.926665544509888
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 3.397768497467041,
      "step": 3337
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 3337,
      "training_step_time": 10.173539161682129
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 3.136753797531128,
      "step": 3337
    },
    {
      "dynamics_loss_avg": 0.05658170394599438,
      "epoch": 1.2729644775390626e-05,
      "step": 3337
    },
    {
      "action_loss_avg": 0.022791787981987,
      "epoch": 1.2729644775390626e-05,
      "step": 3337
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 3337,
      "training_step_time": 9.91455364227295
    },
    {
      "epoch": 1.2729644775390626e-05,
      "model_forward_time": 3.133927822113037,
      "step": 3337
    },
    {
      "epoch": 1.2729644775390626e-05,
      "step": 3337,
      "training_step_time": 10.01579761505127
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 3.1568243503570557,
      "step": 3338
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 3338,
      "training_step_time": 9.925716876983643
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 3.1373353004455566,
      "step": 3338
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 3338,
      "training_step_time": 9.91125226020813
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 3.4161462783813477,
      "step": 3338
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 3338,
      "training_step_time": 10.190969467163086
    },
    {
      "epoch": 1.273345947265625e-05,
      "model_forward_time": 3.1382522583007812,
      "step": 3338
    },
    {
      "epoch": 1.273345947265625e-05,
      "step": 3338,
      "training_step_time": 10.322692632675171
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 3.1547513008117676,
      "step": 3339
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 3339,
      "training_step_time": 9.93601679801941
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 3.1412510871887207,
      "step": 3339
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 3339,
      "training_step_time": 9.923927068710327
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 3.136979103088379,
      "step": 3339
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 3339,
      "training_step_time": 9.913098335266113
    },
    {
      "epoch": 1.2737274169921875e-05,
      "model_forward_time": 3.4009056091308594,
      "step": 3339
    },
    {
      "epoch": 1.2737274169921875e-05,
      "step": 3339,
      "training_step_time": 10.30410385131836
    },
    {
      "epoch": 1.27410888671875e-05,
      "grad_norm": 0.5030019879341125,
      "learning_rate": 1e-05,
      "loss": 0.0932,
      "step": 3340
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 3.1460890769958496,
      "step": 3340
    },
    {
      "dynamics_loss_avg": 0.049874990433454516,
      "epoch": 1.27410888671875e-05,
      "step": 3340
    },
    {
      "action_loss_avg": 0.025253577530384062,
      "epoch": 1.27410888671875e-05,
      "step": 3340
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 3340,
      "training_step_time": 9.927191495895386
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 3.1421902179718018,
      "step": 3340
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 3340,
      "training_step_time": 9.9196195602417
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 3.1379787921905518,
      "step": 3340
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 3340,
      "training_step_time": 9.918280839920044
    },
    {
      "epoch": 1.27410888671875e-05,
      "model_forward_time": 3.1419289112091064,
      "step": 3340
    },
    {
      "epoch": 1.27410888671875e-05,
      "step": 3340,
      "training_step_time": 10.304677963256836
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 3.4378902912139893,
      "step": 3341
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 3341,
      "training_step_time": 10.238202810287476
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 3.141528606414795,
      "step": 3341
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 3341,
      "training_step_time": 9.924690246582031
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 3.1377580165863037,
      "step": 3341
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 3341,
      "training_step_time": 9.915308237075806
    },
    {
      "epoch": 1.2744903564453126e-05,
      "model_forward_time": 3.1402552127838135,
      "step": 3341
    },
    {
      "epoch": 1.2744903564453126e-05,
      "step": 3341,
      "training_step_time": 10.33546257019043
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 3.153865337371826,
      "step": 3342
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 3342,
      "training_step_time": 9.93887996673584
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 2.4351468086242676,
      "step": 3342
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 3342,
      "training_step_time": 7.748054265975952
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 3.1344430446624756,
      "step": 3342
    },
    {
      "dynamics_loss_avg": 0.061162319779396054,
      "epoch": 1.274871826171875e-05,
      "step": 3342
    },
    {
      "action_loss_avg": 0.0340316619258374,
      "epoch": 1.274871826171875e-05,
      "step": 3342
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 3342,
      "training_step_time": 9.908770561218262
    },
    {
      "epoch": 1.274871826171875e-05,
      "model_forward_time": 3.1377170085906982,
      "step": 3342
    },
    {
      "epoch": 1.274871826171875e-05,
      "step": 3342,
      "training_step_time": 10.294361352920532
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 3.150078535079956,
      "step": 3343
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 3343,
      "training_step_time": 9.926525831222534
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 3.13787579536438,
      "step": 3343
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 3343,
      "training_step_time": 9.910545825958252
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 3.4127297401428223,
      "step": 3343
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 3343,
      "training_step_time": 10.187810182571411
    },
    {
      "epoch": 1.2752532958984375e-05,
      "model_forward_time": 3.140620470046997,
      "step": 3343
    },
    {
      "epoch": 1.2752532958984375e-05,
      "step": 3343,
      "training_step_time": 10.331213235855103
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 3.15959095954895,
      "step": 3344
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 3344,
      "training_step_time": 9.933017015457153
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 3.1326773166656494,
      "step": 3344
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 3344,
      "training_step_time": 9.908644199371338
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 3.136913537979126,
      "step": 3344
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 3344,
      "training_step_time": 9.908201694488525
    },
    {
      "epoch": 1.275634765625e-05,
      "model_forward_time": 2.42985463142395,
      "step": 3344
    },
    {
      "epoch": 1.275634765625e-05,
      "step": 3344,
      "training_step_time": 10.296270847320557
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 3.143237352371216,
      "step": 3345
    },
    {
      "dynamics_loss_avg": 0.08399397972971201,
      "epoch": 1.2760162353515626e-05,
      "step": 3345
    },
    {
      "action_loss_avg": 0.03597274678759277,
      "epoch": 1.2760162353515626e-05,
      "step": 3345
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 3345,
      "training_step_time": 9.919929027557373
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 3.1360890865325928,
      "step": 3345
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 3345,
      "training_step_time": 9.911112308502197
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 3.138364553451538,
      "step": 3345
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 3345,
      "training_step_time": 9.91661024093628
    },
    {
      "epoch": 1.2760162353515626e-05,
      "model_forward_time": 3.1359283924102783,
      "step": 3345
    },
    {
      "epoch": 1.2760162353515626e-05,
      "step": 3345,
      "training_step_time": 10.28401494026184
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 3.4330058097839355,
      "step": 3346
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 3346,
      "training_step_time": 10.212232112884521
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 3.1422667503356934,
      "step": 3346
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 3346,
      "training_step_time": 9.923392057418823
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 3.143310785293579,
      "step": 3346
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 3346,
      "training_step_time": 9.92165493965149
    },
    {
      "epoch": 1.276397705078125e-05,
      "model_forward_time": 3.1372756958007812,
      "step": 3346
    },
    {
      "epoch": 1.276397705078125e-05,
      "step": 3346,
      "training_step_time": 10.306765794754028
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 3.154587984085083,
      "step": 3347
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 3347,
      "training_step_time": 9.935421705245972
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 3.39947247505188,
      "step": 3347
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 3347,
      "training_step_time": 10.179671049118042
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 3.1399319171905518,
      "step": 3347
    },
    {
      "dynamics_loss_avg": 0.06548874415457248,
      "epoch": 1.2767791748046875e-05,
      "step": 3347
    },
    {
      "action_loss_avg": 0.03402059283107519,
      "epoch": 1.2767791748046875e-05,
      "step": 3347
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 3347,
      "training_step_time": 9.925345182418823
    },
    {
      "epoch": 1.2767791748046875e-05,
      "model_forward_time": 3.14020037651062,
      "step": 3347
    },
    {
      "epoch": 1.2767791748046875e-05,
      "step": 3347,
      "training_step_time": 10.276906967163086
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 3.1496119499206543,
      "step": 3348
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 3348,
      "training_step_time": 9.927495956420898
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 3.1394968032836914,
      "step": 3348
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 3348,
      "training_step_time": 9.91770601272583
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 3.416821241378784,
      "step": 3348
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 3348,
      "training_step_time": 10.19797682762146
    },
    {
      "epoch": 1.27716064453125e-05,
      "model_forward_time": 3.1368701457977295,
      "step": 3348
    },
    {
      "epoch": 1.27716064453125e-05,
      "step": 3348,
      "training_step_time": 10.26034140586853
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 3.143470287322998,
      "step": 3349
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 3349,
      "training_step_time": 9.929280996322632
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 3.1381804943084717,
      "step": 3349
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 3349,
      "training_step_time": 9.913522243499756
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 3.1349589824676514,
      "step": 3349
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 3349,
      "training_step_time": 9.908012866973877
    },
    {
      "epoch": 1.2775421142578126e-05,
      "model_forward_time": 3.3988912105560303,
      "step": 3349
    },
    {
      "epoch": 1.2775421142578126e-05,
      "step": 3349,
      "training_step_time": 10.294358015060425
    },
    {
      "epoch": 1.277923583984375e-05,
      "grad_norm": 0.6145015358924866,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3350
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 3.1539511680603027,
      "step": 3350
    },
    {
      "dynamics_loss_avg": 0.07202693689614534,
      "epoch": 1.277923583984375e-05,
      "step": 3350
    },
    {
      "action_loss_avg": 0.037804489675909284,
      "epoch": 1.277923583984375e-05,
      "step": 3350
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 3350,
      "training_step_time": 9.930925846099854
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 3.1394309997558594,
      "step": 3350
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 3350,
      "training_step_time": 9.911550760269165
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 3.1448605060577393,
      "step": 3350
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 3350,
      "training_step_time": 9.923064947128296
    },
    {
      "epoch": 1.277923583984375e-05,
      "model_forward_time": 3.1401984691619873,
      "step": 3350
    },
    {
      "epoch": 1.277923583984375e-05,
      "step": 3350,
      "training_step_time": 10.028754949569702
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 3.4196131229400635,
      "step": 3351
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 3351,
      "training_step_time": 10.20010232925415
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 3.145754337310791,
      "step": 3351
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 3351,
      "training_step_time": 9.925847053527832
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 3.140089750289917,
      "step": 3351
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 3351,
      "training_step_time": 9.92527151107788
    },
    {
      "epoch": 1.2783050537109375e-05,
      "model_forward_time": 3.14390230178833,
      "step": 3351
    },
    {
      "epoch": 1.2783050537109375e-05,
      "step": 3351,
      "training_step_time": 10.32039499282837
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 3.1500356197357178,
      "step": 3352
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 3352,
      "training_step_time": 9.92887544631958
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 3.4187400341033936,
      "step": 3352
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 3352,
      "training_step_time": 10.195030212402344
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 3.136864423751831,
      "step": 3352
    },
    {
      "dynamics_loss_avg": 0.0641662273555994,
      "epoch": 1.2786865234375e-05,
      "step": 3352
    },
    {
      "action_loss_avg": 0.03331888392567635,
      "epoch": 1.2786865234375e-05,
      "step": 3352
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 3352,
      "training_step_time": 9.916436910629272
    },
    {
      "epoch": 1.2786865234375e-05,
      "model_forward_time": 3.1380467414855957,
      "step": 3352
    },
    {
      "epoch": 1.2786865234375e-05,
      "step": 3352,
      "training_step_time": 10.29825210571289
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 3.147524118423462,
      "step": 3353
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 3353,
      "training_step_time": 9.934263706207275
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 3.135288953781128,
      "step": 3353
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 3353,
      "training_step_time": 9.922175645828247
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 3.143280029296875,
      "step": 3353
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 3353,
      "training_step_time": 10.230585813522339
    },
    {
      "epoch": 1.2790679931640626e-05,
      "model_forward_time": 3.143655776977539,
      "step": 3353
    },
    {
      "epoch": 1.2790679931640626e-05,
      "step": 3353,
      "training_step_time": 10.30647587776184
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 3.158719062805176,
      "step": 3354
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 3354,
      "training_step_time": 9.929109334945679
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 3.1391029357910156,
      "step": 3354
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 3354,
      "training_step_time": 9.910125017166138
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 3.143334150314331,
      "step": 3354
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 3354,
      "training_step_time": 9.915253400802612
    },
    {
      "epoch": 1.279449462890625e-05,
      "model_forward_time": 2.442110538482666,
      "step": 3354
    },
    {
      "epoch": 1.279449462890625e-05,
      "step": 3354,
      "training_step_time": 10.324991703033447
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 3.1511194705963135,
      "step": 3355
    },
    {
      "dynamics_loss_avg": 0.06157054789364338,
      "epoch": 1.2798309326171875e-05,
      "step": 3355
    },
    {
      "action_loss_avg": 0.030614100070670246,
      "epoch": 1.2798309326171875e-05,
      "step": 3355
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 3355,
      "training_step_time": 9.921771764755249
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 3.133589744567871,
      "step": 3355
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 3355,
      "training_step_time": 9.899051189422607
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 3.136439085006714,
      "step": 3355
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 3355,
      "training_step_time": 9.89783763885498
    },
    {
      "epoch": 1.2798309326171875e-05,
      "model_forward_time": 3.132668972015381,
      "step": 3355
    },
    {
      "epoch": 1.2798309326171875e-05,
      "step": 3355,
      "training_step_time": 10.010429620742798
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 3.1405553817749023,
      "step": 3356
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 3356,
      "training_step_time": 10.185474872589111
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 3.1346263885498047,
      "step": 3356
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 3356,
      "training_step_time": 9.901901006698608
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 3.1368274688720703,
      "step": 3356
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 3356,
      "training_step_time": 9.909469366073608
    },
    {
      "epoch": 1.28021240234375e-05,
      "model_forward_time": 3.1362411975860596,
      "step": 3356
    },
    {
      "epoch": 1.28021240234375e-05,
      "step": 3356,
      "training_step_time": 10.28932237625122
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 3.1556243896484375,
      "step": 3357
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 3357,
      "training_step_time": 9.926955938339233
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 3.3948590755462646,
      "step": 3357
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 3357,
      "training_step_time": 10.160605669021606
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 3.1416730880737305,
      "step": 3357
    },
    {
      "dynamics_loss_avg": 0.0676505384966731,
      "epoch": 1.2805938720703126e-05,
      "step": 3357
    },
    {
      "action_loss_avg": 0.027981604821979998,
      "epoch": 1.2805938720703126e-05,
      "step": 3357
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 3357,
      "training_step_time": 9.91446042060852
    },
    {
      "epoch": 1.2805938720703126e-05,
      "model_forward_time": 2.4350039958953857,
      "step": 3357
    },
    {
      "epoch": 1.2805938720703126e-05,
      "step": 3357,
      "training_step_time": 10.314826965332031
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 3.1532158851623535,
      "step": 3358
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 3358,
      "training_step_time": 9.922346591949463
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 3.13337779045105,
      "step": 3358
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 3358,
      "training_step_time": 9.905030250549316
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 3.131948232650757,
      "step": 3358
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 3358,
      "training_step_time": 9.898563385009766
    },
    {
      "epoch": 1.280975341796875e-05,
      "model_forward_time": 3.373678207397461,
      "step": 3358
    },
    {
      "epoch": 1.280975341796875e-05,
      "step": 3358,
      "training_step_time": 10.293270826339722
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 3.1572258472442627,
      "step": 3359
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 3359,
      "training_step_time": 9.928970575332642
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 3.133486270904541,
      "step": 3359
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 3359,
      "training_step_time": 9.90307354927063
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 3.134014129638672,
      "step": 3359
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 3359,
      "training_step_time": 9.90577244758606
    },
    {
      "epoch": 1.2813568115234375e-05,
      "model_forward_time": 3.3947014808654785,
      "step": 3359
    },
    {
      "epoch": 1.2813568115234375e-05,
      "step": 3359,
      "training_step_time": 10.28530764579773
    },
    {
      "epoch": 1.28173828125e-05,
      "grad_norm": 0.5550172924995422,
      "learning_rate": 1e-05,
      "loss": 0.0983,
      "step": 3360
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 3.1495888233184814,
      "step": 3360
    },
    {
      "dynamics_loss_avg": 0.06196577586233616,
      "epoch": 1.28173828125e-05,
      "step": 3360
    },
    {
      "action_loss_avg": 0.031204770598560572,
      "epoch": 1.28173828125e-05,
      "step": 3360
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 3360,
      "training_step_time": 9.92300009727478
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 3.139024019241333,
      "step": 3360
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 3360,
      "training_step_time": 9.910971403121948
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 3.1353330612182617,
      "step": 3360
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 3360,
      "training_step_time": 9.90765380859375
    },
    {
      "epoch": 1.28173828125e-05,
      "model_forward_time": 3.1410412788391113,
      "step": 3360
    },
    {
      "epoch": 1.28173828125e-05,
      "step": 3360,
      "training_step_time": 10.036080598831177
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 3.152791976928711,
      "step": 3361
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 3361,
      "training_step_time": 10.189655303955078
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 3.1400952339172363,
      "step": 3361
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 3361,
      "training_step_time": 9.913106679916382
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 3.1314451694488525,
      "step": 3361
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 3361,
      "training_step_time": 9.900501728057861
    },
    {
      "epoch": 1.2821197509765626e-05,
      "model_forward_time": 3.14176344871521,
      "step": 3361
    },
    {
      "epoch": 1.2821197509765626e-05,
      "step": 3361,
      "training_step_time": 10.316415548324585
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 3.1498794555664062,
      "step": 3362
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 3362,
      "training_step_time": 9.922038078308105
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 3.4170548915863037,
      "step": 3362
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 3362,
      "training_step_time": 10.187111854553223
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 3.1394541263580322,
      "step": 3362
    },
    {
      "dynamics_loss_avg": 0.04965671207755804,
      "epoch": 1.282501220703125e-05,
      "step": 3362
    },
    {
      "action_loss_avg": 0.02057342231273651,
      "epoch": 1.282501220703125e-05,
      "step": 3362
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 3362,
      "training_step_time": 9.911200284957886
    },
    {
      "epoch": 1.282501220703125e-05,
      "model_forward_time": 3.1399285793304443,
      "step": 3362
    },
    {
      "epoch": 1.282501220703125e-05,
      "step": 3362,
      "training_step_time": 10.310518503189087
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 3.151293992996216,
      "step": 3363
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 3363,
      "training_step_time": 9.924921751022339
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 2.4350228309631348,
      "step": 3363
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 3363,
      "training_step_time": 7.4616782665252686
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 3.3977925777435303,
      "step": 3363
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 3363,
      "training_step_time": 10.17071008682251
    },
    {
      "epoch": 1.2828826904296875e-05,
      "model_forward_time": 3.147273063659668,
      "step": 3363
    },
    {
      "epoch": 1.2828826904296875e-05,
      "step": 3363,
      "training_step_time": 10.330008268356323
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 3.1565065383911133,
      "step": 3364
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 3364,
      "training_step_time": 9.932649374008179
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 3.1389245986938477,
      "step": 3364
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 3364,
      "training_step_time": 9.908745527267456
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 3.139638662338257,
      "step": 3364
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 3364,
      "training_step_time": 9.911314725875854
    },
    {
      "epoch": 1.28326416015625e-05,
      "model_forward_time": 3.1384150981903076,
      "step": 3364
    },
    {
      "epoch": 1.28326416015625e-05,
      "step": 3364,
      "training_step_time": 10.277799129486084
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 3.150949478149414,
      "step": 3365
    },
    {
      "dynamics_loss_avg": 0.06891465093940496,
      "epoch": 1.2836456298828126e-05,
      "step": 3365
    },
    {
      "action_loss_avg": 0.03055845117196441,
      "epoch": 1.2836456298828126e-05,
      "step": 3365
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 3365,
      "training_step_time": 9.921501636505127
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 3.1259381771087646,
      "step": 3365
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 3365,
      "training_step_time": 9.890664339065552
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 3.12937593460083,
      "step": 3365
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 3365,
      "training_step_time": 9.892992734909058
    },
    {
      "epoch": 1.2836456298828126e-05,
      "model_forward_time": 3.1309545040130615,
      "step": 3365
    },
    {
      "epoch": 1.2836456298828126e-05,
      "step": 3365,
      "training_step_time": 10.008502006530762
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 3.4502055644989014,
      "step": 3366
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 3366,
      "training_step_time": 10.21572995185852
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 3.129940986633301,
      "step": 3366
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 3366,
      "training_step_time": 9.886428356170654
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 3.1301352977752686,
      "step": 3366
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 3366,
      "training_step_time": 9.888851881027222
    },
    {
      "epoch": 1.284027099609375e-05,
      "model_forward_time": 3.12920880317688,
      "step": 3366
    },
    {
      "epoch": 1.284027099609375e-05,
      "step": 3366,
      "training_step_time": 10.001307725906372
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 3.1480770111083984,
      "step": 3367
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 3367,
      "training_step_time": 9.908135414123535
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 3.1381735801696777,
      "step": 3367
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 3367,
      "training_step_time": 10.18639612197876
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 3.1376190185546875,
      "step": 3367
    },
    {
      "dynamics_loss_avg": 0.06177041530609131,
      "epoch": 1.2844085693359375e-05,
      "step": 3367
    },
    {
      "action_loss_avg": 0.028700876515358686,
      "epoch": 1.2844085693359375e-05,
      "step": 3367
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 3367,
      "training_step_time": 9.9079008102417
    },
    {
      "epoch": 1.2844085693359375e-05,
      "model_forward_time": 3.134885311126709,
      "step": 3367
    },
    {
      "epoch": 1.2844085693359375e-05,
      "step": 3367,
      "training_step_time": 10.267630577087402
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 3.152541160583496,
      "step": 3368
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 3368,
      "training_step_time": 9.920730590820312
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 3.1364307403564453,
      "step": 3368
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 3368,
      "training_step_time": 9.90727710723877
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 3.4157755374908447,
      "step": 3368
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 3368,
      "training_step_time": 10.186790466308594
    },
    {
      "epoch": 1.2847900390625e-05,
      "model_forward_time": 3.1416707038879395,
      "step": 3368
    },
    {
      "epoch": 1.2847900390625e-05,
      "step": 3368,
      "training_step_time": 10.281298160552979
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 2.4460229873657227,
      "step": 3369
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 3369,
      "training_step_time": 7.481383562088013
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 3.135560989379883,
      "step": 3369
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 3369,
      "training_step_time": 9.911066770553589
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 3.1356232166290283,
      "step": 3369
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 3369,
      "training_step_time": 9.920193195343018
    },
    {
      "epoch": 1.2851715087890626e-05,
      "model_forward_time": 3.141180992126465,
      "step": 3369
    },
    {
      "epoch": 1.2851715087890626e-05,
      "step": 3369,
      "training_step_time": 10.329447746276855
    },
    {
      "epoch": 1.285552978515625e-05,
      "grad_norm": 0.363817423582077,
      "learning_rate": 1e-05,
      "loss": 0.0938,
      "step": 3370
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 3.151130437850952,
      "step": 3370
    },
    {
      "dynamics_loss_avg": 0.060195001773536205,
      "epoch": 1.285552978515625e-05,
      "step": 3370
    },
    {
      "action_loss_avg": 0.029531822074204683,
      "epoch": 1.285552978515625e-05,
      "step": 3370
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 3370,
      "training_step_time": 9.941931009292603
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 3.13942813873291,
      "step": 3370
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 3370,
      "training_step_time": 9.91448187828064
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 3.1376495361328125,
      "step": 3370
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 3370,
      "training_step_time": 9.914788961410522
    },
    {
      "epoch": 1.285552978515625e-05,
      "model_forward_time": 3.1434192657470703,
      "step": 3370
    },
    {
      "epoch": 1.285552978515625e-05,
      "step": 3370,
      "training_step_time": 10.03401780128479
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 3.439316511154175,
      "step": 3371
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 3371,
      "training_step_time": 10.218969106674194
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 3.143219232559204,
      "step": 3371
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 3371,
      "training_step_time": 9.922088384628296
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 3.143411874771118,
      "step": 3371
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 3371,
      "training_step_time": 9.920467615127563
    },
    {
      "epoch": 1.2859344482421875e-05,
      "model_forward_time": 3.137977123260498,
      "step": 3371
    },
    {
      "epoch": 1.2859344482421875e-05,
      "step": 3371,
      "training_step_time": 10.008089303970337
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 3.153761386871338,
      "step": 3372
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 3372,
      "training_step_time": 9.937095642089844
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 3.4560470581054688,
      "step": 3372
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 3372,
      "training_step_time": 10.232027769088745
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 3.13744854927063,
      "step": 3372
    },
    {
      "dynamics_loss_avg": 0.0821739099919796,
      "epoch": 1.28631591796875e-05,
      "step": 3372
    },
    {
      "action_loss_avg": 0.03624071357771754,
      "epoch": 1.28631591796875e-05,
      "step": 3372
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 3372,
      "training_step_time": 9.919544696807861
    },
    {
      "epoch": 1.28631591796875e-05,
      "model_forward_time": 3.142693042755127,
      "step": 3372
    },
    {
      "epoch": 1.28631591796875e-05,
      "step": 3372,
      "training_step_time": 10.31827712059021
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 3.153458833694458,
      "step": 3373
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 3373,
      "training_step_time": 9.930057287216187
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 3.135547161102295,
      "step": 3373
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 3373,
      "training_step_time": 9.90828013420105
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 2.4347496032714844,
      "step": 3373
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 3373,
      "training_step_time": 7.7291295528411865
    },
    {
      "epoch": 1.2866973876953126e-05,
      "model_forward_time": 3.139099597930908,
      "step": 3373
    },
    {
      "epoch": 1.2866973876953126e-05,
      "step": 3373,
      "training_step_time": 10.306137800216675
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 3.155634880065918,
      "step": 3374
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 3374,
      "training_step_time": 9.936625003814697
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 3.136644124984741,
      "step": 3374
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 3374,
      "training_step_time": 9.918277978897095
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 3.1524741649627686,
      "step": 3374
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 3374,
      "training_step_time": 9.928439140319824
    },
    {
      "epoch": 1.287078857421875e-05,
      "model_forward_time": 3.440577983856201,
      "step": 3374
    },
    {
      "epoch": 1.287078857421875e-05,
      "step": 3374,
      "training_step_time": 10.308698415756226
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 3.146305799484253,
      "step": 3375
    },
    {
      "dynamics_loss_avg": 0.06578550338745118,
      "epoch": 1.2874603271484375e-05,
      "step": 3375
    },
    {
      "action_loss_avg": 0.02865854911506176,
      "epoch": 1.2874603271484375e-05,
      "step": 3375
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 3375,
      "training_step_time": 9.942085266113281
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 3.1365578174591064,
      "step": 3375
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 3375,
      "training_step_time": 9.920103788375854
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 3.1381616592407227,
      "step": 3375
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 3375,
      "training_step_time": 9.91669487953186
    },
    {
      "epoch": 1.2874603271484375e-05,
      "model_forward_time": 3.1363022327423096,
      "step": 3375
    },
    {
      "epoch": 1.2874603271484375e-05,
      "step": 3375,
      "training_step_time": 10.292655944824219
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 3.587580680847168,
      "step": 3376
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 3376,
      "training_step_time": 10.382264852523804
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 3.1473886966705322,
      "step": 3376
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 3376,
      "training_step_time": 9.931451082229614
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 3.1557960510253906,
      "step": 3376
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 3376,
      "training_step_time": 9.942277193069458
    },
    {
      "epoch": 1.287841796875e-05,
      "model_forward_time": 3.1528594493865967,
      "step": 3376
    },
    {
      "epoch": 1.287841796875e-05,
      "step": 3376,
      "training_step_time": 10.249445676803589
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 3.156773090362549,
      "step": 3377
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 3377,
      "training_step_time": 9.948659420013428
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 3.4735679626464844,
      "step": 3377
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 3377,
      "training_step_time": 10.253200054168701
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 2.440790891647339,
      "step": 3377
    },
    {
      "dynamics_loss_avg": 0.08801626712083817,
      "epoch": 1.2882232666015624e-05,
      "step": 3377
    },
    {
      "action_loss_avg": 0.0459207646548748,
      "epoch": 1.2882232666015624e-05,
      "step": 3377
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 3377,
      "training_step_time": 7.475461721420288
    },
    {
      "epoch": 1.2882232666015624e-05,
      "model_forward_time": 3.133369207382202,
      "step": 3377
    },
    {
      "epoch": 1.2882232666015624e-05,
      "step": 3377,
      "training_step_time": 10.299959421157837
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 3.1466615200042725,
      "step": 3378
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 3378,
      "training_step_time": 9.92391324043274
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 3.1375539302825928,
      "step": 3378
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 3378,
      "training_step_time": 9.912713527679443
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 3.1388742923736572,
      "step": 3378
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 3378,
      "training_step_time": 10.221650838851929
    },
    {
      "epoch": 1.288604736328125e-05,
      "model_forward_time": 3.1425859928131104,
      "step": 3378
    },
    {
      "epoch": 1.288604736328125e-05,
      "step": 3378,
      "training_step_time": 10.286312103271484
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 3.1546356678009033,
      "step": 3379
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 3379,
      "training_step_time": 9.93571949005127
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 3.1389529705047607,
      "step": 3379
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 3379,
      "training_step_time": 9.908048152923584
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 3.139244794845581,
      "step": 3379
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 3379,
      "training_step_time": 9.91693902015686
    },
    {
      "epoch": 1.2889862060546875e-05,
      "model_forward_time": 3.4292080402374268,
      "step": 3379
    },
    {
      "epoch": 1.2889862060546875e-05,
      "step": 3379,
      "training_step_time": 10.299496173858643
    },
    {
      "epoch": 1.28936767578125e-05,
      "grad_norm": 0.38555556535720825,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3380
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 3.154010057449341,
      "step": 3380
    },
    {
      "dynamics_loss_avg": 0.06950346976518632,
      "epoch": 1.28936767578125e-05,
      "step": 3380
    },
    {
      "action_loss_avg": 0.03450921420007944,
      "epoch": 1.28936767578125e-05,
      "step": 3380
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 3380,
      "training_step_time": 9.929383516311646
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 3.155244827270508,
      "step": 3380
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 3380,
      "training_step_time": 9.935529470443726
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 3.1399965286254883,
      "step": 3380
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 3380,
      "training_step_time": 9.910175085067749
    },
    {
      "epoch": 1.28936767578125e-05,
      "model_forward_time": 3.146827220916748,
      "step": 3380
    },
    {
      "epoch": 1.28936767578125e-05,
      "step": 3380,
      "training_step_time": 10.308886766433716
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 3.4390337467193604,
      "step": 3381
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 3381,
      "training_step_time": 10.219897031784058
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 3.1393649578094482,
      "step": 3381
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 3381,
      "training_step_time": 9.912013530731201
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 3.1307430267333984,
      "step": 3381
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 3381,
      "training_step_time": 9.89452338218689
    },
    {
      "epoch": 1.2897491455078124e-05,
      "model_forward_time": 3.1348485946655273,
      "step": 3381
    },
    {
      "epoch": 1.2897491455078124e-05,
      "step": 3381,
      "training_step_time": 10.009881019592285
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 3.154979705810547,
      "step": 3382
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 3382,
      "training_step_time": 9.92573618888855
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 3.1325159072875977,
      "step": 3382
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 3382,
      "training_step_time": 10.219445705413818
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 2.4339020252227783,
      "step": 3382
    },
    {
      "dynamics_loss_avg": 0.05998817551881075,
      "epoch": 1.290130615234375e-05,
      "step": 3382
    },
    {
      "action_loss_avg": 0.03401909451931715,
      "epoch": 1.290130615234375e-05,
      "step": 3382
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 3382,
      "training_step_time": 7.457110643386841
    },
    {
      "epoch": 1.290130615234375e-05,
      "model_forward_time": 3.1352384090423584,
      "step": 3382
    },
    {
      "epoch": 1.290130615234375e-05,
      "step": 3382,
      "training_step_time": 10.017411470413208
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 3.152031421661377,
      "step": 3383
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 3383,
      "training_step_time": 9.928911685943604
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 3.1462831497192383,
      "step": 3383
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 3383,
      "training_step_time": 9.9164559841156
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 3.1370511054992676,
      "step": 3383
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 3383,
      "training_step_time": 9.909846782684326
    },
    {
      "epoch": 1.2905120849609375e-05,
      "model_forward_time": 3.422466993331909,
      "step": 3383
    },
    {
      "epoch": 1.2905120849609375e-05,
      "step": 3383,
      "training_step_time": 10.311594009399414
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 3.1585800647735596,
      "step": 3384
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 3384,
      "training_step_time": 9.933443069458008
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 3.1480178833007812,
      "step": 3384
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 3384,
      "training_step_time": 9.92113471031189
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 3.1550281047821045,
      "step": 3384
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 3384,
      "training_step_time": 9.92984914779663
    },
    {
      "epoch": 1.2908935546875e-05,
      "model_forward_time": 3.1494269371032715,
      "step": 3384
    },
    {
      "epoch": 1.2908935546875e-05,
      "step": 3384,
      "training_step_time": 10.33984923362732
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 3.160620927810669,
      "step": 3385
    },
    {
      "dynamics_loss_avg": 0.0700859684497118,
      "epoch": 1.2912750244140624e-05,
      "step": 3385
    },
    {
      "action_loss_avg": 0.03257762575522065,
      "epoch": 1.2912750244140624e-05,
      "step": 3385
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 3385,
      "training_step_time": 9.946295976638794
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 2.4438836574554443,
      "step": 3385
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 3385,
      "training_step_time": 7.480161666870117
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 3.1559815406799316,
      "step": 3385
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 3385,
      "training_step_time": 9.928200006484985
    },
    {
      "epoch": 1.2912750244140624e-05,
      "model_forward_time": 3.1493189334869385,
      "step": 3385
    },
    {
      "epoch": 1.2912750244140624e-05,
      "step": 3385,
      "training_step_time": 10.275691509246826
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 3.449061393737793,
      "step": 3386
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 3386,
      "training_step_time": 10.222123622894287
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 3.142895221710205,
      "step": 3386
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 3386,
      "training_step_time": 9.918703079223633
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 3.1456165313720703,
      "step": 3386
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 3386,
      "training_step_time": 9.919942617416382
    },
    {
      "epoch": 1.291656494140625e-05,
      "model_forward_time": 3.1437714099884033,
      "step": 3386
    },
    {
      "epoch": 1.291656494140625e-05,
      "step": 3386,
      "training_step_time": 10.253928184509277
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 3.1605889797210693,
      "step": 3387
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 3387,
      "training_step_time": 9.936923742294312
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 3.1470072269439697,
      "step": 3387
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 3387,
      "training_step_time": 10.18812084197998
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 3.149625062942505,
      "step": 3387
    },
    {
      "dynamics_loss_avg": 0.07219245955348015,
      "epoch": 1.2920379638671875e-05,
      "step": 3387
    },
    {
      "action_loss_avg": 0.03622381901368499,
      "epoch": 1.2920379638671875e-05,
      "step": 3387
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 3387,
      "training_step_time": 9.929543495178223
    },
    {
      "epoch": 1.2920379638671875e-05,
      "model_forward_time": 3.1477136611938477,
      "step": 3387
    },
    {
      "epoch": 1.2920379638671875e-05,
      "step": 3387,
      "training_step_time": 10.025484561920166
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 3.1499109268188477,
      "step": 3388
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 3388,
      "training_step_time": 9.926731824874878
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 3.1464200019836426,
      "step": 3388
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 3388,
      "training_step_time": 9.92047119140625
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 3.422022581100464,
      "step": 3388
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 3388,
      "training_step_time": 10.197673559188843
    },
    {
      "epoch": 1.29241943359375e-05,
      "model_forward_time": 3.141664743423462,
      "step": 3388
    },
    {
      "epoch": 1.29241943359375e-05,
      "step": 3388,
      "training_step_time": 10.283873319625854
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 3.15264630317688,
      "step": 3389
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 3389,
      "training_step_time": 9.927576065063477
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 3.1461679935455322,
      "step": 3389
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 3389,
      "training_step_time": 9.918784856796265
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 3.1382336616516113,
      "step": 3389
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 3389,
      "training_step_time": 9.90910267829895
    },
    {
      "epoch": 1.2928009033203124e-05,
      "model_forward_time": 2.4391708374023438,
      "step": 3389
    },
    {
      "epoch": 1.2928009033203124e-05,
      "step": 3389,
      "training_step_time": 10.285402059555054
    },
    {
      "epoch": 1.293182373046875e-05,
      "grad_norm": 0.51474529504776,
      "learning_rate": 1e-05,
      "loss": 0.1022,
      "step": 3390
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 3.1569929122924805,
      "step": 3390
    },
    {
      "dynamics_loss_avg": 0.07971039842814207,
      "epoch": 1.293182373046875e-05,
      "step": 3390
    },
    {
      "action_loss_avg": 0.034361854288727046,
      "epoch": 1.293182373046875e-05,
      "step": 3390
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 3390,
      "training_step_time": 9.932013988494873
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 3.1436917781829834,
      "step": 3390
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 3390,
      "training_step_time": 9.913774251937866
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 3.1384925842285156,
      "step": 3390
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 3390,
      "training_step_time": 9.90698528289795
    },
    {
      "epoch": 1.293182373046875e-05,
      "model_forward_time": 3.136207342147827,
      "step": 3390
    },
    {
      "epoch": 1.293182373046875e-05,
      "step": 3390,
      "training_step_time": 10.008908987045288
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 3.440019369125366,
      "step": 3391
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 3391,
      "training_step_time": 10.21076774597168
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 3.1397788524627686,
      "step": 3391
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 3391,
      "training_step_time": 9.90336298942566
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 3.1441800594329834,
      "step": 3391
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 3391,
      "training_step_time": 9.90863585472107
    },
    {
      "epoch": 1.2935638427734375e-05,
      "model_forward_time": 3.1406028270721436,
      "step": 3391
    },
    {
      "epoch": 1.2935638427734375e-05,
      "step": 3391,
      "training_step_time": 10.25082516670227
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 3.151686191558838,
      "step": 3392
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 3392,
      "training_step_time": 9.91239857673645
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 3.4288506507873535,
      "step": 3392
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 3392,
      "training_step_time": 10.195044040679932
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 3.1395390033721924,
      "step": 3392
    },
    {
      "dynamics_loss_avg": 0.05890843700617552,
      "epoch": 1.2939453125e-05,
      "step": 3392
    },
    {
      "action_loss_avg": 0.02704286193475127,
      "epoch": 1.2939453125e-05,
      "step": 3392
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 3392,
      "training_step_time": 9.914592027664185
    },
    {
      "epoch": 1.2939453125e-05,
      "model_forward_time": 3.1426284313201904,
      "step": 3392
    },
    {
      "epoch": 1.2939453125e-05,
      "step": 3392,
      "training_step_time": 10.294069528579712
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 3.1537768840789795,
      "step": 3393
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 3393,
      "training_step_time": 9.92215347290039
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 3.1397621631622314,
      "step": 3393
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 3393,
      "training_step_time": 9.910848379135132
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 3.144212484359741,
      "step": 3393
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 3393,
      "training_step_time": 10.196439981460571
    },
    {
      "epoch": 1.2943267822265624e-05,
      "model_forward_time": 3.1426196098327637,
      "step": 3393
    },
    {
      "epoch": 1.2943267822265624e-05,
      "step": 3393,
      "training_step_time": 10.31571912765503
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 3.1559717655181885,
      "step": 3394
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 3394,
      "training_step_time": 9.93183183670044
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 3.138103485107422,
      "step": 3394
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 3394,
      "training_step_time": 9.907886505126953
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 3.1428329944610596,
      "step": 3394
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 3394,
      "training_step_time": 9.91235876083374
    },
    {
      "epoch": 1.294708251953125e-05,
      "model_forward_time": 3.4235167503356934,
      "step": 3394
    },
    {
      "epoch": 1.294708251953125e-05,
      "step": 3394,
      "training_step_time": 10.285568237304688
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 3.156766891479492,
      "step": 3395
    },
    {
      "dynamics_loss_avg": 0.07413059454411268,
      "epoch": 1.2950897216796875e-05,
      "step": 3395
    },
    {
      "action_loss_avg": 0.03503319220617414,
      "epoch": 1.2950897216796875e-05,
      "step": 3395
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 3395,
      "training_step_time": 9.92931318283081
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 3.138943672180176,
      "step": 3395
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 3395,
      "training_step_time": 9.909603118896484
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 3.1442928314208984,
      "step": 3395
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 3395,
      "training_step_time": 9.915475606918335
    },
    {
      "epoch": 1.2950897216796875e-05,
      "model_forward_time": 3.143507480621338,
      "step": 3395
    },
    {
      "epoch": 1.2950897216796875e-05,
      "step": 3395,
      "training_step_time": 10.299535751342773
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 3.152165412902832,
      "step": 3396
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 3396,
      "training_step_time": 10.196435928344727
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 3.1425206661224365,
      "step": 3396
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 3396,
      "training_step_time": 9.912060976028442
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 3.1445510387420654,
      "step": 3396
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 3396,
      "training_step_time": 9.917238235473633
    },
    {
      "epoch": 1.29547119140625e-05,
      "model_forward_time": 3.13822340965271,
      "step": 3396
    },
    {
      "epoch": 1.29547119140625e-05,
      "step": 3396,
      "training_step_time": 10.010744333267212
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 3.1556196212768555,
      "step": 3397
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 3397,
      "training_step_time": 9.92654824256897
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 3.3931498527526855,
      "step": 3397
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 3397,
      "training_step_time": 10.163235902786255
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 3.1444015502929688,
      "step": 3397
    },
    {
      "dynamics_loss_avg": 0.07131660114973784,
      "epoch": 1.2958526611328124e-05,
      "step": 3397
    },
    {
      "action_loss_avg": 0.033483909396454695,
      "epoch": 1.2958526611328124e-05,
      "step": 3397
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 3397,
      "training_step_time": 9.917511224746704
    },
    {
      "epoch": 1.2958526611328124e-05,
      "model_forward_time": 3.1384198665618896,
      "step": 3397
    },
    {
      "epoch": 1.2958526611328124e-05,
      "step": 3397,
      "training_step_time": 10.269357204437256
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 3.1553938388824463,
      "step": 3398
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 3398,
      "training_step_time": 9.931777954101562
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 3.141441822052002,
      "step": 3398
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 3398,
      "training_step_time": 9.914989233016968
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 3.4579970836639404,
      "step": 3398
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 3398,
      "training_step_time": 10.227256298065186
    },
    {
      "epoch": 1.296234130859375e-05,
      "model_forward_time": 3.1396608352661133,
      "step": 3398
    },
    {
      "epoch": 1.296234130859375e-05,
      "step": 3398,
      "training_step_time": 10.024696111679077
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 3.149622917175293,
      "step": 3399
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 3399,
      "training_step_time": 9.921183109283447
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 3.138439655303955,
      "step": 3399
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 3399,
      "training_step_time": 9.910105228424072
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 3.1398699283599854,
      "step": 3399
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 3399,
      "training_step_time": 9.907073974609375
    },
    {
      "epoch": 1.2966156005859375e-05,
      "model_forward_time": 3.4025323390960693,
      "step": 3399
    },
    {
      "epoch": 1.2966156005859375e-05,
      "step": 3399,
      "training_step_time": 10.261480808258057
    },
    {
      "epoch": 1.2969970703125e-05,
      "grad_norm": 0.5263925790786743,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 3400
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 3.1405112743377686,
      "step": 3400
    },
    {
      "dynamics_loss_avg": 0.06606476102024317,
      "epoch": 1.2969970703125e-05,
      "step": 3400
    },
    {
      "action_loss_avg": 0.04027212089858949,
      "epoch": 1.2969970703125e-05,
      "step": 3400
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 3400,
      "training_step_time": 9.906673908233643
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 3.1407716274261475,
      "step": 3400
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 3400,
      "training_step_time": 9.90449857711792
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 3.1348166465759277,
      "step": 3400
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 3400,
      "training_step_time": 9.898610591888428
    },
    {
      "epoch": 1.2969970703125e-05,
      "model_forward_time": 3.139012098312378,
      "step": 3400
    },
    {
      "epoch": 1.2969970703125e-05,
      "step": 3400,
      "training_step_time": 10.268490314483643
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 3.4052765369415283,
      "step": 3401
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 3401,
      "training_step_time": 10.173581838607788
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 3.136645793914795,
      "step": 3401
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 3401,
      "training_step_time": 9.902551412582397
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 3.1352078914642334,
      "step": 3401
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 3401,
      "training_step_time": 9.898539066314697
    },
    {
      "epoch": 1.2973785400390624e-05,
      "model_forward_time": 3.1404929161071777,
      "step": 3401
    },
    {
      "epoch": 1.2973785400390624e-05,
      "step": 3401,
      "training_step_time": 10.017616510391235
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 3.150834083557129,
      "step": 3402
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 3402,
      "training_step_time": 9.915523290634155
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 3.135427951812744,
      "step": 3402
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 3402,
      "training_step_time": 10.182428121566772
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 3.1363325119018555,
      "step": 3402
    },
    {
      "dynamics_loss_avg": 0.07104402482509613,
      "epoch": 1.297760009765625e-05,
      "step": 3402
    },
    {
      "action_loss_avg": 0.0302700849249959,
      "epoch": 1.297760009765625e-05,
      "step": 3402
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 3402,
      "training_step_time": 9.908342361450195
    },
    {
      "epoch": 1.297760009765625e-05,
      "model_forward_time": 3.1406610012054443,
      "step": 3402
    },
    {
      "epoch": 1.297760009765625e-05,
      "step": 3402,
      "training_step_time": 10.028809070587158
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 3.1605064868927,
      "step": 3403
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 3403,
      "training_step_time": 9.9340980052948
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 3.1372017860412598,
      "step": 3403
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 3403,
      "training_step_time": 9.91016173362732
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 2.4353981018066406,
      "step": 3403
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 3403,
      "training_step_time": 7.462940454483032
    },
    {
      "epoch": 1.2981414794921875e-05,
      "model_forward_time": 3.407339572906494,
      "step": 3403
    },
    {
      "epoch": 1.2981414794921875e-05,
      "step": 3403,
      "training_step_time": 10.312769889831543
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 3.1488025188446045,
      "step": 3404
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 3404,
      "training_step_time": 9.924755573272705
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 3.14162540435791,
      "step": 3404
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 3404,
      "training_step_time": 9.913453340530396
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 3.138291835784912,
      "step": 3404
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 3404,
      "training_step_time": 9.907169818878174
    },
    {
      "epoch": 1.29852294921875e-05,
      "model_forward_time": 3.135645627975464,
      "step": 3404
    },
    {
      "epoch": 1.29852294921875e-05,
      "step": 3404,
      "training_step_time": 10.264408111572266
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 3.149071455001831,
      "step": 3405
    },
    {
      "dynamics_loss_avg": 0.05177941434085369,
      "epoch": 1.2989044189453124e-05,
      "step": 3405
    },
    {
      "action_loss_avg": 0.022745833732187748,
      "epoch": 1.2989044189453124e-05,
      "step": 3405
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 3405,
      "training_step_time": 9.924199342727661
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 3.140504837036133,
      "step": 3405
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 3405,
      "training_step_time": 9.919748306274414
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 3.1435067653656006,
      "step": 3405
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 3405,
      "training_step_time": 9.923712730407715
    },
    {
      "epoch": 1.2989044189453124e-05,
      "model_forward_time": 3.1383237838745117,
      "step": 3405
    },
    {
      "epoch": 1.2989044189453124e-05,
      "step": 3405,
      "training_step_time": 10.322674036026001
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 3.428271770477295,
      "step": 3406
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 3406,
      "training_step_time": 10.208043098449707
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 3.1362695693969727,
      "step": 3406
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 3406,
      "training_step_time": 9.908447504043579
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 3.133800506591797,
      "step": 3406
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 3406,
      "training_step_time": 9.906771421432495
    },
    {
      "epoch": 1.299285888671875e-05,
      "model_forward_time": 3.1379926204681396,
      "step": 3406
    },
    {
      "epoch": 1.299285888671875e-05,
      "step": 3406,
      "training_step_time": 10.290136098861694
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 3.15862774848938,
      "step": 3407
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 3407,
      "training_step_time": 9.930088520050049
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 3.1336333751678467,
      "step": 3407
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 3407,
      "training_step_time": 10.173701763153076
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 3.138277530670166,
      "step": 3407
    },
    {
      "dynamics_loss_avg": 0.06316972207278013,
      "epoch": 1.2996673583984375e-05,
      "step": 3407
    },
    {
      "action_loss_avg": 0.03691277354955673,
      "epoch": 1.2996673583984375e-05,
      "step": 3407
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 3407,
      "training_step_time": 9.911033868789673
    },
    {
      "epoch": 1.2996673583984375e-05,
      "model_forward_time": 3.1361284255981445,
      "step": 3407
    },
    {
      "epoch": 1.2996673583984375e-05,
      "step": 3407,
      "training_step_time": 10.280843734741211
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 3.166234254837036,
      "step": 3408
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 3408,
      "training_step_time": 9.951279640197754
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 3.1413474082946777,
      "step": 3408
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 3408,
      "training_step_time": 9.914347171783447
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 3.447122812271118,
      "step": 3408
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 3408,
      "training_step_time": 10.224170923233032
    },
    {
      "epoch": 1.300048828125e-05,
      "model_forward_time": 2.4384195804595947,
      "step": 3408
    },
    {
      "epoch": 1.300048828125e-05,
      "step": 3408,
      "training_step_time": 10.245729446411133
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 3.1534624099731445,
      "step": 3409
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 3409,
      "training_step_time": 9.928550720214844
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 3.144537925720215,
      "step": 3409
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 3409,
      "training_step_time": 9.929016828536987
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 3.1385178565979004,
      "step": 3409
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 3409,
      "training_step_time": 9.9197518825531
    },
    {
      "epoch": 1.3004302978515624e-05,
      "model_forward_time": 3.1405529975891113,
      "step": 3409
    },
    {
      "epoch": 1.3004302978515624e-05,
      "step": 3409,
      "training_step_time": 10.294498443603516
    },
    {
      "epoch": 1.300811767578125e-05,
      "grad_norm": 0.6921606659889221,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 3410
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 3.1577842235565186,
      "step": 3410
    },
    {
      "dynamics_loss_avg": 0.058339134231209754,
      "epoch": 1.300811767578125e-05,
      "step": 3410
    },
    {
      "action_loss_avg": 0.024882517149671914,
      "epoch": 1.300811767578125e-05,
      "step": 3410
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 3410,
      "training_step_time": 9.939698219299316
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 3.1407341957092285,
      "step": 3410
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 3410,
      "training_step_time": 9.91507077217102
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 3.1368680000305176,
      "step": 3410
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 3410,
      "training_step_time": 9.909989595413208
    },
    {
      "epoch": 1.300811767578125e-05,
      "model_forward_time": 3.1411805152893066,
      "step": 3410
    },
    {
      "epoch": 1.300811767578125e-05,
      "step": 3410,
      "training_step_time": 10.28047776222229
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 3.3971481323242188,
      "step": 3411
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 3411,
      "training_step_time": 10.1717050075531
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 2.4323885440826416,
      "step": 3411
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 3411,
      "training_step_time": 7.465481996536255
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 3.1313841342926025,
      "step": 3411
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 3411,
      "training_step_time": 9.912981033325195
    },
    {
      "epoch": 1.3011932373046875e-05,
      "model_forward_time": 3.1352243423461914,
      "step": 3411
    },
    {
      "epoch": 1.3011932373046875e-05,
      "step": 3411,
      "training_step_time": 10.262782335281372
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 3.1506540775299072,
      "step": 3412
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 3412,
      "training_step_time": 9.929748773574829
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 3.415698528289795,
      "step": 3412
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 3412,
      "training_step_time": 10.196460247039795
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 3.13154673576355,
      "step": 3412
    },
    {
      "dynamics_loss_avg": 0.07102391906082631,
      "epoch": 1.30157470703125e-05,
      "step": 3412
    },
    {
      "action_loss_avg": 0.052920233365148305,
      "epoch": 1.30157470703125e-05,
      "step": 3412
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 3412,
      "training_step_time": 9.910298109054565
    },
    {
      "epoch": 1.30157470703125e-05,
      "model_forward_time": 2.4359395503997803,
      "step": 3412
    },
    {
      "epoch": 1.30157470703125e-05,
      "step": 3412,
      "training_step_time": 10.309759378433228
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 3.1509876251220703,
      "step": 3413
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 3413,
      "training_step_time": 9.92922592163086
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 3.1315555572509766,
      "step": 3413
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 3413,
      "training_step_time": 9.905069828033447
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 3.134990692138672,
      "step": 3413
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 3413,
      "training_step_time": 10.181576251983643
    },
    {
      "epoch": 1.3019561767578124e-05,
      "model_forward_time": 3.133702516555786,
      "step": 3413
    },
    {
      "epoch": 1.3019561767578124e-05,
      "step": 3413,
      "training_step_time": 10.013856649398804
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 3.1583945751190186,
      "step": 3414
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 3414,
      "training_step_time": 9.935315370559692
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 3.1381280422210693,
      "step": 3414
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 3414,
      "training_step_time": 9.910656213760376
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 3.138392925262451,
      "step": 3414
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 3414,
      "training_step_time": 9.9101402759552
    },
    {
      "epoch": 1.302337646484375e-05,
      "model_forward_time": 3.419618606567383,
      "step": 3414
    },
    {
      "epoch": 1.302337646484375e-05,
      "step": 3414,
      "training_step_time": 10.292427778244019
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 3.161508321762085,
      "step": 3415
    },
    {
      "dynamics_loss_avg": 0.04782312549650669,
      "epoch": 1.3027191162109375e-05,
      "step": 3415
    },
    {
      "action_loss_avg": 0.020743020251393317,
      "epoch": 1.3027191162109375e-05,
      "step": 3415
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 3415,
      "training_step_time": 9.94731855392456
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 3.1389102935791016,
      "step": 3415
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 3415,
      "training_step_time": 9.910062074661255
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 3.1380860805511475,
      "step": 3415
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 3415,
      "training_step_time": 9.912397384643555
    },
    {
      "epoch": 1.3027191162109375e-05,
      "model_forward_time": 3.1349828243255615,
      "step": 3415
    },
    {
      "epoch": 1.3027191162109375e-05,
      "step": 3415,
      "training_step_time": 10.299319505691528
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 3.1446778774261475,
      "step": 3416
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 3416,
      "training_step_time": 10.192304611206055
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 2.4351115226745605,
      "step": 3416
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 3416,
      "training_step_time": 7.468376398086548
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 3.141664981842041,
      "step": 3416
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 3416,
      "training_step_time": 9.919536352157593
    },
    {
      "epoch": 1.3031005859375e-05,
      "model_forward_time": 3.1379270553588867,
      "step": 3416
    },
    {
      "epoch": 1.3031005859375e-05,
      "step": 3416,
      "training_step_time": 10.309065818786621
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 3.149728775024414,
      "step": 3417
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 3417,
      "training_step_time": 9.931275844573975
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 3.404247283935547,
      "step": 3417
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 3417,
      "training_step_time": 10.177879571914673
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 3.139212131500244,
      "step": 3417
    },
    {
      "dynamics_loss_avg": 0.07239085771143436,
      "epoch": 1.3034820556640624e-05,
      "step": 3417
    },
    {
      "action_loss_avg": 0.03922813748940825,
      "epoch": 1.3034820556640624e-05,
      "step": 3417
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 3417,
      "training_step_time": 9.918700218200684
    },
    {
      "epoch": 1.3034820556640624e-05,
      "model_forward_time": 3.1374402046203613,
      "step": 3417
    },
    {
      "epoch": 1.3034820556640624e-05,
      "step": 3417,
      "training_step_time": 10.282389402389526
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 3.1396560668945312,
      "step": 3418
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 3418,
      "training_step_time": 9.91690444946289
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 3.140425205230713,
      "step": 3418
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 3418,
      "training_step_time": 9.911168336868286
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 3.1359503269195557,
      "step": 3418
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 3418,
      "training_step_time": 10.169173955917358
    },
    {
      "epoch": 1.303863525390625e-05,
      "model_forward_time": 3.1346094608306885,
      "step": 3418
    },
    {
      "epoch": 1.303863525390625e-05,
      "step": 3418,
      "training_step_time": 10.017656564712524
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 3.150944709777832,
      "step": 3419
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 3419,
      "training_step_time": 9.93235182762146
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 3.1414995193481445,
      "step": 3419
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 3419,
      "training_step_time": 9.923824787139893
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 3.140894651412964,
      "step": 3419
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 3419,
      "training_step_time": 9.92165732383728
    },
    {
      "epoch": 1.3042449951171875e-05,
      "model_forward_time": 3.391957998275757,
      "step": 3419
    },
    {
      "epoch": 1.3042449951171875e-05,
      "step": 3419,
      "training_step_time": 10.271510124206543
    },
    {
      "epoch": 1.30462646484375e-05,
      "grad_norm": 0.4580574035644531,
      "learning_rate": 1e-05,
      "loss": 0.0983,
      "step": 3420
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 3.1519274711608887,
      "step": 3420
    },
    {
      "dynamics_loss_avg": 0.05928159710019827,
      "epoch": 1.30462646484375e-05,
      "step": 3420
    },
    {
      "action_loss_avg": 0.033939804509282114,
      "epoch": 1.30462646484375e-05,
      "step": 3420
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 3420,
      "training_step_time": 9.934275150299072
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 3.144463062286377,
      "step": 3420
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 3420,
      "training_step_time": 9.922620058059692
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 3.139753580093384,
      "step": 3420
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 3420,
      "training_step_time": 9.918854713439941
    },
    {
      "epoch": 1.30462646484375e-05,
      "model_forward_time": 1.7453575134277344,
      "step": 3420
    },
    {
      "epoch": 1.30462646484375e-05,
      "step": 3420,
      "training_step_time": 10.279979467391968
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 3.1440157890319824,
      "step": 3421
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 3421,
      "training_step_time": 10.188371896743774
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 3.138155221939087,
      "step": 3421
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 3421,
      "training_step_time": 9.907554626464844
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 3.132627248764038,
      "step": 3421
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 3421,
      "training_step_time": 9.901163339614868
    },
    {
      "epoch": 1.3050079345703124e-05,
      "model_forward_time": 3.1376845836639404,
      "step": 3421
    },
    {
      "epoch": 1.3050079345703124e-05,
      "step": 3421,
      "training_step_time": 10.241221189498901
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 3.1501824855804443,
      "step": 3422
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 3422,
      "training_step_time": 9.92288088798523
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 3.1348533630371094,
      "step": 3422
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 3422,
      "training_step_time": 10.19316577911377
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 3.136868476867676,
      "step": 3422
    },
    {
      "dynamics_loss_avg": 0.06369037758558989,
      "epoch": 1.305389404296875e-05,
      "step": 3422
    },
    {
      "action_loss_avg": 0.030901972111314534,
      "epoch": 1.305389404296875e-05,
      "step": 3422
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 3422,
      "training_step_time": 9.907521963119507
    },
    {
      "epoch": 1.305389404296875e-05,
      "model_forward_time": 3.136352300643921,
      "step": 3422
    },
    {
      "epoch": 1.305389404296875e-05,
      "step": 3422,
      "training_step_time": 10.027791500091553
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 3.1524198055267334,
      "step": 3423
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 3423,
      "training_step_time": 9.928251504898071
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 3.131965398788452,
      "step": 3423
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 3423,
      "training_step_time": 9.906972885131836
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 3.405000925064087,
      "step": 3423
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 3423,
      "training_step_time": 10.172115087509155
    },
    {
      "epoch": 1.3057708740234375e-05,
      "model_forward_time": 3.137335777282715,
      "step": 3423
    },
    {
      "epoch": 1.3057708740234375e-05,
      "step": 3423,
      "training_step_time": 10.330142974853516
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 3.1420557498931885,
      "step": 3424
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 3424,
      "training_step_time": 9.904447555541992
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 3.1320960521698,
      "step": 3424
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 3424,
      "training_step_time": 9.895248174667358
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 3.131516218185425,
      "step": 3424
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 3424,
      "training_step_time": 9.893169164657593
    },
    {
      "epoch": 1.30615234375e-05,
      "model_forward_time": 3.132845401763916,
      "step": 3424
    },
    {
      "epoch": 1.30615234375e-05,
      "step": 3424,
      "training_step_time": 10.33027195930481
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 3.1492276191711426,
      "step": 3425
    },
    {
      "dynamics_loss_avg": 0.06416138503700494,
      "epoch": 1.3065338134765624e-05,
      "step": 3425
    },
    {
      "action_loss_avg": 0.028260675398632883,
      "epoch": 1.3065338134765624e-05,
      "step": 3425
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 3425,
      "training_step_time": 9.918404817581177
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 3.1326096057891846,
      "step": 3425
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 3425,
      "training_step_time": 9.897390604019165
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 3.1400551795959473,
      "step": 3425
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 3425,
      "training_step_time": 9.908156871795654
    },
    {
      "epoch": 1.3065338134765624e-05,
      "model_forward_time": 3.1367151737213135,
      "step": 3425
    },
    {
      "epoch": 1.3065338134765624e-05,
      "step": 3425,
      "training_step_time": 10.314298629760742
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 3.433248996734619,
      "step": 3426
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 3426,
      "training_step_time": 10.206122875213623
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 3.136505126953125,
      "step": 3426
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 3426,
      "training_step_time": 9.91324520111084
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 3.137742042541504,
      "step": 3426
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 3426,
      "training_step_time": 9.908422231674194
    },
    {
      "epoch": 1.306915283203125e-05,
      "model_forward_time": 3.1386702060699463,
      "step": 3426
    },
    {
      "epoch": 1.306915283203125e-05,
      "step": 3426,
      "training_step_time": 10.270253419876099
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 3.152742385864258,
      "step": 3427
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 3427,
      "training_step_time": 9.919131994247437
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 3.3873953819274902,
      "step": 3427
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 3427,
      "training_step_time": 10.154181480407715
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 3.1353037357330322,
      "step": 3427
    },
    {
      "dynamics_loss_avg": 0.06310251615941524,
      "epoch": 1.3072967529296875e-05,
      "step": 3427
    },
    {
      "action_loss_avg": 0.028860871307551862,
      "epoch": 1.3072967529296875e-05,
      "step": 3427
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 3427,
      "training_step_time": 9.907760381698608
    },
    {
      "epoch": 1.3072967529296875e-05,
      "model_forward_time": 3.1375439167022705,
      "step": 3427
    },
    {
      "epoch": 1.3072967529296875e-05,
      "step": 3427,
      "training_step_time": 10.275063514709473
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 3.155397653579712,
      "step": 3428
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 3428,
      "training_step_time": 9.925719499588013
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 3.141009569168091,
      "step": 3428
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 3428,
      "training_step_time": 9.911368131637573
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 3.415095567703247,
      "step": 3428
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 3428,
      "training_step_time": 10.183431625366211
    },
    {
      "epoch": 1.30767822265625e-05,
      "model_forward_time": 3.1401710510253906,
      "step": 3428
    },
    {
      "epoch": 1.30767822265625e-05,
      "step": 3428,
      "training_step_time": 10.286540746688843
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 3.160754442214966,
      "step": 3429
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 3429,
      "training_step_time": 9.937906503677368
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 3.1433138847351074,
      "step": 3429
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 3429,
      "training_step_time": 9.913439989089966
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 3.1388936042785645,
      "step": 3429
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 3429,
      "training_step_time": 9.906615257263184
    },
    {
      "epoch": 1.3080596923828124e-05,
      "model_forward_time": 3.395493507385254,
      "step": 3429
    },
    {
      "epoch": 1.3080596923828124e-05,
      "step": 3429,
      "training_step_time": 10.291248559951782
    },
    {
      "epoch": 1.308441162109375e-05,
      "grad_norm": 0.40371713042259216,
      "learning_rate": 1e-05,
      "loss": 0.0972,
      "step": 3430
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 3.154141664505005,
      "step": 3430
    },
    {
      "dynamics_loss_avg": 0.05800334382802248,
      "epoch": 1.308441162109375e-05,
      "step": 3430
    },
    {
      "action_loss_avg": 0.030219065211713313,
      "epoch": 1.308441162109375e-05,
      "step": 3430
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 3430,
      "training_step_time": 9.92906641960144
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 3.136359930038452,
      "step": 3430
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 3430,
      "training_step_time": 9.901267051696777
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 3.1385157108306885,
      "step": 3430
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 3430,
      "training_step_time": 9.910739660263062
    },
    {
      "epoch": 1.308441162109375e-05,
      "model_forward_time": 3.1412699222564697,
      "step": 3430
    },
    {
      "epoch": 1.308441162109375e-05,
      "step": 3430,
      "training_step_time": 10.248482704162598
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 3.4101815223693848,
      "step": 3431
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 3431,
      "training_step_time": 10.188233852386475
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 3.1427700519561768,
      "step": 3431
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 3431,
      "training_step_time": 9.911465883255005
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 3.1348330974578857,
      "step": 3431
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 3431,
      "training_step_time": 9.897307872772217
    },
    {
      "epoch": 1.3088226318359375e-05,
      "model_forward_time": 3.1405229568481445,
      "step": 3431
    },
    {
      "epoch": 1.3088226318359375e-05,
      "step": 3431,
      "training_step_time": 10.25801706314087
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 3.1466763019561768,
      "step": 3432
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 3432,
      "training_step_time": 9.915358543395996
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 3.412754774093628,
      "step": 3432
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 3432,
      "training_step_time": 10.177269220352173
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 3.13388991355896,
      "step": 3432
    },
    {
      "dynamics_loss_avg": 0.07129048872739077,
      "epoch": 1.3092041015625e-05,
      "step": 3432
    },
    {
      "action_loss_avg": 0.029399016965180637,
      "epoch": 1.3092041015625e-05,
      "step": 3432
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 3432,
      "training_step_time": 9.902549982070923
    },
    {
      "epoch": 1.3092041015625e-05,
      "model_forward_time": 2.4302539825439453,
      "step": 3432
    },
    {
      "epoch": 1.3092041015625e-05,
      "step": 3432,
      "training_step_time": 10.292137384414673
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 3.145444631576538,
      "step": 3433
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 3433,
      "training_step_time": 9.909804582595825
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 3.135326623916626,
      "step": 3433
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 3433,
      "training_step_time": 9.898074388504028
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 3.3928534984588623,
      "step": 3433
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 3433,
      "training_step_time": 10.150441646575928
    },
    {
      "epoch": 1.3095855712890624e-05,
      "model_forward_time": 3.1365020275115967,
      "step": 3433
    },
    {
      "epoch": 1.3095855712890624e-05,
      "step": 3433,
      "training_step_time": 10.29210114479065
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 3.1473867893218994,
      "step": 3434
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 3434,
      "training_step_time": 9.915920972824097
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 3.1298909187316895,
      "step": 3434
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 3434,
      "training_step_time": 9.893701076507568
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 2.431288242340088,
      "step": 3434
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 3434,
      "training_step_time": 7.450062274932861
    },
    {
      "epoch": 1.309967041015625e-05,
      "model_forward_time": 3.429978609085083,
      "step": 3434
    },
    {
      "epoch": 1.309967041015625e-05,
      "step": 3434,
      "training_step_time": 10.286287546157837
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 3.147822141647339,
      "step": 3435
    },
    {
      "dynamics_loss_avg": 0.08589686229825019,
      "epoch": 1.3103485107421875e-05,
      "step": 3435
    },
    {
      "action_loss_avg": 0.03911057524383068,
      "epoch": 1.3103485107421875e-05,
      "step": 3435
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 3435,
      "training_step_time": 9.932765007019043
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 3.134244203567505,
      "step": 3435
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 3435,
      "training_step_time": 9.904234886169434
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 3.14011287689209,
      "step": 3435
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 3435,
      "training_step_time": 9.90891432762146
    },
    {
      "epoch": 1.3103485107421875e-05,
      "model_forward_time": 3.1381406784057617,
      "step": 3435
    },
    {
      "epoch": 1.3103485107421875e-05,
      "step": 3435,
      "training_step_time": 10.318141460418701
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 3.148676872253418,
      "step": 3436
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 3436,
      "training_step_time": 10.227596044540405
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 3.139784574508667,
      "step": 3436
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 3436,
      "training_step_time": 9.91367793083191
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 3.1425986289978027,
      "step": 3436
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 3436,
      "training_step_time": 9.921414375305176
    },
    {
      "epoch": 1.31072998046875e-05,
      "model_forward_time": 3.139101266860962,
      "step": 3436
    },
    {
      "epoch": 1.31072998046875e-05,
      "step": 3436,
      "training_step_time": 10.267599105834961
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 3.148059844970703,
      "step": 3437
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 3437,
      "training_step_time": 9.93225884437561
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 3.4017887115478516,
      "step": 3437
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 3437,
      "training_step_time": 10.190711498260498
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 3.1514694690704346,
      "step": 3437
    },
    {
      "dynamics_loss_avg": 0.05627510454505682,
      "epoch": 1.3111114501953124e-05,
      "step": 3437
    },
    {
      "action_loss_avg": 0.03169656461104751,
      "epoch": 1.3111114501953124e-05,
      "step": 3437
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 3437,
      "training_step_time": 9.934193134307861
    },
    {
      "epoch": 1.3111114501953124e-05,
      "model_forward_time": 3.1446917057037354,
      "step": 3437
    },
    {
      "epoch": 1.3111114501953124e-05,
      "step": 3437,
      "training_step_time": 10.267433881759644
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 3.149961233139038,
      "step": 3438
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 3438,
      "training_step_time": 9.930460691452026
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 3.1449451446533203,
      "step": 3438
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 3438,
      "training_step_time": 9.930213212966919
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 3.4730234146118164,
      "step": 3438
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 3438,
      "training_step_time": 10.256644248962402
    },
    {
      "epoch": 1.311492919921875e-05,
      "model_forward_time": 3.1421267986297607,
      "step": 3438
    },
    {
      "epoch": 1.311492919921875e-05,
      "step": 3438,
      "training_step_time": 10.291967391967773
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 3.1445226669311523,
      "step": 3439
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 3439,
      "training_step_time": 9.922831058502197
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 3.1387171745300293,
      "step": 3439
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 3439,
      "training_step_time": 9.915483236312866
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 3.1356594562530518,
      "step": 3439
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 3439,
      "training_step_time": 9.915258646011353
    },
    {
      "epoch": 1.3118743896484375e-05,
      "model_forward_time": 3.1377882957458496,
      "step": 3439
    },
    {
      "epoch": 1.3118743896484375e-05,
      "step": 3439,
      "training_step_time": 10.304238319396973
    },
    {
      "epoch": 1.312255859375e-05,
      "grad_norm": 0.3451438248157501,
      "learning_rate": 1e-05,
      "loss": 0.0968,
      "step": 3440
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 3.15482759475708,
      "step": 3440
    },
    {
      "dynamics_loss_avg": 0.07132513374090195,
      "epoch": 1.312255859375e-05,
      "step": 3440
    },
    {
      "action_loss_avg": 0.028015515208244322,
      "epoch": 1.312255859375e-05,
      "step": 3440
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 3440,
      "training_step_time": 9.942363262176514
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 3.1441643238067627,
      "step": 3440
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 3440,
      "training_step_time": 9.934017896652222
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 3.138364791870117,
      "step": 3440
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 3440,
      "training_step_time": 9.918976783752441
    },
    {
      "epoch": 1.312255859375e-05,
      "model_forward_time": 3.139461040496826,
      "step": 3440
    },
    {
      "epoch": 1.312255859375e-05,
      "step": 3440,
      "training_step_time": 10.282843828201294
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 3.4347023963928223,
      "step": 3441
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 3441,
      "training_step_time": 10.225492238998413
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 3.1385891437530518,
      "step": 3441
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 3441,
      "training_step_time": 9.917977094650269
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 3.1394007205963135,
      "step": 3441
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 3441,
      "training_step_time": 9.930460453033447
    },
    {
      "epoch": 1.3126373291015624e-05,
      "model_forward_time": 3.140479326248169,
      "step": 3441
    },
    {
      "epoch": 1.3126373291015624e-05,
      "step": 3441,
      "training_step_time": 10.260889768600464
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 3.1579411029815674,
      "step": 3442
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 3442,
      "training_step_time": 9.943292617797852
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 3.1405930519104004,
      "step": 3442
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 3442,
      "training_step_time": 10.210604429244995
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 3.1415483951568604,
      "step": 3442
    },
    {
      "dynamics_loss_avg": 0.07988158240914345,
      "epoch": 1.313018798828125e-05,
      "step": 3442
    },
    {
      "action_loss_avg": 0.038239596970379354,
      "epoch": 1.313018798828125e-05,
      "step": 3442
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 3442,
      "training_step_time": 9.928035020828247
    },
    {
      "epoch": 1.313018798828125e-05,
      "model_forward_time": 3.146812677383423,
      "step": 3442
    },
    {
      "epoch": 1.313018798828125e-05,
      "step": 3442,
      "training_step_time": 10.277975082397461
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 3.154379367828369,
      "step": 3443
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 3443,
      "training_step_time": 9.939729452133179
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 3.1407370567321777,
      "step": 3443
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 3443,
      "training_step_time": 9.922175168991089
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 3.402256727218628,
      "step": 3443
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 3443,
      "training_step_time": 10.186529636383057
    },
    {
      "epoch": 1.3134002685546875e-05,
      "model_forward_time": 3.144695281982422,
      "step": 3443
    },
    {
      "epoch": 1.3134002685546875e-05,
      "step": 3443,
      "training_step_time": 10.337002515792847
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 3.1569228172302246,
      "step": 3444
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 3444,
      "training_step_time": 9.94436526298523
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 3.134768486022949,
      "step": 3444
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 3444,
      "training_step_time": 9.916428804397583
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 3.1395230293273926,
      "step": 3444
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 3444,
      "training_step_time": 9.919836044311523
    },
    {
      "epoch": 1.31378173828125e-05,
      "model_forward_time": 3.1352527141571045,
      "step": 3444
    },
    {
      "epoch": 1.31378173828125e-05,
      "step": 3444,
      "training_step_time": 10.314897298812866
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 3.1498308181762695,
      "step": 3445
    },
    {
      "dynamics_loss_avg": 0.06631596256047487,
      "epoch": 1.3141632080078124e-05,
      "step": 3445
    },
    {
      "action_loss_avg": 0.03428981155157089,
      "epoch": 1.3141632080078124e-05,
      "step": 3445
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 3445,
      "training_step_time": 9.9297616481781
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 3.134270191192627,
      "step": 3445
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 3445,
      "training_step_time": 9.90431547164917
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 3.135920524597168,
      "step": 3445
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 3445,
      "training_step_time": 9.905941486358643
    },
    {
      "epoch": 1.3141632080078124e-05,
      "model_forward_time": 3.1348659992218018,
      "step": 3445
    },
    {
      "epoch": 1.3141632080078124e-05,
      "step": 3445,
      "training_step_time": 10.325836181640625
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 3.4296014308929443,
      "step": 3446
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 3446,
      "training_step_time": 10.205975532531738
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 3.135866641998291,
      "step": 3446
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 3446,
      "training_step_time": 9.912011861801147
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 3.13525128364563,
      "step": 3446
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 3446,
      "training_step_time": 9.912071704864502
    },
    {
      "epoch": 1.314544677734375e-05,
      "model_forward_time": 3.1377956867218018,
      "step": 3446
    },
    {
      "epoch": 1.314544677734375e-05,
      "step": 3446,
      "training_step_time": 10.021188259124756
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 3.1527559757232666,
      "step": 3447
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 3447,
      "training_step_time": 9.935739755630493
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 3.3945019245147705,
      "step": 3447
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 3447,
      "training_step_time": 10.176650285720825
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 3.1453328132629395,
      "step": 3447
    },
    {
      "dynamics_loss_avg": 0.05783070381730795,
      "epoch": 1.3149261474609375e-05,
      "step": 3447
    },
    {
      "action_loss_avg": 0.028428338840603827,
      "epoch": 1.3149261474609375e-05,
      "step": 3447
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 3447,
      "training_step_time": 9.929697275161743
    },
    {
      "epoch": 1.3149261474609375e-05,
      "model_forward_time": 3.1389753818511963,
      "step": 3447
    },
    {
      "epoch": 1.3149261474609375e-05,
      "step": 3447,
      "training_step_time": 10.286525249481201
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 3.14656662940979,
      "step": 3448
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 3448,
      "training_step_time": 9.931736946105957
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 3.138244152069092,
      "step": 3448
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 3448,
      "training_step_time": 9.915831565856934
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 3.4134161472320557,
      "step": 3448
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 3448,
      "training_step_time": 10.190220594406128
    },
    {
      "epoch": 1.3153076171875e-05,
      "model_forward_time": 3.140127658843994,
      "step": 3448
    },
    {
      "epoch": 1.3153076171875e-05,
      "step": 3448,
      "training_step_time": 10.299842357635498
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 3.139763832092285,
      "step": 3449
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 3449,
      "training_step_time": 9.9125497341156
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 3.1400187015533447,
      "step": 3449
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 3449,
      "training_step_time": 9.921358108520508
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 3.1377148628234863,
      "step": 3449
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 3449,
      "training_step_time": 9.915318489074707
    },
    {
      "epoch": 1.3156890869140624e-05,
      "model_forward_time": 3.3918535709381104,
      "step": 3449
    },
    {
      "epoch": 1.3156890869140624e-05,
      "step": 3449,
      "training_step_time": 10.333319902420044
    },
    {
      "epoch": 1.316070556640625e-05,
      "grad_norm": 0.4257708489894867,
      "learning_rate": 1e-05,
      "loss": 0.0981,
      "step": 3450
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 3.161287546157837,
      "step": 3450
    },
    {
      "dynamics_loss_avg": 0.06860265210270881,
      "epoch": 1.316070556640625e-05,
      "step": 3450
    },
    {
      "action_loss_avg": 0.030007867980748414,
      "epoch": 1.316070556640625e-05,
      "step": 3450
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 3450,
      "training_step_time": 9.946611166000366
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 3.138688325881958,
      "step": 3450
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 3450,
      "training_step_time": 9.918413400650024
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 3.1371030807495117,
      "step": 3450
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 3450,
      "training_step_time": 9.91205620765686
    },
    {
      "epoch": 1.316070556640625e-05,
      "model_forward_time": 3.136728286743164,
      "step": 3450
    },
    {
      "epoch": 1.316070556640625e-05,
      "step": 3450,
      "training_step_time": 10.284860610961914
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 3.39928936958313,
      "step": 3451
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 3451,
      "training_step_time": 10.174636363983154
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 3.136995553970337,
      "step": 3451
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 3451,
      "training_step_time": 9.91749119758606
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 3.1379506587982178,
      "step": 3451
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 3451,
      "training_step_time": 9.917377233505249
    },
    {
      "epoch": 1.3164520263671875e-05,
      "model_forward_time": 3.1366639137268066,
      "step": 3451
    },
    {
      "epoch": 1.3164520263671875e-05,
      "step": 3451,
      "training_step_time": 10.293789148330688
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 3.152222156524658,
      "step": 3452
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 3452,
      "training_step_time": 9.932864427566528
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 3.442657470703125,
      "step": 3452
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 3452,
      "training_step_time": 10.216530323028564
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 3.1328938007354736,
      "step": 3452
    },
    {
      "dynamics_loss_avg": 0.06127214953303337,
      "epoch": 1.31683349609375e-05,
      "step": 3452
    },
    {
      "action_loss_avg": 0.03430901300162077,
      "epoch": 1.31683349609375e-05,
      "step": 3452
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 3452,
      "training_step_time": 9.908979177474976
    },
    {
      "epoch": 1.31683349609375e-05,
      "model_forward_time": 3.13883113861084,
      "step": 3452
    },
    {
      "epoch": 1.31683349609375e-05,
      "step": 3452,
      "training_step_time": 10.28862476348877
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 3.1477138996124268,
      "step": 3453
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 3453,
      "training_step_time": 9.920476913452148
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 2.4304113388061523,
      "step": 3453
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 3453,
      "training_step_time": 7.458090305328369
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 3.3897545337677,
      "step": 3453
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 3453,
      "training_step_time": 10.159519910812378
    },
    {
      "epoch": 1.3172149658203124e-05,
      "model_forward_time": 3.137303590774536,
      "step": 3453
    },
    {
      "epoch": 1.3172149658203124e-05,
      "step": 3453,
      "training_step_time": 10.26761245727539
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 3.155935049057007,
      "step": 3454
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 3454,
      "training_step_time": 9.930222511291504
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 2.437278985977173,
      "step": 3454
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 3454,
      "training_step_time": 7.46153450012207
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 3.138413906097412,
      "step": 3454
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 3454,
      "training_step_time": 9.920323371887207
    },
    {
      "epoch": 1.317596435546875e-05,
      "model_forward_time": 3.438714027404785,
      "step": 3454
    },
    {
      "epoch": 1.317596435546875e-05,
      "step": 3454,
      "training_step_time": 10.308504819869995
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 3.1594996452331543,
      "step": 3455
    },
    {
      "dynamics_loss_avg": 0.049347952287644145,
      "epoch": 1.3179779052734375e-05,
      "step": 3455
    },
    {
      "action_loss_avg": 0.021934103313833477,
      "epoch": 1.3179779052734375e-05,
      "step": 3455
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 3455,
      "training_step_time": 9.94347882270813
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 3.1411595344543457,
      "step": 3455
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 3455,
      "training_step_time": 9.916674137115479
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 3.1425466537475586,
      "step": 3455
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 3455,
      "training_step_time": 9.922483921051025
    },
    {
      "epoch": 1.3179779052734375e-05,
      "model_forward_time": 3.140033483505249,
      "step": 3455
    },
    {
      "epoch": 1.3179779052734375e-05,
      "step": 3455,
      "training_step_time": 10.298551559448242
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 2.446882486343384,
      "step": 3456
    },
    {
      "epoch": 1.318359375e-05,
      "step": 3456,
      "training_step_time": 7.477341175079346
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 3.3849122524261475,
      "step": 3456
    },
    {
      "epoch": 1.318359375e-05,
      "step": 3456,
      "training_step_time": 10.15173602104187
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 3.140510082244873,
      "step": 3456
    },
    {
      "epoch": 1.318359375e-05,
      "step": 3456,
      "training_step_time": 9.907496452331543
    },
    {
      "epoch": 1.318359375e-05,
      "model_forward_time": 3.1370062828063965,
      "step": 3456
    },
    {
      "epoch": 1.318359375e-05,
      "step": 3456,
      "training_step_time": 10.321471214294434
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 3.1657838821411133,
      "step": 3457
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 3457,
      "training_step_time": 9.948771715164185
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 3.40297269821167,
      "step": 3457
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 3457,
      "training_step_time": 10.181113719940186
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 2.4359207153320312,
      "step": 3457
    },
    {
      "dynamics_loss_avg": 0.06572187170386315,
      "epoch": 1.3187408447265624e-05,
      "step": 3457
    },
    {
      "action_loss_avg": 0.028042875370010732,
      "epoch": 1.3187408447265624e-05,
      "step": 3457
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 3457,
      "training_step_time": 7.4752631187438965
    },
    {
      "epoch": 1.3187408447265624e-05,
      "model_forward_time": 3.139463186264038,
      "step": 3457
    },
    {
      "epoch": 1.3187408447265624e-05,
      "step": 3457,
      "training_step_time": 10.329140186309814
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 3.147365093231201,
      "step": 3458
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 3458,
      "training_step_time": 9.922118663787842
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 3.141430377960205,
      "step": 3458
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 3458,
      "training_step_time": 9.931779623031616
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 3.1432294845581055,
      "step": 3458
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 3458,
      "training_step_time": 9.926753759384155
    },
    {
      "epoch": 1.319122314453125e-05,
      "model_forward_time": 3.4371724128723145,
      "step": 3458
    },
    {
      "epoch": 1.319122314453125e-05,
      "step": 3458,
      "training_step_time": 10.31357717514038
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 3.15600323677063,
      "step": 3459
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 3459,
      "training_step_time": 9.94459080696106
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 3.146637439727783,
      "step": 3459
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 3459,
      "training_step_time": 9.930278301239014
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 2.4359304904937744,
      "step": 3459
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 3459,
      "training_step_time": 7.468425989151001
    },
    {
      "epoch": 1.3195037841796875e-05,
      "model_forward_time": 3.141263723373413,
      "step": 3459
    },
    {
      "epoch": 1.3195037841796875e-05,
      "step": 3459,
      "training_step_time": 10.316593885421753
    },
    {
      "epoch": 1.31988525390625e-05,
      "grad_norm": 0.27707889676094055,
      "learning_rate": 1e-05,
      "loss": 0.0927,
      "step": 3460
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 3.152252435684204,
      "step": 3460
    },
    {
      "dynamics_loss_avg": 0.05568918194621801,
      "epoch": 1.31988525390625e-05,
      "step": 3460
    },
    {
      "action_loss_avg": 0.02851500967517495,
      "epoch": 1.31988525390625e-05,
      "step": 3460
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 3460,
      "training_step_time": 9.937535762786865
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 3.137056350708008,
      "step": 3460
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 3460,
      "training_step_time": 9.910080909729004
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 3.1326050758361816,
      "step": 3460
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 3460,
      "training_step_time": 9.899962186813354
    },
    {
      "epoch": 1.31988525390625e-05,
      "model_forward_time": 3.130455493927002,
      "step": 3460
    },
    {
      "epoch": 1.31988525390625e-05,
      "step": 3460,
      "training_step_time": 10.312947988510132
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 3.3980748653411865,
      "step": 3461
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 3461,
      "training_step_time": 10.167269706726074
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 3.1339356899261475,
      "step": 3461
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 3461,
      "training_step_time": 9.906136751174927
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 3.1332132816314697,
      "step": 3461
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 3461,
      "training_step_time": 9.907294034957886
    },
    {
      "epoch": 1.3202667236328124e-05,
      "model_forward_time": 3.132812261581421,
      "step": 3461
    },
    {
      "epoch": 1.3202667236328124e-05,
      "step": 3461,
      "training_step_time": 10.27268099784851
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 3.150552988052368,
      "step": 3462
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 3462,
      "training_step_time": 9.920185089111328
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 3.451845645904541,
      "step": 3462
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 3462,
      "training_step_time": 10.215658903121948
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 3.1319141387939453,
      "step": 3462
    },
    {
      "dynamics_loss_avg": 0.06799797788262367,
      "epoch": 1.320648193359375e-05,
      "step": 3462
    },
    {
      "action_loss_avg": 0.036477175541222095,
      "epoch": 1.320648193359375e-05,
      "step": 3462
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 3462,
      "training_step_time": 9.90653133392334
    },
    {
      "epoch": 1.320648193359375e-05,
      "model_forward_time": 3.134726047515869,
      "step": 3462
    },
    {
      "epoch": 1.320648193359375e-05,
      "step": 3462,
      "training_step_time": 10.271701097488403
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 3.151379108428955,
      "step": 3463
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 3463,
      "training_step_time": 9.92831563949585
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 3.1318893432617188,
      "step": 3463
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 3463,
      "training_step_time": 9.90767240524292
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 3.407599925994873,
      "step": 3463
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 3463,
      "training_step_time": 10.18364429473877
    },
    {
      "epoch": 1.3210296630859375e-05,
      "model_forward_time": 3.139700174331665,
      "step": 3463
    },
    {
      "epoch": 1.3210296630859375e-05,
      "step": 3463,
      "training_step_time": 10.027689933776855
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 3.1484861373901367,
      "step": 3464
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 3464,
      "training_step_time": 9.924727439880371
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 3.1343164443969727,
      "step": 3464
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 3464,
      "training_step_time": 9.910078287124634
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 3.1398186683654785,
      "step": 3464
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 3464,
      "training_step_time": 9.918045043945312
    },
    {
      "epoch": 1.3214111328125e-05,
      "model_forward_time": 3.4186739921569824,
      "step": 3464
    },
    {
      "epoch": 1.3214111328125e-05,
      "step": 3464,
      "training_step_time": 10.288201808929443
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 3.1451146602630615,
      "step": 3465
    },
    {
      "dynamics_loss_avg": 0.060611442662775517,
      "epoch": 1.3217926025390624e-05,
      "step": 3465
    },
    {
      "action_loss_avg": 0.029298834316432477,
      "epoch": 1.3217926025390624e-05,
      "step": 3465
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 3465,
      "training_step_time": 9.923729658126831
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 3.132314443588257,
      "step": 3465
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 3465,
      "training_step_time": 9.909016370773315
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 3.1357083320617676,
      "step": 3465
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 3465,
      "training_step_time": 9.905483961105347
    },
    {
      "epoch": 1.3217926025390624e-05,
      "model_forward_time": 3.134429931640625,
      "step": 3465
    },
    {
      "epoch": 1.3217926025390624e-05,
      "step": 3465,
      "training_step_time": 10.221712112426758
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 3.4265732765197754,
      "step": 3466
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 3466,
      "training_step_time": 10.19869875907898
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 3.1324808597564697,
      "step": 3466
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 3466,
      "training_step_time": 9.900291919708252
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 2.429225444793701,
      "step": 3466
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 3466,
      "training_step_time": 7.459794998168945
    },
    {
      "epoch": 1.322174072265625e-05,
      "model_forward_time": 3.1284642219543457,
      "step": 3466
    },
    {
      "epoch": 1.322174072265625e-05,
      "step": 3466,
      "training_step_time": 10.347145795822144
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 3.1499860286712646,
      "step": 3467
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 3467,
      "training_step_time": 9.921495914459229
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 3.129725217819214,
      "step": 3467
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 3467,
      "training_step_time": 10.203198909759521
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 3.1344335079193115,
      "step": 3467
    },
    {
      "dynamics_loss_avg": 0.06955606564879417,
      "epoch": 1.3225555419921875e-05,
      "step": 3467
    },
    {
      "action_loss_avg": 0.03288189265877008,
      "epoch": 1.3225555419921875e-05,
      "step": 3467
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 3467,
      "training_step_time": 9.90345048904419
    },
    {
      "epoch": 1.3225555419921875e-05,
      "model_forward_time": 3.13222599029541,
      "step": 3467
    },
    {
      "epoch": 1.3225555419921875e-05,
      "step": 3467,
      "training_step_time": 10.293632745742798
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 3.1472508907318115,
      "step": 3468
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 3468,
      "training_step_time": 9.92007040977478
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 3.137133836746216,
      "step": 3468
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 3468,
      "training_step_time": 9.907856225967407
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 3.4186718463897705,
      "step": 3468
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 3468,
      "training_step_time": 10.18724775314331
    },
    {
      "epoch": 1.32293701171875e-05,
      "model_forward_time": 3.132399320602417,
      "step": 3468
    },
    {
      "epoch": 1.32293701171875e-05,
      "step": 3468,
      "training_step_time": 10.074227333068848
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 3.155102014541626,
      "step": 3469
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 3469,
      "training_step_time": 9.92748212814331
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 3.1354191303253174,
      "step": 3469
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 3469,
      "training_step_time": 9.90926742553711
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 3.1329996585845947,
      "step": 3469
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 3469,
      "training_step_time": 9.908785343170166
    },
    {
      "epoch": 1.3233184814453124e-05,
      "model_forward_time": 3.400467872619629,
      "step": 3469
    },
    {
      "epoch": 1.3233184814453124e-05,
      "step": 3469,
      "training_step_time": 10.271546602249146
    },
    {
      "epoch": 1.323699951171875e-05,
      "grad_norm": 0.47692981362342834,
      "learning_rate": 1e-05,
      "loss": 0.0922,
      "step": 3470
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 3.1443915367126465,
      "step": 3470
    },
    {
      "dynamics_loss_avg": 0.053846141695976256,
      "epoch": 1.323699951171875e-05,
      "step": 3470
    },
    {
      "action_loss_avg": 0.02071918034926057,
      "epoch": 1.323699951171875e-05,
      "step": 3470
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 3470,
      "training_step_time": 9.922650814056396
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 3.1392316818237305,
      "step": 3470
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 3470,
      "training_step_time": 9.91347622871399
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 3.13167667388916,
      "step": 3470
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 3470,
      "training_step_time": 9.900878429412842
    },
    {
      "epoch": 1.323699951171875e-05,
      "model_forward_time": 3.1396024227142334,
      "step": 3470
    },
    {
      "epoch": 1.323699951171875e-05,
      "step": 3470,
      "training_step_time": 10.022586822509766
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 3.417557954788208,
      "step": 3471
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 3471,
      "training_step_time": 10.196639776229858
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 3.143617868423462,
      "step": 3471
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 3471,
      "training_step_time": 9.921655416488647
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 3.143287181854248,
      "step": 3471
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 3471,
      "training_step_time": 9.921835660934448
    },
    {
      "epoch": 1.3240814208984375e-05,
      "model_forward_time": 3.147825002670288,
      "step": 3471
    },
    {
      "epoch": 1.3240814208984375e-05,
      "step": 3471,
      "training_step_time": 10.312193155288696
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 2.4473719596862793,
      "step": 3472
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 3472,
      "training_step_time": 7.484201192855835
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 3.4199326038360596,
      "step": 3472
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 3472,
      "training_step_time": 10.19365382194519
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 3.1413960456848145,
      "step": 3472
    },
    {
      "dynamics_loss_avg": 0.05789967551827431,
      "epoch": 1.324462890625e-05,
      "step": 3472
    },
    {
      "action_loss_avg": 0.02462607081979513,
      "epoch": 1.324462890625e-05,
      "step": 3472
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 3472,
      "training_step_time": 9.921507596969604
    },
    {
      "epoch": 1.324462890625e-05,
      "model_forward_time": 3.142914295196533,
      "step": 3472
    },
    {
      "epoch": 1.324462890625e-05,
      "step": 3472,
      "training_step_time": 10.281665802001953
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 3.154733657836914,
      "step": 3473
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 3473,
      "training_step_time": 9.930810451507568
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 3.1361289024353027,
      "step": 3473
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 3473,
      "training_step_time": 9.912181615829468
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 2.4388575553894043,
      "step": 3473
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 3473,
      "training_step_time": 7.470712423324585
    },
    {
      "epoch": 1.3248443603515624e-05,
      "model_forward_time": 3.4127931594848633,
      "step": 3473
    },
    {
      "epoch": 1.3248443603515624e-05,
      "step": 3473,
      "training_step_time": 10.283626079559326
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 2.453275203704834,
      "step": 3474
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 3474,
      "training_step_time": 7.4877331256866455
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 3.139369010925293,
      "step": 3474
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 3474,
      "training_step_time": 9.909656763076782
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 3.143841505050659,
      "step": 3474
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 3474,
      "training_step_time": 9.918182849884033
    },
    {
      "epoch": 1.325225830078125e-05,
      "model_forward_time": 2.4343276023864746,
      "step": 3474
    },
    {
      "epoch": 1.325225830078125e-05,
      "step": 3474,
      "training_step_time": 10.01873517036438
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 3.4662599563598633,
      "step": 3475
    },
    {
      "dynamics_loss_avg": 0.06510170269757509,
      "epoch": 1.3256072998046875e-05,
      "step": 3475
    },
    {
      "action_loss_avg": 0.027709690621122718,
      "epoch": 1.3256072998046875e-05,
      "step": 3475
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 3475,
      "training_step_time": 10.241695880889893
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 3.141059160232544,
      "step": 3475
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 3475,
      "training_step_time": 9.906136989593506
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 3.1374685764312744,
      "step": 3475
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 3475,
      "training_step_time": 9.90354037284851
    },
    {
      "epoch": 1.3256072998046875e-05,
      "model_forward_time": 3.132864475250244,
      "step": 3475
    },
    {
      "epoch": 1.3256072998046875e-05,
      "step": 3475,
      "training_step_time": 10.00552225112915
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 3.139582395553589,
      "step": 3476
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 3476,
      "training_step_time": 10.214399099349976
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 3.1331424713134766,
      "step": 3476
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 3476,
      "training_step_time": 9.900286197662354
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 3.1376874446868896,
      "step": 3476
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 3476,
      "training_step_time": 9.91009521484375
    },
    {
      "epoch": 1.32598876953125e-05,
      "model_forward_time": 3.1360979080200195,
      "step": 3476
    },
    {
      "epoch": 1.32598876953125e-05,
      "step": 3476,
      "training_step_time": 10.261972427368164
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 3.1462209224700928,
      "step": 3477
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 3477,
      "training_step_time": 9.92330813407898
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 3.4103381633758545,
      "step": 3477
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 3477,
      "training_step_time": 10.182843923568726
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 3.1412997245788574,
      "step": 3477
    },
    {
      "dynamics_loss_avg": 0.07542615309357643,
      "epoch": 1.3263702392578124e-05,
      "step": 3477
    },
    {
      "action_loss_avg": 0.03562938794493675,
      "epoch": 1.3263702392578124e-05,
      "step": 3477
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 3477,
      "training_step_time": 9.92381763458252
    },
    {
      "epoch": 1.3263702392578124e-05,
      "model_forward_time": 3.1386280059814453,
      "step": 3477
    },
    {
      "epoch": 1.3263702392578124e-05,
      "step": 3477,
      "training_step_time": 10.308532238006592
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 3.1545791625976562,
      "step": 3478
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 3478,
      "training_step_time": 9.935240983963013
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 3.1420247554779053,
      "step": 3478
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 3478,
      "training_step_time": 9.91342806816101
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 3.420832633972168,
      "step": 3478
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 3478,
      "training_step_time": 10.192638158798218
    },
    {
      "epoch": 1.326751708984375e-05,
      "model_forward_time": 3.135448694229126,
      "step": 3478
    },
    {
      "epoch": 1.326751708984375e-05,
      "step": 3478,
      "training_step_time": 10.323330879211426
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 3.1470282077789307,
      "step": 3479
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 3479,
      "training_step_time": 9.924688816070557
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 2.435265302658081,
      "step": 3479
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 3479,
      "training_step_time": 7.460993528366089
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 3.1313419342041016,
      "step": 3479
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 3479,
      "training_step_time": 9.902122020721436
    },
    {
      "epoch": 1.3271331787109375e-05,
      "model_forward_time": 3.138798952102661,
      "step": 3479
    },
    {
      "epoch": 1.3271331787109375e-05,
      "step": 3479,
      "training_step_time": 10.274854183197021
    },
    {
      "epoch": 1.3275146484375e-05,
      "grad_norm": 0.4764023721218109,
      "learning_rate": 1e-05,
      "loss": 0.0979,
      "step": 3480
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 3.161146879196167,
      "step": 3480
    },
    {
      "dynamics_loss_avg": 0.07957221232354642,
      "epoch": 1.3275146484375e-05,
      "step": 3480
    },
    {
      "action_loss_avg": 0.03456805534660816,
      "epoch": 1.3275146484375e-05,
      "step": 3480
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 3480,
      "training_step_time": 9.937307596206665
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 3.148512125015259,
      "step": 3480
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 3480,
      "training_step_time": 9.932242155075073
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 3.1412689685821533,
      "step": 3480
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 3480,
      "training_step_time": 9.921401023864746
    },
    {
      "epoch": 1.3275146484375e-05,
      "model_forward_time": 3.1468381881713867,
      "step": 3480
    },
    {
      "epoch": 1.3275146484375e-05,
      "step": 3480,
      "training_step_time": 10.029155969619751
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 3.4314253330230713,
      "step": 3481
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 3481,
      "training_step_time": 10.212324142456055
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 2.4401748180389404,
      "step": 3481
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 3481,
      "training_step_time": 7.471240043640137
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 3.1445236206054688,
      "step": 3481
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 3481,
      "training_step_time": 9.923139095306396
    },
    {
      "epoch": 1.3278961181640625e-05,
      "model_forward_time": 3.1481828689575195,
      "step": 3481
    },
    {
      "epoch": 1.3278961181640625e-05,
      "step": 3481,
      "training_step_time": 10.339135885238647
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 3.1753087043762207,
      "step": 3482
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 3482,
      "training_step_time": 9.949206352233887
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 3.1415793895721436,
      "step": 3482
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 3482,
      "training_step_time": 9.910898208618164
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 3.424656391143799,
      "step": 3482
    },
    {
      "dynamics_loss_avg": 0.07303429841995239,
      "epoch": 1.328277587890625e-05,
      "step": 3482
    },
    {
      "action_loss_avg": 0.03441557241603732,
      "epoch": 1.328277587890625e-05,
      "step": 3482
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 3482,
      "training_step_time": 10.196898698806763
    },
    {
      "epoch": 1.328277587890625e-05,
      "model_forward_time": 3.143749952316284,
      "step": 3482
    },
    {
      "epoch": 1.328277587890625e-05,
      "step": 3482,
      "training_step_time": 10.29917287826538
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 3.150503635406494,
      "step": 3483
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 3483,
      "training_step_time": 9.917605876922607
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 3.137701988220215,
      "step": 3483
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 3483,
      "training_step_time": 9.904585361480713
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 3.4074084758758545,
      "step": 3483
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 3483,
      "training_step_time": 10.181829690933228
    },
    {
      "epoch": 1.3286590576171875e-05,
      "model_forward_time": 3.1400110721588135,
      "step": 3483
    },
    {
      "epoch": 1.3286590576171875e-05,
      "step": 3483,
      "training_step_time": 10.275230407714844
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 3.1514875888824463,
      "step": 3484
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 3484,
      "training_step_time": 9.919817924499512
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 3.141005754470825,
      "step": 3484
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 3484,
      "training_step_time": 9.905004024505615
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 2.438781499862671,
      "step": 3484
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 3484,
      "training_step_time": 7.460923671722412
    },
    {
      "epoch": 1.32904052734375e-05,
      "model_forward_time": 3.1380436420440674,
      "step": 3484
    },
    {
      "epoch": 1.32904052734375e-05,
      "step": 3484,
      "training_step_time": 10.314738035202026
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 3.5077035427093506,
      "step": 3485
    },
    {
      "dynamics_loss_avg": 0.07067675292491912,
      "epoch": 1.3294219970703125e-05,
      "step": 3485
    },
    {
      "action_loss_avg": 0.02738996995612979,
      "epoch": 1.3294219970703125e-05,
      "step": 3485
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 3485,
      "training_step_time": 10.277973651885986
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 3.1390974521636963,
      "step": 3485
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 3485,
      "training_step_time": 9.9022536277771
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 3.149984836578369,
      "step": 3485
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 3485,
      "training_step_time": 9.914948463439941
    },
    {
      "epoch": 1.3294219970703125e-05,
      "model_forward_time": 3.139411211013794,
      "step": 3485
    },
    {
      "epoch": 1.3294219970703125e-05,
      "step": 3485,
      "training_step_time": 10.276715993881226
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 3.4538495540618896,
      "step": 3486
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 3486,
      "training_step_time": 10.220745086669922
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 3.1352407932281494,
      "step": 3486
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 3486,
      "training_step_time": 9.899717569351196
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 3.1431002616882324,
      "step": 3486
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 3486,
      "training_step_time": 9.910109281539917
    },
    {
      "epoch": 1.329803466796875e-05,
      "model_forward_time": 3.142221450805664,
      "step": 3486
    },
    {
      "epoch": 1.329803466796875e-05,
      "step": 3486,
      "training_step_time": 10.0244882106781
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 3.1508867740631104,
      "step": 3487
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 3487,
      "training_step_time": 9.927524328231812
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 3.447038173675537,
      "step": 3487
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 3487,
      "training_step_time": 10.215082168579102
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 3.1419758796691895,
      "step": 3487
    },
    {
      "dynamics_loss_avg": 0.07068331725895405,
      "epoch": 1.3301849365234375e-05,
      "step": 3487
    },
    {
      "action_loss_avg": 0.029831123212352394,
      "epoch": 1.3301849365234375e-05,
      "step": 3487
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 3487,
      "training_step_time": 9.914504766464233
    },
    {
      "epoch": 1.3301849365234375e-05,
      "model_forward_time": 3.138472318649292,
      "step": 3487
    },
    {
      "epoch": 1.3301849365234375e-05,
      "step": 3487,
      "training_step_time": 10.322957515716553
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 3.1553447246551514,
      "step": 3488
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 3488,
      "training_step_time": 9.926498174667358
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 3.1350438594818115,
      "step": 3488
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 3488,
      "training_step_time": 9.904590606689453
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 3.1397650241851807,
      "step": 3488
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 3488,
      "training_step_time": 10.197165727615356
    },
    {
      "epoch": 1.33056640625e-05,
      "model_forward_time": 3.1364710330963135,
      "step": 3488
    },
    {
      "epoch": 1.33056640625e-05,
      "step": 3488,
      "training_step_time": 10.324967622756958
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 3.1419765949249268,
      "step": 3489
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 3489,
      "training_step_time": 9.924829721450806
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 3.13753080368042,
      "step": 3489
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 3489,
      "training_step_time": 9.924397468566895
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 3.138120651245117,
      "step": 3489
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 3489,
      "training_step_time": 9.913411855697632
    },
    {
      "epoch": 1.3309478759765625e-05,
      "model_forward_time": 3.13871169090271,
      "step": 3489
    },
    {
      "epoch": 1.3309478759765625e-05,
      "step": 3489,
      "training_step_time": 10.326428890228271
    },
    {
      "epoch": 1.331329345703125e-05,
      "grad_norm": 0.4131412208080292,
      "learning_rate": 1e-05,
      "loss": 0.0955,
      "step": 3490
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 3.1431522369384766,
      "step": 3490
    },
    {
      "dynamics_loss_avg": 0.07474969383329153,
      "epoch": 1.331329345703125e-05,
      "step": 3490
    },
    {
      "action_loss_avg": 0.04124852241948247,
      "epoch": 1.331329345703125e-05,
      "step": 3490
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 3490,
      "training_step_time": 9.931142568588257
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 3.139657497406006,
      "step": 3490
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 3490,
      "training_step_time": 9.923756837844849
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 3.1352291107177734,
      "step": 3490
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 3490,
      "training_step_time": 9.918928623199463
    },
    {
      "epoch": 1.331329345703125e-05,
      "model_forward_time": 3.1436679363250732,
      "step": 3490
    },
    {
      "epoch": 1.331329345703125e-05,
      "step": 3490,
      "training_step_time": 10.031501054763794
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 3.150902509689331,
      "step": 3491
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 3491,
      "training_step_time": 10.254727125167847
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 3.1445558071136475,
      "step": 3491
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 3491,
      "training_step_time": 9.926867485046387
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 3.139963150024414,
      "step": 3491
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 3491,
      "training_step_time": 9.925411939620972
    },
    {
      "epoch": 1.3317108154296875e-05,
      "model_forward_time": 3.1412930488586426,
      "step": 3491
    },
    {
      "epoch": 1.3317108154296875e-05,
      "step": 3491,
      "training_step_time": 10.027261734008789
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 3.1547586917877197,
      "step": 3492
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 3492,
      "training_step_time": 9.936695098876953
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 3.4599902629852295,
      "step": 3492
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 3492,
      "training_step_time": 10.243198871612549
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 3.1440138816833496,
      "step": 3492
    },
    {
      "dynamics_loss_avg": 0.06585300248116255,
      "epoch": 1.33209228515625e-05,
      "step": 3492
    },
    {
      "action_loss_avg": 0.03364813225343823,
      "epoch": 1.33209228515625e-05,
      "step": 3492
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 3492,
      "training_step_time": 9.927723407745361
    },
    {
      "epoch": 1.33209228515625e-05,
      "model_forward_time": 3.1433494091033936,
      "step": 3492
    },
    {
      "epoch": 1.33209228515625e-05,
      "step": 3492,
      "training_step_time": 10.347984790802002
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 3.1517159938812256,
      "step": 3493
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 3493,
      "training_step_time": 9.939631223678589
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 3.134772777557373,
      "step": 3493
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 3493,
      "training_step_time": 9.915397882461548
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 3.1384694576263428,
      "step": 3493
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 3493,
      "training_step_time": 10.20334529876709
    },
    {
      "epoch": 1.3324737548828125e-05,
      "model_forward_time": 3.139075756072998,
      "step": 3493
    },
    {
      "epoch": 1.3324737548828125e-05,
      "step": 3493,
      "training_step_time": 10.32573127746582
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 3.1431713104248047,
      "step": 3494
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 3494,
      "training_step_time": 9.922454833984375
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 3.1313371658325195,
      "step": 3494
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 3494,
      "training_step_time": 9.91090726852417
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 3.135213851928711,
      "step": 3494
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 3494,
      "training_step_time": 9.912280321121216
    },
    {
      "epoch": 1.332855224609375e-05,
      "model_forward_time": 3.4157326221466064,
      "step": 3494
    },
    {
      "epoch": 1.332855224609375e-05,
      "step": 3494,
      "training_step_time": 10.29641318321228
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 3.1552889347076416,
      "step": 3495
    },
    {
      "dynamics_loss_avg": 0.06988119520246983,
      "epoch": 1.3332366943359375e-05,
      "step": 3495
    },
    {
      "action_loss_avg": 0.027507026307284833,
      "epoch": 1.3332366943359375e-05,
      "step": 3495
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 3495,
      "training_step_time": 9.932373762130737
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 3.132465124130249,
      "step": 3495
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 3495,
      "training_step_time": 9.900495052337646
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 3.1298508644104004,
      "step": 3495
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 3495,
      "training_step_time": 9.895622253417969
    },
    {
      "epoch": 1.3332366943359375e-05,
      "model_forward_time": 3.132568597793579,
      "step": 3495
    },
    {
      "epoch": 1.3332366943359375e-05,
      "step": 3495,
      "training_step_time": 10.005419492721558
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 3.4480254650115967,
      "step": 3496
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 3496,
      "training_step_time": 10.215769052505493
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 3.1374151706695557,
      "step": 3496
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 3496,
      "training_step_time": 9.904255628585815
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 3.1358401775360107,
      "step": 3496
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 3496,
      "training_step_time": 9.902427196502686
    },
    {
      "epoch": 1.3336181640625e-05,
      "model_forward_time": 2.4319052696228027,
      "step": 3496
    },
    {
      "epoch": 1.3336181640625e-05,
      "step": 3496,
      "training_step_time": 10.292897939682007
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 3.1454594135284424,
      "step": 3497
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 3497,
      "training_step_time": 9.918209314346313
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 3.4013917446136475,
      "step": 3497
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 3497,
      "training_step_time": 10.172728300094604
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 3.1368050575256348,
      "step": 3497
    },
    {
      "dynamics_loss_avg": 0.05709289312362671,
      "epoch": 1.3339996337890625e-05,
      "step": 3497
    },
    {
      "action_loss_avg": 0.02810819949954748,
      "epoch": 1.3339996337890625e-05,
      "step": 3497
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 3497,
      "training_step_time": 9.915641784667969
    },
    {
      "epoch": 1.3339996337890625e-05,
      "model_forward_time": 3.1365060806274414,
      "step": 3497
    },
    {
      "epoch": 1.3339996337890625e-05,
      "step": 3497,
      "training_step_time": 10.325532913208008
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 2.4498913288116455,
      "step": 3498
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 3498,
      "training_step_time": 7.477941036224365
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 3.136754274368286,
      "step": 3498
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 3498,
      "training_step_time": 9.909047365188599
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 3.1403114795684814,
      "step": 3498
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 3498,
      "training_step_time": 9.91314435005188
    },
    {
      "epoch": 1.334381103515625e-05,
      "model_forward_time": 3.388054847717285,
      "step": 3498
    },
    {
      "epoch": 1.334381103515625e-05,
      "step": 3498,
      "training_step_time": 10.253773212432861
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 3.1833927631378174,
      "step": 3499
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 3499,
      "training_step_time": 9.96134066581726
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 3.135786771774292,
      "step": 3499
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 3499,
      "training_step_time": 9.907572031021118
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 3.133134126663208,
      "step": 3499
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 3499,
      "training_step_time": 9.90163540840149
    },
    {
      "epoch": 1.3347625732421875e-05,
      "model_forward_time": 3.1373844146728516,
      "step": 3499
    },
    {
      "epoch": 1.3347625732421875e-05,
      "step": 3499,
      "training_step_time": 10.288614749908447
    },
    {
      "epoch": 1.33514404296875e-05,
      "grad_norm": 0.5306731462478638,
      "learning_rate": 1e-05,
      "loss": 0.0935,
      "step": 3500
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 2.457425117492676,
      "step": 3500
    },
    {
      "dynamics_loss_avg": 0.06413809098303318,
      "epoch": 1.33514404296875e-05,
      "step": 3500
    },
    {
      "action_loss_avg": 0.03381378576159477,
      "epoch": 1.33514404296875e-05,
      "step": 3500
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 3500,
      "training_step_time": 7.476989030838013
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 3.117310047149658,
      "step": 3500
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 3500,
      "training_step_time": 9.864480257034302
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 3.1211442947387695,
      "step": 3500
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 3500,
      "training_step_time": 9.870621681213379
    },
    {
      "epoch": 1.33514404296875e-05,
      "model_forward_time": 3.121389865875244,
      "step": 3500
    },
    {
      "epoch": 1.33514404296875e-05,
      "step": 3500,
      "training_step_time": 10.00360369682312
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 5.4640514850616455,
      "step": 3501
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 3501,
      "training_step_time": 12.237078189849854
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 3.149250030517578,
      "step": 3501
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 3501,
      "training_step_time": 9.951785326004028
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 3.1597678661346436,
      "step": 3501
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 3501,
      "training_step_time": 9.964069843292236
    },
    {
      "epoch": 1.3355255126953125e-05,
      "model_forward_time": 3.1562845706939697,
      "step": 3501
    },
    {
      "epoch": 1.3355255126953125e-05,
      "step": 3501,
      "training_step_time": 10.28168272972107
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 3.158247232437134,
      "step": 3502
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 3502,
      "training_step_time": 9.938282012939453
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 3.4459547996520996,
      "step": 3502
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 3502,
      "training_step_time": 10.229789733886719
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 3.1647017002105713,
      "step": 3502
    },
    {
      "dynamics_loss_avg": 0.05919649787247181,
      "epoch": 1.335906982421875e-05,
      "step": 3502
    },
    {
      "action_loss_avg": 0.032461087126284836,
      "epoch": 1.335906982421875e-05,
      "step": 3502
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 3502,
      "training_step_time": 9.957003593444824
    },
    {
      "epoch": 1.335906982421875e-05,
      "model_forward_time": 3.1453545093536377,
      "step": 3502
    },
    {
      "epoch": 1.335906982421875e-05,
      "step": 3502,
      "training_step_time": 10.348487615585327
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 3.160342216491699,
      "step": 3503
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 3503,
      "training_step_time": 9.941110849380493
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 3.440462112426758,
      "step": 3503
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 3503,
      "training_step_time": 10.211665153503418
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 3.1370606422424316,
      "step": 3503
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 3503,
      "training_step_time": 9.912809371948242
    },
    {
      "epoch": 1.3362884521484375e-05,
      "model_forward_time": 3.1308538913726807,
      "step": 3503
    },
    {
      "epoch": 1.3362884521484375e-05,
      "step": 3503,
      "training_step_time": 10.263996839523315
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 3.1803340911865234,
      "step": 3504
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 3504,
      "training_step_time": 9.955349206924438
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 3.138303756713867,
      "step": 3504
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 3504,
      "training_step_time": 9.908226251602173
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 3.1339364051818848,
      "step": 3504
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 3504,
      "training_step_time": 10.214905738830566
    },
    {
      "epoch": 1.336669921875e-05,
      "model_forward_time": 3.1330111026763916,
      "step": 3504
    },
    {
      "epoch": 1.336669921875e-05,
      "step": 3504,
      "training_step_time": 10.248586177825928
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 3.1509006023406982,
      "step": 3505
    },
    {
      "dynamics_loss_avg": 0.08212301395833492,
      "epoch": 1.3370513916015625e-05,
      "step": 3505
    },
    {
      "action_loss_avg": 0.04567782920785248,
      "epoch": 1.3370513916015625e-05,
      "step": 3505
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 3505,
      "training_step_time": 9.92600131034851
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 3.1421079635620117,
      "step": 3505
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 3505,
      "training_step_time": 9.921653747558594
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 3.1383607387542725,
      "step": 3505
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 3505,
      "training_step_time": 9.923517942428589
    },
    {
      "epoch": 1.3370513916015625e-05,
      "model_forward_time": 3.402737617492676,
      "step": 3505
    },
    {
      "epoch": 1.3370513916015625e-05,
      "step": 3505,
      "training_step_time": 10.278816938400269
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 3.147474527359009,
      "step": 3506
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 3506,
      "training_step_time": 9.929783344268799
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 3.146501302719116,
      "step": 3506
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 3506,
      "training_step_time": 9.924707174301147
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 3.138223171234131,
      "step": 3506
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 3506,
      "training_step_time": 9.91964077949524
    },
    {
      "epoch": 1.337432861328125e-05,
      "model_forward_time": 2.4360649585723877,
      "step": 3506
    },
    {
      "epoch": 1.337432861328125e-05,
      "step": 3506,
      "training_step_time": 10.033198595046997
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 3.444253444671631,
      "step": 3507
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 3507,
      "training_step_time": 10.22112226486206
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 3.1408743858337402,
      "step": 3507
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 3507,
      "training_step_time": 9.913280248641968
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 3.140585422515869,
      "step": 3507
    },
    {
      "dynamics_loss_avg": 0.05875473152846098,
      "epoch": 1.3378143310546875e-05,
      "step": 3507
    },
    {
      "action_loss_avg": 0.029694318771362305,
      "epoch": 1.3378143310546875e-05,
      "step": 3507
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 3507,
      "training_step_time": 9.919081687927246
    },
    {
      "epoch": 1.3378143310546875e-05,
      "model_forward_time": 3.1416592597961426,
      "step": 3507
    },
    {
      "epoch": 1.3378143310546875e-05,
      "step": 3507,
      "training_step_time": 10.334073305130005
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 3.153308629989624,
      "step": 3508
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 3508,
      "training_step_time": 9.933000087738037
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 3.137187957763672,
      "step": 3508
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 3508,
      "training_step_time": 10.189516305923462
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 3.143143653869629,
      "step": 3508
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 3508,
      "training_step_time": 9.92137360572815
    },
    {
      "epoch": 1.33819580078125e-05,
      "model_forward_time": 3.143726348876953,
      "step": 3508
    },
    {
      "epoch": 1.33819580078125e-05,
      "step": 3508,
      "training_step_time": 10.284366846084595
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 3.1557135581970215,
      "step": 3509
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 3509,
      "training_step_time": 9.935795545578003
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 3.138796806335449,
      "step": 3509
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 3509,
      "training_step_time": 9.920036315917969
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 3.409599781036377,
      "step": 3509
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 3509,
      "training_step_time": 10.189798593521118
    },
    {
      "epoch": 1.3385772705078125e-05,
      "model_forward_time": 3.1468665599823,
      "step": 3509
    },
    {
      "epoch": 1.3385772705078125e-05,
      "step": 3509,
      "training_step_time": 10.026654958724976
    },
    {
      "epoch": 1.338958740234375e-05,
      "grad_norm": 0.47400906682014465,
      "learning_rate": 1e-05,
      "loss": 0.0995,
      "step": 3510
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 2.4447929859161377,
      "step": 3510
    },
    {
      "dynamics_loss_avg": 0.07295360825955868,
      "epoch": 1.338958740234375e-05,
      "step": 3510
    },
    {
      "action_loss_avg": 0.028401147201657297,
      "epoch": 1.338958740234375e-05,
      "step": 3510
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 3510,
      "training_step_time": 7.483425140380859
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 3.1405880451202393,
      "step": 3510
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 3510,
      "training_step_time": 9.918380975723267
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 3.1453375816345215,
      "step": 3510
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 3510,
      "training_step_time": 9.92518401145935
    },
    {
      "epoch": 1.338958740234375e-05,
      "model_forward_time": 3.1418561935424805,
      "step": 3510
    },
    {
      "epoch": 1.338958740234375e-05,
      "step": 3510,
      "training_step_time": 10.303171873092651
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 3.4804859161376953,
      "step": 3511
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 3511,
      "training_step_time": 10.255713939666748
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 3.138979434967041,
      "step": 3511
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 3511,
      "training_step_time": 9.92016887664795
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 3.1437950134277344,
      "step": 3511
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 3511,
      "training_step_time": 9.91974949836731
    },
    {
      "epoch": 1.3393402099609375e-05,
      "model_forward_time": 3.1379706859588623,
      "step": 3511
    },
    {
      "epoch": 1.3393402099609375e-05,
      "step": 3511,
      "training_step_time": 10.013831377029419
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 2.7089600563049316,
      "step": 3512
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 3512,
      "training_step_time": 7.740577459335327
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 3.1319141387939453,
      "step": 3512
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 3512,
      "training_step_time": 9.912923097610474
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 3.1385698318481445,
      "step": 3512
    },
    {
      "dynamics_loss_avg": 0.07178827989846467,
      "epoch": 1.3397216796875e-05,
      "step": 3512
    },
    {
      "action_loss_avg": 0.02953617125749588,
      "epoch": 1.3397216796875e-05,
      "step": 3512
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 3512,
      "training_step_time": 9.921996831893921
    },
    {
      "epoch": 1.3397216796875e-05,
      "model_forward_time": 3.141572952270508,
      "step": 3512
    },
    {
      "epoch": 1.3397216796875e-05,
      "step": 3512,
      "training_step_time": 10.299078702926636
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 3.1568408012390137,
      "step": 3513
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 3513,
      "training_step_time": 9.93745231628418
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 3.1414401531219482,
      "step": 3513
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 3513,
      "training_step_time": 10.182192087173462
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 2.437647819519043,
      "step": 3513
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 3513,
      "training_step_time": 7.470836400985718
    },
    {
      "epoch": 1.3401031494140625e-05,
      "model_forward_time": 3.1327195167541504,
      "step": 3513
    },
    {
      "epoch": 1.3401031494140625e-05,
      "step": 3513,
      "training_step_time": 10.318302869796753
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 3.179295778274536,
      "step": 3514
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 3514,
      "training_step_time": 9.967288255691528
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 3.1458096504211426,
      "step": 3514
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 3514,
      "training_step_time": 9.966628313064575
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 3.1415390968322754,
      "step": 3514
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 3514,
      "training_step_time": 10.21100640296936
    },
    {
      "epoch": 1.340484619140625e-05,
      "model_forward_time": 3.1421005725860596,
      "step": 3514
    },
    {
      "epoch": 1.340484619140625e-05,
      "step": 3514,
      "training_step_time": 10.293392181396484
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 3.182279109954834,
      "step": 3515
    },
    {
      "dynamics_loss_avg": 0.05913251377642155,
      "epoch": 1.3408660888671875e-05,
      "step": 3515
    },
    {
      "action_loss_avg": 0.028248712047934534,
      "epoch": 1.3408660888671875e-05,
      "step": 3515
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 3515,
      "training_step_time": 9.961225271224976
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 2.432389974594116,
      "step": 3515
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 3515,
      "training_step_time": 7.461970567703247
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 3.1329543590545654,
      "step": 3515
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 3515,
      "training_step_time": 9.909535646438599
    },
    {
      "epoch": 1.3408660888671875e-05,
      "model_forward_time": 3.1400763988494873,
      "step": 3515
    },
    {
      "epoch": 1.3408660888671875e-05,
      "step": 3515,
      "training_step_time": 10.279615640640259
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 3.4527244567871094,
      "step": 3516
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 3516,
      "training_step_time": 10.228772640228271
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 3.1329643726348877,
      "step": 3516
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 3516,
      "training_step_time": 9.904151439666748
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 3.1320159435272217,
      "step": 3516
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 3516,
      "training_step_time": 9.903144359588623
    },
    {
      "epoch": 1.34124755859375e-05,
      "model_forward_time": 2.4312946796417236,
      "step": 3516
    },
    {
      "epoch": 1.34124755859375e-05,
      "step": 3516,
      "training_step_time": 10.007239818572998
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 3.169689416885376,
      "step": 3517
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 3517,
      "training_step_time": 9.944756746292114
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 3.4056167602539062,
      "step": 3517
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 3517,
      "training_step_time": 10.177304744720459
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 3.1333563327789307,
      "step": 3517
    },
    {
      "dynamics_loss_avg": 0.07552562989294528,
      "epoch": 1.3416290283203125e-05,
      "step": 3517
    },
    {
      "action_loss_avg": 0.0322932374663651,
      "epoch": 1.3416290283203125e-05,
      "step": 3517
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 3517,
      "training_step_time": 9.90933346748352
    },
    {
      "epoch": 1.3416290283203125e-05,
      "model_forward_time": 3.134714126586914,
      "step": 3517
    },
    {
      "epoch": 1.3416290283203125e-05,
      "step": 3517,
      "training_step_time": 10.292999029159546
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 2.4440414905548096,
      "step": 3518
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 3518,
      "training_step_time": 7.4748375415802
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 3.1343109607696533,
      "step": 3518
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 3518,
      "training_step_time": 10.178460597991943
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 2.433922052383423,
      "step": 3518
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 3518,
      "training_step_time": 7.459723234176636
    },
    {
      "epoch": 1.342010498046875e-05,
      "model_forward_time": 2.430421829223633,
      "step": 3518
    },
    {
      "epoch": 1.342010498046875e-05,
      "step": 3518,
      "training_step_time": 10.009309768676758
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 3.137479543685913,
      "step": 3519
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 3519,
      "training_step_time": 9.90215539932251
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 3.1297988891601562,
      "step": 3519
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 3519,
      "training_step_time": 9.897471904754639
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 2.4343154430389404,
      "step": 3519
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 3519,
      "training_step_time": 7.459886074066162
    },
    {
      "epoch": 1.3423919677734375e-05,
      "model_forward_time": 3.4385604858398438,
      "step": 3519
    },
    {
      "epoch": 1.3423919677734375e-05,
      "step": 3519,
      "training_step_time": 10.316190958023071
    },
    {
      "epoch": 1.3427734375e-05,
      "grad_norm": 0.39318692684173584,
      "learning_rate": 1e-05,
      "loss": 0.0967,
      "step": 3520
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 2.445707321166992,
      "step": 3520
    },
    {
      "dynamics_loss_avg": 0.06550372913479804,
      "epoch": 1.3427734375e-05,
      "step": 3520
    },
    {
      "action_loss_avg": 0.030263775633648037,
      "epoch": 1.3427734375e-05,
      "step": 3520
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 3520,
      "training_step_time": 7.479454040527344
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 3.1386842727661133,
      "step": 3520
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 3520,
      "training_step_time": 9.913307428359985
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 2.4354710578918457,
      "step": 3520
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 3520,
      "training_step_time": 7.46699857711792
    },
    {
      "epoch": 1.3427734375e-05,
      "model_forward_time": 2.435872793197632,
      "step": 3520
    },
    {
      "epoch": 1.3427734375e-05,
      "step": 3520,
      "training_step_time": 10.025605201721191
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 3.425072431564331,
      "step": 3521
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 3521,
      "training_step_time": 10.199317216873169
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 3.1391379833221436,
      "step": 3521
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 3521,
      "training_step_time": 9.910085678100586
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 3.1359448432922363,
      "step": 3521
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 3521,
      "training_step_time": 9.906915664672852
    },
    {
      "epoch": 1.3431549072265625e-05,
      "model_forward_time": 3.134817123413086,
      "step": 3521
    },
    {
      "epoch": 1.3431549072265625e-05,
      "step": 3521,
      "training_step_time": 10.271175146102905
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 3.15010404586792,
      "step": 3522
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 3522,
      "training_step_time": 9.919790506362915
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 3.3920505046844482,
      "step": 3522
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 3522,
      "training_step_time": 10.165876865386963
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 3.140594959259033,
      "step": 3522
    },
    {
      "dynamics_loss_avg": 0.06768929958343506,
      "epoch": 1.343536376953125e-05,
      "step": 3522
    },
    {
      "action_loss_avg": 0.03698306679725647,
      "epoch": 1.343536376953125e-05,
      "step": 3522
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 3522,
      "training_step_time": 9.916633605957031
    },
    {
      "epoch": 1.343536376953125e-05,
      "model_forward_time": 3.1373214721679688,
      "step": 3522
    },
    {
      "epoch": 1.343536376953125e-05,
      "step": 3522,
      "training_step_time": 10.255186796188354
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 3.176861047744751,
      "step": 3523
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 3523,
      "training_step_time": 9.947941780090332
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 3.137103319168091,
      "step": 3523
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 3523,
      "training_step_time": 9.90677261352539
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 3.4118523597717285,
      "step": 3523
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 3523,
      "training_step_time": 10.180355310440063
    },
    {
      "epoch": 1.3439178466796876e-05,
      "model_forward_time": 3.133549928665161,
      "step": 3523
    },
    {
      "epoch": 1.3439178466796876e-05,
      "step": 3523,
      "training_step_time": 10.260558605194092
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 3.147963047027588,
      "step": 3524
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 3524,
      "training_step_time": 9.922402620315552
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 3.1373493671417236,
      "step": 3524
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 3524,
      "training_step_time": 9.908987760543823
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 3.1373860836029053,
      "step": 3524
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 3524,
      "training_step_time": 9.907711029052734
    },
    {
      "epoch": 1.34429931640625e-05,
      "model_forward_time": 3.3862414360046387,
      "step": 3524
    },
    {
      "epoch": 1.34429931640625e-05,
      "step": 3524,
      "training_step_time": 10.262767553329468
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 3.1439461708068848,
      "step": 3525
    },
    {
      "dynamics_loss_avg": 0.06775908377021551,
      "epoch": 1.3446807861328125e-05,
      "step": 3525
    },
    {
      "action_loss_avg": 0.027625096775591374,
      "epoch": 1.3446807861328125e-05,
      "step": 3525
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 3525,
      "training_step_time": 9.930108785629272
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 3.1378049850463867,
      "step": 3525
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 3525,
      "training_step_time": 9.911019325256348
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 3.1372859477996826,
      "step": 3525
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 3525,
      "training_step_time": 9.913516998291016
    },
    {
      "epoch": 1.3446807861328125e-05,
      "model_forward_time": 3.1387577056884766,
      "step": 3525
    },
    {
      "epoch": 1.3446807861328125e-05,
      "step": 3525,
      "training_step_time": 10.022548913955688
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 3.454160213470459,
      "step": 3526
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 3526,
      "training_step_time": 10.221203565597534
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 3.1365373134613037,
      "step": 3526
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 3526,
      "training_step_time": 9.903771162033081
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 3.13120436668396,
      "step": 3526
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 3526,
      "training_step_time": 9.896119594573975
    },
    {
      "epoch": 1.345062255859375e-05,
      "model_forward_time": 3.134093761444092,
      "step": 3526
    },
    {
      "epoch": 1.345062255859375e-05,
      "step": 3526,
      "training_step_time": 10.011130809783936
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 3.401585102081299,
      "step": 3527
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 3527,
      "training_step_time": 10.169867515563965
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 3.1402413845062256,
      "step": 3527
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 3527,
      "training_step_time": 9.911966800689697
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 3.1338016986846924,
      "step": 3527
    },
    {
      "dynamics_loss_avg": 0.061862630769610405,
      "epoch": 1.3454437255859376e-05,
      "step": 3527
    },
    {
      "action_loss_avg": 0.0311250289902091,
      "epoch": 1.3454437255859376e-05,
      "step": 3527
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 3527,
      "training_step_time": 9.904920101165771
    },
    {
      "epoch": 1.3454437255859376e-05,
      "model_forward_time": 3.1423490047454834,
      "step": 3527
    },
    {
      "epoch": 1.3454437255859376e-05,
      "step": 3527,
      "training_step_time": 10.296261072158813
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 3.1447458267211914,
      "step": 3528
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 3528,
      "training_step_time": 9.914440393447876
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 2.4323620796203613,
      "step": 3528
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 3528,
      "training_step_time": 7.462522506713867
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 3.378281831741333,
      "step": 3528
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 3528,
      "training_step_time": 10.141807079315186
    },
    {
      "epoch": 1.3458251953125e-05,
      "model_forward_time": 3.1331567764282227,
      "step": 3528
    },
    {
      "epoch": 1.3458251953125e-05,
      "step": 3528,
      "training_step_time": 10.263156175613403
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 3.142066240310669,
      "step": 3529
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 3529,
      "training_step_time": 9.904205083847046
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 3.1301791667938232,
      "step": 3529
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 3529,
      "training_step_time": 9.891628503799438
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 3.3913447856903076,
      "step": 3529
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 3529,
      "training_step_time": 10.153097867965698
    },
    {
      "epoch": 1.3462066650390625e-05,
      "model_forward_time": 3.140395402908325,
      "step": 3529
    },
    {
      "epoch": 1.3462066650390625e-05,
      "step": 3529,
      "training_step_time": 10.29921007156372
    },
    {
      "epoch": 1.346588134765625e-05,
      "grad_norm": 0.5898941159248352,
      "learning_rate": 1e-05,
      "loss": 0.0884,
      "step": 3530
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 3.144993543624878,
      "step": 3530
    },
    {
      "dynamics_loss_avg": 0.06539008356630802,
      "epoch": 1.346588134765625e-05,
      "step": 3530
    },
    {
      "action_loss_avg": 0.042238086462020874,
      "epoch": 1.346588134765625e-05,
      "step": 3530
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 3530,
      "training_step_time": 9.916308164596558
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 3.1322333812713623,
      "step": 3530
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 3530,
      "training_step_time": 9.900772094726562
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 3.1374449729919434,
      "step": 3530
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 3530,
      "training_step_time": 9.907909393310547
    },
    {
      "epoch": 1.346588134765625e-05,
      "model_forward_time": 3.1333603858947754,
      "step": 3530
    },
    {
      "epoch": 1.346588134765625e-05,
      "step": 3530,
      "training_step_time": 10.306559085845947
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 3.150254249572754,
      "step": 3531
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 3531,
      "training_step_time": 9.928584337234497
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 3.1421053409576416,
      "step": 3531
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 3531,
      "training_step_time": 9.918345928192139
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 3.1474204063415527,
      "step": 3531
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 3531,
      "training_step_time": 9.923985958099365
    },
    {
      "epoch": 1.3469696044921876e-05,
      "model_forward_time": 2.4370133876800537,
      "step": 3531
    },
    {
      "epoch": 1.3469696044921876e-05,
      "step": 3531,
      "training_step_time": 10.026418924331665
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 2.729581594467163,
      "step": 3532
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 3532,
      "training_step_time": 7.761705636978149
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 3.1415326595306396,
      "step": 3532
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 3532,
      "training_step_time": 9.918463230133057
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 3.152576208114624,
      "step": 3532
    },
    {
      "dynamics_loss_avg": 0.06314926855266094,
      "epoch": 1.34735107421875e-05,
      "step": 3532
    },
    {
      "action_loss_avg": 0.03359669838100672,
      "epoch": 1.34735107421875e-05,
      "step": 3532
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 3532,
      "training_step_time": 9.928643465042114
    },
    {
      "epoch": 1.34735107421875e-05,
      "model_forward_time": 3.137885570526123,
      "step": 3532
    },
    {
      "epoch": 1.34735107421875e-05,
      "step": 3532,
      "training_step_time": 10.274625539779663
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 3.1554298400878906,
      "step": 3533
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 3533,
      "training_step_time": 9.929358720779419
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 3.142544984817505,
      "step": 3533
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 3533,
      "training_step_time": 10.190273761749268
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 3.13946270942688,
      "step": 3533
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 3533,
      "training_step_time": 9.909288167953491
    },
    {
      "epoch": 1.3477325439453125e-05,
      "model_forward_time": 3.1351099014282227,
      "step": 3533
    },
    {
      "epoch": 1.3477325439453125e-05,
      "step": 3533,
      "training_step_time": 10.285566806793213
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 3.1610052585601807,
      "step": 3534
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 3534,
      "training_step_time": 9.927556037902832
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 2.435483455657959,
      "step": 3534
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 3534,
      "training_step_time": 7.456543207168579
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 3.134312152862549,
      "step": 3534
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 3534,
      "training_step_time": 10.194559812545776
    },
    {
      "epoch": 1.348114013671875e-05,
      "model_forward_time": 3.1362898349761963,
      "step": 3534
    },
    {
      "epoch": 1.348114013671875e-05,
      "step": 3534,
      "training_step_time": 10.003364324569702
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 3.1443638801574707,
      "step": 3535
    },
    {
      "dynamics_loss_avg": 0.05030140895396471,
      "epoch": 1.3484954833984376e-05,
      "step": 3535
    },
    {
      "action_loss_avg": 0.022878628317266704,
      "epoch": 1.3484954833984376e-05,
      "step": 3535
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 3535,
      "training_step_time": 9.907337665557861
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 3.136842727661133,
      "step": 3535
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 3535,
      "training_step_time": 9.90165901184082
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 3.1351940631866455,
      "step": 3535
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 3535,
      "training_step_time": 9.898592472076416
    },
    {
      "epoch": 1.3484954833984376e-05,
      "model_forward_time": 3.1406540870666504,
      "step": 3535
    },
    {
      "epoch": 1.3484954833984376e-05,
      "step": 3535,
      "training_step_time": 10.290894508361816
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 2.4434618949890137,
      "step": 3536
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 3536,
      "training_step_time": 7.467589616775513
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 3.135911226272583,
      "step": 3536
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 3536,
      "training_step_time": 9.901756048202515
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 3.1365716457366943,
      "step": 3536
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 3536,
      "training_step_time": 9.905819416046143
    },
    {
      "epoch": 1.348876953125e-05,
      "model_forward_time": 3.142124891281128,
      "step": 3536
    },
    {
      "epoch": 1.348876953125e-05,
      "step": 3536,
      "training_step_time": 10.30305027961731
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 3.1526596546173096,
      "step": 3537
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 3537,
      "training_step_time": 9.921125411987305
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 3.410385847091675,
      "step": 3537
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 3537,
      "training_step_time": 10.177185773849487
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 3.1355574131011963,
      "step": 3537
    },
    {
      "dynamics_loss_avg": 0.08331864979118109,
      "epoch": 1.3492584228515625e-05,
      "step": 3537
    },
    {
      "action_loss_avg": 0.04087769566103816,
      "epoch": 1.3492584228515625e-05,
      "step": 3537
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 3537,
      "training_step_time": 9.90534782409668
    },
    {
      "epoch": 1.3492584228515625e-05,
      "model_forward_time": 3.1424400806427,
      "step": 3537
    },
    {
      "epoch": 1.3492584228515625e-05,
      "step": 3537,
      "training_step_time": 10.239704608917236
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 3.182466983795166,
      "step": 3538
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 3538,
      "training_step_time": 9.964990377426147
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 3.4154999256134033,
      "step": 3538
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 3538,
      "training_step_time": 10.186135053634644
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 3.1368274688720703,
      "step": 3538
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 3538,
      "training_step_time": 9.908448219299316
    },
    {
      "epoch": 1.349639892578125e-05,
      "model_forward_time": 3.139554738998413,
      "step": 3538
    },
    {
      "epoch": 1.349639892578125e-05,
      "step": 3538,
      "training_step_time": 10.339858770370483
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 3.1440460681915283,
      "step": 3539
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 3539,
      "training_step_time": 9.920363903045654
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 3.13338303565979,
      "step": 3539
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 3539,
      "training_step_time": 9.90677809715271
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 3.141052484512329,
      "step": 3539
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 3539,
      "training_step_time": 10.19467306137085
    },
    {
      "epoch": 1.3500213623046876e-05,
      "model_forward_time": 3.1426243782043457,
      "step": 3539
    },
    {
      "epoch": 1.3500213623046876e-05,
      "step": 3539,
      "training_step_time": 10.023930311203003
    },
    {
      "epoch": 1.35040283203125e-05,
      "grad_norm": 0.7037481069564819,
      "learning_rate": 1e-05,
      "loss": 0.0946,
      "step": 3540
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 3.154622793197632,
      "step": 3540
    },
    {
      "dynamics_loss_avg": 0.08636266756802798,
      "epoch": 1.35040283203125e-05,
      "step": 3540
    },
    {
      "action_loss_avg": 0.04330444550141692,
      "epoch": 1.35040283203125e-05,
      "step": 3540
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 3540,
      "training_step_time": 9.936599969863892
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 3.1410470008850098,
      "step": 3540
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 3540,
      "training_step_time": 9.916914463043213
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 3.1385161876678467,
      "step": 3540
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 3540,
      "training_step_time": 9.916043281555176
    },
    {
      "epoch": 1.35040283203125e-05,
      "model_forward_time": 3.411400079727173,
      "step": 3540
    },
    {
      "epoch": 1.35040283203125e-05,
      "step": 3540,
      "training_step_time": 10.279730081558228
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 3.1602845191955566,
      "step": 3541
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 3541,
      "training_step_time": 9.93886137008667
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 3.142861843109131,
      "step": 3541
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 3541,
      "training_step_time": 9.920753479003906
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 3.1457884311676025,
      "step": 3541
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 3541,
      "training_step_time": 9.922484397888184
    },
    {
      "epoch": 1.3507843017578125e-05,
      "model_forward_time": 3.140312671661377,
      "step": 3541
    },
    {
      "epoch": 1.3507843017578125e-05,
      "step": 3541,
      "training_step_time": 10.319807291030884
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 3.4375839233398438,
      "step": 3542
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 3542,
      "training_step_time": 10.213919639587402
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 2.4375970363616943,
      "step": 3542
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 3542,
      "training_step_time": 7.470776081085205
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 3.140828847885132,
      "step": 3542
    },
    {
      "dynamics_loss_avg": 0.054003299213945866,
      "epoch": 1.351165771484375e-05,
      "step": 3542
    },
    {
      "action_loss_avg": 0.02627449478022754,
      "epoch": 1.351165771484375e-05,
      "step": 3542
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 3542,
      "training_step_time": 9.919869661331177
    },
    {
      "epoch": 1.351165771484375e-05,
      "model_forward_time": 2.4368858337402344,
      "step": 3542
    },
    {
      "epoch": 1.351165771484375e-05,
      "step": 3542,
      "training_step_time": 10.311274290084839
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 3.1461262702941895,
      "step": 3543
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 3543,
      "training_step_time": 9.918123722076416
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 3.133666515350342,
      "step": 3543
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 3543,
      "training_step_time": 10.171672344207764
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 3.1304879188537598,
      "step": 3543
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 3543,
      "training_step_time": 9.898128271102905
    },
    {
      "epoch": 1.3515472412109376e-05,
      "model_forward_time": 3.1307168006896973,
      "step": 3543
    },
    {
      "epoch": 1.3515472412109376e-05,
      "step": 3543,
      "training_step_time": 10.290190935134888
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 3.1457552909851074,
      "step": 3544
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 3544,
      "training_step_time": 9.916399240493774
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 3.1337125301361084,
      "step": 3544
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 3544,
      "training_step_time": 9.906020879745483
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 3.409533977508545,
      "step": 3544
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 3544,
      "training_step_time": 10.181854009628296
    },
    {
      "epoch": 1.3519287109375e-05,
      "model_forward_time": 3.13309383392334,
      "step": 3544
    },
    {
      "epoch": 1.3519287109375e-05,
      "step": 3544,
      "training_step_time": 10.018644571304321
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 3.1555817127227783,
      "step": 3545
    },
    {
      "dynamics_loss_avg": 0.0723675824701786,
      "epoch": 1.3523101806640625e-05,
      "step": 3545
    },
    {
      "action_loss_avg": 0.03224604185670614,
      "epoch": 1.3523101806640625e-05,
      "step": 3545
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 3545,
      "training_step_time": 9.931276321411133
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 3.1363489627838135,
      "step": 3545
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 3545,
      "training_step_time": 9.919678688049316
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 3.135340690612793,
      "step": 3545
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 3545,
      "training_step_time": 9.911458730697632
    },
    {
      "epoch": 1.3523101806640625e-05,
      "model_forward_time": 3.1377506256103516,
      "step": 3545
    },
    {
      "epoch": 1.3523101806640625e-05,
      "step": 3545,
      "training_step_time": 10.286294221878052
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 3.1798694133758545,
      "step": 3546
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 3546,
      "training_step_time": 9.958287715911865
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 3.1442394256591797,
      "step": 3546
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 3546,
      "training_step_time": 9.917548418045044
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 3.137885808944702,
      "step": 3546
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 3546,
      "training_step_time": 9.908066987991333
    },
    {
      "epoch": 1.352691650390625e-05,
      "model_forward_time": 3.1413352489471436,
      "step": 3546
    },
    {
      "epoch": 1.352691650390625e-05,
      "step": 3546,
      "training_step_time": 10.255683183670044
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 3.446742534637451,
      "step": 3547
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 3547,
      "training_step_time": 10.212898254394531
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 3.1446340084075928,
      "step": 3547
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 3547,
      "training_step_time": 9.917469501495361
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 3.141611337661743,
      "step": 3547
    },
    {
      "dynamics_loss_avg": 0.0548484206199646,
      "epoch": 1.3530731201171876e-05,
      "step": 3547
    },
    {
      "action_loss_avg": 0.03711715578101575,
      "epoch": 1.3530731201171876e-05,
      "step": 3547
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 3547,
      "training_step_time": 9.920005083084106
    },
    {
      "epoch": 1.3530731201171876e-05,
      "model_forward_time": 3.144098997116089,
      "step": 3547
    },
    {
      "epoch": 1.3530731201171876e-05,
      "step": 3547,
      "training_step_time": 10.254449129104614
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 3.1555392742156982,
      "step": 3548
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 3548,
      "training_step_time": 9.933465003967285
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 3.144573450088501,
      "step": 3548
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 3548,
      "training_step_time": 10.218270301818848
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 3.1458592414855957,
      "step": 3548
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 3548,
      "training_step_time": 9.924989461898804
    },
    {
      "epoch": 1.35345458984375e-05,
      "model_forward_time": 3.14985728263855,
      "step": 3548
    },
    {
      "epoch": 1.35345458984375e-05,
      "step": 3548,
      "training_step_time": 10.307936191558838
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 3.156468152999878,
      "step": 3549
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 3549,
      "training_step_time": 9.938328504562378
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 3.1451833248138428,
      "step": 3549
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 3549,
      "training_step_time": 9.926265478134155
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 2.4420692920684814,
      "step": 3549
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 3549,
      "training_step_time": 7.763308048248291
    },
    {
      "epoch": 1.3538360595703125e-05,
      "model_forward_time": 3.145226001739502,
      "step": 3549
    },
    {
      "epoch": 1.3538360595703125e-05,
      "step": 3549,
      "training_step_time": 10.320623636245728
    },
    {
      "epoch": 1.354217529296875e-05,
      "grad_norm": 0.6755155920982361,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3550
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 3.1521968841552734,
      "step": 3550
    },
    {
      "dynamics_loss_avg": 0.06370000299066306,
      "epoch": 1.354217529296875e-05,
      "step": 3550
    },
    {
      "action_loss_avg": 0.031173814181238414,
      "epoch": 1.354217529296875e-05,
      "step": 3550
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 3550,
      "training_step_time": 9.929300546646118
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 3.14038348197937,
      "step": 3550
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 3550,
      "training_step_time": 9.914618730545044
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 3.1448779106140137,
      "step": 3550
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 3550,
      "training_step_time": 9.916285276412964
    },
    {
      "epoch": 1.354217529296875e-05,
      "model_forward_time": 3.1409096717834473,
      "step": 3550
    },
    {
      "epoch": 1.354217529296875e-05,
      "step": 3550,
      "training_step_time": 10.273637056350708
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 3.1563408374786377,
      "step": 3551
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 3551,
      "training_step_time": 9.929839372634888
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 3.1433019638061523,
      "step": 3551
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 3551,
      "training_step_time": 9.916646003723145
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 3.145665407180786,
      "step": 3551
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 3551,
      "training_step_time": 9.920341491699219
    },
    {
      "epoch": 1.3545989990234376e-05,
      "model_forward_time": 2.4393177032470703,
      "step": 3551
    },
    {
      "epoch": 1.3545989990234376e-05,
      "step": 3551,
      "training_step_time": 10.027935981750488
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 3.433718204498291,
      "step": 3552
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 3552,
      "training_step_time": 10.20450496673584
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 3.1469123363494873,
      "step": 3552
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 3552,
      "training_step_time": 9.914443254470825
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 2.4301576614379883,
      "step": 3552
    },
    {
      "dynamics_loss_avg": 0.04981678705662489,
      "epoch": 1.35498046875e-05,
      "step": 3552
    },
    {
      "action_loss_avg": 0.02088105920702219,
      "epoch": 1.35498046875e-05,
      "step": 3552
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 3552,
      "training_step_time": 7.452383995056152
    },
    {
      "epoch": 1.35498046875e-05,
      "model_forward_time": 3.1289968490600586,
      "step": 3552
    },
    {
      "epoch": 1.35498046875e-05,
      "step": 3552,
      "training_step_time": 10.288488864898682
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 3.1482174396514893,
      "step": 3553
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 3553,
      "training_step_time": 9.908973932266235
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 3.1330933570861816,
      "step": 3553
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 3553,
      "training_step_time": 10.164926528930664
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 3.1323821544647217,
      "step": 3553
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 3553,
      "training_step_time": 9.897361040115356
    },
    {
      "epoch": 1.3553619384765625e-05,
      "model_forward_time": 3.1314172744750977,
      "step": 3553
    },
    {
      "epoch": 1.3553619384765625e-05,
      "step": 3553,
      "training_step_time": 10.28375792503357
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 3.1373679637908936,
      "step": 3554
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 3554,
      "training_step_time": 9.902405023574829
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 3.134611129760742,
      "step": 3554
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 3554,
      "training_step_time": 9.901769876480103
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 3.4127190113067627,
      "step": 3554
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 3554,
      "training_step_time": 10.183658838272095
    },
    {
      "epoch": 1.355743408203125e-05,
      "model_forward_time": 2.43766713142395,
      "step": 3554
    },
    {
      "epoch": 1.355743408203125e-05,
      "step": 3554,
      "training_step_time": 10.312342405319214
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 3.1780314445495605,
      "step": 3555
    },
    {
      "dynamics_loss_avg": 0.06669496316462756,
      "epoch": 1.3561248779296876e-05,
      "step": 3555
    },
    {
      "action_loss_avg": 0.03187074940651655,
      "epoch": 1.3561248779296876e-05,
      "step": 3555
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 3555,
      "training_step_time": 9.951030492782593
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 3.138516902923584,
      "step": 3555
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 3555,
      "training_step_time": 9.906196355819702
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 3.133992910385132,
      "step": 3555
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 3555,
      "training_step_time": 9.905071020126343
    },
    {
      "epoch": 1.3561248779296876e-05,
      "model_forward_time": 3.408390522003174,
      "step": 3555
    },
    {
      "epoch": 1.3561248779296876e-05,
      "step": 3555,
      "training_step_time": 10.27835488319397
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 3.170351266860962,
      "step": 3556
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 3556,
      "training_step_time": 9.942121744155884
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 2.433380126953125,
      "step": 3556
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 3556,
      "training_step_time": 7.460995197296143
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 3.1342780590057373,
      "step": 3556
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 3556,
      "training_step_time": 9.906155109405518
    },
    {
      "epoch": 1.35650634765625e-05,
      "model_forward_time": 3.1371967792510986,
      "step": 3556
    },
    {
      "epoch": 1.35650634765625e-05,
      "step": 3556,
      "training_step_time": 10.026248931884766
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 3.143754482269287,
      "step": 3557
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 3557,
      "training_step_time": 10.170495748519897
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 3.1381256580352783,
      "step": 3557
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 3557,
      "training_step_time": 9.910466432571411
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 3.1302382946014404,
      "step": 3557
    },
    {
      "dynamics_loss_avg": 0.06614947430789471,
      "epoch": 1.3568878173828125e-05,
      "step": 3557
    },
    {
      "action_loss_avg": 0.03416567426174879,
      "epoch": 1.3568878173828125e-05,
      "step": 3557
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 3557,
      "training_step_time": 9.899105787277222
    },
    {
      "epoch": 1.3568878173828125e-05,
      "model_forward_time": 3.1351706981658936,
      "step": 3557
    },
    {
      "epoch": 1.3568878173828125e-05,
      "step": 3557,
      "training_step_time": 10.261338710784912
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 2.4625794887542725,
      "step": 3558
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 3558,
      "training_step_time": 7.48767876625061
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 3.416168451309204,
      "step": 3558
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 3558,
      "training_step_time": 10.180412530899048
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 3.1340866088867188,
      "step": 3558
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 3558,
      "training_step_time": 9.899527788162231
    },
    {
      "epoch": 1.357269287109375e-05,
      "model_forward_time": 3.1383910179138184,
      "step": 3558
    },
    {
      "epoch": 1.357269287109375e-05,
      "step": 3558,
      "training_step_time": 10.287484645843506
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 3.1498847007751465,
      "step": 3559
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 3559,
      "training_step_time": 9.91853666305542
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 3.1355934143066406,
      "step": 3559
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 3559,
      "training_step_time": 9.907270669937134
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 3.135249614715576,
      "step": 3559
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 3559,
      "training_step_time": 10.173239707946777
    },
    {
      "epoch": 1.3576507568359376e-05,
      "model_forward_time": 3.137317657470703,
      "step": 3559
    },
    {
      "epoch": 1.3576507568359376e-05,
      "step": 3559,
      "training_step_time": 10.333992958068848
    },
    {
      "epoch": 1.3580322265625e-05,
      "grad_norm": 0.5674053430557251,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 3560
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 3.17771053314209,
      "step": 3560
    },
    {
      "dynamics_loss_avg": 0.06186278741806746,
      "epoch": 1.3580322265625e-05,
      "step": 3560
    },
    {
      "action_loss_avg": 0.03123055472970009,
      "epoch": 1.3580322265625e-05,
      "step": 3560
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 3560,
      "training_step_time": 9.950477838516235
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 3.1326982975006104,
      "step": 3560
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 3560,
      "training_step_time": 9.897952795028687
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 3.137676477432251,
      "step": 3560
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 3560,
      "training_step_time": 9.901991844177246
    },
    {
      "epoch": 1.3580322265625e-05,
      "model_forward_time": 3.4120595455169678,
      "step": 3560
    },
    {
      "epoch": 1.3580322265625e-05,
      "step": 3560,
      "training_step_time": 10.27043890953064
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 3.1710877418518066,
      "step": 3561
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 3561,
      "training_step_time": 9.938738107681274
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 3.1353442668914795,
      "step": 3561
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 3561,
      "training_step_time": 9.904390573501587
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 3.135772228240967,
      "step": 3561
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 3561,
      "training_step_time": 9.902793407440186
    },
    {
      "epoch": 1.3584136962890625e-05,
      "model_forward_time": 3.1357293128967285,
      "step": 3561
    },
    {
      "epoch": 1.3584136962890625e-05,
      "step": 3561,
      "training_step_time": 10.02527141571045
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 3.4216766357421875,
      "step": 3562
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 3562,
      "training_step_time": 10.194305658340454
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 3.1380321979522705,
      "step": 3562
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 3562,
      "training_step_time": 9.910370588302612
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 2.435401201248169,
      "step": 3562
    },
    {
      "dynamics_loss_avg": 0.07414183989167214,
      "epoch": 1.358795166015625e-05,
      "step": 3562
    },
    {
      "action_loss_avg": 0.03873060625046491,
      "epoch": 1.358795166015625e-05,
      "step": 3562
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 3562,
      "training_step_time": 7.468136548995972
    },
    {
      "epoch": 1.358795166015625e-05,
      "model_forward_time": 3.1288001537323,
      "step": 3562
    },
    {
      "epoch": 1.358795166015625e-05,
      "step": 3562,
      "training_step_time": 10.02314019203186
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 3.14743971824646,
      "step": 3563
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 3563,
      "training_step_time": 9.919090986251831
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 3.426948308944702,
      "step": 3563
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 3563,
      "training_step_time": 10.191807985305786
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 3.1351561546325684,
      "step": 3563
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 3563,
      "training_step_time": 9.90004014968872
    },
    {
      "epoch": 1.3591766357421876e-05,
      "model_forward_time": 3.1294760704040527,
      "step": 3563
    },
    {
      "epoch": 1.3591766357421876e-05,
      "step": 3563,
      "training_step_time": 10.276017427444458
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 3.1432816982269287,
      "step": 3564
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 3564,
      "training_step_time": 9.907834768295288
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 3.1360182762145996,
      "step": 3564
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 3564,
      "training_step_time": 9.903647184371948
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 3.1340248584747314,
      "step": 3564
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 3564,
      "training_step_time": 10.165839433670044
    },
    {
      "epoch": 1.35955810546875e-05,
      "model_forward_time": 3.1334400177001953,
      "step": 3564
    },
    {
      "epoch": 1.35955810546875e-05,
      "step": 3564,
      "training_step_time": 10.297189950942993
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 3.1416656970977783,
      "step": 3565
    },
    {
      "dynamics_loss_avg": 0.06455477774143219,
      "epoch": 1.3599395751953125e-05,
      "step": 3565
    },
    {
      "action_loss_avg": 0.03746727965772152,
      "epoch": 1.3599395751953125e-05,
      "step": 3565
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 3565,
      "training_step_time": 9.915544509887695
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 3.135190725326538,
      "step": 3565
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 3565,
      "training_step_time": 9.90187954902649
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 3.132390022277832,
      "step": 3565
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 3565,
      "training_step_time": 9.901150703430176
    },
    {
      "epoch": 1.3599395751953125e-05,
      "model_forward_time": 3.3927841186523438,
      "step": 3565
    },
    {
      "epoch": 1.3599395751953125e-05,
      "step": 3565,
      "training_step_time": 10.256167888641357
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 3.156294107437134,
      "step": 3566
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 3566,
      "training_step_time": 9.928165912628174
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 2.435013771057129,
      "step": 3566
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 3566,
      "training_step_time": 7.458177804946899
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 3.1351735591888428,
      "step": 3566
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 3566,
      "training_step_time": 9.90272331237793
    },
    {
      "epoch": 1.360321044921875e-05,
      "model_forward_time": 3.1394481658935547,
      "step": 3566
    },
    {
      "epoch": 1.360321044921875e-05,
      "step": 3566,
      "training_step_time": 10.25037932395935
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 3.4129254817962646,
      "step": 3567
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 3567,
      "training_step_time": 10.186758995056152
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 3.134521961212158,
      "step": 3567
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 3567,
      "training_step_time": 9.90960168838501
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 3.1355950832366943,
      "step": 3567
    },
    {
      "dynamics_loss_avg": 0.0639731852337718,
      "epoch": 1.3607025146484376e-05,
      "step": 3567
    },
    {
      "action_loss_avg": 0.03255136106163263,
      "epoch": 1.3607025146484376e-05,
      "step": 3567
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 3567,
      "training_step_time": 9.904188394546509
    },
    {
      "epoch": 1.3607025146484376e-05,
      "model_forward_time": 3.1361162662506104,
      "step": 3567
    },
    {
      "epoch": 1.3607025146484376e-05,
      "step": 3567,
      "training_step_time": 10.291160345077515
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 3.1494243144989014,
      "step": 3568
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 3568,
      "training_step_time": 9.91689944267273
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 3.1367363929748535,
      "step": 3568
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 3568,
      "training_step_time": 10.17237138748169
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 3.139314889907837,
      "step": 3568
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 3568,
      "training_step_time": 9.906632423400879
    },
    {
      "epoch": 1.361083984375e-05,
      "model_forward_time": 3.143544912338257,
      "step": 3568
    },
    {
      "epoch": 1.361083984375e-05,
      "step": 3568,
      "training_step_time": 10.27808952331543
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 3.1456286907196045,
      "step": 3569
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 3569,
      "training_step_time": 9.91989278793335
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 3.13950777053833,
      "step": 3569
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 3569,
      "training_step_time": 9.91727614402771
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 3.403691530227661,
      "step": 3569
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 3569,
      "training_step_time": 10.177065134048462
    },
    {
      "epoch": 1.3614654541015625e-05,
      "model_forward_time": 3.1494178771972656,
      "step": 3569
    },
    {
      "epoch": 1.3614654541015625e-05,
      "step": 3569,
      "training_step_time": 10.343595027923584
    },
    {
      "epoch": 1.361846923828125e-05,
      "grad_norm": 0.4173453450202942,
      "learning_rate": 1e-05,
      "loss": 0.0966,
      "step": 3570
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 3.1533710956573486,
      "step": 3570
    },
    {
      "dynamics_loss_avg": 0.06194733548909426,
      "epoch": 1.361846923828125e-05,
      "step": 3570
    },
    {
      "action_loss_avg": 0.031113449716940522,
      "epoch": 1.361846923828125e-05,
      "step": 3570
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 3570,
      "training_step_time": 9.929861068725586
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 3.141061305999756,
      "step": 3570
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 3570,
      "training_step_time": 9.909215211868286
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 3.141256093978882,
      "step": 3570
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 3570,
      "training_step_time": 9.911123991012573
    },
    {
      "epoch": 1.361846923828125e-05,
      "model_forward_time": 3.1409826278686523,
      "step": 3570
    },
    {
      "epoch": 1.361846923828125e-05,
      "step": 3570,
      "training_step_time": 10.257910966873169
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 2.4504127502441406,
      "step": 3571
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 3571,
      "training_step_time": 7.48438024520874
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 3.135483980178833,
      "step": 3571
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 3571,
      "training_step_time": 9.902517557144165
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 3.1397640705108643,
      "step": 3571
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 3571,
      "training_step_time": 9.906252384185791
    },
    {
      "epoch": 1.3622283935546876e-05,
      "model_forward_time": 2.436187982559204,
      "step": 3571
    },
    {
      "epoch": 1.3622283935546876e-05,
      "step": 3571,
      "training_step_time": 10.021399021148682
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 3.4239861965179443,
      "step": 3572
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 3572,
      "training_step_time": 10.195718050003052
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 3.13466477394104,
      "step": 3572
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 3572,
      "training_step_time": 9.901201725006104
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 3.1353869438171387,
      "step": 3572
    },
    {
      "dynamics_loss_avg": 0.0596118064597249,
      "epoch": 1.36260986328125e-05,
      "step": 3572
    },
    {
      "action_loss_avg": 0.031043751211836935,
      "epoch": 1.36260986328125e-05,
      "step": 3572
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 3572,
      "training_step_time": 9.90772008895874
    },
    {
      "epoch": 1.36260986328125e-05,
      "model_forward_time": 3.135302782058716,
      "step": 3572
    },
    {
      "epoch": 1.36260986328125e-05,
      "step": 3572,
      "training_step_time": 10.282401084899902
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 2.4622461795806885,
      "step": 3573
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 3573,
      "training_step_time": 7.506683111190796
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 3.139838457107544,
      "step": 3573
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 3573,
      "training_step_time": 9.910862684249878
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 3.4004783630371094,
      "step": 3573
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 3573,
      "training_step_time": 10.169221639633179
    },
    {
      "epoch": 1.3629913330078125e-05,
      "model_forward_time": 3.1325907707214355,
      "step": 3573
    },
    {
      "epoch": 1.3629913330078125e-05,
      "step": 3573,
      "training_step_time": 10.314054489135742
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 3.1526548862457275,
      "step": 3574
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 3574,
      "training_step_time": 9.92229151725769
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 3.1383633613586426,
      "step": 3574
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 3574,
      "training_step_time": 9.908384323120117
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 3.1359691619873047,
      "step": 3574
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 3574,
      "training_step_time": 10.226916790008545
    },
    {
      "epoch": 1.363372802734375e-05,
      "model_forward_time": 3.132701873779297,
      "step": 3574
    },
    {
      "epoch": 1.363372802734375e-05,
      "step": 3574,
      "training_step_time": 10.312174797058105
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 2.449338912963867,
      "step": 3575
    },
    {
      "dynamics_loss_avg": 0.05575280282646418,
      "epoch": 1.3637542724609376e-05,
      "step": 3575
    },
    {
      "action_loss_avg": 0.020791646279394626,
      "epoch": 1.3637542724609376e-05,
      "step": 3575
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 3575,
      "training_step_time": 7.479111194610596
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 3.1365532875061035,
      "step": 3575
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 3575,
      "training_step_time": 9.907249450683594
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 3.1365792751312256,
      "step": 3575
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 3575,
      "training_step_time": 9.906993865966797
    },
    {
      "epoch": 1.3637542724609376e-05,
      "model_forward_time": 3.141899585723877,
      "step": 3575
    },
    {
      "epoch": 1.3637542724609376e-05,
      "step": 3575,
      "training_step_time": 10.322480916976929
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 3.4163317680358887,
      "step": 3576
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 3576,
      "training_step_time": 10.186986446380615
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 3.142176628112793,
      "step": 3576
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 3576,
      "training_step_time": 9.917322635650635
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 3.1319632530212402,
      "step": 3576
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 3576,
      "training_step_time": 9.895705699920654
    },
    {
      "epoch": 1.3641357421875e-05,
      "model_forward_time": 3.131772994995117,
      "step": 3576
    },
    {
      "epoch": 1.3641357421875e-05,
      "step": 3576,
      "training_step_time": 10.028280973434448
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 3.149122714996338,
      "step": 3577
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 3577,
      "training_step_time": 10.17501974105835
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 3.135429620742798,
      "step": 3577
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 3577,
      "training_step_time": 9.898984909057617
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 3.1311564445495605,
      "step": 3577
    },
    {
      "dynamics_loss_avg": 0.0629806885495782,
      "epoch": 1.3645172119140625e-05,
      "step": 3577
    },
    {
      "action_loss_avg": 0.039652309473603965,
      "epoch": 1.3645172119140625e-05,
      "step": 3577
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 3577,
      "training_step_time": 9.900583744049072
    },
    {
      "epoch": 1.3645172119140625e-05,
      "model_forward_time": 3.1334877014160156,
      "step": 3577
    },
    {
      "epoch": 1.3645172119140625e-05,
      "step": 3577,
      "training_step_time": 10.023530721664429
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 2.466398239135742,
      "step": 3578
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 3578,
      "training_step_time": 7.4864068031311035
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 3.451392650604248,
      "step": 3578
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 3578,
      "training_step_time": 10.222019672393799
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 3.1328530311584473,
      "step": 3578
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 3578,
      "training_step_time": 9.900096416473389
    },
    {
      "epoch": 1.364898681640625e-05,
      "model_forward_time": 3.134796142578125,
      "step": 3578
    },
    {
      "epoch": 1.364898681640625e-05,
      "step": 3578,
      "training_step_time": 10.034419536590576
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 3.1759393215179443,
      "step": 3579
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 3579,
      "training_step_time": 9.95268201828003
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 3.134654998779297,
      "step": 3579
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 3579,
      "training_step_time": 9.903878688812256
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 3.1365911960601807,
      "step": 3579
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 3579,
      "training_step_time": 10.181623697280884
    },
    {
      "epoch": 1.3652801513671876e-05,
      "model_forward_time": 3.1370720863342285,
      "step": 3579
    },
    {
      "epoch": 1.3652801513671876e-05,
      "step": 3579,
      "training_step_time": 10.327991962432861
    },
    {
      "epoch": 1.36566162109375e-05,
      "grad_norm": 0.4814984202384949,
      "learning_rate": 1e-05,
      "loss": 0.0957,
      "step": 3580
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 3.149843692779541,
      "step": 3580
    },
    {
      "dynamics_loss_avg": 0.06533026173710824,
      "epoch": 1.36566162109375e-05,
      "step": 3580
    },
    {
      "action_loss_avg": 0.027482274128124116,
      "epoch": 1.36566162109375e-05,
      "step": 3580
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 3580,
      "training_step_time": 9.915140151977539
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 3.1331803798675537,
      "step": 3580
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 3580,
      "training_step_time": 9.8918297290802
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 3.1359071731567383,
      "step": 3580
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 3580,
      "training_step_time": 9.897575855255127
    },
    {
      "epoch": 1.36566162109375e-05,
      "model_forward_time": 3.418290138244629,
      "step": 3580
    },
    {
      "epoch": 1.36566162109375e-05,
      "step": 3580,
      "training_step_time": 10.273773193359375
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 3.1415600776672363,
      "step": 3581
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 3581,
      "training_step_time": 9.902405738830566
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 3.1303091049194336,
      "step": 3581
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 3581,
      "training_step_time": 9.888718843460083
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 3.1325416564941406,
      "step": 3581
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 3581,
      "training_step_time": 9.89365816116333
    },
    {
      "epoch": 1.3660430908203125e-05,
      "model_forward_time": 3.1319901943206787,
      "step": 3581
    },
    {
      "epoch": 1.3660430908203125e-05,
      "step": 3581,
      "training_step_time": 10.022345542907715
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 3.416912317276001,
      "step": 3582
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 3582,
      "training_step_time": 10.180330514907837
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 3.1334469318389893,
      "step": 3582
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 3582,
      "training_step_time": 9.899166345596313
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 3.1395864486694336,
      "step": 3582
    },
    {
      "dynamics_loss_avg": 0.07313193790614606,
      "epoch": 1.366424560546875e-05,
      "step": 3582
    },
    {
      "action_loss_avg": 0.031349827721714976,
      "epoch": 1.366424560546875e-05,
      "step": 3582
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 3582,
      "training_step_time": 9.913015842437744
    },
    {
      "epoch": 1.366424560546875e-05,
      "model_forward_time": 3.136615037918091,
      "step": 3582
    },
    {
      "epoch": 1.366424560546875e-05,
      "step": 3582,
      "training_step_time": 10.254032850265503
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 3.1532084941864014,
      "step": 3583
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 3583,
      "training_step_time": 9.928322553634644
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 2.7163031101226807,
      "step": 3583
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 3583,
      "training_step_time": 7.749013662338257
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 2.4329111576080322,
      "step": 3583
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 3583,
      "training_step_time": 7.465091705322266
    },
    {
      "epoch": 1.3668060302734376e-05,
      "model_forward_time": 3.136885404586792,
      "step": 3583
    },
    {
      "epoch": 1.3668060302734376e-05,
      "step": 3583,
      "training_step_time": 10.312591552734375
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 3.1531901359558105,
      "step": 3584
    },
    {
      "epoch": 1.3671875e-05,
      "step": 3584,
      "training_step_time": 9.927629232406616
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 2.435655355453491,
      "step": 3584
    },
    {
      "epoch": 1.3671875e-05,
      "step": 3584,
      "training_step_time": 7.463294982910156
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 3.129006862640381,
      "step": 3584
    },
    {
      "epoch": 1.3671875e-05,
      "step": 3584,
      "training_step_time": 9.892780542373657
    },
    {
      "epoch": 1.3671875e-05,
      "model_forward_time": 3.4277944564819336,
      "step": 3584
    },
    {
      "epoch": 1.3671875e-05,
      "step": 3584,
      "training_step_time": 10.28721833229065
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 3.137357473373413,
      "step": 3585
    },
    {
      "dynamics_loss_avg": 0.06605379581451416,
      "epoch": 1.3675689697265625e-05,
      "step": 3585
    },
    {
      "action_loss_avg": 0.028917251154780388,
      "epoch": 1.3675689697265625e-05,
      "step": 3585
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 3585,
      "training_step_time": 9.90702199935913
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 3.133599042892456,
      "step": 3585
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 3585,
      "training_step_time": 9.897011756896973
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 3.133385181427002,
      "step": 3585
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 3585,
      "training_step_time": 9.894005298614502
    },
    {
      "epoch": 1.3675689697265625e-05,
      "model_forward_time": 3.1364431381225586,
      "step": 3585
    },
    {
      "epoch": 1.3675689697265625e-05,
      "step": 3585,
      "training_step_time": 10.31188154220581
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 3.4312798976898193,
      "step": 3586
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 3586,
      "training_step_time": 10.195066690444946
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 3.134463310241699,
      "step": 3586
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 3586,
      "training_step_time": 9.89626669883728
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 3.1331024169921875,
      "step": 3586
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 3586,
      "training_step_time": 9.899517297744751
    },
    {
      "epoch": 1.367950439453125e-05,
      "model_forward_time": 3.1365013122558594,
      "step": 3586
    },
    {
      "epoch": 1.367950439453125e-05,
      "step": 3586,
      "training_step_time": 10.310372352600098
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 3.424353837966919,
      "step": 3587
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 3587,
      "training_step_time": 10.191120624542236
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 3.13996958732605,
      "step": 3587
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 3587,
      "training_step_time": 9.906114101409912
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 2.432722806930542,
      "step": 3587
    },
    {
      "dynamics_loss_avg": 0.06278970446437597,
      "epoch": 1.3683319091796876e-05,
      "step": 3587
    },
    {
      "action_loss_avg": 0.033508325554430486,
      "epoch": 1.3683319091796876e-05,
      "step": 3587
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 3587,
      "training_step_time": 7.457637310028076
    },
    {
      "epoch": 1.3683319091796876e-05,
      "model_forward_time": 3.136849880218506,
      "step": 3587
    },
    {
      "epoch": 1.3683319091796876e-05,
      "step": 3587,
      "training_step_time": 10.032679796218872
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 2.455399751663208,
      "step": 3588
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 3588,
      "training_step_time": 7.488062381744385
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 3.1323490142822266,
      "step": 3588
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 3588,
      "training_step_time": 9.901533842086792
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 3.3839640617370605,
      "step": 3588
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 3588,
      "training_step_time": 10.143711805343628
    },
    {
      "epoch": 1.36871337890625e-05,
      "model_forward_time": 3.139616012573242,
      "step": 3588
    },
    {
      "epoch": 1.36871337890625e-05,
      "step": 3588,
      "training_step_time": 10.274537801742554
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 3.1453750133514404,
      "step": 3589
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 3589,
      "training_step_time": 9.908820867538452
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 3.1355230808258057,
      "step": 3589
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 3589,
      "training_step_time": 9.898717641830444
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 3.139009952545166,
      "step": 3589
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 3589,
      "training_step_time": 10.171796321868896
    },
    {
      "epoch": 1.3690948486328125e-05,
      "model_forward_time": 2.434115171432495,
      "step": 3589
    },
    {
      "epoch": 1.3690948486328125e-05,
      "step": 3589,
      "training_step_time": 10.028308629989624
    },
    {
      "epoch": 1.369476318359375e-05,
      "grad_norm": 0.7135710120201111,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 3590
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 3.155778408050537,
      "step": 3590
    },
    {
      "dynamics_loss_avg": 0.06964468900114298,
      "epoch": 1.369476318359375e-05,
      "step": 3590
    },
    {
      "action_loss_avg": 0.03227805150672793,
      "epoch": 1.369476318359375e-05,
      "step": 3590
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 3590,
      "training_step_time": 9.933292388916016
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 3.1334993839263916,
      "step": 3590
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 3590,
      "training_step_time": 9.898889780044556
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 3.1394007205963135,
      "step": 3590
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 3590,
      "training_step_time": 9.901726722717285
    },
    {
      "epoch": 1.369476318359375e-05,
      "model_forward_time": 3.136078119277954,
      "step": 3590
    },
    {
      "epoch": 1.369476318359375e-05,
      "step": 3590,
      "training_step_time": 10.32197093963623
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 3.4838953018188477,
      "step": 3591
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 3591,
      "training_step_time": 10.24796462059021
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 3.133958101272583,
      "step": 3591
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 3591,
      "training_step_time": 9.893959522247314
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 3.137502670288086,
      "step": 3591
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 3591,
      "training_step_time": 9.901573657989502
    },
    {
      "epoch": 1.3698577880859376e-05,
      "model_forward_time": 3.1407814025878906,
      "step": 3591
    },
    {
      "epoch": 1.3698577880859376e-05,
      "step": 3591,
      "training_step_time": 10.30448842048645
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 3.445160150527954,
      "step": 3592
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 3592,
      "training_step_time": 10.215887069702148
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 3.1411218643188477,
      "step": 3592
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 3592,
      "training_step_time": 9.909233093261719
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 3.1407864093780518,
      "step": 3592
    },
    {
      "dynamics_loss_avg": 0.06739763580262662,
      "epoch": 1.3702392578125e-05,
      "step": 3592
    },
    {
      "action_loss_avg": 0.032235019560903314,
      "epoch": 1.3702392578125e-05,
      "step": 3592
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 3592,
      "training_step_time": 9.91268515586853
    },
    {
      "epoch": 1.3702392578125e-05,
      "model_forward_time": 3.137444257736206,
      "step": 3592
    },
    {
      "epoch": 1.3702392578125e-05,
      "step": 3592,
      "training_step_time": 10.013327598571777
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 3.144428014755249,
      "step": 3593
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 3593,
      "training_step_time": 9.916633605957031
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 3.4033825397491455,
      "step": 3593
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 3593,
      "training_step_time": 10.175460815429688
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 3.1427321434020996,
      "step": 3593
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 3593,
      "training_step_time": 9.911207914352417
    },
    {
      "epoch": 1.3706207275390625e-05,
      "model_forward_time": 2.435519218444824,
      "step": 3593
    },
    {
      "epoch": 1.3706207275390625e-05,
      "step": 3593,
      "training_step_time": 10.2882719039917
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 3.1498868465423584,
      "step": 3594
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 3594,
      "training_step_time": 9.919610023498535
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 3.1379170417785645,
      "step": 3594
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 3594,
      "training_step_time": 9.904951333999634
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 3.1379289627075195,
      "step": 3594
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 3594,
      "training_step_time": 10.188495635986328
    },
    {
      "epoch": 1.371002197265625e-05,
      "model_forward_time": 3.1322131156921387,
      "step": 3594
    },
    {
      "epoch": 1.371002197265625e-05,
      "step": 3594,
      "training_step_time": 10.014296770095825
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 3.1477973461151123,
      "step": 3595
    },
    {
      "dynamics_loss_avg": 0.07343269512057304,
      "epoch": 1.3713836669921876e-05,
      "step": 3595
    },
    {
      "action_loss_avg": 0.030305511225014924,
      "epoch": 1.3713836669921876e-05,
      "step": 3595
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 3595,
      "training_step_time": 9.915355682373047
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 3.1369996070861816,
      "step": 3595
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 3595,
      "training_step_time": 9.90386176109314
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 3.1306915283203125,
      "step": 3595
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 3595,
      "training_step_time": 9.896293878555298
    },
    {
      "epoch": 1.3713836669921876e-05,
      "model_forward_time": 2.4332275390625,
      "step": 3595
    },
    {
      "epoch": 1.3713836669921876e-05,
      "step": 3595,
      "training_step_time": 10.28598165512085
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 3.4444832801818848,
      "step": 3596
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 3596,
      "training_step_time": 10.220597505569458
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 2.453044891357422,
      "step": 3596
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 3596,
      "training_step_time": 7.486533164978027
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 3.132718324661255,
      "step": 3596
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 3596,
      "training_step_time": 9.897062301635742
    },
    {
      "epoch": 1.37176513671875e-05,
      "model_forward_time": 3.13961124420166,
      "step": 3596
    },
    {
      "epoch": 1.37176513671875e-05,
      "step": 3596,
      "training_step_time": 10.014986753463745
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 3.151350498199463,
      "step": 3597
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 3597,
      "training_step_time": 9.91611385345459
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 3.4023337364196777,
      "step": 3597
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 3597,
      "training_step_time": 10.166072845458984
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 3.1313552856445312,
      "step": 3597
    },
    {
      "dynamics_loss_avg": 0.06932801641523838,
      "epoch": 1.3721466064453125e-05,
      "step": 3597
    },
    {
      "action_loss_avg": 0.03707325560972095,
      "epoch": 1.3721466064453125e-05,
      "step": 3597
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 3597,
      "training_step_time": 9.897918701171875
    },
    {
      "epoch": 1.3721466064453125e-05,
      "model_forward_time": 3.1360437870025635,
      "step": 3597
    },
    {
      "epoch": 1.3721466064453125e-05,
      "step": 3597,
      "training_step_time": 10.271806716918945
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 3.1395809650421143,
      "step": 3598
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 3598,
      "training_step_time": 9.90721583366394
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 3.408968925476074,
      "step": 3598
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 3598,
      "training_step_time": 10.174402236938477
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 3.1322290897369385,
      "step": 3598
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 3598,
      "training_step_time": 9.89655590057373
    },
    {
      "epoch": 1.372528076171875e-05,
      "model_forward_time": 3.134992837905884,
      "step": 3598
    },
    {
      "epoch": 1.372528076171875e-05,
      "step": 3598,
      "training_step_time": 10.272949934005737
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 3.1480650901794434,
      "step": 3599
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 3599,
      "training_step_time": 9.918315410614014
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 3.133709192276001,
      "step": 3599
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 3599,
      "training_step_time": 9.904382944107056
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 3.140509605407715,
      "step": 3599
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 3599,
      "training_step_time": 10.180113554000854
    },
    {
      "epoch": 1.3729095458984376e-05,
      "model_forward_time": 3.1438465118408203,
      "step": 3599
    },
    {
      "epoch": 1.3729095458984376e-05,
      "step": 3599,
      "training_step_time": 10.04275131225586
    },
    {
      "epoch": 1.373291015625e-05,
      "grad_norm": 0.4423436224460602,
      "learning_rate": 1e-05,
      "loss": 0.0955,
      "step": 3600
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 3.1503639221191406,
      "step": 3600
    },
    {
      "dynamics_loss_avg": 0.05405071973800659,
      "epoch": 1.373291015625e-05,
      "step": 3600
    },
    {
      "action_loss_avg": 0.025091536715626715,
      "epoch": 1.373291015625e-05,
      "step": 3600
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 3600,
      "training_step_time": 9.923894166946411
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 3.1425249576568604,
      "step": 3600
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 3600,
      "training_step_time": 9.915849447250366
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 3.1406354904174805,
      "step": 3600
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 3600,
      "training_step_time": 9.91353988647461
    },
    {
      "epoch": 1.373291015625e-05,
      "model_forward_time": 3.4236032962799072,
      "step": 3600
    },
    {
      "epoch": 1.373291015625e-05,
      "step": 3600,
      "training_step_time": 10.287774801254272
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 3.1596853733062744,
      "step": 3601
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 3601,
      "training_step_time": 9.932124376296997
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 3.1403844356536865,
      "step": 3601
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 3601,
      "training_step_time": 9.909690618515015
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 3.147656202316284,
      "step": 3601
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 3601,
      "training_step_time": 9.914958953857422
    },
    {
      "epoch": 1.3736724853515625e-05,
      "model_forward_time": 3.1427290439605713,
      "step": 3601
    },
    {
      "epoch": 1.3736724853515625e-05,
      "step": 3601,
      "training_step_time": 10.349924087524414
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 3.4865403175354004,
      "step": 3602
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 3602,
      "training_step_time": 10.259014129638672
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 3.138716220855713,
      "step": 3602
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 3602,
      "training_step_time": 9.913968801498413
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 3.143658399581909,
      "step": 3602
    },
    {
      "dynamics_loss_avg": 0.05975890085101128,
      "epoch": 1.374053955078125e-05,
      "step": 3602
    },
    {
      "action_loss_avg": 0.0297935510519892,
      "epoch": 1.374053955078125e-05,
      "step": 3602
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 3602,
      "training_step_time": 9.917299032211304
    },
    {
      "epoch": 1.374053955078125e-05,
      "model_forward_time": 3.1392054557800293,
      "step": 3602
    },
    {
      "epoch": 1.374053955078125e-05,
      "step": 3602,
      "training_step_time": 10.32329511642456
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 3.1808245182037354,
      "step": 3603
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 3603,
      "training_step_time": 9.951741695404053
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 2.718360424041748,
      "step": 3603
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 3603,
      "training_step_time": 7.738993167877197
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 3.1332075595855713,
      "step": 3603
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 3603,
      "training_step_time": 9.89174747467041
    },
    {
      "epoch": 1.3744354248046876e-05,
      "model_forward_time": 3.138779640197754,
      "step": 3603
    },
    {
      "epoch": 1.3744354248046876e-05,
      "step": 3603,
      "training_step_time": 10.014274835586548
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 3.140333414077759,
      "step": 3604
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 3604,
      "training_step_time": 9.906489610671997
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 3.1380391120910645,
      "step": 3604
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 3604,
      "training_step_time": 9.901004552841187
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 3.1377508640289307,
      "step": 3604
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 3604,
      "training_step_time": 10.173082113265991
    },
    {
      "epoch": 1.37481689453125e-05,
      "model_forward_time": 3.136946201324463,
      "step": 3604
    },
    {
      "epoch": 1.37481689453125e-05,
      "step": 3604,
      "training_step_time": 10.02118182182312
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 3.137404441833496,
      "step": 3605
    },
    {
      "dynamics_loss_avg": 0.07052553594112396,
      "epoch": 1.3751983642578125e-05,
      "step": 3605
    },
    {
      "action_loss_avg": 0.0258555362932384,
      "epoch": 1.3751983642578125e-05,
      "step": 3605
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 3605,
      "training_step_time": 9.902427911758423
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 3.1319775581359863,
      "step": 3605
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 3605,
      "training_step_time": 9.89762258529663
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 3.132220506668091,
      "step": 3605
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 3605,
      "training_step_time": 9.89663815498352
    },
    {
      "epoch": 1.3751983642578125e-05,
      "model_forward_time": 2.4296371936798096,
      "step": 3605
    },
    {
      "epoch": 1.3751983642578125e-05,
      "step": 3605,
      "training_step_time": 10.024503231048584
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 3.162885904312134,
      "step": 3606
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 3606,
      "training_step_time": 9.934374570846558
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 2.438880681991577,
      "step": 3606
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 3606,
      "training_step_time": 7.469472169876099
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 3.140150785446167,
      "step": 3606
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 3606,
      "training_step_time": 9.911283731460571
    },
    {
      "epoch": 1.375579833984375e-05,
      "model_forward_time": 3.146218776702881,
      "step": 3606
    },
    {
      "epoch": 1.375579833984375e-05,
      "step": 3606,
      "training_step_time": 10.349927425384521
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 3.153757095336914,
      "step": 3607
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 3607,
      "training_step_time": 10.18501091003418
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 3.147735595703125,
      "step": 3607
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 3607,
      "training_step_time": 9.921279907226562
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 3.1487908363342285,
      "step": 3607
    },
    {
      "dynamics_loss_avg": 0.06793758254498243,
      "epoch": 1.3759613037109376e-05,
      "step": 3607
    },
    {
      "action_loss_avg": 0.03245985144749284,
      "epoch": 1.3759613037109376e-05,
      "step": 3607
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 3607,
      "training_step_time": 9.924413442611694
    },
    {
      "epoch": 1.3759613037109376e-05,
      "model_forward_time": 3.1483216285705566,
      "step": 3607
    },
    {
      "epoch": 1.3759613037109376e-05,
      "step": 3607,
      "training_step_time": 10.279489278793335
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 3.173518180847168,
      "step": 3608
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 3608,
      "training_step_time": 9.945208549499512
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 3.1359713077545166,
      "step": 3608
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 3608,
      "training_step_time": 10.20259428024292
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 3.135894536972046,
      "step": 3608
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 3608,
      "training_step_time": 9.911324501037598
    },
    {
      "epoch": 1.3763427734375e-05,
      "model_forward_time": 3.146350860595703,
      "step": 3608
    },
    {
      "epoch": 1.3763427734375e-05,
      "step": 3608,
      "training_step_time": 10.340213060379028
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 3.180410623550415,
      "step": 3609
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 3609,
      "training_step_time": 9.950792074203491
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 3.1423842906951904,
      "step": 3609
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 3609,
      "training_step_time": 9.911191463470459
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 3.4040822982788086,
      "step": 3609
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 3609,
      "training_step_time": 10.185324907302856
    },
    {
      "epoch": 1.3767242431640625e-05,
      "model_forward_time": 3.1488351821899414,
      "step": 3609
    },
    {
      "epoch": 1.3767242431640625e-05,
      "step": 3609,
      "training_step_time": 10.022814273834229
    },
    {
      "epoch": 1.377105712890625e-05,
      "grad_norm": 0.40760987997055054,
      "learning_rate": 1e-05,
      "loss": 0.0924,
      "step": 3610
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 3.1512818336486816,
      "step": 3610
    },
    {
      "dynamics_loss_avg": 0.0404352031648159,
      "epoch": 1.377105712890625e-05,
      "step": 3610
    },
    {
      "action_loss_avg": 0.024842068506404757,
      "epoch": 1.377105712890625e-05,
      "step": 3610
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 3610,
      "training_step_time": 9.925933122634888
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 3.14066743850708,
      "step": 3610
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 3610,
      "training_step_time": 9.91211485862732
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 3.145188808441162,
      "step": 3610
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 3610,
      "training_step_time": 9.918615579605103
    },
    {
      "epoch": 1.377105712890625e-05,
      "model_forward_time": 3.1435320377349854,
      "step": 3610
    },
    {
      "epoch": 1.377105712890625e-05,
      "step": 3610,
      "training_step_time": 10.290264368057251
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 3.1526939868927,
      "step": 3611
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 3611,
      "training_step_time": 9.938730478286743
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 3.137110948562622,
      "step": 3611
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 3611,
      "training_step_time": 9.91152048110962
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 3.1370534896850586,
      "step": 3611
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 3611,
      "training_step_time": 9.910519361495972
    },
    {
      "epoch": 1.3774871826171876e-05,
      "model_forward_time": 3.1330161094665527,
      "step": 3611
    },
    {
      "epoch": 1.3774871826171876e-05,
      "step": 3611,
      "training_step_time": 10.30681300163269
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 3.448122024536133,
      "step": 3612
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 3612,
      "training_step_time": 10.221172571182251
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 3.1367123126983643,
      "step": 3612
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 3612,
      "training_step_time": 9.90440034866333
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 3.138641119003296,
      "step": 3612
    },
    {
      "dynamics_loss_avg": 0.04494011932983995,
      "epoch": 1.37786865234375e-05,
      "step": 3612
    },
    {
      "action_loss_avg": 0.025506063271313904,
      "epoch": 1.37786865234375e-05,
      "step": 3612
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 3612,
      "training_step_time": 9.912723779678345
    },
    {
      "epoch": 1.37786865234375e-05,
      "model_forward_time": 3.1397294998168945,
      "step": 3612
    },
    {
      "epoch": 1.37786865234375e-05,
      "step": 3612,
      "training_step_time": 10.304184198379517
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 3.171052932739258,
      "step": 3613
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 3613,
      "training_step_time": 9.947060108184814
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 3.3946239948272705,
      "step": 3613
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 3613,
      "training_step_time": 10.161107540130615
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 3.1387221813201904,
      "step": 3613
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 3613,
      "training_step_time": 9.915627241134644
    },
    {
      "epoch": 1.3782501220703125e-05,
      "model_forward_time": 3.132612466812134,
      "step": 3613
    },
    {
      "epoch": 1.3782501220703125e-05,
      "step": 3613,
      "training_step_time": 10.30081558227539
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 3.163580894470215,
      "step": 3614
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 3614,
      "training_step_time": 9.934216260910034
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 3.145373582839966,
      "step": 3614
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 3614,
      "training_step_time": 9.916935920715332
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 3.462754249572754,
      "step": 3614
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 3614,
      "training_step_time": 10.22900390625
    },
    {
      "epoch": 1.378631591796875e-05,
      "model_forward_time": 3.1455600261688232,
      "step": 3614
    },
    {
      "epoch": 1.378631591796875e-05,
      "step": 3614,
      "training_step_time": 10.031583309173584
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 3.149322032928467,
      "step": 3615
    },
    {
      "dynamics_loss_avg": 0.05804637735709548,
      "epoch": 1.3790130615234376e-05,
      "step": 3615
    },
    {
      "action_loss_avg": 0.02737687227781862,
      "epoch": 1.3790130615234376e-05,
      "step": 3615
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 3615,
      "training_step_time": 9.923905372619629
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 3.1504929065704346,
      "step": 3615
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 3615,
      "training_step_time": 9.924508810043335
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 3.1450512409210205,
      "step": 3615
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 3615,
      "training_step_time": 9.920023918151855
    },
    {
      "epoch": 1.3790130615234376e-05,
      "model_forward_time": 3.416757106781006,
      "step": 3615
    },
    {
      "epoch": 1.3790130615234376e-05,
      "step": 3615,
      "training_step_time": 10.289056062698364
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 3.180284023284912,
      "step": 3616
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 3616,
      "training_step_time": 9.955293893814087
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 3.1477606296539307,
      "step": 3616
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 3616,
      "training_step_time": 9.923304557800293
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 3.1494245529174805,
      "step": 3616
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 3616,
      "training_step_time": 9.91710615158081
    },
    {
      "epoch": 1.37939453125e-05,
      "model_forward_time": 3.1507644653320312,
      "step": 3616
    },
    {
      "epoch": 1.37939453125e-05,
      "step": 3616,
      "training_step_time": 10.282456874847412
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 3.412065267562866,
      "step": 3617
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 3617,
      "training_step_time": 10.195839405059814
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 3.1509602069854736,
      "step": 3617
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 3617,
      "training_step_time": 9.927515506744385
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 3.146270513534546,
      "step": 3617
    },
    {
      "dynamics_loss_avg": 0.07033126056194305,
      "epoch": 1.3797760009765625e-05,
      "step": 3617
    },
    {
      "action_loss_avg": 0.03149085436016321,
      "epoch": 1.3797760009765625e-05,
      "step": 3617
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 3617,
      "training_step_time": 9.92565369606018
    },
    {
      "epoch": 1.3797760009765625e-05,
      "model_forward_time": 3.1496810913085938,
      "step": 3617
    },
    {
      "epoch": 1.3797760009765625e-05,
      "step": 3617,
      "training_step_time": 10.278757572174072
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 3.155489206314087,
      "step": 3618
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 3618,
      "training_step_time": 9.931236505508423
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 3.420419692993164,
      "step": 3618
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 3618,
      "training_step_time": 10.196226835250854
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 2.435001850128174,
      "step": 3618
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 3618,
      "training_step_time": 7.463037729263306
    },
    {
      "epoch": 1.380157470703125e-05,
      "model_forward_time": 3.137063503265381,
      "step": 3618
    },
    {
      "epoch": 1.380157470703125e-05,
      "step": 3618,
      "training_step_time": 10.037102222442627
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 3.1450726985931396,
      "step": 3619
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 3619,
      "training_step_time": 9.910438299179077
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 3.129446268081665,
      "step": 3619
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 3619,
      "training_step_time": 9.89333176612854
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 2.4280788898468018,
      "step": 3619
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 3619,
      "training_step_time": 7.717912435531616
    },
    {
      "epoch": 1.3805389404296876e-05,
      "model_forward_time": 3.12532639503479,
      "step": 3619
    },
    {
      "epoch": 1.3805389404296876e-05,
      "step": 3619,
      "training_step_time": 10.31803011894226
    },
    {
      "epoch": 1.38092041015625e-05,
      "grad_norm": 0.4579768776893616,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 3620
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 3.1426901817321777,
      "step": 3620
    },
    {
      "dynamics_loss_avg": 0.05512879882007837,
      "epoch": 1.38092041015625e-05,
      "step": 3620
    },
    {
      "action_loss_avg": 0.02699283165857196,
      "epoch": 1.38092041015625e-05,
      "step": 3620
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 3620,
      "training_step_time": 9.905743837356567
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 3.1325371265411377,
      "step": 3620
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 3620,
      "training_step_time": 9.905565023422241
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 3.139366865158081,
      "step": 3620
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 3620,
      "training_step_time": 9.912809371948242
    },
    {
      "epoch": 1.38092041015625e-05,
      "model_forward_time": 3.420900344848633,
      "step": 3620
    },
    {
      "epoch": 1.38092041015625e-05,
      "step": 3620,
      "training_step_time": 10.283631563186646
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 3.1473662853240967,
      "step": 3621
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 3621,
      "training_step_time": 9.930598497390747
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 2.437962770462036,
      "step": 3621
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 3621,
      "training_step_time": 7.472612619400024
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 3.133657217025757,
      "step": 3621
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 3621,
      "training_step_time": 9.904857158660889
    },
    {
      "epoch": 1.3813018798828125e-05,
      "model_forward_time": 3.1336631774902344,
      "step": 3621
    },
    {
      "epoch": 1.3813018798828125e-05,
      "step": 3621,
      "training_step_time": 10.338152170181274
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 3.146211624145508,
      "step": 3622
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 3622,
      "training_step_time": 10.198064088821411
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 3.135838031768799,
      "step": 3622
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 3622,
      "training_step_time": 9.902965307235718
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 3.1396124362945557,
      "step": 3622
    },
    {
      "dynamics_loss_avg": 0.05709752887487411,
      "epoch": 1.381683349609375e-05,
      "step": 3622
    },
    {
      "action_loss_avg": 0.02841195564251393,
      "epoch": 1.381683349609375e-05,
      "step": 3622
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 3622,
      "training_step_time": 9.911376953125
    },
    {
      "epoch": 1.381683349609375e-05,
      "model_forward_time": 2.431129217147827,
      "step": 3622
    },
    {
      "epoch": 1.381683349609375e-05,
      "step": 3622,
      "training_step_time": 10.282589435577393
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 3.1559436321258545,
      "step": 3623
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 3623,
      "training_step_time": 9.92808485031128
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 3.4055583477020264,
      "step": 3623
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 3623,
      "training_step_time": 10.177148342132568
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 3.1384246349334717,
      "step": 3623
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 3623,
      "training_step_time": 9.91401219367981
    },
    {
      "epoch": 1.3820648193359376e-05,
      "model_forward_time": 3.138618230819702,
      "step": 3623
    },
    {
      "epoch": 1.3820648193359376e-05,
      "step": 3623,
      "training_step_time": 10.028223991394043
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 3.143794059753418,
      "step": 3624
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 3624,
      "training_step_time": 9.919920444488525
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 3.145709991455078,
      "step": 3624
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 3624,
      "training_step_time": 9.92246675491333
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 3.1415584087371826,
      "step": 3624
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 3624,
      "training_step_time": 10.197255611419678
    },
    {
      "epoch": 1.3824462890625e-05,
      "model_forward_time": 3.1410579681396484,
      "step": 3624
    },
    {
      "epoch": 1.3824462890625e-05,
      "step": 3624,
      "training_step_time": 10.25430941581726
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 3.1795520782470703,
      "step": 3625
    },
    {
      "dynamics_loss_avg": 0.062083668634295465,
      "epoch": 1.3828277587890625e-05,
      "step": 3625
    },
    {
      "action_loss_avg": 0.032132974546402696,
      "epoch": 1.3828277587890625e-05,
      "step": 3625
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 3625,
      "training_step_time": 9.965360403060913
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 3.148810386657715,
      "step": 3625
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 3625,
      "training_step_time": 9.928760051727295
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 3.142698287963867,
      "step": 3625
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 3625,
      "training_step_time": 9.925753116607666
    },
    {
      "epoch": 1.3828277587890625e-05,
      "model_forward_time": 3.4078445434570312,
      "step": 3625
    },
    {
      "epoch": 1.3828277587890625e-05,
      "step": 3625,
      "training_step_time": 10.296128034591675
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 3.1604535579681396,
      "step": 3626
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 3626,
      "training_step_time": 9.940979480743408
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 3.145167350769043,
      "step": 3626
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 3626,
      "training_step_time": 9.925259828567505
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 2.442230224609375,
      "step": 3626
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 3626,
      "training_step_time": 7.4767701625823975
    },
    {
      "epoch": 1.383209228515625e-05,
      "model_forward_time": 3.1437346935272217,
      "step": 3626
    },
    {
      "epoch": 1.383209228515625e-05,
      "step": 3626,
      "training_step_time": 10.272885799407959
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 3.4904849529266357,
      "step": 3627
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 3627,
      "training_step_time": 10.273375272750854
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 3.1464579105377197,
      "step": 3627
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 3627,
      "training_step_time": 9.92327356338501
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 3.1432559490203857,
      "step": 3627
    },
    {
      "dynamics_loss_avg": 0.07195301465690136,
      "epoch": 1.3835906982421874e-05,
      "step": 3627
    },
    {
      "action_loss_avg": 0.04479286009445786,
      "epoch": 1.3835906982421874e-05,
      "step": 3627
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 3627,
      "training_step_time": 9.924554586410522
    },
    {
      "epoch": 1.3835906982421874e-05,
      "model_forward_time": 3.1461989879608154,
      "step": 3627
    },
    {
      "epoch": 1.3835906982421874e-05,
      "step": 3627,
      "training_step_time": 10.258877754211426
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 3.151115894317627,
      "step": 3628
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 3628,
      "training_step_time": 9.93375301361084
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 3.1431570053100586,
      "step": 3628
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 3628,
      "training_step_time": 10.200246810913086
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 3.1467232704162598,
      "step": 3628
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 3628,
      "training_step_time": 9.928641319274902
    },
    {
      "epoch": 1.38397216796875e-05,
      "model_forward_time": 3.1446659564971924,
      "step": 3628
    },
    {
      "epoch": 1.38397216796875e-05,
      "step": 3628,
      "training_step_time": 10.309821605682373
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 3.1512808799743652,
      "step": 3629
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 3629,
      "training_step_time": 9.930086612701416
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 3.1393721103668213,
      "step": 3629
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 3629,
      "training_step_time": 9.92748212814331
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 3.4167652130126953,
      "step": 3629
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 3629,
      "training_step_time": 10.19149923324585
    },
    {
      "epoch": 1.3843536376953125e-05,
      "model_forward_time": 2.4337146282196045,
      "step": 3629
    },
    {
      "epoch": 1.3843536376953125e-05,
      "step": 3629,
      "training_step_time": 10.23157024383545
    },
    {
      "epoch": 1.384735107421875e-05,
      "grad_norm": 0.563630223274231,
      "learning_rate": 1e-05,
      "loss": 0.0943,
      "step": 3630
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 3.14315128326416,
      "step": 3630
    },
    {
      "dynamics_loss_avg": 0.0600456491112709,
      "epoch": 1.384735107421875e-05,
      "step": 3630
    },
    {
      "action_loss_avg": 0.026893980521708726,
      "epoch": 1.384735107421875e-05,
      "step": 3630
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 3630,
      "training_step_time": 9.921574831008911
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 3.13676381111145,
      "step": 3630
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 3630,
      "training_step_time": 9.905057907104492
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 3.139017105102539,
      "step": 3630
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 3630,
      "training_step_time": 9.922050714492798
    },
    {
      "epoch": 1.384735107421875e-05,
      "model_forward_time": 3.1361427307128906,
      "step": 3630
    },
    {
      "epoch": 1.384735107421875e-05,
      "step": 3630,
      "training_step_time": 10.2779061794281
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 3.4741227626800537,
      "step": 3631
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 3631,
      "training_step_time": 10.249730348587036
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 3.1355133056640625,
      "step": 3631
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 3631,
      "training_step_time": 9.91064190864563
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 3.138942003250122,
      "step": 3631
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 3631,
      "training_step_time": 9.914335489273071
    },
    {
      "epoch": 1.3851165771484374e-05,
      "model_forward_time": 3.139780282974243,
      "step": 3631
    },
    {
      "epoch": 1.3851165771484374e-05,
      "step": 3631,
      "training_step_time": 10.01911473274231
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 3.43639874458313,
      "step": 3632
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 3632,
      "training_step_time": 10.213892459869385
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 3.1390140056610107,
      "step": 3632
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 3632,
      "training_step_time": 9.916045665740967
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 2.4325757026672363,
      "step": 3632
    },
    {
      "dynamics_loss_avg": 0.07367290258407592,
      "epoch": 1.385498046875e-05,
      "step": 3632
    },
    {
      "action_loss_avg": 0.033795825112611054,
      "epoch": 1.385498046875e-05,
      "step": 3632
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 3632,
      "training_step_time": 7.4709131717681885
    },
    {
      "epoch": 1.385498046875e-05,
      "model_forward_time": 3.1372644901275635,
      "step": 3632
    },
    {
      "epoch": 1.385498046875e-05,
      "step": 3632,
      "training_step_time": 10.31395173072815
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 3.1549317836761475,
      "step": 3633
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 3633,
      "training_step_time": 9.93295669555664
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 3.1390626430511475,
      "step": 3633
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 3633,
      "training_step_time": 10.18463397026062
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 3.143343210220337,
      "step": 3633
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 3633,
      "training_step_time": 9.918289184570312
    },
    {
      "epoch": 1.3858795166015625e-05,
      "model_forward_time": 3.140423536300659,
      "step": 3633
    },
    {
      "epoch": 1.3858795166015625e-05,
      "step": 3633,
      "training_step_time": 10.316759824752808
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 3.1463849544525146,
      "step": 3634
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 3634,
      "training_step_time": 9.921603679656982
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 3.13765549659729,
      "step": 3634
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 3634,
      "training_step_time": 9.910086154937744
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 2.4350979328155518,
      "step": 3634
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 3634,
      "training_step_time": 7.744009733200073
    },
    {
      "epoch": 1.386260986328125e-05,
      "model_forward_time": 2.436018943786621,
      "step": 3634
    },
    {
      "epoch": 1.386260986328125e-05,
      "step": 3634,
      "training_step_time": 10.249919176101685
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 3.179401159286499,
      "step": 3635
    },
    {
      "dynamics_loss_avg": 0.07316546514630318,
      "epoch": 1.3866424560546874e-05,
      "step": 3635
    },
    {
      "action_loss_avg": 0.031001450680196285,
      "epoch": 1.3866424560546874e-05,
      "step": 3635
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 3635,
      "training_step_time": 9.95537543296814
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 3.136385202407837,
      "step": 3635
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 3635,
      "training_step_time": 9.906549215316772
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 3.1351702213287354,
      "step": 3635
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 3635,
      "training_step_time": 9.90261173248291
    },
    {
      "epoch": 1.3866424560546874e-05,
      "model_forward_time": 3.1360416412353516,
      "step": 3635
    },
    {
      "epoch": 1.3866424560546874e-05,
      "step": 3635,
      "training_step_time": 10.004577875137329
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 3.4182939529418945,
      "step": 3636
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 3636,
      "training_step_time": 10.191164255142212
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 3.1350033283233643,
      "step": 3636
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 3636,
      "training_step_time": 9.893361806869507
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 3.127168655395508,
      "step": 3636
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 3636,
      "training_step_time": 9.895521402359009
    },
    {
      "epoch": 1.38702392578125e-05,
      "model_forward_time": 3.1367454528808594,
      "step": 3636
    },
    {
      "epoch": 1.38702392578125e-05,
      "step": 3636,
      "training_step_time": 10.254606008529663
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 3.1460185050964355,
      "step": 3637
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 3637,
      "training_step_time": 10.187945365905762
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 3.134366512298584,
      "step": 3637
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 3637,
      "training_step_time": 9.900888204574585
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 3.1331658363342285,
      "step": 3637
    },
    {
      "dynamics_loss_avg": 0.04700515237636864,
      "epoch": 1.3874053955078125e-05,
      "step": 3637
    },
    {
      "action_loss_avg": 0.02314832661068067,
      "epoch": 1.3874053955078125e-05,
      "step": 3637
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 3637,
      "training_step_time": 9.901015043258667
    },
    {
      "epoch": 1.3874053955078125e-05,
      "model_forward_time": 3.1359925270080566,
      "step": 3637
    },
    {
      "epoch": 1.3874053955078125e-05,
      "step": 3637,
      "training_step_time": 10.010238409042358
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 3.150834083557129,
      "step": 3638
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 3638,
      "training_step_time": 9.919808864593506
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 3.4171900749206543,
      "step": 3638
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 3638,
      "training_step_time": 10.186910152435303
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 3.131866693496704,
      "step": 3638
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 3638,
      "training_step_time": 9.904783248901367
    },
    {
      "epoch": 1.387786865234375e-05,
      "model_forward_time": 3.1375718116760254,
      "step": 3638
    },
    {
      "epoch": 1.387786865234375e-05,
      "step": 3638,
      "training_step_time": 10.277451515197754
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 3.1659395694732666,
      "step": 3639
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 3639,
      "training_step_time": 9.941456079483032
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 3.1331608295440674,
      "step": 3639
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 3639,
      "training_step_time": 9.90766429901123
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 3.1365349292755127,
      "step": 3639
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 3639,
      "training_step_time": 10.188877582550049
    },
    {
      "epoch": 1.3881683349609374e-05,
      "model_forward_time": 3.1388397216796875,
      "step": 3639
    },
    {
      "epoch": 1.3881683349609374e-05,
      "step": 3639,
      "training_step_time": 10.040450096130371
    },
    {
      "epoch": 1.3885498046875e-05,
      "grad_norm": 0.4965161383152008,
      "learning_rate": 1e-05,
      "loss": 0.0974,
      "step": 3640
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 3.1539525985717773,
      "step": 3640
    },
    {
      "dynamics_loss_avg": 0.06923563033342361,
      "epoch": 1.3885498046875e-05,
      "step": 3640
    },
    {
      "action_loss_avg": 0.028680283203721048,
      "epoch": 1.3885498046875e-05,
      "step": 3640
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 3640,
      "training_step_time": 9.940930366516113
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 3.1359195709228516,
      "step": 3640
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 3640,
      "training_step_time": 9.916640758514404
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 3.1400394439697266,
      "step": 3640
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 3640,
      "training_step_time": 9.922449111938477
    },
    {
      "epoch": 1.3885498046875e-05,
      "model_forward_time": 3.4456348419189453,
      "step": 3640
    },
    {
      "epoch": 1.3885498046875e-05,
      "step": 3640,
      "training_step_time": 10.320431470870972
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 3.1775946617126465,
      "step": 3641
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 3641,
      "training_step_time": 9.966278314590454
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 3.1575374603271484,
      "step": 3641
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 3641,
      "training_step_time": 9.936798334121704
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 3.141786575317383,
      "step": 3641
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 3641,
      "training_step_time": 9.919617891311646
    },
    {
      "epoch": 1.3889312744140625e-05,
      "model_forward_time": 3.138899326324463,
      "step": 3641
    },
    {
      "epoch": 1.3889312744140625e-05,
      "step": 3641,
      "training_step_time": 10.285478353500366
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 2.4729535579681396,
      "step": 3642
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 3642,
      "training_step_time": 7.784594297409058
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 3.136885166168213,
      "step": 3642
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 3642,
      "training_step_time": 9.916903495788574
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 3.1459245681762695,
      "step": 3642
    },
    {
      "dynamics_loss_avg": 0.06344791762530803,
      "epoch": 1.389312744140625e-05,
      "step": 3642
    },
    {
      "action_loss_avg": 0.0261686728335917,
      "epoch": 1.389312744140625e-05,
      "step": 3642
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 3642,
      "training_step_time": 9.928984642028809
    },
    {
      "epoch": 1.389312744140625e-05,
      "model_forward_time": 3.1457438468933105,
      "step": 3642
    },
    {
      "epoch": 1.389312744140625e-05,
      "step": 3642,
      "training_step_time": 10.035924196243286
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 2.4696738719940186,
      "step": 3643
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 3643,
      "training_step_time": 7.506069660186768
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 3.4046850204467773,
      "step": 3643
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 3643,
      "training_step_time": 10.182186603546143
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 3.144505023956299,
      "step": 3643
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 3643,
      "training_step_time": 9.927143096923828
    },
    {
      "epoch": 1.3896942138671874e-05,
      "model_forward_time": 3.142585515975952,
      "step": 3643
    },
    {
      "epoch": 1.3896942138671874e-05,
      "step": 3643,
      "training_step_time": 10.287609100341797
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 3.150503396987915,
      "step": 3644
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 3644,
      "training_step_time": 9.932581663131714
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 3.1426286697387695,
      "step": 3644
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 3644,
      "training_step_time": 9.919336318969727
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 3.139373779296875,
      "step": 3644
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 3644,
      "training_step_time": 10.179203033447266
    },
    {
      "epoch": 1.39007568359375e-05,
      "model_forward_time": 3.1376335620880127,
      "step": 3644
    },
    {
      "epoch": 1.39007568359375e-05,
      "step": 3644,
      "training_step_time": 10.027925252914429
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 3.1510097980499268,
      "step": 3645
    },
    {
      "dynamics_loss_avg": 0.06535773687064647,
      "epoch": 1.3904571533203125e-05,
      "step": 3645
    },
    {
      "action_loss_avg": 0.037693094462156296,
      "epoch": 1.3904571533203125e-05,
      "step": 3645
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 3645,
      "training_step_time": 9.92760181427002
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 3.141958236694336,
      "step": 3645
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 3645,
      "training_step_time": 9.917765855789185
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 3.138853073120117,
      "step": 3645
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 3645,
      "training_step_time": 9.914864540100098
    },
    {
      "epoch": 1.3904571533203125e-05,
      "model_forward_time": 3.399049997329712,
      "step": 3645
    },
    {
      "epoch": 1.3904571533203125e-05,
      "step": 3645,
      "training_step_time": 10.290709257125854
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 3.144440174102783,
      "step": 3646
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 3646,
      "training_step_time": 9.922471523284912
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 3.144451141357422,
      "step": 3646
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 3646,
      "training_step_time": 9.91896915435791
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 3.1489784717559814,
      "step": 3646
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 3646,
      "training_step_time": 9.923473119735718
    },
    {
      "epoch": 1.390838623046875e-05,
      "model_forward_time": 2.4406332969665527,
      "step": 3646
    },
    {
      "epoch": 1.390838623046875e-05,
      "step": 3646,
      "training_step_time": 10.297333717346191
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 3.4129786491394043,
      "step": 3647
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 3647,
      "training_step_time": 10.189293146133423
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 3.1458230018615723,
      "step": 3647
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 3647,
      "training_step_time": 9.922216415405273
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 3.1383421421051025,
      "step": 3647
    },
    {
      "dynamics_loss_avg": 0.08348894901573659,
      "epoch": 1.3912200927734374e-05,
      "step": 3647
    },
    {
      "action_loss_avg": 0.044044120237231255,
      "epoch": 1.3912200927734374e-05,
      "step": 3647
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 3647,
      "training_step_time": 9.914716958999634
    },
    {
      "epoch": 1.3912200927734374e-05,
      "model_forward_time": 3.141371250152588,
      "step": 3647
    },
    {
      "epoch": 1.3912200927734374e-05,
      "step": 3647,
      "training_step_time": 10.294674396514893
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 2.477046012878418,
      "step": 3648
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 3648,
      "training_step_time": 7.505288362503052
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 3.1354167461395264,
      "step": 3648
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 3648,
      "training_step_time": 9.905211687088013
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 3.39459228515625,
      "step": 3648
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 3648,
      "training_step_time": 10.167396068572998
    },
    {
      "epoch": 1.3916015625e-05,
      "model_forward_time": 3.143313407897949,
      "step": 3648
    },
    {
      "epoch": 1.3916015625e-05,
      "step": 3648,
      "training_step_time": 10.301567077636719
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 2.4468913078308105,
      "step": 3649
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 3649,
      "training_step_time": 7.479265928268433
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 3.1410772800445557,
      "step": 3649
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 3649,
      "training_step_time": 9.917577028274536
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 3.1420376300811768,
      "step": 3649
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 3649,
      "training_step_time": 10.203916311264038
    },
    {
      "epoch": 1.3919830322265625e-05,
      "model_forward_time": 3.1488609313964844,
      "step": 3649
    },
    {
      "epoch": 1.3919830322265625e-05,
      "step": 3649,
      "training_step_time": 10.038567781448364
    },
    {
      "epoch": 1.392364501953125e-05,
      "grad_norm": 0.3563544452190399,
      "learning_rate": 1e-05,
      "loss": 0.0934,
      "step": 3650
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 3.1487579345703125,
      "step": 3650
    },
    {
      "dynamics_loss_avg": 0.06151238102465868,
      "epoch": 1.392364501953125e-05,
      "step": 3650
    },
    {
      "action_loss_avg": 0.026679752627387643,
      "epoch": 1.392364501953125e-05,
      "step": 3650
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 3650,
      "training_step_time": 9.95398235321045
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 3.1359751224517822,
      "step": 3650
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 3650,
      "training_step_time": 9.920472383499146
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 3.1413581371307373,
      "step": 3650
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 3650,
      "training_step_time": 9.9239981174469
    },
    {
      "epoch": 1.392364501953125e-05,
      "model_forward_time": 2.434783697128296,
      "step": 3650
    },
    {
      "epoch": 1.392364501953125e-05,
      "step": 3650,
      "training_step_time": 10.036788702011108
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 3.4388132095336914,
      "step": 3651
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 3651,
      "training_step_time": 10.221275806427002
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 3.135087728500366,
      "step": 3651
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 3651,
      "training_step_time": 9.909076452255249
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 3.1362879276275635,
      "step": 3651
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 3651,
      "training_step_time": 9.906919002532959
    },
    {
      "epoch": 1.3927459716796874e-05,
      "model_forward_time": 3.1304690837860107,
      "step": 3651
    },
    {
      "epoch": 1.3927459716796874e-05,
      "step": 3651,
      "training_step_time": 9.999397039413452
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 3.4433372020721436,
      "step": 3652
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 3652,
      "training_step_time": 10.218717813491821
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 3.1288909912109375,
      "step": 3652
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 3652,
      "training_step_time": 9.898504972457886
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 2.4371025562286377,
      "step": 3652
    },
    {
      "dynamics_loss_avg": 0.058642519451677796,
      "epoch": 1.39312744140625e-05,
      "step": 3652
    },
    {
      "action_loss_avg": 0.025344942323863508,
      "epoch": 1.39312744140625e-05,
      "step": 3652
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 3652,
      "training_step_time": 7.465036630630493
    },
    {
      "epoch": 1.39312744140625e-05,
      "model_forward_time": 3.126582384109497,
      "step": 3652
    },
    {
      "epoch": 1.39312744140625e-05,
      "step": 3652,
      "training_step_time": 10.335627555847168
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 2.4550862312316895,
      "step": 3653
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 3653,
      "training_step_time": 7.480851173400879
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 3.130117416381836,
      "step": 3653
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 3653,
      "training_step_time": 9.900141716003418
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 3.423518657684326,
      "step": 3653
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 3653,
      "training_step_time": 10.193281888961792
    },
    {
      "epoch": 1.3935089111328125e-05,
      "model_forward_time": 3.139432907104492,
      "step": 3653
    },
    {
      "epoch": 1.3935089111328125e-05,
      "step": 3653,
      "training_step_time": 10.278106451034546
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 3.157703161239624,
      "step": 3654
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 3654,
      "training_step_time": 9.92989206314087
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 3.135359525680542,
      "step": 3654
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 3654,
      "training_step_time": 9.90817403793335
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 3.1299290657043457,
      "step": 3654
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 3654,
      "training_step_time": 10.194114446640015
    },
    {
      "epoch": 1.393890380859375e-05,
      "model_forward_time": 3.1298182010650635,
      "step": 3654
    },
    {
      "epoch": 1.393890380859375e-05,
      "step": 3654,
      "training_step_time": 10.012255907058716
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 3.1593120098114014,
      "step": 3655
    },
    {
      "dynamics_loss_avg": 0.07109117545187474,
      "epoch": 1.3942718505859374e-05,
      "step": 3655
    },
    {
      "action_loss_avg": 0.039169259183108804,
      "epoch": 1.3942718505859374e-05,
      "step": 3655
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 3655,
      "training_step_time": 9.93257999420166
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 3.135404109954834,
      "step": 3655
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 3655,
      "training_step_time": 9.911881685256958
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 3.1314542293548584,
      "step": 3655
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 3655,
      "training_step_time": 9.904315710067749
    },
    {
      "epoch": 1.3942718505859374e-05,
      "model_forward_time": 3.1352338790893555,
      "step": 3655
    },
    {
      "epoch": 1.3942718505859374e-05,
      "step": 3655,
      "training_step_time": 10.27930736541748
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 3.181304931640625,
      "step": 3656
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 3656,
      "training_step_time": 9.960691928863525
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 3.1384713649749756,
      "step": 3656
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 3656,
      "training_step_time": 9.914636373519897
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 3.1360690593719482,
      "step": 3656
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 3656,
      "training_step_time": 9.915477752685547
    },
    {
      "epoch": 1.3946533203125e-05,
      "model_forward_time": 3.144639253616333,
      "step": 3656
    },
    {
      "epoch": 1.3946533203125e-05,
      "step": 3656,
      "training_step_time": 10.027734518051147
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 3.1576569080352783,
      "step": 3657
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 3657,
      "training_step_time": 10.223735570907593
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 3.138667345046997,
      "step": 3657
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 3657,
      "training_step_time": 9.915802478790283
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 3.13600754737854,
      "step": 3657
    },
    {
      "dynamics_loss_avg": 0.06815323494374752,
      "epoch": 1.3950347900390625e-05,
      "step": 3657
    },
    {
      "action_loss_avg": 0.03300442397594452,
      "epoch": 1.3950347900390625e-05,
      "step": 3657
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 3657,
      "training_step_time": 9.914905548095703
    },
    {
      "epoch": 1.3950347900390625e-05,
      "model_forward_time": 2.4345543384552,
      "step": 3657
    },
    {
      "epoch": 1.3950347900390625e-05,
      "step": 3657,
      "training_step_time": 10.300487756729126
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 3.151710271835327,
      "step": 3658
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 3658,
      "training_step_time": 9.931076288223267
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 3.1388890743255615,
      "step": 3658
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 3658,
      "training_step_time": 10.183001518249512
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 2.4350788593292236,
      "step": 3658
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 3658,
      "training_step_time": 7.469773054122925
    },
    {
      "epoch": 1.395416259765625e-05,
      "model_forward_time": 3.136770725250244,
      "step": 3658
    },
    {
      "epoch": 1.395416259765625e-05,
      "step": 3658,
      "training_step_time": 10.291658639907837
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 3.156235456466675,
      "step": 3659
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 3659,
      "training_step_time": 9.930089235305786
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 3.1316370964050293,
      "step": 3659
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 3659,
      "training_step_time": 9.912514448165894
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 2.432483434677124,
      "step": 3659
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 3659,
      "training_step_time": 7.466228723526001
    },
    {
      "epoch": 1.3957977294921874e-05,
      "model_forward_time": 3.4153692722320557,
      "step": 3659
    },
    {
      "epoch": 1.3957977294921874e-05,
      "step": 3659,
      "training_step_time": 10.286582708358765
    },
    {
      "epoch": 1.39617919921875e-05,
      "grad_norm": 0.5546126365661621,
      "learning_rate": 1e-05,
      "loss": 0.0958,
      "step": 3660
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 3.1409685611724854,
      "step": 3660
    },
    {
      "dynamics_loss_avg": 0.06593400537967682,
      "epoch": 1.39617919921875e-05,
      "step": 3660
    },
    {
      "action_loss_avg": 0.034677699021995065,
      "epoch": 1.39617919921875e-05,
      "step": 3660
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 3660,
      "training_step_time": 9.919115781784058
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 3.1299428939819336,
      "step": 3660
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 3660,
      "training_step_time": 9.902803182601929
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 3.1336047649383545,
      "step": 3660
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 3660,
      "training_step_time": 9.902819633483887
    },
    {
      "epoch": 1.39617919921875e-05,
      "model_forward_time": 3.4365928173065186,
      "step": 3660
    },
    {
      "epoch": 1.39617919921875e-05,
      "step": 3660,
      "training_step_time": 10.30072283744812
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 3.177356481552124,
      "step": 3661
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 3661,
      "training_step_time": 9.951154232025146
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 3.133751630783081,
      "step": 3661
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 3661,
      "training_step_time": 9.911837816238403
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 3.141488790512085,
      "step": 3661
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 3661,
      "training_step_time": 9.919003963470459
    },
    {
      "epoch": 1.3965606689453125e-05,
      "model_forward_time": 3.140397548675537,
      "step": 3661
    },
    {
      "epoch": 1.3965606689453125e-05,
      "step": 3661,
      "training_step_time": 10.025021314620972
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 3.1583452224731445,
      "step": 3662
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 3662,
      "training_step_time": 10.204042911529541
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 3.1416378021240234,
      "step": 3662
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 3662,
      "training_step_time": 9.921149969100952
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 3.1412222385406494,
      "step": 3662
    },
    {
      "dynamics_loss_avg": 0.0726812470704317,
      "epoch": 1.396942138671875e-05,
      "step": 3662
    },
    {
      "action_loss_avg": 0.027198931016027928,
      "epoch": 1.396942138671875e-05,
      "step": 3662
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 3662,
      "training_step_time": 9.92658019065857
    },
    {
      "epoch": 1.396942138671875e-05,
      "model_forward_time": 3.142549753189087,
      "step": 3662
    },
    {
      "epoch": 1.396942138671875e-05,
      "step": 3662,
      "training_step_time": 10.323691368103027
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 3.1655640602111816,
      "step": 3663
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 3663,
      "training_step_time": 9.957637071609497
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 3.3987579345703125,
      "step": 3663
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 3663,
      "training_step_time": 10.178067207336426
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 3.137314796447754,
      "step": 3663
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 3663,
      "training_step_time": 9.911224126815796
    },
    {
      "epoch": 1.3973236083984374e-05,
      "model_forward_time": 3.1286261081695557,
      "step": 3663
    },
    {
      "epoch": 1.3973236083984374e-05,
      "step": 3663,
      "training_step_time": 10.308093547821045
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 3.1481266021728516,
      "step": 3664
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 3664,
      "training_step_time": 9.922085285186768
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 3.140230655670166,
      "step": 3664
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 3664,
      "training_step_time": 9.909404754638672
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 3.4457383155822754,
      "step": 3664
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 3664,
      "training_step_time": 10.21386456489563
    },
    {
      "epoch": 1.397705078125e-05,
      "model_forward_time": 2.4378743171691895,
      "step": 3664
    },
    {
      "epoch": 1.397705078125e-05,
      "step": 3664,
      "training_step_time": 10.257339477539062
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 3.1633670330047607,
      "step": 3665
    },
    {
      "dynamics_loss_avg": 0.07062508650124073,
      "epoch": 1.3980865478515625e-05,
      "step": 3665
    },
    {
      "action_loss_avg": 0.028396889567375183,
      "epoch": 1.3980865478515625e-05,
      "step": 3665
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 3665,
      "training_step_time": 9.9338538646698
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 3.1288442611694336,
      "step": 3665
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 3665,
      "training_step_time": 9.89278531074524
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 3.126983165740967,
      "step": 3665
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 3665,
      "training_step_time": 9.888849020004272
    },
    {
      "epoch": 1.3980865478515625e-05,
      "model_forward_time": 3.1293468475341797,
      "step": 3665
    },
    {
      "epoch": 1.3980865478515625e-05,
      "step": 3665,
      "training_step_time": 10.272786855697632
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 2.4702095985412598,
      "step": 3666
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 3666,
      "training_step_time": 7.490128040313721
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 3.129262924194336,
      "step": 3666
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 3666,
      "training_step_time": 9.889639616012573
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 3.1248042583465576,
      "step": 3666
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 3666,
      "training_step_time": 9.889904737472534
    },
    {
      "epoch": 1.398468017578125e-05,
      "model_forward_time": 3.134763717651367,
      "step": 3666
    },
    {
      "epoch": 1.398468017578125e-05,
      "step": 3666,
      "training_step_time": 10.28483247756958
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 3.175119638442993,
      "step": 3667
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 3667,
      "training_step_time": 10.208334922790527
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 3.146040678024292,
      "step": 3667
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 3667,
      "training_step_time": 9.910097122192383
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 3.1357579231262207,
      "step": 3667
    },
    {
      "dynamics_loss_avg": 0.06585852019488811,
      "epoch": 1.3988494873046874e-05,
      "step": 3667
    },
    {
      "action_loss_avg": 0.029847642034292222,
      "epoch": 1.3988494873046874e-05,
      "step": 3667
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 3667,
      "training_step_time": 9.903589010238647
    },
    {
      "epoch": 1.3988494873046874e-05,
      "model_forward_time": 3.1431031227111816,
      "step": 3667
    },
    {
      "epoch": 1.3988494873046874e-05,
      "step": 3667,
      "training_step_time": 10.306198358535767
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 2.4783239364624023,
      "step": 3668
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 3668,
      "training_step_time": 7.504568099975586
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 3.136234998703003,
      "step": 3668
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 3668,
      "training_step_time": 10.158012390136719
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 3.139301300048828,
      "step": 3668
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 3668,
      "training_step_time": 9.903061389923096
    },
    {
      "epoch": 1.39923095703125e-05,
      "model_forward_time": 3.142090082168579,
      "step": 3668
    },
    {
      "epoch": 1.39923095703125e-05,
      "step": 3668,
      "training_step_time": 10.281633615493774
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 3.1529762744903564,
      "step": 3669
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 3669,
      "training_step_time": 9.920039415359497
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 3.1338813304901123,
      "step": 3669
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 3669,
      "training_step_time": 9.896636486053467
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 3.3946287631988525,
      "step": 3669
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 3669,
      "training_step_time": 10.159839868545532
    },
    {
      "epoch": 1.3996124267578125e-05,
      "model_forward_time": 3.140702724456787,
      "step": 3669
    },
    {
      "epoch": 1.3996124267578125e-05,
      "step": 3669,
      "training_step_time": 10.254053592681885
    },
    {
      "epoch": 1.399993896484375e-05,
      "grad_norm": 0.42615747451782227,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 3670
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 3.163306951522827,
      "step": 3670
    },
    {
      "dynamics_loss_avg": 0.07962357308715581,
      "epoch": 1.399993896484375e-05,
      "step": 3670
    },
    {
      "action_loss_avg": 0.03645575847476721,
      "epoch": 1.399993896484375e-05,
      "step": 3670
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 3670,
      "training_step_time": 9.928215026855469
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 3.130404472351074,
      "step": 3670
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 3670,
      "training_step_time": 9.891221761703491
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 3.132120370864868,
      "step": 3670
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 3670,
      "training_step_time": 9.892364501953125
    },
    {
      "epoch": 1.399993896484375e-05,
      "model_forward_time": 3.4152655601501465,
      "step": 3670
    },
    {
      "epoch": 1.399993896484375e-05,
      "step": 3670,
      "training_step_time": 10.275807857513428
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 3.1701512336730957,
      "step": 3671
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 3671,
      "training_step_time": 9.955726623535156
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 3.132580280303955,
      "step": 3671
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 3671,
      "training_step_time": 9.908107280731201
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 3.134167432785034,
      "step": 3671
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 3671,
      "training_step_time": 9.909411907196045
    },
    {
      "epoch": 1.4003753662109374e-05,
      "model_forward_time": 3.131791591644287,
      "step": 3671
    },
    {
      "epoch": 1.4003753662109374e-05,
      "step": 3671,
      "training_step_time": 10.027338027954102
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 3.4205739498138428,
      "step": 3672
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 3672,
      "training_step_time": 10.196622610092163
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 3.133693218231201,
      "step": 3672
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 3672,
      "training_step_time": 9.911727905273438
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 2.4291062355041504,
      "step": 3672
    },
    {
      "dynamics_loss_avg": 0.059357645362615584,
      "epoch": 1.4007568359375e-05,
      "step": 3672
    },
    {
      "action_loss_avg": 0.033049667719751596,
      "epoch": 1.4007568359375e-05,
      "step": 3672
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 3672,
      "training_step_time": 7.462180852890015
    },
    {
      "epoch": 1.4007568359375e-05,
      "model_forward_time": 3.1294567584991455,
      "step": 3672
    },
    {
      "epoch": 1.4007568359375e-05,
      "step": 3672,
      "training_step_time": 10.325863122940063
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 3.1471760272979736,
      "step": 3673
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 3673,
      "training_step_time": 9.928879022598267
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 3.140695571899414,
      "step": 3673
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 3673,
      "training_step_time": 10.189028024673462
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 3.1321561336517334,
      "step": 3673
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 3673,
      "training_step_time": 9.911594152450562
    },
    {
      "epoch": 1.4011383056640625e-05,
      "model_forward_time": 2.430360794067383,
      "step": 3673
    },
    {
      "epoch": 1.4011383056640625e-05,
      "step": 3673,
      "training_step_time": 10.306979179382324
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 3.1409783363342285,
      "step": 3674
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 3674,
      "training_step_time": 9.919394493103027
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 3.136288642883301,
      "step": 3674
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 3674,
      "training_step_time": 9.916203022003174
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 3.1381657123565674,
      "step": 3674
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 3674,
      "training_step_time": 10.23283052444458
    },
    {
      "epoch": 1.401519775390625e-05,
      "model_forward_time": 3.1362087726593018,
      "step": 3674
    },
    {
      "epoch": 1.401519775390625e-05,
      "step": 3674,
      "training_step_time": 10.338761329650879
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 3.141223907470703,
      "step": 3675
    },
    {
      "dynamics_loss_avg": 0.05003986917436123,
      "epoch": 1.4019012451171874e-05,
      "step": 3675
    },
    {
      "action_loss_avg": 0.024533180985599756,
      "epoch": 1.4019012451171874e-05,
      "step": 3675
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 3675,
      "training_step_time": 9.926562547683716
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 3.142723321914673,
      "step": 3675
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 3675,
      "training_step_time": 9.925463676452637
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 3.1371116638183594,
      "step": 3675
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 3675,
      "training_step_time": 9.920706033706665
    },
    {
      "epoch": 1.4019012451171874e-05,
      "model_forward_time": 3.1412510871887207,
      "step": 3675
    },
    {
      "epoch": 1.4019012451171874e-05,
      "step": 3675,
      "training_step_time": 10.315248250961304
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 3.147036075592041,
      "step": 3676
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 3676,
      "training_step_time": 9.925841808319092
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 3.1358582973480225,
      "step": 3676
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 3676,
      "training_step_time": 9.910794973373413
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 3.134066581726074,
      "step": 3676
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 3676,
      "training_step_time": 9.911174535751343
    },
    {
      "epoch": 1.40228271484375e-05,
      "model_forward_time": 3.1368861198425293,
      "step": 3676
    },
    {
      "epoch": 1.40228271484375e-05,
      "step": 3676,
      "training_step_time": 10.033475637435913
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 3.1467227935791016,
      "step": 3677
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 3677,
      "training_step_time": 10.213038682937622
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 2.4355599880218506,
      "step": 3677
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 3677,
      "training_step_time": 7.468534469604492
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 3.1369776725769043,
      "step": 3677
    },
    {
      "dynamics_loss_avg": 0.055163140036165716,
      "epoch": 1.4026641845703125e-05,
      "step": 3677
    },
    {
      "action_loss_avg": 0.031180113786831498,
      "epoch": 1.4026641845703125e-05,
      "step": 3677
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 3677,
      "training_step_time": 9.918687105178833
    },
    {
      "epoch": 1.4026641845703125e-05,
      "model_forward_time": 2.431633710861206,
      "step": 3677
    },
    {
      "epoch": 1.4026641845703125e-05,
      "step": 3677,
      "training_step_time": 10.28449535369873
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 3.1734774112701416,
      "step": 3678
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 3678,
      "training_step_time": 9.949489831924438
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 3.1329166889190674,
      "step": 3678
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 3678,
      "training_step_time": 9.908061027526855
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 3.398301362991333,
      "step": 3678
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 3678,
      "training_step_time": 10.1769540309906
    },
    {
      "epoch": 1.403045654296875e-05,
      "model_forward_time": 3.136533260345459,
      "step": 3678
    },
    {
      "epoch": 1.403045654296875e-05,
      "step": 3678,
      "training_step_time": 10.273413181304932
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 3.1539359092712402,
      "step": 3679
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 3679,
      "training_step_time": 9.935713768005371
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 3.1334495544433594,
      "step": 3679
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 3679,
      "training_step_time": 9.914540529251099
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 3.138867139816284,
      "step": 3679
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 3679,
      "training_step_time": 9.915723085403442
    },
    {
      "epoch": 1.4034271240234374e-05,
      "model_forward_time": 3.406245231628418,
      "step": 3679
    },
    {
      "epoch": 1.4034271240234374e-05,
      "step": 3679,
      "training_step_time": 10.274698734283447
    },
    {
      "epoch": 1.40380859375e-05,
      "grad_norm": 0.5029216408729553,
      "learning_rate": 1e-05,
      "loss": 0.0934,
      "step": 3680
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 3.1526854038238525,
      "step": 3680
    },
    {
      "dynamics_loss_avg": 0.0656260259449482,
      "epoch": 1.40380859375e-05,
      "step": 3680
    },
    {
      "action_loss_avg": 0.025940173491835593,
      "epoch": 1.40380859375e-05,
      "step": 3680
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 3680,
      "training_step_time": 9.933592081069946
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 2.4323835372924805,
      "step": 3680
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 3680,
      "training_step_time": 7.4666688442230225
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 2.4274797439575195,
      "step": 3680
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 3680,
      "training_step_time": 7.4529945850372314
    },
    {
      "epoch": 1.40380859375e-05,
      "model_forward_time": 3.1288509368896484,
      "step": 3680
    },
    {
      "epoch": 1.40380859375e-05,
      "step": 3680,
      "training_step_time": 10.017760515213013
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 3.4811935424804688,
      "step": 3681
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 3681,
      "training_step_time": 10.256705045700073
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 3.1332545280456543,
      "step": 3681
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 3681,
      "training_step_time": 9.905510663986206
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 3.1365320682525635,
      "step": 3681
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 3681,
      "training_step_time": 9.913289546966553
    },
    {
      "epoch": 1.4041900634765625e-05,
      "model_forward_time": 3.13485050201416,
      "step": 3681
    },
    {
      "epoch": 1.4041900634765625e-05,
      "step": 3681,
      "training_step_time": 10.262333393096924
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 3.165071725845337,
      "step": 3682
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 3682,
      "training_step_time": 10.230627536773682
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 3.13698148727417,
      "step": 3682
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 3682,
      "training_step_time": 9.915588855743408
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 3.1418797969818115,
      "step": 3682
    },
    {
      "dynamics_loss_avg": 0.0704645898193121,
      "epoch": 1.404571533203125e-05,
      "step": 3682
    },
    {
      "action_loss_avg": 0.03231435711495578,
      "epoch": 1.404571533203125e-05,
      "step": 3682
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 3682,
      "training_step_time": 9.924572944641113
    },
    {
      "epoch": 1.404571533203125e-05,
      "model_forward_time": 3.135902166366577,
      "step": 3682
    },
    {
      "epoch": 1.404571533203125e-05,
      "step": 3682,
      "training_step_time": 10.24979019165039
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 3.170595407485962,
      "step": 3683
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 3683,
      "training_step_time": 9.949769496917725
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 3.388725996017456,
      "step": 3683
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 3683,
      "training_step_time": 10.165879487991333
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 3.1370370388031006,
      "step": 3683
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 3683,
      "training_step_time": 9.915907382965088
    },
    {
      "epoch": 1.4049530029296874e-05,
      "model_forward_time": 3.1354401111602783,
      "step": 3683
    },
    {
      "epoch": 1.4049530029296874e-05,
      "step": 3683,
      "training_step_time": 10.347620964050293
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 3.1554386615753174,
      "step": 3684
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 3684,
      "training_step_time": 9.937399625778198
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 3.14439058303833,
      "step": 3684
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 3684,
      "training_step_time": 9.9203200340271
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 3.4319984912872314,
      "step": 3684
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 3684,
      "training_step_time": 10.201536417007446
    },
    {
      "epoch": 1.40533447265625e-05,
      "model_forward_time": 3.1409966945648193,
      "step": 3684
    },
    {
      "epoch": 1.40533447265625e-05,
      "step": 3684,
      "training_step_time": 10.261814832687378
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 3.1397175788879395,
      "step": 3685
    },
    {
      "dynamics_loss_avg": 0.062373438850045204,
      "epoch": 1.4057159423828125e-05,
      "step": 3685
    },
    {
      "action_loss_avg": 0.03764371033757925,
      "epoch": 1.4057159423828125e-05,
      "step": 3685
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 3685,
      "training_step_time": 9.920226812362671
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 3.1337528228759766,
      "step": 3685
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 3685,
      "training_step_time": 9.907340049743652
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 3.1340279579162598,
      "step": 3685
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 3685,
      "training_step_time": 9.906260013580322
    },
    {
      "epoch": 1.4057159423828125e-05,
      "model_forward_time": 3.1368658542633057,
      "step": 3685
    },
    {
      "epoch": 1.4057159423828125e-05,
      "step": 3685,
      "training_step_time": 10.326621294021606
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 3.141470193862915,
      "step": 3686
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 3686,
      "training_step_time": 9.914490699768066
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 3.133068084716797,
      "step": 3686
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 3686,
      "training_step_time": 9.912991762161255
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 3.13558292388916,
      "step": 3686
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 3686,
      "training_step_time": 9.908722639083862
    },
    {
      "epoch": 1.406097412109375e-05,
      "model_forward_time": 3.1373939514160156,
      "step": 3686
    },
    {
      "epoch": 1.406097412109375e-05,
      "step": 3686,
      "training_step_time": 10.262609004974365
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 3.4548630714416504,
      "step": 3687
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 3687,
      "training_step_time": 10.234014749526978
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 3.1356780529022217,
      "step": 3687
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 3687,
      "training_step_time": 9.918329238891602
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 3.1312062740325928,
      "step": 3687
    },
    {
      "dynamics_loss_avg": 0.06842027865350246,
      "epoch": 1.4064788818359374e-05,
      "step": 3687
    },
    {
      "action_loss_avg": 0.030384546518325804,
      "epoch": 1.4064788818359374e-05,
      "step": 3687
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 3687,
      "training_step_time": 9.909427165985107
    },
    {
      "epoch": 1.4064788818359374e-05,
      "model_forward_time": 3.1381962299346924,
      "step": 3687
    },
    {
      "epoch": 1.4064788818359374e-05,
      "step": 3687,
      "training_step_time": 10.28577446937561
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 2.444885015487671,
      "step": 3688
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 3688,
      "training_step_time": 7.480176210403442
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 3.1363320350646973,
      "step": 3688
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 3688,
      "training_step_time": 9.916488885879517
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 3.397904634475708,
      "step": 3688
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 3688,
      "training_step_time": 10.167454957962036
    },
    {
      "epoch": 1.4068603515625e-05,
      "model_forward_time": 3.142057180404663,
      "step": 3688
    },
    {
      "epoch": 1.4068603515625e-05,
      "step": 3688,
      "training_step_time": 10.277196645736694
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 3.147000551223755,
      "step": 3689
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 3689,
      "training_step_time": 9.92520260810852
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 3.1388018131256104,
      "step": 3689
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 3689,
      "training_step_time": 9.915599346160889
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 2.436586618423462,
      "step": 3689
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 3689,
      "training_step_time": 7.742465972900391
    },
    {
      "epoch": 1.4072418212890625e-05,
      "model_forward_time": 3.142343759536743,
      "step": 3689
    },
    {
      "epoch": 1.4072418212890625e-05,
      "step": 3689,
      "training_step_time": 10.025253772735596
    },
    {
      "epoch": 1.407623291015625e-05,
      "grad_norm": 0.5947638154029846,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 3690
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 3.1737377643585205,
      "step": 3690
    },
    {
      "dynamics_loss_avg": 0.06273008100688457,
      "epoch": 1.407623291015625e-05,
      "step": 3690
    },
    {
      "action_loss_avg": 0.03269213708117604,
      "epoch": 1.407623291015625e-05,
      "step": 3690
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 3690,
      "training_step_time": 9.950622081756592
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 3.1395280361175537,
      "step": 3690
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 3690,
      "training_step_time": 9.919978857040405
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 3.144514322280884,
      "step": 3690
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 3690,
      "training_step_time": 9.9258394241333
    },
    {
      "epoch": 1.407623291015625e-05,
      "model_forward_time": 3.1391594409942627,
      "step": 3690
    },
    {
      "epoch": 1.407623291015625e-05,
      "step": 3690,
      "training_step_time": 10.27745771408081
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 3.1658132076263428,
      "step": 3691
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 3691,
      "training_step_time": 9.94493556022644
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 3.131903886795044,
      "step": 3691
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 3691,
      "training_step_time": 9.907427787780762
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 3.1361536979675293,
      "step": 3691
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 3691,
      "training_step_time": 9.917649030685425
    },
    {
      "epoch": 1.4080047607421874e-05,
      "model_forward_time": 3.1313178539276123,
      "step": 3691
    },
    {
      "epoch": 1.4080047607421874e-05,
      "step": 3691,
      "training_step_time": 10.285948753356934
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 3.448784112930298,
      "step": 3692
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 3692,
      "training_step_time": 10.224925518035889
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 3.1345438957214355,
      "step": 3692
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 3692,
      "training_step_time": 9.909016847610474
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 3.1362648010253906,
      "step": 3692
    },
    {
      "dynamics_loss_avg": 0.06009735614061355,
      "epoch": 1.40838623046875e-05,
      "step": 3692
    },
    {
      "action_loss_avg": 0.04145732098259032,
      "epoch": 1.40838623046875e-05,
      "step": 3692
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 3692,
      "training_step_time": 9.914937019348145
    },
    {
      "epoch": 1.40838623046875e-05,
      "model_forward_time": 3.132650852203369,
      "step": 3692
    },
    {
      "epoch": 1.40838623046875e-05,
      "step": 3692,
      "training_step_time": 10.017460346221924
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 3.1904540061950684,
      "step": 3693
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 3693,
      "training_step_time": 9.969775438308716
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 3.3900375366210938,
      "step": 3693
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 3693,
      "training_step_time": 10.168888807296753
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 3.135128974914551,
      "step": 3693
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 3693,
      "training_step_time": 9.911978244781494
    },
    {
      "epoch": 1.4087677001953125e-05,
      "model_forward_time": 3.130152940750122,
      "step": 3693
    },
    {
      "epoch": 1.4087677001953125e-05,
      "step": 3693,
      "training_step_time": 10.292316913604736
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 3.16678786277771,
      "step": 3694
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 3694,
      "training_step_time": 9.941858768463135
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 3.1349685192108154,
      "step": 3694
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 3694,
      "training_step_time": 9.910017013549805
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 3.129743814468384,
      "step": 3694
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 3694,
      "training_step_time": 10.182041645050049
    },
    {
      "epoch": 1.409149169921875e-05,
      "model_forward_time": 3.1313607692718506,
      "step": 3694
    },
    {
      "epoch": 1.409149169921875e-05,
      "step": 3694,
      "training_step_time": 10.008295774459839
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 3.1650218963623047,
      "step": 3695
    },
    {
      "dynamics_loss_avg": 0.0661420600488782,
      "epoch": 1.4095306396484374e-05,
      "step": 3695
    },
    {
      "action_loss_avg": 0.038115187734365466,
      "epoch": 1.4095306396484374e-05,
      "step": 3695
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 3695,
      "training_step_time": 9.940001726150513
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 3.1326169967651367,
      "step": 3695
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 3695,
      "training_step_time": 9.906503438949585
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 3.130291700363159,
      "step": 3695
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 3695,
      "training_step_time": 9.900644302368164
    },
    {
      "epoch": 1.4095306396484374e-05,
      "model_forward_time": 3.1314687728881836,
      "step": 3695
    },
    {
      "epoch": 1.4095306396484374e-05,
      "step": 3695,
      "training_step_time": 10.29655909538269
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 3.150332450866699,
      "step": 3696
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 3696,
      "training_step_time": 9.927648544311523
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 3.134974956512451,
      "step": 3696
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 3696,
      "training_step_time": 9.912214994430542
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 3.133439779281616,
      "step": 3696
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 3696,
      "training_step_time": 9.906970500946045
    },
    {
      "epoch": 1.409912109375e-05,
      "model_forward_time": 3.137040138244629,
      "step": 3696
    },
    {
      "epoch": 1.409912109375e-05,
      "step": 3696,
      "training_step_time": 10.27067494392395
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 3.171734094619751,
      "step": 3697
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 3697,
      "training_step_time": 10.230924606323242
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 3.1353042125701904,
      "step": 3697
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 3697,
      "training_step_time": 9.909847497940063
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 3.1319072246551514,
      "step": 3697
    },
    {
      "dynamics_loss_avg": 0.06343625299632549,
      "epoch": 1.4102935791015625e-05,
      "step": 3697
    },
    {
      "action_loss_avg": 0.031113792676478624,
      "epoch": 1.4102935791015625e-05,
      "step": 3697
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 3697,
      "training_step_time": 9.908523321151733
    },
    {
      "epoch": 1.4102935791015625e-05,
      "model_forward_time": 3.136284351348877,
      "step": 3697
    },
    {
      "epoch": 1.4102935791015625e-05,
      "step": 3697,
      "training_step_time": 10.006684303283691
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 3.1566338539123535,
      "step": 3698
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 3698,
      "training_step_time": 9.933767318725586
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 3.417219877243042,
      "step": 3698
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 3698,
      "training_step_time": 10.193402767181396
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 3.133798599243164,
      "step": 3698
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 3698,
      "training_step_time": 9.908552169799805
    },
    {
      "epoch": 1.410675048828125e-05,
      "model_forward_time": 3.141613245010376,
      "step": 3698
    },
    {
      "epoch": 1.410675048828125e-05,
      "step": 3698,
      "training_step_time": 10.29502272605896
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 3.186574697494507,
      "step": 3699
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 3699,
      "training_step_time": 9.965928077697754
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 3.1382784843444824,
      "step": 3699
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 3699,
      "training_step_time": 9.913444995880127
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 3.1415016651153564,
      "step": 3699
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 3699,
      "training_step_time": 10.185637950897217
    },
    {
      "epoch": 1.4110565185546874e-05,
      "model_forward_time": 3.1446735858917236,
      "step": 3699
    },
    {
      "epoch": 1.4110565185546874e-05,
      "step": 3699,
      "training_step_time": 10.020841598510742
    },
    {
      "epoch": 1.41143798828125e-05,
      "grad_norm": 0.5089765787124634,
      "learning_rate": 1e-05,
      "loss": 0.09,
      "step": 3700
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 3.155085563659668,
      "step": 3700
    },
    {
      "dynamics_loss_avg": 0.05874042045325041,
      "epoch": 1.41143798828125e-05,
      "step": 3700
    },
    {
      "action_loss_avg": 0.030361569486558437,
      "epoch": 1.41143798828125e-05,
      "step": 3700
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 3700,
      "training_step_time": 9.930079698562622
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 3.137814998626709,
      "step": 3700
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 3700,
      "training_step_time": 9.911141872406006
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 2.4367623329162598,
      "step": 3700
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 3700,
      "training_step_time": 7.470422267913818
    },
    {
      "epoch": 1.41143798828125e-05,
      "model_forward_time": 3.440666913986206,
      "step": 3700
    },
    {
      "epoch": 1.41143798828125e-05,
      "step": 3700,
      "training_step_time": 10.307591676712036
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 3.170609474182129,
      "step": 3701
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 3701,
      "training_step_time": 9.947500467300415
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 3.128074884414673,
      "step": 3701
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 3701,
      "training_step_time": 9.901431798934937
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 3.130946636199951,
      "step": 3701
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 3701,
      "training_step_time": 9.904693365097046
    },
    {
      "epoch": 1.4118194580078125e-05,
      "model_forward_time": 3.1300504207611084,
      "step": 3701
    },
    {
      "epoch": 1.4118194580078125e-05,
      "step": 3701,
      "training_step_time": 10.290353775024414
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 3.1651344299316406,
      "step": 3702
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 3702,
      "training_step_time": 10.218958854675293
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 3.129781484603882,
      "step": 3702
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 3702,
      "training_step_time": 9.901820659637451
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 3.1355881690979004,
      "step": 3702
    },
    {
      "dynamics_loss_avg": 0.05634656399488449,
      "epoch": 1.412200927734375e-05,
      "step": 3702
    },
    {
      "action_loss_avg": 0.028831164166331292,
      "epoch": 1.412200927734375e-05,
      "step": 3702
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 3702,
      "training_step_time": 9.912940502166748
    },
    {
      "epoch": 1.412200927734375e-05,
      "model_forward_time": 3.1312575340270996,
      "step": 3702
    },
    {
      "epoch": 1.412200927734375e-05,
      "step": 3702,
      "training_step_time": 10.322844505310059
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 2.471715211868286,
      "step": 3703
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 3703,
      "training_step_time": 7.503101348876953
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 3.390439510345459,
      "step": 3703
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 3703,
      "training_step_time": 10.169544696807861
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 3.1349077224731445,
      "step": 3703
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 3703,
      "training_step_time": 9.912330627441406
    },
    {
      "epoch": 1.4125823974609374e-05,
      "model_forward_time": 3.133039951324463,
      "step": 3703
    },
    {
      "epoch": 1.4125823974609374e-05,
      "step": 3703,
      "training_step_time": 10.283527851104736
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 3.137726068496704,
      "step": 3704
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 3704,
      "training_step_time": 9.91671872138977
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 3.136519432067871,
      "step": 3704
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 3704,
      "training_step_time": 9.916933536529541
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 3.420520544052124,
      "step": 3704
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 3704,
      "training_step_time": 10.196761131286621
    },
    {
      "epoch": 1.4129638671875e-05,
      "model_forward_time": 3.1342697143554688,
      "step": 3704
    },
    {
      "epoch": 1.4129638671875e-05,
      "step": 3704,
      "training_step_time": 10.306685209274292
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 3.1616461277008057,
      "step": 3705
    },
    {
      "dynamics_loss_avg": 0.05249695777893067,
      "epoch": 1.4133453369140625e-05,
      "step": 3705
    },
    {
      "action_loss_avg": 0.03172647291794419,
      "epoch": 1.4133453369140625e-05,
      "step": 3705
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 3705,
      "training_step_time": 9.938877582550049
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 3.127519130706787,
      "step": 3705
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 3705,
      "training_step_time": 9.90298843383789
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 3.1313509941101074,
      "step": 3705
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 3705,
      "training_step_time": 9.911010503768921
    },
    {
      "epoch": 1.4133453369140625e-05,
      "model_forward_time": 3.1321871280670166,
      "step": 3705
    },
    {
      "epoch": 1.4133453369140625e-05,
      "step": 3705,
      "training_step_time": 10.281972169876099
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 2.4400815963745117,
      "step": 3706
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 3706,
      "training_step_time": 7.471852779388428
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 3.133357048034668,
      "step": 3706
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 3706,
      "training_step_time": 9.905968427658081
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 3.1283140182495117,
      "step": 3706
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 3706,
      "training_step_time": 9.893298387527466
    },
    {
      "epoch": 1.413726806640625e-05,
      "model_forward_time": 3.1311147212982178,
      "step": 3706
    },
    {
      "epoch": 1.413726806640625e-05,
      "step": 3706,
      "training_step_time": 10.206566095352173
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 3.139008045196533,
      "step": 3707
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 3707,
      "training_step_time": 9.908109664916992
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 2.690770387649536,
      "step": 3707
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 3707,
      "training_step_time": 7.713740825653076
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 3.1249966621398926,
      "step": 3707
    },
    {
      "dynamics_loss_avg": 0.06902762316167355,
      "epoch": 1.4141082763671874e-05,
      "step": 3707
    },
    {
      "action_loss_avg": 0.03468290045857429,
      "epoch": 1.4141082763671874e-05,
      "step": 3707
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 3707,
      "training_step_time": 9.899034261703491
    },
    {
      "epoch": 1.4141082763671874e-05,
      "model_forward_time": 3.132702112197876,
      "step": 3707
    },
    {
      "epoch": 1.4141082763671874e-05,
      "step": 3707,
      "training_step_time": 10.32567834854126
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 3.186162233352661,
      "step": 3708
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 3708,
      "training_step_time": 9.963418960571289
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 3.13641095161438,
      "step": 3708
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 3708,
      "training_step_time": 10.181864023208618
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 3.1336686611175537,
      "step": 3708
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 3708,
      "training_step_time": 9.9064359664917
    },
    {
      "epoch": 1.41448974609375e-05,
      "model_forward_time": 3.1380598545074463,
      "step": 3708
    },
    {
      "epoch": 1.41448974609375e-05,
      "step": 3708,
      "training_step_time": 10.010388612747192
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 3.21110463142395,
      "step": 3709
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 3709,
      "training_step_time": 9.993059158325195
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 3.1372201442718506,
      "step": 3709
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 3709,
      "training_step_time": 9.914125442504883
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 3.3968238830566406,
      "step": 3709
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 3709,
      "training_step_time": 10.184738159179688
    },
    {
      "epoch": 1.4148712158203125e-05,
      "model_forward_time": 3.1440672874450684,
      "step": 3709
    },
    {
      "epoch": 1.4148712158203125e-05,
      "step": 3709,
      "training_step_time": 10.30783987045288
    },
    {
      "epoch": 1.415252685546875e-05,
      "grad_norm": 0.4714811146259308,
      "learning_rate": 1e-05,
      "loss": 0.0953,
      "step": 3710
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 3.162281036376953,
      "step": 3710
    },
    {
      "dynamics_loss_avg": 0.052939323335886,
      "epoch": 1.415252685546875e-05,
      "step": 3710
    },
    {
      "action_loss_avg": 0.019920621532946825,
      "epoch": 1.415252685546875e-05,
      "step": 3710
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 3710,
      "training_step_time": 9.945966720581055
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 3.144015073776245,
      "step": 3710
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 3710,
      "training_step_time": 9.93704628944397
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 3.1443235874176025,
      "step": 3710
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 3710,
      "training_step_time": 9.92579436302185
    },
    {
      "epoch": 1.415252685546875e-05,
      "model_forward_time": 2.436281681060791,
      "step": 3710
    },
    {
      "epoch": 1.415252685546875e-05,
      "step": 3710,
      "training_step_time": 10.045408010482788
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 3.155402660369873,
      "step": 3711
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 3711,
      "training_step_time": 9.935473680496216
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 3.1378989219665527,
      "step": 3711
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 3711,
      "training_step_time": 9.911120176315308
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 3.140846014022827,
      "step": 3711
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 3711,
      "training_step_time": 9.915707349777222
    },
    {
      "epoch": 1.4156341552734374e-05,
      "model_forward_time": 3.1345648765563965,
      "step": 3711
    },
    {
      "epoch": 1.4156341552734374e-05,
      "step": 3711,
      "training_step_time": 10.015759468078613
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 3.4241411685943604,
      "step": 3712
    },
    {
      "epoch": 1.416015625e-05,
      "step": 3712,
      "training_step_time": 10.199685335159302
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 3.1379549503326416,
      "step": 3712
    },
    {
      "epoch": 1.416015625e-05,
      "step": 3712,
      "training_step_time": 9.918019771575928
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 3.141469717025757,
      "step": 3712
    },
    {
      "dynamics_loss_avg": 0.0479197546839714,
      "epoch": 1.416015625e-05,
      "step": 3712
    },
    {
      "action_loss_avg": 0.020290086045861244,
      "epoch": 1.416015625e-05,
      "step": 3712
    },
    {
      "epoch": 1.416015625e-05,
      "step": 3712,
      "training_step_time": 9.922903060913086
    },
    {
      "epoch": 1.416015625e-05,
      "model_forward_time": 3.13845157623291,
      "step": 3712
    },
    {
      "epoch": 1.416015625e-05,
      "step": 3712,
      "training_step_time": 10.29752254486084
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 2.4501664638519287,
      "step": 3713
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 3713,
      "training_step_time": 7.502926349639893
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 3.1398026943206787,
      "step": 3713
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 3713,
      "training_step_time": 10.190509796142578
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 2.4374561309814453,
      "step": 3713
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 3713,
      "training_step_time": 7.47119665145874
    },
    {
      "epoch": 1.4163970947265625e-05,
      "model_forward_time": 3.1265318393707275,
      "step": 3713
    },
    {
      "epoch": 1.4163970947265625e-05,
      "step": 3713,
      "training_step_time": 10.296054363250732
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 3.1479742527008057,
      "step": 3714
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 3714,
      "training_step_time": 9.923444986343384
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 3.1316335201263428,
      "step": 3714
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 3714,
      "training_step_time": 9.904876470565796
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 2.430276393890381,
      "step": 3714
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 3714,
      "training_step_time": 7.717809200286865
    },
    {
      "epoch": 1.416778564453125e-05,
      "model_forward_time": 3.1266937255859375,
      "step": 3714
    },
    {
      "epoch": 1.416778564453125e-05,
      "step": 3714,
      "training_step_time": 10.308135509490967
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 3.134546995162964,
      "step": 3715
    },
    {
      "dynamics_loss_avg": 0.06272121034562587,
      "epoch": 1.4171600341796874e-05,
      "step": 3715
    },
    {
      "action_loss_avg": 0.03373169973492622,
      "epoch": 1.4171600341796874e-05,
      "step": 3715
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 3715,
      "training_step_time": 9.910131216049194
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 3.1342251300811768,
      "step": 3715
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 3715,
      "training_step_time": 9.906297206878662
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 3.1273932456970215,
      "step": 3715
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 3715,
      "training_step_time": 9.903494834899902
    },
    {
      "epoch": 1.4171600341796874e-05,
      "model_forward_time": 3.385221481323242,
      "step": 3715
    },
    {
      "epoch": 1.4171600341796874e-05,
      "step": 3715,
      "training_step_time": 10.250421047210693
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 3.146623373031616,
      "step": 3716
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 3716,
      "training_step_time": 9.922670841217041
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 3.1367685794830322,
      "step": 3716
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 3716,
      "training_step_time": 9.913853168487549
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 3.1334388256073,
      "step": 3716
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 3716,
      "training_step_time": 9.910095691680908
    },
    {
      "epoch": 1.41754150390625e-05,
      "model_forward_time": 3.1318228244781494,
      "step": 3716
    },
    {
      "epoch": 1.41754150390625e-05,
      "step": 3716,
      "training_step_time": 10.020774364471436
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 3.1781342029571533,
      "step": 3717
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 3717,
      "training_step_time": 10.238149881362915
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 3.1383354663848877,
      "step": 3717
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 3717,
      "training_step_time": 9.918317556381226
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 3.137021541595459,
      "step": 3717
    },
    {
      "dynamics_loss_avg": 0.06683645397424698,
      "epoch": 1.4179229736328125e-05,
      "step": 3717
    },
    {
      "action_loss_avg": 0.02363038593903184,
      "epoch": 1.4179229736328125e-05,
      "step": 3717
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 3717,
      "training_step_time": 9.91507339477539
    },
    {
      "epoch": 1.4179229736328125e-05,
      "model_forward_time": 3.14225697517395,
      "step": 3717
    },
    {
      "epoch": 1.4179229736328125e-05,
      "step": 3717,
      "training_step_time": 10.028640747070312
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 3.185601234436035,
      "step": 3718
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 3718,
      "training_step_time": 9.96336317062378
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 3.4275453090667725,
      "step": 3718
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 3718,
      "training_step_time": 10.20795488357544
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 3.1358320713043213,
      "step": 3718
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 3718,
      "training_step_time": 9.916839361190796
    },
    {
      "epoch": 1.418304443359375e-05,
      "model_forward_time": 3.13934588432312,
      "step": 3718
    },
    {
      "epoch": 1.418304443359375e-05,
      "step": 3718,
      "training_step_time": 10.310160875320435
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 3.181568145751953,
      "step": 3719
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 3719,
      "training_step_time": 9.975882768630981
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 3.135672092437744,
      "step": 3719
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 3719,
      "training_step_time": 9.912606716156006
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 3.1367251873016357,
      "step": 3719
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 3719,
      "training_step_time": 10.209271430969238
    },
    {
      "epoch": 1.4186859130859374e-05,
      "model_forward_time": 3.137484312057495,
      "step": 3719
    },
    {
      "epoch": 1.4186859130859374e-05,
      "step": 3719,
      "training_step_time": 10.273428678512573
    },
    {
      "epoch": 1.4190673828125e-05,
      "grad_norm": 0.48695486783981323,
      "learning_rate": 1e-05,
      "loss": 0.0893,
      "step": 3720
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 3.151001453399658,
      "step": 3720
    },
    {
      "dynamics_loss_avg": 0.058098793402314185,
      "epoch": 1.4190673828125e-05,
      "step": 3720
    },
    {
      "action_loss_avg": 0.026911695022135973,
      "epoch": 1.4190673828125e-05,
      "step": 3720
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 3720,
      "training_step_time": 9.942615985870361
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 3.1342904567718506,
      "step": 3720
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 3720,
      "training_step_time": 9.91608738899231
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 3.143005132675171,
      "step": 3720
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 3720,
      "training_step_time": 9.927709579467773
    },
    {
      "epoch": 1.4190673828125e-05,
      "model_forward_time": 3.41880202293396,
      "step": 3720
    },
    {
      "epoch": 1.4190673828125e-05,
      "step": 3720,
      "training_step_time": 10.292935371398926
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 3.1713900566101074,
      "step": 3721
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 3721,
      "training_step_time": 9.951361656188965
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 3.1359145641326904,
      "step": 3721
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 3721,
      "training_step_time": 9.919562101364136
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 3.1373989582061768,
      "step": 3721
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 3721,
      "training_step_time": 9.92291259765625
    },
    {
      "epoch": 1.4194488525390625e-05,
      "model_forward_time": 3.138293504714966,
      "step": 3721
    },
    {
      "epoch": 1.4194488525390625e-05,
      "step": 3721,
      "training_step_time": 10.294324159622192
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 3.4209320545196533,
      "step": 3722
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 3722,
      "training_step_time": 10.20460844039917
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 3.1338276863098145,
      "step": 3722
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 3722,
      "training_step_time": 9.915158033370972
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 3.1362531185150146,
      "step": 3722
    },
    {
      "dynamics_loss_avg": 0.07731214165687561,
      "epoch": 1.419830322265625e-05,
      "step": 3722
    },
    {
      "action_loss_avg": 0.04061290677636862,
      "epoch": 1.419830322265625e-05,
      "step": 3722
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 3722,
      "training_step_time": 9.915165662765503
    },
    {
      "epoch": 1.419830322265625e-05,
      "model_forward_time": 3.132310628890991,
      "step": 3722
    },
    {
      "epoch": 1.419830322265625e-05,
      "step": 3722,
      "training_step_time": 10.297979354858398
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 3.141481876373291,
      "step": 3723
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 3723,
      "training_step_time": 9.916658163070679
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 3.391555070877075,
      "step": 3723
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 3723,
      "training_step_time": 10.170311689376831
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 3.1343328952789307,
      "step": 3723
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 3723,
      "training_step_time": 9.912848472595215
    },
    {
      "epoch": 1.4202117919921874e-05,
      "model_forward_time": 3.1337263584136963,
      "step": 3723
    },
    {
      "epoch": 1.4202117919921874e-05,
      "step": 3723,
      "training_step_time": 10.280789136886597
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 3.171839952468872,
      "step": 3724
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 3724,
      "training_step_time": 9.945404767990112
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 3.1454010009765625,
      "step": 3724
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 3724,
      "training_step_time": 9.92025876045227
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 3.430493116378784,
      "step": 3724
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 3724,
      "training_step_time": 10.204577445983887
    },
    {
      "epoch": 1.42059326171875e-05,
      "model_forward_time": 3.140923261642456,
      "step": 3724
    },
    {
      "epoch": 1.42059326171875e-05,
      "step": 3724,
      "training_step_time": 10.062668085098267
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 3.173959732055664,
      "step": 3725
    },
    {
      "dynamics_loss_avg": 0.06896081026643515,
      "epoch": 1.4209747314453125e-05,
      "step": 3725
    },
    {
      "action_loss_avg": 0.036034750193357466,
      "epoch": 1.4209747314453125e-05,
      "step": 3725
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 3725,
      "training_step_time": 9.950421810150146
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 3.1415605545043945,
      "step": 3725
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 3725,
      "training_step_time": 9.91768217086792
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 3.1397597789764404,
      "step": 3725
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 3725,
      "training_step_time": 9.917924642562866
    },
    {
      "epoch": 1.4209747314453125e-05,
      "model_forward_time": 3.1451847553253174,
      "step": 3725
    },
    {
      "epoch": 1.4209747314453125e-05,
      "step": 3725,
      "training_step_time": 10.338731527328491
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 3.164379119873047,
      "step": 3726
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 3726,
      "training_step_time": 9.948265552520752
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 3.146174192428589,
      "step": 3726
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 3726,
      "training_step_time": 9.92520809173584
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 3.141329526901245,
      "step": 3726
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 3726,
      "training_step_time": 9.92275881767273
    },
    {
      "epoch": 1.421356201171875e-05,
      "model_forward_time": 3.1448941230773926,
      "step": 3726
    },
    {
      "epoch": 1.421356201171875e-05,
      "step": 3726,
      "training_step_time": 10.031107187271118
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 3.412297010421753,
      "step": 3727
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 3727,
      "training_step_time": 10.191110849380493
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 3.144105911254883,
      "step": 3727
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 3727,
      "training_step_time": 9.925856351852417
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 3.140355348587036,
      "step": 3727
    },
    {
      "dynamics_loss_avg": 0.05757462698966265,
      "epoch": 1.4217376708984374e-05,
      "step": 3727
    },
    {
      "action_loss_avg": 0.027415412990376354,
      "epoch": 1.4217376708984374e-05,
      "step": 3727
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 3727,
      "training_step_time": 9.926947832107544
    },
    {
      "epoch": 1.4217376708984374e-05,
      "model_forward_time": 2.464510440826416,
      "step": 3727
    },
    {
      "epoch": 1.4217376708984374e-05,
      "step": 3727,
      "training_step_time": 10.03148627281189
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 3.1807878017425537,
      "step": 3728
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 3728,
      "training_step_time": 9.962452173233032
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 3.1464555263519287,
      "step": 3728
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 3728,
      "training_step_time": 9.925747871398926
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 3.3856470584869385,
      "step": 3728
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 3728,
      "training_step_time": 10.161789894104004
    },
    {
      "epoch": 1.422119140625e-05,
      "model_forward_time": 3.142577648162842,
      "step": 3728
    },
    {
      "epoch": 1.422119140625e-05,
      "step": 3728,
      "training_step_time": 10.28623366355896
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 3.159494400024414,
      "step": 3729
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 3729,
      "training_step_time": 9.930958032608032
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 3.1391148567199707,
      "step": 3729
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 3729,
      "training_step_time": 9.913557052612305
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 3.405318021774292,
      "step": 3729
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 3729,
      "training_step_time": 10.183101892471313
    },
    {
      "epoch": 1.4225006103515625e-05,
      "model_forward_time": 3.143695116043091,
      "step": 3729
    },
    {
      "epoch": 1.4225006103515625e-05,
      "step": 3729,
      "training_step_time": 10.326371908187866
    },
    {
      "epoch": 1.422882080078125e-05,
      "grad_norm": 0.5305073857307434,
      "learning_rate": 1e-05,
      "loss": 0.0922,
      "step": 3730
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 3.1489386558532715,
      "step": 3730
    },
    {
      "dynamics_loss_avg": 0.07407446093857288,
      "epoch": 1.422882080078125e-05,
      "step": 3730
    },
    {
      "action_loss_avg": 0.03659368772059679,
      "epoch": 1.422882080078125e-05,
      "step": 3730
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 3730,
      "training_step_time": 9.933653831481934
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 3.14477801322937,
      "step": 3730
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 3730,
      "training_step_time": 9.920374870300293
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 2.441697835922241,
      "step": 3730
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 3730,
      "training_step_time": 7.478179454803467
    },
    {
      "epoch": 1.422882080078125e-05,
      "model_forward_time": 3.1385176181793213,
      "step": 3730
    },
    {
      "epoch": 1.422882080078125e-05,
      "step": 3730,
      "training_step_time": 10.286497592926025
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 3.45619797706604,
      "step": 3731
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 3731,
      "training_step_time": 10.23554515838623
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 3.1410672664642334,
      "step": 3731
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 3731,
      "training_step_time": 9.918707609176636
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 3.142221450805664,
      "step": 3731
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 3731,
      "training_step_time": 9.920067548751831
    },
    {
      "epoch": 1.4232635498046875e-05,
      "model_forward_time": 2.4406564235687256,
      "step": 3731
    },
    {
      "epoch": 1.4232635498046875e-05,
      "step": 3731,
      "training_step_time": 10.013052225112915
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 3.446518659591675,
      "step": 3732
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 3732,
      "training_step_time": 10.218191385269165
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 3.136075973510742,
      "step": 3732
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 3732,
      "training_step_time": 9.909759044647217
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 3.1373798847198486,
      "step": 3732
    },
    {
      "dynamics_loss_avg": 0.06746394569054245,
      "epoch": 1.42364501953125e-05,
      "step": 3732
    },
    {
      "action_loss_avg": 0.040573902684263884,
      "epoch": 1.42364501953125e-05,
      "step": 3732
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 3732,
      "training_step_time": 9.910446166992188
    },
    {
      "epoch": 1.42364501953125e-05,
      "model_forward_time": 3.136352777481079,
      "step": 3732
    },
    {
      "epoch": 1.42364501953125e-05,
      "step": 3732,
      "training_step_time": 10.28841233253479
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 3.1527140140533447,
      "step": 3733
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 3733,
      "training_step_time": 9.926523447036743
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 3.1312334537506104,
      "step": 3733
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 3733,
      "training_step_time": 9.908367395401001
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 2.6964430809020996,
      "step": 3733
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 3733,
      "training_step_time": 7.726076364517212
    },
    {
      "epoch": 1.4240264892578125e-05,
      "model_forward_time": 2.4255826473236084,
      "step": 3733
    },
    {
      "epoch": 1.4240264892578125e-05,
      "step": 3733,
      "training_step_time": 10.28835391998291
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 2.4362986087799072,
      "step": 3734
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 3734,
      "training_step_time": 7.457191228866577
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 3.132111072540283,
      "step": 3734
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 3734,
      "training_step_time": 9.90430235862732
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 3.132290840148926,
      "step": 3734
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 3734,
      "training_step_time": 9.905288457870483
    },
    {
      "epoch": 1.424407958984375e-05,
      "model_forward_time": 3.3733975887298584,
      "step": 3734
    },
    {
      "epoch": 1.424407958984375e-05,
      "step": 3734,
      "training_step_time": 10.284632444381714
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 2.478825330734253,
      "step": 3735
    },
    {
      "dynamics_loss_avg": 0.062285635620355606,
      "epoch": 1.4247894287109375e-05,
      "step": 3735
    },
    {
      "action_loss_avg": 0.03070751894265413,
      "epoch": 1.4247894287109375e-05,
      "step": 3735
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 3735,
      "training_step_time": 7.5157692432403564
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 3.1414859294891357,
      "step": 3735
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 3735,
      "training_step_time": 9.916552782058716
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 3.1438536643981934,
      "step": 3735
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 3735,
      "training_step_time": 9.91445016860962
    },
    {
      "epoch": 1.4247894287109375e-05,
      "model_forward_time": 3.141551971435547,
      "step": 3735
    },
    {
      "epoch": 1.4247894287109375e-05,
      "step": 3735,
      "training_step_time": 10.326883792877197
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 3.4420266151428223,
      "step": 3736
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 3736,
      "training_step_time": 10.207618236541748
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 3.139281749725342,
      "step": 3736
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 3736,
      "training_step_time": 9.906261444091797
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 3.135781764984131,
      "step": 3736
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 3736,
      "training_step_time": 9.902660608291626
    },
    {
      "epoch": 1.4251708984375e-05,
      "model_forward_time": 3.1442925930023193,
      "step": 3736
    },
    {
      "epoch": 1.4251708984375e-05,
      "step": 3736,
      "training_step_time": 10.005266427993774
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 3.169980764389038,
      "step": 3737
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 3737,
      "training_step_time": 10.240906000137329
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 3.1345205307006836,
      "step": 3737
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 3737,
      "training_step_time": 9.905517339706421
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 3.129643678665161,
      "step": 3737
    },
    {
      "dynamics_loss_avg": 0.05100572314113379,
      "epoch": 1.4255523681640625e-05,
      "step": 3737
    },
    {
      "action_loss_avg": 0.020792519301176073,
      "epoch": 1.4255523681640625e-05,
      "step": 3737
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 3737,
      "training_step_time": 9.908525228500366
    },
    {
      "epoch": 1.4255523681640625e-05,
      "model_forward_time": 3.1435136795043945,
      "step": 3737
    },
    {
      "epoch": 1.4255523681640625e-05,
      "step": 3737,
      "training_step_time": 10.016951560974121
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 3.145617961883545,
      "step": 3738
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 3738,
      "training_step_time": 9.924265146255493
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 2.7120168209075928,
      "step": 3738
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 3738,
      "training_step_time": 7.745092391967773
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 2.4316582679748535,
      "step": 3738
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 3738,
      "training_step_time": 7.465370416641235
    },
    {
      "epoch": 1.425933837890625e-05,
      "model_forward_time": 3.1316678524017334,
      "step": 3738
    },
    {
      "epoch": 1.425933837890625e-05,
      "step": 3738,
      "training_step_time": 10.287216186523438
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 3.1491446495056152,
      "step": 3739
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 3739,
      "training_step_time": 9.926701545715332
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 3.132201671600342,
      "step": 3739
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 3739,
      "training_step_time": 9.907113790512085
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 3.1375784873962402,
      "step": 3739
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 3739,
      "training_step_time": 9.918354034423828
    },
    {
      "epoch": 1.4263153076171875e-05,
      "model_forward_time": 3.406858205795288,
      "step": 3739
    },
    {
      "epoch": 1.4263153076171875e-05,
      "step": 3739,
      "training_step_time": 10.27469253540039
    },
    {
      "epoch": 1.42669677734375e-05,
      "grad_norm": 0.384670227766037,
      "learning_rate": 1e-05,
      "loss": 0.0909,
      "step": 3740
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 3.1734254360198975,
      "step": 3740
    },
    {
      "dynamics_loss_avg": 0.0654872426763177,
      "epoch": 1.42669677734375e-05,
      "step": 3740
    },
    {
      "action_loss_avg": 0.035136750619858506,
      "epoch": 1.42669677734375e-05,
      "step": 3740
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 3740,
      "training_step_time": 9.951769828796387
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 3.1353232860565186,
      "step": 3740
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 3740,
      "training_step_time": 9.910391330718994
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 3.134375810623169,
      "step": 3740
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 3740,
      "training_step_time": 9.908612489700317
    },
    {
      "epoch": 1.42669677734375e-05,
      "model_forward_time": 3.421473979949951,
      "step": 3740
    },
    {
      "epoch": 1.42669677734375e-05,
      "step": 3740,
      "training_step_time": 10.29256796836853
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 3.2027366161346436,
      "step": 3741
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 3741,
      "training_step_time": 9.991600275039673
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 3.1366021633148193,
      "step": 3741
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 3741,
      "training_step_time": 9.911728858947754
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 3.140549659729004,
      "step": 3741
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 3741,
      "training_step_time": 9.915826797485352
    },
    {
      "epoch": 1.4270782470703125e-05,
      "model_forward_time": 3.1382713317871094,
      "step": 3741
    },
    {
      "epoch": 1.4270782470703125e-05,
      "step": 3741,
      "training_step_time": 10.255008697509766
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 3.1467387676239014,
      "step": 3742
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 3742,
      "training_step_time": 10.191063165664673
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 3.1376988887786865,
      "step": 3742
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 3742,
      "training_step_time": 9.915684461593628
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 3.1433424949645996,
      "step": 3742
    },
    {
      "dynamics_loss_avg": 0.07631359845399857,
      "epoch": 1.427459716796875e-05,
      "step": 3742
    },
    {
      "action_loss_avg": 0.03347677905112505,
      "epoch": 1.427459716796875e-05,
      "step": 3742
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 3742,
      "training_step_time": 9.920014381408691
    },
    {
      "epoch": 1.427459716796875e-05,
      "model_forward_time": 3.138831853866577,
      "step": 3742
    },
    {
      "epoch": 1.427459716796875e-05,
      "step": 3742,
      "training_step_time": 10.015029191970825
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 2.474935293197632,
      "step": 3743
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 3743,
      "training_step_time": 7.507122755050659
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 3.3960559368133545,
      "step": 3743
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 3743,
      "training_step_time": 10.167067527770996
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 3.1394097805023193,
      "step": 3743
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 3743,
      "training_step_time": 9.914983749389648
    },
    {
      "epoch": 1.4278411865234375e-05,
      "model_forward_time": 3.138800859451294,
      "step": 3743
    },
    {
      "epoch": 1.4278411865234375e-05,
      "step": 3743,
      "training_step_time": 10.298388242721558
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 3.1696388721466064,
      "step": 3744
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 3744,
      "training_step_time": 9.948360681533813
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 3.143805742263794,
      "step": 3744
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 3744,
      "training_step_time": 9.917748212814331
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 3.423867702484131,
      "step": 3744
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 3744,
      "training_step_time": 10.196915626525879
    },
    {
      "epoch": 1.42822265625e-05,
      "model_forward_time": 3.136218309402466,
      "step": 3744
    },
    {
      "epoch": 1.42822265625e-05,
      "step": 3744,
      "training_step_time": 10.267735004425049
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 2.4694061279296875,
      "step": 3745
    },
    {
      "dynamics_loss_avg": 0.05784404259175062,
      "epoch": 1.4286041259765625e-05,
      "step": 3745
    },
    {
      "action_loss_avg": 0.036281809024512765,
      "epoch": 1.4286041259765625e-05,
      "step": 3745
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 3745,
      "training_step_time": 7.497830629348755
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 3.1342532634735107,
      "step": 3745
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 3745,
      "training_step_time": 9.901514530181885
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 3.131136894226074,
      "step": 3745
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 3745,
      "training_step_time": 9.907227516174316
    },
    {
      "epoch": 1.4286041259765625e-05,
      "model_forward_time": 3.136700391769409,
      "step": 3745
    },
    {
      "epoch": 1.4286041259765625e-05,
      "step": 3745,
      "training_step_time": 10.29159164428711
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 3.143889904022217,
      "step": 3746
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 3746,
      "training_step_time": 9.916493892669678
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 3.1295406818389893,
      "step": 3746
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 3746,
      "training_step_time": 9.904511451721191
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 3.1312036514282227,
      "step": 3746
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 3746,
      "training_step_time": 9.903892040252686
    },
    {
      "epoch": 1.428985595703125e-05,
      "model_forward_time": 2.427907943725586,
      "step": 3746
    },
    {
      "epoch": 1.428985595703125e-05,
      "step": 3746,
      "training_step_time": 10.017422676086426
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 3.1676175594329834,
      "step": 3747
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 3747,
      "training_step_time": 9.94572639465332
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 3.3984920978546143,
      "step": 3747
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 3747,
      "training_step_time": 10.172854661941528
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 3.1330273151397705,
      "step": 3747
    },
    {
      "dynamics_loss_avg": 0.0852923447266221,
      "epoch": 1.4293670654296875e-05,
      "step": 3747
    },
    {
      "action_loss_avg": 0.03582496820017696,
      "epoch": 1.4293670654296875e-05,
      "step": 3747
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 3747,
      "training_step_time": 9.907870531082153
    },
    {
      "epoch": 1.4293670654296875e-05,
      "model_forward_time": 3.144087076187134,
      "step": 3747
    },
    {
      "epoch": 1.4293670654296875e-05,
      "step": 3747,
      "training_step_time": 10.023192405700684
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 2.453516721725464,
      "step": 3748
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 3748,
      "training_step_time": 7.483583688735962
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 3.1356704235076904,
      "step": 3748
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 3748,
      "training_step_time": 10.185590982437134
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 3.139733076095581,
      "step": 3748
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 3748,
      "training_step_time": 9.912721395492554
    },
    {
      "epoch": 1.42974853515625e-05,
      "model_forward_time": 2.4370479583740234,
      "step": 3748
    },
    {
      "epoch": 1.42974853515625e-05,
      "step": 3748,
      "training_step_time": 10.272517442703247
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 3.177870750427246,
      "step": 3749
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 3749,
      "training_step_time": 9.952040195465088
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 3.1335034370422363,
      "step": 3749
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 3749,
      "training_step_time": 9.898280143737793
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 3.3930346965789795,
      "step": 3749
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 3749,
      "training_step_time": 10.159456968307495
    },
    {
      "epoch": 1.4301300048828125e-05,
      "model_forward_time": 3.1344151496887207,
      "step": 3749
    },
    {
      "epoch": 1.4301300048828125e-05,
      "step": 3749,
      "training_step_time": 10.005908489227295
    },
    {
      "epoch": 1.430511474609375e-05,
      "grad_norm": 0.563874363899231,
      "learning_rate": 1e-05,
      "loss": 0.094,
      "step": 3750
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 2.474888801574707,
      "step": 3750
    },
    {
      "dynamics_loss_avg": 0.06087119057774544,
      "epoch": 1.430511474609375e-05,
      "step": 3750
    },
    {
      "action_loss_avg": 0.03047601394355297,
      "epoch": 1.430511474609375e-05,
      "step": 3750
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 3750,
      "training_step_time": 7.503496885299683
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 2.4293642044067383,
      "step": 3750
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 3750,
      "training_step_time": 7.453491449356079
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 3.1301629543304443,
      "step": 3750
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 3750,
      "training_step_time": 9.898571968078613
    },
    {
      "epoch": 1.430511474609375e-05,
      "model_forward_time": 3.1321487426757812,
      "step": 3750
    },
    {
      "epoch": 1.430511474609375e-05,
      "step": 3750,
      "training_step_time": 10.28556752204895
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 3.4452433586120605,
      "step": 3751
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 3751,
      "training_step_time": 10.217033624649048
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 2.43544340133667,
      "step": 3751
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 3751,
      "training_step_time": 7.464411497116089
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 2.431245803833008,
      "step": 3751
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 3751,
      "training_step_time": 7.4566874504089355
    },
    {
      "epoch": 1.4308929443359375e-05,
      "model_forward_time": 3.129146099090576,
      "step": 3751
    },
    {
      "epoch": 1.4308929443359375e-05,
      "step": 3751,
      "training_step_time": 10.009684801101685
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 3.1480953693389893,
      "step": 3752
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 3752,
      "training_step_time": 9.921745538711548
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 3.380181074142456,
      "step": 3752
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 3752,
      "training_step_time": 10.153371810913086
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 2.434028387069702,
      "step": 3752
    },
    {
      "dynamics_loss_avg": 0.06377049945294858,
      "epoch": 1.4312744140625e-05,
      "step": 3752
    },
    {
      "action_loss_avg": 0.034170516580343244,
      "epoch": 1.4312744140625e-05,
      "step": 3752
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 3752,
      "training_step_time": 7.466354608535767
    },
    {
      "epoch": 1.4312744140625e-05,
      "model_forward_time": 2.4319379329681396,
      "step": 3752
    },
    {
      "epoch": 1.4312744140625e-05,
      "step": 3752,
      "training_step_time": 10.29760479927063
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 3.142071485519409,
      "step": 3753
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 3753,
      "training_step_time": 9.918447971343994
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 2.4352378845214844,
      "step": 3753
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 3753,
      "training_step_time": 7.470517635345459
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 3.409494161605835,
      "step": 3753
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 3753,
      "training_step_time": 10.187098264694214
    },
    {
      "epoch": 1.4316558837890625e-05,
      "model_forward_time": 3.134890079498291,
      "step": 3753
    },
    {
      "epoch": 1.4316558837890625e-05,
      "step": 3753,
      "training_step_time": 10.345564603805542
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 2.4457039833068848,
      "step": 3754
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 3754,
      "training_step_time": 7.491050958633423
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 3.132171154022217,
      "step": 3754
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 3754,
      "training_step_time": 9.90366506576538
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 3.132023811340332,
      "step": 3754
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 3754,
      "training_step_time": 9.902875185012817
    },
    {
      "epoch": 1.432037353515625e-05,
      "model_forward_time": 2.702202081680298,
      "step": 3754
    },
    {
      "epoch": 1.432037353515625e-05,
      "step": 3754,
      "training_step_time": 10.26881456375122
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 3.138214111328125,
      "step": 3755
    },
    {
      "dynamics_loss_avg": 0.07421433590352536,
      "epoch": 1.4324188232421875e-05,
      "step": 3755
    },
    {
      "action_loss_avg": 0.03402254525572061,
      "epoch": 1.4324188232421875e-05,
      "step": 3755
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 3755,
      "training_step_time": 9.930202007293701
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 3.13034987449646,
      "step": 3755
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 3755,
      "training_step_time": 9.911698818206787
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 2.4329254627227783,
      "step": 3755
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 3755,
      "training_step_time": 7.465017080307007
    },
    {
      "epoch": 1.4324188232421875e-05,
      "model_forward_time": 3.134310722351074,
      "step": 3755
    },
    {
      "epoch": 1.4324188232421875e-05,
      "step": 3755,
      "training_step_time": 10.317179441452026
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 3.427814483642578,
      "step": 3756
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 3756,
      "training_step_time": 10.202088117599487
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 3.133474588394165,
      "step": 3756
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 3756,
      "training_step_time": 9.905266046524048
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 3.128480911254883,
      "step": 3756
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 3756,
      "training_step_time": 9.901501893997192
    },
    {
      "epoch": 1.43280029296875e-05,
      "model_forward_time": 3.132493019104004,
      "step": 3756
    },
    {
      "epoch": 1.43280029296875e-05,
      "step": 3756,
      "training_step_time": 10.007062673568726
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 2.442751884460449,
      "step": 3757
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 3757,
      "training_step_time": 7.475537061691284
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 2.4284374713897705,
      "step": 3757
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 3757,
      "training_step_time": 7.75575065612793
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 3.1297218799591064,
      "step": 3757
    },
    {
      "dynamics_loss_avg": 0.07379622068256139,
      "epoch": 1.4331817626953125e-05,
      "step": 3757
    },
    {
      "action_loss_avg": 0.03620899124071002,
      "epoch": 1.4331817626953125e-05,
      "step": 3757
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 3757,
      "training_step_time": 9.905084609985352
    },
    {
      "epoch": 1.4331817626953125e-05,
      "model_forward_time": 3.136279582977295,
      "step": 3757
    },
    {
      "epoch": 1.4331817626953125e-05,
      "step": 3757,
      "training_step_time": 10.29936408996582
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 3.1447997093200684,
      "step": 3758
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 3758,
      "training_step_time": 9.924331426620483
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 3.129826307296753,
      "step": 3758
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 3758,
      "training_step_time": 9.908654928207397
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 1.7410156726837158,
      "step": 3758
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 3758,
      "training_step_time": 5.20040225982666
    },
    {
      "epoch": 1.433563232421875e-05,
      "model_forward_time": 3.4194564819335938,
      "step": 3758
    },
    {
      "epoch": 1.433563232421875e-05,
      "step": 3758,
      "training_step_time": 10.29896855354309
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 2.443878412246704,
      "step": 3759
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 3759,
      "training_step_time": 7.481194734573364
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 3.1301236152648926,
      "step": 3759
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 3759,
      "training_step_time": 9.902089595794678
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 3.135887861251831,
      "step": 3759
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 3759,
      "training_step_time": 9.908772945404053
    },
    {
      "epoch": 1.4339447021484375e-05,
      "model_forward_time": 2.4295814037323,
      "step": 3759
    },
    {
      "epoch": 1.4339447021484375e-05,
      "step": 3759,
      "training_step_time": 10.256344079971313
    },
    {
      "epoch": 1.434326171875e-05,
      "grad_norm": 0.4830070436000824,
      "learning_rate": 1e-05,
      "loss": 0.0971,
      "step": 3760
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 3.4177474975585938,
      "step": 3760
    },
    {
      "dynamics_loss_avg": 0.0660085216164589,
      "epoch": 1.434326171875e-05,
      "step": 3760
    },
    {
      "action_loss_avg": 0.03457621494308114,
      "epoch": 1.434326171875e-05,
      "step": 3760
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 3760,
      "training_step_time": 10.191851377487183
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 2.4294378757476807,
      "step": 3760
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 3760,
      "training_step_time": 7.455587387084961
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 3.134279251098633,
      "step": 3760
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 3760,
      "training_step_time": 9.909866094589233
    },
    {
      "epoch": 1.434326171875e-05,
      "model_forward_time": 2.440706968307495,
      "step": 3760
    },
    {
      "epoch": 1.434326171875e-05,
      "step": 3760,
      "training_step_time": 10.014097213745117
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 3.1460134983062744,
      "step": 3761
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 3761,
      "training_step_time": 9.929652214050293
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 3.4018499851226807,
      "step": 3761
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 3761,
      "training_step_time": 10.17800259590149
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 2.4335179328918457,
      "step": 3761
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 3761,
      "training_step_time": 7.469168663024902
    },
    {
      "epoch": 1.4347076416015625e-05,
      "model_forward_time": 3.138693332672119,
      "step": 3761
    },
    {
      "epoch": 1.4347076416015625e-05,
      "step": 3761,
      "training_step_time": 10.335488319396973
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 2.4490020275115967,
      "step": 3762
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 3762,
      "training_step_time": 7.488911151885986
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 3.145904302597046,
      "step": 3762
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 3762,
      "training_step_time": 9.929486513137817
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 3.1501991748809814,
      "step": 3762
    },
    {
      "dynamics_loss_avg": 0.07302777264267206,
      "epoch": 1.435089111328125e-05,
      "step": 3762
    },
    {
      "action_loss_avg": 0.03302735798060894,
      "epoch": 1.435089111328125e-05,
      "step": 3762
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 3762,
      "training_step_time": 10.217169761657715
    },
    {
      "epoch": 1.435089111328125e-05,
      "model_forward_time": 2.443147897720337,
      "step": 3762
    },
    {
      "epoch": 1.435089111328125e-05,
      "step": 3762,
      "training_step_time": 10.043236017227173
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 3.1629271507263184,
      "step": 3763
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 3763,
      "training_step_time": 9.945760726928711
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 3.1432361602783203,
      "step": 3763
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 3763,
      "training_step_time": 9.92325758934021
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 3.1398873329162598,
      "step": 3763
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 3763,
      "training_step_time": 9.918009042739868
    },
    {
      "epoch": 1.4354705810546875e-05,
      "model_forward_time": 2.435577392578125,
      "step": 3763
    },
    {
      "epoch": 1.4354705810546875e-05,
      "step": 3763,
      "training_step_time": 10.271631717681885
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 3.4534499645233154,
      "step": 3764
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 3764,
      "training_step_time": 10.22881269454956
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 3.142963171005249,
      "step": 3764
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 3764,
      "training_step_time": 9.922857522964478
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 2.4409003257751465,
      "step": 3764
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 3764,
      "training_step_time": 7.477998971939087
    },
    {
      "epoch": 1.43585205078125e-05,
      "model_forward_time": 2.4365477561950684,
      "step": 3764
    },
    {
      "epoch": 1.43585205078125e-05,
      "step": 3764,
      "training_step_time": 10.32365107536316
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 2.4543023109436035,
      "step": 3765
    },
    {
      "dynamics_loss_avg": 0.0687461744993925,
      "epoch": 1.4362335205078125e-05,
      "step": 3765
    },
    {
      "action_loss_avg": 0.03361804718151688,
      "epoch": 1.4362335205078125e-05,
      "step": 3765
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 3765,
      "training_step_time": 7.493505477905273
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 2.007777452468872,
      "step": 3765
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 3765,
      "training_step_time": 5.471424341201782
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 2.433649778366089,
      "step": 3765
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 3765,
      "training_step_time": 7.466056823730469
    },
    {
      "epoch": 1.4362335205078125e-05,
      "model_forward_time": 3.1417508125305176,
      "step": 3765
    },
    {
      "epoch": 1.4362335205078125e-05,
      "step": 3765,
      "training_step_time": 10.264099836349487
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 3.1573565006256104,
      "step": 3766
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 3766,
      "training_step_time": 9.93511700630188
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 3.1417887210845947,
      "step": 3766
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 3766,
      "training_step_time": 9.922159433364868
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 2.693087339401245,
      "step": 3766
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 3766,
      "training_step_time": 7.726747512817383
    },
    {
      "epoch": 1.436614990234375e-05,
      "model_forward_time": 3.145969867706299,
      "step": 3766
    },
    {
      "epoch": 1.436614990234375e-05,
      "step": 3766,
      "training_step_time": 10.286118030548096
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 3.1533000469207764,
      "step": 3767
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 3767,
      "training_step_time": 9.93766975402832
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 2.438859462738037,
      "step": 3767
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 3767,
      "training_step_time": 7.474406957626343
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 1.7460331916809082,
      "step": 3767
    },
    {
      "dynamics_loss_avg": 0.057054979354143144,
      "epoch": 1.4369964599609375e-05,
      "step": 3767
    },
    {
      "action_loss_avg": 0.029402667563408612,
      "epoch": 1.4369964599609375e-05,
      "step": 3767
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 3767,
      "training_step_time": 5.211528778076172
    },
    {
      "epoch": 1.4369964599609375e-05,
      "model_forward_time": 3.131464958190918,
      "step": 3767
    },
    {
      "epoch": 1.4369964599609375e-05,
      "step": 3767,
      "training_step_time": 10.284869194030762
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 3.4498205184936523,
      "step": 3768
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 3768,
      "training_step_time": 10.22954535484314
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 2.436289072036743,
      "step": 3768
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 3768,
      "training_step_time": 7.470908880233765
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 3.134286403656006,
      "step": 3768
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 3768,
      "training_step_time": 9.913404703140259
    },
    {
      "epoch": 1.4373779296875e-05,
      "model_forward_time": 2.4352288246154785,
      "step": 3768
    },
    {
      "epoch": 1.4373779296875e-05,
      "step": 3768,
      "training_step_time": 10.02263593673706
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 3.14654541015625,
      "step": 3769
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 3769,
      "training_step_time": 9.928219318389893
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 2.6895782947540283,
      "step": 3769
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 3769,
      "training_step_time": 7.725294828414917
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 3.1423115730285645,
      "step": 3769
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 3769,
      "training_step_time": 9.925954103469849
    },
    {
      "epoch": 1.4377593994140625e-05,
      "model_forward_time": 3.149111032485962,
      "step": 3769
    },
    {
      "epoch": 1.4377593994140625e-05,
      "step": 3769,
      "training_step_time": 10.261024475097656
    },
    {
      "epoch": 1.438140869140625e-05,
      "grad_norm": 0.418091356754303,
      "learning_rate": 1e-05,
      "loss": 0.095,
      "step": 3770
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 2.4577624797821045,
      "step": 3770
    },
    {
      "dynamics_loss_avg": 0.05517611168324947,
      "epoch": 1.438140869140625e-05,
      "step": 3770
    },
    {
      "action_loss_avg": 0.029997536167502403,
      "epoch": 1.438140869140625e-05,
      "step": 3770
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 3770,
      "training_step_time": 7.502687931060791
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 2.4346766471862793,
      "step": 3770
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 3770,
      "training_step_time": 7.47060751914978
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 2.6865625381469727,
      "step": 3770
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 3770,
      "training_step_time": 7.718441486358643
    },
    {
      "epoch": 1.438140869140625e-05,
      "model_forward_time": 3.137021541595459,
      "step": 3770
    },
    {
      "epoch": 1.438140869140625e-05,
      "step": 3770,
      "training_step_time": 10.254198789596558
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 3.1471810340881348,
      "step": 3771
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 3771,
      "training_step_time": 9.927273750305176
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 3.139537811279297,
      "step": 3771
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 3771,
      "training_step_time": 9.914850950241089
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 3.137132406234741,
      "step": 3771
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 3771,
      "training_step_time": 9.909552812576294
    },
    {
      "epoch": 1.4385223388671875e-05,
      "model_forward_time": 3.404264211654663,
      "step": 3771
    },
    {
      "epoch": 1.4385223388671875e-05,
      "step": 3771,
      "training_step_time": 10.271459579467773
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 2.4512877464294434,
      "step": 3772
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 3772,
      "training_step_time": 7.484439373016357
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 3.1389472484588623,
      "step": 3772
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 3772,
      "training_step_time": 9.916798830032349
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 2.434522867202759,
      "step": 3772
    },
    {
      "dynamics_loss_avg": 0.052080638520419595,
      "epoch": 1.43890380859375e-05,
      "step": 3772
    },
    {
      "action_loss_avg": 0.029038076195865868,
      "epoch": 1.43890380859375e-05,
      "step": 3772
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 3772,
      "training_step_time": 7.472613096237183
    },
    {
      "epoch": 1.43890380859375e-05,
      "model_forward_time": 2.4357669353485107,
      "step": 3772
    },
    {
      "epoch": 1.43890380859375e-05,
      "step": 3772,
      "training_step_time": 10.277104139328003
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 2.4497523307800293,
      "step": 3773
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 3773,
      "training_step_time": 7.502748012542725
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 2.6867308616638184,
      "step": 3773
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 3773,
      "training_step_time": 7.715646505355835
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 2.4396445751190186,
      "step": 3773
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 3773,
      "training_step_time": 7.468841075897217
    },
    {
      "epoch": 1.4392852783203126e-05,
      "model_forward_time": 3.1397957801818848,
      "step": 3773
    },
    {
      "epoch": 1.4392852783203126e-05,
      "step": 3773,
      "training_step_time": 10.021230697631836
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 2.4540631771087646,
      "step": 3774
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 3774,
      "training_step_time": 7.48380184173584
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 3.1352808475494385,
      "step": 3774
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 3774,
      "training_step_time": 9.907837390899658
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 3.4023616313934326,
      "step": 3774
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 3774,
      "training_step_time": 10.17300295829773
    },
    {
      "epoch": 1.439666748046875e-05,
      "model_forward_time": 2.439688205718994,
      "step": 3774
    },
    {
      "epoch": 1.439666748046875e-05,
      "step": 3774,
      "training_step_time": 10.292259693145752
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 2.4553639888763428,
      "step": 3775
    },
    {
      "dynamics_loss_avg": 0.07768232077360153,
      "epoch": 1.4400482177734375e-05,
      "step": 3775
    },
    {
      "action_loss_avg": 0.038122791424393655,
      "epoch": 1.4400482177734375e-05,
      "step": 3775
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 3775,
      "training_step_time": 7.487753629684448
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 3.1373839378356934,
      "step": 3775
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 3775,
      "training_step_time": 9.909642219543457
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 1.7435743808746338,
      "step": 3775
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 3775,
      "training_step_time": 5.197578191757202
    },
    {
      "epoch": 1.4400482177734375e-05,
      "model_forward_time": 2.430943489074707,
      "step": 3775
    },
    {
      "epoch": 1.4400482177734375e-05,
      "step": 3775,
      "training_step_time": 10.021219968795776
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 3.4624879360198975,
      "step": 3776
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 3776,
      "training_step_time": 10.238056182861328
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 3.1391263008117676,
      "step": 3776
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 3776,
      "training_step_time": 9.913041591644287
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 2.4363107681274414,
      "step": 3776
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 3776,
      "training_step_time": 7.463388919830322
    },
    {
      "epoch": 1.4404296875e-05,
      "model_forward_time": 3.1353580951690674,
      "step": 3776
    },
    {
      "epoch": 1.4404296875e-05,
      "step": 3776,
      "training_step_time": 10.003462791442871
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 2.4467222690582275,
      "step": 3777
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 3777,
      "training_step_time": 7.469025135040283
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 3.3941807746887207,
      "step": 3777
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 3777,
      "training_step_time": 10.160246133804321
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 3.143904209136963,
      "step": 3777
    },
    {
      "dynamics_loss_avg": 0.07039832156151533,
      "epoch": 1.4408111572265626e-05,
      "step": 3777
    },
    {
      "action_loss_avg": 0.04477207576856017,
      "epoch": 1.4408111572265626e-05,
      "step": 3777
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 3777,
      "training_step_time": 9.912928581237793
    },
    {
      "epoch": 1.4408111572265626e-05,
      "model_forward_time": 3.1409153938293457,
      "step": 3777
    },
    {
      "epoch": 1.4408111572265626e-05,
      "step": 3777,
      "training_step_time": 10.339227437973022
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 3.1420576572418213,
      "step": 3778
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 3778,
      "training_step_time": 9.918831825256348
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 1.7470996379852295,
      "step": 3778
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 3778,
      "training_step_time": 5.2084057331085205
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 3.4251906871795654,
      "step": 3778
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 3778,
      "training_step_time": 10.197766542434692
    },
    {
      "epoch": 1.441192626953125e-05,
      "model_forward_time": 2.4328579902648926,
      "step": 3778
    },
    {
      "epoch": 1.441192626953125e-05,
      "step": 3778,
      "training_step_time": 10.319616079330444
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 3.150728225708008,
      "step": 3779
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 3779,
      "training_step_time": 9.936809539794922
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 1.7457985877990723,
      "step": 3779
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 3779,
      "training_step_time": 5.204907178878784
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 2.431629180908203,
      "step": 3779
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 3779,
      "training_step_time": 7.453995704650879
    },
    {
      "epoch": 1.4415740966796875e-05,
      "model_forward_time": 3.3963406085968018,
      "step": 3779
    },
    {
      "epoch": 1.4415740966796875e-05,
      "step": 3779,
      "training_step_time": 10.317078113555908
    },
    {
      "epoch": 1.44195556640625e-05,
      "grad_norm": 0.6233951449394226,
      "learning_rate": 1e-05,
      "loss": 0.0918,
      "step": 3780
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 1.7558581829071045,
      "step": 3780
    },
    {
      "dynamics_loss_avg": 0.06545385904610157,
      "epoch": 1.44195556640625e-05,
      "step": 3780
    },
    {
      "action_loss_avg": 0.041100828908383846,
      "epoch": 1.44195556640625e-05,
      "step": 3780
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 3780,
      "training_step_time": 5.2115232944488525
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 2.4297335147857666,
      "step": 3780
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 3780,
      "training_step_time": 7.448904991149902
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 3.128044605255127,
      "step": 3780
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 3780,
      "training_step_time": 9.891664266586304
    },
    {
      "epoch": 1.44195556640625e-05,
      "model_forward_time": 3.1348211765289307,
      "step": 3780
    },
    {
      "epoch": 1.44195556640625e-05,
      "step": 3780,
      "training_step_time": 10.023104906082153
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 3.1437275409698486,
      "step": 3781
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 3781,
      "training_step_time": 9.909470558166504
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 2.6908397674560547,
      "step": 3781
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 3781,
      "training_step_time": 7.711742639541626
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 3.127450466156006,
      "step": 3781
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 3781,
      "training_step_time": 9.889359951019287
    },
    {
      "epoch": 1.4423370361328126e-05,
      "model_forward_time": 3.132213592529297,
      "step": 3781
    },
    {
      "epoch": 1.4423370361328126e-05,
      "step": 3781,
      "training_step_time": 10.293390989303589
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 3.1615467071533203,
      "step": 3782
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 3782,
      "training_step_time": 9.931689500808716
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 3.1304683685302734,
      "step": 3782
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 3782,
      "training_step_time": 9.90000033378601
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 2.677324056625366,
      "step": 3782
    },
    {
      "dynamics_loss_avg": 0.0882175724953413,
      "epoch": 1.442718505859375e-05,
      "step": 3782
    },
    {
      "action_loss_avg": 0.04226719178259373,
      "epoch": 1.442718505859375e-05,
      "step": 3782
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 3782,
      "training_step_time": 7.706640243530273
    },
    {
      "epoch": 1.442718505859375e-05,
      "model_forward_time": 3.133887767791748,
      "step": 3782
    },
    {
      "epoch": 1.442718505859375e-05,
      "step": 3782,
      "training_step_time": 10.294721841812134
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 2.446209192276001,
      "step": 3783
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 3783,
      "training_step_time": 7.475562334060669
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 2.4304862022399902,
      "step": 3783
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 3783,
      "training_step_time": 7.457476377487183
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 2.4320857524871826,
      "step": 3783
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 3783,
      "training_step_time": 7.464227914810181
    },
    {
      "epoch": 1.4430999755859375e-05,
      "model_forward_time": 3.414179801940918,
      "step": 3783
    },
    {
      "epoch": 1.4430999755859375e-05,
      "step": 3783,
      "training_step_time": 10.317774534225464
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 3.155151844024658,
      "step": 3784
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 3784,
      "training_step_time": 9.929248094558716
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 3.1390883922576904,
      "step": 3784
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 3784,
      "training_step_time": 9.909834623336792
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 2.4365224838256836,
      "step": 3784
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 3784,
      "training_step_time": 7.4660797119140625
    },
    {
      "epoch": 1.4434814453125e-05,
      "model_forward_time": 2.4345006942749023,
      "step": 3784
    },
    {
      "epoch": 1.4434814453125e-05,
      "step": 3784,
      "training_step_time": 10.032079696655273
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 2.7374398708343506,
      "step": 3785
    },
    {
      "dynamics_loss_avg": 0.05267992578446865,
      "epoch": 1.4438629150390626e-05,
      "step": 3785
    },
    {
      "action_loss_avg": 0.020741022191941738,
      "epoch": 1.4438629150390626e-05,
      "step": 3785
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 3785,
      "training_step_time": 7.771239757537842
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 3.13637375831604,
      "step": 3785
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 3785,
      "training_step_time": 9.911111116409302
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 3.1407408714294434,
      "step": 3785
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 3785,
      "training_step_time": 9.914632797241211
    },
    {
      "epoch": 1.4438629150390626e-05,
      "model_forward_time": 3.170332670211792,
      "step": 3785
    },
    {
      "epoch": 1.4438629150390626e-05,
      "step": 3785,
      "training_step_time": 10.291895866394043
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 3.150956630706787,
      "step": 3786
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 3786,
      "training_step_time": 9.9281907081604
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 2.7087182998657227,
      "step": 3786
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 3786,
      "training_step_time": 7.737523555755615
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 2.4331514835357666,
      "step": 3786
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 3786,
      "training_step_time": 7.461582660675049
    },
    {
      "epoch": 1.444244384765625e-05,
      "model_forward_time": 2.434309959411621,
      "step": 3786
    },
    {
      "epoch": 1.444244384765625e-05,
      "step": 3786,
      "training_step_time": 10.007230520248413
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 3.1422736644744873,
      "step": 3787
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 3787,
      "training_step_time": 9.9146728515625
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 3.139892339706421,
      "step": 3787
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 3787,
      "training_step_time": 9.910235404968262
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 3.139430046081543,
      "step": 3787
    },
    {
      "dynamics_loss_avg": 0.05722966901957989,
      "epoch": 1.4446258544921875e-05,
      "step": 3787
    },
    {
      "action_loss_avg": 0.03690548399463296,
      "epoch": 1.4446258544921875e-05,
      "step": 3787
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 3787,
      "training_step_time": 9.91278338432312
    },
    {
      "epoch": 1.4446258544921875e-05,
      "model_forward_time": 3.396080255508423,
      "step": 3787
    },
    {
      "epoch": 1.4446258544921875e-05,
      "step": 3787,
      "training_step_time": 10.316231727600098
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 2.4499056339263916,
      "step": 3788
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 3788,
      "training_step_time": 7.483808279037476
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 2.438532590866089,
      "step": 3788
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 3788,
      "training_step_time": 7.467878580093384
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 3.139005422592163,
      "step": 3788
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 3788,
      "training_step_time": 9.911515712738037
    },
    {
      "epoch": 1.44500732421875e-05,
      "model_forward_time": 3.147273302078247,
      "step": 3788
    },
    {
      "epoch": 1.44500732421875e-05,
      "step": 3788,
      "training_step_time": 10.31251335144043
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 2.7307374477386475,
      "step": 3789
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 3789,
      "training_step_time": 7.760319232940674
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 1.7502515316009521,
      "step": 3789
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 3789,
      "training_step_time": 5.209932804107666
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 3.1328084468841553,
      "step": 3789
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 3789,
      "training_step_time": 9.89941930770874
    },
    {
      "epoch": 1.4453887939453126e-05,
      "model_forward_time": 3.1364028453826904,
      "step": 3789
    },
    {
      "epoch": 1.4453887939453126e-05,
      "step": 3789,
      "training_step_time": 10.299161911010742
    },
    {
      "epoch": 1.445770263671875e-05,
      "grad_norm": 0.42216920852661133,
      "learning_rate": 1e-05,
      "loss": 0.0974,
      "step": 3790
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 2.447286367416382,
      "step": 3790
    },
    {
      "dynamics_loss_avg": 0.07193141169846058,
      "epoch": 1.445770263671875e-05,
      "step": 3790
    },
    {
      "action_loss_avg": 0.04132820218801499,
      "epoch": 1.445770263671875e-05,
      "step": 3790
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 3790,
      "training_step_time": 7.47773814201355
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 3.4515702724456787,
      "step": 3790
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 3790,
      "training_step_time": 10.217199802398682
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 2.4385862350463867,
      "step": 3790
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 3790,
      "training_step_time": 7.4674882888793945
    },
    {
      "epoch": 1.445770263671875e-05,
      "model_forward_time": 3.139442205429077,
      "step": 3790
    },
    {
      "epoch": 1.445770263671875e-05,
      "step": 3790,
      "training_step_time": 10.289315700531006
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 3.144104480743408,
      "step": 3791
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 3791,
      "training_step_time": 9.917863130569458
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 2.4306397438049316,
      "step": 3791
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 3791,
      "training_step_time": 7.4575722217559814
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 3.1367313861846924,
      "step": 3791
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 3791,
      "training_step_time": 10.212805271148682
    },
    {
      "epoch": 1.4461517333984375e-05,
      "model_forward_time": 2.4307682514190674,
      "step": 3791
    },
    {
      "epoch": 1.4461517333984375e-05,
      "step": 3791,
      "training_step_time": 10.004617691040039
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 2.4406867027282715,
      "step": 3792
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 3792,
      "training_step_time": 7.469787836074829
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 2.4272117614746094,
      "step": 3792
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 3792,
      "training_step_time": 7.450803756713867
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 3.1300320625305176,
      "step": 3792
    },
    {
      "dynamics_loss_avg": 0.06897214502096176,
      "epoch": 1.446533203125e-05,
      "step": 3792
    },
    {
      "action_loss_avg": 0.03999327579513192,
      "epoch": 1.446533203125e-05,
      "step": 3792
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 3792,
      "training_step_time": 9.894179821014404
    },
    {
      "epoch": 1.446533203125e-05,
      "model_forward_time": 1.7373108863830566,
      "step": 3792
    },
    {
      "epoch": 1.446533203125e-05,
      "step": 3792,
      "training_step_time": 10.258339405059814
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 3.412182092666626,
      "step": 3793
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 3793,
      "training_step_time": 10.180594205856323
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 2.4284892082214355,
      "step": 3793
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 3793,
      "training_step_time": 7.452083349227905
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 3.133349895477295,
      "step": 3793
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 3793,
      "training_step_time": 9.902356386184692
    },
    {
      "epoch": 1.4469146728515626e-05,
      "model_forward_time": 3.130152463912964,
      "step": 3793
    },
    {
      "epoch": 1.4469146728515626e-05,
      "step": 3793,
      "training_step_time": 10.021359920501709
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 3.1507019996643066,
      "step": 3794
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 3794,
      "training_step_time": 9.927249431610107
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 3.3923275470733643,
      "step": 3794
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 3794,
      "training_step_time": 10.174034118652344
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 2.435482978820801,
      "step": 3794
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 3794,
      "training_step_time": 7.472218751907349
    },
    {
      "epoch": 1.447296142578125e-05,
      "model_forward_time": 3.131253242492676,
      "step": 3794
    },
    {
      "epoch": 1.447296142578125e-05,
      "step": 3794,
      "training_step_time": 10.291547775268555
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 2.447709321975708,
      "step": 3795
    },
    {
      "dynamics_loss_avg": 0.07879674378782511,
      "epoch": 1.4476776123046875e-05,
      "step": 3795
    },
    {
      "action_loss_avg": 0.033203174639493224,
      "epoch": 1.4476776123046875e-05,
      "step": 3795
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 3795,
      "training_step_time": 7.484380483627319
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 3.1361382007598877,
      "step": 3795
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 3795,
      "training_step_time": 9.917801856994629
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 2.713338613510132,
      "step": 3795
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 3795,
      "training_step_time": 7.747614860534668
    },
    {
      "epoch": 1.4476776123046875e-05,
      "model_forward_time": 2.4347455501556396,
      "step": 3795
    },
    {
      "epoch": 1.4476776123046875e-05,
      "step": 3795,
      "training_step_time": 10.341128826141357
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 3.1594324111938477,
      "step": 3796
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 3796,
      "training_step_time": 9.944303035736084
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 3.1443898677825928,
      "step": 3796
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 3796,
      "training_step_time": 9.923205614089966
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 3.1404285430908203,
      "step": 3796
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 3796,
      "training_step_time": 9.915437459945679
    },
    {
      "epoch": 1.44805908203125e-05,
      "model_forward_time": 3.4432754516601562,
      "step": 3796
    },
    {
      "epoch": 1.44805908203125e-05,
      "step": 3796,
      "training_step_time": 10.313478946685791
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 1.7655131816864014,
      "step": 3797
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 3797,
      "training_step_time": 5.230004072189331
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 3.1378371715545654,
      "step": 3797
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 3797,
      "training_step_time": 9.906835794448853
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 2.434760332107544,
      "step": 3797
    },
    {
      "dynamics_loss_avg": 0.06670006457716227,
      "epoch": 1.4484405517578126e-05,
      "step": 3797
    },
    {
      "action_loss_avg": 0.03345879344269633,
      "epoch": 1.4484405517578126e-05,
      "step": 3797
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 3797,
      "training_step_time": 7.4668426513671875
    },
    {
      "epoch": 1.4484405517578126e-05,
      "model_forward_time": 3.138406753540039,
      "step": 3797
    },
    {
      "epoch": 1.4484405517578126e-05,
      "step": 3797,
      "training_step_time": 10.01841688156128
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 2.4535388946533203,
      "step": 3798
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 3798,
      "training_step_time": 7.4862449169158936
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 2.713167905807495,
      "step": 3798
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 3798,
      "training_step_time": 7.746727705001831
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 3.1657702922821045,
      "step": 3798
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 3798,
      "training_step_time": 9.938974380493164
    },
    {
      "epoch": 1.448822021484375e-05,
      "model_forward_time": 2.454516887664795,
      "step": 3798
    },
    {
      "epoch": 1.448822021484375e-05,
      "step": 3798,
      "training_step_time": 10.02795934677124
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 3.169264554977417,
      "step": 3799
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 3799,
      "training_step_time": 9.94762110710144
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 3.1423301696777344,
      "step": 3799
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 3799,
      "training_step_time": 9.918674945831299
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 3.396848440170288,
      "step": 3799
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 3799,
      "training_step_time": 10.164920806884766
    },
    {
      "epoch": 1.4492034912109375e-05,
      "model_forward_time": 2.436404228210449,
      "step": 3799
    },
    {
      "epoch": 1.4492034912109375e-05,
      "step": 3799,
      "training_step_time": 10.29741883277893
    },
    {
      "epoch": 1.4495849609375e-05,
      "grad_norm": 0.520972728729248,
      "learning_rate": 1e-05,
      "loss": 0.1022,
      "step": 3800
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 3.1834285259246826,
      "step": 3800
    },
    {
      "dynamics_loss_avg": 0.05768901761621237,
      "epoch": 1.4495849609375e-05,
      "step": 3800
    },
    {
      "action_loss_avg": 0.02797459401190281,
      "epoch": 1.4495849609375e-05,
      "step": 3800
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 3800,
      "training_step_time": 9.963407754898071
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 1.7485356330871582,
      "step": 3800
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 3800,
      "training_step_time": 5.206500053405762
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 3.143141508102417,
      "step": 3800
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 3800,
      "training_step_time": 9.915889024734497
    },
    {
      "epoch": 1.4495849609375e-05,
      "model_forward_time": 3.1436617374420166,
      "step": 3800
    },
    {
      "epoch": 1.4495849609375e-05,
      "step": 3800,
      "training_step_time": 10.286787748336792
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 3.516784191131592,
      "step": 3801
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 3801,
      "training_step_time": 10.288901567459106
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 2.4360969066619873,
      "step": 3801
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 3801,
      "training_step_time": 7.4617979526519775
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 3.1440296173095703,
      "step": 3801
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 3801,
      "training_step_time": 9.917446613311768
    },
    {
      "epoch": 1.4499664306640626e-05,
      "model_forward_time": 3.1416141986846924,
      "step": 3801
    },
    {
      "epoch": 1.4499664306640626e-05,
      "step": 3801,
      "training_step_time": 10.271410703659058
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 3.1478700637817383,
      "step": 3802
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 3802,
      "training_step_time": 10.210470914840698
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 3.136476516723633,
      "step": 3802
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 3802,
      "training_step_time": 9.904922008514404
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 3.1431121826171875,
      "step": 3802
    },
    {
      "dynamics_loss_avg": 0.06260905116796493,
      "epoch": 1.450347900390625e-05,
      "step": 3802
    },
    {
      "action_loss_avg": 0.029933376889675855,
      "epoch": 1.450347900390625e-05,
      "step": 3802
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 3802,
      "training_step_time": 9.916635274887085
    },
    {
      "epoch": 1.450347900390625e-05,
      "model_forward_time": 3.144052743911743,
      "step": 3802
    },
    {
      "epoch": 1.450347900390625e-05,
      "step": 3802,
      "training_step_time": 10.019008159637451
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 1.7685492038726807,
      "step": 3803
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 3803,
      "training_step_time": 5.225925445556641
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 1.7461702823638916,
      "step": 3803
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 3803,
      "training_step_time": 5.203339099884033
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 3.4051640033721924,
      "step": 3803
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 3803,
      "training_step_time": 10.169543504714966
    },
    {
      "epoch": 1.4507293701171875e-05,
      "model_forward_time": 2.432708501815796,
      "step": 3803
    },
    {
      "epoch": 1.4507293701171875e-05,
      "step": 3803,
      "training_step_time": 10.023103952407837
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 2.462920665740967,
      "step": 3804
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 3804,
      "training_step_time": 7.487746000289917
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 3.1337249279022217,
      "step": 3804
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 3804,
      "training_step_time": 9.904430866241455
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 3.132686138153076,
      "step": 3804
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 3804,
      "training_step_time": 9.90736985206604
    },
    {
      "epoch": 1.45111083984375e-05,
      "model_forward_time": 2.4385647773742676,
      "step": 3804
    },
    {
      "epoch": 1.45111083984375e-05,
      "step": 3804,
      "training_step_time": 10.264484405517578
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 2.7545969486236572,
      "step": 3805
    },
    {
      "dynamics_loss_avg": 0.05513296201825142,
      "epoch": 1.4514923095703126e-05,
      "step": 3805
    },
    {
      "action_loss_avg": 0.02379880892112851,
      "epoch": 1.4514923095703126e-05,
      "step": 3805
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 3805,
      "training_step_time": 7.783462762832642
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 2.4330251216888428,
      "step": 3805
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 3805,
      "training_step_time": 7.462395429611206
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 3.132704973220825,
      "step": 3805
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 3805,
      "training_step_time": 9.90285587310791
    },
    {
      "epoch": 1.4514923095703126e-05,
      "model_forward_time": 3.1355490684509277,
      "step": 3805
    },
    {
      "epoch": 1.4514923095703126e-05,
      "step": 3805,
      "training_step_time": 10.283971548080444
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 2.4473965167999268,
      "step": 3806
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 3806,
      "training_step_time": 7.477816343307495
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 3.415689706802368,
      "step": 3806
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 3806,
      "training_step_time": 10.195943355560303
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 3.1438419818878174,
      "step": 3806
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 3806,
      "training_step_time": 9.916037321090698
    },
    {
      "epoch": 1.451873779296875e-05,
      "model_forward_time": 2.433455228805542,
      "step": 3806
    },
    {
      "epoch": 1.451873779296875e-05,
      "step": 3806,
      "training_step_time": 10.347581624984741
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 3.158277750015259,
      "step": 3807
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 3807,
      "training_step_time": 9.932576894760132
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 2.437922716140747,
      "step": 3807
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 3807,
      "training_step_time": 7.463802814483643
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 3.4389259815216064,
      "step": 3807
    },
    {
      "dynamics_loss_avg": 0.06067352294921875,
      "epoch": 1.4522552490234375e-05,
      "step": 3807
    },
    {
      "action_loss_avg": 0.03318784944713116,
      "epoch": 1.4522552490234375e-05,
      "step": 3807
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 3807,
      "training_step_time": 10.211459875106812
    },
    {
      "epoch": 1.4522552490234375e-05,
      "model_forward_time": 3.140735149383545,
      "step": 3807
    },
    {
      "epoch": 1.4522552490234375e-05,
      "step": 3807,
      "training_step_time": 10.311589479446411
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 3.150876522064209,
      "step": 3808
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 3808,
      "training_step_time": 9.91913390159607
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 3.142005443572998,
      "step": 3808
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 3808,
      "training_step_time": 9.910164833068848
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 3.143322467803955,
      "step": 3808
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 3808,
      "training_step_time": 9.910978078842163
    },
    {
      "epoch": 1.45263671875e-05,
      "model_forward_time": 2.71016263961792,
      "step": 3808
    },
    {
      "epoch": 1.45263671875e-05,
      "step": 3808,
      "training_step_time": 10.297600030899048
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 2.4413254261016846,
      "step": 3809
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 3809,
      "training_step_time": 7.473856449127197
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 3.132248878479004,
      "step": 3809
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 3809,
      "training_step_time": 9.90001630783081
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 3.1344289779663086,
      "step": 3809
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 3809,
      "training_step_time": 9.900722026824951
    },
    {
      "epoch": 1.4530181884765626e-05,
      "model_forward_time": 3.1365087032318115,
      "step": 3809
    },
    {
      "epoch": 1.4530181884765626e-05,
      "step": 3809,
      "training_step_time": 10.017549276351929
    },
    {
      "epoch": 1.453399658203125e-05,
      "grad_norm": 0.26227104663848877,
      "learning_rate": 1e-05,
      "loss": 0.0988,
      "step": 3810
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 3.4602773189544678,
      "step": 3810
    },
    {
      "dynamics_loss_avg": 0.06062069199979305,
      "epoch": 1.453399658203125e-05,
      "step": 3810
    },
    {
      "action_loss_avg": 0.026469599222764373,
      "epoch": 1.453399658203125e-05,
      "step": 3810
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 3810,
      "training_step_time": 10.235748529434204
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 3.141190528869629,
      "step": 3810
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 3810,
      "training_step_time": 9.912422180175781
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 3.1419050693511963,
      "step": 3810
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 3810,
      "training_step_time": 9.909883737564087
    },
    {
      "epoch": 1.453399658203125e-05,
      "model_forward_time": 2.438584804534912,
      "step": 3810
    },
    {
      "epoch": 1.453399658203125e-05,
      "step": 3810,
      "training_step_time": 10.304877519607544
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 2.444941520690918,
      "step": 3811
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 3811,
      "training_step_time": 7.481134414672852
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 1.7443146705627441,
      "step": 3811
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 3811,
      "training_step_time": 5.201618194580078
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 3.4178738594055176,
      "step": 3811
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 3811,
      "training_step_time": 10.186235666275024
    },
    {
      "epoch": 1.4537811279296875e-05,
      "model_forward_time": 2.4359021186828613,
      "step": 3811
    },
    {
      "epoch": 1.4537811279296875e-05,
      "step": 3811,
      "training_step_time": 10.303149938583374
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 3.152498245239258,
      "step": 3812
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 3812,
      "training_step_time": 9.931664228439331
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 2.4363889694213867,
      "step": 3812
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 3812,
      "training_step_time": 7.469984531402588
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 1.7466130256652832,
      "step": 3812
    },
    {
      "dynamics_loss_avg": 0.07250886671245098,
      "epoch": 1.45416259765625e-05,
      "step": 3812
    },
    {
      "action_loss_avg": 0.03200776670128107,
      "epoch": 1.45416259765625e-05,
      "step": 3812
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 3812,
      "training_step_time": 5.207433223724365
    },
    {
      "epoch": 1.45416259765625e-05,
      "model_forward_time": 2.719167470932007,
      "step": 3812
    },
    {
      "epoch": 1.45416259765625e-05,
      "step": 3812,
      "training_step_time": 10.316955804824829
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 3.148317813873291,
      "step": 3813
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 3813,
      "training_step_time": 9.928450345993042
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 2.4404296875,
      "step": 3813
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 3813,
      "training_step_time": 7.4711244106292725
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 3.1395370960235596,
      "step": 3813
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 3813,
      "training_step_time": 9.907702684402466
    },
    {
      "epoch": 1.4545440673828126e-05,
      "model_forward_time": 3.142691135406494,
      "step": 3813
    },
    {
      "epoch": 1.4545440673828126e-05,
      "step": 3813,
      "training_step_time": 10.310602903366089
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 2.7640018463134766,
      "step": 3814
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 3814,
      "training_step_time": 7.791491270065308
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 2.4402639865875244,
      "step": 3814
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 3814,
      "training_step_time": 7.465535402297974
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 3.142334461212158,
      "step": 3814
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 3814,
      "training_step_time": 9.915826797485352
    },
    {
      "epoch": 1.454925537109375e-05,
      "model_forward_time": 2.4372828006744385,
      "step": 3814
    },
    {
      "epoch": 1.454925537109375e-05,
      "step": 3814,
      "training_step_time": 10.043336391448975
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 3.180309534072876,
      "step": 3815
    },
    {
      "dynamics_loss_avg": 0.056325079407542944,
      "epoch": 1.4553070068359375e-05,
      "step": 3815
    },
    {
      "action_loss_avg": 0.024313684250228108,
      "epoch": 1.4553070068359375e-05,
      "step": 3815
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 3815,
      "training_step_time": 9.958502531051636
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 2.442667245864868,
      "step": 3815
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 3815,
      "training_step_time": 7.476263761520386
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 2.0589771270751953,
      "step": 3815
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 3815,
      "training_step_time": 5.514861345291138
    },
    {
      "epoch": 1.4553070068359375e-05,
      "model_forward_time": 2.4354209899902344,
      "step": 3815
    },
    {
      "epoch": 1.4553070068359375e-05,
      "step": 3815,
      "training_step_time": 10.307092189788818
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 1.7711906433105469,
      "step": 3816
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 3816,
      "training_step_time": 5.230849027633667
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 3.1433491706848145,
      "step": 3816
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 3816,
      "training_step_time": 9.909815311431885
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 2.43245530128479,
      "step": 3816
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 3816,
      "training_step_time": 7.453590631484985
    },
    {
      "epoch": 1.4556884765625e-05,
      "model_forward_time": 3.452755928039551,
      "step": 3816
    },
    {
      "epoch": 1.4556884765625e-05,
      "step": 3816,
      "training_step_time": 10.307531595230103
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 2.4445109367370605,
      "step": 3817
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 3817,
      "training_step_time": 7.467622518539429
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 2.430713415145874,
      "step": 3817
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 3817,
      "training_step_time": 7.452467679977417
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 2.4312121868133545,
      "step": 3817
    },
    {
      "dynamics_loss_avg": 0.05566630437970162,
      "epoch": 1.4560699462890626e-05,
      "step": 3817
    },
    {
      "action_loss_avg": 0.03382491385564208,
      "epoch": 1.4560699462890626e-05,
      "step": 3817
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 3817,
      "training_step_time": 7.451039791107178
    },
    {
      "epoch": 1.4560699462890626e-05,
      "model_forward_time": 2.4278085231781006,
      "step": 3817
    },
    {
      "epoch": 1.4560699462890626e-05,
      "step": 3817,
      "training_step_time": 10.269882440567017
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 2.433323383331299,
      "step": 3818
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 3818,
      "training_step_time": 7.452932357788086
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 2.730393886566162,
      "step": 3818
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 3818,
      "training_step_time": 7.754320383071899
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 2.4289979934692383,
      "step": 3818
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 3818,
      "training_step_time": 7.452787160873413
    },
    {
      "epoch": 1.456451416015625e-05,
      "model_forward_time": 3.1367526054382324,
      "step": 3818
    },
    {
      "epoch": 1.456451416015625e-05,
      "step": 3818,
      "training_step_time": 10.035905838012695
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 3.152345657348633,
      "step": 3819
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 3819,
      "training_step_time": 9.923815727233887
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 2.4335169792175293,
      "step": 3819
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 3819,
      "training_step_time": 7.469176292419434
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 3.4318385124206543,
      "step": 3819
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 3819,
      "training_step_time": 10.208887100219727
    },
    {
      "epoch": 1.4568328857421875e-05,
      "model_forward_time": 3.146656036376953,
      "step": 3819
    },
    {
      "epoch": 1.4568328857421875e-05,
      "step": 3819,
      "training_step_time": 10.036003589630127
    },
    {
      "epoch": 1.45721435546875e-05,
      "grad_norm": 0.43352848291397095,
      "learning_rate": 1e-05,
      "loss": 0.094,
      "step": 3820
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 3.1936874389648438,
      "step": 3820
    },
    {
      "dynamics_loss_avg": 0.05876940414309502,
      "epoch": 1.45721435546875e-05,
      "step": 3820
    },
    {
      "action_loss_avg": 0.031115592643618583,
      "epoch": 1.45721435546875e-05,
      "step": 3820
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 3820,
      "training_step_time": 9.99589729309082
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 2.443610191345215,
      "step": 3820
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 3820,
      "training_step_time": 7.477249622344971
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 2.4458556175231934,
      "step": 3820
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 3820,
      "training_step_time": 7.484596014022827
    },
    {
      "epoch": 1.45721435546875e-05,
      "model_forward_time": 1.751286506652832,
      "step": 3820
    },
    {
      "epoch": 1.45721435546875e-05,
      "step": 3820,
      "training_step_time": 10.34887146949768
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 2.757016658782959,
      "step": 3821
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 3821,
      "training_step_time": 7.786697149276733
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 2.4360430240631104,
      "step": 3821
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 3821,
      "training_step_time": 7.45855975151062
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 2.4411323070526123,
      "step": 3821
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 3821,
      "training_step_time": 7.469209432601929
    },
    {
      "epoch": 1.4575958251953126e-05,
      "model_forward_time": 3.1490023136138916,
      "step": 3821
    },
    {
      "epoch": 1.4575958251953126e-05,
      "step": 3821,
      "training_step_time": 10.281541585922241
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 1.7613697052001953,
      "step": 3822
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 3822,
      "training_step_time": 5.214705467224121
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 2.4364547729492188,
      "step": 3822
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 3822,
      "training_step_time": 7.458527088165283
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 3.4389545917510986,
      "step": 3822
    },
    {
      "dynamics_loss_avg": 0.05625230725854635,
      "epoch": 1.457977294921875e-05,
      "step": 3822
    },
    {
      "action_loss_avg": 0.031087892688810827,
      "epoch": 1.457977294921875e-05,
      "step": 3822
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 3822,
      "training_step_time": 10.212303638458252
    },
    {
      "epoch": 1.457977294921875e-05,
      "model_forward_time": 3.1429190635681152,
      "step": 3822
    },
    {
      "epoch": 1.457977294921875e-05,
      "step": 3822,
      "training_step_time": 10.278659582138062
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 2.463202714920044,
      "step": 3823
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 3823,
      "training_step_time": 7.487803936004639
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 2.4359092712402344,
      "step": 3823
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 3823,
      "training_step_time": 7.458199977874756
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 3.146953821182251,
      "step": 3823
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 3823,
      "training_step_time": 9.917073011398315
    },
    {
      "epoch": 1.4583587646484375e-05,
      "model_forward_time": 3.468060255050659,
      "step": 3823
    },
    {
      "epoch": 1.4583587646484375e-05,
      "step": 3823,
      "training_step_time": 10.332493782043457
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 2.455519199371338,
      "step": 3824
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 3824,
      "training_step_time": 7.485637664794922
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 1.7463648319244385,
      "step": 3824
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 3824,
      "training_step_time": 5.21421480178833
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 2.438169002532959,
      "step": 3824
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 3824,
      "training_step_time": 7.466731548309326
    },
    {
      "epoch": 1.458740234375e-05,
      "model_forward_time": 2.4335131645202637,
      "step": 3824
    },
    {
      "epoch": 1.458740234375e-05,
      "step": 3824,
      "training_step_time": 9.983345985412598
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 2.4448089599609375,
      "step": 3825
    },
    {
      "dynamics_loss_avg": 0.06463763657957315,
      "epoch": 1.4591217041015626e-05,
      "step": 3825
    },
    {
      "action_loss_avg": 0.03979639117605984,
      "epoch": 1.4591217041015626e-05,
      "step": 3825
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 3825,
      "training_step_time": 7.469134330749512
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 3.433103084564209,
      "step": 3825
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 3825,
      "training_step_time": 10.195605993270874
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 3.1420717239379883,
      "step": 3825
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 3825,
      "training_step_time": 9.908875226974487
    },
    {
      "epoch": 1.4591217041015626e-05,
      "model_forward_time": 3.1497743129730225,
      "step": 3825
    },
    {
      "epoch": 1.4591217041015626e-05,
      "step": 3825,
      "training_step_time": 10.308992147445679
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 2.457796335220337,
      "step": 3826
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 3826,
      "training_step_time": 7.483996868133545
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 2.4350759983062744,
      "step": 3826
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 3826,
      "training_step_time": 7.463639974594116
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 3.432237148284912,
      "step": 3826
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 3826,
      "training_step_time": 10.203478813171387
    },
    {
      "epoch": 1.459503173828125e-05,
      "model_forward_time": 1.7480924129486084,
      "step": 3826
    },
    {
      "epoch": 1.459503173828125e-05,
      "step": 3826,
      "training_step_time": 10.024084329605103
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 2.4512736797332764,
      "step": 3827
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 3827,
      "training_step_time": 7.481619834899902
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 3.139362096786499,
      "step": 3827
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 3827,
      "training_step_time": 9.909560918807983
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 3.1371943950653076,
      "step": 3827
    },
    {
      "dynamics_loss_avg": 0.07716438174247742,
      "epoch": 1.4598846435546875e-05,
      "step": 3827
    },
    {
      "action_loss_avg": 0.03526561753824353,
      "epoch": 1.4598846435546875e-05,
      "step": 3827
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 3827,
      "training_step_time": 9.910896301269531
    },
    {
      "epoch": 1.4598846435546875e-05,
      "model_forward_time": 2.435541868209839,
      "step": 3827
    },
    {
      "epoch": 1.4598846435546875e-05,
      "step": 3827,
      "training_step_time": 10.036671161651611
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 3.4103331565856934,
      "step": 3828
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 3828,
      "training_step_time": 10.17929458618164
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 3.143590211868286,
      "step": 3828
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 3828,
      "training_step_time": 9.905744314193726
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 3.1366019248962402,
      "step": 3828
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 3828,
      "training_step_time": 9.90523076057434
    },
    {
      "epoch": 1.46026611328125e-05,
      "model_forward_time": 3.142062187194824,
      "step": 3828
    },
    {
      "epoch": 1.46026611328125e-05,
      "step": 3828,
      "training_step_time": 10.361764907836914
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 2.4461472034454346,
      "step": 3829
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 3829,
      "training_step_time": 7.471031188964844
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 3.4620325565338135,
      "step": 3829
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 3829,
      "training_step_time": 10.234209299087524
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 3.1402385234832764,
      "step": 3829
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 3829,
      "training_step_time": 9.905208110809326
    },
    {
      "epoch": 1.4606475830078126e-05,
      "model_forward_time": 2.434515953063965,
      "step": 3829
    },
    {
      "epoch": 1.4606475830078126e-05,
      "step": 3829,
      "training_step_time": 10.289721488952637
    },
    {
      "epoch": 1.461029052734375e-05,
      "grad_norm": 0.34505894780158997,
      "learning_rate": 1e-05,
      "loss": 0.0962,
      "step": 3830
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 1.766767978668213,
      "step": 3830
    },
    {
      "dynamics_loss_avg": 0.06346758157014847,
      "epoch": 1.461029052734375e-05,
      "step": 3830
    },
    {
      "action_loss_avg": 0.032828941382467745,
      "epoch": 1.461029052734375e-05,
      "step": 3830
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 3830,
      "training_step_time": 5.226876497268677
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 1.746246337890625,
      "step": 3830
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 3830,
      "training_step_time": 5.198290109634399
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 3.1376073360443115,
      "step": 3830
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 3830,
      "training_step_time": 10.165597915649414
    },
    {
      "epoch": 1.461029052734375e-05,
      "model_forward_time": 3.1431992053985596,
      "step": 3830
    },
    {
      "epoch": 1.461029052734375e-05,
      "step": 3830,
      "training_step_time": 10.31375789642334
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 3.1687562465667725,
      "step": 3831
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 3831,
      "training_step_time": 9.935061931610107
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 2.442462682723999,
      "step": 3831
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 3831,
      "training_step_time": 7.463023662567139
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 3.1405222415924072,
      "step": 3831
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 3831,
      "training_step_time": 9.90238881111145
    },
    {
      "epoch": 1.4614105224609375e-05,
      "model_forward_time": 2.434511661529541,
      "step": 3831
    },
    {
      "epoch": 1.4614105224609375e-05,
      "step": 3831,
      "training_step_time": 10.02487325668335
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 2.765620231628418,
      "step": 3832
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 3832,
      "training_step_time": 7.785242795944214
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 1.737971544265747,
      "step": 3832
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 3832,
      "training_step_time": 5.186944961547852
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 1.7411820888519287,
      "step": 3832
    },
    {
      "dynamics_loss_avg": 0.05897639058530331,
      "epoch": 1.4617919921875e-05,
      "step": 3832
    },
    {
      "action_loss_avg": 0.03848600720521063,
      "epoch": 1.4617919921875e-05,
      "step": 3832
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 3832,
      "training_step_time": 5.1933441162109375
    },
    {
      "epoch": 1.4617919921875e-05,
      "model_forward_time": 3.1324453353881836,
      "step": 3832
    },
    {
      "epoch": 1.4617919921875e-05,
      "step": 3832,
      "training_step_time": 10.02900505065918
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 3.143306016921997,
      "step": 3833
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 3833,
      "training_step_time": 9.912697792053223
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 2.432753324508667,
      "step": 3833
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 3833,
      "training_step_time": 7.46279501914978
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 2.7426559925079346,
      "step": 3833
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 3833,
      "training_step_time": 7.7702860832214355
    },
    {
      "epoch": 1.4621734619140626e-05,
      "model_forward_time": 2.435105800628662,
      "step": 3833
    },
    {
      "epoch": 1.4621734619140626e-05,
      "step": 3833,
      "training_step_time": 10.33445692062378
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 2.4503960609436035,
      "step": 3834
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 3834,
      "training_step_time": 7.48097825050354
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 3.139192819595337,
      "step": 3834
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 3834,
      "training_step_time": 9.905943632125854
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 2.438166618347168,
      "step": 3834
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 3834,
      "training_step_time": 7.466815948486328
    },
    {
      "epoch": 1.462554931640625e-05,
      "model_forward_time": 3.395251512527466,
      "step": 3834
    },
    {
      "epoch": 1.462554931640625e-05,
      "step": 3834,
      "training_step_time": 10.274580717086792
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 1.7683684825897217,
      "step": 3835
    },
    {
      "dynamics_loss_avg": 0.05258727669715881,
      "epoch": 1.4629364013671875e-05,
      "step": 3835
    },
    {
      "action_loss_avg": 0.028384434711188077,
      "epoch": 1.4629364013671875e-05,
      "step": 3835
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 3835,
      "training_step_time": 5.22610068321228
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 2.4365394115448,
      "step": 3835
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 3835,
      "training_step_time": 7.4629199504852295
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 2.4374959468841553,
      "step": 3835
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 3835,
      "training_step_time": 7.466024398803711
    },
    {
      "epoch": 1.4629364013671875e-05,
      "model_forward_time": 3.1434123516082764,
      "step": 3835
    },
    {
      "epoch": 1.4629364013671875e-05,
      "step": 3835,
      "training_step_time": 10.308712244033813
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 3.1528522968292236,
      "step": 3836
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 3836,
      "training_step_time": 9.922104597091675
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 2.7158384323120117,
      "step": 3836
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 3836,
      "training_step_time": 7.742640256881714
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 2.433030843734741,
      "step": 3836
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 3836,
      "training_step_time": 7.457597494125366
    },
    {
      "epoch": 1.46331787109375e-05,
      "model_forward_time": 3.13883638381958,
      "step": 3836
    },
    {
      "epoch": 1.46331787109375e-05,
      "step": 3836,
      "training_step_time": 10.032748699188232
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 3.156111717224121,
      "step": 3837
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 3837,
      "training_step_time": 9.932406663894653
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 3.1383163928985596,
      "step": 3837
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 3837,
      "training_step_time": 9.906535863876343
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 2.7081358432769775,
      "step": 3837
    },
    {
      "dynamics_loss_avg": 0.06567954663187266,
      "epoch": 1.4636993408203126e-05,
      "step": 3837
    },
    {
      "action_loss_avg": 0.051643379777669904,
      "epoch": 1.4636993408203126e-05,
      "step": 3837
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 3837,
      "training_step_time": 7.741367816925049
    },
    {
      "epoch": 1.4636993408203126e-05,
      "model_forward_time": 3.1411936283111572,
      "step": 3837
    },
    {
      "epoch": 1.4636993408203126e-05,
      "step": 3837,
      "training_step_time": 10.034162998199463
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 3.1590776443481445,
      "step": 3838
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 3838,
      "training_step_time": 9.934306859970093
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 3.143045663833618,
      "step": 3838
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 3838,
      "training_step_time": 9.920565128326416
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 2.440535545349121,
      "step": 3838
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 3838,
      "training_step_time": 7.469846248626709
    },
    {
      "epoch": 1.464080810546875e-05,
      "model_forward_time": 3.4229111671447754,
      "step": 3838
    },
    {
      "epoch": 1.464080810546875e-05,
      "step": 3838,
      "training_step_time": 10.291727781295776
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 2.447787046432495,
      "step": 3839
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 3839,
      "training_step_time": 7.4765355587005615
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 2.4353179931640625,
      "step": 3839
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 3839,
      "training_step_time": 7.4656662940979
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 3.141101837158203,
      "step": 3839
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 3839,
      "training_step_time": 9.913215398788452
    },
    {
      "epoch": 1.4644622802734375e-05,
      "model_forward_time": 3.1418659687042236,
      "step": 3839
    },
    {
      "epoch": 1.4644622802734375e-05,
      "step": 3839,
      "training_step_time": 10.336368560791016
    },
    {
      "epoch": 1.46484375e-05,
      "grad_norm": 0.4840107560157776,
      "learning_rate": 1e-05,
      "loss": 0.0978,
      "step": 3840
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 3.429537534713745,
      "step": 3840
    },
    {
      "dynamics_loss_avg": 0.0630641184747219,
      "epoch": 1.46484375e-05,
      "step": 3840
    },
    {
      "action_loss_avg": 0.03494771304540336,
      "epoch": 1.46484375e-05,
      "step": 3840
    },
    {
      "epoch": 1.46484375e-05,
      "step": 3840,
      "training_step_time": 10.207106113433838
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 3.1390738487243652,
      "step": 3840
    },
    {
      "epoch": 1.46484375e-05,
      "step": 3840,
      "training_step_time": 9.91592788696289
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 2.4416451454162598,
      "step": 3840
    },
    {
      "epoch": 1.46484375e-05,
      "step": 3840,
      "training_step_time": 7.477707862854004
    },
    {
      "epoch": 1.46484375e-05,
      "model_forward_time": 3.140679359436035,
      "step": 3840
    },
    {
      "epoch": 1.46484375e-05,
      "step": 3840,
      "training_step_time": 10.308030843734741
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 2.4554929733276367,
      "step": 3841
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 3841,
      "training_step_time": 7.48412013053894
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 3.411975622177124,
      "step": 3841
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 3841,
      "training_step_time": 10.182331800460815
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 3.1438698768615723,
      "step": 3841
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 3841,
      "training_step_time": 9.914128065109253
    },
    {
      "epoch": 1.4652252197265626e-05,
      "model_forward_time": 2.4373371601104736,
      "step": 3841
    },
    {
      "epoch": 1.4652252197265626e-05,
      "step": 3841,
      "training_step_time": 10.264488458633423
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 3.147287607192993,
      "step": 3842
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 3842,
      "training_step_time": 9.918169260025024
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 3.1390554904937744,
      "step": 3842
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 3842,
      "training_step_time": 9.906611204147339
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 2.4312567710876465,
      "step": 3842
    },
    {
      "dynamics_loss_avg": 0.06539276633411646,
      "epoch": 1.465606689453125e-05,
      "step": 3842
    },
    {
      "action_loss_avg": 0.031854874454438686,
      "epoch": 1.465606689453125e-05,
      "step": 3842
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 3842,
      "training_step_time": 7.455052137374878
    },
    {
      "epoch": 1.465606689453125e-05,
      "model_forward_time": 2.7335193157196045,
      "step": 3842
    },
    {
      "epoch": 1.465606689453125e-05,
      "step": 3842,
      "training_step_time": 10.019492626190186
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 3.154684543609619,
      "step": 3843
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 3843,
      "training_step_time": 9.919643640518188
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 3.1377220153808594,
      "step": 3843
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 3843,
      "training_step_time": 9.907280683517456
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 2.4372973442077637,
      "step": 3843
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 3843,
      "training_step_time": 7.459136486053467
    },
    {
      "epoch": 1.4659881591796875e-05,
      "model_forward_time": 3.1349093914031982,
      "step": 3843
    },
    {
      "epoch": 1.4659881591796875e-05,
      "step": 3843,
      "training_step_time": 10.026126146316528
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 3.4268317222595215,
      "step": 3844
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 3844,
      "training_step_time": 10.209546566009521
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 2.4401490688323975,
      "step": 3844
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 3844,
      "training_step_time": 7.4714226722717285
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 2.4358553886413574,
      "step": 3844
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 3844,
      "training_step_time": 7.46797513961792
    },
    {
      "epoch": 1.46636962890625e-05,
      "model_forward_time": 2.434934616088867,
      "step": 3844
    },
    {
      "epoch": 1.46636962890625e-05,
      "step": 3844,
      "training_step_time": 10.294602155685425
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 3.151195526123047,
      "step": 3845
    },
    {
      "dynamics_loss_avg": 0.059253144264221194,
      "epoch": 1.4667510986328126e-05,
      "step": 3845
    },
    {
      "action_loss_avg": 0.029298192542046308,
      "epoch": 1.4667510986328126e-05,
      "step": 3845
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 3845,
      "training_step_time": 9.929569482803345
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 2.720407247543335,
      "step": 3845
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 3845,
      "training_step_time": 7.74912166595459
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 3.1371963024139404,
      "step": 3845
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 3845,
      "training_step_time": 9.912614345550537
    },
    {
      "epoch": 1.4667510986328126e-05,
      "model_forward_time": 3.1459720134735107,
      "step": 3845
    },
    {
      "epoch": 1.4667510986328126e-05,
      "step": 3845,
      "training_step_time": 10.259637594223022
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 2.446504592895508,
      "step": 3846
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 3846,
      "training_step_time": 7.480144739151001
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 2.4391493797302246,
      "step": 3846
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 3846,
      "training_step_time": 7.468809366226196
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 3.407090902328491,
      "step": 3846
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 3846,
      "training_step_time": 10.182318925857544
    },
    {
      "epoch": 1.467132568359375e-05,
      "model_forward_time": 3.1467623710632324,
      "step": 3846
    },
    {
      "epoch": 1.467132568359375e-05,
      "step": 3846,
      "training_step_time": 10.031065702438354
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 2.4516654014587402,
      "step": 3847
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 3847,
      "training_step_time": 7.4825358390808105
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 2.435307502746582,
      "step": 3847
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 3847,
      "training_step_time": 7.463808536529541
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 3.137799024581909,
      "step": 3847
    },
    {
      "dynamics_loss_avg": 0.05867801904678345,
      "epoch": 1.4675140380859375e-05,
      "step": 3847
    },
    {
      "action_loss_avg": 0.029757332988083363,
      "epoch": 1.4675140380859375e-05,
      "step": 3847
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 3847,
      "training_step_time": 9.909714460372925
    },
    {
      "epoch": 1.4675140380859375e-05,
      "model_forward_time": 3.140542984008789,
      "step": 3847
    },
    {
      "epoch": 1.4675140380859375e-05,
      "step": 3847,
      "training_step_time": 10.287802457809448
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 2.757551670074463,
      "step": 3848
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 3848,
      "training_step_time": 7.788530588150024
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 2.435879945755005,
      "step": 3848
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 3848,
      "training_step_time": 7.4596850872039795
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 3.135836124420166,
      "step": 3848
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 3848,
      "training_step_time": 9.900789260864258
    },
    {
      "epoch": 1.4678955078125e-05,
      "model_forward_time": 3.136345863342285,
      "step": 3848
    },
    {
      "epoch": 1.4678955078125e-05,
      "step": 3848,
      "training_step_time": 10.011476039886475
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 3.1463661193847656,
      "step": 3849
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 3849,
      "training_step_time": 9.916783571243286
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 3.418182849884033,
      "step": 3849
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 3849,
      "training_step_time": 10.18335223197937
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 3.1368484497070312,
      "step": 3849
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 3849,
      "training_step_time": 9.905638456344604
    },
    {
      "epoch": 1.4682769775390626e-05,
      "model_forward_time": 2.4327902793884277,
      "step": 3849
    },
    {
      "epoch": 1.4682769775390626e-05,
      "step": 3849,
      "training_step_time": 10.325706958770752
    },
    {
      "epoch": 1.468658447265625e-05,
      "grad_norm": 0.4334598481655121,
      "learning_rate": 1e-05,
      "loss": 0.0981,
      "step": 3850
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 2.4450478553771973,
      "step": 3850
    },
    {
      "dynamics_loss_avg": 0.06543107479810714,
      "epoch": 1.468658447265625e-05,
      "step": 3850
    },
    {
      "action_loss_avg": 0.03444930734112859,
      "epoch": 1.468658447265625e-05,
      "step": 3850
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 3850,
      "training_step_time": 7.467499494552612
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 3.135964870452881,
      "step": 3850
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 3850,
      "training_step_time": 9.898971319198608
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 3.427489757537842,
      "step": 3850
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 3850,
      "training_step_time": 10.192504167556763
    },
    {
      "epoch": 1.468658447265625e-05,
      "model_forward_time": 2.431199550628662,
      "step": 3850
    },
    {
      "epoch": 1.468658447265625e-05,
      "step": 3850,
      "training_step_time": 10.27097773551941
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 3.158656597137451,
      "step": 3851
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 3851,
      "training_step_time": 9.931057691574097
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 3.142927885055542,
      "step": 3851
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 3851,
      "training_step_time": 9.915246486663818
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 3.1476826667785645,
      "step": 3851
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 3851,
      "training_step_time": 9.921422004699707
    },
    {
      "epoch": 1.4690399169921875e-05,
      "model_forward_time": 3.3990297317504883,
      "step": 3851
    },
    {
      "epoch": 1.4690399169921875e-05,
      "step": 3851,
      "training_step_time": 10.287670135498047
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 3.15220308303833,
      "step": 3852
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 3852,
      "training_step_time": 9.927909851074219
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 3.1456902027130127,
      "step": 3852
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 3852,
      "training_step_time": 9.922868490219116
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 2.4449622631073,
      "step": 3852
    },
    {
      "dynamics_loss_avg": 0.04652402717620134,
      "epoch": 1.46942138671875e-05,
      "step": 3852
    },
    {
      "action_loss_avg": 0.028450480848550796,
      "epoch": 1.46942138671875e-05,
      "step": 3852
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 3852,
      "training_step_time": 7.480319976806641
    },
    {
      "epoch": 1.46942138671875e-05,
      "model_forward_time": 3.1469905376434326,
      "step": 3852
    },
    {
      "epoch": 1.46942138671875e-05,
      "step": 3852,
      "training_step_time": 10.02644419670105
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 3.418154001235962,
      "step": 3853
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 3853,
      "training_step_time": 10.193156242370605
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 2.440551280975342,
      "step": 3853
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 3853,
      "training_step_time": 7.467522859573364
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 3.136958122253418,
      "step": 3853
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 3853,
      "training_step_time": 9.904371738433838
    },
    {
      "epoch": 1.4698028564453126e-05,
      "model_forward_time": 3.1407642364501953,
      "step": 3853
    },
    {
      "epoch": 1.4698028564453126e-05,
      "step": 3853,
      "training_step_time": 10.02782654762268
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 3.145216464996338,
      "step": 3854
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 3854,
      "training_step_time": 9.916295528411865
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 3.4097065925598145,
      "step": 3854
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 3854,
      "training_step_time": 10.183545589447021
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 3.137727975845337,
      "step": 3854
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 3854,
      "training_step_time": 9.92341685295105
    },
    {
      "epoch": 1.470184326171875e-05,
      "model_forward_time": 2.4338483810424805,
      "step": 3854
    },
    {
      "epoch": 1.470184326171875e-05,
      "step": 3854,
      "training_step_time": 10.31300687789917
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 3.1561429500579834,
      "step": 3855
    },
    {
      "dynamics_loss_avg": 0.060496386513113974,
      "epoch": 1.4705657958984375e-05,
      "step": 3855
    },
    {
      "action_loss_avg": 0.029354495368897913,
      "epoch": 1.4705657958984375e-05,
      "step": 3855
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 3855,
      "training_step_time": 9.932985544204712
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 2.438739538192749,
      "step": 3855
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 3855,
      "training_step_time": 7.471876621246338
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 2.710517168045044,
      "step": 3855
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 3855,
      "training_step_time": 7.741568565368652
    },
    {
      "epoch": 1.4705657958984375e-05,
      "model_forward_time": 3.139925241470337,
      "step": 3855
    },
    {
      "epoch": 1.4705657958984375e-05,
      "step": 3855,
      "training_step_time": 10.344101667404175
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 3.151047945022583,
      "step": 3856
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 3856,
      "training_step_time": 9.925561904907227
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 2.4328861236572266,
      "step": 3856
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 3856,
      "training_step_time": 7.463618278503418
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 3.1362056732177734,
      "step": 3856
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 3856,
      "training_step_time": 9.909177541732788
    },
    {
      "epoch": 1.470947265625e-05,
      "model_forward_time": 2.4337921142578125,
      "step": 3856
    },
    {
      "epoch": 1.470947265625e-05,
      "step": 3856,
      "training_step_time": 10.243396759033203
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 2.440913200378418,
      "step": 3857
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 3857,
      "training_step_time": 7.466562509536743
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 3.1267926692962646,
      "step": 3857
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 3857,
      "training_step_time": 9.893096446990967
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 2.42976975440979,
      "step": 3857
    },
    {
      "dynamics_loss_avg": 0.06840563174337148,
      "epoch": 1.4713287353515626e-05,
      "step": 3857
    },
    {
      "action_loss_avg": 0.03269335892982781,
      "epoch": 1.4713287353515626e-05,
      "step": 3857
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 3857,
      "training_step_time": 7.454880237579346
    },
    {
      "epoch": 1.4713287353515626e-05,
      "model_forward_time": 2.429932117462158,
      "step": 3857
    },
    {
      "epoch": 1.4713287353515626e-05,
      "step": 3857,
      "training_step_time": 10.023088455200195
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 2.4459187984466553,
      "step": 3858
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 3858,
      "training_step_time": 7.477113246917725
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 3.3777027130126953,
      "step": 3858
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 3858,
      "training_step_time": 10.145638227462769
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 2.4316141605377197,
      "step": 3858
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 3858,
      "training_step_time": 7.464831352233887
    },
    {
      "epoch": 1.471710205078125e-05,
      "model_forward_time": 2.4267821311950684,
      "step": 3858
    },
    {
      "epoch": 1.471710205078125e-05,
      "step": 3858,
      "training_step_time": 10.311655282974243
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 3.1364223957061768,
      "step": 3859
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 3859,
      "training_step_time": 9.916430711746216
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 3.1474192142486572,
      "step": 3859
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 3859,
      "training_step_time": 9.929796934127808
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 3.436697483062744,
      "step": 3859
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 3859,
      "training_step_time": 10.20562219619751
    },
    {
      "epoch": 1.4720916748046875e-05,
      "model_forward_time": 3.130922317504883,
      "step": 3859
    },
    {
      "epoch": 1.4720916748046875e-05,
      "step": 3859,
      "training_step_time": 10.0171799659729
    },
    {
      "epoch": 1.47247314453125e-05,
      "grad_norm": 0.5145561099052429,
      "learning_rate": 1e-05,
      "loss": 0.0964,
      "step": 3860
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 2.4470770359039307,
      "step": 3860
    },
    {
      "dynamics_loss_avg": 0.06758902743458747,
      "epoch": 1.47247314453125e-05,
      "step": 3860
    },
    {
      "action_loss_avg": 0.04099434893578291,
      "epoch": 1.47247314453125e-05,
      "step": 3860
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 3860,
      "training_step_time": 7.476756572723389
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 2.4262561798095703,
      "step": 3860
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 3860,
      "training_step_time": 7.450002431869507
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 3.128938674926758,
      "step": 3860
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 3860,
      "training_step_time": 9.898000001907349
    },
    {
      "epoch": 1.47247314453125e-05,
      "model_forward_time": 2.7222189903259277,
      "step": 3860
    },
    {
      "epoch": 1.47247314453125e-05,
      "step": 3860,
      "training_step_time": 10.252651691436768
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 1.7637560367584229,
      "step": 3861
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 3861,
      "training_step_time": 5.225708723068237
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 3.1308515071868896,
      "step": 3861
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 3861,
      "training_step_time": 9.902377367019653
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 1.7424969673156738,
      "step": 3861
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 3861,
      "training_step_time": 5.2024006843566895
    },
    {
      "epoch": 1.4728546142578126e-05,
      "model_forward_time": 2.4279797077178955,
      "step": 3861
    },
    {
      "epoch": 1.4728546142578126e-05,
      "step": 3861,
      "training_step_time": 10.29980754852295
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 2.4475388526916504,
      "step": 3862
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 3862,
      "training_step_time": 7.4859302043914795
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 3.4084906578063965,
      "step": 3862
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 3862,
      "training_step_time": 10.179815292358398
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 3.1321773529052734,
      "step": 3862
    },
    {
      "dynamics_loss_avg": 0.06627513617277145,
      "epoch": 1.473236083984375e-05,
      "step": 3862
    },
    {
      "action_loss_avg": 0.028681694809347392,
      "epoch": 1.473236083984375e-05,
      "step": 3862
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 3862,
      "training_step_time": 9.90506100654602
    },
    {
      "epoch": 1.473236083984375e-05,
      "model_forward_time": 2.4320974349975586,
      "step": 3862
    },
    {
      "epoch": 1.473236083984375e-05,
      "step": 3862,
      "training_step_time": 10.288949489593506
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 3.141416311264038,
      "step": 3863
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 3863,
      "training_step_time": 9.910909175872803
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 3.1271603107452393,
      "step": 3863
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 3863,
      "training_step_time": 9.894566059112549
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 3.3894901275634766,
      "step": 3863
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 3863,
      "training_step_time": 10.161664247512817
    },
    {
      "epoch": 1.4736175537109375e-05,
      "model_forward_time": 3.13533091545105,
      "step": 3863
    },
    {
      "epoch": 1.4736175537109375e-05,
      "step": 3863,
      "training_step_time": 10.027715682983398
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 3.1564273834228516,
      "step": 3864
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 3864,
      "training_step_time": 9.93016767501831
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 3.1353938579559326,
      "step": 3864
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 3864,
      "training_step_time": 9.90874981880188
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 1.7444477081298828,
      "step": 3864
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 3864,
      "training_step_time": 5.202443599700928
    },
    {
      "epoch": 1.4739990234375e-05,
      "model_forward_time": 2.4319210052490234,
      "step": 3864
    },
    {
      "epoch": 1.4739990234375e-05,
      "step": 3864,
      "training_step_time": 10.022481203079224
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 3.4463930130004883,
      "step": 3865
    },
    {
      "dynamics_loss_avg": 0.05774882081896067,
      "epoch": 1.4743804931640626e-05,
      "step": 3865
    },
    {
      "action_loss_avg": 0.024588835425674915,
      "epoch": 1.4743804931640626e-05,
      "step": 3865
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 3865,
      "training_step_time": 10.231334447860718
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 3.1356425285339355,
      "step": 3865
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 3865,
      "training_step_time": 9.912737607955933
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 2.4360384941101074,
      "step": 3865
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 3865,
      "training_step_time": 7.46923303604126
    },
    {
      "epoch": 1.4743804931640626e-05,
      "model_forward_time": 3.1386666297912598,
      "step": 3865
    },
    {
      "epoch": 1.4743804931640626e-05,
      "step": 3865,
      "training_step_time": 10.346300840377808
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 3.1770246028900146,
      "step": 3866
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 3866,
      "training_step_time": 9.967161893844604
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 2.678553819656372,
      "step": 3866
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 3866,
      "training_step_time": 7.711710453033447
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 2.43074893951416,
      "step": 3866
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 3866,
      "training_step_time": 7.4628586769104
    },
    {
      "epoch": 1.474761962890625e-05,
      "model_forward_time": 2.4253432750701904,
      "step": 3866
    },
    {
      "epoch": 1.474761962890625e-05,
      "step": 3866,
      "training_step_time": 10.033451080322266
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 3.1485488414764404,
      "step": 3867
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 3867,
      "training_step_time": 9.914373636245728
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 2.4280591011047363,
      "step": 3867
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 3867,
      "training_step_time": 7.46051549911499
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 3.4044156074523926,
      "step": 3867
    },
    {
      "dynamics_loss_avg": 0.05367824230343103,
      "epoch": 1.4751434326171875e-05,
      "step": 3867
    },
    {
      "action_loss_avg": 0.02220877828076482,
      "epoch": 1.4751434326171875e-05,
      "step": 3867
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 3867,
      "training_step_time": 10.17639970779419
    },
    {
      "epoch": 1.4751434326171875e-05,
      "model_forward_time": 3.1305692195892334,
      "step": 3867
    },
    {
      "epoch": 1.4751434326171875e-05,
      "step": 3867,
      "training_step_time": 10.314012289047241
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 2.4414045810699463,
      "step": 3868
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 3868,
      "training_step_time": 7.466277360916138
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 2.426488161087036,
      "step": 3868
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 3868,
      "training_step_time": 7.453981637954712
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 2.4258902072906494,
      "step": 3868
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 3868,
      "training_step_time": 7.4571733474731445
    },
    {
      "epoch": 1.47552490234375e-05,
      "model_forward_time": 3.3897433280944824,
      "step": 3868
    },
    {
      "epoch": 1.47552490234375e-05,
      "step": 3868,
      "training_step_time": 10.258326053619385
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 3.1578309535980225,
      "step": 3869
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 3869,
      "training_step_time": 9.938891172409058
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 3.1383001804351807,
      "step": 3869
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 3869,
      "training_step_time": 9.917231321334839
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 1.744675874710083,
      "step": 3869
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 3869,
      "training_step_time": 5.207219362258911
    },
    {
      "epoch": 1.4759063720703126e-05,
      "model_forward_time": 2.428539752960205,
      "step": 3869
    },
    {
      "epoch": 1.4759063720703126e-05,
      "step": 3869,
      "training_step_time": 10.29377293586731
    },
    {
      "epoch": 1.476287841796875e-05,
      "grad_norm": 0.766221821308136,
      "learning_rate": 1e-05,
      "loss": 0.0943,
      "step": 3870
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 3.146685838699341,
      "step": 3870
    },
    {
      "dynamics_loss_avg": 0.060211675427854064,
      "epoch": 1.476287841796875e-05,
      "step": 3870
    },
    {
      "action_loss_avg": 0.04059263560920954,
      "epoch": 1.476287841796875e-05,
      "step": 3870
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 3870,
      "training_step_time": 9.923870325088501
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 3.448302984237671,
      "step": 3870
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 3870,
      "training_step_time": 10.222176790237427
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 2.434260606765747,
      "step": 3870
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 3870,
      "training_step_time": 7.467609405517578
    },
    {
      "epoch": 1.476287841796875e-05,
      "model_forward_time": 3.1475117206573486,
      "step": 3870
    },
    {
      "epoch": 1.476287841796875e-05,
      "step": 3870,
      "training_step_time": 10.362508296966553
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 3.1677753925323486,
      "step": 3871
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 3871,
      "training_step_time": 9.952030181884766
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 3.1431546211242676,
      "step": 3871
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 3871,
      "training_step_time": 9.92359709739685
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 2.726992607116699,
      "step": 3871
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 3871,
      "training_step_time": 7.760834693908691
    },
    {
      "epoch": 1.4766693115234375e-05,
      "model_forward_time": 3.1413915157318115,
      "step": 3871
    },
    {
      "epoch": 1.4766693115234375e-05,
      "step": 3871,
      "training_step_time": 10.010746717453003
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 3.1719884872436523,
      "step": 3872
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 3872,
      "training_step_time": 9.949556827545166
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 2.4363138675689697,
      "step": 3872
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 3872,
      "training_step_time": 7.465365886688232
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 2.435169219970703,
      "step": 3872
    },
    {
      "dynamics_loss_avg": 0.06248965840786695,
      "epoch": 1.47705078125e-05,
      "step": 3872
    },
    {
      "action_loss_avg": 0.024209662154316903,
      "epoch": 1.47705078125e-05,
      "step": 3872
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 3872,
      "training_step_time": 7.4644246101379395
    },
    {
      "epoch": 1.47705078125e-05,
      "model_forward_time": 2.72751522064209,
      "step": 3872
    },
    {
      "epoch": 1.47705078125e-05,
      "step": 3872,
      "training_step_time": 10.329055070877075
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 3.1644484996795654,
      "step": 3873
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 3873,
      "training_step_time": 9.94322681427002
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 1.7497825622558594,
      "step": 3873
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 3873,
      "training_step_time": 5.209186315536499
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 2.4398231506347656,
      "step": 3873
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 3873,
      "training_step_time": 7.4711809158325195
    },
    {
      "epoch": 1.4774322509765626e-05,
      "model_forward_time": 2.4357757568359375,
      "step": 3873
    },
    {
      "epoch": 1.4774322509765626e-05,
      "step": 3873,
      "training_step_time": 10.28174114227295
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 3.5081446170806885,
      "step": 3874
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 3874,
      "training_step_time": 10.285213470458984
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 3.134101152420044,
      "step": 3874
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 3874,
      "training_step_time": 9.908490896224976
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 3.140678882598877,
      "step": 3874
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 3874,
      "training_step_time": 9.916812181472778
    },
    {
      "epoch": 1.477813720703125e-05,
      "model_forward_time": 3.1340317726135254,
      "step": 3874
    },
    {
      "epoch": 1.477813720703125e-05,
      "step": 3874,
      "training_step_time": 10.306599617004395
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 3.153207302093506,
      "step": 3875
    },
    {
      "dynamics_loss_avg": 0.06512724850326776,
      "epoch": 1.4781951904296875e-05,
      "step": 3875
    },
    {
      "action_loss_avg": 0.03940495727583766,
      "epoch": 1.4781951904296875e-05,
      "step": 3875
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 3875,
      "training_step_time": 9.930545806884766
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 1.745776653289795,
      "step": 3875
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 3875,
      "training_step_time": 5.20206093788147
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 2.743539810180664,
      "step": 3875
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 3875,
      "training_step_time": 7.768152475357056
    },
    {
      "epoch": 1.4781951904296875e-05,
      "model_forward_time": 3.1399261951446533,
      "step": 3875
    },
    {
      "epoch": 1.4781951904296875e-05,
      "step": 3875,
      "training_step_time": 10.288677453994751
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 3.158595561981201,
      "step": 3876
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 3876,
      "training_step_time": 9.938209056854248
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 2.4374473094940186,
      "step": 3876
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 3876,
      "training_step_time": 7.475633382797241
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 3.1441807746887207,
      "step": 3876
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 3876,
      "training_step_time": 9.921643733978271
    },
    {
      "epoch": 1.47857666015625e-05,
      "model_forward_time": 2.745356798171997,
      "step": 3876
    },
    {
      "epoch": 1.47857666015625e-05,
      "step": 3876,
      "training_step_time": 10.284340381622314
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 2.459120512008667,
      "step": 3877
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 3877,
      "training_step_time": 7.499396324157715
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 3.1395857334136963,
      "step": 3877
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 3877,
      "training_step_time": 9.924047470092773
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 3.1476361751556396,
      "step": 3877
    },
    {
      "dynamics_loss_avg": 0.047624476719647646,
      "epoch": 1.4789581298828124e-05,
      "step": 3877
    },
    {
      "action_loss_avg": 0.026875486504286526,
      "epoch": 1.4789581298828124e-05,
      "step": 3877
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 3877,
      "training_step_time": 9.939344882965088
    },
    {
      "epoch": 1.4789581298828124e-05,
      "model_forward_time": 3.153717279434204,
      "step": 3877
    },
    {
      "epoch": 1.4789581298828124e-05,
      "step": 3877,
      "training_step_time": 10.063003301620483
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 3.580965518951416,
      "step": 3878
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 3878,
      "training_step_time": 10.387280225753784
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 2.4528026580810547,
      "step": 3878
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 3878,
      "training_step_time": 7.505510091781616
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 2.4428958892822266,
      "step": 3878
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 3878,
      "training_step_time": 7.484432697296143
    },
    {
      "epoch": 1.479339599609375e-05,
      "model_forward_time": 3.157456159591675,
      "step": 3878
    },
    {
      "epoch": 1.479339599609375e-05,
      "step": 3878,
      "training_step_time": 10.3158438205719
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 2.458184003829956,
      "step": 3879
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 3879,
      "training_step_time": 7.498882293701172
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 2.4411466121673584,
      "step": 3879
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 3879,
      "training_step_time": 7.479284763336182
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 2.7806100845336914,
      "step": 3879
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 3879,
      "training_step_time": 7.818145513534546
    },
    {
      "epoch": 1.4797210693359375e-05,
      "model_forward_time": 3.149184465408325,
      "step": 3879
    },
    {
      "epoch": 1.4797210693359375e-05,
      "step": 3879,
      "training_step_time": 10.322042226791382
    },
    {
      "epoch": 1.4801025390625e-05,
      "grad_norm": 0.4972519874572754,
      "learning_rate": 1e-05,
      "loss": 0.0915,
      "step": 3880
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 1.7725629806518555,
      "step": 3880
    },
    {
      "dynamics_loss_avg": 0.06620752140879631,
      "epoch": 1.4801025390625e-05,
      "step": 3880
    },
    {
      "action_loss_avg": 0.027779203467071058,
      "epoch": 1.4801025390625e-05,
      "step": 3880
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 3880,
      "training_step_time": 5.24676513671875
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 3.1428756713867188,
      "step": 3880
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 3880,
      "training_step_time": 9.928399562835693
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 3.1546437740325928,
      "step": 3880
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 3880,
      "training_step_time": 9.93481707572937
    },
    {
      "epoch": 1.4801025390625e-05,
      "model_forward_time": 2.7384960651397705,
      "step": 3880
    },
    {
      "epoch": 1.4801025390625e-05,
      "step": 3880,
      "training_step_time": 10.350664615631104
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 2.4451797008514404,
      "step": 3881
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 3881,
      "training_step_time": 7.481725454330444
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 3.1298649311065674,
      "step": 3881
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 3881,
      "training_step_time": 9.900880098342896
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 3.1328814029693604,
      "step": 3881
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 3881,
      "training_step_time": 9.903127670288086
    },
    {
      "epoch": 1.4804840087890624e-05,
      "model_forward_time": 3.1312763690948486,
      "step": 3881
    },
    {
      "epoch": 1.4804840087890624e-05,
      "step": 3881,
      "training_step_time": 10.281797885894775
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 1.7789990901947021,
      "step": 3882
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 3882,
      "training_step_time": 5.250646114349365
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 3.4453818798065186,
      "step": 3882
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 3882,
      "training_step_time": 10.216259241104126
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 3.153028726577759,
      "step": 3882
    },
    {
      "dynamics_loss_avg": 0.06936007691547275,
      "epoch": 1.480865478515625e-05,
      "step": 3882
    },
    {
      "action_loss_avg": 0.04077197152655572,
      "epoch": 1.480865478515625e-05,
      "step": 3882
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 3882,
      "training_step_time": 9.93684697151184
    },
    {
      "epoch": 1.480865478515625e-05,
      "model_forward_time": 2.4347145557403564,
      "step": 3882
    },
    {
      "epoch": 1.480865478515625e-05,
      "step": 3882,
      "training_step_time": 10.02938175201416
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 2.465508460998535,
      "step": 3883
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 3883,
      "training_step_time": 7.500234127044678
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 2.4390323162078857,
      "step": 3883
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 3883,
      "training_step_time": 7.474126577377319
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 3.460859537124634,
      "step": 3883
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 3883,
      "training_step_time": 10.237189769744873
    },
    {
      "epoch": 1.4812469482421875e-05,
      "model_forward_time": 2.4429569244384766,
      "step": 3883
    },
    {
      "epoch": 1.4812469482421875e-05,
      "step": 3883,
      "training_step_time": 10.318355083465576
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 2.4529623985290527,
      "step": 3884
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 3884,
      "training_step_time": 7.480696201324463
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 2.432971239089966,
      "step": 3884
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 3884,
      "training_step_time": 7.456676483154297
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 2.4354476928710938,
      "step": 3884
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 3884,
      "training_step_time": 7.464718580245972
    },
    {
      "epoch": 1.48162841796875e-05,
      "model_forward_time": 3.14479923248291,
      "step": 3884
    },
    {
      "epoch": 1.48162841796875e-05,
      "step": 3884,
      "training_step_time": 10.326708793640137
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 3.483123302459717,
      "step": 3885
    },
    {
      "dynamics_loss_avg": 0.05951713165268302,
      "epoch": 1.4820098876953124e-05,
      "step": 3885
    },
    {
      "action_loss_avg": 0.036535862303571774,
      "epoch": 1.4820098876953124e-05,
      "step": 3885
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 3885,
      "training_step_time": 10.254260540008545
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 3.1454105377197266,
      "step": 3885
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 3885,
      "training_step_time": 9.918906450271606
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 3.1516849994659424,
      "step": 3885
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 3885,
      "training_step_time": 9.922908306121826
    },
    {
      "epoch": 1.4820098876953124e-05,
      "model_forward_time": 2.4397168159484863,
      "step": 3885
    },
    {
      "epoch": 1.4820098876953124e-05,
      "step": 3885,
      "training_step_time": 10.309860944747925
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 3.1472580432891846,
      "step": 3886
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 3886,
      "training_step_time": 9.920281887054443
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 3.429377794265747,
      "step": 3886
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 3886,
      "training_step_time": 10.20070195198059
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 3.1361660957336426,
      "step": 3886
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 3886,
      "training_step_time": 9.905290365219116
    },
    {
      "epoch": 1.482391357421875e-05,
      "model_forward_time": 2.435696840286255,
      "step": 3886
    },
    {
      "epoch": 1.482391357421875e-05,
      "step": 3886,
      "training_step_time": 10.050429344177246
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 3.1524617671966553,
      "step": 3887
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 3887,
      "training_step_time": 9.931622982025146
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 3.1319656372070312,
      "step": 3887
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 3887,
      "training_step_time": 10.171218395233154
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 2.436086654663086,
      "step": 3887
    },
    {
      "dynamics_loss_avg": 0.055407238006591794,
      "epoch": 1.4827728271484375e-05,
      "step": 3887
    },
    {
      "action_loss_avg": 0.02449826830998063,
      "epoch": 1.4827728271484375e-05,
      "step": 3887
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 3887,
      "training_step_time": 7.468634843826294
    },
    {
      "epoch": 1.4827728271484375e-05,
      "model_forward_time": 3.1322412490844727,
      "step": 3887
    },
    {
      "epoch": 1.4827728271484375e-05,
      "step": 3887,
      "training_step_time": 10.292911052703857
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 3.151320695877075,
      "step": 3888
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 3888,
      "training_step_time": 9.920215129852295
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 2.4276797771453857,
      "step": 3888
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 3888,
      "training_step_time": 7.450638055801392
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 2.42631459236145,
      "step": 3888
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 3888,
      "training_step_time": 7.4467198848724365
    },
    {
      "epoch": 1.483154296875e-05,
      "model_forward_time": 3.452507734298706,
      "step": 3888
    },
    {
      "epoch": 1.483154296875e-05,
      "step": 3888,
      "training_step_time": 10.319347381591797
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 2.4439518451690674,
      "step": 3889
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 3889,
      "training_step_time": 7.466346263885498
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 2.427969455718994,
      "step": 3889
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 3889,
      "training_step_time": 7.4482581615448
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 3.1255733966827393,
      "step": 3889
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 3889,
      "training_step_time": 9.889864683151245
    },
    {
      "epoch": 1.4835357666015624e-05,
      "model_forward_time": 2.427541971206665,
      "step": 3889
    },
    {
      "epoch": 1.4835357666015624e-05,
      "step": 3889,
      "training_step_time": 10.259670495986938
    },
    {
      "epoch": 1.483917236328125e-05,
      "grad_norm": 0.6968552470207214,
      "learning_rate": 1e-05,
      "loss": 0.0917,
      "step": 3890
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 2.7023301124572754,
      "step": 3890
    },
    {
      "dynamics_loss_avg": 0.05194306243211031,
      "epoch": 1.483917236328125e-05,
      "step": 3890
    },
    {
      "action_loss_avg": 0.030715092923492192,
      "epoch": 1.483917236328125e-05,
      "step": 3890
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 3890,
      "training_step_time": 7.722297430038452
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 2.4265944957733154,
      "step": 3890
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 3890,
      "training_step_time": 7.447000741958618
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 2.426234483718872,
      "step": 3890
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 3890,
      "training_step_time": 7.446125268936157
    },
    {
      "epoch": 1.483917236328125e-05,
      "model_forward_time": 3.12599515914917,
      "step": 3890
    },
    {
      "epoch": 1.483917236328125e-05,
      "step": 3890,
      "training_step_time": 10.31065034866333
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 2.4390320777893066,
      "step": 3891
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 3891,
      "training_step_time": 7.463965177536011
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 3.3881685733795166,
      "step": 3891
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 3891,
      "training_step_time": 10.148436069488525
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 3.1276676654815674,
      "step": 3891
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 3891,
      "training_step_time": 9.889856576919556
    },
    {
      "epoch": 1.4842987060546875e-05,
      "model_forward_time": 3.133733034133911,
      "step": 3891
    },
    {
      "epoch": 1.4842987060546875e-05,
      "step": 3891,
      "training_step_time": 10.007727146148682
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 2.443688154220581,
      "step": 3892
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 3892,
      "training_step_time": 7.469100713729858
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 3.1371352672576904,
      "step": 3892
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 3892,
      "training_step_time": 9.902485847473145
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 3.413397789001465,
      "step": 3892
    },
    {
      "dynamics_loss_avg": 0.06593674905598164,
      "epoch": 1.48468017578125e-05,
      "step": 3892
    },
    {
      "action_loss_avg": 0.03376121539622545,
      "epoch": 1.48468017578125e-05,
      "step": 3892
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 3892,
      "training_step_time": 10.182716369628906
    },
    {
      "epoch": 1.48468017578125e-05,
      "model_forward_time": 3.1340200901031494,
      "step": 3892
    },
    {
      "epoch": 1.48468017578125e-05,
      "step": 3892,
      "training_step_time": 10.03879714012146
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 2.4500489234924316,
      "step": 3893
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 3893,
      "training_step_time": 7.479065656661987
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 3.1418631076812744,
      "step": 3893
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 3893,
      "training_step_time": 9.918304920196533
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 3.156013250350952,
      "step": 3893
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 3893,
      "training_step_time": 9.931355953216553
    },
    {
      "epoch": 1.4850616455078124e-05,
      "model_forward_time": 2.4416534900665283,
      "step": 3893
    },
    {
      "epoch": 1.4850616455078124e-05,
      "step": 3893,
      "training_step_time": 10.032635688781738
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 3.4492268562316895,
      "step": 3894
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 3894,
      "training_step_time": 10.222810506820679
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 2.437098503112793,
      "step": 3894
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 3894,
      "training_step_time": 7.467033863067627
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 3.1505563259124756,
      "step": 3894
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 3894,
      "training_step_time": 9.925786018371582
    },
    {
      "epoch": 1.485443115234375e-05,
      "model_forward_time": 2.43087100982666,
      "step": 3894
    },
    {
      "epoch": 1.485443115234375e-05,
      "step": 3894,
      "training_step_time": 10.285156488418579
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 3.1555583477020264,
      "step": 3895
    },
    {
      "dynamics_loss_avg": 0.0717744804918766,
      "epoch": 1.4858245849609375e-05,
      "step": 3895
    },
    {
      "action_loss_avg": 0.03591004349291325,
      "epoch": 1.4858245849609375e-05,
      "step": 3895
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 3895,
      "training_step_time": 9.932076454162598
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 2.6783576011657715,
      "step": 3895
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 3895,
      "training_step_time": 7.707323312759399
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 3.1367902755737305,
      "step": 3895
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 3895,
      "training_step_time": 9.910434484481812
    },
    {
      "epoch": 1.4858245849609375e-05,
      "model_forward_time": 2.4306540489196777,
      "step": 3895
    },
    {
      "epoch": 1.4858245849609375e-05,
      "step": 3895,
      "training_step_time": 10.287665605545044
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 2.4390883445739746,
      "step": 3896
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 3896,
      "training_step_time": 7.470686435699463
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 3.1364729404449463,
      "step": 3896
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 3896,
      "training_step_time": 9.913140058517456
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 3.390202045440674,
      "step": 3896
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 3896,
      "training_step_time": 10.165953159332275
    },
    {
      "epoch": 1.4862060546875e-05,
      "model_forward_time": 3.1384129524230957,
      "step": 3896
    },
    {
      "epoch": 1.4862060546875e-05,
      "step": 3896,
      "training_step_time": 10.330905675888062
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 1.761343240737915,
      "step": 3897
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 3897,
      "training_step_time": 5.2322211265563965
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 2.4321327209472656,
      "step": 3897
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 3897,
      "training_step_time": 7.464120388031006
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 3.135719060897827,
      "step": 3897
    },
    {
      "dynamics_loss_avg": 0.06097269952297211,
      "epoch": 1.4865875244140624e-05,
      "step": 3897
    },
    {
      "action_loss_avg": 0.03595138676464558,
      "epoch": 1.4865875244140624e-05,
      "step": 3897
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 3897,
      "training_step_time": 9.901880025863647
    },
    {
      "epoch": 1.4865875244140624e-05,
      "model_forward_time": 3.1351637840270996,
      "step": 3897
    },
    {
      "epoch": 1.4865875244140624e-05,
      "step": 3897,
      "training_step_time": 10.310694456100464
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 3.1528496742248535,
      "step": 3898
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 3898,
      "training_step_time": 9.919874906539917
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 3.1294777393341064,
      "step": 3898
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 3898,
      "training_step_time": 9.890883207321167
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 2.4299869537353516,
      "step": 3898
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 3898,
      "training_step_time": 7.4481823444366455
    },
    {
      "epoch": 1.486968994140625e-05,
      "model_forward_time": 2.4349515438079834,
      "step": 3898
    },
    {
      "epoch": 1.486968994140625e-05,
      "step": 3898,
      "training_step_time": 10.230839252471924
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 3.149749517440796,
      "step": 3899
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 3899,
      "training_step_time": 9.930069923400879
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 3.3827199935913086,
      "step": 3899
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 3899,
      "training_step_time": 10.148099660873413
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 3.1418449878692627,
      "step": 3899
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 3899,
      "training_step_time": 9.910590410232544
    },
    {
      "epoch": 1.4873504638671875e-05,
      "model_forward_time": 2.437124013900757,
      "step": 3899
    },
    {
      "epoch": 1.4873504638671875e-05,
      "step": 3899,
      "training_step_time": 10.243081092834473
    },
    {
      "epoch": 1.48773193359375e-05,
      "grad_norm": 0.6644129157066345,
      "learning_rate": 1e-05,
      "loss": 0.0926,
      "step": 3900
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 1.7664356231689453,
      "step": 3900
    },
    {
      "dynamics_loss_avg": 0.051575775258243084,
      "epoch": 1.48773193359375e-05,
      "step": 3900
    },
    {
      "action_loss_avg": 0.028613726515322922,
      "epoch": 1.48773193359375e-05,
      "step": 3900
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 3900,
      "training_step_time": 5.2268455028533936
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 2.436922550201416,
      "step": 3900
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 3900,
      "training_step_time": 7.462411880493164
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 3.4485950469970703,
      "step": 3900
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 3900,
      "training_step_time": 10.220276117324829
    },
    {
      "epoch": 1.48773193359375e-05,
      "model_forward_time": 2.43937611579895,
      "step": 3900
    },
    {
      "epoch": 1.48773193359375e-05,
      "step": 3900,
      "training_step_time": 10.325353860855103
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 2.4610989093780518,
      "step": 3901
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 3901,
      "training_step_time": 7.4874467849731445
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 3.1399428844451904,
      "step": 3901
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 3901,
      "training_step_time": 9.905223369598389
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 2.4352333545684814,
      "step": 3901
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 3901,
      "training_step_time": 7.458935737609863
    },
    {
      "epoch": 1.4881134033203124e-05,
      "model_forward_time": 1.740553855895996,
      "step": 3901
    },
    {
      "epoch": 1.4881134033203124e-05,
      "step": 3901,
      "training_step_time": 10.345547914505005
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 3.406484842300415,
      "step": 3902
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 3902,
      "training_step_time": 10.170061111450195
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 3.1340560913085938,
      "step": 3902
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 3902,
      "training_step_time": 9.895747423171997
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 2.433274745941162,
      "step": 3902
    },
    {
      "dynamics_loss_avg": 0.06559780612587929,
      "epoch": 1.488494873046875e-05,
      "step": 3902
    },
    {
      "action_loss_avg": 0.036820539459586146,
      "epoch": 1.488494873046875e-05,
      "step": 3902
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 3902,
      "training_step_time": 7.4577414989471436
    },
    {
      "epoch": 1.488494873046875e-05,
      "model_forward_time": 3.1361632347106934,
      "step": 3902
    },
    {
      "epoch": 1.488494873046875e-05,
      "step": 3902,
      "training_step_time": 10.018049240112305
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 3.1542775630950928,
      "step": 3903
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 3903,
      "training_step_time": 9.924828290939331
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 2.7083609104156494,
      "step": 3903
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 3903,
      "training_step_time": 7.73555850982666
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 2.4338178634643555,
      "step": 3903
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 3903,
      "training_step_time": 7.464397192001343
    },
    {
      "epoch": 1.4888763427734375e-05,
      "model_forward_time": 2.4374489784240723,
      "step": 3903
    },
    {
      "epoch": 1.4888763427734375e-05,
      "step": 3903,
      "training_step_time": 10.315469980239868
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 3.1508986949920654,
      "step": 3904
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 3904,
      "training_step_time": 9.930916786193848
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 3.1364409923553467,
      "step": 3904
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 3904,
      "training_step_time": 9.914084434509277
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 2.4335649013519287,
      "step": 3904
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 3904,
      "training_step_time": 7.7190515995025635
    },
    {
      "epoch": 1.4892578125e-05,
      "model_forward_time": 3.135910987854004,
      "step": 3904
    },
    {
      "epoch": 1.4892578125e-05,
      "step": 3904,
      "training_step_time": 10.28263545036316
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 2.4406094551086426,
      "step": 3905
    },
    {
      "dynamics_loss_avg": 0.057558062486350536,
      "epoch": 1.4896392822265624e-05,
      "step": 3905
    },
    {
      "action_loss_avg": 0.020392033224925398,
      "epoch": 1.4896392822265624e-05,
      "step": 3905
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 3905,
      "training_step_time": 7.4671056270599365
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 2.4294145107269287,
      "step": 3905
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 3905,
      "training_step_time": 7.451021194458008
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 3.1304209232330322,
      "step": 3905
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 3905,
      "training_step_time": 9.892675876617432
    },
    {
      "epoch": 1.4896392822265624e-05,
      "model_forward_time": 2.439882755279541,
      "step": 3905
    },
    {
      "epoch": 1.4896392822265624e-05,
      "step": 3905,
      "training_step_time": 10.020682334899902
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 2.7654333114624023,
      "step": 3906
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 3906,
      "training_step_time": 7.790921449661255
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 3.128826379776001,
      "step": 3906
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 3906,
      "training_step_time": 9.894312143325806
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 3.1373093128204346,
      "step": 3906
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 3906,
      "training_step_time": 9.90554690361023
    },
    {
      "epoch": 1.490020751953125e-05,
      "model_forward_time": 2.4327056407928467,
      "step": 3906
    },
    {
      "epoch": 1.490020751953125e-05,
      "step": 3906,
      "training_step_time": 10.297888278961182
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 3.1568729877471924,
      "step": 3907
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 3907,
      "training_step_time": 9.94035029411316
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 3.415818214416504,
      "step": 3907
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 3907,
      "training_step_time": 10.181483268737793
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 1.7462754249572754,
      "step": 3907
    },
    {
      "dynamics_loss_avg": 0.054538949951529504,
      "epoch": 1.4904022216796875e-05,
      "step": 3907
    },
    {
      "action_loss_avg": 0.03657987774349749,
      "epoch": 1.4904022216796875e-05,
      "step": 3907
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 3907,
      "training_step_time": 5.209254264831543
    },
    {
      "epoch": 1.4904022216796875e-05,
      "model_forward_time": 2.4296116828918457,
      "step": 3907
    },
    {
      "epoch": 1.4904022216796875e-05,
      "step": 3907,
      "training_step_time": 10.029747724533081
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 2.451172113418579,
      "step": 3908
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 3908,
      "training_step_time": 7.490503549575806
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 3.136178493499756,
      "step": 3908
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 3908,
      "training_step_time": 9.920977115631104
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 3.462122678756714,
      "step": 3908
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 3908,
      "training_step_time": 10.239212036132812
    },
    {
      "epoch": 1.49078369140625e-05,
      "model_forward_time": 3.133568525314331,
      "step": 3908
    },
    {
      "epoch": 1.49078369140625e-05,
      "step": 3908,
      "training_step_time": 10.307338237762451
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 3.153045892715454,
      "step": 3909
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 3909,
      "training_step_time": 9.930322885513306
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 3.135239362716675,
      "step": 3909
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 3909,
      "training_step_time": 9.914060831069946
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 2.430490016937256,
      "step": 3909
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 3909,
      "training_step_time": 7.464240789413452
    },
    {
      "epoch": 1.4911651611328124e-05,
      "model_forward_time": 3.139183759689331,
      "step": 3909
    },
    {
      "epoch": 1.4911651611328124e-05,
      "step": 3909,
      "training_step_time": 10.311264276504517
    },
    {
      "epoch": 1.491546630859375e-05,
      "grad_norm": 0.34661033749580383,
      "learning_rate": 1e-05,
      "loss": 0.0945,
      "step": 3910
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 2.4496285915374756,
      "step": 3910
    },
    {
      "dynamics_loss_avg": 0.06877866797149182,
      "epoch": 1.491546630859375e-05,
      "step": 3910
    },
    {
      "action_loss_avg": 0.03927694680169225,
      "epoch": 1.491546630859375e-05,
      "step": 3910
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 3910,
      "training_step_time": 7.491651296615601
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 2.438505172729492,
      "step": 3910
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 3910,
      "training_step_time": 7.468853235244751
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 2.433866262435913,
      "step": 3910
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 3910,
      "training_step_time": 7.464477062225342
    },
    {
      "epoch": 1.491546630859375e-05,
      "model_forward_time": 2.4360005855560303,
      "step": 3910
    },
    {
      "epoch": 1.491546630859375e-05,
      "step": 3910,
      "training_step_time": 10.251160860061646
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 1.7639906406402588,
      "step": 3911
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 3911,
      "training_step_time": 5.2239320278167725
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 2.7254233360290527,
      "step": 3911
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 3911,
      "training_step_time": 7.7563793659210205
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 3.133460760116577,
      "step": 3911
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 3911,
      "training_step_time": 9.913517236709595
    },
    {
      "epoch": 1.4919281005859375e-05,
      "model_forward_time": 2.430053949356079,
      "step": 3911
    },
    {
      "epoch": 1.4919281005859375e-05,
      "step": 3911,
      "training_step_time": 10.296724081039429
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 3.1398115158081055,
      "step": 3912
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 3912,
      "training_step_time": 9.920291662216187
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 2.434703826904297,
      "step": 3912
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 3912,
      "training_step_time": 7.471239328384399
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 3.4377386569976807,
      "step": 3912
    },
    {
      "dynamics_loss_avg": 0.059170891717076304,
      "epoch": 1.4923095703125e-05,
      "step": 3912
    },
    {
      "action_loss_avg": 0.03667101794853807,
      "epoch": 1.4923095703125e-05,
      "step": 3912
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 3912,
      "training_step_time": 10.213568449020386
    },
    {
      "epoch": 1.4923095703125e-05,
      "model_forward_time": 3.1477439403533936,
      "step": 3912
    },
    {
      "epoch": 1.4923095703125e-05,
      "step": 3912,
      "training_step_time": 10.312379837036133
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 2.4531643390655518,
      "step": 3913
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 3913,
      "training_step_time": 7.489753484725952
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 3.1323556900024414,
      "step": 3913
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 3913,
      "training_step_time": 9.90990138053894
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 1.7431516647338867,
      "step": 3913
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 3913,
      "training_step_time": 5.207251787185669
    },
    {
      "epoch": 1.4926910400390624e-05,
      "model_forward_time": 2.429436445236206,
      "step": 3913
    },
    {
      "epoch": 1.4926910400390624e-05,
      "step": 3913,
      "training_step_time": 10.018391370773315
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 3.4235594272613525,
      "step": 3914
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 3914,
      "training_step_time": 10.203740358352661
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 1.7449510097503662,
      "step": 3914
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 3914,
      "training_step_time": 5.205726146697998
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 2.4312779903411865,
      "step": 3914
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 3914,
      "training_step_time": 7.476248741149902
    },
    {
      "epoch": 1.493072509765625e-05,
      "model_forward_time": 3.135167121887207,
      "step": 3914
    },
    {
      "epoch": 1.493072509765625e-05,
      "step": 3914,
      "training_step_time": 10.294878482818604
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 2.4784188270568848,
      "step": 3915
    },
    {
      "dynamics_loss_avg": 0.057365885749459264,
      "epoch": 1.4934539794921875e-05,
      "step": 3915
    },
    {
      "action_loss_avg": 0.03395335115492344,
      "epoch": 1.4934539794921875e-05,
      "step": 3915
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 3915,
      "training_step_time": 7.516427516937256
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 3.4342143535614014,
      "step": 3915
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 3915,
      "training_step_time": 10.206862211227417
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 3.143486261367798,
      "step": 3915
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 3915,
      "training_step_time": 9.91960096359253
    },
    {
      "epoch": 1.4934539794921875e-05,
      "model_forward_time": 3.138617992401123,
      "step": 3915
    },
    {
      "epoch": 1.4934539794921875e-05,
      "step": 3915,
      "training_step_time": 10.292246341705322
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 3.1500792503356934,
      "step": 3916
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 3916,
      "training_step_time": 9.933823585510254
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 2.4360134601593018,
      "step": 3916
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 3916,
      "training_step_time": 7.468410491943359
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 2.705537796020508,
      "step": 3916
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 3916,
      "training_step_time": 7.73634934425354
    },
    {
      "epoch": 1.49383544921875e-05,
      "model_forward_time": 3.1374542713165283,
      "step": 3916
    },
    {
      "epoch": 1.49383544921875e-05,
      "step": 3916,
      "training_step_time": 10.332859516143799
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 3.1460015773773193,
      "step": 3917
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 3917,
      "training_step_time": 9.924360752105713
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 3.140275478363037,
      "step": 3917
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 3917,
      "training_step_time": 9.91913914680481
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 3.1411991119384766,
      "step": 3917
    },
    {
      "dynamics_loss_avg": 0.06245455406606197,
      "epoch": 1.4942169189453124e-05,
      "step": 3917
    },
    {
      "action_loss_avg": 0.0471435833722353,
      "epoch": 1.4942169189453124e-05,
      "step": 3917
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 3917,
      "training_step_time": 9.927271366119385
    },
    {
      "epoch": 1.4942169189453124e-05,
      "model_forward_time": 2.711897134780884,
      "step": 3917
    },
    {
      "epoch": 1.4942169189453124e-05,
      "step": 3917,
      "training_step_time": 10.31779146194458
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 3.150282859802246,
      "step": 3918
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 3918,
      "training_step_time": 9.932103395462036
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 2.4336140155792236,
      "step": 3918
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 3918,
      "training_step_time": 7.467507600784302
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 2.4386146068573,
      "step": 3918
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 3918,
      "training_step_time": 7.474400520324707
    },
    {
      "epoch": 1.494598388671875e-05,
      "model_forward_time": 3.138917922973633,
      "step": 3918
    },
    {
      "epoch": 1.494598388671875e-05,
      "step": 3918,
      "training_step_time": 10.248195171356201
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 2.449387550354004,
      "step": 3919
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 3919,
      "training_step_time": 7.478903532028198
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 3.4153318405151367,
      "step": 3919
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 3919,
      "training_step_time": 10.185870170593262
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 3.1361308097839355,
      "step": 3919
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 3919,
      "training_step_time": 9.91071343421936
    },
    {
      "epoch": 1.4949798583984375e-05,
      "model_forward_time": 2.4347622394561768,
      "step": 3919
    },
    {
      "epoch": 1.4949798583984375e-05,
      "step": 3919,
      "training_step_time": 10.300341844558716
    },
    {
      "epoch": 1.495361328125e-05,
      "grad_norm": 0.4226701557636261,
      "learning_rate": 1e-05,
      "loss": 0.0939,
      "step": 3920
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 3.159458875656128,
      "step": 3920
    },
    {
      "dynamics_loss_avg": 0.06181218381971121,
      "epoch": 1.495361328125e-05,
      "step": 3920
    },
    {
      "action_loss_avg": 0.035107811586931346,
      "epoch": 1.495361328125e-05,
      "step": 3920
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 3920,
      "training_step_time": 9.943184852600098
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 3.1391258239746094,
      "step": 3920
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 3920,
      "training_step_time": 9.911277532577515
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 2.7435882091522217,
      "step": 3920
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 3920,
      "training_step_time": 7.765779256820679
    },
    {
      "epoch": 1.495361328125e-05,
      "model_forward_time": 2.4438540935516357,
      "step": 3920
    },
    {
      "epoch": 1.495361328125e-05,
      "step": 3920,
      "training_step_time": 10.298920392990112
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 3.1571433544158936,
      "step": 3921
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 3921,
      "training_step_time": 9.92968487739563
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 2.4489200115203857,
      "step": 3921
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 3921,
      "training_step_time": 7.476197957992554
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 3.1367359161376953,
      "step": 3921
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 3921,
      "training_step_time": 9.90459942817688
    },
    {
      "epoch": 1.4957427978515624e-05,
      "model_forward_time": 3.4239704608917236,
      "step": 3921
    },
    {
      "epoch": 1.4957427978515624e-05,
      "step": 3921,
      "training_step_time": 10.29585862159729
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 1.760570764541626,
      "step": 3922
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 3922,
      "training_step_time": 5.219429969787598
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 3.132297992706299,
      "step": 3922
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 3922,
      "training_step_time": 9.89946699142456
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 3.137869358062744,
      "step": 3922
    },
    {
      "dynamics_loss_avg": 0.05564058609306812,
      "epoch": 1.496124267578125e-05,
      "step": 3922
    },
    {
      "action_loss_avg": 0.022163751535117626,
      "epoch": 1.496124267578125e-05,
      "step": 3922
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 3922,
      "training_step_time": 9.915929555892944
    },
    {
      "epoch": 1.496124267578125e-05,
      "model_forward_time": 3.139251947402954,
      "step": 3922
    },
    {
      "epoch": 1.496124267578125e-05,
      "step": 3922,
      "training_step_time": 10.31000828742981
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 3.4769084453582764,
      "step": 3923
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 3923,
      "training_step_time": 10.25729489326477
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 3.1437199115753174,
      "step": 3923
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 3923,
      "training_step_time": 9.92132043838501
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 3.1448299884796143,
      "step": 3923
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 3923,
      "training_step_time": 9.91658353805542
    },
    {
      "epoch": 1.4965057373046875e-05,
      "model_forward_time": 2.435615301132202,
      "step": 3923
    },
    {
      "epoch": 1.4965057373046875e-05,
      "step": 3923,
      "training_step_time": 10.040021657943726
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 3.146389961242676,
      "step": 3924
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 3924,
      "training_step_time": 9.921558380126953
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 2.7097573280334473,
      "step": 3924
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 3924,
      "training_step_time": 7.7423014640808105
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 3.139720916748047,
      "step": 3924
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 3924,
      "training_step_time": 9.911710262298584
    },
    {
      "epoch": 1.49688720703125e-05,
      "model_forward_time": 3.141176700592041,
      "step": 3924
    },
    {
      "epoch": 1.49688720703125e-05,
      "step": 3924,
      "training_step_time": 10.264835834503174
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 2.452796220779419,
      "step": 3925
    },
    {
      "dynamics_loss_avg": 0.04843361284583807,
      "epoch": 1.4972686767578124e-05,
      "step": 3925
    },
    {
      "action_loss_avg": 0.021204882580786944,
      "epoch": 1.4972686767578124e-05,
      "step": 3925
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 3925,
      "training_step_time": 7.488727569580078
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 3.1444504261016846,
      "step": 3925
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 3925,
      "training_step_time": 9.934329748153687
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 3.4413089752197266,
      "step": 3925
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 3925,
      "training_step_time": 10.225545167922974
    },
    {
      "epoch": 1.4972686767578124e-05,
      "model_forward_time": 3.1437807083129883,
      "step": 3925
    },
    {
      "epoch": 1.4972686767578124e-05,
      "step": 3925,
      "training_step_time": 10.326376676559448
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 3.1702487468719482,
      "step": 3926
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 3926,
      "training_step_time": 9.950135469436646
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 2.439120054244995,
      "step": 3926
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 3926,
      "training_step_time": 7.46980357170105
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 2.435699462890625,
      "step": 3926
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 3926,
      "training_step_time": 7.469489097595215
    },
    {
      "epoch": 1.497650146484375e-05,
      "model_forward_time": 3.1433145999908447,
      "step": 3926
    },
    {
      "epoch": 1.497650146484375e-05,
      "step": 3926,
      "training_step_time": 10.332459449768066
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 3.1515774726867676,
      "step": 3927
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 3927,
      "training_step_time": 9.938727140426636
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 3.141815662384033,
      "step": 3927
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 3927,
      "training_step_time": 9.919763803482056
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 2.438829183578491,
      "step": 3927
    },
    {
      "dynamics_loss_avg": 0.06449069753289223,
      "epoch": 1.4980316162109375e-05,
      "step": 3927
    },
    {
      "action_loss_avg": 0.03411983028054237,
      "epoch": 1.4980316162109375e-05,
      "step": 3927
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 3927,
      "training_step_time": 7.4713051319122314
    },
    {
      "epoch": 1.4980316162109375e-05,
      "model_forward_time": 3.1372859477996826,
      "step": 3927
    },
    {
      "epoch": 1.4980316162109375e-05,
      "step": 3927,
      "training_step_time": 10.317341804504395
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 2.444211721420288,
      "step": 3928
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 3928,
      "training_step_time": 7.475306034088135
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 3.411820650100708,
      "step": 3928
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 3928,
      "training_step_time": 10.184330701828003
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 3.1620116233825684,
      "step": 3928
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 3928,
      "training_step_time": 9.943681716918945
    },
    {
      "epoch": 1.4984130859375e-05,
      "model_forward_time": 2.4460465908050537,
      "step": 3928
    },
    {
      "epoch": 1.4984130859375e-05,
      "step": 3928,
      "training_step_time": 10.027230262756348
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 2.448014974594116,
      "step": 3929
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 3929,
      "training_step_time": 7.476349115371704
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 3.130937099456787,
      "step": 3929
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 3929,
      "training_step_time": 9.900020837783813
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 3.385873794555664,
      "step": 3929
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 3929,
      "training_step_time": 10.156400680541992
    },
    {
      "epoch": 1.4987945556640624e-05,
      "model_forward_time": 3.140198230743408,
      "step": 3929
    },
    {
      "epoch": 1.4987945556640624e-05,
      "step": 3929,
      "training_step_time": 10.334793090820312
    },
    {
      "epoch": 1.499176025390625e-05,
      "grad_norm": 0.33825555443763733,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 3930
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 2.4402823448181152,
      "step": 3930
    },
    {
      "dynamics_loss_avg": 0.05025194920599461,
      "epoch": 1.499176025390625e-05,
      "step": 3930
    },
    {
      "action_loss_avg": 0.028538150526583195,
      "epoch": 1.499176025390625e-05,
      "step": 3930
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 3930,
      "training_step_time": 7.4711949825286865
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 3.133366823196411,
      "step": 3930
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 3930,
      "training_step_time": 9.909270763397217
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 2.4325342178344727,
      "step": 3930
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 3930,
      "training_step_time": 7.461603164672852
    },
    {
      "epoch": 1.499176025390625e-05,
      "model_forward_time": 3.4453816413879395,
      "step": 3930
    },
    {
      "epoch": 1.499176025390625e-05,
      "step": 3930,
      "training_step_time": 10.314040899276733
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 3.1474721431732178,
      "step": 3931
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 3931,
      "training_step_time": 9.92039155960083
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 3.1358203887939453,
      "step": 3931
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 3931,
      "training_step_time": 9.915577411651611
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 3.1417367458343506,
      "step": 3931
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 3931,
      "training_step_time": 9.923210144042969
    },
    {
      "epoch": 1.4995574951171875e-05,
      "model_forward_time": 3.1425392627716064,
      "step": 3931
    },
    {
      "epoch": 1.4995574951171875e-05,
      "step": 3931,
      "training_step_time": 10.276087522506714
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 3.43628191947937,
      "step": 3932
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 3932,
      "training_step_time": 10.215328216552734
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 2.437631130218506,
      "step": 3932
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 3932,
      "training_step_time": 7.473531007766724
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 3.1412298679351807,
      "step": 3932
    },
    {
      "dynamics_loss_avg": 0.06762589700520039,
      "epoch": 1.49993896484375e-05,
      "step": 3932
    },
    {
      "action_loss_avg": 0.04321039682254195,
      "epoch": 1.49993896484375e-05,
      "step": 3932
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 3932,
      "training_step_time": 9.92464804649353
    },
    {
      "epoch": 1.49993896484375e-05,
      "model_forward_time": 3.1450679302215576,
      "step": 3932
    },
    {
      "epoch": 1.49993896484375e-05,
      "step": 3932,
      "training_step_time": 10.312355756759644
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 3.15920352935791,
      "step": 3933
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 3933,
      "training_step_time": 9.941211462020874
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 2.732229232788086,
      "step": 3933
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 3933,
      "training_step_time": 7.767228603363037
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 3.136441469192505,
      "step": 3933
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 3933,
      "training_step_time": 9.91497254371643
    },
    {
      "epoch": 1.5003204345703124e-05,
      "model_forward_time": 2.438119888305664,
      "step": 3933
    },
    {
      "epoch": 1.5003204345703124e-05,
      "step": 3933,
      "training_step_time": 10.029871463775635
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 2.4406449794769287,
      "step": 3934
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 3934,
      "training_step_time": 7.475253105163574
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 3.1383650302886963,
      "step": 3934
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 3934,
      "training_step_time": 9.9124436378479
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 3.4212074279785156,
      "step": 3934
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 3934,
      "training_step_time": 10.201063871383667
    },
    {
      "epoch": 1.500701904296875e-05,
      "model_forward_time": 3.1392369270324707,
      "step": 3934
    },
    {
      "epoch": 1.500701904296875e-05,
      "step": 3934,
      "training_step_time": 10.016735076904297
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 2.446352243423462,
      "step": 3935
    },
    {
      "dynamics_loss_avg": 0.058391974121332166,
      "epoch": 1.5010833740234375e-05,
      "step": 3935
    },
    {
      "action_loss_avg": 0.028151760622859002,
      "epoch": 1.5010833740234375e-05,
      "step": 3935
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 3935,
      "training_step_time": 7.481869697570801
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 2.4355194568634033,
      "step": 3935
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 3935,
      "training_step_time": 7.469497203826904
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 3.126582145690918,
      "step": 3935
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 3935,
      "training_step_time": 9.894403457641602
    },
    {
      "epoch": 1.5010833740234375e-05,
      "model_forward_time": 3.3838231563568115,
      "step": 3935
    },
    {
      "epoch": 1.5010833740234375e-05,
      "step": 3935,
      "training_step_time": 10.247968673706055
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 2.4528839588165283,
      "step": 3936
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 3936,
      "training_step_time": 7.4855875968933105
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 2.433448314666748,
      "step": 3936
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 3936,
      "training_step_time": 7.45659613609314
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 2.435814380645752,
      "step": 3936
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 3936,
      "training_step_time": 7.464565277099609
    },
    {
      "epoch": 1.50146484375e-05,
      "model_forward_time": 2.4424855709075928,
      "step": 3936
    },
    {
      "epoch": 1.50146484375e-05,
      "step": 3936,
      "training_step_time": 10.263872861862183
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 3.1494524478912354,
      "step": 3937
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 3937,
      "training_step_time": 9.924684047698975
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 3.4256794452667236,
      "step": 3937
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 3937,
      "training_step_time": 10.20402216911316
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 3.1370348930358887,
      "step": 3937
    },
    {
      "dynamics_loss_avg": 0.06186510706320405,
      "epoch": 1.5018463134765624e-05,
      "step": 3937
    },
    {
      "action_loss_avg": 0.033428100403398275,
      "epoch": 1.5018463134765624e-05,
      "step": 3937
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 3937,
      "training_step_time": 9.92052412033081
    },
    {
      "epoch": 1.5018463134765624e-05,
      "model_forward_time": 3.145886182785034,
      "step": 3937
    },
    {
      "epoch": 1.5018463134765624e-05,
      "step": 3937,
      "training_step_time": 10.32563328742981
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 3.15635085105896,
      "step": 3938
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 3938,
      "training_step_time": 9.94120717048645
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 2.441409111022949,
      "step": 3938
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 3938,
      "training_step_time": 7.478109836578369
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 2.74863338470459,
      "step": 3938
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 3938,
      "training_step_time": 7.777951955795288
    },
    {
      "epoch": 1.502227783203125e-05,
      "model_forward_time": 3.1446282863616943,
      "step": 3938
    },
    {
      "epoch": 1.502227783203125e-05,
      "step": 3938,
      "training_step_time": 10.030418395996094
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 2.450881242752075,
      "step": 3939
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 3939,
      "training_step_time": 7.486718654632568
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 3.1377973556518555,
      "step": 3939
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 3939,
      "training_step_time": 9.913280248641968
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 3.1479501724243164,
      "step": 3939
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 3939,
      "training_step_time": 9.92149829864502
    },
    {
      "epoch": 1.5026092529296875e-05,
      "model_forward_time": 3.4236841201782227,
      "step": 3939
    },
    {
      "epoch": 1.5026092529296875e-05,
      "step": 3939,
      "training_step_time": 10.28845477104187
    },
    {
      "epoch": 1.50299072265625e-05,
      "grad_norm": 0.4483620226383209,
      "learning_rate": 1e-05,
      "loss": 0.0912,
      "step": 3940
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 2.446092367172241,
      "step": 3940
    },
    {
      "dynamics_loss_avg": 0.04862597836181522,
      "epoch": 1.50299072265625e-05,
      "step": 3940
    },
    {
      "action_loss_avg": 0.03001210242509842,
      "epoch": 1.50299072265625e-05,
      "step": 3940
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 3940,
      "training_step_time": 7.477746963500977
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 3.132211446762085,
      "step": 3940
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 3940,
      "training_step_time": 9.90306305885315
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 3.1331326961517334,
      "step": 3940
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 3940,
      "training_step_time": 9.899152040481567
    },
    {
      "epoch": 1.50299072265625e-05,
      "model_forward_time": 3.1307756900787354,
      "step": 3940
    },
    {
      "epoch": 1.50299072265625e-05,
      "step": 3940,
      "training_step_time": 10.22944450378418
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 3.418776512145996,
      "step": 3941
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 3941,
      "training_step_time": 10.193198680877686
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 3.1321613788604736,
      "step": 3941
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 3941,
      "training_step_time": 9.905353307723999
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 3.134645938873291,
      "step": 3941
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 3941,
      "training_step_time": 9.90729284286499
    },
    {
      "epoch": 1.5033721923828124e-05,
      "model_forward_time": 2.433598279953003,
      "step": 3941
    },
    {
      "epoch": 1.5033721923828124e-05,
      "step": 3941,
      "training_step_time": 10.306496143341064
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 2.441718578338623,
      "step": 3942
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 3942,
      "training_step_time": 7.467833757400513
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 3.3965134620666504,
      "step": 3942
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 3942,
      "training_step_time": 10.16485333442688
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 3.1371662616729736,
      "step": 3942
    },
    {
      "dynamics_loss_avg": 0.06617479585111141,
      "epoch": 1.503753662109375e-05,
      "step": 3942
    },
    {
      "action_loss_avg": 0.03507857751101255,
      "epoch": 1.503753662109375e-05,
      "step": 3942
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 3942,
      "training_step_time": 9.90720510482788
    },
    {
      "epoch": 1.503753662109375e-05,
      "model_forward_time": 2.4290883541107178,
      "step": 3942
    },
    {
      "epoch": 1.503753662109375e-05,
      "step": 3942,
      "training_step_time": 10.306983947753906
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 1.7559328079223633,
      "step": 3943
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 3943,
      "training_step_time": 5.216827869415283
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 3.1303036212921143,
      "step": 3943
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 3943,
      "training_step_time": 9.893805503845215
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 3.131091356277466,
      "step": 3943
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 3943,
      "training_step_time": 9.90235185623169
    },
    {
      "epoch": 1.5041351318359375e-05,
      "model_forward_time": 3.389307975769043,
      "step": 3943
    },
    {
      "epoch": 1.5041351318359375e-05,
      "step": 3943,
      "training_step_time": 10.272026777267456
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 3.1555335521698,
      "step": 3944
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 3944,
      "training_step_time": 9.932491302490234
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 2.430769205093384,
      "step": 3944
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 3944,
      "training_step_time": 7.461503982543945
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 2.430995464324951,
      "step": 3944
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 3944,
      "training_step_time": 7.4599220752716064
    },
    {
      "epoch": 1.5045166015625e-05,
      "model_forward_time": 1.7445945739746094,
      "step": 3944
    },
    {
      "epoch": 1.5045166015625e-05,
      "step": 3944,
      "training_step_time": 10.277740240097046
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 3.417907953262329,
      "step": 3945
    },
    {
      "dynamics_loss_avg": 0.08221485745161772,
      "epoch": 1.5048980712890624e-05,
      "step": 3945
    },
    {
      "action_loss_avg": 0.03265797682106495,
      "epoch": 1.5048980712890624e-05,
      "step": 3945
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 3945,
      "training_step_time": 10.200683355331421
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 3.143559455871582,
      "step": 3945
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 3945,
      "training_step_time": 9.92284893989563
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 2.435988187789917,
      "step": 3945
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 3945,
      "training_step_time": 7.465983867645264
    },
    {
      "epoch": 1.5048980712890624e-05,
      "model_forward_time": 2.4311141967773438,
      "step": 3945
    },
    {
      "epoch": 1.5048980712890624e-05,
      "step": 3945,
      "training_step_time": 10.281798839569092
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 3.143554925918579,
      "step": 3946
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 3946,
      "training_step_time": 9.924020051956177
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 3.1303420066833496,
      "step": 3946
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 3946,
      "training_step_time": 9.904685020446777
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 2.6812922954559326,
      "step": 3946
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 3946,
      "training_step_time": 7.713048934936523
    },
    {
      "epoch": 1.505279541015625e-05,
      "model_forward_time": 3.135009527206421,
      "step": 3946
    },
    {
      "epoch": 1.505279541015625e-05,
      "step": 3946,
      "training_step_time": 10.28495168685913
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 3.158240556716919,
      "step": 3947
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 3947,
      "training_step_time": 9.943090438842773
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 2.430516481399536,
      "step": 3947
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 3947,
      "training_step_time": 7.45881199836731
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 2.434211015701294,
      "step": 3947
    },
    {
      "dynamics_loss_avg": 0.07730205021798611,
      "epoch": 1.5056610107421875e-05,
      "step": 3947
    },
    {
      "action_loss_avg": 0.04463634956628084,
      "epoch": 1.5056610107421875e-05,
      "step": 3947
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 3947,
      "training_step_time": 7.4640514850616455
    },
    {
      "epoch": 1.5056610107421875e-05,
      "model_forward_time": 3.410362482070923,
      "step": 3947
    },
    {
      "epoch": 1.5056610107421875e-05,
      "step": 3947,
      "training_step_time": 10.281559944152832
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 3.1536262035369873,
      "step": 3948
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 3948,
      "training_step_time": 9.926903247833252
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 3.1340644359588623,
      "step": 3948
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 3948,
      "training_step_time": 9.90721082687378
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 3.136603593826294,
      "step": 3948
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 3948,
      "training_step_time": 9.914334774017334
    },
    {
      "epoch": 1.50604248046875e-05,
      "model_forward_time": 2.4376940727233887,
      "step": 3948
    },
    {
      "epoch": 1.50604248046875e-05,
      "step": 3948,
      "training_step_time": 10.338554859161377
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 2.712299346923828,
      "step": 3949
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 3949,
      "training_step_time": 7.744279384613037
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 3.13145112991333,
      "step": 3949
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 3949,
      "training_step_time": 9.908891439437866
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 3.140489339828491,
      "step": 3949
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 3949,
      "training_step_time": 9.916155576705933
    },
    {
      "epoch": 1.5064239501953124e-05,
      "model_forward_time": 3.1353554725646973,
      "step": 3949
    },
    {
      "epoch": 1.5064239501953124e-05,
      "step": 3949,
      "training_step_time": 10.031273365020752
    },
    {
      "epoch": 1.506805419921875e-05,
      "grad_norm": 0.36547452211380005,
      "learning_rate": 1e-05,
      "loss": 0.0998,
      "step": 3950
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 3.165893077850342,
      "step": 3950
    },
    {
      "dynamics_loss_avg": 0.06126630380749702,
      "epoch": 1.506805419921875e-05,
      "step": 3950
    },
    {
      "action_loss_avg": 0.031234253896400332,
      "epoch": 1.506805419921875e-05,
      "step": 3950
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 3950,
      "training_step_time": 9.951285600662231
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 3.390535831451416,
      "step": 3950
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 3950,
      "training_step_time": 10.16588830947876
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 3.1354079246520996,
      "step": 3950
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 3950,
      "training_step_time": 9.912638902664185
    },
    {
      "epoch": 1.506805419921875e-05,
      "model_forward_time": 2.437175750732422,
      "step": 3950
    },
    {
      "epoch": 1.506805419921875e-05,
      "step": 3950,
      "training_step_time": 10.314146757125854
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 3.1458048820495605,
      "step": 3951
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 3951,
      "training_step_time": 9.924610137939453
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 1.7446680068969727,
      "step": 3951
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 3951,
      "training_step_time": 5.205522537231445
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 3.127415895462036,
      "step": 3951
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 3951,
      "training_step_time": 9.897300720214844
    },
    {
      "epoch": 1.5071868896484375e-05,
      "model_forward_time": 2.72426700592041,
      "step": 3951
    },
    {
      "epoch": 1.5071868896484375e-05,
      "step": 3951,
      "training_step_time": 10.245468378067017
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 3.151543140411377,
      "step": 3952
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 3952,
      "training_step_time": 9.925204277038574
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 2.4322867393493652,
      "step": 3952
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 3952,
      "training_step_time": 7.462164402008057
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 3.1314456462860107,
      "step": 3952
    },
    {
      "dynamics_loss_avg": 0.06094300895929337,
      "epoch": 1.507568359375e-05,
      "step": 3952
    },
    {
      "action_loss_avg": 0.02479583639651537,
      "epoch": 1.507568359375e-05,
      "step": 3952
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 3952,
      "training_step_time": 9.908548355102539
    },
    {
      "epoch": 1.507568359375e-05,
      "model_forward_time": 2.433685779571533,
      "step": 3952
    },
    {
      "epoch": 1.507568359375e-05,
      "step": 3952,
      "training_step_time": 10.346203327178955
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 3.419339179992676,
      "step": 3953
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 3953,
      "training_step_time": 10.192355871200562
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 3.1346371173858643,
      "step": 3953
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 3953,
      "training_step_time": 9.908228397369385
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 3.134357452392578,
      "step": 3953
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 3953,
      "training_step_time": 9.910214185714722
    },
    {
      "epoch": 1.5079498291015624e-05,
      "model_forward_time": 2.4361188411712646,
      "step": 3953
    },
    {
      "epoch": 1.5079498291015624e-05,
      "step": 3953,
      "training_step_time": 10.02096939086914
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 3.156949281692505,
      "step": 3954
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 3954,
      "training_step_time": 9.938405752182007
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 3.4212687015533447,
      "step": 3954
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 3954,
      "training_step_time": 10.198384761810303
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 1.747185230255127,
      "step": 3954
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 3954,
      "training_step_time": 5.205845832824707
    },
    {
      "epoch": 1.508331298828125e-05,
      "model_forward_time": 3.138518810272217,
      "step": 3954
    },
    {
      "epoch": 1.508331298828125e-05,
      "step": 3954,
      "training_step_time": 10.271429061889648
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 3.149115800857544,
      "step": 3955
    },
    {
      "dynamics_loss_avg": 0.06445810347795486,
      "epoch": 1.5087127685546875e-05,
      "step": 3955
    },
    {
      "action_loss_avg": 0.046525887493044135,
      "epoch": 1.5087127685546875e-05,
      "step": 3955
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 3955,
      "training_step_time": 9.931076765060425
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 2.433939218521118,
      "step": 3955
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 3955,
      "training_step_time": 7.464484930038452
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 2.4332635402679443,
      "step": 3955
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 3955,
      "training_step_time": 7.766814470291138
    },
    {
      "epoch": 1.5087127685546875e-05,
      "model_forward_time": 3.135241746902466,
      "step": 3955
    },
    {
      "epoch": 1.5087127685546875e-05,
      "step": 3955,
      "training_step_time": 10.289999008178711
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 2.444093704223633,
      "step": 3956
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 3956,
      "training_step_time": 7.476449012756348
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 2.4315149784088135,
      "step": 3956
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 3956,
      "training_step_time": 7.46100115776062
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 3.136002540588379,
      "step": 3956
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 3956,
      "training_step_time": 9.905416011810303
    },
    {
      "epoch": 1.50909423828125e-05,
      "model_forward_time": 3.135092258453369,
      "step": 3956
    },
    {
      "epoch": 1.50909423828125e-05,
      "step": 3956,
      "training_step_time": 10.285465955734253
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 3.5120670795440674,
      "step": 3957
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 3957,
      "training_step_time": 10.286882877349854
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 3.1379692554473877,
      "step": 3957
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 3957,
      "training_step_time": 9.908740282058716
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 1.7443482875823975,
      "step": 3957
    },
    {
      "dynamics_loss_avg": 0.06161599829792976,
      "epoch": 1.5094757080078124e-05,
      "step": 3957
    },
    {
      "action_loss_avg": 0.028224868793040515,
      "epoch": 1.5094757080078124e-05,
      "step": 3957
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 3957,
      "training_step_time": 5.210618257522583
    },
    {
      "epoch": 1.5094757080078124e-05,
      "model_forward_time": 2.4335737228393555,
      "step": 3957
    },
    {
      "epoch": 1.5094757080078124e-05,
      "step": 3957,
      "training_step_time": 10.039413452148438
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 2.4473342895507812,
      "step": 3958
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 3958,
      "training_step_time": 7.478922605514526
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 2.7024195194244385,
      "step": 3958
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 3958,
      "training_step_time": 7.730575084686279
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 3.13716197013855,
      "step": 3958
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 3958,
      "training_step_time": 9.90372610092163
    },
    {
      "epoch": 1.509857177734375e-05,
      "model_forward_time": 3.1322450637817383,
      "step": 3958
    },
    {
      "epoch": 1.509857177734375e-05,
      "step": 3958,
      "training_step_time": 10.020080089569092
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 2.451202392578125,
      "step": 3959
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 3959,
      "training_step_time": 7.498099327087402
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 3.134415626525879,
      "step": 3959
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 3959,
      "training_step_time": 9.90241003036499
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 2.742828607559204,
      "step": 3959
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 3959,
      "training_step_time": 7.764532804489136
    },
    {
      "epoch": 1.5102386474609375e-05,
      "model_forward_time": 3.1306910514831543,
      "step": 3959
    },
    {
      "epoch": 1.5102386474609375e-05,
      "step": 3959,
      "training_step_time": 10.277166604995728
    },
    {
      "epoch": 1.5106201171875e-05,
      "grad_norm": 0.44526830315589905,
      "learning_rate": 1e-05,
      "loss": 0.0893,
      "step": 3960
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 3.171039581298828,
      "step": 3960
    },
    {
      "dynamics_loss_avg": 0.051339739747345445,
      "epoch": 1.5106201171875e-05,
      "step": 3960
    },
    {
      "action_loss_avg": 0.024009296111762522,
      "epoch": 1.5106201171875e-05,
      "step": 3960
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 3960,
      "training_step_time": 9.94017744064331
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 3.135586977005005,
      "step": 3960
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 3960,
      "training_step_time": 9.913460969924927
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 1.7466495037078857,
      "step": 3960
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 3960,
      "training_step_time": 5.2026917934417725
    },
    {
      "epoch": 1.5106201171875e-05,
      "model_forward_time": 3.1334922313690186,
      "step": 3960
    },
    {
      "epoch": 1.5106201171875e-05,
      "step": 3960,
      "training_step_time": 10.29421877861023
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 1.7589540481567383,
      "step": 3961
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 3961,
      "training_step_time": 5.212991952896118
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 3.130765199661255,
      "step": 3961
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 3961,
      "training_step_time": 9.891542911529541
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 3.1318461894989014,
      "step": 3961
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 3961,
      "training_step_time": 9.892295122146606
    },
    {
      "epoch": 1.5110015869140624e-05,
      "model_forward_time": 3.1328840255737305,
      "step": 3961
    },
    {
      "epoch": 1.5110015869140624e-05,
      "step": 3961,
      "training_step_time": 10.318653106689453
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 3.1522202491760254,
      "step": 3962
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 3962,
      "training_step_time": 9.917265176773071
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 3.415999174118042,
      "step": 3962
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 3962,
      "training_step_time": 10.179936408996582
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 3.1398229598999023,
      "step": 3962
    },
    {
      "dynamics_loss_avg": 0.04886491037905216,
      "epoch": 1.511383056640625e-05,
      "step": 3962
    },
    {
      "action_loss_avg": 0.026990209706127644,
      "epoch": 1.511383056640625e-05,
      "step": 3962
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 3962,
      "training_step_time": 9.907787322998047
    },
    {
      "epoch": 1.511383056640625e-05,
      "model_forward_time": 2.4375061988830566,
      "step": 3962
    },
    {
      "epoch": 1.511383056640625e-05,
      "step": 3962,
      "training_step_time": 10.282774209976196
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 3.1522960662841797,
      "step": 3963
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 3963,
      "training_step_time": 9.918184757232666
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 2.4345834255218506,
      "step": 3963
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 3963,
      "training_step_time": 7.456973314285278
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 3.3974485397338867,
      "step": 3963
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 3963,
      "training_step_time": 10.159496784210205
    },
    {
      "epoch": 1.5117645263671875e-05,
      "model_forward_time": 2.4403977394104004,
      "step": 3963
    },
    {
      "epoch": 1.5117645263671875e-05,
      "step": 3963,
      "training_step_time": 10.321786880493164
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 2.444085121154785,
      "step": 3964
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 3964,
      "training_step_time": 7.4732887744903564
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 3.133840322494507,
      "step": 3964
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 3964,
      "training_step_time": 9.908207893371582
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 3.1404833793640137,
      "step": 3964
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 3964,
      "training_step_time": 9.916907548904419
    },
    {
      "epoch": 1.51214599609375e-05,
      "model_forward_time": 2.7359302043914795,
      "step": 3964
    },
    {
      "epoch": 1.51214599609375e-05,
      "step": 3964,
      "training_step_time": 10.065115690231323
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 3.1515071392059326,
      "step": 3965
    },
    {
      "dynamics_loss_avg": 0.05748088527470827,
      "epoch": 1.5125274658203124e-05,
      "step": 3965
    },
    {
      "action_loss_avg": 0.029507444240152836,
      "epoch": 1.5125274658203124e-05,
      "step": 3965
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 3965,
      "training_step_time": 9.927229166030884
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 3.1354892253875732,
      "step": 3965
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 3965,
      "training_step_time": 9.910848617553711
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 3.1443724632263184,
      "step": 3965
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 3965,
      "training_step_time": 9.927409410476685
    },
    {
      "epoch": 1.5125274658203124e-05,
      "model_forward_time": 3.142097234725952,
      "step": 3965
    },
    {
      "epoch": 1.5125274658203124e-05,
      "step": 3965,
      "training_step_time": 10.308879613876343
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 2.7288057804107666,
      "step": 3966
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 3966,
      "training_step_time": 7.766357421875
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 3.136545181274414,
      "step": 3966
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 3966,
      "training_step_time": 9.91289234161377
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 3.141638994216919,
      "step": 3966
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 3966,
      "training_step_time": 9.917805433273315
    },
    {
      "epoch": 1.512908935546875e-05,
      "model_forward_time": 3.1341683864593506,
      "step": 3966
    },
    {
      "epoch": 1.512908935546875e-05,
      "step": 3966,
      "training_step_time": 10.294201135635376
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 2.4544179439544678,
      "step": 3967
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 3967,
      "training_step_time": 7.481964826583862
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 3.3939504623413086,
      "step": 3967
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 3967,
      "training_step_time": 10.16228461265564
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 1.7496907711029053,
      "step": 3967
    },
    {
      "dynamics_loss_avg": 0.06738884206861258,
      "epoch": 1.5132904052734375e-05,
      "step": 3967
    },
    {
      "action_loss_avg": 0.0372079738881439,
      "epoch": 1.5132904052734375e-05,
      "step": 3967
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 3967,
      "training_step_time": 5.21232795715332
    },
    {
      "epoch": 1.5132904052734375e-05,
      "model_forward_time": 2.432138442993164,
      "step": 3967
    },
    {
      "epoch": 1.5132904052734375e-05,
      "step": 3967,
      "training_step_time": 10.289278745651245
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 2.4547455310821533,
      "step": 3968
    },
    {
      "epoch": 1.513671875e-05,
      "step": 3968,
      "training_step_time": 7.489580154418945
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 2.430675745010376,
      "step": 3968
    },
    {
      "epoch": 1.513671875e-05,
      "step": 3968,
      "training_step_time": 7.458234548568726
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 1.7408652305603027,
      "step": 3968
    },
    {
      "epoch": 1.513671875e-05,
      "step": 3968,
      "training_step_time": 5.195702075958252
    },
    {
      "epoch": 1.513671875e-05,
      "model_forward_time": 3.1344833374023438,
      "step": 3968
    },
    {
      "epoch": 1.513671875e-05,
      "step": 3968,
      "training_step_time": 10.276575326919556
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 3.137561082839966,
      "step": 3969
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 3969,
      "training_step_time": 9.909403085708618
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 3.135240077972412,
      "step": 3969
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 3969,
      "training_step_time": 9.91215181350708
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 3.1345338821411133,
      "step": 3969
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 3969,
      "training_step_time": 9.910299301147461
    },
    {
      "epoch": 1.5140533447265624e-05,
      "model_forward_time": 2.434499740600586,
      "step": 3969
    },
    {
      "epoch": 1.5140533447265624e-05,
      "step": 3969,
      "training_step_time": 10.269087791442871
    },
    {
      "epoch": 1.514434814453125e-05,
      "grad_norm": 0.7645297050476074,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 3970
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 3.150960683822632,
      "step": 3970
    },
    {
      "dynamics_loss_avg": 0.0627600910142064,
      "epoch": 1.514434814453125e-05,
      "step": 3970
    },
    {
      "action_loss_avg": 0.03135992772877216,
      "epoch": 1.514434814453125e-05,
      "step": 3970
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 3970,
      "training_step_time": 9.940438270568848
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 2.7177252769470215,
      "step": 3970
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 3970,
      "training_step_time": 7.753756761550903
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 2.4419238567352295,
      "step": 3970
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 3970,
      "training_step_time": 7.467881441116333
    },
    {
      "epoch": 1.514434814453125e-05,
      "model_forward_time": 3.142517328262329,
      "step": 3970
    },
    {
      "epoch": 1.514434814453125e-05,
      "step": 3970,
      "training_step_time": 10.324523448944092
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 2.446735382080078,
      "step": 3971
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 3971,
      "training_step_time": 7.476263761520386
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 3.127603054046631,
      "step": 3971
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 3971,
      "training_step_time": 9.898523330688477
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 3.409202814102173,
      "step": 3971
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 3971,
      "training_step_time": 10.179924249649048
    },
    {
      "epoch": 1.5148162841796875e-05,
      "model_forward_time": 2.4306728839874268,
      "step": 3971
    },
    {
      "epoch": 1.5148162841796875e-05,
      "step": 3971,
      "training_step_time": 10.305025577545166
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 3.148160934448242,
      "step": 3972
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 3972,
      "training_step_time": 9.9214448928833
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 3.133110284805298,
      "step": 3972
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 3972,
      "training_step_time": 9.908479690551758
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 3.1374545097351074,
      "step": 3972
    },
    {
      "dynamics_loss_avg": 0.058069700747728346,
      "epoch": 1.51519775390625e-05,
      "step": 3972
    },
    {
      "action_loss_avg": 0.029116850160062314,
      "epoch": 1.51519775390625e-05,
      "step": 3972
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 3972,
      "training_step_time": 9.913227081298828
    },
    {
      "epoch": 1.51519775390625e-05,
      "model_forward_time": 3.449094533920288,
      "step": 3972
    },
    {
      "epoch": 1.51519775390625e-05,
      "step": 3972,
      "training_step_time": 10.314947605133057
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 2.4497902393341064,
      "step": 3973
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 3973,
      "training_step_time": 7.48747181892395
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 3.136981725692749,
      "step": 3973
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 3973,
      "training_step_time": 9.912254571914673
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 3.135371685028076,
      "step": 3973
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 3973,
      "training_step_time": 9.908630132675171
    },
    {
      "epoch": 1.5155792236328124e-05,
      "model_forward_time": 2.431581497192383,
      "step": 3973
    },
    {
      "epoch": 1.5155792236328124e-05,
      "step": 3973,
      "training_step_time": 10.32100796699524
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 2.0688836574554443,
      "step": 3974
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 3974,
      "training_step_time": 5.526037931442261
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 2.434433937072754,
      "step": 3974
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 3974,
      "training_step_time": 7.467137813568115
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 3.136521339416504,
      "step": 3974
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 3974,
      "training_step_time": 9.91234302520752
    },
    {
      "epoch": 1.515960693359375e-05,
      "model_forward_time": 3.1348721981048584,
      "step": 3974
    },
    {
      "epoch": 1.515960693359375e-05,
      "step": 3974,
      "training_step_time": 10.290121078491211
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 2.454299211502075,
      "step": 3975
    },
    {
      "dynamics_loss_avg": 0.0707393754273653,
      "epoch": 1.5163421630859375e-05,
      "step": 3975
    },
    {
      "action_loss_avg": 0.03429566500708461,
      "epoch": 1.5163421630859375e-05,
      "step": 3975
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 3975,
      "training_step_time": 7.489385366439819
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 3.4332942962646484,
      "step": 3975
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 3975,
      "training_step_time": 10.211013078689575
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 1.7423415184020996,
      "step": 3975
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 3975,
      "training_step_time": 5.202299118041992
    },
    {
      "epoch": 1.5163421630859375e-05,
      "model_forward_time": 2.436272144317627,
      "step": 3975
    },
    {
      "epoch": 1.5163421630859375e-05,
      "step": 3975,
      "training_step_time": 10.024038791656494
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 2.444258689880371,
      "step": 3976
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 3976,
      "training_step_time": 7.475894927978516
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 2.4361729621887207,
      "step": 3976
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 3976,
      "training_step_time": 7.467799425125122
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 3.1274683475494385,
      "step": 3976
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 3976,
      "training_step_time": 9.903613328933716
    },
    {
      "epoch": 1.5167236328125e-05,
      "model_forward_time": 2.697514295578003,
      "step": 3976
    },
    {
      "epoch": 1.5167236328125e-05,
      "step": 3976,
      "training_step_time": 10.31980276107788
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 1.7608635425567627,
      "step": 3977
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 3977,
      "training_step_time": 5.214575290679932
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 2.4300460815429688,
      "step": 3977
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 3977,
      "training_step_time": 7.4558610916137695
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 1.7394466400146484,
      "step": 3977
    },
    {
      "dynamics_loss_avg": 0.06531646512448788,
      "epoch": 1.5171051025390624e-05,
      "step": 3977
    },
    {
      "action_loss_avg": 0.03675547866150737,
      "epoch": 1.5171051025390624e-05,
      "step": 3977
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 3977,
      "training_step_time": 5.195655107498169
    },
    {
      "epoch": 1.5171051025390624e-05,
      "model_forward_time": 3.131937265396118,
      "step": 3977
    },
    {
      "epoch": 1.5171051025390624e-05,
      "step": 3977,
      "training_step_time": 10.244691610336304
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 3.1417131423950195,
      "step": 3978
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 3978,
      "training_step_time": 9.912903070449829
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 3.4046895503997803,
      "step": 3978
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 3978,
      "training_step_time": 10.176357984542847
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 3.134570360183716,
      "step": 3978
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 3978,
      "training_step_time": 9.921615600585938
    },
    {
      "epoch": 1.517486572265625e-05,
      "model_forward_time": 3.139031410217285,
      "step": 3978
    },
    {
      "epoch": 1.517486572265625e-05,
      "step": 3978,
      "training_step_time": 10.295557260513306
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 2.448003053665161,
      "step": 3979
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 3979,
      "training_step_time": 7.481479644775391
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 2.4379770755767822,
      "step": 3979
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 3979,
      "training_step_time": 7.470907688140869
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 3.4215312004089355,
      "step": 3979
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 3979,
      "training_step_time": 10.195729732513428
    },
    {
      "epoch": 1.5178680419921875e-05,
      "model_forward_time": 2.4364123344421387,
      "step": 3979
    },
    {
      "epoch": 1.5178680419921875e-05,
      "step": 3979,
      "training_step_time": 10.244600296020508
    },
    {
      "epoch": 1.51824951171875e-05,
      "grad_norm": 0.4225694537162781,
      "learning_rate": 1e-05,
      "loss": 0.0954,
      "step": 3980
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 2.454045295715332,
      "step": 3980
    },
    {
      "dynamics_loss_avg": 0.05724283736199141,
      "epoch": 1.51824951171875e-05,
      "step": 3980
    },
    {
      "action_loss_avg": 0.029327662754803897,
      "epoch": 1.51824951171875e-05,
      "step": 3980
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 3980,
      "training_step_time": 7.490566253662109
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 2.4302775859832764,
      "step": 3980
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 3980,
      "training_step_time": 7.469916582107544
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 2.43721342086792,
      "step": 3980
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 3980,
      "training_step_time": 7.469545364379883
    },
    {
      "epoch": 1.51824951171875e-05,
      "model_forward_time": 2.4309234619140625,
      "step": 3980
    },
    {
      "epoch": 1.51824951171875e-05,
      "step": 3980,
      "training_step_time": 10.298694849014282
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 2.4464986324310303,
      "step": 3981
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 3981,
      "training_step_time": 7.479637145996094
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 3.1346473693847656,
      "step": 3981
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 3981,
      "training_step_time": 9.914249420166016
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 2.4328720569610596,
      "step": 3981
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 3981,
      "training_step_time": 7.468217849731445
    },
    {
      "epoch": 1.5186309814453125e-05,
      "model_forward_time": 3.1395316123962402,
      "step": 3981
    },
    {
      "epoch": 1.5186309814453125e-05,
      "step": 3981,
      "training_step_time": 10.325846195220947
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 2.4503886699676514,
      "step": 3982
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 3982,
      "training_step_time": 7.500129699707031
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 3.428840160369873,
      "step": 3982
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 3982,
      "training_step_time": 10.207558155059814
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 2.4438183307647705,
      "step": 3982
    },
    {
      "dynamics_loss_avg": 0.08171652164310217,
      "epoch": 1.519012451171875e-05,
      "step": 3982
    },
    {
      "action_loss_avg": 0.03424795297905803,
      "epoch": 1.519012451171875e-05,
      "step": 3982
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 3982,
      "training_step_time": 7.517099857330322
    },
    {
      "epoch": 1.519012451171875e-05,
      "model_forward_time": 2.432482957839966,
      "step": 3982
    },
    {
      "epoch": 1.519012451171875e-05,
      "step": 3982,
      "training_step_time": 10.307215213775635
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 3.144181251525879,
      "step": 3983
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 3983,
      "training_step_time": 9.92340874671936
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 2.4368393421173096,
      "step": 3983
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 3983,
      "training_step_time": 7.464929103851318
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 2.7505292892456055,
      "step": 3983
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 3983,
      "training_step_time": 7.774762153625488
    },
    {
      "epoch": 1.5193939208984375e-05,
      "model_forward_time": 3.129276752471924,
      "step": 3983
    },
    {
      "epoch": 1.5193939208984375e-05,
      "step": 3983,
      "training_step_time": 10.019255876541138
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 2.449610710144043,
      "step": 3984
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 3984,
      "training_step_time": 7.48170804977417
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 3.1335315704345703,
      "step": 3984
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 3984,
      "training_step_time": 9.901137351989746
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 3.1276326179504395,
      "step": 3984
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 3984,
      "training_step_time": 9.895702362060547
    },
    {
      "epoch": 1.519775390625e-05,
      "model_forward_time": 2.0494961738586426,
      "step": 3984
    },
    {
      "epoch": 1.519775390625e-05,
      "step": 3984,
      "training_step_time": 10.014993667602539
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 3.1456515789031982,
      "step": 3985
    },
    {
      "dynamics_loss_avg": 0.05247143469750881,
      "epoch": 1.5201568603515625e-05,
      "step": 3985
    },
    {
      "action_loss_avg": 0.024690912757068872,
      "epoch": 1.5201568603515625e-05,
      "step": 3985
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 3985,
      "training_step_time": 9.921098232269287
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 2.429654359817505,
      "step": 3985
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 3985,
      "training_step_time": 7.460849285125732
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 3.1262590885162354,
      "step": 3985
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 3985,
      "training_step_time": 9.90207028388977
    },
    {
      "epoch": 1.5201568603515625e-05,
      "model_forward_time": 2.4323315620422363,
      "step": 3985
    },
    {
      "epoch": 1.5201568603515625e-05,
      "step": 3985,
      "training_step_time": 10.326303720474243
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 3.142241954803467,
      "step": 3986
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 3986,
      "training_step_time": 9.920487403869629
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 2.728466033935547,
      "step": 3986
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 3986,
      "training_step_time": 7.761573314666748
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 3.134396553039551,
      "step": 3986
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 3986,
      "training_step_time": 9.914743661880493
    },
    {
      "epoch": 1.520538330078125e-05,
      "model_forward_time": 3.135932445526123,
      "step": 3986
    },
    {
      "epoch": 1.520538330078125e-05,
      "step": 3986,
      "training_step_time": 10.33262300491333
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 3.1671676635742188,
      "step": 3987
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 3987,
      "training_step_time": 9.953854322433472
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 3.1392569541931152,
      "step": 3987
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 3987,
      "training_step_time": 10.209118127822876
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 1.7492468357086182,
      "step": 3987
    },
    {
      "dynamics_loss_avg": 0.0574789896607399,
      "epoch": 1.5209197998046875e-05,
      "step": 3987
    },
    {
      "action_loss_avg": 0.027840855391696097,
      "epoch": 1.5209197998046875e-05,
      "step": 3987
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 3987,
      "training_step_time": 5.211057901382446
    },
    {
      "epoch": 1.5209197998046875e-05,
      "model_forward_time": 3.140066385269165,
      "step": 3987
    },
    {
      "epoch": 1.5209197998046875e-05,
      "step": 3987,
      "training_step_time": 10.293072938919067
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 3.151121139526367,
      "step": 3988
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 3988,
      "training_step_time": 9.929068565368652
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 3.138016939163208,
      "step": 3988
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 3988,
      "training_step_time": 9.91442608833313
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 2.4350733757019043,
      "step": 3988
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 3988,
      "training_step_time": 7.468330144882202
    },
    {
      "epoch": 1.52130126953125e-05,
      "model_forward_time": 3.412020206451416,
      "step": 3988
    },
    {
      "epoch": 1.52130126953125e-05,
      "step": 3988,
      "training_step_time": 10.285276412963867
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 1.766657829284668,
      "step": 3989
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 3989,
      "training_step_time": 5.228352308273315
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 2.437868118286133,
      "step": 3989
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 3989,
      "training_step_time": 7.472479581832886
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 3.1433589458465576,
      "step": 3989
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 3989,
      "training_step_time": 9.921929359436035
    },
    {
      "epoch": 1.5216827392578125e-05,
      "model_forward_time": 3.1409409046173096,
      "step": 3989
    },
    {
      "epoch": 1.5216827392578125e-05,
      "step": 3989,
      "training_step_time": 10.040800333023071
    },
    {
      "epoch": 1.522064208984375e-05,
      "grad_norm": 0.5789005756378174,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 3990
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 2.762907028198242,
      "step": 3990
    },
    {
      "dynamics_loss_avg": 0.07165681943297386,
      "epoch": 1.522064208984375e-05,
      "step": 3990
    },
    {
      "action_loss_avg": 0.039213348925113675,
      "epoch": 1.522064208984375e-05,
      "step": 3990
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 3990,
      "training_step_time": 7.815888166427612
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 2.4368085861206055,
      "step": 3990
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 3990,
      "training_step_time": 7.468651294708252
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 2.43544864654541,
      "step": 3990
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 3990,
      "training_step_time": 7.467345952987671
    },
    {
      "epoch": 1.522064208984375e-05,
      "model_forward_time": 2.4335949420928955,
      "step": 3990
    },
    {
      "epoch": 1.522064208984375e-05,
      "step": 3990,
      "training_step_time": 10.238636255264282
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 3.1612260341644287,
      "step": 3991
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 3991,
      "training_step_time": 9.939110279083252
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 2.4548630714416504,
      "step": 3991
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 3991,
      "training_step_time": 7.760274171829224
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 3.143507719039917,
      "step": 3991
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 3991,
      "training_step_time": 9.922358512878418
    },
    {
      "epoch": 1.5224456787109375e-05,
      "model_forward_time": 2.442542314529419,
      "step": 3991
    },
    {
      "epoch": 1.5224456787109375e-05,
      "step": 3991,
      "training_step_time": 10.335474252700806
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 3.1603009700775146,
      "step": 3992
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 3992,
      "training_step_time": 9.94535493850708
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 3.1441550254821777,
      "step": 3992
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 3992,
      "training_step_time": 9.918124437332153
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 3.1432104110717773,
      "step": 3992
    },
    {
      "dynamics_loss_avg": 0.06059519089758396,
      "epoch": 1.5228271484375e-05,
      "step": 3992
    },
    {
      "action_loss_avg": 0.029156218050047755,
      "epoch": 1.5228271484375e-05,
      "step": 3992
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 3992,
      "training_step_time": 10.185078859329224
    },
    {
      "epoch": 1.5228271484375e-05,
      "model_forward_time": 2.440006971359253,
      "step": 3992
    },
    {
      "epoch": 1.5228271484375e-05,
      "step": 3992,
      "training_step_time": 10.29911494255066
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 3.1709861755371094,
      "step": 3993
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 3993,
      "training_step_time": 9.966293811798096
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 2.439394950866699,
      "step": 3993
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 3993,
      "training_step_time": 7.469467639923096
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 2.4310202598571777,
      "step": 3993
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 3993,
      "training_step_time": 7.4573655128479
    },
    {
      "epoch": 1.5232086181640625e-05,
      "model_forward_time": 2.4268980026245117,
      "step": 3993
    },
    {
      "epoch": 1.5232086181640625e-05,
      "step": 3993,
      "training_step_time": 10.317019939422607
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 2.798473358154297,
      "step": 3994
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 3994,
      "training_step_time": 7.8260321617126465
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 1.7461762428283691,
      "step": 3994
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 3994,
      "training_step_time": 5.207042455673218
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 3.144857406616211,
      "step": 3994
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 3994,
      "training_step_time": 9.915261507034302
    },
    {
      "epoch": 1.523590087890625e-05,
      "model_forward_time": 2.432318687438965,
      "step": 3994
    },
    {
      "epoch": 1.523590087890625e-05,
      "step": 3994,
      "training_step_time": 10.025281429290771
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 1.763575792312622,
      "step": 3995
    },
    {
      "dynamics_loss_avg": 0.04627022501081228,
      "epoch": 1.5239715576171875e-05,
      "step": 3995
    },
    {
      "action_loss_avg": 0.02604534011334181,
      "epoch": 1.5239715576171875e-05,
      "step": 3995
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 3995,
      "training_step_time": 5.22527003288269
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 2.4323408603668213,
      "step": 3995
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 3995,
      "training_step_time": 7.456115961074829
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 3.4237594604492188,
      "step": 3995
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 3995,
      "training_step_time": 10.199562549591064
    },
    {
      "epoch": 1.5239715576171875e-05,
      "model_forward_time": 2.4308393001556396,
      "step": 3995
    },
    {
      "epoch": 1.5239715576171875e-05,
      "step": 3995,
      "training_step_time": 10.01407766342163
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 3.1425588130950928,
      "step": 3996
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 3996,
      "training_step_time": 9.912560939788818
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 3.134375810623169,
      "step": 3996
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 3996,
      "training_step_time": 9.904970169067383
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 3.137383222579956,
      "step": 3996
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 3996,
      "training_step_time": 9.908483028411865
    },
    {
      "epoch": 1.52435302734375e-05,
      "model_forward_time": 3.4219682216644287,
      "step": 3996
    },
    {
      "epoch": 1.52435302734375e-05,
      "step": 3996,
      "training_step_time": 10.307464599609375
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 3.150538206100464,
      "step": 3997
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 3997,
      "training_step_time": 9.923659086227417
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 3.1333322525024414,
      "step": 3997
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 3997,
      "training_step_time": 9.904532432556152
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 3.1429948806762695,
      "step": 3997
    },
    {
      "dynamics_loss_avg": 0.08874379359185695,
      "epoch": 1.5247344970703125e-05,
      "step": 3997
    },
    {
      "action_loss_avg": 0.03867394421249628,
      "epoch": 1.5247344970703125e-05,
      "step": 3997
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 3997,
      "training_step_time": 9.923853158950806
    },
    {
      "epoch": 1.5247344970703125e-05,
      "model_forward_time": 2.4343795776367188,
      "step": 3997
    },
    {
      "epoch": 1.5247344970703125e-05,
      "step": 3997,
      "training_step_time": 10.272997379302979
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 2.445251941680908,
      "step": 3998
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 3998,
      "training_step_time": 7.470976829528809
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 3.130039930343628,
      "step": 3998
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 3998,
      "training_step_time": 9.896228551864624
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 3.137850761413574,
      "step": 3998
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 3998,
      "training_step_time": 9.905980348587036
    },
    {
      "epoch": 1.525115966796875e-05,
      "model_forward_time": 2.434920310974121,
      "step": 3998
    },
    {
      "epoch": 1.525115966796875e-05,
      "step": 3998,
      "training_step_time": 10.252466440200806
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 2.4812161922454834,
      "step": 3999
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 3999,
      "training_step_time": 7.512275218963623
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 3.41507887840271,
      "step": 3999
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 3999,
      "training_step_time": 10.179826259613037
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 2.4341602325439453,
      "step": 3999
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 3999,
      "training_step_time": 7.461730241775513
    },
    {
      "epoch": 1.5254974365234375e-05,
      "model_forward_time": 2.431192636489868,
      "step": 3999
    },
    {
      "epoch": 1.5254974365234375e-05,
      "step": 3999,
      "training_step_time": 10.306935787200928
    },
    {
      "epoch": 1.52587890625e-05,
      "grad_norm": 0.530875563621521,
      "learning_rate": 1e-05,
      "loss": 0.0947,
      "step": 4000
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 3.163917303085327,
      "step": 4000
    },
    {
      "dynamics_loss_avg": 0.04988016802817583,
      "epoch": 1.52587890625e-05,
      "step": 4000
    },
    {
      "action_loss_avg": 0.021001918218098582,
      "epoch": 1.52587890625e-05,
      "step": 4000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 4000,
      "training_step_time": 9.929754257202148
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 5.267695903778076,
      "step": 4000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 4000,
      "training_step_time": 12.023623943328857
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 3.131671667098999,
      "step": 4000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 4000,
      "training_step_time": 9.894329071044922
    },
    {
      "epoch": 1.52587890625e-05,
      "model_forward_time": 3.1266796588897705,
      "step": 4000
    },
    {
      "epoch": 1.52587890625e-05,
      "step": 4000,
      "training_step_time": 10.315723657608032
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 3.1382334232330322,
      "step": 4001
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 4001,
      "training_step_time": 9.912612438201904
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 3.1469335556030273,
      "step": 4001
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 4001,
      "training_step_time": 9.922497987747192
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 3.469541072845459,
      "step": 4001
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 4001,
      "training_step_time": 10.241891860961914
    },
    {
      "epoch": 1.5262603759765626e-05,
      "model_forward_time": 3.1371166706085205,
      "step": 4001
    },
    {
      "epoch": 1.5262603759765626e-05,
      "step": 4001,
      "training_step_time": 10.282575368881226
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 3.22334361076355,
      "step": 4002
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 4002,
      "training_step_time": 10.019391536712646
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 2.4330830574035645,
      "step": 4002
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 4002,
      "training_step_time": 7.484708547592163
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 3.148007869720459,
      "step": 4002
    },
    {
      "dynamics_loss_avg": 0.06959165297448636,
      "epoch": 1.526641845703125e-05,
      "step": 4002
    },
    {
      "action_loss_avg": 0.03212300138548017,
      "epoch": 1.526641845703125e-05,
      "step": 4002
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 4002,
      "training_step_time": 9.990873336791992
    },
    {
      "epoch": 1.526641845703125e-05,
      "model_forward_time": 3.466200828552246,
      "step": 4002
    },
    {
      "epoch": 1.526641845703125e-05,
      "step": 4002,
      "training_step_time": 10.340447187423706
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 3.1605000495910645,
      "step": 4003
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 4003,
      "training_step_time": 9.944140434265137
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 3.13806414604187,
      "step": 4003
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 4003,
      "training_step_time": 9.916150093078613
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 3.139643669128418,
      "step": 4003
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 4003,
      "training_step_time": 9.929016590118408
    },
    {
      "epoch": 1.5270233154296875e-05,
      "model_forward_time": 3.147596597671509,
      "step": 4003
    },
    {
      "epoch": 1.5270233154296875e-05,
      "step": 4003,
      "training_step_time": 10.263996362686157
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 3.1584832668304443,
      "step": 4004
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 4004,
      "training_step_time": 10.25049614906311
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 3.1407456398010254,
      "step": 4004
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 4004,
      "training_step_time": 9.919906854629517
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 3.1353964805603027,
      "step": 4004
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 4004,
      "training_step_time": 9.912180185317993
    },
    {
      "epoch": 1.52740478515625e-05,
      "model_forward_time": 3.140465259552002,
      "step": 4004
    },
    {
      "epoch": 1.52740478515625e-05,
      "step": 4004,
      "training_step_time": 10.044259309768677
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 3.149949312210083,
      "step": 4005
    },
    {
      "dynamics_loss_avg": 0.06877200044691563,
      "epoch": 1.5277862548828125e-05,
      "step": 4005
    },
    {
      "action_loss_avg": 0.03211633497849107,
      "epoch": 1.5277862548828125e-05,
      "step": 4005
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 4005,
      "training_step_time": 9.941265344619751
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 3.4573802947998047,
      "step": 4005
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 4005,
      "training_step_time": 10.233752489089966
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 3.1352343559265137,
      "step": 4005
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 4005,
      "training_step_time": 9.905855178833008
    },
    {
      "epoch": 1.5277862548828125e-05,
      "model_forward_time": 3.138111114501953,
      "step": 4005
    },
    {
      "epoch": 1.5277862548828125e-05,
      "step": 4005,
      "training_step_time": 10.277674198150635
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 3.1596808433532715,
      "step": 4006
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 4006,
      "training_step_time": 9.944766759872437
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 2.4434025287628174,
      "step": 4006
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 4006,
      "training_step_time": 7.478478193283081
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 3.1445746421813965,
      "step": 4006
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 4006,
      "training_step_time": 10.237635374069214
    },
    {
      "epoch": 1.528167724609375e-05,
      "model_forward_time": 3.1467478275299072,
      "step": 4006
    },
    {
      "epoch": 1.528167724609375e-05,
      "step": 4006,
      "training_step_time": 10.32915711402893
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 3.164874315261841,
      "step": 4007
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 4007,
      "training_step_time": 9.939830780029297
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 3.1395328044891357,
      "step": 4007
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 4007,
      "training_step_time": 9.913861513137817
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 3.1503777503967285,
      "step": 4007
    },
    {
      "dynamics_loss_avg": 0.0653722470626235,
      "epoch": 1.5285491943359374e-05,
      "step": 4007
    },
    {
      "action_loss_avg": 0.029604786075651646,
      "epoch": 1.5285491943359374e-05,
      "step": 4007
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 4007,
      "training_step_time": 9.929283618927002
    },
    {
      "epoch": 1.5285491943359374e-05,
      "model_forward_time": 3.4246370792388916,
      "step": 4007
    },
    {
      "epoch": 1.5285491943359374e-05,
      "step": 4007,
      "training_step_time": 10.316040277481079
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 3.161268472671509,
      "step": 4008
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 4008,
      "training_step_time": 9.940331935882568
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 3.143073797225952,
      "step": 4008
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 4008,
      "training_step_time": 9.911659955978394
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 3.134223222732544,
      "step": 4008
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 4008,
      "training_step_time": 9.908690452575684
    },
    {
      "epoch": 1.5289306640625e-05,
      "model_forward_time": 3.136793613433838,
      "step": 4008
    },
    {
      "epoch": 1.5289306640625e-05,
      "step": 4008,
      "training_step_time": 10.327006340026855
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 3.4304797649383545,
      "step": 4009
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 4009,
      "training_step_time": 10.205929279327393
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 3.1410725116729736,
      "step": 4009
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 4009,
      "training_step_time": 9.91594672203064
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 3.1449332237243652,
      "step": 4009
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 4009,
      "training_step_time": 9.922578811645508
    },
    {
      "epoch": 1.5293121337890626e-05,
      "model_forward_time": 3.1410887241363525,
      "step": 4009
    },
    {
      "epoch": 1.5293121337890626e-05,
      "step": 4009,
      "training_step_time": 10.035619258880615
    },
    {
      "epoch": 1.529693603515625e-05,
      "grad_norm": 0.6599653959274292,
      "learning_rate": 1e-05,
      "loss": 0.0938,
      "step": 4010
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 3.154463529586792,
      "step": 4010
    },
    {
      "dynamics_loss_avg": 0.05981703344732523,
      "epoch": 1.529693603515625e-05,
      "step": 4010
    },
    {
      "action_loss_avg": 0.02794564124196768,
      "epoch": 1.529693603515625e-05,
      "step": 4010
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 4010,
      "training_step_time": 9.942263841629028
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 3.4479050636291504,
      "step": 4010
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 4010,
      "training_step_time": 10.234849452972412
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 3.156249523162842,
      "step": 4010
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 4010,
      "training_step_time": 9.940161943435669
    },
    {
      "epoch": 1.529693603515625e-05,
      "model_forward_time": 3.1456851959228516,
      "step": 4010
    },
    {
      "epoch": 1.529693603515625e-05,
      "step": 4010,
      "training_step_time": 10.305020809173584
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 3.163623094558716,
      "step": 4011
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 4011,
      "training_step_time": 9.961715936660767
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 3.1462550163269043,
      "step": 4011
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 4011,
      "training_step_time": 9.933381795883179
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 3.4284512996673584,
      "step": 4011
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 4011,
      "training_step_time": 10.217285394668579
    },
    {
      "epoch": 1.5300750732421875e-05,
      "model_forward_time": 3.1528189182281494,
      "step": 4011
    },
    {
      "epoch": 1.5300750732421875e-05,
      "step": 4011,
      "training_step_time": 10.303092241287231
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 3.1568212509155273,
      "step": 4012
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 4012,
      "training_step_time": 9.952378511428833
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 3.1507673263549805,
      "step": 4012
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 4012,
      "training_step_time": 9.941118478775024
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 3.148829936981201,
      "step": 4012
    },
    {
      "dynamics_loss_avg": 0.07225084975361824,
      "epoch": 1.53045654296875e-05,
      "step": 4012
    },
    {
      "action_loss_avg": 0.026518424227833747,
      "epoch": 1.53045654296875e-05,
      "step": 4012
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 4012,
      "training_step_time": 9.939202070236206
    },
    {
      "epoch": 1.53045654296875e-05,
      "model_forward_time": 3.4645299911499023,
      "step": 4012
    },
    {
      "epoch": 1.53045654296875e-05,
      "step": 4012,
      "training_step_time": 10.335452795028687
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 3.1599647998809814,
      "step": 4013
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 4013,
      "training_step_time": 9.939171314239502
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 3.1469790935516357,
      "step": 4013
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 4013,
      "training_step_time": 9.921475648880005
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 3.1436469554901123,
      "step": 4013
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 4013,
      "training_step_time": 9.916479587554932
    },
    {
      "epoch": 1.5308380126953125e-05,
      "model_forward_time": 3.1447465419769287,
      "step": 4013
    },
    {
      "epoch": 1.5308380126953125e-05,
      "step": 4013,
      "training_step_time": 10.306339502334595
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 3.448305606842041,
      "step": 4014
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 4014,
      "training_step_time": 10.226072311401367
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 3.1449625492095947,
      "step": 4014
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 4014,
      "training_step_time": 9.915059566497803
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 3.137207269668579,
      "step": 4014
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 4014,
      "training_step_time": 9.908281087875366
    },
    {
      "epoch": 1.531219482421875e-05,
      "model_forward_time": 3.1409220695495605,
      "step": 4014
    },
    {
      "epoch": 1.531219482421875e-05,
      "step": 4014,
      "training_step_time": 10.34182596206665
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 3.15729022026062,
      "step": 4015
    },
    {
      "dynamics_loss_avg": 0.06199326701462269,
      "epoch": 1.5316009521484374e-05,
      "step": 4015
    },
    {
      "action_loss_avg": 0.026394461374729873,
      "epoch": 1.5316009521484374e-05,
      "step": 4015
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 4015,
      "training_step_time": 9.935681819915771
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 2.4309494495391846,
      "step": 4015
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 4015,
      "training_step_time": 7.729727745056152
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 3.1377341747283936,
      "step": 4015
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 4015,
      "training_step_time": 9.903058767318726
    },
    {
      "epoch": 1.5316009521484374e-05,
      "model_forward_time": 3.1425180435180664,
      "step": 4015
    },
    {
      "epoch": 1.5316009521484374e-05,
      "step": 4015,
      "training_step_time": 10.023978471755981
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 3.1474666595458984,
      "step": 4016
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 4016,
      "training_step_time": 9.921637058258057
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 3.1337356567382812,
      "step": 4016
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 4016,
      "training_step_time": 9.90524697303772
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 3.4137392044067383,
      "step": 4016
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 4016,
      "training_step_time": 10.189123392105103
    },
    {
      "epoch": 1.531982421875e-05,
      "model_forward_time": 3.1455564498901367,
      "step": 4016
    },
    {
      "epoch": 1.531982421875e-05,
      "step": 4016,
      "training_step_time": 10.260005474090576
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 3.1545112133026123,
      "step": 4017
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 4017,
      "training_step_time": 9.927182674407959
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 3.1349611282348633,
      "step": 4017
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 4017,
      "training_step_time": 9.923022747039795
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 3.1437954902648926,
      "step": 4017
    },
    {
      "dynamics_loss_avg": 0.06659573949873447,
      "epoch": 1.5323638916015626e-05,
      "step": 4017
    },
    {
      "action_loss_avg": 0.02740236334502697,
      "epoch": 1.5323638916015626e-05,
      "step": 4017
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 4017,
      "training_step_time": 9.920324563980103
    },
    {
      "epoch": 1.5323638916015626e-05,
      "model_forward_time": 3.4695069789886475,
      "step": 4017
    },
    {
      "epoch": 1.5323638916015626e-05,
      "step": 4017,
      "training_step_time": 10.348620176315308
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 3.156973123550415,
      "step": 4018
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 4018,
      "training_step_time": 9.933127641677856
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 3.1441025733947754,
      "step": 4018
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 4018,
      "training_step_time": 9.925051212310791
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 2.443150758743286,
      "step": 4018
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 4018,
      "training_step_time": 7.475059509277344
    },
    {
      "epoch": 1.532745361328125e-05,
      "model_forward_time": 3.143557548522949,
      "step": 4018
    },
    {
      "epoch": 1.532745361328125e-05,
      "step": 4018,
      "training_step_time": 10.052622556686401
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 3.161564588546753,
      "step": 4019
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 4019,
      "training_step_time": 10.22827672958374
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 3.149868965148926,
      "step": 4019
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 4019,
      "training_step_time": 9.9260995388031
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 3.1539664268493652,
      "step": 4019
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 4019,
      "training_step_time": 9.931390285491943
    },
    {
      "epoch": 1.5331268310546875e-05,
      "model_forward_time": 3.151367425918579,
      "step": 4019
    },
    {
      "epoch": 1.5331268310546875e-05,
      "step": 4019,
      "training_step_time": 10.341835498809814
    },
    {
      "epoch": 1.53350830078125e-05,
      "grad_norm": 0.607779860496521,
      "learning_rate": 1e-05,
      "loss": 0.0902,
      "step": 4020
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 3.1394989490509033,
      "step": 4020
    },
    {
      "dynamics_loss_avg": 0.06077175214886665,
      "epoch": 1.53350830078125e-05,
      "step": 4020
    },
    {
      "action_loss_avg": 0.030778464488685132,
      "epoch": 1.53350830078125e-05,
      "step": 4020
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 4020,
      "training_step_time": 9.918325424194336
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 3.4041857719421387,
      "step": 4020
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 4020,
      "training_step_time": 10.172981023788452
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 3.137291431427002,
      "step": 4020
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 4020,
      "training_step_time": 9.905231475830078
    },
    {
      "epoch": 1.53350830078125e-05,
      "model_forward_time": 3.130763292312622,
      "step": 4020
    },
    {
      "epoch": 1.53350830078125e-05,
      "step": 4020,
      "training_step_time": 10.288136005401611
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 3.149600028991699,
      "step": 4021
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 4021,
      "training_step_time": 9.924229621887207
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 3.133232831954956,
      "step": 4021
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 4021,
      "training_step_time": 9.90153431892395
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 3.130971908569336,
      "step": 4021
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 4021,
      "training_step_time": 10.186916828155518
    },
    {
      "epoch": 1.5338897705078125e-05,
      "model_forward_time": 3.131950616836548,
      "step": 4021
    },
    {
      "epoch": 1.5338897705078125e-05,
      "step": 4021,
      "training_step_time": 10.261517524719238
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 2.4444191455841064,
      "step": 4022
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 4022,
      "training_step_time": 7.466729640960693
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 3.1326398849487305,
      "step": 4022
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 4022,
      "training_step_time": 9.89192008972168
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 3.1253325939178467,
      "step": 4022
    },
    {
      "dynamics_loss_avg": 0.06689858883619308,
      "epoch": 1.534271240234375e-05,
      "step": 4022
    },
    {
      "action_loss_avg": 0.036322697345167396,
      "epoch": 1.534271240234375e-05,
      "step": 4022
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 4022,
      "training_step_time": 9.885628700256348
    },
    {
      "epoch": 1.534271240234375e-05,
      "model_forward_time": 3.13470458984375,
      "step": 4022
    },
    {
      "epoch": 1.534271240234375e-05,
      "step": 4022,
      "training_step_time": 10.297827005386353
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 3.1542134284973145,
      "step": 4023
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 4023,
      "training_step_time": 9.92017912864685
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 2.4315943717956543,
      "step": 4023
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 4023,
      "training_step_time": 7.455081462860107
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 3.1324682235717773,
      "step": 4023
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 4023,
      "training_step_time": 9.891257762908936
    },
    {
      "epoch": 1.5346527099609374e-05,
      "model_forward_time": 3.1345081329345703,
      "step": 4023
    },
    {
      "epoch": 1.5346527099609374e-05,
      "step": 4023,
      "training_step_time": 10.273585557937622
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 3.1542959213256836,
      "step": 4024
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 4024,
      "training_step_time": 9.924772500991821
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 3.4202606678009033,
      "step": 4024
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 4024,
      "training_step_time": 10.193593978881836
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 3.1403908729553223,
      "step": 4024
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 4024,
      "training_step_time": 9.914597034454346
    },
    {
      "epoch": 1.5350341796875e-05,
      "model_forward_time": 3.142587900161743,
      "step": 4024
    },
    {
      "epoch": 1.5350341796875e-05,
      "step": 4024,
      "training_step_time": 10.26547884941101
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 3.1549954414367676,
      "step": 4025
    },
    {
      "dynamics_loss_avg": 0.06200030259788036,
      "epoch": 1.5354156494140626e-05,
      "step": 4025
    },
    {
      "action_loss_avg": 0.02979114013724029,
      "epoch": 1.5354156494140626e-05,
      "step": 4025
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 4025,
      "training_step_time": 9.933798789978027
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 3.4328081607818604,
      "step": 4025
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 4025,
      "training_step_time": 10.204238891601562
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 3.144591808319092,
      "step": 4025
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 4025,
      "training_step_time": 9.920243740081787
    },
    {
      "epoch": 1.5354156494140626e-05,
      "model_forward_time": 3.150212287902832,
      "step": 4025
    },
    {
      "epoch": 1.5354156494140626e-05,
      "step": 4025,
      "training_step_time": 10.306147813796997
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 3.1590607166290283,
      "step": 4026
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 4026,
      "training_step_time": 9.937549352645874
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 2.436720371246338,
      "step": 4026
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 4026,
      "training_step_time": 7.471668243408203
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 3.1465749740600586,
      "step": 4026
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 4026,
      "training_step_time": 10.1925630569458
    },
    {
      "epoch": 1.535797119140625e-05,
      "model_forward_time": 3.150435447692871,
      "step": 4026
    },
    {
      "epoch": 1.535797119140625e-05,
      "step": 4026,
      "training_step_time": 10.300938129425049
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 3.166621446609497,
      "step": 4027
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 4027,
      "training_step_time": 9.94697380065918
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 3.1377735137939453,
      "step": 4027
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 4027,
      "training_step_time": 9.91081976890564
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 3.142686367034912,
      "step": 4027
    },
    {
      "dynamics_loss_avg": 0.06062072701752186,
      "epoch": 1.5361785888671876e-05,
      "step": 4027
    },
    {
      "action_loss_avg": 0.02738365614786744,
      "epoch": 1.5361785888671876e-05,
      "step": 4027
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 4027,
      "training_step_time": 9.916633605957031
    },
    {
      "epoch": 1.5361785888671876e-05,
      "model_forward_time": 3.4220497608184814,
      "step": 4027
    },
    {
      "epoch": 1.5361785888671876e-05,
      "step": 4027,
      "training_step_time": 10.290436506271362
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 3.153564929962158,
      "step": 4028
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 4028,
      "training_step_time": 9.920207262039185
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 3.13510799407959,
      "step": 4028
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 4028,
      "training_step_time": 9.907116174697876
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 3.1424248218536377,
      "step": 4028
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 4028,
      "training_step_time": 9.91183066368103
    },
    {
      "epoch": 1.53656005859375e-05,
      "model_forward_time": 3.137918710708618,
      "step": 4028
    },
    {
      "epoch": 1.53656005859375e-05,
      "step": 4028,
      "training_step_time": 10.266134023666382
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 3.4691739082336426,
      "step": 4029
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 4029,
      "training_step_time": 10.251210689544678
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 3.140138626098633,
      "step": 4029
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 4029,
      "training_step_time": 9.909842729568481
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 3.1474087238311768,
      "step": 4029
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 4029,
      "training_step_time": 9.920254707336426
    },
    {
      "epoch": 1.5369415283203125e-05,
      "model_forward_time": 3.1408355236053467,
      "step": 4029
    },
    {
      "epoch": 1.5369415283203125e-05,
      "step": 4029,
      "training_step_time": 10.025027513504028
    },
    {
      "epoch": 1.537322998046875e-05,
      "grad_norm": 0.6146877408027649,
      "learning_rate": 1e-05,
      "loss": 0.0895,
      "step": 4030
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 3.1606740951538086,
      "step": 4030
    },
    {
      "dynamics_loss_avg": 0.06036844551563263,
      "epoch": 1.537322998046875e-05,
      "step": 4030
    },
    {
      "action_loss_avg": 0.023259590193629266,
      "epoch": 1.537322998046875e-05,
      "step": 4030
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 4030,
      "training_step_time": 9.932429075241089
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 3.4205071926116943,
      "step": 4030
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 4030,
      "training_step_time": 10.191573858261108
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 2.439479351043701,
      "step": 4030
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 4030,
      "training_step_time": 7.472266674041748
    },
    {
      "epoch": 1.537322998046875e-05,
      "model_forward_time": 3.1400177478790283,
      "step": 4030
    },
    {
      "epoch": 1.537322998046875e-05,
      "step": 4030,
      "training_step_time": 10.35167145729065
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 3.1577107906341553,
      "step": 4031
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 4031,
      "training_step_time": 9.931535243988037
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 3.1558501720428467,
      "step": 4031
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 4031,
      "training_step_time": 9.932948112487793
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 3.4282596111297607,
      "step": 4031
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 4031,
      "training_step_time": 10.200293779373169
    },
    {
      "epoch": 1.5377044677734374e-05,
      "model_forward_time": 3.1415698528289795,
      "step": 4031
    },
    {
      "epoch": 1.5377044677734374e-05,
      "step": 4031,
      "training_step_time": 10.04478406906128
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 3.154148578643799,
      "step": 4032
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 4032,
      "training_step_time": 9.925260066986084
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 1.7466988563537598,
      "step": 4032
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 4032,
      "training_step_time": 5.204837083816528
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 3.1333491802215576,
      "step": 4032
    },
    {
      "dynamics_loss_avg": 0.05826029721647501,
      "epoch": 1.5380859375e-05,
      "step": 4032
    },
    {
      "action_loss_avg": 0.02583155771717429,
      "epoch": 1.5380859375e-05,
      "step": 4032
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 4032,
      "training_step_time": 9.904330492019653
    },
    {
      "epoch": 1.5380859375e-05,
      "model_forward_time": 3.143662214279175,
      "step": 4032
    },
    {
      "epoch": 1.5380859375e-05,
      "step": 4032,
      "training_step_time": 10.328505516052246
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 3.4115564823150635,
      "step": 4033
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 4033,
      "training_step_time": 10.180516004562378
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 3.1455321311950684,
      "step": 4033
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 4033,
      "training_step_time": 9.913572549819946
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 3.141308307647705,
      "step": 4033
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 4033,
      "training_step_time": 9.91026258468628
    },
    {
      "epoch": 1.5384674072265626e-05,
      "model_forward_time": 3.1409506797790527,
      "step": 4033
    },
    {
      "epoch": 1.5384674072265626e-05,
      "step": 4033,
      "training_step_time": 10.01481819152832
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 2.4447200298309326,
      "step": 4034
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 4034,
      "training_step_time": 7.4770667552948
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 3.420262336730957,
      "step": 4034
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 4034,
      "training_step_time": 10.189687013626099
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 3.142794132232666,
      "step": 4034
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 4034,
      "training_step_time": 9.91364049911499
    },
    {
      "epoch": 1.538848876953125e-05,
      "model_forward_time": 3.150805711746216,
      "step": 4034
    },
    {
      "epoch": 1.538848876953125e-05,
      "step": 4034,
      "training_step_time": 10.298203706741333
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 3.1538655757904053,
      "step": 4035
    },
    {
      "dynamics_loss_avg": 0.07300643920898438,
      "epoch": 1.5392303466796876e-05,
      "step": 4035
    },
    {
      "action_loss_avg": 0.030545828957110645,
      "epoch": 1.5392303466796876e-05,
      "step": 4035
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 4035,
      "training_step_time": 9.946882009506226
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 3.148796796798706,
      "step": 4035
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 4035,
      "training_step_time": 9.938743114471436
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 3.4428963661193848,
      "step": 4035
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 4035,
      "training_step_time": 10.228061199188232
    },
    {
      "epoch": 1.5392303466796876e-05,
      "model_forward_time": 3.1442711353302,
      "step": 4035
    },
    {
      "epoch": 1.5392303466796876e-05,
      "step": 4035,
      "training_step_time": 10.321928262710571
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 3.15969181060791,
      "step": 4036
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 4036,
      "training_step_time": 9.951448202133179
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 3.1461973190307617,
      "step": 4036
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 4036,
      "training_step_time": 9.930482625961304
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 3.143214225769043,
      "step": 4036
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 4036,
      "training_step_time": 9.930381059646606
    },
    {
      "epoch": 1.53961181640625e-05,
      "model_forward_time": 3.4785642623901367,
      "step": 4036
    },
    {
      "epoch": 1.53961181640625e-05,
      "step": 4036,
      "training_step_time": 10.356429815292358
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 3.156752824783325,
      "step": 4037
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 4037,
      "training_step_time": 9.943511724472046
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 3.138594388961792,
      "step": 4037
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 4037,
      "training_step_time": 9.915138721466064
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 3.147016763687134,
      "step": 4037
    },
    {
      "dynamics_loss_avg": 0.056549482978880405,
      "epoch": 1.5399932861328125e-05,
      "step": 4037
    },
    {
      "action_loss_avg": 0.0289390173740685,
      "epoch": 1.5399932861328125e-05,
      "step": 4037
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 4037,
      "training_step_time": 9.929494380950928
    },
    {
      "epoch": 1.5399932861328125e-05,
      "model_forward_time": 3.13981556892395,
      "step": 4037
    },
    {
      "epoch": 1.5399932861328125e-05,
      "step": 4037,
      "training_step_time": 10.293512344360352
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 3.4459099769592285,
      "step": 4038
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 4038,
      "training_step_time": 10.232487440109253
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 3.1446750164031982,
      "step": 4038
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 4038,
      "training_step_time": 9.933106899261475
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 3.1470327377319336,
      "step": 4038
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 4038,
      "training_step_time": 9.927407264709473
    },
    {
      "epoch": 1.540374755859375e-05,
      "model_forward_time": 3.144564628601074,
      "step": 4038
    },
    {
      "epoch": 1.540374755859375e-05,
      "step": 4038,
      "training_step_time": 10.322813272476196
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 3.1534554958343506,
      "step": 4039
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 4039,
      "training_step_time": 9.947075366973877
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 3.445270538330078,
      "step": 4039
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 4039,
      "training_step_time": 10.23106861114502
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 3.1498427391052246,
      "step": 4039
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 4039,
      "training_step_time": 9.93762731552124
    },
    {
      "epoch": 1.5407562255859374e-05,
      "model_forward_time": 3.14398455619812,
      "step": 4039
    },
    {
      "epoch": 1.5407562255859374e-05,
      "step": 4039,
      "training_step_time": 10.332458257675171
    },
    {
      "epoch": 1.5411376953125e-05,
      "grad_norm": 0.44038107991218567,
      "learning_rate": 1e-05,
      "loss": 0.0907,
      "step": 4040
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 3.1530539989471436,
      "step": 4040
    },
    {
      "dynamics_loss_avg": 0.05804959870874882,
      "epoch": 1.5411376953125e-05,
      "step": 4040
    },
    {
      "action_loss_avg": 0.024635436199605464,
      "epoch": 1.5411376953125e-05,
      "step": 4040
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 4040,
      "training_step_time": 9.935287714004517
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 3.1457037925720215,
      "step": 4040
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 4040,
      "training_step_time": 9.925897359848022
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 3.418755292892456,
      "step": 4040
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 4040,
      "training_step_time": 10.199376106262207
    },
    {
      "epoch": 1.5411376953125e-05,
      "model_forward_time": 3.147205114364624,
      "step": 4040
    },
    {
      "epoch": 1.5411376953125e-05,
      "step": 4040,
      "training_step_time": 10.345988035202026
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 3.160250425338745,
      "step": 4041
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 4041,
      "training_step_time": 9.947824239730835
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 3.149017810821533,
      "step": 4041
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 4041,
      "training_step_time": 9.930082559585571
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 3.1456570625305176,
      "step": 4041
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 4041,
      "training_step_time": 9.926963806152344
    },
    {
      "epoch": 1.5415191650390626e-05,
      "model_forward_time": 3.3916146755218506,
      "step": 4041
    },
    {
      "epoch": 1.5415191650390626e-05,
      "step": 4041,
      "training_step_time": 10.302398204803467
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 3.1480319499969482,
      "step": 4042
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 4042,
      "training_step_time": 9.927014350891113
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 3.140977621078491,
      "step": 4042
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 4042,
      "training_step_time": 9.919637203216553
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 3.1377949714660645,
      "step": 4042
    },
    {
      "dynamics_loss_avg": 0.08244418650865555,
      "epoch": 1.541900634765625e-05,
      "step": 4042
    },
    {
      "action_loss_avg": 0.034407312981784344,
      "epoch": 1.541900634765625e-05,
      "step": 4042
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 4042,
      "training_step_time": 9.914008140563965
    },
    {
      "epoch": 1.541900634765625e-05,
      "model_forward_time": 3.4404008388519287,
      "step": 4042
    },
    {
      "epoch": 1.541900634765625e-05,
      "step": 4042,
      "training_step_time": 10.341184616088867
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 3.1609742641448975,
      "step": 4043
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 4043,
      "training_step_time": 9.939260005950928
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 3.1439449787139893,
      "step": 4043
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 4043,
      "training_step_time": 9.921595335006714
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 3.1417524814605713,
      "step": 4043
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 4043,
      "training_step_time": 9.919874906539917
    },
    {
      "epoch": 1.5422821044921876e-05,
      "model_forward_time": 3.1453983783721924,
      "step": 4043
    },
    {
      "epoch": 1.5422821044921876e-05,
      "step": 4043,
      "training_step_time": 10.315398454666138
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 3.160325527191162,
      "step": 4044
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 4044,
      "training_step_time": 10.221160888671875
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 3.1478891372680664,
      "step": 4044
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 4044,
      "training_step_time": 9.930566549301147
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 3.1443283557891846,
      "step": 4044
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 4044,
      "training_step_time": 9.930709600448608
    },
    {
      "epoch": 1.54266357421875e-05,
      "model_forward_time": 3.1456174850463867,
      "step": 4044
    },
    {
      "epoch": 1.54266357421875e-05,
      "step": 4044,
      "training_step_time": 10.281854629516602
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 3.1558115482330322,
      "step": 4045
    },
    {
      "dynamics_loss_avg": 0.05954956188797951,
      "epoch": 1.5430450439453125e-05,
      "step": 4045
    },
    {
      "action_loss_avg": 0.027233545063063504,
      "epoch": 1.5430450439453125e-05,
      "step": 4045
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 4045,
      "training_step_time": 9.9444580078125
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 2.4444947242736816,
      "step": 4045
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 4045,
      "training_step_time": 7.762316703796387
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 3.140164852142334,
      "step": 4045
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 4045,
      "training_step_time": 9.924529552459717
    },
    {
      "epoch": 1.5430450439453125e-05,
      "model_forward_time": 3.146982431411743,
      "step": 4045
    },
    {
      "epoch": 1.5430450439453125e-05,
      "step": 4045,
      "training_step_time": 10.026206493377686
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 3.1580820083618164,
      "step": 4046
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 4046,
      "training_step_time": 9.946070671081543
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 3.1456267833709717,
      "step": 4046
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 4046,
      "training_step_time": 9.930861473083496
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 3.417682409286499,
      "step": 4046
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 4046,
      "training_step_time": 10.203325510025024
    },
    {
      "epoch": 1.543426513671875e-05,
      "model_forward_time": 3.1511900424957275,
      "step": 4046
    },
    {
      "epoch": 1.543426513671875e-05,
      "step": 4046,
      "training_step_time": 10.312425374984741
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 3.1568620204925537,
      "step": 4047
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 4047,
      "training_step_time": 9.944948434829712
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 2.439032793045044,
      "step": 4047
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 4047,
      "training_step_time": 7.4714415073394775
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 3.14434814453125,
      "step": 4047
    },
    {
      "dynamics_loss_avg": 0.07622913792729377,
      "epoch": 1.5438079833984374e-05,
      "step": 4047
    },
    {
      "action_loss_avg": 0.03147649839520454,
      "epoch": 1.5438079833984374e-05,
      "step": 4047
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 4047,
      "training_step_time": 9.922148942947388
    },
    {
      "epoch": 1.5438079833984374e-05,
      "model_forward_time": 3.1414644718170166,
      "step": 4047
    },
    {
      "epoch": 1.5438079833984374e-05,
      "step": 4047,
      "training_step_time": 10.34551215171814
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 3.157282829284668,
      "step": 4048
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 4048,
      "training_step_time": 9.941641330718994
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 3.1421496868133545,
      "step": 4048
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 4048,
      "training_step_time": 9.922187805175781
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 3.1469128131866455,
      "step": 4048
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 4048,
      "training_step_time": 9.927989959716797
    },
    {
      "epoch": 1.544189453125e-05,
      "model_forward_time": 3.144488573074341,
      "step": 4048
    },
    {
      "epoch": 1.544189453125e-05,
      "step": 4048,
      "training_step_time": 10.036723136901855
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 3.4592337608337402,
      "step": 4049
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 4049,
      "training_step_time": 10.243581295013428
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 3.140881061553955,
      "step": 4049
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 4049,
      "training_step_time": 9.918500185012817
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 3.1478960514068604,
      "step": 4049
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 4049,
      "training_step_time": 9.924162864685059
    },
    {
      "epoch": 1.5445709228515626e-05,
      "model_forward_time": 3.1379330158233643,
      "step": 4049
    },
    {
      "epoch": 1.5445709228515626e-05,
      "step": 4049,
      "training_step_time": 10.015657186508179
    },
    {
      "epoch": 1.544952392578125e-05,
      "grad_norm": 0.3437182307243347,
      "learning_rate": 1e-05,
      "loss": 0.0933,
      "step": 4050
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 3.1503193378448486,
      "step": 4050
    },
    {
      "dynamics_loss_avg": 0.07078957911580801,
      "epoch": 1.544952392578125e-05,
      "step": 4050
    },
    {
      "action_loss_avg": 0.02918472308665514,
      "epoch": 1.544952392578125e-05,
      "step": 4050
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 4050,
      "training_step_time": 9.927583932876587
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 3.4113845825195312,
      "step": 4050
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 4050,
      "training_step_time": 10.182373285293579
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 3.1424448490142822,
      "step": 4050
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 4050,
      "training_step_time": 9.91415786743164
    },
    {
      "epoch": 1.544952392578125e-05,
      "model_forward_time": 3.139090061187744,
      "step": 4050
    },
    {
      "epoch": 1.544952392578125e-05,
      "step": 4050,
      "training_step_time": 10.327423095703125
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 2.4483795166015625,
      "step": 4051
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 4051,
      "training_step_time": 7.481497049331665
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 3.1384024620056152,
      "step": 4051
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 4051,
      "training_step_time": 9.911442756652832
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 2.438776969909668,
      "step": 4051
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 4051,
      "training_step_time": 7.747649669647217
    },
    {
      "epoch": 1.5453338623046876e-05,
      "model_forward_time": 3.1385767459869385,
      "step": 4051
    },
    {
      "epoch": 1.5453338623046876e-05,
      "step": 4051,
      "training_step_time": 10.346479415893555
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 3.1462676525115967,
      "step": 4052
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 4052,
      "training_step_time": 9.929060459136963
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 3.139399528503418,
      "step": 4052
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 4052,
      "training_step_time": 9.911882400512695
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 3.1364386081695557,
      "step": 4052
    },
    {
      "dynamics_loss_avg": 0.07152322828769683,
      "epoch": 1.54571533203125e-05,
      "step": 4052
    },
    {
      "action_loss_avg": 0.03237005202099681,
      "epoch": 1.54571533203125e-05,
      "step": 4052
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 4052,
      "training_step_time": 9.910652875900269
    },
    {
      "epoch": 1.54571533203125e-05,
      "model_forward_time": 3.137256383895874,
      "step": 4052
    },
    {
      "epoch": 1.54571533203125e-05,
      "step": 4052,
      "training_step_time": 10.29173755645752
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 3.1524899005889893,
      "step": 4053
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 4053,
      "training_step_time": 9.933640003204346
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 3.137399196624756,
      "step": 4053
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 4053,
      "training_step_time": 9.916906118392944
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 3.136798858642578,
      "step": 4053
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 4053,
      "training_step_time": 9.91313362121582
    },
    {
      "epoch": 1.5460968017578125e-05,
      "model_forward_time": 3.142918109893799,
      "step": 4053
    },
    {
      "epoch": 1.5460968017578125e-05,
      "step": 4053,
      "training_step_time": 10.027545690536499
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 3.4379289150238037,
      "step": 4054
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 4054,
      "training_step_time": 10.217387914657593
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 3.1398165225982666,
      "step": 4054
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 4054,
      "training_step_time": 9.924226999282837
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 3.1416361331939697,
      "step": 4054
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 4054,
      "training_step_time": 9.926737785339355
    },
    {
      "epoch": 1.546478271484375e-05,
      "model_forward_time": 3.148200511932373,
      "step": 4054
    },
    {
      "epoch": 1.546478271484375e-05,
      "step": 4054,
      "training_step_time": 10.025583744049072
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 3.1503171920776367,
      "step": 4055
    },
    {
      "dynamics_loss_avg": 0.05961343944072724,
      "epoch": 1.5468597412109374e-05,
      "step": 4055
    },
    {
      "action_loss_avg": 0.02126577040180564,
      "epoch": 1.5468597412109374e-05,
      "step": 4055
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 4055,
      "training_step_time": 9.942180633544922
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 3.463900566101074,
      "step": 4055
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 4055,
      "training_step_time": 10.233425855636597
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 3.1313259601593018,
      "step": 4055
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 4055,
      "training_step_time": 9.90005612373352
    },
    {
      "epoch": 1.5468597412109374e-05,
      "model_forward_time": 3.138176441192627,
      "step": 4055
    },
    {
      "epoch": 1.5468597412109374e-05,
      "step": 4055,
      "training_step_time": 10.303563833236694
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 3.1440255641937256,
      "step": 4056
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 4056,
      "training_step_time": 9.915854930877686
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 3.1310181617736816,
      "step": 4056
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 4056,
      "training_step_time": 9.901916980743408
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 3.1396918296813965,
      "step": 4056
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 4056,
      "training_step_time": 10.250030517578125
    },
    {
      "epoch": 1.5472412109375e-05,
      "model_forward_time": 3.136878252029419,
      "step": 4056
    },
    {
      "epoch": 1.5472412109375e-05,
      "step": 4056,
      "training_step_time": 10.333252429962158
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 3.155153512954712,
      "step": 4057
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 4057,
      "training_step_time": 9.930341482162476
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 3.134491443634033,
      "step": 4057
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 4057,
      "training_step_time": 9.904414415359497
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 3.1365463733673096,
      "step": 4057
    },
    {
      "dynamics_loss_avg": 0.05719806859269738,
      "epoch": 1.5476226806640626e-05,
      "step": 4057
    },
    {
      "action_loss_avg": 0.025327858095988632,
      "epoch": 1.5476226806640626e-05,
      "step": 4057
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 4057,
      "training_step_time": 9.904801845550537
    },
    {
      "epoch": 1.5476226806640626e-05,
      "model_forward_time": 3.4624173641204834,
      "step": 4057
    },
    {
      "epoch": 1.5476226806640626e-05,
      "step": 4057,
      "training_step_time": 10.33405351638794
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 3.1520676612854004,
      "step": 4058
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 4058,
      "training_step_time": 9.92411184310913
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 3.1353883743286133,
      "step": 4058
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 4058,
      "training_step_time": 9.906630277633667
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 2.4362518787384033,
      "step": 4058
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 4058,
      "training_step_time": 7.462211608886719
    },
    {
      "epoch": 1.548004150390625e-05,
      "model_forward_time": 3.1309006214141846,
      "step": 4058
    },
    {
      "epoch": 1.548004150390625e-05,
      "step": 4058,
      "training_step_time": 10.031991958618164
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 3.139573574066162,
      "step": 4059
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 4059,
      "training_step_time": 9.912392139434814
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 3.4272780418395996,
      "step": 4059
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 4059,
      "training_step_time": 10.198095798492432
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 3.141587972640991,
      "step": 4059
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 4059,
      "training_step_time": 9.91329574584961
    },
    {
      "epoch": 1.5483856201171876e-05,
      "model_forward_time": 3.1384646892547607,
      "step": 4059
    },
    {
      "epoch": 1.5483856201171876e-05,
      "step": 4059,
      "training_step_time": 10.028202056884766
    },
    {
      "epoch": 1.54876708984375e-05,
      "grad_norm": 0.2558862268924713,
      "learning_rate": 1e-05,
      "loss": 0.0906,
      "step": 4060
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 2.4474411010742188,
      "step": 4060
    },
    {
      "dynamics_loss_avg": 0.06040243171155453,
      "epoch": 1.54876708984375e-05,
      "step": 4060
    },
    {
      "action_loss_avg": 0.025289469957351686,
      "epoch": 1.54876708984375e-05,
      "step": 4060
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 4060,
      "training_step_time": 7.4782798290252686
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 3.39062237739563,
      "step": 4060
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 4060,
      "training_step_time": 10.162029504776001
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 3.1404459476470947,
      "step": 4060
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 4060,
      "training_step_time": 9.914015769958496
    },
    {
      "epoch": 1.54876708984375e-05,
      "model_forward_time": 3.1326005458831787,
      "step": 4060
    },
    {
      "epoch": 1.54876708984375e-05,
      "step": 4060,
      "training_step_time": 10.32996416091919
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 3.1443796157836914,
      "step": 4061
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 4061,
      "training_step_time": 9.919671058654785
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 3.138094663619995,
      "step": 4061
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 4061,
      "training_step_time": 9.909514427185059
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 3.1345713138580322,
      "step": 4061
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 4061,
      "training_step_time": 10.167662382125854
    },
    {
      "epoch": 1.5491485595703125e-05,
      "model_forward_time": 3.136810541152954,
      "step": 4061
    },
    {
      "epoch": 1.5491485595703125e-05,
      "step": 4061,
      "training_step_time": 10.026756525039673
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 3.147176504135132,
      "step": 4062
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 4062,
      "training_step_time": 9.92592167854309
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 3.1375818252563477,
      "step": 4062
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 4062,
      "training_step_time": 9.909493923187256
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 3.1382558345794678,
      "step": 4062
    },
    {
      "dynamics_loss_avg": 0.05982524938881397,
      "epoch": 1.549530029296875e-05,
      "step": 4062
    },
    {
      "action_loss_avg": 0.024503884930163623,
      "epoch": 1.549530029296875e-05,
      "step": 4062
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 4062,
      "training_step_time": 9.91494345664978
    },
    {
      "epoch": 1.549530029296875e-05,
      "model_forward_time": 2.738356113433838,
      "step": 4062
    },
    {
      "epoch": 1.549530029296875e-05,
      "step": 4062,
      "training_step_time": 10.312963247299194
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 3.150448799133301,
      "step": 4063
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 4063,
      "training_step_time": 9.92503833770752
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 3.1376914978027344,
      "step": 4063
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 4063,
      "training_step_time": 9.910007238388062
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 3.1500604152679443,
      "step": 4063
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 4063,
      "training_step_time": 9.91980504989624
    },
    {
      "epoch": 1.5499114990234374e-05,
      "model_forward_time": 3.1343350410461426,
      "step": 4063
    },
    {
      "epoch": 1.5499114990234374e-05,
      "step": 4063,
      "training_step_time": 10.31665325164795
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 3.418752670288086,
      "step": 4064
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 4064,
      "training_step_time": 10.18680715560913
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 3.138556957244873,
      "step": 4064
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 4064,
      "training_step_time": 9.903337240219116
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 3.1362407207489014,
      "step": 4064
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 4064,
      "training_step_time": 9.901719093322754
    },
    {
      "epoch": 1.55029296875e-05,
      "model_forward_time": 3.1403422355651855,
      "step": 4064
    },
    {
      "epoch": 1.55029296875e-05,
      "step": 4064,
      "training_step_time": 10.006022453308105
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 3.155669927597046,
      "step": 4065
    },
    {
      "dynamics_loss_avg": 0.06853582710027695,
      "epoch": 1.5506744384765626e-05,
      "step": 4065
    },
    {
      "action_loss_avg": 0.02548096003010869,
      "epoch": 1.5506744384765626e-05,
      "step": 4065
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 4065,
      "training_step_time": 9.948967456817627
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 3.1415889263153076,
      "step": 4065
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 4065,
      "training_step_time": 10.208705425262451
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 2.4456987380981445,
      "step": 4065
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 4065,
      "training_step_time": 7.4750895500183105
    },
    {
      "epoch": 1.5506744384765626e-05,
      "model_forward_time": 3.13777756690979,
      "step": 4065
    },
    {
      "epoch": 1.5506744384765626e-05,
      "step": 4065,
      "training_step_time": 10.267925262451172
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 3.154309034347534,
      "step": 4066
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 4066,
      "training_step_time": 9.92934513092041
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 3.1382949352264404,
      "step": 4066
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 4066,
      "training_step_time": 9.917934894561768
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 3.1407456398010254,
      "step": 4066
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 4066,
      "training_step_time": 9.917124271392822
    },
    {
      "epoch": 1.551055908203125e-05,
      "model_forward_time": 3.4569478034973145,
      "step": 4066
    },
    {
      "epoch": 1.551055908203125e-05,
      "step": 4066,
      "training_step_time": 10.325303792953491
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 3.1624133586883545,
      "step": 4067
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 4067,
      "training_step_time": 9.939059019088745
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 3.148031711578369,
      "step": 4067
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 4067,
      "training_step_time": 9.930181741714478
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 3.1492090225219727,
      "step": 4067
    },
    {
      "dynamics_loss_avg": 0.0646290734410286,
      "epoch": 1.5514373779296876e-05,
      "step": 4067
    },
    {
      "action_loss_avg": 0.035137383034452796,
      "epoch": 1.5514373779296876e-05,
      "step": 4067
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 4067,
      "training_step_time": 9.930022239685059
    },
    {
      "epoch": 1.5514373779296876e-05,
      "model_forward_time": 3.148247003555298,
      "step": 4067
    },
    {
      "epoch": 1.5514373779296876e-05,
      "step": 4067,
      "training_step_time": 10.30817437171936
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 3.1572675704956055,
      "step": 4068
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 4068,
      "training_step_time": 9.940757751464844
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 3.1440951824188232,
      "step": 4068
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 4068,
      "training_step_time": 9.92517876625061
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 3.149473190307617,
      "step": 4068
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 4068,
      "training_step_time": 9.933749198913574
    },
    {
      "epoch": 1.55181884765625e-05,
      "model_forward_time": 3.147822856903076,
      "step": 4068
    },
    {
      "epoch": 1.55181884765625e-05,
      "step": 4068,
      "training_step_time": 10.307945966720581
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 3.454805850982666,
      "step": 4069
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 4069,
      "training_step_time": 10.240357875823975
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 2.4435479640960693,
      "step": 4069
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 4069,
      "training_step_time": 7.47696852684021
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 3.1475648880004883,
      "step": 4069
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 4069,
      "training_step_time": 9.930444478988647
    },
    {
      "epoch": 1.5522003173828125e-05,
      "model_forward_time": 3.145475149154663,
      "step": 4069
    },
    {
      "epoch": 1.5522003173828125e-05,
      "step": 4069,
      "training_step_time": 10.053749084472656
    },
    {
      "epoch": 1.552581787109375e-05,
      "grad_norm": 0.5641014575958252,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 4070
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 3.1557159423828125,
      "step": 4070
    },
    {
      "dynamics_loss_avg": 0.05809209570288658,
      "epoch": 1.552581787109375e-05,
      "step": 4070
    },
    {
      "action_loss_avg": 0.025319174397736788,
      "epoch": 1.552581787109375e-05,
      "step": 4070
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 4070,
      "training_step_time": 9.952925443649292
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 3.146339178085327,
      "step": 4070
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 4070,
      "training_step_time": 9.92730450630188
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 3.4449656009674072,
      "step": 4070
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 4070,
      "training_step_time": 10.221764087677002
    },
    {
      "epoch": 1.552581787109375e-05,
      "model_forward_time": 3.146148204803467,
      "step": 4070
    },
    {
      "epoch": 1.552581787109375e-05,
      "step": 4070,
      "training_step_time": 10.0480477809906
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 3.1615073680877686,
      "step": 4071
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 4071,
      "training_step_time": 9.933737993240356
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 3.14444637298584,
      "step": 4071
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 4071,
      "training_step_time": 9.918660640716553
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 3.424121141433716,
      "step": 4071
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 4071,
      "training_step_time": 10.204071283340454
    },
    {
      "epoch": 1.5529632568359374e-05,
      "model_forward_time": 3.146263837814331,
      "step": 4071
    },
    {
      "epoch": 1.5529632568359374e-05,
      "step": 4071,
      "training_step_time": 10.331969738006592
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 3.1535754203796387,
      "step": 4072
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 4072,
      "training_step_time": 9.929505348205566
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 3.140202760696411,
      "step": 4072
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 4072,
      "training_step_time": 9.917175531387329
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 3.13763689994812,
      "step": 4072
    },
    {
      "dynamics_loss_avg": 0.056862051412463185,
      "epoch": 1.5533447265625e-05,
      "step": 4072
    },
    {
      "action_loss_avg": 0.02803000481799245,
      "epoch": 1.5533447265625e-05,
      "step": 4072
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 4072,
      "training_step_time": 9.921450138092041
    },
    {
      "epoch": 1.5533447265625e-05,
      "model_forward_time": 2.4397008419036865,
      "step": 4072
    },
    {
      "epoch": 1.5533447265625e-05,
      "step": 4072,
      "training_step_time": 10.337146520614624
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 3.4668662548065186,
      "step": 4073
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 4073,
      "training_step_time": 10.24450135231018
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 3.1423416137695312,
      "step": 4073
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 4073,
      "training_step_time": 9.92615032196045
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 3.1385791301727295,
      "step": 4073
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 4073,
      "training_step_time": 9.91661262512207
    },
    {
      "epoch": 1.5537261962890627e-05,
      "model_forward_time": 3.143688440322876,
      "step": 4073
    },
    {
      "epoch": 1.5537261962890627e-05,
      "step": 4073,
      "training_step_time": 10.338839530944824
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 3.4146080017089844,
      "step": 4074
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 4074,
      "training_step_time": 10.190184593200684
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 3.1405606269836426,
      "step": 4074
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 4074,
      "training_step_time": 9.910839796066284
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 3.134164571762085,
      "step": 4074
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 4074,
      "training_step_time": 9.910676002502441
    },
    {
      "epoch": 1.554107666015625e-05,
      "model_forward_time": 3.142522096633911,
      "step": 4074
    },
    {
      "epoch": 1.554107666015625e-05,
      "step": 4074,
      "training_step_time": 10.032485008239746
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 3.152649402618408,
      "step": 4075
    },
    {
      "dynamics_loss_avg": 0.06548705659806728,
      "epoch": 1.5544891357421876e-05,
      "step": 4075
    },
    {
      "action_loss_avg": 0.027026405464857815,
      "epoch": 1.5544891357421876e-05,
      "step": 4075
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 4075,
      "training_step_time": 9.931254863739014
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 3.138029098510742,
      "step": 4075
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 4075,
      "training_step_time": 10.183812856674194
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 2.435560703277588,
      "step": 4075
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 4075,
      "training_step_time": 7.4635748863220215
    },
    {
      "epoch": 1.5544891357421876e-05,
      "model_forward_time": 3.133819103240967,
      "step": 4075
    },
    {
      "epoch": 1.5544891357421876e-05,
      "step": 4075,
      "training_step_time": 10.080112218856812
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 3.1666502952575684,
      "step": 4076
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 4076,
      "training_step_time": 9.959840774536133
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 3.1386501789093018,
      "step": 4076
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 4076,
      "training_step_time": 9.915522575378418
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 3.1456663608551025,
      "step": 4076
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 4076,
      "training_step_time": 10.235941171646118
    },
    {
      "epoch": 1.55487060546875e-05,
      "model_forward_time": 3.148160934448242,
      "step": 4076
    },
    {
      "epoch": 1.55487060546875e-05,
      "step": 4076,
      "training_step_time": 10.330242156982422
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 3.1504061222076416,
      "step": 4077
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 4077,
      "training_step_time": 9.921189308166504
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 3.1358156204223633,
      "step": 4077
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 4077,
      "training_step_time": 9.91285753250122
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 3.1376047134399414,
      "step": 4077
    },
    {
      "dynamics_loss_avg": 0.07308156173676253,
      "epoch": 1.5552520751953125e-05,
      "step": 4077
    },
    {
      "action_loss_avg": 0.03452866636216641,
      "epoch": 1.5552520751953125e-05,
      "step": 4077
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 4077,
      "training_step_time": 9.910159587860107
    },
    {
      "epoch": 1.5552520751953125e-05,
      "model_forward_time": 3.4659464359283447,
      "step": 4077
    },
    {
      "epoch": 1.5552520751953125e-05,
      "step": 4077,
      "training_step_time": 10.327548742294312
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 3.14321231842041,
      "step": 4078
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 4078,
      "training_step_time": 9.911725521087646
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 3.130711078643799,
      "step": 4078
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 4078,
      "training_step_time": 9.897623300552368
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 3.141331911087036,
      "step": 4078
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 4078,
      "training_step_time": 9.916813611984253
    },
    {
      "epoch": 1.555633544921875e-05,
      "model_forward_time": 3.139289617538452,
      "step": 4078
    },
    {
      "epoch": 1.555633544921875e-05,
      "step": 4078,
      "training_step_time": 10.034723281860352
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 3.144429922103882,
      "step": 4079
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 4079,
      "training_step_time": 10.197075366973877
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 3.14989972114563,
      "step": 4079
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 4079,
      "training_step_time": 9.936646461486816
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 3.141754150390625,
      "step": 4079
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 4079,
      "training_step_time": 9.919089794158936
    },
    {
      "epoch": 1.5560150146484374e-05,
      "model_forward_time": 3.143195390701294,
      "step": 4079
    },
    {
      "epoch": 1.5560150146484374e-05,
      "step": 4079,
      "training_step_time": 10.044113397598267
    },
    {
      "epoch": 1.556396484375e-05,
      "grad_norm": 0.4900008738040924,
      "learning_rate": 1e-05,
      "loss": 0.0882,
      "step": 4080
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 3.1628830432891846,
      "step": 4080
    },
    {
      "dynamics_loss_avg": 0.05363055951893329,
      "epoch": 1.556396484375e-05,
      "step": 4080
    },
    {
      "action_loss_avg": 0.026733826007694007,
      "epoch": 1.556396484375e-05,
      "step": 4080
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 4080,
      "training_step_time": 9.956278800964355
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 3.440568685531616,
      "step": 4080
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 4080,
      "training_step_time": 10.22192931175232
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 3.1476237773895264,
      "step": 4080
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 4080,
      "training_step_time": 9.927956104278564
    },
    {
      "epoch": 1.556396484375e-05,
      "model_forward_time": 3.1394879817962646,
      "step": 4080
    },
    {
      "epoch": 1.556396484375e-05,
      "step": 4080,
      "training_step_time": 10.308526515960693
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 3.1777007579803467,
      "step": 4081
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 4081,
      "training_step_time": 9.974218845367432
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 3.146042585372925,
      "step": 4081
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 4081,
      "training_step_time": 9.921611547470093
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 3.4308550357818604,
      "step": 4081
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 4081,
      "training_step_time": 10.207226514816284
    },
    {
      "epoch": 1.5567779541015627e-05,
      "model_forward_time": 2.437537431716919,
      "step": 4081
    },
    {
      "epoch": 1.5567779541015627e-05,
      "step": 4081,
      "training_step_time": 10.326543807983398
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 3.153287172317505,
      "step": 4082
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 4082,
      "training_step_time": 9.925963878631592
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 3.136455535888672,
      "step": 4082
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 4082,
      "training_step_time": 9.903892517089844
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 3.138475179672241,
      "step": 4082
    },
    {
      "dynamics_loss_avg": 0.07662133984267712,
      "epoch": 1.557159423828125e-05,
      "step": 4082
    },
    {
      "action_loss_avg": 0.03816638076677918,
      "epoch": 1.557159423828125e-05,
      "step": 4082
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 4082,
      "training_step_time": 9.908473491668701
    },
    {
      "epoch": 1.557159423828125e-05,
      "model_forward_time": 3.4358408451080322,
      "step": 4082
    },
    {
      "epoch": 1.557159423828125e-05,
      "step": 4082,
      "training_step_time": 10.335581064224243
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 3.1494252681732178,
      "step": 4083
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 4083,
      "training_step_time": 9.924322843551636
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 3.141737699508667,
      "step": 4083
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 4083,
      "training_step_time": 9.916790246963501
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 3.134812355041504,
      "step": 4083
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 4083,
      "training_step_time": 9.909335374832153
    },
    {
      "epoch": 1.5575408935546876e-05,
      "model_forward_time": 3.1458494663238525,
      "step": 4083
    },
    {
      "epoch": 1.5575408935546876e-05,
      "step": 4083,
      "training_step_time": 10.327178478240967
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 3.153716564178467,
      "step": 4084
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 4084,
      "training_step_time": 10.203283309936523
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 3.1454946994781494,
      "step": 4084
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 4084,
      "training_step_time": 9.922374248504639
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 3.144662380218506,
      "step": 4084
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 4084,
      "training_step_time": 9.922173500061035
    },
    {
      "epoch": 1.55792236328125e-05,
      "model_forward_time": 3.1453206539154053,
      "step": 4084
    },
    {
      "epoch": 1.55792236328125e-05,
      "step": 4084,
      "training_step_time": 10.041920185089111
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 3.154674768447876,
      "step": 4085
    },
    {
      "dynamics_loss_avg": 0.06301839090883732,
      "epoch": 1.5583038330078125e-05,
      "step": 4085
    },
    {
      "action_loss_avg": 0.025004688650369644,
      "epoch": 1.5583038330078125e-05,
      "step": 4085
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 4085,
      "training_step_time": 9.938066005706787
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 3.454709768295288,
      "step": 4085
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 4085,
      "training_step_time": 10.231618165969849
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 3.1353397369384766,
      "step": 4085
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 4085,
      "training_step_time": 9.909530401229858
    },
    {
      "epoch": 1.5583038330078125e-05,
      "model_forward_time": 3.1389803886413574,
      "step": 4085
    },
    {
      "epoch": 1.5583038330078125e-05,
      "step": 4085,
      "training_step_time": 10.306225538253784
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 3.1520211696624756,
      "step": 4086
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 4086,
      "training_step_time": 9.927443027496338
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 3.1315865516662598,
      "step": 4086
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 4086,
      "training_step_time": 9.898229360580444
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 3.3808035850524902,
      "step": 4086
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 4086,
      "training_step_time": 10.149479150772095
    },
    {
      "epoch": 1.558685302734375e-05,
      "model_forward_time": 3.13413405418396,
      "step": 4086
    },
    {
      "epoch": 1.558685302734375e-05,
      "step": 4086,
      "training_step_time": 10.290711402893066
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 3.1588516235351562,
      "step": 4087
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 4087,
      "training_step_time": 9.926149606704712
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 3.1394293308258057,
      "step": 4087
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 4087,
      "training_step_time": 9.90442180633545
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 3.1366078853607178,
      "step": 4087
    },
    {
      "dynamics_loss_avg": 0.0620973564684391,
      "epoch": 1.5590667724609374e-05,
      "step": 4087
    },
    {
      "action_loss_avg": 0.028962370567023756,
      "epoch": 1.5590667724609374e-05,
      "step": 4087
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 4087,
      "training_step_time": 9.90536880493164
    },
    {
      "epoch": 1.5590667724609374e-05,
      "model_forward_time": 3.417863130569458,
      "step": 4087
    },
    {
      "epoch": 1.5590667724609374e-05,
      "step": 4087,
      "training_step_time": 10.284538507461548
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 3.1555824279785156,
      "step": 4088
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 4088,
      "training_step_time": 9.936021566390991
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 3.139615774154663,
      "step": 4088
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 4088,
      "training_step_time": 9.906993627548218
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 2.435136079788208,
      "step": 4088
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 4088,
      "training_step_time": 7.461151123046875
    },
    {
      "epoch": 1.5594482421875e-05,
      "model_forward_time": 3.1346018314361572,
      "step": 4088
    },
    {
      "epoch": 1.5594482421875e-05,
      "step": 4088,
      "training_step_time": 10.295793533325195
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 3.4889307022094727,
      "step": 4089
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 4089,
      "training_step_time": 10.261881589889526
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 2.446120023727417,
      "step": 4089
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 4089,
      "training_step_time": 7.464031457901001
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 3.141172170639038,
      "step": 4089
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 4089,
      "training_step_time": 9.90979290008545
    },
    {
      "epoch": 1.5598297119140627e-05,
      "model_forward_time": 3.1378612518310547,
      "step": 4089
    },
    {
      "epoch": 1.5598297119140627e-05,
      "step": 4089,
      "training_step_time": 10.020519733428955
    },
    {
      "epoch": 1.560211181640625e-05,
      "grad_norm": 0.47000566124916077,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 4090
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 3.1665871143341064,
      "step": 4090
    },
    {
      "dynamics_loss_avg": 0.05867942050099373,
      "epoch": 1.560211181640625e-05,
      "step": 4090
    },
    {
      "action_loss_avg": 0.025671028718352317,
      "epoch": 1.560211181640625e-05,
      "step": 4090
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 4090,
      "training_step_time": 9.946034908294678
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 2.442390203475952,
      "step": 4090
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 4090,
      "training_step_time": 7.476584196090698
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 3.4225027561187744,
      "step": 4090
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 4090,
      "training_step_time": 10.19185495376587
    },
    {
      "epoch": 1.560211181640625e-05,
      "model_forward_time": 3.144894599914551,
      "step": 4090
    },
    {
      "epoch": 1.560211181640625e-05,
      "step": 4090,
      "training_step_time": 10.281320810317993
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 3.1625494956970215,
      "step": 4091
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 4091,
      "training_step_time": 9.931975841522217
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 3.147627353668213,
      "step": 4091
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 4091,
      "training_step_time": 9.919139623641968
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 3.144266366958618,
      "step": 4091
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 4091,
      "training_step_time": 10.213345527648926
    },
    {
      "epoch": 1.5605926513671876e-05,
      "model_forward_time": 3.1448347568511963,
      "step": 4091
    },
    {
      "epoch": 1.5605926513671876e-05,
      "step": 4091,
      "training_step_time": 10.28851866722107
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 3.1811907291412354,
      "step": 4092
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 4092,
      "training_step_time": 9.977222919464111
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 3.1530473232269287,
      "step": 4092
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 4092,
      "training_step_time": 9.939196586608887
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 2.4423463344573975,
      "step": 4092
    },
    {
      "dynamics_loss_avg": 0.05539187267422676,
      "epoch": 1.56097412109375e-05,
      "step": 4092
    },
    {
      "action_loss_avg": 0.019623603951185942,
      "epoch": 1.56097412109375e-05,
      "step": 4092
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 4092,
      "training_step_time": 7.480736255645752
    },
    {
      "epoch": 1.56097412109375e-05,
      "model_forward_time": 3.144096851348877,
      "step": 4092
    },
    {
      "epoch": 1.56097412109375e-05,
      "step": 4092,
      "training_step_time": 10.312416076660156
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 3.491629123687744,
      "step": 4093
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 4093,
      "training_step_time": 10.27889895439148
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 3.1486237049102783,
      "step": 4093
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 4093,
      "training_step_time": 9.9290611743927
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 3.144657850265503,
      "step": 4093
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 4093,
      "training_step_time": 9.924756526947021
    },
    {
      "epoch": 1.5613555908203125e-05,
      "model_forward_time": 3.148653745651245,
      "step": 4093
    },
    {
      "epoch": 1.5613555908203125e-05,
      "step": 4093,
      "training_step_time": 10.324591875076294
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 3.4270246028900146,
      "step": 4094
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 4094,
      "training_step_time": 10.211012363433838
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 3.151805877685547,
      "step": 4094
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 4094,
      "training_step_time": 9.937551498413086
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 3.1484978199005127,
      "step": 4094
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 4094,
      "training_step_time": 9.93466830253601
    },
    {
      "epoch": 1.561737060546875e-05,
      "model_forward_time": 3.1485087871551514,
      "step": 4094
    },
    {
      "epoch": 1.561737060546875e-05,
      "step": 4094,
      "training_step_time": 10.276804208755493
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 3.160470962524414,
      "step": 4095
    },
    {
      "dynamics_loss_avg": 0.05442298990674317,
      "epoch": 1.5621185302734374e-05,
      "step": 4095
    },
    {
      "action_loss_avg": 0.02283556165639311,
      "epoch": 1.5621185302734374e-05,
      "step": 4095
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 4095,
      "training_step_time": 9.955996036529541
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 3.1383750438690186,
      "step": 4095
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 4095,
      "training_step_time": 10.23766016960144
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 3.1398024559020996,
      "step": 4095
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 4095,
      "training_step_time": 9.924108266830444
    },
    {
      "epoch": 1.5621185302734374e-05,
      "model_forward_time": 2.4415485858917236,
      "step": 4095
    },
    {
      "epoch": 1.5621185302734374e-05,
      "step": 4095,
      "training_step_time": 10.315866231918335
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 3.145782709121704,
      "step": 4096
    },
    {
      "epoch": 1.5625e-05,
      "step": 4096,
      "training_step_time": 9.933196067810059
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 3.137300968170166,
      "step": 4096
    },
    {
      "epoch": 1.5625e-05,
      "step": 4096,
      "training_step_time": 9.914926767349243
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 3.136854648590088,
      "step": 4096
    },
    {
      "epoch": 1.5625e-05,
      "step": 4096,
      "training_step_time": 10.225185632705688
    },
    {
      "epoch": 1.5625e-05,
      "model_forward_time": 3.1420552730560303,
      "step": 4096
    },
    {
      "epoch": 1.5625e-05,
      "step": 4096,
      "training_step_time": 10.015730619430542
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 3.1516683101654053,
      "step": 4097
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 4097,
      "training_step_time": 9.935057878494263
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 3.1356260776519775,
      "step": 4097
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 4097,
      "training_step_time": 9.911777973175049
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 3.141784191131592,
      "step": 4097
    },
    {
      "dynamics_loss_avg": 0.05438473280519247,
      "epoch": 1.5628814697265627e-05,
      "step": 4097
    },
    {
      "action_loss_avg": 0.02215517722070217,
      "epoch": 1.5628814697265627e-05,
      "step": 4097
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 4097,
      "training_step_time": 9.91929316520691
    },
    {
      "epoch": 1.5628814697265627e-05,
      "model_forward_time": 3.4239554405212402,
      "step": 4097
    },
    {
      "epoch": 1.5628814697265627e-05,
      "step": 4097,
      "training_step_time": 10.32262134552002
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 3.156813144683838,
      "step": 4098
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 4098,
      "training_step_time": 9.937435150146484
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 3.131526231765747,
      "step": 4098
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 4098,
      "training_step_time": 9.905297040939331
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 3.1427836418151855,
      "step": 4098
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 4098,
      "training_step_time": 9.930273294448853
    },
    {
      "epoch": 1.563262939453125e-05,
      "model_forward_time": 3.138974666595459,
      "step": 4098
    },
    {
      "epoch": 1.563262939453125e-05,
      "step": 4098,
      "training_step_time": 10.29117488861084
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 3.142266273498535,
      "step": 4099
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 4099,
      "training_step_time": 10.23167896270752
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 3.1351535320281982,
      "step": 4099
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 4099,
      "training_step_time": 9.91041111946106
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 3.1443471908569336,
      "step": 4099
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 4099,
      "training_step_time": 9.920178890228271
    },
    {
      "epoch": 1.5636444091796876e-05,
      "model_forward_time": 3.1400811672210693,
      "step": 4099
    },
    {
      "epoch": 1.5636444091796876e-05,
      "step": 4099,
      "training_step_time": 10.329144954681396
    },
    {
      "epoch": 1.56402587890625e-05,
      "grad_norm": 0.5278304815292358,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 4100
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 3.152113437652588,
      "step": 4100
    },
    {
      "dynamics_loss_avg": 0.0592723929323256,
      "epoch": 1.56402587890625e-05,
      "step": 4100
    },
    {
      "action_loss_avg": 0.024363978835754096,
      "epoch": 1.56402587890625e-05,
      "step": 4100
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 4100,
      "training_step_time": 9.932432889938354
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 3.3997530937194824,
      "step": 4100
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 4100,
      "training_step_time": 10.172818899154663
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 2.432246208190918,
      "step": 4100
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 4100,
      "training_step_time": 7.461425065994263
    },
    {
      "epoch": 1.56402587890625e-05,
      "model_forward_time": 3.1411290168762207,
      "step": 4100
    },
    {
      "epoch": 1.56402587890625e-05,
      "step": 4100,
      "training_step_time": 10.021593809127808
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 3.151993989944458,
      "step": 4101
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 4101,
      "training_step_time": 9.922338247299194
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 3.136763334274292,
      "step": 4101
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 4101,
      "training_step_time": 9.916717767715454
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 3.136373281478882,
      "step": 4101
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 4101,
      "training_step_time": 10.251438617706299
    },
    {
      "epoch": 1.5644073486328125e-05,
      "model_forward_time": 3.1368050575256348,
      "step": 4101
    },
    {
      "epoch": 1.5644073486328125e-05,
      "step": 4101,
      "training_step_time": 10.32865047454834
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 3.1475398540496826,
      "step": 4102
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 4102,
      "training_step_time": 9.92619776725769
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 3.1317026615142822,
      "step": 4102
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 4102,
      "training_step_time": 9.91031527519226
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 3.132430076599121,
      "step": 4102
    },
    {
      "dynamics_loss_avg": 0.07567900326102972,
      "epoch": 1.564788818359375e-05,
      "step": 4102
    },
    {
      "action_loss_avg": 0.03728357357904315,
      "epoch": 1.564788818359375e-05,
      "step": 4102
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 4102,
      "training_step_time": 9.914270639419556
    },
    {
      "epoch": 1.564788818359375e-05,
      "model_forward_time": 3.3973090648651123,
      "step": 4102
    },
    {
      "epoch": 1.564788818359375e-05,
      "step": 4102,
      "training_step_time": 10.270775556564331
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 2.4447579383850098,
      "step": 4103
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 4103,
      "training_step_time": 7.483727693557739
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 3.132615327835083,
      "step": 4103
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 4103,
      "training_step_time": 9.909843444824219
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 3.132880449295044,
      "step": 4103
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 4103,
      "training_step_time": 9.908416271209717
    },
    {
      "epoch": 1.5651702880859374e-05,
      "model_forward_time": 3.134366273880005,
      "step": 4103
    },
    {
      "epoch": 1.5651702880859374e-05,
      "step": 4103,
      "training_step_time": 10.323238849639893
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 3.144108295440674,
      "step": 4104
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 4104,
      "training_step_time": 10.209581851959229
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 3.1390438079833984,
      "step": 4104
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 4104,
      "training_step_time": 9.917599439620972
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 2.4314818382263184,
      "step": 4104
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 4104,
      "training_step_time": 7.464856147766113
    },
    {
      "epoch": 1.5655517578125e-05,
      "model_forward_time": 3.1335909366607666,
      "step": 4104
    },
    {
      "epoch": 1.5655517578125e-05,
      "step": 4104,
      "training_step_time": 10.302751302719116
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 3.1494038105010986,
      "step": 4105
    },
    {
      "dynamics_loss_avg": 0.07064975313842296,
      "epoch": 1.5659332275390627e-05,
      "step": 4105
    },
    {
      "action_loss_avg": 0.02932439176365733,
      "epoch": 1.5659332275390627e-05,
      "step": 4105
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 4105,
      "training_step_time": 9.940428972244263
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 3.1373472213745117,
      "step": 4105
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 4105,
      "training_step_time": 10.236888647079468
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 3.1344845294952393,
      "step": 4105
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 4105,
      "training_step_time": 9.910435676574707
    },
    {
      "epoch": 1.5659332275390627e-05,
      "model_forward_time": 3.1382057666778564,
      "step": 4105
    },
    {
      "epoch": 1.5659332275390627e-05,
      "step": 4105,
      "training_step_time": 10.025347471237183
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 3.151890754699707,
      "step": 4106
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 4106,
      "training_step_time": 9.932865619659424
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 3.135965347290039,
      "step": 4106
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 4106,
      "training_step_time": 9.915504455566406
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 3.4263811111450195,
      "step": 4106
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 4106,
      "training_step_time": 10.206367254257202
    },
    {
      "epoch": 1.566314697265625e-05,
      "model_forward_time": 3.142918825149536,
      "step": 4106
    },
    {
      "epoch": 1.566314697265625e-05,
      "step": 4106,
      "training_step_time": 10.309889316558838
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 3.1814491748809814,
      "step": 4107
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 4107,
      "training_step_time": 9.963589906692505
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 2.436389923095703,
      "step": 4107
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 4107,
      "training_step_time": 7.473480224609375
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 3.143761396408081,
      "step": 4107
    },
    {
      "dynamics_loss_avg": 0.07810149937868119,
      "epoch": 1.5666961669921876e-05,
      "step": 4107
    },
    {
      "action_loss_avg": 0.03967137783765793,
      "epoch": 1.5666961669921876e-05,
      "step": 4107
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 4107,
      "training_step_time": 9.933525085449219
    },
    {
      "epoch": 1.5666961669921876e-05,
      "model_forward_time": 3.1439998149871826,
      "step": 4107
    },
    {
      "epoch": 1.5666961669921876e-05,
      "step": 4107,
      "training_step_time": 10.344902515411377
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 3.157456159591675,
      "step": 4108
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 4108,
      "training_step_time": 9.943540811538696
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 3.1350114345550537,
      "step": 4108
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 4108,
      "training_step_time": 9.918035984039307
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 3.1437737941741943,
      "step": 4108
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 4108,
      "training_step_time": 9.921077013015747
    },
    {
      "epoch": 1.56707763671875e-05,
      "model_forward_time": 3.138672113418579,
      "step": 4108
    },
    {
      "epoch": 1.56707763671875e-05,
      "step": 4108,
      "training_step_time": 10.015287637710571
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 3.4573755264282227,
      "step": 4109
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 4109,
      "training_step_time": 10.234113693237305
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 3.128185987472534,
      "step": 4109
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 4109,
      "training_step_time": 9.906105279922485
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 3.1377363204956055,
      "step": 4109
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 4109,
      "training_step_time": 9.920234441757202
    },
    {
      "epoch": 1.5674591064453125e-05,
      "model_forward_time": 3.1353719234466553,
      "step": 4109
    },
    {
      "epoch": 1.5674591064453125e-05,
      "step": 4109,
      "training_step_time": 10.307268381118774
    },
    {
      "epoch": 1.567840576171875e-05,
      "grad_norm": 0.6036770343780518,
      "learning_rate": 1e-05,
      "loss": 0.0978,
      "step": 4110
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 3.1506354808807373,
      "step": 4110
    },
    {
      "dynamics_loss_avg": 0.07549621798098087,
      "epoch": 1.567840576171875e-05,
      "step": 4110
    },
    {
      "action_loss_avg": 0.03265176275745034,
      "epoch": 1.567840576171875e-05,
      "step": 4110
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 4110,
      "training_step_time": 9.935333490371704
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 3.388451337814331,
      "step": 4110
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 4110,
      "training_step_time": 10.171802282333374
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 3.1366984844207764,
      "step": 4110
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 4110,
      "training_step_time": 9.926862239837646
    },
    {
      "epoch": 1.567840576171875e-05,
      "model_forward_time": 3.131643533706665,
      "step": 4110
    },
    {
      "epoch": 1.567840576171875e-05,
      "step": 4110,
      "training_step_time": 10.278571367263794
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 2.4456543922424316,
      "step": 4111
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 4111,
      "training_step_time": 7.483463287353516
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 3.1378042697906494,
      "step": 4111
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 4111,
      "training_step_time": 9.920931577682495
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 3.4183690547943115,
      "step": 4111
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 4111,
      "training_step_time": 10.204856634140015
    },
    {
      "epoch": 1.5682220458984374e-05,
      "model_forward_time": 3.1378679275512695,
      "step": 4111
    },
    {
      "epoch": 1.5682220458984374e-05,
      "step": 4111,
      "training_step_time": 10.038418292999268
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 3.1447343826293945,
      "step": 4112
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 4112,
      "training_step_time": 9.927363872528076
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 3.1375794410705566,
      "step": 4112
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 4112,
      "training_step_time": 9.92102313041687
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 3.1328060626983643,
      "step": 4112
    },
    {
      "dynamics_loss_avg": 0.06829229276627302,
      "epoch": 1.568603515625e-05,
      "step": 4112
    },
    {
      "action_loss_avg": 0.028284467197954653,
      "epoch": 1.568603515625e-05,
      "step": 4112
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 4112,
      "training_step_time": 9.922287702560425
    },
    {
      "epoch": 1.568603515625e-05,
      "model_forward_time": 3.141890525817871,
      "step": 4112
    },
    {
      "epoch": 1.568603515625e-05,
      "step": 4112,
      "training_step_time": 10.307849168777466
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 3.477835178375244,
      "step": 4113
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 4113,
      "training_step_time": 10.263333320617676
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 3.146669864654541,
      "step": 4113
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 4113,
      "training_step_time": 9.931783437728882
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 3.1416513919830322,
      "step": 4113
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 4113,
      "training_step_time": 9.928164720535278
    },
    {
      "epoch": 1.5689849853515627e-05,
      "model_forward_time": 3.1406824588775635,
      "step": 4113
    },
    {
      "epoch": 1.5689849853515627e-05,
      "step": 4113,
      "training_step_time": 10.040918350219727
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 3.408890962600708,
      "step": 4114
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 4114,
      "training_step_time": 10.198091268539429
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 3.1415750980377197,
      "step": 4114
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 4114,
      "training_step_time": 9.922356128692627
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 3.1389405727386475,
      "step": 4114
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 4114,
      "training_step_time": 9.92132830619812
    },
    {
      "epoch": 1.569366455078125e-05,
      "model_forward_time": 3.1414337158203125,
      "step": 4114
    },
    {
      "epoch": 1.569366455078125e-05,
      "step": 4114,
      "training_step_time": 10.295536756515503
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 3.1462109088897705,
      "step": 4115
    },
    {
      "dynamics_loss_avg": 0.0744399033486843,
      "epoch": 1.5697479248046876e-05,
      "step": 4115
    },
    {
      "action_loss_avg": 0.02737396312877536,
      "epoch": 1.5697479248046876e-05,
      "step": 4115
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 4115,
      "training_step_time": 9.923145771026611
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 3.136855125427246,
      "step": 4115
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 4115,
      "training_step_time": 10.196386337280273
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 3.1320459842681885,
      "step": 4115
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 4115,
      "training_step_time": 9.905670881271362
    },
    {
      "epoch": 1.5697479248046876e-05,
      "model_forward_time": 3.1338694095611572,
      "step": 4115
    },
    {
      "epoch": 1.5697479248046876e-05,
      "step": 4115,
      "training_step_time": 10.288893222808838
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 3.148287057876587,
      "step": 4116
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 4116,
      "training_step_time": 9.929441452026367
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 3.1361007690429688,
      "step": 4116
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 4116,
      "training_step_time": 9.910280227661133
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 3.1388111114501953,
      "step": 4116
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 4116,
      "training_step_time": 10.199560642242432
    },
    {
      "epoch": 1.57012939453125e-05,
      "model_forward_time": 3.136665105819702,
      "step": 4116
    },
    {
      "epoch": 1.57012939453125e-05,
      "step": 4116,
      "training_step_time": 10.334204196929932
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 3.158480644226074,
      "step": 4117
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 4117,
      "training_step_time": 9.934414386749268
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 3.1442363262176514,
      "step": 4117
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 4117,
      "training_step_time": 9.917468070983887
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 3.1430232524871826,
      "step": 4117
    },
    {
      "dynamics_loss_avg": 0.05992935076355934,
      "epoch": 1.5705108642578125e-05,
      "step": 4117
    },
    {
      "action_loss_avg": 0.023321084771305324,
      "epoch": 1.5705108642578125e-05,
      "step": 4117
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 4117,
      "training_step_time": 9.92007064819336
    },
    {
      "epoch": 1.5705108642578125e-05,
      "model_forward_time": 3.458928108215332,
      "step": 4117
    },
    {
      "epoch": 1.5705108642578125e-05,
      "step": 4117,
      "training_step_time": 10.326485395431519
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 3.1513235569000244,
      "step": 4118
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 4118,
      "training_step_time": 9.926776885986328
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 3.1393847465515137,
      "step": 4118
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 4118,
      "training_step_time": 9.910004138946533
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 3.1448614597320557,
      "step": 4118
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 4118,
      "training_step_time": 9.919501543045044
    },
    {
      "epoch": 1.570892333984375e-05,
      "model_forward_time": 3.1374080181121826,
      "step": 4118
    },
    {
      "epoch": 1.570892333984375e-05,
      "step": 4118,
      "training_step_time": 10.313988447189331
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 3.1439802646636963,
      "step": 4119
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 4119,
      "training_step_time": 10.201590776443481
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 3.135526180267334,
      "step": 4119
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 4119,
      "training_step_time": 9.90645170211792
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 3.141536235809326,
      "step": 4119
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 4119,
      "training_step_time": 9.913886785507202
    },
    {
      "epoch": 1.5712738037109374e-05,
      "model_forward_time": 3.1378135681152344,
      "step": 4119
    },
    {
      "epoch": 1.5712738037109374e-05,
      "step": 4119,
      "training_step_time": 10.035059928894043
    },
    {
      "epoch": 1.5716552734375e-05,
      "grad_norm": 0.7380896806716919,
      "learning_rate": 1e-05,
      "loss": 0.0909,
      "step": 4120
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 3.146038770675659,
      "step": 4120
    },
    {
      "dynamics_loss_avg": 0.06260819006711245,
      "epoch": 1.5716552734375e-05,
      "step": 4120
    },
    {
      "action_loss_avg": 0.029680885514244436,
      "epoch": 1.5716552734375e-05,
      "step": 4120
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 4120,
      "training_step_time": 9.914574146270752
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 3.395465850830078,
      "step": 4120
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 4120,
      "training_step_time": 10.167460680007935
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 3.143094778060913,
      "step": 4120
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 4120,
      "training_step_time": 9.914823770523071
    },
    {
      "epoch": 1.5716552734375e-05,
      "model_forward_time": 3.1405818462371826,
      "step": 4120
    },
    {
      "epoch": 1.5716552734375e-05,
      "step": 4120,
      "training_step_time": 10.350465059280396
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 3.1572659015655518,
      "step": 4121
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 4121,
      "training_step_time": 9.929237365722656
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 3.143237590789795,
      "step": 4121
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 4121,
      "training_step_time": 9.907780647277832
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 3.4203248023986816,
      "step": 4121
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 4121,
      "training_step_time": 10.187690258026123
    },
    {
      "epoch": 1.5720367431640627e-05,
      "model_forward_time": 3.1402087211608887,
      "step": 4121
    },
    {
      "epoch": 1.5720367431640627e-05,
      "step": 4121,
      "training_step_time": 10.325801134109497
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 3.1399669647216797,
      "step": 4122
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 4122,
      "training_step_time": 9.907506704330444
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 3.1378746032714844,
      "step": 4122
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 4122,
      "training_step_time": 9.909718990325928
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 3.133683681488037,
      "step": 4122
    },
    {
      "dynamics_loss_avg": 0.07870894260704517,
      "epoch": 1.572418212890625e-05,
      "step": 4122
    },
    {
      "action_loss_avg": 0.03757706042379141,
      "epoch": 1.572418212890625e-05,
      "step": 4122
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 4122,
      "training_step_time": 9.910089492797852
    },
    {
      "epoch": 1.572418212890625e-05,
      "model_forward_time": 3.3849358558654785,
      "step": 4122
    },
    {
      "epoch": 1.572418212890625e-05,
      "step": 4122,
      "training_step_time": 10.35631513595581
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 3.1388237476348877,
      "step": 4123
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 4123,
      "training_step_time": 9.91445517539978
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 3.138744354248047,
      "step": 4123
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 4123,
      "training_step_time": 9.90946888923645
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 3.135291576385498,
      "step": 4123
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 4123,
      "training_step_time": 9.90522313117981
    },
    {
      "epoch": 1.5727996826171876e-05,
      "model_forward_time": 3.1376147270202637,
      "step": 4123
    },
    {
      "epoch": 1.5727996826171876e-05,
      "step": 4123,
      "training_step_time": 10.299099206924438
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 3.4225330352783203,
      "step": 4124
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 4124,
      "training_step_time": 10.190009593963623
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 3.1351535320281982,
      "step": 4124
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 4124,
      "training_step_time": 9.909733057022095
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 3.1314187049865723,
      "step": 4124
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 4124,
      "training_step_time": 9.904984951019287
    },
    {
      "epoch": 1.57318115234375e-05,
      "model_forward_time": 3.1330721378326416,
      "step": 4124
    },
    {
      "epoch": 1.57318115234375e-05,
      "step": 4124,
      "training_step_time": 10.30770754814148
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 3.155648946762085,
      "step": 4125
    },
    {
      "dynamics_loss_avg": 0.05866601523011923,
      "epoch": 1.5735626220703125e-05,
      "step": 4125
    },
    {
      "action_loss_avg": 0.02538755489513278,
      "epoch": 1.5735626220703125e-05,
      "step": 4125
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 4125,
      "training_step_time": 9.950568914413452
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 3.145020008087158,
      "step": 4125
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 4125,
      "training_step_time": 10.200817584991455
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 3.147700548171997,
      "step": 4125
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 4125,
      "training_step_time": 9.919982433319092
    },
    {
      "epoch": 1.5735626220703125e-05,
      "model_forward_time": 3.146763801574707,
      "step": 4125
    },
    {
      "epoch": 1.5735626220703125e-05,
      "step": 4125,
      "training_step_time": 10.28866195678711
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 3.157792568206787,
      "step": 4126
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 4126,
      "training_step_time": 9.932852268218994
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 3.145406723022461,
      "step": 4126
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 4126,
      "training_step_time": 9.915555953979492
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 2.4406635761260986,
      "step": 4126
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 4126,
      "training_step_time": 7.466178894042969
    },
    {
      "epoch": 1.573944091796875e-05,
      "model_forward_time": 3.4527883529663086,
      "step": 4126
    },
    {
      "epoch": 1.573944091796875e-05,
      "step": 4126,
      "training_step_time": 10.322346687316895
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 3.1568286418914795,
      "step": 4127
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 4127,
      "training_step_time": 9.9309720993042
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 3.14467453956604,
      "step": 4127
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 4127,
      "training_step_time": 9.91169023513794
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 3.148909568786621,
      "step": 4127
    },
    {
      "dynamics_loss_avg": 0.06574470400810242,
      "epoch": 1.5743255615234374e-05,
      "step": 4127
    },
    {
      "action_loss_avg": 0.037024219334125516,
      "epoch": 1.5743255615234374e-05,
      "step": 4127
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 4127,
      "training_step_time": 9.922909021377563
    },
    {
      "epoch": 1.5743255615234374e-05,
      "model_forward_time": 3.1405041217803955,
      "step": 4127
    },
    {
      "epoch": 1.5743255615234374e-05,
      "step": 4127,
      "training_step_time": 10.296196222305298
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 3.165604591369629,
      "step": 4128
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 4128,
      "training_step_time": 9.94264554977417
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 3.145603656768799,
      "step": 4128
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 4128,
      "training_step_time": 9.918543577194214
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 3.149216651916504,
      "step": 4128
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 4128,
      "training_step_time": 9.923133373260498
    },
    {
      "epoch": 1.57470703125e-05,
      "model_forward_time": 3.150573968887329,
      "step": 4128
    },
    {
      "epoch": 1.57470703125e-05,
      "step": 4128,
      "training_step_time": 10.291321277618408
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 3.4539971351623535,
      "step": 4129
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 4129,
      "training_step_time": 10.228461980819702
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 3.1478044986724854,
      "step": 4129
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 4129,
      "training_step_time": 9.926400184631348
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 3.148693561553955,
      "step": 4129
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 4129,
      "training_step_time": 9.923511028289795
    },
    {
      "epoch": 1.5750885009765623e-05,
      "model_forward_time": 3.144594192504883,
      "step": 4129
    },
    {
      "epoch": 1.5750885009765623e-05,
      "step": 4129,
      "training_step_time": 10.022956609725952
    },
    {
      "epoch": 1.575469970703125e-05,
      "grad_norm": 0.6946805715560913,
      "learning_rate": 1e-05,
      "loss": 0.0891,
      "step": 4130
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 3.153028726577759,
      "step": 4130
    },
    {
      "dynamics_loss_avg": 0.06565366256982089,
      "epoch": 1.575469970703125e-05,
      "step": 4130
    },
    {
      "action_loss_avg": 0.0362924637272954,
      "epoch": 1.575469970703125e-05,
      "step": 4130
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 4130,
      "training_step_time": 9.934226512908936
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 3.149418830871582,
      "step": 4130
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 4130,
      "training_step_time": 10.182883977890015
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 3.1520771980285645,
      "step": 4130
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 4130,
      "training_step_time": 9.92690110206604
    },
    {
      "epoch": 1.575469970703125e-05,
      "model_forward_time": 3.149031639099121,
      "step": 4130
    },
    {
      "epoch": 1.575469970703125e-05,
      "step": 4130,
      "training_step_time": 10.027122259140015
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 3.154541492462158,
      "step": 4131
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 4131,
      "training_step_time": 9.937114238739014
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 3.1502792835235596,
      "step": 4131
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 4131,
      "training_step_time": 9.927611589431763
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 3.4284963607788086,
      "step": 4131
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 4131,
      "training_step_time": 10.204562187194824
    },
    {
      "epoch": 1.5758514404296876e-05,
      "model_forward_time": 3.1492724418640137,
      "step": 4131
    },
    {
      "epoch": 1.5758514404296876e-05,
      "step": 4131,
      "training_step_time": 10.291083812713623
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 3.170961380004883,
      "step": 4132
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 4132,
      "training_step_time": 9.952516317367554
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 3.140402317047119,
      "step": 4132
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 4132,
      "training_step_time": 9.919834852218628
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 3.139023780822754,
      "step": 4132
    },
    {
      "dynamics_loss_avg": 0.05680184513330459,
      "epoch": 1.57623291015625e-05,
      "step": 4132
    },
    {
      "action_loss_avg": 0.02496769721619785,
      "epoch": 1.57623291015625e-05,
      "step": 4132
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 4132,
      "training_step_time": 9.915754795074463
    },
    {
      "epoch": 1.57623291015625e-05,
      "model_forward_time": 3.4010720252990723,
      "step": 4132
    },
    {
      "epoch": 1.57623291015625e-05,
      "step": 4132,
      "training_step_time": 10.350303411483765
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 3.1659867763519287,
      "step": 4133
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 4133,
      "training_step_time": 9.957715511322021
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 3.1486592292785645,
      "step": 4133
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 4133,
      "training_step_time": 9.927099704742432
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 3.14678692817688,
      "step": 4133
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 4133,
      "training_step_time": 9.926230669021606
    },
    {
      "epoch": 1.5766143798828125e-05,
      "model_forward_time": 3.1490399837493896,
      "step": 4133
    },
    {
      "epoch": 1.5766143798828125e-05,
      "step": 4133,
      "training_step_time": 10.24889612197876
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 3.4156131744384766,
      "step": 4134
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 4134,
      "training_step_time": 10.193162679672241
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 3.1509323120117188,
      "step": 4134
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 4134,
      "training_step_time": 9.927939653396606
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 3.1403555870056152,
      "step": 4134
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 4134,
      "training_step_time": 9.915548086166382
    },
    {
      "epoch": 1.576995849609375e-05,
      "model_forward_time": 3.1427218914031982,
      "step": 4134
    },
    {
      "epoch": 1.576995849609375e-05,
      "step": 4134,
      "training_step_time": 10.292284727096558
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 3.1605052947998047,
      "step": 4135
    },
    {
      "dynamics_loss_avg": 0.07706210762262344,
      "epoch": 1.5773773193359374e-05,
      "step": 4135
    },
    {
      "action_loss_avg": 0.037866994924843314,
      "epoch": 1.5773773193359374e-05,
      "step": 4135
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 4135,
      "training_step_time": 9.944323539733887
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 3.4650654792785645,
      "step": 4135
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 4135,
      "training_step_time": 10.24351978302002
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 3.1400012969970703,
      "step": 4135
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 4135,
      "training_step_time": 9.914713621139526
    },
    {
      "epoch": 1.5773773193359374e-05,
      "model_forward_time": 3.147745370864868,
      "step": 4135
    },
    {
      "epoch": 1.5773773193359374e-05,
      "step": 4135,
      "training_step_time": 10.338748455047607
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 2.4506778717041016,
      "step": 4136
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 4136,
      "training_step_time": 7.485058069229126
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 3.1433279514312744,
      "step": 4136
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 4136,
      "training_step_time": 9.925041913986206
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 3.1478400230407715,
      "step": 4136
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 4136,
      "training_step_time": 10.205825090408325
    },
    {
      "epoch": 1.5777587890625e-05,
      "model_forward_time": 3.1480016708374023,
      "step": 4136
    },
    {
      "epoch": 1.5777587890625e-05,
      "step": 4136,
      "training_step_time": 10.027727365493774
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 3.1632728576660156,
      "step": 4137
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 4137,
      "training_step_time": 9.943312406539917
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 3.143455982208252,
      "step": 4137
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 4137,
      "training_step_time": 9.921291589736938
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 3.145678758621216,
      "step": 4137
    },
    {
      "dynamics_loss_avg": 0.052414466999471186,
      "epoch": 1.5781402587890623e-05,
      "step": 4137
    },
    {
      "action_loss_avg": 0.023484402941539882,
      "epoch": 1.5781402587890623e-05,
      "step": 4137
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 4137,
      "training_step_time": 9.930113077163696
    },
    {
      "epoch": 1.5781402587890623e-05,
      "model_forward_time": 3.437711715698242,
      "step": 4137
    },
    {
      "epoch": 1.5781402587890623e-05,
      "step": 4137,
      "training_step_time": 10.319126844406128
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 3.167396068572998,
      "step": 4138
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 4138,
      "training_step_time": 9.94688367843628
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 3.1429972648620605,
      "step": 4138
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 4138,
      "training_step_time": 9.918458938598633
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 3.1460795402526855,
      "step": 4138
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 4138,
      "training_step_time": 9.919625282287598
    },
    {
      "epoch": 1.578521728515625e-05,
      "model_forward_time": 3.1420037746429443,
      "step": 4138
    },
    {
      "epoch": 1.578521728515625e-05,
      "step": 4138,
      "training_step_time": 10.027050971984863
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 3.1489603519439697,
      "step": 4139
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 4139,
      "training_step_time": 10.200780630111694
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 3.139036178588867,
      "step": 4139
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 4139,
      "training_step_time": 9.912776470184326
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 3.13954758644104,
      "step": 4139
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 4139,
      "training_step_time": 9.905691623687744
    },
    {
      "epoch": 1.5789031982421876e-05,
      "model_forward_time": 2.4362897872924805,
      "step": 4139
    },
    {
      "epoch": 1.5789031982421876e-05,
      "step": 4139,
      "training_step_time": 10.289172410964966
    },
    {
      "epoch": 1.57928466796875e-05,
      "grad_norm": 0.5127875208854675,
      "learning_rate": 1e-05,
      "loss": 0.0902,
      "step": 4140
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 3.151179075241089,
      "step": 4140
    },
    {
      "dynamics_loss_avg": 0.06786506697535515,
      "epoch": 1.57928466796875e-05,
      "step": 4140
    },
    {
      "action_loss_avg": 0.025472524762153625,
      "epoch": 1.57928466796875e-05,
      "step": 4140
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 4140,
      "training_step_time": 9.926419496536255
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 3.4108517169952393,
      "step": 4140
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 4140,
      "training_step_time": 10.176482200622559
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 3.139943838119507,
      "step": 4140
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 4140,
      "training_step_time": 9.908945322036743
    },
    {
      "epoch": 1.57928466796875e-05,
      "model_forward_time": 3.138601064682007,
      "step": 4140
    },
    {
      "epoch": 1.57928466796875e-05,
      "step": 4140,
      "training_step_time": 10.01919937133789
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 3.143965005874634,
      "step": 4141
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 4141,
      "training_step_time": 9.914491415023804
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 3.143263816833496,
      "step": 4141
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 4141,
      "training_step_time": 9.91684889793396
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 3.1412901878356934,
      "step": 4141
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 4141,
      "training_step_time": 10.174420833587646
    },
    {
      "epoch": 1.5796661376953125e-05,
      "model_forward_time": 3.14483904838562,
      "step": 4141
    },
    {
      "epoch": 1.5796661376953125e-05,
      "step": 4141,
      "training_step_time": 10.274909734725952
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 3.154202699661255,
      "step": 4142
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 4142,
      "training_step_time": 9.937616348266602
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 3.1485159397125244,
      "step": 4142
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 4142,
      "training_step_time": 9.926881551742554
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 3.1442017555236816,
      "step": 4142
    },
    {
      "dynamics_loss_avg": 0.07322795186191797,
      "epoch": 1.580047607421875e-05,
      "step": 4142
    },
    {
      "action_loss_avg": 0.02999641573987901,
      "epoch": 1.580047607421875e-05,
      "step": 4142
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 4142,
      "training_step_time": 9.924959182739258
    },
    {
      "epoch": 1.580047607421875e-05,
      "model_forward_time": 3.409196376800537,
      "step": 4142
    },
    {
      "epoch": 1.580047607421875e-05,
      "step": 4142,
      "training_step_time": 10.289671659469604
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 3.181670904159546,
      "step": 4143
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 4143,
      "training_step_time": 9.959411144256592
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 3.1471736431121826,
      "step": 4143
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 4143,
      "training_step_time": 9.927967309951782
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 3.147395133972168,
      "step": 4143
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 4143,
      "training_step_time": 9.928672790527344
    },
    {
      "epoch": 1.5804290771484374e-05,
      "model_forward_time": 3.1484193801879883,
      "step": 4143
    },
    {
      "epoch": 1.5804290771484374e-05,
      "step": 4143,
      "training_step_time": 10.35512900352478
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 3.4439284801483154,
      "step": 4144
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 4144,
      "training_step_time": 10.219290018081665
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 3.1464486122131348,
      "step": 4144
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 4144,
      "training_step_time": 9.924137353897095
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 3.141296863555908,
      "step": 4144
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 4144,
      "training_step_time": 9.918346643447876
    },
    {
      "epoch": 1.580810546875e-05,
      "model_forward_time": 3.14697265625,
      "step": 4144
    },
    {
      "epoch": 1.580810546875e-05,
      "step": 4144,
      "training_step_time": 10.295155763626099
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 3.150134801864624,
      "step": 4145
    },
    {
      "dynamics_loss_avg": 0.07977025136351586,
      "epoch": 1.5811920166015623e-05,
      "step": 4145
    },
    {
      "action_loss_avg": 0.04019185472279787,
      "epoch": 1.5811920166015623e-05,
      "step": 4145
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 4145,
      "training_step_time": 9.930761575698853
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 3.44016432762146,
      "step": 4145
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 4145,
      "training_step_time": 10.219086408615112
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 3.145442247390747,
      "step": 4145
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 4145,
      "training_step_time": 9.923320531845093
    },
    {
      "epoch": 1.5811920166015623e-05,
      "model_forward_time": 3.1459269523620605,
      "step": 4145
    },
    {
      "epoch": 1.5811920166015623e-05,
      "step": 4145,
      "training_step_time": 10.314305543899536
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 3.1578896045684814,
      "step": 4146
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 4146,
      "training_step_time": 9.935575485229492
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 3.1436219215393066,
      "step": 4146
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 4146,
      "training_step_time": 9.917514085769653
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 3.4128451347351074,
      "step": 4146
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 4146,
      "training_step_time": 10.18905234336853
    },
    {
      "epoch": 1.581573486328125e-05,
      "model_forward_time": 3.149146556854248,
      "step": 4146
    },
    {
      "epoch": 1.581573486328125e-05,
      "step": 4146,
      "training_step_time": 10.035358667373657
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 3.1568644046783447,
      "step": 4147
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 4147,
      "training_step_time": 9.93491268157959
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 3.1494219303131104,
      "step": 4147
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 4147,
      "training_step_time": 9.92568826675415
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 3.1502230167388916,
      "step": 4147
    },
    {
      "dynamics_loss_avg": 0.05282139889895916,
      "epoch": 1.5819549560546876e-05,
      "step": 4147
    },
    {
      "action_loss_avg": 0.02133239237591624,
      "epoch": 1.5819549560546876e-05,
      "step": 4147
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 4147,
      "training_step_time": 9.942225933074951
    },
    {
      "epoch": 1.5819549560546876e-05,
      "model_forward_time": 3.432217597961426,
      "step": 4147
    },
    {
      "epoch": 1.5819549560546876e-05,
      "step": 4147,
      "training_step_time": 10.313338041305542
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 3.164773941040039,
      "step": 4148
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 4148,
      "training_step_time": 9.951514482498169
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 3.147916793823242,
      "step": 4148
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 4148,
      "training_step_time": 9.93027400970459
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 3.1522650718688965,
      "step": 4148
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 4148,
      "training_step_time": 9.9319748878479
    },
    {
      "epoch": 1.58233642578125e-05,
      "model_forward_time": 3.149148941040039,
      "step": 4148
    },
    {
      "epoch": 1.58233642578125e-05,
      "step": 4148,
      "training_step_time": 10.337682962417603
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 3.4484617710113525,
      "step": 4149
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 4149,
      "training_step_time": 10.231509923934937
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 3.14265775680542,
      "step": 4149
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 4149,
      "training_step_time": 9.919265985488892
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 3.146569013595581,
      "step": 4149
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 4149,
      "training_step_time": 9.926491498947144
    },
    {
      "epoch": 1.5827178955078125e-05,
      "model_forward_time": 3.143602132797241,
      "step": 4149
    },
    {
      "epoch": 1.5827178955078125e-05,
      "step": 4149,
      "training_step_time": 10.285462141036987
    },
    {
      "epoch": 1.583099365234375e-05,
      "grad_norm": 0.5178654789924622,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 4150
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 3.1509480476379395,
      "step": 4150
    },
    {
      "dynamics_loss_avg": 0.0600857887417078,
      "epoch": 1.583099365234375e-05,
      "step": 4150
    },
    {
      "action_loss_avg": 0.026067951507866383,
      "epoch": 1.583099365234375e-05,
      "step": 4150
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 4150,
      "training_step_time": 9.929673433303833
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 2.694594621658325,
      "step": 4150
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 4150,
      "training_step_time": 7.731760025024414
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 3.1380386352539062,
      "step": 4150
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 4150,
      "training_step_time": 9.914090394973755
    },
    {
      "epoch": 1.583099365234375e-05,
      "model_forward_time": 3.137415885925293,
      "step": 4150
    },
    {
      "epoch": 1.583099365234375e-05,
      "step": 4150,
      "training_step_time": 10.048951625823975
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 3.1653332710266113,
      "step": 4151
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 4151,
      "training_step_time": 9.954374074935913
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 3.1514229774475098,
      "step": 4151
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 4151,
      "training_step_time": 9.930250406265259
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 3.425924301147461,
      "step": 4151
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 4151,
      "training_step_time": 10.207365989685059
    },
    {
      "epoch": 1.5834808349609374e-05,
      "model_forward_time": 3.1442294120788574,
      "step": 4151
    },
    {
      "epoch": 1.5834808349609374e-05,
      "step": 4151,
      "training_step_time": 10.276995182037354
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 3.155947208404541,
      "step": 4152
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 4152,
      "training_step_time": 9.9387845993042
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 3.138822078704834,
      "step": 4152
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 4152,
      "training_step_time": 9.919672966003418
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 3.1345574855804443,
      "step": 4152
    },
    {
      "dynamics_loss_avg": 0.07657215539366007,
      "epoch": 1.5838623046875e-05,
      "step": 4152
    },
    {
      "action_loss_avg": 0.0315992671996355,
      "epoch": 1.5838623046875e-05,
      "step": 4152
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 4152,
      "training_step_time": 9.912438154220581
    },
    {
      "epoch": 1.5838623046875e-05,
      "model_forward_time": 3.428901433944702,
      "step": 4152
    },
    {
      "epoch": 1.5838623046875e-05,
      "step": 4152,
      "training_step_time": 10.300556421279907
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 3.1372954845428467,
      "step": 4153
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 4153,
      "training_step_time": 9.916664600372314
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 3.1347291469573975,
      "step": 4153
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 4153,
      "training_step_time": 9.909820079803467
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 3.135698080062866,
      "step": 4153
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 4153,
      "training_step_time": 9.908757448196411
    },
    {
      "epoch": 1.5842437744140623e-05,
      "model_forward_time": 3.134223222732544,
      "step": 4153
    },
    {
      "epoch": 1.5842437744140623e-05,
      "step": 4153,
      "training_step_time": 10.292737007141113
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 3.4062206745147705,
      "step": 4154
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 4154,
      "training_step_time": 10.179465532302856
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 3.133366107940674,
      "step": 4154
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 4154,
      "training_step_time": 9.910677433013916
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 3.1320223808288574,
      "step": 4154
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 4154,
      "training_step_time": 9.903717517852783
    },
    {
      "epoch": 1.584625244140625e-05,
      "model_forward_time": 3.13457989692688,
      "step": 4154
    },
    {
      "epoch": 1.584625244140625e-05,
      "step": 4154,
      "training_step_time": 10.272499799728394
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 3.144366979598999,
      "step": 4155
    },
    {
      "dynamics_loss_avg": 0.06419739741832017,
      "epoch": 1.5850067138671876e-05,
      "step": 4155
    },
    {
      "action_loss_avg": 0.029818890616297722,
      "epoch": 1.5850067138671876e-05,
      "step": 4155
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 4155,
      "training_step_time": 9.922889709472656
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 3.424161434173584,
      "step": 4155
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 4155,
      "training_step_time": 10.19651985168457
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 3.142648458480835,
      "step": 4155
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 4155,
      "training_step_time": 9.919775247573853
    },
    {
      "epoch": 1.5850067138671876e-05,
      "model_forward_time": 2.43837833404541,
      "step": 4155
    },
    {
      "epoch": 1.5850067138671876e-05,
      "step": 4155,
      "training_step_time": 10.020736932754517
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 3.157801389694214,
      "step": 4156
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 4156,
      "training_step_time": 9.935354709625244
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 3.1393656730651855,
      "step": 4156
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 4156,
      "training_step_time": 9.910999774932861
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 3.156996488571167,
      "step": 4156
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 4156,
      "training_step_time": 10.206042528152466
    },
    {
      "epoch": 1.58538818359375e-05,
      "model_forward_time": 3.1427206993103027,
      "step": 4156
    },
    {
      "epoch": 1.58538818359375e-05,
      "step": 4156,
      "training_step_time": 10.018513441085815
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 3.1559510231018066,
      "step": 4157
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 4157,
      "training_step_time": 9.932168006896973
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 3.144359588623047,
      "step": 4157
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 4157,
      "training_step_time": 9.925302743911743
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 3.148524284362793,
      "step": 4157
    },
    {
      "dynamics_loss_avg": 0.06516106966882944,
      "epoch": 1.5857696533203125e-05,
      "step": 4157
    },
    {
      "action_loss_avg": 0.02347197257913649,
      "epoch": 1.5857696533203125e-05,
      "step": 4157
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 4157,
      "training_step_time": 9.93239450454712
    },
    {
      "epoch": 1.5857696533203125e-05,
      "model_forward_time": 3.4286587238311768,
      "step": 4157
    },
    {
      "epoch": 1.5857696533203125e-05,
      "step": 4157,
      "training_step_time": 10.303918361663818
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 3.1567914485931396,
      "step": 4158
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 4158,
      "training_step_time": 9.93920111656189
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 3.1455962657928467,
      "step": 4158
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 4158,
      "training_step_time": 9.923338890075684
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 3.1490676403045654,
      "step": 4158
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 4158,
      "training_step_time": 9.924168825149536
    },
    {
      "epoch": 1.586151123046875e-05,
      "model_forward_time": 3.1438143253326416,
      "step": 4158
    },
    {
      "epoch": 1.586151123046875e-05,
      "step": 4158,
      "training_step_time": 10.335707902908325
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 3.1405889987945557,
      "step": 4159
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 4159,
      "training_step_time": 10.198209762573242
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 3.1355950832366943,
      "step": 4159
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 4159,
      "training_step_time": 9.912341356277466
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 3.135279655456543,
      "step": 4159
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 4159,
      "training_step_time": 9.910224199295044
    },
    {
      "epoch": 1.5865325927734374e-05,
      "model_forward_time": 3.1348350048065186,
      "step": 4159
    },
    {
      "epoch": 1.5865325927734374e-05,
      "step": 4159,
      "training_step_time": 10.25203561782837
    },
    {
      "epoch": 1.5869140625e-05,
      "grad_norm": 0.5041714906692505,
      "learning_rate": 1e-05,
      "loss": 0.0898,
      "step": 4160
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 3.1516048908233643,
      "step": 4160
    },
    {
      "dynamics_loss_avg": 0.06642638593912124,
      "epoch": 1.5869140625e-05,
      "step": 4160
    },
    {
      "action_loss_avg": 0.030284149665385486,
      "epoch": 1.5869140625e-05,
      "step": 4160
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 4160,
      "training_step_time": 9.934825658798218
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 3.38905930519104,
      "step": 4160
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 4160,
      "training_step_time": 10.166232824325562
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 3.1388967037200928,
      "step": 4160
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 4160,
      "training_step_time": 9.916020631790161
    },
    {
      "epoch": 1.5869140625e-05,
      "model_forward_time": 3.136385202407837,
      "step": 4160
    },
    {
      "epoch": 1.5869140625e-05,
      "step": 4160,
      "training_step_time": 10.332082748413086
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 3.143571615219116,
      "step": 4161
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 4161,
      "training_step_time": 9.924832105636597
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 3.1387624740600586,
      "step": 4161
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 4161,
      "training_step_time": 9.918680906295776
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 3.420928716659546,
      "step": 4161
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 4161,
      "training_step_time": 10.205249786376953
    },
    {
      "epoch": 1.5872955322265623e-05,
      "model_forward_time": 3.138373613357544,
      "step": 4161
    },
    {
      "epoch": 1.5872955322265623e-05,
      "step": 4161,
      "training_step_time": 10.028231859207153
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 3.1503448486328125,
      "step": 4162
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 4162,
      "training_step_time": 9.925888538360596
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 3.135568141937256,
      "step": 4162
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 4162,
      "training_step_time": 9.912167072296143
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 3.132838726043701,
      "step": 4162
    },
    {
      "dynamics_loss_avg": 0.06041271425783634,
      "epoch": 1.587677001953125e-05,
      "step": 4162
    },
    {
      "action_loss_avg": 0.02981798518449068,
      "epoch": 1.587677001953125e-05,
      "step": 4162
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 4162,
      "training_step_time": 9.914192914962769
    },
    {
      "epoch": 1.587677001953125e-05,
      "model_forward_time": 3.422816276550293,
      "step": 4162
    },
    {
      "epoch": 1.587677001953125e-05,
      "step": 4162,
      "training_step_time": 10.299036741256714
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 3.1472020149230957,
      "step": 4163
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 4163,
      "training_step_time": 9.93488621711731
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 3.140126943588257,
      "step": 4163
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 4163,
      "training_step_time": 9.924474477767944
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 3.135854482650757,
      "step": 4163
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 4163,
      "training_step_time": 9.918385744094849
    },
    {
      "epoch": 1.5880584716796876e-05,
      "model_forward_time": 3.138171672821045,
      "step": 4163
    },
    {
      "epoch": 1.5880584716796876e-05,
      "step": 4163,
      "training_step_time": 10.298713445663452
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 3.408923387527466,
      "step": 4164
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 4164,
      "training_step_time": 10.186129570007324
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 3.1322121620178223,
      "step": 4164
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 4164,
      "training_step_time": 9.913054704666138
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 3.136995315551758,
      "step": 4164
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 4164,
      "training_step_time": 9.916757822036743
    },
    {
      "epoch": 1.58843994140625e-05,
      "model_forward_time": 3.135981559753418,
      "step": 4164
    },
    {
      "epoch": 1.58843994140625e-05,
      "step": 4164,
      "training_step_time": 10.036010026931763
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 3.1483566761016846,
      "step": 4165
    },
    {
      "dynamics_loss_avg": 0.06500022429972888,
      "epoch": 1.5888214111328125e-05,
      "step": 4165
    },
    {
      "action_loss_avg": 0.02772109480574727,
      "epoch": 1.5888214111328125e-05,
      "step": 4165
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 4165,
      "training_step_time": 9.931769132614136
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 3.138115406036377,
      "step": 4165
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 4165,
      "training_step_time": 10.201048135757446
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 3.1358587741851807,
      "step": 4165
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 4165,
      "training_step_time": 9.917619228363037
    },
    {
      "epoch": 1.5888214111328125e-05,
      "model_forward_time": 2.4391791820526123,
      "step": 4165
    },
    {
      "epoch": 1.5888214111328125e-05,
      "step": 4165,
      "training_step_time": 10.29141902923584
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 3.146031379699707,
      "step": 4166
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 4166,
      "training_step_time": 9.926612854003906
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 3.137913465499878,
      "step": 4166
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 4166,
      "training_step_time": 9.922749757766724
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 3.1400022506713867,
      "step": 4166
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 4166,
      "training_step_time": 9.917046785354614
    },
    {
      "epoch": 1.589202880859375e-05,
      "model_forward_time": 3.4088022708892822,
      "step": 4166
    },
    {
      "epoch": 1.589202880859375e-05,
      "step": 4166,
      "training_step_time": 10.284414291381836
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 3.1527318954467773,
      "step": 4167
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 4167,
      "training_step_time": 9.941336631774902
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 3.137953042984009,
      "step": 4167
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 4167,
      "training_step_time": 9.917538404464722
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 3.1390271186828613,
      "step": 4167
    },
    {
      "dynamics_loss_avg": 0.06368431504815816,
      "epoch": 1.5895843505859374e-05,
      "step": 4167
    },
    {
      "action_loss_avg": 0.02858527209609747,
      "epoch": 1.5895843505859374e-05,
      "step": 4167
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 4167,
      "training_step_time": 9.917521238327026
    },
    {
      "epoch": 1.5895843505859374e-05,
      "model_forward_time": 3.1380889415740967,
      "step": 4167
    },
    {
      "epoch": 1.5895843505859374e-05,
      "step": 4167,
      "training_step_time": 10.28249192237854
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 3.157879114151001,
      "step": 4168
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 4168,
      "training_step_time": 9.95088791847229
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 2.4355623722076416,
      "step": 4168
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 4168,
      "training_step_time": 7.469053268432617
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 3.138148307800293,
      "step": 4168
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 4168,
      "training_step_time": 9.919013977050781
    },
    {
      "epoch": 1.5899658203125e-05,
      "model_forward_time": 3.1356377601623535,
      "step": 4168
    },
    {
      "epoch": 1.5899658203125e-05,
      "step": 4168,
      "training_step_time": 10.29295802116394
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 3.423330307006836,
      "step": 4169
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 4169,
      "training_step_time": 10.20185923576355
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 3.1368203163146973,
      "step": 4169
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 4169,
      "training_step_time": 9.918673753738403
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 3.1419200897216797,
      "step": 4169
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 4169,
      "training_step_time": 9.921267032623291
    },
    {
      "epoch": 1.5903472900390623e-05,
      "model_forward_time": 3.1382181644439697,
      "step": 4169
    },
    {
      "epoch": 1.5903472900390623e-05,
      "step": 4169,
      "training_step_time": 10.040743112564087
    },
    {
      "epoch": 1.590728759765625e-05,
      "grad_norm": 0.5876103043556213,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 4170
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 2.4434738159179688,
      "step": 4170
    },
    {
      "dynamics_loss_avg": 0.06339239701628685,
      "epoch": 1.590728759765625e-05,
      "step": 4170
    },
    {
      "action_loss_avg": 0.02883139136247337,
      "epoch": 1.590728759765625e-05,
      "step": 4170
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 4170,
      "training_step_time": 7.478825330734253
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 3.1328158378601074,
      "step": 4170
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 4170,
      "training_step_time": 9.913092613220215
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 3.4066531658172607,
      "step": 4170
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 4170,
      "training_step_time": 10.18108582496643
    },
    {
      "epoch": 1.590728759765625e-05,
      "model_forward_time": 3.1489200592041016,
      "step": 4170
    },
    {
      "epoch": 1.590728759765625e-05,
      "step": 4170,
      "training_step_time": 10.280020475387573
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 3.139937400817871,
      "step": 4171
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 4171,
      "training_step_time": 9.919003009796143
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 3.1386232376098633,
      "step": 4171
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 4171,
      "training_step_time": 9.919216632843018
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 3.420706272125244,
      "step": 4171
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 4171,
      "training_step_time": 10.190270185470581
    },
    {
      "epoch": 1.5911102294921876e-05,
      "model_forward_time": 3.151615858078003,
      "step": 4171
    },
    {
      "epoch": 1.5911102294921876e-05,
      "step": 4171,
      "training_step_time": 10.029240369796753
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 3.142078161239624,
      "step": 4172
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 4172,
      "training_step_time": 9.920801162719727
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 3.138817310333252,
      "step": 4172
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 4172,
      "training_step_time": 9.914066314697266
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 3.1340410709381104,
      "step": 4172
    },
    {
      "dynamics_loss_avg": 0.06638753712177277,
      "epoch": 1.59149169921875e-05,
      "step": 4172
    },
    {
      "action_loss_avg": 0.026570067554712296,
      "epoch": 1.59149169921875e-05,
      "step": 4172
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 4172,
      "training_step_time": 9.912413120269775
    },
    {
      "epoch": 1.59149169921875e-05,
      "model_forward_time": 3.135576009750366,
      "step": 4172
    },
    {
      "epoch": 1.59149169921875e-05,
      "step": 4172,
      "training_step_time": 10.023910999298096
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 3.416975736618042,
      "step": 4173
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 4173,
      "training_step_time": 10.193568468093872
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 3.138214111328125,
      "step": 4173
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 4173,
      "training_step_time": 9.918346643447876
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 3.1356770992279053,
      "step": 4173
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 4173,
      "training_step_time": 9.919878005981445
    },
    {
      "epoch": 1.5918731689453125e-05,
      "model_forward_time": 3.1400210857391357,
      "step": 4173
    },
    {
      "epoch": 1.5918731689453125e-05,
      "step": 4173,
      "training_step_time": 10.357291460037231
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 3.398244619369507,
      "step": 4174
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 4174,
      "training_step_time": 10.180843353271484
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 3.141108512878418,
      "step": 4174
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 4174,
      "training_step_time": 9.921730518341064
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 3.137422561645508,
      "step": 4174
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 4174,
      "training_step_time": 9.915939569473267
    },
    {
      "epoch": 1.592254638671875e-05,
      "model_forward_time": 3.1378698348999023,
      "step": 4174
    },
    {
      "epoch": 1.592254638671875e-05,
      "step": 4174,
      "training_step_time": 10.310641765594482
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 2.449079990386963,
      "step": 4175
    },
    {
      "dynamics_loss_avg": 0.05559723172336817,
      "epoch": 1.5926361083984374e-05,
      "step": 4175
    },
    {
      "action_loss_avg": 0.018797254608944058,
      "epoch": 1.5926361083984374e-05,
      "step": 4175
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 4175,
      "training_step_time": 7.484333515167236
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 2.4328835010528564,
      "step": 4175
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 4175,
      "training_step_time": 7.464282989501953
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 3.4214117527008057,
      "step": 4175
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 4175,
      "training_step_time": 10.20202898979187
    },
    {
      "epoch": 1.5926361083984374e-05,
      "model_forward_time": 3.146056652069092,
      "step": 4175
    },
    {
      "epoch": 1.5926361083984374e-05,
      "step": 4175,
      "training_step_time": 10.289481163024902
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 3.1456527709960938,
      "step": 4176
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 4176,
      "training_step_time": 9.92739224433899
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 3.1430461406707764,
      "step": 4176
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 4176,
      "training_step_time": 9.924607276916504
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 3.1438941955566406,
      "step": 4176
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 4176,
      "training_step_time": 9.926569938659668
    },
    {
      "epoch": 1.593017578125e-05,
      "model_forward_time": 3.4274396896362305,
      "step": 4176
    },
    {
      "epoch": 1.593017578125e-05,
      "step": 4176,
      "training_step_time": 10.307685136795044
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 3.184330940246582,
      "step": 4177
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 4177,
      "training_step_time": 9.97313904762268
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 2.4404733180999756,
      "step": 4177
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 4177,
      "training_step_time": 7.476803779602051
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 3.1430399417877197,
      "step": 4177
    },
    {
      "dynamics_loss_avg": 0.05604166146367788,
      "epoch": 1.5933990478515623e-05,
      "step": 4177
    },
    {
      "action_loss_avg": 0.019960025139153004,
      "epoch": 1.5933990478515623e-05,
      "step": 4177
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 4177,
      "training_step_time": 9.92734694480896
    },
    {
      "epoch": 1.5933990478515623e-05,
      "model_forward_time": 3.1440601348876953,
      "step": 4177
    },
    {
      "epoch": 1.5933990478515623e-05,
      "step": 4177,
      "training_step_time": 10.304887771606445
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 3.163593292236328,
      "step": 4178
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 4178,
      "training_step_time": 9.943143129348755
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 3.139356851577759,
      "step": 4178
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 4178,
      "training_step_time": 9.916041612625122
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 3.143436908721924,
      "step": 4178
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 4178,
      "training_step_time": 9.91870665550232
    },
    {
      "epoch": 1.593780517578125e-05,
      "model_forward_time": 3.137532949447632,
      "step": 4178
    },
    {
      "epoch": 1.593780517578125e-05,
      "step": 4178,
      "training_step_time": 10.005988359451294
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 3.1478872299194336,
      "step": 4179
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 4179,
      "training_step_time": 10.200761079788208
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 3.1311745643615723,
      "step": 4179
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 4179,
      "training_step_time": 9.903873682022095
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 3.152151107788086,
      "step": 4179
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 4179,
      "training_step_time": 9.926419019699097
    },
    {
      "epoch": 1.5941619873046876e-05,
      "model_forward_time": 3.1407718658447266,
      "step": 4179
    },
    {
      "epoch": 1.5941619873046876e-05,
      "step": 4179,
      "training_step_time": 10.303012132644653
    },
    {
      "epoch": 1.59454345703125e-05,
      "grad_norm": 0.624270498752594,
      "learning_rate": 1e-05,
      "loss": 0.0897,
      "step": 4180
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 3.1473333835601807,
      "step": 4180
    },
    {
      "dynamics_loss_avg": 0.07097199950367213,
      "epoch": 1.59454345703125e-05,
      "step": 4180
    },
    {
      "action_loss_avg": 0.024663320649415253,
      "epoch": 1.59454345703125e-05,
      "step": 4180
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 4180,
      "training_step_time": 9.925055503845215
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 3.4169840812683105,
      "step": 4180
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 4180,
      "training_step_time": 10.191951751708984
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 3.138162612915039,
      "step": 4180
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 4180,
      "training_step_time": 9.910221099853516
    },
    {
      "epoch": 1.59454345703125e-05,
      "model_forward_time": 2.432573080062866,
      "step": 4180
    },
    {
      "epoch": 1.59454345703125e-05,
      "step": 4180,
      "training_step_time": 10.269022941589355
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 3.1475367546081543,
      "step": 4181
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 4181,
      "training_step_time": 9.928614854812622
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 3.1356146335601807,
      "step": 4181
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 4181,
      "training_step_time": 9.91137957572937
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 3.135568141937256,
      "step": 4181
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 4181,
      "training_step_time": 9.901650667190552
    },
    {
      "epoch": 1.5949249267578125e-05,
      "model_forward_time": 3.422914743423462,
      "step": 4181
    },
    {
      "epoch": 1.5949249267578125e-05,
      "step": 4181,
      "training_step_time": 10.286731243133545
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 3.1519534587860107,
      "step": 4182
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 4182,
      "training_step_time": 9.926183223724365
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 3.138455390930176,
      "step": 4182
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 4182,
      "training_step_time": 9.911816835403442
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 3.141979932785034,
      "step": 4182
    },
    {
      "dynamics_loss_avg": 0.06050507165491581,
      "epoch": 1.595306396484375e-05,
      "step": 4182
    },
    {
      "action_loss_avg": 0.02270114468410611,
      "epoch": 1.595306396484375e-05,
      "step": 4182
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 4182,
      "training_step_time": 9.931752681732178
    },
    {
      "epoch": 1.595306396484375e-05,
      "model_forward_time": 3.434800148010254,
      "step": 4182
    },
    {
      "epoch": 1.595306396484375e-05,
      "step": 4182,
      "training_step_time": 10.318328380584717
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 3.1617321968078613,
      "step": 4183
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 4183,
      "training_step_time": 9.94109296798706
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 3.1406378746032715,
      "step": 4183
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 4183,
      "training_step_time": 9.91929578781128
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 3.137066602706909,
      "step": 4183
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 4183,
      "training_step_time": 9.912047386169434
    },
    {
      "epoch": 1.5956878662109374e-05,
      "model_forward_time": 3.138854503631592,
      "step": 4183
    },
    {
      "epoch": 1.5956878662109374e-05,
      "step": 4183,
      "training_step_time": 10.025795698165894
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 3.1522068977355957,
      "step": 4184
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 4184,
      "training_step_time": 10.215991020202637
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 3.144019603729248,
      "step": 4184
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 4184,
      "training_step_time": 9.924198627471924
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 3.138272285461426,
      "step": 4184
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 4184,
      "training_step_time": 9.916784763336182
    },
    {
      "epoch": 1.5960693359375e-05,
      "model_forward_time": 3.144308567047119,
      "step": 4184
    },
    {
      "epoch": 1.5960693359375e-05,
      "step": 4184,
      "training_step_time": 10.34935975074768
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 3.155609130859375,
      "step": 4185
    },
    {
      "dynamics_loss_avg": 0.04770263973623514,
      "epoch": 1.5964508056640624e-05,
      "step": 4185
    },
    {
      "action_loss_avg": 0.02327567096799612,
      "epoch": 1.5964508056640624e-05,
      "step": 4185
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 4185,
      "training_step_time": 9.943085432052612
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 3.471571683883667,
      "step": 4185
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 4185,
      "training_step_time": 10.25457501411438
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 3.143397569656372,
      "step": 4185
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 4185,
      "training_step_time": 9.919552087783813
    },
    {
      "epoch": 1.5964508056640624e-05,
      "model_forward_time": 3.142730474472046,
      "step": 4185
    },
    {
      "epoch": 1.5964508056640624e-05,
      "step": 4185,
      "training_step_time": 10.302800416946411
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 3.154780149459839,
      "step": 4186
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 4186,
      "training_step_time": 9.938390731811523
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 3.1359827518463135,
      "step": 4186
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 4186,
      "training_step_time": 9.903499126434326
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 3.3813893795013428,
      "step": 4186
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 4186,
      "training_step_time": 10.143633127212524
    },
    {
      "epoch": 1.596832275390625e-05,
      "model_forward_time": 3.13181209564209,
      "step": 4186
    },
    {
      "epoch": 1.596832275390625e-05,
      "step": 4186,
      "training_step_time": 10.289025068283081
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 3.139833927154541,
      "step": 4187
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 4187,
      "training_step_time": 9.926913976669312
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 3.1288068294525146,
      "step": 4187
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 4187,
      "training_step_time": 9.897600173950195
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 3.1302177906036377,
      "step": 4187
    },
    {
      "dynamics_loss_avg": 0.0517888817936182,
      "epoch": 1.5972137451171876e-05,
      "step": 4187
    },
    {
      "action_loss_avg": 0.02517217556014657,
      "epoch": 1.5972137451171876e-05,
      "step": 4187
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 4187,
      "training_step_time": 9.904207229614258
    },
    {
      "epoch": 1.5972137451171876e-05,
      "model_forward_time": 3.4525563716888428,
      "step": 4187
    },
    {
      "epoch": 1.5972137451171876e-05,
      "step": 4187,
      "training_step_time": 10.325130224227905
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 3.1406233310699463,
      "step": 4188
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 4188,
      "training_step_time": 9.917029857635498
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 3.1332716941833496,
      "step": 4188
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 4188,
      "training_step_time": 9.898508787155151
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 3.13555908203125,
      "step": 4188
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 4188,
      "training_step_time": 9.906226396560669
    },
    {
      "epoch": 1.59759521484375e-05,
      "model_forward_time": 3.136150360107422,
      "step": 4188
    },
    {
      "epoch": 1.59759521484375e-05,
      "step": 4188,
      "training_step_time": 10.019294500350952
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 3.4727227687835693,
      "step": 4189
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 4189,
      "training_step_time": 10.252654552459717
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 3.138007402420044,
      "step": 4189
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 4189,
      "training_step_time": 9.91265320777893
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 3.1412367820739746,
      "step": 4189
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 4189,
      "training_step_time": 9.930744171142578
    },
    {
      "epoch": 1.5979766845703125e-05,
      "model_forward_time": 3.142188549041748,
      "step": 4189
    },
    {
      "epoch": 1.5979766845703125e-05,
      "step": 4189,
      "training_step_time": 10.343898057937622
    },
    {
      "epoch": 1.598358154296875e-05,
      "grad_norm": 0.5055835247039795,
      "learning_rate": 1e-05,
      "loss": 0.0912,
      "step": 4190
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 3.141937255859375,
      "step": 4190
    },
    {
      "dynamics_loss_avg": 0.06115699335932732,
      "epoch": 1.598358154296875e-05,
      "step": 4190
    },
    {
      "action_loss_avg": 0.024046073481440545,
      "epoch": 1.598358154296875e-05,
      "step": 4190
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 4190,
      "training_step_time": 9.938257217407227
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 3.460883378982544,
      "step": 4190
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 4190,
      "training_step_time": 10.240672588348389
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 3.140564441680908,
      "step": 4190
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 4190,
      "training_step_time": 9.923497200012207
    },
    {
      "epoch": 1.598358154296875e-05,
      "model_forward_time": 3.1435129642486572,
      "step": 4190
    },
    {
      "epoch": 1.598358154296875e-05,
      "step": 4190,
      "training_step_time": 10.019749164581299
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 3.160440683364868,
      "step": 4191
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 4191,
      "training_step_time": 9.944387674331665
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 3.1394312381744385,
      "step": 4191
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 4191,
      "training_step_time": 9.920847177505493
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 3.466139316558838,
      "step": 4191
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 4191,
      "training_step_time": 10.24142837524414
    },
    {
      "epoch": 1.5987396240234374e-05,
      "model_forward_time": 3.1420176029205322,
      "step": 4191
    },
    {
      "epoch": 1.5987396240234374e-05,
      "step": 4191,
      "training_step_time": 10.355289459228516
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 3.14776873588562,
      "step": 4192
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 4192,
      "training_step_time": 9.925475597381592
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 3.1390151977539062,
      "step": 4192
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 4192,
      "training_step_time": 9.914578914642334
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 3.1374428272247314,
      "step": 4192
    },
    {
      "dynamics_loss_avg": 0.07353790029883385,
      "epoch": 1.59912109375e-05,
      "step": 4192
    },
    {
      "action_loss_avg": 0.029364616610109806,
      "epoch": 1.59912109375e-05,
      "step": 4192
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 4192,
      "training_step_time": 9.916931390762329
    },
    {
      "epoch": 1.59912109375e-05,
      "model_forward_time": 3.4079456329345703,
      "step": 4192
    },
    {
      "epoch": 1.59912109375e-05,
      "step": 4192,
      "training_step_time": 10.280245304107666
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 3.138920545578003,
      "step": 4193
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 4193,
      "training_step_time": 9.916492700576782
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 3.1358230113983154,
      "step": 4193
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 4193,
      "training_step_time": 9.915382385253906
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 3.133939027786255,
      "step": 4193
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 4193,
      "training_step_time": 9.911120653152466
    },
    {
      "epoch": 1.5995025634765624e-05,
      "model_forward_time": 3.139268636703491,
      "step": 4193
    },
    {
      "epoch": 1.5995025634765624e-05,
      "step": 4193,
      "training_step_time": 10.022749423980713
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 3.4749367237091064,
      "step": 4194
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 4194,
      "training_step_time": 10.263477563858032
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 3.1494052410125732,
      "step": 4194
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 4194,
      "training_step_time": 9.930663108825684
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 3.142899513244629,
      "step": 4194
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 4194,
      "training_step_time": 9.92206335067749
    },
    {
      "epoch": 1.599884033203125e-05,
      "model_forward_time": 3.1451845169067383,
      "step": 4194
    },
    {
      "epoch": 1.599884033203125e-05,
      "step": 4194,
      "training_step_time": 10.300368309020996
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 3.1590309143066406,
      "step": 4195
    },
    {
      "dynamics_loss_avg": 0.06645413059741259,
      "epoch": 1.6002655029296876e-05,
      "step": 4195
    },
    {
      "action_loss_avg": 0.028219123464077712,
      "epoch": 1.6002655029296876e-05,
      "step": 4195
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 4195,
      "training_step_time": 9.949098825454712
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 3.428110361099243,
      "step": 4195
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 4195,
      "training_step_time": 10.213093996047974
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 3.1461360454559326,
      "step": 4195
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 4195,
      "training_step_time": 9.931795597076416
    },
    {
      "epoch": 1.6002655029296876e-05,
      "model_forward_time": 3.1696407794952393,
      "step": 4195
    },
    {
      "epoch": 1.6002655029296876e-05,
      "step": 4195,
      "training_step_time": 10.048113822937012
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 3.1579947471618652,
      "step": 4196
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 4196,
      "training_step_time": 9.941681385040283
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 3.143749713897705,
      "step": 4196
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 4196,
      "training_step_time": 9.923735618591309
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 3.148420810699463,
      "step": 4196
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 4196,
      "training_step_time": 10.252439737319946
    },
    {
      "epoch": 1.60064697265625e-05,
      "model_forward_time": 3.1496007442474365,
      "step": 4196
    },
    {
      "epoch": 1.60064697265625e-05,
      "step": 4196,
      "training_step_time": 10.287900686264038
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 3.1568338871002197,
      "step": 4197
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 4197,
      "training_step_time": 9.939058542251587
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 2.4405012130737305,
      "step": 4197
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 4197,
      "training_step_time": 7.474404573440552
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 3.1433963775634766,
      "step": 4197
    },
    {
      "dynamics_loss_avg": 0.05929944217205048,
      "epoch": 1.6010284423828125e-05,
      "step": 4197
    },
    {
      "action_loss_avg": 0.028355532232671975,
      "epoch": 1.6010284423828125e-05,
      "step": 4197
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 4197,
      "training_step_time": 9.921447515487671
    },
    {
      "epoch": 1.6010284423828125e-05,
      "model_forward_time": 3.1443984508514404,
      "step": 4197
    },
    {
      "epoch": 1.6010284423828125e-05,
      "step": 4197,
      "training_step_time": 10.297876834869385
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 3.163229465484619,
      "step": 4198
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 4198,
      "training_step_time": 9.937642812728882
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 3.1414458751678467,
      "step": 4198
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 4198,
      "training_step_time": 9.915168046951294
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 3.1393356323242188,
      "step": 4198
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 4198,
      "training_step_time": 9.91817307472229
    },
    {
      "epoch": 1.601409912109375e-05,
      "model_forward_time": 3.1421682834625244,
      "step": 4198
    },
    {
      "epoch": 1.601409912109375e-05,
      "step": 4198,
      "training_step_time": 10.336020231246948
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 3.1537394523620605,
      "step": 4199
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 4199,
      "training_step_time": 10.216361999511719
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 3.1430132389068604,
      "step": 4199
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 4199,
      "training_step_time": 9.921534061431885
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 3.148477077484131,
      "step": 4199
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 4199,
      "training_step_time": 9.927049160003662
    },
    {
      "epoch": 1.6017913818359374e-05,
      "model_forward_time": 2.4395692348480225,
      "step": 4199
    },
    {
      "epoch": 1.6017913818359374e-05,
      "step": 4199,
      "training_step_time": 10.311689853668213
    },
    {
      "epoch": 1.6021728515625e-05,
      "grad_norm": 0.5691599249839783,
      "learning_rate": 1e-05,
      "loss": 0.097,
      "step": 4200
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 3.163403034210205,
      "step": 4200
    },
    {
      "dynamics_loss_avg": 0.0608719103038311,
      "epoch": 1.6021728515625e-05,
      "step": 4200
    },
    {
      "action_loss_avg": 0.023259137757122517,
      "epoch": 1.6021728515625e-05,
      "step": 4200
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 4200,
      "training_step_time": 9.939162731170654
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 3.4089982509613037,
      "step": 4200
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 4200,
      "training_step_time": 10.186364650726318
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 3.149212598800659,
      "step": 4200
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 4200,
      "training_step_time": 9.926312685012817
    },
    {
      "epoch": 1.6021728515625e-05,
      "model_forward_time": 2.441817045211792,
      "step": 4200
    },
    {
      "epoch": 1.6021728515625e-05,
      "step": 4200,
      "training_step_time": 10.345945119857788
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 3.150714874267578,
      "step": 4201
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 4201,
      "training_step_time": 9.92748761177063
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 3.146348714828491,
      "step": 4201
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 4201,
      "training_step_time": 9.921464443206787
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 3.1398417949676514,
      "step": 4201
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 4201,
      "training_step_time": 9.918382167816162
    },
    {
      "epoch": 1.6025543212890624e-05,
      "model_forward_time": 3.439168691635132,
      "step": 4201
    },
    {
      "epoch": 1.6025543212890624e-05,
      "step": 4201,
      "training_step_time": 10.307520627975464
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 3.1618170738220215,
      "step": 4202
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 4202,
      "training_step_time": 9.942271709442139
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 3.147117853164673,
      "step": 4202
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 4202,
      "training_step_time": 9.92668604850769
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 3.14705753326416,
      "step": 4202
    },
    {
      "dynamics_loss_avg": 0.04392117839306593,
      "epoch": 1.602935791015625e-05,
      "step": 4202
    },
    {
      "action_loss_avg": 0.02165824780240655,
      "epoch": 1.602935791015625e-05,
      "step": 4202
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 4202,
      "training_step_time": 9.927512645721436
    },
    {
      "epoch": 1.602935791015625e-05,
      "model_forward_time": 3.4207756519317627,
      "step": 4202
    },
    {
      "epoch": 1.602935791015625e-05,
      "step": 4202,
      "training_step_time": 10.2950279712677
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 3.1604037284851074,
      "step": 4203
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 4203,
      "training_step_time": 9.945871591567993
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 3.14477801322937,
      "step": 4203
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 4203,
      "training_step_time": 9.923885107040405
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 3.140301465988159,
      "step": 4203
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 4203,
      "training_step_time": 9.924334526062012
    },
    {
      "epoch": 1.6033172607421876e-05,
      "model_forward_time": 3.1415817737579346,
      "step": 4203
    },
    {
      "epoch": 1.6033172607421876e-05,
      "step": 4203,
      "training_step_time": 10.322277307510376
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 3.1498050689697266,
      "step": 4204
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 4204,
      "training_step_time": 10.195291996002197
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 3.142939567565918,
      "step": 4204
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 4204,
      "training_step_time": 9.920471429824829
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 3.1398732662200928,
      "step": 4204
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 4204,
      "training_step_time": 9.916419506072998
    },
    {
      "epoch": 1.60369873046875e-05,
      "model_forward_time": 3.1434316635131836,
      "step": 4204
    },
    {
      "epoch": 1.60369873046875e-05,
      "step": 4204,
      "training_step_time": 10.322793245315552
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 3.1605355739593506,
      "step": 4205
    },
    {
      "dynamics_loss_avg": 0.0817389540374279,
      "epoch": 1.6040802001953125e-05,
      "step": 4205
    },
    {
      "action_loss_avg": 0.03845741860568523,
      "epoch": 1.6040802001953125e-05,
      "step": 4205
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 4205,
      "training_step_time": 9.945114850997925
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 3.4252243041992188,
      "step": 4205
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 4205,
      "training_step_time": 10.202671766281128
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 3.1406869888305664,
      "step": 4205
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 4205,
      "training_step_time": 9.914881706237793
    },
    {
      "epoch": 1.6040802001953125e-05,
      "model_forward_time": 3.138942003250122,
      "step": 4205
    },
    {
      "epoch": 1.6040802001953125e-05,
      "step": 4205,
      "training_step_time": 10.250943422317505
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 3.1572654247283936,
      "step": 4206
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 4206,
      "training_step_time": 9.946693897247314
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 3.1390159130096436,
      "step": 4206
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 4206,
      "training_step_time": 9.913018226623535
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 3.4170215129852295,
      "step": 4206
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 4206,
      "training_step_time": 10.196103811264038
    },
    {
      "epoch": 1.604461669921875e-05,
      "model_forward_time": 3.1450514793395996,
      "step": 4206
    },
    {
      "epoch": 1.604461669921875e-05,
      "step": 4206,
      "training_step_time": 10.256540775299072
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 3.151183605194092,
      "step": 4207
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 4207,
      "training_step_time": 9.928739786148071
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 3.1386961936950684,
      "step": 4207
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 4207,
      "training_step_time": 9.910839080810547
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 2.435224771499634,
      "step": 4207
    },
    {
      "dynamics_loss_avg": 0.07961764708161354,
      "epoch": 1.6048431396484374e-05,
      "step": 4207
    },
    {
      "action_loss_avg": 0.024540469422936438,
      "epoch": 1.6048431396484374e-05,
      "step": 4207
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 4207,
      "training_step_time": 7.463123321533203
    },
    {
      "epoch": 1.6048431396484374e-05,
      "model_forward_time": 3.136302947998047,
      "step": 4207
    },
    {
      "epoch": 1.6048431396484374e-05,
      "step": 4207,
      "training_step_time": 10.310548067092896
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 3.157519578933716,
      "step": 4208
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 4208,
      "training_step_time": 9.938349485397339
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 3.13861346244812,
      "step": 4208
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 4208,
      "training_step_time": 9.913617134094238
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 2.4328606128692627,
      "step": 4208
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 4208,
      "training_step_time": 7.465717315673828
    },
    {
      "epoch": 1.605224609375e-05,
      "model_forward_time": 3.137436628341675,
      "step": 4208
    },
    {
      "epoch": 1.605224609375e-05,
      "step": 4208,
      "training_step_time": 10.288301467895508
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 3.14501690864563,
      "step": 4209
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 4209,
      "training_step_time": 10.240707397460938
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 3.1543679237365723,
      "step": 4209
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 4209,
      "training_step_time": 9.946195363998413
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 3.150494337081909,
      "step": 4209
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 4209,
      "training_step_time": 9.930551528930664
    },
    {
      "epoch": 1.6056060791015624e-05,
      "model_forward_time": 3.141115427017212,
      "step": 4209
    },
    {
      "epoch": 1.6056060791015624e-05,
      "step": 4209,
      "training_step_time": 10.31781792640686
    },
    {
      "epoch": 1.605987548828125e-05,
      "grad_norm": 0.6816439628601074,
      "learning_rate": 1e-05,
      "loss": 0.0928,
      "step": 4210
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 3.1689016819000244,
      "step": 4210
    },
    {
      "dynamics_loss_avg": 0.05295478347688913,
      "epoch": 1.605987548828125e-05,
      "step": 4210
    },
    {
      "action_loss_avg": 0.021993768215179444,
      "epoch": 1.605987548828125e-05,
      "step": 4210
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 4210,
      "training_step_time": 9.949787616729736
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 3.142174243927002,
      "step": 4210
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 4210,
      "training_step_time": 10.208251953125
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 3.1497552394866943,
      "step": 4210
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 4210,
      "training_step_time": 9.929476261138916
    },
    {
      "epoch": 1.605987548828125e-05,
      "model_forward_time": 3.1384055614471436,
      "step": 4210
    },
    {
      "epoch": 1.605987548828125e-05,
      "step": 4210,
      "training_step_time": 10.336366415023804
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 3.14499831199646,
      "step": 4211
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 4211,
      "training_step_time": 9.922641515731812
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 3.134268283843994,
      "step": 4211
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 4211,
      "training_step_time": 9.911846160888672
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 3.460392713546753,
      "step": 4211
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 4211,
      "training_step_time": 10.238091945648193
    },
    {
      "epoch": 1.6063690185546876e-05,
      "model_forward_time": 3.1336700916290283,
      "step": 4211
    },
    {
      "epoch": 1.6063690185546876e-05,
      "step": 4211,
      "training_step_time": 10.224229574203491
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 3.1507272720336914,
      "step": 4212
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 4212,
      "training_step_time": 9.931710481643677
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 3.1353352069854736,
      "step": 4212
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 4212,
      "training_step_time": 9.912451267242432
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 3.135094165802002,
      "step": 4212
    },
    {
      "dynamics_loss_avg": 0.07260492257773876,
      "epoch": 1.60675048828125e-05,
      "step": 4212
    },
    {
      "action_loss_avg": 0.03436994533985853,
      "epoch": 1.60675048828125e-05,
      "step": 4212
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 4212,
      "training_step_time": 9.916339635848999
    },
    {
      "epoch": 1.60675048828125e-05,
      "model_forward_time": 3.1381101608276367,
      "step": 4212
    },
    {
      "epoch": 1.60675048828125e-05,
      "step": 4212,
      "training_step_time": 10.313591241836548
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 3.4441213607788086,
      "step": 4213
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 4213,
      "training_step_time": 10.224660634994507
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 3.1456072330474854,
      "step": 4213
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 4213,
      "training_step_time": 9.932865142822266
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 3.13533091545105,
      "step": 4213
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 4213,
      "training_step_time": 9.915504455566406
    },
    {
      "epoch": 1.6071319580078125e-05,
      "model_forward_time": 3.1418275833129883,
      "step": 4213
    },
    {
      "epoch": 1.6071319580078125e-05,
      "step": 4213,
      "training_step_time": 10.030624151229858
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 3.4065146446228027,
      "step": 4214
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 4214,
      "training_step_time": 10.183551549911499
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 3.1460623741149902,
      "step": 4214
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 4214,
      "training_step_time": 9.926377058029175
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 3.1380295753479004,
      "step": 4214
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 4214,
      "training_step_time": 9.917254209518433
    },
    {
      "epoch": 1.607513427734375e-05,
      "model_forward_time": 3.1403005123138428,
      "step": 4214
    },
    {
      "epoch": 1.607513427734375e-05,
      "step": 4214,
      "training_step_time": 10.037566184997559
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 3.1587979793548584,
      "step": 4215
    },
    {
      "dynamics_loss_avg": 0.06595814637839795,
      "epoch": 1.6078948974609374e-05,
      "step": 4215
    },
    {
      "action_loss_avg": 0.02614584630355239,
      "epoch": 1.6078948974609374e-05,
      "step": 4215
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 4215,
      "training_step_time": 9.945785284042358
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 3.1443257331848145,
      "step": 4215
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 4215,
      "training_step_time": 10.23009967803955
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 3.1466917991638184,
      "step": 4215
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 4215,
      "training_step_time": 9.925586223602295
    },
    {
      "epoch": 1.6078948974609374e-05,
      "model_forward_time": 3.1468846797943115,
      "step": 4215
    },
    {
      "epoch": 1.6078948974609374e-05,
      "step": 4215,
      "training_step_time": 10.37993836402893
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 3.155978202819824,
      "step": 4216
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 4216,
      "training_step_time": 9.933666467666626
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 3.142977476119995,
      "step": 4216
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 4216,
      "training_step_time": 9.928763389587402
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 3.1454038619995117,
      "step": 4216
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 4216,
      "training_step_time": 10.213478565216064
    },
    {
      "epoch": 1.6082763671875e-05,
      "model_forward_time": 3.148033380508423,
      "step": 4216
    },
    {
      "epoch": 1.6082763671875e-05,
      "step": 4216,
      "training_step_time": 10.02152681350708
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 3.157900094985962,
      "step": 4217
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 4217,
      "training_step_time": 9.937777996063232
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 3.1392223834991455,
      "step": 4217
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 4217,
      "training_step_time": 9.91007685661316
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 3.1426100730895996,
      "step": 4217
    },
    {
      "dynamics_loss_avg": 0.06793495453894138,
      "epoch": 1.6086578369140624e-05,
      "step": 4217
    },
    {
      "action_loss_avg": 0.028006007662042975,
      "epoch": 1.6086578369140624e-05,
      "step": 4217
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 4217,
      "training_step_time": 9.920016288757324
    },
    {
      "epoch": 1.6086578369140624e-05,
      "model_forward_time": 3.4335765838623047,
      "step": 4217
    },
    {
      "epoch": 1.6086578369140624e-05,
      "step": 4217,
      "training_step_time": 10.298218965530396
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 3.1543705463409424,
      "step": 4218
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 4218,
      "training_step_time": 9.924203157424927
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 3.137965202331543,
      "step": 4218
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 4218,
      "training_step_time": 9.911869764328003
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 3.1387388706207275,
      "step": 4218
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 4218,
      "training_step_time": 9.906621932983398
    },
    {
      "epoch": 1.609039306640625e-05,
      "model_forward_time": 3.1408538818359375,
      "step": 4218
    },
    {
      "epoch": 1.609039306640625e-05,
      "step": 4218,
      "training_step_time": 10.266202688217163
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 2.4466991424560547,
      "step": 4219
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 4219,
      "training_step_time": 7.475920677185059
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 3.3930556774139404,
      "step": 4219
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 4219,
      "training_step_time": 10.166642904281616
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 3.14106822013855,
      "step": 4219
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 4219,
      "training_step_time": 9.914096593856812
    },
    {
      "epoch": 1.6094207763671876e-05,
      "model_forward_time": 3.1382808685302734,
      "step": 4219
    },
    {
      "epoch": 1.6094207763671876e-05,
      "step": 4219,
      "training_step_time": 10.03406572341919
    },
    {
      "epoch": 1.60980224609375e-05,
      "grad_norm": 0.9851928949356079,
      "learning_rate": 1e-05,
      "loss": 0.0938,
      "step": 4220
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 3.1577882766723633,
      "step": 4220
    },
    {
      "dynamics_loss_avg": 0.06541166193783284,
      "epoch": 1.60980224609375e-05,
      "step": 4220
    },
    {
      "action_loss_avg": 0.02257475471124053,
      "epoch": 1.60980224609375e-05,
      "step": 4220
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 4220,
      "training_step_time": 9.938257694244385
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 3.413363218307495,
      "step": 4220
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 4220,
      "training_step_time": 10.182288646697998
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 3.143054962158203,
      "step": 4220
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 4220,
      "training_step_time": 9.91685175895691
    },
    {
      "epoch": 1.60980224609375e-05,
      "model_forward_time": 3.1362102031707764,
      "step": 4220
    },
    {
      "epoch": 1.60980224609375e-05,
      "step": 4220,
      "training_step_time": 10.281598567962646
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 3.1524734497070312,
      "step": 4221
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 4221,
      "training_step_time": 9.925038576126099
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 3.140604019165039,
      "step": 4221
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 4221,
      "training_step_time": 9.907580137252808
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 3.138150930404663,
      "step": 4221
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 4221,
      "training_step_time": 10.176033735275269
    },
    {
      "epoch": 1.6101837158203125e-05,
      "model_forward_time": 3.1529412269592285,
      "step": 4221
    },
    {
      "epoch": 1.6101837158203125e-05,
      "step": 4221,
      "training_step_time": 10.286054372787476
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 3.1564528942108154,
      "step": 4222
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 4222,
      "training_step_time": 9.930665969848633
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 3.138171672821045,
      "step": 4222
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 4222,
      "training_step_time": 9.909869909286499
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 3.135892629623413,
      "step": 4222
    },
    {
      "dynamics_loss_avg": 0.06967838313430548,
      "epoch": 1.610565185546875e-05,
      "step": 4222
    },
    {
      "action_loss_avg": 0.02749959770590067,
      "epoch": 1.610565185546875e-05,
      "step": 4222
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 4222,
      "training_step_time": 9.908079624176025
    },
    {
      "epoch": 1.610565185546875e-05,
      "model_forward_time": 3.4147353172302246,
      "step": 4222
    },
    {
      "epoch": 1.610565185546875e-05,
      "step": 4222,
      "training_step_time": 10.279099464416504
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 3.1453158855438232,
      "step": 4223
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 4223,
      "training_step_time": 9.917840719223022
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 3.1424336433410645,
      "step": 4223
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 4223,
      "training_step_time": 9.915705919265747
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 3.1435909271240234,
      "step": 4223
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 4223,
      "training_step_time": 9.921456098556519
    },
    {
      "epoch": 1.6109466552734374e-05,
      "model_forward_time": 3.1418087482452393,
      "step": 4223
    },
    {
      "epoch": 1.6109466552734374e-05,
      "step": 4223,
      "training_step_time": 10.302546262741089
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 3.4296109676361084,
      "step": 4224
    },
    {
      "epoch": 1.611328125e-05,
      "step": 4224,
      "training_step_time": 10.204191446304321
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 2.441352128982544,
      "step": 4224
    },
    {
      "epoch": 1.611328125e-05,
      "step": 4224,
      "training_step_time": 7.4716103076934814
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 3.13909912109375,
      "step": 4224
    },
    {
      "epoch": 1.611328125e-05,
      "step": 4224,
      "training_step_time": 9.924633264541626
    },
    {
      "epoch": 1.611328125e-05,
      "model_forward_time": 3.143247365951538,
      "step": 4224
    },
    {
      "epoch": 1.611328125e-05,
      "step": 4224,
      "training_step_time": 10.343812465667725
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 3.1550397872924805,
      "step": 4225
    },
    {
      "dynamics_loss_avg": 0.04859609566628933,
      "epoch": 1.6117095947265624e-05,
      "step": 4225
    },
    {
      "action_loss_avg": 0.02268292782828212,
      "epoch": 1.6117095947265624e-05,
      "step": 4225
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 4225,
      "training_step_time": 9.930112361907959
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 3.429564952850342,
      "step": 4225
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 4225,
      "training_step_time": 10.199862480163574
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 3.1367249488830566,
      "step": 4225
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 4225,
      "training_step_time": 9.903559923171997
    },
    {
      "epoch": 1.6117095947265624e-05,
      "model_forward_time": 3.1366310119628906,
      "step": 4225
    },
    {
      "epoch": 1.6117095947265624e-05,
      "step": 4225,
      "training_step_time": 10.30315637588501
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 3.1497111320495605,
      "step": 4226
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 4226,
      "training_step_time": 9.921483039855957
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 3.1334547996520996,
      "step": 4226
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 4226,
      "training_step_time": 9.90115737915039
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 3.394179582595825,
      "step": 4226
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 4226,
      "training_step_time": 10.161109209060669
    },
    {
      "epoch": 1.612091064453125e-05,
      "model_forward_time": 3.141165256500244,
      "step": 4226
    },
    {
      "epoch": 1.612091064453125e-05,
      "step": 4226,
      "training_step_time": 10.300608396530151
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 3.1460983753204346,
      "step": 4227
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 4227,
      "training_step_time": 9.927122354507446
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 3.137474775314331,
      "step": 4227
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 4227,
      "training_step_time": 9.914799451828003
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 3.1509861946105957,
      "step": 4227
    },
    {
      "dynamics_loss_avg": 0.07758644036948681,
      "epoch": 1.6124725341796876e-05,
      "step": 4227
    },
    {
      "action_loss_avg": 0.032350336574018,
      "epoch": 1.6124725341796876e-05,
      "step": 4227
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 4227,
      "training_step_time": 9.93307375907898
    },
    {
      "epoch": 1.6124725341796876e-05,
      "model_forward_time": 3.440739393234253,
      "step": 4227
    },
    {
      "epoch": 1.6124725341796876e-05,
      "step": 4227,
      "training_step_time": 10.315968036651611
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 3.1586289405822754,
      "step": 4228
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 4228,
      "training_step_time": 9.93927550315857
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 3.1456189155578613,
      "step": 4228
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 4228,
      "training_step_time": 9.922975063323975
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 3.1407930850982666,
      "step": 4228
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 4228,
      "training_step_time": 9.910784006118774
    },
    {
      "epoch": 1.61285400390625e-05,
      "model_forward_time": 3.136248826980591,
      "step": 4228
    },
    {
      "epoch": 1.61285400390625e-05,
      "step": 4228,
      "training_step_time": 10.02948260307312
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 3.427210569381714,
      "step": 4229
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 4229,
      "training_step_time": 10.196870803833008
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 3.13496732711792,
      "step": 4229
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 4229,
      "training_step_time": 9.905465126037598
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 3.1473658084869385,
      "step": 4229
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 4229,
      "training_step_time": 9.919735670089722
    },
    {
      "epoch": 1.6132354736328125e-05,
      "model_forward_time": 3.140981435775757,
      "step": 4229
    },
    {
      "epoch": 1.6132354736328125e-05,
      "step": 4229,
      "training_step_time": 10.323770761489868
    },
    {
      "epoch": 1.613616943359375e-05,
      "grad_norm": 0.43409407138824463,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 4230
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 3.1471333503723145,
      "step": 4230
    },
    {
      "dynamics_loss_avg": 0.06473121307790279,
      "epoch": 1.613616943359375e-05,
      "step": 4230
    },
    {
      "action_loss_avg": 0.03356430660933256,
      "epoch": 1.613616943359375e-05,
      "step": 4230
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 4230,
      "training_step_time": 9.92624020576477
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 3.438985824584961,
      "step": 4230
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 4230,
      "training_step_time": 10.216343879699707
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 3.1382620334625244,
      "step": 4230
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 4230,
      "training_step_time": 9.914273500442505
    },
    {
      "epoch": 1.613616943359375e-05,
      "model_forward_time": 3.1384167671203613,
      "step": 4230
    },
    {
      "epoch": 1.613616943359375e-05,
      "step": 4230,
      "training_step_time": 10.319270133972168
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 3.1531314849853516,
      "step": 4231
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 4231,
      "training_step_time": 9.933400630950928
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 3.1483380794525146,
      "step": 4231
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 4231,
      "training_step_time": 9.929961442947388
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 3.429462194442749,
      "step": 4231
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 4231,
      "training_step_time": 10.216369390487671
    },
    {
      "epoch": 1.6139984130859375e-05,
      "model_forward_time": 3.150987386703491,
      "step": 4231
    },
    {
      "epoch": 1.6139984130859375e-05,
      "step": 4231,
      "training_step_time": 10.046278953552246
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 3.1624693870544434,
      "step": 4232
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 4232,
      "training_step_time": 9.946137428283691
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 3.1491565704345703,
      "step": 4232
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 4232,
      "training_step_time": 9.930893659591675
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 3.148075819015503,
      "step": 4232
    },
    {
      "dynamics_loss_avg": 0.06276459973305464,
      "epoch": 1.6143798828125e-05,
      "step": 4232
    },
    {
      "action_loss_avg": 0.02859492339193821,
      "epoch": 1.6143798828125e-05,
      "step": 4232
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 4232,
      "training_step_time": 9.93261981010437
    },
    {
      "epoch": 1.6143798828125e-05,
      "model_forward_time": 3.4329843521118164,
      "step": 4232
    },
    {
      "epoch": 1.6143798828125e-05,
      "step": 4232,
      "training_step_time": 10.304859399795532
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 3.160080909729004,
      "step": 4233
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 4233,
      "training_step_time": 9.939948081970215
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 3.1479222774505615,
      "step": 4233
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 4233,
      "training_step_time": 9.925178289413452
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 2.441394805908203,
      "step": 4233
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 4233,
      "training_step_time": 7.471797227859497
    },
    {
      "epoch": 1.6147613525390624e-05,
      "model_forward_time": 3.1424598693847656,
      "step": 4233
    },
    {
      "epoch": 1.6147613525390624e-05,
      "step": 4233,
      "training_step_time": 10.040579557418823
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 2.4492852687835693,
      "step": 4234
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 4234,
      "training_step_time": 7.741398572921753
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 3.1525285243988037,
      "step": 4234
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 4234,
      "training_step_time": 9.930372953414917
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 3.1464076042175293,
      "step": 4234
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 4234,
      "training_step_time": 9.924469232559204
    },
    {
      "epoch": 1.615142822265625e-05,
      "model_forward_time": 3.151158332824707,
      "step": 4234
    },
    {
      "epoch": 1.615142822265625e-05,
      "step": 4234,
      "training_step_time": 10.320726156234741
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 3.165468215942383,
      "step": 4235
    },
    {
      "dynamics_loss_avg": 0.06308337599039078,
      "epoch": 1.6155242919921876e-05,
      "step": 4235
    },
    {
      "action_loss_avg": 0.025093565788120033,
      "epoch": 1.6155242919921876e-05,
      "step": 4235
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 4235,
      "training_step_time": 9.949172019958496
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 3.4403703212738037,
      "step": 4235
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 4235,
      "training_step_time": 10.211570024490356
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 3.1436238288879395,
      "step": 4235
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 4235,
      "training_step_time": 9.921303510665894
    },
    {
      "epoch": 1.6155242919921876e-05,
      "model_forward_time": 3.146024227142334,
      "step": 4235
    },
    {
      "epoch": 1.6155242919921876e-05,
      "step": 4235,
      "training_step_time": 10.33433198928833
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 3.149820566177368,
      "step": 4236
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 4236,
      "training_step_time": 9.92438006401062
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 3.140903949737549,
      "step": 4236
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 4236,
      "training_step_time": 9.907336950302124
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 3.138913631439209,
      "step": 4236
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 4236,
      "training_step_time": 10.186794757843018
    },
    {
      "epoch": 1.61590576171875e-05,
      "model_forward_time": 3.141239643096924,
      "step": 4236
    },
    {
      "epoch": 1.61590576171875e-05,
      "step": 4236,
      "training_step_time": 10.296611547470093
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 3.1484246253967285,
      "step": 4237
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 4237,
      "training_step_time": 9.919595718383789
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 2.4470713138580322,
      "step": 4237
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 4237,
      "training_step_time": 7.475377321243286
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 3.1389358043670654,
      "step": 4237
    },
    {
      "dynamics_loss_avg": 0.06025587562471628,
      "epoch": 1.6162872314453125e-05,
      "step": 4237
    },
    {
      "action_loss_avg": 0.02628480680286884,
      "epoch": 1.6162872314453125e-05,
      "step": 4237
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 4237,
      "training_step_time": 9.914427757263184
    },
    {
      "epoch": 1.6162872314453125e-05,
      "model_forward_time": 3.139906406402588,
      "step": 4237
    },
    {
      "epoch": 1.6162872314453125e-05,
      "step": 4237,
      "training_step_time": 10.271878480911255
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 3.1571731567382812,
      "step": 4238
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 4238,
      "training_step_time": 9.931047439575195
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 3.142533540725708,
      "step": 4238
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 4238,
      "training_step_time": 9.917649507522583
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 3.144319772720337,
      "step": 4238
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 4238,
      "training_step_time": 9.925529479980469
    },
    {
      "epoch": 1.616668701171875e-05,
      "model_forward_time": 3.1448729038238525,
      "step": 4238
    },
    {
      "epoch": 1.616668701171875e-05,
      "step": 4238,
      "training_step_time": 10.032640933990479
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 3.151625871658325,
      "step": 4239
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 4239,
      "training_step_time": 10.20518708229065
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 2.4409847259521484,
      "step": 4239
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 4239,
      "training_step_time": 7.470890998840332
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 3.146564483642578,
      "step": 4239
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 4239,
      "training_step_time": 9.924722671508789
    },
    {
      "epoch": 1.6170501708984375e-05,
      "model_forward_time": 3.1445677280426025,
      "step": 4239
    },
    {
      "epoch": 1.6170501708984375e-05,
      "step": 4239,
      "training_step_time": 10.038662910461426
    },
    {
      "epoch": 1.617431640625e-05,
      "grad_norm": 0.5374722480773926,
      "learning_rate": 1e-05,
      "loss": 0.089,
      "step": 4240
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 3.150895595550537,
      "step": 4240
    },
    {
      "dynamics_loss_avg": 0.07631653286516667,
      "epoch": 1.617431640625e-05,
      "step": 4240
    },
    {
      "action_loss_avg": 0.033138811402022836,
      "epoch": 1.617431640625e-05,
      "step": 4240
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 4240,
      "training_step_time": 9.937061309814453
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 3.418072462081909,
      "step": 4240
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 4240,
      "training_step_time": 10.195210218429565
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 3.14707612991333,
      "step": 4240
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 4240,
      "training_step_time": 9.924944639205933
    },
    {
      "epoch": 1.617431640625e-05,
      "model_forward_time": 3.1416196823120117,
      "step": 4240
    },
    {
      "epoch": 1.617431640625e-05,
      "step": 4240,
      "training_step_time": 10.297240734100342
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 3.1592254638671875,
      "step": 4241
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 4241,
      "training_step_time": 9.93520712852478
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 3.144974946975708,
      "step": 4241
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 4241,
      "training_step_time": 9.919935941696167
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 3.1438310146331787,
      "step": 4241
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 4241,
      "training_step_time": 10.176174879074097
    },
    {
      "epoch": 1.6178131103515624e-05,
      "model_forward_time": 3.1401352882385254,
      "step": 4241
    },
    {
      "epoch": 1.6178131103515624e-05,
      "step": 4241,
      "training_step_time": 10.341426372528076
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 3.1503920555114746,
      "step": 4242
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 4242,
      "training_step_time": 9.930890560150146
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 3.144507646560669,
      "step": 4242
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 4242,
      "training_step_time": 9.919703245162964
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 3.136159896850586,
      "step": 4242
    },
    {
      "dynamics_loss_avg": 0.06186587847769261,
      "epoch": 1.618194580078125e-05,
      "step": 4242
    },
    {
      "action_loss_avg": 0.021983883809298277,
      "epoch": 1.618194580078125e-05,
      "step": 4242
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 4242,
      "training_step_time": 9.908652305603027
    },
    {
      "epoch": 1.618194580078125e-05,
      "model_forward_time": 3.393599510192871,
      "step": 4242
    },
    {
      "epoch": 1.618194580078125e-05,
      "step": 4242,
      "training_step_time": 10.277325868606567
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 3.1464924812316895,
      "step": 4243
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 4243,
      "training_step_time": 9.924104452133179
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 3.1382410526275635,
      "step": 4243
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 4243,
      "training_step_time": 9.91039514541626
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 3.136061668395996,
      "step": 4243
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 4243,
      "training_step_time": 9.911974906921387
    },
    {
      "epoch": 1.6185760498046876e-05,
      "model_forward_time": 3.1388351917266846,
      "step": 4243
    },
    {
      "epoch": 1.6185760498046876e-05,
      "step": 4243,
      "training_step_time": 10.305165529251099
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 3.447362184524536,
      "step": 4244
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 4244,
      "training_step_time": 10.223568439483643
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 3.154829740524292,
      "step": 4244
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 4244,
      "training_step_time": 9.933340311050415
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 3.1464662551879883,
      "step": 4244
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 4244,
      "training_step_time": 9.921646118164062
    },
    {
      "epoch": 1.61895751953125e-05,
      "model_forward_time": 3.150743246078491,
      "step": 4244
    },
    {
      "epoch": 1.61895751953125e-05,
      "step": 4244,
      "training_step_time": 10.024767875671387
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 2.4511845111846924,
      "step": 4245
    },
    {
      "dynamics_loss_avg": 0.06951560117304326,
      "epoch": 1.6193389892578125e-05,
      "step": 4245
    },
    {
      "action_loss_avg": 0.028968026116490363,
      "epoch": 1.6193389892578125e-05,
      "step": 4245
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 4245,
      "training_step_time": 7.487919807434082
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 3.440324306488037,
      "step": 4245
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 4245,
      "training_step_time": 10.2173490524292
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 3.14827299118042,
      "step": 4245
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 4245,
      "training_step_time": 9.925640344619751
    },
    {
      "epoch": 1.6193389892578125e-05,
      "model_forward_time": 3.146214008331299,
      "step": 4245
    },
    {
      "epoch": 1.6193389892578125e-05,
      "step": 4245,
      "training_step_time": 10.31326174736023
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 3.1551578044891357,
      "step": 4246
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 4246,
      "training_step_time": 9.931102752685547
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 3.143738031387329,
      "step": 4246
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 4246,
      "training_step_time": 9.92103624343872
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 3.4311933517456055,
      "step": 4246
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 4246,
      "training_step_time": 10.20781946182251
    },
    {
      "epoch": 1.619720458984375e-05,
      "model_forward_time": 3.1516435146331787,
      "step": 4246
    },
    {
      "epoch": 1.619720458984375e-05,
      "step": 4246,
      "training_step_time": 10.34019160270691
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 3.162349224090576,
      "step": 4247
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 4247,
      "training_step_time": 9.94340181350708
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 3.1516733169555664,
      "step": 4247
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 4247,
      "training_step_time": 9.928786039352417
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 3.147876024246216,
      "step": 4247
    },
    {
      "dynamics_loss_avg": 0.07167237512767315,
      "epoch": 1.6201019287109375e-05,
      "step": 4247
    },
    {
      "action_loss_avg": 0.03397379983216524,
      "epoch": 1.6201019287109375e-05,
      "step": 4247
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 4247,
      "training_step_time": 9.925407409667969
    },
    {
      "epoch": 1.6201019287109375e-05,
      "model_forward_time": 3.4239840507507324,
      "step": 4247
    },
    {
      "epoch": 1.6201019287109375e-05,
      "step": 4247,
      "training_step_time": 10.2930006980896
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 3.156397819519043,
      "step": 4248
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 4248,
      "training_step_time": 9.933502674102783
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 3.142834424972534,
      "step": 4248
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 4248,
      "training_step_time": 9.922123670578003
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 3.151108741760254,
      "step": 4248
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 4248,
      "training_step_time": 9.931298017501831
    },
    {
      "epoch": 1.6204833984375e-05,
      "model_forward_time": 3.148681879043579,
      "step": 4248
    },
    {
      "epoch": 1.6204833984375e-05,
      "step": 4248,
      "training_step_time": 10.044553995132446
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 3.4778382778167725,
      "step": 4249
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 4249,
      "training_step_time": 10.258624792098999
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 3.141685724258423,
      "step": 4249
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 4249,
      "training_step_time": 9.914201259613037
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 3.142204523086548,
      "step": 4249
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 4249,
      "training_step_time": 9.915320873260498
    },
    {
      "epoch": 1.6208648681640624e-05,
      "model_forward_time": 3.1395351886749268,
      "step": 4249
    },
    {
      "epoch": 1.6208648681640624e-05,
      "step": 4249,
      "training_step_time": 10.267998933792114
    },
    {
      "epoch": 1.621246337890625e-05,
      "grad_norm": 0.46327412128448486,
      "learning_rate": 1e-05,
      "loss": 0.0927,
      "step": 4250
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 3.1564483642578125,
      "step": 4250
    },
    {
      "dynamics_loss_avg": 0.08096064813435078,
      "epoch": 1.621246337890625e-05,
      "step": 4250
    },
    {
      "action_loss_avg": 0.037411504751071335,
      "epoch": 1.621246337890625e-05,
      "step": 4250
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 4250,
      "training_step_time": 9.93440556526184
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 3.398460865020752,
      "step": 4250
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 4250,
      "training_step_time": 10.166656732559204
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 3.139533758163452,
      "step": 4250
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 4250,
      "training_step_time": 9.907373189926147
    },
    {
      "epoch": 1.621246337890625e-05,
      "model_forward_time": 3.13533353805542,
      "step": 4250
    },
    {
      "epoch": 1.621246337890625e-05,
      "step": 4250,
      "training_step_time": 10.007910013198853
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 3.14752197265625,
      "step": 4251
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 4251,
      "training_step_time": 9.915164709091187
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 3.140292167663574,
      "step": 4251
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 4251,
      "training_step_time": 9.917054414749146
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 3.4245545864105225,
      "step": 4251
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 4251,
      "training_step_time": 10.191462516784668
    },
    {
      "epoch": 1.6216278076171876e-05,
      "model_forward_time": 2.4376678466796875,
      "step": 4251
    },
    {
      "epoch": 1.6216278076171876e-05,
      "step": 4251,
      "training_step_time": 10.278706789016724
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 2.443941831588745,
      "step": 4252
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 4252,
      "training_step_time": 7.464724063873291
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 3.134847402572632,
      "step": 4252
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 4252,
      "training_step_time": 9.898955583572388
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 3.134931802749634,
      "step": 4252
    },
    {
      "dynamics_loss_avg": 0.06277516931295395,
      "epoch": 1.62200927734375e-05,
      "step": 4252
    },
    {
      "action_loss_avg": 0.026806361507624388,
      "epoch": 1.62200927734375e-05,
      "step": 4252
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 4252,
      "training_step_time": 9.908835649490356
    },
    {
      "epoch": 1.62200927734375e-05,
      "model_forward_time": 3.1459083557128906,
      "step": 4252
    },
    {
      "epoch": 1.62200927734375e-05,
      "step": 4252,
      "training_step_time": 10.263257265090942
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 3.45206618309021,
      "step": 4253
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 4253,
      "training_step_time": 10.224143505096436
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 3.1404895782470703,
      "step": 4253
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 4253,
      "training_step_time": 9.913588762283325
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 3.1419942378997803,
      "step": 4253
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 4253,
      "training_step_time": 9.911870002746582
    },
    {
      "epoch": 1.6223907470703125e-05,
      "model_forward_time": 3.1404526233673096,
      "step": 4253
    },
    {
      "epoch": 1.6223907470703125e-05,
      "step": 4253,
      "training_step_time": 10.020264148712158
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 3.421527147293091,
      "step": 4254
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 4254,
      "training_step_time": 10.196810245513916
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 3.1369709968566895,
      "step": 4254
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 4254,
      "training_step_time": 9.91291332244873
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 3.136115550994873,
      "step": 4254
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 4254,
      "training_step_time": 9.907352685928345
    },
    {
      "epoch": 1.622772216796875e-05,
      "model_forward_time": 3.141063690185547,
      "step": 4254
    },
    {
      "epoch": 1.622772216796875e-05,
      "step": 4254,
      "training_step_time": 10.274794578552246
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 3.155289649963379,
      "step": 4255
    },
    {
      "dynamics_loss_avg": 0.06753962337970734,
      "epoch": 1.6231536865234375e-05,
      "step": 4255
    },
    {
      "action_loss_avg": 0.027043414209038018,
      "epoch": 1.6231536865234375e-05,
      "step": 4255
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 4255,
      "training_step_time": 9.925373077392578
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 3.137143611907959,
      "step": 4255
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 4255,
      "training_step_time": 10.166566848754883
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 3.137516975402832,
      "step": 4255
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 4255,
      "training_step_time": 9.909674644470215
    },
    {
      "epoch": 1.6231536865234375e-05,
      "model_forward_time": 3.1424458026885986,
      "step": 4255
    },
    {
      "epoch": 1.6231536865234375e-05,
      "step": 4255,
      "training_step_time": 10.32500410079956
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 3.144831657409668,
      "step": 4256
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 4256,
      "training_step_time": 9.916964769363403
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 3.145940065383911,
      "step": 4256
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 4256,
      "training_step_time": 9.936900615692139
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 3.1456494331359863,
      "step": 4256
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 4256,
      "training_step_time": 10.19946002960205
    },
    {
      "epoch": 1.62353515625e-05,
      "model_forward_time": 3.1456470489501953,
      "step": 4256
    },
    {
      "epoch": 1.62353515625e-05,
      "step": 4256,
      "training_step_time": 10.310859203338623
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 3.155839204788208,
      "step": 4257
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 4257,
      "training_step_time": 9.92999815940857
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 3.138472557067871,
      "step": 4257
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 4257,
      "training_step_time": 9.908526420593262
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 3.1379239559173584,
      "step": 4257
    },
    {
      "dynamics_loss_avg": 0.060838913358747956,
      "epoch": 1.6239166259765624e-05,
      "step": 4257
    },
    {
      "action_loss_avg": 0.022880238015204668,
      "epoch": 1.6239166259765624e-05,
      "step": 4257
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 4257,
      "training_step_time": 9.916060209274292
    },
    {
      "epoch": 1.6239166259765624e-05,
      "model_forward_time": 3.425957202911377,
      "step": 4257
    },
    {
      "epoch": 1.6239166259765624e-05,
      "step": 4257,
      "training_step_time": 10.30930781364441
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 3.1452667713165283,
      "step": 4258
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 4258,
      "training_step_time": 9.922363519668579
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 3.138293981552124,
      "step": 4258
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 4258,
      "training_step_time": 9.918692827224731
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 3.144444704055786,
      "step": 4258
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 4258,
      "training_step_time": 9.92299509048462
    },
    {
      "epoch": 1.624298095703125e-05,
      "model_forward_time": 3.139854907989502,
      "step": 4258
    },
    {
      "epoch": 1.624298095703125e-05,
      "step": 4258,
      "training_step_time": 10.314379215240479
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 3.1492879390716553,
      "step": 4259
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 4259,
      "training_step_time": 10.209118127822876
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 3.137221097946167,
      "step": 4259
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 4259,
      "training_step_time": 9.909771919250488
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 3.1441800594329834,
      "step": 4259
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 4259,
      "training_step_time": 9.922326803207397
    },
    {
      "epoch": 1.6246795654296876e-05,
      "model_forward_time": 2.436134099960327,
      "step": 4259
    },
    {
      "epoch": 1.6246795654296876e-05,
      "step": 4259,
      "training_step_time": 10.293856143951416
    },
    {
      "epoch": 1.62506103515625e-05,
      "grad_norm": 0.416593462228775,
      "learning_rate": 1e-05,
      "loss": 0.0931,
      "step": 4260
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 3.146164894104004,
      "step": 4260
    },
    {
      "dynamics_loss_avg": 0.05554693732410669,
      "epoch": 1.62506103515625e-05,
      "step": 4260
    },
    {
      "action_loss_avg": 0.025775883300229906,
      "epoch": 1.62506103515625e-05,
      "step": 4260
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 4260,
      "training_step_time": 9.926533937454224
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 3.4083337783813477,
      "step": 4260
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 4260,
      "training_step_time": 10.183159828186035
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 3.1429901123046875,
      "step": 4260
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 4260,
      "training_step_time": 9.917694807052612
    },
    {
      "epoch": 1.62506103515625e-05,
      "model_forward_time": 3.1373958587646484,
      "step": 4260
    },
    {
      "epoch": 1.62506103515625e-05,
      "step": 4260,
      "training_step_time": 10.017416715621948
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 3.1566824913024902,
      "step": 4261
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 4261,
      "training_step_time": 9.936768054962158
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 3.144880533218384,
      "step": 4261
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 4261,
      "training_step_time": 9.912156343460083
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 3.137310743331909,
      "step": 4261
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 4261,
      "training_step_time": 10.161451816558838
    },
    {
      "epoch": 1.6254425048828125e-05,
      "model_forward_time": 2.4321985244750977,
      "step": 4261
    },
    {
      "epoch": 1.6254425048828125e-05,
      "step": 4261,
      "training_step_time": 10.268157005310059
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 3.1501083374023438,
      "step": 4262
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 4262,
      "training_step_time": 9.917130470275879
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 3.1369543075561523,
      "step": 4262
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 4262,
      "training_step_time": 9.903845071792603
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 3.141834259033203,
      "step": 4262
    },
    {
      "dynamics_loss_avg": 0.06272031813859939,
      "epoch": 1.625823974609375e-05,
      "step": 4262
    },
    {
      "action_loss_avg": 0.023380645643919708,
      "epoch": 1.625823974609375e-05,
      "step": 4262
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 4262,
      "training_step_time": 9.916503429412842
    },
    {
      "epoch": 1.625823974609375e-05,
      "model_forward_time": 3.4020769596099854,
      "step": 4262
    },
    {
      "epoch": 1.625823974609375e-05,
      "step": 4262,
      "training_step_time": 10.318026065826416
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 3.1514909267425537,
      "step": 4263
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 4263,
      "training_step_time": 9.922815322875977
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 3.143575429916382,
      "step": 4263
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 4263,
      "training_step_time": 9.915361881256104
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 3.1445236206054688,
      "step": 4263
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 4263,
      "training_step_time": 9.919174909591675
    },
    {
      "epoch": 1.6262054443359375e-05,
      "model_forward_time": 3.1462504863739014,
      "step": 4263
    },
    {
      "epoch": 1.6262054443359375e-05,
      "step": 4263,
      "training_step_time": 10.3225998878479
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 3.4215240478515625,
      "step": 4264
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 4264,
      "training_step_time": 10.199057579040527
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 3.145305633544922,
      "step": 4264
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 4264,
      "training_step_time": 9.917482614517212
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 3.1413261890411377,
      "step": 4264
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 4264,
      "training_step_time": 9.915257215499878
    },
    {
      "epoch": 1.6265869140625e-05,
      "model_forward_time": 3.144218921661377,
      "step": 4264
    },
    {
      "epoch": 1.6265869140625e-05,
      "step": 4264,
      "training_step_time": 10.013864755630493
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 3.161102056503296,
      "step": 4265
    },
    {
      "dynamics_loss_avg": 0.07740015909075737,
      "epoch": 1.6269683837890624e-05,
      "step": 4265
    },
    {
      "action_loss_avg": 0.029783729650080203,
      "epoch": 1.6269683837890624e-05,
      "step": 4265
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 4265,
      "training_step_time": 9.93804931640625
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 2.4401402473449707,
      "step": 4265
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 4265,
      "training_step_time": 7.737085342407227
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 3.1376607418060303,
      "step": 4265
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 4265,
      "training_step_time": 9.911373615264893
    },
    {
      "epoch": 1.6269683837890624e-05,
      "model_forward_time": 3.1410486698150635,
      "step": 4265
    },
    {
      "epoch": 1.6269683837890624e-05,
      "step": 4265,
      "training_step_time": 10.32655930519104
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 3.1521506309509277,
      "step": 4266
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 4266,
      "training_step_time": 9.924133777618408
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 3.1378870010375977,
      "step": 4266
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 4266,
      "training_step_time": 9.909802675247192
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 3.4028351306915283,
      "step": 4266
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 4266,
      "training_step_time": 10.178097486495972
    },
    {
      "epoch": 1.627349853515625e-05,
      "model_forward_time": 3.149231195449829,
      "step": 4266
    },
    {
      "epoch": 1.627349853515625e-05,
      "step": 4266,
      "training_step_time": 10.29374647140503
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 3.1604933738708496,
      "step": 4267
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 4267,
      "training_step_time": 9.938805103302002
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 3.1379430294036865,
      "step": 4267
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 4267,
      "training_step_time": 9.914396286010742
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 3.139590263366699,
      "step": 4267
    },
    {
      "dynamics_loss_avg": 0.05535948211327195,
      "epoch": 1.6277313232421876e-05,
      "step": 4267
    },
    {
      "action_loss_avg": 0.025287855858914556,
      "epoch": 1.6277313232421876e-05,
      "step": 4267
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 4267,
      "training_step_time": 9.919165849685669
    },
    {
      "epoch": 1.6277313232421876e-05,
      "model_forward_time": 3.1390957832336426,
      "step": 4267
    },
    {
      "epoch": 1.6277313232421876e-05,
      "step": 4267,
      "training_step_time": 10.302175283432007
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 3.144443988800049,
      "step": 4268
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 4268,
      "training_step_time": 9.927962064743042
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 3.1384212970733643,
      "step": 4268
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 4268,
      "training_step_time": 9.915574073791504
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 3.1413865089416504,
      "step": 4268
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 4268,
      "training_step_time": 9.916647672653198
    },
    {
      "epoch": 1.62811279296875e-05,
      "model_forward_time": 3.135334014892578,
      "step": 4268
    },
    {
      "epoch": 1.62811279296875e-05,
      "step": 4268,
      "training_step_time": 10.290938377380371
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 3.420353651046753,
      "step": 4269
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 4269,
      "training_step_time": 10.193719625473022
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 3.1314785480499268,
      "step": 4269
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 4269,
      "training_step_time": 9.899974346160889
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 3.1361453533172607,
      "step": 4269
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 4269,
      "training_step_time": 9.909832954406738
    },
    {
      "epoch": 1.6284942626953125e-05,
      "model_forward_time": 3.1355960369110107,
      "step": 4269
    },
    {
      "epoch": 1.6284942626953125e-05,
      "step": 4269,
      "training_step_time": 10.312734842300415
    },
    {
      "epoch": 1.628875732421875e-05,
      "grad_norm": 0.5943332314491272,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 4270
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 3.1541526317596436,
      "step": 4270
    },
    {
      "dynamics_loss_avg": 0.0654199980199337,
      "epoch": 1.628875732421875e-05,
      "step": 4270
    },
    {
      "action_loss_avg": 0.026935991272330283,
      "epoch": 1.628875732421875e-05,
      "step": 4270
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 4270,
      "training_step_time": 9.938482522964478
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 2.4406492710113525,
      "step": 4270
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 4270,
      "training_step_time": 7.471468448638916
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 3.4143118858337402,
      "step": 4270
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 4270,
      "training_step_time": 10.189436912536621
    },
    {
      "epoch": 1.628875732421875e-05,
      "model_forward_time": 3.1487932205200195,
      "step": 4270
    },
    {
      "epoch": 1.628875732421875e-05,
      "step": 4270,
      "training_step_time": 10.333089351654053
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 3.1547513008117676,
      "step": 4271
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 4271,
      "training_step_time": 9.934242725372314
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 3.148261547088623,
      "step": 4271
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 4271,
      "training_step_time": 9.928955554962158
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 3.444833278656006,
      "step": 4271
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 4271,
      "training_step_time": 10.220922470092773
    },
    {
      "epoch": 1.6292572021484375e-05,
      "model_forward_time": 3.1517038345336914,
      "step": 4271
    },
    {
      "epoch": 1.6292572021484375e-05,
      "step": 4271,
      "training_step_time": 10.021856307983398
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 3.1605844497680664,
      "step": 4272
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 4272,
      "training_step_time": 9.97970700263977
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 3.1498799324035645,
      "step": 4272
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 4272,
      "training_step_time": 9.924061059951782
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 3.1452770233154297,
      "step": 4272
    },
    {
      "dynamics_loss_avg": 0.0707786187529564,
      "epoch": 1.629638671875e-05,
      "step": 4272
    },
    {
      "action_loss_avg": 0.034553711954504254,
      "epoch": 1.629638671875e-05,
      "step": 4272
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 4272,
      "training_step_time": 9.916919469833374
    },
    {
      "epoch": 1.629638671875e-05,
      "model_forward_time": 3.143439292907715,
      "step": 4272
    },
    {
      "epoch": 1.629638671875e-05,
      "step": 4272,
      "training_step_time": 10.015722036361694
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 3.4723846912384033,
      "step": 4273
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 4273,
      "training_step_time": 10.244763135910034
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 2.441284418106079,
      "step": 4273
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 4273,
      "training_step_time": 7.468508958816528
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 3.1402671337127686,
      "step": 4273
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 4273,
      "training_step_time": 9.907644748687744
    },
    {
      "epoch": 1.6300201416015624e-05,
      "model_forward_time": 3.1468279361724854,
      "step": 4273
    },
    {
      "epoch": 1.6300201416015624e-05,
      "step": 4273,
      "training_step_time": 10.337139129638672
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 2.4490933418273926,
      "step": 4274
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 4274,
      "training_step_time": 7.479494571685791
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 3.4323973655700684,
      "step": 4274
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 4274,
      "training_step_time": 10.203495979309082
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 3.1466569900512695,
      "step": 4274
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 4274,
      "training_step_time": 9.920596361160278
    },
    {
      "epoch": 1.630401611328125e-05,
      "model_forward_time": 3.1523284912109375,
      "step": 4274
    },
    {
      "epoch": 1.630401611328125e-05,
      "step": 4274,
      "training_step_time": 10.258729696273804
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 3.1656172275543213,
      "step": 4275
    },
    {
      "dynamics_loss_avg": 0.08181963786482811,
      "epoch": 1.6307830810546876e-05,
      "step": 4275
    },
    {
      "action_loss_avg": 0.029651937261223792,
      "epoch": 1.6307830810546876e-05,
      "step": 4275
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 4275,
      "training_step_time": 9.947488784790039
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 3.420940399169922,
      "step": 4275
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 4275,
      "training_step_time": 10.19087266921997
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 3.1389615535736084,
      "step": 4275
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 4275,
      "training_step_time": 9.91319990158081
    },
    {
      "epoch": 1.6307830810546876e-05,
      "model_forward_time": 3.144301176071167,
      "step": 4275
    },
    {
      "epoch": 1.6307830810546876e-05,
      "step": 4275,
      "training_step_time": 10.297693729400635
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 3.1500654220581055,
      "step": 4276
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 4276,
      "training_step_time": 9.921976804733276
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 3.1340534687042236,
      "step": 4276
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 4276,
      "training_step_time": 9.904425621032715
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 3.136030673980713,
      "step": 4276
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 4276,
      "training_step_time": 10.179234266281128
    },
    {
      "epoch": 1.63116455078125e-05,
      "model_forward_time": 3.138845443725586,
      "step": 4276
    },
    {
      "epoch": 1.63116455078125e-05,
      "step": 4276,
      "training_step_time": 10.003818273544312
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 3.160386323928833,
      "step": 4277
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 4277,
      "training_step_time": 9.928238868713379
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 3.1392197608947754,
      "step": 4277
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 4277,
      "training_step_time": 9.909548282623291
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 3.1470396518707275,
      "step": 4277
    },
    {
      "dynamics_loss_avg": 0.055192156508564946,
      "epoch": 1.6315460205078126e-05,
      "step": 4277
    },
    {
      "action_loss_avg": 0.024312828155234455,
      "epoch": 1.6315460205078126e-05,
      "step": 4277
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 4277,
      "training_step_time": 9.92410659790039
    },
    {
      "epoch": 1.6315460205078126e-05,
      "model_forward_time": 3.428767204284668,
      "step": 4277
    },
    {
      "epoch": 1.6315460205078126e-05,
      "step": 4277,
      "training_step_time": 10.291114807128906
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 3.14997935295105,
      "step": 4278
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 4278,
      "training_step_time": 9.921889066696167
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 3.1445255279541016,
      "step": 4278
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 4278,
      "training_step_time": 9.91508936882019
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 3.146402597427368,
      "step": 4278
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 4278,
      "training_step_time": 9.923488855361938
    },
    {
      "epoch": 1.631927490234375e-05,
      "model_forward_time": 3.1451621055603027,
      "step": 4278
    },
    {
      "epoch": 1.631927490234375e-05,
      "step": 4278,
      "training_step_time": 10.263922214508057
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 3.1484785079956055,
      "step": 4279
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 4279,
      "training_step_time": 10.190864562988281
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 3.140960931777954,
      "step": 4279
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 4279,
      "training_step_time": 9.915773630142212
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 3.1478958129882812,
      "step": 4279
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 4279,
      "training_step_time": 9.926602125167847
    },
    {
      "epoch": 1.6323089599609375e-05,
      "model_forward_time": 3.1470000743865967,
      "step": 4279
    },
    {
      "epoch": 1.6323089599609375e-05,
      "step": 4279,
      "training_step_time": 10.292643547058105
    },
    {
      "epoch": 1.6326904296875e-05,
      "grad_norm": 0.5626899600028992,
      "learning_rate": 1e-05,
      "loss": 0.0897,
      "step": 4280
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 3.156371831893921,
      "step": 4280
    },
    {
      "dynamics_loss_avg": 0.07436810452491045,
      "epoch": 1.6326904296875e-05,
      "step": 4280
    },
    {
      "action_loss_avg": 0.03601367045193911,
      "epoch": 1.6326904296875e-05,
      "step": 4280
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 4280,
      "training_step_time": 9.932801008224487
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 3.388808250427246,
      "step": 4280
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 4280,
      "training_step_time": 10.1621994972229
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 3.1374151706695557,
      "step": 4280
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 4280,
      "training_step_time": 9.911147356033325
    },
    {
      "epoch": 1.6326904296875e-05,
      "model_forward_time": 3.1456151008605957,
      "step": 4280
    },
    {
      "epoch": 1.6326904296875e-05,
      "step": 4280,
      "training_step_time": 10.305100440979004
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 3.1552603244781494,
      "step": 4281
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 4281,
      "training_step_time": 9.934232711791992
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 3.1355209350585938,
      "step": 4281
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 4281,
      "training_step_time": 9.904812574386597
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 3.4115989208221436,
      "step": 4281
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 4281,
      "training_step_time": 10.180450916290283
    },
    {
      "epoch": 1.6330718994140624e-05,
      "model_forward_time": 3.1341605186462402,
      "step": 4281
    },
    {
      "epoch": 1.6330718994140624e-05,
      "step": 4281,
      "training_step_time": 10.000451564788818
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 3.171074867248535,
      "step": 4282
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 4282,
      "training_step_time": 9.952331066131592
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 3.145862340927124,
      "step": 4282
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 4282,
      "training_step_time": 9.919283151626587
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 3.1378369331359863,
      "step": 4282
    },
    {
      "dynamics_loss_avg": 0.06782428082078695,
      "epoch": 1.633453369140625e-05,
      "step": 4282
    },
    {
      "action_loss_avg": 0.030661600548774003,
      "epoch": 1.633453369140625e-05,
      "step": 4282
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 4282,
      "training_step_time": 9.91100001335144
    },
    {
      "epoch": 1.633453369140625e-05,
      "model_forward_time": 3.4000988006591797,
      "step": 4282
    },
    {
      "epoch": 1.633453369140625e-05,
      "step": 4282,
      "training_step_time": 10.258902072906494
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 3.1389927864074707,
      "step": 4283
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 4283,
      "training_step_time": 9.907794713973999
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 3.1342523097991943,
      "step": 4283
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 4283,
      "training_step_time": 9.898982763290405
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 3.1318106651306152,
      "step": 4283
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 4283,
      "training_step_time": 9.897924661636353
    },
    {
      "epoch": 1.6338348388671876e-05,
      "model_forward_time": 3.1319634914398193,
      "step": 4283
    },
    {
      "epoch": 1.6338348388671876e-05,
      "step": 4283,
      "training_step_time": 10.275922060012817
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 3.396467447280884,
      "step": 4284
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 4284,
      "training_step_time": 10.162775754928589
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 3.1367335319519043,
      "step": 4284
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 4284,
      "training_step_time": 9.905177116394043
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 3.1296472549438477,
      "step": 4284
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 4284,
      "training_step_time": 9.89684796333313
    },
    {
      "epoch": 1.63421630859375e-05,
      "model_forward_time": 3.1342883110046387,
      "step": 4284
    },
    {
      "epoch": 1.63421630859375e-05,
      "step": 4284,
      "training_step_time": 10.296292066574097
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 3.154433488845825,
      "step": 4285
    },
    {
      "dynamics_loss_avg": 0.07118273563683034,
      "epoch": 1.6345977783203126e-05,
      "step": 4285
    },
    {
      "action_loss_avg": 0.02943383976817131,
      "epoch": 1.6345977783203126e-05,
      "step": 4285
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 4285,
      "training_step_time": 9.931510210037231
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 3.1347436904907227,
      "step": 4285
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 4285,
      "training_step_time": 10.185298681259155
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 3.1371681690216064,
      "step": 4285
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 4285,
      "training_step_time": 9.908457517623901
    },
    {
      "epoch": 1.6345977783203126e-05,
      "model_forward_time": 3.1375892162323,
      "step": 4285
    },
    {
      "epoch": 1.6345977783203126e-05,
      "step": 4285,
      "training_step_time": 10.298104524612427
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 3.1549630165100098,
      "step": 4286
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 4286,
      "training_step_time": 9.939016342163086
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 3.1333587169647217,
      "step": 4286
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 4286,
      "training_step_time": 9.903156280517578
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 3.4263360500335693,
      "step": 4286
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 4286,
      "training_step_time": 10.194462776184082
    },
    {
      "epoch": 1.634979248046875e-05,
      "model_forward_time": 3.1347081661224365,
      "step": 4286
    },
    {
      "epoch": 1.634979248046875e-05,
      "step": 4286,
      "training_step_time": 10.010571002960205
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 3.1453299522399902,
      "step": 4287
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 4287,
      "training_step_time": 9.93021035194397
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 3.1340131759643555,
      "step": 4287
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 4287,
      "training_step_time": 9.903350353240967
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 3.133843421936035,
      "step": 4287
    },
    {
      "dynamics_loss_avg": 0.06747591271996498,
      "epoch": 1.6353607177734375e-05,
      "step": 4287
    },
    {
      "action_loss_avg": 0.027086254954338074,
      "epoch": 1.6353607177734375e-05,
      "step": 4287
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 4287,
      "training_step_time": 9.905194759368896
    },
    {
      "epoch": 1.6353607177734375e-05,
      "model_forward_time": 3.13280987739563,
      "step": 4287
    },
    {
      "epoch": 1.6353607177734375e-05,
      "step": 4287,
      "training_step_time": 10.29339051246643
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 3.1542651653289795,
      "step": 4288
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 4288,
      "training_step_time": 9.928154706954956
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 3.1348822116851807,
      "step": 4288
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 4288,
      "training_step_time": 9.906885147094727
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 3.1364269256591797,
      "step": 4288
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 4288,
      "training_step_time": 9.915314674377441
    },
    {
      "epoch": 1.6357421875e-05,
      "model_forward_time": 3.1340975761413574,
      "step": 4288
    },
    {
      "epoch": 1.6357421875e-05,
      "step": 4288,
      "training_step_time": 10.010654211044312
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 3.465852737426758,
      "step": 4289
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 4289,
      "training_step_time": 10.240683555603027
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 3.1386330127716064,
      "step": 4289
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 4289,
      "training_step_time": 9.919433355331421
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 3.146970748901367,
      "step": 4289
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 4289,
      "training_step_time": 9.932513952255249
    },
    {
      "epoch": 1.6361236572265624e-05,
      "model_forward_time": 3.139277219772339,
      "step": 4289
    },
    {
      "epoch": 1.6361236572265624e-05,
      "step": 4289,
      "training_step_time": 10.255007266998291
    },
    {
      "epoch": 1.636505126953125e-05,
      "grad_norm": 0.49842214584350586,
      "learning_rate": 1e-05,
      "loss": 0.0929,
      "step": 4290
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 3.156670093536377,
      "step": 4290
    },
    {
      "dynamics_loss_avg": 0.053582153469324115,
      "epoch": 1.636505126953125e-05,
      "step": 4290
    },
    {
      "action_loss_avg": 0.02283151256851852,
      "epoch": 1.636505126953125e-05,
      "step": 4290
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 4290,
      "training_step_time": 9.941698789596558
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 3.150423526763916,
      "step": 4290
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 4290,
      "training_step_time": 10.213198184967041
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 3.147542715072632,
      "step": 4290
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 4290,
      "training_step_time": 9.937310695648193
    },
    {
      "epoch": 1.636505126953125e-05,
      "model_forward_time": 3.1467583179473877,
      "step": 4290
    },
    {
      "epoch": 1.636505126953125e-05,
      "step": 4290,
      "training_step_time": 10.313489437103271
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 3.161350727081299,
      "step": 4291
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 4291,
      "training_step_time": 9.939620971679688
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 3.14701509475708,
      "step": 4291
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 4291,
      "training_step_time": 9.926780223846436
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 3.471381902694702,
      "step": 4291
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 4291,
      "training_step_time": 10.249137163162231
    },
    {
      "epoch": 1.6368865966796876e-05,
      "model_forward_time": 3.1471190452575684,
      "step": 4291
    },
    {
      "epoch": 1.6368865966796876e-05,
      "step": 4291,
      "training_step_time": 10.290243864059448
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 3.1518185138702393,
      "step": 4292
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 4292,
      "training_step_time": 9.935481309890747
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 3.145791530609131,
      "step": 4292
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 4292,
      "training_step_time": 9.926193475723267
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 3.141261577606201,
      "step": 4292
    },
    {
      "dynamics_loss_avg": 0.06245103739202022,
      "epoch": 1.63726806640625e-05,
      "step": 4292
    },
    {
      "action_loss_avg": 0.02747227009385824,
      "epoch": 1.63726806640625e-05,
      "step": 4292
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 4292,
      "training_step_time": 9.919861316680908
    },
    {
      "epoch": 1.63726806640625e-05,
      "model_forward_time": 3.415487766265869,
      "step": 4292
    },
    {
      "epoch": 1.63726806640625e-05,
      "step": 4292,
      "training_step_time": 10.28505539894104
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 3.165747880935669,
      "step": 4293
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 4293,
      "training_step_time": 9.942073345184326
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 2.4408514499664307,
      "step": 4293
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 4293,
      "training_step_time": 7.466871023178101
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 3.137986660003662,
      "step": 4293
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 4293,
      "training_step_time": 9.910306215286255
    },
    {
      "epoch": 1.6376495361328126e-05,
      "model_forward_time": 2.437553882598877,
      "step": 4293
    },
    {
      "epoch": 1.6376495361328126e-05,
      "step": 4293,
      "training_step_time": 10.26508116722107
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 3.1532816886901855,
      "step": 4294
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 4294,
      "training_step_time": 10.210066318511963
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 3.1462390422821045,
      "step": 4294
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 4294,
      "training_step_time": 9.918657541275024
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 3.1416895389556885,
      "step": 4294
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 4294,
      "training_step_time": 9.91454005241394
    },
    {
      "epoch": 1.638031005859375e-05,
      "model_forward_time": 3.1488218307495117,
      "step": 4294
    },
    {
      "epoch": 1.638031005859375e-05,
      "step": 4294,
      "training_step_time": 10.268173217773438
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 3.1633126735687256,
      "step": 4295
    },
    {
      "dynamics_loss_avg": 0.06602956224232912,
      "epoch": 1.6384124755859375e-05,
      "step": 4295
    },
    {
      "action_loss_avg": 0.034607597161084414,
      "epoch": 1.6384124755859375e-05,
      "step": 4295
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 4295,
      "training_step_time": 9.946929931640625
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 3.4462623596191406,
      "step": 4295
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 4295,
      "training_step_time": 10.22439980506897
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 3.148819923400879,
      "step": 4295
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 4295,
      "training_step_time": 9.929691791534424
    },
    {
      "epoch": 1.6384124755859375e-05,
      "model_forward_time": 2.4441113471984863,
      "step": 4295
    },
    {
      "epoch": 1.6384124755859375e-05,
      "step": 4295,
      "training_step_time": 10.321819543838501
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 3.168834924697876,
      "step": 4296
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 4296,
      "training_step_time": 9.944794654846191
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 3.1415622234344482,
      "step": 4296
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 4296,
      "training_step_time": 9.920107364654541
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 3.1447761058807373,
      "step": 4296
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 4296,
      "training_step_time": 10.234400033950806
    },
    {
      "epoch": 1.6387939453125e-05,
      "model_forward_time": 3.1495273113250732,
      "step": 4296
    },
    {
      "epoch": 1.6387939453125e-05,
      "step": 4296,
      "training_step_time": 10.292733192443848
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 3.1635758876800537,
      "step": 4297
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 4297,
      "training_step_time": 9.948551893234253
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 3.14326548576355,
      "step": 4297
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 4297,
      "training_step_time": 9.9218590259552
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 3.15181303024292,
      "step": 4297
    },
    {
      "dynamics_loss_avg": 0.05296078734099865,
      "epoch": 1.6391754150390624e-05,
      "step": 4297
    },
    {
      "action_loss_avg": 0.02156477691605687,
      "epoch": 1.6391754150390624e-05,
      "step": 4297
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 4297,
      "training_step_time": 9.940208196640015
    },
    {
      "epoch": 1.6391754150390624e-05,
      "model_forward_time": 3.427762985229492,
      "step": 4297
    },
    {
      "epoch": 1.6391754150390624e-05,
      "step": 4297,
      "training_step_time": 10.30545449256897
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 3.1627869606018066,
      "step": 4298
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 4298,
      "training_step_time": 9.949733972549438
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 3.137820243835449,
      "step": 4298
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 4298,
      "training_step_time": 9.922940731048584
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 3.1441266536712646,
      "step": 4298
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 4298,
      "training_step_time": 9.928519248962402
    },
    {
      "epoch": 1.639556884765625e-05,
      "model_forward_time": 3.1412911415100098,
      "step": 4298
    },
    {
      "epoch": 1.639556884765625e-05,
      "step": 4298,
      "training_step_time": 10.029888391494751
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 3.160773992538452,
      "step": 4299
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 4299,
      "training_step_time": 10.222501516342163
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 3.1383752822875977,
      "step": 4299
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 4299,
      "training_step_time": 9.918925523757935
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 3.141711473464966,
      "step": 4299
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 4299,
      "training_step_time": 9.92300009727478
    },
    {
      "epoch": 1.6399383544921876e-05,
      "model_forward_time": 3.1395673751831055,
      "step": 4299
    },
    {
      "epoch": 1.6399383544921876e-05,
      "step": 4299,
      "training_step_time": 10.30160903930664
    },
    {
      "epoch": 1.64031982421875e-05,
      "grad_norm": 0.46119973063468933,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 4300
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 3.1484577655792236,
      "step": 4300
    },
    {
      "dynamics_loss_avg": 0.05767385847866535,
      "epoch": 1.64031982421875e-05,
      "step": 4300
    },
    {
      "action_loss_avg": 0.025323971174657345,
      "epoch": 1.64031982421875e-05,
      "step": 4300
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 4300,
      "training_step_time": 9.9315505027771
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 3.4050967693328857,
      "step": 4300
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 4300,
      "training_step_time": 10.18411660194397
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 3.1430039405822754,
      "step": 4300
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 4300,
      "training_step_time": 9.927133560180664
    },
    {
      "epoch": 1.64031982421875e-05,
      "model_forward_time": 2.4393386840820312,
      "step": 4300
    },
    {
      "epoch": 1.64031982421875e-05,
      "step": 4300,
      "training_step_time": 10.302018165588379
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 3.1521685123443604,
      "step": 4301
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 4301,
      "training_step_time": 9.931119203567505
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 3.1390116214752197,
      "step": 4301
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 4301,
      "training_step_time": 9.91611385345459
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 3.1405794620513916,
      "step": 4301
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 4301,
      "training_step_time": 10.227769613265991
    },
    {
      "epoch": 1.6407012939453126e-05,
      "model_forward_time": 3.1383779048919678,
      "step": 4301
    },
    {
      "epoch": 1.6407012939453126e-05,
      "step": 4301,
      "training_step_time": 10.295015811920166
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 3.1561012268066406,
      "step": 4302
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 4302,
      "training_step_time": 9.935494899749756
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 3.138047933578491,
      "step": 4302
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 4302,
      "training_step_time": 9.917752027511597
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 3.1353187561035156,
      "step": 4302
    },
    {
      "dynamics_loss_avg": 0.06283454112708568,
      "epoch": 1.641082763671875e-05,
      "step": 4302
    },
    {
      "action_loss_avg": 0.028239472117275,
      "epoch": 1.641082763671875e-05,
      "step": 4302
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 4302,
      "training_step_time": 9.918562173843384
    },
    {
      "epoch": 1.641082763671875e-05,
      "model_forward_time": 3.4017326831817627,
      "step": 4302
    },
    {
      "epoch": 1.641082763671875e-05,
      "step": 4302,
      "training_step_time": 10.316038846969604
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 3.1517603397369385,
      "step": 4303
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 4303,
      "training_step_time": 9.934168815612793
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 3.1463420391082764,
      "step": 4303
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 4303,
      "training_step_time": 9.930927276611328
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 2.4387295246124268,
      "step": 4303
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 4303,
      "training_step_time": 7.471736192703247
    },
    {
      "epoch": 1.6414642333984375e-05,
      "model_forward_time": 3.145108938217163,
      "step": 4303
    },
    {
      "epoch": 1.6414642333984375e-05,
      "step": 4303,
      "training_step_time": 10.303337335586548
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 3.1658341884613037,
      "step": 4304
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 4304,
      "training_step_time": 10.237883806228638
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 3.1450397968292236,
      "step": 4304
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 4304,
      "training_step_time": 9.929476737976074
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 3.1445178985595703,
      "step": 4304
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 4304,
      "training_step_time": 9.928392171859741
    },
    {
      "epoch": 1.641845703125e-05,
      "model_forward_time": 3.1470251083374023,
      "step": 4304
    },
    {
      "epoch": 1.641845703125e-05,
      "step": 4304,
      "training_step_time": 10.2297203540802
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 3.1610538959503174,
      "step": 4305
    },
    {
      "dynamics_loss_avg": 0.060040184482932094,
      "epoch": 1.6422271728515624e-05,
      "step": 4305
    },
    {
      "action_loss_avg": 0.022833923855796457,
      "epoch": 1.6422271728515624e-05,
      "step": 4305
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 4305,
      "training_step_time": 9.9447340965271
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 3.1439995765686035,
      "step": 4305
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 4305,
      "training_step_time": 10.209867715835571
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 3.1423206329345703,
      "step": 4305
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 4305,
      "training_step_time": 9.917373895645142
    },
    {
      "epoch": 1.6422271728515624e-05,
      "model_forward_time": 3.1474769115448,
      "step": 4305
    },
    {
      "epoch": 1.6422271728515624e-05,
      "step": 4305,
      "training_step_time": 10.278021812438965
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 3.1616101264953613,
      "step": 4306
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 4306,
      "training_step_time": 9.941097259521484
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 3.1355884075164795,
      "step": 4306
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 4306,
      "training_step_time": 9.911697387695312
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 3.1420505046844482,
      "step": 4306
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 4306,
      "training_step_time": 10.19655966758728
    },
    {
      "epoch": 1.642608642578125e-05,
      "model_forward_time": 3.1423258781433105,
      "step": 4306
    },
    {
      "epoch": 1.642608642578125e-05,
      "step": 4306,
      "training_step_time": 10.310546159744263
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 3.1621029376983643,
      "step": 4307
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 4307,
      "training_step_time": 9.939079761505127
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 3.140291452407837,
      "step": 4307
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 4307,
      "training_step_time": 9.91705060005188
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 2.438532829284668,
      "step": 4307
    },
    {
      "dynamics_loss_avg": 0.06324095427989959,
      "epoch": 1.6429901123046876e-05,
      "step": 4307
    },
    {
      "action_loss_avg": 0.03110960330814123,
      "epoch": 1.6429901123046876e-05,
      "step": 4307
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 4307,
      "training_step_time": 7.479233503341675
    },
    {
      "epoch": 1.6429901123046876e-05,
      "model_forward_time": 3.1379706859588623,
      "step": 4307
    },
    {
      "epoch": 1.6429901123046876e-05,
      "step": 4307,
      "training_step_time": 10.302225351333618
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 3.161271810531616,
      "step": 4308
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 4308,
      "training_step_time": 9.940102577209473
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 3.1449856758117676,
      "step": 4308
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 4308,
      "training_step_time": 9.925789833068848
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 3.1411056518554688,
      "step": 4308
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 4308,
      "training_step_time": 9.919820070266724
    },
    {
      "epoch": 1.64337158203125e-05,
      "model_forward_time": 3.143594741821289,
      "step": 4308
    },
    {
      "epoch": 1.64337158203125e-05,
      "step": 4308,
      "training_step_time": 10.019120931625366
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 3.4361016750335693,
      "step": 4309
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 4309,
      "training_step_time": 10.218764543533325
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 3.1437642574310303,
      "step": 4309
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 4309,
      "training_step_time": 9.923480987548828
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 3.1439669132232666,
      "step": 4309
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 4309,
      "training_step_time": 9.925739765167236
    },
    {
      "epoch": 1.6437530517578126e-05,
      "model_forward_time": 3.1414928436279297,
      "step": 4309
    },
    {
      "epoch": 1.6437530517578126e-05,
      "step": 4309,
      "training_step_time": 10.019017696380615
    },
    {
      "epoch": 1.644134521484375e-05,
      "grad_norm": 0.397225022315979,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 4310
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 3.1633498668670654,
      "step": 4310
    },
    {
      "dynamics_loss_avg": 0.04533624537289142,
      "epoch": 1.644134521484375e-05,
      "step": 4310
    },
    {
      "action_loss_avg": 0.020729468297213317,
      "epoch": 1.644134521484375e-05,
      "step": 4310
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 4310,
      "training_step_time": 9.953662157058716
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 3.4083359241485596,
      "step": 4310
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 4310,
      "training_step_time": 10.190560340881348
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 3.1471011638641357,
      "step": 4310
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 4310,
      "training_step_time": 9.930001020431519
    },
    {
      "epoch": 1.644134521484375e-05,
      "model_forward_time": 3.146843433380127,
      "step": 4310
    },
    {
      "epoch": 1.644134521484375e-05,
      "step": 4310,
      "training_step_time": 10.2825288772583
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 3.1507935523986816,
      "step": 4311
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 4311,
      "training_step_time": 9.934622287750244
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 3.149707078933716,
      "step": 4311
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 4311,
      "training_step_time": 9.931259870529175
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 3.4572079181671143,
      "step": 4311
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 4311,
      "training_step_time": 10.241623640060425
    },
    {
      "epoch": 1.6445159912109375e-05,
      "model_forward_time": 3.1444761753082275,
      "step": 4311
    },
    {
      "epoch": 1.6445159912109375e-05,
      "step": 4311,
      "training_step_time": 10.289392232894897
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 3.164580821990967,
      "step": 4312
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 4312,
      "training_step_time": 9.946191310882568
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 3.1450085639953613,
      "step": 4312
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 4312,
      "training_step_time": 9.92427134513855
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 3.1403350830078125,
      "step": 4312
    },
    {
      "dynamics_loss_avg": 0.06165028307586908,
      "epoch": 1.6448974609375e-05,
      "step": 4312
    },
    {
      "action_loss_avg": 0.027751311007887126,
      "epoch": 1.6448974609375e-05,
      "step": 4312
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 4312,
      "training_step_time": 9.923128604888916
    },
    {
      "epoch": 1.6448974609375e-05,
      "model_forward_time": 3.4179203510284424,
      "step": 4312
    },
    {
      "epoch": 1.6448974609375e-05,
      "step": 4312,
      "training_step_time": 10.307276964187622
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 3.166236400604248,
      "step": 4313
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 4313,
      "training_step_time": 9.946902990341187
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 3.1447463035583496,
      "step": 4313
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 4313,
      "training_step_time": 9.92386245727539
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 3.1376492977142334,
      "step": 4313
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 4313,
      "training_step_time": 9.917479038238525
    },
    {
      "epoch": 1.6452789306640624e-05,
      "model_forward_time": 3.144871711730957,
      "step": 4313
    },
    {
      "epoch": 1.6452789306640624e-05,
      "step": 4313,
      "training_step_time": 10.325470685958862
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 3.444152355194092,
      "step": 4314
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 4314,
      "training_step_time": 10.222455739974976
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 3.1456332206726074,
      "step": 4314
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 4314,
      "training_step_time": 9.929318904876709
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 3.144273281097412,
      "step": 4314
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 4314,
      "training_step_time": 9.92500925064087
    },
    {
      "epoch": 1.645660400390625e-05,
      "model_forward_time": 3.149801731109619,
      "step": 4314
    },
    {
      "epoch": 1.645660400390625e-05,
      "step": 4314,
      "training_step_time": 10.29158353805542
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 3.1809000968933105,
      "step": 4315
    },
    {
      "dynamics_loss_avg": 0.07874508276581764,
      "epoch": 1.6460418701171876e-05,
      "step": 4315
    },
    {
      "action_loss_avg": 0.035623466689139606,
      "epoch": 1.6460418701171876e-05,
      "step": 4315
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 4315,
      "training_step_time": 9.962624073028564
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 3.4262843132019043,
      "step": 4315
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 4315,
      "training_step_time": 10.208001375198364
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 3.1424496173858643,
      "step": 4315
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 4315,
      "training_step_time": 9.922998905181885
    },
    {
      "epoch": 1.6460418701171876e-05,
      "model_forward_time": 3.1446778774261475,
      "step": 4315
    },
    {
      "epoch": 1.6460418701171876e-05,
      "step": 4315,
      "training_step_time": 10.274759769439697
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 3.1615731716156006,
      "step": 4316
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 4316,
      "training_step_time": 9.943542242050171
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 3.1412365436553955,
      "step": 4316
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 4316,
      "training_step_time": 9.919039249420166
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 3.1431307792663574,
      "step": 4316
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 4316,
      "training_step_time": 10.209823846817017
    },
    {
      "epoch": 1.64642333984375e-05,
      "model_forward_time": 3.140094757080078,
      "step": 4316
    },
    {
      "epoch": 1.64642333984375e-05,
      "step": 4316,
      "training_step_time": 10.326852798461914
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 3.1735260486602783,
      "step": 4317
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 4317,
      "training_step_time": 9.950942754745483
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 3.139321804046631,
      "step": 4317
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 4317,
      "training_step_time": 9.918687582015991
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 3.1481776237487793,
      "step": 4317
    },
    {
      "dynamics_loss_avg": 0.06350276134908199,
      "epoch": 1.6468048095703126e-05,
      "step": 4317
    },
    {
      "action_loss_avg": 0.02546475697308779,
      "epoch": 1.6468048095703126e-05,
      "step": 4317
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 4317,
      "training_step_time": 9.931222915649414
    },
    {
      "epoch": 1.6468048095703126e-05,
      "model_forward_time": 3.426128387451172,
      "step": 4317
    },
    {
      "epoch": 1.6468048095703126e-05,
      "step": 4317,
      "training_step_time": 10.301070928573608
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 3.15370774269104,
      "step": 4318
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 4318,
      "training_step_time": 9.938118696212769
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 3.140756607055664,
      "step": 4318
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 4318,
      "training_step_time": 9.921343564987183
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 3.1470789909362793,
      "step": 4318
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 4318,
      "training_step_time": 9.930755138397217
    },
    {
      "epoch": 1.647186279296875e-05,
      "model_forward_time": 3.14400315284729,
      "step": 4318
    },
    {
      "epoch": 1.647186279296875e-05,
      "step": 4318,
      "training_step_time": 10.2870774269104
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 3.176877975463867,
      "step": 4319
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 4319,
      "training_step_time": 10.229348182678223
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 3.13950252532959,
      "step": 4319
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 4319,
      "training_step_time": 9.919001340866089
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 3.146062135696411,
      "step": 4319
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 4319,
      "training_step_time": 9.9246346950531
    },
    {
      "epoch": 1.6475677490234375e-05,
      "model_forward_time": 3.140873432159424,
      "step": 4319
    },
    {
      "epoch": 1.6475677490234375e-05,
      "step": 4319,
      "training_step_time": 10.033688306808472
    },
    {
      "epoch": 1.64794921875e-05,
      "grad_norm": 0.35682979226112366,
      "learning_rate": 1e-05,
      "loss": 0.089,
      "step": 4320
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 3.159166097640991,
      "step": 4320
    },
    {
      "dynamics_loss_avg": 0.0622555548325181,
      "epoch": 1.64794921875e-05,
      "step": 4320
    },
    {
      "action_loss_avg": 0.023789649363607168,
      "epoch": 1.64794921875e-05,
      "step": 4320
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 4320,
      "training_step_time": 9.939229011535645
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 2.7177529335021973,
      "step": 4320
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 4320,
      "training_step_time": 7.746591806411743
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 3.1427974700927734,
      "step": 4320
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 4320,
      "training_step_time": 9.920877695083618
    },
    {
      "epoch": 1.64794921875e-05,
      "model_forward_time": 3.1417126655578613,
      "step": 4320
    },
    {
      "epoch": 1.64794921875e-05,
      "step": 4320,
      "training_step_time": 10.26930046081543
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 3.1617517471313477,
      "step": 4321
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 4321,
      "training_step_time": 9.9431893825531
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 3.143925428390503,
      "step": 4321
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 4321,
      "training_step_time": 9.93389105796814
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 3.143561363220215,
      "step": 4321
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 4321,
      "training_step_time": 10.199879884719849
    },
    {
      "epoch": 1.6483306884765624e-05,
      "model_forward_time": 2.437566041946411,
      "step": 4321
    },
    {
      "epoch": 1.6483306884765624e-05,
      "step": 4321,
      "training_step_time": 10.303268432617188
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 3.183058500289917,
      "step": 4322
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 4322,
      "training_step_time": 9.963095664978027
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 3.1430771350860596,
      "step": 4322
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 4322,
      "training_step_time": 9.922478914260864
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 3.1419637203216553,
      "step": 4322
    },
    {
      "dynamics_loss_avg": 0.06135910674929619,
      "epoch": 1.648712158203125e-05,
      "step": 4322
    },
    {
      "action_loss_avg": 0.020363904442638157,
      "epoch": 1.648712158203125e-05,
      "step": 4322
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 4322,
      "training_step_time": 9.923028469085693
    },
    {
      "epoch": 1.648712158203125e-05,
      "model_forward_time": 3.4103219509124756,
      "step": 4322
    },
    {
      "epoch": 1.648712158203125e-05,
      "step": 4322,
      "training_step_time": 10.285855054855347
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 3.167904853820801,
      "step": 4323
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 4323,
      "training_step_time": 9.94626259803772
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 3.140392541885376,
      "step": 4323
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 4323,
      "training_step_time": 9.918465614318848
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 3.143496513366699,
      "step": 4323
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 4323,
      "training_step_time": 9.922173976898193
    },
    {
      "epoch": 1.6490936279296877e-05,
      "model_forward_time": 3.143291473388672,
      "step": 4323
    },
    {
      "epoch": 1.6490936279296877e-05,
      "step": 4323,
      "training_step_time": 10.319982290267944
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 3.174243688583374,
      "step": 4324
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 4324,
      "training_step_time": 10.227967500686646
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 3.14269757270813,
      "step": 4324
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 4324,
      "training_step_time": 9.921302556991577
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 3.1377084255218506,
      "step": 4324
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 4324,
      "training_step_time": 9.917799949645996
    },
    {
      "epoch": 1.64947509765625e-05,
      "model_forward_time": 3.142441749572754,
      "step": 4324
    },
    {
      "epoch": 1.64947509765625e-05,
      "step": 4324,
      "training_step_time": 10.313454389572144
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 3.15175724029541,
      "step": 4325
    },
    {
      "dynamics_loss_avg": 0.07319913096725941,
      "epoch": 1.6498565673828126e-05,
      "step": 4325
    },
    {
      "action_loss_avg": 0.029735776875168086,
      "epoch": 1.6498565673828126e-05,
      "step": 4325
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 4325,
      "training_step_time": 9.939397096633911
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 3.416719675064087,
      "step": 4325
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 4325,
      "training_step_time": 10.198564291000366
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 3.1382088661193848,
      "step": 4325
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 4325,
      "training_step_time": 9.923927068710327
    },
    {
      "epoch": 1.6498565673828126e-05,
      "model_forward_time": 3.1444766521453857,
      "step": 4325
    },
    {
      "epoch": 1.6498565673828126e-05,
      "step": 4325,
      "training_step_time": 10.29843282699585
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 3.146289825439453,
      "step": 4326
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 4326,
      "training_step_time": 9.927273511886597
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 3.1371166706085205,
      "step": 4326
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 4326,
      "training_step_time": 9.916332483291626
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 3.3944454193115234,
      "step": 4326
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 4326,
      "training_step_time": 10.172831058502197
    },
    {
      "epoch": 1.650238037109375e-05,
      "model_forward_time": 3.1412994861602783,
      "step": 4326
    },
    {
      "epoch": 1.650238037109375e-05,
      "step": 4326,
      "training_step_time": 10.032469749450684
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 3.150158166885376,
      "step": 4327
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 4327,
      "training_step_time": 9.932555437088013
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 3.137457847595215,
      "step": 4327
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 4327,
      "training_step_time": 9.915903329849243
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 3.1396892070770264,
      "step": 4327
    },
    {
      "dynamics_loss_avg": 0.0635527279227972,
      "epoch": 1.6506195068359375e-05,
      "step": 4327
    },
    {
      "action_loss_avg": 0.026444311905652284,
      "epoch": 1.6506195068359375e-05,
      "step": 4327
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 4327,
      "training_step_time": 9.923082113265991
    },
    {
      "epoch": 1.6506195068359375e-05,
      "model_forward_time": 3.4530155658721924,
      "step": 4327
    },
    {
      "epoch": 1.6506195068359375e-05,
      "step": 4327,
      "training_step_time": 10.32764220237732
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 3.1755359172821045,
      "step": 4328
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 4328,
      "training_step_time": 9.959295988082886
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 3.137399673461914,
      "step": 4328
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 4328,
      "training_step_time": 9.918591737747192
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 3.141486883163452,
      "step": 4328
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 4328,
      "training_step_time": 9.923565149307251
    },
    {
      "epoch": 1.6510009765625e-05,
      "model_forward_time": 3.1357743740081787,
      "step": 4328
    },
    {
      "epoch": 1.6510009765625e-05,
      "step": 4328,
      "training_step_time": 10.022755861282349
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 3.449986696243286,
      "step": 4329
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 4329,
      "training_step_time": 10.22707223892212
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 3.1367640495300293,
      "step": 4329
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 4329,
      "training_step_time": 9.912616729736328
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 3.1429219245910645,
      "step": 4329
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 4329,
      "training_step_time": 9.924137353897095
    },
    {
      "epoch": 1.6513824462890624e-05,
      "model_forward_time": 3.1423964500427246,
      "step": 4329
    },
    {
      "epoch": 1.6513824462890624e-05,
      "step": 4329,
      "training_step_time": 10.29073429107666
    },
    {
      "epoch": 1.651763916015625e-05,
      "grad_norm": 0.37908411026000977,
      "learning_rate": 1e-05,
      "loss": 0.0907,
      "step": 4330
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 3.1720447540283203,
      "step": 4330
    },
    {
      "dynamics_loss_avg": 0.0553280470892787,
      "epoch": 1.651763916015625e-05,
      "step": 4330
    },
    {
      "action_loss_avg": 0.030622602533549072,
      "epoch": 1.651763916015625e-05,
      "step": 4330
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 4330,
      "training_step_time": 9.955795049667358
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 3.4073429107666016,
      "step": 4330
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 4330,
      "training_step_time": 10.182636260986328
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 3.139653444290161,
      "step": 4330
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 4330,
      "training_step_time": 9.91614294052124
    },
    {
      "epoch": 1.651763916015625e-05,
      "model_forward_time": 3.1375699043273926,
      "step": 4330
    },
    {
      "epoch": 1.651763916015625e-05,
      "step": 4330,
      "training_step_time": 10.31967043876648
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 3.143808364868164,
      "step": 4331
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 4331,
      "training_step_time": 9.92055058479309
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 3.1375555992126465,
      "step": 4331
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 4331,
      "training_step_time": 9.916006803512573
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 3.4094998836517334,
      "step": 4331
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 4331,
      "training_step_time": 10.1764657497406
    },
    {
      "epoch": 1.6521453857421877e-05,
      "model_forward_time": 3.1372056007385254,
      "step": 4331
    },
    {
      "epoch": 1.6521453857421877e-05,
      "step": 4331,
      "training_step_time": 10.020655632019043
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 3.150442600250244,
      "step": 4332
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 4332,
      "training_step_time": 9.927100419998169
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 3.140841245651245,
      "step": 4332
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 4332,
      "training_step_time": 9.9177885055542
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 3.1365790367126465,
      "step": 4332
    },
    {
      "dynamics_loss_avg": 0.05467072781175375,
      "epoch": 1.65252685546875e-05,
      "step": 4332
    },
    {
      "action_loss_avg": 0.019289253279566765,
      "epoch": 1.65252685546875e-05,
      "step": 4332
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 4332,
      "training_step_time": 9.924902200698853
    },
    {
      "epoch": 1.65252685546875e-05,
      "model_forward_time": 3.40350341796875,
      "step": 4332
    },
    {
      "epoch": 1.65252685546875e-05,
      "step": 4332,
      "training_step_time": 10.304754972457886
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 3.146984577178955,
      "step": 4333
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 4333,
      "training_step_time": 9.930098533630371
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 3.1404614448547363,
      "step": 4333
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 4333,
      "training_step_time": 9.925127506256104
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 3.1389613151550293,
      "step": 4333
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 4333,
      "training_step_time": 9.923017501831055
    },
    {
      "epoch": 1.6529083251953126e-05,
      "model_forward_time": 3.1430318355560303,
      "step": 4333
    },
    {
      "epoch": 1.6529083251953126e-05,
      "step": 4333,
      "training_step_time": 10.042382955551147
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 3.4298596382141113,
      "step": 4334
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 4334,
      "training_step_time": 10.214820861816406
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 3.144826650619507,
      "step": 4334
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 4334,
      "training_step_time": 9.930217504501343
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 3.14347243309021,
      "step": 4334
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 4334,
      "training_step_time": 9.928114414215088
    },
    {
      "epoch": 1.653289794921875e-05,
      "model_forward_time": 3.1471292972564697,
      "step": 4334
    },
    {
      "epoch": 1.653289794921875e-05,
      "step": 4334,
      "training_step_time": 10.336633205413818
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 3.157315492630005,
      "step": 4335
    },
    {
      "dynamics_loss_avg": 0.06705551333725453,
      "epoch": 1.6536712646484375e-05,
      "step": 4335
    },
    {
      "action_loss_avg": 0.025595233682543038,
      "epoch": 1.6536712646484375e-05,
      "step": 4335
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 4335,
      "training_step_time": 9.944072246551514
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 3.4133591651916504,
      "step": 4335
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 4335,
      "training_step_time": 10.192610740661621
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 3.1375551223754883,
      "step": 4335
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 4335,
      "training_step_time": 9.91791319847107
    },
    {
      "epoch": 1.6536712646484375e-05,
      "model_forward_time": 3.1374566555023193,
      "step": 4335
    },
    {
      "epoch": 1.6536712646484375e-05,
      "step": 4335,
      "training_step_time": 10.292584896087646
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 3.1445114612579346,
      "step": 4336
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 4336,
      "training_step_time": 9.919732093811035
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 3.1285834312438965,
      "step": 4336
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 4336,
      "training_step_time": 9.903395414352417
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 3.1324622631073,
      "step": 4336
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 4336,
      "training_step_time": 10.191309213638306
    },
    {
      "epoch": 1.654052734375e-05,
      "model_forward_time": 3.136384963989258,
      "step": 4336
    },
    {
      "epoch": 1.654052734375e-05,
      "step": 4336,
      "training_step_time": 10.246246099472046
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 3.155089855194092,
      "step": 4337
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 4337,
      "training_step_time": 9.93031930923462
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 3.1316447257995605,
      "step": 4337
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 4337,
      "training_step_time": 9.901479244232178
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 3.139691114425659,
      "step": 4337
    },
    {
      "dynamics_loss_avg": 0.04745870530605316,
      "epoch": 1.6544342041015624e-05,
      "step": 4337
    },
    {
      "action_loss_avg": 0.017219450301490725,
      "epoch": 1.6544342041015624e-05,
      "step": 4337
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 4337,
      "training_step_time": 9.916674375534058
    },
    {
      "epoch": 1.6544342041015624e-05,
      "model_forward_time": 3.4200141429901123,
      "step": 4337
    },
    {
      "epoch": 1.6544342041015624e-05,
      "step": 4337,
      "training_step_time": 10.288540840148926
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 3.1541693210601807,
      "step": 4338
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 4338,
      "training_step_time": 9.929946899414062
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 3.138812780380249,
      "step": 4338
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 4338,
      "training_step_time": 9.915992021560669
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 3.141432046890259,
      "step": 4338
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 4338,
      "training_step_time": 9.924425601959229
    },
    {
      "epoch": 1.654815673828125e-05,
      "model_forward_time": 3.1408631801605225,
      "step": 4338
    },
    {
      "epoch": 1.654815673828125e-05,
      "step": 4338,
      "training_step_time": 10.292612552642822
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 3.1666910648345947,
      "step": 4339
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 4339,
      "training_step_time": 10.227001428604126
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 3.1355583667755127,
      "step": 4339
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 4339,
      "training_step_time": 9.923616886138916
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 3.1401052474975586,
      "step": 4339
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 4339,
      "training_step_time": 9.918678045272827
    },
    {
      "epoch": 1.6551971435546877e-05,
      "model_forward_time": 3.135732650756836,
      "step": 4339
    },
    {
      "epoch": 1.6551971435546877e-05,
      "step": 4339,
      "training_step_time": 10.032346963882446
    },
    {
      "epoch": 1.65557861328125e-05,
      "grad_norm": 0.3589949607849121,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 4340
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 3.1668150424957275,
      "step": 4340
    },
    {
      "dynamics_loss_avg": 0.07410582676529884,
      "epoch": 1.65557861328125e-05,
      "step": 4340
    },
    {
      "action_loss_avg": 0.03372572902590036,
      "epoch": 1.65557861328125e-05,
      "step": 4340
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 4340,
      "training_step_time": 9.95014500617981
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 3.4063186645507812,
      "step": 4340
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 4340,
      "training_step_time": 10.181200742721558
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 2.438483238220215,
      "step": 4340
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 4340,
      "training_step_time": 7.471973896026611
    },
    {
      "epoch": 1.65557861328125e-05,
      "model_forward_time": 3.140964984893799,
      "step": 4340
    },
    {
      "epoch": 1.65557861328125e-05,
      "step": 4340,
      "training_step_time": 10.293140172958374
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 3.1933536529541016,
      "step": 4341
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 4341,
      "training_step_time": 9.979593515396118
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 3.146911144256592,
      "step": 4341
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 4341,
      "training_step_time": 9.927972793579102
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 3.137892007827759,
      "step": 4341
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 4341,
      "training_step_time": 10.18882441520691
    },
    {
      "epoch": 1.6559600830078126e-05,
      "model_forward_time": 3.1436655521392822,
      "step": 4341
    },
    {
      "epoch": 1.6559600830078126e-05,
      "step": 4341,
      "training_step_time": 10.246642112731934
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 3.1537587642669678,
      "step": 4342
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 4342,
      "training_step_time": 9.932651996612549
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 3.1414060592651367,
      "step": 4342
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 4342,
      "training_step_time": 9.918928623199463
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 3.139134168624878,
      "step": 4342
    },
    {
      "dynamics_loss_avg": 0.06916142497211694,
      "epoch": 1.656341552734375e-05,
      "step": 4342
    },
    {
      "action_loss_avg": 0.02676838687621057,
      "epoch": 1.656341552734375e-05,
      "step": 4342
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 4342,
      "training_step_time": 9.916899919509888
    },
    {
      "epoch": 1.656341552734375e-05,
      "model_forward_time": 3.4007654190063477,
      "step": 4342
    },
    {
      "epoch": 1.656341552734375e-05,
      "step": 4342,
      "training_step_time": 10.28298020362854
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 3.1577930450439453,
      "step": 4343
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 4343,
      "training_step_time": 9.934278964996338
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 3.149850606918335,
      "step": 4343
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 4343,
      "training_step_time": 9.92737102508545
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 3.142348289489746,
      "step": 4343
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 4343,
      "training_step_time": 9.922468900680542
    },
    {
      "epoch": 1.6567230224609375e-05,
      "model_forward_time": 3.141467332839966,
      "step": 4343
    },
    {
      "epoch": 1.6567230224609375e-05,
      "step": 4343,
      "training_step_time": 10.295485973358154
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 3.4308676719665527,
      "step": 4344
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 4344,
      "training_step_time": 10.211627721786499
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 3.1383934020996094,
      "step": 4344
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 4344,
      "training_step_time": 9.916739463806152
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 3.133345365524292,
      "step": 4344
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 4344,
      "training_step_time": 9.908600091934204
    },
    {
      "epoch": 1.6571044921875e-05,
      "model_forward_time": 3.1295077800750732,
      "step": 4344
    },
    {
      "epoch": 1.6571044921875e-05,
      "step": 4344,
      "training_step_time": 10.027705192565918
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 3.1436851024627686,
      "step": 4345
    },
    {
      "dynamics_loss_avg": 0.059296334721148015,
      "epoch": 1.6574859619140624e-05,
      "step": 4345
    },
    {
      "action_loss_avg": 0.02354589644819498,
      "epoch": 1.6574859619140624e-05,
      "step": 4345
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 4345,
      "training_step_time": 9.924863338470459
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 3.1372058391571045,
      "step": 4345
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 4345,
      "training_step_time": 10.241642951965332
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 3.1383402347564697,
      "step": 4345
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 4345,
      "training_step_time": 9.918669939041138
    },
    {
      "epoch": 1.6574859619140624e-05,
      "model_forward_time": 2.4305272102355957,
      "step": 4345
    },
    {
      "epoch": 1.6574859619140624e-05,
      "step": 4345,
      "training_step_time": 10.016132354736328
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 3.1744236946105957,
      "step": 4346
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 4346,
      "training_step_time": 9.956786870956421
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 3.1357643604278564,
      "step": 4346
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 4346,
      "training_step_time": 9.915335416793823
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 3.1423633098602295,
      "step": 4346
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 4346,
      "training_step_time": 9.924599170684814
    },
    {
      "epoch": 1.657867431640625e-05,
      "model_forward_time": 3.411372661590576,
      "step": 4346
    },
    {
      "epoch": 1.657867431640625e-05,
      "step": 4346,
      "training_step_time": 10.329364776611328
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 3.1565568447113037,
      "step": 4347
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 4347,
      "training_step_time": 9.935950517654419
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 3.137676954269409,
      "step": 4347
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 4347,
      "training_step_time": 9.906563997268677
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 2.4349770545959473,
      "step": 4347
    },
    {
      "dynamics_loss_avg": 0.055992725491523745,
      "epoch": 1.6582489013671877e-05,
      "step": 4347
    },
    {
      "action_loss_avg": 0.023284751223400235,
      "epoch": 1.6582489013671877e-05,
      "step": 4347
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 4347,
      "training_step_time": 7.468445539474487
    },
    {
      "epoch": 1.6582489013671877e-05,
      "model_forward_time": 3.1360299587249756,
      "step": 4347
    },
    {
      "epoch": 1.6582489013671877e-05,
      "step": 4347,
      "training_step_time": 10.317433595657349
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 3.470197916030884,
      "step": 4348
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 4348,
      "training_step_time": 10.249950408935547
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 3.1448044776916504,
      "step": 4348
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 4348,
      "training_step_time": 9.918179512023926
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 3.1435723304748535,
      "step": 4348
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 4348,
      "training_step_time": 9.91672682762146
    },
    {
      "epoch": 1.65863037109375e-05,
      "model_forward_time": 3.142861843109131,
      "step": 4348
    },
    {
      "epoch": 1.65863037109375e-05,
      "step": 4348,
      "training_step_time": 10.011873960494995
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 3.467707633972168,
      "step": 4349
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 4349,
      "training_step_time": 10.238640546798706
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 3.145655870437622,
      "step": 4349
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 4349,
      "training_step_time": 9.915789604187012
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 3.1469643115997314,
      "step": 4349
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 4349,
      "training_step_time": 9.919236898422241
    },
    {
      "epoch": 1.6590118408203126e-05,
      "model_forward_time": 3.143968343734741,
      "step": 4349
    },
    {
      "epoch": 1.6590118408203126e-05,
      "step": 4349,
      "training_step_time": 10.30999207496643
    },
    {
      "epoch": 1.659393310546875e-05,
      "grad_norm": 0.7988442182540894,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 4350
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 3.1790027618408203,
      "step": 4350
    },
    {
      "dynamics_loss_avg": 0.04118429273366928,
      "epoch": 1.659393310546875e-05,
      "step": 4350
    },
    {
      "action_loss_avg": 0.0203226272482425,
      "epoch": 1.659393310546875e-05,
      "step": 4350
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 4350,
      "training_step_time": 9.95555305480957
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 3.138046979904175,
      "step": 4350
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 4350,
      "training_step_time": 10.167083263397217
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 3.136482000350952,
      "step": 4350
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 4350,
      "training_step_time": 9.90611720085144
    },
    {
      "epoch": 1.659393310546875e-05,
      "model_forward_time": 3.131495952606201,
      "step": 4350
    },
    {
      "epoch": 1.659393310546875e-05,
      "step": 4350,
      "training_step_time": 10.27731466293335
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 3.1518099308013916,
      "step": 4351
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 4351,
      "training_step_time": 9.91813349723816
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 3.13466739654541,
      "step": 4351
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 4351,
      "training_step_time": 9.902273178100586
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 2.7086808681488037,
      "step": 4351
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 4351,
      "training_step_time": 7.730363130569458
    },
    {
      "epoch": 1.6597747802734375e-05,
      "model_forward_time": 3.1313555240631104,
      "step": 4351
    },
    {
      "epoch": 1.6597747802734375e-05,
      "step": 4351,
      "training_step_time": 10.295532941818237
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 3.176069974899292,
      "step": 4352
    },
    {
      "epoch": 1.66015625e-05,
      "step": 4352,
      "training_step_time": 9.94663953781128
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 3.1370491981506348,
      "step": 4352
    },
    {
      "epoch": 1.66015625e-05,
      "step": 4352,
      "training_step_time": 9.906500101089478
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 3.135040283203125,
      "step": 4352
    },
    {
      "dynamics_loss_avg": 0.06338031254708767,
      "epoch": 1.66015625e-05,
      "step": 4352
    },
    {
      "action_loss_avg": 0.031151151633821428,
      "epoch": 1.66015625e-05,
      "step": 4352
    },
    {
      "epoch": 1.66015625e-05,
      "step": 4352,
      "training_step_time": 9.905443906784058
    },
    {
      "epoch": 1.66015625e-05,
      "model_forward_time": 3.1393680572509766,
      "step": 4352
    },
    {
      "epoch": 1.66015625e-05,
      "step": 4352,
      "training_step_time": 10.33559274673462
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 3.1740152835845947,
      "step": 4353
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 4353,
      "training_step_time": 9.942476987838745
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 3.1378955841064453,
      "step": 4353
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 4353,
      "training_step_time": 9.908893585205078
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 3.136143445968628,
      "step": 4353
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 4353,
      "training_step_time": 9.909900665283203
    },
    {
      "epoch": 1.6605377197265624e-05,
      "model_forward_time": 3.139674663543701,
      "step": 4353
    },
    {
      "epoch": 1.6605377197265624e-05,
      "step": 4353,
      "training_step_time": 10.021813869476318
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 3.430217981338501,
      "step": 4354
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 4354,
      "training_step_time": 10.212910890579224
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 3.1399412155151367,
      "step": 4354
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 4354,
      "training_step_time": 9.910510301589966
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 3.1387903690338135,
      "step": 4354
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 4354,
      "training_step_time": 9.909753561019897
    },
    {
      "epoch": 1.660919189453125e-05,
      "model_forward_time": 3.139230489730835,
      "step": 4354
    },
    {
      "epoch": 1.660919189453125e-05,
      "step": 4354,
      "training_step_time": 10.304833889007568
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 3.154569387435913,
      "step": 4355
    },
    {
      "dynamics_loss_avg": 0.05996937602758408,
      "epoch": 1.6613006591796877e-05,
      "step": 4355
    },
    {
      "action_loss_avg": 0.024841408361680806,
      "epoch": 1.6613006591796877e-05,
      "step": 4355
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 4355,
      "training_step_time": 9.931599617004395
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 3.415090560913086,
      "step": 4355
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 4355,
      "training_step_time": 10.184693813323975
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 3.138524293899536,
      "step": 4355
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 4355,
      "training_step_time": 9.909380912780762
    },
    {
      "epoch": 1.6613006591796877e-05,
      "model_forward_time": 3.1417343616485596,
      "step": 4355
    },
    {
      "epoch": 1.6613006591796877e-05,
      "step": 4355,
      "training_step_time": 10.306263208389282
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 3.153001308441162,
      "step": 4356
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 4356,
      "training_step_time": 9.926793575286865
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 3.136244297027588,
      "step": 4356
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 4356,
      "training_step_time": 9.907137870788574
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 3.3936965465545654,
      "step": 4356
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 4356,
      "training_step_time": 10.164788007736206
    },
    {
      "epoch": 1.66168212890625e-05,
      "model_forward_time": 3.1423070430755615,
      "step": 4356
    },
    {
      "epoch": 1.66168212890625e-05,
      "step": 4356,
      "training_step_time": 10.018979787826538
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 3.158266305923462,
      "step": 4357
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 4357,
      "training_step_time": 9.932082891464233
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 3.1361782550811768,
      "step": 4357
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 4357,
      "training_step_time": 9.907757997512817
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 3.1423568725585938,
      "step": 4357
    },
    {
      "dynamics_loss_avg": 0.059451911225914957,
      "epoch": 1.6620635986328126e-05,
      "step": 4357
    },
    {
      "action_loss_avg": 0.026283119153231383,
      "epoch": 1.6620635986328126e-05,
      "step": 4357
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 4357,
      "training_step_time": 9.917335987091064
    },
    {
      "epoch": 1.6620635986328126e-05,
      "model_forward_time": 2.698694944381714,
      "step": 4357
    },
    {
      "epoch": 1.6620635986328126e-05,
      "step": 4357,
      "training_step_time": 10.33483338356018
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 3.1456351280212402,
      "step": 4358
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 4358,
      "training_step_time": 9.920708894729614
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 3.1346168518066406,
      "step": 4358
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 4358,
      "training_step_time": 9.905906677246094
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 3.1379549503326416,
      "step": 4358
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 4358,
      "training_step_time": 9.906874895095825
    },
    {
      "epoch": 1.662445068359375e-05,
      "model_forward_time": 3.134896993637085,
      "step": 4358
    },
    {
      "epoch": 1.662445068359375e-05,
      "step": 4358,
      "training_step_time": 10.019551515579224
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 3.4506499767303467,
      "step": 4359
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 4359,
      "training_step_time": 10.2228262424469
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 3.133465528488159,
      "step": 4359
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 4359,
      "training_step_time": 9.900418043136597
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 3.1387710571289062,
      "step": 4359
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 4359,
      "training_step_time": 9.906013250350952
    },
    {
      "epoch": 1.6628265380859375e-05,
      "model_forward_time": 3.1367454528808594,
      "step": 4359
    },
    {
      "epoch": 1.6628265380859375e-05,
      "step": 4359,
      "training_step_time": 10.312567710876465
    },
    {
      "epoch": 1.6632080078125e-05,
      "grad_norm": 0.3469895124435425,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 4360
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 3.1447482109069824,
      "step": 4360
    },
    {
      "dynamics_loss_avg": 0.06255371831357479,
      "epoch": 1.6632080078125e-05,
      "step": 4360
    },
    {
      "action_loss_avg": 0.024126709857955574,
      "epoch": 1.6632080078125e-05,
      "step": 4360
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 4360,
      "training_step_time": 9.915625810623169
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 3.39471173286438,
      "step": 4360
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 4360,
      "training_step_time": 10.16414213180542
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 3.1403675079345703,
      "step": 4360
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 4360,
      "training_step_time": 9.915106534957886
    },
    {
      "epoch": 1.6632080078125e-05,
      "model_forward_time": 3.1383395195007324,
      "step": 4360
    },
    {
      "epoch": 1.6632080078125e-05,
      "step": 4360,
      "training_step_time": 10.28206205368042
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 3.1773738861083984,
      "step": 4361
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 4361,
      "training_step_time": 9.95305347442627
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 3.142522096633911,
      "step": 4361
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 4361,
      "training_step_time": 9.911999940872192
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 3.4388363361358643,
      "step": 4361
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 4361,
      "training_step_time": 10.208094120025635
    },
    {
      "epoch": 1.6635894775390624e-05,
      "model_forward_time": 3.133464813232422,
      "step": 4361
    },
    {
      "epoch": 1.6635894775390624e-05,
      "step": 4361,
      "training_step_time": 10.018255472183228
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 3.14326810836792,
      "step": 4362
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 4362,
      "training_step_time": 9.910060167312622
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 3.1370644569396973,
      "step": 4362
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 4362,
      "training_step_time": 9.902336359024048
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 3.1342873573303223,
      "step": 4362
    },
    {
      "dynamics_loss_avg": 0.06365370526909828,
      "epoch": 1.663970947265625e-05,
      "step": 4362
    },
    {
      "action_loss_avg": 0.03649081643670797,
      "epoch": 1.663970947265625e-05,
      "step": 4362
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 4362,
      "training_step_time": 9.905694723129272
    },
    {
      "epoch": 1.663970947265625e-05,
      "model_forward_time": 3.3789823055267334,
      "step": 4362
    },
    {
      "epoch": 1.663970947265625e-05,
      "step": 4362,
      "training_step_time": 10.287455797195435
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 3.142991542816162,
      "step": 4363
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 4363,
      "training_step_time": 9.917569637298584
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 3.1390466690063477,
      "step": 4363
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 4363,
      "training_step_time": 9.911974906921387
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 3.1383838653564453,
      "step": 4363
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 4363,
      "training_step_time": 9.911468505859375
    },
    {
      "epoch": 1.6643524169921877e-05,
      "model_forward_time": 3.4053592681884766,
      "step": 4363
    },
    {
      "epoch": 1.6643524169921877e-05,
      "step": 4363,
      "training_step_time": 10.273157358169556
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 3.1440675258636475,
      "step": 4364
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 4364,
      "training_step_time": 9.916664838790894
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 3.137932538986206,
      "step": 4364
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 4364,
      "training_step_time": 9.9087233543396
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 2.4327123165130615,
      "step": 4364
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 4364,
      "training_step_time": 7.458022594451904
    },
    {
      "epoch": 1.66473388671875e-05,
      "model_forward_time": 3.1361799240112305,
      "step": 4364
    },
    {
      "epoch": 1.66473388671875e-05,
      "step": 4364,
      "training_step_time": 10.02414083480835
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 3.1515393257141113,
      "step": 4365
    },
    {
      "dynamics_loss_avg": 0.06881561260670424,
      "epoch": 1.6651153564453126e-05,
      "step": 4365
    },
    {
      "action_loss_avg": 0.031183349853381514,
      "epoch": 1.6651153564453126e-05,
      "step": 4365
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 4365,
      "training_step_time": 10.179591655731201
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 3.138195514678955,
      "step": 4365
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 4365,
      "training_step_time": 9.90744662284851
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 3.1358766555786133,
      "step": 4365
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 4365,
      "training_step_time": 9.9052734375
    },
    {
      "epoch": 1.6651153564453126e-05,
      "model_forward_time": 2.432912826538086,
      "step": 4365
    },
    {
      "epoch": 1.6651153564453126e-05,
      "step": 4365,
      "training_step_time": 10.020750045776367
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 3.165252685546875,
      "step": 4366
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 4366,
      "training_step_time": 9.93251919746399
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 3.413233757019043,
      "step": 4366
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 4366,
      "training_step_time": 10.193708896636963
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 3.143880605697632,
      "step": 4366
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 4366,
      "training_step_time": 9.920284509658813
    },
    {
      "epoch": 1.665496826171875e-05,
      "model_forward_time": 3.1393918991088867,
      "step": 4366
    },
    {
      "epoch": 1.665496826171875e-05,
      "step": 4366,
      "training_step_time": 10.299520492553711
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 3.1743993759155273,
      "step": 4367
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 4367,
      "training_step_time": 9.954341173171997
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 3.132976770401001,
      "step": 4367
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 4367,
      "training_step_time": 9.910223722457886
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 3.1402995586395264,
      "step": 4367
    },
    {
      "dynamics_loss_avg": 0.05812352634966374,
      "epoch": 1.6658782958984375e-05,
      "step": 4367
    },
    {
      "action_loss_avg": 0.02110278271138668,
      "epoch": 1.6658782958984375e-05,
      "step": 4367
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 4367,
      "training_step_time": 10.189121723175049
    },
    {
      "epoch": 1.6658782958984375e-05,
      "model_forward_time": 3.148503065109253,
      "step": 4367
    },
    {
      "epoch": 1.6658782958984375e-05,
      "step": 4367,
      "training_step_time": 10.036099672317505
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 3.1525702476501465,
      "step": 4368
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 4368,
      "training_step_time": 9.932093143463135
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 3.1395039558410645,
      "step": 4368
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 4368,
      "training_step_time": 9.915778875350952
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 3.145352602005005,
      "step": 4368
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 4368,
      "training_step_time": 9.923597574234009
    },
    {
      "epoch": 1.666259765625e-05,
      "model_forward_time": 3.4302754402160645,
      "step": 4368
    },
    {
      "epoch": 1.666259765625e-05,
      "step": 4368,
      "training_step_time": 10.300868272781372
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 3.188896894454956,
      "step": 4369
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 4369,
      "training_step_time": 9.969208478927612
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 3.1469759941101074,
      "step": 4369
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 4369,
      "training_step_time": 9.920720338821411
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 3.148433208465576,
      "step": 4369
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 4369,
      "training_step_time": 9.924646854400635
    },
    {
      "epoch": 1.6666412353515624e-05,
      "model_forward_time": 3.145559549331665,
      "step": 4369
    },
    {
      "epoch": 1.6666412353515624e-05,
      "step": 4369,
      "training_step_time": 10.026394605636597
    },
    {
      "epoch": 1.667022705078125e-05,
      "grad_norm": 0.8363675475120544,
      "learning_rate": 1e-05,
      "loss": 0.0948,
      "step": 4370
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 3.422238826751709,
      "step": 4370
    },
    {
      "dynamics_loss_avg": 0.07046664319932461,
      "epoch": 1.667022705078125e-05,
      "step": 4370
    },
    {
      "action_loss_avg": 0.031687617674469946,
      "epoch": 1.667022705078125e-05,
      "step": 4370
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 4370,
      "training_step_time": 10.211503744125366
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 3.137612819671631,
      "step": 4370
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 4370,
      "training_step_time": 9.914823055267334
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 3.1474502086639404,
      "step": 4370
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 4370,
      "training_step_time": 9.929283142089844
    },
    {
      "epoch": 1.667022705078125e-05,
      "model_forward_time": 3.148153305053711,
      "step": 4370
    },
    {
      "epoch": 1.667022705078125e-05,
      "step": 4370,
      "training_step_time": 10.024325370788574
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 3.1913795471191406,
      "step": 4371
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 4371,
      "training_step_time": 9.970560312271118
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 3.402266025543213,
      "step": 4371
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 4371,
      "training_step_time": 10.173584461212158
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 3.1452560424804688,
      "step": 4371
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 4371,
      "training_step_time": 9.914177179336548
    },
    {
      "epoch": 1.6674041748046877e-05,
      "model_forward_time": 3.1396758556365967,
      "step": 4371
    },
    {
      "epoch": 1.6674041748046877e-05,
      "step": 4371,
      "training_step_time": 10.028240203857422
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 3.1476569175720215,
      "step": 4372
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 4372,
      "training_step_time": 9.92223310470581
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 3.146855115890503,
      "step": 4372
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 4372,
      "training_step_time": 9.92105770111084
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 3.421541690826416,
      "step": 4372
    },
    {
      "dynamics_loss_avg": 0.0835512887686491,
      "epoch": 1.66778564453125e-05,
      "step": 4372
    },
    {
      "action_loss_avg": 0.03503658398985863,
      "epoch": 1.66778564453125e-05,
      "step": 4372
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 4372,
      "training_step_time": 10.197141647338867
    },
    {
      "epoch": 1.66778564453125e-05,
      "model_forward_time": 3.1438980102539062,
      "step": 4372
    },
    {
      "epoch": 1.66778564453125e-05,
      "step": 4372,
      "training_step_time": 10.313254117965698
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 3.158324718475342,
      "step": 4373
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 4373,
      "training_step_time": 9.933918952941895
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 3.144235610961914,
      "step": 4373
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 4373,
      "training_step_time": 9.91999363899231
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 3.1455726623535156,
      "step": 4373
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 4373,
      "training_step_time": 9.922548055648804
    },
    {
      "epoch": 1.6681671142578126e-05,
      "model_forward_time": 3.148393392562866,
      "step": 4373
    },
    {
      "epoch": 1.6681671142578126e-05,
      "step": 4373,
      "training_step_time": 10.301734685897827
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 3.1656832695007324,
      "step": 4374
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 4374,
      "training_step_time": 9.945107221603394
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 3.147256374359131,
      "step": 4374
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 4374,
      "training_step_time": 9.925513982772827
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 3.1427764892578125,
      "step": 4374
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 4374,
      "training_step_time": 9.919785261154175
    },
    {
      "epoch": 1.668548583984375e-05,
      "model_forward_time": 3.148181200027466,
      "step": 4374
    },
    {
      "epoch": 1.668548583984375e-05,
      "step": 4374,
      "training_step_time": 10.301835775375366
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 3.4792633056640625,
      "step": 4375
    },
    {
      "dynamics_loss_avg": 0.06064318455755711,
      "epoch": 1.6689300537109375e-05,
      "step": 4375
    },
    {
      "action_loss_avg": 0.02778209401294589,
      "epoch": 1.6689300537109375e-05,
      "step": 4375
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 4375,
      "training_step_time": 10.260062456130981
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 3.1446352005004883,
      "step": 4375
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 4375,
      "training_step_time": 9.918521642684937
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 3.1389474868774414,
      "step": 4375
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 4375,
      "training_step_time": 9.910325050354004
    },
    {
      "epoch": 1.6689300537109375e-05,
      "model_forward_time": 3.1408936977386475,
      "step": 4375
    },
    {
      "epoch": 1.6689300537109375e-05,
      "step": 4375,
      "training_step_time": 10.016915082931519
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 3.1591548919677734,
      "step": 4376
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 4376,
      "training_step_time": 9.932365655899048
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 3.1362383365631104,
      "step": 4376
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 4376,
      "training_step_time": 10.190169334411621
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 3.1445393562316895,
      "step": 4376
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 4376,
      "training_step_time": 9.922817945480347
    },
    {
      "epoch": 1.6693115234375e-05,
      "model_forward_time": 3.147948980331421,
      "step": 4376
    },
    {
      "epoch": 1.6693115234375e-05,
      "step": 4376,
      "training_step_time": 10.269280910491943
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 3.1598117351531982,
      "step": 4377
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 4377,
      "training_step_time": 9.932554244995117
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 3.14353084564209,
      "step": 4377
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 4377,
      "training_step_time": 9.915877103805542
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 3.408118486404419,
      "step": 4377
    },
    {
      "dynamics_loss_avg": 0.07135090827941895,
      "epoch": 1.6696929931640624e-05,
      "step": 4377
    },
    {
      "action_loss_avg": 0.03461136817932129,
      "epoch": 1.6696929931640624e-05,
      "step": 4377
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 4377,
      "training_step_time": 10.183651685714722
    },
    {
      "epoch": 1.6696929931640624e-05,
      "model_forward_time": 3.1488351821899414,
      "step": 4377
    },
    {
      "epoch": 1.6696929931640624e-05,
      "step": 4377,
      "training_step_time": 10.260412216186523
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 3.1536123752593994,
      "step": 4378
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 4378,
      "training_step_time": 9.928167581558228
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 3.1340034008026123,
      "step": 4378
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 4378,
      "training_step_time": 9.908499240875244
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 3.135241746902466,
      "step": 4378
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 4378,
      "training_step_time": 9.910794973373413
    },
    {
      "epoch": 1.670074462890625e-05,
      "model_forward_time": 3.1311473846435547,
      "step": 4378
    },
    {
      "epoch": 1.670074462890625e-05,
      "step": 4378,
      "training_step_time": 10.277271270751953
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 3.1806628704071045,
      "step": 4379
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 4379,
      "training_step_time": 9.954431056976318
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 3.1309375762939453,
      "step": 4379
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 4379,
      "training_step_time": 9.906481266021729
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 3.1303017139434814,
      "step": 4379
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 4379,
      "training_step_time": 9.904591083526611
    },
    {
      "epoch": 1.6704559326171873e-05,
      "model_forward_time": 3.1314196586608887,
      "step": 4379
    },
    {
      "epoch": 1.6704559326171873e-05,
      "step": 4379,
      "training_step_time": 10.28410291671753
    },
    {
      "epoch": 1.67083740234375e-05,
      "grad_norm": 0.3597458302974701,
      "learning_rate": 1e-05,
      "loss": 0.0937,
      "step": 4380
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 3.417374610900879,
      "step": 4380
    },
    {
      "dynamics_loss_avg": 0.06630511209368706,
      "epoch": 1.67083740234375e-05,
      "step": 4380
    },
    {
      "action_loss_avg": 0.026279667019844057,
      "epoch": 1.67083740234375e-05,
      "step": 4380
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 4380,
      "training_step_time": 10.199155330657959
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 3.132279396057129,
      "step": 4380
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 4380,
      "training_step_time": 9.903059482574463
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 3.1396639347076416,
      "step": 4380
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 4380,
      "training_step_time": 9.912248849868774
    },
    {
      "epoch": 1.67083740234375e-05,
      "model_forward_time": 3.133014440536499,
      "step": 4380
    },
    {
      "epoch": 1.67083740234375e-05,
      "step": 4380,
      "training_step_time": 10.279393196105957
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 3.143329620361328,
      "step": 4381
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 4381,
      "training_step_time": 9.91246747970581
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 3.3876748085021973,
      "step": 4381
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 4381,
      "training_step_time": 10.158971548080444
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 2.432451009750366,
      "step": 4381
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 4381,
      "training_step_time": 7.46097731590271
    },
    {
      "epoch": 1.6712188720703126e-05,
      "model_forward_time": 3.1299126148223877,
      "step": 4381
    },
    {
      "epoch": 1.6712188720703126e-05,
      "step": 4381,
      "training_step_time": 10.286695003509521
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 3.1780190467834473,
      "step": 4382
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 4382,
      "training_step_time": 9.951264142990112
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 3.13822603225708,
      "step": 4382
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 4382,
      "training_step_time": 9.910761833190918
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 3.4196016788482666,
      "step": 4382
    },
    {
      "dynamics_loss_avg": 0.06664296593517065,
      "epoch": 1.671600341796875e-05,
      "step": 4382
    },
    {
      "action_loss_avg": 0.03267891029827297,
      "epoch": 1.671600341796875e-05,
      "step": 4382
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 4382,
      "training_step_time": 10.201089859008789
    },
    {
      "epoch": 1.671600341796875e-05,
      "model_forward_time": 3.1367316246032715,
      "step": 4382
    },
    {
      "epoch": 1.671600341796875e-05,
      "step": 4382,
      "training_step_time": 10.263715267181396
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 3.1875736713409424,
      "step": 4383
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 4383,
      "training_step_time": 9.968337059020996
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 3.139080762863159,
      "step": 4383
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 4383,
      "training_step_time": 9.919629573822021
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 3.141979694366455,
      "step": 4383
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 4383,
      "training_step_time": 9.924417734146118
    },
    {
      "epoch": 1.6719818115234375e-05,
      "model_forward_time": 3.4140472412109375,
      "step": 4383
    },
    {
      "epoch": 1.6719818115234375e-05,
      "step": 4383,
      "training_step_time": 10.320940017700195
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 3.1695687770843506,
      "step": 4384
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 4384,
      "training_step_time": 9.948254823684692
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 2.4372434616088867,
      "step": 4384
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 4384,
      "training_step_time": 7.470727205276489
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 3.131113290786743,
      "step": 4384
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 4384,
      "training_step_time": 9.903263568878174
    },
    {
      "epoch": 1.67236328125e-05,
      "model_forward_time": 3.1387274265289307,
      "step": 4384
    },
    {
      "epoch": 1.67236328125e-05,
      "step": 4384,
      "training_step_time": 10.007024049758911
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 3.1537930965423584,
      "step": 4385
    },
    {
      "dynamics_loss_avg": 0.06648855283856392,
      "epoch": 1.6727447509765624e-05,
      "step": 4385
    },
    {
      "action_loss_avg": 0.03976714052259922,
      "epoch": 1.6727447509765624e-05,
      "step": 4385
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 4385,
      "training_step_time": 10.231723308563232
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 3.1358063220977783,
      "step": 4385
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 4385,
      "training_step_time": 9.913935899734497
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 2.432865619659424,
      "step": 4385
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 4385,
      "training_step_time": 7.460801839828491
    },
    {
      "epoch": 1.6727447509765624e-05,
      "model_forward_time": 3.1360554695129395,
      "step": 4385
    },
    {
      "epoch": 1.6727447509765624e-05,
      "step": 4385,
      "training_step_time": 10.26919436454773
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 3.166872501373291,
      "step": 4386
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 4386,
      "training_step_time": 9.936965465545654
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 3.132568120956421,
      "step": 4386
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 4386,
      "training_step_time": 10.168250560760498
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 3.133758306503296,
      "step": 4386
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 4386,
      "training_step_time": 9.905970335006714
    },
    {
      "epoch": 1.673126220703125e-05,
      "model_forward_time": 3.1391470432281494,
      "step": 4386
    },
    {
      "epoch": 1.673126220703125e-05,
      "step": 4386,
      "training_step_time": 10.268443822860718
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 3.1752536296844482,
      "step": 4387
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 4387,
      "training_step_time": 9.950756072998047
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 3.1342098712921143,
      "step": 4387
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 4387,
      "training_step_time": 9.914506673812866
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 3.135010242462158,
      "step": 4387
    },
    {
      "dynamics_loss_avg": 0.05980559196323156,
      "epoch": 1.6735076904296873e-05,
      "step": 4387
    },
    {
      "action_loss_avg": 0.030103056831285356,
      "epoch": 1.6735076904296873e-05,
      "step": 4387
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 4387,
      "training_step_time": 10.181772947311401
    },
    {
      "epoch": 1.6735076904296873e-05,
      "model_forward_time": 3.141247510910034,
      "step": 4387
    },
    {
      "epoch": 1.6735076904296873e-05,
      "step": 4387,
      "training_step_time": 10.030043601989746
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 3.1430771350860596,
      "step": 4388
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 4388,
      "training_step_time": 9.919928073883057
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 3.135378360748291,
      "step": 4388
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 4388,
      "training_step_time": 9.913823127746582
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 3.1374378204345703,
      "step": 4388
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 4388,
      "training_step_time": 9.912620544433594
    },
    {
      "epoch": 1.67388916015625e-05,
      "model_forward_time": 3.4254324436187744,
      "step": 4388
    },
    {
      "epoch": 1.67388916015625e-05,
      "step": 4388,
      "training_step_time": 10.300968885421753
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 3.18249773979187,
      "step": 4389
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 4389,
      "training_step_time": 9.964892148971558
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 3.137801170349121,
      "step": 4389
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 4389,
      "training_step_time": 9.918938875198364
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 3.1412675380706787,
      "step": 4389
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 4389,
      "training_step_time": 9.918668031692505
    },
    {
      "epoch": 1.6742706298828126e-05,
      "model_forward_time": 3.139537811279297,
      "step": 4389
    },
    {
      "epoch": 1.6742706298828126e-05,
      "step": 4389,
      "training_step_time": 10.313340663909912
    },
    {
      "epoch": 1.674652099609375e-05,
      "grad_norm": 0.4784305691719055,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 4390
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 3.451361656188965,
      "step": 4390
    },
    {
      "dynamics_loss_avg": 0.06234769206494093,
      "epoch": 1.674652099609375e-05,
      "step": 4390
    },
    {
      "action_loss_avg": 0.026990982051938773,
      "epoch": 1.674652099609375e-05,
      "step": 4390
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 4390,
      "training_step_time": 10.235659122467041
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 3.1335790157318115,
      "step": 4390
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 4390,
      "training_step_time": 9.911871671676636
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 3.139049768447876,
      "step": 4390
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 4390,
      "training_step_time": 9.914644956588745
    },
    {
      "epoch": 1.674652099609375e-05,
      "model_forward_time": 3.1322975158691406,
      "step": 4390
    },
    {
      "epoch": 1.674652099609375e-05,
      "step": 4390,
      "training_step_time": 10.263620376586914
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 3.1607472896575928,
      "step": 4391
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 4391,
      "training_step_time": 9.933738946914673
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 3.4033315181732178,
      "step": 4391
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 4391,
      "training_step_time": 10.176571369171143
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 3.1408865451812744,
      "step": 4391
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 4391,
      "training_step_time": 9.920996189117432
    },
    {
      "epoch": 1.6750335693359375e-05,
      "model_forward_time": 3.145752429962158,
      "step": 4391
    },
    {
      "epoch": 1.6750335693359375e-05,
      "step": 4391,
      "training_step_time": 10.294442892074585
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 3.148850202560425,
      "step": 4392
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 4392,
      "training_step_time": 9.928000688552856
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 3.142791986465454,
      "step": 4392
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 4392,
      "training_step_time": 9.916843891143799
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 3.432589292526245,
      "step": 4392
    },
    {
      "dynamics_loss_avg": 0.06625679302960634,
      "epoch": 1.6754150390625e-05,
      "step": 4392
    },
    {
      "action_loss_avg": 0.024601356219500305,
      "epoch": 1.6754150390625e-05,
      "step": 4392
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 4392,
      "training_step_time": 10.206120252609253
    },
    {
      "epoch": 1.6754150390625e-05,
      "model_forward_time": 3.13952898979187,
      "step": 4392
    },
    {
      "epoch": 1.6754150390625e-05,
      "step": 4392,
      "training_step_time": 10.008601188659668
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 3.1652607917785645,
      "step": 4393
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 4393,
      "training_step_time": 9.942542791366577
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 3.1427090167999268,
      "step": 4393
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 4393,
      "training_step_time": 9.92507028579712
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 3.1404545307159424,
      "step": 4393
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 4393,
      "training_step_time": 9.919110298156738
    },
    {
      "epoch": 1.6757965087890624e-05,
      "model_forward_time": 3.1459848880767822,
      "step": 4393
    },
    {
      "epoch": 1.6757965087890624e-05,
      "step": 4393,
      "training_step_time": 10.300571918487549
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 3.18096923828125,
      "step": 4394
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 4394,
      "training_step_time": 9.954082727432251
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 3.1458256244659424,
      "step": 4394
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 4394,
      "training_step_time": 9.916892051696777
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 3.1496477127075195,
      "step": 4394
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 4394,
      "training_step_time": 9.924591541290283
    },
    {
      "epoch": 1.676177978515625e-05,
      "model_forward_time": 3.1486129760742188,
      "step": 4394
    },
    {
      "epoch": 1.676177978515625e-05,
      "step": 4394,
      "training_step_time": 10.330308437347412
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 3.4381463527679443,
      "step": 4395
    },
    {
      "dynamics_loss_avg": 0.06526581160724163,
      "epoch": 1.6765594482421873e-05,
      "step": 4395
    },
    {
      "action_loss_avg": 0.0276583488099277,
      "epoch": 1.6765594482421873e-05,
      "step": 4395
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 4395,
      "training_step_time": 10.216597318649292
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 3.145846128463745,
      "step": 4395
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 4395,
      "training_step_time": 9.917924165725708
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 3.144425392150879,
      "step": 4395
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 4395,
      "training_step_time": 9.915433883666992
    },
    {
      "epoch": 1.6765594482421873e-05,
      "model_forward_time": 3.1447067260742188,
      "step": 4395
    },
    {
      "epoch": 1.6765594482421873e-05,
      "step": 4395,
      "training_step_time": 10.28601861000061
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 3.1758527755737305,
      "step": 4396
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 4396,
      "training_step_time": 9.944395542144775
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 3.137939214706421,
      "step": 4396
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 4396,
      "training_step_time": 10.178494930267334
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 2.4362759590148926,
      "step": 4396
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 4396,
      "training_step_time": 7.460398197174072
    },
    {
      "epoch": 1.67694091796875e-05,
      "model_forward_time": 3.1382806301116943,
      "step": 4396
    },
    {
      "epoch": 1.67694091796875e-05,
      "step": 4396,
      "training_step_time": 10.291582345962524
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 3.1539928913116455,
      "step": 4397
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 4397,
      "training_step_time": 9.941879034042358
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 3.1578516960144043,
      "step": 4397
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 4397,
      "training_step_time": 9.929422855377197
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 3.4387214183807373,
      "step": 4397
    },
    {
      "dynamics_loss_avg": 0.05183925554156303,
      "epoch": 1.6773223876953126e-05,
      "step": 4397
    },
    {
      "action_loss_avg": 0.023414271138608455,
      "epoch": 1.6773223876953126e-05,
      "step": 4397
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 4397,
      "training_step_time": 10.20514988899231
    },
    {
      "epoch": 1.6773223876953126e-05,
      "model_forward_time": 3.1396777629852295,
      "step": 4397
    },
    {
      "epoch": 1.6773223876953126e-05,
      "step": 4397,
      "training_step_time": 10.001610279083252
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 3.1747629642486572,
      "step": 4398
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 4398,
      "training_step_time": 9.93899130821228
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 3.1389057636260986,
      "step": 4398
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 4398,
      "training_step_time": 9.904922246932983
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 3.1384196281433105,
      "step": 4398
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 4398,
      "training_step_time": 9.905118227005005
    },
    {
      "epoch": 1.677703857421875e-05,
      "model_forward_time": 3.1387226581573486,
      "step": 4398
    },
    {
      "epoch": 1.677703857421875e-05,
      "step": 4398,
      "training_step_time": 10.27446699142456
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 3.1546077728271484,
      "step": 4399
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 4399,
      "training_step_time": 9.927977085113525
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 3.137579917907715,
      "step": 4399
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 4399,
      "training_step_time": 9.90882396697998
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 3.141624689102173,
      "step": 4399
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 4399,
      "training_step_time": 9.912897109985352
    },
    {
      "epoch": 1.6780853271484375e-05,
      "model_forward_time": 3.140216588973999,
      "step": 4399
    },
    {
      "epoch": 1.6780853271484375e-05,
      "step": 4399,
      "training_step_time": 10.027409553527832
    },
    {
      "epoch": 1.678466796875e-05,
      "grad_norm": 0.26306822896003723,
      "learning_rate": 1e-05,
      "loss": 0.0848,
      "step": 4400
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 3.505030632019043,
      "step": 4400
    },
    {
      "dynamics_loss_avg": 0.053976740688085556,
      "epoch": 1.678466796875e-05,
      "step": 4400
    },
    {
      "action_loss_avg": 0.024012432992458345,
      "epoch": 1.678466796875e-05,
      "step": 4400
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 4400,
      "training_step_time": 10.290756940841675
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 3.1432037353515625,
      "step": 4400
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 4400,
      "training_step_time": 9.92198920249939
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 3.1451327800750732,
      "step": 4400
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 4400,
      "training_step_time": 9.922693729400635
    },
    {
      "epoch": 1.678466796875e-05,
      "model_forward_time": 3.149275541305542,
      "step": 4400
    },
    {
      "epoch": 1.678466796875e-05,
      "step": 4400,
      "training_step_time": 10.32670783996582
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 3.1860694885253906,
      "step": 4401
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 4401,
      "training_step_time": 9.972844123840332
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 3.394218683242798,
      "step": 4401
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 4401,
      "training_step_time": 10.169390678405762
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 3.1421546936035156,
      "step": 4401
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 4401,
      "training_step_time": 9.917747974395752
    },
    {
      "epoch": 1.6788482666015624e-05,
      "model_forward_time": 3.140094757080078,
      "step": 4401
    },
    {
      "epoch": 1.6788482666015624e-05,
      "step": 4401,
      "training_step_time": 10.322406530380249
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 3.174896478652954,
      "step": 4402
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 4402,
      "training_step_time": 9.947153806686401
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 3.1426033973693848,
      "step": 4402
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 4402,
      "training_step_time": 9.9122953414917
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 3.421098470687866,
      "step": 4402
    },
    {
      "dynamics_loss_avg": 0.06275154203176499,
      "epoch": 1.679229736328125e-05,
      "step": 4402
    },
    {
      "action_loss_avg": 0.028048592107370495,
      "epoch": 1.679229736328125e-05,
      "step": 4402
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 4402,
      "training_step_time": 10.1975736618042
    },
    {
      "epoch": 1.679229736328125e-05,
      "model_forward_time": 3.1443796157836914,
      "step": 4402
    },
    {
      "epoch": 1.679229736328125e-05,
      "step": 4402,
      "training_step_time": 10.327776670455933
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 3.1794211864471436,
      "step": 4403
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 4403,
      "training_step_time": 9.950645685195923
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 3.1362922191619873,
      "step": 4403
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 4403,
      "training_step_time": 9.903614044189453
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 3.132612943649292,
      "step": 4403
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 4403,
      "training_step_time": 9.902435064315796
    },
    {
      "epoch": 1.6796112060546873e-05,
      "model_forward_time": 3.4097423553466797,
      "step": 4403
    },
    {
      "epoch": 1.6796112060546873e-05,
      "step": 4403,
      "training_step_time": 10.306704044342041
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 3.142439842224121,
      "step": 4404
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 4404,
      "training_step_time": 9.918671607971191
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 3.1357057094573975,
      "step": 4404
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 4404,
      "training_step_time": 9.90819239616394
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 3.138157844543457,
      "step": 4404
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 4404,
      "training_step_time": 9.910730123519897
    },
    {
      "epoch": 1.67999267578125e-05,
      "model_forward_time": 3.140352249145508,
      "step": 4404
    },
    {
      "epoch": 1.67999267578125e-05,
      "step": 4404,
      "training_step_time": 10.279417037963867
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 3.434598207473755,
      "step": 4405
    },
    {
      "dynamics_loss_avg": 0.06416717283427716,
      "epoch": 1.6803741455078126e-05,
      "step": 4405
    },
    {
      "action_loss_avg": 0.02868927698582411,
      "epoch": 1.6803741455078126e-05,
      "step": 4405
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 4405,
      "training_step_time": 10.214235544204712
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 3.140658140182495,
      "step": 4405
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 4405,
      "training_step_time": 9.91511082649231
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 3.138571262359619,
      "step": 4405
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 4405,
      "training_step_time": 9.915722608566284
    },
    {
      "epoch": 1.6803741455078126e-05,
      "model_forward_time": 3.138850450515747,
      "step": 4405
    },
    {
      "epoch": 1.6803741455078126e-05,
      "step": 4405,
      "training_step_time": 10.311658143997192
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 3.153332233428955,
      "step": 4406
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 4406,
      "training_step_time": 9.924666166305542
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 3.4136111736297607,
      "step": 4406
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 4406,
      "training_step_time": 10.181098699569702
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 2.4307656288146973,
      "step": 4406
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 4406,
      "training_step_time": 7.453459978103638
    },
    {
      "epoch": 1.680755615234375e-05,
      "model_forward_time": 3.128647804260254,
      "step": 4406
    },
    {
      "epoch": 1.680755615234375e-05,
      "step": 4406,
      "training_step_time": 10.254650115966797
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 3.157679319381714,
      "step": 4407
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 4407,
      "training_step_time": 9.925301790237427
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 3.1346328258514404,
      "step": 4407
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 4407,
      "training_step_time": 9.900226831436157
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 3.401982545852661,
      "step": 4407
    },
    {
      "dynamics_loss_avg": 0.07059559300541877,
      "epoch": 1.6811370849609375e-05,
      "step": 4407
    },
    {
      "action_loss_avg": 0.03435964621603489,
      "epoch": 1.6811370849609375e-05,
      "step": 4407
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 4407,
      "training_step_time": 10.171137571334839
    },
    {
      "epoch": 1.6811370849609375e-05,
      "model_forward_time": 3.137319564819336,
      "step": 4407
    },
    {
      "epoch": 1.6811370849609375e-05,
      "step": 4407,
      "training_step_time": 10.269896507263184
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 3.149746894836426,
      "step": 4408
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 4408,
      "training_step_time": 9.919707536697388
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 3.134917974472046,
      "step": 4408
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 4408,
      "training_step_time": 9.906068086624146
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 3.1396477222442627,
      "step": 4408
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 4408,
      "training_step_time": 9.906500339508057
    },
    {
      "epoch": 1.6815185546875e-05,
      "model_forward_time": 3.421989679336548,
      "step": 4408
    },
    {
      "epoch": 1.6815185546875e-05,
      "step": 4408,
      "training_step_time": 10.284737586975098
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 3.166274309158325,
      "step": 4409
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 4409,
      "training_step_time": 9.94032597541809
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 2.429949998855591,
      "step": 4409
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 4409,
      "training_step_time": 7.457685232162476
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 3.137951135635376,
      "step": 4409
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 4409,
      "training_step_time": 9.910466432571411
    },
    {
      "epoch": 1.6819000244140624e-05,
      "model_forward_time": 3.1376571655273438,
      "step": 4409
    },
    {
      "epoch": 1.6819000244140624e-05,
      "step": 4409,
      "training_step_time": 10.0197172164917
    },
    {
      "epoch": 1.682281494140625e-05,
      "grad_norm": 0.49191978573799133,
      "learning_rate": 1e-05,
      "loss": 0.0895,
      "step": 4410
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 3.1712894439697266,
      "step": 4410
    },
    {
      "dynamics_loss_avg": 0.058461911231279376,
      "epoch": 1.682281494140625e-05,
      "step": 4410
    },
    {
      "action_loss_avg": 0.02236896250396967,
      "epoch": 1.682281494140625e-05,
      "step": 4410
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 4410,
      "training_step_time": 10.248004913330078
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 3.139317035675049,
      "step": 4410
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 4410,
      "training_step_time": 9.906594276428223
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 3.1435248851776123,
      "step": 4410
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 4410,
      "training_step_time": 9.910381317138672
    },
    {
      "epoch": 1.682281494140625e-05,
      "model_forward_time": 3.1355440616607666,
      "step": 4410
    },
    {
      "epoch": 1.682281494140625e-05,
      "step": 4410,
      "training_step_time": 10.282024621963501
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 2.4513182640075684,
      "step": 4411
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 4411,
      "training_step_time": 7.475677728652954
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 3.4046318531036377,
      "step": 4411
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 4411,
      "training_step_time": 10.17090630531311
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 3.1397600173950195,
      "step": 4411
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 4411,
      "training_step_time": 9.906868696212769
    },
    {
      "epoch": 1.6826629638671873e-05,
      "model_forward_time": 3.134096384048462,
      "step": 4411
    },
    {
      "epoch": 1.6826629638671873e-05,
      "step": 4411,
      "training_step_time": 10.331870794296265
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 3.170438528060913,
      "step": 4412
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 4412,
      "training_step_time": 9.941466808319092
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 3.1414310932159424,
      "step": 4412
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 4412,
      "training_step_time": 9.910603284835815
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 3.142463207244873,
      "step": 4412
    },
    {
      "dynamics_loss_avg": 0.06612288672477007,
      "epoch": 1.68304443359375e-05,
      "step": 4412
    },
    {
      "action_loss_avg": 0.023614677134901285,
      "epoch": 1.68304443359375e-05,
      "step": 4412
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 4412,
      "training_step_time": 10.187819004058838
    },
    {
      "epoch": 1.68304443359375e-05,
      "model_forward_time": 3.1431121826171875,
      "step": 4412
    },
    {
      "epoch": 1.68304443359375e-05,
      "step": 4412,
      "training_step_time": 10.319478034973145
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 3.1510508060455322,
      "step": 4413
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 4413,
      "training_step_time": 9.926185369491577
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 3.1406607627868652,
      "step": 4413
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 4413,
      "training_step_time": 9.911937713623047
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 3.1399643421173096,
      "step": 4413
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 4413,
      "training_step_time": 9.913158893585205
    },
    {
      "epoch": 1.6834259033203126e-05,
      "model_forward_time": 2.437957525253296,
      "step": 4413
    },
    {
      "epoch": 1.6834259033203126e-05,
      "step": 4413,
      "training_step_time": 10.012243270874023
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 3.1442058086395264,
      "step": 4414
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 4414,
      "training_step_time": 9.914325952529907
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 3.1399881839752197,
      "step": 4414
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 4414,
      "training_step_time": 9.908255815505981
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 3.1306610107421875,
      "step": 4414
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 4414,
      "training_step_time": 9.89843487739563
    },
    {
      "epoch": 1.683807373046875e-05,
      "model_forward_time": 3.1390538215637207,
      "step": 4414
    },
    {
      "epoch": 1.683807373046875e-05,
      "step": 4414,
      "training_step_time": 10.003223896026611
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 2.445908546447754,
      "step": 4415
    },
    {
      "dynamics_loss_avg": 0.05740760341286659,
      "epoch": 1.6841888427734375e-05,
      "step": 4415
    },
    {
      "action_loss_avg": 0.025199738191440703,
      "epoch": 1.6841888427734375e-05,
      "step": 4415
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 4415,
      "training_step_time": 7.740508317947388
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 3.1320905685424805,
      "step": 4415
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 4415,
      "training_step_time": 9.906683206558228
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 3.1328675746917725,
      "step": 4415
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 4415,
      "training_step_time": 9.902865648269653
    },
    {
      "epoch": 1.6841888427734375e-05,
      "model_forward_time": 3.1349294185638428,
      "step": 4415
    },
    {
      "epoch": 1.6841888427734375e-05,
      "step": 4415,
      "training_step_time": 10.002615690231323
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 3.1437265872955322,
      "step": 4416
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 4416,
      "training_step_time": 9.925020933151245
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 3.134958028793335,
      "step": 4416
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 4416,
      "training_step_time": 10.187272071838379
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 3.133833169937134,
      "step": 4416
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 4416,
      "training_step_time": 9.909437656402588
    },
    {
      "epoch": 1.6845703125e-05,
      "model_forward_time": 3.135247230529785,
      "step": 4416
    },
    {
      "epoch": 1.6845703125e-05,
      "step": 4416,
      "training_step_time": 10.28420615196228
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 3.1710097789764404,
      "step": 4417
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 4417,
      "training_step_time": 9.946261405944824
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 3.1332509517669678,
      "step": 4417
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 4417,
      "training_step_time": 9.913721323013306
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 3.3991506099700928,
      "step": 4417
    },
    {
      "dynamics_loss_avg": 0.07761558406054973,
      "epoch": 1.6849517822265624e-05,
      "step": 4417
    },
    {
      "action_loss_avg": 0.03579721283167601,
      "epoch": 1.6849517822265624e-05,
      "step": 4417
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 4417,
      "training_step_time": 10.17030143737793
    },
    {
      "epoch": 1.6849517822265624e-05,
      "model_forward_time": 3.138387680053711,
      "step": 4417
    },
    {
      "epoch": 1.6849517822265624e-05,
      "step": 4417,
      "training_step_time": 10.277647733688354
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 3.174772262573242,
      "step": 4418
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 4418,
      "training_step_time": 9.953538417816162
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 3.1374218463897705,
      "step": 4418
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 4418,
      "training_step_time": 9.910297155380249
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 3.1430273056030273,
      "step": 4418
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 4418,
      "training_step_time": 9.915174961090088
    },
    {
      "epoch": 1.685333251953125e-05,
      "model_forward_time": 3.1350343227386475,
      "step": 4418
    },
    {
      "epoch": 1.685333251953125e-05,
      "step": 4418,
      "training_step_time": 10.29548692703247
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 3.1836299896240234,
      "step": 4419
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 4419,
      "training_step_time": 9.95741319656372
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 3.1319921016693115,
      "step": 4419
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 4419,
      "training_step_time": 9.900383710861206
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 3.135546922683716,
      "step": 4419
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 4419,
      "training_step_time": 9.908262491226196
    },
    {
      "epoch": 1.6857147216796873e-05,
      "model_forward_time": 3.1334574222564697,
      "step": 4419
    },
    {
      "epoch": 1.6857147216796873e-05,
      "step": 4419,
      "training_step_time": 10.271571636199951
    },
    {
      "epoch": 1.68609619140625e-05,
      "grad_norm": 0.3792085647583008,
      "learning_rate": 1e-05,
      "loss": 0.0886,
      "step": 4420
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 3.4304492473602295,
      "step": 4420
    },
    {
      "dynamics_loss_avg": 0.07027769312262536,
      "epoch": 1.68609619140625e-05,
      "step": 4420
    },
    {
      "action_loss_avg": 0.032129863556474444,
      "epoch": 1.68609619140625e-05,
      "step": 4420
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 4420,
      "training_step_time": 10.205763339996338
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 3.139371633529663,
      "step": 4420
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 4420,
      "training_step_time": 9.914407968521118
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 3.1368963718414307,
      "step": 4420
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 4420,
      "training_step_time": 9.909264326095581
    },
    {
      "epoch": 1.68609619140625e-05,
      "model_forward_time": 3.1309731006622314,
      "step": 4420
    },
    {
      "epoch": 1.68609619140625e-05,
      "step": 4420,
      "training_step_time": 10.003095388412476
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 3.1729907989501953,
      "step": 4421
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 4421,
      "training_step_time": 9.950345993041992
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 3.402385950088501,
      "step": 4421
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 4421,
      "training_step_time": 10.174449443817139
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 3.1429800987243652,
      "step": 4421
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 4421,
      "training_step_time": 9.921484231948853
    },
    {
      "epoch": 1.6864776611328126e-05,
      "model_forward_time": 3.1371891498565674,
      "step": 4421
    },
    {
      "epoch": 1.6864776611328126e-05,
      "step": 4421,
      "training_step_time": 10.254175662994385
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 3.1560513973236084,
      "step": 4422
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 4422,
      "training_step_time": 9.934265613555908
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 3.140702724456787,
      "step": 4422
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 4422,
      "training_step_time": 9.916844844818115
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 3.4128811359405518,
      "step": 4422
    },
    {
      "dynamics_loss_avg": 0.074650352075696,
      "epoch": 1.686859130859375e-05,
      "step": 4422
    },
    {
      "action_loss_avg": 0.02566006323322654,
      "epoch": 1.686859130859375e-05,
      "step": 4422
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 4422,
      "training_step_time": 10.1897714138031
    },
    {
      "epoch": 1.686859130859375e-05,
      "model_forward_time": 3.134026288986206,
      "step": 4422
    },
    {
      "epoch": 1.686859130859375e-05,
      "step": 4422,
      "training_step_time": 10.28933572769165
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 3.168292999267578,
      "step": 4423
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 4423,
      "training_step_time": 9.939749479293823
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 3.1386454105377197,
      "step": 4423
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 4423,
      "training_step_time": 9.909255504608154
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 3.134930372238159,
      "step": 4423
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 4423,
      "training_step_time": 9.910241842269897
    },
    {
      "epoch": 1.6872406005859375e-05,
      "model_forward_time": 3.390744924545288,
      "step": 4423
    },
    {
      "epoch": 1.6872406005859375e-05,
      "step": 4423,
      "training_step_time": 10.309639692306519
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 3.1664886474609375,
      "step": 4424
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 4424,
      "training_step_time": 9.951847791671753
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 3.146742343902588,
      "step": 4424
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 4424,
      "training_step_time": 9.923166275024414
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 2.43713641166687,
      "step": 4424
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 4424,
      "training_step_time": 7.4710164070129395
    },
    {
      "epoch": 1.6876220703125e-05,
      "model_forward_time": 3.145750045776367,
      "step": 4424
    },
    {
      "epoch": 1.6876220703125e-05,
      "step": 4424,
      "training_step_time": 10.302185535430908
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 3.4164178371429443,
      "step": 4425
    },
    {
      "dynamics_loss_avg": 0.05459311231970787,
      "epoch": 1.6880035400390624e-05,
      "step": 4425
    },
    {
      "action_loss_avg": 0.02307186513207853,
      "epoch": 1.6880035400390624e-05,
      "step": 4425
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 4425,
      "training_step_time": 10.198947429656982
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 3.145766496658325,
      "step": 4425
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 4425,
      "training_step_time": 9.919472694396973
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 3.144315004348755,
      "step": 4425
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 4425,
      "training_step_time": 9.916414737701416
    },
    {
      "epoch": 1.6880035400390624e-05,
      "model_forward_time": 3.1459555625915527,
      "step": 4425
    },
    {
      "epoch": 1.6880035400390624e-05,
      "step": 4425,
      "training_step_time": 10.014431715011597
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 3.154296636581421,
      "step": 4426
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 4426,
      "training_step_time": 9.932064294815063
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 3.43890643119812,
      "step": 4426
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 4426,
      "training_step_time": 10.214078664779663
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 3.1476142406463623,
      "step": 4426
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 4426,
      "training_step_time": 9.928972482681274
    },
    {
      "epoch": 1.688385009765625e-05,
      "model_forward_time": 3.1550345420837402,
      "step": 4426
    },
    {
      "epoch": 1.688385009765625e-05,
      "step": 4426,
      "training_step_time": 10.288756132125854
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 3.1687936782836914,
      "step": 4427
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 4427,
      "training_step_time": 9.94862699508667
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 3.1464099884033203,
      "step": 4427
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 4427,
      "training_step_time": 9.925869226455688
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 3.4191091060638428,
      "step": 4427
    },
    {
      "dynamics_loss_avg": 0.0631975071504712,
      "epoch": 1.6887664794921873e-05,
      "step": 4427
    },
    {
      "action_loss_avg": 0.02844929900020361,
      "epoch": 1.6887664794921873e-05,
      "step": 4427
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 4427,
      "training_step_time": 10.205922365188599
    },
    {
      "epoch": 1.6887664794921873e-05,
      "model_forward_time": 3.1505401134490967,
      "step": 4427
    },
    {
      "epoch": 1.6887664794921873e-05,
      "step": 4427,
      "training_step_time": 10.023409128189087
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 3.1875298023223877,
      "step": 4428
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 4428,
      "training_step_time": 9.970085620880127
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 3.143857002258301,
      "step": 4428
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 4428,
      "training_step_time": 9.918005466461182
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 3.1469123363494873,
      "step": 4428
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 4428,
      "training_step_time": 9.921808242797852
    },
    {
      "epoch": 1.68914794921875e-05,
      "model_forward_time": 3.4478297233581543,
      "step": 4428
    },
    {
      "epoch": 1.68914794921875e-05,
      "step": 4428,
      "training_step_time": 10.315798044204712
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 2.4487404823303223,
      "step": 4429
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 4429,
      "training_step_time": 7.480654001235962
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 2.4365620613098145,
      "step": 4429
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 4429,
      "training_step_time": 7.463132619857788
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 3.1429080963134766,
      "step": 4429
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 4429,
      "training_step_time": 9.914238929748535
    },
    {
      "epoch": 1.6895294189453126e-05,
      "model_forward_time": 3.142052173614502,
      "step": 4429
    },
    {
      "epoch": 1.6895294189453126e-05,
      "step": 4429,
      "training_step_time": 10.274779558181763
    },
    {
      "epoch": 1.689910888671875e-05,
      "grad_norm": 0.32005518674850464,
      "learning_rate": 1e-05,
      "loss": 0.0924,
      "step": 4430
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 3.1523921489715576,
      "step": 4430
    },
    {
      "dynamics_loss_avg": 0.05780662801116705,
      "epoch": 1.689910888671875e-05,
      "step": 4430
    },
    {
      "action_loss_avg": 0.02152993017807603,
      "epoch": 1.689910888671875e-05,
      "step": 4430
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 4430,
      "training_step_time": 9.9271399974823
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 3.4112114906311035,
      "step": 4430
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 4430,
      "training_step_time": 10.183724403381348
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 2.442411422729492,
      "step": 4430
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 4430,
      "training_step_time": 7.470484972000122
    },
    {
      "epoch": 1.689910888671875e-05,
      "model_forward_time": 3.141615152359009,
      "step": 4430
    },
    {
      "epoch": 1.689910888671875e-05,
      "step": 4430,
      "training_step_time": 10.013710975646973
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 3.193606376647949,
      "step": 4431
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 4431,
      "training_step_time": 9.972227573394775
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 3.14593505859375,
      "step": 4431
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 4431,
      "training_step_time": 10.20690369606018
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 3.1503803730010986,
      "step": 4431
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 4431,
      "training_step_time": 9.931741952896118
    },
    {
      "epoch": 1.6902923583984375e-05,
      "model_forward_time": 3.146911144256592,
      "step": 4431
    },
    {
      "epoch": 1.6902923583984375e-05,
      "step": 4431,
      "training_step_time": 10.024778842926025
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 3.1864821910858154,
      "step": 4432
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 4432,
      "training_step_time": 9.972339868545532
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 3.145962715148926,
      "step": 4432
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 4432,
      "training_step_time": 9.926117897033691
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 3.148334503173828,
      "step": 4432
    },
    {
      "dynamics_loss_avg": 0.06824646685272455,
      "epoch": 1.690673828125e-05,
      "step": 4432
    },
    {
      "action_loss_avg": 0.03069143765605986,
      "epoch": 1.690673828125e-05,
      "step": 4432
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 4432,
      "training_step_time": 10.199586153030396
    },
    {
      "epoch": 1.690673828125e-05,
      "model_forward_time": 3.144378662109375,
      "step": 4432
    },
    {
      "epoch": 1.690673828125e-05,
      "step": 4432,
      "training_step_time": 10.304309844970703
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 3.1633121967315674,
      "step": 4433
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 4433,
      "training_step_time": 9.939281702041626
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 3.1485707759857178,
      "step": 4433
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 4433,
      "training_step_time": 9.921802759170532
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 3.143500804901123,
      "step": 4433
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 4433,
      "training_step_time": 9.920985221862793
    },
    {
      "epoch": 1.6910552978515624e-05,
      "model_forward_time": 3.395426034927368,
      "step": 4433
    },
    {
      "epoch": 1.6910552978515624e-05,
      "step": 4433,
      "training_step_time": 10.278724431991577
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 3.1715221405029297,
      "step": 4434
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 4434,
      "training_step_time": 9.948596477508545
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 3.144443988800049,
      "step": 4434
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 4434,
      "training_step_time": 9.92221212387085
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 3.1409108638763428,
      "step": 4434
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 4434,
      "training_step_time": 9.917611360549927
    },
    {
      "epoch": 1.691436767578125e-05,
      "model_forward_time": 3.144787073135376,
      "step": 4434
    },
    {
      "epoch": 1.691436767578125e-05,
      "step": 4434,
      "training_step_time": 10.27434253692627
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 3.424027442932129,
      "step": 4435
    },
    {
      "dynamics_loss_avg": 0.059607143700122836,
      "epoch": 1.6918182373046874e-05,
      "step": 4435
    },
    {
      "action_loss_avg": 0.02971086697652936,
      "epoch": 1.6918182373046874e-05,
      "step": 4435
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 4435,
      "training_step_time": 10.21015453338623
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 3.1462559700012207,
      "step": 4435
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 4435,
      "training_step_time": 9.926510334014893
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 2.437455654144287,
      "step": 4435
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 4435,
      "training_step_time": 7.471205472946167
    },
    {
      "epoch": 1.6918182373046874e-05,
      "model_forward_time": 3.1412229537963867,
      "step": 4435
    },
    {
      "epoch": 1.6918182373046874e-05,
      "step": 4435,
      "training_step_time": 10.02371072769165
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 3.150552988052368,
      "step": 4436
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 4436,
      "training_step_time": 9.938272953033447
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 3.1344153881073,
      "step": 4436
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 4436,
      "training_step_time": 10.184159517288208
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 3.134568452835083,
      "step": 4436
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 4436,
      "training_step_time": 9.914424419403076
    },
    {
      "epoch": 1.69219970703125e-05,
      "model_forward_time": 2.4313623905181885,
      "step": 4436
    },
    {
      "epoch": 1.69219970703125e-05,
      "step": 4436,
      "training_step_time": 10.020138263702393
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 3.1546714305877686,
      "step": 4437
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 4437,
      "training_step_time": 9.932228565216064
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 3.1437528133392334,
      "step": 4437
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 4437,
      "training_step_time": 9.92415452003479
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 3.416325569152832,
      "step": 4437
    },
    {
      "dynamics_loss_avg": 0.05840884801000357,
      "epoch": 1.6925811767578126e-05,
      "step": 4437
    },
    {
      "action_loss_avg": 0.022527293115854264,
      "epoch": 1.6925811767578126e-05,
      "step": 4437
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 4437,
      "training_step_time": 10.198150396347046
    },
    {
      "epoch": 1.6925811767578126e-05,
      "model_forward_time": 3.15102481842041,
      "step": 4437
    },
    {
      "epoch": 1.6925811767578126e-05,
      "step": 4437,
      "training_step_time": 10.336790084838867
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 3.1549715995788574,
      "step": 4438
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 4438,
      "training_step_time": 9.935444593429565
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 3.1346726417541504,
      "step": 4438
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 4438,
      "training_step_time": 9.91291069984436
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 3.1408140659332275,
      "step": 4438
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 4438,
      "training_step_time": 9.921390056610107
    },
    {
      "epoch": 1.692962646484375e-05,
      "model_forward_time": 3.134155750274658,
      "step": 4438
    },
    {
      "epoch": 1.692962646484375e-05,
      "step": 4438,
      "training_step_time": 10.26762342453003
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 3.1527092456817627,
      "step": 4439
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 4439,
      "training_step_time": 9.930752277374268
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 3.1345908641815186,
      "step": 4439
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 4439,
      "training_step_time": 9.909648418426514
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 3.137880325317383,
      "step": 4439
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 4439,
      "training_step_time": 9.909169435501099
    },
    {
      "epoch": 1.6933441162109375e-05,
      "model_forward_time": 3.135796546936035,
      "step": 4439
    },
    {
      "epoch": 1.6933441162109375e-05,
      "step": 4439,
      "training_step_time": 10.271087408065796
    },
    {
      "epoch": 1.6937255859375e-05,
      "grad_norm": 0.4125949442386627,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 4440
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 3.4185237884521484,
      "step": 4440
    },
    {
      "dynamics_loss_avg": 0.06370576843619347,
      "epoch": 1.6937255859375e-05,
      "step": 4440
    },
    {
      "action_loss_avg": 0.031066617462784052,
      "epoch": 1.6937255859375e-05,
      "step": 4440
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 4440,
      "training_step_time": 10.205039024353027
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 3.13346529006958,
      "step": 4440
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 4440,
      "training_step_time": 9.909669876098633
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 3.135422706604004,
      "step": 4440
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 4440,
      "training_step_time": 9.916887521743774
    },
    {
      "epoch": 1.6937255859375e-05,
      "model_forward_time": 3.137097120285034,
      "step": 4440
    },
    {
      "epoch": 1.6937255859375e-05,
      "step": 4440,
      "training_step_time": 10.283566951751709
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 3.180716037750244,
      "step": 4441
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 4441,
      "training_step_time": 9.961278200149536
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 3.3855643272399902,
      "step": 4441
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 4441,
      "training_step_time": 10.16140365600586
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 3.1395106315612793,
      "step": 4441
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 4441,
      "training_step_time": 9.914481401443481
    },
    {
      "epoch": 1.6941070556640624e-05,
      "model_forward_time": 3.1303904056549072,
      "step": 4441
    },
    {
      "epoch": 1.6941070556640624e-05,
      "step": 4441,
      "training_step_time": 10.31615138053894
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 3.1701500415802,
      "step": 4442
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 4442,
      "training_step_time": 9.95335841178894
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 3.140169858932495,
      "step": 4442
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 4442,
      "training_step_time": 9.920087814331055
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 3.4392590522766113,
      "step": 4442
    },
    {
      "dynamics_loss_avg": 0.052287394739687444,
      "epoch": 1.694488525390625e-05,
      "step": 4442
    },
    {
      "action_loss_avg": 0.019103549188002944,
      "epoch": 1.694488525390625e-05,
      "step": 4442
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 4442,
      "training_step_time": 10.22022271156311
    },
    {
      "epoch": 1.694488525390625e-05,
      "model_forward_time": 3.1391079425811768,
      "step": 4442
    },
    {
      "epoch": 1.694488525390625e-05,
      "step": 4442,
      "training_step_time": 10.275627851486206
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 2.470186471939087,
      "step": 4443
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 4443,
      "training_step_time": 7.506098031997681
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 3.1374447345733643,
      "step": 4443
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 4443,
      "training_step_time": 9.916862726211548
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 3.1325697898864746,
      "step": 4443
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 4443,
      "training_step_time": 9.912904262542725
    },
    {
      "epoch": 1.6948699951171874e-05,
      "model_forward_time": 3.388430595397949,
      "step": 4443
    },
    {
      "epoch": 1.6948699951171874e-05,
      "step": 4443,
      "training_step_time": 10.299315690994263
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 3.172497510910034,
      "step": 4444
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 4444,
      "training_step_time": 9.957587480545044
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 3.1389636993408203,
      "step": 4444
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 4444,
      "training_step_time": 9.91757345199585
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 3.141270160675049,
      "step": 4444
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 4444,
      "training_step_time": 9.920823574066162
    },
    {
      "epoch": 1.69525146484375e-05,
      "model_forward_time": 3.144453763961792,
      "step": 4444
    },
    {
      "epoch": 1.69525146484375e-05,
      "step": 4444,
      "training_step_time": 10.302233695983887
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 3.4417269229888916,
      "step": 4445
    },
    {
      "dynamics_loss_avg": 0.055389569327235225,
      "epoch": 1.6956329345703126e-05,
      "step": 4445
    },
    {
      "action_loss_avg": 0.0207483587320894,
      "epoch": 1.6956329345703126e-05,
      "step": 4445
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 4445,
      "training_step_time": 10.223267316818237
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 3.1436469554901123,
      "step": 4445
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 4445,
      "training_step_time": 9.922329187393188
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 3.1436216831207275,
      "step": 4445
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 4445,
      "training_step_time": 9.929722309112549
    },
    {
      "epoch": 1.6956329345703126e-05,
      "model_forward_time": 3.1431593894958496,
      "step": 4445
    },
    {
      "epoch": 1.6956329345703126e-05,
      "step": 4445,
      "training_step_time": 10.324384927749634
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 3.1770997047424316,
      "step": 4446
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 4446,
      "training_step_time": 9.959604740142822
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 3.4227635860443115,
      "step": 4446
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 4446,
      "training_step_time": 10.203986883163452
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 3.139899730682373,
      "step": 4446
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 4446,
      "training_step_time": 9.916509628295898
    },
    {
      "epoch": 1.696014404296875e-05,
      "model_forward_time": 3.141587257385254,
      "step": 4446
    },
    {
      "epoch": 1.696014404296875e-05,
      "step": 4446,
      "training_step_time": 10.013057231903076
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 3.175560474395752,
      "step": 4447
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 4447,
      "training_step_time": 9.950906991958618
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 3.135576009750366,
      "step": 4447
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 4447,
      "training_step_time": 9.914289474487305
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 3.399641752243042,
      "step": 4447
    },
    {
      "dynamics_loss_avg": 0.06518970765173435,
      "epoch": 1.6963958740234375e-05,
      "step": 4447
    },
    {
      "action_loss_avg": 0.03061332618817687,
      "epoch": 1.6963958740234375e-05,
      "step": 4447
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 4447,
      "training_step_time": 10.183196306228638
    },
    {
      "epoch": 1.6963958740234375e-05,
      "model_forward_time": 3.144801378250122,
      "step": 4447
    },
    {
      "epoch": 1.6963958740234375e-05,
      "step": 4447,
      "training_step_time": 10.016223430633545
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 3.1828958988189697,
      "step": 4448
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 4448,
      "training_step_time": 9.962532758712769
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 3.1366782188415527,
      "step": 4448
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 4448,
      "training_step_time": 9.914640426635742
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 3.1394307613372803,
      "step": 4448
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 4448,
      "training_step_time": 9.917212963104248
    },
    {
      "epoch": 1.69677734375e-05,
      "model_forward_time": 3.4149982929229736,
      "step": 4448
    },
    {
      "epoch": 1.69677734375e-05,
      "step": 4448,
      "training_step_time": 10.291809320449829
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 3.1737794876098633,
      "step": 4449
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 4449,
      "training_step_time": 9.960313320159912
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 3.1400554180145264,
      "step": 4449
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 4449,
      "training_step_time": 9.919068336486816
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 3.1433942317962646,
      "step": 4449
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 4449,
      "training_step_time": 9.924211740493774
    },
    {
      "epoch": 1.6971588134765624e-05,
      "model_forward_time": 3.143488883972168,
      "step": 4449
    },
    {
      "epoch": 1.6971588134765624e-05,
      "step": 4449,
      "training_step_time": 10.2733633518219
    },
    {
      "epoch": 1.697540283203125e-05,
      "grad_norm": 0.41736242175102234,
      "learning_rate": 1e-05,
      "loss": 0.0925,
      "step": 4450
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 3.453629493713379,
      "step": 4450
    },
    {
      "dynamics_loss_avg": 0.059710892476141454,
      "epoch": 1.697540283203125e-05,
      "step": 4450
    },
    {
      "action_loss_avg": 0.024300436908379196,
      "epoch": 1.697540283203125e-05,
      "step": 4450
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 4450,
      "training_step_time": 10.235702514648438
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 3.1445722579956055,
      "step": 4450
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 4450,
      "training_step_time": 9.925966262817383
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 3.1502928733825684,
      "step": 4450
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 4450,
      "training_step_time": 9.933932065963745
    },
    {
      "epoch": 1.697540283203125e-05,
      "model_forward_time": 3.1442372798919678,
      "step": 4450
    },
    {
      "epoch": 1.697540283203125e-05,
      "step": 4450,
      "training_step_time": 10.021110773086548
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 3.1643097400665283,
      "step": 4451
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 4451,
      "training_step_time": 9.949544906616211
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 3.3957059383392334,
      "step": 4451
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 4451,
      "training_step_time": 10.171833276748657
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 2.4352331161499023,
      "step": 4451
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 4451,
      "training_step_time": 7.468074798583984
    },
    {
      "epoch": 1.6979217529296874e-05,
      "model_forward_time": 3.1350674629211426,
      "step": 4451
    },
    {
      "epoch": 1.6979217529296874e-05,
      "step": 4451,
      "training_step_time": 10.278666496276855
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 3.1589269638061523,
      "step": 4452
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 4452,
      "training_step_time": 9.934342622756958
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 3.144815444946289,
      "step": 4452
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 4452,
      "training_step_time": 9.919265270233154
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 3.1363525390625,
      "step": 4452
    },
    {
      "dynamics_loss_avg": 0.05510903149843216,
      "epoch": 1.69830322265625e-05,
      "step": 4452
    },
    {
      "action_loss_avg": 0.023831924330443142,
      "epoch": 1.69830322265625e-05,
      "step": 4452
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 4452,
      "training_step_time": 10.193234205245972
    },
    {
      "epoch": 1.69830322265625e-05,
      "model_forward_time": 3.13427734375,
      "step": 4452
    },
    {
      "epoch": 1.69830322265625e-05,
      "step": 4452,
      "training_step_time": 10.014543771743774
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 2.478286027908325,
      "step": 4453
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 4453,
      "training_step_time": 7.5119102001190186
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 3.138888120651245,
      "step": 4453
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 4453,
      "training_step_time": 9.917548656463623
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 3.136859655380249,
      "step": 4453
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 4453,
      "training_step_time": 9.915565013885498
    },
    {
      "epoch": 1.6986846923828126e-05,
      "model_forward_time": 3.138937473297119,
      "step": 4453
    },
    {
      "epoch": 1.6986846923828126e-05,
      "step": 4453,
      "training_step_time": 10.284441471099854
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 3.189626455307007,
      "step": 4454
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 4454,
      "training_step_time": 9.983374118804932
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 3.1452646255493164,
      "step": 4454
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 4454,
      "training_step_time": 9.928712129592896
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 3.1421451568603516,
      "step": 4454
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 4454,
      "training_step_time": 9.925016641616821
    },
    {
      "epoch": 1.699066162109375e-05,
      "model_forward_time": 3.147482395172119,
      "step": 4454
    },
    {
      "epoch": 1.699066162109375e-05,
      "step": 4454,
      "training_step_time": 10.279062271118164
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 3.432762622833252,
      "step": 4455
    },
    {
      "dynamics_loss_avg": 0.06228054165840149,
      "epoch": 1.6994476318359375e-05,
      "step": 4455
    },
    {
      "action_loss_avg": 0.026067940844222905,
      "epoch": 1.6994476318359375e-05,
      "step": 4455
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 4455,
      "training_step_time": 10.216003179550171
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 3.1430394649505615,
      "step": 4455
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 4455,
      "training_step_time": 9.924429893493652
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 3.1429636478424072,
      "step": 4455
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 4455,
      "training_step_time": 9.927245140075684
    },
    {
      "epoch": 1.6994476318359375e-05,
      "model_forward_time": 3.144196033477783,
      "step": 4455
    },
    {
      "epoch": 1.6994476318359375e-05,
      "step": 4455,
      "training_step_time": 10.028244018554688
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 3.1560091972351074,
      "step": 4456
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 4456,
      "training_step_time": 9.937913656234741
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 3.1344077587127686,
      "step": 4456
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 4456,
      "training_step_time": 10.216721534729004
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 2.4316794872283936,
      "step": 4456
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 4456,
      "training_step_time": 7.4643659591674805
    },
    {
      "epoch": 1.6998291015625e-05,
      "model_forward_time": 3.1351051330566406,
      "step": 4456
    },
    {
      "epoch": 1.6998291015625e-05,
      "step": 4456,
      "training_step_time": 10.265652894973755
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 3.1795005798339844,
      "step": 4457
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 4457,
      "training_step_time": 9.950189590454102
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 3.1288082599639893,
      "step": 4457
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 4457,
      "training_step_time": 9.89711308479309
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 3.399120330810547,
      "step": 4457
    },
    {
      "dynamics_loss_avg": 0.06374799646437168,
      "epoch": 1.7002105712890624e-05,
      "step": 4457
    },
    {
      "action_loss_avg": 0.01919529833830893,
      "epoch": 1.7002105712890624e-05,
      "step": 4457
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 4457,
      "training_step_time": 10.17101764678955
    },
    {
      "epoch": 1.7002105712890624e-05,
      "model_forward_time": 3.1286773681640625,
      "step": 4457
    },
    {
      "epoch": 1.7002105712890624e-05,
      "step": 4457,
      "training_step_time": 10.2968909740448
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 3.170931577682495,
      "step": 4458
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 4458,
      "training_step_time": 9.940536975860596
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 3.131805419921875,
      "step": 4458
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 4458,
      "training_step_time": 9.90242600440979
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 3.1372663974761963,
      "step": 4458
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 4458,
      "training_step_time": 9.912155628204346
    },
    {
      "epoch": 1.700592041015625e-05,
      "model_forward_time": 3.1360690593719482,
      "step": 4458
    },
    {
      "epoch": 1.700592041015625e-05,
      "step": 4458,
      "training_step_time": 10.316333055496216
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 3.171886682510376,
      "step": 4459
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 4459,
      "training_step_time": 9.950064182281494
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 2.4319400787353516,
      "step": 4459
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 4459,
      "training_step_time": 7.466463088989258
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 3.1382434368133545,
      "step": 4459
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 4459,
      "training_step_time": 9.918300151824951
    },
    {
      "epoch": 1.7009735107421874e-05,
      "model_forward_time": 3.1392476558685303,
      "step": 4459
    },
    {
      "epoch": 1.7009735107421874e-05,
      "step": 4459,
      "training_step_time": 10.015259504318237
    },
    {
      "epoch": 1.70135498046875e-05,
      "grad_norm": 0.6004653573036194,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 4460
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 3.4571444988250732,
      "step": 4460
    },
    {
      "dynamics_loss_avg": 0.060593714378774165,
      "epoch": 1.70135498046875e-05,
      "step": 4460
    },
    {
      "action_loss_avg": 0.023074275301769375,
      "epoch": 1.70135498046875e-05,
      "step": 4460
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 4460,
      "training_step_time": 10.24045205116272
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 3.1417195796966553,
      "step": 4460
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 4460,
      "training_step_time": 9.923779487609863
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 3.137725353240967,
      "step": 4460
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 4460,
      "training_step_time": 9.917594194412231
    },
    {
      "epoch": 1.70135498046875e-05,
      "model_forward_time": 3.1381986141204834,
      "step": 4460
    },
    {
      "epoch": 1.70135498046875e-05,
      "step": 4460,
      "training_step_time": 10.280324220657349
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 3.1684916019439697,
      "step": 4461
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 4461,
      "training_step_time": 9.944944620132446
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 3.395691156387329,
      "step": 4461
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 4461,
      "training_step_time": 10.17084550857544
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 3.140486240386963,
      "step": 4461
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 4461,
      "training_step_time": 9.922786235809326
    },
    {
      "epoch": 1.7017364501953126e-05,
      "model_forward_time": 3.1363399028778076,
      "step": 4461
    },
    {
      "epoch": 1.7017364501953126e-05,
      "step": 4461,
      "training_step_time": 10.275043487548828
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 3.1952192783355713,
      "step": 4462
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 4462,
      "training_step_time": 9.975853204727173
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 3.1384966373443604,
      "step": 4462
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 4462,
      "training_step_time": 9.916768074035645
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 3.13285493850708,
      "step": 4462
    },
    {
      "dynamics_loss_avg": 0.05379174128174782,
      "epoch": 1.702117919921875e-05,
      "step": 4462
    },
    {
      "action_loss_avg": 0.020816108398139476,
      "epoch": 1.702117919921875e-05,
      "step": 4462
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 4462,
      "training_step_time": 10.197747230529785
    },
    {
      "epoch": 1.702117919921875e-05,
      "model_forward_time": 3.1339149475097656,
      "step": 4462
    },
    {
      "epoch": 1.702117919921875e-05,
      "step": 4462,
      "training_step_time": 10.281047105789185
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 3.1650242805480957,
      "step": 4463
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 4463,
      "training_step_time": 9.945828437805176
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 3.1372714042663574,
      "step": 4463
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 4463,
      "training_step_time": 9.91220736503601
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 3.1356117725372314,
      "step": 4463
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 4463,
      "training_step_time": 9.9111647605896
    },
    {
      "epoch": 1.7024993896484375e-05,
      "model_forward_time": 3.1365325450897217,
      "step": 4463
    },
    {
      "epoch": 1.7024993896484375e-05,
      "step": 4463,
      "training_step_time": 10.291933298110962
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 3.179516315460205,
      "step": 4464
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 4464,
      "training_step_time": 9.957656145095825
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 3.1397407054901123,
      "step": 4464
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 4464,
      "training_step_time": 9.919915914535522
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 3.1382908821105957,
      "step": 4464
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 4464,
      "training_step_time": 9.918184518814087
    },
    {
      "epoch": 1.702880859375e-05,
      "model_forward_time": 3.140655040740967,
      "step": 4464
    },
    {
      "epoch": 1.702880859375e-05,
      "step": 4464,
      "training_step_time": 10.278422594070435
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 2.469642162322998,
      "step": 4465
    },
    {
      "dynamics_loss_avg": 0.07074579373002052,
      "epoch": 1.7032623291015624e-05,
      "step": 4465
    },
    {
      "action_loss_avg": 0.03423945251852274,
      "epoch": 1.7032623291015624e-05,
      "step": 4465
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 4465,
      "training_step_time": 7.5041162967681885
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 3.400869131088257,
      "step": 4465
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 4465,
      "training_step_time": 10.177133321762085
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 3.1332621574401855,
      "step": 4465
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 4465,
      "training_step_time": 9.908617734909058
    },
    {
      "epoch": 1.7032623291015624e-05,
      "model_forward_time": 3.137174129486084,
      "step": 4465
    },
    {
      "epoch": 1.7032623291015624e-05,
      "step": 4465,
      "training_step_time": 10.288687467575073
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 3.19063401222229,
      "step": 4466
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 4466,
      "training_step_time": 9.960438013076782
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 3.4100849628448486,
      "step": 4466
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 4466,
      "training_step_time": 10.178296327590942
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 3.1298341751098633,
      "step": 4466
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 4466,
      "training_step_time": 9.901751518249512
    },
    {
      "epoch": 1.703643798828125e-05,
      "model_forward_time": 3.1295199394226074,
      "step": 4466
    },
    {
      "epoch": 1.703643798828125e-05,
      "step": 4466,
      "training_step_time": 10.001180410385132
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 3.17147159576416,
      "step": 4467
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 4467,
      "training_step_time": 9.94437575340271
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 3.1306028366088867,
      "step": 4467
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 4467,
      "training_step_time": 9.909972190856934
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 3.1381261348724365,
      "step": 4467
    },
    {
      "dynamics_loss_avg": 0.07632546871900558,
      "epoch": 1.7040252685546874e-05,
      "step": 4467
    },
    {
      "action_loss_avg": 0.031867379788309336,
      "epoch": 1.7040252685546874e-05,
      "step": 4467
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 4467,
      "training_step_time": 10.18578839302063
    },
    {
      "epoch": 1.7040252685546874e-05,
      "model_forward_time": 3.1346209049224854,
      "step": 4467
    },
    {
      "epoch": 1.7040252685546874e-05,
      "step": 4467,
      "training_step_time": 10.276854515075684
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 3.165621519088745,
      "step": 4468
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 4468,
      "training_step_time": 9.943541765213013
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 3.1387994289398193,
      "step": 4468
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 4468,
      "training_step_time": 9.913593769073486
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 3.1393587589263916,
      "step": 4468
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 4468,
      "training_step_time": 9.914188385009766
    },
    {
      "epoch": 1.70440673828125e-05,
      "model_forward_time": 3.422955274581909,
      "step": 4468
    },
    {
      "epoch": 1.70440673828125e-05,
      "step": 4468,
      "training_step_time": 10.290263891220093
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 3.1884636878967285,
      "step": 4469
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 4469,
      "training_step_time": 9.964585781097412
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 3.1374990940093994,
      "step": 4469
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 4469,
      "training_step_time": 9.911813735961914
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 3.140838861465454,
      "step": 4469
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 4469,
      "training_step_time": 9.91251254081726
    },
    {
      "epoch": 1.7047882080078126e-05,
      "model_forward_time": 3.133033514022827,
      "step": 4469
    },
    {
      "epoch": 1.7047882080078126e-05,
      "step": 4469,
      "training_step_time": 10.300992250442505
    },
    {
      "epoch": 1.705169677734375e-05,
      "grad_norm": 0.45994338393211365,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 4470
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 3.459226369857788,
      "step": 4470
    },
    {
      "dynamics_loss_avg": 0.061256558261811735,
      "epoch": 1.705169677734375e-05,
      "step": 4470
    },
    {
      "action_loss_avg": 0.02116753808222711,
      "epoch": 1.705169677734375e-05,
      "step": 4470
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 4470,
      "training_step_time": 10.242425680160522
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 3.132615804672241,
      "step": 4470
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 4470,
      "training_step_time": 9.90284538269043
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 3.134777307510376,
      "step": 4470
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 4470,
      "training_step_time": 9.909294843673706
    },
    {
      "epoch": 1.705169677734375e-05,
      "model_forward_time": 3.1339783668518066,
      "step": 4470
    },
    {
      "epoch": 1.705169677734375e-05,
      "step": 4470,
      "training_step_time": 10.244161605834961
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 3.193711757659912,
      "step": 4471
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 4471,
      "training_step_time": 9.968297958374023
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 3.3901662826538086,
      "step": 4471
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 4471,
      "training_step_time": 10.168460369110107
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 3.1430654525756836,
      "step": 4471
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 4471,
      "training_step_time": 9.925252199172974
    },
    {
      "epoch": 1.7055511474609375e-05,
      "model_forward_time": 3.134371280670166,
      "step": 4471
    },
    {
      "epoch": 1.7055511474609375e-05,
      "step": 4471,
      "training_step_time": 10.312788248062134
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 3.1803746223449707,
      "step": 4472
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 4472,
      "training_step_time": 9.963308095932007
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 3.14443039894104,
      "step": 4472
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 4472,
      "training_step_time": 9.925124645233154
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 3.439185857772827,
      "step": 4472
    },
    {
      "dynamics_loss_avg": 0.0653216402977705,
      "epoch": 1.7059326171875e-05,
      "step": 4472
    },
    {
      "action_loss_avg": 0.026826943270862103,
      "epoch": 1.7059326171875e-05,
      "step": 4472
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 4472,
      "training_step_time": 10.223475933074951
    },
    {
      "epoch": 1.7059326171875e-05,
      "model_forward_time": 3.1471669673919678,
      "step": 4472
    },
    {
      "epoch": 1.7059326171875e-05,
      "step": 4472,
      "training_step_time": 10.033344984054565
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 3.1680502891540527,
      "step": 4473
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 4473,
      "training_step_time": 9.949524879455566
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 3.1416923999786377,
      "step": 4473
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 4473,
      "training_step_time": 9.920572280883789
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 3.136831521987915,
      "step": 4473
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 4473,
      "training_step_time": 9.91642951965332
    },
    {
      "epoch": 1.7063140869140624e-05,
      "model_forward_time": 3.137988328933716,
      "step": 4473
    },
    {
      "epoch": 1.7063140869140624e-05,
      "step": 4473,
      "training_step_time": 10.306751251220703
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 3.1793148517608643,
      "step": 4474
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 4474,
      "training_step_time": 9.951019763946533
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 3.138411045074463,
      "step": 4474
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 4474,
      "training_step_time": 9.915330648422241
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 3.135251045227051,
      "step": 4474
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 4474,
      "training_step_time": 9.910527467727661
    },
    {
      "epoch": 1.706695556640625e-05,
      "model_forward_time": 3.135873556137085,
      "step": 4474
    },
    {
      "epoch": 1.706695556640625e-05,
      "step": 4474,
      "training_step_time": 10.298707246780396
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 3.4408774375915527,
      "step": 4475
    },
    {
      "dynamics_loss_avg": 0.07200627774000168,
      "epoch": 1.7070770263671874e-05,
      "step": 4475
    },
    {
      "action_loss_avg": 0.02409036085009575,
      "epoch": 1.7070770263671874e-05,
      "step": 4475
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 4475,
      "training_step_time": 10.22623872756958
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 3.138340950012207,
      "step": 4475
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 4475,
      "training_step_time": 9.918547630310059
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 3.1397151947021484,
      "step": 4475
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 4475,
      "training_step_time": 9.921236991882324
    },
    {
      "epoch": 1.7070770263671874e-05,
      "model_forward_time": 3.1449263095855713,
      "step": 4475
    },
    {
      "epoch": 1.7070770263671874e-05,
      "step": 4475,
      "training_step_time": 10.3001549243927
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 3.180983543395996,
      "step": 4476
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 4476,
      "training_step_time": 9.97313141822815
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 3.140310525894165,
      "step": 4476
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 4476,
      "training_step_time": 10.204243183135986
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 3.132620096206665,
      "step": 4476
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 4476,
      "training_step_time": 9.909828424453735
    },
    {
      "epoch": 1.70745849609375e-05,
      "model_forward_time": 3.1349868774414062,
      "step": 4476
    },
    {
      "epoch": 1.70745849609375e-05,
      "step": 4476,
      "training_step_time": 10.005521059036255
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 3.1907520294189453,
      "step": 4477
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 4477,
      "training_step_time": 9.96916651725769
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 3.140284776687622,
      "step": 4477
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 4477,
      "training_step_time": 9.916393041610718
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 3.4050772190093994,
      "step": 4477
    },
    {
      "dynamics_loss_avg": 0.08178000301122665,
      "epoch": 1.7078399658203126e-05,
      "step": 4477
    },
    {
      "action_loss_avg": 0.03791603660210967,
      "epoch": 1.7078399658203126e-05,
      "step": 4477
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 4477,
      "training_step_time": 10.187687158584595
    },
    {
      "epoch": 1.7078399658203126e-05,
      "model_forward_time": 3.1475889682769775,
      "step": 4477
    },
    {
      "epoch": 1.7078399658203126e-05,
      "step": 4477,
      "training_step_time": 10.023839235305786
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 3.1915042400360107,
      "step": 4478
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 4478,
      "training_step_time": 9.97344970703125
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 3.146998167037964,
      "step": 4478
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 4478,
      "training_step_time": 9.927634954452515
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 3.1543643474578857,
      "step": 4478
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 4478,
      "training_step_time": 9.944665670394897
    },
    {
      "epoch": 1.708221435546875e-05,
      "model_forward_time": 3.1484127044677734,
      "step": 4478
    },
    {
      "epoch": 1.708221435546875e-05,
      "step": 4478,
      "training_step_time": 10.31130075454712
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 3.188261032104492,
      "step": 4479
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 4479,
      "training_step_time": 9.972590684890747
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 3.1485064029693604,
      "step": 4479
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 4479,
      "training_step_time": 9.926708698272705
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 3.144564390182495,
      "step": 4479
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 4479,
      "training_step_time": 9.923933506011963
    },
    {
      "epoch": 1.7086029052734375e-05,
      "model_forward_time": 2.438166618347168,
      "step": 4479
    },
    {
      "epoch": 1.7086029052734375e-05,
      "step": 4479,
      "training_step_time": 10.306751728057861
    },
    {
      "epoch": 1.708984375e-05,
      "grad_norm": 0.24121983349323273,
      "learning_rate": 1e-05,
      "loss": 0.0907,
      "step": 4480
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 3.4712438583374023,
      "step": 4480
    },
    {
      "dynamics_loss_avg": 0.06492396853864194,
      "epoch": 1.708984375e-05,
      "step": 4480
    },
    {
      "action_loss_avg": 0.02978263758122921,
      "epoch": 1.708984375e-05,
      "step": 4480
    },
    {
      "epoch": 1.708984375e-05,
      "step": 4480,
      "training_step_time": 10.247830152511597
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 3.1429927349090576,
      "step": 4480
    },
    {
      "epoch": 1.708984375e-05,
      "step": 4480,
      "training_step_time": 9.915142297744751
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 3.145458936691284,
      "step": 4480
    },
    {
      "epoch": 1.708984375e-05,
      "step": 4480,
      "training_step_time": 9.920047044754028
    },
    {
      "epoch": 1.708984375e-05,
      "model_forward_time": 3.1438379287719727,
      "step": 4480
    },
    {
      "epoch": 1.708984375e-05,
      "step": 4480,
      "training_step_time": 10.269728183746338
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 3.183018207550049,
      "step": 4481
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 4481,
      "training_step_time": 9.96005654335022
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 3.4268970489501953,
      "step": 4481
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 4481,
      "training_step_time": 10.203849792480469
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 3.146024227142334,
      "step": 4481
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 4481,
      "training_step_time": 9.924078226089478
    },
    {
      "epoch": 1.7093658447265625e-05,
      "model_forward_time": 3.1429014205932617,
      "step": 4481
    },
    {
      "epoch": 1.7093658447265625e-05,
      "step": 4481,
      "training_step_time": 10.013984441757202
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 3.180074691772461,
      "step": 4482
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 4482,
      "training_step_time": 9.954445600509644
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 3.1424200534820557,
      "step": 4482
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 4482,
      "training_step_time": 9.919221878051758
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 3.136779308319092,
      "step": 4482
    },
    {
      "dynamics_loss_avg": 0.07825396973639727,
      "epoch": 1.709747314453125e-05,
      "step": 4482
    },
    {
      "action_loss_avg": 0.03172379690222442,
      "epoch": 1.709747314453125e-05,
      "step": 4482
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 4482,
      "training_step_time": 10.199185371398926
    },
    {
      "epoch": 1.709747314453125e-05,
      "model_forward_time": 3.143883466720581,
      "step": 4482
    },
    {
      "epoch": 1.709747314453125e-05,
      "step": 4482,
      "training_step_time": 10.016436338424683
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 3.1686649322509766,
      "step": 4483
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 4483,
      "training_step_time": 9.94741153717041
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 3.1431963443756104,
      "step": 4483
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 4483,
      "training_step_time": 9.91989254951477
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 3.145409345626831,
      "step": 4483
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 4483,
      "training_step_time": 9.925410032272339
    },
    {
      "epoch": 1.7101287841796874e-05,
      "model_forward_time": 3.146106243133545,
      "step": 4483
    },
    {
      "epoch": 1.7101287841796874e-05,
      "step": 4483,
      "training_step_time": 10.29387640953064
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 3.1794826984405518,
      "step": 4484
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 4484,
      "training_step_time": 9.962899923324585
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 3.1481997966766357,
      "step": 4484
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 4484,
      "training_step_time": 9.923319578170776
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 3.1432719230651855,
      "step": 4484
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 4484,
      "training_step_time": 9.924325227737427
    },
    {
      "epoch": 1.71051025390625e-05,
      "model_forward_time": 3.1500244140625,
      "step": 4484
    },
    {
      "epoch": 1.71051025390625e-05,
      "step": 4484,
      "training_step_time": 10.296651124954224
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 3.169586420059204,
      "step": 4485
    },
    {
      "dynamics_loss_avg": 0.06723092496395111,
      "epoch": 1.7108917236328126e-05,
      "step": 4485
    },
    {
      "action_loss_avg": 0.026465922966599463,
      "epoch": 1.7108917236328126e-05,
      "step": 4485
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 4485,
      "training_step_time": 10.228647232055664
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 3.1425299644470215,
      "step": 4485
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 4485,
      "training_step_time": 9.919513702392578
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 3.147010326385498,
      "step": 4485
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 4485,
      "training_step_time": 9.930585145950317
    },
    {
      "epoch": 1.7108917236328126e-05,
      "model_forward_time": 3.140972375869751,
      "step": 4485
    },
    {
      "epoch": 1.7108917236328126e-05,
      "step": 4485,
      "training_step_time": 10.018982887268066
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 3.1850779056549072,
      "step": 4486
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 4486,
      "training_step_time": 9.957620620727539
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 3.419142723083496,
      "step": 4486
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 4486,
      "training_step_time": 10.18809461593628
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 3.1383426189422607,
      "step": 4486
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 4486,
      "training_step_time": 9.90635871887207
    },
    {
      "epoch": 1.711273193359375e-05,
      "model_forward_time": 3.1441378593444824,
      "step": 4486
    },
    {
      "epoch": 1.711273193359375e-05,
      "step": 4486,
      "training_step_time": 10.27124834060669
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 3.180687189102173,
      "step": 4487
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 4487,
      "training_step_time": 9.949889898300171
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 3.1354496479034424,
      "step": 4487
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 4487,
      "training_step_time": 9.909115314483643
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 3.1371326446533203,
      "step": 4487
    },
    {
      "dynamics_loss_avg": 0.07659922689199447,
      "epoch": 1.7116546630859375e-05,
      "step": 4487
    },
    {
      "action_loss_avg": 0.032871304731816056,
      "epoch": 1.7116546630859375e-05,
      "step": 4487
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 4487,
      "training_step_time": 10.186649322509766
    },
    {
      "epoch": 1.7116546630859375e-05,
      "model_forward_time": 3.1373562812805176,
      "step": 4487
    },
    {
      "epoch": 1.7116546630859375e-05,
      "step": 4487,
      "training_step_time": 10.016116380691528
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 3.1928067207336426,
      "step": 4488
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 4488,
      "training_step_time": 9.969107866287231
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 3.144914150238037,
      "step": 4488
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 4488,
      "training_step_time": 9.925092697143555
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 3.148111581802368,
      "step": 4488
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 4488,
      "training_step_time": 9.922265768051147
    },
    {
      "epoch": 1.7120361328125e-05,
      "model_forward_time": 3.4250268936157227,
      "step": 4488
    },
    {
      "epoch": 1.7120361328125e-05,
      "step": 4488,
      "training_step_time": 10.29462194442749
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 3.1838932037353516,
      "step": 4489
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 4489,
      "training_step_time": 9.965880393981934
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 3.142793655395508,
      "step": 4489
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 4489,
      "training_step_time": 9.921695947647095
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 3.1438419818878174,
      "step": 4489
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 4489,
      "training_step_time": 9.92467474937439
    },
    {
      "epoch": 1.7124176025390625e-05,
      "model_forward_time": 2.4394850730895996,
      "step": 4489
    },
    {
      "epoch": 1.7124176025390625e-05,
      "step": 4489,
      "training_step_time": 10.314607381820679
    },
    {
      "epoch": 1.712799072265625e-05,
      "grad_norm": 0.32789021730422974,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 4490
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 3.1702256202697754,
      "step": 4490
    },
    {
      "dynamics_loss_avg": 0.06706365495920182,
      "epoch": 1.712799072265625e-05,
      "step": 4490
    },
    {
      "action_loss_avg": 0.02185864655766636,
      "epoch": 1.712799072265625e-05,
      "step": 4490
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 4490,
      "training_step_time": 10.21191143989563
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 2.4360086917877197,
      "step": 4490
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 4490,
      "training_step_time": 7.462806701660156
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 3.1427810192108154,
      "step": 4490
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 4490,
      "training_step_time": 9.913269996643066
    },
    {
      "epoch": 1.712799072265625e-05,
      "model_forward_time": 3.142728090286255,
      "step": 4490
    },
    {
      "epoch": 1.712799072265625e-05,
      "step": 4490,
      "training_step_time": 10.313684225082397
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 3.1741175651550293,
      "step": 4491
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 4491,
      "training_step_time": 9.948355436325073
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 3.401198387145996,
      "step": 4491
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 4491,
      "training_step_time": 10.17491602897644
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 3.1489367485046387,
      "step": 4491
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 4491,
      "training_step_time": 9.92260193824768
    },
    {
      "epoch": 1.7131805419921874e-05,
      "model_forward_time": 3.1441757678985596,
      "step": 4491
    },
    {
      "epoch": 1.7131805419921874e-05,
      "step": 4491,
      "training_step_time": 10.283611297607422
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 3.1713290214538574,
      "step": 4492
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 4492,
      "training_step_time": 9.950353384017944
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 3.141731023788452,
      "step": 4492
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 4492,
      "training_step_time": 9.918838500976562
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 3.14858341217041,
      "step": 4492
    },
    {
      "dynamics_loss_avg": 0.05747376475483179,
      "epoch": 1.71356201171875e-05,
      "step": 4492
    },
    {
      "action_loss_avg": 0.025432649813592435,
      "epoch": 1.71356201171875e-05,
      "step": 4492
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 4492,
      "training_step_time": 10.205147981643677
    },
    {
      "epoch": 1.71356201171875e-05,
      "model_forward_time": 3.1399004459381104,
      "step": 4492
    },
    {
      "epoch": 1.71356201171875e-05,
      "step": 4492,
      "training_step_time": 10.30066728591919
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 3.184006452560425,
      "step": 4493
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 4493,
      "training_step_time": 9.960845708847046
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 3.1401901245117188,
      "step": 4493
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 4493,
      "training_step_time": 9.911531925201416
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 3.1326708793640137,
      "step": 4493
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 4493,
      "training_step_time": 9.901453733444214
    },
    {
      "epoch": 1.7139434814453126e-05,
      "model_forward_time": 3.423797369003296,
      "step": 4493
    },
    {
      "epoch": 1.7139434814453126e-05,
      "step": 4493,
      "training_step_time": 10.288175582885742
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 3.187473773956299,
      "step": 4494
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 4494,
      "training_step_time": 9.961360216140747
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 3.1401960849761963,
      "step": 4494
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 4494,
      "training_step_time": 9.915628671646118
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 3.1397178173065186,
      "step": 4494
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 4494,
      "training_step_time": 9.916781425476074
    },
    {
      "epoch": 1.714324951171875e-05,
      "model_forward_time": 3.1429672241210938,
      "step": 4494
    },
    {
      "epoch": 1.714324951171875e-05,
      "step": 4494,
      "training_step_time": 10.019479751586914
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 3.4378323554992676,
      "step": 4495
    },
    {
      "dynamics_loss_avg": 0.04487207885831594,
      "epoch": 1.7147064208984375e-05,
      "step": 4495
    },
    {
      "action_loss_avg": 0.022650975920259952,
      "epoch": 1.7147064208984375e-05,
      "step": 4495
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 4495,
      "training_step_time": 10.216172456741333
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 3.1459546089172363,
      "step": 4495
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 4495,
      "training_step_time": 9.92978572845459
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 3.139862537384033,
      "step": 4495
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 4495,
      "training_step_time": 9.918231964111328
    },
    {
      "epoch": 1.7147064208984375e-05,
      "model_forward_time": 3.1443798542022705,
      "step": 4495
    },
    {
      "epoch": 1.7147064208984375e-05,
      "step": 4495,
      "training_step_time": 10.298173666000366
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 3.1854748725891113,
      "step": 4496
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 4496,
      "training_step_time": 9.961530685424805
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 3.435499429702759,
      "step": 4496
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 4496,
      "training_step_time": 10.218271970748901
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 3.1432220935821533,
      "step": 4496
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 4496,
      "training_step_time": 9.925342082977295
    },
    {
      "epoch": 1.715087890625e-05,
      "model_forward_time": 3.1457102298736572,
      "step": 4496
    },
    {
      "epoch": 1.715087890625e-05,
      "step": 4496,
      "training_step_time": 10.286099195480347
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 3.1791884899139404,
      "step": 4497
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 4497,
      "training_step_time": 9.95603322982788
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 3.1463701725006104,
      "step": 4497
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 4497,
      "training_step_time": 9.925746440887451
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 3.4045357704162598,
      "step": 4497
    },
    {
      "dynamics_loss_avg": 0.06881461441516876,
      "epoch": 1.7154693603515625e-05,
      "step": 4497
    },
    {
      "action_loss_avg": 0.030752849020063876,
      "epoch": 1.7154693603515625e-05,
      "step": 4497
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 4497,
      "training_step_time": 10.181877851486206
    },
    {
      "epoch": 1.7154693603515625e-05,
      "model_forward_time": 2.4407756328582764,
      "step": 4497
    },
    {
      "epoch": 1.7154693603515625e-05,
      "step": 4497,
      "training_step_time": 10.320405721664429
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 3.1918888092041016,
      "step": 4498
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 4498,
      "training_step_time": 9.966780424118042
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 3.145301103591919,
      "step": 4498
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 4498,
      "training_step_time": 9.915488958358765
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 3.144686460494995,
      "step": 4498
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 4498,
      "training_step_time": 9.915171146392822
    },
    {
      "epoch": 1.715850830078125e-05,
      "model_forward_time": 3.1381452083587646,
      "step": 4498
    },
    {
      "epoch": 1.715850830078125e-05,
      "step": 4498,
      "training_step_time": 10.273819208145142
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 3.218235731124878,
      "step": 4499
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 4499,
      "training_step_time": 9.995949268341064
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 3.13374662399292,
      "step": 4499
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 4499,
      "training_step_time": 9.905264854431152
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 3.1395981311798096,
      "step": 4499
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 4499,
      "training_step_time": 9.91791558265686
    },
    {
      "epoch": 1.7162322998046874e-05,
      "model_forward_time": 3.136409044265747,
      "step": 4499
    },
    {
      "epoch": 1.7162322998046874e-05,
      "step": 4499,
      "training_step_time": 10.270824432373047
    },
    {
      "epoch": 1.71661376953125e-05,
      "grad_norm": 0.6114079356193542,
      "learning_rate": 1e-05,
      "loss": 0.0889,
      "step": 4500
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 3.152431011199951,
      "step": 4500
    },
    {
      "dynamics_loss_avg": 0.08340183384716511,
      "epoch": 1.71661376953125e-05,
      "step": 4500
    },
    {
      "action_loss_avg": 0.02570588868111372,
      "epoch": 1.71661376953125e-05,
      "step": 4500
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 4500,
      "training_step_time": 9.911858558654785
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 3.1190521717071533,
      "step": 4500
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 4500,
      "training_step_time": 9.874848127365112
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 3.128556251525879,
      "step": 4500
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 4500,
      "training_step_time": 9.883312702178955
    },
    {
      "epoch": 1.71661376953125e-05,
      "model_forward_time": 3.1318466663360596,
      "step": 4500
    },
    {
      "epoch": 1.71661376953125e-05,
      "step": 4500,
      "training_step_time": 10.270768404006958
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 2.45745849609375,
      "step": 4501
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 4501,
      "training_step_time": 7.479737043380737
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 5.30059027671814,
      "step": 4501
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 4501,
      "training_step_time": 12.064242839813232
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 2.4637858867645264,
      "step": 4501
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 4501,
      "training_step_time": 7.495694875717163
    },
    {
      "epoch": 1.7169952392578126e-05,
      "model_forward_time": 3.1292693614959717,
      "step": 4501
    },
    {
      "epoch": 1.7169952392578126e-05,
      "step": 4501,
      "training_step_time": 10.323212623596191
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 2.4719173908233643,
      "step": 4502
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 4502,
      "training_step_time": 7.5027241706848145
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 2.4296135902404785,
      "step": 4502
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 4502,
      "training_step_time": 7.45576810836792
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 3.411344289779663,
      "step": 4502
    },
    {
      "dynamics_loss_avg": 0.052896563149988654,
      "epoch": 1.717376708984375e-05,
      "step": 4502
    },
    {
      "action_loss_avg": 0.026763176592066885,
      "epoch": 1.717376708984375e-05,
      "step": 4502
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 4502,
      "training_step_time": 10.188510417938232
    },
    {
      "epoch": 1.717376708984375e-05,
      "model_forward_time": 3.1338744163513184,
      "step": 4502
    },
    {
      "epoch": 1.717376708984375e-05,
      "step": 4502,
      "training_step_time": 10.302647352218628
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 3.1649343967437744,
      "step": 4503
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 4503,
      "training_step_time": 9.94487452507019
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 3.1353018283843994,
      "step": 4503
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 4503,
      "training_step_time": 9.906018257141113
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 3.139160633087158,
      "step": 4503
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 4503,
      "training_step_time": 9.91041111946106
    },
    {
      "epoch": 1.7177581787109375e-05,
      "model_forward_time": 3.405466079711914,
      "step": 4503
    },
    {
      "epoch": 1.7177581787109375e-05,
      "step": 4503,
      "training_step_time": 10.268972158432007
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 3.1719210147857666,
      "step": 4504
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 4504,
      "training_step_time": 9.945422172546387
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 3.145789623260498,
      "step": 4504
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 4504,
      "training_step_time": 9.91907548904419
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 3.140016555786133,
      "step": 4504
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 4504,
      "training_step_time": 9.913297891616821
    },
    {
      "epoch": 1.7181396484375e-05,
      "model_forward_time": 3.1446971893310547,
      "step": 4504
    },
    {
      "epoch": 1.7181396484375e-05,
      "step": 4504,
      "training_step_time": 10.017385959625244
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 3.4476654529571533,
      "step": 4505
    },
    {
      "dynamics_loss_avg": 0.06916861031204462,
      "epoch": 1.7185211181640625e-05,
      "step": 4505
    },
    {
      "action_loss_avg": 0.03091527707874775,
      "epoch": 1.7185211181640625e-05,
      "step": 4505
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 4505,
      "training_step_time": 10.230411529541016
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 3.1400222778320312,
      "step": 4505
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 4505,
      "training_step_time": 9.91610312461853
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 3.1385366916656494,
      "step": 4505
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 4505,
      "training_step_time": 9.917282342910767
    },
    {
      "epoch": 1.7185211181640625e-05,
      "model_forward_time": 2.433586597442627,
      "step": 4505
    },
    {
      "epoch": 1.7185211181640625e-05,
      "step": 4505,
      "training_step_time": 10.03104543685913
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 3.1559464931488037,
      "step": 4506
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 4506,
      "training_step_time": 9.933917760848999
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 3.42649245262146,
      "step": 4506
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 4506,
      "training_step_time": 10.20427942276001
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 2.436075210571289,
      "step": 4506
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 4506,
      "training_step_time": 7.465593338012695
    },
    {
      "epoch": 1.718902587890625e-05,
      "model_forward_time": 2.431386947631836,
      "step": 4506
    },
    {
      "epoch": 1.718902587890625e-05,
      "step": 4506,
      "training_step_time": 10.262056827545166
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 3.1815297603607178,
      "step": 4507
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 4507,
      "training_step_time": 9.957568645477295
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 3.1382479667663574,
      "step": 4507
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 4507,
      "training_step_time": 9.913460731506348
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 3.3957793712615967,
      "step": 4507
    },
    {
      "dynamics_loss_avg": 0.059320314414799215,
      "epoch": 1.7192840576171874e-05,
      "step": 4507
    },
    {
      "action_loss_avg": 0.029176024021580815,
      "epoch": 1.7192840576171874e-05,
      "step": 4507
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 4507,
      "training_step_time": 10.172975540161133
    },
    {
      "epoch": 1.7192840576171874e-05,
      "model_forward_time": 3.1492037773132324,
      "step": 4507
    },
    {
      "epoch": 1.7192840576171874e-05,
      "step": 4507,
      "training_step_time": 10.346198320388794
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 3.1927542686462402,
      "step": 4508
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 4508,
      "training_step_time": 9.96907925605774
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 3.1451401710510254,
      "step": 4508
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 4508,
      "training_step_time": 9.921079158782959
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 3.147752285003662,
      "step": 4508
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 4508,
      "training_step_time": 9.920196056365967
    },
    {
      "epoch": 1.71966552734375e-05,
      "model_forward_time": 3.419801712036133,
      "step": 4508
    },
    {
      "epoch": 1.71966552734375e-05,
      "step": 4508,
      "training_step_time": 10.294877052307129
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 3.1606388092041016,
      "step": 4509
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 4509,
      "training_step_time": 9.936251401901245
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 3.1438326835632324,
      "step": 4509
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 4509,
      "training_step_time": 9.911131858825684
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 2.439379930496216,
      "step": 4509
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 4509,
      "training_step_time": 7.463437080383301
    },
    {
      "epoch": 1.7200469970703126e-05,
      "model_forward_time": 3.141083240509033,
      "step": 4509
    },
    {
      "epoch": 1.7200469970703126e-05,
      "step": 4509,
      "training_step_time": 10.02023196220398
    },
    {
      "epoch": 1.720428466796875e-05,
      "grad_norm": 0.563660740852356,
      "learning_rate": 1e-05,
      "loss": 0.0921,
      "step": 4510
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 3.4560937881469727,
      "step": 4510
    },
    {
      "dynamics_loss_avg": 0.06627734433859586,
      "epoch": 1.720428466796875e-05,
      "step": 4510
    },
    {
      "action_loss_avg": 0.034824040811508895,
      "epoch": 1.720428466796875e-05,
      "step": 4510
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 4510,
      "training_step_time": 10.228875398635864
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 2.439558982849121,
      "step": 4510
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 4510,
      "training_step_time": 7.470567941665649
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 2.441481113433838,
      "step": 4510
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 4510,
      "training_step_time": 7.474453926086426
    },
    {
      "epoch": 1.720428466796875e-05,
      "model_forward_time": 3.142171621322632,
      "step": 4510
    },
    {
      "epoch": 1.720428466796875e-05,
      "step": 4510,
      "training_step_time": 10.03467607498169
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 3.161315441131592,
      "step": 4511
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 4511,
      "training_step_time": 9.9371178150177
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 3.4043970108032227,
      "step": 4511
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 4511,
      "training_step_time": 10.179395198822021
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 2.441206455230713,
      "step": 4511
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 4511,
      "training_step_time": 7.473544597625732
    },
    {
      "epoch": 1.7208099365234375e-05,
      "model_forward_time": 3.142704486846924,
      "step": 4511
    },
    {
      "epoch": 1.7208099365234375e-05,
      "step": 4511,
      "training_step_time": 10.035334587097168
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 3.1877453327178955,
      "step": 4512
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 4512,
      "training_step_time": 9.965104341506958
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 3.1504533290863037,
      "step": 4512
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 4512,
      "training_step_time": 9.92470932006836
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 3.4280333518981934,
      "step": 4512
    },
    {
      "dynamics_loss_avg": 0.058394729159772396,
      "epoch": 1.72119140625e-05,
      "step": 4512
    },
    {
      "action_loss_avg": 0.03553266199305653,
      "epoch": 1.72119140625e-05,
      "step": 4512
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 4512,
      "training_step_time": 10.200139999389648
    },
    {
      "epoch": 1.72119140625e-05,
      "model_forward_time": 3.1442384719848633,
      "step": 4512
    },
    {
      "epoch": 1.72119140625e-05,
      "step": 4512,
      "training_step_time": 10.322834968566895
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 3.179535388946533,
      "step": 4513
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 4513,
      "training_step_time": 9.961272954940796
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 3.143641471862793,
      "step": 4513
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 4513,
      "training_step_time": 9.910524845123291
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 3.1406655311584473,
      "step": 4513
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 4513,
      "training_step_time": 9.905625104904175
    },
    {
      "epoch": 1.7215728759765625e-05,
      "model_forward_time": 3.4001569747924805,
      "step": 4513
    },
    {
      "epoch": 1.7215728759765625e-05,
      "step": 4513,
      "training_step_time": 10.258601188659668
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 3.187674045562744,
      "step": 4514
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 4514,
      "training_step_time": 9.957812309265137
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 3.143556833267212,
      "step": 4514
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 4514,
      "training_step_time": 9.914664506912231
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 3.144561529159546,
      "step": 4514
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 4514,
      "training_step_time": 9.912071943283081
    },
    {
      "epoch": 1.721954345703125e-05,
      "model_forward_time": 3.148510217666626,
      "step": 4514
    },
    {
      "epoch": 1.721954345703125e-05,
      "step": 4514,
      "training_step_time": 10.017040491104126
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 3.453984022140503,
      "step": 4515
    },
    {
      "dynamics_loss_avg": 0.06394221838563681,
      "epoch": 1.7223358154296874e-05,
      "step": 4515
    },
    {
      "action_loss_avg": 0.03603840013965964,
      "epoch": 1.7223358154296874e-05,
      "step": 4515
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 4515,
      "training_step_time": 10.227281332015991
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 3.1448378562927246,
      "step": 4515
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 4515,
      "training_step_time": 9.920613765716553
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 3.139634132385254,
      "step": 4515
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 4515,
      "training_step_time": 9.909605979919434
    },
    {
      "epoch": 1.7223358154296874e-05,
      "model_forward_time": 3.1449830532073975,
      "step": 4515
    },
    {
      "epoch": 1.7223358154296874e-05,
      "step": 4515,
      "training_step_time": 10.049424886703491
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 3.1939761638641357,
      "step": 4516
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 4516,
      "training_step_time": 9.976243734359741
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 3.4302046298980713,
      "step": 4516
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 4516,
      "training_step_time": 10.207120656967163
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 3.1430296897888184,
      "step": 4516
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 4516,
      "training_step_time": 9.916508674621582
    },
    {
      "epoch": 1.72271728515625e-05,
      "model_forward_time": 3.1455769538879395,
      "step": 4516
    },
    {
      "epoch": 1.72271728515625e-05,
      "step": 4516,
      "training_step_time": 10.302124261856079
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 3.1870386600494385,
      "step": 4517
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 4517,
      "training_step_time": 9.959270000457764
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 3.446256160736084,
      "step": 4517
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 4517,
      "training_step_time": 10.2161386013031
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 3.144043207168579,
      "step": 4517
    },
    {
      "dynamics_loss_avg": 0.0659957081079483,
      "epoch": 1.7230987548828126e-05,
      "step": 4517
    },
    {
      "action_loss_avg": 0.03688128348439932,
      "epoch": 1.7230987548828126e-05,
      "step": 4517
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 4517,
      "training_step_time": 9.923694372177124
    },
    {
      "epoch": 1.7230987548828126e-05,
      "model_forward_time": 3.1484732627868652,
      "step": 4517
    },
    {
      "epoch": 1.7230987548828126e-05,
      "step": 4517,
      "training_step_time": 10.299088954925537
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 2.476219892501831,
      "step": 4518
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 4518,
      "training_step_time": 7.5028297901153564
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 2.435129404067993,
      "step": 4518
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 4518,
      "training_step_time": 7.462689161300659
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 3.1407012939453125,
      "step": 4518
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 4518,
      "training_step_time": 9.912681102752686
    },
    {
      "epoch": 1.723480224609375e-05,
      "model_forward_time": 3.434464693069458,
      "step": 4518
    },
    {
      "epoch": 1.723480224609375e-05,
      "step": 4518,
      "training_step_time": 10.318411111831665
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 3.1715879440307617,
      "step": 4519
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 4519,
      "training_step_time": 9.942027568817139
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 3.1429224014282227,
      "step": 4519
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 4519,
      "training_step_time": 9.913733005523682
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 2.4387505054473877,
      "step": 4519
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 4519,
      "training_step_time": 7.4699790477752686
    },
    {
      "epoch": 1.7238616943359375e-05,
      "model_forward_time": 3.1404151916503906,
      "step": 4519
    },
    {
      "epoch": 1.7238616943359375e-05,
      "step": 4519,
      "training_step_time": 10.041063785552979
    },
    {
      "epoch": 1.7242431640625e-05,
      "grad_norm": 0.6350346207618713,
      "learning_rate": 1e-05,
      "loss": 0.0944,
      "step": 4520
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 3.4807817935943604,
      "step": 4520
    },
    {
      "dynamics_loss_avg": 0.05708009656518698,
      "epoch": 1.7242431640625e-05,
      "step": 4520
    },
    {
      "action_loss_avg": 0.02914476115256548,
      "epoch": 1.7242431640625e-05,
      "step": 4520
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 4520,
      "training_step_time": 10.258824825286865
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 3.1389429569244385,
      "step": 4520
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 4520,
      "training_step_time": 9.913808584213257
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 3.1374495029449463,
      "step": 4520
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 4520,
      "training_step_time": 9.910309314727783
    },
    {
      "epoch": 1.7242431640625e-05,
      "model_forward_time": 2.4391591548919678,
      "step": 4520
    },
    {
      "epoch": 1.7242431640625e-05,
      "step": 4520,
      "training_step_time": 10.252490043640137
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 3.1706151962280273,
      "step": 4521
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 4521,
      "training_step_time": 9.939701318740845
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 3.381331205368042,
      "step": 4521
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 4521,
      "training_step_time": 10.145267724990845
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 3.1369552612304688,
      "step": 4521
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 4521,
      "training_step_time": 9.901882648468018
    },
    {
      "epoch": 1.7246246337890625e-05,
      "model_forward_time": 3.1307637691497803,
      "step": 4521
    },
    {
      "epoch": 1.7246246337890625e-05,
      "step": 4521,
      "training_step_time": 10.021286964416504
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 3.171365261077881,
      "step": 4522
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 4522,
      "training_step_time": 9.933254480361938
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 2.433042049407959,
      "step": 4522
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 4522,
      "training_step_time": 7.457328796386719
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 3.407376527786255,
      "step": 4522
    },
    {
      "dynamics_loss_avg": 0.052293213456869124,
      "epoch": 1.725006103515625e-05,
      "step": 4522
    },
    {
      "action_loss_avg": 0.023115498758852483,
      "epoch": 1.725006103515625e-05,
      "step": 4522
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 4522,
      "training_step_time": 10.17817759513855
    },
    {
      "epoch": 1.725006103515625e-05,
      "model_forward_time": 3.136061668395996,
      "step": 4522
    },
    {
      "epoch": 1.725006103515625e-05,
      "step": 4522,
      "training_step_time": 10.288940668106079
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 3.1706271171569824,
      "step": 4523
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 4523,
      "training_step_time": 9.957992553710938
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 3.142800807952881,
      "step": 4523
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 4523,
      "training_step_time": 9.91562032699585
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 3.137545108795166,
      "step": 4523
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 4523,
      "training_step_time": 9.908112049102783
    },
    {
      "epoch": 1.7253875732421874e-05,
      "model_forward_time": 3.393062114715576,
      "step": 4523
    },
    {
      "epoch": 1.7253875732421874e-05,
      "step": 4523,
      "training_step_time": 10.272976398468018
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 3.187495231628418,
      "step": 4524
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 4524,
      "training_step_time": 9.957107782363892
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 3.1375255584716797,
      "step": 4524
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 4524,
      "training_step_time": 9.906148433685303
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 3.130181312561035,
      "step": 4524
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 4524,
      "training_step_time": 9.902250289916992
    },
    {
      "epoch": 1.72576904296875e-05,
      "model_forward_time": 3.13814115524292,
      "step": 4524
    },
    {
      "epoch": 1.72576904296875e-05,
      "step": 4524,
      "training_step_time": 10.27467656135559
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 3.1434316635131836,
      "step": 4525
    },
    {
      "dynamics_loss_avg": 0.06368138715624809,
      "epoch": 1.7261505126953126e-05,
      "step": 4525
    },
    {
      "action_loss_avg": 0.027241607150062918,
      "epoch": 1.7261505126953126e-05,
      "step": 4525
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 4525,
      "training_step_time": 9.917614936828613
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 3.1352155208587646,
      "step": 4525
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 4525,
      "training_step_time": 9.904218196868896
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 3.1338934898376465,
      "step": 4525
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 4525,
      "training_step_time": 9.90376091003418
    },
    {
      "epoch": 1.7261505126953126e-05,
      "model_forward_time": 2.4280056953430176,
      "step": 4525
    },
    {
      "epoch": 1.7261505126953126e-05,
      "step": 4525,
      "training_step_time": 10.023988485336304
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 3.417487382888794,
      "step": 4526
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 4526,
      "training_step_time": 10.194653987884521
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 3.1382017135620117,
      "step": 4526
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 4526,
      "training_step_time": 9.909291505813599
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 3.1332321166992188,
      "step": 4526
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 4526,
      "training_step_time": 9.904226064682007
    },
    {
      "epoch": 1.726531982421875e-05,
      "model_forward_time": 3.133749008178711,
      "step": 4526
    },
    {
      "epoch": 1.726531982421875e-05,
      "step": 4526,
      "training_step_time": 10.014904260635376
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 3.174316883087158,
      "step": 4527
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 4527,
      "training_step_time": 9.953999280929565
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 3.140496253967285,
      "step": 4527
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 4527,
      "training_step_time": 10.192173957824707
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 3.141658067703247,
      "step": 4527
    },
    {
      "dynamics_loss_avg": 0.06549364142119884,
      "epoch": 1.7269134521484376e-05,
      "step": 4527
    },
    {
      "action_loss_avg": 0.0343223437666893,
      "epoch": 1.7269134521484376e-05,
      "step": 4527
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 4527,
      "training_step_time": 9.914721250534058
    },
    {
      "epoch": 1.7269134521484376e-05,
      "model_forward_time": 3.140669345855713,
      "step": 4527
    },
    {
      "epoch": 1.7269134521484376e-05,
      "step": 4527,
      "training_step_time": 10.004192590713501
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 3.1762986183166504,
      "step": 4528
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 4528,
      "training_step_time": 9.945489883422852
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 3.1314048767089844,
      "step": 4528
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 4528,
      "training_step_time": 9.901353120803833
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 3.4011433124542236,
      "step": 4528
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 4528,
      "training_step_time": 10.170320272445679
    },
    {
      "epoch": 1.727294921875e-05,
      "model_forward_time": 3.1409642696380615,
      "step": 4528
    },
    {
      "epoch": 1.727294921875e-05,
      "step": 4528,
      "training_step_time": 10.288570880889893
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 3.172825336456299,
      "step": 4529
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 4529,
      "training_step_time": 9.943457126617432
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 3.1361911296844482,
      "step": 4529
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 4529,
      "training_step_time": 9.910199165344238
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 3.141723155975342,
      "step": 4529
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 4529,
      "training_step_time": 9.921388626098633
    },
    {
      "epoch": 1.7276763916015625e-05,
      "model_forward_time": 3.4280991554260254,
      "step": 4529
    },
    {
      "epoch": 1.7276763916015625e-05,
      "step": 4529,
      "training_step_time": 10.29939079284668
    },
    {
      "epoch": 1.728057861328125e-05,
      "grad_norm": 0.3629254400730133,
      "learning_rate": 1e-05,
      "loss": 0.0911,
      "step": 4530
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 3.1822853088378906,
      "step": 4530
    },
    {
      "dynamics_loss_avg": 0.05580885503441095,
      "epoch": 1.728057861328125e-05,
      "step": 4530
    },
    {
      "action_loss_avg": 0.028759058564901352,
      "epoch": 1.728057861328125e-05,
      "step": 4530
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 4530,
      "training_step_time": 9.962851524353027
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 3.138869524002075,
      "step": 4530
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 4530,
      "training_step_time": 9.916218996047974
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 3.1436643600463867,
      "step": 4530
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 4530,
      "training_step_time": 9.918838500976562
    },
    {
      "epoch": 1.728057861328125e-05,
      "model_forward_time": 3.1401567459106445,
      "step": 4530
    },
    {
      "epoch": 1.728057861328125e-05,
      "step": 4530,
      "training_step_time": 10.020781517028809
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 3.164931058883667,
      "step": 4531
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 4531,
      "training_step_time": 10.235417366027832
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 3.1404714584350586,
      "step": 4531
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 4531,
      "training_step_time": 9.918903350830078
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 2.4385359287261963,
      "step": 4531
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 4531,
      "training_step_time": 7.474911212921143
    },
    {
      "epoch": 1.7284393310546874e-05,
      "model_forward_time": 3.1392297744750977,
      "step": 4531
    },
    {
      "epoch": 1.7284393310546874e-05,
      "step": 4531,
      "training_step_time": 10.29758906364441
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 3.1866772174835205,
      "step": 4532
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 4532,
      "training_step_time": 9.97119927406311
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 3.1507949829101562,
      "step": 4532
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 4532,
      "training_step_time": 9.93265676498413
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 3.4353175163269043,
      "step": 4532
    },
    {
      "dynamics_loss_avg": 0.0641018271446228,
      "epoch": 1.72882080078125e-05,
      "step": 4532
    },
    {
      "action_loss_avg": 0.030347528308629988,
      "epoch": 1.72882080078125e-05,
      "step": 4532
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 4532,
      "training_step_time": 10.21659255027771
    },
    {
      "epoch": 1.72882080078125e-05,
      "model_forward_time": 3.1420886516571045,
      "step": 4532
    },
    {
      "epoch": 1.72882080078125e-05,
      "step": 4532,
      "training_step_time": 10.289379358291626
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 2.4552435874938965,
      "step": 4533
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 4533,
      "training_step_time": 7.4927589893341064
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 3.1461682319641113,
      "step": 4533
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 4533,
      "training_step_time": 9.92603850364685
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 2.4401321411132812,
      "step": 4533
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 4533,
      "training_step_time": 7.474298715591431
    },
    {
      "epoch": 1.7292022705078126e-05,
      "model_forward_time": 3.4002010822296143,
      "step": 4533
    },
    {
      "epoch": 1.7292022705078126e-05,
      "step": 4533,
      "training_step_time": 10.275822639465332
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 3.1845481395721436,
      "step": 4534
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 4534,
      "training_step_time": 9.961844682693481
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 3.1487908363342285,
      "step": 4534
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 4534,
      "training_step_time": 9.931684017181396
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 3.145573139190674,
      "step": 4534
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 4534,
      "training_step_time": 9.928205490112305
    },
    {
      "epoch": 1.729583740234375e-05,
      "model_forward_time": 3.1511170864105225,
      "step": 4534
    },
    {
      "epoch": 1.729583740234375e-05,
      "step": 4534,
      "training_step_time": 10.305991411209106
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 3.1816585063934326,
      "step": 4535
    },
    {
      "dynamics_loss_avg": 0.06458061579614878,
      "epoch": 1.7299652099609376e-05,
      "step": 4535
    },
    {
      "action_loss_avg": 0.02781418729573488,
      "epoch": 1.7299652099609376e-05,
      "step": 4535
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 4535,
      "training_step_time": 9.964427709579468
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 3.147672176361084,
      "step": 4535
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 4535,
      "training_step_time": 9.92798399925232
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 3.142749309539795,
      "step": 4535
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 4535,
      "training_step_time": 9.920673847198486
    },
    {
      "epoch": 1.7299652099609376e-05,
      "model_forward_time": 3.1436126232147217,
      "step": 4535
    },
    {
      "epoch": 1.7299652099609376e-05,
      "step": 4535,
      "training_step_time": 10.050413846969604
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 3.1795806884765625,
      "step": 4536
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 4536,
      "training_step_time": 10.229368209838867
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 3.1366090774536133,
      "step": 4536
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 4536,
      "training_step_time": 9.911222696304321
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 3.1330618858337402,
      "step": 4536
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 4536,
      "training_step_time": 9.911941051483154
    },
    {
      "epoch": 1.7303466796875e-05,
      "model_forward_time": 3.137157440185547,
      "step": 4536
    },
    {
      "epoch": 1.7303466796875e-05,
      "step": 4536,
      "training_step_time": 10.016971111297607
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 3.1851000785827637,
      "step": 4537
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 4537,
      "training_step_time": 9.970160961151123
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 2.7130846977233887,
      "step": 4537
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 4537,
      "training_step_time": 7.751532793045044
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 3.137903928756714,
      "step": 4537
    },
    {
      "dynamics_loss_avg": 0.05431900396943092,
      "epoch": 1.7307281494140625e-05,
      "step": 4537
    },
    {
      "action_loss_avg": 0.025661608204245567,
      "epoch": 1.7307281494140625e-05,
      "step": 4537
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 4537,
      "training_step_time": 9.921714544296265
    },
    {
      "epoch": 1.7307281494140625e-05,
      "model_forward_time": 3.142064094543457,
      "step": 4537
    },
    {
      "epoch": 1.7307281494140625e-05,
      "step": 4537,
      "training_step_time": 10.031574249267578
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 2.4454424381256104,
      "step": 4538
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 4538,
      "training_step_time": 7.480564117431641
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 3.1352968215942383,
      "step": 4538
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 4538,
      "training_step_time": 9.915034294128418
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 3.141055107116699,
      "step": 4538
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 4538,
      "training_step_time": 9.922497034072876
    },
    {
      "epoch": 1.731109619140625e-05,
      "model_forward_time": 2.7038400173187256,
      "step": 4538
    },
    {
      "epoch": 1.731109619140625e-05,
      "step": 4538,
      "training_step_time": 10.021931648254395
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 3.1611244678497314,
      "step": 4539
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 4539,
      "training_step_time": 9.939702987670898
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 3.139814853668213,
      "step": 4539
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 4539,
      "training_step_time": 9.920722484588623
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 2.4396939277648926,
      "step": 4539
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 4539,
      "training_step_time": 7.476992130279541
    },
    {
      "epoch": 1.7314910888671874e-05,
      "model_forward_time": 3.1389451026916504,
      "step": 4539
    },
    {
      "epoch": 1.7314910888671874e-05,
      "step": 4539,
      "training_step_time": 10.344496011734009
    },
    {
      "epoch": 1.73187255859375e-05,
      "grad_norm": 0.8000726103782654,
      "learning_rate": 1e-05,
      "loss": 0.092,
      "step": 4540
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 2.7778964042663574,
      "step": 4540
    },
    {
      "dynamics_loss_avg": 0.0769137505441904,
      "epoch": 1.73187255859375e-05,
      "step": 4540
    },
    {
      "action_loss_avg": 0.03393358048051596,
      "epoch": 1.73187255859375e-05,
      "step": 4540
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 4540,
      "training_step_time": 7.817063331604004
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 3.133754014968872,
      "step": 4540
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 4540,
      "training_step_time": 9.916857957839966
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 3.139939785003662,
      "step": 4540
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 4540,
      "training_step_time": 9.922611236572266
    },
    {
      "epoch": 1.73187255859375e-05,
      "model_forward_time": 3.139068126678467,
      "step": 4540
    },
    {
      "epoch": 1.73187255859375e-05,
      "step": 4540,
      "training_step_time": 10.319969177246094
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 3.1941025257110596,
      "step": 4541
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 4541,
      "training_step_time": 9.97557020187378
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 3.374720573425293,
      "step": 4541
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 4541,
      "training_step_time": 10.158201694488525
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 3.1405956745147705,
      "step": 4541
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 4541,
      "training_step_time": 9.921375751495361
    },
    {
      "epoch": 1.7322540283203126e-05,
      "model_forward_time": 3.1321489810943604,
      "step": 4541
    },
    {
      "epoch": 1.7322540283203126e-05,
      "step": 4541,
      "training_step_time": 10.00260329246521
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 3.1718499660491943,
      "step": 4542
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 4542,
      "training_step_time": 9.950406312942505
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 3.1383042335510254,
      "step": 4542
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 4542,
      "training_step_time": 10.188188314437866
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 3.136423110961914,
      "step": 4542
    },
    {
      "dynamics_loss_avg": 0.053671720810234545,
      "epoch": 1.732635498046875e-05,
      "step": 4542
    },
    {
      "action_loss_avg": 0.03213241677731275,
      "epoch": 1.732635498046875e-05,
      "step": 4542
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 4542,
      "training_step_time": 9.91602897644043
    },
    {
      "epoch": 1.732635498046875e-05,
      "model_forward_time": 3.1358213424682617,
      "step": 4542
    },
    {
      "epoch": 1.732635498046875e-05,
      "step": 4542,
      "training_step_time": 10.019457817077637
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 3.1743521690368652,
      "step": 4543
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 4543,
      "training_step_time": 9.953607082366943
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 3.133761405944824,
      "step": 4543
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 4543,
      "training_step_time": 9.915710687637329
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 3.4128148555755615,
      "step": 4543
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 4543,
      "training_step_time": 10.19066071510315
    },
    {
      "epoch": 1.7330169677734376e-05,
      "model_forward_time": 3.1331875324249268,
      "step": 4543
    },
    {
      "epoch": 1.7330169677734376e-05,
      "step": 4543,
      "training_step_time": 10.008784294128418
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 3.182237148284912,
      "step": 4544
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 4544,
      "training_step_time": 9.960771322250366
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 2.4404139518737793,
      "step": 4544
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 4544,
      "training_step_time": 7.47354531288147
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 3.127681255340576,
      "step": 4544
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 4544,
      "training_step_time": 9.900496006011963
    },
    {
      "epoch": 1.7333984375e-05,
      "model_forward_time": 3.1398212909698486,
      "step": 4544
    },
    {
      "epoch": 1.7333984375e-05,
      "step": 4544,
      "training_step_time": 10.283587217330933
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 3.459559917449951,
      "step": 4545
    },
    {
      "dynamics_loss_avg": 0.0499456986784935,
      "epoch": 1.7337799072265625e-05,
      "step": 4545
    },
    {
      "action_loss_avg": 0.024361908575519918,
      "epoch": 1.7337799072265625e-05,
      "step": 4545
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 4545,
      "training_step_time": 10.243110179901123
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 3.138036012649536,
      "step": 4545
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 4545,
      "training_step_time": 9.921882629394531
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 3.1370253562927246,
      "step": 4545
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 4545,
      "training_step_time": 9.92235279083252
    },
    {
      "epoch": 1.7337799072265625e-05,
      "model_forward_time": 3.145510196685791,
      "step": 4545
    },
    {
      "epoch": 1.7337799072265625e-05,
      "step": 4545,
      "training_step_time": 10.30339527130127
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 3.463871717453003,
      "step": 4546
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 4546,
      "training_step_time": 10.240114688873291
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 3.140268325805664,
      "step": 4546
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 4546,
      "training_step_time": 9.922112464904785
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 3.132288932800293,
      "step": 4546
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 4546,
      "training_step_time": 9.909316062927246
    },
    {
      "epoch": 1.734161376953125e-05,
      "model_forward_time": 3.1416015625,
      "step": 4546
    },
    {
      "epoch": 1.734161376953125e-05,
      "step": 4546,
      "training_step_time": 10.241008281707764
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 3.172124147415161,
      "step": 4547
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 4547,
      "training_step_time": 9.948846578598022
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 3.4195172786712646,
      "step": 4547
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 4547,
      "training_step_time": 10.194030523300171
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 3.132159948348999,
      "step": 4547
    },
    {
      "dynamics_loss_avg": 0.05990313086658716,
      "epoch": 1.7345428466796874e-05,
      "step": 4547
    },
    {
      "action_loss_avg": 0.025752392038702963,
      "epoch": 1.7345428466796874e-05,
      "step": 4547
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 4547,
      "training_step_time": 9.909578561782837
    },
    {
      "epoch": 1.7345428466796874e-05,
      "model_forward_time": 3.135955810546875,
      "step": 4547
    },
    {
      "epoch": 1.7345428466796874e-05,
      "step": 4547,
      "training_step_time": 10.028031349182129
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 3.1821203231811523,
      "step": 4548
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 4548,
      "training_step_time": 9.95849084854126
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 3.137141704559326,
      "step": 4548
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 4548,
      "training_step_time": 9.9124915599823
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 3.3980658054351807,
      "step": 4548
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 4548,
      "training_step_time": 10.173166513442993
    },
    {
      "epoch": 1.73492431640625e-05,
      "model_forward_time": 3.1348330974578857,
      "step": 4548
    },
    {
      "epoch": 1.73492431640625e-05,
      "step": 4548,
      "training_step_time": 10.012963056564331
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 3.1825177669525146,
      "step": 4549
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 4549,
      "training_step_time": 9.95633316040039
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 3.1335954666137695,
      "step": 4549
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 4549,
      "training_step_time": 9.906879901885986
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 2.4286603927612305,
      "step": 4549
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 4549,
      "training_step_time": 7.456875324249268
    },
    {
      "epoch": 1.7353057861328126e-05,
      "model_forward_time": 3.1256370544433594,
      "step": 4549
    },
    {
      "epoch": 1.7353057861328126e-05,
      "step": 4549,
      "training_step_time": 10.289123773574829
    },
    {
      "epoch": 1.735687255859375e-05,
      "grad_norm": 0.37188342213630676,
      "learning_rate": 1e-05,
      "loss": 0.0888,
      "step": 4550
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 3.169009208679199,
      "step": 4550
    },
    {
      "dynamics_loss_avg": 0.07064520381391048,
      "epoch": 1.735687255859375e-05,
      "step": 4550
    },
    {
      "action_loss_avg": 0.034657100681215525,
      "epoch": 1.735687255859375e-05,
      "step": 4550
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 4550,
      "training_step_time": 9.943397283554077
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 3.1306064128875732,
      "step": 4550
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 4550,
      "training_step_time": 9.90010929107666
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 3.1333775520324707,
      "step": 4550
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 4550,
      "training_step_time": 9.904005765914917
    },
    {
      "epoch": 1.735687255859375e-05,
      "model_forward_time": 2.4263522624969482,
      "step": 4550
    },
    {
      "epoch": 1.735687255859375e-05,
      "step": 4550,
      "training_step_time": 10.010430574417114
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 2.4577248096466064,
      "step": 4551
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 4551,
      "training_step_time": 7.48506760597229
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 3.3764803409576416,
      "step": 4551
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 4551,
      "training_step_time": 10.152272939682007
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 3.1380841732025146,
      "step": 4551
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 4551,
      "training_step_time": 9.913421154022217
    },
    {
      "epoch": 1.7360687255859376e-05,
      "model_forward_time": 3.136788845062256,
      "step": 4551
    },
    {
      "epoch": 1.7360687255859376e-05,
      "step": 4551,
      "training_step_time": 10.341835737228394
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 3.1693198680877686,
      "step": 4552
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 4552,
      "training_step_time": 9.949955701828003
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 3.4017245769500732,
      "step": 4552
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 4552,
      "training_step_time": 10.180538177490234
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 3.1351065635681152,
      "step": 4552
    },
    {
      "dynamics_loss_avg": 0.05716892937198281,
      "epoch": 1.7364501953125e-05,
      "step": 4552
    },
    {
      "action_loss_avg": 0.02845978965051472,
      "epoch": 1.7364501953125e-05,
      "step": 4552
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 4552,
      "training_step_time": 9.918064594268799
    },
    {
      "epoch": 1.7364501953125e-05,
      "model_forward_time": 3.1365323066711426,
      "step": 4552
    },
    {
      "epoch": 1.7364501953125e-05,
      "step": 4552,
      "training_step_time": 10.275669813156128
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 3.177492380142212,
      "step": 4553
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 4553,
      "training_step_time": 9.955138683319092
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 3.135568380355835,
      "step": 4553
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 4553,
      "training_step_time": 9.909727334976196
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 3.135474443435669,
      "step": 4553
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 4553,
      "training_step_time": 10.186767816543579
    },
    {
      "epoch": 1.7368316650390625e-05,
      "model_forward_time": 3.135254144668579,
      "step": 4553
    },
    {
      "epoch": 1.7368316650390625e-05,
      "step": 4553,
      "training_step_time": 10.006146430969238
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 2.4507107734680176,
      "step": 4554
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 4554,
      "training_step_time": 7.483479022979736
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 3.136713743209839,
      "step": 4554
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 4554,
      "training_step_time": 9.913377523422241
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 2.4332473278045654,
      "step": 4554
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 4554,
      "training_step_time": 7.463720083236694
    },
    {
      "epoch": 1.737213134765625e-05,
      "model_forward_time": 3.136425018310547,
      "step": 4554
    },
    {
      "epoch": 1.737213134765625e-05,
      "step": 4554,
      "training_step_time": 10.287843704223633
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 2.7127010822296143,
      "step": 4555
    },
    {
      "dynamics_loss_avg": 0.050938315037637946,
      "epoch": 1.7375946044921874e-05,
      "step": 4555
    },
    {
      "action_loss_avg": 0.024037098768167196,
      "epoch": 1.7375946044921874e-05,
      "step": 4555
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 4555,
      "training_step_time": 7.748657703399658
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 3.137382984161377,
      "step": 4555
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 4555,
      "training_step_time": 9.916360855102539
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 2.4360010623931885,
      "step": 4555
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 4555,
      "training_step_time": 7.469295501708984
    },
    {
      "epoch": 1.7375946044921874e-05,
      "model_forward_time": 3.1347310543060303,
      "step": 4555
    },
    {
      "epoch": 1.7375946044921874e-05,
      "step": 4555,
      "training_step_time": 10.024604082107544
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 3.1861884593963623,
      "step": 4556
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 4556,
      "training_step_time": 9.964773893356323
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 3.41322922706604,
      "step": 4556
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 4556,
      "training_step_time": 10.183392524719238
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 3.1309337615966797,
      "step": 4556
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 4556,
      "training_step_time": 9.903115510940552
    },
    {
      "epoch": 1.73797607421875e-05,
      "model_forward_time": 3.13288950920105,
      "step": 4556
    },
    {
      "epoch": 1.73797607421875e-05,
      "step": 4556,
      "training_step_time": 10.271530628204346
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 3.190074920654297,
      "step": 4557
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 4557,
      "training_step_time": 9.96353530883789
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 3.134521722793579,
      "step": 4557
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 4557,
      "training_step_time": 9.9121253490448
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 2.7326951026916504,
      "step": 4557
    },
    {
      "dynamics_loss_avg": 0.0597821407020092,
      "epoch": 1.7383575439453126e-05,
      "step": 4557
    },
    {
      "action_loss_avg": 0.030051288846880197,
      "epoch": 1.7383575439453126e-05,
      "step": 4557
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 4557,
      "training_step_time": 7.76756477355957
    },
    {
      "epoch": 1.7383575439453126e-05,
      "model_forward_time": 3.1396989822387695,
      "step": 4557
    },
    {
      "epoch": 1.7383575439453126e-05,
      "step": 4557,
      "training_step_time": 10.298107624053955
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 3.176335334777832,
      "step": 4558
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 4558,
      "training_step_time": 9.963874578475952
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 3.1463818550109863,
      "step": 4558
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 4558,
      "training_step_time": 9.928554773330688
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 3.147484302520752,
      "step": 4558
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 4558,
      "training_step_time": 9.923426866531372
    },
    {
      "epoch": 1.738739013671875e-05,
      "model_forward_time": 3.4216225147247314,
      "step": 4558
    },
    {
      "epoch": 1.738739013671875e-05,
      "step": 4558,
      "training_step_time": 10.294229745864868
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 2.466611623764038,
      "step": 4559
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 4559,
      "training_step_time": 7.502626895904541
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 3.133420944213867,
      "step": 4559
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 4559,
      "training_step_time": 9.91388487815857
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 3.1402032375335693,
      "step": 4559
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 4559,
      "training_step_time": 9.923354864120483
    },
    {
      "epoch": 1.7391204833984376e-05,
      "model_forward_time": 3.1430487632751465,
      "step": 4559
    },
    {
      "epoch": 1.7391204833984376e-05,
      "step": 4559,
      "training_step_time": 10.321758031845093
    },
    {
      "epoch": 1.739501953125e-05,
      "grad_norm": 0.5557261109352112,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 4560
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 3.4818460941314697,
      "step": 4560
    },
    {
      "dynamics_loss_avg": 0.06922087389975787,
      "epoch": 1.739501953125e-05,
      "step": 4560
    },
    {
      "action_loss_avg": 0.040620229486376044,
      "epoch": 1.739501953125e-05,
      "step": 4560
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 4560,
      "training_step_time": 10.264936447143555
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 3.1382088661193848,
      "step": 4560
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 4560,
      "training_step_time": 9.91928219795227
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 2.429839611053467,
      "step": 4560
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 4560,
      "training_step_time": 7.4643449783325195
    },
    {
      "epoch": 1.739501953125e-05,
      "model_forward_time": 3.132798910140991,
      "step": 4560
    },
    {
      "epoch": 1.739501953125e-05,
      "step": 4560,
      "training_step_time": 10.020702362060547
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 3.170302152633667,
      "step": 4561
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 4561,
      "training_step_time": 9.953054666519165
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 3.377873182296753,
      "step": 4561
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 4561,
      "training_step_time": 10.155930042266846
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 2.4357728958129883,
      "step": 4561
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 4561,
      "training_step_time": 7.471769571304321
    },
    {
      "epoch": 1.7398834228515625e-05,
      "model_forward_time": 3.1335246562957764,
      "step": 4561
    },
    {
      "epoch": 1.7398834228515625e-05,
      "step": 4561,
      "training_step_time": 10.24517011642456
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 3.180746078491211,
      "step": 4562
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 4562,
      "training_step_time": 9.985498428344727
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 3.1393702030181885,
      "step": 4562
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 4562,
      "training_step_time": 9.926996946334839
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 3.4231302738189697,
      "step": 4562
    },
    {
      "dynamics_loss_avg": 0.06781824324280024,
      "epoch": 1.740264892578125e-05,
      "step": 4562
    },
    {
      "action_loss_avg": 0.04477890424896032,
      "epoch": 1.740264892578125e-05,
      "step": 4562
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 4562,
      "training_step_time": 10.21111011505127
    },
    {
      "epoch": 1.740264892578125e-05,
      "model_forward_time": 3.1424992084503174,
      "step": 4562
    },
    {
      "epoch": 1.740264892578125e-05,
      "step": 4562,
      "training_step_time": 10.321028232574463
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 3.183600902557373,
      "step": 4563
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 4563,
      "training_step_time": 9.964220523834229
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 3.138219118118286,
      "step": 4563
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 4563,
      "training_step_time": 9.916805505752563
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 3.136312246322632,
      "step": 4563
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 4563,
      "training_step_time": 9.91623592376709
    },
    {
      "epoch": 1.7406463623046874e-05,
      "model_forward_time": 2.67389178276062,
      "step": 4563
    },
    {
      "epoch": 1.7406463623046874e-05,
      "step": 4563,
      "training_step_time": 10.03080677986145
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 2.4535419940948486,
      "step": 4564
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 4564,
      "training_step_time": 7.487573623657227
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 3.133948564529419,
      "step": 4564
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 4564,
      "training_step_time": 9.908793210983276
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 3.137416124343872,
      "step": 4564
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 4564,
      "training_step_time": 9.916639804840088
    },
    {
      "epoch": 1.74102783203125e-05,
      "model_forward_time": 2.437880754470825,
      "step": 4564
    },
    {
      "epoch": 1.74102783203125e-05,
      "step": 4564,
      "training_step_time": 10.324404001235962
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 3.434410333633423,
      "step": 4565
    },
    {
      "dynamics_loss_avg": 0.051128761097788814,
      "epoch": 1.7414093017578126e-05,
      "step": 4565
    },
    {
      "action_loss_avg": 0.02523480337113142,
      "epoch": 1.7414093017578126e-05,
      "step": 4565
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 4565,
      "training_step_time": 10.213714599609375
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 3.142502784729004,
      "step": 4565
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 4565,
      "training_step_time": 9.930214405059814
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 3.14040207862854,
      "step": 4565
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 4565,
      "training_step_time": 9.922723531723022
    },
    {
      "epoch": 1.7414093017578126e-05,
      "model_forward_time": 2.4351205825805664,
      "step": 4565
    },
    {
      "epoch": 1.7414093017578126e-05,
      "step": 4565,
      "training_step_time": 10.293771266937256
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 3.1777663230895996,
      "step": 4566
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 4566,
      "training_step_time": 9.95823884010315
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 3.411731243133545,
      "step": 4566
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 4566,
      "training_step_time": 10.190666675567627
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 3.140995502471924,
      "step": 4566
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 4566,
      "training_step_time": 9.91763162612915
    },
    {
      "epoch": 1.741790771484375e-05,
      "model_forward_time": 3.1386585235595703,
      "step": 4566
    },
    {
      "epoch": 1.741790771484375e-05,
      "step": 4566,
      "training_step_time": 10.012219905853271
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 3.1848034858703613,
      "step": 4567
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 4567,
      "training_step_time": 9.960546016693115
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 3.1341121196746826,
      "step": 4567
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 4567,
      "training_step_time": 9.909007549285889
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 3.412203550338745,
      "step": 4567
    },
    {
      "dynamics_loss_avg": 0.06690155193209649,
      "epoch": 1.7421722412109376e-05,
      "step": 4567
    },
    {
      "action_loss_avg": 0.03342286925762892,
      "epoch": 1.7421722412109376e-05,
      "step": 4567
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 4567,
      "training_step_time": 10.191052198410034
    },
    {
      "epoch": 1.7421722412109376e-05,
      "model_forward_time": 2.4328792095184326,
      "step": 4567
    },
    {
      "epoch": 1.7421722412109376e-05,
      "step": 4567,
      "training_step_time": 10.30435299873352
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 2.4664947986602783,
      "step": 4568
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 4568,
      "training_step_time": 7.499765396118164
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 3.1366360187530518,
      "step": 4568
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 4568,
      "training_step_time": 9.91968059539795
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 3.139923572540283,
      "step": 4568
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 4568,
      "training_step_time": 9.928220510482788
    },
    {
      "epoch": 1.7425537109375e-05,
      "model_forward_time": 3.425004243850708,
      "step": 4568
    },
    {
      "epoch": 1.7425537109375e-05,
      "step": 4568,
      "training_step_time": 10.301965713500977
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 3.183154582977295,
      "step": 4569
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 4569,
      "training_step_time": 9.97121000289917
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 2.4376535415649414,
      "step": 4569
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 4569,
      "training_step_time": 7.4774088859558105
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 3.1462650299072266,
      "step": 4569
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 4569,
      "training_step_time": 9.930439233779907
    },
    {
      "epoch": 1.7429351806640625e-05,
      "model_forward_time": 3.1377434730529785,
      "step": 4569
    },
    {
      "epoch": 1.7429351806640625e-05,
      "step": 4569,
      "training_step_time": 10.306790828704834
    },
    {
      "epoch": 1.743316650390625e-05,
      "grad_norm": 0.8334054350852966,
      "learning_rate": 1e-05,
      "loss": 0.0883,
      "step": 4570
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 3.4607391357421875,
      "step": 4570
    },
    {
      "dynamics_loss_avg": 0.06762142973020673,
      "epoch": 1.743316650390625e-05,
      "step": 4570
    },
    {
      "action_loss_avg": 0.0367150311358273,
      "epoch": 1.743316650390625e-05,
      "step": 4570
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 4570,
      "training_step_time": 10.244465827941895
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 3.137329578399658,
      "step": 4570
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 4570,
      "training_step_time": 9.917741775512695
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 3.1377031803131104,
      "step": 4570
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 4570,
      "training_step_time": 9.916927099227905
    },
    {
      "epoch": 1.743316650390625e-05,
      "model_forward_time": 3.1381633281707764,
      "step": 4570
    },
    {
      "epoch": 1.743316650390625e-05,
      "step": 4570,
      "training_step_time": 10.338746309280396
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 3.1652257442474365,
      "step": 4571
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 4571,
      "training_step_time": 10.243041276931763
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 3.131711721420288,
      "step": 4571
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 4571,
      "training_step_time": 9.90916919708252
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 2.4283909797668457,
      "step": 4571
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 4571,
      "training_step_time": 7.457959175109863
    },
    {
      "epoch": 1.7436981201171874e-05,
      "model_forward_time": 3.1283187866210938,
      "step": 4571
    },
    {
      "epoch": 1.7436981201171874e-05,
      "step": 4571,
      "training_step_time": 10.00004243850708
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 3.1813151836395264,
      "step": 4572
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 4572,
      "training_step_time": 9.952338218688965
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 3.1383044719696045,
      "step": 4572
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 4572,
      "training_step_time": 9.912368535995483
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 3.398458957672119,
      "step": 4572
    },
    {
      "dynamics_loss_avg": 0.05720656868070364,
      "epoch": 1.74407958984375e-05,
      "step": 4572
    },
    {
      "action_loss_avg": 0.023767945310100912,
      "epoch": 1.74407958984375e-05,
      "step": 4572
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 4572,
      "training_step_time": 10.176270246505737
    },
    {
      "epoch": 1.74407958984375e-05,
      "model_forward_time": 3.133910655975342,
      "step": 4572
    },
    {
      "epoch": 1.74407958984375e-05,
      "step": 4572,
      "training_step_time": 10.289466142654419
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 3.164466619491577,
      "step": 4573
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 4573,
      "training_step_time": 9.942067384719849
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 2.4325506687164307,
      "step": 4573
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 4573,
      "training_step_time": 7.460797548294067
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 3.1316564083099365,
      "step": 4573
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 4573,
      "training_step_time": 9.906874656677246
    },
    {
      "epoch": 1.7444610595703127e-05,
      "model_forward_time": 3.377917766571045,
      "step": 4573
    },
    {
      "epoch": 1.7444610595703127e-05,
      "step": 4573,
      "training_step_time": 10.288031816482544
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 3.159397602081299,
      "step": 4574
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 4574,
      "training_step_time": 9.935418844223022
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 3.139302968978882,
      "step": 4574
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 4574,
      "training_step_time": 9.916559934616089
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 3.1398584842681885,
      "step": 4574
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 4574,
      "training_step_time": 9.919222593307495
    },
    {
      "epoch": 1.744842529296875e-05,
      "model_forward_time": 3.393657684326172,
      "step": 4574
    },
    {
      "epoch": 1.744842529296875e-05,
      "step": 4574,
      "training_step_time": 10.30851149559021
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 3.1724281311035156,
      "step": 4575
    },
    {
      "dynamics_loss_avg": 0.06657019266858696,
      "epoch": 1.7452239990234376e-05,
      "step": 4575
    },
    {
      "action_loss_avg": 0.03089686306193471,
      "epoch": 1.7452239990234376e-05,
      "step": 4575
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 4575,
      "training_step_time": 9.952851295471191
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 2.4339873790740967,
      "step": 4575
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 4575,
      "training_step_time": 7.486455917358398
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 3.1308352947235107,
      "step": 4575
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 4575,
      "training_step_time": 9.898738861083984
    },
    {
      "epoch": 1.7452239990234376e-05,
      "model_forward_time": 3.134047269821167,
      "step": 4575
    },
    {
      "epoch": 1.7452239990234376e-05,
      "step": 4575,
      "training_step_time": 10.009992837905884
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 3.1666653156280518,
      "step": 4576
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 4576,
      "training_step_time": 9.94071650505066
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 3.4498424530029297,
      "step": 4576
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 4576,
      "training_step_time": 10.22409725189209
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 3.1415297985076904,
      "step": 4576
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 4576,
      "training_step_time": 9.919535398483276
    },
    {
      "epoch": 1.74560546875e-05,
      "model_forward_time": 3.1378486156463623,
      "step": 4576
    },
    {
      "epoch": 1.74560546875e-05,
      "step": 4576,
      "training_step_time": 10.284870624542236
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 3.209423303604126,
      "step": 4577
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 4577,
      "training_step_time": 9.984283924102783
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 2.7372193336486816,
      "step": 4577
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 4577,
      "training_step_time": 7.767013788223267
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 3.137922763824463,
      "step": 4577
    },
    {
      "dynamics_loss_avg": 0.0519258476793766,
      "epoch": 1.7459869384765625e-05,
      "step": 4577
    },
    {
      "action_loss_avg": 0.03506125472486019,
      "epoch": 1.7459869384765625e-05,
      "step": 4577
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 4577,
      "training_step_time": 9.91425633430481
    },
    {
      "epoch": 1.7459869384765625e-05,
      "model_forward_time": 3.1410880088806152,
      "step": 4577
    },
    {
      "epoch": 1.7459869384765625e-05,
      "step": 4577,
      "training_step_time": 10.26857304573059
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 3.1735827922821045,
      "step": 4578
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 4578,
      "training_step_time": 9.947569131851196
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 3.1349802017211914,
      "step": 4578
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 4578,
      "training_step_time": 9.907890796661377
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 3.1429460048675537,
      "step": 4578
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 4578,
      "training_step_time": 10.177535772323608
    },
    {
      "epoch": 1.746368408203125e-05,
      "model_forward_time": 2.43630051612854,
      "step": 4578
    },
    {
      "epoch": 1.746368408203125e-05,
      "step": 4578,
      "training_step_time": 10.298343658447266
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 3.188347101211548,
      "step": 4579
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 4579,
      "training_step_time": 9.964340448379517
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 2.4339756965637207,
      "step": 4579
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 4579,
      "training_step_time": 7.4706196784973145
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 3.140251874923706,
      "step": 4579
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 4579,
      "training_step_time": 9.917582988739014
    },
    {
      "epoch": 1.7467498779296874e-05,
      "model_forward_time": 3.1385152339935303,
      "step": 4579
    },
    {
      "epoch": 1.7467498779296874e-05,
      "step": 4579,
      "training_step_time": 10.293316841125488
    },
    {
      "epoch": 1.74713134765625e-05,
      "grad_norm": 0.3223266899585724,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 4580
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 2.4862701892852783,
      "step": 4580
    },
    {
      "dynamics_loss_avg": 0.04375738617964089,
      "epoch": 1.74713134765625e-05,
      "step": 4580
    },
    {
      "action_loss_avg": 0.021039073867723346,
      "epoch": 1.74713134765625e-05,
      "step": 4580
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 4580,
      "training_step_time": 7.525323152542114
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 3.1372950077056885,
      "step": 4580
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 4580,
      "training_step_time": 9.915253639221191
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 3.1414713859558105,
      "step": 4580
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 4580,
      "training_step_time": 9.919458866119385
    },
    {
      "epoch": 1.74713134765625e-05,
      "model_forward_time": 3.1393330097198486,
      "step": 4580
    },
    {
      "epoch": 1.74713134765625e-05,
      "step": 4580,
      "training_step_time": 10.026471614837646
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 3.1695139408111572,
      "step": 4581
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 4581,
      "training_step_time": 9.954218864440918
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 3.397538185119629,
      "step": 4581
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 4581,
      "training_step_time": 10.173980951309204
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 3.140774965286255,
      "step": 4581
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 4581,
      "training_step_time": 9.913966417312622
    },
    {
      "epoch": 1.7475128173828127e-05,
      "model_forward_time": 3.135812997817993,
      "step": 4581
    },
    {
      "epoch": 1.7475128173828127e-05,
      "step": 4581,
      "training_step_time": 10.279037952423096
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 3.176161527633667,
      "step": 4582
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 4582,
      "training_step_time": 9.956153869628906
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 2.436858892440796,
      "step": 4582
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 4582,
      "training_step_time": 7.475458860397339
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 3.4187402725219727,
      "step": 4582
    },
    {
      "dynamics_loss_avg": 0.04245799910277128,
      "epoch": 1.747894287109375e-05,
      "step": 4582
    },
    {
      "action_loss_avg": 0.019499551970511675,
      "epoch": 1.747894287109375e-05,
      "step": 4582
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 4582,
      "training_step_time": 10.201545476913452
    },
    {
      "epoch": 1.747894287109375e-05,
      "model_forward_time": 3.1415441036224365,
      "step": 4582
    },
    {
      "epoch": 1.747894287109375e-05,
      "step": 4582,
      "training_step_time": 10.312710523605347
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 3.1778013706207275,
      "step": 4583
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 4583,
      "training_step_time": 9.957097291946411
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 3.140432596206665,
      "step": 4583
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 4583,
      "training_step_time": 9.91702389717102
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 3.445096254348755,
      "step": 4583
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 4583,
      "training_step_time": 10.2181875705719
    },
    {
      "epoch": 1.7482757568359376e-05,
      "model_forward_time": 2.4394967555999756,
      "step": 4583
    },
    {
      "epoch": 1.7482757568359376e-05,
      "step": 4583,
      "training_step_time": 10.021468877792358
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 3.1656484603881836,
      "step": 4584
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 4584,
      "training_step_time": 9.93679690361023
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 3.136777400970459,
      "step": 4584
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 4584,
      "training_step_time": 9.90531849861145
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 3.1357316970825195,
      "step": 4584
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 4584,
      "training_step_time": 9.901021957397461
    },
    {
      "epoch": 1.7486572265625e-05,
      "model_forward_time": 3.141984701156616,
      "step": 4584
    },
    {
      "epoch": 1.7486572265625e-05,
      "step": 4584,
      "training_step_time": 10.278411626815796
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 3.485567569732666,
      "step": 4585
    },
    {
      "dynamics_loss_avg": 0.06221917048096657,
      "epoch": 1.7490386962890625e-05,
      "step": 4585
    },
    {
      "action_loss_avg": 0.029316330421715976,
      "epoch": 1.7490386962890625e-05,
      "step": 4585
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 4585,
      "training_step_time": 10.259242534637451
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 3.136714458465576,
      "step": 4585
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 4585,
      "training_step_time": 9.90482473373413
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 3.1296801567077637,
      "step": 4585
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 4585,
      "training_step_time": 9.89903473854065
    },
    {
      "epoch": 1.7490386962890625e-05,
      "model_forward_time": 3.133631944656372,
      "step": 4585
    },
    {
      "epoch": 1.7490386962890625e-05,
      "step": 4585,
      "training_step_time": 10.293915033340454
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 2.773494243621826,
      "step": 4586
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 4586,
      "training_step_time": 7.798161745071411
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 2.4314889907836914,
      "step": 4586
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 4586,
      "training_step_time": 7.457984685897827
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 3.132333278656006,
      "step": 4586
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 4586,
      "training_step_time": 9.895635604858398
    },
    {
      "epoch": 1.749420166015625e-05,
      "model_forward_time": 3.1373536586761475,
      "step": 4586
    },
    {
      "epoch": 1.749420166015625e-05,
      "step": 4586,
      "training_step_time": 10.011160850524902
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 3.1486313343048096,
      "step": 4587
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 4587,
      "training_step_time": 9.917470216751099
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 3.136033058166504,
      "step": 4587
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 4587,
      "training_step_time": 9.899989366531372
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 3.3863039016723633,
      "step": 4587
    },
    {
      "dynamics_loss_avg": 0.07678866796195508,
      "epoch": 1.7498016357421874e-05,
      "step": 4587
    },
    {
      "action_loss_avg": 0.03331028930842876,
      "epoch": 1.7498016357421874e-05,
      "step": 4587
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 4587,
      "training_step_time": 10.157529830932617
    },
    {
      "epoch": 1.7498016357421874e-05,
      "model_forward_time": 2.4364888668060303,
      "step": 4587
    },
    {
      "epoch": 1.7498016357421874e-05,
      "step": 4587,
      "training_step_time": 10.280720710754395
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 3.155963897705078,
      "step": 4588
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 4588,
      "training_step_time": 9.924644231796265
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 3.1378092765808105,
      "step": 4588
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 4588,
      "training_step_time": 9.908604860305786
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 2.435288667678833,
      "step": 4588
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 4588,
      "training_step_time": 7.459106683731079
    },
    {
      "epoch": 1.75018310546875e-05,
      "model_forward_time": 3.4107673168182373,
      "step": 4588
    },
    {
      "epoch": 1.75018310546875e-05,
      "step": 4588,
      "training_step_time": 10.26765513420105
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 2.4731950759887695,
      "step": 4589
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 4589,
      "training_step_time": 7.500828266143799
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 3.1345901489257812,
      "step": 4589
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 4589,
      "training_step_time": 9.900632858276367
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 3.1409623622894287,
      "step": 4589
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 4589,
      "training_step_time": 9.904874563217163
    },
    {
      "epoch": 1.7505645751953127e-05,
      "model_forward_time": 3.1409242153167725,
      "step": 4589
    },
    {
      "epoch": 1.7505645751953127e-05,
      "step": 4589,
      "training_step_time": 10.34398078918457
    },
    {
      "epoch": 1.750946044921875e-05,
      "grad_norm": 0.5263230204582214,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 4590
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 3.4756767749786377,
      "step": 4590
    },
    {
      "dynamics_loss_avg": 0.06353790797293186,
      "epoch": 1.750946044921875e-05,
      "step": 4590
    },
    {
      "action_loss_avg": 0.025019486248493195,
      "epoch": 1.750946044921875e-05,
      "step": 4590
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 4590,
      "training_step_time": 10.255142211914062
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 3.1411595344543457,
      "step": 4590
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 4590,
      "training_step_time": 9.917920589447021
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 3.155708074569702,
      "step": 4590
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 4590,
      "training_step_time": 9.931286811828613
    },
    {
      "epoch": 1.750946044921875e-05,
      "model_forward_time": 3.147758960723877,
      "step": 4590
    },
    {
      "epoch": 1.750946044921875e-05,
      "step": 4590,
      "training_step_time": 10.311422348022461
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 2.4770309925079346,
      "step": 4591
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 4591,
      "training_step_time": 7.764610528945923
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 3.1374671459198,
      "step": 4591
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 4591,
      "training_step_time": 9.911099672317505
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 3.139925241470337,
      "step": 4591
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 4591,
      "training_step_time": 9.910305976867676
    },
    {
      "epoch": 1.7513275146484376e-05,
      "model_forward_time": 2.43917179107666,
      "step": 4591
    },
    {
      "epoch": 1.7513275146484376e-05,
      "step": 4591,
      "training_step_time": 10.022139072418213
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 3.1637072563171387,
      "step": 4592
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 4592,
      "training_step_time": 9.934353351593018
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 3.41359281539917,
      "step": 4592
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 4592,
      "training_step_time": 10.177973985671997
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 2.435298204421997,
      "step": 4592
    },
    {
      "dynamics_loss_avg": 0.056812547519803046,
      "epoch": 1.751708984375e-05,
      "step": 4592
    },
    {
      "action_loss_avg": 0.030462488904595374,
      "epoch": 1.751708984375e-05,
      "step": 4592
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 4592,
      "training_step_time": 7.458756685256958
    },
    {
      "epoch": 1.751708984375e-05,
      "model_forward_time": 2.429246425628662,
      "step": 4592
    },
    {
      "epoch": 1.751708984375e-05,
      "step": 4592,
      "training_step_time": 10.26780915260315
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 3.160954475402832,
      "step": 4593
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 4593,
      "training_step_time": 9.927794456481934
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 3.150148868560791,
      "step": 4593
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 4593,
      "training_step_time": 9.92004108428955
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 3.1291708946228027,
      "step": 4593
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 4593,
      "training_step_time": 9.896700620651245
    },
    {
      "epoch": 1.7520904541015625e-05,
      "model_forward_time": 3.3845081329345703,
      "step": 4593
    },
    {
      "epoch": 1.7520904541015625e-05,
      "step": 4593,
      "training_step_time": 10.24598741531372
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 2.4696338176727295,
      "step": 4594
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 4594,
      "training_step_time": 7.492525577545166
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 3.1350245475769043,
      "step": 4594
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 4594,
      "training_step_time": 9.899998664855957
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 3.1387219429016113,
      "step": 4594
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 4594,
      "training_step_time": 9.908647775650024
    },
    {
      "epoch": 1.752471923828125e-05,
      "model_forward_time": 3.1407744884490967,
      "step": 4594
    },
    {
      "epoch": 1.752471923828125e-05,
      "step": 4594,
      "training_step_time": 10.30974006652832
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 2.739664316177368,
      "step": 4595
    },
    {
      "dynamics_loss_avg": 0.06440633833408356,
      "epoch": 1.7528533935546874e-05,
      "step": 4595
    },
    {
      "action_loss_avg": 0.028160115564242007,
      "epoch": 1.7528533935546874e-05,
      "step": 4595
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 4595,
      "training_step_time": 7.7717039585113525
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 3.1386141777038574,
      "step": 4595
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 4595,
      "training_step_time": 9.907206535339355
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 2.4308440685272217,
      "step": 4595
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 4595,
      "training_step_time": 7.451467990875244
    },
    {
      "epoch": 1.7528533935546874e-05,
      "model_forward_time": 3.1369595527648926,
      "step": 4595
    },
    {
      "epoch": 1.7528533935546874e-05,
      "step": 4595,
      "training_step_time": 10.269160747528076
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 3.1550910472869873,
      "step": 4596
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 4596,
      "training_step_time": 9.924701452255249
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 3.4332199096679688,
      "step": 4596
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 4596,
      "training_step_time": 10.20310640335083
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 3.1438300609588623,
      "step": 4596
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 4596,
      "training_step_time": 9.915042400360107
    },
    {
      "epoch": 1.75323486328125e-05,
      "model_forward_time": 2.434948682785034,
      "step": 4596
    },
    {
      "epoch": 1.75323486328125e-05,
      "step": 4596,
      "training_step_time": 10.28139853477478
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 3.158123016357422,
      "step": 4597
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 4597,
      "training_step_time": 9.932218313217163
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 3.140099287033081,
      "step": 4597
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 4597,
      "training_step_time": 9.915191173553467
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 3.4070510864257812,
      "step": 4597
    },
    {
      "dynamics_loss_avg": 0.051827603578567506,
      "epoch": 1.7536163330078127e-05,
      "step": 4597
    },
    {
      "action_loss_avg": 0.02885234197601676,
      "epoch": 1.7536163330078127e-05,
      "step": 4597
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 4597,
      "training_step_time": 10.179557085037231
    },
    {
      "epoch": 1.7536163330078127e-05,
      "model_forward_time": 2.4363274574279785,
      "step": 4597
    },
    {
      "epoch": 1.7536163330078127e-05,
      "step": 4597,
      "training_step_time": 10.272387027740479
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 3.171893835067749,
      "step": 4598
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 4598,
      "training_step_time": 9.939929723739624
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 3.135963201522827,
      "step": 4598
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 4598,
      "training_step_time": 9.903250694274902
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 3.138871192932129,
      "step": 4598
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 4598,
      "training_step_time": 9.901858806610107
    },
    {
      "epoch": 1.753997802734375e-05,
      "model_forward_time": 3.4156880378723145,
      "step": 4598
    },
    {
      "epoch": 1.753997802734375e-05,
      "step": 4598,
      "training_step_time": 10.281337022781372
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 3.168435573577881,
      "step": 4599
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 4599,
      "training_step_time": 9.947676658630371
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 2.433899402618408,
      "step": 4599
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 4599,
      "training_step_time": 7.466658115386963
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 3.1369097232818604,
      "step": 4599
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 4599,
      "training_step_time": 9.909276247024536
    },
    {
      "epoch": 1.7543792724609376e-05,
      "model_forward_time": 3.1405749320983887,
      "step": 4599
    },
    {
      "epoch": 1.7543792724609376e-05,
      "step": 4599,
      "training_step_time": 10.297224998474121
    },
    {
      "epoch": 1.7547607421875e-05,
      "grad_norm": 0.4865167737007141,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 4600
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 3.474085569381714,
      "step": 4600
    },
    {
      "dynamics_loss_avg": 0.0749287735670805,
      "epoch": 1.7547607421875e-05,
      "step": 4600
    },
    {
      "action_loss_avg": 0.03775048637762666,
      "epoch": 1.7547607421875e-05,
      "step": 4600
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 4600,
      "training_step_time": 10.250560522079468
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 3.140808582305908,
      "step": 4600
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 4600,
      "training_step_time": 9.913729906082153
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 3.146876096725464,
      "step": 4600
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 4600,
      "training_step_time": 9.920279264450073
    },
    {
      "epoch": 1.7547607421875e-05,
      "model_forward_time": 3.1420040130615234,
      "step": 4600
    },
    {
      "epoch": 1.7547607421875e-05,
      "step": 4600,
      "training_step_time": 10.022406578063965
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 2.47788667678833,
      "step": 4601
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 4601,
      "training_step_time": 7.509657621383667
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 3.3955488204956055,
      "step": 4601
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 4601,
      "training_step_time": 10.16979432106018
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 3.1406967639923096,
      "step": 4601
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 4601,
      "training_step_time": 9.917548179626465
    },
    {
      "epoch": 1.7551422119140625e-05,
      "model_forward_time": 3.143876552581787,
      "step": 4601
    },
    {
      "epoch": 1.7551422119140625e-05,
      "step": 4601,
      "training_step_time": 10.292701482772827
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 2.4828684329986572,
      "step": 4602
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 4602,
      "training_step_time": 7.513896465301514
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 3.140904426574707,
      "step": 4602
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 4602,
      "training_step_time": 9.914749383926392
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 3.4185235500335693,
      "step": 4602
    },
    {
      "dynamics_loss_avg": 0.060258184745907786,
      "epoch": 1.755523681640625e-05,
      "step": 4602
    },
    {
      "action_loss_avg": 0.027542464435100555,
      "epoch": 1.755523681640625e-05,
      "step": 4602
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 4602,
      "training_step_time": 10.198537349700928
    },
    {
      "epoch": 1.755523681640625e-05,
      "model_forward_time": 3.143660068511963,
      "step": 4602
    },
    {
      "epoch": 1.755523681640625e-05,
      "step": 4602,
      "training_step_time": 10.306669235229492
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 2.4824016094207764,
      "step": 4603
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 4603,
      "training_step_time": 7.512239694595337
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 3.1387178897857666,
      "step": 4603
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 4603,
      "training_step_time": 9.9073965549469
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 3.137138605117798,
      "step": 4603
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 4603,
      "training_step_time": 9.906956911087036
    },
    {
      "epoch": 1.7559051513671874e-05,
      "model_forward_time": 3.4002366065979004,
      "step": 4603
    },
    {
      "epoch": 1.7559051513671874e-05,
      "step": 4603,
      "training_step_time": 10.265092849731445
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 3.181804895401001,
      "step": 4604
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 4604,
      "training_step_time": 9.953961372375488
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 3.144965887069702,
      "step": 4604
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 4604,
      "training_step_time": 9.92121171951294
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 3.1524486541748047,
      "step": 4604
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 4604,
      "training_step_time": 9.929234981536865
    },
    {
      "epoch": 1.75628662109375e-05,
      "model_forward_time": 3.1472878456115723,
      "step": 4604
    },
    {
      "epoch": 1.75628662109375e-05,
      "step": 4604,
      "training_step_time": 10.270906448364258
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 3.4524738788604736,
      "step": 4605
    },
    {
      "dynamics_loss_avg": 0.054102050326764586,
      "epoch": 1.7566680908203127e-05,
      "step": 4605
    },
    {
      "action_loss_avg": 0.027578945783898235,
      "epoch": 1.7566680908203127e-05,
      "step": 4605
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 4605,
      "training_step_time": 10.228840589523315
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 2.4393866062164307,
      "step": 4605
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 4605,
      "training_step_time": 7.470622777938843
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 3.1366665363311768,
      "step": 4605
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 4605,
      "training_step_time": 9.904144763946533
    },
    {
      "epoch": 1.7566680908203127e-05,
      "model_forward_time": 2.436371088027954,
      "step": 4605
    },
    {
      "epoch": 1.7566680908203127e-05,
      "step": 4605,
      "training_step_time": 10.015366077423096
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 3.190823793411255,
      "step": 4606
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 4606,
      "training_step_time": 9.968234062194824
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 3.4475529193878174,
      "step": 4606
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 4606,
      "training_step_time": 10.220735311508179
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 2.4425930976867676,
      "step": 4606
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 4606,
      "training_step_time": 7.475116968154907
    },
    {
      "epoch": 1.757049560546875e-05,
      "model_forward_time": 3.1465654373168945,
      "step": 4606
    },
    {
      "epoch": 1.757049560546875e-05,
      "step": 4606,
      "training_step_time": 10.237738132476807
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 3.175199508666992,
      "step": 4607
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 4607,
      "training_step_time": 9.951630592346191
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 3.140364408493042,
      "step": 4607
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 4607,
      "training_step_time": 9.915180683135986
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 3.1353323459625244,
      "step": 4607
    },
    {
      "dynamics_loss_avg": 0.06960784140974283,
      "epoch": 1.7574310302734376e-05,
      "step": 4607
    },
    {
      "action_loss_avg": 0.03699726536870003,
      "epoch": 1.7574310302734376e-05,
      "step": 4607
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 4607,
      "training_step_time": 10.186929941177368
    },
    {
      "epoch": 1.7574310302734376e-05,
      "model_forward_time": 3.136894941329956,
      "step": 4607
    },
    {
      "epoch": 1.7574310302734376e-05,
      "step": 4607,
      "training_step_time": 10.265544414520264
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 3.1700966358184814,
      "step": 4608
    },
    {
      "epoch": 1.7578125e-05,
      "step": 4608,
      "training_step_time": 9.943171739578247
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 3.136293411254883,
      "step": 4608
    },
    {
      "epoch": 1.7578125e-05,
      "step": 4608,
      "training_step_time": 9.906577110290527
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 3.1412758827209473,
      "step": 4608
    },
    {
      "epoch": 1.7578125e-05,
      "step": 4608,
      "training_step_time": 9.910790205001831
    },
    {
      "epoch": 1.7578125e-05,
      "model_forward_time": 3.428454637527466,
      "step": 4608
    },
    {
      "epoch": 1.7578125e-05,
      "step": 4608,
      "training_step_time": 10.300800323486328
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 2.483673334121704,
      "step": 4609
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 4609,
      "training_step_time": 7.511598110198975
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 3.135587692260742,
      "step": 4609
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 4609,
      "training_step_time": 9.908127069473267
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 3.1391165256500244,
      "step": 4609
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 4609,
      "training_step_time": 9.913408041000366
    },
    {
      "epoch": 1.7581939697265625e-05,
      "model_forward_time": 3.1396021842956543,
      "step": 4609
    },
    {
      "epoch": 1.7581939697265625e-05,
      "step": 4609,
      "training_step_time": 10.037578821182251
    },
    {
      "epoch": 1.758575439453125e-05,
      "grad_norm": 0.7298262715339661,
      "learning_rate": 1e-05,
      "loss": 0.0945,
      "step": 4610
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 3.168670177459717,
      "step": 4610
    },
    {
      "dynamics_loss_avg": 0.04644895568490028,
      "epoch": 1.758575439453125e-05,
      "step": 4610
    },
    {
      "action_loss_avg": 0.029614798026159406,
      "epoch": 1.758575439453125e-05,
      "step": 4610
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 4610,
      "training_step_time": 9.93879222869873
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 3.3863935470581055,
      "step": 4610
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 4610,
      "training_step_time": 10.16094160079956
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 3.1460647583007812,
      "step": 4610
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 4610,
      "training_step_time": 9.92277479171753
    },
    {
      "epoch": 1.758575439453125e-05,
      "model_forward_time": 3.143749237060547,
      "step": 4610
    },
    {
      "epoch": 1.758575439453125e-05,
      "step": 4610,
      "training_step_time": 10.04228925704956
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 3.1603522300720215,
      "step": 4611
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 4611,
      "training_step_time": 9.938390254974365
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 3.403658628463745,
      "step": 4611
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 4611,
      "training_step_time": 10.177635669708252
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 3.145775556564331,
      "step": 4611
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 4611,
      "training_step_time": 9.9210844039917
    },
    {
      "epoch": 1.7589569091796874e-05,
      "model_forward_time": 2.4379360675811768,
      "step": 4611
    },
    {
      "epoch": 1.7589569091796874e-05,
      "step": 4611,
      "training_step_time": 10.308616161346436
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 3.1628189086914062,
      "step": 4612
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 4612,
      "training_step_time": 9.934557676315308
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 3.1349456310272217,
      "step": 4612
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 4612,
      "training_step_time": 9.901519536972046
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 2.431241273880005,
      "step": 4612
    },
    {
      "dynamics_loss_avg": 0.052425979264080526,
      "epoch": 1.75933837890625e-05,
      "step": 4612
    },
    {
      "action_loss_avg": 0.02628478305414319,
      "epoch": 1.75933837890625e-05,
      "step": 4612
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 4612,
      "training_step_time": 7.457907199859619
    },
    {
      "epoch": 1.75933837890625e-05,
      "model_forward_time": 3.3878133296966553,
      "step": 4612
    },
    {
      "epoch": 1.75933837890625e-05,
      "step": 4612,
      "training_step_time": 10.280765295028687
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 2.4436845779418945,
      "step": 4613
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 4613,
      "training_step_time": 7.463929176330566
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 3.1377112865448,
      "step": 4613
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 4613,
      "training_step_time": 9.903956890106201
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 2.4325063228607178,
      "step": 4613
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 4613,
      "training_step_time": 7.455050230026245
    },
    {
      "epoch": 1.7597198486328127e-05,
      "model_forward_time": 3.13663649559021,
      "step": 4613
    },
    {
      "epoch": 1.7597198486328127e-05,
      "step": 4613,
      "training_step_time": 10.265136241912842
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 3.4688169956207275,
      "step": 4614
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 4614,
      "training_step_time": 10.23697805404663
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 3.140937089920044,
      "step": 4614
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 4614,
      "training_step_time": 9.921854734420776
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 3.1386494636535645,
      "step": 4614
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 4614,
      "training_step_time": 9.918389081954956
    },
    {
      "epoch": 1.760101318359375e-05,
      "model_forward_time": 2.436624526977539,
      "step": 4614
    },
    {
      "epoch": 1.760101318359375e-05,
      "step": 4614,
      "training_step_time": 10.254604578018188
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 3.18969988822937,
      "step": 4615
    },
    {
      "dynamics_loss_avg": 0.059175568632781506,
      "epoch": 1.7604827880859376e-05,
      "step": 4615
    },
    {
      "action_loss_avg": 0.025249846931546926,
      "epoch": 1.7604827880859376e-05,
      "step": 4615
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 4615,
      "training_step_time": 9.968606948852539
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 3.4262535572052,
      "step": 4615
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 4615,
      "training_step_time": 10.206302642822266
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 3.1427321434020996,
      "step": 4615
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 4615,
      "training_step_time": 9.920782566070557
    },
    {
      "epoch": 1.7604827880859376e-05,
      "model_forward_time": 3.145407199859619,
      "step": 4615
    },
    {
      "epoch": 1.7604827880859376e-05,
      "step": 4615,
      "training_step_time": 10.029249906539917
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 3.1798272132873535,
      "step": 4616
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 4616,
      "training_step_time": 9.961605072021484
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 3.1415765285491943,
      "step": 4616
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 4616,
      "training_step_time": 9.920741319656372
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 3.3981075286865234,
      "step": 4616
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 4616,
      "training_step_time": 10.173166513442993
    },
    {
      "epoch": 1.7608642578125e-05,
      "model_forward_time": 3.14724063873291,
      "step": 4616
    },
    {
      "epoch": 1.7608642578125e-05,
      "step": 4616,
      "training_step_time": 10.29792857170105
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 3.1873416900634766,
      "step": 4617
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 4617,
      "training_step_time": 9.968958616256714
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 2.4418749809265137,
      "step": 4617
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 4617,
      "training_step_time": 7.476089715957642
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 3.1419997215270996,
      "step": 4617
    },
    {
      "dynamics_loss_avg": 0.07078686393797398,
      "epoch": 1.7612457275390625e-05,
      "step": 4617
    },
    {
      "action_loss_avg": 0.036293689906597135,
      "epoch": 1.7612457275390625e-05,
      "step": 4617
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 4617,
      "training_step_time": 9.916651487350464
    },
    {
      "epoch": 1.7612457275390625e-05,
      "model_forward_time": 3.4141101837158203,
      "step": 4617
    },
    {
      "epoch": 1.7612457275390625e-05,
      "step": 4617,
      "training_step_time": 10.285972595214844
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 2.451143980026245,
      "step": 4618
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 4618,
      "training_step_time": 7.488464593887329
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 3.132100820541382,
      "step": 4618
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 4618,
      "training_step_time": 9.900285243988037
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 2.439905881881714,
      "step": 4618
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 4618,
      "training_step_time": 7.461184740066528
    },
    {
      "epoch": 1.761627197265625e-05,
      "model_forward_time": 2.4311165809631348,
      "step": 4618
    },
    {
      "epoch": 1.761627197265625e-05,
      "step": 4618,
      "training_step_time": 10.29323673248291
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 3.4365334510803223,
      "step": 4619
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 4619,
      "training_step_time": 10.194509744644165
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 3.1362736225128174,
      "step": 4619
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 4619,
      "training_step_time": 9.900293350219727
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 3.1402885913848877,
      "step": 4619
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 4619,
      "training_step_time": 9.90849494934082
    },
    {
      "epoch": 1.7620086669921874e-05,
      "model_forward_time": 3.1419551372528076,
      "step": 4619
    },
    {
      "epoch": 1.7620086669921874e-05,
      "step": 4619,
      "training_step_time": 10.277104377746582
    },
    {
      "epoch": 1.76239013671875e-05,
      "grad_norm": 0.34600579738616943,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 4620
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 3.182215929031372,
      "step": 4620
    },
    {
      "dynamics_loss_avg": 0.05849310215562582,
      "epoch": 1.76239013671875e-05,
      "step": 4620
    },
    {
      "action_loss_avg": 0.025537449051626027,
      "epoch": 1.76239013671875e-05,
      "step": 4620
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 4620,
      "training_step_time": 9.955040693283081
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 3.393779754638672,
      "step": 4620
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 4620,
      "training_step_time": 10.161498546600342
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 3.143972873687744,
      "step": 4620
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 4620,
      "training_step_time": 9.909013748168945
    },
    {
      "epoch": 1.76239013671875e-05,
      "model_forward_time": 3.1410176753997803,
      "step": 4620
    },
    {
      "epoch": 1.76239013671875e-05,
      "step": 4620,
      "training_step_time": 10.286713361740112
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 3.18588924407959,
      "step": 4621
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 4621,
      "training_step_time": 9.951496362686157
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 3.140408515930176,
      "step": 4621
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 4621,
      "training_step_time": 9.906422853469849
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 3.4170801639556885,
      "step": 4621
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 4621,
      "training_step_time": 10.178325891494751
    },
    {
      "epoch": 1.7627716064453127e-05,
      "model_forward_time": 3.1377296447753906,
      "step": 4621
    },
    {
      "epoch": 1.7627716064453127e-05,
      "step": 4621,
      "training_step_time": 10.013161420822144
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 3.178720712661743,
      "step": 4622
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 4622,
      "training_step_time": 9.944594860076904
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 2.4400672912597656,
      "step": 4622
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 4622,
      "training_step_time": 7.467133283615112
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 3.1392433643341064,
      "step": 4622
    },
    {
      "dynamics_loss_avg": 0.05488338097929955,
      "epoch": 1.763153076171875e-05,
      "step": 4622
    },
    {
      "action_loss_avg": 0.027336080186069013,
      "epoch": 1.763153076171875e-05,
      "step": 4622
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 4622,
      "training_step_time": 9.907704830169678
    },
    {
      "epoch": 1.763153076171875e-05,
      "model_forward_time": 3.394033193588257,
      "step": 4622
    },
    {
      "epoch": 1.763153076171875e-05,
      "step": 4622,
      "training_step_time": 10.260040521621704
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 2.4854328632354736,
      "step": 4623
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 4623,
      "training_step_time": 7.512815952301025
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 3.1372909545898438,
      "step": 4623
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 4623,
      "training_step_time": 9.902594327926636
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 3.136195182800293,
      "step": 4623
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 4623,
      "training_step_time": 9.900607585906982
    },
    {
      "epoch": 1.7635345458984376e-05,
      "model_forward_time": 3.141700267791748,
      "step": 4623
    },
    {
      "epoch": 1.7635345458984376e-05,
      "step": 4623,
      "training_step_time": 10.254270792007446
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 3.451359510421753,
      "step": 4624
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 4624,
      "training_step_time": 10.219032049179077
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 3.141739845275879,
      "step": 4624
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 4624,
      "training_step_time": 9.911367893218994
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 3.139157772064209,
      "step": 4624
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 4624,
      "training_step_time": 9.908178567886353
    },
    {
      "epoch": 1.763916015625e-05,
      "model_forward_time": 3.146646499633789,
      "step": 4624
    },
    {
      "epoch": 1.763916015625e-05,
      "step": 4624,
      "training_step_time": 10.298788070678711
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 3.1937026977539062,
      "step": 4625
    },
    {
      "dynamics_loss_avg": 0.05175809133797884,
      "epoch": 1.7642974853515625e-05,
      "step": 4625
    },
    {
      "action_loss_avg": 0.029193395841866733,
      "epoch": 1.7642974853515625e-05,
      "step": 4625
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 4625,
      "training_step_time": 9.97113037109375
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 3.4289634227752686,
      "step": 4625
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 4625,
      "training_step_time": 10.197922706604004
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 3.1459274291992188,
      "step": 4625
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 4625,
      "training_step_time": 9.922234535217285
    },
    {
      "epoch": 1.7642974853515625e-05,
      "model_forward_time": 3.145470380783081,
      "step": 4625
    },
    {
      "epoch": 1.7642974853515625e-05,
      "step": 4625,
      "training_step_time": 10.035516262054443
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 3.1914453506469727,
      "step": 4626
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 4626,
      "training_step_time": 9.963936805725098
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 3.1418535709381104,
      "step": 4626
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 4626,
      "training_step_time": 9.910705089569092
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 3.4006083011627197,
      "step": 4626
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 4626,
      "training_step_time": 10.17141318321228
    },
    {
      "epoch": 1.764678955078125e-05,
      "model_forward_time": 3.146881103515625,
      "step": 4626
    },
    {
      "epoch": 1.764678955078125e-05,
      "step": 4626,
      "training_step_time": 10.261046886444092
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 3.212433099746704,
      "step": 4627
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 4627,
      "training_step_time": 9.983359813690186
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 3.1390750408172607,
      "step": 4627
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 4627,
      "training_step_time": 9.906799554824829
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 3.134413242340088,
      "step": 4627
    },
    {
      "dynamics_loss_avg": 0.07024939469993115,
      "epoch": 1.7650604248046874e-05,
      "step": 4627
    },
    {
      "action_loss_avg": 0.029592234641313553,
      "epoch": 1.7650604248046874e-05,
      "step": 4627
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 4627,
      "training_step_time": 9.898630619049072
    },
    {
      "epoch": 1.7650604248046874e-05,
      "model_forward_time": 3.4031319618225098,
      "step": 4627
    },
    {
      "epoch": 1.7650604248046874e-05,
      "step": 4627,
      "training_step_time": 10.280439376831055
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 3.1705923080444336,
      "step": 4628
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 4628,
      "training_step_time": 9.93825626373291
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 3.137964963912964,
      "step": 4628
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 4628,
      "training_step_time": 9.90292763710022
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 3.144218683242798,
      "step": 4628
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 4628,
      "training_step_time": 9.904781103134155
    },
    {
      "epoch": 1.76544189453125e-05,
      "model_forward_time": 3.135704278945923,
      "step": 4628
    },
    {
      "epoch": 1.76544189453125e-05,
      "step": 4628,
      "training_step_time": 10.301280975341797
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 3.4747393131256104,
      "step": 4629
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 4629,
      "training_step_time": 10.239818334579468
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 3.131990432739258,
      "step": 4629
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 4629,
      "training_step_time": 9.89205813407898
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 3.1393651962280273,
      "step": 4629
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 4629,
      "training_step_time": 9.907696962356567
    },
    {
      "epoch": 1.7658233642578123e-05,
      "model_forward_time": 2.4386982917785645,
      "step": 4629
    },
    {
      "epoch": 1.7658233642578123e-05,
      "step": 4629,
      "training_step_time": 10.301789999008179
    },
    {
      "epoch": 1.766204833984375e-05,
      "grad_norm": 0.6737249493598938,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 4630
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 3.1790754795074463,
      "step": 4630
    },
    {
      "dynamics_loss_avg": 0.0508576637133956,
      "epoch": 1.766204833984375e-05,
      "step": 4630
    },
    {
      "action_loss_avg": 0.034124648198485374,
      "epoch": 1.766204833984375e-05,
      "step": 4630
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 4630,
      "training_step_time": 9.94752550125122
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 2.692044973373413,
      "step": 4630
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 4630,
      "training_step_time": 7.72001051902771
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 3.1411683559417725,
      "step": 4630
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 4630,
      "training_step_time": 9.913637638092041
    },
    {
      "epoch": 1.766204833984375e-05,
      "model_forward_time": 3.1441450119018555,
      "step": 4630
    },
    {
      "epoch": 1.766204833984375e-05,
      "step": 4630,
      "training_step_time": 10.309057474136353
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 3.1955196857452393,
      "step": 4631
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 4631,
      "training_step_time": 9.969641923904419
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 3.1440601348876953,
      "step": 4631
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 4631,
      "training_step_time": 9.91783356666565
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 3.4187428951263428,
      "step": 4631
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 4631,
      "training_step_time": 10.189810752868652
    },
    {
      "epoch": 1.7665863037109376e-05,
      "model_forward_time": 2.441171169281006,
      "step": 4631
    },
    {
      "epoch": 1.7665863037109376e-05,
      "step": 4631,
      "training_step_time": 10.270604848861694
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 3.191390037536621,
      "step": 4632
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 4632,
      "training_step_time": 9.959821701049805
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 3.142505645751953,
      "step": 4632
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 4632,
      "training_step_time": 9.915402173995972
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 3.1428866386413574,
      "step": 4632
    },
    {
      "dynamics_loss_avg": 0.07478410042822362,
      "epoch": 1.7669677734375e-05,
      "step": 4632
    },
    {
      "action_loss_avg": 0.03384857214987278,
      "epoch": 1.7669677734375e-05,
      "step": 4632
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 4632,
      "training_step_time": 9.918070554733276
    },
    {
      "epoch": 1.7669677734375e-05,
      "model_forward_time": 3.3959224224090576,
      "step": 4632
    },
    {
      "epoch": 1.7669677734375e-05,
      "step": 4632,
      "training_step_time": 10.30316948890686
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 3.1846835613250732,
      "step": 4633
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 4633,
      "training_step_time": 9.950571298599243
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 3.1417348384857178,
      "step": 4633
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 4633,
      "training_step_time": 9.908510446548462
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 3.142030954360962,
      "step": 4633
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 4633,
      "training_step_time": 9.910632371902466
    },
    {
      "epoch": 1.7673492431640625e-05,
      "model_forward_time": 3.1436562538146973,
      "step": 4633
    },
    {
      "epoch": 1.7673492431640625e-05,
      "step": 4633,
      "training_step_time": 10.033418416976929
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 3.4633114337921143,
      "step": 4634
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 4634,
      "training_step_time": 10.23362135887146
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 3.1428983211517334,
      "step": 4634
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 4634,
      "training_step_time": 9.910284280776978
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 3.137267827987671,
      "step": 4634
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 4634,
      "training_step_time": 9.9024498462677
    },
    {
      "epoch": 1.767730712890625e-05,
      "model_forward_time": 3.144662857055664,
      "step": 4634
    },
    {
      "epoch": 1.767730712890625e-05,
      "step": 4634,
      "training_step_time": 10.322628021240234
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 2.4791181087493896,
      "step": 4635
    },
    {
      "dynamics_loss_avg": 0.07153015080839395,
      "epoch": 1.7681121826171874e-05,
      "step": 4635
    },
    {
      "action_loss_avg": 0.03367701955139637,
      "epoch": 1.7681121826171874e-05,
      "step": 4635
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 4635,
      "training_step_time": 7.762371063232422
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 3.139357089996338,
      "step": 4635
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 4635,
      "training_step_time": 9.905897855758667
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 3.134241819381714,
      "step": 4635
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 4635,
      "training_step_time": 9.903984069824219
    },
    {
      "epoch": 1.7681121826171874e-05,
      "model_forward_time": 3.1423487663269043,
      "step": 4635
    },
    {
      "epoch": 1.7681121826171874e-05,
      "step": 4635,
      "training_step_time": 10.280999422073364
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 3.1936755180358887,
      "step": 4636
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 4636,
      "training_step_time": 9.964138984680176
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 3.1354856491088867,
      "step": 4636
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 4636,
      "training_step_time": 10.186242818832397
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 3.1308066844940186,
      "step": 4636
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 4636,
      "training_step_time": 9.895406246185303
    },
    {
      "epoch": 1.76849365234375e-05,
      "model_forward_time": 3.1376678943634033,
      "step": 4636
    },
    {
      "epoch": 1.76849365234375e-05,
      "step": 4636,
      "training_step_time": 10.279185771942139
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 3.1912004947662354,
      "step": 4637
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 4637,
      "training_step_time": 9.957834005355835
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 3.136043071746826,
      "step": 4637
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 4637,
      "training_step_time": 9.899641752243042
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 3.427264451980591,
      "step": 4637
    },
    {
      "dynamics_loss_avg": 0.07085182685405016,
      "epoch": 1.7688751220703123e-05,
      "step": 4637
    },
    {
      "action_loss_avg": 0.03735151942819357,
      "epoch": 1.7688751220703123e-05,
      "step": 4637
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 4637,
      "training_step_time": 10.190224647521973
    },
    {
      "epoch": 1.7688751220703123e-05,
      "model_forward_time": 3.1363770961761475,
      "step": 4637
    },
    {
      "epoch": 1.7688751220703123e-05,
      "step": 4637,
      "training_step_time": 10.262344360351562
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 3.187742233276367,
      "step": 4638
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 4638,
      "training_step_time": 9.952425718307495
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 3.1411843299865723,
      "step": 4638
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 4638,
      "training_step_time": 9.905821561813354
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 3.1342906951904297,
      "step": 4638
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 4638,
      "training_step_time": 9.900672435760498
    },
    {
      "epoch": 1.769256591796875e-05,
      "model_forward_time": 3.1392552852630615,
      "step": 4638
    },
    {
      "epoch": 1.769256591796875e-05,
      "step": 4638,
      "training_step_time": 9.998952388763428
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 3.4896228313446045,
      "step": 4639
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 4639,
      "training_step_time": 10.253924131393433
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 2.4363596439361572,
      "step": 4639
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 4639,
      "training_step_time": 7.459575653076172
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 3.1338372230529785,
      "step": 4639
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 4639,
      "training_step_time": 9.898595571517944
    },
    {
      "epoch": 1.7696380615234376e-05,
      "model_forward_time": 3.1394803524017334,
      "step": 4639
    },
    {
      "epoch": 1.7696380615234376e-05,
      "step": 4639,
      "training_step_time": 10.30177640914917
    },
    {
      "epoch": 1.77001953125e-05,
      "grad_norm": 0.7820988893508911,
      "learning_rate": 1e-05,
      "loss": 0.0962,
      "step": 4640
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 3.472604990005493,
      "step": 4640
    },
    {
      "dynamics_loss_avg": 0.0643431756645441,
      "epoch": 1.77001953125e-05,
      "step": 4640
    },
    {
      "action_loss_avg": 0.03752887547016144,
      "epoch": 1.77001953125e-05,
      "step": 4640
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 4640,
      "training_step_time": 10.23888611793518
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 2.4306282997131348,
      "step": 4640
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 4640,
      "training_step_time": 7.453039884567261
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 3.132451295852661,
      "step": 4640
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 4640,
      "training_step_time": 9.898509979248047
    },
    {
      "epoch": 1.77001953125e-05,
      "model_forward_time": 3.1424500942230225,
      "step": 4640
    },
    {
      "epoch": 1.77001953125e-05,
      "step": 4640,
      "training_step_time": 10.30379056930542
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 3.180673837661743,
      "step": 4641
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 4641,
      "training_step_time": 9.950515508651733
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 3.1387014389038086,
      "step": 4641
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 4641,
      "training_step_time": 9.902119874954224
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 3.3919007778167725,
      "step": 4641
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 4641,
      "training_step_time": 10.160090923309326
    },
    {
      "epoch": 1.7704010009765625e-05,
      "model_forward_time": 3.1463708877563477,
      "step": 4641
    },
    {
      "epoch": 1.7704010009765625e-05,
      "step": 4641,
      "training_step_time": 10.251113653182983
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 3.1751880645751953,
      "step": 4642
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 4642,
      "training_step_time": 9.946353912353516
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 3.144602060317993,
      "step": 4642
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 4642,
      "training_step_time": 9.923342227935791
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 3.1478993892669678,
      "step": 4642
    },
    {
      "dynamics_loss_avg": 0.05859703905880451,
      "epoch": 1.770782470703125e-05,
      "step": 4642
    },
    {
      "action_loss_avg": 0.0308167215436697,
      "epoch": 1.770782470703125e-05,
      "step": 4642
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 4642,
      "training_step_time": 10.202126026153564
    },
    {
      "epoch": 1.770782470703125e-05,
      "model_forward_time": 3.1466903686523438,
      "step": 4642
    },
    {
      "epoch": 1.770782470703125e-05,
      "step": 4642,
      "training_step_time": 10.280475854873657
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 3.1754207611083984,
      "step": 4643
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 4643,
      "training_step_time": 9.949211120605469
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 3.1443071365356445,
      "step": 4643
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 4643,
      "training_step_time": 9.916544914245605
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 3.1429824829101562,
      "step": 4643
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 4643,
      "training_step_time": 9.913289546966553
    },
    {
      "epoch": 1.7711639404296874e-05,
      "model_forward_time": 3.4336979389190674,
      "step": 4643
    },
    {
      "epoch": 1.7711639404296874e-05,
      "step": 4643,
      "training_step_time": 10.299561738967896
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 3.1616358757019043,
      "step": 4644
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 4644,
      "training_step_time": 9.934544801712036
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 3.1335208415985107,
      "step": 4644
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 4644,
      "training_step_time": 9.901319980621338
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 3.134357452392578,
      "step": 4644
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 4644,
      "training_step_time": 9.898572444915771
    },
    {
      "epoch": 1.77154541015625e-05,
      "model_forward_time": 2.4312169551849365,
      "step": 4644
    },
    {
      "epoch": 1.77154541015625e-05,
      "step": 4644,
      "training_step_time": 10.2923104763031
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 3.1699085235595703,
      "step": 4645
    },
    {
      "dynamics_loss_avg": 0.06198191046714783,
      "epoch": 1.7719268798828123e-05,
      "step": 4645
    },
    {
      "action_loss_avg": 0.02928630318492651,
      "epoch": 1.7719268798828123e-05,
      "step": 4645
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 4645,
      "training_step_time": 9.946193933486938
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 3.391791582107544,
      "step": 4645
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 4645,
      "training_step_time": 10.18460202217102
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 3.1419148445129395,
      "step": 4645
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 4645,
      "training_step_time": 9.915616512298584
    },
    {
      "epoch": 1.7719268798828123e-05,
      "model_forward_time": 3.1356594562530518,
      "step": 4645
    },
    {
      "epoch": 1.7719268798828123e-05,
      "step": 4645,
      "training_step_time": 10.294727563858032
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 3.1740760803222656,
      "step": 4646
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 4646,
      "training_step_time": 9.94725513458252
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 2.718104124069214,
      "step": 4646
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 4646,
      "training_step_time": 7.74912166595459
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 3.1451754570007324,
      "step": 4646
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 4646,
      "training_step_time": 9.927870750427246
    },
    {
      "epoch": 1.772308349609375e-05,
      "model_forward_time": 3.144857406616211,
      "step": 4646
    },
    {
      "epoch": 1.772308349609375e-05,
      "step": 4646,
      "training_step_time": 10.260616064071655
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 3.1666741371154785,
      "step": 4647
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 4647,
      "training_step_time": 9.946072578430176
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 3.148289918899536,
      "step": 4647
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 4647,
      "training_step_time": 9.923330545425415
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 3.1426515579223633,
      "step": 4647
    },
    {
      "dynamics_loss_avg": 0.05360922999680042,
      "epoch": 1.7726898193359376e-05,
      "step": 4647
    },
    {
      "action_loss_avg": 0.030176989641040562,
      "epoch": 1.7726898193359376e-05,
      "step": 4647
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 4647,
      "training_step_time": 10.217169046401978
    },
    {
      "epoch": 1.7726898193359376e-05,
      "model_forward_time": 3.138638496398926,
      "step": 4647
    },
    {
      "epoch": 1.7726898193359376e-05,
      "step": 4647,
      "training_step_time": 10.271363019943237
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 3.176625967025757,
      "step": 4648
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 4648,
      "training_step_time": 9.964248180389404
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 3.147481918334961,
      "step": 4648
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 4648,
      "training_step_time": 9.92829942703247
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 3.1412360668182373,
      "step": 4648
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 4648,
      "training_step_time": 9.919503450393677
    },
    {
      "epoch": 1.7730712890625e-05,
      "model_forward_time": 3.405925989151001,
      "step": 4648
    },
    {
      "epoch": 1.7730712890625e-05,
      "step": 4648,
      "training_step_time": 10.277137041091919
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 3.181968927383423,
      "step": 4649
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 4649,
      "training_step_time": 9.964405298233032
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 2.4371938705444336,
      "step": 4649
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 4649,
      "training_step_time": 7.471418380737305
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 3.1398391723632812,
      "step": 4649
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 4649,
      "training_step_time": 9.921511173248291
    },
    {
      "epoch": 1.7734527587890625e-05,
      "model_forward_time": 3.1461734771728516,
      "step": 4649
    },
    {
      "epoch": 1.7734527587890625e-05,
      "step": 4649,
      "training_step_time": 10.270157098770142
    },
    {
      "epoch": 1.773834228515625e-05,
      "grad_norm": 0.5493478178977966,
      "learning_rate": 1e-05,
      "loss": 0.0855,
      "step": 4650
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 3.184072732925415,
      "step": 4650
    },
    {
      "dynamics_loss_avg": 0.0636430462822318,
      "epoch": 1.773834228515625e-05,
      "step": 4650
    },
    {
      "action_loss_avg": 0.033728588139638305,
      "epoch": 1.773834228515625e-05,
      "step": 4650
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 4650,
      "training_step_time": 10.252024412155151
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 3.146907091140747,
      "step": 4650
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 4650,
      "training_step_time": 9.924835205078125
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 3.1379621028900146,
      "step": 4650
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 4650,
      "training_step_time": 9.913130283355713
    },
    {
      "epoch": 1.773834228515625e-05,
      "model_forward_time": 3.143874168395996,
      "step": 4650
    },
    {
      "epoch": 1.773834228515625e-05,
      "step": 4650,
      "training_step_time": 10.313386678695679
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 3.1797549724578857,
      "step": 4651
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 4651,
      "training_step_time": 9.955753564834595
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 3.1378731727600098,
      "step": 4651
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 4651,
      "training_step_time": 10.198378086090088
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 3.1378204822540283,
      "step": 4651
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 4651,
      "training_step_time": 9.914365291595459
    },
    {
      "epoch": 1.7742156982421874e-05,
      "model_forward_time": 3.138960123062134,
      "step": 4651
    },
    {
      "epoch": 1.7742156982421874e-05,
      "step": 4651,
      "training_step_time": 10.312006711959839
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 3.1761350631713867,
      "step": 4652
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 4652,
      "training_step_time": 9.956506729125977
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 3.139801263809204,
      "step": 4652
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 4652,
      "training_step_time": 9.912959337234497
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 3.1413137912750244,
      "step": 4652
    },
    {
      "dynamics_loss_avg": 0.05067832209169865,
      "epoch": 1.77459716796875e-05,
      "step": 4652
    },
    {
      "action_loss_avg": 0.02421395571436733,
      "epoch": 1.77459716796875e-05,
      "step": 4652
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 4652,
      "training_step_time": 10.193476915359497
    },
    {
      "epoch": 1.77459716796875e-05,
      "model_forward_time": 3.1429872512817383,
      "step": 4652
    },
    {
      "epoch": 1.77459716796875e-05,
      "step": 4652,
      "training_step_time": 10.280763626098633
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 3.180267810821533,
      "step": 4653
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 4653,
      "training_step_time": 9.954102039337158
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 3.1367831230163574,
      "step": 4653
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 4653,
      "training_step_time": 9.913346290588379
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 3.1427152156829834,
      "step": 4653
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 4653,
      "training_step_time": 9.92042088508606
    },
    {
      "epoch": 1.7749786376953123e-05,
      "model_forward_time": 3.1370723247528076,
      "step": 4653
    },
    {
      "epoch": 1.7749786376953123e-05,
      "step": 4653,
      "training_step_time": 10.28356122970581
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 3.1919660568237305,
      "step": 4654
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 4654,
      "training_step_time": 9.96947431564331
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 3.1373579502105713,
      "step": 4654
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 4654,
      "training_step_time": 9.91408109664917
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 3.141723871231079,
      "step": 4654
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 4654,
      "training_step_time": 9.919260501861572
    },
    {
      "epoch": 1.775360107421875e-05,
      "model_forward_time": 3.1356005668640137,
      "step": 4654
    },
    {
      "epoch": 1.775360107421875e-05,
      "step": 4654,
      "training_step_time": 10.233934164047241
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 3.480818748474121,
      "step": 4655
    },
    {
      "dynamics_loss_avg": 0.07127673402428628,
      "epoch": 1.7757415771484376e-05,
      "step": 4655
    },
    {
      "action_loss_avg": 0.03566683139652014,
      "epoch": 1.7757415771484376e-05,
      "step": 4655
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 4655,
      "training_step_time": 10.262814998626709
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 2.4397668838500977,
      "step": 4655
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 4655,
      "training_step_time": 7.467407703399658
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 3.13864803314209,
      "step": 4655
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 4655,
      "training_step_time": 9.910355806350708
    },
    {
      "epoch": 1.7757415771484376e-05,
      "model_forward_time": 3.1347784996032715,
      "step": 4655
    },
    {
      "epoch": 1.7757415771484376e-05,
      "step": 4655,
      "training_step_time": 10.309922695159912
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 3.1636452674865723,
      "step": 4656
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 4656,
      "training_step_time": 9.934165239334106
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 3.1385316848754883,
      "step": 4656
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 4656,
      "training_step_time": 9.914846420288086
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 3.402216672897339,
      "step": 4656
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 4656,
      "training_step_time": 10.176684379577637
    },
    {
      "epoch": 1.776123046875e-05,
      "model_forward_time": 3.130758762359619,
      "step": 4656
    },
    {
      "epoch": 1.776123046875e-05,
      "step": 4656,
      "training_step_time": 10.016244411468506
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 3.1717960834503174,
      "step": 4657
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 4657,
      "training_step_time": 9.948131799697876
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 3.1411476135253906,
      "step": 4657
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 4657,
      "training_step_time": 9.91865587234497
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 3.4255573749542236,
      "step": 4657
    },
    {
      "dynamics_loss_avg": 0.06220770962536335,
      "epoch": 1.7765045166015625e-05,
      "step": 4657
    },
    {
      "action_loss_avg": 0.030335792060941457,
      "epoch": 1.7765045166015625e-05,
      "step": 4657
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 4657,
      "training_step_time": 10.206918478012085
    },
    {
      "epoch": 1.7765045166015625e-05,
      "model_forward_time": 3.141007423400879,
      "step": 4657
    },
    {
      "epoch": 1.7765045166015625e-05,
      "step": 4657,
      "training_step_time": 10.300297498703003
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 3.155689001083374,
      "step": 4658
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 4658,
      "training_step_time": 9.937580585479736
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 3.1440539360046387,
      "step": 4658
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 4658,
      "training_step_time": 9.921475172042847
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 2.438358783721924,
      "step": 4658
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 4658,
      "training_step_time": 7.466429710388184
    },
    {
      "epoch": 1.776885986328125e-05,
      "model_forward_time": 3.141800880432129,
      "step": 4658
    },
    {
      "epoch": 1.776885986328125e-05,
      "step": 4658,
      "training_step_time": 10.066729545593262
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 3.4638640880584717,
      "step": 4659
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 4659,
      "training_step_time": 10.243069887161255
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 3.140014171600342,
      "step": 4659
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 4659,
      "training_step_time": 9.9093599319458
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 3.1363613605499268,
      "step": 4659
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 4659,
      "training_step_time": 9.907734394073486
    },
    {
      "epoch": 1.7772674560546874e-05,
      "model_forward_time": 2.4371931552886963,
      "step": 4659
    },
    {
      "epoch": 1.7772674560546874e-05,
      "step": 4659,
      "training_step_time": 10.000735759735107
    },
    {
      "epoch": 1.77764892578125e-05,
      "grad_norm": 0.5216313004493713,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 4660
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 3.4532525539398193,
      "step": 4660
    },
    {
      "dynamics_loss_avg": 0.05425110310316086,
      "epoch": 1.77764892578125e-05,
      "step": 4660
    },
    {
      "action_loss_avg": 0.025877763424068688,
      "epoch": 1.77764892578125e-05,
      "step": 4660
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 4660,
      "training_step_time": 10.226060628890991
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 3.137704610824585,
      "step": 4660
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 4660,
      "training_step_time": 9.910118818283081
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 2.4394073486328125,
      "step": 4660
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 4660,
      "training_step_time": 7.470414161682129
    },
    {
      "epoch": 1.77764892578125e-05,
      "model_forward_time": 3.1383605003356934,
      "step": 4660
    },
    {
      "epoch": 1.77764892578125e-05,
      "step": 4660,
      "training_step_time": 10.281105995178223
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 3.1828346252441406,
      "step": 4661
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 4661,
      "training_step_time": 9.957122564315796
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 3.1336605548858643,
      "step": 4661
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 4661,
      "training_step_time": 9.90747356414795
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 3.37955641746521,
      "step": 4661
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 4661,
      "training_step_time": 10.151952505111694
    },
    {
      "epoch": 1.7780303955078123e-05,
      "model_forward_time": 3.1390278339385986,
      "step": 4661
    },
    {
      "epoch": 1.7780303955078123e-05,
      "step": 4661,
      "training_step_time": 10.295022964477539
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 3.1858816146850586,
      "step": 4662
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 4662,
      "training_step_time": 9.96164083480835
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 3.1371185779571533,
      "step": 4662
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 4662,
      "training_step_time": 9.910158395767212
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 3.1395576000213623,
      "step": 4662
    },
    {
      "dynamics_loss_avg": 0.05157963521778584,
      "epoch": 1.778411865234375e-05,
      "step": 4662
    },
    {
      "action_loss_avg": 0.024907912500202654,
      "epoch": 1.778411865234375e-05,
      "step": 4662
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 4662,
      "training_step_time": 10.197019815444946
    },
    {
      "epoch": 1.778411865234375e-05,
      "model_forward_time": 3.138455867767334,
      "step": 4662
    },
    {
      "epoch": 1.778411865234375e-05,
      "step": 4662,
      "training_step_time": 10.248512744903564
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 3.172715902328491,
      "step": 4663
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 4663,
      "training_step_time": 9.949362754821777
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 3.141505479812622,
      "step": 4663
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 4663,
      "training_step_time": 9.919191598892212
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 3.1375956535339355,
      "step": 4663
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 4663,
      "training_step_time": 9.913222551345825
    },
    {
      "epoch": 1.7787933349609376e-05,
      "model_forward_time": 3.4203100204467773,
      "step": 4663
    },
    {
      "epoch": 1.7787933349609376e-05,
      "step": 4663,
      "training_step_time": 10.290054321289062
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 3.1708505153656006,
      "step": 4664
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 4664,
      "training_step_time": 9.945313215255737
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 3.1361618041992188,
      "step": 4664
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 4664,
      "training_step_time": 9.913033246994019
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 3.140970230102539,
      "step": 4664
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 4664,
      "training_step_time": 9.917459726333618
    },
    {
      "epoch": 1.7791748046875e-05,
      "model_forward_time": 3.135373592376709,
      "step": 4664
    },
    {
      "epoch": 1.7791748046875e-05,
      "step": 4664,
      "training_step_time": 10.022843837738037
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 3.160133123397827,
      "step": 4665
    },
    {
      "dynamics_loss_avg": 0.06375017780810595,
      "epoch": 1.7795562744140625e-05,
      "step": 4665
    },
    {
      "action_loss_avg": 0.02456985116004944,
      "epoch": 1.7795562744140625e-05,
      "step": 4665
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 4665,
      "training_step_time": 10.21437692642212
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 3.1380908489227295,
      "step": 4665
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 4665,
      "training_step_time": 9.917129039764404
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 3.146601915359497,
      "step": 4665
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 4665,
      "training_step_time": 9.924596786499023
    },
    {
      "epoch": 1.7795562744140625e-05,
      "model_forward_time": 2.4385783672332764,
      "step": 4665
    },
    {
      "epoch": 1.7795562744140625e-05,
      "step": 4665,
      "training_step_time": 10.28201937675476
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 3.16727352142334,
      "step": 4666
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 4666,
      "training_step_time": 9.955006837844849
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 2.4348483085632324,
      "step": 4666
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 4666,
      "training_step_time": 7.469099283218384
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 3.4092040061950684,
      "step": 4666
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 4666,
      "training_step_time": 10.182051658630371
    },
    {
      "epoch": 1.779937744140625e-05,
      "model_forward_time": 3.13677716255188,
      "step": 4666
    },
    {
      "epoch": 1.779937744140625e-05,
      "step": 4666,
      "training_step_time": 10.301733255386353
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 3.1596837043762207,
      "step": 4667
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 4667,
      "training_step_time": 9.944063425064087
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 3.1384546756744385,
      "step": 4667
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 4667,
      "training_step_time": 9.911201477050781
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 2.434450149536133,
      "step": 4667
    },
    {
      "dynamics_loss_avg": 0.0656455798074603,
      "epoch": 1.7803192138671874e-05,
      "step": 4667
    },
    {
      "action_loss_avg": 0.040521867014467715,
      "epoch": 1.7803192138671874e-05,
      "step": 4667
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 4667,
      "training_step_time": 7.4718921184539795
    },
    {
      "epoch": 1.7803192138671874e-05,
      "model_forward_time": 3.3773128986358643,
      "step": 4667
    },
    {
      "epoch": 1.7803192138671874e-05,
      "step": 4667,
      "training_step_time": 10.278475999832153
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 3.1810643672943115,
      "step": 4668
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 4668,
      "training_step_time": 9.961673021316528
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 3.141390085220337,
      "step": 4668
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 4668,
      "training_step_time": 9.924748420715332
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 3.1384432315826416,
      "step": 4668
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 4668,
      "training_step_time": 9.919557571411133
    },
    {
      "epoch": 1.78070068359375e-05,
      "model_forward_time": 3.400848865509033,
      "step": 4668
    },
    {
      "epoch": 1.78070068359375e-05,
      "step": 4668,
      "training_step_time": 10.283947706222534
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 3.181830883026123,
      "step": 4669
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 4669,
      "training_step_time": 9.960206747055054
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 3.1406195163726807,
      "step": 4669
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 4669,
      "training_step_time": 9.915011405944824
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 3.141029119491577,
      "step": 4669
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 4669,
      "training_step_time": 9.915844202041626
    },
    {
      "epoch": 1.7810821533203123e-05,
      "model_forward_time": 3.1416175365448,
      "step": 4669
    },
    {
      "epoch": 1.7810821533203123e-05,
      "step": 4669,
      "training_step_time": 10.294599056243896
    },
    {
      "epoch": 1.781463623046875e-05,
      "grad_norm": 0.673571765422821,
      "learning_rate": 1e-05,
      "loss": 0.0949,
      "step": 4670
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 3.1782007217407227,
      "step": 4670
    },
    {
      "dynamics_loss_avg": 0.06463355980813504,
      "epoch": 1.781463623046875e-05,
      "step": 4670
    },
    {
      "action_loss_avg": 0.039585803635418415,
      "epoch": 1.781463623046875e-05,
      "step": 4670
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 4670,
      "training_step_time": 10.219599485397339
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 3.1417641639709473,
      "step": 4670
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 4670,
      "training_step_time": 9.920170307159424
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 3.137470245361328,
      "step": 4670
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 4670,
      "training_step_time": 9.919361114501953
    },
    {
      "epoch": 1.781463623046875e-05,
      "model_forward_time": 3.1424005031585693,
      "step": 4670
    },
    {
      "epoch": 1.781463623046875e-05,
      "step": 4670,
      "training_step_time": 10.283041715621948
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 3.1901357173919678,
      "step": 4671
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 4671,
      "training_step_time": 9.969213962554932
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 3.424588918685913,
      "step": 4671
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 4671,
      "training_step_time": 10.198871850967407
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 3.1345436573028564,
      "step": 4671
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 4671,
      "training_step_time": 9.909070253372192
    },
    {
      "epoch": 1.7818450927734376e-05,
      "model_forward_time": 3.1387126445770264,
      "step": 4671
    },
    {
      "epoch": 1.7818450927734376e-05,
      "step": 4671,
      "training_step_time": 10.274101495742798
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 3.1481542587280273,
      "step": 4672
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 4672,
      "training_step_time": 9.923466920852661
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 2.427819013595581,
      "step": 4672
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 4672,
      "training_step_time": 7.453262090682983
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 2.429832935333252,
      "step": 4672
    },
    {
      "dynamics_loss_avg": 0.04685363583266735,
      "epoch": 1.7822265625e-05,
      "step": 4672
    },
    {
      "action_loss_avg": 0.020904519781470298,
      "epoch": 1.7822265625e-05,
      "step": 4672
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 4672,
      "training_step_time": 7.742160320281982
    },
    {
      "epoch": 1.7822265625e-05,
      "model_forward_time": 3.1284213066101074,
      "step": 4672
    },
    {
      "epoch": 1.7822265625e-05,
      "step": 4672,
      "training_step_time": 10.302640914916992
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 3.1717841625213623,
      "step": 4673
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 4673,
      "training_step_time": 9.944499015808105
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 3.129793167114258,
      "step": 4673
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 4673,
      "training_step_time": 9.905419111251831
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 3.1309444904327393,
      "step": 4673
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 4673,
      "training_step_time": 9.90834093093872
    },
    {
      "epoch": 1.7826080322265625e-05,
      "model_forward_time": 3.411428689956665,
      "step": 4673
    },
    {
      "epoch": 1.7826080322265625e-05,
      "step": 4673,
      "training_step_time": 10.288722515106201
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 3.1472902297973633,
      "step": 4674
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 4674,
      "training_step_time": 9.924570083618164
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 3.138275623321533,
      "step": 4674
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 4674,
      "training_step_time": 9.92649793624878
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 2.4357118606567383,
      "step": 4674
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 4674,
      "training_step_time": 7.471548318862915
    },
    {
      "epoch": 1.782989501953125e-05,
      "model_forward_time": 3.125864028930664,
      "step": 4674
    },
    {
      "epoch": 1.782989501953125e-05,
      "step": 4674,
      "training_step_time": 10.287451028823853
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 3.4587202072143555,
      "step": 4675
    },
    {
      "dynamics_loss_avg": 0.05562841966748237,
      "epoch": 1.7833709716796874e-05,
      "step": 4675
    },
    {
      "action_loss_avg": 0.023677392955869438,
      "epoch": 1.7833709716796874e-05,
      "step": 4675
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 4675,
      "training_step_time": 10.23318886756897
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 2.434126138687134,
      "step": 4675
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 4675,
      "training_step_time": 7.46113657951355
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 3.124997138977051,
      "step": 4675
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 4675,
      "training_step_time": 9.893434762954712
    },
    {
      "epoch": 1.7833709716796874e-05,
      "model_forward_time": 3.1274573802948,
      "step": 4675
    },
    {
      "epoch": 1.7833709716796874e-05,
      "step": 4675,
      "training_step_time": 10.248548746109009
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 3.172245740890503,
      "step": 4676
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 4676,
      "training_step_time": 9.940809726715088
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 3.1311192512512207,
      "step": 4676
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 4676,
      "training_step_time": 9.901769399642944
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 3.403813123703003,
      "step": 4676
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 4676,
      "training_step_time": 10.178171634674072
    },
    {
      "epoch": 1.78375244140625e-05,
      "model_forward_time": 3.1341865062713623,
      "step": 4676
    },
    {
      "epoch": 1.78375244140625e-05,
      "step": 4676,
      "training_step_time": 10.278239727020264
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 3.1715807914733887,
      "step": 4677
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 4677,
      "training_step_time": 9.943537950515747
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 3.143362522125244,
      "step": 4677
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 4677,
      "training_step_time": 9.920556783676147
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 3.4054112434387207,
      "step": 4677
    },
    {
      "dynamics_loss_avg": 0.08127091387286782,
      "epoch": 1.7841339111328123e-05,
      "step": 4677
    },
    {
      "action_loss_avg": 0.03268330385908484,
      "epoch": 1.7841339111328123e-05,
      "step": 4677
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 4677,
      "training_step_time": 10.177487850189209
    },
    {
      "epoch": 1.7841339111328123e-05,
      "model_forward_time": 3.1289169788360596,
      "step": 4677
    },
    {
      "epoch": 1.7841339111328123e-05,
      "step": 4677,
      "training_step_time": 10.263654947280884
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 3.1827754974365234,
      "step": 4678
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 4678,
      "training_step_time": 9.95870065689087
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 3.1374168395996094,
      "step": 4678
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 4678,
      "training_step_time": 9.913063049316406
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 3.1325395107269287,
      "step": 4678
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 4678,
      "training_step_time": 9.906180381774902
    },
    {
      "epoch": 1.784515380859375e-05,
      "model_forward_time": 3.134389638900757,
      "step": 4678
    },
    {
      "epoch": 1.784515380859375e-05,
      "step": 4678,
      "training_step_time": 10.253771305084229
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 3.4575562477111816,
      "step": 4679
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 4679,
      "training_step_time": 10.228081464767456
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 3.1395680904388428,
      "step": 4679
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 4679,
      "training_step_time": 9.910358905792236
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 3.134850263595581,
      "step": 4679
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 4679,
      "training_step_time": 9.910047054290771
    },
    {
      "epoch": 1.7848968505859376e-05,
      "model_forward_time": 3.141864061355591,
      "step": 4679
    },
    {
      "epoch": 1.7848968505859376e-05,
      "step": 4679,
      "training_step_time": 10.271117925643921
    },
    {
      "epoch": 1.7852783203125e-05,
      "grad_norm": 0.6000580787658691,
      "learning_rate": 1e-05,
      "loss": 0.0884,
      "step": 4680
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 3.443988084793091,
      "step": 4680
    },
    {
      "dynamics_loss_avg": 0.06307897586375474,
      "epoch": 1.7852783203125e-05,
      "step": 4680
    },
    {
      "action_loss_avg": 0.03254474867135286,
      "epoch": 1.7852783203125e-05,
      "step": 4680
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 4680,
      "training_step_time": 10.216154098510742
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 3.135730266571045,
      "step": 4680
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 4680,
      "training_step_time": 9.901349306106567
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 2.4342615604400635,
      "step": 4680
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 4680,
      "training_step_time": 7.457148551940918
    },
    {
      "epoch": 1.7852783203125e-05,
      "model_forward_time": 3.1325337886810303,
      "step": 4680
    },
    {
      "epoch": 1.7852783203125e-05,
      "step": 4680,
      "training_step_time": 10.267072439193726
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 2.481264114379883,
      "step": 4681
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 4681,
      "training_step_time": 7.501460790634155
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 3.1283812522888184,
      "step": 4681
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 4681,
      "training_step_time": 9.89474368095398
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 3.380044460296631,
      "step": 4681
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 4681,
      "training_step_time": 10.148210048675537
    },
    {
      "epoch": 1.7856597900390625e-05,
      "model_forward_time": 3.133911609649658,
      "step": 4681
    },
    {
      "epoch": 1.7856597900390625e-05,
      "step": 4681,
      "training_step_time": 10.251102685928345
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 3.1541619300842285,
      "step": 4682
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 4682,
      "training_step_time": 9.928813219070435
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 2.4352309703826904,
      "step": 4682
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 4682,
      "training_step_time": 7.464651584625244
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 3.1377339363098145,
      "step": 4682
    },
    {
      "dynamics_loss_avg": 0.057531500421464446,
      "epoch": 1.786041259765625e-05,
      "step": 4682
    },
    {
      "action_loss_avg": 0.030242635123431682,
      "epoch": 1.786041259765625e-05,
      "step": 4682
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 4682,
      "training_step_time": 9.910694599151611
    },
    {
      "epoch": 1.786041259765625e-05,
      "model_forward_time": 2.6817831993103027,
      "step": 4682
    },
    {
      "epoch": 1.786041259765625e-05,
      "step": 4682,
      "training_step_time": 10.01694917678833
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 3.1819229125976562,
      "step": 4683
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 4683,
      "training_step_time": 9.951173782348633
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 3.1327972412109375,
      "step": 4683
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 4683,
      "training_step_time": 9.904062747955322
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 3.1359806060791016,
      "step": 4683
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 4683,
      "training_step_time": 9.912314176559448
    },
    {
      "epoch": 1.7864227294921874e-05,
      "model_forward_time": 3.1351773738861084,
      "step": 4683
    },
    {
      "epoch": 1.7864227294921874e-05,
      "step": 4683,
      "training_step_time": 10.296525239944458
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 2.7521634101867676,
      "step": 4684
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 4684,
      "training_step_time": 7.786821126937866
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 3.138136148452759,
      "step": 4684
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 4684,
      "training_step_time": 9.918063402175903
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 3.141514539718628,
      "step": 4684
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 4684,
      "training_step_time": 9.919061660766602
    },
    {
      "epoch": 1.78680419921875e-05,
      "model_forward_time": 2.433892250061035,
      "step": 4684
    },
    {
      "epoch": 1.78680419921875e-05,
      "step": 4684,
      "training_step_time": 10.282914638519287
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 3.157435655593872,
      "step": 4685
    },
    {
      "dynamics_loss_avg": 0.08701330237090588,
      "epoch": 1.7871856689453124e-05,
      "step": 4685
    },
    {
      "action_loss_avg": 0.03899189690127969,
      "epoch": 1.7871856689453124e-05,
      "step": 4685
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 4685,
      "training_step_time": 9.935173988342285
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 3.3809401988983154,
      "step": 4685
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 4685,
      "training_step_time": 10.151796340942383
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 3.136843204498291,
      "step": 4685
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 4685,
      "training_step_time": 9.90964412689209
    },
    {
      "epoch": 1.7871856689453124e-05,
      "model_forward_time": 2.429779291152954,
      "step": 4685
    },
    {
      "epoch": 1.7871856689453124e-05,
      "step": 4685,
      "training_step_time": 10.261247873306274
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 3.169851779937744,
      "step": 4686
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 4686,
      "training_step_time": 9.941261529922485
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 3.1338322162628174,
      "step": 4686
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 4686,
      "training_step_time": 9.903800010681152
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 3.4040892124176025,
      "step": 4686
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 4686,
      "training_step_time": 10.174667835235596
    },
    {
      "epoch": 1.787567138671875e-05,
      "model_forward_time": 3.13057279586792,
      "step": 4686
    },
    {
      "epoch": 1.787567138671875e-05,
      "step": 4686,
      "training_step_time": 10.26490044593811
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 3.166902542114258,
      "step": 4687
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 4687,
      "training_step_time": 9.937915325164795
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 3.1336259841918945,
      "step": 4687
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 4687,
      "training_step_time": 9.899341821670532
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 3.1302969455718994,
      "step": 4687
    },
    {
      "dynamics_loss_avg": 0.04763407185673714,
      "epoch": 1.7879486083984376e-05,
      "step": 4687
    },
    {
      "action_loss_avg": 0.023730663768947124,
      "epoch": 1.7879486083984376e-05,
      "step": 4687
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 4687,
      "training_step_time": 9.901625156402588
    },
    {
      "epoch": 1.7879486083984376e-05,
      "model_forward_time": 3.3751683235168457,
      "step": 4687
    },
    {
      "epoch": 1.7879486083984376e-05,
      "step": 4687,
      "training_step_time": 10.319332122802734
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 3.19474196434021,
      "step": 4688
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 4688,
      "training_step_time": 9.989342212677002
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 3.1363725662231445,
      "step": 4688
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 4688,
      "training_step_time": 9.916133880615234
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 3.1320958137512207,
      "step": 4688
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 4688,
      "training_step_time": 9.919269323348999
    },
    {
      "epoch": 1.788330078125e-05,
      "model_forward_time": 3.1336467266082764,
      "step": 4688
    },
    {
      "epoch": 1.788330078125e-05,
      "step": 4688,
      "training_step_time": 10.285123825073242
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 3.151479721069336,
      "step": 4689
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 4689,
      "training_step_time": 9.935247421264648
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 3.1380035877227783,
      "step": 4689
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 4689,
      "training_step_time": 9.918407201766968
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 2.4378533363342285,
      "step": 4689
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 4689,
      "training_step_time": 7.4774065017700195
    },
    {
      "epoch": 1.7887115478515625e-05,
      "model_forward_time": 3.1382343769073486,
      "step": 4689
    },
    {
      "epoch": 1.7887115478515625e-05,
      "step": 4689,
      "training_step_time": 10.030396461486816
    },
    {
      "epoch": 1.789093017578125e-05,
      "grad_norm": 0.3370085656642914,
      "learning_rate": 1e-05,
      "loss": 0.0899,
      "step": 4690
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 3.176872491836548,
      "step": 4690
    },
    {
      "dynamics_loss_avg": 0.04174194000661373,
      "epoch": 1.789093017578125e-05,
      "step": 4690
    },
    {
      "action_loss_avg": 0.020876576052978636,
      "epoch": 1.789093017578125e-05,
      "step": 4690
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 4690,
      "training_step_time": 9.963801860809326
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 3.419090747833252,
      "step": 4690
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 4690,
      "training_step_time": 10.199378252029419
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 3.1424360275268555,
      "step": 4690
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 4690,
      "training_step_time": 9.92436170578003
    },
    {
      "epoch": 1.789093017578125e-05,
      "model_forward_time": 3.147059440612793,
      "step": 4690
    },
    {
      "epoch": 1.789093017578125e-05,
      "step": 4690,
      "training_step_time": 10.289805173873901
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 3.182845115661621,
      "step": 4691
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 4691,
      "training_step_time": 9.968559741973877
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 3.14438533782959,
      "step": 4691
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 4691,
      "training_step_time": 10.226346254348755
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 3.1405045986175537,
      "step": 4691
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 4691,
      "training_step_time": 9.91876220703125
    },
    {
      "epoch": 1.7894744873046874e-05,
      "model_forward_time": 3.1424195766448975,
      "step": 4691
    },
    {
      "epoch": 1.7894744873046874e-05,
      "step": 4691,
      "training_step_time": 10.014389276504517
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 3.1889166831970215,
      "step": 4692
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 4692,
      "training_step_time": 9.96736764907837
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 3.1407124996185303,
      "step": 4692
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 4692,
      "training_step_time": 9.91633677482605
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 3.146317720413208,
      "step": 4692
    },
    {
      "dynamics_loss_avg": 0.05211781226098537,
      "epoch": 1.78985595703125e-05,
      "step": 4692
    },
    {
      "action_loss_avg": 0.030772599298506974,
      "epoch": 1.78985595703125e-05,
      "step": 4692
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 4692,
      "training_step_time": 10.209176540374756
    },
    {
      "epoch": 1.78985595703125e-05,
      "model_forward_time": 3.149632453918457,
      "step": 4692
    },
    {
      "epoch": 1.78985595703125e-05,
      "step": 4692,
      "training_step_time": 10.027036190032959
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 3.1523067951202393,
      "step": 4693
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 4693,
      "training_step_time": 9.93487024307251
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 3.1396539211273193,
      "step": 4693
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 4693,
      "training_step_time": 9.918061017990112
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 3.14316987991333,
      "step": 4693
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 4693,
      "training_step_time": 9.919838190078735
    },
    {
      "epoch": 1.7902374267578124e-05,
      "model_forward_time": 3.1411540508270264,
      "step": 4693
    },
    {
      "epoch": 1.7902374267578124e-05,
      "step": 4693,
      "training_step_time": 10.311768293380737
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 3.1835672855377197,
      "step": 4694
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 4694,
      "training_step_time": 9.971120119094849
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 3.1386594772338867,
      "step": 4694
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 4694,
      "training_step_time": 9.920759439468384
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 3.1422173976898193,
      "step": 4694
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 4694,
      "training_step_time": 9.924535751342773
    },
    {
      "epoch": 1.790618896484375e-05,
      "model_forward_time": 3.137641429901123,
      "step": 4694
    },
    {
      "epoch": 1.790618896484375e-05,
      "step": 4694,
      "training_step_time": 10.318292617797852
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 3.425997495651245,
      "step": 4695
    },
    {
      "dynamics_loss_avg": 0.061072833649814126,
      "epoch": 1.7910003662109376e-05,
      "step": 4695
    },
    {
      "action_loss_avg": 0.03126980494707823,
      "epoch": 1.7910003662109376e-05,
      "step": 4695
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 4695,
      "training_step_time": 10.206537246704102
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 2.4383888244628906,
      "step": 4695
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 4695,
      "training_step_time": 7.4726622104644775
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 2.4314587116241455,
      "step": 4695
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 4695,
      "training_step_time": 7.461153030395508
    },
    {
      "epoch": 1.7910003662109376e-05,
      "model_forward_time": 3.134427070617676,
      "step": 4695
    },
    {
      "epoch": 1.7910003662109376e-05,
      "step": 4695,
      "training_step_time": 10.298821926116943
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 3.170098304748535,
      "step": 4696
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 4696,
      "training_step_time": 9.963388442993164
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 3.152606964111328,
      "step": 4696
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 4696,
      "training_step_time": 9.928689002990723
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 3.4145610332489014,
      "step": 4696
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 4696,
      "training_step_time": 10.192055225372314
    },
    {
      "epoch": 1.7913818359375e-05,
      "model_forward_time": 3.133988380432129,
      "step": 4696
    },
    {
      "epoch": 1.7913818359375e-05,
      "step": 4696,
      "training_step_time": 10.289855003356934
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 3.1718780994415283,
      "step": 4697
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 4697,
      "training_step_time": 9.951829195022583
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 3.136958122253418,
      "step": 4697
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 4697,
      "training_step_time": 9.914453506469727
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 3.1349151134490967,
      "step": 4697
    },
    {
      "dynamics_loss_avg": 0.06716952919960022,
      "epoch": 1.7917633056640625e-05,
      "step": 4697
    },
    {
      "action_loss_avg": 0.024948872765526174,
      "epoch": 1.7917633056640625e-05,
      "step": 4697
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 4697,
      "training_step_time": 10.180768728256226
    },
    {
      "epoch": 1.7917633056640625e-05,
      "model_forward_time": 3.1307694911956787,
      "step": 4697
    },
    {
      "epoch": 1.7917633056640625e-05,
      "step": 4697,
      "training_step_time": 10.294984102249146
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 3.1752166748046875,
      "step": 4698
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 4698,
      "training_step_time": 9.952425241470337
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 3.1387829780578613,
      "step": 4698
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 4698,
      "training_step_time": 9.918604135513306
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 3.1305172443389893,
      "step": 4698
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 4698,
      "training_step_time": 9.90917420387268
    },
    {
      "epoch": 1.792144775390625e-05,
      "model_forward_time": 3.1390390396118164,
      "step": 4698
    },
    {
      "epoch": 1.792144775390625e-05,
      "step": 4698,
      "training_step_time": 10.293685913085938
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 3.1653194427490234,
      "step": 4699
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 4699,
      "training_step_time": 9.93767261505127
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 3.1325314044952393,
      "step": 4699
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 4699,
      "training_step_time": 9.907913446426392
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 3.1300556659698486,
      "step": 4699
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 4699,
      "training_step_time": 9.904863595962524
    },
    {
      "epoch": 1.7925262451171874e-05,
      "model_forward_time": 3.133446455001831,
      "step": 4699
    },
    {
      "epoch": 1.7925262451171874e-05,
      "step": 4699,
      "training_step_time": 10.00761866569519
    },
    {
      "epoch": 1.79290771484375e-05,
      "grad_norm": 0.5011589527130127,
      "learning_rate": 1e-05,
      "loss": 0.0902,
      "step": 4700
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 3.4250845909118652,
      "step": 4700
    },
    {
      "dynamics_loss_avg": 0.04680534601211548,
      "epoch": 1.79290771484375e-05,
      "step": 4700
    },
    {
      "action_loss_avg": 0.02100878208875656,
      "epoch": 1.79290771484375e-05,
      "step": 4700
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 4700,
      "training_step_time": 10.198389291763306
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 3.1377429962158203,
      "step": 4700
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 4700,
      "training_step_time": 9.906740188598633
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 3.133336067199707,
      "step": 4700
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 4700,
      "training_step_time": 9.901033639907837
    },
    {
      "epoch": 1.79290771484375e-05,
      "model_forward_time": 3.135207414627075,
      "step": 4700
    },
    {
      "epoch": 1.79290771484375e-05,
      "step": 4700,
      "training_step_time": 10.289279222488403
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 3.151305675506592,
      "step": 4701
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 4701,
      "training_step_time": 9.93047571182251
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 2.7095937728881836,
      "step": 4701
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 4701,
      "training_step_time": 7.737901926040649
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 2.431910514831543,
      "step": 4701
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 4701,
      "training_step_time": 7.457982778549194
    },
    {
      "epoch": 1.7932891845703124e-05,
      "model_forward_time": 2.433751344680786,
      "step": 4701
    },
    {
      "epoch": 1.7932891845703124e-05,
      "step": 4701,
      "training_step_time": 10.03506326675415
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 3.195307970046997,
      "step": 4702
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 4702,
      "training_step_time": 9.974640846252441
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 3.1381921768188477,
      "step": 4702
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 4702,
      "training_step_time": 9.919159412384033
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 3.1469056606292725,
      "step": 4702
    },
    {
      "dynamics_loss_avg": 0.055702174082398415,
      "epoch": 1.793670654296875e-05,
      "step": 4702
    },
    {
      "action_loss_avg": 0.02768757870653644,
      "epoch": 1.793670654296875e-05,
      "step": 4702
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 4702,
      "training_step_time": 9.934277772903442
    },
    {
      "epoch": 1.793670654296875e-05,
      "model_forward_time": 3.4113736152648926,
      "step": 4702
    },
    {
      "epoch": 1.793670654296875e-05,
      "step": 4702,
      "training_step_time": 10.29456615447998
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 3.180109977722168,
      "step": 4703
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 4703,
      "training_step_time": 9.960360527038574
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 3.1371500492095947,
      "step": 4703
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 4703,
      "training_step_time": 9.918540000915527
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 3.1452770233154297,
      "step": 4703
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 4703,
      "training_step_time": 9.925867557525635
    },
    {
      "epoch": 1.7940521240234376e-05,
      "model_forward_time": 3.4251227378845215,
      "step": 4703
    },
    {
      "epoch": 1.7940521240234376e-05,
      "step": 4703,
      "training_step_time": 10.320947170257568
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 3.1862363815307617,
      "step": 4704
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 4704,
      "training_step_time": 9.970791578292847
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 3.1390199661254883,
      "step": 4704
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 4704,
      "training_step_time": 9.922890663146973
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 3.1431708335876465,
      "step": 4704
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 4704,
      "training_step_time": 9.924571752548218
    },
    {
      "epoch": 1.79443359375e-05,
      "model_forward_time": 3.141645908355713,
      "step": 4704
    },
    {
      "epoch": 1.79443359375e-05,
      "step": 4704,
      "training_step_time": 10.027337551116943
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 3.177366018295288,
      "step": 4705
    },
    {
      "dynamics_loss_avg": 0.08531344197690487,
      "epoch": 1.7948150634765625e-05,
      "step": 4705
    },
    {
      "action_loss_avg": 0.04120574723929167,
      "epoch": 1.7948150634765625e-05,
      "step": 4705
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 4705,
      "training_step_time": 10.235349655151367
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 3.1385302543640137,
      "step": 4705
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 4705,
      "training_step_time": 9.918193578720093
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 3.140110492706299,
      "step": 4705
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 4705,
      "training_step_time": 9.916543960571289
    },
    {
      "epoch": 1.7948150634765625e-05,
      "model_forward_time": 3.133307456970215,
      "step": 4705
    },
    {
      "epoch": 1.7948150634765625e-05,
      "step": 4705,
      "training_step_time": 10.265671968460083
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 3.1883764266967773,
      "step": 4706
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 4706,
      "training_step_time": 9.965402841567993
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 1.7453749179840088,
      "step": 4706
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 4706,
      "training_step_time": 5.205155611038208
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 3.407853603363037,
      "step": 4706
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 4706,
      "training_step_time": 10.19649362564087
    },
    {
      "epoch": 1.795196533203125e-05,
      "model_forward_time": 3.1404967308044434,
      "step": 4706
    },
    {
      "epoch": 1.795196533203125e-05,
      "step": 4706,
      "training_step_time": 10.31485390663147
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 2.469501495361328,
      "step": 4707
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 4707,
      "training_step_time": 7.50385856628418
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 3.1366207599639893,
      "step": 4707
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 4707,
      "training_step_time": 9.917185068130493
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 3.139785051345825,
      "step": 4707
    },
    {
      "dynamics_loss_avg": 0.0533796489238739,
      "epoch": 1.7955780029296874e-05,
      "step": 4707
    },
    {
      "action_loss_avg": 0.028788065491244196,
      "epoch": 1.7955780029296874e-05,
      "step": 4707
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 4707,
      "training_step_time": 9.92411756515503
    },
    {
      "epoch": 1.7955780029296874e-05,
      "model_forward_time": 3.390183210372925,
      "step": 4707
    },
    {
      "epoch": 1.7955780029296874e-05,
      "step": 4707,
      "training_step_time": 10.304205656051636
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 3.2025601863861084,
      "step": 4708
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 4708,
      "training_step_time": 9.990321397781372
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 3.146103620529175,
      "step": 4708
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 4708,
      "training_step_time": 9.929373979568481
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 3.1416285037994385,
      "step": 4708
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 4708,
      "training_step_time": 9.922949314117432
    },
    {
      "epoch": 1.79595947265625e-05,
      "model_forward_time": 3.14384388923645,
      "step": 4708
    },
    {
      "epoch": 1.79595947265625e-05,
      "step": 4708,
      "training_step_time": 10.021941900253296
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 3.4708685874938965,
      "step": 4709
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 4709,
      "training_step_time": 10.247472763061523
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 3.140289306640625,
      "step": 4709
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 4709,
      "training_step_time": 9.919948816299438
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 3.1351828575134277,
      "step": 4709
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 4709,
      "training_step_time": 9.915247440338135
    },
    {
      "epoch": 1.7963409423828124e-05,
      "model_forward_time": 3.1362688541412354,
      "step": 4709
    },
    {
      "epoch": 1.7963409423828124e-05,
      "step": 4709,
      "training_step_time": 10.013278722763062
    },
    {
      "epoch": 1.796722412109375e-05,
      "grad_norm": 0.7511388659477234,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 4710
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 3.4629976749420166,
      "step": 4710
    },
    {
      "dynamics_loss_avg": 0.06311206761747598,
      "epoch": 1.796722412109375e-05,
      "step": 4710
    },
    {
      "action_loss_avg": 0.034812047891318795,
      "epoch": 1.796722412109375e-05,
      "step": 4710
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 4710,
      "training_step_time": 10.246294736862183
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 2.433852195739746,
      "step": 4710
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 4710,
      "training_step_time": 7.465956926345825
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 3.1295053958892822,
      "step": 4710
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 4710,
      "training_step_time": 9.90359377861023
    },
    {
      "epoch": 1.796722412109375e-05,
      "model_forward_time": 3.133601427078247,
      "step": 4710
    },
    {
      "epoch": 1.796722412109375e-05,
      "step": 4710,
      "training_step_time": 10.017780065536499
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 2.4762697219848633,
      "step": 4711
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 4711,
      "training_step_time": 7.5069708824157715
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 3.1280505657196045,
      "step": 4711
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 4711,
      "training_step_time": 9.904422283172607
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 3.405632257461548,
      "step": 4711
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 4711,
      "training_step_time": 10.185309410095215
    },
    {
      "epoch": 1.7971038818359376e-05,
      "model_forward_time": 3.1375417709350586,
      "step": 4711
    },
    {
      "epoch": 1.7971038818359376e-05,
      "step": 4711,
      "training_step_time": 10.017553091049194
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 3.189124584197998,
      "step": 4712
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 4712,
      "training_step_time": 9.969240665435791
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 2.4364383220672607,
      "step": 4712
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 4712,
      "training_step_time": 7.4728100299835205
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 3.1341211795806885,
      "step": 4712
    },
    {
      "dynamics_loss_avg": 0.07510327585041524,
      "epoch": 1.7974853515625e-05,
      "step": 4712
    },
    {
      "action_loss_avg": 0.03890125704929233,
      "epoch": 1.7974853515625e-05,
      "step": 4712
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 4712,
      "training_step_time": 9.913424015045166
    },
    {
      "epoch": 1.7974853515625e-05,
      "model_forward_time": 3.407726287841797,
      "step": 4712
    },
    {
      "epoch": 1.7974853515625e-05,
      "step": 4712,
      "training_step_time": 10.28078007698059
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 3.1876220703125,
      "step": 4713
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 4713,
      "training_step_time": 9.963351488113403
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 3.136265516281128,
      "step": 4713
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 4713,
      "training_step_time": 9.91030478477478
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 3.1397314071655273,
      "step": 4713
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 4713,
      "training_step_time": 9.919138431549072
    },
    {
      "epoch": 1.7978668212890625e-05,
      "model_forward_time": 3.1372575759887695,
      "step": 4713
    },
    {
      "epoch": 1.7978668212890625e-05,
      "step": 4713,
      "training_step_time": 10.310820817947388
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 3.475062608718872,
      "step": 4714
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 4714,
      "training_step_time": 10.255507707595825
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 3.1438732147216797,
      "step": 4714
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 4714,
      "training_step_time": 9.923269271850586
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 3.14874005317688,
      "step": 4714
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 4714,
      "training_step_time": 9.924914121627808
    },
    {
      "epoch": 1.798248291015625e-05,
      "model_forward_time": 2.4425740242004395,
      "step": 4714
    },
    {
      "epoch": 1.798248291015625e-05,
      "step": 4714,
      "training_step_time": 10.012875318527222
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 3.466454267501831,
      "step": 4715
    },
    {
      "dynamics_loss_avg": 0.06207403503358364,
      "epoch": 1.7986297607421874e-05,
      "step": 4715
    },
    {
      "action_loss_avg": 0.036241370160132645,
      "epoch": 1.7986297607421874e-05,
      "step": 4715
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 4715,
      "training_step_time": 10.251639604568481
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 3.1428966522216797,
      "step": 4715
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 4715,
      "training_step_time": 9.916956424713135
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 3.1442489624023438,
      "step": 4715
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 4715,
      "training_step_time": 9.920299053192139
    },
    {
      "epoch": 1.7986297607421874e-05,
      "model_forward_time": 3.1400675773620605,
      "step": 4715
    },
    {
      "epoch": 1.7986297607421874e-05,
      "step": 4715,
      "training_step_time": 10.247272729873657
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 3.178657054901123,
      "step": 4716
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 4716,
      "training_step_time": 9.95688772201538
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 3.1423354148864746,
      "step": 4716
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 4716,
      "training_step_time": 9.92201852798462
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 3.412581443786621,
      "step": 4716
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 4716,
      "training_step_time": 10.190224170684814
    },
    {
      "epoch": 1.79901123046875e-05,
      "model_forward_time": 2.435971260070801,
      "step": 4716
    },
    {
      "epoch": 1.79901123046875e-05,
      "step": 4716,
      "training_step_time": 10.239267349243164
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 3.1798059940338135,
      "step": 4717
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 4717,
      "training_step_time": 9.956605434417725
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 3.1417973041534424,
      "step": 4717
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 4717,
      "training_step_time": 9.92105507850647
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 3.1413626670837402,
      "step": 4717
    },
    {
      "dynamics_loss_avg": 0.053061773627996446,
      "epoch": 1.7993927001953124e-05,
      "step": 4717
    },
    {
      "action_loss_avg": 0.026620023418217897,
      "epoch": 1.7993927001953124e-05,
      "step": 4717
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 4717,
      "training_step_time": 10.208914756774902
    },
    {
      "epoch": 1.7993927001953124e-05,
      "model_forward_time": 3.141324043273926,
      "step": 4717
    },
    {
      "epoch": 1.7993927001953124e-05,
      "step": 4717,
      "training_step_time": 10.308696985244751
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 3.1700053215026855,
      "step": 4718
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 4718,
      "training_step_time": 9.953553676605225
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 3.146394968032837,
      "step": 4718
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 4718,
      "training_step_time": 9.924727439880371
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 3.1388330459594727,
      "step": 4718
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 4718,
      "training_step_time": 9.92071533203125
    },
    {
      "epoch": 1.799774169921875e-05,
      "model_forward_time": 3.1471781730651855,
      "step": 4718
    },
    {
      "epoch": 1.799774169921875e-05,
      "step": 4718,
      "training_step_time": 10.299283266067505
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 3.1790735721588135,
      "step": 4719
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 4719,
      "training_step_time": 9.959984064102173
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 3.1449472904205322,
      "step": 4719
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 4719,
      "training_step_time": 9.932009935379028
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 3.143258571624756,
      "step": 4719
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 4719,
      "training_step_time": 9.924980878829956
    },
    {
      "epoch": 1.8001556396484376e-05,
      "model_forward_time": 3.1419756412506104,
      "step": 4719
    },
    {
      "epoch": 1.8001556396484376e-05,
      "step": 4719,
      "training_step_time": 10.02609896659851
    },
    {
      "epoch": 1.800537109375e-05,
      "grad_norm": 0.6023716330528259,
      "learning_rate": 1e-05,
      "loss": 0.0923,
      "step": 4720
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 3.18215274810791,
      "step": 4720
    },
    {
      "dynamics_loss_avg": 0.06621916983276606,
      "epoch": 1.800537109375e-05,
      "step": 4720
    },
    {
      "action_loss_avg": 0.031610967591404915,
      "epoch": 1.800537109375e-05,
      "step": 4720
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 4720,
      "training_step_time": 10.26917314529419
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 3.1470589637756348,
      "step": 4720
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 4720,
      "training_step_time": 9.929369926452637
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 3.1436402797698975,
      "step": 4720
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 4720,
      "training_step_time": 9.927252531051636
    },
    {
      "epoch": 1.800537109375e-05,
      "model_forward_time": 3.147017478942871,
      "step": 4720
    },
    {
      "epoch": 1.800537109375e-05,
      "step": 4720,
      "training_step_time": 10.027125597000122
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 3.189362049102783,
      "step": 4721
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 4721,
      "training_step_time": 9.97367787361145
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 2.4378397464752197,
      "step": 4721
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 4721,
      "training_step_time": 7.7445056438446045
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 3.1341559886932373,
      "step": 4721
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 4721,
      "training_step_time": 9.911561250686646
    },
    {
      "epoch": 1.8009185791015625e-05,
      "model_forward_time": 3.139782190322876,
      "step": 4721
    },
    {
      "epoch": 1.8009185791015625e-05,
      "step": 4721,
      "training_step_time": 10.271203517913818
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 3.191382884979248,
      "step": 4722
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 4722,
      "training_step_time": 9.9749174118042
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 3.136622667312622,
      "step": 4722
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 4722,
      "training_step_time": 9.915127515792847
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 3.137338399887085,
      "step": 4722
    },
    {
      "dynamics_loss_avg": 0.051376554928719995,
      "epoch": 1.801300048828125e-05,
      "step": 4722
    },
    {
      "action_loss_avg": 0.02471370152197778,
      "epoch": 1.801300048828125e-05,
      "step": 4722
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 4722,
      "training_step_time": 10.194806814193726
    },
    {
      "epoch": 1.801300048828125e-05,
      "model_forward_time": 3.137571334838867,
      "step": 4722
    },
    {
      "epoch": 1.801300048828125e-05,
      "step": 4722,
      "training_step_time": 10.011484146118164
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 2.4707493782043457,
      "step": 4723
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 4723,
      "training_step_time": 7.50217866897583
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 3.1305177211761475,
      "step": 4723
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 4723,
      "training_step_time": 9.900453805923462
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 3.1340317726135254,
      "step": 4723
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 4723,
      "training_step_time": 9.908827543258667
    },
    {
      "epoch": 1.8016815185546874e-05,
      "model_forward_time": 3.421556234359741,
      "step": 4723
    },
    {
      "epoch": 1.8016815185546874e-05,
      "step": 4723,
      "training_step_time": 10.289214849472046
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 3.1749565601348877,
      "step": 4724
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 4724,
      "training_step_time": 9.951075553894043
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 3.136892557144165,
      "step": 4724
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 4724,
      "training_step_time": 9.91225528717041
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 3.139716625213623,
      "step": 4724
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 4724,
      "training_step_time": 9.911290645599365
    },
    {
      "epoch": 1.80206298828125e-05,
      "model_forward_time": 3.1341679096221924,
      "step": 4724
    },
    {
      "epoch": 1.80206298828125e-05,
      "step": 4724,
      "training_step_time": 10.273473024368286
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 2.4631381034851074,
      "step": 4725
    },
    {
      "dynamics_loss_avg": 0.06821517162024975,
      "epoch": 1.8024444580078124e-05,
      "step": 4725
    },
    {
      "action_loss_avg": 0.03893435364589095,
      "epoch": 1.8024444580078124e-05,
      "step": 4725
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 4725,
      "training_step_time": 7.503436803817749
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 3.3811533451080322,
      "step": 4725
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 4725,
      "training_step_time": 10.15433144569397
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 3.13550066947937,
      "step": 4725
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 4725,
      "training_step_time": 9.914206504821777
    },
    {
      "epoch": 1.8024444580078124e-05,
      "model_forward_time": 3.1357266902923584,
      "step": 4725
    },
    {
      "epoch": 1.8024444580078124e-05,
      "step": 4725,
      "training_step_time": 10.29673171043396
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 3.174076557159424,
      "step": 4726
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 4726,
      "training_step_time": 9.95212173461914
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 3.3874671459198,
      "step": 4726
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 4726,
      "training_step_time": 10.16776442527771
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 3.137660264968872,
      "step": 4726
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 4726,
      "training_step_time": 9.914018154144287
    },
    {
      "epoch": 1.802825927734375e-05,
      "model_forward_time": 3.1358301639556885,
      "step": 4726
    },
    {
      "epoch": 1.802825927734375e-05,
      "step": 4726,
      "training_step_time": 10.022752523422241
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 3.153494119644165,
      "step": 4727
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 4727,
      "training_step_time": 9.935161113739014
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 3.1331653594970703,
      "step": 4727
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 4727,
      "training_step_time": 9.909611463546753
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 2.429067373275757,
      "step": 4727
    },
    {
      "dynamics_loss_avg": 0.07443308718502521,
      "epoch": 1.8032073974609376e-05,
      "step": 4727
    },
    {
      "action_loss_avg": 0.03735697101801634,
      "epoch": 1.8032073974609376e-05,
      "step": 4727
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 4727,
      "training_step_time": 7.459931135177612
    },
    {
      "epoch": 1.8032073974609376e-05,
      "model_forward_time": 3.3825881481170654,
      "step": 4727
    },
    {
      "epoch": 1.8032073974609376e-05,
      "step": 4727,
      "training_step_time": 10.27378249168396
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 2.4398677349090576,
      "step": 4728
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 4728,
      "training_step_time": 7.467265844345093
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 3.131013870239258,
      "step": 4728
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 4728,
      "training_step_time": 9.903600692749023
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 3.1261579990386963,
      "step": 4728
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 4728,
      "training_step_time": 9.897145509719849
    },
    {
      "epoch": 1.8035888671875e-05,
      "model_forward_time": 2.4293324947357178,
      "step": 4728
    },
    {
      "epoch": 1.8035888671875e-05,
      "step": 4728,
      "training_step_time": 10.283962488174438
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 3.4487080574035645,
      "step": 4729
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 4729,
      "training_step_time": 10.222803115844727
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 3.141868829727173,
      "step": 4729
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 4729,
      "training_step_time": 9.91456151008606
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 2.4379923343658447,
      "step": 4729
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 4729,
      "training_step_time": 7.472993850708008
    },
    {
      "epoch": 1.8039703369140625e-05,
      "model_forward_time": 3.139343738555908,
      "step": 4729
    },
    {
      "epoch": 1.8039703369140625e-05,
      "step": 4729,
      "training_step_time": 10.285980939865112
    },
    {
      "epoch": 1.804351806640625e-05,
      "grad_norm": 0.6110652685165405,
      "learning_rate": 1e-05,
      "loss": 0.0942,
      "step": 4730
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 2.4782726764678955,
      "step": 4730
    },
    {
      "dynamics_loss_avg": 0.05119585786014795,
      "epoch": 1.804351806640625e-05,
      "step": 4730
    },
    {
      "action_loss_avg": 0.01972896815277636,
      "epoch": 1.804351806640625e-05,
      "step": 4730
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 4730,
      "training_step_time": 7.5111236572265625
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 3.40179705619812,
      "step": 4730
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 4730,
      "training_step_time": 10.178932666778564
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 3.1406540870666504,
      "step": 4730
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 4730,
      "training_step_time": 9.923471927642822
    },
    {
      "epoch": 1.804351806640625e-05,
      "model_forward_time": 3.1458191871643066,
      "step": 4730
    },
    {
      "epoch": 1.804351806640625e-05,
      "step": 4730,
      "training_step_time": 10.021955728530884
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 3.1604251861572266,
      "step": 4731
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 4731,
      "training_step_time": 9.9479660987854
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 3.1429378986358643,
      "step": 4731
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 4731,
      "training_step_time": 9.92649221420288
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 3.38854718208313,
      "step": 4731
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 4731,
      "training_step_time": 10.172127485275269
    },
    {
      "epoch": 1.8047332763671875e-05,
      "model_forward_time": 2.4386353492736816,
      "step": 4731
    },
    {
      "epoch": 1.8047332763671875e-05,
      "step": 4731,
      "training_step_time": 10.027115106582642
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 3.1655054092407227,
      "step": 4732
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 4732,
      "training_step_time": 9.946498155593872
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 2.431994676589966,
      "step": 4732
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 4732,
      "training_step_time": 7.466994285583496
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 3.1323978900909424,
      "step": 4732
    },
    {
      "dynamics_loss_avg": 0.06630587410181761,
      "epoch": 1.80511474609375e-05,
      "step": 4732
    },
    {
      "action_loss_avg": 0.04588240692391991,
      "epoch": 1.80511474609375e-05,
      "step": 4732
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 4732,
      "training_step_time": 9.91059136390686
    },
    {
      "epoch": 1.80511474609375e-05,
      "model_forward_time": 3.4141650199890137,
      "step": 4732
    },
    {
      "epoch": 1.80511474609375e-05,
      "step": 4732,
      "training_step_time": 10.288377046585083
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 2.476257801055908,
      "step": 4733
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 4733,
      "training_step_time": 7.508495569229126
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 3.135834217071533,
      "step": 4733
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 4733,
      "training_step_time": 9.915390968322754
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 3.1438148021698,
      "step": 4733
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 4733,
      "training_step_time": 9.93038272857666
    },
    {
      "epoch": 1.8054962158203124e-05,
      "model_forward_time": 3.141479015350342,
      "step": 4733
    },
    {
      "epoch": 1.8054962158203124e-05,
      "step": 4733,
      "training_step_time": 10.304992198944092
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 3.4745824337005615,
      "step": 4734
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 4734,
      "training_step_time": 10.256579160690308
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 3.1393582820892334,
      "step": 4734
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 4734,
      "training_step_time": 9.915672063827515
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 3.1436500549316406,
      "step": 4734
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 4734,
      "training_step_time": 9.920355081558228
    },
    {
      "epoch": 1.805877685546875e-05,
      "model_forward_time": 3.1437489986419678,
      "step": 4734
    },
    {
      "epoch": 1.805877685546875e-05,
      "step": 4734,
      "training_step_time": 10.302022218704224
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 3.1649818420410156,
      "step": 4735
    },
    {
      "dynamics_loss_avg": 0.06345054768025875,
      "epoch": 1.8062591552734376e-05,
      "step": 4735
    },
    {
      "action_loss_avg": 0.035589448362588885,
      "epoch": 1.8062591552734376e-05,
      "step": 4735
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 4735,
      "training_step_time": 9.943176031112671
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 3.394296884536743,
      "step": 4735
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 4735,
      "training_step_time": 10.16562819480896
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 3.136528491973877,
      "step": 4735
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 4735,
      "training_step_time": 9.90335988998413
    },
    {
      "epoch": 1.8062591552734376e-05,
      "model_forward_time": 3.135789394378662,
      "step": 4735
    },
    {
      "epoch": 1.8062591552734376e-05,
      "step": 4735,
      "training_step_time": 10.292633295059204
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 3.143462657928467,
      "step": 4736
    },
    {
      "epoch": 1.806640625e-05,
      "step": 4736,
      "training_step_time": 9.921526432037354
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 3.1415140628814697,
      "step": 4736
    },
    {
      "epoch": 1.806640625e-05,
      "step": 4736,
      "training_step_time": 9.907217741012573
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 2.708650827407837,
      "step": 4736
    },
    {
      "epoch": 1.806640625e-05,
      "step": 4736,
      "training_step_time": 7.730459451675415
    },
    {
      "epoch": 1.806640625e-05,
      "model_forward_time": 3.1385366916656494,
      "step": 4736
    },
    {
      "epoch": 1.806640625e-05,
      "step": 4736,
      "training_step_time": 10.012150049209595
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 3.1827573776245117,
      "step": 4737
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 4737,
      "training_step_time": 9.95374083518982
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 3.137824058532715,
      "step": 4737
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 4737,
      "training_step_time": 9.903212070465088
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 3.127845287322998,
      "step": 4737
    },
    {
      "dynamics_loss_avg": 0.06051918584853411,
      "epoch": 1.8070220947265625e-05,
      "step": 4737
    },
    {
      "action_loss_avg": 0.027361750323325395,
      "epoch": 1.8070220947265625e-05,
      "step": 4737
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 4737,
      "training_step_time": 10.169753551483154
    },
    {
      "epoch": 1.8070220947265625e-05,
      "model_forward_time": 3.1360762119293213,
      "step": 4737
    },
    {
      "epoch": 1.8070220947265625e-05,
      "step": 4737,
      "training_step_time": 10.276506662368774
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 3.153632640838623,
      "step": 4738
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 4738,
      "training_step_time": 9.919731616973877
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 3.1366491317749023,
      "step": 4738
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 4738,
      "training_step_time": 9.908116817474365
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 2.4323153495788574,
      "step": 4738
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 4738,
      "training_step_time": 7.459264755249023
    },
    {
      "epoch": 1.807403564453125e-05,
      "model_forward_time": 2.4272301197052,
      "step": 4738
    },
    {
      "epoch": 1.807403564453125e-05,
      "step": 4738,
      "training_step_time": 10.3148512840271
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 3.457873821258545,
      "step": 4739
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 4739,
      "training_step_time": 10.230839967727661
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 2.4335923194885254,
      "step": 4739
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 4739,
      "training_step_time": 7.463911533355713
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 3.131603717803955,
      "step": 4739
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 4739,
      "training_step_time": 9.911874055862427
    },
    {
      "epoch": 1.8077850341796875e-05,
      "model_forward_time": 2.433743953704834,
      "step": 4739
    },
    {
      "epoch": 1.8077850341796875e-05,
      "step": 4739,
      "training_step_time": 10.354023456573486
    },
    {
      "epoch": 1.80816650390625e-05,
      "grad_norm": 0.43245410919189453,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 4740
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 2.4792850017547607,
      "step": 4740
    },
    {
      "dynamics_loss_avg": 0.040229678992182015,
      "epoch": 1.80816650390625e-05,
      "step": 4740
    },
    {
      "action_loss_avg": 0.018340656370855867,
      "epoch": 1.80816650390625e-05,
      "step": 4740
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 4740,
      "training_step_time": 7.509187459945679
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 3.425537109375,
      "step": 4740
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 4740,
      "training_step_time": 10.199352979660034
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 3.14314341545105,
      "step": 4740
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 4740,
      "training_step_time": 9.914130210876465
    },
    {
      "epoch": 1.80816650390625e-05,
      "model_forward_time": 3.147944450378418,
      "step": 4740
    },
    {
      "epoch": 1.80816650390625e-05,
      "step": 4740,
      "training_step_time": 10.281519651412964
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 3.20129132270813,
      "step": 4741
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 4741,
      "training_step_time": 9.974447965621948
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 3.1404669284820557,
      "step": 4741
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 4741,
      "training_step_time": 9.912574529647827
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 3.39218807220459,
      "step": 4741
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 4741,
      "training_step_time": 10.16478443145752
    },
    {
      "epoch": 1.8085479736328124e-05,
      "model_forward_time": 3.1468355655670166,
      "step": 4741
    },
    {
      "epoch": 1.8085479736328124e-05,
      "step": 4741,
      "training_step_time": 10.31184196472168
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 3.1861584186553955,
      "step": 4742
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 4742,
      "training_step_time": 9.960197448730469
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 3.1428091526031494,
      "step": 4742
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 4742,
      "training_step_time": 9.912353754043579
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 3.148012161254883,
      "step": 4742
    },
    {
      "dynamics_loss_avg": 0.06095119006931782,
      "epoch": 1.808929443359375e-05,
      "step": 4742
    },
    {
      "action_loss_avg": 0.02912330823019147,
      "epoch": 1.808929443359375e-05,
      "step": 4742
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 4742,
      "training_step_time": 9.92460298538208
    },
    {
      "epoch": 1.808929443359375e-05,
      "model_forward_time": 3.419609308242798,
      "step": 4742
    },
    {
      "epoch": 1.808929443359375e-05,
      "step": 4742,
      "training_step_time": 10.288142681121826
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 3.1585047245025635,
      "step": 4743
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 4743,
      "training_step_time": 9.935442209243774
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 2.441311836242676,
      "step": 4743
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 4743,
      "training_step_time": 7.469395875930786
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 2.4359259605407715,
      "step": 4743
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 4743,
      "training_step_time": 7.464556455612183
    },
    {
      "epoch": 1.8093109130859376e-05,
      "model_forward_time": 3.1361701488494873,
      "step": 4743
    },
    {
      "epoch": 1.8093109130859376e-05,
      "step": 4743,
      "training_step_time": 10.014199495315552
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 3.4623215198516846,
      "step": 4744
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 4744,
      "training_step_time": 10.223863124847412
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 3.1333670616149902,
      "step": 4744
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 4744,
      "training_step_time": 9.896115064620972
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 3.1315557956695557,
      "step": 4744
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 4744,
      "training_step_time": 9.905747652053833
    },
    {
      "epoch": 1.8096923828125e-05,
      "model_forward_time": 3.1311442852020264,
      "step": 4744
    },
    {
      "epoch": 1.8096923828125e-05,
      "step": 4744,
      "training_step_time": 10.276615858078003
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 3.1682252883911133,
      "step": 4745
    },
    {
      "dynamics_loss_avg": 0.049378047604113816,
      "epoch": 1.8100738525390625e-05,
      "step": 4745
    },
    {
      "action_loss_avg": 0.03190910797566175,
      "epoch": 1.8100738525390625e-05,
      "step": 4745
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 4745,
      "training_step_time": 9.938577890396118
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 2.6799509525299072,
      "step": 4745
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 4745,
      "training_step_time": 7.703593730926514
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 3.1310126781463623,
      "step": 4745
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 4745,
      "training_step_time": 9.897265672683716
    },
    {
      "epoch": 1.8100738525390625e-05,
      "model_forward_time": 3.1302459239959717,
      "step": 4745
    },
    {
      "epoch": 1.8100738525390625e-05,
      "step": 4745,
      "training_step_time": 10.294801235198975
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 3.179945468902588,
      "step": 4746
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 4746,
      "training_step_time": 9.961247205734253
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 3.134335517883301,
      "step": 4746
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 4746,
      "training_step_time": 9.900733232498169
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 3.4007511138916016,
      "step": 4746
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 4746,
      "training_step_time": 10.168964147567749
    },
    {
      "epoch": 1.810455322265625e-05,
      "model_forward_time": 3.132568836212158,
      "step": 4746
    },
    {
      "epoch": 1.810455322265625e-05,
      "step": 4746,
      "training_step_time": 10.278724431991577
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 3.1847517490386963,
      "step": 4747
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 4747,
      "training_step_time": 9.957706689834595
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 3.134111166000366,
      "step": 4747
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 4747,
      "training_step_time": 9.913199424743652
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 3.134314775466919,
      "step": 4747
    },
    {
      "dynamics_loss_avg": 0.08195649236440658,
      "epoch": 1.8108367919921875e-05,
      "step": 4747
    },
    {
      "action_loss_avg": 0.032149313762784,
      "epoch": 1.8108367919921875e-05,
      "step": 4747
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 4747,
      "training_step_time": 9.907821893692017
    },
    {
      "epoch": 1.8108367919921875e-05,
      "model_forward_time": 2.6831820011138916,
      "step": 4747
    },
    {
      "epoch": 1.8108367919921875e-05,
      "step": 4747,
      "training_step_time": 10.255249261856079
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 3.173584222793579,
      "step": 4748
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 4748,
      "training_step_time": 9.947200775146484
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 3.138695240020752,
      "step": 4748
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 4748,
      "training_step_time": 9.911609411239624
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 3.133538246154785,
      "step": 4748
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 4748,
      "training_step_time": 9.903276205062866
    },
    {
      "epoch": 1.81121826171875e-05,
      "model_forward_time": 3.139904737472534,
      "step": 4748
    },
    {
      "epoch": 1.81121826171875e-05,
      "step": 4748,
      "training_step_time": 10.3121497631073
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 3.5773332118988037,
      "step": 4749
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 4749,
      "training_step_time": 10.346316576004028
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 3.14684796333313,
      "step": 4749
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 4749,
      "training_step_time": 9.915891408920288
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 3.144131898880005,
      "step": 4749
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 4749,
      "training_step_time": 9.91038203239441
    },
    {
      "epoch": 1.8115997314453124e-05,
      "model_forward_time": 3.1443660259246826,
      "step": 4749
    },
    {
      "epoch": 1.8115997314453124e-05,
      "step": 4749,
      "training_step_time": 10.230215311050415
    },
    {
      "epoch": 1.811981201171875e-05,
      "grad_norm": 0.6546267867088318,
      "learning_rate": 1e-05,
      "loss": 0.0901,
      "step": 4750
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 3.4688472747802734,
      "step": 4750
    },
    {
      "dynamics_loss_avg": 0.06894662417471409,
      "epoch": 1.811981201171875e-05,
      "step": 4750
    },
    {
      "action_loss_avg": 0.029373092018067835,
      "epoch": 1.811981201171875e-05,
      "step": 4750
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 4750,
      "training_step_time": 10.244443416595459
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 3.1367924213409424,
      "step": 4750
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 4750,
      "training_step_time": 9.903077602386475
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 3.1322836875915527,
      "step": 4750
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 4750,
      "training_step_time": 9.901880502700806
    },
    {
      "epoch": 1.811981201171875e-05,
      "model_forward_time": 3.1335883140563965,
      "step": 4750
    },
    {
      "epoch": 1.811981201171875e-05,
      "step": 4750,
      "training_step_time": 10.315761089324951
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 3.1765573024749756,
      "step": 4751
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 4751,
      "training_step_time": 9.947356462478638
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 3.132225513458252,
      "step": 4751
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 4751,
      "training_step_time": 10.171226739883423
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 3.132146120071411,
      "step": 4751
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 4751,
      "training_step_time": 9.901131391525269
    },
    {
      "epoch": 1.8123626708984376e-05,
      "model_forward_time": 3.1350343227386475,
      "step": 4751
    },
    {
      "epoch": 1.8123626708984376e-05,
      "step": 4751,
      "training_step_time": 10.02038311958313
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 3.16145396232605,
      "step": 4752
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 4752,
      "training_step_time": 9.932709455490112
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 3.1341207027435303,
      "step": 4752
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 4752,
      "training_step_time": 9.909857988357544
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 3.3944547176361084,
      "step": 4752
    },
    {
      "dynamics_loss_avg": 0.06634453013539314,
      "epoch": 1.812744140625e-05,
      "step": 4752
    },
    {
      "action_loss_avg": 0.03283601077273488,
      "epoch": 1.812744140625e-05,
      "step": 4752
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 4752,
      "training_step_time": 10.171396493911743
    },
    {
      "epoch": 1.812744140625e-05,
      "model_forward_time": 3.136197566986084,
      "step": 4752
    },
    {
      "epoch": 1.812744140625e-05,
      "step": 4752,
      "training_step_time": 10.28178882598877
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 3.1698927879333496,
      "step": 4753
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 4753,
      "training_step_time": 9.951052904129028
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 3.138859748840332,
      "step": 4753
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 4753,
      "training_step_time": 9.9170663356781
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 3.139477252960205,
      "step": 4753
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 4753,
      "training_step_time": 9.920460224151611
    },
    {
      "epoch": 1.8131256103515625e-05,
      "model_forward_time": 3.1389353275299072,
      "step": 4753
    },
    {
      "epoch": 1.8131256103515625e-05,
      "step": 4753,
      "training_step_time": 10.335499286651611
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 3.178069829940796,
      "step": 4754
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 4754,
      "training_step_time": 9.956561803817749
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 3.1383402347564697,
      "step": 4754
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 4754,
      "training_step_time": 9.916605949401855
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 3.137855052947998,
      "step": 4754
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 4754,
      "training_step_time": 9.910969257354736
    },
    {
      "epoch": 1.813507080078125e-05,
      "model_forward_time": 3.1352379322052,
      "step": 4754
    },
    {
      "epoch": 1.813507080078125e-05,
      "step": 4754,
      "training_step_time": 10.013926029205322
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 3.462714195251465,
      "step": 4755
    },
    {
      "dynamics_loss_avg": 0.06386765390634537,
      "epoch": 1.8138885498046875e-05,
      "step": 4755
    },
    {
      "action_loss_avg": 0.03857789123430848,
      "epoch": 1.8138885498046875e-05,
      "step": 4755
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 4755,
      "training_step_time": 10.239571809768677
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 3.134989023208618,
      "step": 4755
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 4755,
      "training_step_time": 9.907958507537842
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 3.139509916305542,
      "step": 4755
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 4755,
      "training_step_time": 9.916514158248901
    },
    {
      "epoch": 1.8138885498046875e-05,
      "model_forward_time": 3.136657476425171,
      "step": 4755
    },
    {
      "epoch": 1.8138885498046875e-05,
      "step": 4755,
      "training_step_time": 10.265912294387817
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 3.166182041168213,
      "step": 4756
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 4756,
      "training_step_time": 9.945121049880981
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 3.3895342350006104,
      "step": 4756
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 4756,
      "training_step_time": 10.171318769454956
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 3.138847827911377,
      "step": 4756
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 4756,
      "training_step_time": 9.920959711074829
    },
    {
      "epoch": 1.81427001953125e-05,
      "model_forward_time": 3.1369614601135254,
      "step": 4756
    },
    {
      "epoch": 1.81427001953125e-05,
      "step": 4756,
      "training_step_time": 10.246070146560669
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 3.185960531234741,
      "step": 4757
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 4757,
      "training_step_time": 9.972728490829468
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 3.144043445587158,
      "step": 4757
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 4757,
      "training_step_time": 9.922983407974243
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 3.4238440990448,
      "step": 4757
    },
    {
      "dynamics_loss_avg": 0.05576799977570772,
      "epoch": 1.8146514892578124e-05,
      "step": 4757
    },
    {
      "action_loss_avg": 0.02760760299861431,
      "epoch": 1.8146514892578124e-05,
      "step": 4757
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 4757,
      "training_step_time": 10.207448244094849
    },
    {
      "epoch": 1.8146514892578124e-05,
      "model_forward_time": 3.1399292945861816,
      "step": 4757
    },
    {
      "epoch": 1.8146514892578124e-05,
      "step": 4757,
      "training_step_time": 10.025143146514893
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 3.1856162548065186,
      "step": 4758
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 4758,
      "training_step_time": 9.96611762046814
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 3.1430397033691406,
      "step": 4758
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 4758,
      "training_step_time": 9.925138711929321
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 3.139904737472534,
      "step": 4758
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 4758,
      "training_step_time": 9.923158407211304
    },
    {
      "epoch": 1.815032958984375e-05,
      "model_forward_time": 3.3931479454040527,
      "step": 4758
    },
    {
      "epoch": 1.815032958984375e-05,
      "step": 4758,
      "training_step_time": 10.302677392959595
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 3.195422887802124,
      "step": 4759
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 4759,
      "training_step_time": 9.977680444717407
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 3.1433498859405518,
      "step": 4759
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 4759,
      "training_step_time": 9.920346736907959
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 3.137151002883911,
      "step": 4759
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 4759,
      "training_step_time": 9.911154508590698
    },
    {
      "epoch": 1.8154144287109376e-05,
      "model_forward_time": 3.1371874809265137,
      "step": 4759
    },
    {
      "epoch": 1.8154144287109376e-05,
      "step": 4759,
      "training_step_time": 10.008570671081543
    },
    {
      "epoch": 1.8157958984375e-05,
      "grad_norm": 0.6586617827415466,
      "learning_rate": 1e-05,
      "loss": 0.0888,
      "step": 4760
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 2.7441368103027344,
      "step": 4760
    },
    {
      "dynamics_loss_avg": 0.0643948744982481,
      "epoch": 1.8157958984375e-05,
      "step": 4760
    },
    {
      "action_loss_avg": 0.03319648634642362,
      "epoch": 1.8157958984375e-05,
      "step": 4760
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 4760,
      "training_step_time": 7.777596712112427
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 3.1335558891296387,
      "step": 4760
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 4760,
      "training_step_time": 9.904510736465454
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 3.1307930946350098,
      "step": 4760
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 4760,
      "training_step_time": 9.902687549591064
    },
    {
      "epoch": 1.8157958984375e-05,
      "model_forward_time": 3.1363682746887207,
      "step": 4760
    },
    {
      "epoch": 1.8157958984375e-05,
      "step": 4760,
      "training_step_time": 10.286770343780518
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 3.1793882846832275,
      "step": 4761
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 4761,
      "training_step_time": 9.952936887741089
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 3.4090309143066406,
      "step": 4761
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 4761,
      "training_step_time": 10.180071830749512
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 3.134594440460205,
      "step": 4761
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 4761,
      "training_step_time": 9.906757593154907
    },
    {
      "epoch": 1.8161773681640625e-05,
      "model_forward_time": 3.1378188133239746,
      "step": 4761
    },
    {
      "epoch": 1.8161773681640625e-05,
      "step": 4761,
      "training_step_time": 10.289440631866455
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 3.1629226207733154,
      "step": 4762
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 4762,
      "training_step_time": 9.936362266540527
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 3.1339375972747803,
      "step": 4762
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 4762,
      "training_step_time": 9.905955791473389
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 3.3950119018554688,
      "step": 4762
    },
    {
      "dynamics_loss_avg": 0.063852321729064,
      "epoch": 1.816558837890625e-05,
      "step": 4762
    },
    {
      "action_loss_avg": 0.02855928810313344,
      "epoch": 1.816558837890625e-05,
      "step": 4762
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 4762,
      "training_step_time": 10.173498630523682
    },
    {
      "epoch": 1.816558837890625e-05,
      "model_forward_time": 3.1355178356170654,
      "step": 4762
    },
    {
      "epoch": 1.816558837890625e-05,
      "step": 4762,
      "training_step_time": 10.026324272155762
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 3.18141508102417,
      "step": 4763
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 4763,
      "training_step_time": 9.962430000305176
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 3.135931968688965,
      "step": 4763
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 4763,
      "training_step_time": 9.91323208808899
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 3.1360063552856445,
      "step": 4763
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 4763,
      "training_step_time": 9.910872459411621
    },
    {
      "epoch": 1.8169403076171875e-05,
      "model_forward_time": 3.4101603031158447,
      "step": 4763
    },
    {
      "epoch": 1.8169403076171875e-05,
      "step": 4763,
      "training_step_time": 10.2876718044281
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 3.1904914379119873,
      "step": 4764
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 4764,
      "training_step_time": 9.970533847808838
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 3.1348063945770264,
      "step": 4764
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 4764,
      "training_step_time": 9.91267442703247
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 3.1392176151275635,
      "step": 4764
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 4764,
      "training_step_time": 9.915869951248169
    },
    {
      "epoch": 1.81732177734375e-05,
      "model_forward_time": 3.1331310272216797,
      "step": 4764
    },
    {
      "epoch": 1.81732177734375e-05,
      "step": 4764,
      "training_step_time": 10.298941612243652
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 3.4386889934539795,
      "step": 4765
    },
    {
      "dynamics_loss_avg": 0.058269471116364005,
      "epoch": 1.8177032470703124e-05,
      "step": 4765
    },
    {
      "action_loss_avg": 0.028508811071515084,
      "epoch": 1.8177032470703124e-05,
      "step": 4765
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 4765,
      "training_step_time": 10.219107866287231
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 3.132889747619629,
      "step": 4765
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 4765,
      "training_step_time": 9.91168475151062
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 3.1344151496887207,
      "step": 4765
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 4765,
      "training_step_time": 9.9143545627594
    },
    {
      "epoch": 1.8177032470703124e-05,
      "model_forward_time": 3.1357667446136475,
      "step": 4765
    },
    {
      "epoch": 1.8177032470703124e-05,
      "step": 4765,
      "training_step_time": 10.282295942306519
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 3.1840274333953857,
      "step": 4766
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 4766,
      "training_step_time": 9.96626329421997
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 3.3882219791412354,
      "step": 4766
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 4766,
      "training_step_time": 10.166223049163818
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 3.1399052143096924,
      "step": 4766
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 4766,
      "training_step_time": 9.915029525756836
    },
    {
      "epoch": 1.818084716796875e-05,
      "model_forward_time": 3.1349527835845947,
      "step": 4766
    },
    {
      "epoch": 1.818084716796875e-05,
      "step": 4766,
      "training_step_time": 10.276304483413696
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 3.194269895553589,
      "step": 4767
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 4767,
      "training_step_time": 9.976352214813232
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 3.1460087299346924,
      "step": 4767
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 4767,
      "training_step_time": 9.92350172996521
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 3.4284536838531494,
      "step": 4767
    },
    {
      "dynamics_loss_avg": 0.06634440049529075,
      "epoch": 1.8184661865234376e-05,
      "step": 4767
    },
    {
      "action_loss_avg": 0.03389279525727033,
      "epoch": 1.8184661865234376e-05,
      "step": 4767
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 4767,
      "training_step_time": 10.21104884147644
    },
    {
      "epoch": 1.8184661865234376e-05,
      "model_forward_time": 3.143234968185425,
      "step": 4767
    },
    {
      "epoch": 1.8184661865234376e-05,
      "step": 4767,
      "training_step_time": 10.016247510910034
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 3.1620657444000244,
      "step": 4768
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 4768,
      "training_step_time": 9.94399094581604
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 3.1496171951293945,
      "step": 4768
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 4768,
      "training_step_time": 9.93216323852539
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 3.149583101272583,
      "step": 4768
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 4768,
      "training_step_time": 9.927518129348755
    },
    {
      "epoch": 1.81884765625e-05,
      "model_forward_time": 2.440971851348877,
      "step": 4768
    },
    {
      "epoch": 1.81884765625e-05,
      "step": 4768,
      "training_step_time": 10.018295049667358
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 3.171759843826294,
      "step": 4769
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 4769,
      "training_step_time": 9.949237823486328
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 3.146733522415161,
      "step": 4769
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 4769,
      "training_step_time": 9.92889404296875
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 3.1432645320892334,
      "step": 4769
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 4769,
      "training_step_time": 9.917778253555298
    },
    {
      "epoch": 1.8192291259765625e-05,
      "model_forward_time": 3.1457085609436035,
      "step": 4769
    },
    {
      "epoch": 1.8192291259765625e-05,
      "step": 4769,
      "training_step_time": 10.285733461380005
    },
    {
      "epoch": 1.819610595703125e-05,
      "grad_norm": 0.4066225588321686,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 4770
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 3.434828281402588,
      "step": 4770
    },
    {
      "dynamics_loss_avg": 0.05906717199832201,
      "epoch": 1.819610595703125e-05,
      "step": 4770
    },
    {
      "action_loss_avg": 0.03129494581371546,
      "epoch": 1.819610595703125e-05,
      "step": 4770
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 4770,
      "training_step_time": 10.215084791183472
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 3.1445326805114746,
      "step": 4770
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 4770,
      "training_step_time": 9.921026229858398
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 3.1398985385894775,
      "step": 4770
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 4770,
      "training_step_time": 9.914292812347412
    },
    {
      "epoch": 1.819610595703125e-05,
      "model_forward_time": 3.147101402282715,
      "step": 4770
    },
    {
      "epoch": 1.819610595703125e-05,
      "step": 4770,
      "training_step_time": 10.249917984008789
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 3.1761281490325928,
      "step": 4771
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 4771,
      "training_step_time": 9.952505350112915
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 3.143860101699829,
      "step": 4771
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 4771,
      "training_step_time": 10.218552350997925
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 3.1455483436584473,
      "step": 4771
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 4771,
      "training_step_time": 9.923108577728271
    },
    {
      "epoch": 1.8199920654296875e-05,
      "model_forward_time": 3.143162250518799,
      "step": 4771
    },
    {
      "epoch": 1.8199920654296875e-05,
      "step": 4771,
      "training_step_time": 10.300248861312866
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 3.1986031532287598,
      "step": 4772
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 4772,
      "training_step_time": 9.977754354476929
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 3.140113592147827,
      "step": 4772
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 4772,
      "training_step_time": 9.918988466262817
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 3.4315316677093506,
      "step": 4772
    },
    {
      "dynamics_loss_avg": 0.049818433821201324,
      "epoch": 1.82037353515625e-05,
      "step": 4772
    },
    {
      "action_loss_avg": 0.02298530051484704,
      "epoch": 1.82037353515625e-05,
      "step": 4772
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 4772,
      "training_step_time": 10.211523056030273
    },
    {
      "epoch": 1.82037353515625e-05,
      "model_forward_time": 3.142706871032715,
      "step": 4772
    },
    {
      "epoch": 1.82037353515625e-05,
      "step": 4772,
      "training_step_time": 10.010694742202759
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 3.1666173934936523,
      "step": 4773
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 4773,
      "training_step_time": 9.940207481384277
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 3.139329433441162,
      "step": 4773
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 4773,
      "training_step_time": 9.910723209381104
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 3.132725238800049,
      "step": 4773
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 4773,
      "training_step_time": 9.906035900115967
    },
    {
      "epoch": 1.8207550048828124e-05,
      "model_forward_time": 3.4068751335144043,
      "step": 4773
    },
    {
      "epoch": 1.8207550048828124e-05,
      "step": 4773,
      "training_step_time": 10.304922342300415
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 3.1867330074310303,
      "step": 4774
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 4774,
      "training_step_time": 9.968612432479858
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 3.1407227516174316,
      "step": 4774
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 4774,
      "training_step_time": 9.917983531951904
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 3.139429807662964,
      "step": 4774
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 4774,
      "training_step_time": 9.920681476593018
    },
    {
      "epoch": 1.821136474609375e-05,
      "model_forward_time": 3.1411640644073486,
      "step": 4774
    },
    {
      "epoch": 1.821136474609375e-05,
      "step": 4774,
      "training_step_time": 10.021204710006714
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 3.427283525466919,
      "step": 4775
    },
    {
      "dynamics_loss_avg": 0.053800266422331336,
      "epoch": 1.8215179443359376e-05,
      "step": 4775
    },
    {
      "action_loss_avg": 0.0265999099239707,
      "epoch": 1.8215179443359376e-05,
      "step": 4775
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 4775,
      "training_step_time": 10.210344314575195
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 3.1413803100585938,
      "step": 4775
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 4775,
      "training_step_time": 9.91920018196106
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 3.141831159591675,
      "step": 4775
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 4775,
      "training_step_time": 9.919494152069092
    },
    {
      "epoch": 1.8215179443359376e-05,
      "model_forward_time": 3.14105486869812,
      "step": 4775
    },
    {
      "epoch": 1.8215179443359376e-05,
      "step": 4775,
      "training_step_time": 10.276984930038452
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 3.1916756629943848,
      "step": 4776
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 4776,
      "training_step_time": 9.971916913986206
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 3.4191458225250244,
      "step": 4776
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 4776,
      "training_step_time": 10.19039797782898
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 3.1327295303344727,
      "step": 4776
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 4776,
      "training_step_time": 9.904098749160767
    },
    {
      "epoch": 1.8218994140625e-05,
      "model_forward_time": 3.1356089115142822,
      "step": 4776
    },
    {
      "epoch": 1.8218994140625e-05,
      "step": 4776,
      "training_step_time": 10.275209665298462
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 3.1768462657928467,
      "step": 4777
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 4777,
      "training_step_time": 9.943865060806274
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 3.1335883140563965,
      "step": 4777
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 4777,
      "training_step_time": 9.903172492980957
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 3.138246536254883,
      "step": 4777
    },
    {
      "dynamics_loss_avg": 0.05847746655344963,
      "epoch": 1.8222808837890626e-05,
      "step": 4777
    },
    {
      "action_loss_avg": 0.021952259121462703,
      "epoch": 1.8222808837890626e-05,
      "step": 4777
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 4777,
      "training_step_time": 10.189823865890503
    },
    {
      "epoch": 1.8222808837890626e-05,
      "model_forward_time": 3.1386208534240723,
      "step": 4777
    },
    {
      "epoch": 1.8222808837890626e-05,
      "step": 4777,
      "training_step_time": 10.27462100982666
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 3.170595407485962,
      "step": 4778
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 4778,
      "training_step_time": 9.945783615112305
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 3.1355443000793457,
      "step": 4778
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 4778,
      "training_step_time": 9.905541181564331
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 3.1441268920898438,
      "step": 4778
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 4778,
      "training_step_time": 9.917490720748901
    },
    {
      "epoch": 1.822662353515625e-05,
      "model_forward_time": 3.4225854873657227,
      "step": 4778
    },
    {
      "epoch": 1.822662353515625e-05,
      "step": 4778,
      "training_step_time": 10.296832084655762
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 3.178392171859741,
      "step": 4779
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 4779,
      "training_step_time": 9.951149702072144
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 3.1406142711639404,
      "step": 4779
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 4779,
      "training_step_time": 9.90935730934143
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 3.144721031188965,
      "step": 4779
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 4779,
      "training_step_time": 9.918704271316528
    },
    {
      "epoch": 1.8230438232421875e-05,
      "model_forward_time": 3.1419568061828613,
      "step": 4779
    },
    {
      "epoch": 1.8230438232421875e-05,
      "step": 4779,
      "training_step_time": 10.281716585159302
    },
    {
      "epoch": 1.82342529296875e-05,
      "grad_norm": 0.7550568580627441,
      "learning_rate": 1e-05,
      "loss": 0.0924,
      "step": 4780
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 3.1730849742889404,
      "step": 4780
    },
    {
      "dynamics_loss_avg": 0.051331153698265554,
      "epoch": 1.82342529296875e-05,
      "step": 4780
    },
    {
      "action_loss_avg": 0.021108587784692646,
      "epoch": 1.82342529296875e-05,
      "step": 4780
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 4780,
      "training_step_time": 10.216819286346436
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 3.1400392055511475,
      "step": 4780
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 4780,
      "training_step_time": 9.911730289459229
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 3.1404500007629395,
      "step": 4780
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 4780,
      "training_step_time": 9.913226842880249
    },
    {
      "epoch": 1.82342529296875e-05,
      "model_forward_time": 3.1371328830718994,
      "step": 4780
    },
    {
      "epoch": 1.82342529296875e-05,
      "step": 4780,
      "training_step_time": 10.27587342262268
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 3.183119535446167,
      "step": 4781
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 4781,
      "training_step_time": 9.966268062591553
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 3.404632329940796,
      "step": 4781
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 4781,
      "training_step_time": 10.172528266906738
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 3.144479990005493,
      "step": 4781
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 4781,
      "training_step_time": 9.918314456939697
    },
    {
      "epoch": 1.8238067626953124e-05,
      "model_forward_time": 3.1406209468841553,
      "step": 4781
    },
    {
      "epoch": 1.8238067626953124e-05,
      "step": 4781,
      "training_step_time": 10.26489806175232
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 3.176138162612915,
      "step": 4782
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 4782,
      "training_step_time": 9.944993257522583
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 3.1421918869018555,
      "step": 4782
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 4782,
      "training_step_time": 9.913407802581787
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 3.4301228523254395,
      "step": 4782
    },
    {
      "dynamics_loss_avg": 0.06225382126867771,
      "epoch": 1.824188232421875e-05,
      "step": 4782
    },
    {
      "action_loss_avg": 0.026891141943633557,
      "epoch": 1.824188232421875e-05,
      "step": 4782
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 4782,
      "training_step_time": 10.206759452819824
    },
    {
      "epoch": 1.824188232421875e-05,
      "model_forward_time": 3.1469380855560303,
      "step": 4782
    },
    {
      "epoch": 1.824188232421875e-05,
      "step": 4782,
      "training_step_time": 10.289523601531982
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 3.1853156089782715,
      "step": 4783
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 4783,
      "training_step_time": 9.962555408477783
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 3.1415762901306152,
      "step": 4783
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 4783,
      "training_step_time": 9.910338640213013
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 3.143725872039795,
      "step": 4783
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 4783,
      "training_step_time": 9.915376901626587
    },
    {
      "epoch": 1.8245697021484376e-05,
      "model_forward_time": 3.41599178314209,
      "step": 4783
    },
    {
      "epoch": 1.8245697021484376e-05,
      "step": 4783,
      "training_step_time": 10.302957773208618
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 3.1812047958374023,
      "step": 4784
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 4784,
      "training_step_time": 9.95748257637024
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 3.146385908126831,
      "step": 4784
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 4784,
      "training_step_time": 9.919647932052612
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 3.1450679302215576,
      "step": 4784
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 4784,
      "training_step_time": 9.92225980758667
    },
    {
      "epoch": 1.824951171875e-05,
      "model_forward_time": 3.1489222049713135,
      "step": 4784
    },
    {
      "epoch": 1.824951171875e-05,
      "step": 4784,
      "training_step_time": 10.304839611053467
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 3.4263129234313965,
      "step": 4785
    },
    {
      "dynamics_loss_avg": 0.05385929979383945,
      "epoch": 1.8253326416015626e-05,
      "step": 4785
    },
    {
      "action_loss_avg": 0.024912874680012465,
      "epoch": 1.8253326416015626e-05,
      "step": 4785
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 4785,
      "training_step_time": 10.20514726638794
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 3.146756649017334,
      "step": 4785
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 4785,
      "training_step_time": 9.921929836273193
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 3.1412618160247803,
      "step": 4785
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 4785,
      "training_step_time": 9.912834167480469
    },
    {
      "epoch": 1.8253326416015626e-05,
      "model_forward_time": 3.1421425342559814,
      "step": 4785
    },
    {
      "epoch": 1.8253326416015626e-05,
      "step": 4785,
      "training_step_time": 10.012163400650024
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 3.169086456298828,
      "step": 4786
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 4786,
      "training_step_time": 9.94120979309082
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 3.1409225463867188,
      "step": 4786
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 4786,
      "training_step_time": 10.195796251296997
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 3.1409645080566406,
      "step": 4786
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 4786,
      "training_step_time": 9.912700176239014
    },
    {
      "epoch": 1.825714111328125e-05,
      "model_forward_time": 3.146104574203491,
      "step": 4786
    },
    {
      "epoch": 1.825714111328125e-05,
      "step": 4786,
      "training_step_time": 10.284082174301147
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 3.1868948936462402,
      "step": 4787
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 4787,
      "training_step_time": 9.961467981338501
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 3.146827459335327,
      "step": 4787
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 4787,
      "training_step_time": 9.922211647033691
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 3.4456677436828613,
      "step": 4787
    },
    {
      "dynamics_loss_avg": 0.06585291754454374,
      "epoch": 1.8260955810546875e-05,
      "step": 4787
    },
    {
      "action_loss_avg": 0.02335877837613225,
      "epoch": 1.8260955810546875e-05,
      "step": 4787
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 4787,
      "training_step_time": 10.223747968673706
    },
    {
      "epoch": 1.8260955810546875e-05,
      "model_forward_time": 3.1489310264587402,
      "step": 4787
    },
    {
      "epoch": 1.8260955810546875e-05,
      "step": 4787,
      "training_step_time": 10.274994134902954
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 3.177834987640381,
      "step": 4788
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 4788,
      "training_step_time": 9.954679489135742
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 3.1441805362701416,
      "step": 4788
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 4788,
      "training_step_time": 9.920779943466187
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 3.1468546390533447,
      "step": 4788
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 4788,
      "training_step_time": 9.916127920150757
    },
    {
      "epoch": 1.82647705078125e-05,
      "model_forward_time": 3.1454758644104004,
      "step": 4788
    },
    {
      "epoch": 1.82647705078125e-05,
      "step": 4788,
      "training_step_time": 10.306123495101929
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 3.150974988937378,
      "step": 4789
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 4789,
      "training_step_time": 9.920866966247559
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 3.131380081176758,
      "step": 4789
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 4789,
      "training_step_time": 9.904757261276245
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 2.4316234588623047,
      "step": 4789
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 4789,
      "training_step_time": 7.46225380897522
    },
    {
      "epoch": 1.8268585205078124e-05,
      "model_forward_time": 3.135185956954956,
      "step": 4789
    },
    {
      "epoch": 1.8268585205078124e-05,
      "step": 4789,
      "training_step_time": 10.289667844772339
    },
    {
      "epoch": 1.827239990234375e-05,
      "grad_norm": 0.30231016874313354,
      "learning_rate": 1e-05,
      "loss": 0.0919,
      "step": 4790
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 3.175628900527954,
      "step": 4790
    },
    {
      "dynamics_loss_avg": 0.050173462741076945,
      "epoch": 1.827239990234375e-05,
      "step": 4790
    },
    {
      "action_loss_avg": 0.01957567408680916,
      "epoch": 1.827239990234375e-05,
      "step": 4790
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 4790,
      "training_step_time": 10.23981261253357
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 3.1363162994384766,
      "step": 4790
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 4790,
      "training_step_time": 9.908971786499023
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 3.1405701637268066,
      "step": 4790
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 4790,
      "training_step_time": 9.919637203216553
    },
    {
      "epoch": 1.827239990234375e-05,
      "model_forward_time": 3.134713888168335,
      "step": 4790
    },
    {
      "epoch": 1.827239990234375e-05,
      "step": 4790,
      "training_step_time": 10.021990537643433
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 3.181241989135742,
      "step": 4791
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 4791,
      "training_step_time": 9.967469215393066
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 3.1451759338378906,
      "step": 4791
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 4791,
      "training_step_time": 10.205613613128662
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 3.144301414489746,
      "step": 4791
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 4791,
      "training_step_time": 9.925530672073364
    },
    {
      "epoch": 1.8276214599609376e-05,
      "model_forward_time": 3.1448028087615967,
      "step": 4791
    },
    {
      "epoch": 1.8276214599609376e-05,
      "step": 4791,
      "training_step_time": 10.281871318817139
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 3.180478572845459,
      "step": 4792
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 4792,
      "training_step_time": 9.957188844680786
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 3.1439261436462402,
      "step": 4792
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 4792,
      "training_step_time": 9.915204286575317
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 3.4212560653686523,
      "step": 4792
    },
    {
      "dynamics_loss_avg": 0.04842404695227742,
      "epoch": 1.8280029296875e-05,
      "step": 4792
    },
    {
      "action_loss_avg": 0.02714146105572581,
      "epoch": 1.8280029296875e-05,
      "step": 4792
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 4792,
      "training_step_time": 10.196346998214722
    },
    {
      "epoch": 1.8280029296875e-05,
      "model_forward_time": 3.138389825820923,
      "step": 4792
    },
    {
      "epoch": 1.8280029296875e-05,
      "step": 4792,
      "training_step_time": 10.287862777709961
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 3.1713171005249023,
      "step": 4793
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 4793,
      "training_step_time": 9.945144891738892
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 3.140317440032959,
      "step": 4793
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 4793,
      "training_step_time": 9.909892797470093
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 3.139503240585327,
      "step": 4793
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 4793,
      "training_step_time": 9.914766550064087
    },
    {
      "epoch": 1.8283843994140626e-05,
      "model_forward_time": 3.400451898574829,
      "step": 4793
    },
    {
      "epoch": 1.8283843994140626e-05,
      "step": 4793,
      "training_step_time": 10.298673868179321
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 3.197134494781494,
      "step": 4794
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 4794,
      "training_step_time": 9.968587398529053
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 3.140943765640259,
      "step": 4794
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 4794,
      "training_step_time": 9.912553548812866
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 3.135091781616211,
      "step": 4794
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 4794,
      "training_step_time": 9.906444549560547
    },
    {
      "epoch": 1.828765869140625e-05,
      "model_forward_time": 3.140824317932129,
      "step": 4794
    },
    {
      "epoch": 1.828765869140625e-05,
      "step": 4794,
      "training_step_time": 10.019850254058838
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 3.4496500492095947,
      "step": 4795
    },
    {
      "dynamics_loss_avg": 0.06529079433530569,
      "epoch": 1.8291473388671875e-05,
      "step": 4795
    },
    {
      "action_loss_avg": 0.03452189927920699,
      "epoch": 1.8291473388671875e-05,
      "step": 4795
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 4795,
      "training_step_time": 10.227510690689087
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 3.135411024093628,
      "step": 4795
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 4795,
      "training_step_time": 9.906426429748535
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 3.1377005577087402,
      "step": 4795
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 4795,
      "training_step_time": 9.913365125656128
    },
    {
      "epoch": 1.8291473388671875e-05,
      "model_forward_time": 3.1396148204803467,
      "step": 4795
    },
    {
      "epoch": 1.8291473388671875e-05,
      "step": 4795,
      "training_step_time": 10.013183116912842
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 3.1883656978607178,
      "step": 4796
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 4796,
      "training_step_time": 9.962308645248413
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 3.4152629375457764,
      "step": 4796
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 4796,
      "training_step_time": 10.187211036682129
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 3.139106512069702,
      "step": 4796
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 4796,
      "training_step_time": 9.91202425956726
    },
    {
      "epoch": 1.82952880859375e-05,
      "model_forward_time": 3.1373202800750732,
      "step": 4796
    },
    {
      "epoch": 1.82952880859375e-05,
      "step": 4796,
      "training_step_time": 10.25275182723999
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 3.1822502613067627,
      "step": 4797
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 4797,
      "training_step_time": 9.949279069900513
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 3.1263647079467773,
      "step": 4797
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 4797,
      "training_step_time": 9.891229629516602
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 3.1342077255249023,
      "step": 4797
    },
    {
      "dynamics_loss_avg": 0.08097968548536301,
      "epoch": 1.8299102783203124e-05,
      "step": 4797
    },
    {
      "action_loss_avg": 0.033601338230073455,
      "epoch": 1.8299102783203124e-05,
      "step": 4797
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 4797,
      "training_step_time": 10.183496236801147
    },
    {
      "epoch": 1.8299102783203124e-05,
      "model_forward_time": 3.1373209953308105,
      "step": 4797
    },
    {
      "epoch": 1.8299102783203124e-05,
      "step": 4797,
      "training_step_time": 10.259058952331543
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 3.1800854206085205,
      "step": 4798
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 4798,
      "training_step_time": 9.953248739242554
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 3.133998394012451,
      "step": 4798
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 4798,
      "training_step_time": 9.903129816055298
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 3.1398873329162598,
      "step": 4798
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 4798,
      "training_step_time": 9.909335613250732
    },
    {
      "epoch": 1.830291748046875e-05,
      "model_forward_time": 3.424255132675171,
      "step": 4798
    },
    {
      "epoch": 1.830291748046875e-05,
      "step": 4798,
      "training_step_time": 10.285836696624756
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 3.1698479652404785,
      "step": 4799
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 4799,
      "training_step_time": 9.942018747329712
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 3.1370208263397217,
      "step": 4799
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 4799,
      "training_step_time": 9.908541679382324
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 3.1380503177642822,
      "step": 4799
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 4799,
      "training_step_time": 9.907415628433228
    },
    {
      "epoch": 1.8306732177734376e-05,
      "model_forward_time": 3.1373753547668457,
      "step": 4799
    },
    {
      "epoch": 1.8306732177734376e-05,
      "step": 4799,
      "training_step_time": 10.290948629379272
    },
    {
      "epoch": 1.8310546875e-05,
      "grad_norm": 0.6068446040153503,
      "learning_rate": 1e-05,
      "loss": 0.0863,
      "step": 4800
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 3.1630656719207764,
      "step": 4800
    },
    {
      "dynamics_loss_avg": 0.05662286747246981,
      "epoch": 1.8310546875e-05,
      "step": 4800
    },
    {
      "action_loss_avg": 0.02266465239226818,
      "epoch": 1.8310546875e-05,
      "step": 4800
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 4800,
      "training_step_time": 10.211149454116821
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 3.135969638824463,
      "step": 4800
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 4800,
      "training_step_time": 9.905732870101929
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 3.1404058933258057,
      "step": 4800
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 4800,
      "training_step_time": 9.918550491333008
    },
    {
      "epoch": 1.8310546875e-05,
      "model_forward_time": 3.1431281566619873,
      "step": 4800
    },
    {
      "epoch": 1.8310546875e-05,
      "step": 4800,
      "training_step_time": 10.032466173171997
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 3.165158748626709,
      "step": 4801
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 4801,
      "training_step_time": 9.937203645706177
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 3.4037094116210938,
      "step": 4801
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 4801,
      "training_step_time": 10.174852848052979
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 3.138578414916992,
      "step": 4801
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 4801,
      "training_step_time": 9.91141414642334
    },
    {
      "epoch": 1.8314361572265626e-05,
      "model_forward_time": 3.139587879180908,
      "step": 4801
    },
    {
      "epoch": 1.8314361572265626e-05,
      "step": 4801,
      "training_step_time": 10.024056434631348
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 3.1777429580688477,
      "step": 4802
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 4802,
      "training_step_time": 9.963572978973389
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 3.1434528827667236,
      "step": 4802
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 4802,
      "training_step_time": 9.918649911880493
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 3.457853078842163,
      "step": 4802
    },
    {
      "dynamics_loss_avg": 0.05660565085709095,
      "epoch": 1.831817626953125e-05,
      "step": 4802
    },
    {
      "action_loss_avg": 0.0262576412409544,
      "epoch": 1.831817626953125e-05,
      "step": 4802
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 4802,
      "training_step_time": 10.233753681182861
    },
    {
      "epoch": 1.831817626953125e-05,
      "model_forward_time": 3.146892547607422,
      "step": 4802
    },
    {
      "epoch": 1.831817626953125e-05,
      "step": 4802,
      "training_step_time": 10.247353792190552
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 3.173194408416748,
      "step": 4803
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 4803,
      "training_step_time": 9.95021367073059
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 3.150325298309326,
      "step": 4803
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 4803,
      "training_step_time": 9.931248664855957
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 3.1496083736419678,
      "step": 4803
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 4803,
      "training_step_time": 9.925902366638184
    },
    {
      "epoch": 1.8321990966796875e-05,
      "model_forward_time": 3.410402774810791,
      "step": 4803
    },
    {
      "epoch": 1.8321990966796875e-05,
      "step": 4803,
      "training_step_time": 10.315656185150146
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 2.500476837158203,
      "step": 4804
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 4804,
      "training_step_time": 7.535177946090698
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 3.14727520942688,
      "step": 4804
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 4804,
      "training_step_time": 9.924097299575806
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 3.1502625942230225,
      "step": 4804
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 4804,
      "training_step_time": 9.927616357803345
    },
    {
      "epoch": 1.83258056640625e-05,
      "model_forward_time": 3.152263641357422,
      "step": 4804
    },
    {
      "epoch": 1.83258056640625e-05,
      "step": 4804,
      "training_step_time": 10.028105735778809
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 3.1846585273742676,
      "step": 4805
    },
    {
      "dynamics_loss_avg": 0.06597075928002596,
      "epoch": 1.8329620361328124e-05,
      "step": 4805
    },
    {
      "action_loss_avg": 0.029340470675379038,
      "epoch": 1.8329620361328124e-05,
      "step": 4805
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 4805,
      "training_step_time": 10.248331308364868
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 3.1513679027557373,
      "step": 4805
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 4805,
      "training_step_time": 9.930341958999634
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 3.1429874897003174,
      "step": 4805
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 4805,
      "training_step_time": 9.920500040054321
    },
    {
      "epoch": 1.8329620361328124e-05,
      "model_forward_time": 3.15358304977417,
      "step": 4805
    },
    {
      "epoch": 1.8329620361328124e-05,
      "step": 4805,
      "training_step_time": 10.333066940307617
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 3.187852144241333,
      "step": 4806
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 4806,
      "training_step_time": 9.968181371688843
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 3.1499905586242676,
      "step": 4806
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 4806,
      "training_step_time": 10.21314549446106
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 3.147024393081665,
      "step": 4806
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 4806,
      "training_step_time": 9.92574429512024
    },
    {
      "epoch": 1.833343505859375e-05,
      "model_forward_time": 3.148623466491699,
      "step": 4806
    },
    {
      "epoch": 1.833343505859375e-05,
      "step": 4806,
      "training_step_time": 10.026819705963135
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 3.1823225021362305,
      "step": 4807
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 4807,
      "training_step_time": 9.958365440368652
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 3.1463568210601807,
      "step": 4807
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 4807,
      "training_step_time": 9.92196011543274
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 3.410238265991211,
      "step": 4807
    },
    {
      "dynamics_loss_avg": 0.053187605924904346,
      "epoch": 1.8337249755859376e-05,
      "step": 4807
    },
    {
      "action_loss_avg": 0.024249689048156144,
      "epoch": 1.8337249755859376e-05,
      "step": 4807
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 4807,
      "training_step_time": 10.188986539840698
    },
    {
      "epoch": 1.8337249755859376e-05,
      "model_forward_time": 3.1512222290039062,
      "step": 4807
    },
    {
      "epoch": 1.8337249755859376e-05,
      "step": 4807,
      "training_step_time": 10.26662826538086
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 3.174036979675293,
      "step": 4808
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 4808,
      "training_step_time": 9.951684474945068
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 3.145663261413574,
      "step": 4808
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 4808,
      "training_step_time": 9.919384717941284
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 3.148195505142212,
      "step": 4808
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 4808,
      "training_step_time": 9.922122240066528
    },
    {
      "epoch": 1.8341064453125e-05,
      "model_forward_time": 3.1478278636932373,
      "step": 4808
    },
    {
      "epoch": 1.8341064453125e-05,
      "step": 4808,
      "training_step_time": 10.288956880569458
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 2.4814438819885254,
      "step": 4809
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 4809,
      "training_step_time": 7.513737916946411
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 3.143132209777832,
      "step": 4809
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 4809,
      "training_step_time": 9.918924570083618
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 3.1489808559417725,
      "step": 4809
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 4809,
      "training_step_time": 9.923212051391602
    },
    {
      "epoch": 1.8344879150390626e-05,
      "model_forward_time": 3.143803119659424,
      "step": 4809
    },
    {
      "epoch": 1.8344879150390626e-05,
      "step": 4809,
      "training_step_time": 10.286695003509521
    },
    {
      "epoch": 1.834869384765625e-05,
      "grad_norm": 0.6704612970352173,
      "learning_rate": 1e-05,
      "loss": 0.0889,
      "step": 4810
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 3.146392822265625,
      "step": 4810
    },
    {
      "dynamics_loss_avg": 0.08072609547525644,
      "epoch": 1.834869384765625e-05,
      "step": 4810
    },
    {
      "action_loss_avg": 0.039244643226265905,
      "epoch": 1.834869384765625e-05,
      "step": 4810
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 4810,
      "training_step_time": 10.187844276428223
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 3.135113477706909,
      "step": 4810
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 4810,
      "training_step_time": 9.90782642364502
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 2.43167781829834,
      "step": 4810
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 4810,
      "training_step_time": 7.46375036239624
    },
    {
      "epoch": 1.834869384765625e-05,
      "model_forward_time": 3.1281344890594482,
      "step": 4810
    },
    {
      "epoch": 1.834869384765625e-05,
      "step": 4810,
      "training_step_time": 10.256382703781128
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 3.171908140182495,
      "step": 4811
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 4811,
      "training_step_time": 9.942694187164307
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 3.137061595916748,
      "step": 4811
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 4811,
      "training_step_time": 9.9037504196167
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 3.404452323913574,
      "step": 4811
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 4811,
      "training_step_time": 10.1738440990448
    },
    {
      "epoch": 1.8352508544921875e-05,
      "model_forward_time": 3.1375575065612793,
      "step": 4811
    },
    {
      "epoch": 1.8352508544921875e-05,
      "step": 4811,
      "training_step_time": 10.012670040130615
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 3.185882091522217,
      "step": 4812
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 4812,
      "training_step_time": 9.957018375396729
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 3.1437439918518066,
      "step": 4812
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 4812,
      "training_step_time": 9.915050983428955
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 3.4268600940704346,
      "step": 4812
    },
    {
      "dynamics_loss_avg": 0.057698026113212106,
      "epoch": 1.83563232421875e-05,
      "step": 4812
    },
    {
      "action_loss_avg": 0.02322055269032717,
      "epoch": 1.83563232421875e-05,
      "step": 4812
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 4812,
      "training_step_time": 10.204927444458008
    },
    {
      "epoch": 1.83563232421875e-05,
      "model_forward_time": 3.1436564922332764,
      "step": 4812
    },
    {
      "epoch": 1.83563232421875e-05,
      "step": 4812,
      "training_step_time": 10.272672891616821
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 3.1906208992004395,
      "step": 4813
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 4813,
      "training_step_time": 9.980370044708252
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 3.150984048843384,
      "step": 4813
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 4813,
      "training_step_time": 9.930004119873047
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 3.1453001499176025,
      "step": 4813
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 4813,
      "training_step_time": 9.920185327529907
    },
    {
      "epoch": 1.8360137939453124e-05,
      "model_forward_time": 3.144926071166992,
      "step": 4813
    },
    {
      "epoch": 1.8360137939453124e-05,
      "step": 4813,
      "training_step_time": 10.287692070007324
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 3.4724385738372803,
      "step": 4814
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 4814,
      "training_step_time": 10.24067211151123
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 3.146074056625366,
      "step": 4814
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 4814,
      "training_step_time": 9.9137442111969
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 3.141942262649536,
      "step": 4814
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 4814,
      "training_step_time": 9.91476559638977
    },
    {
      "epoch": 1.836395263671875e-05,
      "model_forward_time": 3.1444668769836426,
      "step": 4814
    },
    {
      "epoch": 1.836395263671875e-05,
      "step": 4814,
      "training_step_time": 10.288679599761963
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 3.483438491821289,
      "step": 4815
    },
    {
      "dynamics_loss_avg": 0.05430921968072653,
      "epoch": 1.8367767333984376e-05,
      "step": 4815
    },
    {
      "action_loss_avg": 0.030140737700276077,
      "epoch": 1.8367767333984376e-05,
      "step": 4815
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 4815,
      "training_step_time": 10.255715370178223
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 3.145699977874756,
      "step": 4815
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 4815,
      "training_step_time": 9.916000366210938
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 3.1416726112365723,
      "step": 4815
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 4815,
      "training_step_time": 9.909990072250366
    },
    {
      "epoch": 1.8367767333984376e-05,
      "model_forward_time": 3.1451187133789062,
      "step": 4815
    },
    {
      "epoch": 1.8367767333984376e-05,
      "step": 4815,
      "training_step_time": 10.01887559890747
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 3.172037363052368,
      "step": 4816
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 4816,
      "training_step_time": 9.953285455703735
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 3.139125108718872,
      "step": 4816
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 4816,
      "training_step_time": 10.182879209518433
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 3.139188766479492,
      "step": 4816
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 4816,
      "training_step_time": 9.910632848739624
    },
    {
      "epoch": 1.837158203125e-05,
      "model_forward_time": 3.1471188068389893,
      "step": 4816
    },
    {
      "epoch": 1.837158203125e-05,
      "step": 4816,
      "training_step_time": 10.300843954086304
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 3.1837985515594482,
      "step": 4817
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 4817,
      "training_step_time": 9.958357334136963
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 3.1407577991485596,
      "step": 4817
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 4817,
      "training_step_time": 9.912696599960327
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 3.145108699798584,
      "step": 4817
    },
    {
      "dynamics_loss_avg": 0.05435292366892099,
      "epoch": 1.8375396728515626e-05,
      "step": 4817
    },
    {
      "action_loss_avg": 0.032276162132620814,
      "epoch": 1.8375396728515626e-05,
      "step": 4817
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 4817,
      "training_step_time": 10.217381477355957
    },
    {
      "epoch": 1.8375396728515626e-05,
      "model_forward_time": 3.1437060832977295,
      "step": 4817
    },
    {
      "epoch": 1.8375396728515626e-05,
      "step": 4817,
      "training_step_time": 10.270342350006104
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 3.1726114749908447,
      "step": 4818
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 4818,
      "training_step_time": 9.94705843925476
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 3.1400868892669678,
      "step": 4818
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 4818,
      "training_step_time": 9.91335654258728
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 3.1504178047180176,
      "step": 4818
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 4818,
      "training_step_time": 9.927808046340942
    },
    {
      "epoch": 1.837921142578125e-05,
      "model_forward_time": 3.458183765411377,
      "step": 4818
    },
    {
      "epoch": 1.837921142578125e-05,
      "step": 4818,
      "training_step_time": 10.33760380744934
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 3.178572416305542,
      "step": 4819
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 4819,
      "training_step_time": 9.957024335861206
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 3.149444818496704,
      "step": 4819
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 4819,
      "training_step_time": 9.927839517593384
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 3.1510396003723145,
      "step": 4819
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 4819,
      "training_step_time": 9.933589696884155
    },
    {
      "epoch": 1.8383026123046875e-05,
      "model_forward_time": 3.1477582454681396,
      "step": 4819
    },
    {
      "epoch": 1.8383026123046875e-05,
      "step": 4819,
      "training_step_time": 10.29643726348877
    },
    {
      "epoch": 1.83868408203125e-05,
      "grad_norm": 0.5697858333587646,
      "learning_rate": 1e-05,
      "loss": 0.0914,
      "step": 4820
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 3.178407669067383,
      "step": 4820
    },
    {
      "dynamics_loss_avg": 0.07297036275267602,
      "epoch": 1.83868408203125e-05,
      "step": 4820
    },
    {
      "action_loss_avg": 0.030246684327721596,
      "epoch": 1.83868408203125e-05,
      "step": 4820
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 4820,
      "training_step_time": 10.237412691116333
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 3.150759696960449,
      "step": 4820
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 4820,
      "training_step_time": 9.933186292648315
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 3.1511282920837402,
      "step": 4820
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 4820,
      "training_step_time": 9.929937839508057
    },
    {
      "epoch": 1.83868408203125e-05,
      "model_forward_time": 3.149510622024536,
      "step": 4820
    },
    {
      "epoch": 1.83868408203125e-05,
      "step": 4820,
      "training_step_time": 10.230650663375854
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 3.1955325603485107,
      "step": 4821
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 4821,
      "training_step_time": 9.975359678268433
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 3.416072130203247,
      "step": 4821
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 4821,
      "training_step_time": 10.1946702003479
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 3.1522934436798096,
      "step": 4821
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 4821,
      "training_step_time": 9.9322669506073
    },
    {
      "epoch": 1.8390655517578124e-05,
      "model_forward_time": 3.145099401473999,
      "step": 4821
    },
    {
      "epoch": 1.8390655517578124e-05,
      "step": 4821,
      "training_step_time": 10.01495099067688
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 3.173259973526001,
      "step": 4822
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 4822,
      "training_step_time": 9.948194026947021
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 3.138174057006836,
      "step": 4822
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 4822,
      "training_step_time": 9.910415172576904
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 3.4172263145446777,
      "step": 4822
    },
    {
      "dynamics_loss_avg": 0.0597304243594408,
      "epoch": 1.839447021484375e-05,
      "step": 4822
    },
    {
      "action_loss_avg": 0.031437347456812856,
      "epoch": 1.839447021484375e-05,
      "step": 4822
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 4822,
      "training_step_time": 10.187514305114746
    },
    {
      "epoch": 1.839447021484375e-05,
      "model_forward_time": 3.1344542503356934,
      "step": 4822
    },
    {
      "epoch": 1.839447021484375e-05,
      "step": 4822,
      "training_step_time": 10.248504877090454
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 3.188218593597412,
      "step": 4823
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 4823,
      "training_step_time": 9.963118314743042
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 3.1368565559387207,
      "step": 4823
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 4823,
      "training_step_time": 9.90919017791748
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 3.13435959815979,
      "step": 4823
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 4823,
      "training_step_time": 9.906222343444824
    },
    {
      "epoch": 1.8398284912109377e-05,
      "model_forward_time": 3.388852596282959,
      "step": 4823
    },
    {
      "epoch": 1.8398284912109377e-05,
      "step": 4823,
      "training_step_time": 10.25177550315857
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 3.191127061843872,
      "step": 4824
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 4824,
      "training_step_time": 9.964060306549072
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 3.135045051574707,
      "step": 4824
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 4824,
      "training_step_time": 9.908485174179077
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 3.1372292041778564,
      "step": 4824
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 4824,
      "training_step_time": 9.916333436965942
    },
    {
      "epoch": 1.8402099609375e-05,
      "model_forward_time": 3.143643856048584,
      "step": 4824
    },
    {
      "epoch": 1.8402099609375e-05,
      "step": 4824,
      "training_step_time": 10.289093971252441
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 3.432622194290161,
      "step": 4825
    },
    {
      "dynamics_loss_avg": 0.05081639476120472,
      "epoch": 1.8405914306640626e-05,
      "step": 4825
    },
    {
      "action_loss_avg": 0.022854601498693228,
      "epoch": 1.8405914306640626e-05,
      "step": 4825
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 4825,
      "training_step_time": 10.218016147613525
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 3.1460120677948,
      "step": 4825
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 4825,
      "training_step_time": 9.927193641662598
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 3.1449296474456787,
      "step": 4825
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 4825,
      "training_step_time": 9.929465770721436
    },
    {
      "epoch": 1.8405914306640626e-05,
      "model_forward_time": 3.151075601577759,
      "step": 4825
    },
    {
      "epoch": 1.8405914306640626e-05,
      "step": 4825,
      "training_step_time": 10.293418407440186
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 3.1717514991760254,
      "step": 4826
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 4826,
      "training_step_time": 9.954546928405762
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 3.1425411701202393,
      "step": 4826
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 4826,
      "training_step_time": 10.210310220718384
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 3.14326810836792,
      "step": 4826
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 4826,
      "training_step_time": 9.929207801818848
    },
    {
      "epoch": 1.840972900390625e-05,
      "model_forward_time": 3.1478824615478516,
      "step": 4826
    },
    {
      "epoch": 1.840972900390625e-05,
      "step": 4826,
      "training_step_time": 10.28579044342041
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 3.1817374229431152,
      "step": 4827
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 4827,
      "training_step_time": 9.963761329650879
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 3.140024423599243,
      "step": 4827
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 4827,
      "training_step_time": 9.91617488861084
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 3.405655860900879,
      "step": 4827
    },
    {
      "dynamics_loss_avg": 0.05914880242198706,
      "epoch": 1.8413543701171875e-05,
      "step": 4827
    },
    {
      "action_loss_avg": 0.03112747254781425,
      "epoch": 1.8413543701171875e-05,
      "step": 4827
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 4827,
      "training_step_time": 10.179234027862549
    },
    {
      "epoch": 1.8413543701171875e-05,
      "model_forward_time": 3.139204978942871,
      "step": 4827
    },
    {
      "epoch": 1.8413543701171875e-05,
      "step": 4827,
      "training_step_time": 10.2688467502594
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 3.16828989982605,
      "step": 4828
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 4828,
      "training_step_time": 9.940832138061523
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 3.134343385696411,
      "step": 4828
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 4828,
      "training_step_time": 9.90576696395874
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 3.1373350620269775,
      "step": 4828
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 4828,
      "training_step_time": 9.909093141555786
    },
    {
      "epoch": 1.84173583984375e-05,
      "model_forward_time": 3.138469934463501,
      "step": 4828
    },
    {
      "epoch": 1.84173583984375e-05,
      "step": 4828,
      "training_step_time": 10.279990911483765
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 3.1517863273620605,
      "step": 4829
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 4829,
      "training_step_time": 9.927173376083374
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 3.138561725616455,
      "step": 4829
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 4829,
      "training_step_time": 9.912024021148682
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 2.4334487915039062,
      "step": 4829
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 4829,
      "training_step_time": 7.466920375823975
    },
    {
      "epoch": 1.8421173095703124e-05,
      "model_forward_time": 3.144592046737671,
      "step": 4829
    },
    {
      "epoch": 1.8421173095703124e-05,
      "step": 4829,
      "training_step_time": 10.015548944473267
    },
    {
      "epoch": 1.842498779296875e-05,
      "grad_norm": 0.40566426515579224,
      "learning_rate": 1e-05,
      "loss": 0.0897,
      "step": 4830
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 3.188339948654175,
      "step": 4830
    },
    {
      "dynamics_loss_avg": 0.05999425407499075,
      "epoch": 1.842498779296875e-05,
      "step": 4830
    },
    {
      "action_loss_avg": 0.030994865181855857,
      "epoch": 1.842498779296875e-05,
      "step": 4830
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 4830,
      "training_step_time": 10.251362800598145
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 3.144364356994629,
      "step": 4830
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 4830,
      "training_step_time": 9.924379587173462
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 3.1450929641723633,
      "step": 4830
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 4830,
      "training_step_time": 9.923901796340942
    },
    {
      "epoch": 1.842498779296875e-05,
      "model_forward_time": 3.140907049179077,
      "step": 4830
    },
    {
      "epoch": 1.842498779296875e-05,
      "step": 4830,
      "training_step_time": 10.273609399795532
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 3.183605432510376,
      "step": 4831
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 4831,
      "training_step_time": 9.962941408157349
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 3.136674642562866,
      "step": 4831
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 4831,
      "training_step_time": 10.206535577774048
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 3.142670154571533,
      "step": 4831
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 4831,
      "training_step_time": 9.923794269561768
    },
    {
      "epoch": 1.8428802490234377e-05,
      "model_forward_time": 3.136530876159668,
      "step": 4831
    },
    {
      "epoch": 1.8428802490234377e-05,
      "step": 4831,
      "training_step_time": 10.272769927978516
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 3.1838762760162354,
      "step": 4832
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 4832,
      "training_step_time": 9.962043762207031
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 3.138428211212158,
      "step": 4832
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 4832,
      "training_step_time": 9.913628578186035
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 3.420271158218384,
      "step": 4832
    },
    {
      "dynamics_loss_avg": 0.06064287181943655,
      "epoch": 1.84326171875e-05,
      "step": 4832
    },
    {
      "action_loss_avg": 0.02444983320310712,
      "epoch": 1.84326171875e-05,
      "step": 4832
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 4832,
      "training_step_time": 10.20132851600647
    },
    {
      "epoch": 1.84326171875e-05,
      "model_forward_time": 3.134456157684326,
      "step": 4832
    },
    {
      "epoch": 1.84326171875e-05,
      "step": 4832,
      "training_step_time": 10.020678281784058
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 3.1827666759490967,
      "step": 4833
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 4833,
      "training_step_time": 9.956233739852905
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 3.1349239349365234,
      "step": 4833
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 4833,
      "training_step_time": 9.904738426208496
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 3.133648633956909,
      "step": 4833
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 4833,
      "training_step_time": 9.909532308578491
    },
    {
      "epoch": 1.8436431884765626e-05,
      "model_forward_time": 3.4041197299957275,
      "step": 4833
    },
    {
      "epoch": 1.8436431884765626e-05,
      "step": 4833,
      "training_step_time": 10.2878999710083
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 3.1759226322174072,
      "step": 4834
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 4834,
      "training_step_time": 9.951725244522095
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 3.1396992206573486,
      "step": 4834
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 4834,
      "training_step_time": 9.916078090667725
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 3.143385171890259,
      "step": 4834
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 4834,
      "training_step_time": 9.920197248458862
    },
    {
      "epoch": 1.844024658203125e-05,
      "model_forward_time": 3.1450328826904297,
      "step": 4834
    },
    {
      "epoch": 1.844024658203125e-05,
      "step": 4834,
      "training_step_time": 10.018259525299072
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 3.4656362533569336,
      "step": 4835
    },
    {
      "dynamics_loss_avg": 0.06196709908545017,
      "epoch": 1.8444061279296875e-05,
      "step": 4835
    },
    {
      "action_loss_avg": 0.025444680266082285,
      "epoch": 1.8444061279296875e-05,
      "step": 4835
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 4835,
      "training_step_time": 10.245808839797974
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 3.146477222442627,
      "step": 4835
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 4835,
      "training_step_time": 9.919485569000244
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 3.1421315670013428,
      "step": 4835
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 4835,
      "training_step_time": 9.919903039932251
    },
    {
      "epoch": 1.8444061279296875e-05,
      "model_forward_time": 3.1396737098693848,
      "step": 4835
    },
    {
      "epoch": 1.8444061279296875e-05,
      "step": 4835,
      "training_step_time": 10.28768253326416
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 3.171255111694336,
      "step": 4836
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 4836,
      "training_step_time": 9.935523271560669
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 3.4341883659362793,
      "step": 4836
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 4836,
      "training_step_time": 10.19819951057434
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 3.1429991722106934,
      "step": 4836
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 4836,
      "training_step_time": 9.906638622283936
    },
    {
      "epoch": 1.84478759765625e-05,
      "model_forward_time": 3.1475517749786377,
      "step": 4836
    },
    {
      "epoch": 1.84478759765625e-05,
      "step": 4836,
      "training_step_time": 10.279319286346436
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 3.18437123298645,
      "step": 4837
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 4837,
      "training_step_time": 9.956658124923706
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 3.133085012435913,
      "step": 4837
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 4837,
      "training_step_time": 9.900102376937866
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 3.1472601890563965,
      "step": 4837
    },
    {
      "dynamics_loss_avg": 0.07096776068210602,
      "epoch": 1.8451690673828124e-05,
      "step": 4837
    },
    {
      "action_loss_avg": 0.031124954484403132,
      "epoch": 1.8451690673828124e-05,
      "step": 4837
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 4837,
      "training_step_time": 10.202330589294434
    },
    {
      "epoch": 1.8451690673828124e-05,
      "model_forward_time": 3.1349291801452637,
      "step": 4837
    },
    {
      "epoch": 1.8451690673828124e-05,
      "step": 4837,
      "training_step_time": 10.302923440933228
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 3.1682658195495605,
      "step": 4838
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 4838,
      "training_step_time": 9.939245700836182
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 3.135178565979004,
      "step": 4838
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 4838,
      "training_step_time": 9.90710186958313
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 3.1407206058502197,
      "step": 4838
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 4838,
      "training_step_time": 9.910973310470581
    },
    {
      "epoch": 1.845550537109375e-05,
      "model_forward_time": 3.415431022644043,
      "step": 4838
    },
    {
      "epoch": 1.845550537109375e-05,
      "step": 4838,
      "training_step_time": 10.295889139175415
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 3.183962345123291,
      "step": 4839
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 4839,
      "training_step_time": 9.95542573928833
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 3.140868663787842,
      "step": 4839
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 4839,
      "training_step_time": 9.909362554550171
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 3.1454052925109863,
      "step": 4839
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 4839,
      "training_step_time": 9.912758588790894
    },
    {
      "epoch": 1.8459320068359377e-05,
      "model_forward_time": 3.133082866668701,
      "step": 4839
    },
    {
      "epoch": 1.8459320068359377e-05,
      "step": 4839,
      "training_step_time": 10.02255916595459
    },
    {
      "epoch": 1.8463134765625e-05,
      "grad_norm": 0.29726642370224,
      "learning_rate": 1e-05,
      "loss": 0.0866,
      "step": 4840
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 3.1699907779693604,
      "step": 4840
    },
    {
      "dynamics_loss_avg": 0.05447012409567833,
      "epoch": 1.8463134765625e-05,
      "step": 4840
    },
    {
      "action_loss_avg": 0.018985549360513686,
      "epoch": 1.8463134765625e-05,
      "step": 4840
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 4840,
      "training_step_time": 10.213596820831299
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 3.138901472091675,
      "step": 4840
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 4840,
      "training_step_time": 9.907788515090942
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 3.1420445442199707,
      "step": 4840
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 4840,
      "training_step_time": 9.911197185516357
    },
    {
      "epoch": 1.8463134765625e-05,
      "model_forward_time": 3.1392173767089844,
      "step": 4840
    },
    {
      "epoch": 1.8463134765625e-05,
      "step": 4840,
      "training_step_time": 10.276813507080078
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 3.1921005249023438,
      "step": 4841
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 4841,
      "training_step_time": 9.962987661361694
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 3.42204213142395,
      "step": 4841
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 4841,
      "training_step_time": 10.190612077713013
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 3.140580892562866,
      "step": 4841
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 4841,
      "training_step_time": 9.90878939628601
    },
    {
      "epoch": 1.8466949462890626e-05,
      "model_forward_time": 3.1345574855804443,
      "step": 4841
    },
    {
      "epoch": 1.8466949462890626e-05,
      "step": 4841,
      "training_step_time": 10.256640911102295
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 3.1728596687316895,
      "step": 4842
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 4842,
      "training_step_time": 9.939544439315796
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 3.1367812156677246,
      "step": 4842
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 4842,
      "training_step_time": 9.904640674591064
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 3.417924404144287,
      "step": 4842
    },
    {
      "dynamics_loss_avg": 0.05349623151123524,
      "epoch": 1.847076416015625e-05,
      "step": 4842
    },
    {
      "action_loss_avg": 0.019253035355359317,
      "epoch": 1.847076416015625e-05,
      "step": 4842
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 4842,
      "training_step_time": 10.18052625656128
    },
    {
      "epoch": 1.847076416015625e-05,
      "model_forward_time": 3.1333165168762207,
      "step": 4842
    },
    {
      "epoch": 1.847076416015625e-05,
      "step": 4842,
      "training_step_time": 10.25877857208252
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 3.168975353240967,
      "step": 4843
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 4843,
      "training_step_time": 9.933879137039185
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 3.1363956928253174,
      "step": 4843
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 4843,
      "training_step_time": 9.900935649871826
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 3.1280903816223145,
      "step": 4843
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 4843,
      "training_step_time": 9.888422012329102
    },
    {
      "epoch": 1.8474578857421875e-05,
      "model_forward_time": 3.3926379680633545,
      "step": 4843
    },
    {
      "epoch": 1.8474578857421875e-05,
      "step": 4843,
      "training_step_time": 10.280794382095337
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 3.1733386516571045,
      "step": 4844
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 4844,
      "training_step_time": 9.941020488739014
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 3.1370503902435303,
      "step": 4844
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 4844,
      "training_step_time": 9.905275583267212
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 3.1300089359283447,
      "step": 4844
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 4844,
      "training_step_time": 9.897359371185303
    },
    {
      "epoch": 1.84783935546875e-05,
      "model_forward_time": 3.1367039680480957,
      "step": 4844
    },
    {
      "epoch": 1.84783935546875e-05,
      "step": 4844,
      "training_step_time": 10.002955436706543
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 3.4208574295043945,
      "step": 4845
    },
    {
      "dynamics_loss_avg": 0.058324958011507985,
      "epoch": 1.8482208251953124e-05,
      "step": 4845
    },
    {
      "action_loss_avg": 0.027689338382333516,
      "epoch": 1.8482208251953124e-05,
      "step": 4845
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 4845,
      "training_step_time": 10.18723201751709
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 3.1367146968841553,
      "step": 4845
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 4845,
      "training_step_time": 9.90334415435791
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 3.1419951915740967,
      "step": 4845
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 4845,
      "training_step_time": 9.913414239883423
    },
    {
      "epoch": 1.8482208251953124e-05,
      "model_forward_time": 3.1420211791992188,
      "step": 4845
    },
    {
      "epoch": 1.8482208251953124e-05,
      "step": 4845,
      "training_step_time": 10.287176132202148
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 3.172132730484009,
      "step": 4846
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 4846,
      "training_step_time": 9.944782018661499
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 3.134629249572754,
      "step": 4846
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 4846,
      "training_step_time": 10.186755657196045
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 3.1346051692962646,
      "step": 4846
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 4846,
      "training_step_time": 9.907834529876709
    },
    {
      "epoch": 1.848602294921875e-05,
      "model_forward_time": 3.1400701999664307,
      "step": 4846
    },
    {
      "epoch": 1.848602294921875e-05,
      "step": 4846,
      "training_step_time": 10.29025673866272
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 3.1717705726623535,
      "step": 4847
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 4847,
      "training_step_time": 9.949557542800903
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 3.141893148422241,
      "step": 4847
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 4847,
      "training_step_time": 9.914299964904785
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 3.427515745162964,
      "step": 4847
    },
    {
      "dynamics_loss_avg": 0.07405048385262489,
      "epoch": 1.8489837646484377e-05,
      "step": 4847
    },
    {
      "action_loss_avg": 0.04118769709020853,
      "epoch": 1.8489837646484377e-05,
      "step": 4847
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 4847,
      "training_step_time": 10.198171377182007
    },
    {
      "epoch": 1.8489837646484377e-05,
      "model_forward_time": 3.143017292022705,
      "step": 4847
    },
    {
      "epoch": 1.8489837646484377e-05,
      "step": 4847,
      "training_step_time": 10.277310848236084
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 3.1718525886535645,
      "step": 4848
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 4848,
      "training_step_time": 9.938747644424438
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 3.141432523727417,
      "step": 4848
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 4848,
      "training_step_time": 9.91120195388794
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 3.1394882202148438,
      "step": 4848
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 4848,
      "training_step_time": 9.906174182891846
    },
    {
      "epoch": 1.849365234375e-05,
      "model_forward_time": 3.1341118812561035,
      "step": 4848
    },
    {
      "epoch": 1.849365234375e-05,
      "step": 4848,
      "training_step_time": 10.268288850784302
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 3.1760900020599365,
      "step": 4849
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 4849,
      "training_step_time": 9.94786286354065
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 3.13706111907959,
      "step": 4849
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 4849,
      "training_step_time": 9.909869909286499
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 3.1355652809143066,
      "step": 4849
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 4849,
      "training_step_time": 9.913506507873535
    },
    {
      "epoch": 1.8497467041015626e-05,
      "model_forward_time": 3.141746759414673,
      "step": 4849
    },
    {
      "epoch": 1.8497467041015626e-05,
      "step": 4849,
      "training_step_time": 10.029609441757202
    },
    {
      "epoch": 1.850128173828125e-05,
      "grad_norm": 0.49247097969055176,
      "learning_rate": 1e-05,
      "loss": 0.0904,
      "step": 4850
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 3.4935665130615234,
      "step": 4850
    },
    {
      "dynamics_loss_avg": 0.046584862470626834,
      "epoch": 1.850128173828125e-05,
      "step": 4850
    },
    {
      "action_loss_avg": 0.01608866499736905,
      "epoch": 1.850128173828125e-05,
      "step": 4850
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 4850,
      "training_step_time": 10.272727727890015
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 3.1443934440612793,
      "step": 4850
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 4850,
      "training_step_time": 9.91515040397644
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 3.150521993637085,
      "step": 4850
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 4850,
      "training_step_time": 9.924184560775757
    },
    {
      "epoch": 1.850128173828125e-05,
      "model_forward_time": 3.1450107097625732,
      "step": 4850
    },
    {
      "epoch": 1.850128173828125e-05,
      "step": 4850,
      "training_step_time": 10.009631872177124
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 3.1840908527374268,
      "step": 4851
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 4851,
      "training_step_time": 9.958338499069214
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 3.1453263759613037,
      "step": 4851
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 4851,
      "training_step_time": 10.18139910697937
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 3.1418163776397705,
      "step": 4851
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 4851,
      "training_step_time": 9.910672903060913
    },
    {
      "epoch": 1.8505096435546875e-05,
      "model_forward_time": 3.1393990516662598,
      "step": 4851
    },
    {
      "epoch": 1.8505096435546875e-05,
      "step": 4851,
      "training_step_time": 10.284334421157837
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 3.1935789585113525,
      "step": 4852
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 4852,
      "training_step_time": 9.962919235229492
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 2.4359118938446045,
      "step": 4852
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 4852,
      "training_step_time": 7.461029767990112
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 3.4303438663482666,
      "step": 4852
    },
    {
      "dynamics_loss_avg": 0.07195643335580826,
      "epoch": 1.85089111328125e-05,
      "step": 4852
    },
    {
      "action_loss_avg": 0.03287594951689243,
      "epoch": 1.85089111328125e-05,
      "step": 4852
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 4852,
      "training_step_time": 10.197046995162964
    },
    {
      "epoch": 1.85089111328125e-05,
      "model_forward_time": 3.141571044921875,
      "step": 4852
    },
    {
      "epoch": 1.85089111328125e-05,
      "step": 4852,
      "training_step_time": 10.247857570648193
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 3.1655335426330566,
      "step": 4853
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 4853,
      "training_step_time": 9.936963081359863
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 3.142753839492798,
      "step": 4853
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 4853,
      "training_step_time": 9.91051173210144
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 3.141679525375366,
      "step": 4853
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 4853,
      "training_step_time": 9.912292718887329
    },
    {
      "epoch": 1.8512725830078124e-05,
      "model_forward_time": 3.1439292430877686,
      "step": 4853
    },
    {
      "epoch": 1.8512725830078124e-05,
      "step": 4853,
      "training_step_time": 10.343194723129272
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 3.458601951599121,
      "step": 4854
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 4854,
      "training_step_time": 10.231844663619995
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 3.142629384994507,
      "step": 4854
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 4854,
      "training_step_time": 9.910877466201782
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 3.136352062225342,
      "step": 4854
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 4854,
      "training_step_time": 9.904823541641235
    },
    {
      "epoch": 1.851654052734375e-05,
      "model_forward_time": 3.1390793323516846,
      "step": 4854
    },
    {
      "epoch": 1.851654052734375e-05,
      "step": 4854,
      "training_step_time": 10.009374618530273
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 3.4311962127685547,
      "step": 4855
    },
    {
      "dynamics_loss_avg": 0.06172127965837717,
      "epoch": 1.8520355224609377e-05,
      "step": 4855
    },
    {
      "action_loss_avg": 0.029793252423405647,
      "epoch": 1.8520355224609377e-05,
      "step": 4855
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 4855,
      "training_step_time": 10.201527118682861
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 3.1490466594696045,
      "step": 4855
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 4855,
      "training_step_time": 9.91718077659607
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 3.136932611465454,
      "step": 4855
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 4855,
      "training_step_time": 9.905158996582031
    },
    {
      "epoch": 1.8520355224609377e-05,
      "model_forward_time": 3.139010429382324,
      "step": 4855
    },
    {
      "epoch": 1.8520355224609377e-05,
      "step": 4855,
      "training_step_time": 10.012766599655151
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 3.171083450317383,
      "step": 4856
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 4856,
      "training_step_time": 9.938811540603638
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 3.134807825088501,
      "step": 4856
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 4856,
      "training_step_time": 10.170608282089233
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 3.1345057487487793,
      "step": 4856
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 4856,
      "training_step_time": 9.908467292785645
    },
    {
      "epoch": 1.8524169921875e-05,
      "model_forward_time": 3.1403720378875732,
      "step": 4856
    },
    {
      "epoch": 1.8524169921875e-05,
      "step": 4856,
      "training_step_time": 10.257798194885254
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 3.1895339488983154,
      "step": 4857
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 4857,
      "training_step_time": 9.957556009292603
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 3.1350274085998535,
      "step": 4857
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 4857,
      "training_step_time": 9.901102066040039
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 3.137300491333008,
      "step": 4857
    },
    {
      "dynamics_loss_avg": 0.07103804238140583,
      "epoch": 1.8527984619140626e-05,
      "step": 4857
    },
    {
      "action_loss_avg": 0.029740778170526026,
      "epoch": 1.8527984619140626e-05,
      "step": 4857
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 4857,
      "training_step_time": 10.182945966720581
    },
    {
      "epoch": 1.8527984619140626e-05,
      "model_forward_time": 3.1421947479248047,
      "step": 4857
    },
    {
      "epoch": 1.8527984619140626e-05,
      "step": 4857,
      "training_step_time": 10.313145875930786
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 3.1820456981658936,
      "step": 4858
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 4858,
      "training_step_time": 9.95382833480835
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 3.1325125694274902,
      "step": 4858
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 4858,
      "training_step_time": 9.899533033370972
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 3.1389780044555664,
      "step": 4858
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 4858,
      "training_step_time": 9.903120756149292
    },
    {
      "epoch": 1.853179931640625e-05,
      "model_forward_time": 3.418607234954834,
      "step": 4858
    },
    {
      "epoch": 1.853179931640625e-05,
      "step": 4858,
      "training_step_time": 10.287062406539917
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 3.1727678775787354,
      "step": 4859
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 4859,
      "training_step_time": 9.942961931228638
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 2.4346141815185547,
      "step": 4859
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 4859,
      "training_step_time": 7.463193655014038
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 3.140658140182495,
      "step": 4859
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 4859,
      "training_step_time": 9.91497015953064
    },
    {
      "epoch": 1.8535614013671875e-05,
      "model_forward_time": 3.1378896236419678,
      "step": 4859
    },
    {
      "epoch": 1.8535614013671875e-05,
      "step": 4859,
      "training_step_time": 10.018314361572266
    },
    {
      "epoch": 1.85394287109375e-05,
      "grad_norm": 0.4657436013221741,
      "learning_rate": 1e-05,
      "loss": 0.0898,
      "step": 4860
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 3.183534622192383,
      "step": 4860
    },
    {
      "dynamics_loss_avg": 0.04682825841009617,
      "epoch": 1.85394287109375e-05,
      "step": 4860
    },
    {
      "action_loss_avg": 0.021875719912350176,
      "epoch": 1.85394287109375e-05,
      "step": 4860
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 4860,
      "training_step_time": 9.96261739730835
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 3.388150453567505,
      "step": 4860
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 4860,
      "training_step_time": 10.157914400100708
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 3.1411244869232178,
      "step": 4860
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 4860,
      "training_step_time": 9.909428834915161
    },
    {
      "epoch": 1.85394287109375e-05,
      "model_forward_time": 3.136106491088867,
      "step": 4860
    },
    {
      "epoch": 1.85394287109375e-05,
      "step": 4860,
      "training_step_time": 10.255561590194702
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 3.1815075874328613,
      "step": 4861
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 4861,
      "training_step_time": 9.95634150505066
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 3.405885696411133,
      "step": 4861
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 4861,
      "training_step_time": 10.176887512207031
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 3.1404120922088623,
      "step": 4861
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 4861,
      "training_step_time": 9.915877103805542
    },
    {
      "epoch": 1.8543243408203124e-05,
      "model_forward_time": 3.1395702362060547,
      "step": 4861
    },
    {
      "epoch": 1.8543243408203124e-05,
      "step": 4861,
      "training_step_time": 10.021879434585571
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 3.1862919330596924,
      "step": 4862
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 4862,
      "training_step_time": 9.965043783187866
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 3.15063738822937,
      "step": 4862
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 4862,
      "training_step_time": 9.930607318878174
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 3.149742841720581,
      "step": 4862
    },
    {
      "dynamics_loss_avg": 0.06110396105796099,
      "epoch": 1.854705810546875e-05,
      "step": 4862
    },
    {
      "action_loss_avg": 0.027499700896441936,
      "epoch": 1.854705810546875e-05,
      "step": 4862
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 4862,
      "training_step_time": 10.213184356689453
    },
    {
      "epoch": 1.854705810546875e-05,
      "model_forward_time": 3.1505892276763916,
      "step": 4862
    },
    {
      "epoch": 1.854705810546875e-05,
      "step": 4862,
      "training_step_time": 10.308035612106323
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 3.174931526184082,
      "step": 4863
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 4863,
      "training_step_time": 9.95143747329712
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 3.1411309242248535,
      "step": 4863
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 4863,
      "training_step_time": 9.919775485992432
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 3.137814521789551,
      "step": 4863
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 4863,
      "training_step_time": 9.911801815032959
    },
    {
      "epoch": 1.8550872802734377e-05,
      "model_forward_time": 3.4189085960388184,
      "step": 4863
    },
    {
      "epoch": 1.8550872802734377e-05,
      "step": 4863,
      "training_step_time": 10.303750991821289
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 3.1664483547210693,
      "step": 4864
    },
    {
      "epoch": 1.85546875e-05,
      "step": 4864,
      "training_step_time": 9.945189237594604
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 3.1412644386291504,
      "step": 4864
    },
    {
      "epoch": 1.85546875e-05,
      "step": 4864,
      "training_step_time": 9.916014671325684
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 3.1393635272979736,
      "step": 4864
    },
    {
      "epoch": 1.85546875e-05,
      "step": 4864,
      "training_step_time": 9.91411566734314
    },
    {
      "epoch": 1.85546875e-05,
      "model_forward_time": 3.1399056911468506,
      "step": 4864
    },
    {
      "epoch": 1.85546875e-05,
      "step": 4864,
      "training_step_time": 10.02360224723816
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 3.469233274459839,
      "step": 4865
    },
    {
      "dynamics_loss_avg": 0.05895736962556839,
      "epoch": 1.8558502197265626e-05,
      "step": 4865
    },
    {
      "action_loss_avg": 0.023604161385446788,
      "epoch": 1.8558502197265626e-05,
      "step": 4865
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 4865,
      "training_step_time": 10.252960920333862
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 3.1476316452026367,
      "step": 4865
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 4865,
      "training_step_time": 9.91866683959961
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 3.14165997505188,
      "step": 4865
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 4865,
      "training_step_time": 9.909721851348877
    },
    {
      "epoch": 1.8558502197265626e-05,
      "model_forward_time": 3.1421260833740234,
      "step": 4865
    },
    {
      "epoch": 1.8558502197265626e-05,
      "step": 4865,
      "training_step_time": 10.014817476272583
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 3.186129331588745,
      "step": 4866
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 4866,
      "training_step_time": 9.956728219985962
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 3.4171876907348633,
      "step": 4866
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 4866,
      "training_step_time": 10.190410375595093
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 3.1492066383361816,
      "step": 4866
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 4866,
      "training_step_time": 9.927236080169678
    },
    {
      "epoch": 1.856231689453125e-05,
      "model_forward_time": 3.1456305980682373,
      "step": 4866
    },
    {
      "epoch": 1.856231689453125e-05,
      "step": 4866,
      "training_step_time": 10.324872255325317
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 3.188727617263794,
      "step": 4867
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 4867,
      "training_step_time": 9.968616008758545
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 3.141827344894409,
      "step": 4867
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 4867,
      "training_step_time": 9.92016077041626
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 3.395878791809082,
      "step": 4867
    },
    {
      "dynamics_loss_avg": 0.06740557551383972,
      "epoch": 1.8566131591796875e-05,
      "step": 4867
    },
    {
      "action_loss_avg": 0.032568056229501964,
      "epoch": 1.8566131591796875e-05,
      "step": 4867
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 4867,
      "training_step_time": 10.17033863067627
    },
    {
      "epoch": 1.8566131591796875e-05,
      "model_forward_time": 3.142962694168091,
      "step": 4867
    },
    {
      "epoch": 1.8566131591796875e-05,
      "step": 4867,
      "training_step_time": 10.274615049362183
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 3.188021421432495,
      "step": 4868
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 4868,
      "training_step_time": 9.94980764389038
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 3.1350722312927246,
      "step": 4868
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 4868,
      "training_step_time": 9.897436380386353
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 3.1420648097991943,
      "step": 4868
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 4868,
      "training_step_time": 9.904107093811035
    },
    {
      "epoch": 1.85699462890625e-05,
      "model_forward_time": 3.4143388271331787,
      "step": 4868
    },
    {
      "epoch": 1.85699462890625e-05,
      "step": 4868,
      "training_step_time": 10.286335229873657
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 3.1814815998077393,
      "step": 4869
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 4869,
      "training_step_time": 9.947519063949585
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 3.1374762058258057,
      "step": 4869
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 4869,
      "training_step_time": 9.90448522567749
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 3.147810459136963,
      "step": 4869
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 4869,
      "training_step_time": 9.915360927581787
    },
    {
      "epoch": 1.8573760986328124e-05,
      "model_forward_time": 3.1421244144439697,
      "step": 4869
    },
    {
      "epoch": 1.8573760986328124e-05,
      "step": 4869,
      "training_step_time": 10.029525995254517
    },
    {
      "epoch": 1.857757568359375e-05,
      "grad_norm": 0.6323255300521851,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 4870
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 3.4503116607666016,
      "step": 4870
    },
    {
      "dynamics_loss_avg": 0.06313646622002125,
      "epoch": 1.857757568359375e-05,
      "step": 4870
    },
    {
      "action_loss_avg": 0.03555097607895732,
      "epoch": 1.857757568359375e-05,
      "step": 4870
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 4870,
      "training_step_time": 10.22430419921875
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 3.1418960094451904,
      "step": 4870
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 4870,
      "training_step_time": 9.909270286560059
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 3.141547441482544,
      "step": 4870
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 4870,
      "training_step_time": 9.902770519256592
    },
    {
      "epoch": 1.857757568359375e-05,
      "model_forward_time": 3.1423816680908203,
      "step": 4870
    },
    {
      "epoch": 1.857757568359375e-05,
      "step": 4870,
      "training_step_time": 10.011396884918213
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 3.179244041442871,
      "step": 4871
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 4871,
      "training_step_time": 9.944917917251587
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 3.3918211460113525,
      "step": 4871
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 4871,
      "training_step_time": 10.15809941291809
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 3.1431148052215576,
      "step": 4871
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 4871,
      "training_step_time": 9.914416313171387
    },
    {
      "epoch": 1.8581390380859377e-05,
      "model_forward_time": 3.142119884490967,
      "step": 4871
    },
    {
      "epoch": 1.8581390380859377e-05,
      "step": 4871,
      "training_step_time": 10.274463176727295
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 3.172290563583374,
      "step": 4872
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 4872,
      "training_step_time": 9.946905612945557
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 3.14652156829834,
      "step": 4872
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 4872,
      "training_step_time": 9.919708728790283
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 3.427037477493286,
      "step": 4872
    },
    {
      "dynamics_loss_avg": 0.06587136760354043,
      "epoch": 1.8585205078125e-05,
      "step": 4872
    },
    {
      "action_loss_avg": 0.031511380523443225,
      "epoch": 1.8585205078125e-05,
      "step": 4872
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 4872,
      "training_step_time": 10.20361614227295
    },
    {
      "epoch": 1.8585205078125e-05,
      "model_forward_time": 3.1501963138580322,
      "step": 4872
    },
    {
      "epoch": 1.8585205078125e-05,
      "step": 4872,
      "training_step_time": 10.272900342941284
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 3.185391902923584,
      "step": 4873
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 4873,
      "training_step_time": 9.962188243865967
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 3.139537811279297,
      "step": 4873
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 4873,
      "training_step_time": 9.915682315826416
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 3.1410036087036133,
      "step": 4873
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 4873,
      "training_step_time": 9.92073369026184
    },
    {
      "epoch": 1.8589019775390626e-05,
      "model_forward_time": 3.3940579891204834,
      "step": 4873
    },
    {
      "epoch": 1.8589019775390626e-05,
      "step": 4873,
      "training_step_time": 10.307578802108765
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 3.168386697769165,
      "step": 4874
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 4874,
      "training_step_time": 9.942737817764282
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 3.1444294452667236,
      "step": 4874
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 4874,
      "training_step_time": 9.921390056610107
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 3.142618179321289,
      "step": 4874
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 4874,
      "training_step_time": 9.924602270126343
    },
    {
      "epoch": 1.859283447265625e-05,
      "model_forward_time": 3.1467413902282715,
      "step": 4874
    },
    {
      "epoch": 1.859283447265625e-05,
      "step": 4874,
      "training_step_time": 10.036157846450806
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 3.4530599117279053,
      "step": 4875
    },
    {
      "dynamics_loss_avg": 0.05154891479760408,
      "epoch": 1.8596649169921875e-05,
      "step": 4875
    },
    {
      "action_loss_avg": 0.023949236422777177,
      "epoch": 1.8596649169921875e-05,
      "step": 4875
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 4875,
      "training_step_time": 10.229466676712036
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 2.435255527496338,
      "step": 4875
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 4875,
      "training_step_time": 7.46523380279541
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 3.1392321586608887,
      "step": 4875
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 4875,
      "training_step_time": 9.910577774047852
    },
    {
      "epoch": 1.8596649169921875e-05,
      "model_forward_time": 3.1409313678741455,
      "step": 4875
    },
    {
      "epoch": 1.8596649169921875e-05,
      "step": 4875,
      "training_step_time": 10.247006177902222
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 3.1739590167999268,
      "step": 4876
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 4876,
      "training_step_time": 9.947201490402222
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 3.437211036682129,
      "step": 4876
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 4876,
      "training_step_time": 10.212602615356445
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 3.143080234527588,
      "step": 4876
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 4876,
      "training_step_time": 9.919032573699951
    },
    {
      "epoch": 1.86004638671875e-05,
      "model_forward_time": 3.1468420028686523,
      "step": 4876
    },
    {
      "epoch": 1.86004638671875e-05,
      "step": 4876,
      "training_step_time": 10.307997941970825
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 3.1858789920806885,
      "step": 4877
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 4877,
      "training_step_time": 9.961610078811646
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 3.139188528060913,
      "step": 4877
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 4877,
      "training_step_time": 9.914193153381348
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 3.421555280685425,
      "step": 4877
    },
    {
      "dynamics_loss_avg": 0.0699070168659091,
      "epoch": 1.8604278564453124e-05,
      "step": 4877
    },
    {
      "action_loss_avg": 0.029324750974774362,
      "epoch": 1.8604278564453124e-05,
      "step": 4877
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 4877,
      "training_step_time": 10.194632768630981
    },
    {
      "epoch": 1.8604278564453124e-05,
      "model_forward_time": 3.1469860076904297,
      "step": 4877
    },
    {
      "epoch": 1.8604278564453124e-05,
      "step": 4877,
      "training_step_time": 10.293721437454224
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 3.178894281387329,
      "step": 4878
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 4878,
      "training_step_time": 9.956055879592896
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 3.1421923637390137,
      "step": 4878
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 4878,
      "training_step_time": 9.912587881088257
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 3.1426920890808105,
      "step": 4878
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 4878,
      "training_step_time": 9.911123991012573
    },
    {
      "epoch": 1.860809326171875e-05,
      "model_forward_time": 3.427844762802124,
      "step": 4878
    },
    {
      "epoch": 1.860809326171875e-05,
      "step": 4878,
      "training_step_time": 10.289175987243652
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 3.181303024291992,
      "step": 4879
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 4879,
      "training_step_time": 9.95572018623352
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 3.1380257606506348,
      "step": 4879
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 4879,
      "training_step_time": 9.909421682357788
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 3.1410038471221924,
      "step": 4879
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 4879,
      "training_step_time": 9.913301944732666
    },
    {
      "epoch": 1.8611907958984373e-05,
      "model_forward_time": 3.139216661453247,
      "step": 4879
    },
    {
      "epoch": 1.8611907958984373e-05,
      "step": 4879,
      "training_step_time": 10.027417421340942
    },
    {
      "epoch": 1.861572265625e-05,
      "grad_norm": 0.5486325025558472,
      "learning_rate": 1e-05,
      "loss": 0.093,
      "step": 4880
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 3.4436299800872803,
      "step": 4880
    },
    {
      "dynamics_loss_avg": 0.07480740509927272,
      "epoch": 1.861572265625e-05,
      "step": 4880
    },
    {
      "action_loss_avg": 0.033218704909086225,
      "epoch": 1.861572265625e-05,
      "step": 4880
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 4880,
      "training_step_time": 10.220402717590332
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 3.1401586532592773,
      "step": 4880
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 4880,
      "training_step_time": 9.913061380386353
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 3.141515016555786,
      "step": 4880
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 4880,
      "training_step_time": 9.925595045089722
    },
    {
      "epoch": 1.861572265625e-05,
      "model_forward_time": 3.1439154148101807,
      "step": 4880
    },
    {
      "epoch": 1.861572265625e-05,
      "step": 4880,
      "training_step_time": 10.020477771759033
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 3.177953004837036,
      "step": 4881
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 4881,
      "training_step_time": 9.957992792129517
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 3.4016687870025635,
      "step": 4881
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 4881,
      "training_step_time": 10.181010246276855
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 3.143643617630005,
      "step": 4881
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 4881,
      "training_step_time": 9.926267385482788
    },
    {
      "epoch": 1.8619537353515626e-05,
      "model_forward_time": 3.1400656700134277,
      "step": 4881
    },
    {
      "epoch": 1.8619537353515626e-05,
      "step": 4881,
      "training_step_time": 10.308825492858887
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 3.168275833129883,
      "step": 4882
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 4882,
      "training_step_time": 9.948232650756836
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 3.144007682800293,
      "step": 4882
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 4882,
      "training_step_time": 9.920547008514404
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 3.421201467514038,
      "step": 4882
    },
    {
      "dynamics_loss_avg": 0.06201475784182549,
      "epoch": 1.862335205078125e-05,
      "step": 4882
    },
    {
      "action_loss_avg": 0.02721219165250659,
      "epoch": 1.862335205078125e-05,
      "step": 4882
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 4882,
      "training_step_time": 10.19923734664917
    },
    {
      "epoch": 1.862335205078125e-05,
      "model_forward_time": 3.141939401626587,
      "step": 4882
    },
    {
      "epoch": 1.862335205078125e-05,
      "step": 4882,
      "training_step_time": 10.277448892593384
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 3.1689727306365967,
      "step": 4883
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 4883,
      "training_step_time": 9.945943832397461
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 3.141252040863037,
      "step": 4883
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 4883,
      "training_step_time": 9.925431489944458
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 3.1437838077545166,
      "step": 4883
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 4883,
      "training_step_time": 9.923137187957764
    },
    {
      "epoch": 1.8627166748046875e-05,
      "model_forward_time": 3.3986704349517822,
      "step": 4883
    },
    {
      "epoch": 1.8627166748046875e-05,
      "step": 4883,
      "training_step_time": 10.322926044464111
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 3.169522523880005,
      "step": 4884
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 4884,
      "training_step_time": 9.953349590301514
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 3.1451869010925293,
      "step": 4884
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 4884,
      "training_step_time": 9.924473285675049
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 3.1434857845306396,
      "step": 4884
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 4884,
      "training_step_time": 9.922768354415894
    },
    {
      "epoch": 1.86309814453125e-05,
      "model_forward_time": 3.144679069519043,
      "step": 4884
    },
    {
      "epoch": 1.86309814453125e-05,
      "step": 4884,
      "training_step_time": 10.3109450340271
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 3.4478418827056885,
      "step": 4885
    },
    {
      "dynamics_loss_avg": 0.07225769646465778,
      "epoch": 1.8634796142578124e-05,
      "step": 4885
    },
    {
      "action_loss_avg": 0.02762833908200264,
      "epoch": 1.8634796142578124e-05,
      "step": 4885
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 4885,
      "training_step_time": 10.229292392730713
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 3.151655673980713,
      "step": 4885
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 4885,
      "training_step_time": 9.929842233657837
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 3.1449992656707764,
      "step": 4885
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 4885,
      "training_step_time": 9.921830415725708
    },
    {
      "epoch": 1.8634796142578124e-05,
      "model_forward_time": 3.1447627544403076,
      "step": 4885
    },
    {
      "epoch": 1.8634796142578124e-05,
      "step": 4885,
      "training_step_time": 10.016638994216919
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 2.4800407886505127,
      "step": 4886
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 4886,
      "training_step_time": 7.512179851531982
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 3.138977527618408,
      "step": 4886
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 4886,
      "training_step_time": 10.190244674682617
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 3.141986131668091,
      "step": 4886
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 4886,
      "training_step_time": 9.914330959320068
    },
    {
      "epoch": 1.863861083984375e-05,
      "model_forward_time": 3.147810459136963,
      "step": 4886
    },
    {
      "epoch": 1.863861083984375e-05,
      "step": 4886,
      "training_step_time": 10.016904354095459
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 3.184445381164551,
      "step": 4887
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 4887,
      "training_step_time": 9.957340240478516
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 3.1425113677978516,
      "step": 4887
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 4887,
      "training_step_time": 9.912961959838867
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 3.4097065925598145,
      "step": 4887
    },
    {
      "dynamics_loss_avg": 0.07362268511205912,
      "epoch": 1.8642425537109373e-05,
      "step": 4887
    },
    {
      "action_loss_avg": 0.0346579615958035,
      "epoch": 1.8642425537109373e-05,
      "step": 4887
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 4887,
      "training_step_time": 10.187863111495972
    },
    {
      "epoch": 1.8642425537109373e-05,
      "model_forward_time": 3.146695375442505,
      "step": 4887
    },
    {
      "epoch": 1.8642425537109373e-05,
      "step": 4887,
      "training_step_time": 10.288959503173828
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 3.1873929500579834,
      "step": 4888
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 4888,
      "training_step_time": 9.962267398834229
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 3.143430709838867,
      "step": 4888
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 4888,
      "training_step_time": 9.921350002288818
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 3.15238094329834,
      "step": 4888
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 4888,
      "training_step_time": 9.930284976959229
    },
    {
      "epoch": 1.8646240234375e-05,
      "model_forward_time": 3.1467342376708984,
      "step": 4888
    },
    {
      "epoch": 1.8646240234375e-05,
      "step": 4888,
      "training_step_time": 10.296157836914062
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 3.183845043182373,
      "step": 4889
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 4889,
      "training_step_time": 9.973713397979736
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 3.1354379653930664,
      "step": 4889
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 4889,
      "training_step_time": 9.916055917739868
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 3.1427230834960938,
      "step": 4889
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 4889,
      "training_step_time": 9.920748233795166
    },
    {
      "epoch": 1.8650054931640626e-05,
      "model_forward_time": 3.134566068649292,
      "step": 4889
    },
    {
      "epoch": 1.8650054931640626e-05,
      "step": 4889,
      "training_step_time": 10.287833452224731
    },
    {
      "epoch": 1.865386962890625e-05,
      "grad_norm": 0.45858582854270935,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 4890
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 3.4837684631347656,
      "step": 4890
    },
    {
      "dynamics_loss_avg": 0.061264405772089955,
      "epoch": 1.865386962890625e-05,
      "step": 4890
    },
    {
      "action_loss_avg": 0.023936043307185174,
      "epoch": 1.865386962890625e-05,
      "step": 4890
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 4890,
      "training_step_time": 10.261775016784668
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 3.1412460803985596,
      "step": 4890
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 4890,
      "training_step_time": 9.916043043136597
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 3.1421329975128174,
      "step": 4890
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 4890,
      "training_step_time": 9.91285753250122
    },
    {
      "epoch": 1.865386962890625e-05,
      "model_forward_time": 3.1395726203918457,
      "step": 4890
    },
    {
      "epoch": 1.865386962890625e-05,
      "step": 4890,
      "training_step_time": 10.008052110671997
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 3.1826834678649902,
      "step": 4891
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 4891,
      "training_step_time": 9.954829216003418
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 3.144937038421631,
      "step": 4891
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 4891,
      "training_step_time": 10.184054374694824
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 3.1488118171691895,
      "step": 4891
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 4891,
      "training_step_time": 9.923188209533691
    },
    {
      "epoch": 1.8657684326171875e-05,
      "model_forward_time": 3.143193006515503,
      "step": 4891
    },
    {
      "epoch": 1.8657684326171875e-05,
      "step": 4891,
      "training_step_time": 10.273950576782227
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 3.1809189319610596,
      "step": 4892
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 4892,
      "training_step_time": 9.95971393585205
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 3.140383005142212,
      "step": 4892
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 4892,
      "training_step_time": 9.91494607925415
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 3.4272313117980957,
      "step": 4892
    },
    {
      "dynamics_loss_avg": 0.05803984981030226,
      "epoch": 1.86614990234375e-05,
      "step": 4892
    },
    {
      "action_loss_avg": 0.03039619466289878,
      "epoch": 1.86614990234375e-05,
      "step": 4892
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 4892,
      "training_step_time": 10.204432249069214
    },
    {
      "epoch": 1.86614990234375e-05,
      "model_forward_time": 3.13826322555542,
      "step": 4892
    },
    {
      "epoch": 1.86614990234375e-05,
      "step": 4892,
      "training_step_time": 10.251622200012207
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 3.1921775341033936,
      "step": 4893
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 4893,
      "training_step_time": 9.970773458480835
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 3.1399710178375244,
      "step": 4893
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 4893,
      "training_step_time": 9.919086933135986
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 3.137324810028076,
      "step": 4893
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 4893,
      "training_step_time": 9.918726682662964
    },
    {
      "epoch": 1.8665313720703124e-05,
      "model_forward_time": 3.420388698577881,
      "step": 4893
    },
    {
      "epoch": 1.8665313720703124e-05,
      "step": 4893,
      "training_step_time": 10.29727554321289
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 3.189941167831421,
      "step": 4894
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 4894,
      "training_step_time": 9.977494955062866
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 3.144256591796875,
      "step": 4894
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 4894,
      "training_step_time": 9.92439317703247
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 3.1378841400146484,
      "step": 4894
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 4894,
      "training_step_time": 9.926453590393066
    },
    {
      "epoch": 1.866912841796875e-05,
      "model_forward_time": 3.139296054840088,
      "step": 4894
    },
    {
      "epoch": 1.866912841796875e-05,
      "step": 4894,
      "training_step_time": 10.239060878753662
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 3.41796875,
      "step": 4895
    },
    {
      "dynamics_loss_avg": 0.05930924154818058,
      "epoch": 1.8672943115234373e-05,
      "step": 4895
    },
    {
      "action_loss_avg": 0.02951413537375629,
      "epoch": 1.8672943115234373e-05,
      "step": 4895
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 4895,
      "training_step_time": 10.203155279159546
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 3.140744924545288,
      "step": 4895
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 4895,
      "training_step_time": 9.921340703964233
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 2.4338083267211914,
      "step": 4895
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 4895,
      "training_step_time": 7.472433090209961
    },
    {
      "epoch": 1.8672943115234373e-05,
      "model_forward_time": 3.143864154815674,
      "step": 4895
    },
    {
      "epoch": 1.8672943115234373e-05,
      "step": 4895,
      "training_step_time": 10.256815910339355
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 3.1679139137268066,
      "step": 4896
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 4896,
      "training_step_time": 9.944797039031982
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 3.4563517570495605,
      "step": 4896
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 4896,
      "training_step_time": 10.22594666481018
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 3.1328670978546143,
      "step": 4896
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 4896,
      "training_step_time": 9.909102439880371
    },
    {
      "epoch": 1.86767578125e-05,
      "model_forward_time": 3.1325390338897705,
      "step": 4896
    },
    {
      "epoch": 1.86767578125e-05,
      "step": 4896,
      "training_step_time": 10.022465705871582
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 3.165914535522461,
      "step": 4897
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 4897,
      "training_step_time": 9.948290824890137
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 3.1329731941223145,
      "step": 4897
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 4897,
      "training_step_time": 9.912657022476196
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 3.137578010559082,
      "step": 4897
    },
    {
      "dynamics_loss_avg": 0.06398375295102596,
      "epoch": 1.8680572509765626e-05,
      "step": 4897
    },
    {
      "action_loss_avg": 0.029976280499249698,
      "epoch": 1.8680572509765626e-05,
      "step": 4897
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 4897,
      "training_step_time": 10.20564866065979
    },
    {
      "epoch": 1.8680572509765626e-05,
      "model_forward_time": 3.1370551586151123,
      "step": 4897
    },
    {
      "epoch": 1.8680572509765626e-05,
      "step": 4897,
      "training_step_time": 10.314631223678589
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 3.1812493801116943,
      "step": 4898
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 4898,
      "training_step_time": 9.961666584014893
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 3.1324782371520996,
      "step": 4898
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 4898,
      "training_step_time": 9.913062334060669
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 3.140421152114868,
      "step": 4898
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 4898,
      "training_step_time": 9.933653831481934
    },
    {
      "epoch": 1.868438720703125e-05,
      "model_forward_time": 3.46305513381958,
      "step": 4898
    },
    {
      "epoch": 1.868438720703125e-05,
      "step": 4898,
      "training_step_time": 10.342923879623413
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 3.18404483795166,
      "step": 4899
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 4899,
      "training_step_time": 9.962271928787231
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 3.1295952796936035,
      "step": 4899
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 4899,
      "training_step_time": 9.907140970230103
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 3.1366307735443115,
      "step": 4899
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 4899,
      "training_step_time": 9.914207220077515
    },
    {
      "epoch": 1.8688201904296875e-05,
      "model_forward_time": 3.1383631229400635,
      "step": 4899
    },
    {
      "epoch": 1.8688201904296875e-05,
      "step": 4899,
      "training_step_time": 10.248191118240356
    },
    {
      "epoch": 1.86920166015625e-05,
      "grad_norm": 0.5886891484260559,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 4900
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 3.1670053005218506,
      "step": 4900
    },
    {
      "dynamics_loss_avg": 0.06680948063731193,
      "epoch": 1.86920166015625e-05,
      "step": 4900
    },
    {
      "action_loss_avg": 0.029014912201091647,
      "epoch": 1.86920166015625e-05,
      "step": 4900
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 4900,
      "training_step_time": 10.223572015762329
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 3.1353800296783447,
      "step": 4900
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 4900,
      "training_step_time": 9.915788412094116
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 3.137061595916748,
      "step": 4900
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 4900,
      "training_step_time": 9.91154408454895
    },
    {
      "epoch": 1.86920166015625e-05,
      "model_forward_time": 3.135561466217041,
      "step": 4900
    },
    {
      "epoch": 1.86920166015625e-05,
      "step": 4900,
      "training_step_time": 10.281391620635986
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 3.181523323059082,
      "step": 4901
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 4901,
      "training_step_time": 9.952805042266846
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 3.4299750328063965,
      "step": 4901
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 4901,
      "training_step_time": 10.20062804222107
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 3.1372768878936768,
      "step": 4901
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 4901,
      "training_step_time": 9.909156560897827
    },
    {
      "epoch": 1.8695831298828124e-05,
      "model_forward_time": 3.132957935333252,
      "step": 4901
    },
    {
      "epoch": 1.8695831298828124e-05,
      "step": 4901,
      "training_step_time": 10.22279405593872
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 3.1702163219451904,
      "step": 4902
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 4902,
      "training_step_time": 9.940548181533813
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 3.1360509395599365,
      "step": 4902
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 4902,
      "training_step_time": 9.906933069229126
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 3.4217958450317383,
      "step": 4902
    },
    {
      "dynamics_loss_avg": 0.05393330305814743,
      "epoch": 1.869964599609375e-05,
      "step": 4902
    },
    {
      "action_loss_avg": 0.0218652724288404,
      "epoch": 1.869964599609375e-05,
      "step": 4902
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 4902,
      "training_step_time": 10.195558309555054
    },
    {
      "epoch": 1.869964599609375e-05,
      "model_forward_time": 3.1321778297424316,
      "step": 4902
    },
    {
      "epoch": 1.869964599609375e-05,
      "step": 4902,
      "training_step_time": 10.293997526168823
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 3.1538422107696533,
      "step": 4903
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 4903,
      "training_step_time": 9.926903009414673
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 3.1363725662231445,
      "step": 4903
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 4903,
      "training_step_time": 9.90947413444519
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 3.131517171859741,
      "step": 4903
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 4903,
      "training_step_time": 9.905519247055054
    },
    {
      "epoch": 1.8703460693359373e-05,
      "model_forward_time": 2.7056400775909424,
      "step": 4903
    },
    {
      "epoch": 1.8703460693359373e-05,
      "step": 4903,
      "training_step_time": 10.280240535736084
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 3.1710293292999268,
      "step": 4904
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 4904,
      "training_step_time": 9.945662021636963
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 3.1386075019836426,
      "step": 4904
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 4904,
      "training_step_time": 9.923448085784912
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 3.137500524520874,
      "step": 4904
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 4904,
      "training_step_time": 9.917789697647095
    },
    {
      "epoch": 1.8707275390625e-05,
      "model_forward_time": 3.137835741043091,
      "step": 4904
    },
    {
      "epoch": 1.8707275390625e-05,
      "step": 4904,
      "training_step_time": 10.032412767410278
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 3.1728627681732178,
      "step": 4905
    },
    {
      "dynamics_loss_avg": 0.07121104151010513,
      "epoch": 1.8711090087890626e-05,
      "step": 4905
    },
    {
      "action_loss_avg": 0.025618477165699004,
      "epoch": 1.8711090087890626e-05,
      "step": 4905
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 4905,
      "training_step_time": 10.235447645187378
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 3.1383328437805176,
      "step": 4905
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 4905,
      "training_step_time": 9.91981315612793
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 3.1373233795166016,
      "step": 4905
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 4905,
      "training_step_time": 9.915029287338257
    },
    {
      "epoch": 1.8711090087890626e-05,
      "model_forward_time": 3.140961170196533,
      "step": 4905
    },
    {
      "epoch": 1.8711090087890626e-05,
      "step": 4905,
      "training_step_time": 10.274420022964478
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 3.1778311729431152,
      "step": 4906
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 4906,
      "training_step_time": 9.958329439163208
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 3.1358652114868164,
      "step": 4906
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 4906,
      "training_step_time": 10.235748767852783
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 3.1337645053863525,
      "step": 4906
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 4906,
      "training_step_time": 9.911378860473633
    },
    {
      "epoch": 1.871490478515625e-05,
      "model_forward_time": 3.137240409851074,
      "step": 4906
    },
    {
      "epoch": 1.871490478515625e-05,
      "step": 4906,
      "training_step_time": 10.261884689331055
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 3.170935869216919,
      "step": 4907
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 4907,
      "training_step_time": 9.948586702346802
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 3.1339590549468994,
      "step": 4907
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 4907,
      "training_step_time": 9.913076877593994
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 3.1351518630981445,
      "step": 4907
    },
    {
      "dynamics_loss_avg": 0.0708728762343526,
      "epoch": 1.8718719482421875e-05,
      "step": 4907
    },
    {
      "action_loss_avg": 0.03380845608189702,
      "epoch": 1.8718719482421875e-05,
      "step": 4907
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 4907,
      "training_step_time": 10.206632137298584
    },
    {
      "epoch": 1.8718719482421875e-05,
      "model_forward_time": 3.1357648372650146,
      "step": 4907
    },
    {
      "epoch": 1.8718719482421875e-05,
      "step": 4907,
      "training_step_time": 10.29838490486145
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 3.172102928161621,
      "step": 4908
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 4908,
      "training_step_time": 9.948302745819092
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 3.1339776515960693,
      "step": 4908
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 4908,
      "training_step_time": 9.904784917831421
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 3.1363327503204346,
      "step": 4908
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 4908,
      "training_step_time": 9.914483070373535
    },
    {
      "epoch": 1.87225341796875e-05,
      "model_forward_time": 3.1304609775543213,
      "step": 4908
    },
    {
      "epoch": 1.87225341796875e-05,
      "step": 4908,
      "training_step_time": 10.299269914627075
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 3.173398733139038,
      "step": 4909
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 4909,
      "training_step_time": 9.95100998878479
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 3.1347899436950684,
      "step": 4909
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 4909,
      "training_step_time": 9.915258884429932
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 3.1379387378692627,
      "step": 4909
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 4909,
      "training_step_time": 9.916633367538452
    },
    {
      "epoch": 1.8726348876953124e-05,
      "model_forward_time": 3.132535696029663,
      "step": 4909
    },
    {
      "epoch": 1.8726348876953124e-05,
      "step": 4909,
      "training_step_time": 10.311620950698853
    },
    {
      "epoch": 1.873016357421875e-05,
      "grad_norm": 0.3844752013683319,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 4910
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 3.4646477699279785,
      "step": 4910
    },
    {
      "dynamics_loss_avg": 0.03956555239856243,
      "epoch": 1.873016357421875e-05,
      "step": 4910
    },
    {
      "action_loss_avg": 0.015284256311133504,
      "epoch": 1.873016357421875e-05,
      "step": 4910
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 4910,
      "training_step_time": 10.252875328063965
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 3.1352005004882812,
      "step": 4910
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 4910,
      "training_step_time": 9.913471221923828
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 3.1361069679260254,
      "step": 4910
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 4910,
      "training_step_time": 9.911499738693237
    },
    {
      "epoch": 1.873016357421875e-05,
      "model_forward_time": 3.1318111419677734,
      "step": 4910
    },
    {
      "epoch": 1.873016357421875e-05,
      "step": 4910,
      "training_step_time": 10.228898286819458
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 3.166384220123291,
      "step": 4911
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 4911,
      "training_step_time": 9.942999601364136
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 3.1346065998077393,
      "step": 4911
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 4911,
      "training_step_time": 10.189441680908203
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 3.1331729888916016,
      "step": 4911
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 4911,
      "training_step_time": 9.90895414352417
    },
    {
      "epoch": 1.8733978271484373e-05,
      "model_forward_time": 3.1305694580078125,
      "step": 4911
    },
    {
      "epoch": 1.8733978271484373e-05,
      "step": 4911,
      "training_step_time": 10.299597024917603
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 3.1558661460876465,
      "step": 4912
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 4912,
      "training_step_time": 9.93515920639038
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 3.134915590286255,
      "step": 4912
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 4912,
      "training_step_time": 9.909888982772827
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 3.422964334487915,
      "step": 4912
    },
    {
      "dynamics_loss_avg": 0.07243281910195946,
      "epoch": 1.873779296875e-05,
      "step": 4912
    },
    {
      "action_loss_avg": 0.02645029304549098,
      "epoch": 1.873779296875e-05,
      "step": 4912
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 4912,
      "training_step_time": 10.200218677520752
    },
    {
      "epoch": 1.873779296875e-05,
      "model_forward_time": 3.126272678375244,
      "step": 4912
    },
    {
      "epoch": 1.873779296875e-05,
      "step": 4912,
      "training_step_time": 10.263291835784912
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 3.1665515899658203,
      "step": 4913
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 4913,
      "training_step_time": 9.94081974029541
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 3.1290054321289062,
      "step": 4913
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 4913,
      "training_step_time": 9.899205207824707
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 3.127603769302368,
      "step": 4913
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 4913,
      "training_step_time": 9.901435613632202
    },
    {
      "epoch": 1.8741607666015626e-05,
      "model_forward_time": 3.420990228652954,
      "step": 4913
    },
    {
      "epoch": 1.8741607666015626e-05,
      "step": 4913,
      "training_step_time": 10.30136489868164
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 3.158681869506836,
      "step": 4914
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 4914,
      "training_step_time": 9.939923524856567
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 3.132098436355591,
      "step": 4914
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 4914,
      "training_step_time": 9.91856074333191
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 3.1315340995788574,
      "step": 4914
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 4914,
      "training_step_time": 9.905633926391602
    },
    {
      "epoch": 1.874542236328125e-05,
      "model_forward_time": 3.1321277618408203,
      "step": 4914
    },
    {
      "epoch": 1.874542236328125e-05,
      "step": 4914,
      "training_step_time": 10.241182088851929
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 3.4503045082092285,
      "step": 4915
    },
    {
      "dynamics_loss_avg": 0.061537666246294975,
      "epoch": 1.8749237060546875e-05,
      "step": 4915
    },
    {
      "action_loss_avg": 0.0285511028021574,
      "epoch": 1.8749237060546875e-05,
      "step": 4915
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 4915,
      "training_step_time": 10.227449655532837
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 3.1382036209106445,
      "step": 4915
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 4915,
      "training_step_time": 9.912944078445435
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 3.1355552673339844,
      "step": 4915
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 4915,
      "training_step_time": 9.926065683364868
    },
    {
      "epoch": 1.8749237060546875e-05,
      "model_forward_time": 3.1391923427581787,
      "step": 4915
    },
    {
      "epoch": 1.8749237060546875e-05,
      "step": 4915,
      "training_step_time": 10.027719259262085
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 3.1681325435638428,
      "step": 4916
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 4916,
      "training_step_time": 9.946532487869263
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 3.413729190826416,
      "step": 4916
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 4916,
      "training_step_time": 10.192821502685547
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 3.132517099380493,
      "step": 4916
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 4916,
      "training_step_time": 9.9092857837677
    },
    {
      "epoch": 1.87530517578125e-05,
      "model_forward_time": 3.1388461589813232,
      "step": 4916
    },
    {
      "epoch": 1.87530517578125e-05,
      "step": 4916,
      "training_step_time": 10.279759407043457
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 3.176062822341919,
      "step": 4917
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 4917,
      "training_step_time": 9.96004319190979
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 3.138643741607666,
      "step": 4917
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 4917,
      "training_step_time": 9.916174173355103
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 3.1365742683410645,
      "step": 4917
    },
    {
      "dynamics_loss_avg": 0.06380345970392227,
      "epoch": 1.8756866455078124e-05,
      "step": 4917
    },
    {
      "action_loss_avg": 0.027177490945905446,
      "epoch": 1.8756866455078124e-05,
      "step": 4917
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 4917,
      "training_step_time": 10.20028305053711
    },
    {
      "epoch": 1.8756866455078124e-05,
      "model_forward_time": 3.1370720863342285,
      "step": 4917
    },
    {
      "epoch": 1.8756866455078124e-05,
      "step": 4917,
      "training_step_time": 10.025983095169067
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 3.161161184310913,
      "step": 4918
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 4918,
      "training_step_time": 9.93946886062622
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 3.1345996856689453,
      "step": 4918
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 4918,
      "training_step_time": 9.916626691818237
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 3.14290714263916,
      "step": 4918
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 4918,
      "training_step_time": 9.918581247329712
    },
    {
      "epoch": 1.876068115234375e-05,
      "model_forward_time": 3.430595636367798,
      "step": 4918
    },
    {
      "epoch": 1.876068115234375e-05,
      "step": 4918,
      "training_step_time": 10.304563999176025
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 3.1846113204956055,
      "step": 4919
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 4919,
      "training_step_time": 9.965789079666138
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 3.1364879608154297,
      "step": 4919
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 4919,
      "training_step_time": 9.920831680297852
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 3.1380186080932617,
      "step": 4919
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 4919,
      "training_step_time": 9.913438320159912
    },
    {
      "epoch": 1.8764495849609373e-05,
      "model_forward_time": 3.1391890048980713,
      "step": 4919
    },
    {
      "epoch": 1.8764495849609373e-05,
      "step": 4919,
      "training_step_time": 10.27989649772644
    },
    {
      "epoch": 1.8768310546875e-05,
      "grad_norm": 0.4655334949493408,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 4920
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 3.166687250137329,
      "step": 4920
    },
    {
      "dynamics_loss_avg": 0.059925630874931815,
      "epoch": 1.8768310546875e-05,
      "step": 4920
    },
    {
      "action_loss_avg": 0.026085302606225012,
      "epoch": 1.8768310546875e-05,
      "step": 4920
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 4920,
      "training_step_time": 10.243237733840942
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 3.1332671642303467,
      "step": 4920
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 4920,
      "training_step_time": 9.910275220870972
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 3.1356940269470215,
      "step": 4920
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 4920,
      "training_step_time": 9.914992809295654
    },
    {
      "epoch": 1.8768310546875e-05,
      "model_forward_time": 3.136312484741211,
      "step": 4920
    },
    {
      "epoch": 1.8768310546875e-05,
      "step": 4920,
      "training_step_time": 10.02939510345459
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 3.1813466548919678,
      "step": 4921
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 4921,
      "training_step_time": 9.970243692398071
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 3.436380386352539,
      "step": 4921
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 4921,
      "training_step_time": 10.218535900115967
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 3.154092311859131,
      "step": 4921
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 4921,
      "training_step_time": 9.931791305541992
    },
    {
      "epoch": 1.8772125244140626e-05,
      "model_forward_time": 3.1476235389709473,
      "step": 4921
    },
    {
      "epoch": 1.8772125244140626e-05,
      "step": 4921,
      "training_step_time": 10.021105527877808
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 3.182992458343506,
      "step": 4922
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 4922,
      "training_step_time": 9.958499908447266
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 3.1436057090759277,
      "step": 4922
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 4922,
      "training_step_time": 9.91296100616455
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 3.426459550857544,
      "step": 4922
    },
    {
      "dynamics_loss_avg": 0.07208109349012375,
      "epoch": 1.877593994140625e-05,
      "step": 4922
    },
    {
      "action_loss_avg": 0.04085659440606833,
      "epoch": 1.877593994140625e-05,
      "step": 4922
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 4922,
      "training_step_time": 10.201289892196655
    },
    {
      "epoch": 1.877593994140625e-05,
      "model_forward_time": 2.433929681777954,
      "step": 4922
    },
    {
      "epoch": 1.877593994140625e-05,
      "step": 4922,
      "training_step_time": 10.27460527420044
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 3.1587331295013428,
      "step": 4923
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 4923,
      "training_step_time": 9.936411142349243
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 3.133673906326294,
      "step": 4923
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 4923,
      "training_step_time": 9.903745889663696
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 3.1406478881835938,
      "step": 4923
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 4923,
      "training_step_time": 9.9146728515625
    },
    {
      "epoch": 1.8779754638671875e-05,
      "model_forward_time": 3.4013211727142334,
      "step": 4923
    },
    {
      "epoch": 1.8779754638671875e-05,
      "step": 4923,
      "training_step_time": 10.295525074005127
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 3.171027421951294,
      "step": 4924
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 4924,
      "training_step_time": 9.95088529586792
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 3.1333200931549072,
      "step": 4924
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 4924,
      "training_step_time": 9.902654886245728
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 3.130735158920288,
      "step": 4924
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 4924,
      "training_step_time": 9.89518690109253
    },
    {
      "epoch": 1.87835693359375e-05,
      "model_forward_time": 3.1384832859039307,
      "step": 4924
    },
    {
      "epoch": 1.87835693359375e-05,
      "step": 4924,
      "training_step_time": 10.2677481174469
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 3.1882147789001465,
      "step": 4925
    },
    {
      "dynamics_loss_avg": 0.06220881715416908,
      "epoch": 1.8787384033203124e-05,
      "step": 4925
    },
    {
      "action_loss_avg": 0.02668529599905014,
      "epoch": 1.8787384033203124e-05,
      "step": 4925
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 4925,
      "training_step_time": 10.242481470108032
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 3.142894983291626,
      "step": 4925
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 4925,
      "training_step_time": 9.919013261795044
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 3.134385585784912,
      "step": 4925
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 4925,
      "training_step_time": 9.906198978424072
    },
    {
      "epoch": 1.8787384033203124e-05,
      "model_forward_time": 3.13864803314209,
      "step": 4925
    },
    {
      "epoch": 1.8787384033203124e-05,
      "step": 4925,
      "training_step_time": 10.01217532157898
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 3.1586663722991943,
      "step": 4926
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 4926,
      "training_step_time": 9.93448281288147
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 3.425769090652466,
      "step": 4926
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 4926,
      "training_step_time": 10.20051884651184
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 3.1390507221221924,
      "step": 4926
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 4926,
      "training_step_time": 9.915164232254028
    },
    {
      "epoch": 1.879119873046875e-05,
      "model_forward_time": 3.1380534172058105,
      "step": 4926
    },
    {
      "epoch": 1.879119873046875e-05,
      "step": 4926,
      "training_step_time": 10.285858631134033
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 3.180590867996216,
      "step": 4927
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 4927,
      "training_step_time": 9.959032773971558
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 3.135833263397217,
      "step": 4927
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 4927,
      "training_step_time": 9.911890029907227
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 3.388824462890625,
      "step": 4927
    },
    {
      "dynamics_loss_avg": 0.05232021138072014,
      "epoch": 1.8795013427734373e-05,
      "step": 4927
    },
    {
      "action_loss_avg": 0.022934066597372293,
      "epoch": 1.8795013427734373e-05,
      "step": 4927
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 4927,
      "training_step_time": 10.167484760284424
    },
    {
      "epoch": 1.8795013427734373e-05,
      "model_forward_time": 3.13783860206604,
      "step": 4927
    },
    {
      "epoch": 1.8795013427734373e-05,
      "step": 4927,
      "training_step_time": 10.287466287612915
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 3.1836726665496826,
      "step": 4928
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 4928,
      "training_step_time": 9.9550621509552
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 3.1343791484832764,
      "step": 4928
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 4928,
      "training_step_time": 9.911650657653809
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 3.136049270629883,
      "step": 4928
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 4928,
      "training_step_time": 9.909535646438599
    },
    {
      "epoch": 1.8798828125e-05,
      "model_forward_time": 3.4127814769744873,
      "step": 4928
    },
    {
      "epoch": 1.8798828125e-05,
      "step": 4928,
      "training_step_time": 10.283210515975952
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 3.173164129257202,
      "step": 4929
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 4929,
      "training_step_time": 9.953563213348389
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 3.1347591876983643,
      "step": 4929
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 4929,
      "training_step_time": 9.914786577224731
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 3.145890474319458,
      "step": 4929
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 4929,
      "training_step_time": 9.940040111541748
    },
    {
      "epoch": 1.8802642822265626e-05,
      "model_forward_time": 3.16064190864563,
      "step": 4929
    },
    {
      "epoch": 1.8802642822265626e-05,
      "step": 4929,
      "training_step_time": 10.32195234298706
    },
    {
      "epoch": 1.880645751953125e-05,
      "grad_norm": 0.6048769950866699,
      "learning_rate": 1e-05,
      "loss": 0.0912,
      "step": 4930
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 3.453185796737671,
      "step": 4930
    },
    {
      "dynamics_loss_avg": 0.06203442178666592,
      "epoch": 1.880645751953125e-05,
      "step": 4930
    },
    {
      "action_loss_avg": 0.028001463413238524,
      "epoch": 1.880645751953125e-05,
      "step": 4930
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 4930,
      "training_step_time": 10.23555326461792
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 3.1416797637939453,
      "step": 4930
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 4930,
      "training_step_time": 9.921047687530518
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 3.1369571685791016,
      "step": 4930
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 4930,
      "training_step_time": 9.914231538772583
    },
    {
      "epoch": 1.880645751953125e-05,
      "model_forward_time": 3.1348483562469482,
      "step": 4930
    },
    {
      "epoch": 1.880645751953125e-05,
      "step": 4930,
      "training_step_time": 10.006471633911133
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 3.1495323181152344,
      "step": 4931
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 4931,
      "training_step_time": 9.925621271133423
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 3.418984889984131,
      "step": 4931
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 4931,
      "training_step_time": 10.187739372253418
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 3.1351706981658936,
      "step": 4931
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 4931,
      "training_step_time": 9.905769109725952
    },
    {
      "epoch": 1.8810272216796875e-05,
      "model_forward_time": 2.4253692626953125,
      "step": 4931
    },
    {
      "epoch": 1.8810272216796875e-05,
      "step": 4931,
      "training_step_time": 10.007952213287354
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 3.1870555877685547,
      "step": 4932
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 4932,
      "training_step_time": 9.95500135421753
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 3.1343703269958496,
      "step": 4932
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 4932,
      "training_step_time": 9.907199382781982
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 3.4176294803619385,
      "step": 4932
    },
    {
      "dynamics_loss_avg": 0.06701614521443844,
      "epoch": 1.88140869140625e-05,
      "step": 4932
    },
    {
      "action_loss_avg": 0.029059047508053482,
      "epoch": 1.88140869140625e-05,
      "step": 4932
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 4932,
      "training_step_time": 10.190419435501099
    },
    {
      "epoch": 1.88140869140625e-05,
      "model_forward_time": 3.135432004928589,
      "step": 4932
    },
    {
      "epoch": 1.88140869140625e-05,
      "step": 4932,
      "training_step_time": 10.29812240600586
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 3.1655433177948,
      "step": 4933
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 4933,
      "training_step_time": 9.93859577178955
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 3.1391642093658447,
      "step": 4933
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 4933,
      "training_step_time": 9.913947105407715
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 3.1417593955993652,
      "step": 4933
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 4933,
      "training_step_time": 9.924524545669556
    },
    {
      "epoch": 1.8817901611328124e-05,
      "model_forward_time": 3.138476610183716,
      "step": 4933
    },
    {
      "epoch": 1.8817901611328124e-05,
      "step": 4933,
      "training_step_time": 10.282548904418945
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 3.45944881439209,
      "step": 4934
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 4934,
      "training_step_time": 10.238433361053467
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 3.141573667526245,
      "step": 4934
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 4934,
      "training_step_time": 9.924196481704712
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 3.138216495513916,
      "step": 4934
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 4934,
      "training_step_time": 9.915206909179688
    },
    {
      "epoch": 1.882171630859375e-05,
      "model_forward_time": 3.140338182449341,
      "step": 4934
    },
    {
      "epoch": 1.882171630859375e-05,
      "step": 4934,
      "training_step_time": 10.314619779586792
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 3.4737298488616943,
      "step": 4935
    },
    {
      "dynamics_loss_avg": 0.05473592486232519,
      "epoch": 1.8825531005859374e-05,
      "step": 4935
    },
    {
      "action_loss_avg": 0.024596709897741674,
      "epoch": 1.8825531005859374e-05,
      "step": 4935
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 4935,
      "training_step_time": 10.257156372070312
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 3.1473941802978516,
      "step": 4935
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 4935,
      "training_step_time": 9.929441928863525
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 3.13862681388855,
      "step": 4935
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 4935,
      "training_step_time": 9.919013261795044
    },
    {
      "epoch": 1.8825531005859374e-05,
      "model_forward_time": 3.1395788192749023,
      "step": 4935
    },
    {
      "epoch": 1.8825531005859374e-05,
      "step": 4935,
      "training_step_time": 10.026434659957886
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 3.1674704551696777,
      "step": 4936
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 4936,
      "training_step_time": 9.940444231033325
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 3.1357522010803223,
      "step": 4936
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 4936,
      "training_step_time": 10.184778928756714
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 3.140003204345703,
      "step": 4936
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 4936,
      "training_step_time": 9.924686670303345
    },
    {
      "epoch": 1.8829345703125e-05,
      "model_forward_time": 3.144312858581543,
      "step": 4936
    },
    {
      "epoch": 1.8829345703125e-05,
      "step": 4936,
      "training_step_time": 10.28260612487793
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 3.1727118492126465,
      "step": 4937
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 4937,
      "training_step_time": 9.949812889099121
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 3.1368002891540527,
      "step": 4937
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 4937,
      "training_step_time": 9.91526174545288
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 3.142228603363037,
      "step": 4937
    },
    {
      "dynamics_loss_avg": 0.06579474844038487,
      "epoch": 1.8833160400390626e-05,
      "step": 4937
    },
    {
      "action_loss_avg": 0.03122796886600554,
      "epoch": 1.8833160400390626e-05,
      "step": 4937
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 4937,
      "training_step_time": 10.202468395233154
    },
    {
      "epoch": 1.8833160400390626e-05,
      "model_forward_time": 3.143475294113159,
      "step": 4937
    },
    {
      "epoch": 1.8833160400390626e-05,
      "step": 4937,
      "training_step_time": 10.270660161972046
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 3.197648763656616,
      "step": 4938
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 4938,
      "training_step_time": 9.979154586791992
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 3.133835792541504,
      "step": 4938
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 4938,
      "training_step_time": 9.912098407745361
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 3.137507438659668,
      "step": 4938
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 4938,
      "training_step_time": 9.912474393844604
    },
    {
      "epoch": 1.883697509765625e-05,
      "model_forward_time": 3.428222894668579,
      "step": 4938
    },
    {
      "epoch": 1.883697509765625e-05,
      "step": 4938,
      "training_step_time": 10.301493167877197
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 3.172621250152588,
      "step": 4939
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 4939,
      "training_step_time": 9.950970649719238
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 3.1293790340423584,
      "step": 4939
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 4939,
      "training_step_time": 9.90901494026184
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 3.139017343521118,
      "step": 4939
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 4939,
      "training_step_time": 9.922568798065186
    },
    {
      "epoch": 1.8840789794921875e-05,
      "model_forward_time": 3.1372318267822266,
      "step": 4939
    },
    {
      "epoch": 1.8840789794921875e-05,
      "step": 4939,
      "training_step_time": 10.29472541809082
    },
    {
      "epoch": 1.88446044921875e-05,
      "grad_norm": 0.4163573682308197,
      "learning_rate": 1e-05,
      "loss": 0.0918,
      "step": 4940
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 3.169050693511963,
      "step": 4940
    },
    {
      "dynamics_loss_avg": 0.0653709415346384,
      "epoch": 1.88446044921875e-05,
      "step": 4940
    },
    {
      "action_loss_avg": 0.029870682582259177,
      "epoch": 1.88446044921875e-05,
      "step": 4940
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 4940,
      "training_step_time": 10.229557991027832
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 3.1407105922698975,
      "step": 4940
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 4940,
      "training_step_time": 9.921159505844116
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 3.1401212215423584,
      "step": 4940
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 4940,
      "training_step_time": 9.91948390007019
    },
    {
      "epoch": 1.88446044921875e-05,
      "model_forward_time": 3.132847547531128,
      "step": 4940
    },
    {
      "epoch": 1.88446044921875e-05,
      "step": 4940,
      "training_step_time": 10.008936643600464
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 3.1624205112457275,
      "step": 4941
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 4941,
      "training_step_time": 9.94181203842163
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 3.3908584117889404,
      "step": 4941
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 4941,
      "training_step_time": 10.166623592376709
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 3.136150598526001,
      "step": 4941
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 4941,
      "training_step_time": 9.913193702697754
    },
    {
      "epoch": 1.8848419189453124e-05,
      "model_forward_time": 3.1340348720550537,
      "step": 4941
    },
    {
      "epoch": 1.8848419189453124e-05,
      "step": 4941,
      "training_step_time": 10.263980388641357
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 3.1811745166778564,
      "step": 4942
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 4942,
      "training_step_time": 9.956291198730469
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 3.142104387283325,
      "step": 4942
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 4942,
      "training_step_time": 9.921348333358765
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 3.4155893325805664,
      "step": 4942
    },
    {
      "dynamics_loss_avg": 0.051340189017355445,
      "epoch": 1.885223388671875e-05,
      "step": 4942
    },
    {
      "action_loss_avg": 0.020629419200122356,
      "epoch": 1.885223388671875e-05,
      "step": 4942
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 4942,
      "training_step_time": 10.194111108779907
    },
    {
      "epoch": 1.885223388671875e-05,
      "model_forward_time": 3.1384479999542236,
      "step": 4942
    },
    {
      "epoch": 1.885223388671875e-05,
      "step": 4942,
      "training_step_time": 10.02095890045166
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 3.1869919300079346,
      "step": 4943
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 4943,
      "training_step_time": 9.971837282180786
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 3.1461093425750732,
      "step": 4943
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 4943,
      "training_step_time": 9.930589199066162
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 3.141456365585327,
      "step": 4943
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 4943,
      "training_step_time": 9.931852102279663
    },
    {
      "epoch": 1.8856048583984374e-05,
      "model_forward_time": 3.420572519302368,
      "step": 4943
    },
    {
      "epoch": 1.8856048583984374e-05,
      "step": 4943,
      "training_step_time": 10.301982164382935
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 3.1928486824035645,
      "step": 4944
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 4944,
      "training_step_time": 9.966865301132202
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 3.1565420627593994,
      "step": 4944
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 4944,
      "training_step_time": 9.932220458984375
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 3.156656265258789,
      "step": 4944
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 4944,
      "training_step_time": 9.93671441078186
    },
    {
      "epoch": 1.885986328125e-05,
      "model_forward_time": 3.15645170211792,
      "step": 4944
    },
    {
      "epoch": 1.885986328125e-05,
      "step": 4944,
      "training_step_time": 10.283045530319214
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 3.4625790119171143,
      "step": 4945
    },
    {
      "dynamics_loss_avg": 0.056766111962497236,
      "epoch": 1.8863677978515626e-05,
      "step": 4945
    },
    {
      "action_loss_avg": 0.02970692222006619,
      "epoch": 1.8863677978515626e-05,
      "step": 4945
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 4945,
      "training_step_time": 10.241546154022217
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 3.153350591659546,
      "step": 4945
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 4945,
      "training_step_time": 9.922682523727417
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 3.1475651264190674,
      "step": 4945
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 4945,
      "training_step_time": 9.915957927703857
    },
    {
      "epoch": 1.8863677978515626e-05,
      "model_forward_time": 3.149441719055176,
      "step": 4945
    },
    {
      "epoch": 1.8863677978515626e-05,
      "step": 4945,
      "training_step_time": 10.293132305145264
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 3.1893160343170166,
      "step": 4946
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 4946,
      "training_step_time": 9.958024024963379
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 3.141214609146118,
      "step": 4946
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 4946,
      "training_step_time": 10.187377214431763
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 3.142212390899658,
      "step": 4946
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 4946,
      "training_step_time": 9.90587592124939
    },
    {
      "epoch": 1.886749267578125e-05,
      "model_forward_time": 3.1460957527160645,
      "step": 4946
    },
    {
      "epoch": 1.886749267578125e-05,
      "step": 4946,
      "training_step_time": 10.015522003173828
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 3.1810944080352783,
      "step": 4947
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 4947,
      "training_step_time": 9.945934772491455
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 3.141556739807129,
      "step": 4947
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 4947,
      "training_step_time": 9.906839847564697
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 3.3902766704559326,
      "step": 4947
    },
    {
      "dynamics_loss_avg": 0.04519776590168476,
      "epoch": 1.8871307373046875e-05,
      "step": 4947
    },
    {
      "action_loss_avg": 0.021192850498482584,
      "epoch": 1.8871307373046875e-05,
      "step": 4947
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 4947,
      "training_step_time": 10.165152311325073
    },
    {
      "epoch": 1.8871307373046875e-05,
      "model_forward_time": 3.1354522705078125,
      "step": 4947
    },
    {
      "epoch": 1.8871307373046875e-05,
      "step": 4947,
      "training_step_time": 10.284307718276978
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 3.174518585205078,
      "step": 4948
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 4948,
      "training_step_time": 9.948997259140015
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 3.135232448577881,
      "step": 4948
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 4948,
      "training_step_time": 9.905099153518677
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 3.1482412815093994,
      "step": 4948
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 4948,
      "training_step_time": 9.919170379638672
    },
    {
      "epoch": 1.88751220703125e-05,
      "model_forward_time": 3.145211935043335,
      "step": 4948
    },
    {
      "epoch": 1.88751220703125e-05,
      "step": 4948,
      "training_step_time": 10.297565221786499
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 3.184307813644409,
      "step": 4949
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 4949,
      "training_step_time": 9.959366798400879
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 3.1338415145874023,
      "step": 4949
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 4949,
      "training_step_time": 9.906551837921143
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 3.1390373706817627,
      "step": 4949
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 4949,
      "training_step_time": 9.907589197158813
    },
    {
      "epoch": 1.8878936767578124e-05,
      "model_forward_time": 3.1360745429992676,
      "step": 4949
    },
    {
      "epoch": 1.8878936767578124e-05,
      "step": 4949,
      "training_step_time": 10.004440307617188
    },
    {
      "epoch": 1.888275146484375e-05,
      "grad_norm": 0.3945888876914978,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 4950
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 3.467212200164795,
      "step": 4950
    },
    {
      "dynamics_loss_avg": 0.06800358183681965,
      "epoch": 1.888275146484375e-05,
      "step": 4950
    },
    {
      "action_loss_avg": 0.027991453371942043,
      "epoch": 1.888275146484375e-05,
      "step": 4950
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 4950,
      "training_step_time": 10.256747961044312
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 3.1356194019317627,
      "step": 4950
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 4950,
      "training_step_time": 9.91284441947937
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 3.1403470039367676,
      "step": 4950
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 4950,
      "training_step_time": 9.918359994888306
    },
    {
      "epoch": 1.888275146484375e-05,
      "model_forward_time": 3.13962721824646,
      "step": 4950
    },
    {
      "epoch": 1.888275146484375e-05,
      "step": 4950,
      "training_step_time": 10.306273937225342
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 3.1572659015655518,
      "step": 4951
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 4951,
      "training_step_time": 9.931868553161621
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 3.1450037956237793,
      "step": 4951
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 4951,
      "training_step_time": 10.205585241317749
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 3.1415867805480957,
      "step": 4951
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 4951,
      "training_step_time": 9.916404724121094
    },
    {
      "epoch": 1.8886566162109374e-05,
      "model_forward_time": 2.43408203125,
      "step": 4951
    },
    {
      "epoch": 1.8886566162109374e-05,
      "step": 4951,
      "training_step_time": 10.298762321472168
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 3.1830525398254395,
      "step": 4952
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 4952,
      "training_step_time": 9.953493595123291
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 3.1368398666381836,
      "step": 4952
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 4952,
      "training_step_time": 9.907342672348022
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 3.4110872745513916,
      "step": 4952
    },
    {
      "dynamics_loss_avg": 0.061683239787817,
      "epoch": 1.8890380859375e-05,
      "step": 4952
    },
    {
      "action_loss_avg": 0.03214087309315801,
      "epoch": 1.8890380859375e-05,
      "step": 4952
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 4952,
      "training_step_time": 10.18314266204834
    },
    {
      "epoch": 1.8890380859375e-05,
      "model_forward_time": 3.1346476078033447,
      "step": 4952
    },
    {
      "epoch": 1.8890380859375e-05,
      "step": 4952,
      "training_step_time": 10.300288438796997
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 3.177358627319336,
      "step": 4953
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 4953,
      "training_step_time": 9.940624237060547
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 3.1351847648620605,
      "step": 4953
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 4953,
      "training_step_time": 9.902183294296265
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 3.1322848796844482,
      "step": 4953
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 4953,
      "training_step_time": 9.904930830001831
    },
    {
      "epoch": 1.8894195556640626e-05,
      "model_forward_time": 3.134068250656128,
      "step": 4953
    },
    {
      "epoch": 1.8894195556640626e-05,
      "step": 4953,
      "training_step_time": 10.253878831863403
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 3.43768572807312,
      "step": 4954
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 4954,
      "training_step_time": 10.20693826675415
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 3.134082794189453,
      "step": 4954
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 4954,
      "training_step_time": 9.907394170761108
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 3.1378562450408936,
      "step": 4954
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 4954,
      "training_step_time": 9.914572477340698
    },
    {
      "epoch": 1.889801025390625e-05,
      "model_forward_time": 3.1402432918548584,
      "step": 4954
    },
    {
      "epoch": 1.889801025390625e-05,
      "step": 4954,
      "training_step_time": 10.031081676483154
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 3.4285101890563965,
      "step": 4955
    },
    {
      "dynamics_loss_avg": 0.06602325402200222,
      "epoch": 1.8901824951171875e-05,
      "step": 4955
    },
    {
      "action_loss_avg": 0.03934728037565947,
      "epoch": 1.8901824951171875e-05,
      "step": 4955
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 4955,
      "training_step_time": 10.204167366027832
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 3.141873359680176,
      "step": 4955
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 4955,
      "training_step_time": 9.916774034500122
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 3.1340603828430176,
      "step": 4955
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 4955,
      "training_step_time": 9.908788681030273
    },
    {
      "epoch": 1.8901824951171875e-05,
      "model_forward_time": 3.141288995742798,
      "step": 4955
    },
    {
      "epoch": 1.8901824951171875e-05,
      "step": 4955,
      "training_step_time": 10.299682140350342
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 3.165817975997925,
      "step": 4956
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 4956,
      "training_step_time": 9.935682535171509
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 3.145315408706665,
      "step": 4956
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 4956,
      "training_step_time": 10.19043254852295
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 3.1331052780151367,
      "step": 4956
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 4956,
      "training_step_time": 9.905609846115112
    },
    {
      "epoch": 1.89056396484375e-05,
      "model_forward_time": 3.1371521949768066,
      "step": 4956
    },
    {
      "epoch": 1.89056396484375e-05,
      "step": 4956,
      "training_step_time": 10.014785766601562
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 3.1848256587982178,
      "step": 4957
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 4957,
      "training_step_time": 9.953965902328491
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 3.133007049560547,
      "step": 4957
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 4957,
      "training_step_time": 9.89814567565918
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 3.1339943408966064,
      "step": 4957
    },
    {
      "dynamics_loss_avg": 0.05779799669981003,
      "epoch": 1.8909454345703124e-05,
      "step": 4957
    },
    {
      "action_loss_avg": 0.025984920840710402,
      "epoch": 1.8909454345703124e-05,
      "step": 4957
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 4957,
      "training_step_time": 10.188210010528564
    },
    {
      "epoch": 1.8909454345703124e-05,
      "model_forward_time": 3.132678508758545,
      "step": 4957
    },
    {
      "epoch": 1.8909454345703124e-05,
      "step": 4957,
      "training_step_time": 10.290730476379395
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 3.1903445720672607,
      "step": 4958
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 4958,
      "training_step_time": 9.959404230117798
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 3.132880687713623,
      "step": 4958
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 4958,
      "training_step_time": 9.905939102172852
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 3.132708787918091,
      "step": 4958
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 4958,
      "training_step_time": 9.907201528549194
    },
    {
      "epoch": 1.891326904296875e-05,
      "model_forward_time": 3.4203121662139893,
      "step": 4958
    },
    {
      "epoch": 1.891326904296875e-05,
      "step": 4958,
      "training_step_time": 10.294822454452515
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 3.200939893722534,
      "step": 4959
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 4959,
      "training_step_time": 9.978656768798828
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 3.1373825073242188,
      "step": 4959
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 4959,
      "training_step_time": 9.913888454437256
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 3.1358859539031982,
      "step": 4959
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 4959,
      "training_step_time": 9.907549619674683
    },
    {
      "epoch": 1.8917083740234374e-05,
      "model_forward_time": 3.1303253173828125,
      "step": 4959
    },
    {
      "epoch": 1.8917083740234374e-05,
      "step": 4959,
      "training_step_time": 10.023235082626343
    },
    {
      "epoch": 1.89208984375e-05,
      "grad_norm": 0.732893705368042,
      "learning_rate": 1e-05,
      "loss": 0.09,
      "step": 4960
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 3.1714751720428467,
      "step": 4960
    },
    {
      "dynamics_loss_avg": 0.05555259007960558,
      "epoch": 1.89208984375e-05,
      "step": 4960
    },
    {
      "action_loss_avg": 0.0276587575674057,
      "epoch": 1.89208984375e-05,
      "step": 4960
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 4960,
      "training_step_time": 10.212413549423218
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 3.1350977420806885,
      "step": 4960
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 4960,
      "training_step_time": 9.90848183631897
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 3.1410176753997803,
      "step": 4960
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 4960,
      "training_step_time": 9.91706657409668
    },
    {
      "epoch": 1.89208984375e-05,
      "model_forward_time": 3.1409072875976562,
      "step": 4960
    },
    {
      "epoch": 1.89208984375e-05,
      "step": 4960,
      "training_step_time": 10.315035104751587
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 3.167498826980591,
      "step": 4961
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 4961,
      "training_step_time": 9.939940690994263
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 3.389098882675171,
      "step": 4961
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 4961,
      "training_step_time": 10.164143800735474
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 3.1486732959747314,
      "step": 4961
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 4961,
      "training_step_time": 9.926120519638062
    },
    {
      "epoch": 1.8924713134765626e-05,
      "model_forward_time": 3.1373918056488037,
      "step": 4961
    },
    {
      "epoch": 1.8924713134765626e-05,
      "step": 4961,
      "training_step_time": 10.024531602859497
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 3.1622748374938965,
      "step": 4962
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 4962,
      "training_step_time": 9.937400579452515
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 3.1397721767425537,
      "step": 4962
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 4962,
      "training_step_time": 9.908010005950928
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 3.440463066101074,
      "step": 4962
    },
    {
      "dynamics_loss_avg": 0.05285576768219471,
      "epoch": 1.892852783203125e-05,
      "step": 4962
    },
    {
      "action_loss_avg": 0.02085996512323618,
      "epoch": 1.892852783203125e-05,
      "step": 4962
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 4962,
      "training_step_time": 10.210232496261597
    },
    {
      "epoch": 1.892852783203125e-05,
      "model_forward_time": 3.135838031768799,
      "step": 4962
    },
    {
      "epoch": 1.892852783203125e-05,
      "step": 4962,
      "training_step_time": 10.289804458618164
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 3.18194580078125,
      "step": 4963
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 4963,
      "training_step_time": 9.953537464141846
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 3.1375479698181152,
      "step": 4963
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 4963,
      "training_step_time": 9.914562940597534
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 3.1345396041870117,
      "step": 4963
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 4963,
      "training_step_time": 9.906466722488403
    },
    {
      "epoch": 1.8932342529296875e-05,
      "model_forward_time": 3.3922348022460938,
      "step": 4963
    },
    {
      "epoch": 1.8932342529296875e-05,
      "step": 4963,
      "training_step_time": 10.276845216751099
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 3.1966476440429688,
      "step": 4964
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 4964,
      "training_step_time": 9.971627950668335
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 3.144242763519287,
      "step": 4964
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 4964,
      "training_step_time": 9.91931962966919
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 3.136275053024292,
      "step": 4964
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 4964,
      "training_step_time": 9.914450645446777
    },
    {
      "epoch": 1.89361572265625e-05,
      "model_forward_time": 3.1462132930755615,
      "step": 4964
    },
    {
      "epoch": 1.89361572265625e-05,
      "step": 4964,
      "training_step_time": 10.022557973861694
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 3.432724952697754,
      "step": 4965
    },
    {
      "dynamics_loss_avg": 0.05762410964816809,
      "epoch": 1.8939971923828124e-05,
      "step": 4965
    },
    {
      "action_loss_avg": 0.02412653909996152,
      "epoch": 1.8939971923828124e-05,
      "step": 4965
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 4965,
      "training_step_time": 10.208847761154175
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 3.1399378776550293,
      "step": 4965
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 4965,
      "training_step_time": 9.916367769241333
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 3.137544631958008,
      "step": 4965
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 4965,
      "training_step_time": 9.911484003067017
    },
    {
      "epoch": 1.8939971923828124e-05,
      "model_forward_time": 3.139023780822754,
      "step": 4965
    },
    {
      "epoch": 1.8939971923828124e-05,
      "step": 4965,
      "training_step_time": 10.295939683914185
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 3.1826376914978027,
      "step": 4966
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 4966,
      "training_step_time": 9.962833404541016
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 3.139249324798584,
      "step": 4966
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 4966,
      "training_step_time": 10.201594114303589
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 3.136686086654663,
      "step": 4966
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 4966,
      "training_step_time": 9.903991222381592
    },
    {
      "epoch": 1.894378662109375e-05,
      "model_forward_time": 3.1412124633789062,
      "step": 4966
    },
    {
      "epoch": 1.894378662109375e-05,
      "step": 4966,
      "training_step_time": 10.255330562591553
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 3.166797399520874,
      "step": 4967
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 4967,
      "training_step_time": 9.936914205551147
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 3.1306979656219482,
      "step": 4967
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 4967,
      "training_step_time": 9.893841743469238
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 3.4155802726745605,
      "step": 4967
    },
    {
      "dynamics_loss_avg": 0.08162737786769866,
      "epoch": 1.8947601318359374e-05,
      "step": 4967
    },
    {
      "action_loss_avg": 0.03678009510040283,
      "epoch": 1.8947601318359374e-05,
      "step": 4967
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 4967,
      "training_step_time": 10.18215012550354
    },
    {
      "epoch": 1.8947601318359374e-05,
      "model_forward_time": 3.131722927093506,
      "step": 4967
    },
    {
      "epoch": 1.8947601318359374e-05,
      "step": 4967,
      "training_step_time": 10.310017347335815
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 3.1647942066192627,
      "step": 4968
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 4968,
      "training_step_time": 9.930724859237671
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 3.130025625228882,
      "step": 4968
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 4968,
      "training_step_time": 9.895593404769897
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 3.135254144668579,
      "step": 4968
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 4968,
      "training_step_time": 9.903027057647705
    },
    {
      "epoch": 1.8951416015625e-05,
      "model_forward_time": 3.135608434677124,
      "step": 4968
    },
    {
      "epoch": 1.8951416015625e-05,
      "step": 4968,
      "training_step_time": 10.30353593826294
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 3.1740732192993164,
      "step": 4969
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 4969,
      "training_step_time": 9.95357894897461
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 3.1431541442871094,
      "step": 4969
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 4969,
      "training_step_time": 9.918468713760376
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 3.1466872692108154,
      "step": 4969
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 4969,
      "training_step_time": 9.925622701644897
    },
    {
      "epoch": 1.8955230712890626e-05,
      "model_forward_time": 3.144535541534424,
      "step": 4969
    },
    {
      "epoch": 1.8955230712890626e-05,
      "step": 4969,
      "training_step_time": 10.070059061050415
    },
    {
      "epoch": 1.895904541015625e-05,
      "grad_norm": 0.3737166225910187,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 4970
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 3.4674243927001953,
      "step": 4970
    },
    {
      "dynamics_loss_avg": 0.06980232764035463,
      "epoch": 1.895904541015625e-05,
      "step": 4970
    },
    {
      "action_loss_avg": 0.03283786196261644,
      "epoch": 1.895904541015625e-05,
      "step": 4970
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 4970,
      "training_step_time": 10.244967699050903
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 3.1426656246185303,
      "step": 4970
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 4970,
      "training_step_time": 9.92656135559082
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 3.145366668701172,
      "step": 4970
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 4970,
      "training_step_time": 9.91603708267212
    },
    {
      "epoch": 1.895904541015625e-05,
      "model_forward_time": 3.1413257122039795,
      "step": 4970
    },
    {
      "epoch": 1.895904541015625e-05,
      "step": 4970,
      "training_step_time": 10.283371925354004
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 3.1742002964019775,
      "step": 4971
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 4971,
      "training_step_time": 9.94820261001587
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 3.1399643421173096,
      "step": 4971
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 4971,
      "training_step_time": 10.18050479888916
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 3.141636371612549,
      "step": 4971
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 4971,
      "training_step_time": 9.913971185684204
    },
    {
      "epoch": 1.8962860107421875e-05,
      "model_forward_time": 3.1417148113250732,
      "step": 4971
    },
    {
      "epoch": 1.8962860107421875e-05,
      "step": 4971,
      "training_step_time": 10.286479711532593
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 3.1768102645874023,
      "step": 4972
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 4972,
      "training_step_time": 9.9535551071167
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 3.1433804035186768,
      "step": 4972
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 4972,
      "training_step_time": 9.919665336608887
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 3.421602964401245,
      "step": 4972
    },
    {
      "dynamics_loss_avg": 0.06781389228999615,
      "epoch": 1.89666748046875e-05,
      "step": 4972
    },
    {
      "action_loss_avg": 0.02231081323698163,
      "epoch": 1.89666748046875e-05,
      "step": 4972
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 4972,
      "training_step_time": 10.196096181869507
    },
    {
      "epoch": 1.89666748046875e-05,
      "model_forward_time": 3.141853094100952,
      "step": 4972
    },
    {
      "epoch": 1.89666748046875e-05,
      "step": 4972,
      "training_step_time": 10.32052206993103
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 3.1788718700408936,
      "step": 4973
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 4973,
      "training_step_time": 9.959595203399658
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 3.1467320919036865,
      "step": 4973
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 4973,
      "training_step_time": 9.91808533668518
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 3.139397621154785,
      "step": 4973
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 4973,
      "training_step_time": 9.911504745483398
    },
    {
      "epoch": 1.8970489501953124e-05,
      "model_forward_time": 3.4004013538360596,
      "step": 4973
    },
    {
      "epoch": 1.8970489501953124e-05,
      "step": 4973,
      "training_step_time": 10.281012773513794
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 3.1581413745880127,
      "step": 4974
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 4974,
      "training_step_time": 9.93372893333435
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 3.141953229904175,
      "step": 4974
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 4974,
      "training_step_time": 9.917049407958984
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 3.1425671577453613,
      "step": 4974
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 4974,
      "training_step_time": 9.91671085357666
    },
    {
      "epoch": 1.897430419921875e-05,
      "model_forward_time": 3.1446807384490967,
      "step": 4974
    },
    {
      "epoch": 1.897430419921875e-05,
      "step": 4974,
      "training_step_time": 10.307398319244385
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 3.4262235164642334,
      "step": 4975
    },
    {
      "dynamics_loss_avg": 0.04793870765715837,
      "epoch": 1.8978118896484374e-05,
      "step": 4975
    },
    {
      "action_loss_avg": 0.02014186279848218,
      "epoch": 1.8978118896484374e-05,
      "step": 4975
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 4975,
      "training_step_time": 10.201099872589111
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 3.1382224559783936,
      "step": 4975
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 4975,
      "training_step_time": 9.90906286239624
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 3.1363940238952637,
      "step": 4975
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 4975,
      "training_step_time": 9.90872859954834
    },
    {
      "epoch": 1.8978118896484374e-05,
      "model_forward_time": 3.1429975032806396,
      "step": 4975
    },
    {
      "epoch": 1.8978118896484374e-05,
      "step": 4975,
      "training_step_time": 10.028392553329468
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 2.460422992706299,
      "step": 4976
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 4976,
      "training_step_time": 7.490569114685059
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 3.4120683670043945,
      "step": 4976
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 4976,
      "training_step_time": 10.182318925857544
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 3.135899543762207,
      "step": 4976
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 4976,
      "training_step_time": 9.910226345062256
    },
    {
      "epoch": 1.898193359375e-05,
      "model_forward_time": 3.140979290008545,
      "step": 4976
    },
    {
      "epoch": 1.898193359375e-05,
      "step": 4976,
      "training_step_time": 10.017721176147461
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 3.1787896156311035,
      "step": 4977
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 4977,
      "training_step_time": 9.959415197372437
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 3.140043020248413,
      "step": 4977
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 4977,
      "training_step_time": 9.920453786849976
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 3.1421895027160645,
      "step": 4977
    },
    {
      "dynamics_loss_avg": 0.07181209307163954,
      "epoch": 1.8985748291015626e-05,
      "step": 4977
    },
    {
      "action_loss_avg": 0.030907196458429098,
      "epoch": 1.8985748291015626e-05,
      "step": 4977
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 4977,
      "training_step_time": 10.201587677001953
    },
    {
      "epoch": 1.8985748291015626e-05,
      "model_forward_time": 3.145141363143921,
      "step": 4977
    },
    {
      "epoch": 1.8985748291015626e-05,
      "step": 4977,
      "training_step_time": 10.27216625213623
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 3.187753200531006,
      "step": 4978
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 4978,
      "training_step_time": 9.967552661895752
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 3.1380889415740967,
      "step": 4978
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 4978,
      "training_step_time": 9.912512302398682
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 3.136894464492798,
      "step": 4978
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 4978,
      "training_step_time": 9.908650636672974
    },
    {
      "epoch": 1.898956298828125e-05,
      "model_forward_time": 3.41475772857666,
      "step": 4978
    },
    {
      "epoch": 1.898956298828125e-05,
      "step": 4978,
      "training_step_time": 10.285391569137573
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 3.1812522411346436,
      "step": 4979
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 4979,
      "training_step_time": 9.954123497009277
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 3.1287319660186768,
      "step": 4979
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 4979,
      "training_step_time": 9.898905515670776
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 3.135411024093628,
      "step": 4979
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 4979,
      "training_step_time": 9.903558492660522
    },
    {
      "epoch": 1.8993377685546875e-05,
      "model_forward_time": 3.133521318435669,
      "step": 4979
    },
    {
      "epoch": 1.8993377685546875e-05,
      "step": 4979,
      "training_step_time": 10.312073945999146
    },
    {
      "epoch": 1.89971923828125e-05,
      "grad_norm": 0.5622144937515259,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 4980
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 3.1757829189300537,
      "step": 4980
    },
    {
      "dynamics_loss_avg": 0.0643908909521997,
      "epoch": 1.89971923828125e-05,
      "step": 4980
    },
    {
      "action_loss_avg": 0.02660803203471005,
      "epoch": 1.89971923828125e-05,
      "step": 4980
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 4980,
      "training_step_time": 10.362533569335938
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 3.1348459720611572,
      "step": 4980
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 4980,
      "training_step_time": 9.915346145629883
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 3.137507915496826,
      "step": 4980
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 4980,
      "training_step_time": 9.922950744628906
    },
    {
      "epoch": 1.89971923828125e-05,
      "model_forward_time": 3.1390254497528076,
      "step": 4980
    },
    {
      "epoch": 1.89971923828125e-05,
      "step": 4980,
      "training_step_time": 10.278070449829102
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 3.17417049407959,
      "step": 4981
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 4981,
      "training_step_time": 9.948590517044067
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 3.4368274211883545,
      "step": 4981
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 4981,
      "training_step_time": 10.217477560043335
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 3.145920515060425,
      "step": 4981
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 4981,
      "training_step_time": 9.926467180252075
    },
    {
      "epoch": 1.9001007080078125e-05,
      "model_forward_time": 3.1384782791137695,
      "step": 4981
    },
    {
      "epoch": 1.9001007080078125e-05,
      "step": 4981,
      "training_step_time": 10.248862981796265
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 3.1708827018737793,
      "step": 4982
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 4982,
      "training_step_time": 9.957462787628174
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 3.144564390182495,
      "step": 4982
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 4982,
      "training_step_time": 9.932894706726074
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 3.463196277618408,
      "step": 4982
    },
    {
      "dynamics_loss_avg": 0.0570248132571578,
      "epoch": 1.900482177734375e-05,
      "step": 4982
    },
    {
      "action_loss_avg": 0.02580563146620989,
      "epoch": 1.900482177734375e-05,
      "step": 4982
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 4982,
      "training_step_time": 10.246002435684204
    },
    {
      "epoch": 1.900482177734375e-05,
      "model_forward_time": 3.139845371246338,
      "step": 4982
    },
    {
      "epoch": 1.900482177734375e-05,
      "step": 4982,
      "training_step_time": 10.328786611557007
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 3.1603503227233887,
      "step": 4983
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 4983,
      "training_step_time": 9.941821098327637
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 3.1386234760284424,
      "step": 4983
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 4983,
      "training_step_time": 9.92445182800293
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 3.136173725128174,
      "step": 4983
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 4983,
      "training_step_time": 9.9188973903656
    },
    {
      "epoch": 1.9008636474609374e-05,
      "model_forward_time": 3.406999349594116,
      "step": 4983
    },
    {
      "epoch": 1.9008636474609374e-05,
      "step": 4983,
      "training_step_time": 10.277933835983276
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 3.1576387882232666,
      "step": 4984
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 4984,
      "training_step_time": 9.934767007827759
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 3.135998487472534,
      "step": 4984
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 4984,
      "training_step_time": 9.910649061203003
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 3.132427215576172,
      "step": 4984
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 4984,
      "training_step_time": 9.905317544937134
    },
    {
      "epoch": 1.9012451171875e-05,
      "model_forward_time": 3.135263204574585,
      "step": 4984
    },
    {
      "epoch": 1.9012451171875e-05,
      "step": 4984,
      "training_step_time": 10.29522156715393
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 3.470935106277466,
      "step": 4985
    },
    {
      "dynamics_loss_avg": 0.07163034696131945,
      "epoch": 1.9016265869140626e-05,
      "step": 4985
    },
    {
      "action_loss_avg": 0.030896965228021146,
      "epoch": 1.9016265869140626e-05,
      "step": 4985
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 4985,
      "training_step_time": 10.245438814163208
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 3.1343929767608643,
      "step": 4985
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 4985,
      "training_step_time": 9.901381254196167
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 3.1310245990753174,
      "step": 4985
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 4985,
      "training_step_time": 9.899895429611206
    },
    {
      "epoch": 1.9016265869140626e-05,
      "model_forward_time": 3.1376116275787354,
      "step": 4985
    },
    {
      "epoch": 1.9016265869140626e-05,
      "step": 4985,
      "training_step_time": 10.248695611953735
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 3.1858184337615967,
      "step": 4986
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 4986,
      "training_step_time": 9.9586923122406
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 3.132267713546753,
      "step": 4986
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 4986,
      "training_step_time": 10.202708959579468
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 3.1346843242645264,
      "step": 4986
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 4986,
      "training_step_time": 9.907334804534912
    },
    {
      "epoch": 1.902008056640625e-05,
      "model_forward_time": 3.1365749835968018,
      "step": 4986
    },
    {
      "epoch": 1.902008056640625e-05,
      "step": 4986,
      "training_step_time": 10.274852752685547
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 3.191675901412964,
      "step": 4987
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 4987,
      "training_step_time": 9.975911378860474
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 3.136796236038208,
      "step": 4987
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 4987,
      "training_step_time": 9.920506715774536
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 3.408106565475464,
      "step": 4987
    },
    {
      "dynamics_loss_avg": 0.051352407597005365,
      "epoch": 1.9023895263671875e-05,
      "step": 4987
    },
    {
      "action_loss_avg": 0.023190626967698337,
      "epoch": 1.9023895263671875e-05,
      "step": 4987
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 4987,
      "training_step_time": 10.19985318183899
    },
    {
      "epoch": 1.9023895263671875e-05,
      "model_forward_time": 3.1493637561798096,
      "step": 4987
    },
    {
      "epoch": 1.9023895263671875e-05,
      "step": 4987,
      "training_step_time": 10.246253490447998
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 3.174736499786377,
      "step": 4988
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 4988,
      "training_step_time": 9.960885524749756
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 3.1426565647125244,
      "step": 4988
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 4988,
      "training_step_time": 9.930784463882446
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 3.1461050510406494,
      "step": 4988
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 4988,
      "training_step_time": 9.931257486343384
    },
    {
      "epoch": 1.90277099609375e-05,
      "model_forward_time": 3.1381072998046875,
      "step": 4988
    },
    {
      "epoch": 1.90277099609375e-05,
      "step": 4988,
      "training_step_time": 10.322996616363525
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 3.17132568359375,
      "step": 4989
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 4989,
      "training_step_time": 9.952503204345703
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 3.141855001449585,
      "step": 4989
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 4989,
      "training_step_time": 9.916774272918701
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 3.142573833465576,
      "step": 4989
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 4989,
      "training_step_time": 9.926665782928467
    },
    {
      "epoch": 1.9031524658203125e-05,
      "model_forward_time": 3.145617723464966,
      "step": 4989
    },
    {
      "epoch": 1.9031524658203125e-05,
      "step": 4989,
      "training_step_time": 10.265025854110718
    },
    {
      "epoch": 1.903533935546875e-05,
      "grad_norm": 0.3349017798900604,
      "learning_rate": 1e-05,
      "loss": 0.0854,
      "step": 4990
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 3.4656689167022705,
      "step": 4990
    },
    {
      "dynamics_loss_avg": 0.06691266484558582,
      "epoch": 1.903533935546875e-05,
      "step": 4990
    },
    {
      "action_loss_avg": 0.02818829445168376,
      "epoch": 1.903533935546875e-05,
      "step": 4990
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 4990,
      "training_step_time": 10.245500802993774
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 3.1396262645721436,
      "step": 4990
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 4990,
      "training_step_time": 9.912398099899292
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 3.140423059463501,
      "step": 4990
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 4990,
      "training_step_time": 9.917030096054077
    },
    {
      "epoch": 1.903533935546875e-05,
      "model_forward_time": 3.1381189823150635,
      "step": 4990
    },
    {
      "epoch": 1.903533935546875e-05,
      "step": 4990,
      "training_step_time": 10.290624141693115
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 3.1688387393951416,
      "step": 4991
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 4991,
      "training_step_time": 9.945175170898438
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 3.1441595554351807,
      "step": 4991
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 4991,
      "training_step_time": 10.197884321212769
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 3.143463373184204,
      "step": 4991
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 4991,
      "training_step_time": 9.925812244415283
    },
    {
      "epoch": 1.9039154052734374e-05,
      "model_forward_time": 3.1437394618988037,
      "step": 4991
    },
    {
      "epoch": 1.9039154052734374e-05,
      "step": 4991,
      "training_step_time": 10.294314622879028
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 3.167790651321411,
      "step": 4992
    },
    {
      "epoch": 1.904296875e-05,
      "step": 4992,
      "training_step_time": 9.95535922050476
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 3.1405656337738037,
      "step": 4992
    },
    {
      "epoch": 1.904296875e-05,
      "step": 4992,
      "training_step_time": 9.921047687530518
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 3.4223215579986572,
      "step": 4992
    },
    {
      "dynamics_loss_avg": 0.08383046723902225,
      "epoch": 1.904296875e-05,
      "step": 4992
    },
    {
      "action_loss_avg": 0.040619566850364205,
      "epoch": 1.904296875e-05,
      "step": 4992
    },
    {
      "epoch": 1.904296875e-05,
      "step": 4992,
      "training_step_time": 10.208768129348755
    },
    {
      "epoch": 1.904296875e-05,
      "model_forward_time": 3.141364574432373,
      "step": 4992
    },
    {
      "epoch": 1.904296875e-05,
      "step": 4992,
      "training_step_time": 10.020511865615845
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 3.1748998165130615,
      "step": 4993
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 4993,
      "training_step_time": 9.956645965576172
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 3.1435556411743164,
      "step": 4993
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 4993,
      "training_step_time": 9.923629522323608
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 3.138770580291748,
      "step": 4993
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 4993,
      "training_step_time": 9.922290325164795
    },
    {
      "epoch": 1.9046783447265626e-05,
      "model_forward_time": 3.404294729232788,
      "step": 4993
    },
    {
      "epoch": 1.9046783447265626e-05,
      "step": 4993,
      "training_step_time": 10.296306371688843
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 3.179677963256836,
      "step": 4994
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 4994,
      "training_step_time": 9.96679949760437
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 3.147831678390503,
      "step": 4994
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 4994,
      "training_step_time": 9.934936761856079
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 3.142751932144165,
      "step": 4994
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 4994,
      "training_step_time": 9.930586814880371
    },
    {
      "epoch": 1.905059814453125e-05,
      "model_forward_time": 3.147625684738159,
      "step": 4994
    },
    {
      "epoch": 1.905059814453125e-05,
      "step": 4994,
      "training_step_time": 10.256099462509155
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 3.449904203414917,
      "step": 4995
    },
    {
      "dynamics_loss_avg": 0.06559505909681321,
      "epoch": 1.9054412841796875e-05,
      "step": 4995
    },
    {
      "action_loss_avg": 0.031648145709186795,
      "epoch": 1.9054412841796875e-05,
      "step": 4995
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 4995,
      "training_step_time": 10.236093759536743
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 3.1447360515594482,
      "step": 4995
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 4995,
      "training_step_time": 9.928138732910156
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 3.1475436687469482,
      "step": 4995
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 4995,
      "training_step_time": 9.93323302268982
    },
    {
      "epoch": 1.9054412841796875e-05,
      "model_forward_time": 3.151914596557617,
      "step": 4995
    },
    {
      "epoch": 1.9054412841796875e-05,
      "step": 4995,
      "training_step_time": 10.272014141082764
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 3.174797773361206,
      "step": 4996
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 4996,
      "training_step_time": 9.9648597240448
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 3.449218988418579,
      "step": 4996
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 4996,
      "training_step_time": 10.234009504318237
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 3.145169496536255,
      "step": 4996
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 4996,
      "training_step_time": 9.92956018447876
    },
    {
      "epoch": 1.90582275390625e-05,
      "model_forward_time": 3.1438486576080322,
      "step": 4996
    },
    {
      "epoch": 1.90582275390625e-05,
      "step": 4996,
      "training_step_time": 10.300466060638428
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 3.196969509124756,
      "step": 4997
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 4997,
      "training_step_time": 9.982492446899414
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 3.137723207473755,
      "step": 4997
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 4997,
      "training_step_time": 9.915301322937012
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 3.1415774822235107,
      "step": 4997
    },
    {
      "dynamics_loss_avg": 0.04764838106930256,
      "epoch": 1.9062042236328125e-05,
      "step": 4997
    },
    {
      "action_loss_avg": 0.01734005520120263,
      "epoch": 1.9062042236328125e-05,
      "step": 4997
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 4997,
      "training_step_time": 10.22623610496521
    },
    {
      "epoch": 1.9062042236328125e-05,
      "model_forward_time": 3.143314838409424,
      "step": 4997
    },
    {
      "epoch": 1.9062042236328125e-05,
      "step": 4997,
      "training_step_time": 10.021307468414307
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 3.190376043319702,
      "step": 4998
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 4998,
      "training_step_time": 9.971753358840942
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 3.1419484615325928,
      "step": 4998
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 4998,
      "training_step_time": 9.925867795944214
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 3.1424643993377686,
      "step": 4998
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 4998,
      "training_step_time": 9.913273572921753
    },
    {
      "epoch": 1.906585693359375e-05,
      "model_forward_time": 3.427757501602173,
      "step": 4998
    },
    {
      "epoch": 1.906585693359375e-05,
      "step": 4998,
      "training_step_time": 10.292170524597168
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 3.2132644653320312,
      "step": 4999
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 4999,
      "training_step_time": 9.984898567199707
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 3.135545253753662,
      "step": 4999
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 4999,
      "training_step_time": 9.905702590942383
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 3.1382639408111572,
      "step": 4999
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 4999,
      "training_step_time": 9.903566837310791
    },
    {
      "epoch": 1.9069671630859374e-05,
      "model_forward_time": 3.135282039642334,
      "step": 4999
    },
    {
      "epoch": 1.9069671630859374e-05,
      "step": 4999,
      "training_step_time": 10.268639087677002
    },
    {
      "epoch": 1.9073486328125e-05,
      "grad_norm": 0.3571014106273651,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 5000
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 5.280989646911621,
      "step": 5000
    },
    {
      "dynamics_loss_avg": 0.05953315962105989,
      "epoch": 1.9073486328125e-05,
      "step": 5000
    },
    {
      "action_loss_avg": 0.022660785168409348,
      "epoch": 1.9073486328125e-05,
      "step": 5000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 5000,
      "training_step_time": 12.03698992729187
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 3.1218454837799072,
      "step": 5000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 5000,
      "training_step_time": 9.863496541976929
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 3.1262598037719727,
      "step": 5000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 5000,
      "training_step_time": 9.877880096435547
    },
    {
      "epoch": 1.9073486328125e-05,
      "model_forward_time": 3.132403612136841,
      "step": 5000
    },
    {
      "epoch": 1.9073486328125e-05,
      "step": 5000,
      "training_step_time": 10.313987016677856
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 3.177034854888916,
      "step": 5001
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 5001,
      "training_step_time": 9.93453598022461
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 3.401521921157837,
      "step": 5001
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 5001,
      "training_step_time": 10.163301229476929
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 3.129808187484741,
      "step": 5001
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 5001,
      "training_step_time": 9.893497943878174
    },
    {
      "epoch": 1.9077301025390626e-05,
      "model_forward_time": 3.1348583698272705,
      "step": 5001
    },
    {
      "epoch": 1.9077301025390626e-05,
      "step": 5001,
      "training_step_time": 10.323831558227539
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 3.169198751449585,
      "step": 5002
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 5002,
      "training_step_time": 9.943897008895874
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 3.446018695831299,
      "step": 5002
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 5002,
      "training_step_time": 10.213488340377808
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 3.137028932571411,
      "step": 5002
    },
    {
      "dynamics_loss_avg": 0.06568363271653652,
      "epoch": 1.908111572265625e-05,
      "step": 5002
    },
    {
      "action_loss_avg": 0.034516058210283515,
      "epoch": 1.908111572265625e-05,
      "step": 5002
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 5002,
      "training_step_time": 9.90917420387268
    },
    {
      "epoch": 1.908111572265625e-05,
      "model_forward_time": 3.142829179763794,
      "step": 5002
    },
    {
      "epoch": 1.908111572265625e-05,
      "step": 5002,
      "training_step_time": 10.017767667770386
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 3.1782829761505127,
      "step": 5003
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 5003,
      "training_step_time": 9.944499731063843
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 3.135010242462158,
      "step": 5003
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 5003,
      "training_step_time": 9.908832788467407
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 3.1431210041046143,
      "step": 5003
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 5003,
      "training_step_time": 9.918040990829468
    },
    {
      "epoch": 1.9084930419921875e-05,
      "model_forward_time": 3.418785572052002,
      "step": 5003
    },
    {
      "epoch": 1.9084930419921875e-05,
      "step": 5003,
      "training_step_time": 10.288307189941406
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 3.177215814590454,
      "step": 5004
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 5004,
      "training_step_time": 9.957677125930786
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 3.142798900604248,
      "step": 5004
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 5004,
      "training_step_time": 9.922447681427002
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 3.144695281982422,
      "step": 5004
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 5004,
      "training_step_time": 9.922524690628052
    },
    {
      "epoch": 1.90887451171875e-05,
      "model_forward_time": 3.4181296825408936,
      "step": 5004
    },
    {
      "epoch": 1.90887451171875e-05,
      "step": 5004,
      "training_step_time": 10.289698123931885
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 3.1946358680725098,
      "step": 5005
    },
    {
      "dynamics_loss_avg": 0.0681520402431488,
      "epoch": 1.9092559814453125e-05,
      "step": 5005
    },
    {
      "action_loss_avg": 0.034297469351440665,
      "epoch": 1.9092559814453125e-05,
      "step": 5005
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 5005,
      "training_step_time": 9.97353744506836
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 3.142101764678955,
      "step": 5005
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 5005,
      "training_step_time": 9.921235799789429
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 3.1468470096588135,
      "step": 5005
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 5005,
      "training_step_time": 9.923902988433838
    },
    {
      "epoch": 1.9092559814453125e-05,
      "model_forward_time": 3.1421539783477783,
      "step": 5005
    },
    {
      "epoch": 1.9092559814453125e-05,
      "step": 5005,
      "training_step_time": 10.312074899673462
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 3.454582691192627,
      "step": 5006
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 5006,
      "training_step_time": 10.232968091964722
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 3.141707181930542,
      "step": 5006
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 5006,
      "training_step_time": 9.914346933364868
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 3.142517328262329,
      "step": 5006
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 5006,
      "training_step_time": 9.917051076889038
    },
    {
      "epoch": 1.909637451171875e-05,
      "model_forward_time": 3.137523889541626,
      "step": 5006
    },
    {
      "epoch": 1.909637451171875e-05,
      "step": 5006,
      "training_step_time": 10.296955108642578
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 2.4772064685821533,
      "step": 5007
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 5007,
      "training_step_time": 7.507547616958618
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 3.3896987438201904,
      "step": 5007
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 5007,
      "training_step_time": 10.154090404510498
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 3.1375057697296143,
      "step": 5007
    },
    {
      "dynamics_loss_avg": 0.0719104502350092,
      "epoch": 1.9100189208984374e-05,
      "step": 5007
    },
    {
      "action_loss_avg": 0.0364374415948987,
      "epoch": 1.9100189208984374e-05,
      "step": 5007
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 5007,
      "training_step_time": 9.905058145523071
    },
    {
      "epoch": 1.9100189208984374e-05,
      "model_forward_time": 3.1349384784698486,
      "step": 5007
    },
    {
      "epoch": 1.9100189208984374e-05,
      "step": 5007,
      "training_step_time": 10.288491487503052
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 3.164994239807129,
      "step": 5008
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 5008,
      "training_step_time": 9.934348106384277
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 3.1317765712738037,
      "step": 5008
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 5008,
      "training_step_time": 9.892382860183716
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 3.4045071601867676,
      "step": 5008
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 5008,
      "training_step_time": 10.16405987739563
    },
    {
      "epoch": 1.910400390625e-05,
      "model_forward_time": 3.1269371509552,
      "step": 5008
    },
    {
      "epoch": 1.910400390625e-05,
      "step": 5008,
      "training_step_time": 10.24480128288269
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 3.1639888286590576,
      "step": 5009
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 5009,
      "training_step_time": 9.926786184310913
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 2.428130865097046,
      "step": 5009
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 5009,
      "training_step_time": 7.443313360214233
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 3.125678062438965,
      "step": 5009
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 5009,
      "training_step_time": 9.881535053253174
    },
    {
      "epoch": 1.9107818603515626e-05,
      "model_forward_time": 3.385920286178589,
      "step": 5009
    },
    {
      "epoch": 1.9107818603515626e-05,
      "step": 5009,
      "training_step_time": 10.312270164489746
    },
    {
      "epoch": 1.911163330078125e-05,
      "grad_norm": 0.4676371216773987,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 5010
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 3.1589505672454834,
      "step": 5010
    },
    {
      "dynamics_loss_avg": 0.05792054682970047,
      "epoch": 1.911163330078125e-05,
      "step": 5010
    },
    {
      "action_loss_avg": 0.02954617589712143,
      "epoch": 1.911163330078125e-05,
      "step": 5010
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 5010,
      "training_step_time": 9.92514157295227
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 3.130669593811035,
      "step": 5010
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 5010,
      "training_step_time": 9.903328895568848
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 3.131037950515747,
      "step": 5010
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 5010,
      "training_step_time": 9.90000295639038
    },
    {
      "epoch": 1.911163330078125e-05,
      "model_forward_time": 3.130957841873169,
      "step": 5010
    },
    {
      "epoch": 1.911163330078125e-05,
      "step": 5010,
      "training_step_time": 10.277260780334473
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 3.4262428283691406,
      "step": 5011
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 5011,
      "training_step_time": 10.19527530670166
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 3.14054012298584,
      "step": 5011
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 5011,
      "training_step_time": 9.907419681549072
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 3.1328134536743164,
      "step": 5011
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 5011,
      "training_step_time": 9.897893190383911
    },
    {
      "epoch": 1.9115447998046875e-05,
      "model_forward_time": 3.1348891258239746,
      "step": 5011
    },
    {
      "epoch": 1.9115447998046875e-05,
      "step": 5011,
      "training_step_time": 10.324522018432617
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 3.1760776042938232,
      "step": 5012
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 5012,
      "training_step_time": 9.944052696228027
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 3.4412240982055664,
      "step": 5012
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 5012,
      "training_step_time": 10.207447290420532
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 3.1319539546966553,
      "step": 5012
    },
    {
      "dynamics_loss_avg": 0.049528113566339016,
      "epoch": 1.91192626953125e-05,
      "step": 5012
    },
    {
      "action_loss_avg": 0.019223279831930996,
      "epoch": 1.91192626953125e-05,
      "step": 5012
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 5012,
      "training_step_time": 9.897138357162476
    },
    {
      "epoch": 1.91192626953125e-05,
      "model_forward_time": 3.137155294418335,
      "step": 5012
    },
    {
      "epoch": 1.91192626953125e-05,
      "step": 5012,
      "training_step_time": 10.275640487670898
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 3.138808012008667,
      "step": 5013
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 5013,
      "training_step_time": 9.90305471420288
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 3.128638505935669,
      "step": 5013
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 5013,
      "training_step_time": 9.891572952270508
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 3.394582748413086,
      "step": 5013
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 5013,
      "training_step_time": 10.159642934799194
    },
    {
      "epoch": 1.9123077392578125e-05,
      "model_forward_time": 2.430634021759033,
      "step": 5013
    },
    {
      "epoch": 1.9123077392578125e-05,
      "step": 5013,
      "training_step_time": 10.021799802780151
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 3.1633522510528564,
      "step": 5014
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 5014,
      "training_step_time": 9.93107557296753
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 3.1316332817077637,
      "step": 5014
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 5014,
      "training_step_time": 9.901367664337158
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 3.1351397037506104,
      "step": 5014
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 5014,
      "training_step_time": 9.907017707824707
    },
    {
      "epoch": 1.912689208984375e-05,
      "model_forward_time": 3.1361773014068604,
      "step": 5014
    },
    {
      "epoch": 1.912689208984375e-05,
      "step": 5014,
      "training_step_time": 10.285192489624023
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 3.186826705932617,
      "step": 5015
    },
    {
      "dynamics_loss_avg": 0.05429152343422174,
      "epoch": 1.9130706787109374e-05,
      "step": 5015
    },
    {
      "action_loss_avg": 0.02277756417170167,
      "epoch": 1.9130706787109374e-05,
      "step": 5015
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 5015,
      "training_step_time": 9.962421655654907
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 3.135455846786499,
      "step": 5015
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 5015,
      "training_step_time": 9.901900053024292
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 3.131916046142578,
      "step": 5015
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 5015,
      "training_step_time": 9.891449689865112
    },
    {
      "epoch": 1.9130706787109374e-05,
      "model_forward_time": 3.1273295879364014,
      "step": 5015
    },
    {
      "epoch": 1.9130706787109374e-05,
      "step": 5015,
      "training_step_time": 10.274258136749268
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 3.161421537399292,
      "step": 5016
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 5016,
      "training_step_time": 10.206069469451904
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 3.130476951599121,
      "step": 5016
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 5016,
      "training_step_time": 9.89213228225708
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 3.133533477783203,
      "step": 5016
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 5016,
      "training_step_time": 9.898937225341797
    },
    {
      "epoch": 1.9134521484375e-05,
      "model_forward_time": 3.131690263748169,
      "step": 5016
    },
    {
      "epoch": 1.9134521484375e-05,
      "step": 5016,
      "training_step_time": 10.279163599014282
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 3.1801509857177734,
      "step": 5017
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 5017,
      "training_step_time": 9.944411516189575
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 3.3901798725128174,
      "step": 5017
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 5017,
      "training_step_time": 10.155178785324097
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 3.1347882747650146,
      "step": 5017
    },
    {
      "dynamics_loss_avg": 0.052100732922554016,
      "epoch": 1.9138336181640626e-05,
      "step": 5017
    },
    {
      "action_loss_avg": 0.021014139289036393,
      "epoch": 1.9138336181640626e-05,
      "step": 5017
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 5017,
      "training_step_time": 9.901539087295532
    },
    {
      "epoch": 1.9138336181640626e-05,
      "model_forward_time": 3.133920431137085,
      "step": 5017
    },
    {
      "epoch": 1.9138336181640626e-05,
      "step": 5017,
      "training_step_time": 10.305713176727295
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 3.1690022945404053,
      "step": 5018
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 5018,
      "training_step_time": 9.937439680099487
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 3.1319875717163086,
      "step": 5018
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 5018,
      "training_step_time": 9.89227294921875
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 3.128788471221924,
      "step": 5018
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 5018,
      "training_step_time": 10.157129049301147
    },
    {
      "epoch": 1.914215087890625e-05,
      "model_forward_time": 3.1288723945617676,
      "step": 5018
    },
    {
      "epoch": 1.914215087890625e-05,
      "step": 5018,
      "training_step_time": 10.250725030899048
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 3.159349203109741,
      "step": 5019
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 5019,
      "training_step_time": 9.90785527229309
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 3.126483201980591,
      "step": 5019
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 5019,
      "training_step_time": 9.87941312789917
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 3.126521587371826,
      "step": 5019
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 5019,
      "training_step_time": 9.88062047958374
    },
    {
      "epoch": 1.9145965576171875e-05,
      "model_forward_time": 3.3850321769714355,
      "step": 5019
    },
    {
      "epoch": 1.9145965576171875e-05,
      "step": 5019,
      "training_step_time": 10.232244968414307
    },
    {
      "epoch": 1.91497802734375e-05,
      "grad_norm": 0.4765147566795349,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 5020
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 3.1610982418060303,
      "step": 5020
    },
    {
      "dynamics_loss_avg": 0.0553473174571991,
      "epoch": 1.91497802734375e-05,
      "step": 5020
    },
    {
      "action_loss_avg": 0.025749732623808086,
      "epoch": 1.91497802734375e-05,
      "step": 5020
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 5020,
      "training_step_time": 9.925673723220825
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 3.135796308517456,
      "step": 5020
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 5020,
      "training_step_time": 9.897461175918579
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 3.130620241165161,
      "step": 5020
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 5020,
      "training_step_time": 9.893672704696655
    },
    {
      "epoch": 1.91497802734375e-05,
      "model_forward_time": 3.1341845989227295,
      "step": 5020
    },
    {
      "epoch": 1.91497802734375e-05,
      "step": 5020,
      "training_step_time": 10.269883394241333
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 3.1414201259613037,
      "step": 5021
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 5021,
      "training_step_time": 10.170798301696777
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 3.130479335784912,
      "step": 5021
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 5021,
      "training_step_time": 9.894636392593384
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 3.133246421813965,
      "step": 5021
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 5021,
      "training_step_time": 9.898502349853516
    },
    {
      "epoch": 1.9153594970703125e-05,
      "model_forward_time": 2.43125581741333,
      "step": 5021
    },
    {
      "epoch": 1.9153594970703125e-05,
      "step": 5021,
      "training_step_time": 10.257996797561646
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 3.1845130920410156,
      "step": 5022
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 5022,
      "training_step_time": 9.94894814491272
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 3.40962553024292,
      "step": 5022
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 5022,
      "training_step_time": 10.17760705947876
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 3.1318068504333496,
      "step": 5022
    },
    {
      "dynamics_loss_avg": 0.05490078264847398,
      "epoch": 1.915740966796875e-05,
      "step": 5022
    },
    {
      "action_loss_avg": 0.018507882091216742,
      "epoch": 1.915740966796875e-05,
      "step": 5022
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 5022,
      "training_step_time": 9.899186372756958
    },
    {
      "epoch": 1.915740966796875e-05,
      "model_forward_time": 3.139535903930664,
      "step": 5022
    },
    {
      "epoch": 1.915740966796875e-05,
      "step": 5022,
      "training_step_time": 10.289062261581421
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 3.1659839153289795,
      "step": 5023
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 5023,
      "training_step_time": 9.94138216972351
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 3.133573055267334,
      "step": 5023
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 5023,
      "training_step_time": 9.90276288986206
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 3.396740674972534,
      "step": 5023
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 5023,
      "training_step_time": 10.172957420349121
    },
    {
      "epoch": 1.9161224365234374e-05,
      "model_forward_time": 3.134681224822998,
      "step": 5023
    },
    {
      "epoch": 1.9161224365234374e-05,
      "step": 5023,
      "training_step_time": 10.27552604675293
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 3.1707091331481934,
      "step": 5024
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 5024,
      "training_step_time": 9.936309814453125
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 3.1316473484039307,
      "step": 5024
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 5024,
      "training_step_time": 9.897475719451904
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 3.1321895122528076,
      "step": 5024
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 5024,
      "training_step_time": 9.901479244232178
    },
    {
      "epoch": 1.91650390625e-05,
      "model_forward_time": 3.4228274822235107,
      "step": 5024
    },
    {
      "epoch": 1.91650390625e-05,
      "step": 5024,
      "training_step_time": 10.287349939346313
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 3.1735730171203613,
      "step": 5025
    },
    {
      "dynamics_loss_avg": 0.053928757458925246,
      "epoch": 1.9168853759765626e-05,
      "step": 5025
    },
    {
      "action_loss_avg": 0.016546376841142775,
      "epoch": 1.9168853759765626e-05,
      "step": 5025
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 5025,
      "training_step_time": 9.952367782592773
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 3.135469913482666,
      "step": 5025
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 5025,
      "training_step_time": 9.905339479446411
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 3.1320061683654785,
      "step": 5025
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 5025,
      "training_step_time": 9.903022050857544
    },
    {
      "epoch": 1.9168853759765626e-05,
      "model_forward_time": 3.1299800872802734,
      "step": 5025
    },
    {
      "epoch": 1.9168853759765626e-05,
      "step": 5025,
      "training_step_time": 10.28515887260437
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 3.4676971435546875,
      "step": 5026
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 5026,
      "training_step_time": 10.24012279510498
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 3.1380581855773926,
      "step": 5026
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 5026,
      "training_step_time": 9.91266417503357
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 3.1444499492645264,
      "step": 5026
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 5026,
      "training_step_time": 9.920339107513428
    },
    {
      "epoch": 1.917266845703125e-05,
      "model_forward_time": 3.143362283706665,
      "step": 5026
    },
    {
      "epoch": 1.917266845703125e-05,
      "step": 5026,
      "training_step_time": 10.281924486160278
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 3.1756772994995117,
      "step": 5027
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 5027,
      "training_step_time": 9.950002908706665
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 3.4040141105651855,
      "step": 5027
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 5027,
      "training_step_time": 10.171562194824219
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 3.1386184692382812,
      "step": 5027
    },
    {
      "dynamics_loss_avg": 0.05414619389921427,
      "epoch": 1.9176483154296876e-05,
      "step": 5027
    },
    {
      "action_loss_avg": 0.028152776090428232,
      "epoch": 1.9176483154296876e-05,
      "step": 5027
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 5027,
      "training_step_time": 9.90974760055542
    },
    {
      "epoch": 1.9176483154296876e-05,
      "model_forward_time": 3.13462495803833,
      "step": 5027
    },
    {
      "epoch": 1.9176483154296876e-05,
      "step": 5027,
      "training_step_time": 10.264276504516602
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 3.1554388999938965,
      "step": 5028
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 5028,
      "training_step_time": 9.925421714782715
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 3.136626720428467,
      "step": 5028
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 5028,
      "training_step_time": 9.904256820678711
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 3.4153859615325928,
      "step": 5028
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 5028,
      "training_step_time": 10.181089401245117
    },
    {
      "epoch": 1.91802978515625e-05,
      "model_forward_time": 3.137295961380005,
      "step": 5028
    },
    {
      "epoch": 1.91802978515625e-05,
      "step": 5028,
      "training_step_time": 10.28317642211914
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 2.4768502712249756,
      "step": 5029
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 5029,
      "training_step_time": 7.505994558334351
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 3.144434690475464,
      "step": 5029
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 5029,
      "training_step_time": 9.918222665786743
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 3.14273738861084,
      "step": 5029
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 5029,
      "training_step_time": 9.911603450775146
    },
    {
      "epoch": 1.9184112548828125e-05,
      "model_forward_time": 3.142827272415161,
      "step": 5029
    },
    {
      "epoch": 1.9184112548828125e-05,
      "step": 5029,
      "training_step_time": 10.274861574172974
    },
    {
      "epoch": 1.918792724609375e-05,
      "grad_norm": 0.32285594940185547,
      "learning_rate": 1e-05,
      "loss": 0.0838,
      "step": 5030
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 3.1884357929229736,
      "step": 5030
    },
    {
      "dynamics_loss_avg": 0.055697495117783546,
      "epoch": 1.918792724609375e-05,
      "step": 5030
    },
    {
      "action_loss_avg": 0.024172327481210232,
      "epoch": 1.918792724609375e-05,
      "step": 5030
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 5030,
      "training_step_time": 9.962510824203491
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 3.1429529190063477,
      "step": 5030
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 5030,
      "training_step_time": 9.912814378738403
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 3.1406142711639404,
      "step": 5030
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 5030,
      "training_step_time": 9.907381772994995
    },
    {
      "epoch": 1.918792724609375e-05,
      "model_forward_time": 3.148483991622925,
      "step": 5030
    },
    {
      "epoch": 1.918792724609375e-05,
      "step": 5030,
      "training_step_time": 10.009376287460327
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 3.4427378177642822,
      "step": 5031
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 5031,
      "training_step_time": 10.208497762680054
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 3.141847848892212,
      "step": 5031
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 5031,
      "training_step_time": 9.908392906188965
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 3.1378014087677,
      "step": 5031
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 5031,
      "training_step_time": 9.907032489776611
    },
    {
      "epoch": 1.9191741943359374e-05,
      "model_forward_time": 3.1400837898254395,
      "step": 5031
    },
    {
      "epoch": 1.9191741943359374e-05,
      "step": 5031,
      "training_step_time": 10.280346155166626
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 3.1746950149536133,
      "step": 5032
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 5032,
      "training_step_time": 9.94604754447937
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 3.4267046451568604,
      "step": 5032
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 5032,
      "training_step_time": 10.191757440567017
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 3.1397151947021484,
      "step": 5032
    },
    {
      "dynamics_loss_avg": 0.05276195853948593,
      "epoch": 1.9195556640625e-05,
      "step": 5032
    },
    {
      "action_loss_avg": 0.018143447674810888,
      "epoch": 1.9195556640625e-05,
      "step": 5032
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 5032,
      "training_step_time": 9.907855987548828
    },
    {
      "epoch": 1.9195556640625e-05,
      "model_forward_time": 3.141864061355591,
      "step": 5032
    },
    {
      "epoch": 1.9195556640625e-05,
      "step": 5032,
      "training_step_time": 10.011653423309326
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 3.17630672454834,
      "step": 5033
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 5033,
      "training_step_time": 9.943970918655396
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 3.1381802558898926,
      "step": 5033
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 5033,
      "training_step_time": 9.913395881652832
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 3.4036428928375244,
      "step": 5033
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 5033,
      "training_step_time": 10.172530174255371
    },
    {
      "epoch": 1.9199371337890626e-05,
      "model_forward_time": 3.1434805393218994,
      "step": 5033
    },
    {
      "epoch": 1.9199371337890626e-05,
      "step": 5033,
      "training_step_time": 10.298619747161865
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 3.1837048530578613,
      "step": 5034
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 5034,
      "training_step_time": 9.957042694091797
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 3.1395537853240967,
      "step": 5034
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 5034,
      "training_step_time": 9.909717798233032
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 3.1428492069244385,
      "step": 5034
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 5034,
      "training_step_time": 9.91441011428833
    },
    {
      "epoch": 1.920318603515625e-05,
      "model_forward_time": 3.4264519214630127,
      "step": 5034
    },
    {
      "epoch": 1.920318603515625e-05,
      "step": 5034,
      "training_step_time": 10.319759368896484
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 3.184816837310791,
      "step": 5035
    },
    {
      "dynamics_loss_avg": 0.05491666402667761,
      "epoch": 1.9207000732421876e-05,
      "step": 5035
    },
    {
      "action_loss_avg": 0.02226227717474103,
      "epoch": 1.9207000732421876e-05,
      "step": 5035
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 5035,
      "training_step_time": 9.958004713058472
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 3.1335339546203613,
      "step": 5035
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 5035,
      "training_step_time": 9.900908470153809
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 3.1346805095672607,
      "step": 5035
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 5035,
      "training_step_time": 9.896055221557617
    },
    {
      "epoch": 1.9207000732421876e-05,
      "model_forward_time": 3.133790969848633,
      "step": 5035
    },
    {
      "epoch": 1.9207000732421876e-05,
      "step": 5035,
      "training_step_time": 10.006018161773682
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 3.1825273036956787,
      "step": 5036
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 5036,
      "training_step_time": 10.229288816452026
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 3.1371610164642334,
      "step": 5036
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 5036,
      "training_step_time": 9.90557336807251
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 3.1378424167633057,
      "step": 5036
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 5036,
      "training_step_time": 9.902394771575928
    },
    {
      "epoch": 1.92108154296875e-05,
      "model_forward_time": 3.1355597972869873,
      "step": 5036
    },
    {
      "epoch": 1.92108154296875e-05,
      "step": 5036,
      "training_step_time": 10.301365852355957
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 3.1907708644866943,
      "step": 5037
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 5037,
      "training_step_time": 9.957892656326294
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 3.4015705585479736,
      "step": 5037
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 5037,
      "training_step_time": 10.166204929351807
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 3.138123035430908,
      "step": 5037
    },
    {
      "dynamics_loss_avg": 0.06186361610889435,
      "epoch": 1.9214630126953125e-05,
      "step": 5037
    },
    {
      "action_loss_avg": 0.025523518631234764,
      "epoch": 1.9214630126953125e-05,
      "step": 5037
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 5037,
      "training_step_time": 9.90697693824768
    },
    {
      "epoch": 1.9214630126953125e-05,
      "model_forward_time": 3.1320080757141113,
      "step": 5037
    },
    {
      "epoch": 1.9214630126953125e-05,
      "step": 5037,
      "training_step_time": 10.273085355758667
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 3.1716041564941406,
      "step": 5038
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 5038,
      "training_step_time": 9.938714742660522
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 3.137450695037842,
      "step": 5038
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 5038,
      "training_step_time": 9.904399156570435
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 3.135899543762207,
      "step": 5038
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 5038,
      "training_step_time": 10.20912504196167
    },
    {
      "epoch": 1.921844482421875e-05,
      "model_forward_time": 3.137486457824707,
      "step": 5038
    },
    {
      "epoch": 1.921844482421875e-05,
      "step": 5038,
      "training_step_time": 10.268829584121704
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 3.174039840698242,
      "step": 5039
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 5039,
      "training_step_time": 9.948760032653809
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 3.138653516769409,
      "step": 5039
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 5039,
      "training_step_time": 9.909839630126953
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 3.1346991062164307,
      "step": 5039
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 5039,
      "training_step_time": 9.904620885848999
    },
    {
      "epoch": 1.9222259521484374e-05,
      "model_forward_time": 3.398507833480835,
      "step": 5039
    },
    {
      "epoch": 1.9222259521484374e-05,
      "step": 5039,
      "training_step_time": 10.265697479248047
    },
    {
      "epoch": 1.922607421875e-05,
      "grad_norm": 0.22300703823566437,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 5040
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 3.153200387954712,
      "step": 5040
    },
    {
      "dynamics_loss_avg": 0.04885658901184797,
      "epoch": 1.922607421875e-05,
      "step": 5040
    },
    {
      "action_loss_avg": 0.02563312612473965,
      "epoch": 1.922607421875e-05,
      "step": 5040
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 5040,
      "training_step_time": 9.926825284957886
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 3.134427309036255,
      "step": 5040
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 5040,
      "training_step_time": 9.911724090576172
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 3.1368565559387207,
      "step": 5040
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 5040,
      "training_step_time": 9.909505367279053
    },
    {
      "epoch": 1.922607421875e-05,
      "model_forward_time": 3.12963604927063,
      "step": 5040
    },
    {
      "epoch": 1.922607421875e-05,
      "step": 5040,
      "training_step_time": 10.013283729553223
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 3.1751224994659424,
      "step": 5041
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 5041,
      "training_step_time": 10.21543836593628
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 3.13592267036438,
      "step": 5041
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 5041,
      "training_step_time": 9.900983810424805
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 3.129951238632202,
      "step": 5041
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 5041,
      "training_step_time": 9.894822597503662
    },
    {
      "epoch": 1.9229888916015626e-05,
      "model_forward_time": 3.131633758544922,
      "step": 5041
    },
    {
      "epoch": 1.9229888916015626e-05,
      "step": 5041,
      "training_step_time": 10.295957565307617
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 3.171674966812134,
      "step": 5042
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 5042,
      "training_step_time": 9.93885588645935
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 3.3989434242248535,
      "step": 5042
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 5042,
      "training_step_time": 10.164108276367188
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 3.135335922241211,
      "step": 5042
    },
    {
      "dynamics_loss_avg": 0.06214517522603273,
      "epoch": 1.923370361328125e-05,
      "step": 5042
    },
    {
      "action_loss_avg": 0.02729592556133866,
      "epoch": 1.923370361328125e-05,
      "step": 5042
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 5042,
      "training_step_time": 9.909432172775269
    },
    {
      "epoch": 1.923370361328125e-05,
      "model_forward_time": 3.137831449508667,
      "step": 5042
    },
    {
      "epoch": 1.923370361328125e-05,
      "step": 5042,
      "training_step_time": 10.014656066894531
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 3.167532205581665,
      "step": 5043
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 5043,
      "training_step_time": 9.933869361877441
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 3.136200189590454,
      "step": 5043
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 5043,
      "training_step_time": 9.904266119003296
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 3.387171983718872,
      "step": 5043
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 5043,
      "training_step_time": 10.152086734771729
    },
    {
      "epoch": 1.9237518310546876e-05,
      "model_forward_time": 3.1372151374816895,
      "step": 5043
    },
    {
      "epoch": 1.9237518310546876e-05,
      "step": 5043,
      "training_step_time": 10.013880252838135
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 3.1756210327148438,
      "step": 5044
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 5044,
      "training_step_time": 9.945396423339844
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 3.1317317485809326,
      "step": 5044
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 5044,
      "training_step_time": 9.897246360778809
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 2.4335460662841797,
      "step": 5044
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 5044,
      "training_step_time": 7.455969572067261
    },
    {
      "epoch": 1.92413330078125e-05,
      "model_forward_time": 3.430983304977417,
      "step": 5044
    },
    {
      "epoch": 1.92413330078125e-05,
      "step": 5044,
      "training_step_time": 10.295287370681763
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 3.16353440284729,
      "step": 5045
    },
    {
      "dynamics_loss_avg": 0.07108202017843723,
      "epoch": 1.9245147705078125e-05,
      "step": 5045
    },
    {
      "action_loss_avg": 0.03388176690787077,
      "epoch": 1.9245147705078125e-05,
      "step": 5045
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 5045,
      "training_step_time": 9.93389344215393
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 3.1315369606018066,
      "step": 5045
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 5045,
      "training_step_time": 9.899729490280151
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 3.133333921432495,
      "step": 5045
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 5045,
      "training_step_time": 9.90248417854309
    },
    {
      "epoch": 1.9245147705078125e-05,
      "model_forward_time": 3.133561611175537,
      "step": 5045
    },
    {
      "epoch": 1.9245147705078125e-05,
      "step": 5045,
      "training_step_time": 10.02112102508545
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 3.428237199783325,
      "step": 5046
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 5046,
      "training_step_time": 10.208774089813232
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 3.1362311840057373,
      "step": 5046
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 5046,
      "training_step_time": 9.913241386413574
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 3.142331600189209,
      "step": 5046
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 5046,
      "training_step_time": 9.920184850692749
    },
    {
      "epoch": 1.924896240234375e-05,
      "model_forward_time": 3.1331357955932617,
      "step": 5046
    },
    {
      "epoch": 1.924896240234375e-05,
      "step": 5046,
      "training_step_time": 10.288459539413452
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 3.179246187210083,
      "step": 5047
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 5047,
      "training_step_time": 9.958606958389282
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 3.392565965652466,
      "step": 5047
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 5047,
      "training_step_time": 10.170776844024658
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 3.1405162811279297,
      "step": 5047
    },
    {
      "dynamics_loss_avg": 0.06652734726667404,
      "epoch": 1.9252777099609374e-05,
      "step": 5047
    },
    {
      "action_loss_avg": 0.03748972360044718,
      "epoch": 1.9252777099609374e-05,
      "step": 5047
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 5047,
      "training_step_time": 9.9185209274292
    },
    {
      "epoch": 1.9252777099609374e-05,
      "model_forward_time": 3.135207414627075,
      "step": 5047
    },
    {
      "epoch": 1.9252777099609374e-05,
      "step": 5047,
      "training_step_time": 10.236888408660889
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 3.1766388416290283,
      "step": 5048
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 5048,
      "training_step_time": 9.946373224258423
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 3.1339685916900635,
      "step": 5048
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 5048,
      "training_step_time": 9.904422998428345
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 3.4085910320281982,
      "step": 5048
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 5048,
      "training_step_time": 10.178682327270508
    },
    {
      "epoch": 1.9256591796875e-05,
      "model_forward_time": 3.130970001220703,
      "step": 5048
    },
    {
      "epoch": 1.9256591796875e-05,
      "step": 5048,
      "training_step_time": 10.016347408294678
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 3.153416395187378,
      "step": 5049
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 5049,
      "training_step_time": 9.923434257507324
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 3.1366677284240723,
      "step": 5049
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 5049,
      "training_step_time": 9.904038190841675
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 3.1321585178375244,
      "step": 5049
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 5049,
      "training_step_time": 9.899312019348145
    },
    {
      "epoch": 1.9260406494140626e-05,
      "model_forward_time": 3.386554002761841,
      "step": 5049
    },
    {
      "epoch": 1.9260406494140626e-05,
      "step": 5049,
      "training_step_time": 10.2557954788208
    },
    {
      "epoch": 1.926422119140625e-05,
      "grad_norm": 0.5303308367729187,
      "learning_rate": 1e-05,
      "loss": 0.0855,
      "step": 5050
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 3.1752610206604004,
      "step": 5050
    },
    {
      "dynamics_loss_avg": 0.06144311297684908,
      "epoch": 1.926422119140625e-05,
      "step": 5050
    },
    {
      "action_loss_avg": 0.029781862860545516,
      "epoch": 1.926422119140625e-05,
      "step": 5050
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 5050,
      "training_step_time": 9.95039987564087
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 3.1370232105255127,
      "step": 5050
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 5050,
      "training_step_time": 9.912601232528687
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 3.1347625255584717,
      "step": 5050
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 5050,
      "training_step_time": 9.909960508346558
    },
    {
      "epoch": 1.926422119140625e-05,
      "model_forward_time": 3.1372628211975098,
      "step": 5050
    },
    {
      "epoch": 1.926422119140625e-05,
      "step": 5050,
      "training_step_time": 10.30495285987854
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 3.443895101547241,
      "step": 5051
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 5051,
      "training_step_time": 10.223273754119873
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 3.1405551433563232,
      "step": 5051
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 5051,
      "training_step_time": 9.91644287109375
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 3.136073350906372,
      "step": 5051
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 5051,
      "training_step_time": 9.917281150817871
    },
    {
      "epoch": 1.9268035888671876e-05,
      "model_forward_time": 3.1411406993865967,
      "step": 5051
    },
    {
      "epoch": 1.9268035888671876e-05,
      "step": 5051,
      "training_step_time": 10.321037292480469
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 3.1677751541137695,
      "step": 5052
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 5052,
      "training_step_time": 9.946968793869019
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 3.423990488052368,
      "step": 5052
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 5052,
      "training_step_time": 10.194331645965576
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 3.1304800510406494,
      "step": 5052
    },
    {
      "dynamics_loss_avg": 0.06145186685025692,
      "epoch": 1.92718505859375e-05,
      "step": 5052
    },
    {
      "action_loss_avg": 0.0279630895704031,
      "epoch": 1.92718505859375e-05,
      "step": 5052
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 5052,
      "training_step_time": 9.90596890449524
    },
    {
      "epoch": 1.92718505859375e-05,
      "model_forward_time": 3.133286714553833,
      "step": 5052
    },
    {
      "epoch": 1.92718505859375e-05,
      "step": 5052,
      "training_step_time": 10.24852991104126
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 3.1775052547454834,
      "step": 5053
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 5053,
      "training_step_time": 9.94922137260437
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 3.129671096801758,
      "step": 5053
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 5053,
      "training_step_time": 9.897145748138428
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 3.381625175476074,
      "step": 5053
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 5053,
      "training_step_time": 10.145636081695557
    },
    {
      "epoch": 1.9275665283203125e-05,
      "model_forward_time": 3.130690336227417,
      "step": 5053
    },
    {
      "epoch": 1.9275665283203125e-05,
      "step": 5053,
      "training_step_time": 10.018827199935913
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 3.1660470962524414,
      "step": 5054
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 5054,
      "training_step_time": 9.930144786834717
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 3.1266114711761475,
      "step": 5054
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 5054,
      "training_step_time": 9.894712924957275
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 3.1383612155914307,
      "step": 5054
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 5054,
      "training_step_time": 9.90644383430481
    },
    {
      "epoch": 1.927947998046875e-05,
      "model_forward_time": 3.4175634384155273,
      "step": 5054
    },
    {
      "epoch": 1.927947998046875e-05,
      "step": 5054,
      "training_step_time": 10.283401250839233
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 3.165858030319214,
      "step": 5055
    },
    {
      "dynamics_loss_avg": 0.05508954497054219,
      "epoch": 1.9283294677734374e-05,
      "step": 5055
    },
    {
      "action_loss_avg": 0.027961461804807186,
      "epoch": 1.9283294677734374e-05,
      "step": 5055
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 5055,
      "training_step_time": 9.943575143814087
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 3.134366989135742,
      "step": 5055
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 5055,
      "training_step_time": 9.903286457061768
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 3.1364457607269287,
      "step": 5055
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 5055,
      "training_step_time": 9.90696096420288
    },
    {
      "epoch": 1.9283294677734374e-05,
      "model_forward_time": 3.133854627609253,
      "step": 5055
    },
    {
      "epoch": 1.9283294677734374e-05,
      "step": 5055,
      "training_step_time": 10.272355556488037
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 3.1769168376922607,
      "step": 5056
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 5056,
      "training_step_time": 10.233173608779907
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 3.13531756401062,
      "step": 5056
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 5056,
      "training_step_time": 9.907407283782959
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 3.14001727104187,
      "step": 5056
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 5056,
      "training_step_time": 9.913267612457275
    },
    {
      "epoch": 1.9287109375e-05,
      "model_forward_time": 3.1362578868865967,
      "step": 5056
    },
    {
      "epoch": 1.9287109375e-05,
      "step": 5056,
      "training_step_time": 10.330575227737427
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 3.159471035003662,
      "step": 5057
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 5057,
      "training_step_time": 9.938936710357666
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 3.409120559692383,
      "step": 5057
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 5057,
      "training_step_time": 10.18570852279663
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 3.137838125228882,
      "step": 5057
    },
    {
      "dynamics_loss_avg": 0.06200310736894608,
      "epoch": 1.9290924072265626e-05,
      "step": 5057
    },
    {
      "action_loss_avg": 0.03034001775085926,
      "epoch": 1.9290924072265626e-05,
      "step": 5057
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 5057,
      "training_step_time": 9.91470217704773
    },
    {
      "epoch": 1.9290924072265626e-05,
      "model_forward_time": 3.131673574447632,
      "step": 5057
    },
    {
      "epoch": 1.9290924072265626e-05,
      "step": 5057,
      "training_step_time": 10.034498453140259
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 3.158104181289673,
      "step": 5058
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 5058,
      "training_step_time": 9.929256677627563
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 3.1361565589904785,
      "step": 5058
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 5058,
      "training_step_time": 9.908780574798584
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 3.131828546524048,
      "step": 5058
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 5058,
      "training_step_time": 10.180984020233154
    },
    {
      "epoch": 1.929473876953125e-05,
      "model_forward_time": 3.132446527481079,
      "step": 5058
    },
    {
      "epoch": 1.929473876953125e-05,
      "step": 5058,
      "training_step_time": 10.03486180305481
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 3.1813268661499023,
      "step": 5059
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 5059,
      "training_step_time": 9.954787969589233
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 3.1427814960479736,
      "step": 5059
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 5059,
      "training_step_time": 9.913724899291992
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 3.139096260070801,
      "step": 5059
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 5059,
      "training_step_time": 9.902093172073364
    },
    {
      "epoch": 1.9298553466796876e-05,
      "model_forward_time": 3.394780397415161,
      "step": 5059
    },
    {
      "epoch": 1.9298553466796876e-05,
      "step": 5059,
      "training_step_time": 10.268039464950562
    },
    {
      "epoch": 1.93023681640625e-05,
      "grad_norm": 0.4548547565937042,
      "learning_rate": 1e-05,
      "loss": 0.0888,
      "step": 5060
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 3.1553261280059814,
      "step": 5060
    },
    {
      "dynamics_loss_avg": 0.061028543300926684,
      "epoch": 1.93023681640625e-05,
      "step": 5060
    },
    {
      "action_loss_avg": 0.023500124178826808,
      "epoch": 1.93023681640625e-05,
      "step": 5060
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 5060,
      "training_step_time": 9.921063423156738
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 3.129793405532837,
      "step": 5060
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 5060,
      "training_step_time": 9.888400077819824
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 3.1282413005828857,
      "step": 5060
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 5060,
      "training_step_time": 9.887968301773071
    },
    {
      "epoch": 1.93023681640625e-05,
      "model_forward_time": 3.1348323822021484,
      "step": 5060
    },
    {
      "epoch": 1.93023681640625e-05,
      "step": 5060,
      "training_step_time": 10.233093500137329
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 3.156916379928589,
      "step": 5061
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 5061,
      "training_step_time": 10.191065073013306
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 3.129725933074951,
      "step": 5061
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 5061,
      "training_step_time": 9.898078918457031
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 3.1385488510131836,
      "step": 5061
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 5061,
      "training_step_time": 9.911118745803833
    },
    {
      "epoch": 1.9306182861328125e-05,
      "model_forward_time": 3.1351308822631836,
      "step": 5061
    },
    {
      "epoch": 1.9306182861328125e-05,
      "step": 5061,
      "training_step_time": 10.280460357666016
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 3.16458797454834,
      "step": 5062
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 5062,
      "training_step_time": 9.934000492095947
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 3.409498929977417,
      "step": 5062
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 5062,
      "training_step_time": 10.178382158279419
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 3.133373737335205,
      "step": 5062
    },
    {
      "dynamics_loss_avg": 0.0702486664056778,
      "epoch": 1.930999755859375e-05,
      "step": 5062
    },
    {
      "action_loss_avg": 0.028914911858737468,
      "epoch": 1.930999755859375e-05,
      "step": 5062
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 5062,
      "training_step_time": 9.906563520431519
    },
    {
      "epoch": 1.930999755859375e-05,
      "model_forward_time": 3.137833833694458,
      "step": 5062
    },
    {
      "epoch": 1.930999755859375e-05,
      "step": 5062,
      "training_step_time": 10.021768808364868
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 3.1817753314971924,
      "step": 5063
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 5063,
      "training_step_time": 9.959957122802734
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 3.1352946758270264,
      "step": 5063
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 5063,
      "training_step_time": 9.91174578666687
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 3.394918203353882,
      "step": 5063
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 5063,
      "training_step_time": 10.171700716018677
    },
    {
      "epoch": 1.9313812255859374e-05,
      "model_forward_time": 3.1402814388275146,
      "step": 5063
    },
    {
      "epoch": 1.9313812255859374e-05,
      "step": 5063,
      "training_step_time": 10.07027530670166
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 3.1651220321655273,
      "step": 5064
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 5064,
      "training_step_time": 9.939814567565918
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 3.1372721195220947,
      "step": 5064
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 5064,
      "training_step_time": 9.913540840148926
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 3.139267683029175,
      "step": 5064
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 5064,
      "training_step_time": 9.918002605438232
    },
    {
      "epoch": 1.9317626953125e-05,
      "model_forward_time": 3.4456076622009277,
      "step": 5064
    },
    {
      "epoch": 1.9317626953125e-05,
      "step": 5064,
      "training_step_time": 10.311135530471802
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 3.186850070953369,
      "step": 5065
    },
    {
      "dynamics_loss_avg": 0.046571196801960466,
      "epoch": 1.9321441650390626e-05,
      "step": 5065
    },
    {
      "action_loss_avg": 0.020850703213363887,
      "epoch": 1.9321441650390626e-05,
      "step": 5065
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 5065,
      "training_step_time": 9.964757204055786
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 3.134899377822876,
      "step": 5065
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 5065,
      "training_step_time": 9.90476942062378
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 3.1405625343322754,
      "step": 5065
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 5065,
      "training_step_time": 9.916224718093872
    },
    {
      "epoch": 1.9321441650390626e-05,
      "model_forward_time": 3.146009922027588,
      "step": 5065
    },
    {
      "epoch": 1.9321441650390626e-05,
      "step": 5065,
      "training_step_time": 10.014634132385254
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 3.447502851486206,
      "step": 5066
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 5066,
      "training_step_time": 10.221757173538208
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 3.1309080123901367,
      "step": 5066
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 5066,
      "training_step_time": 9.905049562454224
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 3.1342105865478516,
      "step": 5066
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 5066,
      "training_step_time": 9.911361932754517
    },
    {
      "epoch": 1.932525634765625e-05,
      "model_forward_time": 3.1360864639282227,
      "step": 5066
    },
    {
      "epoch": 1.932525634765625e-05,
      "step": 5066,
      "training_step_time": 10.282577514648438
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 3.178403377532959,
      "step": 5067
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 5067,
      "training_step_time": 9.959559440612793
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 3.3995299339294434,
      "step": 5067
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 5067,
      "training_step_time": 10.16466474533081
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 3.1344401836395264,
      "step": 5067
    },
    {
      "dynamics_loss_avg": 0.06429130882024765,
      "epoch": 1.9329071044921876e-05,
      "step": 5067
    },
    {
      "action_loss_avg": 0.028268968034535647,
      "epoch": 1.9329071044921876e-05,
      "step": 5067
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 5067,
      "training_step_time": 9.901031732559204
    },
    {
      "epoch": 1.9329071044921876e-05,
      "model_forward_time": 3.1291391849517822,
      "step": 5067
    },
    {
      "epoch": 1.9329071044921876e-05,
      "step": 5067,
      "training_step_time": 10.312071800231934
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 3.191243886947632,
      "step": 5068
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 5068,
      "training_step_time": 9.960860967636108
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 3.1379575729370117,
      "step": 5068
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 5068,
      "training_step_time": 9.90695571899414
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 3.414907693862915,
      "step": 5068
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 5068,
      "training_step_time": 10.187931060791016
    },
    {
      "epoch": 1.93328857421875e-05,
      "model_forward_time": 3.1369802951812744,
      "step": 5068
    },
    {
      "epoch": 1.93328857421875e-05,
      "step": 5068,
      "training_step_time": 10.25558590888977
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 3.1887869834899902,
      "step": 5069
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 5069,
      "training_step_time": 9.971493005752563
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 3.135070323944092,
      "step": 5069
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 5069,
      "training_step_time": 9.90158748626709
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 3.1299939155578613,
      "step": 5069
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 5069,
      "training_step_time": 9.890505313873291
    },
    {
      "epoch": 1.9336700439453125e-05,
      "model_forward_time": 3.3769867420196533,
      "step": 5069
    },
    {
      "epoch": 1.9336700439453125e-05,
      "step": 5069,
      "training_step_time": 10.237746000289917
    },
    {
      "epoch": 1.934051513671875e-05,
      "grad_norm": 0.6299375891685486,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 5070
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 3.184048652648926,
      "step": 5070
    },
    {
      "dynamics_loss_avg": 0.06117896437644958,
      "epoch": 1.934051513671875e-05,
      "step": 5070
    },
    {
      "action_loss_avg": 0.02752910554409027,
      "epoch": 1.934051513671875e-05,
      "step": 5070
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 5070,
      "training_step_time": 9.95178747177124
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 2.4322564601898193,
      "step": 5070
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 5070,
      "training_step_time": 7.4541003704071045
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 3.131135940551758,
      "step": 5070
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 5070,
      "training_step_time": 9.89565658569336
    },
    {
      "epoch": 1.934051513671875e-05,
      "model_forward_time": 3.136246681213379,
      "step": 5070
    },
    {
      "epoch": 1.934051513671875e-05,
      "step": 5070,
      "training_step_time": 10.027376890182495
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 3.4291574954986572,
      "step": 5071
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 5071,
      "training_step_time": 10.197531700134277
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 3.139003038406372,
      "step": 5071
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 5071,
      "training_step_time": 9.908057451248169
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 3.132951021194458,
      "step": 5071
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 5071,
      "training_step_time": 9.905260562896729
    },
    {
      "epoch": 1.9344329833984374e-05,
      "model_forward_time": 3.132798194885254,
      "step": 5071
    },
    {
      "epoch": 1.9344329833984374e-05,
      "step": 5071,
      "training_step_time": 10.309766292572021
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 3.1741819381713867,
      "step": 5072
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 5072,
      "training_step_time": 9.937209606170654
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 3.4109036922454834,
      "step": 5072
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 5072,
      "training_step_time": 10.172712802886963
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 3.1335983276367188,
      "step": 5072
    },
    {
      "dynamics_loss_avg": 0.062008580379188064,
      "epoch": 1.934814453125e-05,
      "step": 5072
    },
    {
      "action_loss_avg": 0.02486425437964499,
      "epoch": 1.934814453125e-05,
      "step": 5072
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 5072,
      "training_step_time": 9.897292613983154
    },
    {
      "epoch": 1.934814453125e-05,
      "model_forward_time": 3.137427806854248,
      "step": 5072
    },
    {
      "epoch": 1.934814453125e-05,
      "step": 5072,
      "training_step_time": 10.298505783081055
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 3.1665849685668945,
      "step": 5073
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 5073,
      "training_step_time": 9.94157862663269
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 3.1328225135803223,
      "step": 5073
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 5073,
      "training_step_time": 9.906068325042725
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 3.3968136310577393,
      "step": 5073
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 5073,
      "training_step_time": 10.174787044525146
    },
    {
      "epoch": 1.9351959228515627e-05,
      "model_forward_time": 3.1392784118652344,
      "step": 5073
    },
    {
      "epoch": 1.9351959228515627e-05,
      "step": 5073,
      "training_step_time": 10.300148963928223
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 3.17378830909729,
      "step": 5074
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 5074,
      "training_step_time": 9.950704097747803
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 3.1328179836273193,
      "step": 5074
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 5074,
      "training_step_time": 9.90925908088684
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 3.1358017921447754,
      "step": 5074
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 5074,
      "training_step_time": 9.908628225326538
    },
    {
      "epoch": 1.935577392578125e-05,
      "model_forward_time": 2.6973495483398438,
      "step": 5074
    },
    {
      "epoch": 1.935577392578125e-05,
      "step": 5074,
      "training_step_time": 10.269833087921143
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 3.178382158279419,
      "step": 5075
    },
    {
      "dynamics_loss_avg": 0.0566523764282465,
      "epoch": 1.9359588623046876e-05,
      "step": 5075
    },
    {
      "action_loss_avg": 0.02683774083852768,
      "epoch": 1.9359588623046876e-05,
      "step": 5075
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 5075,
      "training_step_time": 9.954773664474487
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 3.1316897869110107,
      "step": 5075
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 5075,
      "training_step_time": 9.904296636581421
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 3.1360995769500732,
      "step": 5075
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 5075,
      "training_step_time": 9.911711931228638
    },
    {
      "epoch": 1.9359588623046876e-05,
      "model_forward_time": 3.131838083267212,
      "step": 5075
    },
    {
      "epoch": 1.9359588623046876e-05,
      "step": 5075,
      "training_step_time": 10.026737451553345
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 3.4312729835510254,
      "step": 5076
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 5076,
      "training_step_time": 10.20209288597107
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 3.132486343383789,
      "step": 5076
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 5076,
      "training_step_time": 9.907202005386353
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 3.1356260776519775,
      "step": 5076
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 5076,
      "training_step_time": 9.904560327529907
    },
    {
      "epoch": 1.93634033203125e-05,
      "model_forward_time": 3.1335928440093994,
      "step": 5076
    },
    {
      "epoch": 1.93634033203125e-05,
      "step": 5076,
      "training_step_time": 10.274903774261475
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 3.180346965789795,
      "step": 5077
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 5077,
      "training_step_time": 9.953532934188843
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 3.3890130519866943,
      "step": 5077
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 5077,
      "training_step_time": 10.163531064987183
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 3.140854835510254,
      "step": 5077
    },
    {
      "dynamics_loss_avg": 0.04814274329692125,
      "epoch": 1.9367218017578125e-05,
      "step": 5077
    },
    {
      "action_loss_avg": 0.023572403378784655,
      "epoch": 1.9367218017578125e-05,
      "step": 5077
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 5077,
      "training_step_time": 9.920616149902344
    },
    {
      "epoch": 1.9367218017578125e-05,
      "model_forward_time": 3.1388444900512695,
      "step": 5077
    },
    {
      "epoch": 1.9367218017578125e-05,
      "step": 5077,
      "training_step_time": 10.283393621444702
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 3.2108020782470703,
      "step": 5078
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 5078,
      "training_step_time": 9.988263130187988
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 3.141129493713379,
      "step": 5078
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 5078,
      "training_step_time": 9.92351746559143
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 3.415041923522949,
      "step": 5078
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 5078,
      "training_step_time": 10.192041158676147
    },
    {
      "epoch": 1.937103271484375e-05,
      "model_forward_time": 3.1394121646881104,
      "step": 5078
    },
    {
      "epoch": 1.937103271484375e-05,
      "step": 5078,
      "training_step_time": 10.290759086608887
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 3.1892919540405273,
      "step": 5079
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 5079,
      "training_step_time": 9.963420629501343
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 3.1358604431152344,
      "step": 5079
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 5079,
      "training_step_time": 9.90639352798462
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 3.1358320713043213,
      "step": 5079
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 5079,
      "training_step_time": 9.902596712112427
    },
    {
      "epoch": 1.9374847412109374e-05,
      "model_forward_time": 3.395624876022339,
      "step": 5079
    },
    {
      "epoch": 1.9374847412109374e-05,
      "step": 5079,
      "training_step_time": 10.260869264602661
    },
    {
      "epoch": 1.9378662109375e-05,
      "grad_norm": 0.44172403216362,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 5080
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 3.1839921474456787,
      "step": 5080
    },
    {
      "dynamics_loss_avg": 0.046957929991185664,
      "epoch": 1.9378662109375e-05,
      "step": 5080
    },
    {
      "action_loss_avg": 0.018238523881882428,
      "epoch": 1.9378662109375e-05,
      "step": 5080
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 5080,
      "training_step_time": 9.958688974380493
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 3.141256809234619,
      "step": 5080
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 5080,
      "training_step_time": 9.91587519645691
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 3.1345906257629395,
      "step": 5080
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 5080,
      "training_step_time": 9.907224178314209
    },
    {
      "epoch": 1.9378662109375e-05,
      "model_forward_time": 3.143057346343994,
      "step": 5080
    },
    {
      "epoch": 1.9378662109375e-05,
      "step": 5080,
      "training_step_time": 10.024642705917358
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 3.441283702850342,
      "step": 5081
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 5081,
      "training_step_time": 10.22323203086853
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 3.1422226428985596,
      "step": 5081
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 5081,
      "training_step_time": 9.921453475952148
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 3.142212390899658,
      "step": 5081
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 5081,
      "training_step_time": 9.9141526222229
    },
    {
      "epoch": 1.9382476806640627e-05,
      "model_forward_time": 3.1415398120880127,
      "step": 5081
    },
    {
      "epoch": 1.9382476806640627e-05,
      "step": 5081,
      "training_step_time": 10.282207250595093
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 3.170748233795166,
      "step": 5082
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 5082,
      "training_step_time": 9.947705507278442
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 3.414104700088501,
      "step": 5082
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 5082,
      "training_step_time": 10.188681602478027
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 3.1351735591888428,
      "step": 5082
    },
    {
      "dynamics_loss_avg": 0.03940956946462393,
      "epoch": 1.938629150390625e-05,
      "step": 5082
    },
    {
      "action_loss_avg": 0.023935857322067024,
      "epoch": 1.938629150390625e-05,
      "step": 5082
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 5082,
      "training_step_time": 9.910744190216064
    },
    {
      "epoch": 1.938629150390625e-05,
      "model_forward_time": 3.136610984802246,
      "step": 5082
    },
    {
      "epoch": 1.938629150390625e-05,
      "step": 5082,
      "training_step_time": 10.29520058631897
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 3.1811811923980713,
      "step": 5083
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 5083,
      "training_step_time": 9.946263790130615
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 3.130638360977173,
      "step": 5083
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 5083,
      "training_step_time": 9.892446517944336
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 3.3915679454803467,
      "step": 5083
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 5083,
      "training_step_time": 10.159024477005005
    },
    {
      "epoch": 1.9390106201171876e-05,
      "model_forward_time": 3.1379554271698,
      "step": 5083
    },
    {
      "epoch": 1.9390106201171876e-05,
      "step": 5083,
      "training_step_time": 10.27307677268982
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 3.1715991497039795,
      "step": 5084
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 5084,
      "training_step_time": 9.93901252746582
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 3.128587484359741,
      "step": 5084
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 5084,
      "training_step_time": 9.89136266708374
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 3.12994647026062,
      "step": 5084
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 5084,
      "training_step_time": 9.89631986618042
    },
    {
      "epoch": 1.93939208984375e-05,
      "model_forward_time": 3.434920310974121,
      "step": 5084
    },
    {
      "epoch": 1.93939208984375e-05,
      "step": 5084,
      "training_step_time": 10.293720006942749
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 3.159644603729248,
      "step": 5085
    },
    {
      "dynamics_loss_avg": 0.060599977523088454,
      "epoch": 1.9397735595703125e-05,
      "step": 5085
    },
    {
      "action_loss_avg": 0.024316971376538276,
      "epoch": 1.9397735595703125e-05,
      "step": 5085
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 5085,
      "training_step_time": 9.926560878753662
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 3.124765634536743,
      "step": 5085
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 5085,
      "training_step_time": 9.892711639404297
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 3.132585287094116,
      "step": 5085
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 5085,
      "training_step_time": 9.89818263053894
    },
    {
      "epoch": 1.9397735595703125e-05,
      "model_forward_time": 3.13089919090271,
      "step": 5085
    },
    {
      "epoch": 1.9397735595703125e-05,
      "step": 5085,
      "training_step_time": 10.016728639602661
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 3.4373528957366943,
      "step": 5086
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 5086,
      "training_step_time": 10.210984706878662
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 3.141235828399658,
      "step": 5086
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 5086,
      "training_step_time": 9.912429571151733
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 3.14192271232605,
      "step": 5086
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 5086,
      "training_step_time": 9.912098407745361
    },
    {
      "epoch": 1.940155029296875e-05,
      "model_forward_time": 3.1404850482940674,
      "step": 5086
    },
    {
      "epoch": 1.940155029296875e-05,
      "step": 5086,
      "training_step_time": 10.274534940719604
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 3.182183027267456,
      "step": 5087
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 5087,
      "training_step_time": 9.950162172317505
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 3.39239764213562,
      "step": 5087
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 5087,
      "training_step_time": 10.16183853149414
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 3.136441946029663,
      "step": 5087
    },
    {
      "dynamics_loss_avg": 0.05161138698458671,
      "epoch": 1.9405364990234374e-05,
      "step": 5087
    },
    {
      "action_loss_avg": 0.026667419355362654,
      "epoch": 1.9405364990234374e-05,
      "step": 5087
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 5087,
      "training_step_time": 9.913282632827759
    },
    {
      "epoch": 1.9405364990234374e-05,
      "model_forward_time": 3.1354663372039795,
      "step": 5087
    },
    {
      "epoch": 1.9405364990234374e-05,
      "step": 5087,
      "training_step_time": 10.319194793701172
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 3.192551612854004,
      "step": 5088
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 5088,
      "training_step_time": 9.965943813323975
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 3.1376571655273438,
      "step": 5088
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 5088,
      "training_step_time": 9.91089129447937
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 3.4172728061676025,
      "step": 5088
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 5088,
      "training_step_time": 10.18893575668335
    },
    {
      "epoch": 1.94091796875e-05,
      "model_forward_time": 3.138489007949829,
      "step": 5088
    },
    {
      "epoch": 1.94091796875e-05,
      "step": 5088,
      "training_step_time": 10.034150123596191
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 3.148526668548584,
      "step": 5089
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 5089,
      "training_step_time": 9.92707872390747
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 3.144571304321289,
      "step": 5089
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 5089,
      "training_step_time": 9.920745134353638
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 3.1395263671875,
      "step": 5089
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 5089,
      "training_step_time": 9.922590017318726
    },
    {
      "epoch": 1.9412994384765627e-05,
      "model_forward_time": 2.7020013332366943,
      "step": 5089
    },
    {
      "epoch": 1.9412994384765627e-05,
      "step": 5089,
      "training_step_time": 10.293381214141846
    },
    {
      "epoch": 1.941680908203125e-05,
      "grad_norm": 0.4198293685913086,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 5090
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 3.1780645847320557,
      "step": 5090
    },
    {
      "dynamics_loss_avg": 0.03878650013357401,
      "epoch": 1.941680908203125e-05,
      "step": 5090
    },
    {
      "action_loss_avg": 0.01420812695287168,
      "epoch": 1.941680908203125e-05,
      "step": 5090
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 5090,
      "training_step_time": 9.952642679214478
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 3.1373467445373535,
      "step": 5090
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 5090,
      "training_step_time": 9.904512405395508
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 3.1358935832977295,
      "step": 5090
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 5090,
      "training_step_time": 9.902228355407715
    },
    {
      "epoch": 1.941680908203125e-05,
      "model_forward_time": 3.3955070972442627,
      "step": 5090
    },
    {
      "epoch": 1.941680908203125e-05,
      "step": 5090,
      "training_step_time": 10.252573490142822
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 3.1746270656585693,
      "step": 5091
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 5091,
      "training_step_time": 9.94351601600647
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 3.1362593173980713,
      "step": 5091
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 5091,
      "training_step_time": 9.903934955596924
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 3.1329545974731445,
      "step": 5091
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 5091,
      "training_step_time": 9.904582023620605
    },
    {
      "epoch": 1.9420623779296876e-05,
      "model_forward_time": 3.13934326171875,
      "step": 5091
    },
    {
      "epoch": 1.9420623779296876e-05,
      "step": 5091,
      "training_step_time": 10.262538194656372
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 3.169426679611206,
      "step": 5092
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 5092,
      "training_step_time": 10.207547187805176
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 3.1420211791992188,
      "step": 5092
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 5092,
      "training_step_time": 9.916334390640259
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 3.142404794692993,
      "step": 5092
    },
    {
      "dynamics_loss_avg": 0.06086442433297634,
      "epoch": 1.94244384765625e-05,
      "step": 5092
    },
    {
      "action_loss_avg": 0.026686922647058964,
      "epoch": 1.94244384765625e-05,
      "step": 5092
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 5092,
      "training_step_time": 9.924410104751587
    },
    {
      "epoch": 1.94244384765625e-05,
      "model_forward_time": 3.1460275650024414,
      "step": 5092
    },
    {
      "epoch": 1.94244384765625e-05,
      "step": 5092,
      "training_step_time": 10.298062562942505
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 3.183037042617798,
      "step": 5093
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 5093,
      "training_step_time": 9.961748600006104
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 3.4167611598968506,
      "step": 5093
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 5093,
      "training_step_time": 10.19425082206726
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 3.138923406600952,
      "step": 5093
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 5093,
      "training_step_time": 9.919607639312744
    },
    {
      "epoch": 1.9428253173828125e-05,
      "model_forward_time": 3.1564674377441406,
      "step": 5093
    },
    {
      "epoch": 1.9428253173828125e-05,
      "step": 5093,
      "training_step_time": 10.29728889465332
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 3.177548885345459,
      "step": 5094
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 5094,
      "training_step_time": 9.954126119613647
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 3.14067006111145,
      "step": 5094
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 5094,
      "training_step_time": 9.922508478164673
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 3.4074132442474365,
      "step": 5094
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 5094,
      "training_step_time": 10.1893470287323
    },
    {
      "epoch": 1.943206787109375e-05,
      "model_forward_time": 3.1457679271698,
      "step": 5094
    },
    {
      "epoch": 1.943206787109375e-05,
      "step": 5094,
      "training_step_time": 10.289446353912354
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 3.1752421855926514,
      "step": 5095
    },
    {
      "dynamics_loss_avg": 0.062004503607749936,
      "epoch": 1.9435882568359374e-05,
      "step": 5095
    },
    {
      "action_loss_avg": 0.02795644272118807,
      "epoch": 1.9435882568359374e-05,
      "step": 5095
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 5095,
      "training_step_time": 9.958484172821045
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 3.1399624347686768,
      "step": 5095
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 5095,
      "training_step_time": 9.918960809707642
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 3.1433987617492676,
      "step": 5095
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 5095,
      "training_step_time": 9.919289350509644
    },
    {
      "epoch": 1.9435882568359374e-05,
      "model_forward_time": 3.425447940826416,
      "step": 5095
    },
    {
      "epoch": 1.9435882568359374e-05,
      "step": 5095,
      "training_step_time": 10.299133062362671
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 3.181239604949951,
      "step": 5096
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 5096,
      "training_step_time": 9.952207088470459
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 3.130831241607666,
      "step": 5096
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 5096,
      "training_step_time": 9.900396585464478
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 3.133922815322876,
      "step": 5096
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 5096,
      "training_step_time": 9.906262397766113
    },
    {
      "epoch": 1.9439697265625e-05,
      "model_forward_time": 3.126221179962158,
      "step": 5096
    },
    {
      "epoch": 1.9439697265625e-05,
      "step": 5096,
      "training_step_time": 10.008777379989624
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 3.465078115463257,
      "step": 5097
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 5097,
      "training_step_time": 10.226242065429688
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 3.1312549114227295,
      "step": 5097
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 5097,
      "training_step_time": 9.894721269607544
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 3.1360924243927,
      "step": 5097
    },
    {
      "dynamics_loss_avg": 0.052679865062236785,
      "epoch": 1.9443511962890627e-05,
      "step": 5097
    },
    {
      "action_loss_avg": 0.020217420207336544,
      "epoch": 1.9443511962890627e-05,
      "step": 5097
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 5097,
      "training_step_time": 9.903452634811401
    },
    {
      "epoch": 1.9443511962890627e-05,
      "model_forward_time": 3.133643627166748,
      "step": 5097
    },
    {
      "epoch": 1.9443511962890627e-05,
      "step": 5097,
      "training_step_time": 10.283475160598755
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 3.1664974689483643,
      "step": 5098
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 5098,
      "training_step_time": 9.936673879623413
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 3.390817880630493,
      "step": 5098
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 5098,
      "training_step_time": 10.15535283088684
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 3.1299667358398438,
      "step": 5098
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 5098,
      "training_step_time": 9.892646551132202
    },
    {
      "epoch": 1.944732666015625e-05,
      "model_forward_time": 3.124366044998169,
      "step": 5098
    },
    {
      "epoch": 1.944732666015625e-05,
      "step": 5098,
      "training_step_time": 9.995232343673706
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 3.1463279724121094,
      "step": 5099
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 5099,
      "training_step_time": 9.912868976593018
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 3.1302459239959717,
      "step": 5099
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 5099,
      "training_step_time": 9.894230127334595
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 3.414666175842285,
      "step": 5099
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 5099,
      "training_step_time": 10.180237531661987
    },
    {
      "epoch": 1.9451141357421876e-05,
      "model_forward_time": 3.1298277378082275,
      "step": 5099
    },
    {
      "epoch": 1.9451141357421876e-05,
      "step": 5099,
      "training_step_time": 10.018578290939331
    },
    {
      "epoch": 1.94549560546875e-05,
      "grad_norm": 0.6201066374778748,
      "learning_rate": 1e-05,
      "loss": 0.0907,
      "step": 5100
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 3.168217420578003,
      "step": 5100
    },
    {
      "dynamics_loss_avg": 0.07017889879643917,
      "epoch": 1.94549560546875e-05,
      "step": 5100
    },
    {
      "action_loss_avg": 0.031165617145597936,
      "epoch": 1.94549560546875e-05,
      "step": 5100
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 5100,
      "training_step_time": 9.943734884262085
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 3.1285862922668457,
      "step": 5100
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 5100,
      "training_step_time": 9.905047416687012
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 3.131335735321045,
      "step": 5100
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 5100,
      "training_step_time": 9.907845973968506
    },
    {
      "epoch": 1.94549560546875e-05,
      "model_forward_time": 3.421689510345459,
      "step": 5100
    },
    {
      "epoch": 1.94549560546875e-05,
      "step": 5100,
      "training_step_time": 10.302278757095337
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 3.1746609210968018,
      "step": 5101
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 5101,
      "training_step_time": 9.953892707824707
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 3.138136863708496,
      "step": 5101
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 5101,
      "training_step_time": 9.924326181411743
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 3.138512134552002,
      "step": 5101
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 5101,
      "training_step_time": 9.917450904846191
    },
    {
      "epoch": 1.9458770751953125e-05,
      "model_forward_time": 3.134702682495117,
      "step": 5101
    },
    {
      "epoch": 1.9458770751953125e-05,
      "step": 5101,
      "training_step_time": 10.272478342056274
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 3.431014060974121,
      "step": 5102
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 5102,
      "training_step_time": 10.212703704833984
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 3.1405258178710938,
      "step": 5102
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 5102,
      "training_step_time": 9.918539047241211
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 3.1374971866607666,
      "step": 5102
    },
    {
      "dynamics_loss_avg": 0.06919969916343689,
      "epoch": 1.946258544921875e-05,
      "step": 5102
    },
    {
      "action_loss_avg": 0.02519889217801392,
      "epoch": 1.946258544921875e-05,
      "step": 5102
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 5102,
      "training_step_time": 9.919804334640503
    },
    {
      "epoch": 1.946258544921875e-05,
      "model_forward_time": 3.137404441833496,
      "step": 5102
    },
    {
      "epoch": 1.946258544921875e-05,
      "step": 5102,
      "training_step_time": 10.273816108703613
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 3.16235089302063,
      "step": 5103
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 5103,
      "training_step_time": 9.941162586212158
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 3.4100465774536133,
      "step": 5103
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 5103,
      "training_step_time": 10.1881844997406
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 3.1349947452545166,
      "step": 5103
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 5103,
      "training_step_time": 9.910789728164673
    },
    {
      "epoch": 1.9466400146484374e-05,
      "model_forward_time": 3.1344478130340576,
      "step": 5103
    },
    {
      "epoch": 1.9466400146484374e-05,
      "step": 5103,
      "training_step_time": 10.00841236114502
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 3.1712028980255127,
      "step": 5104
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 5104,
      "training_step_time": 9.947887420654297
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 3.127563714981079,
      "step": 5104
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 5104,
      "training_step_time": 9.901894807815552
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 3.1341569423675537,
      "step": 5104
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 5104,
      "training_step_time": 10.198328256607056
    },
    {
      "epoch": 1.947021484375e-05,
      "model_forward_time": 3.134772300720215,
      "step": 5104
    },
    {
      "epoch": 1.947021484375e-05,
      "step": 5104,
      "training_step_time": 10.282655239105225
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 3.1796131134033203,
      "step": 5105
    },
    {
      "dynamics_loss_avg": 0.06372412927448749,
      "epoch": 1.9474029541015627e-05,
      "step": 5105
    },
    {
      "action_loss_avg": 0.029851592425256968,
      "epoch": 1.9474029541015627e-05,
      "step": 5105
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 5105,
      "training_step_time": 9.96000051498413
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 3.1282827854156494,
      "step": 5105
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 5105,
      "training_step_time": 9.904106140136719
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 2.4251527786254883,
      "step": 5105
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 5105,
      "training_step_time": 7.4528703689575195
    },
    {
      "epoch": 1.9474029541015627e-05,
      "model_forward_time": 3.122690439224243,
      "step": 5105
    },
    {
      "epoch": 1.9474029541015627e-05,
      "step": 5105,
      "training_step_time": 10.282116174697876
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 3.4530889987945557,
      "step": 5106
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 5106,
      "training_step_time": 10.223464727401733
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 3.1267404556274414,
      "step": 5106
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 5106,
      "training_step_time": 9.89687967300415
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 3.1306018829345703,
      "step": 5106
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 5106,
      "training_step_time": 9.900642156600952
    },
    {
      "epoch": 1.947784423828125e-05,
      "model_forward_time": 3.1283295154571533,
      "step": 5106
    },
    {
      "epoch": 1.947784423828125e-05,
      "step": 5106,
      "training_step_time": 10.006426334381104
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 3.166382074356079,
      "step": 5107
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 5107,
      "training_step_time": 10.210609436035156
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 3.1254003047943115,
      "step": 5107
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 5107,
      "training_step_time": 9.893843412399292
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 3.128786087036133,
      "step": 5107
    },
    {
      "dynamics_loss_avg": 0.05784163922071457,
      "epoch": 1.9481658935546876e-05,
      "step": 5107
    },
    {
      "action_loss_avg": 0.03228433597832918,
      "epoch": 1.9481658935546876e-05,
      "step": 5107
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 5107,
      "training_step_time": 9.900528907775879
    },
    {
      "epoch": 1.9481658935546876e-05,
      "model_forward_time": 3.1272125244140625,
      "step": 5107
    },
    {
      "epoch": 1.9481658935546876e-05,
      "step": 5107,
      "training_step_time": 10.286176443099976
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 3.1772525310516357,
      "step": 5108
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 5108,
      "training_step_time": 9.952722549438477
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 3.3851757049560547,
      "step": 5108
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 5108,
      "training_step_time": 10.158587455749512
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 3.13551664352417,
      "step": 5108
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 5108,
      "training_step_time": 9.9105384349823
    },
    {
      "epoch": 1.94854736328125e-05,
      "model_forward_time": 3.1313607692718506,
      "step": 5108
    },
    {
      "epoch": 1.94854736328125e-05,
      "step": 5108,
      "training_step_time": 10.017889261245728
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 3.1641428470611572,
      "step": 5109
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 5109,
      "training_step_time": 9.944091320037842
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 3.1327455043792725,
      "step": 5109
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 5109,
      "training_step_time": 9.907816648483276
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 3.1296749114990234,
      "step": 5109
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 5109,
      "training_step_time": 10.167316436767578
    },
    {
      "epoch": 1.9489288330078125e-05,
      "model_forward_time": 3.1291239261627197,
      "step": 5109
    },
    {
      "epoch": 1.9489288330078125e-05,
      "step": 5109,
      "training_step_time": 10.305785417556763
    },
    {
      "epoch": 1.949310302734375e-05,
      "grad_norm": 0.5240788459777832,
      "learning_rate": 1e-05,
      "loss": 0.087,
      "step": 5110
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 3.153886318206787,
      "step": 5110
    },
    {
      "dynamics_loss_avg": 0.06624146774411202,
      "epoch": 1.949310302734375e-05,
      "step": 5110
    },
    {
      "action_loss_avg": 0.024496762175112964,
      "epoch": 1.949310302734375e-05,
      "step": 5110
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 5110,
      "training_step_time": 9.932853698730469
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 3.129606246948242,
      "step": 5110
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 5110,
      "training_step_time": 9.902649879455566
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 3.127779245376587,
      "step": 5110
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 5110,
      "training_step_time": 9.902904748916626
    },
    {
      "epoch": 1.949310302734375e-05,
      "model_forward_time": 3.3784241676330566,
      "step": 5110
    },
    {
      "epoch": 1.949310302734375e-05,
      "step": 5110,
      "training_step_time": 10.249523639678955
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 3.1658787727355957,
      "step": 5111
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 5111,
      "training_step_time": 9.940398454666138
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 3.1286845207214355,
      "step": 5111
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 5111,
      "training_step_time": 9.900700330734253
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 3.1285548210144043,
      "step": 5111
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 5111,
      "training_step_time": 9.90125584602356
    },
    {
      "epoch": 1.9496917724609374e-05,
      "model_forward_time": 3.1282742023468018,
      "step": 5111
    },
    {
      "epoch": 1.9496917724609374e-05,
      "step": 5111,
      "training_step_time": 10.286423683166504
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 3.442091226577759,
      "step": 5112
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 5112,
      "training_step_time": 10.214353799819946
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 3.1326417922973633,
      "step": 5112
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 5112,
      "training_step_time": 9.903443813323975
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 3.1295979022979736,
      "step": 5112
    },
    {
      "dynamics_loss_avg": 0.054043402709066865,
      "epoch": 1.9500732421875e-05,
      "step": 5112
    },
    {
      "action_loss_avg": 0.0247776385396719,
      "epoch": 1.9500732421875e-05,
      "step": 5112
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 5112,
      "training_step_time": 9.906120777130127
    },
    {
      "epoch": 1.9500732421875e-05,
      "model_forward_time": 3.133084774017334,
      "step": 5112
    },
    {
      "epoch": 1.9500732421875e-05,
      "step": 5112,
      "training_step_time": 10.292882204055786
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 3.1663973331451416,
      "step": 5113
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 5113,
      "training_step_time": 9.941108226776123
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 3.4246020317077637,
      "step": 5113
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 5113,
      "training_step_time": 10.201423406600952
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 3.1306304931640625,
      "step": 5113
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 5113,
      "training_step_time": 9.900544881820679
    },
    {
      "epoch": 1.9504547119140627e-05,
      "model_forward_time": 3.1368930339813232,
      "step": 5113
    },
    {
      "epoch": 1.9504547119140627e-05,
      "step": 5113,
      "training_step_time": 10.258304119110107
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 3.182696580886841,
      "step": 5114
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 5114,
      "training_step_time": 9.95987606048584
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 3.1320674419403076,
      "step": 5114
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 5114,
      "training_step_time": 9.903937578201294
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 3.399648666381836,
      "step": 5114
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 5114,
      "training_step_time": 10.171509027481079
    },
    {
      "epoch": 1.950836181640625e-05,
      "model_forward_time": 3.142324686050415,
      "step": 5114
    },
    {
      "epoch": 1.950836181640625e-05,
      "step": 5114,
      "training_step_time": 10.008818864822388
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 3.1735219955444336,
      "step": 5115
    },
    {
      "dynamics_loss_avg": 0.062366426922380924,
      "epoch": 1.9512176513671876e-05,
      "step": 5115
    },
    {
      "action_loss_avg": 0.024707944504916667,
      "epoch": 1.9512176513671876e-05,
      "step": 5115
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 5115,
      "training_step_time": 9.949543714523315
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 3.127654552459717,
      "step": 5115
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 5115,
      "training_step_time": 9.899529457092285
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 3.130505084991455,
      "step": 5115
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 5115,
      "training_step_time": 9.897159814834595
    },
    {
      "epoch": 1.9512176513671876e-05,
      "model_forward_time": 3.4083242416381836,
      "step": 5115
    },
    {
      "epoch": 1.9512176513671876e-05,
      "step": 5115,
      "training_step_time": 10.268100500106812
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 3.156083345413208,
      "step": 5116
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 5116,
      "training_step_time": 9.930108308792114
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 3.1354265213012695,
      "step": 5116
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 5116,
      "training_step_time": 9.910407304763794
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 3.138171672821045,
      "step": 5116
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 5116,
      "training_step_time": 9.917046308517456
    },
    {
      "epoch": 1.95159912109375e-05,
      "model_forward_time": 3.13508939743042,
      "step": 5116
    },
    {
      "epoch": 1.95159912109375e-05,
      "step": 5116,
      "training_step_time": 10.033377885818481
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 3.457155466079712,
      "step": 5117
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 5117,
      "training_step_time": 10.235313892364502
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 3.135389566421509,
      "step": 5117
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 5117,
      "training_step_time": 9.914476156234741
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 3.1402764320373535,
      "step": 5117
    },
    {
      "dynamics_loss_avg": 0.05885545844212174,
      "epoch": 1.9519805908203125e-05,
      "step": 5117
    },
    {
      "action_loss_avg": 0.028352203662507236,
      "epoch": 1.9519805908203125e-05,
      "step": 5117
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 5117,
      "training_step_time": 9.926798343658447
    },
    {
      "epoch": 1.9519805908203125e-05,
      "model_forward_time": 3.1355295181274414,
      "step": 5117
    },
    {
      "epoch": 1.9519805908203125e-05,
      "step": 5117,
      "training_step_time": 10.280832052230835
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 3.173379898071289,
      "step": 5118
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 5118,
      "training_step_time": 9.952434778213501
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 3.4021053314208984,
      "step": 5118
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 5118,
      "training_step_time": 10.180819272994995
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 3.138070821762085,
      "step": 5118
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 5118,
      "training_step_time": 9.917503833770752
    },
    {
      "epoch": 1.952362060546875e-05,
      "model_forward_time": 3.1357829570770264,
      "step": 5118
    },
    {
      "epoch": 1.952362060546875e-05,
      "step": 5118,
      "training_step_time": 10.33149790763855
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 3.162571668624878,
      "step": 5119
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 5119,
      "training_step_time": 9.94067096710205
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 3.1318182945251465,
      "step": 5119
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 5119,
      "training_step_time": 9.90751838684082
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 3.4080066680908203,
      "step": 5119
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 5119,
      "training_step_time": 10.186003923416138
    },
    {
      "epoch": 1.9527435302734374e-05,
      "model_forward_time": 3.136873483657837,
      "step": 5119
    },
    {
      "epoch": 1.9527435302734374e-05,
      "step": 5119,
      "training_step_time": 10.222355842590332
    },
    {
      "epoch": 1.953125e-05,
      "grad_norm": 0.650286853313446,
      "learning_rate": 1e-05,
      "loss": 0.0891,
      "step": 5120
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 3.1870627403259277,
      "step": 5120
    },
    {
      "dynamics_loss_avg": 0.05625906214118004,
      "epoch": 1.953125e-05,
      "step": 5120
    },
    {
      "action_loss_avg": 0.03311924999579787,
      "epoch": 1.953125e-05,
      "step": 5120
    },
    {
      "epoch": 1.953125e-05,
      "step": 5120,
      "training_step_time": 9.971384763717651
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 3.1472790241241455,
      "step": 5120
    },
    {
      "epoch": 1.953125e-05,
      "step": 5120,
      "training_step_time": 9.933613538742065
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 3.142916679382324,
      "step": 5120
    },
    {
      "epoch": 1.953125e-05,
      "step": 5120,
      "training_step_time": 9.923950433731079
    },
    {
      "epoch": 1.953125e-05,
      "model_forward_time": 3.402813196182251,
      "step": 5120
    },
    {
      "epoch": 1.953125e-05,
      "step": 5120,
      "training_step_time": 10.32550048828125
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 3.185530424118042,
      "step": 5121
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 5121,
      "training_step_time": 9.964323043823242
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 3.1399648189544678,
      "step": 5121
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 5121,
      "training_step_time": 9.918011665344238
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 3.1397459506988525,
      "step": 5121
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 5121,
      "training_step_time": 9.918531656265259
    },
    {
      "epoch": 1.9535064697265627e-05,
      "model_forward_time": 3.1414530277252197,
      "step": 5121
    },
    {
      "epoch": 1.9535064697265627e-05,
      "step": 5121,
      "training_step_time": 10.23488736152649
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 3.4347469806671143,
      "step": 5122
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 5122,
      "training_step_time": 10.207069158554077
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 3.1414597034454346,
      "step": 5122
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 5122,
      "training_step_time": 9.917209386825562
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 3.140402317047119,
      "step": 5122
    },
    {
      "dynamics_loss_avg": 0.07417415529489517,
      "epoch": 1.953887939453125e-05,
      "step": 5122
    },
    {
      "action_loss_avg": 0.03174605667591095,
      "epoch": 1.953887939453125e-05,
      "step": 5122
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 5122,
      "training_step_time": 9.919924020767212
    },
    {
      "epoch": 1.953887939453125e-05,
      "model_forward_time": 3.141353130340576,
      "step": 5122
    },
    {
      "epoch": 1.953887939453125e-05,
      "step": 5122,
      "training_step_time": 10.298508167266846
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 3.187575340270996,
      "step": 5123
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 5123,
      "training_step_time": 9.967382669448853
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 3.4562206268310547,
      "step": 5123
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 5123,
      "training_step_time": 10.23376727104187
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 3.1393704414367676,
      "step": 5123
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 5123,
      "training_step_time": 9.921549797058105
    },
    {
      "epoch": 1.9542694091796876e-05,
      "model_forward_time": 3.1486330032348633,
      "step": 5123
    },
    {
      "epoch": 1.9542694091796876e-05,
      "step": 5123,
      "training_step_time": 10.295194149017334
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 3.181346893310547,
      "step": 5124
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 5124,
      "training_step_time": 9.963238954544067
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 3.141437292098999,
      "step": 5124
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 5124,
      "training_step_time": 9.922789573669434
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 3.4085562229156494,
      "step": 5124
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 5124,
      "training_step_time": 10.199145317077637
    },
    {
      "epoch": 1.95465087890625e-05,
      "model_forward_time": 3.147991180419922,
      "step": 5124
    },
    {
      "epoch": 1.95465087890625e-05,
      "step": 5124,
      "training_step_time": 10.26883864402771
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 3.1809427738189697,
      "step": 5125
    },
    {
      "dynamics_loss_avg": 0.05743115544319153,
      "epoch": 1.9550323486328125e-05,
      "step": 5125
    },
    {
      "action_loss_avg": 0.023893648851662876,
      "epoch": 1.9550323486328125e-05,
      "step": 5125
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 5125,
      "training_step_time": 9.965691566467285
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 3.1400482654571533,
      "step": 5125
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 5125,
      "training_step_time": 9.915823221206665
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 3.1430532932281494,
      "step": 5125
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 5125,
      "training_step_time": 9.920718431472778
    },
    {
      "epoch": 1.9550323486328125e-05,
      "model_forward_time": 3.412733793258667,
      "step": 5125
    },
    {
      "epoch": 1.9550323486328125e-05,
      "step": 5125,
      "training_step_time": 10.285028219223022
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 3.1672163009643555,
      "step": 5126
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 5126,
      "training_step_time": 9.947657346725464
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 3.1391546726226807,
      "step": 5126
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 5126,
      "training_step_time": 9.912930250167847
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 3.1379663944244385,
      "step": 5126
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 5126,
      "training_step_time": 9.908282995223999
    },
    {
      "epoch": 1.955413818359375e-05,
      "model_forward_time": 3.138214349746704,
      "step": 5126
    },
    {
      "epoch": 1.955413818359375e-05,
      "step": 5126,
      "training_step_time": 10.300801992416382
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 3.450817823410034,
      "step": 5127
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 5127,
      "training_step_time": 10.224295139312744
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 3.1380603313446045,
      "step": 5127
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 5127,
      "training_step_time": 9.909645318984985
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 3.1414008140563965,
      "step": 5127
    },
    {
      "dynamics_loss_avg": 0.05435638148337603,
      "epoch": 1.9557952880859374e-05,
      "step": 5127
    },
    {
      "action_loss_avg": 0.028972485288977622,
      "epoch": 1.9557952880859374e-05,
      "step": 5127
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 5127,
      "training_step_time": 9.923378467559814
    },
    {
      "epoch": 1.9557952880859374e-05,
      "model_forward_time": 3.1379470825195312,
      "step": 5127
    },
    {
      "epoch": 1.9557952880859374e-05,
      "step": 5127,
      "training_step_time": 10.26723337173462
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 3.1920511722564697,
      "step": 5128
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 5128,
      "training_step_time": 9.96916389465332
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 3.3858726024627686,
      "step": 5128
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 5128,
      "training_step_time": 10.15105938911438
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 3.1366474628448486,
      "step": 5128
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 5128,
      "training_step_time": 9.903151750564575
    },
    {
      "epoch": 1.9561767578125e-05,
      "model_forward_time": 3.1347553730010986,
      "step": 5128
    },
    {
      "epoch": 1.9561767578125e-05,
      "step": 5128,
      "training_step_time": 10.28599762916565
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 3.1834230422973633,
      "step": 5129
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 5129,
      "training_step_time": 9.949794292449951
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 3.140044927597046,
      "step": 5129
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 5129,
      "training_step_time": 9.903613805770874
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 3.409618377685547,
      "step": 5129
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 5129,
      "training_step_time": 10.173213720321655
    },
    {
      "epoch": 1.9565582275390623e-05,
      "model_forward_time": 3.136605739593506,
      "step": 5129
    },
    {
      "epoch": 1.9565582275390623e-05,
      "step": 5129,
      "training_step_time": 10.277852773666382
    },
    {
      "epoch": 1.956939697265625e-05,
      "grad_norm": 0.5044054985046387,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 5130
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 3.1754188537597656,
      "step": 5130
    },
    {
      "dynamics_loss_avg": 0.06066088192164898,
      "epoch": 1.956939697265625e-05,
      "step": 5130
    },
    {
      "action_loss_avg": 0.027057341672480105,
      "epoch": 1.956939697265625e-05,
      "step": 5130
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 5130,
      "training_step_time": 9.949456691741943
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 3.1395812034606934,
      "step": 5130
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 5130,
      "training_step_time": 9.913850784301758
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 3.142507791519165,
      "step": 5130
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 5130,
      "training_step_time": 10.192288398742676
    },
    {
      "epoch": 1.956939697265625e-05,
      "model_forward_time": 3.144181489944458,
      "step": 5130
    },
    {
      "epoch": 1.956939697265625e-05,
      "step": 5130,
      "training_step_time": 10.249754667282104
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 3.188660144805908,
      "step": 5131
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 5131,
      "training_step_time": 9.960319995880127
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 3.1465930938720703,
      "step": 5131
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 5131,
      "training_step_time": 9.913039922714233
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 3.1410982608795166,
      "step": 5131
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 5131,
      "training_step_time": 9.912727355957031
    },
    {
      "epoch": 1.9573211669921876e-05,
      "model_forward_time": 3.145691394805908,
      "step": 5131
    },
    {
      "epoch": 1.9573211669921876e-05,
      "step": 5131,
      "training_step_time": 10.3104407787323
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 3.188140869140625,
      "step": 5132
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 5132,
      "training_step_time": 9.959171295166016
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 3.1419436931610107,
      "step": 5132
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 5132,
      "training_step_time": 9.91283893585205
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 3.1387195587158203,
      "step": 5132
    },
    {
      "dynamics_loss_avg": 0.06131581617519259,
      "epoch": 1.95770263671875e-05,
      "step": 5132
    },
    {
      "action_loss_avg": 0.027809604443609713,
      "epoch": 1.95770263671875e-05,
      "step": 5132
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 5132,
      "training_step_time": 9.908613681793213
    },
    {
      "epoch": 1.95770263671875e-05,
      "model_forward_time": 3.138244867324829,
      "step": 5132
    },
    {
      "epoch": 1.95770263671875e-05,
      "step": 5132,
      "training_step_time": 10.294577836990356
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 3.1750905513763428,
      "step": 5133
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 5133,
      "training_step_time": 10.23855209350586
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 3.146430730819702,
      "step": 5133
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 5133,
      "training_step_time": 9.913995742797852
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 3.143739938735962,
      "step": 5133
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 5133,
      "training_step_time": 9.916525602340698
    },
    {
      "epoch": 1.9580841064453125e-05,
      "model_forward_time": 3.146329164505005,
      "step": 5133
    },
    {
      "epoch": 1.9580841064453125e-05,
      "step": 5133,
      "training_step_time": 10.277666330337524
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 3.1845152378082275,
      "step": 5134
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 5134,
      "training_step_time": 9.952453374862671
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 3.4298760890960693,
      "step": 5134
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 5134,
      "training_step_time": 10.198500871658325
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 3.143665313720703,
      "step": 5134
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 5134,
      "training_step_time": 9.908619403839111
    },
    {
      "epoch": 1.958465576171875e-05,
      "model_forward_time": 3.1382529735565186,
      "step": 5134
    },
    {
      "epoch": 1.958465576171875e-05,
      "step": 5134,
      "training_step_time": 10.01108193397522
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 3.196728467941284,
      "step": 5135
    },
    {
      "dynamics_loss_avg": 0.0676736954599619,
      "epoch": 1.9588470458984374e-05,
      "step": 5135
    },
    {
      "action_loss_avg": 0.025209891237318517,
      "epoch": 1.9588470458984374e-05,
      "step": 5135
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 5135,
      "training_step_time": 9.967026233673096
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 3.139930486679077,
      "step": 5135
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 5135,
      "training_step_time": 9.910962104797363
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 3.1467537879943848,
      "step": 5135
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 5135,
      "training_step_time": 10.178736686706543
    },
    {
      "epoch": 1.9588470458984374e-05,
      "model_forward_time": 3.145204782485962,
      "step": 5135
    },
    {
      "epoch": 1.9588470458984374e-05,
      "step": 5135,
      "training_step_time": 10.281202554702759
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 3.1840736865997314,
      "step": 5136
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 5136,
      "training_step_time": 9.949060440063477
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 3.1393485069274902,
      "step": 5136
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 5136,
      "training_step_time": 9.903162002563477
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 3.1428980827331543,
      "step": 5136
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 5136,
      "training_step_time": 9.91293478012085
    },
    {
      "epoch": 1.959228515625e-05,
      "model_forward_time": 3.4241135120391846,
      "step": 5136
    },
    {
      "epoch": 1.959228515625e-05,
      "step": 5136,
      "training_step_time": 10.286240577697754
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 3.19350266456604,
      "step": 5137
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 5137,
      "training_step_time": 9.96841311454773
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 3.141854763031006,
      "step": 5137
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 5137,
      "training_step_time": 9.911157846450806
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 3.1451473236083984,
      "step": 5137
    },
    {
      "dynamics_loss_avg": 0.05676426813006401,
      "epoch": 1.9596099853515623e-05,
      "step": 5137
    },
    {
      "action_loss_avg": 0.02420173240825534,
      "epoch": 1.9596099853515623e-05,
      "step": 5137
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 5137,
      "training_step_time": 9.918524026870728
    },
    {
      "epoch": 1.9596099853515623e-05,
      "model_forward_time": 3.1415021419525146,
      "step": 5137
    },
    {
      "epoch": 1.9596099853515623e-05,
      "step": 5137,
      "training_step_time": 10.276553869247437
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 3.4462549686431885,
      "step": 5138
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 5138,
      "training_step_time": 10.211487054824829
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 3.132225751876831,
      "step": 5138
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 5138,
      "training_step_time": 9.898147106170654
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 3.135484457015991,
      "step": 5138
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 5138,
      "training_step_time": 9.903999090194702
    },
    {
      "epoch": 1.959991455078125e-05,
      "model_forward_time": 3.13436222076416,
      "step": 5138
    },
    {
      "epoch": 1.959991455078125e-05,
      "step": 5138,
      "training_step_time": 10.284277439117432
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 3.1739418506622314,
      "step": 5139
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 5139,
      "training_step_time": 9.950716495513916
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 3.417322874069214,
      "step": 5139
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 5139,
      "training_step_time": 10.179704189300537
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 3.1321861743927,
      "step": 5139
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 5139,
      "training_step_time": 9.892083406448364
    },
    {
      "epoch": 1.9603729248046876e-05,
      "model_forward_time": 3.1254072189331055,
      "step": 5139
    },
    {
      "epoch": 1.9603729248046876e-05,
      "step": 5139,
      "training_step_time": 10.293484926223755
    },
    {
      "epoch": 1.96075439453125e-05,
      "grad_norm": 0.48253703117370605,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 5140
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 3.1835286617279053,
      "step": 5140
    },
    {
      "dynamics_loss_avg": 0.058104304037988184,
      "epoch": 1.96075439453125e-05,
      "step": 5140
    },
    {
      "action_loss_avg": 0.02413883153349161,
      "epoch": 1.96075439453125e-05,
      "step": 5140
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 5140,
      "training_step_time": 9.95240330696106
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 3.132387399673462,
      "step": 5140
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 5140,
      "training_step_time": 9.892441272735596
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 3.410520315170288,
      "step": 5140
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 5140,
      "training_step_time": 10.172521352767944
    },
    {
      "epoch": 1.96075439453125e-05,
      "model_forward_time": 3.130929470062256,
      "step": 5140
    },
    {
      "epoch": 1.96075439453125e-05,
      "step": 5140,
      "training_step_time": 10.263544797897339
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 3.1656432151794434,
      "step": 5141
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 5141,
      "training_step_time": 9.928703308105469
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 3.1316685676574707,
      "step": 5141
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 5141,
      "training_step_time": 9.898759126663208
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 3.1323609352111816,
      "step": 5141
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 5141,
      "training_step_time": 9.897607803344727
    },
    {
      "epoch": 1.9611358642578125e-05,
      "model_forward_time": 3.1280014514923096,
      "step": 5141
    },
    {
      "epoch": 1.9611358642578125e-05,
      "step": 5141,
      "training_step_time": 10.265482187271118
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 3.169304847717285,
      "step": 5142
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 5142,
      "training_step_time": 9.924965620040894
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 3.1256890296936035,
      "step": 5142
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 5142,
      "training_step_time": 9.880969047546387
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 3.1276252269744873,
      "step": 5142
    },
    {
      "dynamics_loss_avg": 0.0681799165904522,
      "epoch": 1.961517333984375e-05,
      "step": 5142
    },
    {
      "action_loss_avg": 0.03068096302449703,
      "epoch": 1.961517333984375e-05,
      "step": 5142
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 5142,
      "training_step_time": 9.886515378952026
    },
    {
      "epoch": 1.961517333984375e-05,
      "model_forward_time": 3.1329123973846436,
      "step": 5142
    },
    {
      "epoch": 1.961517333984375e-05,
      "step": 5142,
      "training_step_time": 10.311596632003784
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 3.427464246749878,
      "step": 5143
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 5143,
      "training_step_time": 10.189108610153198
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 3.1306114196777344,
      "step": 5143
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 5143,
      "training_step_time": 9.891952991485596
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 3.1294639110565186,
      "step": 5143
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 5143,
      "training_step_time": 9.891409397125244
    },
    {
      "epoch": 1.9618988037109374e-05,
      "model_forward_time": 3.131448745727539,
      "step": 5143
    },
    {
      "epoch": 1.9618988037109374e-05,
      "step": 5143,
      "training_step_time": 10.259665250778198
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 3.1749119758605957,
      "step": 5144
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 5144,
      "training_step_time": 9.943147420883179
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 3.1359832286834717,
      "step": 5144
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 5144,
      "training_step_time": 10.176518678665161
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 3.134268045425415,
      "step": 5144
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 5144,
      "training_step_time": 9.896111488342285
    },
    {
      "epoch": 1.9622802734375e-05,
      "model_forward_time": 3.1370458602905273,
      "step": 5144
    },
    {
      "epoch": 1.9622802734375e-05,
      "step": 5144,
      "training_step_time": 10.282542943954468
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 3.1763739585876465,
      "step": 5145
    },
    {
      "dynamics_loss_avg": 0.0601536326110363,
      "epoch": 1.9626617431640623e-05,
      "step": 5145
    },
    {
      "action_loss_avg": 0.02311056973412633,
      "epoch": 1.9626617431640623e-05,
      "step": 5145
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 5145,
      "training_step_time": 9.941534280776978
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 3.1353394985198975,
      "step": 5145
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 5145,
      "training_step_time": 9.901489496231079
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 3.406691312789917,
      "step": 5145
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 5145,
      "training_step_time": 10.17676043510437
    },
    {
      "epoch": 1.9626617431640623e-05,
      "model_forward_time": 3.1441140174865723,
      "step": 5145
    },
    {
      "epoch": 1.9626617431640623e-05,
      "step": 5145,
      "training_step_time": 10.289027452468872
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 3.183075428009033,
      "step": 5146
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 5146,
      "training_step_time": 9.957432985305786
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 3.144166946411133,
      "step": 5146
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 5146,
      "training_step_time": 9.91985559463501
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 3.142479658126831,
      "step": 5146
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 5146,
      "training_step_time": 9.914664030075073
    },
    {
      "epoch": 1.963043212890625e-05,
      "model_forward_time": 3.1425983905792236,
      "step": 5146
    },
    {
      "epoch": 1.963043212890625e-05,
      "step": 5146,
      "training_step_time": 10.301848888397217
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 3.185412883758545,
      "step": 5147
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 5147,
      "training_step_time": 9.958561420440674
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 3.1353960037231445,
      "step": 5147
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 5147,
      "training_step_time": 9.901672124862671
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 3.1421172618865967,
      "step": 5147
    },
    {
      "dynamics_loss_avg": 0.05962495021522045,
      "epoch": 1.9634246826171876e-05,
      "step": 5147
    },
    {
      "action_loss_avg": 0.03087114430963993,
      "epoch": 1.9634246826171876e-05,
      "step": 5147
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 5147,
      "training_step_time": 9.910059452056885
    },
    {
      "epoch": 1.9634246826171876e-05,
      "model_forward_time": 3.1369521617889404,
      "step": 5147
    },
    {
      "epoch": 1.9634246826171876e-05,
      "step": 5147,
      "training_step_time": 10.221631288528442
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 3.4637091159820557,
      "step": 5148
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 5148,
      "training_step_time": 10.228679895401001
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 3.13932204246521,
      "step": 5148
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 5148,
      "training_step_time": 9.917297601699829
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 3.141866445541382,
      "step": 5148
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 5148,
      "training_step_time": 9.910062313079834
    },
    {
      "epoch": 1.96380615234375e-05,
      "model_forward_time": 3.141970157623291,
      "step": 5148
    },
    {
      "epoch": 1.96380615234375e-05,
      "step": 5148,
      "training_step_time": 10.308199882507324
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 3.171200752258301,
      "step": 5149
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 5149,
      "training_step_time": 9.945953607559204
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 3.416374444961548,
      "step": 5149
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 5149,
      "training_step_time": 10.186884641647339
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 3.143848419189453,
      "step": 5149
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 5149,
      "training_step_time": 9.917438268661499
    },
    {
      "epoch": 1.9641876220703125e-05,
      "model_forward_time": 3.1437036991119385,
      "step": 5149
    },
    {
      "epoch": 1.9641876220703125e-05,
      "step": 5149,
      "training_step_time": 10.277074098587036
    },
    {
      "epoch": 1.964569091796875e-05,
      "grad_norm": 0.3252028524875641,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 5150
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 3.1784861087799072,
      "step": 5150
    },
    {
      "dynamics_loss_avg": 0.06527643315494061,
      "epoch": 1.964569091796875e-05,
      "step": 5150
    },
    {
      "action_loss_avg": 0.026909607090055944,
      "epoch": 1.964569091796875e-05,
      "step": 5150
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 5150,
      "training_step_time": 9.953837156295776
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 3.146069288253784,
      "step": 5150
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 5150,
      "training_step_time": 9.918060064315796
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 3.4260287284851074,
      "step": 5150
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 5150,
      "training_step_time": 10.197314262390137
    },
    {
      "epoch": 1.964569091796875e-05,
      "model_forward_time": 3.140519380569458,
      "step": 5150
    },
    {
      "epoch": 1.964569091796875e-05,
      "step": 5150,
      "training_step_time": 10.278736591339111
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 3.1742186546325684,
      "step": 5151
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 5151,
      "training_step_time": 9.94469928741455
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 3.1406140327453613,
      "step": 5151
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 5151,
      "training_step_time": 9.90959644317627
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 3.136245012283325,
      "step": 5151
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 5151,
      "training_step_time": 9.907230854034424
    },
    {
      "epoch": 1.9649505615234374e-05,
      "model_forward_time": 3.4024155139923096,
      "step": 5151
    },
    {
      "epoch": 1.9649505615234374e-05,
      "step": 5151,
      "training_step_time": 10.263975143432617
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 3.1844663619995117,
      "step": 5152
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 5152,
      "training_step_time": 9.954270124435425
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 3.1418445110321045,
      "step": 5152
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 5152,
      "training_step_time": 9.910022020339966
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 3.138554573059082,
      "step": 5152
    },
    {
      "dynamics_loss_avg": 0.05411401074379683,
      "epoch": 1.96533203125e-05,
      "step": 5152
    },
    {
      "action_loss_avg": 0.02157628503628075,
      "epoch": 1.96533203125e-05,
      "step": 5152
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 5152,
      "training_step_time": 9.913499593734741
    },
    {
      "epoch": 1.96533203125e-05,
      "model_forward_time": 3.142726421356201,
      "step": 5152
    },
    {
      "epoch": 1.96533203125e-05,
      "step": 5152,
      "training_step_time": 10.008037805557251
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 3.4312281608581543,
      "step": 5153
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 5153,
      "training_step_time": 10.199293375015259
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 3.1427981853485107,
      "step": 5153
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 5153,
      "training_step_time": 9.913012742996216
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 3.142033100128174,
      "step": 5153
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 5153,
      "training_step_time": 9.913033723831177
    },
    {
      "epoch": 1.9657135009765623e-05,
      "model_forward_time": 3.1476476192474365,
      "step": 5153
    },
    {
      "epoch": 1.9657135009765623e-05,
      "step": 5153,
      "training_step_time": 10.332590818405151
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 3.1707515716552734,
      "step": 5154
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 5154,
      "training_step_time": 9.9463050365448
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 3.421388864517212,
      "step": 5154
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 5154,
      "training_step_time": 10.194345235824585
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 3.1420974731445312,
      "step": 5154
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 5154,
      "training_step_time": 9.917511463165283
    },
    {
      "epoch": 1.966094970703125e-05,
      "model_forward_time": 3.1438052654266357,
      "step": 5154
    },
    {
      "epoch": 1.966094970703125e-05,
      "step": 5154,
      "training_step_time": 10.016647577285767
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 3.179381847381592,
      "step": 5155
    },
    {
      "dynamics_loss_avg": 0.052467817813158034,
      "epoch": 1.9664764404296876e-05,
      "step": 5155
    },
    {
      "action_loss_avg": 0.02286386340856552,
      "epoch": 1.9664764404296876e-05,
      "step": 5155
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 5155,
      "training_step_time": 9.949743032455444
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 3.137927532196045,
      "step": 5155
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 5155,
      "training_step_time": 9.911866664886475
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 3.405240058898926,
      "step": 5155
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 5155,
      "training_step_time": 10.177824020385742
    },
    {
      "epoch": 1.9664764404296876e-05,
      "model_forward_time": 3.146185874938965,
      "step": 5155
    },
    {
      "epoch": 1.9664764404296876e-05,
      "step": 5155,
      "training_step_time": 10.292166948318481
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 3.1793768405914307,
      "step": 5156
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 5156,
      "training_step_time": 9.95222783088684
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 3.138554573059082,
      "step": 5156
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 5156,
      "training_step_time": 9.902721166610718
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 3.13999605178833,
      "step": 5156
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 5156,
      "training_step_time": 9.90182089805603
    },
    {
      "epoch": 1.96685791015625e-05,
      "model_forward_time": 3.4149348735809326,
      "step": 5156
    },
    {
      "epoch": 1.96685791015625e-05,
      "step": 5156,
      "training_step_time": 10.299529552459717
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 3.1737921237945557,
      "step": 5157
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 5157,
      "training_step_time": 9.939268827438354
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 3.126810312271118,
      "step": 5157
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 5157,
      "training_step_time": 9.892260313034058
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 3.133730173110962,
      "step": 5157
    },
    {
      "dynamics_loss_avg": 0.0651429558172822,
      "epoch": 1.9672393798828125e-05,
      "step": 5157
    },
    {
      "action_loss_avg": 0.02885243631899357,
      "epoch": 1.9672393798828125e-05,
      "step": 5157
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 5157,
      "training_step_time": 9.899280548095703
    },
    {
      "epoch": 1.9672393798828125e-05,
      "model_forward_time": 3.1335623264312744,
      "step": 5157
    },
    {
      "epoch": 1.9672393798828125e-05,
      "step": 5157,
      "training_step_time": 10.01321029663086
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 3.461237668991089,
      "step": 5158
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 5158,
      "training_step_time": 10.234352350234985
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 3.133946657180786,
      "step": 5158
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 5158,
      "training_step_time": 9.90323519706726
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 3.133498430252075,
      "step": 5158
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 5158,
      "training_step_time": 9.90623426437378
    },
    {
      "epoch": 1.967620849609375e-05,
      "model_forward_time": 3.1341629028320312,
      "step": 5158
    },
    {
      "epoch": 1.967620849609375e-05,
      "step": 5158,
      "training_step_time": 10.255658864974976
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 3.1730284690856934,
      "step": 5159
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 5159,
      "training_step_time": 9.949117422103882
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 3.372795343399048,
      "step": 5159
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 5159,
      "training_step_time": 10.143789052963257
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 3.139587879180908,
      "step": 5159
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 5159,
      "training_step_time": 9.91387128829956
    },
    {
      "epoch": 1.9680023193359374e-05,
      "model_forward_time": 3.13832688331604,
      "step": 5159
    },
    {
      "epoch": 1.9680023193359374e-05,
      "step": 5159,
      "training_step_time": 10.318412065505981
    },
    {
      "epoch": 1.9683837890625e-05,
      "grad_norm": 0.36994102597236633,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 5160
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 3.190046787261963,
      "step": 5160
    },
    {
      "dynamics_loss_avg": 0.06394972540438175,
      "epoch": 1.9683837890625e-05,
      "step": 5160
    },
    {
      "action_loss_avg": 0.02443774975836277,
      "epoch": 1.9683837890625e-05,
      "step": 5160
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 5160,
      "training_step_time": 9.963887929916382
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 3.4083192348480225,
      "step": 5160
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 5160,
      "training_step_time": 10.17522144317627
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 3.133535623550415,
      "step": 5160
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 5160,
      "training_step_time": 9.901160478591919
    },
    {
      "epoch": 1.9683837890625e-05,
      "model_forward_time": 3.1307578086853027,
      "step": 5160
    },
    {
      "epoch": 1.9683837890625e-05,
      "step": 5160,
      "training_step_time": 10.007397651672363
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 3.1684389114379883,
      "step": 5161
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 5161,
      "training_step_time": 9.938002824783325
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 3.1339292526245117,
      "step": 5161
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 5161,
      "training_step_time": 9.898324966430664
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 3.1318838596343994,
      "step": 5161
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 5161,
      "training_step_time": 10.187602043151855
    },
    {
      "epoch": 1.9687652587890623e-05,
      "model_forward_time": 3.131146192550659,
      "step": 5161
    },
    {
      "epoch": 1.9687652587890623e-05,
      "step": 5161,
      "training_step_time": 10.28885793685913
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 3.1723520755767822,
      "step": 5162
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 5162,
      "training_step_time": 9.93848705291748
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 3.1319329738616943,
      "step": 5162
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 5162,
      "training_step_time": 9.899013757705688
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 3.13257098197937,
      "step": 5162
    },
    {
      "dynamics_loss_avg": 0.053552799858152866,
      "epoch": 1.969146728515625e-05,
      "step": 5162
    },
    {
      "action_loss_avg": 0.025580454105511307,
      "epoch": 1.969146728515625e-05,
      "step": 5162
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 5162,
      "training_step_time": 9.905121088027954
    },
    {
      "epoch": 1.969146728515625e-05,
      "model_forward_time": 3.3950469493865967,
      "step": 5162
    },
    {
      "epoch": 1.969146728515625e-05,
      "step": 5162,
      "training_step_time": 10.252000093460083
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 3.158661127090454,
      "step": 5163
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 5163,
      "training_step_time": 9.926426887512207
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 3.133859395980835,
      "step": 5163
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 5163,
      "training_step_time": 9.892484664916992
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 3.131237268447876,
      "step": 5163
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 5163,
      "training_step_time": 9.893234252929688
    },
    {
      "epoch": 1.9695281982421876e-05,
      "model_forward_time": 3.132678508758545,
      "step": 5163
    },
    {
      "epoch": 1.9695281982421876e-05,
      "step": 5163,
      "training_step_time": 10.271934509277344
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 3.170117139816284,
      "step": 5164
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 5164,
      "training_step_time": 10.195980310440063
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 3.1376450061798096,
      "step": 5164
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 5164,
      "training_step_time": 9.903308868408203
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 3.143031597137451,
      "step": 5164
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 5164,
      "training_step_time": 9.913495540618896
    },
    {
      "epoch": 1.96990966796875e-05,
      "model_forward_time": 3.1419200897216797,
      "step": 5164
    },
    {
      "epoch": 1.96990966796875e-05,
      "step": 5164,
      "training_step_time": 10.316110372543335
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 3.174536943435669,
      "step": 5165
    },
    {
      "dynamics_loss_avg": 0.07737821638584137,
      "epoch": 1.9702911376953125e-05,
      "step": 5165
    },
    {
      "action_loss_avg": 0.03157700002193451,
      "epoch": 1.9702911376953125e-05,
      "step": 5165
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 5165,
      "training_step_time": 9.952377319335938
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 2.690159797668457,
      "step": 5165
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 5165,
      "training_step_time": 7.717114448547363
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 3.13205623626709,
      "step": 5165
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 5165,
      "training_step_time": 9.89677381515503
    },
    {
      "epoch": 1.9702911376953125e-05,
      "model_forward_time": 3.1358561515808105,
      "step": 5165
    },
    {
      "epoch": 1.9702911376953125e-05,
      "step": 5165,
      "training_step_time": 10.00429916381836
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 3.1643450260162354,
      "step": 5166
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 5166,
      "training_step_time": 9.933196783065796
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 3.1297590732574463,
      "step": 5166
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 5166,
      "training_step_time": 9.896922826766968
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 3.3869941234588623,
      "step": 5166
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 5166,
      "training_step_time": 10.154081344604492
    },
    {
      "epoch": 1.970672607421875e-05,
      "model_forward_time": 3.1337292194366455,
      "step": 5166
    },
    {
      "epoch": 1.970672607421875e-05,
      "step": 5166,
      "training_step_time": 10.283193826675415
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 3.188499689102173,
      "step": 5167
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 5167,
      "training_step_time": 9.954544305801392
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 3.1427769660949707,
      "step": 5167
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 5167,
      "training_step_time": 9.909735441207886
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 3.143439531326294,
      "step": 5167
    },
    {
      "dynamics_loss_avg": 0.08085112199187279,
      "epoch": 1.9710540771484374e-05,
      "step": 5167
    },
    {
      "action_loss_avg": 0.032304642535746095,
      "epoch": 1.9710540771484374e-05,
      "step": 5167
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 5167,
      "training_step_time": 9.910229206085205
    },
    {
      "epoch": 1.9710540771484374e-05,
      "model_forward_time": 3.4537723064422607,
      "step": 5167
    },
    {
      "epoch": 1.9710540771484374e-05,
      "step": 5167,
      "training_step_time": 10.304415225982666
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 3.174326181411743,
      "step": 5168
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 5168,
      "training_step_time": 9.931421995162964
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 3.13118052482605,
      "step": 5168
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 5168,
      "training_step_time": 9.89264464378357
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 3.14597749710083,
      "step": 5168
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 5168,
      "training_step_time": 9.9149489402771
    },
    {
      "epoch": 1.971435546875e-05,
      "model_forward_time": 3.135067939758301,
      "step": 5168
    },
    {
      "epoch": 1.971435546875e-05,
      "step": 5168,
      "training_step_time": 10.009682893753052
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 3.4673924446105957,
      "step": 5169
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 5169,
      "training_step_time": 10.236213445663452
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 3.1347789764404297,
      "step": 5169
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 5169,
      "training_step_time": 9.90062403678894
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 3.1365225315093994,
      "step": 5169
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 5169,
      "training_step_time": 9.901097297668457
    },
    {
      "epoch": 1.9718170166015623e-05,
      "model_forward_time": 3.1321358680725098,
      "step": 5169
    },
    {
      "epoch": 1.9718170166015623e-05,
      "step": 5169,
      "training_step_time": 10.28525447845459
    },
    {
      "epoch": 1.972198486328125e-05,
      "grad_norm": 0.48799142241477966,
      "learning_rate": 1e-05,
      "loss": 0.0886,
      "step": 5170
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 3.1824982166290283,
      "step": 5170
    },
    {
      "dynamics_loss_avg": 0.05025840252637863,
      "epoch": 1.972198486328125e-05,
      "step": 5170
    },
    {
      "action_loss_avg": 0.02054429017007351,
      "epoch": 1.972198486328125e-05,
      "step": 5170
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 5170,
      "training_step_time": 9.95408844947815
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 3.400460720062256,
      "step": 5170
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 5170,
      "training_step_time": 10.162728071212769
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 3.1368110179901123,
      "step": 5170
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 5170,
      "training_step_time": 9.898499011993408
    },
    {
      "epoch": 1.972198486328125e-05,
      "model_forward_time": 3.1299619674682617,
      "step": 5170
    },
    {
      "epoch": 1.972198486328125e-05,
      "step": 5170,
      "training_step_time": 10.002837657928467
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 3.1771180629730225,
      "step": 5171
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 5171,
      "training_step_time": 9.941441774368286
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 3.1345090866088867,
      "step": 5171
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 5171,
      "training_step_time": 9.894800662994385
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 3.4483118057250977,
      "step": 5171
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 5171,
      "training_step_time": 10.212615013122559
    },
    {
      "epoch": 1.9725799560546876e-05,
      "model_forward_time": 3.134342908859253,
      "step": 5171
    },
    {
      "epoch": 1.9725799560546876e-05,
      "step": 5171,
      "training_step_time": 10.234438180923462
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 3.1921703815460205,
      "step": 5172
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 5172,
      "training_step_time": 9.966010093688965
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 3.143493175506592,
      "step": 5172
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 5172,
      "training_step_time": 9.912097215652466
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 3.135650396347046,
      "step": 5172
    },
    {
      "dynamics_loss_avg": 0.06592238917946816,
      "epoch": 1.97296142578125e-05,
      "step": 5172
    },
    {
      "action_loss_avg": 0.028434662614017725,
      "epoch": 1.97296142578125e-05,
      "step": 5172
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 5172,
      "training_step_time": 9.9065260887146
    },
    {
      "epoch": 1.97296142578125e-05,
      "model_forward_time": 3.4132368564605713,
      "step": 5172
    },
    {
      "epoch": 1.97296142578125e-05,
      "step": 5172,
      "training_step_time": 10.293334245681763
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 3.175163745880127,
      "step": 5173
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 5173,
      "training_step_time": 9.943983316421509
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 3.140026330947876,
      "step": 5173
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 5173,
      "training_step_time": 9.906821727752686
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 3.1392407417297363,
      "step": 5173
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 5173,
      "training_step_time": 9.9128258228302
    },
    {
      "epoch": 1.9733428955078125e-05,
      "model_forward_time": 3.1384501457214355,
      "step": 5173
    },
    {
      "epoch": 1.9733428955078125e-05,
      "step": 5173,
      "training_step_time": 10.028513669967651
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 3.4392001628875732,
      "step": 5174
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 5174,
      "training_step_time": 10.202664852142334
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 3.1404638290405273,
      "step": 5174
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 5174,
      "training_step_time": 9.901847124099731
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 3.135204792022705,
      "step": 5174
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 5174,
      "training_step_time": 9.904048204421997
    },
    {
      "epoch": 1.973724365234375e-05,
      "model_forward_time": 3.139838218688965,
      "step": 5174
    },
    {
      "epoch": 1.973724365234375e-05,
      "step": 5174,
      "training_step_time": 10.302217721939087
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 3.1872589588165283,
      "step": 5175
    },
    {
      "dynamics_loss_avg": 0.06824748571962118,
      "epoch": 1.9741058349609374e-05,
      "step": 5175
    },
    {
      "action_loss_avg": 0.0272967922501266,
      "epoch": 1.9741058349609374e-05,
      "step": 5175
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 5175,
      "training_step_time": 9.958194017410278
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 3.4186034202575684,
      "step": 5175
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 5175,
      "training_step_time": 10.180245876312256
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 3.1326968669891357,
      "step": 5175
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 5175,
      "training_step_time": 9.899410724639893
    },
    {
      "epoch": 1.9741058349609374e-05,
      "model_forward_time": 3.137176513671875,
      "step": 5175
    },
    {
      "epoch": 1.9741058349609374e-05,
      "step": 5175,
      "training_step_time": 10.006713628768921
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 3.1646475791931152,
      "step": 5176
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 5176,
      "training_step_time": 9.92711615562439
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 3.1338858604431152,
      "step": 5176
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 5176,
      "training_step_time": 9.901526689529419
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 3.4195098876953125,
      "step": 5176
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 5176,
      "training_step_time": 10.187849044799805
    },
    {
      "epoch": 1.9744873046875e-05,
      "model_forward_time": 3.1437172889709473,
      "step": 5176
    },
    {
      "epoch": 1.9744873046875e-05,
      "step": 5176,
      "training_step_time": 10.029236078262329
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 3.1858291625976562,
      "step": 5177
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 5177,
      "training_step_time": 9.954455375671387
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 3.1403441429138184,
      "step": 5177
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 5177,
      "training_step_time": 9.911535501480103
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 3.145747423171997,
      "step": 5177
    },
    {
      "dynamics_loss_avg": 0.05757713504135609,
      "epoch": 1.9748687744140623e-05,
      "step": 5177
    },
    {
      "action_loss_avg": 0.022506764880381523,
      "epoch": 1.9748687744140623e-05,
      "step": 5177
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 5177,
      "training_step_time": 9.921453714370728
    },
    {
      "epoch": 1.9748687744140623e-05,
      "model_forward_time": 3.4546258449554443,
      "step": 5177
    },
    {
      "epoch": 1.9748687744140623e-05,
      "step": 5177,
      "training_step_time": 10.33818507194519
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 3.189207077026367,
      "step": 5178
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 5178,
      "training_step_time": 9.963062524795532
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 3.1452224254608154,
      "step": 5178
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 5178,
      "training_step_time": 9.916805744171143
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 3.1503994464874268,
      "step": 5178
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 5178,
      "training_step_time": 9.923829793930054
    },
    {
      "epoch": 1.975250244140625e-05,
      "model_forward_time": 3.14176607131958,
      "step": 5178
    },
    {
      "epoch": 1.975250244140625e-05,
      "step": 5178,
      "training_step_time": 10.007522344589233
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 3.194946765899658,
      "step": 5179
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 5179,
      "training_step_time": 10.258288383483887
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 3.1426899433135986,
      "step": 5179
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 5179,
      "training_step_time": 9.918941259384155
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 3.145902633666992,
      "step": 5179
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 5179,
      "training_step_time": 9.9207923412323
    },
    {
      "epoch": 1.9756317138671876e-05,
      "model_forward_time": 3.141939878463745,
      "step": 5179
    },
    {
      "epoch": 1.9756317138671876e-05,
      "step": 5179,
      "training_step_time": 10.021342515945435
    },
    {
      "epoch": 1.97601318359375e-05,
      "grad_norm": 0.4535236954689026,
      "learning_rate": 1e-05,
      "loss": 0.085,
      "step": 5180
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 3.187852621078491,
      "step": 5180
    },
    {
      "dynamics_loss_avg": 0.06237804032862186,
      "epoch": 1.97601318359375e-05,
      "step": 5180
    },
    {
      "action_loss_avg": 0.02852742373943329,
      "epoch": 1.97601318359375e-05,
      "step": 5180
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 5180,
      "training_step_time": 9.963043928146362
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 3.409621238708496,
      "step": 5180
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 5180,
      "training_step_time": 10.181456089019775
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 3.146695375442505,
      "step": 5180
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 5180,
      "training_step_time": 9.930212259292603
    },
    {
      "epoch": 1.97601318359375e-05,
      "model_forward_time": 3.143547534942627,
      "step": 5180
    },
    {
      "epoch": 1.97601318359375e-05,
      "step": 5180,
      "training_step_time": 10.296685695648193
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 3.1725776195526123,
      "step": 5181
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 5181,
      "training_step_time": 9.94644808769226
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 3.140483856201172,
      "step": 5181
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 5181,
      "training_step_time": 9.911578893661499
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 3.1376657485961914,
      "step": 5181
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 5181,
      "training_step_time": 10.176499366760254
    },
    {
      "epoch": 1.9763946533203125e-05,
      "model_forward_time": 3.1362807750701904,
      "step": 5181
    },
    {
      "epoch": 1.9763946533203125e-05,
      "step": 5181,
      "training_step_time": 10.000707864761353
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 3.1790151596069336,
      "step": 5182
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 5182,
      "training_step_time": 9.945931196212769
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 3.1375060081481934,
      "step": 5182
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 5182,
      "training_step_time": 9.902244806289673
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 3.1350674629211426,
      "step": 5182
    },
    {
      "dynamics_loss_avg": 0.055263814702630046,
      "epoch": 1.976776123046875e-05,
      "step": 5182
    },
    {
      "action_loss_avg": 0.020855520106852054,
      "epoch": 1.976776123046875e-05,
      "step": 5182
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 5182,
      "training_step_time": 9.900543928146362
    },
    {
      "epoch": 1.976776123046875e-05,
      "model_forward_time": 3.3980674743652344,
      "step": 5182
    },
    {
      "epoch": 1.976776123046875e-05,
      "step": 5182,
      "training_step_time": 10.256210803985596
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 3.169315814971924,
      "step": 5183
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 5183,
      "training_step_time": 9.939530611038208
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 3.14066481590271,
      "step": 5183
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 5183,
      "training_step_time": 9.906871318817139
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 3.1391444206237793,
      "step": 5183
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 5183,
      "training_step_time": 9.909291505813599
    },
    {
      "epoch": 1.9771575927734374e-05,
      "model_forward_time": 3.1412205696105957,
      "step": 5183
    },
    {
      "epoch": 1.9771575927734374e-05,
      "step": 5183,
      "training_step_time": 10.256737232208252
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 3.1649789810180664,
      "step": 5184
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 5184,
      "training_step_time": 10.232951402664185
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 3.140594482421875,
      "step": 5184
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 5184,
      "training_step_time": 9.90765118598938
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 3.1387455463409424,
      "step": 5184
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 5184,
      "training_step_time": 9.910374402999878
    },
    {
      "epoch": 1.9775390625e-05,
      "model_forward_time": 3.135643243789673,
      "step": 5184
    },
    {
      "epoch": 1.9775390625e-05,
      "step": 5184,
      "training_step_time": 10.286737442016602
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 3.1603636741638184,
      "step": 5185
    },
    {
      "dynamics_loss_avg": 0.05225768908858299,
      "epoch": 1.9779205322265624e-05,
      "step": 5185
    },
    {
      "action_loss_avg": 0.025578020233660938,
      "epoch": 1.9779205322265624e-05,
      "step": 5185
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 5185,
      "training_step_time": 9.932255029678345
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 3.4287891387939453,
      "step": 5185
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 5185,
      "training_step_time": 10.199217796325684
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 3.1410768032073975,
      "step": 5185
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 5185,
      "training_step_time": 9.914595365524292
    },
    {
      "epoch": 1.9779205322265624e-05,
      "model_forward_time": 3.1454849243164062,
      "step": 5185
    },
    {
      "epoch": 1.9779205322265624e-05,
      "step": 5185,
      "training_step_time": 10.304908037185669
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 3.155000925064087,
      "step": 5186
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 5186,
      "training_step_time": 9.92888879776001
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 3.141916275024414,
      "step": 5186
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 5186,
      "training_step_time": 9.916536092758179
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 3.40136981010437,
      "step": 5186
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 5186,
      "training_step_time": 10.169753313064575
    },
    {
      "epoch": 1.978302001953125e-05,
      "model_forward_time": 3.137216567993164,
      "step": 5186
    },
    {
      "epoch": 1.978302001953125e-05,
      "step": 5186,
      "training_step_time": 10.015286207199097
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 3.1791138648986816,
      "step": 5187
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 5187,
      "training_step_time": 9.950746774673462
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 3.1347506046295166,
      "step": 5187
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 5187,
      "training_step_time": 9.903325080871582
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 3.136725664138794,
      "step": 5187
    },
    {
      "dynamics_loss_avg": 0.058195669390261176,
      "epoch": 1.9786834716796876e-05,
      "step": 5187
    },
    {
      "action_loss_avg": 0.021365167992189526,
      "epoch": 1.9786834716796876e-05,
      "step": 5187
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 5187,
      "training_step_time": 9.909894704818726
    },
    {
      "epoch": 1.9786834716796876e-05,
      "model_forward_time": 3.4209072589874268,
      "step": 5187
    },
    {
      "epoch": 1.9786834716796876e-05,
      "step": 5187,
      "training_step_time": 10.284059047698975
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 3.1592462062835693,
      "step": 5188
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 5188,
      "training_step_time": 9.927855968475342
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 3.134221315383911,
      "step": 5188
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 5188,
      "training_step_time": 9.902629137039185
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 3.1397149562835693,
      "step": 5188
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 5188,
      "training_step_time": 9.906880855560303
    },
    {
      "epoch": 1.97906494140625e-05,
      "model_forward_time": 3.135507106781006,
      "step": 5188
    },
    {
      "epoch": 1.97906494140625e-05,
      "step": 5188,
      "training_step_time": 10.327651739120483
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 3.4376423358917236,
      "step": 5189
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 5189,
      "training_step_time": 10.204445362091064
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 3.1356794834136963,
      "step": 5189
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 5189,
      "training_step_time": 9.900894165039062
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 3.13649320602417,
      "step": 5189
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 5189,
      "training_step_time": 9.89827823638916
    },
    {
      "epoch": 1.9794464111328125e-05,
      "model_forward_time": 3.137129783630371,
      "step": 5189
    },
    {
      "epoch": 1.9794464111328125e-05,
      "step": 5189,
      "training_step_time": 10.361027002334595
    },
    {
      "epoch": 1.979827880859375e-05,
      "grad_norm": 0.3695584833621979,
      "learning_rate": 1e-05,
      "loss": 0.0835,
      "step": 5190
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 3.1597230434417725,
      "step": 5190
    },
    {
      "dynamics_loss_avg": 0.042083234339952466,
      "epoch": 1.979827880859375e-05,
      "step": 5190
    },
    {
      "action_loss_avg": 0.015330362203530967,
      "epoch": 1.979827880859375e-05,
      "step": 5190
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 5190,
      "training_step_time": 9.92726182937622
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 3.391878366470337,
      "step": 5190
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 5190,
      "training_step_time": 10.154633283615112
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 3.141322135925293,
      "step": 5190
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 5190,
      "training_step_time": 9.907280206680298
    },
    {
      "epoch": 1.979827880859375e-05,
      "model_forward_time": 3.131484031677246,
      "step": 5190
    },
    {
      "epoch": 1.979827880859375e-05,
      "step": 5190,
      "training_step_time": 10.314763069152832
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 3.1767446994781494,
      "step": 5191
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 5191,
      "training_step_time": 9.937358856201172
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 3.132370948791504,
      "step": 5191
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 5191,
      "training_step_time": 9.892867803573608
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 3.414891004562378,
      "step": 5191
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 5191,
      "training_step_time": 10.17675232887268
    },
    {
      "epoch": 1.9802093505859374e-05,
      "model_forward_time": 3.1334118843078613,
      "step": 5191
    },
    {
      "epoch": 1.9802093505859374e-05,
      "step": 5191,
      "training_step_time": 10.264558792114258
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 3.1658573150634766,
      "step": 5192
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 5192,
      "training_step_time": 9.93141770362854
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 3.1331496238708496,
      "step": 5192
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 5192,
      "training_step_time": 9.892433881759644
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 3.1345720291137695,
      "step": 5192
    },
    {
      "dynamics_loss_avg": 0.06679263524711132,
      "epoch": 1.9805908203125e-05,
      "step": 5192
    },
    {
      "action_loss_avg": 0.027350246161222457,
      "epoch": 1.9805908203125e-05,
      "step": 5192
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 5192,
      "training_step_time": 9.90293836593628
    },
    {
      "epoch": 1.9805908203125e-05,
      "model_forward_time": 3.4011218547821045,
      "step": 5192
    },
    {
      "epoch": 1.9805908203125e-05,
      "step": 5192,
      "training_step_time": 10.262866497039795
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 3.166165828704834,
      "step": 5193
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 5193,
      "training_step_time": 9.938884019851685
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 3.1430318355560303,
      "step": 5193
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 5193,
      "training_step_time": 9.915856122970581
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 3.13787579536438,
      "step": 5193
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 5193,
      "training_step_time": 9.907281875610352
    },
    {
      "epoch": 1.9809722900390624e-05,
      "model_forward_time": 3.140451669692993,
      "step": 5193
    },
    {
      "epoch": 1.9809722900390624e-05,
      "step": 5193,
      "training_step_time": 10.269954919815063
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 3.4636759757995605,
      "step": 5194
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 5194,
      "training_step_time": 10.233156681060791
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 3.1393673419952393,
      "step": 5194
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 5194,
      "training_step_time": 9.90323257446289
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 3.1345672607421875,
      "step": 5194
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 5194,
      "training_step_time": 9.902283430099487
    },
    {
      "epoch": 1.981353759765625e-05,
      "model_forward_time": 3.1396961212158203,
      "step": 5194
    },
    {
      "epoch": 1.981353759765625e-05,
      "step": 5194,
      "training_step_time": 10.301642656326294
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 3.1739587783813477,
      "step": 5195
    },
    {
      "dynamics_loss_avg": 0.06310635916888714,
      "epoch": 1.9817352294921876e-05,
      "step": 5195
    },
    {
      "action_loss_avg": 0.02292053001001477,
      "epoch": 1.9817352294921876e-05,
      "step": 5195
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 5195,
      "training_step_time": 9.940712690353394
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 3.4318673610687256,
      "step": 5195
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 5195,
      "training_step_time": 10.195826530456543
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 3.137500524520874,
      "step": 5195
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 5195,
      "training_step_time": 9.904410123825073
    },
    {
      "epoch": 1.9817352294921876e-05,
      "model_forward_time": 3.1430866718292236,
      "step": 5195
    },
    {
      "epoch": 1.9817352294921876e-05,
      "step": 5195,
      "training_step_time": 10.295237064361572
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 3.1758415699005127,
      "step": 5196
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 5196,
      "training_step_time": 9.95044469833374
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 3.1299831867218018,
      "step": 5196
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 5196,
      "training_step_time": 9.902285099029541
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 3.398533344268799,
      "step": 5196
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 5196,
      "training_step_time": 10.161222457885742
    },
    {
      "epoch": 1.98211669921875e-05,
      "model_forward_time": 3.1319096088409424,
      "step": 5196
    },
    {
      "epoch": 1.98211669921875e-05,
      "step": 5196,
      "training_step_time": 10.25407338142395
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 3.162874698638916,
      "step": 5197
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 5197,
      "training_step_time": 9.926916360855103
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 3.126924514770508,
      "step": 5197
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 5197,
      "training_step_time": 9.894613265991211
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 3.13685941696167,
      "step": 5197
    },
    {
      "dynamics_loss_avg": 0.0665315069258213,
      "epoch": 1.9824981689453125e-05,
      "step": 5197
    },
    {
      "action_loss_avg": 0.027603540010750292,
      "epoch": 1.9824981689453125e-05,
      "step": 5197
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 5197,
      "training_step_time": 9.903337240219116
    },
    {
      "epoch": 1.9824981689453125e-05,
      "model_forward_time": 3.4348256587982178,
      "step": 5197
    },
    {
      "epoch": 1.9824981689453125e-05,
      "step": 5197,
      "training_step_time": 10.291455030441284
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 3.1775972843170166,
      "step": 5198
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 5198,
      "training_step_time": 9.945710182189941
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 3.12679386138916,
      "step": 5198
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 5198,
      "training_step_time": 9.890339851379395
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 3.1350512504577637,
      "step": 5198
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 5198,
      "training_step_time": 9.89816164970398
    },
    {
      "epoch": 1.982879638671875e-05,
      "model_forward_time": 3.1309518814086914,
      "step": 5198
    },
    {
      "epoch": 1.982879638671875e-05,
      "step": 5198,
      "training_step_time": 10.312593460083008
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 3.170126438140869,
      "step": 5199
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 5199,
      "training_step_time": 10.214576721191406
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 3.132392644882202,
      "step": 5199
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 5199,
      "training_step_time": 9.892144441604614
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 3.1281864643096924,
      "step": 5199
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 5199,
      "training_step_time": 9.887314319610596
    },
    {
      "epoch": 1.9832611083984374e-05,
      "model_forward_time": 3.128878593444824,
      "step": 5199
    },
    {
      "epoch": 1.9832611083984374e-05,
      "step": 5199,
      "training_step_time": 10.281068086624146
    },
    {
      "epoch": 1.983642578125e-05,
      "grad_norm": 0.4474075138568878,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 5200
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 3.146944284439087,
      "step": 5200
    },
    {
      "dynamics_loss_avg": 0.06176755614578724,
      "epoch": 1.983642578125e-05,
      "step": 5200
    },
    {
      "action_loss_avg": 0.028138956427574156,
      "epoch": 1.983642578125e-05,
      "step": 5200
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 5200,
      "training_step_time": 9.91219162940979
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 3.385909080505371,
      "step": 5200
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 5200,
      "training_step_time": 10.145107507705688
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 3.1337692737579346,
      "step": 5200
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 5200,
      "training_step_time": 9.892727375030518
    },
    {
      "epoch": 1.983642578125e-05,
      "model_forward_time": 3.1320595741271973,
      "step": 5200
    },
    {
      "epoch": 1.983642578125e-05,
      "step": 5200,
      "training_step_time": 10.273925542831421
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 3.181462049484253,
      "step": 5201
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 5201,
      "training_step_time": 9.945615768432617
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 3.13244891166687,
      "step": 5201
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 5201,
      "training_step_time": 9.89570140838623
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 3.132516384124756,
      "step": 5201
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 5201,
      "training_step_time": 10.17825722694397
    },
    {
      "epoch": 1.9840240478515624e-05,
      "model_forward_time": 3.1355464458465576,
      "step": 5201
    },
    {
      "epoch": 1.9840240478515624e-05,
      "step": 5201,
      "training_step_time": 10.026997327804565
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 3.179544687271118,
      "step": 5202
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 5202,
      "training_step_time": 9.95088267326355
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 3.1430211067199707,
      "step": 5202
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 5202,
      "training_step_time": 9.914711236953735
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 3.1402013301849365,
      "step": 5202
    },
    {
      "dynamics_loss_avg": 0.056341746263206004,
      "epoch": 1.984405517578125e-05,
      "step": 5202
    },
    {
      "action_loss_avg": 0.024416669690981507,
      "epoch": 1.984405517578125e-05,
      "step": 5202
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 5202,
      "training_step_time": 9.915097713470459
    },
    {
      "epoch": 1.984405517578125e-05,
      "model_forward_time": 3.4014780521392822,
      "step": 5202
    },
    {
      "epoch": 1.984405517578125e-05,
      "step": 5202,
      "training_step_time": 10.263206005096436
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 3.1736605167388916,
      "step": 5203
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 5203,
      "training_step_time": 9.949462413787842
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 3.139479160308838,
      "step": 5203
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 5203,
      "training_step_time": 9.909879446029663
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 3.13407826423645,
      "step": 5203
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 5203,
      "training_step_time": 9.905336141586304
    },
    {
      "epoch": 1.9847869873046876e-05,
      "model_forward_time": 3.140070676803589,
      "step": 5203
    },
    {
      "epoch": 1.9847869873046876e-05,
      "step": 5203,
      "training_step_time": 10.029362440109253
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 3.177166700363159,
      "step": 5204
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 5204,
      "training_step_time": 10.202884912490845
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 3.134556293487549,
      "step": 5204
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 5204,
      "training_step_time": 9.901758909225464
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 3.1348228454589844,
      "step": 5204
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 5204,
      "training_step_time": 9.901389122009277
    },
    {
      "epoch": 1.98516845703125e-05,
      "model_forward_time": 3.131685256958008,
      "step": 5204
    },
    {
      "epoch": 1.98516845703125e-05,
      "step": 5204,
      "training_step_time": 10.277719020843506
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 3.159954071044922,
      "step": 5205
    },
    {
      "dynamics_loss_avg": 0.059668049961328504,
      "epoch": 1.9855499267578125e-05,
      "step": 5205
    },
    {
      "action_loss_avg": 0.022969696065410973,
      "epoch": 1.9855499267578125e-05,
      "step": 5205
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 5205,
      "training_step_time": 9.926199913024902
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 3.402348518371582,
      "step": 5205
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 5205,
      "training_step_time": 10.161730289459229
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 3.1260218620300293,
      "step": 5205
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 5205,
      "training_step_time": 9.883012056350708
    },
    {
      "epoch": 1.9855499267578125e-05,
      "model_forward_time": 3.1313767433166504,
      "step": 5205
    },
    {
      "epoch": 1.9855499267578125e-05,
      "step": 5205,
      "training_step_time": 10.282540559768677
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 3.170225143432617,
      "step": 5206
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 5206,
      "training_step_time": 9.930317401885986
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 3.13276743888855,
      "step": 5206
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 5206,
      "training_step_time": 9.891054391860962
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 3.3975987434387207,
      "step": 5206
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 5206,
      "training_step_time": 10.153208017349243
    },
    {
      "epoch": 1.985931396484375e-05,
      "model_forward_time": 3.1337149143218994,
      "step": 5206
    },
    {
      "epoch": 1.985931396484375e-05,
      "step": 5206,
      "training_step_time": 9.99561595916748
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 3.165015459060669,
      "step": 5207
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 5207,
      "training_step_time": 9.92356824874878
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 2.429520606994629,
      "step": 5207
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 5207,
      "training_step_time": 7.449352025985718
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 3.132754325866699,
      "step": 5207
    },
    {
      "dynamics_loss_avg": 0.048339783772826196,
      "epoch": 1.9863128662109374e-05,
      "step": 5207
    },
    {
      "action_loss_avg": 0.0196347844786942,
      "epoch": 1.9863128662109374e-05,
      "step": 5207
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 5207,
      "training_step_time": 9.895534753799438
    },
    {
      "epoch": 1.9863128662109374e-05,
      "model_forward_time": 3.131561040878296,
      "step": 5207
    },
    {
      "epoch": 1.9863128662109374e-05,
      "step": 5207,
      "training_step_time": 10.263839960098267
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 3.1640939712524414,
      "step": 5208
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 5208,
      "training_step_time": 9.925093650817871
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 3.13240385055542,
      "step": 5208
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 5208,
      "training_step_time": 9.892547607421875
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 3.141848564147949,
      "step": 5208
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 5208,
      "training_step_time": 9.9113130569458
    },
    {
      "epoch": 1.9866943359375e-05,
      "model_forward_time": 3.1382296085357666,
      "step": 5208
    },
    {
      "epoch": 1.9866943359375e-05,
      "step": 5208,
      "training_step_time": 10.279986381530762
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 3.172895908355713,
      "step": 5209
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 5209,
      "training_step_time": 10.203324556350708
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 3.1391191482543945,
      "step": 5209
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 5209,
      "training_step_time": 9.911160469055176
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 3.1434550285339355,
      "step": 5209
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 5209,
      "training_step_time": 9.91355037689209
    },
    {
      "epoch": 1.9870758056640624e-05,
      "model_forward_time": 3.1378631591796875,
      "step": 5209
    },
    {
      "epoch": 1.9870758056640624e-05,
      "step": 5209,
      "training_step_time": 10.302913904190063
    },
    {
      "epoch": 1.987457275390625e-05,
      "grad_norm": 0.4313672184944153,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 5210
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 3.1752164363861084,
      "step": 5210
    },
    {
      "dynamics_loss_avg": 0.06624171854928136,
      "epoch": 1.987457275390625e-05,
      "step": 5210
    },
    {
      "action_loss_avg": 0.024583957297727464,
      "epoch": 1.987457275390625e-05,
      "step": 5210
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 5210,
      "training_step_time": 9.950952529907227
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 3.1409201622009277,
      "step": 5210
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 5210,
      "training_step_time": 10.185718536376953
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 3.1404449939727783,
      "step": 5210
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 5210,
      "training_step_time": 9.907871961593628
    },
    {
      "epoch": 1.987457275390625e-05,
      "model_forward_time": 3.136718273162842,
      "step": 5210
    },
    {
      "epoch": 1.987457275390625e-05,
      "step": 5210,
      "training_step_time": 10.287001371383667
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 3.154021739959717,
      "step": 5211
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 5211,
      "training_step_time": 9.916657447814941
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 3.1354336738586426,
      "step": 5211
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 5211,
      "training_step_time": 9.900293827056885
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 3.4322593212127686,
      "step": 5211
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 5211,
      "training_step_time": 10.19188642501831
    },
    {
      "epoch": 1.9878387451171876e-05,
      "model_forward_time": 3.136826276779175,
      "step": 5211
    },
    {
      "epoch": 1.9878387451171876e-05,
      "step": 5211,
      "training_step_time": 10.01521611213684
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 3.162407636642456,
      "step": 5212
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 5212,
      "training_step_time": 9.928435325622559
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 3.1394526958465576,
      "step": 5212
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 5212,
      "training_step_time": 9.907571077346802
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 3.137559652328491,
      "step": 5212
    },
    {
      "dynamics_loss_avg": 0.053460993245244025,
      "epoch": 1.98822021484375e-05,
      "step": 5212
    },
    {
      "action_loss_avg": 0.030282063968479635,
      "epoch": 1.98822021484375e-05,
      "step": 5212
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 5212,
      "training_step_time": 9.912036180496216
    },
    {
      "epoch": 1.98822021484375e-05,
      "model_forward_time": 3.1426217555999756,
      "step": 5212
    },
    {
      "epoch": 1.98822021484375e-05,
      "step": 5212,
      "training_step_time": 10.293935060501099
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 3.1567962169647217,
      "step": 5213
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 5213,
      "training_step_time": 9.92829442024231
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 3.1412646770477295,
      "step": 5213
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 5213,
      "training_step_time": 9.912079095840454
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 3.138787031173706,
      "step": 5213
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 5213,
      "training_step_time": 9.906764268875122
    },
    {
      "epoch": 1.9886016845703125e-05,
      "model_forward_time": 3.146761655807495,
      "step": 5213
    },
    {
      "epoch": 1.9886016845703125e-05,
      "step": 5213,
      "training_step_time": 10.272434711456299
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 3.4326839447021484,
      "step": 5214
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 5214,
      "training_step_time": 10.204774379730225
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 3.1494128704071045,
      "step": 5214
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 5214,
      "training_step_time": 9.928277492523193
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 3.14526629447937,
      "step": 5214
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 5214,
      "training_step_time": 9.919250249862671
    },
    {
      "epoch": 1.988983154296875e-05,
      "model_forward_time": 3.1468422412872314,
      "step": 5214
    },
    {
      "epoch": 1.988983154296875e-05,
      "step": 5214,
      "training_step_time": 10.277595520019531
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 3.176896333694458,
      "step": 5215
    },
    {
      "dynamics_loss_avg": 0.043986551091074945,
      "epoch": 1.9893646240234374e-05,
      "step": 5215
    },
    {
      "action_loss_avg": 0.022926574456505477,
      "epoch": 1.9893646240234374e-05,
      "step": 5215
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 5215,
      "training_step_time": 9.972584247589111
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 3.438890218734741,
      "step": 5215
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 5215,
      "training_step_time": 10.220473051071167
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 3.1396143436431885,
      "step": 5215
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 5215,
      "training_step_time": 9.919123888015747
    },
    {
      "epoch": 1.9893646240234374e-05,
      "model_forward_time": 3.1476690769195557,
      "step": 5215
    },
    {
      "epoch": 1.9893646240234374e-05,
      "step": 5215,
      "training_step_time": 10.314299821853638
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 3.17773175239563,
      "step": 5216
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 5216,
      "training_step_time": 9.96083950996399
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 3.139854669570923,
      "step": 5216
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 5216,
      "training_step_time": 9.912646770477295
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 3.437777280807495,
      "step": 5216
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 5216,
      "training_step_time": 10.203217267990112
    },
    {
      "epoch": 1.98974609375e-05,
      "model_forward_time": 3.1360955238342285,
      "step": 5216
    },
    {
      "epoch": 1.98974609375e-05,
      "step": 5216,
      "training_step_time": 10.262751579284668
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 3.1821722984313965,
      "step": 5217
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 5217,
      "training_step_time": 9.960752248764038
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 3.1373775005340576,
      "step": 5217
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 5217,
      "training_step_time": 9.910507917404175
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 3.1373672485351562,
      "step": 5217
    },
    {
      "dynamics_loss_avg": 0.060747180879116056,
      "epoch": 1.9901275634765624e-05,
      "step": 5217
    },
    {
      "action_loss_avg": 0.02473991014994681,
      "epoch": 1.9901275634765624e-05,
      "step": 5217
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 5217,
      "training_step_time": 9.923622608184814
    },
    {
      "epoch": 1.9901275634765624e-05,
      "model_forward_time": 3.4308485984802246,
      "step": 5217
    },
    {
      "epoch": 1.9901275634765624e-05,
      "step": 5217,
      "training_step_time": 10.29642939567566
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 3.1764328479766846,
      "step": 5218
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 5218,
      "training_step_time": 9.953329086303711
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 3.140824794769287,
      "step": 5218
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 5218,
      "training_step_time": 9.917313814163208
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 3.146101951599121,
      "step": 5218
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 5218,
      "training_step_time": 9.936123371124268
    },
    {
      "epoch": 1.990509033203125e-05,
      "model_forward_time": 3.146662712097168,
      "step": 5218
    },
    {
      "epoch": 1.990509033203125e-05,
      "step": 5218,
      "training_step_time": 10.292293310165405
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 3.1945130825042725,
      "step": 5219
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 5219,
      "training_step_time": 10.291269063949585
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 2.436389684677124,
      "step": 5219
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 5219,
      "training_step_time": 7.466371059417725
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 3.1403720378875732,
      "step": 5219
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 5219,
      "training_step_time": 9.919248819351196
    },
    {
      "epoch": 1.9908905029296876e-05,
      "model_forward_time": 3.1386711597442627,
      "step": 5219
    },
    {
      "epoch": 1.9908905029296876e-05,
      "step": 5219,
      "training_step_time": 10.024785041809082
    },
    {
      "epoch": 1.99127197265625e-05,
      "grad_norm": 0.3587492108345032,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 5220
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 3.1844098567962646,
      "step": 5220
    },
    {
      "dynamics_loss_avg": 0.07199722193181515,
      "epoch": 1.99127197265625e-05,
      "step": 5220
    },
    {
      "action_loss_avg": 0.028046803642064334,
      "epoch": 1.99127197265625e-05,
      "step": 5220
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 5220,
      "training_step_time": 9.958233833312988
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 3.1386146545410156,
      "step": 5220
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 5220,
      "training_step_time": 9.90818738937378
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 3.4177889823913574,
      "step": 5220
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 5220,
      "training_step_time": 10.187139987945557
    },
    {
      "epoch": 1.99127197265625e-05,
      "model_forward_time": 3.133507013320923,
      "step": 5220
    },
    {
      "epoch": 1.99127197265625e-05,
      "step": 5220,
      "training_step_time": 10.228184223175049
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 3.168426513671875,
      "step": 5221
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 5221,
      "training_step_time": 9.944177865982056
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 3.147815227508545,
      "step": 5221
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 5221,
      "training_step_time": 9.927409648895264
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 3.1336262226104736,
      "step": 5221
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 5221,
      "training_step_time": 10.218005657196045
    },
    {
      "epoch": 1.9916534423828125e-05,
      "model_forward_time": 3.1344971656799316,
      "step": 5221
    },
    {
      "epoch": 1.9916534423828125e-05,
      "step": 5221,
      "training_step_time": 10.295052528381348
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 3.154911518096924,
      "step": 5222
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 5222,
      "training_step_time": 9.930836915969849
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 3.130108118057251,
      "step": 5222
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 5222,
      "training_step_time": 9.897143125534058
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 3.127086877822876,
      "step": 5222
    },
    {
      "dynamics_loss_avg": 0.05982623845338821,
      "epoch": 1.992034912109375e-05,
      "step": 5222
    },
    {
      "action_loss_avg": 0.026233557052910327,
      "epoch": 1.992034912109375e-05,
      "step": 5222
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 5222,
      "training_step_time": 9.902322053909302
    },
    {
      "epoch": 1.992034912109375e-05,
      "model_forward_time": 3.3891077041625977,
      "step": 5222
    },
    {
      "epoch": 1.992034912109375e-05,
      "step": 5222,
      "training_step_time": 10.249955177307129
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 3.1701903343200684,
      "step": 5223
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 5223,
      "training_step_time": 9.939457654953003
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 3.129634141921997,
      "step": 5223
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 5223,
      "training_step_time": 9.893128156661987
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 3.124495267868042,
      "step": 5223
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 5223,
      "training_step_time": 9.897236824035645
    },
    {
      "epoch": 1.9924163818359374e-05,
      "model_forward_time": 3.1355316638946533,
      "step": 5223
    },
    {
      "epoch": 1.9924163818359374e-05,
      "step": 5223,
      "training_step_time": 10.026158094406128
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 3.433607816696167,
      "step": 5224
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 5224,
      "training_step_time": 10.209733247756958
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 3.135852813720703,
      "step": 5224
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 5224,
      "training_step_time": 9.913492441177368
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 3.1313889026641846,
      "step": 5224
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 5224,
      "training_step_time": 9.905458450317383
    },
    {
      "epoch": 1.9927978515625e-05,
      "model_forward_time": 3.1383421421051025,
      "step": 5224
    },
    {
      "epoch": 1.9927978515625e-05,
      "step": 5224,
      "training_step_time": 10.310325860977173
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 3.1760551929473877,
      "step": 5225
    },
    {
      "dynamics_loss_avg": 0.054599393531680106,
      "epoch": 1.9931793212890624e-05,
      "step": 5225
    },
    {
      "action_loss_avg": 0.025924880476668476,
      "epoch": 1.9931793212890624e-05,
      "step": 5225
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 5225,
      "training_step_time": 9.961650133132935
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 3.4275224208831787,
      "step": 5225
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 5225,
      "training_step_time": 10.203632354736328
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 3.143298625946045,
      "step": 5225
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 5225,
      "training_step_time": 9.925842046737671
    },
    {
      "epoch": 1.9931793212890624e-05,
      "model_forward_time": 3.1456472873687744,
      "step": 5225
    },
    {
      "epoch": 1.9931793212890624e-05,
      "step": 5225,
      "training_step_time": 10.023134708404541
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 3.1779656410217285,
      "step": 5226
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 5226,
      "training_step_time": 9.948956489562988
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 3.1408114433288574,
      "step": 5226
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 5226,
      "training_step_time": 9.912402153015137
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 3.410459280014038,
      "step": 5226
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 5226,
      "training_step_time": 10.189695835113525
    },
    {
      "epoch": 1.993560791015625e-05,
      "model_forward_time": 3.145745277404785,
      "step": 5226
    },
    {
      "epoch": 1.993560791015625e-05,
      "step": 5226,
      "training_step_time": 10.294828653335571
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 3.1909821033477783,
      "step": 5227
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 5227,
      "training_step_time": 9.968343019485474
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 3.138742685317993,
      "step": 5227
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 5227,
      "training_step_time": 9.904780626296997
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 3.1396689414978027,
      "step": 5227
    },
    {
      "dynamics_loss_avg": 0.06455965247005224,
      "epoch": 1.9939422607421876e-05,
      "step": 5227
    },
    {
      "action_loss_avg": 0.02602881216444075,
      "epoch": 1.9939422607421876e-05,
      "step": 5227
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 5227,
      "training_step_time": 9.91037893295288
    },
    {
      "epoch": 1.9939422607421876e-05,
      "model_forward_time": 3.4603569507598877,
      "step": 5227
    },
    {
      "epoch": 1.9939422607421876e-05,
      "step": 5227,
      "training_step_time": 10.319747686386108
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 3.1795742511749268,
      "step": 5228
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 5228,
      "training_step_time": 9.95335602760315
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 3.136573553085327,
      "step": 5228
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 5228,
      "training_step_time": 9.91161060333252
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 3.1407666206359863,
      "step": 5228
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 5228,
      "training_step_time": 9.91577959060669
    },
    {
      "epoch": 1.99432373046875e-05,
      "model_forward_time": 3.1426479816436768,
      "step": 5228
    },
    {
      "epoch": 1.99432373046875e-05,
      "step": 5228,
      "training_step_time": 10.016985177993774
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 3.466930389404297,
      "step": 5229
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 5229,
      "training_step_time": 10.23753547668457
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 3.1353635787963867,
      "step": 5229
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 5229,
      "training_step_time": 9.909511804580688
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 3.1362125873565674,
      "step": 5229
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 5229,
      "training_step_time": 9.915775299072266
    },
    {
      "epoch": 1.9947052001953125e-05,
      "model_forward_time": 3.136380434036255,
      "step": 5229
    },
    {
      "epoch": 1.9947052001953125e-05,
      "step": 5229,
      "training_step_time": 10.269574642181396
    },
    {
      "epoch": 1.995086669921875e-05,
      "grad_norm": 0.4652932584285736,
      "learning_rate": 1e-05,
      "loss": 0.0854,
      "step": 5230
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 3.163078784942627,
      "step": 5230
    },
    {
      "dynamics_loss_avg": 0.05911008585244417,
      "epoch": 1.995086669921875e-05,
      "step": 5230
    },
    {
      "action_loss_avg": 0.023897820524871348,
      "epoch": 1.995086669921875e-05,
      "step": 5230
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 5230,
      "training_step_time": 9.94516110420227
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 3.396670341491699,
      "step": 5230
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 5230,
      "training_step_time": 10.178072929382324
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 3.1425414085388184,
      "step": 5230
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 5230,
      "training_step_time": 9.925790548324585
    },
    {
      "epoch": 1.995086669921875e-05,
      "model_forward_time": 3.139477491378784,
      "step": 5230
    },
    {
      "epoch": 1.995086669921875e-05,
      "step": 5230,
      "training_step_time": 10.026114702224731
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 3.147536039352417,
      "step": 5231
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 5231,
      "training_step_time": 9.934632062911987
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 3.140564203262329,
      "step": 5231
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 5231,
      "training_step_time": 9.917500734329224
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 2.6910693645477295,
      "step": 5231
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 5231,
      "training_step_time": 7.722467660903931
    },
    {
      "epoch": 1.9954681396484375e-05,
      "model_forward_time": 3.130201816558838,
      "step": 5231
    },
    {
      "epoch": 1.9954681396484375e-05,
      "step": 5231,
      "training_step_time": 10.303519248962402
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 3.1691620349884033,
      "step": 5232
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 5232,
      "training_step_time": 9.949724912643433
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 3.1347310543060303,
      "step": 5232
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 5232,
      "training_step_time": 9.907565355300903
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 3.137258529663086,
      "step": 5232
    },
    {
      "dynamics_loss_avg": 0.05441531278192997,
      "epoch": 1.995849609375e-05,
      "step": 5232
    },
    {
      "action_loss_avg": 0.02725113248452544,
      "epoch": 1.995849609375e-05,
      "step": 5232
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 5232,
      "training_step_time": 9.91362714767456
    },
    {
      "epoch": 1.995849609375e-05,
      "model_forward_time": 3.1354682445526123,
      "step": 5232
    },
    {
      "epoch": 1.995849609375e-05,
      "step": 5232,
      "training_step_time": 10.283042907714844
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 3.150162696838379,
      "step": 5233
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 5233,
      "training_step_time": 9.922415256500244
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 3.1303389072418213,
      "step": 5233
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 5233,
      "training_step_time": 9.902398586273193
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 3.1283156871795654,
      "step": 5233
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 5233,
      "training_step_time": 9.898915529251099
    },
    {
      "epoch": 1.9962310791015624e-05,
      "model_forward_time": 2.4274542331695557,
      "step": 5233
    },
    {
      "epoch": 1.9962310791015624e-05,
      "step": 5233,
      "training_step_time": 10.0136137008667
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 3.4263381958007812,
      "step": 5234
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 5234,
      "training_step_time": 10.19163703918457
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 3.1292240619659424,
      "step": 5234
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 5234,
      "training_step_time": 9.900566339492798
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 3.1301000118255615,
      "step": 5234
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 5234,
      "training_step_time": 9.906070232391357
    },
    {
      "epoch": 1.996612548828125e-05,
      "model_forward_time": 3.133446455001831,
      "step": 5234
    },
    {
      "epoch": 1.996612548828125e-05,
      "step": 5234,
      "training_step_time": 10.01774525642395
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 3.14406418800354,
      "step": 5235
    },
    {
      "dynamics_loss_avg": 0.07432625722140074,
      "epoch": 1.9969940185546876e-05,
      "step": 5235
    },
    {
      "action_loss_avg": 0.03819749355316162,
      "epoch": 1.9969940185546876e-05,
      "step": 5235
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 5235,
      "training_step_time": 9.923046827316284
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 3.1346044540405273,
      "step": 5235
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 5235,
      "training_step_time": 10.2099928855896
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 3.14009165763855,
      "step": 5235
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 5235,
      "training_step_time": 9.914986371994019
    },
    {
      "epoch": 1.9969940185546876e-05,
      "model_forward_time": 3.1400070190429688,
      "step": 5235
    },
    {
      "epoch": 1.9969940185546876e-05,
      "step": 5235,
      "training_step_time": 10.300129652023315
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 3.1545677185058594,
      "step": 5236
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 5236,
      "training_step_time": 9.935246706008911
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 3.1380555629730225,
      "step": 5236
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 5236,
      "training_step_time": 9.915103435516357
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 3.4005002975463867,
      "step": 5236
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 5236,
      "training_step_time": 10.175967931747437
    },
    {
      "epoch": 1.99737548828125e-05,
      "model_forward_time": 2.4298174381256104,
      "step": 5236
    },
    {
      "epoch": 1.99737548828125e-05,
      "step": 5236,
      "training_step_time": 10.267317295074463
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 3.1651339530944824,
      "step": 5237
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 5237,
      "training_step_time": 9.939016342163086
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 3.134077548980713,
      "step": 5237
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 5237,
      "training_step_time": 9.905490636825562
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 3.1331140995025635,
      "step": 5237
    },
    {
      "dynamics_loss_avg": 0.06017885790206492,
      "epoch": 1.9977569580078125e-05,
      "step": 5237
    },
    {
      "action_loss_avg": 0.028005007770843804,
      "epoch": 1.9977569580078125e-05,
      "step": 5237
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 5237,
      "training_step_time": 9.904983758926392
    },
    {
      "epoch": 1.9977569580078125e-05,
      "model_forward_time": 3.129451274871826,
      "step": 5237
    },
    {
      "epoch": 1.9977569580078125e-05,
      "step": 5237,
      "training_step_time": 10.29019546508789
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 3.1697912216186523,
      "step": 5238
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 5238,
      "training_step_time": 9.943154096603394
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 3.130593776702881,
      "step": 5238
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 5238,
      "training_step_time": 9.897370100021362
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 3.1303231716156006,
      "step": 5238
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 5238,
      "training_step_time": 9.89674687385559
    },
    {
      "epoch": 1.998138427734375e-05,
      "model_forward_time": 3.1266167163848877,
      "step": 5238
    },
    {
      "epoch": 1.998138427734375e-05,
      "step": 5238,
      "training_step_time": 10.010934591293335
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 3.169022560119629,
      "step": 5239
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 5239,
      "training_step_time": 10.22188663482666
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 3.129934310913086,
      "step": 5239
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 5239,
      "training_step_time": 9.900742769241333
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 3.1364989280700684,
      "step": 5239
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 5239,
      "training_step_time": 9.909172773361206
    },
    {
      "epoch": 1.9985198974609375e-05,
      "model_forward_time": 3.129467010498047,
      "step": 5239
    },
    {
      "epoch": 1.9985198974609375e-05,
      "step": 5239,
      "training_step_time": 10.012615203857422
    },
    {
      "epoch": 1.9989013671875e-05,
      "grad_norm": 0.6932826042175293,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 5240
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 3.1399359703063965,
      "step": 5240
    },
    {
      "dynamics_loss_avg": 0.04572101179510355,
      "epoch": 1.9989013671875e-05,
      "step": 5240
    },
    {
      "action_loss_avg": 0.020916993916034698,
      "epoch": 1.9989013671875e-05,
      "step": 5240
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 5240,
      "training_step_time": 9.915908098220825
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 3.396480083465576,
      "step": 5240
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 5240,
      "training_step_time": 10.166981220245361
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 3.1363332271575928,
      "step": 5240
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 5240,
      "training_step_time": 9.907627582550049
    },
    {
      "epoch": 1.9989013671875e-05,
      "model_forward_time": 3.135709285736084,
      "step": 5240
    },
    {
      "epoch": 1.9989013671875e-05,
      "step": 5240,
      "training_step_time": 10.297942399978638
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 3.1678154468536377,
      "step": 5241
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 5241,
      "training_step_time": 9.947519302368164
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 3.142350196838379,
      "step": 5241
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 5241,
      "training_step_time": 9.920549392700195
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 3.1408984661102295,
      "step": 5241
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 5241,
      "training_step_time": 10.196106433868408
    },
    {
      "epoch": 1.9992828369140624e-05,
      "model_forward_time": 3.139286518096924,
      "step": 5241
    },
    {
      "epoch": 1.9992828369140624e-05,
      "step": 5241,
      "training_step_time": 10.281262636184692
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 3.1763815879821777,
      "step": 5242
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 5242,
      "training_step_time": 9.961698293685913
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 3.1445109844207764,
      "step": 5242
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 5242,
      "training_step_time": 9.927335739135742
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 3.1363179683685303,
      "step": 5242
    },
    {
      "dynamics_loss_avg": 0.056953944638371466,
      "epoch": 1.999664306640625e-05,
      "step": 5242
    },
    {
      "action_loss_avg": 0.028136267978698015,
      "epoch": 1.999664306640625e-05,
      "step": 5242
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 5242,
      "training_step_time": 9.916474103927612
    },
    {
      "epoch": 1.999664306640625e-05,
      "model_forward_time": 3.4005820751190186,
      "step": 5242
    },
    {
      "epoch": 1.999664306640625e-05,
      "step": 5242,
      "training_step_time": 10.273311376571655
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 3.1837522983551025,
      "step": 5243
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 5243,
      "training_step_time": 9.959171056747437
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 3.1382038593292236,
      "step": 5243
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 5243,
      "training_step_time": 9.913328886032104
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 3.1359705924987793,
      "step": 5243
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 5243,
      "training_step_time": 9.916260719299316
    },
    {
      "epoch": 2.0000457763671876e-05,
      "model_forward_time": 3.1452414989471436,
      "step": 5243
    },
    {
      "epoch": 2.0000457763671876e-05,
      "step": 5243,
      "training_step_time": 10.019074440002441
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 3.16973614692688,
      "step": 5244
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 5244,
      "training_step_time": 10.215998888015747
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 3.1415305137634277,
      "step": 5244
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 5244,
      "training_step_time": 9.923652172088623
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 3.1419737339019775,
      "step": 5244
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 5244,
      "training_step_time": 9.922628164291382
    },
    {
      "epoch": 2.00042724609375e-05,
      "model_forward_time": 3.1376023292541504,
      "step": 5244
    },
    {
      "epoch": 2.00042724609375e-05,
      "step": 5244,
      "training_step_time": 10.27107286453247
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 3.17022705078125,
      "step": 5245
    },
    {
      "dynamics_loss_avg": 0.06119819162413478,
      "epoch": 2.0008087158203125e-05,
      "step": 5245
    },
    {
      "action_loss_avg": 0.025250728521496058,
      "epoch": 2.0008087158203125e-05,
      "step": 5245
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 5245,
      "training_step_time": 9.949232578277588
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 3.418536424636841,
      "step": 5245
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 5245,
      "training_step_time": 10.188308715820312
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 3.1345090866088867,
      "step": 5245
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 5245,
      "training_step_time": 9.90130066871643
    },
    {
      "epoch": 2.0008087158203125e-05,
      "model_forward_time": 3.1374330520629883,
      "step": 5245
    },
    {
      "epoch": 2.0008087158203125e-05,
      "step": 5245,
      "training_step_time": 10.017896890640259
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 3.1559224128723145,
      "step": 5246
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 5246,
      "training_step_time": 9.921013593673706
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 3.1301536560058594,
      "step": 5246
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 5246,
      "training_step_time": 9.891547203063965
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 3.4010682106018066,
      "step": 5246
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 5246,
      "training_step_time": 10.17050552368164
    },
    {
      "epoch": 2.001190185546875e-05,
      "model_forward_time": 3.1344902515411377,
      "step": 5246
    },
    {
      "epoch": 2.001190185546875e-05,
      "step": 5246,
      "training_step_time": 10.28921627998352
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 3.163010358810425,
      "step": 5247
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 5247,
      "training_step_time": 9.929442882537842
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 3.130557060241699,
      "step": 5247
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 5247,
      "training_step_time": 9.896557569503784
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 3.1350276470184326,
      "step": 5247
    },
    {
      "dynamics_loss_avg": 0.07600607685744762,
      "epoch": 2.0015716552734375e-05,
      "step": 5247
    },
    {
      "action_loss_avg": 0.03755202507600188,
      "epoch": 2.0015716552734375e-05,
      "step": 5247
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 5247,
      "training_step_time": 9.905627012252808
    },
    {
      "epoch": 2.0015716552734375e-05,
      "model_forward_time": 3.4319539070129395,
      "step": 5247
    },
    {
      "epoch": 2.0015716552734375e-05,
      "step": 5247,
      "training_step_time": 10.29514455795288
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 3.1827709674835205,
      "step": 5248
    },
    {
      "epoch": 2.001953125e-05,
      "step": 5248,
      "training_step_time": 9.95500135421753
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 3.1319854259490967,
      "step": 5248
    },
    {
      "epoch": 2.001953125e-05,
      "step": 5248,
      "training_step_time": 9.904927968978882
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 3.1400506496429443,
      "step": 5248
    },
    {
      "epoch": 2.001953125e-05,
      "step": 5248,
      "training_step_time": 9.91561508178711
    },
    {
      "epoch": 2.001953125e-05,
      "model_forward_time": 3.1361589431762695,
      "step": 5248
    },
    {
      "epoch": 2.001953125e-05,
      "step": 5248,
      "training_step_time": 10.301061868667603
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 3.5225634574890137,
      "step": 5249
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 5249,
      "training_step_time": 10.293885231018066
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 3.142202377319336,
      "step": 5249
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 5249,
      "training_step_time": 9.9144127368927
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 3.1347861289978027,
      "step": 5249
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 5249,
      "training_step_time": 9.904565334320068
    },
    {
      "epoch": 2.0023345947265624e-05,
      "model_forward_time": 3.1337482929229736,
      "step": 5249
    },
    {
      "epoch": 2.0023345947265624e-05,
      "step": 5249,
      "training_step_time": 10.00431203842163
    },
    {
      "epoch": 2.002716064453125e-05,
      "grad_norm": 0.6763427257537842,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 5250
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 3.178347587585449,
      "step": 5250
    },
    {
      "dynamics_loss_avg": 0.06769939623773098,
      "epoch": 2.002716064453125e-05,
      "step": 5250
    },
    {
      "action_loss_avg": 0.028029162902384996,
      "epoch": 2.002716064453125e-05,
      "step": 5250
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 5250,
      "training_step_time": 9.962462186813354
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 3.390393018722534,
      "step": 5250
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 5250,
      "training_step_time": 10.164496183395386
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 3.1385695934295654,
      "step": 5250
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 5250,
      "training_step_time": 9.913979530334473
    },
    {
      "epoch": 2.002716064453125e-05,
      "model_forward_time": 3.137882947921753,
      "step": 5250
    },
    {
      "epoch": 2.002716064453125e-05,
      "step": 5250,
      "training_step_time": 10.028732061386108
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 3.1797053813934326,
      "step": 5251
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 5251,
      "training_step_time": 9.953616380691528
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 3.137651205062866,
      "step": 5251
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 5251,
      "training_step_time": 9.909044027328491
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 3.4097509384155273,
      "step": 5251
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 5251,
      "training_step_time": 10.179964303970337
    },
    {
      "epoch": 2.0030975341796876e-05,
      "model_forward_time": 3.136390209197998,
      "step": 5251
    },
    {
      "epoch": 2.0030975341796876e-05,
      "step": 5251,
      "training_step_time": 10.290419101715088
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 3.1756386756896973,
      "step": 5252
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 5252,
      "training_step_time": 9.952860355377197
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 3.139338254928589,
      "step": 5252
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 5252,
      "training_step_time": 9.913199424743652
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 3.1390881538391113,
      "step": 5252
    },
    {
      "dynamics_loss_avg": 0.05711725130677223,
      "epoch": 2.00347900390625e-05,
      "step": 5252
    },
    {
      "action_loss_avg": 0.023090011905878782,
      "epoch": 2.00347900390625e-05,
      "step": 5252
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 5252,
      "training_step_time": 9.913027286529541
    },
    {
      "epoch": 2.00347900390625e-05,
      "model_forward_time": 3.3964340686798096,
      "step": 5252
    },
    {
      "epoch": 2.00347900390625e-05,
      "step": 5252,
      "training_step_time": 10.263275384902954
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 3.163844108581543,
      "step": 5253
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 5253,
      "training_step_time": 9.929818630218506
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 3.137218952178955,
      "step": 5253
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 5253,
      "training_step_time": 9.904650449752808
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 3.1364729404449463,
      "step": 5253
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 5253,
      "training_step_time": 9.90700626373291
    },
    {
      "epoch": 2.0038604736328125e-05,
      "model_forward_time": 3.139054536819458,
      "step": 5253
    },
    {
      "epoch": 2.0038604736328125e-05,
      "step": 5253,
      "training_step_time": 10.31534457206726
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 3.421797752380371,
      "step": 5254
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 5254,
      "training_step_time": 10.193294525146484
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 3.1450605392456055,
      "step": 5254
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 5254,
      "training_step_time": 9.917216300964355
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 3.1403095722198486,
      "step": 5254
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 5254,
      "training_step_time": 9.914561748504639
    },
    {
      "epoch": 2.004241943359375e-05,
      "model_forward_time": 3.1395645141601562,
      "step": 5254
    },
    {
      "epoch": 2.004241943359375e-05,
      "step": 5254,
      "training_step_time": 10.27783727645874
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 3.1670615673065186,
      "step": 5255
    },
    {
      "dynamics_loss_avg": 0.06367694055661559,
      "epoch": 2.0046234130859375e-05,
      "step": 5255
    },
    {
      "action_loss_avg": 0.02802242466714233,
      "epoch": 2.0046234130859375e-05,
      "step": 5255
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 5255,
      "training_step_time": 9.949384927749634
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 3.426046848297119,
      "step": 5255
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 5255,
      "training_step_time": 10.201459407806396
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 3.1414847373962402,
      "step": 5255
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 5255,
      "training_step_time": 9.918071031570435
    },
    {
      "epoch": 2.0046234130859375e-05,
      "model_forward_time": 3.140413284301758,
      "step": 5255
    },
    {
      "epoch": 2.0046234130859375e-05,
      "step": 5255,
      "training_step_time": 10.298501253128052
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 3.17370343208313,
      "step": 5256
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 5256,
      "training_step_time": 9.949801921844482
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 3.1366920471191406,
      "step": 5256
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 5256,
      "training_step_time": 9.910717010498047
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 3.404672622680664,
      "step": 5256
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 5256,
      "training_step_time": 10.17928171157837
    },
    {
      "epoch": 2.0050048828125e-05,
      "model_forward_time": 3.1386730670928955,
      "step": 5256
    },
    {
      "epoch": 2.0050048828125e-05,
      "step": 5256,
      "training_step_time": 10.017848491668701
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 3.1911935806274414,
      "step": 5257
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 5257,
      "training_step_time": 9.968471765518188
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 3.1381688117980957,
      "step": 5257
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 5257,
      "training_step_time": 9.91315507888794
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 3.1364428997039795,
      "step": 5257
    },
    {
      "dynamics_loss_avg": 0.0525188148021698,
      "epoch": 2.0053863525390624e-05,
      "step": 5257
    },
    {
      "action_loss_avg": 0.024634077865630388,
      "epoch": 2.0053863525390624e-05,
      "step": 5257
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 5257,
      "training_step_time": 9.9143648147583
    },
    {
      "epoch": 2.0053863525390624e-05,
      "model_forward_time": 3.4393835067749023,
      "step": 5257
    },
    {
      "epoch": 2.0053863525390624e-05,
      "step": 5257,
      "training_step_time": 10.309387445449829
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 3.201686143875122,
      "step": 5258
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 5258,
      "training_step_time": 9.974783182144165
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 3.1422479152679443,
      "step": 5258
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 5258,
      "training_step_time": 9.913637399673462
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 3.150946855545044,
      "step": 5258
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 5258,
      "training_step_time": 9.92420220375061
    },
    {
      "epoch": 2.005767822265625e-05,
      "model_forward_time": 3.142230987548828,
      "step": 5258
    },
    {
      "epoch": 2.005767822265625e-05,
      "step": 5258,
      "training_step_time": 10.31211233139038
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 3.187140703201294,
      "step": 5259
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 5259,
      "training_step_time": 10.242588996887207
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 3.142263412475586,
      "step": 5259
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 5259,
      "training_step_time": 9.916510820388794
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 3.1460959911346436,
      "step": 5259
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 5259,
      "training_step_time": 9.918192863464355
    },
    {
      "epoch": 2.0061492919921876e-05,
      "model_forward_time": 3.1460771560668945,
      "step": 5259
    },
    {
      "epoch": 2.0061492919921876e-05,
      "step": 5259,
      "training_step_time": 10.287560939788818
    },
    {
      "epoch": 2.00653076171875e-05,
      "grad_norm": 0.6310799717903137,
      "learning_rate": 1e-05,
      "loss": 0.0877,
      "step": 5260
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 3.1637563705444336,
      "step": 5260
    },
    {
      "dynamics_loss_avg": 0.06812973059713841,
      "epoch": 2.00653076171875e-05,
      "step": 5260
    },
    {
      "action_loss_avg": 0.029565119929611684,
      "epoch": 2.00653076171875e-05,
      "step": 5260
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 5260,
      "training_step_time": 9.943171262741089
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 3.408890724182129,
      "step": 5260
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 5260,
      "training_step_time": 10.17888855934143
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 3.143073558807373,
      "step": 5260
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 5260,
      "training_step_time": 9.911850452423096
    },
    {
      "epoch": 2.00653076171875e-05,
      "model_forward_time": 3.139984607696533,
      "step": 5260
    },
    {
      "epoch": 2.00653076171875e-05,
      "step": 5260,
      "training_step_time": 10.016830921173096
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 3.1634905338287354,
      "step": 5261
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 5261,
      "training_step_time": 9.93053913116455
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 3.1369662284851074,
      "step": 5261
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 5261,
      "training_step_time": 9.904114961624146
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 3.134430170059204,
      "step": 5261
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 5261,
      "training_step_time": 10.18848729133606
    },
    {
      "epoch": 2.0069122314453125e-05,
      "model_forward_time": 3.1367664337158203,
      "step": 5261
    },
    {
      "epoch": 2.0069122314453125e-05,
      "step": 5261,
      "training_step_time": 10.02793836593628
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 3.173168420791626,
      "step": 5262
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 5262,
      "training_step_time": 9.949181318283081
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 3.146430492401123,
      "step": 5262
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 5262,
      "training_step_time": 9.920125484466553
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 3.1427128314971924,
      "step": 5262
    },
    {
      "dynamics_loss_avg": 0.07409940063953399,
      "epoch": 2.007293701171875e-05,
      "step": 5262
    },
    {
      "action_loss_avg": 0.0369907995685935,
      "epoch": 2.007293701171875e-05,
      "step": 5262
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 5262,
      "training_step_time": 9.916688680648804
    },
    {
      "epoch": 2.007293701171875e-05,
      "model_forward_time": 3.426996946334839,
      "step": 5262
    },
    {
      "epoch": 2.007293701171875e-05,
      "step": 5262,
      "training_step_time": 10.293841361999512
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 3.1865739822387695,
      "step": 5263
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 5263,
      "training_step_time": 9.961720943450928
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 3.146582841873169,
      "step": 5263
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 5263,
      "training_step_time": 9.921141147613525
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 3.149207592010498,
      "step": 5263
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 5263,
      "training_step_time": 9.925417423248291
    },
    {
      "epoch": 2.0076751708984375e-05,
      "model_forward_time": 3.1506505012512207,
      "step": 5263
    },
    {
      "epoch": 2.0076751708984375e-05,
      "step": 5263,
      "training_step_time": 10.320327758789062
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 3.172426700592041,
      "step": 5264
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 5264,
      "training_step_time": 10.223520517349243
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 2.4398231506347656,
      "step": 5264
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 5264,
      "training_step_time": 7.4677605628967285
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 3.137120246887207,
      "step": 5264
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 5264,
      "training_step_time": 9.901482343673706
    },
    {
      "epoch": 2.008056640625e-05,
      "model_forward_time": 3.1431796550750732,
      "step": 5264
    },
    {
      "epoch": 2.008056640625e-05,
      "step": 5264,
      "training_step_time": 10.3191237449646
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 3.1731626987457275,
      "step": 5265
    },
    {
      "dynamics_loss_avg": 0.06365692541003228,
      "epoch": 2.0084381103515624e-05,
      "step": 5265
    },
    {
      "action_loss_avg": 0.026301947282627226,
      "epoch": 2.0084381103515624e-05,
      "step": 5265
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 5265,
      "training_step_time": 9.948037147521973
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 3.420888900756836,
      "step": 5265
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 5265,
      "training_step_time": 10.191100120544434
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 3.140789270401001,
      "step": 5265
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 5265,
      "training_step_time": 9.913771629333496
    },
    {
      "epoch": 2.0084381103515624e-05,
      "model_forward_time": 3.148966073989868,
      "step": 5265
    },
    {
      "epoch": 2.0084381103515624e-05,
      "step": 5265,
      "training_step_time": 10.029577732086182
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 3.1881277561187744,
      "step": 5266
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 5266,
      "training_step_time": 9.97409701347351
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 3.146653890609741,
      "step": 5266
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 5266,
      "training_step_time": 9.917992353439331
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 3.404677152633667,
      "step": 5266
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 5266,
      "training_step_time": 10.174377679824829
    },
    {
      "epoch": 2.008819580078125e-05,
      "model_forward_time": 3.1412789821624756,
      "step": 5266
    },
    {
      "epoch": 2.008819580078125e-05,
      "step": 5266,
      "training_step_time": 10.012269496917725
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 3.181596040725708,
      "step": 5267
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 5267,
      "training_step_time": 9.951437711715698
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 3.140814781188965,
      "step": 5267
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 5267,
      "training_step_time": 9.912712812423706
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 3.1436750888824463,
      "step": 5267
    },
    {
      "dynamics_loss_avg": 0.0573205528780818,
      "epoch": 2.0092010498046876e-05,
      "step": 5267
    },
    {
      "action_loss_avg": 0.023798966594040395,
      "epoch": 2.0092010498046876e-05,
      "step": 5267
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 5267,
      "training_step_time": 9.919109582901001
    },
    {
      "epoch": 2.0092010498046876e-05,
      "model_forward_time": 3.4244911670684814,
      "step": 5267
    },
    {
      "epoch": 2.0092010498046876e-05,
      "step": 5267,
      "training_step_time": 10.28934359550476
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 3.167051076889038,
      "step": 5268
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 5268,
      "training_step_time": 9.944080114364624
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 3.146389961242676,
      "step": 5268
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 5268,
      "training_step_time": 9.922082424163818
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 3.1518161296844482,
      "step": 5268
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 5268,
      "training_step_time": 9.9281747341156
    },
    {
      "epoch": 2.00958251953125e-05,
      "model_forward_time": 3.148401975631714,
      "step": 5268
    },
    {
      "epoch": 2.00958251953125e-05,
      "step": 5268,
      "training_step_time": 10.301412343978882
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 3.4741251468658447,
      "step": 5269
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 5269,
      "training_step_time": 10.253209829330444
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 3.1416847705841064,
      "step": 5269
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 5269,
      "training_step_time": 9.918978452682495
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 3.1488864421844482,
      "step": 5269
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 5269,
      "training_step_time": 9.923354387283325
    },
    {
      "epoch": 2.0099639892578125e-05,
      "model_forward_time": 3.145279884338379,
      "step": 5269
    },
    {
      "epoch": 2.0099639892578125e-05,
      "step": 5269,
      "training_step_time": 10.311914682388306
    },
    {
      "epoch": 2.010345458984375e-05,
      "grad_norm": 0.4965723752975464,
      "learning_rate": 1e-05,
      "loss": 0.0897,
      "step": 5270
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 3.1811563968658447,
      "step": 5270
    },
    {
      "dynamics_loss_avg": 0.061444173380732534,
      "epoch": 2.010345458984375e-05,
      "step": 5270
    },
    {
      "action_loss_avg": 0.02477856185287237,
      "epoch": 2.010345458984375e-05,
      "step": 5270
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 5270,
      "training_step_time": 9.965633392333984
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 3.422283172607422,
      "step": 5270
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 5270,
      "training_step_time": 10.196670770645142
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 3.1528139114379883,
      "step": 5270
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 5270,
      "training_step_time": 9.927480697631836
    },
    {
      "epoch": 2.010345458984375e-05,
      "model_forward_time": 3.143448829650879,
      "step": 5270
    },
    {
      "epoch": 2.010345458984375e-05,
      "step": 5270,
      "training_step_time": 10.281351804733276
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 2.48824143409729,
      "step": 5271
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 5271,
      "training_step_time": 7.517022609710693
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 3.137018918991089,
      "step": 5271
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 5271,
      "training_step_time": 9.908421754837036
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 3.4240834712982178,
      "step": 5271
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 5271,
      "training_step_time": 10.193335056304932
    },
    {
      "epoch": 2.0107269287109375e-05,
      "model_forward_time": 3.139887571334839,
      "step": 5271
    },
    {
      "epoch": 2.0107269287109375e-05,
      "step": 5271,
      "training_step_time": 10.010239839553833
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 3.176995038986206,
      "step": 5272
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 5272,
      "training_step_time": 9.949174880981445
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 3.141690254211426,
      "step": 5272
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 5272,
      "training_step_time": 9.915916442871094
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 3.141369581222534,
      "step": 5272
    },
    {
      "dynamics_loss_avg": 0.048503948375582695,
      "epoch": 2.0111083984375e-05,
      "step": 5272
    },
    {
      "action_loss_avg": 0.01814591311849654,
      "epoch": 2.0111083984375e-05,
      "step": 5272
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 5272,
      "training_step_time": 9.918257474899292
    },
    {
      "epoch": 2.0111083984375e-05,
      "model_forward_time": 3.1451966762542725,
      "step": 5272
    },
    {
      "epoch": 2.0111083984375e-05,
      "step": 5272,
      "training_step_time": 10.281437635421753
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 3.188929796218872,
      "step": 5273
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 5273,
      "training_step_time": 9.97667121887207
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 3.145456552505493,
      "step": 5273
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 5273,
      "training_step_time": 9.920829057693481
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 3.1362740993499756,
      "step": 5273
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 5273,
      "training_step_time": 9.906814813613892
    },
    {
      "epoch": 2.0114898681640624e-05,
      "model_forward_time": 3.1445248126983643,
      "step": 5273
    },
    {
      "epoch": 2.0114898681640624e-05,
      "step": 5273,
      "training_step_time": 10.333106756210327
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 3.45281720161438,
      "step": 5274
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 5274,
      "training_step_time": 10.222366333007812
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 3.141380786895752,
      "step": 5274
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 5274,
      "training_step_time": 9.907941818237305
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 3.1378133296966553,
      "step": 5274
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 5274,
      "training_step_time": 9.908458232879639
    },
    {
      "epoch": 2.011871337890625e-05,
      "model_forward_time": 3.1442620754241943,
      "step": 5274
    },
    {
      "epoch": 2.011871337890625e-05,
      "step": 5274,
      "training_step_time": 10.277066946029663
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 3.1879076957702637,
      "step": 5275
    },
    {
      "dynamics_loss_avg": 0.06214613914489746,
      "epoch": 2.0122528076171876e-05,
      "step": 5275
    },
    {
      "action_loss_avg": 0.0272075807210058,
      "epoch": 2.0122528076171876e-05,
      "step": 5275
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 5275,
      "training_step_time": 9.960037231445312
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 3.4282679557800293,
      "step": 5275
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 5275,
      "training_step_time": 10.19616150856018
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 3.139249563217163,
      "step": 5275
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 5275,
      "training_step_time": 9.90394902229309
    },
    {
      "epoch": 2.0122528076171876e-05,
      "model_forward_time": 3.137119770050049,
      "step": 5275
    },
    {
      "epoch": 2.0122528076171876e-05,
      "step": 5275,
      "training_step_time": 10.315483808517456
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 3.177950143814087,
      "step": 5276
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 5276,
      "training_step_time": 9.944440841674805
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 3.1347339153289795,
      "step": 5276
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 5276,
      "training_step_time": 9.898156881332397
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 3.3925812244415283,
      "step": 5276
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 5276,
      "training_step_time": 10.157586336135864
    },
    {
      "epoch": 2.01263427734375e-05,
      "model_forward_time": 3.134988307952881,
      "step": 5276
    },
    {
      "epoch": 2.01263427734375e-05,
      "step": 5276,
      "training_step_time": 10.006119966506958
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 3.17630672454834,
      "step": 5277
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 5277,
      "training_step_time": 9.942911386489868
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 3.1302661895751953,
      "step": 5277
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 5277,
      "training_step_time": 9.903856992721558
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 3.136824607849121,
      "step": 5277
    },
    {
      "dynamics_loss_avg": 0.06772829182446002,
      "epoch": 2.0130157470703126e-05,
      "step": 5277
    },
    {
      "action_loss_avg": 0.026219173986464738,
      "epoch": 2.0130157470703126e-05,
      "step": 5277
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 5277,
      "training_step_time": 9.911181688308716
    },
    {
      "epoch": 2.0130157470703126e-05,
      "model_forward_time": 3.418225049972534,
      "step": 5277
    },
    {
      "epoch": 2.0130157470703126e-05,
      "step": 5277,
      "training_step_time": 10.28259801864624
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 3.1703827381134033,
      "step": 5278
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 5278,
      "training_step_time": 9.941263198852539
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 3.136385917663574,
      "step": 5278
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 5278,
      "training_step_time": 9.913960456848145
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 3.1395676136016846,
      "step": 5278
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 5278,
      "training_step_time": 9.915305376052856
    },
    {
      "epoch": 2.013397216796875e-05,
      "model_forward_time": 3.1370582580566406,
      "step": 5278
    },
    {
      "epoch": 2.013397216796875e-05,
      "step": 5278,
      "training_step_time": 10.302637100219727
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 3.1705141067504883,
      "step": 5279
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 5279,
      "training_step_time": 10.25588846206665
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 3.133735418319702,
      "step": 5279
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 5279,
      "training_step_time": 9.900578498840332
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 3.1342523097991943,
      "step": 5279
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 5279,
      "training_step_time": 9.901865005493164
    },
    {
      "epoch": 2.0137786865234375e-05,
      "model_forward_time": 3.131070613861084,
      "step": 5279
    },
    {
      "epoch": 2.0137786865234375e-05,
      "step": 5279,
      "training_step_time": 10.237045049667358
    },
    {
      "epoch": 2.01416015625e-05,
      "grad_norm": 0.4826701283454895,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 5280
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 3.140953779220581,
      "step": 5280
    },
    {
      "dynamics_loss_avg": 0.059729671664536,
      "epoch": 2.01416015625e-05,
      "step": 5280
    },
    {
      "action_loss_avg": 0.0265321334823966,
      "epoch": 2.01416015625e-05,
      "step": 5280
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 5280,
      "training_step_time": 9.907903909683228
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 3.3906095027923584,
      "step": 5280
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 5280,
      "training_step_time": 10.152339220046997
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 3.134847402572632,
      "step": 5280
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 5280,
      "training_step_time": 9.897493839263916
    },
    {
      "epoch": 2.01416015625e-05,
      "model_forward_time": 2.428833484649658,
      "step": 5280
    },
    {
      "epoch": 2.01416015625e-05,
      "step": 5280,
      "training_step_time": 10.306942701339722
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 3.1786484718322754,
      "step": 5281
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 5281,
      "training_step_time": 9.942748546600342
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 3.132014274597168,
      "step": 5281
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 5281,
      "training_step_time": 9.895369529724121
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 3.1285243034362793,
      "step": 5281
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 5281,
      "training_step_time": 10.174646854400635
    },
    {
      "epoch": 2.0145416259765624e-05,
      "model_forward_time": 3.1305785179138184,
      "step": 5281
    },
    {
      "epoch": 2.0145416259765624e-05,
      "step": 5281,
      "training_step_time": 10.296464681625366
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 3.1746370792388916,
      "step": 5282
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 5282,
      "training_step_time": 9.94150686264038
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 3.133129358291626,
      "step": 5282
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 5282,
      "training_step_time": 9.906252145767212
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 3.1342644691467285,
      "step": 5282
    },
    {
      "dynamics_loss_avg": 0.0556335611268878,
      "epoch": 2.014923095703125e-05,
      "step": 5282
    },
    {
      "action_loss_avg": 0.022593457903712987,
      "epoch": 2.014923095703125e-05,
      "step": 5282
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 5282,
      "training_step_time": 9.904882192611694
    },
    {
      "epoch": 2.014923095703125e-05,
      "model_forward_time": 3.3848304748535156,
      "step": 5282
    },
    {
      "epoch": 2.014923095703125e-05,
      "step": 5282,
      "training_step_time": 10.249096393585205
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 3.191467046737671,
      "step": 5283
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 5283,
      "training_step_time": 9.97043490409851
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 3.1374363899230957,
      "step": 5283
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 5283,
      "training_step_time": 9.914783000946045
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 3.134330987930298,
      "step": 5283
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 5283,
      "training_step_time": 9.910802364349365
    },
    {
      "epoch": 2.0153045654296876e-05,
      "model_forward_time": 3.1340692043304443,
      "step": 5283
    },
    {
      "epoch": 2.0153045654296876e-05,
      "step": 5283,
      "training_step_time": 10.03015685081482
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 3.4835567474365234,
      "step": 5284
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 5284,
      "training_step_time": 10.25958514213562
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 3.1414101123809814,
      "step": 5284
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 5284,
      "training_step_time": 9.916697263717651
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 3.1368448734283447,
      "step": 5284
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 5284,
      "training_step_time": 9.91206669807434
    },
    {
      "epoch": 2.01568603515625e-05,
      "model_forward_time": 3.1379778385162354,
      "step": 5284
    },
    {
      "epoch": 2.01568603515625e-05,
      "step": 5284,
      "training_step_time": 10.30758285522461
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 3.179347276687622,
      "step": 5285
    },
    {
      "dynamics_loss_avg": 0.06231898888945579,
      "epoch": 2.0160675048828126e-05,
      "step": 5285
    },
    {
      "action_loss_avg": 0.024259870406240225,
      "epoch": 2.0160675048828126e-05,
      "step": 5285
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 5285,
      "training_step_time": 9.956125974655151
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 3.4127795696258545,
      "step": 5285
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 5285,
      "training_step_time": 10.185225486755371
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 3.1447339057922363,
      "step": 5285
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 5285,
      "training_step_time": 9.916322469711304
    },
    {
      "epoch": 2.0160675048828126e-05,
      "model_forward_time": 3.1400504112243652,
      "step": 5285
    },
    {
      "epoch": 2.0160675048828126e-05,
      "step": 5285,
      "training_step_time": 10.300892114639282
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 3.177718162536621,
      "step": 5286
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 5286,
      "training_step_time": 9.947339296340942
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 3.1309001445770264,
      "step": 5286
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 5286,
      "training_step_time": 9.895901918411255
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 3.385160207748413,
      "step": 5286
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 5286,
      "training_step_time": 10.153309106826782
    },
    {
      "epoch": 2.016448974609375e-05,
      "model_forward_time": 3.1310887336730957,
      "step": 5286
    },
    {
      "epoch": 2.016448974609375e-05,
      "step": 5286,
      "training_step_time": 10.271244049072266
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 3.1604232788085938,
      "step": 5287
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 5287,
      "training_step_time": 9.931710720062256
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 3.1319708824157715,
      "step": 5287
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 5287,
      "training_step_time": 9.899784088134766
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 3.1353063583374023,
      "step": 5287
    },
    {
      "dynamics_loss_avg": 0.060454710572957995,
      "epoch": 2.0168304443359375e-05,
      "step": 5287
    },
    {
      "action_loss_avg": 0.023738985974341632,
      "epoch": 2.0168304443359375e-05,
      "step": 5287
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 5287,
      "training_step_time": 9.909044742584229
    },
    {
      "epoch": 2.0168304443359375e-05,
      "model_forward_time": 3.409231424331665,
      "step": 5287
    },
    {
      "epoch": 2.0168304443359375e-05,
      "step": 5287,
      "training_step_time": 10.276782274246216
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 3.1732912063598633,
      "step": 5288
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 5288,
      "training_step_time": 9.94942831993103
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 3.135938882827759,
      "step": 5288
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 5288,
      "training_step_time": 9.90808367729187
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 3.1350033283233643,
      "step": 5288
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 5288,
      "training_step_time": 9.909161567687988
    },
    {
      "epoch": 2.0172119140625e-05,
      "model_forward_time": 3.1302716732025146,
      "step": 5288
    },
    {
      "epoch": 2.0172119140625e-05,
      "step": 5288,
      "training_step_time": 10.009924411773682
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 2.745284080505371,
      "step": 5289
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 5289,
      "training_step_time": 7.77403998374939
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 3.131042957305908,
      "step": 5289
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 5289,
      "training_step_time": 9.898235321044922
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 3.138282060623169,
      "step": 5289
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 5289,
      "training_step_time": 9.909651279449463
    },
    {
      "epoch": 2.0175933837890624e-05,
      "model_forward_time": 3.132361888885498,
      "step": 5289
    },
    {
      "epoch": 2.0175933837890624e-05,
      "step": 5289,
      "training_step_time": 10.290733337402344
    },
    {
      "epoch": 2.017974853515625e-05,
      "grad_norm": 0.5850456357002258,
      "learning_rate": 1e-05,
      "loss": 0.0857,
      "step": 5290
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 3.187551498413086,
      "step": 5290
    },
    {
      "dynamics_loss_avg": 0.0655764851719141,
      "epoch": 2.017974853515625e-05,
      "step": 5290
    },
    {
      "action_loss_avg": 0.02440662831068039,
      "epoch": 2.017974853515625e-05,
      "step": 5290
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 5290,
      "training_step_time": 9.96631669998169
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 3.1377506256103516,
      "step": 5290
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 5290,
      "training_step_time": 10.19739294052124
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 3.135941982269287,
      "step": 5290
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 5290,
      "training_step_time": 9.910682916641235
    },
    {
      "epoch": 2.017974853515625e-05,
      "model_forward_time": 3.13763427734375,
      "step": 5290
    },
    {
      "epoch": 2.017974853515625e-05,
      "step": 5290,
      "training_step_time": 10.26611042022705
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 3.1769211292266846,
      "step": 5291
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 5291,
      "training_step_time": 9.958573341369629
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 3.13838791847229,
      "step": 5291
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 5291,
      "training_step_time": 9.91851258277893
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 3.4172847270965576,
      "step": 5291
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 5291,
      "training_step_time": 10.192884922027588
    },
    {
      "epoch": 2.0183563232421876e-05,
      "model_forward_time": 3.1363108158111572,
      "step": 5291
    },
    {
      "epoch": 2.0183563232421876e-05,
      "step": 5291,
      "training_step_time": 10.313157320022583
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 3.182849168777466,
      "step": 5292
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 5292,
      "training_step_time": 9.963623046875
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 3.143325090408325,
      "step": 5292
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 5292,
      "training_step_time": 9.915225744247437
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 3.139608383178711,
      "step": 5292
    },
    {
      "dynamics_loss_avg": 0.06837530359625817,
      "epoch": 2.01873779296875e-05,
      "step": 5292
    },
    {
      "action_loss_avg": 0.030452779773622752,
      "epoch": 2.01873779296875e-05,
      "step": 5292
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 5292,
      "training_step_time": 9.911844968795776
    },
    {
      "epoch": 2.01873779296875e-05,
      "model_forward_time": 3.14790415763855,
      "step": 5292
    },
    {
      "epoch": 2.01873779296875e-05,
      "step": 5292,
      "training_step_time": 10.289549112319946
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 3.1806068420410156,
      "step": 5293
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 5293,
      "training_step_time": 9.953638792037964
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 3.1430740356445312,
      "step": 5293
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 5293,
      "training_step_time": 9.91387414932251
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 3.1380391120910645,
      "step": 5293
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 5293,
      "training_step_time": 9.905164241790771
    },
    {
      "epoch": 2.0191192626953126e-05,
      "model_forward_time": 3.142716646194458,
      "step": 5293
    },
    {
      "epoch": 2.0191192626953126e-05,
      "step": 5293,
      "training_step_time": 10.291636228561401
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 3.4289846420288086,
      "step": 5294
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 5294,
      "training_step_time": 10.198856592178345
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 3.1396450996398926,
      "step": 5294
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 5294,
      "training_step_time": 9.911202907562256
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 3.140038251876831,
      "step": 5294
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 5294,
      "training_step_time": 9.9119131565094
    },
    {
      "epoch": 2.019500732421875e-05,
      "model_forward_time": 3.1452770233154297,
      "step": 5294
    },
    {
      "epoch": 2.019500732421875e-05,
      "step": 5294,
      "training_step_time": 10.312062978744507
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 3.1852259635925293,
      "step": 5295
    },
    {
      "dynamics_loss_avg": 0.06485506109893321,
      "epoch": 2.0198822021484375e-05,
      "step": 5295
    },
    {
      "action_loss_avg": 0.02856023549102247,
      "epoch": 2.0198822021484375e-05,
      "step": 5295
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 5295,
      "training_step_time": 9.960831642150879
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 3.43768048286438,
      "step": 5295
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 5295,
      "training_step_time": 10.20647144317627
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 3.143772840499878,
      "step": 5295
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 5295,
      "training_step_time": 9.916022777557373
    },
    {
      "epoch": 2.0198822021484375e-05,
      "model_forward_time": 3.146458387374878,
      "step": 5295
    },
    {
      "epoch": 2.0198822021484375e-05,
      "step": 5295,
      "training_step_time": 10.271071910858154
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 3.2002251148223877,
      "step": 5296
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 5296,
      "training_step_time": 9.972185373306274
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 3.141948699951172,
      "step": 5296
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 5296,
      "training_step_time": 9.913986921310425
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 3.427579641342163,
      "step": 5296
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 5296,
      "training_step_time": 10.196854829788208
    },
    {
      "epoch": 2.020263671875e-05,
      "model_forward_time": 3.14540958404541,
      "step": 5296
    },
    {
      "epoch": 2.020263671875e-05,
      "step": 5296,
      "training_step_time": 10.32350778579712
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 3.1849398612976074,
      "step": 5297
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 5297,
      "training_step_time": 9.955355882644653
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 3.133232355117798,
      "step": 5297
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 5297,
      "training_step_time": 9.900145769119263
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 3.1387434005737305,
      "step": 5297
    },
    {
      "dynamics_loss_avg": 0.06166442111134529,
      "epoch": 2.0206451416015624e-05,
      "step": 5297
    },
    {
      "action_loss_avg": 0.021657984424382448,
      "epoch": 2.0206451416015624e-05,
      "step": 5297
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 5297,
      "training_step_time": 9.906423091888428
    },
    {
      "epoch": 2.0206451416015624e-05,
      "model_forward_time": 3.452139377593994,
      "step": 5297
    },
    {
      "epoch": 2.0206451416015624e-05,
      "step": 5297,
      "training_step_time": 10.354125022888184
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 3.182478904724121,
      "step": 5298
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 5298,
      "training_step_time": 9.958347082138062
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 3.1340179443359375,
      "step": 5298
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 5298,
      "training_step_time": 9.912400007247925
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 3.1376965045928955,
      "step": 5298
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 5298,
      "training_step_time": 9.908163070678711
    },
    {
      "epoch": 2.021026611328125e-05,
      "model_forward_time": 3.1395680904388428,
      "step": 5298
    },
    {
      "epoch": 2.021026611328125e-05,
      "step": 5298,
      "training_step_time": 10.262667179107666
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 3.1730642318725586,
      "step": 5299
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 5299,
      "training_step_time": 10.234257698059082
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 3.133518695831299,
      "step": 5299
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 5299,
      "training_step_time": 9.905627727508545
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 3.1339993476867676,
      "step": 5299
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 5299,
      "training_step_time": 9.907499074935913
    },
    {
      "epoch": 2.0214080810546876e-05,
      "model_forward_time": 3.136263847351074,
      "step": 5299
    },
    {
      "epoch": 2.0214080810546876e-05,
      "step": 5299,
      "training_step_time": 10.262502431869507
    },
    {
      "epoch": 2.02178955078125e-05,
      "grad_norm": 0.36494970321655273,
      "learning_rate": 1e-05,
      "loss": 0.0916,
      "step": 5300
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 3.1918017864227295,
      "step": 5300
    },
    {
      "dynamics_loss_avg": 0.06453732661902904,
      "epoch": 2.02178955078125e-05,
      "step": 5300
    },
    {
      "action_loss_avg": 0.03091281889937818,
      "epoch": 2.02178955078125e-05,
      "step": 5300
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 5300,
      "training_step_time": 9.962653160095215
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 3.3977768421173096,
      "step": 5300
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 5300,
      "training_step_time": 10.163950443267822
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 3.1378378868103027,
      "step": 5300
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 5300,
      "training_step_time": 9.905313968658447
    },
    {
      "epoch": 2.02178955078125e-05,
      "model_forward_time": 3.132748603820801,
      "step": 5300
    },
    {
      "epoch": 2.02178955078125e-05,
      "step": 5300,
      "training_step_time": 10.257487535476685
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 3.1616363525390625,
      "step": 5301
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 5301,
      "training_step_time": 9.934139251708984
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 3.1399595737457275,
      "step": 5301
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 5301,
      "training_step_time": 9.907391786575317
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 3.1349494457244873,
      "step": 5301
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 5301,
      "training_step_time": 10.174370288848877
    },
    {
      "epoch": 2.0221710205078126e-05,
      "model_forward_time": 3.138991117477417,
      "step": 5301
    },
    {
      "epoch": 2.0221710205078126e-05,
      "step": 5301,
      "training_step_time": 10.299480438232422
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 3.185999631881714,
      "step": 5302
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 5302,
      "training_step_time": 9.956584453582764
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 3.1368463039398193,
      "step": 5302
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 5302,
      "training_step_time": 9.905675172805786
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 3.137850761413574,
      "step": 5302
    },
    {
      "dynamics_loss_avg": 0.06826633494347334,
      "epoch": 2.022552490234375e-05,
      "step": 5302
    },
    {
      "action_loss_avg": 0.022135536745190622,
      "epoch": 2.022552490234375e-05,
      "step": 5302
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 5302,
      "training_step_time": 9.911420822143555
    },
    {
      "epoch": 2.022552490234375e-05,
      "model_forward_time": 3.399907112121582,
      "step": 5302
    },
    {
      "epoch": 2.022552490234375e-05,
      "step": 5302,
      "training_step_time": 10.260582447052002
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 3.179405927658081,
      "step": 5303
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 5303,
      "training_step_time": 9.948347806930542
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 3.1380059719085693,
      "step": 5303
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 5303,
      "training_step_time": 9.903770685195923
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 3.1371185779571533,
      "step": 5303
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 5303,
      "training_step_time": 9.905946731567383
    },
    {
      "epoch": 2.0229339599609375e-05,
      "model_forward_time": 3.139031171798706,
      "step": 5303
    },
    {
      "epoch": 2.0229339599609375e-05,
      "step": 5303,
      "training_step_time": 10.259748220443726
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 3.1895644664764404,
      "step": 5304
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 5304,
      "training_step_time": 10.226804971694946
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 3.1365838050842285,
      "step": 5304
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 5304,
      "training_step_time": 9.906254053115845
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 3.133197546005249,
      "step": 5304
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 5304,
      "training_step_time": 9.898186206817627
    },
    {
      "epoch": 2.0233154296875e-05,
      "model_forward_time": 3.136000633239746,
      "step": 5304
    },
    {
      "epoch": 2.0233154296875e-05,
      "step": 5304,
      "training_step_time": 10.279761552810669
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 3.187004804611206,
      "step": 5305
    },
    {
      "dynamics_loss_avg": 0.05266467910259962,
      "epoch": 2.0236968994140624e-05,
      "step": 5305
    },
    {
      "action_loss_avg": 0.024882781133055688,
      "epoch": 2.0236968994140624e-05,
      "step": 5305
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 5305,
      "training_step_time": 9.960402250289917
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 3.4171369075775146,
      "step": 5305
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 5305,
      "training_step_time": 10.190128564834595
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 3.138822078704834,
      "step": 5305
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 5305,
      "training_step_time": 9.910803318023682
    },
    {
      "epoch": 2.0236968994140624e-05,
      "model_forward_time": 3.1453540325164795,
      "step": 5305
    },
    {
      "epoch": 2.0236968994140624e-05,
      "step": 5305,
      "training_step_time": 10.274837732315063
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 3.178037166595459,
      "step": 5306
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 5306,
      "training_step_time": 9.953505754470825
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 3.13686466217041,
      "step": 5306
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 5306,
      "training_step_time": 9.906985998153687
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 3.3996026515960693,
      "step": 5306
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 5306,
      "training_step_time": 10.179312705993652
    },
    {
      "epoch": 2.024078369140625e-05,
      "model_forward_time": 3.14790415763855,
      "step": 5306
    },
    {
      "epoch": 2.024078369140625e-05,
      "step": 5306,
      "training_step_time": 10.308974504470825
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 3.189832925796509,
      "step": 5307
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 5307,
      "training_step_time": 9.97191858291626
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 3.1460916996002197,
      "step": 5307
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 5307,
      "training_step_time": 9.925583362579346
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 3.147055149078369,
      "step": 5307
    },
    {
      "dynamics_loss_avg": 0.04760222118347883,
      "epoch": 2.0244598388671876e-05,
      "step": 5307
    },
    {
      "action_loss_avg": 0.017592700105160473,
      "epoch": 2.0244598388671876e-05,
      "step": 5307
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 5307,
      "training_step_time": 9.926452875137329
    },
    {
      "epoch": 2.0244598388671876e-05,
      "model_forward_time": 3.428284168243408,
      "step": 5307
    },
    {
      "epoch": 2.0244598388671876e-05,
      "step": 5307,
      "training_step_time": 10.299851417541504
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 3.1926450729370117,
      "step": 5308
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 5308,
      "training_step_time": 9.980867862701416
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 3.142214298248291,
      "step": 5308
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 5308,
      "training_step_time": 9.919814825057983
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 3.146760940551758,
      "step": 5308
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 5308,
      "training_step_time": 9.921923875808716
    },
    {
      "epoch": 2.02484130859375e-05,
      "model_forward_time": 3.1422574520111084,
      "step": 5308
    },
    {
      "epoch": 2.02484130859375e-05,
      "step": 5308,
      "training_step_time": 10.292176008224487
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 3.455383539199829,
      "step": 5309
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 5309,
      "training_step_time": 10.231952428817749
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 3.140862226486206,
      "step": 5309
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 5309,
      "training_step_time": 9.91487169265747
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 3.1459805965423584,
      "step": 5309
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 5309,
      "training_step_time": 9.921832084655762
    },
    {
      "epoch": 2.0252227783203126e-05,
      "model_forward_time": 3.138996124267578,
      "step": 5309
    },
    {
      "epoch": 2.0252227783203126e-05,
      "step": 5309,
      "training_step_time": 10.249931335449219
    },
    {
      "epoch": 2.025604248046875e-05,
      "grad_norm": 0.4883321523666382,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 5310
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 3.1762330532073975,
      "step": 5310
    },
    {
      "dynamics_loss_avg": 0.06958758700639009,
      "epoch": 2.025604248046875e-05,
      "step": 5310
    },
    {
      "action_loss_avg": 0.03324660649523139,
      "epoch": 2.025604248046875e-05,
      "step": 5310
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 5310,
      "training_step_time": 9.950804710388184
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 3.3951733112335205,
      "step": 5310
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 5310,
      "training_step_time": 10.165806770324707
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 3.141181230545044,
      "step": 5310
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 5310,
      "training_step_time": 9.917668342590332
    },
    {
      "epoch": 2.025604248046875e-05,
      "model_forward_time": 3.1412148475646973,
      "step": 5310
    },
    {
      "epoch": 2.025604248046875e-05,
      "step": 5310,
      "training_step_time": 10.266693592071533
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 3.1787655353546143,
      "step": 5311
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 5311,
      "training_step_time": 9.953548669815063
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 3.138726234436035,
      "step": 5311
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 5311,
      "training_step_time": 9.91097617149353
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 3.4475138187408447,
      "step": 5311
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 5311,
      "training_step_time": 10.22116470336914
    },
    {
      "epoch": 2.0259857177734375e-05,
      "model_forward_time": 3.1391725540161133,
      "step": 5311
    },
    {
      "epoch": 2.0259857177734375e-05,
      "step": 5311,
      "training_step_time": 10.277039527893066
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 3.1803531646728516,
      "step": 5312
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 5312,
      "training_step_time": 9.955706357955933
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 3.146832227706909,
      "step": 5312
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 5312,
      "training_step_time": 9.920539379119873
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 3.140148878097534,
      "step": 5312
    },
    {
      "dynamics_loss_avg": 0.05500615146011114,
      "epoch": 2.0263671875e-05,
      "step": 5312
    },
    {
      "action_loss_avg": 0.02604053574614227,
      "epoch": 2.0263671875e-05,
      "step": 5312
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 5312,
      "training_step_time": 9.914593935012817
    },
    {
      "epoch": 2.0263671875e-05,
      "model_forward_time": 3.398787260055542,
      "step": 5312
    },
    {
      "epoch": 2.0263671875e-05,
      "step": 5312,
      "training_step_time": 10.26382040977478
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 3.181009292602539,
      "step": 5313
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 5313,
      "training_step_time": 9.956008911132812
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 3.14003586769104,
      "step": 5313
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 5313,
      "training_step_time": 9.91234827041626
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 3.136850118637085,
      "step": 5313
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 5313,
      "training_step_time": 9.91057300567627
    },
    {
      "epoch": 2.0267486572265624e-05,
      "model_forward_time": 3.140627145767212,
      "step": 5313
    },
    {
      "epoch": 2.0267486572265624e-05,
      "step": 5313,
      "training_step_time": 10.266586303710938
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 3.4550294876098633,
      "step": 5314
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 5314,
      "training_step_time": 10.226433277130127
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 3.1439151763916016,
      "step": 5314
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 5314,
      "training_step_time": 9.916101455688477
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 3.138369083404541,
      "step": 5314
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 5314,
      "training_step_time": 9.9116849899292
    },
    {
      "epoch": 2.027130126953125e-05,
      "model_forward_time": 3.1434600353240967,
      "step": 5314
    },
    {
      "epoch": 2.027130126953125e-05,
      "step": 5314,
      "training_step_time": 10.28394889831543
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 3.171868324279785,
      "step": 5315
    },
    {
      "dynamics_loss_avg": 0.06148365624248982,
      "epoch": 2.0275115966796876e-05,
      "step": 5315
    },
    {
      "action_loss_avg": 0.021234720665961503,
      "epoch": 2.0275115966796876e-05,
      "step": 5315
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 5315,
      "training_step_time": 9.946459531784058
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 3.412083387374878,
      "step": 5315
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 5315,
      "training_step_time": 10.179064750671387
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 3.1359260082244873,
      "step": 5315
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 5315,
      "training_step_time": 9.90819263458252
    },
    {
      "epoch": 2.0275115966796876e-05,
      "model_forward_time": 3.1363203525543213,
      "step": 5315
    },
    {
      "epoch": 2.0275115966796876e-05,
      "step": 5315,
      "training_step_time": 10.316528558731079
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 3.1963887214660645,
      "step": 5316
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 5316,
      "training_step_time": 9.972254514694214
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 3.142198324203491,
      "step": 5316
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 5316,
      "training_step_time": 9.917593717575073
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 3.4143588542938232,
      "step": 5316
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 5316,
      "training_step_time": 10.189536094665527
    },
    {
      "epoch": 2.02789306640625e-05,
      "model_forward_time": 3.153440237045288,
      "step": 5316
    },
    {
      "epoch": 2.02789306640625e-05,
      "step": 5316,
      "training_step_time": 10.324357509613037
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 3.174358606338501,
      "step": 5317
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 5317,
      "training_step_time": 9.948527097702026
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 3.1344823837280273,
      "step": 5317
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 5317,
      "training_step_time": 9.90865421295166
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 3.1414618492126465,
      "step": 5317
    },
    {
      "dynamics_loss_avg": 0.05392973385751247,
      "epoch": 2.0282745361328126e-05,
      "step": 5317
    },
    {
      "action_loss_avg": 0.023661248199641705,
      "epoch": 2.0282745361328126e-05,
      "step": 5317
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 5317,
      "training_step_time": 9.917803049087524
    },
    {
      "epoch": 2.0282745361328126e-05,
      "model_forward_time": 3.4325499534606934,
      "step": 5317
    },
    {
      "epoch": 2.0282745361328126e-05,
      "step": 5317,
      "training_step_time": 10.303378105163574
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 3.1741349697113037,
      "step": 5318
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 5318,
      "training_step_time": 9.951321125030518
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 3.1354312896728516,
      "step": 5318
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 5318,
      "training_step_time": 9.908924341201782
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 3.1427533626556396,
      "step": 5318
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 5318,
      "training_step_time": 9.91615104675293
    },
    {
      "epoch": 2.028656005859375e-05,
      "model_forward_time": 3.1349172592163086,
      "step": 5318
    },
    {
      "epoch": 2.028656005859375e-05,
      "step": 5318,
      "training_step_time": 10.013112306594849
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 3.4516897201538086,
      "step": 5319
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 5319,
      "training_step_time": 10.229659795761108
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 2.43981671333313,
      "step": 5319
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 5319,
      "training_step_time": 7.473799228668213
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 3.139801263809204,
      "step": 5319
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 5319,
      "training_step_time": 9.914970397949219
    },
    {
      "epoch": 2.0290374755859375e-05,
      "model_forward_time": 3.1405274868011475,
      "step": 5319
    },
    {
      "epoch": 2.0290374755859375e-05,
      "step": 5319,
      "training_step_time": 10.274085760116577
    },
    {
      "epoch": 2.0294189453125e-05,
      "grad_norm": 0.2653711438179016,
      "learning_rate": 1e-05,
      "loss": 0.0905,
      "step": 5320
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 3.1645238399505615,
      "step": 5320
    },
    {
      "dynamics_loss_avg": 0.048212513653561474,
      "epoch": 2.0294189453125e-05,
      "step": 5320
    },
    {
      "action_loss_avg": 0.015819420223124325,
      "epoch": 2.0294189453125e-05,
      "step": 5320
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 5320,
      "training_step_time": 9.939699411392212
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 3.390388250350952,
      "step": 5320
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 5320,
      "training_step_time": 10.16277551651001
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 3.139887571334839,
      "step": 5320
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 5320,
      "training_step_time": 9.914358615875244
    },
    {
      "epoch": 2.0294189453125e-05,
      "model_forward_time": 3.137869119644165,
      "step": 5320
    },
    {
      "epoch": 2.0294189453125e-05,
      "step": 5320,
      "training_step_time": 10.265028953552246
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 3.182459592819214,
      "step": 5321
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 5321,
      "training_step_time": 9.953439712524414
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 3.142148494720459,
      "step": 5321
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 5321,
      "training_step_time": 9.920705556869507
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 3.418058395385742,
      "step": 5321
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 5321,
      "training_step_time": 10.196607112884521
    },
    {
      "epoch": 2.0298004150390624e-05,
      "model_forward_time": 3.142301559448242,
      "step": 5321
    },
    {
      "epoch": 2.0298004150390624e-05,
      "step": 5321,
      "training_step_time": 10.31128478050232
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 3.1763346195220947,
      "step": 5322
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 5322,
      "training_step_time": 9.953874111175537
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 3.144399404525757,
      "step": 5322
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 5322,
      "training_step_time": 9.92176604270935
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 3.1409711837768555,
      "step": 5322
    },
    {
      "dynamics_loss_avg": 0.07981434017419815,
      "epoch": 2.030181884765625e-05,
      "step": 5322
    },
    {
      "action_loss_avg": 0.03126824852079153,
      "epoch": 2.030181884765625e-05,
      "step": 5322
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 5322,
      "training_step_time": 9.921659708023071
    },
    {
      "epoch": 2.030181884765625e-05,
      "model_forward_time": 3.399137258529663,
      "step": 5322
    },
    {
      "epoch": 2.030181884765625e-05,
      "step": 5322,
      "training_step_time": 10.26883316040039
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 3.1928961277008057,
      "step": 5323
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 5323,
      "training_step_time": 9.974570512771606
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 3.148831367492676,
      "step": 5323
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 5323,
      "training_step_time": 9.928599834442139
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 3.1447410583496094,
      "step": 5323
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 5323,
      "training_step_time": 9.923369407653809
    },
    {
      "epoch": 2.0305633544921877e-05,
      "model_forward_time": 3.147160053253174,
      "step": 5323
    },
    {
      "epoch": 2.0305633544921877e-05,
      "step": 5323,
      "training_step_time": 10.045478582382202
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 3.4328227043151855,
      "step": 5324
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 5324,
      "training_step_time": 10.210598468780518
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 3.137293815612793,
      "step": 5324
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 5324,
      "training_step_time": 9.914124727249146
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 3.138305425643921,
      "step": 5324
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 5324,
      "training_step_time": 9.916677474975586
    },
    {
      "epoch": 2.03094482421875e-05,
      "model_forward_time": 3.1422278881073,
      "step": 5324
    },
    {
      "epoch": 2.03094482421875e-05,
      "step": 5324,
      "training_step_time": 10.308075904846191
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 3.184788227081299,
      "step": 5325
    },
    {
      "dynamics_loss_avg": 0.0470461618155241,
      "epoch": 2.0313262939453126e-05,
      "step": 5325
    },
    {
      "action_loss_avg": 0.020065005775541066,
      "epoch": 2.0313262939453126e-05,
      "step": 5325
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 5325,
      "training_step_time": 9.95595383644104
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 3.4078783988952637,
      "step": 5325
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 5325,
      "training_step_time": 10.176684141159058
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 3.1322975158691406,
      "step": 5325
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 5325,
      "training_step_time": 9.900200605392456
    },
    {
      "epoch": 2.0313262939453126e-05,
      "model_forward_time": 3.1354880332946777,
      "step": 5325
    },
    {
      "epoch": 2.0313262939453126e-05,
      "step": 5325,
      "training_step_time": 10.255871534347534
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 3.1612274646759033,
      "step": 5326
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 5326,
      "training_step_time": 9.936662197113037
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 3.131479501724243,
      "step": 5326
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 5326,
      "training_step_time": 9.908326625823975
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 3.395254373550415,
      "step": 5326
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 5326,
      "training_step_time": 10.171129703521729
    },
    {
      "epoch": 2.031707763671875e-05,
      "model_forward_time": 3.142991065979004,
      "step": 5326
    },
    {
      "epoch": 2.031707763671875e-05,
      "step": 5326,
      "training_step_time": 10.321790218353271
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 3.166377544403076,
      "step": 5327
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 5327,
      "training_step_time": 9.942761898040771
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 3.133452892303467,
      "step": 5327
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 5327,
      "training_step_time": 9.905847549438477
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 3.1378376483917236,
      "step": 5327
    },
    {
      "dynamics_loss_avg": 0.054367268644273284,
      "epoch": 2.0320892333984375e-05,
      "step": 5327
    },
    {
      "action_loss_avg": 0.022993630450218915,
      "epoch": 2.0320892333984375e-05,
      "step": 5327
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 5327,
      "training_step_time": 9.920085668563843
    },
    {
      "epoch": 2.0320892333984375e-05,
      "model_forward_time": 3.435450792312622,
      "step": 5327
    },
    {
      "epoch": 2.0320892333984375e-05,
      "step": 5327,
      "training_step_time": 10.30574631690979
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 3.1935760974884033,
      "step": 5328
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 5328,
      "training_step_time": 9.967634439468384
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 3.136338233947754,
      "step": 5328
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 5328,
      "training_step_time": 9.909508228302002
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 3.1414856910705566,
      "step": 5328
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 5328,
      "training_step_time": 9.912431001663208
    },
    {
      "epoch": 2.032470703125e-05,
      "model_forward_time": 3.135111093521118,
      "step": 5328
    },
    {
      "epoch": 2.032470703125e-05,
      "step": 5328,
      "training_step_time": 10.01712965965271
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 3.470731735229492,
      "step": 5329
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 5329,
      "training_step_time": 10.242418050765991
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 3.140160083770752,
      "step": 5329
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 5329,
      "training_step_time": 9.910484790802002
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 3.144678831100464,
      "step": 5329
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 5329,
      "training_step_time": 9.915169477462769
    },
    {
      "epoch": 2.0328521728515624e-05,
      "model_forward_time": 3.1397929191589355,
      "step": 5329
    },
    {
      "epoch": 2.0328521728515624e-05,
      "step": 5329,
      "training_step_time": 10.022501468658447
    },
    {
      "epoch": 2.033233642578125e-05,
      "grad_norm": 0.32232198119163513,
      "learning_rate": 1e-05,
      "loss": 0.0877,
      "step": 5330
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 3.1752071380615234,
      "step": 5330
    },
    {
      "dynamics_loss_avg": 0.06717425864189863,
      "epoch": 2.033233642578125e-05,
      "step": 5330
    },
    {
      "action_loss_avg": 0.02646464332938194,
      "epoch": 2.033233642578125e-05,
      "step": 5330
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 5330,
      "training_step_time": 9.947737216949463
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 3.4037256240844727,
      "step": 5330
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 5330,
      "training_step_time": 10.174778461456299
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 3.1507887840270996,
      "step": 5330
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 5330,
      "training_step_time": 9.925543546676636
    },
    {
      "epoch": 2.033233642578125e-05,
      "model_forward_time": 3.138950824737549,
      "step": 5330
    },
    {
      "epoch": 2.033233642578125e-05,
      "step": 5330,
      "training_step_time": 10.276516199111938
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 3.160029172897339,
      "step": 5331
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 5331,
      "training_step_time": 9.93485975265503
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 3.140864372253418,
      "step": 5331
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 5331,
      "training_step_time": 9.912500858306885
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 3.418665647506714,
      "step": 5331
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 5331,
      "training_step_time": 10.194342613220215
    },
    {
      "epoch": 2.0336151123046877e-05,
      "model_forward_time": 3.1416637897491455,
      "step": 5331
    },
    {
      "epoch": 2.0336151123046877e-05,
      "step": 5331,
      "training_step_time": 10.31873345375061
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 3.189650774002075,
      "step": 5332
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 5332,
      "training_step_time": 9.968236684799194
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 3.144820213317871,
      "step": 5332
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 5332,
      "training_step_time": 9.915355205535889
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 3.144028902053833,
      "step": 5332
    },
    {
      "dynamics_loss_avg": 0.048731824941933156,
      "epoch": 2.03399658203125e-05,
      "step": 5332
    },
    {
      "action_loss_avg": 0.019951820373535156,
      "epoch": 2.03399658203125e-05,
      "step": 5332
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 5332,
      "training_step_time": 9.918190479278564
    },
    {
      "epoch": 2.03399658203125e-05,
      "model_forward_time": 3.4455766677856445,
      "step": 5332
    },
    {
      "epoch": 2.03399658203125e-05,
      "step": 5332,
      "training_step_time": 10.316548109054565
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 3.1902520656585693,
      "step": 5333
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 5333,
      "training_step_time": 9.964292049407959
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 3.1530165672302246,
      "step": 5333
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 5333,
      "training_step_time": 9.92648720741272
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 3.1437277793884277,
      "step": 5333
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 5333,
      "training_step_time": 9.913646221160889
    },
    {
      "epoch": 2.0343780517578126e-05,
      "model_forward_time": 3.148681402206421,
      "step": 5333
    },
    {
      "epoch": 2.0343780517578126e-05,
      "step": 5333,
      "training_step_time": 10.038710594177246
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 3.4507293701171875,
      "step": 5334
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 5334,
      "training_step_time": 10.222593069076538
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 3.1481289863586426,
      "step": 5334
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 5334,
      "training_step_time": 9.920748472213745
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 3.145254135131836,
      "step": 5334
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 5334,
      "training_step_time": 9.919674634933472
    },
    {
      "epoch": 2.034759521484375e-05,
      "model_forward_time": 3.156583786010742,
      "step": 5334
    },
    {
      "epoch": 2.034759521484375e-05,
      "step": 5334,
      "training_step_time": 10.042104005813599
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 3.1738831996917725,
      "step": 5335
    },
    {
      "dynamics_loss_avg": 0.05992723740637303,
      "epoch": 2.0351409912109375e-05,
      "step": 5335
    },
    {
      "action_loss_avg": 0.025076747126877307,
      "epoch": 2.0351409912109375e-05,
      "step": 5335
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 5335,
      "training_step_time": 9.9617760181427
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 3.4276225566864014,
      "step": 5335
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 5335,
      "training_step_time": 10.199018001556396
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 3.141338586807251,
      "step": 5335
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 5335,
      "training_step_time": 9.91502046585083
    },
    {
      "epoch": 2.0351409912109375e-05,
      "model_forward_time": 3.1421854496002197,
      "step": 5335
    },
    {
      "epoch": 2.0351409912109375e-05,
      "step": 5335,
      "training_step_time": 10.323247909545898
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 3.1714775562286377,
      "step": 5336
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 5336,
      "training_step_time": 9.948009967803955
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 3.1393485069274902,
      "step": 5336
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 5336,
      "training_step_time": 9.911781072616577
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 3.40459942817688,
      "step": 5336
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 5336,
      "training_step_time": 10.177894353866577
    },
    {
      "epoch": 2.0355224609375e-05,
      "model_forward_time": 3.1439602375030518,
      "step": 5336
    },
    {
      "epoch": 2.0355224609375e-05,
      "step": 5336,
      "training_step_time": 10.329626083374023
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 3.1859233379364014,
      "step": 5337
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 5337,
      "training_step_time": 9.958012104034424
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 3.132235527038574,
      "step": 5337
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 5337,
      "training_step_time": 9.905102729797363
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 3.1338534355163574,
      "step": 5337
    },
    {
      "dynamics_loss_avg": 0.04416350889950991,
      "epoch": 2.0359039306640624e-05,
      "step": 5337
    },
    {
      "action_loss_avg": 0.017469358118250965,
      "epoch": 2.0359039306640624e-05,
      "step": 5337
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 5337,
      "training_step_time": 9.9107825756073
    },
    {
      "epoch": 2.0359039306640624e-05,
      "model_forward_time": 3.4247498512268066,
      "step": 5337
    },
    {
      "epoch": 2.0359039306640624e-05,
      "step": 5337,
      "training_step_time": 10.314491510391235
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 3.1745357513427734,
      "step": 5338
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 5338,
      "training_step_time": 9.944749355316162
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 3.1340010166168213,
      "step": 5338
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 5338,
      "training_step_time": 9.900776147842407
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 3.133620262145996,
      "step": 5338
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 5338,
      "training_step_time": 9.901403665542603
    },
    {
      "epoch": 2.036285400390625e-05,
      "model_forward_time": 3.129551887512207,
      "step": 5338
    },
    {
      "epoch": 2.036285400390625e-05,
      "step": 5338,
      "training_step_time": 10.01547884941101
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 3.4395458698272705,
      "step": 5339
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 5339,
      "training_step_time": 10.2038893699646
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 3.1313107013702393,
      "step": 5339
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 5339,
      "training_step_time": 9.894675731658936
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 3.1340208053588867,
      "step": 5339
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 5339,
      "training_step_time": 9.896997451782227
    },
    {
      "epoch": 2.0366668701171877e-05,
      "model_forward_time": 3.133157968521118,
      "step": 5339
    },
    {
      "epoch": 2.0366668701171877e-05,
      "step": 5339,
      "training_step_time": 10.302528381347656
    },
    {
      "epoch": 2.03704833984375e-05,
      "grad_norm": 0.5073638558387756,
      "learning_rate": 1e-05,
      "loss": 0.0842,
      "step": 5340
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 3.19598650932312,
      "step": 5340
    },
    {
      "dynamics_loss_avg": 0.05193635653704405,
      "epoch": 2.03704833984375e-05,
      "step": 5340
    },
    {
      "action_loss_avg": 0.02790585970506072,
      "epoch": 2.03704833984375e-05,
      "step": 5340
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 5340,
      "training_step_time": 9.969261169433594
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 3.403115749359131,
      "step": 5340
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 5340,
      "training_step_time": 10.175050497055054
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 3.1485698223114014,
      "step": 5340
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 5340,
      "training_step_time": 9.922523260116577
    },
    {
      "epoch": 2.03704833984375e-05,
      "model_forward_time": 3.1426784992218018,
      "step": 5340
    },
    {
      "epoch": 2.03704833984375e-05,
      "step": 5340,
      "training_step_time": 10.037081956863403
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 3.1984729766845703,
      "step": 5341
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 5341,
      "training_step_time": 9.97019648551941
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 3.147007703781128,
      "step": 5341
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 5341,
      "training_step_time": 9.915897130966187
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 3.4133553504943848,
      "step": 5341
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 5341,
      "training_step_time": 10.185213804244995
    },
    {
      "epoch": 2.0374298095703126e-05,
      "model_forward_time": 3.1421024799346924,
      "step": 5341
    },
    {
      "epoch": 2.0374298095703126e-05,
      "step": 5341,
      "training_step_time": 10.275336742401123
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 3.1744778156280518,
      "step": 5342
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 5342,
      "training_step_time": 9.942795991897583
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 2.4328596591949463,
      "step": 5342
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 5342,
      "training_step_time": 7.456167936325073
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 3.133333444595337,
      "step": 5342
    },
    {
      "dynamics_loss_avg": 0.05562048451974988,
      "epoch": 2.037811279296875e-05,
      "step": 5342
    },
    {
      "action_loss_avg": 0.020487312320619823,
      "epoch": 2.037811279296875e-05,
      "step": 5342
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 5342,
      "training_step_time": 10.16434359550476
    },
    {
      "epoch": 2.037811279296875e-05,
      "model_forward_time": 3.133819103240967,
      "step": 5342
    },
    {
      "epoch": 2.037811279296875e-05,
      "step": 5342,
      "training_step_time": 10.281787633895874
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 3.1677448749542236,
      "step": 5343
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 5343,
      "training_step_time": 9.93405556678772
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 3.1377451419830322,
      "step": 5343
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 5343,
      "training_step_time": 9.901319742202759
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 3.1420345306396484,
      "step": 5343
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 5343,
      "training_step_time": 9.909218549728394
    },
    {
      "epoch": 2.0381927490234375e-05,
      "model_forward_time": 3.1455063819885254,
      "step": 5343
    },
    {
      "epoch": 2.0381927490234375e-05,
      "step": 5343,
      "training_step_time": 10.595877885818481
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 3.180634021759033,
      "step": 5344
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 5344,
      "training_step_time": 9.949702024459839
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 3.144902467727661,
      "step": 5344
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 5344,
      "training_step_time": 9.910704612731934
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 3.1404662132263184,
      "step": 5344
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 5344,
      "training_step_time": 9.911142826080322
    },
    {
      "epoch": 2.03857421875e-05,
      "model_forward_time": 3.144137144088745,
      "step": 5344
    },
    {
      "epoch": 2.03857421875e-05,
      "step": 5344,
      "training_step_time": 10.289013624191284
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 3.4274845123291016,
      "step": 5345
    },
    {
      "dynamics_loss_avg": 0.05613277442753315,
      "epoch": 2.0389556884765624e-05,
      "step": 5345
    },
    {
      "action_loss_avg": 0.026338853966444732,
      "epoch": 2.0389556884765624e-05,
      "step": 5345
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 5345,
      "training_step_time": 10.199918985366821
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 3.139289379119873,
      "step": 5345
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 5345,
      "training_step_time": 9.904584169387817
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 3.1374363899230957,
      "step": 5345
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 5345,
      "training_step_time": 9.90198564529419
    },
    {
      "epoch": 2.0389556884765624e-05,
      "model_forward_time": 3.139935255050659,
      "step": 5345
    },
    {
      "epoch": 2.0389556884765624e-05,
      "step": 5345,
      "training_step_time": 10.028038740158081
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 3.1940689086914062,
      "step": 5346
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 5346,
      "training_step_time": 9.9626784324646
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 3.43031907081604,
      "step": 5346
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 5346,
      "training_step_time": 10.200474500656128
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 3.1438004970550537,
      "step": 5346
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 5346,
      "training_step_time": 9.91878604888916
    },
    {
      "epoch": 2.039337158203125e-05,
      "model_forward_time": 3.146022319793701,
      "step": 5346
    },
    {
      "epoch": 2.039337158203125e-05,
      "step": 5346,
      "training_step_time": 10.262200832366943
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 3.1968953609466553,
      "step": 5347
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 5347,
      "training_step_time": 9.971292972564697
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 3.1441309452056885,
      "step": 5347
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 5347,
      "training_step_time": 9.92008090019226
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 3.401320695877075,
      "step": 5347
    },
    {
      "dynamics_loss_avg": 0.06270988509058953,
      "epoch": 2.0397186279296877e-05,
      "step": 5347
    },
    {
      "action_loss_avg": 0.02447115583345294,
      "epoch": 2.0397186279296877e-05,
      "step": 5347
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 5347,
      "training_step_time": 10.175923347473145
    },
    {
      "epoch": 2.0397186279296877e-05,
      "model_forward_time": 3.1452205181121826,
      "step": 5347
    },
    {
      "epoch": 2.0397186279296877e-05,
      "step": 5347,
      "training_step_time": 10.315977811813354
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 3.174145460128784,
      "step": 5348
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 5348,
      "training_step_time": 9.948314666748047
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 3.1394267082214355,
      "step": 5348
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 5348,
      "training_step_time": 9.909027338027954
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 3.14243221282959,
      "step": 5348
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 5348,
      "training_step_time": 9.911125898361206
    },
    {
      "epoch": 2.04010009765625e-05,
      "model_forward_time": 3.4412288665771484,
      "step": 5348
    },
    {
      "epoch": 2.04010009765625e-05,
      "step": 5348,
      "training_step_time": 10.303935527801514
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 3.1747772693634033,
      "step": 5349
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 5349,
      "training_step_time": 9.944126605987549
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 3.130178928375244,
      "step": 5349
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 5349,
      "training_step_time": 9.899005651473999
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 3.1360912322998047,
      "step": 5349
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 5349,
      "training_step_time": 9.904710054397583
    },
    {
      "epoch": 2.0404815673828126e-05,
      "model_forward_time": 3.1326961517333984,
      "step": 5349
    },
    {
      "epoch": 2.0404815673828126e-05,
      "step": 5349,
      "training_step_time": 10.019882202148438
    },
    {
      "epoch": 2.040863037109375e-05,
      "grad_norm": 0.5716856718063354,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 5350
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 3.4827308654785156,
      "step": 5350
    },
    {
      "dynamics_loss_avg": 0.05460767205804586,
      "epoch": 2.040863037109375e-05,
      "step": 5350
    },
    {
      "action_loss_avg": 0.02215852605877444,
      "epoch": 2.040863037109375e-05,
      "step": 5350
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 5350,
      "training_step_time": 10.254812479019165
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 3.144601583480835,
      "step": 5350
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 5350,
      "training_step_time": 9.915043115615845
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 3.143575429916382,
      "step": 5350
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 5350,
      "training_step_time": 9.914514541625977
    },
    {
      "epoch": 2.040863037109375e-05,
      "model_forward_time": 3.144814968109131,
      "step": 5350
    },
    {
      "epoch": 2.040863037109375e-05,
      "step": 5350,
      "training_step_time": 10.026353359222412
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 3.1869068145751953,
      "step": 5351
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 5351,
      "training_step_time": 9.965693235397339
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 3.406627655029297,
      "step": 5351
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 5351,
      "training_step_time": 10.181837320327759
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 3.1536865234375,
      "step": 5351
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 5351,
      "training_step_time": 9.934295415878296
    },
    {
      "epoch": 2.0412445068359375e-05,
      "model_forward_time": 3.142965316772461,
      "step": 5351
    },
    {
      "epoch": 2.0412445068359375e-05,
      "step": 5351,
      "training_step_time": 10.037282943725586
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 3.196631669998169,
      "step": 5352
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 5352,
      "training_step_time": 9.967437267303467
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 3.146275281906128,
      "step": 5352
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 5352,
      "training_step_time": 9.91770052909851
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 3.4125492572784424,
      "step": 5352
    },
    {
      "dynamics_loss_avg": 0.05475254524499178,
      "epoch": 2.0416259765625e-05,
      "step": 5352
    },
    {
      "action_loss_avg": 0.026230819057673216,
      "epoch": 2.0416259765625e-05,
      "step": 5352
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 5352,
      "training_step_time": 10.184634447097778
    },
    {
      "epoch": 2.0416259765625e-05,
      "model_forward_time": 3.1440114974975586,
      "step": 5352
    },
    {
      "epoch": 2.0416259765625e-05,
      "step": 5352,
      "training_step_time": 10.267324447631836
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 3.165931224822998,
      "step": 5353
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 5353,
      "training_step_time": 9.937354564666748
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 3.1399216651916504,
      "step": 5353
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 5353,
      "training_step_time": 9.907438039779663
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 3.1384119987487793,
      "step": 5353
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 5353,
      "training_step_time": 10.186468601226807
    },
    {
      "epoch": 2.0420074462890624e-05,
      "model_forward_time": 3.1362714767456055,
      "step": 5353
    },
    {
      "epoch": 2.0420074462890624e-05,
      "step": 5353,
      "training_step_time": 10.022743463516235
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 3.18620228767395,
      "step": 5354
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 5354,
      "training_step_time": 9.94957184791565
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 3.1376802921295166,
      "step": 5354
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 5354,
      "training_step_time": 9.909998178482056
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 3.140883445739746,
      "step": 5354
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 5354,
      "training_step_time": 9.912062644958496
    },
    {
      "epoch": 2.042388916015625e-05,
      "model_forward_time": 3.1427032947540283,
      "step": 5354
    },
    {
      "epoch": 2.042388916015625e-05,
      "step": 5354,
      "training_step_time": 10.277318477630615
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 3.1827282905578613,
      "step": 5355
    },
    {
      "dynamics_loss_avg": 0.05791013780981302,
      "epoch": 2.0427703857421877e-05,
      "step": 5355
    },
    {
      "action_loss_avg": 0.024460338801145554,
      "epoch": 2.0427703857421877e-05,
      "step": 5355
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 5355,
      "training_step_time": 9.959827423095703
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 3.14353609085083,
      "step": 5355
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 5355,
      "training_step_time": 9.910470724105835
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 3.135045289993286,
      "step": 5355
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 5355,
      "training_step_time": 9.905633211135864
    },
    {
      "epoch": 2.0427703857421877e-05,
      "model_forward_time": 3.147054672241211,
      "step": 5355
    },
    {
      "epoch": 2.0427703857421877e-05,
      "step": 5355,
      "training_step_time": 10.030509948730469
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 3.177459478378296,
      "step": 5356
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 5356,
      "training_step_time": 10.23377799987793
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 3.145751476287842,
      "step": 5356
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 5356,
      "training_step_time": 9.915103673934937
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 3.148167133331299,
      "step": 5356
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 5356,
      "training_step_time": 9.92215895652771
    },
    {
      "epoch": 2.04315185546875e-05,
      "model_forward_time": 3.1464476585388184,
      "step": 5356
    },
    {
      "epoch": 2.04315185546875e-05,
      "step": 5356,
      "training_step_time": 10.020842552185059
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 3.1772923469543457,
      "step": 5357
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 5357,
      "training_step_time": 9.953020572662354
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 3.4364466667175293,
      "step": 5357
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 5357,
      "training_step_time": 10.209826469421387
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 3.1323680877685547,
      "step": 5357
    },
    {
      "dynamics_loss_avg": 0.06346151325851679,
      "epoch": 2.0435333251953126e-05,
      "step": 5357
    },
    {
      "action_loss_avg": 0.026284929644316436,
      "epoch": 2.0435333251953126e-05,
      "step": 5357
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 5357,
      "training_step_time": 9.907204389572144
    },
    {
      "epoch": 2.0435333251953126e-05,
      "model_forward_time": 3.14170241355896,
      "step": 5357
    },
    {
      "epoch": 2.0435333251953126e-05,
      "step": 5357,
      "training_step_time": 10.33802342414856
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 3.178542375564575,
      "step": 5358
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 5358,
      "training_step_time": 9.945868492126465
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 3.13577938079834,
      "step": 5358
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 5358,
      "training_step_time": 9.9028160572052
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 3.136540651321411,
      "step": 5358
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 5358,
      "training_step_time": 10.19031310081482
    },
    {
      "epoch": 2.043914794921875e-05,
      "model_forward_time": 3.134348154067993,
      "step": 5358
    },
    {
      "epoch": 2.043914794921875e-05,
      "step": 5358,
      "training_step_time": 10.308248281478882
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 3.179715633392334,
      "step": 5359
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 5359,
      "training_step_time": 9.9559485912323
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 3.128509283065796,
      "step": 5359
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 5359,
      "training_step_time": 9.898813962936401
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 3.1313748359680176,
      "step": 5359
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 5359,
      "training_step_time": 9.900142192840576
    },
    {
      "epoch": 2.0442962646484375e-05,
      "model_forward_time": 3.4075076580047607,
      "step": 5359
    },
    {
      "epoch": 2.0442962646484375e-05,
      "step": 5359,
      "training_step_time": 10.274435758590698
    },
    {
      "epoch": 2.044677734375e-05,
      "grad_norm": 0.4075731337070465,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 5360
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 3.163527011871338,
      "step": 5360
    },
    {
      "dynamics_loss_avg": 0.06282782051712274,
      "epoch": 2.044677734375e-05,
      "step": 5360
    },
    {
      "action_loss_avg": 0.02395343519747257,
      "epoch": 2.044677734375e-05,
      "step": 5360
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 5360,
      "training_step_time": 9.937398433685303
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 3.132746934890747,
      "step": 5360
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 5360,
      "training_step_time": 9.902198076248169
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 3.1331307888031006,
      "step": 5360
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 5360,
      "training_step_time": 9.910182476043701
    },
    {
      "epoch": 2.044677734375e-05,
      "model_forward_time": 3.1366641521453857,
      "step": 5360
    },
    {
      "epoch": 2.044677734375e-05,
      "step": 5360,
      "training_step_time": 10.335970640182495
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 3.436171531677246,
      "step": 5361
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 5361,
      "training_step_time": 10.213155031204224
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 3.1354310512542725,
      "step": 5361
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 5361,
      "training_step_time": 9.913417100906372
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 3.136390209197998,
      "step": 5361
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 5361,
      "training_step_time": 9.908986568450928
    },
    {
      "epoch": 2.0450592041015624e-05,
      "model_forward_time": 3.128485918045044,
      "step": 5361
    },
    {
      "epoch": 2.0450592041015624e-05,
      "step": 5361,
      "training_step_time": 10.01065468788147
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 3.1636502742767334,
      "step": 5362
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 5362,
      "training_step_time": 9.944846391677856
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 3.419252634048462,
      "step": 5362
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 5362,
      "training_step_time": 10.193244695663452
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 3.135401725769043,
      "step": 5362
    },
    {
      "dynamics_loss_avg": 0.06513793729245662,
      "epoch": 2.045440673828125e-05,
      "step": 5362
    },
    {
      "action_loss_avg": 0.028510470502078535,
      "epoch": 2.045440673828125e-05,
      "step": 5362
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 5362,
      "training_step_time": 9.914315223693848
    },
    {
      "epoch": 2.045440673828125e-05,
      "model_forward_time": 3.1312835216522217,
      "step": 5362
    },
    {
      "epoch": 2.045440673828125e-05,
      "step": 5362,
      "training_step_time": 10.278894901275635
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 3.1602048873901367,
      "step": 5363
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 5363,
      "training_step_time": 9.935976028442383
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 3.132033586502075,
      "step": 5363
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 5363,
      "training_step_time": 9.907092571258545
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 3.4248135089874268,
      "step": 5363
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 5363,
      "training_step_time": 10.197731018066406
    },
    {
      "epoch": 2.0458221435546877e-05,
      "model_forward_time": 3.1288528442382812,
      "step": 5363
    },
    {
      "epoch": 2.0458221435546877e-05,
      "step": 5363,
      "training_step_time": 10.295866250991821
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 3.1688735485076904,
      "step": 5364
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 5364,
      "training_step_time": 9.946429014205933
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 3.132550001144409,
      "step": 5364
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 5364,
      "training_step_time": 9.90687894821167
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 3.131587505340576,
      "step": 5364
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 5364,
      "training_step_time": 9.906378030776978
    },
    {
      "epoch": 2.04620361328125e-05,
      "model_forward_time": 3.1370227336883545,
      "step": 5364
    },
    {
      "epoch": 2.04620361328125e-05,
      "step": 5364,
      "training_step_time": 10.30332064628601
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 3.1766016483306885,
      "step": 5365
    },
    {
      "dynamics_loss_avg": 0.04286171589046717,
      "epoch": 2.0465850830078126e-05,
      "step": 5365
    },
    {
      "action_loss_avg": 0.014852949883788824,
      "epoch": 2.0465850830078126e-05,
      "step": 5365
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 5365,
      "training_step_time": 9.96279239654541
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 3.1357924938201904,
      "step": 5365
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 5365,
      "training_step_time": 9.910674810409546
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 3.1316583156585693,
      "step": 5365
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 5365,
      "training_step_time": 9.909892320632935
    },
    {
      "epoch": 2.0465850830078126e-05,
      "model_forward_time": 3.135141134262085,
      "step": 5365
    },
    {
      "epoch": 2.0465850830078126e-05,
      "step": 5365,
      "training_step_time": 10.247482061386108
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 3.3976593017578125,
      "step": 5366
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 5366,
      "training_step_time": 10.17302131652832
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 3.132061004638672,
      "step": 5366
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 5366,
      "training_step_time": 9.909563064575195
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 2.4294774532318115,
      "step": 5366
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 5366,
      "training_step_time": 7.464207887649536
    },
    {
      "epoch": 2.046966552734375e-05,
      "model_forward_time": 3.134152412414551,
      "step": 5366
    },
    {
      "epoch": 2.046966552734375e-05,
      "step": 5366,
      "training_step_time": 10.024454832077026
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 3.202946424484253,
      "step": 5367
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 5367,
      "training_step_time": 9.98676609992981
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 3.1426548957824707,
      "step": 5367
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 5367,
      "training_step_time": 9.923664093017578
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 3.390632152557373,
      "step": 5367
    },
    {
      "dynamics_loss_avg": 0.06568859219551086,
      "epoch": 2.0473480224609375e-05,
      "step": 5367
    },
    {
      "action_loss_avg": 0.03148441798985004,
      "epoch": 2.0473480224609375e-05,
      "step": 5367
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 5367,
      "training_step_time": 10.175609350204468
    },
    {
      "epoch": 2.0473480224609375e-05,
      "model_forward_time": 3.1498615741729736,
      "step": 5367
    },
    {
      "epoch": 2.0473480224609375e-05,
      "step": 5367,
      "training_step_time": 10.031294345855713
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 3.1847150325775146,
      "step": 5368
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 5368,
      "training_step_time": 9.973275184631348
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 3.1437954902648926,
      "step": 5368
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 5368,
      "training_step_time": 9.930474996566772
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 3.4170830249786377,
      "step": 5368
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 5368,
      "training_step_time": 10.199640274047852
    },
    {
      "epoch": 2.0477294921875e-05,
      "model_forward_time": 3.147310733795166,
      "step": 5368
    },
    {
      "epoch": 2.0477294921875e-05,
      "step": 5368,
      "training_step_time": 10.302817344665527
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 3.1831870079040527,
      "step": 5369
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 5369,
      "training_step_time": 9.964300632476807
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 3.1420154571533203,
      "step": 5369
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 5369,
      "training_step_time": 9.922881126403809
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 3.1449062824249268,
      "step": 5369
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 5369,
      "training_step_time": 9.922966480255127
    },
    {
      "epoch": 2.0481109619140624e-05,
      "model_forward_time": 3.1444051265716553,
      "step": 5369
    },
    {
      "epoch": 2.0481109619140624e-05,
      "step": 5369,
      "training_step_time": 10.31291937828064
    },
    {
      "epoch": 2.048492431640625e-05,
      "grad_norm": 0.6965805292129517,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 5370
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 3.181396722793579,
      "step": 5370
    },
    {
      "dynamics_loss_avg": 0.05613609924912453,
      "epoch": 2.048492431640625e-05,
      "step": 5370
    },
    {
      "action_loss_avg": 0.02183223990723491,
      "epoch": 2.048492431640625e-05,
      "step": 5370
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 5370,
      "training_step_time": 9.955567836761475
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 3.138375759124756,
      "step": 5370
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 5370,
      "training_step_time": 9.921264171600342
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 3.1466102600097656,
      "step": 5370
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 5370,
      "training_step_time": 9.92691445350647
    },
    {
      "epoch": 2.048492431640625e-05,
      "model_forward_time": 3.134917974472046,
      "step": 5370
    },
    {
      "epoch": 2.048492431640625e-05,
      "step": 5370,
      "training_step_time": 10.271576881408691
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 3.4655098915100098,
      "step": 5371
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 5371,
      "training_step_time": 10.258045434951782
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 3.137787103652954,
      "step": 5371
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 5371,
      "training_step_time": 9.920003175735474
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 3.143287181854248,
      "step": 5371
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 5371,
      "training_step_time": 9.92223048210144
    },
    {
      "epoch": 2.0488739013671877e-05,
      "model_forward_time": 3.139875888824463,
      "step": 5371
    },
    {
      "epoch": 2.0488739013671877e-05,
      "step": 5371,
      "training_step_time": 10.04141116142273
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 3.1671671867370605,
      "step": 5372
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 5372,
      "training_step_time": 9.947705030441284
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 3.413571834564209,
      "step": 5372
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 5372,
      "training_step_time": 10.19016432762146
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 3.1419074535369873,
      "step": 5372
    },
    {
      "dynamics_loss_avg": 0.047580782882869245,
      "epoch": 2.04925537109375e-05,
      "step": 5372
    },
    {
      "action_loss_avg": 0.018993106205016375,
      "epoch": 2.04925537109375e-05,
      "step": 5372
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 5372,
      "training_step_time": 9.92524003982544
    },
    {
      "epoch": 2.04925537109375e-05,
      "model_forward_time": 3.1390509605407715,
      "step": 5372
    },
    {
      "epoch": 2.04925537109375e-05,
      "step": 5372,
      "training_step_time": 10.017165660858154
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 3.1807031631469727,
      "step": 5373
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 5373,
      "training_step_time": 9.96161437034607
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 3.135392665863037,
      "step": 5373
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 5373,
      "training_step_time": 9.919162511825562
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 3.4195940494537354,
      "step": 5373
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 5373,
      "training_step_time": 10.202025413513184
    },
    {
      "epoch": 2.0496368408203126e-05,
      "model_forward_time": 3.1366848945617676,
      "step": 5373
    },
    {
      "epoch": 2.0496368408203126e-05,
      "step": 5373,
      "training_step_time": 10.281946897506714
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 3.1807029247283936,
      "step": 5374
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 5374,
      "training_step_time": 9.967059850692749
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 3.1380462646484375,
      "step": 5374
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 5374,
      "training_step_time": 9.91459059715271
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 3.1257545948028564,
      "step": 5374
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 5374,
      "training_step_time": 9.899728298187256
    },
    {
      "epoch": 2.050018310546875e-05,
      "model_forward_time": 3.382584810256958,
      "step": 5374
    },
    {
      "epoch": 2.050018310546875e-05,
      "step": 5374,
      "training_step_time": 10.269428253173828
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 3.1777150630950928,
      "step": 5375
    },
    {
      "dynamics_loss_avg": 0.04905320387333632,
      "epoch": 2.0503997802734375e-05,
      "step": 5375
    },
    {
      "action_loss_avg": 0.02253899918869138,
      "epoch": 2.0503997802734375e-05,
      "step": 5375
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 5375,
      "training_step_time": 9.955534219741821
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 3.12754487991333,
      "step": 5375
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 5375,
      "training_step_time": 9.909590482711792
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 3.1478211879730225,
      "step": 5375
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 5375,
      "training_step_time": 9.927560567855835
    },
    {
      "epoch": 2.0503997802734375e-05,
      "model_forward_time": 3.1561434268951416,
      "step": 5375
    },
    {
      "epoch": 2.0503997802734375e-05,
      "step": 5375,
      "training_step_time": 10.241408109664917
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 3.451307773590088,
      "step": 5376
    },
    {
      "epoch": 2.05078125e-05,
      "step": 5376,
      "training_step_time": 10.23592209815979
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 3.1499149799346924,
      "step": 5376
    },
    {
      "epoch": 2.05078125e-05,
      "step": 5376,
      "training_step_time": 9.934142351150513
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 3.1425840854644775,
      "step": 5376
    },
    {
      "epoch": 2.05078125e-05,
      "step": 5376,
      "training_step_time": 9.9197359085083
    },
    {
      "epoch": 2.05078125e-05,
      "model_forward_time": 3.1469743251800537,
      "step": 5376
    },
    {
      "epoch": 2.05078125e-05,
      "step": 5376,
      "training_step_time": 10.260334253311157
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 3.177633762359619,
      "step": 5377
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 5377,
      "training_step_time": 9.951384782791138
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 3.4130606651306152,
      "step": 5377
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 5377,
      "training_step_time": 10.183900594711304
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 3.1339635848999023,
      "step": 5377
    },
    {
      "dynamics_loss_avg": 0.07370050847530366,
      "epoch": 2.0511627197265624e-05,
      "step": 5377
    },
    {
      "action_loss_avg": 0.03111965851858258,
      "epoch": 2.0511627197265624e-05,
      "step": 5377
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 5377,
      "training_step_time": 9.910296201705933
    },
    {
      "epoch": 2.0511627197265624e-05,
      "model_forward_time": 3.136472463607788,
      "step": 5377
    },
    {
      "epoch": 2.0511627197265624e-05,
      "step": 5377,
      "training_step_time": 10.018711805343628
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 3.1777918338775635,
      "step": 5378
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 5378,
      "training_step_time": 9.955898761749268
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 3.1378421783447266,
      "step": 5378
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 5378,
      "training_step_time": 9.918421030044556
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 3.3937106132507324,
      "step": 5378
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 5378,
      "training_step_time": 10.175403118133545
    },
    {
      "epoch": 2.051544189453125e-05,
      "model_forward_time": 3.1426424980163574,
      "step": 5378
    },
    {
      "epoch": 2.051544189453125e-05,
      "step": 5378,
      "training_step_time": 10.341402769088745
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 3.186659097671509,
      "step": 5379
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 5379,
      "training_step_time": 9.97365689277649
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 3.1384294033050537,
      "step": 5379
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 5379,
      "training_step_time": 9.921857118606567
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 3.1430859565734863,
      "step": 5379
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 5379,
      "training_step_time": 9.930261135101318
    },
    {
      "epoch": 2.0519256591796873e-05,
      "model_forward_time": 3.457915782928467,
      "step": 5379
    },
    {
      "epoch": 2.0519256591796873e-05,
      "step": 5379,
      "training_step_time": 10.333284378051758
    },
    {
      "epoch": 2.05230712890625e-05,
      "grad_norm": 0.3694950342178345,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 5380
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 3.1737618446350098,
      "step": 5380
    },
    {
      "dynamics_loss_avg": 0.05255249887704849,
      "epoch": 2.05230712890625e-05,
      "step": 5380
    },
    {
      "action_loss_avg": 0.02434946075081825,
      "epoch": 2.05230712890625e-05,
      "step": 5380
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 5380,
      "training_step_time": 9.957728147506714
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 3.1367998123168945,
      "step": 5380
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 5380,
      "training_step_time": 9.91662335395813
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 3.144213914871216,
      "step": 5380
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 5380,
      "training_step_time": 9.924283981323242
    },
    {
      "epoch": 2.05230712890625e-05,
      "model_forward_time": 3.141181230545044,
      "step": 5380
    },
    {
      "epoch": 2.05230712890625e-05,
      "step": 5380,
      "training_step_time": 10.277838468551636
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 3.4626846313476562,
      "step": 5381
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 5381,
      "training_step_time": 10.246267318725586
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 3.141211748123169,
      "step": 5381
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 5381,
      "training_step_time": 9.921940088272095
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 3.142115831375122,
      "step": 5381
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 5381,
      "training_step_time": 9.923921823501587
    },
    {
      "epoch": 2.0526885986328126e-05,
      "model_forward_time": 3.138965368270874,
      "step": 5381
    },
    {
      "epoch": 2.0526885986328126e-05,
      "step": 5381,
      "training_step_time": 10.278611421585083
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 3.179551362991333,
      "step": 5382
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 5382,
      "training_step_time": 9.962408781051636
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 3.3946704864501953,
      "step": 5382
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 5382,
      "training_step_time": 10.17852234840393
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 3.1448490619659424,
      "step": 5382
    },
    {
      "dynamics_loss_avg": 0.07778085581958294,
      "epoch": 2.053070068359375e-05,
      "step": 5382
    },
    {
      "action_loss_avg": 0.03591735763475299,
      "epoch": 2.053070068359375e-05,
      "step": 5382
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 5382,
      "training_step_time": 9.932469367980957
    },
    {
      "epoch": 2.053070068359375e-05,
      "model_forward_time": 3.143151044845581,
      "step": 5382
    },
    {
      "epoch": 2.053070068359375e-05,
      "step": 5382,
      "training_step_time": 10.332555770874023
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 3.164527416229248,
      "step": 5383
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 5383,
      "training_step_time": 9.951340436935425
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 3.144859552383423,
      "step": 5383
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 5383,
      "training_step_time": 9.92951488494873
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 3.4173543453216553,
      "step": 5383
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 5383,
      "training_step_time": 10.201542854309082
    },
    {
      "epoch": 2.0534515380859375e-05,
      "model_forward_time": 3.138928174972534,
      "step": 5383
    },
    {
      "epoch": 2.0534515380859375e-05,
      "step": 5383,
      "training_step_time": 10.304323673248291
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 3.188509702682495,
      "step": 5384
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 5384,
      "training_step_time": 9.972548484802246
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 3.1372084617614746,
      "step": 5384
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 5384,
      "training_step_time": 9.918872594833374
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 3.137395143508911,
      "step": 5384
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 5384,
      "training_step_time": 9.925541639328003
    },
    {
      "epoch": 2.0538330078125e-05,
      "model_forward_time": 3.3903040885925293,
      "step": 5384
    },
    {
      "epoch": 2.0538330078125e-05,
      "step": 5384,
      "training_step_time": 10.264997720718384
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 3.1765544414520264,
      "step": 5385
    },
    {
      "dynamics_loss_avg": 0.05330022443085909,
      "epoch": 2.0542144775390624e-05,
      "step": 5385
    },
    {
      "action_loss_avg": 0.027142384089529514,
      "epoch": 2.0542144775390624e-05,
      "step": 5385
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 5385,
      "training_step_time": 9.964205265045166
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 3.143812417984009,
      "step": 5385
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 5385,
      "training_step_time": 9.924934148788452
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 3.133707284927368,
      "step": 5385
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 5385,
      "training_step_time": 9.91287088394165
    },
    {
      "epoch": 2.0542144775390624e-05,
      "model_forward_time": 3.133838653564453,
      "step": 5385
    },
    {
      "epoch": 2.0542144775390624e-05,
      "step": 5385,
      "training_step_time": 10.022637128829956
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 3.469404935836792,
      "step": 5386
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 5386,
      "training_step_time": 10.24997854232788
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 3.1387197971343994,
      "step": 5386
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 5386,
      "training_step_time": 9.922061443328857
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 3.1369917392730713,
      "step": 5386
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 5386,
      "training_step_time": 9.920774936676025
    },
    {
      "epoch": 2.054595947265625e-05,
      "model_forward_time": 3.1413912773132324,
      "step": 5386
    },
    {
      "epoch": 2.054595947265625e-05,
      "step": 5386,
      "training_step_time": 10.31185507774353
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 3.18072509765625,
      "step": 5387
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 5387,
      "training_step_time": 9.96258020401001
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 3.4290878772735596,
      "step": 5387
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 5387,
      "training_step_time": 10.211057186126709
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 3.1368772983551025,
      "step": 5387
    },
    {
      "dynamics_loss_avg": 0.051341095194220546,
      "epoch": 2.0549774169921873e-05,
      "step": 5387
    },
    {
      "action_loss_avg": 0.025361235719174147,
      "epoch": 2.0549774169921873e-05,
      "step": 5387
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 5387,
      "training_step_time": 9.919234037399292
    },
    {
      "epoch": 2.0549774169921873e-05,
      "model_forward_time": 3.140291690826416,
      "step": 5387
    },
    {
      "epoch": 2.0549774169921873e-05,
      "step": 5387,
      "training_step_time": 10.247590065002441
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 3.1944401264190674,
      "step": 5388
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 5388,
      "training_step_time": 9.973695278167725
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 3.140078067779541,
      "step": 5388
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 5388,
      "training_step_time": 9.92061448097229
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 3.4156455993652344,
      "step": 5388
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 5388,
      "training_step_time": 10.190573930740356
    },
    {
      "epoch": 2.05535888671875e-05,
      "model_forward_time": 3.1354098320007324,
      "step": 5388
    },
    {
      "epoch": 2.05535888671875e-05,
      "step": 5388,
      "training_step_time": 10.004355669021606
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 3.179316520690918,
      "step": 5389
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 5389,
      "training_step_time": 9.956657648086548
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 3.1315526962280273,
      "step": 5389
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 5389,
      "training_step_time": 9.91154670715332
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 3.1384427547454834,
      "step": 5389
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 5389,
      "training_step_time": 9.914668798446655
    },
    {
      "epoch": 2.0557403564453126e-05,
      "model_forward_time": 3.413559675216675,
      "step": 5389
    },
    {
      "epoch": 2.0557403564453126e-05,
      "step": 5389,
      "training_step_time": 10.307372808456421
    },
    {
      "epoch": 2.056121826171875e-05,
      "grad_norm": 0.8004908561706543,
      "learning_rate": 1e-05,
      "loss": 0.091,
      "step": 5390
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 3.1734399795532227,
      "step": 5390
    },
    {
      "dynamics_loss_avg": 0.06601052414625883,
      "epoch": 2.056121826171875e-05,
      "step": 5390
    },
    {
      "action_loss_avg": 0.031202217657119037,
      "epoch": 2.056121826171875e-05,
      "step": 5390
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 5390,
      "training_step_time": 9.957510948181152
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 3.133392333984375,
      "step": 5390
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 5390,
      "training_step_time": 9.914106130599976
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 3.138430595397949,
      "step": 5390
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 5390,
      "training_step_time": 9.910272121429443
    },
    {
      "epoch": 2.056121826171875e-05,
      "model_forward_time": 3.133817672729492,
      "step": 5390
    },
    {
      "epoch": 2.056121826171875e-05,
      "step": 5390,
      "training_step_time": 10.011010646820068
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 3.4771456718444824,
      "step": 5391
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 5391,
      "training_step_time": 10.254729986190796
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 3.133945941925049,
      "step": 5391
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 5391,
      "training_step_time": 9.911099672317505
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 3.138859748840332,
      "step": 5391
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 5391,
      "training_step_time": 9.918868780136108
    },
    {
      "epoch": 2.0565032958984375e-05,
      "model_forward_time": 3.1395699977874756,
      "step": 5391
    },
    {
      "epoch": 2.0565032958984375e-05,
      "step": 5391,
      "training_step_time": 10.289985179901123
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 3.184274911880493,
      "step": 5392
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 5392,
      "training_step_time": 9.96671438217163
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 3.386726140975952,
      "step": 5392
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 5392,
      "training_step_time": 10.164745807647705
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 3.141258955001831,
      "step": 5392
    },
    {
      "dynamics_loss_avg": 0.06190593261271715,
      "epoch": 2.056884765625e-05,
      "step": 5392
    },
    {
      "action_loss_avg": 0.02598538910970092,
      "epoch": 2.056884765625e-05,
      "step": 5392
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 5392,
      "training_step_time": 9.92225170135498
    },
    {
      "epoch": 2.056884765625e-05,
      "model_forward_time": 3.1351194381713867,
      "step": 5392
    },
    {
      "epoch": 2.056884765625e-05,
      "step": 5392,
      "training_step_time": 10.30517029762268
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 3.1749370098114014,
      "step": 5393
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 5393,
      "training_step_time": 9.955049514770508
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 3.138944625854492,
      "step": 5393
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 5393,
      "training_step_time": 10.191761255264282
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 3.1389286518096924,
      "step": 5393
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 5393,
      "training_step_time": 9.920189619064331
    },
    {
      "epoch": 2.0572662353515624e-05,
      "model_forward_time": 3.1345698833465576,
      "step": 5393
    },
    {
      "epoch": 2.0572662353515624e-05,
      "step": 5393,
      "training_step_time": 10.26302456855774
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 3.1558709144592285,
      "step": 5394
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 5394,
      "training_step_time": 9.938173532485962
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 3.1411445140838623,
      "step": 5394
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 5394,
      "training_step_time": 9.922470092773438
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 3.424302339553833,
      "step": 5394
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 5394,
      "training_step_time": 10.201827049255371
    },
    {
      "epoch": 2.057647705078125e-05,
      "model_forward_time": 3.1408042907714844,
      "step": 5394
    },
    {
      "epoch": 2.057647705078125e-05,
      "step": 5394,
      "training_step_time": 10.302063226699829
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 3.2021164894104004,
      "step": 5395
    },
    {
      "dynamics_loss_avg": 0.05646068304777145,
      "epoch": 2.0580291748046873e-05,
      "step": 5395
    },
    {
      "action_loss_avg": 0.025796103477478027,
      "epoch": 2.0580291748046873e-05,
      "step": 5395
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 5395,
      "training_step_time": 9.990001678466797
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 3.142357587814331,
      "step": 5395
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 5395,
      "training_step_time": 9.9197096824646
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 3.137934923171997,
      "step": 5395
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 5395,
      "training_step_time": 9.91632890701294
    },
    {
      "epoch": 2.0580291748046873e-05,
      "model_forward_time": 3.1418867111206055,
      "step": 5395
    },
    {
      "epoch": 2.0580291748046873e-05,
      "step": 5395,
      "training_step_time": 10.319998741149902
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 3.192507266998291,
      "step": 5396
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 5396,
      "training_step_time": 9.968004941940308
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 3.1384971141815186,
      "step": 5396
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 5396,
      "training_step_time": 9.918992757797241
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 3.138819932937622,
      "step": 5396
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 5396,
      "training_step_time": 9.909868717193604
    },
    {
      "epoch": 2.05841064453125e-05,
      "model_forward_time": 3.1364049911499023,
      "step": 5396
    },
    {
      "epoch": 2.05841064453125e-05,
      "step": 5396,
      "training_step_time": 10.251887083053589
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 3.4325709342956543,
      "step": 5397
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 5397,
      "training_step_time": 10.208621501922607
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 3.1380016803741455,
      "step": 5397
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 5397,
      "training_step_time": 9.909617185592651
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 3.1323611736297607,
      "step": 5397
    },
    {
      "dynamics_loss_avg": 0.055799487978219986,
      "epoch": 2.0587921142578126e-05,
      "step": 5397
    },
    {
      "action_loss_avg": 0.023893168848007917,
      "epoch": 2.0587921142578126e-05,
      "step": 5397
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 5397,
      "training_step_time": 9.90850043296814
    },
    {
      "epoch": 2.0587921142578126e-05,
      "model_forward_time": 3.1336100101470947,
      "step": 5397
    },
    {
      "epoch": 2.0587921142578126e-05,
      "step": 5397,
      "training_step_time": 10.253766536712646
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 3.1792073249816895,
      "step": 5398
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 5398,
      "training_step_time": 9.955430030822754
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 3.412909984588623,
      "step": 5398
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 5398,
      "training_step_time": 10.189989566802979
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 3.1351771354675293,
      "step": 5398
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 5398,
      "training_step_time": 9.912473440170288
    },
    {
      "epoch": 2.059173583984375e-05,
      "model_forward_time": 3.138641119003296,
      "step": 5398
    },
    {
      "epoch": 2.059173583984375e-05,
      "step": 5398,
      "training_step_time": 10.267448902130127
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 3.174487590789795,
      "step": 5399
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 5399,
      "training_step_time": 9.953554630279541
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 3.138159990310669,
      "step": 5399
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 5399,
      "training_step_time": 9.915521383285522
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 3.3948981761932373,
      "step": 5399
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 5399,
      "training_step_time": 10.173351764678955
    },
    {
      "epoch": 2.0595550537109375e-05,
      "model_forward_time": 3.1414196491241455,
      "step": 5399
    },
    {
      "epoch": 2.0595550537109375e-05,
      "step": 5399,
      "training_step_time": 10.304131031036377
    },
    {
      "epoch": 2.0599365234375e-05,
      "grad_norm": 0.35406532883644104,
      "learning_rate": 1e-05,
      "loss": 0.0822,
      "step": 5400
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 3.177989959716797,
      "step": 5400
    },
    {
      "dynamics_loss_avg": 0.06242747195065022,
      "epoch": 2.0599365234375e-05,
      "step": 5400
    },
    {
      "action_loss_avg": 0.027391523029655217,
      "epoch": 2.0599365234375e-05,
      "step": 5400
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 5400,
      "training_step_time": 9.962935209274292
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 3.137812614440918,
      "step": 5400
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 5400,
      "training_step_time": 9.917954444885254
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 3.139815092086792,
      "step": 5400
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 5400,
      "training_step_time": 9.925275087356567
    },
    {
      "epoch": 2.0599365234375e-05,
      "model_forward_time": 3.423764705657959,
      "step": 5400
    },
    {
      "epoch": 2.0599365234375e-05,
      "step": 5400,
      "training_step_time": 10.306797504425049
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 3.190584182739258,
      "step": 5401
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 5401,
      "training_step_time": 9.97411561012268
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 3.1455883979797363,
      "step": 5401
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 5401,
      "training_step_time": 9.934700012207031
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 3.1492233276367188,
      "step": 5401
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 5401,
      "training_step_time": 9.93252158164978
    },
    {
      "epoch": 2.0603179931640624e-05,
      "model_forward_time": 3.146893262863159,
      "step": 5401
    },
    {
      "epoch": 2.0603179931640624e-05,
      "step": 5401,
      "training_step_time": 10.288646221160889
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 3.1888811588287354,
      "step": 5402
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 5402,
      "training_step_time": 10.296354293823242
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 3.142646551132202,
      "step": 5402
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 5402,
      "training_step_time": 9.924403429031372
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 3.1451032161712646,
      "step": 5402
    },
    {
      "dynamics_loss_avg": 0.05554027445614338,
      "epoch": 2.060699462890625e-05,
      "step": 5402
    },
    {
      "action_loss_avg": 0.021462392574176193,
      "epoch": 2.060699462890625e-05,
      "step": 5402
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 5402,
      "training_step_time": 9.929490804672241
    },
    {
      "epoch": 2.060699462890625e-05,
      "model_forward_time": 3.138814687728882,
      "step": 5402
    },
    {
      "epoch": 2.060699462890625e-05,
      "step": 5402,
      "training_step_time": 10.01288390159607
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 3.1779870986938477,
      "step": 5403
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 5403,
      "training_step_time": 9.958398818969727
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 3.408564329147339,
      "step": 5403
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 5403,
      "training_step_time": 10.19593596458435
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 3.14896559715271,
      "step": 5403
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 5403,
      "training_step_time": 9.93067479133606
    },
    {
      "epoch": 2.0610809326171873e-05,
      "model_forward_time": 3.1390998363494873,
      "step": 5403
    },
    {
      "epoch": 2.0610809326171873e-05,
      "step": 5403,
      "training_step_time": 10.256811141967773
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 3.1839959621429443,
      "step": 5404
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 5404,
      "training_step_time": 9.963093519210815
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 3.142889976501465,
      "step": 5404
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 5404,
      "training_step_time": 9.925787687301636
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 3.1388368606567383,
      "step": 5404
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 5404,
      "training_step_time": 10.211358785629272
    },
    {
      "epoch": 2.06146240234375e-05,
      "model_forward_time": 3.1462340354919434,
      "step": 5404
    },
    {
      "epoch": 2.06146240234375e-05,
      "step": 5404,
      "training_step_time": 10.33465313911438
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 3.1683008670806885,
      "step": 5405
    },
    {
      "dynamics_loss_avg": 0.05947308987379074,
      "epoch": 2.0618438720703126e-05,
      "step": 5405
    },
    {
      "action_loss_avg": 0.026796854054555296,
      "epoch": 2.0618438720703126e-05,
      "step": 5405
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 5405,
      "training_step_time": 9.954634189605713
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 3.146027088165283,
      "step": 5405
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 5405,
      "training_step_time": 9.932903051376343
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 3.1412971019744873,
      "step": 5405
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 5405,
      "training_step_time": 9.93075680732727
    },
    {
      "epoch": 2.0618438720703126e-05,
      "model_forward_time": 3.4113495349884033,
      "step": 5405
    },
    {
      "epoch": 2.0618438720703126e-05,
      "step": 5405,
      "training_step_time": 10.29287338256836
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 3.191301107406616,
      "step": 5406
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 5406,
      "training_step_time": 9.977832555770874
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 3.150254726409912,
      "step": 5406
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 5406,
      "training_step_time": 9.940209150314331
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 3.150346279144287,
      "step": 5406
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 5406,
      "training_step_time": 9.937318325042725
    },
    {
      "epoch": 2.062225341796875e-05,
      "model_forward_time": 3.151494264602661,
      "step": 5406
    },
    {
      "epoch": 2.062225341796875e-05,
      "step": 5406,
      "training_step_time": 10.316270589828491
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 3.183382034301758,
      "step": 5407
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 5407,
      "training_step_time": 10.239378213882446
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 3.148383140563965,
      "step": 5407
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 5407,
      "training_step_time": 9.924301624298096
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 3.1422810554504395,
      "step": 5407
    },
    {
      "dynamics_loss_avg": 0.07335128709673881,
      "epoch": 2.0626068115234375e-05,
      "step": 5407
    },
    {
      "action_loss_avg": 0.03274753745645285,
      "epoch": 2.0626068115234375e-05,
      "step": 5407
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 5407,
      "training_step_time": 9.927764654159546
    },
    {
      "epoch": 2.0626068115234375e-05,
      "model_forward_time": 3.148104429244995,
      "step": 5407
    },
    {
      "epoch": 2.0626068115234375e-05,
      "step": 5407,
      "training_step_time": 10.021326541900635
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 3.1925792694091797,
      "step": 5408
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 5408,
      "training_step_time": 9.972578287124634
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 3.419072389602661,
      "step": 5408
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 5408,
      "training_step_time": 10.200085401535034
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 3.1404788494110107,
      "step": 5408
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 5408,
      "training_step_time": 9.921847581863403
    },
    {
      "epoch": 2.06298828125e-05,
      "model_forward_time": 3.143317222595215,
      "step": 5408
    },
    {
      "epoch": 2.06298828125e-05,
      "step": 5408,
      "training_step_time": 10.039334535598755
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 3.1744937896728516,
      "step": 5409
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 5409,
      "training_step_time": 9.962057113647461
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 3.1385385990142822,
      "step": 5409
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 5409,
      "training_step_time": 9.920785665512085
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 2.436269760131836,
      "step": 5409
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 5409,
      "training_step_time": 7.747655868530273
    },
    {
      "epoch": 2.0633697509765624e-05,
      "model_forward_time": 3.139636993408203,
      "step": 5409
    },
    {
      "epoch": 2.0633697509765624e-05,
      "step": 5409,
      "training_step_time": 10.301058769226074
    },
    {
      "epoch": 2.063751220703125e-05,
      "grad_norm": 0.48393428325653076,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 5410
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 3.1855156421661377,
      "step": 5410
    },
    {
      "dynamics_loss_avg": 0.06618050709366799,
      "epoch": 2.063751220703125e-05,
      "step": 5410
    },
    {
      "action_loss_avg": 0.024711539410054682,
      "epoch": 2.063751220703125e-05,
      "step": 5410
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 5410,
      "training_step_time": 9.967364311218262
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 3.1356871128082275,
      "step": 5410
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 5410,
      "training_step_time": 9.918229103088379
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 3.142078161239624,
      "step": 5410
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 5410,
      "training_step_time": 9.920867204666138
    },
    {
      "epoch": 2.063751220703125e-05,
      "model_forward_time": 3.4549942016601562,
      "step": 5410
    },
    {
      "epoch": 2.063751220703125e-05,
      "step": 5410,
      "training_step_time": 10.327264070510864
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 3.185260534286499,
      "step": 5411
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 5411,
      "training_step_time": 9.967336177825928
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 3.136200428009033,
      "step": 5411
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 5411,
      "training_step_time": 9.91738748550415
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 3.144340991973877,
      "step": 5411
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 5411,
      "training_step_time": 9.92443323135376
    },
    {
      "epoch": 2.0641326904296873e-05,
      "model_forward_time": 3.1386802196502686,
      "step": 5411
    },
    {
      "epoch": 2.0641326904296873e-05,
      "step": 5411,
      "training_step_time": 10.303642511367798
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 3.4614365100860596,
      "step": 5412
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 5412,
      "training_step_time": 10.240903854370117
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 3.137192726135254,
      "step": 5412
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 5412,
      "training_step_time": 9.922289609909058
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 3.139845371246338,
      "step": 5412
    },
    {
      "dynamics_loss_avg": 0.07098517958074808,
      "epoch": 2.06451416015625e-05,
      "step": 5412
    },
    {
      "action_loss_avg": 0.024600836215540767,
      "epoch": 2.06451416015625e-05,
      "step": 5412
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 5412,
      "training_step_time": 9.921941757202148
    },
    {
      "epoch": 2.06451416015625e-05,
      "model_forward_time": 3.135545015335083,
      "step": 5412
    },
    {
      "epoch": 2.06451416015625e-05,
      "step": 5412,
      "training_step_time": 10.275807619094849
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 3.17149019241333,
      "step": 5413
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 5413,
      "training_step_time": 9.950744152069092
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 3.3991551399230957,
      "step": 5413
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 5413,
      "training_step_time": 10.170865058898926
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 3.1311588287353516,
      "step": 5413
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 5413,
      "training_step_time": 9.90043568611145
    },
    {
      "epoch": 2.0648956298828126e-05,
      "model_forward_time": 3.127596378326416,
      "step": 5413
    },
    {
      "epoch": 2.0648956298828126e-05,
      "step": 5413,
      "training_step_time": 10.00668740272522
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 3.174995183944702,
      "step": 5414
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 5414,
      "training_step_time": 9.94821572303772
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 3.1369452476501465,
      "step": 5414
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 5414,
      "training_step_time": 9.9092538356781
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 3.415987968444824,
      "step": 5414
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 5414,
      "training_step_time": 10.192546844482422
    },
    {
      "epoch": 2.065277099609375e-05,
      "model_forward_time": 3.1369454860687256,
      "step": 5414
    },
    {
      "epoch": 2.065277099609375e-05,
      "step": 5414,
      "training_step_time": 10.013919353485107
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 3.1679153442382812,
      "step": 5415
    },
    {
      "dynamics_loss_avg": 0.07133018337190151,
      "epoch": 2.0656585693359375e-05,
      "step": 5415
    },
    {
      "action_loss_avg": 0.03293945845216513,
      "epoch": 2.0656585693359375e-05,
      "step": 5415
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 5415,
      "training_step_time": 9.944889068603516
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 3.1371026039123535,
      "step": 5415
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 5415,
      "training_step_time": 9.9174485206604
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 3.1363236904144287,
      "step": 5415
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 5415,
      "training_step_time": 9.921000003814697
    },
    {
      "epoch": 2.0656585693359375e-05,
      "model_forward_time": 3.394129514694214,
      "step": 5415
    },
    {
      "epoch": 2.0656585693359375e-05,
      "step": 5415,
      "training_step_time": 10.301310539245605
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 3.1835150718688965,
      "step": 5416
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 5416,
      "training_step_time": 9.967216730117798
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 3.144035577774048,
      "step": 5416
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 5416,
      "training_step_time": 9.927403688430786
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 3.1405856609344482,
      "step": 5416
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 5416,
      "training_step_time": 9.922263383865356
    },
    {
      "epoch": 2.0660400390625e-05,
      "model_forward_time": 3.1427016258239746,
      "step": 5416
    },
    {
      "epoch": 2.0660400390625e-05,
      "step": 5416,
      "training_step_time": 10.2524733543396
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 3.4221630096435547,
      "step": 5417
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 5417,
      "training_step_time": 10.201456308364868
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 3.143446683883667,
      "step": 5417
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 5417,
      "training_step_time": 9.92969298362732
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 3.136676073074341,
      "step": 5417
    },
    {
      "dynamics_loss_avg": 0.056628689356148246,
      "epoch": 2.0664215087890624e-05,
      "step": 5417
    },
    {
      "action_loss_avg": 0.023658467922359705,
      "epoch": 2.0664215087890624e-05,
      "step": 5417
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 5417,
      "training_step_time": 9.923120498657227
    },
    {
      "epoch": 2.0664215087890624e-05,
      "model_forward_time": 3.1426608562469482,
      "step": 5417
    },
    {
      "epoch": 2.0664215087890624e-05,
      "step": 5417,
      "training_step_time": 10.260717868804932
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 3.182495594024658,
      "step": 5418
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 5418,
      "training_step_time": 9.96273159980774
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 3.4197707176208496,
      "step": 5418
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 5418,
      "training_step_time": 10.194607257843018
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 3.1341755390167236,
      "step": 5418
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 5418,
      "training_step_time": 9.90986442565918
    },
    {
      "epoch": 2.066802978515625e-05,
      "model_forward_time": 3.138867139816284,
      "step": 5418
    },
    {
      "epoch": 2.066802978515625e-05,
      "step": 5418,
      "training_step_time": 10.01203966140747
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 3.1752121448516846,
      "step": 5419
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 5419,
      "training_step_time": 9.949229001998901
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 3.134256362915039,
      "step": 5419
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 5419,
      "training_step_time": 9.910419940948486
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 3.3932831287384033,
      "step": 5419
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 5419,
      "training_step_time": 10.172605276107788
    },
    {
      "epoch": 2.0671844482421873e-05,
      "model_forward_time": 3.1389048099517822,
      "step": 5419
    },
    {
      "epoch": 2.0671844482421873e-05,
      "step": 5419,
      "training_step_time": 10.033539772033691
    },
    {
      "epoch": 2.06756591796875e-05,
      "grad_norm": 0.599321722984314,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 5420
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 3.183871030807495,
      "step": 5420
    },
    {
      "dynamics_loss_avg": 0.05711125042289496,
      "epoch": 2.06756591796875e-05,
      "step": 5420
    },
    {
      "action_loss_avg": 0.030141236260533333,
      "epoch": 2.06756591796875e-05,
      "step": 5420
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 5420,
      "training_step_time": 9.965538740158081
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 3.137232542037964,
      "step": 5420
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 5420,
      "training_step_time": 9.920044422149658
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 3.140643358230591,
      "step": 5420
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 5420,
      "training_step_time": 9.921319484710693
    },
    {
      "epoch": 2.06756591796875e-05,
      "model_forward_time": 3.423264741897583,
      "step": 5420
    },
    {
      "epoch": 2.06756591796875e-05,
      "step": 5420,
      "training_step_time": 10.332703351974487
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 3.1900882720947266,
      "step": 5421
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 5421,
      "training_step_time": 9.975646734237671
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 3.1378743648529053,
      "step": 5421
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 5421,
      "training_step_time": 9.916156768798828
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 3.140714406967163,
      "step": 5421
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 5421,
      "training_step_time": 9.921995639801025
    },
    {
      "epoch": 2.0679473876953126e-05,
      "model_forward_time": 3.1410539150238037,
      "step": 5421
    },
    {
      "epoch": 2.0679473876953126e-05,
      "step": 5421,
      "training_step_time": 10.269656419754028
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 3.1666905879974365,
      "step": 5422
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 5422,
      "training_step_time": 10.231596231460571
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 3.140861749649048,
      "step": 5422
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 5422,
      "training_step_time": 9.925376892089844
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 3.142535924911499,
      "step": 5422
    },
    {
      "dynamics_loss_avg": 0.07678388357162476,
      "epoch": 2.068328857421875e-05,
      "step": 5422
    },
    {
      "action_loss_avg": 0.03215730134397745,
      "epoch": 2.068328857421875e-05,
      "step": 5422
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 5422,
      "training_step_time": 9.923216342926025
    },
    {
      "epoch": 2.068328857421875e-05,
      "model_forward_time": 3.1348812580108643,
      "step": 5422
    },
    {
      "epoch": 2.068328857421875e-05,
      "step": 5422,
      "training_step_time": 10.312494039535522
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 3.1801440715789795,
      "step": 5423
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 5423,
      "training_step_time": 9.95897102355957
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 3.3976025581359863,
      "step": 5423
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 5423,
      "training_step_time": 10.16680645942688
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 3.1296467781066895,
      "step": 5423
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 5423,
      "training_step_time": 9.89803171157837
    },
    {
      "epoch": 2.0687103271484375e-05,
      "model_forward_time": 3.129774570465088,
      "step": 5423
    },
    {
      "epoch": 2.0687103271484375e-05,
      "step": 5423,
      "training_step_time": 10.006972312927246
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 3.165482997894287,
      "step": 5424
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 5424,
      "training_step_time": 9.940009593963623
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 3.132460355758667,
      "step": 5424
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 5424,
      "training_step_time": 9.904500484466553
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 3.138288974761963,
      "step": 5424
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 5424,
      "training_step_time": 10.205407857894897
    },
    {
      "epoch": 2.069091796875e-05,
      "model_forward_time": 2.433515787124634,
      "step": 5424
    },
    {
      "epoch": 2.069091796875e-05,
      "step": 5424,
      "training_step_time": 10.252493143081665
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 3.172215223312378,
      "step": 5425
    },
    {
      "dynamics_loss_avg": 0.058390813693404195,
      "epoch": 2.0694732666015624e-05,
      "step": 5425
    },
    {
      "action_loss_avg": 0.0235905772075057,
      "epoch": 2.0694732666015624e-05,
      "step": 5425
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 5425,
      "training_step_time": 9.951644659042358
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 3.1463520526885986,
      "step": 5425
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 5425,
      "training_step_time": 9.920993566513062
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 3.1397969722747803,
      "step": 5425
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 5425,
      "training_step_time": 9.918835878372192
    },
    {
      "epoch": 2.0694732666015624e-05,
      "model_forward_time": 3.146765947341919,
      "step": 5425
    },
    {
      "epoch": 2.0694732666015624e-05,
      "step": 5425,
      "training_step_time": 10.325934171676636
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 3.172280788421631,
      "step": 5426
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 5426,
      "training_step_time": 9.959295272827148
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 2.4404733180999756,
      "step": 5426
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 5426,
      "training_step_time": 7.473579168319702
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 3.139442205429077,
      "step": 5426
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 5426,
      "training_step_time": 9.925204992294312
    },
    {
      "epoch": 2.069854736328125e-05,
      "model_forward_time": 3.145400285720825,
      "step": 5426
    },
    {
      "epoch": 2.069854736328125e-05,
      "step": 5426,
      "training_step_time": 10.287487983703613
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 3.1821646690368652,
      "step": 5427
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 5427,
      "training_step_time": 9.973467350006104
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 3.4288330078125,
      "step": 5427
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 5427,
      "training_step_time": 10.212243556976318
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 3.1379506587982178,
      "step": 5427
    },
    {
      "dynamics_loss_avg": 0.07466384414583445,
      "epoch": 2.0702362060546873e-05,
      "step": 5427
    },
    {
      "action_loss_avg": 0.03322301618754864,
      "epoch": 2.0702362060546873e-05,
      "step": 5427
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 5427,
      "training_step_time": 9.918485164642334
    },
    {
      "epoch": 2.0702362060546873e-05,
      "model_forward_time": 3.138442277908325,
      "step": 5427
    },
    {
      "epoch": 2.0702362060546873e-05,
      "step": 5427,
      "training_step_time": 10.281855583190918
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 3.1657917499542236,
      "step": 5428
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 5428,
      "training_step_time": 9.944709539413452
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 3.4230432510375977,
      "step": 5428
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 5428,
      "training_step_time": 10.197640657424927
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 3.139596939086914,
      "step": 5428
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 5428,
      "training_step_time": 9.92014455795288
    },
    {
      "epoch": 2.07061767578125e-05,
      "model_forward_time": 3.1420021057128906,
      "step": 5428
    },
    {
      "epoch": 2.07061767578125e-05,
      "step": 5428,
      "training_step_time": 10.267358779907227
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 2.485635995864868,
      "step": 5429
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 5429,
      "training_step_time": 7.519779920578003
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 3.1370065212249756,
      "step": 5429
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 5429,
      "training_step_time": 9.918558359146118
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 3.1402206420898438,
      "step": 5429
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 5429,
      "training_step_time": 10.189335823059082
    },
    {
      "epoch": 2.0709991455078126e-05,
      "model_forward_time": 3.1421010494232178,
      "step": 5429
    },
    {
      "epoch": 2.0709991455078126e-05,
      "step": 5429,
      "training_step_time": 10.008531332015991
    },
    {
      "epoch": 2.071380615234375e-05,
      "grad_norm": 0.5484733581542969,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 5430
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 3.185671091079712,
      "step": 5430
    },
    {
      "dynamics_loss_avg": 0.060549925081431866,
      "epoch": 2.071380615234375e-05,
      "step": 5430
    },
    {
      "action_loss_avg": 0.035223431326448916,
      "epoch": 2.071380615234375e-05,
      "step": 5430
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 5430,
      "training_step_time": 9.96983790397644
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 3.1364572048187256,
      "step": 5430
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 5430,
      "training_step_time": 9.908864259719849
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 3.140368700027466,
      "step": 5430
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 5430,
      "training_step_time": 9.912643671035767
    },
    {
      "epoch": 2.071380615234375e-05,
      "model_forward_time": 3.418731689453125,
      "step": 5430
    },
    {
      "epoch": 2.071380615234375e-05,
      "step": 5430,
      "training_step_time": 10.290083169937134
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 3.1759421825408936,
      "step": 5431
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 5431,
      "training_step_time": 9.949787139892578
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 3.135770082473755,
      "step": 5431
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 5431,
      "training_step_time": 9.913530111312866
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 3.1490190029144287,
      "step": 5431
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 5431,
      "training_step_time": 9.915035724639893
    },
    {
      "epoch": 2.0717620849609375e-05,
      "model_forward_time": 3.1410160064697266,
      "step": 5431
    },
    {
      "epoch": 2.0717620849609375e-05,
      "step": 5431,
      "training_step_time": 10.317123413085938
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 3.4685256481170654,
      "step": 5432
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 5432,
      "training_step_time": 10.237273216247559
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 3.1444613933563232,
      "step": 5432
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 5432,
      "training_step_time": 9.916042566299438
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 3.1457252502441406,
      "step": 5432
    },
    {
      "dynamics_loss_avg": 0.05966966608539224,
      "epoch": 2.0721435546875e-05,
      "step": 5432
    },
    {
      "action_loss_avg": 0.021105903573334216,
      "epoch": 2.0721435546875e-05,
      "step": 5432
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 5432,
      "training_step_time": 9.913477182388306
    },
    {
      "epoch": 2.0721435546875e-05,
      "model_forward_time": 3.1473817825317383,
      "step": 5432
    },
    {
      "epoch": 2.0721435546875e-05,
      "step": 5432,
      "training_step_time": 10.306226968765259
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 3.178069829940796,
      "step": 5433
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 5433,
      "training_step_time": 9.951836109161377
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 3.4048218727111816,
      "step": 5433
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 5433,
      "training_step_time": 10.174812316894531
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 3.146393299102783,
      "step": 5433
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 5433,
      "training_step_time": 9.916977643966675
    },
    {
      "epoch": 2.0725250244140624e-05,
      "model_forward_time": 3.144218683242798,
      "step": 5433
    },
    {
      "epoch": 2.0725250244140624e-05,
      "step": 5433,
      "training_step_time": 10.285939693450928
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 3.1667094230651855,
      "step": 5434
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 5434,
      "training_step_time": 9.937174797058105
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 3.1488089561462402,
      "step": 5434
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 5434,
      "training_step_time": 9.923957109451294
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 3.429616689682007,
      "step": 5434
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 5434,
      "training_step_time": 10.199331521987915
    },
    {
      "epoch": 2.072906494140625e-05,
      "model_forward_time": 3.144109010696411,
      "step": 5434
    },
    {
      "epoch": 2.072906494140625e-05,
      "step": 5434,
      "training_step_time": 10.027234077453613
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 3.1780178546905518,
      "step": 5435
    },
    {
      "dynamics_loss_avg": 0.05185349192470312,
      "epoch": 2.0732879638671874e-05,
      "step": 5435
    },
    {
      "action_loss_avg": 0.02549667013809085,
      "epoch": 2.0732879638671874e-05,
      "step": 5435
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 5435,
      "training_step_time": 9.953734397888184
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 3.1488049030303955,
      "step": 5435
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 5435,
      "training_step_time": 9.918214797973633
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 3.1447300910949707,
      "step": 5435
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 5435,
      "training_step_time": 9.921440839767456
    },
    {
      "epoch": 2.0732879638671874e-05,
      "model_forward_time": 3.4122605323791504,
      "step": 5435
    },
    {
      "epoch": 2.0732879638671874e-05,
      "step": 5435,
      "training_step_time": 10.272157430648804
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 3.1832778453826904,
      "step": 5436
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 5436,
      "training_step_time": 9.95177936553955
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 3.135037660598755,
      "step": 5436
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 5436,
      "training_step_time": 9.898383617401123
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 3.1320717334747314,
      "step": 5436
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 5436,
      "training_step_time": 9.892277956008911
    },
    {
      "epoch": 2.07366943359375e-05,
      "model_forward_time": 3.1312501430511475,
      "step": 5436
    },
    {
      "epoch": 2.07366943359375e-05,
      "step": 5436,
      "training_step_time": 10.253601789474487
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 3.41749906539917,
      "step": 5437
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 5437,
      "training_step_time": 10.182490110397339
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 3.1322147846221924,
      "step": 5437
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 5437,
      "training_step_time": 9.896349906921387
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 3.1376850605010986,
      "step": 5437
    },
    {
      "dynamics_loss_avg": 0.06394884139299392,
      "epoch": 2.0740509033203126e-05,
      "step": 5437
    },
    {
      "action_loss_avg": 0.024050495214760305,
      "epoch": 2.0740509033203126e-05,
      "step": 5437
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 5437,
      "training_step_time": 9.910459995269775
    },
    {
      "epoch": 2.0740509033203126e-05,
      "model_forward_time": 3.1411168575286865,
      "step": 5437
    },
    {
      "epoch": 2.0740509033203126e-05,
      "step": 5437,
      "training_step_time": 10.283376216888428
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 3.1762869358062744,
      "step": 5438
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 5438,
      "training_step_time": 9.947659492492676
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 3.416724443435669,
      "step": 5438
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 5438,
      "training_step_time": 10.185487985610962
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 3.139986276626587,
      "step": 5438
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 5438,
      "training_step_time": 9.909264326095581
    },
    {
      "epoch": 2.074432373046875e-05,
      "model_forward_time": 3.14311146736145,
      "step": 5438
    },
    {
      "epoch": 2.074432373046875e-05,
      "step": 5438,
      "training_step_time": 10.266841173171997
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 3.176354169845581,
      "step": 5439
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 5439,
      "training_step_time": 9.950434446334839
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 3.1353862285614014,
      "step": 5439
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 5439,
      "training_step_time": 9.906088829040527
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 3.3915061950683594,
      "step": 5439
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 5439,
      "training_step_time": 10.162149429321289
    },
    {
      "epoch": 2.0748138427734375e-05,
      "model_forward_time": 3.140995740890503,
      "step": 5439
    },
    {
      "epoch": 2.0748138427734375e-05,
      "step": 5439,
      "training_step_time": 10.284799814224243
    },
    {
      "epoch": 2.0751953125e-05,
      "grad_norm": 0.5738428831100464,
      "learning_rate": 1e-05,
      "loss": 0.0844,
      "step": 5440
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 3.17930006980896,
      "step": 5440
    },
    {
      "dynamics_loss_avg": 0.06046426799148321,
      "epoch": 2.0751953125e-05,
      "step": 5440
    },
    {
      "action_loss_avg": 0.03597675897181034,
      "epoch": 2.0751953125e-05,
      "step": 5440
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 5440,
      "training_step_time": 9.959784746170044
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 3.1340396404266357,
      "step": 5440
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 5440,
      "training_step_time": 9.906326293945312
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 3.1382176876068115,
      "step": 5440
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 5440,
      "training_step_time": 9.908467292785645
    },
    {
      "epoch": 2.0751953125e-05,
      "model_forward_time": 3.419243335723877,
      "step": 5440
    },
    {
      "epoch": 2.0751953125e-05,
      "step": 5440,
      "training_step_time": 10.333508253097534
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 3.1892032623291016,
      "step": 5441
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 5441,
      "training_step_time": 9.967479228973389
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 3.1471915245056152,
      "step": 5441
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 5441,
      "training_step_time": 9.919466972351074
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 3.1466245651245117,
      "step": 5441
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 5441,
      "training_step_time": 9.915905237197876
    },
    {
      "epoch": 2.0755767822265624e-05,
      "model_forward_time": 3.1417391300201416,
      "step": 5441
    },
    {
      "epoch": 2.0755767822265624e-05,
      "step": 5441,
      "training_step_time": 10.010018587112427
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 3.1813313961029053,
      "step": 5442
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 5442,
      "training_step_time": 10.240247249603271
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 3.1416351795196533,
      "step": 5442
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 5442,
      "training_step_time": 9.906700372695923
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 3.1393656730651855,
      "step": 5442
    },
    {
      "dynamics_loss_avg": 0.07252344861626625,
      "epoch": 2.075958251953125e-05,
      "step": 5442
    },
    {
      "action_loss_avg": 0.028709353134036063,
      "epoch": 2.075958251953125e-05,
      "step": 5442
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 5442,
      "training_step_time": 9.912021160125732
    },
    {
      "epoch": 2.075958251953125e-05,
      "model_forward_time": 3.1428122520446777,
      "step": 5442
    },
    {
      "epoch": 2.075958251953125e-05,
      "step": 5442,
      "training_step_time": 10.298580169677734
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 3.1854453086853027,
      "step": 5443
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 5443,
      "training_step_time": 9.95693826675415
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 3.423733949661255,
      "step": 5443
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 5443,
      "training_step_time": 10.192748785018921
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 3.1458852291107178,
      "step": 5443
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 5443,
      "training_step_time": 9.9151132106781
    },
    {
      "epoch": 2.0763397216796874e-05,
      "model_forward_time": 3.1462249755859375,
      "step": 5443
    },
    {
      "epoch": 2.0763397216796874e-05,
      "step": 5443,
      "training_step_time": 10.295254707336426
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 3.2143759727478027,
      "step": 5444
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 5444,
      "training_step_time": 9.999059200286865
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 3.1410231590270996,
      "step": 5444
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 5444,
      "training_step_time": 9.914299249649048
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 3.136514902114868,
      "step": 5444
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 5444,
      "training_step_time": 10.176757574081421
    },
    {
      "epoch": 2.07672119140625e-05,
      "model_forward_time": 3.1360909938812256,
      "step": 5444
    },
    {
      "epoch": 2.07672119140625e-05,
      "step": 5444,
      "training_step_time": 10.266896486282349
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 3.184244394302368,
      "step": 5445
    },
    {
      "dynamics_loss_avg": 0.059783822298049925,
      "epoch": 2.0771026611328126e-05,
      "step": 5445
    },
    {
      "action_loss_avg": 0.029148235451430082,
      "epoch": 2.0771026611328126e-05,
      "step": 5445
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 5445,
      "training_step_time": 9.9618239402771
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 3.1442532539367676,
      "step": 5445
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 5445,
      "training_step_time": 9.91678786277771
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 3.1411421298980713,
      "step": 5445
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 5445,
      "training_step_time": 9.914215803146362
    },
    {
      "epoch": 2.0771026611328126e-05,
      "model_forward_time": 3.413426399230957,
      "step": 5445
    },
    {
      "epoch": 2.0771026611328126e-05,
      "step": 5445,
      "training_step_time": 10.326060771942139
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 3.182692527770996,
      "step": 5446
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 5446,
      "training_step_time": 9.952117443084717
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 3.144066333770752,
      "step": 5446
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 5446,
      "training_step_time": 9.912003993988037
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 3.1403074264526367,
      "step": 5446
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 5446,
      "training_step_time": 9.909520387649536
    },
    {
      "epoch": 2.077484130859375e-05,
      "model_forward_time": 3.141580104827881,
      "step": 5446
    },
    {
      "epoch": 2.077484130859375e-05,
      "step": 5446,
      "training_step_time": 10.017260789871216
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 3.1911895275115967,
      "step": 5447
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 5447,
      "training_step_time": 10.228473901748657
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 3.1470630168914795,
      "step": 5447
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 5447,
      "training_step_time": 9.917765378952026
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 3.143812417984009,
      "step": 5447
    },
    {
      "dynamics_loss_avg": 0.06907144207507372,
      "epoch": 2.0778656005859375e-05,
      "step": 5447
    },
    {
      "action_loss_avg": 0.044356150273233655,
      "epoch": 2.0778656005859375e-05,
      "step": 5447
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 5447,
      "training_step_time": 9.916599988937378
    },
    {
      "epoch": 2.0778656005859375e-05,
      "model_forward_time": 3.145357370376587,
      "step": 5447
    },
    {
      "epoch": 2.0778656005859375e-05,
      "step": 5447,
      "training_step_time": 10.26840853691101
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 3.1946206092834473,
      "step": 5448
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 5448,
      "training_step_time": 9.972227334976196
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 3.43076229095459,
      "step": 5448
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 5448,
      "training_step_time": 10.202178716659546
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 3.143996477127075,
      "step": 5448
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 5448,
      "training_step_time": 9.914645910263062
    },
    {
      "epoch": 2.0782470703125e-05,
      "model_forward_time": 3.1412315368652344,
      "step": 5448
    },
    {
      "epoch": 2.0782470703125e-05,
      "step": 5448,
      "training_step_time": 10.008655548095703
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 3.1757450103759766,
      "step": 5449
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 5449,
      "training_step_time": 9.946196556091309
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 3.145049810409546,
      "step": 5449
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 5449,
      "training_step_time": 9.916028499603271
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 3.406844139099121,
      "step": 5449
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 5449,
      "training_step_time": 10.182856559753418
    },
    {
      "epoch": 2.0786285400390624e-05,
      "model_forward_time": 3.1509552001953125,
      "step": 5449
    },
    {
      "epoch": 2.0786285400390624e-05,
      "step": 5449,
      "training_step_time": 10.304476976394653
    },
    {
      "epoch": 2.079010009765625e-05,
      "grad_norm": 0.2692432105541229,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 5450
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 3.1745004653930664,
      "step": 5450
    },
    {
      "dynamics_loss_avg": 0.05263345055282116,
      "epoch": 2.079010009765625e-05,
      "step": 5450
    },
    {
      "action_loss_avg": 0.01827041441574693,
      "epoch": 2.079010009765625e-05,
      "step": 5450
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 5450,
      "training_step_time": 9.952136516571045
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 3.1419222354888916,
      "step": 5450
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 5450,
      "training_step_time": 9.915039539337158
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 3.1447622776031494,
      "step": 5450
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 5450,
      "training_step_time": 9.915423393249512
    },
    {
      "epoch": 2.079010009765625e-05,
      "model_forward_time": 3.427358865737915,
      "step": 5450
    },
    {
      "epoch": 2.079010009765625e-05,
      "step": 5450,
      "training_step_time": 10.317568063735962
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 3.195507287979126,
      "step": 5451
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 5451,
      "training_step_time": 9.970799446105957
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 3.1408843994140625,
      "step": 5451
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 5451,
      "training_step_time": 9.90820598602295
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 3.1449742317199707,
      "step": 5451
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 5451,
      "training_step_time": 9.916807889938354
    },
    {
      "epoch": 2.0793914794921874e-05,
      "model_forward_time": 3.1415576934814453,
      "step": 5451
    },
    {
      "epoch": 2.0793914794921874e-05,
      "step": 5451,
      "training_step_time": 10.007017850875854
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 3.4411542415618896,
      "step": 5452
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 5452,
      "training_step_time": 10.215662479400635
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 3.1383204460144043,
      "step": 5452
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 5452,
      "training_step_time": 9.91283631324768
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 3.144249677658081,
      "step": 5452
    },
    {
      "dynamics_loss_avg": 0.05513774212449789,
      "epoch": 2.07977294921875e-05,
      "step": 5452
    },
    {
      "action_loss_avg": 0.024768645782023667,
      "epoch": 2.07977294921875e-05,
      "step": 5452
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 5452,
      "training_step_time": 9.917817115783691
    },
    {
      "epoch": 2.07977294921875e-05,
      "model_forward_time": 3.1431217193603516,
      "step": 5452
    },
    {
      "epoch": 2.07977294921875e-05,
      "step": 5452,
      "training_step_time": 10.018019676208496
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 3.1620049476623535,
      "step": 5453
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 5453,
      "training_step_time": 9.929471254348755
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 3.4005465507507324,
      "step": 5453
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 5453,
      "training_step_time": 10.16653561592102
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 3.145521879196167,
      "step": 5453
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 5453,
      "training_step_time": 9.910052299499512
    },
    {
      "epoch": 2.0801544189453126e-05,
      "model_forward_time": 3.1400973796844482,
      "step": 5453
    },
    {
      "epoch": 2.0801544189453126e-05,
      "step": 5453,
      "training_step_time": 10.25008511543274
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 3.190438747406006,
      "step": 5454
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 5454,
      "training_step_time": 9.963133811950684
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 3.141242742538452,
      "step": 5454
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 5454,
      "training_step_time": 9.911299228668213
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 3.4190165996551514,
      "step": 5454
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 5454,
      "training_step_time": 10.190712928771973
    },
    {
      "epoch": 2.080535888671875e-05,
      "model_forward_time": 2.4418349266052246,
      "step": 5454
    },
    {
      "epoch": 2.080535888671875e-05,
      "step": 5454,
      "training_step_time": 10.33701491355896
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 3.1906628608703613,
      "step": 5455
    },
    {
      "dynamics_loss_avg": 0.06252958998084068,
      "epoch": 2.0809173583984375e-05,
      "step": 5455
    },
    {
      "action_loss_avg": 0.02521309396252036,
      "epoch": 2.0809173583984375e-05,
      "step": 5455
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 5455,
      "training_step_time": 9.964602947235107
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 3.1451621055603027,
      "step": 5455
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 5455,
      "training_step_time": 9.917082071304321
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 3.1400442123413086,
      "step": 5455
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 5455,
      "training_step_time": 9.90988278388977
    },
    {
      "epoch": 2.0809173583984375e-05,
      "model_forward_time": 3.4104301929473877,
      "step": 5455
    },
    {
      "epoch": 2.0809173583984375e-05,
      "step": 5455,
      "training_step_time": 10.278085708618164
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 3.194070339202881,
      "step": 5456
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 5456,
      "training_step_time": 9.96947956085205
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 3.14245343208313,
      "step": 5456
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 5456,
      "training_step_time": 9.915838241577148
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 3.1428616046905518,
      "step": 5456
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 5456,
      "training_step_time": 9.917749881744385
    },
    {
      "epoch": 2.081298828125e-05,
      "model_forward_time": 3.145282506942749,
      "step": 5456
    },
    {
      "epoch": 2.081298828125e-05,
      "step": 5456,
      "training_step_time": 10.018327713012695
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 3.4398751258850098,
      "step": 5457
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 5457,
      "training_step_time": 10.211220502853394
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 3.1435139179229736,
      "step": 5457
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 5457,
      "training_step_time": 9.912354707717896
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 3.138206958770752,
      "step": 5457
    },
    {
      "dynamics_loss_avg": 0.05831743683665991,
      "epoch": 2.0816802978515624e-05,
      "step": 5457
    },
    {
      "action_loss_avg": 0.02395071662031114,
      "epoch": 2.0816802978515624e-05,
      "step": 5457
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 5457,
      "training_step_time": 9.911314249038696
    },
    {
      "epoch": 2.0816802978515624e-05,
      "model_forward_time": 3.141775369644165,
      "step": 5457
    },
    {
      "epoch": 2.0816802978515624e-05,
      "step": 5457,
      "training_step_time": 10.287113904953003
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 3.1671273708343506,
      "step": 5458
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 5458,
      "training_step_time": 9.942115306854248
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 3.425480842590332,
      "step": 5458
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 5458,
      "training_step_time": 10.196588039398193
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 2.435056209564209,
      "step": 5458
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 5458,
      "training_step_time": 7.462726354598999
    },
    {
      "epoch": 2.082061767578125e-05,
      "model_forward_time": 3.1394834518432617,
      "step": 5458
    },
    {
      "epoch": 2.082061767578125e-05,
      "step": 5458,
      "training_step_time": 10.271140336990356
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 3.152277708053589,
      "step": 5459
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 5459,
      "training_step_time": 9.924196481704712
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 3.135725498199463,
      "step": 5459
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 5459,
      "training_step_time": 9.90490198135376
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 2.4316041469573975,
      "step": 5459
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 5459,
      "training_step_time": 7.714879751205444
    },
    {
      "epoch": 2.0824432373046874e-05,
      "model_forward_time": 3.1307220458984375,
      "step": 5459
    },
    {
      "epoch": 2.0824432373046874e-05,
      "step": 5459,
      "training_step_time": 10.007658004760742
    },
    {
      "epoch": 2.08282470703125e-05,
      "grad_norm": 0.3849577009677887,
      "learning_rate": 1e-05,
      "loss": 0.0857,
      "step": 5460
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 3.185659170150757,
      "step": 5460
    },
    {
      "dynamics_loss_avg": 0.06177739817649126,
      "epoch": 2.08282470703125e-05,
      "step": 5460
    },
    {
      "action_loss_avg": 0.022256421530619262,
      "epoch": 2.08282470703125e-05,
      "step": 5460
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 5460,
      "training_step_time": 9.952763557434082
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 3.131898880004883,
      "step": 5460
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 5460,
      "training_step_time": 9.900012731552124
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 3.138091564178467,
      "step": 5460
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 5460,
      "training_step_time": 9.90717363357544
    },
    {
      "epoch": 2.08282470703125e-05,
      "model_forward_time": 3.4350383281707764,
      "step": 5460
    },
    {
      "epoch": 2.08282470703125e-05,
      "step": 5460,
      "training_step_time": 10.304707050323486
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 3.1913862228393555,
      "step": 5461
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 5461,
      "training_step_time": 9.965405225753784
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 3.138188600540161,
      "step": 5461
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 5461,
      "training_step_time": 9.912159204483032
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 3.1404271125793457,
      "step": 5461
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 5461,
      "training_step_time": 9.914250135421753
    },
    {
      "epoch": 2.0832061767578126e-05,
      "model_forward_time": 3.1405043601989746,
      "step": 5461
    },
    {
      "epoch": 2.0832061767578126e-05,
      "step": 5461,
      "training_step_time": 10.318829774856567
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 3.1609320640563965,
      "step": 5462
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 5462,
      "training_step_time": 10.22865891456604
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 3.140063762664795,
      "step": 5462
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 5462,
      "training_step_time": 9.913819074630737
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 3.1391494274139404,
      "step": 5462
    },
    {
      "dynamics_loss_avg": 0.055326526239514354,
      "epoch": 2.083587646484375e-05,
      "step": 5462
    },
    {
      "action_loss_avg": 0.021343272738158704,
      "epoch": 2.083587646484375e-05,
      "step": 5462
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 5462,
      "training_step_time": 9.919121742248535
    },
    {
      "epoch": 2.083587646484375e-05,
      "model_forward_time": 3.1346957683563232,
      "step": 5462
    },
    {
      "epoch": 2.083587646484375e-05,
      "step": 5462,
      "training_step_time": 10.295698165893555
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 3.1719791889190674,
      "step": 5463
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 5463,
      "training_step_time": 9.941112518310547
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 3.403499126434326,
      "step": 5463
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 5463,
      "training_step_time": 10.176777601242065
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 3.1352579593658447,
      "step": 5463
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 5463,
      "training_step_time": 9.909639358520508
    },
    {
      "epoch": 2.0839691162109375e-05,
      "model_forward_time": 3.1343016624450684,
      "step": 5463
    },
    {
      "epoch": 2.0839691162109375e-05,
      "step": 5463,
      "training_step_time": 10.300300359725952
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 3.1828560829162598,
      "step": 5464
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 5464,
      "training_step_time": 9.953989744186401
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 3.1340010166168213,
      "step": 5464
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 5464,
      "training_step_time": 9.902620315551758
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 3.136003255844116,
      "step": 5464
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 5464,
      "training_step_time": 10.178103685379028
    },
    {
      "epoch": 2.0843505859375e-05,
      "model_forward_time": 3.1318745613098145,
      "step": 5464
    },
    {
      "epoch": 2.0843505859375e-05,
      "step": 5464,
      "training_step_time": 10.29212760925293
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 3.1926424503326416,
      "step": 5465
    },
    {
      "dynamics_loss_avg": 0.05987032540142536,
      "epoch": 2.0847320556640624e-05,
      "step": 5465
    },
    {
      "action_loss_avg": 0.02574730757623911,
      "epoch": 2.0847320556640624e-05,
      "step": 5465
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 5465,
      "training_step_time": 9.963978290557861
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 3.136032819747925,
      "step": 5465
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 5465,
      "training_step_time": 9.902314186096191
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 3.1329288482666016,
      "step": 5465
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 5465,
      "training_step_time": 9.896621942520142
    },
    {
      "epoch": 2.0847320556640624e-05,
      "model_forward_time": 3.3980886936187744,
      "step": 5465
    },
    {
      "epoch": 2.0847320556640624e-05,
      "step": 5465,
      "training_step_time": 10.26639199256897
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 3.171543836593628,
      "step": 5466
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 5466,
      "training_step_time": 9.9418044090271
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 3.1337170600891113,
      "step": 5466
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 5466,
      "training_step_time": 9.902745962142944
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 3.1312947273254395,
      "step": 5466
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 5466,
      "training_step_time": 9.898488759994507
    },
    {
      "epoch": 2.085113525390625e-05,
      "model_forward_time": 3.1376278400421143,
      "step": 5466
    },
    {
      "epoch": 2.085113525390625e-05,
      "step": 5466,
      "training_step_time": 10.28705620765686
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 3.1513001918792725,
      "step": 5467
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 5467,
      "training_step_time": 10.19935917854309
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 3.1345720291137695,
      "step": 5467
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 5467,
      "training_step_time": 9.899818897247314
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 2.4298510551452637,
      "step": 5467
    },
    {
      "dynamics_loss_avg": 0.06502653248608112,
      "epoch": 2.0854949951171874e-05,
      "step": 5467
    },
    {
      "action_loss_avg": 0.027554600033909082,
      "epoch": 2.0854949951171874e-05,
      "step": 5467
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 5467,
      "training_step_time": 7.455815076828003
    },
    {
      "epoch": 2.0854949951171874e-05,
      "model_forward_time": 3.132249116897583,
      "step": 5467
    },
    {
      "epoch": 2.0854949951171874e-05,
      "step": 5467,
      "training_step_time": 10.014591693878174
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 3.175952196121216,
      "step": 5468
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 5468,
      "training_step_time": 9.955718517303467
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 3.4114577770233154,
      "step": 5468
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 5468,
      "training_step_time": 10.190289735794067
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 3.1406502723693848,
      "step": 5468
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 5468,
      "training_step_time": 9.926604270935059
    },
    {
      "epoch": 2.08587646484375e-05,
      "model_forward_time": 3.1398773193359375,
      "step": 5468
    },
    {
      "epoch": 2.08587646484375e-05,
      "step": 5468,
      "training_step_time": 10.281097173690796
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 3.1795883178710938,
      "step": 5469
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 5469,
      "training_step_time": 9.967739582061768
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 3.1412293910980225,
      "step": 5469
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 5469,
      "training_step_time": 9.931020259857178
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 3.4081506729125977,
      "step": 5469
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 5469,
      "training_step_time": 10.188276529312134
    },
    {
      "epoch": 2.0862579345703126e-05,
      "model_forward_time": 3.1449553966522217,
      "step": 5469
    },
    {
      "epoch": 2.0862579345703126e-05,
      "step": 5469,
      "training_step_time": 10.038345098495483
    },
    {
      "epoch": 2.086639404296875e-05,
      "grad_norm": 0.5287845730781555,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 5470
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 3.1800167560577393,
      "step": 5470
    },
    {
      "dynamics_loss_avg": 0.06102517731487751,
      "epoch": 2.086639404296875e-05,
      "step": 5470
    },
    {
      "action_loss_avg": 0.02933131097815931,
      "epoch": 2.086639404296875e-05,
      "step": 5470
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 5470,
      "training_step_time": 9.962913990020752
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 3.1352689266204834,
      "step": 5470
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 5470,
      "training_step_time": 9.91396713256836
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 3.1398634910583496,
      "step": 5470
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 5470,
      "training_step_time": 9.917716264724731
    },
    {
      "epoch": 2.086639404296875e-05,
      "model_forward_time": 3.4155094623565674,
      "step": 5470
    },
    {
      "epoch": 2.086639404296875e-05,
      "step": 5470,
      "training_step_time": 10.289254188537598
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 3.1958088874816895,
      "step": 5471
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 5471,
      "training_step_time": 9.974751710891724
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 3.13818621635437,
      "step": 5471
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 5471,
      "training_step_time": 9.920247793197632
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 3.14658784866333,
      "step": 5471
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 5471,
      "training_step_time": 9.924654006958008
    },
    {
      "epoch": 2.0870208740234375e-05,
      "model_forward_time": 3.1380598545074463,
      "step": 5471
    },
    {
      "epoch": 2.0870208740234375e-05,
      "step": 5471,
      "training_step_time": 10.277004480361938
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 3.446533679962158,
      "step": 5472
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 5472,
      "training_step_time": 10.230725288391113
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 3.1371371746063232,
      "step": 5472
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 5472,
      "training_step_time": 9.919284582138062
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 3.1411900520324707,
      "step": 5472
    },
    {
      "dynamics_loss_avg": 0.05173522867262363,
      "epoch": 2.08740234375e-05,
      "step": 5472
    },
    {
      "action_loss_avg": 0.02263838374055922,
      "epoch": 2.08740234375e-05,
      "step": 5472
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 5472,
      "training_step_time": 9.926047801971436
    },
    {
      "epoch": 2.08740234375e-05,
      "model_forward_time": 3.138279438018799,
      "step": 5472
    },
    {
      "epoch": 2.08740234375e-05,
      "step": 5472,
      "training_step_time": 10.294703960418701
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 3.1596670150756836,
      "step": 5473
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 5473,
      "training_step_time": 9.942567825317383
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 3.39621901512146,
      "step": 5473
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 5473,
      "training_step_time": 10.173261642456055
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 3.140723466873169,
      "step": 5473
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 5473,
      "training_step_time": 9.916653156280518
    },
    {
      "epoch": 2.0877838134765624e-05,
      "model_forward_time": 3.1348326206207275,
      "step": 5473
    },
    {
      "epoch": 2.0877838134765624e-05,
      "step": 5473,
      "training_step_time": 10.302162885665894
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 3.187530040740967,
      "step": 5474
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 5474,
      "training_step_time": 9.967737913131714
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 3.139108657836914,
      "step": 5474
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 5474,
      "training_step_time": 9.91088056564331
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 3.4159622192382812,
      "step": 5474
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 5474,
      "training_step_time": 10.189288139343262
    },
    {
      "epoch": 2.088165283203125e-05,
      "model_forward_time": 3.139960289001465,
      "step": 5474
    },
    {
      "epoch": 2.088165283203125e-05,
      "step": 5474,
      "training_step_time": 10.248366355895996
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 3.186275005340576,
      "step": 5475
    },
    {
      "dynamics_loss_avg": 0.04492034576833248,
      "epoch": 2.0885467529296874e-05,
      "step": 5475
    },
    {
      "action_loss_avg": 0.019891570322215557,
      "epoch": 2.0885467529296874e-05,
      "step": 5475
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 5475,
      "training_step_time": 9.966366291046143
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 3.135244846343994,
      "step": 5475
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 5475,
      "training_step_time": 9.908889293670654
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 3.135902166366577,
      "step": 5475
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 5475,
      "training_step_time": 9.911375045776367
    },
    {
      "epoch": 2.0885467529296874e-05,
      "model_forward_time": 3.395972490310669,
      "step": 5475
    },
    {
      "epoch": 2.0885467529296874e-05,
      "step": 5475,
      "training_step_time": 10.266875267028809
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 3.1720070838928223,
      "step": 5476
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 5476,
      "training_step_time": 9.951205492019653
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 3.135702133178711,
      "step": 5476
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 5476,
      "training_step_time": 9.907081127166748
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 3.130096912384033,
      "step": 5476
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 5476,
      "training_step_time": 9.899198770523071
    },
    {
      "epoch": 2.08892822265625e-05,
      "model_forward_time": 3.1343111991882324,
      "step": 5476
    },
    {
      "epoch": 2.08892822265625e-05,
      "step": 5476,
      "training_step_time": 10.31785273551941
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 3.414111614227295,
      "step": 5477
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 5477,
      "training_step_time": 10.187191009521484
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 3.1342265605926514,
      "step": 5477
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 5477,
      "training_step_time": 9.911264657974243
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 3.1361234188079834,
      "step": 5477
    },
    {
      "dynamics_loss_avg": 0.08452295139431953,
      "epoch": 2.0893096923828126e-05,
      "step": 5477
    },
    {
      "action_loss_avg": 0.03313489491119981,
      "epoch": 2.0893096923828126e-05,
      "step": 5477
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 5477,
      "training_step_time": 9.919385433197021
    },
    {
      "epoch": 2.0893096923828126e-05,
      "model_forward_time": 3.140700578689575,
      "step": 5477
    },
    {
      "epoch": 2.0893096923828126e-05,
      "step": 5477,
      "training_step_time": 10.292305707931519
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 3.1715290546417236,
      "step": 5478
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 5478,
      "training_step_time": 9.947704076766968
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 3.417166233062744,
      "step": 5478
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 5478,
      "training_step_time": 10.195453882217407
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 3.1329715251922607,
      "step": 5478
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 5478,
      "training_step_time": 9.90964937210083
    },
    {
      "epoch": 2.089691162109375e-05,
      "model_forward_time": 3.1385750770568848,
      "step": 5478
    },
    {
      "epoch": 2.089691162109375e-05,
      "step": 5478,
      "training_step_time": 10.014718532562256
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 3.1661415100097656,
      "step": 5479
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 5479,
      "training_step_time": 9.945434093475342
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 3.138317346572876,
      "step": 5479
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 5479,
      "training_step_time": 9.914105653762817
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 3.3973491191864014,
      "step": 5479
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 5479,
      "training_step_time": 10.172499895095825
    },
    {
      "epoch": 2.0900726318359375e-05,
      "model_forward_time": 3.1384665966033936,
      "step": 5479
    },
    {
      "epoch": 2.0900726318359375e-05,
      "step": 5479,
      "training_step_time": 10.290866136550903
    },
    {
      "epoch": 2.0904541015625e-05,
      "grad_norm": 0.45523887872695923,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 5480
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 3.1849541664123535,
      "step": 5480
    },
    {
      "dynamics_loss_avg": 0.059957274422049525,
      "epoch": 2.0904541015625e-05,
      "step": 5480
    },
    {
      "action_loss_avg": 0.02444008425809443,
      "epoch": 2.0904541015625e-05,
      "step": 5480
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 5480,
      "training_step_time": 9.968222379684448
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 3.1360602378845215,
      "step": 5480
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 5480,
      "training_step_time": 9.91801404953003
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 3.141625165939331,
      "step": 5480
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 5480,
      "training_step_time": 9.918776988983154
    },
    {
      "epoch": 2.0904541015625e-05,
      "model_forward_time": 3.4189791679382324,
      "step": 5480
    },
    {
      "epoch": 2.0904541015625e-05,
      "step": 5480,
      "training_step_time": 10.290038824081421
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 3.182431221008301,
      "step": 5481
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 5481,
      "training_step_time": 9.953627109527588
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 3.131284713745117,
      "step": 5481
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 5481,
      "training_step_time": 9.90073299407959
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 3.136543035507202,
      "step": 5481
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 5481,
      "training_step_time": 9.908486127853394
    },
    {
      "epoch": 2.0908355712890625e-05,
      "model_forward_time": 3.1382739543914795,
      "step": 5481
    },
    {
      "epoch": 2.0908355712890625e-05,
      "step": 5481,
      "training_step_time": 10.319342374801636
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 3.1556153297424316,
      "step": 5482
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 5482,
      "training_step_time": 10.221672296524048
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 3.134345769882202,
      "step": 5482
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 5482,
      "training_step_time": 9.91312313079834
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 3.1400465965270996,
      "step": 5482
    },
    {
      "dynamics_loss_avg": 0.05783272851258516,
      "epoch": 2.091217041015625e-05,
      "step": 5482
    },
    {
      "action_loss_avg": 0.028411710914224387,
      "epoch": 2.091217041015625e-05,
      "step": 5482
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 5482,
      "training_step_time": 9.922901153564453
    },
    {
      "epoch": 2.091217041015625e-05,
      "model_forward_time": 3.136134624481201,
      "step": 5482
    },
    {
      "epoch": 2.091217041015625e-05,
      "step": 5482,
      "training_step_time": 10.27916407585144
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 3.183228015899658,
      "step": 5483
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 5483,
      "training_step_time": 9.965676307678223
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 3.393550157546997,
      "step": 5483
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 5483,
      "training_step_time": 10.175230264663696
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 3.1403234004974365,
      "step": 5483
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 5483,
      "training_step_time": 9.9145188331604
    },
    {
      "epoch": 2.0915985107421874e-05,
      "model_forward_time": 3.1365315914154053,
      "step": 5483
    },
    {
      "epoch": 2.0915985107421874e-05,
      "step": 5483,
      "training_step_time": 10.270671606063843
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 3.1546568870544434,
      "step": 5484
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 5484,
      "training_step_time": 9.931394338607788
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 3.1381494998931885,
      "step": 5484
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 5484,
      "training_step_time": 9.916868448257446
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 3.1370253562927246,
      "step": 5484
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 5484,
      "training_step_time": 10.196523427963257
    },
    {
      "epoch": 2.09197998046875e-05,
      "model_forward_time": 3.1355385780334473,
      "step": 5484
    },
    {
      "epoch": 2.09197998046875e-05,
      "step": 5484,
      "training_step_time": 10.292586326599121
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 3.186396360397339,
      "step": 5485
    },
    {
      "dynamics_loss_avg": 0.06418828871101141,
      "epoch": 2.0923614501953126e-05,
      "step": 5485
    },
    {
      "action_loss_avg": 0.02404389837756753,
      "epoch": 2.0923614501953126e-05,
      "step": 5485
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 5485,
      "training_step_time": 9.968372821807861
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 3.1380271911621094,
      "step": 5485
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 5485,
      "training_step_time": 9.919481039047241
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 3.1374588012695312,
      "step": 5485
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 5485,
      "training_step_time": 9.917537927627563
    },
    {
      "epoch": 2.0923614501953126e-05,
      "model_forward_time": 3.4056291580200195,
      "step": 5485
    },
    {
      "epoch": 2.0923614501953126e-05,
      "step": 5485,
      "training_step_time": 10.28591799736023
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 3.1852149963378906,
      "step": 5486
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 5486,
      "training_step_time": 9.968456745147705
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 3.137925148010254,
      "step": 5486
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 5486,
      "training_step_time": 9.919482707977295
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 3.1391303539276123,
      "step": 5486
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 5486,
      "training_step_time": 9.919699668884277
    },
    {
      "epoch": 2.092742919921875e-05,
      "model_forward_time": 3.1416335105895996,
      "step": 5486
    },
    {
      "epoch": 2.092742919921875e-05,
      "step": 5486,
      "training_step_time": 10.287612915039062
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 3.1664533615112305,
      "step": 5487
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 5487,
      "training_step_time": 10.236346006393433
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 3.1404504776000977,
      "step": 5487
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 5487,
      "training_step_time": 9.921032190322876
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 3.1403329372406006,
      "step": 5487
    },
    {
      "dynamics_loss_avg": 0.0583990665152669,
      "epoch": 2.0931243896484375e-05,
      "step": 5487
    },
    {
      "action_loss_avg": 0.024161852709949017,
      "epoch": 2.0931243896484375e-05,
      "step": 5487
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 5487,
      "training_step_time": 9.922979354858398
    },
    {
      "epoch": 2.0931243896484375e-05,
      "model_forward_time": 3.144381046295166,
      "step": 5487
    },
    {
      "epoch": 2.0931243896484375e-05,
      "step": 5487,
      "training_step_time": 10.310433626174927
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 3.175795555114746,
      "step": 5488
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 5488,
      "training_step_time": 9.961815595626831
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 3.422123670578003,
      "step": 5488
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 5488,
      "training_step_time": 10.200798988342285
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 3.1456916332244873,
      "step": 5488
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 5488,
      "training_step_time": 9.924926280975342
    },
    {
      "epoch": 2.093505859375e-05,
      "model_forward_time": 3.1476497650146484,
      "step": 5488
    },
    {
      "epoch": 2.093505859375e-05,
      "step": 5488,
      "training_step_time": 10.279278993606567
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 3.189512252807617,
      "step": 5489
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 5489,
      "training_step_time": 9.961558103561401
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 3.1346487998962402,
      "step": 5489
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 5489,
      "training_step_time": 9.903219223022461
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 3.387099504470825,
      "step": 5489
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 5489,
      "training_step_time": 10.152872562408447
    },
    {
      "epoch": 2.0938873291015625e-05,
      "model_forward_time": 3.1385421752929688,
      "step": 5489
    },
    {
      "epoch": 2.0938873291015625e-05,
      "step": 5489,
      "training_step_time": 10.251893997192383
    },
    {
      "epoch": 2.094268798828125e-05,
      "grad_norm": 0.46323639154434204,
      "learning_rate": 1e-05,
      "loss": 0.086,
      "step": 5490
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 3.145097494125366,
      "step": 5490
    },
    {
      "dynamics_loss_avg": 0.06044472493231297,
      "epoch": 2.094268798828125e-05,
      "step": 5490
    },
    {
      "action_loss_avg": 0.02766996156424284,
      "epoch": 2.094268798828125e-05,
      "step": 5490
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 5490,
      "training_step_time": 9.9215989112854
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 3.1339385509490967,
      "step": 5490
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 5490,
      "training_step_time": 9.903931379318237
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 3.1370038986206055,
      "step": 5490
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 5490,
      "training_step_time": 9.905226469039917
    },
    {
      "epoch": 2.094268798828125e-05,
      "model_forward_time": 2.7071847915649414,
      "step": 5490
    },
    {
      "epoch": 2.094268798828125e-05,
      "step": 5490,
      "training_step_time": 10.00778579711914
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 3.175903797149658,
      "step": 5491
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 5491,
      "training_step_time": 9.942902088165283
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 3.128561019897461,
      "step": 5491
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 5491,
      "training_step_time": 9.894603729248047
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 3.1355478763580322,
      "step": 5491
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 5491,
      "training_step_time": 9.903058528900146
    },
    {
      "epoch": 2.0946502685546874e-05,
      "model_forward_time": 3.1327385902404785,
      "step": 5491
    },
    {
      "epoch": 2.0946502685546874e-05,
      "step": 5491,
      "training_step_time": 10.30614447593689
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 3.4542429447174072,
      "step": 5492
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 5492,
      "training_step_time": 10.221768140792847
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 3.12957501411438,
      "step": 5492
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 5492,
      "training_step_time": 9.893046617507935
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 3.1317434310913086,
      "step": 5492
    },
    {
      "dynamics_loss_avg": 0.07356314733624458,
      "epoch": 2.09503173828125e-05,
      "step": 5492
    },
    {
      "action_loss_avg": 0.02868403997272253,
      "epoch": 2.09503173828125e-05,
      "step": 5492
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 5492,
      "training_step_time": 9.896973848342896
    },
    {
      "epoch": 2.09503173828125e-05,
      "model_forward_time": 3.1298811435699463,
      "step": 5492
    },
    {
      "epoch": 2.09503173828125e-05,
      "step": 5492,
      "training_step_time": 10.263627052307129
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 3.1630451679229736,
      "step": 5493
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 5493,
      "training_step_time": 9.932749509811401
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 3.393052577972412,
      "step": 5493
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 5493,
      "training_step_time": 10.159202575683594
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 3.1412367820739746,
      "step": 5493
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 5493,
      "training_step_time": 9.914950370788574
    },
    {
      "epoch": 2.0954132080078126e-05,
      "model_forward_time": 3.140155792236328,
      "step": 5493
    },
    {
      "epoch": 2.0954132080078126e-05,
      "step": 5493,
      "training_step_time": 10.034425497055054
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 3.1825220584869385,
      "step": 5494
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 5494,
      "training_step_time": 9.95292854309082
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 3.1444878578186035,
      "step": 5494
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 5494,
      "training_step_time": 9.924517631530762
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 3.4195358753204346,
      "step": 5494
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 5494,
      "training_step_time": 10.193756103515625
    },
    {
      "epoch": 2.095794677734375e-05,
      "model_forward_time": 3.1508140563964844,
      "step": 5494
    },
    {
      "epoch": 2.095794677734375e-05,
      "step": 5494,
      "training_step_time": 10.273348093032837
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 3.1810953617095947,
      "step": 5495
    },
    {
      "dynamics_loss_avg": 0.06052019707858562,
      "epoch": 2.0961761474609375e-05,
      "step": 5495
    },
    {
      "action_loss_avg": 0.032120220549404624,
      "epoch": 2.0961761474609375e-05,
      "step": 5495
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 5495,
      "training_step_time": 9.956212520599365
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 3.1433403491973877,
      "step": 5495
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 5495,
      "training_step_time": 9.919483184814453
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 3.1371283531188965,
      "step": 5495
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 5495,
      "training_step_time": 9.91488242149353
    },
    {
      "epoch": 2.0961761474609375e-05,
      "model_forward_time": 3.4056015014648438,
      "step": 5495
    },
    {
      "epoch": 2.0961761474609375e-05,
      "step": 5495,
      "training_step_time": 10.287426948547363
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 3.18931245803833,
      "step": 5496
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 5496,
      "training_step_time": 9.966890096664429
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 3.139533758163452,
      "step": 5496
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 5496,
      "training_step_time": 9.917293548583984
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 3.1362905502319336,
      "step": 5496
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 5496,
      "training_step_time": 9.912811517715454
    },
    {
      "epoch": 2.0965576171875e-05,
      "model_forward_time": 3.1419336795806885,
      "step": 5496
    },
    {
      "epoch": 2.0965576171875e-05,
      "step": 5496,
      "training_step_time": 10.250632047653198
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 3.4363441467285156,
      "step": 5497
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 5497,
      "training_step_time": 10.218340635299683
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 3.1404213905334473,
      "step": 5497
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 5497,
      "training_step_time": 9.920898914337158
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 3.1377036571502686,
      "step": 5497
    },
    {
      "dynamics_loss_avg": 0.059537476487457754,
      "epoch": 2.0969390869140625e-05,
      "step": 5497
    },
    {
      "action_loss_avg": 0.026258377404883505,
      "epoch": 2.0969390869140625e-05,
      "step": 5497
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 5497,
      "training_step_time": 9.918928384780884
    },
    {
      "epoch": 2.0969390869140625e-05,
      "model_forward_time": 2.435255289077759,
      "step": 5497
    },
    {
      "epoch": 2.0969390869140625e-05,
      "step": 5497,
      "training_step_time": 10.29920220375061
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 3.1900458335876465,
      "step": 5498
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 5498,
      "training_step_time": 9.967546701431274
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 3.138662338256836,
      "step": 5498
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 5498,
      "training_step_time": 10.195963382720947
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 3.140450954437256,
      "step": 5498
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 5498,
      "training_step_time": 9.922481060028076
    },
    {
      "epoch": 2.097320556640625e-05,
      "model_forward_time": 3.1486802101135254,
      "step": 5498
    },
    {
      "epoch": 2.097320556640625e-05,
      "step": 5498,
      "training_step_time": 10.29081678390503
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 3.2073898315429688,
      "step": 5499
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 5499,
      "training_step_time": 9.98767876625061
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 3.143852949142456,
      "step": 5499
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 5499,
      "training_step_time": 9.91869330406189
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 3.409442901611328,
      "step": 5499
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 5499,
      "training_step_time": 10.186137437820435
    },
    {
      "epoch": 2.0977020263671874e-05,
      "model_forward_time": 3.1451876163482666,
      "step": 5499
    },
    {
      "epoch": 2.0977020263671874e-05,
      "step": 5499,
      "training_step_time": 10.013561487197876
    },
    {
      "epoch": 2.09808349609375e-05,
      "grad_norm": 0.3955205976963043,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 5500
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 3.15564227104187,
      "step": 5500
    },
    {
      "dynamics_loss_avg": 0.05597408097237348,
      "epoch": 2.09808349609375e-05,
      "step": 5500
    },
    {
      "action_loss_avg": 0.016939033288508654,
      "epoch": 2.09808349609375e-05,
      "step": 5500
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 5500,
      "training_step_time": 9.905410766601562
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 3.1273109912872314,
      "step": 5500
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 5500,
      "training_step_time": 9.876338005065918
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 3.1310250759124756,
      "step": 5500
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 5500,
      "training_step_time": 9.887727737426758
    },
    {
      "epoch": 2.09808349609375e-05,
      "model_forward_time": 5.29963755607605,
      "step": 5500
    },
    {
      "epoch": 2.09808349609375e-05,
      "step": 5500,
      "training_step_time": 12.157870292663574
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 3.2001900672912598,
      "step": 5501
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 5501,
      "training_step_time": 9.963218688964844
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 3.1412785053253174,
      "step": 5501
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 5501,
      "training_step_time": 9.91515827178955
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 3.1376333236694336,
      "step": 5501
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 5501,
      "training_step_time": 9.906145572662354
    },
    {
      "epoch": 2.0984649658203126e-05,
      "model_forward_time": 3.142716646194458,
      "step": 5501
    },
    {
      "epoch": 2.0984649658203126e-05,
      "step": 5501,
      "training_step_time": 10.286466598510742
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 3.1793055534362793,
      "step": 5502
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 5502,
      "training_step_time": 9.954332113265991
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 3.14223575592041,
      "step": 5502
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 5502,
      "training_step_time": 9.919437885284424
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 3.1426215171813965,
      "step": 5502
    },
    {
      "dynamics_loss_avg": 0.0606162054464221,
      "epoch": 2.098846435546875e-05,
      "step": 5502
    },
    {
      "action_loss_avg": 0.028033026354387404,
      "epoch": 2.098846435546875e-05,
      "step": 5502
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 5502,
      "training_step_time": 9.917700290679932
    },
    {
      "epoch": 2.098846435546875e-05,
      "model_forward_time": 3.149510145187378,
      "step": 5502
    },
    {
      "epoch": 2.098846435546875e-05,
      "step": 5502,
      "training_step_time": 10.254392385482788
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 3.446352481842041,
      "step": 5503
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 5503,
      "training_step_time": 10.219173908233643
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 3.1501879692077637,
      "step": 5503
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 5503,
      "training_step_time": 9.924372434616089
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 3.1424028873443604,
      "step": 5503
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 5503,
      "training_step_time": 9.91788935661316
    },
    {
      "epoch": 2.0992279052734375e-05,
      "model_forward_time": 3.1480727195739746,
      "step": 5503
    },
    {
      "epoch": 2.0992279052734375e-05,
      "step": 5503,
      "training_step_time": 10.046873331069946
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 3.1635711193084717,
      "step": 5504
    },
    {
      "epoch": 2.099609375e-05,
      "step": 5504,
      "training_step_time": 9.940192937850952
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 3.1426892280578613,
      "step": 5504
    },
    {
      "epoch": 2.099609375e-05,
      "step": 5504,
      "training_step_time": 10.213942766189575
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 3.1407899856567383,
      "step": 5504
    },
    {
      "epoch": 2.099609375e-05,
      "step": 5504,
      "training_step_time": 9.917637586593628
    },
    {
      "epoch": 2.099609375e-05,
      "model_forward_time": 2.4380156993865967,
      "step": 5504
    },
    {
      "epoch": 2.099609375e-05,
      "step": 5504,
      "training_step_time": 10.256991386413574
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 3.1823911666870117,
      "step": 5505
    },
    {
      "dynamics_loss_avg": 0.0669088114053011,
      "epoch": 2.0999908447265625e-05,
      "step": 5505
    },
    {
      "action_loss_avg": 0.02505804765969515,
      "epoch": 2.0999908447265625e-05,
      "step": 5505
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 5505,
      "training_step_time": 9.955151557922363
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 3.1373677253723145,
      "step": 5505
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 5505,
      "training_step_time": 9.909932613372803
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 3.4217936992645264,
      "step": 5505
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 5505,
      "training_step_time": 10.195204496383667
    },
    {
      "epoch": 2.0999908447265625e-05,
      "model_forward_time": 3.1481878757476807,
      "step": 5505
    },
    {
      "epoch": 2.0999908447265625e-05,
      "step": 5505,
      "training_step_time": 10.329687118530273
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 3.148036241531372,
      "step": 5506
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 5506,
      "training_step_time": 9.919042348861694
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 3.1371572017669678,
      "step": 5506
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 5506,
      "training_step_time": 9.907409906387329
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 2.4397079944610596,
      "step": 5506
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 5506,
      "training_step_time": 7.470928907394409
    },
    {
      "epoch": 2.100372314453125e-05,
      "model_forward_time": 3.1370482444763184,
      "step": 5506
    },
    {
      "epoch": 2.100372314453125e-05,
      "step": 5506,
      "training_step_time": 10.312020301818848
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 3.469480276107788,
      "step": 5507
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 5507,
      "training_step_time": 10.244319200515747
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 2.4368631839752197,
      "step": 5507
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 5507,
      "training_step_time": 7.465334892272949
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 3.1388156414031982,
      "step": 5507
    },
    {
      "dynamics_loss_avg": 0.0630474278703332,
      "epoch": 2.1007537841796874e-05,
      "step": 5507
    },
    {
      "action_loss_avg": 0.02334350598976016,
      "epoch": 2.1007537841796874e-05,
      "step": 5507
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 5507,
      "training_step_time": 9.912671089172363
    },
    {
      "epoch": 2.1007537841796874e-05,
      "model_forward_time": 3.135617733001709,
      "step": 5507
    },
    {
      "epoch": 2.1007537841796874e-05,
      "step": 5507,
      "training_step_time": 10.02173399925232
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 3.470632553100586,
      "step": 5508
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 5508,
      "training_step_time": 10.2429838180542
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 3.143217086791992,
      "step": 5508
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 5508,
      "training_step_time": 9.919804334640503
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 3.144373893737793,
      "step": 5508
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 5508,
      "training_step_time": 9.920865058898926
    },
    {
      "epoch": 2.10113525390625e-05,
      "model_forward_time": 3.140532970428467,
      "step": 5508
    },
    {
      "epoch": 2.10113525390625e-05,
      "step": 5508,
      "training_step_time": 10.022945642471313
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 3.1746537685394287,
      "step": 5509
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 5509,
      "training_step_time": 9.948959589004517
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 3.1494810581207275,
      "step": 5509
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 5509,
      "training_step_time": 9.921003103256226
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 3.41626238822937,
      "step": 5509
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 5509,
      "training_step_time": 10.192406415939331
    },
    {
      "epoch": 2.1015167236328126e-05,
      "model_forward_time": 3.144770622253418,
      "step": 5509
    },
    {
      "epoch": 2.1015167236328126e-05,
      "step": 5509,
      "training_step_time": 10.056065082550049
    },
    {
      "epoch": 2.101898193359375e-05,
      "grad_norm": 0.5936101675033569,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 5510
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 3.159512996673584,
      "step": 5510
    },
    {
      "dynamics_loss_avg": 0.05815493073314428,
      "epoch": 2.101898193359375e-05,
      "step": 5510
    },
    {
      "action_loss_avg": 0.02299977340735495,
      "epoch": 2.101898193359375e-05,
      "step": 5510
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 5510,
      "training_step_time": 9.935069799423218
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 3.1419005393981934,
      "step": 5510
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 5510,
      "training_step_time": 9.907937288284302
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 2.438462972640991,
      "step": 5510
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 5510,
      "training_step_time": 7.744212627410889
    },
    {
      "epoch": 2.101898193359375e-05,
      "model_forward_time": 2.4347901344299316,
      "step": 5510
    },
    {
      "epoch": 2.101898193359375e-05,
      "step": 5510,
      "training_step_time": 10.285377502441406
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 3.176102638244629,
      "step": 5511
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 5511,
      "training_step_time": 9.949457883834839
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 3.1391165256500244,
      "step": 5511
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 5511,
      "training_step_time": 9.899522304534912
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 3.132744550704956,
      "step": 5511
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 5511,
      "training_step_time": 9.890425682067871
    },
    {
      "epoch": 2.1022796630859375e-05,
      "model_forward_time": 3.1347789764404297,
      "step": 5511
    },
    {
      "epoch": 2.1022796630859375e-05,
      "step": 5511,
      "training_step_time": 10.317797422409058
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 3.4685823917388916,
      "step": 5512
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 5512,
      "training_step_time": 10.22825026512146
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 3.139662504196167,
      "step": 5512
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 5512,
      "training_step_time": 9.897832870483398
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 3.138929605484009,
      "step": 5512
    },
    {
      "dynamics_loss_avg": 0.0701028972864151,
      "epoch": 2.1026611328125e-05,
      "step": 5512
    },
    {
      "action_loss_avg": 0.02276992779225111,
      "epoch": 2.1026611328125e-05,
      "step": 5512
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 5512,
      "training_step_time": 9.909636974334717
    },
    {
      "epoch": 2.1026611328125e-05,
      "model_forward_time": 3.140799045562744,
      "step": 5512
    },
    {
      "epoch": 2.1026611328125e-05,
      "step": 5512,
      "training_step_time": 10.015380620956421
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 3.177605152130127,
      "step": 5513
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 5513,
      "training_step_time": 10.218444347381592
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 3.144963026046753,
      "step": 5513
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 5513,
      "training_step_time": 9.96511459350586
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 3.1379220485687256,
      "step": 5513
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 5513,
      "training_step_time": 9.909459829330444
    },
    {
      "epoch": 2.1030426025390625e-05,
      "model_forward_time": 3.1538891792297363,
      "step": 5513
    },
    {
      "epoch": 2.1030426025390625e-05,
      "step": 5513,
      "training_step_time": 10.024080038070679
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 3.188333749771118,
      "step": 5514
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 5514,
      "training_step_time": 9.964035749435425
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 3.427309989929199,
      "step": 5514
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 5514,
      "training_step_time": 10.200446128845215
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 3.1428897380828857,
      "step": 5514
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 5514,
      "training_step_time": 9.919477224349976
    },
    {
      "epoch": 2.103424072265625e-05,
      "model_forward_time": 3.142887592315674,
      "step": 5514
    },
    {
      "epoch": 2.103424072265625e-05,
      "step": 5514,
      "training_step_time": 10.281485557556152
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 3.1632540225982666,
      "step": 5515
    },
    {
      "dynamics_loss_avg": 0.05798284336924553,
      "epoch": 2.1038055419921874e-05,
      "step": 5515
    },
    {
      "action_loss_avg": 0.020921073760837315,
      "epoch": 2.1038055419921874e-05,
      "step": 5515
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 5515,
      "training_step_time": 9.94261360168457
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 3.1326093673706055,
      "step": 5515
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 5515,
      "training_step_time": 9.910556077957153
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 3.136322498321533,
      "step": 5515
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 5515,
      "training_step_time": 10.207353353500366
    },
    {
      "epoch": 2.1038055419921874e-05,
      "model_forward_time": 3.138423204421997,
      "step": 5515
    },
    {
      "epoch": 2.1038055419921874e-05,
      "step": 5515,
      "training_step_time": 10.2925124168396
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 3.184365749359131,
      "step": 5516
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 5516,
      "training_step_time": 9.957876920700073
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 3.130030393600464,
      "step": 5516
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 5516,
      "training_step_time": 9.900025367736816
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 3.1315441131591797,
      "step": 5516
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 5516,
      "training_step_time": 9.90135145187378
    },
    {
      "epoch": 2.10418701171875e-05,
      "model_forward_time": 3.4313509464263916,
      "step": 5516
    },
    {
      "epoch": 2.10418701171875e-05,
      "step": 5516,
      "training_step_time": 10.302107810974121
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 3.167534589767456,
      "step": 5517
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 5517,
      "training_step_time": 9.940722227096558
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 3.1340460777282715,
      "step": 5517
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 5517,
      "training_step_time": 9.912431001663208
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 3.1392035484313965,
      "step": 5517
    },
    {
      "dynamics_loss_avg": 0.0687117874622345,
      "epoch": 2.1045684814453126e-05,
      "step": 5517
    },
    {
      "action_loss_avg": 0.028483508620411158,
      "epoch": 2.1045684814453126e-05,
      "step": 5517
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 5517,
      "training_step_time": 9.917255640029907
    },
    {
      "epoch": 2.1045684814453126e-05,
      "model_forward_time": 3.134666681289673,
      "step": 5517
    },
    {
      "epoch": 2.1045684814453126e-05,
      "step": 5517,
      "training_step_time": 10.037786483764648
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 3.462773561477661,
      "step": 5518
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 5518,
      "training_step_time": 10.241965293884277
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 3.1400763988494873,
      "step": 5518
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 5518,
      "training_step_time": 9.913711547851562
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 3.1388449668884277,
      "step": 5518
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 5518,
      "training_step_time": 9.911505222320557
    },
    {
      "epoch": 2.104949951171875e-05,
      "model_forward_time": 3.1361300945281982,
      "step": 5518
    },
    {
      "epoch": 2.104949951171875e-05,
      "step": 5518,
      "training_step_time": 10.021276473999023
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 3.142437696456909,
      "step": 5519
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 5519,
      "training_step_time": 9.914100408554077
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 3.450984477996826,
      "step": 5519
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 5519,
      "training_step_time": 10.219613313674927
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 3.135868787765503,
      "step": 5519
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 5519,
      "training_step_time": 9.902753353118896
    },
    {
      "epoch": 2.1053314208984375e-05,
      "model_forward_time": 2.428346633911133,
      "step": 5519
    },
    {
      "epoch": 2.1053314208984375e-05,
      "step": 5519,
      "training_step_time": 10.018923282623291
    },
    {
      "epoch": 2.105712890625e-05,
      "grad_norm": 0.3001748323440552,
      "learning_rate": 1e-05,
      "loss": 0.0866,
      "step": 5520
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 3.15922212600708,
      "step": 5520
    },
    {
      "dynamics_loss_avg": 0.05039458926767111,
      "epoch": 2.105712890625e-05,
      "step": 5520
    },
    {
      "action_loss_avg": 0.020460901595652104,
      "epoch": 2.105712890625e-05,
      "step": 5520
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 5520,
      "training_step_time": 9.925729751586914
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 2.4295120239257812,
      "step": 5520
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 5520,
      "training_step_time": 7.449342727661133
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 3.133056163787842,
      "step": 5520
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 5520,
      "training_step_time": 10.164801836013794
    },
    {
      "epoch": 2.105712890625e-05,
      "model_forward_time": 3.1328418254852295,
      "step": 5520
    },
    {
      "epoch": 2.105712890625e-05,
      "step": 5520,
      "training_step_time": 10.023385047912598
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 3.162013530731201,
      "step": 5521
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 5521,
      "training_step_time": 9.933042764663696
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 3.1314289569854736,
      "step": 5521
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 5521,
      "training_step_time": 9.89918041229248
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 3.1332857608795166,
      "step": 5521
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 5521,
      "training_step_time": 9.898515462875366
    },
    {
      "epoch": 2.1060943603515625e-05,
      "model_forward_time": 3.1320013999938965,
      "step": 5521
    },
    {
      "epoch": 2.1060943603515625e-05,
      "step": 5521,
      "training_step_time": 10.27423906326294
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 3.1426596641540527,
      "step": 5522
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 5522,
      "training_step_time": 9.911584854125977
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 3.130762815475464,
      "step": 5522
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 5522,
      "training_step_time": 9.905395984649658
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 3.1353061199188232,
      "step": 5522
    },
    {
      "dynamics_loss_avg": 0.05810441225767136,
      "epoch": 2.106475830078125e-05,
      "step": 5522
    },
    {
      "action_loss_avg": 0.01930546360090375,
      "epoch": 2.106475830078125e-05,
      "step": 5522
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 5522,
      "training_step_time": 9.908949613571167
    },
    {
      "epoch": 2.106475830078125e-05,
      "model_forward_time": 2.433701515197754,
      "step": 5522
    },
    {
      "epoch": 2.106475830078125e-05,
      "step": 5522,
      "training_step_time": 10.275223731994629
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 3.159501552581787,
      "step": 5523
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 5523,
      "training_step_time": 10.211437940597534
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 2.4302315711975098,
      "step": 5523
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 5523,
      "training_step_time": 7.456331253051758
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 3.129838466644287,
      "step": 5523
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 5523,
      "training_step_time": 9.901042461395264
    },
    {
      "epoch": 2.1068572998046874e-05,
      "model_forward_time": 3.1304683685302734,
      "step": 5523
    },
    {
      "epoch": 2.1068572998046874e-05,
      "step": 5523,
      "training_step_time": 10.287349939346313
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 3.1846985816955566,
      "step": 5524
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 5524,
      "training_step_time": 9.962007522583008
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 3.1412789821624756,
      "step": 5524
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 5524,
      "training_step_time": 9.913394927978516
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 3.4313197135925293,
      "step": 5524
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 5524,
      "training_step_time": 10.206144332885742
    },
    {
      "epoch": 2.10723876953125e-05,
      "model_forward_time": 3.1391828060150146,
      "step": 5524
    },
    {
      "epoch": 2.10723876953125e-05,
      "step": 5524,
      "training_step_time": 10.021920204162598
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 3.1821389198303223,
      "step": 5525
    },
    {
      "dynamics_loss_avg": 0.06558232987299562,
      "epoch": 2.1076202392578126e-05,
      "step": 5525
    },
    {
      "action_loss_avg": 0.02517334669828415,
      "epoch": 2.1076202392578126e-05,
      "step": 5525
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 5525,
      "training_step_time": 9.96402645111084
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 3.141578197479248,
      "step": 5525
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 5525,
      "training_step_time": 9.915725469589233
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 3.426375389099121,
      "step": 5525
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 5525,
      "training_step_time": 10.196956872940063
    },
    {
      "epoch": 2.1076202392578126e-05,
      "model_forward_time": 3.135153293609619,
      "step": 5525
    },
    {
      "epoch": 2.1076202392578126e-05,
      "step": 5525,
      "training_step_time": 10.012566089630127
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 3.143965244293213,
      "step": 5526
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 5526,
      "training_step_time": 9.911312818527222
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 3.1380538940429688,
      "step": 5526
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 5526,
      "training_step_time": 9.908222913742065
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 2.4336841106414795,
      "step": 5526
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 5526,
      "training_step_time": 7.464517116546631
    },
    {
      "epoch": 2.108001708984375e-05,
      "model_forward_time": 3.1402153968811035,
      "step": 5526
    },
    {
      "epoch": 2.108001708984375e-05,
      "step": 5526,
      "training_step_time": 10.327486276626587
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 3.453106641769409,
      "step": 5527
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 5527,
      "training_step_time": 10.229125738143921
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 3.1418285369873047,
      "step": 5527
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 5527,
      "training_step_time": 9.919087886810303
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 3.141650676727295,
      "step": 5527
    },
    {
      "dynamics_loss_avg": 0.0697294794023037,
      "epoch": 2.1083831787109376e-05,
      "step": 5527
    },
    {
      "action_loss_avg": 0.028017344418913126,
      "epoch": 2.1083831787109376e-05,
      "step": 5527
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 5527,
      "training_step_time": 9.91657304763794
    },
    {
      "epoch": 2.1083831787109376e-05,
      "model_forward_time": 3.1434831619262695,
      "step": 5527
    },
    {
      "epoch": 2.1083831787109376e-05,
      "step": 5527,
      "training_step_time": 10.31324028968811
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 3.4451904296875,
      "step": 5528
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 5528,
      "training_step_time": 10.216358423233032
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 3.142599582672119,
      "step": 5528
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 5528,
      "training_step_time": 9.91275429725647
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 3.1350324153900146,
      "step": 5528
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 5528,
      "training_step_time": 9.90717601776123
    },
    {
      "epoch": 2.1087646484375e-05,
      "model_forward_time": 3.1438722610473633,
      "step": 5528
    },
    {
      "epoch": 2.1087646484375e-05,
      "step": 5528,
      "training_step_time": 10.01994514465332
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 3.186572313308716,
      "step": 5529
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 5529,
      "training_step_time": 9.952695846557617
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 2.434035539627075,
      "step": 5529
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 5529,
      "training_step_time": 7.459119558334351
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 3.382274866104126,
      "step": 5529
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 5529,
      "training_step_time": 10.146039247512817
    },
    {
      "epoch": 2.1091461181640625e-05,
      "model_forward_time": 3.134575128555298,
      "step": 5529
    },
    {
      "epoch": 2.1091461181640625e-05,
      "step": 5529,
      "training_step_time": 10.014175176620483
    },
    {
      "epoch": 2.109527587890625e-05,
      "grad_norm": 0.51922607421875,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 5530
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 3.1807591915130615,
      "step": 5530
    },
    {
      "dynamics_loss_avg": 0.0617114519700408,
      "epoch": 2.109527587890625e-05,
      "step": 5530
    },
    {
      "action_loss_avg": 0.023969849455170333,
      "epoch": 2.109527587890625e-05,
      "step": 5530
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 5530,
      "training_step_time": 9.953600883483887
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 3.1384799480438232,
      "step": 5530
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 5530,
      "training_step_time": 9.907161712646484
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 3.1394004821777344,
      "step": 5530
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 5530,
      "training_step_time": 10.19185495376587
    },
    {
      "epoch": 2.109527587890625e-05,
      "model_forward_time": 3.149991512298584,
      "step": 5530
    },
    {
      "epoch": 2.109527587890625e-05,
      "step": 5530,
      "training_step_time": 10.241597652435303
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 3.172306776046753,
      "step": 5531
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 5531,
      "training_step_time": 9.94475269317627
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 3.1406285762786865,
      "step": 5531
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 5531,
      "training_step_time": 9.914172172546387
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 3.145066022872925,
      "step": 5531
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 5531,
      "training_step_time": 9.919211149215698
    },
    {
      "epoch": 2.1099090576171874e-05,
      "model_forward_time": 3.4546444416046143,
      "step": 5531
    },
    {
      "epoch": 2.1099090576171874e-05,
      "step": 5531,
      "training_step_time": 10.32126784324646
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 3.1915078163146973,
      "step": 5532
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 5532,
      "training_step_time": 9.970603704452515
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 3.14430832862854,
      "step": 5532
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 5532,
      "training_step_time": 9.91780400276184
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 3.1430437564849854,
      "step": 5532
    },
    {
      "dynamics_loss_avg": 0.06134493667632342,
      "epoch": 2.11029052734375e-05,
      "step": 5532
    },
    {
      "action_loss_avg": 0.025424278061836957,
      "epoch": 2.11029052734375e-05,
      "step": 5532
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 5532,
      "training_step_time": 9.915152311325073
    },
    {
      "epoch": 2.11029052734375e-05,
      "model_forward_time": 3.1479015350341797,
      "step": 5532
    },
    {
      "epoch": 2.11029052734375e-05,
      "step": 5532,
      "training_step_time": 10.321932077407837
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 3.1485865116119385,
      "step": 5533
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 5533,
      "training_step_time": 10.204476594924927
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 3.1379058361053467,
      "step": 5533
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 5533,
      "training_step_time": 9.908905029296875
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 2.43757700920105,
      "step": 5533
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 5533,
      "training_step_time": 7.471757173538208
    },
    {
      "epoch": 2.1106719970703126e-05,
      "model_forward_time": 3.1358373165130615,
      "step": 5533
    },
    {
      "epoch": 2.1106719970703126e-05,
      "step": 5533,
      "training_step_time": 10.304423332214355
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 3.1815054416656494,
      "step": 5534
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 5534,
      "training_step_time": 9.952735185623169
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 3.3933191299438477,
      "step": 5534
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 5534,
      "training_step_time": 10.15549349784851
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 3.1340579986572266,
      "step": 5534
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 5534,
      "training_step_time": 9.896993637084961
    },
    {
      "epoch": 2.111053466796875e-05,
      "model_forward_time": 3.126851797103882,
      "step": 5534
    },
    {
      "epoch": 2.111053466796875e-05,
      "step": 5534,
      "training_step_time": 10.017534732818604
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 3.191654920578003,
      "step": 5535
    },
    {
      "dynamics_loss_avg": 0.06286514215171338,
      "epoch": 2.1114349365234376e-05,
      "step": 5535
    },
    {
      "action_loss_avg": 0.022242658818140627,
      "epoch": 2.1114349365234376e-05,
      "step": 5535
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 5535,
      "training_step_time": 9.965814590454102
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 3.130070209503174,
      "step": 5535
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 5535,
      "training_step_time": 9.898409128189087
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 3.1326141357421875,
      "step": 5535
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 5535,
      "training_step_time": 10.16957950592041
    },
    {
      "epoch": 2.1114349365234376e-05,
      "model_forward_time": 3.1354050636291504,
      "step": 5535
    },
    {
      "epoch": 2.1114349365234376e-05,
      "step": 5535,
      "training_step_time": 10.310694217681885
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 3.187366247177124,
      "step": 5536
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 5536,
      "training_step_time": 9.961036205291748
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 3.1381330490112305,
      "step": 5536
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 5536,
      "training_step_time": 9.910394668579102
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 3.1345975399017334,
      "step": 5536
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 5536,
      "training_step_time": 9.901351690292358
    },
    {
      "epoch": 2.11181640625e-05,
      "model_forward_time": 3.387244701385498,
      "step": 5536
    },
    {
      "epoch": 2.11181640625e-05,
      "step": 5536,
      "training_step_time": 10.299960136413574
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 3.154895305633545,
      "step": 5537
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 5537,
      "training_step_time": 9.928511142730713
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 3.1322219371795654,
      "step": 5537
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 5537,
      "training_step_time": 9.900092124938965
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 3.135007619857788,
      "step": 5537
    },
    {
      "dynamics_loss_avg": 0.07552288044244052,
      "epoch": 2.1121978759765625e-05,
      "step": 5537
    },
    {
      "action_loss_avg": 0.027205259213224052,
      "epoch": 2.1121978759765625e-05,
      "step": 5537
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 5537,
      "training_step_time": 9.90555214881897
    },
    {
      "epoch": 2.1121978759765625e-05,
      "model_forward_time": 3.13606858253479,
      "step": 5537
    },
    {
      "epoch": 2.1121978759765625e-05,
      "step": 5537,
      "training_step_time": 10.023352146148682
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 3.435436964035034,
      "step": 5538
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 5538,
      "training_step_time": 10.20816707611084
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 3.1376471519470215,
      "step": 5538
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 5538,
      "training_step_time": 9.908764362335205
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 3.1330299377441406,
      "step": 5538
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 5538,
      "training_step_time": 9.903502941131592
    },
    {
      "epoch": 2.112579345703125e-05,
      "model_forward_time": 3.1347458362579346,
      "step": 5538
    },
    {
      "epoch": 2.112579345703125e-05,
      "step": 5538,
      "training_step_time": 10.293160915374756
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 3.183262825012207,
      "step": 5539
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 5539,
      "training_step_time": 9.954367637634277
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 3.4133851528167725,
      "step": 5539
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 5539,
      "training_step_time": 10.184280157089233
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 3.1337664127349854,
      "step": 5539
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 5539,
      "training_step_time": 9.899046897888184
    },
    {
      "epoch": 2.1129608154296874e-05,
      "model_forward_time": 3.1335370540618896,
      "step": 5539
    },
    {
      "epoch": 2.1129608154296874e-05,
      "step": 5539,
      "training_step_time": 10.023097038269043
    },
    {
      "epoch": 2.11334228515625e-05,
      "grad_norm": 0.27702081203460693,
      "learning_rate": 1e-05,
      "loss": 0.092,
      "step": 5540
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 3.175806760787964,
      "step": 5540
    },
    {
      "dynamics_loss_avg": 0.060992293432354926,
      "epoch": 2.11334228515625e-05,
      "step": 5540
    },
    {
      "action_loss_avg": 0.018907170556485654,
      "epoch": 2.11334228515625e-05,
      "step": 5540
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 5540,
      "training_step_time": 9.94624638557434
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 3.1305999755859375,
      "step": 5540
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 5540,
      "training_step_time": 9.895245790481567
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 3.3892033100128174,
      "step": 5540
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 5540,
      "training_step_time": 10.152665853500366
    },
    {
      "epoch": 2.11334228515625e-05,
      "model_forward_time": 3.131807565689087,
      "step": 5540
    },
    {
      "epoch": 2.11334228515625e-05,
      "step": 5540,
      "training_step_time": 10.276692390441895
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 3.1674649715423584,
      "step": 5541
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 5541,
      "training_step_time": 9.92619776725769
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 3.1269400119781494,
      "step": 5541
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 5541,
      "training_step_time": 9.884971141815186
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 3.1315317153930664,
      "step": 5541
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 5541,
      "training_step_time": 9.901057958602905
    },
    {
      "epoch": 2.1137237548828126e-05,
      "model_forward_time": 3.408174991607666,
      "step": 5541
    },
    {
      "epoch": 2.1137237548828126e-05,
      "step": 5541,
      "training_step_time": 10.303553819656372
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 3.1634180545806885,
      "step": 5542
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 5542,
      "training_step_time": 9.928105115890503
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 3.131786346435547,
      "step": 5542
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 5542,
      "training_step_time": 9.898497819900513
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 2.431349992752075,
      "step": 5542
    },
    {
      "dynamics_loss_avg": 0.0711141237989068,
      "epoch": 2.114105224609375e-05,
      "step": 5542
    },
    {
      "action_loss_avg": 0.022731700073927642,
      "epoch": 2.114105224609375e-05,
      "step": 5542
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 5542,
      "training_step_time": 7.458280801773071
    },
    {
      "epoch": 2.114105224609375e-05,
      "model_forward_time": 3.1289777755737305,
      "step": 5542
    },
    {
      "epoch": 2.114105224609375e-05,
      "step": 5542,
      "training_step_time": 10.019574165344238
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 3.484783887863159,
      "step": 5543
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 5543,
      "training_step_time": 10.253398656845093
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 3.1315906047821045,
      "step": 5543
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 5543,
      "training_step_time": 9.8973228931427
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 3.137465715408325,
      "step": 5543
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 5543,
      "training_step_time": 9.902729034423828
    },
    {
      "epoch": 2.1144866943359376e-05,
      "model_forward_time": 3.1290457248687744,
      "step": 5543
    },
    {
      "epoch": 2.1144866943359376e-05,
      "step": 5543,
      "training_step_time": 10.008406162261963
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 3.1582586765289307,
      "step": 5544
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 5544,
      "training_step_time": 9.92570185661316
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 3.400979518890381,
      "step": 5544
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 5544,
      "training_step_time": 10.168409585952759
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 3.140995502471924,
      "step": 5544
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 5544,
      "training_step_time": 9.910122394561768
    },
    {
      "epoch": 2.1148681640625e-05,
      "model_forward_time": 3.139744997024536,
      "step": 5544
    },
    {
      "epoch": 2.1148681640625e-05,
      "step": 5544,
      "training_step_time": 10.321741580963135
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 2.448770761489868,
      "step": 5545
    },
    {
      "dynamics_loss_avg": 0.06976677440106868,
      "epoch": 2.1152496337890625e-05,
      "step": 5545
    },
    {
      "action_loss_avg": 0.02480843896046281,
      "epoch": 2.1152496337890625e-05,
      "step": 5545
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 5545,
      "training_step_time": 7.4814677238464355
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 3.138591766357422,
      "step": 5545
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 5545,
      "training_step_time": 9.898629665374756
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 3.435899257659912,
      "step": 5545
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 5545,
      "training_step_time": 10.203306674957275
    },
    {
      "epoch": 2.1152496337890625e-05,
      "model_forward_time": 2.4371516704559326,
      "step": 5545
    },
    {
      "epoch": 2.1152496337890625e-05,
      "step": 5545,
      "training_step_time": 10.013463735580444
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 3.1552505493164062,
      "step": 5546
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 5546,
      "training_step_time": 9.920082092285156
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 3.1387364864349365,
      "step": 5546
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 5546,
      "training_step_time": 9.905253648757935
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 2.432677984237671,
      "step": 5546
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 5546,
      "training_step_time": 7.455808401107788
    },
    {
      "epoch": 2.115631103515625e-05,
      "model_forward_time": 3.1331841945648193,
      "step": 5546
    },
    {
      "epoch": 2.115631103515625e-05,
      "step": 5546,
      "training_step_time": 10.265776872634888
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 3.4193546772003174,
      "step": 5547
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 5547,
      "training_step_time": 10.187443256378174
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 3.1382665634155273,
      "step": 5547
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 5547,
      "training_step_time": 9.9026038646698
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 3.130927085876465,
      "step": 5547
    },
    {
      "dynamics_loss_avg": 0.06155438963323832,
      "epoch": 2.1160125732421874e-05,
      "step": 5547
    },
    {
      "action_loss_avg": 0.024829412624239922,
      "epoch": 2.1160125732421874e-05,
      "step": 5547
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 5547,
      "training_step_time": 9.894860982894897
    },
    {
      "epoch": 2.1160125732421874e-05,
      "model_forward_time": 3.1363346576690674,
      "step": 5547
    },
    {
      "epoch": 2.1160125732421874e-05,
      "step": 5547,
      "training_step_time": 10.268280267715454
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 3.161043167114258,
      "step": 5548
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 5548,
      "training_step_time": 10.190502405166626
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 3.1376616954803467,
      "step": 5548
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 5548,
      "training_step_time": 9.90189266204834
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 3.1349098682403564,
      "step": 5548
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 5548,
      "training_step_time": 9.9034583568573
    },
    {
      "epoch": 2.11639404296875e-05,
      "model_forward_time": 3.1437671184539795,
      "step": 5548
    },
    {
      "epoch": 2.11639404296875e-05,
      "step": 5548,
      "training_step_time": 10.236513137817383
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 3.1712849140167236,
      "step": 5549
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 5549,
      "training_step_time": 9.946390390396118
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 3.4251859188079834,
      "step": 5549
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 5549,
      "training_step_time": 10.193359375
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 3.13576078414917,
      "step": 5549
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 5549,
      "training_step_time": 9.908586025238037
    },
    {
      "epoch": 2.1167755126953126e-05,
      "model_forward_time": 3.144507884979248,
      "step": 5549
    },
    {
      "epoch": 2.1167755126953126e-05,
      "step": 5549,
      "training_step_time": 10.291594505310059
    },
    {
      "epoch": 2.117156982421875e-05,
      "grad_norm": 0.35063284635543823,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 5550
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 3.173129081726074,
      "step": 5550
    },
    {
      "dynamics_loss_avg": 0.07665897831320763,
      "epoch": 2.117156982421875e-05,
      "step": 5550
    },
    {
      "action_loss_avg": 0.031485125049948694,
      "epoch": 2.117156982421875e-05,
      "step": 5550
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 5550,
      "training_step_time": 9.947041988372803
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 3.138155460357666,
      "step": 5550
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 5550,
      "training_step_time": 9.909131526947021
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 3.1520981788635254,
      "step": 5550
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 5550,
      "training_step_time": 10.209629535675049
    },
    {
      "epoch": 2.117156982421875e-05,
      "model_forward_time": 3.139514923095703,
      "step": 5550
    },
    {
      "epoch": 2.117156982421875e-05,
      "step": 5550,
      "training_step_time": 10.306705951690674
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 3.1740078926086426,
      "step": 5551
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 5551,
      "training_step_time": 9.95109510421753
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 2.4362847805023193,
      "step": 5551
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 5551,
      "training_step_time": 7.467059135437012
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 3.1383180618286133,
      "step": 5551
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 5551,
      "training_step_time": 9.912989854812622
    },
    {
      "epoch": 2.1175384521484376e-05,
      "model_forward_time": 3.140613555908203,
      "step": 5551
    },
    {
      "epoch": 2.1175384521484376e-05,
      "step": 5551,
      "training_step_time": 10.28495740890503
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 3.1908206939697266,
      "step": 5552
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 5552,
      "training_step_time": 9.965895414352417
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 3.1428370475769043,
      "step": 5552
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 5552,
      "training_step_time": 9.916521310806274
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 3.1469545364379883,
      "step": 5552
    },
    {
      "dynamics_loss_avg": 0.052697858028113845,
      "epoch": 2.117919921875e-05,
      "step": 5552
    },
    {
      "action_loss_avg": 0.022057279665023088,
      "epoch": 2.117919921875e-05,
      "step": 5552
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 5552,
      "training_step_time": 9.926832675933838
    },
    {
      "epoch": 2.117919921875e-05,
      "model_forward_time": 3.1402461528778076,
      "step": 5552
    },
    {
      "epoch": 2.117919921875e-05,
      "step": 5552,
      "training_step_time": 10.309723138809204
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 3.166173219680786,
      "step": 5553
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 5553,
      "training_step_time": 10.21634292602539
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 3.1564462184906006,
      "step": 5553
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 5553,
      "training_step_time": 9.932933330535889
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 3.148681879043579,
      "step": 5553
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 5553,
      "training_step_time": 9.924489259719849
    },
    {
      "epoch": 2.1183013916015625e-05,
      "model_forward_time": 3.1440186500549316,
      "step": 5553
    },
    {
      "epoch": 2.1183013916015625e-05,
      "step": 5553,
      "training_step_time": 10.291122674942017
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 3.1691839694976807,
      "step": 5554
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 5554,
      "training_step_time": 9.94419240951538
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 3.3997416496276855,
      "step": 5554
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 5554,
      "training_step_time": 10.168346881866455
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 3.1407368183135986,
      "step": 5554
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 5554,
      "training_step_time": 9.913739204406738
    },
    {
      "epoch": 2.118682861328125e-05,
      "model_forward_time": 3.134824752807617,
      "step": 5554
    },
    {
      "epoch": 2.118682861328125e-05,
      "step": 5554,
      "training_step_time": 10.303704023361206
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 3.172644853591919,
      "step": 5555
    },
    {
      "dynamics_loss_avg": 0.08060592850670219,
      "epoch": 2.1190643310546874e-05,
      "step": 5555
    },
    {
      "action_loss_avg": 0.030930406460538507,
      "epoch": 2.1190643310546874e-05,
      "step": 5555
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 5555,
      "training_step_time": 9.94299054145813
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 3.141207695007324,
      "step": 5555
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 5555,
      "training_step_time": 9.91189169883728
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 3.1326234340667725,
      "step": 5555
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 5555,
      "training_step_time": 10.168532609939575
    },
    {
      "epoch": 2.1190643310546874e-05,
      "model_forward_time": 3.1364734172821045,
      "step": 5555
    },
    {
      "epoch": 2.1190643310546874e-05,
      "step": 5555,
      "training_step_time": 10.019390344619751
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 3.14178466796875,
      "step": 5556
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 5556,
      "training_step_time": 9.914545059204102
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 3.13950252532959,
      "step": 5556
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 5556,
      "training_step_time": 9.905507326126099
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 3.1386919021606445,
      "step": 5556
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 5556,
      "training_step_time": 9.903591871261597
    },
    {
      "epoch": 2.11944580078125e-05,
      "model_forward_time": 2.7286956310272217,
      "step": 5556
    },
    {
      "epoch": 2.11944580078125e-05,
      "step": 5556,
      "training_step_time": 10.017467737197876
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 3.168424129486084,
      "step": 5557
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 5557,
      "training_step_time": 9.934763193130493
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 3.1407647132873535,
      "step": 5557
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 5557,
      "training_step_time": 9.905361652374268
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 3.138475179672241,
      "step": 5557
    },
    {
      "dynamics_loss_avg": 0.06788338646292687,
      "epoch": 2.1198272705078126e-05,
      "step": 5557
    },
    {
      "action_loss_avg": 0.02828646469861269,
      "epoch": 2.1198272705078126e-05,
      "step": 5557
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 5557,
      "training_step_time": 9.904188871383667
    },
    {
      "epoch": 2.1198272705078126e-05,
      "model_forward_time": 3.1452536582946777,
      "step": 5557
    },
    {
      "epoch": 2.1198272705078126e-05,
      "step": 5557,
      "training_step_time": 10.321957349777222
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 3.477452039718628,
      "step": 5558
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 5558,
      "training_step_time": 10.247286558151245
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 3.145781993865967,
      "step": 5558
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 5558,
      "training_step_time": 9.910622835159302
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 3.1425063610076904,
      "step": 5558
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 5558,
      "training_step_time": 9.909922122955322
    },
    {
      "epoch": 2.120208740234375e-05,
      "model_forward_time": 3.149160385131836,
      "step": 5558
    },
    {
      "epoch": 2.120208740234375e-05,
      "step": 5558,
      "training_step_time": 10.32247257232666
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 3.1860241889953613,
      "step": 5559
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 5559,
      "training_step_time": 9.965414762496948
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 3.143084764480591,
      "step": 5559
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 5559,
      "training_step_time": 10.204647064208984
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 3.144301652908325,
      "step": 5559
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 5559,
      "training_step_time": 9.913740873336792
    },
    {
      "epoch": 2.1205902099609376e-05,
      "model_forward_time": 3.1459248065948486,
      "step": 5559
    },
    {
      "epoch": 2.1205902099609376e-05,
      "step": 5559,
      "training_step_time": 10.334474325180054
    },
    {
      "epoch": 2.1209716796875e-05,
      "grad_norm": 0.6137920618057251,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 5560
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 3.181610584259033,
      "step": 5560
    },
    {
      "dynamics_loss_avg": 0.051233100891113284,
      "epoch": 2.1209716796875e-05,
      "step": 5560
    },
    {
      "action_loss_avg": 0.025368005223572253,
      "epoch": 2.1209716796875e-05,
      "step": 5560
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 5560,
      "training_step_time": 9.958605527877808
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 3.1518287658691406,
      "step": 5560
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 5560,
      "training_step_time": 9.933443784713745
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 3.4050066471099854,
      "step": 5560
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 5560,
      "training_step_time": 10.17340350151062
    },
    {
      "epoch": 2.1209716796875e-05,
      "model_forward_time": 3.1486785411834717,
      "step": 5560
    },
    {
      "epoch": 2.1209716796875e-05,
      "step": 5560,
      "training_step_time": 10.030794382095337
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 3.1781020164489746,
      "step": 5561
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 5561,
      "training_step_time": 9.964413166046143
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 3.134408473968506,
      "step": 5561
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 5561,
      "training_step_time": 9.908525705337524
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 3.144260883331299,
      "step": 5561
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 5561,
      "training_step_time": 9.916872501373291
    },
    {
      "epoch": 2.1213531494140625e-05,
      "model_forward_time": 3.139864444732666,
      "step": 5561
    },
    {
      "epoch": 2.1213531494140625e-05,
      "step": 5561,
      "training_step_time": 10.30878758430481
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 3.176379442214966,
      "step": 5562
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 5562,
      "training_step_time": 9.952121257781982
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 3.1390533447265625,
      "step": 5562
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 5562,
      "training_step_time": 9.910831928253174
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 3.141850471496582,
      "step": 5562
    },
    {
      "dynamics_loss_avg": 0.05776070896536112,
      "epoch": 2.121734619140625e-05,
      "step": 5562
    },
    {
      "action_loss_avg": 0.026211747666820884,
      "epoch": 2.121734619140625e-05,
      "step": 5562
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 5562,
      "training_step_time": 9.912371397018433
    },
    {
      "epoch": 2.121734619140625e-05,
      "model_forward_time": 3.1432549953460693,
      "step": 5562
    },
    {
      "epoch": 2.121734619140625e-05,
      "step": 5562,
      "training_step_time": 10.272570610046387
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 3.4468817710876465,
      "step": 5563
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 5563,
      "training_step_time": 10.219080686569214
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 3.1421921253204346,
      "step": 5563
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 5563,
      "training_step_time": 9.912852048873901
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 3.146613121032715,
      "step": 5563
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 5563,
      "training_step_time": 9.920377254486084
    },
    {
      "epoch": 2.1221160888671874e-05,
      "model_forward_time": 3.1427102088928223,
      "step": 5563
    },
    {
      "epoch": 2.1221160888671874e-05,
      "step": 5563,
      "training_step_time": 10.041315793991089
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 3.1727473735809326,
      "step": 5564
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 5564,
      "training_step_time": 9.948290586471558
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 3.1482291221618652,
      "step": 5564
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 5564,
      "training_step_time": 10.184852838516235
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 3.1425771713256836,
      "step": 5564
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 5564,
      "training_step_time": 9.914710760116577
    },
    {
      "epoch": 2.12249755859375e-05,
      "model_forward_time": 3.1353864669799805,
      "step": 5564
    },
    {
      "epoch": 2.12249755859375e-05,
      "step": 5564,
      "training_step_time": 10.29558801651001
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 3.1649248600006104,
      "step": 5565
    },
    {
      "dynamics_loss_avg": 0.0665585121139884,
      "epoch": 2.1228790283203126e-05,
      "step": 5565
    },
    {
      "action_loss_avg": 0.026061677653342486,
      "epoch": 2.1228790283203126e-05,
      "step": 5565
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 5565,
      "training_step_time": 9.93605351448059
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 3.138432264328003,
      "step": 5565
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 5565,
      "training_step_time": 9.910137176513672
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 3.411271810531616,
      "step": 5565
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 5565,
      "training_step_time": 10.181476593017578
    },
    {
      "epoch": 2.1228790283203126e-05,
      "model_forward_time": 3.1350619792938232,
      "step": 5565
    },
    {
      "epoch": 2.1228790283203126e-05,
      "step": 5565,
      "training_step_time": 10.308643341064453
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 3.158691644668579,
      "step": 5566
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 5566,
      "training_step_time": 9.934855937957764
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 2.433265209197998,
      "step": 5566
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 5566,
      "training_step_time": 7.460140943527222
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 3.1285367012023926,
      "step": 5566
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 5566,
      "training_step_time": 9.895792484283447
    },
    {
      "epoch": 2.123260498046875e-05,
      "model_forward_time": 3.136542320251465,
      "step": 5566
    },
    {
      "epoch": 2.123260498046875e-05,
      "step": 5566,
      "training_step_time": 10.30312967300415
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 3.181431531906128,
      "step": 5567
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 5567,
      "training_step_time": 9.952069282531738
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 3.136958599090576,
      "step": 5567
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 5567,
      "training_step_time": 9.913272380828857
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 3.135528564453125,
      "step": 5567
    },
    {
      "dynamics_loss_avg": 0.05101324375718832,
      "epoch": 2.1236419677734376e-05,
      "step": 5567
    },
    {
      "action_loss_avg": 0.017585266940295698,
      "epoch": 2.1236419677734376e-05,
      "step": 5567
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 5567,
      "training_step_time": 9.906910419464111
    },
    {
      "epoch": 2.1236419677734376e-05,
      "model_forward_time": 3.1398279666900635,
      "step": 5567
    },
    {
      "epoch": 2.1236419677734376e-05,
      "step": 5567,
      "training_step_time": 10.300484657287598
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 3.4442882537841797,
      "step": 5568
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 5568,
      "training_step_time": 10.214642763137817
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 3.1412014961242676,
      "step": 5568
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 5568,
      "training_step_time": 9.911219358444214
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 3.1382408142089844,
      "step": 5568
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 5568,
      "training_step_time": 9.913472414016724
    },
    {
      "epoch": 2.1240234375e-05,
      "model_forward_time": 3.1426713466644287,
      "step": 5568
    },
    {
      "epoch": 2.1240234375e-05,
      "step": 5568,
      "training_step_time": 10.022459983825684
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 2.467512369155884,
      "step": 5569
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 5569,
      "training_step_time": 7.497654438018799
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 3.4384467601776123,
      "step": 5569
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 5569,
      "training_step_time": 10.212766170501709
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 3.143036127090454,
      "step": 5569
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 5569,
      "training_step_time": 9.91745400428772
    },
    {
      "epoch": 2.1244049072265625e-05,
      "model_forward_time": 3.1423838138580322,
      "step": 5569
    },
    {
      "epoch": 2.1244049072265625e-05,
      "step": 5569,
      "training_step_time": 10.019163608551025
    },
    {
      "epoch": 2.124786376953125e-05,
      "grad_norm": 0.45839032530784607,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 5570
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 3.17816424369812,
      "step": 5570
    },
    {
      "dynamics_loss_avg": 0.0610954437404871,
      "epoch": 2.124786376953125e-05,
      "step": 5570
    },
    {
      "action_loss_avg": 0.02095663156360388,
      "epoch": 2.124786376953125e-05,
      "step": 5570
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 5570,
      "training_step_time": 9.956884860992432
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 3.139439821243286,
      "step": 5570
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 5570,
      "training_step_time": 9.913048267364502
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 3.1379730701446533,
      "step": 5570
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 5570,
      "training_step_time": 10.207396268844604
    },
    {
      "epoch": 2.124786376953125e-05,
      "model_forward_time": 3.1413934230804443,
      "step": 5570
    },
    {
      "epoch": 2.124786376953125e-05,
      "step": 5570,
      "training_step_time": 10.297825813293457
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 3.186188220977783,
      "step": 5571
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 5571,
      "training_step_time": 9.963830947875977
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 3.136979579925537,
      "step": 5571
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 5571,
      "training_step_time": 9.907991886138916
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 3.136122703552246,
      "step": 5571
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 5571,
      "training_step_time": 9.905956745147705
    },
    {
      "epoch": 2.1251678466796874e-05,
      "model_forward_time": 3.414618730545044,
      "step": 5571
    },
    {
      "epoch": 2.1251678466796874e-05,
      "step": 5571,
      "training_step_time": 10.27750849723816
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 3.179818630218506,
      "step": 5572
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 5572,
      "training_step_time": 9.950850009918213
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 3.1321749687194824,
      "step": 5572
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 5572,
      "training_step_time": 9.906126260757446
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 3.1377012729644775,
      "step": 5572
    },
    {
      "dynamics_loss_avg": 0.06218731626868248,
      "epoch": 2.12554931640625e-05,
      "step": 5572
    },
    {
      "action_loss_avg": 0.028062459360808135,
      "epoch": 2.12554931640625e-05,
      "step": 5572
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 5572,
      "training_step_time": 9.914306640625
    },
    {
      "epoch": 2.12554931640625e-05,
      "model_forward_time": 3.134535074234009,
      "step": 5572
    },
    {
      "epoch": 2.12554931640625e-05,
      "step": 5572,
      "training_step_time": 10.032476425170898
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 3.149693250656128,
      "step": 5573
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 5573,
      "training_step_time": 10.200302600860596
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 3.133737325668335,
      "step": 5573
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 5573,
      "training_step_time": 9.907464981079102
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 2.4373462200164795,
      "step": 5573
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 5573,
      "training_step_time": 7.46685528755188
    },
    {
      "epoch": 2.1259307861328127e-05,
      "model_forward_time": 2.431917905807495,
      "step": 5573
    },
    {
      "epoch": 2.1259307861328127e-05,
      "step": 5573,
      "training_step_time": 10.306367874145508
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 3.1758131980895996,
      "step": 5574
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 5574,
      "training_step_time": 9.947280883789062
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 3.141549587249756,
      "step": 5574
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 5574,
      "training_step_time": 10.194389820098877
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 3.1398799419403076,
      "step": 5574
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 5574,
      "training_step_time": 9.913152933120728
    },
    {
      "epoch": 2.126312255859375e-05,
      "model_forward_time": 3.1358110904693604,
      "step": 5574
    },
    {
      "epoch": 2.126312255859375e-05,
      "step": 5574,
      "training_step_time": 10.030939817428589
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 3.168701171875,
      "step": 5575
    },
    {
      "dynamics_loss_avg": 0.07298203520476818,
      "epoch": 2.1266937255859376e-05,
      "step": 5575
    },
    {
      "action_loss_avg": 0.025928351283073425,
      "epoch": 2.1266937255859376e-05,
      "step": 5575
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 5575,
      "training_step_time": 9.944215774536133
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 3.1354548931121826,
      "step": 5575
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 5575,
      "training_step_time": 9.907012224197388
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 3.1372199058532715,
      "step": 5575
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 5575,
      "training_step_time": 10.187520980834961
    },
    {
      "epoch": 2.1266937255859376e-05,
      "model_forward_time": 3.141721248626709,
      "step": 5575
    },
    {
      "epoch": 2.1266937255859376e-05,
      "step": 5575,
      "training_step_time": 10.319963455200195
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 3.157803535461426,
      "step": 5576
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 5576,
      "training_step_time": 9.936529874801636
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 3.1434288024902344,
      "step": 5576
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 5576,
      "training_step_time": 9.919433116912842
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 3.1402952671051025,
      "step": 5576
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 5576,
      "training_step_time": 9.921924352645874
    },
    {
      "epoch": 2.1270751953125e-05,
      "model_forward_time": 3.399714708328247,
      "step": 5576
    },
    {
      "epoch": 2.1270751953125e-05,
      "step": 5576,
      "training_step_time": 10.27352261543274
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 3.186854362487793,
      "step": 5577
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 5577,
      "training_step_time": 9.968067646026611
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 3.143685817718506,
      "step": 5577
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 5577,
      "training_step_time": 9.918206453323364
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 3.1375887393951416,
      "step": 5577
    },
    {
      "dynamics_loss_avg": 0.062078004330396654,
      "epoch": 2.1274566650390625e-05,
      "step": 5577
    },
    {
      "action_loss_avg": 0.02248764969408512,
      "epoch": 2.1274566650390625e-05,
      "step": 5577
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 5577,
      "training_step_time": 9.917659282684326
    },
    {
      "epoch": 2.1274566650390625e-05,
      "model_forward_time": 3.1389455795288086,
      "step": 5577
    },
    {
      "epoch": 2.1274566650390625e-05,
      "step": 5577,
      "training_step_time": 10.042112827301025
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 3.4297544956207275,
      "step": 5578
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 5578,
      "training_step_time": 10.202454090118408
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 3.1401262283325195,
      "step": 5578
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 5578,
      "training_step_time": 9.911959171295166
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 3.135533094406128,
      "step": 5578
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 5578,
      "training_step_time": 9.908972978591919
    },
    {
      "epoch": 2.127838134765625e-05,
      "model_forward_time": 3.1375439167022705,
      "step": 5578
    },
    {
      "epoch": 2.127838134765625e-05,
      "step": 5578,
      "training_step_time": 10.300675868988037
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 3.146440267562866,
      "step": 5579
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 5579,
      "training_step_time": 9.914927244186401
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 3.1329586505889893,
      "step": 5579
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 5579,
      "training_step_time": 10.183650970458984
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 3.131517171859741,
      "step": 5579
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 5579,
      "training_step_time": 9.901281356811523
    },
    {
      "epoch": 2.1282196044921874e-05,
      "model_forward_time": 3.135955572128296,
      "step": 5579
    },
    {
      "epoch": 2.1282196044921874e-05,
      "step": 5579,
      "training_step_time": 10.025345087051392
    },
    {
      "epoch": 2.12860107421875e-05,
      "grad_norm": 0.7074832320213318,
      "learning_rate": 1e-05,
      "loss": 0.0852,
      "step": 5580
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 3.1585466861724854,
      "step": 5580
    },
    {
      "dynamics_loss_avg": 0.06610840894281864,
      "epoch": 2.12860107421875e-05,
      "step": 5580
    },
    {
      "action_loss_avg": 0.02462987871840596,
      "epoch": 2.12860107421875e-05,
      "step": 5580
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 5580,
      "training_step_time": 9.936211109161377
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 3.1487464904785156,
      "step": 5580
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 5580,
      "training_step_time": 9.922377347946167
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 3.1496779918670654,
      "step": 5580
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 5580,
      "training_step_time": 10.199077606201172
    },
    {
      "epoch": 2.12860107421875e-05,
      "model_forward_time": 2.441603660583496,
      "step": 5580
    },
    {
      "epoch": 2.12860107421875e-05,
      "step": 5580,
      "training_step_time": 10.297686100006104
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 3.1581337451934814,
      "step": 5581
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 5581,
      "training_step_time": 9.941315174102783
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 3.1408753395080566,
      "step": 5581
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 5581,
      "training_step_time": 9.913599491119385
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 3.148098945617676,
      "step": 5581
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 5581,
      "training_step_time": 9.9201979637146
    },
    {
      "epoch": 2.1289825439453127e-05,
      "model_forward_time": 3.1421589851379395,
      "step": 5581
    },
    {
      "epoch": 2.1289825439453127e-05,
      "step": 5581,
      "training_step_time": 10.302167177200317
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 3.183753252029419,
      "step": 5582
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 5582,
      "training_step_time": 9.957762002944946
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 3.1379923820495605,
      "step": 5582
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 5582,
      "training_step_time": 9.912119626998901
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 3.1416091918945312,
      "step": 5582
    },
    {
      "dynamics_loss_avg": 0.06269548628479242,
      "epoch": 2.129364013671875e-05,
      "step": 5582
    },
    {
      "action_loss_avg": 0.022530581802129745,
      "epoch": 2.129364013671875e-05,
      "step": 5582
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 5582,
      "training_step_time": 9.92060375213623
    },
    {
      "epoch": 2.129364013671875e-05,
      "model_forward_time": 3.143317699432373,
      "step": 5582
    },
    {
      "epoch": 2.129364013671875e-05,
      "step": 5582,
      "training_step_time": 10.01440167427063
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 3.4591028690338135,
      "step": 5583
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 5583,
      "training_step_time": 10.232948064804077
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 3.137531042098999,
      "step": 5583
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 5583,
      "training_step_time": 9.908127307891846
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 3.1343729496002197,
      "step": 5583
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 5583,
      "training_step_time": 9.901020288467407
    },
    {
      "epoch": 2.1297454833984376e-05,
      "model_forward_time": 3.139828681945801,
      "step": 5583
    },
    {
      "epoch": 2.1297454833984376e-05,
      "step": 5583,
      "training_step_time": 10.279179096221924
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 3.1716420650482178,
      "step": 5584
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 5584,
      "training_step_time": 9.942073106765747
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 3.4067137241363525,
      "step": 5584
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 5584,
      "training_step_time": 10.17917275428772
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 3.136794090270996,
      "step": 5584
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 5584,
      "training_step_time": 9.908920288085938
    },
    {
      "epoch": 2.130126953125e-05,
      "model_forward_time": 3.133854866027832,
      "step": 5584
    },
    {
      "epoch": 2.130126953125e-05,
      "step": 5584,
      "training_step_time": 10.024165153503418
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 3.1720659732818604,
      "step": 5585
    },
    {
      "dynamics_loss_avg": 0.06988455206155778,
      "epoch": 2.1305084228515625e-05,
      "step": 5585
    },
    {
      "action_loss_avg": 0.025640706717967986,
      "epoch": 2.1305084228515625e-05,
      "step": 5585
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 5585,
      "training_step_time": 9.94882583618164
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 3.1386289596557617,
      "step": 5585
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 5585,
      "training_step_time": 9.911211490631104
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 3.4239866733551025,
      "step": 5585
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 5585,
      "training_step_time": 10.206860065460205
    },
    {
      "epoch": 2.1305084228515625e-05,
      "model_forward_time": 3.1321425437927246,
      "step": 5585
    },
    {
      "epoch": 2.1305084228515625e-05,
      "step": 5585,
      "training_step_time": 10.303695917129517
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 3.1677422523498535,
      "step": 5586
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 5586,
      "training_step_time": 9.94226884841919
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 3.135418176651001,
      "step": 5586
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 5586,
      "training_step_time": 9.908501625061035
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 3.1321704387664795,
      "step": 5586
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 5586,
      "training_step_time": 9.901206970214844
    },
    {
      "epoch": 2.130889892578125e-05,
      "model_forward_time": 3.390219211578369,
      "step": 5586
    },
    {
      "epoch": 2.130889892578125e-05,
      "step": 5586,
      "training_step_time": 10.324391841888428
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 3.143728494644165,
      "step": 5587
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 5587,
      "training_step_time": 9.91602873802185
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 3.1365952491760254,
      "step": 5587
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 5587,
      "training_step_time": 9.903968334197998
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 3.135881185531616,
      "step": 5587
    },
    {
      "dynamics_loss_avg": 0.0818142168223858,
      "epoch": 2.1312713623046874e-05,
      "step": 5587
    },
    {
      "action_loss_avg": 0.04068174734711647,
      "epoch": 2.1312713623046874e-05,
      "step": 5587
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 5587,
      "training_step_time": 9.901232719421387
    },
    {
      "epoch": 2.1312713623046874e-05,
      "model_forward_time": 3.1344122886657715,
      "step": 5587
    },
    {
      "epoch": 2.1312713623046874e-05,
      "step": 5587,
      "training_step_time": 10.02407193183899
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 3.4064478874206543,
      "step": 5588
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 5588,
      "training_step_time": 10.175826072692871
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 3.1352438926696777,
      "step": 5588
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 5588,
      "training_step_time": 9.907545328140259
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 3.1377205848693848,
      "step": 5588
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 5588,
      "training_step_time": 9.908026218414307
    },
    {
      "epoch": 2.13165283203125e-05,
      "model_forward_time": 3.1382627487182617,
      "step": 5588
    },
    {
      "epoch": 2.13165283203125e-05,
      "step": 5588,
      "training_step_time": 10.025142431259155
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 3.192647933959961,
      "step": 5589
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 5589,
      "training_step_time": 9.969128370285034
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 3.423311471939087,
      "step": 5589
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 5589,
      "training_step_time": 10.18460726737976
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 3.134880781173706,
      "step": 5589
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 5589,
      "training_step_time": 9.89891529083252
    },
    {
      "epoch": 2.1320343017578127e-05,
      "model_forward_time": 3.1388373374938965,
      "step": 5589
    },
    {
      "epoch": 2.1320343017578127e-05,
      "step": 5589,
      "training_step_time": 10.29180383682251
    },
    {
      "epoch": 2.132415771484375e-05,
      "grad_norm": 0.514741063117981,
      "learning_rate": 1e-05,
      "loss": 0.0896,
      "step": 5590
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 3.1736247539520264,
      "step": 5590
    },
    {
      "dynamics_loss_avg": 0.059679345972836015,
      "epoch": 2.132415771484375e-05,
      "step": 5590
    },
    {
      "action_loss_avg": 0.020844065025448798,
      "epoch": 2.132415771484375e-05,
      "step": 5590
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 5590,
      "training_step_time": 9.940609216690063
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 3.1382155418395996,
      "step": 5590
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 5590,
      "training_step_time": 9.907787084579468
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 3.1355795860290527,
      "step": 5590
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 5590,
      "training_step_time": 10.179264545440674
    },
    {
      "epoch": 2.132415771484375e-05,
      "model_forward_time": 3.134477376937866,
      "step": 5590
    },
    {
      "epoch": 2.132415771484375e-05,
      "step": 5590,
      "training_step_time": 10.015377283096313
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 3.18353533744812,
      "step": 5591
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 5591,
      "training_step_time": 9.95759892463684
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 3.132711172103882,
      "step": 5591
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 5591,
      "training_step_time": 9.901184558868408
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 3.1405749320983887,
      "step": 5591
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 5591,
      "training_step_time": 9.909100532531738
    },
    {
      "epoch": 2.1327972412109376e-05,
      "model_forward_time": 3.415884256362915,
      "step": 5591
    },
    {
      "epoch": 2.1327972412109376e-05,
      "step": 5591,
      "training_step_time": 10.324897050857544
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 3.17107892036438,
      "step": 5592
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 5592,
      "training_step_time": 9.945895433425903
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 3.134246826171875,
      "step": 5592
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 5592,
      "training_step_time": 9.908917427062988
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 3.141723155975342,
      "step": 5592
    },
    {
      "dynamics_loss_avg": 0.07281682286411524,
      "epoch": 2.1331787109375e-05,
      "step": 5592
    },
    {
      "action_loss_avg": 0.02521004984155297,
      "epoch": 2.1331787109375e-05,
      "step": 5592
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 5592,
      "training_step_time": 9.921079158782959
    },
    {
      "epoch": 2.1331787109375e-05,
      "model_forward_time": 3.1400794982910156,
      "step": 5592
    },
    {
      "epoch": 2.1331787109375e-05,
      "step": 5592,
      "training_step_time": 10.30623197555542
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 3.1750893592834473,
      "step": 5593
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 5593,
      "training_step_time": 10.220754861831665
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 3.1371004581451416,
      "step": 5593
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 5593,
      "training_step_time": 9.907766342163086
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 3.138072967529297,
      "step": 5593
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 5593,
      "training_step_time": 9.915435314178467
    },
    {
      "epoch": 2.1335601806640625e-05,
      "model_forward_time": 3.1341664791107178,
      "step": 5593
    },
    {
      "epoch": 2.1335601806640625e-05,
      "step": 5593,
      "training_step_time": 10.019233465194702
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 3.141731023788452,
      "step": 5594
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 5594,
      "training_step_time": 9.91733169555664
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 3.3991715908050537,
      "step": 5594
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 5594,
      "training_step_time": 10.170102834701538
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 3.137827157974243,
      "step": 5594
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 5594,
      "training_step_time": 9.910966634750366
    },
    {
      "epoch": 2.133941650390625e-05,
      "model_forward_time": 3.1330158710479736,
      "step": 5594
    },
    {
      "epoch": 2.133941650390625e-05,
      "step": 5594,
      "training_step_time": 10.268405199050903
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 3.1813879013061523,
      "step": 5595
    },
    {
      "dynamics_loss_avg": 0.07410423494875432,
      "epoch": 2.1343231201171874e-05,
      "step": 5595
    },
    {
      "action_loss_avg": 0.026984817814081907,
      "epoch": 2.1343231201171874e-05,
      "step": 5595
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 5595,
      "training_step_time": 9.954192638397217
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 3.138211965560913,
      "step": 5595
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 5595,
      "training_step_time": 9.907263994216919
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 3.4303576946258545,
      "step": 5595
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 5595,
      "training_step_time": 10.202488422393799
    },
    {
      "epoch": 2.1343231201171874e-05,
      "model_forward_time": 3.1309969425201416,
      "step": 5595
    },
    {
      "epoch": 2.1343231201171874e-05,
      "step": 5595,
      "training_step_time": 10.30172348022461
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 3.157910108566284,
      "step": 5596
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 5596,
      "training_step_time": 9.924280881881714
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 3.1330337524414062,
      "step": 5596
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 5596,
      "training_step_time": 9.900202751159668
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 3.1330718994140625,
      "step": 5596
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 5596,
      "training_step_time": 9.900046586990356
    },
    {
      "epoch": 2.13470458984375e-05,
      "model_forward_time": 3.4223952293395996,
      "step": 5596
    },
    {
      "epoch": 2.13470458984375e-05,
      "step": 5596,
      "training_step_time": 10.331110000610352
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 3.1465728282928467,
      "step": 5597
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 5597,
      "training_step_time": 9.926007747650146
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 2.4276413917541504,
      "step": 5597
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 5597,
      "training_step_time": 7.460109233856201
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 3.1287875175476074,
      "step": 5597
    },
    {
      "dynamics_loss_avg": 0.05913669299334288,
      "epoch": 2.1350860595703127e-05,
      "step": 5597
    },
    {
      "action_loss_avg": 0.02458668078761548,
      "epoch": 2.1350860595703127e-05,
      "step": 5597
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 5597,
      "training_step_time": 9.903384685516357
    },
    {
      "epoch": 2.1350860595703127e-05,
      "model_forward_time": 3.1343700885772705,
      "step": 5597
    },
    {
      "epoch": 2.1350860595703127e-05,
      "step": 5597,
      "training_step_time": 10.036086320877075
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 3.1708221435546875,
      "step": 5598
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 5598,
      "training_step_time": 10.228176832199097
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 3.135073184967041,
      "step": 5598
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 5598,
      "training_step_time": 9.906007528305054
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 3.130275249481201,
      "step": 5598
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 5598,
      "training_step_time": 9.902242660522461
    },
    {
      "epoch": 2.135467529296875e-05,
      "model_forward_time": 3.135159730911255,
      "step": 5598
    },
    {
      "epoch": 2.135467529296875e-05,
      "step": 5598,
      "training_step_time": 10.017382621765137
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 3.1524267196655273,
      "step": 5599
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 5599,
      "training_step_time": 9.930447578430176
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 3.1372604370117188,
      "step": 5599
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 5599,
      "training_step_time": 10.20821213722229
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 3.133409261703491,
      "step": 5599
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 5599,
      "training_step_time": 9.906317234039307
    },
    {
      "epoch": 2.1358489990234376e-05,
      "model_forward_time": 3.137179136276245,
      "step": 5599
    },
    {
      "epoch": 2.1358489990234376e-05,
      "step": 5599,
      "training_step_time": 10.264893770217896
    },
    {
      "epoch": 2.13623046875e-05,
      "grad_norm": 0.2853213846683502,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 5600
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 3.1779565811157227,
      "step": 5600
    },
    {
      "dynamics_loss_avg": 0.06982802413403988,
      "epoch": 2.13623046875e-05,
      "step": 5600
    },
    {
      "action_loss_avg": 0.028608197625726462,
      "epoch": 2.13623046875e-05,
      "step": 5600
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 5600,
      "training_step_time": 9.958391666412354
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 3.130821704864502,
      "step": 5600
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 5600,
      "training_step_time": 9.905438423156738
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 3.3952677249908447,
      "step": 5600
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 5600,
      "training_step_time": 10.172865390777588
    },
    {
      "epoch": 2.13623046875e-05,
      "model_forward_time": 3.136143922805786,
      "step": 5600
    },
    {
      "epoch": 2.13623046875e-05,
      "step": 5600,
      "training_step_time": 10.334490060806274
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 3.1459457874298096,
      "step": 5601
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 5601,
      "training_step_time": 9.927573442459106
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 3.1326255798339844,
      "step": 5601
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 5601,
      "training_step_time": 9.908310174942017
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 3.1336395740509033,
      "step": 5601
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 5601,
      "training_step_time": 9.909689903259277
    },
    {
      "epoch": 2.1366119384765625e-05,
      "model_forward_time": 3.1343462467193604,
      "step": 5601
    },
    {
      "epoch": 2.1366119384765625e-05,
      "step": 5601,
      "training_step_time": 10.28881049156189
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 3.178175210952759,
      "step": 5602
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 5602,
      "training_step_time": 9.96011471748352
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 3.132050037384033,
      "step": 5602
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 5602,
      "training_step_time": 9.907133340835571
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 3.1378097534179688,
      "step": 5602
    },
    {
      "dynamics_loss_avg": 0.06889806967228651,
      "epoch": 2.136993408203125e-05,
      "step": 5602
    },
    {
      "action_loss_avg": 0.024521125853061675,
      "epoch": 2.136993408203125e-05,
      "step": 5602
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 5602,
      "training_step_time": 9.911224842071533
    },
    {
      "epoch": 2.136993408203125e-05,
      "model_forward_time": 3.1351850032806396,
      "step": 5602
    },
    {
      "epoch": 2.136993408203125e-05,
      "step": 5602,
      "training_step_time": 10.287214517593384
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 3.448293447494507,
      "step": 5603
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 5603,
      "training_step_time": 10.228522539138794
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 3.136533260345459,
      "step": 5603
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 5603,
      "training_step_time": 9.90714406967163
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 3.13924503326416,
      "step": 5603
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 5603,
      "training_step_time": 9.912048101425171
    },
    {
      "epoch": 2.1373748779296874e-05,
      "model_forward_time": 3.135146379470825,
      "step": 5603
    },
    {
      "epoch": 2.1373748779296874e-05,
      "step": 5603,
      "training_step_time": 10.031018733978271
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 3.1694018840789795,
      "step": 5604
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 5604,
      "training_step_time": 9.944669246673584
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 3.1421854496002197,
      "step": 5604
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 5604,
      "training_step_time": 10.237213850021362
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 3.1460602283477783,
      "step": 5604
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 5604,
      "training_step_time": 9.93081283569336
    },
    {
      "epoch": 2.13775634765625e-05,
      "model_forward_time": 3.138634204864502,
      "step": 5604
    },
    {
      "epoch": 2.13775634765625e-05,
      "step": 5604,
      "training_step_time": 10.329253673553467
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 3.167320489883423,
      "step": 5605
    },
    {
      "dynamics_loss_avg": 0.05975562520325184,
      "epoch": 2.1381378173828127e-05,
      "step": 5605
    },
    {
      "action_loss_avg": 0.02518020635470748,
      "epoch": 2.1381378173828127e-05,
      "step": 5605
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 5605,
      "training_step_time": 9.957119464874268
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 3.1386423110961914,
      "step": 5605
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 5605,
      "training_step_time": 9.916483879089355
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 3.447397232055664,
      "step": 5605
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 5605,
      "training_step_time": 10.221264839172363
    },
    {
      "epoch": 2.1381378173828127e-05,
      "model_forward_time": 3.133486747741699,
      "step": 5605
    },
    {
      "epoch": 2.1381378173828127e-05,
      "step": 5605,
      "training_step_time": 10.264763355255127
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 3.1691792011260986,
      "step": 5606
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 5606,
      "training_step_time": 9.950294017791748
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 3.1334798336029053,
      "step": 5606
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 5606,
      "training_step_time": 9.904356956481934
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 3.133043050765991,
      "step": 5606
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 5606,
      "training_step_time": 9.90623426437378
    },
    {
      "epoch": 2.138519287109375e-05,
      "model_forward_time": 3.4081549644470215,
      "step": 5606
    },
    {
      "epoch": 2.138519287109375e-05,
      "step": 5606,
      "training_step_time": 10.299560785293579
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 3.1465656757354736,
      "step": 5607
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 5607,
      "training_step_time": 9.921586036682129
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 3.1338608264923096,
      "step": 5607
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 5607,
      "training_step_time": 9.903133153915405
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 3.1290783882141113,
      "step": 5607
    },
    {
      "dynamics_loss_avg": 0.08819490596652031,
      "epoch": 2.1389007568359376e-05,
      "step": 5607
    },
    {
      "action_loss_avg": 0.035401396453380585,
      "epoch": 2.1389007568359376e-05,
      "step": 5607
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 5607,
      "training_step_time": 9.902980089187622
    },
    {
      "epoch": 2.1389007568359376e-05,
      "model_forward_time": 3.1325602531433105,
      "step": 5607
    },
    {
      "epoch": 2.1389007568359376e-05,
      "step": 5607,
      "training_step_time": 10.016521215438843
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 3.416630983352661,
      "step": 5608
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 5608,
      "training_step_time": 10.192823886871338
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 3.13932204246521,
      "step": 5608
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 5608,
      "training_step_time": 9.913830995559692
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 3.1368136405944824,
      "step": 5608
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 5608,
      "training_step_time": 9.918868064880371
    },
    {
      "epoch": 2.1392822265625e-05,
      "model_forward_time": 3.1402628421783447,
      "step": 5608
    },
    {
      "epoch": 2.1392822265625e-05,
      "step": 5608,
      "training_step_time": 10.297303676605225
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 3.147709369659424,
      "step": 5609
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 5609,
      "training_step_time": 9.92714548110962
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 3.452138900756836,
      "step": 5609
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 5609,
      "training_step_time": 10.231735944747925
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 3.139092206954956,
      "step": 5609
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 5609,
      "training_step_time": 9.924421548843384
    },
    {
      "epoch": 2.1396636962890625e-05,
      "model_forward_time": 3.1432905197143555,
      "step": 5609
    },
    {
      "epoch": 2.1396636962890625e-05,
      "step": 5609,
      "training_step_time": 10.028183460235596
    },
    {
      "epoch": 2.140045166015625e-05,
      "grad_norm": 0.5777698755264282,
      "learning_rate": 1e-05,
      "loss": 0.0908,
      "step": 5610
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 2.4427030086517334,
      "step": 5610
    },
    {
      "dynamics_loss_avg": 0.070718327537179,
      "epoch": 2.140045166015625e-05,
      "step": 5610
    },
    {
      "action_loss_avg": 0.02240977166220546,
      "epoch": 2.140045166015625e-05,
      "step": 5610
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 5610,
      "training_step_time": 7.477295637130737
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 3.1336169242858887,
      "step": 5610
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 5610,
      "training_step_time": 9.911049365997314
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 3.1388168334960938,
      "step": 5610
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 5610,
      "training_step_time": 10.19973611831665
    },
    {
      "epoch": 2.140045166015625e-05,
      "model_forward_time": 3.1388351917266846,
      "step": 5610
    },
    {
      "epoch": 2.140045166015625e-05,
      "step": 5610,
      "training_step_time": 10.032778978347778
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 3.175053119659424,
      "step": 5611
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 5611,
      "training_step_time": 9.955988645553589
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 3.136133909225464,
      "step": 5611
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 5611,
      "training_step_time": 9.917013168334961
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 3.1378378868103027,
      "step": 5611
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 5611,
      "training_step_time": 9.916950941085815
    },
    {
      "epoch": 2.1404266357421874e-05,
      "model_forward_time": 3.4174251556396484,
      "step": 5611
    },
    {
      "epoch": 2.1404266357421874e-05,
      "step": 5611,
      "training_step_time": 10.289743661880493
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 3.148998260498047,
      "step": 5612
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 5612,
      "training_step_time": 9.928369760513306
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 3.1348876953125,
      "step": 5612
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 5612,
      "training_step_time": 9.913159847259521
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 3.134986162185669,
      "step": 5612
    },
    {
      "dynamics_loss_avg": 0.06814444940537215,
      "epoch": 2.14080810546875e-05,
      "step": 5612
    },
    {
      "action_loss_avg": 0.030046179331839083,
      "epoch": 2.14080810546875e-05,
      "step": 5612
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 5612,
      "training_step_time": 9.912368059158325
    },
    {
      "epoch": 2.14080810546875e-05,
      "model_forward_time": 3.132136583328247,
      "step": 5612
    },
    {
      "epoch": 2.14080810546875e-05,
      "step": 5612,
      "training_step_time": 10.322946071624756
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 3.146151065826416,
      "step": 5613
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 5613,
      "training_step_time": 10.193673133850098
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 3.136617422103882,
      "step": 5613
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 5613,
      "training_step_time": 9.910732507705688
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 3.1412997245788574,
      "step": 5613
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 5613,
      "training_step_time": 9.919634342193604
    },
    {
      "epoch": 2.1411895751953127e-05,
      "model_forward_time": 3.1346991062164307,
      "step": 5613
    },
    {
      "epoch": 2.1411895751953127e-05,
      "step": 5613,
      "training_step_time": 10.29482364654541
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 3.1625988483428955,
      "step": 5614
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 5614,
      "training_step_time": 9.937301874160767
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 3.422024726867676,
      "step": 5614
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 5614,
      "training_step_time": 10.195503234863281
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 3.139519691467285,
      "step": 5614
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 5614,
      "training_step_time": 9.91475534439087
    },
    {
      "epoch": 2.141571044921875e-05,
      "model_forward_time": 3.1364524364471436,
      "step": 5614
    },
    {
      "epoch": 2.141571044921875e-05,
      "step": 5614,
      "training_step_time": 10.254782438278198
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 3.185234546661377,
      "step": 5615
    },
    {
      "dynamics_loss_avg": 0.07297462373971939,
      "epoch": 2.1419525146484376e-05,
      "step": 5615
    },
    {
      "action_loss_avg": 0.02482274267822504,
      "epoch": 2.1419525146484376e-05,
      "step": 5615
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 5615,
      "training_step_time": 9.972503662109375
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 3.144289255142212,
      "step": 5615
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 5615,
      "training_step_time": 9.926909446716309
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 3.4286160469055176,
      "step": 5615
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 5615,
      "training_step_time": 10.210469484329224
    },
    {
      "epoch": 2.1419525146484376e-05,
      "model_forward_time": 3.1393983364105225,
      "step": 5615
    },
    {
      "epoch": 2.1419525146484376e-05,
      "step": 5615,
      "training_step_time": 10.30142879486084
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 3.1879899501800537,
      "step": 5616
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 5616,
      "training_step_time": 9.97035264968872
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 3.149627685546875,
      "step": 5616
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 5616,
      "training_step_time": 9.93564510345459
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 3.1517677307128906,
      "step": 5616
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 5616,
      "training_step_time": 9.94111180305481
    },
    {
      "epoch": 2.142333984375e-05,
      "model_forward_time": 3.4108660221099854,
      "step": 5616
    },
    {
      "epoch": 2.142333984375e-05,
      "step": 5616,
      "training_step_time": 10.289838790893555
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 3.180950880050659,
      "step": 5617
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 5617,
      "training_step_time": 9.966340780258179
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 3.1401336193084717,
      "step": 5617
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 5617,
      "training_step_time": 9.91356086730957
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 3.1345856189727783,
      "step": 5617
    },
    {
      "dynamics_loss_avg": 0.07503595445305108,
      "epoch": 2.1427154541015625e-05,
      "step": 5617
    },
    {
      "action_loss_avg": 0.025974827725440262,
      "epoch": 2.1427154541015625e-05,
      "step": 5617
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 5617,
      "training_step_time": 9.909572124481201
    },
    {
      "epoch": 2.1427154541015625e-05,
      "model_forward_time": 3.1380465030670166,
      "step": 5617
    },
    {
      "epoch": 2.1427154541015625e-05,
      "step": 5617,
      "training_step_time": 10.257609605789185
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 3.452967882156372,
      "step": 5618
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 5618,
      "training_step_time": 10.232582330703735
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 3.13692307472229,
      "step": 5618
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 5618,
      "training_step_time": 9.908984422683716
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 3.132901430130005,
      "step": 5618
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 5618,
      "training_step_time": 9.903995037078857
    },
    {
      "epoch": 2.143096923828125e-05,
      "model_forward_time": 3.1373989582061768,
      "step": 5618
    },
    {
      "epoch": 2.143096923828125e-05,
      "step": 5618,
      "training_step_time": 10.020274639129639
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 3.182432174682617,
      "step": 5619
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 5619,
      "training_step_time": 9.961669921875
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 3.1359705924987793,
      "step": 5619
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 5619,
      "training_step_time": 10.203984022140503
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 3.136451005935669,
      "step": 5619
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 5619,
      "training_step_time": 9.908000946044922
    },
    {
      "epoch": 2.1434783935546874e-05,
      "model_forward_time": 3.136831283569336,
      "step": 5619
    },
    {
      "epoch": 2.1434783935546874e-05,
      "step": 5619,
      "training_step_time": 10.01146125793457
    },
    {
      "epoch": 2.14385986328125e-05,
      "grad_norm": 0.6240181922912598,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 5620
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 3.181619882583618,
      "step": 5620
    },
    {
      "dynamics_loss_avg": 0.0720671635121107,
      "epoch": 2.14385986328125e-05,
      "step": 5620
    },
    {
      "action_loss_avg": 0.021695437002927066,
      "epoch": 2.14385986328125e-05,
      "step": 5620
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 5620,
      "training_step_time": 9.957990169525146
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 3.139221429824829,
      "step": 5620
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 5620,
      "training_step_time": 9.910289764404297
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 3.4269304275512695,
      "step": 5620
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 5620,
      "training_step_time": 10.201653242111206
    },
    {
      "epoch": 2.14385986328125e-05,
      "model_forward_time": 3.14801287651062,
      "step": 5620
    },
    {
      "epoch": 2.14385986328125e-05,
      "step": 5620,
      "training_step_time": 10.28346061706543
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 3.1606364250183105,
      "step": 5621
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 5621,
      "training_step_time": 9.93208622932434
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 3.1327455043792725,
      "step": 5621
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 5621,
      "training_step_time": 9.897153377532959
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 3.1389994621276855,
      "step": 5621
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 5621,
      "training_step_time": 9.901100635528564
    },
    {
      "epoch": 2.1442413330078127e-05,
      "model_forward_time": 3.1295230388641357,
      "step": 5621
    },
    {
      "epoch": 2.1442413330078127e-05,
      "step": 5621,
      "training_step_time": 10.280341863632202
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 3.1394174098968506,
      "step": 5622
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 5622,
      "training_step_time": 9.910343885421753
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 3.1324753761291504,
      "step": 5622
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 5622,
      "training_step_time": 9.897109746932983
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 3.1386516094207764,
      "step": 5622
    },
    {
      "dynamics_loss_avg": 0.06923066182062029,
      "epoch": 2.144622802734375e-05,
      "step": 5622
    },
    {
      "action_loss_avg": 0.021403127815574408,
      "epoch": 2.144622802734375e-05,
      "step": 5622
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 5622,
      "training_step_time": 9.912424564361572
    },
    {
      "epoch": 2.144622802734375e-05,
      "model_forward_time": 3.1365010738372803,
      "step": 5622
    },
    {
      "epoch": 2.144622802734375e-05,
      "step": 5622,
      "training_step_time": 10.271754503250122
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 3.4386777877807617,
      "step": 5623
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 5623,
      "training_step_time": 10.20862364768982
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 3.141382932662964,
      "step": 5623
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 5623,
      "training_step_time": 9.908199787139893
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 3.151442289352417,
      "step": 5623
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 5623,
      "training_step_time": 9.920865058898926
    },
    {
      "epoch": 2.1450042724609376e-05,
      "model_forward_time": 3.1445510387420654,
      "step": 5623
    },
    {
      "epoch": 2.1450042724609376e-05,
      "step": 5623,
      "training_step_time": 10.015808582305908
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 3.156186580657959,
      "step": 5624
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 5624,
      "training_step_time": 9.927881479263306
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 3.1446728706359863,
      "step": 5624
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 5624,
      "training_step_time": 10.21257996559143
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 3.144195795059204,
      "step": 5624
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 5624,
      "training_step_time": 9.910325765609741
    },
    {
      "epoch": 2.1453857421875e-05,
      "model_forward_time": 3.1446456909179688,
      "step": 5624
    },
    {
      "epoch": 2.1453857421875e-05,
      "step": 5624,
      "training_step_time": 10.008864402770996
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 3.1802289485931396,
      "step": 5625
    },
    {
      "dynamics_loss_avg": 0.0581750575453043,
      "epoch": 2.1457672119140625e-05,
      "step": 5625
    },
    {
      "action_loss_avg": 0.023483035946264864,
      "epoch": 2.1457672119140625e-05,
      "step": 5625
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 5625,
      "training_step_time": 9.949658393859863
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 3.140359878540039,
      "step": 5625
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 5625,
      "training_step_time": 9.907366037368774
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 3.4293737411499023,
      "step": 5625
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 5625,
      "training_step_time": 10.193368434906006
    },
    {
      "epoch": 2.1457672119140625e-05,
      "model_forward_time": 3.140066385269165,
      "step": 5625
    },
    {
      "epoch": 2.1457672119140625e-05,
      "step": 5625,
      "training_step_time": 10.023269891738892
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 3.1474926471710205,
      "step": 5626
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 5626,
      "training_step_time": 9.921199321746826
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 2.4400668144226074,
      "step": 5626
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 5626,
      "training_step_time": 7.475854873657227
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 3.138765811920166,
      "step": 5626
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 5626,
      "training_step_time": 9.91722822189331
    },
    {
      "epoch": 2.146148681640625e-05,
      "model_forward_time": 3.142359495162964,
      "step": 5626
    },
    {
      "epoch": 2.146148681640625e-05,
      "step": 5626,
      "training_step_time": 10.314891338348389
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 3.1814463138580322,
      "step": 5627
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 5627,
      "training_step_time": 9.965691089630127
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 3.139760971069336,
      "step": 5627
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 5627,
      "training_step_time": 9.91849398612976
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 3.1432132720947266,
      "step": 5627
    },
    {
      "dynamics_loss_avg": 0.08012126926332712,
      "epoch": 2.1465301513671874e-05,
      "step": 5627
    },
    {
      "action_loss_avg": 0.030588961392641067,
      "epoch": 2.1465301513671874e-05,
      "step": 5627
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 5627,
      "training_step_time": 9.928342580795288
    },
    {
      "epoch": 2.1465301513671874e-05,
      "model_forward_time": 3.146376609802246,
      "step": 5627
    },
    {
      "epoch": 2.1465301513671874e-05,
      "step": 5627,
      "training_step_time": 10.255509853363037
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 3.4231669902801514,
      "step": 5628
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 5628,
      "training_step_time": 10.210017919540405
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 3.1422204971313477,
      "step": 5628
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 5628,
      "training_step_time": 9.919212579727173
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 3.135230541229248,
      "step": 5628
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 5628,
      "training_step_time": 9.907289266586304
    },
    {
      "epoch": 2.14691162109375e-05,
      "model_forward_time": 3.1391921043395996,
      "step": 5628
    },
    {
      "epoch": 2.14691162109375e-05,
      "step": 5628,
      "training_step_time": 10.27604603767395
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 3.182617425918579,
      "step": 5629
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 5629,
      "training_step_time": 9.959388256072998
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 3.4212584495544434,
      "step": 5629
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 5629,
      "training_step_time": 10.19536805152893
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 3.137080669403076,
      "step": 5629
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 5629,
      "training_step_time": 9.911744117736816
    },
    {
      "epoch": 2.1472930908203123e-05,
      "model_forward_time": 3.1428887844085693,
      "step": 5629
    },
    {
      "epoch": 2.1472930908203123e-05,
      "step": 5629,
      "training_step_time": 10.312023401260376
    },
    {
      "epoch": 2.147674560546875e-05,
      "grad_norm": 0.27510544657707214,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 5630
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 3.1712427139282227,
      "step": 5630
    },
    {
      "dynamics_loss_avg": 0.0702776562422514,
      "epoch": 2.147674560546875e-05,
      "step": 5630
    },
    {
      "action_loss_avg": 0.024453655630350114,
      "epoch": 2.147674560546875e-05,
      "step": 5630
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 5630,
      "training_step_time": 9.957741022109985
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 2.4354448318481445,
      "step": 5630
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 5630,
      "training_step_time": 7.4695658683776855
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 3.142568826675415,
      "step": 5630
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 5630,
      "training_step_time": 10.229642391204834
    },
    {
      "epoch": 2.147674560546875e-05,
      "model_forward_time": 3.145573139190674,
      "step": 5630
    },
    {
      "epoch": 2.147674560546875e-05,
      "step": 5630,
      "training_step_time": 10.040226936340332
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 3.187052011489868,
      "step": 5631
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 5631,
      "training_step_time": 9.968316555023193
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 3.1423122882843018,
      "step": 5631
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 5631,
      "training_step_time": 9.921141147613525
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 3.146897077560425,
      "step": 5631
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 5631,
      "training_step_time": 9.925201892852783
    },
    {
      "epoch": 2.1480560302734376e-05,
      "model_forward_time": 3.443061590194702,
      "step": 5631
    },
    {
      "epoch": 2.1480560302734376e-05,
      "step": 5631,
      "training_step_time": 10.336650133132935
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 3.186145782470703,
      "step": 5632
    },
    {
      "epoch": 2.1484375e-05,
      "step": 5632,
      "training_step_time": 9.968021631240845
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 3.145875930786133,
      "step": 5632
    },
    {
      "epoch": 2.1484375e-05,
      "step": 5632,
      "training_step_time": 9.925703763961792
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 3.1528072357177734,
      "step": 5632
    },
    {
      "dynamics_loss_avg": 0.06789310239255428,
      "epoch": 2.1484375e-05,
      "step": 5632
    },
    {
      "action_loss_avg": 0.028985810279846192,
      "epoch": 2.1484375e-05,
      "step": 5632
    },
    {
      "epoch": 2.1484375e-05,
      "step": 5632,
      "training_step_time": 9.93684697151184
    },
    {
      "epoch": 2.1484375e-05,
      "model_forward_time": 3.1481144428253174,
      "step": 5632
    },
    {
      "epoch": 2.1484375e-05,
      "step": 5632,
      "training_step_time": 10.290624618530273
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 3.1934573650360107,
      "step": 5633
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 5633,
      "training_step_time": 10.24255895614624
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 3.1653647422790527,
      "step": 5633
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 5633,
      "training_step_time": 9.947340726852417
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 3.1376686096191406,
      "step": 5633
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 5633,
      "training_step_time": 9.912005424499512
    },
    {
      "epoch": 2.1488189697265625e-05,
      "model_forward_time": 3.134542942047119,
      "step": 5633
    },
    {
      "epoch": 2.1488189697265625e-05,
      "step": 5633,
      "training_step_time": 10.27825927734375
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 3.147660255432129,
      "step": 5634
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 5634,
      "training_step_time": 9.923065423965454
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 3.4485039710998535,
      "step": 5634
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 5634,
      "training_step_time": 10.222198486328125
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 3.1416826248168945,
      "step": 5634
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 5634,
      "training_step_time": 9.92402458190918
    },
    {
      "epoch": 2.149200439453125e-05,
      "model_forward_time": 2.437039375305176,
      "step": 5634
    },
    {
      "epoch": 2.149200439453125e-05,
      "step": 5634,
      "training_step_time": 10.02755880355835
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 3.14691162109375,
      "step": 5635
    },
    {
      "dynamics_loss_avg": 0.06390120275318623,
      "epoch": 2.1495819091796874e-05,
      "step": 5635
    },
    {
      "action_loss_avg": 0.019347768928855657,
      "epoch": 2.1495819091796874e-05,
      "step": 5635
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 5635,
      "training_step_time": 9.927755355834961
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 3.1421799659729004,
      "step": 5635
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 5635,
      "training_step_time": 9.92286729812622
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 3.138866424560547,
      "step": 5635
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 5635,
      "training_step_time": 10.187582969665527
    },
    {
      "epoch": 2.1495819091796874e-05,
      "model_forward_time": 3.139341115951538,
      "step": 5635
    },
    {
      "epoch": 2.1495819091796874e-05,
      "step": 5635,
      "training_step_time": 10.280689716339111
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 3.1746957302093506,
      "step": 5636
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 5636,
      "training_step_time": 9.956305742263794
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 2.4381024837493896,
      "step": 5636
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 5636,
      "training_step_time": 7.468188524246216
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 3.1405022144317627,
      "step": 5636
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 5636,
      "training_step_time": 9.916314125061035
    },
    {
      "epoch": 2.14996337890625e-05,
      "model_forward_time": 3.4031765460968018,
      "step": 5636
    },
    {
      "epoch": 2.14996337890625e-05,
      "step": 5636,
      "training_step_time": 10.267378091812134
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 3.1742489337921143,
      "step": 5637
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 5637,
      "training_step_time": 9.944905519485474
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 3.1441891193389893,
      "step": 5637
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 5637,
      "training_step_time": 9.912437438964844
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 3.1542892456054688,
      "step": 5637
    },
    {
      "dynamics_loss_avg": 0.0524319339543581,
      "epoch": 2.1503448486328123e-05,
      "step": 5637
    },
    {
      "action_loss_avg": 0.018593764491379262,
      "epoch": 2.1503448486328123e-05,
      "step": 5637
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 5637,
      "training_step_time": 9.929472208023071
    },
    {
      "epoch": 2.1503448486328123e-05,
      "model_forward_time": 3.141625165939331,
      "step": 5637
    },
    {
      "epoch": 2.1503448486328123e-05,
      "step": 5637,
      "training_step_time": 10.28807258605957
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 3.1484508514404297,
      "step": 5638
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 5638,
      "training_step_time": 10.221756219863892
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 3.1401138305664062,
      "step": 5638
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 5638,
      "training_step_time": 9.912967443466187
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 3.139692544937134,
      "step": 5638
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 5638,
      "training_step_time": 9.911202430725098
    },
    {
      "epoch": 2.150726318359375e-05,
      "model_forward_time": 3.143000364303589,
      "step": 5638
    },
    {
      "epoch": 2.150726318359375e-05,
      "step": 5638,
      "training_step_time": 10.271146059036255
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 3.1844756603240967,
      "step": 5639
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 5639,
      "training_step_time": 9.959761381149292
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 3.43304181098938,
      "step": 5639
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 5639,
      "training_step_time": 10.209329843521118
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 3.1432366371154785,
      "step": 5639
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 5639,
      "training_step_time": 9.920626401901245
    },
    {
      "epoch": 2.1511077880859376e-05,
      "model_forward_time": 3.1462161540985107,
      "step": 5639
    },
    {
      "epoch": 2.1511077880859376e-05,
      "step": 5639,
      "training_step_time": 10.321181297302246
    },
    {
      "epoch": 2.1514892578125e-05,
      "grad_norm": 0.49038442969322205,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 5640
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 3.1464383602142334,
      "step": 5640
    },
    {
      "dynamics_loss_avg": 0.0689714384265244,
      "epoch": 2.1514892578125e-05,
      "step": 5640
    },
    {
      "action_loss_avg": 0.02713520664256066,
      "epoch": 2.1514892578125e-05,
      "step": 5640
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 5640,
      "training_step_time": 9.930127620697021
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 3.1511361598968506,
      "step": 5640
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 5640,
      "training_step_time": 9.930531978607178
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 3.406134843826294,
      "step": 5640
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 5640,
      "training_step_time": 10.186428546905518
    },
    {
      "epoch": 2.1514892578125e-05,
      "model_forward_time": 3.150470495223999,
      "step": 5640
    },
    {
      "epoch": 2.1514892578125e-05,
      "step": 5640,
      "training_step_time": 10.025911092758179
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 3.1846835613250732,
      "step": 5641
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 5641,
      "training_step_time": 9.9662184715271
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 3.1428349018096924,
      "step": 5641
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 5641,
      "training_step_time": 9.929604768753052
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 3.146690607070923,
      "step": 5641
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 5641,
      "training_step_time": 9.923932790756226
    },
    {
      "epoch": 2.1518707275390625e-05,
      "model_forward_time": 3.4296841621398926,
      "step": 5641
    },
    {
      "epoch": 2.1518707275390625e-05,
      "step": 5641,
      "training_step_time": 10.302410125732422
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 3.158357858657837,
      "step": 5642
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 5642,
      "training_step_time": 9.937256813049316
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 3.141866445541382,
      "step": 5642
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 5642,
      "training_step_time": 9.91589093208313
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 3.1416211128234863,
      "step": 5642
    },
    {
      "dynamics_loss_avg": 0.05994526706635952,
      "epoch": 2.152252197265625e-05,
      "step": 5642
    },
    {
      "action_loss_avg": 0.018739802297204733,
      "epoch": 2.152252197265625e-05,
      "step": 5642
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 5642,
      "training_step_time": 9.913969278335571
    },
    {
      "epoch": 2.152252197265625e-05,
      "model_forward_time": 3.1370046138763428,
      "step": 5642
    },
    {
      "epoch": 2.152252197265625e-05,
      "step": 5642,
      "training_step_time": 10.329657316207886
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 3.470674753189087,
      "step": 5643
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 5643,
      "training_step_time": 10.245445728302002
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 3.141688346862793,
      "step": 5643
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 5643,
      "training_step_time": 9.918370962142944
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 3.1484909057617188,
      "step": 5643
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 5643,
      "training_step_time": 9.924900531768799
    },
    {
      "epoch": 2.1526336669921874e-05,
      "model_forward_time": 3.1402475833892822,
      "step": 5643
    },
    {
      "epoch": 2.1526336669921874e-05,
      "step": 5643,
      "training_step_time": 10.290319442749023
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 3.172985553741455,
      "step": 5644
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 5644,
      "training_step_time": 9.950489282608032
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 3.3975136280059814,
      "step": 5644
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 5644,
      "training_step_time": 10.176419258117676
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 3.1463048458099365,
      "step": 5644
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 5644,
      "training_step_time": 9.924912691116333
    },
    {
      "epoch": 2.15301513671875e-05,
      "model_forward_time": 3.1434457302093506,
      "step": 5644
    },
    {
      "epoch": 2.15301513671875e-05,
      "step": 5644,
      "training_step_time": 10.283928632736206
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 3.1529057025909424,
      "step": 5645
    },
    {
      "dynamics_loss_avg": 0.055973477102816104,
      "epoch": 2.1533966064453123e-05,
      "step": 5645
    },
    {
      "action_loss_avg": 0.025542121846228838,
      "epoch": 2.1533966064453123e-05,
      "step": 5645
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 5645,
      "training_step_time": 9.950968980789185
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 3.158623695373535,
      "step": 5645
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 5645,
      "training_step_time": 9.949899196624756
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 3.5704734325408936,
      "step": 5645
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 5645,
      "training_step_time": 10.365324258804321
    },
    {
      "epoch": 2.1533966064453123e-05,
      "model_forward_time": 3.151346206665039,
      "step": 5645
    },
    {
      "epoch": 2.1533966064453123e-05,
      "step": 5645,
      "training_step_time": 10.053419589996338
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 3.1618199348449707,
      "step": 5646
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 5646,
      "training_step_time": 9.937179565429688
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 3.1366286277770996,
      "step": 5646
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 5646,
      "training_step_time": 9.909245491027832
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 3.1380932331085205,
      "step": 5646
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 5646,
      "training_step_time": 9.914422035217285
    },
    {
      "epoch": 2.153778076171875e-05,
      "model_forward_time": 3.4376986026763916,
      "step": 5646
    },
    {
      "epoch": 2.153778076171875e-05,
      "step": 5646,
      "training_step_time": 10.300409317016602
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 3.1415674686431885,
      "step": 5647
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 5647,
      "training_step_time": 9.915143489837646
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 3.134941577911377,
      "step": 5647
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 5647,
      "training_step_time": 9.90581202507019
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 1.7513744831085205,
      "step": 5647
    },
    {
      "dynamics_loss_avg": 0.05164263155311346,
      "epoch": 2.1541595458984376e-05,
      "step": 5647
    },
    {
      "action_loss_avg": 0.01983916275203228,
      "epoch": 2.1541595458984376e-05,
      "step": 5647
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 5647,
      "training_step_time": 5.214481353759766
    },
    {
      "epoch": 2.1541595458984376e-05,
      "model_forward_time": 3.132024049758911,
      "step": 5647
    },
    {
      "epoch": 2.1541595458984376e-05,
      "step": 5647,
      "training_step_time": 10.301834344863892
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 3.178605079650879,
      "step": 5648
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 5648,
      "training_step_time": 9.956878662109375
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 3.4269015789031982,
      "step": 5648
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 5648,
      "training_step_time": 10.202611207962036
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 3.1305360794067383,
      "step": 5648
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 5648,
      "training_step_time": 9.901849746704102
    },
    {
      "epoch": 2.154541015625e-05,
      "model_forward_time": 3.138914108276367,
      "step": 5648
    },
    {
      "epoch": 2.154541015625e-05,
      "step": 5648,
      "training_step_time": 10.30319881439209
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 3.1741724014282227,
      "step": 5649
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 5649,
      "training_step_time": 9.94714879989624
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 3.4599854946136475,
      "step": 5649
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 5649,
      "training_step_time": 10.226532936096191
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 3.133918046951294,
      "step": 5649
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 5649,
      "training_step_time": 9.904316663742065
    },
    {
      "epoch": 2.1549224853515625e-05,
      "model_forward_time": 3.138214111328125,
      "step": 5649
    },
    {
      "epoch": 2.1549224853515625e-05,
      "step": 5649,
      "training_step_time": 10.256351470947266
    },
    {
      "epoch": 2.155303955078125e-05,
      "grad_norm": 0.4059607684612274,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 5650
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 3.1748063564300537,
      "step": 5650
    },
    {
      "dynamics_loss_avg": 0.06270387805998326,
      "epoch": 2.155303955078125e-05,
      "step": 5650
    },
    {
      "action_loss_avg": 0.029062716569751502,
      "epoch": 2.155303955078125e-05,
      "step": 5650
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 5650,
      "training_step_time": 9.956791400909424
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 3.1379008293151855,
      "step": 5650
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 5650,
      "training_step_time": 9.913677453994751
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 3.1402885913848877,
      "step": 5650
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 5650,
      "training_step_time": 10.187960624694824
    },
    {
      "epoch": 2.155303955078125e-05,
      "model_forward_time": 3.1403989791870117,
      "step": 5650
    },
    {
      "epoch": 2.155303955078125e-05,
      "step": 5650,
      "training_step_time": 10.031904697418213
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 3.1750833988189697,
      "step": 5651
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 5651,
      "training_step_time": 9.957354545593262
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 3.1331136226654053,
      "step": 5651
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 5651,
      "training_step_time": 9.908226013183594
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 3.150211811065674,
      "step": 5651
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 5651,
      "training_step_time": 9.92843508720398
    },
    {
      "epoch": 2.1556854248046874e-05,
      "model_forward_time": 3.4140923023223877,
      "step": 5651
    },
    {
      "epoch": 2.1556854248046874e-05,
      "step": 5651,
      "training_step_time": 10.300116777420044
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 3.1680636405944824,
      "step": 5652
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 5652,
      "training_step_time": 9.94497013092041
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 3.1318771839141846,
      "step": 5652
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 5652,
      "training_step_time": 9.910736799240112
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 3.1387171745300293,
      "step": 5652
    },
    {
      "dynamics_loss_avg": 0.0456645155325532,
      "epoch": 2.15606689453125e-05,
      "step": 5652
    },
    {
      "action_loss_avg": 0.014936779625713825,
      "epoch": 2.15606689453125e-05,
      "step": 5652
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 5652,
      "training_step_time": 9.923281192779541
    },
    {
      "epoch": 2.15606689453125e-05,
      "model_forward_time": 3.138895273208618,
      "step": 5652
    },
    {
      "epoch": 2.15606689453125e-05,
      "step": 5652,
      "training_step_time": 10.30876350402832
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 3.444805860519409,
      "step": 5653
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 5653,
      "training_step_time": 10.223852634429932
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 3.132251739501953,
      "step": 5653
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 5653,
      "training_step_time": 9.905537128448486
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 3.134762763977051,
      "step": 5653
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 5653,
      "training_step_time": 9.909661769866943
    },
    {
      "epoch": 2.1564483642578123e-05,
      "model_forward_time": 3.1295037269592285,
      "step": 5653
    },
    {
      "epoch": 2.1564483642578123e-05,
      "step": 5653,
      "training_step_time": 10.314907550811768
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 2.4715375900268555,
      "step": 5654
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 5654,
      "training_step_time": 7.497424364089966
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 3.431067705154419,
      "step": 5654
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 5654,
      "training_step_time": 10.205198287963867
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 3.138584852218628,
      "step": 5654
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 5654,
      "training_step_time": 9.90928864479065
    },
    {
      "epoch": 2.156829833984375e-05,
      "model_forward_time": 3.132678747177124,
      "step": 5654
    },
    {
      "epoch": 2.156829833984375e-05,
      "step": 5654,
      "training_step_time": 10.317608118057251
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 3.135377883911133,
      "step": 5655
    },
    {
      "dynamics_loss_avg": 0.07153729177080095,
      "epoch": 2.1572113037109376e-05,
      "step": 5655
    },
    {
      "action_loss_avg": 0.02475942187011242,
      "epoch": 2.1572113037109376e-05,
      "step": 5655
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 5655,
      "training_step_time": 9.90970253944397
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 3.132713794708252,
      "step": 5655
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 5655,
      "training_step_time": 9.902706384658813
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 3.1337368488311768,
      "step": 5655
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 5655,
      "training_step_time": 10.179739236831665
    },
    {
      "epoch": 2.1572113037109376e-05,
      "model_forward_time": 3.138702630996704,
      "step": 5655
    },
    {
      "epoch": 2.1572113037109376e-05,
      "step": 5655,
      "training_step_time": 10.009023189544678
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 3.1399049758911133,
      "step": 5656
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 5656,
      "training_step_time": 9.913825988769531
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 3.1340558528900146,
      "step": 5656
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 5656,
      "training_step_time": 9.909026145935059
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 3.135291814804077,
      "step": 5656
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 5656,
      "training_step_time": 9.91167664527893
    },
    {
      "epoch": 2.1575927734375e-05,
      "model_forward_time": 3.3977725505828857,
      "step": 5656
    },
    {
      "epoch": 2.1575927734375e-05,
      "step": 5656,
      "training_step_time": 10.276841640472412
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 3.1742782592773438,
      "step": 5657
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 5657,
      "training_step_time": 9.960497379302979
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 3.139908790588379,
      "step": 5657
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 5657,
      "training_step_time": 9.91773509979248
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 3.137556314468384,
      "step": 5657
    },
    {
      "dynamics_loss_avg": 0.05374479629099369,
      "epoch": 2.1579742431640625e-05,
      "step": 5657
    },
    {
      "action_loss_avg": 0.01998304607113823,
      "epoch": 2.1579742431640625e-05,
      "step": 5657
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 5657,
      "training_step_time": 9.915940046310425
    },
    {
      "epoch": 2.1579742431640625e-05,
      "model_forward_time": 3.140326499938965,
      "step": 5657
    },
    {
      "epoch": 2.1579742431640625e-05,
      "step": 5657,
      "training_step_time": 10.025352239608765
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 3.427372694015503,
      "step": 5658
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 5658,
      "training_step_time": 10.202980041503906
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 3.142263174057007,
      "step": 5658
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 5658,
      "training_step_time": 9.916101932525635
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 3.1353275775909424,
      "step": 5658
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 5658,
      "training_step_time": 9.911481618881226
    },
    {
      "epoch": 2.158355712890625e-05,
      "model_forward_time": 3.138549327850342,
      "step": 5658
    },
    {
      "epoch": 2.158355712890625e-05,
      "step": 5658,
      "training_step_time": 10.277124881744385
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 3.1829869747161865,
      "step": 5659
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 5659,
      "training_step_time": 9.95652151107788
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 3.4514658451080322,
      "step": 5659
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 5659,
      "training_step_time": 10.217937707901001
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 3.1375865936279297,
      "step": 5659
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 5659,
      "training_step_time": 9.909541130065918
    },
    {
      "epoch": 2.1587371826171874e-05,
      "model_forward_time": 3.137568235397339,
      "step": 5659
    },
    {
      "epoch": 2.1587371826171874e-05,
      "step": 5659,
      "training_step_time": 10.247350692749023
    },
    {
      "epoch": 2.15911865234375e-05,
      "grad_norm": 0.5810136795043945,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 5660
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 3.147263526916504,
      "step": 5660
    },
    {
      "dynamics_loss_avg": 0.07197928801178932,
      "epoch": 2.15911865234375e-05,
      "step": 5660
    },
    {
      "action_loss_avg": 0.0317415090277791,
      "epoch": 2.15911865234375e-05,
      "step": 5660
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 5660,
      "training_step_time": 9.92592740058899
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 3.131603479385376,
      "step": 5660
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 5660,
      "training_step_time": 9.902582883834839
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 3.4106526374816895,
      "step": 5660
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 5660,
      "training_step_time": 10.183332920074463
    },
    {
      "epoch": 2.15911865234375e-05,
      "model_forward_time": 3.143435478210449,
      "step": 5660
    },
    {
      "epoch": 2.15911865234375e-05,
      "step": 5660,
      "training_step_time": 10.344080924987793
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 3.1797873973846436,
      "step": 5661
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 5661,
      "training_step_time": 9.957285165786743
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 3.1410136222839355,
      "step": 5661
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 5661,
      "training_step_time": 9.922534942626953
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 3.144548177719116,
      "step": 5661
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 5661,
      "training_step_time": 9.921783447265625
    },
    {
      "epoch": 2.1595001220703123e-05,
      "model_forward_time": 3.4325499534606934,
      "step": 5661
    },
    {
      "epoch": 2.1595001220703123e-05,
      "step": 5661,
      "training_step_time": 10.300971984863281
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 3.184537887573242,
      "step": 5662
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 5662,
      "training_step_time": 9.965283155441284
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 3.1370279788970947,
      "step": 5662
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 5662,
      "training_step_time": 9.910372257232666
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 3.1418099403381348,
      "step": 5662
    },
    {
      "dynamics_loss_avg": 0.0621572844684124,
      "epoch": 2.159881591796875e-05,
      "step": 5662
    },
    {
      "action_loss_avg": 0.022523953579366207,
      "epoch": 2.159881591796875e-05,
      "step": 5662
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 5662,
      "training_step_time": 9.916449069976807
    },
    {
      "epoch": 2.159881591796875e-05,
      "model_forward_time": 3.1418113708496094,
      "step": 5662
    },
    {
      "epoch": 2.159881591796875e-05,
      "step": 5662,
      "training_step_time": 10.022574186325073
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 3.4515037536621094,
      "step": 5663
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 5663,
      "training_step_time": 10.22115159034729
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 3.135359525680542,
      "step": 5663
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 5663,
      "training_step_time": 9.90588927268982
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 3.135005235671997,
      "step": 5663
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 5663,
      "training_step_time": 9.904500246047974
    },
    {
      "epoch": 2.1602630615234376e-05,
      "model_forward_time": 3.1343398094177246,
      "step": 5663
    },
    {
      "epoch": 2.1602630615234376e-05,
      "step": 5663,
      "training_step_time": 10.28403115272522
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 3.1515913009643555,
      "step": 5664
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 5664,
      "training_step_time": 9.939159154891968
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 3.4314961433410645,
      "step": 5664
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 5664,
      "training_step_time": 10.207756280899048
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 3.1367433071136475,
      "step": 5664
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 5664,
      "training_step_time": 9.915346622467041
    },
    {
      "epoch": 2.16064453125e-05,
      "model_forward_time": 2.430314540863037,
      "step": 5664
    },
    {
      "epoch": 2.16064453125e-05,
      "step": 5664,
      "training_step_time": 10.275959968566895
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 3.1509816646575928,
      "step": 5665
    },
    {
      "dynamics_loss_avg": 0.057254190742969516,
      "epoch": 2.1610260009765625e-05,
      "step": 5665
    },
    {
      "action_loss_avg": 0.028850399330258368,
      "epoch": 2.1610260009765625e-05,
      "step": 5665
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 5665,
      "training_step_time": 9.939343214035034
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 3.1543989181518555,
      "step": 5665
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 5665,
      "training_step_time": 9.948667287826538
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 3.154615879058838,
      "step": 5665
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 5665,
      "training_step_time": 10.287384271621704
    },
    {
      "epoch": 2.1610260009765625e-05,
      "model_forward_time": 3.1465940475463867,
      "step": 5665
    },
    {
      "epoch": 2.1610260009765625e-05,
      "step": 5665,
      "training_step_time": 10.34939980506897
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 3.2112815380096436,
      "step": 5666
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 5666,
      "training_step_time": 10.026447772979736
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 3.1527888774871826,
      "step": 5666
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 5666,
      "training_step_time": 9.927756071090698
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 3.140479564666748,
      "step": 5666
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 5666,
      "training_step_time": 9.91204285621643
    },
    {
      "epoch": 2.161407470703125e-05,
      "model_forward_time": 3.143817663192749,
      "step": 5666
    },
    {
      "epoch": 2.161407470703125e-05,
      "step": 5666,
      "training_step_time": 10.340855598449707
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 3.163987636566162,
      "step": 5667
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 5667,
      "training_step_time": 9.953309535980225
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 3.147343635559082,
      "step": 5667
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 5667,
      "training_step_time": 9.926949262619019
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 3.140925645828247,
      "step": 5667
    },
    {
      "dynamics_loss_avg": 0.061947745829820634,
      "epoch": 2.1617889404296874e-05,
      "step": 5667
    },
    {
      "action_loss_avg": 0.029513728991150855,
      "epoch": 2.1617889404296874e-05,
      "step": 5667
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 5667,
      "training_step_time": 9.924824476242065
    },
    {
      "epoch": 2.1617889404296874e-05,
      "model_forward_time": 3.143892288208008,
      "step": 5667
    },
    {
      "epoch": 2.1617889404296874e-05,
      "step": 5667,
      "training_step_time": 10.022598505020142
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 2.4816460609436035,
      "step": 5668
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 5668,
      "training_step_time": 7.517960071563721
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 3.4460976123809814,
      "step": 5668
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 5668,
      "training_step_time": 10.230790615081787
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 3.1342146396636963,
      "step": 5668
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 5668,
      "training_step_time": 9.90948486328125
    },
    {
      "epoch": 2.16217041015625e-05,
      "model_forward_time": 3.136218547821045,
      "step": 5668
    },
    {
      "epoch": 2.16217041015625e-05,
      "step": 5668,
      "training_step_time": 10.044467210769653
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 3.159554958343506,
      "step": 5669
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 5669,
      "training_step_time": 9.936596632003784
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 3.442908763885498,
      "step": 5669
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 5669,
      "training_step_time": 10.213603734970093
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 3.130934238433838,
      "step": 5669
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 5669,
      "training_step_time": 9.901580572128296
    },
    {
      "epoch": 2.1625518798828123e-05,
      "model_forward_time": 3.1431682109832764,
      "step": 5669
    },
    {
      "epoch": 2.1625518798828123e-05,
      "step": 5669,
      "training_step_time": 10.273918390274048
    },
    {
      "epoch": 2.162933349609375e-05,
      "grad_norm": 0.7429137229919434,
      "learning_rate": 1e-05,
      "loss": 0.0842,
      "step": 5670
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 3.162186861038208,
      "step": 5670
    },
    {
      "dynamics_loss_avg": 0.04740300923585892,
      "epoch": 2.162933349609375e-05,
      "step": 5670
    },
    {
      "action_loss_avg": 0.01922585438005626,
      "epoch": 2.162933349609375e-05,
      "step": 5670
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 5670,
      "training_step_time": 9.945096969604492
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 3.1419155597686768,
      "step": 5670
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 5670,
      "training_step_time": 9.921461582183838
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 2.444167137145996,
      "step": 5670
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 5670,
      "training_step_time": 7.478825569152832
    },
    {
      "epoch": 2.162933349609375e-05,
      "model_forward_time": 3.4261398315429688,
      "step": 5670
    },
    {
      "epoch": 2.162933349609375e-05,
      "step": 5670,
      "training_step_time": 10.301013469696045
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 2.4786489009857178,
      "step": 5671
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 5671,
      "training_step_time": 7.512327671051025
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 3.141597270965576,
      "step": 5671
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 5671,
      "training_step_time": 9.922888278961182
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 3.1465516090393066,
      "step": 5671
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 5671,
      "training_step_time": 9.924146890640259
    },
    {
      "epoch": 2.1633148193359376e-05,
      "model_forward_time": 3.14336895942688,
      "step": 5671
    },
    {
      "epoch": 2.1633148193359376e-05,
      "step": 5671,
      "training_step_time": 10.33344316482544
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 3.185654401779175,
      "step": 5672
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 5672,
      "training_step_time": 9.969945669174194
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 3.1435039043426514,
      "step": 5672
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 5672,
      "training_step_time": 9.921875238418579
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 3.1449944972991943,
      "step": 5672
    },
    {
      "dynamics_loss_avg": 0.07347925752401352,
      "epoch": 2.1636962890625e-05,
      "step": 5672
    },
    {
      "action_loss_avg": 0.025043010292574764,
      "epoch": 2.1636962890625e-05,
      "step": 5672
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 5672,
      "training_step_time": 9.928022623062134
    },
    {
      "epoch": 2.1636962890625e-05,
      "model_forward_time": 3.142371892929077,
      "step": 5672
    },
    {
      "epoch": 2.1636962890625e-05,
      "step": 5672,
      "training_step_time": 10.28087043762207
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 3.1655218601226807,
      "step": 5673
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 5673,
      "training_step_time": 10.248910427093506
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 3.14791202545166,
      "step": 5673
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 5673,
      "training_step_time": 9.930723667144775
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 3.1469156742095947,
      "step": 5673
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 5673,
      "training_step_time": 9.930366039276123
    },
    {
      "epoch": 2.1640777587890625e-05,
      "model_forward_time": 3.146662950515747,
      "step": 5673
    },
    {
      "epoch": 2.1640777587890625e-05,
      "step": 5673,
      "training_step_time": 10.302499771118164
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 3.170557975769043,
      "step": 5674
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 5674,
      "training_step_time": 9.952232599258423
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 3.420226573944092,
      "step": 5674
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 5674,
      "training_step_time": 10.192551136016846
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 3.1408722400665283,
      "step": 5674
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 5674,
      "training_step_time": 9.912952423095703
    },
    {
      "epoch": 2.164459228515625e-05,
      "model_forward_time": 3.1391124725341797,
      "step": 5674
    },
    {
      "epoch": 2.164459228515625e-05,
      "step": 5674,
      "training_step_time": 10.357589960098267
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 3.196115255355835,
      "step": 5675
    },
    {
      "dynamics_loss_avg": 0.0391296080313623,
      "epoch": 2.1648406982421874e-05,
      "step": 5675
    },
    {
      "action_loss_avg": 0.014552579238079488,
      "epoch": 2.1648406982421874e-05,
      "step": 5675
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 5675,
      "training_step_time": 9.972245931625366
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 3.1629862785339355,
      "step": 5675
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 5675,
      "training_step_time": 9.9579496383667
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 3.5653257369995117,
      "step": 5675
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 5675,
      "training_step_time": 10.35708475112915
    },
    {
      "epoch": 2.1648406982421874e-05,
      "model_forward_time": 3.151411294937134,
      "step": 5675
    },
    {
      "epoch": 2.1648406982421874e-05,
      "step": 5675,
      "training_step_time": 10.027380466461182
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 3.179011583328247,
      "step": 5676
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 5676,
      "training_step_time": 9.956549167633057
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 3.148113250732422,
      "step": 5676
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 5676,
      "training_step_time": 9.924678564071655
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 3.142495632171631,
      "step": 5676
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 5676,
      "training_step_time": 9.92117190361023
    },
    {
      "epoch": 2.16522216796875e-05,
      "model_forward_time": 3.150907039642334,
      "step": 5676
    },
    {
      "epoch": 2.16522216796875e-05,
      "step": 5676,
      "training_step_time": 10.364066362380981
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 3.1986372470855713,
      "step": 5677
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 5677,
      "training_step_time": 9.976502180099487
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 3.1409573554992676,
      "step": 5677
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 5677,
      "training_step_time": 9.913866519927979
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 3.1390023231506348,
      "step": 5677
    },
    {
      "dynamics_loss_avg": 0.06636097654700279,
      "epoch": 2.1656036376953123e-05,
      "step": 5677
    },
    {
      "action_loss_avg": 0.02326713940128684,
      "epoch": 2.1656036376953123e-05,
      "step": 5677
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 5677,
      "training_step_time": 9.918154954910278
    },
    {
      "epoch": 2.1656036376953123e-05,
      "model_forward_time": 3.1414101123809814,
      "step": 5677
    },
    {
      "epoch": 2.1656036376953123e-05,
      "step": 5677,
      "training_step_time": 10.311440706253052
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 3.4394123554229736,
      "step": 5678
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 5678,
      "training_step_time": 10.22169804573059
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 2.435169219970703,
      "step": 5678
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 5678,
      "training_step_time": 7.470817804336548
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 3.137873411178589,
      "step": 5678
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 5678,
      "training_step_time": 9.9165198802948
    },
    {
      "epoch": 2.165985107421875e-05,
      "model_forward_time": 3.148618698120117,
      "step": 5678
    },
    {
      "epoch": 2.165985107421875e-05,
      "step": 5678,
      "training_step_time": 10.260814189910889
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 3.1595451831817627,
      "step": 5679
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 5679,
      "training_step_time": 9.941351175308228
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 3.1386759281158447,
      "step": 5679
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 5679,
      "training_step_time": 9.913902044296265
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 2.747356414794922,
      "step": 5679
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 5679,
      "training_step_time": 7.777791500091553
    },
    {
      "epoch": 2.1663665771484376e-05,
      "model_forward_time": 3.1352760791778564,
      "step": 5679
    },
    {
      "epoch": 2.1663665771484376e-05,
      "step": 5679,
      "training_step_time": 10.032672882080078
    },
    {
      "epoch": 2.166748046875e-05,
      "grad_norm": 0.6999729871749878,
      "learning_rate": 1e-05,
      "loss": 0.0873,
      "step": 5680
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 3.1710922718048096,
      "step": 5680
    },
    {
      "dynamics_loss_avg": 0.06109077036380768,
      "epoch": 2.166748046875e-05,
      "step": 5680
    },
    {
      "action_loss_avg": 0.025164423044770956,
      "epoch": 2.166748046875e-05,
      "step": 5680
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 5680,
      "training_step_time": 9.950563430786133
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 3.135704517364502,
      "step": 5680
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 5680,
      "training_step_time": 9.907663106918335
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 3.407841444015503,
      "step": 5680
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 5680,
      "training_step_time": 10.1824049949646
    },
    {
      "epoch": 2.166748046875e-05,
      "model_forward_time": 3.1389002799987793,
      "step": 5680
    },
    {
      "epoch": 2.166748046875e-05,
      "step": 5680,
      "training_step_time": 10.020347833633423
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 3.1524384021759033,
      "step": 5681
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 5681,
      "training_step_time": 9.929611444473267
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 3.135125160217285,
      "step": 5681
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 5681,
      "training_step_time": 9.914407014846802
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 3.1443328857421875,
      "step": 5681
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 5681,
      "training_step_time": 9.921093940734863
    },
    {
      "epoch": 2.1671295166015625e-05,
      "model_forward_time": 3.1377766132354736,
      "step": 5681
    },
    {
      "epoch": 2.1671295166015625e-05,
      "step": 5681,
      "training_step_time": 10.34069561958313
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 3.153963088989258,
      "step": 5682
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 5682,
      "training_step_time": 9.93329381942749
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 3.137343168258667,
      "step": 5682
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 5682,
      "training_step_time": 9.927100658416748
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 3.1437630653381348,
      "step": 5682
    },
    {
      "dynamics_loss_avg": 0.0745039762929082,
      "epoch": 2.167510986328125e-05,
      "step": 5682
    },
    {
      "action_loss_avg": 0.029009946063160895,
      "epoch": 2.167510986328125e-05,
      "step": 5682
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 5682,
      "training_step_time": 9.927196502685547
    },
    {
      "epoch": 2.167510986328125e-05,
      "model_forward_time": 2.4361870288848877,
      "step": 5682
    },
    {
      "epoch": 2.167510986328125e-05,
      "step": 5682,
      "training_step_time": 10.04276704788208
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 3.4655137062072754,
      "step": 5683
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 5683,
      "training_step_time": 10.24651312828064
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 3.1363613605499268,
      "step": 5683
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 5683,
      "training_step_time": 9.918564558029175
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 3.1431639194488525,
      "step": 5683
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 5683,
      "training_step_time": 9.928333759307861
    },
    {
      "epoch": 2.1678924560546874e-05,
      "model_forward_time": 3.137871026992798,
      "step": 5683
    },
    {
      "epoch": 2.1678924560546874e-05,
      "step": 5683,
      "training_step_time": 10.055182456970215
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 3.144613265991211,
      "step": 5684
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 5684,
      "training_step_time": 9.924768447875977
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 3.4326460361480713,
      "step": 5684
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 5684,
      "training_step_time": 10.207590579986572
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 3.1391379833221436,
      "step": 5684
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 5684,
      "training_step_time": 9.914382696151733
    },
    {
      "epoch": 2.16827392578125e-05,
      "model_forward_time": 3.134838104248047,
      "step": 5684
    },
    {
      "epoch": 2.16827392578125e-05,
      "step": 5684,
      "training_step_time": 10.032243728637695
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 3.1679604053497314,
      "step": 5685
    },
    {
      "dynamics_loss_avg": 0.05874189306050539,
      "epoch": 2.1686553955078124e-05,
      "step": 5685
    },
    {
      "action_loss_avg": 0.019814338907599448,
      "epoch": 2.1686553955078124e-05,
      "step": 5685
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 5685,
      "training_step_time": 9.957460880279541
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 3.1399223804473877,
      "step": 5685
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 5685,
      "training_step_time": 9.919928550720215
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 3.1355040073394775,
      "step": 5685
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 5685,
      "training_step_time": 10.211353302001953
    },
    {
      "epoch": 2.1686553955078124e-05,
      "model_forward_time": 3.135847806930542,
      "step": 5685
    },
    {
      "epoch": 2.1686553955078124e-05,
      "step": 5685,
      "training_step_time": 10.30902361869812
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 3.1919105052948,
      "step": 5686
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 5686,
      "training_step_time": 9.975704193115234
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 3.1379828453063965,
      "step": 5686
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 5686,
      "training_step_time": 9.916637182235718
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 3.137324333190918,
      "step": 5686
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 5686,
      "training_step_time": 9.92290997505188
    },
    {
      "epoch": 2.169036865234375e-05,
      "model_forward_time": 3.1405396461486816,
      "step": 5686
    },
    {
      "epoch": 2.169036865234375e-05,
      "step": 5686,
      "training_step_time": 10.35549283027649
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 2.444342613220215,
      "step": 5687
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 5687,
      "training_step_time": 7.483060836791992
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 3.1352884769439697,
      "step": 5687
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 5687,
      "training_step_time": 9.915484428405762
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 2.4302291870117188,
      "step": 5687
    },
    {
      "dynamics_loss_avg": 0.0675764162093401,
      "epoch": 2.1694183349609376e-05,
      "step": 5687
    },
    {
      "action_loss_avg": 0.024942884407937526,
      "epoch": 2.1694183349609376e-05,
      "step": 5687
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 5687,
      "training_step_time": 7.467846393585205
    },
    {
      "epoch": 2.1694183349609376e-05,
      "model_forward_time": 3.1357665061950684,
      "step": 5687
    },
    {
      "epoch": 2.1694183349609376e-05,
      "step": 5687,
      "training_step_time": 10.250221252441406
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 3.14062762260437,
      "step": 5688
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 5688,
      "training_step_time": 9.92180848121643
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 3.4430460929870605,
      "step": 5688
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 5688,
      "training_step_time": 10.215473413467407
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 3.1326279640197754,
      "step": 5688
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 5688,
      "training_step_time": 9.921818017959595
    },
    {
      "epoch": 2.1697998046875e-05,
      "model_forward_time": 3.1390843391418457,
      "step": 5688
    },
    {
      "epoch": 2.1697998046875e-05,
      "step": 5688,
      "training_step_time": 10.265674829483032
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 3.1500003337860107,
      "step": 5689
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 5689,
      "training_step_time": 9.930742025375366
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 3.1322388648986816,
      "step": 5689
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 5689,
      "training_step_time": 9.910595655441284
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 3.391967296600342,
      "step": 5689
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 5689,
      "training_step_time": 10.170376062393188
    },
    {
      "epoch": 2.1701812744140625e-05,
      "model_forward_time": 3.1361992359161377,
      "step": 5689
    },
    {
      "epoch": 2.1701812744140625e-05,
      "step": 5689,
      "training_step_time": 10.020835399627686
    },
    {
      "epoch": 2.170562744140625e-05,
      "grad_norm": 0.5852240920066833,
      "learning_rate": 1e-05,
      "loss": 0.0899,
      "step": 5690
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 3.1711597442626953,
      "step": 5690
    },
    {
      "dynamics_loss_avg": 0.06954334992915392,
      "epoch": 2.170562744140625e-05,
      "step": 5690
    },
    {
      "action_loss_avg": 0.02422152701765299,
      "epoch": 2.170562744140625e-05,
      "step": 5690
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 5690,
      "training_step_time": 9.94892406463623
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 3.1332356929779053,
      "step": 5690
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 5690,
      "training_step_time": 9.907636880874634
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 3.134453296661377,
      "step": 5690
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 5690,
      "training_step_time": 10.201594352722168
    },
    {
      "epoch": 2.170562744140625e-05,
      "model_forward_time": 3.1371989250183105,
      "step": 5690
    },
    {
      "epoch": 2.170562744140625e-05,
      "step": 5690,
      "training_step_time": 10.275095224380493
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 3.1497395038604736,
      "step": 5691
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 5691,
      "training_step_time": 9.927151918411255
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 3.1341264247894287,
      "step": 5691
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 5691,
      "training_step_time": 9.910557508468628
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 3.1368978023529053,
      "step": 5691
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 5691,
      "training_step_time": 9.913601875305176
    },
    {
      "epoch": 2.1709442138671874e-05,
      "model_forward_time": 3.4467074871063232,
      "step": 5691
    },
    {
      "epoch": 2.1709442138671874e-05,
      "step": 5691,
      "training_step_time": 10.317106485366821
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 3.178091526031494,
      "step": 5692
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 5692,
      "training_step_time": 9.962586402893066
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 3.1354870796203613,
      "step": 5692
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 5692,
      "training_step_time": 9.919963836669922
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 3.1404364109039307,
      "step": 5692
    },
    {
      "dynamics_loss_avg": 0.0725922618061304,
      "epoch": 2.17132568359375e-05,
      "step": 5692
    },
    {
      "action_loss_avg": 0.02272342350333929,
      "epoch": 2.17132568359375e-05,
      "step": 5692
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 5692,
      "training_step_time": 9.920580863952637
    },
    {
      "epoch": 2.17132568359375e-05,
      "model_forward_time": 3.1355972290039062,
      "step": 5692
    },
    {
      "epoch": 2.17132568359375e-05,
      "step": 5692,
      "training_step_time": 10.288849353790283
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 3.1679751873016357,
      "step": 5693
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 5693,
      "training_step_time": 10.208761930465698
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 3.131169557571411,
      "step": 5693
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 5693,
      "training_step_time": 9.910985469818115
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 3.138906240463257,
      "step": 5693
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 5693,
      "training_step_time": 9.920595407485962
    },
    {
      "epoch": 2.1717071533203124e-05,
      "model_forward_time": 3.1379847526550293,
      "step": 5693
    },
    {
      "epoch": 2.1717071533203124e-05,
      "step": 5693,
      "training_step_time": 10.254510879516602
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 3.1475625038146973,
      "step": 5694
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 5694,
      "training_step_time": 9.926243543624878
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 3.392277956008911,
      "step": 5694
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 5694,
      "training_step_time": 10.16902232170105
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 3.136716842651367,
      "step": 5694
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 5694,
      "training_step_time": 9.914757013320923
    },
    {
      "epoch": 2.172088623046875e-05,
      "model_forward_time": 3.133394479751587,
      "step": 5694
    },
    {
      "epoch": 2.172088623046875e-05,
      "step": 5694,
      "training_step_time": 10.227646589279175
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 3.149653911590576,
      "step": 5695
    },
    {
      "dynamics_loss_avg": 0.05612399559468031,
      "epoch": 2.1724700927734376e-05,
      "step": 5695
    },
    {
      "action_loss_avg": 0.020356307341717183,
      "epoch": 2.1724700927734376e-05,
      "step": 5695
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 5695,
      "training_step_time": 9.937762975692749
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 3.145707607269287,
      "step": 5695
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 5695,
      "training_step_time": 9.928827285766602
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 3.428640604019165,
      "step": 5695
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 5695,
      "training_step_time": 10.208434343338013
    },
    {
      "epoch": 2.1724700927734376e-05,
      "model_forward_time": 3.1468403339385986,
      "step": 5695
    },
    {
      "epoch": 2.1724700927734376e-05,
      "step": 5695,
      "training_step_time": 10.29546594619751
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 3.1913719177246094,
      "step": 5696
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 5696,
      "training_step_time": 9.974576473236084
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 3.144063949584961,
      "step": 5696
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 5696,
      "training_step_time": 9.92561411857605
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 3.1408843994140625,
      "step": 5696
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 5696,
      "training_step_time": 9.920156002044678
    },
    {
      "epoch": 2.1728515625e-05,
      "model_forward_time": 3.143120288848877,
      "step": 5696
    },
    {
      "epoch": 2.1728515625e-05,
      "step": 5696,
      "training_step_time": 10.302510499954224
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 3.1625609397888184,
      "step": 5697
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 5697,
      "training_step_time": 9.944228172302246
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 3.1405909061431885,
      "step": 5697
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 5697,
      "training_step_time": 9.916237592697144
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 2.441678047180176,
      "step": 5697
    },
    {
      "dynamics_loss_avg": 0.06640384690836072,
      "epoch": 2.1732330322265625e-05,
      "step": 5697
    },
    {
      "action_loss_avg": 0.02234553135931492,
      "epoch": 2.1732330322265625e-05,
      "step": 5697
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 5697,
      "training_step_time": 7.474305868148804
    },
    {
      "epoch": 2.1732330322265625e-05,
      "model_forward_time": 3.1439383029937744,
      "step": 5697
    },
    {
      "epoch": 2.1732330322265625e-05,
      "step": 5697,
      "training_step_time": 10.346114873886108
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 3.1832146644592285,
      "step": 5698
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 5698,
      "training_step_time": 9.961623191833496
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 3.447450637817383,
      "step": 5698
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 5698,
      "training_step_time": 10.22709608078003
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 3.141071319580078,
      "step": 5698
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 5698,
      "training_step_time": 9.925166130065918
    },
    {
      "epoch": 2.173614501953125e-05,
      "model_forward_time": 3.157289981842041,
      "step": 5698
    },
    {
      "epoch": 2.173614501953125e-05,
      "step": 5698,
      "training_step_time": 10.268961191177368
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 3.152740478515625,
      "step": 5699
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 5699,
      "training_step_time": 9.954474210739136
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 3.1677141189575195,
      "step": 5699
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 5699,
      "training_step_time": 10.300782680511475
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 3.146591901779175,
      "step": 5699
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 5699,
      "training_step_time": 9.930808305740356
    },
    {
      "epoch": 2.1739959716796874e-05,
      "model_forward_time": 3.14624285697937,
      "step": 5699
    },
    {
      "epoch": 2.1739959716796874e-05,
      "step": 5699,
      "training_step_time": 10.309859275817871
    },
    {
      "epoch": 2.17437744140625e-05,
      "grad_norm": 0.75218665599823,
      "learning_rate": 1e-05,
      "loss": 0.0945,
      "step": 5700
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 3.187443256378174,
      "step": 5700
    },
    {
      "dynamics_loss_avg": 0.07144740074872971,
      "epoch": 2.17437744140625e-05,
      "step": 5700
    },
    {
      "action_loss_avg": 0.028632648382335903,
      "epoch": 2.17437744140625e-05,
      "step": 5700
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 5700,
      "training_step_time": 9.971590995788574
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 3.1496427059173584,
      "step": 5700
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 5700,
      "training_step_time": 9.92890191078186
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 3.4173576831817627,
      "step": 5700
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 5700,
      "training_step_time": 10.196730375289917
    },
    {
      "epoch": 2.17437744140625e-05,
      "model_forward_time": 3.1476094722747803,
      "step": 5700
    },
    {
      "epoch": 2.17437744140625e-05,
      "step": 5700,
      "training_step_time": 10.279862403869629
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 3.165703058242798,
      "step": 5701
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 5701,
      "training_step_time": 9.94332242012024
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 3.1461594104766846,
      "step": 5701
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 5701,
      "training_step_time": 9.924694776535034
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 3.144273519515991,
      "step": 5701
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 5701,
      "training_step_time": 9.920997858047485
    },
    {
      "epoch": 2.1747589111328124e-05,
      "model_forward_time": 3.140690803527832,
      "step": 5701
    },
    {
      "epoch": 2.1747589111328124e-05,
      "step": 5701,
      "training_step_time": 10.315024852752686
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 3.1601929664611816,
      "step": 5702
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 5702,
      "training_step_time": 9.949725151062012
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 3.147463321685791,
      "step": 5702
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 5702,
      "training_step_time": 9.92804217338562
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 3.1459801197052,
      "step": 5702
    },
    {
      "dynamics_loss_avg": 0.06914015971124172,
      "epoch": 2.175140380859375e-05,
      "step": 5702
    },
    {
      "action_loss_avg": 0.02436174135655165,
      "epoch": 2.175140380859375e-05,
      "step": 5702
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 5702,
      "training_step_time": 9.93142819404602
    },
    {
      "epoch": 2.175140380859375e-05,
      "model_forward_time": 3.152536153793335,
      "step": 5702
    },
    {
      "epoch": 2.175140380859375e-05,
      "step": 5702,
      "training_step_time": 10.307507753372192
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 3.4813294410705566,
      "step": 5703
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 5703,
      "training_step_time": 10.258034944534302
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 3.134448766708374,
      "step": 5703
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 5703,
      "training_step_time": 9.911445379257202
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 3.1385741233825684,
      "step": 5703
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 5703,
      "training_step_time": 9.911547660827637
    },
    {
      "epoch": 2.1755218505859376e-05,
      "model_forward_time": 3.1424055099487305,
      "step": 5703
    },
    {
      "epoch": 2.1755218505859376e-05,
      "step": 5703,
      "training_step_time": 10.225342750549316
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 3.146536111831665,
      "step": 5704
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 5704,
      "training_step_time": 9.920854568481445
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 2.7310805320739746,
      "step": 5704
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 5704,
      "training_step_time": 7.754547834396362
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 3.1363067626953125,
      "step": 5704
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 5704,
      "training_step_time": 9.900797843933105
    },
    {
      "epoch": 2.1759033203125e-05,
      "model_forward_time": 3.1286628246307373,
      "step": 5704
    },
    {
      "epoch": 2.1759033203125e-05,
      "step": 5704,
      "training_step_time": 10.006007194519043
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 3.1411852836608887,
      "step": 5705
    },
    {
      "dynamics_loss_avg": 0.06233777701854706,
      "epoch": 2.1762847900390625e-05,
      "step": 5705
    },
    {
      "action_loss_avg": 0.019562786864116787,
      "epoch": 2.1762847900390625e-05,
      "step": 5705
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 5705,
      "training_step_time": 9.925740957260132
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 3.128469228744507,
      "step": 5705
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 5705,
      "training_step_time": 9.897750854492188
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 3.4183719158172607,
      "step": 5705
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 5705,
      "training_step_time": 10.189741373062134
    },
    {
      "epoch": 2.1762847900390625e-05,
      "model_forward_time": 3.126620054244995,
      "step": 5705
    },
    {
      "epoch": 2.1762847900390625e-05,
      "step": 5705,
      "training_step_time": 10.009721517562866
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 3.1489779949188232,
      "step": 5706
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 5706,
      "training_step_time": 9.921328783035278
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 3.133331298828125,
      "step": 5706
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 5706,
      "training_step_time": 9.910544633865356
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 3.128288507461548,
      "step": 5706
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 5706,
      "training_step_time": 9.898920059204102
    },
    {
      "epoch": 2.176666259765625e-05,
      "model_forward_time": 3.3942573070526123,
      "step": 5706
    },
    {
      "epoch": 2.176666259765625e-05,
      "step": 5706,
      "training_step_time": 10.321296453475952
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 3.1678285598754883,
      "step": 5707
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 5707,
      "training_step_time": 9.938061952590942
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 3.1358401775360107,
      "step": 5707
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 5707,
      "training_step_time": 9.910306930541992
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 3.133310556411743,
      "step": 5707
    },
    {
      "dynamics_loss_avg": 0.06043727546930313,
      "epoch": 2.1770477294921874e-05,
      "step": 5707
    },
    {
      "action_loss_avg": 0.02473878124728799,
      "epoch": 2.1770477294921874e-05,
      "step": 5707
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 5707,
      "training_step_time": 9.915584087371826
    },
    {
      "epoch": 2.1770477294921874e-05,
      "model_forward_time": 3.1372225284576416,
      "step": 5707
    },
    {
      "epoch": 2.1770477294921874e-05,
      "step": 5707,
      "training_step_time": 10.02471923828125
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 3.442561388015747,
      "step": 5708
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 5708,
      "training_step_time": 10.219637155532837
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 3.1407525539398193,
      "step": 5708
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 5708,
      "training_step_time": 9.919595718383789
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 3.1318039894104004,
      "step": 5708
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 5708,
      "training_step_time": 9.909581899642944
    },
    {
      "epoch": 2.17742919921875e-05,
      "model_forward_time": 3.1355161666870117,
      "step": 5708
    },
    {
      "epoch": 2.17742919921875e-05,
      "step": 5708,
      "training_step_time": 10.018715620040894
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 3.1568381786346436,
      "step": 5709
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 5709,
      "training_step_time": 9.946449518203735
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 3.4487218856811523,
      "step": 5709
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 5709,
      "training_step_time": 10.233978748321533
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 3.1393680572509766,
      "step": 5709
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 5709,
      "training_step_time": 9.925528764724731
    },
    {
      "epoch": 2.1778106689453124e-05,
      "model_forward_time": 3.143592119216919,
      "step": 5709
    },
    {
      "epoch": 2.1778106689453124e-05,
      "step": 5709,
      "training_step_time": 10.31385850906372
    },
    {
      "epoch": 2.178192138671875e-05,
      "grad_norm": 0.7202146649360657,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 5710
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 3.1874160766601562,
      "step": 5710
    },
    {
      "dynamics_loss_avg": 0.06029564831405878,
      "epoch": 2.178192138671875e-05,
      "step": 5710
    },
    {
      "action_loss_avg": 0.022478135116398335,
      "epoch": 2.178192138671875e-05,
      "step": 5710
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 5710,
      "training_step_time": 9.977370738983154
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 2.4354307651519775,
      "step": 5710
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 5710,
      "training_step_time": 7.475358724594116
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 3.1425158977508545,
      "step": 5710
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 5710,
      "training_step_time": 10.208995342254639
    },
    {
      "epoch": 2.178192138671875e-05,
      "model_forward_time": 3.1439614295959473,
      "step": 5710
    },
    {
      "epoch": 2.178192138671875e-05,
      "step": 5710,
      "training_step_time": 10.269904851913452
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 3.1514153480529785,
      "step": 5711
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 5711,
      "training_step_time": 9.93683123588562
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 3.135490894317627,
      "step": 5711
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 5711,
      "training_step_time": 9.909568309783936
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 3.13476300239563,
      "step": 5711
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 5711,
      "training_step_time": 9.909721374511719
    },
    {
      "epoch": 2.1785736083984376e-05,
      "model_forward_time": 3.419661521911621,
      "step": 5711
    },
    {
      "epoch": 2.1785736083984376e-05,
      "step": 5711,
      "training_step_time": 10.2877676486969
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 3.1517817974090576,
      "step": 5712
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 5712,
      "training_step_time": 9.937351703643799
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 3.137833595275879,
      "step": 5712
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 5712,
      "training_step_time": 9.911193132400513
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 3.1503963470458984,
      "step": 5712
    },
    {
      "dynamics_loss_avg": 0.06628549974411727,
      "epoch": 2.178955078125e-05,
      "step": 5712
    },
    {
      "action_loss_avg": 0.02936064461246133,
      "epoch": 2.178955078125e-05,
      "step": 5712
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 5712,
      "training_step_time": 9.933161735534668
    },
    {
      "epoch": 2.178955078125e-05,
      "model_forward_time": 3.142733573913574,
      "step": 5712
    },
    {
      "epoch": 2.178955078125e-05,
      "step": 5712,
      "training_step_time": 10.279762268066406
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 3.498016834259033,
      "step": 5713
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 5713,
      "training_step_time": 10.278430461883545
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 3.1479108333587646,
      "step": 5713
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 5713,
      "training_step_time": 9.93128228187561
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 3.143610954284668,
      "step": 5713
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 5713,
      "training_step_time": 9.923308849334717
    },
    {
      "epoch": 2.1793365478515625e-05,
      "model_forward_time": 3.1402151584625244,
      "step": 5713
    },
    {
      "epoch": 2.1793365478515625e-05,
      "step": 5713,
      "training_step_time": 10.284572839736938
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 3.164015769958496,
      "step": 5714
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 5714,
      "training_step_time": 9.944624900817871
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 3.4101364612579346,
      "step": 5714
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 5714,
      "training_step_time": 10.185657501220703
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 3.1424334049224854,
      "step": 5714
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 5714,
      "training_step_time": 9.922402620315552
    },
    {
      "epoch": 2.179718017578125e-05,
      "model_forward_time": 3.138002634048462,
      "step": 5714
    },
    {
      "epoch": 2.179718017578125e-05,
      "step": 5714,
      "training_step_time": 10.325626850128174
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 3.149829626083374,
      "step": 5715
    },
    {
      "dynamics_loss_avg": 0.06644066274166108,
      "epoch": 2.1800994873046874e-05,
      "step": 5715
    },
    {
      "action_loss_avg": 0.02359894383698702,
      "epoch": 2.1800994873046874e-05,
      "step": 5715
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 5715,
      "training_step_time": 9.927411317825317
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 3.1415703296661377,
      "step": 5715
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 5715,
      "training_step_time": 9.92049241065979
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 3.4263644218444824,
      "step": 5715
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 5715,
      "training_step_time": 10.200915336608887
    },
    {
      "epoch": 2.1800994873046874e-05,
      "model_forward_time": 3.1422533988952637,
      "step": 5715
    },
    {
      "epoch": 2.1800994873046874e-05,
      "step": 5715,
      "training_step_time": 10.032729625701904
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 3.179870367050171,
      "step": 5716
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 5716,
      "training_step_time": 9.96131420135498
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 3.1479852199554443,
      "step": 5716
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 5716,
      "training_step_time": 9.930641174316406
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 3.148392677307129,
      "step": 5716
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 5716,
      "training_step_time": 9.93506669998169
    },
    {
      "epoch": 2.18048095703125e-05,
      "model_forward_time": 3.147235870361328,
      "step": 5716
    },
    {
      "epoch": 2.18048095703125e-05,
      "step": 5716,
      "training_step_time": 10.327840566635132
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 2.483234405517578,
      "step": 5717
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 5717,
      "training_step_time": 7.526879549026489
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 3.133127212524414,
      "step": 5717
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 5717,
      "training_step_time": 9.926461458206177
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 3.1281282901763916,
      "step": 5717
    },
    {
      "dynamics_loss_avg": 0.048003040440380576,
      "epoch": 2.1808624267578124e-05,
      "step": 5717
    },
    {
      "action_loss_avg": 0.017880812007933854,
      "epoch": 2.1808624267578124e-05,
      "step": 5717
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 5717,
      "training_step_time": 9.900927782058716
    },
    {
      "epoch": 2.1808624267578124e-05,
      "model_forward_time": 3.138597249984741,
      "step": 5717
    },
    {
      "epoch": 2.1808624267578124e-05,
      "step": 5717,
      "training_step_time": 10.299055337905884
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 3.1709563732147217,
      "step": 5718
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 5718,
      "training_step_time": 9.94455623626709
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 3.442474603652954,
      "step": 5718
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 5718,
      "training_step_time": 10.22094464302063
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 3.132049798965454,
      "step": 5718
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 5718,
      "training_step_time": 9.906739473342896
    },
    {
      "epoch": 2.181243896484375e-05,
      "model_forward_time": 3.1360466480255127,
      "step": 5718
    },
    {
      "epoch": 2.181243896484375e-05,
      "step": 5718,
      "training_step_time": 10.25533413887024
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 3.1520025730133057,
      "step": 5719
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 5719,
      "training_step_time": 9.931339979171753
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 3.144707441329956,
      "step": 5719
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 5719,
      "training_step_time": 10.240661382675171
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 3.147871494293213,
      "step": 5719
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 5719,
      "training_step_time": 9.929495573043823
    },
    {
      "epoch": 2.1816253662109376e-05,
      "model_forward_time": 3.1596527099609375,
      "step": 5719
    },
    {
      "epoch": 2.1816253662109376e-05,
      "step": 5719,
      "training_step_time": 10.335224866867065
    },
    {
      "epoch": 2.1820068359375e-05,
      "grad_norm": 0.3573676645755768,
      "learning_rate": 1e-05,
      "loss": 0.0858,
      "step": 5720
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 3.189161539077759,
      "step": 5720
    },
    {
      "dynamics_loss_avg": 0.05560374353080988,
      "epoch": 2.1820068359375e-05,
      "step": 5720
    },
    {
      "action_loss_avg": 0.02167349332012236,
      "epoch": 2.1820068359375e-05,
      "step": 5720
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 5720,
      "training_step_time": 9.97730803489685
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 3.1449451446533203,
      "step": 5720
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 5720,
      "training_step_time": 9.927059173583984
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 3.4141383171081543,
      "step": 5720
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 5720,
      "training_step_time": 10.190843343734741
    },
    {
      "epoch": 2.1820068359375e-05,
      "model_forward_time": 3.1463963985443115,
      "step": 5720
    },
    {
      "epoch": 2.1820068359375e-05,
      "step": 5720,
      "training_step_time": 10.253156185150146
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 3.1506617069244385,
      "step": 5721
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 5721,
      "training_step_time": 9.931261539459229
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 3.1434826850891113,
      "step": 5721
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 5721,
      "training_step_time": 9.918063640594482
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 3.153083324432373,
      "step": 5721
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 5721,
      "training_step_time": 9.932913541793823
    },
    {
      "epoch": 2.1823883056640625e-05,
      "model_forward_time": 3.142836809158325,
      "step": 5721
    },
    {
      "epoch": 2.1823883056640625e-05,
      "step": 5721,
      "training_step_time": 10.283716440200806
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 3.186176300048828,
      "step": 5722
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 5722,
      "training_step_time": 9.960193634033203
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 3.1433956623077393,
      "step": 5722
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 5722,
      "training_step_time": 9.9154794216156
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 3.142387628555298,
      "step": 5722
    },
    {
      "dynamics_loss_avg": 0.058664731308817865,
      "epoch": 2.182769775390625e-05,
      "step": 5722
    },
    {
      "action_loss_avg": 0.023521758802235126,
      "epoch": 2.182769775390625e-05,
      "step": 5722
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 5722,
      "training_step_time": 9.917985439300537
    },
    {
      "epoch": 2.182769775390625e-05,
      "model_forward_time": 3.1416847705841064,
      "step": 5722
    },
    {
      "epoch": 2.182769775390625e-05,
      "step": 5722,
      "training_step_time": 10.02307915687561
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 3.4745566844940186,
      "step": 5723
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 5723,
      "training_step_time": 10.251281499862671
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 3.143042802810669,
      "step": 5723
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 5723,
      "training_step_time": 9.91862154006958
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 3.145725965499878,
      "step": 5723
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 5723,
      "training_step_time": 9.922274589538574
    },
    {
      "epoch": 2.1831512451171874e-05,
      "model_forward_time": 3.1508703231811523,
      "step": 5723
    },
    {
      "epoch": 2.1831512451171874e-05,
      "step": 5723,
      "training_step_time": 10.257370948791504
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 3.1517014503479004,
      "step": 5724
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 5724,
      "training_step_time": 9.93159532546997
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 3.4062869548797607,
      "step": 5724
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 5724,
      "training_step_time": 10.183613300323486
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 3.1456098556518555,
      "step": 5724
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 5724,
      "training_step_time": 9.921514511108398
    },
    {
      "epoch": 2.18353271484375e-05,
      "model_forward_time": 3.139566421508789,
      "step": 5724
    },
    {
      "epoch": 2.18353271484375e-05,
      "step": 5724,
      "training_step_time": 10.313028812408447
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 3.156787872314453,
      "step": 5725
    },
    {
      "dynamics_loss_avg": 0.051179122552275656,
      "epoch": 2.1839141845703124e-05,
      "step": 5725
    },
    {
      "action_loss_avg": 0.016240489669144154,
      "epoch": 2.1839141845703124e-05,
      "step": 5725
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 5725,
      "training_step_time": 9.929964780807495
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 3.1387710571289062,
      "step": 5725
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 5725,
      "training_step_time": 9.912734985351562
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 3.42289662361145,
      "step": 5725
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 5725,
      "training_step_time": 10.196680784225464
    },
    {
      "epoch": 2.1839141845703124e-05,
      "model_forward_time": 3.1414456367492676,
      "step": 5725
    },
    {
      "epoch": 2.1839141845703124e-05,
      "step": 5725,
      "training_step_time": 10.318872213363647
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 3.1533615589141846,
      "step": 5726
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 5726,
      "training_step_time": 9.929653644561768
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 3.140777587890625,
      "step": 5726
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 5726,
      "training_step_time": 9.91043734550476
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 3.139791250228882,
      "step": 5726
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 5726,
      "training_step_time": 9.911037921905518
    },
    {
      "epoch": 2.184295654296875e-05,
      "model_forward_time": 3.435317039489746,
      "step": 5726
    },
    {
      "epoch": 2.184295654296875e-05,
      "step": 5726,
      "training_step_time": 10.297771692276001
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 3.16772723197937,
      "step": 5727
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 5727,
      "training_step_time": 9.936268329620361
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 3.144483804702759,
      "step": 5727
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 5727,
      "training_step_time": 9.915062189102173
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 3.1405580043792725,
      "step": 5727
    },
    {
      "dynamics_loss_avg": 0.058293388411402705,
      "epoch": 2.1846771240234376e-05,
      "step": 5727
    },
    {
      "action_loss_avg": 0.02308632303029299,
      "epoch": 2.1846771240234376e-05,
      "step": 5727
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 5727,
      "training_step_time": 9.91757321357727
    },
    {
      "epoch": 2.1846771240234376e-05,
      "model_forward_time": 3.1423683166503906,
      "step": 5727
    },
    {
      "epoch": 2.1846771240234376e-05,
      "step": 5727,
      "training_step_time": 10.305058002471924
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 3.418954372406006,
      "step": 5728
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 5728,
      "training_step_time": 10.18584132194519
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 3.1440389156341553,
      "step": 5728
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 5728,
      "training_step_time": 9.92130422592163
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 2.4389171600341797,
      "step": 5728
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 5728,
      "training_step_time": 7.470453262329102
    },
    {
      "epoch": 2.18505859375e-05,
      "model_forward_time": 3.1452667713165283,
      "step": 5728
    },
    {
      "epoch": 2.18505859375e-05,
      "step": 5728,
      "training_step_time": 10.321018934249878
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 3.1551687717437744,
      "step": 5729
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 5729,
      "training_step_time": 9.93198561668396
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 3.4259564876556396,
      "step": 5729
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 5729,
      "training_step_time": 10.201451301574707
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 3.140737295150757,
      "step": 5729
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 5729,
      "training_step_time": 9.912095785140991
    },
    {
      "epoch": 2.1854400634765625e-05,
      "model_forward_time": 3.1428041458129883,
      "step": 5729
    },
    {
      "epoch": 2.1854400634765625e-05,
      "step": 5729,
      "training_step_time": 10.023584842681885
    },
    {
      "epoch": 2.185821533203125e-05,
      "grad_norm": 0.6638956665992737,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 5730
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 3.1640260219573975,
      "step": 5730
    },
    {
      "dynamics_loss_avg": 0.06050690487027168,
      "epoch": 2.185821533203125e-05,
      "step": 5730
    },
    {
      "action_loss_avg": 0.024014505557715893,
      "epoch": 2.185821533203125e-05,
      "step": 5730
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 5730,
      "training_step_time": 9.94481086730957
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 3.1465091705322266,
      "step": 5730
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 5730,
      "training_step_time": 9.92518138885498
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 3.40686297416687,
      "step": 5730
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 5730,
      "training_step_time": 10.180795669555664
    },
    {
      "epoch": 2.185821533203125e-05,
      "model_forward_time": 3.147101879119873,
      "step": 5730
    },
    {
      "epoch": 2.185821533203125e-05,
      "step": 5730,
      "training_step_time": 10.326845407485962
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 3.189657688140869,
      "step": 5731
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 5731,
      "training_step_time": 9.963788747787476
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 3.1360998153686523,
      "step": 5731
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 5731,
      "training_step_time": 9.910130023956299
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 3.144540786743164,
      "step": 5731
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 5731,
      "training_step_time": 9.920024633407593
    },
    {
      "epoch": 2.1862030029296875e-05,
      "model_forward_time": 3.4238152503967285,
      "step": 5731
    },
    {
      "epoch": 2.1862030029296875e-05,
      "step": 5731,
      "training_step_time": 10.285882949829102
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 3.181385040283203,
      "step": 5732
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 5732,
      "training_step_time": 9.956612825393677
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 3.1383657455444336,
      "step": 5732
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 5732,
      "training_step_time": 9.9073007106781
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 3.1396536827087402,
      "step": 5732
    },
    {
      "dynamics_loss_avg": 0.06021798066794872,
      "epoch": 2.18658447265625e-05,
      "step": 5732
    },
    {
      "action_loss_avg": 0.021676924731582402,
      "epoch": 2.18658447265625e-05,
      "step": 5732
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 5732,
      "training_step_time": 9.908507108688354
    },
    {
      "epoch": 2.18658447265625e-05,
      "model_forward_time": 3.139399766921997,
      "step": 5732
    },
    {
      "epoch": 2.18658447265625e-05,
      "step": 5732,
      "training_step_time": 10.281371593475342
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 3.4605257511138916,
      "step": 5733
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 5733,
      "training_step_time": 10.231220006942749
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 3.1389660835266113,
      "step": 5733
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 5733,
      "training_step_time": 9.905577421188354
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 3.138817071914673,
      "step": 5733
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 5733,
      "training_step_time": 9.905935287475586
    },
    {
      "epoch": 2.1869659423828124e-05,
      "model_forward_time": 3.143320083618164,
      "step": 5733
    },
    {
      "epoch": 2.1869659423828124e-05,
      "step": 5733,
      "training_step_time": 10.276241064071655
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 3.17504620552063,
      "step": 5734
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 5734,
      "training_step_time": 9.946501970291138
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 3.4186816215515137,
      "step": 5734
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 5734,
      "training_step_time": 10.191700458526611
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 3.1400938034057617,
      "step": 5734
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 5734,
      "training_step_time": 9.912976503372192
    },
    {
      "epoch": 2.187347412109375e-05,
      "model_forward_time": 3.137619733810425,
      "step": 5734
    },
    {
      "epoch": 2.187347412109375e-05,
      "step": 5734,
      "training_step_time": 10.010117530822754
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 3.1572420597076416,
      "step": 5735
    },
    {
      "dynamics_loss_avg": 0.05584066472947598,
      "epoch": 2.1877288818359376e-05,
      "step": 5735
    },
    {
      "action_loss_avg": 0.020669105276465417,
      "epoch": 2.1877288818359376e-05,
      "step": 5735
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 5735,
      "training_step_time": 9.936465501785278
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 3.1406502723693848,
      "step": 5735
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 5735,
      "training_step_time": 9.912456750869751
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 3.4487648010253906,
      "step": 5735
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 5735,
      "training_step_time": 10.22722578048706
    },
    {
      "epoch": 2.1877288818359376e-05,
      "model_forward_time": 3.141923189163208,
      "step": 5735
    },
    {
      "epoch": 2.1877288818359376e-05,
      "step": 5735,
      "training_step_time": 10.267802715301514
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 3.167484998703003,
      "step": 5736
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 5736,
      "training_step_time": 9.938799381256104
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 3.145192861557007,
      "step": 5736
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 5736,
      "training_step_time": 9.919275045394897
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 2.438688039779663,
      "step": 5736
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 5736,
      "training_step_time": 7.469425916671753
    },
    {
      "epoch": 2.1881103515625e-05,
      "model_forward_time": 3.4247918128967285,
      "step": 5736
    },
    {
      "epoch": 2.1881103515625e-05,
      "step": 5736,
      "training_step_time": 10.289827346801758
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 3.162010669708252,
      "step": 5737
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 5737,
      "training_step_time": 9.945722579956055
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 3.145298957824707,
      "step": 5737
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 5737,
      "training_step_time": 9.917238235473633
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 3.1402688026428223,
      "step": 5737
    },
    {
      "dynamics_loss_avg": 0.06737694777548313,
      "epoch": 2.1884918212890625e-05,
      "step": 5737
    },
    {
      "action_loss_avg": 0.025344215612858535,
      "epoch": 2.1884918212890625e-05,
      "step": 5737
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 5737,
      "training_step_time": 9.915211915969849
    },
    {
      "epoch": 2.1884918212890625e-05,
      "model_forward_time": 3.1414172649383545,
      "step": 5737
    },
    {
      "epoch": 2.1884918212890625e-05,
      "step": 5737,
      "training_step_time": 10.021129131317139
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 3.467756509780884,
      "step": 5738
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 5738,
      "training_step_time": 10.239289999008179
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 3.140540361404419,
      "step": 5738
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 5738,
      "training_step_time": 9.904818534851074
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 3.142838716506958,
      "step": 5738
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 5738,
      "training_step_time": 9.91289734840393
    },
    {
      "epoch": 2.188873291015625e-05,
      "model_forward_time": 3.138737678527832,
      "step": 5738
    },
    {
      "epoch": 2.188873291015625e-05,
      "step": 5738,
      "training_step_time": 10.293833494186401
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 3.147719383239746,
      "step": 5739
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 5739,
      "training_step_time": 9.914771556854248
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 3.4281129837036133,
      "step": 5739
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 5739,
      "training_step_time": 10.191927909851074
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 3.13960599899292,
      "step": 5739
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 5739,
      "training_step_time": 9.904248714447021
    },
    {
      "epoch": 2.1892547607421875e-05,
      "model_forward_time": 3.1355316638946533,
      "step": 5739
    },
    {
      "epoch": 2.1892547607421875e-05,
      "step": 5739,
      "training_step_time": 10.267902612686157
    },
    {
      "epoch": 2.18963623046875e-05,
      "grad_norm": 0.3871297836303711,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 5740
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 3.1736841201782227,
      "step": 5740
    },
    {
      "dynamics_loss_avg": 0.054937346186488864,
      "epoch": 2.18963623046875e-05,
      "step": 5740
    },
    {
      "action_loss_avg": 0.01671775826252997,
      "epoch": 2.18963623046875e-05,
      "step": 5740
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 5740,
      "training_step_time": 9.939003944396973
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 3.1386377811431885,
      "step": 5740
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 5740,
      "training_step_time": 9.902234077453613
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 3.3997621536254883,
      "step": 5740
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 5740,
      "training_step_time": 10.166882276535034
    },
    {
      "epoch": 2.18963623046875e-05,
      "model_forward_time": 3.1435604095458984,
      "step": 5740
    },
    {
      "epoch": 2.18963623046875e-05,
      "step": 5740,
      "training_step_time": 10.331767082214355
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 3.1408345699310303,
      "step": 5741
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 5741,
      "training_step_time": 9.907172679901123
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 3.135042190551758,
      "step": 5741
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 5741,
      "training_step_time": 9.906425952911377
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 3.135138988494873,
      "step": 5741
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 5741,
      "training_step_time": 9.910502672195435
    },
    {
      "epoch": 2.1900177001953124e-05,
      "model_forward_time": 3.414923667907715,
      "step": 5741
    },
    {
      "epoch": 2.1900177001953124e-05,
      "step": 5741,
      "training_step_time": 10.283932447433472
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 3.1807963848114014,
      "step": 5742
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 5742,
      "training_step_time": 9.962206602096558
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 3.1345911026000977,
      "step": 5742
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 5742,
      "training_step_time": 9.905617952346802
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 3.1363255977630615,
      "step": 5742
    },
    {
      "dynamics_loss_avg": 0.051842847373336554,
      "epoch": 2.190399169921875e-05,
      "step": 5742
    },
    {
      "action_loss_avg": 0.02363123344257474,
      "epoch": 2.190399169921875e-05,
      "step": 5742
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 5742,
      "training_step_time": 9.913946866989136
    },
    {
      "epoch": 2.190399169921875e-05,
      "model_forward_time": 3.130985975265503,
      "step": 5742
    },
    {
      "epoch": 2.190399169921875e-05,
      "step": 5742,
      "training_step_time": 10.018046379089355
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 2.7053990364074707,
      "step": 5743
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 5743,
      "training_step_time": 7.746120452880859
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 3.141404151916504,
      "step": 5743
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 5743,
      "training_step_time": 9.906192779541016
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 3.1405067443847656,
      "step": 5743
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 5743,
      "training_step_time": 9.905433177947998
    },
    {
      "epoch": 2.1907806396484376e-05,
      "model_forward_time": 3.1345088481903076,
      "step": 5743
    },
    {
      "epoch": 2.1907806396484376e-05,
      "step": 5743,
      "training_step_time": 10.01701807975769
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 3.1535756587982178,
      "step": 5744
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 5744,
      "training_step_time": 9.93028473854065
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 3.4485275745391846,
      "step": 5744
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 5744,
      "training_step_time": 10.217269659042358
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 3.148181200027466,
      "step": 5744
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 5744,
      "training_step_time": 9.929833173751831
    },
    {
      "epoch": 2.191162109375e-05,
      "model_forward_time": 3.1436450481414795,
      "step": 5744
    },
    {
      "epoch": 2.191162109375e-05,
      "step": 5744,
      "training_step_time": 10.309277057647705
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 3.1869256496429443,
      "step": 5745
    },
    {
      "dynamics_loss_avg": 0.04833784764632583,
      "epoch": 2.1915435791015625e-05,
      "step": 5745
    },
    {
      "action_loss_avg": 0.02100889743305743,
      "epoch": 2.1915435791015625e-05,
      "step": 5745
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 5745,
      "training_step_time": 9.966567516326904
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 3.1497044563293457,
      "step": 5745
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 5745,
      "training_step_time": 9.927500009536743
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 3.1468517780303955,
      "step": 5745
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 5745,
      "training_step_time": 10.247124671936035
    },
    {
      "epoch": 2.1915435791015625e-05,
      "model_forward_time": 3.1494078636169434,
      "step": 5745
    },
    {
      "epoch": 2.1915435791015625e-05,
      "step": 5745,
      "training_step_time": 10.321316957473755
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 3.1573305130004883,
      "step": 5746
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 5746,
      "training_step_time": 9.936600685119629
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 3.1451618671417236,
      "step": 5746
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 5746,
      "training_step_time": 9.914581060409546
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 3.1405467987060547,
      "step": 5746
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 5746,
      "training_step_time": 9.910396099090576
    },
    {
      "epoch": 2.191925048828125e-05,
      "model_forward_time": 3.1412851810455322,
      "step": 5746
    },
    {
      "epoch": 2.191925048828125e-05,
      "step": 5746,
      "training_step_time": 10.294234991073608
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 3.1611130237579346,
      "step": 5747
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 5747,
      "training_step_time": 9.935100078582764
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 3.147125482559204,
      "step": 5747
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 5747,
      "training_step_time": 9.915975570678711
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 3.139045238494873,
      "step": 5747
    },
    {
      "dynamics_loss_avg": 0.060596195980906487,
      "epoch": 2.1923065185546875e-05,
      "step": 5747
    },
    {
      "action_loss_avg": 0.0222651201300323,
      "epoch": 2.1923065185546875e-05,
      "step": 5747
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 5747,
      "training_step_time": 9.909025192260742
    },
    {
      "epoch": 2.1923065185546875e-05,
      "model_forward_time": 3.142764091491699,
      "step": 5747
    },
    {
      "epoch": 2.1923065185546875e-05,
      "step": 5747,
      "training_step_time": 10.018444538116455
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 3.149392604827881,
      "step": 5748
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 5748,
      "training_step_time": 10.24074125289917
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 3.1420016288757324,
      "step": 5748
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 5748,
      "training_step_time": 9.91368293762207
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 3.1377675533294678,
      "step": 5748
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 5748,
      "training_step_time": 9.90412449836731
    },
    {
      "epoch": 2.19268798828125e-05,
      "model_forward_time": 3.1350412368774414,
      "step": 5748
    },
    {
      "epoch": 2.19268798828125e-05,
      "step": 5748,
      "training_step_time": 10.004737615585327
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 3.224353551864624,
      "step": 5749
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 5749,
      "training_step_time": 9.995035886764526
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 3.4265458583831787,
      "step": 5749
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 5749,
      "training_step_time": 10.19108533859253
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 3.1350831985473633,
      "step": 5749
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 5749,
      "training_step_time": 9.90104055404663
    },
    {
      "epoch": 2.1930694580078124e-05,
      "model_forward_time": 3.1422059535980225,
      "step": 5749
    },
    {
      "epoch": 2.1930694580078124e-05,
      "step": 5749,
      "training_step_time": 10.284753799438477
    },
    {
      "epoch": 2.193450927734375e-05,
      "grad_norm": 0.46588334441185,
      "learning_rate": 1e-05,
      "loss": 0.0889,
      "step": 5750
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 3.156649589538574,
      "step": 5750
    },
    {
      "dynamics_loss_avg": 0.0623982734978199,
      "epoch": 2.193450927734375e-05,
      "step": 5750
    },
    {
      "action_loss_avg": 0.02261264491826296,
      "epoch": 2.193450927734375e-05,
      "step": 5750
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 5750,
      "training_step_time": 9.929116487503052
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 3.1374082565307617,
      "step": 5750
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 5750,
      "training_step_time": 9.909777879714966
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 3.1402859687805176,
      "step": 5750
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 5750,
      "training_step_time": 10.211186170578003
    },
    {
      "epoch": 2.193450927734375e-05,
      "model_forward_time": 2.4370524883270264,
      "step": 5750
    },
    {
      "epoch": 2.193450927734375e-05,
      "step": 5750,
      "training_step_time": 10.29590106010437
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 3.1571171283721924,
      "step": 5751
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 5751,
      "training_step_time": 9.927666664123535
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 3.1373581886291504,
      "step": 5751
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 5751,
      "training_step_time": 9.908697128295898
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 3.143348217010498,
      "step": 5751
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 5751,
      "training_step_time": 9.91227912902832
    },
    {
      "epoch": 2.1938323974609376e-05,
      "model_forward_time": 2.437286138534546,
      "step": 5751
    },
    {
      "epoch": 2.1938323974609376e-05,
      "step": 5751,
      "training_step_time": 10.029453039169312
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 3.155634880065918,
      "step": 5752
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 5752,
      "training_step_time": 9.924439430236816
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 3.140089273452759,
      "step": 5752
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 5752,
      "training_step_time": 9.900576829910278
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 3.1435577869415283,
      "step": 5752
    },
    {
      "dynamics_loss_avg": 0.05122518204152584,
      "epoch": 2.1942138671875e-05,
      "step": 5752
    },
    {
      "action_loss_avg": 0.029790995875373483,
      "epoch": 2.1942138671875e-05,
      "step": 5752
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 5752,
      "training_step_time": 9.910080909729004
    },
    {
      "epoch": 2.1942138671875e-05,
      "model_forward_time": 2.4332022666931152,
      "step": 5752
    },
    {
      "epoch": 2.1942138671875e-05,
      "step": 5752,
      "training_step_time": 10.323556661605835
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 3.1449875831604004,
      "step": 5753
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 5753,
      "training_step_time": 9.909263134002686
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 3.386486768722534,
      "step": 5753
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 5753,
      "training_step_time": 10.147820711135864
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 3.140637159347534,
      "step": 5753
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 5753,
      "training_step_time": 9.903068780899048
    },
    {
      "epoch": 2.1945953369140625e-05,
      "model_forward_time": 3.135732650756836,
      "step": 5753
    },
    {
      "epoch": 2.1945953369140625e-05,
      "step": 5753,
      "training_step_time": 10.306354522705078
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 3.1466948986053467,
      "step": 5754
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 5754,
      "training_step_time": 9.913687705993652
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 3.4085605144500732,
      "step": 5754
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 5754,
      "training_step_time": 10.16710615158081
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 3.1400187015533447,
      "step": 5754
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 5754,
      "training_step_time": 9.901868343353271
    },
    {
      "epoch": 2.194976806640625e-05,
      "model_forward_time": 3.1323089599609375,
      "step": 5754
    },
    {
      "epoch": 2.194976806640625e-05,
      "step": 5754,
      "training_step_time": 10.248124122619629
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 3.169198751449585,
      "step": 5755
    },
    {
      "dynamics_loss_avg": 0.05527394451200962,
      "epoch": 2.1953582763671875e-05,
      "step": 5755
    },
    {
      "action_loss_avg": 0.028324292646721004,
      "epoch": 2.1953582763671875e-05,
      "step": 5755
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 5755,
      "training_step_time": 9.933687210083008
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 3.13491153717041,
      "step": 5755
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 5755,
      "training_step_time": 9.900160074234009
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 3.135338544845581,
      "step": 5755
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 5755,
      "training_step_time": 10.18259048461914
    },
    {
      "epoch": 2.1953582763671875e-05,
      "model_forward_time": 3.1371638774871826,
      "step": 5755
    },
    {
      "epoch": 2.1953582763671875e-05,
      "step": 5755,
      "training_step_time": 10.304551601409912
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 2.4451417922973633,
      "step": 5756
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 5756,
      "training_step_time": 7.4685444831848145
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 3.137657403945923,
      "step": 5756
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 5756,
      "training_step_time": 9.902770280838013
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 3.1388745307922363,
      "step": 5756
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 5756,
      "training_step_time": 9.905332326889038
    },
    {
      "epoch": 2.19573974609375e-05,
      "model_forward_time": 3.1483590602874756,
      "step": 5756
    },
    {
      "epoch": 2.19573974609375e-05,
      "step": 5756,
      "training_step_time": 10.312240839004517
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 3.164883852005005,
      "step": 5757
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 5757,
      "training_step_time": 9.939976692199707
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 3.1489672660827637,
      "step": 5757
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 5757,
      "training_step_time": 9.919369459152222
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 3.1366751194000244,
      "step": 5757
    },
    {
      "dynamics_loss_avg": 0.052662523835897444,
      "epoch": 2.1961212158203124e-05,
      "step": 5757
    },
    {
      "action_loss_avg": 0.018783801048994065,
      "epoch": 2.1961212158203124e-05,
      "step": 5757
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 5757,
      "training_step_time": 9.902791261672974
    },
    {
      "epoch": 2.1961212158203124e-05,
      "model_forward_time": 2.4304862022399902,
      "step": 5757
    },
    {
      "epoch": 2.1961212158203124e-05,
      "step": 5757,
      "training_step_time": 10.30929708480835
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 3.1742734909057617,
      "step": 5758
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 5758,
      "training_step_time": 9.937555074691772
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 2.020019054412842,
      "step": 5758
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 5758,
      "training_step_time": 5.470996618270874
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 3.125854730606079,
      "step": 5758
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 5758,
      "training_step_time": 9.875715732574463
    },
    {
      "epoch": 2.196502685546875e-05,
      "model_forward_time": 3.1287293434143066,
      "step": 5758
    },
    {
      "epoch": 2.196502685546875e-05,
      "step": 5758,
      "training_step_time": 10.035989761352539
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 3.1423189640045166,
      "step": 5759
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 5759,
      "training_step_time": 9.894327640533447
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 3.1275105476379395,
      "step": 5759
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 5759,
      "training_step_time": 9.882414817810059
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 3.3931727409362793,
      "step": 5759
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 5759,
      "training_step_time": 10.151098489761353
    },
    {
      "epoch": 2.1968841552734376e-05,
      "model_forward_time": 2.4346139430999756,
      "step": 5759
    },
    {
      "epoch": 2.1968841552734376e-05,
      "step": 5759,
      "training_step_time": 10.304671287536621
    },
    {
      "epoch": 2.197265625e-05,
      "grad_norm": 0.5397168397903442,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 5760
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 2.4557251930236816,
      "step": 5760
    },
    {
      "dynamics_loss_avg": 0.06080726236104965,
      "epoch": 2.197265625e-05,
      "step": 5760
    },
    {
      "action_loss_avg": 0.032145375944674016,
      "epoch": 2.197265625e-05,
      "step": 5760
    },
    {
      "epoch": 2.197265625e-05,
      "step": 5760,
      "training_step_time": 7.472782611846924
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 2.429654359817505,
      "step": 5760
    },
    {
      "epoch": 2.197265625e-05,
      "step": 5760,
      "training_step_time": 7.446750164031982
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 3.132620096206665,
      "step": 5760
    },
    {
      "epoch": 2.197265625e-05,
      "step": 5760,
      "training_step_time": 9.892124652862549
    },
    {
      "epoch": 2.197265625e-05,
      "model_forward_time": 3.4227468967437744,
      "step": 5760
    },
    {
      "epoch": 2.197265625e-05,
      "step": 5760,
      "training_step_time": 10.347827434539795
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 2.445258378982544,
      "step": 5761
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 5761,
      "training_step_time": 7.46890926361084
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 3.130880117416382,
      "step": 5761
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 5761,
      "training_step_time": 9.892771005630493
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 3.1376376152038574,
      "step": 5761
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 5761,
      "training_step_time": 9.900696277618408
    },
    {
      "epoch": 2.1976470947265625e-05,
      "model_forward_time": 3.134902238845825,
      "step": 5761
    },
    {
      "epoch": 2.1976470947265625e-05,
      "step": 5761,
      "training_step_time": 10.120988607406616
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 3.4437899589538574,
      "step": 5762
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 5762,
      "training_step_time": 10.213135004043579
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 3.1276068687438965,
      "step": 5762
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 5762,
      "training_step_time": 9.894598960876465
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 3.1285860538482666,
      "step": 5762
    },
    {
      "dynamics_loss_avg": 0.06571405977010727,
      "epoch": 2.198028564453125e-05,
      "step": 5762
    },
    {
      "action_loss_avg": 0.027311115339398385,
      "epoch": 2.198028564453125e-05,
      "step": 5762
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 5762,
      "training_step_time": 9.89283275604248
    },
    {
      "epoch": 2.198028564453125e-05,
      "model_forward_time": 3.1275620460510254,
      "step": 5762
    },
    {
      "epoch": 2.198028564453125e-05,
      "step": 5762,
      "training_step_time": 10.30541443824768
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 3.1445472240448,
      "step": 5763
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 5763,
      "training_step_time": 9.908802270889282
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 3.376278877258301,
      "step": 5763
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 5763,
      "training_step_time": 10.135082006454468
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 3.129668951034546,
      "step": 5763
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 5763,
      "training_step_time": 9.890174627304077
    },
    {
      "epoch": 2.1984100341796875e-05,
      "model_forward_time": 2.439260959625244,
      "step": 5763
    },
    {
      "epoch": 2.1984100341796875e-05,
      "step": 5763,
      "training_step_time": 10.010329961776733
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 3.1603477001190186,
      "step": 5764
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 5764,
      "training_step_time": 9.924472093582153
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 3.132131338119507,
      "step": 5764
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 5764,
      "training_step_time": 9.91093921661377
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 3.4024341106414795,
      "step": 5764
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 5764,
      "training_step_time": 10.170232057571411
    },
    {
      "epoch": 2.19879150390625e-05,
      "model_forward_time": 3.131227493286133,
      "step": 5764
    },
    {
      "epoch": 2.19879150390625e-05,
      "step": 5764,
      "training_step_time": 10.021749496459961
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 3.1512904167175293,
      "step": 5765
    },
    {
      "dynamics_loss_avg": 0.058145546540617946,
      "epoch": 2.1991729736328124e-05,
      "step": 5765
    },
    {
      "action_loss_avg": 0.02505324063822627,
      "epoch": 2.1991729736328124e-05,
      "step": 5765
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 5765,
      "training_step_time": 9.919814825057983
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 3.1292271614074707,
      "step": 5765
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 5765,
      "training_step_time": 9.89196515083313
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 3.1312246322631836,
      "step": 5765
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 5765,
      "training_step_time": 9.89469027519226
    },
    {
      "epoch": 2.1991729736328124e-05,
      "model_forward_time": 2.7257559299468994,
      "step": 5765
    },
    {
      "epoch": 2.1991729736328124e-05,
      "step": 5765,
      "training_step_time": 10.336012363433838
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 3.1523218154907227,
      "step": 5766
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 5766,
      "training_step_time": 9.920676469802856
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 3.134519100189209,
      "step": 5766
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 5766,
      "training_step_time": 9.900665044784546
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 3.1328840255737305,
      "step": 5766
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 5766,
      "training_step_time": 9.90636134147644
    },
    {
      "epoch": 2.199554443359375e-05,
      "model_forward_time": 2.4276251792907715,
      "step": 5766
    },
    {
      "epoch": 2.199554443359375e-05,
      "step": 5766,
      "training_step_time": 10.31048583984375
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 3.429462432861328,
      "step": 5767
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 5767,
      "training_step_time": 10.188823938369751
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 3.1288022994995117,
      "step": 5767
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 5767,
      "training_step_time": 9.889049530029297
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 3.127182722091675,
      "step": 5767
    },
    {
      "dynamics_loss_avg": 0.06047971136868,
      "epoch": 2.1999359130859376e-05,
      "step": 5767
    },
    {
      "action_loss_avg": 0.025406602956354617,
      "epoch": 2.1999359130859376e-05,
      "step": 5767
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 5767,
      "training_step_time": 9.883286476135254
    },
    {
      "epoch": 2.1999359130859376e-05,
      "model_forward_time": 3.1297967433929443,
      "step": 5767
    },
    {
      "epoch": 2.1999359130859376e-05,
      "step": 5767,
      "training_step_time": 10.228646993637085
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 2.4696285724639893,
      "step": 5768
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 5768,
      "training_step_time": 7.4857704639434814
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 3.414203405380249,
      "step": 5768
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 5768,
      "training_step_time": 10.174368381500244
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 3.1259355545043945,
      "step": 5768
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 5768,
      "training_step_time": 9.88028359413147
    },
    {
      "epoch": 2.2003173828125e-05,
      "model_forward_time": 3.1334927082061768,
      "step": 5768
    },
    {
      "epoch": 2.2003173828125e-05,
      "step": 5768,
      "training_step_time": 10.294867753982544
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 3.1473312377929688,
      "step": 5769
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 5769,
      "training_step_time": 9.908440828323364
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 3.1330270767211914,
      "step": 5769
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 5769,
      "training_step_time": 9.895769834518433
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 2.711533546447754,
      "step": 5769
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 5769,
      "training_step_time": 7.729661464691162
    },
    {
      "epoch": 2.2006988525390625e-05,
      "model_forward_time": 3.1323301792144775,
      "step": 5769
    },
    {
      "epoch": 2.2006988525390625e-05,
      "step": 5769,
      "training_step_time": 10.012640714645386
    },
    {
      "epoch": 2.201080322265625e-05,
      "grad_norm": 0.6300995945930481,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 5770
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 3.1700539588928223,
      "step": 5770
    },
    {
      "dynamics_loss_avg": 0.06748362183570862,
      "epoch": 2.201080322265625e-05,
      "step": 5770
    },
    {
      "action_loss_avg": 0.03019356671720743,
      "epoch": 2.201080322265625e-05,
      "step": 5770
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 5770,
      "training_step_time": 9.934042692184448
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 3.1367647647857666,
      "step": 5770
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 5770,
      "training_step_time": 9.900718688964844
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 3.13885235786438,
      "step": 5770
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 5770,
      "training_step_time": 9.901992082595825
    },
    {
      "epoch": 2.201080322265625e-05,
      "model_forward_time": 3.4270544052124023,
      "step": 5770
    },
    {
      "epoch": 2.201080322265625e-05,
      "step": 5770,
      "training_step_time": 10.289582014083862
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 3.1474077701568604,
      "step": 5771
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 5771,
      "training_step_time": 9.916501760482788
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 2.4311742782592773,
      "step": 5771
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 5771,
      "training_step_time": 7.454419374465942
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 2.434715986251831,
      "step": 5771
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 5771,
      "training_step_time": 7.456228256225586
    },
    {
      "epoch": 2.2014617919921875e-05,
      "model_forward_time": 3.1314663887023926,
      "step": 5771
    },
    {
      "epoch": 2.2014617919921875e-05,
      "step": 5771,
      "training_step_time": 10.302196264266968
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 3.422288656234741,
      "step": 5772
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 5772,
      "training_step_time": 10.18883728981018
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 3.1375725269317627,
      "step": 5772
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 5772,
      "training_step_time": 9.901029348373413
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 3.1333553791046143,
      "step": 5772
    },
    {
      "dynamics_loss_avg": 0.05842733308672905,
      "epoch": 2.20184326171875e-05,
      "step": 5772
    },
    {
      "action_loss_avg": 0.028092192602343857,
      "epoch": 2.20184326171875e-05,
      "step": 5772
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 5772,
      "training_step_time": 9.895896911621094
    },
    {
      "epoch": 2.20184326171875e-05,
      "model_forward_time": 3.130733013153076,
      "step": 5772
    },
    {
      "epoch": 2.20184326171875e-05,
      "step": 5772,
      "training_step_time": 10.011488437652588
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 3.144695997238159,
      "step": 5773
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 5773,
      "training_step_time": 9.910460948944092
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 3.3797988891601562,
      "step": 5773
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 5773,
      "training_step_time": 10.146676301956177
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 2.4364566802978516,
      "step": 5773
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 5773,
      "training_step_time": 7.464951276779175
    },
    {
      "epoch": 2.2022247314453124e-05,
      "model_forward_time": 2.4273602962493896,
      "step": 5773
    },
    {
      "epoch": 2.2022247314453124e-05,
      "step": 5773,
      "training_step_time": 10.324304103851318
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 2.474931001663208,
      "step": 5774
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 5774,
      "training_step_time": 7.502598285675049
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 3.140005350112915,
      "step": 5774
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 5774,
      "training_step_time": 9.903635263442993
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 3.410012722015381,
      "step": 5774
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 5774,
      "training_step_time": 10.168534755706787
    },
    {
      "epoch": 2.202606201171875e-05,
      "model_forward_time": 3.125638723373413,
      "step": 5774
    },
    {
      "epoch": 2.202606201171875e-05,
      "step": 5774,
      "training_step_time": 10.250462770462036
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 3.1502225399017334,
      "step": 5775
    },
    {
      "dynamics_loss_avg": 0.057065197825431825,
      "epoch": 2.2029876708984376e-05,
      "step": 5775
    },
    {
      "action_loss_avg": 0.023285625968128443,
      "epoch": 2.2029876708984376e-05,
      "step": 5775
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 5775,
      "training_step_time": 9.920192003250122
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 3.1359047889709473,
      "step": 5775
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 5775,
      "training_step_time": 9.903070449829102
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 2.436800479888916,
      "step": 5775
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 5775,
      "training_step_time": 7.458928346633911
    },
    {
      "epoch": 2.2029876708984376e-05,
      "model_forward_time": 3.419116735458374,
      "step": 5775
    },
    {
      "epoch": 2.2029876708984376e-05,
      "step": 5775,
      "training_step_time": 10.283437252044678
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 3.1712121963500977,
      "step": 5776
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 5776,
      "training_step_time": 9.934680223464966
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 3.139465808868408,
      "step": 5776
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 5776,
      "training_step_time": 9.911446332931519
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 3.138934373855591,
      "step": 5776
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 5776,
      "training_step_time": 9.910389423370361
    },
    {
      "epoch": 2.203369140625e-05,
      "model_forward_time": 3.141352891921997,
      "step": 5776
    },
    {
      "epoch": 2.203369140625e-05,
      "step": 5776,
      "training_step_time": 10.280280590057373
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 3.409646511077881,
      "step": 5777
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 5777,
      "training_step_time": 10.182191371917725
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 3.1411848068237305,
      "step": 5777
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 5777,
      "training_step_time": 9.910090208053589
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 3.137259006500244,
      "step": 5777
    },
    {
      "dynamics_loss_avg": 0.0645141938701272,
      "epoch": 2.2037506103515626e-05,
      "step": 5777
    },
    {
      "action_loss_avg": 0.02927437494508922,
      "epoch": 2.2037506103515626e-05,
      "step": 5777
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 5777,
      "training_step_time": 9.908823251724243
    },
    {
      "epoch": 2.2037506103515626e-05,
      "model_forward_time": 3.1448276042938232,
      "step": 5777
    },
    {
      "epoch": 2.2037506103515626e-05,
      "step": 5777,
      "training_step_time": 10.2991042137146
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 2.474928140640259,
      "step": 5778
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 5778,
      "training_step_time": 7.504650354385376
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 2.6926157474517822,
      "step": 5778
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 5778,
      "training_step_time": 7.720336198806763
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 3.1347556114196777,
      "step": 5778
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 5778,
      "training_step_time": 9.900970697402954
    },
    {
      "epoch": 2.204132080078125e-05,
      "model_forward_time": 3.1404871940612793,
      "step": 5778
    },
    {
      "epoch": 2.204132080078125e-05,
      "step": 5778,
      "training_step_time": 10.272661924362183
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 3.148176670074463,
      "step": 5779
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 5779,
      "training_step_time": 9.924281597137451
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 3.1340534687042236,
      "step": 5779
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 5779,
      "training_step_time": 9.898800134658813
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 2.435849666595459,
      "step": 5779
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 5779,
      "training_step_time": 7.724686622619629
    },
    {
      "epoch": 2.2045135498046875e-05,
      "model_forward_time": 3.13193416595459,
      "step": 5779
    },
    {
      "epoch": 2.2045135498046875e-05,
      "step": 5779,
      "training_step_time": 10.283413171768188
    },
    {
      "epoch": 2.20489501953125e-05,
      "grad_norm": 0.574698269367218,
      "learning_rate": 1e-05,
      "loss": 0.0854,
      "step": 5780
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 3.171581268310547,
      "step": 5780
    },
    {
      "dynamics_loss_avg": 0.05259250588715077,
      "epoch": 2.20489501953125e-05,
      "step": 5780
    },
    {
      "action_loss_avg": 0.02107750568538904,
      "epoch": 2.20489501953125e-05,
      "step": 5780
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 5780,
      "training_step_time": 9.93862271308899
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 3.1331558227539062,
      "step": 5780
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 5780,
      "training_step_time": 9.894879579544067
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 3.1352782249450684,
      "step": 5780
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 5780,
      "training_step_time": 9.897518634796143
    },
    {
      "epoch": 2.20489501953125e-05,
      "model_forward_time": 3.4129722118377686,
      "step": 5780
    },
    {
      "epoch": 2.20489501953125e-05,
      "step": 5780,
      "training_step_time": 10.294296979904175
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 3.16646146774292,
      "step": 5781
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 5781,
      "training_step_time": 9.928021907806396
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 2.429755210876465,
      "step": 5781
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 5781,
      "training_step_time": 7.448185443878174
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 3.1347367763519287,
      "step": 5781
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 5781,
      "training_step_time": 9.894221067428589
    },
    {
      "epoch": 2.2052764892578124e-05,
      "model_forward_time": 3.1316237449645996,
      "step": 5781
    },
    {
      "epoch": 2.2052764892578124e-05,
      "step": 5781,
      "training_step_time": 10.236326694488525
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 2.461799144744873,
      "step": 5782
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 5782,
      "training_step_time": 7.480844736099243
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 3.3728067874908447,
      "step": 5782
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 5782,
      "training_step_time": 10.133424997329712
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 3.132115364074707,
      "step": 5782
    },
    {
      "dynamics_loss_avg": 0.06152381431311369,
      "epoch": 2.205657958984375e-05,
      "step": 5782
    },
    {
      "action_loss_avg": 0.029317894577980043,
      "epoch": 2.205657958984375e-05,
      "step": 5782
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 5782,
      "training_step_time": 9.900511503219604
    },
    {
      "epoch": 2.205657958984375e-05,
      "model_forward_time": 3.132850170135498,
      "step": 5782
    },
    {
      "epoch": 2.205657958984375e-05,
      "step": 5782,
      "training_step_time": 10.28601336479187
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 3.1405296325683594,
      "step": 5783
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 5783,
      "training_step_time": 9.907545566558838
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 3.3964474201202393,
      "step": 5783
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 5783,
      "training_step_time": 10.167585134506226
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 2.43384051322937,
      "step": 5783
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 5783,
      "training_step_time": 7.459510803222656
    },
    {
      "epoch": 2.2060394287109376e-05,
      "model_forward_time": 3.1340701580047607,
      "step": 5783
    },
    {
      "epoch": 2.2060394287109376e-05,
      "step": 5783,
      "training_step_time": 10.293610334396362
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 3.1807682514190674,
      "step": 5784
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 5784,
      "training_step_time": 9.953330516815186
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 3.1372287273406982,
      "step": 5784
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 5784,
      "training_step_time": 9.9095139503479
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 3.1362931728363037,
      "step": 5784
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 5784,
      "training_step_time": 9.905543565750122
    },
    {
      "epoch": 2.2064208984375e-05,
      "model_forward_time": 3.3923356533050537,
      "step": 5784
    },
    {
      "epoch": 2.2064208984375e-05,
      "step": 5784,
      "training_step_time": 10.262490034103394
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 3.185286283493042,
      "step": 5785
    },
    {
      "dynamics_loss_avg": 0.058501063659787175,
      "epoch": 2.2068023681640626e-05,
      "step": 5785
    },
    {
      "action_loss_avg": 0.02862659953534603,
      "epoch": 2.2068023681640626e-05,
      "step": 5785
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 5785,
      "training_step_time": 9.962585926055908
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 3.1381640434265137,
      "step": 5785
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 5785,
      "training_step_time": 9.911811828613281
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 3.139876127243042,
      "step": 5785
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 5785,
      "training_step_time": 9.91274356842041
    },
    {
      "epoch": 2.2068023681640626e-05,
      "model_forward_time": 3.1381094455718994,
      "step": 5785
    },
    {
      "epoch": 2.2068023681640626e-05,
      "step": 5785,
      "training_step_time": 10.28798794746399
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 3.1737520694732666,
      "step": 5786
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 5786,
      "training_step_time": 9.94990086555481
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 3.140796661376953,
      "step": 5786
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 5786,
      "training_step_time": 9.915710687637329
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 3.138181447982788,
      "step": 5786
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 5786,
      "training_step_time": 9.91585087776184
    },
    {
      "epoch": 2.207183837890625e-05,
      "model_forward_time": 3.1425888538360596,
      "step": 5786
    },
    {
      "epoch": 2.207183837890625e-05,
      "step": 5786,
      "training_step_time": 10.286745309829712
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 3.439750909805298,
      "step": 5787
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 5787,
      "training_step_time": 10.212616682052612
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 3.1322426795959473,
      "step": 5787
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 5787,
      "training_step_time": 9.900183200836182
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 3.1285715103149414,
      "step": 5787
    },
    {
      "dynamics_loss_avg": 0.05603852067142725,
      "epoch": 2.2075653076171875e-05,
      "step": 5787
    },
    {
      "action_loss_avg": 0.024219251610338687,
      "epoch": 2.2075653076171875e-05,
      "step": 5787
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 5787,
      "training_step_time": 9.904431581497192
    },
    {
      "epoch": 2.2075653076171875e-05,
      "model_forward_time": 3.132791757583618,
      "step": 5787
    },
    {
      "epoch": 2.2075653076171875e-05,
      "step": 5787,
      "training_step_time": 10.035014152526855
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 3.16951584815979,
      "step": 5788
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 5788,
      "training_step_time": 9.944281816482544
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 3.1307742595672607,
      "step": 5788
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 5788,
      "training_step_time": 10.174870491027832
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 3.130500555038452,
      "step": 5788
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 5788,
      "training_step_time": 9.899054765701294
    },
    {
      "epoch": 2.20794677734375e-05,
      "model_forward_time": 3.136523723602295,
      "step": 5788
    },
    {
      "epoch": 2.20794677734375e-05,
      "step": 5788,
      "training_step_time": 10.283921957015991
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 3.1793971061706543,
      "step": 5789
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 5789,
      "training_step_time": 9.951260089874268
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 3.130211353302002,
      "step": 5789
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 5789,
      "training_step_time": 9.898857593536377
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 3.3925423622131348,
      "step": 5789
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 5789,
      "training_step_time": 10.161287307739258
    },
    {
      "epoch": 2.2083282470703124e-05,
      "model_forward_time": 3.1422810554504395,
      "step": 5789
    },
    {
      "epoch": 2.2083282470703124e-05,
      "step": 5789,
      "training_step_time": 10.312832832336426
    },
    {
      "epoch": 2.208709716796875e-05,
      "grad_norm": 0.6377077698707581,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 5790
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 3.154459238052368,
      "step": 5790
    },
    {
      "dynamics_loss_avg": 0.06362666450440883,
      "epoch": 2.208709716796875e-05,
      "step": 5790
    },
    {
      "action_loss_avg": 0.02961030788719654,
      "epoch": 2.208709716796875e-05,
      "step": 5790
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 5790,
      "training_step_time": 9.923938751220703
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 3.1351537704467773,
      "step": 5790
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 5790,
      "training_step_time": 9.90859341621399
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 3.1425235271453857,
      "step": 5790
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 5790,
      "training_step_time": 9.9136803150177
    },
    {
      "epoch": 2.208709716796875e-05,
      "model_forward_time": 3.1372313499450684,
      "step": 5790
    },
    {
      "epoch": 2.208709716796875e-05,
      "step": 5790,
      "training_step_time": 10.288845539093018
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 3.158613681793213,
      "step": 5791
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 5791,
      "training_step_time": 9.930196285247803
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 3.134042978286743,
      "step": 5791
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 5791,
      "training_step_time": 9.905226707458496
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 3.1351563930511475,
      "step": 5791
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 5791,
      "training_step_time": 9.907509326934814
    },
    {
      "epoch": 2.2090911865234376e-05,
      "model_forward_time": 2.4402129650115967,
      "step": 5791
    },
    {
      "epoch": 2.2090911865234376e-05,
      "step": 5791,
      "training_step_time": 10.026329040527344
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 3.4451684951782227,
      "step": 5792
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 5792,
      "training_step_time": 10.215784788131714
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 3.140284776687622,
      "step": 5792
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 5792,
      "training_step_time": 9.914148092269897
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 3.147118091583252,
      "step": 5792
    },
    {
      "dynamics_loss_avg": 0.046753219328820705,
      "epoch": 2.20947265625e-05,
      "step": 5792
    },
    {
      "action_loss_avg": 0.025342382211238145,
      "epoch": 2.20947265625e-05,
      "step": 5792
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 5792,
      "training_step_time": 9.938987016677856
    },
    {
      "epoch": 2.20947265625e-05,
      "model_forward_time": 3.1380527019500732,
      "step": 5792
    },
    {
      "epoch": 2.20947265625e-05,
      "step": 5792,
      "training_step_time": 10.268743991851807
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 2.4749977588653564,
      "step": 5793
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 5793,
      "training_step_time": 7.520227909088135
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 3.1372506618499756,
      "step": 5793
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 5793,
      "training_step_time": 10.181895971298218
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 3.1376430988311768,
      "step": 5793
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 5793,
      "training_step_time": 9.91049575805664
    },
    {
      "epoch": 2.2098541259765626e-05,
      "model_forward_time": 3.134427309036255,
      "step": 5793
    },
    {
      "epoch": 2.2098541259765626e-05,
      "step": 5793,
      "training_step_time": 10.286552906036377
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 3.1584525108337402,
      "step": 5794
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 5794,
      "training_step_time": 9.934981346130371
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 3.1382110118865967,
      "step": 5794
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 5794,
      "training_step_time": 9.913194179534912
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 2.441499948501587,
      "step": 5794
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 5794,
      "training_step_time": 7.759102821350098
    },
    {
      "epoch": 2.210235595703125e-05,
      "model_forward_time": 3.1401424407958984,
      "step": 5794
    },
    {
      "epoch": 2.210235595703125e-05,
      "step": 5794,
      "training_step_time": 10.287873983383179
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 3.165139675140381,
      "step": 5795
    },
    {
      "dynamics_loss_avg": 0.05942755863070488,
      "epoch": 2.2106170654296875e-05,
      "step": 5795
    },
    {
      "action_loss_avg": 0.019309447333216666,
      "epoch": 2.2106170654296875e-05,
      "step": 5795
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 5795,
      "training_step_time": 9.945992946624756
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 3.150150775909424,
      "step": 5795
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 5795,
      "training_step_time": 9.931250810623169
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 3.142183542251587,
      "step": 5795
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 5795,
      "training_step_time": 9.922417163848877
    },
    {
      "epoch": 2.2106170654296875e-05,
      "model_forward_time": 3.4096593856811523,
      "step": 5795
    },
    {
      "epoch": 2.2106170654296875e-05,
      "step": 5795,
      "training_step_time": 10.333542108535767
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 3.181596517562866,
      "step": 5796
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 5796,
      "training_step_time": 9.959458589553833
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 3.141951322555542,
      "step": 5796
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 5796,
      "training_step_time": 9.914722919464111
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 3.1404359340667725,
      "step": 5796
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 5796,
      "training_step_time": 9.911821842193604
    },
    {
      "epoch": 2.21099853515625e-05,
      "model_forward_time": 3.1371710300445557,
      "step": 5796
    },
    {
      "epoch": 2.21099853515625e-05,
      "step": 5796,
      "training_step_time": 10.016754627227783
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 3.150892496109009,
      "step": 5797
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 5797,
      "training_step_time": 10.19717264175415
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 3.1419050693511963,
      "step": 5797
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 5797,
      "training_step_time": 9.913777351379395
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 3.1348021030426025,
      "step": 5797
    },
    {
      "dynamics_loss_avg": 0.05694101303815842,
      "epoch": 2.2113800048828124e-05,
      "step": 5797
    },
    {
      "action_loss_avg": 0.023203375190496443,
      "epoch": 2.2113800048828124e-05,
      "step": 5797
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 5797,
      "training_step_time": 9.905282974243164
    },
    {
      "epoch": 2.2113800048828124e-05,
      "model_forward_time": 3.1391751766204834,
      "step": 5797
    },
    {
      "epoch": 2.2113800048828124e-05,
      "step": 5797,
      "training_step_time": 10.230382442474365
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 3.173367738723755,
      "step": 5798
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 5798,
      "training_step_time": 9.942671298980713
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 3.4211745262145996,
      "step": 5798
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 5798,
      "training_step_time": 10.188104629516602
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 3.137073040008545,
      "step": 5798
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 5798,
      "training_step_time": 9.907568454742432
    },
    {
      "epoch": 2.211761474609375e-05,
      "model_forward_time": 3.1388909816741943,
      "step": 5798
    },
    {
      "epoch": 2.211761474609375e-05,
      "step": 5798,
      "training_step_time": 10.301063299179077
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 3.1855313777923584,
      "step": 5799
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 5799,
      "training_step_time": 9.960384607315063
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 3.14100980758667,
      "step": 5799
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 5799,
      "training_step_time": 9.916965007781982
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 3.1441495418548584,
      "step": 5799
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 5799,
      "training_step_time": 10.189152956008911
    },
    {
      "epoch": 2.2121429443359376e-05,
      "model_forward_time": 3.148172378540039,
      "step": 5799
    },
    {
      "epoch": 2.2121429443359376e-05,
      "step": 5799,
      "training_step_time": 10.299813985824585
    },
    {
      "epoch": 2.2125244140625e-05,
      "grad_norm": 0.5639790892601013,
      "learning_rate": 1e-05,
      "loss": 0.0837,
      "step": 5800
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 3.187786340713501,
      "step": 5800
    },
    {
      "dynamics_loss_avg": 0.06023936811834574,
      "epoch": 2.2125244140625e-05,
      "step": 5800
    },
    {
      "action_loss_avg": 0.03084878418594599,
      "epoch": 2.2125244140625e-05,
      "step": 5800
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 5800,
      "training_step_time": 9.9738929271698
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 3.14370059967041,
      "step": 5800
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 5800,
      "training_step_time": 9.918797969818115
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 3.1494245529174805,
      "step": 5800
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 5800,
      "training_step_time": 9.932915449142456
    },
    {
      "epoch": 2.2125244140625e-05,
      "model_forward_time": 3.4341418743133545,
      "step": 5800
    },
    {
      "epoch": 2.2125244140625e-05,
      "step": 5800,
      "training_step_time": 10.306905269622803
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 3.1645710468292236,
      "step": 5801
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 5801,
      "training_step_time": 9.939208984375
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 3.1393322944641113,
      "step": 5801
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 5801,
      "training_step_time": 9.90925121307373
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 2.437056541442871,
      "step": 5801
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 5801,
      "training_step_time": 7.46695876121521
    },
    {
      "epoch": 2.2129058837890626e-05,
      "model_forward_time": 2.4339144229888916,
      "step": 5801
    },
    {
      "epoch": 2.2129058837890626e-05,
      "step": 5801,
      "training_step_time": 10.02796220779419
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 3.1706604957580566,
      "step": 5802
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 5802,
      "training_step_time": 9.943178176879883
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 3.3770666122436523,
      "step": 5802
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 5802,
      "training_step_time": 10.143372535705566
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 3.1330199241638184,
      "step": 5802
    },
    {
      "dynamics_loss_avg": 0.06925488887354732,
      "epoch": 2.213287353515625e-05,
      "step": 5802
    },
    {
      "action_loss_avg": 0.03144680514233187,
      "epoch": 2.213287353515625e-05,
      "step": 5802
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 5802,
      "training_step_time": 9.899892330169678
    },
    {
      "epoch": 2.213287353515625e-05,
      "model_forward_time": 3.1263811588287354,
      "step": 5802
    },
    {
      "epoch": 2.213287353515625e-05,
      "step": 5802,
      "training_step_time": 10.26694107055664
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 3.1524369716644287,
      "step": 5803
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 5803,
      "training_step_time": 9.929689407348633
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 3.389665126800537,
      "step": 5803
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 5803,
      "training_step_time": 10.15481185913086
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 2.4279978275299072,
      "step": 5803
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 5803,
      "training_step_time": 7.45025110244751
    },
    {
      "epoch": 2.2136688232421875e-05,
      "model_forward_time": 3.130988121032715,
      "step": 5803
    },
    {
      "epoch": 2.2136688232421875e-05,
      "step": 5803,
      "training_step_time": 10.31984281539917
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 3.1408495903015137,
      "step": 5804
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 5804,
      "training_step_time": 9.908538818359375
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 3.137537956237793,
      "step": 5804
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 5804,
      "training_step_time": 9.908093214035034
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 3.134384870529175,
      "step": 5804
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 5804,
      "training_step_time": 9.902914762496948
    },
    {
      "epoch": 2.21405029296875e-05,
      "model_forward_time": 3.378061294555664,
      "step": 5804
    },
    {
      "epoch": 2.21405029296875e-05,
      "step": 5804,
      "training_step_time": 10.245235919952393
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 3.167846918106079,
      "step": 5805
    },
    {
      "dynamics_loss_avg": 0.05960719306021929,
      "epoch": 2.2144317626953124e-05,
      "step": 5805
    },
    {
      "action_loss_avg": 0.02160218358039856,
      "epoch": 2.2144317626953124e-05,
      "step": 5805
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 5805,
      "training_step_time": 9.945488214492798
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 3.1365091800689697,
      "step": 5805
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 5805,
      "training_step_time": 9.907663583755493
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 3.1323916912078857,
      "step": 5805
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 5805,
      "training_step_time": 9.904853820800781
    },
    {
      "epoch": 2.2144317626953124e-05,
      "model_forward_time": 3.141263484954834,
      "step": 5805
    },
    {
      "epoch": 2.2144317626953124e-05,
      "step": 5805,
      "training_step_time": 10.31373119354248
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 3.1611862182617188,
      "step": 5806
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 5806,
      "training_step_time": 9.936642408370972
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 3.13877272605896,
      "step": 5806
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 5806,
      "training_step_time": 9.90765118598938
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 3.132317066192627,
      "step": 5806
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 5806,
      "training_step_time": 9.902974843978882
    },
    {
      "epoch": 2.214813232421875e-05,
      "model_forward_time": 3.133366107940674,
      "step": 5806
    },
    {
      "epoch": 2.214813232421875e-05,
      "step": 5806,
      "training_step_time": 10.037751913070679
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 3.42148756980896,
      "step": 5807
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 5807,
      "training_step_time": 10.19173288345337
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 3.132035255432129,
      "step": 5807
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 5807,
      "training_step_time": 9.896993160247803
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 2.4288878440856934,
      "step": 5807
    },
    {
      "dynamics_loss_avg": 0.06735473852604627,
      "epoch": 2.2151947021484376e-05,
      "step": 5807
    },
    {
      "action_loss_avg": 0.030281612556427716,
      "epoch": 2.2151947021484376e-05,
      "step": 5807
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 5807,
      "training_step_time": 7.450814962387085
    },
    {
      "epoch": 2.2151947021484376e-05,
      "model_forward_time": 3.129697799682617,
      "step": 5807
    },
    {
      "epoch": 2.2151947021484376e-05,
      "step": 5807,
      "training_step_time": 10.237214088439941
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 3.170952796936035,
      "step": 5808
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 5808,
      "training_step_time": 9.940217733383179
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 2.4252543449401855,
      "step": 5808
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 5808,
      "training_step_time": 7.446991443634033
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 3.3750791549682617,
      "step": 5808
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 5808,
      "training_step_time": 10.145947217941284
    },
    {
      "epoch": 2.215576171875e-05,
      "model_forward_time": 3.132028341293335,
      "step": 5808
    },
    {
      "epoch": 2.215576171875e-05,
      "step": 5808,
      "training_step_time": 10.013600826263428
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 3.147519111633301,
      "step": 5809
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 5809,
      "training_step_time": 9.912704467773438
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 3.1236186027526855,
      "step": 5809
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 5809,
      "training_step_time": 9.890476942062378
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 1.7390053272247314,
      "step": 5809
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 5809,
      "training_step_time": 5.1921398639678955
    },
    {
      "epoch": 2.2159576416015626e-05,
      "model_forward_time": 3.4098732471466064,
      "step": 5809
    },
    {
      "epoch": 2.2159576416015626e-05,
      "step": 5809,
      "training_step_time": 10.271878480911255
    },
    {
      "epoch": 2.216339111328125e-05,
      "grad_norm": 0.4738594889640808,
      "learning_rate": 1e-05,
      "loss": 0.0881,
      "step": 5810
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 3.143373489379883,
      "step": 5810
    },
    {
      "dynamics_loss_avg": 0.059383615106344226,
      "epoch": 2.216339111328125e-05,
      "step": 5810
    },
    {
      "action_loss_avg": 0.026259659952484072,
      "epoch": 2.216339111328125e-05,
      "step": 5810
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 5810,
      "training_step_time": 9.91753101348877
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 3.1330502033233643,
      "step": 5810
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 5810,
      "training_step_time": 9.906199932098389
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 2.4316389560699463,
      "step": 5810
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 5810,
      "training_step_time": 7.465159177780151
    },
    {
      "epoch": 2.216339111328125e-05,
      "model_forward_time": 3.1326565742492676,
      "step": 5810
    },
    {
      "epoch": 2.216339111328125e-05,
      "step": 5810,
      "training_step_time": 10.324799060821533
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 3.446913003921509,
      "step": 5811
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 5811,
      "training_step_time": 10.222261905670166
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 3.133502244949341,
      "step": 5811
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 5811,
      "training_step_time": 9.911506175994873
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 3.1382555961608887,
      "step": 5811
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 5811,
      "training_step_time": 9.913465023040771
    },
    {
      "epoch": 2.2167205810546875e-05,
      "model_forward_time": 3.140774726867676,
      "step": 5811
    },
    {
      "epoch": 2.2167205810546875e-05,
      "step": 5811,
      "training_step_time": 10.299193143844604
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 3.161381483078003,
      "step": 5812
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 5812,
      "training_step_time": 9.942518949508667
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 3.385223627090454,
      "step": 5812
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 5812,
      "training_step_time": 10.16389012336731
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 3.1397905349731445,
      "step": 5812
    },
    {
      "dynamics_loss_avg": 0.06391539983451366,
      "epoch": 2.21710205078125e-05,
      "step": 5812
    },
    {
      "action_loss_avg": 0.027390080224722623,
      "epoch": 2.21710205078125e-05,
      "step": 5812
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 5812,
      "training_step_time": 9.923857688903809
    },
    {
      "epoch": 2.21710205078125e-05,
      "model_forward_time": 3.1412620544433594,
      "step": 5812
    },
    {
      "epoch": 2.21710205078125e-05,
      "step": 5812,
      "training_step_time": 10.298691272735596
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 3.1705403327941895,
      "step": 5813
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 5813,
      "training_step_time": 9.95244550704956
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 3.1497962474823,
      "step": 5813
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 5813,
      "training_step_time": 9.929887294769287
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 2.7153728008270264,
      "step": 5813
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 5813,
      "training_step_time": 7.743798017501831
    },
    {
      "epoch": 2.2174835205078124e-05,
      "model_forward_time": 3.14211106300354,
      "step": 5813
    },
    {
      "epoch": 2.2174835205078124e-05,
      "step": 5813,
      "training_step_time": 10.057097673416138
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 2.447044849395752,
      "step": 5814
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 5814,
      "training_step_time": 7.475362300872803
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 3.143676996231079,
      "step": 5814
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 5814,
      "training_step_time": 9.915385484695435
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 3.13795804977417,
      "step": 5814
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 5814,
      "training_step_time": 9.910163879394531
    },
    {
      "epoch": 2.217864990234375e-05,
      "model_forward_time": 3.3915534019470215,
      "step": 5814
    },
    {
      "epoch": 2.217864990234375e-05,
      "step": 5814,
      "training_step_time": 10.260706186294556
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 3.16145920753479,
      "step": 5815
    },
    {
      "dynamics_loss_avg": 0.061886041238904,
      "epoch": 2.2182464599609376e-05,
      "step": 5815
    },
    {
      "action_loss_avg": 0.028429485578089952,
      "epoch": 2.2182464599609376e-05,
      "step": 5815
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 5815,
      "training_step_time": 9.935831546783447
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 2.4333083629608154,
      "step": 5815
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 5815,
      "training_step_time": 7.46035099029541
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 2.437786102294922,
      "step": 5815
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 5815,
      "training_step_time": 7.4697864055633545
    },
    {
      "epoch": 2.2182464599609376e-05,
      "model_forward_time": 3.140179395675659,
      "step": 5815
    },
    {
      "epoch": 2.2182464599609376e-05,
      "step": 5815,
      "training_step_time": 10.261704206466675
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 3.3952250480651855,
      "step": 5816
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 5816,
      "training_step_time": 10.168669700622559
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 3.134896993637085,
      "step": 5816
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 5816,
      "training_step_time": 9.908758640289307
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 3.1309986114501953,
      "step": 5816
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 5816,
      "training_step_time": 9.915403127670288
    },
    {
      "epoch": 2.2186279296875e-05,
      "model_forward_time": 3.1398508548736572,
      "step": 5816
    },
    {
      "epoch": 2.2186279296875e-05,
      "step": 5816,
      "training_step_time": 10.267934799194336
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 3.1465790271759033,
      "step": 5817
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 5817,
      "training_step_time": 9.93138837814331
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 3.3986473083496094,
      "step": 5817
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 5817,
      "training_step_time": 10.177543878555298
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 3.1335623264312744,
      "step": 5817
    },
    {
      "dynamics_loss_avg": 0.05147843463346362,
      "epoch": 2.2190093994140626e-05,
      "step": 5817
    },
    {
      "action_loss_avg": 0.023286540899425744,
      "epoch": 2.2190093994140626e-05,
      "step": 5817
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 5817,
      "training_step_time": 9.916033267974854
    },
    {
      "epoch": 2.2190093994140626e-05,
      "model_forward_time": 3.1349997520446777,
      "step": 5817
    },
    {
      "epoch": 2.2190093994140626e-05,
      "step": 5817,
      "training_step_time": 10.272446632385254
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 3.150848627090454,
      "step": 5818
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 5818,
      "training_step_time": 9.933411598205566
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 3.133669376373291,
      "step": 5818
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 5818,
      "training_step_time": 10.19687819480896
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 2.434523820877075,
      "step": 5818
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 5818,
      "training_step_time": 7.4683778285980225
    },
    {
      "epoch": 2.219390869140625e-05,
      "model_forward_time": 3.1341745853424072,
      "step": 5818
    },
    {
      "epoch": 2.219390869140625e-05,
      "step": 5818,
      "training_step_time": 10.23906660079956
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 3.1495163440704346,
      "step": 5819
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 5819,
      "training_step_time": 9.920698404312134
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 3.1386148929595947,
      "step": 5819
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 5819,
      "training_step_time": 9.918033838272095
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 3.1866281032562256,
      "step": 5819
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 5819,
      "training_step_time": 9.96494174003601
    },
    {
      "epoch": 2.2197723388671875e-05,
      "model_forward_time": 3.4178998470306396,
      "step": 5819
    },
    {
      "epoch": 2.2197723388671875e-05,
      "step": 5819,
      "training_step_time": 10.294060468673706
    },
    {
      "epoch": 2.22015380859375e-05,
      "grad_norm": 0.5670578479766846,
      "learning_rate": 1e-05,
      "loss": 0.0852,
      "step": 5820
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 3.1648285388946533,
      "step": 5820
    },
    {
      "dynamics_loss_avg": 0.04063219875097275,
      "epoch": 2.22015380859375e-05,
      "step": 5820
    },
    {
      "action_loss_avg": 0.020244132773950697,
      "epoch": 2.22015380859375e-05,
      "step": 5820
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 5820,
      "training_step_time": 9.953832387924194
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 3.1371586322784424,
      "step": 5820
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 5820,
      "training_step_time": 9.914029836654663
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 3.1732981204986572,
      "step": 5820
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 5820,
      "training_step_time": 9.947209596633911
    },
    {
      "epoch": 2.22015380859375e-05,
      "model_forward_time": 3.1364848613739014,
      "step": 5820
    },
    {
      "epoch": 2.22015380859375e-05,
      "step": 5820,
      "training_step_time": 10.333574533462524
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 3.1850972175598145,
      "step": 5821
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 5821,
      "training_step_time": 9.964335918426514
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 2.4365739822387695,
      "step": 5821
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 5821,
      "training_step_time": 7.472767114639282
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 3.1318740844726562,
      "step": 5821
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 5821,
      "training_step_time": 9.897588968276978
    },
    {
      "epoch": 2.2205352783203124e-05,
      "model_forward_time": 3.128647565841675,
      "step": 5821
    },
    {
      "epoch": 2.2205352783203124e-05,
      "step": 5821,
      "training_step_time": 10.014286041259766
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 3.172494649887085,
      "step": 5822
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 5822,
      "training_step_time": 9.936729669570923
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 3.4174628257751465,
      "step": 5822
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 5822,
      "training_step_time": 10.178884267807007
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 3.1359188556671143,
      "step": 5822
    },
    {
      "dynamics_loss_avg": 0.06224059164524078,
      "epoch": 2.220916748046875e-05,
      "step": 5822
    },
    {
      "action_loss_avg": 0.02965703085064888,
      "epoch": 2.220916748046875e-05,
      "step": 5822
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 5822,
      "training_step_time": 9.908708572387695
    },
    {
      "epoch": 2.220916748046875e-05,
      "model_forward_time": 3.1341779232025146,
      "step": 5822
    },
    {
      "epoch": 2.220916748046875e-05,
      "step": 5822,
      "training_step_time": 10.268883228302002
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 3.1443543434143066,
      "step": 5823
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 5823,
      "training_step_time": 9.913666725158691
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 3.140000820159912,
      "step": 5823
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 5823,
      "training_step_time": 10.182052850723267
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 3.1424596309661865,
      "step": 5823
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 5823,
      "training_step_time": 9.915632247924805
    },
    {
      "epoch": 2.2212982177734377e-05,
      "model_forward_time": 3.1385934352874756,
      "step": 5823
    },
    {
      "epoch": 2.2212982177734377e-05,
      "step": 5823,
      "training_step_time": 10.266520023345947
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 3.155076503753662,
      "step": 5824
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 5824,
      "training_step_time": 9.930461645126343
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 2.435030937194824,
      "step": 5824
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 5824,
      "training_step_time": 7.4655468463897705
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 2.4366354942321777,
      "step": 5824
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 5824,
      "training_step_time": 7.468308448791504
    },
    {
      "epoch": 2.2216796875e-05,
      "model_forward_time": 3.4008092880249023,
      "step": 5824
    },
    {
      "epoch": 2.2216796875e-05,
      "step": 5824,
      "training_step_time": 10.262498617172241
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 3.150073766708374,
      "step": 5825
    },
    {
      "dynamics_loss_avg": 0.049217276461422446,
      "epoch": 2.2220611572265626e-05,
      "step": 5825
    },
    {
      "action_loss_avg": 0.025656120292842387,
      "epoch": 2.2220611572265626e-05,
      "step": 5825
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 5825,
      "training_step_time": 9.923954725265503
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 2.436952829360962,
      "step": 5825
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 5825,
      "training_step_time": 7.470686674118042
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 3.1350481510162354,
      "step": 5825
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 5825,
      "training_step_time": 9.90969729423523
    },
    {
      "epoch": 2.2220611572265626e-05,
      "model_forward_time": 3.133824586868286,
      "step": 5825
    },
    {
      "epoch": 2.2220611572265626e-05,
      "step": 5825,
      "training_step_time": 10.282328844070435
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 3.423875331878662,
      "step": 5826
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 5826,
      "training_step_time": 10.196751832962036
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 3.14093017578125,
      "step": 5826
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 5826,
      "training_step_time": 9.906849145889282
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 3.1302297115325928,
      "step": 5826
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 5826,
      "training_step_time": 9.894487857818604
    },
    {
      "epoch": 2.222442626953125e-05,
      "model_forward_time": 3.137441873550415,
      "step": 5826
    },
    {
      "epoch": 2.222442626953125e-05,
      "step": 5826,
      "training_step_time": 10.299280166625977
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 3.145813465118408,
      "step": 5827
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 5827,
      "training_step_time": 10.177881002426147
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 3.13291597366333,
      "step": 5827
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 5827,
      "training_step_time": 9.887742757797241
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 3.1333367824554443,
      "step": 5827
    },
    {
      "dynamics_loss_avg": 0.049290735833346845,
      "epoch": 2.2228240966796875e-05,
      "step": 5827
    },
    {
      "action_loss_avg": 0.02086109984666109,
      "epoch": 2.2228240966796875e-05,
      "step": 5827
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 5827,
      "training_step_time": 9.899796724319458
    },
    {
      "epoch": 2.2228240966796875e-05,
      "model_forward_time": 3.12969970703125,
      "step": 5827
    },
    {
      "epoch": 2.2228240966796875e-05,
      "step": 5827,
      "training_step_time": 10.002777814865112
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 3.1453824043273926,
      "step": 5828
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 5828,
      "training_step_time": 9.914292097091675
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 2.4307637214660645,
      "step": 5828
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 5828,
      "training_step_time": 7.45116114616394
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 2.6930789947509766,
      "step": 5828
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 5828,
      "training_step_time": 7.721198081970215
    },
    {
      "epoch": 2.22320556640625e-05,
      "model_forward_time": 3.13916015625,
      "step": 5828
    },
    {
      "epoch": 2.22320556640625e-05,
      "step": 5828,
      "training_step_time": 10.258412599563599
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 3.149815082550049,
      "step": 5829
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 5829,
      "training_step_time": 9.91803240776062
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 3.1325268745422363,
      "step": 5829
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 5829,
      "training_step_time": 9.898106098175049
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 3.1392416954040527,
      "step": 5829
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 5829,
      "training_step_time": 10.183191776275635
    },
    {
      "epoch": 2.2235870361328124e-05,
      "model_forward_time": 3.1373040676116943,
      "step": 5829
    },
    {
      "epoch": 2.2235870361328124e-05,
      "step": 5829,
      "training_step_time": 10.015915632247925
    },
    {
      "epoch": 2.223968505859375e-05,
      "grad_norm": 0.3970579206943512,
      "learning_rate": 1e-05,
      "loss": 0.0895,
      "step": 5830
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 3.1617045402526855,
      "step": 5830
    },
    {
      "dynamics_loss_avg": 0.05596824213862419,
      "epoch": 2.223968505859375e-05,
      "step": 5830
    },
    {
      "action_loss_avg": 0.019388703955337407,
      "epoch": 2.223968505859375e-05,
      "step": 5830
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 5830,
      "training_step_time": 9.936185121536255
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 3.1342272758483887,
      "step": 5830
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 5830,
      "training_step_time": 9.901854991912842
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 2.434210777282715,
      "step": 5830
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 5830,
      "training_step_time": 7.45372200012207
    },
    {
      "epoch": 2.223968505859375e-05,
      "model_forward_time": 3.12898588180542,
      "step": 5830
    },
    {
      "epoch": 2.223968505859375e-05,
      "step": 5830,
      "training_step_time": 10.28527569770813
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 2.7653729915618896,
      "step": 5831
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 5831,
      "training_step_time": 7.788674354553223
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 3.1289780139923096,
      "step": 5831
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 5831,
      "training_step_time": 9.888668298721313
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 3.129429817199707,
      "step": 5831
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 5831,
      "training_step_time": 9.890270948410034
    },
    {
      "epoch": 2.2243499755859377e-05,
      "model_forward_time": 3.1268203258514404,
      "step": 5831
    },
    {
      "epoch": 2.2243499755859377e-05,
      "step": 5831,
      "training_step_time": 10.30956768989563
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 3.1502685546875,
      "step": 5832
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 5832,
      "training_step_time": 10.175673961639404
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 3.1289162635803223,
      "step": 5832
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 5832,
      "training_step_time": 9.897403001785278
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 3.1430838108062744,
      "step": 5832
    },
    {
      "dynamics_loss_avg": 0.06238232776522636,
      "epoch": 2.2247314453125e-05,
      "step": 5832
    },
    {
      "action_loss_avg": 0.02030829950235784,
      "epoch": 2.2247314453125e-05,
      "step": 5832
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 5832,
      "training_step_time": 9.90844988822937
    },
    {
      "epoch": 2.2247314453125e-05,
      "model_forward_time": 3.138831853866577,
      "step": 5832
    },
    {
      "epoch": 2.2247314453125e-05,
      "step": 5832,
      "training_step_time": 10.271622657775879
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 3.147117853164673,
      "step": 5833
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 5833,
      "training_step_time": 9.918212413787842
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 3.407857656478882,
      "step": 5833
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 5833,
      "training_step_time": 10.18194842338562
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 2.4382524490356445,
      "step": 5833
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 5833,
      "training_step_time": 7.468914270401001
    },
    {
      "epoch": 2.2251129150390626e-05,
      "model_forward_time": 2.437960624694824,
      "step": 5833
    },
    {
      "epoch": 2.2251129150390626e-05,
      "step": 5833,
      "training_step_time": 10.329533100128174
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 3.156925916671753,
      "step": 5834
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 5834,
      "training_step_time": 9.929429531097412
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 2.4390571117401123,
      "step": 5834
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 5834,
      "training_step_time": 7.466403007507324
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 3.1326143741607666,
      "step": 5834
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 5834,
      "training_step_time": 9.891854763031006
    },
    {
      "epoch": 2.225494384765625e-05,
      "model_forward_time": 3.389479637145996,
      "step": 5834
    },
    {
      "epoch": 2.225494384765625e-05,
      "step": 5834,
      "training_step_time": 10.244561433792114
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 3.152998447418213,
      "step": 5835
    },
    {
      "dynamics_loss_avg": 0.05624636970460415,
      "epoch": 2.2258758544921875e-05,
      "step": 5835
    },
    {
      "action_loss_avg": 0.028650462906807662,
      "epoch": 2.2258758544921875e-05,
      "step": 5835
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 5835,
      "training_step_time": 9.918230295181274
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 3.139130115509033,
      "step": 5835
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 5835,
      "training_step_time": 9.900328159332275
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 3.133781909942627,
      "step": 5835
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 5835,
      "training_step_time": 9.893364667892456
    },
    {
      "epoch": 2.2258758544921875e-05,
      "model_forward_time": 3.1406219005584717,
      "step": 5835
    },
    {
      "epoch": 2.2258758544921875e-05,
      "step": 5835,
      "training_step_time": 10.259452819824219
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 3.156346321105957,
      "step": 5836
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 5836,
      "training_step_time": 9.927241086959839
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 3.140000820159912,
      "step": 5836
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 5836,
      "training_step_time": 9.906513929367065
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 3.1373164653778076,
      "step": 5836
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 5836,
      "training_step_time": 9.91039490699768
    },
    {
      "epoch": 2.22625732421875e-05,
      "model_forward_time": 3.1456491947174072,
      "step": 5836
    },
    {
      "epoch": 2.22625732421875e-05,
      "step": 5836,
      "training_step_time": 10.325197458267212
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 3.150444746017456,
      "step": 5837
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 5837,
      "training_step_time": 10.19300103187561
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 2.4371278285980225,
      "step": 5837
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 5837,
      "training_step_time": 7.456592559814453
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 2.431169033050537,
      "step": 5837
    },
    {
      "dynamics_loss_avg": 0.05171958729624748,
      "epoch": 2.2266387939453124e-05,
      "step": 5837
    },
    {
      "action_loss_avg": 0.020649877749383448,
      "epoch": 2.2266387939453124e-05,
      "step": 5837
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 5837,
      "training_step_time": 7.4511399269104
    },
    {
      "epoch": 2.2266387939453124e-05,
      "model_forward_time": 3.1372005939483643,
      "step": 5837
    },
    {
      "epoch": 2.2266387939453124e-05,
      "step": 5837,
      "training_step_time": 10.017235040664673
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 3.1847078800201416,
      "step": 5838
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 5838,
      "training_step_time": 9.945895671844482
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 2.434187412261963,
      "step": 5838
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 5838,
      "training_step_time": 7.452138900756836
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 3.3952784538269043,
      "step": 5838
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 5838,
      "training_step_time": 10.156970500946045
    },
    {
      "epoch": 2.227020263671875e-05,
      "model_forward_time": 3.137434959411621,
      "step": 5838
    },
    {
      "epoch": 2.227020263671875e-05,
      "step": 5838,
      "training_step_time": 10.013417482376099
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 3.149576425552368,
      "step": 5839
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 5839,
      "training_step_time": 9.91254186630249
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 3.135521173477173,
      "step": 5839
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 5839,
      "training_step_time": 9.902806758880615
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 3.139831781387329,
      "step": 5839
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 5839,
      "training_step_time": 9.912944555282593
    },
    {
      "epoch": 2.2274017333984377e-05,
      "model_forward_time": 2.7313132286071777,
      "step": 5839
    },
    {
      "epoch": 2.2274017333984377e-05,
      "step": 5839,
      "training_step_time": 10.27706241607666
    },
    {
      "epoch": 2.227783203125e-05,
      "grad_norm": 0.47846364974975586,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 5840
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 3.1816022396087646,
      "step": 5840
    },
    {
      "dynamics_loss_avg": 0.04727649437263608,
      "epoch": 2.227783203125e-05,
      "step": 5840
    },
    {
      "action_loss_avg": 0.02096333270892501,
      "epoch": 2.227783203125e-05,
      "step": 5840
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 5840,
      "training_step_time": 9.952483177185059
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 3.136892557144165,
      "step": 5840
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 5840,
      "training_step_time": 9.902094841003418
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 3.144045114517212,
      "step": 5840
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 5840,
      "training_step_time": 9.91025686264038
    },
    {
      "epoch": 2.227783203125e-05,
      "model_forward_time": 3.1403448581695557,
      "step": 5840
    },
    {
      "epoch": 2.227783203125e-05,
      "step": 5840,
      "training_step_time": 10.283427476882935
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 3.186981678009033,
      "step": 5841
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 5841,
      "training_step_time": 9.955549240112305
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 3.1411025524139404,
      "step": 5841
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 5841,
      "training_step_time": 9.911195755004883
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 3.141768455505371,
      "step": 5841
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 5841,
      "training_step_time": 9.914918422698975
    },
    {
      "epoch": 2.2281646728515626e-05,
      "model_forward_time": 3.1439788341522217,
      "step": 5841
    },
    {
      "epoch": 2.2281646728515626e-05,
      "step": 5841,
      "training_step_time": 10.287760734558105
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 3.1768136024475098,
      "step": 5842
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 5842,
      "training_step_time": 10.224936246871948
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 3.1386663913726807,
      "step": 5842
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 5842,
      "training_step_time": 9.906344890594482
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 3.1414787769317627,
      "step": 5842
    },
    {
      "dynamics_loss_avg": 0.0562157865613699,
      "epoch": 2.228546142578125e-05,
      "step": 5842
    },
    {
      "action_loss_avg": 0.03709203749895096,
      "epoch": 2.228546142578125e-05,
      "step": 5842
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 5842,
      "training_step_time": 9.910723447799683
    },
    {
      "epoch": 2.228546142578125e-05,
      "model_forward_time": 3.1400389671325684,
      "step": 5842
    },
    {
      "epoch": 2.228546142578125e-05,
      "step": 5842,
      "training_step_time": 10.014318943023682
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 3.1493618488311768,
      "step": 5843
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 5843,
      "training_step_time": 9.919492959976196
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 3.143828868865967,
      "step": 5843
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 5843,
      "training_step_time": 10.192543029785156
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 2.4366259574890137,
      "step": 5843
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 5843,
      "training_step_time": 7.473177433013916
    },
    {
      "epoch": 2.2289276123046875e-05,
      "model_forward_time": 3.1396050453186035,
      "step": 5843
    },
    {
      "epoch": 2.2289276123046875e-05,
      "step": 5843,
      "training_step_time": 10.264266014099121
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 2.4505181312561035,
      "step": 5844
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 5844,
      "training_step_time": 7.481323003768921
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 3.143341064453125,
      "step": 5844
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 5844,
      "training_step_time": 9.916208744049072
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 3.139737844467163,
      "step": 5844
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 5844,
      "training_step_time": 9.913527011871338
    },
    {
      "epoch": 2.22930908203125e-05,
      "model_forward_time": 3.3923161029815674,
      "step": 5844
    },
    {
      "epoch": 2.22930908203125e-05,
      "step": 5844,
      "training_step_time": 10.255980968475342
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 3.161599636077881,
      "step": 5845
    },
    {
      "dynamics_loss_avg": 0.0607037715613842,
      "epoch": 2.2296905517578124e-05,
      "step": 5845
    },
    {
      "action_loss_avg": 0.02564528360962868,
      "epoch": 2.2296905517578124e-05,
      "step": 5845
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 5845,
      "training_step_time": 9.936598777770996
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 3.137298583984375,
      "step": 5845
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 5845,
      "training_step_time": 9.906294822692871
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 3.131751298904419,
      "step": 5845
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 5845,
      "training_step_time": 9.903102159500122
    },
    {
      "epoch": 2.2296905517578124e-05,
      "model_forward_time": 3.14247465133667,
      "step": 5845
    },
    {
      "epoch": 2.2296905517578124e-05,
      "step": 5845,
      "training_step_time": 10.247946977615356
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 2.7832233905792236,
      "step": 5846
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 5846,
      "training_step_time": 7.804733514785767
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 3.1385929584503174,
      "step": 5846
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 5846,
      "training_step_time": 9.913943529129028
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 3.1396870613098145,
      "step": 5846
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 5846,
      "training_step_time": 9.905131578445435
    },
    {
      "epoch": 2.230072021484375e-05,
      "model_forward_time": 3.1403489112854004,
      "step": 5846
    },
    {
      "epoch": 2.230072021484375e-05,
      "step": 5846,
      "training_step_time": 10.256342887878418
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 3.4446628093719482,
      "step": 5847
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 5847,
      "training_step_time": 10.209569692611694
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 3.139765977859497,
      "step": 5847
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 5847,
      "training_step_time": 9.902987718582153
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 3.134706735610962,
      "step": 5847
    },
    {
      "dynamics_loss_avg": 0.06529167015105486,
      "epoch": 2.2304534912109377e-05,
      "step": 5847
    },
    {
      "action_loss_avg": 0.022334794234484433,
      "epoch": 2.2304534912109377e-05,
      "step": 5847
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 5847,
      "training_step_time": 9.902403116226196
    },
    {
      "epoch": 2.2304534912109377e-05,
      "model_forward_time": 3.142733335494995,
      "step": 5847
    },
    {
      "epoch": 2.2304534912109377e-05,
      "step": 5847,
      "training_step_time": 10.327258586883545
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 3.1445672512054443,
      "step": 5848
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 5848,
      "training_step_time": 9.913979768753052
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 2.431072473526001,
      "step": 5848
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 5848,
      "training_step_time": 7.4551308155059814
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 3.388336420059204,
      "step": 5848
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 5848,
      "training_step_time": 10.156085968017578
    },
    {
      "epoch": 2.2308349609375e-05,
      "model_forward_time": 3.1390514373779297,
      "step": 5848
    },
    {
      "epoch": 2.2308349609375e-05,
      "step": 5848,
      "training_step_time": 10.235686302185059
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 3.152812957763672,
      "step": 5849
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 5849,
      "training_step_time": 9.929935693740845
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 3.140165090560913,
      "step": 5849
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 5849,
      "training_step_time": 9.912030220031738
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 3.406586170196533,
      "step": 5849
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 5849,
      "training_step_time": 10.180888414382935
    },
    {
      "epoch": 2.2312164306640626e-05,
      "model_forward_time": 3.1437652111053467,
      "step": 5849
    },
    {
      "epoch": 2.2312164306640626e-05,
      "step": 5849,
      "training_step_time": 10.317963600158691
    },
    {
      "epoch": 2.231597900390625e-05,
      "grad_norm": 0.38524574041366577,
      "learning_rate": 1e-05,
      "loss": 0.0849,
      "step": 5850
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 3.1832032203674316,
      "step": 5850
    },
    {
      "dynamics_loss_avg": 0.06488138101994992,
      "epoch": 2.231597900390625e-05,
      "step": 5850
    },
    {
      "action_loss_avg": 0.0338789539411664,
      "epoch": 2.231597900390625e-05,
      "step": 5850
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 5850,
      "training_step_time": 9.957065105438232
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 3.1451961994171143,
      "step": 5850
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 5850,
      "training_step_time": 9.9157133102417
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 3.1474037170410156,
      "step": 5850
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 5850,
      "training_step_time": 9.92267394065857
    },
    {
      "epoch": 2.231597900390625e-05,
      "model_forward_time": 3.1464450359344482,
      "step": 5850
    },
    {
      "epoch": 2.231597900390625e-05,
      "step": 5850,
      "training_step_time": 10.285227298736572
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 3.1856439113616943,
      "step": 5851
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 5851,
      "training_step_time": 9.960740327835083
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 3.14056396484375,
      "step": 5851
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 5851,
      "training_step_time": 9.908462762832642
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 3.144669532775879,
      "step": 5851
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 5851,
      "training_step_time": 9.913039684295654
    },
    {
      "epoch": 2.2319793701171875e-05,
      "model_forward_time": 3.139026641845703,
      "step": 5851
    },
    {
      "epoch": 2.2319793701171875e-05,
      "step": 5851,
      "training_step_time": 10.32037615776062
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 3.4709506034851074,
      "step": 5852
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 5852,
      "training_step_time": 10.241618633270264
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 3.1390209197998047,
      "step": 5852
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 5852,
      "training_step_time": 9.904767751693726
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 3.14109468460083,
      "step": 5852
    },
    {
      "dynamics_loss_avg": 0.05882584899663925,
      "epoch": 2.23236083984375e-05,
      "step": 5852
    },
    {
      "action_loss_avg": 0.026514746621251105,
      "epoch": 2.23236083984375e-05,
      "step": 5852
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 5852,
      "training_step_time": 9.909922122955322
    },
    {
      "epoch": 2.23236083984375e-05,
      "model_forward_time": 3.142977714538574,
      "step": 5852
    },
    {
      "epoch": 2.23236083984375e-05,
      "step": 5852,
      "training_step_time": 10.26157522201538
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 3.179043769836426,
      "step": 5853
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 5853,
      "training_step_time": 9.94344425201416
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 3.4002513885498047,
      "step": 5853
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 5853,
      "training_step_time": 10.1614830493927
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 3.1409382820129395,
      "step": 5853
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 5853,
      "training_step_time": 9.902192831039429
    },
    {
      "epoch": 2.2327423095703124e-05,
      "model_forward_time": 3.1343305110931396,
      "step": 5853
    },
    {
      "epoch": 2.2327423095703124e-05,
      "step": 5853,
      "training_step_time": 10.29692006111145
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 3.1363768577575684,
      "step": 5854
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 5854,
      "training_step_time": 9.896878242492676
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 3.125955104827881,
      "step": 5854
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 5854,
      "training_step_time": 9.887576818466187
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 3.4036571979522705,
      "step": 5854
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 5854,
      "training_step_time": 10.16219449043274
    },
    {
      "epoch": 2.233123779296875e-05,
      "model_forward_time": 3.1314961910247803,
      "step": 5854
    },
    {
      "epoch": 2.233123779296875e-05,
      "step": 5854,
      "training_step_time": 10.00505518913269
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 2.467900276184082,
      "step": 5855
    },
    {
      "dynamics_loss_avg": 0.06111814398318529,
      "epoch": 2.2335052490234377e-05,
      "step": 5855
    },
    {
      "action_loss_avg": 0.024929805565625428,
      "epoch": 2.2335052490234377e-05,
      "step": 5855
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 5855,
      "training_step_time": 7.490043640136719
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 3.1252710819244385,
      "step": 5855
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 5855,
      "training_step_time": 9.883750200271606
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 3.1281206607818604,
      "step": 5855
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 5855,
      "training_step_time": 9.887177228927612
    },
    {
      "epoch": 2.2335052490234377e-05,
      "model_forward_time": 3.4051108360290527,
      "step": 5855
    },
    {
      "epoch": 2.2335052490234377e-05,
      "step": 5855,
      "training_step_time": 10.2711660861969
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 3.1681787967681885,
      "step": 5856
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 5856,
      "training_step_time": 9.948395013809204
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 3.13446307182312,
      "step": 5856
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 5856,
      "training_step_time": 9.905598878860474
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 3.135286569595337,
      "step": 5856
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 5856,
      "training_step_time": 9.908653259277344
    },
    {
      "epoch": 2.23388671875e-05,
      "model_forward_time": 3.1408822536468506,
      "step": 5856
    },
    {
      "epoch": 2.23388671875e-05,
      "step": 5856,
      "training_step_time": 10.257546424865723
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 3.14644455909729,
      "step": 5857
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 5857,
      "training_step_time": 10.20577597618103
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 3.1388120651245117,
      "step": 5857
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 5857,
      "training_step_time": 9.90798306465149
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 3.1366069316864014,
      "step": 5857
    },
    {
      "dynamics_loss_avg": 0.08338056094944477,
      "epoch": 2.2342681884765626e-05,
      "step": 5857
    },
    {
      "action_loss_avg": 0.038522783666849136,
      "epoch": 2.2342681884765626e-05,
      "step": 5857
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 5857,
      "training_step_time": 9.910973310470581
    },
    {
      "epoch": 2.2342681884765626e-05,
      "model_forward_time": 3.138712167739868,
      "step": 5857
    },
    {
      "epoch": 2.2342681884765626e-05,
      "step": 5857,
      "training_step_time": 10.008929967880249
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 3.146000385284424,
      "step": 5858
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 5858,
      "training_step_time": 9.917771816253662
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 3.4333584308624268,
      "step": 5858
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 5858,
      "training_step_time": 10.20005178451538
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 2.439225673675537,
      "step": 5858
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 5858,
      "training_step_time": 7.459954738616943
    },
    {
      "epoch": 2.234649658203125e-05,
      "model_forward_time": 3.13867449760437,
      "step": 5858
    },
    {
      "epoch": 2.234649658203125e-05,
      "step": 5858,
      "training_step_time": 10.289108514785767
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 3.153979778289795,
      "step": 5859
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 5859,
      "training_step_time": 9.919955730438232
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 3.1360483169555664,
      "step": 5859
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 5859,
      "training_step_time": 9.900213479995728
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 3.401254415512085,
      "step": 5859
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 5859,
      "training_step_time": 10.166038513183594
    },
    {
      "epoch": 2.2350311279296875e-05,
      "model_forward_time": 3.141498327255249,
      "step": 5859
    },
    {
      "epoch": 2.2350311279296875e-05,
      "step": 5859,
      "training_step_time": 10.2283775806427
    },
    {
      "epoch": 2.23541259765625e-05,
      "grad_norm": 0.6178678274154663,
      "learning_rate": 1e-05,
      "loss": 0.0892,
      "step": 5860
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 3.1814398765563965,
      "step": 5860
    },
    {
      "dynamics_loss_avg": 0.04857059381902218,
      "epoch": 2.23541259765625e-05,
      "step": 5860
    },
    {
      "action_loss_avg": 0.021783330477774143,
      "epoch": 2.23541259765625e-05,
      "step": 5860
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 5860,
      "training_step_time": 9.95257043838501
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 3.1320714950561523,
      "step": 5860
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 5860,
      "training_step_time": 9.90023136138916
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 3.141578435897827,
      "step": 5860
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 5860,
      "training_step_time": 9.908204555511475
    },
    {
      "epoch": 2.23541259765625e-05,
      "model_forward_time": 3.4253485202789307,
      "step": 5860
    },
    {
      "epoch": 2.23541259765625e-05,
      "step": 5860,
      "training_step_time": 10.287238836288452
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 3.151538610458374,
      "step": 5861
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 5861,
      "training_step_time": 9.924991130828857
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 3.138087034225464,
      "step": 5861
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 5861,
      "training_step_time": 9.908912658691406
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 2.4334425926208496,
      "step": 5861
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 5861,
      "training_step_time": 7.459224224090576
    },
    {
      "epoch": 2.2357940673828124e-05,
      "model_forward_time": 3.1361947059631348,
      "step": 5861
    },
    {
      "epoch": 2.2357940673828124e-05,
      "step": 5861,
      "training_step_time": 10.270843505859375
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 3.4154326915740967,
      "step": 5862
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 5862,
      "training_step_time": 10.179826736450195
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 3.127809762954712,
      "step": 5862
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 5862,
      "training_step_time": 9.89072060585022
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 3.1347644329071045,
      "step": 5862
    },
    {
      "dynamics_loss_avg": 0.05131069775670767,
      "epoch": 2.236175537109375e-05,
      "step": 5862
    },
    {
      "action_loss_avg": 0.02076902911067009,
      "epoch": 2.236175537109375e-05,
      "step": 5862
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 5862,
      "training_step_time": 9.899533748626709
    },
    {
      "epoch": 2.236175537109375e-05,
      "model_forward_time": 3.131465435028076,
      "step": 5862
    },
    {
      "epoch": 2.236175537109375e-05,
      "step": 5862,
      "training_step_time": 10.28273606300354
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 3.1765356063842773,
      "step": 5863
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 5863,
      "training_step_time": 9.943719148635864
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 3.1319162845611572,
      "step": 5863
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 5863,
      "training_step_time": 10.176120519638062
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 3.1308374404907227,
      "step": 5863
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 5863,
      "training_step_time": 9.897342205047607
    },
    {
      "epoch": 2.2365570068359377e-05,
      "model_forward_time": 3.1306629180908203,
      "step": 5863
    },
    {
      "epoch": 2.2365570068359377e-05,
      "step": 5863,
      "training_step_time": 10.261557579040527
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 3.137721300125122,
      "step": 5864
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 5864,
      "training_step_time": 9.902749300003052
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 3.129006862640381,
      "step": 5864
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 5864,
      "training_step_time": 9.892612934112549
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 3.407824993133545,
      "step": 5864
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 5864,
      "training_step_time": 10.173277854919434
    },
    {
      "epoch": 2.2369384765625e-05,
      "model_forward_time": 3.128340244293213,
      "step": 5864
    },
    {
      "epoch": 2.2369384765625e-05,
      "step": 5864,
      "training_step_time": 10.279771566390991
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 3.141261339187622,
      "step": 5865
    },
    {
      "dynamics_loss_avg": 0.058190458081662656,
      "epoch": 2.2373199462890626e-05,
      "step": 5865
    },
    {
      "action_loss_avg": 0.02577579990029335,
      "epoch": 2.2373199462890626e-05,
      "step": 5865
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 5865,
      "training_step_time": 9.904402017593384
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 3.1311118602752686,
      "step": 5865
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 5865,
      "training_step_time": 9.89392352104187
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 2.425379991531372,
      "step": 5865
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 5865,
      "training_step_time": 7.444115161895752
    },
    {
      "epoch": 2.2373199462890626e-05,
      "model_forward_time": 3.1275808811187744,
      "step": 5865
    },
    {
      "epoch": 2.2373199462890626e-05,
      "step": 5865,
      "training_step_time": 10.00093126296997
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 3.432955026626587,
      "step": 5866
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 5866,
      "training_step_time": 10.198845624923706
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 2.42781662940979,
      "step": 5866
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 5866,
      "training_step_time": 7.450472593307495
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 3.124145984649658,
      "step": 5866
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 5866,
      "training_step_time": 9.88686490058899
    },
    {
      "epoch": 2.237701416015625e-05,
      "model_forward_time": 3.132366180419922,
      "step": 5866
    },
    {
      "epoch": 2.237701416015625e-05,
      "step": 5866,
      "training_step_time": 10.005340576171875
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 2.442629814147949,
      "step": 5867
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 5867,
      "training_step_time": 7.461695909500122
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 3.3924098014831543,
      "step": 5867
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 5867,
      "training_step_time": 10.15813136100769
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 3.129317283630371,
      "step": 5867
    },
    {
      "dynamics_loss_avg": 0.05430175624787807,
      "epoch": 2.2380828857421875e-05,
      "step": 5867
    },
    {
      "action_loss_avg": 0.02654516976326704,
      "epoch": 2.2380828857421875e-05,
      "step": 5867
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 5867,
      "training_step_time": 9.895935535430908
    },
    {
      "epoch": 2.2380828857421875e-05,
      "model_forward_time": 3.1344082355499268,
      "step": 5867
    },
    {
      "epoch": 2.2380828857421875e-05,
      "step": 5867,
      "training_step_time": 10.265149116516113
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 2.446042776107788,
      "step": 5868
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 5868,
      "training_step_time": 7.467804670333862
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 3.1235337257385254,
      "step": 5868
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 5868,
      "training_step_time": 9.88488507270813
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 3.38175892829895,
      "step": 5868
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 5868,
      "training_step_time": 10.142662048339844
    },
    {
      "epoch": 2.23846435546875e-05,
      "model_forward_time": 2.4306530952453613,
      "step": 5868
    },
    {
      "epoch": 2.23846435546875e-05,
      "step": 5868,
      "training_step_time": 10.013727188110352
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 3.140965461730957,
      "step": 5869
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 5869,
      "training_step_time": 9.902646541595459
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 3.126370429992676,
      "step": 5869
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 5869,
      "training_step_time": 9.886203527450562
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 3.126047134399414,
      "step": 5869
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 5869,
      "training_step_time": 10.179036378860474
    },
    {
      "epoch": 2.2388458251953124e-05,
      "model_forward_time": 3.129879951477051,
      "step": 5869
    },
    {
      "epoch": 2.2388458251953124e-05,
      "step": 5869,
      "training_step_time": 10.297048568725586
    },
    {
      "epoch": 2.239227294921875e-05,
      "grad_norm": 0.5099148750305176,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 5870
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 3.170473575592041,
      "step": 5870
    },
    {
      "dynamics_loss_avg": 0.04551907302811742,
      "epoch": 2.239227294921875e-05,
      "step": 5870
    },
    {
      "action_loss_avg": 0.018756329407915474,
      "epoch": 2.239227294921875e-05,
      "step": 5870
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 5870,
      "training_step_time": 9.939427375793457
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 3.132981061935425,
      "step": 5870
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 5870,
      "training_step_time": 9.893923044204712
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 3.1338613033294678,
      "step": 5870
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 5870,
      "training_step_time": 9.898381471633911
    },
    {
      "epoch": 2.239227294921875e-05,
      "model_forward_time": 3.1334969997406006,
      "step": 5870
    },
    {
      "epoch": 2.239227294921875e-05,
      "step": 5870,
      "training_step_time": 10.266449451446533
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 3.1421420574188232,
      "step": 5871
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 5871,
      "training_step_time": 9.9036123752594
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 3.1330173015594482,
      "step": 5871
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 5871,
      "training_step_time": 9.896567821502686
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 3.1370739936828613,
      "step": 5871
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 5871,
      "training_step_time": 9.90372633934021
    },
    {
      "epoch": 2.2396087646484377e-05,
      "model_forward_time": 2.4286086559295654,
      "step": 5871
    },
    {
      "epoch": 2.2396087646484377e-05,
      "step": 5871,
      "training_step_time": 10.021517753601074
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 3.138491153717041,
      "step": 5872
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 5872,
      "training_step_time": 10.179964303970337
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 3.1328489780426025,
      "step": 5872
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 5872,
      "training_step_time": 9.890560865402222
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 3.127779245376587,
      "step": 5872
    },
    {
      "dynamics_loss_avg": 0.062092102319002154,
      "epoch": 2.239990234375e-05,
      "step": 5872
    },
    {
      "action_loss_avg": 0.02565506724640727,
      "epoch": 2.239990234375e-05,
      "step": 5872
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 5872,
      "training_step_time": 9.889894723892212
    },
    {
      "epoch": 2.239990234375e-05,
      "model_forward_time": 3.12619948387146,
      "step": 5872
    },
    {
      "epoch": 2.239990234375e-05,
      "step": 5872,
      "training_step_time": 10.281821727752686
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 3.140630006790161,
      "step": 5873
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 5873,
      "training_step_time": 9.904094219207764
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 3.391857862472534,
      "step": 5873
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 5873,
      "training_step_time": 10.152910709381104
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 3.1319580078125,
      "step": 5873
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 5873,
      "training_step_time": 9.893039226531982
    },
    {
      "epoch": 2.2403717041015626e-05,
      "model_forward_time": 3.131422758102417,
      "step": 5873
    },
    {
      "epoch": 2.2403717041015626e-05,
      "step": 5873,
      "training_step_time": 10.278464317321777
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 2.4672510623931885,
      "step": 5874
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 5874,
      "training_step_time": 7.490428686141968
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 2.4298479557037354,
      "step": 5874
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 5874,
      "training_step_time": 7.4538774490356445
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 3.132558584213257,
      "step": 5874
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 5874,
      "training_step_time": 9.901897192001343
    },
    {
      "epoch": 2.240753173828125e-05,
      "model_forward_time": 3.376711130142212,
      "step": 5874
    },
    {
      "epoch": 2.240753173828125e-05,
      "step": 5874,
      "training_step_time": 10.320063352584839
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 3.1430764198303223,
      "step": 5875
    },
    {
      "dynamics_loss_avg": 0.06315484307706357,
      "epoch": 2.2411346435546875e-05,
      "step": 5875
    },
    {
      "action_loss_avg": 0.028753406181931496,
      "epoch": 2.2411346435546875e-05,
      "step": 5875
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 5875,
      "training_step_time": 9.918081521987915
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 2.433382272720337,
      "step": 5875
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 5875,
      "training_step_time": 7.459648370742798
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 2.434953451156616,
      "step": 5875
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 5875,
      "training_step_time": 7.463771820068359
    },
    {
      "epoch": 2.2411346435546875e-05,
      "model_forward_time": 3.133427858352661,
      "step": 5875
    },
    {
      "epoch": 2.2411346435546875e-05,
      "step": 5875,
      "training_step_time": 10.280198335647583
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 3.4018304347991943,
      "step": 5876
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 5876,
      "training_step_time": 10.169963836669922
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 3.133845329284668,
      "step": 5876
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 5876,
      "training_step_time": 9.89846134185791
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 3.127995014190674,
      "step": 5876
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 5876,
      "training_step_time": 9.89128565788269
    },
    {
      "epoch": 2.24151611328125e-05,
      "model_forward_time": 3.129746913909912,
      "step": 5876
    },
    {
      "epoch": 2.24151611328125e-05,
      "step": 5876,
      "training_step_time": 10.018243789672852
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 3.164116621017456,
      "step": 5877
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 5877,
      "training_step_time": 9.9269437789917
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 3.39534854888916,
      "step": 5877
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 5877,
      "training_step_time": 10.16130781173706
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 3.1294121742248535,
      "step": 5877
    },
    {
      "dynamics_loss_avg": 0.0534479595720768,
      "epoch": 2.2418975830078124e-05,
      "step": 5877
    },
    {
      "action_loss_avg": 0.03100910410284996,
      "epoch": 2.2418975830078124e-05,
      "step": 5877
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 5877,
      "training_step_time": 9.8967125415802
    },
    {
      "epoch": 2.2418975830078124e-05,
      "model_forward_time": 3.1314916610717773,
      "step": 5877
    },
    {
      "epoch": 2.2418975830078124e-05,
      "step": 5877,
      "training_step_time": 10.291611909866333
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 3.1425862312316895,
      "step": 5878
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 5878,
      "training_step_time": 9.90650224685669
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 3.126303195953369,
      "step": 5878
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 5878,
      "training_step_time": 10.154705286026001
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 3.130645751953125,
      "step": 5878
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 5878,
      "training_step_time": 9.89747166633606
    },
    {
      "epoch": 2.242279052734375e-05,
      "model_forward_time": 3.133714199066162,
      "step": 5878
    },
    {
      "epoch": 2.242279052734375e-05,
      "step": 5878,
      "training_step_time": 10.280314683914185
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 2.4733636379241943,
      "step": 5879
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 5879,
      "training_step_time": 7.4928624629974365
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 3.1280109882354736,
      "step": 5879
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 5879,
      "training_step_time": 9.890774726867676
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 3.1309030055999756,
      "step": 5879
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 5879,
      "training_step_time": 9.895084381103516
    },
    {
      "epoch": 2.2426605224609373e-05,
      "model_forward_time": 3.3945705890655518,
      "step": 5879
    },
    {
      "epoch": 2.2426605224609373e-05,
      "step": 5879,
      "training_step_time": 10.252991676330566
    },
    {
      "epoch": 2.2430419921875e-05,
      "grad_norm": 0.4333094656467438,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 5880
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 2.4423999786376953,
      "step": 5880
    },
    {
      "dynamics_loss_avg": 0.06515070740133524,
      "epoch": 2.2430419921875e-05,
      "step": 5880
    },
    {
      "action_loss_avg": 0.026536542177200317,
      "epoch": 2.2430419921875e-05,
      "step": 5880
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 5880,
      "training_step_time": 7.474746942520142
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 3.124732255935669,
      "step": 5880
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 5880,
      "training_step_time": 9.885804891586304
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 3.127553701400757,
      "step": 5880
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 5880,
      "training_step_time": 9.888874769210815
    },
    {
      "epoch": 2.2430419921875e-05,
      "model_forward_time": 3.125948667526245,
      "step": 5880
    },
    {
      "epoch": 2.2430419921875e-05,
      "step": 5880,
      "training_step_time": 10.265696287155151
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 3.1767399311065674,
      "step": 5881
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 5881,
      "training_step_time": 9.943670749664307
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 2.4341399669647217,
      "step": 5881
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 5881,
      "training_step_time": 7.455776214599609
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 3.128389596939087,
      "step": 5881
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 5881,
      "training_step_time": 9.895502090454102
    },
    {
      "epoch": 2.2434234619140626e-05,
      "model_forward_time": 3.1289820671081543,
      "step": 5881
    },
    {
      "epoch": 2.2434234619140626e-05,
      "step": 5881,
      "training_step_time": 10.052294254302979
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 2.437584161758423,
      "step": 5882
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 5882,
      "training_step_time": 7.460874319076538
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 3.373887300491333,
      "step": 5882
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 5882,
      "training_step_time": 10.140308380126953
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 3.1333980560302734,
      "step": 5882
    },
    {
      "dynamics_loss_avg": 0.050956799741834403,
      "epoch": 2.243804931640625e-05,
      "step": 5882
    },
    {
      "action_loss_avg": 0.02514152377843857,
      "epoch": 2.243804931640625e-05,
      "step": 5882
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 5882,
      "training_step_time": 9.899861812591553
    },
    {
      "epoch": 2.243804931640625e-05,
      "model_forward_time": 3.129382371902466,
      "step": 5882
    },
    {
      "epoch": 2.243804931640625e-05,
      "step": 5882,
      "training_step_time": 10.280807733535767
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 3.1532504558563232,
      "step": 5883
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 5883,
      "training_step_time": 9.913714408874512
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 2.431130886077881,
      "step": 5883
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 5883,
      "training_step_time": 7.452500343322754
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 3.390817403793335,
      "step": 5883
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 5883,
      "training_step_time": 10.15405821800232
    },
    {
      "epoch": 2.2441864013671875e-05,
      "model_forward_time": 3.126973867416382,
      "step": 5883
    },
    {
      "epoch": 2.2441864013671875e-05,
      "step": 5883,
      "training_step_time": 10.261032819747925
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 3.15159010887146,
      "step": 5884
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 5884,
      "training_step_time": 9.910758256912231
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 3.1371355056762695,
      "step": 5884
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 5884,
      "training_step_time": 9.901927471160889
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 3.137500047683716,
      "step": 5884
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 5884,
      "training_step_time": 9.90296220779419
    },
    {
      "epoch": 2.24456787109375e-05,
      "model_forward_time": 2.010852575302124,
      "step": 5884
    },
    {
      "epoch": 2.24456787109375e-05,
      "step": 5884,
      "training_step_time": 10.24080514907837
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 3.1515660285949707,
      "step": 5885
    },
    {
      "dynamics_loss_avg": 0.07462546676397323,
      "epoch": 2.2449493408203124e-05,
      "step": 5885
    },
    {
      "action_loss_avg": 0.028728198260068893,
      "epoch": 2.2449493408203124e-05,
      "step": 5885
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 5885,
      "training_step_time": 9.921299457550049
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 3.138719081878662,
      "step": 5885
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 5885,
      "training_step_time": 9.909102201461792
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 3.135654926300049,
      "step": 5885
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 5885,
      "training_step_time": 9.904887676239014
    },
    {
      "epoch": 2.2449493408203124e-05,
      "model_forward_time": 3.1417088508605957,
      "step": 5885
    },
    {
      "epoch": 2.2449493408203124e-05,
      "step": 5885,
      "training_step_time": 10.309090614318848
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 3.4339144229888916,
      "step": 5886
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 5886,
      "training_step_time": 10.202564239501953
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 3.1437556743621826,
      "step": 5886
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 5886,
      "training_step_time": 9.914850950241089
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 3.1409173011779785,
      "step": 5886
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 5886,
      "training_step_time": 9.91249966621399
    },
    {
      "epoch": 2.245330810546875e-05,
      "model_forward_time": 3.140331745147705,
      "step": 5886
    },
    {
      "epoch": 2.245330810546875e-05,
      "step": 5886,
      "training_step_time": 10.02386474609375
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 3.165070056915283,
      "step": 5887
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 5887,
      "training_step_time": 9.946172952651978
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 3.4338581562042236,
      "step": 5887
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 5887,
      "training_step_time": 10.214224576950073
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 3.1457602977752686,
      "step": 5887
    },
    {
      "dynamics_loss_avg": 0.06717365076765418,
      "epoch": 2.2457122802734373e-05,
      "step": 5887
    },
    {
      "action_loss_avg": 0.02398269670084119,
      "epoch": 2.2457122802734373e-05,
      "step": 5887
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 5887,
      "training_step_time": 9.924474477767944
    },
    {
      "epoch": 2.2457122802734373e-05,
      "model_forward_time": 3.1479365825653076,
      "step": 5887
    },
    {
      "epoch": 2.2457122802734373e-05,
      "step": 5887,
      "training_step_time": 10.026405096054077
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 3.1600422859191895,
      "step": 5888
    },
    {
      "epoch": 2.24609375e-05,
      "step": 5888,
      "training_step_time": 9.934834957122803
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 3.13913631439209,
      "step": 5888
    },
    {
      "epoch": 2.24609375e-05,
      "step": 5888,
      "training_step_time": 9.90992021560669
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 3.4123737812042236,
      "step": 5888
    },
    {
      "epoch": 2.24609375e-05,
      "step": 5888,
      "training_step_time": 10.18643045425415
    },
    {
      "epoch": 2.24609375e-05,
      "model_forward_time": 3.1422412395477295,
      "step": 5888
    },
    {
      "epoch": 2.24609375e-05,
      "step": 5888,
      "training_step_time": 10.024005651473999
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 3.17989182472229,
      "step": 5889
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 5889,
      "training_step_time": 9.950373888015747
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 3.134948968887329,
      "step": 5889
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 5889,
      "training_step_time": 9.904700994491577
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 3.1382992267608643,
      "step": 5889
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 5889,
      "training_step_time": 10.193955659866333
    },
    {
      "epoch": 2.2464752197265626e-05,
      "model_forward_time": 3.1401267051696777,
      "step": 5889
    },
    {
      "epoch": 2.2464752197265626e-05,
      "step": 5889,
      "training_step_time": 10.333563327789307
    },
    {
      "epoch": 2.246856689453125e-05,
      "grad_norm": 0.5196723341941833,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 5890
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 3.1851062774658203,
      "step": 5890
    },
    {
      "dynamics_loss_avg": 0.0690944479778409,
      "epoch": 2.246856689453125e-05,
      "step": 5890
    },
    {
      "action_loss_avg": 0.027582898456603288,
      "epoch": 2.246856689453125e-05,
      "step": 5890
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 5890,
      "training_step_time": 9.963249444961548
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 3.13696026802063,
      "step": 5890
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 5890,
      "training_step_time": 9.903737783432007
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 3.140338182449341,
      "step": 5890
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 5890,
      "training_step_time": 9.907453536987305
    },
    {
      "epoch": 2.246856689453125e-05,
      "model_forward_time": 3.431265354156494,
      "step": 5890
    },
    {
      "epoch": 2.246856689453125e-05,
      "step": 5890,
      "training_step_time": 10.302572965621948
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 3.1587350368499756,
      "step": 5891
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 5891,
      "training_step_time": 9.930944204330444
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 3.138456344604492,
      "step": 5891
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 5891,
      "training_step_time": 9.905446290969849
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 2.4359757900238037,
      "step": 5891
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 5891,
      "training_step_time": 7.459882020950317
    },
    {
      "epoch": 2.2472381591796875e-05,
      "model_forward_time": 3.1354100704193115,
      "step": 5891
    },
    {
      "epoch": 2.2472381591796875e-05,
      "step": 5891,
      "training_step_time": 10.2638418674469
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 3.166322708129883,
      "step": 5892
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 5892,
      "training_step_time": 9.931320667266846
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 3.3952925205230713,
      "step": 5892
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 5892,
      "training_step_time": 10.160341739654541
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 3.141364097595215,
      "step": 5892
    },
    {
      "dynamics_loss_avg": 0.06117057800292969,
      "epoch": 2.24761962890625e-05,
      "step": 5892
    },
    {
      "action_loss_avg": 0.027448731148615478,
      "epoch": 2.24761962890625e-05,
      "step": 5892
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 5892,
      "training_step_time": 9.91600227355957
    },
    {
      "epoch": 2.24761962890625e-05,
      "model_forward_time": 3.1425366401672363,
      "step": 5892
    },
    {
      "epoch": 2.24761962890625e-05,
      "step": 5892,
      "training_step_time": 10.022778987884521
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 3.175077199935913,
      "step": 5893
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 5893,
      "training_step_time": 9.953882694244385
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 3.4179930686950684,
      "step": 5893
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 5893,
      "training_step_time": 10.191111087799072
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 3.1473820209503174,
      "step": 5893
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 5893,
      "training_step_time": 9.916906118392944
    },
    {
      "epoch": 2.2480010986328124e-05,
      "model_forward_time": 3.140876054763794,
      "step": 5893
    },
    {
      "epoch": 2.2480010986328124e-05,
      "step": 5893,
      "training_step_time": 10.011186361312866
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 3.1651012897491455,
      "step": 5894
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 5894,
      "training_step_time": 9.943130254745483
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 3.145181894302368,
      "step": 5894
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 5894,
      "training_step_time": 9.916476488113403
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 3.1407456398010254,
      "step": 5894
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 5894,
      "training_step_time": 10.172417640686035
    },
    {
      "epoch": 2.248382568359375e-05,
      "model_forward_time": 2.438308000564575,
      "step": 5894
    },
    {
      "epoch": 2.248382568359375e-05,
      "step": 5894,
      "training_step_time": 10.29741644859314
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 3.191796064376831,
      "step": 5895
    },
    {
      "dynamics_loss_avg": 0.06155305225402117,
      "epoch": 2.2487640380859373e-05,
      "step": 5895
    },
    {
      "action_loss_avg": 0.0220039586070925,
      "epoch": 2.2487640380859373e-05,
      "step": 5895
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 5895,
      "training_step_time": 9.964738607406616
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 3.1417336463928223,
      "step": 5895
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 5895,
      "training_step_time": 9.913041591644287
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 3.1442744731903076,
      "step": 5895
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 5895,
      "training_step_time": 9.916468381881714
    },
    {
      "epoch": 2.2487640380859373e-05,
      "model_forward_time": 3.4005701541900635,
      "step": 5895
    },
    {
      "epoch": 2.2487640380859373e-05,
      "step": 5895,
      "training_step_time": 10.322007656097412
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 3.1838817596435547,
      "step": 5896
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 5896,
      "training_step_time": 9.956689834594727
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 2.4368324279785156,
      "step": 5896
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 5896,
      "training_step_time": 7.464541435241699
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 3.1373422145843506,
      "step": 5896
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 5896,
      "training_step_time": 9.912097454071045
    },
    {
      "epoch": 2.2491455078125e-05,
      "model_forward_time": 3.1482439041137695,
      "step": 5896
    },
    {
      "epoch": 2.2491455078125e-05,
      "step": 5896,
      "training_step_time": 10.265649795532227
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 3.1608119010925293,
      "step": 5897
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 5897,
      "training_step_time": 10.216625928878784
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 3.147512674331665,
      "step": 5897
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 5897,
      "training_step_time": 9.922674655914307
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 3.1448111534118652,
      "step": 5897
    },
    {
      "dynamics_loss_avg": 0.06442822851240634,
      "epoch": 2.2495269775390626e-05,
      "step": 5897
    },
    {
      "action_loss_avg": 0.023798521887511016,
      "epoch": 2.2495269775390626e-05,
      "step": 5897
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 5897,
      "training_step_time": 9.922602415084839
    },
    {
      "epoch": 2.2495269775390626e-05,
      "model_forward_time": 3.148709774017334,
      "step": 5897
    },
    {
      "epoch": 2.2495269775390626e-05,
      "step": 5897,
      "training_step_time": 10.28051233291626
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 3.1900947093963623,
      "step": 5898
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 5898,
      "training_step_time": 9.967701196670532
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 3.148242950439453,
      "step": 5898
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 5898,
      "training_step_time": 10.212551355361938
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 3.140709400177002,
      "step": 5898
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 5898,
      "training_step_time": 9.913120985031128
    },
    {
      "epoch": 2.249908447265625e-05,
      "model_forward_time": 3.141572952270508,
      "step": 5898
    },
    {
      "epoch": 2.249908447265625e-05,
      "step": 5898,
      "training_step_time": 10.270529985427856
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 3.1785924434661865,
      "step": 5899
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 5899,
      "training_step_time": 9.945473194122314
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 3.13134765625,
      "step": 5899
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 5899,
      "training_step_time": 9.895783185958862
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 3.139845848083496,
      "step": 5899
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 5899,
      "training_step_time": 10.184363603591919
    },
    {
      "epoch": 2.2502899169921875e-05,
      "model_forward_time": 3.143900156021118,
      "step": 5899
    },
    {
      "epoch": 2.2502899169921875e-05,
      "step": 5899,
      "training_step_time": 10.313610553741455
    },
    {
      "epoch": 2.25067138671875e-05,
      "grad_norm": 0.4526304602622986,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 5900
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 3.1878180503845215,
      "step": 5900
    },
    {
      "dynamics_loss_avg": 0.055048163048923016,
      "epoch": 2.25067138671875e-05,
      "step": 5900
    },
    {
      "action_loss_avg": 0.021237343829125167,
      "epoch": 2.25067138671875e-05,
      "step": 5900
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 5900,
      "training_step_time": 9.958334922790527
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 3.1356077194213867,
      "step": 5900
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 5900,
      "training_step_time": 9.90866231918335
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 3.1334168910980225,
      "step": 5900
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 5900,
      "training_step_time": 9.89557671546936
    },
    {
      "epoch": 2.25067138671875e-05,
      "model_forward_time": 3.135920524597168,
      "step": 5900
    },
    {
      "epoch": 2.25067138671875e-05,
      "step": 5900,
      "training_step_time": 10.27163290977478
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 3.179656982421875,
      "step": 5901
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 5901,
      "training_step_time": 9.967909097671509
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 2.4298102855682373,
      "step": 5901
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 5901,
      "training_step_time": 7.446497917175293
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 3.1257619857788086,
      "step": 5901
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 5901,
      "training_step_time": 9.884169340133667
    },
    {
      "epoch": 2.2510528564453124e-05,
      "model_forward_time": 3.128373384475708,
      "step": 5901
    },
    {
      "epoch": 2.2510528564453124e-05,
      "step": 5901,
      "training_step_time": 10.297821283340454
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 3.1596505641937256,
      "step": 5902
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 5902,
      "training_step_time": 10.193900346755981
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 2.4328997135162354,
      "step": 5902
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 5902,
      "training_step_time": 7.45758843421936
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 3.14032244682312,
      "step": 5902
    },
    {
      "dynamics_loss_avg": 0.05369055215269327,
      "epoch": 2.251434326171875e-05,
      "step": 5902
    },
    {
      "action_loss_avg": 0.023497684672474862,
      "epoch": 2.251434326171875e-05,
      "step": 5902
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 5902,
      "training_step_time": 9.913792610168457
    },
    {
      "epoch": 2.251434326171875e-05,
      "model_forward_time": 3.130383014678955,
      "step": 5902
    },
    {
      "epoch": 2.251434326171875e-05,
      "step": 5902,
      "training_step_time": 10.0228431224823
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 3.165320873260498,
      "step": 5903
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 5903,
      "training_step_time": 9.942965745925903
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 3.1355066299438477,
      "step": 5903
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 5903,
      "training_step_time": 9.905679941177368
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 3.3993051052093506,
      "step": 5903
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 5903,
      "training_step_time": 10.168558835983276
    },
    {
      "epoch": 2.2518157958984373e-05,
      "model_forward_time": 3.1319286823272705,
      "step": 5903
    },
    {
      "epoch": 2.2518157958984373e-05,
      "step": 5903,
      "training_step_time": 10.30716586112976
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 3.1713452339172363,
      "step": 5904
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 5904,
      "training_step_time": 9.9379301071167
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 3.135251522064209,
      "step": 5904
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 5904,
      "training_step_time": 9.902017593383789
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 3.423046827316284,
      "step": 5904
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 5904,
      "training_step_time": 10.185804605484009
    },
    {
      "epoch": 2.252197265625e-05,
      "model_forward_time": 3.1290442943573,
      "step": 5904
    },
    {
      "epoch": 2.252197265625e-05,
      "step": 5904,
      "training_step_time": 10.254949808120728
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 3.1710236072540283,
      "step": 5905
    },
    {
      "dynamics_loss_avg": 0.05111750401556492,
      "epoch": 2.2525787353515626e-05,
      "step": 5905
    },
    {
      "action_loss_avg": 0.022097612079232932,
      "epoch": 2.2525787353515626e-05,
      "step": 5905
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 5905,
      "training_step_time": 9.938686609268188
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 3.133673667907715,
      "step": 5905
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 5905,
      "training_step_time": 9.898450136184692
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 3.1300766468048096,
      "step": 5905
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 5905,
      "training_step_time": 9.892968893051147
    },
    {
      "epoch": 2.2525787353515626e-05,
      "model_forward_time": 3.1347551345825195,
      "step": 5905
    },
    {
      "epoch": 2.2525787353515626e-05,
      "step": 5905,
      "training_step_time": 10.25905990600586
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 3.5094079971313477,
      "step": 5906
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 5906,
      "training_step_time": 10.27464771270752
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 3.1369338035583496,
      "step": 5906
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 5906,
      "training_step_time": 9.90515398979187
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 3.1343812942504883,
      "step": 5906
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 5906,
      "training_step_time": 9.90690803527832
    },
    {
      "epoch": 2.252960205078125e-05,
      "model_forward_time": 3.1382064819335938,
      "step": 5906
    },
    {
      "epoch": 2.252960205078125e-05,
      "step": 5906,
      "training_step_time": 10.032351732254028
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 3.4149281978607178,
      "step": 5907
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 5907,
      "training_step_time": 10.18640661239624
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 2.4350380897521973,
      "step": 5907
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 5907,
      "training_step_time": 7.46689510345459
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 2.432659864425659,
      "step": 5907
    },
    {
      "dynamics_loss_avg": 0.06400298792868853,
      "epoch": 2.2533416748046875e-05,
      "step": 5907
    },
    {
      "action_loss_avg": 0.023996564745903014,
      "epoch": 2.2533416748046875e-05,
      "step": 5907
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 5907,
      "training_step_time": 7.470509767532349
    },
    {
      "epoch": 2.2533416748046875e-05,
      "model_forward_time": 3.134716033935547,
      "step": 5907
    },
    {
      "epoch": 2.2533416748046875e-05,
      "step": 5907,
      "training_step_time": 10.29090428352356
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 3.175173044204712,
      "step": 5908
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 5908,
      "training_step_time": 9.95405888557434
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 3.137587308883667,
      "step": 5908
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 5908,
      "training_step_time": 9.908907651901245
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 3.3921830654144287,
      "step": 5908
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 5908,
      "training_step_time": 10.162290573120117
    },
    {
      "epoch": 2.25372314453125e-05,
      "model_forward_time": 3.1403615474700928,
      "step": 5908
    },
    {
      "epoch": 2.25372314453125e-05,
      "step": 5908,
      "training_step_time": 10.29711103439331
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 3.170287847518921,
      "step": 5909
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 5909,
      "training_step_time": 9.944060325622559
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 3.1318795680999756,
      "step": 5909
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 5909,
      "training_step_time": 9.903387784957886
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 3.1461403369903564,
      "step": 5909
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 5909,
      "training_step_time": 9.913215398788452
    },
    {
      "epoch": 2.2541046142578124e-05,
      "model_forward_time": 3.3969345092773438,
      "step": 5909
    },
    {
      "epoch": 2.2541046142578124e-05,
      "step": 5909,
      "training_step_time": 10.317935228347778
    },
    {
      "epoch": 2.254486083984375e-05,
      "grad_norm": 0.46799567341804504,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 5910
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 3.158233880996704,
      "step": 5910
    },
    {
      "dynamics_loss_avg": 0.05380944134667516,
      "epoch": 2.254486083984375e-05,
      "step": 5910
    },
    {
      "action_loss_avg": 0.023021534411236645,
      "epoch": 2.254486083984375e-05,
      "step": 5910
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 5910,
      "training_step_time": 9.928781986236572
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 3.1269683837890625,
      "step": 5910
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 5910,
      "training_step_time": 9.89335298538208
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 3.133742570877075,
      "step": 5910
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 5910,
      "training_step_time": 9.89944314956665
    },
    {
      "epoch": 2.254486083984375e-05,
      "model_forward_time": 3.41221284866333,
      "step": 5910
    },
    {
      "epoch": 2.254486083984375e-05,
      "step": 5910,
      "training_step_time": 10.292762041091919
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 3.168383836746216,
      "step": 5911
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 5911,
      "training_step_time": 9.94071912765503
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 3.1326427459716797,
      "step": 5911
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 5911,
      "training_step_time": 9.901926755905151
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 3.137937068939209,
      "step": 5911
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 5911,
      "training_step_time": 9.90587830543518
    },
    {
      "epoch": 2.2548675537109373e-05,
      "model_forward_time": 3.1346287727355957,
      "step": 5911
    },
    {
      "epoch": 2.2548675537109373e-05,
      "step": 5911,
      "training_step_time": 10.02468729019165
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 3.143071413040161,
      "step": 5912
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 5912,
      "training_step_time": 10.187231063842773
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 3.1338462829589844,
      "step": 5912
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 5912,
      "training_step_time": 9.908024072647095
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 2.434587001800537,
      "step": 5912
    },
    {
      "dynamics_loss_avg": 0.04661061745136976,
      "epoch": 2.2552490234375e-05,
      "step": 5912
    },
    {
      "action_loss_avg": 0.021351417154073717,
      "epoch": 2.2552490234375e-05,
      "step": 5912
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 5912,
      "training_step_time": 7.467271089553833
    },
    {
      "epoch": 2.2552490234375e-05,
      "model_forward_time": 3.129478931427002,
      "step": 5912
    },
    {
      "epoch": 2.2552490234375e-05,
      "step": 5912,
      "training_step_time": 10.308465719223022
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 3.1660141944885254,
      "step": 5913
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 5913,
      "training_step_time": 9.936107873916626
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 2.4288909435272217,
      "step": 5913
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 5913,
      "training_step_time": 7.454730033874512
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 3.3924314975738525,
      "step": 5913
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 5913,
      "training_step_time": 10.167155027389526
    },
    {
      "epoch": 2.2556304931640626e-05,
      "model_forward_time": 3.1350247859954834,
      "step": 5913
    },
    {
      "epoch": 2.2556304931640626e-05,
      "step": 5913,
      "training_step_time": 10.296783685684204
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 3.2012815475463867,
      "step": 5914
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 5914,
      "training_step_time": 9.986013650894165
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 3.140197992324829,
      "step": 5914
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 5914,
      "training_step_time": 9.909974575042725
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 3.1386919021606445,
      "step": 5914
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 5914,
      "training_step_time": 10.184325218200684
    },
    {
      "epoch": 2.256011962890625e-05,
      "model_forward_time": 3.1398749351501465,
      "step": 5914
    },
    {
      "epoch": 2.256011962890625e-05,
      "step": 5914,
      "training_step_time": 10.316476106643677
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 3.148085355758667,
      "step": 5915
    },
    {
      "dynamics_loss_avg": 0.05510725099593401,
      "epoch": 2.2563934326171875e-05,
      "step": 5915
    },
    {
      "action_loss_avg": 0.01970697659999132,
      "epoch": 2.2563934326171875e-05,
      "step": 5915
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 5915,
      "training_step_time": 9.920240640640259
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 3.139817476272583,
      "step": 5915
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 5915,
      "training_step_time": 9.907867908477783
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 2.43937349319458,
      "step": 5915
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 5915,
      "training_step_time": 7.466958045959473
    },
    {
      "epoch": 2.2563934326171875e-05,
      "model_forward_time": 3.446444272994995,
      "step": 5915
    },
    {
      "epoch": 2.2563934326171875e-05,
      "step": 5915,
      "training_step_time": 10.306697607040405
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 3.15368914604187,
      "step": 5916
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 5916,
      "training_step_time": 9.920890092849731
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 3.128682851791382,
      "step": 5916
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 5916,
      "training_step_time": 9.901895761489868
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 3.134739398956299,
      "step": 5916
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 5916,
      "training_step_time": 9.913107633590698
    },
    {
      "epoch": 2.25677490234375e-05,
      "model_forward_time": 3.137441635131836,
      "step": 5916
    },
    {
      "epoch": 2.25677490234375e-05,
      "step": 5916,
      "training_step_time": 10.028771162033081
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 3.168837785720825,
      "step": 5917
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 5917,
      "training_step_time": 10.216917514801025
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 3.1338789463043213,
      "step": 5917
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 5917,
      "training_step_time": 9.902870893478394
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 3.127866506576538,
      "step": 5917
    },
    {
      "dynamics_loss_avg": 0.061757024750113486,
      "epoch": 2.2571563720703124e-05,
      "step": 5917
    },
    {
      "action_loss_avg": 0.032115024514496326,
      "epoch": 2.2571563720703124e-05,
      "step": 5917
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 5917,
      "training_step_time": 9.902818441390991
    },
    {
      "epoch": 2.2571563720703124e-05,
      "model_forward_time": 3.1328930854797363,
      "step": 5917
    },
    {
      "epoch": 2.2571563720703124e-05,
      "step": 5917,
      "training_step_time": 10.251766204833984
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 3.188049793243408,
      "step": 5918
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 5918,
      "training_step_time": 9.959840059280396
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 3.437840223312378,
      "step": 5918
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 5918,
      "training_step_time": 10.207421064376831
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 3.1298112869262695,
      "step": 5918
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 5918,
      "training_step_time": 9.898027420043945
    },
    {
      "epoch": 2.257537841796875e-05,
      "model_forward_time": 3.129000425338745,
      "step": 5918
    },
    {
      "epoch": 2.257537841796875e-05,
      "step": 5918,
      "training_step_time": 10.2931227684021
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 3.168246269226074,
      "step": 5919
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 5919,
      "training_step_time": 9.931809186935425
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 3.1297976970672607,
      "step": 5919
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 5919,
      "training_step_time": 9.897261619567871
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 3.3815011978149414,
      "step": 5919
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 5919,
      "training_step_time": 10.144177913665771
    },
    {
      "epoch": 2.2579193115234373e-05,
      "model_forward_time": 3.134366989135742,
      "step": 5919
    },
    {
      "epoch": 2.2579193115234373e-05,
      "step": 5919,
      "training_step_time": 10.27332592010498
    },
    {
      "epoch": 2.25830078125e-05,
      "grad_norm": 0.5763951539993286,
      "learning_rate": 1e-05,
      "loss": 0.081,
      "step": 5920
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 3.1818602085113525,
      "step": 5920
    },
    {
      "dynamics_loss_avg": 0.05984009765088558,
      "epoch": 2.25830078125e-05,
      "step": 5920
    },
    {
      "action_loss_avg": 0.031670233234763145,
      "epoch": 2.25830078125e-05,
      "step": 5920
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 5920,
      "training_step_time": 9.9508535861969
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 3.137237787246704,
      "step": 5920
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 5920,
      "training_step_time": 9.904509544372559
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 3.1412739753723145,
      "step": 5920
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 5920,
      "training_step_time": 9.908264636993408
    },
    {
      "epoch": 2.25830078125e-05,
      "model_forward_time": 3.4275615215301514,
      "step": 5920
    },
    {
      "epoch": 2.25830078125e-05,
      "step": 5920,
      "training_step_time": 10.289060831069946
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 3.1516144275665283,
      "step": 5921
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 5921,
      "training_step_time": 9.927802801132202
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 3.1330339908599854,
      "step": 5921
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 5921,
      "training_step_time": 9.904911041259766
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 3.1392123699188232,
      "step": 5921
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 5921,
      "training_step_time": 9.913427114486694
    },
    {
      "epoch": 2.2586822509765626e-05,
      "model_forward_time": 3.1379079818725586,
      "step": 5921
    },
    {
      "epoch": 2.2586822509765626e-05,
      "step": 5921,
      "training_step_time": 10.29119610786438
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 3.504805564880371,
      "step": 5922
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 5922,
      "training_step_time": 10.287581205368042
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 3.1411142349243164,
      "step": 5922
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 5922,
      "training_step_time": 9.913909435272217
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 3.1403985023498535,
      "step": 5922
    },
    {
      "dynamics_loss_avg": 0.06058562537655234,
      "epoch": 2.259063720703125e-05,
      "step": 5922
    },
    {
      "action_loss_avg": 0.023224777774885297,
      "epoch": 2.259063720703125e-05,
      "step": 5922
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 5922,
      "training_step_time": 9.912616491317749
    },
    {
      "epoch": 2.259063720703125e-05,
      "model_forward_time": 3.1365127563476562,
      "step": 5922
    },
    {
      "epoch": 2.259063720703125e-05,
      "step": 5922,
      "training_step_time": 10.014021396636963
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 3.1883487701416016,
      "step": 5923
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 5923,
      "training_step_time": 9.9573073387146
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 3.399723768234253,
      "step": 5923
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 5923,
      "training_step_time": 10.165011167526245
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 3.1448678970336914,
      "step": 5923
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 5923,
      "training_step_time": 9.915773391723633
    },
    {
      "epoch": 2.2594451904296875e-05,
      "model_forward_time": 3.1387267112731934,
      "step": 5923
    },
    {
      "epoch": 2.2594451904296875e-05,
      "step": 5923,
      "training_step_time": 10.289162397384644
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 3.155782461166382,
      "step": 5924
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 5924,
      "training_step_time": 9.921544790267944
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 3.1361806392669678,
      "step": 5924
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 5924,
      "training_step_time": 9.902875661849976
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 3.414844274520874,
      "step": 5924
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 5924,
      "training_step_time": 10.177265644073486
    },
    {
      "epoch": 2.25982666015625e-05,
      "model_forward_time": 2.4354653358459473,
      "step": 5924
    },
    {
      "epoch": 2.25982666015625e-05,
      "step": 5924,
      "training_step_time": 10.330002307891846
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 3.170330047607422,
      "step": 5925
    },
    {
      "dynamics_loss_avg": 0.06816296726465225,
      "epoch": 2.2602081298828124e-05,
      "step": 5925
    },
    {
      "action_loss_avg": 0.03149287812411785,
      "epoch": 2.2602081298828124e-05,
      "step": 5925
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 5925,
      "training_step_time": 9.953857660293579
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 3.1345865726470947,
      "step": 5925
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 5925,
      "training_step_time": 9.902377605438232
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 3.1335017681121826,
      "step": 5925
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 5925,
      "training_step_time": 9.900471448898315
    },
    {
      "epoch": 2.2602081298828124e-05,
      "model_forward_time": 3.137720823287964,
      "step": 5925
    },
    {
      "epoch": 2.2602081298828124e-05,
      "step": 5925,
      "training_step_time": 10.269528865814209
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 3.1731302738189697,
      "step": 5926
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 5926,
      "training_step_time": 9.942782402038574
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 3.1371653079986572,
      "step": 5926
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 5926,
      "training_step_time": 9.91134762763977
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 3.1388349533081055,
      "step": 5926
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 5926,
      "training_step_time": 9.912344932556152
    },
    {
      "epoch": 2.260589599609375e-05,
      "model_forward_time": 3.1466763019561768,
      "step": 5926
    },
    {
      "epoch": 2.260589599609375e-05,
      "step": 5926,
      "training_step_time": 10.301193714141846
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 3.471052885055542,
      "step": 5927
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 5927,
      "training_step_time": 10.243458032608032
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 3.1451163291931152,
      "step": 5927
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 5927,
      "training_step_time": 9.91771674156189
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 3.1356050968170166,
      "step": 5927
    },
    {
      "dynamics_loss_avg": 0.05981452353298664,
      "epoch": 2.2609710693359373e-05,
      "step": 5927
    },
    {
      "action_loss_avg": 0.02656025355681777,
      "epoch": 2.2609710693359373e-05,
      "step": 5927
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 5927,
      "training_step_time": 9.913112878799438
    },
    {
      "epoch": 2.2609710693359373e-05,
      "model_forward_time": 3.143270969390869,
      "step": 5927
    },
    {
      "epoch": 2.2609710693359373e-05,
      "step": 5927,
      "training_step_time": 10.30996298789978
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 3.160513162612915,
      "step": 5928
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 5928,
      "training_step_time": 9.939880132675171
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 3.43496036529541,
      "step": 5928
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 5928,
      "training_step_time": 10.203922986984253
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 3.1372525691986084,
      "step": 5928
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 5928,
      "training_step_time": 9.904408693313599
    },
    {
      "epoch": 2.2613525390625e-05,
      "model_forward_time": 2.435137987136841,
      "step": 5928
    },
    {
      "epoch": 2.2613525390625e-05,
      "step": 5928,
      "training_step_time": 10.256727457046509
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 3.155388832092285,
      "step": 5929
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 5929,
      "training_step_time": 9.925344705581665
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 3.1367852687835693,
      "step": 5929
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 5929,
      "training_step_time": 9.903988361358643
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 2.4356021881103516,
      "step": 5929
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 5929,
      "training_step_time": 7.460078001022339
    },
    {
      "epoch": 2.2617340087890626e-05,
      "model_forward_time": 3.410067081451416,
      "step": 5929
    },
    {
      "epoch": 2.2617340087890626e-05,
      "step": 5929,
      "training_step_time": 10.285609006881714
    },
    {
      "epoch": 2.262115478515625e-05,
      "grad_norm": 0.5917835831642151,
      "learning_rate": 1e-05,
      "loss": 0.087,
      "step": 5930
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 2.4704184532165527,
      "step": 5930
    },
    {
      "dynamics_loss_avg": 0.06346264220774174,
      "epoch": 2.262115478515625e-05,
      "step": 5930
    },
    {
      "action_loss_avg": 0.02711592186242342,
      "epoch": 2.262115478515625e-05,
      "step": 5930
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 5930,
      "training_step_time": 7.500679969787598
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 3.1318089962005615,
      "step": 5930
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 5930,
      "training_step_time": 9.897513151168823
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 3.140220880508423,
      "step": 5930
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 5930,
      "training_step_time": 9.90830683708191
    },
    {
      "epoch": 2.262115478515625e-05,
      "model_forward_time": 3.1347155570983887,
      "step": 5930
    },
    {
      "epoch": 2.262115478515625e-05,
      "step": 5930,
      "training_step_time": 10.277604341506958
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 3.17010760307312,
      "step": 5931
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 5931,
      "training_step_time": 9.936843156814575
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 3.133847236633301,
      "step": 5931
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 5931,
      "training_step_time": 9.898287296295166
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 3.1406474113464355,
      "step": 5931
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 5931,
      "training_step_time": 9.909409523010254
    },
    {
      "epoch": 2.2624969482421875e-05,
      "model_forward_time": 3.131767511367798,
      "step": 5931
    },
    {
      "epoch": 2.2624969482421875e-05,
      "step": 5931,
      "training_step_time": 10.002002477645874
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 2.4758975505828857,
      "step": 5932
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 5932,
      "training_step_time": 7.7670512199401855
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 3.1344175338745117,
      "step": 5932
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 5932,
      "training_step_time": 9.903562307357788
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 3.137282371520996,
      "step": 5932
    },
    {
      "dynamics_loss_avg": 0.050220455043017864,
      "epoch": 2.26287841796875e-05,
      "step": 5932
    },
    {
      "action_loss_avg": 0.019598629511892795,
      "epoch": 2.26287841796875e-05,
      "step": 5932
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 5932,
      "training_step_time": 9.915424108505249
    },
    {
      "epoch": 2.26287841796875e-05,
      "model_forward_time": 3.1355884075164795,
      "step": 5932
    },
    {
      "epoch": 2.26287841796875e-05,
      "step": 5932,
      "training_step_time": 10.272250890731812
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 3.1817426681518555,
      "step": 5933
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 5933,
      "training_step_time": 9.950191259384155
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 3.4004766941070557,
      "step": 5933
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 5933,
      "training_step_time": 10.16942548751831
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 3.141369342803955,
      "step": 5933
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 5933,
      "training_step_time": 9.90636658668518
    },
    {
      "epoch": 2.2632598876953124e-05,
      "model_forward_time": 3.1336288452148438,
      "step": 5933
    },
    {
      "epoch": 2.2632598876953124e-05,
      "step": 5933,
      "training_step_time": 10.255784034729004
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 3.143714427947998,
      "step": 5934
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 5934,
      "training_step_time": 9.910726547241211
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 3.1374974250793457,
      "step": 5934
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 5934,
      "training_step_time": 9.903296947479248
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 3.4504013061523438,
      "step": 5934
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 5934,
      "training_step_time": 10.21696662902832
    },
    {
      "epoch": 2.263641357421875e-05,
      "model_forward_time": 3.1370739936828613,
      "step": 5934
    },
    {
      "epoch": 2.263641357421875e-05,
      "step": 5934,
      "training_step_time": 10.3085618019104
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 3.1663477420806885,
      "step": 5935
    },
    {
      "dynamics_loss_avg": 0.05293248668313026,
      "epoch": 2.2640228271484374e-05,
      "step": 5935
    },
    {
      "action_loss_avg": 0.028250162629410623,
      "epoch": 2.2640228271484374e-05,
      "step": 5935
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 5935,
      "training_step_time": 9.939422845840454
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 3.1300718784332275,
      "step": 5935
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 5935,
      "training_step_time": 9.893926620483398
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 3.128793954849243,
      "step": 5935
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 5935,
      "training_step_time": 9.892987966537476
    },
    {
      "epoch": 2.2640228271484374e-05,
      "model_forward_time": 3.3988466262817383,
      "step": 5935
    },
    {
      "epoch": 2.2640228271484374e-05,
      "step": 5935,
      "training_step_time": 10.260809421539307
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 3.16447114944458,
      "step": 5936
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 5936,
      "training_step_time": 9.935372114181519
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 3.1341798305511475,
      "step": 5936
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 5936,
      "training_step_time": 9.901720523834229
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 3.1322789192199707,
      "step": 5936
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 5936,
      "training_step_time": 9.90302562713623
    },
    {
      "epoch": 2.264404296875e-05,
      "model_forward_time": 3.136496067047119,
      "step": 5936
    },
    {
      "epoch": 2.264404296875e-05,
      "step": 5936,
      "training_step_time": 10.010619878768921
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 3.440617799758911,
      "step": 5937
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 5937,
      "training_step_time": 10.229743242263794
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 3.142022132873535,
      "step": 5937
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 5937,
      "training_step_time": 9.919732332229614
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 3.143702268600464,
      "step": 5937
    },
    {
      "dynamics_loss_avg": 0.06712343599647283,
      "epoch": 2.2647857666015626e-05,
      "step": 5937
    },
    {
      "action_loss_avg": 0.027915679290890692,
      "epoch": 2.2647857666015626e-05,
      "step": 5937
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 5937,
      "training_step_time": 9.930651187896729
    },
    {
      "epoch": 2.2647857666015626e-05,
      "model_forward_time": 3.145979642868042,
      "step": 5937
    },
    {
      "epoch": 2.2647857666015626e-05,
      "step": 5937,
      "training_step_time": 10.278711795806885
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 3.183708429336548,
      "step": 5938
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 5938,
      "training_step_time": 9.960279703140259
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 2.4377341270446777,
      "step": 5938
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 5938,
      "training_step_time": 7.748867750167847
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 3.1342782974243164,
      "step": 5938
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 5938,
      "training_step_time": 9.911005735397339
    },
    {
      "epoch": 2.265167236328125e-05,
      "model_forward_time": 3.1442291736602783,
      "step": 5938
    },
    {
      "epoch": 2.265167236328125e-05,
      "step": 5938,
      "training_step_time": 10.277419328689575
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 3.152287483215332,
      "step": 5939
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 5939,
      "training_step_time": 9.927100419998169
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 3.132657051086426,
      "step": 5939
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 5939,
      "training_step_time": 9.90198540687561
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 3.403019666671753,
      "step": 5939
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 5939,
      "training_step_time": 10.169502258300781
    },
    {
      "epoch": 2.2655487060546875e-05,
      "model_forward_time": 2.4326324462890625,
      "step": 5939
    },
    {
      "epoch": 2.2655487060546875e-05,
      "step": 5939,
      "training_step_time": 10.277230978012085
    },
    {
      "epoch": 2.26593017578125e-05,
      "grad_norm": 0.48908689618110657,
      "learning_rate": 1e-05,
      "loss": 0.0877,
      "step": 5940
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 3.1544718742370605,
      "step": 5940
    },
    {
      "dynamics_loss_avg": 0.05605257246643305,
      "epoch": 2.26593017578125e-05,
      "step": 5940
    },
    {
      "action_loss_avg": 0.02527327658608556,
      "epoch": 2.26593017578125e-05,
      "step": 5940
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 5940,
      "training_step_time": 9.917409181594849
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 3.131923198699951,
      "step": 5940
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 5940,
      "training_step_time": 9.898554563522339
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 2.434479236602783,
      "step": 5940
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 5940,
      "training_step_time": 7.456927299499512
    },
    {
      "epoch": 2.26593017578125e-05,
      "model_forward_time": 2.4318196773529053,
      "step": 5940
    },
    {
      "epoch": 2.26593017578125e-05,
      "step": 5940,
      "training_step_time": 10.275794506072998
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 2.7212820053100586,
      "step": 5941
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 5941,
      "training_step_time": 7.746516466140747
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 3.1306376457214355,
      "step": 5941
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 5941,
      "training_step_time": 9.895200252532959
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 3.136969804763794,
      "step": 5941
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 5941,
      "training_step_time": 9.911307573318481
    },
    {
      "epoch": 2.2663116455078124e-05,
      "model_forward_time": 3.136810541152954,
      "step": 5941
    },
    {
      "epoch": 2.2663116455078124e-05,
      "step": 5941,
      "training_step_time": 10.017435073852539
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 3.1770966053009033,
      "step": 5942
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 5942,
      "training_step_time": 9.946887969970703
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 3.38131046295166,
      "step": 5942
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 5942,
      "training_step_time": 10.147253513336182
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 3.1323540210723877,
      "step": 5942
    },
    {
      "dynamics_loss_avg": 0.04764385335147381,
      "epoch": 2.266693115234375e-05,
      "step": 5942
    },
    {
      "action_loss_avg": 0.017740247142501175,
      "epoch": 2.266693115234375e-05,
      "step": 5942
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 5942,
      "training_step_time": 9.900051832199097
    },
    {
      "epoch": 2.266693115234375e-05,
      "model_forward_time": 3.1338539123535156,
      "step": 5942
    },
    {
      "epoch": 2.266693115234375e-05,
      "step": 5942,
      "training_step_time": 10.002317428588867
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 3.1645264625549316,
      "step": 5943
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 5943,
      "training_step_time": 9.931387424468994
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 3.136777400970459,
      "step": 5943
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 5943,
      "training_step_time": 9.908653020858765
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 3.4032859802246094,
      "step": 5943
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 5943,
      "training_step_time": 10.171478271484375
    },
    {
      "epoch": 2.2670745849609374e-05,
      "model_forward_time": 3.1363794803619385,
      "step": 5943
    },
    {
      "epoch": 2.2670745849609374e-05,
      "step": 5943,
      "training_step_time": 10.269160270690918
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 3.1744132041931152,
      "step": 5944
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 5944,
      "training_step_time": 9.941941738128662
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 3.1329736709594727,
      "step": 5944
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 5944,
      "training_step_time": 9.902608871459961
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 3.4086318016052246,
      "step": 5944
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 5944,
      "training_step_time": 10.180128335952759
    },
    {
      "epoch": 2.2674560546875e-05,
      "model_forward_time": 3.1308727264404297,
      "step": 5944
    },
    {
      "epoch": 2.2674560546875e-05,
      "step": 5944,
      "training_step_time": 10.268165826797485
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 3.155458927154541,
      "step": 5945
    },
    {
      "dynamics_loss_avg": 0.06005691047757864,
      "epoch": 2.2678375244140626e-05,
      "step": 5945
    },
    {
      "action_loss_avg": 0.027986391074955465,
      "epoch": 2.2678375244140626e-05,
      "step": 5945
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 5945,
      "training_step_time": 9.931030988693237
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 3.134631633758545,
      "step": 5945
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 5945,
      "training_step_time": 9.910655975341797
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 3.1340296268463135,
      "step": 5945
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 5945,
      "training_step_time": 9.906259775161743
    },
    {
      "epoch": 2.2678375244140626e-05,
      "model_forward_time": 3.1361517906188965,
      "step": 5945
    },
    {
      "epoch": 2.2678375244140626e-05,
      "step": 5945,
      "training_step_time": 10.333248853683472
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 3.502103328704834,
      "step": 5946
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 5946,
      "training_step_time": 10.279664993286133
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 3.1395201683044434,
      "step": 5946
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 5946,
      "training_step_time": 9.914325714111328
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 3.1361606121063232,
      "step": 5946
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 5946,
      "training_step_time": 9.91406798362732
    },
    {
      "epoch": 2.268218994140625e-05,
      "model_forward_time": 3.1350390911102295,
      "step": 5946
    },
    {
      "epoch": 2.268218994140625e-05,
      "step": 5946,
      "training_step_time": 10.031684875488281
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 3.4517104625701904,
      "step": 5947
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 5947,
      "training_step_time": 10.23662543296814
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 3.1436920166015625,
      "step": 5947
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 5947,
      "training_step_time": 9.920814514160156
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 3.1386828422546387,
      "step": 5947
    },
    {
      "dynamics_loss_avg": 0.056245225667953494,
      "epoch": 2.2686004638671875e-05,
      "step": 5947
    },
    {
      "action_loss_avg": 0.021071741729974745,
      "epoch": 2.2686004638671875e-05,
      "step": 5947
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 5947,
      "training_step_time": 9.915869951248169
    },
    {
      "epoch": 2.2686004638671875e-05,
      "model_forward_time": 3.136465072631836,
      "step": 5947
    },
    {
      "epoch": 2.2686004638671875e-05,
      "step": 5947,
      "training_step_time": 10.003884077072144
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 3.1692564487457275,
      "step": 5948
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 5948,
      "training_step_time": 9.94029951095581
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 3.127350330352783,
      "step": 5948
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 5948,
      "training_step_time": 10.180224657058716
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 3.1300041675567627,
      "step": 5948
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 5948,
      "training_step_time": 9.89699101448059
    },
    {
      "epoch": 2.26898193359375e-05,
      "model_forward_time": 3.131908655166626,
      "step": 5948
    },
    {
      "epoch": 2.26898193359375e-05,
      "step": 5948,
      "training_step_time": 10.006035804748535
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 3.147150993347168,
      "step": 5949
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 5949,
      "training_step_time": 9.915655612945557
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 3.130575656890869,
      "step": 5949
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 5949,
      "training_step_time": 9.90254521369934
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 3.136627197265625,
      "step": 5949
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 5949,
      "training_step_time": 10.190661907196045
    },
    {
      "epoch": 2.2693634033203124e-05,
      "model_forward_time": 2.432370662689209,
      "step": 5949
    },
    {
      "epoch": 2.2693634033203124e-05,
      "step": 5949,
      "training_step_time": 10.298167705535889
    },
    {
      "epoch": 2.269744873046875e-05,
      "grad_norm": 0.6351513862609863,
      "learning_rate": 1e-05,
      "loss": 0.0848,
      "step": 5950
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 3.145102024078369,
      "step": 5950
    },
    {
      "dynamics_loss_avg": 0.05064706765115261,
      "epoch": 2.269744873046875e-05,
      "step": 5950
    },
    {
      "action_loss_avg": 0.020331720355898142,
      "epoch": 2.269744873046875e-05,
      "step": 5950
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 5950,
      "training_step_time": 9.925472259521484
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 3.1362085342407227,
      "step": 5950
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 5950,
      "training_step_time": 9.911110639572144
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 3.1426708698272705,
      "step": 5950
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 5950,
      "training_step_time": 9.92432427406311
    },
    {
      "epoch": 2.269744873046875e-05,
      "model_forward_time": 3.1397037506103516,
      "step": 5950
    },
    {
      "epoch": 2.269744873046875e-05,
      "step": 5950,
      "training_step_time": 10.330008268356323
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 3.1765663623809814,
      "step": 5951
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 5951,
      "training_step_time": 9.953291416168213
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 3.132852554321289,
      "step": 5951
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 5951,
      "training_step_time": 9.905513286590576
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 3.1344027519226074,
      "step": 5951
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 5951,
      "training_step_time": 9.910138368606567
    },
    {
      "epoch": 2.2701263427734374e-05,
      "model_forward_time": 3.134467363357544,
      "step": 5951
    },
    {
      "epoch": 2.2701263427734374e-05,
      "step": 5951,
      "training_step_time": 10.026601791381836
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 3.1705708503723145,
      "step": 5952
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 5952,
      "training_step_time": 10.217168092727661
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 3.12678861618042,
      "step": 5952
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 5952,
      "training_step_time": 9.895824670791626
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 3.134481906890869,
      "step": 5952
    },
    {
      "dynamics_loss_avg": 0.05205999370664358,
      "epoch": 2.2705078125e-05,
      "step": 5952
    },
    {
      "action_loss_avg": 0.023207604605704545,
      "epoch": 2.2705078125e-05,
      "step": 5952
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 5952,
      "training_step_time": 9.907817363739014
    },
    {
      "epoch": 2.2705078125e-05,
      "model_forward_time": 3.127692937850952,
      "step": 5952
    },
    {
      "epoch": 2.2705078125e-05,
      "step": 5952,
      "training_step_time": 10.029927253723145
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 3.1571922302246094,
      "step": 5953
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 5953,
      "training_step_time": 9.928371906280518
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 2.429380178451538,
      "step": 5953
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 5953,
      "training_step_time": 7.725724697113037
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 3.1299116611480713,
      "step": 5953
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 5953,
      "training_step_time": 9.892725706100464
    },
    {
      "epoch": 2.2708892822265626e-05,
      "model_forward_time": 3.129056930541992,
      "step": 5953
    },
    {
      "epoch": 2.2708892822265626e-05,
      "step": 5953,
      "training_step_time": 10.021605730056763
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 2.4654529094696045,
      "step": 5954
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 5954,
      "training_step_time": 7.4850239753723145
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 3.123478412628174,
      "step": 5954
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 5954,
      "training_step_time": 9.888713598251343
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 3.126884937286377,
      "step": 5954
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 5954,
      "training_step_time": 9.889533042907715
    },
    {
      "epoch": 2.271270751953125e-05,
      "model_forward_time": 3.382291793823242,
      "step": 5954
    },
    {
      "epoch": 2.271270751953125e-05,
      "step": 5954,
      "training_step_time": 10.303441762924194
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 3.1678545475006104,
      "step": 5955
    },
    {
      "dynamics_loss_avg": 0.057502823509275915,
      "epoch": 2.2716522216796875e-05,
      "step": 5955
    },
    {
      "action_loss_avg": 0.03217382594011724,
      "epoch": 2.2716522216796875e-05,
      "step": 5955
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 5955,
      "training_step_time": 9.934805154800415
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 3.1305136680603027,
      "step": 5955
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 5955,
      "training_step_time": 9.894649267196655
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 3.1325767040252686,
      "step": 5955
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 5955,
      "training_step_time": 9.895690441131592
    },
    {
      "epoch": 2.2716522216796875e-05,
      "model_forward_time": 3.4143409729003906,
      "step": 5955
    },
    {
      "epoch": 2.2716522216796875e-05,
      "step": 5955,
      "training_step_time": 10.343381643295288
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 3.191765069961548,
      "step": 5956
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 5956,
      "training_step_time": 9.961565732955933
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 3.131880283355713,
      "step": 5956
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 5956,
      "training_step_time": 9.901395082473755
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 3.1352646350860596,
      "step": 5956
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 5956,
      "training_step_time": 9.905810117721558
    },
    {
      "epoch": 2.27203369140625e-05,
      "model_forward_time": 3.135411024093628,
      "step": 5956
    },
    {
      "epoch": 2.27203369140625e-05,
      "step": 5956,
      "training_step_time": 10.048574686050415
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 3.16020131111145,
      "step": 5957
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 5957,
      "training_step_time": 10.231838464736938
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 3.1500120162963867,
      "step": 5957
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 5957,
      "training_step_time": 9.924875974655151
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 3.142944812774658,
      "step": 5957
    },
    {
      "dynamics_loss_avg": 0.056306453235447404,
      "epoch": 2.2724151611328124e-05,
      "step": 5957
    },
    {
      "action_loss_avg": 0.028636434860527516,
      "epoch": 2.2724151611328124e-05,
      "step": 5957
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 5957,
      "training_step_time": 9.918617725372314
    },
    {
      "epoch": 2.2724151611328124e-05,
      "model_forward_time": 3.1450226306915283,
      "step": 5957
    },
    {
      "epoch": 2.2724151611328124e-05,
      "step": 5957,
      "training_step_time": 10.03487753868103
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 3.1533827781677246,
      "step": 5958
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 5958,
      "training_step_time": 9.929322481155396
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 3.4284543991088867,
      "step": 5958
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 5958,
      "training_step_time": 10.196576833724976
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 3.138369083404541,
      "step": 5958
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 5958,
      "training_step_time": 9.907348155975342
    },
    {
      "epoch": 2.272796630859375e-05,
      "model_forward_time": 3.1423559188842773,
      "step": 5958
    },
    {
      "epoch": 2.272796630859375e-05,
      "step": 5958,
      "training_step_time": 10.011655330657959
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 2.475722551345825,
      "step": 5959
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 5959,
      "training_step_time": 7.4989259243011475
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 3.13248348236084,
      "step": 5959
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 5959,
      "training_step_time": 9.898682117462158
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 3.402216672897339,
      "step": 5959
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 5959,
      "training_step_time": 10.16821575164795
    },
    {
      "epoch": 2.2731781005859374e-05,
      "model_forward_time": 3.1393415927886963,
      "step": 5959
    },
    {
      "epoch": 2.2731781005859374e-05,
      "step": 5959,
      "training_step_time": 10.248649597167969
    },
    {
      "epoch": 2.2735595703125e-05,
      "grad_norm": 0.49239885807037354,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 5960
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 2.4515249729156494,
      "step": 5960
    },
    {
      "dynamics_loss_avg": 0.07204680666327476,
      "epoch": 2.2735595703125e-05,
      "step": 5960
    },
    {
      "action_loss_avg": 0.03189355339854956,
      "epoch": 2.2735595703125e-05,
      "step": 5960
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 5960,
      "training_step_time": 7.487206935882568
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 3.139559030532837,
      "step": 5960
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 5960,
      "training_step_time": 9.917786836624146
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 3.1483547687530518,
      "step": 5960
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 5960,
      "training_step_time": 9.932432174682617
    },
    {
      "epoch": 2.2735595703125e-05,
      "model_forward_time": 2.435889959335327,
      "step": 5960
    },
    {
      "epoch": 2.2735595703125e-05,
      "step": 5960,
      "training_step_time": 10.339085817337036
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 3.488633871078491,
      "step": 5961
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 5961,
      "training_step_time": 10.27414321899414
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 3.143038511276245,
      "step": 5961
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 5961,
      "training_step_time": 9.921443700790405
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 3.1455163955688477,
      "step": 5961
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 5961,
      "training_step_time": 9.92811393737793
    },
    {
      "epoch": 2.2739410400390626e-05,
      "model_forward_time": 3.1424407958984375,
      "step": 5961
    },
    {
      "epoch": 2.2739410400390626e-05,
      "step": 5961,
      "training_step_time": 10.27500581741333
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 2.479069471359253,
      "step": 5962
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 5962,
      "training_step_time": 7.784763813018799
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 2.4317142963409424,
      "step": 5962
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 5962,
      "training_step_time": 7.457749128341675
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 3.132749319076538,
      "step": 5962
    },
    {
      "dynamics_loss_avg": 0.05189746487885714,
      "epoch": 2.274322509765625e-05,
      "step": 5962
    },
    {
      "action_loss_avg": 0.02607811139896512,
      "epoch": 2.274322509765625e-05,
      "step": 5962
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 5962,
      "training_step_time": 9.901909112930298
    },
    {
      "epoch": 2.274322509765625e-05,
      "model_forward_time": 3.1305017471313477,
      "step": 5962
    },
    {
      "epoch": 2.274322509765625e-05,
      "step": 5962,
      "training_step_time": 10.008022785186768
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 2.4721639156341553,
      "step": 5963
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 5963,
      "training_step_time": 7.4967920780181885
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 3.1314127445220947,
      "step": 5963
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 5963,
      "training_step_time": 9.900137901306152
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 3.4161391258239746,
      "step": 5963
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 5963,
      "training_step_time": 10.190316438674927
    },
    {
      "epoch": 2.2747039794921875e-05,
      "model_forward_time": 3.1344566345214844,
      "step": 5963
    },
    {
      "epoch": 2.2747039794921875e-05,
      "step": 5963,
      "training_step_time": 10.029719591140747
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 2.469313859939575,
      "step": 5964
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 5964,
      "training_step_time": 7.495838642120361
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 3.142016887664795,
      "step": 5964
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 5964,
      "training_step_time": 9.904832363128662
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 3.139455795288086,
      "step": 5964
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 5964,
      "training_step_time": 9.906610250473022
    },
    {
      "epoch": 2.27508544921875e-05,
      "model_forward_time": 3.394361734390259,
      "step": 5964
    },
    {
      "epoch": 2.27508544921875e-05,
      "step": 5964,
      "training_step_time": 10.288962841033936
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 3.147045850753784,
      "step": 5965
    },
    {
      "dynamics_loss_avg": 0.049386654049158096,
      "epoch": 2.2754669189453124e-05,
      "step": 5965
    },
    {
      "action_loss_avg": 0.02069023549556732,
      "epoch": 2.2754669189453124e-05,
      "step": 5965
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 5965,
      "training_step_time": 9.913022756576538
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 3.139474391937256,
      "step": 5965
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 5965,
      "training_step_time": 9.901328802108765
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 2.4400739669799805,
      "step": 5965
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 5965,
      "training_step_time": 7.462351560592651
    },
    {
      "epoch": 2.2754669189453124e-05,
      "model_forward_time": 3.1444594860076904,
      "step": 5965
    },
    {
      "epoch": 2.2754669189453124e-05,
      "step": 5965,
      "training_step_time": 10.302792310714722
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 3.4508652687072754,
      "step": 5966
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 5966,
      "training_step_time": 10.218491315841675
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 2.4301931858062744,
      "step": 5966
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 5966,
      "training_step_time": 7.454279661178589
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 3.123447895050049,
      "step": 5966
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 5966,
      "training_step_time": 9.891090869903564
    },
    {
      "epoch": 2.275848388671875e-05,
      "model_forward_time": 3.1352643966674805,
      "step": 5966
    },
    {
      "epoch": 2.275848388671875e-05,
      "step": 5966,
      "training_step_time": 10.295543193817139
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 3.1464319229125977,
      "step": 5967
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 5967,
      "training_step_time": 9.914612293243408
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 3.403181314468384,
      "step": 5967
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 5967,
      "training_step_time": 10.166797876358032
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 2.4313595294952393,
      "step": 5967
    },
    {
      "dynamics_loss_avg": 0.06321330065838993,
      "epoch": 2.2762298583984374e-05,
      "step": 5967
    },
    {
      "action_loss_avg": 0.025396771170198916,
      "epoch": 2.2762298583984374e-05,
      "step": 5967
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 5967,
      "training_step_time": 7.451387643814087
    },
    {
      "epoch": 2.2762298583984374e-05,
      "model_forward_time": 3.1282920837402344,
      "step": 5967
    },
    {
      "epoch": 2.2762298583984374e-05,
      "step": 5967,
      "training_step_time": 10.258419275283813
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 3.1436212062835693,
      "step": 5968
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 5968,
      "training_step_time": 9.908755540847778
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 3.1334619522094727,
      "step": 5968
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 5968,
      "training_step_time": 9.892638206481934
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 3.3821182250976562,
      "step": 5968
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 5968,
      "training_step_time": 10.146364450454712
    },
    {
      "epoch": 2.276611328125e-05,
      "model_forward_time": 2.4287168979644775,
      "step": 5968
    },
    {
      "epoch": 2.276611328125e-05,
      "step": 5968,
      "training_step_time": 10.000497579574585
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 3.139699935913086,
      "step": 5969
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 5969,
      "training_step_time": 9.899235963821411
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 3.1219911575317383,
      "step": 5969
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 5969,
      "training_step_time": 9.881651163101196
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 3.1273415088653564,
      "step": 5969
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 5969,
      "training_step_time": 9.88917350769043
    },
    {
      "epoch": 2.2769927978515626e-05,
      "model_forward_time": 3.395080089569092,
      "step": 5969
    },
    {
      "epoch": 2.2769927978515626e-05,
      "step": 5969,
      "training_step_time": 10.30410361289978
    },
    {
      "epoch": 2.277374267578125e-05,
      "grad_norm": 0.4032292366027832,
      "learning_rate": 1e-05,
      "loss": 0.0872,
      "step": 5970
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 3.1697821617126465,
      "step": 5970
    },
    {
      "dynamics_loss_avg": 0.06426082868129016,
      "epoch": 2.277374267578125e-05,
      "step": 5970
    },
    {
      "action_loss_avg": 0.02925753896124661,
      "epoch": 2.277374267578125e-05,
      "step": 5970
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 5970,
      "training_step_time": 9.940356492996216
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 2.4317264556884766,
      "step": 5970
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 5970,
      "training_step_time": 7.455414056777954
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 3.131725549697876,
      "step": 5970
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 5970,
      "training_step_time": 9.897387742996216
    },
    {
      "epoch": 2.277374267578125e-05,
      "model_forward_time": 3.1344282627105713,
      "step": 5970
    },
    {
      "epoch": 2.277374267578125e-05,
      "step": 5970,
      "training_step_time": 10.281998872756958
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 2.768817186355591,
      "step": 5971
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 5971,
      "training_step_time": 7.790405035018921
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 3.12872576713562,
      "step": 5971
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 5971,
      "training_step_time": 9.895973682403564
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 3.1371095180511475,
      "step": 5971
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 5971,
      "training_step_time": 9.905051469802856
    },
    {
      "epoch": 2.2777557373046875e-05,
      "model_forward_time": 3.1385018825531006,
      "step": 5971
    },
    {
      "epoch": 2.2777557373046875e-05,
      "step": 5971,
      "training_step_time": 10.318190336227417
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 2.47312593460083,
      "step": 5972
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 5972,
      "training_step_time": 7.50084114074707
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 3.3745927810668945,
      "step": 5972
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 5972,
      "training_step_time": 10.14242672920227
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 3.133552312850952,
      "step": 5972
    },
    {
      "dynamics_loss_avg": 0.05743393339216709,
      "epoch": 2.27813720703125e-05,
      "step": 5972
    },
    {
      "action_loss_avg": 0.022450930392369627,
      "epoch": 2.27813720703125e-05,
      "step": 5972
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 5972,
      "training_step_time": 9.904081583023071
    },
    {
      "epoch": 2.27813720703125e-05,
      "model_forward_time": 3.1408140659332275,
      "step": 5972
    },
    {
      "epoch": 2.27813720703125e-05,
      "step": 5972,
      "training_step_time": 10.297348737716675
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 3.168342351913452,
      "step": 5973
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 5973,
      "training_step_time": 9.940464973449707
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 3.135000467300415,
      "step": 5973
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 5973,
      "training_step_time": 9.901705980300903
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 3.4077565670013428,
      "step": 5973
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 5973,
      "training_step_time": 10.177087306976318
    },
    {
      "epoch": 2.2785186767578124e-05,
      "model_forward_time": 3.135246753692627,
      "step": 5973
    },
    {
      "epoch": 2.2785186767578124e-05,
      "step": 5973,
      "training_step_time": 10.019111156463623
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 3.1720736026763916,
      "step": 5974
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 5974,
      "training_step_time": 9.943379878997803
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 3.136237144470215,
      "step": 5974
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 5974,
      "training_step_time": 9.90951657295227
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 3.1309094429016113,
      "step": 5974
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 5974,
      "training_step_time": 9.901886701583862
    },
    {
      "epoch": 2.278900146484375e-05,
      "model_forward_time": 3.3818728923797607,
      "step": 5974
    },
    {
      "epoch": 2.278900146484375e-05,
      "step": 5974,
      "training_step_time": 10.247761964797974
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 2.4724864959716797,
      "step": 5975
    },
    {
      "dynamics_loss_avg": 0.06544304843991995,
      "epoch": 2.2792816162109374e-05,
      "step": 5975
    },
    {
      "action_loss_avg": 0.037951197009533644,
      "epoch": 2.2792816162109374e-05,
      "step": 5975
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 5975,
      "training_step_time": 7.501938343048096
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 3.130878210067749,
      "step": 5975
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 5975,
      "training_step_time": 9.89937138557434
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 3.133800506591797,
      "step": 5975
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 5975,
      "training_step_time": 9.905059576034546
    },
    {
      "epoch": 2.2792816162109374e-05,
      "model_forward_time": 3.3937630653381348,
      "step": 5975
    },
    {
      "epoch": 2.2792816162109374e-05,
      "step": 5975,
      "training_step_time": 10.288790941238403
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 3.1782824993133545,
      "step": 5976
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 5976,
      "training_step_time": 9.947321891784668
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 3.133798360824585,
      "step": 5976
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 5976,
      "training_step_time": 9.899941682815552
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 3.132647752761841,
      "step": 5976
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 5976,
      "training_step_time": 9.898124933242798
    },
    {
      "epoch": 2.2796630859375e-05,
      "model_forward_time": 3.1354382038116455,
      "step": 5976
    },
    {
      "epoch": 2.2796630859375e-05,
      "step": 5976,
      "training_step_time": 10.002570867538452
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 3.148559331893921,
      "step": 5977
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 5977,
      "training_step_time": 10.202396631240845
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 3.1367177963256836,
      "step": 5977
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 5977,
      "training_step_time": 9.903276681900024
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 3.1324989795684814,
      "step": 5977
    },
    {
      "dynamics_loss_avg": 0.04703651238232851,
      "epoch": 2.2800445556640626e-05,
      "step": 5977
    },
    {
      "action_loss_avg": 0.0197712704539299,
      "epoch": 2.2800445556640626e-05,
      "step": 5977
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 5977,
      "training_step_time": 9.902931928634644
    },
    {
      "epoch": 2.2800445556640626e-05,
      "model_forward_time": 2.427696704864502,
      "step": 5977
    },
    {
      "epoch": 2.2800445556640626e-05,
      "step": 5977,
      "training_step_time": 10.324466943740845
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 3.1439497470855713,
      "step": 5978
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 5978,
      "training_step_time": 9.907713890075684
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 3.124509572982788,
      "step": 5978
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 5978,
      "training_step_time": 10.189882040023804
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 3.1271579265594482,
      "step": 5978
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 5978,
      "training_step_time": 9.897480964660645
    },
    {
      "epoch": 2.280426025390625e-05,
      "model_forward_time": 3.1373064517974854,
      "step": 5978
    },
    {
      "epoch": 2.280426025390625e-05,
      "step": 5978,
      "training_step_time": 10.30335807800293
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 3.1743228435516357,
      "step": 5979
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 5979,
      "training_step_time": 9.949783563613892
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 3.1299920082092285,
      "step": 5979
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 5979,
      "training_step_time": 9.901268005371094
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 3.1371936798095703,
      "step": 5979
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 5979,
      "training_step_time": 10.215218544006348
    },
    {
      "epoch": 2.2808074951171875e-05,
      "model_forward_time": 3.134364128112793,
      "step": 5979
    },
    {
      "epoch": 2.2808074951171875e-05,
      "step": 5979,
      "training_step_time": 10.277910947799683
    },
    {
      "epoch": 2.28118896484375e-05,
      "grad_norm": 0.48488807678222656,
      "learning_rate": 1e-05,
      "loss": 0.0841,
      "step": 5980
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 3.1471357345581055,
      "step": 5980
    },
    {
      "dynamics_loss_avg": 0.06990890949964523,
      "epoch": 2.28118896484375e-05,
      "step": 5980
    },
    {
      "action_loss_avg": 0.03105794037692249,
      "epoch": 2.28118896484375e-05,
      "step": 5980
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 5980,
      "training_step_time": 9.923221349716187
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 3.1321804523468018,
      "step": 5980
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 5980,
      "training_step_time": 9.902589797973633
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 3.134242534637451,
      "step": 5980
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 5980,
      "training_step_time": 9.909096479415894
    },
    {
      "epoch": 2.28118896484375e-05,
      "model_forward_time": 3.135420322418213,
      "step": 5980
    },
    {
      "epoch": 2.28118896484375e-05,
      "step": 5980,
      "training_step_time": 10.292460203170776
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 3.1789352893829346,
      "step": 5981
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 5981,
      "training_step_time": 9.958357810974121
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 3.137070894241333,
      "step": 5981
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 5981,
      "training_step_time": 9.916953086853027
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 3.1393420696258545,
      "step": 5981
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 5981,
      "training_step_time": 9.91843056678772
    },
    {
      "epoch": 2.2815704345703125e-05,
      "model_forward_time": 3.135061025619507,
      "step": 5981
    },
    {
      "epoch": 2.2815704345703125e-05,
      "step": 5981,
      "training_step_time": 10.035048961639404
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 3.458883285522461,
      "step": 5982
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 5982,
      "training_step_time": 10.241562604904175
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 2.4361705780029297,
      "step": 5982
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 5982,
      "training_step_time": 7.481351852416992
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 3.137108564376831,
      "step": 5982
    },
    {
      "dynamics_loss_avg": 0.060343002527952196,
      "epoch": 2.281951904296875e-05,
      "step": 5982
    },
    {
      "action_loss_avg": 0.022753837704658508,
      "epoch": 2.281951904296875e-05,
      "step": 5982
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 5982,
      "training_step_time": 9.921136379241943
    },
    {
      "epoch": 2.281951904296875e-05,
      "model_forward_time": 3.147747039794922,
      "step": 5982
    },
    {
      "epoch": 2.281951904296875e-05,
      "step": 5982,
      "training_step_time": 10.288573980331421
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 1.7808785438537598,
      "step": 5983
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 5983,
      "training_step_time": 5.243387937545776
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 3.142493724822998,
      "step": 5983
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 5983,
      "training_step_time": 9.922068357467651
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 3.424112319946289,
      "step": 5983
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 5983,
      "training_step_time": 10.201628923416138
    },
    {
      "epoch": 2.2823333740234374e-05,
      "model_forward_time": 3.143939971923828,
      "step": 5983
    },
    {
      "epoch": 2.2823333740234374e-05,
      "step": 5983,
      "training_step_time": 10.036602973937988
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 3.189436912536621,
      "step": 5984
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 5984,
      "training_step_time": 9.97143006324768
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 3.1426539421081543,
      "step": 5984
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 5984,
      "training_step_time": 9.919840335845947
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 3.141150951385498,
      "step": 5984
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 5984,
      "training_step_time": 9.914742708206177
    },
    {
      "epoch": 2.28271484375e-05,
      "model_forward_time": 3.396132707595825,
      "step": 5984
    },
    {
      "epoch": 2.28271484375e-05,
      "step": 5984,
      "training_step_time": 10.259584188461304
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 3.1703293323516846,
      "step": 5985
    },
    {
      "dynamics_loss_avg": 0.06961516151204705,
      "epoch": 2.2830963134765626e-05,
      "step": 5985
    },
    {
      "action_loss_avg": 0.03459862135350704,
      "epoch": 2.2830963134765626e-05,
      "step": 5985
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 5985,
      "training_step_time": 9.943143367767334
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 3.133474111557007,
      "step": 5985
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 5985,
      "training_step_time": 9.90362286567688
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 3.1324563026428223,
      "step": 5985
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 5985,
      "training_step_time": 9.895431756973267
    },
    {
      "epoch": 2.2830963134765626e-05,
      "model_forward_time": 3.134507179260254,
      "step": 5985
    },
    {
      "epoch": 2.2830963134765626e-05,
      "step": 5985,
      "training_step_time": 10.298393487930298
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 2.7174696922302246,
      "step": 5986
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 5986,
      "training_step_time": 7.735854148864746
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 2.4289402961730957,
      "step": 5986
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 5986,
      "training_step_time": 7.448909521102905
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 3.1281542778015137,
      "step": 5986
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 5986,
      "training_step_time": 9.88233208656311
    },
    {
      "epoch": 2.283477783203125e-05,
      "model_forward_time": 3.134310245513916,
      "step": 5986
    },
    {
      "epoch": 2.283477783203125e-05,
      "step": 5986,
      "training_step_time": 10.249282121658325
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 3.1469428539276123,
      "step": 5987
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 5987,
      "training_step_time": 9.91582703590393
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 2.716749668121338,
      "step": 5987
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 5987,
      "training_step_time": 7.7401323318481445
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 3.1354031562805176,
      "step": 5987
    },
    {
      "dynamics_loss_avg": 0.05636911783367395,
      "epoch": 2.2838592529296875e-05,
      "step": 5987
    },
    {
      "action_loss_avg": 0.02642057663761079,
      "epoch": 2.2838592529296875e-05,
      "step": 5987
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 5987,
      "training_step_time": 9.909335374832153
    },
    {
      "epoch": 2.2838592529296875e-05,
      "model_forward_time": 2.434626340866089,
      "step": 5987
    },
    {
      "epoch": 2.2838592529296875e-05,
      "step": 5987,
      "training_step_time": 10.300582885742188
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 3.1515862941741943,
      "step": 5988
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 5988,
      "training_step_time": 9.927951335906982
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 3.13777494430542,
      "step": 5988
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 5988,
      "training_step_time": 9.910798788070679
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 3.398787260055542,
      "step": 5988
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 5988,
      "training_step_time": 10.16985011100769
    },
    {
      "epoch": 2.28424072265625e-05,
      "model_forward_time": 3.1377952098846436,
      "step": 5988
    },
    {
      "epoch": 2.28424072265625e-05,
      "step": 5988,
      "training_step_time": 10.315372705459595
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 2.473787307739258,
      "step": 5989
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 5989,
      "training_step_time": 7.50006890296936
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 3.129030227661133,
      "step": 5989
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 5989,
      "training_step_time": 9.895471811294556
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 3.1348845958709717,
      "step": 5989
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 5989,
      "training_step_time": 9.901456594467163
    },
    {
      "epoch": 2.2846221923828125e-05,
      "model_forward_time": 3.133105516433716,
      "step": 5989
    },
    {
      "epoch": 2.2846221923828125e-05,
      "step": 5989,
      "training_step_time": 10.264884233474731
    },
    {
      "epoch": 2.285003662109375e-05,
      "grad_norm": 0.7572925090789795,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 5990
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 3.181154251098633,
      "step": 5990
    },
    {
      "dynamics_loss_avg": 0.045787197165191175,
      "epoch": 2.285003662109375e-05,
      "step": 5990
    },
    {
      "action_loss_avg": 0.02709364490583539,
      "epoch": 2.285003662109375e-05,
      "step": 5990
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 5990,
      "training_step_time": 9.954357385635376
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 2.436431407928467,
      "step": 5990
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 5990,
      "training_step_time": 7.461282968521118
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 3.1352479457855225,
      "step": 5990
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 5990,
      "training_step_time": 9.908586025238037
    },
    {
      "epoch": 2.285003662109375e-05,
      "model_forward_time": 3.131664514541626,
      "step": 5990
    },
    {
      "epoch": 2.285003662109375e-05,
      "step": 5990,
      "training_step_time": 10.25169849395752
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 3.45180082321167,
      "step": 5991
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 5991,
      "training_step_time": 10.22747802734375
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 3.1300954818725586,
      "step": 5991
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 5991,
      "training_step_time": 9.899134159088135
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 3.1357150077819824,
      "step": 5991
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 5991,
      "training_step_time": 9.903858184814453
    },
    {
      "epoch": 2.2853851318359374e-05,
      "model_forward_time": 3.1331305503845215,
      "step": 5991
    },
    {
      "epoch": 2.2853851318359374e-05,
      "step": 5991,
      "training_step_time": 10.00342345237732
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 3.1436195373535156,
      "step": 5992
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 5992,
      "training_step_time": 9.915742874145508
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 3.1364173889160156,
      "step": 5992
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 5992,
      "training_step_time": 10.178870677947998
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 3.143109083175659,
      "step": 5992
    },
    {
      "dynamics_loss_avg": 0.04700653124600649,
      "epoch": 2.2857666015625e-05,
      "step": 5992
    },
    {
      "action_loss_avg": 0.018664856255054475,
      "epoch": 2.2857666015625e-05,
      "step": 5992
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 5992,
      "training_step_time": 9.917802810668945
    },
    {
      "epoch": 2.2857666015625e-05,
      "model_forward_time": 3.134817123413086,
      "step": 5992
    },
    {
      "epoch": 2.2857666015625e-05,
      "step": 5992,
      "training_step_time": 10.007952213287354
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 3.1460072994232178,
      "step": 5993
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 5993,
      "training_step_time": 9.923740863800049
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 3.1362948417663574,
      "step": 5993
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 5993,
      "training_step_time": 9.911520957946777
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 3.4204375743865967,
      "step": 5993
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 5993,
      "training_step_time": 10.19568943977356
    },
    {
      "epoch": 2.2861480712890626e-05,
      "model_forward_time": 3.1416304111480713,
      "step": 5993
    },
    {
      "epoch": 2.2861480712890626e-05,
      "step": 5993,
      "training_step_time": 10.261474370956421
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 3.1463210582733154,
      "step": 5994
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 5994,
      "training_step_time": 9.92550277709961
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 3.143342971801758,
      "step": 5994
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 5994,
      "training_step_time": 9.919847249984741
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 3.1427958011627197,
      "step": 5994
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 5994,
      "training_step_time": 9.92278003692627
    },
    {
      "epoch": 2.286529541015625e-05,
      "model_forward_time": 2.4367644786834717,
      "step": 5994
    },
    {
      "epoch": 2.286529541015625e-05,
      "step": 5994,
      "training_step_time": 10.270906925201416
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 2.4477808475494385,
      "step": 5995
    },
    {
      "dynamics_loss_avg": 0.04801495596766472,
      "epoch": 2.2869110107421875e-05,
      "step": 5995
    },
    {
      "action_loss_avg": 0.020396507205441596,
      "epoch": 2.2869110107421875e-05,
      "step": 5995
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 5995,
      "training_step_time": 7.484052419662476
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 3.1420226097106934,
      "step": 5995
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 5995,
      "training_step_time": 9.919582843780518
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 3.1396396160125732,
      "step": 5995
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 5995,
      "training_step_time": 9.916826486587524
    },
    {
      "epoch": 2.2869110107421875e-05,
      "model_forward_time": 3.1433496475219727,
      "step": 5995
    },
    {
      "epoch": 2.2869110107421875e-05,
      "step": 5995,
      "training_step_time": 10.297088861465454
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 3.4242732524871826,
      "step": 5996
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 5996,
      "training_step_time": 10.193999290466309
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 3.137296676635742,
      "step": 5996
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 5996,
      "training_step_time": 9.909292936325073
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 3.1346864700317383,
      "step": 5996
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 5996,
      "training_step_time": 9.906429290771484
    },
    {
      "epoch": 2.28729248046875e-05,
      "model_forward_time": 2.433382749557495,
      "step": 5996
    },
    {
      "epoch": 2.28729248046875e-05,
      "step": 5996,
      "training_step_time": 10.000405073165894
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 3.1791107654571533,
      "step": 5997
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 5997,
      "training_step_time": 9.949758768081665
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 3.1311850547790527,
      "step": 5997
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 5997,
      "training_step_time": 9.900725841522217
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 3.4240448474884033,
      "step": 5997
    },
    {
      "dynamics_loss_avg": 0.06173875071108341,
      "epoch": 2.2876739501953125e-05,
      "step": 5997
    },
    {
      "action_loss_avg": 0.03463684683665633,
      "epoch": 2.2876739501953125e-05,
      "step": 5997
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 5997,
      "training_step_time": 10.193589210510254
    },
    {
      "epoch": 2.2876739501953125e-05,
      "model_forward_time": 3.1344077587127686,
      "step": 5997
    },
    {
      "epoch": 2.2876739501953125e-05,
      "step": 5997,
      "training_step_time": 9.99711298942566
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 3.149355888366699,
      "step": 5998
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 5998,
      "training_step_time": 9.920316219329834
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 3.1345980167388916,
      "step": 5998
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 5998,
      "training_step_time": 9.903693199157715
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 3.1364798545837402,
      "step": 5998
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 5998,
      "training_step_time": 10.19112491607666
    },
    {
      "epoch": 2.288055419921875e-05,
      "model_forward_time": 3.1398417949676514,
      "step": 5998
    },
    {
      "epoch": 2.288055419921875e-05,
      "step": 5998,
      "training_step_time": 10.29629111289978
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 3.2161176204681396,
      "step": 5999
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 5999,
      "training_step_time": 9.993283748626709
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 3.1314921379089355,
      "step": 5999
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 5999,
      "training_step_time": 9.90880298614502
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 3.1425058841705322,
      "step": 5999
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 5999,
      "training_step_time": 9.924707412719727
    },
    {
      "epoch": 2.2884368896484374e-05,
      "model_forward_time": 3.425987958908081,
      "step": 5999
    },
    {
      "epoch": 2.2884368896484374e-05,
      "step": 5999,
      "training_step_time": 10.331561088562012
    },
    {
      "epoch": 2.288818359375e-05,
      "grad_norm": 0.8887127637863159,
      "learning_rate": 1e-05,
      "loss": 0.0814,
      "step": 6000
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 3.167248010635376,
      "step": 6000
    },
    {
      "dynamics_loss_avg": 0.05557930488139391,
      "epoch": 2.288818359375e-05,
      "step": 6000
    },
    {
      "action_loss_avg": 0.022079536225646734,
      "epoch": 2.288818359375e-05,
      "step": 6000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 6000,
      "training_step_time": 9.921705961227417
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 3.1312389373779297,
      "step": 6000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 6000,
      "training_step_time": 9.88622236251831
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 3.1283512115478516,
      "step": 6000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 6000,
      "training_step_time": 9.887061595916748
    },
    {
      "epoch": 2.288818359375e-05,
      "model_forward_time": 2.4286558628082275,
      "step": 6000
    },
    {
      "epoch": 2.288818359375e-05,
      "step": 6000,
      "training_step_time": 10.314239263534546
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 5.248364448547363,
      "step": 6001
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 6001,
      "training_step_time": 12.018007755279541
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 3.1311028003692627,
      "step": 6001
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 6001,
      "training_step_time": 9.901156663894653
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 3.129655122756958,
      "step": 6001
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 6001,
      "training_step_time": 9.902592420578003
    },
    {
      "epoch": 2.2891998291015626e-05,
      "model_forward_time": 3.1365749835968018,
      "step": 6001
    },
    {
      "epoch": 2.2891998291015626e-05,
      "step": 6001,
      "training_step_time": 10.046639204025269
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 3.1662099361419678,
      "step": 6002
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 6002,
      "training_step_time": 9.940139770507812
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 3.4064857959747314,
      "step": 6002
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 6002,
      "training_step_time": 10.182963371276855
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 3.1416428089141846,
      "step": 6002
    },
    {
      "dynamics_loss_avg": 0.059134325757622716,
      "epoch": 2.289581298828125e-05,
      "step": 6002
    },
    {
      "action_loss_avg": 0.0260429160669446,
      "epoch": 2.289581298828125e-05,
      "step": 6002
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 6002,
      "training_step_time": 9.915905475616455
    },
    {
      "epoch": 2.289581298828125e-05,
      "model_forward_time": 3.1365888118743896,
      "step": 6002
    },
    {
      "epoch": 2.289581298828125e-05,
      "step": 6002,
      "training_step_time": 10.01440691947937
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 3.149979591369629,
      "step": 6003
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 6003,
      "training_step_time": 9.926628828048706
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 3.4322574138641357,
      "step": 6003
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 6003,
      "training_step_time": 10.209091663360596
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 2.4415440559387207,
      "step": 6003
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 6003,
      "training_step_time": 7.4729650020599365
    },
    {
      "epoch": 2.2899627685546875e-05,
      "model_forward_time": 3.146599531173706,
      "step": 6003
    },
    {
      "epoch": 2.2899627685546875e-05,
      "step": 6003,
      "training_step_time": 10.313429117202759
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 3.172222137451172,
      "step": 6004
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 6004,
      "training_step_time": 9.957845211029053
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 2.4349417686462402,
      "step": 6004
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 6004,
      "training_step_time": 7.472238779067993
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 3.1418895721435547,
      "step": 6004
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 6004,
      "training_step_time": 9.929532289505005
    },
    {
      "epoch": 2.29034423828125e-05,
      "model_forward_time": 3.4103691577911377,
      "step": 6004
    },
    {
      "epoch": 2.29034423828125e-05,
      "step": 6004,
      "training_step_time": 10.29405689239502
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 3.1820411682128906,
      "step": 6005
    },
    {
      "dynamics_loss_avg": 0.06413466744124889,
      "epoch": 2.2907257080078125e-05,
      "step": 6005
    },
    {
      "action_loss_avg": 0.025303343031555414,
      "epoch": 2.2907257080078125e-05,
      "step": 6005
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 6005,
      "training_step_time": 9.968708038330078
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 3.1353042125701904,
      "step": 6005
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 6005,
      "training_step_time": 9.914807558059692
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 3.140857696533203,
      "step": 6005
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 6005,
      "training_step_time": 9.922597169876099
    },
    {
      "epoch": 2.2907257080078125e-05,
      "model_forward_time": 3.139028787612915,
      "step": 6005
    },
    {
      "epoch": 2.2907257080078125e-05,
      "step": 6005,
      "training_step_time": 10.32532000541687
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 1.758296251296997,
      "step": 6006
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 6006,
      "training_step_time": 5.2210328578948975
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 2.4304537773132324,
      "step": 6006
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 6006,
      "training_step_time": 7.460737466812134
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 3.131735324859619,
      "step": 6006
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 6006,
      "training_step_time": 9.914775848388672
    },
    {
      "epoch": 2.291107177734375e-05,
      "model_forward_time": 3.138777494430542,
      "step": 6006
    },
    {
      "epoch": 2.291107177734375e-05,
      "step": 6006,
      "training_step_time": 10.29955792427063
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 3.1499898433685303,
      "step": 6007
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 6007,
      "training_step_time": 9.933014869689941
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 3.3889222145080566,
      "step": 6007
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 6007,
      "training_step_time": 10.170823097229004
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 3.1456682682037354,
      "step": 6007
    },
    {
      "dynamics_loss_avg": 0.06719210464507341,
      "epoch": 2.2914886474609374e-05,
      "step": 6007
    },
    {
      "action_loss_avg": 0.025724540976807476,
      "epoch": 2.2914886474609374e-05,
      "step": 6007
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 6007,
      "training_step_time": 9.929786443710327
    },
    {
      "epoch": 2.2914886474609374e-05,
      "model_forward_time": 2.435065984725952,
      "step": 6007
    },
    {
      "epoch": 2.2914886474609374e-05,
      "step": 6007,
      "training_step_time": 10.033172369003296
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 3.1562225818634033,
      "step": 6008
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 6008,
      "training_step_time": 9.94533371925354
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 3.1387202739715576,
      "step": 6008
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 6008,
      "training_step_time": 9.918199062347412
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 3.4036097526550293,
      "step": 6008
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 6008,
      "training_step_time": 10.178106307983398
    },
    {
      "epoch": 2.2918701171875e-05,
      "model_forward_time": 2.4318575859069824,
      "step": 6008
    },
    {
      "epoch": 2.2918701171875e-05,
      "step": 6008,
      "training_step_time": 10.293743133544922
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 3.159818649291992,
      "step": 6009
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 6009,
      "training_step_time": 9.933887004852295
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 3.1311371326446533,
      "step": 6009
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 6009,
      "training_step_time": 9.909754276275635
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 3.1323742866516113,
      "step": 6009
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 6009,
      "training_step_time": 9.910598993301392
    },
    {
      "epoch": 2.2922515869140626e-05,
      "model_forward_time": 3.3773586750030518,
      "step": 6009
    },
    {
      "epoch": 2.2922515869140626e-05,
      "step": 6009,
      "training_step_time": 10.30703067779541
    },
    {
      "epoch": 2.292633056640625e-05,
      "grad_norm": 0.5135846138000488,
      "learning_rate": 1e-05,
      "loss": 0.084,
      "step": 6010
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 3.177394390106201,
      "step": 6010
    },
    {
      "dynamics_loss_avg": 0.053870771266520026,
      "epoch": 2.292633056640625e-05,
      "step": 6010
    },
    {
      "action_loss_avg": 0.02446648934856057,
      "epoch": 2.292633056640625e-05,
      "step": 6010
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 6010,
      "training_step_time": 9.960587739944458
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 3.1415138244628906,
      "step": 6010
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 6010,
      "training_step_time": 9.922338485717773
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 3.142775297164917,
      "step": 6010
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 6010,
      "training_step_time": 9.928351163864136
    },
    {
      "epoch": 2.292633056640625e-05,
      "model_forward_time": 3.160909414291382,
      "step": 6010
    },
    {
      "epoch": 2.292633056640625e-05,
      "step": 6010,
      "training_step_time": 10.332472562789917
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 3.439302444458008,
      "step": 6011
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 6011,
      "training_step_time": 10.224350452423096
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 3.1482622623443604,
      "step": 6011
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 6011,
      "training_step_time": 9.93139362335205
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 3.1457879543304443,
      "step": 6011
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 6011,
      "training_step_time": 9.926196813583374
    },
    {
      "epoch": 2.2930145263671875e-05,
      "model_forward_time": 3.1504344940185547,
      "step": 6011
    },
    {
      "epoch": 2.2930145263671875e-05,
      "step": 6011,
      "training_step_time": 10.287185907363892
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 2.4443700313568115,
      "step": 6012
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 6012,
      "training_step_time": 7.480942010879517
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 3.463474750518799,
      "step": 6012
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 6012,
      "training_step_time": 10.26460599899292
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 3.1459763050079346,
      "step": 6012
    },
    {
      "dynamics_loss_avg": 0.06961145736277104,
      "epoch": 2.29339599609375e-05,
      "step": 6012
    },
    {
      "action_loss_avg": 0.032081476785242556,
      "epoch": 2.29339599609375e-05,
      "step": 6012
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 6012,
      "training_step_time": 9.923768043518066
    },
    {
      "epoch": 2.29339599609375e-05,
      "model_forward_time": 3.1501214504241943,
      "step": 6012
    },
    {
      "epoch": 2.29339599609375e-05,
      "step": 6012,
      "training_step_time": 10.022057056427002
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 3.189166307449341,
      "step": 6013
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 6013,
      "training_step_time": 9.96652102470398
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 3.1423163414001465,
      "step": 6013
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 6013,
      "training_step_time": 9.913254261016846
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 3.378422737121582,
      "step": 6013
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 6013,
      "training_step_time": 10.154061317443848
    },
    {
      "epoch": 2.2937774658203125e-05,
      "model_forward_time": 3.134597063064575,
      "step": 6013
    },
    {
      "epoch": 2.2937774658203125e-05,
      "step": 6013,
      "training_step_time": 10.272786617279053
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 3.150451421737671,
      "step": 6014
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 6014,
      "training_step_time": 9.927987813949585
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 3.1316068172454834,
      "step": 6014
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 6014,
      "training_step_time": 9.90676474571228
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 2.429102659225464,
      "step": 6014
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 6014,
      "training_step_time": 7.455448150634766
    },
    {
      "epoch": 2.294158935546875e-05,
      "model_forward_time": 3.403677463531494,
      "step": 6014
    },
    {
      "epoch": 2.294158935546875e-05,
      "step": 6014,
      "training_step_time": 10.272349119186401
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 3.1622045040130615,
      "step": 6015
    },
    {
      "dynamics_loss_avg": 0.060010576993227004,
      "epoch": 2.2945404052734374e-05,
      "step": 6015
    },
    {
      "action_loss_avg": 0.023489065933972597,
      "epoch": 2.2945404052734374e-05,
      "step": 6015
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 6015,
      "training_step_time": 9.936651706695557
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 3.126331329345703,
      "step": 6015
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 6015,
      "training_step_time": 9.897021770477295
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 3.1331264972686768,
      "step": 6015
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 6015,
      "training_step_time": 9.901482820510864
    },
    {
      "epoch": 2.2945404052734374e-05,
      "model_forward_time": 3.1284444332122803,
      "step": 6015
    },
    {
      "epoch": 2.2945404052734374e-05,
      "step": 6015,
      "training_step_time": 10.292406558990479
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 2.785645008087158,
      "step": 6016
    },
    {
      "epoch": 2.294921875e-05,
      "step": 6016,
      "training_step_time": 7.8087263107299805
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 3.1293246746063232,
      "step": 6016
    },
    {
      "epoch": 2.294921875e-05,
      "step": 6016,
      "training_step_time": 9.903337240219116
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 3.1326661109924316,
      "step": 6016
    },
    {
      "epoch": 2.294921875e-05,
      "step": 6016,
      "training_step_time": 9.906054258346558
    },
    {
      "epoch": 2.294921875e-05,
      "model_forward_time": 3.1339797973632812,
      "step": 6016
    },
    {
      "epoch": 2.294921875e-05,
      "step": 6016,
      "training_step_time": 10.249433755874634
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 3.1551692485809326,
      "step": 6017
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 6017,
      "training_step_time": 10.200398206710815
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 3.13437819480896,
      "step": 6017
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 6017,
      "training_step_time": 9.909958600997925
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 3.1323704719543457,
      "step": 6017
    },
    {
      "dynamics_loss_avg": 0.0623917106539011,
      "epoch": 2.2953033447265626e-05,
      "step": 6017
    },
    {
      "action_loss_avg": 0.02429620213806629,
      "epoch": 2.2953033447265626e-05,
      "step": 6017
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 6017,
      "training_step_time": 9.910181283950806
    },
    {
      "epoch": 2.2953033447265626e-05,
      "model_forward_time": 3.1344335079193115,
      "step": 6017
    },
    {
      "epoch": 2.2953033447265626e-05,
      "step": 6017,
      "training_step_time": 10.305770874023438
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 3.187671661376953,
      "step": 6018
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 6018,
      "training_step_time": 9.966829538345337
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 3.3973865509033203,
      "step": 6018
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 6018,
      "training_step_time": 10.176984071731567
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 3.1417040824890137,
      "step": 6018
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 6018,
      "training_step_time": 9.92048168182373
    },
    {
      "epoch": 2.295684814453125e-05,
      "model_forward_time": 3.1352667808532715,
      "step": 6018
    },
    {
      "epoch": 2.295684814453125e-05,
      "step": 6018,
      "training_step_time": 10.012813329696655
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 3.1444051265716553,
      "step": 6019
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 6019,
      "training_step_time": 9.923146724700928
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 3.1422643661499023,
      "step": 6019
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 6019,
      "training_step_time": 9.924049377441406
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 3.4231953620910645,
      "step": 6019
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 6019,
      "training_step_time": 10.191872119903564
    },
    {
      "epoch": 2.2960662841796875e-05,
      "model_forward_time": 3.1397793292999268,
      "step": 6019
    },
    {
      "epoch": 2.2960662841796875e-05,
      "step": 6019,
      "training_step_time": 10.007378339767456
    },
    {
      "epoch": 2.29644775390625e-05,
      "grad_norm": 0.524816632270813,
      "learning_rate": 1e-05,
      "loss": 0.0878,
      "step": 6020
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 3.1454858779907227,
      "step": 6020
    },
    {
      "dynamics_loss_avg": 0.0555931456387043,
      "epoch": 2.29644775390625e-05,
      "step": 6020
    },
    {
      "action_loss_avg": 0.021338747069239616,
      "epoch": 2.29644775390625e-05,
      "step": 6020
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 6020,
      "training_step_time": 9.91609811782837
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 3.1384544372558594,
      "step": 6020
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 6020,
      "training_step_time": 9.905673503875732
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 3.1359922885894775,
      "step": 6020
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 6020,
      "training_step_time": 9.905575275421143
    },
    {
      "epoch": 2.29644775390625e-05,
      "model_forward_time": 3.40398907661438,
      "step": 6020
    },
    {
      "epoch": 2.29644775390625e-05,
      "step": 6020,
      "training_step_time": 10.298136234283447
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 3.1475117206573486,
      "step": 6021
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 6021,
      "training_step_time": 9.921526908874512
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 3.143263339996338,
      "step": 6021
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 6021,
      "training_step_time": 9.91437315940857
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 3.139085531234741,
      "step": 6021
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 6021,
      "training_step_time": 9.911764860153198
    },
    {
      "epoch": 2.2968292236328125e-05,
      "model_forward_time": 3.1400063037872314,
      "step": 6021
    },
    {
      "epoch": 2.2968292236328125e-05,
      "step": 6021,
      "training_step_time": 10.02403450012207
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 3.4611687660217285,
      "step": 6022
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 6022,
      "training_step_time": 10.233406782150269
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 3.1484618186950684,
      "step": 6022
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 6022,
      "training_step_time": 9.918467998504639
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 3.1464781761169434,
      "step": 6022
    },
    {
      "dynamics_loss_avg": 0.07634728811681271,
      "epoch": 2.297210693359375e-05,
      "step": 6022
    },
    {
      "action_loss_avg": 0.030718103982508183,
      "epoch": 2.297210693359375e-05,
      "step": 6022
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 6022,
      "training_step_time": 9.917375087738037
    },
    {
      "epoch": 2.297210693359375e-05,
      "model_forward_time": 3.147649049758911,
      "step": 6022
    },
    {
      "epoch": 2.297210693359375e-05,
      "step": 6022,
      "training_step_time": 10.016928672790527
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 3.154409408569336,
      "step": 6023
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 6023,
      "training_step_time": 9.929603338241577
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 2.442195177078247,
      "step": 6023
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 6023,
      "training_step_time": 7.746589660644531
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 3.1407461166381836,
      "step": 6023
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 6023,
      "training_step_time": 9.911741495132446
    },
    {
      "epoch": 2.2975921630859374e-05,
      "model_forward_time": 3.142813205718994,
      "step": 6023
    },
    {
      "epoch": 2.2975921630859374e-05,
      "step": 6023,
      "training_step_time": 10.024749279022217
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 3.1686289310455322,
      "step": 6024
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 6024,
      "training_step_time": 9.936420440673828
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 2.4381704330444336,
      "step": 6024
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 6024,
      "training_step_time": 7.461634874343872
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 3.143458604812622,
      "step": 6024
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 6024,
      "training_step_time": 9.910638332366943
    },
    {
      "epoch": 2.2979736328125e-05,
      "model_forward_time": 2.7187721729278564,
      "step": 6024
    },
    {
      "epoch": 2.2979736328125e-05,
      "step": 6024,
      "training_step_time": 10.25721549987793
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 3.177129030227661,
      "step": 6025
    },
    {
      "dynamics_loss_avg": 0.06629013307392598,
      "epoch": 2.2983551025390626e-05,
      "step": 6025
    },
    {
      "action_loss_avg": 0.031795705668628216,
      "epoch": 2.2983551025390626e-05,
      "step": 6025
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 6025,
      "training_step_time": 9.953316926956177
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 3.1440160274505615,
      "step": 6025
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 6025,
      "training_step_time": 9.915821552276611
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 3.144930601119995,
      "step": 6025
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 6025,
      "training_step_time": 9.915820598602295
    },
    {
      "epoch": 2.2983551025390626e-05,
      "model_forward_time": 3.137866258621216,
      "step": 6025
    },
    {
      "epoch": 2.2983551025390626e-05,
      "step": 6025,
      "training_step_time": 10.281495571136475
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 2.475727081298828,
      "step": 6026
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 6026,
      "training_step_time": 7.5055341720581055
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 3.1390674114227295,
      "step": 6026
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 6026,
      "training_step_time": 9.914720296859741
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 3.1475868225097656,
      "step": 6026
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 6026,
      "training_step_time": 9.920535326004028
    },
    {
      "epoch": 2.298736572265625e-05,
      "model_forward_time": 3.141918182373047,
      "step": 6026
    },
    {
      "epoch": 2.298736572265625e-05,
      "step": 6026,
      "training_step_time": 10.28967833518982
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 3.1797211170196533,
      "step": 6027
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 6027,
      "training_step_time": 10.231518030166626
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 2.443753480911255,
      "step": 6027
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 6027,
      "training_step_time": 7.476922035217285
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 3.1446449756622314,
      "step": 6027
    },
    {
      "dynamics_loss_avg": 0.06780331451445817,
      "epoch": 2.2991180419921876e-05,
      "step": 6027
    },
    {
      "action_loss_avg": 0.02938653449527919,
      "epoch": 2.2991180419921876e-05,
      "step": 6027
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 6027,
      "training_step_time": 9.938051700592041
    },
    {
      "epoch": 2.2991180419921876e-05,
      "model_forward_time": 3.1488332748413086,
      "step": 6027
    },
    {
      "epoch": 2.2991180419921876e-05,
      "step": 6027,
      "training_step_time": 10.299091815948486
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 3.192091464996338,
      "step": 6028
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 6028,
      "training_step_time": 9.966827869415283
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 3.146883964538574,
      "step": 6028
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 6028,
      "training_step_time": 9.926336288452148
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 3.4337916374206543,
      "step": 6028
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 6028,
      "training_step_time": 10.207870721817017
    },
    {
      "epoch": 2.29949951171875e-05,
      "model_forward_time": 3.148721218109131,
      "step": 6028
    },
    {
      "epoch": 2.29949951171875e-05,
      "step": 6028,
      "training_step_time": 10.020459651947021
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 3.1739790439605713,
      "step": 6029
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 6029,
      "training_step_time": 9.953552007675171
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 3.1395535469055176,
      "step": 6029
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 6029,
      "training_step_time": 9.905938625335693
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 3.4128894805908203,
      "step": 6029
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 6029,
      "training_step_time": 10.181875944137573
    },
    {
      "epoch": 2.2998809814453125e-05,
      "model_forward_time": 3.131566286087036,
      "step": 6029
    },
    {
      "epoch": 2.2998809814453125e-05,
      "step": 6029,
      "training_step_time": 10.004263877868652
    },
    {
      "epoch": 2.300262451171875e-05,
      "grad_norm": 0.5060502290725708,
      "learning_rate": 1e-05,
      "loss": 0.0894,
      "step": 6030
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 2.4689202308654785,
      "step": 6030
    },
    {
      "dynamics_loss_avg": 0.0590103754773736,
      "epoch": 2.300262451171875e-05,
      "step": 6030
    },
    {
      "action_loss_avg": 0.029657906340435147,
      "epoch": 2.300262451171875e-05,
      "step": 6030
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 6030,
      "training_step_time": 7.497021913528442
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 3.134748935699463,
      "step": 6030
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 6030,
      "training_step_time": 9.906489610671997
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 3.1365509033203125,
      "step": 6030
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 6030,
      "training_step_time": 9.906312465667725
    },
    {
      "epoch": 2.300262451171875e-05,
      "model_forward_time": 3.134918212890625,
      "step": 6030
    },
    {
      "epoch": 2.300262451171875e-05,
      "step": 6030,
      "training_step_time": 10.273755550384521
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 3.4812140464782715,
      "step": 6031
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 6031,
      "training_step_time": 10.251957893371582
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 3.1412460803985596,
      "step": 6031
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 6031,
      "training_step_time": 9.915435314178467
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 3.147442579269409,
      "step": 6031
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 6031,
      "training_step_time": 9.920150995254517
    },
    {
      "epoch": 2.3006439208984374e-05,
      "model_forward_time": 3.143747329711914,
      "step": 6031
    },
    {
      "epoch": 2.3006439208984374e-05,
      "step": 6031,
      "training_step_time": 10.325623989105225
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 3.4723284244537354,
      "step": 6032
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 6032,
      "training_step_time": 10.246075630187988
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 3.1388330459594727,
      "step": 6032
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 6032,
      "training_step_time": 9.911241292953491
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 3.138183116912842,
      "step": 6032
    },
    {
      "dynamics_loss_avg": 0.05678938888013363,
      "epoch": 2.301025390625e-05,
      "step": 6032
    },
    {
      "action_loss_avg": 0.022752878814935686,
      "epoch": 2.301025390625e-05,
      "step": 6032
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 6032,
      "training_step_time": 9.908899784088135
    },
    {
      "epoch": 2.301025390625e-05,
      "model_forward_time": 3.1384899616241455,
      "step": 6032
    },
    {
      "epoch": 2.301025390625e-05,
      "step": 6032,
      "training_step_time": 10.32745909690857
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 3.163881778717041,
      "step": 6033
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 6033,
      "training_step_time": 9.933204412460327
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 3.1330654621124268,
      "step": 6033
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 6033,
      "training_step_time": 10.17558479309082
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 3.131589412689209,
      "step": 6033
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 6033,
      "training_step_time": 9.896892786026001
    },
    {
      "epoch": 2.3014068603515626e-05,
      "model_forward_time": 3.1327078342437744,
      "step": 6033
    },
    {
      "epoch": 2.3014068603515626e-05,
      "step": 6033,
      "training_step_time": 10.001341342926025
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 3.173778772354126,
      "step": 6034
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 6034,
      "training_step_time": 9.941680908203125
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 2.429135322570801,
      "step": 6034
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 6034,
      "training_step_time": 7.454282522201538
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 3.1357202529907227,
      "step": 6034
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 6034,
      "training_step_time": 10.17392611503601
    },
    {
      "epoch": 2.301788330078125e-05,
      "model_forward_time": 3.138313055038452,
      "step": 6034
    },
    {
      "epoch": 2.301788330078125e-05,
      "step": 6034,
      "training_step_time": 10.011744499206543
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 3.177548408508301,
      "step": 6035
    },
    {
      "dynamics_loss_avg": 0.07042440958321095,
      "epoch": 2.3021697998046876e-05,
      "step": 6035
    },
    {
      "action_loss_avg": 0.032879261299967766,
      "epoch": 2.3021697998046876e-05,
      "step": 6035
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 6035,
      "training_step_time": 9.960030794143677
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 3.138185739517212,
      "step": 6035
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 6035,
      "training_step_time": 9.90485167503357
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 3.138958692550659,
      "step": 6035
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 6035,
      "training_step_time": 9.906400442123413
    },
    {
      "epoch": 2.3021697998046876e-05,
      "model_forward_time": 3.4219651222229004,
      "step": 6035
    },
    {
      "epoch": 2.3021697998046876e-05,
      "step": 6035,
      "training_step_time": 10.283385038375854
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 3.1561217308044434,
      "step": 6036
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 6036,
      "training_step_time": 9.932826042175293
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 3.1339921951293945,
      "step": 6036
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 6036,
      "training_step_time": 9.901209592819214
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 2.4369823932647705,
      "step": 6036
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 6036,
      "training_step_time": 7.462656497955322
    },
    {
      "epoch": 2.30255126953125e-05,
      "model_forward_time": 2.4343016147613525,
      "step": 6036
    },
    {
      "epoch": 2.30255126953125e-05,
      "step": 6036,
      "training_step_time": 10.319764614105225
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 3.1713225841522217,
      "step": 6037
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 6037,
      "training_step_time": 9.939838647842407
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 3.37955904006958,
      "step": 6037
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 6037,
      "training_step_time": 10.144219160079956
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 3.137953758239746,
      "step": 6037
    },
    {
      "dynamics_loss_avg": 0.06068700812757015,
      "epoch": 2.3029327392578125e-05,
      "step": 6037
    },
    {
      "action_loss_avg": 0.02784752966836095,
      "epoch": 2.3029327392578125e-05,
      "step": 6037
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 6037,
      "training_step_time": 9.913947582244873
    },
    {
      "epoch": 2.3029327392578125e-05,
      "model_forward_time": 3.1441969871520996,
      "step": 6037
    },
    {
      "epoch": 2.3029327392578125e-05,
      "step": 6037,
      "training_step_time": 10.278757572174072
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 3.1738028526306152,
      "step": 6038
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 6038,
      "training_step_time": 9.939018249511719
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 3.138577699661255,
      "step": 6038
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 6038,
      "training_step_time": 10.172750234603882
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 3.1367104053497314,
      "step": 6038
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 6038,
      "training_step_time": 9.900947332382202
    },
    {
      "epoch": 2.303314208984375e-05,
      "model_forward_time": 3.135695457458496,
      "step": 6038
    },
    {
      "epoch": 2.303314208984375e-05,
      "step": 6038,
      "training_step_time": 10.260022640228271
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 3.1750118732452393,
      "step": 6039
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 6039,
      "training_step_time": 9.945040941238403
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 3.1416025161743164,
      "step": 6039
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 6039,
      "training_step_time": 9.917212963104248
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 3.1371116638183594,
      "step": 6039
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 6039,
      "training_step_time": 10.184627532958984
    },
    {
      "epoch": 2.3036956787109374e-05,
      "model_forward_time": 3.1426990032196045,
      "step": 6039
    },
    {
      "epoch": 2.3036956787109374e-05,
      "step": 6039,
      "training_step_time": 10.269660711288452
    },
    {
      "epoch": 2.3040771484375e-05,
      "grad_norm": 0.5499476194381714,
      "learning_rate": 1e-05,
      "loss": 0.0845,
      "step": 6040
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 3.1547279357910156,
      "step": 6040
    },
    {
      "dynamics_loss_avg": 0.07129070777446031,
      "epoch": 2.3040771484375e-05,
      "step": 6040
    },
    {
      "action_loss_avg": 0.028535846062004565,
      "epoch": 2.3040771484375e-05,
      "step": 6040
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 6040,
      "training_step_time": 9.937386989593506
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 2.445277214050293,
      "step": 6040
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 6040,
      "training_step_time": 7.475320339202881
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 3.13590407371521,
      "step": 6040
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 6040,
      "training_step_time": 9.907140493392944
    },
    {
      "epoch": 2.3040771484375e-05,
      "model_forward_time": 2.4286797046661377,
      "step": 6040
    },
    {
      "epoch": 2.3040771484375e-05,
      "step": 6040,
      "training_step_time": 10.288287162780762
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 3.1531291007995605,
      "step": 6041
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 6041,
      "training_step_time": 9.91939401626587
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 3.13777494430542,
      "step": 6041
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 6041,
      "training_step_time": 9.911349058151245
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 1.740983009338379,
      "step": 6041
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 6041,
      "training_step_time": 5.195215940475464
    },
    {
      "epoch": 2.3044586181640626e-05,
      "model_forward_time": 3.131937026977539,
      "step": 6041
    },
    {
      "epoch": 2.3044586181640626e-05,
      "step": 6041,
      "training_step_time": 10.027837753295898
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 3.1507039070129395,
      "step": 6042
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 6042,
      "training_step_time": 9.925506830215454
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 2.701181650161743,
      "step": 6042
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 6042,
      "training_step_time": 7.726458549499512
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 3.1260623931884766,
      "step": 6042
    },
    {
      "dynamics_loss_avg": 0.047525372728705405,
      "epoch": 2.304840087890625e-05,
      "step": 6042
    },
    {
      "action_loss_avg": 0.027117210952565075,
      "epoch": 2.304840087890625e-05,
      "step": 6042
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 6042,
      "training_step_time": 9.895399570465088
    },
    {
      "epoch": 2.304840087890625e-05,
      "model_forward_time": 3.128682851791382,
      "step": 6042
    },
    {
      "epoch": 2.304840087890625e-05,
      "step": 6042,
      "training_step_time": 10.314005613327026
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 3.184337615966797,
      "step": 6043
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 6043,
      "training_step_time": 9.957032918930054
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 3.13189697265625,
      "step": 6043
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 6043,
      "training_step_time": 9.898102521896362
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 3.3791589736938477,
      "step": 6043
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 6043,
      "training_step_time": 10.148690938949585
    },
    {
      "epoch": 2.3052215576171876e-05,
      "model_forward_time": 3.1350719928741455,
      "step": 6043
    },
    {
      "epoch": 2.3052215576171876e-05,
      "step": 6043,
      "training_step_time": 10.271078824996948
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 3.1549012660980225,
      "step": 6044
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 6044,
      "training_step_time": 9.925933837890625
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 3.1310267448425293,
      "step": 6044
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 6044,
      "training_step_time": 9.901885032653809
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 2.43182110786438,
      "step": 6044
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 6044,
      "training_step_time": 7.466843128204346
    },
    {
      "epoch": 2.30560302734375e-05,
      "model_forward_time": 3.4149739742279053,
      "step": 6044
    },
    {
      "epoch": 2.30560302734375e-05,
      "step": 6044,
      "training_step_time": 10.291256666183472
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 3.1522600650787354,
      "step": 6045
    },
    {
      "dynamics_loss_avg": 0.05937941912561655,
      "epoch": 2.3059844970703125e-05,
      "step": 6045
    },
    {
      "action_loss_avg": 0.028039384353905916,
      "epoch": 2.3059844970703125e-05,
      "step": 6045
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 6045,
      "training_step_time": 9.93005657196045
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 3.1333377361297607,
      "step": 6045
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 6045,
      "training_step_time": 9.90559720993042
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 3.1300387382507324,
      "step": 6045
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 6045,
      "training_step_time": 9.897138118743896
    },
    {
      "epoch": 2.3059844970703125e-05,
      "model_forward_time": 2.429265022277832,
      "step": 6045
    },
    {
      "epoch": 2.3059844970703125e-05,
      "step": 6045,
      "training_step_time": 10.22597599029541
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 3.442833185195923,
      "step": 6046
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 6046,
      "training_step_time": 10.21643853187561
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 2.430280923843384,
      "step": 6046
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 6046,
      "training_step_time": 7.456471920013428
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 2.4319639205932617,
      "step": 6046
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 6046,
      "training_step_time": 7.462568521499634
    },
    {
      "epoch": 2.306365966796875e-05,
      "model_forward_time": 3.12603497505188,
      "step": 6046
    },
    {
      "epoch": 2.306365966796875e-05,
      "step": 6046,
      "training_step_time": 10.012121677398682
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 3.14540696144104,
      "step": 6047
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 6047,
      "training_step_time": 9.921794891357422
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 3.399951696395874,
      "step": 6047
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 6047,
      "training_step_time": 10.175277709960938
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 3.15382981300354,
      "step": 6047
    },
    {
      "dynamics_loss_avg": 0.06975752683356404,
      "epoch": 2.3067474365234374e-05,
      "step": 6047
    },
    {
      "action_loss_avg": 0.029818449227605015,
      "epoch": 2.3067474365234374e-05,
      "step": 6047
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 6047,
      "training_step_time": 9.933029651641846
    },
    {
      "epoch": 2.3067474365234374e-05,
      "model_forward_time": 3.1362674236297607,
      "step": 6047
    },
    {
      "epoch": 2.3067474365234374e-05,
      "step": 6047,
      "training_step_time": 10.299427509307861
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 3.1677558422088623,
      "step": 6048
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 6048,
      "training_step_time": 9.946973085403442
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 2.437406063079834,
      "step": 6048
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 6048,
      "training_step_time": 7.472882986068726
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 3.4134581089019775,
      "step": 6048
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 6048,
      "training_step_time": 10.191239833831787
    },
    {
      "epoch": 2.30712890625e-05,
      "model_forward_time": 3.1393585205078125,
      "step": 6048
    },
    {
      "epoch": 2.30712890625e-05,
      "step": 6048,
      "training_step_time": 10.284252643585205
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 3.1632957458496094,
      "step": 6049
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 6049,
      "training_step_time": 9.942553043365479
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 3.137345552444458,
      "step": 6049
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 6049,
      "training_step_time": 9.913949251174927
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 3.1343531608581543,
      "step": 6049
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 6049,
      "training_step_time": 9.905519962310791
    },
    {
      "epoch": 2.3075103759765626e-05,
      "model_forward_time": 3.3866536617279053,
      "step": 6049
    },
    {
      "epoch": 2.3075103759765626e-05,
      "step": 6049,
      "training_step_time": 10.25663423538208
    },
    {
      "epoch": 2.307891845703125e-05,
      "grad_norm": 0.6048067808151245,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 6050
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 3.1651220321655273,
      "step": 6050
    },
    {
      "dynamics_loss_avg": 0.0771242605522275,
      "epoch": 2.307891845703125e-05,
      "step": 6050
    },
    {
      "action_loss_avg": 0.040716790780425074,
      "epoch": 2.307891845703125e-05,
      "step": 6050
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 6050,
      "training_step_time": 9.94717526435852
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 3.137786626815796,
      "step": 6050
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 6050,
      "training_step_time": 9.914466381072998
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 3.135749578475952,
      "step": 6050
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 6050,
      "training_step_time": 9.91078782081604
    },
    {
      "epoch": 2.307891845703125e-05,
      "model_forward_time": 3.138512372970581,
      "step": 6050
    },
    {
      "epoch": 2.307891845703125e-05,
      "step": 6050,
      "training_step_time": 10.286580801010132
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 3.3929550647735596,
      "step": 6051
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 6051,
      "training_step_time": 10.175469398498535
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 3.1414268016815186,
      "step": 6051
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 6051,
      "training_step_time": 9.918269872665405
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 3.1366894245147705,
      "step": 6051
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 6051,
      "training_step_time": 9.916994094848633
    },
    {
      "epoch": 2.3082733154296876e-05,
      "model_forward_time": 3.1394569873809814,
      "step": 6051
    },
    {
      "epoch": 2.3082733154296876e-05,
      "step": 6051,
      "training_step_time": 10.020744562149048
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 3.178481101989746,
      "step": 6052
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 6052,
      "training_step_time": 9.957067728042603
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 2.7077579498291016,
      "step": 6052
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 6052,
      "training_step_time": 7.739243745803833
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 3.1323986053466797,
      "step": 6052
    },
    {
      "dynamics_loss_avg": 0.06165993474423885,
      "epoch": 2.30865478515625e-05,
      "step": 6052
    },
    {
      "action_loss_avg": 0.02693795906379819,
      "epoch": 2.30865478515625e-05,
      "step": 6052
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 6052,
      "training_step_time": 9.902408599853516
    },
    {
      "epoch": 2.30865478515625e-05,
      "model_forward_time": 3.1362383365631104,
      "step": 6052
    },
    {
      "epoch": 2.30865478515625e-05,
      "step": 6052,
      "training_step_time": 10.313833475112915
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 2.4440884590148926,
      "step": 6053
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 6053,
      "training_step_time": 7.469222068786621
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 3.130934000015259,
      "step": 6053
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 6053,
      "training_step_time": 9.903982639312744
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 3.387035608291626,
      "step": 6053
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 6053,
      "training_step_time": 10.159815073013306
    },
    {
      "epoch": 2.3090362548828125e-05,
      "model_forward_time": 3.141418218612671,
      "step": 6053
    },
    {
      "epoch": 2.3090362548828125e-05,
      "step": 6053,
      "training_step_time": 10.346202611923218
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 3.1531119346618652,
      "step": 6054
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 6054,
      "training_step_time": 9.925077438354492
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 3.131160259246826,
      "step": 6054
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 6054,
      "training_step_time": 9.91445279121399
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 3.152251958847046,
      "step": 6054
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 6054,
      "training_step_time": 9.921706438064575
    },
    {
      "epoch": 2.309417724609375e-05,
      "model_forward_time": 3.4244823455810547,
      "step": 6054
    },
    {
      "epoch": 2.309417724609375e-05,
      "step": 6054,
      "training_step_time": 10.290784358978271
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 2.4937734603881836,
      "step": 6055
    },
    {
      "dynamics_loss_avg": 0.06214235424995422,
      "epoch": 2.3097991943359374e-05,
      "step": 6055
    },
    {
      "action_loss_avg": 0.024363067001104355,
      "epoch": 2.3097991943359374e-05,
      "step": 6055
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 6055,
      "training_step_time": 7.527742862701416
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 3.1292903423309326,
      "step": 6055
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 6055,
      "training_step_time": 9.90482234954834
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 3.1322860717773438,
      "step": 6055
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 6055,
      "training_step_time": 9.911263227462769
    },
    {
      "epoch": 2.3097991943359374e-05,
      "model_forward_time": 3.1331474781036377,
      "step": 6055
    },
    {
      "epoch": 2.3097991943359374e-05,
      "step": 6055,
      "training_step_time": 10.004835367202759
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 3.419734239578247,
      "step": 6056
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 6056,
      "training_step_time": 10.189874649047852
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 3.1337857246398926,
      "step": 6056
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 6056,
      "training_step_time": 9.912335634231567
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 2.4354307651519775,
      "step": 6056
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 6056,
      "training_step_time": 7.468048572540283
    },
    {
      "epoch": 2.3101806640625e-05,
      "model_forward_time": 3.1339287757873535,
      "step": 6056
    },
    {
      "epoch": 2.3101806640625e-05,
      "step": 6056,
      "training_step_time": 10.290022134780884
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 3.1500983238220215,
      "step": 6057
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 6057,
      "training_step_time": 9.931082963943481
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 3.4070584774017334,
      "step": 6057
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 6057,
      "training_step_time": 10.185997486114502
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 3.140475034713745,
      "step": 6057
    },
    {
      "dynamics_loss_avg": 0.060148551128804684,
      "epoch": 2.3105621337890626e-05,
      "step": 6057
    },
    {
      "action_loss_avg": 0.02281406761612743,
      "epoch": 2.3105621337890626e-05,
      "step": 6057
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 6057,
      "training_step_time": 9.920786380767822
    },
    {
      "epoch": 2.3105621337890626e-05,
      "model_forward_time": 3.138068199157715,
      "step": 6057
    },
    {
      "epoch": 2.3105621337890626e-05,
      "step": 6057,
      "training_step_time": 10.29092264175415
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 3.16571307182312,
      "step": 6058
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 6058,
      "training_step_time": 9.946048259735107
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 3.1353635787963867,
      "step": 6058
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 6058,
      "training_step_time": 9.911659717559814
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 3.4284934997558594,
      "step": 6058
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 6058,
      "training_step_time": 10.207300662994385
    },
    {
      "epoch": 2.310943603515625e-05,
      "model_forward_time": 3.1296350955963135,
      "step": 6058
    },
    {
      "epoch": 2.310943603515625e-05,
      "step": 6058,
      "training_step_time": 10.267542362213135
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 3.140428304672241,
      "step": 6059
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 6059,
      "training_step_time": 9.917845010757446
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 3.1363518238067627,
      "step": 6059
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 6059,
      "training_step_time": 9.91334581375122
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 3.13185977935791,
      "step": 6059
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 6059,
      "training_step_time": 9.909996032714844
    },
    {
      "epoch": 2.3113250732421876e-05,
      "model_forward_time": 2.6900417804718018,
      "step": 6059
    },
    {
      "epoch": 2.3113250732421876e-05,
      "step": 6059,
      "training_step_time": 10.265507936477661
    },
    {
      "epoch": 2.31170654296875e-05,
      "grad_norm": 0.4562627077102661,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 6060
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 2.4399125576019287,
      "step": 6060
    },
    {
      "dynamics_loss_avg": 0.052041835710406305,
      "epoch": 2.31170654296875e-05,
      "step": 6060
    },
    {
      "action_loss_avg": 0.020019686268642544,
      "epoch": 2.31170654296875e-05,
      "step": 6060
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 6060,
      "training_step_time": 7.4840075969696045
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 2.4309051036834717,
      "step": 6060
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 6060,
      "training_step_time": 7.45197868347168
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 3.13321590423584,
      "step": 6060
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 6060,
      "training_step_time": 9.901371717453003
    },
    {
      "epoch": 2.31170654296875e-05,
      "model_forward_time": 3.1360981464385986,
      "step": 6060
    },
    {
      "epoch": 2.31170654296875e-05,
      "step": 6060,
      "training_step_time": 10.017623901367188
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 3.4423580169677734,
      "step": 6061
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 6061,
      "training_step_time": 10.22032880783081
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 3.1413378715515137,
      "step": 6061
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 6061,
      "training_step_time": 9.908868312835693
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 3.137524127960205,
      "step": 6061
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 6061,
      "training_step_time": 9.905452489852905
    },
    {
      "epoch": 2.3120880126953125e-05,
      "model_forward_time": 3.1419146060943604,
      "step": 6061
    },
    {
      "epoch": 2.3120880126953125e-05,
      "step": 6061,
      "training_step_time": 10.295379638671875
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 3.1879045963287354,
      "step": 6062
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 6062,
      "training_step_time": 9.958137273788452
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 3.410038948059082,
      "step": 6062
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 6062,
      "training_step_time": 10.176758050918579
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 3.138289213180542,
      "step": 6062
    },
    {
      "dynamics_loss_avg": 0.061626937985420224,
      "epoch": 2.312469482421875e-05,
      "step": 6062
    },
    {
      "action_loss_avg": 0.020998962875455617,
      "epoch": 2.312469482421875e-05,
      "step": 6062
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 6062,
      "training_step_time": 9.909259557723999
    },
    {
      "epoch": 2.312469482421875e-05,
      "model_forward_time": 3.1522910594940186,
      "step": 6062
    },
    {
      "epoch": 2.312469482421875e-05,
      "step": 6062,
      "training_step_time": 10.265926599502563
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 3.181551694869995,
      "step": 6063
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 6063,
      "training_step_time": 9.950074195861816
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 3.1336874961853027,
      "step": 6063
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 6063,
      "training_step_time": 10.175671339035034
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 3.1348352432250977,
      "step": 6063
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 6063,
      "training_step_time": 9.899695873260498
    },
    {
      "epoch": 2.3128509521484374e-05,
      "model_forward_time": 3.1366589069366455,
      "step": 6063
    },
    {
      "epoch": 2.3128509521484374e-05,
      "step": 6063,
      "training_step_time": 10.27973985671997
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 3.152965545654297,
      "step": 6064
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 6064,
      "training_step_time": 9.9275803565979
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 3.1351304054260254,
      "step": 6064
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 6064,
      "training_step_time": 9.903660297393799
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 2.710178852081299,
      "step": 6064
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 6064,
      "training_step_time": 7.733031988143921
    },
    {
      "epoch": 2.313232421875e-05,
      "model_forward_time": 3.134084701538086,
      "step": 6064
    },
    {
      "epoch": 2.313232421875e-05,
      "step": 6064,
      "training_step_time": 10.284960985183716
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 3.1669259071350098,
      "step": 6065
    },
    {
      "dynamics_loss_avg": 0.05052713165059686,
      "epoch": 2.3136138916015626e-05,
      "step": 6065
    },
    {
      "action_loss_avg": 0.021397690242156386,
      "epoch": 2.3136138916015626e-05,
      "step": 6065
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 6065,
      "training_step_time": 9.935248374938965
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 3.1339242458343506,
      "step": 6065
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 6065,
      "training_step_time": 9.906622886657715
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 3.1386303901672363,
      "step": 6065
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 6065,
      "training_step_time": 9.909155368804932
    },
    {
      "epoch": 2.3136138916015626e-05,
      "model_forward_time": 3.137545585632324,
      "step": 6065
    },
    {
      "epoch": 2.3136138916015626e-05,
      "step": 6065,
      "training_step_time": 10.271151304244995
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 3.150876522064209,
      "step": 6066
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 6066,
      "training_step_time": 9.922160148620605
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 3.133193016052246,
      "step": 6066
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 6066,
      "training_step_time": 9.900233030319214
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 3.136329174041748,
      "step": 6066
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 6066,
      "training_step_time": 9.905255317687988
    },
    {
      "epoch": 2.313995361328125e-05,
      "model_forward_time": 3.1319963932037354,
      "step": 6066
    },
    {
      "epoch": 2.313995361328125e-05,
      "step": 6066,
      "training_step_time": 10.011173248291016
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 3.4754226207733154,
      "step": 6067
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 6067,
      "training_step_time": 10.245517253875732
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 3.141649007797241,
      "step": 6067
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 6067,
      "training_step_time": 9.90842318534851
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 3.144768238067627,
      "step": 6067
    },
    {
      "dynamics_loss_avg": 0.05808682553470135,
      "epoch": 2.3143768310546876e-05,
      "step": 6067
    },
    {
      "action_loss_avg": 0.02599760997109115,
      "epoch": 2.3143768310546876e-05,
      "step": 6067
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 6067,
      "training_step_time": 9.915864944458008
    },
    {
      "epoch": 2.3143768310546876e-05,
      "model_forward_time": 3.1429519653320312,
      "step": 6067
    },
    {
      "epoch": 2.3143768310546876e-05,
      "step": 6067,
      "training_step_time": 10.258406162261963
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 3.1461663246154785,
      "step": 6068
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 6068,
      "training_step_time": 9.920054197311401
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 2.4388554096221924,
      "step": 6068
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 6068,
      "training_step_time": 7.738997459411621
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 3.1398632526397705,
      "step": 6068
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 6068,
      "training_step_time": 9.909444570541382
    },
    {
      "epoch": 2.31475830078125e-05,
      "model_forward_time": 3.137726306915283,
      "step": 6068
    },
    {
      "epoch": 2.31475830078125e-05,
      "step": 6068,
      "training_step_time": 10.293090343475342
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 3.162318229675293,
      "step": 6069
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 6069,
      "training_step_time": 9.937414407730103
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 3.1361920833587646,
      "step": 6069
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 6069,
      "training_step_time": 9.911277055740356
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 2.4382712841033936,
      "step": 6069
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 6069,
      "training_step_time": 7.7773261070251465
    },
    {
      "epoch": 2.3151397705078125e-05,
      "model_forward_time": 3.142561674118042,
      "step": 6069
    },
    {
      "epoch": 2.3151397705078125e-05,
      "step": 6069,
      "training_step_time": 10.298725366592407
    },
    {
      "epoch": 2.315521240234375e-05,
      "grad_norm": 0.5665556192398071,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 6070
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 3.1691040992736816,
      "step": 6070
    },
    {
      "dynamics_loss_avg": 0.0661770474165678,
      "epoch": 2.315521240234375e-05,
      "step": 6070
    },
    {
      "action_loss_avg": 0.029431446641683578,
      "epoch": 2.315521240234375e-05,
      "step": 6070
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 6070,
      "training_step_time": 9.950866460800171
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 3.1474246978759766,
      "step": 6070
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 6070,
      "training_step_time": 9.919430255889893
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 2.440098762512207,
      "step": 6070
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 6070,
      "training_step_time": 7.468530893325806
    },
    {
      "epoch": 2.315521240234375e-05,
      "model_forward_time": 3.146217107772827,
      "step": 6070
    },
    {
      "epoch": 2.315521240234375e-05,
      "step": 6070,
      "training_step_time": 10.28537654876709
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 3.4531333446502686,
      "step": 6071
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 6071,
      "training_step_time": 10.226645708084106
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 3.1422080993652344,
      "step": 6071
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 6071,
      "training_step_time": 9.911638498306274
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 3.136256456375122,
      "step": 6071
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 6071,
      "training_step_time": 9.898895978927612
    },
    {
      "epoch": 2.3159027099609374e-05,
      "model_forward_time": 3.135657548904419,
      "step": 6071
    },
    {
      "epoch": 2.3159027099609374e-05,
      "step": 6071,
      "training_step_time": 10.268348217010498
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 2.464810609817505,
      "step": 6072
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 6072,
      "training_step_time": 7.487316846847534
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 3.4039180278778076,
      "step": 6072
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 6072,
      "training_step_time": 10.168846845626831
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 3.1388120651245117,
      "step": 6072
    },
    {
      "dynamics_loss_avg": 0.05014958754181862,
      "epoch": 2.3162841796875e-05,
      "step": 6072
    },
    {
      "action_loss_avg": 0.020317395590245722,
      "epoch": 2.3162841796875e-05,
      "step": 6072
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 6072,
      "training_step_time": 9.911494970321655
    },
    {
      "epoch": 2.3162841796875e-05,
      "model_forward_time": 3.1446166038513184,
      "step": 6072
    },
    {
      "epoch": 2.3162841796875e-05,
      "step": 6072,
      "training_step_time": 10.01064944267273
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 3.153730869293213,
      "step": 6073
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 6073,
      "training_step_time": 9.926570653915405
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 3.4263670444488525,
      "step": 6073
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 6073,
      "training_step_time": 10.201260089874268
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 3.1431519985198975,
      "step": 6073
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 6073,
      "training_step_time": 9.91749119758606
    },
    {
      "epoch": 2.3166656494140627e-05,
      "model_forward_time": 2.4384665489196777,
      "step": 6073
    },
    {
      "epoch": 2.3166656494140627e-05,
      "step": 6073,
      "training_step_time": 10.292852878570557
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 3.161179542541504,
      "step": 6074
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 6074,
      "training_step_time": 9.936136722564697
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 3.142025947570801,
      "step": 6074
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 6074,
      "training_step_time": 9.91488790512085
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 3.145332098007202,
      "step": 6074
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 6074,
      "training_step_time": 9.919647693634033
    },
    {
      "epoch": 2.317047119140625e-05,
      "model_forward_time": 2.710038423538208,
      "step": 6074
    },
    {
      "epoch": 2.317047119140625e-05,
      "step": 6074,
      "training_step_time": 10.017913341522217
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 3.1790246963500977,
      "step": 6075
    },
    {
      "dynamics_loss_avg": 0.05802565459161997,
      "epoch": 2.3174285888671876e-05,
      "step": 6075
    },
    {
      "action_loss_avg": 0.023660630360245703,
      "epoch": 2.3174285888671876e-05,
      "step": 6075
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 6075,
      "training_step_time": 9.951314687728882
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 3.1337037086486816,
      "step": 6075
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 6075,
      "training_step_time": 9.89869737625122
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 3.138197422027588,
      "step": 6075
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 6075,
      "training_step_time": 9.9022216796875
    },
    {
      "epoch": 2.3174285888671876e-05,
      "model_forward_time": 3.4218125343322754,
      "step": 6075
    },
    {
      "epoch": 2.3174285888671876e-05,
      "step": 6075,
      "training_step_time": 10.310667514801025
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 3.187678337097168,
      "step": 6076
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 6076,
      "training_step_time": 9.952842235565186
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 3.1387500762939453,
      "step": 6076
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 6076,
      "training_step_time": 9.897918939590454
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 3.1390459537506104,
      "step": 6076
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 6076,
      "training_step_time": 9.903797626495361
    },
    {
      "epoch": 2.31781005859375e-05,
      "model_forward_time": 3.1403746604919434,
      "step": 6076
    },
    {
      "epoch": 2.31781005859375e-05,
      "step": 6076,
      "training_step_time": 10.009883880615234
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 3.1478219032287598,
      "step": 6077
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 6077,
      "training_step_time": 10.1741464138031
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 3.1352367401123047,
      "step": 6077
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 6077,
      "training_step_time": 9.90395998954773
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 3.1411490440368652,
      "step": 6077
    },
    {
      "dynamics_loss_avg": 0.059018587321043016,
      "epoch": 2.3181915283203125e-05,
      "step": 6077
    },
    {
      "action_loss_avg": 0.028268987592309714,
      "epoch": 2.3181915283203125e-05,
      "step": 6077
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 6077,
      "training_step_time": 9.911355018615723
    },
    {
      "epoch": 2.3181915283203125e-05,
      "model_forward_time": 3.1387228965759277,
      "step": 6077
    },
    {
      "epoch": 2.3181915283203125e-05,
      "step": 6077,
      "training_step_time": 10.274747371673584
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 3.173422336578369,
      "step": 6078
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 6078,
      "training_step_time": 9.947243452072144
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 3.4104275703430176,
      "step": 6078
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 6078,
      "training_step_time": 10.190287828445435
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 3.145193099975586,
      "step": 6078
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 6078,
      "training_step_time": 9.920101165771484
    },
    {
      "epoch": 2.318572998046875e-05,
      "model_forward_time": 3.14750337600708,
      "step": 6078
    },
    {
      "epoch": 2.318572998046875e-05,
      "step": 6078,
      "training_step_time": 10.30351185798645
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 3.1752593517303467,
      "step": 6079
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 6079,
      "training_step_time": 9.955161094665527
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 2.4401252269744873,
      "step": 6079
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 6079,
      "training_step_time": 7.467447280883789
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 3.140169382095337,
      "step": 6079
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 6079,
      "training_step_time": 10.182124376296997
    },
    {
      "epoch": 2.3189544677734374e-05,
      "model_forward_time": 3.1384143829345703,
      "step": 6079
    },
    {
      "epoch": 2.3189544677734374e-05,
      "step": 6079,
      "training_step_time": 10.304188013076782
    },
    {
      "epoch": 2.3193359375e-05,
      "grad_norm": 0.39679765701293945,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 6080
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 3.168139696121216,
      "step": 6080
    },
    {
      "dynamics_loss_avg": 0.061620728485286234,
      "epoch": 2.3193359375e-05,
      "step": 6080
    },
    {
      "action_loss_avg": 0.026714064367115497,
      "epoch": 2.3193359375e-05,
      "step": 6080
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 6080,
      "training_step_time": 9.944981336593628
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 3.132521867752075,
      "step": 6080
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 6080,
      "training_step_time": 9.908864736557007
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 3.131831407546997,
      "step": 6080
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 6080,
      "training_step_time": 9.909368515014648
    },
    {
      "epoch": 2.3193359375e-05,
      "model_forward_time": 3.134328603744507,
      "step": 6080
    },
    {
      "epoch": 2.3193359375e-05,
      "step": 6080,
      "training_step_time": 10.330390453338623
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 3.1905670166015625,
      "step": 6081
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 6081,
      "training_step_time": 9.971391439437866
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 3.1468844413757324,
      "step": 6081
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 6081,
      "training_step_time": 9.923025846481323
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 3.1424124240875244,
      "step": 6081
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 6081,
      "training_step_time": 9.924193620681763
    },
    {
      "epoch": 2.3197174072265627e-05,
      "model_forward_time": 3.1428935527801514,
      "step": 6081
    },
    {
      "epoch": 2.3197174072265627e-05,
      "step": 6081,
      "training_step_time": 10.27811074256897
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 3.4395334720611572,
      "step": 6082
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 6082,
      "training_step_time": 10.216699600219727
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 3.1425371170043945,
      "step": 6082
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 6082,
      "training_step_time": 9.922234773635864
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 2.43863844871521,
      "step": 6082
    },
    {
      "dynamics_loss_avg": 0.06404312569648027,
      "epoch": 2.320098876953125e-05,
      "step": 6082
    },
    {
      "action_loss_avg": 0.025769960973411797,
      "epoch": 2.320098876953125e-05,
      "step": 6082
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 6082,
      "training_step_time": 7.474146366119385
    },
    {
      "epoch": 2.320098876953125e-05,
      "model_forward_time": 3.1395263671875,
      "step": 6082
    },
    {
      "epoch": 2.320098876953125e-05,
      "step": 6082,
      "training_step_time": 10.298604011535645
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 3.1547257900238037,
      "step": 6083
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 6083,
      "training_step_time": 9.938687562942505
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 3.1347947120666504,
      "step": 6083
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 6083,
      "training_step_time": 9.907609224319458
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 3.383915662765503,
      "step": 6083
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 6083,
      "training_step_time": 10.156417846679688
    },
    {
      "epoch": 2.3204803466796876e-05,
      "model_forward_time": 2.43764591217041,
      "step": 6083
    },
    {
      "epoch": 2.3204803466796876e-05,
      "step": 6083,
      "training_step_time": 10.280954122543335
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 2.452829360961914,
      "step": 6084
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 6084,
      "training_step_time": 7.480405330657959
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 3.127532482147217,
      "step": 6084
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 6084,
      "training_step_time": 9.899337768554688
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 3.133937120437622,
      "step": 6084
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 6084,
      "training_step_time": 9.906691074371338
    },
    {
      "epoch": 2.32086181640625e-05,
      "model_forward_time": 3.4027771949768066,
      "step": 6084
    },
    {
      "epoch": 2.32086181640625e-05,
      "step": 6084,
      "training_step_time": 10.271788597106934
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 3.1746110916137695,
      "step": 6085
    },
    {
      "dynamics_loss_avg": 0.045753059722483155,
      "epoch": 2.3212432861328125e-05,
      "step": 6085
    },
    {
      "action_loss_avg": 0.01461192206479609,
      "epoch": 2.3212432861328125e-05,
      "step": 6085
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 6085,
      "training_step_time": 9.959531307220459
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 3.1353719234466553,
      "step": 6085
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 6085,
      "training_step_time": 9.91175889968872
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 3.1383466720581055,
      "step": 6085
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 6085,
      "training_step_time": 9.917361497879028
    },
    {
      "epoch": 2.3212432861328125e-05,
      "model_forward_time": 3.1337525844573975,
      "step": 6085
    },
    {
      "epoch": 2.3212432861328125e-05,
      "step": 6085,
      "training_step_time": 10.287466049194336
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 2.7827606201171875,
      "step": 6086
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 6086,
      "training_step_time": 7.8134026527404785
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 3.128774404525757,
      "step": 6086
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 6086,
      "training_step_time": 9.902897119522095
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 3.1349668502807617,
      "step": 6086
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 6086,
      "training_step_time": 9.908839464187622
    },
    {
      "epoch": 2.321624755859375e-05,
      "model_forward_time": 3.1319186687469482,
      "step": 6086
    },
    {
      "epoch": 2.321624755859375e-05,
      "step": 6086,
      "training_step_time": 10.301388263702393
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 3.416954755783081,
      "step": 6087
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 6087,
      "training_step_time": 10.188639402389526
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 2.432476282119751,
      "step": 6087
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 6087,
      "training_step_time": 7.457981109619141
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 3.1311323642730713,
      "step": 6087
    },
    {
      "dynamics_loss_avg": 0.0443417850881815,
      "epoch": 2.3220062255859374e-05,
      "step": 6087
    },
    {
      "action_loss_avg": 0.02059124279767275,
      "epoch": 2.3220062255859374e-05,
      "step": 6087
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 6087,
      "training_step_time": 9.906761884689331
    },
    {
      "epoch": 2.3220062255859374e-05,
      "model_forward_time": 3.130699872970581,
      "step": 6087
    },
    {
      "epoch": 2.3220062255859374e-05,
      "step": 6087,
      "training_step_time": 10.276811838150024
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 2.4431545734405518,
      "step": 6088
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 6088,
      "training_step_time": 7.475438833236694
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 3.132805824279785,
      "step": 6088
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 6088,
      "training_step_time": 9.90615701675415
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 2.7006733417510986,
      "step": 6088
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 6088,
      "training_step_time": 7.730238676071167
    },
    {
      "epoch": 2.3223876953125e-05,
      "model_forward_time": 3.130147695541382,
      "step": 6088
    },
    {
      "epoch": 2.3223876953125e-05,
      "step": 6088,
      "training_step_time": 10.244318962097168
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 3.162977695465088,
      "step": 6089
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 6089,
      "training_step_time": 9.936847686767578
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 3.144679069519043,
      "step": 6089
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 6089,
      "training_step_time": 9.922353029251099
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 3.139651298522949,
      "step": 6089
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 6089,
      "training_step_time": 9.913752317428589
    },
    {
      "epoch": 2.3227691650390627e-05,
      "model_forward_time": 3.3851735591888428,
      "step": 6089
    },
    {
      "epoch": 2.3227691650390627e-05,
      "step": 6089,
      "training_step_time": 10.326196432113647
    },
    {
      "epoch": 2.323150634765625e-05,
      "grad_norm": 0.48746371269226074,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 6090
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 3.1625237464904785,
      "step": 6090
    },
    {
      "dynamics_loss_avg": 0.05552836786955595,
      "epoch": 2.323150634765625e-05,
      "step": 6090
    },
    {
      "action_loss_avg": 0.017851097323000432,
      "epoch": 2.323150634765625e-05,
      "step": 6090
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 6090,
      "training_step_time": 9.940006017684937
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 3.1418793201446533,
      "step": 6090
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 6090,
      "training_step_time": 9.920995712280273
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 3.137279987335205,
      "step": 6090
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 6090,
      "training_step_time": 9.909996271133423
    },
    {
      "epoch": 2.323150634765625e-05,
      "model_forward_time": 3.1508426666259766,
      "step": 6090
    },
    {
      "epoch": 2.323150634765625e-05,
      "step": 6090,
      "training_step_time": 10.251955509185791
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 3.458874225616455,
      "step": 6091
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 6091,
      "training_step_time": 10.23841667175293
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 3.1427745819091797,
      "step": 6091
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 6091,
      "training_step_time": 9.918996572494507
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 3.136345863342285,
      "step": 6091
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 6091,
      "training_step_time": 9.90903639793396
    },
    {
      "epoch": 2.3235321044921876e-05,
      "model_forward_time": 3.137902021408081,
      "step": 6091
    },
    {
      "epoch": 2.3235321044921876e-05,
      "step": 6091,
      "training_step_time": 10.239746570587158
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 3.186513662338257,
      "step": 6092
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 6092,
      "training_step_time": 10.230006217956543
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 2.438300609588623,
      "step": 6092
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 6092,
      "training_step_time": 7.467171907424927
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 3.1342451572418213,
      "step": 6092
    },
    {
      "dynamics_loss_avg": 0.07072525024414063,
      "epoch": 2.32391357421875e-05,
      "step": 6092
    },
    {
      "action_loss_avg": 0.0320530379191041,
      "epoch": 2.32391357421875e-05,
      "step": 6092
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 6092,
      "training_step_time": 9.905720233917236
    },
    {
      "epoch": 2.32391357421875e-05,
      "model_forward_time": 3.146275520324707,
      "step": 6092
    },
    {
      "epoch": 2.32391357421875e-05,
      "step": 6092,
      "training_step_time": 10.303500652313232
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 3.151501417160034,
      "step": 6093
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 6093,
      "training_step_time": 9.9267418384552
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 3.419005870819092,
      "step": 6093
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 6093,
      "training_step_time": 10.188906192779541
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 3.1398441791534424,
      "step": 6093
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 6093,
      "training_step_time": 9.912813663482666
    },
    {
      "epoch": 2.3242950439453125e-05,
      "model_forward_time": 2.4363884925842285,
      "step": 6093
    },
    {
      "epoch": 2.3242950439453125e-05,
      "step": 6093,
      "training_step_time": 10.009527444839478
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 3.1864054203033447,
      "step": 6094
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 6094,
      "training_step_time": 9.95711374282837
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 3.133923292160034,
      "step": 6094
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 6094,
      "training_step_time": 9.904671430587769
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 3.1405749320983887,
      "step": 6094
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 6094,
      "training_step_time": 9.909475326538086
    },
    {
      "epoch": 2.324676513671875e-05,
      "model_forward_time": 3.420606851577759,
      "step": 6094
    },
    {
      "epoch": 2.324676513671875e-05,
      "step": 6094,
      "training_step_time": 10.299343824386597
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 3.162311315536499,
      "step": 6095
    },
    {
      "dynamics_loss_avg": 0.05350617710500956,
      "epoch": 2.3250579833984374e-05,
      "step": 6095
    },
    {
      "action_loss_avg": 0.023717524390667676,
      "epoch": 2.3250579833984374e-05,
      "step": 6095
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 6095,
      "training_step_time": 9.940170764923096
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 3.1358158588409424,
      "step": 6095
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 6095,
      "training_step_time": 9.907557725906372
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 2.4351370334625244,
      "step": 6095
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 6095,
      "training_step_time": 7.462610483169556
    },
    {
      "epoch": 2.3250579833984374e-05,
      "model_forward_time": 3.133103847503662,
      "step": 6095
    },
    {
      "epoch": 2.3250579833984374e-05,
      "step": 6095,
      "training_step_time": 10.285753726959229
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 3.158496141433716,
      "step": 6096
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 6096,
      "training_step_time": 9.928219318389893
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 3.1377079486846924,
      "step": 6096
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 6096,
      "training_step_time": 9.910452127456665
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 2.434577465057373,
      "step": 6096
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 6096,
      "training_step_time": 7.462716341018677
    },
    {
      "epoch": 2.325439453125e-05,
      "model_forward_time": 2.436039924621582,
      "step": 6096
    },
    {
      "epoch": 2.325439453125e-05,
      "step": 6096,
      "training_step_time": 10.272743701934814
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 3.1542251110076904,
      "step": 6097
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 6097,
      "training_step_time": 9.934032917022705
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 3.401313543319702,
      "step": 6097
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 6097,
      "training_step_time": 10.172465562820435
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 3.1432249546051025,
      "step": 6097
    },
    {
      "dynamics_loss_avg": 0.06337932916358113,
      "epoch": 2.3258209228515627e-05,
      "step": 6097
    },
    {
      "action_loss_avg": 0.022086557932198048,
      "epoch": 2.3258209228515627e-05,
      "step": 6097
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 6097,
      "training_step_time": 9.91718053817749
    },
    {
      "epoch": 2.3258209228515627e-05,
      "model_forward_time": 2.4358396530151367,
      "step": 6097
    },
    {
      "epoch": 2.3258209228515627e-05,
      "step": 6097,
      "training_step_time": 10.282264709472656
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 2.475400924682617,
      "step": 6098
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 6098,
      "training_step_time": 7.497790098190308
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 3.1348323822021484,
      "step": 6098
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 6098,
      "training_step_time": 9.904099225997925
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 3.407710313796997,
      "step": 6098
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 6098,
      "training_step_time": 10.17632269859314
    },
    {
      "epoch": 2.326202392578125e-05,
      "model_forward_time": 3.1314857006073,
      "step": 6098
    },
    {
      "epoch": 2.326202392578125e-05,
      "step": 6098,
      "training_step_time": 10.005967378616333
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 2.4717535972595215,
      "step": 6099
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 6099,
      "training_step_time": 7.502997636795044
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 3.1303153038024902,
      "step": 6099
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 6099,
      "training_step_time": 9.898117780685425
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 3.1321189403533936,
      "step": 6099
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 6099,
      "training_step_time": 9.902170658111572
    },
    {
      "epoch": 2.3265838623046876e-05,
      "model_forward_time": 3.385944128036499,
      "step": 6099
    },
    {
      "epoch": 2.3265838623046876e-05,
      "step": 6099,
      "training_step_time": 10.27306079864502
    },
    {
      "epoch": 2.32696533203125e-05,
      "grad_norm": 0.46633151173591614,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 6100
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 3.1801161766052246,
      "step": 6100
    },
    {
      "dynamics_loss_avg": 0.057512703351676464,
      "epoch": 2.32696533203125e-05,
      "step": 6100
    },
    {
      "action_loss_avg": 0.021090183313935994,
      "epoch": 2.32696533203125e-05,
      "step": 6100
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 6100,
      "training_step_time": 9.956793308258057
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 3.139399766921997,
      "step": 6100
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 6100,
      "training_step_time": 9.915111064910889
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 3.1415607929229736,
      "step": 6100
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 6100,
      "training_step_time": 9.915799856185913
    },
    {
      "epoch": 2.32696533203125e-05,
      "model_forward_time": 3.1380650997161865,
      "step": 6100
    },
    {
      "epoch": 2.32696533203125e-05,
      "step": 6100,
      "training_step_time": 10.005752325057983
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 3.4473061561584473,
      "step": 6101
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 6101,
      "training_step_time": 10.219973802566528
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 3.1369528770446777,
      "step": 6101
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 6101,
      "training_step_time": 9.905070066452026
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 3.132732629776001,
      "step": 6101
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 6101,
      "training_step_time": 9.898483037948608
    },
    {
      "epoch": 2.3273468017578125e-05,
      "model_forward_time": 3.1290225982666016,
      "step": 6101
    },
    {
      "epoch": 2.3273468017578125e-05,
      "step": 6101,
      "training_step_time": 10.257883071899414
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 3.143472194671631,
      "step": 6102
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 6102,
      "training_step_time": 9.90984582901001
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 3.398674726486206,
      "step": 6102
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 6102,
      "training_step_time": 10.160017967224121
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 3.129312038421631,
      "step": 6102
    },
    {
      "dynamics_loss_avg": 0.05730808200314641,
      "epoch": 2.327728271484375e-05,
      "step": 6102
    },
    {
      "action_loss_avg": 0.02233234872110188,
      "epoch": 2.327728271484375e-05,
      "step": 6102
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 6102,
      "training_step_time": 9.894988059997559
    },
    {
      "epoch": 2.327728271484375e-05,
      "model_forward_time": 2.4300730228424072,
      "step": 6102
    },
    {
      "epoch": 2.327728271484375e-05,
      "step": 6102,
      "training_step_time": 10.287334203720093
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 3.1647114753723145,
      "step": 6103
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 6103,
      "training_step_time": 9.925453901290894
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 3.1268372535705566,
      "step": 6103
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 6103,
      "training_step_time": 9.895411014556885
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 3.3729591369628906,
      "step": 6103
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 6103,
      "training_step_time": 10.138410568237305
    },
    {
      "epoch": 2.3281097412109374e-05,
      "model_forward_time": 3.12921404838562,
      "step": 6103
    },
    {
      "epoch": 2.3281097412109374e-05,
      "step": 6103,
      "training_step_time": 10.273679256439209
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 2.466691493988037,
      "step": 6104
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 6104,
      "training_step_time": 7.491374492645264
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 3.1247754096984863,
      "step": 6104
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 6104,
      "training_step_time": 9.89134693145752
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 3.1311466693878174,
      "step": 6104
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 6104,
      "training_step_time": 10.170454978942871
    },
    {
      "epoch": 2.3284912109375e-05,
      "model_forward_time": 3.138266086578369,
      "step": 6104
    },
    {
      "epoch": 2.3284912109375e-05,
      "step": 6104,
      "training_step_time": 10.290609121322632
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 3.1879539489746094,
      "step": 6105
    },
    {
      "dynamics_loss_avg": 0.059398791007697585,
      "epoch": 2.3288726806640627e-05,
      "step": 6105
    },
    {
      "action_loss_avg": 0.03005608106032014,
      "epoch": 2.3288726806640627e-05,
      "step": 6105
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 6105,
      "training_step_time": 9.96265172958374
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 3.1408450603485107,
      "step": 6105
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 6105,
      "training_step_time": 9.911412715911865
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 3.141223192214966,
      "step": 6105
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 6105,
      "training_step_time": 9.907170057296753
    },
    {
      "epoch": 2.3288726806640627e-05,
      "model_forward_time": 3.1368567943573,
      "step": 6105
    },
    {
      "epoch": 2.3288726806640627e-05,
      "step": 6105,
      "training_step_time": 10.257762432098389
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 3.163670778274536,
      "step": 6106
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 6106,
      "training_step_time": 9.940416097640991
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 3.1395211219787598,
      "step": 6106
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 6106,
      "training_step_time": 9.908597469329834
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 2.4347009658813477,
      "step": 6106
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 6106,
      "training_step_time": 7.456894159317017
    },
    {
      "epoch": 2.329254150390625e-05,
      "model_forward_time": 3.1330366134643555,
      "step": 6106
    },
    {
      "epoch": 2.329254150390625e-05,
      "step": 6106,
      "training_step_time": 10.273539304733276
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 3.4507806301116943,
      "step": 6107
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 6107,
      "training_step_time": 10.218718767166138
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 3.1369171142578125,
      "step": 6107
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 6107,
      "training_step_time": 9.898179054260254
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 3.1408891677856445,
      "step": 6107
    },
    {
      "dynamics_loss_avg": 0.05600219015032053,
      "epoch": 2.3296356201171876e-05,
      "step": 6107
    },
    {
      "action_loss_avg": 0.02517156843096018,
      "epoch": 2.3296356201171876e-05,
      "step": 6107
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 6107,
      "training_step_time": 9.908178091049194
    },
    {
      "epoch": 2.3296356201171876e-05,
      "model_forward_time": 3.1332502365112305,
      "step": 6107
    },
    {
      "epoch": 2.3296356201171876e-05,
      "step": 6107,
      "training_step_time": 10.26406979560852
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 3.153366804122925,
      "step": 6108
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 6108,
      "training_step_time": 9.918088436126709
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 3.405848503112793,
      "step": 6108
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 6108,
      "training_step_time": 10.170761108398438
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 3.1403489112854004,
      "step": 6108
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 6108,
      "training_step_time": 9.904853582382202
    },
    {
      "epoch": 2.33001708984375e-05,
      "model_forward_time": 3.140580654144287,
      "step": 6108
    },
    {
      "epoch": 2.33001708984375e-05,
      "step": 6108,
      "training_step_time": 10.304035425186157
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 3.155823230743408,
      "step": 6109
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 6109,
      "training_step_time": 9.926841497421265
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 3.1389126777648926,
      "step": 6109
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 6109,
      "training_step_time": 9.90446162223816
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 3.1376471519470215,
      "step": 6109
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 6109,
      "training_step_time": 10.185917854309082
    },
    {
      "epoch": 2.3303985595703125e-05,
      "model_forward_time": 3.1371500492095947,
      "step": 6109
    },
    {
      "epoch": 2.3303985595703125e-05,
      "step": 6109,
      "training_step_time": 10.290053367614746
    },
    {
      "epoch": 2.330780029296875e-05,
      "grad_norm": 0.46703484654426575,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 6110
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 3.149686574935913,
      "step": 6110
    },
    {
      "dynamics_loss_avg": 0.05725608784705401,
      "epoch": 2.330780029296875e-05,
      "step": 6110
    },
    {
      "action_loss_avg": 0.021598402224481106,
      "epoch": 2.330780029296875e-05,
      "step": 6110
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 6110,
      "training_step_time": 9.92444109916687
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 3.142448902130127,
      "step": 6110
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 6110,
      "training_step_time": 9.921842813491821
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 2.4393441677093506,
      "step": 6110
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 6110,
      "training_step_time": 7.469910144805908
    },
    {
      "epoch": 2.330780029296875e-05,
      "model_forward_time": 3.1410226821899414,
      "step": 6110
    },
    {
      "epoch": 2.330780029296875e-05,
      "step": 6110,
      "training_step_time": 10.015971660614014
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 3.4502477645874023,
      "step": 6111
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 6111,
      "training_step_time": 10.226062297821045
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 3.1484832763671875,
      "step": 6111
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 6111,
      "training_step_time": 9.92195463180542
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 3.14036226272583,
      "step": 6111
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 6111,
      "training_step_time": 9.909442663192749
    },
    {
      "epoch": 2.3311614990234374e-05,
      "model_forward_time": 3.1428751945495605,
      "step": 6111
    },
    {
      "epoch": 2.3311614990234374e-05,
      "step": 6111,
      "training_step_time": 10.243756532669067
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 3.17704439163208,
      "step": 6112
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 6112,
      "training_step_time": 10.241645097732544
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 3.143878221511841,
      "step": 6112
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 6112,
      "training_step_time": 9.915170907974243
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 3.140153169631958,
      "step": 6112
    },
    {
      "dynamics_loss_avg": 0.057309767603874205,
      "epoch": 2.33154296875e-05,
      "step": 6112
    },
    {
      "action_loss_avg": 0.02315277690067887,
      "epoch": 2.33154296875e-05,
      "step": 6112
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 6112,
      "training_step_time": 9.91084885597229
    },
    {
      "epoch": 2.33154296875e-05,
      "model_forward_time": 3.1415092945098877,
      "step": 6112
    },
    {
      "epoch": 2.33154296875e-05,
      "step": 6112,
      "training_step_time": 10.277614831924438
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 3.176407814025879,
      "step": 6113
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 6113,
      "training_step_time": 9.95209550857544
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 3.425717830657959,
      "step": 6113
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 6113,
      "training_step_time": 10.193573236465454
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 3.140903949737549,
      "step": 6113
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 6113,
      "training_step_time": 9.908875226974487
    },
    {
      "epoch": 2.3319244384765627e-05,
      "model_forward_time": 3.144420862197876,
      "step": 6113
    },
    {
      "epoch": 2.3319244384765627e-05,
      "step": 6113,
      "training_step_time": 10.308902025222778
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 3.1859869956970215,
      "step": 6114
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 6114,
      "training_step_time": 9.951912879943848
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 3.135650634765625,
      "step": 6114
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 6114,
      "training_step_time": 9.900646448135376
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 3.4146790504455566,
      "step": 6114
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 6114,
      "training_step_time": 10.179752826690674
    },
    {
      "epoch": 2.332305908203125e-05,
      "model_forward_time": 3.1464154720306396,
      "step": 6114
    },
    {
      "epoch": 2.332305908203125e-05,
      "step": 6114,
      "training_step_time": 10.274827241897583
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 3.179028034210205,
      "step": 6115
    },
    {
      "dynamics_loss_avg": 0.06443349160254001,
      "epoch": 2.3326873779296876e-05,
      "step": 6115
    },
    {
      "action_loss_avg": 0.03245258303359151,
      "epoch": 2.3326873779296876e-05,
      "step": 6115
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 6115,
      "training_step_time": 9.950496435165405
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 3.1358516216278076,
      "step": 6115
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 6115,
      "training_step_time": 9.901702165603638
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 3.1404576301574707,
      "step": 6115
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 6115,
      "training_step_time": 9.909881591796875
    },
    {
      "epoch": 2.3326873779296876e-05,
      "model_forward_time": 3.4108529090881348,
      "step": 6115
    },
    {
      "epoch": 2.3326873779296876e-05,
      "step": 6115,
      "training_step_time": 10.275746583938599
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 3.1744370460510254,
      "step": 6116
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 6116,
      "training_step_time": 9.941413640975952
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 3.13142728805542,
      "step": 6116
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 6116,
      "training_step_time": 9.897942066192627
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 3.137646198272705,
      "step": 6116
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 6116,
      "training_step_time": 9.90545392036438
    },
    {
      "epoch": 2.33306884765625e-05,
      "model_forward_time": 3.1340372562408447,
      "step": 6116
    },
    {
      "epoch": 2.33306884765625e-05,
      "step": 6116,
      "training_step_time": 10.001871347427368
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 3.4206206798553467,
      "step": 6117
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 6117,
      "training_step_time": 10.190493106842041
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 2.43331241607666,
      "step": 6117
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 6117,
      "training_step_time": 7.4643166065216064
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 3.1288204193115234,
      "step": 6117
    },
    {
      "dynamics_loss_avg": 0.04182139541953802,
      "epoch": 2.3334503173828125e-05,
      "step": 6117
    },
    {
      "action_loss_avg": 0.01787264822050929,
      "epoch": 2.3334503173828125e-05,
      "step": 6117
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 6117,
      "training_step_time": 9.907070398330688
    },
    {
      "epoch": 2.3334503173828125e-05,
      "model_forward_time": 3.134397268295288,
      "step": 6117
    },
    {
      "epoch": 2.3334503173828125e-05,
      "step": 6117,
      "training_step_time": 10.270780563354492
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 3.1415228843688965,
      "step": 6118
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 6118,
      "training_step_time": 9.917096614837646
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 3.38590145111084,
      "step": 6118
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 6118,
      "training_step_time": 10.154112339019775
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 2.430746555328369,
      "step": 6118
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 6118,
      "training_step_time": 7.453861474990845
    },
    {
      "epoch": 2.333831787109375e-05,
      "model_forward_time": 3.130028486251831,
      "step": 6118
    },
    {
      "epoch": 2.333831787109375e-05,
      "step": 6118,
      "training_step_time": 10.326438188552856
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 3.1739625930786133,
      "step": 6119
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 6119,
      "training_step_time": 9.938714504241943
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 3.1328041553497314,
      "step": 6119
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 6119,
      "training_step_time": 9.899865865707397
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 3.1320013999938965,
      "step": 6119
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 6119,
      "training_step_time": 9.905255556106567
    },
    {
      "epoch": 2.3342132568359374e-05,
      "model_forward_time": 3.3938217163085938,
      "step": 6119
    },
    {
      "epoch": 2.3342132568359374e-05,
      "step": 6119,
      "training_step_time": 10.321926593780518
    },
    {
      "epoch": 2.3345947265625e-05,
      "grad_norm": 0.4424014985561371,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 6120
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 2.4627766609191895,
      "step": 6120
    },
    {
      "dynamics_loss_avg": 0.06667614355683327,
      "epoch": 2.3345947265625e-05,
      "step": 6120
    },
    {
      "action_loss_avg": 0.023493759939447044,
      "epoch": 2.3345947265625e-05,
      "step": 6120
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 6120,
      "training_step_time": 7.493229150772095
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 3.1337037086486816,
      "step": 6120
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 6120,
      "training_step_time": 9.910639524459839
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 3.1342928409576416,
      "step": 6120
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 6120,
      "training_step_time": 9.908635139465332
    },
    {
      "epoch": 2.3345947265625e-05,
      "model_forward_time": 3.1341569423675537,
      "step": 6120
    },
    {
      "epoch": 2.3345947265625e-05,
      "step": 6120,
      "training_step_time": 10.2949697971344
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 3.172391414642334,
      "step": 6121
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 6121,
      "training_step_time": 9.948231220245361
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 3.137648344039917,
      "step": 6121
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 6121,
      "training_step_time": 9.912197351455688
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 3.139526605606079,
      "step": 6121
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 6121,
      "training_step_time": 9.91170597076416
    },
    {
      "epoch": 2.3349761962890627e-05,
      "model_forward_time": 3.1372241973876953,
      "step": 6121
    },
    {
      "epoch": 2.3349761962890627e-05,
      "step": 6121,
      "training_step_time": 10.020481586456299
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 3.4290993213653564,
      "step": 6122
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 6122,
      "training_step_time": 10.202791452407837
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 3.138225555419922,
      "step": 6122
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 6122,
      "training_step_time": 9.913886070251465
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 3.1357321739196777,
      "step": 6122
    },
    {
      "dynamics_loss_avg": 0.06319011617451906,
      "epoch": 2.335357666015625e-05,
      "step": 6122
    },
    {
      "action_loss_avg": 0.028485319670289756,
      "epoch": 2.335357666015625e-05,
      "step": 6122
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 6122,
      "training_step_time": 9.91229796409607
    },
    {
      "epoch": 2.335357666015625e-05,
      "model_forward_time": 3.1370863914489746,
      "step": 6122
    },
    {
      "epoch": 2.335357666015625e-05,
      "step": 6122,
      "training_step_time": 10.01346731185913
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 3.179149627685547,
      "step": 6123
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 6123,
      "training_step_time": 9.948363542556763
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 3.4240880012512207,
      "step": 6123
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 6123,
      "training_step_time": 10.193858623504639
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 3.1311612129211426,
      "step": 6123
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 6123,
      "training_step_time": 9.899746656417847
    },
    {
      "epoch": 2.3357391357421876e-05,
      "model_forward_time": 3.132143020629883,
      "step": 6123
    },
    {
      "epoch": 2.3357391357421876e-05,
      "step": 6123,
      "training_step_time": 10.253837823867798
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 3.1750075817108154,
      "step": 6124
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 6124,
      "training_step_time": 9.94329047203064
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 3.129899024963379,
      "step": 6124
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 6124,
      "training_step_time": 9.901941061019897
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 3.401401996612549,
      "step": 6124
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 6124,
      "training_step_time": 10.172499418258667
    },
    {
      "epoch": 2.33612060546875e-05,
      "model_forward_time": 3.136775016784668,
      "step": 6124
    },
    {
      "epoch": 2.33612060546875e-05,
      "step": 6124,
      "training_step_time": 10.3116455078125
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 3.1651620864868164,
      "step": 6125
    },
    {
      "dynamics_loss_avg": 0.06622030548751354,
      "epoch": 2.3365020751953125e-05,
      "step": 6125
    },
    {
      "action_loss_avg": 0.02360205799341202,
      "epoch": 2.3365020751953125e-05,
      "step": 6125
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 6125,
      "training_step_time": 9.945424318313599
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 3.1352953910827637,
      "step": 6125
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 6125,
      "training_step_time": 9.917734622955322
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 3.135063409805298,
      "step": 6125
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 6125,
      "training_step_time": 9.910597324371338
    },
    {
      "epoch": 2.3365020751953125e-05,
      "model_forward_time": 3.4299983978271484,
      "step": 6125
    },
    {
      "epoch": 2.3365020751953125e-05,
      "step": 6125,
      "training_step_time": 10.325645208358765
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 3.175079822540283,
      "step": 6126
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 6126,
      "training_step_time": 9.954696655273438
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 3.137112617492676,
      "step": 6126
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 6126,
      "training_step_time": 9.912864208221436
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 3.1376137733459473,
      "step": 6126
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 6126,
      "training_step_time": 9.913957118988037
    },
    {
      "epoch": 2.336883544921875e-05,
      "model_forward_time": 3.135551929473877,
      "step": 6126
    },
    {
      "epoch": 2.336883544921875e-05,
      "step": 6126,
      "training_step_time": 10.020150899887085
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 3.459456443786621,
      "step": 6127
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 6127,
      "training_step_time": 10.238840341567993
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 3.140136957168579,
      "step": 6127
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 6127,
      "training_step_time": 9.91267728805542
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 3.1391162872314453,
      "step": 6127
    },
    {
      "dynamics_loss_avg": 0.05569901876151562,
      "epoch": 2.3372650146484374e-05,
      "step": 6127
    },
    {
      "action_loss_avg": 0.024868285842239858,
      "epoch": 2.3372650146484374e-05,
      "step": 6127
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 6127,
      "training_step_time": 9.913854122161865
    },
    {
      "epoch": 2.3372650146484374e-05,
      "model_forward_time": 3.136831760406494,
      "step": 6127
    },
    {
      "epoch": 2.3372650146484374e-05,
      "step": 6127,
      "training_step_time": 10.014577150344849
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 3.1658143997192383,
      "step": 6128
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 6128,
      "training_step_time": 9.942423820495605
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 3.4451804161071777,
      "step": 6128
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 6128,
      "training_step_time": 10.215113162994385
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 3.139880895614624,
      "step": 6128
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 6128,
      "training_step_time": 9.916426181793213
    },
    {
      "epoch": 2.337646484375e-05,
      "model_forward_time": 3.1367359161376953,
      "step": 6128
    },
    {
      "epoch": 2.337646484375e-05,
      "step": 6128,
      "training_step_time": 10.277600526809692
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 3.1617228984832764,
      "step": 6129
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 6129,
      "training_step_time": 9.93625545501709
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 2.4374642372131348,
      "step": 6129
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 6129,
      "training_step_time": 7.469165325164795
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 3.140024423599243,
      "step": 6129
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 6129,
      "training_step_time": 10.203617572784424
    },
    {
      "epoch": 2.3380279541015623e-05,
      "model_forward_time": 2.439129114151001,
      "step": 6129
    },
    {
      "epoch": 2.3380279541015623e-05,
      "step": 6129,
      "training_step_time": 10.318767786026001
    },
    {
      "epoch": 2.338409423828125e-05,
      "grad_norm": 0.4172932505607605,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 6130
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 3.164785146713257,
      "step": 6130
    },
    {
      "dynamics_loss_avg": 0.04880228750407696,
      "epoch": 2.338409423828125e-05,
      "step": 6130
    },
    {
      "action_loss_avg": 0.023551377840340136,
      "epoch": 2.338409423828125e-05,
      "step": 6130
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 6130,
      "training_step_time": 9.94515609741211
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 3.1425065994262695,
      "step": 6130
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 6130,
      "training_step_time": 9.917166233062744
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 3.1389129161834717,
      "step": 6130
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 6130,
      "training_step_time": 9.913501262664795
    },
    {
      "epoch": 2.338409423828125e-05,
      "model_forward_time": 3.1393983364105225,
      "step": 6130
    },
    {
      "epoch": 2.338409423828125e-05,
      "step": 6130,
      "training_step_time": 10.301278352737427
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 3.4964663982391357,
      "step": 6131
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 6131,
      "training_step_time": 10.263715267181396
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 3.1412503719329834,
      "step": 6131
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 6131,
      "training_step_time": 9.909020900726318
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 3.1375625133514404,
      "step": 6131
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 6131,
      "training_step_time": 9.90670657157898
    },
    {
      "epoch": 2.3387908935546876e-05,
      "model_forward_time": 3.1436607837677,
      "step": 6131
    },
    {
      "epoch": 2.3387908935546876e-05,
      "step": 6131,
      "training_step_time": 10.019665002822876
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 3.1519479751586914,
      "step": 6132
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 6132,
      "training_step_time": 10.199459314346313
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 3.1395676136016846,
      "step": 6132
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 6132,
      "training_step_time": 9.908565998077393
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 3.1380817890167236,
      "step": 6132
    },
    {
      "dynamics_loss_avg": 0.08172606974840164,
      "epoch": 2.33917236328125e-05,
      "step": 6132
    },
    {
      "action_loss_avg": 0.030348738376051187,
      "epoch": 2.33917236328125e-05,
      "step": 6132
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 6132,
      "training_step_time": 9.90906810760498
    },
    {
      "epoch": 2.33917236328125e-05,
      "model_forward_time": 3.1465513706207275,
      "step": 6132
    },
    {
      "epoch": 2.33917236328125e-05,
      "step": 6132,
      "training_step_time": 10.014095783233643
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 3.1562387943267822,
      "step": 6133
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 6133,
      "training_step_time": 9.926796197891235
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 3.4380924701690674,
      "step": 6133
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 6133,
      "training_step_time": 10.205790281295776
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 3.1377882957458496,
      "step": 6133
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 6133,
      "training_step_time": 9.91017770767212
    },
    {
      "epoch": 2.3395538330078125e-05,
      "model_forward_time": 3.143098831176758,
      "step": 6133
    },
    {
      "epoch": 2.3395538330078125e-05,
      "step": 6133,
      "training_step_time": 10.250818014144897
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 3.1775410175323486,
      "step": 6134
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 6134,
      "training_step_time": 9.951680660247803
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 3.143810749053955,
      "step": 6134
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 6134,
      "training_step_time": 9.916388750076294
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 3.140563488006592,
      "step": 6134
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 6134,
      "training_step_time": 10.180320501327515
    },
    {
      "epoch": 2.339935302734375e-05,
      "model_forward_time": 3.139331817626953,
      "step": 6134
    },
    {
      "epoch": 2.339935302734375e-05,
      "step": 6134,
      "training_step_time": 10.311945915222168
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 3.152311325073242,
      "step": 6135
    },
    {
      "dynamics_loss_avg": 0.04782405737787485,
      "epoch": 2.3403167724609374e-05,
      "step": 6135
    },
    {
      "action_loss_avg": 0.01988504589535296,
      "epoch": 2.3403167724609374e-05,
      "step": 6135
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 6135,
      "training_step_time": 9.926830053329468
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 3.133232593536377,
      "step": 6135
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 6135,
      "training_step_time": 9.903351783752441
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 2.4338090419769287,
      "step": 6135
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 6135,
      "training_step_time": 7.4651198387146
    },
    {
      "epoch": 2.3403167724609374e-05,
      "model_forward_time": 3.4144163131713867,
      "step": 6135
    },
    {
      "epoch": 2.3403167724609374e-05,
      "step": 6135,
      "training_step_time": 10.294432878494263
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 3.150076150894165,
      "step": 6136
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 6136,
      "training_step_time": 9.923784494400024
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 3.134411096572876,
      "step": 6136
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 6136,
      "training_step_time": 9.908299207687378
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 3.139481782913208,
      "step": 6136
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 6136,
      "training_step_time": 9.917008638381958
    },
    {
      "epoch": 2.3406982421875e-05,
      "model_forward_time": 2.4359824657440186,
      "step": 6136
    },
    {
      "epoch": 2.3406982421875e-05,
      "step": 6136,
      "training_step_time": 10.28148889541626
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 3.173137664794922,
      "step": 6137
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 6137,
      "training_step_time": 9.950637578964233
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 3.381248712539673,
      "step": 6137
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 6137,
      "training_step_time": 10.15102744102478
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 3.136064291000366,
      "step": 6137
    },
    {
      "dynamics_loss_avg": 0.07007484473288059,
      "epoch": 2.3410797119140623e-05,
      "step": 6137
    },
    {
      "action_loss_avg": 0.025890772184357046,
      "epoch": 2.3410797119140623e-05,
      "step": 6137
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 6137,
      "training_step_time": 9.904730796813965
    },
    {
      "epoch": 2.3410797119140623e-05,
      "model_forward_time": 3.130419969558716,
      "step": 6137
    },
    {
      "epoch": 2.3410797119140623e-05,
      "step": 6137,
      "training_step_time": 10.021127700805664
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 3.138746976852417,
      "step": 6138
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 6138,
      "training_step_time": 9.905624389648438
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 3.3897736072540283,
      "step": 6138
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 6138,
      "training_step_time": 10.15625
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 3.136328935623169,
      "step": 6138
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 6138,
      "training_step_time": 9.907344818115234
    },
    {
      "epoch": 2.341461181640625e-05,
      "model_forward_time": 3.1329164505004883,
      "step": 6138
    },
    {
      "epoch": 2.341461181640625e-05,
      "step": 6138,
      "training_step_time": 10.010555505752563
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 3.191441535949707,
      "step": 6139
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 6139,
      "training_step_time": 9.963619470596313
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 3.14851713180542,
      "step": 6139
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 6139,
      "training_step_time": 9.920040369033813
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 3.1402182579040527,
      "step": 6139
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 6139,
      "training_step_time": 10.171306371688843
    },
    {
      "epoch": 2.3418426513671876e-05,
      "model_forward_time": 3.142822742462158,
      "step": 6139
    },
    {
      "epoch": 2.3418426513671876e-05,
      "step": 6139,
      "training_step_time": 10.278290510177612
    },
    {
      "epoch": 2.34222412109375e-05,
      "grad_norm": 0.6532696485519409,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 6140
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 3.1579887866973877,
      "step": 6140
    },
    {
      "dynamics_loss_avg": 0.061806130781769754,
      "epoch": 2.34222412109375e-05,
      "step": 6140
    },
    {
      "action_loss_avg": 0.028736860072240234,
      "epoch": 2.34222412109375e-05,
      "step": 6140
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 6140,
      "training_step_time": 9.93088698387146
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 3.1424076557159424,
      "step": 6140
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 6140,
      "training_step_time": 9.913300037384033
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 3.142409324645996,
      "step": 6140
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 6140,
      "training_step_time": 9.910786628723145
    },
    {
      "epoch": 2.34222412109375e-05,
      "model_forward_time": 3.406501054763794,
      "step": 6140
    },
    {
      "epoch": 2.34222412109375e-05,
      "step": 6140,
      "training_step_time": 10.307607650756836
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 3.1521787643432617,
      "step": 6141
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 6141,
      "training_step_time": 9.922958374023438
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 3.140964984893799,
      "step": 6141
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 6141,
      "training_step_time": 9.912915706634521
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 3.1410117149353027,
      "step": 6141
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 6141,
      "training_step_time": 9.912686347961426
    },
    {
      "epoch": 2.3426055908203125e-05,
      "model_forward_time": 3.143550157546997,
      "step": 6141
    },
    {
      "epoch": 2.3426055908203125e-05,
      "step": 6141,
      "training_step_time": 10.274071216583252
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 3.4212722778320312,
      "step": 6142
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 6142,
      "training_step_time": 10.192952394485474
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 3.1472859382629395,
      "step": 6142
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 6142,
      "training_step_time": 9.919600248336792
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 3.1427619457244873,
      "step": 6142
    },
    {
      "dynamics_loss_avg": 0.06559290774166585,
      "epoch": 2.342987060546875e-05,
      "step": 6142
    },
    {
      "action_loss_avg": 0.026617617486044763,
      "epoch": 2.342987060546875e-05,
      "step": 6142
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 6142,
      "training_step_time": 9.917037963867188
    },
    {
      "epoch": 2.342987060546875e-05,
      "model_forward_time": 3.1477437019348145,
      "step": 6142
    },
    {
      "epoch": 2.342987060546875e-05,
      "step": 6142,
      "training_step_time": 10.021993160247803
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 3.1882314682006836,
      "step": 6143
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 6143,
      "training_step_time": 9.961156606674194
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 3.4361069202423096,
      "step": 6143
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 6143,
      "training_step_time": 10.204368591308594
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 3.1425108909606934,
      "step": 6143
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 6143,
      "training_step_time": 9.910358667373657
    },
    {
      "epoch": 2.3433685302734374e-05,
      "model_forward_time": 3.1465811729431152,
      "step": 6143
    },
    {
      "epoch": 2.3433685302734374e-05,
      "step": 6143,
      "training_step_time": 10.009441614151001
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 3.17582368850708,
      "step": 6144
    },
    {
      "epoch": 2.34375e-05,
      "step": 6144,
      "training_step_time": 9.942972660064697
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 3.141122817993164,
      "step": 6144
    },
    {
      "epoch": 2.34375e-05,
      "step": 6144,
      "training_step_time": 9.909411907196045
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 3.4274744987487793,
      "step": 6144
    },
    {
      "epoch": 2.34375e-05,
      "step": 6144,
      "training_step_time": 10.194292306900024
    },
    {
      "epoch": 2.34375e-05,
      "model_forward_time": 3.1481595039367676,
      "step": 6144
    },
    {
      "epoch": 2.34375e-05,
      "step": 6144,
      "training_step_time": 10.308262348175049
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 3.15759015083313,
      "step": 6145
    },
    {
      "dynamics_loss_avg": 0.05108266919851303,
      "epoch": 2.3441314697265623e-05,
      "step": 6145
    },
    {
      "action_loss_avg": 0.022534614149481057,
      "epoch": 2.3441314697265623e-05,
      "step": 6145
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 6145,
      "training_step_time": 9.935468912124634
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 3.143494129180908,
      "step": 6145
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 6145,
      "training_step_time": 9.916170597076416
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 3.1520495414733887,
      "step": 6145
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 6145,
      "training_step_time": 9.926600694656372
    },
    {
      "epoch": 2.3441314697265623e-05,
      "model_forward_time": 2.437875986099243,
      "step": 6145
    },
    {
      "epoch": 2.3441314697265623e-05,
      "step": 6145,
      "training_step_time": 10.271385908126831
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 3.1555864810943604,
      "step": 6146
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 6146,
      "training_step_time": 9.93127179145813
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 3.147430658340454,
      "step": 6146
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 6146,
      "training_step_time": 9.919734001159668
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 3.147162914276123,
      "step": 6146
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 6146,
      "training_step_time": 9.916129112243652
    },
    {
      "epoch": 2.344512939453125e-05,
      "model_forward_time": 3.143848180770874,
      "step": 6146
    },
    {
      "epoch": 2.344512939453125e-05,
      "step": 6146,
      "training_step_time": 10.266839981079102
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 3.468715190887451,
      "step": 6147
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 6147,
      "training_step_time": 10.241020441055298
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 3.1457438468933105,
      "step": 6147
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 6147,
      "training_step_time": 9.921494007110596
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 3.147270917892456,
      "step": 6147
    },
    {
      "dynamics_loss_avg": 0.06960275918245315,
      "epoch": 2.3448944091796876e-05,
      "step": 6147
    },
    {
      "action_loss_avg": 0.02510800678282976,
      "epoch": 2.3448944091796876e-05,
      "step": 6147
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 6147,
      "training_step_time": 9.920589923858643
    },
    {
      "epoch": 2.3448944091796876e-05,
      "model_forward_time": 3.1397063732147217,
      "step": 6147
    },
    {
      "epoch": 2.3448944091796876e-05,
      "step": 6147,
      "training_step_time": 10.021556377410889
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 3.164263963699341,
      "step": 6148
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 6148,
      "training_step_time": 9.939151525497437
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 3.414320230484009,
      "step": 6148
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 6148,
      "training_step_time": 10.18602728843689
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 3.14725661277771,
      "step": 6148
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 6148,
      "training_step_time": 9.919358491897583
    },
    {
      "epoch": 2.34527587890625e-05,
      "model_forward_time": 3.1429121494293213,
      "step": 6148
    },
    {
      "epoch": 2.34527587890625e-05,
      "step": 6148,
      "training_step_time": 10.01827335357666
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 3.1534361839294434,
      "step": 6149
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 6149,
      "training_step_time": 9.926520347595215
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 3.146176815032959,
      "step": 6149
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 6149,
      "training_step_time": 9.916320085525513
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 2.438112735748291,
      "step": 6149
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 6149,
      "training_step_time": 7.744086265563965
    },
    {
      "epoch": 2.3456573486328125e-05,
      "model_forward_time": 3.144090414047241,
      "step": 6149
    },
    {
      "epoch": 2.3456573486328125e-05,
      "step": 6149,
      "training_step_time": 10.303224086761475
    },
    {
      "epoch": 2.346038818359375e-05,
      "grad_norm": 0.8029784560203552,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 6150
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 2.4548726081848145,
      "step": 6150
    },
    {
      "dynamics_loss_avg": 0.0693790815770626,
      "epoch": 2.346038818359375e-05,
      "step": 6150
    },
    {
      "action_loss_avg": 0.032524429447948935,
      "epoch": 2.346038818359375e-05,
      "step": 6150
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 6150,
      "training_step_time": 7.491594314575195
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 3.1429624557495117,
      "step": 6150
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 6150,
      "training_step_time": 9.918343544006348
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 3.1429386138916016,
      "step": 6150
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 6150,
      "training_step_time": 9.92050290107727
    },
    {
      "epoch": 2.346038818359375e-05,
      "model_forward_time": 3.1484384536743164,
      "step": 6150
    },
    {
      "epoch": 2.346038818359375e-05,
      "step": 6150,
      "training_step_time": 10.306591033935547
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 2.818155527114868,
      "step": 6151
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 6151,
      "training_step_time": 7.852729558944702
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 3.1438136100769043,
      "step": 6151
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 6151,
      "training_step_time": 9.924317359924316
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 3.146574020385742,
      "step": 6151
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 6151,
      "training_step_time": 9.93088698387146
    },
    {
      "epoch": 2.3464202880859374e-05,
      "model_forward_time": 3.1530873775482178,
      "step": 6151
    },
    {
      "epoch": 2.3464202880859374e-05,
      "step": 6151,
      "training_step_time": 10.306292533874512
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 3.1683461666107178,
      "step": 6152
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 6152,
      "training_step_time": 9.950216054916382
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 3.432676315307617,
      "step": 6152
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 6152,
      "training_step_time": 10.21680474281311
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 3.15080189704895,
      "step": 6152
    },
    {
      "dynamics_loss_avg": 0.07122951000928879,
      "epoch": 2.3468017578125e-05,
      "step": 6152
    },
    {
      "action_loss_avg": 0.024651250708848237,
      "epoch": 2.3468017578125e-05,
      "step": 6152
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 6152,
      "training_step_time": 9.943891763687134
    },
    {
      "epoch": 2.3468017578125e-05,
      "model_forward_time": 2.4405574798583984,
      "step": 6152
    },
    {
      "epoch": 2.3468017578125e-05,
      "step": 6152,
      "training_step_time": 10.02825379371643
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 3.1547279357910156,
      "step": 6153
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 6153,
      "training_step_time": 9.938189268112183
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 3.1435582637786865,
      "step": 6153
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 6153,
      "training_step_time": 10.246637344360352
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 3.146719217300415,
      "step": 6153
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 6153,
      "training_step_time": 9.932197093963623
    },
    {
      "epoch": 2.3471832275390623e-05,
      "model_forward_time": 3.1476757526397705,
      "step": 6153
    },
    {
      "epoch": 2.3471832275390623e-05,
      "step": 6153,
      "training_step_time": 10.02883529663086
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 3.184237003326416,
      "step": 6154
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 6154,
      "training_step_time": 9.971492528915405
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 3.1464335918426514,
      "step": 6154
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 6154,
      "training_step_time": 9.93425178527832
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 3.151125907897949,
      "step": 6154
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 6154,
      "training_step_time": 10.251726627349854
    },
    {
      "epoch": 2.347564697265625e-05,
      "model_forward_time": 3.15718674659729,
      "step": 6154
    },
    {
      "epoch": 2.347564697265625e-05,
      "step": 6154,
      "training_step_time": 10.323151588439941
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 1.765385389328003,
      "step": 6155
    },
    {
      "dynamics_loss_avg": 0.06162587832659483,
      "epoch": 2.3479461669921876e-05,
      "step": 6155
    },
    {
      "action_loss_avg": 0.02911572828888893,
      "epoch": 2.3479461669921876e-05,
      "step": 6155
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 6155,
      "training_step_time": 5.233540058135986
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 3.1411662101745605,
      "step": 6155
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 6155,
      "training_step_time": 9.925031423568726
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 3.148367404937744,
      "step": 6155
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 6155,
      "training_step_time": 9.928224563598633
    },
    {
      "epoch": 2.3479461669921876e-05,
      "model_forward_time": 3.1420845985412598,
      "step": 6155
    },
    {
      "epoch": 2.3479461669921876e-05,
      "step": 6155,
      "training_step_time": 10.022156715393066
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 2.801337480545044,
      "step": 6156
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 6156,
      "training_step_time": 7.837145805358887
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 3.138012409210205,
      "step": 6156
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 6156,
      "training_step_time": 9.91407060623169
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 3.148040771484375,
      "step": 6156
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 6156,
      "training_step_time": 9.933081865310669
    },
    {
      "epoch": 2.34832763671875e-05,
      "model_forward_time": 3.1517820358276367,
      "step": 6156
    },
    {
      "epoch": 2.34832763671875e-05,
      "step": 6156,
      "training_step_time": 10.326552629470825
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 3.164618730545044,
      "step": 6157
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 6157,
      "training_step_time": 9.942888975143433
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 3.3927743434906006,
      "step": 6157
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 6157,
      "training_step_time": 10.168787240982056
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 2.440840482711792,
      "step": 6157
    },
    {
      "dynamics_loss_avg": 0.05604485888034105,
      "epoch": 2.3487091064453125e-05,
      "step": 6157
    },
    {
      "action_loss_avg": 0.022838453669101,
      "epoch": 2.3487091064453125e-05,
      "step": 6157
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 6157,
      "training_step_time": 7.477716684341431
    },
    {
      "epoch": 2.3487091064453125e-05,
      "model_forward_time": 3.1437370777130127,
      "step": 6157
    },
    {
      "epoch": 2.3487091064453125e-05,
      "step": 6157,
      "training_step_time": 10.262784957885742
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 3.175870656967163,
      "step": 6158
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 6158,
      "training_step_time": 9.953766345977783
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 3.146052122116089,
      "step": 6158
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 6158,
      "training_step_time": 9.92323613166809
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 3.4560539722442627,
      "step": 6158
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 6158,
      "training_step_time": 10.22938346862793
    },
    {
      "epoch": 2.349090576171875e-05,
      "model_forward_time": 3.1403884887695312,
      "step": 6158
    },
    {
      "epoch": 2.349090576171875e-05,
      "step": 6158,
      "training_step_time": 10.265154123306274
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 3.1855733394622803,
      "step": 6159
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 6159,
      "training_step_time": 9.95999526977539
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 3.1448700428009033,
      "step": 6159
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 6159,
      "training_step_time": 9.919372320175171
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 3.4496161937713623,
      "step": 6159
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 6159,
      "training_step_time": 10.22896146774292
    },
    {
      "epoch": 2.3494720458984374e-05,
      "model_forward_time": 3.146190881729126,
      "step": 6159
    },
    {
      "epoch": 2.3494720458984374e-05,
      "step": 6159,
      "training_step_time": 10.289108991622925
    },
    {
      "epoch": 2.349853515625e-05,
      "grad_norm": 0.42543378472328186,
      "learning_rate": 1e-05,
      "loss": 0.0851,
      "step": 6160
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 3.159108877182007,
      "step": 6160
    },
    {
      "dynamics_loss_avg": 0.045948141068220136,
      "epoch": 2.349853515625e-05,
      "step": 6160
    },
    {
      "action_loss_avg": 0.013883064035326242,
      "epoch": 2.349853515625e-05,
      "step": 6160
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 6160,
      "training_step_time": 9.944650888442993
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 3.1461663246154785,
      "step": 6160
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 6160,
      "training_step_time": 9.92153000831604
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 3.1424126625061035,
      "step": 6160
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 6160,
      "training_step_time": 9.919457912445068
    },
    {
      "epoch": 2.349853515625e-05,
      "model_forward_time": 3.1368520259857178,
      "step": 6160
    },
    {
      "epoch": 2.349853515625e-05,
      "step": 6160,
      "training_step_time": 10.275240182876587
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 3.152590036392212,
      "step": 6161
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 6161,
      "training_step_time": 9.930031061172485
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 3.1418163776397705,
      "step": 6161
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 6161,
      "training_step_time": 9.915791511535645
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 2.437453508377075,
      "step": 6161
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 6161,
      "training_step_time": 7.467053413391113
    },
    {
      "epoch": 2.3502349853515623e-05,
      "model_forward_time": 2.4329512119293213,
      "step": 6161
    },
    {
      "epoch": 2.3502349853515623e-05,
      "step": 6161,
      "training_step_time": 10.318590641021729
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 2.4711246490478516,
      "step": 6162
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 6162,
      "training_step_time": 7.49604606628418
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 3.4291269779205322,
      "step": 6162
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 6162,
      "training_step_time": 10.204886436462402
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 3.1387479305267334,
      "step": 6162
    },
    {
      "dynamics_loss_avg": 0.0658332196995616,
      "epoch": 2.350616455078125e-05,
      "step": 6162
    },
    {
      "action_loss_avg": 0.034027309902012345,
      "epoch": 2.350616455078125e-05,
      "step": 6162
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 6162,
      "training_step_time": 9.913114547729492
    },
    {
      "epoch": 2.350616455078125e-05,
      "model_forward_time": 3.1463003158569336,
      "step": 6162
    },
    {
      "epoch": 2.350616455078125e-05,
      "step": 6162,
      "training_step_time": 10.334859609603882
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 3.1751105785369873,
      "step": 6163
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 6163,
      "training_step_time": 9.948505401611328
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 2.434316396713257,
      "step": 6163
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 6163,
      "training_step_time": 7.461807489395142
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 3.3832879066467285,
      "step": 6163
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 6163,
      "training_step_time": 10.148428440093994
    },
    {
      "epoch": 2.3509979248046876e-05,
      "model_forward_time": 3.14556884765625,
      "step": 6163
    },
    {
      "epoch": 2.3509979248046876e-05,
      "step": 6163,
      "training_step_time": 10.009652376174927
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 3.157308340072632,
      "step": 6164
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 6164,
      "training_step_time": 9.92716360092163
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 3.1350514888763428,
      "step": 6164
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 6164,
      "training_step_time": 9.903554439544678
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 2.432633399963379,
      "step": 6164
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 6164,
      "training_step_time": 7.455057859420776
    },
    {
      "epoch": 2.35137939453125e-05,
      "model_forward_time": 3.4371843338012695,
      "step": 6164
    },
    {
      "epoch": 2.35137939453125e-05,
      "step": 6164,
      "training_step_time": 10.300949573516846
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 3.1554436683654785,
      "step": 6165
    },
    {
      "dynamics_loss_avg": 0.057733663357794285,
      "epoch": 2.3517608642578125e-05,
      "step": 6165
    },
    {
      "action_loss_avg": 0.026266466360539198,
      "epoch": 2.3517608642578125e-05,
      "step": 6165
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 6165,
      "training_step_time": 9.927210569381714
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 3.1381514072418213,
      "step": 6165
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 6165,
      "training_step_time": 9.906349658966064
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 3.141979932785034,
      "step": 6165
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 6165,
      "training_step_time": 9.90980863571167
    },
    {
      "epoch": 2.3517608642578125e-05,
      "model_forward_time": 3.136477470397949,
      "step": 6165
    },
    {
      "epoch": 2.3517608642578125e-05,
      "step": 6165,
      "training_step_time": 10.279813051223755
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 3.501615285873413,
      "step": 6166
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 6166,
      "training_step_time": 10.269248485565186
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 2.4327383041381836,
      "step": 6166
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 6166,
      "training_step_time": 7.456318140029907
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 3.137505292892456,
      "step": 6166
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 6166,
      "training_step_time": 9.907221555709839
    },
    {
      "epoch": 2.352142333984375e-05,
      "model_forward_time": 3.1381449699401855,
      "step": 6166
    },
    {
      "epoch": 2.352142333984375e-05,
      "step": 6166,
      "training_step_time": 10.234750986099243
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 2.452359199523926,
      "step": 6167
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 6167,
      "training_step_time": 7.473705530166626
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 3.3805484771728516,
      "step": 6167
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 6167,
      "training_step_time": 10.144001722335815
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 3.1560134887695312,
      "step": 6167
    },
    {
      "dynamics_loss_avg": 0.059081704914569856,
      "epoch": 2.3525238037109374e-05,
      "step": 6167
    },
    {
      "action_loss_avg": 0.024052365869283675,
      "epoch": 2.3525238037109374e-05,
      "step": 6167
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 6167,
      "training_step_time": 9.921456813812256
    },
    {
      "epoch": 2.3525238037109374e-05,
      "model_forward_time": 2.4343960285186768,
      "step": 6167
    },
    {
      "epoch": 2.3525238037109374e-05,
      "step": 6167,
      "training_step_time": 10.280816316604614
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 3.1721930503845215,
      "step": 6168
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 6168,
      "training_step_time": 9.940142154693604
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 3.139575481414795,
      "step": 6168
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 6168,
      "training_step_time": 9.903008222579956
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 3.4044060707092285,
      "step": 6168
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 6168,
      "training_step_time": 10.166064262390137
    },
    {
      "epoch": 2.3529052734375e-05,
      "model_forward_time": 3.1331393718719482,
      "step": 6168
    },
    {
      "epoch": 2.3529052734375e-05,
      "step": 6168,
      "training_step_time": 10.004520654678345
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 3.1568803787231445,
      "step": 6169
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 6169,
      "training_step_time": 9.91875672340393
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 3.137298107147217,
      "step": 6169
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 6169,
      "training_step_time": 9.90061902999878
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 3.1331987380981445,
      "step": 6169
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 6169,
      "training_step_time": 10.160029888153076
    },
    {
      "epoch": 2.3532867431640623e-05,
      "model_forward_time": 3.1374764442443848,
      "step": 6169
    },
    {
      "epoch": 2.3532867431640623e-05,
      "step": 6169,
      "training_step_time": 10.008758306503296
    },
    {
      "epoch": 2.353668212890625e-05,
      "grad_norm": 0.4393240213394165,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 6170
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 2.4842066764831543,
      "step": 6170
    },
    {
      "dynamics_loss_avg": 0.06687017567455769,
      "epoch": 2.353668212890625e-05,
      "step": 6170
    },
    {
      "action_loss_avg": 0.024045493081212045,
      "epoch": 2.353668212890625e-05,
      "step": 6170
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 6170,
      "training_step_time": 7.508011341094971
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 3.1348812580108643,
      "step": 6170
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 6170,
      "training_step_time": 9.894080877304077
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 3.1318113803863525,
      "step": 6170
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 6170,
      "training_step_time": 9.893482446670532
    },
    {
      "epoch": 2.353668212890625e-05,
      "model_forward_time": 3.3987655639648438,
      "step": 6170
    },
    {
      "epoch": 2.353668212890625e-05,
      "step": 6170,
      "training_step_time": 10.2553129196167
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 3.185192346572876,
      "step": 6171
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 6171,
      "training_step_time": 9.949376583099365
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 3.1401491165161133,
      "step": 6171
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 6171,
      "training_step_time": 9.90948486328125
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 3.138002395629883,
      "step": 6171
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 6171,
      "training_step_time": 9.90821361541748
    },
    {
      "epoch": 2.3540496826171876e-05,
      "model_forward_time": 3.140590190887451,
      "step": 6171
    },
    {
      "epoch": 2.3540496826171876e-05,
      "step": 6171,
      "training_step_time": 10.020143747329712
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 3.145031690597534,
      "step": 6172
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 6172,
      "training_step_time": 10.193765878677368
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 3.1397688388824463,
      "step": 6172
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 6172,
      "training_step_time": 9.928140878677368
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 3.128743886947632,
      "step": 6172
    },
    {
      "dynamics_loss_avg": 0.060850700736045836,
      "epoch": 2.35443115234375e-05,
      "step": 6172
    },
    {
      "action_loss_avg": 0.025950932409614325,
      "epoch": 2.35443115234375e-05,
      "step": 6172
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 6172,
      "training_step_time": 9.90973424911499
    },
    {
      "epoch": 2.35443115234375e-05,
      "model_forward_time": 2.4298148155212402,
      "step": 6172
    },
    {
      "epoch": 2.35443115234375e-05,
      "step": 6172,
      "training_step_time": 10.326744079589844
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 2.4364962577819824,
      "step": 6173
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 6173,
      "training_step_time": 7.466130495071411
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 3.401942014694214,
      "step": 6173
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 6173,
      "training_step_time": 10.167879581451416
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 3.1309914588928223,
      "step": 6173
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 6173,
      "training_step_time": 9.901331663131714
    },
    {
      "epoch": 2.3548126220703125e-05,
      "model_forward_time": 3.1314268112182617,
      "step": 6173
    },
    {
      "epoch": 2.3548126220703125e-05,
      "step": 6173,
      "training_step_time": 10.266465902328491
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 3.173651695251465,
      "step": 6174
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 6174,
      "training_step_time": 9.944644927978516
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 3.1303913593292236,
      "step": 6174
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 6174,
      "training_step_time": 9.898033380508423
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 3.130573272705078,
      "step": 6174
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 6174,
      "training_step_time": 9.900145053863525
    },
    {
      "epoch": 2.355194091796875e-05,
      "model_forward_time": 3.3999764919281006,
      "step": 6174
    },
    {
      "epoch": 2.355194091796875e-05,
      "step": 6174,
      "training_step_time": 10.294170618057251
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 3.152458667755127,
      "step": 6175
    },
    {
      "dynamics_loss_avg": 0.07220243364572525,
      "epoch": 2.3555755615234374e-05,
      "step": 6175
    },
    {
      "action_loss_avg": 0.024954501166939737,
      "epoch": 2.3555755615234374e-05,
      "step": 6175
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 6175,
      "training_step_time": 9.927022695541382
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 3.1293983459472656,
      "step": 6175
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 6175,
      "training_step_time": 9.900291681289673
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 3.1335582733154297,
      "step": 6175
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 6175,
      "training_step_time": 9.90862488746643
    },
    {
      "epoch": 2.3555755615234374e-05,
      "model_forward_time": 3.4237313270568848,
      "step": 6175
    },
    {
      "epoch": 2.3555755615234374e-05,
      "step": 6175,
      "training_step_time": 10.293049335479736
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 3.16776967048645,
      "step": 6176
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 6176,
      "training_step_time": 9.94678783416748
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 3.1380677223205566,
      "step": 6176
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 6176,
      "training_step_time": 9.916821479797363
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 3.14028000831604,
      "step": 6176
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 6176,
      "training_step_time": 9.918651819229126
    },
    {
      "epoch": 2.35595703125e-05,
      "model_forward_time": 3.1379663944244385,
      "step": 6176
    },
    {
      "epoch": 2.35595703125e-05,
      "step": 6176,
      "training_step_time": 10.288491010665894
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 3.4382541179656982,
      "step": 6177
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 6177,
      "training_step_time": 10.219180822372437
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 3.1364855766296387,
      "step": 6177
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 6177,
      "training_step_time": 9.91594910621643
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 3.142608642578125,
      "step": 6177
    },
    {
      "dynamics_loss_avg": 0.05818093530833721,
      "epoch": 2.3563385009765623e-05,
      "step": 6177
    },
    {
      "action_loss_avg": 0.024317778833210468,
      "epoch": 2.3563385009765623e-05,
      "step": 6177
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 6177,
      "training_step_time": 9.92919921875
    },
    {
      "epoch": 2.3563385009765623e-05,
      "model_forward_time": 2.4354710578918457,
      "step": 6177
    },
    {
      "epoch": 2.3563385009765623e-05,
      "step": 6177,
      "training_step_time": 10.307932138442993
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 3.180924415588379,
      "step": 6178
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 6178,
      "training_step_time": 9.967982053756714
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 3.399529218673706,
      "step": 6178
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 6178,
      "training_step_time": 10.180268287658691
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 3.140005588531494,
      "step": 6178
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 6178,
      "training_step_time": 9.920803308486938
    },
    {
      "epoch": 2.356719970703125e-05,
      "model_forward_time": 3.1367242336273193,
      "step": 6178
    },
    {
      "epoch": 2.356719970703125e-05,
      "step": 6178,
      "training_step_time": 10.269718885421753
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 3.1736087799072266,
      "step": 6179
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 6179,
      "training_step_time": 9.95685887336731
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 3.137873411178589,
      "step": 6179
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 6179,
      "training_step_time": 9.917586326599121
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 3.4143238067626953,
      "step": 6179
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 6179,
      "training_step_time": 10.190644264221191
    },
    {
      "epoch": 2.3571014404296876e-05,
      "model_forward_time": 3.1333751678466797,
      "step": 6179
    },
    {
      "epoch": 2.3571014404296876e-05,
      "step": 6179,
      "training_step_time": 10.261987209320068
    },
    {
      "epoch": 2.35748291015625e-05,
      "grad_norm": 0.4525676667690277,
      "learning_rate": 1e-05,
      "loss": 0.0882,
      "step": 6180
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 3.173729181289673,
      "step": 6180
    },
    {
      "dynamics_loss_avg": 0.07813989836722612,
      "epoch": 2.35748291015625e-05,
      "step": 6180
    },
    {
      "action_loss_avg": 0.02932180950883776,
      "epoch": 2.35748291015625e-05,
      "step": 6180
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 6180,
      "training_step_time": 9.943700313568115
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 3.1390185356140137,
      "step": 6180
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 6180,
      "training_step_time": 9.907651901245117
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 3.1391148567199707,
      "step": 6180
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 6180,
      "training_step_time": 9.911304473876953
    },
    {
      "epoch": 2.35748291015625e-05,
      "model_forward_time": 3.398784637451172,
      "step": 6180
    },
    {
      "epoch": 2.35748291015625e-05,
      "step": 6180,
      "training_step_time": 10.266861915588379
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 2.4453814029693604,
      "step": 6181
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 6181,
      "training_step_time": 7.475511312484741
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 2.4282283782958984,
      "step": 6181
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 6181,
      "training_step_time": 7.453976392745972
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 3.130582571029663,
      "step": 6181
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 6181,
      "training_step_time": 9.89931845664978
    },
    {
      "epoch": 2.3578643798828125e-05,
      "model_forward_time": 3.1365177631378174,
      "step": 6181
    },
    {
      "epoch": 2.3578643798828125e-05,
      "step": 6181,
      "training_step_time": 10.291949987411499
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 3.418178081512451,
      "step": 6182
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 6182,
      "training_step_time": 10.18798828125
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 3.137118339538574,
      "step": 6182
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 6182,
      "training_step_time": 9.906034231185913
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 3.1286754608154297,
      "step": 6182
    },
    {
      "dynamics_loss_avg": 0.05934142768383026,
      "epoch": 2.358245849609375e-05,
      "step": 6182
    },
    {
      "action_loss_avg": 0.021880069095641375,
      "epoch": 2.358245849609375e-05,
      "step": 6182
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 6182,
      "training_step_time": 9.897765636444092
    },
    {
      "epoch": 2.358245849609375e-05,
      "model_forward_time": 3.137383222579956,
      "step": 6182
    },
    {
      "epoch": 2.358245849609375e-05,
      "step": 6182,
      "training_step_time": 10.313550233840942
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 3.1733343601226807,
      "step": 6183
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 6183,
      "training_step_time": 9.942598581314087
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 3.4121286869049072,
      "step": 6183
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 6183,
      "training_step_time": 10.181081056594849
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 3.132707118988037,
      "step": 6183
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 6183,
      "training_step_time": 9.9000883102417
    },
    {
      "epoch": 2.3586273193359374e-05,
      "model_forward_time": 3.132227659225464,
      "step": 6183
    },
    {
      "epoch": 2.3586273193359374e-05,
      "step": 6183,
      "training_step_time": 10.30494213104248
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 3.141331911087036,
      "step": 6184
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 6184,
      "training_step_time": 9.908276796340942
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 3.126185417175293,
      "step": 6184
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 6184,
      "training_step_time": 9.892814874649048
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 3.3890771865844727,
      "step": 6184
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 6184,
      "training_step_time": 10.151347398757935
    },
    {
      "epoch": 2.3590087890625e-05,
      "model_forward_time": 3.1376900672912598,
      "step": 6184
    },
    {
      "epoch": 2.3590087890625e-05,
      "step": 6184,
      "training_step_time": 10.289536476135254
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 3.143052577972412,
      "step": 6185
    },
    {
      "dynamics_loss_avg": 0.07206413503736257,
      "epoch": 2.3593902587890624e-05,
      "step": 6185
    },
    {
      "action_loss_avg": 0.026066087651997805,
      "epoch": 2.3593902587890624e-05,
      "step": 6185
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 6185,
      "training_step_time": 9.91598391532898
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 3.127067804336548,
      "step": 6185
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 6185,
      "training_step_time": 9.895667552947998
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 3.1349949836730957,
      "step": 6185
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 6185,
      "training_step_time": 9.905151128768921
    },
    {
      "epoch": 2.3593902587890624e-05,
      "model_forward_time": 3.4144320487976074,
      "step": 6185
    },
    {
      "epoch": 2.3593902587890624e-05,
      "step": 6185,
      "training_step_time": 10.273850679397583
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 3.1675620079040527,
      "step": 6186
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 6186,
      "training_step_time": 9.938103914260864
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 3.1294405460357666,
      "step": 6186
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 6186,
      "training_step_time": 9.902447938919067
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 3.1322906017303467,
      "step": 6186
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 6186,
      "training_step_time": 9.90573525428772
    },
    {
      "epoch": 2.359771728515625e-05,
      "model_forward_time": 3.1353347301483154,
      "step": 6186
    },
    {
      "epoch": 2.359771728515625e-05,
      "step": 6186,
      "training_step_time": 10.025228023529053
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 3.1475305557250977,
      "step": 6187
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 6187,
      "training_step_time": 10.208987951278687
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 3.13252854347229,
      "step": 6187
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 6187,
      "training_step_time": 9.906098127365112
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 3.1406009197235107,
      "step": 6187
    },
    {
      "dynamics_loss_avg": 0.06296126395463944,
      "epoch": 2.3601531982421876e-05,
      "step": 6187
    },
    {
      "action_loss_avg": 0.023784347996115684,
      "epoch": 2.3601531982421876e-05,
      "step": 6187
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 6187,
      "training_step_time": 9.920520305633545
    },
    {
      "epoch": 2.3601531982421876e-05,
      "model_forward_time": 2.432997941970825,
      "step": 6187
    },
    {
      "epoch": 2.3601531982421876e-05,
      "step": 6187,
      "training_step_time": 10.023600339889526
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 3.1508607864379883,
      "step": 6188
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 6188,
      "training_step_time": 9.921953439712524
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 3.137375831604004,
      "step": 6188
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 6188,
      "training_step_time": 9.910590410232544
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 3.4134411811828613,
      "step": 6188
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 6188,
      "training_step_time": 10.185954332351685
    },
    {
      "epoch": 2.36053466796875e-05,
      "model_forward_time": 3.135984182357788,
      "step": 6188
    },
    {
      "epoch": 2.36053466796875e-05,
      "step": 6188,
      "training_step_time": 10.284877061843872
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 3.145442247390747,
      "step": 6189
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 6189,
      "training_step_time": 9.921393632888794
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 3.133938789367676,
      "step": 6189
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 6189,
      "training_step_time": 9.90544080734253
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 3.1330204010009766,
      "step": 6189
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 6189,
      "training_step_time": 10.186644315719604
    },
    {
      "epoch": 2.3609161376953125e-05,
      "model_forward_time": 2.4311156272888184,
      "step": 6189
    },
    {
      "epoch": 2.3609161376953125e-05,
      "step": 6189,
      "training_step_time": 10.299781560897827
    },
    {
      "epoch": 2.361297607421875e-05,
      "grad_norm": 0.4750877618789673,
      "learning_rate": 1e-05,
      "loss": 0.088,
      "step": 6190
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 3.1778640747070312,
      "step": 6190
    },
    {
      "dynamics_loss_avg": 0.06218617493286729,
      "epoch": 2.361297607421875e-05,
      "step": 6190
    },
    {
      "action_loss_avg": 0.027639613021165133,
      "epoch": 2.361297607421875e-05,
      "step": 6190
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 6190,
      "training_step_time": 9.94771122932434
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 3.1349375247955322,
      "step": 6190
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 6190,
      "training_step_time": 9.902000188827515
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 3.129056692123413,
      "step": 6190
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 6190,
      "training_step_time": 9.899933815002441
    },
    {
      "epoch": 2.361297607421875e-05,
      "model_forward_time": 3.399792194366455,
      "step": 6190
    },
    {
      "epoch": 2.361297607421875e-05,
      "step": 6190,
      "training_step_time": 10.296568870544434
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 3.16723895072937,
      "step": 6191
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 6191,
      "training_step_time": 9.933746814727783
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 2.4273526668548584,
      "step": 6191
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 6191,
      "training_step_time": 7.448590517044067
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 3.122678518295288,
      "step": 6191
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 6191,
      "training_step_time": 9.884582757949829
    },
    {
      "epoch": 2.3616790771484374e-05,
      "model_forward_time": 3.131096124649048,
      "step": 6191
    },
    {
      "epoch": 2.3616790771484374e-05,
      "step": 6191,
      "training_step_time": 10.285654544830322
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 3.144695997238159,
      "step": 6192
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 6192,
      "training_step_time": 9.916228532791138
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 3.3997275829315186,
      "step": 6192
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 6192,
      "training_step_time": 10.169109106063843
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 2.425219774246216,
      "step": 6192
    },
    {
      "dynamics_loss_avg": 0.053557921666651964,
      "epoch": 2.362060546875e-05,
      "step": 6192
    },
    {
      "action_loss_avg": 0.027483388781547546,
      "epoch": 2.362060546875e-05,
      "step": 6192
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 6192,
      "training_step_time": 7.45328164100647
    },
    {
      "epoch": 2.362060546875e-05,
      "model_forward_time": 3.132329225540161,
      "step": 6192
    },
    {
      "epoch": 2.362060546875e-05,
      "step": 6192,
      "training_step_time": 10.016456604003906
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 3.181334972381592,
      "step": 6193
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 6193,
      "training_step_time": 9.952059030532837
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 3.406919002532959,
      "step": 6193
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 6193,
      "training_step_time": 10.174023151397705
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 3.131042242050171,
      "step": 6193
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 6193,
      "training_step_time": 9.899041414260864
    },
    {
      "epoch": 2.3624420166015624e-05,
      "model_forward_time": 3.131537914276123,
      "step": 6193
    },
    {
      "epoch": 2.3624420166015624e-05,
      "step": 6193,
      "training_step_time": 10.318039417266846
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 3.15293288230896,
      "step": 6194
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 6194,
      "training_step_time": 9.921643733978271
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 3.1293740272521973,
      "step": 6194
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 6194,
      "training_step_time": 9.894334077835083
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 3.132673501968384,
      "step": 6194
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 6194,
      "training_step_time": 9.903515815734863
    },
    {
      "epoch": 2.362823486328125e-05,
      "model_forward_time": 3.4052975177764893,
      "step": 6194
    },
    {
      "epoch": 2.362823486328125e-05,
      "step": 6194,
      "training_step_time": 10.298452854156494
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 3.143974781036377,
      "step": 6195
    },
    {
      "dynamics_loss_avg": 0.05281798131763935,
      "epoch": 2.3632049560546876e-05,
      "step": 6195
    },
    {
      "action_loss_avg": 0.021725956303998827,
      "epoch": 2.3632049560546876e-05,
      "step": 6195
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 6195,
      "training_step_time": 9.917859077453613
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 3.1303482055664062,
      "step": 6195
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 6195,
      "training_step_time": 9.895405769348145
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 3.134310483932495,
      "step": 6195
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 6195,
      "training_step_time": 9.90150785446167
    },
    {
      "epoch": 2.3632049560546876e-05,
      "model_forward_time": 2.7059595584869385,
      "step": 6195
    },
    {
      "epoch": 2.3632049560546876e-05,
      "step": 6195,
      "training_step_time": 10.287515640258789
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 3.152568817138672,
      "step": 6196
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 6196,
      "training_step_time": 9.920371294021606
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 3.1418070793151855,
      "step": 6196
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 6196,
      "training_step_time": 9.910266399383545
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 3.1478271484375,
      "step": 6196
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 6196,
      "training_step_time": 9.919987916946411
    },
    {
      "epoch": 2.36358642578125e-05,
      "model_forward_time": 2.4428656101226807,
      "step": 6196
    },
    {
      "epoch": 2.36358642578125e-05,
      "step": 6196,
      "training_step_time": 10.319240808486938
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 3.4357526302337646,
      "step": 6197
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 6197,
      "training_step_time": 10.215011596679688
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 3.1330175399780273,
      "step": 6197
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 6197,
      "training_step_time": 9.919054508209229
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 2.4354329109191895,
      "step": 6197
    },
    {
      "dynamics_loss_avg": 0.047017054399475455,
      "epoch": 2.3639678955078125e-05,
      "step": 6197
    },
    {
      "action_loss_avg": 0.015536717628128827,
      "epoch": 2.3639678955078125e-05,
      "step": 6197
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 6197,
      "training_step_time": 7.473287343978882
    },
    {
      "epoch": 2.3639678955078125e-05,
      "model_forward_time": 3.132830858230591,
      "step": 6197
    },
    {
      "epoch": 2.3639678955078125e-05,
      "step": 6197,
      "training_step_time": 10.023728132247925
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 3.144066572189331,
      "step": 6198
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 6198,
      "training_step_time": 9.92926287651062
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 3.1391072273254395,
      "step": 6198
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 6198,
      "training_step_time": 10.187281847000122
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 3.142000198364258,
      "step": 6198
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 6198,
      "training_step_time": 9.923149108886719
    },
    {
      "epoch": 2.364349365234375e-05,
      "model_forward_time": 3.137481927871704,
      "step": 6198
    },
    {
      "epoch": 2.364349365234375e-05,
      "step": 6198,
      "training_step_time": 10.293102502822876
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 3.151686668395996,
      "step": 6199
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 6199,
      "training_step_time": 9.93750810623169
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 3.1400468349456787,
      "step": 6199
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 6199,
      "training_step_time": 9.91949200630188
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 3.420694351196289,
      "step": 6199
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 6199,
      "training_step_time": 10.199813604354858
    },
    {
      "epoch": 2.3647308349609374e-05,
      "model_forward_time": 3.132153272628784,
      "step": 6199
    },
    {
      "epoch": 2.3647308349609374e-05,
      "step": 6199,
      "training_step_time": 10.28523588180542
    },
    {
      "epoch": 2.3651123046875e-05,
      "grad_norm": 0.47582414746284485,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 6200
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 3.160935878753662,
      "step": 6200
    },
    {
      "dynamics_loss_avg": 0.06011693775653839,
      "epoch": 2.3651123046875e-05,
      "step": 6200
    },
    {
      "action_loss_avg": 0.025194545090198518,
      "epoch": 2.3651123046875e-05,
      "step": 6200
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 6200,
      "training_step_time": 9.950597524642944
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 3.13942289352417,
      "step": 6200
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 6200,
      "training_step_time": 9.919637203216553
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 3.13559889793396,
      "step": 6200
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 6200,
      "training_step_time": 9.912009239196777
    },
    {
      "epoch": 2.3651123046875e-05,
      "model_forward_time": 3.3952553272247314,
      "step": 6200
    },
    {
      "epoch": 2.3651123046875e-05,
      "step": 6200,
      "training_step_time": 10.287472248077393
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 2.4731147289276123,
      "step": 6201
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 6201,
      "training_step_time": 7.508355140686035
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 3.1361773014068604,
      "step": 6201
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 6201,
      "training_step_time": 9.909878492355347
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 3.1327836513519287,
      "step": 6201
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 6201,
      "training_step_time": 9.907569408416748
    },
    {
      "epoch": 2.3654937744140624e-05,
      "model_forward_time": 3.1374642848968506,
      "step": 6201
    },
    {
      "epoch": 2.3654937744140624e-05,
      "step": 6201,
      "training_step_time": 10.31759786605835
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 3.4588985443115234,
      "step": 6202
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 6202,
      "training_step_time": 10.231807470321655
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 3.1355490684509277,
      "step": 6202
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 6202,
      "training_step_time": 9.906781196594238
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 3.1344752311706543,
      "step": 6202
    },
    {
      "dynamics_loss_avg": 0.07855163346976042,
      "epoch": 2.365875244140625e-05,
      "step": 6202
    },
    {
      "action_loss_avg": 0.03401831965893507,
      "epoch": 2.365875244140625e-05,
      "step": 6202
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 6202,
      "training_step_time": 9.911457777023315
    },
    {
      "epoch": 2.365875244140625e-05,
      "model_forward_time": 3.139216899871826,
      "step": 6202
    },
    {
      "epoch": 2.365875244140625e-05,
      "step": 6202,
      "training_step_time": 10.018020153045654
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 3.15423583984375,
      "step": 6203
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 6203,
      "training_step_time": 9.939931392669678
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 2.433690071105957,
      "step": 6203
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 6203,
      "training_step_time": 7.468318462371826
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 3.3876681327819824,
      "step": 6203
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 6203,
      "training_step_time": 10.16116738319397
    },
    {
      "epoch": 2.3662567138671876e-05,
      "model_forward_time": 3.1334943771362305,
      "step": 6203
    },
    {
      "epoch": 2.3662567138671876e-05,
      "step": 6203,
      "training_step_time": 10.022311210632324
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 3.1729073524475098,
      "step": 6204
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 6204,
      "training_step_time": 9.955814123153687
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 2.430913209915161,
      "step": 6204
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 6204,
      "training_step_time": 7.4667370319366455
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 3.1372971534729004,
      "step": 6204
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 6204,
      "training_step_time": 10.185233354568481
    },
    {
      "epoch": 2.36663818359375e-05,
      "model_forward_time": 3.136308193206787,
      "step": 6204
    },
    {
      "epoch": 2.36663818359375e-05,
      "step": 6204,
      "training_step_time": 10.3167884349823
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 3.150824785232544,
      "step": 6205
    },
    {
      "dynamics_loss_avg": 0.052225915156304835,
      "epoch": 2.3670196533203125e-05,
      "step": 6205
    },
    {
      "action_loss_avg": 0.02213765513151884,
      "epoch": 2.3670196533203125e-05,
      "step": 6205
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 6205,
      "training_step_time": 9.931610107421875
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 3.134167194366455,
      "step": 6205
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 6205,
      "training_step_time": 9.909112691879272
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 3.1387600898742676,
      "step": 6205
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 6205,
      "training_step_time": 9.918383598327637
    },
    {
      "epoch": 2.3670196533203125e-05,
      "model_forward_time": 3.136434555053711,
      "step": 6205
    },
    {
      "epoch": 2.3670196533203125e-05,
      "step": 6205,
      "training_step_time": 10.285963296890259
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 3.18330717086792,
      "step": 6206
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 6206,
      "training_step_time": 9.960386514663696
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 3.129775285720825,
      "step": 6206
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 6206,
      "training_step_time": 9.906203031539917
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 3.13791561126709,
      "step": 6206
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 6206,
      "training_step_time": 9.91058087348938
    },
    {
      "epoch": 2.367401123046875e-05,
      "model_forward_time": 3.134516477584839,
      "step": 6206
    },
    {
      "epoch": 2.367401123046875e-05,
      "step": 6206,
      "training_step_time": 10.294216632843018
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 3.454357624053955,
      "step": 6207
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 6207,
      "training_step_time": 10.230751037597656
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 3.1432676315307617,
      "step": 6207
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 6207,
      "training_step_time": 9.922868728637695
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 3.144733190536499,
      "step": 6207
    },
    {
      "dynamics_loss_avg": 0.055312860384583475,
      "epoch": 2.3677825927734374e-05,
      "step": 6207
    },
    {
      "action_loss_avg": 0.023905450478196144,
      "epoch": 2.3677825927734374e-05,
      "step": 6207
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 6207,
      "training_step_time": 9.926791667938232
    },
    {
      "epoch": 2.3677825927734374e-05,
      "model_forward_time": 3.1450254917144775,
      "step": 6207
    },
    {
      "epoch": 2.3677825927734374e-05,
      "step": 6207,
      "training_step_time": 10.031120300292969
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 3.1633477210998535,
      "step": 6208
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 6208,
      "training_step_time": 9.945150136947632
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 3.399467945098877,
      "step": 6208
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 6208,
      "training_step_time": 10.174256324768066
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 3.146740674972534,
      "step": 6208
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 6208,
      "training_step_time": 9.922900915145874
    },
    {
      "epoch": 2.3681640625e-05,
      "model_forward_time": 2.4388859272003174,
      "step": 6208
    },
    {
      "epoch": 2.3681640625e-05,
      "step": 6208,
      "training_step_time": 10.288222789764404
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 3.155759811401367,
      "step": 6209
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 6209,
      "training_step_time": 9.930120706558228
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 3.1413068771362305,
      "step": 6209
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 6209,
      "training_step_time": 9.911310195922852
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 3.1358892917633057,
      "step": 6209
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 6209,
      "training_step_time": 10.184021711349487
    },
    {
      "epoch": 2.3685455322265624e-05,
      "model_forward_time": 3.1379222869873047,
      "step": 6209
    },
    {
      "epoch": 2.3685455322265624e-05,
      "step": 6209,
      "training_step_time": 10.263502359390259
    },
    {
      "epoch": 2.368927001953125e-05,
      "grad_norm": 0.5011510848999023,
      "learning_rate": 1e-05,
      "loss": 0.0876,
      "step": 6210
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 3.1464030742645264,
      "step": 6210
    },
    {
      "dynamics_loss_avg": 0.06905349791049957,
      "epoch": 2.368927001953125e-05,
      "step": 6210
    },
    {
      "action_loss_avg": 0.03492411114275455,
      "epoch": 2.368927001953125e-05,
      "step": 6210
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 6210,
      "training_step_time": 9.91734790802002
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 3.138026237487793,
      "step": 6210
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 6210,
      "training_step_time": 9.906416893005371
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 3.1424434185028076,
      "step": 6210
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 6210,
      "training_step_time": 9.918558120727539
    },
    {
      "epoch": 2.368927001953125e-05,
      "model_forward_time": 3.4069364070892334,
      "step": 6210
    },
    {
      "epoch": 2.368927001953125e-05,
      "step": 6210,
      "training_step_time": 10.274861335754395
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 3.1644535064697266,
      "step": 6211
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 6211,
      "training_step_time": 9.936924695968628
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 3.1374058723449707,
      "step": 6211
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 6211,
      "training_step_time": 9.914429903030396
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 3.1331825256347656,
      "step": 6211
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 6211,
      "training_step_time": 9.906450271606445
    },
    {
      "epoch": 2.3693084716796876e-05,
      "model_forward_time": 3.141275405883789,
      "step": 6211
    },
    {
      "epoch": 2.3693084716796876e-05,
      "step": 6211,
      "training_step_time": 10.263193845748901
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 3.151158571243286,
      "step": 6212
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 6212,
      "training_step_time": 10.191692113876343
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 3.1353538036346436,
      "step": 6212
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 6212,
      "training_step_time": 9.909953117370605
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 3.1371850967407227,
      "step": 6212
    },
    {
      "dynamics_loss_avg": 0.05592278260737658,
      "epoch": 2.36968994140625e-05,
      "step": 6212
    },
    {
      "action_loss_avg": 0.025839715264737606,
      "epoch": 2.36968994140625e-05,
      "step": 6212
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 6212,
      "training_step_time": 9.912200212478638
    },
    {
      "epoch": 2.36968994140625e-05,
      "model_forward_time": 2.4336979389190674,
      "step": 6212
    },
    {
      "epoch": 2.36968994140625e-05,
      "step": 6212,
      "training_step_time": 10.280116319656372
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 3.1806554794311523,
      "step": 6213
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 6213,
      "training_step_time": 9.958077430725098
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 2.71441650390625,
      "step": 6213
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 6213,
      "training_step_time": 7.740512132644653
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 3.128525733947754,
      "step": 6213
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 6213,
      "training_step_time": 9.898422241210938
    },
    {
      "epoch": 2.3700714111328125e-05,
      "model_forward_time": 3.140038013458252,
      "step": 6213
    },
    {
      "epoch": 2.3700714111328125e-05,
      "step": 6213,
      "training_step_time": 10.032092809677124
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 3.181894063949585,
      "step": 6214
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 6214,
      "training_step_time": 9.961220979690552
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 3.1365561485290527,
      "step": 6214
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 6214,
      "training_step_time": 9.913785696029663
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 3.141401529312134,
      "step": 6214
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 6214,
      "training_step_time": 10.186181545257568
    },
    {
      "epoch": 2.370452880859375e-05,
      "model_forward_time": 3.1394894123077393,
      "step": 6214
    },
    {
      "epoch": 2.370452880859375e-05,
      "step": 6214,
      "training_step_time": 10.102319717407227
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 3.1537926197052,
      "step": 6215
    },
    {
      "dynamics_loss_avg": 0.07364453878253699,
      "epoch": 2.3708343505859374e-05,
      "step": 6215
    },
    {
      "action_loss_avg": 0.028544284822419285,
      "epoch": 2.3708343505859374e-05,
      "step": 6215
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 6215,
      "training_step_time": 9.926656484603882
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 3.1332473754882812,
      "step": 6215
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 6215,
      "training_step_time": 9.904783010482788
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 3.132443904876709,
      "step": 6215
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 6215,
      "training_step_time": 9.901741743087769
    },
    {
      "epoch": 2.3708343505859374e-05,
      "model_forward_time": 3.4394257068634033,
      "step": 6215
    },
    {
      "epoch": 2.3708343505859374e-05,
      "step": 6215,
      "training_step_time": 10.324119091033936
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 3.1490228176116943,
      "step": 6216
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 6216,
      "training_step_time": 9.917136430740356
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 1.7426865100860596,
      "step": 6216
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 6216,
      "training_step_time": 5.193300485610962
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 3.130957841873169,
      "step": 6216
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 6216,
      "training_step_time": 9.895182847976685
    },
    {
      "epoch": 2.3712158203125e-05,
      "model_forward_time": 2.4308881759643555,
      "step": 6216
    },
    {
      "epoch": 2.3712158203125e-05,
      "step": 6216,
      "training_step_time": 10.29440426826477
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 3.1688060760498047,
      "step": 6217
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 6217,
      "training_step_time": 9.936473846435547
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 3.374476671218872,
      "step": 6217
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 6217,
      "training_step_time": 10.143051385879517
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 3.1341657638549805,
      "step": 6217
    },
    {
      "dynamics_loss_avg": 0.056361717917025086,
      "epoch": 2.3715972900390624e-05,
      "step": 6217
    },
    {
      "action_loss_avg": 0.023800395661965013,
      "epoch": 2.3715972900390624e-05,
      "step": 6217
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 6217,
      "training_step_time": 9.903613328933716
    },
    {
      "epoch": 2.3715972900390624e-05,
      "model_forward_time": 3.1360011100769043,
      "step": 6217
    },
    {
      "epoch": 2.3715972900390624e-05,
      "step": 6217,
      "training_step_time": 10.275840520858765
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 3.1504313945770264,
      "step": 6218
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 6218,
      "training_step_time": 9.92067003250122
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 3.1400346755981445,
      "step": 6218
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 6218,
      "training_step_time": 9.903098583221436
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 3.427424907684326,
      "step": 6218
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 6218,
      "training_step_time": 10.195419311523438
    },
    {
      "epoch": 2.371978759765625e-05,
      "model_forward_time": 3.1357104778289795,
      "step": 6218
    },
    {
      "epoch": 2.371978759765625e-05,
      "step": 6218,
      "training_step_time": 9.998940706253052
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 3.1706879138946533,
      "step": 6219
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 6219,
      "training_step_time": 9.938978672027588
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 3.1389341354370117,
      "step": 6219
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 6219,
      "training_step_time": 9.90987229347229
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 3.4199395179748535,
      "step": 6219
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 6219,
      "training_step_time": 10.189470767974854
    },
    {
      "epoch": 2.3723602294921876e-05,
      "model_forward_time": 3.1422007083892822,
      "step": 6219
    },
    {
      "epoch": 2.3723602294921876e-05,
      "step": 6219,
      "training_step_time": 10.020936965942383
    },
    {
      "epoch": 2.37274169921875e-05,
      "grad_norm": 0.4650699496269226,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 6220
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 3.185234308242798,
      "step": 6220
    },
    {
      "dynamics_loss_avg": 0.057876495830714705,
      "epoch": 2.37274169921875e-05,
      "step": 6220
    },
    {
      "action_loss_avg": 0.021796574257314204,
      "epoch": 2.37274169921875e-05,
      "step": 6220
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 6220,
      "training_step_time": 9.96501088142395
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 3.1460962295532227,
      "step": 6220
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 6220,
      "training_step_time": 9.919491052627563
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 3.143946647644043,
      "step": 6220
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 6220,
      "training_step_time": 9.92024850845337
    },
    {
      "epoch": 2.37274169921875e-05,
      "model_forward_time": 3.1442229747772217,
      "step": 6220
    },
    {
      "epoch": 2.37274169921875e-05,
      "step": 6220,
      "training_step_time": 10.273136138916016
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 2.7967278957366943,
      "step": 6221
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 6221,
      "training_step_time": 7.824706315994263
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 3.1477205753326416,
      "step": 6221
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 6221,
      "training_step_time": 9.922865629196167
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 3.1422412395477295,
      "step": 6221
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 6221,
      "training_step_time": 9.91553783416748
    },
    {
      "epoch": 2.3731231689453125e-05,
      "model_forward_time": 3.1405887603759766,
      "step": 6221
    },
    {
      "epoch": 2.3731231689453125e-05,
      "step": 6221,
      "training_step_time": 10.287164211273193
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 3.4477646350860596,
      "step": 6222
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 6222,
      "training_step_time": 10.225656747817993
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 3.143751621246338,
      "step": 6222
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 6222,
      "training_step_time": 9.919525623321533
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 3.142805576324463,
      "step": 6222
    },
    {
      "dynamics_loss_avg": 0.07225011382251978,
      "epoch": 2.373504638671875e-05,
      "step": 6222
    },
    {
      "action_loss_avg": 0.02574396417476237,
      "epoch": 2.373504638671875e-05,
      "step": 6222
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 6222,
      "training_step_time": 9.921216487884521
    },
    {
      "epoch": 2.373504638671875e-05,
      "model_forward_time": 2.4372470378875732,
      "step": 6222
    },
    {
      "epoch": 2.373504638671875e-05,
      "step": 6222,
      "training_step_time": 10.29804253578186
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 3.195491075515747,
      "step": 6223
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 6223,
      "training_step_time": 9.980438709259033
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 3.1436803340911865,
      "step": 6223
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 6223,
      "training_step_time": 9.925406455993652
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 3.393265724182129,
      "step": 6223
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 6223,
      "training_step_time": 10.164968490600586
    },
    {
      "epoch": 2.3738861083984374e-05,
      "model_forward_time": 3.1467337608337402,
      "step": 6223
    },
    {
      "epoch": 2.3738861083984374e-05,
      "step": 6223,
      "training_step_time": 10.025479078292847
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 3.1658215522766113,
      "step": 6224
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 6224,
      "training_step_time": 9.942676782608032
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 3.142332077026367,
      "step": 6224
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 6224,
      "training_step_time": 9.916908740997314
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 2.438544511795044,
      "step": 6224
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 6224,
      "training_step_time": 7.465123176574707
    },
    {
      "epoch": 2.374267578125e-05,
      "model_forward_time": 3.4114737510681152,
      "step": 6224
    },
    {
      "epoch": 2.374267578125e-05,
      "step": 6224,
      "training_step_time": 10.27788782119751
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 3.157249927520752,
      "step": 6225
    },
    {
      "dynamics_loss_avg": 0.057600609958171844,
      "epoch": 2.3746490478515624e-05,
      "step": 6225
    },
    {
      "action_loss_avg": 0.031008313596248626,
      "epoch": 2.3746490478515624e-05,
      "step": 6225
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 6225,
      "training_step_time": 9.933879375457764
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 3.1377828121185303,
      "step": 6225
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 6225,
      "training_step_time": 9.911265850067139
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 2.436509847640991,
      "step": 6225
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 6225,
      "training_step_time": 7.465002775192261
    },
    {
      "epoch": 2.3746490478515624e-05,
      "model_forward_time": 2.4375815391540527,
      "step": 6225
    },
    {
      "epoch": 2.3746490478515624e-05,
      "step": 6225,
      "training_step_time": 10.293317079544067
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 3.4611363410949707,
      "step": 6226
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 6226,
      "training_step_time": 10.228526592254639
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 3.1394705772399902,
      "step": 6226
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 6226,
      "training_step_time": 9.912110090255737
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 3.1382453441619873,
      "step": 6226
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 6226,
      "training_step_time": 9.906444072723389
    },
    {
      "epoch": 2.375030517578125e-05,
      "model_forward_time": 3.1354708671569824,
      "step": 6226
    },
    {
      "epoch": 2.375030517578125e-05,
      "step": 6226,
      "training_step_time": 10.312649011611938
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 3.145667552947998,
      "step": 6227
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 6227,
      "training_step_time": 9.91697359085083
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 3.382456064224243,
      "step": 6227
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 6227,
      "training_step_time": 10.15287971496582
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 1.748063325881958,
      "step": 6227
    },
    {
      "dynamics_loss_avg": 0.047088841814547776,
      "epoch": 2.3754119873046876e-05,
      "step": 6227
    },
    {
      "action_loss_avg": 0.019896577089093627,
      "epoch": 2.3754119873046876e-05,
      "step": 6227
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 6227,
      "training_step_time": 5.21080470085144
    },
    {
      "epoch": 2.3754119873046876e-05,
      "model_forward_time": 3.1316113471984863,
      "step": 6227
    },
    {
      "epoch": 2.3754119873046876e-05,
      "step": 6227,
      "training_step_time": 10.281558752059937
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 3.1615078449249268,
      "step": 6228
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 6228,
      "training_step_time": 9.940644025802612
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 3.143611192703247,
      "step": 6228
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 6228,
      "training_step_time": 9.915646314620972
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 3.401254892349243,
      "step": 6228
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 6228,
      "training_step_time": 10.172834157943726
    },
    {
      "epoch": 2.37579345703125e-05,
      "model_forward_time": 3.1382439136505127,
      "step": 6228
    },
    {
      "epoch": 2.37579345703125e-05,
      "step": 6228,
      "training_step_time": 10.27209758758545
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 3.153207302093506,
      "step": 6229
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 6229,
      "training_step_time": 9.926101446151733
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 2.4301931858062744,
      "step": 6229
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 6229,
      "training_step_time": 7.455702066421509
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 3.1270501613616943,
      "step": 6229
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 6229,
      "training_step_time": 9.895192623138428
    },
    {
      "epoch": 2.3761749267578125e-05,
      "model_forward_time": 3.3805487155914307,
      "step": 6229
    },
    {
      "epoch": 2.3761749267578125e-05,
      "step": 6229,
      "training_step_time": 10.242704391479492
    },
    {
      "epoch": 2.376556396484375e-05,
      "grad_norm": 0.40718865394592285,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 6230
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 3.161271572113037,
      "step": 6230
    },
    {
      "dynamics_loss_avg": 0.05175887383520603,
      "epoch": 2.376556396484375e-05,
      "step": 6230
    },
    {
      "action_loss_avg": 0.01944605388562195,
      "epoch": 2.376556396484375e-05,
      "step": 6230
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 6230,
      "training_step_time": 9.929322719573975
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 3.134232997894287,
      "step": 6230
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 6230,
      "training_step_time": 9.902698278427124
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 3.132216453552246,
      "step": 6230
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 6230,
      "training_step_time": 9.899045944213867
    },
    {
      "epoch": 2.376556396484375e-05,
      "model_forward_time": 3.1349709033966064,
      "step": 6230
    },
    {
      "epoch": 2.376556396484375e-05,
      "step": 6230,
      "training_step_time": 10.317911148071289
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 3.145759105682373,
      "step": 6231
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 6231,
      "training_step_time": 9.922092199325562
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 3.1330816745758057,
      "step": 6231
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 6231,
      "training_step_time": 9.906981945037842
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 2.435025691986084,
      "step": 6231
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 6231,
      "training_step_time": 7.467263460159302
    },
    {
      "epoch": 2.3769378662109375e-05,
      "model_forward_time": 2.426816940307617,
      "step": 6231
    },
    {
      "epoch": 2.3769378662109375e-05,
      "step": 6231,
      "training_step_time": 10.317646026611328
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 3.1691994667053223,
      "step": 6232
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 6232,
      "training_step_time": 9.93674349784851
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 2.7164602279663086,
      "step": 6232
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 6232,
      "training_step_time": 7.738128900527954
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 3.1289584636688232,
      "step": 6232
    },
    {
      "dynamics_loss_avg": 0.04971222598105669,
      "epoch": 2.3773193359375e-05,
      "step": 6232
    },
    {
      "action_loss_avg": 0.024614083487540482,
      "epoch": 2.3773193359375e-05,
      "step": 6232
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 6232,
      "training_step_time": 9.905897378921509
    },
    {
      "epoch": 2.3773193359375e-05,
      "model_forward_time": 3.131627082824707,
      "step": 6232
    },
    {
      "epoch": 2.3773193359375e-05,
      "step": 6232,
      "training_step_time": 10.293076992034912
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 3.150325298309326,
      "step": 6233
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 6233,
      "training_step_time": 9.918108940124512
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 3.12619948387146,
      "step": 6233
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 6233,
      "training_step_time": 9.88990330696106
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 3.38297438621521,
      "step": 6233
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 6233,
      "training_step_time": 10.144756317138672
    },
    {
      "epoch": 2.3777008056640624e-05,
      "model_forward_time": 2.4384493827819824,
      "step": 6233
    },
    {
      "epoch": 2.3777008056640624e-05,
      "step": 6233,
      "training_step_time": 10.222254514694214
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 3.157383441925049,
      "step": 6234
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 6234,
      "training_step_time": 9.916184425354004
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 2.429208755493164,
      "step": 6234
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 6234,
      "training_step_time": 7.44739294052124
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 3.128551959991455,
      "step": 6234
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 6234,
      "training_step_time": 9.88933777809143
    },
    {
      "epoch": 2.378082275390625e-05,
      "model_forward_time": 3.39595103263855,
      "step": 6234
    },
    {
      "epoch": 2.378082275390625e-05,
      "step": 6234,
      "training_step_time": 10.25896954536438
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 3.138695478439331,
      "step": 6235
    },
    {
      "dynamics_loss_avg": 0.04901004228740931,
      "epoch": 2.3784637451171876e-05,
      "step": 6235
    },
    {
      "action_loss_avg": 0.02228312212973833,
      "epoch": 2.3784637451171876e-05,
      "step": 6235
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 6235,
      "training_step_time": 9.902518272399902
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 3.131603240966797,
      "step": 6235
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 6235,
      "training_step_time": 9.894461631774902
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 2.436366081237793,
      "step": 6235
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 6235,
      "training_step_time": 7.456502437591553
    },
    {
      "epoch": 2.3784637451171876e-05,
      "model_forward_time": 3.1344456672668457,
      "step": 6235
    },
    {
      "epoch": 2.3784637451171876e-05,
      "step": 6235,
      "training_step_time": 10.005090713500977
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 3.4378600120544434,
      "step": 6236
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 6236,
      "training_step_time": 10.197792530059814
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 3.13031005859375,
      "step": 6236
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 6236,
      "training_step_time": 9.898284912109375
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 3.1371493339538574,
      "step": 6236
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 6236,
      "training_step_time": 9.912903308868408
    },
    {
      "epoch": 2.37884521484375e-05,
      "model_forward_time": 3.1343023777008057,
      "step": 6236
    },
    {
      "epoch": 2.37884521484375e-05,
      "step": 6236,
      "training_step_time": 10.337074041366577
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 3.176166296005249,
      "step": 6237
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 6237,
      "training_step_time": 9.94971776008606
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 3.380673408508301,
      "step": 6237
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 6237,
      "training_step_time": 10.155187129974365
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 3.1407296657562256,
      "step": 6237
    },
    {
      "dynamics_loss_avg": 0.05048489663749933,
      "epoch": 2.3792266845703125e-05,
      "step": 6237
    },
    {
      "action_loss_avg": 0.023530772794038056,
      "epoch": 2.3792266845703125e-05,
      "step": 6237
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 6237,
      "training_step_time": 9.917320013046265
    },
    {
      "epoch": 2.3792266845703125e-05,
      "model_forward_time": 3.1400129795074463,
      "step": 6237
    },
    {
      "epoch": 2.3792266845703125e-05,
      "step": 6237,
      "training_step_time": 10.320859432220459
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 2.4726953506469727,
      "step": 6238
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 6238,
      "training_step_time": 7.504093408584595
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 2.4317567348480225,
      "step": 6238
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 6238,
      "training_step_time": 7.458383321762085
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 3.4016597270965576,
      "step": 6238
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 6238,
      "training_step_time": 10.168697595596313
    },
    {
      "epoch": 2.379608154296875e-05,
      "model_forward_time": 3.1300101280212402,
      "step": 6238
    },
    {
      "epoch": 2.379608154296875e-05,
      "step": 6238,
      "training_step_time": 10.26655101776123
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 2.445565938949585,
      "step": 6239
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 6239,
      "training_step_time": 7.470137119293213
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 3.131772756576538,
      "step": 6239
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 6239,
      "training_step_time": 9.901803970336914
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 3.1299397945404053,
      "step": 6239
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 6239,
      "training_step_time": 9.897575378417969
    },
    {
      "epoch": 2.3799896240234375e-05,
      "model_forward_time": 2.002753257751465,
      "step": 6239
    },
    {
      "epoch": 2.3799896240234375e-05,
      "step": 6239,
      "training_step_time": 10.28149962425232
    },
    {
      "epoch": 2.38037109375e-05,
      "grad_norm": 0.516812264919281,
      "learning_rate": 1e-05,
      "loss": 0.085,
      "step": 6240
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 3.1468141078948975,
      "step": 6240
    },
    {
      "dynamics_loss_avg": 0.058039613626897336,
      "epoch": 2.38037109375e-05,
      "step": 6240
    },
    {
      "action_loss_avg": 0.023993863770738245,
      "epoch": 2.38037109375e-05,
      "step": 6240
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 6240,
      "training_step_time": 9.911532640457153
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 3.134230613708496,
      "step": 6240
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 6240,
      "training_step_time": 9.901299238204956
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 2.4293651580810547,
      "step": 6240
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 6240,
      "training_step_time": 7.452996492385864
    },
    {
      "epoch": 2.38037109375e-05,
      "model_forward_time": 3.1328980922698975,
      "step": 6240
    },
    {
      "epoch": 2.38037109375e-05,
      "step": 6240,
      "training_step_time": 10.015327453613281
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 3.4156787395477295,
      "step": 6241
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 6241,
      "training_step_time": 10.183793067932129
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 3.1312122344970703,
      "step": 6241
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 6241,
      "training_step_time": 9.895590782165527
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 3.1314806938171387,
      "step": 6241
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 6241,
      "training_step_time": 9.898308992385864
    },
    {
      "epoch": 2.3807525634765624e-05,
      "model_forward_time": 3.1317636966705322,
      "step": 6241
    },
    {
      "epoch": 2.3807525634765624e-05,
      "step": 6241,
      "training_step_time": 10.306092023849487
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 3.155602216720581,
      "step": 6242
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 6242,
      "training_step_time": 9.937008619308472
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 3.4766385555267334,
      "step": 6242
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 6242,
      "training_step_time": 10.256463766098022
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 3.142061233520508,
      "step": 6242
    },
    {
      "dynamics_loss_avg": 0.06252746898680925,
      "epoch": 2.381134033203125e-05,
      "step": 6242
    },
    {
      "action_loss_avg": 0.030839358549565078,
      "epoch": 2.381134033203125e-05,
      "step": 6242
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 6242,
      "training_step_time": 9.918872356414795
    },
    {
      "epoch": 2.381134033203125e-05,
      "model_forward_time": 3.14392352104187,
      "step": 6242
    },
    {
      "epoch": 2.381134033203125e-05,
      "step": 6242,
      "training_step_time": 10.28418517112732
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 2.4691691398620605,
      "step": 6243
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 6243,
      "training_step_time": 7.497748136520386
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 2.4359090328216553,
      "step": 6243
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 6243,
      "training_step_time": 7.46457839012146
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 3.412440538406372,
      "step": 6243
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 6243,
      "training_step_time": 10.189096450805664
    },
    {
      "epoch": 2.3815155029296876e-05,
      "model_forward_time": 3.1508452892303467,
      "step": 6243
    },
    {
      "epoch": 2.3815155029296876e-05,
      "step": 6243,
      "training_step_time": 10.022249221801758
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 3.1632437705993652,
      "step": 6244
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 6244,
      "training_step_time": 9.935585975646973
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 2.437938690185547,
      "step": 6244
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 6244,
      "training_step_time": 7.464884042739868
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 3.1371681690216064,
      "step": 6244
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 6244,
      "training_step_time": 9.913566589355469
    },
    {
      "epoch": 2.38189697265625e-05,
      "model_forward_time": 2.4398064613342285,
      "step": 6244
    },
    {
      "epoch": 2.38189697265625e-05,
      "step": 6244,
      "training_step_time": 10.257856845855713
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 3.5158517360687256,
      "step": 6245
    },
    {
      "dynamics_loss_avg": 0.07157321590930224,
      "epoch": 2.3822784423828125e-05,
      "step": 6245
    },
    {
      "action_loss_avg": 0.036777478130534294,
      "epoch": 2.3822784423828125e-05,
      "step": 6245
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 6245,
      "training_step_time": 10.287755012512207
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 3.1378936767578125,
      "step": 6245
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 6245,
      "training_step_time": 9.905903100967407
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 3.143904447555542,
      "step": 6245
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 6245,
      "training_step_time": 9.915987968444824
    },
    {
      "epoch": 2.3822784423828125e-05,
      "model_forward_time": 3.142594814300537,
      "step": 6245
    },
    {
      "epoch": 2.3822784423828125e-05,
      "step": 6245,
      "training_step_time": 10.298428535461426
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 3.179414987564087,
      "step": 6246
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 6246,
      "training_step_time": 10.22022557258606
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 3.1445343494415283,
      "step": 6246
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 6246,
      "training_step_time": 9.918569803237915
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 3.140855550765991,
      "step": 6246
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 6246,
      "training_step_time": 9.914020299911499
    },
    {
      "epoch": 2.382659912109375e-05,
      "model_forward_time": 3.1395108699798584,
      "step": 6246
    },
    {
      "epoch": 2.382659912109375e-05,
      "step": 6246,
      "training_step_time": 10.309628009796143
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 2.4513769149780273,
      "step": 6247
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 6247,
      "training_step_time": 7.481434345245361
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 3.138432025909424,
      "step": 6247
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 6247,
      "training_step_time": 9.901213884353638
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 2.712292432785034,
      "step": 6247
    },
    {
      "dynamics_loss_avg": 0.06119655445218086,
      "epoch": 2.3830413818359375e-05,
      "step": 6247
    },
    {
      "action_loss_avg": 0.031229126453399658,
      "epoch": 2.3830413818359375e-05,
      "step": 6247
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 6247,
      "training_step_time": 7.73697566986084
    },
    {
      "epoch": 2.3830413818359375e-05,
      "model_forward_time": 3.134282350540161,
      "step": 6247
    },
    {
      "epoch": 2.3830413818359375e-05,
      "step": 6247,
      "training_step_time": 10.300454139709473
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 3.162797689437866,
      "step": 6248
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 6248,
      "training_step_time": 9.926702737808228
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 3.132478952407837,
      "step": 6248
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 6248,
      "training_step_time": 9.897958993911743
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 3.1285715103149414,
      "step": 6248
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 6248,
      "training_step_time": 10.19968581199646
    },
    {
      "epoch": 2.3834228515625e-05,
      "model_forward_time": 3.135110855102539,
      "step": 6248
    },
    {
      "epoch": 2.3834228515625e-05,
      "step": 6248,
      "training_step_time": 10.30292797088623
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 3.192795515060425,
      "step": 6249
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 6249,
      "training_step_time": 9.962675094604492
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 3.141223669052124,
      "step": 6249
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 6249,
      "training_step_time": 9.905656099319458
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 3.1340749263763428,
      "step": 6249
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 6249,
      "training_step_time": 9.900145530700684
    },
    {
      "epoch": 2.3838043212890624e-05,
      "model_forward_time": 3.397517681121826,
      "step": 6249
    },
    {
      "epoch": 2.3838043212890624e-05,
      "step": 6249,
      "training_step_time": 10.25349497795105
    },
    {
      "epoch": 2.384185791015625e-05,
      "grad_norm": 0.316860556602478,
      "learning_rate": 1e-05,
      "loss": 0.0887,
      "step": 6250
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 2.4450106620788574,
      "step": 6250
    },
    {
      "dynamics_loss_avg": 0.05289427768439055,
      "epoch": 2.384185791015625e-05,
      "step": 6250
    },
    {
      "action_loss_avg": 0.01806408944539726,
      "epoch": 2.384185791015625e-05,
      "step": 6250
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 6250,
      "training_step_time": 7.46645188331604
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 2.431640625,
      "step": 6250
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 6250,
      "training_step_time": 7.449753761291504
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 3.1308372020721436,
      "step": 6250
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 6250,
      "training_step_time": 9.892318487167358
    },
    {
      "epoch": 2.384185791015625e-05,
      "model_forward_time": 3.1406571865081787,
      "step": 6250
    },
    {
      "epoch": 2.384185791015625e-05,
      "step": 6250,
      "training_step_time": 10.273884296417236
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 3.1705915927886963,
      "step": 6251
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 6251,
      "training_step_time": 9.944451332092285
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 2.684709072113037,
      "step": 6251
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 6251,
      "training_step_time": 7.716098308563232
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 2.439350128173828,
      "step": 6251
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 6251,
      "training_step_time": 7.4669599533081055
    },
    {
      "epoch": 2.3845672607421876e-05,
      "model_forward_time": 3.1374666690826416,
      "step": 6251
    },
    {
      "epoch": 2.3845672607421876e-05,
      "step": 6251,
      "training_step_time": 10.28460168838501
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 2.4522948265075684,
      "step": 6252
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 6252,
      "training_step_time": 7.486204385757446
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 3.1441497802734375,
      "step": 6252
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 6252,
      "training_step_time": 9.918961524963379
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 3.410796880722046,
      "step": 6252
    },
    {
      "dynamics_loss_avg": 0.07512428909540177,
      "epoch": 2.38494873046875e-05,
      "step": 6252
    },
    {
      "action_loss_avg": 0.03556137690320611,
      "epoch": 2.38494873046875e-05,
      "step": 6252
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 6252,
      "training_step_time": 10.181884527206421
    },
    {
      "epoch": 2.38494873046875e-05,
      "model_forward_time": 2.4420557022094727,
      "step": 6252
    },
    {
      "epoch": 2.38494873046875e-05,
      "step": 6252,
      "training_step_time": 10.365612030029297
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 2.4549808502197266,
      "step": 6253
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 6253,
      "training_step_time": 7.487148284912109
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 3.1387012004852295,
      "step": 6253
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 6253,
      "training_step_time": 9.908938884735107
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 2.4392967224121094,
      "step": 6253
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 6253,
      "training_step_time": 7.461759090423584
    },
    {
      "epoch": 2.3853302001953125e-05,
      "model_forward_time": 3.4280030727386475,
      "step": 6253
    },
    {
      "epoch": 2.3853302001953125e-05,
      "step": 6253,
      "training_step_time": 10.284638166427612
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 3.161198377609253,
      "step": 6254
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 6254,
      "training_step_time": 9.93293309211731
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 3.135944366455078,
      "step": 6254
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 6254,
      "training_step_time": 9.904001951217651
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 3.1413991451263428,
      "step": 6254
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 6254,
      "training_step_time": 9.911608934402466
    },
    {
      "epoch": 2.385711669921875e-05,
      "model_forward_time": 2.4384586811065674,
      "step": 6254
    },
    {
      "epoch": 2.385711669921875e-05,
      "step": 6254,
      "training_step_time": 10.305084943771362
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 2.4491336345672607,
      "step": 6255
    },
    {
      "dynamics_loss_avg": 0.06938365176320076,
      "epoch": 2.3860931396484375e-05,
      "step": 6255
    },
    {
      "action_loss_avg": 0.02950077401474118,
      "epoch": 2.3860931396484375e-05,
      "step": 6255
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 6255,
      "training_step_time": 7.760721445083618
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 2.435447931289673,
      "step": 6255
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 6255,
      "training_step_time": 7.4580137729644775
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 3.13663649559021,
      "step": 6255
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 6255,
      "training_step_time": 9.900568962097168
    },
    {
      "epoch": 2.3860931396484375e-05,
      "model_forward_time": 2.4319968223571777,
      "step": 6255
    },
    {
      "epoch": 2.3860931396484375e-05,
      "step": 6255,
      "training_step_time": 10.22136926651001
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 3.1532084941864014,
      "step": 6256
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 6256,
      "training_step_time": 9.921119928359985
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 3.142411708831787,
      "step": 6256
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 6256,
      "training_step_time": 9.924012660980225
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 2.708513021469116,
      "step": 6256
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 6256,
      "training_step_time": 7.7304911613464355
    },
    {
      "epoch": 2.386474609375e-05,
      "model_forward_time": 1.7424800395965576,
      "step": 6256
    },
    {
      "epoch": 2.386474609375e-05,
      "step": 6256,
      "training_step_time": 10.313533306121826
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 3.155454158782959,
      "step": 6257
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 6257,
      "training_step_time": 9.926316499710083
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 2.4358487129211426,
      "step": 6257
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 6257,
      "training_step_time": 7.462209939956665
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 2.435729742050171,
      "step": 6257
    },
    {
      "dynamics_loss_avg": 0.057835440151393414,
      "epoch": 2.3868560791015624e-05,
      "step": 6257
    },
    {
      "action_loss_avg": 0.023430534498766066,
      "epoch": 2.3868560791015624e-05,
      "step": 6257
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 6257,
      "training_step_time": 7.471853971481323
    },
    {
      "epoch": 2.3868560791015624e-05,
      "model_forward_time": 2.7591331005096436,
      "step": 6257
    },
    {
      "epoch": 2.3868560791015624e-05,
      "step": 6257,
      "training_step_time": 10.278180599212646
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 2.4501373767852783,
      "step": 6258
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 6258,
      "training_step_time": 7.477057695388794
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 2.4322776794433594,
      "step": 6258
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 6258,
      "training_step_time": 7.457785129547119
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 3.1322431564331055,
      "step": 6258
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 6258,
      "training_step_time": 9.8995943069458
    },
    {
      "epoch": 2.387237548828125e-05,
      "model_forward_time": 2.434657335281372,
      "step": 6258
    },
    {
      "epoch": 2.387237548828125e-05,
      "step": 6258,
      "training_step_time": 10.327641248703003
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 3.157839298248291,
      "step": 6259
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 6259,
      "training_step_time": 9.927874088287354
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 3.4102916717529297,
      "step": 6259
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 6259,
      "training_step_time": 10.176171779632568
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 2.433706283569336,
      "step": 6259
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 6259,
      "training_step_time": 7.458812475204468
    },
    {
      "epoch": 2.3876190185546876e-05,
      "model_forward_time": 3.139901638031006,
      "step": 6259
    },
    {
      "epoch": 2.3876190185546876e-05,
      "step": 6259,
      "training_step_time": 10.280174493789673
    },
    {
      "epoch": 2.38800048828125e-05,
      "grad_norm": 0.436259388923645,
      "learning_rate": 1e-05,
      "loss": 0.0903,
      "step": 6260
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 3.170367479324341,
      "step": 6260
    },
    {
      "dynamics_loss_avg": 0.06650203187018633,
      "epoch": 2.38800048828125e-05,
      "step": 6260
    },
    {
      "action_loss_avg": 0.029107725573703648,
      "epoch": 2.38800048828125e-05,
      "step": 6260
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 6260,
      "training_step_time": 9.941470861434937
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 3.127547025680542,
      "step": 6260
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 6260,
      "training_step_time": 10.168753862380981
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 2.4315755367279053,
      "step": 6260
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 6260,
      "training_step_time": 7.4543538093566895
    },
    {
      "epoch": 2.38800048828125e-05,
      "model_forward_time": 3.1310386657714844,
      "step": 6260
    },
    {
      "epoch": 2.38800048828125e-05,
      "step": 6260,
      "training_step_time": 10.29306960105896
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 3.146843433380127,
      "step": 6261
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 6261,
      "training_step_time": 9.9164719581604
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 3.145120859146118,
      "step": 6261
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 6261,
      "training_step_time": 9.920185089111328
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 3.4205713272094727,
      "step": 6261
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 6261,
      "training_step_time": 10.18634033203125
    },
    {
      "epoch": 2.3883819580078125e-05,
      "model_forward_time": 3.1346020698547363,
      "step": 6261
    },
    {
      "epoch": 2.3883819580078125e-05,
      "step": 6261,
      "training_step_time": 10.304940700531006
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 3.145294666290283,
      "step": 6262
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 6262,
      "training_step_time": 9.917885541915894
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 2.4335885047912598,
      "step": 6262
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 6262,
      "training_step_time": 7.465181827545166
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 3.13457989692688,
      "step": 6262
    },
    {
      "dynamics_loss_avg": 0.04485415266826749,
      "epoch": 2.388763427734375e-05,
      "step": 6262
    },
    {
      "action_loss_avg": 0.018022906687110662,
      "epoch": 2.388763427734375e-05,
      "step": 6262
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 6262,
      "training_step_time": 9.904343843460083
    },
    {
      "epoch": 2.388763427734375e-05,
      "model_forward_time": 2.430156946182251,
      "step": 6262
    },
    {
      "epoch": 2.388763427734375e-05,
      "step": 6262,
      "training_step_time": 10.019014120101929
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 3.4701130390167236,
      "step": 6263
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 6263,
      "training_step_time": 10.246002197265625
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 3.1343600749969482,
      "step": 6263
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 6263,
      "training_step_time": 9.90126633644104
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 3.134554624557495,
      "step": 6263
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 6263,
      "training_step_time": 9.901390075683594
    },
    {
      "epoch": 2.3891448974609375e-05,
      "model_forward_time": 2.4302492141723633,
      "step": 6263
    },
    {
      "epoch": 2.3891448974609375e-05,
      "step": 6263,
      "training_step_time": 10.30538010597229
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 3.1357932090759277,
      "step": 6264
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 6264,
      "training_step_time": 10.175740242004395
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 1.7417683601379395,
      "step": 6264
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 6264,
      "training_step_time": 5.201093673706055
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 3.135976791381836,
      "step": 6264
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 6264,
      "training_step_time": 9.902184963226318
    },
    {
      "epoch": 2.3895263671875e-05,
      "model_forward_time": 2.4293315410614014,
      "step": 6264
    },
    {
      "epoch": 2.3895263671875e-05,
      "step": 6264,
      "training_step_time": 10.301963090896606
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 3.137803792953491,
      "step": 6265
    },
    {
      "dynamics_loss_avg": 0.06494689267128706,
      "epoch": 2.3899078369140624e-05,
      "step": 6265
    },
    {
      "action_loss_avg": 0.026711721532046796,
      "epoch": 2.3899078369140624e-05,
      "step": 6265
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 6265,
      "training_step_time": 9.908277988433838
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 2.4320297241210938,
      "step": 6265
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 6265,
      "training_step_time": 7.455118656158447
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 3.457085609436035,
      "step": 6265
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 6265,
      "training_step_time": 10.223917484283447
    },
    {
      "epoch": 2.3899078369140624e-05,
      "model_forward_time": 2.430805206298828,
      "step": 6265
    },
    {
      "epoch": 2.3899078369140624e-05,
      "step": 6265,
      "training_step_time": 10.263214349746704
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 3.153261184692383,
      "step": 6266
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 6266,
      "training_step_time": 9.92748737335205
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 3.131187915802002,
      "step": 6266
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 6266,
      "training_step_time": 9.89700198173523
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 2.428947687149048,
      "step": 6266
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 6266,
      "training_step_time": 7.448570489883423
    },
    {
      "epoch": 2.390289306640625e-05,
      "model_forward_time": 3.3880703449249268,
      "step": 6266
    },
    {
      "epoch": 2.390289306640625e-05,
      "step": 6266,
      "training_step_time": 10.283663511276245
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 3.1489298343658447,
      "step": 6267
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 6267,
      "training_step_time": 9.918652057647705
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 3.131131410598755,
      "step": 6267
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 6267,
      "training_step_time": 9.894855737686157
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 3.1801581382751465,
      "step": 6267
    },
    {
      "dynamics_loss_avg": 0.06955404849722982,
      "epoch": 2.3906707763671876e-05,
      "step": 6267
    },
    {
      "action_loss_avg": 0.027364346012473107,
      "epoch": 2.3906707763671876e-05,
      "step": 6267
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 6267,
      "training_step_time": 9.95163369178772
    },
    {
      "epoch": 2.3906707763671876e-05,
      "model_forward_time": 2.4678521156311035,
      "step": 6267
    },
    {
      "epoch": 2.3906707763671876e-05,
      "step": 6267,
      "training_step_time": 10.012453556060791
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 3.1464266777038574,
      "step": 6268
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 6268,
      "training_step_time": 10.219197511672974
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 2.432173013687134,
      "step": 6268
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 6268,
      "training_step_time": 7.455355644226074
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 2.4359657764434814,
      "step": 6268
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 6268,
      "training_step_time": 7.461393117904663
    },
    {
      "epoch": 2.39105224609375e-05,
      "model_forward_time": 2.4316163063049316,
      "step": 6268
    },
    {
      "epoch": 2.39105224609375e-05,
      "step": 6268,
      "training_step_time": 10.316964626312256
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 2.441711902618408,
      "step": 6269
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 6269,
      "training_step_time": 7.467709064483643
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 3.132244348526001,
      "step": 6269
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 6269,
      "training_step_time": 9.900665521621704
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 2.715223550796509,
      "step": 6269
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 6269,
      "training_step_time": 7.7390525341033936
    },
    {
      "epoch": 2.3914337158203125e-05,
      "model_forward_time": 3.13470721244812,
      "step": 6269
    },
    {
      "epoch": 2.3914337158203125e-05,
      "step": 6269,
      "training_step_time": 10.301366329193115
    },
    {
      "epoch": 2.391815185546875e-05,
      "grad_norm": 0.26727744936943054,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 6270
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 3.1797726154327393,
      "step": 6270
    },
    {
      "dynamics_loss_avg": 0.04970342703163624,
      "epoch": 2.391815185546875e-05,
      "step": 6270
    },
    {
      "action_loss_avg": 0.0230513134971261,
      "epoch": 2.391815185546875e-05,
      "step": 6270
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 6270,
      "training_step_time": 9.950057029724121
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 2.4345333576202393,
      "step": 6270
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 6270,
      "training_step_time": 7.45828652381897
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 3.138786554336548,
      "step": 6270
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 6270,
      "training_step_time": 9.90756869316101
    },
    {
      "epoch": 2.391815185546875e-05,
      "model_forward_time": 3.445089817047119,
      "step": 6270
    },
    {
      "epoch": 2.391815185546875e-05,
      "step": 6270,
      "training_step_time": 10.32199740409851
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 2.4515254497528076,
      "step": 6271
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 6271,
      "training_step_time": 7.477337598800659
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 3.133046865463257,
      "step": 6271
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 6271,
      "training_step_time": 9.894951820373535
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 2.43434739112854,
      "step": 6271
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 6271,
      "training_step_time": 7.45895528793335
    },
    {
      "epoch": 2.3921966552734375e-05,
      "model_forward_time": 2.43084454536438,
      "step": 6271
    },
    {
      "epoch": 2.3921966552734375e-05,
      "step": 6271,
      "training_step_time": 10.31988000869751
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 3.141728401184082,
      "step": 6272
    },
    {
      "epoch": 2.392578125e-05,
      "step": 6272,
      "training_step_time": 10.193634033203125
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 2.4291117191314697,
      "step": 6272
    },
    {
      "epoch": 2.392578125e-05,
      "step": 6272,
      "training_step_time": 7.449172258377075
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 3.1298983097076416,
      "step": 6272
    },
    {
      "dynamics_loss_avg": 0.061071842163801196,
      "epoch": 2.392578125e-05,
      "step": 6272
    },
    {
      "action_loss_avg": 0.02763686222024262,
      "epoch": 2.392578125e-05,
      "step": 6272
    },
    {
      "epoch": 2.392578125e-05,
      "step": 6272,
      "training_step_time": 9.890613079071045
    },
    {
      "epoch": 2.392578125e-05,
      "model_forward_time": 2.431666851043701,
      "step": 6272
    },
    {
      "epoch": 2.392578125e-05,
      "step": 6272,
      "training_step_time": 10.302777290344238
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 3.1466336250305176,
      "step": 6273
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 6273,
      "training_step_time": 9.910647869110107
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 3.138349771499634,
      "step": 6273
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 6273,
      "training_step_time": 9.901761531829834
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 2.707571268081665,
      "step": 6273
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 6273,
      "training_step_time": 7.725650310516357
    },
    {
      "epoch": 2.3929595947265624e-05,
      "model_forward_time": 2.429521322250366,
      "step": 6273
    },
    {
      "epoch": 2.3929595947265624e-05,
      "step": 6273,
      "training_step_time": 10.265841245651245
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 2.4434380531311035,
      "step": 6274
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 6274,
      "training_step_time": 7.45526647567749
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 3.1245267391204834,
      "step": 6274
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 6274,
      "training_step_time": 9.881563186645508
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 2.4266250133514404,
      "step": 6274
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 6274,
      "training_step_time": 7.444411277770996
    },
    {
      "epoch": 2.393341064453125e-05,
      "model_forward_time": 2.7110071182250977,
      "step": 6274
    },
    {
      "epoch": 2.393341064453125e-05,
      "step": 6274,
      "training_step_time": 10.285501956939697
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 2.4387643337249756,
      "step": 6275
    },
    {
      "dynamics_loss_avg": 0.059296478517353535,
      "epoch": 2.3937225341796876e-05,
      "step": 6275
    },
    {
      "action_loss_avg": 0.02376289265230298,
      "epoch": 2.3937225341796876e-05,
      "step": 6275
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 6275,
      "training_step_time": 7.461164712905884
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 2.427229404449463,
      "step": 6275
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 6275,
      "training_step_time": 7.449542045593262
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 3.1274542808532715,
      "step": 6275
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 6275,
      "training_step_time": 9.889675617218018
    },
    {
      "epoch": 2.3937225341796876e-05,
      "model_forward_time": 3.1412220001220703,
      "step": 6275
    },
    {
      "epoch": 2.3937225341796876e-05,
      "step": 6275,
      "training_step_time": 10.011338949203491
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 2.726780414581299,
      "step": 6276
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 6276,
      "training_step_time": 7.751437425613403
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 2.4282045364379883,
      "step": 6276
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 6276,
      "training_step_time": 7.449373960494995
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 2.4299261569976807,
      "step": 6276
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 6276,
      "training_step_time": 7.452127933502197
    },
    {
      "epoch": 2.39410400390625e-05,
      "model_forward_time": 3.1324679851531982,
      "step": 6276
    },
    {
      "epoch": 2.39410400390625e-05,
      "step": 6276,
      "training_step_time": 10.290924787521362
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 2.467524528503418,
      "step": 6277
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 6277,
      "training_step_time": 7.491708755493164
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 3.1349074840545654,
      "step": 6277
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 6277,
      "training_step_time": 9.900990009307861
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 3.3820788860321045,
      "step": 6277
    },
    {
      "dynamics_loss_avg": 0.05512782521545887,
      "epoch": 2.3944854736328126e-05,
      "step": 6277
    },
    {
      "action_loss_avg": 0.02507700528949499,
      "epoch": 2.3944854736328126e-05,
      "step": 6277
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 6277,
      "training_step_time": 10.150370836257935
    },
    {
      "epoch": 2.3944854736328126e-05,
      "model_forward_time": 3.140476703643799,
      "step": 6277
    },
    {
      "epoch": 2.3944854736328126e-05,
      "step": 6277,
      "training_step_time": 10.246705770492554
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 3.1492247581481934,
      "step": 6278
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 6278,
      "training_step_time": 9.917226076126099
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 2.4295709133148193,
      "step": 6278
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 6278,
      "training_step_time": 7.451632976531982
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 3.132481098175049,
      "step": 6278
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 6278,
      "training_step_time": 9.898161888122559
    },
    {
      "epoch": 2.394866943359375e-05,
      "model_forward_time": 2.7006518840789795,
      "step": 6278
    },
    {
      "epoch": 2.394866943359375e-05,
      "step": 6278,
      "training_step_time": 10.257189512252808
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 3.1593406200408936,
      "step": 6279
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 6279,
      "training_step_time": 9.918956518173218
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 2.4350132942199707,
      "step": 6279
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 6279,
      "training_step_time": 7.453944683074951
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 3.1364588737487793,
      "step": 6279
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 6279,
      "training_step_time": 9.899168491363525
    },
    {
      "epoch": 2.3952484130859375e-05,
      "model_forward_time": 2.4339234828948975,
      "step": 6279
    },
    {
      "epoch": 2.3952484130859375e-05,
      "step": 6279,
      "training_step_time": 10.011469602584839
    },
    {
      "epoch": 2.3956298828125e-05,
      "grad_norm": 0.2951928973197937,
      "learning_rate": 1e-05,
      "loss": 0.0853,
      "step": 6280
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 3.4877326488494873,
      "step": 6280
    },
    {
      "dynamics_loss_avg": 0.044123804476112125,
      "epoch": 2.3956298828125e-05,
      "step": 6280
    },
    {
      "action_loss_avg": 0.025682517886161806,
      "epoch": 2.3956298828125e-05,
      "step": 6280
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 6280,
      "training_step_time": 10.256404399871826
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 3.1285340785980225,
      "step": 6280
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 6280,
      "training_step_time": 9.893391847610474
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 3.136174440383911,
      "step": 6280
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 6280,
      "training_step_time": 9.898394107818604
    },
    {
      "epoch": 2.3956298828125e-05,
      "model_forward_time": 3.1364119052886963,
      "step": 6280
    },
    {
      "epoch": 2.3956298828125e-05,
      "step": 6280,
      "training_step_time": 10.266229629516602
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 3.1499428749084473,
      "step": 6281
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 6281,
      "training_step_time": 9.913459300994873
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 2.0273678302764893,
      "step": 6281
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 6281,
      "training_step_time": 5.477308988571167
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 3.1303646564483643,
      "step": 6281
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 6281,
      "training_step_time": 9.889034509658813
    },
    {
      "epoch": 2.3960113525390624e-05,
      "model_forward_time": 2.433568239212036,
      "step": 6281
    },
    {
      "epoch": 2.3960113525390624e-05,
      "step": 6281,
      "training_step_time": 10.31267786026001
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 2.442821741104126,
      "step": 6282
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 6282,
      "training_step_time": 7.4621992111206055
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 2.427555799484253,
      "step": 6282
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 6282,
      "training_step_time": 7.444924592971802
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 3.1308422088623047,
      "step": 6282
    },
    {
      "dynamics_loss_avg": 0.04605286438018084,
      "epoch": 2.396392822265625e-05,
      "step": 6282
    },
    {
      "action_loss_avg": 0.01865662313066423,
      "epoch": 2.396392822265625e-05,
      "step": 6282
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 6282,
      "training_step_time": 10.158452033996582
    },
    {
      "epoch": 2.396392822265625e-05,
      "model_forward_time": 3.132497787475586,
      "step": 6282
    },
    {
      "epoch": 2.396392822265625e-05,
      "step": 6282,
      "training_step_time": 10.022058248519897
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 3.1554062366485596,
      "step": 6283
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 6283,
      "training_step_time": 9.920214653015137
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 3.133282423019409,
      "step": 6283
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 6283,
      "training_step_time": 9.900964736938477
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 2.429705858230591,
      "step": 6283
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 6283,
      "training_step_time": 7.452299356460571
    },
    {
      "epoch": 2.3967742919921876e-05,
      "model_forward_time": 2.429857015609741,
      "step": 6283
    },
    {
      "epoch": 2.3967742919921876e-05,
      "step": 6283,
      "training_step_time": 10.289036750793457
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 3.172598361968994,
      "step": 6284
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 6284,
      "training_step_time": 9.938998699188232
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 3.139124870300293,
      "step": 6284
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 6284,
      "training_step_time": 9.901758193969727
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 3.131998300552368,
      "step": 6284
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 6284,
      "training_step_time": 9.899883508682251
    },
    {
      "epoch": 2.39715576171875e-05,
      "model_forward_time": 3.1319491863250732,
      "step": 6284
    },
    {
      "epoch": 2.39715576171875e-05,
      "step": 6284,
      "training_step_time": 10.260365009307861
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 3.405111312866211,
      "step": 6285
    },
    {
      "dynamics_loss_avg": 0.04191039660945535,
      "epoch": 2.3975372314453126e-05,
      "step": 6285
    },
    {
      "action_loss_avg": 0.019103808631189167,
      "epoch": 2.3975372314453126e-05,
      "step": 6285
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 6285,
      "training_step_time": 10.174984216690063
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 3.136354446411133,
      "step": 6285
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 6285,
      "training_step_time": 9.89886474609375
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 2.4306769371032715,
      "step": 6285
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 6285,
      "training_step_time": 7.451039552688599
    },
    {
      "epoch": 2.3975372314453126e-05,
      "model_forward_time": 2.431459665298462,
      "step": 6285
    },
    {
      "epoch": 2.3975372314453126e-05,
      "step": 6285,
      "training_step_time": 10.017158269882202
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 2.454639196395874,
      "step": 6286
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 6286,
      "training_step_time": 7.477950096130371
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 3.1302695274353027,
      "step": 6286
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 6286,
      "training_step_time": 9.892045497894287
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 2.697763681411743,
      "step": 6286
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 6286,
      "training_step_time": 7.716429233551025
    },
    {
      "epoch": 2.397918701171875e-05,
      "model_forward_time": 2.4266202449798584,
      "step": 6286
    },
    {
      "epoch": 2.397918701171875e-05,
      "step": 6286,
      "training_step_time": 10.292787313461304
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 3.158888101577759,
      "step": 6287
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 6287,
      "training_step_time": 9.922923803329468
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 3.129398822784424,
      "step": 6287
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 6287,
      "training_step_time": 9.888987064361572
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 2.4273524284362793,
      "step": 6287
    },
    {
      "dynamics_loss_avg": 0.07254669517278671,
      "epoch": 2.3983001708984375e-05,
      "step": 6287
    },
    {
      "action_loss_avg": 0.02832707725465298,
      "epoch": 2.3983001708984375e-05,
      "step": 6287
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 6287,
      "training_step_time": 7.443924427032471
    },
    {
      "epoch": 2.3983001708984375e-05,
      "model_forward_time": 2.687255620956421,
      "step": 6287
    },
    {
      "epoch": 2.3983001708984375e-05,
      "step": 6287,
      "training_step_time": 9.998754024505615
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 3.1422078609466553,
      "step": 6288
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 6288,
      "training_step_time": 9.908051490783691
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 2.4279987812042236,
      "step": 6288
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 6288,
      "training_step_time": 7.448393821716309
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 2.4291954040527344,
      "step": 6288
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 6288,
      "training_step_time": 7.447011470794678
    },
    {
      "epoch": 2.398681640625e-05,
      "model_forward_time": 2.426392078399658,
      "step": 6288
    },
    {
      "epoch": 2.398681640625e-05,
      "step": 6288,
      "training_step_time": 10.005213499069214
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 2.4495701789855957,
      "step": 6289
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 6289,
      "training_step_time": 7.471293926239014
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 2.690425395965576,
      "step": 6289
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 6289,
      "training_step_time": 7.711100339889526
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 3.133875846862793,
      "step": 6289
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 6289,
      "training_step_time": 9.897214651107788
    },
    {
      "epoch": 2.3990631103515624e-05,
      "model_forward_time": 3.133005380630493,
      "step": 6289
    },
    {
      "epoch": 2.3990631103515624e-05,
      "step": 6289,
      "training_step_time": 10.283975601196289
    },
    {
      "epoch": 2.399444580078125e-05,
      "grad_norm": 0.5778221487998962,
      "learning_rate": 1e-05,
      "loss": 0.0883,
      "step": 6290
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 1.7623491287231445,
      "step": 6290
    },
    {
      "dynamics_loss_avg": 0.05866013970226049,
      "epoch": 2.399444580078125e-05,
      "step": 6290
    },
    {
      "action_loss_avg": 0.025654822308570146,
      "epoch": 2.399444580078125e-05,
      "step": 6290
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 6290,
      "training_step_time": 5.223548889160156
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 3.1312854290008545,
      "step": 6290
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 6290,
      "training_step_time": 9.897075653076172
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 2.6920840740203857,
      "step": 6290
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 6290,
      "training_step_time": 7.711562633514404
    },
    {
      "epoch": 2.399444580078125e-05,
      "model_forward_time": 2.431509256362915,
      "step": 6290
    },
    {
      "epoch": 2.399444580078125e-05,
      "step": 6290,
      "training_step_time": 10.004706859588623
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 3.1744210720062256,
      "step": 6291
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 6291,
      "training_step_time": 9.938831806182861
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 3.1383957862854004,
      "step": 6291
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 6291,
      "training_step_time": 9.905883312225342
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 3.135138988494873,
      "step": 6291
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 6291,
      "training_step_time": 9.899621963500977
    },
    {
      "epoch": 2.3998260498046876e-05,
      "model_forward_time": 3.414658784866333,
      "step": 6291
    },
    {
      "epoch": 2.3998260498046876e-05,
      "step": 6291,
      "training_step_time": 10.322187423706055
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 2.448660135269165,
      "step": 6292
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 6292,
      "training_step_time": 7.476935148239136
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 2.435312271118164,
      "step": 6292
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 6292,
      "training_step_time": 7.4665117263793945
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 3.137357234954834,
      "step": 6292
    },
    {
      "dynamics_loss_avg": 0.06381768714636564,
      "epoch": 2.40020751953125e-05,
      "step": 6292
    },
    {
      "action_loss_avg": 0.029091898119077086,
      "epoch": 2.40020751953125e-05,
      "step": 6292
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 6292,
      "training_step_time": 9.919398546218872
    },
    {
      "epoch": 2.40020751953125e-05,
      "model_forward_time": 2.436309814453125,
      "step": 6292
    },
    {
      "epoch": 2.40020751953125e-05,
      "step": 6292,
      "training_step_time": 10.333578824996948
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 3.473947048187256,
      "step": 6293
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 6293,
      "training_step_time": 10.257704496383667
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 3.143333911895752,
      "step": 6293
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 6293,
      "training_step_time": 9.917613506317139
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 3.1455399990081787,
      "step": 6293
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 6293,
      "training_step_time": 9.917389631271362
    },
    {
      "epoch": 2.4005889892578126e-05,
      "model_forward_time": 2.4376728534698486,
      "step": 6293
    },
    {
      "epoch": 2.4005889892578126e-05,
      "step": 6293,
      "training_step_time": 10.029756784439087
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 3.149984121322632,
      "step": 6294
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 6294,
      "training_step_time": 9.942452192306519
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 3.136293649673462,
      "step": 6294
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 6294,
      "training_step_time": 9.905611276626587
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 2.678370952606201,
      "step": 6294
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 6294,
      "training_step_time": 7.704291105270386
    },
    {
      "epoch": 2.400970458984375e-05,
      "model_forward_time": 3.1373822689056396,
      "step": 6294
    },
    {
      "epoch": 2.400970458984375e-05,
      "step": 6294,
      "training_step_time": 10.286271095275879
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 2.4478323459625244,
      "step": 6295
    },
    {
      "dynamics_loss_avg": 0.06234505418688059,
      "epoch": 2.4013519287109375e-05,
      "step": 6295
    },
    {
      "action_loss_avg": 0.03668143190443516,
      "epoch": 2.4013519287109375e-05,
      "step": 6295
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 6295,
      "training_step_time": 7.479743480682373
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 2.4287843704223633,
      "step": 6295
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 6295,
      "training_step_time": 7.458371162414551
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 2.4325385093688965,
      "step": 6295
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 6295,
      "training_step_time": 7.4673237800598145
    },
    {
      "epoch": 2.4013519287109375e-05,
      "model_forward_time": 3.4086155891418457,
      "step": 6295
    },
    {
      "epoch": 2.4013519287109375e-05,
      "step": 6295,
      "training_step_time": 10.304031133651733
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 3.14164400100708,
      "step": 6296
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 6296,
      "training_step_time": 9.910223960876465
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 3.1331803798675537,
      "step": 6296
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 6296,
      "training_step_time": 9.903149604797363
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 3.1357836723327637,
      "step": 6296
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 6296,
      "training_step_time": 9.905229806900024
    },
    {
      "epoch": 2.4017333984375e-05,
      "model_forward_time": 3.1319522857666016,
      "step": 6296
    },
    {
      "epoch": 2.4017333984375e-05,
      "step": 6296,
      "training_step_time": 10.036912679672241
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 2.7288873195648193,
      "step": 6297
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 6297,
      "training_step_time": 7.757333278656006
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 2.427466869354248,
      "step": 6297
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 6297,
      "training_step_time": 7.4547812938690186
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 3.134652853012085,
      "step": 6297
    },
    {
      "dynamics_loss_avg": 0.06781297419220209,
      "epoch": 2.4021148681640624e-05,
      "step": 6297
    },
    {
      "action_loss_avg": 0.02662732624448836,
      "epoch": 2.4021148681640624e-05,
      "step": 6297
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 6297,
      "training_step_time": 9.908228158950806
    },
    {
      "epoch": 2.4021148681640624e-05,
      "model_forward_time": 2.4320194721221924,
      "step": 6297
    },
    {
      "epoch": 2.4021148681640624e-05,
      "step": 6297,
      "training_step_time": 10.322632551193237
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 3.1461524963378906,
      "step": 6298
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 6298,
      "training_step_time": 9.91910457611084
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 2.4387354850769043,
      "step": 6298
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 6298,
      "training_step_time": 7.745219945907593
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 3.1424319744110107,
      "step": 6298
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 6298,
      "training_step_time": 9.912652015686035
    },
    {
      "epoch": 2.402496337890625e-05,
      "model_forward_time": 3.1407582759857178,
      "step": 6298
    },
    {
      "epoch": 2.402496337890625e-05,
      "step": 6298,
      "training_step_time": 10.04784870147705
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 2.4588940143585205,
      "step": 6299
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 6299,
      "training_step_time": 7.491015672683716
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 3.1370785236358643,
      "step": 6299
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 6299,
      "training_step_time": 9.910508155822754
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 2.4387547969818115,
      "step": 6299
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 6299,
      "training_step_time": 7.466236352920532
    },
    {
      "epoch": 2.4028778076171876e-05,
      "model_forward_time": 2.710993766784668,
      "step": 6299
    },
    {
      "epoch": 2.4028778076171876e-05,
      "step": 6299,
      "training_step_time": 10.297991514205933
    },
    {
      "epoch": 2.40325927734375e-05,
      "grad_norm": 0.4390122890472412,
      "learning_rate": 1e-05,
      "loss": 0.0847,
      "step": 6300
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 3.1679084300994873,
      "step": 6300
    },
    {
      "dynamics_loss_avg": 0.05384974852204323,
      "epoch": 2.40325927734375e-05,
      "step": 6300
    },
    {
      "action_loss_avg": 0.02170174764469266,
      "epoch": 2.40325927734375e-05,
      "step": 6300
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 6300,
      "training_step_time": 9.94047212600708
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 3.1427392959594727,
      "step": 6300
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 6300,
      "training_step_time": 9.910876512527466
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 3.130751609802246,
      "step": 6300
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 6300,
      "training_step_time": 9.89774203300476
    },
    {
      "epoch": 2.40325927734375e-05,
      "model_forward_time": 3.138902425765991,
      "step": 6300
    },
    {
      "epoch": 2.40325927734375e-05,
      "step": 6300,
      "training_step_time": 10.273686170578003
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 2.7619614601135254,
      "step": 6301
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 6301,
      "training_step_time": 7.7883429527282715
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 3.131185531616211,
      "step": 6301
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 6301,
      "training_step_time": 9.897367000579834
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 3.137495994567871,
      "step": 6301
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 6301,
      "training_step_time": 9.906234741210938
    },
    {
      "epoch": 2.4036407470703126e-05,
      "model_forward_time": 2.433837413787842,
      "step": 6301
    },
    {
      "epoch": 2.4036407470703126e-05,
      "step": 6301,
      "training_step_time": 10.318910598754883
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 3.1558542251586914,
      "step": 6302
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 6302,
      "training_step_time": 9.933557033538818
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 2.717542886734009,
      "step": 6302
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 6302,
      "training_step_time": 7.74223780632019
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 2.4323577880859375,
      "step": 6302
    },
    {
      "dynamics_loss_avg": 0.05788593702018261,
      "epoch": 2.404022216796875e-05,
      "step": 6302
    },
    {
      "action_loss_avg": 0.028327189781703056,
      "epoch": 2.404022216796875e-05,
      "step": 6302
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 6302,
      "training_step_time": 7.4590232372283936
    },
    {
      "epoch": 2.404022216796875e-05,
      "model_forward_time": 3.1328723430633545,
      "step": 6302
    },
    {
      "epoch": 2.404022216796875e-05,
      "step": 6302,
      "training_step_time": 10.289299726486206
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 2.4476168155670166,
      "step": 6303
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 6303,
      "training_step_time": 7.470266342163086
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 3.126620292663574,
      "step": 6303
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 6303,
      "training_step_time": 9.894563436508179
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 3.419466733932495,
      "step": 6303
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 6303,
      "training_step_time": 10.190377712249756
    },
    {
      "epoch": 2.4044036865234375e-05,
      "model_forward_time": 2.4323527812957764,
      "step": 6303
    },
    {
      "epoch": 2.4044036865234375e-05,
      "step": 6303,
      "training_step_time": 10.289590120315552
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 3.149090528488159,
      "step": 6304
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 6304,
      "training_step_time": 9.92414903640747
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 2.4315402507781982,
      "step": 6304
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 6304,
      "training_step_time": 7.463786840438843
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 2.426217555999756,
      "step": 6304
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 6304,
      "training_step_time": 7.453201055526733
    },
    {
      "epoch": 2.40478515625e-05,
      "model_forward_time": 2.429924726486206,
      "step": 6304
    },
    {
      "epoch": 2.40478515625e-05,
      "step": 6304,
      "training_step_time": 10.237345695495605
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 2.444002389907837,
      "step": 6305
    },
    {
      "dynamics_loss_avg": 0.03762457258999348,
      "epoch": 2.4051666259765624e-05,
      "step": 6305
    },
    {
      "action_loss_avg": 0.017485922621563076,
      "epoch": 2.4051666259765624e-05,
      "step": 6305
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 6305,
      "training_step_time": 7.471102952957153
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 3.13187837600708,
      "step": 6305
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 6305,
      "training_step_time": 9.897480487823486
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 2.427180051803589,
      "step": 6305
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 6305,
      "training_step_time": 7.444628000259399
    },
    {
      "epoch": 2.4051666259765624e-05,
      "model_forward_time": 3.124864101409912,
      "step": 6305
    },
    {
      "epoch": 2.4051666259765624e-05,
      "step": 6305,
      "training_step_time": 10.30225396156311
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 2.442195415496826,
      "step": 6306
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 6306,
      "training_step_time": 7.460475921630859
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 2.6915886402130127,
      "step": 6306
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 6306,
      "training_step_time": 7.712060928344727
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 2.4282283782958984,
      "step": 6306
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 6306,
      "training_step_time": 7.449512004852295
    },
    {
      "epoch": 2.405548095703125e-05,
      "model_forward_time": 3.1272075176239014,
      "step": 6306
    },
    {
      "epoch": 2.405548095703125e-05,
      "step": 6306,
      "training_step_time": 10.296091556549072
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 3.1373062133789062,
      "step": 6307
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 6307,
      "training_step_time": 9.90610957145691
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 3.13130259513855,
      "step": 6307
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 6307,
      "training_step_time": 9.894024848937988
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 3.4102468490600586,
      "step": 6307
    },
    {
      "dynamics_loss_avg": 0.06822564965113997,
      "epoch": 2.4059295654296876e-05,
      "step": 6307
    },
    {
      "action_loss_avg": 0.03987249722704291,
      "epoch": 2.4059295654296876e-05,
      "step": 6307
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 6307,
      "training_step_time": 10.179407358169556
    },
    {
      "epoch": 2.4059295654296876e-05,
      "model_forward_time": 3.1322810649871826,
      "step": 6307
    },
    {
      "epoch": 2.4059295654296876e-05,
      "step": 6307,
      "training_step_time": 10.026888370513916
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 2.4493699073791504,
      "step": 6308
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 6308,
      "training_step_time": 7.4789345264434814
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 2.4321188926696777,
      "step": 6308
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 6308,
      "training_step_time": 7.458843946456909
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 2.4256739616394043,
      "step": 6308
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 6308,
      "training_step_time": 7.449029207229614
    },
    {
      "epoch": 2.40631103515625e-05,
      "model_forward_time": 2.7071139812469482,
      "step": 6308
    },
    {
      "epoch": 2.40631103515625e-05,
      "step": 6308,
      "training_step_time": 10.299686431884766
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 3.1515583992004395,
      "step": 6309
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 6309,
      "training_step_time": 9.918293714523315
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 3.133944034576416,
      "step": 6309
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 6309,
      "training_step_time": 9.896052360534668
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 2.436479330062866,
      "step": 6309
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 6309,
      "training_step_time": 7.465840101242065
    },
    {
      "epoch": 2.4066925048828126e-05,
      "model_forward_time": 3.1348023414611816,
      "step": 6309
    },
    {
      "epoch": 2.4066925048828126e-05,
      "step": 6309,
      "training_step_time": 10.012981176376343
    },
    {
      "epoch": 2.407073974609375e-05,
      "grad_norm": 0.7011648416519165,
      "learning_rate": 1e-05,
      "loss": 0.0866,
      "step": 6310
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 2.4451611042022705,
      "step": 6310
    },
    {
      "dynamics_loss_avg": 0.06076850313693285,
      "epoch": 2.407073974609375e-05,
      "step": 6310
    },
    {
      "action_loss_avg": 0.031779388431459665,
      "epoch": 2.407073974609375e-05,
      "step": 6310
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 6310,
      "training_step_time": 7.740606784820557
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 2.429603338241577,
      "step": 6310
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 6310,
      "training_step_time": 7.447319030761719
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 3.126471519470215,
      "step": 6310
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 6310,
      "training_step_time": 9.888103008270264
    },
    {
      "epoch": 2.407073974609375e-05,
      "model_forward_time": 2.427475690841675,
      "step": 6310
    },
    {
      "epoch": 2.407073974609375e-05,
      "step": 6310,
      "training_step_time": 10.309407949447632
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 2.4484076499938965,
      "step": 6311
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 6311,
      "training_step_time": 7.467698812484741
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 2.427048683166504,
      "step": 6311
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 6311,
      "training_step_time": 7.4409472942352295
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 2.6743106842041016,
      "step": 6311
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 6311,
      "training_step_time": 7.690798282623291
    },
    {
      "epoch": 2.4074554443359375e-05,
      "model_forward_time": 2.428419828414917,
      "step": 6311
    },
    {
      "epoch": 2.4074554443359375e-05,
      "step": 6311,
      "training_step_time": 10.29258918762207
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 3.1716744899749756,
      "step": 6312
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 6312,
      "training_step_time": 9.934069871902466
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 3.131967306137085,
      "step": 6312
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 6312,
      "training_step_time": 9.90092945098877
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 3.1313047409057617,
      "step": 6312
    },
    {
      "dynamics_loss_avg": 0.05846296213567257,
      "epoch": 2.4078369140625e-05,
      "step": 6312
    },
    {
      "action_loss_avg": 0.020087701035663486,
      "epoch": 2.4078369140625e-05,
      "step": 6312
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 6312,
      "training_step_time": 9.904781818389893
    },
    {
      "epoch": 2.4078369140625e-05,
      "model_forward_time": 3.40921950340271,
      "step": 6312
    },
    {
      "epoch": 2.4078369140625e-05,
      "step": 6312,
      "training_step_time": 10.324437618255615
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 2.452104330062866,
      "step": 6313
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 6313,
      "training_step_time": 7.484009504318237
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 3.1320316791534424,
      "step": 6313
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 6313,
      "training_step_time": 9.905517339706421
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 2.429111957550049,
      "step": 6313
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 6313,
      "training_step_time": 7.459468126296997
    },
    {
      "epoch": 2.4082183837890624e-05,
      "model_forward_time": 1.7414357662200928,
      "step": 6313
    },
    {
      "epoch": 2.4082183837890624e-05,
      "step": 6313,
      "training_step_time": 10.014283418655396
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 2.4497578144073486,
      "step": 6314
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 6314,
      "training_step_time": 7.475372791290283
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 3.385599374771118,
      "step": 6314
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 6314,
      "training_step_time": 10.154086112976074
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 3.135903835296631,
      "step": 6314
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 6314,
      "training_step_time": 9.906249046325684
    },
    {
      "epoch": 2.408599853515625e-05,
      "model_forward_time": 1.745185375213623,
      "step": 6314
    },
    {
      "epoch": 2.408599853515625e-05,
      "step": 6314,
      "training_step_time": 10.32572054862976
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 3.1421573162078857,
      "step": 6315
    },
    {
      "dynamics_loss_avg": 0.0529246224090457,
      "epoch": 2.4089813232421876e-05,
      "step": 6315
    },
    {
      "action_loss_avg": 0.02287366627715528,
      "epoch": 2.4089813232421876e-05,
      "step": 6315
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 6315,
      "training_step_time": 9.917209386825562
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 3.1335744857788086,
      "step": 6315
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 6315,
      "training_step_time": 9.90748929977417
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 2.726097822189331,
      "step": 6315
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 6315,
      "training_step_time": 7.7508864402771
    },
    {
      "epoch": 2.4089813232421876e-05,
      "model_forward_time": 3.1266109943389893,
      "step": 6315
    },
    {
      "epoch": 2.4089813232421876e-05,
      "step": 6315,
      "training_step_time": 10.288571119308472
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 3.144209146499634,
      "step": 6316
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 6316,
      "training_step_time": 9.912041664123535
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 2.4312803745269775,
      "step": 6316
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 6316,
      "training_step_time": 7.4576497077941895
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 2.4325854778289795,
      "step": 6316
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 6316,
      "training_step_time": 7.4580559730529785
    },
    {
      "epoch": 2.40936279296875e-05,
      "model_forward_time": 2.711383819580078,
      "step": 6316
    },
    {
      "epoch": 2.40936279296875e-05,
      "step": 6316,
      "training_step_time": 10.031745910644531
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 2.447206974029541,
      "step": 6317
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 6317,
      "training_step_time": 7.474125862121582
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 3.1323907375335693,
      "step": 6317
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 6317,
      "training_step_time": 9.89664363861084
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 3.1311399936676025,
      "step": 6317
    },
    {
      "dynamics_loss_avg": 0.05328539446927607,
      "epoch": 2.4097442626953126e-05,
      "step": 6317
    },
    {
      "action_loss_avg": 0.022206082986667753,
      "epoch": 2.4097442626953126e-05,
      "step": 6317
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 6317,
      "training_step_time": 9.900467157363892
    },
    {
      "epoch": 2.4097442626953126e-05,
      "model_forward_time": 3.1327433586120605,
      "step": 6317
    },
    {
      "epoch": 2.4097442626953126e-05,
      "step": 6317,
      "training_step_time": 10.285796880722046
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 3.16176438331604,
      "step": 6318
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 6318,
      "training_step_time": 9.942596197128296
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 3.406714677810669,
      "step": 6318
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 6318,
      "training_step_time": 10.180829763412476
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 1.741253137588501,
      "step": 6318
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 6318,
      "training_step_time": 5.203557252883911
    },
    {
      "epoch": 2.410125732421875e-05,
      "model_forward_time": 2.431535482406616,
      "step": 6318
    },
    {
      "epoch": 2.410125732421875e-05,
      "step": 6318,
      "training_step_time": 10.27034306526184
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 3.1668813228607178,
      "step": 6319
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 6319,
      "training_step_time": 9.950587749481201
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 3.1329164505004883,
      "step": 6319
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 6319,
      "training_step_time": 9.907179117202759
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 2.7547154426574707,
      "step": 6319
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 6319,
      "training_step_time": 7.782902717590332
    },
    {
      "epoch": 2.4105072021484375e-05,
      "model_forward_time": 3.1316277980804443,
      "step": 6319
    },
    {
      "epoch": 2.4105072021484375e-05,
      "step": 6319,
      "training_step_time": 10.283367395401001
    },
    {
      "epoch": 2.410888671875e-05,
      "grad_norm": 0.4131011664867401,
      "learning_rate": 1e-05,
      "loss": 0.0845,
      "step": 6320
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 3.158393621444702,
      "step": 6320
    },
    {
      "dynamics_loss_avg": 0.06254330705851316,
      "epoch": 2.410888671875e-05,
      "step": 6320
    },
    {
      "action_loss_avg": 0.022672323416918515,
      "epoch": 2.410888671875e-05,
      "step": 6320
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 6320,
      "training_step_time": 9.935772895812988
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 2.430669069290161,
      "step": 6320
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 6320,
      "training_step_time": 7.4576029777526855
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 3.1296448707580566,
      "step": 6320
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 6320,
      "training_step_time": 9.906019926071167
    },
    {
      "epoch": 2.410888671875e-05,
      "model_forward_time": 3.454035758972168,
      "step": 6320
    },
    {
      "epoch": 2.410888671875e-05,
      "step": 6320,
      "training_step_time": 10.325082778930664
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 3.145974636077881,
      "step": 6321
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 6321,
      "training_step_time": 9.92064642906189
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 2.431386947631836,
      "step": 6321
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 6321,
      "training_step_time": 7.456737518310547
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 3.132009506225586,
      "step": 6321
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 6321,
      "training_step_time": 9.90390133857727
    },
    {
      "epoch": 2.4112701416015624e-05,
      "model_forward_time": 3.130815029144287,
      "step": 6321
    },
    {
      "epoch": 2.4112701416015624e-05,
      "step": 6321,
      "training_step_time": 10.29027509689331
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 3.4453141689300537,
      "step": 6322
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 6322,
      "training_step_time": 10.224913120269775
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 3.134373188018799,
      "step": 6322
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 6322,
      "training_step_time": 9.912198066711426
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 2.4315571784973145,
      "step": 6322
    },
    {
      "dynamics_loss_avg": 0.048041352164000274,
      "epoch": 2.411651611328125e-05,
      "step": 6322
    },
    {
      "action_loss_avg": 0.02344516753219068,
      "epoch": 2.411651611328125e-05,
      "step": 6322
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 6322,
      "training_step_time": 7.465942621231079
    },
    {
      "epoch": 2.411651611328125e-05,
      "model_forward_time": 3.1342415809631348,
      "step": 6322
    },
    {
      "epoch": 2.411651611328125e-05,
      "step": 6322,
      "training_step_time": 10.018261194229126
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 3.1826648712158203,
      "step": 6323
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 6323,
      "training_step_time": 9.963658094406128
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 2.7502992153167725,
      "step": 6323
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 6323,
      "training_step_time": 7.778578519821167
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 3.133983850479126,
      "step": 6323
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 6323,
      "training_step_time": 9.908571481704712
    },
    {
      "epoch": 2.4120330810546877e-05,
      "model_forward_time": 3.1318130493164062,
      "step": 6323
    },
    {
      "epoch": 2.4120330810546877e-05,
      "step": 6323,
      "training_step_time": 10.242434024810791
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 2.4441936016082764,
      "step": 6324
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 6324,
      "training_step_time": 7.478766679763794
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 1.7392387390136719,
      "step": 6324
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 6324,
      "training_step_time": 5.199852705001831
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 3.1310927867889404,
      "step": 6324
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 6324,
      "training_step_time": 9.908297538757324
    },
    {
      "epoch": 2.41241455078125e-05,
      "model_forward_time": 2.7123870849609375,
      "step": 6324
    },
    {
      "epoch": 2.41241455078125e-05,
      "step": 6324,
      "training_step_time": 10.336715698242188
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 2.4477083683013916,
      "step": 6325
    },
    {
      "dynamics_loss_avg": 0.07579440772533416,
      "epoch": 2.4127960205078126e-05,
      "step": 6325
    },
    {
      "action_loss_avg": 0.03169857095927,
      "epoch": 2.4127960205078126e-05,
      "step": 6325
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 6325,
      "training_step_time": 7.488967418670654
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 3.1358911991119385,
      "step": 6325
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 6325,
      "training_step_time": 9.913445949554443
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 2.4300084114074707,
      "step": 6325
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 6325,
      "training_step_time": 7.456895589828491
    },
    {
      "epoch": 2.4127960205078126e-05,
      "model_forward_time": 3.1328177452087402,
      "step": 6325
    },
    {
      "epoch": 2.4127960205078126e-05,
      "step": 6325,
      "training_step_time": 10.306381702423096
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 2.7675650119781494,
      "step": 6326
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 6326,
      "training_step_time": 7.796624183654785
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 3.15543794631958,
      "step": 6326
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 6326,
      "training_step_time": 9.930130243301392
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 3.139347553253174,
      "step": 6326
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 6326,
      "training_step_time": 9.91728663444519
    },
    {
      "epoch": 2.413177490234375e-05,
      "model_forward_time": 3.1504526138305664,
      "step": 6326
    },
    {
      "epoch": 2.413177490234375e-05,
      "step": 6326,
      "training_step_time": 10.303274393081665
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 2.4588944911956787,
      "step": 6327
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 6327,
      "training_step_time": 7.4983603954315186
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 3.1408302783966064,
      "step": 6327
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 6327,
      "training_step_time": 9.917707443237305
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 2.742586612701416,
      "step": 6327
    },
    {
      "dynamics_loss_avg": 0.04702141620218754,
      "epoch": 2.4135589599609375e-05,
      "step": 6327
    },
    {
      "action_loss_avg": 0.022088391100987793,
      "epoch": 2.4135589599609375e-05,
      "step": 6327
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 6327,
      "training_step_time": 7.775794506072998
    },
    {
      "epoch": 2.4135589599609375e-05,
      "model_forward_time": 2.442229747772217,
      "step": 6327
    },
    {
      "epoch": 2.4135589599609375e-05,
      "step": 6327,
      "training_step_time": 10.037296533584595
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 3.1615052223205566,
      "step": 6328
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 6328,
      "training_step_time": 9.94410490989685
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 3.1458966732025146,
      "step": 6328
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 6328,
      "training_step_time": 9.931347608566284
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 3.1505167484283447,
      "step": 6328
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 6328,
      "training_step_time": 9.929483413696289
    },
    {
      "epoch": 2.4139404296875e-05,
      "model_forward_time": 3.4724457263946533,
      "step": 6328
    },
    {
      "epoch": 2.4139404296875e-05,
      "step": 6328,
      "training_step_time": 10.346197605133057
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 2.449988842010498,
      "step": 6329
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 6329,
      "training_step_time": 7.490360260009766
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 2.4383554458618164,
      "step": 6329
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 6329,
      "training_step_time": 7.46812891960144
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 2.434002161026001,
      "step": 6329
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 6329,
      "training_step_time": 7.464610815048218
    },
    {
      "epoch": 2.4143218994140624e-05,
      "model_forward_time": 2.44079852104187,
      "step": 6329
    },
    {
      "epoch": 2.4143218994140624e-05,
      "step": 6329,
      "training_step_time": 10.015312671661377
    },
    {
      "epoch": 2.414703369140625e-05,
      "grad_norm": 0.6278831362724304,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 6330
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 3.5032522678375244,
      "step": 6330
    },
    {
      "dynamics_loss_avg": 0.06691713184118271,
      "epoch": 2.414703369140625e-05,
      "step": 6330
    },
    {
      "action_loss_avg": 0.024812314473092557,
      "epoch": 2.414703369140625e-05,
      "step": 6330
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 6330,
      "training_step_time": 10.278011560440063
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 2.4382901191711426,
      "step": 6330
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 6330,
      "training_step_time": 7.468237400054932
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 3.14679217338562,
      "step": 6330
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 6330,
      "training_step_time": 9.92192792892456
    },
    {
      "epoch": 2.414703369140625e-05,
      "model_forward_time": 3.1413474082946777,
      "step": 6330
    },
    {
      "epoch": 2.414703369140625e-05,
      "step": 6330,
      "training_step_time": 10.320339679718018
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 2.4543957710266113,
      "step": 6331
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 6331,
      "training_step_time": 7.4834184646606445
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 3.420184373855591,
      "step": 6331
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 6331,
      "training_step_time": 10.191096067428589
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 3.1427102088928223,
      "step": 6331
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 6331,
      "training_step_time": 9.909296751022339
    },
    {
      "epoch": 2.4150848388671877e-05,
      "model_forward_time": 2.4410665035247803,
      "step": 6331
    },
    {
      "epoch": 2.4150848388671877e-05,
      "step": 6331,
      "training_step_time": 10.299495458602905
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 3.1489014625549316,
      "step": 6332
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 6332,
      "training_step_time": 9.925812721252441
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 2.437394618988037,
      "step": 6332
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 6332,
      "training_step_time": 7.460499286651611
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 2.4381000995635986,
      "step": 6332
    },
    {
      "dynamics_loss_avg": 0.07300836257636548,
      "epoch": 2.41546630859375e-05,
      "step": 6332
    },
    {
      "action_loss_avg": 0.029570358991622924,
      "epoch": 2.41546630859375e-05,
      "step": 6332
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 6332,
      "training_step_time": 7.460679292678833
    },
    {
      "epoch": 2.41546630859375e-05,
      "model_forward_time": 3.4251129627227783,
      "step": 6332
    },
    {
      "epoch": 2.41546630859375e-05,
      "step": 6332,
      "training_step_time": 10.27992582321167
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 2.4495065212249756,
      "step": 6333
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 6333,
      "training_step_time": 7.478838205337524
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 2.4433348178863525,
      "step": 6333
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 6333,
      "training_step_time": 7.4685304164886475
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 2.432147264480591,
      "step": 6333
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 6333,
      "training_step_time": 7.4517035484313965
    },
    {
      "epoch": 2.4158477783203126e-05,
      "model_forward_time": 3.1393537521362305,
      "step": 6333
    },
    {
      "epoch": 2.4158477783203126e-05,
      "step": 6333,
      "training_step_time": 10.011462450027466
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 3.418099880218506,
      "step": 6334
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 6334,
      "training_step_time": 10.178683757781982
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 3.1418063640594482,
      "step": 6334
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 6334,
      "training_step_time": 9.9063880443573
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 2.4357688426971436,
      "step": 6334
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 6334,
      "training_step_time": 7.4550862312316895
    },
    {
      "epoch": 2.416229248046875e-05,
      "model_forward_time": 3.1482343673706055,
      "step": 6334
    },
    {
      "epoch": 2.416229248046875e-05,
      "step": 6334,
      "training_step_time": 10.284486293792725
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 2.4416303634643555,
      "step": 6335
    },
    {
      "dynamics_loss_avg": 0.05520454347133637,
      "epoch": 2.4166107177734375e-05,
      "step": 6335
    },
    {
      "action_loss_avg": 0.023544747568666934,
      "epoch": 2.4166107177734375e-05,
      "step": 6335
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 6335,
      "training_step_time": 7.459944486618042
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 2.001777172088623,
      "step": 6335
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 6335,
      "training_step_time": 5.447654485702515
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 2.435436248779297,
      "step": 6335
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 6335,
      "training_step_time": 7.452594995498657
    },
    {
      "epoch": 2.4166107177734375e-05,
      "model_forward_time": 3.13531231880188,
      "step": 6335
    },
    {
      "epoch": 2.4166107177734375e-05,
      "step": 6335,
      "training_step_time": 10.010436534881592
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 2.4487805366516113,
      "step": 6336
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 6336,
      "training_step_time": 7.489516019821167
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 1.741910457611084,
      "step": 6336
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 6336,
      "training_step_time": 5.198467254638672
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 3.1326208114624023,
      "step": 6336
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 6336,
      "training_step_time": 9.90175199508667
    },
    {
      "epoch": 2.4169921875e-05,
      "model_forward_time": 2.753997325897217,
      "step": 6336
    },
    {
      "epoch": 2.4169921875e-05,
      "step": 6336,
      "training_step_time": 10.300396203994751
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 2.4561378955841064,
      "step": 6337
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 6337,
      "training_step_time": 7.485834360122681
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 3.135059356689453,
      "step": 6337
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 6337,
      "training_step_time": 9.910327434539795
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 1.7495801448822021,
      "step": 6337
    },
    {
      "dynamics_loss_avg": 0.04283265871927142,
      "epoch": 2.4173736572265624e-05,
      "step": 6337
    },
    {
      "action_loss_avg": 0.016718672297429293,
      "epoch": 2.4173736572265624e-05,
      "step": 6337
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 6337,
      "training_step_time": 5.217602252960205
    },
    {
      "epoch": 2.4173736572265624e-05,
      "model_forward_time": 2.430713653564453,
      "step": 6337
    },
    {
      "epoch": 2.4173736572265624e-05,
      "step": 6337,
      "training_step_time": 10.306750535964966
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 3.1539852619171143,
      "step": 6338
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 6338,
      "training_step_time": 10.254886627197266
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 3.1341822147369385,
      "step": 6338
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 6338,
      "training_step_time": 9.900602340698242
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 2.4463722705841064,
      "step": 6338
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 6338,
      "training_step_time": 7.477967977523804
    },
    {
      "epoch": 2.417755126953125e-05,
      "model_forward_time": 2.4304652214050293,
      "step": 6338
    },
    {
      "epoch": 2.417755126953125e-05,
      "step": 6338,
      "training_step_time": 10.021615982055664
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 2.4784514904022217,
      "step": 6339
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 6339,
      "training_step_time": 7.510792016983032
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 2.4394445419311523,
      "step": 6339
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 6339,
      "training_step_time": 7.465183734893799
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 3.472892999649048,
      "step": 6339
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 6339,
      "training_step_time": 10.236606121063232
    },
    {
      "epoch": 2.4181365966796877e-05,
      "model_forward_time": 3.1342644691467285,
      "step": 6339
    },
    {
      "epoch": 2.4181365966796877e-05,
      "step": 6339,
      "training_step_time": 10.274862289428711
    },
    {
      "epoch": 2.41851806640625e-05,
      "grad_norm": 0.4770094156265259,
      "learning_rate": 1e-05,
      "loss": 0.0869,
      "step": 6340
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 3.1429126262664795,
      "step": 6340
    },
    {
      "dynamics_loss_avg": 0.06915044970810413,
      "epoch": 2.41851806640625e-05,
      "step": 6340
    },
    {
      "action_loss_avg": 0.02809035051614046,
      "epoch": 2.41851806640625e-05,
      "step": 6340
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 6340,
      "training_step_time": 9.918458223342896
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 3.1426918506622314,
      "step": 6340
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 6340,
      "training_step_time": 9.913338422775269
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 3.1399168968200684,
      "step": 6340
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 6340,
      "training_step_time": 9.915253400802612
    },
    {
      "epoch": 2.41851806640625e-05,
      "model_forward_time": 3.4031150341033936,
      "step": 6340
    },
    {
      "epoch": 2.41851806640625e-05,
      "step": 6340,
      "training_step_time": 10.284645318984985
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 3.1553382873535156,
      "step": 6341
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 6341,
      "training_step_time": 9.929341316223145
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 2.4322092533111572,
      "step": 6341
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 6341,
      "training_step_time": 7.459781646728516
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 2.4280922412872314,
      "step": 6341
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 6341,
      "training_step_time": 7.454264402389526
    },
    {
      "epoch": 2.4188995361328126e-05,
      "model_forward_time": 3.140571117401123,
      "step": 6341
    },
    {
      "epoch": 2.4188995361328126e-05,
      "step": 6341,
      "training_step_time": 10.301886320114136
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 2.80863881111145,
      "step": 6342
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 6342,
      "training_step_time": 7.836252689361572
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 3.135936975479126,
      "step": 6342
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 6342,
      "training_step_time": 9.897967338562012
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 3.12624192237854,
      "step": 6342
    },
    {
      "dynamics_loss_avg": 0.05691201835870743,
      "epoch": 2.419281005859375e-05,
      "step": 6342
    },
    {
      "action_loss_avg": 0.02326351273804903,
      "epoch": 2.419281005859375e-05,
      "step": 6342
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 6342,
      "training_step_time": 9.890104055404663
    },
    {
      "epoch": 2.419281005859375e-05,
      "model_forward_time": 3.1271185874938965,
      "step": 6342
    },
    {
      "epoch": 2.419281005859375e-05,
      "step": 6342,
      "training_step_time": 10.281604290008545
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 3.1582412719726562,
      "step": 6343
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 6343,
      "training_step_time": 9.927284240722656
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 3.4300379753112793,
      "step": 6343
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 6343,
      "training_step_time": 10.198080062866211
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 2.4282150268554688,
      "step": 6343
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 6343,
      "training_step_time": 7.453056573867798
    },
    {
      "epoch": 2.4196624755859375e-05,
      "model_forward_time": 3.130488157272339,
      "step": 6343
    },
    {
      "epoch": 2.4196624755859375e-05,
      "step": 6343,
      "training_step_time": 10.0128812789917
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 2.4481849670410156,
      "step": 6344
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 6344,
      "training_step_time": 7.486762523651123
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 2.4344232082366943,
      "step": 6344
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 6344,
      "training_step_time": 7.462430715560913
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 3.410162925720215,
      "step": 6344
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 6344,
      "training_step_time": 10.179386377334595
    },
    {
      "epoch": 2.4200439453125e-05,
      "model_forward_time": 3.139569044113159,
      "step": 6344
    },
    {
      "epoch": 2.4200439453125e-05,
      "step": 6344,
      "training_step_time": 10.01475214958191
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 3.1582627296447754,
      "step": 6345
    },
    {
      "dynamics_loss_avg": 0.05547707062214613,
      "epoch": 2.4204254150390624e-05,
      "step": 6345
    },
    {
      "action_loss_avg": 0.018099309504032136,
      "epoch": 2.4204254150390624e-05,
      "step": 6345
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 6345,
      "training_step_time": 9.93335771560669
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 3.1392312049865723,
      "step": 6345
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 6345,
      "training_step_time": 9.908708810806274
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 3.1382675170898438,
      "step": 6345
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 6345,
      "training_step_time": 9.903390407562256
    },
    {
      "epoch": 2.4204254150390624e-05,
      "model_forward_time": 2.7483866214752197,
      "step": 6345
    },
    {
      "epoch": 2.4204254150390624e-05,
      "step": 6345,
      "training_step_time": 10.30794644355774
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 3.1638011932373047,
      "step": 6346
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 6346,
      "training_step_time": 9.951438426971436
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 2.4340755939483643,
      "step": 6346
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 6346,
      "training_step_time": 7.463502645492554
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 2.4345314502716064,
      "step": 6346
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 6346,
      "training_step_time": 7.4677698612213135
    },
    {
      "epoch": 2.420806884765625e-05,
      "model_forward_time": 2.430997848510742,
      "step": 6346
    },
    {
      "epoch": 2.420806884765625e-05,
      "step": 6346,
      "training_step_time": 10.303271293640137
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 1.7865889072418213,
      "step": 6347
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 6347,
      "training_step_time": 5.246871471405029
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 3.3949575424194336,
      "step": 6347
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 6347,
      "training_step_time": 10.16952657699585
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 3.141857385635376,
      "step": 6347
    },
    {
      "dynamics_loss_avg": 0.05812493283301592,
      "epoch": 2.4211883544921877e-05,
      "step": 6347
    },
    {
      "action_loss_avg": 0.030357481352984904,
      "epoch": 2.4211883544921877e-05,
      "step": 6347
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 6347,
      "training_step_time": 9.920276403427124
    },
    {
      "epoch": 2.4211883544921877e-05,
      "model_forward_time": 3.141566276550293,
      "step": 6347
    },
    {
      "epoch": 2.4211883544921877e-05,
      "step": 6347,
      "training_step_time": 10.286221265792847
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 2.4594032764434814,
      "step": 6348
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 6348,
      "training_step_time": 7.4959471225738525
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 2.431698799133301,
      "step": 6348
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 6348,
      "training_step_time": 7.4621429443359375
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 2.699347496032715,
      "step": 6348
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 6348,
      "training_step_time": 7.728591680526733
    },
    {
      "epoch": 2.42156982421875e-05,
      "model_forward_time": 2.4372100830078125,
      "step": 6348
    },
    {
      "epoch": 2.42156982421875e-05,
      "step": 6348,
      "training_step_time": 10.019624710083008
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 3.1558656692504883,
      "step": 6349
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 6349,
      "training_step_time": 9.930553436279297
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 2.434678077697754,
      "step": 6349
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 6349,
      "training_step_time": 7.459063291549683
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 2.4355311393737793,
      "step": 6349
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 6349,
      "training_step_time": 7.4591453075408936
    },
    {
      "epoch": 2.4219512939453126e-05,
      "model_forward_time": 3.4333648681640625,
      "step": 6349
    },
    {
      "epoch": 2.4219512939453126e-05,
      "step": 6349,
      "training_step_time": 10.297896146774292
    },
    {
      "epoch": 2.422332763671875e-05,
      "grad_norm": 0.4688355326652527,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 6350
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 3.1739251613616943,
      "step": 6350
    },
    {
      "dynamics_loss_avg": 0.05813237633556127,
      "epoch": 2.422332763671875e-05,
      "step": 6350
    },
    {
      "action_loss_avg": 0.023194333212450145,
      "epoch": 2.422332763671875e-05,
      "step": 6350
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 6350,
      "training_step_time": 9.955396175384521
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 3.1445863246917725,
      "step": 6350
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 6350,
      "training_step_time": 9.91484546661377
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 3.135927200317383,
      "step": 6350
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 6350,
      "training_step_time": 9.908515453338623
    },
    {
      "epoch": 2.422332763671875e-05,
      "model_forward_time": 3.141173839569092,
      "step": 6350
    },
    {
      "epoch": 2.422332763671875e-05,
      "step": 6350,
      "training_step_time": 10.282190799713135
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 2.7784457206726074,
      "step": 6351
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 6351,
      "training_step_time": 7.806157350540161
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 3.1344573497772217,
      "step": 6351
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 6351,
      "training_step_time": 9.906805038452148
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 2.4500250816345215,
      "step": 6351
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 6351,
      "training_step_time": 7.487301349639893
    },
    {
      "epoch": 2.4227142333984375e-05,
      "model_forward_time": 3.132521629333496,
      "step": 6351
    },
    {
      "epoch": 2.4227142333984375e-05,
      "step": 6351,
      "training_step_time": 10.329905986785889
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 3.1453847885131836,
      "step": 6352
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 6352,
      "training_step_time": 9.927411079406738
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 3.4048376083374023,
      "step": 6352
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 6352,
      "training_step_time": 10.180779933929443
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 2.4285173416137695,
      "step": 6352
    },
    {
      "dynamics_loss_avg": 0.06338913943618536,
      "epoch": 2.423095703125e-05,
      "step": 6352
    },
    {
      "action_loss_avg": 0.026098337210714816,
      "epoch": 2.423095703125e-05,
      "step": 6352
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 6352,
      "training_step_time": 7.461931228637695
    },
    {
      "epoch": 2.423095703125e-05,
      "model_forward_time": 3.1284923553466797,
      "step": 6352
    },
    {
      "epoch": 2.423095703125e-05,
      "step": 6352,
      "training_step_time": 10.33291482925415
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 2.446486473083496,
      "step": 6353
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 6353,
      "training_step_time": 7.479398012161255
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 2.426955461502075,
      "step": 6353
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 6353,
      "training_step_time": 7.455673694610596
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 2.695728063583374,
      "step": 6353
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 6353,
      "training_step_time": 7.7223920822143555
    },
    {
      "epoch": 2.4234771728515624e-05,
      "model_forward_time": 2.4269819259643555,
      "step": 6353
    },
    {
      "epoch": 2.4234771728515624e-05,
      "step": 6353,
      "training_step_time": 10.017059803009033
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 2.4383115768432617,
      "step": 6354
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 6354,
      "training_step_time": 7.465466022491455
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 3.132664918899536,
      "step": 6354
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 6354,
      "training_step_time": 9.913963079452515
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 3.135671615600586,
      "step": 6354
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 6354,
      "training_step_time": 9.913931846618652
    },
    {
      "epoch": 2.423858642578125e-05,
      "model_forward_time": 3.135817766189575,
      "step": 6354
    },
    {
      "epoch": 2.423858642578125e-05,
      "step": 6354,
      "training_step_time": 10.021703958511353
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 2.7677178382873535,
      "step": 6355
    },
    {
      "dynamics_loss_avg": 0.07172930175438523,
      "epoch": 2.4242401123046877e-05,
      "step": 6355
    },
    {
      "action_loss_avg": 0.0239130049245432,
      "epoch": 2.4242401123046877e-05,
      "step": 6355
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 6355,
      "training_step_time": 7.8070361614227295
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 3.1345608234405518,
      "step": 6355
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 6355,
      "training_step_time": 9.917799234390259
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 3.1404757499694824,
      "step": 6355
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 6355,
      "training_step_time": 9.92628788948059
    },
    {
      "epoch": 2.4242401123046877e-05,
      "model_forward_time": 2.438554286956787,
      "step": 6355
    },
    {
      "epoch": 2.4242401123046877e-05,
      "step": 6355,
      "training_step_time": 10.367095708847046
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 2.4648115634918213,
      "step": 6356
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 6356,
      "training_step_time": 7.519201040267944
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 2.734954595565796,
      "step": 6356
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 6356,
      "training_step_time": 7.770338773727417
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 2.4347898960113525,
      "step": 6356
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 6356,
      "training_step_time": 7.472670316696167
    },
    {
      "epoch": 2.42462158203125e-05,
      "model_forward_time": 2.4344913959503174,
      "step": 6356
    },
    {
      "epoch": 2.42462158203125e-05,
      "step": 6356,
      "training_step_time": 10.2636137008667
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 2.44340443611145,
      "step": 6357
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 6357,
      "training_step_time": 7.478663444519043
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 2.427455186843872,
      "step": 6357
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 6357,
      "training_step_time": 7.4523093700408936
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 3.4214463233947754,
      "step": 6357
    },
    {
      "dynamics_loss_avg": 0.04407316781580448,
      "epoch": 2.4250030517578126e-05,
      "step": 6357
    },
    {
      "action_loss_avg": 0.019752277340739966,
      "epoch": 2.4250030517578126e-05,
      "step": 6357
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 6357,
      "training_step_time": 10.192544937133789
    },
    {
      "epoch": 2.4250030517578126e-05,
      "model_forward_time": 2.4307568073272705,
      "step": 6357
    },
    {
      "epoch": 2.4250030517578126e-05,
      "step": 6357,
      "training_step_time": 10.303284883499146
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 2.4416964054107666,
      "step": 6358
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 6358,
      "training_step_time": 7.465129375457764
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 2.4291157722473145,
      "step": 6358
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 6358,
      "training_step_time": 7.455588340759277
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 3.127596616744995,
      "step": 6358
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 6358,
      "training_step_time": 9.895863771438599
    },
    {
      "epoch": 2.425384521484375e-05,
      "model_forward_time": 2.4311418533325195,
      "step": 6358
    },
    {
      "epoch": 2.425384521484375e-05,
      "step": 6358,
      "training_step_time": 10.301578283309937
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 3.44368577003479,
      "step": 6359
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 6359,
      "training_step_time": 10.2162606716156
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 2.4265003204345703,
      "step": 6359
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 6359,
      "training_step_time": 7.454969882965088
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 3.132582426071167,
      "step": 6359
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 6359,
      "training_step_time": 9.909565687179565
    },
    {
      "epoch": 2.4257659912109375e-05,
      "model_forward_time": 3.136601209640503,
      "step": 6359
    },
    {
      "epoch": 2.4257659912109375e-05,
      "step": 6359,
      "training_step_time": 10.022968769073486
    },
    {
      "epoch": 2.4261474609375e-05,
      "grad_norm": 0.42648935317993164,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 6360
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 3.1620781421661377,
      "step": 6360
    },
    {
      "dynamics_loss_avg": 0.05697639845311642,
      "epoch": 2.4261474609375e-05,
      "step": 6360
    },
    {
      "action_loss_avg": 0.02746027698740363,
      "epoch": 2.4261474609375e-05,
      "step": 6360
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 6360,
      "training_step_time": 9.949082374572754
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 2.703380584716797,
      "step": 6360
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 6360,
      "training_step_time": 7.73645281791687
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 2.432633638381958,
      "step": 6360
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 6360,
      "training_step_time": 7.4663755893707275
    },
    {
      "epoch": 2.4261474609375e-05,
      "model_forward_time": 2.4333252906799316,
      "step": 6360
    },
    {
      "epoch": 2.4261474609375e-05,
      "step": 6360,
      "training_step_time": 10.267652750015259
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 3.169102191925049,
      "step": 6361
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 6361,
      "training_step_time": 9.943941354751587
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 2.4281275272369385,
      "step": 6361
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 6361,
      "training_step_time": 7.463166952133179
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 3.3869571685791016,
      "step": 6361
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 6361,
      "training_step_time": 10.155856132507324
    },
    {
      "epoch": 2.4265289306640624e-05,
      "model_forward_time": 3.135274648666382,
      "step": 6361
    },
    {
      "epoch": 2.4265289306640624e-05,
      "step": 6361,
      "training_step_time": 10.257688283920288
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 3.1449646949768066,
      "step": 6362
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 6362,
      "training_step_time": 9.922137975692749
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 2.434647798538208,
      "step": 6362
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 6362,
      "training_step_time": 7.460882186889648
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 3.1323301792144775,
      "step": 6362
    },
    {
      "dynamics_loss_avg": 0.056979725882411,
      "epoch": 2.426910400390625e-05,
      "step": 6362
    },
    {
      "action_loss_avg": 0.018990937853232026,
      "epoch": 2.426910400390625e-05,
      "step": 6362
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 6362,
      "training_step_time": 9.907028436660767
    },
    {
      "epoch": 2.426910400390625e-05,
      "model_forward_time": 3.4257984161376953,
      "step": 6362
    },
    {
      "epoch": 2.426910400390625e-05,
      "step": 6362,
      "training_step_time": 10.291121244430542
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 3.1489787101745605,
      "step": 6363
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 6363,
      "training_step_time": 9.923666715621948
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 2.4299724102020264,
      "step": 6363
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 6363,
      "training_step_time": 7.459233522415161
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 2.431488037109375,
      "step": 6363
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 6363,
      "training_step_time": 7.459091424942017
    },
    {
      "epoch": 2.4272918701171877e-05,
      "model_forward_time": 2.4271626472473145,
      "step": 6363
    },
    {
      "epoch": 2.4272918701171877e-05,
      "step": 6363,
      "training_step_time": 10.322931289672852
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 2.445718765258789,
      "step": 6364
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 6364,
      "training_step_time": 7.778571128845215
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 3.133975028991699,
      "step": 6364
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 6364,
      "training_step_time": 9.908585786819458
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 3.138364791870117,
      "step": 6364
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 6364,
      "training_step_time": 9.918781757354736
    },
    {
      "epoch": 2.42767333984375e-05,
      "model_forward_time": 3.134824752807617,
      "step": 6364
    },
    {
      "epoch": 2.42767333984375e-05,
      "step": 6364,
      "training_step_time": 10.028438806533813
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 1.7629015445709229,
      "step": 6365
    },
    {
      "dynamics_loss_avg": 0.054556605964899064,
      "epoch": 2.4280548095703126e-05,
      "step": 6365
    },
    {
      "action_loss_avg": 0.028761530620977284,
      "epoch": 2.4280548095703126e-05,
      "step": 6365
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 6365,
      "training_step_time": 5.229895114898682
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 2.4292216300964355,
      "step": 6365
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 6365,
      "training_step_time": 7.457498073577881
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 3.4168989658355713,
      "step": 6365
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 6365,
      "training_step_time": 10.190799236297607
    },
    {
      "epoch": 2.4280548095703126e-05,
      "model_forward_time": 3.134812355041504,
      "step": 6365
    },
    {
      "epoch": 2.4280548095703126e-05,
      "step": 6365,
      "training_step_time": 10.274052381515503
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 2.4647250175476074,
      "step": 6366
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 6366,
      "training_step_time": 7.499129056930542
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 2.432243585586548,
      "step": 6366
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 6366,
      "training_step_time": 7.466681718826294
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 3.127549648284912,
      "step": 6366
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 6366,
      "training_step_time": 9.902767658233643
    },
    {
      "epoch": 2.428436279296875e-05,
      "model_forward_time": 3.3825244903564453,
      "step": 6366
    },
    {
      "epoch": 2.428436279296875e-05,
      "step": 6366,
      "training_step_time": 10.270096063613892
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 3.15067458152771,
      "step": 6367
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 6367,
      "training_step_time": 9.924426555633545
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 3.135887384414673,
      "step": 6367
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 6367,
      "training_step_time": 9.908360481262207
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 3.132988214492798,
      "step": 6367
    },
    {
      "dynamics_loss_avg": 0.06551381852477789,
      "epoch": 2.4288177490234375e-05,
      "step": 6367
    },
    {
      "action_loss_avg": 0.03513716300949454,
      "epoch": 2.4288177490234375e-05,
      "step": 6367
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 6367,
      "training_step_time": 9.913513898849487
    },
    {
      "epoch": 2.4288177490234375e-05,
      "model_forward_time": 3.136997699737549,
      "step": 6367
    },
    {
      "epoch": 2.4288177490234375e-05,
      "step": 6367,
      "training_step_time": 10.307876825332642
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 2.7749810218811035,
      "step": 6368
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 6368,
      "training_step_time": 7.808258056640625
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 1.744861125946045,
      "step": 6368
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 6368,
      "training_step_time": 5.210813522338867
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 2.4337692260742188,
      "step": 6368
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 6368,
      "training_step_time": 7.465490341186523
    },
    {
      "epoch": 2.42919921875e-05,
      "model_forward_time": 3.1383490562438965,
      "step": 6368
    },
    {
      "epoch": 2.42919921875e-05,
      "step": 6368,
      "training_step_time": 10.296841382980347
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 2.4492197036743164,
      "step": 6369
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 6369,
      "training_step_time": 7.485557556152344
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 3.4157607555389404,
      "step": 6369
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 6369,
      "training_step_time": 10.190839290618896
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 1.7446496486663818,
      "step": 6369
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 6369,
      "training_step_time": 5.205562353134155
    },
    {
      "epoch": 2.4295806884765624e-05,
      "model_forward_time": 2.4309186935424805,
      "step": 6369
    },
    {
      "epoch": 2.4295806884765624e-05,
      "step": 6369,
      "training_step_time": 10.29245901107788
    },
    {
      "epoch": 2.429962158203125e-05,
      "grad_norm": 0.5963994860649109,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 6370
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 3.1537230014801025,
      "step": 6370
    },
    {
      "dynamics_loss_avg": 0.06484974846243859,
      "epoch": 2.429962158203125e-05,
      "step": 6370
    },
    {
      "action_loss_avg": 0.022515100613236427,
      "epoch": 2.429962158203125e-05,
      "step": 6370
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 6370,
      "training_step_time": 9.937088966369629
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 3.136603593826294,
      "step": 6370
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 6370,
      "training_step_time": 9.916106224060059
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 2.4368085861206055,
      "step": 6370
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 6370,
      "training_step_time": 7.4722678661346436
    },
    {
      "epoch": 2.429962158203125e-05,
      "model_forward_time": 3.415051221847534,
      "step": 6370
    },
    {
      "epoch": 2.429962158203125e-05,
      "step": 6370,
      "training_step_time": 10.282678127288818
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 3.150200366973877,
      "step": 6371
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 6371,
      "training_step_time": 9.936279296875
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 3.1421191692352295,
      "step": 6371
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 6371,
      "training_step_time": 9.919211387634277
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 2.435286045074463,
      "step": 6371
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 6371,
      "training_step_time": 7.4616124629974365
    },
    {
      "epoch": 2.4303436279296877e-05,
      "model_forward_time": 3.1272401809692383,
      "step": 6371
    },
    {
      "epoch": 2.4303436279296877e-05,
      "step": 6371,
      "training_step_time": 10.249975442886353
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 2.769411325454712,
      "step": 6372
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 6372,
      "training_step_time": 7.796709775924683
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 2.430067539215088,
      "step": 6372
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 6372,
      "training_step_time": 7.457464218139648
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 3.136446714401245,
      "step": 6372
    },
    {
      "dynamics_loss_avg": 0.07415878847241401,
      "epoch": 2.43072509765625e-05,
      "step": 6372
    },
    {
      "action_loss_avg": 0.028385291900485755,
      "epoch": 2.43072509765625e-05,
      "step": 6372
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 6372,
      "training_step_time": 9.91710638999939
    },
    {
      "epoch": 2.43072509765625e-05,
      "model_forward_time": 2.434758186340332,
      "step": 6372
    },
    {
      "epoch": 2.43072509765625e-05,
      "step": 6372,
      "training_step_time": 10.285460233688354
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 3.1478824615478516,
      "step": 6373
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 6373,
      "training_step_time": 9.93278956413269
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 3.4088315963745117,
      "step": 6373
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 6373,
      "training_step_time": 10.187005996704102
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 2.4384686946868896,
      "step": 6373
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 6373,
      "training_step_time": 7.474486827850342
    },
    {
      "epoch": 2.4311065673828126e-05,
      "model_forward_time": 3.1323564052581787,
      "step": 6373
    },
    {
      "epoch": 2.4311065673828126e-05,
      "step": 6373,
      "training_step_time": 10.321530103683472
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 2.450526237487793,
      "step": 6374
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 6374,
      "training_step_time": 7.4878249168396
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 2.4313528537750244,
      "step": 6374
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 6374,
      "training_step_time": 7.460049152374268
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 3.12898588180542,
      "step": 6374
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 6374,
      "training_step_time": 10.17080020904541
    },
    {
      "epoch": 2.431488037109375e-05,
      "model_forward_time": 3.131497383117676,
      "step": 6374
    },
    {
      "epoch": 2.431488037109375e-05,
      "step": 6374,
      "training_step_time": 10.328927278518677
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 2.4725115299224854,
      "step": 6375
    },
    {
      "dynamics_loss_avg": 0.04059681324288249,
      "epoch": 2.4318695068359375e-05,
      "step": 6375
    },
    {
      "action_loss_avg": 0.014256175002083182,
      "epoch": 2.4318695068359375e-05,
      "step": 6375
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 6375,
      "training_step_time": 7.500901699066162
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 3.1330509185791016,
      "step": 6375
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 6375,
      "training_step_time": 9.904016733169556
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 3.1294984817504883,
      "step": 6375
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 6375,
      "training_step_time": 9.900477886199951
    },
    {
      "epoch": 2.4318695068359375e-05,
      "model_forward_time": 3.131760597229004,
      "step": 6375
    },
    {
      "epoch": 2.4318695068359375e-05,
      "step": 6375,
      "training_step_time": 10.27606749534607
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 3.1603167057037354,
      "step": 6376
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 6376,
      "training_step_time": 9.938868522644043
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 2.4322147369384766,
      "step": 6376
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 6376,
      "training_step_time": 7.465728044509888
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 3.130856513977051,
      "step": 6376
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 6376,
      "training_step_time": 9.908220291137695
    },
    {
      "epoch": 2.4322509765625e-05,
      "model_forward_time": 3.140124797821045,
      "step": 6376
    },
    {
      "epoch": 2.4322509765625e-05,
      "step": 6376,
      "training_step_time": 10.327243328094482
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 3.1497416496276855,
      "step": 6377
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 6377,
      "training_step_time": 10.225355863571167
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 3.1392009258270264,
      "step": 6377
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 6377,
      "training_step_time": 9.923064231872559
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 2.437603235244751,
      "step": 6377
    },
    {
      "dynamics_loss_avg": 0.06401964575052262,
      "epoch": 2.4326324462890624e-05,
      "step": 6377
    },
    {
      "action_loss_avg": 0.03258806141093373,
      "epoch": 2.4326324462890624e-05,
      "step": 6377
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 6377,
      "training_step_time": 7.478161573410034
    },
    {
      "epoch": 2.4326324462890624e-05,
      "model_forward_time": 3.1377856731414795,
      "step": 6377
    },
    {
      "epoch": 2.4326324462890624e-05,
      "step": 6377,
      "training_step_time": 10.317610263824463
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 3.158446788787842,
      "step": 6378
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 6378,
      "training_step_time": 9.943446159362793
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 3.135706901550293,
      "step": 6378
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 6378,
      "training_step_time": 10.188548564910889
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 2.4337286949157715,
      "step": 6378
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 6378,
      "training_step_time": 7.466584920883179
    },
    {
      "epoch": 2.433013916015625e-05,
      "model_forward_time": 3.1344900131225586,
      "step": 6378
    },
    {
      "epoch": 2.433013916015625e-05,
      "step": 6378,
      "training_step_time": 10.285744905471802
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 2.4504764080047607,
      "step": 6379
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 6379,
      "training_step_time": 7.4874045848846436
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 2.428884744644165,
      "step": 6379
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 6379,
      "training_step_time": 7.458567380905151
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 3.1376214027404785,
      "step": 6379
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 6379,
      "training_step_time": 9.921056270599365
    },
    {
      "epoch": 2.4333953857421873e-05,
      "model_forward_time": 2.7529854774475098,
      "step": 6379
    },
    {
      "epoch": 2.4333953857421873e-05,
      "step": 6379,
      "training_step_time": 10.309800863265991
    },
    {
      "epoch": 2.43377685546875e-05,
      "grad_norm": 0.33362579345703125,
      "learning_rate": 1e-05,
      "loss": 0.0874,
      "step": 6380
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 3.150742292404175,
      "step": 6380
    },
    {
      "dynamics_loss_avg": 0.055223706550896166,
      "epoch": 2.43377685546875e-05,
      "step": 6380
    },
    {
      "action_loss_avg": 0.021767979394644497,
      "epoch": 2.43377685546875e-05,
      "step": 6380
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 6380,
      "training_step_time": 9.9298415184021
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 2.4351022243499756,
      "step": 6380
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 6380,
      "training_step_time": 7.4707255363464355
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 3.138653516769409,
      "step": 6380
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 6380,
      "training_step_time": 9.920518398284912
    },
    {
      "epoch": 2.43377685546875e-05,
      "model_forward_time": 2.4375553131103516,
      "step": 6380
    },
    {
      "epoch": 2.43377685546875e-05,
      "step": 6380,
      "training_step_time": 10.29174017906189
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 3.4956047534942627,
      "step": 6381
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 6381,
      "training_step_time": 10.276714324951172
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 3.137934446334839,
      "step": 6381
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 6381,
      "training_step_time": 9.906864404678345
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 2.4328725337982178,
      "step": 6381
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 6381,
      "training_step_time": 7.466594457626343
    },
    {
      "epoch": 2.4341583251953126e-05,
      "model_forward_time": 2.432802438735962,
      "step": 6381
    },
    {
      "epoch": 2.4341583251953126e-05,
      "step": 6381,
      "training_step_time": 10.254080533981323
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 3.1529576778411865,
      "step": 6382
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 6382,
      "training_step_time": 9.92852234840393
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 3.410799026489258,
      "step": 6382
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 6382,
      "training_step_time": 10.192745923995972
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 3.136939525604248,
      "step": 6382
    },
    {
      "dynamics_loss_avg": 0.04955162638798356,
      "epoch": 2.434539794921875e-05,
      "step": 6382
    },
    {
      "action_loss_avg": 0.022227234626188874,
      "epoch": 2.434539794921875e-05,
      "step": 6382
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 6382,
      "training_step_time": 9.910219669342041
    },
    {
      "epoch": 2.434539794921875e-05,
      "model_forward_time": 3.1294970512390137,
      "step": 6382
    },
    {
      "epoch": 2.434539794921875e-05,
      "step": 6382,
      "training_step_time": 10.057109594345093
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 2.44805908203125,
      "step": 6383
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 6383,
      "training_step_time": 7.501993894577026
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 2.4280693531036377,
      "step": 6383
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 6383,
      "training_step_time": 7.455505609512329
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 3.4521875381469727,
      "step": 6383
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 6383,
      "training_step_time": 10.229271411895752
    },
    {
      "epoch": 2.4349212646484375e-05,
      "model_forward_time": 3.1370344161987305,
      "step": 6383
    },
    {
      "epoch": 2.4349212646484375e-05,
      "step": 6383,
      "training_step_time": 10.301610231399536
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 3.1453654766082764,
      "step": 6384
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 6384,
      "training_step_time": 9.918902397155762
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 2.4280993938446045,
      "step": 6384
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 6384,
      "training_step_time": 7.45857572555542
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 3.12829852104187,
      "step": 6384
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 6384,
      "training_step_time": 9.895216226577759
    },
    {
      "epoch": 2.435302734375e-05,
      "model_forward_time": 2.4322078227996826,
      "step": 6384
    },
    {
      "epoch": 2.435302734375e-05,
      "step": 6384,
      "training_step_time": 10.286214113235474
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 2.448476552963257,
      "step": 6385
    },
    {
      "dynamics_loss_avg": 0.05492623206228018,
      "epoch": 2.4356842041015624e-05,
      "step": 6385
    },
    {
      "action_loss_avg": 0.0222058501560241,
      "epoch": 2.4356842041015624e-05,
      "step": 6385
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 6385,
      "training_step_time": 7.478654861450195
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 3.1333789825439453,
      "step": 6385
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 6385,
      "training_step_time": 9.909679174423218
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 2.4289357662200928,
      "step": 6385
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 6385,
      "training_step_time": 7.459223031997681
    },
    {
      "epoch": 2.4356842041015624e-05,
      "model_forward_time": 2.4302704334259033,
      "step": 6385
    },
    {
      "epoch": 2.4356842041015624e-05,
      "step": 6385,
      "training_step_time": 10.31428074836731
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 2.4412686824798584,
      "step": 6386
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 6386,
      "training_step_time": 7.474072456359863
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 3.4469659328460693,
      "step": 6386
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 6386,
      "training_step_time": 10.217985391616821
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 3.134915828704834,
      "step": 6386
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 6386,
      "training_step_time": 9.909141778945923
    },
    {
      "epoch": 2.436065673828125e-05,
      "model_forward_time": 3.1361258029937744,
      "step": 6386
    },
    {
      "epoch": 2.436065673828125e-05,
      "step": 6386,
      "training_step_time": 10.028136253356934
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 3.152097463607788,
      "step": 6387
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 6387,
      "training_step_time": 9.926547765731812
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 2.433457374572754,
      "step": 6387
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 6387,
      "training_step_time": 7.465329647064209
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 2.675483465194702,
      "step": 6387
    },
    {
      "dynamics_loss_avg": 0.0897297978401184,
      "epoch": 2.4364471435546873e-05,
      "step": 6387
    },
    {
      "action_loss_avg": 0.03719912711530924,
      "epoch": 2.4364471435546873e-05,
      "step": 6387
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 6387,
      "training_step_time": 7.70542311668396
    },
    {
      "epoch": 2.4364471435546873e-05,
      "model_forward_time": 1.742218255996704,
      "step": 6387
    },
    {
      "epoch": 2.4364471435546873e-05,
      "step": 6387,
      "training_step_time": 10.293146133422852
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 3.1714131832122803,
      "step": 6388
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 6388,
      "training_step_time": 9.949800252914429
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 3.1327223777770996,
      "step": 6388
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 6388,
      "training_step_time": 9.91141128540039
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 3.1405062675476074,
      "step": 6388
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 6388,
      "training_step_time": 9.920656681060791
    },
    {
      "epoch": 2.43682861328125e-05,
      "model_forward_time": 3.4108948707580566,
      "step": 6388
    },
    {
      "epoch": 2.43682861328125e-05,
      "step": 6388,
      "training_step_time": 10.285966396331787
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 2.4438018798828125,
      "step": 6389
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 6389,
      "training_step_time": 7.478008508682251
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 3.1351661682128906,
      "step": 6389
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 6389,
      "training_step_time": 9.92289400100708
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 3.138857364654541,
      "step": 6389
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 6389,
      "training_step_time": 9.919985055923462
    },
    {
      "epoch": 2.4372100830078126e-05,
      "model_forward_time": 3.14007568359375,
      "step": 6389
    },
    {
      "epoch": 2.4372100830078126e-05,
      "step": 6389,
      "training_step_time": 10.272910833358765
    },
    {
      "epoch": 2.437591552734375e-05,
      "grad_norm": 0.39535096287727356,
      "learning_rate": 1e-05,
      "loss": 0.0873,
      "step": 6390
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 3.4236133098602295,
      "step": 6390
    },
    {
      "dynamics_loss_avg": 0.06141626257449388,
      "epoch": 2.437591552734375e-05,
      "step": 6390
    },
    {
      "action_loss_avg": 0.027128860214725138,
      "epoch": 2.437591552734375e-05,
      "step": 6390
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 6390,
      "training_step_time": 10.204695701599121
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 2.4343857765197754,
      "step": 6390
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 6390,
      "training_step_time": 7.467238187789917
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 3.1310346126556396,
      "step": 6390
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 6390,
      "training_step_time": 9.909957885742188
    },
    {
      "epoch": 2.437591552734375e-05,
      "model_forward_time": 3.1340510845184326,
      "step": 6390
    },
    {
      "epoch": 2.437591552734375e-05,
      "step": 6390,
      "training_step_time": 10.295142650604248
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 2.439540147781372,
      "step": 6391
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 6391,
      "training_step_time": 7.46788763999939
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 3.1354453563690186,
      "step": 6391
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 6391,
      "training_step_time": 9.903889179229736
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 3.44285249710083,
      "step": 6391
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 6391,
      "training_step_time": 10.210062026977539
    },
    {
      "epoch": 2.4379730224609375e-05,
      "model_forward_time": 3.1317312717437744,
      "step": 6391
    },
    {
      "epoch": 2.4379730224609375e-05,
      "step": 6391,
      "training_step_time": 10.321008443832397
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 3.1677212715148926,
      "step": 6392
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 6392,
      "training_step_time": 9.943416595458984
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 2.4286534786224365,
      "step": 6392
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 6392,
      "training_step_time": 7.460443735122681
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 3.130582571029663,
      "step": 6392
    },
    {
      "dynamics_loss_avg": 0.07481192126870155,
      "epoch": 2.4383544921875e-05,
      "step": 6392
    },
    {
      "action_loss_avg": 0.027121583744883538,
      "epoch": 2.4383544921875e-05,
      "step": 6392
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 6392,
      "training_step_time": 9.90571928024292
    },
    {
      "epoch": 2.4383544921875e-05,
      "model_forward_time": 3.377370834350586,
      "step": 6392
    },
    {
      "epoch": 2.4383544921875e-05,
      "step": 6392,
      "training_step_time": 10.266958475112915
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 2.4532740116119385,
      "step": 6393
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 6393,
      "training_step_time": 7.490414619445801
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 2.4322152137756348,
      "step": 6393
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 6393,
      "training_step_time": 7.467770338058472
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 2.4442458152770996,
      "step": 6393
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 6393,
      "training_step_time": 7.480266332626343
    },
    {
      "epoch": 2.4387359619140624e-05,
      "model_forward_time": 2.4329006671905518,
      "step": 6393
    },
    {
      "epoch": 2.4387359619140624e-05,
      "step": 6393,
      "training_step_time": 10.035667419433594
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 3.418163537979126,
      "step": 6394
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 6394,
      "training_step_time": 10.202454328536987
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 3.137643575668335,
      "step": 6394
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 6394,
      "training_step_time": 9.915164947509766
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 2.432060480117798,
      "step": 6394
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 6394,
      "training_step_time": 7.464777708053589
    },
    {
      "epoch": 2.439117431640625e-05,
      "model_forward_time": 2.4292800426483154,
      "step": 6394
    },
    {
      "epoch": 2.439117431640625e-05,
      "step": 6394,
      "training_step_time": 10.283612251281738
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 3.195042848587036,
      "step": 6395
    },
    {
      "dynamics_loss_avg": 0.06199457328766585,
      "epoch": 2.4394989013671873e-05,
      "step": 6395
    },
    {
      "action_loss_avg": 0.032829553540796044,
      "epoch": 2.4394989013671873e-05,
      "step": 6395
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 6395,
      "training_step_time": 9.97759485244751
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 3.4579625129699707,
      "step": 6395
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 6395,
      "training_step_time": 10.231122255325317
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 3.135633707046509,
      "step": 6395
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 6395,
      "training_step_time": 9.913209676742554
    },
    {
      "epoch": 2.4394989013671873e-05,
      "model_forward_time": 3.148391008377075,
      "step": 6395
    },
    {
      "epoch": 2.4394989013671873e-05,
      "step": 6395,
      "training_step_time": 10.309704303741455
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 2.4577250480651855,
      "step": 6396
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 6396,
      "training_step_time": 7.495605707168579
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 3.1443986892700195,
      "step": 6396
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 6396,
      "training_step_time": 9.933587312698364
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 2.4401185512542725,
      "step": 6396
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 6396,
      "training_step_time": 7.470996141433716
    },
    {
      "epoch": 2.43988037109375e-05,
      "model_forward_time": 2.749722957611084,
      "step": 6396
    },
    {
      "epoch": 2.43988037109375e-05,
      "step": 6396,
      "training_step_time": 10.329270124435425
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 2.4512929916381836,
      "step": 6397
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 6397,
      "training_step_time": 7.480189085006714
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 3.1434290409088135,
      "step": 6397
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 6397,
      "training_step_time": 9.91300654411316
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 2.434495210647583,
      "step": 6397
    },
    {
      "dynamics_loss_avg": 0.042132724449038504,
      "epoch": 2.4402618408203126e-05,
      "step": 6397
    },
    {
      "action_loss_avg": 0.021261784713715315,
      "epoch": 2.4402618408203126e-05,
      "step": 6397
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 6397,
      "training_step_time": 7.4657838344573975
    },
    {
      "epoch": 2.4402618408203126e-05,
      "model_forward_time": 3.1373674869537354,
      "step": 6397
    },
    {
      "epoch": 2.4402618408203126e-05,
      "step": 6397,
      "training_step_time": 10.3197922706604
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 3.480311393737793,
      "step": 6398
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 6398,
      "training_step_time": 10.25408148765564
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 3.133100986480713,
      "step": 6398
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 6398,
      "training_step_time": 9.903655529022217
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 2.4346671104431152,
      "step": 6398
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 6398,
      "training_step_time": 7.4609904289245605
    },
    {
      "epoch": 2.440643310546875e-05,
      "model_forward_time": 3.13309383392334,
      "step": 6398
    },
    {
      "epoch": 2.440643310546875e-05,
      "step": 6398,
      "training_step_time": 10.042738437652588
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 3.149244546890259,
      "step": 6399
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 6399,
      "training_step_time": 9.927562952041626
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 3.4263546466827393,
      "step": 6399
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 6399,
      "training_step_time": 10.201921224594116
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 2.4317595958709717,
      "step": 6399
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 6399,
      "training_step_time": 7.467066049575806
    },
    {
      "epoch": 2.4410247802734375e-05,
      "model_forward_time": 2.42850399017334,
      "step": 6399
    },
    {
      "epoch": 2.4410247802734375e-05,
      "step": 6399,
      "training_step_time": 10.029311656951904
    },
    {
      "epoch": 2.44140625e-05,
      "grad_norm": 0.3247873783111572,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 6400
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 1.762357473373413,
      "step": 6400
    },
    {
      "dynamics_loss_avg": 0.0642023291438818,
      "epoch": 2.44140625e-05,
      "step": 6400
    },
    {
      "action_loss_avg": 0.02394348382949829,
      "epoch": 2.44140625e-05,
      "step": 6400
    },
    {
      "epoch": 2.44140625e-05,
      "step": 6400,
      "training_step_time": 5.227257013320923
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 2.430708169937134,
      "step": 6400
    },
    {
      "epoch": 2.44140625e-05,
      "step": 6400,
      "training_step_time": 7.455349922180176
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 2.4244439601898193,
      "step": 6400
    },
    {
      "epoch": 2.44140625e-05,
      "step": 6400,
      "training_step_time": 7.743541479110718
    },
    {
      "epoch": 2.44140625e-05,
      "model_forward_time": 3.1316347122192383,
      "step": 6400
    },
    {
      "epoch": 2.44140625e-05,
      "step": 6400,
      "training_step_time": 10.305319786071777
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 3.1411101818084717,
      "step": 6401
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 6401,
      "training_step_time": 9.914273500442505
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 2.4286184310913086,
      "step": 6401
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 6401,
      "training_step_time": 7.457771062850952
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 3.1222193241119385,
      "step": 6401
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 6401,
      "training_step_time": 9.890319347381592
    },
    {
      "epoch": 2.4417877197265624e-05,
      "model_forward_time": 3.429448127746582,
      "step": 6401
    },
    {
      "epoch": 2.4417877197265624e-05,
      "step": 6401,
      "training_step_time": 10.290892601013184
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 3.140380382537842,
      "step": 6402
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 6402,
      "training_step_time": 9.915426015853882
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 2.4285295009613037,
      "step": 6402
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 6402,
      "training_step_time": 7.4518723487854
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 2.427969455718994,
      "step": 6402
    },
    {
      "dynamics_loss_avg": 0.0451089221984148,
      "epoch": 2.442169189453125e-05,
      "step": 6402
    },
    {
      "action_loss_avg": 0.02219754848629236,
      "epoch": 2.442169189453125e-05,
      "step": 6402
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 6402,
      "training_step_time": 7.455298185348511
    },
    {
      "epoch": 2.442169189453125e-05,
      "model_forward_time": 2.428516387939453,
      "step": 6402
    },
    {
      "epoch": 2.442169189453125e-05,
      "step": 6402,
      "training_step_time": 10.269109725952148
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 2.473902463912964,
      "step": 6403
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 6403,
      "training_step_time": 7.503793478012085
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 2.716151237487793,
      "step": 6403
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 6403,
      "training_step_time": 7.746372699737549
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 3.133955955505371,
      "step": 6403
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 6403,
      "training_step_time": 9.9083890914917
    },
    {
      "epoch": 2.4425506591796873e-05,
      "model_forward_time": 3.1426970958709717,
      "step": 6403
    },
    {
      "epoch": 2.4425506591796873e-05,
      "step": 6403,
      "training_step_time": 10.303711414337158
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 2.4435904026031494,
      "step": 6404
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 6404,
      "training_step_time": 7.4805943965911865
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 3.128134250640869,
      "step": 6404
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 6404,
      "training_step_time": 9.90143346786499
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 3.402463674545288,
      "step": 6404
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 6404,
      "training_step_time": 10.177356719970703
    },
    {
      "epoch": 2.44293212890625e-05,
      "model_forward_time": 2.4336509704589844,
      "step": 6404
    },
    {
      "epoch": 2.44293212890625e-05,
      "step": 6404,
      "training_step_time": 10.011465072631836
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 3.148258686065674,
      "step": 6405
    },
    {
      "dynamics_loss_avg": 0.06776705998927354,
      "epoch": 2.4433135986328126e-05,
      "step": 6405
    },
    {
      "action_loss_avg": 0.032051194459199905,
      "epoch": 2.4433135986328126e-05,
      "step": 6405
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 6405,
      "training_step_time": 9.925101041793823
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 3.1299750804901123,
      "step": 6405
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 6405,
      "training_step_time": 9.90100908279419
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 2.431964874267578,
      "step": 6405
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 6405,
      "training_step_time": 7.45863938331604
    },
    {
      "epoch": 2.4433135986328126e-05,
      "model_forward_time": 3.1335644721984863,
      "step": 6405
    },
    {
      "epoch": 2.4433135986328126e-05,
      "step": 6405,
      "training_step_time": 10.323739051818848
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 3.1613428592681885,
      "step": 6406
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 6406,
      "training_step_time": 9.943825483322144
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 2.4416041374206543,
      "step": 6406
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 6406,
      "training_step_time": 7.4730870723724365
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 2.4370720386505127,
      "step": 6406
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 6406,
      "training_step_time": 7.463393211364746
    },
    {
      "epoch": 2.443695068359375e-05,
      "model_forward_time": 3.1352603435516357,
      "step": 6406
    },
    {
      "epoch": 2.443695068359375e-05,
      "step": 6406,
      "training_step_time": 10.019633054733276
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 2.4573514461517334,
      "step": 6407
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 6407,
      "training_step_time": 7.487585544586182
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 3.3882179260253906,
      "step": 6407
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 6407,
      "training_step_time": 10.162575483322144
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 3.148757219314575,
      "step": 6407
    },
    {
      "dynamics_loss_avg": 0.0540166437625885,
      "epoch": 2.4440765380859375e-05,
      "step": 6407
    },
    {
      "action_loss_avg": 0.0228119432926178,
      "epoch": 2.4440765380859375e-05,
      "step": 6407
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 6407,
      "training_step_time": 9.930433988571167
    },
    {
      "epoch": 2.4440765380859375e-05,
      "model_forward_time": 3.1409757137298584,
      "step": 6407
    },
    {
      "epoch": 2.4440765380859375e-05,
      "step": 6407,
      "training_step_time": 10.318689584732056
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 3.1495838165283203,
      "step": 6408
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 6408,
      "training_step_time": 9.929169654846191
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 3.1468424797058105,
      "step": 6408
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 6408,
      "training_step_time": 9.926915407180786
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 2.7178049087524414,
      "step": 6408
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 6408,
      "training_step_time": 7.755321025848389
    },
    {
      "epoch": 2.4444580078125e-05,
      "model_forward_time": 3.1420702934265137,
      "step": 6408
    },
    {
      "epoch": 2.4444580078125e-05,
      "step": 6408,
      "training_step_time": 10.331964015960693
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 2.4597127437591553,
      "step": 6409
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 6409,
      "training_step_time": 7.4974684715271
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 3.146554470062256,
      "step": 6409
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 6409,
      "training_step_time": 9.926271200180054
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 2.438575267791748,
      "step": 6409
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 6409,
      "training_step_time": 7.471640110015869
    },
    {
      "epoch": 2.4448394775390624e-05,
      "model_forward_time": 2.712263345718384,
      "step": 6409
    },
    {
      "epoch": 2.4448394775390624e-05,
      "step": 6409,
      "training_step_time": 10.00960636138916
    },
    {
      "epoch": 2.445220947265625e-05,
      "grad_norm": 0.2994491755962372,
      "learning_rate": 1e-05,
      "loss": 0.0865,
      "step": 6410
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 3.1555323600769043,
      "step": 6410
    },
    {
      "dynamics_loss_avg": 0.05387392174452543,
      "epoch": 2.445220947265625e-05,
      "step": 6410
    },
    {
      "action_loss_avg": 0.024074284918606283,
      "epoch": 2.445220947265625e-05,
      "step": 6410
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 6410,
      "training_step_time": 9.927922010421753
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 3.137657403945923,
      "step": 6410
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 6410,
      "training_step_time": 9.909118175506592
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 3.1352477073669434,
      "step": 6410
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 6410,
      "training_step_time": 9.903691530227661
    },
    {
      "epoch": 2.445220947265625e-05,
      "model_forward_time": 2.433727741241455,
      "step": 6410
    },
    {
      "epoch": 2.445220947265625e-05,
      "step": 6410,
      "training_step_time": 10.250442504882812
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 3.4030959606170654,
      "step": 6411
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 6411,
      "training_step_time": 10.176138877868652
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 2.4379265308380127,
      "step": 6411
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 6411,
      "training_step_time": 7.465088605880737
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 2.432229518890381,
      "step": 6411
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 6411,
      "training_step_time": 7.4528892040252686
    },
    {
      "epoch": 2.4456024169921873e-05,
      "model_forward_time": 3.1405301094055176,
      "step": 6411
    },
    {
      "epoch": 2.4456024169921873e-05,
      "step": 6411,
      "training_step_time": 10.309881925582886
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 2.444627523422241,
      "step": 6412
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 6412,
      "training_step_time": 7.47289252281189
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 3.4450418949127197,
      "step": 6412
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 6412,
      "training_step_time": 10.211986541748047
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 3.136481761932373,
      "step": 6412
    },
    {
      "dynamics_loss_avg": 0.054086904786527154,
      "epoch": 2.44598388671875e-05,
      "step": 6412
    },
    {
      "action_loss_avg": 0.01859982619062066,
      "epoch": 2.44598388671875e-05,
      "step": 6412
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 6412,
      "training_step_time": 9.907990455627441
    },
    {
      "epoch": 2.44598388671875e-05,
      "model_forward_time": 3.140840530395508,
      "step": 6412
    },
    {
      "epoch": 2.44598388671875e-05,
      "step": 6412,
      "training_step_time": 10.270697355270386
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 2.4783480167388916,
      "step": 6413
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 6413,
      "training_step_time": 7.512932777404785
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 2.4351887702941895,
      "step": 6413
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 6413,
      "training_step_time": 7.466814994812012
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 3.4133124351501465,
      "step": 6413
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 6413,
      "training_step_time": 10.181514501571655
    },
    {
      "epoch": 2.4463653564453126e-05,
      "model_forward_time": 3.141880989074707,
      "step": 6413
    },
    {
      "epoch": 2.4463653564453126e-05,
      "step": 6413,
      "training_step_time": 10.324496507644653
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 2.48060941696167,
      "step": 6414
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 6414,
      "training_step_time": 7.507917165756226
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 3.1402878761291504,
      "step": 6414
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 6414,
      "training_step_time": 9.910466432571411
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 3.1424224376678467,
      "step": 6414
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 6414,
      "training_step_time": 9.913287878036499
    },
    {
      "epoch": 2.446746826171875e-05,
      "model_forward_time": 3.1379168033599854,
      "step": 6414
    },
    {
      "epoch": 2.446746826171875e-05,
      "step": 6414,
      "training_step_time": 10.274827241897583
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 3.15818190574646,
      "step": 6415
    },
    {
      "dynamics_loss_avg": 0.04492682460695505,
      "epoch": 2.4471282958984375e-05,
      "step": 6415
    },
    {
      "action_loss_avg": 0.01995775639079511,
      "epoch": 2.4471282958984375e-05,
      "step": 6415
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 6415,
      "training_step_time": 9.945635795593262
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 3.142972946166992,
      "step": 6415
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 6415,
      "training_step_time": 9.91522216796875
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 3.145496368408203,
      "step": 6415
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 6415,
      "training_step_time": 9.919645309448242
    },
    {
      "epoch": 2.4471282958984375e-05,
      "model_forward_time": 2.442476272583008,
      "step": 6415
    },
    {
      "epoch": 2.4471282958984375e-05,
      "step": 6415,
      "training_step_time": 10.26567554473877
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 2.4562201499938965,
      "step": 6416
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 6416,
      "training_step_time": 7.798310995101929
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 3.1388463973999023,
      "step": 6416
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 6416,
      "training_step_time": 9.911109924316406
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 2.4367623329162598,
      "step": 6416
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 6416,
      "training_step_time": 7.469386577606201
    },
    {
      "epoch": 2.447509765625e-05,
      "model_forward_time": 3.1387557983398438,
      "step": 6416
    },
    {
      "epoch": 2.447509765625e-05,
      "step": 6416,
      "training_step_time": 10.32258677482605
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 1.7682476043701172,
      "step": 6417
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 6417,
      "training_step_time": 5.229476690292358
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 3.13847279548645,
      "step": 6417
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 6417,
      "training_step_time": 9.910460233688354
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 2.717125415802002,
      "step": 6417
    },
    {
      "dynamics_loss_avg": 0.05974240731447935,
      "epoch": 2.4478912353515624e-05,
      "step": 6417
    },
    {
      "action_loss_avg": 0.02472826191224158,
      "epoch": 2.4478912353515624e-05,
      "step": 6417
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 6417,
      "training_step_time": 7.744624137878418
    },
    {
      "epoch": 2.4478912353515624e-05,
      "model_forward_time": 3.1362359523773193,
      "step": 6417
    },
    {
      "epoch": 2.4478912353515624e-05,
      "step": 6417,
      "training_step_time": 10.259405612945557
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 3.14939284324646,
      "step": 6418
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 6418,
      "training_step_time": 9.92414665222168
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 2.4374632835388184,
      "step": 6418
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 6418,
      "training_step_time": 7.464275360107422
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 3.1374473571777344,
      "step": 6418
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 6418,
      "training_step_time": 9.903243780136108
    },
    {
      "epoch": 2.448272705078125e-05,
      "model_forward_time": 2.7574985027313232,
      "step": 6418
    },
    {
      "epoch": 2.448272705078125e-05,
      "step": 6418,
      "training_step_time": 10.330840349197388
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 3.154093027114868,
      "step": 6419
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 6419,
      "training_step_time": 9.920286893844604
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 3.1391735076904297,
      "step": 6419
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 6419,
      "training_step_time": 9.90901231765747
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 3.1400113105773926,
      "step": 6419
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 6419,
      "training_step_time": 9.905339241027832
    },
    {
      "epoch": 2.4486541748046873e-05,
      "model_forward_time": 2.4344260692596436,
      "step": 6419
    },
    {
      "epoch": 2.4486541748046873e-05,
      "step": 6419,
      "training_step_time": 10.30351710319519
    },
    {
      "epoch": 2.44903564453125e-05,
      "grad_norm": 0.4083288311958313,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 6420
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 2.446908950805664,
      "step": 6420
    },
    {
      "dynamics_loss_avg": 0.06440729442983865,
      "epoch": 2.44903564453125e-05,
      "step": 6420
    },
    {
      "action_loss_avg": 0.027292759250849487,
      "epoch": 2.44903564453125e-05,
      "step": 6420
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 6420,
      "training_step_time": 7.484341382980347
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 2.695091962814331,
      "step": 6420
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 6420,
      "training_step_time": 7.716344594955444
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 3.127680540084839,
      "step": 6420
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 6420,
      "training_step_time": 9.892282009124756
    },
    {
      "epoch": 2.44903564453125e-05,
      "model_forward_time": 2.4334919452667236,
      "step": 6420
    },
    {
      "epoch": 2.44903564453125e-05,
      "step": 6420,
      "training_step_time": 10.039896726608276
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 2.442551612854004,
      "step": 6421
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 6421,
      "training_step_time": 7.475210666656494
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 3.129770040512085,
      "step": 6421
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 6421,
      "training_step_time": 9.90429139137268
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 3.4172661304473877,
      "step": 6421
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 6421,
      "training_step_time": 10.189066648483276
    },
    {
      "epoch": 2.4494171142578126e-05,
      "model_forward_time": 2.4338293075561523,
      "step": 6421
    },
    {
      "epoch": 2.4494171142578126e-05,
      "step": 6421,
      "training_step_time": 10.323593854904175
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 2.4521827697753906,
      "step": 6422
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 6422,
      "training_step_time": 7.47989559173584
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 2.4327166080474854,
      "step": 6422
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 6422,
      "training_step_time": 7.460078954696655
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 2.4360599517822266,
      "step": 6422
    },
    {
      "dynamics_loss_avg": 0.046112663298845294,
      "epoch": 2.449798583984375e-05,
      "step": 6422
    },
    {
      "action_loss_avg": 0.020186554128304123,
      "epoch": 2.449798583984375e-05,
      "step": 6422
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 6422,
      "training_step_time": 7.5050578117370605
    },
    {
      "epoch": 2.449798583984375e-05,
      "model_forward_time": 2.0180232524871826,
      "step": 6422
    },
    {
      "epoch": 2.449798583984375e-05,
      "step": 6422,
      "training_step_time": 10.037641286849976
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 3.148906946182251,
      "step": 6423
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 6423,
      "training_step_time": 9.916830062866211
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 3.124298334121704,
      "step": 6423
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 6423,
      "training_step_time": 9.887536525726318
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 2.4278452396392822,
      "step": 6423
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 6423,
      "training_step_time": 7.446597576141357
    },
    {
      "epoch": 2.4501800537109375e-05,
      "model_forward_time": 3.1233866214752197,
      "step": 6423
    },
    {
      "epoch": 2.4501800537109375e-05,
      "step": 6423,
      "training_step_time": 10.339707374572754
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 3.1325888633728027,
      "step": 6424
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 6424,
      "training_step_time": 10.161457538604736
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 2.4263265132904053,
      "step": 6424
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 6424,
      "training_step_time": 7.443984746932983
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 2.423903465270996,
      "step": 6424
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 6424,
      "training_step_time": 7.440423011779785
    },
    {
      "epoch": 2.4505615234375e-05,
      "model_forward_time": 3.1254475116729736,
      "step": 6424
    },
    {
      "epoch": 2.4505615234375e-05,
      "step": 6424,
      "training_step_time": 10.273410558700562
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 2.4464404582977295,
      "step": 6425
    },
    {
      "dynamics_loss_avg": 0.04972925689071417,
      "epoch": 2.4509429931640624e-05,
      "step": 6425
    },
    {
      "action_loss_avg": 0.021106779389083386,
      "epoch": 2.4509429931640624e-05,
      "step": 6425
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 6425,
      "training_step_time": 7.465136766433716
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 3.1225404739379883,
      "step": 6425
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 6425,
      "training_step_time": 9.877237558364868
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 2.689727306365967,
      "step": 6425
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 6425,
      "training_step_time": 7.704294919967651
    },
    {
      "epoch": 2.4509429931640624e-05,
      "model_forward_time": 2.4280622005462646,
      "step": 6425
    },
    {
      "epoch": 2.4509429931640624e-05,
      "step": 6425,
      "training_step_time": 10.263896942138672
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 3.145716428756714,
      "step": 6426
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 6426,
      "training_step_time": 9.904964208602905
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 2.4310789108276367,
      "step": 6426
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 6426,
      "training_step_time": 7.4530651569366455
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 2.4217870235443115,
      "step": 6426
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 6426,
      "training_step_time": 7.439350605010986
    },
    {
      "epoch": 2.451324462890625e-05,
      "model_forward_time": 3.399676561355591,
      "step": 6426
    },
    {
      "epoch": 2.451324462890625e-05,
      "step": 6426,
      "training_step_time": 10.314465045928955
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 2.4660730361938477,
      "step": 6427
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 6427,
      "training_step_time": 7.496397256851196
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 3.1296579837799072,
      "step": 6427
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 6427,
      "training_step_time": 9.897534132003784
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 3.1315319538116455,
      "step": 6427
    },
    {
      "dynamics_loss_avg": 0.07445074934512377,
      "epoch": 2.4517059326171873e-05,
      "step": 6427
    },
    {
      "action_loss_avg": 0.02976101413369179,
      "epoch": 2.4517059326171873e-05,
      "step": 6427
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 6427,
      "training_step_time": 9.899869680404663
    },
    {
      "epoch": 2.4517059326171873e-05,
      "model_forward_time": 3.1310360431671143,
      "step": 6427
    },
    {
      "epoch": 2.4517059326171873e-05,
      "step": 6427,
      "training_step_time": 10.03281545639038
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 2.766075849533081,
      "step": 6428
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 6428,
      "training_step_time": 7.790987730026245
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 3.1318278312683105,
      "step": 6428
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 6428,
      "training_step_time": 9.897678136825562
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 2.4397616386413574,
      "step": 6428
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 6428,
      "training_step_time": 7.464093923568726
    },
    {
      "epoch": 2.45208740234375e-05,
      "model_forward_time": 3.1352992057800293,
      "step": 6428
    },
    {
      "epoch": 2.45208740234375e-05,
      "step": 6428,
      "training_step_time": 10.329920768737793
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 3.1735646724700928,
      "step": 6429
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 6429,
      "training_step_time": 9.939603328704834
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 2.427971363067627,
      "step": 6429
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 6429,
      "training_step_time": 7.714101076126099
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 3.1295723915100098,
      "step": 6429
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 6429,
      "training_step_time": 9.895209074020386
    },
    {
      "epoch": 2.4524688720703126e-05,
      "model_forward_time": 3.1403133869171143,
      "step": 6429
    },
    {
      "epoch": 2.4524688720703126e-05,
      "step": 6429,
      "training_step_time": 10.267850875854492
    },
    {
      "epoch": 2.452850341796875e-05,
      "grad_norm": 0.47326749563217163,
      "learning_rate": 1e-05,
      "loss": 0.084,
      "step": 6430
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 3.1506669521331787,
      "step": 6430
    },
    {
      "dynamics_loss_avg": 0.060261805541813374,
      "epoch": 2.452850341796875e-05,
      "step": 6430
    },
    {
      "action_loss_avg": 0.029294694773852825,
      "epoch": 2.452850341796875e-05,
      "step": 6430
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 6430,
      "training_step_time": 9.922376871109009
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 2.4318275451660156,
      "step": 6430
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 6430,
      "training_step_time": 7.457542419433594
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 3.387420415878296,
      "step": 6430
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 6430,
      "training_step_time": 10.155317068099976
    },
    {
      "epoch": 2.452850341796875e-05,
      "model_forward_time": 2.4324967861175537,
      "step": 6430
    },
    {
      "epoch": 2.452850341796875e-05,
      "step": 6430,
      "training_step_time": 10.319051504135132
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 3.1496424674987793,
      "step": 6431
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 6431,
      "training_step_time": 9.915733575820923
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 3.1332149505615234,
      "step": 6431
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 6431,
      "training_step_time": 9.909564018249512
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 3.1384129524230957,
      "step": 6431
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 6431,
      "training_step_time": 9.911610841751099
    },
    {
      "epoch": 2.4532318115234375e-05,
      "model_forward_time": 2.430370569229126,
      "step": 6431
    },
    {
      "epoch": 2.4532318115234375e-05,
      "step": 6431,
      "training_step_time": 10.302590131759644
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 2.759704828262329,
      "step": 6432
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 6432,
      "training_step_time": 7.781078577041626
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 3.1282918453216553,
      "step": 6432
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 6432,
      "training_step_time": 9.89542031288147
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 2.4305472373962402,
      "step": 6432
    },
    {
      "dynamics_loss_avg": 0.055185495782643555,
      "epoch": 2.45361328125e-05,
      "step": 6432
    },
    {
      "action_loss_avg": 0.02155897137708962,
      "epoch": 2.45361328125e-05,
      "step": 6432
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 6432,
      "training_step_time": 7.457885026931763
    },
    {
      "epoch": 2.45361328125e-05,
      "model_forward_time": 2.4260056018829346,
      "step": 6432
    },
    {
      "epoch": 2.45361328125e-05,
      "step": 6432,
      "training_step_time": 10.325592041015625
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 2.4513094425201416,
      "step": 6433
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 6433,
      "training_step_time": 7.474519491195679
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 2.696805715560913,
      "step": 6433
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 6433,
      "training_step_time": 7.716435670852661
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 1.7405738830566406,
      "step": 6433
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 6433,
      "training_step_time": 5.1916162967681885
    },
    {
      "epoch": 2.4539947509765624e-05,
      "model_forward_time": 2.429295063018799,
      "step": 6433
    },
    {
      "epoch": 2.4539947509765624e-05,
      "step": 6433,
      "training_step_time": 10.2722749710083
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 2.447882652282715,
      "step": 6434
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 6434,
      "training_step_time": 7.4691948890686035
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 3.1300575733184814,
      "step": 6434
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 6434,
      "training_step_time": 9.892760276794434
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 2.4347312450408936,
      "step": 6434
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 6434,
      "training_step_time": 7.454684734344482
    },
    {
      "epoch": 2.454376220703125e-05,
      "model_forward_time": 3.397706985473633,
      "step": 6434
    },
    {
      "epoch": 2.454376220703125e-05,
      "step": 6434,
      "training_step_time": 10.25687861442566
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 3.163027048110962,
      "step": 6435
    },
    {
      "dynamics_loss_avg": 0.06636632308363914,
      "epoch": 2.4547576904296874e-05,
      "step": 6435
    },
    {
      "action_loss_avg": 0.025915591232478618,
      "epoch": 2.4547576904296874e-05,
      "step": 6435
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 6435,
      "training_step_time": 9.930591821670532
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 3.1382925510406494,
      "step": 6435
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 6435,
      "training_step_time": 9.903793811798096
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 2.442835569381714,
      "step": 6435
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 6435,
      "training_step_time": 7.469084739685059
    },
    {
      "epoch": 2.4547576904296874e-05,
      "model_forward_time": 2.4328114986419678,
      "step": 6435
    },
    {
      "epoch": 2.4547576904296874e-05,
      "step": 6435,
      "training_step_time": 10.307268142700195
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 3.465564250946045,
      "step": 6436
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 6436,
      "training_step_time": 10.23120665550232
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 2.435579299926758,
      "step": 6436
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 6436,
      "training_step_time": 7.457261085510254
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 3.1336543560028076,
      "step": 6436
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 6436,
      "training_step_time": 9.8980872631073
    },
    {
      "epoch": 2.45513916015625e-05,
      "model_forward_time": 3.1424789428710938,
      "step": 6436
    },
    {
      "epoch": 2.45513916015625e-05,
      "step": 6436,
      "training_step_time": 10.31458830833435
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 2.450683832168579,
      "step": 6437
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 6437,
      "training_step_time": 7.478106498718262
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 2.7144875526428223,
      "step": 6437
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 6437,
      "training_step_time": 7.7385687828063965
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 2.4373795986175537,
      "step": 6437
    },
    {
      "dynamics_loss_avg": 0.057529470324516295,
      "epoch": 2.4555206298828126e-05,
      "step": 6437
    },
    {
      "action_loss_avg": 0.02383629335090518,
      "epoch": 2.4555206298828126e-05,
      "step": 6437
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 6437,
      "training_step_time": 7.466387748718262
    },
    {
      "epoch": 2.4555206298828126e-05,
      "model_forward_time": 2.4315922260284424,
      "step": 6437
    },
    {
      "epoch": 2.4555206298828126e-05,
      "step": 6437,
      "training_step_time": 10.030933380126953
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 3.162689208984375,
      "step": 6438
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 6438,
      "training_step_time": 9.932341814041138
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 2.4341299533843994,
      "step": 6438
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 6438,
      "training_step_time": 7.4662744998931885
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 2.696319580078125,
      "step": 6438
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 6438,
      "training_step_time": 7.721816301345825
    },
    {
      "epoch": 2.455902099609375e-05,
      "model_forward_time": 2.4336302280426025,
      "step": 6438
    },
    {
      "epoch": 2.455902099609375e-05,
      "step": 6438,
      "training_step_time": 10.340218544006348
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 2.4450714588165283,
      "step": 6439
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 6439,
      "training_step_time": 7.4677605628967285
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 2.4305381774902344,
      "step": 6439
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 6439,
      "training_step_time": 7.446481466293335
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 3.1334145069122314,
      "step": 6439
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 6439,
      "training_step_time": 9.896405220031738
    },
    {
      "epoch": 2.4562835693359375e-05,
      "model_forward_time": 2.6781091690063477,
      "step": 6439
    },
    {
      "epoch": 2.4562835693359375e-05,
      "step": 6439,
      "training_step_time": 10.272979974746704
    },
    {
      "epoch": 2.4566650390625e-05,
      "grad_norm": 0.5766101479530334,
      "learning_rate": 1e-05,
      "loss": 0.0832,
      "step": 6440
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 3.1675009727478027,
      "step": 6440
    },
    {
      "dynamics_loss_avg": 0.05925948601216078,
      "epoch": 2.4566650390625e-05,
      "step": 6440
    },
    {
      "action_loss_avg": 0.02247585724107921,
      "epoch": 2.4566650390625e-05,
      "step": 6440
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 6440,
      "training_step_time": 9.932428121566772
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 3.133665084838867,
      "step": 6440
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 6440,
      "training_step_time": 9.896585464477539
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 3.137566328048706,
      "step": 6440
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 6440,
      "training_step_time": 9.901183366775513
    },
    {
      "epoch": 2.4566650390625e-05,
      "model_forward_time": 3.1367135047912598,
      "step": 6440
    },
    {
      "epoch": 2.4566650390625e-05,
      "step": 6440,
      "training_step_time": 10.253134489059448
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 2.4515860080718994,
      "step": 6441
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 6441,
      "training_step_time": 7.754764795303345
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 3.1346983909606934,
      "step": 6441
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 6441,
      "training_step_time": 9.903540134429932
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 2.437413454055786,
      "step": 6441
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 6441,
      "training_step_time": 7.462828159332275
    },
    {
      "epoch": 2.4570465087890624e-05,
      "model_forward_time": 2.4327139854431152,
      "step": 6441
    },
    {
      "epoch": 2.4570465087890624e-05,
      "step": 6441,
      "training_step_time": 10.293385982513428
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 3.1592700481414795,
      "step": 6442
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 6442,
      "training_step_time": 9.930058717727661
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 2.434199810028076,
      "step": 6442
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 6442,
      "training_step_time": 7.462080717086792
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 3.4189469814300537,
      "step": 6442
    },
    {
      "dynamics_loss_avg": 0.05455398336052895,
      "epoch": 2.457427978515625e-05,
      "step": 6442
    },
    {
      "action_loss_avg": 0.020913375727832317,
      "epoch": 2.457427978515625e-05,
      "step": 6442
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 6442,
      "training_step_time": 10.194628477096558
    },
    {
      "epoch": 2.457427978515625e-05,
      "model_forward_time": 2.4418630599975586,
      "step": 6442
    },
    {
      "epoch": 2.457427978515625e-05,
      "step": 6442,
      "training_step_time": 10.32603120803833
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 3.153468132019043,
      "step": 6443
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 6443,
      "training_step_time": 9.925032377243042
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 1.7477455139160156,
      "step": 6443
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 6443,
      "training_step_time": 5.206357479095459
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 2.4356889724731445,
      "step": 6443
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 6443,
      "training_step_time": 7.462576389312744
    },
    {
      "epoch": 2.4578094482421874e-05,
      "model_forward_time": 2.7102677822113037,
      "step": 6443
    },
    {
      "epoch": 2.4578094482421874e-05,
      "step": 6443,
      "training_step_time": 10.29975438117981
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 3.1443898677825928,
      "step": 6444
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 6444,
      "training_step_time": 9.912675142288208
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 2.4335696697235107,
      "step": 6444
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 6444,
      "training_step_time": 7.453064441680908
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 2.4281911849975586,
      "step": 6444
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 6444,
      "training_step_time": 7.445166349411011
    },
    {
      "epoch": 2.45819091796875e-05,
      "model_forward_time": 2.4300291538238525,
      "step": 6444
    },
    {
      "epoch": 2.45819091796875e-05,
      "step": 6444,
      "training_step_time": 10.242432832717896
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 2.436475992202759,
      "step": 6445
    },
    {
      "dynamics_loss_avg": 0.061559957824647425,
      "epoch": 2.4585723876953126e-05,
      "step": 6445
    },
    {
      "action_loss_avg": 0.02459244355559349,
      "epoch": 2.4585723876953126e-05,
      "step": 6445
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 6445,
      "training_step_time": 7.459635972976685
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 2.712848424911499,
      "step": 6445
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 6445,
      "training_step_time": 7.73005747795105
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 3.1300339698791504,
      "step": 6445
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 6445,
      "training_step_time": 9.887115478515625
    },
    {
      "epoch": 2.4585723876953126e-05,
      "model_forward_time": 3.128169298171997,
      "step": 6445
    },
    {
      "epoch": 2.4585723876953126e-05,
      "step": 6445,
      "training_step_time": 10.218162775039673
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 2.4426827430725098,
      "step": 6446
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 6446,
      "training_step_time": 7.461896657943726
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 3.126574993133545,
      "step": 6446
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 6446,
      "training_step_time": 9.883100509643555
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 2.7072794437408447,
      "step": 6446
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 6446,
      "training_step_time": 7.725215196609497
    },
    {
      "epoch": 2.458953857421875e-05,
      "model_forward_time": 3.1338019371032715,
      "step": 6446
    },
    {
      "epoch": 2.458953857421875e-05,
      "step": 6446,
      "training_step_time": 10.256523609161377
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 2.4714794158935547,
      "step": 6447
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 6447,
      "training_step_time": 7.503488302230835
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 3.1306140422821045,
      "step": 6447
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 6447,
      "training_step_time": 9.89433765411377
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 3.1331839561462402,
      "step": 6447
    },
    {
      "dynamics_loss_avg": 0.03833114486187696,
      "epoch": 2.4593353271484375e-05,
      "step": 6447
    },
    {
      "action_loss_avg": 0.015777193661779165,
      "epoch": 2.4593353271484375e-05,
      "step": 6447
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 6447,
      "training_step_time": 9.900003910064697
    },
    {
      "epoch": 2.4593353271484375e-05,
      "model_forward_time": 3.406172275543213,
      "step": 6447
    },
    {
      "epoch": 2.4593353271484375e-05,
      "step": 6447,
      "training_step_time": 10.34597659111023
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 2.4474575519561768,
      "step": 6448
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 6448,
      "training_step_time": 7.4781951904296875
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 3.128124952316284,
      "step": 6448
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 6448,
      "training_step_time": 9.897946119308472
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 2.4293129444122314,
      "step": 6448
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 6448,
      "training_step_time": 7.456448793411255
    },
    {
      "epoch": 2.459716796875e-05,
      "model_forward_time": 2.430739641189575,
      "step": 6448
    },
    {
      "epoch": 2.459716796875e-05,
      "step": 6448,
      "training_step_time": 10.33101749420166
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 2.450209856033325,
      "step": 6449
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 6449,
      "training_step_time": 7.476383686065674
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 3.3958284854888916,
      "step": 6449
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 6449,
      "training_step_time": 10.165876626968384
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 2.429232120513916,
      "step": 6449
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 6449,
      "training_step_time": 7.451374530792236
    },
    {
      "epoch": 2.4600982666015624e-05,
      "model_forward_time": 2.4260473251342773,
      "step": 6449
    },
    {
      "epoch": 2.4600982666015624e-05,
      "step": 6449,
      "training_step_time": 10.2646484375
    },
    {
      "epoch": 2.460479736328125e-05,
      "grad_norm": 0.32057130336761475,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 6450
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 2.4413177967071533,
      "step": 6450
    },
    {
      "dynamics_loss_avg": 0.056288976781070234,
      "epoch": 2.460479736328125e-05,
      "step": 6450
    },
    {
      "action_loss_avg": 0.024305727705359458,
      "epoch": 2.460479736328125e-05,
      "step": 6450
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 6450,
      "training_step_time": 7.459310293197632
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 3.1247494220733643,
      "step": 6450
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 6450,
      "training_step_time": 9.884955406188965
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 2.7144429683685303,
      "step": 6450
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 6450,
      "training_step_time": 7.730047941207886
    },
    {
      "epoch": 2.460479736328125e-05,
      "model_forward_time": 3.12459135055542,
      "step": 6450
    },
    {
      "epoch": 2.460479736328125e-05,
      "step": 6450,
      "training_step_time": 10.270480871200562
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 2.448072910308838,
      "step": 6451
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 6451,
      "training_step_time": 7.466344833374023
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 2.431443214416504,
      "step": 6451
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 6451,
      "training_step_time": 7.446728944778442
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 3.1387939453125,
      "step": 6451
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 6451,
      "training_step_time": 9.900021076202393
    },
    {
      "epoch": 2.4608612060546874e-05,
      "model_forward_time": 2.7177164554595947,
      "step": 6451
    },
    {
      "epoch": 2.4608612060546874e-05,
      "step": 6451,
      "training_step_time": 10.264792203903198
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 2.4494168758392334,
      "step": 6452
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 6452,
      "training_step_time": 7.467848300933838
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 3.1373767852783203,
      "step": 6452
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 6452,
      "training_step_time": 9.900779724121094
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 2.4344801902770996,
      "step": 6452
    },
    {
      "dynamics_loss_avg": 0.06794738732278346,
      "epoch": 2.46124267578125e-05,
      "step": 6452
    },
    {
      "action_loss_avg": 0.02532572951167822,
      "epoch": 2.46124267578125e-05,
      "step": 6452
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 6452,
      "training_step_time": 7.456083297729492
    },
    {
      "epoch": 2.46124267578125e-05,
      "model_forward_time": 3.1346096992492676,
      "step": 6452
    },
    {
      "epoch": 2.46124267578125e-05,
      "step": 6452,
      "training_step_time": 10.30856704711914
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 3.431152820587158,
      "step": 6453
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 6453,
      "training_step_time": 10.19917368888855
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 2.435340404510498,
      "step": 6453
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 6453,
      "training_step_time": 7.459321022033691
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 3.134650468826294,
      "step": 6453
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 6453,
      "training_step_time": 9.902239084243774
    },
    {
      "epoch": 2.4616241455078126e-05,
      "model_forward_time": 3.141420364379883,
      "step": 6453
    },
    {
      "epoch": 2.4616241455078126e-05,
      "step": 6453,
      "training_step_time": 10.305418491363525
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 2.47395920753479,
      "step": 6454
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 6454,
      "training_step_time": 7.497645616531372
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 3.4111008644104004,
      "step": 6454
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 6454,
      "training_step_time": 10.171236991882324
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 3.133582830429077,
      "step": 6454
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 6454,
      "training_step_time": 9.89291524887085
    },
    {
      "epoch": 2.462005615234375e-05,
      "model_forward_time": 3.137026786804199,
      "step": 6454
    },
    {
      "epoch": 2.462005615234375e-05,
      "step": 6454,
      "training_step_time": 10.006839275360107
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.4486491680145264,
      "step": 6455
    },
    {
      "dynamics_loss_avg": 0.050931991543620825,
      "epoch": 2.4623870849609375e-05,
      "step": 6455
    },
    {
      "action_loss_avg": 0.024016787903383373,
      "epoch": 2.4623870849609375e-05,
      "step": 6455
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 6455,
      "training_step_time": 7.477459669113159
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.427263021469116,
      "step": 6455
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 6455,
      "training_step_time": 7.449644327163696
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.426788568496704,
      "step": 6455
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 6455,
      "training_step_time": 7.4459474086761475
    },
    {
      "epoch": 2.4623870849609375e-05,
      "model_forward_time": 2.698885679244995,
      "step": 6455
    },
    {
      "epoch": 2.4623870849609375e-05,
      "step": 6455,
      "training_step_time": 10.33618450164795
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 3.172865629196167,
      "step": 6456
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 6456,
      "training_step_time": 9.944278955459595
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 3.1363039016723633,
      "step": 6456
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 6456,
      "training_step_time": 9.903742790222168
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 3.142915725708008,
      "step": 6456
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 6456,
      "training_step_time": 9.913432121276855
    },
    {
      "epoch": 2.4627685546875e-05,
      "model_forward_time": 3.1402621269226074,
      "step": 6456
    },
    {
      "epoch": 2.4627685546875e-05,
      "step": 6456,
      "training_step_time": 10.288998365402222
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 3.4785306453704834,
      "step": 6457
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 6457,
      "training_step_time": 10.250447034835815
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 2.4419877529144287,
      "step": 6457
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 6457,
      "training_step_time": 7.4679670333862305
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 3.145328998565674,
      "step": 6457
    },
    {
      "dynamics_loss_avg": 0.06200542990118265,
      "epoch": 2.4631500244140624e-05,
      "step": 6457
    },
    {
      "action_loss_avg": 0.02142698848620057,
      "epoch": 2.4631500244140624e-05,
      "step": 6457
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 6457,
      "training_step_time": 9.920075416564941
    },
    {
      "epoch": 2.4631500244140624e-05,
      "model_forward_time": 3.1437056064605713,
      "step": 6457
    },
    {
      "epoch": 2.4631500244140624e-05,
      "step": 6457,
      "training_step_time": 10.297827243804932
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 2.44805908203125,
      "step": 6458
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 6458,
      "training_step_time": 7.480482816696167
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 2.691272735595703,
      "step": 6458
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 6458,
      "training_step_time": 7.723649024963379
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 3.1382811069488525,
      "step": 6458
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 6458,
      "training_step_time": 9.910593271255493
    },
    {
      "epoch": 2.463531494140625e-05,
      "model_forward_time": 3.1381680965423584,
      "step": 6458
    },
    {
      "epoch": 2.463531494140625e-05,
      "step": 6458,
      "training_step_time": 10.296197891235352
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 3.1621270179748535,
      "step": 6459
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 6459,
      "training_step_time": 9.934865474700928
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 3.1445977687835693,
      "step": 6459
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 6459,
      "training_step_time": 9.919941425323486
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 2.734621047973633,
      "step": 6459
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 6459,
      "training_step_time": 7.766063451766968
    },
    {
      "epoch": 2.4639129638671874e-05,
      "model_forward_time": 1.7462491989135742,
      "step": 6459
    },
    {
      "epoch": 2.4639129638671874e-05,
      "step": 6459,
      "training_step_time": 10.021475553512573
    },
    {
      "epoch": 2.46429443359375e-05,
      "grad_norm": 0.37251290678977966,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 6460
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 3.151453971862793,
      "step": 6460
    },
    {
      "dynamics_loss_avg": 0.07639578748494387,
      "epoch": 2.46429443359375e-05,
      "step": 6460
    },
    {
      "action_loss_avg": 0.031909510307013986,
      "epoch": 2.46429443359375e-05,
      "step": 6460
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 6460,
      "training_step_time": 9.92513108253479
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 2.4310755729675293,
      "step": 6460
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 6460,
      "training_step_time": 7.456287860870361
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 3.1340973377227783,
      "step": 6460
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 6460,
      "training_step_time": 9.909060001373291
    },
    {
      "epoch": 2.46429443359375e-05,
      "model_forward_time": 2.4341375827789307,
      "step": 6460
    },
    {
      "epoch": 2.46429443359375e-05,
      "step": 6460,
      "training_step_time": 10.290029287338257
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 3.4883360862731934,
      "step": 6461
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 6461,
      "training_step_time": 10.264480113983154
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 2.435521364212036,
      "step": 6461
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 6461,
      "training_step_time": 7.465962648391724
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 3.1344709396362305,
      "step": 6461
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 6461,
      "training_step_time": 9.904512643814087
    },
    {
      "epoch": 2.4646759033203126e-05,
      "model_forward_time": 2.4381372928619385,
      "step": 6461
    },
    {
      "epoch": 2.4646759033203126e-05,
      "step": 6461,
      "training_step_time": 10.310700416564941
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 3.1522679328918457,
      "step": 6462
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 6462,
      "training_step_time": 9.93172574043274
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 3.410778760910034,
      "step": 6462
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 6462,
      "training_step_time": 10.179510116577148
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 2.4335808753967285,
      "step": 6462
    },
    {
      "dynamics_loss_avg": 0.058237684145569804,
      "epoch": 2.465057373046875e-05,
      "step": 6462
    },
    {
      "action_loss_avg": 0.023140000738203527,
      "epoch": 2.465057373046875e-05,
      "step": 6462
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 6462,
      "training_step_time": 7.461014270782471
    },
    {
      "epoch": 2.465057373046875e-05,
      "model_forward_time": 3.13211727142334,
      "step": 6462
    },
    {
      "epoch": 2.465057373046875e-05,
      "step": 6462,
      "training_step_time": 10.049803495407104
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 3.148892641067505,
      "step": 6463
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 6463,
      "training_step_time": 9.91660761833191
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 3.1347782611846924,
      "step": 6463
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 6463,
      "training_step_time": 9.905447959899902
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 2.7106170654296875,
      "step": 6463
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 6463,
      "training_step_time": 7.740910530090332
    },
    {
      "epoch": 2.4654388427734375e-05,
      "model_forward_time": 3.1381494998931885,
      "step": 6463
    },
    {
      "epoch": 2.4654388427734375e-05,
      "step": 6463,
      "training_step_time": 10.301841974258423
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 3.1482160091400146,
      "step": 6464
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 6464,
      "training_step_time": 9.923523902893066
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 2.4337821006774902,
      "step": 6464
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 6464,
      "training_step_time": 7.467314004898071
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 3.1414287090301514,
      "step": 6464
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 6464,
      "training_step_time": 9.918448209762573
    },
    {
      "epoch": 2.4658203125e-05,
      "model_forward_time": 3.4141225814819336,
      "step": 6464
    },
    {
      "epoch": 2.4658203125e-05,
      "step": 6464,
      "training_step_time": 10.29999566078186
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 3.1580350399017334,
      "step": 6465
    },
    {
      "dynamics_loss_avg": 0.060292262211441995,
      "epoch": 2.4662017822265624e-05,
      "step": 6465
    },
    {
      "action_loss_avg": 0.025918645318597555,
      "epoch": 2.4662017822265624e-05,
      "step": 6465
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 6465,
      "training_step_time": 9.94197702407837
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 3.141836643218994,
      "step": 6465
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 6465,
      "training_step_time": 9.923187017440796
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 3.145225763320923,
      "step": 6465
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 6465,
      "training_step_time": 9.924718856811523
    },
    {
      "epoch": 2.4662017822265624e-05,
      "model_forward_time": 3.1409285068511963,
      "step": 6465
    },
    {
      "epoch": 2.4662017822265624e-05,
      "step": 6465,
      "training_step_time": 10.31021499633789
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 3.444380521774292,
      "step": 6466
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 6466,
      "training_step_time": 10.222049474716187
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 3.140165328979492,
      "step": 6466
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 6466,
      "training_step_time": 9.919113397598267
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 3.1430928707122803,
      "step": 6466
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 6466,
      "training_step_time": 9.921533584594727
    },
    {
      "epoch": 2.466583251953125e-05,
      "model_forward_time": 2.438328981399536,
      "step": 6466
    },
    {
      "epoch": 2.466583251953125e-05,
      "step": 6466,
      "training_step_time": 10.02474331855774
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 3.161388874053955,
      "step": 6467
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 6467,
      "training_step_time": 9.940396070480347
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 3.4209327697753906,
      "step": 6467
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 6467,
      "training_step_time": 10.199235916137695
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 3.1423110961914062,
      "step": 6467
    },
    {
      "dynamics_loss_avg": 0.05791410934180021,
      "epoch": 2.4669647216796874e-05,
      "step": 6467
    },
    {
      "action_loss_avg": 0.030690562073141338,
      "epoch": 2.4669647216796874e-05,
      "step": 6467
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 6467,
      "training_step_time": 9.925530195236206
    },
    {
      "epoch": 2.4669647216796874e-05,
      "model_forward_time": 2.436459541320801,
      "step": 6467
    },
    {
      "epoch": 2.4669647216796874e-05,
      "step": 6467,
      "training_step_time": 10.23812747001648
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 2.4609243869781494,
      "step": 6468
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 6468,
      "training_step_time": 7.4947192668914795
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 3.1487977504730225,
      "step": 6468
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 6468,
      "training_step_time": 9.931747913360596
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 2.762705087661743,
      "step": 6468
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 6468,
      "training_step_time": 7.795026779174805
    },
    {
      "epoch": 2.46734619140625e-05,
      "model_forward_time": 2.4368956089019775,
      "step": 6468
    },
    {
      "epoch": 2.46734619140625e-05,
      "step": 6468,
      "training_step_time": 10.291396141052246
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 3.162551164627075,
      "step": 6469
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 6469,
      "training_step_time": 9.939882755279541
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 2.4399642944335938,
      "step": 6469
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 6469,
      "training_step_time": 7.468377351760864
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 2.4366941452026367,
      "step": 6469
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 6469,
      "training_step_time": 7.4686055183410645
    },
    {
      "epoch": 2.4677276611328126e-05,
      "model_forward_time": 2.432021379470825,
      "step": 6469
    },
    {
      "epoch": 2.4677276611328126e-05,
      "step": 6469,
      "training_step_time": 10.247665405273438
    },
    {
      "epoch": 2.468109130859375e-05,
      "grad_norm": 0.34294313192367554,
      "learning_rate": 1e-05,
      "loss": 0.0871,
      "step": 6470
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 3.4785382747650146,
      "step": 6470
    },
    {
      "dynamics_loss_avg": 0.06930226795375347,
      "epoch": 2.468109130859375e-05,
      "step": 6470
    },
    {
      "action_loss_avg": 0.023410484846681356,
      "epoch": 2.468109130859375e-05,
      "step": 6470
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 6470,
      "training_step_time": 10.255250453948975
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 3.139702796936035,
      "step": 6470
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 6470,
      "training_step_time": 9.913036823272705
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 2.43503999710083,
      "step": 6470
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 6470,
      "training_step_time": 7.466899156570435
    },
    {
      "epoch": 2.468109130859375e-05,
      "model_forward_time": 3.132859468460083,
      "step": 6470
    },
    {
      "epoch": 2.468109130859375e-05,
      "step": 6470,
      "training_step_time": 10.309139490127563
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 2.4499995708465576,
      "step": 6471
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 6471,
      "training_step_time": 7.482804536819458
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 3.4421355724334717,
      "step": 6471
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 6471,
      "training_step_time": 10.209900140762329
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 2.429769515991211,
      "step": 6471
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 6471,
      "training_step_time": 7.454299688339233
    },
    {
      "epoch": 2.4684906005859375e-05,
      "model_forward_time": 2.428697347640991,
      "step": 6471
    },
    {
      "epoch": 2.4684906005859375e-05,
      "step": 6471,
      "training_step_time": 10.028259515762329
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 2.442680835723877,
      "step": 6472
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 6472,
      "training_step_time": 7.4728920459747314
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 3.133620262145996,
      "step": 6472
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 6472,
      "training_step_time": 9.9010648727417
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 3.3870553970336914,
      "step": 6472
    },
    {
      "dynamics_loss_avg": 0.050975416973233226,
      "epoch": 2.4688720703125e-05,
      "step": 6472
    },
    {
      "action_loss_avg": 0.02146552484482527,
      "epoch": 2.4688720703125e-05,
      "step": 6472
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 6472,
      "training_step_time": 10.171243906021118
    },
    {
      "epoch": 2.4688720703125e-05,
      "model_forward_time": 2.432655096054077,
      "step": 6472
    },
    {
      "epoch": 2.4688720703125e-05,
      "step": 6472,
      "training_step_time": 10.031624794006348
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 2.466038227081299,
      "step": 6473
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 6473,
      "training_step_time": 7.499181747436523
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 3.1412770748138428,
      "step": 6473
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 6473,
      "training_step_time": 9.921236753463745
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 3.136491537094116,
      "step": 6473
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 6473,
      "training_step_time": 9.910077571868896
    },
    {
      "epoch": 2.4692535400390624e-05,
      "model_forward_time": 3.4170451164245605,
      "step": 6473
    },
    {
      "epoch": 2.4692535400390624e-05,
      "step": 6473,
      "training_step_time": 10.30088210105896
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 3.1533398628234863,
      "step": 6474
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 6474,
      "training_step_time": 9.926599502563477
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 3.136094570159912,
      "step": 6474
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 6474,
      "training_step_time": 9.907626152038574
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 1.7462866306304932,
      "step": 6474
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 6474,
      "training_step_time": 5.206573724746704
    },
    {
      "epoch": 2.469635009765625e-05,
      "model_forward_time": 2.4308929443359375,
      "step": 6474
    },
    {
      "epoch": 2.469635009765625e-05,
      "step": 6474,
      "training_step_time": 10.310625791549683
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 1.7615203857421875,
      "step": 6475
    },
    {
      "dynamics_loss_avg": 0.044624712131917474,
      "epoch": 2.4700164794921874e-05,
      "step": 6475
    },
    {
      "action_loss_avg": 0.022404151409864424,
      "epoch": 2.4700164794921874e-05,
      "step": 6475
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 6475,
      "training_step_time": 5.221573352813721
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 3.431699752807617,
      "step": 6475
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 6475,
      "training_step_time": 10.208155632019043
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 3.139813184738159,
      "step": 6475
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 6475,
      "training_step_time": 9.915029048919678
    },
    {
      "epoch": 2.4700164794921874e-05,
      "model_forward_time": 2.4409637451171875,
      "step": 6475
    },
    {
      "epoch": 2.4700164794921874e-05,
      "step": 6475,
      "training_step_time": 10.331790447235107
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 3.14530348777771,
      "step": 6476
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 6476,
      "training_step_time": 9.927289724349976
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 2.4353885650634766,
      "step": 6476
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 6476,
      "training_step_time": 7.4688560962677
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 2.005094051361084,
      "step": 6476
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 6476,
      "training_step_time": 5.467645883560181
    },
    {
      "epoch": 2.47039794921875e-05,
      "model_forward_time": 3.1278634071350098,
      "step": 6476
    },
    {
      "epoch": 2.47039794921875e-05,
      "step": 6476,
      "training_step_time": 10.04771876335144
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 2.4461071491241455,
      "step": 6477
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 6477,
      "training_step_time": 7.481016635894775
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 3.1319518089294434,
      "step": 6477
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 6477,
      "training_step_time": 9.904489278793335
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 2.431807041168213,
      "step": 6477
    },
    {
      "dynamics_loss_avg": 0.06819611107930541,
      "epoch": 2.4707794189453126e-05,
      "step": 6477
    },
    {
      "action_loss_avg": 0.030807158071547748,
      "epoch": 2.4707794189453126e-05,
      "step": 6477
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 6477,
      "training_step_time": 7.460107326507568
    },
    {
      "epoch": 2.4707794189453126e-05,
      "model_forward_time": 3.1294355392456055,
      "step": 6477
    },
    {
      "epoch": 2.4707794189453126e-05,
      "step": 6477,
      "training_step_time": 10.021766424179077
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 3.422858953475952,
      "step": 6478
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 6478,
      "training_step_time": 10.195939064025879
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 1.744459629058838,
      "step": 6478
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 6478,
      "training_step_time": 5.2042882442474365
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 3.128244638442993,
      "step": 6478
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 6478,
      "training_step_time": 9.900879383087158
    },
    {
      "epoch": 2.471160888671875e-05,
      "model_forward_time": 2.432410478591919,
      "step": 6478
    },
    {
      "epoch": 2.471160888671875e-05,
      "step": 6478,
      "training_step_time": 10.28716254234314
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 1.7647910118103027,
      "step": 6479
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 6479,
      "training_step_time": 5.2299158573150635
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 2.4303131103515625,
      "step": 6479
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 6479,
      "training_step_time": 7.785137176513672
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 1.7415289878845215,
      "step": 6479
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 6479,
      "training_step_time": 5.20013427734375
    },
    {
      "epoch": 2.4715423583984375e-05,
      "model_forward_time": 2.429802656173706,
      "step": 6479
    },
    {
      "epoch": 2.4715423583984375e-05,
      "step": 6479,
      "training_step_time": 10.301570653915405
    },
    {
      "epoch": 2.471923828125e-05,
      "grad_norm": 0.5204232931137085,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 6480
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 3.1493024826049805,
      "step": 6480
    },
    {
      "dynamics_loss_avg": 0.058209870802238585,
      "epoch": 2.471923828125e-05,
      "step": 6480
    },
    {
      "action_loss_avg": 0.022432719252537935,
      "epoch": 2.471923828125e-05,
      "step": 6480
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 6480,
      "training_step_time": 9.920299291610718
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 2.4312937259674072,
      "step": 6480
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 6480,
      "training_step_time": 7.458814859390259
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 2.428553581237793,
      "step": 6480
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 6480,
      "training_step_time": 7.4578564167022705
    },
    {
      "epoch": 2.471923828125e-05,
      "model_forward_time": 2.6857378482818604,
      "step": 6480
    },
    {
      "epoch": 2.471923828125e-05,
      "step": 6480,
      "training_step_time": 10.284366369247437
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 2.442782402038574,
      "step": 6481
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 6481,
      "training_step_time": 7.466192245483398
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 2.431511402130127,
      "step": 6481
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 6481,
      "training_step_time": 7.452059745788574
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 2.4347808361053467,
      "step": 6481
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 6481,
      "training_step_time": 7.454256057739258
    },
    {
      "epoch": 2.4723052978515625e-05,
      "model_forward_time": 3.13484787940979,
      "step": 6481
    },
    {
      "epoch": 2.4723052978515625e-05,
      "step": 6481,
      "training_step_time": 10.283661603927612
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 3.142477512359619,
      "step": 6482
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 6482,
      "training_step_time": 9.906852722167969
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 2.699626922607422,
      "step": 6482
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 6482,
      "training_step_time": 7.722126722335815
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 3.133127450942993,
      "step": 6482
    },
    {
      "dynamics_loss_avg": 0.05101243183016777,
      "epoch": 2.472686767578125e-05,
      "step": 6482
    },
    {
      "action_loss_avg": 0.02117675666231662,
      "epoch": 2.472686767578125e-05,
      "step": 6482
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 6482,
      "training_step_time": 9.901370763778687
    },
    {
      "epoch": 2.472686767578125e-05,
      "model_forward_time": 3.1353819370269775,
      "step": 6482
    },
    {
      "epoch": 2.472686767578125e-05,
      "step": 6482,
      "training_step_time": 10.022552251815796
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 2.4453728199005127,
      "step": 6483
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 6483,
      "training_step_time": 7.475872755050659
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 3.1331593990325928,
      "step": 6483
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 6483,
      "training_step_time": 9.894721984863281
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 3.4105770587921143,
      "step": 6483
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 6483,
      "training_step_time": 10.180623292922974
    },
    {
      "epoch": 2.4730682373046874e-05,
      "model_forward_time": 3.1367239952087402,
      "step": 6483
    },
    {
      "epoch": 2.4730682373046874e-05,
      "step": 6483,
      "training_step_time": 10.31058382987976
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 3.1665773391723633,
      "step": 6484
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 6484,
      "training_step_time": 9.937621355056763
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 3.1425251960754395,
      "step": 6484
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 6484,
      "training_step_time": 9.911916971206665
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 3.1454317569732666,
      "step": 6484
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 6484,
      "training_step_time": 10.22548532485962
    },
    {
      "epoch": 2.47344970703125e-05,
      "model_forward_time": 2.436891555786133,
      "step": 6484
    },
    {
      "epoch": 2.47344970703125e-05,
      "step": 6484,
      "training_step_time": 10.293940782546997
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 3.146669387817383,
      "step": 6485
    },
    {
      "dynamics_loss_avg": 0.06274044588208198,
      "epoch": 2.4738311767578126e-05,
      "step": 6485
    },
    {
      "action_loss_avg": 0.024619545694440602,
      "epoch": 2.4738311767578126e-05,
      "step": 6485
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 6485,
      "training_step_time": 9.920034408569336
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 2.428598165512085,
      "step": 6485
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 6485,
      "training_step_time": 7.453115224838257
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 2.427232027053833,
      "step": 6485
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 6485,
      "training_step_time": 7.453297138214111
    },
    {
      "epoch": 2.4738311767578126e-05,
      "model_forward_time": 2.4249939918518066,
      "step": 6485
    },
    {
      "epoch": 2.4738311767578126e-05,
      "step": 6485,
      "training_step_time": 10.324417352676392
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 3.4104626178741455,
      "step": 6486
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 6486,
      "training_step_time": 10.181627750396729
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 3.134922504425049,
      "step": 6486
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 6486,
      "training_step_time": 9.90568995475769
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 2.4313998222351074,
      "step": 6486
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 6486,
      "training_step_time": 7.4586687088012695
    },
    {
      "epoch": 2.474212646484375e-05,
      "model_forward_time": 3.1329965591430664,
      "step": 6486
    },
    {
      "epoch": 2.474212646484375e-05,
      "step": 6486,
      "training_step_time": 10.243719816207886
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 3.1514508724212646,
      "step": 6487
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 6487,
      "training_step_time": 9.919560432434082
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 3.402195930480957,
      "step": 6487
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 6487,
      "training_step_time": 10.179269790649414
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 3.12957501411438,
      "step": 6487
    },
    {
      "dynamics_loss_avg": 0.04988362919539213,
      "epoch": 2.4745941162109375e-05,
      "step": 6487
    },
    {
      "action_loss_avg": 0.02127369251102209,
      "epoch": 2.4745941162109375e-05,
      "step": 6487
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 6487,
      "training_step_time": 9.902051210403442
    },
    {
      "epoch": 2.4745941162109375e-05,
      "model_forward_time": 2.431976556777954,
      "step": 6487
    },
    {
      "epoch": 2.4745941162109375e-05,
      "step": 6487,
      "training_step_time": 10.019784927368164
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 2.439038038253784,
      "step": 6488
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 6488,
      "training_step_time": 7.463880777359009
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 2.427764654159546,
      "step": 6488
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 6488,
      "training_step_time": 7.449995040893555
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 3.3782687187194824,
      "step": 6488
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 6488,
      "training_step_time": 10.146941900253296
    },
    {
      "epoch": 2.4749755859375e-05,
      "model_forward_time": 3.133401393890381,
      "step": 6488
    },
    {
      "epoch": 2.4749755859375e-05,
      "step": 6488,
      "training_step_time": 10.2391357421875
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 3.156721353530884,
      "step": 6489
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 6489,
      "training_step_time": 9.940441370010376
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 2.4301671981811523,
      "step": 6489
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 6489,
      "training_step_time": 7.457720041275024
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 3.1360738277435303,
      "step": 6489
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 6489,
      "training_step_time": 9.909586668014526
    },
    {
      "epoch": 2.4753570556640625e-05,
      "model_forward_time": 2.7174978256225586,
      "step": 6489
    },
    {
      "epoch": 2.4753570556640625e-05,
      "step": 6489,
      "training_step_time": 10.30473017692566
    },
    {
      "epoch": 2.475738525390625e-05,
      "grad_norm": 0.31711870431900024,
      "learning_rate": 1e-05,
      "loss": 0.0823,
      "step": 6490
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 3.162186622619629,
      "step": 6490
    },
    {
      "dynamics_loss_avg": 0.07481645462103188,
      "epoch": 2.475738525390625e-05,
      "step": 6490
    },
    {
      "action_loss_avg": 0.028892336739227177,
      "epoch": 2.475738525390625e-05,
      "step": 6490
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 6490,
      "training_step_time": 9.936012506484985
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 2.4319207668304443,
      "step": 6490
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 6490,
      "training_step_time": 7.458245515823364
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 3.141730308532715,
      "step": 6490
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 6490,
      "training_step_time": 9.914397239685059
    },
    {
      "epoch": 2.475738525390625e-05,
      "model_forward_time": 3.1360325813293457,
      "step": 6490
    },
    {
      "epoch": 2.475738525390625e-05,
      "step": 6490,
      "training_step_time": 10.280296802520752
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 3.1714141368865967,
      "step": 6491
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 6491,
      "training_step_time": 10.23505449295044
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 3.1369259357452393,
      "step": 6491
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 6491,
      "training_step_time": 9.909573793411255
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 3.1440727710723877,
      "step": 6491
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 6491,
      "training_step_time": 9.917017221450806
    },
    {
      "epoch": 2.4761199951171874e-05,
      "model_forward_time": 3.135944128036499,
      "step": 6491
    },
    {
      "epoch": 2.4761199951171874e-05,
      "step": 6491,
      "training_step_time": 10.321103811264038
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.455664873123169,
      "step": 6492
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 6492,
      "training_step_time": 7.4851226806640625
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.4340271949768066,
      "step": 6492
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 6492,
      "training_step_time": 7.4605631828308105
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.7348098754882812,
      "step": 6492
    },
    {
      "dynamics_loss_avg": 0.08423349596560001,
      "epoch": 2.47650146484375e-05,
      "step": 6492
    },
    {
      "action_loss_avg": 0.03378619458526373,
      "epoch": 2.47650146484375e-05,
      "step": 6492
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 6492,
      "training_step_time": 7.763651609420776
    },
    {
      "epoch": 2.47650146484375e-05,
      "model_forward_time": 2.4310965538024902,
      "step": 6492
    },
    {
      "epoch": 2.47650146484375e-05,
      "step": 6492,
      "training_step_time": 10.298052787780762
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 2.44657826423645,
      "step": 6493
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 6493,
      "training_step_time": 7.491745710372925
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 3.1378564834594727,
      "step": 6493
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 6493,
      "training_step_time": 9.911224365234375
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 2.433476448059082,
      "step": 6493
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 6493,
      "training_step_time": 7.466224908828735
    },
    {
      "epoch": 2.4768829345703126e-05,
      "model_forward_time": 3.3981220722198486,
      "step": 6493
    },
    {
      "epoch": 2.4768829345703126e-05,
      "step": 6493,
      "training_step_time": 10.272810459136963
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 3.1575841903686523,
      "step": 6494
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 6494,
      "training_step_time": 9.940008878707886
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 3.139293670654297,
      "step": 6494
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 6494,
      "training_step_time": 9.918655633926392
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 3.1380841732025146,
      "step": 6494
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 6494,
      "training_step_time": 9.914260387420654
    },
    {
      "epoch": 2.477264404296875e-05,
      "model_forward_time": 2.4353981018066406,
      "step": 6494
    },
    {
      "epoch": 2.477264404296875e-05,
      "step": 6494,
      "training_step_time": 10.302956342697144
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 2.738192081451416,
      "step": 6495
    },
    {
      "dynamics_loss_avg": 0.04470290597528219,
      "epoch": 2.4776458740234375e-05,
      "step": 6495
    },
    {
      "action_loss_avg": 0.01874893018975854,
      "epoch": 2.4776458740234375e-05,
      "step": 6495
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 6495,
      "training_step_time": 7.773868799209595
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 3.13362455368042,
      "step": 6495
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 6495,
      "training_step_time": 9.907989501953125
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 3.1332502365112305,
      "step": 6495
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 6495,
      "training_step_time": 9.912720441818237
    },
    {
      "epoch": 2.4776458740234375e-05,
      "model_forward_time": 2.4373910427093506,
      "step": 6495
    },
    {
      "epoch": 2.4776458740234375e-05,
      "step": 6495,
      "training_step_time": 10.279329061508179
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 2.446321964263916,
      "step": 6496
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 6496,
      "training_step_time": 7.488542556762695
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 3.4088358879089355,
      "step": 6496
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 6496,
      "training_step_time": 10.179609298706055
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 2.4292805194854736,
      "step": 6496
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 6496,
      "training_step_time": 7.458210706710815
    },
    {
      "epoch": 2.47802734375e-05,
      "model_forward_time": 3.13165020942688,
      "step": 6496
    },
    {
      "epoch": 2.47802734375e-05,
      "step": 6496,
      "training_step_time": 10.344570875167847
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 2.4420182704925537,
      "step": 6497
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 6497,
      "training_step_time": 7.472864627838135
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 3.1285016536712646,
      "step": 6497
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 6497,
      "training_step_time": 9.897526979446411
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 2.427912712097168,
      "step": 6497
    },
    {
      "dynamics_loss_avg": 0.04992790445685387,
      "epoch": 2.4784088134765625e-05,
      "step": 6497
    },
    {
      "action_loss_avg": 0.01844366230070591,
      "epoch": 2.4784088134765625e-05,
      "step": 6497
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 6497,
      "training_step_time": 7.451600074768066
    },
    {
      "epoch": 2.4784088134765625e-05,
      "model_forward_time": 2.689116954803467,
      "step": 6497
    },
    {
      "epoch": 2.4784088134765625e-05,
      "step": 6497,
      "training_step_time": 10.278570175170898
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 1.7544541358947754,
      "step": 6498
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 6498,
      "training_step_time": 5.204066038131714
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 2.4270341396331787,
      "step": 6498
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 6498,
      "training_step_time": 7.447473764419556
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 3.1350836753845215,
      "step": 6498
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 6498,
      "training_step_time": 9.90468168258667
    },
    {
      "epoch": 2.478790283203125e-05,
      "model_forward_time": 3.1286261081695557,
      "step": 6498
    },
    {
      "epoch": 2.478790283203125e-05,
      "step": 6498,
      "training_step_time": 10.012444257736206
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 3.514939785003662,
      "step": 6499
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 6499,
      "training_step_time": 10.291475534439087
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 3.129706621170044,
      "step": 6499
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 6499,
      "training_step_time": 9.907261610031128
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 2.43027400970459,
      "step": 6499
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 6499,
      "training_step_time": 7.46259617805481
    },
    {
      "epoch": 2.4791717529296874e-05,
      "model_forward_time": 3.1310458183288574,
      "step": 6499
    },
    {
      "epoch": 2.4791717529296874e-05,
      "step": 6499,
      "training_step_time": 10.305633068084717
    },
    {
      "epoch": 2.47955322265625e-05,
      "grad_norm": 0.2616739869117737,
      "learning_rate": 1e-05,
      "loss": 0.085,
      "step": 6500
    },
    {
      "epoch": 2.47955322265625e-05,
      "model_forward_time": 5.122313499450684,
      "step": 6500
    },
    {
      "dynamics_loss_avg": 0.06079503726214171,
      "epoch": 2.47955322265625e-05,
      "step": 6500
    },
    {
      "action_loss_avg": 0.022989442083053292,
      "epoch": 2.47955322265625e-05,
      "step": 6500
    },
    {
      "epoch": 2.47955322265625e-05,
      "step": 6500,
      "training_step_time": 11.881048679351807
    },
    {
      "epoch": 2.47955322265625e-05,
      "model_forward_time": 3.149569272994995,
      "step": 6500
    },
    {
      "epoch": 2.47955322265625e-05,
      "step": 6500,
      "training_step_time": 9.909728288650513
    },
    {
      "epoch": 2.47955322265625e-05,
      "model_forward_time": 3.124569892883301,
      "step": 6500
    },
    {
      "epoch": 2.47955322265625e-05,
      "step": 6500,
      "training_step_time": 9.881812810897827
    },
    {
      "epoch": 2.47955322265625e-05,
      "model_forward_time": 3.125020742416382,
      "step": 6500
    },
    {
      "epoch": 2.47955322265625e-05,
      "step": 6500,
      "training_step_time": 10.27381944656372
    },
    {
      "epoch": 2.4799346923828126e-05,
      "model_forward_time": 3.1556925773620605,
      "step": 6501
    },
    {
      "epoch": 2.4799346923828126e-05,
      "step": 6501,
      "training_step_time": 9.920418977737427
    },
    {
      "epoch": 2.4799346923828126e-05,
      "model_forward_time": 3.3905129432678223,
      "step": 6501
    },
    {
      "epoch": 2.4799346923828126e-05,
      "step": 6501,
      "training_step_time": 10.15393614768982
    },
    {
      "epoch": 2.4799346923828126e-05,
      "model_forward_time": 3.140214681625366,
      "step": 6501
    },
    {
      "epoch": 2.4799346923828126e-05,
      "step": 6501,
      "training_step_time": 9.913255453109741
    },
    {
      "epoch": 2.4799346923828126e-05,
      "model_forward_time": 3.1272971630096436,
      "step": 6501
    },
    {
      "epoch": 2.4799346923828126e-05,
      "step": 6501,
      "training_step_time": 10.258569955825806
    },
    {
      "epoch": 2.480316162109375e-05,
      "model_forward_time": 3.1972475051879883,
      "step": 6502
    },
    {
      "epoch": 2.480316162109375e-05,
      "step": 6502,
      "training_step_time": 9.97568655014038
    },
    {
      "epoch": 2.480316162109375e-05,
      "model_forward_time": 3.146547317504883,
      "step": 6502
    },
    {
      "epoch": 2.480316162109375e-05,
      "step": 6502,
      "training_step_time": 9.925827264785767
    },
    {
      "epoch": 2.480316162109375e-05,
      "model_forward_time": 3.149683952331543,
      "step": 6502
    },
    {
      "dynamics_loss_avg": 0.05099691916257143,
      "epoch": 2.480316162109375e-05,
      "step": 6502
    },
    {
      "action_loss_avg": 0.018880266323685647,
      "epoch": 2.480316162109375e-05,
      "step": 6502
    },
    {
      "epoch": 2.480316162109375e-05,
      "step": 6502,
      "training_step_time": 10.259975671768188
    },
    {
      "epoch": 2.480316162109375e-05,
      "model_forward_time": 3.141042947769165,
      "step": 6502
    },
    {
      "epoch": 2.480316162109375e-05,
      "step": 6502,
      "training_step_time": 10.290585994720459
    },
    {
      "epoch": 2.4806976318359375e-05,
      "model_forward_time": 3.1924448013305664,
      "step": 6503
    },
    {
      "epoch": 2.4806976318359375e-05,
      "step": 6503,
      "training_step_time": 9.975447654724121
    },
    {
      "epoch": 2.4806976318359375e-05,
      "model_forward_time": 3.1436238288879395,
      "step": 6503
    },
    {
      "epoch": 2.4806976318359375e-05,
      "step": 6503,
      "training_step_time": 9.926406621932983
    },
    {
      "epoch": 2.4806976318359375e-05,
      "model_forward_time": 3.1407289505004883,
      "step": 6503
    },
    {
      "epoch": 2.4806976318359375e-05,
      "step": 6503,
      "training_step_time": 9.9250009059906
    },
    {
      "epoch": 2.4806976318359375e-05,
      "model_forward_time": 3.142080068588257,
      "step": 6503
    },
    {
      "epoch": 2.4806976318359375e-05,
      "step": 6503,
      "training_step_time": 10.329949140548706
    },
    {
      "epoch": 2.4810791015625e-05,
      "model_forward_time": 3.1926658153533936,
      "step": 6504
    },
    {
      "epoch": 2.4810791015625e-05,
      "step": 6504,
      "training_step_time": 9.975889444351196
    },
    {
      "epoch": 2.4810791015625e-05,
      "model_forward_time": 3.1476757526397705,
      "step": 6504
    },
    {
      "epoch": 2.4810791015625e-05,
      "step": 6504,
      "training_step_time": 9.93821382522583
    },
    {
      "epoch": 2.4810791015625e-05,
      "model_forward_time": 3.138947010040283,
      "step": 6504
    },
    {
      "epoch": 2.4810791015625e-05,
      "step": 6504,
      "training_step_time": 9.919323205947876
    },
    {
      "epoch": 2.4810791015625e-05,
      "model_forward_time": 3.145841121673584,
      "step": 6504
    },
    {
      "epoch": 2.4810791015625e-05,
      "step": 6504,
      "training_step_time": 10.282219886779785
    },
    {
      "epoch": 2.4814605712890625e-05,
      "model_forward_time": 3.1684410572052,
      "step": 6505
    },
    {
      "dynamics_loss_avg": 0.05477816425263882,
      "epoch": 2.4814605712890625e-05,
      "step": 6505
    },
    {
      "action_loss_avg": 0.030669789388775827,
      "epoch": 2.4814605712890625e-05,
      "step": 6505
    },
    {
      "epoch": 2.4814605712890625e-05,
      "step": 6505,
      "training_step_time": 10.237046480178833
    },
    {
      "epoch": 2.4814605712890625e-05,
      "model_forward_time": 3.140164613723755,
      "step": 6505
    },
    {
      "epoch": 2.4814605712890625e-05,
      "step": 6505,
      "training_step_time": 9.962224006652832
    },
    {
      "epoch": 2.4814605712890625e-05,
      "model_forward_time": 3.136916160583496,
      "step": 6505
    },
    {
      "epoch": 2.4814605712890625e-05,
      "step": 6505,
      "training_step_time": 9.91751480102539
    },
    {
      "epoch": 2.4814605712890625e-05,
      "model_forward_time": 3.1340928077697754,
      "step": 6505
    },
    {
      "epoch": 2.4814605712890625e-05,
      "step": 6505,
      "training_step_time": 10.290437698364258
    },
    {
      "epoch": 2.481842041015625e-05,
      "model_forward_time": 3.161855459213257,
      "step": 6506
    },
    {
      "epoch": 2.481842041015625e-05,
      "step": 6506,
      "training_step_time": 9.937817335128784
    },
    {
      "epoch": 2.481842041015625e-05,
      "model_forward_time": 3.4733738899230957,
      "step": 6506
    },
    {
      "epoch": 2.481842041015625e-05,
      "step": 6506,
      "training_step_time": 10.252280950546265
    },
    {
      "epoch": 2.481842041015625e-05,
      "model_forward_time": 3.1399054527282715,
      "step": 6506
    },
    {
      "epoch": 2.481842041015625e-05,
      "step": 6506,
      "training_step_time": 9.914914608001709
    },
    {
      "epoch": 2.481842041015625e-05,
      "model_forward_time": 3.13655161857605,
      "step": 6506
    },
    {
      "epoch": 2.481842041015625e-05,
      "step": 6506,
      "training_step_time": 10.308945417404175
    },
    {
      "epoch": 2.4822235107421874e-05,
      "model_forward_time": 3.1784324645996094,
      "step": 6507
    },
    {
      "epoch": 2.4822235107421874e-05,
      "step": 6507,
      "training_step_time": 9.959902048110962
    },
    {
      "epoch": 2.4822235107421874e-05,
      "model_forward_time": 3.1362264156341553,
      "step": 6507
    },
    {
      "epoch": 2.4822235107421874e-05,
      "step": 6507,
      "training_step_time": 9.911812782287598
    },
    {
      "epoch": 2.4822235107421874e-05,
      "model_forward_time": 3.1374497413635254,
      "step": 6507
    },
    {
      "dynamics_loss_avg": 0.05494870813563466,
      "epoch": 2.4822235107421874e-05,
      "step": 6507
    },
    {
      "action_loss_avg": 0.027481347834691406,
      "epoch": 2.4822235107421874e-05,
      "step": 6507
    },
    {
      "epoch": 2.4822235107421874e-05,
      "step": 6507,
      "training_step_time": 10.190999746322632
    },
    {
      "epoch": 2.4822235107421874e-05,
      "model_forward_time": 3.1406614780426025,
      "step": 6507
    },
    {
      "epoch": 2.4822235107421874e-05,
      "step": 6507,
      "training_step_time": 10.316374063491821
    },
    {
      "epoch": 2.48260498046875e-05,
      "model_forward_time": 3.1859915256500244,
      "step": 6508
    },
    {
      "epoch": 2.48260498046875e-05,
      "step": 6508,
      "training_step_time": 9.96588921546936
    },
    {
      "epoch": 2.48260498046875e-05,
      "model_forward_time": 3.1414403915405273,
      "step": 6508
    },
    {
      "epoch": 2.48260498046875e-05,
      "step": 6508,
      "training_step_time": 9.918733596801758
    },
    {
      "epoch": 2.48260498046875e-05,
      "model_forward_time": 3.151240587234497,
      "step": 6508
    },
    {
      "epoch": 2.48260498046875e-05,
      "step": 6508,
      "training_step_time": 9.930618047714233
    },
    {
      "epoch": 2.48260498046875e-05,
      "model_forward_time": 3.4419727325439453,
      "step": 6508
    },
    {
      "epoch": 2.48260498046875e-05,
      "step": 6508,
      "training_step_time": 10.326850175857544
    },
    {
      "epoch": 2.4829864501953126e-05,
      "model_forward_time": 3.2009925842285156,
      "step": 6509
    },
    {
      "epoch": 2.4829864501953126e-05,
      "step": 6509,
      "training_step_time": 9.982897996902466
    },
    {
      "epoch": 2.4829864501953126e-05,
      "model_forward_time": 3.152527332305908,
      "step": 6509
    },
    {
      "epoch": 2.4829864501953126e-05,
      "step": 6509,
      "training_step_time": 9.933713912963867
    },
    {
      "epoch": 2.4829864501953126e-05,
      "model_forward_time": 3.150129556655884,
      "step": 6509
    },
    {
      "epoch": 2.4829864501953126e-05,
      "step": 6509,
      "training_step_time": 9.93379282951355
    },
    {
      "epoch": 2.4829864501953126e-05,
      "model_forward_time": 3.1504452228546143,
      "step": 6509
    },
    {
      "epoch": 2.4829864501953126e-05,
      "step": 6509,
      "training_step_time": 10.032030820846558
    },
    {
      "epoch": 2.483367919921875e-05,
      "grad_norm": 0.2820837199687958,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 6510
    },
    {
      "epoch": 2.483367919921875e-05,
      "model_forward_time": 3.447247266769409,
      "step": 6510
    },
    {
      "dynamics_loss_avg": 0.05871667489409447,
      "epoch": 2.483367919921875e-05,
      "step": 6510
    },
    {
      "action_loss_avg": 0.024872958939522506,
      "epoch": 2.483367919921875e-05,
      "step": 6510
    },
    {
      "epoch": 2.483367919921875e-05,
      "step": 6510,
      "training_step_time": 10.228892087936401
    },
    {
      "epoch": 2.483367919921875e-05,
      "model_forward_time": 3.142133951187134,
      "step": 6510
    },
    {
      "epoch": 2.483367919921875e-05,
      "step": 6510,
      "training_step_time": 9.921537637710571
    },
    {
      "epoch": 2.483367919921875e-05,
      "model_forward_time": 3.1370339393615723,
      "step": 6510
    },
    {
      "epoch": 2.483367919921875e-05,
      "step": 6510,
      "training_step_time": 9.918559789657593
    },
    {
      "epoch": 2.483367919921875e-05,
      "model_forward_time": 3.1390156745910645,
      "step": 6510
    },
    {
      "epoch": 2.483367919921875e-05,
      "step": 6510,
      "training_step_time": 10.270082950592041
    },
    {
      "epoch": 2.4837493896484375e-05,
      "model_forward_time": 3.1806857585906982,
      "step": 6511
    },
    {
      "epoch": 2.4837493896484375e-05,
      "step": 6511,
      "training_step_time": 9.96262812614441
    },
    {
      "epoch": 2.4837493896484375e-05,
      "model_forward_time": 3.145519971847534,
      "step": 6511
    },
    {
      "epoch": 2.4837493896484375e-05,
      "step": 6511,
      "training_step_time": 10.277664184570312
    },
    {
      "epoch": 2.4837493896484375e-05,
      "model_forward_time": 3.146254301071167,
      "step": 6511
    },
    {
      "epoch": 2.4837493896484375e-05,
      "step": 6511,
      "training_step_time": 9.93052339553833
    },
    {
      "epoch": 2.4837493896484375e-05,
      "model_forward_time": 3.1462020874023438,
      "step": 6511
    },
    {
      "epoch": 2.4837493896484375e-05,
      "step": 6511,
      "training_step_time": 10.023237466812134
    },
    {
      "epoch": 2.484130859375e-05,
      "model_forward_time": 3.168949604034424,
      "step": 6512
    },
    {
      "epoch": 2.484130859375e-05,
      "step": 6512,
      "training_step_time": 9.953436374664307
    },
    {
      "epoch": 2.484130859375e-05,
      "model_forward_time": 3.137437582015991,
      "step": 6512
    },
    {
      "epoch": 2.484130859375e-05,
      "step": 6512,
      "training_step_time": 9.918753623962402
    },
    {
      "epoch": 2.484130859375e-05,
      "model_forward_time": 3.4063076972961426,
      "step": 6512
    },
    {
      "dynamics_loss_avg": 0.0561911491677165,
      "epoch": 2.484130859375e-05,
      "step": 6512
    },
    {
      "action_loss_avg": 0.02636198718100786,
      "epoch": 2.484130859375e-05,
      "step": 6512
    },
    {
      "epoch": 2.484130859375e-05,
      "step": 6512,
      "training_step_time": 10.186829566955566
    },
    {
      "epoch": 2.484130859375e-05,
      "model_forward_time": 3.1402060985565186,
      "step": 6512
    },
    {
      "epoch": 2.484130859375e-05,
      "step": 6512,
      "training_step_time": 10.282232284545898
    },
    {
      "epoch": 2.4845123291015625e-05,
      "model_forward_time": 3.161391258239746,
      "step": 6513
    },
    {
      "epoch": 2.4845123291015625e-05,
      "step": 6513,
      "training_step_time": 9.938506603240967
    },
    {
      "epoch": 2.4845123291015625e-05,
      "model_forward_time": 3.132683038711548,
      "step": 6513
    },
    {
      "epoch": 2.4845123291015625e-05,
      "step": 6513,
      "training_step_time": 9.912211894989014
    },
    {
      "epoch": 2.4845123291015625e-05,
      "model_forward_time": 3.141731023788452,
      "step": 6513
    },
    {
      "epoch": 2.4845123291015625e-05,
      "step": 6513,
      "training_step_time": 9.918263673782349
    },
    {
      "epoch": 2.4845123291015625e-05,
      "model_forward_time": 3.1352672576904297,
      "step": 6513
    },
    {
      "epoch": 2.4845123291015625e-05,
      "step": 6513,
      "training_step_time": 10.311527490615845
    },
    {
      "epoch": 2.484893798828125e-05,
      "model_forward_time": 3.1783056259155273,
      "step": 6514
    },
    {
      "epoch": 2.484893798828125e-05,
      "step": 6514,
      "training_step_time": 9.9590322971344
    },
    {
      "epoch": 2.484893798828125e-05,
      "model_forward_time": 3.133686065673828,
      "step": 6514
    },
    {
      "epoch": 2.484893798828125e-05,
      "step": 6514,
      "training_step_time": 9.911367654800415
    },
    {
      "epoch": 2.484893798828125e-05,
      "model_forward_time": 3.1423535346984863,
      "step": 6514
    },
    {
      "epoch": 2.484893798828125e-05,
      "step": 6514,
      "training_step_time": 9.920804738998413
    },
    {
      "epoch": 2.484893798828125e-05,
      "model_forward_time": 3.1396260261535645,
      "step": 6514
    },
    {
      "epoch": 2.484893798828125e-05,
      "step": 6514,
      "training_step_time": 10.016656637191772
    },
    {
      "epoch": 2.4852752685546874e-05,
      "model_forward_time": 3.464099884033203,
      "step": 6515
    },
    {
      "dynamics_loss_avg": 0.06882292814552785,
      "epoch": 2.4852752685546874e-05,
      "step": 6515
    },
    {
      "action_loss_avg": 0.024057985097169877,
      "epoch": 2.4852752685546874e-05,
      "step": 6515
    },
    {
      "epoch": 2.4852752685546874e-05,
      "step": 6515,
      "training_step_time": 10.260071516036987
    },
    {
      "epoch": 2.4852752685546874e-05,
      "model_forward_time": 3.145951747894287,
      "step": 6515
    },
    {
      "epoch": 2.4852752685546874e-05,
      "step": 6515,
      "training_step_time": 9.930821180343628
    },
    {
      "epoch": 2.4852752685546874e-05,
      "model_forward_time": 3.1512575149536133,
      "step": 6515
    },
    {
      "epoch": 2.4852752685546874e-05,
      "step": 6515,
      "training_step_time": 9.935216903686523
    },
    {
      "epoch": 2.4852752685546874e-05,
      "model_forward_time": 3.143904685974121,
      "step": 6515
    },
    {
      "epoch": 2.4852752685546874e-05,
      "step": 6515,
      "training_step_time": 10.33791995048523
    },
    {
      "epoch": 2.48565673828125e-05,
      "model_forward_time": 3.1868131160736084,
      "step": 6516
    },
    {
      "epoch": 2.48565673828125e-05,
      "step": 6516,
      "training_step_time": 9.971636295318604
    },
    {
      "epoch": 2.48565673828125e-05,
      "model_forward_time": 3.140096664428711,
      "step": 6516
    },
    {
      "epoch": 2.48565673828125e-05,
      "step": 6516,
      "training_step_time": 10.18598461151123
    },
    {
      "epoch": 2.48565673828125e-05,
      "model_forward_time": 3.141947031021118,
      "step": 6516
    },
    {
      "epoch": 2.48565673828125e-05,
      "step": 6516,
      "training_step_time": 9.919095516204834
    },
    {
      "epoch": 2.48565673828125e-05,
      "model_forward_time": 3.1403400897979736,
      "step": 6516
    },
    {
      "epoch": 2.48565673828125e-05,
      "step": 6516,
      "training_step_time": 10.030204772949219
    },
    {
      "epoch": 2.4860382080078126e-05,
      "model_forward_time": 3.1832950115203857,
      "step": 6517
    },
    {
      "epoch": 2.4860382080078126e-05,
      "step": 6517,
      "training_step_time": 9.957638025283813
    },
    {
      "epoch": 2.4860382080078126e-05,
      "model_forward_time": 3.139042854309082,
      "step": 6517
    },
    {
      "epoch": 2.4860382080078126e-05,
      "step": 6517,
      "training_step_time": 9.913773775100708
    },
    {
      "epoch": 2.4860382080078126e-05,
      "model_forward_time": 3.453989028930664,
      "step": 6517
    },
    {
      "dynamics_loss_avg": 0.06669493354856967,
      "epoch": 2.4860382080078126e-05,
      "step": 6517
    },
    {
      "action_loss_avg": 0.029927856801077724,
      "epoch": 2.4860382080078126e-05,
      "step": 6517
    },
    {
      "epoch": 2.4860382080078126e-05,
      "step": 6517,
      "training_step_time": 10.231307744979858
    },
    {
      "epoch": 2.4860382080078126e-05,
      "model_forward_time": 3.135287284851074,
      "step": 6517
    },
    {
      "epoch": 2.4860382080078126e-05,
      "step": 6517,
      "training_step_time": 10.298294305801392
    },
    {
      "epoch": 2.486419677734375e-05,
      "model_forward_time": 3.1935391426086426,
      "step": 6518
    },
    {
      "epoch": 2.486419677734375e-05,
      "step": 6518,
      "training_step_time": 9.973554372787476
    },
    {
      "epoch": 2.486419677734375e-05,
      "model_forward_time": 3.143507480621338,
      "step": 6518
    },
    {
      "epoch": 2.486419677734375e-05,
      "step": 6518,
      "training_step_time": 9.92756199836731
    },
    {
      "epoch": 2.486419677734375e-05,
      "model_forward_time": 3.1393516063690186,
      "step": 6518
    },
    {
      "epoch": 2.486419677734375e-05,
      "step": 6518,
      "training_step_time": 9.919427394866943
    },
    {
      "epoch": 2.486419677734375e-05,
      "model_forward_time": 3.412559747695923,
      "step": 6518
    },
    {
      "epoch": 2.486419677734375e-05,
      "step": 6518,
      "training_step_time": 10.36893892288208
    },
    {
      "epoch": 2.4868011474609375e-05,
      "model_forward_time": 3.171565294265747,
      "step": 6519
    },
    {
      "epoch": 2.4868011474609375e-05,
      "step": 6519,
      "training_step_time": 9.949494361877441
    },
    {
      "epoch": 2.4868011474609375e-05,
      "model_forward_time": 3.1372549533843994,
      "step": 6519
    },
    {
      "epoch": 2.4868011474609375e-05,
      "step": 6519,
      "training_step_time": 9.91296648979187
    },
    {
      "epoch": 2.4868011474609375e-05,
      "model_forward_time": 3.1336965560913086,
      "step": 6519
    },
    {
      "epoch": 2.4868011474609375e-05,
      "step": 6519,
      "training_step_time": 9.907312631607056
    },
    {
      "epoch": 2.4868011474609375e-05,
      "model_forward_time": 3.1425318717956543,
      "step": 6519
    },
    {
      "epoch": 2.4868011474609375e-05,
      "step": 6519,
      "training_step_time": 10.251027822494507
    },
    {
      "epoch": 2.4871826171875e-05,
      "grad_norm": 0.4601542055606842,
      "learning_rate": 1e-05,
      "loss": 0.0868,
      "step": 6520
    },
    {
      "epoch": 2.4871826171875e-05,
      "model_forward_time": 3.430803060531616,
      "step": 6520
    },
    {
      "dynamics_loss_avg": 0.06443827580660581,
      "epoch": 2.4871826171875e-05,
      "step": 6520
    },
    {
      "action_loss_avg": 0.02258431278169155,
      "epoch": 2.4871826171875e-05,
      "step": 6520
    },
    {
      "epoch": 2.4871826171875e-05,
      "step": 6520,
      "training_step_time": 10.205164432525635
    },
    {
      "epoch": 2.4871826171875e-05,
      "model_forward_time": 3.1300621032714844,
      "step": 6520
    },
    {
      "epoch": 2.4871826171875e-05,
      "step": 6520,
      "training_step_time": 9.897566556930542
    },
    {
      "epoch": 2.4871826171875e-05,
      "model_forward_time": 3.1313226222991943,
      "step": 6520
    },
    {
      "epoch": 2.4871826171875e-05,
      "step": 6520,
      "training_step_time": 9.898669481277466
    },
    {
      "epoch": 2.4871826171875e-05,
      "model_forward_time": 2.428698778152466,
      "step": 6520
    },
    {
      "epoch": 2.4871826171875e-05,
      "step": 6520,
      "training_step_time": 10.242719888687134
    },
    {
      "epoch": 2.4875640869140625e-05,
      "model_forward_time": 2.4643361568450928,
      "step": 6521
    },
    {
      "epoch": 2.4875640869140625e-05,
      "step": 6521,
      "training_step_time": 7.4950852394104
    },
    {
      "epoch": 2.4875640869140625e-05,
      "model_forward_time": 3.1334891319274902,
      "step": 6521
    },
    {
      "epoch": 2.4875640869140625e-05,
      "step": 6521,
      "training_step_time": 9.908243417739868
    },
    {
      "epoch": 2.4875640869140625e-05,
      "model_forward_time": 3.382777214050293,
      "step": 6521
    },
    {
      "epoch": 2.4875640869140625e-05,
      "step": 6521,
      "training_step_time": 10.159925937652588
    },
    {
      "epoch": 2.4875640869140625e-05,
      "model_forward_time": 3.1390321254730225,
      "step": 6521
    },
    {
      "epoch": 2.4875640869140625e-05,
      "step": 6521,
      "training_step_time": 10.039255142211914
    },
    {
      "epoch": 2.487945556640625e-05,
      "model_forward_time": 3.183655023574829,
      "step": 6522
    },
    {
      "epoch": 2.487945556640625e-05,
      "step": 6522,
      "training_step_time": 9.963849544525146
    },
    {
      "epoch": 2.487945556640625e-05,
      "model_forward_time": 3.1406726837158203,
      "step": 6522
    },
    {
      "epoch": 2.487945556640625e-05,
      "step": 6522,
      "training_step_time": 9.922961473464966
    },
    {
      "epoch": 2.487945556640625e-05,
      "model_forward_time": 3.146235227584839,
      "step": 6522
    },
    {
      "dynamics_loss_avg": 0.05949754016473889,
      "epoch": 2.487945556640625e-05,
      "step": 6522
    },
    {
      "action_loss_avg": 0.02286711020860821,
      "epoch": 2.487945556640625e-05,
      "step": 6522
    },
    {
      "epoch": 2.487945556640625e-05,
      "step": 6522,
      "training_step_time": 10.226548671722412
    },
    {
      "epoch": 2.487945556640625e-05,
      "model_forward_time": 3.145820140838623,
      "step": 6522
    },
    {
      "epoch": 2.487945556640625e-05,
      "step": 6522,
      "training_step_time": 10.32503342628479
    },
    {
      "epoch": 2.4883270263671874e-05,
      "model_forward_time": 3.1738977432250977,
      "step": 6523
    },
    {
      "epoch": 2.4883270263671874e-05,
      "step": 6523,
      "training_step_time": 9.954161167144775
    },
    {
      "epoch": 2.4883270263671874e-05,
      "model_forward_time": 3.1384520530700684,
      "step": 6523
    },
    {
      "epoch": 2.4883270263671874e-05,
      "step": 6523,
      "training_step_time": 9.916620016098022
    },
    {
      "epoch": 2.4883270263671874e-05,
      "model_forward_time": 3.13541579246521,
      "step": 6523
    },
    {
      "epoch": 2.4883270263671874e-05,
      "step": 6523,
      "training_step_time": 9.914936304092407
    },
    {
      "epoch": 2.4883270263671874e-05,
      "model_forward_time": 3.419234275817871,
      "step": 6523
    },
    {
      "epoch": 2.4883270263671874e-05,
      "step": 6523,
      "training_step_time": 10.291921854019165
    },
    {
      "epoch": 2.48870849609375e-05,
      "model_forward_time": 3.176194190979004,
      "step": 6524
    },
    {
      "epoch": 2.48870849609375e-05,
      "step": 6524,
      "training_step_time": 9.948403358459473
    },
    {
      "epoch": 2.48870849609375e-05,
      "model_forward_time": 3.1342194080352783,
      "step": 6524
    },
    {
      "epoch": 2.48870849609375e-05,
      "step": 6524,
      "training_step_time": 9.904353618621826
    },
    {
      "epoch": 2.48870849609375e-05,
      "model_forward_time": 3.137446880340576,
      "step": 6524
    },
    {
      "epoch": 2.48870849609375e-05,
      "step": 6524,
      "training_step_time": 9.909759283065796
    },
    {
      "epoch": 2.48870849609375e-05,
      "model_forward_time": 2.436901092529297,
      "step": 6524
    },
    {
      "epoch": 2.48870849609375e-05,
      "step": 6524,
      "training_step_time": 10.294250011444092
    },
    {
      "epoch": 2.4890899658203126e-05,
      "model_forward_time": 3.1603922843933105,
      "step": 6525
    },
    {
      "dynamics_loss_avg": 0.05033228993415832,
      "epoch": 2.4890899658203126e-05,
      "step": 6525
    },
    {
      "action_loss_avg": 0.020384465949609874,
      "epoch": 2.4890899658203126e-05,
      "step": 6525
    },
    {
      "epoch": 2.4890899658203126e-05,
      "step": 6525,
      "training_step_time": 9.93110728263855
    },
    {
      "epoch": 2.4890899658203126e-05,
      "model_forward_time": 3.3835837841033936,
      "step": 6525
    },
    {
      "epoch": 2.4890899658203126e-05,
      "step": 6525,
      "training_step_time": 10.156479358673096
    },
    {
      "epoch": 2.4890899658203126e-05,
      "model_forward_time": 3.135080337524414,
      "step": 6525
    },
    {
      "epoch": 2.4890899658203126e-05,
      "step": 6525,
      "training_step_time": 9.913028955459595
    },
    {
      "epoch": 2.4890899658203126e-05,
      "model_forward_time": 3.1357154846191406,
      "step": 6525
    },
    {
      "epoch": 2.4890899658203126e-05,
      "step": 6525,
      "training_step_time": 10.260993719100952
    },
    {
      "epoch": 2.489471435546875e-05,
      "model_forward_time": 3.1964216232299805,
      "step": 6526
    },
    {
      "epoch": 2.489471435546875e-05,
      "step": 6526,
      "training_step_time": 9.9909029006958
    },
    {
      "epoch": 2.489471435546875e-05,
      "model_forward_time": 3.462247133255005,
      "step": 6526
    },
    {
      "epoch": 2.489471435546875e-05,
      "step": 6526,
      "training_step_time": 10.237201929092407
    },
    {
      "epoch": 2.489471435546875e-05,
      "model_forward_time": 3.1470463275909424,
      "step": 6526
    },
    {
      "epoch": 2.489471435546875e-05,
      "step": 6526,
      "training_step_time": 9.926275730133057
    },
    {
      "epoch": 2.489471435546875e-05,
      "model_forward_time": 3.1457924842834473,
      "step": 6526
    },
    {
      "epoch": 2.489471435546875e-05,
      "step": 6526,
      "training_step_time": 10.035005331039429
    },
    {
      "epoch": 2.4898529052734376e-05,
      "model_forward_time": 3.1729750633239746,
      "step": 6527
    },
    {
      "epoch": 2.4898529052734376e-05,
      "step": 6527,
      "training_step_time": 9.95943808555603
    },
    {
      "epoch": 2.4898529052734376e-05,
      "model_forward_time": 3.1469597816467285,
      "step": 6527
    },
    {
      "epoch": 2.4898529052734376e-05,
      "step": 6527,
      "training_step_time": 9.928022623062134
    },
    {
      "epoch": 2.4898529052734376e-05,
      "model_forward_time": 3.1423757076263428,
      "step": 6527
    },
    {
      "dynamics_loss_avg": 0.049868234619498256,
      "epoch": 2.4898529052734376e-05,
      "step": 6527
    },
    {
      "action_loss_avg": 0.021289576729759573,
      "epoch": 2.4898529052734376e-05,
      "step": 6527
    },
    {
      "epoch": 2.4898529052734376e-05,
      "step": 6527,
      "training_step_time": 10.183328628540039
    },
    {
      "epoch": 2.4898529052734376e-05,
      "model_forward_time": 3.1400985717773438,
      "step": 6527
    },
    {
      "epoch": 2.4898529052734376e-05,
      "step": 6527,
      "training_step_time": 10.300721406936646
    },
    {
      "epoch": 2.490234375e-05,
      "model_forward_time": 3.172760009765625,
      "step": 6528
    },
    {
      "epoch": 2.490234375e-05,
      "step": 6528,
      "training_step_time": 9.954661846160889
    },
    {
      "epoch": 2.490234375e-05,
      "model_forward_time": 3.1427791118621826,
      "step": 6528
    },
    {
      "epoch": 2.490234375e-05,
      "step": 6528,
      "training_step_time": 9.921117544174194
    },
    {
      "epoch": 2.490234375e-05,
      "model_forward_time": 3.139385938644409,
      "step": 6528
    },
    {
      "epoch": 2.490234375e-05,
      "step": 6528,
      "training_step_time": 9.917860507965088
    },
    {
      "epoch": 2.490234375e-05,
      "model_forward_time": 3.3925271034240723,
      "step": 6528
    },
    {
      "epoch": 2.490234375e-05,
      "step": 6528,
      "training_step_time": 10.313058614730835
    },
    {
      "epoch": 2.4906158447265625e-05,
      "model_forward_time": 3.1927382946014404,
      "step": 6529
    },
    {
      "epoch": 2.4906158447265625e-05,
      "step": 6529,
      "training_step_time": 9.968506097793579
    },
    {
      "epoch": 2.4906158447265625e-05,
      "model_forward_time": 3.143563747406006,
      "step": 6529
    },
    {
      "epoch": 2.4906158447265625e-05,
      "step": 6529,
      "training_step_time": 9.921393632888794
    },
    {
      "epoch": 2.4906158447265625e-05,
      "model_forward_time": 3.1381242275238037,
      "step": 6529
    },
    {
      "epoch": 2.4906158447265625e-05,
      "step": 6529,
      "training_step_time": 9.914454936981201
    },
    {
      "epoch": 2.4906158447265625e-05,
      "model_forward_time": 3.138183355331421,
      "step": 6529
    },
    {
      "epoch": 2.4906158447265625e-05,
      "step": 6529,
      "training_step_time": 10.260056257247925
    },
    {
      "epoch": 2.490997314453125e-05,
      "grad_norm": 0.46899691224098206,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 6530
    },
    {
      "epoch": 2.490997314453125e-05,
      "model_forward_time": 3.477496385574341,
      "step": 6530
    },
    {
      "dynamics_loss_avg": 0.07383682914078235,
      "epoch": 2.490997314453125e-05,
      "step": 6530
    },
    {
      "action_loss_avg": 0.030785509385168552,
      "epoch": 2.490997314453125e-05,
      "step": 6530
    },
    {
      "epoch": 2.490997314453125e-05,
      "step": 6530,
      "training_step_time": 10.251405477523804
    },
    {
      "epoch": 2.490997314453125e-05,
      "model_forward_time": 3.148921251296997,
      "step": 6530
    },
    {
      "epoch": 2.490997314453125e-05,
      "step": 6530,
      "training_step_time": 9.922120332717896
    },
    {
      "epoch": 2.490997314453125e-05,
      "model_forward_time": 3.14603590965271,
      "step": 6530
    },
    {
      "epoch": 2.490997314453125e-05,
      "step": 6530,
      "training_step_time": 9.925703525543213
    },
    {
      "epoch": 2.490997314453125e-05,
      "model_forward_time": 3.1460609436035156,
      "step": 6530
    },
    {
      "epoch": 2.490997314453125e-05,
      "step": 6530,
      "training_step_time": 10.251485824584961
    },
    {
      "epoch": 2.4913787841796874e-05,
      "model_forward_time": 3.1718432903289795,
      "step": 6531
    },
    {
      "epoch": 2.4913787841796874e-05,
      "step": 6531,
      "training_step_time": 9.954935073852539
    },
    {
      "epoch": 2.4913787841796874e-05,
      "model_forward_time": 3.43450927734375,
      "step": 6531
    },
    {
      "epoch": 2.4913787841796874e-05,
      "step": 6531,
      "training_step_time": 10.209270238876343
    },
    {
      "epoch": 2.4913787841796874e-05,
      "model_forward_time": 3.145686388015747,
      "step": 6531
    },
    {
      "epoch": 2.4913787841796874e-05,
      "step": 6531,
      "training_step_time": 9.922436952590942
    },
    {
      "epoch": 2.4913787841796874e-05,
      "model_forward_time": 3.1474931240081787,
      "step": 6531
    },
    {
      "epoch": 2.4913787841796874e-05,
      "step": 6531,
      "training_step_time": 10.020143747329712
    },
    {
      "epoch": 2.49176025390625e-05,
      "model_forward_time": 3.195453643798828,
      "step": 6532
    },
    {
      "epoch": 2.49176025390625e-05,
      "step": 6532,
      "training_step_time": 9.973893404006958
    },
    {
      "epoch": 2.49176025390625e-05,
      "model_forward_time": 3.146233320236206,
      "step": 6532
    },
    {
      "epoch": 2.49176025390625e-05,
      "step": 6532,
      "training_step_time": 9.926294803619385
    },
    {
      "epoch": 2.49176025390625e-05,
      "model_forward_time": 3.4037890434265137,
      "step": 6532
    },
    {
      "dynamics_loss_avg": 0.06336637027561665,
      "epoch": 2.49176025390625e-05,
      "step": 6532
    },
    {
      "action_loss_avg": 0.023598400026094167,
      "epoch": 2.49176025390625e-05,
      "step": 6532
    },
    {
      "epoch": 2.49176025390625e-05,
      "step": 6532,
      "training_step_time": 10.186798095703125
    },
    {
      "epoch": 2.49176025390625e-05,
      "model_forward_time": 3.147010326385498,
      "step": 6532
    },
    {
      "epoch": 2.49176025390625e-05,
      "step": 6532,
      "training_step_time": 10.32136845588684
    },
    {
      "epoch": 2.4921417236328126e-05,
      "model_forward_time": 3.196787118911743,
      "step": 6533
    },
    {
      "epoch": 2.4921417236328126e-05,
      "step": 6533,
      "training_step_time": 9.973766326904297
    },
    {
      "epoch": 2.4921417236328126e-05,
      "model_forward_time": 3.1440742015838623,
      "step": 6533
    },
    {
      "epoch": 2.4921417236328126e-05,
      "step": 6533,
      "training_step_time": 9.924670696258545
    },
    {
      "epoch": 2.4921417236328126e-05,
      "model_forward_time": 3.1491236686706543,
      "step": 6533
    },
    {
      "epoch": 2.4921417236328126e-05,
      "step": 6533,
      "training_step_time": 9.924788475036621
    },
    {
      "epoch": 2.4921417236328126e-05,
      "model_forward_time": 3.4421300888061523,
      "step": 6533
    },
    {
      "epoch": 2.4921417236328126e-05,
      "step": 6533,
      "training_step_time": 10.315034866333008
    },
    {
      "epoch": 2.492523193359375e-05,
      "model_forward_time": 3.1793549060821533,
      "step": 6534
    },
    {
      "epoch": 2.492523193359375e-05,
      "step": 6534,
      "training_step_time": 9.961038589477539
    },
    {
      "epoch": 2.492523193359375e-05,
      "model_forward_time": 3.144045829772949,
      "step": 6534
    },
    {
      "epoch": 2.492523193359375e-05,
      "step": 6534,
      "training_step_time": 9.923986673355103
    },
    {
      "epoch": 2.492523193359375e-05,
      "model_forward_time": 3.1502127647399902,
      "step": 6534
    },
    {
      "epoch": 2.492523193359375e-05,
      "step": 6534,
      "training_step_time": 9.932857275009155
    },
    {
      "epoch": 2.492523193359375e-05,
      "model_forward_time": 3.149419069290161,
      "step": 6534
    },
    {
      "epoch": 2.492523193359375e-05,
      "step": 6534,
      "training_step_time": 10.290540218353271
    },
    {
      "epoch": 2.4929046630859376e-05,
      "model_forward_time": 3.4885051250457764,
      "step": 6535
    },
    {
      "dynamics_loss_avg": 0.07356781959533691,
      "epoch": 2.4929046630859376e-05,
      "step": 6535
    },
    {
      "action_loss_avg": 0.03532655239105224,
      "epoch": 2.4929046630859376e-05,
      "step": 6535
    },
    {
      "epoch": 2.4929046630859376e-05,
      "step": 6535,
      "training_step_time": 10.268954992294312
    },
    {
      "epoch": 2.4929046630859376e-05,
      "model_forward_time": 3.1452131271362305,
      "step": 6535
    },
    {
      "epoch": 2.4929046630859376e-05,
      "step": 6535,
      "training_step_time": 9.919113636016846
    },
    {
      "epoch": 2.4929046630859376e-05,
      "model_forward_time": 3.1494061946868896,
      "step": 6535
    },
    {
      "epoch": 2.4929046630859376e-05,
      "step": 6535,
      "training_step_time": 9.926958322525024
    },
    {
      "epoch": 2.4929046630859376e-05,
      "model_forward_time": 3.143321990966797,
      "step": 6535
    },
    {
      "epoch": 2.4929046630859376e-05,
      "step": 6535,
      "training_step_time": 10.315436840057373
    },
    {
      "epoch": 2.4932861328125e-05,
      "model_forward_time": 3.1796679496765137,
      "step": 6536
    },
    {
      "epoch": 2.4932861328125e-05,
      "step": 6536,
      "training_step_time": 9.958808183670044
    },
    {
      "epoch": 2.4932861328125e-05,
      "model_forward_time": 3.4098756313323975,
      "step": 6536
    },
    {
      "epoch": 2.4932861328125e-05,
      "step": 6536,
      "training_step_time": 10.185978651046753
    },
    {
      "epoch": 2.4932861328125e-05,
      "model_forward_time": 3.142482280731201,
      "step": 6536
    },
    {
      "epoch": 2.4932861328125e-05,
      "step": 6536,
      "training_step_time": 9.92356538772583
    },
    {
      "epoch": 2.4932861328125e-05,
      "model_forward_time": 3.142923355102539,
      "step": 6536
    },
    {
      "epoch": 2.4932861328125e-05,
      "step": 6536,
      "training_step_time": 10.255420446395874
    },
    {
      "epoch": 2.4936676025390625e-05,
      "model_forward_time": 3.175112009048462,
      "step": 6537
    },
    {
      "epoch": 2.4936676025390625e-05,
      "step": 6537,
      "training_step_time": 9.95465874671936
    },
    {
      "epoch": 2.4936676025390625e-05,
      "model_forward_time": 3.135895013809204,
      "step": 6537
    },
    {
      "epoch": 2.4936676025390625e-05,
      "step": 6537,
      "training_step_time": 9.912176132202148
    },
    {
      "epoch": 2.4936676025390625e-05,
      "model_forward_time": 3.410477638244629,
      "step": 6537
    },
    {
      "dynamics_loss_avg": 0.0658358346670866,
      "epoch": 2.4936676025390625e-05,
      "step": 6537
    },
    {
      "action_loss_avg": 0.022808332834392785,
      "epoch": 2.4936676025390625e-05,
      "step": 6537
    },
    {
      "epoch": 2.4936676025390625e-05,
      "step": 6537,
      "training_step_time": 10.187244176864624
    },
    {
      "epoch": 2.4936676025390625e-05,
      "model_forward_time": 3.134382486343384,
      "step": 6537
    },
    {
      "epoch": 2.4936676025390625e-05,
      "step": 6537,
      "training_step_time": 10.309434652328491
    },
    {
      "epoch": 2.494049072265625e-05,
      "model_forward_time": 3.165445327758789,
      "step": 6538
    },
    {
      "epoch": 2.494049072265625e-05,
      "step": 6538,
      "training_step_time": 9.943034648895264
    },
    {
      "epoch": 2.494049072265625e-05,
      "model_forward_time": 3.1398186683654785,
      "step": 6538
    },
    {
      "epoch": 2.494049072265625e-05,
      "step": 6538,
      "training_step_time": 9.920082092285156
    },
    {
      "epoch": 2.494049072265625e-05,
      "model_forward_time": 3.1354429721832275,
      "step": 6538
    },
    {
      "epoch": 2.494049072265625e-05,
      "step": 6538,
      "training_step_time": 9.916420936584473
    },
    {
      "epoch": 2.494049072265625e-05,
      "model_forward_time": 3.3928709030151367,
      "step": 6538
    },
    {
      "epoch": 2.494049072265625e-05,
      "step": 6538,
      "training_step_time": 10.295090913772583
    },
    {
      "epoch": 2.4944305419921874e-05,
      "model_forward_time": 3.172718048095703,
      "step": 6539
    },
    {
      "epoch": 2.4944305419921874e-05,
      "step": 6539,
      "training_step_time": 9.952842235565186
    },
    {
      "epoch": 2.4944305419921874e-05,
      "model_forward_time": 3.137653350830078,
      "step": 6539
    },
    {
      "epoch": 2.4944305419921874e-05,
      "step": 6539,
      "training_step_time": 9.917970657348633
    },
    {
      "epoch": 2.4944305419921874e-05,
      "model_forward_time": 3.1370394229888916,
      "step": 6539
    },
    {
      "epoch": 2.4944305419921874e-05,
      "step": 6539,
      "training_step_time": 9.912670373916626
    },
    {
      "epoch": 2.4944305419921874e-05,
      "model_forward_time": 3.140230417251587,
      "step": 6539
    },
    {
      "epoch": 2.4944305419921874e-05,
      "step": 6539,
      "training_step_time": 10.304035663604736
    },
    {
      "epoch": 2.49481201171875e-05,
      "grad_norm": 0.43517574667930603,
      "learning_rate": 1e-05,
      "loss": 0.0836,
      "step": 6540
    },
    {
      "epoch": 2.49481201171875e-05,
      "model_forward_time": 3.575791835784912,
      "step": 6540
    },
    {
      "dynamics_loss_avg": 0.059126023389399054,
      "epoch": 2.49481201171875e-05,
      "step": 6540
    },
    {
      "action_loss_avg": 0.01923772522713989,
      "epoch": 2.49481201171875e-05,
      "step": 6540
    },
    {
      "epoch": 2.49481201171875e-05,
      "step": 6540,
      "training_step_time": 10.357254266738892
    },
    {
      "epoch": 2.49481201171875e-05,
      "model_forward_time": 3.1614646911621094,
      "step": 6540
    },
    {
      "epoch": 2.49481201171875e-05,
      "step": 6540,
      "training_step_time": 9.948039054870605
    },
    {
      "epoch": 2.49481201171875e-05,
      "model_forward_time": 3.1441633701324463,
      "step": 6540
    },
    {
      "epoch": 2.49481201171875e-05,
      "step": 6540,
      "training_step_time": 9.927819967269897
    },
    {
      "epoch": 2.49481201171875e-05,
      "model_forward_time": 3.1519460678100586,
      "step": 6540
    },
    {
      "epoch": 2.49481201171875e-05,
      "step": 6540,
      "training_step_time": 10.318757057189941
    },
    {
      "epoch": 2.4951934814453126e-05,
      "model_forward_time": 3.190216302871704,
      "step": 6541
    },
    {
      "epoch": 2.4951934814453126e-05,
      "step": 6541,
      "training_step_time": 9.968603134155273
    },
    {
      "epoch": 2.4951934814453126e-05,
      "model_forward_time": 3.47819447517395,
      "step": 6541
    },
    {
      "epoch": 2.4951934814453126e-05,
      "step": 6541,
      "training_step_time": 10.256235837936401
    },
    {
      "epoch": 2.4951934814453126e-05,
      "model_forward_time": 3.142055034637451,
      "step": 6541
    },
    {
      "epoch": 2.4951934814453126e-05,
      "step": 6541,
      "training_step_time": 9.91185736656189
    },
    {
      "epoch": 2.4951934814453126e-05,
      "model_forward_time": 3.141678810119629,
      "step": 6541
    },
    {
      "epoch": 2.4951934814453126e-05,
      "step": 6541,
      "training_step_time": 10.253060579299927
    },
    {
      "epoch": 2.495574951171875e-05,
      "model_forward_time": 3.212120771408081,
      "step": 6542
    },
    {
      "epoch": 2.495574951171875e-05,
      "step": 6542,
      "training_step_time": 9.991855144500732
    },
    {
      "epoch": 2.495574951171875e-05,
      "model_forward_time": 3.1342835426330566,
      "step": 6542
    },
    {
      "epoch": 2.495574951171875e-05,
      "step": 6542,
      "training_step_time": 9.910980463027954
    },
    {
      "epoch": 2.495574951171875e-05,
      "model_forward_time": 3.4261772632598877,
      "step": 6542
    },
    {
      "dynamics_loss_avg": 0.05315079838037491,
      "epoch": 2.495574951171875e-05,
      "step": 6542
    },
    {
      "action_loss_avg": 0.02064777258783579,
      "epoch": 2.495574951171875e-05,
      "step": 6542
    },
    {
      "epoch": 2.495574951171875e-05,
      "step": 6542,
      "training_step_time": 10.197498559951782
    },
    {
      "epoch": 2.495574951171875e-05,
      "model_forward_time": 3.1451056003570557,
      "step": 6542
    },
    {
      "epoch": 2.495574951171875e-05,
      "step": 6542,
      "training_step_time": 10.014897584915161
    },
    {
      "epoch": 2.4959564208984376e-05,
      "model_forward_time": 3.1649460792541504,
      "step": 6543
    },
    {
      "epoch": 2.4959564208984376e-05,
      "step": 6543,
      "training_step_time": 9.944503784179688
    },
    {
      "epoch": 2.4959564208984376e-05,
      "model_forward_time": 3.1350271701812744,
      "step": 6543
    },
    {
      "epoch": 2.4959564208984376e-05,
      "step": 6543,
      "training_step_time": 9.909149408340454
    },
    {
      "epoch": 2.4959564208984376e-05,
      "model_forward_time": 3.138084888458252,
      "step": 6543
    },
    {
      "epoch": 2.4959564208984376e-05,
      "step": 6543,
      "training_step_time": 9.919057369232178
    },
    {
      "epoch": 2.4959564208984376e-05,
      "model_forward_time": 3.420480251312256,
      "step": 6543
    },
    {
      "epoch": 2.4959564208984376e-05,
      "step": 6543,
      "training_step_time": 10.289750576019287
    },
    {
      "epoch": 2.496337890625e-05,
      "model_forward_time": 3.1738498210906982,
      "step": 6544
    },
    {
      "epoch": 2.496337890625e-05,
      "step": 6544,
      "training_step_time": 9.955872058868408
    },
    {
      "epoch": 2.496337890625e-05,
      "model_forward_time": 3.142611265182495,
      "step": 6544
    },
    {
      "epoch": 2.496337890625e-05,
      "step": 6544,
      "training_step_time": 9.915484189987183
    },
    {
      "epoch": 2.496337890625e-05,
      "model_forward_time": 3.136929750442505,
      "step": 6544
    },
    {
      "epoch": 2.496337890625e-05,
      "step": 6544,
      "training_step_time": 9.909630060195923
    },
    {
      "epoch": 2.496337890625e-05,
      "model_forward_time": 3.1416144371032715,
      "step": 6544
    },
    {
      "epoch": 2.496337890625e-05,
      "step": 6544,
      "training_step_time": 10.28129529953003
    },
    {
      "epoch": 2.4967193603515625e-05,
      "model_forward_time": 3.4871559143066406,
      "step": 6545
    },
    {
      "dynamics_loss_avg": 0.0536420188844204,
      "epoch": 2.4967193603515625e-05,
      "step": 6545
    },
    {
      "action_loss_avg": 0.015029707178473472,
      "epoch": 2.4967193603515625e-05,
      "step": 6545
    },
    {
      "epoch": 2.4967193603515625e-05,
      "step": 6545,
      "training_step_time": 10.265844106674194
    },
    {
      "epoch": 2.4967193603515625e-05,
      "model_forward_time": 3.1394665241241455,
      "step": 6545
    },
    {
      "epoch": 2.4967193603515625e-05,
      "step": 6545,
      "training_step_time": 9.91641640663147
    },
    {
      "epoch": 2.4967193603515625e-05,
      "model_forward_time": 3.1414291858673096,
      "step": 6545
    },
    {
      "epoch": 2.4967193603515625e-05,
      "step": 6545,
      "training_step_time": 9.915048122406006
    },
    {
      "epoch": 2.4967193603515625e-05,
      "model_forward_time": 3.143972158432007,
      "step": 6545
    },
    {
      "epoch": 2.4967193603515625e-05,
      "step": 6545,
      "training_step_time": 10.232073068618774
    },
    {
      "epoch": 2.497100830078125e-05,
      "model_forward_time": 3.179349660873413,
      "step": 6546
    },
    {
      "epoch": 2.497100830078125e-05,
      "step": 6546,
      "training_step_time": 9.95869779586792
    },
    {
      "epoch": 2.497100830078125e-05,
      "model_forward_time": 3.429367780685425,
      "step": 6546
    },
    {
      "epoch": 2.497100830078125e-05,
      "step": 6546,
      "training_step_time": 10.21654224395752
    },
    {
      "epoch": 2.497100830078125e-05,
      "model_forward_time": 3.148892879486084,
      "step": 6546
    },
    {
      "epoch": 2.497100830078125e-05,
      "step": 6546,
      "training_step_time": 9.926809549331665
    },
    {
      "epoch": 2.497100830078125e-05,
      "model_forward_time": 3.145477294921875,
      "step": 6546
    },
    {
      "epoch": 2.497100830078125e-05,
      "step": 6546,
      "training_step_time": 10.324609994888306
    },
    {
      "epoch": 2.4974822998046874e-05,
      "model_forward_time": 3.1759538650512695,
      "step": 6547
    },
    {
      "epoch": 2.4974822998046874e-05,
      "step": 6547,
      "training_step_time": 9.951809406280518
    },
    {
      "epoch": 2.4974822998046874e-05,
      "model_forward_time": 3.145817756652832,
      "step": 6547
    },
    {
      "epoch": 2.4974822998046874e-05,
      "step": 6547,
      "training_step_time": 9.92020034790039
    },
    {
      "epoch": 2.4974822998046874e-05,
      "model_forward_time": 3.422497510910034,
      "step": 6547
    },
    {
      "dynamics_loss_avg": 0.07293946482241154,
      "epoch": 2.4974822998046874e-05,
      "step": 6547
    },
    {
      "action_loss_avg": 0.02824619384482503,
      "epoch": 2.4974822998046874e-05,
      "step": 6547
    },
    {
      "epoch": 2.4974822998046874e-05,
      "step": 6547,
      "training_step_time": 10.199448823928833
    },
    {
      "epoch": 2.4974822998046874e-05,
      "model_forward_time": 3.1478025913238525,
      "step": 6547
    },
    {
      "epoch": 2.4974822998046874e-05,
      "step": 6547,
      "training_step_time": 10.024170637130737
    },
    {
      "epoch": 2.49786376953125e-05,
      "model_forward_time": 3.1961429119110107,
      "step": 6548
    },
    {
      "epoch": 2.49786376953125e-05,
      "step": 6548,
      "training_step_time": 9.974365711212158
    },
    {
      "epoch": 2.49786376953125e-05,
      "model_forward_time": 3.1396379470825195,
      "step": 6548
    },
    {
      "epoch": 2.49786376953125e-05,
      "step": 6548,
      "training_step_time": 9.91422414779663
    },
    {
      "epoch": 2.49786376953125e-05,
      "model_forward_time": 3.1406660079956055,
      "step": 6548
    },
    {
      "epoch": 2.49786376953125e-05,
      "step": 6548,
      "training_step_time": 10.213125705718994
    },
    {
      "epoch": 2.49786376953125e-05,
      "model_forward_time": 3.1417458057403564,
      "step": 6548
    },
    {
      "epoch": 2.49786376953125e-05,
      "step": 6548,
      "training_step_time": 10.02569055557251
    },
    {
      "epoch": 2.4982452392578126e-05,
      "model_forward_time": 3.182131767272949,
      "step": 6549
    },
    {
      "epoch": 2.4982452392578126e-05,
      "step": 6549,
      "training_step_time": 9.963648319244385
    },
    {
      "epoch": 2.4982452392578126e-05,
      "model_forward_time": 3.1494157314300537,
      "step": 6549
    },
    {
      "epoch": 2.4982452392578126e-05,
      "step": 6549,
      "training_step_time": 9.925436735153198
    },
    {
      "epoch": 2.4982452392578126e-05,
      "model_forward_time": 3.1380438804626465,
      "step": 6549
    },
    {
      "epoch": 2.4982452392578126e-05,
      "step": 6549,
      "training_step_time": 9.91213870048523
    },
    {
      "epoch": 2.4982452392578126e-05,
      "model_forward_time": 3.1461539268493652,
      "step": 6549
    },
    {
      "epoch": 2.4982452392578126e-05,
      "step": 6549,
      "training_step_time": 10.326817989349365
    },
    {
      "epoch": 2.498626708984375e-05,
      "grad_norm": 0.596373438835144,
      "learning_rate": 1e-05,
      "loss": 0.0854,
      "step": 6550
    },
    {
      "epoch": 2.498626708984375e-05,
      "model_forward_time": 3.157226085662842,
      "step": 6550
    },
    {
      "dynamics_loss_avg": 0.06461217775940895,
      "epoch": 2.498626708984375e-05,
      "step": 6550
    },
    {
      "action_loss_avg": 0.026494104834273457,
      "epoch": 2.498626708984375e-05,
      "step": 6550
    },
    {
      "epoch": 2.498626708984375e-05,
      "step": 6550,
      "training_step_time": 9.934176206588745
    },
    {
      "epoch": 2.498626708984375e-05,
      "model_forward_time": 3.143761396408081,
      "step": 6550
    },
    {
      "epoch": 2.498626708984375e-05,
      "step": 6550,
      "training_step_time": 9.918673515319824
    },
    {
      "epoch": 2.498626708984375e-05,
      "model_forward_time": 2.4362831115722656,
      "step": 6550
    },
    {
      "epoch": 2.498626708984375e-05,
      "step": 6550,
      "training_step_time": 7.466436862945557
    },
    {
      "epoch": 2.498626708984375e-05,
      "model_forward_time": 3.1367743015289307,
      "step": 6550
    },
    {
      "epoch": 2.498626708984375e-05,
      "step": 6550,
      "training_step_time": 10.304057598114014
    },
    {
      "epoch": 2.4990081787109376e-05,
      "model_forward_time": 3.1587226390838623,
      "step": 6551
    },
    {
      "epoch": 2.4990081787109376e-05,
      "step": 6551,
      "training_step_time": 10.203255414962769
    },
    {
      "epoch": 2.4990081787109376e-05,
      "model_forward_time": 3.1332287788391113,
      "step": 6551
    },
    {
      "epoch": 2.4990081787109376e-05,
      "step": 6551,
      "training_step_time": 9.902898073196411
    },
    {
      "epoch": 2.4990081787109376e-05,
      "model_forward_time": 3.1316776275634766,
      "step": 6551
    },
    {
      "epoch": 2.4990081787109376e-05,
      "step": 6551,
      "training_step_time": 9.899736642837524
    },
    {
      "epoch": 2.4990081787109376e-05,
      "model_forward_time": 3.1335902214050293,
      "step": 6551
    },
    {
      "epoch": 2.4990081787109376e-05,
      "step": 6551,
      "training_step_time": 10.001531839370728
    },
    {
      "epoch": 2.4993896484375e-05,
      "model_forward_time": 3.1828668117523193,
      "step": 6552
    },
    {
      "epoch": 2.4993896484375e-05,
      "step": 6552,
      "training_step_time": 9.954426527023315
    },
    {
      "epoch": 2.4993896484375e-05,
      "model_forward_time": 3.423450469970703,
      "step": 6552
    },
    {
      "epoch": 2.4993896484375e-05,
      "step": 6552,
      "training_step_time": 10.191413402557373
    },
    {
      "epoch": 2.4993896484375e-05,
      "model_forward_time": 3.1354024410247803,
      "step": 6552
    },
    {
      "dynamics_loss_avg": 0.06610692348331212,
      "epoch": 2.4993896484375e-05,
      "step": 6552
    },
    {
      "action_loss_avg": 0.026560769649222493,
      "epoch": 2.4993896484375e-05,
      "step": 6552
    },
    {
      "epoch": 2.4993896484375e-05,
      "step": 6552,
      "training_step_time": 9.909100532531738
    },
    {
      "epoch": 2.4993896484375e-05,
      "model_forward_time": 3.139864206314087,
      "step": 6552
    },
    {
      "epoch": 2.4993896484375e-05,
      "step": 6552,
      "training_step_time": 10.306326150894165
    },
    {
      "epoch": 2.4997711181640625e-05,
      "model_forward_time": 3.1739840507507324,
      "step": 6553
    },
    {
      "epoch": 2.4997711181640625e-05,
      "step": 6553,
      "training_step_time": 9.94673228263855
    },
    {
      "epoch": 2.4997711181640625e-05,
      "model_forward_time": 3.139136791229248,
      "step": 6553
    },
    {
      "epoch": 2.4997711181640625e-05,
      "step": 6553,
      "training_step_time": 9.916111707687378
    },
    {
      "epoch": 2.4997711181640625e-05,
      "model_forward_time": 3.4087650775909424,
      "step": 6553
    },
    {
      "epoch": 2.4997711181640625e-05,
      "step": 6553,
      "training_step_time": 10.185033798217773
    },
    {
      "epoch": 2.4997711181640625e-05,
      "model_forward_time": 3.144124746322632,
      "step": 6553
    },
    {
      "epoch": 2.4997711181640625e-05,
      "step": 6553,
      "training_step_time": 10.017498254776001
    },
    {
      "epoch": 2.500152587890625e-05,
      "model_forward_time": 3.182417631149292,
      "step": 6554
    },
    {
      "epoch": 2.500152587890625e-05,
      "step": 6554,
      "training_step_time": 9.956587791442871
    },
    {
      "epoch": 2.500152587890625e-05,
      "model_forward_time": 3.1426141262054443,
      "step": 6554
    },
    {
      "epoch": 2.500152587890625e-05,
      "step": 6554,
      "training_step_time": 9.914332389831543
    },
    {
      "epoch": 2.500152587890625e-05,
      "model_forward_time": 3.1412956714630127,
      "step": 6554
    },
    {
      "epoch": 2.500152587890625e-05,
      "step": 6554,
      "training_step_time": 9.918104887008667
    },
    {
      "epoch": 2.500152587890625e-05,
      "model_forward_time": 3.463704824447632,
      "step": 6554
    },
    {
      "epoch": 2.500152587890625e-05,
      "step": 6554,
      "training_step_time": 10.34062385559082
    },
    {
      "epoch": 2.5005340576171874e-05,
      "model_forward_time": 3.1939005851745605,
      "step": 6555
    },
    {
      "dynamics_loss_avg": 0.07046398427337408,
      "epoch": 2.5005340576171874e-05,
      "step": 6555
    },
    {
      "action_loss_avg": 0.025036588334478437,
      "epoch": 2.5005340576171874e-05,
      "step": 6555
    },
    {
      "epoch": 2.5005340576171874e-05,
      "step": 6555,
      "training_step_time": 9.971071720123291
    },
    {
      "epoch": 2.5005340576171874e-05,
      "model_forward_time": 3.1429481506347656,
      "step": 6555
    },
    {
      "epoch": 2.5005340576171874e-05,
      "step": 6555,
      "training_step_time": 9.919526815414429
    },
    {
      "epoch": 2.5005340576171874e-05,
      "model_forward_time": 3.1465396881103516,
      "step": 6555
    },
    {
      "epoch": 2.5005340576171874e-05,
      "step": 6555,
      "training_step_time": 9.925466537475586
    },
    {
      "epoch": 2.5005340576171874e-05,
      "model_forward_time": 3.1441290378570557,
      "step": 6555
    },
    {
      "epoch": 2.5005340576171874e-05,
      "step": 6555,
      "training_step_time": 10.036571025848389
    },
    {
      "epoch": 2.50091552734375e-05,
      "model_forward_time": 3.4595251083374023,
      "step": 6556
    },
    {
      "epoch": 2.50091552734375e-05,
      "step": 6556,
      "training_step_time": 10.24414610862732
    },
    {
      "epoch": 2.50091552734375e-05,
      "model_forward_time": 3.1417489051818848,
      "step": 6556
    },
    {
      "epoch": 2.50091552734375e-05,
      "step": 6556,
      "training_step_time": 9.9166259765625
    },
    {
      "epoch": 2.50091552734375e-05,
      "model_forward_time": 3.1426494121551514,
      "step": 6556
    },
    {
      "epoch": 2.50091552734375e-05,
      "step": 6556,
      "training_step_time": 9.916819095611572
    },
    {
      "epoch": 2.50091552734375e-05,
      "model_forward_time": 3.1359000205993652,
      "step": 6556
    },
    {
      "epoch": 2.50091552734375e-05,
      "step": 6556,
      "training_step_time": 10.32460331916809
    },
    {
      "epoch": 2.5012969970703126e-05,
      "model_forward_time": 3.2116827964782715,
      "step": 6557
    },
    {
      "epoch": 2.5012969970703126e-05,
      "step": 6557,
      "training_step_time": 10.001792192459106
    },
    {
      "epoch": 2.5012969970703126e-05,
      "model_forward_time": 3.419066905975342,
      "step": 6557
    },
    {
      "epoch": 2.5012969970703126e-05,
      "step": 6557,
      "training_step_time": 10.197636604309082
    },
    {
      "epoch": 2.5012969970703126e-05,
      "model_forward_time": 3.1516306400299072,
      "step": 6557
    },
    {
      "dynamics_loss_avg": 0.05394646730273962,
      "epoch": 2.5012969970703126e-05,
      "step": 6557
    },
    {
      "action_loss_avg": 0.02241978603415191,
      "epoch": 2.5012969970703126e-05,
      "step": 6557
    },
    {
      "epoch": 2.5012969970703126e-05,
      "step": 6557,
      "training_step_time": 9.929479122161865
    },
    {
      "epoch": 2.5012969970703126e-05,
      "model_forward_time": 3.14281964302063,
      "step": 6557
    },
    {
      "epoch": 2.5012969970703126e-05,
      "step": 6557,
      "training_step_time": 10.263956785202026
    },
    {
      "epoch": 2.501678466796875e-05,
      "model_forward_time": 3.172372817993164,
      "step": 6558
    },
    {
      "epoch": 2.501678466796875e-05,
      "step": 6558,
      "training_step_time": 9.962706327438354
    },
    {
      "epoch": 2.501678466796875e-05,
      "model_forward_time": 3.1362874507904053,
      "step": 6558
    },
    {
      "epoch": 2.501678466796875e-05,
      "step": 6558,
      "training_step_time": 9.925126075744629
    },
    {
      "epoch": 2.501678466796875e-05,
      "model_forward_time": 3.460763454437256,
      "step": 6558
    },
    {
      "epoch": 2.501678466796875e-05,
      "step": 6558,
      "training_step_time": 10.230652332305908
    },
    {
      "epoch": 2.501678466796875e-05,
      "model_forward_time": 3.1386711597442627,
      "step": 6558
    },
    {
      "epoch": 2.501678466796875e-05,
      "step": 6558,
      "training_step_time": 10.023410320281982
    },
    {
      "epoch": 2.5020599365234376e-05,
      "model_forward_time": 3.163804054260254,
      "step": 6559
    },
    {
      "epoch": 2.5020599365234376e-05,
      "step": 6559,
      "training_step_time": 9.94087529182434
    },
    {
      "epoch": 2.5020599365234376e-05,
      "model_forward_time": 3.1426842212677,
      "step": 6559
    },
    {
      "epoch": 2.5020599365234376e-05,
      "step": 6559,
      "training_step_time": 9.924040079116821
    },
    {
      "epoch": 2.5020599365234376e-05,
      "model_forward_time": 3.141172409057617,
      "step": 6559
    },
    {
      "epoch": 2.5020599365234376e-05,
      "step": 6559,
      "training_step_time": 9.919816255569458
    },
    {
      "epoch": 2.5020599365234376e-05,
      "model_forward_time": 3.1375534534454346,
      "step": 6559
    },
    {
      "epoch": 2.5020599365234376e-05,
      "step": 6559,
      "training_step_time": 10.336947202682495
    },
    {
      "epoch": 2.50244140625e-05,
      "grad_norm": 0.28087279200553894,
      "learning_rate": 1e-05,
      "loss": 0.0835,
      "step": 6560
    },
    {
      "epoch": 2.50244140625e-05,
      "model_forward_time": 3.1774306297302246,
      "step": 6560
    },
    {
      "dynamics_loss_avg": 0.052520958613604306,
      "epoch": 2.50244140625e-05,
      "step": 6560
    },
    {
      "action_loss_avg": 0.01749051036313176,
      "epoch": 2.50244140625e-05,
      "step": 6560
    },
    {
      "epoch": 2.50244140625e-05,
      "step": 6560,
      "training_step_time": 9.958703994750977
    },
    {
      "epoch": 2.50244140625e-05,
      "model_forward_time": 3.136532783508301,
      "step": 6560
    },
    {
      "epoch": 2.50244140625e-05,
      "step": 6560,
      "training_step_time": 9.91360092163086
    },
    {
      "epoch": 2.50244140625e-05,
      "model_forward_time": 3.139883041381836,
      "step": 6560
    },
    {
      "epoch": 2.50244140625e-05,
      "step": 6560,
      "training_step_time": 9.916705846786499
    },
    {
      "epoch": 2.50244140625e-05,
      "model_forward_time": 3.143822193145752,
      "step": 6560
    },
    {
      "epoch": 2.50244140625e-05,
      "step": 6560,
      "training_step_time": 10.305345058441162
    },
    {
      "epoch": 2.5028228759765625e-05,
      "model_forward_time": 3.444737434387207,
      "step": 6561
    },
    {
      "epoch": 2.5028228759765625e-05,
      "step": 6561,
      "training_step_time": 10.220538854598999
    },
    {
      "epoch": 2.5028228759765625e-05,
      "model_forward_time": 3.136622905731201,
      "step": 6561
    },
    {
      "epoch": 2.5028228759765625e-05,
      "step": 6561,
      "training_step_time": 9.910773038864136
    },
    {
      "epoch": 2.5028228759765625e-05,
      "model_forward_time": 3.1354563236236572,
      "step": 6561
    },
    {
      "epoch": 2.5028228759765625e-05,
      "step": 6561,
      "training_step_time": 9.910407781600952
    },
    {
      "epoch": 2.5028228759765625e-05,
      "model_forward_time": 3.1398963928222656,
      "step": 6561
    },
    {
      "epoch": 2.5028228759765625e-05,
      "step": 6561,
      "training_step_time": 10.277980089187622
    },
    {
      "epoch": 2.503204345703125e-05,
      "model_forward_time": 3.1706535816192627,
      "step": 6562
    },
    {
      "epoch": 2.503204345703125e-05,
      "step": 6562,
      "training_step_time": 9.950888395309448
    },
    {
      "epoch": 2.503204345703125e-05,
      "model_forward_time": 3.1327531337738037,
      "step": 6562
    },
    {
      "epoch": 2.503204345703125e-05,
      "step": 6562,
      "training_step_time": 10.184887886047363
    },
    {
      "epoch": 2.503204345703125e-05,
      "model_forward_time": 2.4292173385620117,
      "step": 6562
    },
    {
      "dynamics_loss_avg": 0.07421106565743685,
      "epoch": 2.503204345703125e-05,
      "step": 6562
    },
    {
      "action_loss_avg": 0.027296986617147922,
      "epoch": 2.503204345703125e-05,
      "step": 6562
    },
    {
      "epoch": 2.503204345703125e-05,
      "step": 6562,
      "training_step_time": 7.458416700363159
    },
    {
      "epoch": 2.503204345703125e-05,
      "model_forward_time": 3.1329691410064697,
      "step": 6562
    },
    {
      "epoch": 2.503204345703125e-05,
      "step": 6562,
      "training_step_time": 10.010565519332886
    },
    {
      "epoch": 2.5035858154296874e-05,
      "model_forward_time": 2.4571359157562256,
      "step": 6563
    },
    {
      "epoch": 2.5035858154296874e-05,
      "step": 6563,
      "training_step_time": 7.483911752700806
    },
    {
      "epoch": 2.5035858154296874e-05,
      "model_forward_time": 3.1262753009796143,
      "step": 6563
    },
    {
      "epoch": 2.5035858154296874e-05,
      "step": 6563,
      "training_step_time": 9.89924168586731
    },
    {
      "epoch": 2.5035858154296874e-05,
      "model_forward_time": 3.136573076248169,
      "step": 6563
    },
    {
      "epoch": 2.5035858154296874e-05,
      "step": 6563,
      "training_step_time": 9.912341117858887
    },
    {
      "epoch": 2.5035858154296874e-05,
      "model_forward_time": 3.4558088779449463,
      "step": 6563
    },
    {
      "epoch": 2.5035858154296874e-05,
      "step": 6563,
      "training_step_time": 10.330612182617188
    },
    {
      "epoch": 2.50396728515625e-05,
      "model_forward_time": 3.1643404960632324,
      "step": 6564
    },
    {
      "epoch": 2.50396728515625e-05,
      "step": 6564,
      "training_step_time": 9.95087194442749
    },
    {
      "epoch": 2.50396728515625e-05,
      "model_forward_time": 3.151700019836426,
      "step": 6564
    },
    {
      "epoch": 2.50396728515625e-05,
      "step": 6564,
      "training_step_time": 9.935252666473389
    },
    {
      "epoch": 2.50396728515625e-05,
      "model_forward_time": 2.44325590133667,
      "step": 6564
    },
    {
      "epoch": 2.50396728515625e-05,
      "step": 6564,
      "training_step_time": 7.482325077056885
    },
    {
      "epoch": 2.50396728515625e-05,
      "model_forward_time": 3.1436033248901367,
      "step": 6564
    },
    {
      "epoch": 2.50396728515625e-05,
      "step": 6564,
      "training_step_time": 10.295790433883667
    },
    {
      "epoch": 2.5043487548828126e-05,
      "model_forward_time": 3.494279623031616,
      "step": 6565
    },
    {
      "dynamics_loss_avg": 0.05395951867103577,
      "epoch": 2.5043487548828126e-05,
      "step": 6565
    },
    {
      "action_loss_avg": 0.02405702348332852,
      "epoch": 2.5043487548828126e-05,
      "step": 6565
    },
    {
      "epoch": 2.5043487548828126e-05,
      "step": 6565,
      "training_step_time": 10.275342226028442
    },
    {
      "epoch": 2.5043487548828126e-05,
      "model_forward_time": 3.1424856185913086,
      "step": 6565
    },
    {
      "epoch": 2.5043487548828126e-05,
      "step": 6565,
      "training_step_time": 9.920511245727539
    },
    {
      "epoch": 2.5043487548828126e-05,
      "model_forward_time": 3.1462676525115967,
      "step": 6565
    },
    {
      "epoch": 2.5043487548828126e-05,
      "step": 6565,
      "training_step_time": 9.928874015808105
    },
    {
      "epoch": 2.5043487548828126e-05,
      "model_forward_time": 3.1422791481018066,
      "step": 6565
    },
    {
      "epoch": 2.5043487548828126e-05,
      "step": 6565,
      "training_step_time": 10.276691436767578
    },
    {
      "epoch": 2.504730224609375e-05,
      "model_forward_time": 3.456592321395874,
      "step": 6566
    },
    {
      "epoch": 2.504730224609375e-05,
      "step": 6566,
      "training_step_time": 10.236963510513306
    },
    {
      "epoch": 2.504730224609375e-05,
      "model_forward_time": 3.140425205230713,
      "step": 6566
    },
    {
      "epoch": 2.504730224609375e-05,
      "step": 6566,
      "training_step_time": 9.916683435440063
    },
    {
      "epoch": 2.504730224609375e-05,
      "model_forward_time": 3.144918918609619,
      "step": 6566
    },
    {
      "epoch": 2.504730224609375e-05,
      "step": 6566,
      "training_step_time": 9.921430826187134
    },
    {
      "epoch": 2.504730224609375e-05,
      "model_forward_time": 3.139740228652954,
      "step": 6566
    },
    {
      "epoch": 2.504730224609375e-05,
      "step": 6566,
      "training_step_time": 10.267534494400024
    },
    {
      "epoch": 2.5051116943359376e-05,
      "model_forward_time": 3.1685588359832764,
      "step": 6567
    },
    {
      "epoch": 2.5051116943359376e-05,
      "step": 6567,
      "training_step_time": 9.943304777145386
    },
    {
      "epoch": 2.5051116943359376e-05,
      "model_forward_time": 3.4070825576782227,
      "step": 6567
    },
    {
      "epoch": 2.5051116943359376e-05,
      "step": 6567,
      "training_step_time": 10.181772470474243
    },
    {
      "epoch": 2.5051116943359376e-05,
      "model_forward_time": 3.1409413814544678,
      "step": 6567
    },
    {
      "dynamics_loss_avg": 0.05603925138711929,
      "epoch": 2.5051116943359376e-05,
      "step": 6567
    },
    {
      "action_loss_avg": 0.022649484919384122,
      "epoch": 2.5051116943359376e-05,
      "step": 6567
    },
    {
      "epoch": 2.5051116943359376e-05,
      "step": 6567,
      "training_step_time": 9.912137746810913
    },
    {
      "epoch": 2.5051116943359376e-05,
      "model_forward_time": 3.137418270111084,
      "step": 6567
    },
    {
      "epoch": 2.5051116943359376e-05,
      "step": 6567,
      "training_step_time": 10.009121656417847
    },
    {
      "epoch": 2.5054931640625e-05,
      "model_forward_time": 3.1794676780700684,
      "step": 6568
    },
    {
      "epoch": 2.5054931640625e-05,
      "step": 6568,
      "training_step_time": 9.94261360168457
    },
    {
      "epoch": 2.5054931640625e-05,
      "model_forward_time": 3.1416101455688477,
      "step": 6568
    },
    {
      "epoch": 2.5054931640625e-05,
      "step": 6568,
      "training_step_time": 9.906357526779175
    },
    {
      "epoch": 2.5054931640625e-05,
      "model_forward_time": 3.138474941253662,
      "step": 6568
    },
    {
      "epoch": 2.5054931640625e-05,
      "step": 6568,
      "training_step_time": 10.1857590675354
    },
    {
      "epoch": 2.5054931640625e-05,
      "model_forward_time": 3.1390318870544434,
      "step": 6568
    },
    {
      "epoch": 2.5054931640625e-05,
      "step": 6568,
      "training_step_time": 10.304812669754028
    },
    {
      "epoch": 2.5058746337890625e-05,
      "model_forward_time": 3.1835572719573975,
      "step": 6569
    },
    {
      "epoch": 2.5058746337890625e-05,
      "step": 6569,
      "training_step_time": 9.954236030578613
    },
    {
      "epoch": 2.5058746337890625e-05,
      "model_forward_time": 3.1426758766174316,
      "step": 6569
    },
    {
      "epoch": 2.5058746337890625e-05,
      "step": 6569,
      "training_step_time": 9.91292691230774
    },
    {
      "epoch": 2.5058746337890625e-05,
      "model_forward_time": 3.1386020183563232,
      "step": 6569
    },
    {
      "epoch": 2.5058746337890625e-05,
      "step": 6569,
      "training_step_time": 9.909067153930664
    },
    {
      "epoch": 2.5058746337890625e-05,
      "model_forward_time": 3.143138885498047,
      "step": 6569
    },
    {
      "epoch": 2.5058746337890625e-05,
      "step": 6569,
      "training_step_time": 10.305389642715454
    },
    {
      "epoch": 2.506256103515625e-05,
      "grad_norm": 0.3940463066101074,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 6570
    },
    {
      "epoch": 2.506256103515625e-05,
      "model_forward_time": 3.2065303325653076,
      "step": 6570
    },
    {
      "dynamics_loss_avg": 0.05648173987865448,
      "epoch": 2.506256103515625e-05,
      "step": 6570
    },
    {
      "action_loss_avg": 0.01769352052360773,
      "epoch": 2.506256103515625e-05,
      "step": 6570
    },
    {
      "epoch": 2.506256103515625e-05,
      "step": 6570,
      "training_step_time": 9.983146667480469
    },
    {
      "epoch": 2.506256103515625e-05,
      "model_forward_time": 3.144390106201172,
      "step": 6570
    },
    {
      "epoch": 2.506256103515625e-05,
      "step": 6570,
      "training_step_time": 9.917738437652588
    },
    {
      "epoch": 2.506256103515625e-05,
      "model_forward_time": 3.139930486679077,
      "step": 6570
    },
    {
      "epoch": 2.506256103515625e-05,
      "step": 6570,
      "training_step_time": 9.913414001464844
    },
    {
      "epoch": 2.506256103515625e-05,
      "model_forward_time": 3.1466808319091797,
      "step": 6570
    },
    {
      "epoch": 2.506256103515625e-05,
      "step": 6570,
      "training_step_time": 10.289915561676025
    },
    {
      "epoch": 2.5066375732421874e-05,
      "model_forward_time": 3.1934778690338135,
      "step": 6571
    },
    {
      "epoch": 2.5066375732421874e-05,
      "step": 6571,
      "training_step_time": 10.285269260406494
    },
    {
      "epoch": 2.5066375732421874e-05,
      "model_forward_time": 3.1703691482543945,
      "step": 6571
    },
    {
      "epoch": 2.5066375732421874e-05,
      "step": 6571,
      "training_step_time": 9.95639419555664
    },
    {
      "epoch": 2.5066375732421874e-05,
      "model_forward_time": 3.152252674102783,
      "step": 6571
    },
    {
      "epoch": 2.5066375732421874e-05,
      "step": 6571,
      "training_step_time": 9.938754320144653
    },
    {
      "epoch": 2.5066375732421874e-05,
      "model_forward_time": 3.1520512104034424,
      "step": 6571
    },
    {
      "epoch": 2.5066375732421874e-05,
      "step": 6571,
      "training_step_time": 10.354975700378418
    },
    {
      "epoch": 2.50701904296875e-05,
      "model_forward_time": 3.1833629608154297,
      "step": 6572
    },
    {
      "epoch": 2.50701904296875e-05,
      "step": 6572,
      "training_step_time": 9.958638668060303
    },
    {
      "epoch": 2.50701904296875e-05,
      "model_forward_time": 3.428856611251831,
      "step": 6572
    },
    {
      "epoch": 2.50701904296875e-05,
      "step": 6572,
      "training_step_time": 10.206804037094116
    },
    {
      "epoch": 2.50701904296875e-05,
      "model_forward_time": 3.147186517715454,
      "step": 6572
    },
    {
      "dynamics_loss_avg": 0.06654832744970918,
      "epoch": 2.50701904296875e-05,
      "step": 6572
    },
    {
      "action_loss_avg": 0.02461371081881225,
      "epoch": 2.50701904296875e-05,
      "step": 6572
    },
    {
      "epoch": 2.50701904296875e-05,
      "step": 6572,
      "training_step_time": 9.928454875946045
    },
    {
      "epoch": 2.50701904296875e-05,
      "model_forward_time": 3.146515369415283,
      "step": 6572
    },
    {
      "epoch": 2.50701904296875e-05,
      "step": 6572,
      "training_step_time": 10.027068853378296
    },
    {
      "epoch": 2.5074005126953127e-05,
      "model_forward_time": 3.1987662315368652,
      "step": 6573
    },
    {
      "epoch": 2.5074005126953127e-05,
      "step": 6573,
      "training_step_time": 9.977423429489136
    },
    {
      "epoch": 2.5074005126953127e-05,
      "model_forward_time": 3.1448824405670166,
      "step": 6573
    },
    {
      "epoch": 2.5074005126953127e-05,
      "step": 6573,
      "training_step_time": 9.923585176467896
    },
    {
      "epoch": 2.5074005126953127e-05,
      "model_forward_time": 3.1552937030792236,
      "step": 6573
    },
    {
      "epoch": 2.5074005126953127e-05,
      "step": 6573,
      "training_step_time": 10.232397317886353
    },
    {
      "epoch": 2.5074005126953127e-05,
      "model_forward_time": 2.450845241546631,
      "step": 6573
    },
    {
      "epoch": 2.5074005126953127e-05,
      "step": 6573,
      "training_step_time": 10.317274332046509
    },
    {
      "epoch": 2.507781982421875e-05,
      "model_forward_time": 3.1955039501190186,
      "step": 6574
    },
    {
      "epoch": 2.507781982421875e-05,
      "step": 6574,
      "training_step_time": 9.969856023788452
    },
    {
      "epoch": 2.507781982421875e-05,
      "model_forward_time": 3.1414456367492676,
      "step": 6574
    },
    {
      "epoch": 2.507781982421875e-05,
      "step": 6574,
      "training_step_time": 9.918335914611816
    },
    {
      "epoch": 2.507781982421875e-05,
      "model_forward_time": 3.142298460006714,
      "step": 6574
    },
    {
      "epoch": 2.507781982421875e-05,
      "step": 6574,
      "training_step_time": 9.915530443191528
    },
    {
      "epoch": 2.507781982421875e-05,
      "model_forward_time": 2.443344831466675,
      "step": 6574
    },
    {
      "epoch": 2.507781982421875e-05,
      "step": 6574,
      "training_step_time": 10.316571474075317
    },
    {
      "epoch": 2.5081634521484376e-05,
      "model_forward_time": 3.198288679122925,
      "step": 6575
    },
    {
      "dynamics_loss_avg": 0.0695667915046215,
      "epoch": 2.5081634521484376e-05,
      "step": 6575
    },
    {
      "action_loss_avg": 0.028710872773081063,
      "epoch": 2.5081634521484376e-05,
      "step": 6575
    },
    {
      "epoch": 2.5081634521484376e-05,
      "step": 6575,
      "training_step_time": 9.970924615859985
    },
    {
      "epoch": 2.5081634521484376e-05,
      "model_forward_time": 3.1392788887023926,
      "step": 6575
    },
    {
      "epoch": 2.5081634521484376e-05,
      "step": 6575,
      "training_step_time": 9.911585330963135
    },
    {
      "epoch": 2.5081634521484376e-05,
      "model_forward_time": 3.1451549530029297,
      "step": 6575
    },
    {
      "epoch": 2.5081634521484376e-05,
      "step": 6575,
      "training_step_time": 9.916346073150635
    },
    {
      "epoch": 2.5081634521484376e-05,
      "model_forward_time": 3.14085054397583,
      "step": 6575
    },
    {
      "epoch": 2.5081634521484376e-05,
      "step": 6575,
      "training_step_time": 10.00968074798584
    },
    {
      "epoch": 2.508544921875e-05,
      "model_forward_time": 3.1774861812591553,
      "step": 6576
    },
    {
      "epoch": 2.508544921875e-05,
      "step": 6576,
      "training_step_time": 10.22797417640686
    },
    {
      "epoch": 2.508544921875e-05,
      "model_forward_time": 3.138544797897339,
      "step": 6576
    },
    {
      "epoch": 2.508544921875e-05,
      "step": 6576,
      "training_step_time": 9.914365291595459
    },
    {
      "epoch": 2.508544921875e-05,
      "model_forward_time": 3.143923044204712,
      "step": 6576
    },
    {
      "epoch": 2.508544921875e-05,
      "step": 6576,
      "training_step_time": 9.92102861404419
    },
    {
      "epoch": 2.508544921875e-05,
      "model_forward_time": 3.1424875259399414,
      "step": 6576
    },
    {
      "epoch": 2.508544921875e-05,
      "step": 6576,
      "training_step_time": 10.283841133117676
    },
    {
      "epoch": 2.5089263916015625e-05,
      "model_forward_time": 3.1753485202789307,
      "step": 6577
    },
    {
      "epoch": 2.5089263916015625e-05,
      "step": 6577,
      "training_step_time": 9.952053308486938
    },
    {
      "epoch": 2.5089263916015625e-05,
      "model_forward_time": 3.44706654548645,
      "step": 6577
    },
    {
      "epoch": 2.5089263916015625e-05,
      "step": 6577,
      "training_step_time": 10.217681169509888
    },
    {
      "epoch": 2.5089263916015625e-05,
      "model_forward_time": 3.1419177055358887,
      "step": 6577
    },
    {
      "dynamics_loss_avg": 0.06222406160086393,
      "epoch": 2.5089263916015625e-05,
      "step": 6577
    },
    {
      "action_loss_avg": 0.02172396732494235,
      "epoch": 2.5089263916015625e-05,
      "step": 6577
    },
    {
      "epoch": 2.5089263916015625e-05,
      "step": 6577,
      "training_step_time": 9.916501998901367
    },
    {
      "epoch": 2.5089263916015625e-05,
      "model_forward_time": 3.133277416229248,
      "step": 6577
    },
    {
      "epoch": 2.5089263916015625e-05,
      "step": 6577,
      "training_step_time": 10.316151142120361
    },
    {
      "epoch": 2.509307861328125e-05,
      "model_forward_time": 3.183321475982666,
      "step": 6578
    },
    {
      "epoch": 2.509307861328125e-05,
      "step": 6578,
      "training_step_time": 9.964475154876709
    },
    {
      "epoch": 2.509307861328125e-05,
      "model_forward_time": 3.1441121101379395,
      "step": 6578
    },
    {
      "epoch": 2.509307861328125e-05,
      "step": 6578,
      "training_step_time": 9.921140432357788
    },
    {
      "epoch": 2.509307861328125e-05,
      "model_forward_time": 3.434974431991577,
      "step": 6578
    },
    {
      "epoch": 2.509307861328125e-05,
      "step": 6578,
      "training_step_time": 10.208728551864624
    },
    {
      "epoch": 2.509307861328125e-05,
      "model_forward_time": 3.14168643951416,
      "step": 6578
    },
    {
      "epoch": 2.509307861328125e-05,
      "step": 6578,
      "training_step_time": 10.026272296905518
    },
    {
      "epoch": 2.5096893310546874e-05,
      "model_forward_time": 3.1688740253448486,
      "step": 6579
    },
    {
      "epoch": 2.5096893310546874e-05,
      "step": 6579,
      "training_step_time": 9.941195964813232
    },
    {
      "epoch": 2.5096893310546874e-05,
      "model_forward_time": 3.1428380012512207,
      "step": 6579
    },
    {
      "epoch": 2.5096893310546874e-05,
      "step": 6579,
      "training_step_time": 9.932616233825684
    },
    {
      "epoch": 2.5096893310546874e-05,
      "model_forward_time": 3.150055170059204,
      "step": 6579
    },
    {
      "epoch": 2.5096893310546874e-05,
      "step": 6579,
      "training_step_time": 9.932738542556763
    },
    {
      "epoch": 2.5096893310546874e-05,
      "model_forward_time": 3.148341417312622,
      "step": 6579
    },
    {
      "epoch": 2.5096893310546874e-05,
      "step": 6579,
      "training_step_time": 10.350400924682617
    },
    {
      "epoch": 2.51007080078125e-05,
      "grad_norm": 0.3784235119819641,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 6580
    },
    {
      "epoch": 2.51007080078125e-05,
      "model_forward_time": 3.1752355098724365,
      "step": 6580
    },
    {
      "dynamics_loss_avg": 0.06634223163127899,
      "epoch": 2.51007080078125e-05,
      "step": 6580
    },
    {
      "action_loss_avg": 0.03586462875828147,
      "epoch": 2.51007080078125e-05,
      "step": 6580
    },
    {
      "epoch": 2.51007080078125e-05,
      "step": 6580,
      "training_step_time": 9.960169315338135
    },
    {
      "epoch": 2.51007080078125e-05,
      "model_forward_time": 3.1439638137817383,
      "step": 6580
    },
    {
      "epoch": 2.51007080078125e-05,
      "step": 6580,
      "training_step_time": 9.926216840744019
    },
    {
      "epoch": 2.51007080078125e-05,
      "model_forward_time": 3.140913248062134,
      "step": 6580
    },
    {
      "epoch": 2.51007080078125e-05,
      "step": 6580,
      "training_step_time": 9.92323112487793
    },
    {
      "epoch": 2.51007080078125e-05,
      "model_forward_time": 3.144487142562866,
      "step": 6580
    },
    {
      "epoch": 2.51007080078125e-05,
      "step": 6580,
      "training_step_time": 10.027414798736572
    },
    {
      "epoch": 2.5104522705078127e-05,
      "model_forward_time": 3.4748828411102295,
      "step": 6581
    },
    {
      "epoch": 2.5104522705078127e-05,
      "step": 6581,
      "training_step_time": 10.256104230880737
    },
    {
      "epoch": 2.5104522705078127e-05,
      "model_forward_time": 3.137956380844116,
      "step": 6581
    },
    {
      "epoch": 2.5104522705078127e-05,
      "step": 6581,
      "training_step_time": 9.918877124786377
    },
    {
      "epoch": 2.5104522705078127e-05,
      "model_forward_time": 3.135793447494507,
      "step": 6581
    },
    {
      "epoch": 2.5104522705078127e-05,
      "step": 6581,
      "training_step_time": 9.92061448097229
    },
    {
      "epoch": 2.5104522705078127e-05,
      "model_forward_time": 3.142601251602173,
      "step": 6581
    },
    {
      "epoch": 2.5104522705078127e-05,
      "step": 6581,
      "training_step_time": 10.299160718917847
    },
    {
      "epoch": 2.510833740234375e-05,
      "model_forward_time": 3.1864466667175293,
      "step": 6582
    },
    {
      "epoch": 2.510833740234375e-05,
      "step": 6582,
      "training_step_time": 9.968914270401001
    },
    {
      "epoch": 2.510833740234375e-05,
      "model_forward_time": 3.1415820121765137,
      "step": 6582
    },
    {
      "epoch": 2.510833740234375e-05,
      "step": 6582,
      "training_step_time": 10.2058584690094
    },
    {
      "epoch": 2.510833740234375e-05,
      "model_forward_time": 3.1456611156463623,
      "step": 6582
    },
    {
      "dynamics_loss_avg": 0.054602443985641,
      "epoch": 2.510833740234375e-05,
      "step": 6582
    },
    {
      "action_loss_avg": 0.024815340340137482,
      "epoch": 2.510833740234375e-05,
      "step": 6582
    },
    {
      "epoch": 2.510833740234375e-05,
      "step": 6582,
      "training_step_time": 9.922710180282593
    },
    {
      "epoch": 2.510833740234375e-05,
      "model_forward_time": 3.1505839824676514,
      "step": 6582
    },
    {
      "epoch": 2.510833740234375e-05,
      "step": 6582,
      "training_step_time": 10.334707975387573
    },
    {
      "epoch": 2.5112152099609376e-05,
      "model_forward_time": 3.18717098236084,
      "step": 6583
    },
    {
      "epoch": 2.5112152099609376e-05,
      "step": 6583,
      "training_step_time": 9.963653802871704
    },
    {
      "epoch": 2.5112152099609376e-05,
      "model_forward_time": 3.138313055038452,
      "step": 6583
    },
    {
      "epoch": 2.5112152099609376e-05,
      "step": 6583,
      "training_step_time": 9.911932229995728
    },
    {
      "epoch": 2.5112152099609376e-05,
      "model_forward_time": 3.4417459964752197,
      "step": 6583
    },
    {
      "epoch": 2.5112152099609376e-05,
      "step": 6583,
      "training_step_time": 10.214065074920654
    },
    {
      "epoch": 2.5112152099609376e-05,
      "model_forward_time": 3.1397757530212402,
      "step": 6583
    },
    {
      "epoch": 2.5112152099609376e-05,
      "step": 6583,
      "training_step_time": 10.315866708755493
    },
    {
      "epoch": 2.5115966796875e-05,
      "model_forward_time": 3.1659159660339355,
      "step": 6584
    },
    {
      "epoch": 2.5115966796875e-05,
      "step": 6584,
      "training_step_time": 9.951828479766846
    },
    {
      "epoch": 2.5115966796875e-05,
      "model_forward_time": 3.130147695541382,
      "step": 6584
    },
    {
      "epoch": 2.5115966796875e-05,
      "step": 6584,
      "training_step_time": 9.908691644668579
    },
    {
      "epoch": 2.5115966796875e-05,
      "model_forward_time": 3.1416420936584473,
      "step": 6584
    },
    {
      "epoch": 2.5115966796875e-05,
      "step": 6584,
      "training_step_time": 9.924197912216187
    },
    {
      "epoch": 2.5115966796875e-05,
      "model_forward_time": 3.14237642288208,
      "step": 6584
    },
    {
      "epoch": 2.5115966796875e-05,
      "step": 6584,
      "training_step_time": 10.316062927246094
    },
    {
      "epoch": 2.5119781494140625e-05,
      "model_forward_time": 2.4877679347991943,
      "step": 6585
    },
    {
      "dynamics_loss_avg": 0.05327414385974407,
      "epoch": 2.5119781494140625e-05,
      "step": 6585
    },
    {
      "action_loss_avg": 0.020468449220061302,
      "epoch": 2.5119781494140625e-05,
      "step": 6585
    },
    {
      "epoch": 2.5119781494140625e-05,
      "step": 6585,
      "training_step_time": 7.52463698387146
    },
    {
      "epoch": 2.5119781494140625e-05,
      "model_forward_time": 3.1374316215515137,
      "step": 6585
    },
    {
      "epoch": 2.5119781494140625e-05,
      "step": 6585,
      "training_step_time": 9.91903567314148
    },
    {
      "epoch": 2.5119781494140625e-05,
      "model_forward_time": 3.139284372329712,
      "step": 6585
    },
    {
      "epoch": 2.5119781494140625e-05,
      "step": 6585,
      "training_step_time": 9.919461250305176
    },
    {
      "epoch": 2.5119781494140625e-05,
      "model_forward_time": 3.1400792598724365,
      "step": 6585
    },
    {
      "epoch": 2.5119781494140625e-05,
      "step": 6585,
      "training_step_time": 10.307327508926392
    },
    {
      "epoch": 2.512359619140625e-05,
      "model_forward_time": 3.4563803672790527,
      "step": 6586
    },
    {
      "epoch": 2.512359619140625e-05,
      "step": 6586,
      "training_step_time": 10.236998319625854
    },
    {
      "epoch": 2.512359619140625e-05,
      "model_forward_time": 3.1355295181274414,
      "step": 6586
    },
    {
      "epoch": 2.512359619140625e-05,
      "step": 6586,
      "training_step_time": 9.919398307800293
    },
    {
      "epoch": 2.512359619140625e-05,
      "model_forward_time": 3.146425247192383,
      "step": 6586
    },
    {
      "epoch": 2.512359619140625e-05,
      "step": 6586,
      "training_step_time": 9.926702737808228
    },
    {
      "epoch": 2.512359619140625e-05,
      "model_forward_time": 3.1333019733428955,
      "step": 6586
    },
    {
      "epoch": 2.512359619140625e-05,
      "step": 6586,
      "training_step_time": 10.269876956939697
    },
    {
      "epoch": 2.5127410888671874e-05,
      "model_forward_time": 3.172268867492676,
      "step": 6587
    },
    {
      "epoch": 2.5127410888671874e-05,
      "step": 6587,
      "training_step_time": 9.949886798858643
    },
    {
      "epoch": 2.5127410888671874e-05,
      "model_forward_time": 3.4040684700012207,
      "step": 6587
    },
    {
      "epoch": 2.5127410888671874e-05,
      "step": 6587,
      "training_step_time": 10.18760895729065
    },
    {
      "epoch": 2.5127410888671874e-05,
      "model_forward_time": 3.1414573192596436,
      "step": 6587
    },
    {
      "dynamics_loss_avg": 0.054870708473026754,
      "epoch": 2.5127410888671874e-05,
      "step": 6587
    },
    {
      "action_loss_avg": 0.019786673597991467,
      "epoch": 2.5127410888671874e-05,
      "step": 6587
    },
    {
      "epoch": 2.5127410888671874e-05,
      "step": 6587,
      "training_step_time": 9.916913747787476
    },
    {
      "epoch": 2.5127410888671874e-05,
      "model_forward_time": 3.1410906314849854,
      "step": 6587
    },
    {
      "epoch": 2.5127410888671874e-05,
      "step": 6587,
      "training_step_time": 10.3128342628479
    },
    {
      "epoch": 2.51312255859375e-05,
      "model_forward_time": 3.19028902053833,
      "step": 6588
    },
    {
      "epoch": 2.51312255859375e-05,
      "step": 6588,
      "training_step_time": 9.967280387878418
    },
    {
      "epoch": 2.51312255859375e-05,
      "model_forward_time": 3.1441609859466553,
      "step": 6588
    },
    {
      "epoch": 2.51312255859375e-05,
      "step": 6588,
      "training_step_time": 9.924365997314453
    },
    {
      "epoch": 2.51312255859375e-05,
      "model_forward_time": 3.142831563949585,
      "step": 6588
    },
    {
      "epoch": 2.51312255859375e-05,
      "step": 6588,
      "training_step_time": 10.21373176574707
    },
    {
      "epoch": 2.51312255859375e-05,
      "model_forward_time": 3.1425564289093018,
      "step": 6588
    },
    {
      "epoch": 2.51312255859375e-05,
      "step": 6588,
      "training_step_time": 10.0203857421875
    },
    {
      "epoch": 2.5135040283203127e-05,
      "model_forward_time": 3.1773364543914795,
      "step": 6589
    },
    {
      "epoch": 2.5135040283203127e-05,
      "step": 6589,
      "training_step_time": 9.965317249298096
    },
    {
      "epoch": 2.5135040283203127e-05,
      "model_forward_time": 3.141045093536377,
      "step": 6589
    },
    {
      "epoch": 2.5135040283203127e-05,
      "step": 6589,
      "training_step_time": 9.921897172927856
    },
    {
      "epoch": 2.5135040283203127e-05,
      "model_forward_time": 3.135085105895996,
      "step": 6589
    },
    {
      "epoch": 2.5135040283203127e-05,
      "step": 6589,
      "training_step_time": 9.914860963821411
    },
    {
      "epoch": 2.5135040283203127e-05,
      "model_forward_time": 3.13881778717041,
      "step": 6589
    },
    {
      "epoch": 2.5135040283203127e-05,
      "step": 6589,
      "training_step_time": 10.32395339012146
    },
    {
      "epoch": 2.513885498046875e-05,
      "grad_norm": 0.4351724088191986,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 6590
    },
    {
      "epoch": 2.513885498046875e-05,
      "model_forward_time": 3.189905881881714,
      "step": 6590
    },
    {
      "dynamics_loss_avg": 0.04816662035882473,
      "epoch": 2.513885498046875e-05,
      "step": 6590
    },
    {
      "action_loss_avg": 0.020036160107702018,
      "epoch": 2.513885498046875e-05,
      "step": 6590
    },
    {
      "epoch": 2.513885498046875e-05,
      "step": 6590,
      "training_step_time": 9.975218534469604
    },
    {
      "epoch": 2.513885498046875e-05,
      "model_forward_time": 3.150878667831421,
      "step": 6590
    },
    {
      "epoch": 2.513885498046875e-05,
      "step": 6590,
      "training_step_time": 9.93625545501709
    },
    {
      "epoch": 2.513885498046875e-05,
      "model_forward_time": 3.1484408378601074,
      "step": 6590
    },
    {
      "epoch": 2.513885498046875e-05,
      "step": 6590,
      "training_step_time": 9.932440996170044
    },
    {
      "epoch": 2.513885498046875e-05,
      "model_forward_time": 3.157956600189209,
      "step": 6590
    },
    {
      "epoch": 2.513885498046875e-05,
      "step": 6590,
      "training_step_time": 10.300884008407593
    },
    {
      "epoch": 2.5142669677734376e-05,
      "model_forward_time": 3.1632492542266846,
      "step": 6591
    },
    {
      "epoch": 2.5142669677734376e-05,
      "step": 6591,
      "training_step_time": 10.234840154647827
    },
    {
      "epoch": 2.5142669677734376e-05,
      "model_forward_time": 3.1431384086608887,
      "step": 6591
    },
    {
      "epoch": 2.5142669677734376e-05,
      "step": 6591,
      "training_step_time": 9.926162958145142
    },
    {
      "epoch": 2.5142669677734376e-05,
      "model_forward_time": 3.134045124053955,
      "step": 6591
    },
    {
      "epoch": 2.5142669677734376e-05,
      "step": 6591,
      "training_step_time": 9.914320707321167
    },
    {
      "epoch": 2.5142669677734376e-05,
      "model_forward_time": 3.139085531234741,
      "step": 6591
    },
    {
      "epoch": 2.5142669677734376e-05,
      "step": 6591,
      "training_step_time": 10.299386501312256
    },
    {
      "epoch": 2.5146484375e-05,
      "model_forward_time": 3.185476779937744,
      "step": 6592
    },
    {
      "epoch": 2.5146484375e-05,
      "step": 6592,
      "training_step_time": 9.965009212493896
    },
    {
      "epoch": 2.5146484375e-05,
      "model_forward_time": 3.4109714031219482,
      "step": 6592
    },
    {
      "epoch": 2.5146484375e-05,
      "step": 6592,
      "training_step_time": 10.182637214660645
    },
    {
      "epoch": 2.5146484375e-05,
      "model_forward_time": 3.1310811042785645,
      "step": 6592
    },
    {
      "dynamics_loss_avg": 0.05894717946648598,
      "epoch": 2.5146484375e-05,
      "step": 6592
    },
    {
      "action_loss_avg": 0.019816956715658307,
      "epoch": 2.5146484375e-05,
      "step": 6592
    },
    {
      "epoch": 2.5146484375e-05,
      "step": 6592,
      "training_step_time": 9.913054466247559
    },
    {
      "epoch": 2.5146484375e-05,
      "model_forward_time": 3.1364142894744873,
      "step": 6592
    },
    {
      "epoch": 2.5146484375e-05,
      "step": 6592,
      "training_step_time": 10.320221900939941
    },
    {
      "epoch": 2.5150299072265625e-05,
      "model_forward_time": 3.1837780475616455,
      "step": 6593
    },
    {
      "epoch": 2.5150299072265625e-05,
      "step": 6593,
      "training_step_time": 9.96029543876648
    },
    {
      "epoch": 2.5150299072265625e-05,
      "model_forward_time": 3.1329739093780518,
      "step": 6593
    },
    {
      "epoch": 2.5150299072265625e-05,
      "step": 6593,
      "training_step_time": 9.902487754821777
    },
    {
      "epoch": 2.5150299072265625e-05,
      "model_forward_time": 3.133814573287964,
      "step": 6593
    },
    {
      "epoch": 2.5150299072265625e-05,
      "step": 6593,
      "training_step_time": 10.177597761154175
    },
    {
      "epoch": 2.5150299072265625e-05,
      "model_forward_time": 3.13266658782959,
      "step": 6593
    },
    {
      "epoch": 2.5150299072265625e-05,
      "step": 6593,
      "training_step_time": 10.34168553352356
    },
    {
      "epoch": 2.515411376953125e-05,
      "model_forward_time": 3.173078775405884,
      "step": 6594
    },
    {
      "epoch": 2.515411376953125e-05,
      "step": 6594,
      "training_step_time": 9.949901580810547
    },
    {
      "epoch": 2.515411376953125e-05,
      "model_forward_time": 3.1345508098602295,
      "step": 6594
    },
    {
      "epoch": 2.515411376953125e-05,
      "step": 6594,
      "training_step_time": 9.908592462539673
    },
    {
      "epoch": 2.515411376953125e-05,
      "model_forward_time": 3.1415600776672363,
      "step": 6594
    },
    {
      "epoch": 2.515411376953125e-05,
      "step": 6594,
      "training_step_time": 9.91541337966919
    },
    {
      "epoch": 2.515411376953125e-05,
      "model_forward_time": 3.411383867263794,
      "step": 6594
    },
    {
      "epoch": 2.515411376953125e-05,
      "step": 6594,
      "training_step_time": 10.28502082824707
    },
    {
      "epoch": 2.5157928466796874e-05,
      "model_forward_time": 3.1827733516693115,
      "step": 6595
    },
    {
      "dynamics_loss_avg": 0.05135854315012693,
      "epoch": 2.5157928466796874e-05,
      "step": 6595
    },
    {
      "action_loss_avg": 0.021458082972094415,
      "epoch": 2.5157928466796874e-05,
      "step": 6595
    },
    {
      "epoch": 2.5157928466796874e-05,
      "step": 6595,
      "training_step_time": 9.957306385040283
    },
    {
      "epoch": 2.5157928466796874e-05,
      "model_forward_time": 3.138225793838501,
      "step": 6595
    },
    {
      "epoch": 2.5157928466796874e-05,
      "step": 6595,
      "training_step_time": 9.912941217422485
    },
    {
      "epoch": 2.5157928466796874e-05,
      "model_forward_time": 3.138896942138672,
      "step": 6595
    },
    {
      "epoch": 2.5157928466796874e-05,
      "step": 6595,
      "training_step_time": 9.914145469665527
    },
    {
      "epoch": 2.5157928466796874e-05,
      "model_forward_time": 3.136540651321411,
      "step": 6595
    },
    {
      "epoch": 2.5157928466796874e-05,
      "step": 6595,
      "training_step_time": 10.336437463760376
    },
    {
      "epoch": 2.51617431640625e-05,
      "model_forward_time": 3.477588415145874,
      "step": 6596
    },
    {
      "epoch": 2.51617431640625e-05,
      "step": 6596,
      "training_step_time": 10.249210357666016
    },
    {
      "epoch": 2.51617431640625e-05,
      "model_forward_time": 3.1351757049560547,
      "step": 6596
    },
    {
      "epoch": 2.51617431640625e-05,
      "step": 6596,
      "training_step_time": 9.910634756088257
    },
    {
      "epoch": 2.51617431640625e-05,
      "model_forward_time": 3.1374318599700928,
      "step": 6596
    },
    {
      "epoch": 2.51617431640625e-05,
      "step": 6596,
      "training_step_time": 9.911248683929443
    },
    {
      "epoch": 2.51617431640625e-05,
      "model_forward_time": 3.1334707736968994,
      "step": 6596
    },
    {
      "epoch": 2.51617431640625e-05,
      "step": 6596,
      "training_step_time": 10.347060441970825
    },
    {
      "epoch": 2.5165557861328127e-05,
      "model_forward_time": 3.175185441970825,
      "step": 6597
    },
    {
      "epoch": 2.5165557861328127e-05,
      "step": 6597,
      "training_step_time": 9.956414222717285
    },
    {
      "epoch": 2.5165557861328127e-05,
      "model_forward_time": 3.4289960861206055,
      "step": 6597
    },
    {
      "epoch": 2.5165557861328127e-05,
      "step": 6597,
      "training_step_time": 10.201581716537476
    },
    {
      "epoch": 2.5165557861328127e-05,
      "model_forward_time": 3.13801908493042,
      "step": 6597
    },
    {
      "dynamics_loss_avg": 0.06391145493835211,
      "epoch": 2.5165557861328127e-05,
      "step": 6597
    },
    {
      "action_loss_avg": 0.024098568968474864,
      "epoch": 2.5165557861328127e-05,
      "step": 6597
    },
    {
      "epoch": 2.5165557861328127e-05,
      "step": 6597,
      "training_step_time": 9.913050174713135
    },
    {
      "epoch": 2.5165557861328127e-05,
      "model_forward_time": 3.1430890560150146,
      "step": 6597
    },
    {
      "epoch": 2.5165557861328127e-05,
      "step": 6597,
      "training_step_time": 10.266578674316406
    },
    {
      "epoch": 2.516937255859375e-05,
      "model_forward_time": 3.1748342514038086,
      "step": 6598
    },
    {
      "epoch": 2.516937255859375e-05,
      "step": 6598,
      "training_step_time": 9.950305461883545
    },
    {
      "epoch": 2.516937255859375e-05,
      "model_forward_time": 3.1369102001190186,
      "step": 6598
    },
    {
      "epoch": 2.516937255859375e-05,
      "step": 6598,
      "training_step_time": 9.908564805984497
    },
    {
      "epoch": 2.516937255859375e-05,
      "model_forward_time": 3.411389112472534,
      "step": 6598
    },
    {
      "epoch": 2.516937255859375e-05,
      "step": 6598,
      "training_step_time": 10.179253101348877
    },
    {
      "epoch": 2.516937255859375e-05,
      "model_forward_time": 3.131103277206421,
      "step": 6598
    },
    {
      "epoch": 2.516937255859375e-05,
      "step": 6598,
      "training_step_time": 10.292325019836426
    },
    {
      "epoch": 2.5173187255859376e-05,
      "model_forward_time": 3.1703779697418213,
      "step": 6599
    },
    {
      "epoch": 2.5173187255859376e-05,
      "step": 6599,
      "training_step_time": 9.938822984695435
    },
    {
      "epoch": 2.5173187255859376e-05,
      "model_forward_time": 3.1371827125549316,
      "step": 6599
    },
    {
      "epoch": 2.5173187255859376e-05,
      "step": 6599,
      "training_step_time": 9.905359745025635
    },
    {
      "epoch": 2.5173187255859376e-05,
      "model_forward_time": 3.133467435836792,
      "step": 6599
    },
    {
      "epoch": 2.5173187255859376e-05,
      "step": 6599,
      "training_step_time": 9.905858755111694
    },
    {
      "epoch": 2.5173187255859376e-05,
      "model_forward_time": 3.1438610553741455,
      "step": 6599
    },
    {
      "epoch": 2.5173187255859376e-05,
      "step": 6599,
      "training_step_time": 10.29404354095459
    },
    {
      "epoch": 2.5177001953125e-05,
      "grad_norm": 0.37427395582199097,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 6600
    },
    {
      "epoch": 2.5177001953125e-05,
      "model_forward_time": 3.1873719692230225,
      "step": 6600
    },
    {
      "dynamics_loss_avg": 0.07013558074831963,
      "epoch": 2.5177001953125e-05,
      "step": 6600
    },
    {
      "action_loss_avg": 0.025433466769754887,
      "epoch": 2.5177001953125e-05,
      "step": 6600
    },
    {
      "epoch": 2.5177001953125e-05,
      "step": 6600,
      "training_step_time": 9.965415000915527
    },
    {
      "epoch": 2.5177001953125e-05,
      "model_forward_time": 3.144836664199829,
      "step": 6600
    },
    {
      "epoch": 2.5177001953125e-05,
      "step": 6600,
      "training_step_time": 9.918769836425781
    },
    {
      "epoch": 2.5177001953125e-05,
      "model_forward_time": 3.143949508666992,
      "step": 6600
    },
    {
      "epoch": 2.5177001953125e-05,
      "step": 6600,
      "training_step_time": 9.916478157043457
    },
    {
      "epoch": 2.5177001953125e-05,
      "model_forward_time": 3.1402783393859863,
      "step": 6600
    },
    {
      "epoch": 2.5177001953125e-05,
      "step": 6600,
      "training_step_time": 10.034961223602295
    },
    {
      "epoch": 2.5180816650390625e-05,
      "model_forward_time": 3.498304605484009,
      "step": 6601
    },
    {
      "epoch": 2.5180816650390625e-05,
      "step": 6601,
      "training_step_time": 10.27379035949707
    },
    {
      "epoch": 2.5180816650390625e-05,
      "model_forward_time": 3.1444358825683594,
      "step": 6601
    },
    {
      "epoch": 2.5180816650390625e-05,
      "step": 6601,
      "training_step_time": 9.915205001831055
    },
    {
      "epoch": 2.5180816650390625e-05,
      "model_forward_time": 3.142085552215576,
      "step": 6601
    },
    {
      "epoch": 2.5180816650390625e-05,
      "step": 6601,
      "training_step_time": 9.913838386535645
    },
    {
      "epoch": 2.5180816650390625e-05,
      "model_forward_time": 3.1456944942474365,
      "step": 6601
    },
    {
      "epoch": 2.5180816650390625e-05,
      "step": 6601,
      "training_step_time": 10.311255931854248
    },
    {
      "epoch": 2.518463134765625e-05,
      "model_forward_time": 3.187013864517212,
      "step": 6602
    },
    {
      "epoch": 2.518463134765625e-05,
      "step": 6602,
      "training_step_time": 9.963351011276245
    },
    {
      "epoch": 2.518463134765625e-05,
      "model_forward_time": 3.1427688598632812,
      "step": 6602
    },
    {
      "epoch": 2.518463134765625e-05,
      "step": 6602,
      "training_step_time": 10.190892934799194
    },
    {
      "epoch": 2.518463134765625e-05,
      "model_forward_time": 3.146573066711426,
      "step": 6602
    },
    {
      "dynamics_loss_avg": 0.05892653279006481,
      "epoch": 2.518463134765625e-05,
      "step": 6602
    },
    {
      "action_loss_avg": 0.025108943972736598,
      "epoch": 2.518463134765625e-05,
      "step": 6602
    },
    {
      "epoch": 2.518463134765625e-05,
      "step": 6602,
      "training_step_time": 9.924603700637817
    },
    {
      "epoch": 2.518463134765625e-05,
      "model_forward_time": 3.1450014114379883,
      "step": 6602
    },
    {
      "epoch": 2.518463134765625e-05,
      "step": 6602,
      "training_step_time": 10.284106016159058
    },
    {
      "epoch": 2.5188446044921874e-05,
      "model_forward_time": 3.1762750148773193,
      "step": 6603
    },
    {
      "epoch": 2.5188446044921874e-05,
      "step": 6603,
      "training_step_time": 9.949928283691406
    },
    {
      "epoch": 2.5188446044921874e-05,
      "model_forward_time": 3.1427369117736816,
      "step": 6603
    },
    {
      "epoch": 2.5188446044921874e-05,
      "step": 6603,
      "training_step_time": 9.921790361404419
    },
    {
      "epoch": 2.5188446044921874e-05,
      "model_forward_time": 3.4191572666168213,
      "step": 6603
    },
    {
      "epoch": 2.5188446044921874e-05,
      "step": 6603,
      "training_step_time": 10.196866512298584
    },
    {
      "epoch": 2.5188446044921874e-05,
      "model_forward_time": 3.1499249935150146,
      "step": 6603
    },
    {
      "epoch": 2.5188446044921874e-05,
      "step": 6603,
      "training_step_time": 10.323880910873413
    },
    {
      "epoch": 2.51922607421875e-05,
      "model_forward_time": 3.1882729530334473,
      "step": 6604
    },
    {
      "epoch": 2.51922607421875e-05,
      "step": 6604,
      "training_step_time": 9.97693657875061
    },
    {
      "epoch": 2.51922607421875e-05,
      "model_forward_time": 3.1401681900024414,
      "step": 6604
    },
    {
      "epoch": 2.51922607421875e-05,
      "step": 6604,
      "training_step_time": 9.91270136833191
    },
    {
      "epoch": 2.51922607421875e-05,
      "model_forward_time": 3.135254383087158,
      "step": 6604
    },
    {
      "epoch": 2.51922607421875e-05,
      "step": 6604,
      "training_step_time": 9.902559280395508
    },
    {
      "epoch": 2.51922607421875e-05,
      "model_forward_time": 3.1292123794555664,
      "step": 6604
    },
    {
      "epoch": 2.51922607421875e-05,
      "step": 6604,
      "training_step_time": 10.275101900100708
    },
    {
      "epoch": 2.5196075439453127e-05,
      "model_forward_time": 3.1650147438049316,
      "step": 6605
    },
    {
      "dynamics_loss_avg": 0.05473623424768448,
      "epoch": 2.5196075439453127e-05,
      "step": 6605
    },
    {
      "action_loss_avg": 0.02619317462667823,
      "epoch": 2.5196075439453127e-05,
      "step": 6605
    },
    {
      "epoch": 2.5196075439453127e-05,
      "step": 6605,
      "training_step_time": 9.939450740814209
    },
    {
      "epoch": 2.5196075439453127e-05,
      "model_forward_time": 3.133331775665283,
      "step": 6605
    },
    {
      "epoch": 2.5196075439453127e-05,
      "step": 6605,
      "training_step_time": 9.903449535369873
    },
    {
      "epoch": 2.5196075439453127e-05,
      "model_forward_time": 3.1372482776641846,
      "step": 6605
    },
    {
      "epoch": 2.5196075439453127e-05,
      "step": 6605,
      "training_step_time": 9.904923439025879
    },
    {
      "epoch": 2.5196075439453127e-05,
      "model_forward_time": 3.1326162815093994,
      "step": 6605
    },
    {
      "epoch": 2.5196075439453127e-05,
      "step": 6605,
      "training_step_time": 10.02300238609314
    },
    {
      "epoch": 2.519989013671875e-05,
      "model_forward_time": 3.458899736404419,
      "step": 6606
    },
    {
      "epoch": 2.519989013671875e-05,
      "step": 6606,
      "training_step_time": 10.2262282371521
    },
    {
      "epoch": 2.519989013671875e-05,
      "model_forward_time": 3.1412596702575684,
      "step": 6606
    },
    {
      "epoch": 2.519989013671875e-05,
      "step": 6606,
      "training_step_time": 9.908894300460815
    },
    {
      "epoch": 2.519989013671875e-05,
      "model_forward_time": 3.140711784362793,
      "step": 6606
    },
    {
      "epoch": 2.519989013671875e-05,
      "step": 6606,
      "training_step_time": 9.90901494026184
    },
    {
      "epoch": 2.519989013671875e-05,
      "model_forward_time": 3.140307903289795,
      "step": 6606
    },
    {
      "epoch": 2.519989013671875e-05,
      "step": 6606,
      "training_step_time": 10.291089057922363
    },
    {
      "epoch": 2.5203704833984376e-05,
      "model_forward_time": 3.1820719242095947,
      "step": 6607
    },
    {
      "epoch": 2.5203704833984376e-05,
      "step": 6607,
      "training_step_time": 9.951476097106934
    },
    {
      "epoch": 2.5203704833984376e-05,
      "model_forward_time": 3.405707836151123,
      "step": 6607
    },
    {
      "epoch": 2.5203704833984376e-05,
      "step": 6607,
      "training_step_time": 10.184788465499878
    },
    {
      "epoch": 2.5203704833984376e-05,
      "model_forward_time": 3.152679920196533,
      "step": 6607
    },
    {
      "dynamics_loss_avg": 0.04842299371957779,
      "epoch": 2.5203704833984376e-05,
      "step": 6607
    },
    {
      "action_loss_avg": 0.019012762559577824,
      "epoch": 2.5203704833984376e-05,
      "step": 6607
    },
    {
      "epoch": 2.5203704833984376e-05,
      "step": 6607,
      "training_step_time": 9.932503461837769
    },
    {
      "epoch": 2.5203704833984376e-05,
      "model_forward_time": 3.141233444213867,
      "step": 6607
    },
    {
      "epoch": 2.5203704833984376e-05,
      "step": 6607,
      "training_step_time": 10.274921417236328
    },
    {
      "epoch": 2.520751953125e-05,
      "model_forward_time": 3.178682327270508,
      "step": 6608
    },
    {
      "epoch": 2.520751953125e-05,
      "step": 6608,
      "training_step_time": 9.968980550765991
    },
    {
      "epoch": 2.520751953125e-05,
      "model_forward_time": 3.1456196308135986,
      "step": 6608
    },
    {
      "epoch": 2.520751953125e-05,
      "step": 6608,
      "training_step_time": 9.924099683761597
    },
    {
      "epoch": 2.520751953125e-05,
      "model_forward_time": 3.4713687896728516,
      "step": 6608
    },
    {
      "epoch": 2.520751953125e-05,
      "step": 6608,
      "training_step_time": 10.246845483779907
    },
    {
      "epoch": 2.520751953125e-05,
      "model_forward_time": 3.154761552810669,
      "step": 6608
    },
    {
      "epoch": 2.520751953125e-05,
      "step": 6608,
      "training_step_time": 10.327326536178589
    },
    {
      "epoch": 2.5211334228515625e-05,
      "model_forward_time": 3.19641375541687,
      "step": 6609
    },
    {
      "epoch": 2.5211334228515625e-05,
      "step": 6609,
      "training_step_time": 9.977301597595215
    },
    {
      "epoch": 2.5211334228515625e-05,
      "model_forward_time": 3.152782917022705,
      "step": 6609
    },
    {
      "epoch": 2.5211334228515625e-05,
      "step": 6609,
      "training_step_time": 9.928000926971436
    },
    {
      "epoch": 2.5211334228515625e-05,
      "model_forward_time": 3.1425058841705322,
      "step": 6609
    },
    {
      "epoch": 2.5211334228515625e-05,
      "step": 6609,
      "training_step_time": 9.914957761764526
    },
    {
      "epoch": 2.5211334228515625e-05,
      "model_forward_time": 3.4312307834625244,
      "step": 6609
    },
    {
      "epoch": 2.5211334228515625e-05,
      "step": 6609,
      "training_step_time": 10.295363664627075
    },
    {
      "epoch": 2.521514892578125e-05,
      "grad_norm": 0.4586222171783447,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 6610
    },
    {
      "epoch": 2.521514892578125e-05,
      "model_forward_time": 3.1810555458068848,
      "step": 6610
    },
    {
      "dynamics_loss_avg": 0.06227670628577471,
      "epoch": 2.521514892578125e-05,
      "step": 6610
    },
    {
      "action_loss_avg": 0.02298576645553112,
      "epoch": 2.521514892578125e-05,
      "step": 6610
    },
    {
      "epoch": 2.521514892578125e-05,
      "step": 6610,
      "training_step_time": 9.958229780197144
    },
    {
      "epoch": 2.521514892578125e-05,
      "model_forward_time": 3.152815341949463,
      "step": 6610
    },
    {
      "epoch": 2.521514892578125e-05,
      "step": 6610,
      "training_step_time": 9.926048040390015
    },
    {
      "epoch": 2.521514892578125e-05,
      "model_forward_time": 3.1466097831726074,
      "step": 6610
    },
    {
      "epoch": 2.521514892578125e-05,
      "step": 6610,
      "training_step_time": 9.917383670806885
    },
    {
      "epoch": 2.521514892578125e-05,
      "model_forward_time": 3.1532557010650635,
      "step": 6610
    },
    {
      "epoch": 2.521514892578125e-05,
      "step": 6610,
      "training_step_time": 10.032846450805664
    },
    {
      "epoch": 2.5218963623046874e-05,
      "model_forward_time": 3.4969143867492676,
      "step": 6611
    },
    {
      "epoch": 2.5218963623046874e-05,
      "step": 6611,
      "training_step_time": 10.272418975830078
    },
    {
      "epoch": 2.5218963623046874e-05,
      "model_forward_time": 3.1437761783599854,
      "step": 6611
    },
    {
      "epoch": 2.5218963623046874e-05,
      "step": 6611,
      "training_step_time": 9.913946151733398
    },
    {
      "epoch": 2.5218963623046874e-05,
      "model_forward_time": 3.143256425857544,
      "step": 6611
    },
    {
      "epoch": 2.5218963623046874e-05,
      "step": 6611,
      "training_step_time": 9.913076877593994
    },
    {
      "epoch": 2.5218963623046874e-05,
      "model_forward_time": 3.14371919631958,
      "step": 6611
    },
    {
      "epoch": 2.5218963623046874e-05,
      "step": 6611,
      "training_step_time": 10.262691259384155
    },
    {
      "epoch": 2.52227783203125e-05,
      "model_forward_time": 3.1792032718658447,
      "step": 6612
    },
    {
      "epoch": 2.52227783203125e-05,
      "step": 6612,
      "training_step_time": 9.954355239868164
    },
    {
      "epoch": 2.52227783203125e-05,
      "model_forward_time": 3.4310123920440674,
      "step": 6612
    },
    {
      "epoch": 2.52227783203125e-05,
      "step": 6612,
      "training_step_time": 10.201309442520142
    },
    {
      "epoch": 2.52227783203125e-05,
      "model_forward_time": 3.139230728149414,
      "step": 6612
    },
    {
      "dynamics_loss_avg": 0.058171288669109346,
      "epoch": 2.52227783203125e-05,
      "step": 6612
    },
    {
      "action_loss_avg": 0.026802263036370278,
      "epoch": 2.52227783203125e-05,
      "step": 6612
    },
    {
      "epoch": 2.52227783203125e-05,
      "step": 6612,
      "training_step_time": 9.905613899230957
    },
    {
      "epoch": 2.52227783203125e-05,
      "model_forward_time": 3.1439661979675293,
      "step": 6612
    },
    {
      "epoch": 2.52227783203125e-05,
      "step": 6612,
      "training_step_time": 10.251614332199097
    },
    {
      "epoch": 2.5226593017578127e-05,
      "model_forward_time": 3.1983630657196045,
      "step": 6613
    },
    {
      "epoch": 2.5226593017578127e-05,
      "step": 6613,
      "training_step_time": 9.968740224838257
    },
    {
      "epoch": 2.5226593017578127e-05,
      "model_forward_time": 3.1364760398864746,
      "step": 6613
    },
    {
      "epoch": 2.5226593017578127e-05,
      "step": 6613,
      "training_step_time": 9.90562391281128
    },
    {
      "epoch": 2.5226593017578127e-05,
      "model_forward_time": 3.4111011028289795,
      "step": 6613
    },
    {
      "epoch": 2.5226593017578127e-05,
      "step": 6613,
      "training_step_time": 10.19184398651123
    },
    {
      "epoch": 2.5226593017578127e-05,
      "model_forward_time": 3.154212474822998,
      "step": 6613
    },
    {
      "epoch": 2.5226593017578127e-05,
      "step": 6613,
      "training_step_time": 10.313090562820435
    },
    {
      "epoch": 2.523040771484375e-05,
      "model_forward_time": 3.186283826828003,
      "step": 6614
    },
    {
      "epoch": 2.523040771484375e-05,
      "step": 6614,
      "training_step_time": 9.97607135772705
    },
    {
      "epoch": 2.523040771484375e-05,
      "model_forward_time": 3.143601655960083,
      "step": 6614
    },
    {
      "epoch": 2.523040771484375e-05,
      "step": 6614,
      "training_step_time": 9.917617321014404
    },
    {
      "epoch": 2.523040771484375e-05,
      "model_forward_time": 3.143310785293579,
      "step": 6614
    },
    {
      "epoch": 2.523040771484375e-05,
      "step": 6614,
      "training_step_time": 9.922291040420532
    },
    {
      "epoch": 2.523040771484375e-05,
      "model_forward_time": 3.4560678005218506,
      "step": 6614
    },
    {
      "epoch": 2.523040771484375e-05,
      "step": 6614,
      "training_step_time": 10.325270175933838
    },
    {
      "epoch": 2.5234222412109376e-05,
      "model_forward_time": 3.1980323791503906,
      "step": 6615
    },
    {
      "dynamics_loss_avg": 0.05201630312949419,
      "epoch": 2.5234222412109376e-05,
      "step": 6615
    },
    {
      "action_loss_avg": 0.018783597834408285,
      "epoch": 2.5234222412109376e-05,
      "step": 6615
    },
    {
      "epoch": 2.5234222412109376e-05,
      "step": 6615,
      "training_step_time": 9.977442026138306
    },
    {
      "epoch": 2.5234222412109376e-05,
      "model_forward_time": 3.13505482673645,
      "step": 6615
    },
    {
      "epoch": 2.5234222412109376e-05,
      "step": 6615,
      "training_step_time": 9.904613256454468
    },
    {
      "epoch": 2.5234222412109376e-05,
      "model_forward_time": 3.1378915309906006,
      "step": 6615
    },
    {
      "epoch": 2.5234222412109376e-05,
      "step": 6615,
      "training_step_time": 9.909374713897705
    },
    {
      "epoch": 2.5234222412109376e-05,
      "model_forward_time": 3.1330602169036865,
      "step": 6615
    },
    {
      "epoch": 2.5234222412109376e-05,
      "step": 6615,
      "training_step_time": 10.016594171524048
    },
    {
      "epoch": 2.5238037109375e-05,
      "model_forward_time": 3.5324060916900635,
      "step": 6616
    },
    {
      "epoch": 2.5238037109375e-05,
      "step": 6616,
      "training_step_time": 10.308834791183472
    },
    {
      "epoch": 2.5238037109375e-05,
      "model_forward_time": 3.1440963745117188,
      "step": 6616
    },
    {
      "epoch": 2.5238037109375e-05,
      "step": 6616,
      "training_step_time": 9.918105840682983
    },
    {
      "epoch": 2.5238037109375e-05,
      "model_forward_time": 3.1487138271331787,
      "step": 6616
    },
    {
      "epoch": 2.5238037109375e-05,
      "step": 6616,
      "training_step_time": 9.923786640167236
    },
    {
      "epoch": 2.5238037109375e-05,
      "model_forward_time": 3.143846273422241,
      "step": 6616
    },
    {
      "epoch": 2.5238037109375e-05,
      "step": 6616,
      "training_step_time": 10.034210443496704
    },
    {
      "epoch": 2.5241851806640625e-05,
      "model_forward_time": 3.1829004287719727,
      "step": 6617
    },
    {
      "epoch": 2.5241851806640625e-05,
      "step": 6617,
      "training_step_time": 9.957881927490234
    },
    {
      "epoch": 2.5241851806640625e-05,
      "model_forward_time": 3.380106210708618,
      "step": 6617
    },
    {
      "epoch": 2.5241851806640625e-05,
      "step": 6617,
      "training_step_time": 10.156797170639038
    },
    {
      "epoch": 2.5241851806640625e-05,
      "model_forward_time": 3.14115571975708,
      "step": 6617
    },
    {
      "dynamics_loss_avg": 0.06415495723485946,
      "epoch": 2.5241851806640625e-05,
      "step": 6617
    },
    {
      "action_loss_avg": 0.025398334581404926,
      "epoch": 2.5241851806640625e-05,
      "step": 6617
    },
    {
      "epoch": 2.5241851806640625e-05,
      "step": 6617,
      "training_step_time": 9.918021440505981
    },
    {
      "epoch": 2.5241851806640625e-05,
      "model_forward_time": 3.141814947128296,
      "step": 6617
    },
    {
      "epoch": 2.5241851806640625e-05,
      "step": 6617,
      "training_step_time": 10.31553864479065
    },
    {
      "epoch": 2.524566650390625e-05,
      "model_forward_time": 3.1574907302856445,
      "step": 6618
    },
    {
      "epoch": 2.524566650390625e-05,
      "step": 6618,
      "training_step_time": 9.931960344314575
    },
    {
      "epoch": 2.524566650390625e-05,
      "model_forward_time": 3.4262893199920654,
      "step": 6618
    },
    {
      "epoch": 2.524566650390625e-05,
      "step": 6618,
      "training_step_time": 10.194336175918579
    },
    {
      "epoch": 2.524566650390625e-05,
      "model_forward_time": 3.1326992511749268,
      "step": 6618
    },
    {
      "epoch": 2.524566650390625e-05,
      "step": 6618,
      "training_step_time": 9.898288488388062
    },
    {
      "epoch": 2.524566650390625e-05,
      "model_forward_time": 3.1351537704467773,
      "step": 6618
    },
    {
      "epoch": 2.524566650390625e-05,
      "step": 6618,
      "training_step_time": 10.298920154571533
    },
    {
      "epoch": 2.5249481201171874e-05,
      "model_forward_time": 3.1618764400482178,
      "step": 6619
    },
    {
      "epoch": 2.5249481201171874e-05,
      "step": 6619,
      "training_step_time": 9.92617392539978
    },
    {
      "epoch": 2.5249481201171874e-05,
      "model_forward_time": 3.138758420944214,
      "step": 6619
    },
    {
      "epoch": 2.5249481201171874e-05,
      "step": 6619,
      "training_step_time": 9.905258893966675
    },
    {
      "epoch": 2.5249481201171874e-05,
      "model_forward_time": 3.1291816234588623,
      "step": 6619
    },
    {
      "epoch": 2.5249481201171874e-05,
      "step": 6619,
      "training_step_time": 10.17865800857544
    },
    {
      "epoch": 2.5249481201171874e-05,
      "model_forward_time": 3.127962589263916,
      "step": 6619
    },
    {
      "epoch": 2.5249481201171874e-05,
      "step": 6619,
      "training_step_time": 10.00744080543518
    },
    {
      "epoch": 2.52532958984375e-05,
      "grad_norm": 0.33188873529434204,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 6620
    },
    {
      "epoch": 2.52532958984375e-05,
      "model_forward_time": 3.169218063354492,
      "step": 6620
    },
    {
      "dynamics_loss_avg": 0.07517426088452339,
      "epoch": 2.52532958984375e-05,
      "step": 6620
    },
    {
      "action_loss_avg": 0.02630314165726304,
      "epoch": 2.52532958984375e-05,
      "step": 6620
    },
    {
      "epoch": 2.52532958984375e-05,
      "step": 6620,
      "training_step_time": 9.93611741065979
    },
    {
      "epoch": 2.52532958984375e-05,
      "model_forward_time": 3.1332051753997803,
      "step": 6620
    },
    {
      "epoch": 2.52532958984375e-05,
      "step": 6620,
      "training_step_time": 9.900733470916748
    },
    {
      "epoch": 2.52532958984375e-05,
      "model_forward_time": 3.130885124206543,
      "step": 6620
    },
    {
      "epoch": 2.52532958984375e-05,
      "step": 6620,
      "training_step_time": 9.902744770050049
    },
    {
      "epoch": 2.52532958984375e-05,
      "model_forward_time": 3.3926889896392822,
      "step": 6620
    },
    {
      "epoch": 2.52532958984375e-05,
      "step": 6620,
      "training_step_time": 10.258163452148438
    },
    {
      "epoch": 2.5257110595703127e-05,
      "model_forward_time": 3.1718344688415527,
      "step": 6621
    },
    {
      "epoch": 2.5257110595703127e-05,
      "step": 6621,
      "training_step_time": 9.943262815475464
    },
    {
      "epoch": 2.5257110595703127e-05,
      "model_forward_time": 3.1617202758789062,
      "step": 6621
    },
    {
      "epoch": 2.5257110595703127e-05,
      "step": 6621,
      "training_step_time": 9.939810752868652
    },
    {
      "epoch": 2.5257110595703127e-05,
      "model_forward_time": 3.141460657119751,
      "step": 6621
    },
    {
      "epoch": 2.5257110595703127e-05,
      "step": 6621,
      "training_step_time": 9.913578510284424
    },
    {
      "epoch": 2.5257110595703127e-05,
      "model_forward_time": 3.1430182456970215,
      "step": 6621
    },
    {
      "epoch": 2.5257110595703127e-05,
      "step": 6621,
      "training_step_time": 10.307541847229004
    },
    {
      "epoch": 2.526092529296875e-05,
      "model_forward_time": 3.1829380989074707,
      "step": 6622
    },
    {
      "epoch": 2.526092529296875e-05,
      "step": 6622,
      "training_step_time": 10.26149582862854
    },
    {
      "epoch": 2.526092529296875e-05,
      "model_forward_time": 3.145479202270508,
      "step": 6622
    },
    {
      "epoch": 2.526092529296875e-05,
      "step": 6622,
      "training_step_time": 9.921806573867798
    },
    {
      "epoch": 2.526092529296875e-05,
      "model_forward_time": 3.1410417556762695,
      "step": 6622
    },
    {
      "dynamics_loss_avg": 0.05731095820665359,
      "epoch": 2.526092529296875e-05,
      "step": 6622
    },
    {
      "action_loss_avg": 0.021456203237175942,
      "epoch": 2.526092529296875e-05,
      "step": 6622
    },
    {
      "epoch": 2.526092529296875e-05,
      "step": 6622,
      "training_step_time": 9.911038637161255
    },
    {
      "epoch": 2.526092529296875e-05,
      "model_forward_time": 3.1426174640655518,
      "step": 6622
    },
    {
      "epoch": 2.526092529296875e-05,
      "step": 6622,
      "training_step_time": 10.300165891647339
    },
    {
      "epoch": 2.5264739990234376e-05,
      "model_forward_time": 3.1923022270202637,
      "step": 6623
    },
    {
      "epoch": 2.5264739990234376e-05,
      "step": 6623,
      "training_step_time": 9.967111825942993
    },
    {
      "epoch": 2.5264739990234376e-05,
      "model_forward_time": 3.4738011360168457,
      "step": 6623
    },
    {
      "epoch": 2.5264739990234376e-05,
      "step": 6623,
      "training_step_time": 10.24025559425354
    },
    {
      "epoch": 2.5264739990234376e-05,
      "model_forward_time": 3.138204336166382,
      "step": 6623
    },
    {
      "epoch": 2.5264739990234376e-05,
      "step": 6623,
      "training_step_time": 9.9055814743042
    },
    {
      "epoch": 2.5264739990234376e-05,
      "model_forward_time": 3.143686294555664,
      "step": 6623
    },
    {
      "epoch": 2.5264739990234376e-05,
      "step": 6623,
      "training_step_time": 10.2919602394104
    },
    {
      "epoch": 2.52685546875e-05,
      "model_forward_time": 3.171721935272217,
      "step": 6624
    },
    {
      "epoch": 2.52685546875e-05,
      "step": 6624,
      "training_step_time": 9.938498497009277
    },
    {
      "epoch": 2.52685546875e-05,
      "model_forward_time": 3.1370372772216797,
      "step": 6624
    },
    {
      "epoch": 2.52685546875e-05,
      "step": 6624,
      "training_step_time": 9.902206182479858
    },
    {
      "epoch": 2.52685546875e-05,
      "model_forward_time": 3.4046645164489746,
      "step": 6624
    },
    {
      "epoch": 2.52685546875e-05,
      "step": 6624,
      "training_step_time": 10.17235255241394
    },
    {
      "epoch": 2.52685546875e-05,
      "model_forward_time": 3.145611524581909,
      "step": 6624
    },
    {
      "epoch": 2.52685546875e-05,
      "step": 6624,
      "training_step_time": 10.022274494171143
    },
    {
      "epoch": 2.5272369384765625e-05,
      "model_forward_time": 3.1987967491149902,
      "step": 6625
    },
    {
      "dynamics_loss_avg": 0.06804669052362441,
      "epoch": 2.5272369384765625e-05,
      "step": 6625
    },
    {
      "action_loss_avg": 0.027106967754662037,
      "epoch": 2.5272369384765625e-05,
      "step": 6625
    },
    {
      "epoch": 2.5272369384765625e-05,
      "step": 6625,
      "training_step_time": 9.995864391326904
    },
    {
      "epoch": 2.5272369384765625e-05,
      "model_forward_time": 3.1522278785705566,
      "step": 6625
    },
    {
      "epoch": 2.5272369384765625e-05,
      "step": 6625,
      "training_step_time": 9.927194118499756
    },
    {
      "epoch": 2.5272369384765625e-05,
      "model_forward_time": 3.143681526184082,
      "step": 6625
    },
    {
      "epoch": 2.5272369384765625e-05,
      "step": 6625,
      "training_step_time": 9.918118715286255
    },
    {
      "epoch": 2.5272369384765625e-05,
      "model_forward_time": 3.4830477237701416,
      "step": 6625
    },
    {
      "epoch": 2.5272369384765625e-05,
      "step": 6625,
      "training_step_time": 10.353166818618774
    },
    {
      "epoch": 2.527618408203125e-05,
      "model_forward_time": 3.16561222076416,
      "step": 6626
    },
    {
      "epoch": 2.527618408203125e-05,
      "step": 6626,
      "training_step_time": 9.941978693008423
    },
    {
      "epoch": 2.527618408203125e-05,
      "model_forward_time": 3.1490001678466797,
      "step": 6626
    },
    {
      "epoch": 2.527618408203125e-05,
      "step": 6626,
      "training_step_time": 9.922312021255493
    },
    {
      "epoch": 2.527618408203125e-05,
      "model_forward_time": 3.1527042388916016,
      "step": 6626
    },
    {
      "epoch": 2.527618408203125e-05,
      "step": 6626,
      "training_step_time": 9.930131196975708
    },
    {
      "epoch": 2.527618408203125e-05,
      "model_forward_time": 2.4471986293792725,
      "step": 6626
    },
    {
      "epoch": 2.527618408203125e-05,
      "step": 6626,
      "training_step_time": 10.302423477172852
    },
    {
      "epoch": 2.5279998779296874e-05,
      "model_forward_time": 3.4811456203460693,
      "step": 6627
    },
    {
      "epoch": 2.5279998779296874e-05,
      "step": 6627,
      "training_step_time": 10.255064725875854
    },
    {
      "epoch": 2.5279998779296874e-05,
      "model_forward_time": 3.137120008468628,
      "step": 6627
    },
    {
      "epoch": 2.5279998779296874e-05,
      "step": 6627,
      "training_step_time": 9.904199361801147
    },
    {
      "epoch": 2.5279998779296874e-05,
      "model_forward_time": 3.1440510749816895,
      "step": 6627
    },
    {
      "dynamics_loss_avg": 0.04936292991042137,
      "epoch": 2.5279998779296874e-05,
      "step": 6627
    },
    {
      "action_loss_avg": 0.015369820734485984,
      "epoch": 2.5279998779296874e-05,
      "step": 6627
    },
    {
      "epoch": 2.5279998779296874e-05,
      "step": 6627,
      "training_step_time": 9.91697645187378
    },
    {
      "epoch": 2.5279998779296874e-05,
      "model_forward_time": 3.1375300884246826,
      "step": 6627
    },
    {
      "epoch": 2.5279998779296874e-05,
      "step": 6627,
      "training_step_time": 10.30221152305603
    },
    {
      "epoch": 2.52838134765625e-05,
      "model_forward_time": 2.4727160930633545,
      "step": 6628
    },
    {
      "epoch": 2.52838134765625e-05,
      "step": 6628,
      "training_step_time": 7.509061336517334
    },
    {
      "epoch": 2.52838134765625e-05,
      "model_forward_time": 3.141225814819336,
      "step": 6628
    },
    {
      "epoch": 2.52838134765625e-05,
      "step": 6628,
      "training_step_time": 10.168177604675293
    },
    {
      "epoch": 2.52838134765625e-05,
      "model_forward_time": 3.143841028213501,
      "step": 6628
    },
    {
      "epoch": 2.52838134765625e-05,
      "step": 6628,
      "training_step_time": 9.916178464889526
    },
    {
      "epoch": 2.52838134765625e-05,
      "model_forward_time": 3.138517141342163,
      "step": 6628
    },
    {
      "epoch": 2.52838134765625e-05,
      "step": 6628,
      "training_step_time": 10.326617240905762
    },
    {
      "epoch": 2.5287628173828123e-05,
      "model_forward_time": 3.1719164848327637,
      "step": 6629
    },
    {
      "epoch": 2.5287628173828123e-05,
      "step": 6629,
      "training_step_time": 9.943300008773804
    },
    {
      "epoch": 2.5287628173828123e-05,
      "model_forward_time": 3.1433534622192383,
      "step": 6629
    },
    {
      "epoch": 2.5287628173828123e-05,
      "step": 6629,
      "training_step_time": 9.916354894638062
    },
    {
      "epoch": 2.5287628173828123e-05,
      "model_forward_time": 3.4654197692871094,
      "step": 6629
    },
    {
      "epoch": 2.5287628173828123e-05,
      "step": 6629,
      "training_step_time": 10.239007711410522
    },
    {
      "epoch": 2.5287628173828123e-05,
      "model_forward_time": 3.144047260284424,
      "step": 6629
    },
    {
      "epoch": 2.5287628173828123e-05,
      "step": 6629,
      "training_step_time": 10.309231042861938
    },
    {
      "epoch": 2.529144287109375e-05,
      "grad_norm": 0.5058490037918091,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 6630
    },
    {
      "epoch": 2.529144287109375e-05,
      "model_forward_time": 3.1804304122924805,
      "step": 6630
    },
    {
      "dynamics_loss_avg": 0.05670791622251272,
      "epoch": 2.529144287109375e-05,
      "step": 6630
    },
    {
      "action_loss_avg": 0.019535115640610456,
      "epoch": 2.529144287109375e-05,
      "step": 6630
    },
    {
      "epoch": 2.529144287109375e-05,
      "step": 6630,
      "training_step_time": 9.957159996032715
    },
    {
      "epoch": 2.529144287109375e-05,
      "model_forward_time": 3.1473703384399414,
      "step": 6630
    },
    {
      "epoch": 2.529144287109375e-05,
      "step": 6630,
      "training_step_time": 9.925799369812012
    },
    {
      "epoch": 2.529144287109375e-05,
      "model_forward_time": 3.1495041847229004,
      "step": 6630
    },
    {
      "epoch": 2.529144287109375e-05,
      "step": 6630,
      "training_step_time": 9.93573808670044
    },
    {
      "epoch": 2.529144287109375e-05,
      "model_forward_time": 3.4433906078338623,
      "step": 6630
    },
    {
      "epoch": 2.529144287109375e-05,
      "step": 6630,
      "training_step_time": 10.310702085494995
    },
    {
      "epoch": 2.5295257568359376e-05,
      "model_forward_time": 3.1681737899780273,
      "step": 6631
    },
    {
      "epoch": 2.5295257568359376e-05,
      "step": 6631,
      "training_step_time": 9.938288450241089
    },
    {
      "epoch": 2.5295257568359376e-05,
      "model_forward_time": 3.1370532512664795,
      "step": 6631
    },
    {
      "epoch": 2.5295257568359376e-05,
      "step": 6631,
      "training_step_time": 9.907639741897583
    },
    {
      "epoch": 2.5295257568359376e-05,
      "model_forward_time": 3.138488292694092,
      "step": 6631
    },
    {
      "epoch": 2.5295257568359376e-05,
      "step": 6631,
      "training_step_time": 9.907939434051514
    },
    {
      "epoch": 2.5295257568359376e-05,
      "model_forward_time": 3.146796464920044,
      "step": 6631
    },
    {
      "epoch": 2.5295257568359376e-05,
      "step": 6631,
      "training_step_time": 10.299721956253052
    },
    {
      "epoch": 2.5299072265625e-05,
      "model_forward_time": 3.4443583488464355,
      "step": 6632
    },
    {
      "epoch": 2.5299072265625e-05,
      "step": 6632,
      "training_step_time": 10.21163034439087
    },
    {
      "epoch": 2.5299072265625e-05,
      "model_forward_time": 3.137273073196411,
      "step": 6632
    },
    {
      "epoch": 2.5299072265625e-05,
      "step": 6632,
      "training_step_time": 9.908478498458862
    },
    {
      "epoch": 2.5299072265625e-05,
      "model_forward_time": 3.135709762573242,
      "step": 6632
    },
    {
      "dynamics_loss_avg": 0.052398450672626495,
      "epoch": 2.5299072265625e-05,
      "step": 6632
    },
    {
      "action_loss_avg": 0.023552227206528185,
      "epoch": 2.5299072265625e-05,
      "step": 6632
    },
    {
      "epoch": 2.5299072265625e-05,
      "step": 6632,
      "training_step_time": 9.91438341140747
    },
    {
      "epoch": 2.5299072265625e-05,
      "model_forward_time": 3.138134241104126,
      "step": 6632
    },
    {
      "epoch": 2.5299072265625e-05,
      "step": 6632,
      "training_step_time": 10.313566207885742
    },
    {
      "epoch": 2.5302886962890625e-05,
      "model_forward_time": 3.159132242202759,
      "step": 6633
    },
    {
      "epoch": 2.5302886962890625e-05,
      "step": 6633,
      "training_step_time": 9.937838077545166
    },
    {
      "epoch": 2.5302886962890625e-05,
      "model_forward_time": 3.4805514812469482,
      "step": 6633
    },
    {
      "epoch": 2.5302886962890625e-05,
      "step": 6633,
      "training_step_time": 10.251898288726807
    },
    {
      "epoch": 2.5302886962890625e-05,
      "model_forward_time": 3.148960590362549,
      "step": 6633
    },
    {
      "epoch": 2.5302886962890625e-05,
      "step": 6633,
      "training_step_time": 9.925183296203613
    },
    {
      "epoch": 2.5302886962890625e-05,
      "model_forward_time": 2.446384906768799,
      "step": 6633
    },
    {
      "epoch": 2.5302886962890625e-05,
      "step": 6633,
      "training_step_time": 10.254374504089355
    },
    {
      "epoch": 2.530670166015625e-05,
      "model_forward_time": 3.1870601177215576,
      "step": 6634
    },
    {
      "epoch": 2.530670166015625e-05,
      "step": 6634,
      "training_step_time": 9.960578680038452
    },
    {
      "epoch": 2.530670166015625e-05,
      "model_forward_time": 3.150177240371704,
      "step": 6634
    },
    {
      "epoch": 2.530670166015625e-05,
      "step": 6634,
      "training_step_time": 9.926884889602661
    },
    {
      "epoch": 2.530670166015625e-05,
      "model_forward_time": 3.450802803039551,
      "step": 6634
    },
    {
      "epoch": 2.530670166015625e-05,
      "step": 6634,
      "training_step_time": 10.226851463317871
    },
    {
      "epoch": 2.530670166015625e-05,
      "model_forward_time": 3.151806592941284,
      "step": 6634
    },
    {
      "epoch": 2.530670166015625e-05,
      "step": 6634,
      "training_step_time": 10.334506273269653
    },
    {
      "epoch": 2.5310516357421874e-05,
      "model_forward_time": 3.1849684715270996,
      "step": 6635
    },
    {
      "dynamics_loss_avg": 0.05997241139411926,
      "epoch": 2.5310516357421874e-05,
      "step": 6635
    },
    {
      "action_loss_avg": 0.020834850380197167,
      "epoch": 2.5310516357421874e-05,
      "step": 6635
    },
    {
      "epoch": 2.5310516357421874e-05,
      "step": 6635,
      "training_step_time": 9.96836256980896
    },
    {
      "epoch": 2.5310516357421874e-05,
      "model_forward_time": 3.1448380947113037,
      "step": 6635
    },
    {
      "epoch": 2.5310516357421874e-05,
      "step": 6635,
      "training_step_time": 9.918022632598877
    },
    {
      "epoch": 2.5310516357421874e-05,
      "model_forward_time": 3.1444363594055176,
      "step": 6635
    },
    {
      "epoch": 2.5310516357421874e-05,
      "step": 6635,
      "training_step_time": 9.916095495223999
    },
    {
      "epoch": 2.5310516357421874e-05,
      "model_forward_time": 3.1463801860809326,
      "step": 6635
    },
    {
      "epoch": 2.5310516357421874e-05,
      "step": 6635,
      "training_step_time": 10.321718454360962
    },
    {
      "epoch": 2.53143310546875e-05,
      "model_forward_time": 3.185622453689575,
      "step": 6636
    },
    {
      "epoch": 2.53143310546875e-05,
      "step": 6636,
      "training_step_time": 9.962864637374878
    },
    {
      "epoch": 2.53143310546875e-05,
      "model_forward_time": 3.151411294937134,
      "step": 6636
    },
    {
      "epoch": 2.53143310546875e-05,
      "step": 6636,
      "training_step_time": 9.92920184135437
    },
    {
      "epoch": 2.53143310546875e-05,
      "model_forward_time": 3.147891044616699,
      "step": 6636
    },
    {
      "epoch": 2.53143310546875e-05,
      "step": 6636,
      "training_step_time": 9.920215845108032
    },
    {
      "epoch": 2.53143310546875e-05,
      "model_forward_time": 3.1436567306518555,
      "step": 6636
    },
    {
      "epoch": 2.53143310546875e-05,
      "step": 6636,
      "training_step_time": 10.037189245223999
    },
    {
      "epoch": 2.5318145751953123e-05,
      "model_forward_time": 3.4809889793395996,
      "step": 6637
    },
    {
      "epoch": 2.5318145751953123e-05,
      "step": 6637,
      "training_step_time": 10.255536556243896
    },
    {
      "epoch": 2.5318145751953123e-05,
      "model_forward_time": 3.142179250717163,
      "step": 6637
    },
    {
      "epoch": 2.5318145751953123e-05,
      "step": 6637,
      "training_step_time": 9.909568786621094
    },
    {
      "epoch": 2.5318145751953123e-05,
      "model_forward_time": 3.1477463245391846,
      "step": 6637
    },
    {
      "dynamics_loss_avg": 0.051937428489327434,
      "epoch": 2.5318145751953123e-05,
      "step": 6637
    },
    {
      "action_loss_avg": 0.021256498340517283,
      "epoch": 2.5318145751953123e-05,
      "step": 6637
    },
    {
      "epoch": 2.5318145751953123e-05,
      "step": 6637,
      "training_step_time": 9.922001123428345
    },
    {
      "epoch": 2.5318145751953123e-05,
      "model_forward_time": 3.1390621662139893,
      "step": 6637
    },
    {
      "epoch": 2.5318145751953123e-05,
      "step": 6637,
      "training_step_time": 10.29708194732666
    },
    {
      "epoch": 2.532196044921875e-05,
      "model_forward_time": 3.177654981613159,
      "step": 6638
    },
    {
      "epoch": 2.532196044921875e-05,
      "step": 6638,
      "training_step_time": 9.948432922363281
    },
    {
      "epoch": 2.532196044921875e-05,
      "model_forward_time": 3.4056622982025146,
      "step": 6638
    },
    {
      "epoch": 2.532196044921875e-05,
      "step": 6638,
      "training_step_time": 10.178376197814941
    },
    {
      "epoch": 2.532196044921875e-05,
      "model_forward_time": 3.148106098175049,
      "step": 6638
    },
    {
      "epoch": 2.532196044921875e-05,
      "step": 6638,
      "training_step_time": 9.920954465866089
    },
    {
      "epoch": 2.532196044921875e-05,
      "model_forward_time": 3.1427483558654785,
      "step": 6638
    },
    {
      "epoch": 2.532196044921875e-05,
      "step": 6638,
      "training_step_time": 10.278416156768799
    },
    {
      "epoch": 2.5325775146484376e-05,
      "model_forward_time": 3.185997724533081,
      "step": 6639
    },
    {
      "epoch": 2.5325775146484376e-05,
      "step": 6639,
      "training_step_time": 9.962764978408813
    },
    {
      "epoch": 2.5325775146484376e-05,
      "model_forward_time": 3.150804042816162,
      "step": 6639
    },
    {
      "epoch": 2.5325775146484376e-05,
      "step": 6639,
      "training_step_time": 9.929160356521606
    },
    {
      "epoch": 2.5325775146484376e-05,
      "model_forward_time": 3.483112096786499,
      "step": 6639
    },
    {
      "epoch": 2.5325775146484376e-05,
      "step": 6639,
      "training_step_time": 10.264277458190918
    },
    {
      "epoch": 2.5325775146484376e-05,
      "model_forward_time": 3.148510217666626,
      "step": 6639
    },
    {
      "epoch": 2.5325775146484376e-05,
      "step": 6639,
      "training_step_time": 10.36265254020691
    },
    {
      "epoch": 2.532958984375e-05,
      "grad_norm": 0.45720672607421875,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 6640
    },
    {
      "epoch": 2.532958984375e-05,
      "model_forward_time": 3.183422088623047,
      "step": 6640
    },
    {
      "dynamics_loss_avg": 0.06320047732442617,
      "epoch": 2.532958984375e-05,
      "step": 6640
    },
    {
      "action_loss_avg": 0.024300961382687093,
      "epoch": 2.532958984375e-05,
      "step": 6640
    },
    {
      "epoch": 2.532958984375e-05,
      "step": 6640,
      "training_step_time": 9.961727857589722
    },
    {
      "epoch": 2.532958984375e-05,
      "model_forward_time": 3.158092737197876,
      "step": 6640
    },
    {
      "epoch": 2.532958984375e-05,
      "step": 6640,
      "training_step_time": 9.930610179901123
    },
    {
      "epoch": 2.532958984375e-05,
      "model_forward_time": 3.1369402408599854,
      "step": 6640
    },
    {
      "epoch": 2.532958984375e-05,
      "step": 6640,
      "training_step_time": 9.902429342269897
    },
    {
      "epoch": 2.532958984375e-05,
      "model_forward_time": 3.3971645832061768,
      "step": 6640
    },
    {
      "epoch": 2.532958984375e-05,
      "step": 6640,
      "training_step_time": 10.311626195907593
    },
    {
      "epoch": 2.5333404541015625e-05,
      "model_forward_time": 3.1696078777313232,
      "step": 6641
    },
    {
      "epoch": 2.5333404541015625e-05,
      "step": 6641,
      "training_step_time": 9.945783615112305
    },
    {
      "epoch": 2.5333404541015625e-05,
      "model_forward_time": 3.1370255947113037,
      "step": 6641
    },
    {
      "epoch": 2.5333404541015625e-05,
      "step": 6641,
      "training_step_time": 9.904327630996704
    },
    {
      "epoch": 2.5333404541015625e-05,
      "model_forward_time": 3.135576009750366,
      "step": 6641
    },
    {
      "epoch": 2.5333404541015625e-05,
      "step": 6641,
      "training_step_time": 9.90510869026184
    },
    {
      "epoch": 2.5333404541015625e-05,
      "model_forward_time": 3.134636163711548,
      "step": 6641
    },
    {
      "epoch": 2.5333404541015625e-05,
      "step": 6641,
      "training_step_time": 10.03226089477539
    },
    {
      "epoch": 2.533721923828125e-05,
      "model_forward_time": 3.4787280559539795,
      "step": 6642
    },
    {
      "epoch": 2.533721923828125e-05,
      "step": 6642,
      "training_step_time": 10.248488664627075
    },
    {
      "epoch": 2.533721923828125e-05,
      "model_forward_time": 3.1519694328308105,
      "step": 6642
    },
    {
      "epoch": 2.533721923828125e-05,
      "step": 6642,
      "training_step_time": 9.927756309509277
    },
    {
      "epoch": 2.533721923828125e-05,
      "model_forward_time": 3.1478848457336426,
      "step": 6642
    },
    {
      "dynamics_loss_avg": 0.058747283183038235,
      "epoch": 2.533721923828125e-05,
      "step": 6642
    },
    {
      "action_loss_avg": 0.026221308019012214,
      "epoch": 2.533721923828125e-05,
      "step": 6642
    },
    {
      "epoch": 2.533721923828125e-05,
      "step": 6642,
      "training_step_time": 9.92044472694397
    },
    {
      "epoch": 2.533721923828125e-05,
      "model_forward_time": 3.150972366333008,
      "step": 6642
    },
    {
      "epoch": 2.533721923828125e-05,
      "step": 6642,
      "training_step_time": 10.307788133621216
    },
    {
      "epoch": 2.5341033935546874e-05,
      "model_forward_time": 3.1953277587890625,
      "step": 6643
    },
    {
      "epoch": 2.5341033935546874e-05,
      "step": 6643,
      "training_step_time": 9.974506855010986
    },
    {
      "epoch": 2.5341033935546874e-05,
      "model_forward_time": 3.4412238597869873,
      "step": 6643
    },
    {
      "epoch": 2.5341033935546874e-05,
      "step": 6643,
      "training_step_time": 10.213405132293701
    },
    {
      "epoch": 2.5341033935546874e-05,
      "model_forward_time": 3.1419901847839355,
      "step": 6643
    },
    {
      "epoch": 2.5341033935546874e-05,
      "step": 6643,
      "training_step_time": 9.913681507110596
    },
    {
      "epoch": 2.5341033935546874e-05,
      "model_forward_time": 3.143278121948242,
      "step": 6643
    },
    {
      "epoch": 2.5341033935546874e-05,
      "step": 6643,
      "training_step_time": 10.231123924255371
    },
    {
      "epoch": 2.53448486328125e-05,
      "model_forward_time": 3.191862106323242,
      "step": 6644
    },
    {
      "epoch": 2.53448486328125e-05,
      "step": 6644,
      "training_step_time": 9.965080738067627
    },
    {
      "epoch": 2.53448486328125e-05,
      "model_forward_time": 3.15053653717041,
      "step": 6644
    },
    {
      "epoch": 2.53448486328125e-05,
      "step": 6644,
      "training_step_time": 9.927192687988281
    },
    {
      "epoch": 2.53448486328125e-05,
      "model_forward_time": 3.453725814819336,
      "step": 6644
    },
    {
      "epoch": 2.53448486328125e-05,
      "step": 6644,
      "training_step_time": 10.230987787246704
    },
    {
      "epoch": 2.53448486328125e-05,
      "model_forward_time": 3.1531662940979004,
      "step": 6644
    },
    {
      "epoch": 2.53448486328125e-05,
      "step": 6644,
      "training_step_time": 10.32481336593628
    },
    {
      "epoch": 2.5348663330078123e-05,
      "model_forward_time": 3.199709415435791,
      "step": 6645
    },
    {
      "dynamics_loss_avg": 0.03952408637851477,
      "epoch": 2.5348663330078123e-05,
      "step": 6645
    },
    {
      "action_loss_avg": 0.01791469338349998,
      "epoch": 2.5348663330078123e-05,
      "step": 6645
    },
    {
      "epoch": 2.5348663330078123e-05,
      "step": 6645,
      "training_step_time": 9.979907751083374
    },
    {
      "epoch": 2.5348663330078123e-05,
      "model_forward_time": 3.158245801925659,
      "step": 6645
    },
    {
      "epoch": 2.5348663330078123e-05,
      "step": 6645,
      "training_step_time": 9.938911199569702
    },
    {
      "epoch": 2.5348663330078123e-05,
      "model_forward_time": 3.1469004154205322,
      "step": 6645
    },
    {
      "epoch": 2.5348663330078123e-05,
      "step": 6645,
      "training_step_time": 9.933647394180298
    },
    {
      "epoch": 2.5348663330078123e-05,
      "model_forward_time": 3.4427430629730225,
      "step": 6645
    },
    {
      "epoch": 2.5348663330078123e-05,
      "step": 6645,
      "training_step_time": 10.319112539291382
    },
    {
      "epoch": 2.535247802734375e-05,
      "model_forward_time": 3.204375743865967,
      "step": 6646
    },
    {
      "epoch": 2.535247802734375e-05,
      "step": 6646,
      "training_step_time": 9.98661494255066
    },
    {
      "epoch": 2.535247802734375e-05,
      "model_forward_time": 3.147108793258667,
      "step": 6646
    },
    {
      "epoch": 2.535247802734375e-05,
      "step": 6646,
      "training_step_time": 9.923277854919434
    },
    {
      "epoch": 2.535247802734375e-05,
      "model_forward_time": 3.1487698554992676,
      "step": 6646
    },
    {
      "epoch": 2.535247802734375e-05,
      "step": 6646,
      "training_step_time": 9.926759481430054
    },
    {
      "epoch": 2.535247802734375e-05,
      "model_forward_time": 3.14339280128479,
      "step": 6646
    },
    {
      "epoch": 2.535247802734375e-05,
      "step": 6646,
      "training_step_time": 10.038797855377197
    },
    {
      "epoch": 2.5356292724609376e-05,
      "model_forward_time": 3.4411027431488037,
      "step": 6647
    },
    {
      "epoch": 2.5356292724609376e-05,
      "step": 6647,
      "training_step_time": 10.21843147277832
    },
    {
      "epoch": 2.5356292724609376e-05,
      "model_forward_time": 3.144495725631714,
      "step": 6647
    },
    {
      "epoch": 2.5356292724609376e-05,
      "step": 6647,
      "training_step_time": 9.930228233337402
    },
    {
      "epoch": 2.5356292724609376e-05,
      "model_forward_time": 2.4389724731445312,
      "step": 6647
    },
    {
      "dynamics_loss_avg": 0.0614375377073884,
      "epoch": 2.5356292724609376e-05,
      "step": 6647
    },
    {
      "action_loss_avg": 0.026102047273889185,
      "epoch": 2.5356292724609376e-05,
      "step": 6647
    },
    {
      "epoch": 2.5356292724609376e-05,
      "step": 6647,
      "training_step_time": 7.475137710571289
    },
    {
      "epoch": 2.5356292724609376e-05,
      "model_forward_time": 3.1384284496307373,
      "step": 6647
    },
    {
      "epoch": 2.5356292724609376e-05,
      "step": 6647,
      "training_step_time": 10.29413652420044
    },
    {
      "epoch": 2.5360107421875e-05,
      "model_forward_time": 3.1950483322143555,
      "step": 6648
    },
    {
      "epoch": 2.5360107421875e-05,
      "step": 6648,
      "training_step_time": 9.975797414779663
    },
    {
      "epoch": 2.5360107421875e-05,
      "model_forward_time": 3.142829418182373,
      "step": 6648
    },
    {
      "epoch": 2.5360107421875e-05,
      "step": 6648,
      "training_step_time": 10.207083463668823
    },
    {
      "epoch": 2.5360107421875e-05,
      "model_forward_time": 3.143026351928711,
      "step": 6648
    },
    {
      "epoch": 2.5360107421875e-05,
      "step": 6648,
      "training_step_time": 9.923858880996704
    },
    {
      "epoch": 2.5360107421875e-05,
      "model_forward_time": 3.146080255508423,
      "step": 6648
    },
    {
      "epoch": 2.5360107421875e-05,
      "step": 6648,
      "training_step_time": 10.33386754989624
    },
    {
      "epoch": 2.5363922119140625e-05,
      "model_forward_time": 3.1758387088775635,
      "step": 6649
    },
    {
      "epoch": 2.5363922119140625e-05,
      "step": 6649,
      "training_step_time": 9.96045207977295
    },
    {
      "epoch": 2.5363922119140625e-05,
      "model_forward_time": 3.145174026489258,
      "step": 6649
    },
    {
      "epoch": 2.5363922119140625e-05,
      "step": 6649,
      "training_step_time": 9.926156282424927
    },
    {
      "epoch": 2.5363922119140625e-05,
      "model_forward_time": 3.464491605758667,
      "step": 6649
    },
    {
      "epoch": 2.5363922119140625e-05,
      "step": 6649,
      "training_step_time": 10.245712518692017
    },
    {
      "epoch": 2.5363922119140625e-05,
      "model_forward_time": 3.139230966567993,
      "step": 6649
    },
    {
      "epoch": 2.5363922119140625e-05,
      "step": 6649,
      "training_step_time": 10.326589584350586
    },
    {
      "epoch": 2.536773681640625e-05,
      "grad_norm": 0.5548143982887268,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 6650
    },
    {
      "epoch": 2.536773681640625e-05,
      "model_forward_time": 3.1811060905456543,
      "step": 6650
    },
    {
      "dynamics_loss_avg": 0.05262570362538099,
      "epoch": 2.536773681640625e-05,
      "step": 6650
    },
    {
      "action_loss_avg": 0.023459074459969996,
      "epoch": 2.536773681640625e-05,
      "step": 6650
    },
    {
      "epoch": 2.536773681640625e-05,
      "step": 6650,
      "training_step_time": 9.962753772735596
    },
    {
      "epoch": 2.536773681640625e-05,
      "model_forward_time": 3.13505482673645,
      "step": 6650
    },
    {
      "epoch": 2.536773681640625e-05,
      "step": 6650,
      "training_step_time": 9.909673690795898
    },
    {
      "epoch": 2.536773681640625e-05,
      "model_forward_time": 3.134483575820923,
      "step": 6650
    },
    {
      "epoch": 2.536773681640625e-05,
      "step": 6650,
      "training_step_time": 9.909703254699707
    },
    {
      "epoch": 2.536773681640625e-05,
      "model_forward_time": 3.135363817214966,
      "step": 6650
    },
    {
      "epoch": 2.536773681640625e-05,
      "step": 6650,
      "training_step_time": 10.280133724212646
    },
    {
      "epoch": 2.5371551513671874e-05,
      "model_forward_time": 3.187636613845825,
      "step": 6651
    },
    {
      "epoch": 2.5371551513671874e-05,
      "step": 6651,
      "training_step_time": 9.962764501571655
    },
    {
      "epoch": 2.5371551513671874e-05,
      "model_forward_time": 3.135910749435425,
      "step": 6651
    },
    {
      "epoch": 2.5371551513671874e-05,
      "step": 6651,
      "training_step_time": 9.912830591201782
    },
    {
      "epoch": 2.5371551513671874e-05,
      "model_forward_time": 3.1425583362579346,
      "step": 6651
    },
    {
      "epoch": 2.5371551513671874e-05,
      "step": 6651,
      "training_step_time": 9.918876647949219
    },
    {
      "epoch": 2.5371551513671874e-05,
      "model_forward_time": 3.1383020877838135,
      "step": 6651
    },
    {
      "epoch": 2.5371551513671874e-05,
      "step": 6651,
      "training_step_time": 10.292819738388062
    },
    {
      "epoch": 2.53753662109375e-05,
      "model_forward_time": 3.436523914337158,
      "step": 6652
    },
    {
      "epoch": 2.53753662109375e-05,
      "step": 6652,
      "training_step_time": 10.217530488967896
    },
    {
      "epoch": 2.53753662109375e-05,
      "model_forward_time": 3.1377408504486084,
      "step": 6652
    },
    {
      "epoch": 2.53753662109375e-05,
      "step": 6652,
      "training_step_time": 9.920092582702637
    },
    {
      "epoch": 2.53753662109375e-05,
      "model_forward_time": 3.138461112976074,
      "step": 6652
    },
    {
      "dynamics_loss_avg": 0.05672796797007322,
      "epoch": 2.53753662109375e-05,
      "step": 6652
    },
    {
      "action_loss_avg": 0.018440541997551917,
      "epoch": 2.53753662109375e-05,
      "step": 6652
    },
    {
      "epoch": 2.53753662109375e-05,
      "step": 6652,
      "training_step_time": 9.920352458953857
    },
    {
      "epoch": 2.53753662109375e-05,
      "model_forward_time": 3.1395957469940186,
      "step": 6652
    },
    {
      "epoch": 2.53753662109375e-05,
      "step": 6652,
      "training_step_time": 10.284953355789185
    },
    {
      "epoch": 2.5379180908203123e-05,
      "model_forward_time": 3.1926562786102295,
      "step": 6653
    },
    {
      "epoch": 2.5379180908203123e-05,
      "step": 6653,
      "training_step_time": 9.97293734550476
    },
    {
      "epoch": 2.5379180908203123e-05,
      "model_forward_time": 3.423353672027588,
      "step": 6653
    },
    {
      "epoch": 2.5379180908203123e-05,
      "step": 6653,
      "training_step_time": 10.203887462615967
    },
    {
      "epoch": 2.5379180908203123e-05,
      "model_forward_time": 3.143263816833496,
      "step": 6653
    },
    {
      "epoch": 2.5379180908203123e-05,
      "step": 6653,
      "training_step_time": 9.927985906600952
    },
    {
      "epoch": 2.5379180908203123e-05,
      "model_forward_time": 3.148484945297241,
      "step": 6653
    },
    {
      "epoch": 2.5379180908203123e-05,
      "step": 6653,
      "training_step_time": 10.30264163017273
    },
    {
      "epoch": 2.538299560546875e-05,
      "model_forward_time": 3.154853343963623,
      "step": 6654
    },
    {
      "epoch": 2.538299560546875e-05,
      "step": 6654,
      "training_step_time": 9.937597036361694
    },
    {
      "epoch": 2.538299560546875e-05,
      "model_forward_time": 2.436344623565674,
      "step": 6654
    },
    {
      "epoch": 2.538299560546875e-05,
      "step": 6654,
      "training_step_time": 7.471640586853027
    },
    {
      "epoch": 2.538299560546875e-05,
      "model_forward_time": 2.431903839111328,
      "step": 6654
    },
    {
      "epoch": 2.538299560546875e-05,
      "step": 6654,
      "training_step_time": 7.732307434082031
    },
    {
      "epoch": 2.538299560546875e-05,
      "model_forward_time": 3.1382858753204346,
      "step": 6654
    },
    {
      "epoch": 2.538299560546875e-05,
      "step": 6654,
      "training_step_time": 10.305303573608398
    },
    {
      "epoch": 2.5386810302734376e-05,
      "model_forward_time": 3.1945950984954834,
      "step": 6655
    },
    {
      "dynamics_loss_avg": 0.0713074654340744,
      "epoch": 2.5386810302734376e-05,
      "step": 6655
    },
    {
      "action_loss_avg": 0.023578627593815328,
      "epoch": 2.5386810302734376e-05,
      "step": 6655
    },
    {
      "epoch": 2.5386810302734376e-05,
      "step": 6655,
      "training_step_time": 9.978143692016602
    },
    {
      "epoch": 2.5386810302734376e-05,
      "model_forward_time": 3.1388297080993652,
      "step": 6655
    },
    {
      "epoch": 2.5386810302734376e-05,
      "step": 6655,
      "training_step_time": 9.916709184646606
    },
    {
      "epoch": 2.5386810302734376e-05,
      "model_forward_time": 3.141810655593872,
      "step": 6655
    },
    {
      "epoch": 2.5386810302734376e-05,
      "step": 6655,
      "training_step_time": 9.917862892150879
    },
    {
      "epoch": 2.5386810302734376e-05,
      "model_forward_time": 3.4503679275512695,
      "step": 6655
    },
    {
      "epoch": 2.5386810302734376e-05,
      "step": 6655,
      "training_step_time": 10.320953607559204
    },
    {
      "epoch": 2.5390625e-05,
      "model_forward_time": 3.1706864833831787,
      "step": 6656
    },
    {
      "epoch": 2.5390625e-05,
      "step": 6656,
      "training_step_time": 9.945042371749878
    },
    {
      "epoch": 2.5390625e-05,
      "model_forward_time": 3.1346607208251953,
      "step": 6656
    },
    {
      "epoch": 2.5390625e-05,
      "step": 6656,
      "training_step_time": 9.905200719833374
    },
    {
      "epoch": 2.5390625e-05,
      "model_forward_time": 3.1360254287719727,
      "step": 6656
    },
    {
      "epoch": 2.5390625e-05,
      "step": 6656,
      "training_step_time": 9.908876180648804
    },
    {
      "epoch": 2.5390625e-05,
      "model_forward_time": 3.1344332695007324,
      "step": 6656
    },
    {
      "epoch": 2.5390625e-05,
      "step": 6656,
      "training_step_time": 10.251976728439331
    },
    {
      "epoch": 2.5394439697265625e-05,
      "model_forward_time": 3.1812562942504883,
      "step": 6657
    },
    {
      "epoch": 2.5394439697265625e-05,
      "step": 6657,
      "training_step_time": 10.24974536895752
    },
    {
      "epoch": 2.5394439697265625e-05,
      "model_forward_time": 3.1404879093170166,
      "step": 6657
    },
    {
      "epoch": 2.5394439697265625e-05,
      "step": 6657,
      "training_step_time": 9.90486192703247
    },
    {
      "epoch": 2.5394439697265625e-05,
      "model_forward_time": 3.139331340789795,
      "step": 6657
    },
    {
      "dynamics_loss_avg": 0.08886347971856594,
      "epoch": 2.5394439697265625e-05,
      "step": 6657
    },
    {
      "action_loss_avg": 0.04732755739241838,
      "epoch": 2.5394439697265625e-05,
      "step": 6657
    },
    {
      "epoch": 2.5394439697265625e-05,
      "step": 6657,
      "training_step_time": 9.904244184494019
    },
    {
      "epoch": 2.5394439697265625e-05,
      "model_forward_time": 3.1388254165649414,
      "step": 6657
    },
    {
      "epoch": 2.5394439697265625e-05,
      "step": 6657,
      "training_step_time": 10.284698963165283
    },
    {
      "epoch": 2.539825439453125e-05,
      "model_forward_time": 3.1920833587646484,
      "step": 6658
    },
    {
      "epoch": 2.539825439453125e-05,
      "step": 6658,
      "training_step_time": 9.961991786956787
    },
    {
      "epoch": 2.539825439453125e-05,
      "model_forward_time": 3.4023048877716064,
      "step": 6658
    },
    {
      "epoch": 2.539825439453125e-05,
      "step": 6658,
      "training_step_time": 10.168577671051025
    },
    {
      "epoch": 2.539825439453125e-05,
      "model_forward_time": 3.1496100425720215,
      "step": 6658
    },
    {
      "epoch": 2.539825439453125e-05,
      "step": 6658,
      "training_step_time": 9.922929525375366
    },
    {
      "epoch": 2.539825439453125e-05,
      "model_forward_time": 3.1430702209472656,
      "step": 6658
    },
    {
      "epoch": 2.539825439453125e-05,
      "step": 6658,
      "training_step_time": 10.294185161590576
    },
    {
      "epoch": 2.5402069091796874e-05,
      "model_forward_time": 3.1678879261016846,
      "step": 6659
    },
    {
      "epoch": 2.5402069091796874e-05,
      "step": 6659,
      "training_step_time": 9.934379577636719
    },
    {
      "epoch": 2.5402069091796874e-05,
      "model_forward_time": 3.145695447921753,
      "step": 6659
    },
    {
      "epoch": 2.5402069091796874e-05,
      "step": 6659,
      "training_step_time": 9.907146215438843
    },
    {
      "epoch": 2.5402069091796874e-05,
      "model_forward_time": 3.138507127761841,
      "step": 6659
    },
    {
      "epoch": 2.5402069091796874e-05,
      "step": 6659,
      "training_step_time": 10.172411918640137
    },
    {
      "epoch": 2.5402069091796874e-05,
      "model_forward_time": 3.1410470008850098,
      "step": 6659
    },
    {
      "epoch": 2.5402069091796874e-05,
      "step": 6659,
      "training_step_time": 10.019391059875488
    },
    {
      "epoch": 2.54058837890625e-05,
      "grad_norm": 0.40710991621017456,
      "learning_rate": 1e-05,
      "loss": 0.0843,
      "step": 6660
    },
    {
      "epoch": 2.54058837890625e-05,
      "model_forward_time": 3.167348623275757,
      "step": 6660
    },
    {
      "dynamics_loss_avg": 0.053122196719050406,
      "epoch": 2.54058837890625e-05,
      "step": 6660
    },
    {
      "action_loss_avg": 0.019563385704532265,
      "epoch": 2.54058837890625e-05,
      "step": 6660
    },
    {
      "epoch": 2.54058837890625e-05,
      "step": 6660,
      "training_step_time": 9.940389156341553
    },
    {
      "epoch": 2.54058837890625e-05,
      "model_forward_time": 3.1416335105895996,
      "step": 6660
    },
    {
      "epoch": 2.54058837890625e-05,
      "step": 6660,
      "training_step_time": 9.907965898513794
    },
    {
      "epoch": 2.54058837890625e-05,
      "model_forward_time": 3.139636516571045,
      "step": 6660
    },
    {
      "epoch": 2.54058837890625e-05,
      "step": 6660,
      "training_step_time": 9.913550615310669
    },
    {
      "epoch": 2.54058837890625e-05,
      "model_forward_time": 3.4095287322998047,
      "step": 6660
    },
    {
      "epoch": 2.54058837890625e-05,
      "step": 6660,
      "training_step_time": 10.276946783065796
    },
    {
      "epoch": 2.5409698486328123e-05,
      "model_forward_time": 3.1844277381896973,
      "step": 6661
    },
    {
      "epoch": 2.5409698486328123e-05,
      "step": 6661,
      "training_step_time": 9.960426568984985
    },
    {
      "epoch": 2.5409698486328123e-05,
      "model_forward_time": 3.1435980796813965,
      "step": 6661
    },
    {
      "epoch": 2.5409698486328123e-05,
      "step": 6661,
      "training_step_time": 9.917797565460205
    },
    {
      "epoch": 2.5409698486328123e-05,
      "model_forward_time": 3.1375303268432617,
      "step": 6661
    },
    {
      "epoch": 2.5409698486328123e-05,
      "step": 6661,
      "training_step_time": 9.91014575958252
    },
    {
      "epoch": 2.5409698486328123e-05,
      "model_forward_time": 3.1394808292388916,
      "step": 6661
    },
    {
      "epoch": 2.5409698486328123e-05,
      "step": 6661,
      "training_step_time": 10.022592544555664
    },
    {
      "epoch": 2.541351318359375e-05,
      "model_forward_time": 3.1667518615722656,
      "step": 6662
    },
    {
      "epoch": 2.541351318359375e-05,
      "step": 6662,
      "training_step_time": 10.195876836776733
    },
    {
      "epoch": 2.541351318359375e-05,
      "model_forward_time": 3.1446025371551514,
      "step": 6662
    },
    {
      "epoch": 2.541351318359375e-05,
      "step": 6662,
      "training_step_time": 9.914518356323242
    },
    {
      "epoch": 2.541351318359375e-05,
      "model_forward_time": 3.1434993743896484,
      "step": 6662
    },
    {
      "dynamics_loss_avg": 0.06927580982446671,
      "epoch": 2.541351318359375e-05,
      "step": 6662
    },
    {
      "action_loss_avg": 0.023949641548097134,
      "epoch": 2.541351318359375e-05,
      "step": 6662
    },
    {
      "epoch": 2.541351318359375e-05,
      "step": 6662,
      "training_step_time": 9.918590307235718
    },
    {
      "epoch": 2.541351318359375e-05,
      "model_forward_time": 3.1459527015686035,
      "step": 6662
    },
    {
      "epoch": 2.541351318359375e-05,
      "step": 6662,
      "training_step_time": 10.29648494720459
    },
    {
      "epoch": 2.5417327880859376e-05,
      "model_forward_time": 2.46732759475708,
      "step": 6663
    },
    {
      "epoch": 2.5417327880859376e-05,
      "step": 6663,
      "training_step_time": 7.493589878082275
    },
    {
      "epoch": 2.5417327880859376e-05,
      "model_forward_time": 3.418937921524048,
      "step": 6663
    },
    {
      "epoch": 2.5417327880859376e-05,
      "step": 6663,
      "training_step_time": 10.196624517440796
    },
    {
      "epoch": 2.5417327880859376e-05,
      "model_forward_time": 3.143080472946167,
      "step": 6663
    },
    {
      "epoch": 2.5417327880859376e-05,
      "step": 6663,
      "training_step_time": 9.916098833084106
    },
    {
      "epoch": 2.5417327880859376e-05,
      "model_forward_time": 3.1461594104766846,
      "step": 6663
    },
    {
      "epoch": 2.5417327880859376e-05,
      "step": 6663,
      "training_step_time": 10.26859164237976
    },
    {
      "epoch": 2.5421142578125e-05,
      "model_forward_time": 3.201408624649048,
      "step": 6664
    },
    {
      "epoch": 2.5421142578125e-05,
      "step": 6664,
      "training_step_time": 9.976367235183716
    },
    {
      "epoch": 2.5421142578125e-05,
      "model_forward_time": 3.1401920318603516,
      "step": 6664
    },
    {
      "epoch": 2.5421142578125e-05,
      "step": 6664,
      "training_step_time": 9.90625786781311
    },
    {
      "epoch": 2.5421142578125e-05,
      "model_forward_time": 3.4347734451293945,
      "step": 6664
    },
    {
      "epoch": 2.5421142578125e-05,
      "step": 6664,
      "training_step_time": 10.20102572441101
    },
    {
      "epoch": 2.5421142578125e-05,
      "model_forward_time": 3.137986660003662,
      "step": 6664
    },
    {
      "epoch": 2.5421142578125e-05,
      "step": 6664,
      "training_step_time": 10.322607517242432
    },
    {
      "epoch": 2.5424957275390625e-05,
      "model_forward_time": 3.174690008163452,
      "step": 6665
    },
    {
      "dynamics_loss_avg": 0.0749487653374672,
      "epoch": 2.5424957275390625e-05,
      "step": 6665
    },
    {
      "action_loss_avg": 0.02741734217852354,
      "epoch": 2.5424957275390625e-05,
      "step": 6665
    },
    {
      "epoch": 2.5424957275390625e-05,
      "step": 6665,
      "training_step_time": 9.943303346633911
    },
    {
      "epoch": 2.5424957275390625e-05,
      "model_forward_time": 3.1368093490600586,
      "step": 6665
    },
    {
      "epoch": 2.5424957275390625e-05,
      "step": 6665,
      "training_step_time": 9.903308868408203
    },
    {
      "epoch": 2.5424957275390625e-05,
      "model_forward_time": 3.154123544692993,
      "step": 6665
    },
    {
      "epoch": 2.5424957275390625e-05,
      "step": 6665,
      "training_step_time": 9.929466485977173
    },
    {
      "epoch": 2.5424957275390625e-05,
      "model_forward_time": 3.4326446056365967,
      "step": 6665
    },
    {
      "epoch": 2.5424957275390625e-05,
      "step": 6665,
      "training_step_time": 10.309305667877197
    },
    {
      "epoch": 2.542877197265625e-05,
      "model_forward_time": 3.185720443725586,
      "step": 6666
    },
    {
      "epoch": 2.542877197265625e-05,
      "step": 6666,
      "training_step_time": 9.962532043457031
    },
    {
      "epoch": 2.542877197265625e-05,
      "model_forward_time": 3.1405956745147705,
      "step": 6666
    },
    {
      "epoch": 2.542877197265625e-05,
      "step": 6666,
      "training_step_time": 9.910443305969238
    },
    {
      "epoch": 2.542877197265625e-05,
      "model_forward_time": 3.1439743041992188,
      "step": 6666
    },
    {
      "epoch": 2.542877197265625e-05,
      "step": 6666,
      "training_step_time": 9.912371158599854
    },
    {
      "epoch": 2.542877197265625e-05,
      "model_forward_time": 3.1279120445251465,
      "step": 6666
    },
    {
      "epoch": 2.542877197265625e-05,
      "step": 6666,
      "training_step_time": 10.276126146316528
    },
    {
      "epoch": 2.5432586669921874e-05,
      "model_forward_time": 3.4424808025360107,
      "step": 6667
    },
    {
      "epoch": 2.5432586669921874e-05,
      "step": 6667,
      "training_step_time": 10.209999322891235
    },
    {
      "epoch": 2.5432586669921874e-05,
      "model_forward_time": 3.1341938972473145,
      "step": 6667
    },
    {
      "epoch": 2.5432586669921874e-05,
      "step": 6667,
      "training_step_time": 9.904770851135254
    },
    {
      "epoch": 2.5432586669921874e-05,
      "model_forward_time": 3.140024423599243,
      "step": 6667
    },
    {
      "dynamics_loss_avg": 0.06326657645404339,
      "epoch": 2.5432586669921874e-05,
      "step": 6667
    },
    {
      "action_loss_avg": 0.030122152669355272,
      "epoch": 2.5432586669921874e-05,
      "step": 6667
    },
    {
      "epoch": 2.5432586669921874e-05,
      "step": 6667,
      "training_step_time": 9.913194179534912
    },
    {
      "epoch": 2.5432586669921874e-05,
      "model_forward_time": 3.135894536972046,
      "step": 6667
    },
    {
      "epoch": 2.5432586669921874e-05,
      "step": 6667,
      "training_step_time": 10.027989149093628
    },
    {
      "epoch": 2.54364013671875e-05,
      "model_forward_time": 3.1832892894744873,
      "step": 6668
    },
    {
      "epoch": 2.54364013671875e-05,
      "step": 6668,
      "training_step_time": 9.958149194717407
    },
    {
      "epoch": 2.54364013671875e-05,
      "model_forward_time": 3.390375852584839,
      "step": 6668
    },
    {
      "epoch": 2.54364013671875e-05,
      "step": 6668,
      "training_step_time": 10.164734363555908
    },
    {
      "epoch": 2.54364013671875e-05,
      "model_forward_time": 3.13802170753479,
      "step": 6668
    },
    {
      "epoch": 2.54364013671875e-05,
      "step": 6668,
      "training_step_time": 9.908405065536499
    },
    {
      "epoch": 2.54364013671875e-05,
      "model_forward_time": 3.1369435787200928,
      "step": 6668
    },
    {
      "epoch": 2.54364013671875e-05,
      "step": 6668,
      "training_step_time": 10.317393779754639
    },
    {
      "epoch": 2.5440216064453123e-05,
      "model_forward_time": 3.1833341121673584,
      "step": 6669
    },
    {
      "epoch": 2.5440216064453123e-05,
      "step": 6669,
      "training_step_time": 9.956974506378174
    },
    {
      "epoch": 2.5440216064453123e-05,
      "model_forward_time": 3.142803192138672,
      "step": 6669
    },
    {
      "epoch": 2.5440216064453123e-05,
      "step": 6669,
      "training_step_time": 9.9160315990448
    },
    {
      "epoch": 2.5440216064453123e-05,
      "model_forward_time": 3.412247657775879,
      "step": 6669
    },
    {
      "epoch": 2.5440216064453123e-05,
      "step": 6669,
      "training_step_time": 10.180082321166992
    },
    {
      "epoch": 2.5440216064453123e-05,
      "model_forward_time": 3.1376543045043945,
      "step": 6669
    },
    {
      "epoch": 2.5440216064453123e-05,
      "step": 6669,
      "training_step_time": 10.296382427215576
    },
    {
      "epoch": 2.544403076171875e-05,
      "grad_norm": 0.3875315189361572,
      "learning_rate": 1e-05,
      "loss": 0.0877,
      "step": 6670
    },
    {
      "epoch": 2.544403076171875e-05,
      "model_forward_time": 3.1591086387634277,
      "step": 6670
    },
    {
      "dynamics_loss_avg": 0.0801035825163126,
      "epoch": 2.544403076171875e-05,
      "step": 6670
    },
    {
      "action_loss_avg": 0.03214147351682186,
      "epoch": 2.544403076171875e-05,
      "step": 6670
    },
    {
      "epoch": 2.544403076171875e-05,
      "step": 6670,
      "training_step_time": 9.933451652526855
    },
    {
      "epoch": 2.544403076171875e-05,
      "model_forward_time": 3.138657808303833,
      "step": 6670
    },
    {
      "epoch": 2.544403076171875e-05,
      "step": 6670,
      "training_step_time": 9.907289266586304
    },
    {
      "epoch": 2.544403076171875e-05,
      "model_forward_time": 3.1374008655548096,
      "step": 6670
    },
    {
      "epoch": 2.544403076171875e-05,
      "step": 6670,
      "training_step_time": 9.907017707824707
    },
    {
      "epoch": 2.544403076171875e-05,
      "model_forward_time": 3.3929741382598877,
      "step": 6670
    },
    {
      "epoch": 2.544403076171875e-05,
      "step": 6670,
      "training_step_time": 10.30910587310791
    },
    {
      "epoch": 2.5447845458984376e-05,
      "model_forward_time": 3.171363353729248,
      "step": 6671
    },
    {
      "epoch": 2.5447845458984376e-05,
      "step": 6671,
      "training_step_time": 9.94802975654602
    },
    {
      "epoch": 2.5447845458984376e-05,
      "model_forward_time": 3.1404590606689453,
      "step": 6671
    },
    {
      "epoch": 2.5447845458984376e-05,
      "step": 6671,
      "training_step_time": 9.912203550338745
    },
    {
      "epoch": 2.5447845458984376e-05,
      "model_forward_time": 3.1364312171936035,
      "step": 6671
    },
    {
      "epoch": 2.5447845458984376e-05,
      "step": 6671,
      "training_step_time": 9.908855199813843
    },
    {
      "epoch": 2.5447845458984376e-05,
      "model_forward_time": 3.1419923305511475,
      "step": 6671
    },
    {
      "epoch": 2.5447845458984376e-05,
      "step": 6671,
      "training_step_time": 10.304266929626465
    },
    {
      "epoch": 2.545166015625e-05,
      "model_forward_time": 3.435950994491577,
      "step": 6672
    },
    {
      "epoch": 2.545166015625e-05,
      "step": 6672,
      "training_step_time": 10.208830833435059
    },
    {
      "epoch": 2.545166015625e-05,
      "model_forward_time": 3.138075828552246,
      "step": 6672
    },
    {
      "epoch": 2.545166015625e-05,
      "step": 6672,
      "training_step_time": 9.911116600036621
    },
    {
      "epoch": 2.545166015625e-05,
      "model_forward_time": 3.1422410011291504,
      "step": 6672
    },
    {
      "dynamics_loss_avg": 0.058413917943835256,
      "epoch": 2.545166015625e-05,
      "step": 6672
    },
    {
      "action_loss_avg": 0.022475119098089635,
      "epoch": 2.545166015625e-05,
      "step": 6672
    },
    {
      "epoch": 2.545166015625e-05,
      "step": 6672,
      "training_step_time": 9.920293807983398
    },
    {
      "epoch": 2.545166015625e-05,
      "model_forward_time": 3.1412270069122314,
      "step": 6672
    },
    {
      "epoch": 2.545166015625e-05,
      "step": 6672,
      "training_step_time": 10.294359683990479
    },
    {
      "epoch": 2.5455474853515625e-05,
      "model_forward_time": 3.148286819458008,
      "step": 6673
    },
    {
      "epoch": 2.5455474853515625e-05,
      "step": 6673,
      "training_step_time": 9.921066761016846
    },
    {
      "epoch": 2.5455474853515625e-05,
      "model_forward_time": 3.434041738510132,
      "step": 6673
    },
    {
      "epoch": 2.5455474853515625e-05,
      "step": 6673,
      "training_step_time": 10.20359492301941
    },
    {
      "epoch": 2.5455474853515625e-05,
      "model_forward_time": 3.1352286338806152,
      "step": 6673
    },
    {
      "epoch": 2.5455474853515625e-05,
      "step": 6673,
      "training_step_time": 9.902793407440186
    },
    {
      "epoch": 2.5455474853515625e-05,
      "model_forward_time": 2.432018756866455,
      "step": 6673
    },
    {
      "epoch": 2.5455474853515625e-05,
      "step": 6673,
      "training_step_time": 10.290656089782715
    },
    {
      "epoch": 2.545928955078125e-05,
      "model_forward_time": 3.15380597114563,
      "step": 6674
    },
    {
      "epoch": 2.545928955078125e-05,
      "step": 6674,
      "training_step_time": 9.923830270767212
    },
    {
      "epoch": 2.545928955078125e-05,
      "model_forward_time": 3.129246950149536,
      "step": 6674
    },
    {
      "epoch": 2.545928955078125e-05,
      "step": 6674,
      "training_step_time": 9.894197940826416
    },
    {
      "epoch": 2.545928955078125e-05,
      "model_forward_time": 2.43200945854187,
      "step": 6674
    },
    {
      "epoch": 2.545928955078125e-05,
      "step": 6674,
      "training_step_time": 7.723647117614746
    },
    {
      "epoch": 2.545928955078125e-05,
      "model_forward_time": 3.1331992149353027,
      "step": 6674
    },
    {
      "epoch": 2.545928955078125e-05,
      "step": 6674,
      "training_step_time": 10.262297868728638
    },
    {
      "epoch": 2.5463104248046874e-05,
      "model_forward_time": 3.1996374130249023,
      "step": 6675
    },
    {
      "dynamics_loss_avg": 0.04953531324863434,
      "epoch": 2.5463104248046874e-05,
      "step": 6675
    },
    {
      "action_loss_avg": 0.01644375827163458,
      "epoch": 2.5463104248046874e-05,
      "step": 6675
    },
    {
      "epoch": 2.5463104248046874e-05,
      "step": 6675,
      "training_step_time": 9.968218564987183
    },
    {
      "epoch": 2.5463104248046874e-05,
      "model_forward_time": 3.142224073410034,
      "step": 6675
    },
    {
      "epoch": 2.5463104248046874e-05,
      "step": 6675,
      "training_step_time": 9.923420429229736
    },
    {
      "epoch": 2.5463104248046874e-05,
      "model_forward_time": 3.1437900066375732,
      "step": 6675
    },
    {
      "epoch": 2.5463104248046874e-05,
      "step": 6675,
      "training_step_time": 9.918750762939453
    },
    {
      "epoch": 2.5463104248046874e-05,
      "model_forward_time": 3.43095326423645,
      "step": 6675
    },
    {
      "epoch": 2.5463104248046874e-05,
      "step": 6675,
      "training_step_time": 10.319579124450684
    },
    {
      "epoch": 2.54669189453125e-05,
      "model_forward_time": 3.187107563018799,
      "step": 6676
    },
    {
      "epoch": 2.54669189453125e-05,
      "step": 6676,
      "training_step_time": 9.980490446090698
    },
    {
      "epoch": 2.54669189453125e-05,
      "model_forward_time": 3.142995595932007,
      "step": 6676
    },
    {
      "epoch": 2.54669189453125e-05,
      "step": 6676,
      "training_step_time": 9.921587228775024
    },
    {
      "epoch": 2.54669189453125e-05,
      "model_forward_time": 3.147975206375122,
      "step": 6676
    },
    {
      "epoch": 2.54669189453125e-05,
      "step": 6676,
      "training_step_time": 9.93071961402893
    },
    {
      "epoch": 2.54669189453125e-05,
      "model_forward_time": 3.1487207412719727,
      "step": 6676
    },
    {
      "epoch": 2.54669189453125e-05,
      "step": 6676,
      "training_step_time": 10.028823137283325
    },
    {
      "epoch": 2.5470733642578123e-05,
      "model_forward_time": 3.1891634464263916,
      "step": 6677
    },
    {
      "epoch": 2.5470733642578123e-05,
      "step": 6677,
      "training_step_time": 10.296099424362183
    },
    {
      "epoch": 2.5470733642578123e-05,
      "model_forward_time": 3.14129376411438,
      "step": 6677
    },
    {
      "epoch": 2.5470733642578123e-05,
      "step": 6677,
      "training_step_time": 9.91947078704834
    },
    {
      "epoch": 2.5470733642578123e-05,
      "model_forward_time": 3.1515398025512695,
      "step": 6677
    },
    {
      "dynamics_loss_avg": 0.05375222489237785,
      "epoch": 2.5470733642578123e-05,
      "step": 6677
    },
    {
      "action_loss_avg": 0.025380121171474458,
      "epoch": 2.5470733642578123e-05,
      "step": 6677
    },
    {
      "epoch": 2.5470733642578123e-05,
      "step": 6677,
      "training_step_time": 9.937573432922363
    },
    {
      "epoch": 2.5470733642578123e-05,
      "model_forward_time": 3.140498638153076,
      "step": 6677
    },
    {
      "epoch": 2.5470733642578123e-05,
      "step": 6677,
      "training_step_time": 10.28910779953003
    },
    {
      "epoch": 2.547454833984375e-05,
      "model_forward_time": 3.1770222187042236,
      "step": 6678
    },
    {
      "epoch": 2.547454833984375e-05,
      "step": 6678,
      "training_step_time": 9.954226970672607
    },
    {
      "epoch": 2.547454833984375e-05,
      "model_forward_time": 3.4142398834228516,
      "step": 6678
    },
    {
      "epoch": 2.547454833984375e-05,
      "step": 6678,
      "training_step_time": 10.187822103500366
    },
    {
      "epoch": 2.547454833984375e-05,
      "model_forward_time": 3.148691415786743,
      "step": 6678
    },
    {
      "epoch": 2.547454833984375e-05,
      "step": 6678,
      "training_step_time": 9.920209407806396
    },
    {
      "epoch": 2.547454833984375e-05,
      "model_forward_time": 3.1383707523345947,
      "step": 6678
    },
    {
      "epoch": 2.547454833984375e-05,
      "step": 6678,
      "training_step_time": 10.013702154159546
    },
    {
      "epoch": 2.5478363037109376e-05,
      "model_forward_time": 3.179832935333252,
      "step": 6679
    },
    {
      "epoch": 2.5478363037109376e-05,
      "step": 6679,
      "training_step_time": 9.955240726470947
    },
    {
      "epoch": 2.5478363037109376e-05,
      "model_forward_time": 3.140307664871216,
      "step": 6679
    },
    {
      "epoch": 2.5478363037109376e-05,
      "step": 6679,
      "training_step_time": 9.91302227973938
    },
    {
      "epoch": 2.5478363037109376e-05,
      "model_forward_time": 3.152435302734375,
      "step": 6679
    },
    {
      "epoch": 2.5478363037109376e-05,
      "step": 6679,
      "training_step_time": 10.246099472045898
    },
    {
      "epoch": 2.5478363037109376e-05,
      "model_forward_time": 3.148463010787964,
      "step": 6679
    },
    {
      "epoch": 2.5478363037109376e-05,
      "step": 6679,
      "training_step_time": 10.292607307434082
    },
    {
      "epoch": 2.5482177734375e-05,
      "grad_norm": 0.5384833812713623,
      "learning_rate": 1e-05,
      "loss": 0.0833,
      "step": 6680
    },
    {
      "epoch": 2.5482177734375e-05,
      "model_forward_time": 3.1898550987243652,
      "step": 6680
    },
    {
      "dynamics_loss_avg": 0.06214622799307108,
      "epoch": 2.5482177734375e-05,
      "step": 6680
    },
    {
      "action_loss_avg": 0.0271233219653368,
      "epoch": 2.5482177734375e-05,
      "step": 6680
    },
    {
      "epoch": 2.5482177734375e-05,
      "step": 6680,
      "training_step_time": 9.970756769180298
    },
    {
      "epoch": 2.5482177734375e-05,
      "model_forward_time": 3.141463041305542,
      "step": 6680
    },
    {
      "epoch": 2.5482177734375e-05,
      "step": 6680,
      "training_step_time": 9.917395830154419
    },
    {
      "epoch": 2.5482177734375e-05,
      "model_forward_time": 3.1432623863220215,
      "step": 6680
    },
    {
      "epoch": 2.5482177734375e-05,
      "step": 6680,
      "training_step_time": 9.922872304916382
    },
    {
      "epoch": 2.5482177734375e-05,
      "model_forward_time": 3.451557159423828,
      "step": 6680
    },
    {
      "epoch": 2.5482177734375e-05,
      "step": 6680,
      "training_step_time": 10.324406862258911
    },
    {
      "epoch": 2.5485992431640625e-05,
      "model_forward_time": 3.1902968883514404,
      "step": 6681
    },
    {
      "epoch": 2.5485992431640625e-05,
      "step": 6681,
      "training_step_time": 9.96847128868103
    },
    {
      "epoch": 2.5485992431640625e-05,
      "model_forward_time": 3.14791202545166,
      "step": 6681
    },
    {
      "epoch": 2.5485992431640625e-05,
      "step": 6681,
      "training_step_time": 9.926156044006348
    },
    {
      "epoch": 2.5485992431640625e-05,
      "model_forward_time": 3.1471502780914307,
      "step": 6681
    },
    {
      "epoch": 2.5485992431640625e-05,
      "step": 6681,
      "training_step_time": 9.92235517501831
    },
    {
      "epoch": 2.5485992431640625e-05,
      "model_forward_time": 3.1483676433563232,
      "step": 6681
    },
    {
      "epoch": 2.5485992431640625e-05,
      "step": 6681,
      "training_step_time": 10.281810522079468
    },
    {
      "epoch": 2.548980712890625e-05,
      "model_forward_time": 3.1717591285705566,
      "step": 6682
    },
    {
      "epoch": 2.548980712890625e-05,
      "step": 6682,
      "training_step_time": 10.21751070022583
    },
    {
      "epoch": 2.548980712890625e-05,
      "model_forward_time": 3.1463146209716797,
      "step": 6682
    },
    {
      "epoch": 2.548980712890625e-05,
      "step": 6682,
      "training_step_time": 9.922363996505737
    },
    {
      "epoch": 2.548980712890625e-05,
      "model_forward_time": 3.1388256549835205,
      "step": 6682
    },
    {
      "dynamics_loss_avg": 0.056064625084400174,
      "epoch": 2.548980712890625e-05,
      "step": 6682
    },
    {
      "action_loss_avg": 0.023185070883482695,
      "epoch": 2.548980712890625e-05,
      "step": 6682
    },
    {
      "epoch": 2.548980712890625e-05,
      "step": 6682,
      "training_step_time": 9.915982484817505
    },
    {
      "epoch": 2.548980712890625e-05,
      "model_forward_time": 3.1403095722198486,
      "step": 6682
    },
    {
      "epoch": 2.548980712890625e-05,
      "step": 6682,
      "training_step_time": 10.012877225875854
    },
    {
      "epoch": 2.5493621826171874e-05,
      "model_forward_time": 3.1949198246002197,
      "step": 6683
    },
    {
      "epoch": 2.5493621826171874e-05,
      "step": 6683,
      "training_step_time": 9.97896409034729
    },
    {
      "epoch": 2.5493621826171874e-05,
      "model_forward_time": 3.465956926345825,
      "step": 6683
    },
    {
      "epoch": 2.5493621826171874e-05,
      "step": 6683,
      "training_step_time": 10.239835262298584
    },
    {
      "epoch": 2.5493621826171874e-05,
      "model_forward_time": 3.1406681537628174,
      "step": 6683
    },
    {
      "epoch": 2.5493621826171874e-05,
      "step": 6683,
      "training_step_time": 9.914442539215088
    },
    {
      "epoch": 2.5493621826171874e-05,
      "model_forward_time": 3.151940107345581,
      "step": 6683
    },
    {
      "epoch": 2.5493621826171874e-05,
      "step": 6683,
      "training_step_time": 10.25601601600647
    },
    {
      "epoch": 2.54974365234375e-05,
      "model_forward_time": 3.172642707824707,
      "step": 6684
    },
    {
      "epoch": 2.54974365234375e-05,
      "step": 6684,
      "training_step_time": 9.956701278686523
    },
    {
      "epoch": 2.54974365234375e-05,
      "model_forward_time": 3.1431174278259277,
      "step": 6684
    },
    {
      "epoch": 2.54974365234375e-05,
      "step": 6684,
      "training_step_time": 9.921926975250244
    },
    {
      "epoch": 2.54974365234375e-05,
      "model_forward_time": 3.4481093883514404,
      "step": 6684
    },
    {
      "epoch": 2.54974365234375e-05,
      "step": 6684,
      "training_step_time": 10.225082397460938
    },
    {
      "epoch": 2.54974365234375e-05,
      "model_forward_time": 3.1465532779693604,
      "step": 6684
    },
    {
      "epoch": 2.54974365234375e-05,
      "step": 6684,
      "training_step_time": 10.027487516403198
    },
    {
      "epoch": 2.5501251220703124e-05,
      "model_forward_time": 3.1732683181762695,
      "step": 6685
    },
    {
      "dynamics_loss_avg": 0.059577585011720655,
      "epoch": 2.5501251220703124e-05,
      "step": 6685
    },
    {
      "action_loss_avg": 0.02149776150472462,
      "epoch": 2.5501251220703124e-05,
      "step": 6685
    },
    {
      "epoch": 2.5501251220703124e-05,
      "step": 6685,
      "training_step_time": 9.96678900718689
    },
    {
      "epoch": 2.5501251220703124e-05,
      "model_forward_time": 3.1468963623046875,
      "step": 6685
    },
    {
      "epoch": 2.5501251220703124e-05,
      "step": 6685,
      "training_step_time": 9.927936792373657
    },
    {
      "epoch": 2.5501251220703124e-05,
      "model_forward_time": 3.1481142044067383,
      "step": 6685
    },
    {
      "epoch": 2.5501251220703124e-05,
      "step": 6685,
      "training_step_time": 9.927446365356445
    },
    {
      "epoch": 2.5501251220703124e-05,
      "model_forward_time": 3.4346067905426025,
      "step": 6685
    },
    {
      "epoch": 2.5501251220703124e-05,
      "step": 6685,
      "training_step_time": 10.30714225769043
    },
    {
      "epoch": 2.550506591796875e-05,
      "model_forward_time": 3.2014987468719482,
      "step": 6686
    },
    {
      "epoch": 2.550506591796875e-05,
      "step": 6686,
      "training_step_time": 9.984630346298218
    },
    {
      "epoch": 2.550506591796875e-05,
      "model_forward_time": 3.1425445079803467,
      "step": 6686
    },
    {
      "epoch": 2.550506591796875e-05,
      "step": 6686,
      "training_step_time": 9.92001748085022
    },
    {
      "epoch": 2.550506591796875e-05,
      "model_forward_time": 3.139226198196411,
      "step": 6686
    },
    {
      "epoch": 2.550506591796875e-05,
      "step": 6686,
      "training_step_time": 9.91207242012024
    },
    {
      "epoch": 2.550506591796875e-05,
      "model_forward_time": 3.137728214263916,
      "step": 6686
    },
    {
      "epoch": 2.550506591796875e-05,
      "step": 6686,
      "training_step_time": 10.00973391532898
    },
    {
      "epoch": 2.5508880615234376e-05,
      "model_forward_time": 3.488790273666382,
      "step": 6687
    },
    {
      "epoch": 2.5508880615234376e-05,
      "step": 6687,
      "training_step_time": 10.265491247177124
    },
    {
      "epoch": 2.5508880615234376e-05,
      "model_forward_time": 3.139857769012451,
      "step": 6687
    },
    {
      "epoch": 2.5508880615234376e-05,
      "step": 6687,
      "training_step_time": 9.91060757637024
    },
    {
      "epoch": 2.5508880615234376e-05,
      "model_forward_time": 3.1455237865448,
      "step": 6687
    },
    {
      "dynamics_loss_avg": 0.047719151340425016,
      "epoch": 2.5508880615234376e-05,
      "step": 6687
    },
    {
      "action_loss_avg": 0.017390767578035593,
      "epoch": 2.5508880615234376e-05,
      "step": 6687
    },
    {
      "epoch": 2.5508880615234376e-05,
      "step": 6687,
      "training_step_time": 9.922630310058594
    },
    {
      "epoch": 2.5508880615234376e-05,
      "model_forward_time": 3.139608860015869,
      "step": 6687
    },
    {
      "epoch": 2.5508880615234376e-05,
      "step": 6687,
      "training_step_time": 10.013959884643555
    },
    {
      "epoch": 2.55126953125e-05,
      "model_forward_time": 3.171381711959839,
      "step": 6688
    },
    {
      "epoch": 2.55126953125e-05,
      "step": 6688,
      "training_step_time": 9.948710441589355
    },
    {
      "epoch": 2.55126953125e-05,
      "model_forward_time": 3.4553897380828857,
      "step": 6688
    },
    {
      "epoch": 2.55126953125e-05,
      "step": 6688,
      "training_step_time": 10.22701358795166
    },
    {
      "epoch": 2.55126953125e-05,
      "model_forward_time": 3.1486542224884033,
      "step": 6688
    },
    {
      "epoch": 2.55126953125e-05,
      "step": 6688,
      "training_step_time": 9.922446727752686
    },
    {
      "epoch": 2.55126953125e-05,
      "model_forward_time": 3.14558482170105,
      "step": 6688
    },
    {
      "epoch": 2.55126953125e-05,
      "step": 6688,
      "training_step_time": 10.027449369430542
    },
    {
      "epoch": 2.5516510009765625e-05,
      "model_forward_time": 3.184767961502075,
      "step": 6689
    },
    {
      "epoch": 2.5516510009765625e-05,
      "step": 6689,
      "training_step_time": 9.960654973983765
    },
    {
      "epoch": 2.5516510009765625e-05,
      "model_forward_time": 3.1367135047912598,
      "step": 6689
    },
    {
      "epoch": 2.5516510009765625e-05,
      "step": 6689,
      "training_step_time": 9.911686658859253
    },
    {
      "epoch": 2.5516510009765625e-05,
      "model_forward_time": 3.419694662094116,
      "step": 6689
    },
    {
      "epoch": 2.5516510009765625e-05,
      "step": 6689,
      "training_step_time": 10.19713830947876
    },
    {
      "epoch": 2.5516510009765625e-05,
      "model_forward_time": 3.1371700763702393,
      "step": 6689
    },
    {
      "epoch": 2.5516510009765625e-05,
      "step": 6689,
      "training_step_time": 10.304824352264404
    },
    {
      "epoch": 2.552032470703125e-05,
      "grad_norm": 0.37536290287971497,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 6690
    },
    {
      "epoch": 2.552032470703125e-05,
      "model_forward_time": 3.174387216567993,
      "step": 6690
    },
    {
      "dynamics_loss_avg": 0.08161931056529284,
      "epoch": 2.552032470703125e-05,
      "step": 6690
    },
    {
      "action_loss_avg": 0.032813158072531226,
      "epoch": 2.552032470703125e-05,
      "step": 6690
    },
    {
      "epoch": 2.552032470703125e-05,
      "step": 6690,
      "training_step_time": 9.952443599700928
    },
    {
      "epoch": 2.552032470703125e-05,
      "model_forward_time": 3.1430952548980713,
      "step": 6690
    },
    {
      "epoch": 2.552032470703125e-05,
      "step": 6690,
      "training_step_time": 9.914048671722412
    },
    {
      "epoch": 2.552032470703125e-05,
      "model_forward_time": 3.1335437297821045,
      "step": 6690
    },
    {
      "epoch": 2.552032470703125e-05,
      "step": 6690,
      "training_step_time": 9.904776096343994
    },
    {
      "epoch": 2.552032470703125e-05,
      "model_forward_time": 3.438911199569702,
      "step": 6690
    },
    {
      "epoch": 2.552032470703125e-05,
      "step": 6690,
      "training_step_time": 10.307193517684937
    },
    {
      "epoch": 2.5524139404296874e-05,
      "model_forward_time": 3.1852171421051025,
      "step": 6691
    },
    {
      "epoch": 2.5524139404296874e-05,
      "step": 6691,
      "training_step_time": 9.965595006942749
    },
    {
      "epoch": 2.5524139404296874e-05,
      "model_forward_time": 3.136434555053711,
      "step": 6691
    },
    {
      "epoch": 2.5524139404296874e-05,
      "step": 6691,
      "training_step_time": 9.911993980407715
    },
    {
      "epoch": 2.5524139404296874e-05,
      "model_forward_time": 3.136254072189331,
      "step": 6691
    },
    {
      "epoch": 2.5524139404296874e-05,
      "step": 6691,
      "training_step_time": 9.91430115699768
    },
    {
      "epoch": 2.5524139404296874e-05,
      "model_forward_time": 3.1407289505004883,
      "step": 6691
    },
    {
      "epoch": 2.5524139404296874e-05,
      "step": 6691,
      "training_step_time": 10.309804439544678
    },
    {
      "epoch": 2.55279541015625e-05,
      "model_forward_time": 3.4400901794433594,
      "step": 6692
    },
    {
      "epoch": 2.55279541015625e-05,
      "step": 6692,
      "training_step_time": 10.218040227890015
    },
    {
      "epoch": 2.55279541015625e-05,
      "model_forward_time": 3.138840913772583,
      "step": 6692
    },
    {
      "epoch": 2.55279541015625e-05,
      "step": 6692,
      "training_step_time": 9.919383525848389
    },
    {
      "epoch": 2.55279541015625e-05,
      "model_forward_time": 3.138803482055664,
      "step": 6692
    },
    {
      "dynamics_loss_avg": 0.07545417658984661,
      "epoch": 2.55279541015625e-05,
      "step": 6692
    },
    {
      "action_loss_avg": 0.02434200285933912,
      "epoch": 2.55279541015625e-05,
      "step": 6692
    },
    {
      "epoch": 2.55279541015625e-05,
      "step": 6692,
      "training_step_time": 9.92589783668518
    },
    {
      "epoch": 2.55279541015625e-05,
      "model_forward_time": 3.1393163204193115,
      "step": 6692
    },
    {
      "epoch": 2.55279541015625e-05,
      "step": 6692,
      "training_step_time": 10.029465675354004
    },
    {
      "epoch": 2.5531768798828124e-05,
      "model_forward_time": 3.1864166259765625,
      "step": 6693
    },
    {
      "epoch": 2.5531768798828124e-05,
      "step": 6693,
      "training_step_time": 9.96757984161377
    },
    {
      "epoch": 2.5531768798828124e-05,
      "model_forward_time": 3.41624116897583,
      "step": 6693
    },
    {
      "epoch": 2.5531768798828124e-05,
      "step": 6693,
      "training_step_time": 10.197898626327515
    },
    {
      "epoch": 2.5531768798828124e-05,
      "model_forward_time": 3.138479232788086,
      "step": 6693
    },
    {
      "epoch": 2.5531768798828124e-05,
      "step": 6693,
      "training_step_time": 9.921150922775269
    },
    {
      "epoch": 2.5531768798828124e-05,
      "model_forward_time": 3.1448349952697754,
      "step": 6693
    },
    {
      "epoch": 2.5531768798828124e-05,
      "step": 6693,
      "training_step_time": 10.228686571121216
    },
    {
      "epoch": 2.553558349609375e-05,
      "model_forward_time": 3.186300754547119,
      "step": 6694
    },
    {
      "epoch": 2.553558349609375e-05,
      "step": 6694,
      "training_step_time": 9.965457916259766
    },
    {
      "epoch": 2.553558349609375e-05,
      "model_forward_time": 3.1339685916900635,
      "step": 6694
    },
    {
      "epoch": 2.553558349609375e-05,
      "step": 6694,
      "training_step_time": 9.907117128372192
    },
    {
      "epoch": 2.553558349609375e-05,
      "model_forward_time": 3.422703981399536,
      "step": 6694
    },
    {
      "epoch": 2.553558349609375e-05,
      "step": 6694,
      "training_step_time": 10.192368268966675
    },
    {
      "epoch": 2.553558349609375e-05,
      "model_forward_time": 3.13824200630188,
      "step": 6694
    },
    {
      "epoch": 2.553558349609375e-05,
      "step": 6694,
      "training_step_time": 10.301252841949463
    },
    {
      "epoch": 2.5539398193359376e-05,
      "model_forward_time": 3.185831308364868,
      "step": 6695
    },
    {
      "dynamics_loss_avg": 0.04615816585719586,
      "epoch": 2.5539398193359376e-05,
      "step": 6695
    },
    {
      "action_loss_avg": 0.018141145072877406,
      "epoch": 2.5539398193359376e-05,
      "step": 6695
    },
    {
      "epoch": 2.5539398193359376e-05,
      "step": 6695,
      "training_step_time": 9.965174674987793
    },
    {
      "epoch": 2.5539398193359376e-05,
      "model_forward_time": 3.1334831714630127,
      "step": 6695
    },
    {
      "epoch": 2.5539398193359376e-05,
      "step": 6695,
      "training_step_time": 9.905250787734985
    },
    {
      "epoch": 2.5539398193359376e-05,
      "model_forward_time": 3.137779951095581,
      "step": 6695
    },
    {
      "epoch": 2.5539398193359376e-05,
      "step": 6695,
      "training_step_time": 9.91319489479065
    },
    {
      "epoch": 2.5539398193359376e-05,
      "model_forward_time": 3.421635150909424,
      "step": 6695
    },
    {
      "epoch": 2.5539398193359376e-05,
      "step": 6695,
      "training_step_time": 10.319742441177368
    },
    {
      "epoch": 2.5543212890625e-05,
      "model_forward_time": 3.1768412590026855,
      "step": 6696
    },
    {
      "epoch": 2.5543212890625e-05,
      "step": 6696,
      "training_step_time": 9.961965799331665
    },
    {
      "epoch": 2.5543212890625e-05,
      "model_forward_time": 3.138413906097412,
      "step": 6696
    },
    {
      "epoch": 2.5543212890625e-05,
      "step": 6696,
      "training_step_time": 9.920755863189697
    },
    {
      "epoch": 2.5543212890625e-05,
      "model_forward_time": 3.1484763622283936,
      "step": 6696
    },
    {
      "epoch": 2.5543212890625e-05,
      "step": 6696,
      "training_step_time": 9.932872533798218
    },
    {
      "epoch": 2.5543212890625e-05,
      "model_forward_time": 3.145658493041992,
      "step": 6696
    },
    {
      "epoch": 2.5543212890625e-05,
      "step": 6696,
      "training_step_time": 10.3284330368042
    },
    {
      "epoch": 2.5547027587890625e-05,
      "model_forward_time": 3.678457260131836,
      "step": 6697
    },
    {
      "epoch": 2.5547027587890625e-05,
      "step": 6697,
      "training_step_time": 10.47344183921814
    },
    {
      "epoch": 2.5547027587890625e-05,
      "model_forward_time": 3.1422178745269775,
      "step": 6697
    },
    {
      "epoch": 2.5547027587890625e-05,
      "step": 6697,
      "training_step_time": 9.912543773651123
    },
    {
      "epoch": 2.5547027587890625e-05,
      "model_forward_time": 3.139263868331909,
      "step": 6697
    },
    {
      "dynamics_loss_avg": 0.06519601494073868,
      "epoch": 2.5547027587890625e-05,
      "step": 6697
    },
    {
      "action_loss_avg": 0.024333867803215982,
      "epoch": 2.5547027587890625e-05,
      "step": 6697
    },
    {
      "epoch": 2.5547027587890625e-05,
      "step": 6697,
      "training_step_time": 9.911881685256958
    },
    {
      "epoch": 2.5547027587890625e-05,
      "model_forward_time": 3.141735792160034,
      "step": 6697
    },
    {
      "epoch": 2.5547027587890625e-05,
      "step": 6697,
      "training_step_time": 10.01837420463562
    },
    {
      "epoch": 2.555084228515625e-05,
      "model_forward_time": 3.1616058349609375,
      "step": 6698
    },
    {
      "epoch": 2.555084228515625e-05,
      "step": 6698,
      "training_step_time": 9.939716577529907
    },
    {
      "epoch": 2.555084228515625e-05,
      "model_forward_time": 3.4375102519989014,
      "step": 6698
    },
    {
      "epoch": 2.555084228515625e-05,
      "step": 6698,
      "training_step_time": 10.209046602249146
    },
    {
      "epoch": 2.555084228515625e-05,
      "model_forward_time": 3.14117169380188,
      "step": 6698
    },
    {
      "epoch": 2.555084228515625e-05,
      "step": 6698,
      "training_step_time": 9.910136222839355
    },
    {
      "epoch": 2.555084228515625e-05,
      "model_forward_time": 3.128133773803711,
      "step": 6698
    },
    {
      "epoch": 2.555084228515625e-05,
      "step": 6698,
      "training_step_time": 10.264548778533936
    },
    {
      "epoch": 2.5554656982421874e-05,
      "model_forward_time": 2.4689853191375732,
      "step": 6699
    },
    {
      "epoch": 2.5554656982421874e-05,
      "step": 6699,
      "training_step_time": 7.494490146636963
    },
    {
      "epoch": 2.5554656982421874e-05,
      "model_forward_time": 3.1414411067962646,
      "step": 6699
    },
    {
      "epoch": 2.5554656982421874e-05,
      "step": 6699,
      "training_step_time": 9.920310020446777
    },
    {
      "epoch": 2.5554656982421874e-05,
      "model_forward_time": 3.4588842391967773,
      "step": 6699
    },
    {
      "epoch": 2.5554656982421874e-05,
      "step": 6699,
      "training_step_time": 10.236180305480957
    },
    {
      "epoch": 2.5554656982421874e-05,
      "model_forward_time": 3.1363444328308105,
      "step": 6699
    },
    {
      "epoch": 2.5554656982421874e-05,
      "step": 6699,
      "training_step_time": 10.040411233901978
    },
    {
      "epoch": 2.55584716796875e-05,
      "grad_norm": 0.3375657796859741,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 6700
    },
    {
      "epoch": 2.55584716796875e-05,
      "model_forward_time": 3.187288761138916,
      "step": 6700
    },
    {
      "dynamics_loss_avg": 0.05653002690523863,
      "epoch": 2.55584716796875e-05,
      "step": 6700
    },
    {
      "action_loss_avg": 0.01874482585117221,
      "epoch": 2.55584716796875e-05,
      "step": 6700
    },
    {
      "epoch": 2.55584716796875e-05,
      "step": 6700,
      "training_step_time": 9.96673059463501
    },
    {
      "epoch": 2.55584716796875e-05,
      "model_forward_time": 3.138434648513794,
      "step": 6700
    },
    {
      "epoch": 2.55584716796875e-05,
      "step": 6700,
      "training_step_time": 9.912118434906006
    },
    {
      "epoch": 2.55584716796875e-05,
      "model_forward_time": 3.1365294456481934,
      "step": 6700
    },
    {
      "epoch": 2.55584716796875e-05,
      "step": 6700,
      "training_step_time": 9.911059617996216
    },
    {
      "epoch": 2.55584716796875e-05,
      "model_forward_time": 3.3865222930908203,
      "step": 6700
    },
    {
      "epoch": 2.55584716796875e-05,
      "step": 6700,
      "training_step_time": 10.275656700134277
    },
    {
      "epoch": 2.5562286376953124e-05,
      "model_forward_time": 3.1805474758148193,
      "step": 6701
    },
    {
      "epoch": 2.5562286376953124e-05,
      "step": 6701,
      "training_step_time": 9.954355716705322
    },
    {
      "epoch": 2.5562286376953124e-05,
      "model_forward_time": 3.137141704559326,
      "step": 6701
    },
    {
      "epoch": 2.5562286376953124e-05,
      "step": 6701,
      "training_step_time": 9.910563230514526
    },
    {
      "epoch": 2.5562286376953124e-05,
      "model_forward_time": 3.139224052429199,
      "step": 6701
    },
    {
      "epoch": 2.5562286376953124e-05,
      "step": 6701,
      "training_step_time": 9.91799783706665
    },
    {
      "epoch": 2.5562286376953124e-05,
      "model_forward_time": 3.139829397201538,
      "step": 6701
    },
    {
      "epoch": 2.5562286376953124e-05,
      "step": 6701,
      "training_step_time": 10.302310466766357
    },
    {
      "epoch": 2.556610107421875e-05,
      "model_forward_time": 3.4649159908294678,
      "step": 6702
    },
    {
      "epoch": 2.556610107421875e-05,
      "step": 6702,
      "training_step_time": 10.243881940841675
    },
    {
      "epoch": 2.556610107421875e-05,
      "model_forward_time": 3.1385602951049805,
      "step": 6702
    },
    {
      "epoch": 2.556610107421875e-05,
      "step": 6702,
      "training_step_time": 9.91355037689209
    },
    {
      "epoch": 2.556610107421875e-05,
      "model_forward_time": 3.1352853775024414,
      "step": 6702
    },
    {
      "dynamics_loss_avg": 0.057101001217961314,
      "epoch": 2.556610107421875e-05,
      "step": 6702
    },
    {
      "action_loss_avg": 0.02107227244414389,
      "epoch": 2.556610107421875e-05,
      "step": 6702
    },
    {
      "epoch": 2.556610107421875e-05,
      "step": 6702,
      "training_step_time": 9.909344673156738
    },
    {
      "epoch": 2.556610107421875e-05,
      "model_forward_time": 3.1378984451293945,
      "step": 6702
    },
    {
      "epoch": 2.556610107421875e-05,
      "step": 6702,
      "training_step_time": 10.033588886260986
    },
    {
      "epoch": 2.5569915771484376e-05,
      "model_forward_time": 3.173004150390625,
      "step": 6703
    },
    {
      "epoch": 2.5569915771484376e-05,
      "step": 6703,
      "training_step_time": 9.951601028442383
    },
    {
      "epoch": 2.5569915771484376e-05,
      "model_forward_time": 3.412299871444702,
      "step": 6703
    },
    {
      "epoch": 2.5569915771484376e-05,
      "step": 6703,
      "training_step_time": 10.196477890014648
    },
    {
      "epoch": 2.5569915771484376e-05,
      "model_forward_time": 3.141779899597168,
      "step": 6703
    },
    {
      "epoch": 2.5569915771484376e-05,
      "step": 6703,
      "training_step_time": 9.92073655128479
    },
    {
      "epoch": 2.5569915771484376e-05,
      "model_forward_time": 3.148930311203003,
      "step": 6703
    },
    {
      "epoch": 2.5569915771484376e-05,
      "step": 6703,
      "training_step_time": 10.040309429168701
    },
    {
      "epoch": 2.557373046875e-05,
      "model_forward_time": 3.1845626831054688,
      "step": 6704
    },
    {
      "epoch": 2.557373046875e-05,
      "step": 6704,
      "training_step_time": 9.962922096252441
    },
    {
      "epoch": 2.557373046875e-05,
      "model_forward_time": 3.138349771499634,
      "step": 6704
    },
    {
      "epoch": 2.557373046875e-05,
      "step": 6704,
      "training_step_time": 9.912588357925415
    },
    {
      "epoch": 2.557373046875e-05,
      "model_forward_time": 3.413498878479004,
      "step": 6704
    },
    {
      "epoch": 2.557373046875e-05,
      "step": 6704,
      "training_step_time": 10.1851327419281
    },
    {
      "epoch": 2.557373046875e-05,
      "model_forward_time": 3.139126777648926,
      "step": 6704
    },
    {
      "epoch": 2.557373046875e-05,
      "step": 6704,
      "training_step_time": 10.314181327819824
    },
    {
      "epoch": 2.5577545166015625e-05,
      "model_forward_time": 3.1548142433166504,
      "step": 6705
    },
    {
      "dynamics_loss_avg": 0.048171301186084745,
      "epoch": 2.5577545166015625e-05,
      "step": 6705
    },
    {
      "action_loss_avg": 0.01659195302054286,
      "epoch": 2.5577545166015625e-05,
      "step": 6705
    },
    {
      "epoch": 2.5577545166015625e-05,
      "step": 6705,
      "training_step_time": 9.930045127868652
    },
    {
      "epoch": 2.5577545166015625e-05,
      "model_forward_time": 3.1328065395355225,
      "step": 6705
    },
    {
      "epoch": 2.5577545166015625e-05,
      "step": 6705,
      "training_step_time": 9.898601531982422
    },
    {
      "epoch": 2.5577545166015625e-05,
      "model_forward_time": 3.138648748397827,
      "step": 6705
    },
    {
      "epoch": 2.5577545166015625e-05,
      "step": 6705,
      "training_step_time": 10.200822830200195
    },
    {
      "epoch": 2.5577545166015625e-05,
      "model_forward_time": 3.1337761878967285,
      "step": 6705
    },
    {
      "epoch": 2.5577545166015625e-05,
      "step": 6705,
      "training_step_time": 10.289098024368286
    },
    {
      "epoch": 2.558135986328125e-05,
      "model_forward_time": 3.175671100616455,
      "step": 6706
    },
    {
      "epoch": 2.558135986328125e-05,
      "step": 6706,
      "training_step_time": 9.947821140289307
    },
    {
      "epoch": 2.558135986328125e-05,
      "model_forward_time": 3.129643678665161,
      "step": 6706
    },
    {
      "epoch": 2.558135986328125e-05,
      "step": 6706,
      "training_step_time": 9.90399169921875
    },
    {
      "epoch": 2.558135986328125e-05,
      "model_forward_time": 3.1361451148986816,
      "step": 6706
    },
    {
      "epoch": 2.558135986328125e-05,
      "step": 6706,
      "training_step_time": 9.909153938293457
    },
    {
      "epoch": 2.558135986328125e-05,
      "model_forward_time": 3.4129981994628906,
      "step": 6706
    },
    {
      "epoch": 2.558135986328125e-05,
      "step": 6706,
      "training_step_time": 10.29618787765503
    },
    {
      "epoch": 2.5585174560546874e-05,
      "model_forward_time": 3.190058469772339,
      "step": 6707
    },
    {
      "epoch": 2.5585174560546874e-05,
      "step": 6707,
      "training_step_time": 9.969141006469727
    },
    {
      "epoch": 2.5585174560546874e-05,
      "model_forward_time": 3.1344399452209473,
      "step": 6707
    },
    {
      "epoch": 2.5585174560546874e-05,
      "step": 6707,
      "training_step_time": 9.908877849578857
    },
    {
      "epoch": 2.5585174560546874e-05,
      "model_forward_time": 3.137897253036499,
      "step": 6707
    },
    {
      "dynamics_loss_avg": 0.07297180648893117,
      "epoch": 2.5585174560546874e-05,
      "step": 6707
    },
    {
      "action_loss_avg": 0.031059625186026098,
      "epoch": 2.5585174560546874e-05,
      "step": 6707
    },
    {
      "epoch": 2.5585174560546874e-05,
      "step": 6707,
      "training_step_time": 9.916447639465332
    },
    {
      "epoch": 2.5585174560546874e-05,
      "model_forward_time": 3.1368141174316406,
      "step": 6707
    },
    {
      "epoch": 2.5585174560546874e-05,
      "step": 6707,
      "training_step_time": 10.014444351196289
    },
    {
      "epoch": 2.55889892578125e-05,
      "model_forward_time": 3.1630241870880127,
      "step": 6708
    },
    {
      "epoch": 2.55889892578125e-05,
      "step": 6708,
      "training_step_time": 10.24918818473816
    },
    {
      "epoch": 2.55889892578125e-05,
      "model_forward_time": 3.1358792781829834,
      "step": 6708
    },
    {
      "epoch": 2.55889892578125e-05,
      "step": 6708,
      "training_step_time": 9.914612054824829
    },
    {
      "epoch": 2.55889892578125e-05,
      "model_forward_time": 3.13643741607666,
      "step": 6708
    },
    {
      "epoch": 2.55889892578125e-05,
      "step": 6708,
      "training_step_time": 9.914665699005127
    },
    {
      "epoch": 2.55889892578125e-05,
      "model_forward_time": 3.1364359855651855,
      "step": 6708
    },
    {
      "epoch": 2.55889892578125e-05,
      "step": 6708,
      "training_step_time": 10.02529764175415
    },
    {
      "epoch": 2.5592803955078124e-05,
      "model_forward_time": 3.1767117977142334,
      "step": 6709
    },
    {
      "epoch": 2.5592803955078124e-05,
      "step": 6709,
      "training_step_time": 9.959760665893555
    },
    {
      "epoch": 2.5592803955078124e-05,
      "model_forward_time": 3.408357858657837,
      "step": 6709
    },
    {
      "epoch": 2.5592803955078124e-05,
      "step": 6709,
      "training_step_time": 10.18746566772461
    },
    {
      "epoch": 2.5592803955078124e-05,
      "model_forward_time": 3.142646074295044,
      "step": 6709
    },
    {
      "epoch": 2.5592803955078124e-05,
      "step": 6709,
      "training_step_time": 9.92046046257019
    },
    {
      "epoch": 2.5592803955078124e-05,
      "model_forward_time": 3.1368722915649414,
      "step": 6709
    },
    {
      "epoch": 2.5592803955078124e-05,
      "step": 6709,
      "training_step_time": 10.27773141860962
    },
    {
      "epoch": 2.559661865234375e-05,
      "grad_norm": 0.42315056920051575,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 6710
    },
    {
      "epoch": 2.559661865234375e-05,
      "model_forward_time": 3.189826250076294,
      "step": 6710
    },
    {
      "dynamics_loss_avg": 0.0611929502338171,
      "epoch": 2.559661865234375e-05,
      "step": 6710
    },
    {
      "action_loss_avg": 0.024125611688941718,
      "epoch": 2.559661865234375e-05,
      "step": 6710
    },
    {
      "epoch": 2.559661865234375e-05,
      "step": 6710,
      "training_step_time": 9.967453241348267
    },
    {
      "epoch": 2.559661865234375e-05,
      "model_forward_time": 3.1381726264953613,
      "step": 6710
    },
    {
      "epoch": 2.559661865234375e-05,
      "step": 6710,
      "training_step_time": 9.915750503540039
    },
    {
      "epoch": 2.559661865234375e-05,
      "model_forward_time": 3.409477472305298,
      "step": 6710
    },
    {
      "epoch": 2.559661865234375e-05,
      "step": 6710,
      "training_step_time": 10.190014362335205
    },
    {
      "epoch": 2.559661865234375e-05,
      "model_forward_time": 3.137946367263794,
      "step": 6710
    },
    {
      "epoch": 2.559661865234375e-05,
      "step": 6710,
      "training_step_time": 10.29936408996582
    },
    {
      "epoch": 2.5600433349609376e-05,
      "model_forward_time": 3.157313585281372,
      "step": 6711
    },
    {
      "epoch": 2.5600433349609376e-05,
      "step": 6711,
      "training_step_time": 9.9324369430542
    },
    {
      "epoch": 2.5600433349609376e-05,
      "model_forward_time": 3.1380083560943604,
      "step": 6711
    },
    {
      "epoch": 2.5600433349609376e-05,
      "step": 6711,
      "training_step_time": 9.913209676742554
    },
    {
      "epoch": 2.5600433349609376e-05,
      "model_forward_time": 3.1341922283172607,
      "step": 6711
    },
    {
      "epoch": 2.5600433349609376e-05,
      "step": 6711,
      "training_step_time": 9.9155912399292
    },
    {
      "epoch": 2.5600433349609376e-05,
      "model_forward_time": 3.392415761947632,
      "step": 6711
    },
    {
      "epoch": 2.5600433349609376e-05,
      "step": 6711,
      "training_step_time": 10.294842958450317
    },
    {
      "epoch": 2.5604248046875e-05,
      "model_forward_time": 3.1827471256256104,
      "step": 6712
    },
    {
      "epoch": 2.5604248046875e-05,
      "step": 6712,
      "training_step_time": 9.960408449172974
    },
    {
      "epoch": 2.5604248046875e-05,
      "model_forward_time": 3.134955406188965,
      "step": 6712
    },
    {
      "epoch": 2.5604248046875e-05,
      "step": 6712,
      "training_step_time": 9.91239595413208
    },
    {
      "epoch": 2.5604248046875e-05,
      "model_forward_time": 3.132235527038574,
      "step": 6712
    },
    {
      "dynamics_loss_avg": 0.05174867697060108,
      "epoch": 2.5604248046875e-05,
      "step": 6712
    },
    {
      "action_loss_avg": 0.020252142380923033,
      "epoch": 2.5604248046875e-05,
      "step": 6712
    },
    {
      "epoch": 2.5604248046875e-05,
      "step": 6712,
      "training_step_time": 9.908393859863281
    },
    {
      "epoch": 2.5604248046875e-05,
      "model_forward_time": 3.137606620788574,
      "step": 6712
    },
    {
      "epoch": 2.5604248046875e-05,
      "step": 6712,
      "training_step_time": 10.283254861831665
    },
    {
      "epoch": 2.5608062744140625e-05,
      "model_forward_time": 3.479158639907837,
      "step": 6713
    },
    {
      "epoch": 2.5608062744140625e-05,
      "step": 6713,
      "training_step_time": 10.25460934638977
    },
    {
      "epoch": 2.5608062744140625e-05,
      "model_forward_time": 3.140424966812134,
      "step": 6713
    },
    {
      "epoch": 2.5608062744140625e-05,
      "step": 6713,
      "training_step_time": 9.918856143951416
    },
    {
      "epoch": 2.5608062744140625e-05,
      "model_forward_time": 3.1442928314208984,
      "step": 6713
    },
    {
      "epoch": 2.5608062744140625e-05,
      "step": 6713,
      "training_step_time": 9.92479920387268
    },
    {
      "epoch": 2.5608062744140625e-05,
      "model_forward_time": 3.1372311115264893,
      "step": 6713
    },
    {
      "epoch": 2.5608062744140625e-05,
      "step": 6713,
      "training_step_time": 10.290908336639404
    },
    {
      "epoch": 2.561187744140625e-05,
      "model_forward_time": 3.1717727184295654,
      "step": 6714
    },
    {
      "epoch": 2.561187744140625e-05,
      "step": 6714,
      "training_step_time": 9.949153661727905
    },
    {
      "epoch": 2.561187744140625e-05,
      "model_forward_time": 3.1358237266540527,
      "step": 6714
    },
    {
      "epoch": 2.561187744140625e-05,
      "step": 6714,
      "training_step_time": 10.195495843887329
    },
    {
      "epoch": 2.561187744140625e-05,
      "model_forward_time": 3.1344220638275146,
      "step": 6714
    },
    {
      "epoch": 2.561187744140625e-05,
      "step": 6714,
      "training_step_time": 9.910741090774536
    },
    {
      "epoch": 2.561187744140625e-05,
      "model_forward_time": 3.135582208633423,
      "step": 6714
    },
    {
      "epoch": 2.561187744140625e-05,
      "step": 6714,
      "training_step_time": 10.0107421875
    },
    {
      "epoch": 2.5615692138671874e-05,
      "model_forward_time": 3.1877520084381104,
      "step": 6715
    },
    {
      "dynamics_loss_avg": 0.04700617361813784,
      "epoch": 2.5615692138671874e-05,
      "step": 6715
    },
    {
      "action_loss_avg": 0.01996548450551927,
      "epoch": 2.5615692138671874e-05,
      "step": 6715
    },
    {
      "epoch": 2.5615692138671874e-05,
      "step": 6715,
      "training_step_time": 9.983531475067139
    },
    {
      "epoch": 2.5615692138671874e-05,
      "model_forward_time": 3.1452102661132812,
      "step": 6715
    },
    {
      "epoch": 2.5615692138671874e-05,
      "step": 6715,
      "training_step_time": 9.921122074127197
    },
    {
      "epoch": 2.5615692138671874e-05,
      "model_forward_time": 3.148794651031494,
      "step": 6715
    },
    {
      "epoch": 2.5615692138671874e-05,
      "step": 6715,
      "training_step_time": 10.203383684158325
    },
    {
      "epoch": 2.5615692138671874e-05,
      "model_forward_time": 3.1460952758789062,
      "step": 6715
    },
    {
      "epoch": 2.5615692138671874e-05,
      "step": 6715,
      "training_step_time": 10.261520862579346
    },
    {
      "epoch": 2.56195068359375e-05,
      "model_forward_time": 3.185502290725708,
      "step": 6716
    },
    {
      "epoch": 2.56195068359375e-05,
      "step": 6716,
      "training_step_time": 9.968674182891846
    },
    {
      "epoch": 2.56195068359375e-05,
      "model_forward_time": 3.145169973373413,
      "step": 6716
    },
    {
      "epoch": 2.56195068359375e-05,
      "step": 6716,
      "training_step_time": 9.92490005493164
    },
    {
      "epoch": 2.56195068359375e-05,
      "model_forward_time": 3.153069257736206,
      "step": 6716
    },
    {
      "epoch": 2.56195068359375e-05,
      "step": 6716,
      "training_step_time": 9.931139707565308
    },
    {
      "epoch": 2.56195068359375e-05,
      "model_forward_time": 3.1450393199920654,
      "step": 6716
    },
    {
      "epoch": 2.56195068359375e-05,
      "step": 6716,
      "training_step_time": 10.33004093170166
    },
    {
      "epoch": 2.5623321533203124e-05,
      "model_forward_time": 3.1964828968048096,
      "step": 6717
    },
    {
      "epoch": 2.5623321533203124e-05,
      "step": 6717,
      "training_step_time": 9.99192476272583
    },
    {
      "epoch": 2.5623321533203124e-05,
      "model_forward_time": 3.140674114227295,
      "step": 6717
    },
    {
      "epoch": 2.5623321533203124e-05,
      "step": 6717,
      "training_step_time": 9.914310455322266
    },
    {
      "epoch": 2.5623321533203124e-05,
      "model_forward_time": 3.1467056274414062,
      "step": 6717
    },
    {
      "dynamics_loss_avg": 0.058519018813967705,
      "epoch": 2.5623321533203124e-05,
      "step": 6717
    },
    {
      "action_loss_avg": 0.01839935565367341,
      "epoch": 2.5623321533203124e-05,
      "step": 6717
    },
    {
      "epoch": 2.5623321533203124e-05,
      "step": 6717,
      "training_step_time": 9.927829027175903
    },
    {
      "epoch": 2.5623321533203124e-05,
      "model_forward_time": 3.1392452716827393,
      "step": 6717
    },
    {
      "epoch": 2.5623321533203124e-05,
      "step": 6717,
      "training_step_time": 10.327860355377197
    },
    {
      "epoch": 2.562713623046875e-05,
      "model_forward_time": 3.464181423187256,
      "step": 6718
    },
    {
      "epoch": 2.562713623046875e-05,
      "step": 6718,
      "training_step_time": 10.241828680038452
    },
    {
      "epoch": 2.562713623046875e-05,
      "model_forward_time": 3.1348581314086914,
      "step": 6718
    },
    {
      "epoch": 2.562713623046875e-05,
      "step": 6718,
      "training_step_time": 9.906259059906006
    },
    {
      "epoch": 2.562713623046875e-05,
      "model_forward_time": 3.135697364807129,
      "step": 6718
    },
    {
      "epoch": 2.562713623046875e-05,
      "step": 6718,
      "training_step_time": 9.90839147567749
    },
    {
      "epoch": 2.562713623046875e-05,
      "model_forward_time": 3.140033006668091,
      "step": 6718
    },
    {
      "epoch": 2.562713623046875e-05,
      "step": 6718,
      "training_step_time": 10.026252746582031
    },
    {
      "epoch": 2.5630950927734376e-05,
      "model_forward_time": 3.1959493160247803,
      "step": 6719
    },
    {
      "epoch": 2.5630950927734376e-05,
      "step": 6719,
      "training_step_time": 9.972377300262451
    },
    {
      "epoch": 2.5630950927734376e-05,
      "model_forward_time": 3.138746500015259,
      "step": 6719
    },
    {
      "epoch": 2.5630950927734376e-05,
      "step": 6719,
      "training_step_time": 10.222283840179443
    },
    {
      "epoch": 2.5630950927734376e-05,
      "model_forward_time": 3.1457858085632324,
      "step": 6719
    },
    {
      "epoch": 2.5630950927734376e-05,
      "step": 6719,
      "training_step_time": 9.91179084777832
    },
    {
      "epoch": 2.5630950927734376e-05,
      "model_forward_time": 3.130866050720215,
      "step": 6719
    },
    {
      "epoch": 2.5630950927734376e-05,
      "step": 6719,
      "training_step_time": 10.012057542800903
    },
    {
      "epoch": 2.5634765625e-05,
      "grad_norm": 0.42844244837760925,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 6720
    },
    {
      "epoch": 2.5634765625e-05,
      "model_forward_time": 3.161949872970581,
      "step": 6720
    },
    {
      "dynamics_loss_avg": 0.06045455206185579,
      "epoch": 2.5634765625e-05,
      "step": 6720
    },
    {
      "action_loss_avg": 0.02346398653462529,
      "epoch": 2.5634765625e-05,
      "step": 6720
    },
    {
      "epoch": 2.5634765625e-05,
      "step": 6720,
      "training_step_time": 9.933451414108276
    },
    {
      "epoch": 2.5634765625e-05,
      "model_forward_time": 3.1405997276306152,
      "step": 6720
    },
    {
      "epoch": 2.5634765625e-05,
      "step": 6720,
      "training_step_time": 9.910336971282959
    },
    {
      "epoch": 2.5634765625e-05,
      "model_forward_time": 3.4173264503479004,
      "step": 6720
    },
    {
      "epoch": 2.5634765625e-05,
      "step": 6720,
      "training_step_time": 10.18225622177124
    },
    {
      "epoch": 2.5634765625e-05,
      "model_forward_time": 3.1332850456237793,
      "step": 6720
    },
    {
      "epoch": 2.5634765625e-05,
      "step": 6720,
      "training_step_time": 10.25520396232605
    },
    {
      "epoch": 2.5638580322265625e-05,
      "model_forward_time": 3.1635029315948486,
      "step": 6721
    },
    {
      "epoch": 2.5638580322265625e-05,
      "step": 6721,
      "training_step_time": 9.932931661605835
    },
    {
      "epoch": 2.5638580322265625e-05,
      "model_forward_time": 3.13226056098938,
      "step": 6721
    },
    {
      "epoch": 2.5638580322265625e-05,
      "step": 6721,
      "training_step_time": 9.90319538116455
    },
    {
      "epoch": 2.5638580322265625e-05,
      "model_forward_time": 3.1344213485717773,
      "step": 6721
    },
    {
      "epoch": 2.5638580322265625e-05,
      "step": 6721,
      "training_step_time": 9.90221357345581
    },
    {
      "epoch": 2.5638580322265625e-05,
      "model_forward_time": 3.3977372646331787,
      "step": 6721
    },
    {
      "epoch": 2.5638580322265625e-05,
      "step": 6721,
      "training_step_time": 10.273345708847046
    },
    {
      "epoch": 2.564239501953125e-05,
      "model_forward_time": 3.194127321243286,
      "step": 6722
    },
    {
      "epoch": 2.564239501953125e-05,
      "step": 6722,
      "training_step_time": 9.966170072555542
    },
    {
      "epoch": 2.564239501953125e-05,
      "model_forward_time": 3.138843536376953,
      "step": 6722
    },
    {
      "epoch": 2.564239501953125e-05,
      "step": 6722,
      "training_step_time": 9.9050612449646
    },
    {
      "epoch": 2.564239501953125e-05,
      "model_forward_time": 3.135077714920044,
      "step": 6722
    },
    {
      "dynamics_loss_avg": 0.059277843311429025,
      "epoch": 2.564239501953125e-05,
      "step": 6722
    },
    {
      "action_loss_avg": 0.022244102833792568,
      "epoch": 2.564239501953125e-05,
      "step": 6722
    },
    {
      "epoch": 2.564239501953125e-05,
      "step": 6722,
      "training_step_time": 9.909854412078857
    },
    {
      "epoch": 2.564239501953125e-05,
      "model_forward_time": 3.1376583576202393,
      "step": 6722
    },
    {
      "epoch": 2.564239501953125e-05,
      "step": 6722,
      "training_step_time": 10.302376508712769
    },
    {
      "epoch": 2.5646209716796874e-05,
      "model_forward_time": 3.489635944366455,
      "step": 6723
    },
    {
      "epoch": 2.5646209716796874e-05,
      "step": 6723,
      "training_step_time": 10.266061782836914
    },
    {
      "epoch": 2.5646209716796874e-05,
      "model_forward_time": 3.1423254013061523,
      "step": 6723
    },
    {
      "epoch": 2.5646209716796874e-05,
      "step": 6723,
      "training_step_time": 9.918411493301392
    },
    {
      "epoch": 2.5646209716796874e-05,
      "model_forward_time": 3.1371214389801025,
      "step": 6723
    },
    {
      "epoch": 2.5646209716796874e-05,
      "step": 6723,
      "training_step_time": 9.910810947418213
    },
    {
      "epoch": 2.5646209716796874e-05,
      "model_forward_time": 3.139542818069458,
      "step": 6723
    },
    {
      "epoch": 2.5646209716796874e-05,
      "step": 6723,
      "training_step_time": 10.031062602996826
    },
    {
      "epoch": 2.56500244140625e-05,
      "model_forward_time": 3.16593599319458,
      "step": 6724
    },
    {
      "epoch": 2.56500244140625e-05,
      "step": 6724,
      "training_step_time": 9.942332983016968
    },
    {
      "epoch": 2.56500244140625e-05,
      "model_forward_time": 3.420403003692627,
      "step": 6724
    },
    {
      "epoch": 2.56500244140625e-05,
      "step": 6724,
      "training_step_time": 10.190944194793701
    },
    {
      "epoch": 2.56500244140625e-05,
      "model_forward_time": 3.138206720352173,
      "step": 6724
    },
    {
      "epoch": 2.56500244140625e-05,
      "step": 6724,
      "training_step_time": 9.913342714309692
    },
    {
      "epoch": 2.56500244140625e-05,
      "model_forward_time": 3.137799024581909,
      "step": 6724
    },
    {
      "epoch": 2.56500244140625e-05,
      "step": 6724,
      "training_step_time": 10.03172779083252
    },
    {
      "epoch": 2.5653839111328124e-05,
      "model_forward_time": 3.1760878562927246,
      "step": 6725
    },
    {
      "dynamics_loss_avg": 0.0655309360474348,
      "epoch": 2.5653839111328124e-05,
      "step": 6725
    },
    {
      "action_loss_avg": 0.024932358600199222,
      "epoch": 2.5653839111328124e-05,
      "step": 6725
    },
    {
      "epoch": 2.5653839111328124e-05,
      "step": 6725,
      "training_step_time": 9.949186086654663
    },
    {
      "epoch": 2.5653839111328124e-05,
      "model_forward_time": 3.1369056701660156,
      "step": 6725
    },
    {
      "epoch": 2.5653839111328124e-05,
      "step": 6725,
      "training_step_time": 9.914734840393066
    },
    {
      "epoch": 2.5653839111328124e-05,
      "model_forward_time": 3.137493848800659,
      "step": 6725
    },
    {
      "epoch": 2.5653839111328124e-05,
      "step": 6725,
      "training_step_time": 10.195485591888428
    },
    {
      "epoch": 2.5653839111328124e-05,
      "model_forward_time": 3.138176202774048,
      "step": 6725
    },
    {
      "epoch": 2.5653839111328124e-05,
      "step": 6725,
      "training_step_time": 10.037713766098022
    },
    {
      "epoch": 2.565765380859375e-05,
      "model_forward_time": 3.198448657989502,
      "step": 6726
    },
    {
      "epoch": 2.565765380859375e-05,
      "step": 6726,
      "training_step_time": 9.974611759185791
    },
    {
      "epoch": 2.565765380859375e-05,
      "model_forward_time": 3.136537790298462,
      "step": 6726
    },
    {
      "epoch": 2.565765380859375e-05,
      "step": 6726,
      "training_step_time": 9.908408164978027
    },
    {
      "epoch": 2.565765380859375e-05,
      "model_forward_time": 3.1339111328125,
      "step": 6726
    },
    {
      "epoch": 2.565765380859375e-05,
      "step": 6726,
      "training_step_time": 9.902746438980103
    },
    {
      "epoch": 2.565765380859375e-05,
      "model_forward_time": 3.455972671508789,
      "step": 6726
    },
    {
      "epoch": 2.565765380859375e-05,
      "step": 6726,
      "training_step_time": 10.313627481460571
    },
    {
      "epoch": 2.5661468505859376e-05,
      "model_forward_time": 3.1640841960906982,
      "step": 6727
    },
    {
      "epoch": 2.5661468505859376e-05,
      "step": 6727,
      "training_step_time": 9.930984735488892
    },
    {
      "epoch": 2.5661468505859376e-05,
      "model_forward_time": 3.1321873664855957,
      "step": 6727
    },
    {
      "epoch": 2.5661468505859376e-05,
      "step": 6727,
      "training_step_time": 9.900050163269043
    },
    {
      "epoch": 2.5661468505859376e-05,
      "model_forward_time": 3.1413888931274414,
      "step": 6727
    },
    {
      "dynamics_loss_avg": 0.053063310123980044,
      "epoch": 2.5661468505859376e-05,
      "step": 6727
    },
    {
      "action_loss_avg": 0.018373874295502902,
      "epoch": 2.5661468505859376e-05,
      "step": 6727
    },
    {
      "epoch": 2.5661468505859376e-05,
      "step": 6727,
      "training_step_time": 9.914840698242188
    },
    {
      "epoch": 2.5661468505859376e-05,
      "model_forward_time": 3.136651039123535,
      "step": 6727
    },
    {
      "epoch": 2.5661468505859376e-05,
      "step": 6727,
      "training_step_time": 10.244454860687256
    },
    {
      "epoch": 2.5665283203125e-05,
      "model_forward_time": 3.1771631240844727,
      "step": 6728
    },
    {
      "epoch": 2.5665283203125e-05,
      "step": 6728,
      "training_step_time": 10.235595464706421
    },
    {
      "epoch": 2.5665283203125e-05,
      "model_forward_time": 3.140043020248413,
      "step": 6728
    },
    {
      "epoch": 2.5665283203125e-05,
      "step": 6728,
      "training_step_time": 9.919220209121704
    },
    {
      "epoch": 2.5665283203125e-05,
      "model_forward_time": 3.137022018432617,
      "step": 6728
    },
    {
      "epoch": 2.5665283203125e-05,
      "step": 6728,
      "training_step_time": 9.908397912979126
    },
    {
      "epoch": 2.5665283203125e-05,
      "model_forward_time": 3.140078544616699,
      "step": 6728
    },
    {
      "epoch": 2.5665283203125e-05,
      "step": 6728,
      "training_step_time": 10.346735715866089
    },
    {
      "epoch": 2.5669097900390625e-05,
      "model_forward_time": 3.182889461517334,
      "step": 6729
    },
    {
      "epoch": 2.5669097900390625e-05,
      "step": 6729,
      "training_step_time": 9.960368156433105
    },
    {
      "epoch": 2.5669097900390625e-05,
      "model_forward_time": 3.417058229446411,
      "step": 6729
    },
    {
      "epoch": 2.5669097900390625e-05,
      "step": 6729,
      "training_step_time": 10.19672155380249
    },
    {
      "epoch": 2.5669097900390625e-05,
      "model_forward_time": 3.142821788787842,
      "step": 6729
    },
    {
      "epoch": 2.5669097900390625e-05,
      "step": 6729,
      "training_step_time": 9.922041654586792
    },
    {
      "epoch": 2.5669097900390625e-05,
      "model_forward_time": 3.1394121646881104,
      "step": 6729
    },
    {
      "epoch": 2.5669097900390625e-05,
      "step": 6729,
      "training_step_time": 10.028157711029053
    },
    {
      "epoch": 2.567291259765625e-05,
      "grad_norm": 0.32143089175224304,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 6730
    },
    {
      "epoch": 2.567291259765625e-05,
      "model_forward_time": 3.175607204437256,
      "step": 6730
    },
    {
      "dynamics_loss_avg": 0.06109988186508417,
      "epoch": 2.567291259765625e-05,
      "step": 6730
    },
    {
      "action_loss_avg": 0.021161020174622534,
      "epoch": 2.567291259765625e-05,
      "step": 6730
    },
    {
      "epoch": 2.567291259765625e-05,
      "step": 6730,
      "training_step_time": 9.960095643997192
    },
    {
      "epoch": 2.567291259765625e-05,
      "model_forward_time": 3.142315149307251,
      "step": 6730
    },
    {
      "epoch": 2.567291259765625e-05,
      "step": 6730,
      "training_step_time": 9.919865608215332
    },
    {
      "epoch": 2.567291259765625e-05,
      "model_forward_time": 3.430757522583008,
      "step": 6730
    },
    {
      "epoch": 2.567291259765625e-05,
      "step": 6730,
      "training_step_time": 10.206291437149048
    },
    {
      "epoch": 2.567291259765625e-05,
      "model_forward_time": 3.1395697593688965,
      "step": 6730
    },
    {
      "epoch": 2.567291259765625e-05,
      "step": 6730,
      "training_step_time": 10.040716409683228
    },
    {
      "epoch": 2.5676727294921875e-05,
      "model_forward_time": 3.1741700172424316,
      "step": 6731
    },
    {
      "epoch": 2.5676727294921875e-05,
      "step": 6731,
      "training_step_time": 9.955852270126343
    },
    {
      "epoch": 2.5676727294921875e-05,
      "model_forward_time": 3.1422626972198486,
      "step": 6731
    },
    {
      "epoch": 2.5676727294921875e-05,
      "step": 6731,
      "training_step_time": 9.920647859573364
    },
    {
      "epoch": 2.5676727294921875e-05,
      "model_forward_time": 3.1369640827178955,
      "step": 6731
    },
    {
      "epoch": 2.5676727294921875e-05,
      "step": 6731,
      "training_step_time": 9.916106700897217
    },
    {
      "epoch": 2.5676727294921875e-05,
      "model_forward_time": 3.4054012298583984,
      "step": 6731
    },
    {
      "epoch": 2.5676727294921875e-05,
      "step": 6731,
      "training_step_time": 10.307218551635742
    },
    {
      "epoch": 2.56805419921875e-05,
      "model_forward_time": 3.1838080883026123,
      "step": 6732
    },
    {
      "epoch": 2.56805419921875e-05,
      "step": 6732,
      "training_step_time": 9.963297605514526
    },
    {
      "epoch": 2.56805419921875e-05,
      "model_forward_time": 3.1379177570343018,
      "step": 6732
    },
    {
      "epoch": 2.56805419921875e-05,
      "step": 6732,
      "training_step_time": 9.915666341781616
    },
    {
      "epoch": 2.56805419921875e-05,
      "model_forward_time": 3.1422371864318848,
      "step": 6732
    },
    {
      "dynamics_loss_avg": 0.05850769225507975,
      "epoch": 2.56805419921875e-05,
      "step": 6732
    },
    {
      "action_loss_avg": 0.017957725375890733,
      "epoch": 2.56805419921875e-05,
      "step": 6732
    },
    {
      "epoch": 2.56805419921875e-05,
      "step": 6732,
      "training_step_time": 9.923705816268921
    },
    {
      "epoch": 2.56805419921875e-05,
      "model_forward_time": 3.140371084213257,
      "step": 6732
    },
    {
      "epoch": 2.56805419921875e-05,
      "step": 6732,
      "training_step_time": 10.274852752685547
    },
    {
      "epoch": 2.5684356689453124e-05,
      "model_forward_time": 3.4224281311035156,
      "step": 6733
    },
    {
      "epoch": 2.5684356689453124e-05,
      "step": 6733,
      "training_step_time": 10.203778505325317
    },
    {
      "epoch": 2.5684356689453124e-05,
      "model_forward_time": 3.1434383392333984,
      "step": 6733
    },
    {
      "epoch": 2.5684356689453124e-05,
      "step": 6733,
      "training_step_time": 9.923277616500854
    },
    {
      "epoch": 2.5684356689453124e-05,
      "model_forward_time": 3.1411139965057373,
      "step": 6733
    },
    {
      "epoch": 2.5684356689453124e-05,
      "step": 6733,
      "training_step_time": 9.917767763137817
    },
    {
      "epoch": 2.5684356689453124e-05,
      "model_forward_time": 2.437249183654785,
      "step": 6733
    },
    {
      "epoch": 2.5684356689453124e-05,
      "step": 6733,
      "training_step_time": 10.308655261993408
    },
    {
      "epoch": 2.568817138671875e-05,
      "model_forward_time": 3.171884059906006,
      "step": 6734
    },
    {
      "epoch": 2.568817138671875e-05,
      "step": 6734,
      "training_step_time": 9.942489862442017
    },
    {
      "epoch": 2.568817138671875e-05,
      "model_forward_time": 3.128948211669922,
      "step": 6734
    },
    {
      "epoch": 2.568817138671875e-05,
      "step": 6734,
      "training_step_time": 10.185776233673096
    },
    {
      "epoch": 2.568817138671875e-05,
      "model_forward_time": 3.133753538131714,
      "step": 6734
    },
    {
      "epoch": 2.568817138671875e-05,
      "step": 6734,
      "training_step_time": 9.907945156097412
    },
    {
      "epoch": 2.568817138671875e-05,
      "model_forward_time": 3.132254123687744,
      "step": 6734
    },
    {
      "epoch": 2.568817138671875e-05,
      "step": 6734,
      "training_step_time": 10.284322500228882
    },
    {
      "epoch": 2.5691986083984376e-05,
      "model_forward_time": 3.1848227977752686,
      "step": 6735
    },
    {
      "dynamics_loss_avg": 0.05539385639131069,
      "epoch": 2.5691986083984376e-05,
      "step": 6735
    },
    {
      "action_loss_avg": 0.023778519872576,
      "epoch": 2.5691986083984376e-05,
      "step": 6735
    },
    {
      "epoch": 2.5691986083984376e-05,
      "step": 6735,
      "training_step_time": 9.959123373031616
    },
    {
      "epoch": 2.5691986083984376e-05,
      "model_forward_time": 3.1338818073272705,
      "step": 6735
    },
    {
      "epoch": 2.5691986083984376e-05,
      "step": 6735,
      "training_step_time": 9.903352499008179
    },
    {
      "epoch": 2.5691986083984376e-05,
      "model_forward_time": 3.429335832595825,
      "step": 6735
    },
    {
      "epoch": 2.5691986083984376e-05,
      "step": 6735,
      "training_step_time": 10.201711416244507
    },
    {
      "epoch": 2.5691986083984376e-05,
      "model_forward_time": 3.1410598754882812,
      "step": 6735
    },
    {
      "epoch": 2.5691986083984376e-05,
      "step": 6735,
      "training_step_time": 10.029918432235718
    },
    {
      "epoch": 2.569580078125e-05,
      "model_forward_time": 3.173832654953003,
      "step": 6736
    },
    {
      "epoch": 2.569580078125e-05,
      "step": 6736,
      "training_step_time": 9.948955059051514
    },
    {
      "epoch": 2.569580078125e-05,
      "model_forward_time": 3.1325888633728027,
      "step": 6736
    },
    {
      "epoch": 2.569580078125e-05,
      "step": 6736,
      "training_step_time": 9.910433292388916
    },
    {
      "epoch": 2.569580078125e-05,
      "model_forward_time": 3.143052577972412,
      "step": 6736
    },
    {
      "epoch": 2.569580078125e-05,
      "step": 6736,
      "training_step_time": 9.923118114471436
    },
    {
      "epoch": 2.569580078125e-05,
      "model_forward_time": 3.1359031200408936,
      "step": 6736
    },
    {
      "epoch": 2.569580078125e-05,
      "step": 6736,
      "training_step_time": 10.29942536354065
    },
    {
      "epoch": 2.5699615478515625e-05,
      "model_forward_time": 3.1758289337158203,
      "step": 6737
    },
    {
      "epoch": 2.5699615478515625e-05,
      "step": 6737,
      "training_step_time": 9.956450462341309
    },
    {
      "epoch": 2.5699615478515625e-05,
      "model_forward_time": 3.145082473754883,
      "step": 6737
    },
    {
      "epoch": 2.5699615478515625e-05,
      "step": 6737,
      "training_step_time": 9.926417112350464
    },
    {
      "epoch": 2.5699615478515625e-05,
      "model_forward_time": 3.141585350036621,
      "step": 6737
    },
    {
      "dynamics_loss_avg": 0.046288214065134524,
      "epoch": 2.5699615478515625e-05,
      "step": 6737
    },
    {
      "action_loss_avg": 0.015072959382086993,
      "epoch": 2.5699615478515625e-05,
      "step": 6737
    },
    {
      "epoch": 2.5699615478515625e-05,
      "step": 6737,
      "training_step_time": 9.922511577606201
    },
    {
      "epoch": 2.5699615478515625e-05,
      "model_forward_time": 3.13920521736145,
      "step": 6737
    },
    {
      "epoch": 2.5699615478515625e-05,
      "step": 6737,
      "training_step_time": 10.31299090385437
    },
    {
      "epoch": 2.570343017578125e-05,
      "model_forward_time": 3.443169116973877,
      "step": 6738
    },
    {
      "epoch": 2.570343017578125e-05,
      "step": 6738,
      "training_step_time": 10.217889785766602
    },
    {
      "epoch": 2.570343017578125e-05,
      "model_forward_time": 3.1309006214141846,
      "step": 6738
    },
    {
      "epoch": 2.570343017578125e-05,
      "step": 6738,
      "training_step_time": 9.904366493225098
    },
    {
      "epoch": 2.570343017578125e-05,
      "model_forward_time": 2.433211326599121,
      "step": 6738
    },
    {
      "epoch": 2.570343017578125e-05,
      "step": 6738,
      "training_step_time": 7.46251368522644
    },
    {
      "epoch": 2.570343017578125e-05,
      "model_forward_time": 3.13049054145813,
      "step": 6738
    },
    {
      "epoch": 2.570343017578125e-05,
      "step": 6738,
      "training_step_time": 10.244486570358276
    },
    {
      "epoch": 2.5707244873046875e-05,
      "model_forward_time": 3.189293384552002,
      "step": 6739
    },
    {
      "epoch": 2.5707244873046875e-05,
      "step": 6739,
      "training_step_time": 9.960866212844849
    },
    {
      "epoch": 2.5707244873046875e-05,
      "model_forward_time": 3.1387298107147217,
      "step": 6739
    },
    {
      "epoch": 2.5707244873046875e-05,
      "step": 6739,
      "training_step_time": 9.912230491638184
    },
    {
      "epoch": 2.5707244873046875e-05,
      "model_forward_time": 3.401205062866211,
      "step": 6739
    },
    {
      "epoch": 2.5707244873046875e-05,
      "step": 6739,
      "training_step_time": 10.1735360622406
    },
    {
      "epoch": 2.5707244873046875e-05,
      "model_forward_time": 3.1375279426574707,
      "step": 6739
    },
    {
      "epoch": 2.5707244873046875e-05,
      "step": 6739,
      "training_step_time": 10.339603185653687
    },
    {
      "epoch": 2.57110595703125e-05,
      "grad_norm": 0.386509507894516,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 6740
    },
    {
      "epoch": 2.57110595703125e-05,
      "model_forward_time": 3.145287275314331,
      "step": 6740
    },
    {
      "dynamics_loss_avg": 0.060974205285310744,
      "epoch": 2.57110595703125e-05,
      "step": 6740
    },
    {
      "action_loss_avg": 0.02303065713495016,
      "epoch": 2.57110595703125e-05,
      "step": 6740
    },
    {
      "epoch": 2.57110595703125e-05,
      "step": 6740,
      "training_step_time": 9.927887916564941
    },
    {
      "epoch": 2.57110595703125e-05,
      "model_forward_time": 3.1370205879211426,
      "step": 6740
    },
    {
      "epoch": 2.57110595703125e-05,
      "step": 6740,
      "training_step_time": 9.913154363632202
    },
    {
      "epoch": 2.57110595703125e-05,
      "model_forward_time": 2.7195963859558105,
      "step": 6740
    },
    {
      "epoch": 2.57110595703125e-05,
      "step": 6740,
      "training_step_time": 7.745033025741577
    },
    {
      "epoch": 2.57110595703125e-05,
      "model_forward_time": 3.1309852600097656,
      "step": 6740
    },
    {
      "epoch": 2.57110595703125e-05,
      "step": 6740,
      "training_step_time": 10.018524169921875
    },
    {
      "epoch": 2.5714874267578124e-05,
      "model_forward_time": 3.149799108505249,
      "step": 6741
    },
    {
      "epoch": 2.5714874267578124e-05,
      "step": 6741,
      "training_step_time": 9.923291683197021
    },
    {
      "epoch": 2.5714874267578124e-05,
      "model_forward_time": 3.136159658432007,
      "step": 6741
    },
    {
      "epoch": 2.5714874267578124e-05,
      "step": 6741,
      "training_step_time": 9.911609172821045
    },
    {
      "epoch": 2.5714874267578124e-05,
      "model_forward_time": 3.133962392807007,
      "step": 6741
    },
    {
      "epoch": 2.5714874267578124e-05,
      "step": 6741,
      "training_step_time": 9.907964706420898
    },
    {
      "epoch": 2.5714874267578124e-05,
      "model_forward_time": 2.437243938446045,
      "step": 6741
    },
    {
      "epoch": 2.5714874267578124e-05,
      "step": 6741,
      "training_step_time": 10.279991388320923
    },
    {
      "epoch": 2.571868896484375e-05,
      "model_forward_time": 3.477123737335205,
      "step": 6742
    },
    {
      "epoch": 2.571868896484375e-05,
      "step": 6742,
      "training_step_time": 10.253750801086426
    },
    {
      "epoch": 2.571868896484375e-05,
      "model_forward_time": 3.1478254795074463,
      "step": 6742
    },
    {
      "epoch": 2.571868896484375e-05,
      "step": 6742,
      "training_step_time": 9.927695512771606
    },
    {
      "epoch": 2.571868896484375e-05,
      "model_forward_time": 3.1378350257873535,
      "step": 6742
    },
    {
      "dynamics_loss_avg": 0.05862460490316153,
      "epoch": 2.571868896484375e-05,
      "step": 6742
    },
    {
      "action_loss_avg": 0.018163517140783368,
      "epoch": 2.571868896484375e-05,
      "step": 6742
    },
    {
      "epoch": 2.571868896484375e-05,
      "step": 6742,
      "training_step_time": 9.917897701263428
    },
    {
      "epoch": 2.571868896484375e-05,
      "model_forward_time": 3.137340784072876,
      "step": 6742
    },
    {
      "epoch": 2.571868896484375e-05,
      "step": 6742,
      "training_step_time": 10.315011024475098
    },
    {
      "epoch": 2.5722503662109376e-05,
      "model_forward_time": 3.505648374557495,
      "step": 6743
    },
    {
      "epoch": 2.5722503662109376e-05,
      "step": 6743,
      "training_step_time": 10.283687114715576
    },
    {
      "epoch": 2.5722503662109376e-05,
      "model_forward_time": 3.1388962268829346,
      "step": 6743
    },
    {
      "epoch": 2.5722503662109376e-05,
      "step": 6743,
      "training_step_time": 9.911575555801392
    },
    {
      "epoch": 2.5722503662109376e-05,
      "model_forward_time": 3.132248640060425,
      "step": 6743
    },
    {
      "epoch": 2.5722503662109376e-05,
      "step": 6743,
      "training_step_time": 9.905632972717285
    },
    {
      "epoch": 2.5722503662109376e-05,
      "model_forward_time": 3.137091636657715,
      "step": 6743
    },
    {
      "epoch": 2.5722503662109376e-05,
      "step": 6743,
      "training_step_time": 10.262198448181152
    },
    {
      "epoch": 2.5726318359375e-05,
      "model_forward_time": 3.1911797523498535,
      "step": 6744
    },
    {
      "epoch": 2.5726318359375e-05,
      "step": 6744,
      "training_step_time": 9.971935987472534
    },
    {
      "epoch": 2.5726318359375e-05,
      "model_forward_time": 3.1546361446380615,
      "step": 6744
    },
    {
      "epoch": 2.5726318359375e-05,
      "step": 6744,
      "training_step_time": 10.249672651290894
    },
    {
      "epoch": 2.5726318359375e-05,
      "model_forward_time": 3.147341012954712,
      "step": 6744
    },
    {
      "epoch": 2.5726318359375e-05,
      "step": 6744,
      "training_step_time": 9.930461168289185
    },
    {
      "epoch": 2.5726318359375e-05,
      "model_forward_time": 3.1492698192596436,
      "step": 6744
    },
    {
      "epoch": 2.5726318359375e-05,
      "step": 6744,
      "training_step_time": 10.334475755691528
    },
    {
      "epoch": 2.5730133056640625e-05,
      "model_forward_time": 3.1885530948638916,
      "step": 6745
    },
    {
      "dynamics_loss_avg": 0.07360202632844448,
      "epoch": 2.5730133056640625e-05,
      "step": 6745
    },
    {
      "action_loss_avg": 0.025631351815536618,
      "epoch": 2.5730133056640625e-05,
      "step": 6745
    },
    {
      "epoch": 2.5730133056640625e-05,
      "step": 6745,
      "training_step_time": 9.97066593170166
    },
    {
      "epoch": 2.5730133056640625e-05,
      "model_forward_time": 3.1509082317352295,
      "step": 6745
    },
    {
      "epoch": 2.5730133056640625e-05,
      "step": 6745,
      "training_step_time": 9.929869890213013
    },
    {
      "epoch": 2.5730133056640625e-05,
      "model_forward_time": 3.149510622024536,
      "step": 6745
    },
    {
      "epoch": 2.5730133056640625e-05,
      "step": 6745,
      "training_step_time": 10.220536708831787
    },
    {
      "epoch": 2.5730133056640625e-05,
      "model_forward_time": 3.1474804878234863,
      "step": 6745
    },
    {
      "epoch": 2.5730133056640625e-05,
      "step": 6745,
      "training_step_time": 10.296998739242554
    },
    {
      "epoch": 2.573394775390625e-05,
      "model_forward_time": 3.19578218460083,
      "step": 6746
    },
    {
      "epoch": 2.573394775390625e-05,
      "step": 6746,
      "training_step_time": 9.972884893417358
    },
    {
      "epoch": 2.573394775390625e-05,
      "model_forward_time": 3.143759250640869,
      "step": 6746
    },
    {
      "epoch": 2.573394775390625e-05,
      "step": 6746,
      "training_step_time": 9.918440580368042
    },
    {
      "epoch": 2.573394775390625e-05,
      "model_forward_time": 3.146073341369629,
      "step": 6746
    },
    {
      "epoch": 2.573394775390625e-05,
      "step": 6746,
      "training_step_time": 9.921512365341187
    },
    {
      "epoch": 2.573394775390625e-05,
      "model_forward_time": 3.431469202041626,
      "step": 6746
    },
    {
      "epoch": 2.573394775390625e-05,
      "step": 6746,
      "training_step_time": 10.300573110580444
    },
    {
      "epoch": 2.5737762451171875e-05,
      "model_forward_time": 3.180288076400757,
      "step": 6747
    },
    {
      "epoch": 2.5737762451171875e-05,
      "step": 6747,
      "training_step_time": 9.961647272109985
    },
    {
      "epoch": 2.5737762451171875e-05,
      "model_forward_time": 3.1399900913238525,
      "step": 6747
    },
    {
      "epoch": 2.5737762451171875e-05,
      "step": 6747,
      "training_step_time": 9.917684078216553
    },
    {
      "epoch": 2.5737762451171875e-05,
      "model_forward_time": 3.135373830795288,
      "step": 6747
    },
    {
      "dynamics_loss_avg": 0.054809030052274464,
      "epoch": 2.5737762451171875e-05,
      "step": 6747
    },
    {
      "action_loss_avg": 0.01549087641760707,
      "epoch": 2.5737762451171875e-05,
      "step": 6747
    },
    {
      "epoch": 2.5737762451171875e-05,
      "step": 6747,
      "training_step_time": 9.902744054794312
    },
    {
      "epoch": 2.5737762451171875e-05,
      "model_forward_time": 3.133714437484741,
      "step": 6747
    },
    {
      "epoch": 2.5737762451171875e-05,
      "step": 6747,
      "training_step_time": 10.017886877059937
    },
    {
      "epoch": 2.57415771484375e-05,
      "model_forward_time": 3.1833276748657227,
      "step": 6748
    },
    {
      "epoch": 2.57415771484375e-05,
      "step": 6748,
      "training_step_time": 10.260198831558228
    },
    {
      "epoch": 2.57415771484375e-05,
      "model_forward_time": 3.136284589767456,
      "step": 6748
    },
    {
      "epoch": 2.57415771484375e-05,
      "step": 6748,
      "training_step_time": 9.90503454208374
    },
    {
      "epoch": 2.57415771484375e-05,
      "model_forward_time": 3.1431221961975098,
      "step": 6748
    },
    {
      "epoch": 2.57415771484375e-05,
      "step": 6748,
      "training_step_time": 9.915177822113037
    },
    {
      "epoch": 2.57415771484375e-05,
      "model_forward_time": 3.1393473148345947,
      "step": 6748
    },
    {
      "epoch": 2.57415771484375e-05,
      "step": 6748,
      "training_step_time": 10.273240566253662
    },
    {
      "epoch": 2.5745391845703124e-05,
      "model_forward_time": 3.180711269378662,
      "step": 6749
    },
    {
      "epoch": 2.5745391845703124e-05,
      "step": 6749,
      "training_step_time": 9.951596975326538
    },
    {
      "epoch": 2.5745391845703124e-05,
      "model_forward_time": 3.4114394187927246,
      "step": 6749
    },
    {
      "epoch": 2.5745391845703124e-05,
      "step": 6749,
      "training_step_time": 10.180680274963379
    },
    {
      "epoch": 2.5745391845703124e-05,
      "model_forward_time": 3.14375638961792,
      "step": 6749
    },
    {
      "epoch": 2.5745391845703124e-05,
      "step": 6749,
      "training_step_time": 9.91184663772583
    },
    {
      "epoch": 2.5745391845703124e-05,
      "model_forward_time": 3.13944411277771,
      "step": 6749
    },
    {
      "epoch": 2.5745391845703124e-05,
      "step": 6749,
      "training_step_time": 10.020088911056519
    },
    {
      "epoch": 2.574920654296875e-05,
      "grad_norm": 0.4227989912033081,
      "learning_rate": 1e-05,
      "loss": 0.0862,
      "step": 6750
    },
    {
      "epoch": 2.574920654296875e-05,
      "model_forward_time": 3.167524814605713,
      "step": 6750
    },
    {
      "dynamics_loss_avg": 0.05649773683398962,
      "epoch": 2.574920654296875e-05,
      "step": 6750
    },
    {
      "action_loss_avg": 0.02378671579062939,
      "epoch": 2.574920654296875e-05,
      "step": 6750
    },
    {
      "epoch": 2.574920654296875e-05,
      "step": 6750,
      "training_step_time": 9.93947172164917
    },
    {
      "epoch": 2.574920654296875e-05,
      "model_forward_time": 3.135631799697876,
      "step": 6750
    },
    {
      "epoch": 2.574920654296875e-05,
      "step": 6750,
      "training_step_time": 9.90446138381958
    },
    {
      "epoch": 2.574920654296875e-05,
      "model_forward_time": 3.41733980178833,
      "step": 6750
    },
    {
      "epoch": 2.574920654296875e-05,
      "step": 6750,
      "training_step_time": 10.186963558197021
    },
    {
      "epoch": 2.574920654296875e-05,
      "model_forward_time": 3.1350228786468506,
      "step": 6750
    },
    {
      "epoch": 2.574920654296875e-05,
      "step": 6750,
      "training_step_time": 10.254696369171143
    },
    {
      "epoch": 2.5753021240234376e-05,
      "model_forward_time": 3.166362762451172,
      "step": 6751
    },
    {
      "epoch": 2.5753021240234376e-05,
      "step": 6751,
      "training_step_time": 9.934352159500122
    },
    {
      "epoch": 2.5753021240234376e-05,
      "model_forward_time": 3.138270139694214,
      "step": 6751
    },
    {
      "epoch": 2.5753021240234376e-05,
      "step": 6751,
      "training_step_time": 9.913519620895386
    },
    {
      "epoch": 2.5753021240234376e-05,
      "model_forward_time": 3.1363332271575928,
      "step": 6751
    },
    {
      "epoch": 2.5753021240234376e-05,
      "step": 6751,
      "training_step_time": 9.907546281814575
    },
    {
      "epoch": 2.5753021240234376e-05,
      "model_forward_time": 3.3921761512756348,
      "step": 6751
    },
    {
      "epoch": 2.5753021240234376e-05,
      "step": 6751,
      "training_step_time": 10.257085800170898
    },
    {
      "epoch": 2.57568359375e-05,
      "model_forward_time": 3.1830286979675293,
      "step": 6752
    },
    {
      "epoch": 2.57568359375e-05,
      "step": 6752,
      "training_step_time": 9.95777940750122
    },
    {
      "epoch": 2.57568359375e-05,
      "model_forward_time": 3.139702796936035,
      "step": 6752
    },
    {
      "epoch": 2.57568359375e-05,
      "step": 6752,
      "training_step_time": 9.913043737411499
    },
    {
      "epoch": 2.57568359375e-05,
      "model_forward_time": 3.1385538578033447,
      "step": 6752
    },
    {
      "dynamics_loss_avg": 0.056893250159919265,
      "epoch": 2.57568359375e-05,
      "step": 6752
    },
    {
      "action_loss_avg": 0.02089755153283477,
      "epoch": 2.57568359375e-05,
      "step": 6752
    },
    {
      "epoch": 2.57568359375e-05,
      "step": 6752,
      "training_step_time": 9.918848037719727
    },
    {
      "epoch": 2.57568359375e-05,
      "model_forward_time": 3.1417322158813477,
      "step": 6752
    },
    {
      "epoch": 2.57568359375e-05,
      "step": 6752,
      "training_step_time": 10.299409866333008
    },
    {
      "epoch": 2.5760650634765625e-05,
      "model_forward_time": 3.434462070465088,
      "step": 6753
    },
    {
      "epoch": 2.5760650634765625e-05,
      "step": 6753,
      "training_step_time": 10.214683055877686
    },
    {
      "epoch": 2.5760650634765625e-05,
      "model_forward_time": 3.142585277557373,
      "step": 6753
    },
    {
      "epoch": 2.5760650634765625e-05,
      "step": 6753,
      "training_step_time": 9.918228387832642
    },
    {
      "epoch": 2.5760650634765625e-05,
      "model_forward_time": 3.132713556289673,
      "step": 6753
    },
    {
      "epoch": 2.5760650634765625e-05,
      "step": 6753,
      "training_step_time": 9.90592336654663
    },
    {
      "epoch": 2.5760650634765625e-05,
      "model_forward_time": 3.139167547225952,
      "step": 6753
    },
    {
      "epoch": 2.5760650634765625e-05,
      "step": 6753,
      "training_step_time": 10.294284343719482
    },
    {
      "epoch": 2.576446533203125e-05,
      "model_forward_time": 3.168147325515747,
      "step": 6754
    },
    {
      "epoch": 2.576446533203125e-05,
      "step": 6754,
      "training_step_time": 9.942172527313232
    },
    {
      "epoch": 2.576446533203125e-05,
      "model_forward_time": 3.1359472274780273,
      "step": 6754
    },
    {
      "epoch": 2.576446533203125e-05,
      "step": 6754,
      "training_step_time": 10.189186096191406
    },
    {
      "epoch": 2.576446533203125e-05,
      "model_forward_time": 3.1340532302856445,
      "step": 6754
    },
    {
      "epoch": 2.576446533203125e-05,
      "step": 6754,
      "training_step_time": 9.90365982055664
    },
    {
      "epoch": 2.576446533203125e-05,
      "model_forward_time": 3.136296272277832,
      "step": 6754
    },
    {
      "epoch": 2.576446533203125e-05,
      "step": 6754,
      "training_step_time": 10.009688138961792
    },
    {
      "epoch": 2.5768280029296875e-05,
      "model_forward_time": 3.1492531299591064,
      "step": 6755
    },
    {
      "dynamics_loss_avg": 0.04733720403164625,
      "epoch": 2.5768280029296875e-05,
      "step": 6755
    },
    {
      "action_loss_avg": 0.014481564611196518,
      "epoch": 2.5768280029296875e-05,
      "step": 6755
    },
    {
      "epoch": 2.5768280029296875e-05,
      "step": 6755,
      "training_step_time": 9.925238370895386
    },
    {
      "epoch": 2.5768280029296875e-05,
      "model_forward_time": 3.137657642364502,
      "step": 6755
    },
    {
      "epoch": 2.5768280029296875e-05,
      "step": 6755,
      "training_step_time": 9.912375926971436
    },
    {
      "epoch": 2.5768280029296875e-05,
      "model_forward_time": 3.4476194381713867,
      "step": 6755
    },
    {
      "epoch": 2.5768280029296875e-05,
      "step": 6755,
      "training_step_time": 10.222246885299683
    },
    {
      "epoch": 2.5768280029296875e-05,
      "model_forward_time": 3.1422300338745117,
      "step": 6755
    },
    {
      "epoch": 2.5768280029296875e-05,
      "step": 6755,
      "training_step_time": 10.285752058029175
    },
    {
      "epoch": 2.57720947265625e-05,
      "model_forward_time": 3.188504457473755,
      "step": 6756
    },
    {
      "epoch": 2.57720947265625e-05,
      "step": 6756,
      "training_step_time": 9.961667776107788
    },
    {
      "epoch": 2.57720947265625e-05,
      "model_forward_time": 3.137450933456421,
      "step": 6756
    },
    {
      "epoch": 2.57720947265625e-05,
      "step": 6756,
      "training_step_time": 9.903263330459595
    },
    {
      "epoch": 2.57720947265625e-05,
      "model_forward_time": 3.1370151042938232,
      "step": 6756
    },
    {
      "epoch": 2.57720947265625e-05,
      "step": 6756,
      "training_step_time": 9.905438899993896
    },
    {
      "epoch": 2.57720947265625e-05,
      "model_forward_time": 3.1294455528259277,
      "step": 6756
    },
    {
      "epoch": 2.57720947265625e-05,
      "step": 6756,
      "training_step_time": 10.266930341720581
    },
    {
      "epoch": 2.5775909423828124e-05,
      "model_forward_time": 3.1799802780151367,
      "step": 6757
    },
    {
      "epoch": 2.5775909423828124e-05,
      "step": 6757,
      "training_step_time": 9.956541538238525
    },
    {
      "epoch": 2.5775909423828124e-05,
      "model_forward_time": 3.1329991817474365,
      "step": 6757
    },
    {
      "epoch": 2.5775909423828124e-05,
      "step": 6757,
      "training_step_time": 9.901737451553345
    },
    {
      "epoch": 2.5775909423828124e-05,
      "model_forward_time": 3.1340067386627197,
      "step": 6757
    },
    {
      "dynamics_loss_avg": 0.06868806593120098,
      "epoch": 2.5775909423828124e-05,
      "step": 6757
    },
    {
      "action_loss_avg": 0.03007965302094817,
      "epoch": 2.5775909423828124e-05,
      "step": 6757
    },
    {
      "epoch": 2.5775909423828124e-05,
      "step": 6757,
      "training_step_time": 9.902453660964966
    },
    {
      "epoch": 2.5775909423828124e-05,
      "model_forward_time": 3.1327314376831055,
      "step": 6757
    },
    {
      "epoch": 2.5775909423828124e-05,
      "step": 6757,
      "training_step_time": 10.265522241592407
    },
    {
      "epoch": 2.577972412109375e-05,
      "model_forward_time": 3.4541070461273193,
      "step": 6758
    },
    {
      "epoch": 2.577972412109375e-05,
      "step": 6758,
      "training_step_time": 10.218628406524658
    },
    {
      "epoch": 2.577972412109375e-05,
      "model_forward_time": 3.132244348526001,
      "step": 6758
    },
    {
      "epoch": 2.577972412109375e-05,
      "step": 6758,
      "training_step_time": 9.90109634399414
    },
    {
      "epoch": 2.577972412109375e-05,
      "model_forward_time": 3.1370532512664795,
      "step": 6758
    },
    {
      "epoch": 2.577972412109375e-05,
      "step": 6758,
      "training_step_time": 9.904263734817505
    },
    {
      "epoch": 2.577972412109375e-05,
      "model_forward_time": 3.1358630657196045,
      "step": 6758
    },
    {
      "epoch": 2.577972412109375e-05,
      "step": 6758,
      "training_step_time": 10.029716968536377
    },
    {
      "epoch": 2.5783538818359376e-05,
      "model_forward_time": 3.167283058166504,
      "step": 6759
    },
    {
      "epoch": 2.5783538818359376e-05,
      "step": 6759,
      "training_step_time": 9.94253921508789
    },
    {
      "epoch": 2.5783538818359376e-05,
      "model_forward_time": 3.1393818855285645,
      "step": 6759
    },
    {
      "epoch": 2.5783538818359376e-05,
      "step": 6759,
      "training_step_time": 10.176477432250977
    },
    {
      "epoch": 2.5783538818359376e-05,
      "model_forward_time": 3.140775680541992,
      "step": 6759
    },
    {
      "epoch": 2.5783538818359376e-05,
      "step": 6759,
      "training_step_time": 9.914573431015015
    },
    {
      "epoch": 2.5783538818359376e-05,
      "model_forward_time": 3.13716197013855,
      "step": 6759
    },
    {
      "epoch": 2.5783538818359376e-05,
      "step": 6759,
      "training_step_time": 10.306563138961792
    },
    {
      "epoch": 2.5787353515625e-05,
      "grad_norm": 0.4335194528102875,
      "learning_rate": 1e-05,
      "loss": 0.0879,
      "step": 6760
    },
    {
      "epoch": 2.5787353515625e-05,
      "model_forward_time": 3.151705503463745,
      "step": 6760
    },
    {
      "dynamics_loss_avg": 0.06062554381787777,
      "epoch": 2.5787353515625e-05,
      "step": 6760
    },
    {
      "action_loss_avg": 0.025732395146042107,
      "epoch": 2.5787353515625e-05,
      "step": 6760
    },
    {
      "epoch": 2.5787353515625e-05,
      "step": 6760,
      "training_step_time": 9.928325891494751
    },
    {
      "epoch": 2.5787353515625e-05,
      "model_forward_time": 3.1425070762634277,
      "step": 6760
    },
    {
      "epoch": 2.5787353515625e-05,
      "step": 6760,
      "training_step_time": 9.913246154785156
    },
    {
      "epoch": 2.5787353515625e-05,
      "model_forward_time": 3.4144177436828613,
      "step": 6760
    },
    {
      "epoch": 2.5787353515625e-05,
      "step": 6760,
      "training_step_time": 10.186578750610352
    },
    {
      "epoch": 2.5787353515625e-05,
      "model_forward_time": 3.137259006500244,
      "step": 6760
    },
    {
      "epoch": 2.5787353515625e-05,
      "step": 6760,
      "training_step_time": 10.283542394638062
    },
    {
      "epoch": 2.5791168212890625e-05,
      "model_forward_time": 3.1714465618133545,
      "step": 6761
    },
    {
      "epoch": 2.5791168212890625e-05,
      "step": 6761,
      "training_step_time": 9.948180675506592
    },
    {
      "epoch": 2.5791168212890625e-05,
      "model_forward_time": 3.135416269302368,
      "step": 6761
    },
    {
      "epoch": 2.5791168212890625e-05,
      "step": 6761,
      "training_step_time": 9.909345626831055
    },
    {
      "epoch": 2.5791168212890625e-05,
      "model_forward_time": 3.1360316276550293,
      "step": 6761
    },
    {
      "epoch": 2.5791168212890625e-05,
      "step": 6761,
      "training_step_time": 9.903286218643188
    },
    {
      "epoch": 2.5791168212890625e-05,
      "model_forward_time": 3.3948819637298584,
      "step": 6761
    },
    {
      "epoch": 2.5791168212890625e-05,
      "step": 6761,
      "training_step_time": 10.32086181640625
    },
    {
      "epoch": 2.579498291015625e-05,
      "model_forward_time": 3.1763854026794434,
      "step": 6762
    },
    {
      "epoch": 2.579498291015625e-05,
      "step": 6762,
      "training_step_time": 9.942387104034424
    },
    {
      "epoch": 2.579498291015625e-05,
      "model_forward_time": 3.1354334354400635,
      "step": 6762
    },
    {
      "epoch": 2.579498291015625e-05,
      "step": 6762,
      "training_step_time": 9.901310920715332
    },
    {
      "epoch": 2.579498291015625e-05,
      "model_forward_time": 3.1353511810302734,
      "step": 6762
    },
    {
      "dynamics_loss_avg": 0.0785881880670786,
      "epoch": 2.579498291015625e-05,
      "step": 6762
    },
    {
      "action_loss_avg": 0.02847735050600022,
      "epoch": 2.579498291015625e-05,
      "step": 6762
    },
    {
      "epoch": 2.579498291015625e-05,
      "step": 6762,
      "training_step_time": 9.91545057296753
    },
    {
      "epoch": 2.579498291015625e-05,
      "model_forward_time": 3.139914035797119,
      "step": 6762
    },
    {
      "epoch": 2.579498291015625e-05,
      "step": 6762,
      "training_step_time": 10.338218927383423
    },
    {
      "epoch": 2.5798797607421875e-05,
      "model_forward_time": 3.4325296878814697,
      "step": 6763
    },
    {
      "epoch": 2.5798797607421875e-05,
      "step": 6763,
      "training_step_time": 10.213733434677124
    },
    {
      "epoch": 2.5798797607421875e-05,
      "model_forward_time": 3.143092393875122,
      "step": 6763
    },
    {
      "epoch": 2.5798797607421875e-05,
      "step": 6763,
      "training_step_time": 9.92506456375122
    },
    {
      "epoch": 2.5798797607421875e-05,
      "model_forward_time": 3.1423113346099854,
      "step": 6763
    },
    {
      "epoch": 2.5798797607421875e-05,
      "step": 6763,
      "training_step_time": 9.925545930862427
    },
    {
      "epoch": 2.5798797607421875e-05,
      "model_forward_time": 3.1467444896698,
      "step": 6763
    },
    {
      "epoch": 2.5798797607421875e-05,
      "step": 6763,
      "training_step_time": 10.341888427734375
    },
    {
      "epoch": 2.58026123046875e-05,
      "model_forward_time": 3.186779737472534,
      "step": 6764
    },
    {
      "epoch": 2.58026123046875e-05,
      "step": 6764,
      "training_step_time": 9.971120119094849
    },
    {
      "epoch": 2.58026123046875e-05,
      "model_forward_time": 3.452815294265747,
      "step": 6764
    },
    {
      "epoch": 2.58026123046875e-05,
      "step": 6764,
      "training_step_time": 10.23320460319519
    },
    {
      "epoch": 2.58026123046875e-05,
      "model_forward_time": 3.1420111656188965,
      "step": 6764
    },
    {
      "epoch": 2.58026123046875e-05,
      "step": 6764,
      "training_step_time": 9.920017957687378
    },
    {
      "epoch": 2.58026123046875e-05,
      "model_forward_time": 3.1522154808044434,
      "step": 6764
    },
    {
      "epoch": 2.58026123046875e-05,
      "step": 6764,
      "training_step_time": 10.255064010620117
    },
    {
      "epoch": 2.5806427001953124e-05,
      "model_forward_time": 3.170854091644287,
      "step": 6765
    },
    {
      "dynamics_loss_avg": 0.04576550330966711,
      "epoch": 2.5806427001953124e-05,
      "step": 6765
    },
    {
      "action_loss_avg": 0.019908425165340304,
      "epoch": 2.5806427001953124e-05,
      "step": 6765
    },
    {
      "epoch": 2.5806427001953124e-05,
      "step": 6765,
      "training_step_time": 9.95702075958252
    },
    {
      "epoch": 2.5806427001953124e-05,
      "model_forward_time": 3.138322591781616,
      "step": 6765
    },
    {
      "epoch": 2.5806427001953124e-05,
      "step": 6765,
      "training_step_time": 9.921528339385986
    },
    {
      "epoch": 2.5806427001953124e-05,
      "model_forward_time": 3.1465890407562256,
      "step": 6765
    },
    {
      "epoch": 2.5806427001953124e-05,
      "step": 6765,
      "training_step_time": 10.218714952468872
    },
    {
      "epoch": 2.5806427001953124e-05,
      "model_forward_time": 3.1490185260772705,
      "step": 6765
    },
    {
      "epoch": 2.5806427001953124e-05,
      "step": 6765,
      "training_step_time": 10.020325660705566
    },
    {
      "epoch": 2.581024169921875e-05,
      "model_forward_time": 3.1831345558166504,
      "step": 6766
    },
    {
      "epoch": 2.581024169921875e-05,
      "step": 6766,
      "training_step_time": 9.960421085357666
    },
    {
      "epoch": 2.581024169921875e-05,
      "model_forward_time": 3.1437346935272217,
      "step": 6766
    },
    {
      "epoch": 2.581024169921875e-05,
      "step": 6766,
      "training_step_time": 9.920670986175537
    },
    {
      "epoch": 2.581024169921875e-05,
      "model_forward_time": 3.143611431121826,
      "step": 6766
    },
    {
      "epoch": 2.581024169921875e-05,
      "step": 6766,
      "training_step_time": 9.920535326004028
    },
    {
      "epoch": 2.581024169921875e-05,
      "model_forward_time": 3.4351351261138916,
      "step": 6766
    },
    {
      "epoch": 2.581024169921875e-05,
      "step": 6766,
      "training_step_time": 10.30677604675293
    },
    {
      "epoch": 2.5814056396484376e-05,
      "model_forward_time": 3.159032106399536,
      "step": 6767
    },
    {
      "epoch": 2.5814056396484376e-05,
      "step": 6767,
      "training_step_time": 9.943885326385498
    },
    {
      "epoch": 2.5814056396484376e-05,
      "model_forward_time": 3.1430790424346924,
      "step": 6767
    },
    {
      "epoch": 2.5814056396484376e-05,
      "step": 6767,
      "training_step_time": 9.927053928375244
    },
    {
      "epoch": 2.5814056396484376e-05,
      "model_forward_time": 2.443631410598755,
      "step": 6767
    },
    {
      "dynamics_loss_avg": 0.06913938634097576,
      "epoch": 2.5814056396484376e-05,
      "step": 6767
    },
    {
      "action_loss_avg": 0.02124291192740202,
      "epoch": 2.5814056396484376e-05,
      "step": 6767
    },
    {
      "epoch": 2.5814056396484376e-05,
      "step": 6767,
      "training_step_time": 7.47968053817749
    },
    {
      "epoch": 2.5814056396484376e-05,
      "model_forward_time": 3.139233350753784,
      "step": 6767
    },
    {
      "epoch": 2.5814056396484376e-05,
      "step": 6767,
      "training_step_time": 10.020784139633179
    },
    {
      "epoch": 2.581787109375e-05,
      "model_forward_time": 3.1616287231445312,
      "step": 6768
    },
    {
      "epoch": 2.581787109375e-05,
      "step": 6768,
      "training_step_time": 9.941840171813965
    },
    {
      "epoch": 2.581787109375e-05,
      "model_forward_time": 3.392132520675659,
      "step": 6768
    },
    {
      "epoch": 2.581787109375e-05,
      "step": 6768,
      "training_step_time": 10.171374559402466
    },
    {
      "epoch": 2.581787109375e-05,
      "model_forward_time": 3.1453616619110107,
      "step": 6768
    },
    {
      "epoch": 2.581787109375e-05,
      "step": 6768,
      "training_step_time": 9.93102502822876
    },
    {
      "epoch": 2.581787109375e-05,
      "model_forward_time": 3.1503210067749023,
      "step": 6768
    },
    {
      "epoch": 2.581787109375e-05,
      "step": 6768,
      "training_step_time": 10.274919033050537
    },
    {
      "epoch": 2.5821685791015625e-05,
      "model_forward_time": 3.182222843170166,
      "step": 6769
    },
    {
      "epoch": 2.5821685791015625e-05,
      "step": 6769,
      "training_step_time": 9.96566367149353
    },
    {
      "epoch": 2.5821685791015625e-05,
      "model_forward_time": 3.407437801361084,
      "step": 6769
    },
    {
      "epoch": 2.5821685791015625e-05,
      "step": 6769,
      "training_step_time": 10.190593481063843
    },
    {
      "epoch": 2.5821685791015625e-05,
      "model_forward_time": 3.142033338546753,
      "step": 6769
    },
    {
      "epoch": 2.5821685791015625e-05,
      "step": 6769,
      "training_step_time": 9.923787355422974
    },
    {
      "epoch": 2.5821685791015625e-05,
      "model_forward_time": 3.1370019912719727,
      "step": 6769
    },
    {
      "epoch": 2.5821685791015625e-05,
      "step": 6769,
      "training_step_time": 10.274214744567871
    },
    {
      "epoch": 2.582550048828125e-05,
      "grad_norm": 0.5255469679832458,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 6770
    },
    {
      "epoch": 2.582550048828125e-05,
      "model_forward_time": 3.1697418689727783,
      "step": 6770
    },
    {
      "dynamics_loss_avg": 0.07006401270627975,
      "epoch": 2.582550048828125e-05,
      "step": 6770
    },
    {
      "action_loss_avg": 0.024018235504627228,
      "epoch": 2.582550048828125e-05,
      "step": 6770
    },
    {
      "epoch": 2.582550048828125e-05,
      "step": 6770,
      "training_step_time": 9.952661037445068
    },
    {
      "epoch": 2.582550048828125e-05,
      "model_forward_time": 3.140545129776001,
      "step": 6770
    },
    {
      "epoch": 2.582550048828125e-05,
      "step": 6770,
      "training_step_time": 9.920106410980225
    },
    {
      "epoch": 2.582550048828125e-05,
      "model_forward_time": 3.136002779006958,
      "step": 6770
    },
    {
      "epoch": 2.582550048828125e-05,
      "step": 6770,
      "training_step_time": 10.182409763336182
    },
    {
      "epoch": 2.582550048828125e-05,
      "model_forward_time": 3.1340575218200684,
      "step": 6770
    },
    {
      "epoch": 2.582550048828125e-05,
      "step": 6770,
      "training_step_time": 10.011330604553223
    },
    {
      "epoch": 2.5829315185546875e-05,
      "model_forward_time": 3.1799073219299316,
      "step": 6771
    },
    {
      "epoch": 2.5829315185546875e-05,
      "step": 6771,
      "training_step_time": 9.961610794067383
    },
    {
      "epoch": 2.5829315185546875e-05,
      "model_forward_time": 3.1397507190704346,
      "step": 6771
    },
    {
      "epoch": 2.5829315185546875e-05,
      "step": 6771,
      "training_step_time": 9.917473793029785
    },
    {
      "epoch": 2.5829315185546875e-05,
      "model_forward_time": 3.136472225189209,
      "step": 6771
    },
    {
      "epoch": 2.5829315185546875e-05,
      "step": 6771,
      "training_step_time": 9.915141820907593
    },
    {
      "epoch": 2.5829315185546875e-05,
      "model_forward_time": 3.4038338661193848,
      "step": 6771
    },
    {
      "epoch": 2.5829315185546875e-05,
      "step": 6771,
      "training_step_time": 10.280283451080322
    },
    {
      "epoch": 2.58331298828125e-05,
      "model_forward_time": 2.4523088932037354,
      "step": 6772
    },
    {
      "epoch": 2.58331298828125e-05,
      "step": 6772,
      "training_step_time": 7.488132476806641
    },
    {
      "epoch": 2.58331298828125e-05,
      "model_forward_time": 3.1374082565307617,
      "step": 6772
    },
    {
      "epoch": 2.58331298828125e-05,
      "step": 6772,
      "training_step_time": 9.921164751052856
    },
    {
      "epoch": 2.58331298828125e-05,
      "model_forward_time": 3.1388847827911377,
      "step": 6772
    },
    {
      "dynamics_loss_avg": 0.06430706568062305,
      "epoch": 2.58331298828125e-05,
      "step": 6772
    },
    {
      "action_loss_avg": 0.023609667364507914,
      "epoch": 2.58331298828125e-05,
      "step": 6772
    },
    {
      "epoch": 2.58331298828125e-05,
      "step": 6772,
      "training_step_time": 9.92293667793274
    },
    {
      "epoch": 2.58331298828125e-05,
      "model_forward_time": 3.1422319412231445,
      "step": 6772
    },
    {
      "epoch": 2.58331298828125e-05,
      "step": 6772,
      "training_step_time": 10.282350063323975
    },
    {
      "epoch": 2.5836944580078124e-05,
      "model_forward_time": 3.4037320613861084,
      "step": 6773
    },
    {
      "epoch": 2.5836944580078124e-05,
      "step": 6773,
      "training_step_time": 10.184059381484985
    },
    {
      "epoch": 2.5836944580078124e-05,
      "model_forward_time": 3.139538288116455,
      "step": 6773
    },
    {
      "epoch": 2.5836944580078124e-05,
      "step": 6773,
      "training_step_time": 9.915088891983032
    },
    {
      "epoch": 2.5836944580078124e-05,
      "model_forward_time": 3.134666681289673,
      "step": 6773
    },
    {
      "epoch": 2.5836944580078124e-05,
      "step": 6773,
      "training_step_time": 9.910077095031738
    },
    {
      "epoch": 2.5836944580078124e-05,
      "model_forward_time": 3.1380720138549805,
      "step": 6773
    },
    {
      "epoch": 2.5836944580078124e-05,
      "step": 6773,
      "training_step_time": 10.274803638458252
    },
    {
      "epoch": 2.584075927734375e-05,
      "model_forward_time": 3.1479623317718506,
      "step": 6774
    },
    {
      "epoch": 2.584075927734375e-05,
      "step": 6774,
      "training_step_time": 9.925922393798828
    },
    {
      "epoch": 2.584075927734375e-05,
      "model_forward_time": 3.4157581329345703,
      "step": 6774
    },
    {
      "epoch": 2.584075927734375e-05,
      "step": 6774,
      "training_step_time": 10.192345142364502
    },
    {
      "epoch": 2.584075927734375e-05,
      "model_forward_time": 3.135099172592163,
      "step": 6774
    },
    {
      "epoch": 2.584075927734375e-05,
      "step": 6774,
      "training_step_time": 9.912198066711426
    },
    {
      "epoch": 2.584075927734375e-05,
      "model_forward_time": 3.139413356781006,
      "step": 6774
    },
    {
      "epoch": 2.584075927734375e-05,
      "step": 6774,
      "training_step_time": 10.28421950340271
    },
    {
      "epoch": 2.5844573974609376e-05,
      "model_forward_time": 3.1504948139190674,
      "step": 6775
    },
    {
      "dynamics_loss_avg": 0.08511234112083912,
      "epoch": 2.5844573974609376e-05,
      "step": 6775
    },
    {
      "action_loss_avg": 0.033293232601135966,
      "epoch": 2.5844573974609376e-05,
      "step": 6775
    },
    {
      "epoch": 2.5844573974609376e-05,
      "step": 6775,
      "training_step_time": 9.933132648468018
    },
    {
      "epoch": 2.5844573974609376e-05,
      "model_forward_time": 3.1389214992523193,
      "step": 6775
    },
    {
      "epoch": 2.5844573974609376e-05,
      "step": 6775,
      "training_step_time": 9.91931939125061
    },
    {
      "epoch": 2.5844573974609376e-05,
      "model_forward_time": 3.394850969314575,
      "step": 6775
    },
    {
      "epoch": 2.5844573974609376e-05,
      "step": 6775,
      "training_step_time": 10.1754891872406
    },
    {
      "epoch": 2.5844573974609376e-05,
      "model_forward_time": 3.1423561573028564,
      "step": 6775
    },
    {
      "epoch": 2.5844573974609376e-05,
      "step": 6775,
      "training_step_time": 10.269883871078491
    },
    {
      "epoch": 2.5848388671875e-05,
      "model_forward_time": 3.185939311981201,
      "step": 6776
    },
    {
      "epoch": 2.5848388671875e-05,
      "step": 6776,
      "training_step_time": 9.980835676193237
    },
    {
      "epoch": 2.5848388671875e-05,
      "model_forward_time": 3.136648178100586,
      "step": 6776
    },
    {
      "epoch": 2.5848388671875e-05,
      "step": 6776,
      "training_step_time": 9.915431022644043
    },
    {
      "epoch": 2.5848388671875e-05,
      "model_forward_time": 3.1401052474975586,
      "step": 6776
    },
    {
      "epoch": 2.5848388671875e-05,
      "step": 6776,
      "training_step_time": 9.922182321548462
    },
    {
      "epoch": 2.5848388671875e-05,
      "model_forward_time": 3.4111828804016113,
      "step": 6776
    },
    {
      "epoch": 2.5848388671875e-05,
      "step": 6776,
      "training_step_time": 10.284553289413452
    },
    {
      "epoch": 2.5852203369140626e-05,
      "model_forward_time": 3.147315502166748,
      "step": 6777
    },
    {
      "epoch": 2.5852203369140626e-05,
      "step": 6777,
      "training_step_time": 9.930759906768799
    },
    {
      "epoch": 2.5852203369140626e-05,
      "model_forward_time": 3.132558822631836,
      "step": 6777
    },
    {
      "epoch": 2.5852203369140626e-05,
      "step": 6777,
      "training_step_time": 9.915235042572021
    },
    {
      "epoch": 2.5852203369140626e-05,
      "model_forward_time": 3.134967565536499,
      "step": 6777
    },
    {
      "dynamics_loss_avg": 0.04498824048787355,
      "epoch": 2.5852203369140626e-05,
      "step": 6777
    },
    {
      "action_loss_avg": 0.018525476241484283,
      "epoch": 2.5852203369140626e-05,
      "step": 6777
    },
    {
      "epoch": 2.5852203369140626e-05,
      "step": 6777,
      "training_step_time": 9.916256666183472
    },
    {
      "epoch": 2.5852203369140626e-05,
      "model_forward_time": 3.1358227729797363,
      "step": 6777
    },
    {
      "epoch": 2.5852203369140626e-05,
      "step": 6777,
      "training_step_time": 10.008270263671875
    },
    {
      "epoch": 2.585601806640625e-05,
      "model_forward_time": 3.4328770637512207,
      "step": 6778
    },
    {
      "epoch": 2.585601806640625e-05,
      "step": 6778,
      "training_step_time": 10.214595079421997
    },
    {
      "epoch": 2.585601806640625e-05,
      "model_forward_time": 3.1415486335754395,
      "step": 6778
    },
    {
      "epoch": 2.585601806640625e-05,
      "step": 6778,
      "training_step_time": 9.92277979850769
    },
    {
      "epoch": 2.585601806640625e-05,
      "model_forward_time": 3.142841100692749,
      "step": 6778
    },
    {
      "epoch": 2.585601806640625e-05,
      "step": 6778,
      "training_step_time": 9.92566704750061
    },
    {
      "epoch": 2.585601806640625e-05,
      "model_forward_time": 3.1391170024871826,
      "step": 6778
    },
    {
      "epoch": 2.585601806640625e-05,
      "step": 6778,
      "training_step_time": 10.015151262283325
    },
    {
      "epoch": 2.5859832763671875e-05,
      "model_forward_time": 3.182257890701294,
      "step": 6779
    },
    {
      "epoch": 2.5859832763671875e-05,
      "step": 6779,
      "training_step_time": 9.961181402206421
    },
    {
      "epoch": 2.5859832763671875e-05,
      "model_forward_time": 3.403169870376587,
      "step": 6779
    },
    {
      "epoch": 2.5859832763671875e-05,
      "step": 6779,
      "training_step_time": 10.185866832733154
    },
    {
      "epoch": 2.5859832763671875e-05,
      "model_forward_time": 3.1433489322662354,
      "step": 6779
    },
    {
      "epoch": 2.5859832763671875e-05,
      "step": 6779,
      "training_step_time": 9.926142930984497
    },
    {
      "epoch": 2.5859832763671875e-05,
      "model_forward_time": 3.142693519592285,
      "step": 6779
    },
    {
      "epoch": 2.5859832763671875e-05,
      "step": 6779,
      "training_step_time": 10.269828081130981
    },
    {
      "epoch": 2.58636474609375e-05,
      "grad_norm": 0.3848026394844055,
      "learning_rate": 1e-05,
      "loss": 0.0885,
      "step": 6780
    },
    {
      "epoch": 2.58636474609375e-05,
      "model_forward_time": 3.1764214038848877,
      "step": 6780
    },
    {
      "dynamics_loss_avg": 0.07155334167182445,
      "epoch": 2.58636474609375e-05,
      "step": 6780
    },
    {
      "action_loss_avg": 0.022895061830058693,
      "epoch": 2.58636474609375e-05,
      "step": 6780
    },
    {
      "epoch": 2.58636474609375e-05,
      "step": 6780,
      "training_step_time": 9.964877367019653
    },
    {
      "epoch": 2.58636474609375e-05,
      "model_forward_time": 3.146367311477661,
      "step": 6780
    },
    {
      "epoch": 2.58636474609375e-05,
      "step": 6780,
      "training_step_time": 9.928044557571411
    },
    {
      "epoch": 2.58636474609375e-05,
      "model_forward_time": 3.458338975906372,
      "step": 6780
    },
    {
      "epoch": 2.58636474609375e-05,
      "step": 6780,
      "training_step_time": 10.240264654159546
    },
    {
      "epoch": 2.58636474609375e-05,
      "model_forward_time": 3.1425864696502686,
      "step": 6780
    },
    {
      "epoch": 2.58636474609375e-05,
      "step": 6780,
      "training_step_time": 10.288410902023315
    },
    {
      "epoch": 2.5867462158203124e-05,
      "model_forward_time": 3.1606390476226807,
      "step": 6781
    },
    {
      "epoch": 2.5867462158203124e-05,
      "step": 6781,
      "training_step_time": 9.944169759750366
    },
    {
      "epoch": 2.5867462158203124e-05,
      "model_forward_time": 3.150015354156494,
      "step": 6781
    },
    {
      "epoch": 2.5867462158203124e-05,
      "step": 6781,
      "training_step_time": 9.934197187423706
    },
    {
      "epoch": 2.5867462158203124e-05,
      "model_forward_time": 3.146545648574829,
      "step": 6781
    },
    {
      "epoch": 2.5867462158203124e-05,
      "step": 6781,
      "training_step_time": 9.92999815940857
    },
    {
      "epoch": 2.5867462158203124e-05,
      "model_forward_time": 3.409109354019165,
      "step": 6781
    },
    {
      "epoch": 2.5867462158203124e-05,
      "step": 6781,
      "training_step_time": 10.286869049072266
    },
    {
      "epoch": 2.587127685546875e-05,
      "model_forward_time": 3.1806986331939697,
      "step": 6782
    },
    {
      "epoch": 2.587127685546875e-05,
      "step": 6782,
      "training_step_time": 9.964922189712524
    },
    {
      "epoch": 2.587127685546875e-05,
      "model_forward_time": 3.1461806297302246,
      "step": 6782
    },
    {
      "epoch": 2.587127685546875e-05,
      "step": 6782,
      "training_step_time": 9.928203344345093
    },
    {
      "epoch": 2.587127685546875e-05,
      "model_forward_time": 3.143303155899048,
      "step": 6782
    },
    {
      "dynamics_loss_avg": 0.05689707733690739,
      "epoch": 2.587127685546875e-05,
      "step": 6782
    },
    {
      "action_loss_avg": 0.02194083328358829,
      "epoch": 2.587127685546875e-05,
      "step": 6782
    },
    {
      "epoch": 2.587127685546875e-05,
      "step": 6782,
      "training_step_time": 9.928030967712402
    },
    {
      "epoch": 2.587127685546875e-05,
      "model_forward_time": 3.150655746459961,
      "step": 6782
    },
    {
      "epoch": 2.587127685546875e-05,
      "step": 6782,
      "training_step_time": 10.306109189987183
    },
    {
      "epoch": 2.5875091552734376e-05,
      "model_forward_time": 3.4204020500183105,
      "step": 6783
    },
    {
      "epoch": 2.5875091552734376e-05,
      "step": 6783,
      "training_step_time": 10.204612970352173
    },
    {
      "epoch": 2.5875091552734376e-05,
      "model_forward_time": 3.147571325302124,
      "step": 6783
    },
    {
      "epoch": 2.5875091552734376e-05,
      "step": 6783,
      "training_step_time": 9.932834386825562
    },
    {
      "epoch": 2.5875091552734376e-05,
      "model_forward_time": 2.4430673122406006,
      "step": 6783
    },
    {
      "epoch": 2.5875091552734376e-05,
      "step": 6783,
      "training_step_time": 7.479414939880371
    },
    {
      "epoch": 2.5875091552734376e-05,
      "model_forward_time": 3.1528453826904297,
      "step": 6783
    },
    {
      "epoch": 2.5875091552734376e-05,
      "step": 6783,
      "training_step_time": 10.235368490219116
    },
    {
      "epoch": 2.587890625e-05,
      "model_forward_time": 2.4849209785461426,
      "step": 6784
    },
    {
      "epoch": 2.587890625e-05,
      "step": 6784,
      "training_step_time": 7.520545482635498
    },
    {
      "epoch": 2.587890625e-05,
      "model_forward_time": 3.4435875415802,
      "step": 6784
    },
    {
      "epoch": 2.587890625e-05,
      "step": 6784,
      "training_step_time": 10.220329284667969
    },
    {
      "epoch": 2.587890625e-05,
      "model_forward_time": 3.145617723464966,
      "step": 6784
    },
    {
      "epoch": 2.587890625e-05,
      "step": 6784,
      "training_step_time": 9.925742864608765
    },
    {
      "epoch": 2.587890625e-05,
      "model_forward_time": 3.150710105895996,
      "step": 6784
    },
    {
      "epoch": 2.587890625e-05,
      "step": 6784,
      "training_step_time": 10.29881501197815
    },
    {
      "epoch": 2.5882720947265626e-05,
      "model_forward_time": 3.1597845554351807,
      "step": 6785
    },
    {
      "dynamics_loss_avg": 0.04749938733875751,
      "epoch": 2.5882720947265626e-05,
      "step": 6785
    },
    {
      "action_loss_avg": 0.022492261603474617,
      "epoch": 2.5882720947265626e-05,
      "step": 6785
    },
    {
      "epoch": 2.5882720947265626e-05,
      "step": 6785,
      "training_step_time": 9.944748640060425
    },
    {
      "epoch": 2.5882720947265626e-05,
      "model_forward_time": 3.14690899848938,
      "step": 6785
    },
    {
      "epoch": 2.5882720947265626e-05,
      "step": 6785,
      "training_step_time": 9.927764415740967
    },
    {
      "epoch": 2.5882720947265626e-05,
      "model_forward_time": 3.1488687992095947,
      "step": 6785
    },
    {
      "epoch": 2.5882720947265626e-05,
      "step": 6785,
      "training_step_time": 10.21709394454956
    },
    {
      "epoch": 2.5882720947265626e-05,
      "model_forward_time": 3.1517093181610107,
      "step": 6785
    },
    {
      "epoch": 2.5882720947265626e-05,
      "step": 6785,
      "training_step_time": 10.286846160888672
    },
    {
      "epoch": 2.588653564453125e-05,
      "model_forward_time": 3.1650893688201904,
      "step": 6786
    },
    {
      "epoch": 2.588653564453125e-05,
      "step": 6786,
      "training_step_time": 9.949625730514526
    },
    {
      "epoch": 2.588653564453125e-05,
      "model_forward_time": 3.1454501152038574,
      "step": 6786
    },
    {
      "epoch": 2.588653564453125e-05,
      "step": 6786,
      "training_step_time": 9.932121753692627
    },
    {
      "epoch": 2.588653564453125e-05,
      "model_forward_time": 3.1540474891662598,
      "step": 6786
    },
    {
      "epoch": 2.588653564453125e-05,
      "step": 6786,
      "training_step_time": 9.939379930496216
    },
    {
      "epoch": 2.588653564453125e-05,
      "model_forward_time": 3.4328079223632812,
      "step": 6786
    },
    {
      "epoch": 2.588653564453125e-05,
      "step": 6786,
      "training_step_time": 10.311402320861816
    },
    {
      "epoch": 2.5890350341796875e-05,
      "model_forward_time": 3.168612480163574,
      "step": 6787
    },
    {
      "epoch": 2.5890350341796875e-05,
      "step": 6787,
      "training_step_time": 9.952415943145752
    },
    {
      "epoch": 2.5890350341796875e-05,
      "model_forward_time": 3.14782452583313,
      "step": 6787
    },
    {
      "epoch": 2.5890350341796875e-05,
      "step": 6787,
      "training_step_time": 9.930797100067139
    },
    {
      "epoch": 2.5890350341796875e-05,
      "model_forward_time": 3.1483373641967773,
      "step": 6787
    },
    {
      "dynamics_loss_avg": 0.03684982499107718,
      "epoch": 2.5890350341796875e-05,
      "step": 6787
    },
    {
      "action_loss_avg": 0.012962630996480585,
      "epoch": 2.5890350341796875e-05,
      "step": 6787
    },
    {
      "epoch": 2.5890350341796875e-05,
      "step": 6787,
      "training_step_time": 9.933878183364868
    },
    {
      "epoch": 2.5890350341796875e-05,
      "model_forward_time": 3.1484897136688232,
      "step": 6787
    },
    {
      "epoch": 2.5890350341796875e-05,
      "step": 6787,
      "training_step_time": 10.285378694534302
    },
    {
      "epoch": 2.58941650390625e-05,
      "model_forward_time": 3.1501946449279785,
      "step": 6788
    },
    {
      "epoch": 2.58941650390625e-05,
      "step": 6788,
      "training_step_time": 10.205750226974487
    },
    {
      "epoch": 2.58941650390625e-05,
      "model_forward_time": 3.1424543857574463,
      "step": 6788
    },
    {
      "epoch": 2.58941650390625e-05,
      "step": 6788,
      "training_step_time": 9.922877073287964
    },
    {
      "epoch": 2.58941650390625e-05,
      "model_forward_time": 3.1439220905303955,
      "step": 6788
    },
    {
      "epoch": 2.58941650390625e-05,
      "step": 6788,
      "training_step_time": 9.921157598495483
    },
    {
      "epoch": 2.58941650390625e-05,
      "model_forward_time": 3.1420583724975586,
      "step": 6788
    },
    {
      "epoch": 2.58941650390625e-05,
      "step": 6788,
      "training_step_time": 10.270158052444458
    },
    {
      "epoch": 2.5897979736328124e-05,
      "model_forward_time": 3.185030698776245,
      "step": 6789
    },
    {
      "epoch": 2.5897979736328124e-05,
      "step": 6789,
      "training_step_time": 9.964531183242798
    },
    {
      "epoch": 2.5897979736328124e-05,
      "model_forward_time": 3.4202680587768555,
      "step": 6789
    },
    {
      "epoch": 2.5897979736328124e-05,
      "step": 6789,
      "training_step_time": 10.201175689697266
    },
    {
      "epoch": 2.5897979736328124e-05,
      "model_forward_time": 3.1452574729919434,
      "step": 6789
    },
    {
      "epoch": 2.5897979736328124e-05,
      "step": 6789,
      "training_step_time": 9.927326917648315
    },
    {
      "epoch": 2.5897979736328124e-05,
      "model_forward_time": 3.1479601860046387,
      "step": 6789
    },
    {
      "epoch": 2.5897979736328124e-05,
      "step": 6789,
      "training_step_time": 10.027111053466797
    },
    {
      "epoch": 2.590179443359375e-05,
      "grad_norm": 0.4016224443912506,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 6790
    },
    {
      "epoch": 2.590179443359375e-05,
      "model_forward_time": 3.1594080924987793,
      "step": 6790
    },
    {
      "dynamics_loss_avg": 0.06356278136372566,
      "epoch": 2.590179443359375e-05,
      "step": 6790
    },
    {
      "action_loss_avg": 0.02133226776495576,
      "epoch": 2.590179443359375e-05,
      "step": 6790
    },
    {
      "epoch": 2.590179443359375e-05,
      "step": 6790,
      "training_step_time": 9.942708492279053
    },
    {
      "epoch": 2.590179443359375e-05,
      "model_forward_time": 3.154681444168091,
      "step": 6790
    },
    {
      "epoch": 2.590179443359375e-05,
      "step": 6790,
      "training_step_time": 9.936299562454224
    },
    {
      "epoch": 2.590179443359375e-05,
      "model_forward_time": 3.436246633529663,
      "step": 6790
    },
    {
      "epoch": 2.590179443359375e-05,
      "step": 6790,
      "training_step_time": 10.22405481338501
    },
    {
      "epoch": 2.590179443359375e-05,
      "model_forward_time": 3.1546003818511963,
      "step": 6790
    },
    {
      "epoch": 2.590179443359375e-05,
      "step": 6790,
      "training_step_time": 10.277886629104614
    },
    {
      "epoch": 2.5905609130859376e-05,
      "model_forward_time": 3.1586246490478516,
      "step": 6791
    },
    {
      "epoch": 2.5905609130859376e-05,
      "step": 6791,
      "training_step_time": 9.943791151046753
    },
    {
      "epoch": 2.5905609130859376e-05,
      "model_forward_time": 3.1524343490600586,
      "step": 6791
    },
    {
      "epoch": 2.5905609130859376e-05,
      "step": 6791,
      "training_step_time": 9.940056800842285
    },
    {
      "epoch": 2.5905609130859376e-05,
      "model_forward_time": 3.1491947174072266,
      "step": 6791
    },
    {
      "epoch": 2.5905609130859376e-05,
      "step": 6791,
      "training_step_time": 9.933348655700684
    },
    {
      "epoch": 2.5905609130859376e-05,
      "model_forward_time": 3.414386034011841,
      "step": 6791
    },
    {
      "epoch": 2.5905609130859376e-05,
      "step": 6791,
      "training_step_time": 10.293081521987915
    },
    {
      "epoch": 2.5909423828125e-05,
      "model_forward_time": 2.4575881958007812,
      "step": 6792
    },
    {
      "epoch": 2.5909423828125e-05,
      "step": 6792,
      "training_step_time": 7.493473529815674
    },
    {
      "epoch": 2.5909423828125e-05,
      "model_forward_time": 3.147592306137085,
      "step": 6792
    },
    {
      "epoch": 2.5909423828125e-05,
      "step": 6792,
      "training_step_time": 9.925017595291138
    },
    {
      "epoch": 2.5909423828125e-05,
      "model_forward_time": 3.1442015171051025,
      "step": 6792
    },
    {
      "dynamics_loss_avg": 0.06237897649407387,
      "epoch": 2.5909423828125e-05,
      "step": 6792
    },
    {
      "action_loss_avg": 0.02186499796807766,
      "epoch": 2.5909423828125e-05,
      "step": 6792
    },
    {
      "epoch": 2.5909423828125e-05,
      "step": 6792,
      "training_step_time": 9.928813219070435
    },
    {
      "epoch": 2.5909423828125e-05,
      "model_forward_time": 3.149336576461792,
      "step": 6792
    },
    {
      "epoch": 2.5909423828125e-05,
      "step": 6792,
      "training_step_time": 10.04341173171997
    },
    {
      "epoch": 2.5913238525390626e-05,
      "model_forward_time": 2.485952138900757,
      "step": 6793
    },
    {
      "epoch": 2.5913238525390626e-05,
      "step": 6793,
      "training_step_time": 7.524743318557739
    },
    {
      "epoch": 2.5913238525390626e-05,
      "model_forward_time": 3.422377109527588,
      "step": 6793
    },
    {
      "epoch": 2.5913238525390626e-05,
      "step": 6793,
      "training_step_time": 10.203017473220825
    },
    {
      "epoch": 2.5913238525390626e-05,
      "model_forward_time": 3.153883457183838,
      "step": 6793
    },
    {
      "epoch": 2.5913238525390626e-05,
      "step": 6793,
      "training_step_time": 9.931835412979126
    },
    {
      "epoch": 2.5913238525390626e-05,
      "model_forward_time": 3.1457953453063965,
      "step": 6793
    },
    {
      "epoch": 2.5913238525390626e-05,
      "step": 6793,
      "training_step_time": 10.329346895217896
    },
    {
      "epoch": 2.591705322265625e-05,
      "model_forward_time": 3.1579322814941406,
      "step": 6794
    },
    {
      "epoch": 2.591705322265625e-05,
      "step": 6794,
      "training_step_time": 9.94306206703186
    },
    {
      "epoch": 2.591705322265625e-05,
      "model_forward_time": 3.150221824645996,
      "step": 6794
    },
    {
      "epoch": 2.591705322265625e-05,
      "step": 6794,
      "training_step_time": 10.217615842819214
    },
    {
      "epoch": 2.591705322265625e-05,
      "model_forward_time": 3.147477388381958,
      "step": 6794
    },
    {
      "epoch": 2.591705322265625e-05,
      "step": 6794,
      "training_step_time": 9.932474374771118
    },
    {
      "epoch": 2.591705322265625e-05,
      "model_forward_time": 3.1470625400543213,
      "step": 6794
    },
    {
      "epoch": 2.591705322265625e-05,
      "step": 6794,
      "training_step_time": 10.027599573135376
    },
    {
      "epoch": 2.5920867919921875e-05,
      "model_forward_time": 3.1670777797698975,
      "step": 6795
    },
    {
      "dynamics_loss_avg": 0.06237582005560398,
      "epoch": 2.5920867919921875e-05,
      "step": 6795
    },
    {
      "action_loss_avg": 0.022260717675089838,
      "epoch": 2.5920867919921875e-05,
      "step": 6795
    },
    {
      "epoch": 2.5920867919921875e-05,
      "step": 6795,
      "training_step_time": 9.954240560531616
    },
    {
      "epoch": 2.5920867919921875e-05,
      "model_forward_time": 3.1452174186706543,
      "step": 6795
    },
    {
      "epoch": 2.5920867919921875e-05,
      "step": 6795,
      "training_step_time": 9.924739599227905
    },
    {
      "epoch": 2.5920867919921875e-05,
      "model_forward_time": 3.149827480316162,
      "step": 6795
    },
    {
      "epoch": 2.5920867919921875e-05,
      "step": 6795,
      "training_step_time": 10.216766595840454
    },
    {
      "epoch": 2.5920867919921875e-05,
      "model_forward_time": 3.1521613597869873,
      "step": 6795
    },
    {
      "epoch": 2.5920867919921875e-05,
      "step": 6795,
      "training_step_time": 10.27828574180603
    },
    {
      "epoch": 2.59246826171875e-05,
      "model_forward_time": 3.1454012393951416,
      "step": 6796
    },
    {
      "epoch": 2.59246826171875e-05,
      "step": 6796,
      "training_step_time": 9.929235696792603
    },
    {
      "epoch": 2.59246826171875e-05,
      "model_forward_time": 3.1365556716918945,
      "step": 6796
    },
    {
      "epoch": 2.59246826171875e-05,
      "step": 6796,
      "training_step_time": 9.92311406135559
    },
    {
      "epoch": 2.59246826171875e-05,
      "model_forward_time": 3.1382546424865723,
      "step": 6796
    },
    {
      "epoch": 2.59246826171875e-05,
      "step": 6796,
      "training_step_time": 9.916102886199951
    },
    {
      "epoch": 2.59246826171875e-05,
      "model_forward_time": 3.138432025909424,
      "step": 6796
    },
    {
      "epoch": 2.59246826171875e-05,
      "step": 6796,
      "training_step_time": 10.30201244354248
    },
    {
      "epoch": 2.5928497314453124e-05,
      "model_forward_time": 3.153143882751465,
      "step": 6797
    },
    {
      "epoch": 2.5928497314453124e-05,
      "step": 6797,
      "training_step_time": 9.936144351959229
    },
    {
      "epoch": 2.5928497314453124e-05,
      "model_forward_time": 3.1328728199005127,
      "step": 6797
    },
    {
      "epoch": 2.5928497314453124e-05,
      "step": 6797,
      "training_step_time": 9.91264009475708
    },
    {
      "epoch": 2.5928497314453124e-05,
      "model_forward_time": 3.141472339630127,
      "step": 6797
    },
    {
      "dynamics_loss_avg": 0.06391759403049946,
      "epoch": 2.5928497314453124e-05,
      "step": 6797
    },
    {
      "action_loss_avg": 0.027394183445721865,
      "epoch": 2.5928497314453124e-05,
      "step": 6797
    },
    {
      "epoch": 2.5928497314453124e-05,
      "step": 6797,
      "training_step_time": 9.926707744598389
    },
    {
      "epoch": 2.5928497314453124e-05,
      "model_forward_time": 3.1382811069488525,
      "step": 6797
    },
    {
      "epoch": 2.5928497314453124e-05,
      "step": 6797,
      "training_step_time": 10.295022010803223
    },
    {
      "epoch": 2.593231201171875e-05,
      "model_forward_time": 3.426736354827881,
      "step": 6798
    },
    {
      "epoch": 2.593231201171875e-05,
      "step": 6798,
      "training_step_time": 10.203406810760498
    },
    {
      "epoch": 2.593231201171875e-05,
      "model_forward_time": 3.1298649311065674,
      "step": 6798
    },
    {
      "epoch": 2.593231201171875e-05,
      "step": 6798,
      "training_step_time": 9.909235715866089
    },
    {
      "epoch": 2.593231201171875e-05,
      "model_forward_time": 3.1346375942230225,
      "step": 6798
    },
    {
      "epoch": 2.593231201171875e-05,
      "step": 6798,
      "training_step_time": 9.914226055145264
    },
    {
      "epoch": 2.593231201171875e-05,
      "model_forward_time": 3.132275342941284,
      "step": 6798
    },
    {
      "epoch": 2.593231201171875e-05,
      "step": 6798,
      "training_step_time": 10.289784908294678
    },
    {
      "epoch": 2.5936126708984376e-05,
      "model_forward_time": 3.142439365386963,
      "step": 6799
    },
    {
      "epoch": 2.5936126708984376e-05,
      "step": 6799,
      "training_step_time": 9.929483413696289
    },
    {
      "epoch": 2.5936126708984376e-05,
      "model_forward_time": 3.1333839893341064,
      "step": 6799
    },
    {
      "epoch": 2.5936126708984376e-05,
      "step": 6799,
      "training_step_time": 10.187120914459229
    },
    {
      "epoch": 2.5936126708984376e-05,
      "model_forward_time": 3.136343479156494,
      "step": 6799
    },
    {
      "epoch": 2.5936126708984376e-05,
      "step": 6799,
      "training_step_time": 9.911956787109375
    },
    {
      "epoch": 2.5936126708984376e-05,
      "model_forward_time": 3.1338722705841064,
      "step": 6799
    },
    {
      "epoch": 2.5936126708984376e-05,
      "step": 6799,
      "training_step_time": 10.285032272338867
    },
    {
      "epoch": 2.593994140625e-05,
      "grad_norm": 0.35785481333732605,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 6800
    },
    {
      "epoch": 2.593994140625e-05,
      "model_forward_time": 3.1678872108459473,
      "step": 6800
    },
    {
      "dynamics_loss_avg": 0.06572662685066462,
      "epoch": 2.593994140625e-05,
      "step": 6800
    },
    {
      "action_loss_avg": 0.02165632164105773,
      "epoch": 2.593994140625e-05,
      "step": 6800
    },
    {
      "epoch": 2.593994140625e-05,
      "step": 6800,
      "training_step_time": 9.947081327438354
    },
    {
      "epoch": 2.593994140625e-05,
      "model_forward_time": 3.1322994232177734,
      "step": 6800
    },
    {
      "epoch": 2.593994140625e-05,
      "step": 6800,
      "training_step_time": 9.908639669418335
    },
    {
      "epoch": 2.593994140625e-05,
      "model_forward_time": 3.442267417907715,
      "step": 6800
    },
    {
      "epoch": 2.593994140625e-05,
      "step": 6800,
      "training_step_time": 10.216137886047363
    },
    {
      "epoch": 2.593994140625e-05,
      "model_forward_time": 3.1316869258880615,
      "step": 6800
    },
    {
      "epoch": 2.593994140625e-05,
      "step": 6800,
      "training_step_time": 10.242585897445679
    },
    {
      "epoch": 2.5943756103515626e-05,
      "model_forward_time": 3.1361582279205322,
      "step": 6801
    },
    {
      "epoch": 2.5943756103515626e-05,
      "step": 6801,
      "training_step_time": 9.913157224655151
    },
    {
      "epoch": 2.5943756103515626e-05,
      "model_forward_time": 3.1318273544311523,
      "step": 6801
    },
    {
      "epoch": 2.5943756103515626e-05,
      "step": 6801,
      "training_step_time": 9.90647268295288
    },
    {
      "epoch": 2.5943756103515626e-05,
      "model_forward_time": 3.1306614875793457,
      "step": 6801
    },
    {
      "epoch": 2.5943756103515626e-05,
      "step": 6801,
      "training_step_time": 9.908474206924438
    },
    {
      "epoch": 2.5943756103515626e-05,
      "model_forward_time": 3.3920435905456543,
      "step": 6801
    },
    {
      "epoch": 2.5943756103515626e-05,
      "step": 6801,
      "training_step_time": 10.271214008331299
    },
    {
      "epoch": 2.594757080078125e-05,
      "model_forward_time": 3.142580986022949,
      "step": 6802
    },
    {
      "epoch": 2.594757080078125e-05,
      "step": 6802,
      "training_step_time": 9.914615631103516
    },
    {
      "epoch": 2.594757080078125e-05,
      "model_forward_time": 3.1428093910217285,
      "step": 6802
    },
    {
      "epoch": 2.594757080078125e-05,
      "step": 6802,
      "training_step_time": 9.91887354850769
    },
    {
      "epoch": 2.594757080078125e-05,
      "model_forward_time": 2.4292356967926025,
      "step": 6802
    },
    {
      "dynamics_loss_avg": 0.0666001170873642,
      "epoch": 2.594757080078125e-05,
      "step": 6802
    },
    {
      "action_loss_avg": 0.022507328633219005,
      "epoch": 2.594757080078125e-05,
      "step": 6802
    },
    {
      "epoch": 2.594757080078125e-05,
      "step": 6802,
      "training_step_time": 7.465043067932129
    },
    {
      "epoch": 2.594757080078125e-05,
      "model_forward_time": 3.1363062858581543,
      "step": 6802
    },
    {
      "epoch": 2.594757080078125e-05,
      "step": 6802,
      "training_step_time": 10.025306701660156
    },
    {
      "epoch": 2.5951385498046875e-05,
      "model_forward_time": 3.422130584716797,
      "step": 6803
    },
    {
      "epoch": 2.5951385498046875e-05,
      "step": 6803,
      "training_step_time": 10.211623191833496
    },
    {
      "epoch": 2.5951385498046875e-05,
      "model_forward_time": 3.141655683517456,
      "step": 6803
    },
    {
      "epoch": 2.5951385498046875e-05,
      "step": 6803,
      "training_step_time": 9.919992685317993
    },
    {
      "epoch": 2.5951385498046875e-05,
      "model_forward_time": 3.138944149017334,
      "step": 6803
    },
    {
      "epoch": 2.5951385498046875e-05,
      "step": 6803,
      "training_step_time": 9.91696810722351
    },
    {
      "epoch": 2.5951385498046875e-05,
      "model_forward_time": 3.140010118484497,
      "step": 6803
    },
    {
      "epoch": 2.5951385498046875e-05,
      "step": 6803,
      "training_step_time": 10.031192541122437
    },
    {
      "epoch": 2.59552001953125e-05,
      "model_forward_time": 3.1551125049591064,
      "step": 6804
    },
    {
      "epoch": 2.59552001953125e-05,
      "step": 6804,
      "training_step_time": 9.934460878372192
    },
    {
      "epoch": 2.59552001953125e-05,
      "model_forward_time": 3.1440954208374023,
      "step": 6804
    },
    {
      "epoch": 2.59552001953125e-05,
      "step": 6804,
      "training_step_time": 10.19074535369873
    },
    {
      "epoch": 2.59552001953125e-05,
      "model_forward_time": 3.1381940841674805,
      "step": 6804
    },
    {
      "epoch": 2.59552001953125e-05,
      "step": 6804,
      "training_step_time": 9.91255521774292
    },
    {
      "epoch": 2.59552001953125e-05,
      "model_forward_time": 3.1438984870910645,
      "step": 6804
    },
    {
      "epoch": 2.59552001953125e-05,
      "step": 6804,
      "training_step_time": 10.03589153289795
    },
    {
      "epoch": 2.5959014892578124e-05,
      "model_forward_time": 3.1568491458892822,
      "step": 6805
    },
    {
      "dynamics_loss_avg": 0.07764026280492545,
      "epoch": 2.5959014892578124e-05,
      "step": 6805
    },
    {
      "action_loss_avg": 0.028380265552550556,
      "epoch": 2.5959014892578124e-05,
      "step": 6805
    },
    {
      "epoch": 2.5959014892578124e-05,
      "step": 6805,
      "training_step_time": 9.937339067459106
    },
    {
      "epoch": 2.5959014892578124e-05,
      "model_forward_time": 3.139953374862671,
      "step": 6805
    },
    {
      "epoch": 2.5959014892578124e-05,
      "step": 6805,
      "training_step_time": 9.917332887649536
    },
    {
      "epoch": 2.5959014892578124e-05,
      "model_forward_time": 3.1398260593414307,
      "step": 6805
    },
    {
      "epoch": 2.5959014892578124e-05,
      "step": 6805,
      "training_step_time": 10.202131509780884
    },
    {
      "epoch": 2.5959014892578124e-05,
      "model_forward_time": 3.137779712677002,
      "step": 6805
    },
    {
      "epoch": 2.5959014892578124e-05,
      "step": 6805,
      "training_step_time": 10.018789052963257
    },
    {
      "epoch": 2.596282958984375e-05,
      "model_forward_time": 3.1548731327056885,
      "step": 6806
    },
    {
      "epoch": 2.596282958984375e-05,
      "step": 6806,
      "training_step_time": 9.930245876312256
    },
    {
      "epoch": 2.596282958984375e-05,
      "model_forward_time": 3.1360814571380615,
      "step": 6806
    },
    {
      "epoch": 2.596282958984375e-05,
      "step": 6806,
      "training_step_time": 9.911257982254028
    },
    {
      "epoch": 2.596282958984375e-05,
      "model_forward_time": 3.1362242698669434,
      "step": 6806
    },
    {
      "epoch": 2.596282958984375e-05,
      "step": 6806,
      "training_step_time": 9.908579587936401
    },
    {
      "epoch": 2.596282958984375e-05,
      "model_forward_time": 3.412818670272827,
      "step": 6806
    },
    {
      "epoch": 2.596282958984375e-05,
      "step": 6806,
      "training_step_time": 10.289706468582153
    },
    {
      "epoch": 2.5966644287109376e-05,
      "model_forward_time": 3.145129442214966,
      "step": 6807
    },
    {
      "epoch": 2.5966644287109376e-05,
      "step": 6807,
      "training_step_time": 9.918851852416992
    },
    {
      "epoch": 2.5966644287109376e-05,
      "model_forward_time": 3.1343040466308594,
      "step": 6807
    },
    {
      "epoch": 2.5966644287109376e-05,
      "step": 6807,
      "training_step_time": 9.906939506530762
    },
    {
      "epoch": 2.5966644287109376e-05,
      "model_forward_time": 3.1364269256591797,
      "step": 6807
    },
    {
      "dynamics_loss_avg": 0.04910854399204254,
      "epoch": 2.5966644287109376e-05,
      "step": 6807
    },
    {
      "action_loss_avg": 0.01775129153393209,
      "epoch": 2.5966644287109376e-05,
      "step": 6807
    },
    {
      "epoch": 2.5966644287109376e-05,
      "step": 6807,
      "training_step_time": 9.91565752029419
    },
    {
      "epoch": 2.5966644287109376e-05,
      "model_forward_time": 3.1369707584381104,
      "step": 6807
    },
    {
      "epoch": 2.5966644287109376e-05,
      "step": 6807,
      "training_step_time": 10.275410890579224
    },
    {
      "epoch": 2.5970458984375e-05,
      "model_forward_time": 3.1484787464141846,
      "step": 6808
    },
    {
      "epoch": 2.5970458984375e-05,
      "step": 6808,
      "training_step_time": 10.206909418106079
    },
    {
      "epoch": 2.5970458984375e-05,
      "model_forward_time": 3.13388729095459,
      "step": 6808
    },
    {
      "epoch": 2.5970458984375e-05,
      "step": 6808,
      "training_step_time": 9.909764289855957
    },
    {
      "epoch": 2.5970458984375e-05,
      "model_forward_time": 3.1379549503326416,
      "step": 6808
    },
    {
      "epoch": 2.5970458984375e-05,
      "step": 6808,
      "training_step_time": 9.912509202957153
    },
    {
      "epoch": 2.5970458984375e-05,
      "model_forward_time": 3.134036064147949,
      "step": 6808
    },
    {
      "epoch": 2.5970458984375e-05,
      "step": 6808,
      "training_step_time": 10.011423826217651
    },
    {
      "epoch": 2.5974273681640626e-05,
      "model_forward_time": 3.150491714477539,
      "step": 6809
    },
    {
      "epoch": 2.5974273681640626e-05,
      "step": 6809,
      "training_step_time": 9.923519372940063
    },
    {
      "epoch": 2.5974273681640626e-05,
      "model_forward_time": 3.386256694793701,
      "step": 6809
    },
    {
      "epoch": 2.5974273681640626e-05,
      "step": 6809,
      "training_step_time": 10.154382467269897
    },
    {
      "epoch": 2.5974273681640626e-05,
      "model_forward_time": 3.136857509613037,
      "step": 6809
    },
    {
      "epoch": 2.5974273681640626e-05,
      "step": 6809,
      "training_step_time": 9.904460191726685
    },
    {
      "epoch": 2.5974273681640626e-05,
      "model_forward_time": 3.1339781284332275,
      "step": 6809
    },
    {
      "epoch": 2.5974273681640626e-05,
      "step": 6809,
      "training_step_time": 10.012065410614014
    },
    {
      "epoch": 2.597808837890625e-05,
      "grad_norm": 0.34951427578926086,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 6810
    },
    {
      "epoch": 2.597808837890625e-05,
      "model_forward_time": 3.1507701873779297,
      "step": 6810
    },
    {
      "dynamics_loss_avg": 0.04632650669664144,
      "epoch": 2.597808837890625e-05,
      "step": 6810
    },
    {
      "action_loss_avg": 0.014639018289744853,
      "epoch": 2.597808837890625e-05,
      "step": 6810
    },
    {
      "epoch": 2.597808837890625e-05,
      "step": 6810,
      "training_step_time": 9.930405378341675
    },
    {
      "epoch": 2.597808837890625e-05,
      "model_forward_time": 3.1386730670928955,
      "step": 6810
    },
    {
      "epoch": 2.597808837890625e-05,
      "step": 6810,
      "training_step_time": 9.913295269012451
    },
    {
      "epoch": 2.597808837890625e-05,
      "model_forward_time": 3.418869972229004,
      "step": 6810
    },
    {
      "epoch": 2.597808837890625e-05,
      "step": 6810,
      "training_step_time": 10.192760944366455
    },
    {
      "epoch": 2.597808837890625e-05,
      "model_forward_time": 3.1373085975646973,
      "step": 6810
    },
    {
      "epoch": 2.597808837890625e-05,
      "step": 6810,
      "training_step_time": 10.273747444152832
    },
    {
      "epoch": 2.5981903076171875e-05,
      "model_forward_time": 3.149296522140503,
      "step": 6811
    },
    {
      "epoch": 2.5981903076171875e-05,
      "step": 6811,
      "training_step_time": 9.928818941116333
    },
    {
      "epoch": 2.5981903076171875e-05,
      "model_forward_time": 3.144554376602173,
      "step": 6811
    },
    {
      "epoch": 2.5981903076171875e-05,
      "step": 6811,
      "training_step_time": 9.924769163131714
    },
    {
      "epoch": 2.5981903076171875e-05,
      "model_forward_time": 3.1400246620178223,
      "step": 6811
    },
    {
      "epoch": 2.5981903076171875e-05,
      "step": 6811,
      "training_step_time": 9.922755479812622
    },
    {
      "epoch": 2.5981903076171875e-05,
      "model_forward_time": 3.3987064361572266,
      "step": 6811
    },
    {
      "epoch": 2.5981903076171875e-05,
      "step": 6811,
      "training_step_time": 10.293193578720093
    },
    {
      "epoch": 2.59857177734375e-05,
      "model_forward_time": 3.1558024883270264,
      "step": 6812
    },
    {
      "epoch": 2.59857177734375e-05,
      "step": 6812,
      "training_step_time": 9.936851978302002
    },
    {
      "epoch": 2.59857177734375e-05,
      "model_forward_time": 3.136293411254883,
      "step": 6812
    },
    {
      "epoch": 2.59857177734375e-05,
      "step": 6812,
      "training_step_time": 9.912071943283081
    },
    {
      "epoch": 2.59857177734375e-05,
      "model_forward_time": 3.139944314956665,
      "step": 6812
    },
    {
      "dynamics_loss_avg": 0.04942045882344246,
      "epoch": 2.59857177734375e-05,
      "step": 6812
    },
    {
      "action_loss_avg": 0.022556520206853746,
      "epoch": 2.59857177734375e-05,
      "step": 6812
    },
    {
      "epoch": 2.59857177734375e-05,
      "step": 6812,
      "training_step_time": 9.916029214859009
    },
    {
      "epoch": 2.59857177734375e-05,
      "model_forward_time": 3.1389589309692383,
      "step": 6812
    },
    {
      "epoch": 2.59857177734375e-05,
      "step": 6812,
      "training_step_time": 10.316442251205444
    },
    {
      "epoch": 2.5989532470703124e-05,
      "model_forward_time": 3.4015674591064453,
      "step": 6813
    },
    {
      "epoch": 2.5989532470703124e-05,
      "step": 6813,
      "training_step_time": 10.178773403167725
    },
    {
      "epoch": 2.5989532470703124e-05,
      "model_forward_time": 3.1426756381988525,
      "step": 6813
    },
    {
      "epoch": 2.5989532470703124e-05,
      "step": 6813,
      "training_step_time": 9.921123504638672
    },
    {
      "epoch": 2.5989532470703124e-05,
      "model_forward_time": 3.136046886444092,
      "step": 6813
    },
    {
      "epoch": 2.5989532470703124e-05,
      "step": 6813,
      "training_step_time": 9.917265892028809
    },
    {
      "epoch": 2.5989532470703124e-05,
      "model_forward_time": 3.139753818511963,
      "step": 6813
    },
    {
      "epoch": 2.5989532470703124e-05,
      "step": 6813,
      "training_step_time": 10.035761594772339
    },
    {
      "epoch": 2.599334716796875e-05,
      "model_forward_time": 3.1561408042907715,
      "step": 6814
    },
    {
      "epoch": 2.599334716796875e-05,
      "step": 6814,
      "training_step_time": 9.933774948120117
    },
    {
      "epoch": 2.599334716796875e-05,
      "model_forward_time": 3.1308741569519043,
      "step": 6814
    },
    {
      "epoch": 2.599334716796875e-05,
      "step": 6814,
      "training_step_time": 10.185256958007812
    },
    {
      "epoch": 2.599334716796875e-05,
      "model_forward_time": 3.13338565826416,
      "step": 6814
    },
    {
      "epoch": 2.599334716796875e-05,
      "step": 6814,
      "training_step_time": 9.909008502960205
    },
    {
      "epoch": 2.599334716796875e-05,
      "model_forward_time": 3.134981393814087,
      "step": 6814
    },
    {
      "epoch": 2.599334716796875e-05,
      "step": 6814,
      "training_step_time": 10.012736082077026
    },
    {
      "epoch": 2.5997161865234376e-05,
      "model_forward_time": 3.14949631690979,
      "step": 6815
    },
    {
      "dynamics_loss_avg": 0.07063901722431183,
      "epoch": 2.5997161865234376e-05,
      "step": 6815
    },
    {
      "action_loss_avg": 0.027508560847491027,
      "epoch": 2.5997161865234376e-05,
      "step": 6815
    },
    {
      "epoch": 2.5997161865234376e-05,
      "step": 6815,
      "training_step_time": 9.927387714385986
    },
    {
      "epoch": 2.5997161865234376e-05,
      "model_forward_time": 3.1298625469207764,
      "step": 6815
    },
    {
      "epoch": 2.5997161865234376e-05,
      "step": 6815,
      "training_step_time": 9.903687715530396
    },
    {
      "epoch": 2.5997161865234376e-05,
      "model_forward_time": 3.3957691192626953,
      "step": 6815
    },
    {
      "epoch": 2.5997161865234376e-05,
      "step": 6815,
      "training_step_time": 10.169442415237427
    },
    {
      "epoch": 2.5997161865234376e-05,
      "model_forward_time": 3.1411213874816895,
      "step": 6815
    },
    {
      "epoch": 2.5997161865234376e-05,
      "step": 6815,
      "training_step_time": 10.019521474838257
    },
    {
      "epoch": 2.60009765625e-05,
      "model_forward_time": 3.153175115585327,
      "step": 6816
    },
    {
      "epoch": 2.60009765625e-05,
      "step": 6816,
      "training_step_time": 9.933058023452759
    },
    {
      "epoch": 2.60009765625e-05,
      "model_forward_time": 3.1367413997650146,
      "step": 6816
    },
    {
      "epoch": 2.60009765625e-05,
      "step": 6816,
      "training_step_time": 9.913113117218018
    },
    {
      "epoch": 2.60009765625e-05,
      "model_forward_time": 3.137521743774414,
      "step": 6816
    },
    {
      "epoch": 2.60009765625e-05,
      "step": 6816,
      "training_step_time": 9.914345026016235
    },
    {
      "epoch": 2.60009765625e-05,
      "model_forward_time": 3.134857416152954,
      "step": 6816
    },
    {
      "epoch": 2.60009765625e-05,
      "step": 6816,
      "training_step_time": 10.295288324356079
    },
    {
      "epoch": 2.6004791259765626e-05,
      "model_forward_time": 3.15224552154541,
      "step": 6817
    },
    {
      "epoch": 2.6004791259765626e-05,
      "step": 6817,
      "training_step_time": 9.928105592727661
    },
    {
      "epoch": 2.6004791259765626e-05,
      "model_forward_time": 3.1317996978759766,
      "step": 6817
    },
    {
      "epoch": 2.6004791259765626e-05,
      "step": 6817,
      "training_step_time": 9.90381908416748
    },
    {
      "epoch": 2.6004791259765626e-05,
      "model_forward_time": 3.1393513679504395,
      "step": 6817
    },
    {
      "dynamics_loss_avg": 0.06502988412976266,
      "epoch": 2.6004791259765626e-05,
      "step": 6817
    },
    {
      "action_loss_avg": 0.021116551477462053,
      "epoch": 2.6004791259765626e-05,
      "step": 6817
    },
    {
      "epoch": 2.6004791259765626e-05,
      "step": 6817,
      "training_step_time": 9.91623044013977
    },
    {
      "epoch": 2.6004791259765626e-05,
      "model_forward_time": 3.1344645023345947,
      "step": 6817
    },
    {
      "epoch": 2.6004791259765626e-05,
      "step": 6817,
      "training_step_time": 10.281172037124634
    },
    {
      "epoch": 2.600860595703125e-05,
      "model_forward_time": 3.4505505561828613,
      "step": 6818
    },
    {
      "epoch": 2.600860595703125e-05,
      "step": 6818,
      "training_step_time": 10.222727298736572
    },
    {
      "epoch": 2.600860595703125e-05,
      "model_forward_time": 3.1314480304718018,
      "step": 6818
    },
    {
      "epoch": 2.600860595703125e-05,
      "step": 6818,
      "training_step_time": 9.90103816986084
    },
    {
      "epoch": 2.600860595703125e-05,
      "model_forward_time": 3.1338605880737305,
      "step": 6818
    },
    {
      "epoch": 2.600860595703125e-05,
      "step": 6818,
      "training_step_time": 9.903777122497559
    },
    {
      "epoch": 2.600860595703125e-05,
      "model_forward_time": 3.1281020641326904,
      "step": 6818
    },
    {
      "epoch": 2.600860595703125e-05,
      "step": 6818,
      "training_step_time": 10.306365966796875
    },
    {
      "epoch": 2.6012420654296875e-05,
      "model_forward_time": 3.15087890625,
      "step": 6819
    },
    {
      "epoch": 2.6012420654296875e-05,
      "step": 6819,
      "training_step_time": 9.923528671264648
    },
    {
      "epoch": 2.6012420654296875e-05,
      "model_forward_time": 3.1347532272338867,
      "step": 6819
    },
    {
      "epoch": 2.6012420654296875e-05,
      "step": 6819,
      "training_step_time": 10.186546564102173
    },
    {
      "epoch": 2.6012420654296875e-05,
      "model_forward_time": 3.1391730308532715,
      "step": 6819
    },
    {
      "epoch": 2.6012420654296875e-05,
      "step": 6819,
      "training_step_time": 9.910567998886108
    },
    {
      "epoch": 2.6012420654296875e-05,
      "model_forward_time": 3.1348538398742676,
      "step": 6819
    },
    {
      "epoch": 2.6012420654296875e-05,
      "step": 6819,
      "training_step_time": 10.297428846359253
    },
    {
      "epoch": 2.60162353515625e-05,
      "grad_norm": 0.39611873030662537,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 6820
    },
    {
      "epoch": 2.60162353515625e-05,
      "model_forward_time": 3.1706035137176514,
      "step": 6820
    },
    {
      "dynamics_loss_avg": 0.047151712514460084,
      "epoch": 2.60162353515625e-05,
      "step": 6820
    },
    {
      "action_loss_avg": 0.01782599538564682,
      "epoch": 2.60162353515625e-05,
      "step": 6820
    },
    {
      "epoch": 2.60162353515625e-05,
      "step": 6820,
      "training_step_time": 9.951580047607422
    },
    {
      "epoch": 2.60162353515625e-05,
      "model_forward_time": 3.142972707748413,
      "step": 6820
    },
    {
      "epoch": 2.60162353515625e-05,
      "step": 6820,
      "training_step_time": 9.91910696029663
    },
    {
      "epoch": 2.60162353515625e-05,
      "model_forward_time": 3.4239723682403564,
      "step": 6820
    },
    {
      "epoch": 2.60162353515625e-05,
      "step": 6820,
      "training_step_time": 10.19919753074646
    },
    {
      "epoch": 2.60162353515625e-05,
      "model_forward_time": 3.1385996341705322,
      "step": 6820
    },
    {
      "epoch": 2.60162353515625e-05,
      "step": 6820,
      "training_step_time": 10.014340877532959
    },
    {
      "epoch": 2.6020050048828124e-05,
      "model_forward_time": 3.164912700653076,
      "step": 6821
    },
    {
      "epoch": 2.6020050048828124e-05,
      "step": 6821,
      "training_step_time": 9.954347372055054
    },
    {
      "epoch": 2.6020050048828124e-05,
      "model_forward_time": 3.1444578170776367,
      "step": 6821
    },
    {
      "epoch": 2.6020050048828124e-05,
      "step": 6821,
      "training_step_time": 9.918431043624878
    },
    {
      "epoch": 2.6020050048828124e-05,
      "model_forward_time": 3.1405460834503174,
      "step": 6821
    },
    {
      "epoch": 2.6020050048828124e-05,
      "step": 6821,
      "training_step_time": 9.913616180419922
    },
    {
      "epoch": 2.6020050048828124e-05,
      "model_forward_time": 3.4063634872436523,
      "step": 6821
    },
    {
      "epoch": 2.6020050048828124e-05,
      "step": 6821,
      "training_step_time": 10.2885901927948
    },
    {
      "epoch": 2.602386474609375e-05,
      "model_forward_time": 3.1649487018585205,
      "step": 6822
    },
    {
      "epoch": 2.602386474609375e-05,
      "step": 6822,
      "training_step_time": 9.943467855453491
    },
    {
      "epoch": 2.602386474609375e-05,
      "model_forward_time": 3.145550012588501,
      "step": 6822
    },
    {
      "epoch": 2.602386474609375e-05,
      "step": 6822,
      "training_step_time": 9.923583745956421
    },
    {
      "epoch": 2.602386474609375e-05,
      "model_forward_time": 3.1477701663970947,
      "step": 6822
    },
    {
      "dynamics_loss_avg": 0.056520835123956205,
      "epoch": 2.602386474609375e-05,
      "step": 6822
    },
    {
      "action_loss_avg": 0.01887996927835047,
      "epoch": 2.602386474609375e-05,
      "step": 6822
    },
    {
      "epoch": 2.602386474609375e-05,
      "step": 6822,
      "training_step_time": 9.927358150482178
    },
    {
      "epoch": 2.602386474609375e-05,
      "model_forward_time": 3.1467814445495605,
      "step": 6822
    },
    {
      "epoch": 2.602386474609375e-05,
      "step": 6822,
      "training_step_time": 10.030595302581787
    },
    {
      "epoch": 2.6027679443359377e-05,
      "model_forward_time": 3.445812225341797,
      "step": 6823
    },
    {
      "epoch": 2.6027679443359377e-05,
      "step": 6823,
      "training_step_time": 10.227723121643066
    },
    {
      "epoch": 2.6027679443359377e-05,
      "model_forward_time": 3.146897315979004,
      "step": 6823
    },
    {
      "epoch": 2.6027679443359377e-05,
      "step": 6823,
      "training_step_time": 9.920837879180908
    },
    {
      "epoch": 2.6027679443359377e-05,
      "model_forward_time": 3.1407885551452637,
      "step": 6823
    },
    {
      "epoch": 2.6027679443359377e-05,
      "step": 6823,
      "training_step_time": 9.910399913787842
    },
    {
      "epoch": 2.6027679443359377e-05,
      "model_forward_time": 3.1468143463134766,
      "step": 6823
    },
    {
      "epoch": 2.6027679443359377e-05,
      "step": 6823,
      "training_step_time": 10.266680002212524
    },
    {
      "epoch": 2.6031494140625e-05,
      "model_forward_time": 3.1495015621185303,
      "step": 6824
    },
    {
      "epoch": 2.6031494140625e-05,
      "step": 6824,
      "training_step_time": 9.921239852905273
    },
    {
      "epoch": 2.6031494140625e-05,
      "model_forward_time": 3.427962064743042,
      "step": 6824
    },
    {
      "epoch": 2.6031494140625e-05,
      "step": 6824,
      "training_step_time": 10.199870824813843
    },
    {
      "epoch": 2.6031494140625e-05,
      "model_forward_time": 3.145559310913086,
      "step": 6824
    },
    {
      "epoch": 2.6031494140625e-05,
      "step": 6824,
      "training_step_time": 9.927584648132324
    },
    {
      "epoch": 2.6031494140625e-05,
      "model_forward_time": 3.146498680114746,
      "step": 6824
    },
    {
      "epoch": 2.6031494140625e-05,
      "step": 6824,
      "training_step_time": 10.029488563537598
    },
    {
      "epoch": 2.6035308837890626e-05,
      "model_forward_time": 3.166747570037842,
      "step": 6825
    },
    {
      "dynamics_loss_avg": 0.06185982115566731,
      "epoch": 2.6035308837890626e-05,
      "step": 6825
    },
    {
      "action_loss_avg": 0.021965521946549417,
      "epoch": 2.6035308837890626e-05,
      "step": 6825
    },
    {
      "epoch": 2.6035308837890626e-05,
      "step": 6825,
      "training_step_time": 9.948360443115234
    },
    {
      "epoch": 2.6035308837890626e-05,
      "model_forward_time": 3.1395251750946045,
      "step": 6825
    },
    {
      "epoch": 2.6035308837890626e-05,
      "step": 6825,
      "training_step_time": 9.919025182723999
    },
    {
      "epoch": 2.6035308837890626e-05,
      "model_forward_time": 3.148793935775757,
      "step": 6825
    },
    {
      "epoch": 2.6035308837890626e-05,
      "step": 6825,
      "training_step_time": 10.209557056427002
    },
    {
      "epoch": 2.6035308837890626e-05,
      "model_forward_time": 3.1532087326049805,
      "step": 6825
    },
    {
      "epoch": 2.6035308837890626e-05,
      "step": 6825,
      "training_step_time": 10.314573049545288
    },
    {
      "epoch": 2.603912353515625e-05,
      "model_forward_time": 3.1593973636627197,
      "step": 6826
    },
    {
      "epoch": 2.603912353515625e-05,
      "step": 6826,
      "training_step_time": 9.944427013397217
    },
    {
      "epoch": 2.603912353515625e-05,
      "model_forward_time": 3.1510720252990723,
      "step": 6826
    },
    {
      "epoch": 2.603912353515625e-05,
      "step": 6826,
      "training_step_time": 9.929087162017822
    },
    {
      "epoch": 2.603912353515625e-05,
      "model_forward_time": 3.148155689239502,
      "step": 6826
    },
    {
      "epoch": 2.603912353515625e-05,
      "step": 6826,
      "training_step_time": 9.914877891540527
    },
    {
      "epoch": 2.603912353515625e-05,
      "model_forward_time": 3.4283032417297363,
      "step": 6826
    },
    {
      "epoch": 2.603912353515625e-05,
      "step": 6826,
      "training_step_time": 10.29646348953247
    },
    {
      "epoch": 2.6042938232421875e-05,
      "model_forward_time": 3.1545491218566895,
      "step": 6827
    },
    {
      "epoch": 2.6042938232421875e-05,
      "step": 6827,
      "training_step_time": 9.925348043441772
    },
    {
      "epoch": 2.6042938232421875e-05,
      "model_forward_time": 3.1359050273895264,
      "step": 6827
    },
    {
      "epoch": 2.6042938232421875e-05,
      "step": 6827,
      "training_step_time": 9.90911340713501
    },
    {
      "epoch": 2.6042938232421875e-05,
      "model_forward_time": 3.1432390213012695,
      "step": 6827
    },
    {
      "dynamics_loss_avg": 0.06022821720689535,
      "epoch": 2.6042938232421875e-05,
      "step": 6827
    },
    {
      "action_loss_avg": 0.022637419356033205,
      "epoch": 2.6042938232421875e-05,
      "step": 6827
    },
    {
      "epoch": 2.6042938232421875e-05,
      "step": 6827,
      "training_step_time": 9.910428047180176
    },
    {
      "epoch": 2.6042938232421875e-05,
      "model_forward_time": 3.1377811431884766,
      "step": 6827
    },
    {
      "epoch": 2.6042938232421875e-05,
      "step": 6827,
      "training_step_time": 10.002669095993042
    },
    {
      "epoch": 2.60467529296875e-05,
      "model_forward_time": 3.1477737426757812,
      "step": 6828
    },
    {
      "epoch": 2.60467529296875e-05,
      "step": 6828,
      "training_step_time": 10.190177917480469
    },
    {
      "epoch": 2.60467529296875e-05,
      "model_forward_time": 3.140855073928833,
      "step": 6828
    },
    {
      "epoch": 2.60467529296875e-05,
      "step": 6828,
      "training_step_time": 9.911570310592651
    },
    {
      "epoch": 2.60467529296875e-05,
      "model_forward_time": 3.143841505050659,
      "step": 6828
    },
    {
      "epoch": 2.60467529296875e-05,
      "step": 6828,
      "training_step_time": 9.915567398071289
    },
    {
      "epoch": 2.60467529296875e-05,
      "model_forward_time": 3.1410274505615234,
      "step": 6828
    },
    {
      "epoch": 2.60467529296875e-05,
      "step": 6828,
      "training_step_time": 10.257774591445923
    },
    {
      "epoch": 2.6050567626953124e-05,
      "model_forward_time": 3.1533327102661133,
      "step": 6829
    },
    {
      "epoch": 2.6050567626953124e-05,
      "step": 6829,
      "training_step_time": 9.93007755279541
    },
    {
      "epoch": 2.6050567626953124e-05,
      "model_forward_time": 3.4031708240509033,
      "step": 6829
    },
    {
      "epoch": 2.6050567626953124e-05,
      "step": 6829,
      "training_step_time": 10.178752899169922
    },
    {
      "epoch": 2.6050567626953124e-05,
      "model_forward_time": 3.1484930515289307,
      "step": 6829
    },
    {
      "epoch": 2.6050567626953124e-05,
      "step": 6829,
      "training_step_time": 9.925919771194458
    },
    {
      "epoch": 2.6050567626953124e-05,
      "model_forward_time": 3.145965576171875,
      "step": 6829
    },
    {
      "epoch": 2.6050567626953124e-05,
      "step": 6829,
      "training_step_time": 10.299039363861084
    },
    {
      "epoch": 2.605438232421875e-05,
      "grad_norm": 0.32859358191490173,
      "learning_rate": 1e-05,
      "loss": 0.0784,
      "step": 6830
    },
    {
      "epoch": 2.605438232421875e-05,
      "model_forward_time": 2.479670763015747,
      "step": 6830
    },
    {
      "dynamics_loss_avg": 0.05879141874611378,
      "epoch": 2.605438232421875e-05,
      "step": 6830
    },
    {
      "action_loss_avg": 0.02485669367015362,
      "epoch": 2.605438232421875e-05,
      "step": 6830
    },
    {
      "epoch": 2.605438232421875e-05,
      "step": 6830,
      "training_step_time": 7.510861873626709
    },
    {
      "epoch": 2.605438232421875e-05,
      "model_forward_time": 3.1441731452941895,
      "step": 6830
    },
    {
      "epoch": 2.605438232421875e-05,
      "step": 6830,
      "training_step_time": 9.914111614227295
    },
    {
      "epoch": 2.605438232421875e-05,
      "model_forward_time": 3.1400957107543945,
      "step": 6830
    },
    {
      "epoch": 2.605438232421875e-05,
      "step": 6830,
      "training_step_time": 10.177263736724854
    },
    {
      "epoch": 2.605438232421875e-05,
      "model_forward_time": 3.1390323638916016,
      "step": 6830
    },
    {
      "epoch": 2.605438232421875e-05,
      "step": 6830,
      "training_step_time": 10.002563238143921
    },
    {
      "epoch": 2.6058197021484377e-05,
      "model_forward_time": 3.159740447998047,
      "step": 6831
    },
    {
      "epoch": 2.6058197021484377e-05,
      "step": 6831,
      "training_step_time": 9.933093786239624
    },
    {
      "epoch": 2.6058197021484377e-05,
      "model_forward_time": 3.1386680603027344,
      "step": 6831
    },
    {
      "epoch": 2.6058197021484377e-05,
      "step": 6831,
      "training_step_time": 9.907079458236694
    },
    {
      "epoch": 2.6058197021484377e-05,
      "model_forward_time": 3.1373536586761475,
      "step": 6831
    },
    {
      "epoch": 2.6058197021484377e-05,
      "step": 6831,
      "training_step_time": 9.910883903503418
    },
    {
      "epoch": 2.6058197021484377e-05,
      "model_forward_time": 3.402981996536255,
      "step": 6831
    },
    {
      "epoch": 2.6058197021484377e-05,
      "step": 6831,
      "training_step_time": 10.270014762878418
    },
    {
      "epoch": 2.606201171875e-05,
      "model_forward_time": 3.1604387760162354,
      "step": 6832
    },
    {
      "epoch": 2.606201171875e-05,
      "step": 6832,
      "training_step_time": 9.931817531585693
    },
    {
      "epoch": 2.606201171875e-05,
      "model_forward_time": 3.144686460494995,
      "step": 6832
    },
    {
      "epoch": 2.606201171875e-05,
      "step": 6832,
      "training_step_time": 9.919708490371704
    },
    {
      "epoch": 2.606201171875e-05,
      "model_forward_time": 3.141433000564575,
      "step": 6832
    },
    {
      "dynamics_loss_avg": 0.07613549456000328,
      "epoch": 2.606201171875e-05,
      "step": 6832
    },
    {
      "action_loss_avg": 0.029903862066566943,
      "epoch": 2.606201171875e-05,
      "step": 6832
    },
    {
      "epoch": 2.606201171875e-05,
      "step": 6832,
      "training_step_time": 9.920053482055664
    },
    {
      "epoch": 2.606201171875e-05,
      "model_forward_time": 3.146425724029541,
      "step": 6832
    },
    {
      "epoch": 2.606201171875e-05,
      "step": 6832,
      "training_step_time": 10.253182888031006
    },
    {
      "epoch": 2.6065826416015626e-05,
      "model_forward_time": 3.423083543777466,
      "step": 6833
    },
    {
      "epoch": 2.6065826416015626e-05,
      "step": 6833,
      "training_step_time": 10.199116230010986
    },
    {
      "epoch": 2.6065826416015626e-05,
      "model_forward_time": 3.150564670562744,
      "step": 6833
    },
    {
      "epoch": 2.6065826416015626e-05,
      "step": 6833,
      "training_step_time": 9.926324605941772
    },
    {
      "epoch": 2.6065826416015626e-05,
      "model_forward_time": 3.1484200954437256,
      "step": 6833
    },
    {
      "epoch": 2.6065826416015626e-05,
      "step": 6833,
      "training_step_time": 9.926998615264893
    },
    {
      "epoch": 2.6065826416015626e-05,
      "model_forward_time": 3.1481969356536865,
      "step": 6833
    },
    {
      "epoch": 2.6065826416015626e-05,
      "step": 6833,
      "training_step_time": 10.273065567016602
    },
    {
      "epoch": 2.606964111328125e-05,
      "model_forward_time": 3.155864715576172,
      "step": 6834
    },
    {
      "epoch": 2.606964111328125e-05,
      "step": 6834,
      "training_step_time": 9.936245441436768
    },
    {
      "epoch": 2.606964111328125e-05,
      "model_forward_time": 3.1444504261016846,
      "step": 6834
    },
    {
      "epoch": 2.606964111328125e-05,
      "step": 6834,
      "training_step_time": 10.238145112991333
    },
    {
      "epoch": 2.606964111328125e-05,
      "model_forward_time": 3.141521453857422,
      "step": 6834
    },
    {
      "epoch": 2.606964111328125e-05,
      "step": 6834,
      "training_step_time": 9.921730518341064
    },
    {
      "epoch": 2.606964111328125e-05,
      "model_forward_time": 3.150264024734497,
      "step": 6834
    },
    {
      "epoch": 2.606964111328125e-05,
      "step": 6834,
      "training_step_time": 10.255651235580444
    },
    {
      "epoch": 2.6073455810546875e-05,
      "model_forward_time": 3.1508452892303467,
      "step": 6835
    },
    {
      "dynamics_loss_avg": 0.0473457065410912,
      "epoch": 2.6073455810546875e-05,
      "step": 6835
    },
    {
      "action_loss_avg": 0.017843074211850763,
      "epoch": 2.6073455810546875e-05,
      "step": 6835
    },
    {
      "epoch": 2.6073455810546875e-05,
      "step": 6835,
      "training_step_time": 9.928737163543701
    },
    {
      "epoch": 2.6073455810546875e-05,
      "model_forward_time": 3.1455016136169434,
      "step": 6835
    },
    {
      "epoch": 2.6073455810546875e-05,
      "step": 6835,
      "training_step_time": 9.922868013381958
    },
    {
      "epoch": 2.6073455810546875e-05,
      "model_forward_time": 3.4102256298065186,
      "step": 6835
    },
    {
      "epoch": 2.6073455810546875e-05,
      "step": 6835,
      "training_step_time": 10.189567565917969
    },
    {
      "epoch": 2.6073455810546875e-05,
      "model_forward_time": 3.1492462158203125,
      "step": 6835
    },
    {
      "epoch": 2.6073455810546875e-05,
      "step": 6835,
      "training_step_time": 10.02713656425476
    },
    {
      "epoch": 2.60772705078125e-05,
      "model_forward_time": 3.153635025024414,
      "step": 6836
    },
    {
      "epoch": 2.60772705078125e-05,
      "step": 6836,
      "training_step_time": 9.934334754943848
    },
    {
      "epoch": 2.60772705078125e-05,
      "model_forward_time": 3.1430885791778564,
      "step": 6836
    },
    {
      "epoch": 2.60772705078125e-05,
      "step": 6836,
      "training_step_time": 9.922999858856201
    },
    {
      "epoch": 2.60772705078125e-05,
      "model_forward_time": 3.1539645195007324,
      "step": 6836
    },
    {
      "epoch": 2.60772705078125e-05,
      "step": 6836,
      "training_step_time": 9.93456768989563
    },
    {
      "epoch": 2.60772705078125e-05,
      "model_forward_time": 3.1436030864715576,
      "step": 6836
    },
    {
      "epoch": 2.60772705078125e-05,
      "step": 6836,
      "training_step_time": 10.309782981872559
    },
    {
      "epoch": 2.6081085205078124e-05,
      "model_forward_time": 3.162523031234741,
      "step": 6837
    },
    {
      "epoch": 2.6081085205078124e-05,
      "step": 6837,
      "training_step_time": 9.943838119506836
    },
    {
      "epoch": 2.6081085205078124e-05,
      "model_forward_time": 3.146343469619751,
      "step": 6837
    },
    {
      "epoch": 2.6081085205078124e-05,
      "step": 6837,
      "training_step_time": 9.925740003585815
    },
    {
      "epoch": 2.6081085205078124e-05,
      "model_forward_time": 3.143489360809326,
      "step": 6837
    },
    {
      "dynamics_loss_avg": 0.0568853709846735,
      "epoch": 2.6081085205078124e-05,
      "step": 6837
    },
    {
      "action_loss_avg": 0.023537169117480515,
      "epoch": 2.6081085205078124e-05,
      "step": 6837
    },
    {
      "epoch": 2.6081085205078124e-05,
      "step": 6837,
      "training_step_time": 9.922950267791748
    },
    {
      "epoch": 2.6081085205078124e-05,
      "model_forward_time": 3.138261556625366,
      "step": 6837
    },
    {
      "epoch": 2.6081085205078124e-05,
      "step": 6837,
      "training_step_time": 10.286305665969849
    },
    {
      "epoch": 2.608489990234375e-05,
      "model_forward_time": 3.430934190750122,
      "step": 6838
    },
    {
      "epoch": 2.608489990234375e-05,
      "step": 6838,
      "training_step_time": 10.199422121047974
    },
    {
      "epoch": 2.608489990234375e-05,
      "model_forward_time": 3.142583131790161,
      "step": 6838
    },
    {
      "epoch": 2.608489990234375e-05,
      "step": 6838,
      "training_step_time": 9.916620016098022
    },
    {
      "epoch": 2.608489990234375e-05,
      "model_forward_time": 3.14674973487854,
      "step": 6838
    },
    {
      "epoch": 2.608489990234375e-05,
      "step": 6838,
      "training_step_time": 9.921207666397095
    },
    {
      "epoch": 2.608489990234375e-05,
      "model_forward_time": 3.143620252609253,
      "step": 6838
    },
    {
      "epoch": 2.608489990234375e-05,
      "step": 6838,
      "training_step_time": 10.014897584915161
    },
    {
      "epoch": 2.6088714599609377e-05,
      "model_forward_time": 3.158869743347168,
      "step": 6839
    },
    {
      "epoch": 2.6088714599609377e-05,
      "step": 6839,
      "training_step_time": 9.936060667037964
    },
    {
      "epoch": 2.6088714599609377e-05,
      "model_forward_time": 3.147542953491211,
      "step": 6839
    },
    {
      "epoch": 2.6088714599609377e-05,
      "step": 6839,
      "training_step_time": 10.19432783126831
    },
    {
      "epoch": 2.6088714599609377e-05,
      "model_forward_time": 3.148798942565918,
      "step": 6839
    },
    {
      "epoch": 2.6088714599609377e-05,
      "step": 6839,
      "training_step_time": 9.924514055252075
    },
    {
      "epoch": 2.6088714599609377e-05,
      "model_forward_time": 3.1437251567840576,
      "step": 6839
    },
    {
      "epoch": 2.6088714599609377e-05,
      "step": 6839,
      "training_step_time": 10.030602216720581
    },
    {
      "epoch": 2.6092529296875e-05,
      "grad_norm": 0.33396902680397034,
      "learning_rate": 1e-05,
      "loss": 0.0844,
      "step": 6840
    },
    {
      "epoch": 2.6092529296875e-05,
      "model_forward_time": 3.157872438430786,
      "step": 6840
    },
    {
      "dynamics_loss_avg": 0.06266312841325998,
      "epoch": 2.6092529296875e-05,
      "step": 6840
    },
    {
      "action_loss_avg": 0.023319906182587148,
      "epoch": 2.6092529296875e-05,
      "step": 6840
    },
    {
      "epoch": 2.6092529296875e-05,
      "step": 6840,
      "training_step_time": 9.942072868347168
    },
    {
      "epoch": 2.6092529296875e-05,
      "model_forward_time": 3.1491057872772217,
      "step": 6840
    },
    {
      "epoch": 2.6092529296875e-05,
      "step": 6840,
      "training_step_time": 9.926167249679565
    },
    {
      "epoch": 2.6092529296875e-05,
      "model_forward_time": 3.4266273975372314,
      "step": 6840
    },
    {
      "epoch": 2.6092529296875e-05,
      "step": 6840,
      "training_step_time": 10.204609155654907
    },
    {
      "epoch": 2.6092529296875e-05,
      "model_forward_time": 3.145750045776367,
      "step": 6840
    },
    {
      "epoch": 2.6092529296875e-05,
      "step": 6840,
      "training_step_time": 10.29110836982727
    },
    {
      "epoch": 2.6096343994140626e-05,
      "model_forward_time": 3.17881178855896,
      "step": 6841
    },
    {
      "epoch": 2.6096343994140626e-05,
      "step": 6841,
      "training_step_time": 9.952779769897461
    },
    {
      "epoch": 2.6096343994140626e-05,
      "model_forward_time": 3.1442885398864746,
      "step": 6841
    },
    {
      "epoch": 2.6096343994140626e-05,
      "step": 6841,
      "training_step_time": 9.918513774871826
    },
    {
      "epoch": 2.6096343994140626e-05,
      "model_forward_time": 3.145665168762207,
      "step": 6841
    },
    {
      "epoch": 2.6096343994140626e-05,
      "step": 6841,
      "training_step_time": 9.920121669769287
    },
    {
      "epoch": 2.6096343994140626e-05,
      "model_forward_time": 3.409341812133789,
      "step": 6841
    },
    {
      "epoch": 2.6096343994140626e-05,
      "step": 6841,
      "training_step_time": 10.278071403503418
    },
    {
      "epoch": 2.610015869140625e-05,
      "model_forward_time": 3.1496901512145996,
      "step": 6842
    },
    {
      "epoch": 2.610015869140625e-05,
      "step": 6842,
      "training_step_time": 9.924375772476196
    },
    {
      "epoch": 2.610015869140625e-05,
      "model_forward_time": 3.1411101818084717,
      "step": 6842
    },
    {
      "epoch": 2.610015869140625e-05,
      "step": 6842,
      "training_step_time": 9.914030075073242
    },
    {
      "epoch": 2.610015869140625e-05,
      "model_forward_time": 3.139023780822754,
      "step": 6842
    },
    {
      "dynamics_loss_avg": 0.05165970679372549,
      "epoch": 2.610015869140625e-05,
      "step": 6842
    },
    {
      "action_loss_avg": 0.019771629013121128,
      "epoch": 2.610015869140625e-05,
      "step": 6842
    },
    {
      "epoch": 2.610015869140625e-05,
      "step": 6842,
      "training_step_time": 9.912962913513184
    },
    {
      "epoch": 2.610015869140625e-05,
      "model_forward_time": 3.1418471336364746,
      "step": 6842
    },
    {
      "epoch": 2.610015869140625e-05,
      "step": 6842,
      "training_step_time": 10.291643619537354
    },
    {
      "epoch": 2.6103973388671875e-05,
      "model_forward_time": 3.4088308811187744,
      "step": 6843
    },
    {
      "epoch": 2.6103973388671875e-05,
      "step": 6843,
      "training_step_time": 10.180622816085815
    },
    {
      "epoch": 2.6103973388671875e-05,
      "model_forward_time": 3.144040822982788,
      "step": 6843
    },
    {
      "epoch": 2.6103973388671875e-05,
      "step": 6843,
      "training_step_time": 9.91253924369812
    },
    {
      "epoch": 2.6103973388671875e-05,
      "model_forward_time": 3.140014886856079,
      "step": 6843
    },
    {
      "epoch": 2.6103973388671875e-05,
      "step": 6843,
      "training_step_time": 9.911781787872314
    },
    {
      "epoch": 2.6103973388671875e-05,
      "model_forward_time": 3.139876127243042,
      "step": 6843
    },
    {
      "epoch": 2.6103973388671875e-05,
      "step": 6843,
      "training_step_time": 10.013973236083984
    },
    {
      "epoch": 2.61077880859375e-05,
      "model_forward_time": 3.1561193466186523,
      "step": 6844
    },
    {
      "epoch": 2.61077880859375e-05,
      "step": 6844,
      "training_step_time": 9.929848670959473
    },
    {
      "epoch": 2.61077880859375e-05,
      "model_forward_time": 3.4240660667419434,
      "step": 6844
    },
    {
      "epoch": 2.61077880859375e-05,
      "step": 6844,
      "training_step_time": 10.19462776184082
    },
    {
      "epoch": 2.61077880859375e-05,
      "model_forward_time": 3.1390440464019775,
      "step": 6844
    },
    {
      "epoch": 2.61077880859375e-05,
      "step": 6844,
      "training_step_time": 9.906913995742798
    },
    {
      "epoch": 2.61077880859375e-05,
      "model_forward_time": 3.1428749561309814,
      "step": 6844
    },
    {
      "epoch": 2.61077880859375e-05,
      "step": 6844,
      "training_step_time": 10.326272964477539
    },
    {
      "epoch": 2.6111602783203124e-05,
      "model_forward_time": 3.1511664390563965,
      "step": 6845
    },
    {
      "dynamics_loss_avg": 0.06941265370696784,
      "epoch": 2.6111602783203124e-05,
      "step": 6845
    },
    {
      "action_loss_avg": 0.02413079785183072,
      "epoch": 2.6111602783203124e-05,
      "step": 6845
    },
    {
      "epoch": 2.6111602783203124e-05,
      "step": 6845,
      "training_step_time": 9.926674604415894
    },
    {
      "epoch": 2.6111602783203124e-05,
      "model_forward_time": 3.141373634338379,
      "step": 6845
    },
    {
      "epoch": 2.6111602783203124e-05,
      "step": 6845,
      "training_step_time": 9.913631677627563
    },
    {
      "epoch": 2.6111602783203124e-05,
      "model_forward_time": 3.145347833633423,
      "step": 6845
    },
    {
      "epoch": 2.6111602783203124e-05,
      "step": 6845,
      "training_step_time": 10.21237850189209
    },
    {
      "epoch": 2.6111602783203124e-05,
      "model_forward_time": 3.145080327987671,
      "step": 6845
    },
    {
      "epoch": 2.6111602783203124e-05,
      "step": 6845,
      "training_step_time": 10.02393889427185
    },
    {
      "epoch": 2.611541748046875e-05,
      "model_forward_time": 3.160132884979248,
      "step": 6846
    },
    {
      "epoch": 2.611541748046875e-05,
      "step": 6846,
      "training_step_time": 9.936584949493408
    },
    {
      "epoch": 2.611541748046875e-05,
      "model_forward_time": 3.1376845836639404,
      "step": 6846
    },
    {
      "epoch": 2.611541748046875e-05,
      "step": 6846,
      "training_step_time": 9.914650678634644
    },
    {
      "epoch": 2.611541748046875e-05,
      "model_forward_time": 3.1477081775665283,
      "step": 6846
    },
    {
      "epoch": 2.611541748046875e-05,
      "step": 6846,
      "training_step_time": 9.923335790634155
    },
    {
      "epoch": 2.611541748046875e-05,
      "model_forward_time": 3.4264049530029297,
      "step": 6846
    },
    {
      "epoch": 2.611541748046875e-05,
      "step": 6846,
      "training_step_time": 10.299105405807495
    },
    {
      "epoch": 2.6119232177734377e-05,
      "model_forward_time": 3.1555113792419434,
      "step": 6847
    },
    {
      "epoch": 2.6119232177734377e-05,
      "step": 6847,
      "training_step_time": 9.933459997177124
    },
    {
      "epoch": 2.6119232177734377e-05,
      "model_forward_time": 3.140859842300415,
      "step": 6847
    },
    {
      "epoch": 2.6119232177734377e-05,
      "step": 6847,
      "training_step_time": 9.917248249053955
    },
    {
      "epoch": 2.6119232177734377e-05,
      "model_forward_time": 3.144709348678589,
      "step": 6847
    },
    {
      "dynamics_loss_avg": 0.05243172980844975,
      "epoch": 2.6119232177734377e-05,
      "step": 6847
    },
    {
      "action_loss_avg": 0.019967377185821533,
      "epoch": 2.6119232177734377e-05,
      "step": 6847
    },
    {
      "epoch": 2.6119232177734377e-05,
      "step": 6847,
      "training_step_time": 9.923659801483154
    },
    {
      "epoch": 2.6119232177734377e-05,
      "model_forward_time": 3.1403050422668457,
      "step": 6847
    },
    {
      "epoch": 2.6119232177734377e-05,
      "step": 6847,
      "training_step_time": 10.305240631103516
    },
    {
      "epoch": 2.6123046875e-05,
      "model_forward_time": 3.216418504714966,
      "step": 6848
    },
    {
      "epoch": 2.6123046875e-05,
      "step": 6848,
      "training_step_time": 10.270397663116455
    },
    {
      "epoch": 2.6123046875e-05,
      "model_forward_time": 3.1403262615203857,
      "step": 6848
    },
    {
      "epoch": 2.6123046875e-05,
      "step": 6848,
      "training_step_time": 9.913996934890747
    },
    {
      "epoch": 2.6123046875e-05,
      "model_forward_time": 3.1451380252838135,
      "step": 6848
    },
    {
      "epoch": 2.6123046875e-05,
      "step": 6848,
      "training_step_time": 9.918402194976807
    },
    {
      "epoch": 2.6123046875e-05,
      "model_forward_time": 3.14376163482666,
      "step": 6848
    },
    {
      "epoch": 2.6123046875e-05,
      "step": 6848,
      "training_step_time": 10.025980472564697
    },
    {
      "epoch": 2.6126861572265626e-05,
      "model_forward_time": 3.155231237411499,
      "step": 6849
    },
    {
      "epoch": 2.6126861572265626e-05,
      "step": 6849,
      "training_step_time": 9.930395126342773
    },
    {
      "epoch": 2.6126861572265626e-05,
      "model_forward_time": 3.426673412322998,
      "step": 6849
    },
    {
      "epoch": 2.6126861572265626e-05,
      "step": 6849,
      "training_step_time": 10.196677923202515
    },
    {
      "epoch": 2.6126861572265626e-05,
      "model_forward_time": 3.1449244022369385,
      "step": 6849
    },
    {
      "epoch": 2.6126861572265626e-05,
      "step": 6849,
      "training_step_time": 9.919615745544434
    },
    {
      "epoch": 2.6126861572265626e-05,
      "model_forward_time": 3.1432716846466064,
      "step": 6849
    },
    {
      "epoch": 2.6126861572265626e-05,
      "step": 6849,
      "training_step_time": 10.262483358383179
    },
    {
      "epoch": 2.613067626953125e-05,
      "grad_norm": 0.5017421245574951,
      "learning_rate": 1e-05,
      "loss": 0.0829,
      "step": 6850
    },
    {
      "epoch": 2.613067626953125e-05,
      "model_forward_time": 3.1658833026885986,
      "step": 6850
    },
    {
      "dynamics_loss_avg": 0.06491523273289204,
      "epoch": 2.613067626953125e-05,
      "step": 6850
    },
    {
      "action_loss_avg": 0.031114376708865167,
      "epoch": 2.613067626953125e-05,
      "step": 6850
    },
    {
      "epoch": 2.613067626953125e-05,
      "step": 6850,
      "training_step_time": 9.944058895111084
    },
    {
      "epoch": 2.613067626953125e-05,
      "model_forward_time": 3.1473703384399414,
      "step": 6850
    },
    {
      "epoch": 2.613067626953125e-05,
      "step": 6850,
      "training_step_time": 9.917977333068848
    },
    {
      "epoch": 2.613067626953125e-05,
      "model_forward_time": 3.425114154815674,
      "step": 6850
    },
    {
      "epoch": 2.613067626953125e-05,
      "step": 6850,
      "training_step_time": 10.201058387756348
    },
    {
      "epoch": 2.613067626953125e-05,
      "model_forward_time": 3.1420140266418457,
      "step": 6850
    },
    {
      "epoch": 2.613067626953125e-05,
      "step": 6850,
      "training_step_time": 10.288633346557617
    },
    {
      "epoch": 2.6134490966796875e-05,
      "model_forward_time": 3.14974045753479,
      "step": 6851
    },
    {
      "epoch": 2.6134490966796875e-05,
      "step": 6851,
      "training_step_time": 9.924958229064941
    },
    {
      "epoch": 2.6134490966796875e-05,
      "model_forward_time": 3.145472764968872,
      "step": 6851
    },
    {
      "epoch": 2.6134490966796875e-05,
      "step": 6851,
      "training_step_time": 9.923190593719482
    },
    {
      "epoch": 2.6134490966796875e-05,
      "model_forward_time": 3.1362226009368896,
      "step": 6851
    },
    {
      "epoch": 2.6134490966796875e-05,
      "step": 6851,
      "training_step_time": 9.911540508270264
    },
    {
      "epoch": 2.6134490966796875e-05,
      "model_forward_time": 3.409533977508545,
      "step": 6851
    },
    {
      "epoch": 2.6134490966796875e-05,
      "step": 6851,
      "training_step_time": 10.290751218795776
    },
    {
      "epoch": 2.61383056640625e-05,
      "model_forward_time": 3.1572518348693848,
      "step": 6852
    },
    {
      "epoch": 2.61383056640625e-05,
      "step": 6852,
      "training_step_time": 9.933533668518066
    },
    {
      "epoch": 2.61383056640625e-05,
      "model_forward_time": 3.1419079303741455,
      "step": 6852
    },
    {
      "epoch": 2.61383056640625e-05,
      "step": 6852,
      "training_step_time": 9.913109302520752
    },
    {
      "epoch": 2.61383056640625e-05,
      "model_forward_time": 3.1351852416992188,
      "step": 6852
    },
    {
      "dynamics_loss_avg": 0.06298861987888812,
      "epoch": 2.61383056640625e-05,
      "step": 6852
    },
    {
      "action_loss_avg": 0.018004499841481447,
      "epoch": 2.61383056640625e-05,
      "step": 6852
    },
    {
      "epoch": 2.61383056640625e-05,
      "step": 6852,
      "training_step_time": 9.910467863082886
    },
    {
      "epoch": 2.61383056640625e-05,
      "model_forward_time": 3.141636610031128,
      "step": 6852
    },
    {
      "epoch": 2.61383056640625e-05,
      "step": 6852,
      "training_step_time": 10.27286148071289
    },
    {
      "epoch": 2.6142120361328124e-05,
      "model_forward_time": 3.4402801990509033,
      "step": 6853
    },
    {
      "epoch": 2.6142120361328124e-05,
      "step": 6853,
      "training_step_time": 10.2099449634552
    },
    {
      "epoch": 2.6142120361328124e-05,
      "model_forward_time": 3.1453089714050293,
      "step": 6853
    },
    {
      "epoch": 2.6142120361328124e-05,
      "step": 6853,
      "training_step_time": 9.917363405227661
    },
    {
      "epoch": 2.6142120361328124e-05,
      "model_forward_time": 3.1385669708251953,
      "step": 6853
    },
    {
      "epoch": 2.6142120361328124e-05,
      "step": 6853,
      "training_step_time": 9.907944679260254
    },
    {
      "epoch": 2.6142120361328124e-05,
      "model_forward_time": 3.146103858947754,
      "step": 6853
    },
    {
      "epoch": 2.6142120361328124e-05,
      "step": 6853,
      "training_step_time": 10.245965242385864
    },
    {
      "epoch": 2.614593505859375e-05,
      "model_forward_time": 3.152994155883789,
      "step": 6854
    },
    {
      "epoch": 2.614593505859375e-05,
      "step": 6854,
      "training_step_time": 9.93332314491272
    },
    {
      "epoch": 2.614593505859375e-05,
      "model_forward_time": 3.1433937549591064,
      "step": 6854
    },
    {
      "epoch": 2.614593505859375e-05,
      "step": 6854,
      "training_step_time": 10.203360080718994
    },
    {
      "epoch": 2.614593505859375e-05,
      "model_forward_time": 3.1451172828674316,
      "step": 6854
    },
    {
      "epoch": 2.614593505859375e-05,
      "step": 6854,
      "training_step_time": 9.922967195510864
    },
    {
      "epoch": 2.614593505859375e-05,
      "model_forward_time": 3.1441826820373535,
      "step": 6854
    },
    {
      "epoch": 2.614593505859375e-05,
      "step": 6854,
      "training_step_time": 10.01859164237976
    },
    {
      "epoch": 2.6149749755859377e-05,
      "model_forward_time": 3.164614200592041,
      "step": 6855
    },
    {
      "dynamics_loss_avg": 0.07418015468865632,
      "epoch": 2.6149749755859377e-05,
      "step": 6855
    },
    {
      "action_loss_avg": 0.03095485894009471,
      "epoch": 2.6149749755859377e-05,
      "step": 6855
    },
    {
      "epoch": 2.6149749755859377e-05,
      "step": 6855,
      "training_step_time": 9.944035768508911
    },
    {
      "epoch": 2.6149749755859377e-05,
      "model_forward_time": 3.140089988708496,
      "step": 6855
    },
    {
      "epoch": 2.6149749755859377e-05,
      "step": 6855,
      "training_step_time": 9.915997743606567
    },
    {
      "epoch": 2.6149749755859377e-05,
      "model_forward_time": 3.146329164505005,
      "step": 6855
    },
    {
      "epoch": 2.6149749755859377e-05,
      "step": 6855,
      "training_step_time": 10.206301927566528
    },
    {
      "epoch": 2.6149749755859377e-05,
      "model_forward_time": 3.1474335193634033,
      "step": 6855
    },
    {
      "epoch": 2.6149749755859377e-05,
      "step": 6855,
      "training_step_time": 10.287203550338745
    },
    {
      "epoch": 2.6153564453125e-05,
      "model_forward_time": 3.156860589981079,
      "step": 6856
    },
    {
      "epoch": 2.6153564453125e-05,
      "step": 6856,
      "training_step_time": 9.934121370315552
    },
    {
      "epoch": 2.6153564453125e-05,
      "model_forward_time": 3.1387529373168945,
      "step": 6856
    },
    {
      "epoch": 2.6153564453125e-05,
      "step": 6856,
      "training_step_time": 9.912450551986694
    },
    {
      "epoch": 2.6153564453125e-05,
      "model_forward_time": 3.1419129371643066,
      "step": 6856
    },
    {
      "epoch": 2.6153564453125e-05,
      "step": 6856,
      "training_step_time": 9.914721488952637
    },
    {
      "epoch": 2.6153564453125e-05,
      "model_forward_time": 3.135268211364746,
      "step": 6856
    },
    {
      "epoch": 2.6153564453125e-05,
      "step": 6856,
      "training_step_time": 10.313444137573242
    },
    {
      "epoch": 2.6157379150390626e-05,
      "model_forward_time": 3.15501070022583,
      "step": 6857
    },
    {
      "epoch": 2.6157379150390626e-05,
      "step": 6857,
      "training_step_time": 9.92716908454895
    },
    {
      "epoch": 2.6157379150390626e-05,
      "model_forward_time": 3.1387038230895996,
      "step": 6857
    },
    {
      "epoch": 2.6157379150390626e-05,
      "step": 6857,
      "training_step_time": 9.912646532058716
    },
    {
      "epoch": 2.6157379150390626e-05,
      "model_forward_time": 3.1426589488983154,
      "step": 6857
    },
    {
      "dynamics_loss_avg": 0.05501676183193922,
      "epoch": 2.6157379150390626e-05,
      "step": 6857
    },
    {
      "action_loss_avg": 0.022800904326140882,
      "epoch": 2.6157379150390626e-05,
      "step": 6857
    },
    {
      "epoch": 2.6157379150390626e-05,
      "step": 6857,
      "training_step_time": 9.918117761611938
    },
    {
      "epoch": 2.6157379150390626e-05,
      "model_forward_time": 3.1408345699310303,
      "step": 6857
    },
    {
      "epoch": 2.6157379150390626e-05,
      "step": 6857,
      "training_step_time": 10.278481483459473
    },
    {
      "epoch": 2.616119384765625e-05,
      "model_forward_time": 3.433274269104004,
      "step": 6858
    },
    {
      "epoch": 2.616119384765625e-05,
      "step": 6858,
      "training_step_time": 10.209081888198853
    },
    {
      "epoch": 2.616119384765625e-05,
      "model_forward_time": 3.144852638244629,
      "step": 6858
    },
    {
      "epoch": 2.616119384765625e-05,
      "step": 6858,
      "training_step_time": 9.920385599136353
    },
    {
      "epoch": 2.616119384765625e-05,
      "model_forward_time": 3.1451821327209473,
      "step": 6858
    },
    {
      "epoch": 2.616119384765625e-05,
      "step": 6858,
      "training_step_time": 9.923495531082153
    },
    {
      "epoch": 2.616119384765625e-05,
      "model_forward_time": 3.1416609287261963,
      "step": 6858
    },
    {
      "epoch": 2.616119384765625e-05,
      "step": 6858,
      "training_step_time": 10.294797897338867
    },
    {
      "epoch": 2.6165008544921875e-05,
      "model_forward_time": 3.1515133380889893,
      "step": 6859
    },
    {
      "epoch": 2.6165008544921875e-05,
      "step": 6859,
      "training_step_time": 9.926712989807129
    },
    {
      "epoch": 2.6165008544921875e-05,
      "model_forward_time": 3.1463963985443115,
      "step": 6859
    },
    {
      "epoch": 2.6165008544921875e-05,
      "step": 6859,
      "training_step_time": 10.188387870788574
    },
    {
      "epoch": 2.6165008544921875e-05,
      "model_forward_time": 3.149264097213745,
      "step": 6859
    },
    {
      "epoch": 2.6165008544921875e-05,
      "step": 6859,
      "training_step_time": 9.931541681289673
    },
    {
      "epoch": 2.6165008544921875e-05,
      "model_forward_time": 3.146153450012207,
      "step": 6859
    },
    {
      "epoch": 2.6165008544921875e-05,
      "step": 6859,
      "training_step_time": 10.336246490478516
    },
    {
      "epoch": 2.61688232421875e-05,
      "grad_norm": 0.33582085371017456,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 6860
    },
    {
      "epoch": 2.61688232421875e-05,
      "model_forward_time": 3.1623117923736572,
      "step": 6860
    },
    {
      "dynamics_loss_avg": 0.04138615019619465,
      "epoch": 2.61688232421875e-05,
      "step": 6860
    },
    {
      "action_loss_avg": 0.01657656519673765,
      "epoch": 2.61688232421875e-05,
      "step": 6860
    },
    {
      "epoch": 2.61688232421875e-05,
      "step": 6860,
      "training_step_time": 9.943968296051025
    },
    {
      "epoch": 2.61688232421875e-05,
      "model_forward_time": 3.143005609512329,
      "step": 6860
    },
    {
      "epoch": 2.61688232421875e-05,
      "step": 6860,
      "training_step_time": 9.919600248336792
    },
    {
      "epoch": 2.61688232421875e-05,
      "model_forward_time": 3.419020891189575,
      "step": 6860
    },
    {
      "epoch": 2.61688232421875e-05,
      "step": 6860,
      "training_step_time": 10.195692300796509
    },
    {
      "epoch": 2.61688232421875e-05,
      "model_forward_time": 3.1414852142333984,
      "step": 6860
    },
    {
      "epoch": 2.61688232421875e-05,
      "step": 6860,
      "training_step_time": 10.289467096328735
    },
    {
      "epoch": 2.6172637939453124e-05,
      "model_forward_time": 3.1588809490203857,
      "step": 6861
    },
    {
      "epoch": 2.6172637939453124e-05,
      "step": 6861,
      "training_step_time": 9.935084819793701
    },
    {
      "epoch": 2.6172637939453124e-05,
      "model_forward_time": 3.1405770778656006,
      "step": 6861
    },
    {
      "epoch": 2.6172637939453124e-05,
      "step": 6861,
      "training_step_time": 9.913357496261597
    },
    {
      "epoch": 2.6172637939453124e-05,
      "model_forward_time": 3.13916015625,
      "step": 6861
    },
    {
      "epoch": 2.6172637939453124e-05,
      "step": 6861,
      "training_step_time": 9.910641193389893
    },
    {
      "epoch": 2.6172637939453124e-05,
      "model_forward_time": 3.4045815467834473,
      "step": 6861
    },
    {
      "epoch": 2.6172637939453124e-05,
      "step": 6861,
      "training_step_time": 10.271701335906982
    },
    {
      "epoch": 2.617645263671875e-05,
      "model_forward_time": 3.1489057540893555,
      "step": 6862
    },
    {
      "epoch": 2.617645263671875e-05,
      "step": 6862,
      "training_step_time": 9.926498413085938
    },
    {
      "epoch": 2.617645263671875e-05,
      "model_forward_time": 3.143782615661621,
      "step": 6862
    },
    {
      "epoch": 2.617645263671875e-05,
      "step": 6862,
      "training_step_time": 9.916243076324463
    },
    {
      "epoch": 2.617645263671875e-05,
      "model_forward_time": 3.143683433532715,
      "step": 6862
    },
    {
      "dynamics_loss_avg": 0.04080870654433966,
      "epoch": 2.617645263671875e-05,
      "step": 6862
    },
    {
      "action_loss_avg": 0.016358589008450507,
      "epoch": 2.617645263671875e-05,
      "step": 6862
    },
    {
      "epoch": 2.617645263671875e-05,
      "step": 6862,
      "training_step_time": 9.919886112213135
    },
    {
      "epoch": 2.617645263671875e-05,
      "model_forward_time": 3.1426568031311035,
      "step": 6862
    },
    {
      "epoch": 2.617645263671875e-05,
      "step": 6862,
      "training_step_time": 10.293536186218262
    },
    {
      "epoch": 2.6180267333984377e-05,
      "model_forward_time": 3.4099624156951904,
      "step": 6863
    },
    {
      "epoch": 2.6180267333984377e-05,
      "step": 6863,
      "training_step_time": 10.185142755508423
    },
    {
      "epoch": 2.6180267333984377e-05,
      "model_forward_time": 3.146099328994751,
      "step": 6863
    },
    {
      "epoch": 2.6180267333984377e-05,
      "step": 6863,
      "training_step_time": 9.92301893234253
    },
    {
      "epoch": 2.6180267333984377e-05,
      "model_forward_time": 3.1443047523498535,
      "step": 6863
    },
    {
      "epoch": 2.6180267333984377e-05,
      "step": 6863,
      "training_step_time": 9.923320770263672
    },
    {
      "epoch": 2.6180267333984377e-05,
      "model_forward_time": 3.1443305015563965,
      "step": 6863
    },
    {
      "epoch": 2.6180267333984377e-05,
      "step": 6863,
      "training_step_time": 10.332631587982178
    },
    {
      "epoch": 2.618408203125e-05,
      "model_forward_time": 3.1622414588928223,
      "step": 6864
    },
    {
      "epoch": 2.618408203125e-05,
      "step": 6864,
      "training_step_time": 9.940226554870605
    },
    {
      "epoch": 2.618408203125e-05,
      "model_forward_time": 3.420562982559204,
      "step": 6864
    },
    {
      "epoch": 2.618408203125e-05,
      "step": 6864,
      "training_step_time": 10.194473266601562
    },
    {
      "epoch": 2.618408203125e-05,
      "model_forward_time": 3.1414570808410645,
      "step": 6864
    },
    {
      "epoch": 2.618408203125e-05,
      "step": 6864,
      "training_step_time": 9.913819789886475
    },
    {
      "epoch": 2.618408203125e-05,
      "model_forward_time": 3.1389951705932617,
      "step": 6864
    },
    {
      "epoch": 2.618408203125e-05,
      "step": 6864,
      "training_step_time": 10.008517742156982
    },
    {
      "epoch": 2.6187896728515626e-05,
      "model_forward_time": 3.1495094299316406,
      "step": 6865
    },
    {
      "dynamics_loss_avg": 0.06113814748823643,
      "epoch": 2.6187896728515626e-05,
      "step": 6865
    },
    {
      "action_loss_avg": 0.021759941801428796,
      "epoch": 2.6187896728515626e-05,
      "step": 6865
    },
    {
      "epoch": 2.6187896728515626e-05,
      "step": 6865,
      "training_step_time": 9.921627521514893
    },
    {
      "epoch": 2.6187896728515626e-05,
      "model_forward_time": 3.1421096324920654,
      "step": 6865
    },
    {
      "epoch": 2.6187896728515626e-05,
      "step": 6865,
      "training_step_time": 9.91623592376709
    },
    {
      "epoch": 2.6187896728515626e-05,
      "model_forward_time": 3.14206862449646,
      "step": 6865
    },
    {
      "epoch": 2.6187896728515626e-05,
      "step": 6865,
      "training_step_time": 10.224591970443726
    },
    {
      "epoch": 2.6187896728515626e-05,
      "model_forward_time": 3.143230438232422,
      "step": 6865
    },
    {
      "epoch": 2.6187896728515626e-05,
      "step": 6865,
      "training_step_time": 10.267687797546387
    },
    {
      "epoch": 2.619171142578125e-05,
      "model_forward_time": 3.148810386657715,
      "step": 6866
    },
    {
      "epoch": 2.619171142578125e-05,
      "step": 6866,
      "training_step_time": 9.927278518676758
    },
    {
      "epoch": 2.619171142578125e-05,
      "model_forward_time": 2.4367525577545166,
      "step": 6866
    },
    {
      "epoch": 2.619171142578125e-05,
      "step": 6866,
      "training_step_time": 7.462830305099487
    },
    {
      "epoch": 2.619171142578125e-05,
      "model_forward_time": 3.133901357650757,
      "step": 6866
    },
    {
      "epoch": 2.619171142578125e-05,
      "step": 6866,
      "training_step_time": 9.90922474861145
    },
    {
      "epoch": 2.619171142578125e-05,
      "model_forward_time": 3.14035701751709,
      "step": 6866
    },
    {
      "epoch": 2.619171142578125e-05,
      "step": 6866,
      "training_step_time": 10.279259204864502
    },
    {
      "epoch": 2.6195526123046875e-05,
      "model_forward_time": 3.1609771251678467,
      "step": 6867
    },
    {
      "epoch": 2.6195526123046875e-05,
      "step": 6867,
      "training_step_time": 9.938313245773315
    },
    {
      "epoch": 2.6195526123046875e-05,
      "model_forward_time": 3.141510248184204,
      "step": 6867
    },
    {
      "epoch": 2.6195526123046875e-05,
      "step": 6867,
      "training_step_time": 9.920319080352783
    },
    {
      "epoch": 2.6195526123046875e-05,
      "model_forward_time": 3.1465463638305664,
      "step": 6867
    },
    {
      "dynamics_loss_avg": 0.05731235276907683,
      "epoch": 2.6195526123046875e-05,
      "step": 6867
    },
    {
      "action_loss_avg": 0.017836204264312983,
      "epoch": 2.6195526123046875e-05,
      "step": 6867
    },
    {
      "epoch": 2.6195526123046875e-05,
      "step": 6867,
      "training_step_time": 9.928015947341919
    },
    {
      "epoch": 2.6195526123046875e-05,
      "model_forward_time": 3.1390810012817383,
      "step": 6867
    },
    {
      "epoch": 2.6195526123046875e-05,
      "step": 6867,
      "training_step_time": 10.014219045639038
    },
    {
      "epoch": 2.61993408203125e-05,
      "model_forward_time": 3.1529011726379395,
      "step": 6868
    },
    {
      "epoch": 2.61993408203125e-05,
      "step": 6868,
      "training_step_time": 10.218801975250244
    },
    {
      "epoch": 2.61993408203125e-05,
      "model_forward_time": 3.1416139602661133,
      "step": 6868
    },
    {
      "epoch": 2.61993408203125e-05,
      "step": 6868,
      "training_step_time": 9.916165113449097
    },
    {
      "epoch": 2.61993408203125e-05,
      "model_forward_time": 3.1413064002990723,
      "step": 6868
    },
    {
      "epoch": 2.61993408203125e-05,
      "step": 6868,
      "training_step_time": 9.912734270095825
    },
    {
      "epoch": 2.61993408203125e-05,
      "model_forward_time": 3.141511917114258,
      "step": 6868
    },
    {
      "epoch": 2.61993408203125e-05,
      "step": 6868,
      "training_step_time": 10.010096549987793
    },
    {
      "epoch": 2.6203155517578124e-05,
      "model_forward_time": 2.447399616241455,
      "step": 6869
    },
    {
      "epoch": 2.6203155517578124e-05,
      "step": 6869,
      "training_step_time": 7.477077007293701
    },
    {
      "epoch": 2.6203155517578124e-05,
      "model_forward_time": 3.400132656097412,
      "step": 6869
    },
    {
      "epoch": 2.6203155517578124e-05,
      "step": 6869,
      "training_step_time": 10.16962218284607
    },
    {
      "epoch": 2.6203155517578124e-05,
      "model_forward_time": 3.1435999870300293,
      "step": 6869
    },
    {
      "epoch": 2.6203155517578124e-05,
      "step": 6869,
      "training_step_time": 9.919968605041504
    },
    {
      "epoch": 2.6203155517578124e-05,
      "model_forward_time": 3.1390695571899414,
      "step": 6869
    },
    {
      "epoch": 2.6203155517578124e-05,
      "step": 6869,
      "training_step_time": 10.264743328094482
    },
    {
      "epoch": 2.620697021484375e-05,
      "grad_norm": 0.40078461170196533,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 6870
    },
    {
      "epoch": 2.620697021484375e-05,
      "model_forward_time": 3.147233009338379,
      "step": 6870
    },
    {
      "dynamics_loss_avg": 0.055994106363505124,
      "epoch": 2.620697021484375e-05,
      "step": 6870
    },
    {
      "action_loss_avg": 0.029279160196892917,
      "epoch": 2.620697021484375e-05,
      "step": 6870
    },
    {
      "epoch": 2.620697021484375e-05,
      "step": 6870,
      "training_step_time": 9.929084777832031
    },
    {
      "epoch": 2.620697021484375e-05,
      "model_forward_time": 3.141903877258301,
      "step": 6870
    },
    {
      "epoch": 2.620697021484375e-05,
      "step": 6870,
      "training_step_time": 9.918737888336182
    },
    {
      "epoch": 2.620697021484375e-05,
      "model_forward_time": 3.1409976482391357,
      "step": 6870
    },
    {
      "epoch": 2.620697021484375e-05,
      "step": 6870,
      "training_step_time": 10.180853366851807
    },
    {
      "epoch": 2.620697021484375e-05,
      "model_forward_time": 3.1422414779663086,
      "step": 6870
    },
    {
      "epoch": 2.620697021484375e-05,
      "step": 6870,
      "training_step_time": 10.283145904541016
    },
    {
      "epoch": 2.6210784912109377e-05,
      "model_forward_time": 3.16031551361084,
      "step": 6871
    },
    {
      "epoch": 2.6210784912109377e-05,
      "step": 6871,
      "training_step_time": 9.938211679458618
    },
    {
      "epoch": 2.6210784912109377e-05,
      "model_forward_time": 2.4350411891937256,
      "step": 6871
    },
    {
      "epoch": 2.6210784912109377e-05,
      "step": 6871,
      "training_step_time": 7.468290090560913
    },
    {
      "epoch": 2.6210784912109377e-05,
      "model_forward_time": 3.1372087001800537,
      "step": 6871
    },
    {
      "epoch": 2.6210784912109377e-05,
      "step": 6871,
      "training_step_time": 9.914888143539429
    },
    {
      "epoch": 2.6210784912109377e-05,
      "model_forward_time": 3.4041965007781982,
      "step": 6871
    },
    {
      "epoch": 2.6210784912109377e-05,
      "step": 6871,
      "training_step_time": 10.27346420288086
    },
    {
      "epoch": 2.6214599609375e-05,
      "model_forward_time": 3.148162364959717,
      "step": 6872
    },
    {
      "epoch": 2.6214599609375e-05,
      "step": 6872,
      "training_step_time": 9.928532123565674
    },
    {
      "epoch": 2.6214599609375e-05,
      "model_forward_time": 3.1435165405273438,
      "step": 6872
    },
    {
      "epoch": 2.6214599609375e-05,
      "step": 6872,
      "training_step_time": 9.92647671699524
    },
    {
      "epoch": 2.6214599609375e-05,
      "model_forward_time": 3.140458345413208,
      "step": 6872
    },
    {
      "dynamics_loss_avg": 0.06394286071881652,
      "epoch": 2.6214599609375e-05,
      "step": 6872
    },
    {
      "action_loss_avg": 0.022160901012830435,
      "epoch": 2.6214599609375e-05,
      "step": 6872
    },
    {
      "epoch": 2.6214599609375e-05,
      "step": 6872,
      "training_step_time": 9.918145179748535
    },
    {
      "epoch": 2.6214599609375e-05,
      "model_forward_time": 3.143397808074951,
      "step": 6872
    },
    {
      "epoch": 2.6214599609375e-05,
      "step": 6872,
      "training_step_time": 10.014471054077148
    },
    {
      "epoch": 2.6218414306640626e-05,
      "model_forward_time": 3.4080471992492676,
      "step": 6873
    },
    {
      "epoch": 2.6218414306640626e-05,
      "step": 6873,
      "training_step_time": 10.185699939727783
    },
    {
      "epoch": 2.6218414306640626e-05,
      "model_forward_time": 3.1437923908233643,
      "step": 6873
    },
    {
      "epoch": 2.6218414306640626e-05,
      "step": 6873,
      "training_step_time": 9.919743537902832
    },
    {
      "epoch": 2.6218414306640626e-05,
      "model_forward_time": 3.1405069828033447,
      "step": 6873
    },
    {
      "epoch": 2.6218414306640626e-05,
      "step": 6873,
      "training_step_time": 9.916420698165894
    },
    {
      "epoch": 2.6218414306640626e-05,
      "model_forward_time": 3.1482303142547607,
      "step": 6873
    },
    {
      "epoch": 2.6218414306640626e-05,
      "step": 6873,
      "training_step_time": 10.297962665557861
    },
    {
      "epoch": 2.622222900390625e-05,
      "model_forward_time": 3.157585382461548,
      "step": 6874
    },
    {
      "epoch": 2.622222900390625e-05,
      "step": 6874,
      "training_step_time": 9.932754755020142
    },
    {
      "epoch": 2.622222900390625e-05,
      "model_forward_time": 3.1368472576141357,
      "step": 6874
    },
    {
      "epoch": 2.622222900390625e-05,
      "step": 6874,
      "training_step_time": 10.2003014087677
    },
    {
      "epoch": 2.622222900390625e-05,
      "model_forward_time": 3.1414003372192383,
      "step": 6874
    },
    {
      "epoch": 2.622222900390625e-05,
      "step": 6874,
      "training_step_time": 9.917762041091919
    },
    {
      "epoch": 2.622222900390625e-05,
      "model_forward_time": 3.1468594074249268,
      "step": 6874
    },
    {
      "epoch": 2.622222900390625e-05,
      "step": 6874,
      "training_step_time": 10.289436101913452
    },
    {
      "epoch": 2.6226043701171875e-05,
      "model_forward_time": 3.1541507244110107,
      "step": 6875
    },
    {
      "dynamics_loss_avg": 0.056599709391593936,
      "epoch": 2.6226043701171875e-05,
      "step": 6875
    },
    {
      "action_loss_avg": 0.023714331910014153,
      "epoch": 2.6226043701171875e-05,
      "step": 6875
    },
    {
      "epoch": 2.6226043701171875e-05,
      "step": 6875,
      "training_step_time": 9.937905550003052
    },
    {
      "epoch": 2.6226043701171875e-05,
      "model_forward_time": 3.14324951171875,
      "step": 6875
    },
    {
      "epoch": 2.6226043701171875e-05,
      "step": 6875,
      "training_step_time": 9.91954779624939
    },
    {
      "epoch": 2.6226043701171875e-05,
      "model_forward_time": 3.405667781829834,
      "step": 6875
    },
    {
      "epoch": 2.6226043701171875e-05,
      "step": 6875,
      "training_step_time": 10.177967071533203
    },
    {
      "epoch": 2.6226043701171875e-05,
      "model_forward_time": 3.145331859588623,
      "step": 6875
    },
    {
      "epoch": 2.6226043701171875e-05,
      "step": 6875,
      "training_step_time": 10.272945165634155
    },
    {
      "epoch": 2.62298583984375e-05,
      "model_forward_time": 3.152231216430664,
      "step": 6876
    },
    {
      "epoch": 2.62298583984375e-05,
      "step": 6876,
      "training_step_time": 9.928541898727417
    },
    {
      "epoch": 2.62298583984375e-05,
      "model_forward_time": 3.1425092220306396,
      "step": 6876
    },
    {
      "epoch": 2.62298583984375e-05,
      "step": 6876,
      "training_step_time": 9.91623854637146
    },
    {
      "epoch": 2.62298583984375e-05,
      "model_forward_time": 3.146343469619751,
      "step": 6876
    },
    {
      "epoch": 2.62298583984375e-05,
      "step": 6876,
      "training_step_time": 9.929105997085571
    },
    {
      "epoch": 2.62298583984375e-05,
      "model_forward_time": 3.146340847015381,
      "step": 6876
    },
    {
      "epoch": 2.62298583984375e-05,
      "step": 6876,
      "training_step_time": 10.302960872650146
    },
    {
      "epoch": 2.6233673095703124e-05,
      "model_forward_time": 3.156050682067871,
      "step": 6877
    },
    {
      "epoch": 2.6233673095703124e-05,
      "step": 6877,
      "training_step_time": 9.933358669281006
    },
    {
      "epoch": 2.6233673095703124e-05,
      "model_forward_time": 3.142491579055786,
      "step": 6877
    },
    {
      "epoch": 2.6233673095703124e-05,
      "step": 6877,
      "training_step_time": 9.91960072517395
    },
    {
      "epoch": 2.6233673095703124e-05,
      "model_forward_time": 3.143071174621582,
      "step": 6877
    },
    {
      "dynamics_loss_avg": 0.07557355277240277,
      "epoch": 2.6233673095703124e-05,
      "step": 6877
    },
    {
      "action_loss_avg": 0.028516737278550863,
      "epoch": 2.6233673095703124e-05,
      "step": 6877
    },
    {
      "epoch": 2.6233673095703124e-05,
      "step": 6877,
      "training_step_time": 9.926084041595459
    },
    {
      "epoch": 2.6233673095703124e-05,
      "model_forward_time": 3.144887685775757,
      "step": 6877
    },
    {
      "epoch": 2.6233673095703124e-05,
      "step": 6877,
      "training_step_time": 10.249690294265747
    },
    {
      "epoch": 2.623748779296875e-05,
      "model_forward_time": 3.440917730331421,
      "step": 6878
    },
    {
      "epoch": 2.623748779296875e-05,
      "step": 6878,
      "training_step_time": 10.21962857246399
    },
    {
      "epoch": 2.623748779296875e-05,
      "model_forward_time": 3.1468138694763184,
      "step": 6878
    },
    {
      "epoch": 2.623748779296875e-05,
      "step": 6878,
      "training_step_time": 9.92236852645874
    },
    {
      "epoch": 2.623748779296875e-05,
      "model_forward_time": 3.143580198287964,
      "step": 6878
    },
    {
      "epoch": 2.623748779296875e-05,
      "step": 6878,
      "training_step_time": 9.920512199401855
    },
    {
      "epoch": 2.623748779296875e-05,
      "model_forward_time": 3.1431827545166016,
      "step": 6878
    },
    {
      "epoch": 2.623748779296875e-05,
      "step": 6878,
      "training_step_time": 10.028093576431274
    },
    {
      "epoch": 2.6241302490234373e-05,
      "model_forward_time": 3.1528618335723877,
      "step": 6879
    },
    {
      "epoch": 2.6241302490234373e-05,
      "step": 6879,
      "training_step_time": 9.929194450378418
    },
    {
      "epoch": 2.6241302490234373e-05,
      "model_forward_time": 3.149740695953369,
      "step": 6879
    },
    {
      "epoch": 2.6241302490234373e-05,
      "step": 6879,
      "training_step_time": 10.194856643676758
    },
    {
      "epoch": 2.6241302490234373e-05,
      "model_forward_time": 3.1493747234344482,
      "step": 6879
    },
    {
      "epoch": 2.6241302490234373e-05,
      "step": 6879,
      "training_step_time": 9.92960524559021
    },
    {
      "epoch": 2.6241302490234373e-05,
      "model_forward_time": 3.144287109375,
      "step": 6879
    },
    {
      "epoch": 2.6241302490234373e-05,
      "step": 6879,
      "training_step_time": 10.303776979446411
    },
    {
      "epoch": 2.62451171875e-05,
      "grad_norm": 0.45579785108566284,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 6880
    },
    {
      "epoch": 2.62451171875e-05,
      "model_forward_time": 3.148040294647217,
      "step": 6880
    },
    {
      "dynamics_loss_avg": 0.05162944570183754,
      "epoch": 2.62451171875e-05,
      "step": 6880
    },
    {
      "action_loss_avg": 0.01743419428821653,
      "epoch": 2.62451171875e-05,
      "step": 6880
    },
    {
      "epoch": 2.62451171875e-05,
      "step": 6880,
      "training_step_time": 9.93085503578186
    },
    {
      "epoch": 2.62451171875e-05,
      "model_forward_time": 3.139470100402832,
      "step": 6880
    },
    {
      "epoch": 2.62451171875e-05,
      "step": 6880,
      "training_step_time": 9.912668943405151
    },
    {
      "epoch": 2.62451171875e-05,
      "model_forward_time": 3.4227569103240967,
      "step": 6880
    },
    {
      "epoch": 2.62451171875e-05,
      "step": 6880,
      "training_step_time": 10.195768117904663
    },
    {
      "epoch": 2.62451171875e-05,
      "model_forward_time": 3.1395976543426514,
      "step": 6880
    },
    {
      "epoch": 2.62451171875e-05,
      "step": 6880,
      "training_step_time": 10.020763635635376
    },
    {
      "epoch": 2.6248931884765626e-05,
      "model_forward_time": 3.1749236583709717,
      "step": 6881
    },
    {
      "epoch": 2.6248931884765626e-05,
      "step": 6881,
      "training_step_time": 9.950050830841064
    },
    {
      "epoch": 2.6248931884765626e-05,
      "model_forward_time": 3.139432907104492,
      "step": 6881
    },
    {
      "epoch": 2.6248931884765626e-05,
      "step": 6881,
      "training_step_time": 9.913715600967407
    },
    {
      "epoch": 2.6248931884765626e-05,
      "model_forward_time": 3.139199733734131,
      "step": 6881
    },
    {
      "epoch": 2.6248931884765626e-05,
      "step": 6881,
      "training_step_time": 9.917850971221924
    },
    {
      "epoch": 2.6248931884765626e-05,
      "model_forward_time": 3.407557249069214,
      "step": 6881
    },
    {
      "epoch": 2.6248931884765626e-05,
      "step": 6881,
      "training_step_time": 10.277303218841553
    },
    {
      "epoch": 2.625274658203125e-05,
      "model_forward_time": 3.1647427082061768,
      "step": 6882
    },
    {
      "epoch": 2.625274658203125e-05,
      "step": 6882,
      "training_step_time": 9.94710898399353
    },
    {
      "epoch": 2.625274658203125e-05,
      "model_forward_time": 3.1514523029327393,
      "step": 6882
    },
    {
      "epoch": 2.625274658203125e-05,
      "step": 6882,
      "training_step_time": 9.934536933898926
    },
    {
      "epoch": 2.625274658203125e-05,
      "model_forward_time": 3.145109176635742,
      "step": 6882
    },
    {
      "dynamics_loss_avg": 0.042024905979633334,
      "epoch": 2.625274658203125e-05,
      "step": 6882
    },
    {
      "action_loss_avg": 0.013014434836804867,
      "epoch": 2.625274658203125e-05,
      "step": 6882
    },
    {
      "epoch": 2.625274658203125e-05,
      "step": 6882,
      "training_step_time": 9.927521228790283
    },
    {
      "epoch": 2.625274658203125e-05,
      "model_forward_time": 2.4396891593933105,
      "step": 6882
    },
    {
      "epoch": 2.625274658203125e-05,
      "step": 6882,
      "training_step_time": 10.027499437332153
    },
    {
      "epoch": 2.6256561279296875e-05,
      "model_forward_time": 3.41607403755188,
      "step": 6883
    },
    {
      "epoch": 2.6256561279296875e-05,
      "step": 6883,
      "training_step_time": 10.192171812057495
    },
    {
      "epoch": 2.6256561279296875e-05,
      "model_forward_time": 3.1381845474243164,
      "step": 6883
    },
    {
      "epoch": 2.6256561279296875e-05,
      "step": 6883,
      "training_step_time": 9.911926984786987
    },
    {
      "epoch": 2.6256561279296875e-05,
      "model_forward_time": 3.1368978023529053,
      "step": 6883
    },
    {
      "epoch": 2.6256561279296875e-05,
      "step": 6883,
      "training_step_time": 9.917553663253784
    },
    {
      "epoch": 2.6256561279296875e-05,
      "model_forward_time": 3.138788938522339,
      "step": 6883
    },
    {
      "epoch": 2.6256561279296875e-05,
      "step": 6883,
      "training_step_time": 10.232495069503784
    },
    {
      "epoch": 2.62603759765625e-05,
      "model_forward_time": 3.155500888824463,
      "step": 6884
    },
    {
      "epoch": 2.62603759765625e-05,
      "step": 6884,
      "training_step_time": 9.942458391189575
    },
    {
      "epoch": 2.62603759765625e-05,
      "model_forward_time": 3.1419575214385986,
      "step": 6884
    },
    {
      "epoch": 2.62603759765625e-05,
      "step": 6884,
      "training_step_time": 10.216970682144165
    },
    {
      "epoch": 2.62603759765625e-05,
      "model_forward_time": 3.1401236057281494,
      "step": 6884
    },
    {
      "epoch": 2.62603759765625e-05,
      "step": 6884,
      "training_step_time": 9.913270711898804
    },
    {
      "epoch": 2.62603759765625e-05,
      "model_forward_time": 3.1427299976348877,
      "step": 6884
    },
    {
      "epoch": 2.62603759765625e-05,
      "step": 6884,
      "training_step_time": 10.321048498153687
    },
    {
      "epoch": 2.6264190673828124e-05,
      "model_forward_time": 3.1459438800811768,
      "step": 6885
    },
    {
      "dynamics_loss_avg": 0.06721306927502155,
      "epoch": 2.6264190673828124e-05,
      "step": 6885
    },
    {
      "action_loss_avg": 0.028039575507864355,
      "epoch": 2.6264190673828124e-05,
      "step": 6885
    },
    {
      "epoch": 2.6264190673828124e-05,
      "step": 6885,
      "training_step_time": 9.92342495918274
    },
    {
      "epoch": 2.6264190673828124e-05,
      "model_forward_time": 3.1366631984710693,
      "step": 6885
    },
    {
      "epoch": 2.6264190673828124e-05,
      "step": 6885,
      "training_step_time": 9.90791130065918
    },
    {
      "epoch": 2.6264190673828124e-05,
      "model_forward_time": 3.1431005001068115,
      "step": 6885
    },
    {
      "epoch": 2.6264190673828124e-05,
      "step": 6885,
      "training_step_time": 10.197776556015015
    },
    {
      "epoch": 2.6264190673828124e-05,
      "model_forward_time": 3.1471641063690186,
      "step": 6885
    },
    {
      "epoch": 2.6264190673828124e-05,
      "step": 6885,
      "training_step_time": 10.0218346118927
    },
    {
      "epoch": 2.626800537109375e-05,
      "model_forward_time": 3.154174327850342,
      "step": 6886
    },
    {
      "epoch": 2.626800537109375e-05,
      "step": 6886,
      "training_step_time": 9.936523914337158
    },
    {
      "epoch": 2.626800537109375e-05,
      "model_forward_time": 3.141688823699951,
      "step": 6886
    },
    {
      "epoch": 2.626800537109375e-05,
      "step": 6886,
      "training_step_time": 9.928100347518921
    },
    {
      "epoch": 2.626800537109375e-05,
      "model_forward_time": 3.149902582168579,
      "step": 6886
    },
    {
      "epoch": 2.626800537109375e-05,
      "step": 6886,
      "training_step_time": 9.928222417831421
    },
    {
      "epoch": 2.626800537109375e-05,
      "model_forward_time": 3.4379348754882812,
      "step": 6886
    },
    {
      "epoch": 2.626800537109375e-05,
      "step": 6886,
      "training_step_time": 10.305640697479248
    },
    {
      "epoch": 2.6271820068359373e-05,
      "model_forward_time": 3.153716802597046,
      "step": 6887
    },
    {
      "epoch": 2.6271820068359373e-05,
      "step": 6887,
      "training_step_time": 9.936743974685669
    },
    {
      "epoch": 2.6271820068359373e-05,
      "model_forward_time": 3.1385176181793213,
      "step": 6887
    },
    {
      "epoch": 2.6271820068359373e-05,
      "step": 6887,
      "training_step_time": 9.921648979187012
    },
    {
      "epoch": 2.6271820068359373e-05,
      "model_forward_time": 3.1420586109161377,
      "step": 6887
    },
    {
      "dynamics_loss_avg": 0.07319232076406479,
      "epoch": 2.6271820068359373e-05,
      "step": 6887
    },
    {
      "action_loss_avg": 0.031081174965947866,
      "epoch": 2.6271820068359373e-05,
      "step": 6887
    },
    {
      "epoch": 2.6271820068359373e-05,
      "step": 6887,
      "training_step_time": 9.921855449676514
    },
    {
      "epoch": 2.6271820068359373e-05,
      "model_forward_time": 3.1363773345947266,
      "step": 6887
    },
    {
      "epoch": 2.6271820068359373e-05,
      "step": 6887,
      "training_step_time": 10.013808488845825
    },
    {
      "epoch": 2.6275634765625e-05,
      "model_forward_time": 3.143683910369873,
      "step": 6888
    },
    {
      "epoch": 2.6275634765625e-05,
      "step": 6888,
      "training_step_time": 10.19179391860962
    },
    {
      "epoch": 2.6275634765625e-05,
      "model_forward_time": 3.134105920791626,
      "step": 6888
    },
    {
      "epoch": 2.6275634765625e-05,
      "step": 6888,
      "training_step_time": 9.911548852920532
    },
    {
      "epoch": 2.6275634765625e-05,
      "model_forward_time": 3.1396400928497314,
      "step": 6888
    },
    {
      "epoch": 2.6275634765625e-05,
      "step": 6888,
      "training_step_time": 9.916208028793335
    },
    {
      "epoch": 2.6275634765625e-05,
      "model_forward_time": 3.13887357711792,
      "step": 6888
    },
    {
      "epoch": 2.6275634765625e-05,
      "step": 6888,
      "training_step_time": 10.020936965942383
    },
    {
      "epoch": 2.6279449462890626e-05,
      "model_forward_time": 3.1465818881988525,
      "step": 6889
    },
    {
      "epoch": 2.6279449462890626e-05,
      "step": 6889,
      "training_step_time": 9.924977540969849
    },
    {
      "epoch": 2.6279449462890626e-05,
      "model_forward_time": 3.4001364707946777,
      "step": 6889
    },
    {
      "epoch": 2.6279449462890626e-05,
      "step": 6889,
      "training_step_time": 10.18301010131836
    },
    {
      "epoch": 2.6279449462890626e-05,
      "model_forward_time": 3.142702341079712,
      "step": 6889
    },
    {
      "epoch": 2.6279449462890626e-05,
      "step": 6889,
      "training_step_time": 9.912526369094849
    },
    {
      "epoch": 2.6279449462890626e-05,
      "model_forward_time": 3.143558979034424,
      "step": 6889
    },
    {
      "epoch": 2.6279449462890626e-05,
      "step": 6889,
      "training_step_time": 10.271545886993408
    },
    {
      "epoch": 2.628326416015625e-05,
      "grad_norm": 0.2235783487558365,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 6890
    },
    {
      "epoch": 2.628326416015625e-05,
      "model_forward_time": 3.1640570163726807,
      "step": 6890
    },
    {
      "dynamics_loss_avg": 0.05291339997202158,
      "epoch": 2.628326416015625e-05,
      "step": 6890
    },
    {
      "action_loss_avg": 0.017593650612980127,
      "epoch": 2.628326416015625e-05,
      "step": 6890
    },
    {
      "epoch": 2.628326416015625e-05,
      "step": 6890,
      "training_step_time": 9.937856435775757
    },
    {
      "epoch": 2.628326416015625e-05,
      "model_forward_time": 3.146312713623047,
      "step": 6890
    },
    {
      "epoch": 2.628326416015625e-05,
      "step": 6890,
      "training_step_time": 9.918459415435791
    },
    {
      "epoch": 2.628326416015625e-05,
      "model_forward_time": 3.4278745651245117,
      "step": 6890
    },
    {
      "epoch": 2.628326416015625e-05,
      "step": 6890,
      "training_step_time": 10.205086708068848
    },
    {
      "epoch": 2.628326416015625e-05,
      "model_forward_time": 3.139538526535034,
      "step": 6890
    },
    {
      "epoch": 2.628326416015625e-05,
      "step": 6890,
      "training_step_time": 10.011479139328003
    },
    {
      "epoch": 2.6287078857421875e-05,
      "model_forward_time": 3.1429836750030518,
      "step": 6891
    },
    {
      "epoch": 2.6287078857421875e-05,
      "step": 6891,
      "training_step_time": 9.911442518234253
    },
    {
      "epoch": 2.6287078857421875e-05,
      "model_forward_time": 3.146250009536743,
      "step": 6891
    },
    {
      "epoch": 2.6287078857421875e-05,
      "step": 6891,
      "training_step_time": 9.917182683944702
    },
    {
      "epoch": 2.6287078857421875e-05,
      "model_forward_time": 3.143113136291504,
      "step": 6891
    },
    {
      "epoch": 2.6287078857421875e-05,
      "step": 6891,
      "training_step_time": 9.913000345230103
    },
    {
      "epoch": 2.6287078857421875e-05,
      "model_forward_time": 3.4077374935150146,
      "step": 6891
    },
    {
      "epoch": 2.6287078857421875e-05,
      "step": 6891,
      "training_step_time": 10.275917530059814
    },
    {
      "epoch": 2.62908935546875e-05,
      "model_forward_time": 3.146043300628662,
      "step": 6892
    },
    {
      "epoch": 2.62908935546875e-05,
      "step": 6892,
      "training_step_time": 9.929046154022217
    },
    {
      "epoch": 2.62908935546875e-05,
      "model_forward_time": 3.142245054244995,
      "step": 6892
    },
    {
      "epoch": 2.62908935546875e-05,
      "step": 6892,
      "training_step_time": 9.915367841720581
    },
    {
      "epoch": 2.62908935546875e-05,
      "model_forward_time": 3.1337945461273193,
      "step": 6892
    },
    {
      "dynamics_loss_avg": 0.06679174415767193,
      "epoch": 2.62908935546875e-05,
      "step": 6892
    },
    {
      "action_loss_avg": 0.026176675967872143,
      "epoch": 2.62908935546875e-05,
      "step": 6892
    },
    {
      "epoch": 2.62908935546875e-05,
      "step": 6892,
      "training_step_time": 9.910189390182495
    },
    {
      "epoch": 2.62908935546875e-05,
      "model_forward_time": 3.135335683822632,
      "step": 6892
    },
    {
      "epoch": 2.62908935546875e-05,
      "step": 6892,
      "training_step_time": 10.276891231536865
    },
    {
      "epoch": 2.6294708251953124e-05,
      "model_forward_time": 3.401048183441162,
      "step": 6893
    },
    {
      "epoch": 2.6294708251953124e-05,
      "step": 6893,
      "training_step_time": 10.171795845031738
    },
    {
      "epoch": 2.6294708251953124e-05,
      "model_forward_time": 3.137204170227051,
      "step": 6893
    },
    {
      "epoch": 2.6294708251953124e-05,
      "step": 6893,
      "training_step_time": 9.905397415161133
    },
    {
      "epoch": 2.6294708251953124e-05,
      "model_forward_time": 3.1345577239990234,
      "step": 6893
    },
    {
      "epoch": 2.6294708251953124e-05,
      "step": 6893,
      "training_step_time": 9.905632257461548
    },
    {
      "epoch": 2.6294708251953124e-05,
      "model_forward_time": 3.133308172225952,
      "step": 6893
    },
    {
      "epoch": 2.6294708251953124e-05,
      "step": 6893,
      "training_step_time": 10.01219367980957
    },
    {
      "epoch": 2.629852294921875e-05,
      "model_forward_time": 3.1467175483703613,
      "step": 6894
    },
    {
      "epoch": 2.629852294921875e-05,
      "step": 6894,
      "training_step_time": 9.925916194915771
    },
    {
      "epoch": 2.629852294921875e-05,
      "model_forward_time": 3.1353495121002197,
      "step": 6894
    },
    {
      "epoch": 2.629852294921875e-05,
      "step": 6894,
      "training_step_time": 10.18648624420166
    },
    {
      "epoch": 2.629852294921875e-05,
      "model_forward_time": 3.1381876468658447,
      "step": 6894
    },
    {
      "epoch": 2.629852294921875e-05,
      "step": 6894,
      "training_step_time": 9.91581416130066
    },
    {
      "epoch": 2.629852294921875e-05,
      "model_forward_time": 3.144669771194458,
      "step": 6894
    },
    {
      "epoch": 2.629852294921875e-05,
      "step": 6894,
      "training_step_time": 10.024269104003906
    },
    {
      "epoch": 2.6302337646484373e-05,
      "model_forward_time": 2.447190523147583,
      "step": 6895
    },
    {
      "dynamics_loss_avg": 0.05642086546868086,
      "epoch": 2.6302337646484373e-05,
      "step": 6895
    },
    {
      "action_loss_avg": 0.02235249960795045,
      "epoch": 2.6302337646484373e-05,
      "step": 6895
    },
    {
      "epoch": 2.6302337646484373e-05,
      "step": 6895,
      "training_step_time": 7.476614475250244
    },
    {
      "epoch": 2.6302337646484373e-05,
      "model_forward_time": 3.1350138187408447,
      "step": 6895
    },
    {
      "epoch": 2.6302337646484373e-05,
      "step": 6895,
      "training_step_time": 9.903639793395996
    },
    {
      "epoch": 2.6302337646484373e-05,
      "model_forward_time": 3.1377601623535156,
      "step": 6895
    },
    {
      "epoch": 2.6302337646484373e-05,
      "step": 6895,
      "training_step_time": 9.910160303115845
    },
    {
      "epoch": 2.6302337646484373e-05,
      "model_forward_time": 3.406860113143921,
      "step": 6895
    },
    {
      "epoch": 2.6302337646484373e-05,
      "step": 6895,
      "training_step_time": 10.291847944259644
    },
    {
      "epoch": 2.630615234375e-05,
      "model_forward_time": 3.1575777530670166,
      "step": 6896
    },
    {
      "epoch": 2.630615234375e-05,
      "step": 6896,
      "training_step_time": 9.930317401885986
    },
    {
      "epoch": 2.630615234375e-05,
      "model_forward_time": 3.154783010482788,
      "step": 6896
    },
    {
      "epoch": 2.630615234375e-05,
      "step": 6896,
      "training_step_time": 9.927107810974121
    },
    {
      "epoch": 2.630615234375e-05,
      "model_forward_time": 3.1444504261016846,
      "step": 6896
    },
    {
      "epoch": 2.630615234375e-05,
      "step": 6896,
      "training_step_time": 9.916301012039185
    },
    {
      "epoch": 2.630615234375e-05,
      "model_forward_time": 3.138056755065918,
      "step": 6896
    },
    {
      "epoch": 2.630615234375e-05,
      "step": 6896,
      "training_step_time": 10.288110733032227
    },
    {
      "epoch": 2.6309967041015626e-05,
      "model_forward_time": 3.1480510234832764,
      "step": 6897
    },
    {
      "epoch": 2.6309967041015626e-05,
      "step": 6897,
      "training_step_time": 9.92525577545166
    },
    {
      "epoch": 2.6309967041015626e-05,
      "model_forward_time": 3.1330525875091553,
      "step": 6897
    },
    {
      "epoch": 2.6309967041015626e-05,
      "step": 6897,
      "training_step_time": 9.90712833404541
    },
    {
      "epoch": 2.6309967041015626e-05,
      "model_forward_time": 3.1419787406921387,
      "step": 6897
    },
    {
      "dynamics_loss_avg": 0.05048149321228266,
      "epoch": 2.6309967041015626e-05,
      "step": 6897
    },
    {
      "action_loss_avg": 0.018176006665453315,
      "epoch": 2.6309967041015626e-05,
      "step": 6897
    },
    {
      "epoch": 2.6309967041015626e-05,
      "step": 6897,
      "training_step_time": 9.919079542160034
    },
    {
      "epoch": 2.6309967041015626e-05,
      "model_forward_time": 3.1414737701416016,
      "step": 6897
    },
    {
      "epoch": 2.6309967041015626e-05,
      "step": 6897,
      "training_step_time": 10.297247171401978
    },
    {
      "epoch": 2.631378173828125e-05,
      "model_forward_time": 3.4246103763580322,
      "step": 6898
    },
    {
      "epoch": 2.631378173828125e-05,
      "step": 6898,
      "training_step_time": 10.202682733535767
    },
    {
      "epoch": 2.631378173828125e-05,
      "model_forward_time": 3.1413204669952393,
      "step": 6898
    },
    {
      "epoch": 2.631378173828125e-05,
      "step": 6898,
      "training_step_time": 9.91384243965149
    },
    {
      "epoch": 2.631378173828125e-05,
      "model_forward_time": 3.1415488719940186,
      "step": 6898
    },
    {
      "epoch": 2.631378173828125e-05,
      "step": 6898,
      "training_step_time": 9.908963441848755
    },
    {
      "epoch": 2.631378173828125e-05,
      "model_forward_time": 3.1407034397125244,
      "step": 6898
    },
    {
      "epoch": 2.631378173828125e-05,
      "step": 6898,
      "training_step_time": 10.032601118087769
    },
    {
      "epoch": 2.6317596435546875e-05,
      "model_forward_time": 3.1541249752044678,
      "step": 6899
    },
    {
      "epoch": 2.6317596435546875e-05,
      "step": 6899,
      "training_step_time": 9.929454326629639
    },
    {
      "epoch": 2.6317596435546875e-05,
      "model_forward_time": 3.145831823348999,
      "step": 6899
    },
    {
      "epoch": 2.6317596435546875e-05,
      "step": 6899,
      "training_step_time": 10.187772512435913
    },
    {
      "epoch": 2.6317596435546875e-05,
      "model_forward_time": 3.144242525100708,
      "step": 6899
    },
    {
      "epoch": 2.6317596435546875e-05,
      "step": 6899,
      "training_step_time": 9.91471004486084
    },
    {
      "epoch": 2.6317596435546875e-05,
      "model_forward_time": 3.138352870941162,
      "step": 6899
    },
    {
      "epoch": 2.6317596435546875e-05,
      "step": 6899,
      "training_step_time": 10.281249046325684
    },
    {
      "epoch": 2.63214111328125e-05,
      "grad_norm": 0.49640944600105286,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 6900
    },
    {
      "epoch": 2.63214111328125e-05,
      "model_forward_time": 3.1541671752929688,
      "step": 6900
    },
    {
      "dynamics_loss_avg": 0.0528373846784234,
      "epoch": 2.63214111328125e-05,
      "step": 6900
    },
    {
      "action_loss_avg": 0.021569705521687865,
      "epoch": 2.63214111328125e-05,
      "step": 6900
    },
    {
      "epoch": 2.63214111328125e-05,
      "step": 6900,
      "training_step_time": 9.92989182472229
    },
    {
      "epoch": 2.63214111328125e-05,
      "model_forward_time": 3.1415042877197266,
      "step": 6900
    },
    {
      "epoch": 2.63214111328125e-05,
      "step": 6900,
      "training_step_time": 9.910171270370483
    },
    {
      "epoch": 2.63214111328125e-05,
      "model_forward_time": 3.4176173210144043,
      "step": 6900
    },
    {
      "epoch": 2.63214111328125e-05,
      "step": 6900,
      "training_step_time": 10.18992280960083
    },
    {
      "epoch": 2.63214111328125e-05,
      "model_forward_time": 3.134773015975952,
      "step": 6900
    },
    {
      "epoch": 2.63214111328125e-05,
      "step": 6900,
      "training_step_time": 10.32024073600769
    },
    {
      "epoch": 2.6325225830078124e-05,
      "model_forward_time": 3.1443123817443848,
      "step": 6901
    },
    {
      "epoch": 2.6325225830078124e-05,
      "step": 6901,
      "training_step_time": 9.91647982597351
    },
    {
      "epoch": 2.6325225830078124e-05,
      "model_forward_time": 3.137968063354492,
      "step": 6901
    },
    {
      "epoch": 2.6325225830078124e-05,
      "step": 6901,
      "training_step_time": 9.908534526824951
    },
    {
      "epoch": 2.6325225830078124e-05,
      "model_forward_time": 3.1322522163391113,
      "step": 6901
    },
    {
      "epoch": 2.6325225830078124e-05,
      "step": 6901,
      "training_step_time": 9.898735046386719
    },
    {
      "epoch": 2.6325225830078124e-05,
      "model_forward_time": 3.4057388305664062,
      "step": 6901
    },
    {
      "epoch": 2.6325225830078124e-05,
      "step": 6901,
      "training_step_time": 10.310864686965942
    },
    {
      "epoch": 2.632904052734375e-05,
      "model_forward_time": 3.1475231647491455,
      "step": 6902
    },
    {
      "epoch": 2.632904052734375e-05,
      "step": 6902,
      "training_step_time": 9.91831111907959
    },
    {
      "epoch": 2.632904052734375e-05,
      "model_forward_time": 3.1318628787994385,
      "step": 6902
    },
    {
      "epoch": 2.632904052734375e-05,
      "step": 6902,
      "training_step_time": 9.899369239807129
    },
    {
      "epoch": 2.632904052734375e-05,
      "model_forward_time": 3.1305360794067383,
      "step": 6902
    },
    {
      "dynamics_loss_avg": 0.054396983608603476,
      "epoch": 2.632904052734375e-05,
      "step": 6902
    },
    {
      "action_loss_avg": 0.019590748101472856,
      "epoch": 2.632904052734375e-05,
      "step": 6902
    },
    {
      "epoch": 2.632904052734375e-05,
      "step": 6902,
      "training_step_time": 9.902363300323486
    },
    {
      "epoch": 2.632904052734375e-05,
      "model_forward_time": 3.136568784713745,
      "step": 6902
    },
    {
      "epoch": 2.632904052734375e-05,
      "step": 6902,
      "training_step_time": 10.290053367614746
    },
    {
      "epoch": 2.6332855224609373e-05,
      "model_forward_time": 3.403503894805908,
      "step": 6903
    },
    {
      "epoch": 2.6332855224609373e-05,
      "step": 6903,
      "training_step_time": 10.17800498008728
    },
    {
      "epoch": 2.6332855224609373e-05,
      "model_forward_time": 3.138667106628418,
      "step": 6903
    },
    {
      "epoch": 2.6332855224609373e-05,
      "step": 6903,
      "training_step_time": 9.915002584457397
    },
    {
      "epoch": 2.6332855224609373e-05,
      "model_forward_time": 3.138455390930176,
      "step": 6903
    },
    {
      "epoch": 2.6332855224609373e-05,
      "step": 6903,
      "training_step_time": 9.909111261367798
    },
    {
      "epoch": 2.6332855224609373e-05,
      "model_forward_time": 3.136514663696289,
      "step": 6903
    },
    {
      "epoch": 2.6332855224609373e-05,
      "step": 6903,
      "training_step_time": 10.027896642684937
    },
    {
      "epoch": 2.6336669921875e-05,
      "model_forward_time": 3.1515049934387207,
      "step": 6904
    },
    {
      "epoch": 2.6336669921875e-05,
      "step": 6904,
      "training_step_time": 9.925110340118408
    },
    {
      "epoch": 2.6336669921875e-05,
      "model_forward_time": 3.425307035446167,
      "step": 6904
    },
    {
      "epoch": 2.6336669921875e-05,
      "step": 6904,
      "training_step_time": 10.191720008850098
    },
    {
      "epoch": 2.6336669921875e-05,
      "model_forward_time": 3.1318368911743164,
      "step": 6904
    },
    {
      "epoch": 2.6336669921875e-05,
      "step": 6904,
      "training_step_time": 9.90249228477478
    },
    {
      "epoch": 2.6336669921875e-05,
      "model_forward_time": 3.1367766857147217,
      "step": 6904
    },
    {
      "epoch": 2.6336669921875e-05,
      "step": 6904,
      "training_step_time": 10.019602298736572
    },
    {
      "epoch": 2.6340484619140626e-05,
      "model_forward_time": 3.15659236907959,
      "step": 6905
    },
    {
      "dynamics_loss_avg": 0.07667082250118255,
      "epoch": 2.6340484619140626e-05,
      "step": 6905
    },
    {
      "action_loss_avg": 0.02877771407365799,
      "epoch": 2.6340484619140626e-05,
      "step": 6905
    },
    {
      "epoch": 2.6340484619140626e-05,
      "step": 6905,
      "training_step_time": 9.930095672607422
    },
    {
      "epoch": 2.6340484619140626e-05,
      "model_forward_time": 3.1372580528259277,
      "step": 6905
    },
    {
      "epoch": 2.6340484619140626e-05,
      "step": 6905,
      "training_step_time": 9.90896987915039
    },
    {
      "epoch": 2.6340484619140626e-05,
      "model_forward_time": 3.1419968605041504,
      "step": 6905
    },
    {
      "epoch": 2.6340484619140626e-05,
      "step": 6905,
      "training_step_time": 10.200653791427612
    },
    {
      "epoch": 2.6340484619140626e-05,
      "model_forward_time": 3.142205238342285,
      "step": 6905
    },
    {
      "epoch": 2.6340484619140626e-05,
      "step": 6905,
      "training_step_time": 10.29330825805664
    },
    {
      "epoch": 2.634429931640625e-05,
      "model_forward_time": 3.179067373275757,
      "step": 6906
    },
    {
      "epoch": 2.634429931640625e-05,
      "step": 6906,
      "training_step_time": 9.947940349578857
    },
    {
      "epoch": 2.634429931640625e-05,
      "model_forward_time": 3.1321589946746826,
      "step": 6906
    },
    {
      "epoch": 2.634429931640625e-05,
      "step": 6906,
      "training_step_time": 9.900362014770508
    },
    {
      "epoch": 2.634429931640625e-05,
      "model_forward_time": 3.1390187740325928,
      "step": 6906
    },
    {
      "epoch": 2.634429931640625e-05,
      "step": 6906,
      "training_step_time": 9.909842014312744
    },
    {
      "epoch": 2.634429931640625e-05,
      "model_forward_time": 3.451854944229126,
      "step": 6906
    },
    {
      "epoch": 2.634429931640625e-05,
      "step": 6906,
      "training_step_time": 10.31739330291748
    },
    {
      "epoch": 2.6348114013671875e-05,
      "model_forward_time": 3.1481740474700928,
      "step": 6907
    },
    {
      "epoch": 2.6348114013671875e-05,
      "step": 6907,
      "training_step_time": 9.917173385620117
    },
    {
      "epoch": 2.6348114013671875e-05,
      "model_forward_time": 3.1319127082824707,
      "step": 6907
    },
    {
      "epoch": 2.6348114013671875e-05,
      "step": 6907,
      "training_step_time": 9.904594421386719
    },
    {
      "epoch": 2.6348114013671875e-05,
      "model_forward_time": 3.139554977416992,
      "step": 6907
    },
    {
      "dynamics_loss_avg": 0.08560301959514618,
      "epoch": 2.6348114013671875e-05,
      "step": 6907
    },
    {
      "action_loss_avg": 0.029585908353328704,
      "epoch": 2.6348114013671875e-05,
      "step": 6907
    },
    {
      "epoch": 2.6348114013671875e-05,
      "step": 6907,
      "training_step_time": 9.91495394706726
    },
    {
      "epoch": 2.6348114013671875e-05,
      "model_forward_time": 3.1393327713012695,
      "step": 6907
    },
    {
      "epoch": 2.6348114013671875e-05,
      "step": 6907,
      "training_step_time": 10.31620192527771
    },
    {
      "epoch": 2.63519287109375e-05,
      "model_forward_time": 3.1741693019866943,
      "step": 6908
    },
    {
      "epoch": 2.63519287109375e-05,
      "step": 6908,
      "training_step_time": 10.214366436004639
    },
    {
      "epoch": 2.63519287109375e-05,
      "model_forward_time": 3.1357696056365967,
      "step": 6908
    },
    {
      "epoch": 2.63519287109375e-05,
      "step": 6908,
      "training_step_time": 9.906896352767944
    },
    {
      "epoch": 2.63519287109375e-05,
      "model_forward_time": 3.13517165184021,
      "step": 6908
    },
    {
      "epoch": 2.63519287109375e-05,
      "step": 6908,
      "training_step_time": 9.90666937828064
    },
    {
      "epoch": 2.63519287109375e-05,
      "model_forward_time": 3.134918689727783,
      "step": 6908
    },
    {
      "epoch": 2.63519287109375e-05,
      "step": 6908,
      "training_step_time": 10.266703844070435
    },
    {
      "epoch": 2.6355743408203124e-05,
      "model_forward_time": 3.1943657398223877,
      "step": 6909
    },
    {
      "epoch": 2.6355743408203124e-05,
      "step": 6909,
      "training_step_time": 9.960801601409912
    },
    {
      "epoch": 2.6355743408203124e-05,
      "model_forward_time": 3.4181230068206787,
      "step": 6909
    },
    {
      "epoch": 2.6355743408203124e-05,
      "step": 6909,
      "training_step_time": 10.183897733688354
    },
    {
      "epoch": 2.6355743408203124e-05,
      "model_forward_time": 3.13830304145813,
      "step": 6909
    },
    {
      "epoch": 2.6355743408203124e-05,
      "step": 6909,
      "training_step_time": 9.91208004951477
    },
    {
      "epoch": 2.6355743408203124e-05,
      "model_forward_time": 3.136180877685547,
      "step": 6909
    },
    {
      "epoch": 2.6355743408203124e-05,
      "step": 6909,
      "training_step_time": 10.02974796295166
    },
    {
      "epoch": 2.635955810546875e-05,
      "grad_norm": 0.42329755425453186,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 6910
    },
    {
      "epoch": 2.635955810546875e-05,
      "model_forward_time": 3.1712658405303955,
      "step": 6910
    },
    {
      "dynamics_loss_avg": 0.06761384550482034,
      "epoch": 2.635955810546875e-05,
      "step": 6910
    },
    {
      "action_loss_avg": 0.02300518867559731,
      "epoch": 2.635955810546875e-05,
      "step": 6910
    },
    {
      "epoch": 2.635955810546875e-05,
      "step": 6910,
      "training_step_time": 9.946657657623291
    },
    {
      "epoch": 2.635955810546875e-05,
      "model_forward_time": 3.143556833267212,
      "step": 6910
    },
    {
      "epoch": 2.635955810546875e-05,
      "step": 6910,
      "training_step_time": 9.918338298797607
    },
    {
      "epoch": 2.635955810546875e-05,
      "model_forward_time": 3.4345028400421143,
      "step": 6910
    },
    {
      "epoch": 2.635955810546875e-05,
      "step": 6910,
      "training_step_time": 10.20712685585022
    },
    {
      "epoch": 2.635955810546875e-05,
      "model_forward_time": 3.13393497467041,
      "step": 6910
    },
    {
      "epoch": 2.635955810546875e-05,
      "step": 6910,
      "training_step_time": 10.262684345245361
    },
    {
      "epoch": 2.6363372802734373e-05,
      "model_forward_time": 3.1629984378814697,
      "step": 6911
    },
    {
      "epoch": 2.6363372802734373e-05,
      "step": 6911,
      "training_step_time": 9.933662414550781
    },
    {
      "epoch": 2.6363372802734373e-05,
      "model_forward_time": 3.1389873027801514,
      "step": 6911
    },
    {
      "epoch": 2.6363372802734373e-05,
      "step": 6911,
      "training_step_time": 9.906797647476196
    },
    {
      "epoch": 2.6363372802734373e-05,
      "model_forward_time": 3.1361143589019775,
      "step": 6911
    },
    {
      "epoch": 2.6363372802734373e-05,
      "step": 6911,
      "training_step_time": 9.903804540634155
    },
    {
      "epoch": 2.6363372802734373e-05,
      "model_forward_time": 3.410083532333374,
      "step": 6911
    },
    {
      "epoch": 2.6363372802734373e-05,
      "step": 6911,
      "training_step_time": 10.274077653884888
    },
    {
      "epoch": 2.63671875e-05,
      "model_forward_time": 3.1686201095581055,
      "step": 6912
    },
    {
      "epoch": 2.63671875e-05,
      "step": 6912,
      "training_step_time": 9.939970254898071
    },
    {
      "epoch": 2.63671875e-05,
      "model_forward_time": 3.1362006664276123,
      "step": 6912
    },
    {
      "epoch": 2.63671875e-05,
      "step": 6912,
      "training_step_time": 9.906580448150635
    },
    {
      "epoch": 2.63671875e-05,
      "model_forward_time": 3.1352200508117676,
      "step": 6912
    },
    {
      "dynamics_loss_avg": 0.064267086237669,
      "epoch": 2.63671875e-05,
      "step": 6912
    },
    {
      "action_loss_avg": 0.02701577786356211,
      "epoch": 2.63671875e-05,
      "step": 6912
    },
    {
      "epoch": 2.63671875e-05,
      "step": 6912,
      "training_step_time": 9.912210702896118
    },
    {
      "epoch": 2.63671875e-05,
      "model_forward_time": 3.1425626277923584,
      "step": 6912
    },
    {
      "epoch": 2.63671875e-05,
      "step": 6912,
      "training_step_time": 10.26312804222107
    },
    {
      "epoch": 2.6371002197265626e-05,
      "model_forward_time": 3.416119337081909,
      "step": 6913
    },
    {
      "epoch": 2.6371002197265626e-05,
      "step": 6913,
      "training_step_time": 10.192272424697876
    },
    {
      "epoch": 2.6371002197265626e-05,
      "model_forward_time": 3.1451377868652344,
      "step": 6913
    },
    {
      "epoch": 2.6371002197265626e-05,
      "step": 6913,
      "training_step_time": 9.917443037033081
    },
    {
      "epoch": 2.6371002197265626e-05,
      "model_forward_time": 3.141828775405884,
      "step": 6913
    },
    {
      "epoch": 2.6371002197265626e-05,
      "step": 6913,
      "training_step_time": 9.911229848861694
    },
    {
      "epoch": 2.6371002197265626e-05,
      "model_forward_time": 3.1392951011657715,
      "step": 6913
    },
    {
      "epoch": 2.6371002197265626e-05,
      "step": 6913,
      "training_step_time": 10.019619226455688
    },
    {
      "epoch": 2.637481689453125e-05,
      "model_forward_time": 3.1500020027160645,
      "step": 6914
    },
    {
      "epoch": 2.637481689453125e-05,
      "step": 6914,
      "training_step_time": 9.917933702468872
    },
    {
      "epoch": 2.637481689453125e-05,
      "model_forward_time": 3.1325910091400146,
      "step": 6914
    },
    {
      "epoch": 2.637481689453125e-05,
      "step": 6914,
      "training_step_time": 10.178731918334961
    },
    {
      "epoch": 2.637481689453125e-05,
      "model_forward_time": 3.1432993412017822,
      "step": 6914
    },
    {
      "epoch": 2.637481689453125e-05,
      "step": 6914,
      "training_step_time": 9.911449193954468
    },
    {
      "epoch": 2.637481689453125e-05,
      "model_forward_time": 3.137700080871582,
      "step": 6914
    },
    {
      "epoch": 2.637481689453125e-05,
      "step": 6914,
      "training_step_time": 10.017762184143066
    },
    {
      "epoch": 2.6378631591796875e-05,
      "model_forward_time": 3.149087429046631,
      "step": 6915
    },
    {
      "dynamics_loss_avg": 0.06616183631122112,
      "epoch": 2.6378631591796875e-05,
      "step": 6915
    },
    {
      "action_loss_avg": 0.02718278141692281,
      "epoch": 2.6378631591796875e-05,
      "step": 6915
    },
    {
      "epoch": 2.6378631591796875e-05,
      "step": 6915,
      "training_step_time": 9.921375751495361
    },
    {
      "epoch": 2.6378631591796875e-05,
      "model_forward_time": 3.1301963329315186,
      "step": 6915
    },
    {
      "epoch": 2.6378631591796875e-05,
      "step": 6915,
      "training_step_time": 9.899958848953247
    },
    {
      "epoch": 2.6378631591796875e-05,
      "model_forward_time": 3.402656078338623,
      "step": 6915
    },
    {
      "epoch": 2.6378631591796875e-05,
      "step": 6915,
      "training_step_time": 10.1747465133667
    },
    {
      "epoch": 2.6378631591796875e-05,
      "model_forward_time": 3.142862319946289,
      "step": 6915
    },
    {
      "epoch": 2.6378631591796875e-05,
      "step": 6915,
      "training_step_time": 10.230303049087524
    },
    {
      "epoch": 2.63824462890625e-05,
      "model_forward_time": 3.1535251140594482,
      "step": 6916
    },
    {
      "epoch": 2.63824462890625e-05,
      "step": 6916,
      "training_step_time": 9.936719417572021
    },
    {
      "epoch": 2.63824462890625e-05,
      "model_forward_time": 3.1459102630615234,
      "step": 6916
    },
    {
      "epoch": 2.63824462890625e-05,
      "step": 6916,
      "training_step_time": 9.923585414886475
    },
    {
      "epoch": 2.63824462890625e-05,
      "model_forward_time": 3.143531322479248,
      "step": 6916
    },
    {
      "epoch": 2.63824462890625e-05,
      "step": 6916,
      "training_step_time": 9.922923564910889
    },
    {
      "epoch": 2.63824462890625e-05,
      "model_forward_time": 3.1517579555511475,
      "step": 6916
    },
    {
      "epoch": 2.63824462890625e-05,
      "step": 6916,
      "training_step_time": 10.298343181610107
    },
    {
      "epoch": 2.6386260986328124e-05,
      "model_forward_time": 3.1620383262634277,
      "step": 6917
    },
    {
      "epoch": 2.6386260986328124e-05,
      "step": 6917,
      "training_step_time": 9.939604997634888
    },
    {
      "epoch": 2.6386260986328124e-05,
      "model_forward_time": 3.157249927520752,
      "step": 6917
    },
    {
      "epoch": 2.6386260986328124e-05,
      "step": 6917,
      "training_step_time": 9.934820890426636
    },
    {
      "epoch": 2.6386260986328124e-05,
      "model_forward_time": 3.1519265174865723,
      "step": 6917
    },
    {
      "dynamics_loss_avg": 0.0432195742148906,
      "epoch": 2.6386260986328124e-05,
      "step": 6917
    },
    {
      "action_loss_avg": 0.013864857458975166,
      "epoch": 2.6386260986328124e-05,
      "step": 6917
    },
    {
      "epoch": 2.6386260986328124e-05,
      "step": 6917,
      "training_step_time": 9.925361156463623
    },
    {
      "epoch": 2.6386260986328124e-05,
      "model_forward_time": 3.1416072845458984,
      "step": 6917
    },
    {
      "epoch": 2.6386260986328124e-05,
      "step": 6917,
      "training_step_time": 10.019010543823242
    },
    {
      "epoch": 2.639007568359375e-05,
      "model_forward_time": 3.4706246852874756,
      "step": 6918
    },
    {
      "epoch": 2.639007568359375e-05,
      "step": 6918,
      "training_step_time": 10.24247407913208
    },
    {
      "epoch": 2.639007568359375e-05,
      "model_forward_time": 3.1453938484191895,
      "step": 6918
    },
    {
      "epoch": 2.639007568359375e-05,
      "step": 6918,
      "training_step_time": 9.916054964065552
    },
    {
      "epoch": 2.639007568359375e-05,
      "model_forward_time": 3.1480181217193604,
      "step": 6918
    },
    {
      "epoch": 2.639007568359375e-05,
      "step": 6918,
      "training_step_time": 9.920785188674927
    },
    {
      "epoch": 2.639007568359375e-05,
      "model_forward_time": 3.1514110565185547,
      "step": 6918
    },
    {
      "epoch": 2.639007568359375e-05,
      "step": 6918,
      "training_step_time": 10.297866106033325
    },
    {
      "epoch": 2.6393890380859373e-05,
      "model_forward_time": 3.169983386993408,
      "step": 6919
    },
    {
      "epoch": 2.6393890380859373e-05,
      "step": 6919,
      "training_step_time": 9.942108392715454
    },
    {
      "epoch": 2.6393890380859373e-05,
      "model_forward_time": 3.149142265319824,
      "step": 6919
    },
    {
      "epoch": 2.6393890380859373e-05,
      "step": 6919,
      "training_step_time": 10.183731079101562
    },
    {
      "epoch": 2.6393890380859373e-05,
      "model_forward_time": 3.1471879482269287,
      "step": 6919
    },
    {
      "epoch": 2.6393890380859373e-05,
      "step": 6919,
      "training_step_time": 9.91989254951477
    },
    {
      "epoch": 2.6393890380859373e-05,
      "model_forward_time": 3.14404034614563,
      "step": 6919
    },
    {
      "epoch": 2.6393890380859373e-05,
      "step": 6919,
      "training_step_time": 10.24507761001587
    },
    {
      "epoch": 2.6397705078125e-05,
      "grad_norm": 0.6125026345252991,
      "learning_rate": 1e-05,
      "loss": 0.0834,
      "step": 6920
    },
    {
      "epoch": 2.6397705078125e-05,
      "model_forward_time": 3.1910791397094727,
      "step": 6920
    },
    {
      "dynamics_loss_avg": 0.06146290563046932,
      "epoch": 2.6397705078125e-05,
      "step": 6920
    },
    {
      "action_loss_avg": 0.022472689114511012,
      "epoch": 2.6397705078125e-05,
      "step": 6920
    },
    {
      "epoch": 2.6397705078125e-05,
      "step": 6920,
      "training_step_time": 9.970204591751099
    },
    {
      "epoch": 2.6397705078125e-05,
      "model_forward_time": 3.1484603881835938,
      "step": 6920
    },
    {
      "epoch": 2.6397705078125e-05,
      "step": 6920,
      "training_step_time": 9.920893430709839
    },
    {
      "epoch": 2.6397705078125e-05,
      "model_forward_time": 3.4296863079071045,
      "step": 6920
    },
    {
      "epoch": 2.6397705078125e-05,
      "step": 6920,
      "training_step_time": 10.201021432876587
    },
    {
      "epoch": 2.6397705078125e-05,
      "model_forward_time": 3.1484670639038086,
      "step": 6920
    },
    {
      "epoch": 2.6397705078125e-05,
      "step": 6920,
      "training_step_time": 10.306539058685303
    },
    {
      "epoch": 2.6401519775390626e-05,
      "model_forward_time": 3.1908674240112305,
      "step": 6921
    },
    {
      "epoch": 2.6401519775390626e-05,
      "step": 6921,
      "training_step_time": 9.968838691711426
    },
    {
      "epoch": 2.6401519775390626e-05,
      "model_forward_time": 3.148155450820923,
      "step": 6921
    },
    {
      "epoch": 2.6401519775390626e-05,
      "step": 6921,
      "training_step_time": 9.920023441314697
    },
    {
      "epoch": 2.6401519775390626e-05,
      "model_forward_time": 3.146959066390991,
      "step": 6921
    },
    {
      "epoch": 2.6401519775390626e-05,
      "step": 6921,
      "training_step_time": 9.922120809555054
    },
    {
      "epoch": 2.6401519775390626e-05,
      "model_forward_time": 3.4085164070129395,
      "step": 6921
    },
    {
      "epoch": 2.6401519775390626e-05,
      "step": 6921,
      "training_step_time": 10.270890474319458
    },
    {
      "epoch": 2.640533447265625e-05,
      "model_forward_time": 3.1780245304107666,
      "step": 6922
    },
    {
      "epoch": 2.640533447265625e-05,
      "step": 6922,
      "training_step_time": 9.949011087417603
    },
    {
      "epoch": 2.640533447265625e-05,
      "model_forward_time": 3.1449553966522217,
      "step": 6922
    },
    {
      "epoch": 2.640533447265625e-05,
      "step": 6922,
      "training_step_time": 9.91295337677002
    },
    {
      "epoch": 2.640533447265625e-05,
      "model_forward_time": 3.1433145999908447,
      "step": 6922
    },
    {
      "dynamics_loss_avg": 0.06751979496330023,
      "epoch": 2.640533447265625e-05,
      "step": 6922
    },
    {
      "action_loss_avg": 0.02535421708598733,
      "epoch": 2.640533447265625e-05,
      "step": 6922
    },
    {
      "epoch": 2.640533447265625e-05,
      "step": 6922,
      "training_step_time": 9.914958953857422
    },
    {
      "epoch": 2.640533447265625e-05,
      "model_forward_time": 3.1463844776153564,
      "step": 6922
    },
    {
      "epoch": 2.640533447265625e-05,
      "step": 6922,
      "training_step_time": 10.277563095092773
    },
    {
      "epoch": 2.6409149169921875e-05,
      "model_forward_time": 3.4453113079071045,
      "step": 6923
    },
    {
      "epoch": 2.6409149169921875e-05,
      "step": 6923,
      "training_step_time": 10.214430332183838
    },
    {
      "epoch": 2.6409149169921875e-05,
      "model_forward_time": 2.4309911727905273,
      "step": 6923
    },
    {
      "epoch": 2.6409149169921875e-05,
      "step": 6923,
      "training_step_time": 7.454762697219849
    },
    {
      "epoch": 2.6409149169921875e-05,
      "model_forward_time": 3.1313016414642334,
      "step": 6923
    },
    {
      "epoch": 2.6409149169921875e-05,
      "step": 6923,
      "training_step_time": 9.898913145065308
    },
    {
      "epoch": 2.6409149169921875e-05,
      "model_forward_time": 3.138834238052368,
      "step": 6923
    },
    {
      "epoch": 2.6409149169921875e-05,
      "step": 6923,
      "training_step_time": 10.264472484588623
    },
    {
      "epoch": 2.64129638671875e-05,
      "model_forward_time": 3.160829782485962,
      "step": 6924
    },
    {
      "epoch": 2.64129638671875e-05,
      "step": 6924,
      "training_step_time": 9.930266618728638
    },
    {
      "epoch": 2.64129638671875e-05,
      "model_forward_time": 3.419074058532715,
      "step": 6924
    },
    {
      "epoch": 2.64129638671875e-05,
      "step": 6924,
      "training_step_time": 10.194735288619995
    },
    {
      "epoch": 2.64129638671875e-05,
      "model_forward_time": 2.4361722469329834,
      "step": 6924
    },
    {
      "epoch": 2.64129638671875e-05,
      "step": 6924,
      "training_step_time": 7.468116521835327
    },
    {
      "epoch": 2.64129638671875e-05,
      "model_forward_time": 3.138150930404663,
      "step": 6924
    },
    {
      "epoch": 2.64129638671875e-05,
      "step": 6924,
      "training_step_time": 10.026298522949219
    },
    {
      "epoch": 2.6416778564453124e-05,
      "model_forward_time": 3.1601855754852295,
      "step": 6925
    },
    {
      "dynamics_loss_avg": 0.059778769686818126,
      "epoch": 2.6416778564453124e-05,
      "step": 6925
    },
    {
      "action_loss_avg": 0.021182967629283667,
      "epoch": 2.6416778564453124e-05,
      "step": 6925
    },
    {
      "epoch": 2.6416778564453124e-05,
      "step": 6925,
      "training_step_time": 9.946596384048462
    },
    {
      "epoch": 2.6416778564453124e-05,
      "model_forward_time": 3.1390275955200195,
      "step": 6925
    },
    {
      "epoch": 2.6416778564453124e-05,
      "step": 6925,
      "training_step_time": 9.914294004440308
    },
    {
      "epoch": 2.6416778564453124e-05,
      "model_forward_time": 3.1366677284240723,
      "step": 6925
    },
    {
      "epoch": 2.6416778564453124e-05,
      "step": 6925,
      "training_step_time": 10.22215723991394
    },
    {
      "epoch": 2.6416778564453124e-05,
      "model_forward_time": 3.143298864364624,
      "step": 6925
    },
    {
      "epoch": 2.6416778564453124e-05,
      "step": 6925,
      "training_step_time": 10.26514983177185
    },
    {
      "epoch": 2.642059326171875e-05,
      "model_forward_time": 3.148264169692993,
      "step": 6926
    },
    {
      "epoch": 2.642059326171875e-05,
      "step": 6926,
      "training_step_time": 9.920815706253052
    },
    {
      "epoch": 2.642059326171875e-05,
      "model_forward_time": 3.135732650756836,
      "step": 6926
    },
    {
      "epoch": 2.642059326171875e-05,
      "step": 6926,
      "training_step_time": 9.91214108467102
    },
    {
      "epoch": 2.642059326171875e-05,
      "model_forward_time": 3.1398584842681885,
      "step": 6926
    },
    {
      "epoch": 2.642059326171875e-05,
      "step": 6926,
      "training_step_time": 9.918641805648804
    },
    {
      "epoch": 2.642059326171875e-05,
      "model_forward_time": 3.4539012908935547,
      "step": 6926
    },
    {
      "epoch": 2.642059326171875e-05,
      "step": 6926,
      "training_step_time": 10.32991647720337
    },
    {
      "epoch": 2.6424407958984373e-05,
      "model_forward_time": 3.1550211906433105,
      "step": 6927
    },
    {
      "epoch": 2.6424407958984373e-05,
      "step": 6927,
      "training_step_time": 9.934918642044067
    },
    {
      "epoch": 2.6424407958984373e-05,
      "model_forward_time": 3.135037422180176,
      "step": 6927
    },
    {
      "epoch": 2.6424407958984373e-05,
      "step": 6927,
      "training_step_time": 9.912048816680908
    },
    {
      "epoch": 2.6424407958984373e-05,
      "model_forward_time": 3.139418840408325,
      "step": 6927
    },
    {
      "dynamics_loss_avg": 0.06647697538137436,
      "epoch": 2.6424407958984373e-05,
      "step": 6927
    },
    {
      "action_loss_avg": 0.025982115417718887,
      "epoch": 2.6424407958984373e-05,
      "step": 6927
    },
    {
      "epoch": 2.6424407958984373e-05,
      "step": 6927,
      "training_step_time": 9.917280435562134
    },
    {
      "epoch": 2.6424407958984373e-05,
      "model_forward_time": 3.1374974250793457,
      "step": 6927
    },
    {
      "epoch": 2.6424407958984373e-05,
      "step": 6927,
      "training_step_time": 10.293012857437134
    },
    {
      "epoch": 2.642822265625e-05,
      "model_forward_time": 3.158590793609619,
      "step": 6928
    },
    {
      "epoch": 2.642822265625e-05,
      "step": 6928,
      "training_step_time": 10.211432933807373
    },
    {
      "epoch": 2.642822265625e-05,
      "model_forward_time": 3.1314401626586914,
      "step": 6928
    },
    {
      "epoch": 2.642822265625e-05,
      "step": 6928,
      "training_step_time": 9.90390133857727
    },
    {
      "epoch": 2.642822265625e-05,
      "model_forward_time": 3.137211799621582,
      "step": 6928
    },
    {
      "epoch": 2.642822265625e-05,
      "step": 6928,
      "training_step_time": 9.91926646232605
    },
    {
      "epoch": 2.642822265625e-05,
      "model_forward_time": 3.1396539211273193,
      "step": 6928
    },
    {
      "epoch": 2.642822265625e-05,
      "step": 6928,
      "training_step_time": 10.26188850402832
    },
    {
      "epoch": 2.6432037353515626e-05,
      "model_forward_time": 3.1670758724212646,
      "step": 6929
    },
    {
      "epoch": 2.6432037353515626e-05,
      "step": 6929,
      "training_step_time": 9.94278073310852
    },
    {
      "epoch": 2.6432037353515626e-05,
      "model_forward_time": 3.432405471801758,
      "step": 6929
    },
    {
      "epoch": 2.6432037353515626e-05,
      "step": 6929,
      "training_step_time": 10.206488609313965
    },
    {
      "epoch": 2.6432037353515626e-05,
      "model_forward_time": 3.1420397758483887,
      "step": 6929
    },
    {
      "epoch": 2.6432037353515626e-05,
      "step": 6929,
      "training_step_time": 9.919622659683228
    },
    {
      "epoch": 2.6432037353515626e-05,
      "model_forward_time": 3.139876127243042,
      "step": 6929
    },
    {
      "epoch": 2.6432037353515626e-05,
      "step": 6929,
      "training_step_time": 10.01174783706665
    },
    {
      "epoch": 2.643585205078125e-05,
      "grad_norm": 0.5395582318305969,
      "learning_rate": 1e-05,
      "loss": 0.0849,
      "step": 6930
    },
    {
      "epoch": 2.643585205078125e-05,
      "model_forward_time": 3.1466064453125,
      "step": 6930
    },
    {
      "dynamics_loss_avg": 0.061953655630350116,
      "epoch": 2.643585205078125e-05,
      "step": 6930
    },
    {
      "action_loss_avg": 0.027587727084755896,
      "epoch": 2.643585205078125e-05,
      "step": 6930
    },
    {
      "epoch": 2.643585205078125e-05,
      "step": 6930,
      "training_step_time": 9.930140018463135
    },
    {
      "epoch": 2.643585205078125e-05,
      "model_forward_time": 3.1405720710754395,
      "step": 6930
    },
    {
      "epoch": 2.643585205078125e-05,
      "step": 6930,
      "training_step_time": 9.918385028839111
    },
    {
      "epoch": 2.643585205078125e-05,
      "model_forward_time": 3.437098979949951,
      "step": 6930
    },
    {
      "epoch": 2.643585205078125e-05,
      "step": 6930,
      "training_step_time": 10.213286876678467
    },
    {
      "epoch": 2.643585205078125e-05,
      "model_forward_time": 3.1377270221710205,
      "step": 6930
    },
    {
      "epoch": 2.643585205078125e-05,
      "step": 6930,
      "training_step_time": 10.00580883026123
    },
    {
      "epoch": 2.6439666748046875e-05,
      "model_forward_time": 3.171905517578125,
      "step": 6931
    },
    {
      "epoch": 2.6439666748046875e-05,
      "step": 6931,
      "training_step_time": 9.948244333267212
    },
    {
      "epoch": 2.6439666748046875e-05,
      "model_forward_time": 3.140510320663452,
      "step": 6931
    },
    {
      "epoch": 2.6439666748046875e-05,
      "step": 6931,
      "training_step_time": 9.922752141952515
    },
    {
      "epoch": 2.6439666748046875e-05,
      "model_forward_time": 3.135654926300049,
      "step": 6931
    },
    {
      "epoch": 2.6439666748046875e-05,
      "step": 6931,
      "training_step_time": 9.91395092010498
    },
    {
      "epoch": 2.6439666748046875e-05,
      "model_forward_time": 3.3944270610809326,
      "step": 6931
    },
    {
      "epoch": 2.6439666748046875e-05,
      "step": 6931,
      "training_step_time": 10.268530368804932
    },
    {
      "epoch": 2.64434814453125e-05,
      "model_forward_time": 3.167616844177246,
      "step": 6932
    },
    {
      "epoch": 2.64434814453125e-05,
      "step": 6932,
      "training_step_time": 9.946183204650879
    },
    {
      "epoch": 2.64434814453125e-05,
      "model_forward_time": 3.1398043632507324,
      "step": 6932
    },
    {
      "epoch": 2.64434814453125e-05,
      "step": 6932,
      "training_step_time": 9.918516635894775
    },
    {
      "epoch": 2.64434814453125e-05,
      "model_forward_time": 3.1400930881500244,
      "step": 6932
    },
    {
      "dynamics_loss_avg": 0.057702387869358066,
      "epoch": 2.64434814453125e-05,
      "step": 6932
    },
    {
      "action_loss_avg": 0.02098501930013299,
      "epoch": 2.64434814453125e-05,
      "step": 6932
    },
    {
      "epoch": 2.64434814453125e-05,
      "step": 6932,
      "training_step_time": 9.918683290481567
    },
    {
      "epoch": 2.64434814453125e-05,
      "model_forward_time": 3.141996383666992,
      "step": 6932
    },
    {
      "epoch": 2.64434814453125e-05,
      "step": 6932,
      "training_step_time": 10.037321090698242
    },
    {
      "epoch": 2.6447296142578124e-05,
      "model_forward_time": 3.4450647830963135,
      "step": 6933
    },
    {
      "epoch": 2.6447296142578124e-05,
      "step": 6933,
      "training_step_time": 10.227404117584229
    },
    {
      "epoch": 2.6447296142578124e-05,
      "model_forward_time": 3.1415011882781982,
      "step": 6933
    },
    {
      "epoch": 2.6447296142578124e-05,
      "step": 6933,
      "training_step_time": 9.9170401096344
    },
    {
      "epoch": 2.6447296142578124e-05,
      "model_forward_time": 3.134657382965088,
      "step": 6933
    },
    {
      "epoch": 2.6447296142578124e-05,
      "step": 6933,
      "training_step_time": 9.912733554840088
    },
    {
      "epoch": 2.6447296142578124e-05,
      "model_forward_time": 3.1409637928009033,
      "step": 6933
    },
    {
      "epoch": 2.6447296142578124e-05,
      "step": 6933,
      "training_step_time": 10.267999410629272
    },
    {
      "epoch": 2.645111083984375e-05,
      "model_forward_time": 3.151519298553467,
      "step": 6934
    },
    {
      "epoch": 2.645111083984375e-05,
      "step": 6934,
      "training_step_time": 9.925471782684326
    },
    {
      "epoch": 2.645111083984375e-05,
      "model_forward_time": 3.136314630508423,
      "step": 6934
    },
    {
      "epoch": 2.645111083984375e-05,
      "step": 6934,
      "training_step_time": 10.220353126525879
    },
    {
      "epoch": 2.645111083984375e-05,
      "model_forward_time": 3.1355326175689697,
      "step": 6934
    },
    {
      "epoch": 2.645111083984375e-05,
      "step": 6934,
      "training_step_time": 9.908159255981445
    },
    {
      "epoch": 2.645111083984375e-05,
      "model_forward_time": 3.137816905975342,
      "step": 6934
    },
    {
      "epoch": 2.645111083984375e-05,
      "step": 6934,
      "training_step_time": 10.01183009147644
    },
    {
      "epoch": 2.6454925537109374e-05,
      "model_forward_time": 3.1490230560302734,
      "step": 6935
    },
    {
      "dynamics_loss_avg": 0.05043344553560018,
      "epoch": 2.6454925537109374e-05,
      "step": 6935
    },
    {
      "action_loss_avg": 0.019443607516586782,
      "epoch": 2.6454925537109374e-05,
      "step": 6935
    },
    {
      "epoch": 2.6454925537109374e-05,
      "step": 6935,
      "training_step_time": 9.92034649848938
    },
    {
      "epoch": 2.6454925537109374e-05,
      "model_forward_time": 3.1337244510650635,
      "step": 6935
    },
    {
      "epoch": 2.6454925537109374e-05,
      "step": 6935,
      "training_step_time": 9.907817125320435
    },
    {
      "epoch": 2.6454925537109374e-05,
      "model_forward_time": 3.4130516052246094,
      "step": 6935
    },
    {
      "epoch": 2.6454925537109374e-05,
      "step": 6935,
      "training_step_time": 10.189250469207764
    },
    {
      "epoch": 2.6454925537109374e-05,
      "model_forward_time": 3.139570713043213,
      "step": 6935
    },
    {
      "epoch": 2.6454925537109374e-05,
      "step": 6935,
      "training_step_time": 10.015891551971436
    },
    {
      "epoch": 2.6458740234375e-05,
      "model_forward_time": 3.151918888092041,
      "step": 6936
    },
    {
      "epoch": 2.6458740234375e-05,
      "step": 6936,
      "training_step_time": 9.93001937866211
    },
    {
      "epoch": 2.6458740234375e-05,
      "model_forward_time": 3.137491464614868,
      "step": 6936
    },
    {
      "epoch": 2.6458740234375e-05,
      "step": 6936,
      "training_step_time": 9.916716575622559
    },
    {
      "epoch": 2.6458740234375e-05,
      "model_forward_time": 3.1414902210235596,
      "step": 6936
    },
    {
      "epoch": 2.6458740234375e-05,
      "step": 6936,
      "training_step_time": 9.91592264175415
    },
    {
      "epoch": 2.6458740234375e-05,
      "model_forward_time": 3.13369083404541,
      "step": 6936
    },
    {
      "epoch": 2.6458740234375e-05,
      "step": 6936,
      "training_step_time": 10.276753902435303
    },
    {
      "epoch": 2.6462554931640626e-05,
      "model_forward_time": 3.163524866104126,
      "step": 6937
    },
    {
      "epoch": 2.6462554931640626e-05,
      "step": 6937,
      "training_step_time": 9.93833613395691
    },
    {
      "epoch": 2.6462554931640626e-05,
      "model_forward_time": 3.1364686489105225,
      "step": 6937
    },
    {
      "epoch": 2.6462554931640626e-05,
      "step": 6937,
      "training_step_time": 9.911199808120728
    },
    {
      "epoch": 2.6462554931640626e-05,
      "model_forward_time": 3.140815258026123,
      "step": 6937
    },
    {
      "dynamics_loss_avg": 0.058392465487122534,
      "epoch": 2.6462554931640626e-05,
      "step": 6937
    },
    {
      "action_loss_avg": 0.026283145137131213,
      "epoch": 2.6462554931640626e-05,
      "step": 6937
    },
    {
      "epoch": 2.6462554931640626e-05,
      "step": 6937,
      "training_step_time": 9.919588804244995
    },
    {
      "epoch": 2.6462554931640626e-05,
      "model_forward_time": 3.141047477722168,
      "step": 6937
    },
    {
      "epoch": 2.6462554931640626e-05,
      "step": 6937,
      "training_step_time": 10.016870021820068
    },
    {
      "epoch": 2.646636962890625e-05,
      "model_forward_time": 3.4275383949279785,
      "step": 6938
    },
    {
      "epoch": 2.646636962890625e-05,
      "step": 6938,
      "training_step_time": 10.204441785812378
    },
    {
      "epoch": 2.646636962890625e-05,
      "model_forward_time": 3.1374499797821045,
      "step": 6938
    },
    {
      "epoch": 2.646636962890625e-05,
      "step": 6938,
      "training_step_time": 9.920390605926514
    },
    {
      "epoch": 2.646636962890625e-05,
      "model_forward_time": 3.1426422595977783,
      "step": 6938
    },
    {
      "epoch": 2.646636962890625e-05,
      "step": 6938,
      "training_step_time": 9.923582077026367
    },
    {
      "epoch": 2.646636962890625e-05,
      "model_forward_time": 3.142063856124878,
      "step": 6938
    },
    {
      "epoch": 2.646636962890625e-05,
      "step": 6938,
      "training_step_time": 10.299091100692749
    },
    {
      "epoch": 2.6470184326171875e-05,
      "model_forward_time": 3.1783061027526855,
      "step": 6939
    },
    {
      "epoch": 2.6470184326171875e-05,
      "step": 6939,
      "training_step_time": 9.95969295501709
    },
    {
      "epoch": 2.6470184326171875e-05,
      "model_forward_time": 3.14536190032959,
      "step": 6939
    },
    {
      "epoch": 2.6470184326171875e-05,
      "step": 6939,
      "training_step_time": 10.192854166030884
    },
    {
      "epoch": 2.6470184326171875e-05,
      "model_forward_time": 3.146418571472168,
      "step": 6939
    },
    {
      "epoch": 2.6470184326171875e-05,
      "step": 6939,
      "training_step_time": 9.929867267608643
    },
    {
      "epoch": 2.6470184326171875e-05,
      "model_forward_time": 3.146250009536743,
      "step": 6939
    },
    {
      "epoch": 2.6470184326171875e-05,
      "step": 6939,
      "training_step_time": 10.057952165603638
    },
    {
      "epoch": 2.64739990234375e-05,
      "grad_norm": 0.4881359934806824,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 6940
    },
    {
      "epoch": 2.64739990234375e-05,
      "model_forward_time": 3.159170150756836,
      "step": 6940
    },
    {
      "dynamics_loss_avg": 0.07278832979500294,
      "epoch": 2.64739990234375e-05,
      "step": 6940
    },
    {
      "action_loss_avg": 0.03317682622000575,
      "epoch": 2.64739990234375e-05,
      "step": 6940
    },
    {
      "epoch": 2.64739990234375e-05,
      "step": 6940,
      "training_step_time": 9.943219900131226
    },
    {
      "epoch": 2.64739990234375e-05,
      "model_forward_time": 3.141781806945801,
      "step": 6940
    },
    {
      "epoch": 2.64739990234375e-05,
      "step": 6940,
      "training_step_time": 9.917848825454712
    },
    {
      "epoch": 2.64739990234375e-05,
      "model_forward_time": 3.430959463119507,
      "step": 6940
    },
    {
      "epoch": 2.64739990234375e-05,
      "step": 6940,
      "training_step_time": 10.210672855377197
    },
    {
      "epoch": 2.64739990234375e-05,
      "model_forward_time": 3.1392953395843506,
      "step": 6940
    },
    {
      "epoch": 2.64739990234375e-05,
      "step": 6940,
      "training_step_time": 10.026898384094238
    },
    {
      "epoch": 2.6477813720703124e-05,
      "model_forward_time": 3.1742043495178223,
      "step": 6941
    },
    {
      "epoch": 2.6477813720703124e-05,
      "step": 6941,
      "training_step_time": 9.951206684112549
    },
    {
      "epoch": 2.6477813720703124e-05,
      "model_forward_time": 3.1501519680023193,
      "step": 6941
    },
    {
      "epoch": 2.6477813720703124e-05,
      "step": 6941,
      "training_step_time": 9.931249618530273
    },
    {
      "epoch": 2.6477813720703124e-05,
      "model_forward_time": 3.138808250427246,
      "step": 6941
    },
    {
      "epoch": 2.6477813720703124e-05,
      "step": 6941,
      "training_step_time": 9.916513919830322
    },
    {
      "epoch": 2.6477813720703124e-05,
      "model_forward_time": 3.422259569168091,
      "step": 6941
    },
    {
      "epoch": 2.6477813720703124e-05,
      "step": 6941,
      "training_step_time": 10.311120510101318
    },
    {
      "epoch": 2.648162841796875e-05,
      "model_forward_time": 3.17165207862854,
      "step": 6942
    },
    {
      "epoch": 2.648162841796875e-05,
      "step": 6942,
      "training_step_time": 9.945791959762573
    },
    {
      "epoch": 2.648162841796875e-05,
      "model_forward_time": 3.1345653533935547,
      "step": 6942
    },
    {
      "epoch": 2.648162841796875e-05,
      "step": 6942,
      "training_step_time": 9.908447265625
    },
    {
      "epoch": 2.648162841796875e-05,
      "model_forward_time": 3.138932228088379,
      "step": 6942
    },
    {
      "dynamics_loss_avg": 0.0515323307365179,
      "epoch": 2.648162841796875e-05,
      "step": 6942
    },
    {
      "action_loss_avg": 0.017267325520515443,
      "epoch": 2.648162841796875e-05,
      "step": 6942
    },
    {
      "epoch": 2.648162841796875e-05,
      "step": 6942,
      "training_step_time": 9.91954255104065
    },
    {
      "epoch": 2.648162841796875e-05,
      "model_forward_time": 3.1384401321411133,
      "step": 6942
    },
    {
      "epoch": 2.648162841796875e-05,
      "step": 6942,
      "training_step_time": 10.319463968276978
    },
    {
      "epoch": 2.6485443115234374e-05,
      "model_forward_time": 3.4190149307250977,
      "step": 6943
    },
    {
      "epoch": 2.6485443115234374e-05,
      "step": 6943,
      "training_step_time": 10.191097497940063
    },
    {
      "epoch": 2.6485443115234374e-05,
      "model_forward_time": 3.1362721920013428,
      "step": 6943
    },
    {
      "epoch": 2.6485443115234374e-05,
      "step": 6943,
      "training_step_time": 9.908634185791016
    },
    {
      "epoch": 2.6485443115234374e-05,
      "model_forward_time": 3.1352362632751465,
      "step": 6943
    },
    {
      "epoch": 2.6485443115234374e-05,
      "step": 6943,
      "training_step_time": 9.90605115890503
    },
    {
      "epoch": 2.6485443115234374e-05,
      "model_forward_time": 3.141375780105591,
      "step": 6943
    },
    {
      "epoch": 2.6485443115234374e-05,
      "step": 6943,
      "training_step_time": 10.292641401290894
    },
    {
      "epoch": 2.64892578125e-05,
      "model_forward_time": 3.1741092205047607,
      "step": 6944
    },
    {
      "epoch": 2.64892578125e-05,
      "step": 6944,
      "training_step_time": 9.953478336334229
    },
    {
      "epoch": 2.64892578125e-05,
      "model_forward_time": 3.4238383769989014,
      "step": 6944
    },
    {
      "epoch": 2.64892578125e-05,
      "step": 6944,
      "training_step_time": 10.203349590301514
    },
    {
      "epoch": 2.64892578125e-05,
      "model_forward_time": 3.151304244995117,
      "step": 6944
    },
    {
      "epoch": 2.64892578125e-05,
      "step": 6944,
      "training_step_time": 9.931752443313599
    },
    {
      "epoch": 2.64892578125e-05,
      "model_forward_time": 3.1436731815338135,
      "step": 6944
    },
    {
      "epoch": 2.64892578125e-05,
      "step": 6944,
      "training_step_time": 10.274035453796387
    },
    {
      "epoch": 2.6493072509765626e-05,
      "model_forward_time": 3.149895191192627,
      "step": 6945
    },
    {
      "dynamics_loss_avg": 0.05604271125048399,
      "epoch": 2.6493072509765626e-05,
      "step": 6945
    },
    {
      "action_loss_avg": 0.016170579008758068,
      "epoch": 2.6493072509765626e-05,
      "step": 6945
    },
    {
      "epoch": 2.6493072509765626e-05,
      "step": 6945,
      "training_step_time": 9.932271242141724
    },
    {
      "epoch": 2.6493072509765626e-05,
      "model_forward_time": 3.139069080352783,
      "step": 6945
    },
    {
      "epoch": 2.6493072509765626e-05,
      "step": 6945,
      "training_step_time": 9.921354055404663
    },
    {
      "epoch": 2.6493072509765626e-05,
      "model_forward_time": 3.1450488567352295,
      "step": 6945
    },
    {
      "epoch": 2.6493072509765626e-05,
      "step": 6945,
      "training_step_time": 10.211873054504395
    },
    {
      "epoch": 2.6493072509765626e-05,
      "model_forward_time": 3.1492815017700195,
      "step": 6945
    },
    {
      "epoch": 2.6493072509765626e-05,
      "step": 6945,
      "training_step_time": 10.361949443817139
    },
    {
      "epoch": 2.649688720703125e-05,
      "model_forward_time": 3.15425705909729,
      "step": 6946
    },
    {
      "epoch": 2.649688720703125e-05,
      "step": 6946,
      "training_step_time": 9.93491005897522
    },
    {
      "epoch": 2.649688720703125e-05,
      "model_forward_time": 3.1405937671661377,
      "step": 6946
    },
    {
      "epoch": 2.649688720703125e-05,
      "step": 6946,
      "training_step_time": 9.920382261276245
    },
    {
      "epoch": 2.649688720703125e-05,
      "model_forward_time": 3.1456375122070312,
      "step": 6946
    },
    {
      "epoch": 2.649688720703125e-05,
      "step": 6946,
      "training_step_time": 9.922186136245728
    },
    {
      "epoch": 2.649688720703125e-05,
      "model_forward_time": 3.414757251739502,
      "step": 6946
    },
    {
      "epoch": 2.649688720703125e-05,
      "step": 6946,
      "training_step_time": 10.28446340560913
    },
    {
      "epoch": 2.6500701904296875e-05,
      "model_forward_time": 3.1601386070251465,
      "step": 6947
    },
    {
      "epoch": 2.6500701904296875e-05,
      "step": 6947,
      "training_step_time": 9.934345006942749
    },
    {
      "epoch": 2.6500701904296875e-05,
      "model_forward_time": 3.1321215629577637,
      "step": 6947
    },
    {
      "epoch": 2.6500701904296875e-05,
      "step": 6947,
      "training_step_time": 9.90379810333252
    },
    {
      "epoch": 2.6500701904296875e-05,
      "model_forward_time": 3.134662628173828,
      "step": 6947
    },
    {
      "dynamics_loss_avg": 0.053647108376026154,
      "epoch": 2.6500701904296875e-05,
      "step": 6947
    },
    {
      "action_loss_avg": 0.023871381161734463,
      "epoch": 2.6500701904296875e-05,
      "step": 6947
    },
    {
      "epoch": 2.6500701904296875e-05,
      "step": 6947,
      "training_step_time": 9.905096769332886
    },
    {
      "epoch": 2.6500701904296875e-05,
      "model_forward_time": 3.129002332687378,
      "step": 6947
    },
    {
      "epoch": 2.6500701904296875e-05,
      "step": 6947,
      "training_step_time": 10.227779388427734
    },
    {
      "epoch": 2.65045166015625e-05,
      "model_forward_time": 3.134882926940918,
      "step": 6948
    },
    {
      "epoch": 2.65045166015625e-05,
      "step": 6948,
      "training_step_time": 10.18012809753418
    },
    {
      "epoch": 2.65045166015625e-05,
      "model_forward_time": 3.137683868408203,
      "step": 6948
    },
    {
      "epoch": 2.65045166015625e-05,
      "step": 6948,
      "training_step_time": 9.909950017929077
    },
    {
      "epoch": 2.65045166015625e-05,
      "model_forward_time": 3.1423208713531494,
      "step": 6948
    },
    {
      "epoch": 2.65045166015625e-05,
      "step": 6948,
      "training_step_time": 9.909352540969849
    },
    {
      "epoch": 2.65045166015625e-05,
      "model_forward_time": 3.135282278060913,
      "step": 6948
    },
    {
      "epoch": 2.65045166015625e-05,
      "step": 6948,
      "training_step_time": 10.27431035041809
    },
    {
      "epoch": 2.6508331298828124e-05,
      "model_forward_time": 3.1717097759246826,
      "step": 6949
    },
    {
      "epoch": 2.6508331298828124e-05,
      "step": 6949,
      "training_step_time": 9.943217277526855
    },
    {
      "epoch": 2.6508331298828124e-05,
      "model_forward_time": 3.394129514694214,
      "step": 6949
    },
    {
      "epoch": 2.6508331298828124e-05,
      "step": 6949,
      "training_step_time": 10.160066843032837
    },
    {
      "epoch": 2.6508331298828124e-05,
      "model_forward_time": 3.1416208744049072,
      "step": 6949
    },
    {
      "epoch": 2.6508331298828124e-05,
      "step": 6949,
      "training_step_time": 9.907689809799194
    },
    {
      "epoch": 2.6508331298828124e-05,
      "model_forward_time": 3.142119884490967,
      "step": 6949
    },
    {
      "epoch": 2.6508331298828124e-05,
      "step": 6949,
      "training_step_time": 10.276505470275879
    },
    {
      "epoch": 2.651214599609375e-05,
      "grad_norm": 0.3297647535800934,
      "learning_rate": 1e-05,
      "loss": 0.0828,
      "step": 6950
    },
    {
      "epoch": 2.651214599609375e-05,
      "model_forward_time": 3.184027671813965,
      "step": 6950
    },
    {
      "dynamics_loss_avg": 0.06899151820689439,
      "epoch": 2.651214599609375e-05,
      "step": 6950
    },
    {
      "action_loss_avg": 0.0249935339204967,
      "epoch": 2.651214599609375e-05,
      "step": 6950
    },
    {
      "epoch": 2.651214599609375e-05,
      "step": 6950,
      "training_step_time": 9.95468282699585
    },
    {
      "epoch": 2.651214599609375e-05,
      "model_forward_time": 3.1397759914398193,
      "step": 6950
    },
    {
      "epoch": 2.651214599609375e-05,
      "step": 6950,
      "training_step_time": 9.9092538356781
    },
    {
      "epoch": 2.651214599609375e-05,
      "model_forward_time": 3.4262802600860596,
      "step": 6950
    },
    {
      "epoch": 2.651214599609375e-05,
      "step": 6950,
      "training_step_time": 10.197129487991333
    },
    {
      "epoch": 2.651214599609375e-05,
      "model_forward_time": 3.1411638259887695,
      "step": 6950
    },
    {
      "epoch": 2.651214599609375e-05,
      "step": 6950,
      "training_step_time": 10.295845031738281
    },
    {
      "epoch": 2.6515960693359374e-05,
      "model_forward_time": 3.149657964706421,
      "step": 6951
    },
    {
      "epoch": 2.6515960693359374e-05,
      "step": 6951,
      "training_step_time": 9.923423290252686
    },
    {
      "epoch": 2.6515960693359374e-05,
      "model_forward_time": 3.147542715072632,
      "step": 6951
    },
    {
      "epoch": 2.6515960693359374e-05,
      "step": 6951,
      "training_step_time": 9.918527126312256
    },
    {
      "epoch": 2.6515960693359374e-05,
      "model_forward_time": 3.1424379348754883,
      "step": 6951
    },
    {
      "epoch": 2.6515960693359374e-05,
      "step": 6951,
      "training_step_time": 9.914858341217041
    },
    {
      "epoch": 2.6515960693359374e-05,
      "model_forward_time": 3.4050772190093994,
      "step": 6951
    },
    {
      "epoch": 2.6515960693359374e-05,
      "step": 6951,
      "training_step_time": 10.268948793411255
    },
    {
      "epoch": 2.6519775390625e-05,
      "model_forward_time": 3.1642262935638428,
      "step": 6952
    },
    {
      "epoch": 2.6519775390625e-05,
      "step": 6952,
      "training_step_time": 9.94363522529602
    },
    {
      "epoch": 2.6519775390625e-05,
      "model_forward_time": 3.148552417755127,
      "step": 6952
    },
    {
      "epoch": 2.6519775390625e-05,
      "step": 6952,
      "training_step_time": 9.919633626937866
    },
    {
      "epoch": 2.6519775390625e-05,
      "model_forward_time": 3.1441729068756104,
      "step": 6952
    },
    {
      "dynamics_loss_avg": 0.04580745678395033,
      "epoch": 2.6519775390625e-05,
      "step": 6952
    },
    {
      "action_loss_avg": 0.01803316380828619,
      "epoch": 2.6519775390625e-05,
      "step": 6952
    },
    {
      "epoch": 2.6519775390625e-05,
      "step": 6952,
      "training_step_time": 9.921037673950195
    },
    {
      "epoch": 2.6519775390625e-05,
      "model_forward_time": 3.1487109661102295,
      "step": 6952
    },
    {
      "epoch": 2.6519775390625e-05,
      "step": 6952,
      "training_step_time": 10.262083292007446
    },
    {
      "epoch": 2.6523590087890626e-05,
      "model_forward_time": 3.425837755203247,
      "step": 6953
    },
    {
      "epoch": 2.6523590087890626e-05,
      "step": 6953,
      "training_step_time": 10.197226762771606
    },
    {
      "epoch": 2.6523590087890626e-05,
      "model_forward_time": 3.13893985748291,
      "step": 6953
    },
    {
      "epoch": 2.6523590087890626e-05,
      "step": 6953,
      "training_step_time": 9.9099760055542
    },
    {
      "epoch": 2.6523590087890626e-05,
      "model_forward_time": 3.1360950469970703,
      "step": 6953
    },
    {
      "epoch": 2.6523590087890626e-05,
      "step": 6953,
      "training_step_time": 9.911453485488892
    },
    {
      "epoch": 2.6523590087890626e-05,
      "model_forward_time": 3.140547275543213,
      "step": 6953
    },
    {
      "epoch": 2.6523590087890626e-05,
      "step": 6953,
      "training_step_time": 10.266708612442017
    },
    {
      "epoch": 2.652740478515625e-05,
      "model_forward_time": 3.204850912094116,
      "step": 6954
    },
    {
      "epoch": 2.652740478515625e-05,
      "step": 6954,
      "training_step_time": 9.98513126373291
    },
    {
      "epoch": 2.652740478515625e-05,
      "model_forward_time": 3.145887851715088,
      "step": 6954
    },
    {
      "epoch": 2.652740478515625e-05,
      "step": 6954,
      "training_step_time": 10.224738359451294
    },
    {
      "epoch": 2.652740478515625e-05,
      "model_forward_time": 3.147374391555786,
      "step": 6954
    },
    {
      "epoch": 2.652740478515625e-05,
      "step": 6954,
      "training_step_time": 9.923234224319458
    },
    {
      "epoch": 2.652740478515625e-05,
      "model_forward_time": 3.147883415222168,
      "step": 6954
    },
    {
      "epoch": 2.652740478515625e-05,
      "step": 6954,
      "training_step_time": 10.043365478515625
    },
    {
      "epoch": 2.6531219482421875e-05,
      "model_forward_time": 3.161257743835449,
      "step": 6955
    },
    {
      "dynamics_loss_avg": 0.06649166401475667,
      "epoch": 2.6531219482421875e-05,
      "step": 6955
    },
    {
      "action_loss_avg": 0.0270217580255121,
      "epoch": 2.6531219482421875e-05,
      "step": 6955
    },
    {
      "epoch": 2.6531219482421875e-05,
      "step": 6955,
      "training_step_time": 9.939401388168335
    },
    {
      "epoch": 2.6531219482421875e-05,
      "model_forward_time": 3.1364150047302246,
      "step": 6955
    },
    {
      "epoch": 2.6531219482421875e-05,
      "step": 6955,
      "training_step_time": 9.908184051513672
    },
    {
      "epoch": 2.6531219482421875e-05,
      "model_forward_time": 3.143415927886963,
      "step": 6955
    },
    {
      "epoch": 2.6531219482421875e-05,
      "step": 6955,
      "training_step_time": 10.188535928726196
    },
    {
      "epoch": 2.6531219482421875e-05,
      "model_forward_time": 3.1475844383239746,
      "step": 6955
    },
    {
      "epoch": 2.6531219482421875e-05,
      "step": 6955,
      "training_step_time": 10.301417112350464
    },
    {
      "epoch": 2.65350341796875e-05,
      "model_forward_time": 3.1525890827178955,
      "step": 6956
    },
    {
      "epoch": 2.65350341796875e-05,
      "step": 6956,
      "training_step_time": 9.922165393829346
    },
    {
      "epoch": 2.65350341796875e-05,
      "model_forward_time": 3.1390886306762695,
      "step": 6956
    },
    {
      "epoch": 2.65350341796875e-05,
      "step": 6956,
      "training_step_time": 9.907305240631104
    },
    {
      "epoch": 2.65350341796875e-05,
      "model_forward_time": 3.13974666595459,
      "step": 6956
    },
    {
      "epoch": 2.65350341796875e-05,
      "step": 6956,
      "training_step_time": 9.903995513916016
    },
    {
      "epoch": 2.65350341796875e-05,
      "model_forward_time": 3.1385295391082764,
      "step": 6956
    },
    {
      "epoch": 2.65350341796875e-05,
      "step": 6956,
      "training_step_time": 10.285782098770142
    },
    {
      "epoch": 2.6538848876953124e-05,
      "model_forward_time": 3.1558029651641846,
      "step": 6957
    },
    {
      "epoch": 2.6538848876953124e-05,
      "step": 6957,
      "training_step_time": 9.925452470779419
    },
    {
      "epoch": 2.6538848876953124e-05,
      "model_forward_time": 3.1356184482574463,
      "step": 6957
    },
    {
      "epoch": 2.6538848876953124e-05,
      "step": 6957,
      "training_step_time": 9.899243354797363
    },
    {
      "epoch": 2.6538848876953124e-05,
      "model_forward_time": 3.1425695419311523,
      "step": 6957
    },
    {
      "dynamics_loss_avg": 0.04383820816874504,
      "epoch": 2.6538848876953124e-05,
      "step": 6957
    },
    {
      "action_loss_avg": 0.015367998648434878,
      "epoch": 2.6538848876953124e-05,
      "step": 6957
    },
    {
      "epoch": 2.6538848876953124e-05,
      "step": 6957,
      "training_step_time": 9.908002614974976
    },
    {
      "epoch": 2.6538848876953124e-05,
      "model_forward_time": 3.1368019580841064,
      "step": 6957
    },
    {
      "epoch": 2.6538848876953124e-05,
      "step": 6957,
      "training_step_time": 10.271981716156006
    },
    {
      "epoch": 2.654266357421875e-05,
      "model_forward_time": 3.443768262863159,
      "step": 6958
    },
    {
      "epoch": 2.654266357421875e-05,
      "step": 6958,
      "training_step_time": 10.212725639343262
    },
    {
      "epoch": 2.654266357421875e-05,
      "model_forward_time": 3.1369576454162598,
      "step": 6958
    },
    {
      "epoch": 2.654266357421875e-05,
      "step": 6958,
      "training_step_time": 9.902351379394531
    },
    {
      "epoch": 2.654266357421875e-05,
      "model_forward_time": 3.1422557830810547,
      "step": 6958
    },
    {
      "epoch": 2.654266357421875e-05,
      "step": 6958,
      "training_step_time": 9.908839702606201
    },
    {
      "epoch": 2.654266357421875e-05,
      "model_forward_time": 3.1403050422668457,
      "step": 6958
    },
    {
      "epoch": 2.654266357421875e-05,
      "step": 6958,
      "training_step_time": 10.270118951797485
    },
    {
      "epoch": 2.6546478271484374e-05,
      "model_forward_time": 3.172738790512085,
      "step": 6959
    },
    {
      "epoch": 2.6546478271484374e-05,
      "step": 6959,
      "training_step_time": 9.936431646347046
    },
    {
      "epoch": 2.6546478271484374e-05,
      "model_forward_time": 3.1412816047668457,
      "step": 6959
    },
    {
      "epoch": 2.6546478271484374e-05,
      "step": 6959,
      "training_step_time": 10.173366069793701
    },
    {
      "epoch": 2.6546478271484374e-05,
      "model_forward_time": 3.140491008758545,
      "step": 6959
    },
    {
      "epoch": 2.6546478271484374e-05,
      "step": 6959,
      "training_step_time": 9.90638542175293
    },
    {
      "epoch": 2.6546478271484374e-05,
      "model_forward_time": 3.1383183002471924,
      "step": 6959
    },
    {
      "epoch": 2.6546478271484374e-05,
      "step": 6959,
      "training_step_time": 10.262198686599731
    },
    {
      "epoch": 2.655029296875e-05,
      "grad_norm": 0.6088270545005798,
      "learning_rate": 1e-05,
      "loss": 0.081,
      "step": 6960
    },
    {
      "epoch": 2.655029296875e-05,
      "model_forward_time": 3.1554877758026123,
      "step": 6960
    },
    {
      "dynamics_loss_avg": 0.07059205174446107,
      "epoch": 2.655029296875e-05,
      "step": 6960
    },
    {
      "action_loss_avg": 0.02361172642558813,
      "epoch": 2.655029296875e-05,
      "step": 6960
    },
    {
      "epoch": 2.655029296875e-05,
      "step": 6960,
      "training_step_time": 9.928448677062988
    },
    {
      "epoch": 2.655029296875e-05,
      "model_forward_time": 3.14146089553833,
      "step": 6960
    },
    {
      "epoch": 2.655029296875e-05,
      "step": 6960,
      "training_step_time": 9.910415410995483
    },
    {
      "epoch": 2.655029296875e-05,
      "model_forward_time": 3.4350154399871826,
      "step": 6960
    },
    {
      "epoch": 2.655029296875e-05,
      "step": 6960,
      "training_step_time": 10.201876163482666
    },
    {
      "epoch": 2.655029296875e-05,
      "model_forward_time": 3.1388659477233887,
      "step": 6960
    },
    {
      "epoch": 2.655029296875e-05,
      "step": 6960,
      "training_step_time": 10.024070501327515
    },
    {
      "epoch": 2.6554107666015626e-05,
      "model_forward_time": 3.147813081741333,
      "step": 6961
    },
    {
      "epoch": 2.6554107666015626e-05,
      "step": 6961,
      "training_step_time": 9.92041563987732
    },
    {
      "epoch": 2.6554107666015626e-05,
      "model_forward_time": 3.1392557621002197,
      "step": 6961
    },
    {
      "epoch": 2.6554107666015626e-05,
      "step": 6961,
      "training_step_time": 9.909623146057129
    },
    {
      "epoch": 2.6554107666015626e-05,
      "model_forward_time": 3.1304337978363037,
      "step": 6961
    },
    {
      "epoch": 2.6554107666015626e-05,
      "step": 6961,
      "training_step_time": 9.905449628829956
    },
    {
      "epoch": 2.6554107666015626e-05,
      "model_forward_time": 3.392488479614258,
      "step": 6961
    },
    {
      "epoch": 2.6554107666015626e-05,
      "step": 6961,
      "training_step_time": 10.291476488113403
    },
    {
      "epoch": 2.655792236328125e-05,
      "model_forward_time": 3.189040422439575,
      "step": 6962
    },
    {
      "epoch": 2.655792236328125e-05,
      "step": 6962,
      "training_step_time": 9.961050748825073
    },
    {
      "epoch": 2.655792236328125e-05,
      "model_forward_time": 3.141996383666992,
      "step": 6962
    },
    {
      "epoch": 2.655792236328125e-05,
      "step": 6962,
      "training_step_time": 9.912478685379028
    },
    {
      "epoch": 2.655792236328125e-05,
      "model_forward_time": 3.142317056655884,
      "step": 6962
    },
    {
      "dynamics_loss_avg": 0.07325145546346903,
      "epoch": 2.655792236328125e-05,
      "step": 6962
    },
    {
      "action_loss_avg": 0.023777720518410206,
      "epoch": 2.655792236328125e-05,
      "step": 6962
    },
    {
      "epoch": 2.655792236328125e-05,
      "step": 6962,
      "training_step_time": 9.916598320007324
    },
    {
      "epoch": 2.655792236328125e-05,
      "model_forward_time": 3.1419405937194824,
      "step": 6962
    },
    {
      "epoch": 2.655792236328125e-05,
      "step": 6962,
      "training_step_time": 10.245092391967773
    },
    {
      "epoch": 2.6561737060546875e-05,
      "model_forward_time": 3.4172003269195557,
      "step": 6963
    },
    {
      "epoch": 2.6561737060546875e-05,
      "step": 6963,
      "training_step_time": 10.192248582839966
    },
    {
      "epoch": 2.6561737060546875e-05,
      "model_forward_time": 3.1428942680358887,
      "step": 6963
    },
    {
      "epoch": 2.6561737060546875e-05,
      "step": 6963,
      "training_step_time": 9.912748575210571
    },
    {
      "epoch": 2.6561737060546875e-05,
      "model_forward_time": 2.4358808994293213,
      "step": 6963
    },
    {
      "epoch": 2.6561737060546875e-05,
      "step": 6963,
      "training_step_time": 7.4625091552734375
    },
    {
      "epoch": 2.6561737060546875e-05,
      "model_forward_time": 3.143847942352295,
      "step": 6963
    },
    {
      "epoch": 2.6561737060546875e-05,
      "step": 6963,
      "training_step_time": 10.327786445617676
    },
    {
      "epoch": 2.65655517578125e-05,
      "model_forward_time": 3.154987335205078,
      "step": 6964
    },
    {
      "epoch": 2.65655517578125e-05,
      "step": 6964,
      "training_step_time": 9.925750732421875
    },
    {
      "epoch": 2.65655517578125e-05,
      "model_forward_time": 3.419671058654785,
      "step": 6964
    },
    {
      "epoch": 2.65655517578125e-05,
      "step": 6964,
      "training_step_time": 10.188079833984375
    },
    {
      "epoch": 2.65655517578125e-05,
      "model_forward_time": 3.141483783721924,
      "step": 6964
    },
    {
      "epoch": 2.65655517578125e-05,
      "step": 6964,
      "training_step_time": 9.90191388130188
    },
    {
      "epoch": 2.65655517578125e-05,
      "model_forward_time": 3.143371343612671,
      "step": 6964
    },
    {
      "epoch": 2.65655517578125e-05,
      "step": 6964,
      "training_step_time": 10.27032208442688
    },
    {
      "epoch": 2.6569366455078124e-05,
      "model_forward_time": 3.1510391235351562,
      "step": 6965
    },
    {
      "dynamics_loss_avg": 0.05709430631250143,
      "epoch": 2.6569366455078124e-05,
      "step": 6965
    },
    {
      "action_loss_avg": 0.02492043897509575,
      "epoch": 2.6569366455078124e-05,
      "step": 6965
    },
    {
      "epoch": 2.6569366455078124e-05,
      "step": 6965,
      "training_step_time": 9.92418098449707
    },
    {
      "epoch": 2.6569366455078124e-05,
      "model_forward_time": 3.138251304626465,
      "step": 6965
    },
    {
      "epoch": 2.6569366455078124e-05,
      "step": 6965,
      "training_step_time": 9.907142639160156
    },
    {
      "epoch": 2.6569366455078124e-05,
      "model_forward_time": 3.13961124420166,
      "step": 6965
    },
    {
      "epoch": 2.6569366455078124e-05,
      "step": 6965,
      "training_step_time": 10.189007997512817
    },
    {
      "epoch": 2.6569366455078124e-05,
      "model_forward_time": 3.139413356781006,
      "step": 6965
    },
    {
      "epoch": 2.6569366455078124e-05,
      "step": 6965,
      "training_step_time": 10.011330127716064
    },
    {
      "epoch": 2.657318115234375e-05,
      "model_forward_time": 3.1800522804260254,
      "step": 6966
    },
    {
      "epoch": 2.657318115234375e-05,
      "step": 6966,
      "training_step_time": 9.94884467124939
    },
    {
      "epoch": 2.657318115234375e-05,
      "model_forward_time": 3.138270854949951,
      "step": 6966
    },
    {
      "epoch": 2.657318115234375e-05,
      "step": 6966,
      "training_step_time": 9.90757966041565
    },
    {
      "epoch": 2.657318115234375e-05,
      "model_forward_time": 3.152635097503662,
      "step": 6966
    },
    {
      "epoch": 2.657318115234375e-05,
      "step": 6966,
      "training_step_time": 9.92676043510437
    },
    {
      "epoch": 2.657318115234375e-05,
      "model_forward_time": 3.423114776611328,
      "step": 6966
    },
    {
      "epoch": 2.657318115234375e-05,
      "step": 6966,
      "training_step_time": 10.297329902648926
    },
    {
      "epoch": 2.6576995849609374e-05,
      "model_forward_time": 3.1805100440979004,
      "step": 6967
    },
    {
      "epoch": 2.6576995849609374e-05,
      "step": 6967,
      "training_step_time": 9.954380750656128
    },
    {
      "epoch": 2.6576995849609374e-05,
      "model_forward_time": 3.1424179077148438,
      "step": 6967
    },
    {
      "epoch": 2.6576995849609374e-05,
      "step": 6967,
      "training_step_time": 9.91562819480896
    },
    {
      "epoch": 2.6576995849609374e-05,
      "model_forward_time": 3.1476123332977295,
      "step": 6967
    },
    {
      "dynamics_loss_avg": 0.059672006592154506,
      "epoch": 2.6576995849609374e-05,
      "step": 6967
    },
    {
      "action_loss_avg": 0.030487219523638486,
      "epoch": 2.6576995849609374e-05,
      "step": 6967
    },
    {
      "epoch": 2.6576995849609374e-05,
      "step": 6967,
      "training_step_time": 9.923817157745361
    },
    {
      "epoch": 2.6576995849609374e-05,
      "model_forward_time": 3.1435012817382812,
      "step": 6967
    },
    {
      "epoch": 2.6576995849609374e-05,
      "step": 6967,
      "training_step_time": 10.283372402191162
    },
    {
      "epoch": 2.6580810546875e-05,
      "model_forward_time": 3.1508536338806152,
      "step": 6968
    },
    {
      "epoch": 2.6580810546875e-05,
      "step": 6968,
      "training_step_time": 10.201334476470947
    },
    {
      "epoch": 2.6580810546875e-05,
      "model_forward_time": 3.1374731063842773,
      "step": 6968
    },
    {
      "epoch": 2.6580810546875e-05,
      "step": 6968,
      "training_step_time": 9.913527011871338
    },
    {
      "epoch": 2.6580810546875e-05,
      "model_forward_time": 3.1416220664978027,
      "step": 6968
    },
    {
      "epoch": 2.6580810546875e-05,
      "step": 6968,
      "training_step_time": 9.916332721710205
    },
    {
      "epoch": 2.6580810546875e-05,
      "model_forward_time": 3.1351983547210693,
      "step": 6968
    },
    {
      "epoch": 2.6580810546875e-05,
      "step": 6968,
      "training_step_time": 10.314087152481079
    },
    {
      "epoch": 2.6584625244140626e-05,
      "model_forward_time": 3.1528892517089844,
      "step": 6969
    },
    {
      "epoch": 2.6584625244140626e-05,
      "step": 6969,
      "training_step_time": 9.924243211746216
    },
    {
      "epoch": 2.6584625244140626e-05,
      "model_forward_time": 3.391176700592041,
      "step": 6969
    },
    {
      "epoch": 2.6584625244140626e-05,
      "step": 6969,
      "training_step_time": 10.160837888717651
    },
    {
      "epoch": 2.6584625244140626e-05,
      "model_forward_time": 3.1364591121673584,
      "step": 6969
    },
    {
      "epoch": 2.6584625244140626e-05,
      "step": 6969,
      "training_step_time": 9.906600952148438
    },
    {
      "epoch": 2.6584625244140626e-05,
      "model_forward_time": 3.1355459690093994,
      "step": 6969
    },
    {
      "epoch": 2.6584625244140626e-05,
      "step": 6969,
      "training_step_time": 10.279374599456787
    },
    {
      "epoch": 2.658843994140625e-05,
      "grad_norm": 0.4571607708930969,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 6970
    },
    {
      "epoch": 2.658843994140625e-05,
      "model_forward_time": 3.1747488975524902,
      "step": 6970
    },
    {
      "dynamics_loss_avg": 0.05353178912773728,
      "epoch": 2.658843994140625e-05,
      "step": 6970
    },
    {
      "action_loss_avg": 0.01747838072478771,
      "epoch": 2.658843994140625e-05,
      "step": 6970
    },
    {
      "epoch": 2.658843994140625e-05,
      "step": 6970,
      "training_step_time": 9.946918487548828
    },
    {
      "epoch": 2.658843994140625e-05,
      "model_forward_time": 3.13399338722229,
      "step": 6970
    },
    {
      "epoch": 2.658843994140625e-05,
      "step": 6970,
      "training_step_time": 9.899787902832031
    },
    {
      "epoch": 2.658843994140625e-05,
      "model_forward_time": 3.408162832260132,
      "step": 6970
    },
    {
      "epoch": 2.658843994140625e-05,
      "step": 6970,
      "training_step_time": 10.176634550094604
    },
    {
      "epoch": 2.658843994140625e-05,
      "model_forward_time": 3.1387109756469727,
      "step": 6970
    },
    {
      "epoch": 2.658843994140625e-05,
      "step": 6970,
      "training_step_time": 10.011974334716797
    },
    {
      "epoch": 2.6592254638671875e-05,
      "model_forward_time": 3.141920804977417,
      "step": 6971
    },
    {
      "epoch": 2.6592254638671875e-05,
      "step": 6971,
      "training_step_time": 9.912649393081665
    },
    {
      "epoch": 2.6592254638671875e-05,
      "model_forward_time": 2.4299564361572266,
      "step": 6971
    },
    {
      "epoch": 2.6592254638671875e-05,
      "step": 6971,
      "training_step_time": 7.457709312438965
    },
    {
      "epoch": 2.6592254638671875e-05,
      "model_forward_time": 3.131622552871704,
      "step": 6971
    },
    {
      "epoch": 2.6592254638671875e-05,
      "step": 6971,
      "training_step_time": 9.905394554138184
    },
    {
      "epoch": 2.6592254638671875e-05,
      "model_forward_time": 2.430866003036499,
      "step": 6971
    },
    {
      "epoch": 2.6592254638671875e-05,
      "step": 6971,
      "training_step_time": 10.034619808197021
    },
    {
      "epoch": 2.65960693359375e-05,
      "model_forward_time": 3.1465349197387695,
      "step": 6972
    },
    {
      "epoch": 2.65960693359375e-05,
      "step": 6972,
      "training_step_time": 9.925025939941406
    },
    {
      "epoch": 2.65960693359375e-05,
      "model_forward_time": 3.1407227516174316,
      "step": 6972
    },
    {
      "epoch": 2.65960693359375e-05,
      "step": 6972,
      "training_step_time": 9.914201736450195
    },
    {
      "epoch": 2.65960693359375e-05,
      "model_forward_time": 3.131324291229248,
      "step": 6972
    },
    {
      "dynamics_loss_avg": 0.056241314858198166,
      "epoch": 2.65960693359375e-05,
      "step": 6972
    },
    {
      "action_loss_avg": 0.02070811940357089,
      "epoch": 2.65960693359375e-05,
      "step": 6972
    },
    {
      "epoch": 2.65960693359375e-05,
      "step": 6972,
      "training_step_time": 9.905996561050415
    },
    {
      "epoch": 2.65960693359375e-05,
      "model_forward_time": 3.1377265453338623,
      "step": 6972
    },
    {
      "epoch": 2.65960693359375e-05,
      "step": 6972,
      "training_step_time": 10.299289464950562
    },
    {
      "epoch": 2.6599884033203124e-05,
      "model_forward_time": 3.401965379714966,
      "step": 6973
    },
    {
      "epoch": 2.6599884033203124e-05,
      "step": 6973,
      "training_step_time": 10.173395156860352
    },
    {
      "epoch": 2.6599884033203124e-05,
      "model_forward_time": 3.1327691078186035,
      "step": 6973
    },
    {
      "epoch": 2.6599884033203124e-05,
      "step": 6973,
      "training_step_time": 9.900955438613892
    },
    {
      "epoch": 2.6599884033203124e-05,
      "model_forward_time": 3.1368777751922607,
      "step": 6973
    },
    {
      "epoch": 2.6599884033203124e-05,
      "step": 6973,
      "training_step_time": 9.905840158462524
    },
    {
      "epoch": 2.6599884033203124e-05,
      "model_forward_time": 3.136644124984741,
      "step": 6973
    },
    {
      "epoch": 2.6599884033203124e-05,
      "step": 6973,
      "training_step_time": 10.263208866119385
    },
    {
      "epoch": 2.660369873046875e-05,
      "model_forward_time": 3.153480291366577,
      "step": 6974
    },
    {
      "epoch": 2.660369873046875e-05,
      "step": 6974,
      "training_step_time": 9.939832210540771
    },
    {
      "epoch": 2.660369873046875e-05,
      "model_forward_time": 3.1364824771881104,
      "step": 6974
    },
    {
      "epoch": 2.660369873046875e-05,
      "step": 6974,
      "training_step_time": 10.217091798782349
    },
    {
      "epoch": 2.660369873046875e-05,
      "model_forward_time": 3.1422314643859863,
      "step": 6974
    },
    {
      "epoch": 2.660369873046875e-05,
      "step": 6974,
      "training_step_time": 9.913584470748901
    },
    {
      "epoch": 2.660369873046875e-05,
      "model_forward_time": 3.1454854011535645,
      "step": 6974
    },
    {
      "epoch": 2.660369873046875e-05,
      "step": 6974,
      "training_step_time": 10.251360177993774
    },
    {
      "epoch": 2.6607513427734374e-05,
      "model_forward_time": 3.180182695388794,
      "step": 6975
    },
    {
      "dynamics_loss_avg": 0.052341495826840403,
      "epoch": 2.6607513427734374e-05,
      "step": 6975
    },
    {
      "action_loss_avg": 0.023849973827600478,
      "epoch": 2.6607513427734374e-05,
      "step": 6975
    },
    {
      "epoch": 2.6607513427734374e-05,
      "step": 6975,
      "training_step_time": 9.955317974090576
    },
    {
      "epoch": 2.6607513427734374e-05,
      "model_forward_time": 3.1360440254211426,
      "step": 6975
    },
    {
      "epoch": 2.6607513427734374e-05,
      "step": 6975,
      "training_step_time": 9.902518272399902
    },
    {
      "epoch": 2.6607513427734374e-05,
      "model_forward_time": 3.399496555328369,
      "step": 6975
    },
    {
      "epoch": 2.6607513427734374e-05,
      "step": 6975,
      "training_step_time": 10.16226601600647
    },
    {
      "epoch": 2.6607513427734374e-05,
      "model_forward_time": 3.14121675491333,
      "step": 6975
    },
    {
      "epoch": 2.6607513427734374e-05,
      "step": 6975,
      "training_step_time": 10.00463581085205
    },
    {
      "epoch": 2.6611328125e-05,
      "model_forward_time": 3.1416404247283936,
      "step": 6976
    },
    {
      "epoch": 2.6611328125e-05,
      "step": 6976,
      "training_step_time": 9.912178039550781
    },
    {
      "epoch": 2.6611328125e-05,
      "model_forward_time": 3.1369409561157227,
      "step": 6976
    },
    {
      "epoch": 2.6611328125e-05,
      "step": 6976,
      "training_step_time": 9.908013582229614
    },
    {
      "epoch": 2.6611328125e-05,
      "model_forward_time": 3.135016918182373,
      "step": 6976
    },
    {
      "epoch": 2.6611328125e-05,
      "step": 6976,
      "training_step_time": 9.897900819778442
    },
    {
      "epoch": 2.6611328125e-05,
      "model_forward_time": 3.131608724594116,
      "step": 6976
    },
    {
      "epoch": 2.6611328125e-05,
      "step": 6976,
      "training_step_time": 10.279546022415161
    },
    {
      "epoch": 2.6615142822265626e-05,
      "model_forward_time": 3.1734540462493896,
      "step": 6977
    },
    {
      "epoch": 2.6615142822265626e-05,
      "step": 6977,
      "training_step_time": 9.942415714263916
    },
    {
      "epoch": 2.6615142822265626e-05,
      "model_forward_time": 3.130499839782715,
      "step": 6977
    },
    {
      "epoch": 2.6615142822265626e-05,
      "step": 6977,
      "training_step_time": 9.90000581741333
    },
    {
      "epoch": 2.6615142822265626e-05,
      "model_forward_time": 3.136167049407959,
      "step": 6977
    },
    {
      "dynamics_loss_avg": 0.057300962135195734,
      "epoch": 2.6615142822265626e-05,
      "step": 6977
    },
    {
      "action_loss_avg": 0.02204320775344968,
      "epoch": 2.6615142822265626e-05,
      "step": 6977
    },
    {
      "epoch": 2.6615142822265626e-05,
      "step": 6977,
      "training_step_time": 9.910001993179321
    },
    {
      "epoch": 2.6615142822265626e-05,
      "model_forward_time": 3.1341493129730225,
      "step": 6977
    },
    {
      "epoch": 2.6615142822265626e-05,
      "step": 6977,
      "training_step_time": 10.315702199935913
    },
    {
      "epoch": 2.661895751953125e-05,
      "model_forward_time": 3.4227499961853027,
      "step": 6978
    },
    {
      "epoch": 2.661895751953125e-05,
      "step": 6978,
      "training_step_time": 10.194275140762329
    },
    {
      "epoch": 2.661895751953125e-05,
      "model_forward_time": 3.1322484016418457,
      "step": 6978
    },
    {
      "epoch": 2.661895751953125e-05,
      "step": 6978,
      "training_step_time": 9.903249263763428
    },
    {
      "epoch": 2.661895751953125e-05,
      "model_forward_time": 2.4308879375457764,
      "step": 6978
    },
    {
      "epoch": 2.661895751953125e-05,
      "step": 6978,
      "training_step_time": 7.460827350616455
    },
    {
      "epoch": 2.661895751953125e-05,
      "model_forward_time": 3.1284098625183105,
      "step": 6978
    },
    {
      "epoch": 2.661895751953125e-05,
      "step": 6978,
      "training_step_time": 10.282591581344604
    },
    {
      "epoch": 2.6622772216796875e-05,
      "model_forward_time": 3.1477882862091064,
      "step": 6979
    },
    {
      "epoch": 2.6622772216796875e-05,
      "step": 6979,
      "training_step_time": 9.915512323379517
    },
    {
      "epoch": 2.6622772216796875e-05,
      "model_forward_time": 3.130070447921753,
      "step": 6979
    },
    {
      "epoch": 2.6622772216796875e-05,
      "step": 6979,
      "training_step_time": 9.90047311782837
    },
    {
      "epoch": 2.6622772216796875e-05,
      "model_forward_time": 3.39689564704895,
      "step": 6979
    },
    {
      "epoch": 2.6622772216796875e-05,
      "step": 6979,
      "training_step_time": 10.165549755096436
    },
    {
      "epoch": 2.6622772216796875e-05,
      "model_forward_time": 3.1338565349578857,
      "step": 6979
    },
    {
      "epoch": 2.6622772216796875e-05,
      "step": 6979,
      "training_step_time": 10.285008192062378
    },
    {
      "epoch": 2.66265869140625e-05,
      "grad_norm": 0.5622042417526245,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 6980
    },
    {
      "epoch": 2.66265869140625e-05,
      "model_forward_time": 2.4742965698242188,
      "step": 6980
    },
    {
      "dynamics_loss_avg": 0.05133553761988878,
      "epoch": 2.66265869140625e-05,
      "step": 6980
    },
    {
      "action_loss_avg": 0.017507085599936544,
      "epoch": 2.66265869140625e-05,
      "step": 6980
    },
    {
      "epoch": 2.66265869140625e-05,
      "step": 6980,
      "training_step_time": 7.507267236709595
    },
    {
      "epoch": 2.66265869140625e-05,
      "model_forward_time": 3.132697343826294,
      "step": 6980
    },
    {
      "epoch": 2.66265869140625e-05,
      "step": 6980,
      "training_step_time": 9.900356769561768
    },
    {
      "epoch": 2.66265869140625e-05,
      "model_forward_time": 3.414566993713379,
      "step": 6980
    },
    {
      "epoch": 2.66265869140625e-05,
      "step": 6980,
      "training_step_time": 10.186031103134155
    },
    {
      "epoch": 2.66265869140625e-05,
      "model_forward_time": 3.1338348388671875,
      "step": 6980
    },
    {
      "epoch": 2.66265869140625e-05,
      "step": 6980,
      "training_step_time": 10.017055749893188
    },
    {
      "epoch": 2.6630401611328125e-05,
      "model_forward_time": 3.155409097671509,
      "step": 6981
    },
    {
      "epoch": 2.6630401611328125e-05,
      "step": 6981,
      "training_step_time": 9.93867039680481
    },
    {
      "epoch": 2.6630401611328125e-05,
      "model_forward_time": 3.138859987258911,
      "step": 6981
    },
    {
      "epoch": 2.6630401611328125e-05,
      "step": 6981,
      "training_step_time": 9.911326885223389
    },
    {
      "epoch": 2.6630401611328125e-05,
      "model_forward_time": 3.13777494430542,
      "step": 6981
    },
    {
      "epoch": 2.6630401611328125e-05,
      "step": 6981,
      "training_step_time": 9.911675930023193
    },
    {
      "epoch": 2.6630401611328125e-05,
      "model_forward_time": 3.1384119987487793,
      "step": 6981
    },
    {
      "epoch": 2.6630401611328125e-05,
      "step": 6981,
      "training_step_time": 10.283840656280518
    },
    {
      "epoch": 2.663421630859375e-05,
      "model_forward_time": 3.4388585090637207,
      "step": 6982
    },
    {
      "epoch": 2.663421630859375e-05,
      "step": 6982,
      "training_step_time": 10.212054252624512
    },
    {
      "epoch": 2.663421630859375e-05,
      "model_forward_time": 3.1426076889038086,
      "step": 6982
    },
    {
      "epoch": 2.663421630859375e-05,
      "step": 6982,
      "training_step_time": 9.913150787353516
    },
    {
      "epoch": 2.663421630859375e-05,
      "model_forward_time": 3.1348657608032227,
      "step": 6982
    },
    {
      "dynamics_loss_avg": 0.06023248936980963,
      "epoch": 2.663421630859375e-05,
      "step": 6982
    },
    {
      "action_loss_avg": 0.019859890267252923,
      "epoch": 2.663421630859375e-05,
      "step": 6982
    },
    {
      "epoch": 2.663421630859375e-05,
      "step": 6982,
      "training_step_time": 9.907115936279297
    },
    {
      "epoch": 2.663421630859375e-05,
      "model_forward_time": 3.1413378715515137,
      "step": 6982
    },
    {
      "epoch": 2.663421630859375e-05,
      "step": 6982,
      "training_step_time": 10.030583143234253
    },
    {
      "epoch": 2.6638031005859374e-05,
      "model_forward_time": 3.4860928058624268,
      "step": 6983
    },
    {
      "epoch": 2.6638031005859374e-05,
      "step": 6983,
      "training_step_time": 10.262053489685059
    },
    {
      "epoch": 2.6638031005859374e-05,
      "model_forward_time": 3.150230884552002,
      "step": 6983
    },
    {
      "epoch": 2.6638031005859374e-05,
      "step": 6983,
      "training_step_time": 9.92491888999939
    },
    {
      "epoch": 2.6638031005859374e-05,
      "model_forward_time": 3.1465582847595215,
      "step": 6983
    },
    {
      "epoch": 2.6638031005859374e-05,
      "step": 6983,
      "training_step_time": 9.918716430664062
    },
    {
      "epoch": 2.6638031005859374e-05,
      "model_forward_time": 3.146117925643921,
      "step": 6983
    },
    {
      "epoch": 2.6638031005859374e-05,
      "step": 6983,
      "training_step_time": 10.282454490661621
    },
    {
      "epoch": 2.6641845703125e-05,
      "model_forward_time": 3.1501424312591553,
      "step": 6984
    },
    {
      "epoch": 2.6641845703125e-05,
      "step": 6984,
      "training_step_time": 9.922048568725586
    },
    {
      "epoch": 2.6641845703125e-05,
      "model_forward_time": 3.138681411743164,
      "step": 6984
    },
    {
      "epoch": 2.6641845703125e-05,
      "step": 6984,
      "training_step_time": 10.178211212158203
    },
    {
      "epoch": 2.6641845703125e-05,
      "model_forward_time": 3.1387839317321777,
      "step": 6984
    },
    {
      "epoch": 2.6641845703125e-05,
      "step": 6984,
      "training_step_time": 9.901342630386353
    },
    {
      "epoch": 2.6641845703125e-05,
      "model_forward_time": 3.142657995223999,
      "step": 6984
    },
    {
      "epoch": 2.6641845703125e-05,
      "step": 6984,
      "training_step_time": 10.245097875595093
    },
    {
      "epoch": 2.6645660400390626e-05,
      "model_forward_time": 3.18581485748291,
      "step": 6985
    },
    {
      "dynamics_loss_avg": 0.06127799265086651,
      "epoch": 2.6645660400390626e-05,
      "step": 6985
    },
    {
      "action_loss_avg": 0.01829157308675349,
      "epoch": 2.6645660400390626e-05,
      "step": 6985
    },
    {
      "epoch": 2.6645660400390626e-05,
      "step": 6985,
      "training_step_time": 9.956923961639404
    },
    {
      "epoch": 2.6645660400390626e-05,
      "model_forward_time": 3.1361985206604004,
      "step": 6985
    },
    {
      "epoch": 2.6645660400390626e-05,
      "step": 6985,
      "training_step_time": 9.904792785644531
    },
    {
      "epoch": 2.6645660400390626e-05,
      "model_forward_time": 3.1429569721221924,
      "step": 6985
    },
    {
      "epoch": 2.6645660400390626e-05,
      "step": 6985,
      "training_step_time": 10.195823669433594
    },
    {
      "epoch": 2.6645660400390626e-05,
      "model_forward_time": 3.142793893814087,
      "step": 6985
    },
    {
      "epoch": 2.6645660400390626e-05,
      "step": 6985,
      "training_step_time": 10.02000379562378
    },
    {
      "epoch": 2.664947509765625e-05,
      "model_forward_time": 3.163597345352173,
      "step": 6986
    },
    {
      "epoch": 2.664947509765625e-05,
      "step": 6986,
      "training_step_time": 9.934294700622559
    },
    {
      "epoch": 2.664947509765625e-05,
      "model_forward_time": 3.1427714824676514,
      "step": 6986
    },
    {
      "epoch": 2.664947509765625e-05,
      "step": 6986,
      "training_step_time": 9.915236234664917
    },
    {
      "epoch": 2.664947509765625e-05,
      "model_forward_time": 3.1484618186950684,
      "step": 6986
    },
    {
      "epoch": 2.664947509765625e-05,
      "step": 6986,
      "training_step_time": 9.92235279083252
    },
    {
      "epoch": 2.664947509765625e-05,
      "model_forward_time": 3.429896354675293,
      "step": 6986
    },
    {
      "epoch": 2.664947509765625e-05,
      "step": 6986,
      "training_step_time": 10.297338247299194
    },
    {
      "epoch": 2.6653289794921875e-05,
      "model_forward_time": 3.1642394065856934,
      "step": 6987
    },
    {
      "epoch": 2.6653289794921875e-05,
      "step": 6987,
      "training_step_time": 9.943501710891724
    },
    {
      "epoch": 2.6653289794921875e-05,
      "model_forward_time": 3.144911050796509,
      "step": 6987
    },
    {
      "epoch": 2.6653289794921875e-05,
      "step": 6987,
      "training_step_time": 9.921818256378174
    },
    {
      "epoch": 2.6653289794921875e-05,
      "model_forward_time": 3.149327039718628,
      "step": 6987
    },
    {
      "dynamics_loss_avg": 0.05540689993649721,
      "epoch": 2.6653289794921875e-05,
      "step": 6987
    },
    {
      "action_loss_avg": 0.01947574382647872,
      "epoch": 2.6653289794921875e-05,
      "step": 6987
    },
    {
      "epoch": 2.6653289794921875e-05,
      "step": 6987,
      "training_step_time": 9.925799131393433
    },
    {
      "epoch": 2.6653289794921875e-05,
      "model_forward_time": 3.1458473205566406,
      "step": 6987
    },
    {
      "epoch": 2.6653289794921875e-05,
      "step": 6987,
      "training_step_time": 10.017597675323486
    },
    {
      "epoch": 2.66571044921875e-05,
      "model_forward_time": 3.178741931915283,
      "step": 6988
    },
    {
      "epoch": 2.66571044921875e-05,
      "step": 6988,
      "training_step_time": 10.22603440284729
    },
    {
      "epoch": 2.66571044921875e-05,
      "model_forward_time": 3.1354944705963135,
      "step": 6988
    },
    {
      "epoch": 2.66571044921875e-05,
      "step": 6988,
      "training_step_time": 9.911082983016968
    },
    {
      "epoch": 2.66571044921875e-05,
      "model_forward_time": 3.1409640312194824,
      "step": 6988
    },
    {
      "epoch": 2.66571044921875e-05,
      "step": 6988,
      "training_step_time": 9.917011022567749
    },
    {
      "epoch": 2.66571044921875e-05,
      "model_forward_time": 3.136810541152954,
      "step": 6988
    },
    {
      "epoch": 2.66571044921875e-05,
      "step": 6988,
      "training_step_time": 10.270238637924194
    },
    {
      "epoch": 2.6660919189453125e-05,
      "model_forward_time": 3.177110433578491,
      "step": 6989
    },
    {
      "epoch": 2.6660919189453125e-05,
      "step": 6989,
      "training_step_time": 9.955121278762817
    },
    {
      "epoch": 2.6660919189453125e-05,
      "model_forward_time": 3.3888778686523438,
      "step": 6989
    },
    {
      "epoch": 2.6660919189453125e-05,
      "step": 6989,
      "training_step_time": 10.15744137763977
    },
    {
      "epoch": 2.6660919189453125e-05,
      "model_forward_time": 3.1352546215057373,
      "step": 6989
    },
    {
      "epoch": 2.6660919189453125e-05,
      "step": 6989,
      "training_step_time": 9.903825283050537
    },
    {
      "epoch": 2.6660919189453125e-05,
      "model_forward_time": 3.1338207721710205,
      "step": 6989
    },
    {
      "epoch": 2.6660919189453125e-05,
      "step": 6989,
      "training_step_time": 10.261605262756348
    },
    {
      "epoch": 2.666473388671875e-05,
      "grad_norm": 0.3401319682598114,
      "learning_rate": 1e-05,
      "loss": 0.0783,
      "step": 6990
    },
    {
      "epoch": 2.666473388671875e-05,
      "model_forward_time": 3.1466104984283447,
      "step": 6990
    },
    {
      "dynamics_loss_avg": 0.07238128706812859,
      "epoch": 2.666473388671875e-05,
      "step": 6990
    },
    {
      "action_loss_avg": 0.02685169531032443,
      "epoch": 2.666473388671875e-05,
      "step": 6990
    },
    {
      "epoch": 2.666473388671875e-05,
      "step": 6990,
      "training_step_time": 9.921193599700928
    },
    {
      "epoch": 2.666473388671875e-05,
      "model_forward_time": 3.1363027095794678,
      "step": 6990
    },
    {
      "epoch": 2.666473388671875e-05,
      "step": 6990,
      "training_step_time": 9.907618522644043
    },
    {
      "epoch": 2.666473388671875e-05,
      "model_forward_time": 3.4228224754333496,
      "step": 6990
    },
    {
      "epoch": 2.666473388671875e-05,
      "step": 6990,
      "training_step_time": 10.190446615219116
    },
    {
      "epoch": 2.666473388671875e-05,
      "model_forward_time": 3.1366522312164307,
      "step": 6990
    },
    {
      "epoch": 2.666473388671875e-05,
      "step": 6990,
      "training_step_time": 10.295955419540405
    },
    {
      "epoch": 2.6668548583984374e-05,
      "model_forward_time": 3.1811814308166504,
      "step": 6991
    },
    {
      "epoch": 2.6668548583984374e-05,
      "step": 6991,
      "training_step_time": 9.954838991165161
    },
    {
      "epoch": 2.6668548583984374e-05,
      "model_forward_time": 3.1407229900360107,
      "step": 6991
    },
    {
      "epoch": 2.6668548583984374e-05,
      "step": 6991,
      "training_step_time": 9.914999723434448
    },
    {
      "epoch": 2.6668548583984374e-05,
      "model_forward_time": 3.142416477203369,
      "step": 6991
    },
    {
      "epoch": 2.6668548583984374e-05,
      "step": 6991,
      "training_step_time": 9.918965816497803
    },
    {
      "epoch": 2.6668548583984374e-05,
      "model_forward_time": 3.4267189502716064,
      "step": 6991
    },
    {
      "epoch": 2.6668548583984374e-05,
      "step": 6991,
      "training_step_time": 10.295595169067383
    },
    {
      "epoch": 2.667236328125e-05,
      "model_forward_time": 3.1532394886016846,
      "step": 6992
    },
    {
      "epoch": 2.667236328125e-05,
      "step": 6992,
      "training_step_time": 9.932284355163574
    },
    {
      "epoch": 2.667236328125e-05,
      "model_forward_time": 3.1412980556488037,
      "step": 6992
    },
    {
      "epoch": 2.667236328125e-05,
      "step": 6992,
      "training_step_time": 9.919521808624268
    },
    {
      "epoch": 2.667236328125e-05,
      "model_forward_time": 3.14656925201416,
      "step": 6992
    },
    {
      "dynamics_loss_avg": 0.05155752189457417,
      "epoch": 2.667236328125e-05,
      "step": 6992
    },
    {
      "action_loss_avg": 0.016473282827064395,
      "epoch": 2.667236328125e-05,
      "step": 6992
    },
    {
      "epoch": 2.667236328125e-05,
      "step": 6992,
      "training_step_time": 9.92830204963684
    },
    {
      "epoch": 2.667236328125e-05,
      "model_forward_time": 3.1393187046051025,
      "step": 6992
    },
    {
      "epoch": 2.667236328125e-05,
      "step": 6992,
      "training_step_time": 10.021215438842773
    },
    {
      "epoch": 2.6676177978515626e-05,
      "model_forward_time": 3.435300827026367,
      "step": 6993
    },
    {
      "epoch": 2.6676177978515626e-05,
      "step": 6993,
      "training_step_time": 10.230677127838135
    },
    {
      "epoch": 2.6676177978515626e-05,
      "model_forward_time": 3.1385340690612793,
      "step": 6993
    },
    {
      "epoch": 2.6676177978515626e-05,
      "step": 6993,
      "training_step_time": 9.922998905181885
    },
    {
      "epoch": 2.6676177978515626e-05,
      "model_forward_time": 3.1452796459198,
      "step": 6993
    },
    {
      "epoch": 2.6676177978515626e-05,
      "step": 6993,
      "training_step_time": 9.929286241531372
    },
    {
      "epoch": 2.6676177978515626e-05,
      "model_forward_time": 3.1414690017700195,
      "step": 6993
    },
    {
      "epoch": 2.6676177978515626e-05,
      "step": 6993,
      "training_step_time": 10.275589227676392
    },
    {
      "epoch": 2.667999267578125e-05,
      "model_forward_time": 3.149479627609253,
      "step": 6994
    },
    {
      "epoch": 2.667999267578125e-05,
      "step": 6994,
      "training_step_time": 9.93026876449585
    },
    {
      "epoch": 2.667999267578125e-05,
      "model_forward_time": 3.417015314102173,
      "step": 6994
    },
    {
      "epoch": 2.667999267578125e-05,
      "step": 6994,
      "training_step_time": 10.194287538528442
    },
    {
      "epoch": 2.667999267578125e-05,
      "model_forward_time": 3.1385254859924316,
      "step": 6994
    },
    {
      "epoch": 2.667999267578125e-05,
      "step": 6994,
      "training_step_time": 9.916034698486328
    },
    {
      "epoch": 2.667999267578125e-05,
      "model_forward_time": 3.135314702987671,
      "step": 6994
    },
    {
      "epoch": 2.667999267578125e-05,
      "step": 6994,
      "training_step_time": 10.284911870956421
    },
    {
      "epoch": 2.6683807373046875e-05,
      "model_forward_time": 3.1597530841827393,
      "step": 6995
    },
    {
      "dynamics_loss_avg": 0.07297636885195971,
      "epoch": 2.6683807373046875e-05,
      "step": 6995
    },
    {
      "action_loss_avg": 0.028486208990216255,
      "epoch": 2.6683807373046875e-05,
      "step": 6995
    },
    {
      "epoch": 2.6683807373046875e-05,
      "step": 6995,
      "training_step_time": 9.938647747039795
    },
    {
      "epoch": 2.6683807373046875e-05,
      "model_forward_time": 3.1379027366638184,
      "step": 6995
    },
    {
      "epoch": 2.6683807373046875e-05,
      "step": 6995,
      "training_step_time": 9.915617227554321
    },
    {
      "epoch": 2.6683807373046875e-05,
      "model_forward_time": 3.4180374145507812,
      "step": 6995
    },
    {
      "epoch": 2.6683807373046875e-05,
      "step": 6995,
      "training_step_time": 10.18993854522705
    },
    {
      "epoch": 2.6683807373046875e-05,
      "model_forward_time": 3.139392852783203,
      "step": 6995
    },
    {
      "epoch": 2.6683807373046875e-05,
      "step": 6995,
      "training_step_time": 10.33154845237732
    },
    {
      "epoch": 2.66876220703125e-05,
      "model_forward_time": 3.1481099128723145,
      "step": 6996
    },
    {
      "epoch": 2.66876220703125e-05,
      "step": 6996,
      "training_step_time": 9.931122541427612
    },
    {
      "epoch": 2.66876220703125e-05,
      "model_forward_time": 3.139589548110962,
      "step": 6996
    },
    {
      "epoch": 2.66876220703125e-05,
      "step": 6996,
      "training_step_time": 9.918055772781372
    },
    {
      "epoch": 2.66876220703125e-05,
      "model_forward_time": 3.1398634910583496,
      "step": 6996
    },
    {
      "epoch": 2.66876220703125e-05,
      "step": 6996,
      "training_step_time": 9.922932624816895
    },
    {
      "epoch": 2.66876220703125e-05,
      "model_forward_time": 3.4158058166503906,
      "step": 6996
    },
    {
      "epoch": 2.66876220703125e-05,
      "step": 6996,
      "training_step_time": 10.288309812545776
    },
    {
      "epoch": 2.6691436767578125e-05,
      "model_forward_time": 3.157388925552368,
      "step": 6997
    },
    {
      "epoch": 2.6691436767578125e-05,
      "step": 6997,
      "training_step_time": 9.933264017105103
    },
    {
      "epoch": 2.6691436767578125e-05,
      "model_forward_time": 3.143064498901367,
      "step": 6997
    },
    {
      "epoch": 2.6691436767578125e-05,
      "step": 6997,
      "training_step_time": 9.916340351104736
    },
    {
      "epoch": 2.6691436767578125e-05,
      "model_forward_time": 3.139037609100342,
      "step": 6997
    },
    {
      "dynamics_loss_avg": 0.0687162283807993,
      "epoch": 2.6691436767578125e-05,
      "step": 6997
    },
    {
      "action_loss_avg": 0.023481682967394592,
      "epoch": 2.6691436767578125e-05,
      "step": 6997
    },
    {
      "epoch": 2.6691436767578125e-05,
      "step": 6997,
      "training_step_time": 9.918147563934326
    },
    {
      "epoch": 2.6691436767578125e-05,
      "model_forward_time": 3.139301061630249,
      "step": 6997
    },
    {
      "epoch": 2.6691436767578125e-05,
      "step": 6997,
      "training_step_time": 10.014616012573242
    },
    {
      "epoch": 2.669525146484375e-05,
      "model_forward_time": 3.402503252029419,
      "step": 6998
    },
    {
      "epoch": 2.669525146484375e-05,
      "step": 6998,
      "training_step_time": 10.179296016693115
    },
    {
      "epoch": 2.669525146484375e-05,
      "model_forward_time": 3.1354079246520996,
      "step": 6998
    },
    {
      "epoch": 2.669525146484375e-05,
      "step": 6998,
      "training_step_time": 9.907468557357788
    },
    {
      "epoch": 2.669525146484375e-05,
      "model_forward_time": 3.133904457092285,
      "step": 6998
    },
    {
      "epoch": 2.669525146484375e-05,
      "step": 6998,
      "training_step_time": 9.905559539794922
    },
    {
      "epoch": 2.669525146484375e-05,
      "model_forward_time": 3.137698173522949,
      "step": 6998
    },
    {
      "epoch": 2.669525146484375e-05,
      "step": 6998,
      "training_step_time": 10.299690246582031
    },
    {
      "epoch": 2.6699066162109374e-05,
      "model_forward_time": 3.179997682571411,
      "step": 6999
    },
    {
      "epoch": 2.6699066162109374e-05,
      "step": 6999,
      "training_step_time": 9.954282283782959
    },
    {
      "epoch": 2.6699066162109374e-05,
      "model_forward_time": 3.4196457862854004,
      "step": 6999
    },
    {
      "epoch": 2.6699066162109374e-05,
      "step": 6999,
      "training_step_time": 10.186991930007935
    },
    {
      "epoch": 2.6699066162109374e-05,
      "model_forward_time": 3.136863946914673,
      "step": 6999
    },
    {
      "epoch": 2.6699066162109374e-05,
      "step": 6999,
      "training_step_time": 9.909386396408081
    },
    {
      "epoch": 2.6699066162109374e-05,
      "model_forward_time": 3.136265993118286,
      "step": 6999
    },
    {
      "epoch": 2.6699066162109374e-05,
      "step": 6999,
      "training_step_time": 10.007436752319336
    },
    {
      "epoch": 2.6702880859375e-05,
      "grad_norm": 0.5259419083595276,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 7000
    },
    {
      "epoch": 2.6702880859375e-05,
      "model_forward_time": 3.166487693786621,
      "step": 7000
    },
    {
      "dynamics_loss_avg": 0.0507304429076612,
      "epoch": 2.6702880859375e-05,
      "step": 7000
    },
    {
      "action_loss_avg": 0.019580386765301227,
      "epoch": 2.6702880859375e-05,
      "step": 7000
    },
    {
      "epoch": 2.6702880859375e-05,
      "step": 7000,
      "training_step_time": 9.92469573020935
    },
    {
      "epoch": 2.6702880859375e-05,
      "model_forward_time": 3.1268210411071777,
      "step": 7000
    },
    {
      "epoch": 2.6702880859375e-05,
      "step": 7000,
      "training_step_time": 9.885126113891602
    },
    {
      "epoch": 2.6702880859375e-05,
      "model_forward_time": 5.06401252746582,
      "step": 7000
    },
    {
      "epoch": 2.6702880859375e-05,
      "step": 7000,
      "training_step_time": 11.820027828216553
    },
    {
      "epoch": 2.6702880859375e-05,
      "model_forward_time": 3.13433575630188,
      "step": 7000
    },
    {
      "epoch": 2.6702880859375e-05,
      "step": 7000,
      "training_step_time": 10.300371885299683
    },
    {
      "epoch": 2.6706695556640626e-05,
      "model_forward_time": 3.144376039505005,
      "step": 7001
    },
    {
      "epoch": 2.6706695556640626e-05,
      "step": 7001,
      "training_step_time": 9.911990880966187
    },
    {
      "epoch": 2.6706695556640626e-05,
      "model_forward_time": 3.1289615631103516,
      "step": 7001
    },
    {
      "epoch": 2.6706695556640626e-05,
      "step": 7001,
      "training_step_time": 9.89701223373413
    },
    {
      "epoch": 2.6706695556640626e-05,
      "model_forward_time": 3.400118112564087,
      "step": 7001
    },
    {
      "epoch": 2.6706695556640626e-05,
      "step": 7001,
      "training_step_time": 10.170257568359375
    },
    {
      "epoch": 2.6706695556640626e-05,
      "model_forward_time": 3.1403555870056152,
      "step": 7001
    },
    {
      "epoch": 2.6706695556640626e-05,
      "step": 7001,
      "training_step_time": 10.28860354423523
    },
    {
      "epoch": 2.671051025390625e-05,
      "model_forward_time": 3.181725025177002,
      "step": 7002
    },
    {
      "epoch": 2.671051025390625e-05,
      "step": 7002,
      "training_step_time": 9.955565452575684
    },
    {
      "epoch": 2.671051025390625e-05,
      "model_forward_time": 3.139524459838867,
      "step": 7002
    },
    {
      "epoch": 2.671051025390625e-05,
      "step": 7002,
      "training_step_time": 9.913341999053955
    },
    {
      "epoch": 2.671051025390625e-05,
      "model_forward_time": 3.1453194618225098,
      "step": 7002
    },
    {
      "dynamics_loss_avg": 0.06399417128413916,
      "epoch": 2.671051025390625e-05,
      "step": 7002
    },
    {
      "action_loss_avg": 0.021762088965624572,
      "epoch": 2.671051025390625e-05,
      "step": 7002
    },
    {
      "epoch": 2.671051025390625e-05,
      "step": 7002,
      "training_step_time": 9.922949075698853
    },
    {
      "epoch": 2.671051025390625e-05,
      "model_forward_time": 3.141974687576294,
      "step": 7002
    },
    {
      "epoch": 2.671051025390625e-05,
      "step": 7002,
      "training_step_time": 10.284893989562988
    },
    {
      "epoch": 2.6714324951171875e-05,
      "model_forward_time": 3.1888129711151123,
      "step": 7003
    },
    {
      "epoch": 2.6714324951171875e-05,
      "step": 7003,
      "training_step_time": 9.969281673431396
    },
    {
      "epoch": 2.6714324951171875e-05,
      "model_forward_time": 2.436074733734131,
      "step": 7003
    },
    {
      "epoch": 2.6714324951171875e-05,
      "step": 7003,
      "training_step_time": 7.472933530807495
    },
    {
      "epoch": 2.6714324951171875e-05,
      "model_forward_time": 3.138031005859375,
      "step": 7003
    },
    {
      "epoch": 2.6714324951171875e-05,
      "step": 7003,
      "training_step_time": 9.906946420669556
    },
    {
      "epoch": 2.6714324951171875e-05,
      "model_forward_time": 3.1404266357421875,
      "step": 7003
    },
    {
      "epoch": 2.6714324951171875e-05,
      "step": 7003,
      "training_step_time": 10.262598037719727
    },
    {
      "epoch": 2.67181396484375e-05,
      "model_forward_time": 3.163414239883423,
      "step": 7004
    },
    {
      "epoch": 2.67181396484375e-05,
      "step": 7004,
      "training_step_time": 10.213462114334106
    },
    {
      "epoch": 2.67181396484375e-05,
      "model_forward_time": 3.136077642440796,
      "step": 7004
    },
    {
      "epoch": 2.67181396484375e-05,
      "step": 7004,
      "training_step_time": 9.908005475997925
    },
    {
      "epoch": 2.67181396484375e-05,
      "model_forward_time": 3.141979932785034,
      "step": 7004
    },
    {
      "epoch": 2.67181396484375e-05,
      "step": 7004,
      "training_step_time": 9.916075468063354
    },
    {
      "epoch": 2.67181396484375e-05,
      "model_forward_time": 3.1475753784179688,
      "step": 7004
    },
    {
      "epoch": 2.67181396484375e-05,
      "step": 7004,
      "training_step_time": 10.30053973197937
    },
    {
      "epoch": 2.6721954345703125e-05,
      "model_forward_time": 3.1779234409332275,
      "step": 7005
    },
    {
      "dynamics_loss_avg": 0.05275424290448427,
      "epoch": 2.6721954345703125e-05,
      "step": 7005
    },
    {
      "action_loss_avg": 0.018162472546100615,
      "epoch": 2.6721954345703125e-05,
      "step": 7005
    },
    {
      "epoch": 2.6721954345703125e-05,
      "step": 7005,
      "training_step_time": 9.958810806274414
    },
    {
      "epoch": 2.6721954345703125e-05,
      "model_forward_time": 3.415611982345581,
      "step": 7005
    },
    {
      "epoch": 2.6721954345703125e-05,
      "step": 7005,
      "training_step_time": 10.183831691741943
    },
    {
      "epoch": 2.6721954345703125e-05,
      "model_forward_time": 3.1508448123931885,
      "step": 7005
    },
    {
      "epoch": 2.6721954345703125e-05,
      "step": 7005,
      "training_step_time": 9.92324185371399
    },
    {
      "epoch": 2.6721954345703125e-05,
      "model_forward_time": 3.1428914070129395,
      "step": 7005
    },
    {
      "epoch": 2.6721954345703125e-05,
      "step": 7005,
      "training_step_time": 10.250584363937378
    },
    {
      "epoch": 2.672576904296875e-05,
      "model_forward_time": 3.179128408432007,
      "step": 7006
    },
    {
      "epoch": 2.672576904296875e-05,
      "step": 7006,
      "training_step_time": 9.949971914291382
    },
    {
      "epoch": 2.672576904296875e-05,
      "model_forward_time": 3.1445467472076416,
      "step": 7006
    },
    {
      "epoch": 2.672576904296875e-05,
      "step": 7006,
      "training_step_time": 9.91020655632019
    },
    {
      "epoch": 2.672576904296875e-05,
      "model_forward_time": 3.1406362056732178,
      "step": 7006
    },
    {
      "epoch": 2.672576904296875e-05,
      "step": 7006,
      "training_step_time": 10.176125049591064
    },
    {
      "epoch": 2.672576904296875e-05,
      "model_forward_time": 3.1400747299194336,
      "step": 7006
    },
    {
      "epoch": 2.672576904296875e-05,
      "step": 7006,
      "training_step_time": 10.26612138748169
    },
    {
      "epoch": 2.6729583740234374e-05,
      "model_forward_time": 3.169966697692871,
      "step": 7007
    },
    {
      "epoch": 2.6729583740234374e-05,
      "step": 7007,
      "training_step_time": 9.938332319259644
    },
    {
      "epoch": 2.6729583740234374e-05,
      "model_forward_time": 3.130506992340088,
      "step": 7007
    },
    {
      "epoch": 2.6729583740234374e-05,
      "step": 7007,
      "training_step_time": 9.895089864730835
    },
    {
      "epoch": 2.6729583740234374e-05,
      "model_forward_time": 3.134209156036377,
      "step": 7007
    },
    {
      "dynamics_loss_avg": 0.06031613163650036,
      "epoch": 2.6729583740234374e-05,
      "step": 7007
    },
    {
      "action_loss_avg": 0.028157241083681582,
      "epoch": 2.6729583740234374e-05,
      "step": 7007
    },
    {
      "epoch": 2.6729583740234374e-05,
      "step": 7007,
      "training_step_time": 9.910260677337646
    },
    {
      "epoch": 2.6729583740234374e-05,
      "model_forward_time": 3.418962001800537,
      "step": 7007
    },
    {
      "epoch": 2.6729583740234374e-05,
      "step": 7007,
      "training_step_time": 10.278096199035645
    },
    {
      "epoch": 2.67333984375e-05,
      "model_forward_time": 3.1790287494659424,
      "step": 7008
    },
    {
      "epoch": 2.67333984375e-05,
      "step": 7008,
      "training_step_time": 9.946413040161133
    },
    {
      "epoch": 2.67333984375e-05,
      "model_forward_time": 3.1354832649230957,
      "step": 7008
    },
    {
      "epoch": 2.67333984375e-05,
      "step": 7008,
      "training_step_time": 9.899100065231323
    },
    {
      "epoch": 2.67333984375e-05,
      "model_forward_time": 3.131296157836914,
      "step": 7008
    },
    {
      "epoch": 2.67333984375e-05,
      "step": 7008,
      "training_step_time": 9.898494243621826
    },
    {
      "epoch": 2.67333984375e-05,
      "model_forward_time": 3.137275218963623,
      "step": 7008
    },
    {
      "epoch": 2.67333984375e-05,
      "step": 7008,
      "training_step_time": 10.233994960784912
    },
    {
      "epoch": 2.6737213134765626e-05,
      "model_forward_time": 3.15224289894104,
      "step": 7009
    },
    {
      "epoch": 2.6737213134765626e-05,
      "step": 7009,
      "training_step_time": 10.201840877532959
    },
    {
      "epoch": 2.6737213134765626e-05,
      "model_forward_time": 3.143305540084839,
      "step": 7009
    },
    {
      "epoch": 2.6737213134765626e-05,
      "step": 7009,
      "training_step_time": 9.917185068130493
    },
    {
      "epoch": 2.6737213134765626e-05,
      "model_forward_time": 3.1370840072631836,
      "step": 7009
    },
    {
      "epoch": 2.6737213134765626e-05,
      "step": 7009,
      "training_step_time": 9.908083200454712
    },
    {
      "epoch": 2.6737213134765626e-05,
      "model_forward_time": 3.1407201290130615,
      "step": 7009
    },
    {
      "epoch": 2.6737213134765626e-05,
      "step": 7009,
      "training_step_time": 10.318389415740967
    },
    {
      "epoch": 2.674102783203125e-05,
      "grad_norm": 0.48367923498153687,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 7010
    },
    {
      "epoch": 2.674102783203125e-05,
      "model_forward_time": 3.182570219039917,
      "step": 7010
    },
    {
      "dynamics_loss_avg": 0.061996202915906906,
      "epoch": 2.674102783203125e-05,
      "step": 7010
    },
    {
      "action_loss_avg": 0.031688410136848685,
      "epoch": 2.674102783203125e-05,
      "step": 7010
    },
    {
      "epoch": 2.674102783203125e-05,
      "step": 7010,
      "training_step_time": 9.958242177963257
    },
    {
      "epoch": 2.674102783203125e-05,
      "model_forward_time": 3.4221296310424805,
      "step": 7010
    },
    {
      "epoch": 2.674102783203125e-05,
      "step": 7010,
      "training_step_time": 10.201822996139526
    },
    {
      "epoch": 2.674102783203125e-05,
      "model_forward_time": 3.1394264698028564,
      "step": 7010
    },
    {
      "epoch": 2.674102783203125e-05,
      "step": 7010,
      "training_step_time": 9.906368017196655
    },
    {
      "epoch": 2.674102783203125e-05,
      "model_forward_time": 3.1409356594085693,
      "step": 7010
    },
    {
      "epoch": 2.674102783203125e-05,
      "step": 7010,
      "training_step_time": 10.020639419555664
    },
    {
      "epoch": 2.6744842529296875e-05,
      "model_forward_time": 3.152693271636963,
      "step": 7011
    },
    {
      "epoch": 2.6744842529296875e-05,
      "step": 7011,
      "training_step_time": 9.926432847976685
    },
    {
      "epoch": 2.6744842529296875e-05,
      "model_forward_time": 3.1353542804718018,
      "step": 7011
    },
    {
      "epoch": 2.6744842529296875e-05,
      "step": 7011,
      "training_step_time": 9.917652130126953
    },
    {
      "epoch": 2.6744842529296875e-05,
      "model_forward_time": 3.4009957313537598,
      "step": 7011
    },
    {
      "epoch": 2.6744842529296875e-05,
      "step": 7011,
      "training_step_time": 10.177029609680176
    },
    {
      "epoch": 2.6744842529296875e-05,
      "model_forward_time": 3.1477317810058594,
      "step": 7011
    },
    {
      "epoch": 2.6744842529296875e-05,
      "step": 7011,
      "training_step_time": 10.029731750488281
    },
    {
      "epoch": 2.67486572265625e-05,
      "model_forward_time": 3.1810948848724365,
      "step": 7012
    },
    {
      "epoch": 2.67486572265625e-05,
      "step": 7012,
      "training_step_time": 9.955084085464478
    },
    {
      "epoch": 2.67486572265625e-05,
      "model_forward_time": 3.1390790939331055,
      "step": 7012
    },
    {
      "epoch": 2.67486572265625e-05,
      "step": 7012,
      "training_step_time": 9.908945560455322
    },
    {
      "epoch": 2.67486572265625e-05,
      "model_forward_time": 3.1428277492523193,
      "step": 7012
    },
    {
      "dynamics_loss_avg": 0.051226720213890076,
      "epoch": 2.67486572265625e-05,
      "step": 7012
    },
    {
      "action_loss_avg": 0.01680969400331378,
      "epoch": 2.67486572265625e-05,
      "step": 7012
    },
    {
      "epoch": 2.67486572265625e-05,
      "step": 7012,
      "training_step_time": 9.918787717819214
    },
    {
      "epoch": 2.67486572265625e-05,
      "model_forward_time": 3.417773723602295,
      "step": 7012
    },
    {
      "epoch": 2.67486572265625e-05,
      "step": 7012,
      "training_step_time": 10.292253971099854
    },
    {
      "epoch": 2.6752471923828125e-05,
      "model_forward_time": 3.1865899562835693,
      "step": 7013
    },
    {
      "epoch": 2.6752471923828125e-05,
      "step": 7013,
      "training_step_time": 9.969507694244385
    },
    {
      "epoch": 2.6752471923828125e-05,
      "model_forward_time": 3.138266086578369,
      "step": 7013
    },
    {
      "epoch": 2.6752471923828125e-05,
      "step": 7013,
      "training_step_time": 9.910610675811768
    },
    {
      "epoch": 2.6752471923828125e-05,
      "model_forward_time": 3.139188051223755,
      "step": 7013
    },
    {
      "epoch": 2.6752471923828125e-05,
      "step": 7013,
      "training_step_time": 9.913479328155518
    },
    {
      "epoch": 2.6752471923828125e-05,
      "model_forward_time": 3.1392171382904053,
      "step": 7013
    },
    {
      "epoch": 2.6752471923828125e-05,
      "step": 7013,
      "training_step_time": 10.289517402648926
    },
    {
      "epoch": 2.675628662109375e-05,
      "model_forward_time": 3.466510534286499,
      "step": 7014
    },
    {
      "epoch": 2.675628662109375e-05,
      "step": 7014,
      "training_step_time": 10.238296747207642
    },
    {
      "epoch": 2.675628662109375e-05,
      "model_forward_time": 3.1367828845977783,
      "step": 7014
    },
    {
      "epoch": 2.675628662109375e-05,
      "step": 7014,
      "training_step_time": 9.904104471206665
    },
    {
      "epoch": 2.675628662109375e-05,
      "model_forward_time": 3.1415798664093018,
      "step": 7014
    },
    {
      "epoch": 2.675628662109375e-05,
      "step": 7014,
      "training_step_time": 9.914783239364624
    },
    {
      "epoch": 2.675628662109375e-05,
      "model_forward_time": 3.138685464859009,
      "step": 7014
    },
    {
      "epoch": 2.675628662109375e-05,
      "step": 7014,
      "training_step_time": 10.290186405181885
    },
    {
      "epoch": 2.6760101318359374e-05,
      "model_forward_time": 3.1422629356384277,
      "step": 7015
    },
    {
      "dynamics_loss_avg": 0.06543025672435761,
      "epoch": 2.6760101318359374e-05,
      "step": 7015
    },
    {
      "action_loss_avg": 0.03146258071064949,
      "epoch": 2.6760101318359374e-05,
      "step": 7015
    },
    {
      "epoch": 2.6760101318359374e-05,
      "step": 7015,
      "training_step_time": 9.917786598205566
    },
    {
      "epoch": 2.6760101318359374e-05,
      "model_forward_time": 3.411241054534912,
      "step": 7015
    },
    {
      "epoch": 2.6760101318359374e-05,
      "step": 7015,
      "training_step_time": 10.178104639053345
    },
    {
      "epoch": 2.6760101318359374e-05,
      "model_forward_time": 3.1386184692382812,
      "step": 7015
    },
    {
      "epoch": 2.6760101318359374e-05,
      "step": 7015,
      "training_step_time": 9.907915115356445
    },
    {
      "epoch": 2.6760101318359374e-05,
      "model_forward_time": 3.1368367671966553,
      "step": 7015
    },
    {
      "epoch": 2.6760101318359374e-05,
      "step": 7015,
      "training_step_time": 10.031599044799805
    },
    {
      "epoch": 2.6763916015625e-05,
      "model_forward_time": 2.477102279663086,
      "step": 7016
    },
    {
      "epoch": 2.6763916015625e-05,
      "step": 7016,
      "training_step_time": 7.508238792419434
    },
    {
      "epoch": 2.6763916015625e-05,
      "model_forward_time": 3.132967710494995,
      "step": 7016
    },
    {
      "epoch": 2.6763916015625e-05,
      "step": 7016,
      "training_step_time": 9.90211009979248
    },
    {
      "epoch": 2.6763916015625e-05,
      "model_forward_time": 3.426557779312134,
      "step": 7016
    },
    {
      "epoch": 2.6763916015625e-05,
      "step": 7016,
      "training_step_time": 10.1941237449646
    },
    {
      "epoch": 2.6763916015625e-05,
      "model_forward_time": 3.138899087905884,
      "step": 7016
    },
    {
      "epoch": 2.6763916015625e-05,
      "step": 7016,
      "training_step_time": 10.030246496200562
    },
    {
      "epoch": 2.6767730712890626e-05,
      "model_forward_time": 3.153324842453003,
      "step": 7017
    },
    {
      "epoch": 2.6767730712890626e-05,
      "step": 7017,
      "training_step_time": 9.92180609703064
    },
    {
      "epoch": 2.6767730712890626e-05,
      "model_forward_time": 3.1392290592193604,
      "step": 7017
    },
    {
      "epoch": 2.6767730712890626e-05,
      "step": 7017,
      "training_step_time": 9.903783321380615
    },
    {
      "epoch": 2.6767730712890626e-05,
      "model_forward_time": 3.1371371746063232,
      "step": 7017
    },
    {
      "dynamics_loss_avg": 0.05898034945130348,
      "epoch": 2.6767730712890626e-05,
      "step": 7017
    },
    {
      "action_loss_avg": 0.026711852941662072,
      "epoch": 2.6767730712890626e-05,
      "step": 7017
    },
    {
      "epoch": 2.6767730712890626e-05,
      "step": 7017,
      "training_step_time": 9.905842781066895
    },
    {
      "epoch": 2.6767730712890626e-05,
      "model_forward_time": 3.1373519897460938,
      "step": 7017
    },
    {
      "epoch": 2.6767730712890626e-05,
      "step": 7017,
      "training_step_time": 10.308692216873169
    },
    {
      "epoch": 2.677154541015625e-05,
      "model_forward_time": 3.152083396911621,
      "step": 7018
    },
    {
      "epoch": 2.677154541015625e-05,
      "step": 7018,
      "training_step_time": 9.918683767318726
    },
    {
      "epoch": 2.677154541015625e-05,
      "model_forward_time": 3.138237237930298,
      "step": 7018
    },
    {
      "epoch": 2.677154541015625e-05,
      "step": 7018,
      "training_step_time": 9.905884742736816
    },
    {
      "epoch": 2.677154541015625e-05,
      "model_forward_time": 3.133535861968994,
      "step": 7018
    },
    {
      "epoch": 2.677154541015625e-05,
      "step": 7018,
      "training_step_time": 9.899892330169678
    },
    {
      "epoch": 2.677154541015625e-05,
      "model_forward_time": 3.144113540649414,
      "step": 7018
    },
    {
      "epoch": 2.677154541015625e-05,
      "step": 7018,
      "training_step_time": 10.26124620437622
    },
    {
      "epoch": 2.6775360107421875e-05,
      "model_forward_time": 3.4367854595184326,
      "step": 7019
    },
    {
      "epoch": 2.6775360107421875e-05,
      "step": 7019,
      "training_step_time": 10.208992719650269
    },
    {
      "epoch": 2.6775360107421875e-05,
      "model_forward_time": 3.138882875442505,
      "step": 7019
    },
    {
      "epoch": 2.6775360107421875e-05,
      "step": 7019,
      "training_step_time": 9.909553527832031
    },
    {
      "epoch": 2.6775360107421875e-05,
      "model_forward_time": 3.138998031616211,
      "step": 7019
    },
    {
      "epoch": 2.6775360107421875e-05,
      "step": 7019,
      "training_step_time": 9.90809154510498
    },
    {
      "epoch": 2.6775360107421875e-05,
      "model_forward_time": 3.1428890228271484,
      "step": 7019
    },
    {
      "epoch": 2.6775360107421875e-05,
      "step": 7019,
      "training_step_time": 10.32890510559082
    },
    {
      "epoch": 2.67791748046875e-05,
      "grad_norm": 0.5186468958854675,
      "learning_rate": 1e-05,
      "loss": 0.0817,
      "step": 7020
    },
    {
      "epoch": 2.67791748046875e-05,
      "model_forward_time": 3.152167558670044,
      "step": 7020
    },
    {
      "dynamics_loss_avg": 0.0573038199916482,
      "epoch": 2.67791748046875e-05,
      "step": 7020
    },
    {
      "action_loss_avg": 0.023607354518026112,
      "epoch": 2.67791748046875e-05,
      "step": 7020
    },
    {
      "epoch": 2.67791748046875e-05,
      "step": 7020,
      "training_step_time": 9.932570457458496
    },
    {
      "epoch": 2.67791748046875e-05,
      "model_forward_time": 3.416916608810425,
      "step": 7020
    },
    {
      "epoch": 2.67791748046875e-05,
      "step": 7020,
      "training_step_time": 10.185178756713867
    },
    {
      "epoch": 2.67791748046875e-05,
      "model_forward_time": 3.13464093208313,
      "step": 7020
    },
    {
      "epoch": 2.67791748046875e-05,
      "step": 7020,
      "training_step_time": 9.908585548400879
    },
    {
      "epoch": 2.67791748046875e-05,
      "model_forward_time": 3.1390881538391113,
      "step": 7020
    },
    {
      "epoch": 2.67791748046875e-05,
      "step": 7020,
      "training_step_time": 10.015294551849365
    },
    {
      "epoch": 2.6782989501953125e-05,
      "model_forward_time": 3.1475305557250977,
      "step": 7021
    },
    {
      "epoch": 2.6782989501953125e-05,
      "step": 7021,
      "training_step_time": 9.921834468841553
    },
    {
      "epoch": 2.6782989501953125e-05,
      "model_forward_time": 3.1345081329345703,
      "step": 7021
    },
    {
      "epoch": 2.6782989501953125e-05,
      "step": 7021,
      "training_step_time": 9.906365633010864
    },
    {
      "epoch": 2.6782989501953125e-05,
      "model_forward_time": 3.4092304706573486,
      "step": 7021
    },
    {
      "epoch": 2.6782989501953125e-05,
      "step": 7021,
      "training_step_time": 10.182092905044556
    },
    {
      "epoch": 2.6782989501953125e-05,
      "model_forward_time": 3.1403934955596924,
      "step": 7021
    },
    {
      "epoch": 2.6782989501953125e-05,
      "step": 7021,
      "training_step_time": 10.030362129211426
    },
    {
      "epoch": 2.678680419921875e-05,
      "model_forward_time": 3.15303373336792,
      "step": 7022
    },
    {
      "epoch": 2.678680419921875e-05,
      "step": 7022,
      "training_step_time": 9.923966884613037
    },
    {
      "epoch": 2.678680419921875e-05,
      "model_forward_time": 3.135446786880493,
      "step": 7022
    },
    {
      "epoch": 2.678680419921875e-05,
      "step": 7022,
      "training_step_time": 9.909647941589355
    },
    {
      "epoch": 2.678680419921875e-05,
      "model_forward_time": 3.136916160583496,
      "step": 7022
    },
    {
      "dynamics_loss_avg": 0.07008968014270067,
      "epoch": 2.678680419921875e-05,
      "step": 7022
    },
    {
      "action_loss_avg": 0.027090131817385555,
      "epoch": 2.678680419921875e-05,
      "step": 7022
    },
    {
      "epoch": 2.678680419921875e-05,
      "step": 7022,
      "training_step_time": 9.912245750427246
    },
    {
      "epoch": 2.678680419921875e-05,
      "model_forward_time": 3.4172394275665283,
      "step": 7022
    },
    {
      "epoch": 2.678680419921875e-05,
      "step": 7022,
      "training_step_time": 10.287886142730713
    },
    {
      "epoch": 2.6790618896484374e-05,
      "model_forward_time": 2.482248067855835,
      "step": 7023
    },
    {
      "epoch": 2.6790618896484374e-05,
      "step": 7023,
      "training_step_time": 7.515077590942383
    },
    {
      "epoch": 2.6790618896484374e-05,
      "model_forward_time": 3.133537530899048,
      "step": 7023
    },
    {
      "epoch": 2.6790618896484374e-05,
      "step": 7023,
      "training_step_time": 9.907770156860352
    },
    {
      "epoch": 2.6790618896484374e-05,
      "model_forward_time": 3.137651205062866,
      "step": 7023
    },
    {
      "epoch": 2.6790618896484374e-05,
      "step": 7023,
      "training_step_time": 9.914085865020752
    },
    {
      "epoch": 2.6790618896484374e-05,
      "model_forward_time": 3.1361310482025146,
      "step": 7023
    },
    {
      "epoch": 2.6790618896484374e-05,
      "step": 7023,
      "training_step_time": 10.028173923492432
    },
    {
      "epoch": 2.679443359375e-05,
      "model_forward_time": 3.1453006267547607,
      "step": 7024
    },
    {
      "epoch": 2.679443359375e-05,
      "step": 7024,
      "training_step_time": 10.202493906021118
    },
    {
      "epoch": 2.679443359375e-05,
      "model_forward_time": 3.1342239379882812,
      "step": 7024
    },
    {
      "epoch": 2.679443359375e-05,
      "step": 7024,
      "training_step_time": 9.907274961471558
    },
    {
      "epoch": 2.679443359375e-05,
      "model_forward_time": 3.1398696899414062,
      "step": 7024
    },
    {
      "epoch": 2.679443359375e-05,
      "step": 7024,
      "training_step_time": 9.910202264785767
    },
    {
      "epoch": 2.679443359375e-05,
      "model_forward_time": 3.135380506515503,
      "step": 7024
    },
    {
      "epoch": 2.679443359375e-05,
      "step": 7024,
      "training_step_time": 10.308485269546509
    },
    {
      "epoch": 2.6798248291015626e-05,
      "model_forward_time": 3.1840415000915527,
      "step": 7025
    },
    {
      "dynamics_loss_avg": 0.04717586506158113,
      "epoch": 2.6798248291015626e-05,
      "step": 7025
    },
    {
      "action_loss_avg": 0.021823785407468675,
      "epoch": 2.6798248291015626e-05,
      "step": 7025
    },
    {
      "epoch": 2.6798248291015626e-05,
      "step": 7025,
      "training_step_time": 9.956870079040527
    },
    {
      "epoch": 2.6798248291015626e-05,
      "model_forward_time": 3.398695230484009,
      "step": 7025
    },
    {
      "epoch": 2.6798248291015626e-05,
      "step": 7025,
      "training_step_time": 10.165401697158813
    },
    {
      "epoch": 2.6798248291015626e-05,
      "model_forward_time": 3.1362054347991943,
      "step": 7025
    },
    {
      "epoch": 2.6798248291015626e-05,
      "step": 7025,
      "training_step_time": 9.90720820426941
    },
    {
      "epoch": 2.6798248291015626e-05,
      "model_forward_time": 3.1326634883880615,
      "step": 7025
    },
    {
      "epoch": 2.6798248291015626e-05,
      "step": 7025,
      "training_step_time": 10.304868459701538
    },
    {
      "epoch": 2.680206298828125e-05,
      "model_forward_time": 3.1500816345214844,
      "step": 7026
    },
    {
      "epoch": 2.680206298828125e-05,
      "step": 7026,
      "training_step_time": 9.92557668685913
    },
    {
      "epoch": 2.680206298828125e-05,
      "model_forward_time": 3.138763189315796,
      "step": 7026
    },
    {
      "epoch": 2.680206298828125e-05,
      "step": 7026,
      "training_step_time": 9.912162780761719
    },
    {
      "epoch": 2.680206298828125e-05,
      "model_forward_time": 3.137375593185425,
      "step": 7026
    },
    {
      "epoch": 2.680206298828125e-05,
      "step": 7026,
      "training_step_time": 10.180125713348389
    },
    {
      "epoch": 2.680206298828125e-05,
      "model_forward_time": 3.1396872997283936,
      "step": 7026
    },
    {
      "epoch": 2.680206298828125e-05,
      "step": 7026,
      "training_step_time": 10.252185583114624
    },
    {
      "epoch": 2.6805877685546876e-05,
      "model_forward_time": 3.1746695041656494,
      "step": 7027
    },
    {
      "epoch": 2.6805877685546876e-05,
      "step": 7027,
      "training_step_time": 9.947302341461182
    },
    {
      "epoch": 2.6805877685546876e-05,
      "model_forward_time": 3.1374478340148926,
      "step": 7027
    },
    {
      "epoch": 2.6805877685546876e-05,
      "step": 7027,
      "training_step_time": 9.905553102493286
    },
    {
      "epoch": 2.6805877685546876e-05,
      "model_forward_time": 3.1332902908325195,
      "step": 7027
    },
    {
      "dynamics_loss_avg": 0.06950573697686195,
      "epoch": 2.6805877685546876e-05,
      "step": 7027
    },
    {
      "action_loss_avg": 0.02528941254131496,
      "epoch": 2.6805877685546876e-05,
      "step": 7027
    },
    {
      "epoch": 2.6805877685546876e-05,
      "step": 7027,
      "training_step_time": 9.910073041915894
    },
    {
      "epoch": 2.6805877685546876e-05,
      "model_forward_time": 3.396480083465576,
      "step": 7027
    },
    {
      "epoch": 2.6805877685546876e-05,
      "step": 7027,
      "training_step_time": 10.26315450668335
    },
    {
      "epoch": 2.68096923828125e-05,
      "model_forward_time": 3.169904947280884,
      "step": 7028
    },
    {
      "epoch": 2.68096923828125e-05,
      "step": 7028,
      "training_step_time": 9.943591117858887
    },
    {
      "epoch": 2.68096923828125e-05,
      "model_forward_time": 3.137798547744751,
      "step": 7028
    },
    {
      "epoch": 2.68096923828125e-05,
      "step": 7028,
      "training_step_time": 9.910253286361694
    },
    {
      "epoch": 2.68096923828125e-05,
      "model_forward_time": 3.133382558822632,
      "step": 7028
    },
    {
      "epoch": 2.68096923828125e-05,
      "step": 7028,
      "training_step_time": 9.904470920562744
    },
    {
      "epoch": 2.68096923828125e-05,
      "model_forward_time": 3.1352438926696777,
      "step": 7028
    },
    {
      "epoch": 2.68096923828125e-05,
      "step": 7028,
      "training_step_time": 10.315890550613403
    },
    {
      "epoch": 2.6813507080078125e-05,
      "model_forward_time": 3.1746883392333984,
      "step": 7029
    },
    {
      "epoch": 2.6813507080078125e-05,
      "step": 7029,
      "training_step_time": 10.247578144073486
    },
    {
      "epoch": 2.6813507080078125e-05,
      "model_forward_time": 3.1365396976470947,
      "step": 7029
    },
    {
      "epoch": 2.6813507080078125e-05,
      "step": 7029,
      "training_step_time": 9.910669565200806
    },
    {
      "epoch": 2.6813507080078125e-05,
      "model_forward_time": 3.141118049621582,
      "step": 7029
    },
    {
      "epoch": 2.6813507080078125e-05,
      "step": 7029,
      "training_step_time": 9.91568374633789
    },
    {
      "epoch": 2.6813507080078125e-05,
      "model_forward_time": 3.14453387260437,
      "step": 7029
    },
    {
      "epoch": 2.6813507080078125e-05,
      "step": 7029,
      "training_step_time": 10.250180721282959
    },
    {
      "epoch": 2.681732177734375e-05,
      "grad_norm": 0.46760642528533936,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 7030
    },
    {
      "epoch": 2.681732177734375e-05,
      "model_forward_time": 3.1908154487609863,
      "step": 7030
    },
    {
      "dynamics_loss_avg": 0.051861372590065,
      "epoch": 2.681732177734375e-05,
      "step": 7030
    },
    {
      "action_loss_avg": 0.019041345175355672,
      "epoch": 2.681732177734375e-05,
      "step": 7030
    },
    {
      "epoch": 2.681732177734375e-05,
      "step": 7030,
      "training_step_time": 9.97152042388916
    },
    {
      "epoch": 2.681732177734375e-05,
      "model_forward_time": 3.426867961883545,
      "step": 7030
    },
    {
      "epoch": 2.681732177734375e-05,
      "step": 7030,
      "training_step_time": 10.198261737823486
    },
    {
      "epoch": 2.681732177734375e-05,
      "model_forward_time": 3.132765054702759,
      "step": 7030
    },
    {
      "epoch": 2.681732177734375e-05,
      "step": 7030,
      "training_step_time": 9.907686948776245
    },
    {
      "epoch": 2.681732177734375e-05,
      "model_forward_time": 3.138115167617798,
      "step": 7030
    },
    {
      "epoch": 2.681732177734375e-05,
      "step": 7030,
      "training_step_time": 10.298074722290039
    },
    {
      "epoch": 2.6821136474609374e-05,
      "model_forward_time": 3.1522271633148193,
      "step": 7031
    },
    {
      "epoch": 2.6821136474609374e-05,
      "step": 7031,
      "training_step_time": 9.925501108169556
    },
    {
      "epoch": 2.6821136474609374e-05,
      "model_forward_time": 3.1441726684570312,
      "step": 7031
    },
    {
      "epoch": 2.6821136474609374e-05,
      "step": 7031,
      "training_step_time": 9.916007280349731
    },
    {
      "epoch": 2.6821136474609374e-05,
      "model_forward_time": 3.3958663940429688,
      "step": 7031
    },
    {
      "epoch": 2.6821136474609374e-05,
      "step": 7031,
      "training_step_time": 10.169843196868896
    },
    {
      "epoch": 2.6821136474609374e-05,
      "model_forward_time": 2.435934543609619,
      "step": 7031
    },
    {
      "epoch": 2.6821136474609374e-05,
      "step": 7031,
      "training_step_time": 10.026390314102173
    },
    {
      "epoch": 2.6824951171875e-05,
      "model_forward_time": 3.1474545001983643,
      "step": 7032
    },
    {
      "epoch": 2.6824951171875e-05,
      "step": 7032,
      "training_step_time": 9.921879291534424
    },
    {
      "epoch": 2.6824951171875e-05,
      "model_forward_time": 2.432642936706543,
      "step": 7032
    },
    {
      "epoch": 2.6824951171875e-05,
      "step": 7032,
      "training_step_time": 7.459219455718994
    },
    {
      "epoch": 2.6824951171875e-05,
      "model_forward_time": 3.1324095726013184,
      "step": 7032
    },
    {
      "dynamics_loss_avg": 0.056254766695201394,
      "epoch": 2.6824951171875e-05,
      "step": 7032
    },
    {
      "action_loss_avg": 0.02146424213424325,
      "epoch": 2.6824951171875e-05,
      "step": 7032
    },
    {
      "epoch": 2.6824951171875e-05,
      "step": 7032,
      "training_step_time": 9.903220415115356
    },
    {
      "epoch": 2.6824951171875e-05,
      "model_forward_time": 3.1325628757476807,
      "step": 7032
    },
    {
      "epoch": 2.6824951171875e-05,
      "step": 7032,
      "training_step_time": 10.286164999008179
    },
    {
      "epoch": 2.6828765869140626e-05,
      "model_forward_time": 3.1793572902679443,
      "step": 7033
    },
    {
      "epoch": 2.6828765869140626e-05,
      "step": 7033,
      "training_step_time": 9.952549934387207
    },
    {
      "epoch": 2.6828765869140626e-05,
      "model_forward_time": 3.135806083679199,
      "step": 7033
    },
    {
      "epoch": 2.6828765869140626e-05,
      "step": 7033,
      "training_step_time": 9.915104627609253
    },
    {
      "epoch": 2.6828765869140626e-05,
      "model_forward_time": 3.1391937732696533,
      "step": 7033
    },
    {
      "epoch": 2.6828765869140626e-05,
      "step": 7033,
      "training_step_time": 9.916814804077148
    },
    {
      "epoch": 2.6828765869140626e-05,
      "model_forward_time": 3.1393985748291016,
      "step": 7033
    },
    {
      "epoch": 2.6828765869140626e-05,
      "step": 7033,
      "training_step_time": 10.331112146377563
    },
    {
      "epoch": 2.683258056640625e-05,
      "model_forward_time": 3.1753079891204834,
      "step": 7034
    },
    {
      "epoch": 2.683258056640625e-05,
      "step": 7034,
      "training_step_time": 10.229727506637573
    },
    {
      "epoch": 2.683258056640625e-05,
      "model_forward_time": 3.1381006240844727,
      "step": 7034
    },
    {
      "epoch": 2.683258056640625e-05,
      "step": 7034,
      "training_step_time": 9.917603254318237
    },
    {
      "epoch": 2.683258056640625e-05,
      "model_forward_time": 3.1429126262664795,
      "step": 7034
    },
    {
      "epoch": 2.683258056640625e-05,
      "step": 7034,
      "training_step_time": 9.92264437675476
    },
    {
      "epoch": 2.683258056640625e-05,
      "model_forward_time": 3.137462854385376,
      "step": 7034
    },
    {
      "epoch": 2.683258056640625e-05,
      "step": 7034,
      "training_step_time": 10.307874917984009
    },
    {
      "epoch": 2.6836395263671876e-05,
      "model_forward_time": 3.1607494354248047,
      "step": 7035
    },
    {
      "dynamics_loss_avg": 0.05165490824729204,
      "epoch": 2.6836395263671876e-05,
      "step": 7035
    },
    {
      "action_loss_avg": 0.018941976875066758,
      "epoch": 2.6836395263671876e-05,
      "step": 7035
    },
    {
      "epoch": 2.6836395263671876e-05,
      "step": 7035,
      "training_step_time": 9.946322679519653
    },
    {
      "epoch": 2.6836395263671876e-05,
      "model_forward_time": 3.1369636058807373,
      "step": 7035
    },
    {
      "epoch": 2.6836395263671876e-05,
      "step": 7035,
      "training_step_time": 10.199063777923584
    },
    {
      "epoch": 2.6836395263671876e-05,
      "model_forward_time": 3.1391537189483643,
      "step": 7035
    },
    {
      "epoch": 2.6836395263671876e-05,
      "step": 7035,
      "training_step_time": 9.918792963027954
    },
    {
      "epoch": 2.6836395263671876e-05,
      "model_forward_time": 3.1392874717712402,
      "step": 7035
    },
    {
      "epoch": 2.6836395263671876e-05,
      "step": 7035,
      "training_step_time": 10.315596580505371
    },
    {
      "epoch": 2.68402099609375e-05,
      "model_forward_time": 3.184394121170044,
      "step": 7036
    },
    {
      "epoch": 2.68402099609375e-05,
      "step": 7036,
      "training_step_time": 9.961894989013672
    },
    {
      "epoch": 2.68402099609375e-05,
      "model_forward_time": 3.1384549140930176,
      "step": 7036
    },
    {
      "epoch": 2.68402099609375e-05,
      "step": 7036,
      "training_step_time": 9.917010068893433
    },
    {
      "epoch": 2.68402099609375e-05,
      "model_forward_time": 3.4168646335601807,
      "step": 7036
    },
    {
      "epoch": 2.68402099609375e-05,
      "step": 7036,
      "training_step_time": 10.189917802810669
    },
    {
      "epoch": 2.68402099609375e-05,
      "model_forward_time": 3.1341657638549805,
      "step": 7036
    },
    {
      "epoch": 2.68402099609375e-05,
      "step": 7036,
      "training_step_time": 10.272199869155884
    },
    {
      "epoch": 2.6844024658203125e-05,
      "model_forward_time": 3.16676926612854,
      "step": 7037
    },
    {
      "epoch": 2.6844024658203125e-05,
      "step": 7037,
      "training_step_time": 9.946423768997192
    },
    {
      "epoch": 2.6844024658203125e-05,
      "model_forward_time": 3.1351919174194336,
      "step": 7037
    },
    {
      "epoch": 2.6844024658203125e-05,
      "step": 7037,
      "training_step_time": 9.910742282867432
    },
    {
      "epoch": 2.6844024658203125e-05,
      "model_forward_time": 3.136786460876465,
      "step": 7037
    },
    {
      "dynamics_loss_avg": 0.05936795938760042,
      "epoch": 2.6844024658203125e-05,
      "step": 7037
    },
    {
      "action_loss_avg": 0.02001745654270053,
      "epoch": 2.6844024658203125e-05,
      "step": 7037
    },
    {
      "epoch": 2.6844024658203125e-05,
      "step": 7037,
      "training_step_time": 9.913525342941284
    },
    {
      "epoch": 2.6844024658203125e-05,
      "model_forward_time": 3.1410839557647705,
      "step": 7037
    },
    {
      "epoch": 2.6844024658203125e-05,
      "step": 7037,
      "training_step_time": 10.283693313598633
    },
    {
      "epoch": 2.684783935546875e-05,
      "model_forward_time": 3.1708662509918213,
      "step": 7038
    },
    {
      "epoch": 2.684783935546875e-05,
      "step": 7038,
      "training_step_time": 9.95459771156311
    },
    {
      "epoch": 2.684783935546875e-05,
      "model_forward_time": 3.1491851806640625,
      "step": 7038
    },
    {
      "epoch": 2.684783935546875e-05,
      "step": 7038,
      "training_step_time": 9.93022084236145
    },
    {
      "epoch": 2.684783935546875e-05,
      "model_forward_time": 3.147054672241211,
      "step": 7038
    },
    {
      "epoch": 2.684783935546875e-05,
      "step": 7038,
      "training_step_time": 9.929768085479736
    },
    {
      "epoch": 2.684783935546875e-05,
      "model_forward_time": 3.147026538848877,
      "step": 7038
    },
    {
      "epoch": 2.684783935546875e-05,
      "step": 7038,
      "training_step_time": 10.29511547088623
    },
    {
      "epoch": 2.6851654052734374e-05,
      "model_forward_time": 3.42687726020813,
      "step": 7039
    },
    {
      "epoch": 2.6851654052734374e-05,
      "step": 7039,
      "training_step_time": 10.209996461868286
    },
    {
      "epoch": 2.6851654052734374e-05,
      "model_forward_time": 3.1509809494018555,
      "step": 7039
    },
    {
      "epoch": 2.6851654052734374e-05,
      "step": 7039,
      "training_step_time": 9.931239366531372
    },
    {
      "epoch": 2.6851654052734374e-05,
      "model_forward_time": 3.149050712585449,
      "step": 7039
    },
    {
      "epoch": 2.6851654052734374e-05,
      "step": 7039,
      "training_step_time": 9.93065619468689
    },
    {
      "epoch": 2.6851654052734374e-05,
      "model_forward_time": 3.1575734615325928,
      "step": 7039
    },
    {
      "epoch": 2.6851654052734374e-05,
      "step": 7039,
      "training_step_time": 10.032579183578491
    },
    {
      "epoch": 2.685546875e-05,
      "grad_norm": 0.5005838871002197,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7040
    },
    {
      "epoch": 2.685546875e-05,
      "model_forward_time": 3.161771774291992,
      "step": 7040
    },
    {
      "dynamics_loss_avg": 0.06565926894545555,
      "epoch": 2.685546875e-05,
      "step": 7040
    },
    {
      "action_loss_avg": 0.02535968217998743,
      "epoch": 2.685546875e-05,
      "step": 7040
    },
    {
      "epoch": 2.685546875e-05,
      "step": 7040,
      "training_step_time": 9.9427969455719
    },
    {
      "epoch": 2.685546875e-05,
      "model_forward_time": 3.413407325744629,
      "step": 7040
    },
    {
      "epoch": 2.685546875e-05,
      "step": 7040,
      "training_step_time": 10.185474872589111
    },
    {
      "epoch": 2.685546875e-05,
      "model_forward_time": 3.1328377723693848,
      "step": 7040
    },
    {
      "epoch": 2.685546875e-05,
      "step": 7040,
      "training_step_time": 9.904457569122314
    },
    {
      "epoch": 2.685546875e-05,
      "model_forward_time": 3.131342887878418,
      "step": 7040
    },
    {
      "epoch": 2.685546875e-05,
      "step": 7040,
      "training_step_time": 10.33003854751587
    },
    {
      "epoch": 2.6859283447265626e-05,
      "model_forward_time": 3.168731689453125,
      "step": 7041
    },
    {
      "epoch": 2.6859283447265626e-05,
      "step": 7041,
      "training_step_time": 9.939541816711426
    },
    {
      "epoch": 2.6859283447265626e-05,
      "model_forward_time": 3.1446597576141357,
      "step": 7041
    },
    {
      "epoch": 2.6859283447265626e-05,
      "step": 7041,
      "training_step_time": 9.917892456054688
    },
    {
      "epoch": 2.6859283447265626e-05,
      "model_forward_time": 3.427448034286499,
      "step": 7041
    },
    {
      "epoch": 2.6859283447265626e-05,
      "step": 7041,
      "training_step_time": 10.19723105430603
    },
    {
      "epoch": 2.6859283447265626e-05,
      "model_forward_time": 3.141355514526367,
      "step": 7041
    },
    {
      "epoch": 2.6859283447265626e-05,
      "step": 7041,
      "training_step_time": 10.271306276321411
    },
    {
      "epoch": 2.686309814453125e-05,
      "model_forward_time": 3.180675745010376,
      "step": 7042
    },
    {
      "epoch": 2.686309814453125e-05,
      "step": 7042,
      "training_step_time": 9.956471920013428
    },
    {
      "epoch": 2.686309814453125e-05,
      "model_forward_time": 3.1366381645202637,
      "step": 7042
    },
    {
      "epoch": 2.686309814453125e-05,
      "step": 7042,
      "training_step_time": 9.911552667617798
    },
    {
      "epoch": 2.686309814453125e-05,
      "model_forward_time": 3.140129327774048,
      "step": 7042
    },
    {
      "dynamics_loss_avg": 0.04375414224341512,
      "epoch": 2.686309814453125e-05,
      "step": 7042
    },
    {
      "action_loss_avg": 0.014262927463278174,
      "epoch": 2.686309814453125e-05,
      "step": 7042
    },
    {
      "epoch": 2.686309814453125e-05,
      "step": 7042,
      "training_step_time": 9.921890258789062
    },
    {
      "epoch": 2.686309814453125e-05,
      "model_forward_time": 3.4289770126342773,
      "step": 7042
    },
    {
      "epoch": 2.686309814453125e-05,
      "step": 7042,
      "training_step_time": 10.296130180358887
    },
    {
      "epoch": 2.6866912841796876e-05,
      "model_forward_time": 2.478398323059082,
      "step": 7043
    },
    {
      "epoch": 2.6866912841796876e-05,
      "step": 7043,
      "training_step_time": 7.507457733154297
    },
    {
      "epoch": 2.6866912841796876e-05,
      "model_forward_time": 3.1303112506866455,
      "step": 7043
    },
    {
      "epoch": 2.6866912841796876e-05,
      "step": 7043,
      "training_step_time": 9.902235507965088
    },
    {
      "epoch": 2.6866912841796876e-05,
      "model_forward_time": 3.1393656730651855,
      "step": 7043
    },
    {
      "epoch": 2.6866912841796876e-05,
      "step": 7043,
      "training_step_time": 9.911638259887695
    },
    {
      "epoch": 2.6866912841796876e-05,
      "model_forward_time": 3.1399614810943604,
      "step": 7043
    },
    {
      "epoch": 2.6866912841796876e-05,
      "step": 7043,
      "training_step_time": 10.317046165466309
    },
    {
      "epoch": 2.68707275390625e-05,
      "model_forward_time": 3.175506830215454,
      "step": 7044
    },
    {
      "epoch": 2.68707275390625e-05,
      "step": 7044,
      "training_step_time": 10.253591299057007
    },
    {
      "epoch": 2.68707275390625e-05,
      "model_forward_time": 3.1349823474884033,
      "step": 7044
    },
    {
      "epoch": 2.68707275390625e-05,
      "step": 7044,
      "training_step_time": 9.907649517059326
    },
    {
      "epoch": 2.68707275390625e-05,
      "model_forward_time": 3.1365160942077637,
      "step": 7044
    },
    {
      "epoch": 2.68707275390625e-05,
      "step": 7044,
      "training_step_time": 9.90777850151062
    },
    {
      "epoch": 2.68707275390625e-05,
      "model_forward_time": 3.1342873573303223,
      "step": 7044
    },
    {
      "epoch": 2.68707275390625e-05,
      "step": 7044,
      "training_step_time": 10.010964393615723
    },
    {
      "epoch": 2.6874542236328125e-05,
      "model_forward_time": 3.1798040866851807,
      "step": 7045
    },
    {
      "dynamics_loss_avg": 0.06547122187912464,
      "epoch": 2.6874542236328125e-05,
      "step": 7045
    },
    {
      "action_loss_avg": 0.02311000139452517,
      "epoch": 2.6874542236328125e-05,
      "step": 7045
    },
    {
      "epoch": 2.6874542236328125e-05,
      "step": 7045,
      "training_step_time": 9.955753564834595
    },
    {
      "epoch": 2.6874542236328125e-05,
      "model_forward_time": 3.4081661701202393,
      "step": 7045
    },
    {
      "epoch": 2.6874542236328125e-05,
      "step": 7045,
      "training_step_time": 10.183167934417725
    },
    {
      "epoch": 2.6874542236328125e-05,
      "model_forward_time": 3.141228437423706,
      "step": 7045
    },
    {
      "epoch": 2.6874542236328125e-05,
      "step": 7045,
      "training_step_time": 9.923959970474243
    },
    {
      "epoch": 2.6874542236328125e-05,
      "model_forward_time": 3.1423797607421875,
      "step": 7045
    },
    {
      "epoch": 2.6874542236328125e-05,
      "step": 7045,
      "training_step_time": 10.325279474258423
    },
    {
      "epoch": 2.687835693359375e-05,
      "model_forward_time": 3.1493067741394043,
      "step": 7046
    },
    {
      "epoch": 2.687835693359375e-05,
      "step": 7046,
      "training_step_time": 9.936158657073975
    },
    {
      "epoch": 2.687835693359375e-05,
      "model_forward_time": 3.142627000808716,
      "step": 7046
    },
    {
      "epoch": 2.687835693359375e-05,
      "step": 7046,
      "training_step_time": 9.921098947525024
    },
    {
      "epoch": 2.687835693359375e-05,
      "model_forward_time": 3.137551784515381,
      "step": 7046
    },
    {
      "epoch": 2.687835693359375e-05,
      "step": 7046,
      "training_step_time": 10.191668033599854
    },
    {
      "epoch": 2.687835693359375e-05,
      "model_forward_time": 3.137158155441284,
      "step": 7046
    },
    {
      "epoch": 2.687835693359375e-05,
      "step": 7046,
      "training_step_time": 10.29816484451294
    },
    {
      "epoch": 2.6882171630859374e-05,
      "model_forward_time": 3.1755754947662354,
      "step": 7047
    },
    {
      "epoch": 2.6882171630859374e-05,
      "step": 7047,
      "training_step_time": 9.946887969970703
    },
    {
      "epoch": 2.6882171630859374e-05,
      "model_forward_time": 3.1414506435394287,
      "step": 7047
    },
    {
      "epoch": 2.6882171630859374e-05,
      "step": 7047,
      "training_step_time": 9.918017625808716
    },
    {
      "epoch": 2.6882171630859374e-05,
      "model_forward_time": 3.138847589492798,
      "step": 7047
    },
    {
      "dynamics_loss_avg": 0.06641047038137912,
      "epoch": 2.6882171630859374e-05,
      "step": 7047
    },
    {
      "action_loss_avg": 0.025139212608337402,
      "epoch": 2.6882171630859374e-05,
      "step": 7047
    },
    {
      "epoch": 2.6882171630859374e-05,
      "step": 7047,
      "training_step_time": 9.922141075134277
    },
    {
      "epoch": 2.6882171630859374e-05,
      "model_forward_time": 3.4005961418151855,
      "step": 7047
    },
    {
      "epoch": 2.6882171630859374e-05,
      "step": 7047,
      "training_step_time": 10.27584457397461
    },
    {
      "epoch": 2.6885986328125e-05,
      "model_forward_time": 3.1546900272369385,
      "step": 7048
    },
    {
      "epoch": 2.6885986328125e-05,
      "step": 7048,
      "training_step_time": 9.939860343933105
    },
    {
      "epoch": 2.6885986328125e-05,
      "model_forward_time": 3.1467716693878174,
      "step": 7048
    },
    {
      "epoch": 2.6885986328125e-05,
      "step": 7048,
      "training_step_time": 9.928846836090088
    },
    {
      "epoch": 2.6885986328125e-05,
      "model_forward_time": 3.1466898918151855,
      "step": 7048
    },
    {
      "epoch": 2.6885986328125e-05,
      "step": 7048,
      "training_step_time": 9.931692838668823
    },
    {
      "epoch": 2.6885986328125e-05,
      "model_forward_time": 3.1473429203033447,
      "step": 7048
    },
    {
      "epoch": 2.6885986328125e-05,
      "step": 7048,
      "training_step_time": 10.298106908798218
    },
    {
      "epoch": 2.6889801025390626e-05,
      "model_forward_time": 3.182723045349121,
      "step": 7049
    },
    {
      "epoch": 2.6889801025390626e-05,
      "step": 7049,
      "training_step_time": 10.233556270599365
    },
    {
      "epoch": 2.6889801025390626e-05,
      "model_forward_time": 3.1460177898406982,
      "step": 7049
    },
    {
      "epoch": 2.6889801025390626e-05,
      "step": 7049,
      "training_step_time": 9.930974960327148
    },
    {
      "epoch": 2.6889801025390626e-05,
      "model_forward_time": 3.1412997245788574,
      "step": 7049
    },
    {
      "epoch": 2.6889801025390626e-05,
      "step": 7049,
      "training_step_time": 9.92823338508606
    },
    {
      "epoch": 2.6889801025390626e-05,
      "model_forward_time": 3.1468024253845215,
      "step": 7049
    },
    {
      "epoch": 2.6889801025390626e-05,
      "step": 7049,
      "training_step_time": 10.041818857192993
    },
    {
      "epoch": 2.689361572265625e-05,
      "grad_norm": 0.3682999610900879,
      "learning_rate": 1e-05,
      "loss": 0.0852,
      "step": 7050
    },
    {
      "epoch": 2.689361572265625e-05,
      "model_forward_time": 3.1669552326202393,
      "step": 7050
    },
    {
      "dynamics_loss_avg": 0.06952038519084454,
      "epoch": 2.689361572265625e-05,
      "step": 7050
    },
    {
      "action_loss_avg": 0.035208352655172345,
      "epoch": 2.689361572265625e-05,
      "step": 7050
    },
    {
      "epoch": 2.689361572265625e-05,
      "step": 7050,
      "training_step_time": 9.956643342971802
    },
    {
      "epoch": 2.689361572265625e-05,
      "model_forward_time": 3.4130566120147705,
      "step": 7050
    },
    {
      "epoch": 2.689361572265625e-05,
      "step": 7050,
      "training_step_time": 10.186692237854004
    },
    {
      "epoch": 2.689361572265625e-05,
      "model_forward_time": 3.139732837677002,
      "step": 7050
    },
    {
      "epoch": 2.689361572265625e-05,
      "step": 7050,
      "training_step_time": 9.913794040679932
    },
    {
      "epoch": 2.689361572265625e-05,
      "model_forward_time": 3.1375861167907715,
      "step": 7050
    },
    {
      "epoch": 2.689361572265625e-05,
      "step": 7050,
      "training_step_time": 10.300097465515137
    },
    {
      "epoch": 2.6897430419921876e-05,
      "model_forward_time": 3.1526389122009277,
      "step": 7051
    },
    {
      "epoch": 2.6897430419921876e-05,
      "step": 7051,
      "training_step_time": 9.928959369659424
    },
    {
      "epoch": 2.6897430419921876e-05,
      "model_forward_time": 3.134544849395752,
      "step": 7051
    },
    {
      "epoch": 2.6897430419921876e-05,
      "step": 7051,
      "training_step_time": 9.905730724334717
    },
    {
      "epoch": 2.6897430419921876e-05,
      "model_forward_time": 2.707582712173462,
      "step": 7051
    },
    {
      "epoch": 2.6897430419921876e-05,
      "step": 7051,
      "training_step_time": 7.731903314590454
    },
    {
      "epoch": 2.6897430419921876e-05,
      "model_forward_time": 3.134993553161621,
      "step": 7051
    },
    {
      "epoch": 2.6897430419921876e-05,
      "step": 7051,
      "training_step_time": 10.327345848083496
    },
    {
      "epoch": 2.69012451171875e-05,
      "model_forward_time": 3.1751317977905273,
      "step": 7052
    },
    {
      "epoch": 2.69012451171875e-05,
      "step": 7052,
      "training_step_time": 9.948618173599243
    },
    {
      "epoch": 2.69012451171875e-05,
      "model_forward_time": 3.134093761444092,
      "step": 7052
    },
    {
      "epoch": 2.69012451171875e-05,
      "step": 7052,
      "training_step_time": 9.90576696395874
    },
    {
      "epoch": 2.69012451171875e-05,
      "model_forward_time": 3.13384747505188,
      "step": 7052
    },
    {
      "dynamics_loss_avg": 0.04848107360303402,
      "epoch": 2.69012451171875e-05,
      "step": 7052
    },
    {
      "action_loss_avg": 0.020820984710007907,
      "epoch": 2.69012451171875e-05,
      "step": 7052
    },
    {
      "epoch": 2.69012451171875e-05,
      "step": 7052,
      "training_step_time": 9.906436681747437
    },
    {
      "epoch": 2.69012451171875e-05,
      "model_forward_time": 3.4126877784729004,
      "step": 7052
    },
    {
      "epoch": 2.69012451171875e-05,
      "step": 7052,
      "training_step_time": 10.312857389450073
    },
    {
      "epoch": 2.6905059814453125e-05,
      "model_forward_time": 3.1701929569244385,
      "step": 7053
    },
    {
      "epoch": 2.6905059814453125e-05,
      "step": 7053,
      "training_step_time": 9.948598146438599
    },
    {
      "epoch": 2.6905059814453125e-05,
      "model_forward_time": 3.137702703475952,
      "step": 7053
    },
    {
      "epoch": 2.6905059814453125e-05,
      "step": 7053,
      "training_step_time": 9.903136253356934
    },
    {
      "epoch": 2.6905059814453125e-05,
      "model_forward_time": 3.1415979862213135,
      "step": 7053
    },
    {
      "epoch": 2.6905059814453125e-05,
      "step": 7053,
      "training_step_time": 9.90699028968811
    },
    {
      "epoch": 2.6905059814453125e-05,
      "model_forward_time": 3.1402432918548584,
      "step": 7053
    },
    {
      "epoch": 2.6905059814453125e-05,
      "step": 7053,
      "training_step_time": 10.288624048233032
    },
    {
      "epoch": 2.690887451171875e-05,
      "model_forward_time": 3.4298624992370605,
      "step": 7054
    },
    {
      "epoch": 2.690887451171875e-05,
      "step": 7054,
      "training_step_time": 10.198888540267944
    },
    {
      "epoch": 2.690887451171875e-05,
      "model_forward_time": 3.140734910964966,
      "step": 7054
    },
    {
      "epoch": 2.690887451171875e-05,
      "step": 7054,
      "training_step_time": 9.908968210220337
    },
    {
      "epoch": 2.690887451171875e-05,
      "model_forward_time": 3.1409690380096436,
      "step": 7054
    },
    {
      "epoch": 2.690887451171875e-05,
      "step": 7054,
      "training_step_time": 9.910519361495972
    },
    {
      "epoch": 2.690887451171875e-05,
      "model_forward_time": 3.1366188526153564,
      "step": 7054
    },
    {
      "epoch": 2.690887451171875e-05,
      "step": 7054,
      "training_step_time": 10.251734972000122
    },
    {
      "epoch": 2.6912689208984374e-05,
      "model_forward_time": 2.477642297744751,
      "step": 7055
    },
    {
      "dynamics_loss_avg": 0.048288670182228086,
      "epoch": 2.6912689208984374e-05,
      "step": 7055
    },
    {
      "action_loss_avg": 0.018283876474015415,
      "epoch": 2.6912689208984374e-05,
      "step": 7055
    },
    {
      "epoch": 2.6912689208984374e-05,
      "step": 7055,
      "training_step_time": 7.510584354400635
    },
    {
      "epoch": 2.6912689208984374e-05,
      "model_forward_time": 3.1446521282196045,
      "step": 7055
    },
    {
      "epoch": 2.6912689208984374e-05,
      "step": 7055,
      "training_step_time": 10.189409494400024
    },
    {
      "epoch": 2.6912689208984374e-05,
      "model_forward_time": 3.1390678882598877,
      "step": 7055
    },
    {
      "epoch": 2.6912689208984374e-05,
      "step": 7055,
      "training_step_time": 9.904430389404297
    },
    {
      "epoch": 2.6912689208984374e-05,
      "model_forward_time": 3.136079788208008,
      "step": 7055
    },
    {
      "epoch": 2.6912689208984374e-05,
      "step": 7055,
      "training_step_time": 10.030030250549316
    },
    {
      "epoch": 2.691650390625e-05,
      "model_forward_time": 3.164367437362671,
      "step": 7056
    },
    {
      "epoch": 2.691650390625e-05,
      "step": 7056,
      "training_step_time": 9.933699369430542
    },
    {
      "epoch": 2.691650390625e-05,
      "model_forward_time": 3.1391103267669678,
      "step": 7056
    },
    {
      "epoch": 2.691650390625e-05,
      "step": 7056,
      "training_step_time": 9.91248869895935
    },
    {
      "epoch": 2.691650390625e-05,
      "model_forward_time": 3.4539601802825928,
      "step": 7056
    },
    {
      "epoch": 2.691650390625e-05,
      "step": 7056,
      "training_step_time": 10.229635238647461
    },
    {
      "epoch": 2.691650390625e-05,
      "model_forward_time": 3.141664743423462,
      "step": 7056
    },
    {
      "epoch": 2.691650390625e-05,
      "step": 7056,
      "training_step_time": 10.346753358840942
    },
    {
      "epoch": 2.6920318603515626e-05,
      "model_forward_time": 3.1851348876953125,
      "step": 7057
    },
    {
      "epoch": 2.6920318603515626e-05,
      "step": 7057,
      "training_step_time": 9.960851430892944
    },
    {
      "epoch": 2.6920318603515626e-05,
      "model_forward_time": 3.149214029312134,
      "step": 7057
    },
    {
      "epoch": 2.6920318603515626e-05,
      "step": 7057,
      "training_step_time": 9.926102876663208
    },
    {
      "epoch": 2.6920318603515626e-05,
      "model_forward_time": 3.1458027362823486,
      "step": 7057
    },
    {
      "dynamics_loss_avg": 0.08588211052119732,
      "epoch": 2.6920318603515626e-05,
      "step": 7057
    },
    {
      "action_loss_avg": 0.028560301521793008,
      "epoch": 2.6920318603515626e-05,
      "step": 7057
    },
    {
      "epoch": 2.6920318603515626e-05,
      "step": 7057,
      "training_step_time": 9.92294716835022
    },
    {
      "epoch": 2.6920318603515626e-05,
      "model_forward_time": 3.1442136764526367,
      "step": 7057
    },
    {
      "epoch": 2.6920318603515626e-05,
      "step": 7057,
      "training_step_time": 10.311848402023315
    },
    {
      "epoch": 2.692413330078125e-05,
      "model_forward_time": 3.1958625316619873,
      "step": 7058
    },
    {
      "epoch": 2.692413330078125e-05,
      "step": 7058,
      "training_step_time": 9.973194599151611
    },
    {
      "epoch": 2.692413330078125e-05,
      "model_forward_time": 3.1395297050476074,
      "step": 7058
    },
    {
      "epoch": 2.692413330078125e-05,
      "step": 7058,
      "training_step_time": 9.911192655563354
    },
    {
      "epoch": 2.692413330078125e-05,
      "model_forward_time": 3.1386542320251465,
      "step": 7058
    },
    {
      "epoch": 2.692413330078125e-05,
      "step": 7058,
      "training_step_time": 9.906954050064087
    },
    {
      "epoch": 2.692413330078125e-05,
      "model_forward_time": 3.145284652709961,
      "step": 7058
    },
    {
      "epoch": 2.692413330078125e-05,
      "step": 7058,
      "training_step_time": 10.233221054077148
    },
    {
      "epoch": 2.6927947998046876e-05,
      "model_forward_time": 3.4480206966400146,
      "step": 7059
    },
    {
      "epoch": 2.6927947998046876e-05,
      "step": 7059,
      "training_step_time": 10.217932224273682
    },
    {
      "epoch": 2.6927947998046876e-05,
      "model_forward_time": 3.1385467052459717,
      "step": 7059
    },
    {
      "epoch": 2.6927947998046876e-05,
      "step": 7059,
      "training_step_time": 9.9155912399292
    },
    {
      "epoch": 2.6927947998046876e-05,
      "model_forward_time": 3.1388609409332275,
      "step": 7059
    },
    {
      "epoch": 2.6927947998046876e-05,
      "step": 7059,
      "training_step_time": 9.907313585281372
    },
    {
      "epoch": 2.6927947998046876e-05,
      "model_forward_time": 3.1493706703186035,
      "step": 7059
    },
    {
      "epoch": 2.6927947998046876e-05,
      "step": 7059,
      "training_step_time": 10.306989431381226
    },
    {
      "epoch": 2.69317626953125e-05,
      "grad_norm": 0.34377428889274597,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 7060
    },
    {
      "epoch": 2.69317626953125e-05,
      "model_forward_time": 3.176621198654175,
      "step": 7060
    },
    {
      "dynamics_loss_avg": 0.05716408025473356,
      "epoch": 2.69317626953125e-05,
      "step": 7060
    },
    {
      "action_loss_avg": 0.021309383679181337,
      "epoch": 2.69317626953125e-05,
      "step": 7060
    },
    {
      "epoch": 2.69317626953125e-05,
      "step": 7060,
      "training_step_time": 9.95154070854187
    },
    {
      "epoch": 2.69317626953125e-05,
      "model_forward_time": 3.4241621494293213,
      "step": 7060
    },
    {
      "epoch": 2.69317626953125e-05,
      "step": 7060,
      "training_step_time": 10.192760467529297
    },
    {
      "epoch": 2.69317626953125e-05,
      "model_forward_time": 3.136589527130127,
      "step": 7060
    },
    {
      "epoch": 2.69317626953125e-05,
      "step": 7060,
      "training_step_time": 9.904595136642456
    },
    {
      "epoch": 2.69317626953125e-05,
      "model_forward_time": 3.13822078704834,
      "step": 7060
    },
    {
      "epoch": 2.69317626953125e-05,
      "step": 7060,
      "training_step_time": 10.274943351745605
    },
    {
      "epoch": 2.6935577392578125e-05,
      "model_forward_time": 3.147116184234619,
      "step": 7061
    },
    {
      "epoch": 2.6935577392578125e-05,
      "step": 7061,
      "training_step_time": 9.912754774093628
    },
    {
      "epoch": 2.6935577392578125e-05,
      "model_forward_time": 3.130998134613037,
      "step": 7061
    },
    {
      "epoch": 2.6935577392578125e-05,
      "step": 7061,
      "training_step_time": 9.89583683013916
    },
    {
      "epoch": 2.6935577392578125e-05,
      "model_forward_time": 3.391467809677124,
      "step": 7061
    },
    {
      "epoch": 2.6935577392578125e-05,
      "step": 7061,
      "training_step_time": 10.155322551727295
    },
    {
      "epoch": 2.6935577392578125e-05,
      "model_forward_time": 3.139193296432495,
      "step": 7061
    },
    {
      "epoch": 2.6935577392578125e-05,
      "step": 7061,
      "training_step_time": 10.305773496627808
    },
    {
      "epoch": 2.693939208984375e-05,
      "model_forward_time": 3.1708269119262695,
      "step": 7062
    },
    {
      "epoch": 2.693939208984375e-05,
      "step": 7062,
      "training_step_time": 9.943759679794312
    },
    {
      "epoch": 2.693939208984375e-05,
      "model_forward_time": 2.444812059402466,
      "step": 7062
    },
    {
      "epoch": 2.693939208984375e-05,
      "step": 7062,
      "training_step_time": 7.472140312194824
    },
    {
      "epoch": 2.693939208984375e-05,
      "model_forward_time": 3.1348822116851807,
      "step": 7062
    },
    {
      "dynamics_loss_avg": 0.062489939667284486,
      "epoch": 2.693939208984375e-05,
      "step": 7062
    },
    {
      "action_loss_avg": 0.021190203540027142,
      "epoch": 2.693939208984375e-05,
      "step": 7062
    },
    {
      "epoch": 2.693939208984375e-05,
      "step": 7062,
      "training_step_time": 9.903093099594116
    },
    {
      "epoch": 2.693939208984375e-05,
      "model_forward_time": 3.133289098739624,
      "step": 7062
    },
    {
      "epoch": 2.693939208984375e-05,
      "step": 7062,
      "training_step_time": 10.319685459136963
    },
    {
      "epoch": 2.6943206787109374e-05,
      "model_forward_time": 3.172218084335327,
      "step": 7063
    },
    {
      "epoch": 2.6943206787109374e-05,
      "step": 7063,
      "training_step_time": 9.943122863769531
    },
    {
      "epoch": 2.6943206787109374e-05,
      "model_forward_time": 3.1375672817230225,
      "step": 7063
    },
    {
      "epoch": 2.6943206787109374e-05,
      "step": 7063,
      "training_step_time": 9.906795024871826
    },
    {
      "epoch": 2.6943206787109374e-05,
      "model_forward_time": 3.144011974334717,
      "step": 7063
    },
    {
      "epoch": 2.6943206787109374e-05,
      "step": 7063,
      "training_step_time": 9.91938304901123
    },
    {
      "epoch": 2.6943206787109374e-05,
      "model_forward_time": 3.1441657543182373,
      "step": 7063
    },
    {
      "epoch": 2.6943206787109374e-05,
      "step": 7063,
      "training_step_time": 10.037826299667358
    },
    {
      "epoch": 2.6947021484375e-05,
      "model_forward_time": 3.149726390838623,
      "step": 7064
    },
    {
      "epoch": 2.6947021484375e-05,
      "step": 7064,
      "training_step_time": 10.203486919403076
    },
    {
      "epoch": 2.6947021484375e-05,
      "model_forward_time": 3.140918016433716,
      "step": 7064
    },
    {
      "epoch": 2.6947021484375e-05,
      "step": 7064,
      "training_step_time": 9.918014287948608
    },
    {
      "epoch": 2.6947021484375e-05,
      "model_forward_time": 3.1444756984710693,
      "step": 7064
    },
    {
      "epoch": 2.6947021484375e-05,
      "step": 7064,
      "training_step_time": 9.921669721603394
    },
    {
      "epoch": 2.6947021484375e-05,
      "model_forward_time": 3.1403472423553467,
      "step": 7064
    },
    {
      "epoch": 2.6947021484375e-05,
      "step": 7064,
      "training_step_time": 10.277724027633667
    },
    {
      "epoch": 2.6950836181640626e-05,
      "model_forward_time": 3.162294387817383,
      "step": 7065
    },
    {
      "dynamics_loss_avg": 0.05759614668786526,
      "epoch": 2.6950836181640626e-05,
      "step": 7065
    },
    {
      "action_loss_avg": 0.020289731863886117,
      "epoch": 2.6950836181640626e-05,
      "step": 7065
    },
    {
      "epoch": 2.6950836181640626e-05,
      "step": 7065,
      "training_step_time": 9.944295644760132
    },
    {
      "epoch": 2.6950836181640626e-05,
      "model_forward_time": 3.403663396835327,
      "step": 7065
    },
    {
      "epoch": 2.6950836181640626e-05,
      "step": 7065,
      "training_step_time": 10.173553943634033
    },
    {
      "epoch": 2.6950836181640626e-05,
      "model_forward_time": 3.138608932495117,
      "step": 7065
    },
    {
      "epoch": 2.6950836181640626e-05,
      "step": 7065,
      "training_step_time": 9.907693147659302
    },
    {
      "epoch": 2.6950836181640626e-05,
      "model_forward_time": 3.1326215267181396,
      "step": 7065
    },
    {
      "epoch": 2.6950836181640626e-05,
      "step": 7065,
      "training_step_time": 10.292205333709717
    },
    {
      "epoch": 2.695465087890625e-05,
      "model_forward_time": 3.1659631729125977,
      "step": 7066
    },
    {
      "epoch": 2.695465087890625e-05,
      "step": 7066,
      "training_step_time": 9.933801412582397
    },
    {
      "epoch": 2.695465087890625e-05,
      "model_forward_time": 3.135711669921875,
      "step": 7066
    },
    {
      "epoch": 2.695465087890625e-05,
      "step": 7066,
      "training_step_time": 9.902702569961548
    },
    {
      "epoch": 2.695465087890625e-05,
      "model_forward_time": 3.1359269618988037,
      "step": 7066
    },
    {
      "epoch": 2.695465087890625e-05,
      "step": 7066,
      "training_step_time": 10.18200397491455
    },
    {
      "epoch": 2.695465087890625e-05,
      "model_forward_time": 3.1356585025787354,
      "step": 7066
    },
    {
      "epoch": 2.695465087890625e-05,
      "step": 7066,
      "training_step_time": 10.32915449142456
    },
    {
      "epoch": 2.6958465576171876e-05,
      "model_forward_time": 3.145104169845581,
      "step": 7067
    },
    {
      "epoch": 2.6958465576171876e-05,
      "step": 7067,
      "training_step_time": 9.915819883346558
    },
    {
      "epoch": 2.6958465576171876e-05,
      "model_forward_time": 3.1392834186553955,
      "step": 7067
    },
    {
      "epoch": 2.6958465576171876e-05,
      "step": 7067,
      "training_step_time": 9.903496265411377
    },
    {
      "epoch": 2.6958465576171876e-05,
      "model_forward_time": 3.136230707168579,
      "step": 7067
    },
    {
      "dynamics_loss_avg": 0.06645897552371025,
      "epoch": 2.6958465576171876e-05,
      "step": 7067
    },
    {
      "action_loss_avg": 0.0272925210185349,
      "epoch": 2.6958465576171876e-05,
      "step": 7067
    },
    {
      "epoch": 2.6958465576171876e-05,
      "step": 7067,
      "training_step_time": 9.903123378753662
    },
    {
      "epoch": 2.6958465576171876e-05,
      "model_forward_time": 3.405043363571167,
      "step": 7067
    },
    {
      "epoch": 2.6958465576171876e-05,
      "step": 7067,
      "training_step_time": 10.294392347335815
    },
    {
      "epoch": 2.69622802734375e-05,
      "model_forward_time": 3.1676836013793945,
      "step": 7068
    },
    {
      "epoch": 2.69622802734375e-05,
      "step": 7068,
      "training_step_time": 9.935184478759766
    },
    {
      "epoch": 2.69622802734375e-05,
      "model_forward_time": 3.140425205230713,
      "step": 7068
    },
    {
      "epoch": 2.69622802734375e-05,
      "step": 7068,
      "training_step_time": 9.908194065093994
    },
    {
      "epoch": 2.69622802734375e-05,
      "model_forward_time": 3.135453939437866,
      "step": 7068
    },
    {
      "epoch": 2.69622802734375e-05,
      "step": 7068,
      "training_step_time": 9.902524471282959
    },
    {
      "epoch": 2.69622802734375e-05,
      "model_forward_time": 3.1396026611328125,
      "step": 7068
    },
    {
      "epoch": 2.69622802734375e-05,
      "step": 7068,
      "training_step_time": 10.244040250778198
    },
    {
      "epoch": 2.6966094970703125e-05,
      "model_forward_time": 3.1746816635131836,
      "step": 7069
    },
    {
      "epoch": 2.6966094970703125e-05,
      "step": 7069,
      "training_step_time": 10.207322359085083
    },
    {
      "epoch": 2.6966094970703125e-05,
      "model_forward_time": 3.140644073486328,
      "step": 7069
    },
    {
      "epoch": 2.6966094970703125e-05,
      "step": 7069,
      "training_step_time": 9.914968967437744
    },
    {
      "epoch": 2.6966094970703125e-05,
      "model_forward_time": 3.1415300369262695,
      "step": 7069
    },
    {
      "epoch": 2.6966094970703125e-05,
      "step": 7069,
      "training_step_time": 9.91125750541687
    },
    {
      "epoch": 2.6966094970703125e-05,
      "model_forward_time": 3.1350064277648926,
      "step": 7069
    },
    {
      "epoch": 2.6966094970703125e-05,
      "step": 7069,
      "training_step_time": 10.011610984802246
    },
    {
      "epoch": 2.696990966796875e-05,
      "grad_norm": 0.3545319139957428,
      "learning_rate": 1e-05,
      "loss": 0.0832,
      "step": 7070
    },
    {
      "epoch": 2.696990966796875e-05,
      "model_forward_time": 3.167386770248413,
      "step": 7070
    },
    {
      "dynamics_loss_avg": 0.06524397470057011,
      "epoch": 2.696990966796875e-05,
      "step": 7070
    },
    {
      "action_loss_avg": 0.027846572734415532,
      "epoch": 2.696990966796875e-05,
      "step": 7070
    },
    {
      "epoch": 2.696990966796875e-05,
      "step": 7070,
      "training_step_time": 9.942111253738403
    },
    {
      "epoch": 2.696990966796875e-05,
      "model_forward_time": 3.4196677207946777,
      "step": 7070
    },
    {
      "epoch": 2.696990966796875e-05,
      "step": 7070,
      "training_step_time": 10.186605215072632
    },
    {
      "epoch": 2.696990966796875e-05,
      "model_forward_time": 3.139880895614624,
      "step": 7070
    },
    {
      "epoch": 2.696990966796875e-05,
      "step": 7070,
      "training_step_time": 9.916100025177002
    },
    {
      "epoch": 2.696990966796875e-05,
      "model_forward_time": 3.145416736602783,
      "step": 7070
    },
    {
      "epoch": 2.696990966796875e-05,
      "step": 7070,
      "training_step_time": 10.288483381271362
    },
    {
      "epoch": 2.6973724365234374e-05,
      "model_forward_time": 3.1720151901245117,
      "step": 7071
    },
    {
      "epoch": 2.6973724365234374e-05,
      "step": 7071,
      "training_step_time": 9.946665525436401
    },
    {
      "epoch": 2.6973724365234374e-05,
      "model_forward_time": 3.133518934249878,
      "step": 7071
    },
    {
      "epoch": 2.6973724365234374e-05,
      "step": 7071,
      "training_step_time": 9.902709007263184
    },
    {
      "epoch": 2.6973724365234374e-05,
      "model_forward_time": 3.393296957015991,
      "step": 7071
    },
    {
      "epoch": 2.6973724365234374e-05,
      "step": 7071,
      "training_step_time": 10.159324169158936
    },
    {
      "epoch": 2.6973724365234374e-05,
      "model_forward_time": 3.139101266860962,
      "step": 7071
    },
    {
      "epoch": 2.6973724365234374e-05,
      "step": 7071,
      "training_step_time": 10.307071924209595
    },
    {
      "epoch": 2.69775390625e-05,
      "model_forward_time": 3.1846208572387695,
      "step": 7072
    },
    {
      "epoch": 2.69775390625e-05,
      "step": 7072,
      "training_step_time": 9.953414678573608
    },
    {
      "epoch": 2.69775390625e-05,
      "model_forward_time": 3.131321430206299,
      "step": 7072
    },
    {
      "epoch": 2.69775390625e-05,
      "step": 7072,
      "training_step_time": 9.897728443145752
    },
    {
      "epoch": 2.69775390625e-05,
      "model_forward_time": 3.13214111328125,
      "step": 7072
    },
    {
      "dynamics_loss_avg": 0.051403725147247316,
      "epoch": 2.69775390625e-05,
      "step": 7072
    },
    {
      "action_loss_avg": 0.016391583997756243,
      "epoch": 2.69775390625e-05,
      "step": 7072
    },
    {
      "epoch": 2.69775390625e-05,
      "step": 7072,
      "training_step_time": 9.902710437774658
    },
    {
      "epoch": 2.69775390625e-05,
      "model_forward_time": 3.4153873920440674,
      "step": 7072
    },
    {
      "epoch": 2.69775390625e-05,
      "step": 7072,
      "training_step_time": 10.308699369430542
    },
    {
      "epoch": 2.6981353759765627e-05,
      "model_forward_time": 3.1723709106445312,
      "step": 7073
    },
    {
      "epoch": 2.6981353759765627e-05,
      "step": 7073,
      "training_step_time": 9.941774606704712
    },
    {
      "epoch": 2.6981353759765627e-05,
      "model_forward_time": 3.1302096843719482,
      "step": 7073
    },
    {
      "epoch": 2.6981353759765627e-05,
      "step": 7073,
      "training_step_time": 9.899101257324219
    },
    {
      "epoch": 2.6981353759765627e-05,
      "model_forward_time": 3.1505019664764404,
      "step": 7073
    },
    {
      "epoch": 2.6981353759765627e-05,
      "step": 7073,
      "training_step_time": 9.920085668563843
    },
    {
      "epoch": 2.6981353759765627e-05,
      "model_forward_time": 3.1336560249328613,
      "step": 7073
    },
    {
      "epoch": 2.6981353759765627e-05,
      "step": 7073,
      "training_step_time": 10.272854566574097
    },
    {
      "epoch": 2.698516845703125e-05,
      "model_forward_time": 3.478135824203491,
      "step": 7074
    },
    {
      "epoch": 2.698516845703125e-05,
      "step": 7074,
      "training_step_time": 10.244846820831299
    },
    {
      "epoch": 2.698516845703125e-05,
      "model_forward_time": 3.1339712142944336,
      "step": 7074
    },
    {
      "epoch": 2.698516845703125e-05,
      "step": 7074,
      "training_step_time": 9.906801462173462
    },
    {
      "epoch": 2.698516845703125e-05,
      "model_forward_time": 3.1371169090270996,
      "step": 7074
    },
    {
      "epoch": 2.698516845703125e-05,
      "step": 7074,
      "training_step_time": 9.911853313446045
    },
    {
      "epoch": 2.698516845703125e-05,
      "model_forward_time": 3.1384549140930176,
      "step": 7074
    },
    {
      "epoch": 2.698516845703125e-05,
      "step": 7074,
      "training_step_time": 10.040266275405884
    },
    {
      "epoch": 2.6988983154296876e-05,
      "model_forward_time": 3.156554937362671,
      "step": 7075
    },
    {
      "dynamics_loss_avg": 0.06792650874704123,
      "epoch": 2.6988983154296876e-05,
      "step": 7075
    },
    {
      "action_loss_avg": 0.025175606831908226,
      "epoch": 2.6988983154296876e-05,
      "step": 7075
    },
    {
      "epoch": 2.6988983154296876e-05,
      "step": 7075,
      "training_step_time": 9.93492078781128
    },
    {
      "epoch": 2.6988983154296876e-05,
      "model_forward_time": 3.4028687477111816,
      "step": 7075
    },
    {
      "epoch": 2.6988983154296876e-05,
      "step": 7075,
      "training_step_time": 10.167088985443115
    },
    {
      "epoch": 2.6988983154296876e-05,
      "model_forward_time": 3.1389529705047607,
      "step": 7075
    },
    {
      "epoch": 2.6988983154296876e-05,
      "step": 7075,
      "training_step_time": 9.906023740768433
    },
    {
      "epoch": 2.6988983154296876e-05,
      "model_forward_time": 3.1336116790771484,
      "step": 7075
    },
    {
      "epoch": 2.6988983154296876e-05,
      "step": 7075,
      "training_step_time": 10.26159954071045
    },
    {
      "epoch": 2.69927978515625e-05,
      "model_forward_time": 3.162435293197632,
      "step": 7076
    },
    {
      "epoch": 2.69927978515625e-05,
      "step": 7076,
      "training_step_time": 9.929304122924805
    },
    {
      "epoch": 2.69927978515625e-05,
      "model_forward_time": 3.1388514041900635,
      "step": 7076
    },
    {
      "epoch": 2.69927978515625e-05,
      "step": 7076,
      "training_step_time": 9.910419940948486
    },
    {
      "epoch": 2.69927978515625e-05,
      "model_forward_time": 3.4072182178497314,
      "step": 7076
    },
    {
      "epoch": 2.69927978515625e-05,
      "step": 7076,
      "training_step_time": 10.178760290145874
    },
    {
      "epoch": 2.69927978515625e-05,
      "model_forward_time": 3.133800506591797,
      "step": 7076
    },
    {
      "epoch": 2.69927978515625e-05,
      "step": 7076,
      "training_step_time": 10.316779613494873
    },
    {
      "epoch": 2.6996612548828125e-05,
      "model_forward_time": 3.149284839630127,
      "step": 7077
    },
    {
      "epoch": 2.6996612548828125e-05,
      "step": 7077,
      "training_step_time": 9.920908451080322
    },
    {
      "epoch": 2.6996612548828125e-05,
      "model_forward_time": 3.1344690322875977,
      "step": 7077
    },
    {
      "epoch": 2.6996612548828125e-05,
      "step": 7077,
      "training_step_time": 9.902329206466675
    },
    {
      "epoch": 2.6996612548828125e-05,
      "model_forward_time": 3.132960319519043,
      "step": 7077
    },
    {
      "dynamics_loss_avg": 0.06116869561374187,
      "epoch": 2.6996612548828125e-05,
      "step": 7077
    },
    {
      "action_loss_avg": 0.021265834849327803,
      "epoch": 2.6996612548828125e-05,
      "step": 7077
    },
    {
      "epoch": 2.6996612548828125e-05,
      "step": 7077,
      "training_step_time": 9.908823728561401
    },
    {
      "epoch": 2.6996612548828125e-05,
      "model_forward_time": 3.3961997032165527,
      "step": 7077
    },
    {
      "epoch": 2.6996612548828125e-05,
      "step": 7077,
      "training_step_time": 10.282710552215576
    },
    {
      "epoch": 2.700042724609375e-05,
      "model_forward_time": 3.1777431964874268,
      "step": 7078
    },
    {
      "epoch": 2.700042724609375e-05,
      "step": 7078,
      "training_step_time": 9.944734334945679
    },
    {
      "epoch": 2.700042724609375e-05,
      "model_forward_time": 3.135438919067383,
      "step": 7078
    },
    {
      "epoch": 2.700042724609375e-05,
      "step": 7078,
      "training_step_time": 9.905998229980469
    },
    {
      "epoch": 2.700042724609375e-05,
      "model_forward_time": 3.1343166828155518,
      "step": 7078
    },
    {
      "epoch": 2.700042724609375e-05,
      "step": 7078,
      "training_step_time": 9.903481721878052
    },
    {
      "epoch": 2.700042724609375e-05,
      "model_forward_time": 3.140366792678833,
      "step": 7078
    },
    {
      "epoch": 2.700042724609375e-05,
      "step": 7078,
      "training_step_time": 10.300620555877686
    },
    {
      "epoch": 2.7004241943359374e-05,
      "model_forward_time": 3.4421448707580566,
      "step": 7079
    },
    {
      "epoch": 2.7004241943359374e-05,
      "step": 7079,
      "training_step_time": 10.218355178833008
    },
    {
      "epoch": 2.7004241943359374e-05,
      "model_forward_time": 3.1455423831939697,
      "step": 7079
    },
    {
      "epoch": 2.7004241943359374e-05,
      "step": 7079,
      "training_step_time": 9.915446043014526
    },
    {
      "epoch": 2.7004241943359374e-05,
      "model_forward_time": 3.1464107036590576,
      "step": 7079
    },
    {
      "epoch": 2.7004241943359374e-05,
      "step": 7079,
      "training_step_time": 9.919052839279175
    },
    {
      "epoch": 2.7004241943359374e-05,
      "model_forward_time": 3.150628089904785,
      "step": 7079
    },
    {
      "epoch": 2.7004241943359374e-05,
      "step": 7079,
      "training_step_time": 10.031125783920288
    },
    {
      "epoch": 2.7008056640625e-05,
      "grad_norm": 0.37927132844924927,
      "learning_rate": 1e-05,
      "loss": 0.0836,
      "step": 7080
    },
    {
      "epoch": 2.7008056640625e-05,
      "model_forward_time": 3.1606178283691406,
      "step": 7080
    },
    {
      "dynamics_loss_avg": 0.06524788700044155,
      "epoch": 2.7008056640625e-05,
      "step": 7080
    },
    {
      "action_loss_avg": 0.022296778578311204,
      "epoch": 2.7008056640625e-05,
      "step": 7080
    },
    {
      "epoch": 2.7008056640625e-05,
      "step": 7080,
      "training_step_time": 9.935276508331299
    },
    {
      "epoch": 2.7008056640625e-05,
      "model_forward_time": 3.4287757873535156,
      "step": 7080
    },
    {
      "epoch": 2.7008056640625e-05,
      "step": 7080,
      "training_step_time": 10.196184873580933
    },
    {
      "epoch": 2.7008056640625e-05,
      "model_forward_time": 3.1405768394470215,
      "step": 7080
    },
    {
      "epoch": 2.7008056640625e-05,
      "step": 7080,
      "training_step_time": 9.910274982452393
    },
    {
      "epoch": 2.7008056640625e-05,
      "model_forward_time": 3.141563653945923,
      "step": 7080
    },
    {
      "epoch": 2.7008056640625e-05,
      "step": 7080,
      "training_step_time": 10.273296594619751
    },
    {
      "epoch": 2.7011871337890627e-05,
      "model_forward_time": 3.1797966957092285,
      "step": 7081
    },
    {
      "epoch": 2.7011871337890627e-05,
      "step": 7081,
      "training_step_time": 9.946325063705444
    },
    {
      "epoch": 2.7011871337890627e-05,
      "model_forward_time": 3.1421585083007812,
      "step": 7081
    },
    {
      "epoch": 2.7011871337890627e-05,
      "step": 7081,
      "training_step_time": 9.908611536026001
    },
    {
      "epoch": 2.7011871337890627e-05,
      "model_forward_time": 3.4028353691101074,
      "step": 7081
    },
    {
      "epoch": 2.7011871337890627e-05,
      "step": 7081,
      "training_step_time": 10.168252944946289
    },
    {
      "epoch": 2.7011871337890627e-05,
      "model_forward_time": 3.142073392868042,
      "step": 7081
    },
    {
      "epoch": 2.7011871337890627e-05,
      "step": 7081,
      "training_step_time": 10.297705173492432
    },
    {
      "epoch": 2.701568603515625e-05,
      "model_forward_time": 3.164185047149658,
      "step": 7082
    },
    {
      "epoch": 2.701568603515625e-05,
      "step": 7082,
      "training_step_time": 9.929599523544312
    },
    {
      "epoch": 2.701568603515625e-05,
      "model_forward_time": 3.1428029537200928,
      "step": 7082
    },
    {
      "epoch": 2.701568603515625e-05,
      "step": 7082,
      "training_step_time": 9.916020393371582
    },
    {
      "epoch": 2.701568603515625e-05,
      "model_forward_time": 3.146705389022827,
      "step": 7082
    },
    {
      "dynamics_loss_avg": 0.042874593194574116,
      "epoch": 2.701568603515625e-05,
      "step": 7082
    },
    {
      "action_loss_avg": 0.015328806499019266,
      "epoch": 2.701568603515625e-05,
      "step": 7082
    },
    {
      "epoch": 2.701568603515625e-05,
      "step": 7082,
      "training_step_time": 9.919783592224121
    },
    {
      "epoch": 2.701568603515625e-05,
      "model_forward_time": 2.4405300617218018,
      "step": 7082
    },
    {
      "epoch": 2.701568603515625e-05,
      "step": 7082,
      "training_step_time": 10.348309755325317
    },
    {
      "epoch": 2.7019500732421876e-05,
      "model_forward_time": 3.1861398220062256,
      "step": 7083
    },
    {
      "epoch": 2.7019500732421876e-05,
      "step": 7083,
      "training_step_time": 9.960441827774048
    },
    {
      "epoch": 2.7019500732421876e-05,
      "model_forward_time": 3.1492838859558105,
      "step": 7083
    },
    {
      "epoch": 2.7019500732421876e-05,
      "step": 7083,
      "training_step_time": 9.925452470779419
    },
    {
      "epoch": 2.7019500732421876e-05,
      "model_forward_time": 3.1489498615264893,
      "step": 7083
    },
    {
      "epoch": 2.7019500732421876e-05,
      "step": 7083,
      "training_step_time": 9.921581745147705
    },
    {
      "epoch": 2.7019500732421876e-05,
      "model_forward_time": 3.1400489807128906,
      "step": 7083
    },
    {
      "epoch": 2.7019500732421876e-05,
      "step": 7083,
      "training_step_time": 10.29765248298645
    },
    {
      "epoch": 2.70233154296875e-05,
      "model_forward_time": 3.1815600395202637,
      "step": 7084
    },
    {
      "epoch": 2.70233154296875e-05,
      "step": 7084,
      "training_step_time": 10.241376399993896
    },
    {
      "epoch": 2.70233154296875e-05,
      "model_forward_time": 3.142953872680664,
      "step": 7084
    },
    {
      "epoch": 2.70233154296875e-05,
      "step": 7084,
      "training_step_time": 9.918931245803833
    },
    {
      "epoch": 2.70233154296875e-05,
      "model_forward_time": 3.1483545303344727,
      "step": 7084
    },
    {
      "epoch": 2.70233154296875e-05,
      "step": 7084,
      "training_step_time": 9.923596858978271
    },
    {
      "epoch": 2.70233154296875e-05,
      "model_forward_time": 3.1428682804107666,
      "step": 7084
    },
    {
      "epoch": 2.70233154296875e-05,
      "step": 7084,
      "training_step_time": 10.285545349121094
    },
    {
      "epoch": 2.7027130126953125e-05,
      "model_forward_time": 3.1696348190307617,
      "step": 7085
    },
    {
      "dynamics_loss_avg": 0.05557698104530573,
      "epoch": 2.7027130126953125e-05,
      "step": 7085
    },
    {
      "action_loss_avg": 0.021418539388105274,
      "epoch": 2.7027130126953125e-05,
      "step": 7085
    },
    {
      "epoch": 2.7027130126953125e-05,
      "step": 7085,
      "training_step_time": 9.95540452003479
    },
    {
      "epoch": 2.7027130126953125e-05,
      "model_forward_time": 3.401327133178711,
      "step": 7085
    },
    {
      "epoch": 2.7027130126953125e-05,
      "step": 7085,
      "training_step_time": 10.170050382614136
    },
    {
      "epoch": 2.7027130126953125e-05,
      "model_forward_time": 3.1412222385406494,
      "step": 7085
    },
    {
      "epoch": 2.7027130126953125e-05,
      "step": 7085,
      "training_step_time": 9.909656286239624
    },
    {
      "epoch": 2.7027130126953125e-05,
      "model_forward_time": 3.1369051933288574,
      "step": 7085
    },
    {
      "epoch": 2.7027130126953125e-05,
      "step": 7085,
      "training_step_time": 10.282698631286621
    },
    {
      "epoch": 2.703094482421875e-05,
      "model_forward_time": 3.177471876144409,
      "step": 7086
    },
    {
      "epoch": 2.703094482421875e-05,
      "step": 7086,
      "training_step_time": 9.950852155685425
    },
    {
      "epoch": 2.703094482421875e-05,
      "model_forward_time": 3.1328437328338623,
      "step": 7086
    },
    {
      "epoch": 2.703094482421875e-05,
      "step": 7086,
      "training_step_time": 9.902242422103882
    },
    {
      "epoch": 2.703094482421875e-05,
      "model_forward_time": 3.135730028152466,
      "step": 7086
    },
    {
      "epoch": 2.703094482421875e-05,
      "step": 7086,
      "training_step_time": 10.17970061302185
    },
    {
      "epoch": 2.703094482421875e-05,
      "model_forward_time": 3.1406593322753906,
      "step": 7086
    },
    {
      "epoch": 2.703094482421875e-05,
      "step": 7086,
      "training_step_time": 10.285696268081665
    },
    {
      "epoch": 2.7034759521484374e-05,
      "model_forward_time": 3.177600622177124,
      "step": 7087
    },
    {
      "epoch": 2.7034759521484374e-05,
      "step": 7087,
      "training_step_time": 9.951247215270996
    },
    {
      "epoch": 2.7034759521484374e-05,
      "model_forward_time": 3.1449384689331055,
      "step": 7087
    },
    {
      "epoch": 2.7034759521484374e-05,
      "step": 7087,
      "training_step_time": 9.916472911834717
    },
    {
      "epoch": 2.7034759521484374e-05,
      "model_forward_time": 3.134593963623047,
      "step": 7087
    },
    {
      "dynamics_loss_avg": 0.05286492398008704,
      "epoch": 2.7034759521484374e-05,
      "step": 7087
    },
    {
      "action_loss_avg": 0.01898873974569142,
      "epoch": 2.7034759521484374e-05,
      "step": 7087
    },
    {
      "epoch": 2.7034759521484374e-05,
      "step": 7087,
      "training_step_time": 9.900944948196411
    },
    {
      "epoch": 2.7034759521484374e-05,
      "model_forward_time": 3.4010910987854004,
      "step": 7087
    },
    {
      "epoch": 2.7034759521484374e-05,
      "step": 7087,
      "training_step_time": 10.279709100723267
    },
    {
      "epoch": 2.703857421875e-05,
      "model_forward_time": 3.1707966327667236,
      "step": 7088
    },
    {
      "epoch": 2.703857421875e-05,
      "step": 7088,
      "training_step_time": 9.9362633228302
    },
    {
      "epoch": 2.703857421875e-05,
      "model_forward_time": 3.1373050212860107,
      "step": 7088
    },
    {
      "epoch": 2.703857421875e-05,
      "step": 7088,
      "training_step_time": 9.901220083236694
    },
    {
      "epoch": 2.703857421875e-05,
      "model_forward_time": 3.1347694396972656,
      "step": 7088
    },
    {
      "epoch": 2.703857421875e-05,
      "step": 7088,
      "training_step_time": 9.899587631225586
    },
    {
      "epoch": 2.703857421875e-05,
      "model_forward_time": 3.1379098892211914,
      "step": 7088
    },
    {
      "epoch": 2.703857421875e-05,
      "step": 7088,
      "training_step_time": 10.024083614349365
    },
    {
      "epoch": 2.7042388916015627e-05,
      "model_forward_time": 3.1724226474761963,
      "step": 7089
    },
    {
      "epoch": 2.7042388916015627e-05,
      "step": 7089,
      "training_step_time": 10.230852842330933
    },
    {
      "epoch": 2.7042388916015627e-05,
      "model_forward_time": 3.1398305892944336,
      "step": 7089
    },
    {
      "epoch": 2.7042388916015627e-05,
      "step": 7089,
      "training_step_time": 9.903803825378418
    },
    {
      "epoch": 2.7042388916015627e-05,
      "model_forward_time": 3.135223627090454,
      "step": 7089
    },
    {
      "epoch": 2.7042388916015627e-05,
      "step": 7089,
      "training_step_time": 9.903000593185425
    },
    {
      "epoch": 2.7042388916015627e-05,
      "model_forward_time": 3.1421380043029785,
      "step": 7089
    },
    {
      "epoch": 2.7042388916015627e-05,
      "step": 7089,
      "training_step_time": 10.291098833084106
    },
    {
      "epoch": 2.704620361328125e-05,
      "grad_norm": 0.5233691930770874,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 7090
    },
    {
      "epoch": 2.704620361328125e-05,
      "model_forward_time": 3.1783690452575684,
      "step": 7090
    },
    {
      "dynamics_loss_avg": 0.06018208619207144,
      "epoch": 2.704620361328125e-05,
      "step": 7090
    },
    {
      "action_loss_avg": 0.02196465227752924,
      "epoch": 2.704620361328125e-05,
      "step": 7090
    },
    {
      "epoch": 2.704620361328125e-05,
      "step": 7090,
      "training_step_time": 9.954059600830078
    },
    {
      "epoch": 2.704620361328125e-05,
      "model_forward_time": 3.4391591548919678,
      "step": 7090
    },
    {
      "epoch": 2.704620361328125e-05,
      "step": 7090,
      "training_step_time": 10.213197946548462
    },
    {
      "epoch": 2.704620361328125e-05,
      "model_forward_time": 3.137849807739258,
      "step": 7090
    },
    {
      "epoch": 2.704620361328125e-05,
      "step": 7090,
      "training_step_time": 9.90607500076294
    },
    {
      "epoch": 2.704620361328125e-05,
      "model_forward_time": 3.137573480606079,
      "step": 7090
    },
    {
      "epoch": 2.704620361328125e-05,
      "step": 7090,
      "training_step_time": 10.016764402389526
    },
    {
      "epoch": 2.7050018310546876e-05,
      "model_forward_time": 3.16688871383667,
      "step": 7091
    },
    {
      "epoch": 2.7050018310546876e-05,
      "step": 7091,
      "training_step_time": 9.932716131210327
    },
    {
      "epoch": 2.7050018310546876e-05,
      "model_forward_time": 3.1330819129943848,
      "step": 7091
    },
    {
      "epoch": 2.7050018310546876e-05,
      "step": 7091,
      "training_step_time": 9.891187906265259
    },
    {
      "epoch": 2.7050018310546876e-05,
      "model_forward_time": 3.3893072605133057,
      "step": 7091
    },
    {
      "epoch": 2.7050018310546876e-05,
      "step": 7091,
      "training_step_time": 10.150758266448975
    },
    {
      "epoch": 2.7050018310546876e-05,
      "model_forward_time": 3.1368978023529053,
      "step": 7091
    },
    {
      "epoch": 2.7050018310546876e-05,
      "step": 7091,
      "training_step_time": 10.300498008728027
    },
    {
      "epoch": 2.70538330078125e-05,
      "model_forward_time": 3.1771023273468018,
      "step": 7092
    },
    {
      "epoch": 2.70538330078125e-05,
      "step": 7092,
      "training_step_time": 9.940903186798096
    },
    {
      "epoch": 2.70538330078125e-05,
      "model_forward_time": 3.1324007511138916,
      "step": 7092
    },
    {
      "epoch": 2.70538330078125e-05,
      "step": 7092,
      "training_step_time": 9.892470121383667
    },
    {
      "epoch": 2.70538330078125e-05,
      "model_forward_time": 3.1334455013275146,
      "step": 7092
    },
    {
      "dynamics_loss_avg": 0.06654550544917584,
      "epoch": 2.70538330078125e-05,
      "step": 7092
    },
    {
      "action_loss_avg": 0.023341993149369956,
      "epoch": 2.70538330078125e-05,
      "step": 7092
    },
    {
      "epoch": 2.70538330078125e-05,
      "step": 7092,
      "training_step_time": 9.90118408203125
    },
    {
      "epoch": 2.70538330078125e-05,
      "model_forward_time": 3.4129233360290527,
      "step": 7092
    },
    {
      "epoch": 2.70538330078125e-05,
      "step": 7092,
      "training_step_time": 10.273901462554932
    },
    {
      "epoch": 2.7057647705078125e-05,
      "model_forward_time": 3.1482906341552734,
      "step": 7093
    },
    {
      "epoch": 2.7057647705078125e-05,
      "step": 7093,
      "training_step_time": 9.92090654373169
    },
    {
      "epoch": 2.7057647705078125e-05,
      "model_forward_time": 3.1365950107574463,
      "step": 7093
    },
    {
      "epoch": 2.7057647705078125e-05,
      "step": 7093,
      "training_step_time": 9.906985998153687
    },
    {
      "epoch": 2.7057647705078125e-05,
      "model_forward_time": 3.1424171924591064,
      "step": 7093
    },
    {
      "epoch": 2.7057647705078125e-05,
      "step": 7093,
      "training_step_time": 9.911311388015747
    },
    {
      "epoch": 2.7057647705078125e-05,
      "model_forward_time": 3.1384005546569824,
      "step": 7093
    },
    {
      "epoch": 2.7057647705078125e-05,
      "step": 7093,
      "training_step_time": 10.308416366577148
    },
    {
      "epoch": 2.706146240234375e-05,
      "model_forward_time": 3.420515537261963,
      "step": 7094
    },
    {
      "epoch": 2.706146240234375e-05,
      "step": 7094,
      "training_step_time": 10.193162202835083
    },
    {
      "epoch": 2.706146240234375e-05,
      "model_forward_time": 3.137777328491211,
      "step": 7094
    },
    {
      "epoch": 2.706146240234375e-05,
      "step": 7094,
      "training_step_time": 9.905685901641846
    },
    {
      "epoch": 2.706146240234375e-05,
      "model_forward_time": 3.1412806510925293,
      "step": 7094
    },
    {
      "epoch": 2.706146240234375e-05,
      "step": 7094,
      "training_step_time": 9.918189525604248
    },
    {
      "epoch": 2.706146240234375e-05,
      "model_forward_time": 3.133715867996216,
      "step": 7094
    },
    {
      "epoch": 2.706146240234375e-05,
      "step": 7094,
      "training_step_time": 10.239213466644287
    },
    {
      "epoch": 2.7065277099609374e-05,
      "model_forward_time": 3.1527249813079834,
      "step": 7095
    },
    {
      "dynamics_loss_avg": 0.05076429322361946,
      "epoch": 2.7065277099609374e-05,
      "step": 7095
    },
    {
      "action_loss_avg": 0.02477112179622054,
      "epoch": 2.7065277099609374e-05,
      "step": 7095
    },
    {
      "epoch": 2.7065277099609374e-05,
      "step": 7095,
      "training_step_time": 9.927386999130249
    },
    {
      "epoch": 2.7065277099609374e-05,
      "model_forward_time": 3.3973031044006348,
      "step": 7095
    },
    {
      "epoch": 2.7065277099609374e-05,
      "step": 7095,
      "training_step_time": 10.169752836227417
    },
    {
      "epoch": 2.7065277099609374e-05,
      "model_forward_time": 3.138301372528076,
      "step": 7095
    },
    {
      "epoch": 2.7065277099609374e-05,
      "step": 7095,
      "training_step_time": 9.911450386047363
    },
    {
      "epoch": 2.7065277099609374e-05,
      "model_forward_time": 3.134986400604248,
      "step": 7095
    },
    {
      "epoch": 2.7065277099609374e-05,
      "step": 7095,
      "training_step_time": 10.314653873443604
    },
    {
      "epoch": 2.7069091796875e-05,
      "model_forward_time": 3.1476330757141113,
      "step": 7096
    },
    {
      "epoch": 2.7069091796875e-05,
      "step": 7096,
      "training_step_time": 9.928775072097778
    },
    {
      "epoch": 2.7069091796875e-05,
      "model_forward_time": 3.1442511081695557,
      "step": 7096
    },
    {
      "epoch": 2.7069091796875e-05,
      "step": 7096,
      "training_step_time": 9.921880722045898
    },
    {
      "epoch": 2.7069091796875e-05,
      "model_forward_time": 3.4198153018951416,
      "step": 7096
    },
    {
      "epoch": 2.7069091796875e-05,
      "step": 7096,
      "training_step_time": 10.19761347770691
    },
    {
      "epoch": 2.7069091796875e-05,
      "model_forward_time": 3.142296075820923,
      "step": 7096
    },
    {
      "epoch": 2.7069091796875e-05,
      "step": 7096,
      "training_step_time": 10.292872428894043
    },
    {
      "epoch": 2.7072906494140627e-05,
      "model_forward_time": 3.164036750793457,
      "step": 7097
    },
    {
      "epoch": 2.7072906494140627e-05,
      "step": 7097,
      "training_step_time": 9.941649913787842
    },
    {
      "epoch": 2.7072906494140627e-05,
      "model_forward_time": 3.146787405014038,
      "step": 7097
    },
    {
      "epoch": 2.7072906494140627e-05,
      "step": 7097,
      "training_step_time": 9.933957815170288
    },
    {
      "epoch": 2.7072906494140627e-05,
      "model_forward_time": 3.1396138668060303,
      "step": 7097
    },
    {
      "dynamics_loss_avg": 0.06599908843636512,
      "epoch": 2.7072906494140627e-05,
      "step": 7097
    },
    {
      "action_loss_avg": 0.022680287715047598,
      "epoch": 2.7072906494140627e-05,
      "step": 7097
    },
    {
      "epoch": 2.7072906494140627e-05,
      "step": 7097,
      "training_step_time": 9.91958498954773
    },
    {
      "epoch": 2.7072906494140627e-05,
      "model_forward_time": 3.402139902114868,
      "step": 7097
    },
    {
      "epoch": 2.7072906494140627e-05,
      "step": 7097,
      "training_step_time": 10.274364233016968
    },
    {
      "epoch": 2.707672119140625e-05,
      "model_forward_time": 3.173891067504883,
      "step": 7098
    },
    {
      "epoch": 2.707672119140625e-05,
      "step": 7098,
      "training_step_time": 9.954570770263672
    },
    {
      "epoch": 2.707672119140625e-05,
      "model_forward_time": 3.1456921100616455,
      "step": 7098
    },
    {
      "epoch": 2.707672119140625e-05,
      "step": 7098,
      "training_step_time": 9.922791957855225
    },
    {
      "epoch": 2.707672119140625e-05,
      "model_forward_time": 3.1379635334014893,
      "step": 7098
    },
    {
      "epoch": 2.707672119140625e-05,
      "step": 7098,
      "training_step_time": 9.922001361846924
    },
    {
      "epoch": 2.707672119140625e-05,
      "model_forward_time": 3.149348497390747,
      "step": 7098
    },
    {
      "epoch": 2.707672119140625e-05,
      "step": 7098,
      "training_step_time": 10.308303117752075
    },
    {
      "epoch": 2.7080535888671876e-05,
      "model_forward_time": 3.444125175476074,
      "step": 7099
    },
    {
      "epoch": 2.7080535888671876e-05,
      "step": 7099,
      "training_step_time": 10.21455979347229
    },
    {
      "epoch": 2.7080535888671876e-05,
      "model_forward_time": 3.144232749938965,
      "step": 7099
    },
    {
      "epoch": 2.7080535888671876e-05,
      "step": 7099,
      "training_step_time": 9.916298627853394
    },
    {
      "epoch": 2.7080535888671876e-05,
      "model_forward_time": 3.137454032897949,
      "step": 7099
    },
    {
      "epoch": 2.7080535888671876e-05,
      "step": 7099,
      "training_step_time": 9.906402111053467
    },
    {
      "epoch": 2.7080535888671876e-05,
      "model_forward_time": 3.136711359024048,
      "step": 7099
    },
    {
      "epoch": 2.7080535888671876e-05,
      "step": 7099,
      "training_step_time": 10.28958511352539
    },
    {
      "epoch": 2.70843505859375e-05,
      "grad_norm": 0.28469353914260864,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 7100
    },
    {
      "epoch": 2.70843505859375e-05,
      "model_forward_time": 3.1730782985687256,
      "step": 7100
    },
    {
      "dynamics_loss_avg": 0.05525343231856823,
      "epoch": 2.70843505859375e-05,
      "step": 7100
    },
    {
      "action_loss_avg": 0.019197099190205336,
      "epoch": 2.70843505859375e-05,
      "step": 7100
    },
    {
      "epoch": 2.70843505859375e-05,
      "step": 7100,
      "training_step_time": 9.954659700393677
    },
    {
      "epoch": 2.70843505859375e-05,
      "model_forward_time": 3.414865016937256,
      "step": 7100
    },
    {
      "epoch": 2.70843505859375e-05,
      "step": 7100,
      "training_step_time": 10.185960531234741
    },
    {
      "epoch": 2.70843505859375e-05,
      "model_forward_time": 3.1359145641326904,
      "step": 7100
    },
    {
      "epoch": 2.70843505859375e-05,
      "step": 7100,
      "training_step_time": 9.91156554222107
    },
    {
      "epoch": 2.70843505859375e-05,
      "model_forward_time": 3.14662504196167,
      "step": 7100
    },
    {
      "epoch": 2.70843505859375e-05,
      "step": 7100,
      "training_step_time": 10.297651767730713
    },
    {
      "epoch": 2.7088165283203125e-05,
      "model_forward_time": 3.178260564804077,
      "step": 7101
    },
    {
      "epoch": 2.7088165283203125e-05,
      "step": 7101,
      "training_step_time": 9.957849740982056
    },
    {
      "epoch": 2.7088165283203125e-05,
      "model_forward_time": 3.1416714191436768,
      "step": 7101
    },
    {
      "epoch": 2.7088165283203125e-05,
      "step": 7101,
      "training_step_time": 9.914536714553833
    },
    {
      "epoch": 2.7088165283203125e-05,
      "model_forward_time": 3.398934841156006,
      "step": 7101
    },
    {
      "epoch": 2.7088165283203125e-05,
      "step": 7101,
      "training_step_time": 10.173544883728027
    },
    {
      "epoch": 2.7088165283203125e-05,
      "model_forward_time": 3.1403234004974365,
      "step": 7101
    },
    {
      "epoch": 2.7088165283203125e-05,
      "step": 7101,
      "training_step_time": 10.282679319381714
    },
    {
      "epoch": 2.709197998046875e-05,
      "model_forward_time": 3.179831027984619,
      "step": 7102
    },
    {
      "epoch": 2.709197998046875e-05,
      "step": 7102,
      "training_step_time": 9.950647592544556
    },
    {
      "epoch": 2.709197998046875e-05,
      "model_forward_time": 3.140913724899292,
      "step": 7102
    },
    {
      "epoch": 2.709197998046875e-05,
      "step": 7102,
      "training_step_time": 9.907193183898926
    },
    {
      "epoch": 2.709197998046875e-05,
      "model_forward_time": 3.141535997390747,
      "step": 7102
    },
    {
      "dynamics_loss_avg": 0.0492715222761035,
      "epoch": 2.709197998046875e-05,
      "step": 7102
    },
    {
      "action_loss_avg": 0.019942962331697345,
      "epoch": 2.709197998046875e-05,
      "step": 7102
    },
    {
      "epoch": 2.709197998046875e-05,
      "step": 7102,
      "training_step_time": 9.91183614730835
    },
    {
      "epoch": 2.709197998046875e-05,
      "model_forward_time": 3.448831081390381,
      "step": 7102
    },
    {
      "epoch": 2.709197998046875e-05,
      "step": 7102,
      "training_step_time": 10.314713954925537
    },
    {
      "epoch": 2.7095794677734374e-05,
      "model_forward_time": 3.1820054054260254,
      "step": 7103
    },
    {
      "epoch": 2.7095794677734374e-05,
      "step": 7103,
      "training_step_time": 9.955852508544922
    },
    {
      "epoch": 2.7095794677734374e-05,
      "model_forward_time": 3.144015312194824,
      "step": 7103
    },
    {
      "epoch": 2.7095794677734374e-05,
      "step": 7103,
      "training_step_time": 9.91912031173706
    },
    {
      "epoch": 2.7095794677734374e-05,
      "model_forward_time": 3.146353006362915,
      "step": 7103
    },
    {
      "epoch": 2.7095794677734374e-05,
      "step": 7103,
      "training_step_time": 9.921882629394531
    },
    {
      "epoch": 2.7095794677734374e-05,
      "model_forward_time": 3.1416125297546387,
      "step": 7103
    },
    {
      "epoch": 2.7095794677734374e-05,
      "step": 7103,
      "training_step_time": 10.327864408493042
    },
    {
      "epoch": 2.7099609375e-05,
      "model_forward_time": 3.4525842666625977,
      "step": 7104
    },
    {
      "epoch": 2.7099609375e-05,
      "step": 7104,
      "training_step_time": 10.234673261642456
    },
    {
      "epoch": 2.7099609375e-05,
      "model_forward_time": 3.1467578411102295,
      "step": 7104
    },
    {
      "epoch": 2.7099609375e-05,
      "step": 7104,
      "training_step_time": 9.927227020263672
    },
    {
      "epoch": 2.7099609375e-05,
      "model_forward_time": 3.145108461380005,
      "step": 7104
    },
    {
      "epoch": 2.7099609375e-05,
      "step": 7104,
      "training_step_time": 9.922034502029419
    },
    {
      "epoch": 2.7099609375e-05,
      "model_forward_time": 3.1405441761016846,
      "step": 7104
    },
    {
      "epoch": 2.7099609375e-05,
      "step": 7104,
      "training_step_time": 10.300771236419678
    },
    {
      "epoch": 2.7103424072265627e-05,
      "model_forward_time": 3.1600375175476074,
      "step": 7105
    },
    {
      "dynamics_loss_avg": 0.04675358906388283,
      "epoch": 2.7103424072265627e-05,
      "step": 7105
    },
    {
      "action_loss_avg": 0.013976971432566643,
      "epoch": 2.7103424072265627e-05,
      "step": 7105
    },
    {
      "epoch": 2.7103424072265627e-05,
      "step": 7105,
      "training_step_time": 9.938854932785034
    },
    {
      "epoch": 2.7103424072265627e-05,
      "model_forward_time": 3.4037678241729736,
      "step": 7105
    },
    {
      "epoch": 2.7103424072265627e-05,
      "step": 7105,
      "training_step_time": 10.18306279182434
    },
    {
      "epoch": 2.7103424072265627e-05,
      "model_forward_time": 3.1468141078948975,
      "step": 7105
    },
    {
      "epoch": 2.7103424072265627e-05,
      "step": 7105,
      "training_step_time": 9.926980257034302
    },
    {
      "epoch": 2.7103424072265627e-05,
      "model_forward_time": 3.141723155975342,
      "step": 7105
    },
    {
      "epoch": 2.7103424072265627e-05,
      "step": 7105,
      "training_step_time": 10.297732830047607
    },
    {
      "epoch": 2.710723876953125e-05,
      "model_forward_time": 3.1465883255004883,
      "step": 7106
    },
    {
      "epoch": 2.710723876953125e-05,
      "step": 7106,
      "training_step_time": 9.920792818069458
    },
    {
      "epoch": 2.710723876953125e-05,
      "model_forward_time": 3.1417973041534424,
      "step": 7106
    },
    {
      "epoch": 2.710723876953125e-05,
      "step": 7106,
      "training_step_time": 9.911926984786987
    },
    {
      "epoch": 2.710723876953125e-05,
      "model_forward_time": 3.4181084632873535,
      "step": 7106
    },
    {
      "epoch": 2.710723876953125e-05,
      "step": 7106,
      "training_step_time": 10.186389207839966
    },
    {
      "epoch": 2.710723876953125e-05,
      "model_forward_time": 3.138300895690918,
      "step": 7106
    },
    {
      "epoch": 2.710723876953125e-05,
      "step": 7106,
      "training_step_time": 10.263172149658203
    },
    {
      "epoch": 2.7111053466796876e-05,
      "model_forward_time": 3.176187038421631,
      "step": 7107
    },
    {
      "epoch": 2.7111053466796876e-05,
      "step": 7107,
      "training_step_time": 9.956526517868042
    },
    {
      "epoch": 2.7111053466796876e-05,
      "model_forward_time": 3.135976791381836,
      "step": 7107
    },
    {
      "epoch": 2.7111053466796876e-05,
      "step": 7107,
      "training_step_time": 9.90660047531128
    },
    {
      "epoch": 2.7111053466796876e-05,
      "model_forward_time": 3.1314697265625,
      "step": 7107
    },
    {
      "dynamics_loss_avg": 0.07133755926042795,
      "epoch": 2.7111053466796876e-05,
      "step": 7107
    },
    {
      "action_loss_avg": 0.0245882548391819,
      "epoch": 2.7111053466796876e-05,
      "step": 7107
    },
    {
      "epoch": 2.7111053466796876e-05,
      "step": 7107,
      "training_step_time": 9.90082573890686
    },
    {
      "epoch": 2.7111053466796876e-05,
      "model_forward_time": 3.378063201904297,
      "step": 7107
    },
    {
      "epoch": 2.7111053466796876e-05,
      "step": 7107,
      "training_step_time": 10.251620054244995
    },
    {
      "epoch": 2.71148681640625e-05,
      "model_forward_time": 3.1765921115875244,
      "step": 7108
    },
    {
      "epoch": 2.71148681640625e-05,
      "step": 7108,
      "training_step_time": 9.94663143157959
    },
    {
      "epoch": 2.71148681640625e-05,
      "model_forward_time": 3.1304969787597656,
      "step": 7108
    },
    {
      "epoch": 2.71148681640625e-05,
      "step": 7108,
      "training_step_time": 9.899868249893188
    },
    {
      "epoch": 2.71148681640625e-05,
      "model_forward_time": 3.1414825916290283,
      "step": 7108
    },
    {
      "epoch": 2.71148681640625e-05,
      "step": 7108,
      "training_step_time": 9.919021844863892
    },
    {
      "epoch": 2.71148681640625e-05,
      "model_forward_time": 3.1367080211639404,
      "step": 7108
    },
    {
      "epoch": 2.71148681640625e-05,
      "step": 7108,
      "training_step_time": 10.027538537979126
    },
    {
      "epoch": 2.7118682861328125e-05,
      "model_forward_time": 3.431020975112915,
      "step": 7109
    },
    {
      "epoch": 2.7118682861328125e-05,
      "step": 7109,
      "training_step_time": 10.207202434539795
    },
    {
      "epoch": 2.7118682861328125e-05,
      "model_forward_time": 2.4341681003570557,
      "step": 7109
    },
    {
      "epoch": 2.7118682861328125e-05,
      "step": 7109,
      "training_step_time": 7.463468313217163
    },
    {
      "epoch": 2.7118682861328125e-05,
      "model_forward_time": 3.1289138793945312,
      "step": 7109
    },
    {
      "epoch": 2.7118682861328125e-05,
      "step": 7109,
      "training_step_time": 9.902481317520142
    },
    {
      "epoch": 2.7118682861328125e-05,
      "model_forward_time": 3.1420178413391113,
      "step": 7109
    },
    {
      "epoch": 2.7118682861328125e-05,
      "step": 7109,
      "training_step_time": 10.30981707572937
    },
    {
      "epoch": 2.712249755859375e-05,
      "grad_norm": 0.461375892162323,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 7110
    },
    {
      "epoch": 2.712249755859375e-05,
      "model_forward_time": 3.442444086074829,
      "step": 7110
    },
    {
      "dynamics_loss_avg": 0.05438407957553863,
      "epoch": 2.712249755859375e-05,
      "step": 7110
    },
    {
      "action_loss_avg": 0.0195356079377234,
      "epoch": 2.712249755859375e-05,
      "step": 7110
    },
    {
      "epoch": 2.712249755859375e-05,
      "step": 7110,
      "training_step_time": 10.216898441314697
    },
    {
      "epoch": 2.712249755859375e-05,
      "model_forward_time": 3.1429762840270996,
      "step": 7110
    },
    {
      "epoch": 2.712249755859375e-05,
      "step": 7110,
      "training_step_time": 9.917383909225464
    },
    {
      "epoch": 2.712249755859375e-05,
      "model_forward_time": 3.137049436569214,
      "step": 7110
    },
    {
      "epoch": 2.712249755859375e-05,
      "step": 7110,
      "training_step_time": 9.910140991210938
    },
    {
      "epoch": 2.712249755859375e-05,
      "model_forward_time": 3.1390841007232666,
      "step": 7110
    },
    {
      "epoch": 2.712249755859375e-05,
      "step": 7110,
      "training_step_time": 10.291787385940552
    },
    {
      "epoch": 2.7126312255859374e-05,
      "model_forward_time": 3.1865296363830566,
      "step": 7111
    },
    {
      "epoch": 2.7126312255859374e-05,
      "step": 7111,
      "training_step_time": 9.958576917648315
    },
    {
      "epoch": 2.7126312255859374e-05,
      "model_forward_time": 3.134582042694092,
      "step": 7111
    },
    {
      "epoch": 2.7126312255859374e-05,
      "step": 7111,
      "training_step_time": 10.188443899154663
    },
    {
      "epoch": 2.7126312255859374e-05,
      "model_forward_time": 3.1402242183685303,
      "step": 7111
    },
    {
      "epoch": 2.7126312255859374e-05,
      "step": 7111,
      "training_step_time": 9.91818642616272
    },
    {
      "epoch": 2.7126312255859374e-05,
      "model_forward_time": 3.1476633548736572,
      "step": 7111
    },
    {
      "epoch": 2.7126312255859374e-05,
      "step": 7111,
      "training_step_time": 10.294387340545654
    },
    {
      "epoch": 2.7130126953125e-05,
      "model_forward_time": 3.1819779872894287,
      "step": 7112
    },
    {
      "epoch": 2.7130126953125e-05,
      "step": 7112,
      "training_step_time": 9.966205596923828
    },
    {
      "epoch": 2.7130126953125e-05,
      "model_forward_time": 3.139439105987549,
      "step": 7112
    },
    {
      "epoch": 2.7130126953125e-05,
      "step": 7112,
      "training_step_time": 9.914966821670532
    },
    {
      "epoch": 2.7130126953125e-05,
      "model_forward_time": 3.144212245941162,
      "step": 7112
    },
    {
      "dynamics_loss_avg": 0.05617283480241895,
      "epoch": 2.7130126953125e-05,
      "step": 7112
    },
    {
      "action_loss_avg": 0.02312753889709711,
      "epoch": 2.7130126953125e-05,
      "step": 7112
    },
    {
      "epoch": 2.7130126953125e-05,
      "step": 7112,
      "training_step_time": 10.208615779876709
    },
    {
      "epoch": 2.7130126953125e-05,
      "model_forward_time": 3.151254653930664,
      "step": 7112
    },
    {
      "epoch": 2.7130126953125e-05,
      "step": 7112,
      "training_step_time": 10.281685829162598
    },
    {
      "epoch": 2.7133941650390627e-05,
      "model_forward_time": 3.176830768585205,
      "step": 7113
    },
    {
      "epoch": 2.7133941650390627e-05,
      "step": 7113,
      "training_step_time": 9.950834512710571
    },
    {
      "epoch": 2.7133941650390627e-05,
      "model_forward_time": 3.136371612548828,
      "step": 7113
    },
    {
      "epoch": 2.7133941650390627e-05,
      "step": 7113,
      "training_step_time": 9.905811548233032
    },
    {
      "epoch": 2.7133941650390627e-05,
      "model_forward_time": 3.1455814838409424,
      "step": 7113
    },
    {
      "epoch": 2.7133941650390627e-05,
      "step": 7113,
      "training_step_time": 9.917442798614502
    },
    {
      "epoch": 2.7133941650390627e-05,
      "model_forward_time": 3.4303131103515625,
      "step": 7113
    },
    {
      "epoch": 2.7133941650390627e-05,
      "step": 7113,
      "training_step_time": 10.29697036743164
    },
    {
      "epoch": 2.713775634765625e-05,
      "model_forward_time": 3.1516013145446777,
      "step": 7114
    },
    {
      "epoch": 2.713775634765625e-05,
      "step": 7114,
      "training_step_time": 9.92776083946228
    },
    {
      "epoch": 2.713775634765625e-05,
      "model_forward_time": 3.1362051963806152,
      "step": 7114
    },
    {
      "epoch": 2.713775634765625e-05,
      "step": 7114,
      "training_step_time": 9.905768632888794
    },
    {
      "epoch": 2.713775634765625e-05,
      "model_forward_time": 3.1355504989624023,
      "step": 7114
    },
    {
      "epoch": 2.713775634765625e-05,
      "step": 7114,
      "training_step_time": 9.906082391738892
    },
    {
      "epoch": 2.713775634765625e-05,
      "model_forward_time": 3.1389214992523193,
      "step": 7114
    },
    {
      "epoch": 2.713775634765625e-05,
      "step": 7114,
      "training_step_time": 10.28553056716919
    },
    {
      "epoch": 2.7141571044921876e-05,
      "model_forward_time": 3.1447322368621826,
      "step": 7115
    },
    {
      "dynamics_loss_avg": 0.06871852986514568,
      "epoch": 2.7141571044921876e-05,
      "step": 7115
    },
    {
      "action_loss_avg": 0.023379762191325426,
      "epoch": 2.7141571044921876e-05,
      "step": 7115
    },
    {
      "epoch": 2.7141571044921876e-05,
      "step": 7115,
      "training_step_time": 10.22508692741394
    },
    {
      "epoch": 2.7141571044921876e-05,
      "model_forward_time": 3.138681650161743,
      "step": 7115
    },
    {
      "epoch": 2.7141571044921876e-05,
      "step": 7115,
      "training_step_time": 9.912477254867554
    },
    {
      "epoch": 2.7141571044921876e-05,
      "model_forward_time": 3.1461050510406494,
      "step": 7115
    },
    {
      "epoch": 2.7141571044921876e-05,
      "step": 7115,
      "training_step_time": 9.92275857925415
    },
    {
      "epoch": 2.7141571044921876e-05,
      "model_forward_time": 3.1419460773468018,
      "step": 7115
    },
    {
      "epoch": 2.7141571044921876e-05,
      "step": 7115,
      "training_step_time": 10.265713930130005
    },
    {
      "epoch": 2.71453857421875e-05,
      "model_forward_time": 3.1734981536865234,
      "step": 7116
    },
    {
      "epoch": 2.71453857421875e-05,
      "step": 7116,
      "training_step_time": 9.947440147399902
    },
    {
      "epoch": 2.71453857421875e-05,
      "model_forward_time": 3.4055237770080566,
      "step": 7116
    },
    {
      "epoch": 2.71453857421875e-05,
      "step": 7116,
      "training_step_time": 10.178638458251953
    },
    {
      "epoch": 2.71453857421875e-05,
      "model_forward_time": 3.145965337753296,
      "step": 7116
    },
    {
      "epoch": 2.71453857421875e-05,
      "step": 7116,
      "training_step_time": 9.92072868347168
    },
    {
      "epoch": 2.71453857421875e-05,
      "model_forward_time": 3.1390557289123535,
      "step": 7116
    },
    {
      "epoch": 2.71453857421875e-05,
      "step": 7116,
      "training_step_time": 10.273157358169556
    },
    {
      "epoch": 2.7149200439453125e-05,
      "model_forward_time": 3.1702146530151367,
      "step": 7117
    },
    {
      "epoch": 2.7149200439453125e-05,
      "step": 7117,
      "training_step_time": 9.941120386123657
    },
    {
      "epoch": 2.7149200439453125e-05,
      "model_forward_time": 3.138688325881958,
      "step": 7117
    },
    {
      "epoch": 2.7149200439453125e-05,
      "step": 7117,
      "training_step_time": 9.907015562057495
    },
    {
      "epoch": 2.7149200439453125e-05,
      "model_forward_time": 3.4318273067474365,
      "step": 7117
    },
    {
      "dynamics_loss_avg": 0.047613066248595716,
      "epoch": 2.7149200439453125e-05,
      "step": 7117
    },
    {
      "action_loss_avg": 0.016781359538435937,
      "epoch": 2.7149200439453125e-05,
      "step": 7117
    },
    {
      "epoch": 2.7149200439453125e-05,
      "step": 7117,
      "training_step_time": 10.200064659118652
    },
    {
      "epoch": 2.7149200439453125e-05,
      "model_forward_time": 3.138418674468994,
      "step": 7117
    },
    {
      "epoch": 2.7149200439453125e-05,
      "step": 7117,
      "training_step_time": 10.257308721542358
    },
    {
      "epoch": 2.715301513671875e-05,
      "model_forward_time": 3.1777844429016113,
      "step": 7118
    },
    {
      "epoch": 2.715301513671875e-05,
      "step": 7118,
      "training_step_time": 9.94583249092102
    },
    {
      "epoch": 2.715301513671875e-05,
      "model_forward_time": 3.1396334171295166,
      "step": 7118
    },
    {
      "epoch": 2.715301513671875e-05,
      "step": 7118,
      "training_step_time": 9.908883810043335
    },
    {
      "epoch": 2.715301513671875e-05,
      "model_forward_time": 3.1410725116729736,
      "step": 7118
    },
    {
      "epoch": 2.715301513671875e-05,
      "step": 7118,
      "training_step_time": 9.906696319580078
    },
    {
      "epoch": 2.715301513671875e-05,
      "model_forward_time": 3.3971447944641113,
      "step": 7118
    },
    {
      "epoch": 2.715301513671875e-05,
      "step": 7118,
      "training_step_time": 10.26735544204712
    },
    {
      "epoch": 2.7156829833984374e-05,
      "model_forward_time": 3.1468446254730225,
      "step": 7119
    },
    {
      "epoch": 2.7156829833984374e-05,
      "step": 7119,
      "training_step_time": 9.917070627212524
    },
    {
      "epoch": 2.7156829833984374e-05,
      "model_forward_time": 3.142301082611084,
      "step": 7119
    },
    {
      "epoch": 2.7156829833984374e-05,
      "step": 7119,
      "training_step_time": 9.906623125076294
    },
    {
      "epoch": 2.7156829833984374e-05,
      "model_forward_time": 3.1372721195220947,
      "step": 7119
    },
    {
      "epoch": 2.7156829833984374e-05,
      "step": 7119,
      "training_step_time": 9.90944528579712
    },
    {
      "epoch": 2.7156829833984374e-05,
      "model_forward_time": 3.14254093170166,
      "step": 7119
    },
    {
      "epoch": 2.7156829833984374e-05,
      "step": 7119,
      "training_step_time": 10.296920776367188
    },
    {
      "epoch": 2.716064453125e-05,
      "grad_norm": 0.517691969871521,
      "learning_rate": 1e-05,
      "loss": 0.0859,
      "step": 7120
    },
    {
      "epoch": 2.716064453125e-05,
      "model_forward_time": 3.417177438735962,
      "step": 7120
    },
    {
      "dynamics_loss_avg": 0.06484488919377326,
      "epoch": 2.716064453125e-05,
      "step": 7120
    },
    {
      "action_loss_avg": 0.025457292003557085,
      "epoch": 2.716064453125e-05,
      "step": 7120
    },
    {
      "epoch": 2.716064453125e-05,
      "step": 7120,
      "training_step_time": 10.190348386764526
    },
    {
      "epoch": 2.716064453125e-05,
      "model_forward_time": 3.144577980041504,
      "step": 7120
    },
    {
      "epoch": 2.716064453125e-05,
      "step": 7120,
      "training_step_time": 9.912118673324585
    },
    {
      "epoch": 2.716064453125e-05,
      "model_forward_time": 3.1422276496887207,
      "step": 7120
    },
    {
      "epoch": 2.716064453125e-05,
      "step": 7120,
      "training_step_time": 9.911257982254028
    },
    {
      "epoch": 2.716064453125e-05,
      "model_forward_time": 3.146430253982544,
      "step": 7120
    },
    {
      "epoch": 2.716064453125e-05,
      "step": 7120,
      "training_step_time": 10.018086671829224
    },
    {
      "epoch": 2.7164459228515627e-05,
      "model_forward_time": 3.1805739402770996,
      "step": 7121
    },
    {
      "epoch": 2.7164459228515627e-05,
      "step": 7121,
      "training_step_time": 9.95374846458435
    },
    {
      "epoch": 2.7164459228515627e-05,
      "model_forward_time": 3.1397933959960938,
      "step": 7121
    },
    {
      "epoch": 2.7164459228515627e-05,
      "step": 7121,
      "training_step_time": 10.195943117141724
    },
    {
      "epoch": 2.7164459228515627e-05,
      "model_forward_time": 3.1412739753723145,
      "step": 7121
    },
    {
      "epoch": 2.7164459228515627e-05,
      "step": 7121,
      "training_step_time": 9.912070274353027
    },
    {
      "epoch": 2.7164459228515627e-05,
      "model_forward_time": 3.1461069583892822,
      "step": 7121
    },
    {
      "epoch": 2.7164459228515627e-05,
      "step": 7121,
      "training_step_time": 10.291377305984497
    },
    {
      "epoch": 2.716827392578125e-05,
      "model_forward_time": 3.1735100746154785,
      "step": 7122
    },
    {
      "epoch": 2.716827392578125e-05,
      "step": 7122,
      "training_step_time": 9.942924737930298
    },
    {
      "epoch": 2.716827392578125e-05,
      "model_forward_time": 3.1429431438446045,
      "step": 7122
    },
    {
      "epoch": 2.716827392578125e-05,
      "step": 7122,
      "training_step_time": 9.90953540802002
    },
    {
      "epoch": 2.716827392578125e-05,
      "model_forward_time": 3.411668539047241,
      "step": 7122
    },
    {
      "dynamics_loss_avg": 0.06071097161620855,
      "epoch": 2.716827392578125e-05,
      "step": 7122
    },
    {
      "action_loss_avg": 0.023234587023034693,
      "epoch": 2.716827392578125e-05,
      "step": 7122
    },
    {
      "epoch": 2.716827392578125e-05,
      "step": 7122,
      "training_step_time": 10.187354326248169
    },
    {
      "epoch": 2.716827392578125e-05,
      "model_forward_time": 3.146711826324463,
      "step": 7122
    },
    {
      "epoch": 2.716827392578125e-05,
      "step": 7122,
      "training_step_time": 10.293405294418335
    },
    {
      "epoch": 2.7172088623046876e-05,
      "model_forward_time": 3.156707763671875,
      "step": 7123
    },
    {
      "epoch": 2.7172088623046876e-05,
      "step": 7123,
      "training_step_time": 9.9300537109375
    },
    {
      "epoch": 2.7172088623046876e-05,
      "model_forward_time": 3.141712188720703,
      "step": 7123
    },
    {
      "epoch": 2.7172088623046876e-05,
      "step": 7123,
      "training_step_time": 9.909387826919556
    },
    {
      "epoch": 2.7172088623046876e-05,
      "model_forward_time": 3.147873640060425,
      "step": 7123
    },
    {
      "epoch": 2.7172088623046876e-05,
      "step": 7123,
      "training_step_time": 9.934877872467041
    },
    {
      "epoch": 2.7172088623046876e-05,
      "model_forward_time": 3.1421046257019043,
      "step": 7123
    },
    {
      "epoch": 2.7172088623046876e-05,
      "step": 7123,
      "training_step_time": 10.282480716705322
    },
    {
      "epoch": 2.71759033203125e-05,
      "model_forward_time": 3.153374195098877,
      "step": 7124
    },
    {
      "epoch": 2.71759033203125e-05,
      "step": 7124,
      "training_step_time": 9.923514366149902
    },
    {
      "epoch": 2.71759033203125e-05,
      "model_forward_time": 3.134864330291748,
      "step": 7124
    },
    {
      "epoch": 2.71759033203125e-05,
      "step": 7124,
      "training_step_time": 9.903688907623291
    },
    {
      "epoch": 2.71759033203125e-05,
      "model_forward_time": 3.137115716934204,
      "step": 7124
    },
    {
      "epoch": 2.71759033203125e-05,
      "step": 7124,
      "training_step_time": 9.908867835998535
    },
    {
      "epoch": 2.71759033203125e-05,
      "model_forward_time": 3.1327314376831055,
      "step": 7124
    },
    {
      "epoch": 2.71759033203125e-05,
      "step": 7124,
      "training_step_time": 10.019146203994751
    },
    {
      "epoch": 2.7179718017578125e-05,
      "model_forward_time": 3.441178798675537,
      "step": 7125
    },
    {
      "dynamics_loss_avg": 0.0481883592903614,
      "epoch": 2.7179718017578125e-05,
      "step": 7125
    },
    {
      "action_loss_avg": 0.019579982897266746,
      "epoch": 2.7179718017578125e-05,
      "step": 7125
    },
    {
      "epoch": 2.7179718017578125e-05,
      "step": 7125,
      "training_step_time": 10.214123010635376
    },
    {
      "epoch": 2.7179718017578125e-05,
      "model_forward_time": 3.1342344284057617,
      "step": 7125
    },
    {
      "epoch": 2.7179718017578125e-05,
      "step": 7125,
      "training_step_time": 9.907260179519653
    },
    {
      "epoch": 2.7179718017578125e-05,
      "model_forward_time": 3.1351821422576904,
      "step": 7125
    },
    {
      "epoch": 2.7179718017578125e-05,
      "step": 7125,
      "training_step_time": 9.90191650390625
    },
    {
      "epoch": 2.7179718017578125e-05,
      "model_forward_time": 3.1334877014160156,
      "step": 7125
    },
    {
      "epoch": 2.7179718017578125e-05,
      "step": 7125,
      "training_step_time": 10.273093461990356
    },
    {
      "epoch": 2.718353271484375e-05,
      "model_forward_time": 3.171156167984009,
      "step": 7126
    },
    {
      "epoch": 2.718353271484375e-05,
      "step": 7126,
      "training_step_time": 9.940166473388672
    },
    {
      "epoch": 2.718353271484375e-05,
      "model_forward_time": 3.13480281829834,
      "step": 7126
    },
    {
      "epoch": 2.718353271484375e-05,
      "step": 7126,
      "training_step_time": 10.16700005531311
    },
    {
      "epoch": 2.718353271484375e-05,
      "model_forward_time": 3.132408618927002,
      "step": 7126
    },
    {
      "epoch": 2.718353271484375e-05,
      "step": 7126,
      "training_step_time": 9.899296045303345
    },
    {
      "epoch": 2.718353271484375e-05,
      "model_forward_time": 3.131321668624878,
      "step": 7126
    },
    {
      "epoch": 2.718353271484375e-05,
      "step": 7126,
      "training_step_time": 10.004038095474243
    },
    {
      "epoch": 2.7187347412109374e-05,
      "model_forward_time": 3.185466766357422,
      "step": 7127
    },
    {
      "epoch": 2.7187347412109374e-05,
      "step": 7127,
      "training_step_time": 9.952035427093506
    },
    {
      "epoch": 2.7187347412109374e-05,
      "model_forward_time": 3.1413514614105225,
      "step": 7127
    },
    {
      "epoch": 2.7187347412109374e-05,
      "step": 7127,
      "training_step_time": 9.911792755126953
    },
    {
      "epoch": 2.7187347412109374e-05,
      "model_forward_time": 3.4218502044677734,
      "step": 7127
    },
    {
      "dynamics_loss_avg": 0.05762110408395529,
      "epoch": 2.7187347412109374e-05,
      "step": 7127
    },
    {
      "action_loss_avg": 0.02085341876372695,
      "epoch": 2.7187347412109374e-05,
      "step": 7127
    },
    {
      "epoch": 2.7187347412109374e-05,
      "step": 7127,
      "training_step_time": 10.196532011032104
    },
    {
      "epoch": 2.7187347412109374e-05,
      "model_forward_time": 3.144742727279663,
      "step": 7127
    },
    {
      "epoch": 2.7187347412109374e-05,
      "step": 7127,
      "training_step_time": 10.25161600112915
    },
    {
      "epoch": 2.7191162109375e-05,
      "model_forward_time": 3.17080020904541,
      "step": 7128
    },
    {
      "epoch": 2.7191162109375e-05,
      "step": 7128,
      "training_step_time": 9.947965383529663
    },
    {
      "epoch": 2.7191162109375e-05,
      "model_forward_time": 3.1425442695617676,
      "step": 7128
    },
    {
      "epoch": 2.7191162109375e-05,
      "step": 7128,
      "training_step_time": 9.915323734283447
    },
    {
      "epoch": 2.7191162109375e-05,
      "model_forward_time": 3.1417577266693115,
      "step": 7128
    },
    {
      "epoch": 2.7191162109375e-05,
      "step": 7128,
      "training_step_time": 9.910773992538452
    },
    {
      "epoch": 2.7191162109375e-05,
      "model_forward_time": 3.414047956466675,
      "step": 7128
    },
    {
      "epoch": 2.7191162109375e-05,
      "step": 7128,
      "training_step_time": 10.279922246932983
    },
    {
      "epoch": 2.7194976806640623e-05,
      "model_forward_time": 3.1586148738861084,
      "step": 7129
    },
    {
      "epoch": 2.7194976806640623e-05,
      "step": 7129,
      "training_step_time": 9.932225465774536
    },
    {
      "epoch": 2.7194976806640623e-05,
      "model_forward_time": 3.14174222946167,
      "step": 7129
    },
    {
      "epoch": 2.7194976806640623e-05,
      "step": 7129,
      "training_step_time": 9.913248062133789
    },
    {
      "epoch": 2.7194976806640623e-05,
      "model_forward_time": 3.1451938152313232,
      "step": 7129
    },
    {
      "epoch": 2.7194976806640623e-05,
      "step": 7129,
      "training_step_time": 9.918821811676025
    },
    {
      "epoch": 2.7194976806640623e-05,
      "model_forward_time": 3.1447079181671143,
      "step": 7129
    },
    {
      "epoch": 2.7194976806640623e-05,
      "step": 7129,
      "training_step_time": 10.25703501701355
    },
    {
      "epoch": 2.719879150390625e-05,
      "grad_norm": 0.49461907148361206,
      "learning_rate": 1e-05,
      "loss": 0.0823,
      "step": 7130
    },
    {
      "epoch": 2.719879150390625e-05,
      "model_forward_time": 3.4499893188476562,
      "step": 7130
    },
    {
      "dynamics_loss_avg": 0.07184096463024617,
      "epoch": 2.719879150390625e-05,
      "step": 7130
    },
    {
      "action_loss_avg": 0.029310712567530574,
      "epoch": 2.719879150390625e-05,
      "step": 7130
    },
    {
      "epoch": 2.719879150390625e-05,
      "step": 7130,
      "training_step_time": 10.223527431488037
    },
    {
      "epoch": 2.719879150390625e-05,
      "model_forward_time": 3.1434648036956787,
      "step": 7130
    },
    {
      "epoch": 2.719879150390625e-05,
      "step": 7130,
      "training_step_time": 9.914080381393433
    },
    {
      "epoch": 2.719879150390625e-05,
      "model_forward_time": 3.1407053470611572,
      "step": 7130
    },
    {
      "epoch": 2.719879150390625e-05,
      "step": 7130,
      "training_step_time": 9.915923833847046
    },
    {
      "epoch": 2.719879150390625e-05,
      "model_forward_time": 3.1486663818359375,
      "step": 7130
    },
    {
      "epoch": 2.719879150390625e-05,
      "step": 7130,
      "training_step_time": 10.324315071105957
    },
    {
      "epoch": 2.7202606201171876e-05,
      "model_forward_time": 3.166560649871826,
      "step": 7131
    },
    {
      "epoch": 2.7202606201171876e-05,
      "step": 7131,
      "training_step_time": 9.93880558013916
    },
    {
      "epoch": 2.7202606201171876e-05,
      "model_forward_time": 3.4382359981536865,
      "step": 7131
    },
    {
      "epoch": 2.7202606201171876e-05,
      "step": 7131,
      "training_step_time": 10.210925340652466
    },
    {
      "epoch": 2.7202606201171876e-05,
      "model_forward_time": 3.1457931995391846,
      "step": 7131
    },
    {
      "epoch": 2.7202606201171876e-05,
      "step": 7131,
      "training_step_time": 9.91938328742981
    },
    {
      "epoch": 2.7202606201171876e-05,
      "model_forward_time": 3.1460440158843994,
      "step": 7131
    },
    {
      "epoch": 2.7202606201171876e-05,
      "step": 7131,
      "training_step_time": 10.039766311645508
    },
    {
      "epoch": 2.72064208984375e-05,
      "model_forward_time": 3.160604476928711,
      "step": 7132
    },
    {
      "epoch": 2.72064208984375e-05,
      "step": 7132,
      "training_step_time": 9.933659791946411
    },
    {
      "epoch": 2.72064208984375e-05,
      "model_forward_time": 3.1436495780944824,
      "step": 7132
    },
    {
      "epoch": 2.72064208984375e-05,
      "step": 7132,
      "training_step_time": 9.921406269073486
    },
    {
      "epoch": 2.72064208984375e-05,
      "model_forward_time": 3.1512913703918457,
      "step": 7132
    },
    {
      "dynamics_loss_avg": 0.06122639891691506,
      "epoch": 2.72064208984375e-05,
      "step": 7132
    },
    {
      "action_loss_avg": 0.025632294802926482,
      "epoch": 2.72064208984375e-05,
      "step": 7132
    },
    {
      "epoch": 2.72064208984375e-05,
      "step": 7132,
      "training_step_time": 10.212255239486694
    },
    {
      "epoch": 2.72064208984375e-05,
      "model_forward_time": 3.1499366760253906,
      "step": 7132
    },
    {
      "epoch": 2.72064208984375e-05,
      "step": 7132,
      "training_step_time": 10.020776748657227
    },
    {
      "epoch": 2.7210235595703125e-05,
      "model_forward_time": 3.190002679824829,
      "step": 7133
    },
    {
      "epoch": 2.7210235595703125e-05,
      "step": 7133,
      "training_step_time": 9.963855266571045
    },
    {
      "epoch": 2.7210235595703125e-05,
      "model_forward_time": 3.1450388431549072,
      "step": 7133
    },
    {
      "epoch": 2.7210235595703125e-05,
      "step": 7133,
      "training_step_time": 9.918769359588623
    },
    {
      "epoch": 2.7210235595703125e-05,
      "model_forward_time": 3.1478657722473145,
      "step": 7133
    },
    {
      "epoch": 2.7210235595703125e-05,
      "step": 7133,
      "training_step_time": 9.922751426696777
    },
    {
      "epoch": 2.7210235595703125e-05,
      "model_forward_time": 3.471834659576416,
      "step": 7133
    },
    {
      "epoch": 2.7210235595703125e-05,
      "step": 7133,
      "training_step_time": 10.342194557189941
    },
    {
      "epoch": 2.721405029296875e-05,
      "model_forward_time": 3.166510820388794,
      "step": 7134
    },
    {
      "epoch": 2.721405029296875e-05,
      "step": 7134,
      "training_step_time": 9.948119878768921
    },
    {
      "epoch": 2.721405029296875e-05,
      "model_forward_time": 3.1445720195770264,
      "step": 7134
    },
    {
      "epoch": 2.721405029296875e-05,
      "step": 7134,
      "training_step_time": 9.916219234466553
    },
    {
      "epoch": 2.721405029296875e-05,
      "model_forward_time": 3.14312744140625,
      "step": 7134
    },
    {
      "epoch": 2.721405029296875e-05,
      "step": 7134,
      "training_step_time": 9.918041467666626
    },
    {
      "epoch": 2.721405029296875e-05,
      "model_forward_time": 3.14036226272583,
      "step": 7134
    },
    {
      "epoch": 2.721405029296875e-05,
      "step": 7134,
      "training_step_time": 10.27720308303833
    },
    {
      "epoch": 2.7217864990234374e-05,
      "model_forward_time": 3.1506364345550537,
      "step": 7135
    },
    {
      "dynamics_loss_avg": 0.0434581222012639,
      "epoch": 2.7217864990234374e-05,
      "step": 7135
    },
    {
      "action_loss_avg": 0.019810654409229755,
      "epoch": 2.7217864990234374e-05,
      "step": 7135
    },
    {
      "epoch": 2.7217864990234374e-05,
      "step": 7135,
      "training_step_time": 10.218732595443726
    },
    {
      "epoch": 2.7217864990234374e-05,
      "model_forward_time": 3.1413211822509766,
      "step": 7135
    },
    {
      "epoch": 2.7217864990234374e-05,
      "step": 7135,
      "training_step_time": 9.908445835113525
    },
    {
      "epoch": 2.7217864990234374e-05,
      "model_forward_time": 3.14367938041687,
      "step": 7135
    },
    {
      "epoch": 2.7217864990234374e-05,
      "step": 7135,
      "training_step_time": 9.907247543334961
    },
    {
      "epoch": 2.7217864990234374e-05,
      "model_forward_time": 3.138819694519043,
      "step": 7135
    },
    {
      "epoch": 2.7217864990234374e-05,
      "step": 7135,
      "training_step_time": 10.245301246643066
    },
    {
      "epoch": 2.72216796875e-05,
      "model_forward_time": 3.1483006477355957,
      "step": 7136
    },
    {
      "epoch": 2.72216796875e-05,
      "step": 7136,
      "training_step_time": 9.914334058761597
    },
    {
      "epoch": 2.72216796875e-05,
      "model_forward_time": 3.4017226696014404,
      "step": 7136
    },
    {
      "epoch": 2.72216796875e-05,
      "step": 7136,
      "training_step_time": 10.160810470581055
    },
    {
      "epoch": 2.72216796875e-05,
      "model_forward_time": 3.1512393951416016,
      "step": 7136
    },
    {
      "epoch": 2.72216796875e-05,
      "step": 7136,
      "training_step_time": 9.918301343917847
    },
    {
      "epoch": 2.72216796875e-05,
      "model_forward_time": 3.1353323459625244,
      "step": 7136
    },
    {
      "epoch": 2.72216796875e-05,
      "step": 7136,
      "training_step_time": 10.013146877288818
    },
    {
      "epoch": 2.7225494384765623e-05,
      "model_forward_time": 3.191312074661255,
      "step": 7137
    },
    {
      "epoch": 2.7225494384765623e-05,
      "step": 7137,
      "training_step_time": 9.961609601974487
    },
    {
      "epoch": 2.7225494384765623e-05,
      "model_forward_time": 3.1385982036590576,
      "step": 7137
    },
    {
      "epoch": 2.7225494384765623e-05,
      "step": 7137,
      "training_step_time": 9.909111976623535
    },
    {
      "epoch": 2.7225494384765623e-05,
      "model_forward_time": 3.431769847869873,
      "step": 7137
    },
    {
      "dynamics_loss_avg": 0.05644292011857033,
      "epoch": 2.7225494384765623e-05,
      "step": 7137
    },
    {
      "action_loss_avg": 0.024685046635568142,
      "epoch": 2.7225494384765623e-05,
      "step": 7137
    },
    {
      "epoch": 2.7225494384765623e-05,
      "step": 7137,
      "training_step_time": 10.201344728469849
    },
    {
      "epoch": 2.7225494384765623e-05,
      "model_forward_time": 3.1383304595947266,
      "step": 7137
    },
    {
      "epoch": 2.7225494384765623e-05,
      "step": 7137,
      "training_step_time": 10.285786390304565
    },
    {
      "epoch": 2.722930908203125e-05,
      "model_forward_time": 3.184995651245117,
      "step": 7138
    },
    {
      "epoch": 2.722930908203125e-05,
      "step": 7138,
      "training_step_time": 9.950952768325806
    },
    {
      "epoch": 2.722930908203125e-05,
      "model_forward_time": 3.143564224243164,
      "step": 7138
    },
    {
      "epoch": 2.722930908203125e-05,
      "step": 7138,
      "training_step_time": 9.909410238265991
    },
    {
      "epoch": 2.722930908203125e-05,
      "model_forward_time": 3.140435218811035,
      "step": 7138
    },
    {
      "epoch": 2.722930908203125e-05,
      "step": 7138,
      "training_step_time": 9.905314683914185
    },
    {
      "epoch": 2.722930908203125e-05,
      "model_forward_time": 3.4034790992736816,
      "step": 7138
    },
    {
      "epoch": 2.722930908203125e-05,
      "step": 7138,
      "training_step_time": 10.260823965072632
    },
    {
      "epoch": 2.7233123779296876e-05,
      "model_forward_time": 3.1789371967315674,
      "step": 7139
    },
    {
      "epoch": 2.7233123779296876e-05,
      "step": 7139,
      "training_step_time": 9.945521831512451
    },
    {
      "epoch": 2.7233123779296876e-05,
      "model_forward_time": 3.1466825008392334,
      "step": 7139
    },
    {
      "epoch": 2.7233123779296876e-05,
      "step": 7139,
      "training_step_time": 9.920945882797241
    },
    {
      "epoch": 2.7233123779296876e-05,
      "model_forward_time": 3.1449854373931885,
      "step": 7139
    },
    {
      "epoch": 2.7233123779296876e-05,
      "step": 7139,
      "training_step_time": 9.91545295715332
    },
    {
      "epoch": 2.7233123779296876e-05,
      "model_forward_time": 3.144124746322632,
      "step": 7139
    },
    {
      "epoch": 2.7233123779296876e-05,
      "step": 7139,
      "training_step_time": 10.25995135307312
    },
    {
      "epoch": 2.72369384765625e-05,
      "grad_norm": 0.3125889301300049,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 7140
    },
    {
      "epoch": 2.72369384765625e-05,
      "model_forward_time": 3.4487929344177246,
      "step": 7140
    },
    {
      "dynamics_loss_avg": 0.07213966194540262,
      "epoch": 2.72369384765625e-05,
      "step": 7140
    },
    {
      "action_loss_avg": 0.03549908115528524,
      "epoch": 2.72369384765625e-05,
      "step": 7140
    },
    {
      "epoch": 2.72369384765625e-05,
      "step": 7140,
      "training_step_time": 10.216087818145752
    },
    {
      "epoch": 2.72369384765625e-05,
      "model_forward_time": 3.141409158706665,
      "step": 7140
    },
    {
      "epoch": 2.72369384765625e-05,
      "step": 7140,
      "training_step_time": 9.906021118164062
    },
    {
      "epoch": 2.72369384765625e-05,
      "model_forward_time": 3.1390881538391113,
      "step": 7140
    },
    {
      "epoch": 2.72369384765625e-05,
      "step": 7140,
      "training_step_time": 9.903143644332886
    },
    {
      "epoch": 2.72369384765625e-05,
      "model_forward_time": 3.1478323936462402,
      "step": 7140
    },
    {
      "epoch": 2.72369384765625e-05,
      "step": 7140,
      "training_step_time": 10.285506010055542
    },
    {
      "epoch": 2.7240753173828125e-05,
      "model_forward_time": 3.172933340072632,
      "step": 7141
    },
    {
      "epoch": 2.7240753173828125e-05,
      "step": 7141,
      "training_step_time": 9.945545434951782
    },
    {
      "epoch": 2.7240753173828125e-05,
      "model_forward_time": 3.129298686981201,
      "step": 7141
    },
    {
      "epoch": 2.7240753173828125e-05,
      "step": 7141,
      "training_step_time": 10.183202981948853
    },
    {
      "epoch": 2.7240753173828125e-05,
      "model_forward_time": 3.1297881603240967,
      "step": 7141
    },
    {
      "epoch": 2.7240753173828125e-05,
      "step": 7141,
      "training_step_time": 9.898259401321411
    },
    {
      "epoch": 2.7240753173828125e-05,
      "model_forward_time": 3.133479595184326,
      "step": 7141
    },
    {
      "epoch": 2.7240753173828125e-05,
      "step": 7141,
      "training_step_time": 10.305347204208374
    },
    {
      "epoch": 2.724456787109375e-05,
      "model_forward_time": 3.145162582397461,
      "step": 7142
    },
    {
      "epoch": 2.724456787109375e-05,
      "step": 7142,
      "training_step_time": 9.924813032150269
    },
    {
      "epoch": 2.724456787109375e-05,
      "model_forward_time": 3.134410858154297,
      "step": 7142
    },
    {
      "epoch": 2.724456787109375e-05,
      "step": 7142,
      "training_step_time": 9.913441896438599
    },
    {
      "epoch": 2.724456787109375e-05,
      "model_forward_time": 3.393744945526123,
      "step": 7142
    },
    {
      "dynamics_loss_avg": 0.06746340133249759,
      "epoch": 2.724456787109375e-05,
      "step": 7142
    },
    {
      "action_loss_avg": 0.02458961522206664,
      "epoch": 2.724456787109375e-05,
      "step": 7142
    },
    {
      "epoch": 2.724456787109375e-05,
      "step": 7142,
      "training_step_time": 10.175696611404419
    },
    {
      "epoch": 2.724456787109375e-05,
      "model_forward_time": 3.136589527130127,
      "step": 7142
    },
    {
      "epoch": 2.724456787109375e-05,
      "step": 7142,
      "training_step_time": 10.311741828918457
    },
    {
      "epoch": 2.7248382568359374e-05,
      "model_forward_time": 3.1512532234191895,
      "step": 7143
    },
    {
      "epoch": 2.7248382568359374e-05,
      "step": 7143,
      "training_step_time": 9.935582637786865
    },
    {
      "epoch": 2.7248382568359374e-05,
      "model_forward_time": 3.1362462043762207,
      "step": 7143
    },
    {
      "epoch": 2.7248382568359374e-05,
      "step": 7143,
      "training_step_time": 9.919559955596924
    },
    {
      "epoch": 2.7248382568359374e-05,
      "model_forward_time": 3.1413049697875977,
      "step": 7143
    },
    {
      "epoch": 2.7248382568359374e-05,
      "step": 7143,
      "training_step_time": 9.924197673797607
    },
    {
      "epoch": 2.7248382568359374e-05,
      "model_forward_time": 3.1314570903778076,
      "step": 7143
    },
    {
      "epoch": 2.7248382568359374e-05,
      "step": 7143,
      "training_step_time": 10.279205799102783
    },
    {
      "epoch": 2.7252197265625e-05,
      "model_forward_time": 3.169127941131592,
      "step": 7144
    },
    {
      "epoch": 2.7252197265625e-05,
      "step": 7144,
      "training_step_time": 9.950254678726196
    },
    {
      "epoch": 2.7252197265625e-05,
      "model_forward_time": 3.133166551589966,
      "step": 7144
    },
    {
      "epoch": 2.7252197265625e-05,
      "step": 7144,
      "training_step_time": 9.90846872329712
    },
    {
      "epoch": 2.7252197265625e-05,
      "model_forward_time": 3.1373820304870605,
      "step": 7144
    },
    {
      "epoch": 2.7252197265625e-05,
      "step": 7144,
      "training_step_time": 9.911371946334839
    },
    {
      "epoch": 2.7252197265625e-05,
      "model_forward_time": 3.1304986476898193,
      "step": 7144
    },
    {
      "epoch": 2.7252197265625e-05,
      "step": 7144,
      "training_step_time": 10.019556999206543
    },
    {
      "epoch": 2.7256011962890623e-05,
      "model_forward_time": 3.4271984100341797,
      "step": 7145
    },
    {
      "dynamics_loss_avg": 0.06936574913561344,
      "epoch": 2.7256011962890623e-05,
      "step": 7145
    },
    {
      "action_loss_avg": 0.027066652849316596,
      "epoch": 2.7256011962890623e-05,
      "step": 7145
    },
    {
      "epoch": 2.7256011962890623e-05,
      "step": 7145,
      "training_step_time": 10.207277536392212
    },
    {
      "epoch": 2.7256011962890623e-05,
      "model_forward_time": 3.1350393295288086,
      "step": 7145
    },
    {
      "epoch": 2.7256011962890623e-05,
      "step": 7145,
      "training_step_time": 9.915400743484497
    },
    {
      "epoch": 2.7256011962890623e-05,
      "model_forward_time": 3.1377246379852295,
      "step": 7145
    },
    {
      "epoch": 2.7256011962890623e-05,
      "step": 7145,
      "training_step_time": 9.915714502334595
    },
    {
      "epoch": 2.7256011962890623e-05,
      "model_forward_time": 3.1396255493164062,
      "step": 7145
    },
    {
      "epoch": 2.7256011962890623e-05,
      "step": 7145,
      "training_step_time": 10.296303033828735
    },
    {
      "epoch": 2.725982666015625e-05,
      "model_forward_time": 3.180922508239746,
      "step": 7146
    },
    {
      "epoch": 2.725982666015625e-05,
      "step": 7146,
      "training_step_time": 9.96605110168457
    },
    {
      "epoch": 2.725982666015625e-05,
      "model_forward_time": 3.1363019943237305,
      "step": 7146
    },
    {
      "epoch": 2.725982666015625e-05,
      "step": 7146,
      "training_step_time": 10.19463324546814
    },
    {
      "epoch": 2.725982666015625e-05,
      "model_forward_time": 3.139223575592041,
      "step": 7146
    },
    {
      "epoch": 2.725982666015625e-05,
      "step": 7146,
      "training_step_time": 9.922315835952759
    },
    {
      "epoch": 2.725982666015625e-05,
      "model_forward_time": 3.1373515129089355,
      "step": 7146
    },
    {
      "epoch": 2.725982666015625e-05,
      "step": 7146,
      "training_step_time": 10.325760841369629
    },
    {
      "epoch": 2.7263641357421876e-05,
      "model_forward_time": 3.1447103023529053,
      "step": 7147
    },
    {
      "epoch": 2.7263641357421876e-05,
      "step": 7147,
      "training_step_time": 9.926398277282715
    },
    {
      "epoch": 2.7263641357421876e-05,
      "model_forward_time": 3.135005235671997,
      "step": 7147
    },
    {
      "epoch": 2.7263641357421876e-05,
      "step": 7147,
      "training_step_time": 9.91249132156372
    },
    {
      "epoch": 2.7263641357421876e-05,
      "model_forward_time": 3.419163703918457,
      "step": 7147
    },
    {
      "dynamics_loss_avg": 0.053264778852462766,
      "epoch": 2.7263641357421876e-05,
      "step": 7147
    },
    {
      "action_loss_avg": 0.01495910263620317,
      "epoch": 2.7263641357421876e-05,
      "step": 7147
    },
    {
      "epoch": 2.7263641357421876e-05,
      "step": 7147,
      "training_step_time": 10.20091438293457
    },
    {
      "epoch": 2.7263641357421876e-05,
      "model_forward_time": 3.137997627258301,
      "step": 7147
    },
    {
      "epoch": 2.7263641357421876e-05,
      "step": 7147,
      "training_step_time": 10.31039547920227
    },
    {
      "epoch": 2.72674560546875e-05,
      "model_forward_time": 3.142483949661255,
      "step": 7148
    },
    {
      "epoch": 2.72674560546875e-05,
      "step": 7148,
      "training_step_time": 9.923741579055786
    },
    {
      "epoch": 2.72674560546875e-05,
      "model_forward_time": 3.1419906616210938,
      "step": 7148
    },
    {
      "epoch": 2.72674560546875e-05,
      "step": 7148,
      "training_step_time": 9.919938564300537
    },
    {
      "epoch": 2.72674560546875e-05,
      "model_forward_time": 3.138124942779541,
      "step": 7148
    },
    {
      "epoch": 2.72674560546875e-05,
      "step": 7148,
      "training_step_time": 9.91825246810913
    },
    {
      "epoch": 2.72674560546875e-05,
      "model_forward_time": 3.4304003715515137,
      "step": 7148
    },
    {
      "epoch": 2.72674560546875e-05,
      "step": 7148,
      "training_step_time": 10.304710388183594
    },
    {
      "epoch": 2.7271270751953125e-05,
      "model_forward_time": 3.181600570678711,
      "step": 7149
    },
    {
      "epoch": 2.7271270751953125e-05,
      "step": 7149,
      "training_step_time": 9.971109628677368
    },
    {
      "epoch": 2.7271270751953125e-05,
      "model_forward_time": 3.1372010707855225,
      "step": 7149
    },
    {
      "epoch": 2.7271270751953125e-05,
      "step": 7149,
      "training_step_time": 9.922770977020264
    },
    {
      "epoch": 2.7271270751953125e-05,
      "model_forward_time": 3.135274887084961,
      "step": 7149
    },
    {
      "epoch": 2.7271270751953125e-05,
      "step": 7149,
      "training_step_time": 9.916524887084961
    },
    {
      "epoch": 2.7271270751953125e-05,
      "model_forward_time": 3.1373867988586426,
      "step": 7149
    },
    {
      "epoch": 2.7271270751953125e-05,
      "step": 7149,
      "training_step_time": 10.021100044250488
    },
    {
      "epoch": 2.727508544921875e-05,
      "grad_norm": 0.4082852005958557,
      "learning_rate": 1e-05,
      "loss": 0.0875,
      "step": 7150
    },
    {
      "epoch": 2.727508544921875e-05,
      "model_forward_time": 3.4231839179992676,
      "step": 7150
    },
    {
      "dynamics_loss_avg": 0.051232816837728026,
      "epoch": 2.727508544921875e-05,
      "step": 7150
    },
    {
      "action_loss_avg": 0.02123810676857829,
      "epoch": 2.727508544921875e-05,
      "step": 7150
    },
    {
      "epoch": 2.727508544921875e-05,
      "step": 7150,
      "training_step_time": 10.204987525939941
    },
    {
      "epoch": 2.727508544921875e-05,
      "model_forward_time": 3.1379003524780273,
      "step": 7150
    },
    {
      "epoch": 2.727508544921875e-05,
      "step": 7150,
      "training_step_time": 9.916781663894653
    },
    {
      "epoch": 2.727508544921875e-05,
      "model_forward_time": 3.1352996826171875,
      "step": 7150
    },
    {
      "epoch": 2.727508544921875e-05,
      "step": 7150,
      "training_step_time": 9.915173530578613
    },
    {
      "epoch": 2.727508544921875e-05,
      "model_forward_time": 3.139399290084839,
      "step": 7150
    },
    {
      "epoch": 2.727508544921875e-05,
      "step": 7150,
      "training_step_time": 10.263486862182617
    },
    {
      "epoch": 2.7278900146484374e-05,
      "model_forward_time": 3.167292833328247,
      "step": 7151
    },
    {
      "epoch": 2.7278900146484374e-05,
      "step": 7151,
      "training_step_time": 9.951152086257935
    },
    {
      "epoch": 2.7278900146484374e-05,
      "model_forward_time": 3.421623706817627,
      "step": 7151
    },
    {
      "epoch": 2.7278900146484374e-05,
      "step": 7151,
      "training_step_time": 10.20673155784607
    },
    {
      "epoch": 2.7278900146484374e-05,
      "model_forward_time": 3.131685256958008,
      "step": 7151
    },
    {
      "epoch": 2.7278900146484374e-05,
      "step": 7151,
      "training_step_time": 9.915271997451782
    },
    {
      "epoch": 2.7278900146484374e-05,
      "model_forward_time": 3.1388742923736572,
      "step": 7151
    },
    {
      "epoch": 2.7278900146484374e-05,
      "step": 7151,
      "training_step_time": 10.311887502670288
    },
    {
      "epoch": 2.728271484375e-05,
      "model_forward_time": 3.1607396602630615,
      "step": 7152
    },
    {
      "epoch": 2.728271484375e-05,
      "step": 7152,
      "training_step_time": 9.946889162063599
    },
    {
      "epoch": 2.728271484375e-05,
      "model_forward_time": 3.1371779441833496,
      "step": 7152
    },
    {
      "epoch": 2.728271484375e-05,
      "step": 7152,
      "training_step_time": 9.91561245918274
    },
    {
      "epoch": 2.728271484375e-05,
      "model_forward_time": 3.14180064201355,
      "step": 7152
    },
    {
      "dynamics_loss_avg": 0.05153217855840921,
      "epoch": 2.728271484375e-05,
      "step": 7152
    },
    {
      "action_loss_avg": 0.020521745830774308,
      "epoch": 2.728271484375e-05,
      "step": 7152
    },
    {
      "epoch": 2.728271484375e-05,
      "step": 7152,
      "training_step_time": 10.206220149993896
    },
    {
      "epoch": 2.728271484375e-05,
      "model_forward_time": 3.1424596309661865,
      "step": 7152
    },
    {
      "epoch": 2.728271484375e-05,
      "step": 7152,
      "training_step_time": 10.30141830444336
    },
    {
      "epoch": 2.7286529541015623e-05,
      "model_forward_time": 3.1526174545288086,
      "step": 7153
    },
    {
      "epoch": 2.7286529541015623e-05,
      "step": 7153,
      "training_step_time": 9.933005094528198
    },
    {
      "epoch": 2.7286529541015623e-05,
      "model_forward_time": 3.1370034217834473,
      "step": 7153
    },
    {
      "epoch": 2.7286529541015623e-05,
      "step": 7153,
      "training_step_time": 9.919296741485596
    },
    {
      "epoch": 2.7286529541015623e-05,
      "model_forward_time": 3.1438238620758057,
      "step": 7153
    },
    {
      "epoch": 2.7286529541015623e-05,
      "step": 7153,
      "training_step_time": 9.92406702041626
    },
    {
      "epoch": 2.7286529541015623e-05,
      "model_forward_time": 3.427340030670166,
      "step": 7153
    },
    {
      "epoch": 2.7286529541015623e-05,
      "step": 7153,
      "training_step_time": 10.300890684127808
    },
    {
      "epoch": 2.729034423828125e-05,
      "model_forward_time": 3.1588499546051025,
      "step": 7154
    },
    {
      "epoch": 2.729034423828125e-05,
      "step": 7154,
      "training_step_time": 9.939325332641602
    },
    {
      "epoch": 2.729034423828125e-05,
      "model_forward_time": 3.1372101306915283,
      "step": 7154
    },
    {
      "epoch": 2.729034423828125e-05,
      "step": 7154,
      "training_step_time": 9.913252830505371
    },
    {
      "epoch": 2.729034423828125e-05,
      "model_forward_time": 3.1442394256591797,
      "step": 7154
    },
    {
      "epoch": 2.729034423828125e-05,
      "step": 7154,
      "training_step_time": 9.923330307006836
    },
    {
      "epoch": 2.729034423828125e-05,
      "model_forward_time": 3.141129970550537,
      "step": 7154
    },
    {
      "epoch": 2.729034423828125e-05,
      "step": 7154,
      "training_step_time": 10.320243120193481
    },
    {
      "epoch": 2.7294158935546876e-05,
      "model_forward_time": 2.444812059402466,
      "step": 7155
    },
    {
      "dynamics_loss_avg": 0.05525617282837629,
      "epoch": 2.7294158935546876e-05,
      "step": 7155
    },
    {
      "action_loss_avg": 0.018030953127890825,
      "epoch": 2.7294158935546876e-05,
      "step": 7155
    },
    {
      "epoch": 2.7294158935546876e-05,
      "step": 7155,
      "training_step_time": 7.47784686088562
    },
    {
      "epoch": 2.7294158935546876e-05,
      "model_forward_time": 3.378831148147583,
      "step": 7155
    },
    {
      "epoch": 2.7294158935546876e-05,
      "step": 7155,
      "training_step_time": 10.149499893188477
    },
    {
      "epoch": 2.7294158935546876e-05,
      "model_forward_time": 3.1393423080444336,
      "step": 7155
    },
    {
      "epoch": 2.7294158935546876e-05,
      "step": 7155,
      "training_step_time": 9.912960529327393
    },
    {
      "epoch": 2.7294158935546876e-05,
      "model_forward_time": 3.138484239578247,
      "step": 7155
    },
    {
      "epoch": 2.7294158935546876e-05,
      "step": 7155,
      "training_step_time": 10.332457304000854
    },
    {
      "epoch": 2.72979736328125e-05,
      "model_forward_time": 3.1480469703674316,
      "step": 7156
    },
    {
      "epoch": 2.72979736328125e-05,
      "step": 7156,
      "training_step_time": 9.925236701965332
    },
    {
      "epoch": 2.72979736328125e-05,
      "model_forward_time": 3.3990399837493896,
      "step": 7156
    },
    {
      "epoch": 2.72979736328125e-05,
      "step": 7156,
      "training_step_time": 10.172828674316406
    },
    {
      "epoch": 2.72979736328125e-05,
      "model_forward_time": 3.136807441711426,
      "step": 7156
    },
    {
      "epoch": 2.72979736328125e-05,
      "step": 7156,
      "training_step_time": 9.912413120269775
    },
    {
      "epoch": 2.72979736328125e-05,
      "model_forward_time": 3.136066436767578,
      "step": 7156
    },
    {
      "epoch": 2.72979736328125e-05,
      "step": 7156,
      "training_step_time": 10.024747133255005
    },
    {
      "epoch": 2.7301788330078125e-05,
      "model_forward_time": 3.156111717224121,
      "step": 7157
    },
    {
      "epoch": 2.7301788330078125e-05,
      "step": 7157,
      "training_step_time": 9.931897640228271
    },
    {
      "epoch": 2.7301788330078125e-05,
      "model_forward_time": 3.136667013168335,
      "step": 7157
    },
    {
      "epoch": 2.7301788330078125e-05,
      "step": 7157,
      "training_step_time": 9.912519693374634
    },
    {
      "epoch": 2.7301788330078125e-05,
      "model_forward_time": 3.1422054767608643,
      "step": 7157
    },
    {
      "dynamics_loss_avg": 0.06357817687094211,
      "epoch": 2.7301788330078125e-05,
      "step": 7157
    },
    {
      "action_loss_avg": 0.016971257980912924,
      "epoch": 2.7301788330078125e-05,
      "step": 7157
    },
    {
      "epoch": 2.7301788330078125e-05,
      "step": 7157,
      "training_step_time": 10.196526765823364
    },
    {
      "epoch": 2.7301788330078125e-05,
      "model_forward_time": 3.1394705772399902,
      "step": 7157
    },
    {
      "epoch": 2.7301788330078125e-05,
      "step": 7157,
      "training_step_time": 10.299418449401855
    },
    {
      "epoch": 2.730560302734375e-05,
      "model_forward_time": 3.1874961853027344,
      "step": 7158
    },
    {
      "epoch": 2.730560302734375e-05,
      "step": 7158,
      "training_step_time": 9.96803593635559
    },
    {
      "epoch": 2.730560302734375e-05,
      "model_forward_time": 3.142974853515625,
      "step": 7158
    },
    {
      "epoch": 2.730560302734375e-05,
      "step": 7158,
      "training_step_time": 9.921125173568726
    },
    {
      "epoch": 2.730560302734375e-05,
      "model_forward_time": 3.14077091217041,
      "step": 7158
    },
    {
      "epoch": 2.730560302734375e-05,
      "step": 7158,
      "training_step_time": 9.916963338851929
    },
    {
      "epoch": 2.730560302734375e-05,
      "model_forward_time": 3.394083023071289,
      "step": 7158
    },
    {
      "epoch": 2.730560302734375e-05,
      "step": 7158,
      "training_step_time": 10.291311502456665
    },
    {
      "epoch": 2.7309417724609374e-05,
      "model_forward_time": 3.1504435539245605,
      "step": 7159
    },
    {
      "epoch": 2.7309417724609374e-05,
      "step": 7159,
      "training_step_time": 9.934350967407227
    },
    {
      "epoch": 2.7309417724609374e-05,
      "model_forward_time": 3.1425952911376953,
      "step": 7159
    },
    {
      "epoch": 2.7309417724609374e-05,
      "step": 7159,
      "training_step_time": 9.92202091217041
    },
    {
      "epoch": 2.7309417724609374e-05,
      "model_forward_time": 3.14217209815979,
      "step": 7159
    },
    {
      "epoch": 2.7309417724609374e-05,
      "step": 7159,
      "training_step_time": 9.92111587524414
    },
    {
      "epoch": 2.7309417724609374e-05,
      "model_forward_time": 3.145831346511841,
      "step": 7159
    },
    {
      "epoch": 2.7309417724609374e-05,
      "step": 7159,
      "training_step_time": 10.030643701553345
    },
    {
      "epoch": 2.7313232421875e-05,
      "grad_norm": 0.44352343678474426,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 7160
    },
    {
      "epoch": 2.7313232421875e-05,
      "model_forward_time": 3.442058563232422,
      "step": 7160
    },
    {
      "dynamics_loss_avg": 0.054845964536070824,
      "epoch": 2.7313232421875e-05,
      "step": 7160
    },
    {
      "action_loss_avg": 0.026059945300221442,
      "epoch": 2.7313232421875e-05,
      "step": 7160
    },
    {
      "epoch": 2.7313232421875e-05,
      "step": 7160,
      "training_step_time": 10.220403671264648
    },
    {
      "epoch": 2.7313232421875e-05,
      "model_forward_time": 3.144228458404541,
      "step": 7160
    },
    {
      "epoch": 2.7313232421875e-05,
      "step": 7160,
      "training_step_time": 9.922037839889526
    },
    {
      "epoch": 2.7313232421875e-05,
      "model_forward_time": 3.1366496086120605,
      "step": 7160
    },
    {
      "epoch": 2.7313232421875e-05,
      "step": 7160,
      "training_step_time": 9.91204833984375
    },
    {
      "epoch": 2.7313232421875e-05,
      "model_forward_time": 3.140314817428589,
      "step": 7160
    },
    {
      "epoch": 2.7313232421875e-05,
      "step": 7160,
      "training_step_time": 10.277209758758545
    },
    {
      "epoch": 2.7317047119140623e-05,
      "model_forward_time": 3.1765971183776855,
      "step": 7161
    },
    {
      "epoch": 2.7317047119140623e-05,
      "step": 7161,
      "training_step_time": 9.95234227180481
    },
    {
      "epoch": 2.7317047119140623e-05,
      "model_forward_time": 3.4159955978393555,
      "step": 7161
    },
    {
      "epoch": 2.7317047119140623e-05,
      "step": 7161,
      "training_step_time": 10.192853450775146
    },
    {
      "epoch": 2.7317047119140623e-05,
      "model_forward_time": 3.1376214027404785,
      "step": 7161
    },
    {
      "epoch": 2.7317047119140623e-05,
      "step": 7161,
      "training_step_time": 9.91254997253418
    },
    {
      "epoch": 2.7317047119140623e-05,
      "model_forward_time": 3.1454403400421143,
      "step": 7161
    },
    {
      "epoch": 2.7317047119140623e-05,
      "step": 7161,
      "training_step_time": 10.282382726669312
    },
    {
      "epoch": 2.732086181640625e-05,
      "model_forward_time": 3.16945743560791,
      "step": 7162
    },
    {
      "epoch": 2.732086181640625e-05,
      "step": 7162,
      "training_step_time": 9.946452379226685
    },
    {
      "epoch": 2.732086181640625e-05,
      "model_forward_time": 3.137861967086792,
      "step": 7162
    },
    {
      "epoch": 2.732086181640625e-05,
      "step": 7162,
      "training_step_time": 9.912935256958008
    },
    {
      "epoch": 2.732086181640625e-05,
      "model_forward_time": 3.3950414657592773,
      "step": 7162
    },
    {
      "dynamics_loss_avg": 0.04751283004879951,
      "epoch": 2.732086181640625e-05,
      "step": 7162
    },
    {
      "action_loss_avg": 0.016327502392232417,
      "epoch": 2.732086181640625e-05,
      "step": 7162
    },
    {
      "epoch": 2.732086181640625e-05,
      "step": 7162,
      "training_step_time": 10.181425094604492
    },
    {
      "epoch": 2.732086181640625e-05,
      "model_forward_time": 3.1458373069763184,
      "step": 7162
    },
    {
      "epoch": 2.732086181640625e-05,
      "step": 7162,
      "training_step_time": 10.325085163116455
    },
    {
      "epoch": 2.7324676513671876e-05,
      "model_forward_time": 3.155587673187256,
      "step": 7163
    },
    {
      "epoch": 2.7324676513671876e-05,
      "step": 7163,
      "training_step_time": 9.938507318496704
    },
    {
      "epoch": 2.7324676513671876e-05,
      "model_forward_time": 3.142035722732544,
      "step": 7163
    },
    {
      "epoch": 2.7324676513671876e-05,
      "step": 7163,
      "training_step_time": 9.920347452163696
    },
    {
      "epoch": 2.7324676513671876e-05,
      "model_forward_time": 3.1463260650634766,
      "step": 7163
    },
    {
      "epoch": 2.7324676513671876e-05,
      "step": 7163,
      "training_step_time": 9.926935911178589
    },
    {
      "epoch": 2.7324676513671876e-05,
      "model_forward_time": 3.4209442138671875,
      "step": 7163
    },
    {
      "epoch": 2.7324676513671876e-05,
      "step": 7163,
      "training_step_time": 10.297003746032715
    },
    {
      "epoch": 2.73284912109375e-05,
      "model_forward_time": 3.1592018604278564,
      "step": 7164
    },
    {
      "epoch": 2.73284912109375e-05,
      "step": 7164,
      "training_step_time": 9.942447900772095
    },
    {
      "epoch": 2.73284912109375e-05,
      "model_forward_time": 3.1324145793914795,
      "step": 7164
    },
    {
      "epoch": 2.73284912109375e-05,
      "step": 7164,
      "training_step_time": 9.905764818191528
    },
    {
      "epoch": 2.73284912109375e-05,
      "model_forward_time": 3.139477491378784,
      "step": 7164
    },
    {
      "epoch": 2.73284912109375e-05,
      "step": 7164,
      "training_step_time": 9.913068532943726
    },
    {
      "epoch": 2.73284912109375e-05,
      "model_forward_time": 3.132537841796875,
      "step": 7164
    },
    {
      "epoch": 2.73284912109375e-05,
      "step": 7164,
      "training_step_time": 10.02009654045105
    },
    {
      "epoch": 2.7332305908203125e-05,
      "model_forward_time": 3.4308438301086426,
      "step": 7165
    },
    {
      "dynamics_loss_avg": 0.06761856600642205,
      "epoch": 2.7332305908203125e-05,
      "step": 7165
    },
    {
      "action_loss_avg": 0.024221648182719945,
      "epoch": 2.7332305908203125e-05,
      "step": 7165
    },
    {
      "epoch": 2.7332305908203125e-05,
      "step": 7165,
      "training_step_time": 10.208874225616455
    },
    {
      "epoch": 2.7332305908203125e-05,
      "model_forward_time": 3.137193441390991,
      "step": 7165
    },
    {
      "epoch": 2.7332305908203125e-05,
      "step": 7165,
      "training_step_time": 9.906195163726807
    },
    {
      "epoch": 2.7332305908203125e-05,
      "model_forward_time": 3.1413047313690186,
      "step": 7165
    },
    {
      "epoch": 2.7332305908203125e-05,
      "step": 7165,
      "training_step_time": 9.91304087638855
    },
    {
      "epoch": 2.7332305908203125e-05,
      "model_forward_time": 3.1347084045410156,
      "step": 7165
    },
    {
      "epoch": 2.7332305908203125e-05,
      "step": 7165,
      "training_step_time": 10.248060703277588
    },
    {
      "epoch": 2.733612060546875e-05,
      "model_forward_time": 3.1539909839630127,
      "step": 7166
    },
    {
      "epoch": 2.733612060546875e-05,
      "step": 7166,
      "training_step_time": 9.922305822372437
    },
    {
      "epoch": 2.733612060546875e-05,
      "model_forward_time": 3.39593768119812,
      "step": 7166
    },
    {
      "epoch": 2.733612060546875e-05,
      "step": 7166,
      "training_step_time": 10.167367219924927
    },
    {
      "epoch": 2.733612060546875e-05,
      "model_forward_time": 3.141444683074951,
      "step": 7166
    },
    {
      "epoch": 2.733612060546875e-05,
      "step": 7166,
      "training_step_time": 9.912145376205444
    },
    {
      "epoch": 2.733612060546875e-05,
      "model_forward_time": 3.1380064487457275,
      "step": 7166
    },
    {
      "epoch": 2.733612060546875e-05,
      "step": 7166,
      "training_step_time": 10.329199314117432
    },
    {
      "epoch": 2.7339935302734374e-05,
      "model_forward_time": 3.159548759460449,
      "step": 7167
    },
    {
      "epoch": 2.7339935302734374e-05,
      "step": 7167,
      "training_step_time": 9.93737244606018
    },
    {
      "epoch": 2.7339935302734374e-05,
      "model_forward_time": 3.145836591720581,
      "step": 7167
    },
    {
      "epoch": 2.7339935302734374e-05,
      "step": 7167,
      "training_step_time": 9.926559448242188
    },
    {
      "epoch": 2.7339935302734374e-05,
      "model_forward_time": 3.424116849899292,
      "step": 7167
    },
    {
      "dynamics_loss_avg": 0.04645775519311428,
      "epoch": 2.7339935302734374e-05,
      "step": 7167
    },
    {
      "action_loss_avg": 0.0201308264862746,
      "epoch": 2.7339935302734374e-05,
      "step": 7167
    },
    {
      "epoch": 2.7339935302734374e-05,
      "step": 7167,
      "training_step_time": 10.207298040390015
    },
    {
      "epoch": 2.7339935302734374e-05,
      "model_forward_time": 2.4401462078094482,
      "step": 7167
    },
    {
      "epoch": 2.7339935302734374e-05,
      "step": 7167,
      "training_step_time": 10.29973292350769
    },
    {
      "epoch": 2.734375e-05,
      "model_forward_time": 3.2041585445404053,
      "step": 7168
    },
    {
      "epoch": 2.734375e-05,
      "step": 7168,
      "training_step_time": 9.984345436096191
    },
    {
      "epoch": 2.734375e-05,
      "model_forward_time": 3.150026321411133,
      "step": 7168
    },
    {
      "epoch": 2.734375e-05,
      "step": 7168,
      "training_step_time": 9.932308197021484
    },
    {
      "epoch": 2.734375e-05,
      "model_forward_time": 3.14538836479187,
      "step": 7168
    },
    {
      "epoch": 2.734375e-05,
      "step": 7168,
      "training_step_time": 9.924084901809692
    },
    {
      "epoch": 2.734375e-05,
      "model_forward_time": 3.4646146297454834,
      "step": 7168
    },
    {
      "epoch": 2.734375e-05,
      "step": 7168,
      "training_step_time": 10.340588569641113
    },
    {
      "epoch": 2.7347564697265623e-05,
      "model_forward_time": 3.173461437225342,
      "step": 7169
    },
    {
      "epoch": 2.7347564697265623e-05,
      "step": 7169,
      "training_step_time": 9.951317548751831
    },
    {
      "epoch": 2.7347564697265623e-05,
      "model_forward_time": 2.4405622482299805,
      "step": 7169
    },
    {
      "epoch": 2.7347564697265623e-05,
      "step": 7169,
      "training_step_time": 7.473222970962524
    },
    {
      "epoch": 2.7347564697265623e-05,
      "model_forward_time": 3.1370997428894043,
      "step": 7169
    },
    {
      "epoch": 2.7347564697265623e-05,
      "step": 7169,
      "training_step_time": 9.90910792350769
    },
    {
      "epoch": 2.7347564697265623e-05,
      "model_forward_time": 3.1478729248046875,
      "step": 7169
    },
    {
      "epoch": 2.7347564697265623e-05,
      "step": 7169,
      "training_step_time": 10.01871633529663
    },
    {
      "epoch": 2.735137939453125e-05,
      "grad_norm": 0.4991455078125,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 7170
    },
    {
      "epoch": 2.735137939453125e-05,
      "model_forward_time": 2.466366767883301,
      "step": 7170
    },
    {
      "dynamics_loss_avg": 0.053763395827263596,
      "epoch": 2.735137939453125e-05,
      "step": 7170
    },
    {
      "action_loss_avg": 0.016548922029323877,
      "epoch": 2.735137939453125e-05,
      "step": 7170
    },
    {
      "epoch": 2.735137939453125e-05,
      "step": 7170,
      "training_step_time": 7.783069372177124
    },
    {
      "epoch": 2.735137939453125e-05,
      "model_forward_time": 2.433072090148926,
      "step": 7170
    },
    {
      "epoch": 2.735137939453125e-05,
      "step": 7170,
      "training_step_time": 7.460038900375366
    },
    {
      "epoch": 2.735137939453125e-05,
      "model_forward_time": 3.1376683712005615,
      "step": 7170
    },
    {
      "epoch": 2.735137939453125e-05,
      "step": 7170,
      "training_step_time": 9.909168481826782
    },
    {
      "epoch": 2.735137939453125e-05,
      "model_forward_time": 3.1433212757110596,
      "step": 7170
    },
    {
      "epoch": 2.735137939453125e-05,
      "step": 7170,
      "training_step_time": 10.232175827026367
    },
    {
      "epoch": 2.7355194091796876e-05,
      "model_forward_time": 3.1597516536712646,
      "step": 7171
    },
    {
      "epoch": 2.7355194091796876e-05,
      "step": 7171,
      "training_step_time": 9.938586473464966
    },
    {
      "epoch": 2.7355194091796876e-05,
      "model_forward_time": 3.1419472694396973,
      "step": 7171
    },
    {
      "epoch": 2.7355194091796876e-05,
      "step": 7171,
      "training_step_time": 10.18326187133789
    },
    {
      "epoch": 2.7355194091796876e-05,
      "model_forward_time": 3.140638828277588,
      "step": 7171
    },
    {
      "epoch": 2.7355194091796876e-05,
      "step": 7171,
      "training_step_time": 9.91612696647644
    },
    {
      "epoch": 2.7355194091796876e-05,
      "model_forward_time": 3.145321846008301,
      "step": 7171
    },
    {
      "epoch": 2.7355194091796876e-05,
      "step": 7171,
      "training_step_time": 10.266477108001709
    },
    {
      "epoch": 2.73590087890625e-05,
      "model_forward_time": 3.187178134918213,
      "step": 7172
    },
    {
      "epoch": 2.73590087890625e-05,
      "step": 7172,
      "training_step_time": 9.966556072235107
    },
    {
      "epoch": 2.73590087890625e-05,
      "model_forward_time": 3.1416022777557373,
      "step": 7172
    },
    {
      "epoch": 2.73590087890625e-05,
      "step": 7172,
      "training_step_time": 9.911768436431885
    },
    {
      "epoch": 2.73590087890625e-05,
      "model_forward_time": 3.437657594680786,
      "step": 7172
    },
    {
      "dynamics_loss_avg": 0.06127544119954109,
      "epoch": 2.73590087890625e-05,
      "step": 7172
    },
    {
      "action_loss_avg": 0.026037716679275034,
      "epoch": 2.73590087890625e-05,
      "step": 7172
    },
    {
      "epoch": 2.73590087890625e-05,
      "step": 7172,
      "training_step_time": 10.212367057800293
    },
    {
      "epoch": 2.73590087890625e-05,
      "model_forward_time": 3.145303964614868,
      "step": 7172
    },
    {
      "epoch": 2.73590087890625e-05,
      "step": 7172,
      "training_step_time": 10.344614744186401
    },
    {
      "epoch": 2.7362823486328125e-05,
      "model_forward_time": 3.183840036392212,
      "step": 7173
    },
    {
      "epoch": 2.7362823486328125e-05,
      "step": 7173,
      "training_step_time": 9.959125757217407
    },
    {
      "epoch": 2.7362823486328125e-05,
      "model_forward_time": 2.437757730484009,
      "step": 7173
    },
    {
      "epoch": 2.7362823486328125e-05,
      "step": 7173,
      "training_step_time": 7.4628589153289795
    },
    {
      "epoch": 2.7362823486328125e-05,
      "model_forward_time": 3.1399197578430176,
      "step": 7173
    },
    {
      "epoch": 2.7362823486328125e-05,
      "step": 7173,
      "training_step_time": 9.911484241485596
    },
    {
      "epoch": 2.7362823486328125e-05,
      "model_forward_time": 3.4419054985046387,
      "step": 7173
    },
    {
      "epoch": 2.7362823486328125e-05,
      "step": 7173,
      "training_step_time": 10.313690900802612
    },
    {
      "epoch": 2.736663818359375e-05,
      "model_forward_time": 3.191373825073242,
      "step": 7174
    },
    {
      "epoch": 2.736663818359375e-05,
      "step": 7174,
      "training_step_time": 9.96778130531311
    },
    {
      "epoch": 2.736663818359375e-05,
      "model_forward_time": 3.144785165786743,
      "step": 7174
    },
    {
      "epoch": 2.736663818359375e-05,
      "step": 7174,
      "training_step_time": 9.91990041732788
    },
    {
      "epoch": 2.736663818359375e-05,
      "model_forward_time": 3.147531747817993,
      "step": 7174
    },
    {
      "epoch": 2.736663818359375e-05,
      "step": 7174,
      "training_step_time": 9.918949604034424
    },
    {
      "epoch": 2.736663818359375e-05,
      "model_forward_time": 3.1400113105773926,
      "step": 7174
    },
    {
      "epoch": 2.736663818359375e-05,
      "step": 7174,
      "training_step_time": 10.014785289764404
    },
    {
      "epoch": 2.7370452880859374e-05,
      "model_forward_time": 3.173809766769409,
      "step": 7175
    },
    {
      "dynamics_loss_avg": 0.05681977290660143,
      "epoch": 2.7370452880859374e-05,
      "step": 7175
    },
    {
      "action_loss_avg": 0.02384288739413023,
      "epoch": 2.7370452880859374e-05,
      "step": 7175
    },
    {
      "epoch": 2.7370452880859374e-05,
      "step": 7175,
      "training_step_time": 10.222310304641724
    },
    {
      "epoch": 2.7370452880859374e-05,
      "model_forward_time": 3.13340163230896,
      "step": 7175
    },
    {
      "epoch": 2.7370452880859374e-05,
      "step": 7175,
      "training_step_time": 9.903818368911743
    },
    {
      "epoch": 2.7370452880859374e-05,
      "model_forward_time": 3.134643316268921,
      "step": 7175
    },
    {
      "epoch": 2.7370452880859374e-05,
      "step": 7175,
      "training_step_time": 9.908325910568237
    },
    {
      "epoch": 2.7370452880859374e-05,
      "model_forward_time": 3.1339142322540283,
      "step": 7175
    },
    {
      "epoch": 2.7370452880859374e-05,
      "step": 7175,
      "training_step_time": 10.296406030654907
    },
    {
      "epoch": 2.7374267578125e-05,
      "model_forward_time": 3.1722915172576904,
      "step": 7176
    },
    {
      "epoch": 2.7374267578125e-05,
      "step": 7176,
      "training_step_time": 9.946956872940063
    },
    {
      "epoch": 2.7374267578125e-05,
      "model_forward_time": 3.391382932662964,
      "step": 7176
    },
    {
      "epoch": 2.7374267578125e-05,
      "step": 7176,
      "training_step_time": 10.160872220993042
    },
    {
      "epoch": 2.7374267578125e-05,
      "model_forward_time": 3.137859582901001,
      "step": 7176
    },
    {
      "epoch": 2.7374267578125e-05,
      "step": 7176,
      "training_step_time": 9.909816026687622
    },
    {
      "epoch": 2.7374267578125e-05,
      "model_forward_time": 3.1313090324401855,
      "step": 7176
    },
    {
      "epoch": 2.7374267578125e-05,
      "step": 7176,
      "training_step_time": 10.234660148620605
    },
    {
      "epoch": 2.7378082275390623e-05,
      "model_forward_time": 3.1795942783355713,
      "step": 7177
    },
    {
      "epoch": 2.7378082275390623e-05,
      "step": 7177,
      "training_step_time": 9.949304103851318
    },
    {
      "epoch": 2.7378082275390623e-05,
      "model_forward_time": 3.140834093093872,
      "step": 7177
    },
    {
      "epoch": 2.7378082275390623e-05,
      "step": 7177,
      "training_step_time": 9.912078619003296
    },
    {
      "epoch": 2.7378082275390623e-05,
      "model_forward_time": 3.4262630939483643,
      "step": 7177
    },
    {
      "dynamics_loss_avg": 0.055855901911854744,
      "epoch": 2.7378082275390623e-05,
      "step": 7177
    },
    {
      "action_loss_avg": 0.03028989639133215,
      "epoch": 2.7378082275390623e-05,
      "step": 7177
    },
    {
      "epoch": 2.7378082275390623e-05,
      "step": 7177,
      "training_step_time": 10.19408106803894
    },
    {
      "epoch": 2.7378082275390623e-05,
      "model_forward_time": 3.138676643371582,
      "step": 7177
    },
    {
      "epoch": 2.7378082275390623e-05,
      "step": 7177,
      "training_step_time": 10.293538093566895
    },
    {
      "epoch": 2.738189697265625e-05,
      "model_forward_time": 3.1437673568725586,
      "step": 7178
    },
    {
      "epoch": 2.738189697265625e-05,
      "step": 7178,
      "training_step_time": 9.917401552200317
    },
    {
      "epoch": 2.738189697265625e-05,
      "model_forward_time": 3.140601396560669,
      "step": 7178
    },
    {
      "epoch": 2.738189697265625e-05,
      "step": 7178,
      "training_step_time": 9.911320447921753
    },
    {
      "epoch": 2.738189697265625e-05,
      "model_forward_time": 3.140810012817383,
      "step": 7178
    },
    {
      "epoch": 2.738189697265625e-05,
      "step": 7178,
      "training_step_time": 9.919105291366577
    },
    {
      "epoch": 2.738189697265625e-05,
      "model_forward_time": 3.434380531311035,
      "step": 7178
    },
    {
      "epoch": 2.738189697265625e-05,
      "step": 7178,
      "training_step_time": 10.304404735565186
    },
    {
      "epoch": 2.7385711669921876e-05,
      "model_forward_time": 3.187371015548706,
      "step": 7179
    },
    {
      "epoch": 2.7385711669921876e-05,
      "step": 7179,
      "training_step_time": 9.961670160293579
    },
    {
      "epoch": 2.7385711669921876e-05,
      "model_forward_time": 3.143477201461792,
      "step": 7179
    },
    {
      "epoch": 2.7385711669921876e-05,
      "step": 7179,
      "training_step_time": 9.923079490661621
    },
    {
      "epoch": 2.7385711669921876e-05,
      "model_forward_time": 3.149306058883667,
      "step": 7179
    },
    {
      "epoch": 2.7385711669921876e-05,
      "step": 7179,
      "training_step_time": 9.927682638168335
    },
    {
      "epoch": 2.7385711669921876e-05,
      "model_forward_time": 3.143402576446533,
      "step": 7179
    },
    {
      "epoch": 2.7385711669921876e-05,
      "step": 7179,
      "training_step_time": 10.029321193695068
    },
    {
      "epoch": 2.73895263671875e-05,
      "grad_norm": 0.21601232886314392,
      "learning_rate": 1e-05,
      "loss": 0.0856,
      "step": 7180
    },
    {
      "epoch": 2.73895263671875e-05,
      "model_forward_time": 3.425365686416626,
      "step": 7180
    },
    {
      "dynamics_loss_avg": 0.060963697731494904,
      "epoch": 2.73895263671875e-05,
      "step": 7180
    },
    {
      "action_loss_avg": 0.021270344406366347,
      "epoch": 2.73895263671875e-05,
      "step": 7180
    },
    {
      "epoch": 2.73895263671875e-05,
      "step": 7180,
      "training_step_time": 10.212062358856201
    },
    {
      "epoch": 2.73895263671875e-05,
      "model_forward_time": 3.1483988761901855,
      "step": 7180
    },
    {
      "epoch": 2.73895263671875e-05,
      "step": 7180,
      "training_step_time": 9.926597595214844
    },
    {
      "epoch": 2.73895263671875e-05,
      "model_forward_time": 3.13618803024292,
      "step": 7180
    },
    {
      "epoch": 2.73895263671875e-05,
      "step": 7180,
      "training_step_time": 9.911068201065063
    },
    {
      "epoch": 2.73895263671875e-05,
      "model_forward_time": 3.136936902999878,
      "step": 7180
    },
    {
      "epoch": 2.73895263671875e-05,
      "step": 7180,
      "training_step_time": 10.22568678855896
    },
    {
      "epoch": 2.7393341064453125e-05,
      "model_forward_time": 3.155318260192871,
      "step": 7181
    },
    {
      "epoch": 2.7393341064453125e-05,
      "step": 7181,
      "training_step_time": 9.929020643234253
    },
    {
      "epoch": 2.7393341064453125e-05,
      "model_forward_time": 3.138183116912842,
      "step": 7181
    },
    {
      "epoch": 2.7393341064453125e-05,
      "step": 7181,
      "training_step_time": 10.200510025024414
    },
    {
      "epoch": 2.7393341064453125e-05,
      "model_forward_time": 3.138166904449463,
      "step": 7181
    },
    {
      "epoch": 2.7393341064453125e-05,
      "step": 7181,
      "training_step_time": 9.909639596939087
    },
    {
      "epoch": 2.7393341064453125e-05,
      "model_forward_time": 3.13686203956604,
      "step": 7181
    },
    {
      "epoch": 2.7393341064453125e-05,
      "step": 7181,
      "training_step_time": 10.27664041519165
    },
    {
      "epoch": 2.739715576171875e-05,
      "model_forward_time": 3.1742653846740723,
      "step": 7182
    },
    {
      "epoch": 2.739715576171875e-05,
      "step": 7182,
      "training_step_time": 9.947299242019653
    },
    {
      "epoch": 2.739715576171875e-05,
      "model_forward_time": 3.1346280574798584,
      "step": 7182
    },
    {
      "epoch": 2.739715576171875e-05,
      "step": 7182,
      "training_step_time": 9.903883934020996
    },
    {
      "epoch": 2.739715576171875e-05,
      "model_forward_time": 3.3998830318450928,
      "step": 7182
    },
    {
      "dynamics_loss_avg": 0.06266771275550127,
      "epoch": 2.739715576171875e-05,
      "step": 7182
    },
    {
      "action_loss_avg": 0.02430526353418827,
      "epoch": 2.739715576171875e-05,
      "step": 7182
    },
    {
      "epoch": 2.739715576171875e-05,
      "step": 7182,
      "training_step_time": 10.172704219818115
    },
    {
      "epoch": 2.739715576171875e-05,
      "model_forward_time": 3.143954038619995,
      "step": 7182
    },
    {
      "epoch": 2.739715576171875e-05,
      "step": 7182,
      "training_step_time": 10.258492231369019
    },
    {
      "epoch": 2.7400970458984374e-05,
      "model_forward_time": 3.153191328048706,
      "step": 7183
    },
    {
      "epoch": 2.7400970458984374e-05,
      "step": 7183,
      "training_step_time": 9.926894187927246
    },
    {
      "epoch": 2.7400970458984374e-05,
      "model_forward_time": 3.134230852127075,
      "step": 7183
    },
    {
      "epoch": 2.7400970458984374e-05,
      "step": 7183,
      "training_step_time": 9.90808916091919
    },
    {
      "epoch": 2.7400970458984374e-05,
      "model_forward_time": 3.135525703430176,
      "step": 7183
    },
    {
      "epoch": 2.7400970458984374e-05,
      "step": 7183,
      "training_step_time": 9.91328763961792
    },
    {
      "epoch": 2.7400970458984374e-05,
      "model_forward_time": 3.1372241973876953,
      "step": 7183
    },
    {
      "epoch": 2.7400970458984374e-05,
      "step": 7183,
      "training_step_time": 10.29757833480835
    },
    {
      "epoch": 2.740478515625e-05,
      "model_forward_time": 3.1558690071105957,
      "step": 7184
    },
    {
      "epoch": 2.740478515625e-05,
      "step": 7184,
      "training_step_time": 9.934258699417114
    },
    {
      "epoch": 2.740478515625e-05,
      "model_forward_time": 3.1393394470214844,
      "step": 7184
    },
    {
      "epoch": 2.740478515625e-05,
      "step": 7184,
      "training_step_time": 9.91727900505066
    },
    {
      "epoch": 2.740478515625e-05,
      "model_forward_time": 3.1415231227874756,
      "step": 7184
    },
    {
      "epoch": 2.740478515625e-05,
      "step": 7184,
      "training_step_time": 9.919015645980835
    },
    {
      "epoch": 2.740478515625e-05,
      "model_forward_time": 3.138977289199829,
      "step": 7184
    },
    {
      "epoch": 2.740478515625e-05,
      "step": 7184,
      "training_step_time": 10.023242712020874
    },
    {
      "epoch": 2.7408599853515624e-05,
      "model_forward_time": 3.467711925506592,
      "step": 7185
    },
    {
      "dynamics_loss_avg": 0.05695826653391123,
      "epoch": 2.7408599853515624e-05,
      "step": 7185
    },
    {
      "action_loss_avg": 0.020055192848667504,
      "epoch": 2.7408599853515624e-05,
      "step": 7185
    },
    {
      "epoch": 2.7408599853515624e-05,
      "step": 7185,
      "training_step_time": 10.245078086853027
    },
    {
      "epoch": 2.7408599853515624e-05,
      "model_forward_time": 3.1379687786102295,
      "step": 7185
    },
    {
      "epoch": 2.7408599853515624e-05,
      "step": 7185,
      "training_step_time": 9.911070346832275
    },
    {
      "epoch": 2.7408599853515624e-05,
      "model_forward_time": 3.14034104347229,
      "step": 7185
    },
    {
      "epoch": 2.7408599853515624e-05,
      "step": 7185,
      "training_step_time": 9.91204309463501
    },
    {
      "epoch": 2.7408599853515624e-05,
      "model_forward_time": 3.1394944190979004,
      "step": 7185
    },
    {
      "epoch": 2.7408599853515624e-05,
      "step": 7185,
      "training_step_time": 10.282350540161133
    },
    {
      "epoch": 2.741241455078125e-05,
      "model_forward_time": 3.1504287719726562,
      "step": 7186
    },
    {
      "epoch": 2.741241455078125e-05,
      "step": 7186,
      "training_step_time": 9.923293113708496
    },
    {
      "epoch": 2.741241455078125e-05,
      "model_forward_time": 3.1383302211761475,
      "step": 7186
    },
    {
      "epoch": 2.741241455078125e-05,
      "step": 7186,
      "training_step_time": 10.20841097831726
    },
    {
      "epoch": 2.741241455078125e-05,
      "model_forward_time": 3.1330759525299072,
      "step": 7186
    },
    {
      "epoch": 2.741241455078125e-05,
      "step": 7186,
      "training_step_time": 9.903989791870117
    },
    {
      "epoch": 2.741241455078125e-05,
      "model_forward_time": 2.4331374168395996,
      "step": 7186
    },
    {
      "epoch": 2.741241455078125e-05,
      "step": 7186,
      "training_step_time": 10.241381645202637
    },
    {
      "epoch": 2.7416229248046876e-05,
      "model_forward_time": 3.17040753364563,
      "step": 7187
    },
    {
      "epoch": 2.7416229248046876e-05,
      "step": 7187,
      "training_step_time": 9.946277141571045
    },
    {
      "epoch": 2.7416229248046876e-05,
      "model_forward_time": 3.1388328075408936,
      "step": 7187
    },
    {
      "epoch": 2.7416229248046876e-05,
      "step": 7187,
      "training_step_time": 9.92208456993103
    },
    {
      "epoch": 2.7416229248046876e-05,
      "model_forward_time": 3.4251139163970947,
      "step": 7187
    },
    {
      "dynamics_loss_avg": 0.052193901129066946,
      "epoch": 2.7416229248046876e-05,
      "step": 7187
    },
    {
      "action_loss_avg": 0.017599678901024162,
      "epoch": 2.7416229248046876e-05,
      "step": 7187
    },
    {
      "epoch": 2.7416229248046876e-05,
      "step": 7187,
      "training_step_time": 10.200034856796265
    },
    {
      "epoch": 2.7416229248046876e-05,
      "model_forward_time": 3.1350245475769043,
      "step": 7187
    },
    {
      "epoch": 2.7416229248046876e-05,
      "step": 7187,
      "training_step_time": 10.010745286941528
    },
    {
      "epoch": 2.74200439453125e-05,
      "model_forward_time": 3.162327527999878,
      "step": 7188
    },
    {
      "epoch": 2.74200439453125e-05,
      "step": 7188,
      "training_step_time": 9.936898231506348
    },
    {
      "epoch": 2.74200439453125e-05,
      "model_forward_time": 3.1397037506103516,
      "step": 7188
    },
    {
      "epoch": 2.74200439453125e-05,
      "step": 7188,
      "training_step_time": 9.917137145996094
    },
    {
      "epoch": 2.74200439453125e-05,
      "model_forward_time": 3.1305997371673584,
      "step": 7188
    },
    {
      "epoch": 2.74200439453125e-05,
      "step": 7188,
      "training_step_time": 9.904828071594238
    },
    {
      "epoch": 2.74200439453125e-05,
      "model_forward_time": 3.1374659538269043,
      "step": 7188
    },
    {
      "epoch": 2.74200439453125e-05,
      "step": 7188,
      "training_step_time": 10.28891897201538
    },
    {
      "epoch": 2.7423858642578125e-05,
      "model_forward_time": 3.4310264587402344,
      "step": 7189
    },
    {
      "epoch": 2.7423858642578125e-05,
      "step": 7189,
      "training_step_time": 10.204720735549927
    },
    {
      "epoch": 2.7423858642578125e-05,
      "model_forward_time": 3.139707326889038,
      "step": 7189
    },
    {
      "epoch": 2.7423858642578125e-05,
      "step": 7189,
      "training_step_time": 9.915343523025513
    },
    {
      "epoch": 2.7423858642578125e-05,
      "model_forward_time": 3.135258913040161,
      "step": 7189
    },
    {
      "epoch": 2.7423858642578125e-05,
      "step": 7189,
      "training_step_time": 9.912885427474976
    },
    {
      "epoch": 2.7423858642578125e-05,
      "model_forward_time": 3.1371288299560547,
      "step": 7189
    },
    {
      "epoch": 2.7423858642578125e-05,
      "step": 7189,
      "training_step_time": 10.311930418014526
    },
    {
      "epoch": 2.742767333984375e-05,
      "grad_norm": 0.6165053844451904,
      "learning_rate": 1e-05,
      "loss": 0.0837,
      "step": 7190
    },
    {
      "epoch": 2.742767333984375e-05,
      "model_forward_time": 3.410320281982422,
      "step": 7190
    },
    {
      "dynamics_loss_avg": 0.06114410776644945,
      "epoch": 2.742767333984375e-05,
      "step": 7190
    },
    {
      "action_loss_avg": 0.01839275248348713,
      "epoch": 2.742767333984375e-05,
      "step": 7190
    },
    {
      "epoch": 2.742767333984375e-05,
      "step": 7190,
      "training_step_time": 10.193308591842651
    },
    {
      "epoch": 2.742767333984375e-05,
      "model_forward_time": 3.141178846359253,
      "step": 7190
    },
    {
      "epoch": 2.742767333984375e-05,
      "step": 7190,
      "training_step_time": 9.918736457824707
    },
    {
      "epoch": 2.742767333984375e-05,
      "model_forward_time": 3.1366584300994873,
      "step": 7190
    },
    {
      "epoch": 2.742767333984375e-05,
      "step": 7190,
      "training_step_time": 9.914368867874146
    },
    {
      "epoch": 2.742767333984375e-05,
      "model_forward_time": 3.1418213844299316,
      "step": 7190
    },
    {
      "epoch": 2.742767333984375e-05,
      "step": 7190,
      "training_step_time": 10.242273569107056
    },
    {
      "epoch": 2.7431488037109374e-05,
      "model_forward_time": 3.1770660877227783,
      "step": 7191
    },
    {
      "epoch": 2.7431488037109374e-05,
      "step": 7191,
      "training_step_time": 9.957983016967773
    },
    {
      "epoch": 2.7431488037109374e-05,
      "model_forward_time": 3.137179374694824,
      "step": 7191
    },
    {
      "epoch": 2.7431488037109374e-05,
      "step": 7191,
      "training_step_time": 10.189889669418335
    },
    {
      "epoch": 2.7431488037109374e-05,
      "model_forward_time": 3.1386020183563232,
      "step": 7191
    },
    {
      "epoch": 2.7431488037109374e-05,
      "step": 7191,
      "training_step_time": 9.912735939025879
    },
    {
      "epoch": 2.7431488037109374e-05,
      "model_forward_time": 3.1366729736328125,
      "step": 7191
    },
    {
      "epoch": 2.7431488037109374e-05,
      "step": 7191,
      "training_step_time": 10.255915641784668
    },
    {
      "epoch": 2.7435302734375e-05,
      "model_forward_time": 3.151679754257202,
      "step": 7192
    },
    {
      "epoch": 2.7435302734375e-05,
      "step": 7192,
      "training_step_time": 9.925598382949829
    },
    {
      "epoch": 2.7435302734375e-05,
      "model_forward_time": 3.129953145980835,
      "step": 7192
    },
    {
      "epoch": 2.7435302734375e-05,
      "step": 7192,
      "training_step_time": 9.907465934753418
    },
    {
      "epoch": 2.7435302734375e-05,
      "model_forward_time": 2.4338905811309814,
      "step": 7192
    },
    {
      "dynamics_loss_avg": 0.048242265824228525,
      "epoch": 2.7435302734375e-05,
      "step": 7192
    },
    {
      "action_loss_avg": 0.018460229691118,
      "epoch": 2.7435302734375e-05,
      "step": 7192
    },
    {
      "epoch": 2.7435302734375e-05,
      "step": 7192,
      "training_step_time": 7.734243154525757
    },
    {
      "epoch": 2.7435302734375e-05,
      "model_forward_time": 3.133620023727417,
      "step": 7192
    },
    {
      "epoch": 2.7435302734375e-05,
      "step": 7192,
      "training_step_time": 10.002985715866089
    },
    {
      "epoch": 2.7439117431640624e-05,
      "model_forward_time": 3.1767570972442627,
      "step": 7193
    },
    {
      "epoch": 2.7439117431640624e-05,
      "step": 7193,
      "training_step_time": 9.948731422424316
    },
    {
      "epoch": 2.7439117431640624e-05,
      "model_forward_time": 3.1297948360443115,
      "step": 7193
    },
    {
      "epoch": 2.7439117431640624e-05,
      "step": 7193,
      "training_step_time": 9.898711442947388
    },
    {
      "epoch": 2.7439117431640624e-05,
      "model_forward_time": 3.1366961002349854,
      "step": 7193
    },
    {
      "epoch": 2.7439117431640624e-05,
      "step": 7193,
      "training_step_time": 9.905866384506226
    },
    {
      "epoch": 2.7439117431640624e-05,
      "model_forward_time": 3.4138689041137695,
      "step": 7193
    },
    {
      "epoch": 2.7439117431640624e-05,
      "step": 7193,
      "training_step_time": 10.309314966201782
    },
    {
      "epoch": 2.744293212890625e-05,
      "model_forward_time": 3.156313180923462,
      "step": 7194
    },
    {
      "epoch": 2.744293212890625e-05,
      "step": 7194,
      "training_step_time": 9.935464859008789
    },
    {
      "epoch": 2.744293212890625e-05,
      "model_forward_time": 3.136286973953247,
      "step": 7194
    },
    {
      "epoch": 2.744293212890625e-05,
      "step": 7194,
      "training_step_time": 9.911654949188232
    },
    {
      "epoch": 2.744293212890625e-05,
      "model_forward_time": 3.138054132461548,
      "step": 7194
    },
    {
      "epoch": 2.744293212890625e-05,
      "step": 7194,
      "training_step_time": 9.909857511520386
    },
    {
      "epoch": 2.744293212890625e-05,
      "model_forward_time": 3.134330987930298,
      "step": 7194
    },
    {
      "epoch": 2.744293212890625e-05,
      "step": 7194,
      "training_step_time": 10.30936884880066
    },
    {
      "epoch": 2.7446746826171876e-05,
      "model_forward_time": 3.1690573692321777,
      "step": 7195
    },
    {
      "dynamics_loss_avg": 0.05571407657116652,
      "epoch": 2.7446746826171876e-05,
      "step": 7195
    },
    {
      "action_loss_avg": 0.022976226452738047,
      "epoch": 2.7446746826171876e-05,
      "step": 7195
    },
    {
      "epoch": 2.7446746826171876e-05,
      "step": 7195,
      "training_step_time": 10.222078800201416
    },
    {
      "epoch": 2.7446746826171876e-05,
      "model_forward_time": 3.1343886852264404,
      "step": 7195
    },
    {
      "epoch": 2.7446746826171876e-05,
      "step": 7195,
      "training_step_time": 9.912180662155151
    },
    {
      "epoch": 2.7446746826171876e-05,
      "model_forward_time": 3.1364502906799316,
      "step": 7195
    },
    {
      "epoch": 2.7446746826171876e-05,
      "step": 7195,
      "training_step_time": 9.909178018569946
    },
    {
      "epoch": 2.7446746826171876e-05,
      "model_forward_time": 3.1311862468719482,
      "step": 7195
    },
    {
      "epoch": 2.7446746826171876e-05,
      "step": 7195,
      "training_step_time": 10.00740671157837
    },
    {
      "epoch": 2.74505615234375e-05,
      "model_forward_time": 3.1716256141662598,
      "step": 7196
    },
    {
      "epoch": 2.74505615234375e-05,
      "step": 7196,
      "training_step_time": 9.94611406326294
    },
    {
      "epoch": 2.74505615234375e-05,
      "model_forward_time": 3.4025447368621826,
      "step": 7196
    },
    {
      "epoch": 2.74505615234375e-05,
      "step": 7196,
      "training_step_time": 10.171695947647095
    },
    {
      "epoch": 2.74505615234375e-05,
      "model_forward_time": 3.1356494426727295,
      "step": 7196
    },
    {
      "epoch": 2.74505615234375e-05,
      "step": 7196,
      "training_step_time": 9.900846481323242
    },
    {
      "epoch": 2.74505615234375e-05,
      "model_forward_time": 3.130519151687622,
      "step": 7196
    },
    {
      "epoch": 2.74505615234375e-05,
      "step": 7196,
      "training_step_time": 10.27816128730774
    },
    {
      "epoch": 2.7454376220703125e-05,
      "model_forward_time": 3.137871503829956,
      "step": 7197
    },
    {
      "epoch": 2.7454376220703125e-05,
      "step": 7197,
      "training_step_time": 9.904781579971313
    },
    {
      "epoch": 2.7454376220703125e-05,
      "model_forward_time": 3.131157875061035,
      "step": 7197
    },
    {
      "epoch": 2.7454376220703125e-05,
      "step": 7197,
      "training_step_time": 9.897643566131592
    },
    {
      "epoch": 2.7454376220703125e-05,
      "model_forward_time": 3.446270227432251,
      "step": 7197
    },
    {
      "dynamics_loss_avg": 0.051695529744029044,
      "epoch": 2.7454376220703125e-05,
      "step": 7197
    },
    {
      "action_loss_avg": 0.021325863618403672,
      "epoch": 2.7454376220703125e-05,
      "step": 7197
    },
    {
      "epoch": 2.7454376220703125e-05,
      "step": 7197,
      "training_step_time": 10.213896036148071
    },
    {
      "epoch": 2.7454376220703125e-05,
      "model_forward_time": 3.1319615840911865,
      "step": 7197
    },
    {
      "epoch": 2.7454376220703125e-05,
      "step": 7197,
      "training_step_time": 10.01409125328064
    },
    {
      "epoch": 2.745819091796875e-05,
      "model_forward_time": 3.142268657684326,
      "step": 7198
    },
    {
      "epoch": 2.745819091796875e-05,
      "step": 7198,
      "training_step_time": 9.919122219085693
    },
    {
      "epoch": 2.745819091796875e-05,
      "model_forward_time": 3.138338565826416,
      "step": 7198
    },
    {
      "epoch": 2.745819091796875e-05,
      "step": 7198,
      "training_step_time": 9.915738105773926
    },
    {
      "epoch": 2.745819091796875e-05,
      "model_forward_time": 3.1379570960998535,
      "step": 7198
    },
    {
      "epoch": 2.745819091796875e-05,
      "step": 7198,
      "training_step_time": 9.917282819747925
    },
    {
      "epoch": 2.745819091796875e-05,
      "model_forward_time": 3.4010884761810303,
      "step": 7198
    },
    {
      "epoch": 2.745819091796875e-05,
      "step": 7198,
      "training_step_time": 10.270967721939087
    },
    {
      "epoch": 2.7462005615234374e-05,
      "model_forward_time": 3.14630126953125,
      "step": 7199
    },
    {
      "epoch": 2.7462005615234374e-05,
      "step": 7199,
      "training_step_time": 9.925251722335815
    },
    {
      "epoch": 2.7462005615234374e-05,
      "model_forward_time": 3.139616012573242,
      "step": 7199
    },
    {
      "epoch": 2.7462005615234374e-05,
      "step": 7199,
      "training_step_time": 9.918628454208374
    },
    {
      "epoch": 2.7462005615234374e-05,
      "model_forward_time": 3.137073516845703,
      "step": 7199
    },
    {
      "epoch": 2.7462005615234374e-05,
      "step": 7199,
      "training_step_time": 9.919081449508667
    },
    {
      "epoch": 2.7462005615234374e-05,
      "model_forward_time": 3.145751714706421,
      "step": 7199
    },
    {
      "epoch": 2.7462005615234374e-05,
      "step": 7199,
      "training_step_time": 10.30907917022705
    },
    {
      "epoch": 2.74658203125e-05,
      "grad_norm": 0.5143951773643494,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 7200
    },
    {
      "epoch": 2.74658203125e-05,
      "model_forward_time": 3.4325602054595947,
      "step": 7200
    },
    {
      "dynamics_loss_avg": 0.04466585889458656,
      "epoch": 2.74658203125e-05,
      "step": 7200
    },
    {
      "action_loss_avg": 0.015996209951117633,
      "epoch": 2.74658203125e-05,
      "step": 7200
    },
    {
      "epoch": 2.74658203125e-05,
      "step": 7200,
      "training_step_time": 10.214911699295044
    },
    {
      "epoch": 2.74658203125e-05,
      "model_forward_time": 3.1406121253967285,
      "step": 7200
    },
    {
      "epoch": 2.74658203125e-05,
      "step": 7200,
      "training_step_time": 9.920366525650024
    },
    {
      "epoch": 2.74658203125e-05,
      "model_forward_time": 3.1423254013061523,
      "step": 7200
    },
    {
      "epoch": 2.74658203125e-05,
      "step": 7200,
      "training_step_time": 9.921721696853638
    },
    {
      "epoch": 2.74658203125e-05,
      "model_forward_time": 3.1472649574279785,
      "step": 7200
    },
    {
      "epoch": 2.74658203125e-05,
      "step": 7200,
      "training_step_time": 10.02826476097107
    },
    {
      "epoch": 2.7469635009765624e-05,
      "model_forward_time": 3.178997039794922,
      "step": 7201
    },
    {
      "epoch": 2.7469635009765624e-05,
      "step": 7201,
      "training_step_time": 9.965295791625977
    },
    {
      "epoch": 2.7469635009765624e-05,
      "model_forward_time": 3.1359524726867676,
      "step": 7201
    },
    {
      "epoch": 2.7469635009765624e-05,
      "step": 7201,
      "training_step_time": 10.193827629089355
    },
    {
      "epoch": 2.7469635009765624e-05,
      "model_forward_time": 3.134504795074463,
      "step": 7201
    },
    {
      "epoch": 2.7469635009765624e-05,
      "step": 7201,
      "training_step_time": 9.905274152755737
    },
    {
      "epoch": 2.7469635009765624e-05,
      "model_forward_time": 3.1336987018585205,
      "step": 7201
    },
    {
      "epoch": 2.7469635009765624e-05,
      "step": 7201,
      "training_step_time": 10.259533882141113
    },
    {
      "epoch": 2.747344970703125e-05,
      "model_forward_time": 3.1723759174346924,
      "step": 7202
    },
    {
      "epoch": 2.747344970703125e-05,
      "step": 7202,
      "training_step_time": 9.945101737976074
    },
    {
      "epoch": 2.747344970703125e-05,
      "model_forward_time": 3.134810447692871,
      "step": 7202
    },
    {
      "epoch": 2.747344970703125e-05,
      "step": 7202,
      "training_step_time": 9.909310102462769
    },
    {
      "epoch": 2.747344970703125e-05,
      "model_forward_time": 3.137817859649658,
      "step": 7202
    },
    {
      "dynamics_loss_avg": 0.058693009056150915,
      "epoch": 2.747344970703125e-05,
      "step": 7202
    },
    {
      "action_loss_avg": 0.02143205045722425,
      "epoch": 2.747344970703125e-05,
      "step": 7202
    },
    {
      "epoch": 2.747344970703125e-05,
      "step": 7202,
      "training_step_time": 10.191521406173706
    },
    {
      "epoch": 2.747344970703125e-05,
      "model_forward_time": 3.1359102725982666,
      "step": 7202
    },
    {
      "epoch": 2.747344970703125e-05,
      "step": 7202,
      "training_step_time": 10.01040267944336
    },
    {
      "epoch": 2.7477264404296876e-05,
      "model_forward_time": 3.176255702972412,
      "step": 7203
    },
    {
      "epoch": 2.7477264404296876e-05,
      "step": 7203,
      "training_step_time": 9.949164628982544
    },
    {
      "epoch": 2.7477264404296876e-05,
      "model_forward_time": 3.128718376159668,
      "step": 7203
    },
    {
      "epoch": 2.7477264404296876e-05,
      "step": 7203,
      "training_step_time": 9.903493881225586
    },
    {
      "epoch": 2.7477264404296876e-05,
      "model_forward_time": 3.1359775066375732,
      "step": 7203
    },
    {
      "epoch": 2.7477264404296876e-05,
      "step": 7203,
      "training_step_time": 9.913546800613403
    },
    {
      "epoch": 2.7477264404296876e-05,
      "model_forward_time": 3.135650873184204,
      "step": 7203
    },
    {
      "epoch": 2.7477264404296876e-05,
      "step": 7203,
      "training_step_time": 10.294152021408081
    },
    {
      "epoch": 2.74810791015625e-05,
      "model_forward_time": 3.183072805404663,
      "step": 7204
    },
    {
      "epoch": 2.74810791015625e-05,
      "step": 7204,
      "training_step_time": 9.96458101272583
    },
    {
      "epoch": 2.74810791015625e-05,
      "model_forward_time": 3.141106367111206,
      "step": 7204
    },
    {
      "epoch": 2.74810791015625e-05,
      "step": 7204,
      "training_step_time": 9.924036741256714
    },
    {
      "epoch": 2.74810791015625e-05,
      "model_forward_time": 3.1393823623657227,
      "step": 7204
    },
    {
      "epoch": 2.74810791015625e-05,
      "step": 7204,
      "training_step_time": 9.92634630203247
    },
    {
      "epoch": 2.74810791015625e-05,
      "model_forward_time": 3.140230417251587,
      "step": 7204
    },
    {
      "epoch": 2.74810791015625e-05,
      "step": 7204,
      "training_step_time": 10.29860806465149
    },
    {
      "epoch": 2.7484893798828125e-05,
      "model_forward_time": 3.427042245864868,
      "step": 7205
    },
    {
      "dynamics_loss_avg": 0.05043429136276245,
      "epoch": 2.7484893798828125e-05,
      "step": 7205
    },
    {
      "action_loss_avg": 0.014641584083437919,
      "epoch": 2.7484893798828125e-05,
      "step": 7205
    },
    {
      "epoch": 2.7484893798828125e-05,
      "step": 7205,
      "training_step_time": 10.211130619049072
    },
    {
      "epoch": 2.7484893798828125e-05,
      "model_forward_time": 3.134838342666626,
      "step": 7205
    },
    {
      "epoch": 2.7484893798828125e-05,
      "step": 7205,
      "training_step_time": 9.914448738098145
    },
    {
      "epoch": 2.7484893798828125e-05,
      "model_forward_time": 3.1414029598236084,
      "step": 7205
    },
    {
      "epoch": 2.7484893798828125e-05,
      "step": 7205,
      "training_step_time": 9.920077323913574
    },
    {
      "epoch": 2.7484893798828125e-05,
      "model_forward_time": 3.1368532180786133,
      "step": 7205
    },
    {
      "epoch": 2.7484893798828125e-05,
      "step": 7205,
      "training_step_time": 10.027894735336304
    },
    {
      "epoch": 2.748870849609375e-05,
      "model_forward_time": 3.1591105461120605,
      "step": 7206
    },
    {
      "epoch": 2.748870849609375e-05,
      "step": 7206,
      "training_step_time": 9.935946702957153
    },
    {
      "epoch": 2.748870849609375e-05,
      "model_forward_time": 3.138314962387085,
      "step": 7206
    },
    {
      "epoch": 2.748870849609375e-05,
      "step": 7206,
      "training_step_time": 10.184826374053955
    },
    {
      "epoch": 2.748870849609375e-05,
      "model_forward_time": 3.1380362510681152,
      "step": 7206
    },
    {
      "epoch": 2.748870849609375e-05,
      "step": 7206,
      "training_step_time": 9.913938045501709
    },
    {
      "epoch": 2.748870849609375e-05,
      "model_forward_time": 3.1396446228027344,
      "step": 7206
    },
    {
      "epoch": 2.748870849609375e-05,
      "step": 7206,
      "training_step_time": 10.262763261795044
    },
    {
      "epoch": 2.7492523193359374e-05,
      "model_forward_time": 3.1458144187927246,
      "step": 7207
    },
    {
      "epoch": 2.7492523193359374e-05,
      "step": 7207,
      "training_step_time": 9.926808595657349
    },
    {
      "epoch": 2.7492523193359374e-05,
      "model_forward_time": 3.1398744583129883,
      "step": 7207
    },
    {
      "epoch": 2.7492523193359374e-05,
      "step": 7207,
      "training_step_time": 9.916377305984497
    },
    {
      "epoch": 2.7492523193359374e-05,
      "model_forward_time": 3.414875030517578,
      "step": 7207
    },
    {
      "dynamics_loss_avg": 0.06610580142587423,
      "epoch": 2.7492523193359374e-05,
      "step": 7207
    },
    {
      "action_loss_avg": 0.023561984254047274,
      "epoch": 2.7492523193359374e-05,
      "step": 7207
    },
    {
      "epoch": 2.7492523193359374e-05,
      "step": 7207,
      "training_step_time": 10.193485975265503
    },
    {
      "epoch": 2.7492523193359374e-05,
      "model_forward_time": 3.13279390335083,
      "step": 7207
    },
    {
      "epoch": 2.7492523193359374e-05,
      "step": 7207,
      "training_step_time": 10.02202582359314
    },
    {
      "epoch": 2.7496337890625e-05,
      "model_forward_time": 3.1520066261291504,
      "step": 7208
    },
    {
      "epoch": 2.7496337890625e-05,
      "step": 7208,
      "training_step_time": 9.96634554862976
    },
    {
      "epoch": 2.7496337890625e-05,
      "model_forward_time": 3.1377346515655518,
      "step": 7208
    },
    {
      "epoch": 2.7496337890625e-05,
      "step": 7208,
      "training_step_time": 9.911291360855103
    },
    {
      "epoch": 2.7496337890625e-05,
      "model_forward_time": 3.1358680725097656,
      "step": 7208
    },
    {
      "epoch": 2.7496337890625e-05,
      "step": 7208,
      "training_step_time": 9.914163827896118
    },
    {
      "epoch": 2.7496337890625e-05,
      "model_forward_time": 3.424006462097168,
      "step": 7208
    },
    {
      "epoch": 2.7496337890625e-05,
      "step": 7208,
      "training_step_time": 10.291173934936523
    },
    {
      "epoch": 2.7500152587890624e-05,
      "model_forward_time": 3.172118902206421,
      "step": 7209
    },
    {
      "epoch": 2.7500152587890624e-05,
      "step": 7209,
      "training_step_time": 9.948723793029785
    },
    {
      "epoch": 2.7500152587890624e-05,
      "model_forward_time": 3.1453897953033447,
      "step": 7209
    },
    {
      "epoch": 2.7500152587890624e-05,
      "step": 7209,
      "training_step_time": 9.918662309646606
    },
    {
      "epoch": 2.7500152587890624e-05,
      "model_forward_time": 3.141176223754883,
      "step": 7209
    },
    {
      "epoch": 2.7500152587890624e-05,
      "step": 7209,
      "training_step_time": 9.918434143066406
    },
    {
      "epoch": 2.7500152587890624e-05,
      "model_forward_time": 3.1432840824127197,
      "step": 7209
    },
    {
      "epoch": 2.7500152587890624e-05,
      "step": 7209,
      "training_step_time": 10.287264823913574
    },
    {
      "epoch": 2.750396728515625e-05,
      "grad_norm": 0.4550672173500061,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 7210
    },
    {
      "epoch": 2.750396728515625e-05,
      "model_forward_time": 3.4294638633728027,
      "step": 7210
    },
    {
      "dynamics_loss_avg": 0.04958621673285961,
      "epoch": 2.750396728515625e-05,
      "step": 7210
    },
    {
      "action_loss_avg": 0.0181114942766726,
      "epoch": 2.750396728515625e-05,
      "step": 7210
    },
    {
      "epoch": 2.750396728515625e-05,
      "step": 7210,
      "training_step_time": 10.203329086303711
    },
    {
      "epoch": 2.750396728515625e-05,
      "model_forward_time": 3.1436150074005127,
      "step": 7210
    },
    {
      "epoch": 2.750396728515625e-05,
      "step": 7210,
      "training_step_time": 9.915194034576416
    },
    {
      "epoch": 2.750396728515625e-05,
      "model_forward_time": 3.144474983215332,
      "step": 7210
    },
    {
      "epoch": 2.750396728515625e-05,
      "step": 7210,
      "training_step_time": 9.91812777519226
    },
    {
      "epoch": 2.750396728515625e-05,
      "model_forward_time": 3.142993450164795,
      "step": 7210
    },
    {
      "epoch": 2.750396728515625e-05,
      "step": 7210,
      "training_step_time": 10.27155590057373
    },
    {
      "epoch": 2.7507781982421876e-05,
      "model_forward_time": 3.190324544906616,
      "step": 7211
    },
    {
      "epoch": 2.7507781982421876e-05,
      "step": 7211,
      "training_step_time": 9.970078706741333
    },
    {
      "epoch": 2.7507781982421876e-05,
      "model_forward_time": 3.4436259269714355,
      "step": 7211
    },
    {
      "epoch": 2.7507781982421876e-05,
      "step": 7211,
      "training_step_time": 10.21439266204834
    },
    {
      "epoch": 2.7507781982421876e-05,
      "model_forward_time": 3.1453206539154053,
      "step": 7211
    },
    {
      "epoch": 2.7507781982421876e-05,
      "step": 7211,
      "training_step_time": 9.92514157295227
    },
    {
      "epoch": 2.7507781982421876e-05,
      "model_forward_time": 3.1428472995758057,
      "step": 7211
    },
    {
      "epoch": 2.7507781982421876e-05,
      "step": 7211,
      "training_step_time": 10.254014730453491
    },
    {
      "epoch": 2.75115966796875e-05,
      "model_forward_time": 3.18949294090271,
      "step": 7212
    },
    {
      "epoch": 2.75115966796875e-05,
      "step": 7212,
      "training_step_time": 9.969085454940796
    },
    {
      "epoch": 2.75115966796875e-05,
      "model_forward_time": 3.1480348110198975,
      "step": 7212
    },
    {
      "epoch": 2.75115966796875e-05,
      "step": 7212,
      "training_step_time": 9.927465200424194
    },
    {
      "epoch": 2.75115966796875e-05,
      "model_forward_time": 3.147373676300049,
      "step": 7212
    },
    {
      "dynamics_loss_avg": 0.04860524525865913,
      "epoch": 2.75115966796875e-05,
      "step": 7212
    },
    {
      "action_loss_avg": 0.0172274004900828,
      "epoch": 2.75115966796875e-05,
      "step": 7212
    },
    {
      "epoch": 2.75115966796875e-05,
      "step": 7212,
      "training_step_time": 10.211382389068604
    },
    {
      "epoch": 2.75115966796875e-05,
      "model_forward_time": 3.148916482925415,
      "step": 7212
    },
    {
      "epoch": 2.75115966796875e-05,
      "step": 7212,
      "training_step_time": 10.027363777160645
    },
    {
      "epoch": 2.7515411376953125e-05,
      "model_forward_time": 3.1776061058044434,
      "step": 7213
    },
    {
      "epoch": 2.7515411376953125e-05,
      "step": 7213,
      "training_step_time": 9.961808681488037
    },
    {
      "epoch": 2.7515411376953125e-05,
      "model_forward_time": 3.1449694633483887,
      "step": 7213
    },
    {
      "epoch": 2.7515411376953125e-05,
      "step": 7213,
      "training_step_time": 9.921625137329102
    },
    {
      "epoch": 2.7515411376953125e-05,
      "model_forward_time": 3.1474485397338867,
      "step": 7213
    },
    {
      "epoch": 2.7515411376953125e-05,
      "step": 7213,
      "training_step_time": 9.925974369049072
    },
    {
      "epoch": 2.7515411376953125e-05,
      "model_forward_time": 3.43516206741333,
      "step": 7213
    },
    {
      "epoch": 2.7515411376953125e-05,
      "step": 7213,
      "training_step_time": 10.308042049407959
    },
    {
      "epoch": 2.751922607421875e-05,
      "model_forward_time": 3.176785945892334,
      "step": 7214
    },
    {
      "epoch": 2.751922607421875e-05,
      "step": 7214,
      "training_step_time": 9.952102422714233
    },
    {
      "epoch": 2.751922607421875e-05,
      "model_forward_time": 3.1428911685943604,
      "step": 7214
    },
    {
      "epoch": 2.751922607421875e-05,
      "step": 7214,
      "training_step_time": 9.919040203094482
    },
    {
      "epoch": 2.751922607421875e-05,
      "model_forward_time": 3.143282413482666,
      "step": 7214
    },
    {
      "epoch": 2.751922607421875e-05,
      "step": 7214,
      "training_step_time": 9.920567035675049
    },
    {
      "epoch": 2.751922607421875e-05,
      "model_forward_time": 3.141700506210327,
      "step": 7214
    },
    {
      "epoch": 2.751922607421875e-05,
      "step": 7214,
      "training_step_time": 10.294432163238525
    },
    {
      "epoch": 2.7523040771484374e-05,
      "model_forward_time": 3.1718785762786865,
      "step": 7215
    },
    {
      "dynamics_loss_avg": 0.05456398241221905,
      "epoch": 2.7523040771484374e-05,
      "step": 7215
    },
    {
      "action_loss_avg": 0.022060371376574038,
      "epoch": 2.7523040771484374e-05,
      "step": 7215
    },
    {
      "epoch": 2.7523040771484374e-05,
      "step": 7215,
      "training_step_time": 10.237743377685547
    },
    {
      "epoch": 2.7523040771484374e-05,
      "model_forward_time": 3.1442747116088867,
      "step": 7215
    },
    {
      "epoch": 2.7523040771484374e-05,
      "step": 7215,
      "training_step_time": 9.921729803085327
    },
    {
      "epoch": 2.7523040771484374e-05,
      "model_forward_time": 3.1478183269500732,
      "step": 7215
    },
    {
      "epoch": 2.7523040771484374e-05,
      "step": 7215,
      "training_step_time": 9.924430131912231
    },
    {
      "epoch": 2.7523040771484374e-05,
      "model_forward_time": 3.14914608001709,
      "step": 7215
    },
    {
      "epoch": 2.7523040771484374e-05,
      "step": 7215,
      "training_step_time": 10.297630786895752
    },
    {
      "epoch": 2.752685546875e-05,
      "model_forward_time": 3.1790287494659424,
      "step": 7216
    },
    {
      "epoch": 2.752685546875e-05,
      "step": 7216,
      "training_step_time": 9.961550235748291
    },
    {
      "epoch": 2.752685546875e-05,
      "model_forward_time": 3.4004452228546143,
      "step": 7216
    },
    {
      "epoch": 2.752685546875e-05,
      "step": 7216,
      "training_step_time": 10.178626298904419
    },
    {
      "epoch": 2.752685546875e-05,
      "model_forward_time": 3.14906644821167,
      "step": 7216
    },
    {
      "epoch": 2.752685546875e-05,
      "step": 7216,
      "training_step_time": 9.929028034210205
    },
    {
      "epoch": 2.752685546875e-05,
      "model_forward_time": 3.147812604904175,
      "step": 7216
    },
    {
      "epoch": 2.752685546875e-05,
      "step": 7216,
      "training_step_time": 10.29174017906189
    },
    {
      "epoch": 2.7530670166015624e-05,
      "model_forward_time": 3.1874568462371826,
      "step": 7217
    },
    {
      "epoch": 2.7530670166015624e-05,
      "step": 7217,
      "training_step_time": 9.968546390533447
    },
    {
      "epoch": 2.7530670166015624e-05,
      "model_forward_time": 2.4411776065826416,
      "step": 7217
    },
    {
      "epoch": 2.7530670166015624e-05,
      "step": 7217,
      "training_step_time": 7.47467303276062
    },
    {
      "epoch": 2.7530670166015624e-05,
      "model_forward_time": 3.1383166313171387,
      "step": 7217
    },
    {
      "dynamics_loss_avg": 0.06723707765340806,
      "epoch": 2.7530670166015624e-05,
      "step": 7217
    },
    {
      "action_loss_avg": 0.02757021798752248,
      "epoch": 2.7530670166015624e-05,
      "step": 7217
    },
    {
      "epoch": 2.7530670166015624e-05,
      "step": 7217,
      "training_step_time": 10.189327001571655
    },
    {
      "epoch": 2.7530670166015624e-05,
      "model_forward_time": 3.1403791904449463,
      "step": 7217
    },
    {
      "epoch": 2.7530670166015624e-05,
      "step": 7217,
      "training_step_time": 10.021826028823853
    },
    {
      "epoch": 2.753448486328125e-05,
      "model_forward_time": 3.1797823905944824,
      "step": 7218
    },
    {
      "epoch": 2.753448486328125e-05,
      "step": 7218,
      "training_step_time": 9.95840859413147
    },
    {
      "epoch": 2.753448486328125e-05,
      "model_forward_time": 3.145200729370117,
      "step": 7218
    },
    {
      "epoch": 2.753448486328125e-05,
      "step": 7218,
      "training_step_time": 9.923678874969482
    },
    {
      "epoch": 2.753448486328125e-05,
      "model_forward_time": 3.143934488296509,
      "step": 7218
    },
    {
      "epoch": 2.753448486328125e-05,
      "step": 7218,
      "training_step_time": 9.921828985214233
    },
    {
      "epoch": 2.753448486328125e-05,
      "model_forward_time": 3.403106689453125,
      "step": 7218
    },
    {
      "epoch": 2.753448486328125e-05,
      "step": 7218,
      "training_step_time": 10.279601812362671
    },
    {
      "epoch": 2.7538299560546876e-05,
      "model_forward_time": 3.187750816345215,
      "step": 7219
    },
    {
      "epoch": 2.7538299560546876e-05,
      "step": 7219,
      "training_step_time": 9.968698501586914
    },
    {
      "epoch": 2.7538299560546876e-05,
      "model_forward_time": 3.147308111190796,
      "step": 7219
    },
    {
      "epoch": 2.7538299560546876e-05,
      "step": 7219,
      "training_step_time": 9.928295612335205
    },
    {
      "epoch": 2.7538299560546876e-05,
      "model_forward_time": 3.144585132598877,
      "step": 7219
    },
    {
      "epoch": 2.7538299560546876e-05,
      "step": 7219,
      "training_step_time": 9.928275108337402
    },
    {
      "epoch": 2.7538299560546876e-05,
      "model_forward_time": 3.1496973037719727,
      "step": 7219
    },
    {
      "epoch": 2.7538299560546876e-05,
      "step": 7219,
      "training_step_time": 10.289137601852417
    },
    {
      "epoch": 2.75421142578125e-05,
      "grad_norm": 0.8084501624107361,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7220
    },
    {
      "epoch": 2.75421142578125e-05,
      "model_forward_time": 3.443354368209839,
      "step": 7220
    },
    {
      "dynamics_loss_avg": 0.060033435560762884,
      "epoch": 2.75421142578125e-05,
      "step": 7220
    },
    {
      "action_loss_avg": 0.01973021272569895,
      "epoch": 2.75421142578125e-05,
      "step": 7220
    },
    {
      "epoch": 2.75421142578125e-05,
      "step": 7220,
      "training_step_time": 10.222018480300903
    },
    {
      "epoch": 2.75421142578125e-05,
      "model_forward_time": 3.1416447162628174,
      "step": 7220
    },
    {
      "epoch": 2.75421142578125e-05,
      "step": 7220,
      "training_step_time": 9.915870428085327
    },
    {
      "epoch": 2.75421142578125e-05,
      "model_forward_time": 3.1400225162506104,
      "step": 7220
    },
    {
      "epoch": 2.75421142578125e-05,
      "step": 7220,
      "training_step_time": 9.912975072860718
    },
    {
      "epoch": 2.75421142578125e-05,
      "model_forward_time": 3.146543502807617,
      "step": 7220
    },
    {
      "epoch": 2.75421142578125e-05,
      "step": 7220,
      "training_step_time": 10.31046724319458
    },
    {
      "epoch": 2.7545928955078125e-05,
      "model_forward_time": 3.155583620071411,
      "step": 7221
    },
    {
      "epoch": 2.7545928955078125e-05,
      "step": 7221,
      "training_step_time": 9.933438539505005
    },
    {
      "epoch": 2.7545928955078125e-05,
      "model_forward_time": 3.1391348838806152,
      "step": 7221
    },
    {
      "epoch": 2.7545928955078125e-05,
      "step": 7221,
      "training_step_time": 10.200706481933594
    },
    {
      "epoch": 2.7545928955078125e-05,
      "model_forward_time": 3.142145872116089,
      "step": 7221
    },
    {
      "epoch": 2.7545928955078125e-05,
      "step": 7221,
      "training_step_time": 9.918090343475342
    },
    {
      "epoch": 2.7545928955078125e-05,
      "model_forward_time": 3.144732713699341,
      "step": 7221
    },
    {
      "epoch": 2.7545928955078125e-05,
      "step": 7221,
      "training_step_time": 10.29211688041687
    },
    {
      "epoch": 2.754974365234375e-05,
      "model_forward_time": 3.1778557300567627,
      "step": 7222
    },
    {
      "epoch": 2.754974365234375e-05,
      "step": 7222,
      "training_step_time": 9.957417249679565
    },
    {
      "epoch": 2.754974365234375e-05,
      "model_forward_time": 3.144380569458008,
      "step": 7222
    },
    {
      "epoch": 2.754974365234375e-05,
      "step": 7222,
      "training_step_time": 9.920852899551392
    },
    {
      "epoch": 2.754974365234375e-05,
      "model_forward_time": 3.410318374633789,
      "step": 7222
    },
    {
      "dynamics_loss_avg": 0.059437576308846475,
      "epoch": 2.754974365234375e-05,
      "step": 7222
    },
    {
      "action_loss_avg": 0.02558361287228763,
      "epoch": 2.754974365234375e-05,
      "step": 7222
    },
    {
      "epoch": 2.754974365234375e-05,
      "step": 7222,
      "training_step_time": 10.191771745681763
    },
    {
      "epoch": 2.754974365234375e-05,
      "model_forward_time": 3.1468753814697266,
      "step": 7222
    },
    {
      "epoch": 2.754974365234375e-05,
      "step": 7222,
      "training_step_time": 10.0416100025177
    },
    {
      "epoch": 2.7553558349609374e-05,
      "model_forward_time": 3.156435012817383,
      "step": 7223
    },
    {
      "epoch": 2.7553558349609374e-05,
      "step": 7223,
      "training_step_time": 9.937260627746582
    },
    {
      "epoch": 2.7553558349609374e-05,
      "model_forward_time": 3.1395347118377686,
      "step": 7223
    },
    {
      "epoch": 2.7553558349609374e-05,
      "step": 7223,
      "training_step_time": 9.9145348072052
    },
    {
      "epoch": 2.7553558349609374e-05,
      "model_forward_time": 3.150428533554077,
      "step": 7223
    },
    {
      "epoch": 2.7553558349609374e-05,
      "step": 7223,
      "training_step_time": 9.92440128326416
    },
    {
      "epoch": 2.7553558349609374e-05,
      "model_forward_time": 3.1382241249084473,
      "step": 7223
    },
    {
      "epoch": 2.7553558349609374e-05,
      "step": 7223,
      "training_step_time": 10.283758878707886
    },
    {
      "epoch": 2.7557373046875e-05,
      "model_forward_time": 3.1586406230926514,
      "step": 7224
    },
    {
      "epoch": 2.7557373046875e-05,
      "step": 7224,
      "training_step_time": 9.9444580078125
    },
    {
      "epoch": 2.7557373046875e-05,
      "model_forward_time": 3.1435959339141846,
      "step": 7224
    },
    {
      "epoch": 2.7557373046875e-05,
      "step": 7224,
      "training_step_time": 9.920102834701538
    },
    {
      "epoch": 2.7557373046875e-05,
      "model_forward_time": 3.1483047008514404,
      "step": 7224
    },
    {
      "epoch": 2.7557373046875e-05,
      "step": 7224,
      "training_step_time": 9.926322221755981
    },
    {
      "epoch": 2.7557373046875e-05,
      "model_forward_time": 3.1431477069854736,
      "step": 7224
    },
    {
      "epoch": 2.7557373046875e-05,
      "step": 7224,
      "training_step_time": 10.276101112365723
    },
    {
      "epoch": 2.7561187744140624e-05,
      "model_forward_time": 2.4805850982666016,
      "step": 7225
    },
    {
      "dynamics_loss_avg": 0.07061175554990769,
      "epoch": 2.7561187744140624e-05,
      "step": 7225
    },
    {
      "action_loss_avg": 0.030272903479635717,
      "epoch": 2.7561187744140624e-05,
      "step": 7225
    },
    {
      "epoch": 2.7561187744140624e-05,
      "step": 7225,
      "training_step_time": 7.783359527587891
    },
    {
      "epoch": 2.7561187744140624e-05,
      "model_forward_time": 3.1392195224761963,
      "step": 7225
    },
    {
      "epoch": 2.7561187744140624e-05,
      "step": 7225,
      "training_step_time": 9.915644645690918
    },
    {
      "epoch": 2.7561187744140624e-05,
      "model_forward_time": 3.1500625610351562,
      "step": 7225
    },
    {
      "epoch": 2.7561187744140624e-05,
      "step": 7225,
      "training_step_time": 9.92690658569336
    },
    {
      "epoch": 2.7561187744140624e-05,
      "model_forward_time": 3.1458499431610107,
      "step": 7225
    },
    {
      "epoch": 2.7561187744140624e-05,
      "step": 7225,
      "training_step_time": 10.280592203140259
    },
    {
      "epoch": 2.756500244140625e-05,
      "model_forward_time": 3.190918207168579,
      "step": 7226
    },
    {
      "epoch": 2.756500244140625e-05,
      "step": 7226,
      "training_step_time": 9.972132205963135
    },
    {
      "epoch": 2.756500244140625e-05,
      "model_forward_time": 3.1474409103393555,
      "step": 7226
    },
    {
      "epoch": 2.756500244140625e-05,
      "step": 7226,
      "training_step_time": 10.233846664428711
    },
    {
      "epoch": 2.756500244140625e-05,
      "model_forward_time": 3.1436071395874023,
      "step": 7226
    },
    {
      "epoch": 2.756500244140625e-05,
      "step": 7226,
      "training_step_time": 9.916301250457764
    },
    {
      "epoch": 2.756500244140625e-05,
      "model_forward_time": 3.140895366668701,
      "step": 7226
    },
    {
      "epoch": 2.756500244140625e-05,
      "step": 7226,
      "training_step_time": 10.311909675598145
    },
    {
      "epoch": 2.7568817138671876e-05,
      "model_forward_time": 3.1808645725250244,
      "step": 7227
    },
    {
      "epoch": 2.7568817138671876e-05,
      "step": 7227,
      "training_step_time": 9.953379392623901
    },
    {
      "epoch": 2.7568817138671876e-05,
      "model_forward_time": 3.1370134353637695,
      "step": 7227
    },
    {
      "epoch": 2.7568817138671876e-05,
      "step": 7227,
      "training_step_time": 9.911322116851807
    },
    {
      "epoch": 2.7568817138671876e-05,
      "model_forward_time": 3.417492151260376,
      "step": 7227
    },
    {
      "dynamics_loss_avg": 0.04508390333503485,
      "epoch": 2.7568817138671876e-05,
      "step": 7227
    },
    {
      "action_loss_avg": 0.020656986394897103,
      "epoch": 2.7568817138671876e-05,
      "step": 7227
    },
    {
      "epoch": 2.7568817138671876e-05,
      "step": 7227,
      "training_step_time": 10.19345998764038
    },
    {
      "epoch": 2.7568817138671876e-05,
      "model_forward_time": 3.138428211212158,
      "step": 7227
    },
    {
      "epoch": 2.7568817138671876e-05,
      "step": 7227,
      "training_step_time": 10.030799627304077
    },
    {
      "epoch": 2.75726318359375e-05,
      "model_forward_time": 3.176077127456665,
      "step": 7228
    },
    {
      "epoch": 2.75726318359375e-05,
      "step": 7228,
      "training_step_time": 9.959403991699219
    },
    {
      "epoch": 2.75726318359375e-05,
      "model_forward_time": 3.144355535507202,
      "step": 7228
    },
    {
      "epoch": 2.75726318359375e-05,
      "step": 7228,
      "training_step_time": 9.923820734024048
    },
    {
      "epoch": 2.75726318359375e-05,
      "model_forward_time": 3.147592782974243,
      "step": 7228
    },
    {
      "epoch": 2.75726318359375e-05,
      "step": 7228,
      "training_step_time": 9.92551851272583
    },
    {
      "epoch": 2.75726318359375e-05,
      "model_forward_time": 3.4268767833709717,
      "step": 7228
    },
    {
      "epoch": 2.75726318359375e-05,
      "step": 7228,
      "training_step_time": 10.296587228775024
    },
    {
      "epoch": 2.7576446533203125e-05,
      "model_forward_time": 3.1664810180664062,
      "step": 7229
    },
    {
      "epoch": 2.7576446533203125e-05,
      "step": 7229,
      "training_step_time": 9.950260162353516
    },
    {
      "epoch": 2.7576446533203125e-05,
      "model_forward_time": 3.1493937969207764,
      "step": 7229
    },
    {
      "epoch": 2.7576446533203125e-05,
      "step": 7229,
      "training_step_time": 9.926735162734985
    },
    {
      "epoch": 2.7576446533203125e-05,
      "model_forward_time": 3.142221450805664,
      "step": 7229
    },
    {
      "epoch": 2.7576446533203125e-05,
      "step": 7229,
      "training_step_time": 9.918963193893433
    },
    {
      "epoch": 2.7576446533203125e-05,
      "model_forward_time": 3.1423187255859375,
      "step": 7229
    },
    {
      "epoch": 2.7576446533203125e-05,
      "step": 7229,
      "training_step_time": 10.02705717086792
    },
    {
      "epoch": 2.758026123046875e-05,
      "grad_norm": 0.3937731087207794,
      "learning_rate": 1e-05,
      "loss": 0.0833,
      "step": 7230
    },
    {
      "epoch": 2.758026123046875e-05,
      "model_forward_time": 3.460406541824341,
      "step": 7230
    },
    {
      "dynamics_loss_avg": 0.054472157172858714,
      "epoch": 2.758026123046875e-05,
      "step": 7230
    },
    {
      "action_loss_avg": 0.025236184522509576,
      "epoch": 2.758026123046875e-05,
      "step": 7230
    },
    {
      "epoch": 2.758026123046875e-05,
      "step": 7230,
      "training_step_time": 10.238929748535156
    },
    {
      "epoch": 2.758026123046875e-05,
      "model_forward_time": 3.1326282024383545,
      "step": 7230
    },
    {
      "epoch": 2.758026123046875e-05,
      "step": 7230,
      "training_step_time": 9.905827760696411
    },
    {
      "epoch": 2.758026123046875e-05,
      "model_forward_time": 3.15250301361084,
      "step": 7230
    },
    {
      "epoch": 2.758026123046875e-05,
      "step": 7230,
      "training_step_time": 9.929592847824097
    },
    {
      "epoch": 2.758026123046875e-05,
      "model_forward_time": 3.1351158618927,
      "step": 7230
    },
    {
      "epoch": 2.758026123046875e-05,
      "step": 7230,
      "training_step_time": 10.337447166442871
    },
    {
      "epoch": 2.7584075927734375e-05,
      "model_forward_time": 3.168734312057495,
      "step": 7231
    },
    {
      "epoch": 2.7584075927734375e-05,
      "step": 7231,
      "training_step_time": 9.946812152862549
    },
    {
      "epoch": 2.7584075927734375e-05,
      "model_forward_time": 3.389387369155884,
      "step": 7231
    },
    {
      "epoch": 2.7584075927734375e-05,
      "step": 7231,
      "training_step_time": 10.166355609893799
    },
    {
      "epoch": 2.7584075927734375e-05,
      "model_forward_time": 3.1392881870269775,
      "step": 7231
    },
    {
      "epoch": 2.7584075927734375e-05,
      "step": 7231,
      "training_step_time": 9.913560628890991
    },
    {
      "epoch": 2.7584075927734375e-05,
      "model_forward_time": 3.140247344970703,
      "step": 7231
    },
    {
      "epoch": 2.7584075927734375e-05,
      "step": 7231,
      "training_step_time": 10.307695865631104
    },
    {
      "epoch": 2.7587890625e-05,
      "model_forward_time": 3.1815309524536133,
      "step": 7232
    },
    {
      "epoch": 2.7587890625e-05,
      "step": 7232,
      "training_step_time": 9.960594892501831
    },
    {
      "epoch": 2.7587890625e-05,
      "model_forward_time": 3.1395411491394043,
      "step": 7232
    },
    {
      "epoch": 2.7587890625e-05,
      "step": 7232,
      "training_step_time": 9.915463924407959
    },
    {
      "epoch": 2.7587890625e-05,
      "model_forward_time": 3.4139912128448486,
      "step": 7232
    },
    {
      "dynamics_loss_avg": 0.06948008649051189,
      "epoch": 2.7587890625e-05,
      "step": 7232
    },
    {
      "action_loss_avg": 0.026541471807286145,
      "epoch": 2.7587890625e-05,
      "step": 7232
    },
    {
      "epoch": 2.7587890625e-05,
      "step": 7232,
      "training_step_time": 10.190762519836426
    },
    {
      "epoch": 2.7587890625e-05,
      "model_forward_time": 3.137486457824707,
      "step": 7232
    },
    {
      "epoch": 2.7587890625e-05,
      "step": 7232,
      "training_step_time": 10.029687404632568
    },
    {
      "epoch": 2.7591705322265624e-05,
      "model_forward_time": 3.170093059539795,
      "step": 7233
    },
    {
      "epoch": 2.7591705322265624e-05,
      "step": 7233,
      "training_step_time": 9.94620418548584
    },
    {
      "epoch": 2.7591705322265624e-05,
      "model_forward_time": 3.1406455039978027,
      "step": 7233
    },
    {
      "epoch": 2.7591705322265624e-05,
      "step": 7233,
      "training_step_time": 9.921420812606812
    },
    {
      "epoch": 2.7591705322265624e-05,
      "model_forward_time": 3.1347830295562744,
      "step": 7233
    },
    {
      "epoch": 2.7591705322265624e-05,
      "step": 7233,
      "training_step_time": 9.91664457321167
    },
    {
      "epoch": 2.7591705322265624e-05,
      "model_forward_time": 3.3931167125701904,
      "step": 7233
    },
    {
      "epoch": 2.7591705322265624e-05,
      "step": 7233,
      "training_step_time": 10.266386270523071
    },
    {
      "epoch": 2.759552001953125e-05,
      "model_forward_time": 3.1797192096710205,
      "step": 7234
    },
    {
      "epoch": 2.759552001953125e-05,
      "step": 7234,
      "training_step_time": 9.963385820388794
    },
    {
      "epoch": 2.759552001953125e-05,
      "model_forward_time": 3.146360158920288,
      "step": 7234
    },
    {
      "epoch": 2.759552001953125e-05,
      "step": 7234,
      "training_step_time": 9.924148082733154
    },
    {
      "epoch": 2.759552001953125e-05,
      "model_forward_time": 3.1413028240203857,
      "step": 7234
    },
    {
      "epoch": 2.759552001953125e-05,
      "step": 7234,
      "training_step_time": 9.918479442596436
    },
    {
      "epoch": 2.759552001953125e-05,
      "model_forward_time": 3.139620304107666,
      "step": 7234
    },
    {
      "epoch": 2.759552001953125e-05,
      "step": 7234,
      "training_step_time": 10.015692949295044
    },
    {
      "epoch": 2.7599334716796876e-05,
      "model_forward_time": 3.4041194915771484,
      "step": 7235
    },
    {
      "dynamics_loss_avg": 0.06170813636854291,
      "epoch": 2.7599334716796876e-05,
      "step": 7235
    },
    {
      "action_loss_avg": 0.022666912199929355,
      "epoch": 2.7599334716796876e-05,
      "step": 7235
    },
    {
      "epoch": 2.7599334716796876e-05,
      "step": 7235,
      "training_step_time": 10.179924964904785
    },
    {
      "epoch": 2.7599334716796876e-05,
      "model_forward_time": 3.1347217559814453,
      "step": 7235
    },
    {
      "epoch": 2.7599334716796876e-05,
      "step": 7235,
      "training_step_time": 9.908385753631592
    },
    {
      "epoch": 2.7599334716796876e-05,
      "model_forward_time": 3.1359946727752686,
      "step": 7235
    },
    {
      "epoch": 2.7599334716796876e-05,
      "step": 7235,
      "training_step_time": 9.910743713378906
    },
    {
      "epoch": 2.7599334716796876e-05,
      "model_forward_time": 3.1403732299804688,
      "step": 7235
    },
    {
      "epoch": 2.7599334716796876e-05,
      "step": 7235,
      "training_step_time": 10.293565034866333
    },
    {
      "epoch": 2.76031494140625e-05,
      "model_forward_time": 3.147686243057251,
      "step": 7236
    },
    {
      "epoch": 2.76031494140625e-05,
      "step": 7236,
      "training_step_time": 9.93835186958313
    },
    {
      "epoch": 2.76031494140625e-05,
      "model_forward_time": 3.4232935905456543,
      "step": 7236
    },
    {
      "epoch": 2.76031494140625e-05,
      "step": 7236,
      "training_step_time": 10.194387435913086
    },
    {
      "epoch": 2.76031494140625e-05,
      "model_forward_time": 3.1330742835998535,
      "step": 7236
    },
    {
      "epoch": 2.76031494140625e-05,
      "step": 7236,
      "training_step_time": 9.903394222259521
    },
    {
      "epoch": 2.76031494140625e-05,
      "model_forward_time": 3.1339364051818848,
      "step": 7236
    },
    {
      "epoch": 2.76031494140625e-05,
      "step": 7236,
      "training_step_time": 10.273701667785645
    },
    {
      "epoch": 2.7606964111328125e-05,
      "model_forward_time": 3.1704578399658203,
      "step": 7237
    },
    {
      "epoch": 2.7606964111328125e-05,
      "step": 7237,
      "training_step_time": 9.944247961044312
    },
    {
      "epoch": 2.7606964111328125e-05,
      "model_forward_time": 3.1289260387420654,
      "step": 7237
    },
    {
      "epoch": 2.7606964111328125e-05,
      "step": 7237,
      "training_step_time": 9.899199485778809
    },
    {
      "epoch": 2.7606964111328125e-05,
      "model_forward_time": 3.3978092670440674,
      "step": 7237
    },
    {
      "dynamics_loss_avg": 0.054879112914204595,
      "epoch": 2.7606964111328125e-05,
      "step": 7237
    },
    {
      "action_loss_avg": 0.02132913828827441,
      "epoch": 2.7606964111328125e-05,
      "step": 7237
    },
    {
      "epoch": 2.7606964111328125e-05,
      "step": 7237,
      "training_step_time": 10.172089099884033
    },
    {
      "epoch": 2.7606964111328125e-05,
      "model_forward_time": 3.1355152130126953,
      "step": 7237
    },
    {
      "epoch": 2.7606964111328125e-05,
      "step": 7237,
      "training_step_time": 10.334226369857788
    },
    {
      "epoch": 2.761077880859375e-05,
      "model_forward_time": 3.1494686603546143,
      "step": 7238
    },
    {
      "epoch": 2.761077880859375e-05,
      "step": 7238,
      "training_step_time": 9.92391848564148
    },
    {
      "epoch": 2.761077880859375e-05,
      "model_forward_time": 3.1339364051818848,
      "step": 7238
    },
    {
      "epoch": 2.761077880859375e-05,
      "step": 7238,
      "training_step_time": 9.904347658157349
    },
    {
      "epoch": 2.761077880859375e-05,
      "model_forward_time": 3.133587598800659,
      "step": 7238
    },
    {
      "epoch": 2.761077880859375e-05,
      "step": 7238,
      "training_step_time": 9.904142141342163
    },
    {
      "epoch": 2.761077880859375e-05,
      "model_forward_time": 3.405168056488037,
      "step": 7238
    },
    {
      "epoch": 2.761077880859375e-05,
      "step": 7238,
      "training_step_time": 10.27363657951355
    },
    {
      "epoch": 2.7614593505859375e-05,
      "model_forward_time": 3.152092218399048,
      "step": 7239
    },
    {
      "epoch": 2.7614593505859375e-05,
      "step": 7239,
      "training_step_time": 9.927746057510376
    },
    {
      "epoch": 2.7614593505859375e-05,
      "model_forward_time": 3.1340129375457764,
      "step": 7239
    },
    {
      "epoch": 2.7614593505859375e-05,
      "step": 7239,
      "training_step_time": 9.911283016204834
    },
    {
      "epoch": 2.7614593505859375e-05,
      "model_forward_time": 3.1390135288238525,
      "step": 7239
    },
    {
      "epoch": 2.7614593505859375e-05,
      "step": 7239,
      "training_step_time": 9.914278507232666
    },
    {
      "epoch": 2.7614593505859375e-05,
      "model_forward_time": 3.1352062225341797,
      "step": 7239
    },
    {
      "epoch": 2.7614593505859375e-05,
      "step": 7239,
      "training_step_time": 10.26597809791565
    },
    {
      "epoch": 2.7618408203125e-05,
      "grad_norm": 0.44493791460990906,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 7240
    },
    {
      "epoch": 2.7618408203125e-05,
      "model_forward_time": 3.4142744541168213,
      "step": 7240
    },
    {
      "dynamics_loss_avg": 0.06345069017261266,
      "epoch": 2.7618408203125e-05,
      "step": 7240
    },
    {
      "action_loss_avg": 0.024506524112075566,
      "epoch": 2.7618408203125e-05,
      "step": 7240
    },
    {
      "epoch": 2.7618408203125e-05,
      "step": 7240,
      "training_step_time": 10.187681436538696
    },
    {
      "epoch": 2.7618408203125e-05,
      "model_forward_time": 3.1337597370147705,
      "step": 7240
    },
    {
      "epoch": 2.7618408203125e-05,
      "step": 7240,
      "training_step_time": 9.907931327819824
    },
    {
      "epoch": 2.7618408203125e-05,
      "model_forward_time": 3.131727933883667,
      "step": 7240
    },
    {
      "epoch": 2.7618408203125e-05,
      "step": 7240,
      "training_step_time": 9.89536428451538
    },
    {
      "epoch": 2.7618408203125e-05,
      "model_forward_time": 3.1295199394226074,
      "step": 7240
    },
    {
      "epoch": 2.7618408203125e-05,
      "step": 7240,
      "training_step_time": 10.267980813980103
    },
    {
      "epoch": 2.7622222900390624e-05,
      "model_forward_time": 3.1597094535827637,
      "step": 7241
    },
    {
      "epoch": 2.7622222900390624e-05,
      "step": 7241,
      "training_step_time": 9.929917097091675
    },
    {
      "epoch": 2.7622222900390624e-05,
      "model_forward_time": 3.38869047164917,
      "step": 7241
    },
    {
      "epoch": 2.7622222900390624e-05,
      "step": 7241,
      "training_step_time": 10.161433458328247
    },
    {
      "epoch": 2.7622222900390624e-05,
      "model_forward_time": 3.140005111694336,
      "step": 7241
    },
    {
      "epoch": 2.7622222900390624e-05,
      "step": 7241,
      "training_step_time": 9.915165424346924
    },
    {
      "epoch": 2.7622222900390624e-05,
      "model_forward_time": 3.1372482776641846,
      "step": 7241
    },
    {
      "epoch": 2.7622222900390624e-05,
      "step": 7241,
      "training_step_time": 10.020490884780884
    },
    {
      "epoch": 2.762603759765625e-05,
      "model_forward_time": 3.172799825668335,
      "step": 7242
    },
    {
      "epoch": 2.762603759765625e-05,
      "step": 7242,
      "training_step_time": 9.948434114456177
    },
    {
      "epoch": 2.762603759765625e-05,
      "model_forward_time": 3.1369245052337646,
      "step": 7242
    },
    {
      "epoch": 2.762603759765625e-05,
      "step": 7242,
      "training_step_time": 9.9125337600708
    },
    {
      "epoch": 2.762603759765625e-05,
      "model_forward_time": 3.4433462619781494,
      "step": 7242
    },
    {
      "dynamics_loss_avg": 0.049035387113690376,
      "epoch": 2.762603759765625e-05,
      "step": 7242
    },
    {
      "action_loss_avg": 0.019700295850634576,
      "epoch": 2.762603759765625e-05,
      "step": 7242
    },
    {
      "epoch": 2.762603759765625e-05,
      "step": 7242,
      "training_step_time": 10.216398000717163
    },
    {
      "epoch": 2.762603759765625e-05,
      "model_forward_time": 3.136507034301758,
      "step": 7242
    },
    {
      "epoch": 2.762603759765625e-05,
      "step": 7242,
      "training_step_time": 10.31199312210083
    },
    {
      "epoch": 2.7629852294921876e-05,
      "model_forward_time": 3.169064521789551,
      "step": 7243
    },
    {
      "epoch": 2.7629852294921876e-05,
      "step": 7243,
      "training_step_time": 9.947154521942139
    },
    {
      "epoch": 2.7629852294921876e-05,
      "model_forward_time": 3.1374738216400146,
      "step": 7243
    },
    {
      "epoch": 2.7629852294921876e-05,
      "step": 7243,
      "training_step_time": 9.923187732696533
    },
    {
      "epoch": 2.7629852294921876e-05,
      "model_forward_time": 3.1361186504364014,
      "step": 7243
    },
    {
      "epoch": 2.7629852294921876e-05,
      "step": 7243,
      "training_step_time": 9.918391942977905
    },
    {
      "epoch": 2.7629852294921876e-05,
      "model_forward_time": 3.4078288078308105,
      "step": 7243
    },
    {
      "epoch": 2.7629852294921876e-05,
      "step": 7243,
      "training_step_time": 10.288922786712646
    },
    {
      "epoch": 2.76336669921875e-05,
      "model_forward_time": 3.177178144454956,
      "step": 7244
    },
    {
      "epoch": 2.76336669921875e-05,
      "step": 7244,
      "training_step_time": 9.955531120300293
    },
    {
      "epoch": 2.76336669921875e-05,
      "model_forward_time": 3.14216947555542,
      "step": 7244
    },
    {
      "epoch": 2.76336669921875e-05,
      "step": 7244,
      "training_step_time": 9.919878482818604
    },
    {
      "epoch": 2.76336669921875e-05,
      "model_forward_time": 3.1420507431030273,
      "step": 7244
    },
    {
      "epoch": 2.76336669921875e-05,
      "step": 7244,
      "training_step_time": 9.921664714813232
    },
    {
      "epoch": 2.76336669921875e-05,
      "model_forward_time": 3.142085313796997,
      "step": 7244
    },
    {
      "epoch": 2.76336669921875e-05,
      "step": 7244,
      "training_step_time": 10.293145656585693
    },
    {
      "epoch": 2.7637481689453125e-05,
      "model_forward_time": 3.450552463531494,
      "step": 7245
    },
    {
      "dynamics_loss_avg": 0.06318254768848419,
      "epoch": 2.7637481689453125e-05,
      "step": 7245
    },
    {
      "action_loss_avg": 0.026928021013736723,
      "epoch": 2.7637481689453125e-05,
      "step": 7245
    },
    {
      "epoch": 2.7637481689453125e-05,
      "step": 7245,
      "training_step_time": 10.231127500534058
    },
    {
      "epoch": 2.7637481689453125e-05,
      "model_forward_time": 3.1417808532714844,
      "step": 7245
    },
    {
      "epoch": 2.7637481689453125e-05,
      "step": 7245,
      "training_step_time": 9.922311067581177
    },
    {
      "epoch": 2.7637481689453125e-05,
      "model_forward_time": 3.1422197818756104,
      "step": 7245
    },
    {
      "epoch": 2.7637481689453125e-05,
      "step": 7245,
      "training_step_time": 9.919585227966309
    },
    {
      "epoch": 2.7637481689453125e-05,
      "model_forward_time": 3.1468355655670166,
      "step": 7245
    },
    {
      "epoch": 2.7637481689453125e-05,
      "step": 7245,
      "training_step_time": 10.306217432022095
    },
    {
      "epoch": 2.764129638671875e-05,
      "model_forward_time": 3.183126211166382,
      "step": 7246
    },
    {
      "epoch": 2.764129638671875e-05,
      "step": 7246,
      "training_step_time": 9.961891651153564
    },
    {
      "epoch": 2.764129638671875e-05,
      "model_forward_time": 3.4255166053771973,
      "step": 7246
    },
    {
      "epoch": 2.764129638671875e-05,
      "step": 7246,
      "training_step_time": 10.199776411056519
    },
    {
      "epoch": 2.764129638671875e-05,
      "model_forward_time": 3.1431727409362793,
      "step": 7246
    },
    {
      "epoch": 2.764129638671875e-05,
      "step": 7246,
      "training_step_time": 9.917999267578125
    },
    {
      "epoch": 2.764129638671875e-05,
      "model_forward_time": 3.143674373626709,
      "step": 7246
    },
    {
      "epoch": 2.764129638671875e-05,
      "step": 7246,
      "training_step_time": 10.005429029464722
    },
    {
      "epoch": 2.7645111083984375e-05,
      "model_forward_time": 3.1558351516723633,
      "step": 7247
    },
    {
      "epoch": 2.7645111083984375e-05,
      "step": 7247,
      "training_step_time": 9.926953792572021
    },
    {
      "epoch": 2.7645111083984375e-05,
      "model_forward_time": 3.1420087814331055,
      "step": 7247
    },
    {
      "epoch": 2.7645111083984375e-05,
      "step": 7247,
      "training_step_time": 9.920613765716553
    },
    {
      "epoch": 2.7645111083984375e-05,
      "model_forward_time": 3.4044013023376465,
      "step": 7247
    },
    {
      "dynamics_loss_avg": 0.0563711354508996,
      "epoch": 2.7645111083984375e-05,
      "step": 7247
    },
    {
      "action_loss_avg": 0.019900096394121648,
      "epoch": 2.7645111083984375e-05,
      "step": 7247
    },
    {
      "epoch": 2.7645111083984375e-05,
      "step": 7247,
      "training_step_time": 10.183130741119385
    },
    {
      "epoch": 2.7645111083984375e-05,
      "model_forward_time": 3.146559715270996,
      "step": 7247
    },
    {
      "epoch": 2.7645111083984375e-05,
      "step": 7247,
      "training_step_time": 10.294468641281128
    },
    {
      "epoch": 2.764892578125e-05,
      "model_forward_time": 3.1767232418060303,
      "step": 7248
    },
    {
      "epoch": 2.764892578125e-05,
      "step": 7248,
      "training_step_time": 9.957541942596436
    },
    {
      "epoch": 2.764892578125e-05,
      "model_forward_time": 3.13893723487854,
      "step": 7248
    },
    {
      "epoch": 2.764892578125e-05,
      "step": 7248,
      "training_step_time": 9.914732456207275
    },
    {
      "epoch": 2.764892578125e-05,
      "model_forward_time": 3.144325017929077,
      "step": 7248
    },
    {
      "epoch": 2.764892578125e-05,
      "step": 7248,
      "training_step_time": 9.916245222091675
    },
    {
      "epoch": 2.764892578125e-05,
      "model_forward_time": 3.4307284355163574,
      "step": 7248
    },
    {
      "epoch": 2.764892578125e-05,
      "step": 7248,
      "training_step_time": 10.300425291061401
    },
    {
      "epoch": 2.7652740478515624e-05,
      "model_forward_time": 3.197441339492798,
      "step": 7249
    },
    {
      "epoch": 2.7652740478515624e-05,
      "step": 7249,
      "training_step_time": 9.976706504821777
    },
    {
      "epoch": 2.7652740478515624e-05,
      "model_forward_time": 3.1349055767059326,
      "step": 7249
    },
    {
      "epoch": 2.7652740478515624e-05,
      "step": 7249,
      "training_step_time": 9.907601356506348
    },
    {
      "epoch": 2.7652740478515624e-05,
      "model_forward_time": 3.139833688735962,
      "step": 7249
    },
    {
      "epoch": 2.7652740478515624e-05,
      "step": 7249,
      "training_step_time": 9.917282104492188
    },
    {
      "epoch": 2.7652740478515624e-05,
      "model_forward_time": 3.136268377304077,
      "step": 7249
    },
    {
      "epoch": 2.7652740478515624e-05,
      "step": 7249,
      "training_step_time": 10.003490447998047
    },
    {
      "epoch": 2.765655517578125e-05,
      "grad_norm": 0.33952194452285767,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 7250
    },
    {
      "epoch": 2.765655517578125e-05,
      "model_forward_time": 3.1564126014709473,
      "step": 7250
    },
    {
      "dynamics_loss_avg": 0.07103754784911871,
      "epoch": 2.765655517578125e-05,
      "step": 7250
    },
    {
      "action_loss_avg": 0.035811344347894195,
      "epoch": 2.765655517578125e-05,
      "step": 7250
    },
    {
      "epoch": 2.765655517578125e-05,
      "step": 7250,
      "training_step_time": 10.223283767700195
    },
    {
      "epoch": 2.765655517578125e-05,
      "model_forward_time": 3.1450917720794678,
      "step": 7250
    },
    {
      "epoch": 2.765655517578125e-05,
      "step": 7250,
      "training_step_time": 9.916995763778687
    },
    {
      "epoch": 2.765655517578125e-05,
      "model_forward_time": 2.436189889907837,
      "step": 7250
    },
    {
      "epoch": 2.765655517578125e-05,
      "step": 7250,
      "training_step_time": 7.4613196849823
    },
    {
      "epoch": 2.765655517578125e-05,
      "model_forward_time": 3.135791063308716,
      "step": 7250
    },
    {
      "epoch": 2.765655517578125e-05,
      "step": 7250,
      "training_step_time": 10.020334005355835
    },
    {
      "epoch": 2.7660369873046876e-05,
      "model_forward_time": 3.156996488571167,
      "step": 7251
    },
    {
      "epoch": 2.7660369873046876e-05,
      "step": 7251,
      "training_step_time": 9.93213701248169
    },
    {
      "epoch": 2.7660369873046876e-05,
      "model_forward_time": 2.439563751220703,
      "step": 7251
    },
    {
      "epoch": 2.7660369873046876e-05,
      "step": 7251,
      "training_step_time": 7.469941854476929
    },
    {
      "epoch": 2.7660369873046876e-05,
      "model_forward_time": 3.411731481552124,
      "step": 7251
    },
    {
      "epoch": 2.7660369873046876e-05,
      "step": 7251,
      "training_step_time": 10.186585903167725
    },
    {
      "epoch": 2.7660369873046876e-05,
      "model_forward_time": 2.440063238143921,
      "step": 7251
    },
    {
      "epoch": 2.7660369873046876e-05,
      "step": 7251,
      "training_step_time": 10.277136087417603
    },
    {
      "epoch": 2.76641845703125e-05,
      "model_forward_time": 3.16444730758667,
      "step": 7252
    },
    {
      "epoch": 2.76641845703125e-05,
      "step": 7252,
      "training_step_time": 9.94076657295227
    },
    {
      "epoch": 2.76641845703125e-05,
      "model_forward_time": 3.1395390033721924,
      "step": 7252
    },
    {
      "epoch": 2.76641845703125e-05,
      "step": 7252,
      "training_step_time": 9.919116258621216
    },
    {
      "epoch": 2.76641845703125e-05,
      "model_forward_time": 1.745342493057251,
      "step": 7252
    },
    {
      "dynamics_loss_avg": 0.04751522913575172,
      "epoch": 2.76641845703125e-05,
      "step": 7252
    },
    {
      "action_loss_avg": 0.025009887642227113,
      "epoch": 2.76641845703125e-05,
      "step": 7252
    },
    {
      "epoch": 2.76641845703125e-05,
      "step": 7252,
      "training_step_time": 5.207425594329834
    },
    {
      "epoch": 2.76641845703125e-05,
      "model_forward_time": 3.3889803886413574,
      "step": 7252
    },
    {
      "epoch": 2.76641845703125e-05,
      "step": 7252,
      "training_step_time": 10.255061388015747
    },
    {
      "epoch": 2.7667999267578125e-05,
      "model_forward_time": 3.1502368450164795,
      "step": 7253
    },
    {
      "epoch": 2.7667999267578125e-05,
      "step": 7253,
      "training_step_time": 9.925524950027466
    },
    {
      "epoch": 2.7667999267578125e-05,
      "model_forward_time": 3.139474868774414,
      "step": 7253
    },
    {
      "epoch": 2.7667999267578125e-05,
      "step": 7253,
      "training_step_time": 9.919543981552124
    },
    {
      "epoch": 2.7667999267578125e-05,
      "model_forward_time": 3.145113706588745,
      "step": 7253
    },
    {
      "epoch": 2.7667999267578125e-05,
      "step": 7253,
      "training_step_time": 9.92135763168335
    },
    {
      "epoch": 2.7667999267578125e-05,
      "model_forward_time": 2.434507369995117,
      "step": 7253
    },
    {
      "epoch": 2.7667999267578125e-05,
      "step": 7253,
      "training_step_time": 10.31401538848877
    },
    {
      "epoch": 2.767181396484375e-05,
      "model_forward_time": 3.412416934967041,
      "step": 7254
    },
    {
      "epoch": 2.767181396484375e-05,
      "step": 7254,
      "training_step_time": 10.188196182250977
    },
    {
      "epoch": 2.767181396484375e-05,
      "model_forward_time": 3.1431264877319336,
      "step": 7254
    },
    {
      "epoch": 2.767181396484375e-05,
      "step": 7254,
      "training_step_time": 9.921832084655762
    },
    {
      "epoch": 2.767181396484375e-05,
      "model_forward_time": 2.439105987548828,
      "step": 7254
    },
    {
      "epoch": 2.767181396484375e-05,
      "step": 7254,
      "training_step_time": 7.4680187702178955
    },
    {
      "epoch": 2.767181396484375e-05,
      "model_forward_time": 2.4369211196899414,
      "step": 7254
    },
    {
      "epoch": 2.767181396484375e-05,
      "step": 7254,
      "training_step_time": 10.301438331604004
    },
    {
      "epoch": 2.7675628662109375e-05,
      "model_forward_time": 3.1568009853363037,
      "step": 7255
    },
    {
      "dynamics_loss_avg": 0.05064526498317719,
      "epoch": 2.7675628662109375e-05,
      "step": 7255
    },
    {
      "action_loss_avg": 0.028839578526094554,
      "epoch": 2.7675628662109375e-05,
      "step": 7255
    },
    {
      "epoch": 2.7675628662109375e-05,
      "step": 7255,
      "training_step_time": 9.928673028945923
    },
    {
      "epoch": 2.7675628662109375e-05,
      "model_forward_time": 3.1441152095794678,
      "step": 7255
    },
    {
      "epoch": 2.7675628662109375e-05,
      "step": 7255,
      "training_step_time": 10.186975002288818
    },
    {
      "epoch": 2.7675628662109375e-05,
      "model_forward_time": 2.4418327808380127,
      "step": 7255
    },
    {
      "epoch": 2.7675628662109375e-05,
      "step": 7255,
      "training_step_time": 7.473783016204834
    },
    {
      "epoch": 2.7675628662109375e-05,
      "model_forward_time": 3.146723508834839,
      "step": 7255
    },
    {
      "epoch": 2.7675628662109375e-05,
      "step": 7255,
      "training_step_time": 10.038507461547852
    },
    {
      "epoch": 2.7679443359375e-05,
      "model_forward_time": 3.1556246280670166,
      "step": 7256
    },
    {
      "epoch": 2.7679443359375e-05,
      "step": 7256,
      "training_step_time": 9.92996096611023
    },
    {
      "epoch": 2.7679443359375e-05,
      "model_forward_time": 3.1395866870880127,
      "step": 7256
    },
    {
      "epoch": 2.7679443359375e-05,
      "step": 7256,
      "training_step_time": 9.917448997497559
    },
    {
      "epoch": 2.7679443359375e-05,
      "model_forward_time": 3.4057297706604004,
      "step": 7256
    },
    {
      "epoch": 2.7679443359375e-05,
      "step": 7256,
      "training_step_time": 10.181333065032959
    },
    {
      "epoch": 2.7679443359375e-05,
      "model_forward_time": 2.436880111694336,
      "step": 7256
    },
    {
      "epoch": 2.7679443359375e-05,
      "step": 7256,
      "training_step_time": 10.259502172470093
    },
    {
      "epoch": 2.7683258056640624e-05,
      "model_forward_time": 1.7880527973175049,
      "step": 7257
    },
    {
      "epoch": 2.7683258056640624e-05,
      "step": 7257,
      "training_step_time": 5.247082948684692
    },
    {
      "epoch": 2.7683258056640624e-05,
      "model_forward_time": 3.1307334899902344,
      "step": 7257
    },
    {
      "epoch": 2.7683258056640624e-05,
      "step": 7257,
      "training_step_time": 9.902908563613892
    },
    {
      "epoch": 2.7683258056640624e-05,
      "model_forward_time": 3.137286901473999,
      "step": 7257
    },
    {
      "dynamics_loss_avg": 0.05372826866805554,
      "epoch": 2.7683258056640624e-05,
      "step": 7257
    },
    {
      "action_loss_avg": 0.01804460524581373,
      "epoch": 2.7683258056640624e-05,
      "step": 7257
    },
    {
      "epoch": 2.7683258056640624e-05,
      "step": 7257,
      "training_step_time": 9.911774396896362
    },
    {
      "epoch": 2.7683258056640624e-05,
      "model_forward_time": 3.1419105529785156,
      "step": 7257
    },
    {
      "epoch": 2.7683258056640624e-05,
      "step": 7257,
      "training_step_time": 10.300630331039429
    },
    {
      "epoch": 2.768707275390625e-05,
      "model_forward_time": 2.7358407974243164,
      "step": 7258
    },
    {
      "epoch": 2.768707275390625e-05,
      "step": 7258,
      "training_step_time": 7.763280630111694
    },
    {
      "epoch": 2.768707275390625e-05,
      "model_forward_time": 2.4305078983306885,
      "step": 7258
    },
    {
      "epoch": 2.768707275390625e-05,
      "step": 7258,
      "training_step_time": 7.453728199005127
    },
    {
      "epoch": 2.768707275390625e-05,
      "model_forward_time": 3.136021375656128,
      "step": 7258
    },
    {
      "epoch": 2.768707275390625e-05,
      "step": 7258,
      "training_step_time": 9.90736699104309
    },
    {
      "epoch": 2.768707275390625e-05,
      "model_forward_time": 3.1393678188323975,
      "step": 7258
    },
    {
      "epoch": 2.768707275390625e-05,
      "step": 7258,
      "training_step_time": 10.29058027267456
    },
    {
      "epoch": 2.7690887451171876e-05,
      "model_forward_time": 2.451943874359131,
      "step": 7259
    },
    {
      "epoch": 2.7690887451171876e-05,
      "step": 7259,
      "training_step_time": 7.477420091629028
    },
    {
      "epoch": 2.7690887451171876e-05,
      "model_forward_time": 2.429844379425049,
      "step": 7259
    },
    {
      "epoch": 2.7690887451171876e-05,
      "step": 7259,
      "training_step_time": 7.720341682434082
    },
    {
      "epoch": 2.7690887451171876e-05,
      "model_forward_time": 2.4305081367492676,
      "step": 7259
    },
    {
      "epoch": 2.7690887451171876e-05,
      "step": 7259,
      "training_step_time": 7.454413175582886
    },
    {
      "epoch": 2.7690887451171876e-05,
      "model_forward_time": 1.7440259456634521,
      "step": 7259
    },
    {
      "epoch": 2.7690887451171876e-05,
      "step": 7259,
      "training_step_time": 10.2682204246521
    },
    {
      "epoch": 2.76947021484375e-05,
      "grad_norm": 0.43967312574386597,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 7260
    },
    {
      "epoch": 2.76947021484375e-05,
      "model_forward_time": 3.1757116317749023,
      "step": 7260
    },
    {
      "dynamics_loss_avg": 0.050238530151546,
      "epoch": 2.76947021484375e-05,
      "step": 7260
    },
    {
      "action_loss_avg": 0.0280812447424978,
      "epoch": 2.76947021484375e-05,
      "step": 7260
    },
    {
      "epoch": 2.76947021484375e-05,
      "step": 7260,
      "training_step_time": 9.94585132598877
    },
    {
      "epoch": 2.76947021484375e-05,
      "model_forward_time": 3.1418745517730713,
      "step": 7260
    },
    {
      "epoch": 2.76947021484375e-05,
      "step": 7260,
      "training_step_time": 9.908941268920898
    },
    {
      "epoch": 2.76947021484375e-05,
      "model_forward_time": 3.133183240890503,
      "step": 7260
    },
    {
      "epoch": 2.76947021484375e-05,
      "step": 7260,
      "training_step_time": 9.898294448852539
    },
    {
      "epoch": 2.76947021484375e-05,
      "model_forward_time": 3.388361930847168,
      "step": 7260
    },
    {
      "epoch": 2.76947021484375e-05,
      "step": 7260,
      "training_step_time": 10.277950048446655
    },
    {
      "epoch": 2.7698516845703125e-05,
      "model_forward_time": 3.1410701274871826,
      "step": 7261
    },
    {
      "epoch": 2.7698516845703125e-05,
      "step": 7261,
      "training_step_time": 9.907318115234375
    },
    {
      "epoch": 2.7698516845703125e-05,
      "model_forward_time": 3.135218620300293,
      "step": 7261
    },
    {
      "epoch": 2.7698516845703125e-05,
      "step": 7261,
      "training_step_time": 9.903030157089233
    },
    {
      "epoch": 2.7698516845703125e-05,
      "model_forward_time": 3.138709306716919,
      "step": 7261
    },
    {
      "epoch": 2.7698516845703125e-05,
      "step": 7261,
      "training_step_time": 9.908940076828003
    },
    {
      "epoch": 2.7698516845703125e-05,
      "model_forward_time": 3.137849807739258,
      "step": 7261
    },
    {
      "epoch": 2.7698516845703125e-05,
      "step": 7261,
      "training_step_time": 10.016350269317627
    },
    {
      "epoch": 2.770233154296875e-05,
      "model_forward_time": 2.7612674236297607,
      "step": 7262
    },
    {
      "epoch": 2.770233154296875e-05,
      "step": 7262,
      "training_step_time": 7.788572788238525
    },
    {
      "epoch": 2.770233154296875e-05,
      "model_forward_time": 2.4337680339813232,
      "step": 7262
    },
    {
      "epoch": 2.770233154296875e-05,
      "step": 7262,
      "training_step_time": 7.461155891418457
    },
    {
      "epoch": 2.770233154296875e-05,
      "model_forward_time": 2.4319443702697754,
      "step": 7262
    },
    {
      "dynamics_loss_avg": 0.06711104363203049,
      "epoch": 2.770233154296875e-05,
      "step": 7262
    },
    {
      "action_loss_avg": 0.021348144160583615,
      "epoch": 2.770233154296875e-05,
      "step": 7262
    },
    {
      "epoch": 2.770233154296875e-05,
      "step": 7262,
      "training_step_time": 7.463192701339722
    },
    {
      "epoch": 2.770233154296875e-05,
      "model_forward_time": 2.4315357208251953,
      "step": 7262
    },
    {
      "epoch": 2.770233154296875e-05,
      "step": 7262,
      "training_step_time": 10.331462860107422
    },
    {
      "epoch": 2.7706146240234375e-05,
      "model_forward_time": 2.446101427078247,
      "step": 7263
    },
    {
      "epoch": 2.7706146240234375e-05,
      "step": 7263,
      "training_step_time": 7.473518133163452
    },
    {
      "epoch": 2.7706146240234375e-05,
      "model_forward_time": 2.7272539138793945,
      "step": 7263
    },
    {
      "epoch": 2.7706146240234375e-05,
      "step": 7263,
      "training_step_time": 7.751957416534424
    },
    {
      "epoch": 2.7706146240234375e-05,
      "model_forward_time": 2.4313905239105225,
      "step": 7263
    },
    {
      "epoch": 2.7706146240234375e-05,
      "step": 7263,
      "training_step_time": 7.4529807567596436
    },
    {
      "epoch": 2.7706146240234375e-05,
      "model_forward_time": 3.13460636138916,
      "step": 7263
    },
    {
      "epoch": 2.7706146240234375e-05,
      "step": 7263,
      "training_step_time": 10.275802850723267
    },
    {
      "epoch": 2.77099609375e-05,
      "model_forward_time": 2.44956111907959,
      "step": 7264
    },
    {
      "epoch": 2.77099609375e-05,
      "step": 7264,
      "training_step_time": 7.4703710079193115
    },
    {
      "epoch": 2.77099609375e-05,
      "model_forward_time": 2.432368040084839,
      "step": 7264
    },
    {
      "epoch": 2.77099609375e-05,
      "step": 7264,
      "training_step_time": 7.451383113861084
    },
    {
      "epoch": 2.77099609375e-05,
      "model_forward_time": 2.4319400787353516,
      "step": 7264
    },
    {
      "epoch": 2.77099609375e-05,
      "step": 7264,
      "training_step_time": 7.44513463973999
    },
    {
      "epoch": 2.77099609375e-05,
      "model_forward_time": 2.0137100219726562,
      "step": 7264
    },
    {
      "epoch": 2.77099609375e-05,
      "step": 7264,
      "training_step_time": 10.302512168884277
    },
    {
      "epoch": 2.7713775634765624e-05,
      "model_forward_time": 3.172084093093872,
      "step": 7265
    },
    {
      "dynamics_loss_avg": 0.056665233336389065,
      "epoch": 2.7713775634765624e-05,
      "step": 7265
    },
    {
      "action_loss_avg": 0.02409951398149133,
      "epoch": 2.7713775634765624e-05,
      "step": 7265
    },
    {
      "epoch": 2.7713775634765624e-05,
      "step": 7265,
      "training_step_time": 9.936246633529663
    },
    {
      "epoch": 2.7713775634765624e-05,
      "model_forward_time": 3.1331584453582764,
      "step": 7265
    },
    {
      "epoch": 2.7713775634765624e-05,
      "step": 7265,
      "training_step_time": 9.892389059066772
    },
    {
      "epoch": 2.7713775634765624e-05,
      "model_forward_time": 3.1375834941864014,
      "step": 7265
    },
    {
      "epoch": 2.7713775634765624e-05,
      "step": 7265,
      "training_step_time": 9.901168823242188
    },
    {
      "epoch": 2.7713775634765624e-05,
      "model_forward_time": 3.1304969787597656,
      "step": 7265
    },
    {
      "epoch": 2.7713775634765624e-05,
      "step": 7265,
      "training_step_time": 10.282709836959839
    },
    {
      "epoch": 2.771759033203125e-05,
      "model_forward_time": 3.1459474563598633,
      "step": 7266
    },
    {
      "epoch": 2.771759033203125e-05,
      "step": 7266,
      "training_step_time": 10.187602281570435
    },
    {
      "epoch": 2.771759033203125e-05,
      "model_forward_time": 2.432722330093384,
      "step": 7266
    },
    {
      "epoch": 2.771759033203125e-05,
      "step": 7266,
      "training_step_time": 7.451539754867554
    },
    {
      "epoch": 2.771759033203125e-05,
      "model_forward_time": 2.4293172359466553,
      "step": 7266
    },
    {
      "epoch": 2.771759033203125e-05,
      "step": 7266,
      "training_step_time": 7.447703838348389
    },
    {
      "epoch": 2.771759033203125e-05,
      "model_forward_time": 2.4314985275268555,
      "step": 7266
    },
    {
      "epoch": 2.771759033203125e-05,
      "step": 7266,
      "training_step_time": 10.015475988388062
    },
    {
      "epoch": 2.7721405029296876e-05,
      "model_forward_time": 3.157467842102051,
      "step": 7267
    },
    {
      "epoch": 2.7721405029296876e-05,
      "step": 7267,
      "training_step_time": 9.923180341720581
    },
    {
      "epoch": 2.7721405029296876e-05,
      "model_forward_time": 2.4329042434692383,
      "step": 7267
    },
    {
      "epoch": 2.7721405029296876e-05,
      "step": 7267,
      "training_step_time": 7.453451871871948
    },
    {
      "epoch": 2.7721405029296876e-05,
      "model_forward_time": 3.4123308658599854,
      "step": 7267
    },
    {
      "dynamics_loss_avg": 0.06311742532998324,
      "epoch": 2.7721405029296876e-05,
      "step": 7267
    },
    {
      "action_loss_avg": 0.03326101182028651,
      "epoch": 2.7721405029296876e-05,
      "step": 7267
    },
    {
      "epoch": 2.7721405029296876e-05,
      "step": 7267,
      "training_step_time": 10.174156904220581
    },
    {
      "epoch": 2.7721405029296876e-05,
      "model_forward_time": 2.4328701496124268,
      "step": 7267
    },
    {
      "epoch": 2.7721405029296876e-05,
      "step": 7267,
      "training_step_time": 10.25312614440918
    },
    {
      "epoch": 2.77252197265625e-05,
      "model_forward_time": 3.1445794105529785,
      "step": 7268
    },
    {
      "epoch": 2.77252197265625e-05,
      "step": 7268,
      "training_step_time": 9.908207178115845
    },
    {
      "epoch": 2.77252197265625e-05,
      "model_forward_time": 1.745513677597046,
      "step": 7268
    },
    {
      "epoch": 2.77252197265625e-05,
      "step": 7268,
      "training_step_time": 5.196500539779663
    },
    {
      "epoch": 2.77252197265625e-05,
      "model_forward_time": 2.429394245147705,
      "step": 7268
    },
    {
      "epoch": 2.77252197265625e-05,
      "step": 7268,
      "training_step_time": 7.44776463508606
    },
    {
      "epoch": 2.77252197265625e-05,
      "model_forward_time": 3.1332902908325195,
      "step": 7268
    },
    {
      "epoch": 2.77252197265625e-05,
      "step": 7268,
      "training_step_time": 10.30340576171875
    },
    {
      "epoch": 2.7729034423828125e-05,
      "model_forward_time": 3.1550259590148926,
      "step": 7269
    },
    {
      "epoch": 2.7729034423828125e-05,
      "step": 7269,
      "training_step_time": 9.92382287979126
    },
    {
      "epoch": 2.7729034423828125e-05,
      "model_forward_time": 3.1407482624053955,
      "step": 7269
    },
    {
      "epoch": 2.7729034423828125e-05,
      "step": 7269,
      "training_step_time": 9.913161039352417
    },
    {
      "epoch": 2.7729034423828125e-05,
      "model_forward_time": 2.4369006156921387,
      "step": 7269
    },
    {
      "epoch": 2.7729034423828125e-05,
      "step": 7269,
      "training_step_time": 7.466339111328125
    },
    {
      "epoch": 2.7729034423828125e-05,
      "model_forward_time": 3.1402738094329834,
      "step": 7269
    },
    {
      "epoch": 2.7729034423828125e-05,
      "step": 7269,
      "training_step_time": 10.308595895767212
    },
    {
      "epoch": 2.773284912109375e-05,
      "grad_norm": 0.45006322860717773,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 7270
    },
    {
      "epoch": 2.773284912109375e-05,
      "model_forward_time": 3.421809434890747,
      "step": 7270
    },
    {
      "dynamics_loss_avg": 0.06543442597612739,
      "epoch": 2.773284912109375e-05,
      "step": 7270
    },
    {
      "action_loss_avg": 0.02975941466866061,
      "epoch": 2.773284912109375e-05,
      "step": 7270
    },
    {
      "epoch": 2.773284912109375e-05,
      "step": 7270,
      "training_step_time": 10.194353818893433
    },
    {
      "epoch": 2.773284912109375e-05,
      "model_forward_time": 3.1405677795410156,
      "step": 7270
    },
    {
      "epoch": 2.773284912109375e-05,
      "step": 7270,
      "training_step_time": 9.907804727554321
    },
    {
      "epoch": 2.773284912109375e-05,
      "model_forward_time": 3.1356213092803955,
      "step": 7270
    },
    {
      "epoch": 2.773284912109375e-05,
      "step": 7270,
      "training_step_time": 9.899015665054321
    },
    {
      "epoch": 2.773284912109375e-05,
      "model_forward_time": 2.433816432952881,
      "step": 7270
    },
    {
      "epoch": 2.773284912109375e-05,
      "step": 7270,
      "training_step_time": 10.268362045288086
    },
    {
      "epoch": 2.7736663818359375e-05,
      "model_forward_time": 2.4452056884765625,
      "step": 7271
    },
    {
      "epoch": 2.7736663818359375e-05,
      "step": 7271,
      "training_step_time": 7.463662147521973
    },
    {
      "epoch": 2.7736663818359375e-05,
      "model_forward_time": 2.433424234390259,
      "step": 7271
    },
    {
      "epoch": 2.7736663818359375e-05,
      "step": 7271,
      "training_step_time": 7.454656362533569
    },
    {
      "epoch": 2.7736663818359375e-05,
      "model_forward_time": 2.687357187271118,
      "step": 7271
    },
    {
      "epoch": 2.7736663818359375e-05,
      "step": 7271,
      "training_step_time": 7.705541133880615
    },
    {
      "epoch": 2.7736663818359375e-05,
      "model_forward_time": 3.1371757984161377,
      "step": 7271
    },
    {
      "epoch": 2.7736663818359375e-05,
      "step": 7271,
      "training_step_time": 10.004438161849976
    },
    {
      "epoch": 2.7740478515625e-05,
      "model_forward_time": 3.1568117141723633,
      "step": 7272
    },
    {
      "epoch": 2.7740478515625e-05,
      "step": 7272,
      "training_step_time": 9.92201280593872
    },
    {
      "epoch": 2.7740478515625e-05,
      "model_forward_time": 2.4371509552001953,
      "step": 7272
    },
    {
      "epoch": 2.7740478515625e-05,
      "step": 7272,
      "training_step_time": 7.458975076675415
    },
    {
      "epoch": 2.7740478515625e-05,
      "model_forward_time": 3.1409213542938232,
      "step": 7272
    },
    {
      "dynamics_loss_avg": 0.0646192267537117,
      "epoch": 2.7740478515625e-05,
      "step": 7272
    },
    {
      "action_loss_avg": 0.02669799868017435,
      "epoch": 2.7740478515625e-05,
      "step": 7272
    },
    {
      "epoch": 2.7740478515625e-05,
      "step": 7272,
      "training_step_time": 9.9098961353302
    },
    {
      "epoch": 2.7740478515625e-05,
      "model_forward_time": 3.4192559719085693,
      "step": 7272
    },
    {
      "epoch": 2.7740478515625e-05,
      "step": 7272,
      "training_step_time": 10.281995296478271
    },
    {
      "epoch": 2.7744293212890624e-05,
      "model_forward_time": 3.1479125022888184,
      "step": 7273
    },
    {
      "epoch": 2.7744293212890624e-05,
      "step": 7273,
      "training_step_time": 9.915129661560059
    },
    {
      "epoch": 2.7744293212890624e-05,
      "model_forward_time": 3.1389124393463135,
      "step": 7273
    },
    {
      "epoch": 2.7744293212890624e-05,
      "step": 7273,
      "training_step_time": 9.907638549804688
    },
    {
      "epoch": 2.7744293212890624e-05,
      "model_forward_time": 2.438823699951172,
      "step": 7273
    },
    {
      "epoch": 2.7744293212890624e-05,
      "step": 7273,
      "training_step_time": 7.463900804519653
    },
    {
      "epoch": 2.7744293212890624e-05,
      "model_forward_time": 3.136774778366089,
      "step": 7273
    },
    {
      "epoch": 2.7744293212890624e-05,
      "step": 7273,
      "training_step_time": 10.342700481414795
    },
    {
      "epoch": 2.774810791015625e-05,
      "model_forward_time": 2.739516496658325,
      "step": 7274
    },
    {
      "epoch": 2.774810791015625e-05,
      "step": 7274,
      "training_step_time": 7.769778251647949
    },
    {
      "epoch": 2.774810791015625e-05,
      "model_forward_time": 2.432023525238037,
      "step": 7274
    },
    {
      "epoch": 2.774810791015625e-05,
      "step": 7274,
      "training_step_time": 7.452584266662598
    },
    {
      "epoch": 2.774810791015625e-05,
      "model_forward_time": 3.1357719898223877,
      "step": 7274
    },
    {
      "epoch": 2.774810791015625e-05,
      "step": 7274,
      "training_step_time": 9.90147852897644
    },
    {
      "epoch": 2.774810791015625e-05,
      "model_forward_time": 2.4306719303131104,
      "step": 7274
    },
    {
      "epoch": 2.774810791015625e-05,
      "step": 7274,
      "training_step_time": 10.293145418167114
    },
    {
      "epoch": 2.7751922607421876e-05,
      "model_forward_time": 3.1576852798461914,
      "step": 7275
    },
    {
      "dynamics_loss_avg": 0.04900735439732671,
      "epoch": 2.7751922607421876e-05,
      "step": 7275
    },
    {
      "action_loss_avg": 0.020147223584353925,
      "epoch": 2.7751922607421876e-05,
      "step": 7275
    },
    {
      "epoch": 2.7751922607421876e-05,
      "step": 7275,
      "training_step_time": 9.925132751464844
    },
    {
      "epoch": 2.7751922607421876e-05,
      "model_forward_time": 3.4013001918792725,
      "step": 7275
    },
    {
      "epoch": 2.7751922607421876e-05,
      "step": 7275,
      "training_step_time": 10.16554069519043
    },
    {
      "epoch": 2.7751922607421876e-05,
      "model_forward_time": 3.1417734622955322,
      "step": 7275
    },
    {
      "epoch": 2.7751922607421876e-05,
      "step": 7275,
      "training_step_time": 9.906901359558105
    },
    {
      "epoch": 2.7751922607421876e-05,
      "model_forward_time": 2.4377315044403076,
      "step": 7275
    },
    {
      "epoch": 2.7751922607421876e-05,
      "step": 7275,
      "training_step_time": 10.276921510696411
    },
    {
      "epoch": 2.77557373046875e-05,
      "model_forward_time": 2.448145627975464,
      "step": 7276
    },
    {
      "epoch": 2.77557373046875e-05,
      "step": 7276,
      "training_step_time": 7.467757701873779
    },
    {
      "epoch": 2.77557373046875e-05,
      "model_forward_time": 3.1353514194488525,
      "step": 7276
    },
    {
      "epoch": 2.77557373046875e-05,
      "step": 7276,
      "training_step_time": 9.89448857307434
    },
    {
      "epoch": 2.77557373046875e-05,
      "model_forward_time": 3.13156795501709,
      "step": 7276
    },
    {
      "epoch": 2.77557373046875e-05,
      "step": 7276,
      "training_step_time": 9.891249418258667
    },
    {
      "epoch": 2.77557373046875e-05,
      "model_forward_time": 3.3825857639312744,
      "step": 7276
    },
    {
      "epoch": 2.77557373046875e-05,
      "step": 7276,
      "training_step_time": 10.243111371994019
    },
    {
      "epoch": 2.7759552001953126e-05,
      "model_forward_time": 3.1783909797668457,
      "step": 7277
    },
    {
      "epoch": 2.7759552001953126e-05,
      "step": 7277,
      "training_step_time": 9.94335389137268
    },
    {
      "epoch": 2.7759552001953126e-05,
      "model_forward_time": 3.1390304565429688,
      "step": 7277
    },
    {
      "epoch": 2.7759552001953126e-05,
      "step": 7277,
      "training_step_time": 9.903384685516357
    },
    {
      "epoch": 2.7759552001953126e-05,
      "model_forward_time": 3.1376912593841553,
      "step": 7277
    },
    {
      "dynamics_loss_avg": 0.0523360550403595,
      "epoch": 2.7759552001953126e-05,
      "step": 7277
    },
    {
      "action_loss_avg": 0.03117743958719075,
      "epoch": 2.7759552001953126e-05,
      "step": 7277
    },
    {
      "epoch": 2.7759552001953126e-05,
      "step": 7277,
      "training_step_time": 9.909493446350098
    },
    {
      "epoch": 2.7759552001953126e-05,
      "model_forward_time": 3.1446549892425537,
      "step": 7277
    },
    {
      "epoch": 2.7759552001953126e-05,
      "step": 7277,
      "training_step_time": 10.283047199249268
    },
    {
      "epoch": 2.776336669921875e-05,
      "model_forward_time": 2.4569544792175293,
      "step": 7278
    },
    {
      "epoch": 2.776336669921875e-05,
      "step": 7278,
      "training_step_time": 7.483394384384155
    },
    {
      "epoch": 2.776336669921875e-05,
      "model_forward_time": 3.1406519412994385,
      "step": 7278
    },
    {
      "epoch": 2.776336669921875e-05,
      "step": 7278,
      "training_step_time": 9.908263444900513
    },
    {
      "epoch": 2.776336669921875e-05,
      "model_forward_time": 2.4392502307891846,
      "step": 7278
    },
    {
      "epoch": 2.776336669921875e-05,
      "step": 7278,
      "training_step_time": 7.462352752685547
    },
    {
      "epoch": 2.776336669921875e-05,
      "model_forward_time": 2.435142755508423,
      "step": 7278
    },
    {
      "epoch": 2.776336669921875e-05,
      "step": 7278,
      "training_step_time": 10.291649341583252
    },
    {
      "epoch": 2.7767181396484375e-05,
      "model_forward_time": 3.1805102825164795,
      "step": 7279
    },
    {
      "epoch": 2.7767181396484375e-05,
      "step": 7279,
      "training_step_time": 9.948559761047363
    },
    {
      "epoch": 2.7767181396484375e-05,
      "model_forward_time": 3.445279836654663,
      "step": 7279
    },
    {
      "epoch": 2.7767181396484375e-05,
      "step": 7279,
      "training_step_time": 10.21274471282959
    },
    {
      "epoch": 2.7767181396484375e-05,
      "model_forward_time": 3.141021251678467,
      "step": 7279
    },
    {
      "epoch": 2.7767181396484375e-05,
      "step": 7279,
      "training_step_time": 9.909317255020142
    },
    {
      "epoch": 2.7767181396484375e-05,
      "model_forward_time": 3.1454362869262695,
      "step": 7279
    },
    {
      "epoch": 2.7767181396484375e-05,
      "step": 7279,
      "training_step_time": 10.284966468811035
    },
    {
      "epoch": 2.777099609375e-05,
      "grad_norm": 0.4531249701976776,
      "learning_rate": 1e-05,
      "loss": 0.0807,
      "step": 7280
    },
    {
      "epoch": 2.777099609375e-05,
      "model_forward_time": 2.4530746936798096,
      "step": 7280
    },
    {
      "dynamics_loss_avg": 0.05562290940433741,
      "epoch": 2.777099609375e-05,
      "step": 7280
    },
    {
      "action_loss_avg": 0.026826126500964165,
      "epoch": 2.777099609375e-05,
      "step": 7280
    },
    {
      "epoch": 2.777099609375e-05,
      "step": 7280,
      "training_step_time": 7.482163906097412
    },
    {
      "epoch": 2.777099609375e-05,
      "model_forward_time": 2.43135929107666,
      "step": 7280
    },
    {
      "epoch": 2.777099609375e-05,
      "step": 7280,
      "training_step_time": 7.453667640686035
    },
    {
      "epoch": 2.777099609375e-05,
      "model_forward_time": 2.707949638366699,
      "step": 7280
    },
    {
      "epoch": 2.777099609375e-05,
      "step": 7280,
      "training_step_time": 7.726778507232666
    },
    {
      "epoch": 2.777099609375e-05,
      "model_forward_time": 3.134498357772827,
      "step": 7280
    },
    {
      "epoch": 2.777099609375e-05,
      "step": 7280,
      "training_step_time": 10.305320262908936
    },
    {
      "epoch": 2.7774810791015624e-05,
      "model_forward_time": 3.158482313156128,
      "step": 7281
    },
    {
      "epoch": 2.7774810791015624e-05,
      "step": 7281,
      "training_step_time": 9.92446517944336
    },
    {
      "epoch": 2.7774810791015624e-05,
      "model_forward_time": 3.133077621459961,
      "step": 7281
    },
    {
      "epoch": 2.7774810791015624e-05,
      "step": 7281,
      "training_step_time": 9.899268388748169
    },
    {
      "epoch": 2.7774810791015624e-05,
      "model_forward_time": 2.4326961040496826,
      "step": 7281
    },
    {
      "epoch": 2.7774810791015624e-05,
      "step": 7281,
      "training_step_time": 7.451991558074951
    },
    {
      "epoch": 2.7774810791015624e-05,
      "model_forward_time": 3.4163668155670166,
      "step": 7281
    },
    {
      "epoch": 2.7774810791015624e-05,
      "step": 7281,
      "training_step_time": 10.267446756362915
    },
    {
      "epoch": 2.777862548828125e-05,
      "model_forward_time": 2.450213670730591,
      "step": 7282
    },
    {
      "epoch": 2.777862548828125e-05,
      "step": 7282,
      "training_step_time": 7.469971418380737
    },
    {
      "epoch": 2.777862548828125e-05,
      "model_forward_time": 3.134063243865967,
      "step": 7282
    },
    {
      "epoch": 2.777862548828125e-05,
      "step": 7282,
      "training_step_time": 9.89781141281128
    },
    {
      "epoch": 2.777862548828125e-05,
      "model_forward_time": 2.4371752738952637,
      "step": 7282
    },
    {
      "dynamics_loss_avg": 0.07219867650419473,
      "epoch": 2.777862548828125e-05,
      "step": 7282
    },
    {
      "action_loss_avg": 0.020933052292093635,
      "epoch": 2.777862548828125e-05,
      "step": 7282
    },
    {
      "epoch": 2.777862548828125e-05,
      "step": 7282,
      "training_step_time": 7.461499452590942
    },
    {
      "epoch": 2.777862548828125e-05,
      "model_forward_time": 2.432436227798462,
      "step": 7282
    },
    {
      "epoch": 2.777862548828125e-05,
      "step": 7282,
      "training_step_time": 10.01557183265686
    },
    {
      "epoch": 2.7782440185546876e-05,
      "model_forward_time": 2.4464008808135986,
      "step": 7283
    },
    {
      "epoch": 2.7782440185546876e-05,
      "step": 7283,
      "training_step_time": 7.465392351150513
    },
    {
      "epoch": 2.7782440185546876e-05,
      "model_forward_time": 3.385136842727661,
      "step": 7283
    },
    {
      "epoch": 2.7782440185546876e-05,
      "step": 7283,
      "training_step_time": 10.146360874176025
    },
    {
      "epoch": 2.7782440185546876e-05,
      "model_forward_time": 2.437070369720459,
      "step": 7283
    },
    {
      "epoch": 2.7782440185546876e-05,
      "step": 7283,
      "training_step_time": 7.463908910751343
    },
    {
      "epoch": 2.7782440185546876e-05,
      "model_forward_time": 3.137237310409546,
      "step": 7283
    },
    {
      "epoch": 2.7782440185546876e-05,
      "step": 7283,
      "training_step_time": 10.281718254089355
    },
    {
      "epoch": 2.77862548828125e-05,
      "model_forward_time": 2.451444387435913,
      "step": 7284
    },
    {
      "epoch": 2.77862548828125e-05,
      "step": 7284,
      "training_step_time": 7.481194257736206
    },
    {
      "epoch": 2.77862548828125e-05,
      "model_forward_time": 3.1357483863830566,
      "step": 7284
    },
    {
      "epoch": 2.77862548828125e-05,
      "step": 7284,
      "training_step_time": 9.900911331176758
    },
    {
      "epoch": 2.77862548828125e-05,
      "model_forward_time": 3.4040684700012207,
      "step": 7284
    },
    {
      "epoch": 2.77862548828125e-05,
      "step": 7284,
      "training_step_time": 10.172637462615967
    },
    {
      "epoch": 2.77862548828125e-05,
      "model_forward_time": 2.434321641921997,
      "step": 7284
    },
    {
      "epoch": 2.77862548828125e-05,
      "step": 7284,
      "training_step_time": 10.438733339309692
    },
    {
      "epoch": 2.7790069580078126e-05,
      "model_forward_time": 3.1421244144439697,
      "step": 7285
    },
    {
      "dynamics_loss_avg": 0.05845136009156704,
      "epoch": 2.7790069580078126e-05,
      "step": 7285
    },
    {
      "action_loss_avg": 0.020890390500426292,
      "epoch": 2.7790069580078126e-05,
      "step": 7285
    },
    {
      "epoch": 2.7790069580078126e-05,
      "step": 7285,
      "training_step_time": 9.910751104354858
    },
    {
      "epoch": 2.7790069580078126e-05,
      "model_forward_time": 3.135227680206299,
      "step": 7285
    },
    {
      "epoch": 2.7790069580078126e-05,
      "step": 7285,
      "training_step_time": 9.900272846221924
    },
    {
      "epoch": 2.7790069580078126e-05,
      "model_forward_time": 2.42621111869812,
      "step": 7285
    },
    {
      "epoch": 2.7790069580078126e-05,
      "step": 7285,
      "training_step_time": 7.445624828338623
    },
    {
      "epoch": 2.7790069580078126e-05,
      "model_forward_time": 3.3766555786132812,
      "step": 7285
    },
    {
      "epoch": 2.7790069580078126e-05,
      "step": 7285,
      "training_step_time": 10.281449794769287
    },
    {
      "epoch": 2.779388427734375e-05,
      "model_forward_time": 2.458367109298706,
      "step": 7286
    },
    {
      "epoch": 2.779388427734375e-05,
      "step": 7286,
      "training_step_time": 7.4820029735565186
    },
    {
      "epoch": 2.779388427734375e-05,
      "model_forward_time": 3.14178466796875,
      "step": 7286
    },
    {
      "epoch": 2.779388427734375e-05,
      "step": 7286,
      "training_step_time": 9.905433654785156
    },
    {
      "epoch": 2.779388427734375e-05,
      "model_forward_time": 2.439418315887451,
      "step": 7286
    },
    {
      "epoch": 2.779388427734375e-05,
      "step": 7286,
      "training_step_time": 7.465869903564453
    },
    {
      "epoch": 2.779388427734375e-05,
      "model_forward_time": 2.427773952484131,
      "step": 7286
    },
    {
      "epoch": 2.779388427734375e-05,
      "step": 7286,
      "training_step_time": 10.010396242141724
    },
    {
      "epoch": 2.7797698974609375e-05,
      "model_forward_time": 2.0735960006713867,
      "step": 7287
    },
    {
      "epoch": 2.7797698974609375e-05,
      "step": 7287,
      "training_step_time": 5.537662506103516
    },
    {
      "epoch": 2.7797698974609375e-05,
      "model_forward_time": 3.1331663131713867,
      "step": 7287
    },
    {
      "epoch": 2.7797698974609375e-05,
      "step": 7287,
      "training_step_time": 9.899055004119873
    },
    {
      "epoch": 2.7797698974609375e-05,
      "model_forward_time": 2.4353110790252686,
      "step": 7287
    },
    {
      "dynamics_loss_avg": 0.052568502817302944,
      "epoch": 2.7797698974609375e-05,
      "step": 7287
    },
    {
      "action_loss_avg": 0.02960832016542554,
      "epoch": 2.7797698974609375e-05,
      "step": 7287
    },
    {
      "epoch": 2.7797698974609375e-05,
      "step": 7287,
      "training_step_time": 7.463521718978882
    },
    {
      "epoch": 2.7797698974609375e-05,
      "model_forward_time": 2.4299254417419434,
      "step": 7287
    },
    {
      "epoch": 2.7797698974609375e-05,
      "step": 7287,
      "training_step_time": 10.276972532272339
    },
    {
      "epoch": 2.7801513671875e-05,
      "model_forward_time": 2.4728198051452637,
      "step": 7288
    },
    {
      "epoch": 2.7801513671875e-05,
      "step": 7288,
      "training_step_time": 7.500676870346069
    },
    {
      "epoch": 2.7801513671875e-05,
      "model_forward_time": 2.4294183254241943,
      "step": 7288
    },
    {
      "epoch": 2.7801513671875e-05,
      "step": 7288,
      "training_step_time": 7.459786891937256
    },
    {
      "epoch": 2.7801513671875e-05,
      "model_forward_time": 3.3849167823791504,
      "step": 7288
    },
    {
      "epoch": 2.7801513671875e-05,
      "step": 7288,
      "training_step_time": 10.15682077407837
    },
    {
      "epoch": 2.7801513671875e-05,
      "model_forward_time": 3.1406519412994385,
      "step": 7288
    },
    {
      "epoch": 2.7801513671875e-05,
      "step": 7288,
      "training_step_time": 10.288702964782715
    },
    {
      "epoch": 2.7805328369140624e-05,
      "model_forward_time": 3.1584274768829346,
      "step": 7289
    },
    {
      "epoch": 2.7805328369140624e-05,
      "step": 7289,
      "training_step_time": 9.932516813278198
    },
    {
      "epoch": 2.7805328369140624e-05,
      "model_forward_time": 2.4347622394561768,
      "step": 7289
    },
    {
      "epoch": 2.7805328369140624e-05,
      "step": 7289,
      "training_step_time": 7.46670937538147
    },
    {
      "epoch": 2.7805328369140624e-05,
      "model_forward_time": 3.1393322944641113,
      "step": 7289
    },
    {
      "epoch": 2.7805328369140624e-05,
      "step": 7289,
      "training_step_time": 9.91904902458191
    },
    {
      "epoch": 2.7805328369140624e-05,
      "model_forward_time": 2.6815223693847656,
      "step": 7289
    },
    {
      "epoch": 2.7805328369140624e-05,
      "step": 7289,
      "training_step_time": 10.263386726379395
    },
    {
      "epoch": 2.780914306640625e-05,
      "grad_norm": 0.35811254382133484,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 7290
    },
    {
      "epoch": 2.780914306640625e-05,
      "model_forward_time": 2.4452013969421387,
      "step": 7290
    },
    {
      "dynamics_loss_avg": 0.05741619784384966,
      "epoch": 2.780914306640625e-05,
      "step": 7290
    },
    {
      "action_loss_avg": 0.028408439923077823,
      "epoch": 2.780914306640625e-05,
      "step": 7290
    },
    {
      "epoch": 2.780914306640625e-05,
      "step": 7290,
      "training_step_time": 7.476500034332275
    },
    {
      "epoch": 2.780914306640625e-05,
      "model_forward_time": 3.1330406665802,
      "step": 7290
    },
    {
      "epoch": 2.780914306640625e-05,
      "step": 7290,
      "training_step_time": 9.907936096191406
    },
    {
      "epoch": 2.780914306640625e-05,
      "model_forward_time": 2.4345061779022217,
      "step": 7290
    },
    {
      "epoch": 2.780914306640625e-05,
      "step": 7290,
      "training_step_time": 7.466835260391235
    },
    {
      "epoch": 2.780914306640625e-05,
      "model_forward_time": 3.132359266281128,
      "step": 7290
    },
    {
      "epoch": 2.780914306640625e-05,
      "step": 7290,
      "training_step_time": 10.020655393600464
    },
    {
      "epoch": 2.7812957763671876e-05,
      "model_forward_time": 2.763414144515991,
      "step": 7291
    },
    {
      "epoch": 2.7812957763671876e-05,
      "step": 7291,
      "training_step_time": 7.795896768569946
    },
    {
      "epoch": 2.7812957763671876e-05,
      "model_forward_time": 3.131166934967041,
      "step": 7291
    },
    {
      "epoch": 2.7812957763671876e-05,
      "step": 7291,
      "training_step_time": 9.906372547149658
    },
    {
      "epoch": 2.7812957763671876e-05,
      "model_forward_time": 1.7411909103393555,
      "step": 7291
    },
    {
      "epoch": 2.7812957763671876e-05,
      "step": 7291,
      "training_step_time": 5.20147967338562
    },
    {
      "epoch": 2.7812957763671876e-05,
      "model_forward_time": 3.1242856979370117,
      "step": 7291
    },
    {
      "epoch": 2.7812957763671876e-05,
      "step": 7291,
      "training_step_time": 10.305747747421265
    },
    {
      "epoch": 2.78167724609375e-05,
      "model_forward_time": 3.1669211387634277,
      "step": 7292
    },
    {
      "epoch": 2.78167724609375e-05,
      "step": 7292,
      "training_step_time": 9.939080715179443
    },
    {
      "epoch": 2.78167724609375e-05,
      "model_forward_time": 1.7552258968353271,
      "step": 7292
    },
    {
      "epoch": 2.78167724609375e-05,
      "step": 7292,
      "training_step_time": 5.218214988708496
    },
    {
      "epoch": 2.78167724609375e-05,
      "model_forward_time": 3.4004645347595215,
      "step": 7292
    },
    {
      "dynamics_loss_avg": 0.058028624020516874,
      "epoch": 2.78167724609375e-05,
      "step": 7292
    },
    {
      "action_loss_avg": 0.028595313942059875,
      "epoch": 2.78167724609375e-05,
      "step": 7292
    },
    {
      "epoch": 2.78167724609375e-05,
      "step": 7292,
      "training_step_time": 10.180760860443115
    },
    {
      "epoch": 2.78167724609375e-05,
      "model_forward_time": 3.12966251373291,
      "step": 7292
    },
    {
      "epoch": 2.78167724609375e-05,
      "step": 7292,
      "training_step_time": 10.264612674713135
    },
    {
      "epoch": 2.7820587158203126e-05,
      "model_forward_time": 3.150158643722534,
      "step": 7293
    },
    {
      "epoch": 2.7820587158203126e-05,
      "step": 7293,
      "training_step_time": 9.92860198020935
    },
    {
      "epoch": 2.7820587158203126e-05,
      "model_forward_time": 2.4313852787017822,
      "step": 7293
    },
    {
      "epoch": 2.7820587158203126e-05,
      "step": 7293,
      "training_step_time": 7.467597961425781
    },
    {
      "epoch": 2.7820587158203126e-05,
      "model_forward_time": 2.4285433292388916,
      "step": 7293
    },
    {
      "epoch": 2.7820587158203126e-05,
      "step": 7293,
      "training_step_time": 7.460735559463501
    },
    {
      "epoch": 2.7820587158203126e-05,
      "model_forward_time": 3.37565016746521,
      "step": 7293
    },
    {
      "epoch": 2.7820587158203126e-05,
      "step": 7293,
      "training_step_time": 10.313130378723145
    },
    {
      "epoch": 2.782440185546875e-05,
      "model_forward_time": 3.1470906734466553,
      "step": 7294
    },
    {
      "epoch": 2.782440185546875e-05,
      "step": 7294,
      "training_step_time": 9.927308082580566
    },
    {
      "epoch": 2.782440185546875e-05,
      "model_forward_time": 2.4305453300476074,
      "step": 7294
    },
    {
      "epoch": 2.782440185546875e-05,
      "step": 7294,
      "training_step_time": 7.460680723190308
    },
    {
      "epoch": 2.782440185546875e-05,
      "model_forward_time": 3.1361098289489746,
      "step": 7294
    },
    {
      "epoch": 2.782440185546875e-05,
      "step": 7294,
      "training_step_time": 9.910943984985352
    },
    {
      "epoch": 2.782440185546875e-05,
      "model_forward_time": 2.4341280460357666,
      "step": 7294
    },
    {
      "epoch": 2.782440185546875e-05,
      "step": 7294,
      "training_step_time": 10.311982154846191
    },
    {
      "epoch": 2.7828216552734375e-05,
      "model_forward_time": 2.7643749713897705,
      "step": 7295
    },
    {
      "dynamics_loss_avg": 0.059143328480422495,
      "epoch": 2.7828216552734375e-05,
      "step": 7295
    },
    {
      "action_loss_avg": 0.02243647165596485,
      "epoch": 2.7828216552734375e-05,
      "step": 7295
    },
    {
      "epoch": 2.7828216552734375e-05,
      "step": 7295,
      "training_step_time": 7.797304391860962
    },
    {
      "epoch": 2.7828216552734375e-05,
      "model_forward_time": 3.1357030868530273,
      "step": 7295
    },
    {
      "epoch": 2.7828216552734375e-05,
      "step": 7295,
      "training_step_time": 9.904581785202026
    },
    {
      "epoch": 2.7828216552734375e-05,
      "model_forward_time": 3.1367344856262207,
      "step": 7295
    },
    {
      "epoch": 2.7828216552734375e-05,
      "step": 7295,
      "training_step_time": 9.905099868774414
    },
    {
      "epoch": 2.7828216552734375e-05,
      "model_forward_time": 3.1386115550994873,
      "step": 7295
    },
    {
      "epoch": 2.7828216552734375e-05,
      "step": 7295,
      "training_step_time": 10.309433937072754
    },
    {
      "epoch": 2.783203125e-05,
      "model_forward_time": 3.152912139892578,
      "step": 7296
    },
    {
      "epoch": 2.783203125e-05,
      "step": 7296,
      "training_step_time": 9.92194676399231
    },
    {
      "epoch": 2.783203125e-05,
      "model_forward_time": 2.6736812591552734,
      "step": 7296
    },
    {
      "epoch": 2.783203125e-05,
      "step": 7296,
      "training_step_time": 7.702335834503174
    },
    {
      "epoch": 2.783203125e-05,
      "model_forward_time": 3.136258840560913,
      "step": 7296
    },
    {
      "epoch": 2.783203125e-05,
      "step": 7296,
      "training_step_time": 9.90390944480896
    },
    {
      "epoch": 2.783203125e-05,
      "model_forward_time": 2.4302988052368164,
      "step": 7296
    },
    {
      "epoch": 2.783203125e-05,
      "step": 7296,
      "training_step_time": 10.264881372451782
    },
    {
      "epoch": 2.7835845947265624e-05,
      "model_forward_time": 1.7612721920013428,
      "step": 7297
    },
    {
      "epoch": 2.7835845947265624e-05,
      "step": 7297,
      "training_step_time": 5.220077991485596
    },
    {
      "epoch": 2.7835845947265624e-05,
      "model_forward_time": 2.431570291519165,
      "step": 7297
    },
    {
      "epoch": 2.7835845947265624e-05,
      "step": 7297,
      "training_step_time": 7.45516562461853
    },
    {
      "epoch": 2.7835845947265624e-05,
      "model_forward_time": 2.7027792930603027,
      "step": 7297
    },
    {
      "dynamics_loss_avg": 0.05603581033647061,
      "epoch": 2.7835845947265624e-05,
      "step": 7297
    },
    {
      "action_loss_avg": 0.02301909402012825,
      "epoch": 2.7835845947265624e-05,
      "step": 7297
    },
    {
      "epoch": 2.7835845947265624e-05,
      "step": 7297,
      "training_step_time": 7.726344347000122
    },
    {
      "epoch": 2.7835845947265624e-05,
      "model_forward_time": 2.429227828979492,
      "step": 7297
    },
    {
      "epoch": 2.7835845947265624e-05,
      "step": 7297,
      "training_step_time": 10.266433954238892
    },
    {
      "epoch": 2.783966064453125e-05,
      "model_forward_time": 3.1462526321411133,
      "step": 7298
    },
    {
      "epoch": 2.783966064453125e-05,
      "step": 7298,
      "training_step_time": 9.913517236709595
    },
    {
      "epoch": 2.783966064453125e-05,
      "model_forward_time": 3.132068395614624,
      "step": 7298
    },
    {
      "epoch": 2.783966064453125e-05,
      "step": 7298,
      "training_step_time": 9.8941650390625
    },
    {
      "epoch": 2.783966064453125e-05,
      "model_forward_time": 3.1329972743988037,
      "step": 7298
    },
    {
      "epoch": 2.783966064453125e-05,
      "step": 7298,
      "training_step_time": 9.895544052124023
    },
    {
      "epoch": 2.783966064453125e-05,
      "model_forward_time": 3.1294870376586914,
      "step": 7298
    },
    {
      "epoch": 2.783966064453125e-05,
      "step": 7298,
      "training_step_time": 10.288567304611206
    },
    {
      "epoch": 2.7843475341796876e-05,
      "model_forward_time": 2.4696502685546875,
      "step": 7299
    },
    {
      "epoch": 2.7843475341796876e-05,
      "step": 7299,
      "training_step_time": 7.488616704940796
    },
    {
      "epoch": 2.7843475341796876e-05,
      "model_forward_time": 3.128227710723877,
      "step": 7299
    },
    {
      "epoch": 2.7843475341796876e-05,
      "step": 7299,
      "training_step_time": 9.891952753067017
    },
    {
      "epoch": 2.7843475341796876e-05,
      "model_forward_time": 3.132939577102661,
      "step": 7299
    },
    {
      "epoch": 2.7843475341796876e-05,
      "step": 7299,
      "training_step_time": 9.899238348007202
    },
    {
      "epoch": 2.7843475341796876e-05,
      "model_forward_time": 3.1342692375183105,
      "step": 7299
    },
    {
      "epoch": 2.7843475341796876e-05,
      "step": 7299,
      "training_step_time": 10.286827087402344
    },
    {
      "epoch": 2.78472900390625e-05,
      "grad_norm": 0.45048001408576965,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 7300
    },
    {
      "epoch": 2.78472900390625e-05,
      "model_forward_time": 3.4272899627685547,
      "step": 7300
    },
    {
      "dynamics_loss_avg": 0.06411577351391315,
      "epoch": 2.78472900390625e-05,
      "step": 7300
    },
    {
      "action_loss_avg": 0.02594365384429693,
      "epoch": 2.78472900390625e-05,
      "step": 7300
    },
    {
      "epoch": 2.78472900390625e-05,
      "step": 7300,
      "training_step_time": 10.1984384059906
    },
    {
      "epoch": 2.78472900390625e-05,
      "model_forward_time": 2.435271739959717,
      "step": 7300
    },
    {
      "epoch": 2.78472900390625e-05,
      "step": 7300,
      "training_step_time": 7.462574481964111
    },
    {
      "epoch": 2.78472900390625e-05,
      "model_forward_time": 3.1341331005096436,
      "step": 7300
    },
    {
      "epoch": 2.78472900390625e-05,
      "step": 7300,
      "training_step_time": 9.906321048736572
    },
    {
      "epoch": 2.78472900390625e-05,
      "model_forward_time": 2.4413020610809326,
      "step": 7300
    },
    {
      "epoch": 2.78472900390625e-05,
      "step": 7300,
      "training_step_time": 10.259277105331421
    },
    {
      "epoch": 2.7851104736328126e-05,
      "model_forward_time": 2.4760165214538574,
      "step": 7301
    },
    {
      "epoch": 2.7851104736328126e-05,
      "step": 7301,
      "training_step_time": 7.505603075027466
    },
    {
      "epoch": 2.7851104736328126e-05,
      "model_forward_time": 2.435468912124634,
      "step": 7301
    },
    {
      "epoch": 2.7851104736328126e-05,
      "step": 7301,
      "training_step_time": 7.457244396209717
    },
    {
      "epoch": 2.7851104736328126e-05,
      "model_forward_time": 3.427659749984741,
      "step": 7301
    },
    {
      "epoch": 2.7851104736328126e-05,
      "step": 7301,
      "training_step_time": 10.193893194198608
    },
    {
      "epoch": 2.7851104736328126e-05,
      "model_forward_time": 3.136213779449463,
      "step": 7301
    },
    {
      "epoch": 2.7851104736328126e-05,
      "step": 7301,
      "training_step_time": 10.270779371261597
    },
    {
      "epoch": 2.785491943359375e-05,
      "model_forward_time": 2.488769054412842,
      "step": 7302
    },
    {
      "epoch": 2.785491943359375e-05,
      "step": 7302,
      "training_step_time": 7.513931035995483
    },
    {
      "epoch": 2.785491943359375e-05,
      "model_forward_time": 1.7429156303405762,
      "step": 7302
    },
    {
      "epoch": 2.785491943359375e-05,
      "step": 7302,
      "training_step_time": 5.193405628204346
    },
    {
      "epoch": 2.785491943359375e-05,
      "model_forward_time": 3.130157470703125,
      "step": 7302
    },
    {
      "dynamics_loss_avg": 0.05474928729236126,
      "epoch": 2.785491943359375e-05,
      "step": 7302
    },
    {
      "action_loss_avg": 0.02356641269288957,
      "epoch": 2.785491943359375e-05,
      "step": 7302
    },
    {
      "epoch": 2.785491943359375e-05,
      "step": 7302,
      "training_step_time": 9.88922381401062
    },
    {
      "epoch": 2.785491943359375e-05,
      "model_forward_time": 3.1364240646362305,
      "step": 7302
    },
    {
      "epoch": 2.785491943359375e-05,
      "step": 7302,
      "training_step_time": 10.273473978042603
    },
    {
      "epoch": 2.7858734130859375e-05,
      "model_forward_time": 3.1446125507354736,
      "step": 7303
    },
    {
      "epoch": 2.7858734130859375e-05,
      "step": 7303,
      "training_step_time": 9.917746305465698
    },
    {
      "epoch": 2.7858734130859375e-05,
      "model_forward_time": 3.1377146244049072,
      "step": 7303
    },
    {
      "epoch": 2.7858734130859375e-05,
      "step": 7303,
      "training_step_time": 9.90163278579712
    },
    {
      "epoch": 2.7858734130859375e-05,
      "model_forward_time": 3.136725664138794,
      "step": 7303
    },
    {
      "epoch": 2.7858734130859375e-05,
      "step": 7303,
      "training_step_time": 9.898165941238403
    },
    {
      "epoch": 2.7858734130859375e-05,
      "model_forward_time": 2.431438446044922,
      "step": 7303
    },
    {
      "epoch": 2.7858734130859375e-05,
      "step": 7303,
      "training_step_time": 10.261305093765259
    },
    {
      "epoch": 2.7862548828125e-05,
      "model_forward_time": 3.1535937786102295,
      "step": 7304
    },
    {
      "epoch": 2.7862548828125e-05,
      "step": 7304,
      "training_step_time": 9.913795232772827
    },
    {
      "epoch": 2.7862548828125e-05,
      "model_forward_time": 3.4094440937042236,
      "step": 7304
    },
    {
      "epoch": 2.7862548828125e-05,
      "step": 7304,
      "training_step_time": 10.167073965072632
    },
    {
      "epoch": 2.7862548828125e-05,
      "model_forward_time": 2.435171127319336,
      "step": 7304
    },
    {
      "epoch": 2.7862548828125e-05,
      "step": 7304,
      "training_step_time": 7.450900554656982
    },
    {
      "epoch": 2.7862548828125e-05,
      "model_forward_time": 1.7401463985443115,
      "step": 7304
    },
    {
      "epoch": 2.7862548828125e-05,
      "step": 7304,
      "training_step_time": 10.294277906417847
    },
    {
      "epoch": 2.7866363525390624e-05,
      "model_forward_time": 2.4457411766052246,
      "step": 7305
    },
    {
      "dynamics_loss_avg": 0.054392187297344206,
      "epoch": 2.7866363525390624e-05,
      "step": 7305
    },
    {
      "action_loss_avg": 0.027364743407815695,
      "epoch": 2.7866363525390624e-05,
      "step": 7305
    },
    {
      "epoch": 2.7866363525390624e-05,
      "step": 7305,
      "training_step_time": 7.465675115585327
    },
    {
      "epoch": 2.7866363525390624e-05,
      "model_forward_time": 3.1358842849731445,
      "step": 7305
    },
    {
      "epoch": 2.7866363525390624e-05,
      "step": 7305,
      "training_step_time": 9.8934645652771
    },
    {
      "epoch": 2.7866363525390624e-05,
      "model_forward_time": 3.3926289081573486,
      "step": 7305
    },
    {
      "epoch": 2.7866363525390624e-05,
      "step": 7305,
      "training_step_time": 10.149689435958862
    },
    {
      "epoch": 2.7866363525390624e-05,
      "model_forward_time": 3.142102003097534,
      "step": 7305
    },
    {
      "epoch": 2.7866363525390624e-05,
      "step": 7305,
      "training_step_time": 10.344470262527466
    },
    {
      "epoch": 2.787017822265625e-05,
      "model_forward_time": 3.1552209854125977,
      "step": 7306
    },
    {
      "epoch": 2.787017822265625e-05,
      "step": 7306,
      "training_step_time": 9.922754764556885
    },
    {
      "epoch": 2.787017822265625e-05,
      "model_forward_time": 2.4359049797058105,
      "step": 7306
    },
    {
      "epoch": 2.787017822265625e-05,
      "step": 7306,
      "training_step_time": 7.460540771484375
    },
    {
      "epoch": 2.787017822265625e-05,
      "model_forward_time": 2.431297540664673,
      "step": 7306
    },
    {
      "epoch": 2.787017822265625e-05,
      "step": 7306,
      "training_step_time": 7.45145320892334
    },
    {
      "epoch": 2.787017822265625e-05,
      "model_forward_time": 3.416592597961426,
      "step": 7306
    },
    {
      "epoch": 2.787017822265625e-05,
      "step": 7306,
      "training_step_time": 10.274089574813843
    },
    {
      "epoch": 2.7873992919921876e-05,
      "model_forward_time": 3.1548898220062256,
      "step": 7307
    },
    {
      "epoch": 2.7873992919921876e-05,
      "step": 7307,
      "training_step_time": 9.920334339141846
    },
    {
      "epoch": 2.7873992919921876e-05,
      "model_forward_time": 3.135082960128784,
      "step": 7307
    },
    {
      "epoch": 2.7873992919921876e-05,
      "step": 7307,
      "training_step_time": 9.904290914535522
    },
    {
      "epoch": 2.7873992919921876e-05,
      "model_forward_time": 3.1392111778259277,
      "step": 7307
    },
    {
      "dynamics_loss_avg": 0.056932405568659306,
      "epoch": 2.7873992919921876e-05,
      "step": 7307
    },
    {
      "action_loss_avg": 0.017501685163006188,
      "epoch": 2.7873992919921876e-05,
      "step": 7307
    },
    {
      "epoch": 2.7873992919921876e-05,
      "step": 7307,
      "training_step_time": 9.915189027786255
    },
    {
      "epoch": 2.7873992919921876e-05,
      "model_forward_time": 2.439833641052246,
      "step": 7307
    },
    {
      "epoch": 2.7873992919921876e-05,
      "step": 7307,
      "training_step_time": 10.283564567565918
    },
    {
      "epoch": 2.78778076171875e-05,
      "model_forward_time": 3.439383029937744,
      "step": 7308
    },
    {
      "epoch": 2.78778076171875e-05,
      "step": 7308,
      "training_step_time": 10.208470344543457
    },
    {
      "epoch": 2.78778076171875e-05,
      "model_forward_time": 3.1346020698547363,
      "step": 7308
    },
    {
      "epoch": 2.78778076171875e-05,
      "step": 7308,
      "training_step_time": 9.906978130340576
    },
    {
      "epoch": 2.78778076171875e-05,
      "model_forward_time": 3.143455982208252,
      "step": 7308
    },
    {
      "epoch": 2.78778076171875e-05,
      "step": 7308,
      "training_step_time": 9.917379379272461
    },
    {
      "epoch": 2.78778076171875e-05,
      "model_forward_time": 3.1367263793945312,
      "step": 7308
    },
    {
      "epoch": 2.78778076171875e-05,
      "step": 7308,
      "training_step_time": 10.258961915969849
    },
    {
      "epoch": 2.7881622314453126e-05,
      "model_forward_time": 2.486191511154175,
      "step": 7309
    },
    {
      "epoch": 2.7881622314453126e-05,
      "step": 7309,
      "training_step_time": 7.521175146102905
    },
    {
      "epoch": 2.7881622314453126e-05,
      "model_forward_time": 3.4639434814453125,
      "step": 7309
    },
    {
      "epoch": 2.7881622314453126e-05,
      "step": 7309,
      "training_step_time": 10.237980842590332
    },
    {
      "epoch": 2.7881622314453126e-05,
      "model_forward_time": 1.7507226467132568,
      "step": 7309
    },
    {
      "epoch": 2.7881622314453126e-05,
      "step": 7309,
      "training_step_time": 5.209377765655518
    },
    {
      "epoch": 2.7881622314453126e-05,
      "model_forward_time": 1.7467982769012451,
      "step": 7309
    },
    {
      "epoch": 2.7881622314453126e-05,
      "step": 7309,
      "training_step_time": 10.03390383720398
    },
    {
      "epoch": 2.788543701171875e-05,
      "grad_norm": 0.5625168085098267,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 7310
    },
    {
      "epoch": 2.788543701171875e-05,
      "model_forward_time": 2.4497103691101074,
      "step": 7310
    },
    {
      "dynamics_loss_avg": 0.04540460258722305,
      "epoch": 2.788543701171875e-05,
      "step": 7310
    },
    {
      "action_loss_avg": 0.020332937687635423,
      "epoch": 2.788543701171875e-05,
      "step": 7310
    },
    {
      "epoch": 2.788543701171875e-05,
      "step": 7310,
      "training_step_time": 7.477119445800781
    },
    {
      "epoch": 2.788543701171875e-05,
      "model_forward_time": 2.433274984359741,
      "step": 7310
    },
    {
      "epoch": 2.788543701171875e-05,
      "step": 7310,
      "training_step_time": 7.456023931503296
    },
    {
      "epoch": 2.788543701171875e-05,
      "model_forward_time": 3.141500949859619,
      "step": 7310
    },
    {
      "epoch": 2.788543701171875e-05,
      "step": 7310,
      "training_step_time": 9.913291692733765
    },
    {
      "epoch": 2.788543701171875e-05,
      "model_forward_time": 2.71799635887146,
      "step": 7310
    },
    {
      "epoch": 2.788543701171875e-05,
      "step": 7310,
      "training_step_time": 10.035198450088501
    },
    {
      "epoch": 2.7889251708984375e-05,
      "model_forward_time": 3.165104627609253,
      "step": 7311
    },
    {
      "epoch": 2.7889251708984375e-05,
      "step": 7311,
      "training_step_time": 9.934242010116577
    },
    {
      "epoch": 2.7889251708984375e-05,
      "model_forward_time": 3.1425158977508545,
      "step": 7311
    },
    {
      "epoch": 2.7889251708984375e-05,
      "step": 7311,
      "training_step_time": 9.909318685531616
    },
    {
      "epoch": 2.7889251708984375e-05,
      "model_forward_time": 2.4323489665985107,
      "step": 7311
    },
    {
      "epoch": 2.7889251708984375e-05,
      "step": 7311,
      "training_step_time": 7.455671787261963
    },
    {
      "epoch": 2.7889251708984375e-05,
      "model_forward_time": 3.1410233974456787,
      "step": 7311
    },
    {
      "epoch": 2.7889251708984375e-05,
      "step": 7311,
      "training_step_time": 10.298726797103882
    },
    {
      "epoch": 2.789306640625e-05,
      "model_forward_time": 2.4515602588653564,
      "step": 7312
    },
    {
      "epoch": 2.789306640625e-05,
      "step": 7312,
      "training_step_time": 7.474106550216675
    },
    {
      "epoch": 2.789306640625e-05,
      "model_forward_time": 3.4583327770233154,
      "step": 7312
    },
    {
      "epoch": 2.789306640625e-05,
      "step": 7312,
      "training_step_time": 10.225171089172363
    },
    {
      "epoch": 2.789306640625e-05,
      "model_forward_time": 3.138213634490967,
      "step": 7312
    },
    {
      "dynamics_loss_avg": 0.04968552440404892,
      "epoch": 2.789306640625e-05,
      "step": 7312
    },
    {
      "action_loss_avg": 0.019860606687143446,
      "epoch": 2.789306640625e-05,
      "step": 7312
    },
    {
      "epoch": 2.789306640625e-05,
      "step": 7312,
      "training_step_time": 9.914367198944092
    },
    {
      "epoch": 2.789306640625e-05,
      "model_forward_time": 2.437924861907959,
      "step": 7312
    },
    {
      "epoch": 2.789306640625e-05,
      "step": 7312,
      "training_step_time": 10.295743465423584
    },
    {
      "epoch": 2.7896881103515624e-05,
      "model_forward_time": 2.451805591583252,
      "step": 7313
    },
    {
      "epoch": 2.7896881103515624e-05,
      "step": 7313,
      "training_step_time": 7.4765167236328125
    },
    {
      "epoch": 2.7896881103515624e-05,
      "model_forward_time": 2.436791181564331,
      "step": 7313
    },
    {
      "epoch": 2.7896881103515624e-05,
      "step": 7313,
      "training_step_time": 7.459131956100464
    },
    {
      "epoch": 2.7896881103515624e-05,
      "model_forward_time": 2.7103073596954346,
      "step": 7313
    },
    {
      "epoch": 2.7896881103515624e-05,
      "step": 7313,
      "training_step_time": 7.730407476425171
    },
    {
      "epoch": 2.7896881103515624e-05,
      "model_forward_time": 2.4289188385009766,
      "step": 7313
    },
    {
      "epoch": 2.7896881103515624e-05,
      "step": 7313,
      "training_step_time": 10.250893354415894
    },
    {
      "epoch": 2.790069580078125e-05,
      "model_forward_time": 2.4482600688934326,
      "step": 7314
    },
    {
      "epoch": 2.790069580078125e-05,
      "step": 7314,
      "training_step_time": 7.468024730682373
    },
    {
      "epoch": 2.790069580078125e-05,
      "model_forward_time": 3.1365106105804443,
      "step": 7314
    },
    {
      "epoch": 2.790069580078125e-05,
      "step": 7314,
      "training_step_time": 9.912898302078247
    },
    {
      "epoch": 2.790069580078125e-05,
      "model_forward_time": 2.437248945236206,
      "step": 7314
    },
    {
      "epoch": 2.790069580078125e-05,
      "step": 7314,
      "training_step_time": 7.460848808288574
    },
    {
      "epoch": 2.790069580078125e-05,
      "model_forward_time": 3.4264299869537354,
      "step": 7314
    },
    {
      "epoch": 2.790069580078125e-05,
      "step": 7314,
      "training_step_time": 10.288316249847412
    },
    {
      "epoch": 2.7904510498046876e-05,
      "model_forward_time": 2.4746780395507812,
      "step": 7315
    },
    {
      "dynamics_loss_avg": 0.04768055379390716,
      "epoch": 2.7904510498046876e-05,
      "step": 7315
    },
    {
      "action_loss_avg": 0.021420522592961788,
      "epoch": 2.7904510498046876e-05,
      "step": 7315
    },
    {
      "epoch": 2.7904510498046876e-05,
      "step": 7315,
      "training_step_time": 7.500744581222534
    },
    {
      "epoch": 2.7904510498046876e-05,
      "model_forward_time": 3.135859966278076,
      "step": 7315
    },
    {
      "epoch": 2.7904510498046876e-05,
      "step": 7315,
      "training_step_time": 9.905788898468018
    },
    {
      "epoch": 2.7904510498046876e-05,
      "model_forward_time": 3.1405980587005615,
      "step": 7315
    },
    {
      "epoch": 2.7904510498046876e-05,
      "step": 7315,
      "training_step_time": 9.91334319114685
    },
    {
      "epoch": 2.7904510498046876e-05,
      "model_forward_time": 3.140411376953125,
      "step": 7315
    },
    {
      "epoch": 2.7904510498046876e-05,
      "step": 7315,
      "training_step_time": 10.308294534683228
    },
    {
      "epoch": 2.79083251953125e-05,
      "model_forward_time": 3.478234052658081,
      "step": 7316
    },
    {
      "epoch": 2.79083251953125e-05,
      "step": 7316,
      "training_step_time": 10.25130844116211
    },
    {
      "epoch": 2.79083251953125e-05,
      "model_forward_time": 3.1400094032287598,
      "step": 7316
    },
    {
      "epoch": 2.79083251953125e-05,
      "step": 7316,
      "training_step_time": 9.910464525222778
    },
    {
      "epoch": 2.79083251953125e-05,
      "model_forward_time": 3.1439731121063232,
      "step": 7316
    },
    {
      "epoch": 2.79083251953125e-05,
      "step": 7316,
      "training_step_time": 9.922448873519897
    },
    {
      "epoch": 2.79083251953125e-05,
      "model_forward_time": 2.43994402885437,
      "step": 7316
    },
    {
      "epoch": 2.79083251953125e-05,
      "step": 7316,
      "training_step_time": 10.028290271759033
    },
    {
      "epoch": 2.7912139892578126e-05,
      "model_forward_time": 3.156766176223755,
      "step": 7317
    },
    {
      "epoch": 2.7912139892578126e-05,
      "step": 7317,
      "training_step_time": 9.934280395507812
    },
    {
      "epoch": 2.7912139892578126e-05,
      "model_forward_time": 3.1389803886413574,
      "step": 7317
    },
    {
      "epoch": 2.7912139892578126e-05,
      "step": 7317,
      "training_step_time": 9.916721105575562
    },
    {
      "epoch": 2.7912139892578126e-05,
      "model_forward_time": 2.029841184616089,
      "step": 7317
    },
    {
      "dynamics_loss_avg": 0.05956180151551962,
      "epoch": 2.7912139892578126e-05,
      "step": 7317
    },
    {
      "action_loss_avg": 0.0357334207277745,
      "epoch": 2.7912139892578126e-05,
      "step": 7317
    },
    {
      "epoch": 2.7912139892578126e-05,
      "step": 7317,
      "training_step_time": 5.488086938858032
    },
    {
      "epoch": 2.7912139892578126e-05,
      "model_forward_time": 1.741703987121582,
      "step": 7317
    },
    {
      "epoch": 2.7912139892578126e-05,
      "step": 7317,
      "training_step_time": 10.304739236831665
    },
    {
      "epoch": 2.791595458984375e-05,
      "model_forward_time": 2.447688579559326,
      "step": 7318
    },
    {
      "epoch": 2.791595458984375e-05,
      "step": 7318,
      "training_step_time": 7.476970672607422
    },
    {
      "epoch": 2.791595458984375e-05,
      "model_forward_time": 2.4335365295410156,
      "step": 7318
    },
    {
      "epoch": 2.791595458984375e-05,
      "step": 7318,
      "training_step_time": 7.45926308631897
    },
    {
      "epoch": 2.791595458984375e-05,
      "model_forward_time": 2.4308481216430664,
      "step": 7318
    },
    {
      "epoch": 2.791595458984375e-05,
      "step": 7318,
      "training_step_time": 7.459766626358032
    },
    {
      "epoch": 2.791595458984375e-05,
      "model_forward_time": 2.4305708408355713,
      "step": 7318
    },
    {
      "epoch": 2.791595458984375e-05,
      "step": 7318,
      "training_step_time": 10.284518003463745
    },
    {
      "epoch": 2.7919769287109375e-05,
      "model_forward_time": 3.4266889095306396,
      "step": 7319
    },
    {
      "epoch": 2.7919769287109375e-05,
      "step": 7319,
      "training_step_time": 10.209105730056763
    },
    {
      "epoch": 2.7919769287109375e-05,
      "model_forward_time": 2.4275500774383545,
      "step": 7319
    },
    {
      "epoch": 2.7919769287109375e-05,
      "step": 7319,
      "training_step_time": 7.456383466720581
    },
    {
      "epoch": 2.7919769287109375e-05,
      "model_forward_time": 2.428882360458374,
      "step": 7319
    },
    {
      "epoch": 2.7919769287109375e-05,
      "step": 7319,
      "training_step_time": 7.453972339630127
    },
    {
      "epoch": 2.7919769287109375e-05,
      "model_forward_time": 3.1318094730377197,
      "step": 7319
    },
    {
      "epoch": 2.7919769287109375e-05,
      "step": 7319,
      "training_step_time": 10.234504461288452
    },
    {
      "epoch": 2.7923583984375e-05,
      "grad_norm": 0.5358084440231323,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 7320
    },
    {
      "epoch": 2.7923583984375e-05,
      "model_forward_time": 2.4424281120300293,
      "step": 7320
    },
    {
      "dynamics_loss_avg": 0.05313499700278044,
      "epoch": 2.7923583984375e-05,
      "step": 7320
    },
    {
      "action_loss_avg": 0.021271014492958783,
      "epoch": 2.7923583984375e-05,
      "step": 7320
    },
    {
      "epoch": 2.7923583984375e-05,
      "step": 7320,
      "training_step_time": 7.4748005867004395
    },
    {
      "epoch": 2.7923583984375e-05,
      "model_forward_time": 2.6971938610076904,
      "step": 7320
    },
    {
      "epoch": 2.7923583984375e-05,
      "step": 7320,
      "training_step_time": 7.722760200500488
    },
    {
      "epoch": 2.7923583984375e-05,
      "model_forward_time": 2.4262139797210693,
      "step": 7320
    },
    {
      "epoch": 2.7923583984375e-05,
      "step": 7320,
      "training_step_time": 7.446681261062622
    },
    {
      "epoch": 2.7923583984375e-05,
      "model_forward_time": 3.1292548179626465,
      "step": 7320
    },
    {
      "epoch": 2.7923583984375e-05,
      "step": 7320,
      "training_step_time": 10.251302003860474
    },
    {
      "epoch": 2.7927398681640624e-05,
      "model_forward_time": 2.4453296661376953,
      "step": 7321
    },
    {
      "epoch": 2.7927398681640624e-05,
      "step": 7321,
      "training_step_time": 7.4690845012664795
    },
    {
      "epoch": 2.7927398681640624e-05,
      "model_forward_time": 3.1274452209472656,
      "step": 7321
    },
    {
      "epoch": 2.7927398681640624e-05,
      "step": 7321,
      "training_step_time": 9.894345998764038
    },
    {
      "epoch": 2.7927398681640624e-05,
      "model_forward_time": 2.7123186588287354,
      "step": 7321
    },
    {
      "epoch": 2.7927398681640624e-05,
      "step": 7321,
      "training_step_time": 7.73636531829834
    },
    {
      "epoch": 2.7927398681640624e-05,
      "model_forward_time": 3.1306889057159424,
      "step": 7321
    },
    {
      "epoch": 2.7927398681640624e-05,
      "step": 7321,
      "training_step_time": 10.00308609008789
    },
    {
      "epoch": 2.793121337890625e-05,
      "model_forward_time": 2.4391446113586426,
      "step": 7322
    },
    {
      "epoch": 2.793121337890625e-05,
      "step": 7322,
      "training_step_time": 7.464905261993408
    },
    {
      "epoch": 2.793121337890625e-05,
      "model_forward_time": 2.431300401687622,
      "step": 7322
    },
    {
      "epoch": 2.793121337890625e-05,
      "step": 7322,
      "training_step_time": 7.457298278808594
    },
    {
      "epoch": 2.793121337890625e-05,
      "model_forward_time": 3.1297433376312256,
      "step": 7322
    },
    {
      "dynamics_loss_avg": 0.05276577118784189,
      "epoch": 2.793121337890625e-05,
      "step": 7322
    },
    {
      "action_loss_avg": 0.028520776983350517,
      "epoch": 2.793121337890625e-05,
      "step": 7322
    },
    {
      "epoch": 2.793121337890625e-05,
      "step": 7322,
      "training_step_time": 9.903674840927124
    },
    {
      "epoch": 2.793121337890625e-05,
      "model_forward_time": 2.4311769008636475,
      "step": 7322
    },
    {
      "epoch": 2.793121337890625e-05,
      "step": 7322,
      "training_step_time": 10.300163984298706
    },
    {
      "epoch": 2.7935028076171877e-05,
      "model_forward_time": 2.771418809890747,
      "step": 7323
    },
    {
      "epoch": 2.7935028076171877e-05,
      "step": 7323,
      "training_step_time": 7.804893732070923
    },
    {
      "epoch": 2.7935028076171877e-05,
      "model_forward_time": 3.1285605430603027,
      "step": 7323
    },
    {
      "epoch": 2.7935028076171877e-05,
      "step": 7323,
      "training_step_time": 9.902525901794434
    },
    {
      "epoch": 2.7935028076171877e-05,
      "model_forward_time": 2.430972099304199,
      "step": 7323
    },
    {
      "epoch": 2.7935028076171877e-05,
      "step": 7323,
      "training_step_time": 7.457846164703369
    },
    {
      "epoch": 2.7935028076171877e-05,
      "model_forward_time": 3.126908302307129,
      "step": 7323
    },
    {
      "epoch": 2.7935028076171877e-05,
      "step": 7323,
      "training_step_time": 10.312535047531128
    },
    {
      "epoch": 2.79388427734375e-05,
      "model_forward_time": 3.1515464782714844,
      "step": 7324
    },
    {
      "epoch": 2.79388427734375e-05,
      "step": 7324,
      "training_step_time": 9.92414140701294
    },
    {
      "epoch": 2.79388427734375e-05,
      "model_forward_time": 2.679072856903076,
      "step": 7324
    },
    {
      "epoch": 2.79388427734375e-05,
      "step": 7324,
      "training_step_time": 7.701827526092529
    },
    {
      "epoch": 2.79388427734375e-05,
      "model_forward_time": 3.12801456451416,
      "step": 7324
    },
    {
      "epoch": 2.79388427734375e-05,
      "step": 7324,
      "training_step_time": 9.899698495864868
    },
    {
      "epoch": 2.79388427734375e-05,
      "model_forward_time": 2.427410364151001,
      "step": 7324
    },
    {
      "epoch": 2.79388427734375e-05,
      "step": 7324,
      "training_step_time": 10.001538515090942
    },
    {
      "epoch": 2.7942657470703126e-05,
      "model_forward_time": 2.460951805114746,
      "step": 7325
    },
    {
      "dynamics_loss_avg": 0.05099013689905405,
      "epoch": 2.7942657470703126e-05,
      "step": 7325
    },
    {
      "action_loss_avg": 0.021528636384755374,
      "epoch": 2.7942657470703126e-05,
      "step": 7325
    },
    {
      "epoch": 2.7942657470703126e-05,
      "step": 7325,
      "training_step_time": 7.502103567123413
    },
    {
      "epoch": 2.7942657470703126e-05,
      "model_forward_time": 2.43390154838562,
      "step": 7325
    },
    {
      "epoch": 2.7942657470703126e-05,
      "step": 7325,
      "training_step_time": 7.457175254821777
    },
    {
      "epoch": 2.7942657470703126e-05,
      "model_forward_time": 2.6913743019104004,
      "step": 7325
    },
    {
      "epoch": 2.7942657470703126e-05,
      "step": 7325,
      "training_step_time": 7.715704441070557
    },
    {
      "epoch": 2.7942657470703126e-05,
      "model_forward_time": 1.7420763969421387,
      "step": 7325
    },
    {
      "epoch": 2.7942657470703126e-05,
      "step": 7325,
      "training_step_time": 10.288520097732544
    },
    {
      "epoch": 2.794647216796875e-05,
      "model_forward_time": 3.14453387260437,
      "step": 7326
    },
    {
      "epoch": 2.794647216796875e-05,
      "step": 7326,
      "training_step_time": 9.91022801399231
    },
    {
      "epoch": 2.794647216796875e-05,
      "model_forward_time": 3.138195514678955,
      "step": 7326
    },
    {
      "epoch": 2.794647216796875e-05,
      "step": 7326,
      "training_step_time": 9.904847621917725
    },
    {
      "epoch": 2.794647216796875e-05,
      "model_forward_time": 2.43752121925354,
      "step": 7326
    },
    {
      "epoch": 2.794647216796875e-05,
      "step": 7326,
      "training_step_time": 7.462044715881348
    },
    {
      "epoch": 2.794647216796875e-05,
      "model_forward_time": 3.139249086380005,
      "step": 7326
    },
    {
      "epoch": 2.794647216796875e-05,
      "step": 7326,
      "training_step_time": 10.017500400543213
    },
    {
      "epoch": 2.7950286865234375e-05,
      "model_forward_time": 2.0816025733947754,
      "step": 7327
    },
    {
      "epoch": 2.7950286865234375e-05,
      "step": 7327,
      "training_step_time": 5.539976596832275
    },
    {
      "epoch": 2.7950286865234375e-05,
      "model_forward_time": 2.438675880432129,
      "step": 7327
    },
    {
      "epoch": 2.7950286865234375e-05,
      "step": 7327,
      "training_step_time": 7.469969749450684
    },
    {
      "epoch": 2.7950286865234375e-05,
      "model_forward_time": 2.435779571533203,
      "step": 7327
    },
    {
      "dynamics_loss_avg": 0.0505368085578084,
      "epoch": 2.7950286865234375e-05,
      "step": 7327
    },
    {
      "action_loss_avg": 0.018045483715832235,
      "epoch": 2.7950286865234375e-05,
      "step": 7327
    },
    {
      "epoch": 2.7950286865234375e-05,
      "step": 7327,
      "training_step_time": 7.466466426849365
    },
    {
      "epoch": 2.7950286865234375e-05,
      "model_forward_time": 3.141644239425659,
      "step": 7327
    },
    {
      "epoch": 2.7950286865234375e-05,
      "step": 7327,
      "training_step_time": 10.304609060287476
    },
    {
      "epoch": 2.79541015625e-05,
      "model_forward_time": 3.1543099880218506,
      "step": 7328
    },
    {
      "epoch": 2.79541015625e-05,
      "step": 7328,
      "training_step_time": 9.932499885559082
    },
    {
      "epoch": 2.79541015625e-05,
      "model_forward_time": 3.426663398742676,
      "step": 7328
    },
    {
      "epoch": 2.79541015625e-05,
      "step": 7328,
      "training_step_time": 10.20608901977539
    },
    {
      "epoch": 2.79541015625e-05,
      "model_forward_time": 3.142726182937622,
      "step": 7328
    },
    {
      "epoch": 2.79541015625e-05,
      "step": 7328,
      "training_step_time": 9.927811622619629
    },
    {
      "epoch": 2.79541015625e-05,
      "model_forward_time": 2.4349567890167236,
      "step": 7328
    },
    {
      "epoch": 2.79541015625e-05,
      "step": 7328,
      "training_step_time": 10.273180723190308
    },
    {
      "epoch": 2.7957916259765624e-05,
      "model_forward_time": 3.146331787109375,
      "step": 7329
    },
    {
      "epoch": 2.7957916259765624e-05,
      "step": 7329,
      "training_step_time": 9.925768852233887
    },
    {
      "epoch": 2.7957916259765624e-05,
      "model_forward_time": 3.1294708251953125,
      "step": 7329
    },
    {
      "epoch": 2.7957916259765624e-05,
      "step": 7329,
      "training_step_time": 9.909295558929443
    },
    {
      "epoch": 2.7957916259765624e-05,
      "model_forward_time": 2.7065272331237793,
      "step": 7329
    },
    {
      "epoch": 2.7957916259765624e-05,
      "step": 7329,
      "training_step_time": 7.736555099487305
    },
    {
      "epoch": 2.7957916259765624e-05,
      "model_forward_time": 3.1359500885009766,
      "step": 7329
    },
    {
      "epoch": 2.7957916259765624e-05,
      "step": 7329,
      "training_step_time": 10.24429702758789
    },
    {
      "epoch": 2.796173095703125e-05,
      "grad_norm": 0.32067468762397766,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 7330
    },
    {
      "epoch": 2.796173095703125e-05,
      "model_forward_time": 3.173161506652832,
      "step": 7330
    },
    {
      "dynamics_loss_avg": 0.06281272415071726,
      "epoch": 2.796173095703125e-05,
      "step": 7330
    },
    {
      "action_loss_avg": 0.026233050879091023,
      "epoch": 2.796173095703125e-05,
      "step": 7330
    },
    {
      "epoch": 2.796173095703125e-05,
      "step": 7330,
      "training_step_time": 9.956956148147583
    },
    {
      "epoch": 2.796173095703125e-05,
      "model_forward_time": 3.1338353157043457,
      "step": 7330
    },
    {
      "epoch": 2.796173095703125e-05,
      "step": 7330,
      "training_step_time": 9.911103963851929
    },
    {
      "epoch": 2.796173095703125e-05,
      "model_forward_time": 3.132559299468994,
      "step": 7330
    },
    {
      "epoch": 2.796173095703125e-05,
      "step": 7330,
      "training_step_time": 9.904610872268677
    },
    {
      "epoch": 2.796173095703125e-05,
      "model_forward_time": 3.4458742141723633,
      "step": 7330
    },
    {
      "epoch": 2.796173095703125e-05,
      "step": 7330,
      "training_step_time": 10.321174383163452
    },
    {
      "epoch": 2.7965545654296877e-05,
      "model_forward_time": 2.471384048461914,
      "step": 7331
    },
    {
      "epoch": 2.7965545654296877e-05,
      "step": 7331,
      "training_step_time": 7.50663161277771
    },
    {
      "epoch": 2.7965545654296877e-05,
      "model_forward_time": 3.1324844360351562,
      "step": 7331
    },
    {
      "epoch": 2.7965545654296877e-05,
      "step": 7331,
      "training_step_time": 9.914634466171265
    },
    {
      "epoch": 2.7965545654296877e-05,
      "model_forward_time": 3.1406099796295166,
      "step": 7331
    },
    {
      "epoch": 2.7965545654296877e-05,
      "step": 7331,
      "training_step_time": 9.925175428390503
    },
    {
      "epoch": 2.7965545654296877e-05,
      "model_forward_time": 3.136779308319092,
      "step": 7331
    },
    {
      "epoch": 2.7965545654296877e-05,
      "step": 7331,
      "training_step_time": 10.029216527938843
    },
    {
      "epoch": 2.79693603515625e-05,
      "model_forward_time": 3.4412219524383545,
      "step": 7332
    },
    {
      "epoch": 2.79693603515625e-05,
      "step": 7332,
      "training_step_time": 10.226646184921265
    },
    {
      "epoch": 2.79693603515625e-05,
      "model_forward_time": 3.142347812652588,
      "step": 7332
    },
    {
      "epoch": 2.79693603515625e-05,
      "step": 7332,
      "training_step_time": 9.924931526184082
    },
    {
      "epoch": 2.79693603515625e-05,
      "model_forward_time": 1.7447764873504639,
      "step": 7332
    },
    {
      "dynamics_loss_avg": 0.05181524511426687,
      "epoch": 2.79693603515625e-05,
      "step": 7332
    },
    {
      "action_loss_avg": 0.019839126709848644,
      "epoch": 2.79693603515625e-05,
      "step": 7332
    },
    {
      "epoch": 2.79693603515625e-05,
      "step": 7332,
      "training_step_time": 5.209928512573242
    },
    {
      "epoch": 2.79693603515625e-05,
      "model_forward_time": 3.1291518211364746,
      "step": 7332
    },
    {
      "epoch": 2.79693603515625e-05,
      "step": 7332,
      "training_step_time": 10.318897724151611
    },
    {
      "epoch": 2.7973175048828126e-05,
      "model_forward_time": 2.452162981033325,
      "step": 7333
    },
    {
      "epoch": 2.7973175048828126e-05,
      "step": 7333,
      "training_step_time": 7.490095376968384
    },
    {
      "epoch": 2.7973175048828126e-05,
      "model_forward_time": 3.132450580596924,
      "step": 7333
    },
    {
      "epoch": 2.7973175048828126e-05,
      "step": 7333,
      "training_step_time": 9.904861211776733
    },
    {
      "epoch": 2.7973175048828126e-05,
      "model_forward_time": 2.699615955352783,
      "step": 7333
    },
    {
      "epoch": 2.7973175048828126e-05,
      "step": 7333,
      "training_step_time": 7.730343341827393
    },
    {
      "epoch": 2.7973175048828126e-05,
      "model_forward_time": 2.427448272705078,
      "step": 7333
    },
    {
      "epoch": 2.7973175048828126e-05,
      "step": 7333,
      "training_step_time": 10.246067762374878
    },
    {
      "epoch": 2.797698974609375e-05,
      "model_forward_time": 2.4657046794891357,
      "step": 7334
    },
    {
      "epoch": 2.797698974609375e-05,
      "step": 7334,
      "training_step_time": 7.504181385040283
    },
    {
      "epoch": 2.797698974609375e-05,
      "model_forward_time": 2.428366184234619,
      "step": 7334
    },
    {
      "epoch": 2.797698974609375e-05,
      "step": 7334,
      "training_step_time": 7.459677219390869
    },
    {
      "epoch": 2.797698974609375e-05,
      "model_forward_time": 3.1294193267822266,
      "step": 7334
    },
    {
      "epoch": 2.797698974609375e-05,
      "step": 7334,
      "training_step_time": 9.90272831916809
    },
    {
      "epoch": 2.797698974609375e-05,
      "model_forward_time": 3.4008727073669434,
      "step": 7334
    },
    {
      "epoch": 2.797698974609375e-05,
      "step": 7334,
      "training_step_time": 10.268615484237671
    },
    {
      "epoch": 2.7980804443359375e-05,
      "model_forward_time": 3.150477647781372,
      "step": 7335
    },
    {
      "dynamics_loss_avg": 0.06374746356159448,
      "epoch": 2.7980804443359375e-05,
      "step": 7335
    },
    {
      "action_loss_avg": 0.023644372541457414,
      "epoch": 2.7980804443359375e-05,
      "step": 7335
    },
    {
      "epoch": 2.7980804443359375e-05,
      "step": 7335,
      "training_step_time": 9.927013397216797
    },
    {
      "epoch": 2.7980804443359375e-05,
      "model_forward_time": 2.4291772842407227,
      "step": 7335
    },
    {
      "epoch": 2.7980804443359375e-05,
      "step": 7335,
      "training_step_time": 7.458510875701904
    },
    {
      "epoch": 2.7980804443359375e-05,
      "model_forward_time": 3.1289474964141846,
      "step": 7335
    },
    {
      "epoch": 2.7980804443359375e-05,
      "step": 7335,
      "training_step_time": 9.907639026641846
    },
    {
      "epoch": 2.7980804443359375e-05,
      "model_forward_time": 3.138669967651367,
      "step": 7335
    },
    {
      "epoch": 2.7980804443359375e-05,
      "step": 7335,
      "training_step_time": 10.274407386779785
    },
    {
      "epoch": 2.7984619140625e-05,
      "model_forward_time": 3.4427435398101807,
      "step": 7336
    },
    {
      "epoch": 2.7984619140625e-05,
      "step": 7336,
      "training_step_time": 10.2222261428833
    },
    {
      "epoch": 2.7984619140625e-05,
      "model_forward_time": 2.4309847354888916,
      "step": 7336
    },
    {
      "epoch": 2.7984619140625e-05,
      "step": 7336,
      "training_step_time": 7.462420463562012
    },
    {
      "epoch": 2.7984619140625e-05,
      "model_forward_time": 1.7409439086914062,
      "step": 7336
    },
    {
      "epoch": 2.7984619140625e-05,
      "step": 7336,
      "training_step_time": 5.2004172801971436
    },
    {
      "epoch": 2.7984619140625e-05,
      "model_forward_time": 2.428039073944092,
      "step": 7336
    },
    {
      "epoch": 2.7984619140625e-05,
      "step": 7336,
      "training_step_time": 10.259501695632935
    },
    {
      "epoch": 2.7988433837890624e-05,
      "model_forward_time": 2.4972078800201416,
      "step": 7337
    },
    {
      "epoch": 2.7988433837890624e-05,
      "step": 7337,
      "training_step_time": 7.542083024978638
    },
    {
      "epoch": 2.7988433837890624e-05,
      "model_forward_time": 3.1339659690856934,
      "step": 7337
    },
    {
      "epoch": 2.7988433837890624e-05,
      "step": 7337,
      "training_step_time": 9.9101243019104
    },
    {
      "epoch": 2.7988433837890624e-05,
      "model_forward_time": 3.399411678314209,
      "step": 7337
    },
    {
      "dynamics_loss_avg": 0.05751785719767213,
      "epoch": 2.7988433837890624e-05,
      "step": 7337
    },
    {
      "action_loss_avg": 0.024833544716238977,
      "epoch": 2.7988433837890624e-05,
      "step": 7337
    },
    {
      "epoch": 2.7988433837890624e-05,
      "step": 7337,
      "training_step_time": 10.172419548034668
    },
    {
      "epoch": 2.7988433837890624e-05,
      "model_forward_time": 3.14526104927063,
      "step": 7337
    },
    {
      "epoch": 2.7988433837890624e-05,
      "step": 7337,
      "training_step_time": 10.305484533309937
    },
    {
      "epoch": 2.799224853515625e-05,
      "model_forward_time": 2.4464056491851807,
      "step": 7338
    },
    {
      "epoch": 2.799224853515625e-05,
      "step": 7338,
      "training_step_time": 7.473747968673706
    },
    {
      "epoch": 2.799224853515625e-05,
      "model_forward_time": 2.427730083465576,
      "step": 7338
    },
    {
      "epoch": 2.799224853515625e-05,
      "step": 7338,
      "training_step_time": 7.453949689865112
    },
    {
      "epoch": 2.799224853515625e-05,
      "model_forward_time": 2.430830240249634,
      "step": 7338
    },
    {
      "epoch": 2.799224853515625e-05,
      "step": 7338,
      "training_step_time": 7.460877895355225
    },
    {
      "epoch": 2.799224853515625e-05,
      "model_forward_time": 3.139956474304199,
      "step": 7338
    },
    {
      "epoch": 2.799224853515625e-05,
      "step": 7338,
      "training_step_time": 10.311525344848633
    },
    {
      "epoch": 2.7996063232421877e-05,
      "model_forward_time": 3.5007543563842773,
      "step": 7339
    },
    {
      "epoch": 2.7996063232421877e-05,
      "step": 7339,
      "training_step_time": 10.273924827575684
    },
    {
      "epoch": 2.7996063232421877e-05,
      "model_forward_time": 2.4343738555908203,
      "step": 7339
    },
    {
      "epoch": 2.7996063232421877e-05,
      "step": 7339,
      "training_step_time": 7.462062120437622
    },
    {
      "epoch": 2.7996063232421877e-05,
      "model_forward_time": 2.4322023391723633,
      "step": 7339
    },
    {
      "epoch": 2.7996063232421877e-05,
      "step": 7339,
      "training_step_time": 7.459094285964966
    },
    {
      "epoch": 2.7996063232421877e-05,
      "model_forward_time": 3.1304755210876465,
      "step": 7339
    },
    {
      "epoch": 2.7996063232421877e-05,
      "step": 7339,
      "training_step_time": 10.008060693740845
    },
    {
      "epoch": 2.79998779296875e-05,
      "grad_norm": 0.40504151582717896,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 7340
    },
    {
      "epoch": 2.79998779296875e-05,
      "model_forward_time": 3.1441097259521484,
      "step": 7340
    },
    {
      "dynamics_loss_avg": 0.06090114917606115,
      "epoch": 2.79998779296875e-05,
      "step": 7340
    },
    {
      "action_loss_avg": 0.02638543099164963,
      "epoch": 2.79998779296875e-05,
      "step": 7340
    },
    {
      "epoch": 2.79998779296875e-05,
      "step": 7340,
      "training_step_time": 9.916303157806396
    },
    {
      "epoch": 2.79998779296875e-05,
      "model_forward_time": 3.432471513748169,
      "step": 7340
    },
    {
      "epoch": 2.79998779296875e-05,
      "step": 7340,
      "training_step_time": 10.201273679733276
    },
    {
      "epoch": 2.79998779296875e-05,
      "model_forward_time": 2.4349989891052246,
      "step": 7340
    },
    {
      "epoch": 2.79998779296875e-05,
      "step": 7340,
      "training_step_time": 7.464964389801025
    },
    {
      "epoch": 2.79998779296875e-05,
      "model_forward_time": 2.431975841522217,
      "step": 7340
    },
    {
      "epoch": 2.79998779296875e-05,
      "step": 7340,
      "training_step_time": 10.283033847808838
    },
    {
      "epoch": 2.8003692626953126e-05,
      "model_forward_time": 3.1460208892822266,
      "step": 7341
    },
    {
      "epoch": 2.8003692626953126e-05,
      "step": 7341,
      "training_step_time": 9.923516035079956
    },
    {
      "epoch": 2.8003692626953126e-05,
      "model_forward_time": 2.432905673980713,
      "step": 7341
    },
    {
      "epoch": 2.8003692626953126e-05,
      "step": 7341,
      "training_step_time": 7.4670515060424805
    },
    {
      "epoch": 2.8003692626953126e-05,
      "model_forward_time": 2.7149643898010254,
      "step": 7341
    },
    {
      "epoch": 2.8003692626953126e-05,
      "step": 7341,
      "training_step_time": 7.749828815460205
    },
    {
      "epoch": 2.8003692626953126e-05,
      "model_forward_time": 3.134376049041748,
      "step": 7341
    },
    {
      "epoch": 2.8003692626953126e-05,
      "step": 7341,
      "training_step_time": 10.260164976119995
    },
    {
      "epoch": 2.800750732421875e-05,
      "model_forward_time": 3.1462175846099854,
      "step": 7342
    },
    {
      "epoch": 2.800750732421875e-05,
      "step": 7342,
      "training_step_time": 9.925498723983765
    },
    {
      "epoch": 2.800750732421875e-05,
      "model_forward_time": 3.1349024772644043,
      "step": 7342
    },
    {
      "epoch": 2.800750732421875e-05,
      "step": 7342,
      "training_step_time": 9.911859512329102
    },
    {
      "epoch": 2.800750732421875e-05,
      "model_forward_time": 3.1338162422180176,
      "step": 7342
    },
    {
      "dynamics_loss_avg": 0.054347021505236626,
      "epoch": 2.800750732421875e-05,
      "step": 7342
    },
    {
      "action_loss_avg": 0.021913842670619486,
      "epoch": 2.800750732421875e-05,
      "step": 7342
    },
    {
      "epoch": 2.800750732421875e-05,
      "step": 7342,
      "training_step_time": 9.912783861160278
    },
    {
      "epoch": 2.800750732421875e-05,
      "model_forward_time": 2.436612606048584,
      "step": 7342
    },
    {
      "epoch": 2.800750732421875e-05,
      "step": 7342,
      "training_step_time": 10.01791787147522
    },
    {
      "epoch": 2.8011322021484375e-05,
      "model_forward_time": 2.4391820430755615,
      "step": 7343
    },
    {
      "epoch": 2.8011322021484375e-05,
      "step": 7343,
      "training_step_time": 7.4689435958862305
    },
    {
      "epoch": 2.8011322021484375e-05,
      "model_forward_time": 3.133594036102295,
      "step": 7343
    },
    {
      "epoch": 2.8011322021484375e-05,
      "step": 7343,
      "training_step_time": 9.906287908554077
    },
    {
      "epoch": 2.8011322021484375e-05,
      "model_forward_time": 3.1307599544525146,
      "step": 7343
    },
    {
      "epoch": 2.8011322021484375e-05,
      "step": 7343,
      "training_step_time": 9.90241026878357
    },
    {
      "epoch": 2.8011322021484375e-05,
      "model_forward_time": 2.433297634124756,
      "step": 7343
    },
    {
      "epoch": 2.8011322021484375e-05,
      "step": 7343,
      "training_step_time": 10.328184604644775
    },
    {
      "epoch": 2.801513671875e-05,
      "model_forward_time": 2.4487032890319824,
      "step": 7344
    },
    {
      "epoch": 2.801513671875e-05,
      "step": 7344,
      "training_step_time": 7.481900930404663
    },
    {
      "epoch": 2.801513671875e-05,
      "model_forward_time": 2.718327760696411,
      "step": 7344
    },
    {
      "epoch": 2.801513671875e-05,
      "step": 7344,
      "training_step_time": 7.740035533905029
    },
    {
      "epoch": 2.801513671875e-05,
      "model_forward_time": 2.429668664932251,
      "step": 7344
    },
    {
      "epoch": 2.801513671875e-05,
      "step": 7344,
      "training_step_time": 7.449941873550415
    },
    {
      "epoch": 2.801513671875e-05,
      "model_forward_time": 2.427011728286743,
      "step": 7344
    },
    {
      "epoch": 2.801513671875e-05,
      "step": 7344,
      "training_step_time": 10.316880702972412
    },
    {
      "epoch": 2.8018951416015624e-05,
      "model_forward_time": 3.16033935546875,
      "step": 7345
    },
    {
      "dynamics_loss_avg": 0.05749041214585304,
      "epoch": 2.8018951416015624e-05,
      "step": 7345
    },
    {
      "action_loss_avg": 0.022786941519007085,
      "epoch": 2.8018951416015624e-05,
      "step": 7345
    },
    {
      "epoch": 2.8018951416015624e-05,
      "step": 7345,
      "training_step_time": 9.939327239990234
    },
    {
      "epoch": 2.8018951416015624e-05,
      "model_forward_time": 3.1424944400787354,
      "step": 7345
    },
    {
      "epoch": 2.8018951416015624e-05,
      "step": 7345,
      "training_step_time": 9.915422201156616
    },
    {
      "epoch": 2.8018951416015624e-05,
      "model_forward_time": 2.778596878051758,
      "step": 7345
    },
    {
      "epoch": 2.8018951416015624e-05,
      "step": 7345,
      "training_step_time": 7.808685541152954
    },
    {
      "epoch": 2.8018951416015624e-05,
      "model_forward_time": 2.4366159439086914,
      "step": 7345
    },
    {
      "epoch": 2.8018951416015624e-05,
      "step": 7345,
      "training_step_time": 10.019344568252563
    },
    {
      "epoch": 2.802276611328125e-05,
      "model_forward_time": 3.1558141708374023,
      "step": 7346
    },
    {
      "epoch": 2.802276611328125e-05,
      "step": 7346,
      "training_step_time": 9.927402973175049
    },
    {
      "epoch": 2.802276611328125e-05,
      "model_forward_time": 1.7479631900787354,
      "step": 7346
    },
    {
      "epoch": 2.802276611328125e-05,
      "step": 7346,
      "training_step_time": 5.2040510177612305
    },
    {
      "epoch": 2.802276611328125e-05,
      "model_forward_time": 2.4361605644226074,
      "step": 7346
    },
    {
      "epoch": 2.802276611328125e-05,
      "step": 7346,
      "training_step_time": 7.456955432891846
    },
    {
      "epoch": 2.802276611328125e-05,
      "model_forward_time": 3.132596969604492,
      "step": 7346
    },
    {
      "epoch": 2.802276611328125e-05,
      "step": 7346,
      "training_step_time": 10.259153842926025
    },
    {
      "epoch": 2.8026580810546877e-05,
      "model_forward_time": 3.427446126937866,
      "step": 7347
    },
    {
      "epoch": 2.8026580810546877e-05,
      "step": 7347,
      "training_step_time": 10.195642948150635
    },
    {
      "epoch": 2.8026580810546877e-05,
      "model_forward_time": 3.1427929401397705,
      "step": 7347
    },
    {
      "epoch": 2.8026580810546877e-05,
      "step": 7347,
      "training_step_time": 9.90875506401062
    },
    {
      "epoch": 2.8026580810546877e-05,
      "model_forward_time": 3.1366186141967773,
      "step": 7347
    },
    {
      "dynamics_loss_avg": 0.06188396941870451,
      "epoch": 2.8026580810546877e-05,
      "step": 7347
    },
    {
      "action_loss_avg": 0.024948463775217533,
      "epoch": 2.8026580810546877e-05,
      "step": 7347
    },
    {
      "epoch": 2.8026580810546877e-05,
      "step": 7347,
      "training_step_time": 9.906882286071777
    },
    {
      "epoch": 2.8026580810546877e-05,
      "model_forward_time": 3.1357669830322266,
      "step": 7347
    },
    {
      "epoch": 2.8026580810546877e-05,
      "step": 7347,
      "training_step_time": 10.014892578125
    },
    {
      "epoch": 2.80303955078125e-05,
      "model_forward_time": 2.4597065448760986,
      "step": 7348
    },
    {
      "epoch": 2.80303955078125e-05,
      "step": 7348,
      "training_step_time": 7.492467880249023
    },
    {
      "epoch": 2.80303955078125e-05,
      "model_forward_time": 2.7532176971435547,
      "step": 7348
    },
    {
      "epoch": 2.80303955078125e-05,
      "step": 7348,
      "training_step_time": 7.774495840072632
    },
    {
      "epoch": 2.80303955078125e-05,
      "model_forward_time": 3.1319310665130615,
      "step": 7348
    },
    {
      "epoch": 2.80303955078125e-05,
      "step": 7348,
      "training_step_time": 9.896546363830566
    },
    {
      "epoch": 2.80303955078125e-05,
      "model_forward_time": 1.7389349937438965,
      "step": 7348
    },
    {
      "epoch": 2.80303955078125e-05,
      "step": 7348,
      "training_step_time": 10.292890310287476
    },
    {
      "epoch": 2.8034210205078126e-05,
      "model_forward_time": 2.4713282585144043,
      "step": 7349
    },
    {
      "epoch": 2.8034210205078126e-05,
      "step": 7349,
      "training_step_time": 7.490388870239258
    },
    {
      "epoch": 2.8034210205078126e-05,
      "model_forward_time": 3.127389430999756,
      "step": 7349
    },
    {
      "epoch": 2.8034210205078126e-05,
      "step": 7349,
      "training_step_time": 9.883454084396362
    },
    {
      "epoch": 2.8034210205078126e-05,
      "model_forward_time": 3.4443581104278564,
      "step": 7349
    },
    {
      "epoch": 2.8034210205078126e-05,
      "step": 7349,
      "training_step_time": 10.200608015060425
    },
    {
      "epoch": 2.8034210205078126e-05,
      "model_forward_time": 3.138503313064575,
      "step": 7349
    },
    {
      "epoch": 2.8034210205078126e-05,
      "step": 7349,
      "training_step_time": 10.327565908432007
    },
    {
      "epoch": 2.803802490234375e-05,
      "grad_norm": 0.5080481171607971,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 7350
    },
    {
      "epoch": 2.803802490234375e-05,
      "model_forward_time": 3.146902561187744,
      "step": 7350
    },
    {
      "dynamics_loss_avg": 0.055590916983783246,
      "epoch": 2.803802490234375e-05,
      "step": 7350
    },
    {
      "action_loss_avg": 0.02548948205076158,
      "epoch": 2.803802490234375e-05,
      "step": 7350
    },
    {
      "epoch": 2.803802490234375e-05,
      "step": 7350,
      "training_step_time": 9.915274620056152
    },
    {
      "epoch": 2.803802490234375e-05,
      "model_forward_time": 2.430878162384033,
      "step": 7350
    },
    {
      "epoch": 2.803802490234375e-05,
      "step": 7350,
      "training_step_time": 7.451206922531128
    },
    {
      "epoch": 2.803802490234375e-05,
      "model_forward_time": 2.4301018714904785,
      "step": 7350
    },
    {
      "epoch": 2.803802490234375e-05,
      "step": 7350,
      "training_step_time": 7.452378034591675
    },
    {
      "epoch": 2.803802490234375e-05,
      "model_forward_time": 3.1310982704162598,
      "step": 7350
    },
    {
      "epoch": 2.803802490234375e-05,
      "step": 7350,
      "training_step_time": 10.2653648853302
    },
    {
      "epoch": 2.8041839599609375e-05,
      "model_forward_time": 3.1637144088745117,
      "step": 7351
    },
    {
      "epoch": 2.8041839599609375e-05,
      "step": 7351,
      "training_step_time": 9.932185411453247
    },
    {
      "epoch": 2.8041839599609375e-05,
      "model_forward_time": 2.433720111846924,
      "step": 7351
    },
    {
      "epoch": 2.8041839599609375e-05,
      "step": 7351,
      "training_step_time": 7.45995020866394
    },
    {
      "epoch": 2.8041839599609375e-05,
      "model_forward_time": 1.743131160736084,
      "step": 7351
    },
    {
      "epoch": 2.8041839599609375e-05,
      "step": 7351,
      "training_step_time": 5.197397708892822
    },
    {
      "epoch": 2.8041839599609375e-05,
      "model_forward_time": 3.137573719024658,
      "step": 7351
    },
    {
      "epoch": 2.8041839599609375e-05,
      "step": 7351,
      "training_step_time": 10.231925964355469
    },
    {
      "epoch": 2.8045654296875e-05,
      "model_forward_time": 3.147695541381836,
      "step": 7352
    },
    {
      "epoch": 2.8045654296875e-05,
      "step": 7352,
      "training_step_time": 9.915437936782837
    },
    {
      "epoch": 2.8045654296875e-05,
      "model_forward_time": 3.410473346710205,
      "step": 7352
    },
    {
      "epoch": 2.8045654296875e-05,
      "step": 7352,
      "training_step_time": 10.181489706039429
    },
    {
      "epoch": 2.8045654296875e-05,
      "model_forward_time": 3.139007091522217,
      "step": 7352
    },
    {
      "dynamics_loss_avg": 0.06770182494074106,
      "epoch": 2.8045654296875e-05,
      "step": 7352
    },
    {
      "action_loss_avg": 0.03183821188285947,
      "epoch": 2.8045654296875e-05,
      "step": 7352
    },
    {
      "epoch": 2.8045654296875e-05,
      "step": 7352,
      "training_step_time": 9.915714263916016
    },
    {
      "epoch": 2.8045654296875e-05,
      "model_forward_time": 2.4421374797821045,
      "step": 7352
    },
    {
      "epoch": 2.8045654296875e-05,
      "step": 7352,
      "training_step_time": 10.021502017974854
    },
    {
      "epoch": 2.8049468994140624e-05,
      "model_forward_time": 2.4543843269348145,
      "step": 7353
    },
    {
      "epoch": 2.8049468994140624e-05,
      "step": 7353,
      "training_step_time": 7.475445985794067
    },
    {
      "epoch": 2.8049468994140624e-05,
      "model_forward_time": 2.431877374649048,
      "step": 7353
    },
    {
      "epoch": 2.8049468994140624e-05,
      "step": 7353,
      "training_step_time": 7.455068588256836
    },
    {
      "epoch": 2.8049468994140624e-05,
      "model_forward_time": 3.4498445987701416,
      "step": 7353
    },
    {
      "epoch": 2.8049468994140624e-05,
      "step": 7353,
      "training_step_time": 10.218987464904785
    },
    {
      "epoch": 2.8049468994140624e-05,
      "model_forward_time": 1.7476296424865723,
      "step": 7353
    },
    {
      "epoch": 2.8049468994140624e-05,
      "step": 7353,
      "training_step_time": 10.322549104690552
    },
    {
      "epoch": 2.805328369140625e-05,
      "model_forward_time": 2.449969530105591,
      "step": 7354
    },
    {
      "epoch": 2.805328369140625e-05,
      "step": 7354,
      "training_step_time": 7.477656841278076
    },
    {
      "epoch": 2.805328369140625e-05,
      "model_forward_time": 3.132028579711914,
      "step": 7354
    },
    {
      "epoch": 2.805328369140625e-05,
      "step": 7354,
      "training_step_time": 9.89553689956665
    },
    {
      "epoch": 2.805328369140625e-05,
      "model_forward_time": 2.4317338466644287,
      "step": 7354
    },
    {
      "epoch": 2.805328369140625e-05,
      "step": 7354,
      "training_step_time": 7.4543702602386475
    },
    {
      "epoch": 2.805328369140625e-05,
      "model_forward_time": 2.430419683456421,
      "step": 7354
    },
    {
      "epoch": 2.805328369140625e-05,
      "step": 7354,
      "training_step_time": 10.32573914527893
    },
    {
      "epoch": 2.8057098388671877e-05,
      "model_forward_time": 2.089698553085327,
      "step": 7355
    },
    {
      "dynamics_loss_avg": 0.05831993734464049,
      "epoch": 2.8057098388671877e-05,
      "step": 7355
    },
    {
      "action_loss_avg": 0.02676741871982813,
      "epoch": 2.8057098388671877e-05,
      "step": 7355
    },
    {
      "epoch": 2.8057098388671877e-05,
      "step": 7355,
      "training_step_time": 5.538154363632202
    },
    {
      "epoch": 2.8057098388671877e-05,
      "model_forward_time": 2.4311904907226562,
      "step": 7355
    },
    {
      "epoch": 2.8057098388671877e-05,
      "step": 7355,
      "training_step_time": 7.451573610305786
    },
    {
      "epoch": 2.8057098388671877e-05,
      "model_forward_time": 3.1377193927764893,
      "step": 7355
    },
    {
      "epoch": 2.8057098388671877e-05,
      "step": 7355,
      "training_step_time": 9.901900053024292
    },
    {
      "epoch": 2.8057098388671877e-05,
      "model_forward_time": 3.133246660232544,
      "step": 7355
    },
    {
      "epoch": 2.8057098388671877e-05,
      "step": 7355,
      "training_step_time": 10.305870532989502
    },
    {
      "epoch": 2.80609130859375e-05,
      "model_forward_time": 2.4473190307617188,
      "step": 7356
    },
    {
      "epoch": 2.80609130859375e-05,
      "step": 7356,
      "training_step_time": 7.472121238708496
    },
    {
      "epoch": 2.80609130859375e-05,
      "model_forward_time": 2.434074878692627,
      "step": 7356
    },
    {
      "epoch": 2.80609130859375e-05,
      "step": 7356,
      "training_step_time": 7.476478576660156
    },
    {
      "epoch": 2.80609130859375e-05,
      "model_forward_time": 2.702025890350342,
      "step": 7356
    },
    {
      "epoch": 2.80609130859375e-05,
      "step": 7356,
      "training_step_time": 7.72236704826355
    },
    {
      "epoch": 2.80609130859375e-05,
      "model_forward_time": 3.136406660079956,
      "step": 7356
    },
    {
      "epoch": 2.80609130859375e-05,
      "step": 7356,
      "training_step_time": 10.29107141494751
    },
    {
      "epoch": 2.8064727783203126e-05,
      "model_forward_time": 3.1545846462249756,
      "step": 7357
    },
    {
      "epoch": 2.8064727783203126e-05,
      "step": 7357,
      "training_step_time": 9.916133642196655
    },
    {
      "epoch": 2.8064727783203126e-05,
      "model_forward_time": 3.133899211883545,
      "step": 7357
    },
    {
      "epoch": 2.8064727783203126e-05,
      "step": 7357,
      "training_step_time": 9.894824504852295
    },
    {
      "epoch": 2.8064727783203126e-05,
      "model_forward_time": 2.4335789680480957,
      "step": 7357
    },
    {
      "dynamics_loss_avg": 0.05976866483688355,
      "epoch": 2.8064727783203126e-05,
      "step": 7357
    },
    {
      "action_loss_avg": 0.021155424229800702,
      "epoch": 2.8064727783203126e-05,
      "step": 7357
    },
    {
      "epoch": 2.8064727783203126e-05,
      "step": 7357,
      "training_step_time": 7.767747640609741
    },
    {
      "epoch": 2.8064727783203126e-05,
      "model_forward_time": 3.129321813583374,
      "step": 7357
    },
    {
      "epoch": 2.8064727783203126e-05,
      "step": 7357,
      "training_step_time": 10.024760484695435
    },
    {
      "epoch": 2.806854248046875e-05,
      "model_forward_time": 1.7670135498046875,
      "step": 7358
    },
    {
      "epoch": 2.806854248046875e-05,
      "step": 7358,
      "training_step_time": 5.223021030426025
    },
    {
      "epoch": 2.806854248046875e-05,
      "model_forward_time": 3.135066032409668,
      "step": 7358
    },
    {
      "epoch": 2.806854248046875e-05,
      "step": 7358,
      "training_step_time": 9.898797512054443
    },
    {
      "epoch": 2.806854248046875e-05,
      "model_forward_time": 3.1434288024902344,
      "step": 7358
    },
    {
      "epoch": 2.806854248046875e-05,
      "step": 7358,
      "training_step_time": 9.91063380241394
    },
    {
      "epoch": 2.806854248046875e-05,
      "model_forward_time": 2.4330625534057617,
      "step": 7358
    },
    {
      "epoch": 2.806854248046875e-05,
      "step": 7358,
      "training_step_time": 10.27112364768982
    },
    {
      "epoch": 2.8072357177734375e-05,
      "model_forward_time": 3.424257278442383,
      "step": 7359
    },
    {
      "epoch": 2.8072357177734375e-05,
      "step": 7359,
      "training_step_time": 10.190418004989624
    },
    {
      "epoch": 2.8072357177734375e-05,
      "model_forward_time": 3.1398239135742188,
      "step": 7359
    },
    {
      "epoch": 2.8072357177734375e-05,
      "step": 7359,
      "training_step_time": 9.905689001083374
    },
    {
      "epoch": 2.8072357177734375e-05,
      "model_forward_time": 3.1344850063323975,
      "step": 7359
    },
    {
      "epoch": 2.8072357177734375e-05,
      "step": 7359,
      "training_step_time": 9.904645442962646
    },
    {
      "epoch": 2.8072357177734375e-05,
      "model_forward_time": 2.4381420612335205,
      "step": 7359
    },
    {
      "epoch": 2.8072357177734375e-05,
      "step": 7359,
      "training_step_time": 10.316298007965088
    },
    {
      "epoch": 2.8076171875e-05,
      "grad_norm": 0.5446875095367432,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 7360
    },
    {
      "epoch": 2.8076171875e-05,
      "model_forward_time": 2.450490951538086,
      "step": 7360
    },
    {
      "dynamics_loss_avg": 0.06596810817718506,
      "epoch": 2.8076171875e-05,
      "step": 7360
    },
    {
      "action_loss_avg": 0.0351219130679965,
      "epoch": 2.8076171875e-05,
      "step": 7360
    },
    {
      "epoch": 2.8076171875e-05,
      "step": 7360,
      "training_step_time": 7.471303224563599
    },
    {
      "epoch": 2.8076171875e-05,
      "model_forward_time": 3.410604953765869,
      "step": 7360
    },
    {
      "epoch": 2.8076171875e-05,
      "step": 7360,
      "training_step_time": 10.17447566986084
    },
    {
      "epoch": 2.8076171875e-05,
      "model_forward_time": 2.434683322906494,
      "step": 7360
    },
    {
      "epoch": 2.8076171875e-05,
      "step": 7360,
      "training_step_time": 7.457288503646851
    },
    {
      "epoch": 2.8076171875e-05,
      "model_forward_time": 2.433936357498169,
      "step": 7360
    },
    {
      "epoch": 2.8076171875e-05,
      "step": 7360,
      "training_step_time": 10.336088180541992
    },
    {
      "epoch": 2.8079986572265624e-05,
      "model_forward_time": 3.1582937240600586,
      "step": 7361
    },
    {
      "epoch": 2.8079986572265624e-05,
      "step": 7361,
      "training_step_time": 9.928820848464966
    },
    {
      "epoch": 2.8079986572265624e-05,
      "model_forward_time": 3.141589879989624,
      "step": 7361
    },
    {
      "epoch": 2.8079986572265624e-05,
      "step": 7361,
      "training_step_time": 9.909897565841675
    },
    {
      "epoch": 2.8079986572265624e-05,
      "model_forward_time": 3.399855375289917,
      "step": 7361
    },
    {
      "epoch": 2.8079986572265624e-05,
      "step": 7361,
      "training_step_time": 10.165219068527222
    },
    {
      "epoch": 2.8079986572265624e-05,
      "model_forward_time": 2.4383978843688965,
      "step": 7361
    },
    {
      "epoch": 2.8079986572265624e-05,
      "step": 7361,
      "training_step_time": 10.245834827423096
    },
    {
      "epoch": 2.808380126953125e-05,
      "model_forward_time": 3.1850900650024414,
      "step": 7362
    },
    {
      "epoch": 2.808380126953125e-05,
      "step": 7362,
      "training_step_time": 9.950623273849487
    },
    {
      "epoch": 2.808380126953125e-05,
      "model_forward_time": 3.1428797245025635,
      "step": 7362
    },
    {
      "epoch": 2.808380126953125e-05,
      "step": 7362,
      "training_step_time": 9.918442010879517
    },
    {
      "epoch": 2.808380126953125e-05,
      "model_forward_time": 3.1466007232666016,
      "step": 7362
    },
    {
      "dynamics_loss_avg": 0.05009103761985898,
      "epoch": 2.808380126953125e-05,
      "step": 7362
    },
    {
      "action_loss_avg": 0.028314293781295418,
      "epoch": 2.808380126953125e-05,
      "step": 7362
    },
    {
      "epoch": 2.808380126953125e-05,
      "step": 7362,
      "training_step_time": 9.917186260223389
    },
    {
      "epoch": 2.808380126953125e-05,
      "model_forward_time": 3.4308199882507324,
      "step": 7362
    },
    {
      "epoch": 2.808380126953125e-05,
      "step": 7362,
      "training_step_time": 10.291072130203247
    },
    {
      "epoch": 2.8087615966796877e-05,
      "model_forward_time": 2.457341194152832,
      "step": 7363
    },
    {
      "epoch": 2.8087615966796877e-05,
      "step": 7363,
      "training_step_time": 7.485675811767578
    },
    {
      "epoch": 2.8087615966796877e-05,
      "model_forward_time": 3.1424710750579834,
      "step": 7363
    },
    {
      "epoch": 2.8087615966796877e-05,
      "step": 7363,
      "training_step_time": 9.909299612045288
    },
    {
      "epoch": 2.8087615966796877e-05,
      "model_forward_time": 2.441281318664551,
      "step": 7363
    },
    {
      "epoch": 2.8087615966796877e-05,
      "step": 7363,
      "training_step_time": 7.464473009109497
    },
    {
      "epoch": 2.8087615966796877e-05,
      "model_forward_time": 3.1411306858062744,
      "step": 7363
    },
    {
      "epoch": 2.8087615966796877e-05,
      "step": 7363,
      "training_step_time": 10.286645889282227
    },
    {
      "epoch": 2.80914306640625e-05,
      "model_forward_time": 3.150893449783325,
      "step": 7364
    },
    {
      "epoch": 2.80914306640625e-05,
      "step": 7364,
      "training_step_time": 10.196083784103394
    },
    {
      "epoch": 2.80914306640625e-05,
      "model_forward_time": 2.435128927230835,
      "step": 7364
    },
    {
      "epoch": 2.80914306640625e-05,
      "step": 7364,
      "training_step_time": 7.454682111740112
    },
    {
      "epoch": 2.80914306640625e-05,
      "model_forward_time": 2.429598331451416,
      "step": 7364
    },
    {
      "epoch": 2.80914306640625e-05,
      "step": 7364,
      "training_step_time": 7.457358121871948
    },
    {
      "epoch": 2.80914306640625e-05,
      "model_forward_time": 3.133316993713379,
      "step": 7364
    },
    {
      "epoch": 2.80914306640625e-05,
      "step": 7364,
      "training_step_time": 10.324342250823975
    },
    {
      "epoch": 2.8095245361328126e-05,
      "model_forward_time": 3.1423072814941406,
      "step": 7365
    },
    {
      "dynamics_loss_avg": 0.04871522840112448,
      "epoch": 2.8095245361328126e-05,
      "step": 7365
    },
    {
      "action_loss_avg": 0.023251879680901765,
      "epoch": 2.8095245361328126e-05,
      "step": 7365
    },
    {
      "epoch": 2.8095245361328126e-05,
      "step": 7365,
      "training_step_time": 9.916222095489502
    },
    {
      "epoch": 2.8095245361328126e-05,
      "model_forward_time": 1.7428832054138184,
      "step": 7365
    },
    {
      "epoch": 2.8095245361328126e-05,
      "step": 7365,
      "training_step_time": 5.195102691650391
    },
    {
      "epoch": 2.8095245361328126e-05,
      "model_forward_time": 2.699385404586792,
      "step": 7365
    },
    {
      "epoch": 2.8095245361328126e-05,
      "step": 7365,
      "training_step_time": 7.714660406112671
    },
    {
      "epoch": 2.8095245361328126e-05,
      "model_forward_time": 2.4275293350219727,
      "step": 7365
    },
    {
      "epoch": 2.8095245361328126e-05,
      "step": 7365,
      "training_step_time": 10.247804880142212
    },
    {
      "epoch": 2.809906005859375e-05,
      "model_forward_time": 3.1407153606414795,
      "step": 7366
    },
    {
      "epoch": 2.809906005859375e-05,
      "step": 7366,
      "training_step_time": 9.897646188735962
    },
    {
      "epoch": 2.809906005859375e-05,
      "model_forward_time": 2.4294991493225098,
      "step": 7366
    },
    {
      "epoch": 2.809906005859375e-05,
      "step": 7366,
      "training_step_time": 7.4452619552612305
    },
    {
      "epoch": 2.809906005859375e-05,
      "model_forward_time": 2.4242053031921387,
      "step": 7366
    },
    {
      "epoch": 2.809906005859375e-05,
      "step": 7366,
      "training_step_time": 7.445995330810547
    },
    {
      "epoch": 2.809906005859375e-05,
      "model_forward_time": 3.1298208236694336,
      "step": 7366
    },
    {
      "epoch": 2.809906005859375e-05,
      "step": 7366,
      "training_step_time": 10.29982876777649
    },
    {
      "epoch": 2.8102874755859375e-05,
      "model_forward_time": 2.4681458473205566,
      "step": 7367
    },
    {
      "epoch": 2.8102874755859375e-05,
      "step": 7367,
      "training_step_time": 7.482745170593262
    },
    {
      "epoch": 2.8102874755859375e-05,
      "model_forward_time": 3.126589298248291,
      "step": 7367
    },
    {
      "epoch": 2.8102874755859375e-05,
      "step": 7367,
      "training_step_time": 9.886587142944336
    },
    {
      "epoch": 2.8102874755859375e-05,
      "model_forward_time": 3.1266067028045654,
      "step": 7367
    },
    {
      "dynamics_loss_avg": 0.04921774975955486,
      "epoch": 2.8102874755859375e-05,
      "step": 7367
    },
    {
      "action_loss_avg": 0.023688054550439118,
      "epoch": 2.8102874755859375e-05,
      "step": 7367
    },
    {
      "epoch": 2.8102874755859375e-05,
      "step": 7367,
      "training_step_time": 9.889518022537231
    },
    {
      "epoch": 2.8102874755859375e-05,
      "model_forward_time": 3.1337618827819824,
      "step": 7367
    },
    {
      "epoch": 2.8102874755859375e-05,
      "step": 7367,
      "training_step_time": 10.269285202026367
    },
    {
      "epoch": 2.8106689453125e-05,
      "model_forward_time": 3.1442768573760986,
      "step": 7368
    },
    {
      "epoch": 2.8106689453125e-05,
      "step": 7368,
      "training_step_time": 10.214599370956421
    },
    {
      "epoch": 2.8106689453125e-05,
      "model_forward_time": 3.135467052459717,
      "step": 7368
    },
    {
      "epoch": 2.8106689453125e-05,
      "step": 7368,
      "training_step_time": 9.904662132263184
    },
    {
      "epoch": 2.8106689453125e-05,
      "model_forward_time": 2.430833578109741,
      "step": 7368
    },
    {
      "epoch": 2.8106689453125e-05,
      "step": 7368,
      "training_step_time": 7.456584692001343
    },
    {
      "epoch": 2.8106689453125e-05,
      "model_forward_time": 3.136138677597046,
      "step": 7368
    },
    {
      "epoch": 2.8106689453125e-05,
      "step": 7368,
      "training_step_time": 10.011690616607666
    },
    {
      "epoch": 2.8110504150390624e-05,
      "model_forward_time": 3.1568267345428467,
      "step": 7369
    },
    {
      "epoch": 2.8110504150390624e-05,
      "step": 7369,
      "training_step_time": 9.928885221481323
    },
    {
      "epoch": 2.8110504150390624e-05,
      "model_forward_time": 3.1378836631774902,
      "step": 7369
    },
    {
      "epoch": 2.8110504150390624e-05,
      "step": 7369,
      "training_step_time": 9.914172410964966
    },
    {
      "epoch": 2.8110504150390624e-05,
      "model_forward_time": 3.4061691761016846,
      "step": 7369
    },
    {
      "epoch": 2.8110504150390624e-05,
      "step": 7369,
      "training_step_time": 10.17762041091919
    },
    {
      "epoch": 2.8110504150390624e-05,
      "model_forward_time": 2.436744451522827,
      "step": 7369
    },
    {
      "epoch": 2.8110504150390624e-05,
      "step": 7369,
      "training_step_time": 10.266748428344727
    },
    {
      "epoch": 2.811431884765625e-05,
      "grad_norm": 0.26331740617752075,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 7370
    },
    {
      "epoch": 2.811431884765625e-05,
      "model_forward_time": 3.180155038833618,
      "step": 7370
    },
    {
      "dynamics_loss_avg": 0.054426584020257,
      "epoch": 2.811431884765625e-05,
      "step": 7370
    },
    {
      "action_loss_avg": 0.019365843338891863,
      "epoch": 2.811431884765625e-05,
      "step": 7370
    },
    {
      "epoch": 2.811431884765625e-05,
      "step": 7370,
      "training_step_time": 9.958920240402222
    },
    {
      "epoch": 2.811431884765625e-05,
      "model_forward_time": 2.440220355987549,
      "step": 7370
    },
    {
      "epoch": 2.811431884765625e-05,
      "step": 7370,
      "training_step_time": 7.468700885772705
    },
    {
      "epoch": 2.811431884765625e-05,
      "model_forward_time": 3.145228147506714,
      "step": 7370
    },
    {
      "epoch": 2.811431884765625e-05,
      "step": 7370,
      "training_step_time": 9.936474561691284
    },
    {
      "epoch": 2.811431884765625e-05,
      "model_forward_time": 3.4146525859832764,
      "step": 7370
    },
    {
      "epoch": 2.811431884765625e-05,
      "step": 7370,
      "training_step_time": 10.287557125091553
    },
    {
      "epoch": 2.8118133544921877e-05,
      "model_forward_time": 2.452052116394043,
      "step": 7371
    },
    {
      "epoch": 2.8118133544921877e-05,
      "step": 7371,
      "training_step_time": 7.479917287826538
    },
    {
      "epoch": 2.8118133544921877e-05,
      "model_forward_time": 2.4326419830322266,
      "step": 7371
    },
    {
      "epoch": 2.8118133544921877e-05,
      "step": 7371,
      "training_step_time": 7.458461761474609
    },
    {
      "epoch": 2.8118133544921877e-05,
      "model_forward_time": 2.434415102005005,
      "step": 7371
    },
    {
      "epoch": 2.8118133544921877e-05,
      "step": 7371,
      "training_step_time": 7.460490465164185
    },
    {
      "epoch": 2.8118133544921877e-05,
      "model_forward_time": 3.136059284210205,
      "step": 7371
    },
    {
      "epoch": 2.8118133544921877e-05,
      "step": 7371,
      "training_step_time": 10.318451166152954
    },
    {
      "epoch": 2.81219482421875e-05,
      "model_forward_time": 3.422389030456543,
      "step": 7372
    },
    {
      "epoch": 2.81219482421875e-05,
      "step": 7372,
      "training_step_time": 10.191078424453735
    },
    {
      "epoch": 2.81219482421875e-05,
      "model_forward_time": 3.1363744735717773,
      "step": 7372
    },
    {
      "epoch": 2.81219482421875e-05,
      "step": 7372,
      "training_step_time": 9.900858163833618
    },
    {
      "epoch": 2.81219482421875e-05,
      "model_forward_time": 2.432354688644409,
      "step": 7372
    },
    {
      "dynamics_loss_avg": 0.05710648829117417,
      "epoch": 2.81219482421875e-05,
      "step": 7372
    },
    {
      "action_loss_avg": 0.030514504155144097,
      "epoch": 2.81219482421875e-05,
      "step": 7372
    },
    {
      "epoch": 2.81219482421875e-05,
      "step": 7372,
      "training_step_time": 7.457702398300171
    },
    {
      "epoch": 2.81219482421875e-05,
      "model_forward_time": 3.1314408779144287,
      "step": 7372
    },
    {
      "epoch": 2.81219482421875e-05,
      "step": 7372,
      "training_step_time": 10.244206190109253
    },
    {
      "epoch": 2.8125762939453126e-05,
      "model_forward_time": 3.146897077560425,
      "step": 7373
    },
    {
      "epoch": 2.8125762939453126e-05,
      "step": 7373,
      "training_step_time": 9.910816431045532
    },
    {
      "epoch": 2.8125762939453126e-05,
      "model_forward_time": 3.384087562561035,
      "step": 7373
    },
    {
      "epoch": 2.8125762939453126e-05,
      "step": 7373,
      "training_step_time": 10.145923852920532
    },
    {
      "epoch": 2.8125762939453126e-05,
      "model_forward_time": 3.1364474296569824,
      "step": 7373
    },
    {
      "epoch": 2.8125762939453126e-05,
      "step": 7373,
      "training_step_time": 9.90071153640747
    },
    {
      "epoch": 2.8125762939453126e-05,
      "model_forward_time": 3.130453109741211,
      "step": 7373
    },
    {
      "epoch": 2.8125762939453126e-05,
      "step": 7373,
      "training_step_time": 10.261589765548706
    },
    {
      "epoch": 2.812957763671875e-05,
      "model_forward_time": 3.1487984657287598,
      "step": 7374
    },
    {
      "epoch": 2.812957763671875e-05,
      "step": 7374,
      "training_step_time": 9.913809299468994
    },
    {
      "epoch": 2.812957763671875e-05,
      "model_forward_time": 2.4320404529571533,
      "step": 7374
    },
    {
      "epoch": 2.812957763671875e-05,
      "step": 7374,
      "training_step_time": 7.453955888748169
    },
    {
      "epoch": 2.812957763671875e-05,
      "model_forward_time": 3.40644907951355,
      "step": 7374
    },
    {
      "epoch": 2.812957763671875e-05,
      "step": 7374,
      "training_step_time": 10.16769528388977
    },
    {
      "epoch": 2.812957763671875e-05,
      "model_forward_time": 3.1319284439086914,
      "step": 7374
    },
    {
      "epoch": 2.812957763671875e-05,
      "step": 7374,
      "training_step_time": 10.312324523925781
    },
    {
      "epoch": 2.8133392333984375e-05,
      "model_forward_time": 1.7650411128997803,
      "step": 7375
    },
    {
      "dynamics_loss_avg": 0.038836523750796915,
      "epoch": 2.8133392333984375e-05,
      "step": 7375
    },
    {
      "action_loss_avg": 0.01714122488629073,
      "epoch": 2.8133392333984375e-05,
      "step": 7375
    },
    {
      "epoch": 2.8133392333984375e-05,
      "step": 7375,
      "training_step_time": 5.215203523635864
    },
    {
      "epoch": 2.8133392333984375e-05,
      "model_forward_time": 3.1283164024353027,
      "step": 7375
    },
    {
      "epoch": 2.8133392333984375e-05,
      "step": 7375,
      "training_step_time": 9.891997814178467
    },
    {
      "epoch": 2.8133392333984375e-05,
      "model_forward_time": 3.1321706771850586,
      "step": 7375
    },
    {
      "epoch": 2.8133392333984375e-05,
      "step": 7375,
      "training_step_time": 9.896551132202148
    },
    {
      "epoch": 2.8133392333984375e-05,
      "model_forward_time": 3.3949294090270996,
      "step": 7375
    },
    {
      "epoch": 2.8133392333984375e-05,
      "step": 7375,
      "training_step_time": 10.273567914962769
    },
    {
      "epoch": 2.813720703125e-05,
      "model_forward_time": 1.7617783546447754,
      "step": 7376
    },
    {
      "epoch": 2.813720703125e-05,
      "step": 7376,
      "training_step_time": 5.21588659286499
    },
    {
      "epoch": 2.813720703125e-05,
      "model_forward_time": 2.430846929550171,
      "step": 7376
    },
    {
      "epoch": 2.813720703125e-05,
      "step": 7376,
      "training_step_time": 7.456218719482422
    },
    {
      "epoch": 2.813720703125e-05,
      "model_forward_time": 2.4313955307006836,
      "step": 7376
    },
    {
      "epoch": 2.813720703125e-05,
      "step": 7376,
      "training_step_time": 7.457252025604248
    },
    {
      "epoch": 2.813720703125e-05,
      "model_forward_time": 3.136993885040283,
      "step": 7376
    },
    {
      "epoch": 2.813720703125e-05,
      "step": 7376,
      "training_step_time": 10.281548500061035
    },
    {
      "epoch": 2.8141021728515624e-05,
      "model_forward_time": 2.4431307315826416,
      "step": 7377
    },
    {
      "epoch": 2.8141021728515624e-05,
      "step": 7377,
      "training_step_time": 7.725029468536377
    },
    {
      "epoch": 2.8141021728515624e-05,
      "model_forward_time": 3.1371524333953857,
      "step": 7377
    },
    {
      "epoch": 2.8141021728515624e-05,
      "step": 7377,
      "training_step_time": 9.903561353683472
    },
    {
      "epoch": 2.8141021728515624e-05,
      "model_forward_time": 2.4354147911071777,
      "step": 7377
    },
    {
      "dynamics_loss_avg": 0.058876197785139084,
      "epoch": 2.8141021728515624e-05,
      "step": 7377
    },
    {
      "action_loss_avg": 0.025141636189073325,
      "epoch": 2.8141021728515624e-05,
      "step": 7377
    },
    {
      "epoch": 2.8141021728515624e-05,
      "step": 7377,
      "training_step_time": 7.457662343978882
    },
    {
      "epoch": 2.8141021728515624e-05,
      "model_forward_time": 3.1370599269866943,
      "step": 7377
    },
    {
      "epoch": 2.8141021728515624e-05,
      "step": 7377,
      "training_step_time": 10.034681797027588
    },
    {
      "epoch": 2.814483642578125e-05,
      "model_forward_time": 2.4598171710968018,
      "step": 7378
    },
    {
      "epoch": 2.814483642578125e-05,
      "step": 7378,
      "training_step_time": 7.4911048412323
    },
    {
      "epoch": 2.814483642578125e-05,
      "model_forward_time": 2.4310519695281982,
      "step": 7378
    },
    {
      "epoch": 2.814483642578125e-05,
      "step": 7378,
      "training_step_time": 7.4583823680877686
    },
    {
      "epoch": 2.814483642578125e-05,
      "model_forward_time": 2.020534038543701,
      "step": 7378
    },
    {
      "epoch": 2.814483642578125e-05,
      "step": 7378,
      "training_step_time": 5.4785730838775635
    },
    {
      "epoch": 2.814483642578125e-05,
      "model_forward_time": 1.74153733253479,
      "step": 7378
    },
    {
      "epoch": 2.814483642578125e-05,
      "step": 7378,
      "training_step_time": 10.309406280517578
    },
    {
      "epoch": 2.8148651123046873e-05,
      "model_forward_time": 3.1681079864501953,
      "step": 7379
    },
    {
      "epoch": 2.8148651123046873e-05,
      "step": 7379,
      "training_step_time": 9.935035467147827
    },
    {
      "epoch": 2.8148651123046873e-05,
      "model_forward_time": 3.136430025100708,
      "step": 7379
    },
    {
      "epoch": 2.8148651123046873e-05,
      "step": 7379,
      "training_step_time": 9.907753944396973
    },
    {
      "epoch": 2.8148651123046873e-05,
      "model_forward_time": 3.1393022537231445,
      "step": 7379
    },
    {
      "epoch": 2.8148651123046873e-05,
      "step": 7379,
      "training_step_time": 9.909572124481201
    },
    {
      "epoch": 2.8148651123046873e-05,
      "model_forward_time": 3.1349802017211914,
      "step": 7379
    },
    {
      "epoch": 2.8148651123046873e-05,
      "step": 7379,
      "training_step_time": 10.306545495986938
    },
    {
      "epoch": 2.81524658203125e-05,
      "grad_norm": 0.49587371945381165,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7380
    },
    {
      "epoch": 2.81524658203125e-05,
      "model_forward_time": 3.153477191925049,
      "step": 7380
    },
    {
      "dynamics_loss_avg": 0.06524884086102248,
      "epoch": 2.81524658203125e-05,
      "step": 7380
    },
    {
      "action_loss_avg": 0.033083084016107024,
      "epoch": 2.81524658203125e-05,
      "step": 7380
    },
    {
      "epoch": 2.81524658203125e-05,
      "step": 7380,
      "training_step_time": 9.926474571228027
    },
    {
      "epoch": 2.81524658203125e-05,
      "model_forward_time": 3.138826608657837,
      "step": 7380
    },
    {
      "epoch": 2.81524658203125e-05,
      "step": 7380,
      "training_step_time": 9.906781435012817
    },
    {
      "epoch": 2.81524658203125e-05,
      "model_forward_time": 3.1483113765716553,
      "step": 7380
    },
    {
      "epoch": 2.81524658203125e-05,
      "step": 7380,
      "training_step_time": 9.93566370010376
    },
    {
      "epoch": 2.81524658203125e-05,
      "model_forward_time": 3.1398098468780518,
      "step": 7380
    },
    {
      "epoch": 2.81524658203125e-05,
      "step": 7380,
      "training_step_time": 10.27388882637024
    },
    {
      "epoch": 2.8156280517578126e-05,
      "model_forward_time": 3.4415931701660156,
      "step": 7381
    },
    {
      "epoch": 2.8156280517578126e-05,
      "step": 7381,
      "training_step_time": 10.22154974937439
    },
    {
      "epoch": 2.8156280517578126e-05,
      "model_forward_time": 2.4358677864074707,
      "step": 7381
    },
    {
      "epoch": 2.8156280517578126e-05,
      "step": 7381,
      "training_step_time": 7.471579551696777
    },
    {
      "epoch": 2.8156280517578126e-05,
      "model_forward_time": 2.43428897857666,
      "step": 7381
    },
    {
      "epoch": 2.8156280517578126e-05,
      "step": 7381,
      "training_step_time": 7.466233968734741
    },
    {
      "epoch": 2.8156280517578126e-05,
      "model_forward_time": 2.431056022644043,
      "step": 7381
    },
    {
      "epoch": 2.8156280517578126e-05,
      "step": 7381,
      "training_step_time": 10.312981128692627
    },
    {
      "epoch": 2.816009521484375e-05,
      "model_forward_time": 1.7610478401184082,
      "step": 7382
    },
    {
      "epoch": 2.816009521484375e-05,
      "step": 7382,
      "training_step_time": 5.2221996784210205
    },
    {
      "epoch": 2.816009521484375e-05,
      "model_forward_time": 3.1352181434631348,
      "step": 7382
    },
    {
      "epoch": 2.816009521484375e-05,
      "step": 7382,
      "training_step_time": 9.903142929077148
    },
    {
      "epoch": 2.816009521484375e-05,
      "model_forward_time": 3.4178714752197266,
      "step": 7382
    },
    {
      "dynamics_loss_avg": 0.07309115901589394,
      "epoch": 2.816009521484375e-05,
      "step": 7382
    },
    {
      "action_loss_avg": 0.025447236187756062,
      "epoch": 2.816009521484375e-05,
      "step": 7382
    },
    {
      "epoch": 2.816009521484375e-05,
      "step": 7382,
      "training_step_time": 10.187323093414307
    },
    {
      "epoch": 2.816009521484375e-05,
      "model_forward_time": 2.432833433151245,
      "step": 7382
    },
    {
      "epoch": 2.816009521484375e-05,
      "step": 7382,
      "training_step_time": 10.266094207763672
    },
    {
      "epoch": 2.8163909912109375e-05,
      "model_forward_time": 2.4474732875823975,
      "step": 7383
    },
    {
      "epoch": 2.8163909912109375e-05,
      "step": 7383,
      "training_step_time": 7.471567630767822
    },
    {
      "epoch": 2.8163909912109375e-05,
      "model_forward_time": 3.1330690383911133,
      "step": 7383
    },
    {
      "epoch": 2.8163909912109375e-05,
      "step": 7383,
      "training_step_time": 9.900917768478394
    },
    {
      "epoch": 2.8163909912109375e-05,
      "model_forward_time": 2.4334874153137207,
      "step": 7383
    },
    {
      "epoch": 2.8163909912109375e-05,
      "step": 7383,
      "training_step_time": 7.452962636947632
    },
    {
      "epoch": 2.8163909912109375e-05,
      "model_forward_time": 3.1303482055664062,
      "step": 7383
    },
    {
      "epoch": 2.8163909912109375e-05,
      "step": 7383,
      "training_step_time": 10.009966611862183
    },
    {
      "epoch": 2.8167724609375e-05,
      "model_forward_time": 3.467571973800659,
      "step": 7384
    },
    {
      "epoch": 2.8167724609375e-05,
      "step": 7384,
      "training_step_time": 10.236332893371582
    },
    {
      "epoch": 2.8167724609375e-05,
      "model_forward_time": 3.1323301792144775,
      "step": 7384
    },
    {
      "epoch": 2.8167724609375e-05,
      "step": 7384,
      "training_step_time": 9.901994466781616
    },
    {
      "epoch": 2.8167724609375e-05,
      "model_forward_time": 3.1307201385498047,
      "step": 7384
    },
    {
      "epoch": 2.8167724609375e-05,
      "step": 7384,
      "training_step_time": 9.9031240940094
    },
    {
      "epoch": 2.8167724609375e-05,
      "model_forward_time": 3.1427316665649414,
      "step": 7384
    },
    {
      "epoch": 2.8167724609375e-05,
      "step": 7384,
      "training_step_time": 10.311492204666138
    },
    {
      "epoch": 2.8171539306640624e-05,
      "model_forward_time": 3.155827045440674,
      "step": 7385
    },
    {
      "dynamics_loss_avg": 0.07899390216916799,
      "epoch": 2.8171539306640624e-05,
      "step": 7385
    },
    {
      "action_loss_avg": 0.02664591884240508,
      "epoch": 2.8171539306640624e-05,
      "step": 7385
    },
    {
      "epoch": 2.8171539306640624e-05,
      "step": 7385,
      "training_step_time": 10.212724447250366
    },
    {
      "epoch": 2.8171539306640624e-05,
      "model_forward_time": 3.141636848449707,
      "step": 7385
    },
    {
      "epoch": 2.8171539306640624e-05,
      "step": 7385,
      "training_step_time": 9.918853759765625
    },
    {
      "epoch": 2.8171539306640624e-05,
      "model_forward_time": 2.4353861808776855,
      "step": 7385
    },
    {
      "epoch": 2.8171539306640624e-05,
      "step": 7385,
      "training_step_time": 7.469956636428833
    },
    {
      "epoch": 2.8171539306640624e-05,
      "model_forward_time": 2.43804669380188,
      "step": 7385
    },
    {
      "epoch": 2.8171539306640624e-05,
      "step": 7385,
      "training_step_time": 10.264398574829102
    },
    {
      "epoch": 2.817535400390625e-05,
      "model_forward_time": 3.160665512084961,
      "step": 7386
    },
    {
      "epoch": 2.817535400390625e-05,
      "step": 7386,
      "training_step_time": 9.944244384765625
    },
    {
      "epoch": 2.817535400390625e-05,
      "model_forward_time": 3.13638973236084,
      "step": 7386
    },
    {
      "epoch": 2.817535400390625e-05,
      "step": 7386,
      "training_step_time": 9.90752387046814
    },
    {
      "epoch": 2.817535400390625e-05,
      "model_forward_time": 2.0183446407318115,
      "step": 7386
    },
    {
      "epoch": 2.817535400390625e-05,
      "step": 7386,
      "training_step_time": 5.47608757019043
    },
    {
      "epoch": 2.817535400390625e-05,
      "model_forward_time": 3.1345620155334473,
      "step": 7386
    },
    {
      "epoch": 2.817535400390625e-05,
      "step": 7386,
      "training_step_time": 10.296000003814697
    },
    {
      "epoch": 2.8179168701171873e-05,
      "model_forward_time": 3.14200496673584,
      "step": 7387
    },
    {
      "epoch": 2.8179168701171873e-05,
      "step": 7387,
      "training_step_time": 9.914225578308105
    },
    {
      "epoch": 2.8179168701171873e-05,
      "model_forward_time": 2.4302010536193848,
      "step": 7387
    },
    {
      "epoch": 2.8179168701171873e-05,
      "step": 7387,
      "training_step_time": 7.4556872844696045
    },
    {
      "epoch": 2.8179168701171873e-05,
      "model_forward_time": 2.425048589706421,
      "step": 7387
    },
    {
      "dynamics_loss_avg": 0.054189902544021604,
      "epoch": 2.8179168701171873e-05,
      "step": 7387
    },
    {
      "action_loss_avg": 0.028149490617215632,
      "epoch": 2.8179168701171873e-05,
      "step": 7387
    },
    {
      "epoch": 2.8179168701171873e-05,
      "step": 7387,
      "training_step_time": 7.4561944007873535
    },
    {
      "epoch": 2.8179168701171873e-05,
      "model_forward_time": 3.4024484157562256,
      "step": 7387
    },
    {
      "epoch": 2.8179168701171873e-05,
      "step": 7387,
      "training_step_time": 10.285520792007446
    },
    {
      "epoch": 2.81829833984375e-05,
      "model_forward_time": 3.183040142059326,
      "step": 7388
    },
    {
      "epoch": 2.81829833984375e-05,
      "step": 7388,
      "training_step_time": 9.961119413375854
    },
    {
      "epoch": 2.81829833984375e-05,
      "model_forward_time": 3.1322789192199707,
      "step": 7388
    },
    {
      "epoch": 2.81829833984375e-05,
      "step": 7388,
      "training_step_time": 9.908271789550781
    },
    {
      "epoch": 2.81829833984375e-05,
      "model_forward_time": 3.1356101036071777,
      "step": 7388
    },
    {
      "epoch": 2.81829833984375e-05,
      "step": 7388,
      "training_step_time": 9.913263082504272
    },
    {
      "epoch": 2.81829833984375e-05,
      "model_forward_time": 3.1350762844085693,
      "step": 7388
    },
    {
      "epoch": 2.81829833984375e-05,
      "step": 7388,
      "training_step_time": 10.049047946929932
    },
    {
      "epoch": 2.8186798095703126e-05,
      "model_forward_time": 3.428683280944824,
      "step": 7389
    },
    {
      "epoch": 2.8186798095703126e-05,
      "step": 7389,
      "training_step_time": 10.210185766220093
    },
    {
      "epoch": 2.8186798095703126e-05,
      "model_forward_time": 3.1335806846618652,
      "step": 7389
    },
    {
      "epoch": 2.8186798095703126e-05,
      "step": 7389,
      "training_step_time": 9.909150838851929
    },
    {
      "epoch": 2.8186798095703126e-05,
      "model_forward_time": 2.432600736618042,
      "step": 7389
    },
    {
      "epoch": 2.8186798095703126e-05,
      "step": 7389,
      "training_step_time": 7.4628283977508545
    },
    {
      "epoch": 2.8186798095703126e-05,
      "model_forward_time": 1.7438938617706299,
      "step": 7389
    },
    {
      "epoch": 2.8186798095703126e-05,
      "step": 7389,
      "training_step_time": 10.298105239868164
    },
    {
      "epoch": 2.819061279296875e-05,
      "grad_norm": 0.496356725692749,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 7390
    },
    {
      "epoch": 2.819061279296875e-05,
      "model_forward_time": 2.4433159828186035,
      "step": 7390
    },
    {
      "dynamics_loss_avg": 0.06455003879964352,
      "epoch": 2.819061279296875e-05,
      "step": 7390
    },
    {
      "action_loss_avg": 0.031029343046247958,
      "epoch": 2.819061279296875e-05,
      "step": 7390
    },
    {
      "epoch": 2.819061279296875e-05,
      "step": 7390,
      "training_step_time": 7.472073316574097
    },
    {
      "epoch": 2.819061279296875e-05,
      "model_forward_time": 1.740321159362793,
      "step": 7390
    },
    {
      "epoch": 2.819061279296875e-05,
      "step": 7390,
      "training_step_time": 5.196113348007202
    },
    {
      "epoch": 2.819061279296875e-05,
      "model_forward_time": 2.6692872047424316,
      "step": 7390
    },
    {
      "epoch": 2.819061279296875e-05,
      "step": 7390,
      "training_step_time": 7.696251630783081
    },
    {
      "epoch": 2.819061279296875e-05,
      "model_forward_time": 3.128225326538086,
      "step": 7390
    },
    {
      "epoch": 2.819061279296875e-05,
      "step": 7390,
      "training_step_time": 10.289047956466675
    },
    {
      "epoch": 2.8194427490234375e-05,
      "model_forward_time": 2.4427406787872314,
      "step": 7391
    },
    {
      "epoch": 2.8194427490234375e-05,
      "step": 7391,
      "training_step_time": 7.472203731536865
    },
    {
      "epoch": 2.8194427490234375e-05,
      "model_forward_time": 3.1312623023986816,
      "step": 7391
    },
    {
      "epoch": 2.8194427490234375e-05,
      "step": 7391,
      "training_step_time": 9.90463137626648
    },
    {
      "epoch": 2.8194427490234375e-05,
      "model_forward_time": 3.1319375038146973,
      "step": 7391
    },
    {
      "epoch": 2.8194427490234375e-05,
      "step": 7391,
      "training_step_time": 9.899122476577759
    },
    {
      "epoch": 2.8194427490234375e-05,
      "model_forward_time": 3.3779373168945312,
      "step": 7391
    },
    {
      "epoch": 2.8194427490234375e-05,
      "step": 7391,
      "training_step_time": 10.287727117538452
    },
    {
      "epoch": 2.81982421875e-05,
      "model_forward_time": 2.442521095275879,
      "step": 7392
    },
    {
      "epoch": 2.81982421875e-05,
      "step": 7392,
      "training_step_time": 7.4685609340667725
    },
    {
      "epoch": 2.81982421875e-05,
      "model_forward_time": 1.7421469688415527,
      "step": 7392
    },
    {
      "epoch": 2.81982421875e-05,
      "step": 7392,
      "training_step_time": 5.1985626220703125
    },
    {
      "epoch": 2.81982421875e-05,
      "model_forward_time": 3.1270394325256348,
      "step": 7392
    },
    {
      "dynamics_loss_avg": 0.09022454768419266,
      "epoch": 2.81982421875e-05,
      "step": 7392
    },
    {
      "action_loss_avg": 0.03919923715293407,
      "epoch": 2.81982421875e-05,
      "step": 7392
    },
    {
      "epoch": 2.81982421875e-05,
      "step": 7392,
      "training_step_time": 9.902961015701294
    },
    {
      "epoch": 2.81982421875e-05,
      "model_forward_time": 2.4331445693969727,
      "step": 7392
    },
    {
      "epoch": 2.81982421875e-05,
      "step": 7392,
      "training_step_time": 10.26645040512085
    },
    {
      "epoch": 2.8202056884765624e-05,
      "model_forward_time": 2.4484376907348633,
      "step": 7393
    },
    {
      "epoch": 2.8202056884765624e-05,
      "step": 7393,
      "training_step_time": 7.482881307601929
    },
    {
      "epoch": 2.8202056884765624e-05,
      "model_forward_time": 3.406251907348633,
      "step": 7393
    },
    {
      "epoch": 2.8202056884765624e-05,
      "step": 7393,
      "training_step_time": 10.178337335586548
    },
    {
      "epoch": 2.8202056884765624e-05,
      "model_forward_time": 2.4284989833831787,
      "step": 7393
    },
    {
      "epoch": 2.8202056884765624e-05,
      "step": 7393,
      "training_step_time": 7.45145320892334
    },
    {
      "epoch": 2.8202056884765624e-05,
      "model_forward_time": 2.426027536392212,
      "step": 7393
    },
    {
      "epoch": 2.8202056884765624e-05,
      "step": 7393,
      "training_step_time": 10.013901472091675
    },
    {
      "epoch": 2.820587158203125e-05,
      "model_forward_time": 2.4417381286621094,
      "step": 7394
    },
    {
      "epoch": 2.820587158203125e-05,
      "step": 7394,
      "training_step_time": 7.464407682418823
    },
    {
      "epoch": 2.820587158203125e-05,
      "model_forward_time": 2.428306818008423,
      "step": 7394
    },
    {
      "epoch": 2.820587158203125e-05,
      "step": 7394,
      "training_step_time": 7.449769973754883
    },
    {
      "epoch": 2.820587158203125e-05,
      "model_forward_time": 2.7400896549224854,
      "step": 7394
    },
    {
      "epoch": 2.820587158203125e-05,
      "step": 7394,
      "training_step_time": 7.769232511520386
    },
    {
      "epoch": 2.820587158203125e-05,
      "model_forward_time": 2.426377058029175,
      "step": 7394
    },
    {
      "epoch": 2.820587158203125e-05,
      "step": 7394,
      "training_step_time": 10.285435914993286
    },
    {
      "epoch": 2.8209686279296873e-05,
      "model_forward_time": 1.7898354530334473,
      "step": 7395
    },
    {
      "dynamics_loss_avg": 0.061292259208858016,
      "epoch": 2.8209686279296873e-05,
      "step": 7395
    },
    {
      "action_loss_avg": 0.02354733063839376,
      "epoch": 2.8209686279296873e-05,
      "step": 7395
    },
    {
      "epoch": 2.8209686279296873e-05,
      "step": 7395,
      "training_step_time": 5.2499072551727295
    },
    {
      "epoch": 2.8209686279296873e-05,
      "model_forward_time": 3.1344094276428223,
      "step": 7395
    },
    {
      "epoch": 2.8209686279296873e-05,
      "step": 7395,
      "training_step_time": 9.902434587478638
    },
    {
      "epoch": 2.8209686279296873e-05,
      "model_forward_time": 3.143505096435547,
      "step": 7395
    },
    {
      "epoch": 2.8209686279296873e-05,
      "step": 7395,
      "training_step_time": 9.912884712219238
    },
    {
      "epoch": 2.8209686279296873e-05,
      "model_forward_time": 3.141089916229248,
      "step": 7395
    },
    {
      "epoch": 2.8209686279296873e-05,
      "step": 7395,
      "training_step_time": 10.295019626617432
    },
    {
      "epoch": 2.82135009765625e-05,
      "model_forward_time": 2.4682037830352783,
      "step": 7396
    },
    {
      "epoch": 2.82135009765625e-05,
      "step": 7396,
      "training_step_time": 7.504810333251953
    },
    {
      "epoch": 2.82135009765625e-05,
      "model_forward_time": 2.4322409629821777,
      "step": 7396
    },
    {
      "epoch": 2.82135009765625e-05,
      "step": 7396,
      "training_step_time": 7.454607963562012
    },
    {
      "epoch": 2.82135009765625e-05,
      "model_forward_time": 3.128671646118164,
      "step": 7396
    },
    {
      "epoch": 2.82135009765625e-05,
      "step": 7396,
      "training_step_time": 9.900564193725586
    },
    {
      "epoch": 2.82135009765625e-05,
      "model_forward_time": 3.1321377754211426,
      "step": 7396
    },
    {
      "epoch": 2.82135009765625e-05,
      "step": 7396,
      "training_step_time": 10.301773309707642
    },
    {
      "epoch": 2.8217315673828126e-05,
      "model_forward_time": 2.447713613510132,
      "step": 7397
    },
    {
      "epoch": 2.8217315673828126e-05,
      "step": 7397,
      "training_step_time": 7.4747302532196045
    },
    {
      "epoch": 2.8217315673828126e-05,
      "model_forward_time": 2.7047200202941895,
      "step": 7397
    },
    {
      "epoch": 2.8217315673828126e-05,
      "step": 7397,
      "training_step_time": 7.728008031845093
    },
    {
      "epoch": 2.8217315673828126e-05,
      "model_forward_time": 2.4275782108306885,
      "step": 7397
    },
    {
      "dynamics_loss_avg": 0.05244541075080633,
      "epoch": 2.8217315673828126e-05,
      "step": 7397
    },
    {
      "action_loss_avg": 0.022200262802653015,
      "epoch": 2.8217315673828126e-05,
      "step": 7397
    },
    {
      "epoch": 2.8217315673828126e-05,
      "step": 7397,
      "training_step_time": 7.455261468887329
    },
    {
      "epoch": 2.8217315673828126e-05,
      "model_forward_time": 2.428133726119995,
      "step": 7397
    },
    {
      "epoch": 2.8217315673828126e-05,
      "step": 7397,
      "training_step_time": 10.2868070602417
    },
    {
      "epoch": 2.822113037109375e-05,
      "model_forward_time": 2.444443941116333,
      "step": 7398
    },
    {
      "epoch": 2.822113037109375e-05,
      "step": 7398,
      "training_step_time": 7.472754955291748
    },
    {
      "epoch": 2.822113037109375e-05,
      "model_forward_time": 3.1274163722991943,
      "step": 7398
    },
    {
      "epoch": 2.822113037109375e-05,
      "step": 7398,
      "training_step_time": 9.896270036697388
    },
    {
      "epoch": 2.822113037109375e-05,
      "model_forward_time": 2.6973423957824707,
      "step": 7398
    },
    {
      "epoch": 2.822113037109375e-05,
      "step": 7398,
      "training_step_time": 7.721750497817993
    },
    {
      "epoch": 2.822113037109375e-05,
      "model_forward_time": 3.1281957626342773,
      "step": 7398
    },
    {
      "epoch": 2.822113037109375e-05,
      "step": 7398,
      "training_step_time": 10.015687227249146
    },
    {
      "epoch": 2.8224945068359375e-05,
      "model_forward_time": 2.4423000812530518,
      "step": 7399
    },
    {
      "epoch": 2.8224945068359375e-05,
      "step": 7399,
      "training_step_time": 7.468919038772583
    },
    {
      "epoch": 2.8224945068359375e-05,
      "model_forward_time": 3.1323680877685547,
      "step": 7399
    },
    {
      "epoch": 2.8224945068359375e-05,
      "step": 7399,
      "training_step_time": 9.902052879333496
    },
    {
      "epoch": 2.8224945068359375e-05,
      "model_forward_time": 3.1284170150756836,
      "step": 7399
    },
    {
      "epoch": 2.8224945068359375e-05,
      "step": 7399,
      "training_step_time": 9.902795314788818
    },
    {
      "epoch": 2.8224945068359375e-05,
      "model_forward_time": 3.3905580043792725,
      "step": 7399
    },
    {
      "epoch": 2.8224945068359375e-05,
      "step": 7399,
      "training_step_time": 10.295610904693604
    },
    {
      "epoch": 2.8228759765625e-05,
      "grad_norm": 0.38658034801483154,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 7400
    },
    {
      "epoch": 2.8228759765625e-05,
      "model_forward_time": 3.151737689971924,
      "step": 7400
    },
    {
      "dynamics_loss_avg": 0.052171111665666105,
      "epoch": 2.8228759765625e-05,
      "step": 7400
    },
    {
      "action_loss_avg": 0.017949448758736254,
      "epoch": 2.8228759765625e-05,
      "step": 7400
    },
    {
      "epoch": 2.8228759765625e-05,
      "step": 7400,
      "training_step_time": 9.930768966674805
    },
    {
      "epoch": 2.8228759765625e-05,
      "model_forward_time": 2.4350340366363525,
      "step": 7400
    },
    {
      "epoch": 2.8228759765625e-05,
      "step": 7400,
      "training_step_time": 7.467574119567871
    },
    {
      "epoch": 2.8228759765625e-05,
      "model_forward_time": 3.1293439865112305,
      "step": 7400
    },
    {
      "epoch": 2.8228759765625e-05,
      "step": 7400,
      "training_step_time": 9.905279874801636
    },
    {
      "epoch": 2.8228759765625e-05,
      "model_forward_time": 2.4295177459716797,
      "step": 7400
    },
    {
      "epoch": 2.8228759765625e-05,
      "step": 7400,
      "training_step_time": 10.305377721786499
    },
    {
      "epoch": 2.8232574462890624e-05,
      "model_forward_time": 3.143228769302368,
      "step": 7401
    },
    {
      "epoch": 2.8232574462890624e-05,
      "step": 7401,
      "training_step_time": 10.182913541793823
    },
    {
      "epoch": 2.8232574462890624e-05,
      "model_forward_time": 3.133180856704712,
      "step": 7401
    },
    {
      "epoch": 2.8232574462890624e-05,
      "step": 7401,
      "training_step_time": 9.906795978546143
    },
    {
      "epoch": 2.8232574462890624e-05,
      "model_forward_time": 3.1333746910095215,
      "step": 7401
    },
    {
      "epoch": 2.8232574462890624e-05,
      "step": 7401,
      "training_step_time": 9.911476850509644
    },
    {
      "epoch": 2.8232574462890624e-05,
      "model_forward_time": 2.4393222332000732,
      "step": 7401
    },
    {
      "epoch": 2.8232574462890624e-05,
      "step": 7401,
      "training_step_time": 10.278438806533813
    },
    {
      "epoch": 2.823638916015625e-05,
      "model_forward_time": 2.4493424892425537,
      "step": 7402
    },
    {
      "epoch": 2.823638916015625e-05,
      "step": 7402,
      "training_step_time": 7.4752748012542725
    },
    {
      "epoch": 2.823638916015625e-05,
      "model_forward_time": 3.13958740234375,
      "step": 7402
    },
    {
      "epoch": 2.823638916015625e-05,
      "step": 7402,
      "training_step_time": 9.914554357528687
    },
    {
      "epoch": 2.823638916015625e-05,
      "model_forward_time": 3.3905797004699707,
      "step": 7402
    },
    {
      "dynamics_loss_avg": 0.06110811596736312,
      "epoch": 2.823638916015625e-05,
      "step": 7402
    },
    {
      "action_loss_avg": 0.025142471073195338,
      "epoch": 2.823638916015625e-05,
      "step": 7402
    },
    {
      "epoch": 2.823638916015625e-05,
      "step": 7402,
      "training_step_time": 10.167205333709717
    },
    {
      "epoch": 2.823638916015625e-05,
      "model_forward_time": 3.149085760116577,
      "step": 7402
    },
    {
      "epoch": 2.823638916015625e-05,
      "step": 7402,
      "training_step_time": 10.243935823440552
    },
    {
      "epoch": 2.8240203857421873e-05,
      "model_forward_time": 3.165154218673706,
      "step": 7403
    },
    {
      "epoch": 2.8240203857421873e-05,
      "step": 7403,
      "training_step_time": 9.943172216415405
    },
    {
      "epoch": 2.8240203857421873e-05,
      "model_forward_time": 3.138688087463379,
      "step": 7403
    },
    {
      "epoch": 2.8240203857421873e-05,
      "step": 7403,
      "training_step_time": 9.912749528884888
    },
    {
      "epoch": 2.8240203857421873e-05,
      "model_forward_time": 2.434908390045166,
      "step": 7403
    },
    {
      "epoch": 2.8240203857421873e-05,
      "step": 7403,
      "training_step_time": 7.733461618423462
    },
    {
      "epoch": 2.8240203857421873e-05,
      "model_forward_time": 3.1392056941986084,
      "step": 7403
    },
    {
      "epoch": 2.8240203857421873e-05,
      "step": 7403,
      "training_step_time": 10.002838611602783
    },
    {
      "epoch": 2.82440185546875e-05,
      "model_forward_time": 3.151484966278076,
      "step": 7404
    },
    {
      "epoch": 2.82440185546875e-05,
      "step": 7404,
      "training_step_time": 9.923322916030884
    },
    {
      "epoch": 2.82440185546875e-05,
      "model_forward_time": 2.4370217323303223,
      "step": 7404
    },
    {
      "epoch": 2.82440185546875e-05,
      "step": 7404,
      "training_step_time": 7.470096111297607
    },
    {
      "epoch": 2.82440185546875e-05,
      "model_forward_time": 3.1421101093292236,
      "step": 7404
    },
    {
      "epoch": 2.82440185546875e-05,
      "step": 7404,
      "training_step_time": 9.908414125442505
    },
    {
      "epoch": 2.82440185546875e-05,
      "model_forward_time": 2.4366793632507324,
      "step": 7404
    },
    {
      "epoch": 2.82440185546875e-05,
      "step": 7404,
      "training_step_time": 10.326566934585571
    },
    {
      "epoch": 2.8247833251953126e-05,
      "model_forward_time": 3.4677789211273193,
      "step": 7405
    },
    {
      "dynamics_loss_avg": 0.04613186251372099,
      "epoch": 2.8247833251953126e-05,
      "step": 7405
    },
    {
      "action_loss_avg": 0.024665877129882573,
      "epoch": 2.8247833251953126e-05,
      "step": 7405
    },
    {
      "epoch": 2.8247833251953126e-05,
      "step": 7405,
      "training_step_time": 10.253599405288696
    },
    {
      "epoch": 2.8247833251953126e-05,
      "model_forward_time": 3.1399569511413574,
      "step": 7405
    },
    {
      "epoch": 2.8247833251953126e-05,
      "step": 7405,
      "training_step_time": 9.908170223236084
    },
    {
      "epoch": 2.8247833251953126e-05,
      "model_forward_time": 2.436880350112915,
      "step": 7405
    },
    {
      "epoch": 2.8247833251953126e-05,
      "step": 7405,
      "training_step_time": 7.4604103565216064
    },
    {
      "epoch": 2.8247833251953126e-05,
      "model_forward_time": 2.434948682785034,
      "step": 7405
    },
    {
      "epoch": 2.8247833251953126e-05,
      "step": 7405,
      "training_step_time": 10.263438940048218
    },
    {
      "epoch": 2.825164794921875e-05,
      "model_forward_time": 2.481956720352173,
      "step": 7406
    },
    {
      "epoch": 2.825164794921875e-05,
      "step": 7406,
      "training_step_time": 7.503811359405518
    },
    {
      "epoch": 2.825164794921875e-05,
      "model_forward_time": 3.3882250785827637,
      "step": 7406
    },
    {
      "epoch": 2.825164794921875e-05,
      "step": 7406,
      "training_step_time": 10.147773504257202
    },
    {
      "epoch": 2.825164794921875e-05,
      "model_forward_time": 3.1368765830993652,
      "step": 7406
    },
    {
      "epoch": 2.825164794921875e-05,
      "step": 7406,
      "training_step_time": 9.899552345275879
    },
    {
      "epoch": 2.825164794921875e-05,
      "model_forward_time": 3.1367385387420654,
      "step": 7406
    },
    {
      "epoch": 2.825164794921875e-05,
      "step": 7406,
      "training_step_time": 10.271859169006348
    },
    {
      "epoch": 2.8255462646484375e-05,
      "model_forward_time": 2.4499359130859375,
      "step": 7407
    },
    {
      "epoch": 2.8255462646484375e-05,
      "step": 7407,
      "training_step_time": 7.471853256225586
    },
    {
      "epoch": 2.8255462646484375e-05,
      "model_forward_time": 2.4340696334838867,
      "step": 7407
    },
    {
      "epoch": 2.8255462646484375e-05,
      "step": 7407,
      "training_step_time": 7.45146107673645
    },
    {
      "epoch": 2.8255462646484375e-05,
      "model_forward_time": 2.711229085922241,
      "step": 7407
    },
    {
      "dynamics_loss_avg": 0.059008494950830936,
      "epoch": 2.8255462646484375e-05,
      "step": 7407
    },
    {
      "action_loss_avg": 0.030097836256027223,
      "epoch": 2.8255462646484375e-05,
      "step": 7407
    },
    {
      "epoch": 2.8255462646484375e-05,
      "step": 7407,
      "training_step_time": 7.730740547180176
    },
    {
      "epoch": 2.8255462646484375e-05,
      "model_forward_time": 2.426184892654419,
      "step": 7407
    },
    {
      "epoch": 2.8255462646484375e-05,
      "step": 7407,
      "training_step_time": 10.249891519546509
    },
    {
      "epoch": 2.825927734375e-05,
      "model_forward_time": 2.4501473903656006,
      "step": 7408
    },
    {
      "epoch": 2.825927734375e-05,
      "step": 7408,
      "training_step_time": 7.469354152679443
    },
    {
      "epoch": 2.825927734375e-05,
      "model_forward_time": 2.4304165840148926,
      "step": 7408
    },
    {
      "epoch": 2.825927734375e-05,
      "step": 7408,
      "training_step_time": 7.4461469650268555
    },
    {
      "epoch": 2.825927734375e-05,
      "model_forward_time": 2.4315948486328125,
      "step": 7408
    },
    {
      "epoch": 2.825927734375e-05,
      "step": 7408,
      "training_step_time": 7.4487340450286865
    },
    {
      "epoch": 2.825927734375e-05,
      "model_forward_time": 2.4281651973724365,
      "step": 7408
    },
    {
      "epoch": 2.825927734375e-05,
      "step": 7408,
      "training_step_time": 10.324612855911255
    },
    {
      "epoch": 2.8263092041015624e-05,
      "model_forward_time": 2.7991302013397217,
      "step": 7409
    },
    {
      "epoch": 2.8263092041015624e-05,
      "step": 7409,
      "training_step_time": 7.818113803863525
    },
    {
      "epoch": 2.8263092041015624e-05,
      "model_forward_time": 2.431205987930298,
      "step": 7409
    },
    {
      "epoch": 2.8263092041015624e-05,
      "step": 7409,
      "training_step_time": 7.448809862136841
    },
    {
      "epoch": 2.8263092041015624e-05,
      "model_forward_time": 2.433981418609619,
      "step": 7409
    },
    {
      "epoch": 2.8263092041015624e-05,
      "step": 7409,
      "training_step_time": 7.450389385223389
    },
    {
      "epoch": 2.8263092041015624e-05,
      "model_forward_time": 3.1346287727355957,
      "step": 7409
    },
    {
      "epoch": 2.8263092041015624e-05,
      "step": 7409,
      "training_step_time": 10.279442548751831
    },
    {
      "epoch": 2.826690673828125e-05,
      "grad_norm": 0.30305421352386475,
      "learning_rate": 1e-05,
      "loss": 0.0829,
      "step": 7410
    },
    {
      "epoch": 2.826690673828125e-05,
      "model_forward_time": 3.1486496925354004,
      "step": 7410
    },
    {
      "dynamics_loss_avg": 0.058667571470141414,
      "epoch": 2.826690673828125e-05,
      "step": 7410
    },
    {
      "action_loss_avg": 0.023598104994744063,
      "epoch": 2.826690673828125e-05,
      "step": 7410
    },
    {
      "epoch": 2.826690673828125e-05,
      "step": 7410,
      "training_step_time": 9.913603782653809
    },
    {
      "epoch": 2.826690673828125e-05,
      "model_forward_time": 3.4169836044311523,
      "step": 7410
    },
    {
      "epoch": 2.826690673828125e-05,
      "step": 7410,
      "training_step_time": 10.187891483306885
    },
    {
      "epoch": 2.826690673828125e-05,
      "model_forward_time": 2.438890218734741,
      "step": 7410
    },
    {
      "epoch": 2.826690673828125e-05,
      "step": 7410,
      "training_step_time": 7.468419551849365
    },
    {
      "epoch": 2.826690673828125e-05,
      "model_forward_time": 3.1403985023498535,
      "step": 7410
    },
    {
      "epoch": 2.826690673828125e-05,
      "step": 7410,
      "training_step_time": 10.297256708145142
    },
    {
      "epoch": 2.8270721435546873e-05,
      "model_forward_time": 2.44783353805542,
      "step": 7411
    },
    {
      "epoch": 2.8270721435546873e-05,
      "step": 7411,
      "training_step_time": 7.4767186641693115
    },
    {
      "epoch": 2.8270721435546873e-05,
      "model_forward_time": 3.1327991485595703,
      "step": 7411
    },
    {
      "epoch": 2.8270721435546873e-05,
      "step": 7411,
      "training_step_time": 9.907013893127441
    },
    {
      "epoch": 2.8270721435546873e-05,
      "model_forward_time": 3.406198263168335,
      "step": 7411
    },
    {
      "epoch": 2.8270721435546873e-05,
      "step": 7411,
      "training_step_time": 10.17431116104126
    },
    {
      "epoch": 2.8270721435546873e-05,
      "model_forward_time": 2.437875986099243,
      "step": 7411
    },
    {
      "epoch": 2.8270721435546873e-05,
      "step": 7411,
      "training_step_time": 10.29405927658081
    },
    {
      "epoch": 2.82745361328125e-05,
      "model_forward_time": 3.157309055328369,
      "step": 7412
    },
    {
      "epoch": 2.82745361328125e-05,
      "step": 7412,
      "training_step_time": 9.928012609481812
    },
    {
      "epoch": 2.82745361328125e-05,
      "model_forward_time": 3.1401519775390625,
      "step": 7412
    },
    {
      "epoch": 2.82745361328125e-05,
      "step": 7412,
      "training_step_time": 9.907248973846436
    },
    {
      "epoch": 2.82745361328125e-05,
      "model_forward_time": 2.4353761672973633,
      "step": 7412
    },
    {
      "dynamics_loss_avg": 0.07116251755505801,
      "epoch": 2.82745361328125e-05,
      "step": 7412
    },
    {
      "action_loss_avg": 0.02991495979949832,
      "epoch": 2.82745361328125e-05,
      "step": 7412
    },
    {
      "epoch": 2.82745361328125e-05,
      "step": 7412,
      "training_step_time": 7.461137056350708
    },
    {
      "epoch": 2.82745361328125e-05,
      "model_forward_time": 3.4266300201416016,
      "step": 7412
    },
    {
      "epoch": 2.82745361328125e-05,
      "step": 7412,
      "training_step_time": 10.283891677856445
    },
    {
      "epoch": 2.8278350830078126e-05,
      "model_forward_time": 2.447472095489502,
      "step": 7413
    },
    {
      "epoch": 2.8278350830078126e-05,
      "step": 7413,
      "training_step_time": 7.4677300453186035
    },
    {
      "epoch": 2.8278350830078126e-05,
      "model_forward_time": 2.4293291568756104,
      "step": 7413
    },
    {
      "epoch": 2.8278350830078126e-05,
      "step": 7413,
      "training_step_time": 7.446352481842041
    },
    {
      "epoch": 2.8278350830078126e-05,
      "model_forward_time": 3.1362383365631104,
      "step": 7413
    },
    {
      "epoch": 2.8278350830078126e-05,
      "step": 7413,
      "training_step_time": 9.896660804748535
    },
    {
      "epoch": 2.8278350830078126e-05,
      "model_forward_time": 2.4332590103149414,
      "step": 7413
    },
    {
      "epoch": 2.8278350830078126e-05,
      "step": 7413,
      "training_step_time": 10.006771564483643
    },
    {
      "epoch": 2.828216552734375e-05,
      "model_forward_time": 3.1409661769866943,
      "step": 7414
    },
    {
      "epoch": 2.828216552734375e-05,
      "step": 7414,
      "training_step_time": 9.902301549911499
    },
    {
      "epoch": 2.828216552734375e-05,
      "model_forward_time": 3.383906364440918,
      "step": 7414
    },
    {
      "epoch": 2.828216552734375e-05,
      "step": 7414,
      "training_step_time": 10.147024393081665
    },
    {
      "epoch": 2.828216552734375e-05,
      "model_forward_time": 3.139073133468628,
      "step": 7414
    },
    {
      "epoch": 2.828216552734375e-05,
      "step": 7414,
      "training_step_time": 9.906522035598755
    },
    {
      "epoch": 2.828216552734375e-05,
      "model_forward_time": 3.1371750831604004,
      "step": 7414
    },
    {
      "epoch": 2.828216552734375e-05,
      "step": 7414,
      "training_step_time": 10.027300119400024
    },
    {
      "epoch": 2.8285980224609375e-05,
      "model_forward_time": 2.4545950889587402,
      "step": 7415
    },
    {
      "dynamics_loss_avg": 0.06097924094647169,
      "epoch": 2.8285980224609375e-05,
      "step": 7415
    },
    {
      "action_loss_avg": 0.022343394951894878,
      "epoch": 2.8285980224609375e-05,
      "step": 7415
    },
    {
      "epoch": 2.8285980224609375e-05,
      "step": 7415,
      "training_step_time": 7.484158039093018
    },
    {
      "epoch": 2.8285980224609375e-05,
      "model_forward_time": 2.4343864917755127,
      "step": 7415
    },
    {
      "epoch": 2.8285980224609375e-05,
      "step": 7415,
      "training_step_time": 7.453965902328491
    },
    {
      "epoch": 2.8285980224609375e-05,
      "model_forward_time": 2.0212340354919434,
      "step": 7415
    },
    {
      "epoch": 2.8285980224609375e-05,
      "step": 7415,
      "training_step_time": 5.474236249923706
    },
    {
      "epoch": 2.8285980224609375e-05,
      "model_forward_time": 3.13108491897583,
      "step": 7415
    },
    {
      "epoch": 2.8285980224609375e-05,
      "step": 7415,
      "training_step_time": 10.018383026123047
    },
    {
      "epoch": 2.8289794921875e-05,
      "model_forward_time": 2.4448671340942383,
      "step": 7416
    },
    {
      "epoch": 2.8289794921875e-05,
      "step": 7416,
      "training_step_time": 7.464757680892944
    },
    {
      "epoch": 2.8289794921875e-05,
      "model_forward_time": 3.1348259449005127,
      "step": 7416
    },
    {
      "epoch": 2.8289794921875e-05,
      "step": 7416,
      "training_step_time": 9.90010404586792
    },
    {
      "epoch": 2.8289794921875e-05,
      "model_forward_time": 3.1355249881744385,
      "step": 7416
    },
    {
      "epoch": 2.8289794921875e-05,
      "step": 7416,
      "training_step_time": 9.898170471191406
    },
    {
      "epoch": 2.8289794921875e-05,
      "model_forward_time": 2.436879873275757,
      "step": 7416
    },
    {
      "epoch": 2.8289794921875e-05,
      "step": 7416,
      "training_step_time": 10.296099424362183
    },
    {
      "epoch": 2.8293609619140624e-05,
      "model_forward_time": 3.143296241760254,
      "step": 7417
    },
    {
      "epoch": 2.8293609619140624e-05,
      "step": 7417,
      "training_step_time": 9.909388780593872
    },
    {
      "epoch": 2.8293609619140624e-05,
      "model_forward_time": 2.4345362186431885,
      "step": 7417
    },
    {
      "epoch": 2.8293609619140624e-05,
      "step": 7417,
      "training_step_time": 7.456683874130249
    },
    {
      "epoch": 2.8293609619140624e-05,
      "model_forward_time": 3.1302623748779297,
      "step": 7417
    },
    {
      "dynamics_loss_avg": 0.07789266444742679,
      "epoch": 2.8293609619140624e-05,
      "step": 7417
    },
    {
      "action_loss_avg": 0.02963744830340147,
      "epoch": 2.8293609619140624e-05,
      "step": 7417
    },
    {
      "epoch": 2.8293609619140624e-05,
      "step": 7417,
      "training_step_time": 9.893195390701294
    },
    {
      "epoch": 2.8293609619140624e-05,
      "model_forward_time": 3.1370232105255127,
      "step": 7417
    },
    {
      "epoch": 2.8293609619140624e-05,
      "step": 7417,
      "training_step_time": 10.270424604415894
    },
    {
      "epoch": 2.829742431640625e-05,
      "model_forward_time": 3.4149842262268066,
      "step": 7418
    },
    {
      "epoch": 2.829742431640625e-05,
      "step": 7418,
      "training_step_time": 10.174243211746216
    },
    {
      "epoch": 2.829742431640625e-05,
      "model_forward_time": 3.133330821990967,
      "step": 7418
    },
    {
      "epoch": 2.829742431640625e-05,
      "step": 7418,
      "training_step_time": 9.898130655288696
    },
    {
      "epoch": 2.829742431640625e-05,
      "model_forward_time": 3.135099172592163,
      "step": 7418
    },
    {
      "epoch": 2.829742431640625e-05,
      "step": 7418,
      "training_step_time": 9.899004220962524
    },
    {
      "epoch": 2.829742431640625e-05,
      "model_forward_time": 2.4358437061309814,
      "step": 7418
    },
    {
      "epoch": 2.829742431640625e-05,
      "step": 7418,
      "training_step_time": 10.295464277267456
    },
    {
      "epoch": 2.8301239013671873e-05,
      "model_forward_time": 2.453057289123535,
      "step": 7419
    },
    {
      "epoch": 2.8301239013671873e-05,
      "step": 7419,
      "training_step_time": 7.474626779556274
    },
    {
      "epoch": 2.8301239013671873e-05,
      "model_forward_time": 2.431612968444824,
      "step": 7419
    },
    {
      "epoch": 2.8301239013671873e-05,
      "step": 7419,
      "training_step_time": 7.4552202224731445
    },
    {
      "epoch": 2.8301239013671873e-05,
      "model_forward_time": 2.689384698867798,
      "step": 7419
    },
    {
      "epoch": 2.8301239013671873e-05,
      "step": 7419,
      "training_step_time": 7.711355924606323
    },
    {
      "epoch": 2.8301239013671873e-05,
      "model_forward_time": 3.1359360218048096,
      "step": 7419
    },
    {
      "epoch": 2.8301239013671873e-05,
      "step": 7419,
      "training_step_time": 10.30270791053772
    },
    {
      "epoch": 2.83050537109375e-05,
      "grad_norm": 0.3127902150154114,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 7420
    },
    {
      "epoch": 2.83050537109375e-05,
      "model_forward_time": 2.448810577392578,
      "step": 7420
    },
    {
      "dynamics_loss_avg": 0.06127888709306717,
      "epoch": 2.83050537109375e-05,
      "step": 7420
    },
    {
      "action_loss_avg": 0.023003727942705155,
      "epoch": 2.83050537109375e-05,
      "step": 7420
    },
    {
      "epoch": 2.83050537109375e-05,
      "step": 7420,
      "training_step_time": 7.477672815322876
    },
    {
      "epoch": 2.83050537109375e-05,
      "model_forward_time": 3.1372928619384766,
      "step": 7420
    },
    {
      "epoch": 2.83050537109375e-05,
      "step": 7420,
      "training_step_time": 9.906548261642456
    },
    {
      "epoch": 2.83050537109375e-05,
      "model_forward_time": 2.437169313430786,
      "step": 7420
    },
    {
      "epoch": 2.83050537109375e-05,
      "step": 7420,
      "training_step_time": 7.468637228012085
    },
    {
      "epoch": 2.83050537109375e-05,
      "model_forward_time": 3.4161763191223145,
      "step": 7420
    },
    {
      "epoch": 2.83050537109375e-05,
      "step": 7420,
      "training_step_time": 10.282218217849731
    },
    {
      "epoch": 2.8308868408203126e-05,
      "model_forward_time": 3.1656973361968994,
      "step": 7421
    },
    {
      "epoch": 2.8308868408203126e-05,
      "step": 7421,
      "training_step_time": 9.940002918243408
    },
    {
      "epoch": 2.8308868408203126e-05,
      "model_forward_time": 3.143683433532715,
      "step": 7421
    },
    {
      "epoch": 2.8308868408203126e-05,
      "step": 7421,
      "training_step_time": 9.920444011688232
    },
    {
      "epoch": 2.8308868408203126e-05,
      "model_forward_time": 2.438810348510742,
      "step": 7421
    },
    {
      "epoch": 2.8308868408203126e-05,
      "step": 7421,
      "training_step_time": 7.47073221206665
    },
    {
      "epoch": 2.8308868408203126e-05,
      "model_forward_time": 2.438270092010498,
      "step": 7421
    },
    {
      "epoch": 2.8308868408203126e-05,
      "step": 7421,
      "training_step_time": 10.327887058258057
    },
    {
      "epoch": 2.831268310546875e-05,
      "model_forward_time": 3.151092767715454,
      "step": 7422
    },
    {
      "epoch": 2.831268310546875e-05,
      "step": 7422,
      "training_step_time": 10.197245121002197
    },
    {
      "epoch": 2.831268310546875e-05,
      "model_forward_time": 3.1382105350494385,
      "step": 7422
    },
    {
      "epoch": 2.831268310546875e-05,
      "step": 7422,
      "training_step_time": 9.908134698867798
    },
    {
      "epoch": 2.831268310546875e-05,
      "model_forward_time": 2.43805193901062,
      "step": 7422
    },
    {
      "dynamics_loss_avg": 0.06256841346621514,
      "epoch": 2.831268310546875e-05,
      "step": 7422
    },
    {
      "action_loss_avg": 0.024224281997885556,
      "epoch": 2.831268310546875e-05,
      "step": 7422
    },
    {
      "epoch": 2.831268310546875e-05,
      "step": 7422,
      "training_step_time": 7.469201326370239
    },
    {
      "epoch": 2.831268310546875e-05,
      "model_forward_time": 3.1399614810943604,
      "step": 7422
    },
    {
      "epoch": 2.831268310546875e-05,
      "step": 7422,
      "training_step_time": 10.29734468460083
    },
    {
      "epoch": 2.8316497802734375e-05,
      "model_forward_time": 3.149839162826538,
      "step": 7423
    },
    {
      "epoch": 2.8316497802734375e-05,
      "step": 7423,
      "training_step_time": 9.921613931655884
    },
    {
      "epoch": 2.8316497802734375e-05,
      "model_forward_time": 3.403353691101074,
      "step": 7423
    },
    {
      "epoch": 2.8316497802734375e-05,
      "step": 7423,
      "training_step_time": 10.172402143478394
    },
    {
      "epoch": 2.8316497802734375e-05,
      "model_forward_time": 3.1363730430603027,
      "step": 7423
    },
    {
      "epoch": 2.8316497802734375e-05,
      "step": 7423,
      "training_step_time": 9.902122974395752
    },
    {
      "epoch": 2.8316497802734375e-05,
      "model_forward_time": 2.4343879222869873,
      "step": 7423
    },
    {
      "epoch": 2.8316497802734375e-05,
      "step": 7423,
      "training_step_time": 10.267465353012085
    },
    {
      "epoch": 2.83203125e-05,
      "model_forward_time": 1.760169506072998,
      "step": 7424
    },
    {
      "epoch": 2.83203125e-05,
      "step": 7424,
      "training_step_time": 5.213061332702637
    },
    {
      "epoch": 2.83203125e-05,
      "model_forward_time": 3.137479782104492,
      "step": 7424
    },
    {
      "epoch": 2.83203125e-05,
      "step": 7424,
      "training_step_time": 9.90368938446045
    },
    {
      "epoch": 2.83203125e-05,
      "model_forward_time": 3.1340150833129883,
      "step": 7424
    },
    {
      "epoch": 2.83203125e-05,
      "step": 7424,
      "training_step_time": 9.902284622192383
    },
    {
      "epoch": 2.83203125e-05,
      "model_forward_time": 2.71772837638855,
      "step": 7424
    },
    {
      "epoch": 2.83203125e-05,
      "step": 7424,
      "training_step_time": 10.29349136352539
    },
    {
      "epoch": 2.8324127197265624e-05,
      "model_forward_time": 1.7635903358459473,
      "step": 7425
    },
    {
      "dynamics_loss_avg": 0.057994982413947585,
      "epoch": 2.8324127197265624e-05,
      "step": 7425
    },
    {
      "action_loss_avg": 0.029249406047165395,
      "epoch": 2.8324127197265624e-05,
      "step": 7425
    },
    {
      "epoch": 2.8324127197265624e-05,
      "step": 7425,
      "training_step_time": 5.2222816944122314
    },
    {
      "epoch": 2.8324127197265624e-05,
      "model_forward_time": 2.43229079246521,
      "step": 7425
    },
    {
      "epoch": 2.8324127197265624e-05,
      "step": 7425,
      "training_step_time": 7.459988594055176
    },
    {
      "epoch": 2.8324127197265624e-05,
      "model_forward_time": 3.1348817348480225,
      "step": 7425
    },
    {
      "epoch": 2.8324127197265624e-05,
      "step": 7425,
      "training_step_time": 9.90316653251648
    },
    {
      "epoch": 2.8324127197265624e-05,
      "model_forward_time": 2.436537504196167,
      "step": 7425
    },
    {
      "epoch": 2.8324127197265624e-05,
      "step": 7425,
      "training_step_time": 10.017737627029419
    },
    {
      "epoch": 2.832794189453125e-05,
      "model_forward_time": 3.152730941772461,
      "step": 7426
    },
    {
      "epoch": 2.832794189453125e-05,
      "step": 7426,
      "training_step_time": 9.923838138580322
    },
    {
      "epoch": 2.832794189453125e-05,
      "model_forward_time": 3.4119584560394287,
      "step": 7426
    },
    {
      "epoch": 2.832794189453125e-05,
      "step": 7426,
      "training_step_time": 10.180838346481323
    },
    {
      "epoch": 2.832794189453125e-05,
      "model_forward_time": 2.4366323947906494,
      "step": 7426
    },
    {
      "epoch": 2.832794189453125e-05,
      "step": 7426,
      "training_step_time": 7.462038278579712
    },
    {
      "epoch": 2.832794189453125e-05,
      "model_forward_time": 3.140587329864502,
      "step": 7426
    },
    {
      "epoch": 2.832794189453125e-05,
      "step": 7426,
      "training_step_time": 10.288418769836426
    },
    {
      "epoch": 2.8331756591796873e-05,
      "model_forward_time": 2.450955867767334,
      "step": 7427
    },
    {
      "epoch": 2.8331756591796873e-05,
      "step": 7427,
      "training_step_time": 7.472055435180664
    },
    {
      "epoch": 2.8331756591796873e-05,
      "model_forward_time": 3.137361526489258,
      "step": 7427
    },
    {
      "epoch": 2.8331756591796873e-05,
      "step": 7427,
      "training_step_time": 9.905841588973999
    },
    {
      "epoch": 2.8331756591796873e-05,
      "model_forward_time": 3.4170877933502197,
      "step": 7427
    },
    {
      "dynamics_loss_avg": 0.05825069714337587,
      "epoch": 2.8331756591796873e-05,
      "step": 7427
    },
    {
      "action_loss_avg": 0.02560917576774955,
      "epoch": 2.8331756591796873e-05,
      "step": 7427
    },
    {
      "epoch": 2.8331756591796873e-05,
      "step": 7427,
      "training_step_time": 10.185897827148438
    },
    {
      "epoch": 2.8331756591796873e-05,
      "model_forward_time": 2.436239719390869,
      "step": 7427
    },
    {
      "epoch": 2.8331756591796873e-05,
      "step": 7427,
      "training_step_time": 10.294149398803711
    },
    {
      "epoch": 2.83355712890625e-05,
      "model_forward_time": 2.4564032554626465,
      "step": 7428
    },
    {
      "epoch": 2.83355712890625e-05,
      "step": 7428,
      "training_step_time": 7.48114800453186
    },
    {
      "epoch": 2.83355712890625e-05,
      "model_forward_time": 3.133413553237915,
      "step": 7428
    },
    {
      "epoch": 2.83355712890625e-05,
      "step": 7428,
      "training_step_time": 9.899591207504272
    },
    {
      "epoch": 2.83355712890625e-05,
      "model_forward_time": 2.438824415206909,
      "step": 7428
    },
    {
      "epoch": 2.83355712890625e-05,
      "step": 7428,
      "training_step_time": 7.4580981731414795
    },
    {
      "epoch": 2.83355712890625e-05,
      "model_forward_time": 2.7159998416900635,
      "step": 7428
    },
    {
      "epoch": 2.83355712890625e-05,
      "step": 7428,
      "training_step_time": 10.303351879119873
    },
    {
      "epoch": 2.8339385986328126e-05,
      "model_forward_time": 3.160862445831299,
      "step": 7429
    },
    {
      "epoch": 2.8339385986328126e-05,
      "step": 7429,
      "training_step_time": 9.924601793289185
    },
    {
      "epoch": 2.8339385986328126e-05,
      "model_forward_time": 3.1345810890197754,
      "step": 7429
    },
    {
      "epoch": 2.8339385986328126e-05,
      "step": 7429,
      "training_step_time": 9.89595651626587
    },
    {
      "epoch": 2.8339385986328126e-05,
      "model_forward_time": 2.4329819679260254,
      "step": 7429
    },
    {
      "epoch": 2.8339385986328126e-05,
      "step": 7429,
      "training_step_time": 7.4518351554870605
    },
    {
      "epoch": 2.8339385986328126e-05,
      "model_forward_time": 1.7428765296936035,
      "step": 7429
    },
    {
      "epoch": 2.8339385986328126e-05,
      "step": 7429,
      "training_step_time": 10.28126311302185
    },
    {
      "epoch": 2.834320068359375e-05,
      "grad_norm": 0.4107002019882202,
      "learning_rate": 1e-05,
      "loss": 0.0766,
      "step": 7430
    },
    {
      "epoch": 2.834320068359375e-05,
      "model_forward_time": 3.145956039428711,
      "step": 7430
    },
    {
      "dynamics_loss_avg": 0.047034363821148874,
      "epoch": 2.834320068359375e-05,
      "step": 7430
    },
    {
      "action_loss_avg": 0.017774881981313228,
      "epoch": 2.834320068359375e-05,
      "step": 7430
    },
    {
      "epoch": 2.834320068359375e-05,
      "step": 7430,
      "training_step_time": 9.906293630599976
    },
    {
      "epoch": 2.834320068359375e-05,
      "model_forward_time": 2.7069296836853027,
      "step": 7430
    },
    {
      "epoch": 2.834320068359375e-05,
      "step": 7430,
      "training_step_time": 7.721732139587402
    },
    {
      "epoch": 2.834320068359375e-05,
      "model_forward_time": 2.429488182067871,
      "step": 7430
    },
    {
      "epoch": 2.834320068359375e-05,
      "step": 7430,
      "training_step_time": 7.447390794754028
    },
    {
      "epoch": 2.834320068359375e-05,
      "model_forward_time": 2.432180404663086,
      "step": 7430
    },
    {
      "epoch": 2.834320068359375e-05,
      "step": 7430,
      "training_step_time": 10.01219129562378
    },
    {
      "epoch": 2.8347015380859375e-05,
      "model_forward_time": 3.1903457641601562,
      "step": 7431
    },
    {
      "epoch": 2.8347015380859375e-05,
      "step": 7431,
      "training_step_time": 9.947425365447998
    },
    {
      "epoch": 2.8347015380859375e-05,
      "model_forward_time": 3.132047653198242,
      "step": 7431
    },
    {
      "epoch": 2.8347015380859375e-05,
      "step": 7431,
      "training_step_time": 9.889500856399536
    },
    {
      "epoch": 2.8347015380859375e-05,
      "model_forward_time": 3.412179946899414,
      "step": 7431
    },
    {
      "epoch": 2.8347015380859375e-05,
      "step": 7431,
      "training_step_time": 10.163935422897339
    },
    {
      "epoch": 2.8347015380859375e-05,
      "model_forward_time": 3.134629964828491,
      "step": 7431
    },
    {
      "epoch": 2.8347015380859375e-05,
      "step": 7431,
      "training_step_time": 10.267612934112549
    },
    {
      "epoch": 2.8350830078125e-05,
      "model_forward_time": 3.1832053661346436,
      "step": 7432
    },
    {
      "epoch": 2.8350830078125e-05,
      "step": 7432,
      "training_step_time": 9.94679856300354
    },
    {
      "epoch": 2.8350830078125e-05,
      "model_forward_time": 2.4354140758514404,
      "step": 7432
    },
    {
      "epoch": 2.8350830078125e-05,
      "step": 7432,
      "training_step_time": 7.4590935707092285
    },
    {
      "epoch": 2.8350830078125e-05,
      "model_forward_time": 3.1343655586242676,
      "step": 7432
    },
    {
      "dynamics_loss_avg": 0.0555482467636466,
      "epoch": 2.8350830078125e-05,
      "step": 7432
    },
    {
      "action_loss_avg": 0.020332698430866002,
      "epoch": 2.8350830078125e-05,
      "step": 7432
    },
    {
      "epoch": 2.8350830078125e-05,
      "step": 7432,
      "training_step_time": 9.901304721832275
    },
    {
      "epoch": 2.8350830078125e-05,
      "model_forward_time": 3.389742374420166,
      "step": 7432
    },
    {
      "epoch": 2.8350830078125e-05,
      "step": 7432,
      "training_step_time": 10.300557613372803
    },
    {
      "epoch": 2.8354644775390624e-05,
      "model_forward_time": 3.168555498123169,
      "step": 7433
    },
    {
      "epoch": 2.8354644775390624e-05,
      "step": 7433,
      "training_step_time": 9.939951419830322
    },
    {
      "epoch": 2.8354644775390624e-05,
      "model_forward_time": 2.4337940216064453,
      "step": 7433
    },
    {
      "epoch": 2.8354644775390624e-05,
      "step": 7433,
      "training_step_time": 7.463258743286133
    },
    {
      "epoch": 2.8354644775390624e-05,
      "model_forward_time": 1.744974136352539,
      "step": 7433
    },
    {
      "epoch": 2.8354644775390624e-05,
      "step": 7433,
      "training_step_time": 5.198976993560791
    },
    {
      "epoch": 2.8354644775390624e-05,
      "model_forward_time": 1.7427639961242676,
      "step": 7433
    },
    {
      "epoch": 2.8354644775390624e-05,
      "step": 7433,
      "training_step_time": 10.288973331451416
    },
    {
      "epoch": 2.835845947265625e-05,
      "model_forward_time": 3.1509416103363037,
      "step": 7434
    },
    {
      "epoch": 2.835845947265625e-05,
      "step": 7434,
      "training_step_time": 9.91161561012268
    },
    {
      "epoch": 2.835845947265625e-05,
      "model_forward_time": 2.679852247238159,
      "step": 7434
    },
    {
      "epoch": 2.835845947265625e-05,
      "step": 7434,
      "training_step_time": 7.697546720504761
    },
    {
      "epoch": 2.835845947265625e-05,
      "model_forward_time": 2.4311258792877197,
      "step": 7434
    },
    {
      "epoch": 2.835845947265625e-05,
      "step": 7434,
      "training_step_time": 7.4481236934661865
    },
    {
      "epoch": 2.835845947265625e-05,
      "model_forward_time": 2.4303221702575684,
      "step": 7434
    },
    {
      "epoch": 2.835845947265625e-05,
      "step": 7434,
      "training_step_time": 10.268999576568604
    },
    {
      "epoch": 2.8362274169921874e-05,
      "model_forward_time": 3.1456141471862793,
      "step": 7435
    },
    {
      "dynamics_loss_avg": 0.06004969691857696,
      "epoch": 2.8362274169921874e-05,
      "step": 7435
    },
    {
      "action_loss_avg": 0.02194429445080459,
      "epoch": 2.8362274169921874e-05,
      "step": 7435
    },
    {
      "epoch": 2.8362274169921874e-05,
      "step": 7435,
      "training_step_time": 9.907271146774292
    },
    {
      "epoch": 2.8362274169921874e-05,
      "model_forward_time": 3.1288375854492188,
      "step": 7435
    },
    {
      "epoch": 2.8362274169921874e-05,
      "step": 7435,
      "training_step_time": 9.893554925918579
    },
    {
      "epoch": 2.8362274169921874e-05,
      "model_forward_time": 2.712808609008789,
      "step": 7435
    },
    {
      "epoch": 2.8362274169921874e-05,
      "step": 7435,
      "training_step_time": 7.733318567276001
    },
    {
      "epoch": 2.8362274169921874e-05,
      "model_forward_time": 2.4331858158111572,
      "step": 7435
    },
    {
      "epoch": 2.8362274169921874e-05,
      "step": 7435,
      "training_step_time": 10.306623935699463
    },
    {
      "epoch": 2.83660888671875e-05,
      "model_forward_time": 3.1698169708251953,
      "step": 7436
    },
    {
      "epoch": 2.83660888671875e-05,
      "step": 7436,
      "training_step_time": 9.933727264404297
    },
    {
      "epoch": 2.83660888671875e-05,
      "model_forward_time": 2.435924768447876,
      "step": 7436
    },
    {
      "epoch": 2.83660888671875e-05,
      "step": 7436,
      "training_step_time": 7.459739446640015
    },
    {
      "epoch": 2.83660888671875e-05,
      "model_forward_time": 3.1366944313049316,
      "step": 7436
    },
    {
      "epoch": 2.83660888671875e-05,
      "step": 7436,
      "training_step_time": 9.902564287185669
    },
    {
      "epoch": 2.83660888671875e-05,
      "model_forward_time": 3.420675277709961,
      "step": 7436
    },
    {
      "epoch": 2.83660888671875e-05,
      "step": 7436,
      "training_step_time": 10.27886438369751
    },
    {
      "epoch": 2.8369903564453126e-05,
      "model_forward_time": 3.1851632595062256,
      "step": 7437
    },
    {
      "epoch": 2.8369903564453126e-05,
      "step": 7437,
      "training_step_time": 9.952680826187134
    },
    {
      "epoch": 2.8369903564453126e-05,
      "model_forward_time": 2.4355790615081787,
      "step": 7437
    },
    {
      "epoch": 2.8369903564453126e-05,
      "step": 7437,
      "training_step_time": 7.456583023071289
    },
    {
      "epoch": 2.8369903564453126e-05,
      "model_forward_time": 3.1348304748535156,
      "step": 7437
    },
    {
      "dynamics_loss_avg": 0.058674710988998416,
      "epoch": 2.8369903564453126e-05,
      "step": 7437
    },
    {
      "action_loss_avg": 0.022353223338723183,
      "epoch": 2.8369903564453126e-05,
      "step": 7437
    },
    {
      "epoch": 2.8369903564453126e-05,
      "step": 7437,
      "training_step_time": 9.897108554840088
    },
    {
      "epoch": 2.8369903564453126e-05,
      "model_forward_time": 2.4311318397521973,
      "step": 7437
    },
    {
      "epoch": 2.8369903564453126e-05,
      "step": 7437,
      "training_step_time": 10.298383951187134
    },
    {
      "epoch": 2.837371826171875e-05,
      "model_forward_time": 3.1818881034851074,
      "step": 7438
    },
    {
      "epoch": 2.837371826171875e-05,
      "step": 7438,
      "training_step_time": 9.947898149490356
    },
    {
      "epoch": 2.837371826171875e-05,
      "model_forward_time": 2.699347496032715,
      "step": 7438
    },
    {
      "epoch": 2.837371826171875e-05,
      "step": 7438,
      "training_step_time": 7.721198558807373
    },
    {
      "epoch": 2.837371826171875e-05,
      "model_forward_time": 3.1348772048950195,
      "step": 7438
    },
    {
      "epoch": 2.837371826171875e-05,
      "step": 7438,
      "training_step_time": 9.899131059646606
    },
    {
      "epoch": 2.837371826171875e-05,
      "model_forward_time": 3.138017177581787,
      "step": 7438
    },
    {
      "epoch": 2.837371826171875e-05,
      "step": 7438,
      "training_step_time": 10.314443826675415
    },
    {
      "epoch": 2.8377532958984375e-05,
      "model_forward_time": 2.4805543422698975,
      "step": 7439
    },
    {
      "epoch": 2.8377532958984375e-05,
      "step": 7439,
      "training_step_time": 7.5041563510894775
    },
    {
      "epoch": 2.8377532958984375e-05,
      "model_forward_time": 3.138674736022949,
      "step": 7439
    },
    {
      "epoch": 2.8377532958984375e-05,
      "step": 7439,
      "training_step_time": 9.908354997634888
    },
    {
      "epoch": 2.8377532958984375e-05,
      "model_forward_time": 2.684370279312134,
      "step": 7439
    },
    {
      "epoch": 2.8377532958984375e-05,
      "step": 7439,
      "training_step_time": 7.713814735412598
    },
    {
      "epoch": 2.8377532958984375e-05,
      "model_forward_time": 3.1403753757476807,
      "step": 7439
    },
    {
      "epoch": 2.8377532958984375e-05,
      "step": 7439,
      "training_step_time": 10.040898084640503
    },
    {
      "epoch": 2.838134765625e-05,
      "grad_norm": 0.29368749260902405,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 7440
    },
    {
      "epoch": 2.838134765625e-05,
      "model_forward_time": 3.1541759967803955,
      "step": 7440
    },
    {
      "dynamics_loss_avg": 0.05487036127597093,
      "epoch": 2.838134765625e-05,
      "step": 7440
    },
    {
      "action_loss_avg": 0.022890609176829457,
      "epoch": 2.838134765625e-05,
      "step": 7440
    },
    {
      "epoch": 2.838134765625e-05,
      "step": 7440,
      "training_step_time": 9.929083108901978
    },
    {
      "epoch": 2.838134765625e-05,
      "model_forward_time": 3.147697925567627,
      "step": 7440
    },
    {
      "epoch": 2.838134765625e-05,
      "step": 7440,
      "training_step_time": 9.922855377197266
    },
    {
      "epoch": 2.838134765625e-05,
      "model_forward_time": 3.1445600986480713,
      "step": 7440
    },
    {
      "epoch": 2.838134765625e-05,
      "step": 7440,
      "training_step_time": 9.920323848724365
    },
    {
      "epoch": 2.838134765625e-05,
      "model_forward_time": 2.026374578475952,
      "step": 7440
    },
    {
      "epoch": 2.838134765625e-05,
      "step": 7440,
      "training_step_time": 10.305588722229004
    },
    {
      "epoch": 2.8385162353515624e-05,
      "model_forward_time": 3.167649507522583,
      "step": 7441
    },
    {
      "epoch": 2.8385162353515624e-05,
      "step": 7441,
      "training_step_time": 9.941518783569336
    },
    {
      "epoch": 2.8385162353515624e-05,
      "model_forward_time": 2.4398691654205322,
      "step": 7441
    },
    {
      "epoch": 2.8385162353515624e-05,
      "step": 7441,
      "training_step_time": 7.470276355743408
    },
    {
      "epoch": 2.8385162353515624e-05,
      "model_forward_time": 2.4363186359405518,
      "step": 7441
    },
    {
      "epoch": 2.8385162353515624e-05,
      "step": 7441,
      "training_step_time": 7.463449478149414
    },
    {
      "epoch": 2.8385162353515624e-05,
      "model_forward_time": 1.744234561920166,
      "step": 7441
    },
    {
      "epoch": 2.8385162353515624e-05,
      "step": 7441,
      "training_step_time": 10.240081310272217
    },
    {
      "epoch": 2.838897705078125e-05,
      "model_forward_time": 3.1763358116149902,
      "step": 7442
    },
    {
      "epoch": 2.838897705078125e-05,
      "step": 7442,
      "training_step_time": 9.94173288345337
    },
    {
      "epoch": 2.838897705078125e-05,
      "model_forward_time": 3.4537665843963623,
      "step": 7442
    },
    {
      "epoch": 2.838897705078125e-05,
      "step": 7442,
      "training_step_time": 10.219829320907593
    },
    {
      "epoch": 2.838897705078125e-05,
      "model_forward_time": 3.1408767700195312,
      "step": 7442
    },
    {
      "dynamics_loss_avg": 0.06192027721554041,
      "epoch": 2.838897705078125e-05,
      "step": 7442
    },
    {
      "action_loss_avg": 0.028048847033642233,
      "epoch": 2.838897705078125e-05,
      "step": 7442
    },
    {
      "epoch": 2.838897705078125e-05,
      "step": 7442,
      "training_step_time": 9.906754970550537
    },
    {
      "epoch": 2.838897705078125e-05,
      "model_forward_time": 2.4366774559020996,
      "step": 7442
    },
    {
      "epoch": 2.838897705078125e-05,
      "step": 7442,
      "training_step_time": 10.28997254371643
    },
    {
      "epoch": 2.8392791748046874e-05,
      "model_forward_time": 3.155836343765259,
      "step": 7443
    },
    {
      "epoch": 2.8392791748046874e-05,
      "step": 7443,
      "training_step_time": 9.921512365341187
    },
    {
      "epoch": 2.8392791748046874e-05,
      "model_forward_time": 3.1351304054260254,
      "step": 7443
    },
    {
      "epoch": 2.8392791748046874e-05,
      "step": 7443,
      "training_step_time": 10.157855033874512
    },
    {
      "epoch": 2.8392791748046874e-05,
      "model_forward_time": 3.1380081176757812,
      "step": 7443
    },
    {
      "epoch": 2.8392791748046874e-05,
      "step": 7443,
      "training_step_time": 9.904336929321289
    },
    {
      "epoch": 2.8392791748046874e-05,
      "model_forward_time": 2.4333996772766113,
      "step": 7443
    },
    {
      "epoch": 2.8392791748046874e-05,
      "step": 7443,
      "training_step_time": 10.313258647918701
    },
    {
      "epoch": 2.83966064453125e-05,
      "model_forward_time": 1.7711598873138428,
      "step": 7444
    },
    {
      "epoch": 2.83966064453125e-05,
      "step": 7444,
      "training_step_time": 5.220171689987183
    },
    {
      "epoch": 2.83966064453125e-05,
      "model_forward_time": 3.130244493484497,
      "step": 7444
    },
    {
      "epoch": 2.83966064453125e-05,
      "step": 7444,
      "training_step_time": 9.888179779052734
    },
    {
      "epoch": 2.83966064453125e-05,
      "model_forward_time": 3.1362152099609375,
      "step": 7444
    },
    {
      "epoch": 2.83966064453125e-05,
      "step": 7444,
      "training_step_time": 9.892132759094238
    },
    {
      "epoch": 2.83966064453125e-05,
      "model_forward_time": 3.4242329597473145,
      "step": 7444
    },
    {
      "epoch": 2.83966064453125e-05,
      "step": 7444,
      "training_step_time": 10.271841049194336
    },
    {
      "epoch": 2.8400421142578126e-05,
      "model_forward_time": 3.1831881999969482,
      "step": 7445
    },
    {
      "dynamics_loss_avg": 0.061501169577240944,
      "epoch": 2.8400421142578126e-05,
      "step": 7445
    },
    {
      "action_loss_avg": 0.03114529708400369,
      "epoch": 2.8400421142578126e-05,
      "step": 7445
    },
    {
      "epoch": 2.8400421142578126e-05,
      "step": 7445,
      "training_step_time": 9.953130722045898
    },
    {
      "epoch": 2.8400421142578126e-05,
      "model_forward_time": 2.432586431503296,
      "step": 7445
    },
    {
      "epoch": 2.8400421142578126e-05,
      "step": 7445,
      "training_step_time": 7.454316854476929
    },
    {
      "epoch": 2.8400421142578126e-05,
      "model_forward_time": 3.1353437900543213,
      "step": 7445
    },
    {
      "epoch": 2.8400421142578126e-05,
      "step": 7445,
      "training_step_time": 9.898176670074463
    },
    {
      "epoch": 2.8400421142578126e-05,
      "model_forward_time": 2.4345781803131104,
      "step": 7445
    },
    {
      "epoch": 2.8400421142578126e-05,
      "step": 7445,
      "training_step_time": 10.22194218635559
    },
    {
      "epoch": 2.840423583984375e-05,
      "model_forward_time": 2.7486412525177,
      "step": 7446
    },
    {
      "epoch": 2.840423583984375e-05,
      "step": 7446,
      "training_step_time": 7.771825551986694
    },
    {
      "epoch": 2.840423583984375e-05,
      "model_forward_time": 2.4310009479522705,
      "step": 7446
    },
    {
      "epoch": 2.840423583984375e-05,
      "step": 7446,
      "training_step_time": 7.45602822303772
    },
    {
      "epoch": 2.840423583984375e-05,
      "model_forward_time": 3.139036178588867,
      "step": 7446
    },
    {
      "epoch": 2.840423583984375e-05,
      "step": 7446,
      "training_step_time": 9.90744423866272
    },
    {
      "epoch": 2.840423583984375e-05,
      "model_forward_time": 3.1372387409210205,
      "step": 7446
    },
    {
      "epoch": 2.840423583984375e-05,
      "step": 7446,
      "training_step_time": 10.030354499816895
    },
    {
      "epoch": 2.8408050537109375e-05,
      "model_forward_time": 2.475350856781006,
      "step": 7447
    },
    {
      "epoch": 2.8408050537109375e-05,
      "step": 7447,
      "training_step_time": 7.500360012054443
    },
    {
      "epoch": 2.8408050537109375e-05,
      "model_forward_time": 2.7153942584991455,
      "step": 7447
    },
    {
      "epoch": 2.8408050537109375e-05,
      "step": 7447,
      "training_step_time": 7.742105960845947
    },
    {
      "epoch": 2.8408050537109375e-05,
      "model_forward_time": 3.134495735168457,
      "step": 7447
    },
    {
      "dynamics_loss_avg": 0.052984795160591604,
      "epoch": 2.8408050537109375e-05,
      "step": 7447
    },
    {
      "action_loss_avg": 0.017883546138182282,
      "epoch": 2.8408050537109375e-05,
      "step": 7447
    },
    {
      "epoch": 2.8408050537109375e-05,
      "step": 7447,
      "training_step_time": 9.902987480163574
    },
    {
      "epoch": 2.8408050537109375e-05,
      "model_forward_time": 3.136754035949707,
      "step": 7447
    },
    {
      "epoch": 2.8408050537109375e-05,
      "step": 7447,
      "training_step_time": 10.022294521331787
    },
    {
      "epoch": 2.8411865234375e-05,
      "model_forward_time": 2.4836316108703613,
      "step": 7448
    },
    {
      "epoch": 2.8411865234375e-05,
      "step": 7448,
      "training_step_time": 7.519712686538696
    },
    {
      "epoch": 2.8411865234375e-05,
      "model_forward_time": 3.140796184539795,
      "step": 7448
    },
    {
      "epoch": 2.8411865234375e-05,
      "step": 7448,
      "training_step_time": 9.904690980911255
    },
    {
      "epoch": 2.8411865234375e-05,
      "model_forward_time": 3.1457414627075195,
      "step": 7448
    },
    {
      "epoch": 2.8411865234375e-05,
      "step": 7448,
      "training_step_time": 9.91365671157837
    },
    {
      "epoch": 2.8411865234375e-05,
      "model_forward_time": 2.706761121749878,
      "step": 7448
    },
    {
      "epoch": 2.8411865234375e-05,
      "step": 7448,
      "training_step_time": 10.304762840270996
    },
    {
      "epoch": 2.8415679931640624e-05,
      "model_forward_time": 3.1491951942443848,
      "step": 7449
    },
    {
      "epoch": 2.8415679931640624e-05,
      "step": 7449,
      "training_step_time": 9.917493343353271
    },
    {
      "epoch": 2.8415679931640624e-05,
      "model_forward_time": 3.1416404247283936,
      "step": 7449
    },
    {
      "epoch": 2.8415679931640624e-05,
      "step": 7449,
      "training_step_time": 9.911973714828491
    },
    {
      "epoch": 2.8415679931640624e-05,
      "model_forward_time": 3.1398425102233887,
      "step": 7449
    },
    {
      "epoch": 2.8415679931640624e-05,
      "step": 7449,
      "training_step_time": 9.910324096679688
    },
    {
      "epoch": 2.8415679931640624e-05,
      "model_forward_time": 2.434112787246704,
      "step": 7449
    },
    {
      "epoch": 2.8415679931640624e-05,
      "step": 7449,
      "training_step_time": 10.268926858901978
    },
    {
      "epoch": 2.841949462890625e-05,
      "grad_norm": 0.3452734649181366,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 7450
    },
    {
      "epoch": 2.841949462890625e-05,
      "model_forward_time": 3.4452152252197266,
      "step": 7450
    },
    {
      "dynamics_loss_avg": 0.058175386115908626,
      "epoch": 2.841949462890625e-05,
      "step": 7450
    },
    {
      "action_loss_avg": 0.02183169303461909,
      "epoch": 2.841949462890625e-05,
      "step": 7450
    },
    {
      "epoch": 2.841949462890625e-05,
      "step": 7450,
      "training_step_time": 10.212834119796753
    },
    {
      "epoch": 2.841949462890625e-05,
      "model_forward_time": 3.1398327350616455,
      "step": 7450
    },
    {
      "epoch": 2.841949462890625e-05,
      "step": 7450,
      "training_step_time": 9.905332088470459
    },
    {
      "epoch": 2.841949462890625e-05,
      "model_forward_time": 2.437551975250244,
      "step": 7450
    },
    {
      "epoch": 2.841949462890625e-05,
      "step": 7450,
      "training_step_time": 7.460139513015747
    },
    {
      "epoch": 2.841949462890625e-05,
      "model_forward_time": 2.4358766078948975,
      "step": 7450
    },
    {
      "epoch": 2.841949462890625e-05,
      "step": 7450,
      "training_step_time": 10.016459703445435
    },
    {
      "epoch": 2.8423309326171874e-05,
      "model_forward_time": 2.4781436920166016,
      "step": 7451
    },
    {
      "epoch": 2.8423309326171874e-05,
      "step": 7451,
      "training_step_time": 7.497256278991699
    },
    {
      "epoch": 2.8423309326171874e-05,
      "model_forward_time": 2.7157211303710938,
      "step": 7451
    },
    {
      "epoch": 2.8423309326171874e-05,
      "step": 7451,
      "training_step_time": 7.73248553276062
    },
    {
      "epoch": 2.8423309326171874e-05,
      "model_forward_time": 3.131422996520996,
      "step": 7451
    },
    {
      "epoch": 2.8423309326171874e-05,
      "step": 7451,
      "training_step_time": 9.892961263656616
    },
    {
      "epoch": 2.8423309326171874e-05,
      "model_forward_time": 2.435323476791382,
      "step": 7451
    },
    {
      "epoch": 2.8423309326171874e-05,
      "step": 7451,
      "training_step_time": 10.254814863204956
    },
    {
      "epoch": 2.84271240234375e-05,
      "model_forward_time": 3.1544675827026367,
      "step": 7452
    },
    {
      "epoch": 2.84271240234375e-05,
      "step": 7452,
      "training_step_time": 9.92469573020935
    },
    {
      "epoch": 2.84271240234375e-05,
      "model_forward_time": 2.4356577396392822,
      "step": 7452
    },
    {
      "epoch": 2.84271240234375e-05,
      "step": 7452,
      "training_step_time": 7.461267709732056
    },
    {
      "epoch": 2.84271240234375e-05,
      "model_forward_time": 2.729771614074707,
      "step": 7452
    },
    {
      "dynamics_loss_avg": 0.05623790863901377,
      "epoch": 2.84271240234375e-05,
      "step": 7452
    },
    {
      "action_loss_avg": 0.030019257962703706,
      "epoch": 2.84271240234375e-05,
      "step": 7452
    },
    {
      "epoch": 2.84271240234375e-05,
      "step": 7452,
      "training_step_time": 7.754645824432373
    },
    {
      "epoch": 2.84271240234375e-05,
      "model_forward_time": 1.7465884685516357,
      "step": 7452
    },
    {
      "epoch": 2.84271240234375e-05,
      "step": 7452,
      "training_step_time": 10.032439947128296
    },
    {
      "epoch": 2.8430938720703126e-05,
      "model_forward_time": 3.1723597049713135,
      "step": 7453
    },
    {
      "epoch": 2.8430938720703126e-05,
      "step": 7453,
      "training_step_time": 9.942389488220215
    },
    {
      "epoch": 2.8430938720703126e-05,
      "model_forward_time": 2.43829083442688,
      "step": 7453
    },
    {
      "epoch": 2.8430938720703126e-05,
      "step": 7453,
      "training_step_time": 7.463940143585205
    },
    {
      "epoch": 2.8430938720703126e-05,
      "model_forward_time": 3.142794609069824,
      "step": 7453
    },
    {
      "epoch": 2.8430938720703126e-05,
      "step": 7453,
      "training_step_time": 9.91474461555481
    },
    {
      "epoch": 2.8430938720703126e-05,
      "model_forward_time": 3.1453800201416016,
      "step": 7453
    },
    {
      "epoch": 2.8430938720703126e-05,
      "step": 7453,
      "training_step_time": 10.314419746398926
    },
    {
      "epoch": 2.843475341796875e-05,
      "model_forward_time": 2.483842372894287,
      "step": 7454
    },
    {
      "epoch": 2.843475341796875e-05,
      "step": 7454,
      "training_step_time": 7.512984752655029
    },
    {
      "epoch": 2.843475341796875e-05,
      "model_forward_time": 3.138554096221924,
      "step": 7454
    },
    {
      "epoch": 2.843475341796875e-05,
      "step": 7454,
      "training_step_time": 9.910653352737427
    },
    {
      "epoch": 2.843475341796875e-05,
      "model_forward_time": 3.147211790084839,
      "step": 7454
    },
    {
      "epoch": 2.843475341796875e-05,
      "step": 7454,
      "training_step_time": 9.91951847076416
    },
    {
      "epoch": 2.843475341796875e-05,
      "model_forward_time": 2.436544179916382,
      "step": 7454
    },
    {
      "epoch": 2.843475341796875e-05,
      "step": 7454,
      "training_step_time": 10.307422399520874
    },
    {
      "epoch": 2.8438568115234375e-05,
      "model_forward_time": 1.764211893081665,
      "step": 7455
    },
    {
      "dynamics_loss_avg": 0.04898178866133094,
      "epoch": 2.8438568115234375e-05,
      "step": 7455
    },
    {
      "action_loss_avg": 0.020737357321195304,
      "epoch": 2.8438568115234375e-05,
      "step": 7455
    },
    {
      "epoch": 2.8438568115234375e-05,
      "step": 7455,
      "training_step_time": 5.22704005241394
    },
    {
      "epoch": 2.8438568115234375e-05,
      "model_forward_time": 2.696821689605713,
      "step": 7455
    },
    {
      "epoch": 2.8438568115234375e-05,
      "step": 7455,
      "training_step_time": 7.720355749130249
    },
    {
      "epoch": 2.8438568115234375e-05,
      "model_forward_time": 3.135823965072632,
      "step": 7455
    },
    {
      "epoch": 2.8438568115234375e-05,
      "step": 7455,
      "training_step_time": 9.900747299194336
    },
    {
      "epoch": 2.8438568115234375e-05,
      "model_forward_time": 1.747661828994751,
      "step": 7455
    },
    {
      "epoch": 2.8438568115234375e-05,
      "step": 7455,
      "training_step_time": 10.020222902297974
    },
    {
      "epoch": 2.84423828125e-05,
      "model_forward_time": 3.174947500228882,
      "step": 7456
    },
    {
      "epoch": 2.84423828125e-05,
      "step": 7456,
      "training_step_time": 9.942715883255005
    },
    {
      "epoch": 2.84423828125e-05,
      "model_forward_time": 2.4360158443450928,
      "step": 7456
    },
    {
      "epoch": 2.84423828125e-05,
      "step": 7456,
      "training_step_time": 7.461339235305786
    },
    {
      "epoch": 2.84423828125e-05,
      "model_forward_time": 3.1354939937591553,
      "step": 7456
    },
    {
      "epoch": 2.84423828125e-05,
      "step": 7456,
      "training_step_time": 10.217551469802856
    },
    {
      "epoch": 2.84423828125e-05,
      "model_forward_time": 3.138514757156372,
      "step": 7456
    },
    {
      "epoch": 2.84423828125e-05,
      "step": 7456,
      "training_step_time": 10.25414752960205
    },
    {
      "epoch": 2.8446197509765624e-05,
      "model_forward_time": 2.4550788402557373,
      "step": 7457
    },
    {
      "epoch": 2.8446197509765624e-05,
      "step": 7457,
      "training_step_time": 7.47808575630188
    },
    {
      "epoch": 2.8446197509765624e-05,
      "model_forward_time": 2.443859577178955,
      "step": 7457
    },
    {
      "epoch": 2.8446197509765624e-05,
      "step": 7457,
      "training_step_time": 7.464270114898682
    },
    {
      "epoch": 2.8446197509765624e-05,
      "model_forward_time": 3.130481243133545,
      "step": 7457
    },
    {
      "dynamics_loss_avg": 0.05403587929904461,
      "epoch": 2.8446197509765624e-05,
      "step": 7457
    },
    {
      "action_loss_avg": 0.018333655688911675,
      "epoch": 2.8446197509765624e-05,
      "step": 7457
    },
    {
      "epoch": 2.8446197509765624e-05,
      "step": 7457,
      "training_step_time": 9.90030288696289
    },
    {
      "epoch": 2.8446197509765624e-05,
      "model_forward_time": 2.432957410812378,
      "step": 7457
    },
    {
      "epoch": 2.8446197509765624e-05,
      "step": 7457,
      "training_step_time": 10.004832983016968
    },
    {
      "epoch": 2.845001220703125e-05,
      "model_forward_time": 2.7400217056274414,
      "step": 7458
    },
    {
      "epoch": 2.845001220703125e-05,
      "step": 7458,
      "training_step_time": 7.7626447677612305
    },
    {
      "epoch": 2.845001220703125e-05,
      "model_forward_time": 2.431307077407837,
      "step": 7458
    },
    {
      "epoch": 2.845001220703125e-05,
      "step": 7458,
      "training_step_time": 7.449938058853149
    },
    {
      "epoch": 2.845001220703125e-05,
      "model_forward_time": 2.428781270980835,
      "step": 7458
    },
    {
      "epoch": 2.845001220703125e-05,
      "step": 7458,
      "training_step_time": 7.451972961425781
    },
    {
      "epoch": 2.845001220703125e-05,
      "model_forward_time": 2.433293342590332,
      "step": 7458
    },
    {
      "epoch": 2.845001220703125e-05,
      "step": 7458,
      "training_step_time": 10.263736009597778
    },
    {
      "epoch": 2.8453826904296874e-05,
      "model_forward_time": 3.1564626693725586,
      "step": 7459
    },
    {
      "epoch": 2.8453826904296874e-05,
      "step": 7459,
      "training_step_time": 9.92183518409729
    },
    {
      "epoch": 2.8453826904296874e-05,
      "model_forward_time": 2.4325757026672363,
      "step": 7459
    },
    {
      "epoch": 2.8453826904296874e-05,
      "step": 7459,
      "training_step_time": 7.734600067138672
    },
    {
      "epoch": 2.8453826904296874e-05,
      "model_forward_time": 3.127347230911255,
      "step": 7459
    },
    {
      "epoch": 2.8453826904296874e-05,
      "step": 7459,
      "training_step_time": 9.886976957321167
    },
    {
      "epoch": 2.8453826904296874e-05,
      "model_forward_time": 2.430227756500244,
      "step": 7459
    },
    {
      "epoch": 2.8453826904296874e-05,
      "step": 7459,
      "training_step_time": 10.275393009185791
    },
    {
      "epoch": 2.84576416015625e-05,
      "grad_norm": 0.3233489692211151,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 7460
    },
    {
      "epoch": 2.84576416015625e-05,
      "model_forward_time": 3.1711292266845703,
      "step": 7460
    },
    {
      "dynamics_loss_avg": 0.049159903451800345,
      "epoch": 2.84576416015625e-05,
      "step": 7460
    },
    {
      "action_loss_avg": 0.023010466573759915,
      "epoch": 2.84576416015625e-05,
      "step": 7460
    },
    {
      "epoch": 2.84576416015625e-05,
      "step": 7460,
      "training_step_time": 9.939607620239258
    },
    {
      "epoch": 2.84576416015625e-05,
      "model_forward_time": 2.4317145347595215,
      "step": 7460
    },
    {
      "epoch": 2.84576416015625e-05,
      "step": 7460,
      "training_step_time": 7.451536178588867
    },
    {
      "epoch": 2.84576416015625e-05,
      "model_forward_time": 3.1333863735198975,
      "step": 7460
    },
    {
      "epoch": 2.84576416015625e-05,
      "step": 7460,
      "training_step_time": 9.896962642669678
    },
    {
      "epoch": 2.84576416015625e-05,
      "model_forward_time": 3.40024995803833,
      "step": 7460
    },
    {
      "epoch": 2.84576416015625e-05,
      "step": 7460,
      "training_step_time": 10.258825778961182
    },
    {
      "epoch": 2.8461456298828126e-05,
      "model_forward_time": 2.4707717895507812,
      "step": 7461
    },
    {
      "epoch": 2.8461456298828126e-05,
      "step": 7461,
      "training_step_time": 7.4887566566467285
    },
    {
      "epoch": 2.8461456298828126e-05,
      "model_forward_time": 3.1238152980804443,
      "step": 7461
    },
    {
      "epoch": 2.8461456298828126e-05,
      "step": 7461,
      "training_step_time": 9.878940343856812
    },
    {
      "epoch": 2.8461456298828126e-05,
      "model_forward_time": 3.1293540000915527,
      "step": 7461
    },
    {
      "epoch": 2.8461456298828126e-05,
      "step": 7461,
      "training_step_time": 9.883968353271484
    },
    {
      "epoch": 2.8461456298828126e-05,
      "model_forward_time": 3.1255438327789307,
      "step": 7461
    },
    {
      "epoch": 2.8461456298828126e-05,
      "step": 7461,
      "training_step_time": 10.269392013549805
    },
    {
      "epoch": 2.846527099609375e-05,
      "model_forward_time": 3.4651360511779785,
      "step": 7462
    },
    {
      "epoch": 2.846527099609375e-05,
      "step": 7462,
      "training_step_time": 10.226490497589111
    },
    {
      "epoch": 2.846527099609375e-05,
      "model_forward_time": 3.1313490867614746,
      "step": 7462
    },
    {
      "epoch": 2.846527099609375e-05,
      "step": 7462,
      "training_step_time": 9.891366243362427
    },
    {
      "epoch": 2.846527099609375e-05,
      "model_forward_time": 2.427393913269043,
      "step": 7462
    },
    {
      "dynamics_loss_avg": 0.04838333688676357,
      "epoch": 2.846527099609375e-05,
      "step": 7462
    },
    {
      "action_loss_avg": 0.02267247224226594,
      "epoch": 2.846527099609375e-05,
      "step": 7462
    },
    {
      "epoch": 2.846527099609375e-05,
      "step": 7462,
      "training_step_time": 7.448387622833252
    },
    {
      "epoch": 2.846527099609375e-05,
      "model_forward_time": 3.1256861686706543,
      "step": 7462
    },
    {
      "epoch": 2.846527099609375e-05,
      "step": 7462,
      "training_step_time": 10.00871467590332
    },
    {
      "epoch": 2.8469085693359375e-05,
      "model_forward_time": 2.448720932006836,
      "step": 7463
    },
    {
      "epoch": 2.8469085693359375e-05,
      "step": 7463,
      "training_step_time": 7.470635890960693
    },
    {
      "epoch": 2.8469085693359375e-05,
      "model_forward_time": 3.3807413578033447,
      "step": 7463
    },
    {
      "epoch": 2.8469085693359375e-05,
      "step": 7463,
      "training_step_time": 10.14458179473877
    },
    {
      "epoch": 2.8469085693359375e-05,
      "model_forward_time": 2.4306743144989014,
      "step": 7463
    },
    {
      "epoch": 2.8469085693359375e-05,
      "step": 7463,
      "training_step_time": 7.448627948760986
    },
    {
      "epoch": 2.8469085693359375e-05,
      "model_forward_time": 2.428856134414673,
      "step": 7463
    },
    {
      "epoch": 2.8469085693359375e-05,
      "step": 7463,
      "training_step_time": 10.283828020095825
    },
    {
      "epoch": 2.8472900390625e-05,
      "model_forward_time": 3.1479570865631104,
      "step": 7464
    },
    {
      "epoch": 2.8472900390625e-05,
      "step": 7464,
      "training_step_time": 9.909339189529419
    },
    {
      "epoch": 2.8472900390625e-05,
      "model_forward_time": 2.4314188957214355,
      "step": 7464
    },
    {
      "epoch": 2.8472900390625e-05,
      "step": 7464,
      "training_step_time": 7.450019359588623
    },
    {
      "epoch": 2.8472900390625e-05,
      "model_forward_time": 2.7041056156158447,
      "step": 7464
    },
    {
      "epoch": 2.8472900390625e-05,
      "step": 7464,
      "training_step_time": 7.720351696014404
    },
    {
      "epoch": 2.8472900390625e-05,
      "model_forward_time": 2.4245991706848145,
      "step": 7464
    },
    {
      "epoch": 2.8472900390625e-05,
      "step": 7464,
      "training_step_time": 10.245206832885742
    },
    {
      "epoch": 2.8476715087890624e-05,
      "model_forward_time": 2.452909469604492,
      "step": 7465
    },
    {
      "dynamics_loss_avg": 0.05967817697674036,
      "epoch": 2.8476715087890624e-05,
      "step": 7465
    },
    {
      "action_loss_avg": 0.02275673266267404,
      "epoch": 2.8476715087890624e-05,
      "step": 7465
    },
    {
      "epoch": 2.8476715087890624e-05,
      "step": 7465,
      "training_step_time": 7.470599174499512
    },
    {
      "epoch": 2.8476715087890624e-05,
      "model_forward_time": 3.128682851791382,
      "step": 7465
    },
    {
      "epoch": 2.8476715087890624e-05,
      "step": 7465,
      "training_step_time": 9.883591651916504
    },
    {
      "epoch": 2.8476715087890624e-05,
      "model_forward_time": 3.125828981399536,
      "step": 7465
    },
    {
      "epoch": 2.8476715087890624e-05,
      "step": 7465,
      "training_step_time": 9.881319522857666
    },
    {
      "epoch": 2.8476715087890624e-05,
      "model_forward_time": 2.712564468383789,
      "step": 7465
    },
    {
      "epoch": 2.8476715087890624e-05,
      "step": 7465,
      "training_step_time": 10.28830885887146
    },
    {
      "epoch": 2.848052978515625e-05,
      "model_forward_time": 1.7903454303741455,
      "step": 7466
    },
    {
      "epoch": 2.848052978515625e-05,
      "step": 7466,
      "training_step_time": 5.24550461769104
    },
    {
      "epoch": 2.848052978515625e-05,
      "model_forward_time": 3.1302647590637207,
      "step": 7466
    },
    {
      "epoch": 2.848052978515625e-05,
      "step": 7466,
      "training_step_time": 9.886598348617554
    },
    {
      "epoch": 2.848052978515625e-05,
      "model_forward_time": 3.1301820278167725,
      "step": 7466
    },
    {
      "epoch": 2.848052978515625e-05,
      "step": 7466,
      "training_step_time": 9.89354920387268
    },
    {
      "epoch": 2.848052978515625e-05,
      "model_forward_time": 3.134662628173828,
      "step": 7466
    },
    {
      "epoch": 2.848052978515625e-05,
      "step": 7466,
      "training_step_time": 10.01096510887146
    },
    {
      "epoch": 2.8484344482421874e-05,
      "model_forward_time": 3.147972822189331,
      "step": 7467
    },
    {
      "epoch": 2.8484344482421874e-05,
      "step": 7467,
      "training_step_time": 9.9143648147583
    },
    {
      "epoch": 2.8484344482421874e-05,
      "model_forward_time": 2.7329914569854736,
      "step": 7467
    },
    {
      "epoch": 2.8484344482421874e-05,
      "step": 7467,
      "training_step_time": 7.758074760437012
    },
    {
      "epoch": 2.8484344482421874e-05,
      "model_forward_time": 3.1359059810638428,
      "step": 7467
    },
    {
      "dynamics_loss_avg": 0.06446725195273757,
      "epoch": 2.8484344482421874e-05,
      "step": 7467
    },
    {
      "action_loss_avg": 0.02867089435458183,
      "epoch": 2.8484344482421874e-05,
      "step": 7467
    },
    {
      "epoch": 2.8484344482421874e-05,
      "step": 7467,
      "training_step_time": 9.905985116958618
    },
    {
      "epoch": 2.8484344482421874e-05,
      "model_forward_time": 1.7447493076324463,
      "step": 7467
    },
    {
      "epoch": 2.8484344482421874e-05,
      "step": 7467,
      "training_step_time": 10.2480309009552
    },
    {
      "epoch": 2.84881591796875e-05,
      "model_forward_time": 2.4453635215759277,
      "step": 7468
    },
    {
      "epoch": 2.84881591796875e-05,
      "step": 7468,
      "training_step_time": 7.465015649795532
    },
    {
      "epoch": 2.84881591796875e-05,
      "model_forward_time": 2.427471160888672,
      "step": 7468
    },
    {
      "epoch": 2.84881591796875e-05,
      "step": 7468,
      "training_step_time": 7.446558952331543
    },
    {
      "epoch": 2.84881591796875e-05,
      "model_forward_time": 2.716082811355591,
      "step": 7468
    },
    {
      "epoch": 2.84881591796875e-05,
      "step": 7468,
      "training_step_time": 7.737730026245117
    },
    {
      "epoch": 2.84881591796875e-05,
      "model_forward_time": 1.7418196201324463,
      "step": 7468
    },
    {
      "epoch": 2.84881591796875e-05,
      "step": 7468,
      "training_step_time": 10.258630752563477
    },
    {
      "epoch": 2.8491973876953126e-05,
      "model_forward_time": 2.5142664909362793,
      "step": 7469
    },
    {
      "epoch": 2.8491973876953126e-05,
      "step": 7469,
      "training_step_time": 7.5409862995147705
    },
    {
      "epoch": 2.8491973876953126e-05,
      "model_forward_time": 3.123098611831665,
      "step": 7469
    },
    {
      "epoch": 2.8491973876953126e-05,
      "step": 7469,
      "training_step_time": 9.880529880523682
    },
    {
      "epoch": 2.8491973876953126e-05,
      "model_forward_time": 3.1304819583892822,
      "step": 7469
    },
    {
      "epoch": 2.8491973876953126e-05,
      "step": 7469,
      "training_step_time": 9.887660026550293
    },
    {
      "epoch": 2.8491973876953126e-05,
      "model_forward_time": 3.424834728240967,
      "step": 7469
    },
    {
      "epoch": 2.8491973876953126e-05,
      "step": 7469,
      "training_step_time": 10.27713656425476
    },
    {
      "epoch": 2.849578857421875e-05,
      "grad_norm": 0.44752275943756104,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7470
    },
    {
      "epoch": 2.849578857421875e-05,
      "model_forward_time": 3.139383316040039,
      "step": 7470
    },
    {
      "dynamics_loss_avg": 0.06679545119404792,
      "epoch": 2.849578857421875e-05,
      "step": 7470
    },
    {
      "action_loss_avg": 0.030930273979902268,
      "epoch": 2.849578857421875e-05,
      "step": 7470
    },
    {
      "epoch": 2.849578857421875e-05,
      "step": 7470,
      "training_step_time": 9.904228448867798
    },
    {
      "epoch": 2.849578857421875e-05,
      "model_forward_time": 3.1306939125061035,
      "step": 7470
    },
    {
      "epoch": 2.849578857421875e-05,
      "step": 7470,
      "training_step_time": 9.895052194595337
    },
    {
      "epoch": 2.849578857421875e-05,
      "model_forward_time": 2.4302279949188232,
      "step": 7470
    },
    {
      "epoch": 2.849578857421875e-05,
      "step": 7470,
      "training_step_time": 7.454545497894287
    },
    {
      "epoch": 2.849578857421875e-05,
      "model_forward_time": 3.129595994949341,
      "step": 7470
    },
    {
      "epoch": 2.849578857421875e-05,
      "step": 7470,
      "training_step_time": 10.288428544998169
    },
    {
      "epoch": 2.8499603271484375e-05,
      "model_forward_time": 2.4439406394958496,
      "step": 7471
    },
    {
      "epoch": 2.8499603271484375e-05,
      "step": 7471,
      "training_step_time": 7.473573684692383
    },
    {
      "epoch": 2.8499603271484375e-05,
      "model_forward_time": 3.378659725189209,
      "step": 7471
    },
    {
      "epoch": 2.8499603271484375e-05,
      "step": 7471,
      "training_step_time": 10.142841815948486
    },
    {
      "epoch": 2.8499603271484375e-05,
      "model_forward_time": 2.430873394012451,
      "step": 7471
    },
    {
      "epoch": 2.8499603271484375e-05,
      "step": 7471,
      "training_step_time": 7.454594612121582
    },
    {
      "epoch": 2.8499603271484375e-05,
      "model_forward_time": 3.1289467811584473,
      "step": 7471
    },
    {
      "epoch": 2.8499603271484375e-05,
      "step": 7471,
      "training_step_time": 10.279065132141113
    },
    {
      "epoch": 2.850341796875e-05,
      "model_forward_time": 3.1782898902893066,
      "step": 7472
    },
    {
      "epoch": 2.850341796875e-05,
      "step": 7472,
      "training_step_time": 9.94110918045044
    },
    {
      "epoch": 2.850341796875e-05,
      "model_forward_time": 3.1356194019317627,
      "step": 7472
    },
    {
      "epoch": 2.850341796875e-05,
      "step": 7472,
      "training_step_time": 9.895843267440796
    },
    {
      "epoch": 2.850341796875e-05,
      "model_forward_time": 3.4039409160614014,
      "step": 7472
    },
    {
      "dynamics_loss_avg": 0.06364070121198892,
      "epoch": 2.850341796875e-05,
      "step": 7472
    },
    {
      "action_loss_avg": 0.026458350755274297,
      "epoch": 2.850341796875e-05,
      "step": 7472
    },
    {
      "epoch": 2.850341796875e-05,
      "step": 7472,
      "training_step_time": 10.167571544647217
    },
    {
      "epoch": 2.850341796875e-05,
      "model_forward_time": 3.1331465244293213,
      "step": 7472
    },
    {
      "epoch": 2.850341796875e-05,
      "step": 7472,
      "training_step_time": 10.290989398956299
    },
    {
      "epoch": 2.8507232666015624e-05,
      "model_forward_time": 3.1652231216430664,
      "step": 7473
    },
    {
      "epoch": 2.8507232666015624e-05,
      "step": 7473,
      "training_step_time": 9.932803392410278
    },
    {
      "epoch": 2.8507232666015624e-05,
      "model_forward_time": 3.1320950984954834,
      "step": 7473
    },
    {
      "epoch": 2.8507232666015624e-05,
      "step": 7473,
      "training_step_time": 9.895387887954712
    },
    {
      "epoch": 2.8507232666015624e-05,
      "model_forward_time": 3.136434316635132,
      "step": 7473
    },
    {
      "epoch": 2.8507232666015624e-05,
      "step": 7473,
      "training_step_time": 9.897309064865112
    },
    {
      "epoch": 2.8507232666015624e-05,
      "model_forward_time": 2.7035372257232666,
      "step": 7473
    },
    {
      "epoch": 2.8507232666015624e-05,
      "step": 7473,
      "training_step_time": 10.279525756835938
    },
    {
      "epoch": 2.851104736328125e-05,
      "model_forward_time": 3.158169984817505,
      "step": 7474
    },
    {
      "epoch": 2.851104736328125e-05,
      "step": 7474,
      "training_step_time": 9.927396297454834
    },
    {
      "epoch": 2.851104736328125e-05,
      "model_forward_time": 3.1319830417633057,
      "step": 7474
    },
    {
      "epoch": 2.851104736328125e-05,
      "step": 7474,
      "training_step_time": 9.899121284484863
    },
    {
      "epoch": 2.851104736328125e-05,
      "model_forward_time": 3.1359853744506836,
      "step": 7474
    },
    {
      "epoch": 2.851104736328125e-05,
      "step": 7474,
      "training_step_time": 9.904179573059082
    },
    {
      "epoch": 2.851104736328125e-05,
      "model_forward_time": 3.1351077556610107,
      "step": 7474
    },
    {
      "epoch": 2.851104736328125e-05,
      "step": 7474,
      "training_step_time": 10.275110006332397
    },
    {
      "epoch": 2.8514862060546874e-05,
      "model_forward_time": 2.7391347885131836,
      "step": 7475
    },
    {
      "dynamics_loss_avg": 0.05192072447389364,
      "epoch": 2.8514862060546874e-05,
      "step": 7475
    },
    {
      "action_loss_avg": 0.019210727512836458,
      "epoch": 2.8514862060546874e-05,
      "step": 7475
    },
    {
      "epoch": 2.8514862060546874e-05,
      "step": 7475,
      "training_step_time": 7.770552635192871
    },
    {
      "epoch": 2.8514862060546874e-05,
      "model_forward_time": 3.12939190864563,
      "step": 7475
    },
    {
      "epoch": 2.8514862060546874e-05,
      "step": 7475,
      "training_step_time": 9.898009538650513
    },
    {
      "epoch": 2.8514862060546874e-05,
      "model_forward_time": 2.4327774047851562,
      "step": 7475
    },
    {
      "epoch": 2.8514862060546874e-05,
      "step": 7475,
      "training_step_time": 7.461062431335449
    },
    {
      "epoch": 2.8514862060546874e-05,
      "model_forward_time": 2.435742139816284,
      "step": 7475
    },
    {
      "epoch": 2.8514862060546874e-05,
      "step": 7475,
      "training_step_time": 10.284137725830078
    },
    {
      "epoch": 2.85186767578125e-05,
      "model_forward_time": 3.1563117504119873,
      "step": 7476
    },
    {
      "epoch": 2.85186767578125e-05,
      "step": 7476,
      "training_step_time": 9.934147596359253
    },
    {
      "epoch": 2.85186767578125e-05,
      "model_forward_time": 3.3818583488464355,
      "step": 7476
    },
    {
      "epoch": 2.85186767578125e-05,
      "step": 7476,
      "training_step_time": 10.144607305526733
    },
    {
      "epoch": 2.85186767578125e-05,
      "model_forward_time": 2.4296364784240723,
      "step": 7476
    },
    {
      "epoch": 2.85186767578125e-05,
      "step": 7476,
      "training_step_time": 7.451833486557007
    },
    {
      "epoch": 2.85186767578125e-05,
      "model_forward_time": 3.129471778869629,
      "step": 7476
    },
    {
      "epoch": 2.85186767578125e-05,
      "step": 7476,
      "training_step_time": 10.274556398391724
    },
    {
      "epoch": 2.8522491455078126e-05,
      "model_forward_time": 3.140916109085083,
      "step": 7477
    },
    {
      "epoch": 2.8522491455078126e-05,
      "step": 7477,
      "training_step_time": 9.908993244171143
    },
    {
      "epoch": 2.8522491455078126e-05,
      "model_forward_time": 3.1338181495666504,
      "step": 7477
    },
    {
      "epoch": 2.8522491455078126e-05,
      "step": 7477,
      "training_step_time": 9.904216527938843
    },
    {
      "epoch": 2.8522491455078126e-05,
      "model_forward_time": 3.421069383621216,
      "step": 7477
    },
    {
      "dynamics_loss_avg": 0.04520734725520015,
      "epoch": 2.8522491455078126e-05,
      "step": 7477
    },
    {
      "action_loss_avg": 0.018137731798924507,
      "epoch": 2.8522491455078126e-05,
      "step": 7477
    },
    {
      "epoch": 2.8522491455078126e-05,
      "step": 7477,
      "training_step_time": 10.192344427108765
    },
    {
      "epoch": 2.8522491455078126e-05,
      "model_forward_time": 2.4337146282196045,
      "step": 7477
    },
    {
      "epoch": 2.8522491455078126e-05,
      "step": 7477,
      "training_step_time": 10.31675672531128
    },
    {
      "epoch": 2.852630615234375e-05,
      "model_forward_time": 2.450385093688965,
      "step": 7478
    },
    {
      "epoch": 2.852630615234375e-05,
      "step": 7478,
      "training_step_time": 7.480648517608643
    },
    {
      "epoch": 2.852630615234375e-05,
      "model_forward_time": 2.4336259365081787,
      "step": 7478
    },
    {
      "epoch": 2.852630615234375e-05,
      "step": 7478,
      "training_step_time": 7.466066360473633
    },
    {
      "epoch": 2.852630615234375e-05,
      "model_forward_time": 2.4334731101989746,
      "step": 7478
    },
    {
      "epoch": 2.852630615234375e-05,
      "step": 7478,
      "training_step_time": 7.4606688022613525
    },
    {
      "epoch": 2.852630615234375e-05,
      "model_forward_time": 3.4016969203948975,
      "step": 7478
    },
    {
      "epoch": 2.852630615234375e-05,
      "step": 7478,
      "training_step_time": 10.264478206634521
    },
    {
      "epoch": 2.8530120849609375e-05,
      "model_forward_time": 3.1434104442596436,
      "step": 7479
    },
    {
      "epoch": 2.8530120849609375e-05,
      "step": 7479,
      "training_step_time": 9.909108400344849
    },
    {
      "epoch": 2.8530120849609375e-05,
      "model_forward_time": 2.426826000213623,
      "step": 7479
    },
    {
      "epoch": 2.8530120849609375e-05,
      "step": 7479,
      "training_step_time": 7.448042392730713
    },
    {
      "epoch": 2.8530120849609375e-05,
      "model_forward_time": 3.1301493644714355,
      "step": 7479
    },
    {
      "epoch": 2.8530120849609375e-05,
      "step": 7479,
      "training_step_time": 9.892900228500366
    },
    {
      "epoch": 2.8530120849609375e-05,
      "model_forward_time": 2.431560754776001,
      "step": 7479
    },
    {
      "epoch": 2.8530120849609375e-05,
      "step": 7479,
      "training_step_time": 10.292847871780396
    },
    {
      "epoch": 2.8533935546875e-05,
      "grad_norm": 0.48838943243026733,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 7480
    },
    {
      "epoch": 2.8533935546875e-05,
      "model_forward_time": 2.443479061126709,
      "step": 7480
    },
    {
      "dynamics_loss_avg": 0.06870334800332785,
      "epoch": 2.8533935546875e-05,
      "step": 7480
    },
    {
      "action_loss_avg": 0.027813705196604133,
      "epoch": 2.8533935546875e-05,
      "step": 7480
    },
    {
      "epoch": 2.8533935546875e-05,
      "step": 7480,
      "training_step_time": 7.462882995605469
    },
    {
      "epoch": 2.8533935546875e-05,
      "model_forward_time": 3.4127745628356934,
      "step": 7480
    },
    {
      "epoch": 2.8533935546875e-05,
      "step": 7480,
      "training_step_time": 10.175562858581543
    },
    {
      "epoch": 2.8533935546875e-05,
      "model_forward_time": 1.744471788406372,
      "step": 7480
    },
    {
      "epoch": 2.8533935546875e-05,
      "step": 7480,
      "training_step_time": 5.186768293380737
    },
    {
      "epoch": 2.8533935546875e-05,
      "model_forward_time": 3.1365158557891846,
      "step": 7480
    },
    {
      "epoch": 2.8533935546875e-05,
      "step": 7480,
      "training_step_time": 10.287925958633423
    },
    {
      "epoch": 2.8537750244140625e-05,
      "model_forward_time": 3.1567575931549072,
      "step": 7481
    },
    {
      "epoch": 2.8537750244140625e-05,
      "step": 7481,
      "training_step_time": 9.922205924987793
    },
    {
      "epoch": 2.8537750244140625e-05,
      "model_forward_time": 3.1340808868408203,
      "step": 7481
    },
    {
      "epoch": 2.8537750244140625e-05,
      "step": 7481,
      "training_step_time": 9.89697790145874
    },
    {
      "epoch": 2.8537750244140625e-05,
      "model_forward_time": 3.400460958480835,
      "step": 7481
    },
    {
      "epoch": 2.8537750244140625e-05,
      "step": 7481,
      "training_step_time": 10.164639711380005
    },
    {
      "epoch": 2.8537750244140625e-05,
      "model_forward_time": 2.434813976287842,
      "step": 7481
    },
    {
      "epoch": 2.8537750244140625e-05,
      "step": 7481,
      "training_step_time": 10.02819275856018
    },
    {
      "epoch": 2.854156494140625e-05,
      "model_forward_time": 2.480442762374878,
      "step": 7482
    },
    {
      "epoch": 2.854156494140625e-05,
      "step": 7482,
      "training_step_time": 7.513167381286621
    },
    {
      "epoch": 2.854156494140625e-05,
      "model_forward_time": 3.132683515548706,
      "step": 7482
    },
    {
      "epoch": 2.854156494140625e-05,
      "step": 7482,
      "training_step_time": 9.896220922470093
    },
    {
      "epoch": 2.854156494140625e-05,
      "model_forward_time": 2.428649425506592,
      "step": 7482
    },
    {
      "dynamics_loss_avg": 0.05936850439757109,
      "epoch": 2.854156494140625e-05,
      "step": 7482
    },
    {
      "action_loss_avg": 0.025370834837667645,
      "epoch": 2.854156494140625e-05,
      "step": 7482
    },
    {
      "epoch": 2.854156494140625e-05,
      "step": 7482,
      "training_step_time": 7.454481363296509
    },
    {
      "epoch": 2.854156494140625e-05,
      "model_forward_time": 3.4162840843200684,
      "step": 7482
    },
    {
      "epoch": 2.854156494140625e-05,
      "step": 7482,
      "training_step_time": 10.275843143463135
    },
    {
      "epoch": 2.8545379638671874e-05,
      "model_forward_time": 2.453066825866699,
      "step": 7483
    },
    {
      "epoch": 2.8545379638671874e-05,
      "step": 7483,
      "training_step_time": 7.47852349281311
    },
    {
      "epoch": 2.8545379638671874e-05,
      "model_forward_time": 2.424895763397217,
      "step": 7483
    },
    {
      "epoch": 2.8545379638671874e-05,
      "step": 7483,
      "training_step_time": 7.4433393478393555
    },
    {
      "epoch": 2.8545379638671874e-05,
      "model_forward_time": 2.4255645275115967,
      "step": 7483
    },
    {
      "epoch": 2.8545379638671874e-05,
      "step": 7483,
      "training_step_time": 7.439567565917969
    },
    {
      "epoch": 2.8545379638671874e-05,
      "model_forward_time": 2.4245266914367676,
      "step": 7483
    },
    {
      "epoch": 2.8545379638671874e-05,
      "step": 7483,
      "training_step_time": 10.019949674606323
    },
    {
      "epoch": 2.85491943359375e-05,
      "model_forward_time": 2.4459125995635986,
      "step": 7484
    },
    {
      "epoch": 2.85491943359375e-05,
      "step": 7484,
      "training_step_time": 7.4644529819488525
    },
    {
      "epoch": 2.85491943359375e-05,
      "model_forward_time": 3.3765664100646973,
      "step": 7484
    },
    {
      "epoch": 2.85491943359375e-05,
      "step": 7484,
      "training_step_time": 10.140331745147705
    },
    {
      "epoch": 2.85491943359375e-05,
      "model_forward_time": 2.4266676902770996,
      "step": 7484
    },
    {
      "epoch": 2.85491943359375e-05,
      "step": 7484,
      "training_step_time": 7.446230888366699
    },
    {
      "epoch": 2.85491943359375e-05,
      "model_forward_time": 2.4297587871551514,
      "step": 7484
    },
    {
      "epoch": 2.85491943359375e-05,
      "step": 7484,
      "training_step_time": 10.288979530334473
    },
    {
      "epoch": 2.8553009033203126e-05,
      "model_forward_time": 3.20200514793396,
      "step": 7485
    },
    {
      "dynamics_loss_avg": 0.055493044294416906,
      "epoch": 2.8553009033203126e-05,
      "step": 7485
    },
    {
      "action_loss_avg": 0.02592846639454365,
      "epoch": 2.8553009033203126e-05,
      "step": 7485
    },
    {
      "epoch": 2.8553009033203126e-05,
      "step": 7485,
      "training_step_time": 9.975101709365845
    },
    {
      "epoch": 2.8553009033203126e-05,
      "model_forward_time": 3.132629632949829,
      "step": 7485
    },
    {
      "epoch": 2.8553009033203126e-05,
      "step": 7485,
      "training_step_time": 9.894599676132202
    },
    {
      "epoch": 2.8553009033203126e-05,
      "model_forward_time": 3.414572238922119,
      "step": 7485
    },
    {
      "epoch": 2.8553009033203126e-05,
      "step": 7485,
      "training_step_time": 10.181631088256836
    },
    {
      "epoch": 2.8553009033203126e-05,
      "model_forward_time": 3.1318771839141846,
      "step": 7485
    },
    {
      "epoch": 2.8553009033203126e-05,
      "step": 7485,
      "training_step_time": 10.283237218856812
    },
    {
      "epoch": 2.855682373046875e-05,
      "model_forward_time": 3.1650946140289307,
      "step": 7486
    },
    {
      "epoch": 2.855682373046875e-05,
      "step": 7486,
      "training_step_time": 9.93363642692566
    },
    {
      "epoch": 2.855682373046875e-05,
      "model_forward_time": 3.13407564163208,
      "step": 7486
    },
    {
      "epoch": 2.855682373046875e-05,
      "step": 7486,
      "training_step_time": 9.906558990478516
    },
    {
      "epoch": 2.855682373046875e-05,
      "model_forward_time": 3.135003089904785,
      "step": 7486
    },
    {
      "epoch": 2.855682373046875e-05,
      "step": 7486,
      "training_step_time": 9.909148931503296
    },
    {
      "epoch": 2.855682373046875e-05,
      "model_forward_time": 3.4039130210876465,
      "step": 7486
    },
    {
      "epoch": 2.855682373046875e-05,
      "step": 7486,
      "training_step_time": 10.274373531341553
    },
    {
      "epoch": 2.8560638427734375e-05,
      "model_forward_time": 2.478914737701416,
      "step": 7487
    },
    {
      "epoch": 2.8560638427734375e-05,
      "step": 7487,
      "training_step_time": 7.514991998672485
    },
    {
      "epoch": 2.8560638427734375e-05,
      "model_forward_time": 2.4318432807922363,
      "step": 7487
    },
    {
      "epoch": 2.8560638427734375e-05,
      "step": 7487,
      "training_step_time": 7.458317279815674
    },
    {
      "epoch": 2.8560638427734375e-05,
      "model_forward_time": 3.1348893642425537,
      "step": 7487
    },
    {
      "dynamics_loss_avg": 0.049486521445214746,
      "epoch": 2.8560638427734375e-05,
      "step": 7487
    },
    {
      "action_loss_avg": 0.02069140370003879,
      "epoch": 2.8560638427734375e-05,
      "step": 7487
    },
    {
      "epoch": 2.8560638427734375e-05,
      "step": 7487,
      "training_step_time": 9.910584926605225
    },
    {
      "epoch": 2.8560638427734375e-05,
      "model_forward_time": 2.4345805644989014,
      "step": 7487
    },
    {
      "epoch": 2.8560638427734375e-05,
      "step": 7487,
      "training_step_time": 10.271403312683105
    },
    {
      "epoch": 2.8564453125e-05,
      "model_forward_time": 2.7700002193450928,
      "step": 7488
    },
    {
      "epoch": 2.8564453125e-05,
      "step": 7488,
      "training_step_time": 7.799286127090454
    },
    {
      "epoch": 2.8564453125e-05,
      "model_forward_time": 3.141555070877075,
      "step": 7488
    },
    {
      "epoch": 2.8564453125e-05,
      "step": 7488,
      "training_step_time": 9.916548728942871
    },
    {
      "epoch": 2.8564453125e-05,
      "model_forward_time": 2.4399421215057373,
      "step": 7488
    },
    {
      "epoch": 2.8564453125e-05,
      "step": 7488,
      "training_step_time": 7.47108268737793
    },
    {
      "epoch": 2.8564453125e-05,
      "model_forward_time": 2.434839963912964,
      "step": 7488
    },
    {
      "epoch": 2.8564453125e-05,
      "step": 7488,
      "training_step_time": 10.026023626327515
    },
    {
      "epoch": 2.8568267822265625e-05,
      "model_forward_time": 2.4768052101135254,
      "step": 7489
    },
    {
      "epoch": 2.8568267822265625e-05,
      "step": 7489,
      "training_step_time": 7.505821704864502
    },
    {
      "epoch": 2.8568267822265625e-05,
      "model_forward_time": 3.1320998668670654,
      "step": 7489
    },
    {
      "epoch": 2.8568267822265625e-05,
      "step": 7489,
      "training_step_time": 10.193102598190308
    },
    {
      "epoch": 2.8568267822265625e-05,
      "model_forward_time": 2.4306609630584717,
      "step": 7489
    },
    {
      "epoch": 2.8568267822265625e-05,
      "step": 7489,
      "training_step_time": 7.460219383239746
    },
    {
      "epoch": 2.8568267822265625e-05,
      "model_forward_time": 3.127227783203125,
      "step": 7489
    },
    {
      "epoch": 2.8568267822265625e-05,
      "step": 7489,
      "training_step_time": 10.281580924987793
    },
    {
      "epoch": 2.857208251953125e-05,
      "grad_norm": 0.31223389506340027,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 7490
    },
    {
      "epoch": 2.857208251953125e-05,
      "model_forward_time": 2.4724080562591553,
      "step": 7490
    },
    {
      "dynamics_loss_avg": 0.0466100187972188,
      "epoch": 2.857208251953125e-05,
      "step": 7490
    },
    {
      "action_loss_avg": 0.017867008596658705,
      "epoch": 2.857208251953125e-05,
      "step": 7490
    },
    {
      "epoch": 2.857208251953125e-05,
      "step": 7490,
      "training_step_time": 7.5082032680511475
    },
    {
      "epoch": 2.857208251953125e-05,
      "model_forward_time": 3.1302778720855713,
      "step": 7490
    },
    {
      "epoch": 2.857208251953125e-05,
      "step": 7490,
      "training_step_time": 9.891947984695435
    },
    {
      "epoch": 2.857208251953125e-05,
      "model_forward_time": 2.4256556034088135,
      "step": 7490
    },
    {
      "epoch": 2.857208251953125e-05,
      "step": 7490,
      "training_step_time": 7.4447715282440186
    },
    {
      "epoch": 2.857208251953125e-05,
      "model_forward_time": 3.395550012588501,
      "step": 7490
    },
    {
      "epoch": 2.857208251953125e-05,
      "step": 7490,
      "training_step_time": 10.298902988433838
    },
    {
      "epoch": 2.8575897216796874e-05,
      "model_forward_time": 2.442687511444092,
      "step": 7491
    },
    {
      "epoch": 2.8575897216796874e-05,
      "step": 7491,
      "training_step_time": 7.4659435749053955
    },
    {
      "epoch": 2.8575897216796874e-05,
      "model_forward_time": 3.125657081604004,
      "step": 7491
    },
    {
      "epoch": 2.8575897216796874e-05,
      "step": 7491,
      "training_step_time": 9.893559217453003
    },
    {
      "epoch": 2.8575897216796874e-05,
      "model_forward_time": 2.4276673793792725,
      "step": 7491
    },
    {
      "epoch": 2.8575897216796874e-05,
      "step": 7491,
      "training_step_time": 7.445621013641357
    },
    {
      "epoch": 2.8575897216796874e-05,
      "model_forward_time": 3.1209094524383545,
      "step": 7491
    },
    {
      "epoch": 2.8575897216796874e-05,
      "step": 7491,
      "training_step_time": 10.00952434539795
    },
    {
      "epoch": 2.85797119140625e-05,
      "model_forward_time": 2.771415948867798,
      "step": 7492
    },
    {
      "epoch": 2.85797119140625e-05,
      "step": 7492,
      "training_step_time": 7.789273023605347
    },
    {
      "epoch": 2.85797119140625e-05,
      "model_forward_time": 3.1221742630004883,
      "step": 7492
    },
    {
      "epoch": 2.85797119140625e-05,
      "step": 7492,
      "training_step_time": 9.879897356033325
    },
    {
      "epoch": 2.85797119140625e-05,
      "model_forward_time": 2.4261293411254883,
      "step": 7492
    },
    {
      "dynamics_loss_avg": 0.05960559155791998,
      "epoch": 2.85797119140625e-05,
      "step": 7492
    },
    {
      "action_loss_avg": 0.02861235924065113,
      "epoch": 2.85797119140625e-05,
      "step": 7492
    },
    {
      "epoch": 2.85797119140625e-05,
      "step": 7492,
      "training_step_time": 7.4432373046875
    },
    {
      "epoch": 2.85797119140625e-05,
      "model_forward_time": 3.1269969940185547,
      "step": 7492
    },
    {
      "epoch": 2.85797119140625e-05,
      "step": 7492,
      "training_step_time": 10.283326864242554
    },
    {
      "epoch": 2.8583526611328126e-05,
      "model_forward_time": 3.137082099914551,
      "step": 7493
    },
    {
      "epoch": 2.8583526611328126e-05,
      "step": 7493,
      "training_step_time": 9.899612188339233
    },
    {
      "epoch": 2.8583526611328126e-05,
      "model_forward_time": 2.700796365737915,
      "step": 7493
    },
    {
      "epoch": 2.8583526611328126e-05,
      "step": 7493,
      "training_step_time": 7.716348886489868
    },
    {
      "epoch": 2.8583526611328126e-05,
      "model_forward_time": 3.1245951652526855,
      "step": 7493
    },
    {
      "epoch": 2.8583526611328126e-05,
      "step": 7493,
      "training_step_time": 9.883358001708984
    },
    {
      "epoch": 2.8583526611328126e-05,
      "model_forward_time": 2.427021026611328,
      "step": 7493
    },
    {
      "epoch": 2.8583526611328126e-05,
      "step": 7493,
      "training_step_time": 10.011077404022217
    },
    {
      "epoch": 2.858734130859375e-05,
      "model_forward_time": 1.7551720142364502,
      "step": 7494
    },
    {
      "epoch": 2.858734130859375e-05,
      "step": 7494,
      "training_step_time": 5.208537340164185
    },
    {
      "epoch": 2.858734130859375e-05,
      "model_forward_time": 2.424823760986328,
      "step": 7494
    },
    {
      "epoch": 2.858734130859375e-05,
      "step": 7494,
      "training_step_time": 7.443174123764038
    },
    {
      "epoch": 2.858734130859375e-05,
      "model_forward_time": 2.426079273223877,
      "step": 7494
    },
    {
      "epoch": 2.858734130859375e-05,
      "step": 7494,
      "training_step_time": 7.441784858703613
    },
    {
      "epoch": 2.858734130859375e-05,
      "model_forward_time": 2.6718826293945312,
      "step": 7494
    },
    {
      "epoch": 2.858734130859375e-05,
      "step": 7494,
      "training_step_time": 10.30305004119873
    },
    {
      "epoch": 2.8591156005859375e-05,
      "model_forward_time": 2.439232587814331,
      "step": 7495
    },
    {
      "dynamics_loss_avg": 0.068973096460104,
      "epoch": 2.8591156005859375e-05,
      "step": 7495
    },
    {
      "action_loss_avg": 0.03518689451739192,
      "epoch": 2.8591156005859375e-05,
      "step": 7495
    },
    {
      "epoch": 2.8591156005859375e-05,
      "step": 7495,
      "training_step_time": 7.459059476852417
    },
    {
      "epoch": 2.8591156005859375e-05,
      "model_forward_time": 3.1312406063079834,
      "step": 7495
    },
    {
      "epoch": 2.8591156005859375e-05,
      "step": 7495,
      "training_step_time": 9.893399715423584
    },
    {
      "epoch": 2.8591156005859375e-05,
      "model_forward_time": 2.427699565887451,
      "step": 7495
    },
    {
      "epoch": 2.8591156005859375e-05,
      "step": 7495,
      "training_step_time": 7.448969125747681
    },
    {
      "epoch": 2.8591156005859375e-05,
      "model_forward_time": 2.4309122562408447,
      "step": 7495
    },
    {
      "epoch": 2.8591156005859375e-05,
      "step": 7495,
      "training_step_time": 10.288279056549072
    },
    {
      "epoch": 2.8594970703125e-05,
      "model_forward_time": 2.7527997493743896,
      "step": 7496
    },
    {
      "epoch": 2.8594970703125e-05,
      "step": 7496,
      "training_step_time": 7.771596431732178
    },
    {
      "epoch": 2.8594970703125e-05,
      "model_forward_time": 3.1284916400909424,
      "step": 7496
    },
    {
      "epoch": 2.8594970703125e-05,
      "step": 7496,
      "training_step_time": 9.896056890487671
    },
    {
      "epoch": 2.8594970703125e-05,
      "model_forward_time": 3.131324052810669,
      "step": 7496
    },
    {
      "epoch": 2.8594970703125e-05,
      "step": 7496,
      "training_step_time": 9.899333238601685
    },
    {
      "epoch": 2.8594970703125e-05,
      "model_forward_time": 3.1369662284851074,
      "step": 7496
    },
    {
      "epoch": 2.8594970703125e-05,
      "step": 7496,
      "training_step_time": 10.25608777999878
    },
    {
      "epoch": 2.8598785400390625e-05,
      "model_forward_time": 3.1813862323760986,
      "step": 7497
    },
    {
      "epoch": 2.8598785400390625e-05,
      "step": 7497,
      "training_step_time": 9.956157684326172
    },
    {
      "epoch": 2.8598785400390625e-05,
      "model_forward_time": 3.131068229675293,
      "step": 7497
    },
    {
      "epoch": 2.8598785400390625e-05,
      "step": 7497,
      "training_step_time": 10.170411348342896
    },
    {
      "epoch": 2.8598785400390625e-05,
      "model_forward_time": 3.1340229511260986,
      "step": 7497
    },
    {
      "dynamics_loss_avg": 0.05538119748234749,
      "epoch": 2.8598785400390625e-05,
      "step": 7497
    },
    {
      "action_loss_avg": 0.021714451629668473,
      "epoch": 2.8598785400390625e-05,
      "step": 7497
    },
    {
      "epoch": 2.8598785400390625e-05,
      "step": 7497,
      "training_step_time": 9.901520490646362
    },
    {
      "epoch": 2.8598785400390625e-05,
      "model_forward_time": 2.4348886013031006,
      "step": 7497
    },
    {
      "epoch": 2.8598785400390625e-05,
      "step": 7497,
      "training_step_time": 10.315941095352173
    },
    {
      "epoch": 2.860260009765625e-05,
      "model_forward_time": 2.4750499725341797,
      "step": 7498
    },
    {
      "epoch": 2.860260009765625e-05,
      "step": 7498,
      "training_step_time": 7.500065803527832
    },
    {
      "epoch": 2.860260009765625e-05,
      "model_forward_time": 2.4349982738494873,
      "step": 7498
    },
    {
      "epoch": 2.860260009765625e-05,
      "step": 7498,
      "training_step_time": 7.462892055511475
    },
    {
      "epoch": 2.860260009765625e-05,
      "model_forward_time": 3.145787477493286,
      "step": 7498
    },
    {
      "epoch": 2.860260009765625e-05,
      "step": 7498,
      "training_step_time": 9.915741920471191
    },
    {
      "epoch": 2.860260009765625e-05,
      "model_forward_time": 3.424669027328491,
      "step": 7498
    },
    {
      "epoch": 2.860260009765625e-05,
      "step": 7498,
      "training_step_time": 10.288528203964233
    },
    {
      "epoch": 2.8606414794921874e-05,
      "model_forward_time": 3.220945119857788,
      "step": 7499
    },
    {
      "epoch": 2.8606414794921874e-05,
      "step": 7499,
      "training_step_time": 9.99623727798462
    },
    {
      "epoch": 2.8606414794921874e-05,
      "model_forward_time": 2.4402031898498535,
      "step": 7499
    },
    {
      "epoch": 2.8606414794921874e-05,
      "step": 7499,
      "training_step_time": 7.466150999069214
    },
    {
      "epoch": 2.8606414794921874e-05,
      "model_forward_time": 3.1374642848968506,
      "step": 7499
    },
    {
      "epoch": 2.8606414794921874e-05,
      "step": 7499,
      "training_step_time": 9.90403962135315
    },
    {
      "epoch": 2.8606414794921874e-05,
      "model_forward_time": 3.1398274898529053,
      "step": 7499
    },
    {
      "epoch": 2.8606414794921874e-05,
      "step": 7499,
      "training_step_time": 10.285014390945435
    },
    {
      "epoch": 2.86102294921875e-05,
      "grad_norm": 0.3118495047092438,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 7500
    },
    {
      "epoch": 2.86102294921875e-05,
      "model_forward_time": 3.1759281158447266,
      "step": 7500
    },
    {
      "dynamics_loss_avg": 0.0574985571205616,
      "epoch": 2.86102294921875e-05,
      "step": 7500
    },
    {
      "action_loss_avg": 0.025887738214805722,
      "epoch": 2.86102294921875e-05,
      "step": 7500
    },
    {
      "epoch": 2.86102294921875e-05,
      "step": 7500,
      "training_step_time": 9.924991130828857
    },
    {
      "epoch": 2.86102294921875e-05,
      "model_forward_time": 3.126582145690918,
      "step": 7500
    },
    {
      "epoch": 2.86102294921875e-05,
      "step": 7500,
      "training_step_time": 9.880455493927002
    },
    {
      "epoch": 2.86102294921875e-05,
      "model_forward_time": 3.1341497898101807,
      "step": 7500
    },
    {
      "epoch": 2.86102294921875e-05,
      "step": 7500,
      "training_step_time": 9.890799283981323
    },
    {
      "epoch": 2.86102294921875e-05,
      "model_forward_time": 3.1362907886505127,
      "step": 7500
    },
    {
      "epoch": 2.86102294921875e-05,
      "step": 7500,
      "training_step_time": 11.931772232055664
    },
    {
      "epoch": 2.8614044189453126e-05,
      "model_forward_time": 3.169449806213379,
      "step": 7501
    },
    {
      "epoch": 2.8614044189453126e-05,
      "step": 7501,
      "training_step_time": 9.938252449035645
    },
    {
      "epoch": 2.8614044189453126e-05,
      "model_forward_time": 3.1343507766723633,
      "step": 7501
    },
    {
      "epoch": 2.8614044189453126e-05,
      "step": 7501,
      "training_step_time": 9.912189722061157
    },
    {
      "epoch": 2.8614044189453126e-05,
      "model_forward_time": 2.439831256866455,
      "step": 7501
    },
    {
      "epoch": 2.8614044189453126e-05,
      "step": 7501,
      "training_step_time": 7.465437889099121
    },
    {
      "epoch": 2.8614044189453126e-05,
      "model_forward_time": 3.1387929916381836,
      "step": 7501
    },
    {
      "epoch": 2.8614044189453126e-05,
      "step": 7501,
      "training_step_time": 10.021850347518921
    },
    {
      "epoch": 2.861785888671875e-05,
      "model_forward_time": 2.484844446182251,
      "step": 7502
    },
    {
      "epoch": 2.861785888671875e-05,
      "step": 7502,
      "training_step_time": 7.51504921913147
    },
    {
      "epoch": 2.861785888671875e-05,
      "model_forward_time": 3.398300886154175,
      "step": 7502
    },
    {
      "epoch": 2.861785888671875e-05,
      "step": 7502,
      "training_step_time": 10.17690372467041
    },
    {
      "epoch": 2.861785888671875e-05,
      "model_forward_time": 3.14805006980896,
      "step": 7502
    },
    {
      "dynamics_loss_avg": 0.05122173596173525,
      "epoch": 2.861785888671875e-05,
      "step": 7502
    },
    {
      "action_loss_avg": 0.025042641302570702,
      "epoch": 2.861785888671875e-05,
      "step": 7502
    },
    {
      "epoch": 2.861785888671875e-05,
      "step": 7502,
      "training_step_time": 9.930691003799438
    },
    {
      "epoch": 2.861785888671875e-05,
      "model_forward_time": 3.145146369934082,
      "step": 7502
    },
    {
      "epoch": 2.861785888671875e-05,
      "step": 7502,
      "training_step_time": 10.330800533294678
    },
    {
      "epoch": 2.8621673583984375e-05,
      "model_forward_time": 3.1500587463378906,
      "step": 7503
    },
    {
      "epoch": 2.8621673583984375e-05,
      "step": 7503,
      "training_step_time": 9.931031703948975
    },
    {
      "epoch": 2.8621673583984375e-05,
      "model_forward_time": 3.1451144218444824,
      "step": 7503
    },
    {
      "epoch": 2.8621673583984375e-05,
      "step": 7503,
      "training_step_time": 10.21685004234314
    },
    {
      "epoch": 2.8621673583984375e-05,
      "model_forward_time": 2.4382548332214355,
      "step": 7503
    },
    {
      "epoch": 2.8621673583984375e-05,
      "step": 7503,
      "training_step_time": 7.4728944301605225
    },
    {
      "epoch": 2.8621673583984375e-05,
      "model_forward_time": 3.139481782913208,
      "step": 7503
    },
    {
      "epoch": 2.8621673583984375e-05,
      "step": 7503,
      "training_step_time": 10.319563388824463
    },
    {
      "epoch": 2.862548828125e-05,
      "model_forward_time": 3.173725128173828,
      "step": 7504
    },
    {
      "epoch": 2.862548828125e-05,
      "step": 7504,
      "training_step_time": 9.947792768478394
    },
    {
      "epoch": 2.862548828125e-05,
      "model_forward_time": 3.144160270690918,
      "step": 7504
    },
    {
      "epoch": 2.862548828125e-05,
      "step": 7504,
      "training_step_time": 9.921922206878662
    },
    {
      "epoch": 2.862548828125e-05,
      "model_forward_time": 2.438248634338379,
      "step": 7504
    },
    {
      "epoch": 2.862548828125e-05,
      "step": 7504,
      "training_step_time": 7.469823598861694
    },
    {
      "epoch": 2.862548828125e-05,
      "model_forward_time": 3.396780252456665,
      "step": 7504
    },
    {
      "epoch": 2.862548828125e-05,
      "step": 7504,
      "training_step_time": 10.264863014221191
    },
    {
      "epoch": 2.8629302978515625e-05,
      "model_forward_time": 3.193476676940918,
      "step": 7505
    },
    {
      "dynamics_loss_avg": 0.054578579217195514,
      "epoch": 2.8629302978515625e-05,
      "step": 7505
    },
    {
      "action_loss_avg": 0.023038186971098183,
      "epoch": 2.8629302978515625e-05,
      "step": 7505
    },
    {
      "epoch": 2.8629302978515625e-05,
      "step": 7505,
      "training_step_time": 9.973710298538208
    },
    {
      "epoch": 2.8629302978515625e-05,
      "model_forward_time": 3.1471128463745117,
      "step": 7505
    },
    {
      "epoch": 2.8629302978515625e-05,
      "step": 7505,
      "training_step_time": 9.926025867462158
    },
    {
      "epoch": 2.8629302978515625e-05,
      "model_forward_time": 3.1437015533447266,
      "step": 7505
    },
    {
      "epoch": 2.8629302978515625e-05,
      "step": 7505,
      "training_step_time": 9.916480779647827
    },
    {
      "epoch": 2.8629302978515625e-05,
      "model_forward_time": 3.1490423679351807,
      "step": 7505
    },
    {
      "epoch": 2.8629302978515625e-05,
      "step": 7505,
      "training_step_time": 10.305302619934082
    },
    {
      "epoch": 2.863311767578125e-05,
      "model_forward_time": 3.1572728157043457,
      "step": 7506
    },
    {
      "epoch": 2.863311767578125e-05,
      "step": 7506,
      "training_step_time": 9.933669805526733
    },
    {
      "epoch": 2.863311767578125e-05,
      "model_forward_time": 3.147886037826538,
      "step": 7506
    },
    {
      "epoch": 2.863311767578125e-05,
      "step": 7506,
      "training_step_time": 9.918342351913452
    },
    {
      "epoch": 2.863311767578125e-05,
      "model_forward_time": 3.1448347568511963,
      "step": 7506
    },
    {
      "epoch": 2.863311767578125e-05,
      "step": 7506,
      "training_step_time": 9.916587591171265
    },
    {
      "epoch": 2.863311767578125e-05,
      "model_forward_time": 2.442699909210205,
      "step": 7506
    },
    {
      "epoch": 2.863311767578125e-05,
      "step": 7506,
      "training_step_time": 10.283787250518799
    },
    {
      "epoch": 2.8636932373046874e-05,
      "model_forward_time": 3.18489670753479,
      "step": 7507
    },
    {
      "epoch": 2.8636932373046874e-05,
      "step": 7507,
      "training_step_time": 9.955551385879517
    },
    {
      "epoch": 2.8636932373046874e-05,
      "model_forward_time": 3.4204602241516113,
      "step": 7507
    },
    {
      "epoch": 2.8636932373046874e-05,
      "step": 7507,
      "training_step_time": 10.185898542404175
    },
    {
      "epoch": 2.8636932373046874e-05,
      "model_forward_time": 3.1379153728485107,
      "step": 7507
    },
    {
      "dynamics_loss_avg": 0.05880323555320501,
      "epoch": 2.8636932373046874e-05,
      "step": 7507
    },
    {
      "action_loss_avg": 0.022013161797076464,
      "epoch": 2.8636932373046874e-05,
      "step": 7507
    },
    {
      "epoch": 2.8636932373046874e-05,
      "step": 7507,
      "training_step_time": 9.904114007949829
    },
    {
      "epoch": 2.8636932373046874e-05,
      "model_forward_time": 3.1340415477752686,
      "step": 7507
    },
    {
      "epoch": 2.8636932373046874e-05,
      "step": 7507,
      "training_step_time": 10.246574640274048
    },
    {
      "epoch": 2.86407470703125e-05,
      "model_forward_time": 3.1805098056793213,
      "step": 7508
    },
    {
      "epoch": 2.86407470703125e-05,
      "step": 7508,
      "training_step_time": 9.950309038162231
    },
    {
      "epoch": 2.86407470703125e-05,
      "model_forward_time": 3.1365861892700195,
      "step": 7508
    },
    {
      "epoch": 2.86407470703125e-05,
      "step": 7508,
      "training_step_time": 10.175848960876465
    },
    {
      "epoch": 2.86407470703125e-05,
      "model_forward_time": 3.1338095664978027,
      "step": 7508
    },
    {
      "epoch": 2.86407470703125e-05,
      "step": 7508,
      "training_step_time": 9.899015188217163
    },
    {
      "epoch": 2.86407470703125e-05,
      "model_forward_time": 3.134653091430664,
      "step": 7508
    },
    {
      "epoch": 2.86407470703125e-05,
      "step": 7508,
      "training_step_time": 10.315047025680542
    },
    {
      "epoch": 2.8644561767578126e-05,
      "model_forward_time": 3.1749978065490723,
      "step": 7509
    },
    {
      "epoch": 2.8644561767578126e-05,
      "step": 7509,
      "training_step_time": 9.942425966262817
    },
    {
      "epoch": 2.8644561767578126e-05,
      "model_forward_time": 3.130051851272583,
      "step": 7509
    },
    {
      "epoch": 2.8644561767578126e-05,
      "step": 7509,
      "training_step_time": 9.901917219161987
    },
    {
      "epoch": 2.8644561767578126e-05,
      "model_forward_time": 3.4021284580230713,
      "step": 7509
    },
    {
      "epoch": 2.8644561767578126e-05,
      "step": 7509,
      "training_step_time": 10.171047925949097
    },
    {
      "epoch": 2.8644561767578126e-05,
      "model_forward_time": 3.13211727142334,
      "step": 7509
    },
    {
      "epoch": 2.8644561767578126e-05,
      "step": 7509,
      "training_step_time": 10.263998746871948
    },
    {
      "epoch": 2.864837646484375e-05,
      "grad_norm": 0.7755390405654907,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 7510
    },
    {
      "epoch": 2.864837646484375e-05,
      "model_forward_time": 3.1733245849609375,
      "step": 7510
    },
    {
      "dynamics_loss_avg": 0.05101885981857777,
      "epoch": 2.864837646484375e-05,
      "step": 7510
    },
    {
      "action_loss_avg": 0.0200759134022519,
      "epoch": 2.864837646484375e-05,
      "step": 7510
    },
    {
      "epoch": 2.864837646484375e-05,
      "step": 7510,
      "training_step_time": 9.94447922706604
    },
    {
      "epoch": 2.864837646484375e-05,
      "model_forward_time": 3.1336116790771484,
      "step": 7510
    },
    {
      "epoch": 2.864837646484375e-05,
      "step": 7510,
      "training_step_time": 9.898685216903687
    },
    {
      "epoch": 2.864837646484375e-05,
      "model_forward_time": 3.1284284591674805,
      "step": 7510
    },
    {
      "epoch": 2.864837646484375e-05,
      "step": 7510,
      "training_step_time": 9.897513151168823
    },
    {
      "epoch": 2.864837646484375e-05,
      "model_forward_time": 3.129838466644287,
      "step": 7510
    },
    {
      "epoch": 2.864837646484375e-05,
      "step": 7510,
      "training_step_time": 10.265647649765015
    },
    {
      "epoch": 2.8652191162109375e-05,
      "model_forward_time": 3.1853091716766357,
      "step": 7511
    },
    {
      "epoch": 2.8652191162109375e-05,
      "step": 7511,
      "training_step_time": 9.947904348373413
    },
    {
      "epoch": 2.8652191162109375e-05,
      "model_forward_time": 3.128983736038208,
      "step": 7511
    },
    {
      "epoch": 2.8652191162109375e-05,
      "step": 7511,
      "training_step_time": 9.892826080322266
    },
    {
      "epoch": 2.8652191162109375e-05,
      "model_forward_time": 3.1312384605407715,
      "step": 7511
    },
    {
      "epoch": 2.8652191162109375e-05,
      "step": 7511,
      "training_step_time": 9.892657041549683
    },
    {
      "epoch": 2.8652191162109375e-05,
      "model_forward_time": 3.1270320415496826,
      "step": 7511
    },
    {
      "epoch": 2.8652191162109375e-05,
      "step": 7511,
      "training_step_time": 10.285801410675049
    },
    {
      "epoch": 2.8656005859375e-05,
      "model_forward_time": 3.449962615966797,
      "step": 7512
    },
    {
      "epoch": 2.8656005859375e-05,
      "step": 7512,
      "training_step_time": 10.213435649871826
    },
    {
      "epoch": 2.8656005859375e-05,
      "model_forward_time": 3.133713960647583,
      "step": 7512
    },
    {
      "epoch": 2.8656005859375e-05,
      "step": 7512,
      "training_step_time": 9.89799690246582
    },
    {
      "epoch": 2.8656005859375e-05,
      "model_forward_time": 3.1310038566589355,
      "step": 7512
    },
    {
      "dynamics_loss_avg": 0.07975331544876099,
      "epoch": 2.8656005859375e-05,
      "step": 7512
    },
    {
      "action_loss_avg": 0.028687769919633864,
      "epoch": 2.8656005859375e-05,
      "step": 7512
    },
    {
      "epoch": 2.8656005859375e-05,
      "step": 7512,
      "training_step_time": 9.93824577331543
    },
    {
      "epoch": 2.8656005859375e-05,
      "model_forward_time": 3.1320250034332275,
      "step": 7512
    },
    {
      "epoch": 2.8656005859375e-05,
      "step": 7512,
      "training_step_time": 10.309728860855103
    },
    {
      "epoch": 2.8659820556640625e-05,
      "model_forward_time": 3.18473219871521,
      "step": 7513
    },
    {
      "epoch": 2.8659820556640625e-05,
      "step": 7513,
      "training_step_time": 9.951142072677612
    },
    {
      "epoch": 2.8659820556640625e-05,
      "model_forward_time": 3.392230987548828,
      "step": 7513
    },
    {
      "epoch": 2.8659820556640625e-05,
      "step": 7513,
      "training_step_time": 10.156314611434937
    },
    {
      "epoch": 2.8659820556640625e-05,
      "model_forward_time": 3.135385274887085,
      "step": 7513
    },
    {
      "epoch": 2.8659820556640625e-05,
      "step": 7513,
      "training_step_time": 9.897319793701172
    },
    {
      "epoch": 2.8659820556640625e-05,
      "model_forward_time": 3.131723642349243,
      "step": 7513
    },
    {
      "epoch": 2.8659820556640625e-05,
      "step": 7513,
      "training_step_time": 10.273398876190186
    },
    {
      "epoch": 2.866363525390625e-05,
      "model_forward_time": 3.161375045776367,
      "step": 7514
    },
    {
      "epoch": 2.866363525390625e-05,
      "step": 7514,
      "training_step_time": 9.926734447479248
    },
    {
      "epoch": 2.866363525390625e-05,
      "model_forward_time": 3.1329288482666016,
      "step": 7514
    },
    {
      "epoch": 2.866363525390625e-05,
      "step": 7514,
      "training_step_time": 9.894417762756348
    },
    {
      "epoch": 2.866363525390625e-05,
      "model_forward_time": 3.4109814167022705,
      "step": 7514
    },
    {
      "epoch": 2.866363525390625e-05,
      "step": 7514,
      "training_step_time": 10.177732229232788
    },
    {
      "epoch": 2.866363525390625e-05,
      "model_forward_time": 3.133547306060791,
      "step": 7514
    },
    {
      "epoch": 2.866363525390625e-05,
      "step": 7514,
      "training_step_time": 10.269963502883911
    },
    {
      "epoch": 2.8667449951171874e-05,
      "model_forward_time": 3.161670446395874,
      "step": 7515
    },
    {
      "dynamics_loss_avg": 0.050366236735135315,
      "epoch": 2.8667449951171874e-05,
      "step": 7515
    },
    {
      "action_loss_avg": 0.020067567005753517,
      "epoch": 2.8667449951171874e-05,
      "step": 7515
    },
    {
      "epoch": 2.8667449951171874e-05,
      "step": 7515,
      "training_step_time": 9.93399977684021
    },
    {
      "epoch": 2.8667449951171874e-05,
      "model_forward_time": 3.138291597366333,
      "step": 7515
    },
    {
      "epoch": 2.8667449951171874e-05,
      "step": 7515,
      "training_step_time": 9.90162205696106
    },
    {
      "epoch": 2.8667449951171874e-05,
      "model_forward_time": 3.132840394973755,
      "step": 7515
    },
    {
      "epoch": 2.8667449951171874e-05,
      "step": 7515,
      "training_step_time": 9.906894445419312
    },
    {
      "epoch": 2.8667449951171874e-05,
      "model_forward_time": 3.4252984523773193,
      "step": 7515
    },
    {
      "epoch": 2.8667449951171874e-05,
      "step": 7515,
      "training_step_time": 10.310271501541138
    },
    {
      "epoch": 2.86712646484375e-05,
      "model_forward_time": 3.1754720211029053,
      "step": 7516
    },
    {
      "epoch": 2.86712646484375e-05,
      "step": 7516,
      "training_step_time": 9.948675394058228
    },
    {
      "epoch": 2.86712646484375e-05,
      "model_forward_time": 3.1482033729553223,
      "step": 7516
    },
    {
      "epoch": 2.86712646484375e-05,
      "step": 7516,
      "training_step_time": 9.928750276565552
    },
    {
      "epoch": 2.86712646484375e-05,
      "model_forward_time": 3.1454310417175293,
      "step": 7516
    },
    {
      "epoch": 2.86712646484375e-05,
      "step": 7516,
      "training_step_time": 9.91912579536438
    },
    {
      "epoch": 2.86712646484375e-05,
      "model_forward_time": 3.141552209854126,
      "step": 7516
    },
    {
      "epoch": 2.86712646484375e-05,
      "step": 7516,
      "training_step_time": 10.274348020553589
    },
    {
      "epoch": 2.8675079345703126e-05,
      "model_forward_time": 3.4426944255828857,
      "step": 7517
    },
    {
      "epoch": 2.8675079345703126e-05,
      "step": 7517,
      "training_step_time": 10.21522831916809
    },
    {
      "epoch": 2.8675079345703126e-05,
      "model_forward_time": 3.141878604888916,
      "step": 7517
    },
    {
      "epoch": 2.8675079345703126e-05,
      "step": 7517,
      "training_step_time": 9.912527084350586
    },
    {
      "epoch": 2.8675079345703126e-05,
      "model_forward_time": 3.1373863220214844,
      "step": 7517
    },
    {
      "dynamics_loss_avg": 0.05072878021746874,
      "epoch": 2.8675079345703126e-05,
      "step": 7517
    },
    {
      "action_loss_avg": 0.0211487696506083,
      "epoch": 2.8675079345703126e-05,
      "step": 7517
    },
    {
      "epoch": 2.8675079345703126e-05,
      "step": 7517,
      "training_step_time": 9.913525104522705
    },
    {
      "epoch": 2.8675079345703126e-05,
      "model_forward_time": 3.138184070587158,
      "step": 7517
    },
    {
      "epoch": 2.8675079345703126e-05,
      "step": 7517,
      "training_step_time": 10.22750449180603
    },
    {
      "epoch": 2.867889404296875e-05,
      "model_forward_time": 3.2021608352661133,
      "step": 7518
    },
    {
      "epoch": 2.867889404296875e-05,
      "step": 7518,
      "training_step_time": 9.987902641296387
    },
    {
      "epoch": 2.867889404296875e-05,
      "model_forward_time": 3.4315991401672363,
      "step": 7518
    },
    {
      "epoch": 2.867889404296875e-05,
      "step": 7518,
      "training_step_time": 10.206083059310913
    },
    {
      "epoch": 2.867889404296875e-05,
      "model_forward_time": 3.1487338542938232,
      "step": 7518
    },
    {
      "epoch": 2.867889404296875e-05,
      "step": 7518,
      "training_step_time": 9.922811508178711
    },
    {
      "epoch": 2.867889404296875e-05,
      "model_forward_time": 3.1519296169281006,
      "step": 7518
    },
    {
      "epoch": 2.867889404296875e-05,
      "step": 7518,
      "training_step_time": 10.279713869094849
    },
    {
      "epoch": 2.8682708740234375e-05,
      "model_forward_time": 3.2066962718963623,
      "step": 7519
    },
    {
      "epoch": 2.8682708740234375e-05,
      "step": 7519,
      "training_step_time": 9.984797477722168
    },
    {
      "epoch": 2.8682708740234375e-05,
      "model_forward_time": 3.1461851596832275,
      "step": 7519
    },
    {
      "epoch": 2.8682708740234375e-05,
      "step": 7519,
      "training_step_time": 9.919400691986084
    },
    {
      "epoch": 2.8682708740234375e-05,
      "model_forward_time": 3.410520076751709,
      "step": 7519
    },
    {
      "epoch": 2.8682708740234375e-05,
      "step": 7519,
      "training_step_time": 10.181838512420654
    },
    {
      "epoch": 2.8682708740234375e-05,
      "model_forward_time": 3.144923210144043,
      "step": 7519
    },
    {
      "epoch": 2.8682708740234375e-05,
      "step": 7519,
      "training_step_time": 10.329165697097778
    },
    {
      "epoch": 2.86865234375e-05,
      "grad_norm": 0.7212926745414734,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 7520
    },
    {
      "epoch": 2.86865234375e-05,
      "model_forward_time": 3.1842598915100098,
      "step": 7520
    },
    {
      "dynamics_loss_avg": 0.038578327745199203,
      "epoch": 2.86865234375e-05,
      "step": 7520
    },
    {
      "action_loss_avg": 0.02164608840830624,
      "epoch": 2.86865234375e-05,
      "step": 7520
    },
    {
      "epoch": 2.86865234375e-05,
      "step": 7520,
      "training_step_time": 9.958057641983032
    },
    {
      "epoch": 2.86865234375e-05,
      "model_forward_time": 3.1401445865631104,
      "step": 7520
    },
    {
      "epoch": 2.86865234375e-05,
      "step": 7520,
      "training_step_time": 9.910152196884155
    },
    {
      "epoch": 2.86865234375e-05,
      "model_forward_time": 3.149073600769043,
      "step": 7520
    },
    {
      "epoch": 2.86865234375e-05,
      "step": 7520,
      "training_step_time": 9.92623233795166
    },
    {
      "epoch": 2.86865234375e-05,
      "model_forward_time": 3.4266209602355957,
      "step": 7520
    },
    {
      "epoch": 2.86865234375e-05,
      "step": 7520,
      "training_step_time": 10.296595096588135
    },
    {
      "epoch": 2.8690338134765625e-05,
      "model_forward_time": 3.198540687561035,
      "step": 7521
    },
    {
      "epoch": 2.8690338134765625e-05,
      "step": 7521,
      "training_step_time": 9.971878290176392
    },
    {
      "epoch": 2.8690338134765625e-05,
      "model_forward_time": 2.442887306213379,
      "step": 7521
    },
    {
      "epoch": 2.8690338134765625e-05,
      "step": 7521,
      "training_step_time": 7.474296808242798
    },
    {
      "epoch": 2.8690338134765625e-05,
      "model_forward_time": 3.1446847915649414,
      "step": 7521
    },
    {
      "epoch": 2.8690338134765625e-05,
      "step": 7521,
      "training_step_time": 9.919359683990479
    },
    {
      "epoch": 2.8690338134765625e-05,
      "model_forward_time": 3.142988681793213,
      "step": 7521
    },
    {
      "epoch": 2.8690338134765625e-05,
      "step": 7521,
      "training_step_time": 10.016589403152466
    },
    {
      "epoch": 2.869415283203125e-05,
      "model_forward_time": 3.4674158096313477,
      "step": 7522
    },
    {
      "epoch": 2.869415283203125e-05,
      "step": 7522,
      "training_step_time": 10.252636194229126
    },
    {
      "epoch": 2.869415283203125e-05,
      "model_forward_time": 3.1499087810516357,
      "step": 7522
    },
    {
      "epoch": 2.869415283203125e-05,
      "step": 7522,
      "training_step_time": 9.92290711402893
    },
    {
      "epoch": 2.869415283203125e-05,
      "model_forward_time": 3.151059865951538,
      "step": 7522
    },
    {
      "dynamics_loss_avg": 0.051541604474186896,
      "epoch": 2.869415283203125e-05,
      "step": 7522
    },
    {
      "action_loss_avg": 0.022355897491797803,
      "epoch": 2.869415283203125e-05,
      "step": 7522
    },
    {
      "epoch": 2.869415283203125e-05,
      "step": 7522,
      "training_step_time": 9.927908420562744
    },
    {
      "epoch": 2.869415283203125e-05,
      "model_forward_time": 3.1491312980651855,
      "step": 7522
    },
    {
      "epoch": 2.869415283203125e-05,
      "step": 7522,
      "training_step_time": 10.271583557128906
    },
    {
      "epoch": 2.8697967529296874e-05,
      "model_forward_time": 3.1797690391540527,
      "step": 7523
    },
    {
      "epoch": 2.8697967529296874e-05,
      "step": 7523,
      "training_step_time": 9.952220916748047
    },
    {
      "epoch": 2.8697967529296874e-05,
      "model_forward_time": 3.394306182861328,
      "step": 7523
    },
    {
      "epoch": 2.8697967529296874e-05,
      "step": 7523,
      "training_step_time": 10.162317276000977
    },
    {
      "epoch": 2.8697967529296874e-05,
      "model_forward_time": 3.1459224224090576,
      "step": 7523
    },
    {
      "epoch": 2.8697967529296874e-05,
      "step": 7523,
      "training_step_time": 9.916428804397583
    },
    {
      "epoch": 2.8697967529296874e-05,
      "model_forward_time": 3.140888214111328,
      "step": 7523
    },
    {
      "epoch": 2.8697967529296874e-05,
      "step": 7523,
      "training_step_time": 10.266491413116455
    },
    {
      "epoch": 2.87017822265625e-05,
      "model_forward_time": 3.1831421852111816,
      "step": 7524
    },
    {
      "epoch": 2.87017822265625e-05,
      "step": 7524,
      "training_step_time": 9.952266693115234
    },
    {
      "epoch": 2.87017822265625e-05,
      "model_forward_time": 3.1416666507720947,
      "step": 7524
    },
    {
      "epoch": 2.87017822265625e-05,
      "step": 7524,
      "training_step_time": 9.908214330673218
    },
    {
      "epoch": 2.87017822265625e-05,
      "model_forward_time": 3.440509080886841,
      "step": 7524
    },
    {
      "epoch": 2.87017822265625e-05,
      "step": 7524,
      "training_step_time": 10.20565128326416
    },
    {
      "epoch": 2.87017822265625e-05,
      "model_forward_time": 3.1429615020751953,
      "step": 7524
    },
    {
      "epoch": 2.87017822265625e-05,
      "step": 7524,
      "training_step_time": 10.02055311203003
    },
    {
      "epoch": 2.8705596923828126e-05,
      "model_forward_time": 3.1889405250549316,
      "step": 7525
    },
    {
      "dynamics_loss_avg": 0.04778496120125055,
      "epoch": 2.8705596923828126e-05,
      "step": 7525
    },
    {
      "action_loss_avg": 0.016762733832001685,
      "epoch": 2.8705596923828126e-05,
      "step": 7525
    },
    {
      "epoch": 2.8705596923828126e-05,
      "step": 7525,
      "training_step_time": 9.963905334472656
    },
    {
      "epoch": 2.8705596923828126e-05,
      "model_forward_time": 3.1419150829315186,
      "step": 7525
    },
    {
      "epoch": 2.8705596923828126e-05,
      "step": 7525,
      "training_step_time": 9.907748460769653
    },
    {
      "epoch": 2.8705596923828126e-05,
      "model_forward_time": 3.1355271339416504,
      "step": 7525
    },
    {
      "epoch": 2.8705596923828126e-05,
      "step": 7525,
      "training_step_time": 10.174688577651978
    },
    {
      "epoch": 2.8705596923828126e-05,
      "model_forward_time": 3.1389052867889404,
      "step": 7525
    },
    {
      "epoch": 2.8705596923828126e-05,
      "step": 7525,
      "training_step_time": 10.233746767044067
    },
    {
      "epoch": 2.870941162109375e-05,
      "model_forward_time": 3.171783447265625,
      "step": 7526
    },
    {
      "epoch": 2.870941162109375e-05,
      "step": 7526,
      "training_step_time": 9.942376375198364
    },
    {
      "epoch": 2.870941162109375e-05,
      "model_forward_time": 3.145174264907837,
      "step": 7526
    },
    {
      "epoch": 2.870941162109375e-05,
      "step": 7526,
      "training_step_time": 9.923918008804321
    },
    {
      "epoch": 2.870941162109375e-05,
      "model_forward_time": 3.145158290863037,
      "step": 7526
    },
    {
      "epoch": 2.870941162109375e-05,
      "step": 7526,
      "training_step_time": 9.915289878845215
    },
    {
      "epoch": 2.870941162109375e-05,
      "model_forward_time": 3.4036526679992676,
      "step": 7526
    },
    {
      "epoch": 2.870941162109375e-05,
      "step": 7526,
      "training_step_time": 10.268746614456177
    },
    {
      "epoch": 2.8713226318359376e-05,
      "model_forward_time": 3.166916847229004,
      "step": 7527
    },
    {
      "epoch": 2.8713226318359376e-05,
      "step": 7527,
      "training_step_time": 9.940388202667236
    },
    {
      "epoch": 2.8713226318359376e-05,
      "model_forward_time": 3.148480176925659,
      "step": 7527
    },
    {
      "epoch": 2.8713226318359376e-05,
      "step": 7527,
      "training_step_time": 9.923860788345337
    },
    {
      "epoch": 2.8713226318359376e-05,
      "model_forward_time": 3.15041446685791,
      "step": 7527
    },
    {
      "dynamics_loss_avg": 0.06032378561794758,
      "epoch": 2.8713226318359376e-05,
      "step": 7527
    },
    {
      "action_loss_avg": 0.023299374058842658,
      "epoch": 2.8713226318359376e-05,
      "step": 7527
    },
    {
      "epoch": 2.8713226318359376e-05,
      "step": 7527,
      "training_step_time": 9.929045915603638
    },
    {
      "epoch": 2.8713226318359376e-05,
      "model_forward_time": 3.151125192642212,
      "step": 7527
    },
    {
      "epoch": 2.8713226318359376e-05,
      "step": 7527,
      "training_step_time": 10.314503908157349
    },
    {
      "epoch": 2.8717041015625e-05,
      "model_forward_time": 3.4481091499328613,
      "step": 7528
    },
    {
      "epoch": 2.8717041015625e-05,
      "step": 7528,
      "training_step_time": 10.224870443344116
    },
    {
      "epoch": 2.8717041015625e-05,
      "model_forward_time": 3.1488938331604004,
      "step": 7528
    },
    {
      "epoch": 2.8717041015625e-05,
      "step": 7528,
      "training_step_time": 9.921126127243042
    },
    {
      "epoch": 2.8717041015625e-05,
      "model_forward_time": 3.1443538665771484,
      "step": 7528
    },
    {
      "epoch": 2.8717041015625e-05,
      "step": 7528,
      "training_step_time": 9.914443016052246
    },
    {
      "epoch": 2.8717041015625e-05,
      "model_forward_time": 3.1478559970855713,
      "step": 7528
    },
    {
      "epoch": 2.8717041015625e-05,
      "step": 7528,
      "training_step_time": 10.279155969619751
    },
    {
      "epoch": 2.8720855712890625e-05,
      "model_forward_time": 3.1920888423919678,
      "step": 7529
    },
    {
      "epoch": 2.8720855712890625e-05,
      "step": 7529,
      "training_step_time": 9.962907791137695
    },
    {
      "epoch": 2.8720855712890625e-05,
      "model_forward_time": 3.4260482788085938,
      "step": 7529
    },
    {
      "epoch": 2.8720855712890625e-05,
      "step": 7529,
      "training_step_time": 10.190511226654053
    },
    {
      "epoch": 2.8720855712890625e-05,
      "model_forward_time": 3.137874126434326,
      "step": 7529
    },
    {
      "epoch": 2.8720855712890625e-05,
      "step": 7529,
      "training_step_time": 9.900150537490845
    },
    {
      "epoch": 2.8720855712890625e-05,
      "model_forward_time": 3.1415157318115234,
      "step": 7529
    },
    {
      "epoch": 2.8720855712890625e-05,
      "step": 7529,
      "training_step_time": 10.004485368728638
    },
    {
      "epoch": 2.872467041015625e-05,
      "grad_norm": 0.5215729475021362,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 7530
    },
    {
      "epoch": 2.872467041015625e-05,
      "model_forward_time": 3.189622402191162,
      "step": 7530
    },
    {
      "dynamics_loss_avg": 0.06411313302814961,
      "epoch": 2.872467041015625e-05,
      "step": 7530
    },
    {
      "action_loss_avg": 0.025986730027943848,
      "epoch": 2.872467041015625e-05,
      "step": 7530
    },
    {
      "epoch": 2.872467041015625e-05,
      "step": 7530,
      "training_step_time": 9.959513425827026
    },
    {
      "epoch": 2.872467041015625e-05,
      "model_forward_time": 3.1299943923950195,
      "step": 7530
    },
    {
      "epoch": 2.872467041015625e-05,
      "step": 7530,
      "training_step_time": 9.894862651824951
    },
    {
      "epoch": 2.872467041015625e-05,
      "model_forward_time": 3.3999178409576416,
      "step": 7530
    },
    {
      "epoch": 2.872467041015625e-05,
      "step": 7530,
      "training_step_time": 10.171006441116333
    },
    {
      "epoch": 2.872467041015625e-05,
      "model_forward_time": 3.1366357803344727,
      "step": 7530
    },
    {
      "epoch": 2.872467041015625e-05,
      "step": 7530,
      "training_step_time": 10.31923508644104
    },
    {
      "epoch": 2.8728485107421874e-05,
      "model_forward_time": 3.170696258544922,
      "step": 7531
    },
    {
      "epoch": 2.8728485107421874e-05,
      "step": 7531,
      "training_step_time": 9.939208269119263
    },
    {
      "epoch": 2.8728485107421874e-05,
      "model_forward_time": 3.134092330932617,
      "step": 7531
    },
    {
      "epoch": 2.8728485107421874e-05,
      "step": 7531,
      "training_step_time": 9.898530960083008
    },
    {
      "epoch": 2.8728485107421874e-05,
      "model_forward_time": 3.1348249912261963,
      "step": 7531
    },
    {
      "epoch": 2.8728485107421874e-05,
      "step": 7531,
      "training_step_time": 9.89982795715332
    },
    {
      "epoch": 2.8728485107421874e-05,
      "model_forward_time": 3.4143548011779785,
      "step": 7531
    },
    {
      "epoch": 2.8728485107421874e-05,
      "step": 7531,
      "training_step_time": 10.29430103302002
    },
    {
      "epoch": 2.87322998046875e-05,
      "model_forward_time": 3.179647207260132,
      "step": 7532
    },
    {
      "epoch": 2.87322998046875e-05,
      "step": 7532,
      "training_step_time": 9.9462730884552
    },
    {
      "epoch": 2.87322998046875e-05,
      "model_forward_time": 3.131227970123291,
      "step": 7532
    },
    {
      "epoch": 2.87322998046875e-05,
      "step": 7532,
      "training_step_time": 9.89503264427185
    },
    {
      "epoch": 2.87322998046875e-05,
      "model_forward_time": 3.135965347290039,
      "step": 7532
    },
    {
      "dynamics_loss_avg": 0.04796995567157865,
      "epoch": 2.87322998046875e-05,
      "step": 7532
    },
    {
      "action_loss_avg": 0.016417240933515133,
      "epoch": 2.87322998046875e-05,
      "step": 7532
    },
    {
      "epoch": 2.87322998046875e-05,
      "step": 7532,
      "training_step_time": 9.904822587966919
    },
    {
      "epoch": 2.87322998046875e-05,
      "model_forward_time": 2.4320781230926514,
      "step": 7532
    },
    {
      "epoch": 2.87322998046875e-05,
      "step": 7532,
      "training_step_time": 10.299987077713013
    },
    {
      "epoch": 2.8736114501953126e-05,
      "model_forward_time": 3.4437592029571533,
      "step": 7533
    },
    {
      "epoch": 2.8736114501953126e-05,
      "step": 7533,
      "training_step_time": 10.208427906036377
    },
    {
      "epoch": 2.8736114501953126e-05,
      "model_forward_time": 3.132878541946411,
      "step": 7533
    },
    {
      "epoch": 2.8736114501953126e-05,
      "step": 7533,
      "training_step_time": 9.90616512298584
    },
    {
      "epoch": 2.8736114501953126e-05,
      "model_forward_time": 3.1353678703308105,
      "step": 7533
    },
    {
      "epoch": 2.8736114501953126e-05,
      "step": 7533,
      "training_step_time": 9.892196416854858
    },
    {
      "epoch": 2.8736114501953126e-05,
      "model_forward_time": 3.1324830055236816,
      "step": 7533
    },
    {
      "epoch": 2.8736114501953126e-05,
      "step": 7533,
      "training_step_time": 10.268224000930786
    },
    {
      "epoch": 2.873992919921875e-05,
      "model_forward_time": 3.1704254150390625,
      "step": 7534
    },
    {
      "epoch": 2.873992919921875e-05,
      "step": 7534,
      "training_step_time": 9.938058137893677
    },
    {
      "epoch": 2.873992919921875e-05,
      "model_forward_time": 3.1289851665496826,
      "step": 7534
    },
    {
      "epoch": 2.873992919921875e-05,
      "step": 7534,
      "training_step_time": 10.179187536239624
    },
    {
      "epoch": 2.873992919921875e-05,
      "model_forward_time": 3.1303253173828125,
      "step": 7534
    },
    {
      "epoch": 2.873992919921875e-05,
      "step": 7534,
      "training_step_time": 9.895089864730835
    },
    {
      "epoch": 2.873992919921875e-05,
      "model_forward_time": 3.131540060043335,
      "step": 7534
    },
    {
      "epoch": 2.873992919921875e-05,
      "step": 7534,
      "training_step_time": 10.303324222564697
    },
    {
      "epoch": 2.8743743896484376e-05,
      "model_forward_time": 3.1732192039489746,
      "step": 7535
    },
    {
      "dynamics_loss_avg": 0.05767254922538996,
      "epoch": 2.8743743896484376e-05,
      "step": 7535
    },
    {
      "action_loss_avg": 0.02808554908260703,
      "epoch": 2.8743743896484376e-05,
      "step": 7535
    },
    {
      "epoch": 2.8743743896484376e-05,
      "step": 7535,
      "training_step_time": 9.944173097610474
    },
    {
      "epoch": 2.8743743896484376e-05,
      "model_forward_time": 3.1461868286132812,
      "step": 7535
    },
    {
      "epoch": 2.8743743896484376e-05,
      "step": 7535,
      "training_step_time": 9.917140483856201
    },
    {
      "epoch": 2.8743743896484376e-05,
      "model_forward_time": 3.448988199234009,
      "step": 7535
    },
    {
      "epoch": 2.8743743896484376e-05,
      "step": 7535,
      "training_step_time": 10.214301824569702
    },
    {
      "epoch": 2.8743743896484376e-05,
      "model_forward_time": 3.1415319442749023,
      "step": 7535
    },
    {
      "epoch": 2.8743743896484376e-05,
      "step": 7535,
      "training_step_time": 10.287023067474365
    },
    {
      "epoch": 2.874755859375e-05,
      "model_forward_time": 3.1767804622650146,
      "step": 7536
    },
    {
      "epoch": 2.874755859375e-05,
      "step": 7536,
      "training_step_time": 9.950724601745605
    },
    {
      "epoch": 2.874755859375e-05,
      "model_forward_time": 3.1435704231262207,
      "step": 7536
    },
    {
      "epoch": 2.874755859375e-05,
      "step": 7536,
      "training_step_time": 9.913377523422241
    },
    {
      "epoch": 2.874755859375e-05,
      "model_forward_time": 3.141441583633423,
      "step": 7536
    },
    {
      "epoch": 2.874755859375e-05,
      "step": 7536,
      "training_step_time": 9.908298969268799
    },
    {
      "epoch": 2.874755859375e-05,
      "model_forward_time": 3.1518635749816895,
      "step": 7536
    },
    {
      "epoch": 2.874755859375e-05,
      "step": 7536,
      "training_step_time": 10.297165393829346
    },
    {
      "epoch": 2.8751373291015625e-05,
      "model_forward_time": 3.171522378921509,
      "step": 7537
    },
    {
      "epoch": 2.8751373291015625e-05,
      "step": 7537,
      "training_step_time": 9.94633150100708
    },
    {
      "epoch": 2.8751373291015625e-05,
      "model_forward_time": 3.144404888153076,
      "step": 7537
    },
    {
      "epoch": 2.8751373291015625e-05,
      "step": 7537,
      "training_step_time": 9.917908906936646
    },
    {
      "epoch": 2.8751373291015625e-05,
      "model_forward_time": 3.14363956451416,
      "step": 7537
    },
    {
      "dynamics_loss_avg": 0.06589643592014909,
      "epoch": 2.8751373291015625e-05,
      "step": 7537
    },
    {
      "action_loss_avg": 0.02912530521862209,
      "epoch": 2.8751373291015625e-05,
      "step": 7537
    },
    {
      "epoch": 2.8751373291015625e-05,
      "step": 7537,
      "training_step_time": 9.917172193527222
    },
    {
      "epoch": 2.8751373291015625e-05,
      "model_forward_time": 3.143160104751587,
      "step": 7537
    },
    {
      "epoch": 2.8751373291015625e-05,
      "step": 7537,
      "training_step_time": 10.27514100074768
    },
    {
      "epoch": 2.875518798828125e-05,
      "model_forward_time": 3.4752159118652344,
      "step": 7538
    },
    {
      "epoch": 2.875518798828125e-05,
      "step": 7538,
      "training_step_time": 10.243360757827759
    },
    {
      "epoch": 2.875518798828125e-05,
      "model_forward_time": 3.1409008502960205,
      "step": 7538
    },
    {
      "epoch": 2.875518798828125e-05,
      "step": 7538,
      "training_step_time": 9.911995649337769
    },
    {
      "epoch": 2.875518798828125e-05,
      "model_forward_time": 3.1408588886260986,
      "step": 7538
    },
    {
      "epoch": 2.875518798828125e-05,
      "step": 7538,
      "training_step_time": 9.913407564163208
    },
    {
      "epoch": 2.875518798828125e-05,
      "model_forward_time": 3.150285243988037,
      "step": 7538
    },
    {
      "epoch": 2.875518798828125e-05,
      "step": 7538,
      "training_step_time": 10.252881526947021
    },
    {
      "epoch": 2.8759002685546874e-05,
      "model_forward_time": 3.1774888038635254,
      "step": 7539
    },
    {
      "epoch": 2.8759002685546874e-05,
      "step": 7539,
      "training_step_time": 9.954096794128418
    },
    {
      "epoch": 2.8759002685546874e-05,
      "model_forward_time": 3.433199405670166,
      "step": 7539
    },
    {
      "epoch": 2.8759002685546874e-05,
      "step": 7539,
      "training_step_time": 10.203475952148438
    },
    {
      "epoch": 2.8759002685546874e-05,
      "model_forward_time": 3.1491150856018066,
      "step": 7539
    },
    {
      "epoch": 2.8759002685546874e-05,
      "step": 7539,
      "training_step_time": 9.92290472984314
    },
    {
      "epoch": 2.8759002685546874e-05,
      "model_forward_time": 3.1513214111328125,
      "step": 7539
    },
    {
      "epoch": 2.8759002685546874e-05,
      "step": 7539,
      "training_step_time": 10.28238844871521
    },
    {
      "epoch": 2.87628173828125e-05,
      "grad_norm": 0.6577472686767578,
      "learning_rate": 1e-05,
      "loss": 0.0823,
      "step": 7540
    },
    {
      "epoch": 2.87628173828125e-05,
      "model_forward_time": 3.187169313430786,
      "step": 7540
    },
    {
      "dynamics_loss_avg": 0.05645263828337192,
      "epoch": 2.87628173828125e-05,
      "step": 7540
    },
    {
      "action_loss_avg": 0.026346152555197476,
      "epoch": 2.87628173828125e-05,
      "step": 7540
    },
    {
      "epoch": 2.87628173828125e-05,
      "step": 7540,
      "training_step_time": 9.966378450393677
    },
    {
      "epoch": 2.87628173828125e-05,
      "model_forward_time": 2.4442927837371826,
      "step": 7540
    },
    {
      "epoch": 2.87628173828125e-05,
      "step": 7540,
      "training_step_time": 7.478578567504883
    },
    {
      "epoch": 2.87628173828125e-05,
      "model_forward_time": 3.4086496829986572,
      "step": 7540
    },
    {
      "epoch": 2.87628173828125e-05,
      "step": 7540,
      "training_step_time": 10.180095195770264
    },
    {
      "epoch": 2.87628173828125e-05,
      "model_forward_time": 3.1467103958129883,
      "step": 7540
    },
    {
      "epoch": 2.87628173828125e-05,
      "step": 7540,
      "training_step_time": 10.328603506088257
    },
    {
      "epoch": 2.8766632080078126e-05,
      "model_forward_time": 3.1976382732391357,
      "step": 7541
    },
    {
      "epoch": 2.8766632080078126e-05,
      "step": 7541,
      "training_step_time": 9.974535465240479
    },
    {
      "epoch": 2.8766632080078126e-05,
      "model_forward_time": 3.143427610397339,
      "step": 7541
    },
    {
      "epoch": 2.8766632080078126e-05,
      "step": 7541,
      "training_step_time": 9.91405177116394
    },
    {
      "epoch": 2.8766632080078126e-05,
      "model_forward_time": 3.13771915435791,
      "step": 7541
    },
    {
      "epoch": 2.8766632080078126e-05,
      "step": 7541,
      "training_step_time": 9.9075345993042
    },
    {
      "epoch": 2.8766632080078126e-05,
      "model_forward_time": 3.1377811431884766,
      "step": 7541
    },
    {
      "epoch": 2.8766632080078126e-05,
      "step": 7541,
      "training_step_time": 10.261276245117188
    },
    {
      "epoch": 2.877044677734375e-05,
      "model_forward_time": 2.4871087074279785,
      "step": 7542
    },
    {
      "epoch": 2.877044677734375e-05,
      "step": 7542,
      "training_step_time": 7.510363578796387
    },
    {
      "epoch": 2.877044677734375e-05,
      "model_forward_time": 3.1349399089813232,
      "step": 7542
    },
    {
      "epoch": 2.877044677734375e-05,
      "step": 7542,
      "training_step_time": 9.904146194458008
    },
    {
      "epoch": 2.877044677734375e-05,
      "model_forward_time": 3.1430089473724365,
      "step": 7542
    },
    {
      "dynamics_loss_avg": 0.05127360858023167,
      "epoch": 2.877044677734375e-05,
      "step": 7542
    },
    {
      "action_loss_avg": 0.024465734511613844,
      "epoch": 2.877044677734375e-05,
      "step": 7542
    },
    {
      "epoch": 2.877044677734375e-05,
      "step": 7542,
      "training_step_time": 9.91442322731018
    },
    {
      "epoch": 2.877044677734375e-05,
      "model_forward_time": 3.136402130126953,
      "step": 7542
    },
    {
      "epoch": 2.877044677734375e-05,
      "step": 7542,
      "training_step_time": 10.296122550964355
    },
    {
      "epoch": 2.8774261474609376e-05,
      "model_forward_time": 3.1707208156585693,
      "step": 7543
    },
    {
      "epoch": 2.8774261474609376e-05,
      "step": 7543,
      "training_step_time": 10.230648040771484
    },
    {
      "epoch": 2.8774261474609376e-05,
      "model_forward_time": 3.137686252593994,
      "step": 7543
    },
    {
      "epoch": 2.8774261474609376e-05,
      "step": 7543,
      "training_step_time": 9.902666807174683
    },
    {
      "epoch": 2.8774261474609376e-05,
      "model_forward_time": 3.135063886642456,
      "step": 7543
    },
    {
      "epoch": 2.8774261474609376e-05,
      "step": 7543,
      "training_step_time": 9.9005126953125
    },
    {
      "epoch": 2.8774261474609376e-05,
      "model_forward_time": 3.133563280105591,
      "step": 7543
    },
    {
      "epoch": 2.8774261474609376e-05,
      "step": 7543,
      "training_step_time": 10.24906039237976
    },
    {
      "epoch": 2.8778076171875e-05,
      "model_forward_time": 3.1836745738983154,
      "step": 7544
    },
    {
      "epoch": 2.8778076171875e-05,
      "step": 7544,
      "training_step_time": 9.956864595413208
    },
    {
      "epoch": 2.8778076171875e-05,
      "model_forward_time": 3.390821695327759,
      "step": 7544
    },
    {
      "epoch": 2.8778076171875e-05,
      "step": 7544,
      "training_step_time": 10.155142068862915
    },
    {
      "epoch": 2.8778076171875e-05,
      "model_forward_time": 2.4307868480682373,
      "step": 7544
    },
    {
      "epoch": 2.8778076171875e-05,
      "step": 7544,
      "training_step_time": 7.453627347946167
    },
    {
      "epoch": 2.8778076171875e-05,
      "model_forward_time": 3.127904176712036,
      "step": 7544
    },
    {
      "epoch": 2.8778076171875e-05,
      "step": 7544,
      "training_step_time": 10.26599645614624
    },
    {
      "epoch": 2.8781890869140625e-05,
      "model_forward_time": 3.1657626628875732,
      "step": 7545
    },
    {
      "dynamics_loss_avg": 0.05201549381017685,
      "epoch": 2.8781890869140625e-05,
      "step": 7545
    },
    {
      "action_loss_avg": 0.019344986230134965,
      "epoch": 2.8781890869140625e-05,
      "step": 7545
    },
    {
      "epoch": 2.8781890869140625e-05,
      "step": 7545,
      "training_step_time": 9.934879064559937
    },
    {
      "epoch": 2.8781890869140625e-05,
      "model_forward_time": 3.136578321456909,
      "step": 7545
    },
    {
      "epoch": 2.8781890869140625e-05,
      "step": 7545,
      "training_step_time": 9.905370235443115
    },
    {
      "epoch": 2.8781890869140625e-05,
      "model_forward_time": 3.13401198387146,
      "step": 7545
    },
    {
      "epoch": 2.8781890869140625e-05,
      "step": 7545,
      "training_step_time": 9.903664588928223
    },
    {
      "epoch": 2.8781890869140625e-05,
      "model_forward_time": 3.3843226432800293,
      "step": 7545
    },
    {
      "epoch": 2.8781890869140625e-05,
      "step": 7545,
      "training_step_time": 10.302417039871216
    },
    {
      "epoch": 2.878570556640625e-05,
      "model_forward_time": 3.1881253719329834,
      "step": 7546
    },
    {
      "epoch": 2.878570556640625e-05,
      "step": 7546,
      "training_step_time": 9.966357946395874
    },
    {
      "epoch": 2.878570556640625e-05,
      "model_forward_time": 3.1381146907806396,
      "step": 7546
    },
    {
      "epoch": 2.878570556640625e-05,
      "step": 7546,
      "training_step_time": 9.910408973693848
    },
    {
      "epoch": 2.878570556640625e-05,
      "model_forward_time": 3.138150453567505,
      "step": 7546
    },
    {
      "epoch": 2.878570556640625e-05,
      "step": 7546,
      "training_step_time": 9.911749124526978
    },
    {
      "epoch": 2.878570556640625e-05,
      "model_forward_time": 3.3937113285064697,
      "step": 7546
    },
    {
      "epoch": 2.878570556640625e-05,
      "step": 7546,
      "training_step_time": 10.296608686447144
    },
    {
      "epoch": 2.8789520263671874e-05,
      "model_forward_time": 3.1734933853149414,
      "step": 7547
    },
    {
      "epoch": 2.8789520263671874e-05,
      "step": 7547,
      "training_step_time": 9.948297500610352
    },
    {
      "epoch": 2.8789520263671874e-05,
      "model_forward_time": 3.1394803524017334,
      "step": 7547
    },
    {
      "epoch": 2.8789520263671874e-05,
      "step": 7547,
      "training_step_time": 9.914344072341919
    },
    {
      "epoch": 2.8789520263671874e-05,
      "model_forward_time": 3.1363539695739746,
      "step": 7547
    },
    {
      "dynamics_loss_avg": 0.05875974763184786,
      "epoch": 2.8789520263671874e-05,
      "step": 7547
    },
    {
      "action_loss_avg": 0.021636296482756733,
      "epoch": 2.8789520263671874e-05,
      "step": 7547
    },
    {
      "epoch": 2.8789520263671874e-05,
      "step": 7547,
      "training_step_time": 9.912218809127808
    },
    {
      "epoch": 2.8789520263671874e-05,
      "model_forward_time": 3.138888120651245,
      "step": 7547
    },
    {
      "epoch": 2.8789520263671874e-05,
      "step": 7547,
      "training_step_time": 10.319770812988281
    },
    {
      "epoch": 2.87933349609375e-05,
      "model_forward_time": 3.1720478534698486,
      "step": 7548
    },
    {
      "epoch": 2.87933349609375e-05,
      "step": 7548,
      "training_step_time": 10.216015577316284
    },
    {
      "epoch": 2.87933349609375e-05,
      "model_forward_time": 3.1394174098968506,
      "step": 7548
    },
    {
      "epoch": 2.87933349609375e-05,
      "step": 7548,
      "training_step_time": 9.915709495544434
    },
    {
      "epoch": 2.87933349609375e-05,
      "model_forward_time": 3.1373612880706787,
      "step": 7548
    },
    {
      "epoch": 2.87933349609375e-05,
      "step": 7548,
      "training_step_time": 9.919248342514038
    },
    {
      "epoch": 2.87933349609375e-05,
      "model_forward_time": 3.1421327590942383,
      "step": 7548
    },
    {
      "epoch": 2.87933349609375e-05,
      "step": 7548,
      "training_step_time": 10.302520275115967
    },
    {
      "epoch": 2.8797149658203126e-05,
      "model_forward_time": 3.1848864555358887,
      "step": 7549
    },
    {
      "epoch": 2.8797149658203126e-05,
      "step": 7549,
      "training_step_time": 9.968462944030762
    },
    {
      "epoch": 2.8797149658203126e-05,
      "model_forward_time": 3.4365992546081543,
      "step": 7549
    },
    {
      "epoch": 2.8797149658203126e-05,
      "step": 7549,
      "training_step_time": 10.207902193069458
    },
    {
      "epoch": 2.8797149658203126e-05,
      "model_forward_time": 3.143211603164673,
      "step": 7549
    },
    {
      "epoch": 2.8797149658203126e-05,
      "step": 7549,
      "training_step_time": 9.915634393692017
    },
    {
      "epoch": 2.8797149658203126e-05,
      "model_forward_time": 3.149402618408203,
      "step": 7549
    },
    {
      "epoch": 2.8797149658203126e-05,
      "step": 7549,
      "training_step_time": 10.239924669265747
    },
    {
      "epoch": 2.880096435546875e-05,
      "grad_norm": 0.5687077641487122,
      "learning_rate": 1e-05,
      "loss": 0.0831,
      "step": 7550
    },
    {
      "epoch": 2.880096435546875e-05,
      "model_forward_time": 3.176528215408325,
      "step": 7550
    },
    {
      "dynamics_loss_avg": 0.04925199635326862,
      "epoch": 2.880096435546875e-05,
      "step": 7550
    },
    {
      "action_loss_avg": 0.01980765163898468,
      "epoch": 2.880096435546875e-05,
      "step": 7550
    },
    {
      "epoch": 2.880096435546875e-05,
      "step": 7550,
      "training_step_time": 9.953853845596313
    },
    {
      "epoch": 2.880096435546875e-05,
      "model_forward_time": 3.1396498680114746,
      "step": 7550
    },
    {
      "epoch": 2.880096435546875e-05,
      "step": 7550,
      "training_step_time": 9.912078857421875
    },
    {
      "epoch": 2.880096435546875e-05,
      "model_forward_time": 3.427722215652466,
      "step": 7550
    },
    {
      "epoch": 2.880096435546875e-05,
      "step": 7550,
      "training_step_time": 10.192783117294312
    },
    {
      "epoch": 2.880096435546875e-05,
      "model_forward_time": 3.1420938968658447,
      "step": 7550
    },
    {
      "epoch": 2.880096435546875e-05,
      "step": 7550,
      "training_step_time": 10.264559507369995
    },
    {
      "epoch": 2.8804779052734376e-05,
      "model_forward_time": 3.178903102874756,
      "step": 7551
    },
    {
      "epoch": 2.8804779052734376e-05,
      "step": 7551,
      "training_step_time": 9.946611642837524
    },
    {
      "epoch": 2.8804779052734376e-05,
      "model_forward_time": 3.1381447315216064,
      "step": 7551
    },
    {
      "epoch": 2.8804779052734376e-05,
      "step": 7551,
      "training_step_time": 9.904751777648926
    },
    {
      "epoch": 2.8804779052734376e-05,
      "model_forward_time": 3.1404614448547363,
      "step": 7551
    },
    {
      "epoch": 2.8804779052734376e-05,
      "step": 7551,
      "training_step_time": 9.911102771759033
    },
    {
      "epoch": 2.8804779052734376e-05,
      "model_forward_time": 3.431149959564209,
      "step": 7551
    },
    {
      "epoch": 2.8804779052734376e-05,
      "step": 7551,
      "training_step_time": 10.292041778564453
    },
    {
      "epoch": 2.880859375e-05,
      "model_forward_time": 3.1736133098602295,
      "step": 7552
    },
    {
      "epoch": 2.880859375e-05,
      "step": 7552,
      "training_step_time": 9.943711280822754
    },
    {
      "epoch": 2.880859375e-05,
      "model_forward_time": 3.1397206783294678,
      "step": 7552
    },
    {
      "epoch": 2.880859375e-05,
      "step": 7552,
      "training_step_time": 9.905713081359863
    },
    {
      "epoch": 2.880859375e-05,
      "model_forward_time": 3.141366720199585,
      "step": 7552
    },
    {
      "dynamics_loss_avg": 0.04794756732881069,
      "epoch": 2.880859375e-05,
      "step": 7552
    },
    {
      "action_loss_avg": 0.016957518737763168,
      "epoch": 2.880859375e-05,
      "step": 7552
    },
    {
      "epoch": 2.880859375e-05,
      "step": 7552,
      "training_step_time": 9.911196231842041
    },
    {
      "epoch": 2.880859375e-05,
      "model_forward_time": 3.1373331546783447,
      "step": 7552
    },
    {
      "epoch": 2.880859375e-05,
      "step": 7552,
      "training_step_time": 10.27980089187622
    },
    {
      "epoch": 2.8812408447265625e-05,
      "model_forward_time": 3.464094400405884,
      "step": 7553
    },
    {
      "epoch": 2.8812408447265625e-05,
      "step": 7553,
      "training_step_time": 10.22903847694397
    },
    {
      "epoch": 2.8812408447265625e-05,
      "model_forward_time": 3.1360740661621094,
      "step": 7553
    },
    {
      "epoch": 2.8812408447265625e-05,
      "step": 7553,
      "training_step_time": 9.900901794433594
    },
    {
      "epoch": 2.8812408447265625e-05,
      "model_forward_time": 3.1388866901397705,
      "step": 7553
    },
    {
      "epoch": 2.8812408447265625e-05,
      "step": 7553,
      "training_step_time": 9.90463662147522
    },
    {
      "epoch": 2.8812408447265625e-05,
      "model_forward_time": 3.1360788345336914,
      "step": 7553
    },
    {
      "epoch": 2.8812408447265625e-05,
      "step": 7553,
      "training_step_time": 10.283021688461304
    },
    {
      "epoch": 2.881622314453125e-05,
      "model_forward_time": 3.183267116546631,
      "step": 7554
    },
    {
      "epoch": 2.881622314453125e-05,
      "step": 7554,
      "training_step_time": 9.951452016830444
    },
    {
      "epoch": 2.881622314453125e-05,
      "model_forward_time": 3.402496337890625,
      "step": 7554
    },
    {
      "epoch": 2.881622314453125e-05,
      "step": 7554,
      "training_step_time": 10.168128728866577
    },
    {
      "epoch": 2.881622314453125e-05,
      "model_forward_time": 3.140009880065918,
      "step": 7554
    },
    {
      "epoch": 2.881622314453125e-05,
      "step": 7554,
      "training_step_time": 9.907756567001343
    },
    {
      "epoch": 2.881622314453125e-05,
      "model_forward_time": 3.1393325328826904,
      "step": 7554
    },
    {
      "epoch": 2.881622314453125e-05,
      "step": 7554,
      "training_step_time": 10.251685380935669
    },
    {
      "epoch": 2.8820037841796874e-05,
      "model_forward_time": 3.1844019889831543,
      "step": 7555
    },
    {
      "dynamics_loss_avg": 0.047822830826044084,
      "epoch": 2.8820037841796874e-05,
      "step": 7555
    },
    {
      "action_loss_avg": 0.01958230291493237,
      "epoch": 2.8820037841796874e-05,
      "step": 7555
    },
    {
      "epoch": 2.8820037841796874e-05,
      "step": 7555,
      "training_step_time": 9.95799446105957
    },
    {
      "epoch": 2.8820037841796874e-05,
      "model_forward_time": 3.1396429538726807,
      "step": 7555
    },
    {
      "epoch": 2.8820037841796874e-05,
      "step": 7555,
      "training_step_time": 9.907328844070435
    },
    {
      "epoch": 2.8820037841796874e-05,
      "model_forward_time": 3.4215216636657715,
      "step": 7555
    },
    {
      "epoch": 2.8820037841796874e-05,
      "step": 7555,
      "training_step_time": 10.18900179862976
    },
    {
      "epoch": 2.8820037841796874e-05,
      "model_forward_time": 3.141587734222412,
      "step": 7555
    },
    {
      "epoch": 2.8820037841796874e-05,
      "step": 7555,
      "training_step_time": 10.258525609970093
    },
    {
      "epoch": 2.88238525390625e-05,
      "model_forward_time": 3.174258232116699,
      "step": 7556
    },
    {
      "epoch": 2.88238525390625e-05,
      "step": 7556,
      "training_step_time": 9.939959049224854
    },
    {
      "epoch": 2.88238525390625e-05,
      "model_forward_time": 3.1382896900177,
      "step": 7556
    },
    {
      "epoch": 2.88238525390625e-05,
      "step": 7556,
      "training_step_time": 9.903792381286621
    },
    {
      "epoch": 2.88238525390625e-05,
      "model_forward_time": 3.1363720893859863,
      "step": 7556
    },
    {
      "epoch": 2.88238525390625e-05,
      "step": 7556,
      "training_step_time": 9.898120403289795
    },
    {
      "epoch": 2.88238525390625e-05,
      "model_forward_time": 3.419938564300537,
      "step": 7556
    },
    {
      "epoch": 2.88238525390625e-05,
      "step": 7556,
      "training_step_time": 10.293593883514404
    },
    {
      "epoch": 2.8827667236328126e-05,
      "model_forward_time": 3.180293321609497,
      "step": 7557
    },
    {
      "epoch": 2.8827667236328126e-05,
      "step": 7557,
      "training_step_time": 9.955790281295776
    },
    {
      "epoch": 2.8827667236328126e-05,
      "model_forward_time": 3.1420083045959473,
      "step": 7557
    },
    {
      "epoch": 2.8827667236328126e-05,
      "step": 7557,
      "training_step_time": 9.912865400314331
    },
    {
      "epoch": 2.8827667236328126e-05,
      "model_forward_time": 2.439605236053467,
      "step": 7557
    },
    {
      "dynamics_loss_avg": 0.05576534438878298,
      "epoch": 2.8827667236328126e-05,
      "step": 7557
    },
    {
      "action_loss_avg": 0.023790272418409586,
      "epoch": 2.8827667236328126e-05,
      "step": 7557
    },
    {
      "epoch": 2.8827667236328126e-05,
      "step": 7557,
      "training_step_time": 7.474161148071289
    },
    {
      "epoch": 2.8827667236328126e-05,
      "model_forward_time": 3.1400270462036133,
      "step": 7557
    },
    {
      "epoch": 2.8827667236328126e-05,
      "step": 7557,
      "training_step_time": 10.296592950820923
    },
    {
      "epoch": 2.883148193359375e-05,
      "model_forward_time": 3.4433326721191406,
      "step": 7558
    },
    {
      "epoch": 2.883148193359375e-05,
      "step": 7558,
      "training_step_time": 10.217705249786377
    },
    {
      "epoch": 2.883148193359375e-05,
      "model_forward_time": 3.1453046798706055,
      "step": 7558
    },
    {
      "epoch": 2.883148193359375e-05,
      "step": 7558,
      "training_step_time": 9.918787717819214
    },
    {
      "epoch": 2.883148193359375e-05,
      "model_forward_time": 3.1388092041015625,
      "step": 7558
    },
    {
      "epoch": 2.883148193359375e-05,
      "step": 7558,
      "training_step_time": 9.9158935546875
    },
    {
      "epoch": 2.883148193359375e-05,
      "model_forward_time": 3.137375593185425,
      "step": 7558
    },
    {
      "epoch": 2.883148193359375e-05,
      "step": 7558,
      "training_step_time": 10.259947299957275
    },
    {
      "epoch": 2.8835296630859376e-05,
      "model_forward_time": 3.1889402866363525,
      "step": 7559
    },
    {
      "epoch": 2.8835296630859376e-05,
      "step": 7559,
      "training_step_time": 9.961649656295776
    },
    {
      "epoch": 2.8835296630859376e-05,
      "model_forward_time": 3.1326904296875,
      "step": 7559
    },
    {
      "epoch": 2.8835296630859376e-05,
      "step": 7559,
      "training_step_time": 10.17578935623169
    },
    {
      "epoch": 2.8835296630859376e-05,
      "model_forward_time": 3.1396167278289795,
      "step": 7559
    },
    {
      "epoch": 2.8835296630859376e-05,
      "step": 7559,
      "training_step_time": 9.913393020629883
    },
    {
      "epoch": 2.8835296630859376e-05,
      "model_forward_time": 3.1415297985076904,
      "step": 7559
    },
    {
      "epoch": 2.8835296630859376e-05,
      "step": 7559,
      "training_step_time": 10.301452875137329
    },
    {
      "epoch": 2.8839111328125e-05,
      "grad_norm": 0.5503395795822144,
      "learning_rate": 1e-05,
      "loss": 0.0797,
      "step": 7560
    },
    {
      "epoch": 2.8839111328125e-05,
      "model_forward_time": 3.1839182376861572,
      "step": 7560
    },
    {
      "dynamics_loss_avg": 0.07453372552990914,
      "epoch": 2.8839111328125e-05,
      "step": 7560
    },
    {
      "action_loss_avg": 0.029864223673939705,
      "epoch": 2.8839111328125e-05,
      "step": 7560
    },
    {
      "epoch": 2.8839111328125e-05,
      "step": 7560,
      "training_step_time": 9.965041399002075
    },
    {
      "epoch": 2.8839111328125e-05,
      "model_forward_time": 3.1403210163116455,
      "step": 7560
    },
    {
      "epoch": 2.8839111328125e-05,
      "step": 7560,
      "training_step_time": 9.918602228164673
    },
    {
      "epoch": 2.8839111328125e-05,
      "model_forward_time": 3.401731014251709,
      "step": 7560
    },
    {
      "epoch": 2.8839111328125e-05,
      "step": 7560,
      "training_step_time": 10.183518886566162
    },
    {
      "epoch": 2.8839111328125e-05,
      "model_forward_time": 3.146671772003174,
      "step": 7560
    },
    {
      "epoch": 2.8839111328125e-05,
      "step": 7560,
      "training_step_time": 10.30199909210205
    },
    {
      "epoch": 2.8842926025390625e-05,
      "model_forward_time": 3.179975986480713,
      "step": 7561
    },
    {
      "epoch": 2.8842926025390625e-05,
      "step": 7561,
      "training_step_time": 9.965389490127563
    },
    {
      "epoch": 2.8842926025390625e-05,
      "model_forward_time": 3.1396431922912598,
      "step": 7561
    },
    {
      "epoch": 2.8842926025390625e-05,
      "step": 7561,
      "training_step_time": 9.921442031860352
    },
    {
      "epoch": 2.8842926025390625e-05,
      "model_forward_time": 3.1432626247406006,
      "step": 7561
    },
    {
      "epoch": 2.8842926025390625e-05,
      "step": 7561,
      "training_step_time": 9.921310901641846
    },
    {
      "epoch": 2.8842926025390625e-05,
      "model_forward_time": 3.428295373916626,
      "step": 7561
    },
    {
      "epoch": 2.8842926025390625e-05,
      "step": 7561,
      "training_step_time": 10.300907850265503
    },
    {
      "epoch": 2.884674072265625e-05,
      "model_forward_time": 3.1835145950317383,
      "step": 7562
    },
    {
      "epoch": 2.884674072265625e-05,
      "step": 7562,
      "training_step_time": 9.958430528640747
    },
    {
      "epoch": 2.884674072265625e-05,
      "model_forward_time": 3.1411292552948,
      "step": 7562
    },
    {
      "epoch": 2.884674072265625e-05,
      "step": 7562,
      "training_step_time": 9.913644075393677
    },
    {
      "epoch": 2.884674072265625e-05,
      "model_forward_time": 3.1380198001861572,
      "step": 7562
    },
    {
      "dynamics_loss_avg": 0.05680554863065481,
      "epoch": 2.884674072265625e-05,
      "step": 7562
    },
    {
      "action_loss_avg": 0.02555198296904564,
      "epoch": 2.884674072265625e-05,
      "step": 7562
    },
    {
      "epoch": 2.884674072265625e-05,
      "step": 7562,
      "training_step_time": 9.925467014312744
    },
    {
      "epoch": 2.884674072265625e-05,
      "model_forward_time": 3.1410791873931885,
      "step": 7562
    },
    {
      "epoch": 2.884674072265625e-05,
      "step": 7562,
      "training_step_time": 10.258532285690308
    },
    {
      "epoch": 2.8850555419921874e-05,
      "model_forward_time": 3.1841073036193848,
      "step": 7563
    },
    {
      "epoch": 2.8850555419921874e-05,
      "step": 7563,
      "training_step_time": 10.264267444610596
    },
    {
      "epoch": 2.8850555419921874e-05,
      "model_forward_time": 3.139569044113159,
      "step": 7563
    },
    {
      "epoch": 2.8850555419921874e-05,
      "step": 7563,
      "training_step_time": 9.920025110244751
    },
    {
      "epoch": 2.8850555419921874e-05,
      "model_forward_time": 3.1429014205932617,
      "step": 7563
    },
    {
      "epoch": 2.8850555419921874e-05,
      "step": 7563,
      "training_step_time": 9.92098879814148
    },
    {
      "epoch": 2.8850555419921874e-05,
      "model_forward_time": 3.139353036880493,
      "step": 7563
    },
    {
      "epoch": 2.8850555419921874e-05,
      "step": 7563,
      "training_step_time": 10.333046436309814
    },
    {
      "epoch": 2.88543701171875e-05,
      "model_forward_time": 3.1680264472961426,
      "step": 7564
    },
    {
      "epoch": 2.88543701171875e-05,
      "step": 7564,
      "training_step_time": 9.947327613830566
    },
    {
      "epoch": 2.88543701171875e-05,
      "model_forward_time": 3.4392004013061523,
      "step": 7564
    },
    {
      "epoch": 2.88543701171875e-05,
      "step": 7564,
      "training_step_time": 10.216138362884521
    },
    {
      "epoch": 2.88543701171875e-05,
      "model_forward_time": 3.1360983848571777,
      "step": 7564
    },
    {
      "epoch": 2.88543701171875e-05,
      "step": 7564,
      "training_step_time": 9.910238027572632
    },
    {
      "epoch": 2.88543701171875e-05,
      "model_forward_time": 3.1376123428344727,
      "step": 7564
    },
    {
      "epoch": 2.88543701171875e-05,
      "step": 7564,
      "training_step_time": 10.287294387817383
    },
    {
      "epoch": 2.8858184814453126e-05,
      "model_forward_time": 3.1682803630828857,
      "step": 7565
    },
    {
      "dynamics_loss_avg": 0.05321551244705915,
      "epoch": 2.8858184814453126e-05,
      "step": 7565
    },
    {
      "action_loss_avg": 0.025554353510960935,
      "epoch": 2.8858184814453126e-05,
      "step": 7565
    },
    {
      "epoch": 2.8858184814453126e-05,
      "step": 7565,
      "training_step_time": 9.94694209098816
    },
    {
      "epoch": 2.8858184814453126e-05,
      "model_forward_time": 3.1378540992736816,
      "step": 7565
    },
    {
      "epoch": 2.8858184814453126e-05,
      "step": 7565,
      "training_step_time": 9.916275024414062
    },
    {
      "epoch": 2.8858184814453126e-05,
      "model_forward_time": 3.142425060272217,
      "step": 7565
    },
    {
      "epoch": 2.8858184814453126e-05,
      "step": 7565,
      "training_step_time": 10.20711636543274
    },
    {
      "epoch": 2.8858184814453126e-05,
      "model_forward_time": 3.133796215057373,
      "step": 7565
    },
    {
      "epoch": 2.8858184814453126e-05,
      "step": 7565,
      "training_step_time": 10.00831651687622
    },
    {
      "epoch": 2.886199951171875e-05,
      "model_forward_time": 3.184004068374634,
      "step": 7566
    },
    {
      "epoch": 2.886199951171875e-05,
      "step": 7566,
      "training_step_time": 9.957247734069824
    },
    {
      "epoch": 2.886199951171875e-05,
      "model_forward_time": 3.1391820907592773,
      "step": 7566
    },
    {
      "epoch": 2.886199951171875e-05,
      "step": 7566,
      "training_step_time": 9.913160800933838
    },
    {
      "epoch": 2.886199951171875e-05,
      "model_forward_time": 3.136152505874634,
      "step": 7566
    },
    {
      "epoch": 2.886199951171875e-05,
      "step": 7566,
      "training_step_time": 9.916192770004272
    },
    {
      "epoch": 2.886199951171875e-05,
      "model_forward_time": 3.4088969230651855,
      "step": 7566
    },
    {
      "epoch": 2.886199951171875e-05,
      "step": 7566,
      "training_step_time": 10.313491821289062
    },
    {
      "epoch": 2.8865814208984376e-05,
      "model_forward_time": 3.1721303462982178,
      "step": 7567
    },
    {
      "epoch": 2.8865814208984376e-05,
      "step": 7567,
      "training_step_time": 9.956275701522827
    },
    {
      "epoch": 2.8865814208984376e-05,
      "model_forward_time": 3.1470937728881836,
      "step": 7567
    },
    {
      "epoch": 2.8865814208984376e-05,
      "step": 7567,
      "training_step_time": 9.926266193389893
    },
    {
      "epoch": 2.8865814208984376e-05,
      "model_forward_time": 3.142112970352173,
      "step": 7567
    },
    {
      "dynamics_loss_avg": 0.05217060223221779,
      "epoch": 2.8865814208984376e-05,
      "step": 7567
    },
    {
      "action_loss_avg": 0.022479289304465057,
      "epoch": 2.8865814208984376e-05,
      "step": 7567
    },
    {
      "epoch": 2.8865814208984376e-05,
      "step": 7567,
      "training_step_time": 9.927339792251587
    },
    {
      "epoch": 2.8865814208984376e-05,
      "model_forward_time": 3.1435019969940186,
      "step": 7567
    },
    {
      "epoch": 2.8865814208984376e-05,
      "step": 7567,
      "training_step_time": 10.268524885177612
    },
    {
      "epoch": 2.886962890625e-05,
      "model_forward_time": 3.185114860534668,
      "step": 7568
    },
    {
      "epoch": 2.886962890625e-05,
      "step": 7568,
      "training_step_time": 10.23793649673462
    },
    {
      "epoch": 2.886962890625e-05,
      "model_forward_time": 3.1475186347961426,
      "step": 7568
    },
    {
      "epoch": 2.886962890625e-05,
      "step": 7568,
      "training_step_time": 9.928580522537231
    },
    {
      "epoch": 2.886962890625e-05,
      "model_forward_time": 3.1415092945098877,
      "step": 7568
    },
    {
      "epoch": 2.886962890625e-05,
      "step": 7568,
      "training_step_time": 9.920510530471802
    },
    {
      "epoch": 2.886962890625e-05,
      "model_forward_time": 3.145643949508667,
      "step": 7568
    },
    {
      "epoch": 2.886962890625e-05,
      "step": 7568,
      "training_step_time": 10.235217809677124
    },
    {
      "epoch": 2.8873443603515625e-05,
      "model_forward_time": 3.1796340942382812,
      "step": 7569
    },
    {
      "epoch": 2.8873443603515625e-05,
      "step": 7569,
      "training_step_time": 9.960473775863647
    },
    {
      "epoch": 2.8873443603515625e-05,
      "model_forward_time": 3.417370080947876,
      "step": 7569
    },
    {
      "epoch": 2.8873443603515625e-05,
      "step": 7569,
      "training_step_time": 10.188328742980957
    },
    {
      "epoch": 2.8873443603515625e-05,
      "model_forward_time": 3.1358916759490967,
      "step": 7569
    },
    {
      "epoch": 2.8873443603515625e-05,
      "step": 7569,
      "training_step_time": 9.910259485244751
    },
    {
      "epoch": 2.8873443603515625e-05,
      "model_forward_time": 3.1391959190368652,
      "step": 7569
    },
    {
      "epoch": 2.8873443603515625e-05,
      "step": 7569,
      "training_step_time": 10.261637687683105
    },
    {
      "epoch": 2.887725830078125e-05,
      "grad_norm": 0.5087330937385559,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 7570
    },
    {
      "epoch": 2.887725830078125e-05,
      "model_forward_time": 3.1789636611938477,
      "step": 7570
    },
    {
      "dynamics_loss_avg": 0.059754349291324615,
      "epoch": 2.887725830078125e-05,
      "step": 7570
    },
    {
      "action_loss_avg": 0.021775378659367562,
      "epoch": 2.887725830078125e-05,
      "step": 7570
    },
    {
      "epoch": 2.887725830078125e-05,
      "step": 7570,
      "training_step_time": 9.95055866241455
    },
    {
      "epoch": 2.887725830078125e-05,
      "model_forward_time": 3.13336181640625,
      "step": 7570
    },
    {
      "epoch": 2.887725830078125e-05,
      "step": 7570,
      "training_step_time": 9.902327537536621
    },
    {
      "epoch": 2.887725830078125e-05,
      "model_forward_time": 3.3951611518859863,
      "step": 7570
    },
    {
      "epoch": 2.887725830078125e-05,
      "step": 7570,
      "training_step_time": 10.166962623596191
    },
    {
      "epoch": 2.887725830078125e-05,
      "model_forward_time": 3.1358606815338135,
      "step": 7570
    },
    {
      "epoch": 2.887725830078125e-05,
      "step": 7570,
      "training_step_time": 10.268953084945679
    },
    {
      "epoch": 2.8881072998046874e-05,
      "model_forward_time": 3.1826701164245605,
      "step": 7571
    },
    {
      "epoch": 2.8881072998046874e-05,
      "step": 7571,
      "training_step_time": 9.95792818069458
    },
    {
      "epoch": 2.8881072998046874e-05,
      "model_forward_time": 3.137472152709961,
      "step": 7571
    },
    {
      "epoch": 2.8881072998046874e-05,
      "step": 7571,
      "training_step_time": 9.909432888031006
    },
    {
      "epoch": 2.8881072998046874e-05,
      "model_forward_time": 3.140050172805786,
      "step": 7571
    },
    {
      "epoch": 2.8881072998046874e-05,
      "step": 7571,
      "training_step_time": 9.916318416595459
    },
    {
      "epoch": 2.8881072998046874e-05,
      "model_forward_time": 3.4164135456085205,
      "step": 7571
    },
    {
      "epoch": 2.8881072998046874e-05,
      "step": 7571,
      "training_step_time": 10.288239240646362
    },
    {
      "epoch": 2.88848876953125e-05,
      "model_forward_time": 2.476257801055908,
      "step": 7572
    },
    {
      "epoch": 2.88848876953125e-05,
      "step": 7572,
      "training_step_time": 7.51232123374939
    },
    {
      "epoch": 2.88848876953125e-05,
      "model_forward_time": 3.1384189128875732,
      "step": 7572
    },
    {
      "epoch": 2.88848876953125e-05,
      "step": 7572,
      "training_step_time": 9.91894245147705
    },
    {
      "epoch": 2.88848876953125e-05,
      "model_forward_time": 3.1431174278259277,
      "step": 7572
    },
    {
      "dynamics_loss_avg": 0.057697109319269656,
      "epoch": 2.88848876953125e-05,
      "step": 7572
    },
    {
      "action_loss_avg": 0.02242368757724762,
      "epoch": 2.88848876953125e-05,
      "step": 7572
    },
    {
      "epoch": 2.88848876953125e-05,
      "step": 7572,
      "training_step_time": 9.927289247512817
    },
    {
      "epoch": 2.88848876953125e-05,
      "model_forward_time": 3.1418163776397705,
      "step": 7572
    },
    {
      "epoch": 2.88848876953125e-05,
      "step": 7572,
      "training_step_time": 10.24551248550415
    },
    {
      "epoch": 2.8888702392578127e-05,
      "model_forward_time": 3.447906494140625,
      "step": 7573
    },
    {
      "epoch": 2.8888702392578127e-05,
      "step": 7573,
      "training_step_time": 10.229336500167847
    },
    {
      "epoch": 2.8888702392578127e-05,
      "model_forward_time": 3.1406164169311523,
      "step": 7573
    },
    {
      "epoch": 2.8888702392578127e-05,
      "step": 7573,
      "training_step_time": 9.919405937194824
    },
    {
      "epoch": 2.8888702392578127e-05,
      "model_forward_time": 3.1435999870300293,
      "step": 7573
    },
    {
      "epoch": 2.8888702392578127e-05,
      "step": 7573,
      "training_step_time": 9.919859409332275
    },
    {
      "epoch": 2.8888702392578127e-05,
      "model_forward_time": 3.1383016109466553,
      "step": 7573
    },
    {
      "epoch": 2.8888702392578127e-05,
      "step": 7573,
      "training_step_time": 10.306715250015259
    },
    {
      "epoch": 2.889251708984375e-05,
      "model_forward_time": 3.173978328704834,
      "step": 7574
    },
    {
      "epoch": 2.889251708984375e-05,
      "step": 7574,
      "training_step_time": 9.953712463378906
    },
    {
      "epoch": 2.889251708984375e-05,
      "model_forward_time": 3.3994834423065186,
      "step": 7574
    },
    {
      "epoch": 2.889251708984375e-05,
      "step": 7574,
      "training_step_time": 10.179019212722778
    },
    {
      "epoch": 2.889251708984375e-05,
      "model_forward_time": 3.143996000289917,
      "step": 7574
    },
    {
      "epoch": 2.889251708984375e-05,
      "step": 7574,
      "training_step_time": 9.923353433609009
    },
    {
      "epoch": 2.889251708984375e-05,
      "model_forward_time": 3.139570713043213,
      "step": 7574
    },
    {
      "epoch": 2.889251708984375e-05,
      "step": 7574,
      "training_step_time": 10.302764654159546
    },
    {
      "epoch": 2.8896331787109376e-05,
      "model_forward_time": 3.172206163406372,
      "step": 7575
    },
    {
      "dynamics_loss_avg": 0.0583209915086627,
      "epoch": 2.8896331787109376e-05,
      "step": 7575
    },
    {
      "action_loss_avg": 0.018929498502984643,
      "epoch": 2.8896331787109376e-05,
      "step": 7575
    },
    {
      "epoch": 2.8896331787109376e-05,
      "step": 7575,
      "training_step_time": 9.956298351287842
    },
    {
      "epoch": 2.8896331787109376e-05,
      "model_forward_time": 3.145352602005005,
      "step": 7575
    },
    {
      "epoch": 2.8896331787109376e-05,
      "step": 7575,
      "training_step_time": 9.921270847320557
    },
    {
      "epoch": 2.8896331787109376e-05,
      "model_forward_time": 3.423180341720581,
      "step": 7575
    },
    {
      "epoch": 2.8896331787109376e-05,
      "step": 7575,
      "training_step_time": 10.198753118515015
    },
    {
      "epoch": 2.8896331787109376e-05,
      "model_forward_time": 3.1380834579467773,
      "step": 7575
    },
    {
      "epoch": 2.8896331787109376e-05,
      "step": 7575,
      "training_step_time": 10.211808919906616
    },
    {
      "epoch": 2.8900146484375e-05,
      "model_forward_time": 3.1772570610046387,
      "step": 7576
    },
    {
      "epoch": 2.8900146484375e-05,
      "step": 7576,
      "training_step_time": 9.953006744384766
    },
    {
      "epoch": 2.8900146484375e-05,
      "model_forward_time": 3.144577980041504,
      "step": 7576
    },
    {
      "epoch": 2.8900146484375e-05,
      "step": 7576,
      "training_step_time": 9.919393062591553
    },
    {
      "epoch": 2.8900146484375e-05,
      "model_forward_time": 3.141693115234375,
      "step": 7576
    },
    {
      "epoch": 2.8900146484375e-05,
      "step": 7576,
      "training_step_time": 9.921647071838379
    },
    {
      "epoch": 2.8900146484375e-05,
      "model_forward_time": 3.415353536605835,
      "step": 7576
    },
    {
      "epoch": 2.8900146484375e-05,
      "step": 7576,
      "training_step_time": 10.289591789245605
    },
    {
      "epoch": 2.8903961181640625e-05,
      "model_forward_time": 3.1910152435302734,
      "step": 7577
    },
    {
      "epoch": 2.8903961181640625e-05,
      "step": 7577,
      "training_step_time": 9.974173307418823
    },
    {
      "epoch": 2.8903961181640625e-05,
      "model_forward_time": 3.149569511413574,
      "step": 7577
    },
    {
      "epoch": 2.8903961181640625e-05,
      "step": 7577,
      "training_step_time": 9.927490949630737
    },
    {
      "epoch": 2.8903961181640625e-05,
      "model_forward_time": 3.140698194503784,
      "step": 7577
    },
    {
      "dynamics_loss_avg": 0.061947441473603246,
      "epoch": 2.8903961181640625e-05,
      "step": 7577
    },
    {
      "action_loss_avg": 0.01849699979647994,
      "epoch": 2.8903961181640625e-05,
      "step": 7577
    },
    {
      "epoch": 2.8903961181640625e-05,
      "step": 7577,
      "training_step_time": 9.926842451095581
    },
    {
      "epoch": 2.8903961181640625e-05,
      "model_forward_time": 3.143493890762329,
      "step": 7577
    },
    {
      "epoch": 2.8903961181640625e-05,
      "step": 7577,
      "training_step_time": 10.275241613388062
    },
    {
      "epoch": 2.890777587890625e-05,
      "model_forward_time": 3.457777976989746,
      "step": 7578
    },
    {
      "epoch": 2.890777587890625e-05,
      "step": 7578,
      "training_step_time": 10.240755081176758
    },
    {
      "epoch": 2.890777587890625e-05,
      "model_forward_time": 3.1477744579315186,
      "step": 7578
    },
    {
      "epoch": 2.890777587890625e-05,
      "step": 7578,
      "training_step_time": 9.928275346755981
    },
    {
      "epoch": 2.890777587890625e-05,
      "model_forward_time": 3.1443610191345215,
      "step": 7578
    },
    {
      "epoch": 2.890777587890625e-05,
      "step": 7578,
      "training_step_time": 9.925368070602417
    },
    {
      "epoch": 2.890777587890625e-05,
      "model_forward_time": 3.148854970932007,
      "step": 7578
    },
    {
      "epoch": 2.890777587890625e-05,
      "step": 7578,
      "training_step_time": 10.280218839645386
    },
    {
      "epoch": 2.8911590576171874e-05,
      "model_forward_time": 3.1927435398101807,
      "step": 7579
    },
    {
      "epoch": 2.8911590576171874e-05,
      "step": 7579,
      "training_step_time": 9.975245714187622
    },
    {
      "epoch": 2.8911590576171874e-05,
      "model_forward_time": 3.43880033493042,
      "step": 7579
    },
    {
      "epoch": 2.8911590576171874e-05,
      "step": 7579,
      "training_step_time": 10.216190338134766
    },
    {
      "epoch": 2.8911590576171874e-05,
      "model_forward_time": 3.142235279083252,
      "step": 7579
    },
    {
      "epoch": 2.8911590576171874e-05,
      "step": 7579,
      "training_step_time": 9.922481536865234
    },
    {
      "epoch": 2.8911590576171874e-05,
      "model_forward_time": 3.1465773582458496,
      "step": 7579
    },
    {
      "epoch": 2.8911590576171874e-05,
      "step": 7579,
      "training_step_time": 10.313864469528198
    },
    {
      "epoch": 2.89154052734375e-05,
      "grad_norm": 0.5731578469276428,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 7580
    },
    {
      "epoch": 2.89154052734375e-05,
      "model_forward_time": 3.193913459777832,
      "step": 7580
    },
    {
      "dynamics_loss_avg": 0.05821647979319096,
      "epoch": 2.89154052734375e-05,
      "step": 7580
    },
    {
      "action_loss_avg": 0.019571882556192577,
      "epoch": 2.89154052734375e-05,
      "step": 7580
    },
    {
      "epoch": 2.89154052734375e-05,
      "step": 7580,
      "training_step_time": 9.977476119995117
    },
    {
      "epoch": 2.89154052734375e-05,
      "model_forward_time": 3.1424968242645264,
      "step": 7580
    },
    {
      "epoch": 2.89154052734375e-05,
      "step": 7580,
      "training_step_time": 9.919259786605835
    },
    {
      "epoch": 2.89154052734375e-05,
      "model_forward_time": 3.432732343673706,
      "step": 7580
    },
    {
      "epoch": 2.89154052734375e-05,
      "step": 7580,
      "training_step_time": 10.208387613296509
    },
    {
      "epoch": 2.89154052734375e-05,
      "model_forward_time": 3.145864963531494,
      "step": 7580
    },
    {
      "epoch": 2.89154052734375e-05,
      "step": 7580,
      "training_step_time": 10.28324818611145
    },
    {
      "epoch": 2.8919219970703127e-05,
      "model_forward_time": 3.1862823963165283,
      "step": 7581
    },
    {
      "epoch": 2.8919219970703127e-05,
      "step": 7581,
      "training_step_time": 9.959904193878174
    },
    {
      "epoch": 2.8919219970703127e-05,
      "model_forward_time": 3.1402907371520996,
      "step": 7581
    },
    {
      "epoch": 2.8919219970703127e-05,
      "step": 7581,
      "training_step_time": 9.914734601974487
    },
    {
      "epoch": 2.8919219970703127e-05,
      "model_forward_time": 3.14786958694458,
      "step": 7581
    },
    {
      "epoch": 2.8919219970703127e-05,
      "step": 7581,
      "training_step_time": 9.924306392669678
    },
    {
      "epoch": 2.8919219970703127e-05,
      "model_forward_time": 3.441103219985962,
      "step": 7581
    },
    {
      "epoch": 2.8919219970703127e-05,
      "step": 7581,
      "training_step_time": 10.311002492904663
    },
    {
      "epoch": 2.892303466796875e-05,
      "model_forward_time": 3.186248540878296,
      "step": 7582
    },
    {
      "epoch": 2.892303466796875e-05,
      "step": 7582,
      "training_step_time": 9.966784477233887
    },
    {
      "epoch": 2.892303466796875e-05,
      "model_forward_time": 3.137819766998291,
      "step": 7582
    },
    {
      "epoch": 2.892303466796875e-05,
      "step": 7582,
      "training_step_time": 9.921487092971802
    },
    {
      "epoch": 2.892303466796875e-05,
      "model_forward_time": 3.1440043449401855,
      "step": 7582
    },
    {
      "dynamics_loss_avg": 0.0574223343282938,
      "epoch": 2.892303466796875e-05,
      "step": 7582
    },
    {
      "action_loss_avg": 0.021656098775565624,
      "epoch": 2.892303466796875e-05,
      "step": 7582
    },
    {
      "epoch": 2.892303466796875e-05,
      "step": 7582,
      "training_step_time": 9.919300079345703
    },
    {
      "epoch": 2.892303466796875e-05,
      "model_forward_time": 3.142963171005249,
      "step": 7582
    },
    {
      "epoch": 2.892303466796875e-05,
      "step": 7582,
      "training_step_time": 10.244349002838135
    },
    {
      "epoch": 2.8926849365234376e-05,
      "model_forward_time": 3.1715009212493896,
      "step": 7583
    },
    {
      "epoch": 2.8926849365234376e-05,
      "step": 7583,
      "training_step_time": 10.23277473449707
    },
    {
      "epoch": 2.8926849365234376e-05,
      "model_forward_time": 3.1401405334472656,
      "step": 7583
    },
    {
      "epoch": 2.8926849365234376e-05,
      "step": 7583,
      "training_step_time": 9.910234928131104
    },
    {
      "epoch": 2.8926849365234376e-05,
      "model_forward_time": 3.144233465194702,
      "step": 7583
    },
    {
      "epoch": 2.8926849365234376e-05,
      "step": 7583,
      "training_step_time": 9.922119379043579
    },
    {
      "epoch": 2.8926849365234376e-05,
      "model_forward_time": 3.1402246952056885,
      "step": 7583
    },
    {
      "epoch": 2.8926849365234376e-05,
      "step": 7583,
      "training_step_time": 10.276172637939453
    },
    {
      "epoch": 2.89306640625e-05,
      "model_forward_time": 3.1777760982513428,
      "step": 7584
    },
    {
      "epoch": 2.89306640625e-05,
      "step": 7584,
      "training_step_time": 9.953234672546387
    },
    {
      "epoch": 2.89306640625e-05,
      "model_forward_time": 3.4050140380859375,
      "step": 7584
    },
    {
      "epoch": 2.89306640625e-05,
      "step": 7584,
      "training_step_time": 10.179157733917236
    },
    {
      "epoch": 2.89306640625e-05,
      "model_forward_time": 3.144473075866699,
      "step": 7584
    },
    {
      "epoch": 2.89306640625e-05,
      "step": 7584,
      "training_step_time": 9.918323278427124
    },
    {
      "epoch": 2.89306640625e-05,
      "model_forward_time": 3.1371638774871826,
      "step": 7584
    },
    {
      "epoch": 2.89306640625e-05,
      "step": 7584,
      "training_step_time": 10.314396619796753
    },
    {
      "epoch": 2.8934478759765625e-05,
      "model_forward_time": 3.178511142730713,
      "step": 7585
    },
    {
      "dynamics_loss_avg": 0.06420463770627975,
      "epoch": 2.8934478759765625e-05,
      "step": 7585
    },
    {
      "action_loss_avg": 0.026906687393784524,
      "epoch": 2.8934478759765625e-05,
      "step": 7585
    },
    {
      "epoch": 2.8934478759765625e-05,
      "step": 7585,
      "training_step_time": 9.9628746509552
    },
    {
      "epoch": 2.8934478759765625e-05,
      "model_forward_time": 3.1411044597625732,
      "step": 7585
    },
    {
      "epoch": 2.8934478759765625e-05,
      "step": 7585,
      "training_step_time": 9.91362476348877
    },
    {
      "epoch": 2.8934478759765625e-05,
      "model_forward_time": 3.1373980045318604,
      "step": 7585
    },
    {
      "epoch": 2.8934478759765625e-05,
      "step": 7585,
      "training_step_time": 10.185454607009888
    },
    {
      "epoch": 2.8934478759765625e-05,
      "model_forward_time": 3.138319492340088,
      "step": 7585
    },
    {
      "epoch": 2.8934478759765625e-05,
      "step": 7585,
      "training_step_time": 10.252556562423706
    },
    {
      "epoch": 2.893829345703125e-05,
      "model_forward_time": 3.1854543685913086,
      "step": 7586
    },
    {
      "epoch": 2.893829345703125e-05,
      "step": 7586,
      "training_step_time": 9.955787658691406
    },
    {
      "epoch": 2.893829345703125e-05,
      "model_forward_time": 3.1364517211914062,
      "step": 7586
    },
    {
      "epoch": 2.893829345703125e-05,
      "step": 7586,
      "training_step_time": 9.910582304000854
    },
    {
      "epoch": 2.893829345703125e-05,
      "model_forward_time": 3.1356306076049805,
      "step": 7586
    },
    {
      "epoch": 2.893829345703125e-05,
      "step": 7586,
      "training_step_time": 9.907110929489136
    },
    {
      "epoch": 2.893829345703125e-05,
      "model_forward_time": 3.4066309928894043,
      "step": 7586
    },
    {
      "epoch": 2.893829345703125e-05,
      "step": 7586,
      "training_step_time": 10.270863771438599
    },
    {
      "epoch": 2.8942108154296874e-05,
      "model_forward_time": 3.17791748046875,
      "step": 7587
    },
    {
      "epoch": 2.8942108154296874e-05,
      "step": 7587,
      "training_step_time": 9.95061993598938
    },
    {
      "epoch": 2.8942108154296874e-05,
      "model_forward_time": 3.141176700592041,
      "step": 7587
    },
    {
      "epoch": 2.8942108154296874e-05,
      "step": 7587,
      "training_step_time": 9.916350603103638
    },
    {
      "epoch": 2.8942108154296874e-05,
      "model_forward_time": 3.1379711627960205,
      "step": 7587
    },
    {
      "dynamics_loss_avg": 0.06793020479381084,
      "epoch": 2.8942108154296874e-05,
      "step": 7587
    },
    {
      "action_loss_avg": 0.02907817605882883,
      "epoch": 2.8942108154296874e-05,
      "step": 7587
    },
    {
      "epoch": 2.8942108154296874e-05,
      "step": 7587,
      "training_step_time": 9.911717891693115
    },
    {
      "epoch": 2.8942108154296874e-05,
      "model_forward_time": 3.1443636417388916,
      "step": 7587
    },
    {
      "epoch": 2.8942108154296874e-05,
      "step": 7587,
      "training_step_time": 10.016538143157959
    },
    {
      "epoch": 2.89459228515625e-05,
      "model_forward_time": 3.1720974445343018,
      "step": 7588
    },
    {
      "epoch": 2.89459228515625e-05,
      "step": 7588,
      "training_step_time": 10.213910341262817
    },
    {
      "epoch": 2.89459228515625e-05,
      "model_forward_time": 3.1407761573791504,
      "step": 7588
    },
    {
      "epoch": 2.89459228515625e-05,
      "step": 7588,
      "training_step_time": 9.910380363464355
    },
    {
      "epoch": 2.89459228515625e-05,
      "model_forward_time": 3.1391241550445557,
      "step": 7588
    },
    {
      "epoch": 2.89459228515625e-05,
      "step": 7588,
      "training_step_time": 9.908511400222778
    },
    {
      "epoch": 2.89459228515625e-05,
      "model_forward_time": 3.140261650085449,
      "step": 7588
    },
    {
      "epoch": 2.89459228515625e-05,
      "step": 7588,
      "training_step_time": 10.26836371421814
    },
    {
      "epoch": 2.8949737548828127e-05,
      "model_forward_time": 3.167900800704956,
      "step": 7589
    },
    {
      "epoch": 2.8949737548828127e-05,
      "step": 7589,
      "training_step_time": 9.941676139831543
    },
    {
      "epoch": 2.8949737548828127e-05,
      "model_forward_time": 3.427013874053955,
      "step": 7589
    },
    {
      "epoch": 2.8949737548828127e-05,
      "step": 7589,
      "training_step_time": 10.197749853134155
    },
    {
      "epoch": 2.8949737548828127e-05,
      "model_forward_time": 3.1365103721618652,
      "step": 7589
    },
    {
      "epoch": 2.8949737548828127e-05,
      "step": 7589,
      "training_step_time": 9.904528379440308
    },
    {
      "epoch": 2.8949737548828127e-05,
      "model_forward_time": 3.14162015914917,
      "step": 7589
    },
    {
      "epoch": 2.8949737548828127e-05,
      "step": 7589,
      "training_step_time": 10.284019231796265
    },
    {
      "epoch": 2.895355224609375e-05,
      "grad_norm": 0.5749658942222595,
      "learning_rate": 1e-05,
      "loss": 0.0829,
      "step": 7590
    },
    {
      "epoch": 2.895355224609375e-05,
      "model_forward_time": 3.1656248569488525,
      "step": 7590
    },
    {
      "dynamics_loss_avg": 0.07027901485562324,
      "epoch": 2.895355224609375e-05,
      "step": 7590
    },
    {
      "action_loss_avg": 0.02793555874377489,
      "epoch": 2.895355224609375e-05,
      "step": 7590
    },
    {
      "epoch": 2.895355224609375e-05,
      "step": 7590,
      "training_step_time": 9.940073251724243
    },
    {
      "epoch": 2.895355224609375e-05,
      "model_forward_time": 3.13155198097229,
      "step": 7590
    },
    {
      "epoch": 2.895355224609375e-05,
      "step": 7590,
      "training_step_time": 9.90121841430664
    },
    {
      "epoch": 2.895355224609375e-05,
      "model_forward_time": 3.3948731422424316,
      "step": 7590
    },
    {
      "epoch": 2.895355224609375e-05,
      "step": 7590,
      "training_step_time": 10.169898748397827
    },
    {
      "epoch": 2.895355224609375e-05,
      "model_forward_time": 3.1429526805877686,
      "step": 7590
    },
    {
      "epoch": 2.895355224609375e-05,
      "step": 7590,
      "training_step_time": 10.239064693450928
    },
    {
      "epoch": 2.8957366943359376e-05,
      "model_forward_time": 3.168863296508789,
      "step": 7591
    },
    {
      "epoch": 2.8957366943359376e-05,
      "step": 7591,
      "training_step_time": 9.945971965789795
    },
    {
      "epoch": 2.8957366943359376e-05,
      "model_forward_time": 3.1315205097198486,
      "step": 7591
    },
    {
      "epoch": 2.8957366943359376e-05,
      "step": 7591,
      "training_step_time": 9.919551610946655
    },
    {
      "epoch": 2.8957366943359376e-05,
      "model_forward_time": 3.1353678703308105,
      "step": 7591
    },
    {
      "epoch": 2.8957366943359376e-05,
      "step": 7591,
      "training_step_time": 9.915092468261719
    },
    {
      "epoch": 2.8957366943359376e-05,
      "model_forward_time": 3.420311450958252,
      "step": 7591
    },
    {
      "epoch": 2.8957366943359376e-05,
      "step": 7591,
      "training_step_time": 10.292553424835205
    },
    {
      "epoch": 2.8961181640625e-05,
      "model_forward_time": 3.195590019226074,
      "step": 7592
    },
    {
      "epoch": 2.8961181640625e-05,
      "step": 7592,
      "training_step_time": 9.978258848190308
    },
    {
      "epoch": 2.8961181640625e-05,
      "model_forward_time": 3.1352672576904297,
      "step": 7592
    },
    {
      "epoch": 2.8961181640625e-05,
      "step": 7592,
      "training_step_time": 9.913701295852661
    },
    {
      "epoch": 2.8961181640625e-05,
      "model_forward_time": 3.1386678218841553,
      "step": 7592
    },
    {
      "dynamics_loss_avg": 0.07248447239398956,
      "epoch": 2.8961181640625e-05,
      "step": 7592
    },
    {
      "action_loss_avg": 0.025792561192065477,
      "epoch": 2.8961181640625e-05,
      "step": 7592
    },
    {
      "epoch": 2.8961181640625e-05,
      "step": 7592,
      "training_step_time": 9.918163299560547
    },
    {
      "epoch": 2.8961181640625e-05,
      "model_forward_time": 3.135406255722046,
      "step": 7592
    },
    {
      "epoch": 2.8961181640625e-05,
      "step": 7592,
      "training_step_time": 10.318613529205322
    },
    {
      "epoch": 2.8964996337890625e-05,
      "model_forward_time": 3.443402051925659,
      "step": 7593
    },
    {
      "epoch": 2.8964996337890625e-05,
      "step": 7593,
      "training_step_time": 10.219505071640015
    },
    {
      "epoch": 2.8964996337890625e-05,
      "model_forward_time": 3.1401712894439697,
      "step": 7593
    },
    {
      "epoch": 2.8964996337890625e-05,
      "step": 7593,
      "training_step_time": 9.916991472244263
    },
    {
      "epoch": 2.8964996337890625e-05,
      "model_forward_time": 3.1407220363616943,
      "step": 7593
    },
    {
      "epoch": 2.8964996337890625e-05,
      "step": 7593,
      "training_step_time": 9.914283037185669
    },
    {
      "epoch": 2.8964996337890625e-05,
      "model_forward_time": 3.1356210708618164,
      "step": 7593
    },
    {
      "epoch": 2.8964996337890625e-05,
      "step": 7593,
      "training_step_time": 10.250832796096802
    },
    {
      "epoch": 2.896881103515625e-05,
      "model_forward_time": 3.1666324138641357,
      "step": 7594
    },
    {
      "epoch": 2.896881103515625e-05,
      "step": 7594,
      "training_step_time": 9.944089412689209
    },
    {
      "epoch": 2.896881103515625e-05,
      "model_forward_time": 3.3861865997314453,
      "step": 7594
    },
    {
      "epoch": 2.896881103515625e-05,
      "step": 7594,
      "training_step_time": 10.164125919342041
    },
    {
      "epoch": 2.896881103515625e-05,
      "model_forward_time": 3.1446423530578613,
      "step": 7594
    },
    {
      "epoch": 2.896881103515625e-05,
      "step": 7594,
      "training_step_time": 9.92361831665039
    },
    {
      "epoch": 2.896881103515625e-05,
      "model_forward_time": 3.139475107192993,
      "step": 7594
    },
    {
      "epoch": 2.896881103515625e-05,
      "step": 7594,
      "training_step_time": 10.3225417137146
    },
    {
      "epoch": 2.8972625732421874e-05,
      "model_forward_time": 3.169635772705078,
      "step": 7595
    },
    {
      "dynamics_loss_avg": 0.05832974463701248,
      "epoch": 2.8972625732421874e-05,
      "step": 7595
    },
    {
      "action_loss_avg": 0.022447227034717798,
      "epoch": 2.8972625732421874e-05,
      "step": 7595
    },
    {
      "epoch": 2.8972625732421874e-05,
      "step": 7595,
      "training_step_time": 9.952127933502197
    },
    {
      "epoch": 2.8972625732421874e-05,
      "model_forward_time": 3.1387524604797363,
      "step": 7595
    },
    {
      "epoch": 2.8972625732421874e-05,
      "step": 7595,
      "training_step_time": 9.924046993255615
    },
    {
      "epoch": 2.8972625732421874e-05,
      "model_forward_time": 3.4143505096435547,
      "step": 7595
    },
    {
      "epoch": 2.8972625732421874e-05,
      "step": 7595,
      "training_step_time": 10.194822072982788
    },
    {
      "epoch": 2.8972625732421874e-05,
      "model_forward_time": 3.136679172515869,
      "step": 7595
    },
    {
      "epoch": 2.8972625732421874e-05,
      "step": 7595,
      "training_step_time": 10.259024620056152
    },
    {
      "epoch": 2.89764404296875e-05,
      "model_forward_time": 3.1660938262939453,
      "step": 7596
    },
    {
      "epoch": 2.89764404296875e-05,
      "step": 7596,
      "training_step_time": 9.946704864501953
    },
    {
      "epoch": 2.89764404296875e-05,
      "model_forward_time": 3.13625431060791,
      "step": 7596
    },
    {
      "epoch": 2.89764404296875e-05,
      "step": 7596,
      "training_step_time": 9.908346176147461
    },
    {
      "epoch": 2.89764404296875e-05,
      "model_forward_time": 3.1340904235839844,
      "step": 7596
    },
    {
      "epoch": 2.89764404296875e-05,
      "step": 7596,
      "training_step_time": 9.905909776687622
    },
    {
      "epoch": 2.89764404296875e-05,
      "model_forward_time": 3.3987388610839844,
      "step": 7596
    },
    {
      "epoch": 2.89764404296875e-05,
      "step": 7596,
      "training_step_time": 10.265215396881104
    },
    {
      "epoch": 2.8980255126953127e-05,
      "model_forward_time": 3.174640655517578,
      "step": 7597
    },
    {
      "epoch": 2.8980255126953127e-05,
      "step": 7597,
      "training_step_time": 9.946253299713135
    },
    {
      "epoch": 2.8980255126953127e-05,
      "model_forward_time": 3.13493013381958,
      "step": 7597
    },
    {
      "epoch": 2.8980255126953127e-05,
      "step": 7597,
      "training_step_time": 9.909695386886597
    },
    {
      "epoch": 2.8980255126953127e-05,
      "model_forward_time": 3.1317503452301025,
      "step": 7597
    },
    {
      "dynamics_loss_avg": 0.06290197856724263,
      "epoch": 2.8980255126953127e-05,
      "step": 7597
    },
    {
      "action_loss_avg": 0.024188516568392514,
      "epoch": 2.8980255126953127e-05,
      "step": 7597
    },
    {
      "epoch": 2.8980255126953127e-05,
      "step": 7597,
      "training_step_time": 9.910628318786621
    },
    {
      "epoch": 2.8980255126953127e-05,
      "model_forward_time": 3.1395952701568604,
      "step": 7597
    },
    {
      "epoch": 2.8980255126953127e-05,
      "step": 7597,
      "training_step_time": 10.0134437084198
    },
    {
      "epoch": 2.898406982421875e-05,
      "model_forward_time": 3.4407432079315186,
      "step": 7598
    },
    {
      "epoch": 2.898406982421875e-05,
      "step": 7598,
      "training_step_time": 10.217419624328613
    },
    {
      "epoch": 2.898406982421875e-05,
      "model_forward_time": 3.1393988132476807,
      "step": 7598
    },
    {
      "epoch": 2.898406982421875e-05,
      "step": 7598,
      "training_step_time": 9.916658163070679
    },
    {
      "epoch": 2.898406982421875e-05,
      "model_forward_time": 3.1378684043884277,
      "step": 7598
    },
    {
      "epoch": 2.898406982421875e-05,
      "step": 7598,
      "training_step_time": 9.918246030807495
    },
    {
      "epoch": 2.898406982421875e-05,
      "model_forward_time": 3.1387593746185303,
      "step": 7598
    },
    {
      "epoch": 2.898406982421875e-05,
      "step": 7598,
      "training_step_time": 10.0212721824646
    },
    {
      "epoch": 2.8987884521484376e-05,
      "model_forward_time": 3.18182373046875,
      "step": 7599
    },
    {
      "epoch": 2.8987884521484376e-05,
      "step": 7599,
      "training_step_time": 9.96242642402649
    },
    {
      "epoch": 2.8987884521484376e-05,
      "model_forward_time": 3.4189023971557617,
      "step": 7599
    },
    {
      "epoch": 2.8987884521484376e-05,
      "step": 7599,
      "training_step_time": 10.192718744277954
    },
    {
      "epoch": 2.8987884521484376e-05,
      "model_forward_time": 3.138512372970581,
      "step": 7599
    },
    {
      "epoch": 2.8987884521484376e-05,
      "step": 7599,
      "training_step_time": 9.917326211929321
    },
    {
      "epoch": 2.8987884521484376e-05,
      "model_forward_time": 3.14068341255188,
      "step": 7599
    },
    {
      "epoch": 2.8987884521484376e-05,
      "step": 7599,
      "training_step_time": 10.304290533065796
    },
    {
      "epoch": 2.899169921875e-05,
      "grad_norm": 0.4892193377017975,
      "learning_rate": 1e-05,
      "loss": 0.0829,
      "step": 7600
    },
    {
      "epoch": 2.899169921875e-05,
      "model_forward_time": 3.1727983951568604,
      "step": 7600
    },
    {
      "dynamics_loss_avg": 0.05623045414686203,
      "epoch": 2.899169921875e-05,
      "step": 7600
    },
    {
      "action_loss_avg": 0.02314018039032817,
      "epoch": 2.899169921875e-05,
      "step": 7600
    },
    {
      "epoch": 2.899169921875e-05,
      "step": 7600,
      "training_step_time": 9.956939935684204
    },
    {
      "epoch": 2.899169921875e-05,
      "model_forward_time": 3.13588285446167,
      "step": 7600
    },
    {
      "epoch": 2.899169921875e-05,
      "step": 7600,
      "training_step_time": 9.911321640014648
    },
    {
      "epoch": 2.899169921875e-05,
      "model_forward_time": 3.405644655227661,
      "step": 7600
    },
    {
      "epoch": 2.899169921875e-05,
      "step": 7600,
      "training_step_time": 10.181372165679932
    },
    {
      "epoch": 2.899169921875e-05,
      "model_forward_time": 3.141630172729492,
      "step": 7600
    },
    {
      "epoch": 2.899169921875e-05,
      "step": 7600,
      "training_step_time": 10.291729927062988
    },
    {
      "epoch": 2.8995513916015625e-05,
      "model_forward_time": 3.1777572631835938,
      "step": 7601
    },
    {
      "epoch": 2.8995513916015625e-05,
      "step": 7601,
      "training_step_time": 9.954858303070068
    },
    {
      "epoch": 2.8995513916015625e-05,
      "model_forward_time": 3.1333296298980713,
      "step": 7601
    },
    {
      "epoch": 2.8995513916015625e-05,
      "step": 7601,
      "training_step_time": 9.90749979019165
    },
    {
      "epoch": 2.8995513916015625e-05,
      "model_forward_time": 3.137324571609497,
      "step": 7601
    },
    {
      "epoch": 2.8995513916015625e-05,
      "step": 7601,
      "training_step_time": 9.914203643798828
    },
    {
      "epoch": 2.8995513916015625e-05,
      "model_forward_time": 3.418863296508789,
      "step": 7601
    },
    {
      "epoch": 2.8995513916015625e-05,
      "step": 7601,
      "training_step_time": 10.309677600860596
    },
    {
      "epoch": 2.899932861328125e-05,
      "model_forward_time": 3.184415817260742,
      "step": 7602
    },
    {
      "epoch": 2.899932861328125e-05,
      "step": 7602,
      "training_step_time": 9.964441061019897
    },
    {
      "epoch": 2.899932861328125e-05,
      "model_forward_time": 3.1365365982055664,
      "step": 7602
    },
    {
      "epoch": 2.899932861328125e-05,
      "step": 7602,
      "training_step_time": 9.916678190231323
    },
    {
      "epoch": 2.899932861328125e-05,
      "model_forward_time": 3.136990785598755,
      "step": 7602
    },
    {
      "dynamics_loss_avg": 0.054042131453752515,
      "epoch": 2.899932861328125e-05,
      "step": 7602
    },
    {
      "action_loss_avg": 0.01854753540828824,
      "epoch": 2.899932861328125e-05,
      "step": 7602
    },
    {
      "epoch": 2.899932861328125e-05,
      "step": 7602,
      "training_step_time": 9.92544412612915
    },
    {
      "epoch": 2.899932861328125e-05,
      "model_forward_time": 3.1432535648345947,
      "step": 7602
    },
    {
      "epoch": 2.899932861328125e-05,
      "step": 7602,
      "training_step_time": 10.025230169296265
    },
    {
      "epoch": 2.9003143310546874e-05,
      "model_forward_time": 3.4414303302764893,
      "step": 7603
    },
    {
      "epoch": 2.9003143310546874e-05,
      "step": 7603,
      "training_step_time": 10.222336292266846
    },
    {
      "epoch": 2.9003143310546874e-05,
      "model_forward_time": 3.136906147003174,
      "step": 7603
    },
    {
      "epoch": 2.9003143310546874e-05,
      "step": 7603,
      "training_step_time": 9.920769929885864
    },
    {
      "epoch": 2.9003143310546874e-05,
      "model_forward_time": 3.1398985385894775,
      "step": 7603
    },
    {
      "epoch": 2.9003143310546874e-05,
      "step": 7603,
      "training_step_time": 9.916451215744019
    },
    {
      "epoch": 2.9003143310546874e-05,
      "model_forward_time": 3.133535146713257,
      "step": 7603
    },
    {
      "epoch": 2.9003143310546874e-05,
      "step": 7603,
      "training_step_time": 10.008172273635864
    },
    {
      "epoch": 2.90069580078125e-05,
      "model_forward_time": 3.164649248123169,
      "step": 7604
    },
    {
      "epoch": 2.90069580078125e-05,
      "step": 7604,
      "training_step_time": 9.944036483764648
    },
    {
      "epoch": 2.90069580078125e-05,
      "model_forward_time": 3.384876251220703,
      "step": 7604
    },
    {
      "epoch": 2.90069580078125e-05,
      "step": 7604,
      "training_step_time": 10.165737628936768
    },
    {
      "epoch": 2.90069580078125e-05,
      "model_forward_time": 3.139467477798462,
      "step": 7604
    },
    {
      "epoch": 2.90069580078125e-05,
      "step": 7604,
      "training_step_time": 9.920900106430054
    },
    {
      "epoch": 2.90069580078125e-05,
      "model_forward_time": 3.14103102684021,
      "step": 7604
    },
    {
      "epoch": 2.90069580078125e-05,
      "step": 7604,
      "training_step_time": 10.300692796707153
    },
    {
      "epoch": 2.9010772705078127e-05,
      "model_forward_time": 3.156647205352783,
      "step": 7605
    },
    {
      "dynamics_loss_avg": 0.0607687771320343,
      "epoch": 2.9010772705078127e-05,
      "step": 7605
    },
    {
      "action_loss_avg": 0.02659575887955725,
      "epoch": 2.9010772705078127e-05,
      "step": 7605
    },
    {
      "epoch": 2.9010772705078127e-05,
      "step": 7605,
      "training_step_time": 9.938493967056274
    },
    {
      "epoch": 2.9010772705078127e-05,
      "model_forward_time": 3.1364998817443848,
      "step": 7605
    },
    {
      "epoch": 2.9010772705078127e-05,
      "step": 7605,
      "training_step_time": 9.91960620880127
    },
    {
      "epoch": 2.9010772705078127e-05,
      "model_forward_time": 3.4098007678985596,
      "step": 7605
    },
    {
      "epoch": 2.9010772705078127e-05,
      "step": 7605,
      "training_step_time": 10.186385154724121
    },
    {
      "epoch": 2.9010772705078127e-05,
      "model_forward_time": 3.1350600719451904,
      "step": 7605
    },
    {
      "epoch": 2.9010772705078127e-05,
      "step": 7605,
      "training_step_time": 10.29926872253418
    },
    {
      "epoch": 2.901458740234375e-05,
      "model_forward_time": 3.1807315349578857,
      "step": 7606
    },
    {
      "epoch": 2.901458740234375e-05,
      "step": 7606,
      "training_step_time": 9.9585280418396
    },
    {
      "epoch": 2.901458740234375e-05,
      "model_forward_time": 3.1347711086273193,
      "step": 7606
    },
    {
      "epoch": 2.901458740234375e-05,
      "step": 7606,
      "training_step_time": 9.912809610366821
    },
    {
      "epoch": 2.901458740234375e-05,
      "model_forward_time": 3.1372604370117188,
      "step": 7606
    },
    {
      "epoch": 2.901458740234375e-05,
      "step": 7606,
      "training_step_time": 9.915821552276611
    },
    {
      "epoch": 2.901458740234375e-05,
      "model_forward_time": 3.386902332305908,
      "step": 7606
    },
    {
      "epoch": 2.901458740234375e-05,
      "step": 7606,
      "training_step_time": 10.264502763748169
    },
    {
      "epoch": 2.9018402099609376e-05,
      "model_forward_time": 3.1930201053619385,
      "step": 7607
    },
    {
      "epoch": 2.9018402099609376e-05,
      "step": 7607,
      "training_step_time": 9.973820924758911
    },
    {
      "epoch": 2.9018402099609376e-05,
      "model_forward_time": 3.140847682952881,
      "step": 7607
    },
    {
      "epoch": 2.9018402099609376e-05,
      "step": 7607,
      "training_step_time": 9.923252582550049
    },
    {
      "epoch": 2.9018402099609376e-05,
      "model_forward_time": 3.140972375869751,
      "step": 7607
    },
    {
      "dynamics_loss_avg": 0.04881849270313978,
      "epoch": 2.9018402099609376e-05,
      "step": 7607
    },
    {
      "action_loss_avg": 0.018578722700476648,
      "epoch": 2.9018402099609376e-05,
      "step": 7607
    },
    {
      "epoch": 2.9018402099609376e-05,
      "step": 7607,
      "training_step_time": 9.925477743148804
    },
    {
      "epoch": 2.9018402099609376e-05,
      "model_forward_time": 3.142364025115967,
      "step": 7607
    },
    {
      "epoch": 2.9018402099609376e-05,
      "step": 7607,
      "training_step_time": 10.021912574768066
    },
    {
      "epoch": 2.9022216796875e-05,
      "model_forward_time": 3.4622914791107178,
      "step": 7608
    },
    {
      "epoch": 2.9022216796875e-05,
      "step": 7608,
      "training_step_time": 10.24323844909668
    },
    {
      "epoch": 2.9022216796875e-05,
      "model_forward_time": 3.141597032546997,
      "step": 7608
    },
    {
      "epoch": 2.9022216796875e-05,
      "step": 7608,
      "training_step_time": 9.919002532958984
    },
    {
      "epoch": 2.9022216796875e-05,
      "model_forward_time": 3.1344683170318604,
      "step": 7608
    },
    {
      "epoch": 2.9022216796875e-05,
      "step": 7608,
      "training_step_time": 9.916027545928955
    },
    {
      "epoch": 2.9022216796875e-05,
      "model_forward_time": 3.137357234954834,
      "step": 7608
    },
    {
      "epoch": 2.9022216796875e-05,
      "step": 7608,
      "training_step_time": 10.018614292144775
    },
    {
      "epoch": 2.9026031494140625e-05,
      "model_forward_time": 3.4717628955841064,
      "step": 7609
    },
    {
      "epoch": 2.9026031494140625e-05,
      "step": 7609,
      "training_step_time": 10.249352216720581
    },
    {
      "epoch": 2.9026031494140625e-05,
      "model_forward_time": 3.1373846530914307,
      "step": 7609
    },
    {
      "epoch": 2.9026031494140625e-05,
      "step": 7609,
      "training_step_time": 9.92045545578003
    },
    {
      "epoch": 2.9026031494140625e-05,
      "model_forward_time": 3.1407270431518555,
      "step": 7609
    },
    {
      "epoch": 2.9026031494140625e-05,
      "step": 7609,
      "training_step_time": 9.927660942077637
    },
    {
      "epoch": 2.9026031494140625e-05,
      "model_forward_time": 3.143355369567871,
      "step": 7609
    },
    {
      "epoch": 2.9026031494140625e-05,
      "step": 7609,
      "training_step_time": 10.32764220237732
    },
    {
      "epoch": 2.902984619140625e-05,
      "grad_norm": 0.41991886496543884,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 7610
    },
    {
      "epoch": 2.902984619140625e-05,
      "model_forward_time": 3.1918909549713135,
      "step": 7610
    },
    {
      "dynamics_loss_avg": 0.05434451252222061,
      "epoch": 2.902984619140625e-05,
      "step": 7610
    },
    {
      "action_loss_avg": 0.01715727630071342,
      "epoch": 2.902984619140625e-05,
      "step": 7610
    },
    {
      "epoch": 2.902984619140625e-05,
      "step": 7610,
      "training_step_time": 9.979798078536987
    },
    {
      "epoch": 2.902984619140625e-05,
      "model_forward_time": 3.1377480030059814,
      "step": 7610
    },
    {
      "epoch": 2.902984619140625e-05,
      "step": 7610,
      "training_step_time": 10.18880033493042
    },
    {
      "epoch": 2.902984619140625e-05,
      "model_forward_time": 3.1350786685943604,
      "step": 7610
    },
    {
      "epoch": 2.902984619140625e-05,
      "step": 7610,
      "training_step_time": 9.913085222244263
    },
    {
      "epoch": 2.902984619140625e-05,
      "model_forward_time": 3.1384963989257812,
      "step": 7610
    },
    {
      "epoch": 2.902984619140625e-05,
      "step": 7610,
      "training_step_time": 10.28046989440918
    },
    {
      "epoch": 2.9033660888671874e-05,
      "model_forward_time": 3.185173273086548,
      "step": 7611
    },
    {
      "epoch": 2.9033660888671874e-05,
      "step": 7611,
      "training_step_time": 9.961342334747314
    },
    {
      "epoch": 2.9033660888671874e-05,
      "model_forward_time": 3.1339712142944336,
      "step": 7611
    },
    {
      "epoch": 2.9033660888671874e-05,
      "step": 7611,
      "training_step_time": 9.908539056777954
    },
    {
      "epoch": 2.9033660888671874e-05,
      "model_forward_time": 3.1382346153259277,
      "step": 7611
    },
    {
      "epoch": 2.9033660888671874e-05,
      "step": 7611,
      "training_step_time": 10.202409982681274
    },
    {
      "epoch": 2.9033660888671874e-05,
      "model_forward_time": 3.140669822692871,
      "step": 7611
    },
    {
      "epoch": 2.9033660888671874e-05,
      "step": 7611,
      "training_step_time": 10.274136066436768
    },
    {
      "epoch": 2.90374755859375e-05,
      "model_forward_time": 3.1634132862091064,
      "step": 7612
    },
    {
      "epoch": 2.90374755859375e-05,
      "step": 7612,
      "training_step_time": 9.942785263061523
    },
    {
      "epoch": 2.90374755859375e-05,
      "model_forward_time": 3.1330065727233887,
      "step": 7612
    },
    {
      "epoch": 2.90374755859375e-05,
      "step": 7612,
      "training_step_time": 9.911873579025269
    },
    {
      "epoch": 2.90374755859375e-05,
      "model_forward_time": 3.1387505531311035,
      "step": 7612
    },
    {
      "dynamics_loss_avg": 0.054384422674775126,
      "epoch": 2.90374755859375e-05,
      "step": 7612
    },
    {
      "action_loss_avg": 0.01879479824565351,
      "epoch": 2.90374755859375e-05,
      "step": 7612
    },
    {
      "epoch": 2.90374755859375e-05,
      "step": 7612,
      "training_step_time": 9.91853141784668
    },
    {
      "epoch": 2.90374755859375e-05,
      "model_forward_time": 3.4225518703460693,
      "step": 7612
    },
    {
      "epoch": 2.90374755859375e-05,
      "step": 7612,
      "training_step_time": 10.295711755752563
    },
    {
      "epoch": 2.9041290283203127e-05,
      "model_forward_time": 3.1918134689331055,
      "step": 7613
    },
    {
      "epoch": 2.9041290283203127e-05,
      "step": 7613,
      "training_step_time": 9.977251052856445
    },
    {
      "epoch": 2.9041290283203127e-05,
      "model_forward_time": 3.138676166534424,
      "step": 7613
    },
    {
      "epoch": 2.9041290283203127e-05,
      "step": 7613,
      "training_step_time": 9.921593189239502
    },
    {
      "epoch": 2.9041290283203127e-05,
      "model_forward_time": 3.1411516666412354,
      "step": 7613
    },
    {
      "epoch": 2.9041290283203127e-05,
      "step": 7613,
      "training_step_time": 9.922553062438965
    },
    {
      "epoch": 2.9041290283203127e-05,
      "model_forward_time": 3.1365370750427246,
      "step": 7613
    },
    {
      "epoch": 2.9041290283203127e-05,
      "step": 7613,
      "training_step_time": 10.276600122451782
    },
    {
      "epoch": 2.904510498046875e-05,
      "model_forward_time": 3.180832862854004,
      "step": 7614
    },
    {
      "epoch": 2.904510498046875e-05,
      "step": 7614,
      "training_step_time": 10.238516330718994
    },
    {
      "epoch": 2.904510498046875e-05,
      "model_forward_time": 3.139108657836914,
      "step": 7614
    },
    {
      "epoch": 2.904510498046875e-05,
      "step": 7614,
      "training_step_time": 9.923728942871094
    },
    {
      "epoch": 2.904510498046875e-05,
      "model_forward_time": 3.1386210918426514,
      "step": 7614
    },
    {
      "epoch": 2.904510498046875e-05,
      "step": 7614,
      "training_step_time": 9.917725324630737
    },
    {
      "epoch": 2.904510498046875e-05,
      "model_forward_time": 3.1349406242370605,
      "step": 7614
    },
    {
      "epoch": 2.904510498046875e-05,
      "step": 7614,
      "training_step_time": 10.014812469482422
    },
    {
      "epoch": 2.9048919677734376e-05,
      "model_forward_time": 3.194744110107422,
      "step": 7615
    },
    {
      "dynamics_loss_avg": 0.06354445479810238,
      "epoch": 2.9048919677734376e-05,
      "step": 7615
    },
    {
      "action_loss_avg": 0.02241334831342101,
      "epoch": 2.9048919677734376e-05,
      "step": 7615
    },
    {
      "epoch": 2.9048919677734376e-05,
      "step": 7615,
      "training_step_time": 9.978475093841553
    },
    {
      "epoch": 2.9048919677734376e-05,
      "model_forward_time": 3.4067037105560303,
      "step": 7615
    },
    {
      "epoch": 2.9048919677734376e-05,
      "step": 7615,
      "training_step_time": 10.187089920043945
    },
    {
      "epoch": 2.9048919677734376e-05,
      "model_forward_time": 3.140395402908325,
      "step": 7615
    },
    {
      "epoch": 2.9048919677734376e-05,
      "step": 7615,
      "training_step_time": 9.921460628509521
    },
    {
      "epoch": 2.9048919677734376e-05,
      "model_forward_time": 3.140575647354126,
      "step": 7615
    },
    {
      "epoch": 2.9048919677734376e-05,
      "step": 7615,
      "training_step_time": 10.34945273399353
    },
    {
      "epoch": 2.9052734375e-05,
      "model_forward_time": 3.175577163696289,
      "step": 7616
    },
    {
      "epoch": 2.9052734375e-05,
      "step": 7616,
      "training_step_time": 9.95697546005249
    },
    {
      "epoch": 2.9052734375e-05,
      "model_forward_time": 3.145357370376587,
      "step": 7616
    },
    {
      "epoch": 2.9052734375e-05,
      "step": 7616,
      "training_step_time": 9.929812908172607
    },
    {
      "epoch": 2.9052734375e-05,
      "model_forward_time": 3.4432802200317383,
      "step": 7616
    },
    {
      "epoch": 2.9052734375e-05,
      "step": 7616,
      "training_step_time": 10.228458404541016
    },
    {
      "epoch": 2.9052734375e-05,
      "model_forward_time": 3.1416685581207275,
      "step": 7616
    },
    {
      "epoch": 2.9052734375e-05,
      "step": 7616,
      "training_step_time": 10.295143365859985
    },
    {
      "epoch": 2.9056549072265625e-05,
      "model_forward_time": 3.182656764984131,
      "step": 7617
    },
    {
      "epoch": 2.9056549072265625e-05,
      "step": 7617,
      "training_step_time": 9.966078281402588
    },
    {
      "epoch": 2.9056549072265625e-05,
      "model_forward_time": 3.1430819034576416,
      "step": 7617
    },
    {
      "epoch": 2.9056549072265625e-05,
      "step": 7617,
      "training_step_time": 9.930079936981201
    },
    {
      "epoch": 2.9056549072265625e-05,
      "model_forward_time": 3.135913372039795,
      "step": 7617
    },
    {
      "dynamics_loss_avg": 0.06655591949820519,
      "epoch": 2.9056549072265625e-05,
      "step": 7617
    },
    {
      "action_loss_avg": 0.02100518497172743,
      "epoch": 2.9056549072265625e-05,
      "step": 7617
    },
    {
      "epoch": 2.9056549072265625e-05,
      "step": 7617,
      "training_step_time": 9.918475151062012
    },
    {
      "epoch": 2.9056549072265625e-05,
      "model_forward_time": 3.3966221809387207,
      "step": 7617
    },
    {
      "epoch": 2.9056549072265625e-05,
      "step": 7617,
      "training_step_time": 10.270973205566406
    },
    {
      "epoch": 2.906036376953125e-05,
      "model_forward_time": 3.181570529937744,
      "step": 7618
    },
    {
      "epoch": 2.906036376953125e-05,
      "step": 7618,
      "training_step_time": 9.961348056793213
    },
    {
      "epoch": 2.906036376953125e-05,
      "model_forward_time": 3.138664484024048,
      "step": 7618
    },
    {
      "epoch": 2.906036376953125e-05,
      "step": 7618,
      "training_step_time": 9.920661211013794
    },
    {
      "epoch": 2.906036376953125e-05,
      "model_forward_time": 2.438908815383911,
      "step": 7618
    },
    {
      "epoch": 2.906036376953125e-05,
      "step": 7618,
      "training_step_time": 7.477373838424683
    },
    {
      "epoch": 2.906036376953125e-05,
      "model_forward_time": 3.1401216983795166,
      "step": 7618
    },
    {
      "epoch": 2.906036376953125e-05,
      "step": 7618,
      "training_step_time": 10.034410953521729
    },
    {
      "epoch": 2.9064178466796874e-05,
      "model_forward_time": 3.1713473796844482,
      "step": 7619
    },
    {
      "epoch": 2.9064178466796874e-05,
      "step": 7619,
      "training_step_time": 10.242884635925293
    },
    {
      "epoch": 2.9064178466796874e-05,
      "model_forward_time": 3.140617609024048,
      "step": 7619
    },
    {
      "epoch": 2.9064178466796874e-05,
      "step": 7619,
      "training_step_time": 9.926190376281738
    },
    {
      "epoch": 2.9064178466796874e-05,
      "model_forward_time": 3.138303756713867,
      "step": 7619
    },
    {
      "epoch": 2.9064178466796874e-05,
      "step": 7619,
      "training_step_time": 9.918625831604004
    },
    {
      "epoch": 2.9064178466796874e-05,
      "model_forward_time": 3.1431257724761963,
      "step": 7619
    },
    {
      "epoch": 2.9064178466796874e-05,
      "step": 7619,
      "training_step_time": 10.01966643333435
    },
    {
      "epoch": 2.90679931640625e-05,
      "grad_norm": 0.43178680539131165,
      "learning_rate": 1e-05,
      "loss": 0.0818,
      "step": 7620
    },
    {
      "epoch": 2.90679931640625e-05,
      "model_forward_time": 3.177443504333496,
      "step": 7620
    },
    {
      "dynamics_loss_avg": 0.06379964761435986,
      "epoch": 2.90679931640625e-05,
      "step": 7620
    },
    {
      "action_loss_avg": 0.025319104455411433,
      "epoch": 2.90679931640625e-05,
      "step": 7620
    },
    {
      "epoch": 2.90679931640625e-05,
      "step": 7620,
      "training_step_time": 9.963782548904419
    },
    {
      "epoch": 2.90679931640625e-05,
      "model_forward_time": 3.1376616954803467,
      "step": 7620
    },
    {
      "epoch": 2.90679931640625e-05,
      "step": 7620,
      "training_step_time": 10.203733444213867
    },
    {
      "epoch": 2.90679931640625e-05,
      "model_forward_time": 3.1383419036865234,
      "step": 7620
    },
    {
      "epoch": 2.90679931640625e-05,
      "step": 7620,
      "training_step_time": 9.922870397567749
    },
    {
      "epoch": 2.90679931640625e-05,
      "model_forward_time": 3.14198637008667,
      "step": 7620
    },
    {
      "epoch": 2.90679931640625e-05,
      "step": 7620,
      "training_step_time": 10.323775053024292
    },
    {
      "epoch": 2.9071807861328127e-05,
      "model_forward_time": 3.1888866424560547,
      "step": 7621
    },
    {
      "epoch": 2.9071807861328127e-05,
      "step": 7621,
      "training_step_time": 9.971954107284546
    },
    {
      "epoch": 2.9071807861328127e-05,
      "model_forward_time": 3.141160726547241,
      "step": 7621
    },
    {
      "epoch": 2.9071807861328127e-05,
      "step": 7621,
      "training_step_time": 9.924596071243286
    },
    {
      "epoch": 2.9071807861328127e-05,
      "model_forward_time": 3.4131388664245605,
      "step": 7621
    },
    {
      "epoch": 2.9071807861328127e-05,
      "step": 7621,
      "training_step_time": 10.198207139968872
    },
    {
      "epoch": 2.9071807861328127e-05,
      "model_forward_time": 3.1427578926086426,
      "step": 7621
    },
    {
      "epoch": 2.9071807861328127e-05,
      "step": 7621,
      "training_step_time": 10.275900602340698
    },
    {
      "epoch": 2.907562255859375e-05,
      "model_forward_time": 3.1877245903015137,
      "step": 7622
    },
    {
      "epoch": 2.907562255859375e-05,
      "step": 7622,
      "training_step_time": 9.967204570770264
    },
    {
      "epoch": 2.907562255859375e-05,
      "model_forward_time": 3.1398518085479736,
      "step": 7622
    },
    {
      "epoch": 2.907562255859375e-05,
      "step": 7622,
      "training_step_time": 9.92100214958191
    },
    {
      "epoch": 2.907562255859375e-05,
      "model_forward_time": 3.1410624980926514,
      "step": 7622
    },
    {
      "dynamics_loss_avg": 0.05661671389825642,
      "epoch": 2.907562255859375e-05,
      "step": 7622
    },
    {
      "action_loss_avg": 0.027306166931521148,
      "epoch": 2.907562255859375e-05,
      "step": 7622
    },
    {
      "epoch": 2.907562255859375e-05,
      "step": 7622,
      "training_step_time": 9.930337190628052
    },
    {
      "epoch": 2.907562255859375e-05,
      "model_forward_time": 3.1431093215942383,
      "step": 7622
    },
    {
      "epoch": 2.907562255859375e-05,
      "step": 7622,
      "training_step_time": 10.309022426605225
    },
    {
      "epoch": 2.9079437255859376e-05,
      "model_forward_time": 3.1895246505737305,
      "step": 7623
    },
    {
      "epoch": 2.9079437255859376e-05,
      "step": 7623,
      "training_step_time": 9.96996545791626
    },
    {
      "epoch": 2.9079437255859376e-05,
      "model_forward_time": 3.141406774520874,
      "step": 7623
    },
    {
      "epoch": 2.9079437255859376e-05,
      "step": 7623,
      "training_step_time": 9.919357061386108
    },
    {
      "epoch": 2.9079437255859376e-05,
      "model_forward_time": 3.139625310897827,
      "step": 7623
    },
    {
      "epoch": 2.9079437255859376e-05,
      "step": 7623,
      "training_step_time": 9.917907476425171
    },
    {
      "epoch": 2.9079437255859376e-05,
      "model_forward_time": 3.1365513801574707,
      "step": 7623
    },
    {
      "epoch": 2.9079437255859376e-05,
      "step": 7623,
      "training_step_time": 10.018799066543579
    },
    {
      "epoch": 2.9083251953125e-05,
      "model_forward_time": 3.4627697467803955,
      "step": 7624
    },
    {
      "epoch": 2.9083251953125e-05,
      "step": 7624,
      "training_step_time": 10.239680528640747
    },
    {
      "epoch": 2.9083251953125e-05,
      "model_forward_time": 3.138519525527954,
      "step": 7624
    },
    {
      "epoch": 2.9083251953125e-05,
      "step": 7624,
      "training_step_time": 9.918131351470947
    },
    {
      "epoch": 2.9083251953125e-05,
      "model_forward_time": 3.13504958152771,
      "step": 7624
    },
    {
      "epoch": 2.9083251953125e-05,
      "step": 7624,
      "training_step_time": 9.914376497268677
    },
    {
      "epoch": 2.9083251953125e-05,
      "model_forward_time": 3.1319456100463867,
      "step": 7624
    },
    {
      "epoch": 2.9083251953125e-05,
      "step": 7624,
      "training_step_time": 10.020098447799683
    },
    {
      "epoch": 2.9087066650390625e-05,
      "model_forward_time": 3.1888418197631836,
      "step": 7625
    },
    {
      "dynamics_loss_avg": 0.044789958745241165,
      "epoch": 2.9087066650390625e-05,
      "step": 7625
    },
    {
      "action_loss_avg": 0.014771256363019347,
      "epoch": 2.9087066650390625e-05,
      "step": 7625
    },
    {
      "epoch": 2.9087066650390625e-05,
      "step": 7625,
      "training_step_time": 9.967653751373291
    },
    {
      "epoch": 2.9087066650390625e-05,
      "model_forward_time": 3.1395630836486816,
      "step": 7625
    },
    {
      "epoch": 2.9087066650390625e-05,
      "step": 7625,
      "training_step_time": 10.189453601837158
    },
    {
      "epoch": 2.9087066650390625e-05,
      "model_forward_time": 3.1379950046539307,
      "step": 7625
    },
    {
      "epoch": 2.9087066650390625e-05,
      "step": 7625,
      "training_step_time": 9.908703804016113
    },
    {
      "epoch": 2.9087066650390625e-05,
      "model_forward_time": 3.132509469985962,
      "step": 7625
    },
    {
      "epoch": 2.9087066650390625e-05,
      "step": 7625,
      "training_step_time": 10.278159856796265
    },
    {
      "epoch": 2.909088134765625e-05,
      "model_forward_time": 3.1815731525421143,
      "step": 7626
    },
    {
      "epoch": 2.909088134765625e-05,
      "step": 7626,
      "training_step_time": 9.956623315811157
    },
    {
      "epoch": 2.909088134765625e-05,
      "model_forward_time": 3.1399290561676025,
      "step": 7626
    },
    {
      "epoch": 2.909088134765625e-05,
      "step": 7626,
      "training_step_time": 9.918638467788696
    },
    {
      "epoch": 2.909088134765625e-05,
      "model_forward_time": 3.4186437129974365,
      "step": 7626
    },
    {
      "epoch": 2.909088134765625e-05,
      "step": 7626,
      "training_step_time": 10.194709300994873
    },
    {
      "epoch": 2.909088134765625e-05,
      "model_forward_time": 3.1408910751342773,
      "step": 7626
    },
    {
      "epoch": 2.909088134765625e-05,
      "step": 7626,
      "training_step_time": 10.303654432296753
    },
    {
      "epoch": 2.9094696044921874e-05,
      "model_forward_time": 3.163020372390747,
      "step": 7627
    },
    {
      "epoch": 2.9094696044921874e-05,
      "step": 7627,
      "training_step_time": 9.949078798294067
    },
    {
      "epoch": 2.9094696044921874e-05,
      "model_forward_time": 3.1427500247955322,
      "step": 7627
    },
    {
      "epoch": 2.9094696044921874e-05,
      "step": 7627,
      "training_step_time": 9.922386646270752
    },
    {
      "epoch": 2.9094696044921874e-05,
      "model_forward_time": 3.1370556354522705,
      "step": 7627
    },
    {
      "dynamics_loss_avg": 0.0528806148096919,
      "epoch": 2.9094696044921874e-05,
      "step": 7627
    },
    {
      "action_loss_avg": 0.018834751611575485,
      "epoch": 2.9094696044921874e-05,
      "step": 7627
    },
    {
      "epoch": 2.9094696044921874e-05,
      "step": 7627,
      "training_step_time": 9.916382074356079
    },
    {
      "epoch": 2.9094696044921874e-05,
      "model_forward_time": 3.400827646255493,
      "step": 7627
    },
    {
      "epoch": 2.9094696044921874e-05,
      "step": 7627,
      "training_step_time": 10.301860570907593
    },
    {
      "epoch": 2.90985107421875e-05,
      "model_forward_time": 3.169877529144287,
      "step": 7628
    },
    {
      "epoch": 2.90985107421875e-05,
      "step": 7628,
      "training_step_time": 9.948890209197998
    },
    {
      "epoch": 2.90985107421875e-05,
      "model_forward_time": 3.1343095302581787,
      "step": 7628
    },
    {
      "epoch": 2.90985107421875e-05,
      "step": 7628,
      "training_step_time": 9.910993337631226
    },
    {
      "epoch": 2.90985107421875e-05,
      "model_forward_time": 3.1333515644073486,
      "step": 7628
    },
    {
      "epoch": 2.90985107421875e-05,
      "step": 7628,
      "training_step_time": 9.913810729980469
    },
    {
      "epoch": 2.90985107421875e-05,
      "model_forward_time": 3.139580726623535,
      "step": 7628
    },
    {
      "epoch": 2.90985107421875e-05,
      "step": 7628,
      "training_step_time": 10.026956558227539
    },
    {
      "epoch": 2.9102325439453123e-05,
      "model_forward_time": 3.4290173053741455,
      "step": 7629
    },
    {
      "epoch": 2.9102325439453123e-05,
      "step": 7629,
      "training_step_time": 10.209094285964966
    },
    {
      "epoch": 2.9102325439453123e-05,
      "model_forward_time": 3.146771192550659,
      "step": 7629
    },
    {
      "epoch": 2.9102325439453123e-05,
      "step": 7629,
      "training_step_time": 9.928411483764648
    },
    {
      "epoch": 2.9102325439453123e-05,
      "model_forward_time": 3.14223313331604,
      "step": 7629
    },
    {
      "epoch": 2.9102325439453123e-05,
      "step": 7629,
      "training_step_time": 9.926044464111328
    },
    {
      "epoch": 2.9102325439453123e-05,
      "model_forward_time": 3.1474878787994385,
      "step": 7629
    },
    {
      "epoch": 2.9102325439453123e-05,
      "step": 7629,
      "training_step_time": 10.02780270576477
    },
    {
      "epoch": 2.910614013671875e-05,
      "grad_norm": 0.6058488488197327,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 7630
    },
    {
      "epoch": 2.910614013671875e-05,
      "model_forward_time": 3.181591749191284,
      "step": 7630
    },
    {
      "dynamics_loss_avg": 0.048539400845766065,
      "epoch": 2.910614013671875e-05,
      "step": 7630
    },
    {
      "action_loss_avg": 0.0184434721712023,
      "epoch": 2.910614013671875e-05,
      "step": 7630
    },
    {
      "epoch": 2.910614013671875e-05,
      "step": 7630,
      "training_step_time": 9.968213558197021
    },
    {
      "epoch": 2.910614013671875e-05,
      "model_forward_time": 3.424440383911133,
      "step": 7630
    },
    {
      "epoch": 2.910614013671875e-05,
      "step": 7630,
      "training_step_time": 10.204379081726074
    },
    {
      "epoch": 2.910614013671875e-05,
      "model_forward_time": 3.1437320709228516,
      "step": 7630
    },
    {
      "epoch": 2.910614013671875e-05,
      "step": 7630,
      "training_step_time": 9.929016590118408
    },
    {
      "epoch": 2.910614013671875e-05,
      "model_forward_time": 3.1392295360565186,
      "step": 7630
    },
    {
      "epoch": 2.910614013671875e-05,
      "step": 7630,
      "training_step_time": 10.267486095428467
    },
    {
      "epoch": 2.9109954833984376e-05,
      "model_forward_time": 3.175839900970459,
      "step": 7631
    },
    {
      "epoch": 2.9109954833984376e-05,
      "step": 7631,
      "training_step_time": 9.957071304321289
    },
    {
      "epoch": 2.9109954833984376e-05,
      "model_forward_time": 3.1405820846557617,
      "step": 7631
    },
    {
      "epoch": 2.9109954833984376e-05,
      "step": 7631,
      "training_step_time": 9.92546558380127
    },
    {
      "epoch": 2.9109954833984376e-05,
      "model_forward_time": 3.1438167095184326,
      "step": 7631
    },
    {
      "epoch": 2.9109954833984376e-05,
      "step": 7631,
      "training_step_time": 10.22229552268982
    },
    {
      "epoch": 2.9109954833984376e-05,
      "model_forward_time": 3.145843505859375,
      "step": 7631
    },
    {
      "epoch": 2.9109954833984376e-05,
      "step": 7631,
      "training_step_time": 10.264916181564331
    },
    {
      "epoch": 2.911376953125e-05,
      "model_forward_time": 3.176661729812622,
      "step": 7632
    },
    {
      "epoch": 2.911376953125e-05,
      "step": 7632,
      "training_step_time": 9.958478450775146
    },
    {
      "epoch": 2.911376953125e-05,
      "model_forward_time": 3.1405584812164307,
      "step": 7632
    },
    {
      "epoch": 2.911376953125e-05,
      "step": 7632,
      "training_step_time": 9.919867515563965
    },
    {
      "epoch": 2.911376953125e-05,
      "model_forward_time": 3.143320322036743,
      "step": 7632
    },
    {
      "dynamics_loss_avg": 0.051408973336219785,
      "epoch": 2.911376953125e-05,
      "step": 7632
    },
    {
      "action_loss_avg": 0.017647594027221204,
      "epoch": 2.911376953125e-05,
      "step": 7632
    },
    {
      "epoch": 2.911376953125e-05,
      "step": 7632,
      "training_step_time": 9.924763679504395
    },
    {
      "epoch": 2.911376953125e-05,
      "model_forward_time": 3.4487557411193848,
      "step": 7632
    },
    {
      "epoch": 2.911376953125e-05,
      "step": 7632,
      "training_step_time": 10.318105936050415
    },
    {
      "epoch": 2.9117584228515625e-05,
      "model_forward_time": 3.177187919616699,
      "step": 7633
    },
    {
      "epoch": 2.9117584228515625e-05,
      "step": 7633,
      "training_step_time": 9.954060316085815
    },
    {
      "epoch": 2.9117584228515625e-05,
      "model_forward_time": 3.1392829418182373,
      "step": 7633
    },
    {
      "epoch": 2.9117584228515625e-05,
      "step": 7633,
      "training_step_time": 9.91585087776184
    },
    {
      "epoch": 2.9117584228515625e-05,
      "model_forward_time": 3.1384968757629395,
      "step": 7633
    },
    {
      "epoch": 2.9117584228515625e-05,
      "step": 7633,
      "training_step_time": 9.909404993057251
    },
    {
      "epoch": 2.9117584228515625e-05,
      "model_forward_time": 3.137782573699951,
      "step": 7633
    },
    {
      "epoch": 2.9117584228515625e-05,
      "step": 7633,
      "training_step_time": 10.299403429031372
    },
    {
      "epoch": 2.912139892578125e-05,
      "model_forward_time": 3.179002046585083,
      "step": 7634
    },
    {
      "epoch": 2.912139892578125e-05,
      "step": 7634,
      "training_step_time": 10.245440483093262
    },
    {
      "epoch": 2.912139892578125e-05,
      "model_forward_time": 3.137478828430176,
      "step": 7634
    },
    {
      "epoch": 2.912139892578125e-05,
      "step": 7634,
      "training_step_time": 9.913101196289062
    },
    {
      "epoch": 2.912139892578125e-05,
      "model_forward_time": 3.139578104019165,
      "step": 7634
    },
    {
      "epoch": 2.912139892578125e-05,
      "step": 7634,
      "training_step_time": 9.920573472976685
    },
    {
      "epoch": 2.912139892578125e-05,
      "model_forward_time": 3.141650915145874,
      "step": 7634
    },
    {
      "epoch": 2.912139892578125e-05,
      "step": 7634,
      "training_step_time": 10.027952909469604
    },
    {
      "epoch": 2.9125213623046874e-05,
      "model_forward_time": 3.173217535018921,
      "step": 7635
    },
    {
      "dynamics_loss_avg": 0.044674573838710784,
      "epoch": 2.9125213623046874e-05,
      "step": 7635
    },
    {
      "action_loss_avg": 0.01607448561117053,
      "epoch": 2.9125213623046874e-05,
      "step": 7635
    },
    {
      "epoch": 2.9125213623046874e-05,
      "step": 7635,
      "training_step_time": 9.961829900741577
    },
    {
      "epoch": 2.9125213623046874e-05,
      "model_forward_time": 3.4023091793060303,
      "step": 7635
    },
    {
      "epoch": 2.9125213623046874e-05,
      "step": 7635,
      "training_step_time": 10.187793970108032
    },
    {
      "epoch": 2.9125213623046874e-05,
      "model_forward_time": 3.149338483810425,
      "step": 7635
    },
    {
      "epoch": 2.9125213623046874e-05,
      "step": 7635,
      "training_step_time": 9.940031051635742
    },
    {
      "epoch": 2.9125213623046874e-05,
      "model_forward_time": 3.14277982711792,
      "step": 7635
    },
    {
      "epoch": 2.9125213623046874e-05,
      "step": 7635,
      "training_step_time": 10.33738374710083
    },
    {
      "epoch": 2.91290283203125e-05,
      "model_forward_time": 3.1846299171447754,
      "step": 7636
    },
    {
      "epoch": 2.91290283203125e-05,
      "step": 7636,
      "training_step_time": 9.973016500473022
    },
    {
      "epoch": 2.91290283203125e-05,
      "model_forward_time": 3.1402740478515625,
      "step": 7636
    },
    {
      "epoch": 2.91290283203125e-05,
      "step": 7636,
      "training_step_time": 9.924862146377563
    },
    {
      "epoch": 2.91290283203125e-05,
      "model_forward_time": 3.4190938472747803,
      "step": 7636
    },
    {
      "epoch": 2.91290283203125e-05,
      "step": 7636,
      "training_step_time": 10.196309328079224
    },
    {
      "epoch": 2.91290283203125e-05,
      "model_forward_time": 3.1355276107788086,
      "step": 7636
    },
    {
      "epoch": 2.91290283203125e-05,
      "step": 7636,
      "training_step_time": 10.266123056411743
    },
    {
      "epoch": 2.9132843017578123e-05,
      "model_forward_time": 3.1543190479278564,
      "step": 7637
    },
    {
      "epoch": 2.9132843017578123e-05,
      "step": 7637,
      "training_step_time": 9.927736520767212
    },
    {
      "epoch": 2.9132843017578123e-05,
      "model_forward_time": 3.1329329013824463,
      "step": 7637
    },
    {
      "epoch": 2.9132843017578123e-05,
      "step": 7637,
      "training_step_time": 9.904439449310303
    },
    {
      "epoch": 2.9132843017578123e-05,
      "model_forward_time": 3.127474784851074,
      "step": 7637
    },
    {
      "dynamics_loss_avg": 0.053588411025702956,
      "epoch": 2.9132843017578123e-05,
      "step": 7637
    },
    {
      "action_loss_avg": 0.020260009355843066,
      "epoch": 2.9132843017578123e-05,
      "step": 7637
    },
    {
      "epoch": 2.9132843017578123e-05,
      "step": 7637,
      "training_step_time": 9.900472402572632
    },
    {
      "epoch": 2.9132843017578123e-05,
      "model_forward_time": 3.3881442546844482,
      "step": 7637
    },
    {
      "epoch": 2.9132843017578123e-05,
      "step": 7637,
      "training_step_time": 10.303514957427979
    },
    {
      "epoch": 2.913665771484375e-05,
      "model_forward_time": 3.1780192852020264,
      "step": 7638
    },
    {
      "epoch": 2.913665771484375e-05,
      "step": 7638,
      "training_step_time": 9.952564716339111
    },
    {
      "epoch": 2.913665771484375e-05,
      "model_forward_time": 3.1332004070281982,
      "step": 7638
    },
    {
      "epoch": 2.913665771484375e-05,
      "step": 7638,
      "training_step_time": 9.90889286994934
    },
    {
      "epoch": 2.913665771484375e-05,
      "model_forward_time": 3.1317615509033203,
      "step": 7638
    },
    {
      "epoch": 2.913665771484375e-05,
      "step": 7638,
      "training_step_time": 9.906983613967896
    },
    {
      "epoch": 2.913665771484375e-05,
      "model_forward_time": 3.1360421180725098,
      "step": 7638
    },
    {
      "epoch": 2.913665771484375e-05,
      "step": 7638,
      "training_step_time": 10.275531768798828
    },
    {
      "epoch": 2.9140472412109376e-05,
      "model_forward_time": 3.416672468185425,
      "step": 7639
    },
    {
      "epoch": 2.9140472412109376e-05,
      "step": 7639,
      "training_step_time": 10.19267463684082
    },
    {
      "epoch": 2.9140472412109376e-05,
      "model_forward_time": 3.136810779571533,
      "step": 7639
    },
    {
      "epoch": 2.9140472412109376e-05,
      "step": 7639,
      "training_step_time": 9.913317441940308
    },
    {
      "epoch": 2.9140472412109376e-05,
      "model_forward_time": 3.1373252868652344,
      "step": 7639
    },
    {
      "epoch": 2.9140472412109376e-05,
      "step": 7639,
      "training_step_time": 9.918742418289185
    },
    {
      "epoch": 2.9140472412109376e-05,
      "model_forward_time": 3.139920234680176,
      "step": 7639
    },
    {
      "epoch": 2.9140472412109376e-05,
      "step": 7639,
      "training_step_time": 10.032824516296387
    },
    {
      "epoch": 2.9144287109375e-05,
      "grad_norm": 0.34735068678855896,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 7640
    },
    {
      "epoch": 2.9144287109375e-05,
      "model_forward_time": 3.1544253826141357,
      "step": 7640
    },
    {
      "dynamics_loss_avg": 0.061317659169435504,
      "epoch": 2.9144287109375e-05,
      "step": 7640
    },
    {
      "action_loss_avg": 0.020219917129725217,
      "epoch": 2.9144287109375e-05,
      "step": 7640
    },
    {
      "epoch": 2.9144287109375e-05,
      "step": 7640,
      "training_step_time": 9.941272735595703
    },
    {
      "epoch": 2.9144287109375e-05,
      "model_forward_time": 3.135680913925171,
      "step": 7640
    },
    {
      "epoch": 2.9144287109375e-05,
      "step": 7640,
      "training_step_time": 10.213829278945923
    },
    {
      "epoch": 2.9144287109375e-05,
      "model_forward_time": 3.135286331176758,
      "step": 7640
    },
    {
      "epoch": 2.9144287109375e-05,
      "step": 7640,
      "training_step_time": 9.92154312133789
    },
    {
      "epoch": 2.9144287109375e-05,
      "model_forward_time": 3.1469783782958984,
      "step": 7640
    },
    {
      "epoch": 2.9144287109375e-05,
      "step": 7640,
      "training_step_time": 10.301798582077026
    },
    {
      "epoch": 2.9148101806640625e-05,
      "model_forward_time": 3.1479926109313965,
      "step": 7641
    },
    {
      "epoch": 2.9148101806640625e-05,
      "step": 7641,
      "training_step_time": 9.93155813217163
    },
    {
      "epoch": 2.9148101806640625e-05,
      "model_forward_time": 3.138871669769287,
      "step": 7641
    },
    {
      "epoch": 2.9148101806640625e-05,
      "step": 7641,
      "training_step_time": 9.917456865310669
    },
    {
      "epoch": 2.9148101806640625e-05,
      "model_forward_time": 3.3935391902923584,
      "step": 7641
    },
    {
      "epoch": 2.9148101806640625e-05,
      "step": 7641,
      "training_step_time": 10.172916173934937
    },
    {
      "epoch": 2.9148101806640625e-05,
      "model_forward_time": 3.140716314315796,
      "step": 7641
    },
    {
      "epoch": 2.9148101806640625e-05,
      "step": 7641,
      "training_step_time": 10.021024942398071
    },
    {
      "epoch": 2.915191650390625e-05,
      "model_forward_time": 3.150719404220581,
      "step": 7642
    },
    {
      "epoch": 2.915191650390625e-05,
      "step": 7642,
      "training_step_time": 9.92876410484314
    },
    {
      "epoch": 2.915191650390625e-05,
      "model_forward_time": 3.134377956390381,
      "step": 7642
    },
    {
      "epoch": 2.915191650390625e-05,
      "step": 7642,
      "training_step_time": 9.912655591964722
    },
    {
      "epoch": 2.915191650390625e-05,
      "model_forward_time": 3.1405398845672607,
      "step": 7642
    },
    {
      "dynamics_loss_avg": 0.05515118949115276,
      "epoch": 2.915191650390625e-05,
      "step": 7642
    },
    {
      "action_loss_avg": 0.01923709730617702,
      "epoch": 2.915191650390625e-05,
      "step": 7642
    },
    {
      "epoch": 2.915191650390625e-05,
      "step": 7642,
      "training_step_time": 9.926466464996338
    },
    {
      "epoch": 2.915191650390625e-05,
      "model_forward_time": 3.1369311809539795,
      "step": 7642
    },
    {
      "epoch": 2.915191650390625e-05,
      "step": 7642,
      "training_step_time": 10.301233291625977
    },
    {
      "epoch": 2.9155731201171874e-05,
      "model_forward_time": 3.1523048877716064,
      "step": 7643
    },
    {
      "epoch": 2.9155731201171874e-05,
      "step": 7643,
      "training_step_time": 9.93444013595581
    },
    {
      "epoch": 2.9155731201171874e-05,
      "model_forward_time": 3.1360561847686768,
      "step": 7643
    },
    {
      "epoch": 2.9155731201171874e-05,
      "step": 7643,
      "training_step_time": 9.912276983261108
    },
    {
      "epoch": 2.9155731201171874e-05,
      "model_forward_time": 3.134773015975952,
      "step": 7643
    },
    {
      "epoch": 2.9155731201171874e-05,
      "step": 7643,
      "training_step_time": 9.91049337387085
    },
    {
      "epoch": 2.9155731201171874e-05,
      "model_forward_time": 3.133784532546997,
      "step": 7643
    },
    {
      "epoch": 2.9155731201171874e-05,
      "step": 7643,
      "training_step_time": 10.292688846588135
    },
    {
      "epoch": 2.91595458984375e-05,
      "model_forward_time": 3.435054302215576,
      "step": 7644
    },
    {
      "epoch": 2.91595458984375e-05,
      "step": 7644,
      "training_step_time": 10.2115159034729
    },
    {
      "epoch": 2.91595458984375e-05,
      "model_forward_time": 3.1318609714508057,
      "step": 7644
    },
    {
      "epoch": 2.91595458984375e-05,
      "step": 7644,
      "training_step_time": 9.90997052192688
    },
    {
      "epoch": 2.91595458984375e-05,
      "model_forward_time": 3.137319564819336,
      "step": 7644
    },
    {
      "epoch": 2.91595458984375e-05,
      "step": 7644,
      "training_step_time": 9.921231985092163
    },
    {
      "epoch": 2.91595458984375e-05,
      "model_forward_time": 3.1300485134124756,
      "step": 7644
    },
    {
      "epoch": 2.91595458984375e-05,
      "step": 7644,
      "training_step_time": 10.02606725692749
    },
    {
      "epoch": 2.9163360595703123e-05,
      "model_forward_time": 3.1453330516815186,
      "step": 7645
    },
    {
      "dynamics_loss_avg": 0.05464279446750879,
      "epoch": 2.9163360595703123e-05,
      "step": 7645
    },
    {
      "action_loss_avg": 0.017969893850386144,
      "epoch": 2.9163360595703123e-05,
      "step": 7645
    },
    {
      "epoch": 2.9163360595703123e-05,
      "step": 7645,
      "training_step_time": 9.924078226089478
    },
    {
      "epoch": 2.9163360595703123e-05,
      "model_forward_time": 3.135890483856201,
      "step": 7645
    },
    {
      "epoch": 2.9163360595703123e-05,
      "step": 7645,
      "training_step_time": 10.200393915176392
    },
    {
      "epoch": 2.9163360595703123e-05,
      "model_forward_time": 3.1327645778656006,
      "step": 7645
    },
    {
      "epoch": 2.9163360595703123e-05,
      "step": 7645,
      "training_step_time": 9.90688681602478
    },
    {
      "epoch": 2.9163360595703123e-05,
      "model_forward_time": 3.1294853687286377,
      "step": 7645
    },
    {
      "epoch": 2.9163360595703123e-05,
      "step": 7645,
      "training_step_time": 10.329346656799316
    },
    {
      "epoch": 2.916717529296875e-05,
      "model_forward_time": 3.1367712020874023,
      "step": 7646
    },
    {
      "epoch": 2.916717529296875e-05,
      "step": 7646,
      "training_step_time": 9.913869380950928
    },
    {
      "epoch": 2.916717529296875e-05,
      "model_forward_time": 3.1303794384002686,
      "step": 7646
    },
    {
      "epoch": 2.916717529296875e-05,
      "step": 7646,
      "training_step_time": 9.904312372207642
    },
    {
      "epoch": 2.916717529296875e-05,
      "model_forward_time": 3.43930983543396,
      "step": 7646
    },
    {
      "epoch": 2.916717529296875e-05,
      "step": 7646,
      "training_step_time": 10.217683792114258
    },
    {
      "epoch": 2.916717529296875e-05,
      "model_forward_time": 3.132704973220825,
      "step": 7646
    },
    {
      "epoch": 2.916717529296875e-05,
      "step": 7646,
      "training_step_time": 10.305565595626831
    },
    {
      "epoch": 2.9170989990234376e-05,
      "model_forward_time": 3.1527185440063477,
      "step": 7647
    },
    {
      "epoch": 2.9170989990234376e-05,
      "step": 7647,
      "training_step_time": 9.934142112731934
    },
    {
      "epoch": 2.9170989990234376e-05,
      "model_forward_time": 3.1415159702301025,
      "step": 7647
    },
    {
      "epoch": 2.9170989990234376e-05,
      "step": 7647,
      "training_step_time": 9.921586513519287
    },
    {
      "epoch": 2.9170989990234376e-05,
      "model_forward_time": 3.1395277976989746,
      "step": 7647
    },
    {
      "dynamics_loss_avg": 0.05218219216912985,
      "epoch": 2.9170989990234376e-05,
      "step": 7647
    },
    {
      "action_loss_avg": 0.019590160995721816,
      "epoch": 2.9170989990234376e-05,
      "step": 7647
    },
    {
      "epoch": 2.9170989990234376e-05,
      "step": 7647,
      "training_step_time": 9.924500942230225
    },
    {
      "epoch": 2.9170989990234376e-05,
      "model_forward_time": 3.4051425457000732,
      "step": 7647
    },
    {
      "epoch": 2.9170989990234376e-05,
      "step": 7647,
      "training_step_time": 10.31486201286316
    },
    {
      "epoch": 2.91748046875e-05,
      "model_forward_time": 3.154355049133301,
      "step": 7648
    },
    {
      "epoch": 2.91748046875e-05,
      "step": 7648,
      "training_step_time": 9.94152545928955
    },
    {
      "epoch": 2.91748046875e-05,
      "model_forward_time": 2.439530372619629,
      "step": 7648
    },
    {
      "epoch": 2.91748046875e-05,
      "step": 7648,
      "training_step_time": 7.476436614990234
    },
    {
      "epoch": 2.91748046875e-05,
      "model_forward_time": 3.1345512866973877,
      "step": 7648
    },
    {
      "epoch": 2.91748046875e-05,
      "step": 7648,
      "training_step_time": 9.912162065505981
    },
    {
      "epoch": 2.91748046875e-05,
      "model_forward_time": 3.1369287967681885,
      "step": 7648
    },
    {
      "epoch": 2.91748046875e-05,
      "step": 7648,
      "training_step_time": 10.232551097869873
    },
    {
      "epoch": 2.9178619384765625e-05,
      "model_forward_time": 3.4124248027801514,
      "step": 7649
    },
    {
      "epoch": 2.9178619384765625e-05,
      "step": 7649,
      "training_step_time": 10.19098711013794
    },
    {
      "epoch": 2.9178619384765625e-05,
      "model_forward_time": 3.1352651119232178,
      "step": 7649
    },
    {
      "epoch": 2.9178619384765625e-05,
      "step": 7649,
      "training_step_time": 9.916428089141846
    },
    {
      "epoch": 2.9178619384765625e-05,
      "model_forward_time": 3.1323201656341553,
      "step": 7649
    },
    {
      "epoch": 2.9178619384765625e-05,
      "step": 7649,
      "training_step_time": 9.906813144683838
    },
    {
      "epoch": 2.9178619384765625e-05,
      "model_forward_time": 3.133021593093872,
      "step": 7649
    },
    {
      "epoch": 2.9178619384765625e-05,
      "step": 7649,
      "training_step_time": 10.31234884262085
    },
    {
      "epoch": 2.918243408203125e-05,
      "grad_norm": 0.44032981991767883,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 7650
    },
    {
      "epoch": 2.918243408203125e-05,
      "model_forward_time": 3.141040563583374,
      "step": 7650
    },
    {
      "dynamics_loss_avg": 0.03743220204487443,
      "epoch": 2.918243408203125e-05,
      "step": 7650
    },
    {
      "action_loss_avg": 0.014672987698577344,
      "epoch": 2.918243408203125e-05,
      "step": 7650
    },
    {
      "epoch": 2.918243408203125e-05,
      "step": 7650,
      "training_step_time": 9.918211698532104
    },
    {
      "epoch": 2.918243408203125e-05,
      "model_forward_time": 3.1298389434814453,
      "step": 7650
    },
    {
      "epoch": 2.918243408203125e-05,
      "step": 7650,
      "training_step_time": 10.171552658081055
    },
    {
      "epoch": 2.918243408203125e-05,
      "model_forward_time": 3.127577781677246,
      "step": 7650
    },
    {
      "epoch": 2.918243408203125e-05,
      "step": 7650,
      "training_step_time": 9.900768518447876
    },
    {
      "epoch": 2.918243408203125e-05,
      "model_forward_time": 3.128462314605713,
      "step": 7650
    },
    {
      "epoch": 2.918243408203125e-05,
      "step": 7650,
      "training_step_time": 10.016946077346802
    },
    {
      "epoch": 2.9186248779296874e-05,
      "model_forward_time": 3.144190788269043,
      "step": 7651
    },
    {
      "epoch": 2.9186248779296874e-05,
      "step": 7651,
      "training_step_time": 9.92030668258667
    },
    {
      "epoch": 2.9186248779296874e-05,
      "model_forward_time": 3.128720283508301,
      "step": 7651
    },
    {
      "epoch": 2.9186248779296874e-05,
      "step": 7651,
      "training_step_time": 9.902296304702759
    },
    {
      "epoch": 2.9186248779296874e-05,
      "model_forward_time": 3.1360154151916504,
      "step": 7651
    },
    {
      "epoch": 2.9186248779296874e-05,
      "step": 7651,
      "training_step_time": 10.195712566375732
    },
    {
      "epoch": 2.9186248779296874e-05,
      "model_forward_time": 3.1366865634918213,
      "step": 7651
    },
    {
      "epoch": 2.9186248779296874e-05,
      "step": 7651,
      "training_step_time": 10.314706802368164
    },
    {
      "epoch": 2.91900634765625e-05,
      "model_forward_time": 3.1429622173309326,
      "step": 7652
    },
    {
      "epoch": 2.91900634765625e-05,
      "step": 7652,
      "training_step_time": 9.924177169799805
    },
    {
      "epoch": 2.91900634765625e-05,
      "model_forward_time": 3.132192373275757,
      "step": 7652
    },
    {
      "epoch": 2.91900634765625e-05,
      "step": 7652,
      "training_step_time": 9.910775184631348
    },
    {
      "epoch": 2.91900634765625e-05,
      "model_forward_time": 3.134800434112549,
      "step": 7652
    },
    {
      "dynamics_loss_avg": 0.05821835305541754,
      "epoch": 2.91900634765625e-05,
      "step": 7652
    },
    {
      "action_loss_avg": 0.023851606249809264,
      "epoch": 2.91900634765625e-05,
      "step": 7652
    },
    {
      "epoch": 2.91900634765625e-05,
      "step": 7652,
      "training_step_time": 9.910926580429077
    },
    {
      "epoch": 2.91900634765625e-05,
      "model_forward_time": 3.416086196899414,
      "step": 7652
    },
    {
      "epoch": 2.91900634765625e-05,
      "step": 7652,
      "training_step_time": 10.30786657333374
    },
    {
      "epoch": 2.9193878173828123e-05,
      "model_forward_time": 3.1490440368652344,
      "step": 7653
    },
    {
      "epoch": 2.9193878173828123e-05,
      "step": 7653,
      "training_step_time": 9.926022052764893
    },
    {
      "epoch": 2.9193878173828123e-05,
      "model_forward_time": 3.1320009231567383,
      "step": 7653
    },
    {
      "epoch": 2.9193878173828123e-05,
      "step": 7653,
      "training_step_time": 9.910181999206543
    },
    {
      "epoch": 2.9193878173828123e-05,
      "model_forward_time": 3.139357805252075,
      "step": 7653
    },
    {
      "epoch": 2.9193878173828123e-05,
      "step": 7653,
      "training_step_time": 9.92218828201294
    },
    {
      "epoch": 2.9193878173828123e-05,
      "model_forward_time": 3.1427884101867676,
      "step": 7653
    },
    {
      "epoch": 2.9193878173828123e-05,
      "step": 7653,
      "training_step_time": 10.044061660766602
    },
    {
      "epoch": 2.919769287109375e-05,
      "model_forward_time": 3.1431331634521484,
      "step": 7654
    },
    {
      "epoch": 2.919769287109375e-05,
      "step": 7654,
      "training_step_time": 10.204330205917358
    },
    {
      "epoch": 2.919769287109375e-05,
      "model_forward_time": 3.1357526779174805,
      "step": 7654
    },
    {
      "epoch": 2.919769287109375e-05,
      "step": 7654,
      "training_step_time": 9.917906761169434
    },
    {
      "epoch": 2.919769287109375e-05,
      "model_forward_time": 3.1358416080474854,
      "step": 7654
    },
    {
      "epoch": 2.919769287109375e-05,
      "step": 7654,
      "training_step_time": 9.91586971282959
    },
    {
      "epoch": 2.919769287109375e-05,
      "model_forward_time": 3.132615804672241,
      "step": 7654
    },
    {
      "epoch": 2.919769287109375e-05,
      "step": 7654,
      "training_step_time": 10.290764331817627
    },
    {
      "epoch": 2.9201507568359376e-05,
      "model_forward_time": 3.1588590145111084,
      "step": 7655
    },
    {
      "dynamics_loss_avg": 0.049619277752935884,
      "epoch": 2.9201507568359376e-05,
      "step": 7655
    },
    {
      "action_loss_avg": 0.015960394218564034,
      "epoch": 2.9201507568359376e-05,
      "step": 7655
    },
    {
      "epoch": 2.9201507568359376e-05,
      "step": 7655,
      "training_step_time": 9.947833061218262
    },
    {
      "epoch": 2.9201507568359376e-05,
      "model_forward_time": 3.4003798961639404,
      "step": 7655
    },
    {
      "epoch": 2.9201507568359376e-05,
      "step": 7655,
      "training_step_time": 10.181429862976074
    },
    {
      "epoch": 2.9201507568359376e-05,
      "model_forward_time": 3.1455788612365723,
      "step": 7655
    },
    {
      "epoch": 2.9201507568359376e-05,
      "step": 7655,
      "training_step_time": 9.925576210021973
    },
    {
      "epoch": 2.9201507568359376e-05,
      "model_forward_time": 3.141512155532837,
      "step": 7655
    },
    {
      "epoch": 2.9201507568359376e-05,
      "step": 7655,
      "training_step_time": 10.032420873641968
    },
    {
      "epoch": 2.9205322265625e-05,
      "model_forward_time": 3.153184652328491,
      "step": 7656
    },
    {
      "epoch": 2.9205322265625e-05,
      "step": 7656,
      "training_step_time": 9.93294620513916
    },
    {
      "epoch": 2.9205322265625e-05,
      "model_forward_time": 3.1454672813415527,
      "step": 7656
    },
    {
      "epoch": 2.9205322265625e-05,
      "step": 7656,
      "training_step_time": 9.92752194404602
    },
    {
      "epoch": 2.9205322265625e-05,
      "model_forward_time": 3.436203718185425,
      "step": 7656
    },
    {
      "epoch": 2.9205322265625e-05,
      "step": 7656,
      "training_step_time": 10.220038652420044
    },
    {
      "epoch": 2.9205322265625e-05,
      "model_forward_time": 3.1413958072662354,
      "step": 7656
    },
    {
      "epoch": 2.9205322265625e-05,
      "step": 7656,
      "training_step_time": 10.307948112487793
    },
    {
      "epoch": 2.9209136962890625e-05,
      "model_forward_time": 3.148434638977051,
      "step": 7657
    },
    {
      "epoch": 2.9209136962890625e-05,
      "step": 7657,
      "training_step_time": 9.931355237960815
    },
    {
      "epoch": 2.9209136962890625e-05,
      "model_forward_time": 3.1407201290130615,
      "step": 7657
    },
    {
      "epoch": 2.9209136962890625e-05,
      "step": 7657,
      "training_step_time": 9.923117399215698
    },
    {
      "epoch": 2.9209136962890625e-05,
      "model_forward_time": 3.1455912590026855,
      "step": 7657
    },
    {
      "dynamics_loss_avg": 0.05703217294067144,
      "epoch": 2.9209136962890625e-05,
      "step": 7657
    },
    {
      "action_loss_avg": 0.02599390661343932,
      "epoch": 2.9209136962890625e-05,
      "step": 7657
    },
    {
      "epoch": 2.9209136962890625e-05,
      "step": 7657,
      "training_step_time": 9.934002876281738
    },
    {
      "epoch": 2.9209136962890625e-05,
      "model_forward_time": 3.4091711044311523,
      "step": 7657
    },
    {
      "epoch": 2.9209136962890625e-05,
      "step": 7657,
      "training_step_time": 10.312705516815186
    },
    {
      "epoch": 2.921295166015625e-05,
      "model_forward_time": 3.153061866760254,
      "step": 7658
    },
    {
      "epoch": 2.921295166015625e-05,
      "step": 7658,
      "training_step_time": 9.937560319900513
    },
    {
      "epoch": 2.921295166015625e-05,
      "model_forward_time": 3.1498236656188965,
      "step": 7658
    },
    {
      "epoch": 2.921295166015625e-05,
      "step": 7658,
      "training_step_time": 9.930936336517334
    },
    {
      "epoch": 2.921295166015625e-05,
      "model_forward_time": 3.146524429321289,
      "step": 7658
    },
    {
      "epoch": 2.921295166015625e-05,
      "step": 7658,
      "training_step_time": 9.931196689605713
    },
    {
      "epoch": 2.921295166015625e-05,
      "model_forward_time": 3.149320363998413,
      "step": 7658
    },
    {
      "epoch": 2.921295166015625e-05,
      "step": 7658,
      "training_step_time": 10.26217246055603
    },
    {
      "epoch": 2.9216766357421874e-05,
      "model_forward_time": 3.444262981414795,
      "step": 7659
    },
    {
      "epoch": 2.9216766357421874e-05,
      "step": 7659,
      "training_step_time": 10.2267484664917
    },
    {
      "epoch": 2.9216766357421874e-05,
      "model_forward_time": 3.14803409576416,
      "step": 7659
    },
    {
      "epoch": 2.9216766357421874e-05,
      "step": 7659,
      "training_step_time": 9.935481548309326
    },
    {
      "epoch": 2.9216766357421874e-05,
      "model_forward_time": 3.1463053226470947,
      "step": 7659
    },
    {
      "epoch": 2.9216766357421874e-05,
      "step": 7659,
      "training_step_time": 9.930822610855103
    },
    {
      "epoch": 2.9216766357421874e-05,
      "model_forward_time": 3.150728225708008,
      "step": 7659
    },
    {
      "epoch": 2.9216766357421874e-05,
      "step": 7659,
      "training_step_time": 10.240227222442627
    },
    {
      "epoch": 2.92205810546875e-05,
      "grad_norm": 0.3715699017047882,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 7660
    },
    {
      "epoch": 2.92205810546875e-05,
      "model_forward_time": 3.157559394836426,
      "step": 7660
    },
    {
      "dynamics_loss_avg": 0.05705544762313366,
      "epoch": 2.92205810546875e-05,
      "step": 7660
    },
    {
      "action_loss_avg": 0.022606785548850893,
      "epoch": 2.92205810546875e-05,
      "step": 7660
    },
    {
      "epoch": 2.92205810546875e-05,
      "step": 7660,
      "training_step_time": 9.944588899612427
    },
    {
      "epoch": 2.92205810546875e-05,
      "model_forward_time": 3.1432814598083496,
      "step": 7660
    },
    {
      "epoch": 2.92205810546875e-05,
      "step": 7660,
      "training_step_time": 10.236558675765991
    },
    {
      "epoch": 2.92205810546875e-05,
      "model_forward_time": 3.145781993865967,
      "step": 7660
    },
    {
      "epoch": 2.92205810546875e-05,
      "step": 7660,
      "training_step_time": 9.934539318084717
    },
    {
      "epoch": 2.92205810546875e-05,
      "model_forward_time": 3.1557843685150146,
      "step": 7660
    },
    {
      "epoch": 2.92205810546875e-05,
      "step": 7660,
      "training_step_time": 10.30512523651123
    },
    {
      "epoch": 2.9224395751953123e-05,
      "model_forward_time": 3.1695797443389893,
      "step": 7661
    },
    {
      "epoch": 2.9224395751953123e-05,
      "step": 7661,
      "training_step_time": 9.95188856124878
    },
    {
      "epoch": 2.9224395751953123e-05,
      "model_forward_time": 3.1430513858795166,
      "step": 7661
    },
    {
      "epoch": 2.9224395751953123e-05,
      "step": 7661,
      "training_step_time": 9.92724871635437
    },
    {
      "epoch": 2.9224395751953123e-05,
      "model_forward_time": 3.1554183959960938,
      "step": 7661
    },
    {
      "epoch": 2.9224395751953123e-05,
      "step": 7661,
      "training_step_time": 10.233099937438965
    },
    {
      "epoch": 2.9224395751953123e-05,
      "model_forward_time": 3.1526308059692383,
      "step": 7661
    },
    {
      "epoch": 2.9224395751953123e-05,
      "step": 7661,
      "training_step_time": 10.034783124923706
    },
    {
      "epoch": 2.922821044921875e-05,
      "model_forward_time": 3.157381534576416,
      "step": 7662
    },
    {
      "epoch": 2.922821044921875e-05,
      "step": 7662,
      "training_step_time": 9.95168137550354
    },
    {
      "epoch": 2.922821044921875e-05,
      "model_forward_time": 3.1472575664520264,
      "step": 7662
    },
    {
      "epoch": 2.922821044921875e-05,
      "step": 7662,
      "training_step_time": 9.932263135910034
    },
    {
      "epoch": 2.922821044921875e-05,
      "model_forward_time": 3.150648355484009,
      "step": 7662
    },
    {
      "dynamics_loss_avg": 0.060693977214396,
      "epoch": 2.922821044921875e-05,
      "step": 7662
    },
    {
      "action_loss_avg": 0.026055717095732688,
      "epoch": 2.922821044921875e-05,
      "step": 7662
    },
    {
      "epoch": 2.922821044921875e-05,
      "step": 7662,
      "training_step_time": 9.933220863342285
    },
    {
      "epoch": 2.922821044921875e-05,
      "model_forward_time": 3.1420443058013916,
      "step": 7662
    },
    {
      "epoch": 2.922821044921875e-05,
      "step": 7662,
      "training_step_time": 10.299286127090454
    },
    {
      "epoch": 2.9232025146484376e-05,
      "model_forward_time": 3.1580803394317627,
      "step": 7663
    },
    {
      "epoch": 2.9232025146484376e-05,
      "step": 7663,
      "training_step_time": 9.94157886505127
    },
    {
      "epoch": 2.9232025146484376e-05,
      "model_forward_time": 3.138803243637085,
      "step": 7663
    },
    {
      "epoch": 2.9232025146484376e-05,
      "step": 7663,
      "training_step_time": 9.919196844100952
    },
    {
      "epoch": 2.9232025146484376e-05,
      "model_forward_time": 3.1423630714416504,
      "step": 7663
    },
    {
      "epoch": 2.9232025146484376e-05,
      "step": 7663,
      "training_step_time": 9.926920413970947
    },
    {
      "epoch": 2.9232025146484376e-05,
      "model_forward_time": 3.135254144668579,
      "step": 7663
    },
    {
      "epoch": 2.9232025146484376e-05,
      "step": 7663,
      "training_step_time": 10.273085355758667
    },
    {
      "epoch": 2.923583984375e-05,
      "model_forward_time": 3.4329137802124023,
      "step": 7664
    },
    {
      "epoch": 2.923583984375e-05,
      "step": 7664,
      "training_step_time": 10.210381746292114
    },
    {
      "epoch": 2.923583984375e-05,
      "model_forward_time": 3.130789279937744,
      "step": 7664
    },
    {
      "epoch": 2.923583984375e-05,
      "step": 7664,
      "training_step_time": 9.908118486404419
    },
    {
      "epoch": 2.923583984375e-05,
      "model_forward_time": 3.134965181350708,
      "step": 7664
    },
    {
      "epoch": 2.923583984375e-05,
      "step": 7664,
      "training_step_time": 9.911917209625244
    },
    {
      "epoch": 2.923583984375e-05,
      "model_forward_time": 3.1340794563293457,
      "step": 7664
    },
    {
      "epoch": 2.923583984375e-05,
      "step": 7664,
      "training_step_time": 10.3273344039917
    },
    {
      "epoch": 2.9239654541015625e-05,
      "model_forward_time": 3.1598527431488037,
      "step": 7665
    },
    {
      "dynamics_loss_avg": 0.04962080381810665,
      "epoch": 2.9239654541015625e-05,
      "step": 7665
    },
    {
      "action_loss_avg": 0.020041984925046562,
      "epoch": 2.9239654541015625e-05,
      "step": 7665
    },
    {
      "epoch": 2.9239654541015625e-05,
      "step": 7665,
      "training_step_time": 9.946434497833252
    },
    {
      "epoch": 2.9239654541015625e-05,
      "model_forward_time": 3.1447489261627197,
      "step": 7665
    },
    {
      "epoch": 2.9239654541015625e-05,
      "step": 7665,
      "training_step_time": 10.20640778541565
    },
    {
      "epoch": 2.9239654541015625e-05,
      "model_forward_time": 3.1455235481262207,
      "step": 7665
    },
    {
      "epoch": 2.9239654541015625e-05,
      "step": 7665,
      "training_step_time": 9.928877353668213
    },
    {
      "epoch": 2.9239654541015625e-05,
      "model_forward_time": 3.1441915035247803,
      "step": 7665
    },
    {
      "epoch": 2.9239654541015625e-05,
      "step": 7665,
      "training_step_time": 10.300355911254883
    },
    {
      "epoch": 2.924346923828125e-05,
      "model_forward_time": 3.1506667137145996,
      "step": 7666
    },
    {
      "epoch": 2.924346923828125e-05,
      "step": 7666,
      "training_step_time": 9.931648015975952
    },
    {
      "epoch": 2.924346923828125e-05,
      "model_forward_time": 3.1453773975372314,
      "step": 7666
    },
    {
      "epoch": 2.924346923828125e-05,
      "step": 7666,
      "training_step_time": 9.930259943008423
    },
    {
      "epoch": 2.924346923828125e-05,
      "model_forward_time": 3.4229395389556885,
      "step": 7666
    },
    {
      "epoch": 2.924346923828125e-05,
      "step": 7666,
      "training_step_time": 10.20724368095398
    },
    {
      "epoch": 2.924346923828125e-05,
      "model_forward_time": 3.141418218612671,
      "step": 7666
    },
    {
      "epoch": 2.924346923828125e-05,
      "step": 7666,
      "training_step_time": 10.030284881591797
    },
    {
      "epoch": 2.9247283935546874e-05,
      "model_forward_time": 3.159576654434204,
      "step": 7667
    },
    {
      "epoch": 2.9247283935546874e-05,
      "step": 7667,
      "training_step_time": 9.946187973022461
    },
    {
      "epoch": 2.9247283935546874e-05,
      "model_forward_time": 3.144599199295044,
      "step": 7667
    },
    {
      "epoch": 2.9247283935546874e-05,
      "step": 7667,
      "training_step_time": 9.925889015197754
    },
    {
      "epoch": 2.9247283935546874e-05,
      "model_forward_time": 3.141486167907715,
      "step": 7667
    },
    {
      "dynamics_loss_avg": 0.053283180110156536,
      "epoch": 2.9247283935546874e-05,
      "step": 7667
    },
    {
      "action_loss_avg": 0.01648643957450986,
      "epoch": 2.9247283935546874e-05,
      "step": 7667
    },
    {
      "epoch": 2.9247283935546874e-05,
      "step": 7667,
      "training_step_time": 9.92594861984253
    },
    {
      "epoch": 2.9247283935546874e-05,
      "model_forward_time": 3.4135472774505615,
      "step": 7667
    },
    {
      "epoch": 2.9247283935546874e-05,
      "step": 7667,
      "training_step_time": 10.333441495895386
    },
    {
      "epoch": 2.92510986328125e-05,
      "model_forward_time": 3.1618144512176514,
      "step": 7668
    },
    {
      "epoch": 2.92510986328125e-05,
      "step": 7668,
      "training_step_time": 9.9409818649292
    },
    {
      "epoch": 2.92510986328125e-05,
      "model_forward_time": 3.1426210403442383,
      "step": 7668
    },
    {
      "epoch": 2.92510986328125e-05,
      "step": 7668,
      "training_step_time": 9.921780109405518
    },
    {
      "epoch": 2.92510986328125e-05,
      "model_forward_time": 3.136806011199951,
      "step": 7668
    },
    {
      "epoch": 2.92510986328125e-05,
      "step": 7668,
      "training_step_time": 9.913255453109741
    },
    {
      "epoch": 2.92510986328125e-05,
      "model_forward_time": 3.140218496322632,
      "step": 7668
    },
    {
      "epoch": 2.92510986328125e-05,
      "step": 7668,
      "training_step_time": 10.273248195648193
    },
    {
      "epoch": 2.9254913330078123e-05,
      "model_forward_time": 3.41959810256958,
      "step": 7669
    },
    {
      "epoch": 2.9254913330078123e-05,
      "step": 7669,
      "training_step_time": 10.202154159545898
    },
    {
      "epoch": 2.9254913330078123e-05,
      "model_forward_time": 3.142540454864502,
      "step": 7669
    },
    {
      "epoch": 2.9254913330078123e-05,
      "step": 7669,
      "training_step_time": 9.915531396865845
    },
    {
      "epoch": 2.9254913330078123e-05,
      "model_forward_time": 3.1399221420288086,
      "step": 7669
    },
    {
      "epoch": 2.9254913330078123e-05,
      "step": 7669,
      "training_step_time": 9.911162614822388
    },
    {
      "epoch": 2.9254913330078123e-05,
      "model_forward_time": 3.1422677040100098,
      "step": 7669
    },
    {
      "epoch": 2.9254913330078123e-05,
      "step": 7669,
      "training_step_time": 10.288285732269287
    },
    {
      "epoch": 2.925872802734375e-05,
      "grad_norm": 0.41713279485702515,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 7670
    },
    {
      "epoch": 2.925872802734375e-05,
      "model_forward_time": 3.1516196727752686,
      "step": 7670
    },
    {
      "dynamics_loss_avg": 0.04384498670697212,
      "epoch": 2.925872802734375e-05,
      "step": 7670
    },
    {
      "action_loss_avg": 0.020362977497279645,
      "epoch": 2.925872802734375e-05,
      "step": 7670
    },
    {
      "epoch": 2.925872802734375e-05,
      "step": 7670,
      "training_step_time": 9.929418087005615
    },
    {
      "epoch": 2.925872802734375e-05,
      "model_forward_time": 3.4376864433288574,
      "step": 7670
    },
    {
      "epoch": 2.925872802734375e-05,
      "step": 7670,
      "training_step_time": 10.206088304519653
    },
    {
      "epoch": 2.925872802734375e-05,
      "model_forward_time": 3.1399805545806885,
      "step": 7670
    },
    {
      "epoch": 2.925872802734375e-05,
      "step": 7670,
      "training_step_time": 9.913997888565063
    },
    {
      "epoch": 2.925872802734375e-05,
      "model_forward_time": 3.1410553455352783,
      "step": 7670
    },
    {
      "epoch": 2.925872802734375e-05,
      "step": 7670,
      "training_step_time": 10.312591314315796
    },
    {
      "epoch": 2.9262542724609376e-05,
      "model_forward_time": 3.1594066619873047,
      "step": 7671
    },
    {
      "epoch": 2.9262542724609376e-05,
      "step": 7671,
      "training_step_time": 9.93866491317749
    },
    {
      "epoch": 2.9262542724609376e-05,
      "model_forward_time": 3.1400508880615234,
      "step": 7671
    },
    {
      "epoch": 2.9262542724609376e-05,
      "step": 7671,
      "training_step_time": 9.915987968444824
    },
    {
      "epoch": 2.9262542724609376e-05,
      "model_forward_time": 3.1462178230285645,
      "step": 7671
    },
    {
      "epoch": 2.9262542724609376e-05,
      "step": 7671,
      "training_step_time": 10.211973190307617
    },
    {
      "epoch": 2.9262542724609376e-05,
      "model_forward_time": 3.143685817718506,
      "step": 7671
    },
    {
      "epoch": 2.9262542724609376e-05,
      "step": 7671,
      "training_step_time": 10.0267915725708
    },
    {
      "epoch": 2.9266357421875e-05,
      "model_forward_time": 3.1590585708618164,
      "step": 7672
    },
    {
      "epoch": 2.9266357421875e-05,
      "step": 7672,
      "training_step_time": 9.93867540359497
    },
    {
      "epoch": 2.9266357421875e-05,
      "model_forward_time": 3.1436548233032227,
      "step": 7672
    },
    {
      "epoch": 2.9266357421875e-05,
      "step": 7672,
      "training_step_time": 9.92841124534607
    },
    {
      "epoch": 2.9266357421875e-05,
      "model_forward_time": 3.14961838722229,
      "step": 7672
    },
    {
      "dynamics_loss_avg": 0.0569350378587842,
      "epoch": 2.9266357421875e-05,
      "step": 7672
    },
    {
      "action_loss_avg": 0.0196052017621696,
      "epoch": 2.9266357421875e-05,
      "step": 7672
    },
    {
      "epoch": 2.9266357421875e-05,
      "step": 7672,
      "training_step_time": 9.936489343643188
    },
    {
      "epoch": 2.9266357421875e-05,
      "model_forward_time": 3.441286087036133,
      "step": 7672
    },
    {
      "epoch": 2.9266357421875e-05,
      "step": 7672,
      "training_step_time": 10.321150779724121
    },
    {
      "epoch": 2.9270172119140625e-05,
      "model_forward_time": 3.1628499031066895,
      "step": 7673
    },
    {
      "epoch": 2.9270172119140625e-05,
      "step": 7673,
      "training_step_time": 9.94820761680603
    },
    {
      "epoch": 2.9270172119140625e-05,
      "model_forward_time": 3.1419878005981445,
      "step": 7673
    },
    {
      "epoch": 2.9270172119140625e-05,
      "step": 7673,
      "training_step_time": 9.927999019622803
    },
    {
      "epoch": 2.9270172119140625e-05,
      "model_forward_time": 3.147193431854248,
      "step": 7673
    },
    {
      "epoch": 2.9270172119140625e-05,
      "step": 7673,
      "training_step_time": 9.928553104400635
    },
    {
      "epoch": 2.9270172119140625e-05,
      "model_forward_time": 3.1466598510742188,
      "step": 7673
    },
    {
      "epoch": 2.9270172119140625e-05,
      "step": 7673,
      "training_step_time": 10.335559606552124
    },
    {
      "epoch": 2.927398681640625e-05,
      "model_forward_time": 3.156526565551758,
      "step": 7674
    },
    {
      "epoch": 2.927398681640625e-05,
      "step": 7674,
      "training_step_time": 10.24816083908081
    },
    {
      "epoch": 2.927398681640625e-05,
      "model_forward_time": 3.1434662342071533,
      "step": 7674
    },
    {
      "epoch": 2.927398681640625e-05,
      "step": 7674,
      "training_step_time": 9.931385517120361
    },
    {
      "epoch": 2.927398681640625e-05,
      "model_forward_time": 3.1504294872283936,
      "step": 7674
    },
    {
      "epoch": 2.927398681640625e-05,
      "step": 7674,
      "training_step_time": 9.935117959976196
    },
    {
      "epoch": 2.927398681640625e-05,
      "model_forward_time": 3.148442506790161,
      "step": 7674
    },
    {
      "epoch": 2.927398681640625e-05,
      "step": 7674,
      "training_step_time": 10.305518388748169
    },
    {
      "epoch": 2.9277801513671874e-05,
      "model_forward_time": 3.1616404056549072,
      "step": 7675
    },
    {
      "dynamics_loss_avg": 0.05648876763880253,
      "epoch": 2.9277801513671874e-05,
      "step": 7675
    },
    {
      "action_loss_avg": 0.0198038587346673,
      "epoch": 2.9277801513671874e-05,
      "step": 7675
    },
    {
      "epoch": 2.9277801513671874e-05,
      "step": 7675,
      "training_step_time": 9.949226140975952
    },
    {
      "epoch": 2.9277801513671874e-05,
      "model_forward_time": 3.4103150367736816,
      "step": 7675
    },
    {
      "epoch": 2.9277801513671874e-05,
      "step": 7675,
      "training_step_time": 10.193493127822876
    },
    {
      "epoch": 2.9277801513671874e-05,
      "model_forward_time": 3.147742509841919,
      "step": 7675
    },
    {
      "epoch": 2.9277801513671874e-05,
      "step": 7675,
      "training_step_time": 9.927796602249146
    },
    {
      "epoch": 2.9277801513671874e-05,
      "model_forward_time": 3.1378605365753174,
      "step": 7675
    },
    {
      "epoch": 2.9277801513671874e-05,
      "step": 7675,
      "training_step_time": 10.27995753288269
    },
    {
      "epoch": 2.92816162109375e-05,
      "model_forward_time": 3.1424455642700195,
      "step": 7676
    },
    {
      "epoch": 2.92816162109375e-05,
      "step": 7676,
      "training_step_time": 9.92115592956543
    },
    {
      "epoch": 2.92816162109375e-05,
      "model_forward_time": 2.433750867843628,
      "step": 7676
    },
    {
      "epoch": 2.92816162109375e-05,
      "step": 7676,
      "training_step_time": 7.460552215576172
    },
    {
      "epoch": 2.92816162109375e-05,
      "model_forward_time": 3.129490375518799,
      "step": 7676
    },
    {
      "epoch": 2.92816162109375e-05,
      "step": 7676,
      "training_step_time": 10.172786951065063
    },
    {
      "epoch": 2.92816162109375e-05,
      "model_forward_time": 3.1352405548095703,
      "step": 7676
    },
    {
      "epoch": 2.92816162109375e-05,
      "step": 7676,
      "training_step_time": 10.320422887802124
    },
    {
      "epoch": 2.9285430908203123e-05,
      "model_forward_time": 3.1539130210876465,
      "step": 7677
    },
    {
      "epoch": 2.9285430908203123e-05,
      "step": 7677,
      "training_step_time": 9.926532745361328
    },
    {
      "epoch": 2.9285430908203123e-05,
      "model_forward_time": 3.137147903442383,
      "step": 7677
    },
    {
      "epoch": 2.9285430908203123e-05,
      "step": 7677,
      "training_step_time": 9.910593748092651
    },
    {
      "epoch": 2.9285430908203123e-05,
      "model_forward_time": 3.132309913635254,
      "step": 7677
    },
    {
      "dynamics_loss_avg": 0.05151425674557686,
      "epoch": 2.9285430908203123e-05,
      "step": 7677
    },
    {
      "action_loss_avg": 0.020626282133162022,
      "epoch": 2.9285430908203123e-05,
      "step": 7677
    },
    {
      "epoch": 2.9285430908203123e-05,
      "step": 7677,
      "training_step_time": 9.915416479110718
    },
    {
      "epoch": 2.9285430908203123e-05,
      "model_forward_time": 3.391737222671509,
      "step": 7677
    },
    {
      "epoch": 2.9285430908203123e-05,
      "step": 7677,
      "training_step_time": 10.30344295501709
    },
    {
      "epoch": 2.928924560546875e-05,
      "model_forward_time": 3.160865068435669,
      "step": 7678
    },
    {
      "epoch": 2.928924560546875e-05,
      "step": 7678,
      "training_step_time": 9.937408208847046
    },
    {
      "epoch": 2.928924560546875e-05,
      "model_forward_time": 3.1369993686676025,
      "step": 7678
    },
    {
      "epoch": 2.928924560546875e-05,
      "step": 7678,
      "training_step_time": 9.911423683166504
    },
    {
      "epoch": 2.928924560546875e-05,
      "model_forward_time": 3.138643980026245,
      "step": 7678
    },
    {
      "epoch": 2.928924560546875e-05,
      "step": 7678,
      "training_step_time": 9.913551807403564
    },
    {
      "epoch": 2.928924560546875e-05,
      "model_forward_time": 3.1405224800109863,
      "step": 7678
    },
    {
      "epoch": 2.928924560546875e-05,
      "step": 7678,
      "training_step_time": 10.374004602432251
    },
    {
      "epoch": 2.9293060302734376e-05,
      "model_forward_time": 3.4070310592651367,
      "step": 7679
    },
    {
      "epoch": 2.9293060302734376e-05,
      "step": 7679,
      "training_step_time": 10.183512687683105
    },
    {
      "epoch": 2.9293060302734376e-05,
      "model_forward_time": 3.1395421028137207,
      "step": 7679
    },
    {
      "epoch": 2.9293060302734376e-05,
      "step": 7679,
      "training_step_time": 9.915625810623169
    },
    {
      "epoch": 2.9293060302734376e-05,
      "model_forward_time": 3.132659435272217,
      "step": 7679
    },
    {
      "epoch": 2.9293060302734376e-05,
      "step": 7679,
      "training_step_time": 9.908353090286255
    },
    {
      "epoch": 2.9293060302734376e-05,
      "model_forward_time": 3.139495372772217,
      "step": 7679
    },
    {
      "epoch": 2.9293060302734376e-05,
      "step": 7679,
      "training_step_time": 10.288208246231079
    },
    {
      "epoch": 2.9296875e-05,
      "grad_norm": 0.34361040592193604,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 7680
    },
    {
      "epoch": 2.9296875e-05,
      "model_forward_time": 3.1506807804107666,
      "step": 7680
    },
    {
      "dynamics_loss_avg": 0.03904694234952331,
      "epoch": 2.9296875e-05,
      "step": 7680
    },
    {
      "action_loss_avg": 0.011446574935689569,
      "epoch": 2.9296875e-05,
      "step": 7680
    },
    {
      "epoch": 2.9296875e-05,
      "step": 7680,
      "training_step_time": 9.935835838317871
    },
    {
      "epoch": 2.9296875e-05,
      "model_forward_time": 3.135493516921997,
      "step": 7680
    },
    {
      "epoch": 2.9296875e-05,
      "step": 7680,
      "training_step_time": 10.20188856124878
    },
    {
      "epoch": 2.9296875e-05,
      "model_forward_time": 3.1422457695007324,
      "step": 7680
    },
    {
      "epoch": 2.9296875e-05,
      "step": 7680,
      "training_step_time": 9.919986724853516
    },
    {
      "epoch": 2.9296875e-05,
      "model_forward_time": 3.137939691543579,
      "step": 7680
    },
    {
      "epoch": 2.9296875e-05,
      "step": 7680,
      "training_step_time": 10.005191802978516
    },
    {
      "epoch": 2.9300689697265625e-05,
      "model_forward_time": 3.1526365280151367,
      "step": 7681
    },
    {
      "epoch": 2.9300689697265625e-05,
      "step": 7681,
      "training_step_time": 9.921351909637451
    },
    {
      "epoch": 2.9300689697265625e-05,
      "model_forward_time": 3.1309726238250732,
      "step": 7681
    },
    {
      "epoch": 2.9300689697265625e-05,
      "step": 7681,
      "training_step_time": 9.902738332748413
    },
    {
      "epoch": 2.9300689697265625e-05,
      "model_forward_time": 3.137767791748047,
      "step": 7681
    },
    {
      "epoch": 2.9300689697265625e-05,
      "step": 7681,
      "training_step_time": 10.222405433654785
    },
    {
      "epoch": 2.9300689697265625e-05,
      "model_forward_time": 3.142714738845825,
      "step": 7681
    },
    {
      "epoch": 2.9300689697265625e-05,
      "step": 7681,
      "training_step_time": 10.300530910491943
    },
    {
      "epoch": 2.930450439453125e-05,
      "model_forward_time": 3.155947208404541,
      "step": 7682
    },
    {
      "epoch": 2.930450439453125e-05,
      "step": 7682,
      "training_step_time": 9.931601762771606
    },
    {
      "epoch": 2.930450439453125e-05,
      "model_forward_time": 3.137676954269409,
      "step": 7682
    },
    {
      "epoch": 2.930450439453125e-05,
      "step": 7682,
      "training_step_time": 9.91618275642395
    },
    {
      "epoch": 2.930450439453125e-05,
      "model_forward_time": 3.146667242050171,
      "step": 7682
    },
    {
      "dynamics_loss_avg": 0.06791428178548813,
      "epoch": 2.930450439453125e-05,
      "step": 7682
    },
    {
      "action_loss_avg": 0.026593520678579806,
      "epoch": 2.930450439453125e-05,
      "step": 7682
    },
    {
      "epoch": 2.930450439453125e-05,
      "step": 7682,
      "training_step_time": 9.925568103790283
    },
    {
      "epoch": 2.930450439453125e-05,
      "model_forward_time": 3.1427130699157715,
      "step": 7682
    },
    {
      "epoch": 2.930450439453125e-05,
      "step": 7682,
      "training_step_time": 10.331154823303223
    },
    {
      "epoch": 2.9308319091796874e-05,
      "model_forward_time": 3.1710636615753174,
      "step": 7683
    },
    {
      "epoch": 2.9308319091796874e-05,
      "step": 7683,
      "training_step_time": 9.974631547927856
    },
    {
      "epoch": 2.9308319091796874e-05,
      "model_forward_time": 3.1430509090423584,
      "step": 7683
    },
    {
      "epoch": 2.9308319091796874e-05,
      "step": 7683,
      "training_step_time": 9.921731948852539
    },
    {
      "epoch": 2.9308319091796874e-05,
      "model_forward_time": 3.1431782245635986,
      "step": 7683
    },
    {
      "epoch": 2.9308319091796874e-05,
      "step": 7683,
      "training_step_time": 9.923635482788086
    },
    {
      "epoch": 2.9308319091796874e-05,
      "model_forward_time": 3.141455888748169,
      "step": 7683
    },
    {
      "epoch": 2.9308319091796874e-05,
      "step": 7683,
      "training_step_time": 10.313106536865234
    },
    {
      "epoch": 2.93121337890625e-05,
      "model_forward_time": 3.4521257877349854,
      "step": 7684
    },
    {
      "epoch": 2.93121337890625e-05,
      "step": 7684,
      "training_step_time": 10.235400676727295
    },
    {
      "epoch": 2.93121337890625e-05,
      "model_forward_time": 3.1422533988952637,
      "step": 7684
    },
    {
      "epoch": 2.93121337890625e-05,
      "step": 7684,
      "training_step_time": 9.925662517547607
    },
    {
      "epoch": 2.93121337890625e-05,
      "model_forward_time": 3.1536483764648438,
      "step": 7684
    },
    {
      "epoch": 2.93121337890625e-05,
      "step": 7684,
      "training_step_time": 9.937671899795532
    },
    {
      "epoch": 2.93121337890625e-05,
      "model_forward_time": 3.143375873565674,
      "step": 7684
    },
    {
      "epoch": 2.93121337890625e-05,
      "step": 7684,
      "training_step_time": 10.27424168586731
    },
    {
      "epoch": 2.9315948486328124e-05,
      "model_forward_time": 3.1687302589416504,
      "step": 7685
    },
    {
      "dynamics_loss_avg": 0.05224066935479641,
      "epoch": 2.9315948486328124e-05,
      "step": 7685
    },
    {
      "action_loss_avg": 0.01944937938824296,
      "epoch": 2.9315948486328124e-05,
      "step": 7685
    },
    {
      "epoch": 2.9315948486328124e-05,
      "step": 7685,
      "training_step_time": 9.962119340896606
    },
    {
      "epoch": 2.9315948486328124e-05,
      "model_forward_time": 3.1461758613586426,
      "step": 7685
    },
    {
      "epoch": 2.9315948486328124e-05,
      "step": 7685,
      "training_step_time": 10.246941566467285
    },
    {
      "epoch": 2.9315948486328124e-05,
      "model_forward_time": 3.141719102859497,
      "step": 7685
    },
    {
      "epoch": 2.9315948486328124e-05,
      "step": 7685,
      "training_step_time": 9.919525384902954
    },
    {
      "epoch": 2.9315948486328124e-05,
      "model_forward_time": 3.1412441730499268,
      "step": 7685
    },
    {
      "epoch": 2.9315948486328124e-05,
      "step": 7685,
      "training_step_time": 10.280348539352417
    },
    {
      "epoch": 2.931976318359375e-05,
      "model_forward_time": 3.1511027812957764,
      "step": 7686
    },
    {
      "epoch": 2.931976318359375e-05,
      "step": 7686,
      "training_step_time": 9.933277130126953
    },
    {
      "epoch": 2.931976318359375e-05,
      "model_forward_time": 3.14387583732605,
      "step": 7686
    },
    {
      "epoch": 2.931976318359375e-05,
      "step": 7686,
      "training_step_time": 9.92204213142395
    },
    {
      "epoch": 2.931976318359375e-05,
      "model_forward_time": 3.4361507892608643,
      "step": 7686
    },
    {
      "epoch": 2.931976318359375e-05,
      "step": 7686,
      "training_step_time": 10.214103937149048
    },
    {
      "epoch": 2.931976318359375e-05,
      "model_forward_time": 3.1407291889190674,
      "step": 7686
    },
    {
      "epoch": 2.931976318359375e-05,
      "step": 7686,
      "training_step_time": 10.021390676498413
    },
    {
      "epoch": 2.9323577880859376e-05,
      "model_forward_time": 3.1522128582000732,
      "step": 7687
    },
    {
      "epoch": 2.9323577880859376e-05,
      "step": 7687,
      "training_step_time": 9.938173294067383
    },
    {
      "epoch": 2.9323577880859376e-05,
      "model_forward_time": 3.146817684173584,
      "step": 7687
    },
    {
      "epoch": 2.9323577880859376e-05,
      "step": 7687,
      "training_step_time": 9.931495904922485
    },
    {
      "epoch": 2.9323577880859376e-05,
      "model_forward_time": 3.1423914432525635,
      "step": 7687
    },
    {
      "dynamics_loss_avg": 0.06501821484416723,
      "epoch": 2.9323577880859376e-05,
      "step": 7687
    },
    {
      "action_loss_avg": 0.025397123862057926,
      "epoch": 2.9323577880859376e-05,
      "step": 7687
    },
    {
      "epoch": 2.9323577880859376e-05,
      "step": 7687,
      "training_step_time": 9.931460618972778
    },
    {
      "epoch": 2.9323577880859376e-05,
      "model_forward_time": 2.4440994262695312,
      "step": 7687
    },
    {
      "epoch": 2.9323577880859376e-05,
      "step": 7687,
      "training_step_time": 10.32528305053711
    },
    {
      "epoch": 2.9327392578125e-05,
      "model_forward_time": 3.1581573486328125,
      "step": 7688
    },
    {
      "epoch": 2.9327392578125e-05,
      "step": 7688,
      "training_step_time": 9.950993776321411
    },
    {
      "epoch": 2.9327392578125e-05,
      "model_forward_time": 3.1434550285339355,
      "step": 7688
    },
    {
      "epoch": 2.9327392578125e-05,
      "step": 7688,
      "training_step_time": 9.92500114440918
    },
    {
      "epoch": 2.9327392578125e-05,
      "model_forward_time": 3.1393306255340576,
      "step": 7688
    },
    {
      "epoch": 2.9327392578125e-05,
      "step": 7688,
      "training_step_time": 9.9242103099823
    },
    {
      "epoch": 2.9327392578125e-05,
      "model_forward_time": 3.144104242324829,
      "step": 7688
    },
    {
      "epoch": 2.9327392578125e-05,
      "step": 7688,
      "training_step_time": 10.278241395950317
    },
    {
      "epoch": 2.9331207275390625e-05,
      "model_forward_time": 3.420633554458618,
      "step": 7689
    },
    {
      "epoch": 2.9331207275390625e-05,
      "step": 7689,
      "training_step_time": 10.202925443649292
    },
    {
      "epoch": 2.9331207275390625e-05,
      "model_forward_time": 3.1427316665649414,
      "step": 7689
    },
    {
      "epoch": 2.9331207275390625e-05,
      "step": 7689,
      "training_step_time": 9.925191402435303
    },
    {
      "epoch": 2.9331207275390625e-05,
      "model_forward_time": 3.1415371894836426,
      "step": 7689
    },
    {
      "epoch": 2.9331207275390625e-05,
      "step": 7689,
      "training_step_time": 9.92667269706726
    },
    {
      "epoch": 2.9331207275390625e-05,
      "model_forward_time": 3.145568609237671,
      "step": 7689
    },
    {
      "epoch": 2.9331207275390625e-05,
      "step": 7689,
      "training_step_time": 10.330373048782349
    },
    {
      "epoch": 2.933502197265625e-05,
      "grad_norm": 0.36321762204170227,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 7690
    },
    {
      "epoch": 2.933502197265625e-05,
      "model_forward_time": 3.163285970687866,
      "step": 7690
    },
    {
      "dynamics_loss_avg": 0.061395745165646075,
      "epoch": 2.933502197265625e-05,
      "step": 7690
    },
    {
      "action_loss_avg": 0.02028063954785466,
      "epoch": 2.933502197265625e-05,
      "step": 7690
    },
    {
      "epoch": 2.933502197265625e-05,
      "step": 7690,
      "training_step_time": 9.953495502471924
    },
    {
      "epoch": 2.933502197265625e-05,
      "model_forward_time": 3.429931163787842,
      "step": 7690
    },
    {
      "epoch": 2.933502197265625e-05,
      "step": 7690,
      "training_step_time": 10.20959734916687
    },
    {
      "epoch": 2.933502197265625e-05,
      "model_forward_time": 3.1397299766540527,
      "step": 7690
    },
    {
      "epoch": 2.933502197265625e-05,
      "step": 7690,
      "training_step_time": 9.91679048538208
    },
    {
      "epoch": 2.933502197265625e-05,
      "model_forward_time": 3.138511896133423,
      "step": 7690
    },
    {
      "epoch": 2.933502197265625e-05,
      "step": 7690,
      "training_step_time": 10.256319999694824
    },
    {
      "epoch": 2.9338836669921874e-05,
      "model_forward_time": 3.163233757019043,
      "step": 7691
    },
    {
      "epoch": 2.9338836669921874e-05,
      "step": 7691,
      "training_step_time": 9.93935227394104
    },
    {
      "epoch": 2.9338836669921874e-05,
      "model_forward_time": 2.4376227855682373,
      "step": 7691
    },
    {
      "epoch": 2.9338836669921874e-05,
      "step": 7691,
      "training_step_time": 7.46772575378418
    },
    {
      "epoch": 2.9338836669921874e-05,
      "model_forward_time": 3.1510567665100098,
      "step": 7691
    },
    {
      "epoch": 2.9338836669921874e-05,
      "step": 7691,
      "training_step_time": 10.205250024795532
    },
    {
      "epoch": 2.9338836669921874e-05,
      "model_forward_time": 3.145914077758789,
      "step": 7691
    },
    {
      "epoch": 2.9338836669921874e-05,
      "step": 7691,
      "training_step_time": 10.297833919525146
    },
    {
      "epoch": 2.93426513671875e-05,
      "model_forward_time": 3.161515235900879,
      "step": 7692
    },
    {
      "epoch": 2.93426513671875e-05,
      "step": 7692,
      "training_step_time": 9.938267469406128
    },
    {
      "epoch": 2.93426513671875e-05,
      "model_forward_time": 3.1418685913085938,
      "step": 7692
    },
    {
      "epoch": 2.93426513671875e-05,
      "step": 7692,
      "training_step_time": 9.914567708969116
    },
    {
      "epoch": 2.93426513671875e-05,
      "model_forward_time": 3.1451096534729004,
      "step": 7692
    },
    {
      "dynamics_loss_avg": 0.05916986837983131,
      "epoch": 2.93426513671875e-05,
      "step": 7692
    },
    {
      "action_loss_avg": 0.02400926183909178,
      "epoch": 2.93426513671875e-05,
      "step": 7692
    },
    {
      "epoch": 2.93426513671875e-05,
      "step": 7692,
      "training_step_time": 9.923651218414307
    },
    {
      "epoch": 2.93426513671875e-05,
      "model_forward_time": 3.4311363697052,
      "step": 7692
    },
    {
      "epoch": 2.93426513671875e-05,
      "step": 7692,
      "training_step_time": 10.302848815917969
    },
    {
      "epoch": 2.9346466064453124e-05,
      "model_forward_time": 3.159938097000122,
      "step": 7693
    },
    {
      "epoch": 2.9346466064453124e-05,
      "step": 7693,
      "training_step_time": 9.944119930267334
    },
    {
      "epoch": 2.9346466064453124e-05,
      "model_forward_time": 3.1458277702331543,
      "step": 7693
    },
    {
      "epoch": 2.9346466064453124e-05,
      "step": 7693,
      "training_step_time": 9.93225622177124
    },
    {
      "epoch": 2.9346466064453124e-05,
      "model_forward_time": 3.149571657180786,
      "step": 7693
    },
    {
      "epoch": 2.9346466064453124e-05,
      "step": 7693,
      "training_step_time": 9.932762861251831
    },
    {
      "epoch": 2.9346466064453124e-05,
      "model_forward_time": 3.1450867652893066,
      "step": 7693
    },
    {
      "epoch": 2.9346466064453124e-05,
      "step": 7693,
      "training_step_time": 10.041070461273193
    },
    {
      "epoch": 2.935028076171875e-05,
      "model_forward_time": 3.164360523223877,
      "step": 7694
    },
    {
      "epoch": 2.935028076171875e-05,
      "step": 7694,
      "training_step_time": 10.275389432907104
    },
    {
      "epoch": 2.935028076171875e-05,
      "model_forward_time": 3.1498212814331055,
      "step": 7694
    },
    {
      "epoch": 2.935028076171875e-05,
      "step": 7694,
      "training_step_time": 9.93506121635437
    },
    {
      "epoch": 2.935028076171875e-05,
      "model_forward_time": 3.1495251655578613,
      "step": 7694
    },
    {
      "epoch": 2.935028076171875e-05,
      "step": 7694,
      "training_step_time": 9.933520555496216
    },
    {
      "epoch": 2.935028076171875e-05,
      "model_forward_time": 3.1434903144836426,
      "step": 7694
    },
    {
      "epoch": 2.935028076171875e-05,
      "step": 7694,
      "training_step_time": 10.275131940841675
    },
    {
      "epoch": 2.9354095458984376e-05,
      "model_forward_time": 3.154639959335327,
      "step": 7695
    },
    {
      "dynamics_loss_avg": 0.05512247644364834,
      "epoch": 2.9354095458984376e-05,
      "step": 7695
    },
    {
      "action_loss_avg": 0.022778818476945163,
      "epoch": 2.9354095458984376e-05,
      "step": 7695
    },
    {
      "epoch": 2.9354095458984376e-05,
      "step": 7695,
      "training_step_time": 9.94162106513977
    },
    {
      "epoch": 2.9354095458984376e-05,
      "model_forward_time": 3.388648748397827,
      "step": 7695
    },
    {
      "epoch": 2.9354095458984376e-05,
      "step": 7695,
      "training_step_time": 10.172076940536499
    },
    {
      "epoch": 2.9354095458984376e-05,
      "model_forward_time": 3.1422338485717773,
      "step": 7695
    },
    {
      "epoch": 2.9354095458984376e-05,
      "step": 7695,
      "training_step_time": 9.924783706665039
    },
    {
      "epoch": 2.9354095458984376e-05,
      "model_forward_time": 3.1454579830169678,
      "step": 7695
    },
    {
      "epoch": 2.9354095458984376e-05,
      "step": 7695,
      "training_step_time": 10.021209478378296
    },
    {
      "epoch": 2.935791015625e-05,
      "model_forward_time": 3.1691055297851562,
      "step": 7696
    },
    {
      "epoch": 2.935791015625e-05,
      "step": 7696,
      "training_step_time": 9.952642679214478
    },
    {
      "epoch": 2.935791015625e-05,
      "model_forward_time": 3.146803855895996,
      "step": 7696
    },
    {
      "epoch": 2.935791015625e-05,
      "step": 7696,
      "training_step_time": 9.926471948623657
    },
    {
      "epoch": 2.935791015625e-05,
      "model_forward_time": 3.467212200164795,
      "step": 7696
    },
    {
      "epoch": 2.935791015625e-05,
      "step": 7696,
      "training_step_time": 10.248585224151611
    },
    {
      "epoch": 2.935791015625e-05,
      "model_forward_time": 3.1462197303771973,
      "step": 7696
    },
    {
      "epoch": 2.935791015625e-05,
      "step": 7696,
      "training_step_time": 10.349261999130249
    },
    {
      "epoch": 2.9361724853515625e-05,
      "model_forward_time": 3.148526906967163,
      "step": 7697
    },
    {
      "epoch": 2.9361724853515625e-05,
      "step": 7697,
      "training_step_time": 9.937851190567017
    },
    {
      "epoch": 2.9361724853515625e-05,
      "model_forward_time": 3.156470537185669,
      "step": 7697
    },
    {
      "epoch": 2.9361724853515625e-05,
      "step": 7697,
      "training_step_time": 9.947375059127808
    },
    {
      "epoch": 2.9361724853515625e-05,
      "model_forward_time": 3.143901824951172,
      "step": 7697
    },
    {
      "dynamics_loss_avg": 0.06658986993134022,
      "epoch": 2.9361724853515625e-05,
      "step": 7697
    },
    {
      "action_loss_avg": 0.02175003942102194,
      "epoch": 2.9361724853515625e-05,
      "step": 7697
    },
    {
      "epoch": 2.9361724853515625e-05,
      "step": 7697,
      "training_step_time": 9.923483610153198
    },
    {
      "epoch": 2.9361724853515625e-05,
      "model_forward_time": 3.4145946502685547,
      "step": 7697
    },
    {
      "epoch": 2.9361724853515625e-05,
      "step": 7697,
      "training_step_time": 10.290886402130127
    },
    {
      "epoch": 2.936553955078125e-05,
      "model_forward_time": 3.159958600997925,
      "step": 7698
    },
    {
      "epoch": 2.936553955078125e-05,
      "step": 7698,
      "training_step_time": 9.943486452102661
    },
    {
      "epoch": 2.936553955078125e-05,
      "model_forward_time": 3.1427412033081055,
      "step": 7698
    },
    {
      "epoch": 2.936553955078125e-05,
      "step": 7698,
      "training_step_time": 9.919911861419678
    },
    {
      "epoch": 2.936553955078125e-05,
      "model_forward_time": 3.1396665573120117,
      "step": 7698
    },
    {
      "epoch": 2.936553955078125e-05,
      "step": 7698,
      "training_step_time": 9.924286365509033
    },
    {
      "epoch": 2.936553955078125e-05,
      "model_forward_time": 3.142338514328003,
      "step": 7698
    },
    {
      "epoch": 2.936553955078125e-05,
      "step": 7698,
      "training_step_time": 10.263268947601318
    },
    {
      "epoch": 2.9369354248046874e-05,
      "model_forward_time": 3.4255621433258057,
      "step": 7699
    },
    {
      "epoch": 2.9369354248046874e-05,
      "step": 7699,
      "training_step_time": 10.208235740661621
    },
    {
      "epoch": 2.9369354248046874e-05,
      "model_forward_time": 3.1434736251831055,
      "step": 7699
    },
    {
      "epoch": 2.9369354248046874e-05,
      "step": 7699,
      "training_step_time": 9.930552005767822
    },
    {
      "epoch": 2.9369354248046874e-05,
      "model_forward_time": 3.140836715698242,
      "step": 7699
    },
    {
      "epoch": 2.9369354248046874e-05,
      "step": 7699,
      "training_step_time": 9.92185354232788
    },
    {
      "epoch": 2.9369354248046874e-05,
      "model_forward_time": 3.1382880210876465,
      "step": 7699
    },
    {
      "epoch": 2.9369354248046874e-05,
      "step": 7699,
      "training_step_time": 10.236377239227295
    },
    {
      "epoch": 2.93731689453125e-05,
      "grad_norm": 0.5287105441093445,
      "learning_rate": 1e-05,
      "loss": 0.0785,
      "step": 7700
    },
    {
      "epoch": 2.93731689453125e-05,
      "model_forward_time": 3.1480953693389893,
      "step": 7700
    },
    {
      "dynamics_loss_avg": 0.056127327494323254,
      "epoch": 2.93731689453125e-05,
      "step": 7700
    },
    {
      "action_loss_avg": 0.021005330234766008,
      "epoch": 2.93731689453125e-05,
      "step": 7700
    },
    {
      "epoch": 2.93731689453125e-05,
      "step": 7700,
      "training_step_time": 9.93605351448059
    },
    {
      "epoch": 2.93731689453125e-05,
      "model_forward_time": 3.1326568126678467,
      "step": 7700
    },
    {
      "epoch": 2.93731689453125e-05,
      "step": 7700,
      "training_step_time": 10.235357999801636
    },
    {
      "epoch": 2.93731689453125e-05,
      "model_forward_time": 3.141983985900879,
      "step": 7700
    },
    {
      "epoch": 2.93731689453125e-05,
      "step": 7700,
      "training_step_time": 9.92127537727356
    },
    {
      "epoch": 2.93731689453125e-05,
      "model_forward_time": 3.1437952518463135,
      "step": 7700
    },
    {
      "epoch": 2.93731689453125e-05,
      "step": 7700,
      "training_step_time": 10.277313709259033
    },
    {
      "epoch": 2.9376983642578124e-05,
      "model_forward_time": 3.156372308731079,
      "step": 7701
    },
    {
      "epoch": 2.9376983642578124e-05,
      "step": 7701,
      "training_step_time": 9.937171936035156
    },
    {
      "epoch": 2.9376983642578124e-05,
      "model_forward_time": 3.1346940994262695,
      "step": 7701
    },
    {
      "epoch": 2.9376983642578124e-05,
      "step": 7701,
      "training_step_time": 9.916056394577026
    },
    {
      "epoch": 2.9376983642578124e-05,
      "model_forward_time": 3.4168155193328857,
      "step": 7701
    },
    {
      "epoch": 2.9376983642578124e-05,
      "step": 7701,
      "training_step_time": 10.196385145187378
    },
    {
      "epoch": 2.9376983642578124e-05,
      "model_forward_time": 3.147355794906616,
      "step": 7701
    },
    {
      "epoch": 2.9376983642578124e-05,
      "step": 7701,
      "training_step_time": 10.2561354637146
    },
    {
      "epoch": 2.938079833984375e-05,
      "model_forward_time": 3.1429615020751953,
      "step": 7702
    },
    {
      "epoch": 2.938079833984375e-05,
      "step": 7702,
      "training_step_time": 9.924614191055298
    },
    {
      "epoch": 2.938079833984375e-05,
      "model_forward_time": 3.132244825363159,
      "step": 7702
    },
    {
      "epoch": 2.938079833984375e-05,
      "step": 7702,
      "training_step_time": 9.913312435150146
    },
    {
      "epoch": 2.938079833984375e-05,
      "model_forward_time": 3.143686056137085,
      "step": 7702
    },
    {
      "dynamics_loss_avg": 0.06370531376451254,
      "epoch": 2.938079833984375e-05,
      "step": 7702
    },
    {
      "action_loss_avg": 0.020362847205251455,
      "epoch": 2.938079833984375e-05,
      "step": 7702
    },
    {
      "epoch": 2.938079833984375e-05,
      "step": 7702,
      "training_step_time": 9.930239200592041
    },
    {
      "epoch": 2.938079833984375e-05,
      "model_forward_time": 3.1442368030548096,
      "step": 7702
    },
    {
      "epoch": 2.938079833984375e-05,
      "step": 7702,
      "training_step_time": 10.308160781860352
    },
    {
      "epoch": 2.9384613037109376e-05,
      "model_forward_time": 3.154860496520996,
      "step": 7703
    },
    {
      "epoch": 2.9384613037109376e-05,
      "step": 7703,
      "training_step_time": 9.940162658691406
    },
    {
      "epoch": 2.9384613037109376e-05,
      "model_forward_time": 3.135685920715332,
      "step": 7703
    },
    {
      "epoch": 2.9384613037109376e-05,
      "step": 7703,
      "training_step_time": 9.915023803710938
    },
    {
      "epoch": 2.9384613037109376e-05,
      "model_forward_time": 3.1399447917938232,
      "step": 7703
    },
    {
      "epoch": 2.9384613037109376e-05,
      "step": 7703,
      "training_step_time": 9.920843362808228
    },
    {
      "epoch": 2.9384613037109376e-05,
      "model_forward_time": 3.1354169845581055,
      "step": 7703
    },
    {
      "epoch": 2.9384613037109376e-05,
      "step": 7703,
      "training_step_time": 10.28965950012207
    },
    {
      "epoch": 2.9388427734375e-05,
      "model_forward_time": 3.425290107727051,
      "step": 7704
    },
    {
      "epoch": 2.9388427734375e-05,
      "step": 7704,
      "training_step_time": 10.206031560897827
    },
    {
      "epoch": 2.9388427734375e-05,
      "model_forward_time": 3.137845993041992,
      "step": 7704
    },
    {
      "epoch": 2.9388427734375e-05,
      "step": 7704,
      "training_step_time": 9.918907165527344
    },
    {
      "epoch": 2.9388427734375e-05,
      "model_forward_time": 3.1402668952941895,
      "step": 7704
    },
    {
      "epoch": 2.9388427734375e-05,
      "step": 7704,
      "training_step_time": 9.919219255447388
    },
    {
      "epoch": 2.9388427734375e-05,
      "model_forward_time": 3.141211748123169,
      "step": 7704
    },
    {
      "epoch": 2.9388427734375e-05,
      "step": 7704,
      "training_step_time": 10.344194889068604
    },
    {
      "epoch": 2.9392242431640625e-05,
      "model_forward_time": 3.148206949234009,
      "step": 7705
    },
    {
      "dynamics_loss_avg": 0.05831659212708473,
      "epoch": 2.9392242431640625e-05,
      "step": 7705
    },
    {
      "action_loss_avg": 0.02563311317935586,
      "epoch": 2.9392242431640625e-05,
      "step": 7705
    },
    {
      "epoch": 2.9392242431640625e-05,
      "step": 7705,
      "training_step_time": 9.931971788406372
    },
    {
      "epoch": 2.9392242431640625e-05,
      "model_forward_time": 3.145111083984375,
      "step": 7705
    },
    {
      "epoch": 2.9392242431640625e-05,
      "step": 7705,
      "training_step_time": 10.225116729736328
    },
    {
      "epoch": 2.9392242431640625e-05,
      "model_forward_time": 3.1459176540374756,
      "step": 7705
    },
    {
      "epoch": 2.9392242431640625e-05,
      "step": 7705,
      "training_step_time": 9.925509691238403
    },
    {
      "epoch": 2.9392242431640625e-05,
      "model_forward_time": 3.136622905731201,
      "step": 7705
    },
    {
      "epoch": 2.9392242431640625e-05,
      "step": 7705,
      "training_step_time": 10.315925359725952
    },
    {
      "epoch": 2.939605712890625e-05,
      "model_forward_time": 2.4559195041656494,
      "step": 7706
    },
    {
      "epoch": 2.939605712890625e-05,
      "step": 7706,
      "training_step_time": 7.502955913543701
    },
    {
      "epoch": 2.939605712890625e-05,
      "model_forward_time": 3.1361265182495117,
      "step": 7706
    },
    {
      "epoch": 2.939605712890625e-05,
      "step": 7706,
      "training_step_time": 9.911212682723999
    },
    {
      "epoch": 2.939605712890625e-05,
      "model_forward_time": 3.430802345275879,
      "step": 7706
    },
    {
      "epoch": 2.939605712890625e-05,
      "step": 7706,
      "training_step_time": 10.20715069770813
    },
    {
      "epoch": 2.939605712890625e-05,
      "model_forward_time": 3.1377811431884766,
      "step": 7706
    },
    {
      "epoch": 2.939605712890625e-05,
      "step": 7706,
      "training_step_time": 10.273394107818604
    },
    {
      "epoch": 2.9399871826171874e-05,
      "model_forward_time": 3.144756555557251,
      "step": 7707
    },
    {
      "epoch": 2.9399871826171874e-05,
      "step": 7707,
      "training_step_time": 9.920270681381226
    },
    {
      "epoch": 2.9399871826171874e-05,
      "model_forward_time": 3.134711742401123,
      "step": 7707
    },
    {
      "epoch": 2.9399871826171874e-05,
      "step": 7707,
      "training_step_time": 9.904219150543213
    },
    {
      "epoch": 2.9399871826171874e-05,
      "model_forward_time": 3.127964735031128,
      "step": 7707
    },
    {
      "dynamics_loss_avg": 0.06000019814819098,
      "epoch": 2.9399871826171874e-05,
      "step": 7707
    },
    {
      "action_loss_avg": 0.02183340098708868,
      "epoch": 2.9399871826171874e-05,
      "step": 7707
    },
    {
      "epoch": 2.9399871826171874e-05,
      "step": 7707,
      "training_step_time": 9.9048011302948
    },
    {
      "epoch": 2.9399871826171874e-05,
      "model_forward_time": 3.136608600616455,
      "step": 7707
    },
    {
      "epoch": 2.9399871826171874e-05,
      "step": 7707,
      "training_step_time": 10.293363332748413
    },
    {
      "epoch": 2.94036865234375e-05,
      "model_forward_time": 3.473079204559326,
      "step": 7708
    },
    {
      "epoch": 2.94036865234375e-05,
      "step": 7708,
      "training_step_time": 10.240554809570312
    },
    {
      "epoch": 2.94036865234375e-05,
      "model_forward_time": 3.134561538696289,
      "step": 7708
    },
    {
      "epoch": 2.94036865234375e-05,
      "step": 7708,
      "training_step_time": 9.908035516738892
    },
    {
      "epoch": 2.94036865234375e-05,
      "model_forward_time": 3.1312344074249268,
      "step": 7708
    },
    {
      "epoch": 2.94036865234375e-05,
      "step": 7708,
      "training_step_time": 9.904848575592041
    },
    {
      "epoch": 2.94036865234375e-05,
      "model_forward_time": 3.135474920272827,
      "step": 7708
    },
    {
      "epoch": 2.94036865234375e-05,
      "step": 7708,
      "training_step_time": 10.304795980453491
    },
    {
      "epoch": 2.9407501220703124e-05,
      "model_forward_time": 3.4483346939086914,
      "step": 7709
    },
    {
      "epoch": 2.9407501220703124e-05,
      "step": 7709,
      "training_step_time": 10.22462773323059
    },
    {
      "epoch": 2.9407501220703124e-05,
      "model_forward_time": 3.1405885219573975,
      "step": 7709
    },
    {
      "epoch": 2.9407501220703124e-05,
      "step": 7709,
      "training_step_time": 9.913386344909668
    },
    {
      "epoch": 2.9407501220703124e-05,
      "model_forward_time": 3.1365857124328613,
      "step": 7709
    },
    {
      "epoch": 2.9407501220703124e-05,
      "step": 7709,
      "training_step_time": 9.905884265899658
    },
    {
      "epoch": 2.9407501220703124e-05,
      "model_forward_time": 3.1415135860443115,
      "step": 7709
    },
    {
      "epoch": 2.9407501220703124e-05,
      "step": 7709,
      "training_step_time": 10.016507625579834
    },
    {
      "epoch": 2.941131591796875e-05,
      "grad_norm": 0.35250189900398254,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 7710
    },
    {
      "epoch": 2.941131591796875e-05,
      "model_forward_time": 3.1533780097961426,
      "step": 7710
    },
    {
      "dynamics_loss_avg": 0.059195127338171005,
      "epoch": 2.941131591796875e-05,
      "step": 7710
    },
    {
      "action_loss_avg": 0.02298903316259384,
      "epoch": 2.941131591796875e-05,
      "step": 7710
    },
    {
      "epoch": 2.941131591796875e-05,
      "step": 7710,
      "training_step_time": 9.932406902313232
    },
    {
      "epoch": 2.941131591796875e-05,
      "model_forward_time": 3.139470100402832,
      "step": 7710
    },
    {
      "epoch": 2.941131591796875e-05,
      "step": 7710,
      "training_step_time": 10.194646120071411
    },
    {
      "epoch": 2.941131591796875e-05,
      "model_forward_time": 3.1384010314941406,
      "step": 7710
    },
    {
      "epoch": 2.941131591796875e-05,
      "step": 7710,
      "training_step_time": 9.913248777389526
    },
    {
      "epoch": 2.941131591796875e-05,
      "model_forward_time": 3.1417150497436523,
      "step": 7710
    },
    {
      "epoch": 2.941131591796875e-05,
      "step": 7710,
      "training_step_time": 10.325050830841064
    },
    {
      "epoch": 2.9415130615234376e-05,
      "model_forward_time": 3.161764144897461,
      "step": 7711
    },
    {
      "epoch": 2.9415130615234376e-05,
      "step": 7711,
      "training_step_time": 9.96603012084961
    },
    {
      "epoch": 2.9415130615234376e-05,
      "model_forward_time": 2.4434525966644287,
      "step": 7711
    },
    {
      "epoch": 2.9415130615234376e-05,
      "step": 7711,
      "training_step_time": 7.480008363723755
    },
    {
      "epoch": 2.9415130615234376e-05,
      "model_forward_time": 3.144117593765259,
      "step": 7711
    },
    {
      "epoch": 2.9415130615234376e-05,
      "step": 7711,
      "training_step_time": 10.238676309585571
    },
    {
      "epoch": 2.9415130615234376e-05,
      "model_forward_time": 3.1465635299682617,
      "step": 7711
    },
    {
      "epoch": 2.9415130615234376e-05,
      "step": 7711,
      "training_step_time": 10.241509437561035
    },
    {
      "epoch": 2.94189453125e-05,
      "model_forward_time": 3.1599819660186768,
      "step": 7712
    },
    {
      "epoch": 2.94189453125e-05,
      "step": 7712,
      "training_step_time": 9.953107833862305
    },
    {
      "epoch": 2.94189453125e-05,
      "model_forward_time": 3.141477346420288,
      "step": 7712
    },
    {
      "epoch": 2.94189453125e-05,
      "step": 7712,
      "training_step_time": 9.92077350616455
    },
    {
      "epoch": 2.94189453125e-05,
      "model_forward_time": 3.1597776412963867,
      "step": 7712
    },
    {
      "dynamics_loss_avg": 0.0668921923264861,
      "epoch": 2.94189453125e-05,
      "step": 7712
    },
    {
      "action_loss_avg": 0.027070954884402455,
      "epoch": 2.94189453125e-05,
      "step": 7712
    },
    {
      "epoch": 2.94189453125e-05,
      "step": 7712,
      "training_step_time": 9.947735786437988
    },
    {
      "epoch": 2.94189453125e-05,
      "model_forward_time": 3.4852209091186523,
      "step": 7712
    },
    {
      "epoch": 2.94189453125e-05,
      "step": 7712,
      "training_step_time": 10.359723806381226
    },
    {
      "epoch": 2.9422760009765625e-05,
      "model_forward_time": 3.1609854698181152,
      "step": 7713
    },
    {
      "epoch": 2.9422760009765625e-05,
      "step": 7713,
      "training_step_time": 9.950148105621338
    },
    {
      "epoch": 2.9422760009765625e-05,
      "model_forward_time": 3.148123264312744,
      "step": 7713
    },
    {
      "epoch": 2.9422760009765625e-05,
      "step": 7713,
      "training_step_time": 9.931839942932129
    },
    {
      "epoch": 2.9422760009765625e-05,
      "model_forward_time": 3.146958351135254,
      "step": 7713
    },
    {
      "epoch": 2.9422760009765625e-05,
      "step": 7713,
      "training_step_time": 9.92446780204773
    },
    {
      "epoch": 2.9422760009765625e-05,
      "model_forward_time": 3.141512393951416,
      "step": 7713
    },
    {
      "epoch": 2.9422760009765625e-05,
      "step": 7713,
      "training_step_time": 10.29304051399231
    },
    {
      "epoch": 2.942657470703125e-05,
      "model_forward_time": 3.147491693496704,
      "step": 7714
    },
    {
      "epoch": 2.942657470703125e-05,
      "step": 7714,
      "training_step_time": 10.252914667129517
    },
    {
      "epoch": 2.942657470703125e-05,
      "model_forward_time": 3.144014358520508,
      "step": 7714
    },
    {
      "epoch": 2.942657470703125e-05,
      "step": 7714,
      "training_step_time": 9.92190170288086
    },
    {
      "epoch": 2.942657470703125e-05,
      "model_forward_time": 3.145111083984375,
      "step": 7714
    },
    {
      "epoch": 2.942657470703125e-05,
      "step": 7714,
      "training_step_time": 9.92250680923462
    },
    {
      "epoch": 2.942657470703125e-05,
      "model_forward_time": 3.1387012004852295,
      "step": 7714
    },
    {
      "epoch": 2.942657470703125e-05,
      "step": 7714,
      "training_step_time": 10.015132665634155
    },
    {
      "epoch": 2.9430389404296874e-05,
      "model_forward_time": 3.1870999336242676,
      "step": 7715
    },
    {
      "dynamics_loss_avg": 0.055847573466598986,
      "epoch": 2.9430389404296874e-05,
      "step": 7715
    },
    {
      "action_loss_avg": 0.017533602192997932,
      "epoch": 2.9430389404296874e-05,
      "step": 7715
    },
    {
      "epoch": 2.9430389404296874e-05,
      "step": 7715,
      "training_step_time": 9.975618362426758
    },
    {
      "epoch": 2.9430389404296874e-05,
      "model_forward_time": 3.4443612098693848,
      "step": 7715
    },
    {
      "epoch": 2.9430389404296874e-05,
      "step": 7715,
      "training_step_time": 10.226752758026123
    },
    {
      "epoch": 2.9430389404296874e-05,
      "model_forward_time": 3.147571086883545,
      "step": 7715
    },
    {
      "epoch": 2.9430389404296874e-05,
      "step": 7715,
      "training_step_time": 9.930393695831299
    },
    {
      "epoch": 2.9430389404296874e-05,
      "model_forward_time": 3.1464922428131104,
      "step": 7715
    },
    {
      "epoch": 2.9430389404296874e-05,
      "step": 7715,
      "training_step_time": 10.281843423843384
    },
    {
      "epoch": 2.94342041015625e-05,
      "model_forward_time": 3.1547048091888428,
      "step": 7716
    },
    {
      "epoch": 2.94342041015625e-05,
      "step": 7716,
      "training_step_time": 9.93748927116394
    },
    {
      "epoch": 2.94342041015625e-05,
      "model_forward_time": 3.154660224914551,
      "step": 7716
    },
    {
      "epoch": 2.94342041015625e-05,
      "step": 7716,
      "training_step_time": 9.942437887191772
    },
    {
      "epoch": 2.94342041015625e-05,
      "model_forward_time": 3.452953577041626,
      "step": 7716
    },
    {
      "epoch": 2.94342041015625e-05,
      "step": 7716,
      "training_step_time": 10.235333681106567
    },
    {
      "epoch": 2.94342041015625e-05,
      "model_forward_time": 3.1518890857696533,
      "step": 7716
    },
    {
      "epoch": 2.94342041015625e-05,
      "step": 7716,
      "training_step_time": 10.29076623916626
    },
    {
      "epoch": 2.9438018798828124e-05,
      "model_forward_time": 3.1777970790863037,
      "step": 7717
    },
    {
      "epoch": 2.9438018798828124e-05,
      "step": 7717,
      "training_step_time": 9.966959953308105
    },
    {
      "epoch": 2.9438018798828124e-05,
      "model_forward_time": 3.1491613388061523,
      "step": 7717
    },
    {
      "epoch": 2.9438018798828124e-05,
      "step": 7717,
      "training_step_time": 9.935177564620972
    },
    {
      "epoch": 2.9438018798828124e-05,
      "model_forward_time": 3.1430506706237793,
      "step": 7717
    },
    {
      "dynamics_loss_avg": 0.05626314040273428,
      "epoch": 2.9438018798828124e-05,
      "step": 7717
    },
    {
      "action_loss_avg": 0.021334848226979376,
      "epoch": 2.9438018798828124e-05,
      "step": 7717
    },
    {
      "epoch": 2.9438018798828124e-05,
      "step": 7717,
      "training_step_time": 9.926444292068481
    },
    {
      "epoch": 2.9438018798828124e-05,
      "model_forward_time": 3.4387810230255127,
      "step": 7717
    },
    {
      "epoch": 2.9438018798828124e-05,
      "step": 7717,
      "training_step_time": 10.306479454040527
    },
    {
      "epoch": 2.944183349609375e-05,
      "model_forward_time": 3.1805953979492188,
      "step": 7718
    },
    {
      "epoch": 2.944183349609375e-05,
      "step": 7718,
      "training_step_time": 9.967579364776611
    },
    {
      "epoch": 2.944183349609375e-05,
      "model_forward_time": 3.146655559539795,
      "step": 7718
    },
    {
      "epoch": 2.944183349609375e-05,
      "step": 7718,
      "training_step_time": 9.923181533813477
    },
    {
      "epoch": 2.944183349609375e-05,
      "model_forward_time": 3.1411898136138916,
      "step": 7718
    },
    {
      "epoch": 2.944183349609375e-05,
      "step": 7718,
      "training_step_time": 9.919719457626343
    },
    {
      "epoch": 2.944183349609375e-05,
      "model_forward_time": 3.149907112121582,
      "step": 7718
    },
    {
      "epoch": 2.944183349609375e-05,
      "step": 7718,
      "training_step_time": 10.252280235290527
    },
    {
      "epoch": 2.9445648193359376e-05,
      "model_forward_time": 3.43778920173645,
      "step": 7719
    },
    {
      "epoch": 2.9445648193359376e-05,
      "step": 7719,
      "training_step_time": 10.218892097473145
    },
    {
      "epoch": 2.9445648193359376e-05,
      "model_forward_time": 3.146684408187866,
      "step": 7719
    },
    {
      "epoch": 2.9445648193359376e-05,
      "step": 7719,
      "training_step_time": 9.925172090530396
    },
    {
      "epoch": 2.9445648193359376e-05,
      "model_forward_time": 3.153125762939453,
      "step": 7719
    },
    {
      "epoch": 2.9445648193359376e-05,
      "step": 7719,
      "training_step_time": 9.932331323623657
    },
    {
      "epoch": 2.9445648193359376e-05,
      "model_forward_time": 3.1433074474334717,
      "step": 7719
    },
    {
      "epoch": 2.9445648193359376e-05,
      "step": 7719,
      "training_step_time": 10.068387985229492
    },
    {
      "epoch": 2.9449462890625e-05,
      "grad_norm": 0.43309658765792847,
      "learning_rate": 1e-05,
      "loss": 0.0835,
      "step": 7720
    },
    {
      "epoch": 2.9449462890625e-05,
      "model_forward_time": 3.156942367553711,
      "step": 7720
    },
    {
      "dynamics_loss_avg": 0.0641808751039207,
      "epoch": 2.9449462890625e-05,
      "step": 7720
    },
    {
      "action_loss_avg": 0.01878528050146997,
      "epoch": 2.9449462890625e-05,
      "step": 7720
    },
    {
      "epoch": 2.9449462890625e-05,
      "step": 7720,
      "training_step_time": 9.937263250350952
    },
    {
      "epoch": 2.9449462890625e-05,
      "model_forward_time": 3.1456243991851807,
      "step": 7720
    },
    {
      "epoch": 2.9449462890625e-05,
      "step": 7720,
      "training_step_time": 10.219637870788574
    },
    {
      "epoch": 2.9449462890625e-05,
      "model_forward_time": 3.1431941986083984,
      "step": 7720
    },
    {
      "epoch": 2.9449462890625e-05,
      "step": 7720,
      "training_step_time": 9.925273895263672
    },
    {
      "epoch": 2.9449462890625e-05,
      "model_forward_time": 3.151890277862549,
      "step": 7720
    },
    {
      "epoch": 2.9449462890625e-05,
      "step": 7720,
      "training_step_time": 10.256438493728638
    },
    {
      "epoch": 2.9453277587890625e-05,
      "model_forward_time": 3.1623756885528564,
      "step": 7721
    },
    {
      "epoch": 2.9453277587890625e-05,
      "step": 7721,
      "training_step_time": 9.945043563842773
    },
    {
      "epoch": 2.9453277587890625e-05,
      "model_forward_time": 3.1482748985290527,
      "step": 7721
    },
    {
      "epoch": 2.9453277587890625e-05,
      "step": 7721,
      "training_step_time": 9.92583417892456
    },
    {
      "epoch": 2.9453277587890625e-05,
      "model_forward_time": 3.3998355865478516,
      "step": 7721
    },
    {
      "epoch": 2.9453277587890625e-05,
      "step": 7721,
      "training_step_time": 10.180114030838013
    },
    {
      "epoch": 2.9453277587890625e-05,
      "model_forward_time": 3.1357219219207764,
      "step": 7721
    },
    {
      "epoch": 2.9453277587890625e-05,
      "step": 7721,
      "training_step_time": 10.284773588180542
    },
    {
      "epoch": 2.945709228515625e-05,
      "model_forward_time": 3.148310661315918,
      "step": 7722
    },
    {
      "epoch": 2.945709228515625e-05,
      "step": 7722,
      "training_step_time": 9.926404476165771
    },
    {
      "epoch": 2.945709228515625e-05,
      "model_forward_time": 3.1334261894226074,
      "step": 7722
    },
    {
      "epoch": 2.945709228515625e-05,
      "step": 7722,
      "training_step_time": 9.910711765289307
    },
    {
      "epoch": 2.945709228515625e-05,
      "model_forward_time": 3.1386642456054688,
      "step": 7722
    },
    {
      "dynamics_loss_avg": 0.05764185208827257,
      "epoch": 2.945709228515625e-05,
      "step": 7722
    },
    {
      "action_loss_avg": 0.02177452277392149,
      "epoch": 2.945709228515625e-05,
      "step": 7722
    },
    {
      "epoch": 2.945709228515625e-05,
      "step": 7722,
      "training_step_time": 9.920708179473877
    },
    {
      "epoch": 2.945709228515625e-05,
      "model_forward_time": 3.136685371398926,
      "step": 7722
    },
    {
      "epoch": 2.945709228515625e-05,
      "step": 7722,
      "training_step_time": 10.322871446609497
    },
    {
      "epoch": 2.9460906982421874e-05,
      "model_forward_time": 3.150845766067505,
      "step": 7723
    },
    {
      "epoch": 2.9460906982421874e-05,
      "step": 7723,
      "training_step_time": 9.928687334060669
    },
    {
      "epoch": 2.9460906982421874e-05,
      "model_forward_time": 3.1332058906555176,
      "step": 7723
    },
    {
      "epoch": 2.9460906982421874e-05,
      "step": 7723,
      "training_step_time": 9.91056776046753
    },
    {
      "epoch": 2.9460906982421874e-05,
      "model_forward_time": 3.1376473903656006,
      "step": 7723
    },
    {
      "epoch": 2.9460906982421874e-05,
      "step": 7723,
      "training_step_time": 9.917446613311768
    },
    {
      "epoch": 2.9460906982421874e-05,
      "model_forward_time": 3.148594617843628,
      "step": 7723
    },
    {
      "epoch": 2.9460906982421874e-05,
      "step": 7723,
      "training_step_time": 10.238842248916626
    },
    {
      "epoch": 2.94647216796875e-05,
      "model_forward_time": 3.4547855854034424,
      "step": 7724
    },
    {
      "epoch": 2.94647216796875e-05,
      "step": 7724,
      "training_step_time": 10.238122463226318
    },
    {
      "epoch": 2.94647216796875e-05,
      "model_forward_time": 3.1423516273498535,
      "step": 7724
    },
    {
      "epoch": 2.94647216796875e-05,
      "step": 7724,
      "training_step_time": 9.922533512115479
    },
    {
      "epoch": 2.94647216796875e-05,
      "model_forward_time": 3.138814926147461,
      "step": 7724
    },
    {
      "epoch": 2.94647216796875e-05,
      "step": 7724,
      "training_step_time": 9.917149305343628
    },
    {
      "epoch": 2.94647216796875e-05,
      "model_forward_time": 3.1370790004730225,
      "step": 7724
    },
    {
      "epoch": 2.94647216796875e-05,
      "step": 7724,
      "training_step_time": 10.321558237075806
    },
    {
      "epoch": 2.9468536376953124e-05,
      "model_forward_time": 3.162191867828369,
      "step": 7725
    },
    {
      "dynamics_loss_avg": 0.06129264086484909,
      "epoch": 2.9468536376953124e-05,
      "step": 7725
    },
    {
      "action_loss_avg": 0.019686334254220127,
      "epoch": 2.9468536376953124e-05,
      "step": 7725
    },
    {
      "epoch": 2.9468536376953124e-05,
      "step": 7725,
      "training_step_time": 9.950494527816772
    },
    {
      "epoch": 2.9468536376953124e-05,
      "model_forward_time": 3.1411662101745605,
      "step": 7725
    },
    {
      "epoch": 2.9468536376953124e-05,
      "step": 7725,
      "training_step_time": 10.21782112121582
    },
    {
      "epoch": 2.9468536376953124e-05,
      "model_forward_time": 3.1387462615966797,
      "step": 7725
    },
    {
      "epoch": 2.9468536376953124e-05,
      "step": 7725,
      "training_step_time": 9.911998510360718
    },
    {
      "epoch": 2.9468536376953124e-05,
      "model_forward_time": 3.1372411251068115,
      "step": 7725
    },
    {
      "epoch": 2.9468536376953124e-05,
      "step": 7725,
      "training_step_time": 10.265546798706055
    },
    {
      "epoch": 2.947235107421875e-05,
      "model_forward_time": 3.154902219772339,
      "step": 7726
    },
    {
      "epoch": 2.947235107421875e-05,
      "step": 7726,
      "training_step_time": 9.933234930038452
    },
    {
      "epoch": 2.947235107421875e-05,
      "model_forward_time": 3.1405351161956787,
      "step": 7726
    },
    {
      "epoch": 2.947235107421875e-05,
      "step": 7726,
      "training_step_time": 9.917173147201538
    },
    {
      "epoch": 2.947235107421875e-05,
      "model_forward_time": 3.4529011249542236,
      "step": 7726
    },
    {
      "epoch": 2.947235107421875e-05,
      "step": 7726,
      "training_step_time": 10.231448888778687
    },
    {
      "epoch": 2.947235107421875e-05,
      "model_forward_time": 3.1414384841918945,
      "step": 7726
    },
    {
      "epoch": 2.947235107421875e-05,
      "step": 7726,
      "training_step_time": 10.313152551651001
    },
    {
      "epoch": 2.9476165771484376e-05,
      "model_forward_time": 3.15090012550354,
      "step": 7727
    },
    {
      "epoch": 2.9476165771484376e-05,
      "step": 7727,
      "training_step_time": 9.928632020950317
    },
    {
      "epoch": 2.9476165771484376e-05,
      "model_forward_time": 3.1426279544830322,
      "step": 7727
    },
    {
      "epoch": 2.9476165771484376e-05,
      "step": 7727,
      "training_step_time": 9.91727900505066
    },
    {
      "epoch": 2.9476165771484376e-05,
      "model_forward_time": 3.1388282775878906,
      "step": 7727
    },
    {
      "dynamics_loss_avg": 0.056535684317350385,
      "epoch": 2.9476165771484376e-05,
      "step": 7727
    },
    {
      "action_loss_avg": 0.016604118328541518,
      "epoch": 2.9476165771484376e-05,
      "step": 7727
    },
    {
      "epoch": 2.9476165771484376e-05,
      "step": 7727,
      "training_step_time": 9.91611623764038
    },
    {
      "epoch": 2.9476165771484376e-05,
      "model_forward_time": 3.407435417175293,
      "step": 7727
    },
    {
      "epoch": 2.9476165771484376e-05,
      "step": 7727,
      "training_step_time": 10.273998022079468
    },
    {
      "epoch": 2.947998046875e-05,
      "model_forward_time": 3.157066583633423,
      "step": 7728
    },
    {
      "epoch": 2.947998046875e-05,
      "step": 7728,
      "training_step_time": 9.930565595626831
    },
    {
      "epoch": 2.947998046875e-05,
      "model_forward_time": 3.14694881439209,
      "step": 7728
    },
    {
      "epoch": 2.947998046875e-05,
      "step": 7728,
      "training_step_time": 9.917861461639404
    },
    {
      "epoch": 2.947998046875e-05,
      "model_forward_time": 3.1364524364471436,
      "step": 7728
    },
    {
      "epoch": 2.947998046875e-05,
      "step": 7728,
      "training_step_time": 9.911060333251953
    },
    {
      "epoch": 2.947998046875e-05,
      "model_forward_time": 3.1404645442962646,
      "step": 7728
    },
    {
      "epoch": 2.947998046875e-05,
      "step": 7728,
      "training_step_time": 10.291956663131714
    },
    {
      "epoch": 2.9483795166015625e-05,
      "model_forward_time": 3.466437339782715,
      "step": 7729
    },
    {
      "epoch": 2.9483795166015625e-05,
      "step": 7729,
      "training_step_time": 10.242263555526733
    },
    {
      "epoch": 2.9483795166015625e-05,
      "model_forward_time": 3.147988796234131,
      "step": 7729
    },
    {
      "epoch": 2.9483795166015625e-05,
      "step": 7729,
      "training_step_time": 9.92194414138794
    },
    {
      "epoch": 2.9483795166015625e-05,
      "model_forward_time": 3.1431496143341064,
      "step": 7729
    },
    {
      "epoch": 2.9483795166015625e-05,
      "step": 7729,
      "training_step_time": 9.920575857162476
    },
    {
      "epoch": 2.9483795166015625e-05,
      "model_forward_time": 3.1460466384887695,
      "step": 7729
    },
    {
      "epoch": 2.9483795166015625e-05,
      "step": 7729,
      "training_step_time": 10.018261432647705
    },
    {
      "epoch": 2.948760986328125e-05,
      "grad_norm": 0.5315247774124146,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 7730
    },
    {
      "epoch": 2.948760986328125e-05,
      "model_forward_time": 3.1607139110565186,
      "step": 7730
    },
    {
      "dynamics_loss_avg": 0.07141714282333851,
      "epoch": 2.948760986328125e-05,
      "step": 7730
    },
    {
      "action_loss_avg": 0.022047143895179033,
      "epoch": 2.948760986328125e-05,
      "step": 7730
    },
    {
      "epoch": 2.948760986328125e-05,
      "step": 7730,
      "training_step_time": 9.940441608428955
    },
    {
      "epoch": 2.948760986328125e-05,
      "model_forward_time": 3.4357330799102783,
      "step": 7730
    },
    {
      "epoch": 2.948760986328125e-05,
      "step": 7730,
      "training_step_time": 10.209131240844727
    },
    {
      "epoch": 2.948760986328125e-05,
      "model_forward_time": 3.149736166000366,
      "step": 7730
    },
    {
      "epoch": 2.948760986328125e-05,
      "step": 7730,
      "training_step_time": 9.925504684448242
    },
    {
      "epoch": 2.948760986328125e-05,
      "model_forward_time": 3.146559000015259,
      "step": 7730
    },
    {
      "epoch": 2.948760986328125e-05,
      "step": 7730,
      "training_step_time": 10.305692195892334
    },
    {
      "epoch": 2.9491424560546875e-05,
      "model_forward_time": 3.16593074798584,
      "step": 7731
    },
    {
      "epoch": 2.9491424560546875e-05,
      "step": 7731,
      "training_step_time": 9.945667028427124
    },
    {
      "epoch": 2.9491424560546875e-05,
      "model_forward_time": 3.1455435752868652,
      "step": 7731
    },
    {
      "epoch": 2.9491424560546875e-05,
      "step": 7731,
      "training_step_time": 9.924826860427856
    },
    {
      "epoch": 2.9491424560546875e-05,
      "model_forward_time": 3.1580028533935547,
      "step": 7731
    },
    {
      "epoch": 2.9491424560546875e-05,
      "step": 7731,
      "training_step_time": 10.266554355621338
    },
    {
      "epoch": 2.9491424560546875e-05,
      "model_forward_time": 3.147437572479248,
      "step": 7731
    },
    {
      "epoch": 2.9491424560546875e-05,
      "step": 7731,
      "training_step_time": 10.298963069915771
    },
    {
      "epoch": 2.94952392578125e-05,
      "model_forward_time": 3.14927077293396,
      "step": 7732
    },
    {
      "epoch": 2.94952392578125e-05,
      "step": 7732,
      "training_step_time": 9.926657438278198
    },
    {
      "epoch": 2.94952392578125e-05,
      "model_forward_time": 3.136378765106201,
      "step": 7732
    },
    {
      "epoch": 2.94952392578125e-05,
      "step": 7732,
      "training_step_time": 9.912033557891846
    },
    {
      "epoch": 2.94952392578125e-05,
      "model_forward_time": 3.142801284790039,
      "step": 7732
    },
    {
      "dynamics_loss_avg": 0.04695238210260868,
      "epoch": 2.94952392578125e-05,
      "step": 7732
    },
    {
      "action_loss_avg": 0.01522367070429027,
      "epoch": 2.94952392578125e-05,
      "step": 7732
    },
    {
      "epoch": 2.94952392578125e-05,
      "step": 7732,
      "training_step_time": 9.919928789138794
    },
    {
      "epoch": 2.94952392578125e-05,
      "model_forward_time": 3.433948278427124,
      "step": 7732
    },
    {
      "epoch": 2.94952392578125e-05,
      "step": 7732,
      "training_step_time": 10.305162906646729
    },
    {
      "epoch": 2.9499053955078124e-05,
      "model_forward_time": 3.1550893783569336,
      "step": 7733
    },
    {
      "epoch": 2.9499053955078124e-05,
      "step": 7733,
      "training_step_time": 9.931665897369385
    },
    {
      "epoch": 2.9499053955078124e-05,
      "model_forward_time": 3.1336448192596436,
      "step": 7733
    },
    {
      "epoch": 2.9499053955078124e-05,
      "step": 7733,
      "training_step_time": 9.90575385093689
    },
    {
      "epoch": 2.9499053955078124e-05,
      "model_forward_time": 3.1393299102783203,
      "step": 7733
    },
    {
      "epoch": 2.9499053955078124e-05,
      "step": 7733,
      "training_step_time": 9.91504168510437
    },
    {
      "epoch": 2.9499053955078124e-05,
      "model_forward_time": 3.1326425075531006,
      "step": 7733
    },
    {
      "epoch": 2.9499053955078124e-05,
      "step": 7733,
      "training_step_time": 10.292229413986206
    },
    {
      "epoch": 2.950286865234375e-05,
      "model_forward_time": 3.1428048610687256,
      "step": 7734
    },
    {
      "epoch": 2.950286865234375e-05,
      "step": 7734,
      "training_step_time": 10.206707000732422
    },
    {
      "epoch": 2.950286865234375e-05,
      "model_forward_time": 3.1362264156341553,
      "step": 7734
    },
    {
      "epoch": 2.950286865234375e-05,
      "step": 7734,
      "training_step_time": 9.915395021438599
    },
    {
      "epoch": 2.950286865234375e-05,
      "model_forward_time": 3.1377012729644775,
      "step": 7734
    },
    {
      "epoch": 2.950286865234375e-05,
      "step": 7734,
      "training_step_time": 9.915626287460327
    },
    {
      "epoch": 2.950286865234375e-05,
      "model_forward_time": 3.137728691101074,
      "step": 7734
    },
    {
      "epoch": 2.950286865234375e-05,
      "step": 7734,
      "training_step_time": 10.262823820114136
    },
    {
      "epoch": 2.9506683349609376e-05,
      "model_forward_time": 3.1536049842834473,
      "step": 7735
    },
    {
      "dynamics_loss_avg": 0.05865231342613697,
      "epoch": 2.9506683349609376e-05,
      "step": 7735
    },
    {
      "action_loss_avg": 0.02260049032047391,
      "epoch": 2.9506683349609376e-05,
      "step": 7735
    },
    {
      "epoch": 2.9506683349609376e-05,
      "step": 7735,
      "training_step_time": 9.948508501052856
    },
    {
      "epoch": 2.9506683349609376e-05,
      "model_forward_time": 3.4713175296783447,
      "step": 7735
    },
    {
      "epoch": 2.9506683349609376e-05,
      "step": 7735,
      "training_step_time": 10.251924514770508
    },
    {
      "epoch": 2.9506683349609376e-05,
      "model_forward_time": 3.1517176628112793,
      "step": 7735
    },
    {
      "epoch": 2.9506683349609376e-05,
      "step": 7735,
      "training_step_time": 9.934845447540283
    },
    {
      "epoch": 2.9506683349609376e-05,
      "model_forward_time": 3.1444668769836426,
      "step": 7735
    },
    {
      "epoch": 2.9506683349609376e-05,
      "step": 7735,
      "training_step_time": 10.326040983200073
    },
    {
      "epoch": 2.9510498046875e-05,
      "model_forward_time": 3.15480637550354,
      "step": 7736
    },
    {
      "epoch": 2.9510498046875e-05,
      "step": 7736,
      "training_step_time": 9.951199293136597
    },
    {
      "epoch": 2.9510498046875e-05,
      "model_forward_time": 3.144731044769287,
      "step": 7736
    },
    {
      "epoch": 2.9510498046875e-05,
      "step": 7736,
      "training_step_time": 9.922950029373169
    },
    {
      "epoch": 2.9510498046875e-05,
      "model_forward_time": 3.460461139678955,
      "step": 7736
    },
    {
      "epoch": 2.9510498046875e-05,
      "step": 7736,
      "training_step_time": 10.234513759613037
    },
    {
      "epoch": 2.9510498046875e-05,
      "model_forward_time": 3.1324548721313477,
      "step": 7736
    },
    {
      "epoch": 2.9510498046875e-05,
      "step": 7736,
      "training_step_time": 10.260884523391724
    },
    {
      "epoch": 2.9514312744140625e-05,
      "model_forward_time": 3.1533989906311035,
      "step": 7737
    },
    {
      "epoch": 2.9514312744140625e-05,
      "step": 7737,
      "training_step_time": 9.928248882293701
    },
    {
      "epoch": 2.9514312744140625e-05,
      "model_forward_time": 3.1386516094207764,
      "step": 7737
    },
    {
      "epoch": 2.9514312744140625e-05,
      "step": 7737,
      "training_step_time": 9.913840293884277
    },
    {
      "epoch": 2.9514312744140625e-05,
      "model_forward_time": 3.1451339721679688,
      "step": 7737
    },
    {
      "dynamics_loss_avg": 0.05888646915555,
      "epoch": 2.9514312744140625e-05,
      "step": 7737
    },
    {
      "action_loss_avg": 0.025173921138048172,
      "epoch": 2.9514312744140625e-05,
      "step": 7737
    },
    {
      "epoch": 2.9514312744140625e-05,
      "step": 7737,
      "training_step_time": 9.926282405853271
    },
    {
      "epoch": 2.9514312744140625e-05,
      "model_forward_time": 3.443819046020508,
      "step": 7737
    },
    {
      "epoch": 2.9514312744140625e-05,
      "step": 7737,
      "training_step_time": 10.310468196868896
    },
    {
      "epoch": 2.951812744140625e-05,
      "model_forward_time": 3.170309066772461,
      "step": 7738
    },
    {
      "epoch": 2.951812744140625e-05,
      "step": 7738,
      "training_step_time": 9.942185401916504
    },
    {
      "epoch": 2.951812744140625e-05,
      "model_forward_time": 3.1461544036865234,
      "step": 7738
    },
    {
      "epoch": 2.951812744140625e-05,
      "step": 7738,
      "training_step_time": 9.917977571487427
    },
    {
      "epoch": 2.951812744140625e-05,
      "model_forward_time": 3.1429061889648438,
      "step": 7738
    },
    {
      "epoch": 2.951812744140625e-05,
      "step": 7738,
      "training_step_time": 9.91507339477539
    },
    {
      "epoch": 2.951812744140625e-05,
      "model_forward_time": 3.1446659564971924,
      "step": 7738
    },
    {
      "epoch": 2.951812744140625e-05,
      "step": 7738,
      "training_step_time": 10.280311584472656
    },
    {
      "epoch": 2.9521942138671875e-05,
      "model_forward_time": 3.43428111076355,
      "step": 7739
    },
    {
      "epoch": 2.9521942138671875e-05,
      "step": 7739,
      "training_step_time": 10.208173274993896
    },
    {
      "epoch": 2.9521942138671875e-05,
      "model_forward_time": 2.4339966773986816,
      "step": 7739
    },
    {
      "epoch": 2.9521942138671875e-05,
      "step": 7739,
      "training_step_time": 7.455043077468872
    },
    {
      "epoch": 2.9521942138671875e-05,
      "model_forward_time": 3.127473831176758,
      "step": 7739
    },
    {
      "epoch": 2.9521942138671875e-05,
      "step": 7739,
      "training_step_time": 9.88089394569397
    },
    {
      "epoch": 2.9521942138671875e-05,
      "model_forward_time": 3.1322503089904785,
      "step": 7739
    },
    {
      "epoch": 2.9521942138671875e-05,
      "step": 7739,
      "training_step_time": 10.291185140609741
    },
    {
      "epoch": 2.95257568359375e-05,
      "grad_norm": 0.26879629492759705,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 7740
    },
    {
      "epoch": 2.95257568359375e-05,
      "model_forward_time": 3.1477084159851074,
      "step": 7740
    },
    {
      "dynamics_loss_avg": 0.062756722047925,
      "epoch": 2.95257568359375e-05,
      "step": 7740
    },
    {
      "action_loss_avg": 0.022671496216207742,
      "epoch": 2.95257568359375e-05,
      "step": 7740
    },
    {
      "epoch": 2.95257568359375e-05,
      "step": 7740,
      "training_step_time": 9.924654006958008
    },
    {
      "epoch": 2.95257568359375e-05,
      "model_forward_time": 3.1304662227630615,
      "step": 7740
    },
    {
      "epoch": 2.95257568359375e-05,
      "step": 7740,
      "training_step_time": 10.21779179573059
    },
    {
      "epoch": 2.95257568359375e-05,
      "model_forward_time": 3.134648084640503,
      "step": 7740
    },
    {
      "epoch": 2.95257568359375e-05,
      "step": 7740,
      "training_step_time": 9.903759479522705
    },
    {
      "epoch": 2.95257568359375e-05,
      "model_forward_time": 3.137599229812622,
      "step": 7740
    },
    {
      "epoch": 2.95257568359375e-05,
      "step": 7740,
      "training_step_time": 10.023841142654419
    },
    {
      "epoch": 2.9529571533203124e-05,
      "model_forward_time": 3.1473705768585205,
      "step": 7741
    },
    {
      "epoch": 2.9529571533203124e-05,
      "step": 7741,
      "training_step_time": 9.921830415725708
    },
    {
      "epoch": 2.9529571533203124e-05,
      "model_forward_time": 3.1385607719421387,
      "step": 7741
    },
    {
      "epoch": 2.9529571533203124e-05,
      "step": 7741,
      "training_step_time": 9.915802001953125
    },
    {
      "epoch": 2.9529571533203124e-05,
      "model_forward_time": 3.398719549179077,
      "step": 7741
    },
    {
      "epoch": 2.9529571533203124e-05,
      "step": 7741,
      "training_step_time": 10.177913904190063
    },
    {
      "epoch": 2.9529571533203124e-05,
      "model_forward_time": 3.151031970977783,
      "step": 7741
    },
    {
      "epoch": 2.9529571533203124e-05,
      "step": 7741,
      "training_step_time": 10.293396472930908
    },
    {
      "epoch": 2.953338623046875e-05,
      "model_forward_time": 3.151762008666992,
      "step": 7742
    },
    {
      "epoch": 2.953338623046875e-05,
      "step": 7742,
      "training_step_time": 9.92995548248291
    },
    {
      "epoch": 2.953338623046875e-05,
      "model_forward_time": 3.140650987625122,
      "step": 7742
    },
    {
      "epoch": 2.953338623046875e-05,
      "step": 7742,
      "training_step_time": 9.915207862854004
    },
    {
      "epoch": 2.953338623046875e-05,
      "model_forward_time": 3.1340742111206055,
      "step": 7742
    },
    {
      "dynamics_loss_avg": 0.054955686815083025,
      "epoch": 2.953338623046875e-05,
      "step": 7742
    },
    {
      "action_loss_avg": 0.025759372767060994,
      "epoch": 2.953338623046875e-05,
      "step": 7742
    },
    {
      "epoch": 2.953338623046875e-05,
      "step": 7742,
      "training_step_time": 9.909879922866821
    },
    {
      "epoch": 2.953338623046875e-05,
      "model_forward_time": 3.1360301971435547,
      "step": 7742
    },
    {
      "epoch": 2.953338623046875e-05,
      "step": 7742,
      "training_step_time": 10.331245183944702
    },
    {
      "epoch": 2.9537200927734376e-05,
      "model_forward_time": 3.148838520050049,
      "step": 7743
    },
    {
      "epoch": 2.9537200927734376e-05,
      "step": 7743,
      "training_step_time": 9.921340227127075
    },
    {
      "epoch": 2.9537200927734376e-05,
      "model_forward_time": 3.1342196464538574,
      "step": 7743
    },
    {
      "epoch": 2.9537200927734376e-05,
      "step": 7743,
      "training_step_time": 9.905671834945679
    },
    {
      "epoch": 2.9537200927734376e-05,
      "model_forward_time": 3.14103364944458,
      "step": 7743
    },
    {
      "epoch": 2.9537200927734376e-05,
      "step": 7743,
      "training_step_time": 9.91340708732605
    },
    {
      "epoch": 2.9537200927734376e-05,
      "model_forward_time": 3.1389670372009277,
      "step": 7743
    },
    {
      "epoch": 2.9537200927734376e-05,
      "step": 7743,
      "training_step_time": 10.298927068710327
    },
    {
      "epoch": 2.9541015625e-05,
      "model_forward_time": 3.459617853164673,
      "step": 7744
    },
    {
      "epoch": 2.9541015625e-05,
      "step": 7744,
      "training_step_time": 10.238198280334473
    },
    {
      "epoch": 2.9541015625e-05,
      "model_forward_time": 3.1386990547180176,
      "step": 7744
    },
    {
      "epoch": 2.9541015625e-05,
      "step": 7744,
      "training_step_time": 9.914309978485107
    },
    {
      "epoch": 2.9541015625e-05,
      "model_forward_time": 3.1444928646087646,
      "step": 7744
    },
    {
      "epoch": 2.9541015625e-05,
      "step": 7744,
      "training_step_time": 9.923221111297607
    },
    {
      "epoch": 2.9541015625e-05,
      "model_forward_time": 3.139444589614868,
      "step": 7744
    },
    {
      "epoch": 2.9541015625e-05,
      "step": 7744,
      "training_step_time": 10.29273772239685
    },
    {
      "epoch": 2.9544830322265625e-05,
      "model_forward_time": 3.1458539962768555,
      "step": 7745
    },
    {
      "dynamics_loss_avg": 0.054761518724262714,
      "epoch": 2.9544830322265625e-05,
      "step": 7745
    },
    {
      "action_loss_avg": 0.019072136469185354,
      "epoch": 2.9544830322265625e-05,
      "step": 7745
    },
    {
      "epoch": 2.9544830322265625e-05,
      "step": 7745,
      "training_step_time": 9.925722122192383
    },
    {
      "epoch": 2.9544830322265625e-05,
      "model_forward_time": 3.1395137310028076,
      "step": 7745
    },
    {
      "epoch": 2.9544830322265625e-05,
      "step": 7745,
      "training_step_time": 10.171792268753052
    },
    {
      "epoch": 2.9544830322265625e-05,
      "model_forward_time": 3.1365180015563965,
      "step": 7745
    },
    {
      "epoch": 2.9544830322265625e-05,
      "step": 7745,
      "training_step_time": 9.909762144088745
    },
    {
      "epoch": 2.9544830322265625e-05,
      "model_forward_time": 3.1339828968048096,
      "step": 7745
    },
    {
      "epoch": 2.9544830322265625e-05,
      "step": 7745,
      "training_step_time": 10.020691871643066
    },
    {
      "epoch": 2.954864501953125e-05,
      "model_forward_time": 3.142301082611084,
      "step": 7746
    },
    {
      "epoch": 2.954864501953125e-05,
      "step": 7746,
      "training_step_time": 9.923832893371582
    },
    {
      "epoch": 2.954864501953125e-05,
      "model_forward_time": 3.138791561126709,
      "step": 7746
    },
    {
      "epoch": 2.954864501953125e-05,
      "step": 7746,
      "training_step_time": 9.911263704299927
    },
    {
      "epoch": 2.954864501953125e-05,
      "model_forward_time": 3.41036057472229,
      "step": 7746
    },
    {
      "epoch": 2.954864501953125e-05,
      "step": 7746,
      "training_step_time": 10.179933786392212
    },
    {
      "epoch": 2.954864501953125e-05,
      "model_forward_time": 3.1352429389953613,
      "step": 7746
    },
    {
      "epoch": 2.954864501953125e-05,
      "step": 7746,
      "training_step_time": 10.31109356880188
    },
    {
      "epoch": 2.9552459716796875e-05,
      "model_forward_time": 3.1515369415283203,
      "step": 7747
    },
    {
      "epoch": 2.9552459716796875e-05,
      "step": 7747,
      "training_step_time": 9.924766063690186
    },
    {
      "epoch": 2.9552459716796875e-05,
      "model_forward_time": 3.132582902908325,
      "step": 7747
    },
    {
      "epoch": 2.9552459716796875e-05,
      "step": 7747,
      "training_step_time": 9.899119138717651
    },
    {
      "epoch": 2.9552459716796875e-05,
      "model_forward_time": 3.135082483291626,
      "step": 7747
    },
    {
      "dynamics_loss_avg": 0.06297941468656063,
      "epoch": 2.9552459716796875e-05,
      "step": 7747
    },
    {
      "action_loss_avg": 0.022513786051422357,
      "epoch": 2.9552459716796875e-05,
      "step": 7747
    },
    {
      "epoch": 2.9552459716796875e-05,
      "step": 7747,
      "training_step_time": 9.906639337539673
    },
    {
      "epoch": 2.9552459716796875e-05,
      "model_forward_time": 3.3909270763397217,
      "step": 7747
    },
    {
      "epoch": 2.9552459716796875e-05,
      "step": 7747,
      "training_step_time": 10.294631481170654
    },
    {
      "epoch": 2.95562744140625e-05,
      "model_forward_time": 3.138432264328003,
      "step": 7748
    },
    {
      "epoch": 2.95562744140625e-05,
      "step": 7748,
      "training_step_time": 9.904400825500488
    },
    {
      "epoch": 2.95562744140625e-05,
      "model_forward_time": 3.1330618858337402,
      "step": 7748
    },
    {
      "epoch": 2.95562744140625e-05,
      "step": 7748,
      "training_step_time": 9.898434162139893
    },
    {
      "epoch": 2.95562744140625e-05,
      "model_forward_time": 3.128317356109619,
      "step": 7748
    },
    {
      "epoch": 2.95562744140625e-05,
      "step": 7748,
      "training_step_time": 9.892199993133545
    },
    {
      "epoch": 2.95562744140625e-05,
      "model_forward_time": 3.130269765853882,
      "step": 7748
    },
    {
      "epoch": 2.95562744140625e-05,
      "step": 7748,
      "training_step_time": 10.002779006958008
    },
    {
      "epoch": 2.9560089111328124e-05,
      "model_forward_time": 3.4412739276885986,
      "step": 7749
    },
    {
      "epoch": 2.9560089111328124e-05,
      "step": 7749,
      "training_step_time": 10.207085609436035
    },
    {
      "epoch": 2.9560089111328124e-05,
      "model_forward_time": 3.1373071670532227,
      "step": 7749
    },
    {
      "epoch": 2.9560089111328124e-05,
      "step": 7749,
      "training_step_time": 9.90688419342041
    },
    {
      "epoch": 2.9560089111328124e-05,
      "model_forward_time": 3.1370668411254883,
      "step": 7749
    },
    {
      "epoch": 2.9560089111328124e-05,
      "step": 7749,
      "training_step_time": 9.908973693847656
    },
    {
      "epoch": 2.9560089111328124e-05,
      "model_forward_time": 3.1420867443084717,
      "step": 7749
    },
    {
      "epoch": 2.9560089111328124e-05,
      "step": 7749,
      "training_step_time": 10.286524295806885
    },
    {
      "epoch": 2.956390380859375e-05,
      "grad_norm": 0.42784327268600464,
      "learning_rate": 1e-05,
      "loss": 0.0817,
      "step": 7750
    },
    {
      "epoch": 2.956390380859375e-05,
      "model_forward_time": 3.1594314575195312,
      "step": 7750
    },
    {
      "dynamics_loss_avg": 0.04844478787854314,
      "epoch": 2.956390380859375e-05,
      "step": 7750
    },
    {
      "action_loss_avg": 0.01951543753966689,
      "epoch": 2.956390380859375e-05,
      "step": 7750
    },
    {
      "epoch": 2.956390380859375e-05,
      "step": 7750,
      "training_step_time": 9.9400315284729
    },
    {
      "epoch": 2.956390380859375e-05,
      "model_forward_time": 3.4669840335845947,
      "step": 7750
    },
    {
      "epoch": 2.956390380859375e-05,
      "step": 7750,
      "training_step_time": 10.240068674087524
    },
    {
      "epoch": 2.956390380859375e-05,
      "model_forward_time": 3.1388628482818604,
      "step": 7750
    },
    {
      "epoch": 2.956390380859375e-05,
      "step": 7750,
      "training_step_time": 9.914313077926636
    },
    {
      "epoch": 2.956390380859375e-05,
      "model_forward_time": 3.1426377296447754,
      "step": 7750
    },
    {
      "epoch": 2.956390380859375e-05,
      "step": 7750,
      "training_step_time": 10.290982007980347
    },
    {
      "epoch": 2.9567718505859376e-05,
      "model_forward_time": 3.1513493061065674,
      "step": 7751
    },
    {
      "epoch": 2.9567718505859376e-05,
      "step": 7751,
      "training_step_time": 9.924136400222778
    },
    {
      "epoch": 2.9567718505859376e-05,
      "model_forward_time": 3.134856700897217,
      "step": 7751
    },
    {
      "epoch": 2.9567718505859376e-05,
      "step": 7751,
      "training_step_time": 9.904187440872192
    },
    {
      "epoch": 2.9567718505859376e-05,
      "model_forward_time": 3.144367218017578,
      "step": 7751
    },
    {
      "epoch": 2.9567718505859376e-05,
      "step": 7751,
      "training_step_time": 10.205644607543945
    },
    {
      "epoch": 2.9567718505859376e-05,
      "model_forward_time": 3.140979290008545,
      "step": 7751
    },
    {
      "epoch": 2.9567718505859376e-05,
      "step": 7751,
      "training_step_time": 10.28961968421936
    },
    {
      "epoch": 2.9571533203125e-05,
      "model_forward_time": 3.150296211242676,
      "step": 7752
    },
    {
      "epoch": 2.9571533203125e-05,
      "step": 7752,
      "training_step_time": 9.919011116027832
    },
    {
      "epoch": 2.9571533203125e-05,
      "model_forward_time": 3.1350033283233643,
      "step": 7752
    },
    {
      "epoch": 2.9571533203125e-05,
      "step": 7752,
      "training_step_time": 9.900280714035034
    },
    {
      "epoch": 2.9571533203125e-05,
      "model_forward_time": 3.13806414604187,
      "step": 7752
    },
    {
      "dynamics_loss_avg": 0.05171375572681427,
      "epoch": 2.9571533203125e-05,
      "step": 7752
    },
    {
      "action_loss_avg": 0.014754295535385608,
      "epoch": 2.9571533203125e-05,
      "step": 7752
    },
    {
      "epoch": 2.9571533203125e-05,
      "step": 7752,
      "training_step_time": 9.90547227859497
    },
    {
      "epoch": 2.9571533203125e-05,
      "model_forward_time": 3.419924736022949,
      "step": 7752
    },
    {
      "epoch": 2.9571533203125e-05,
      "step": 7752,
      "training_step_time": 10.320633888244629
    },
    {
      "epoch": 2.9575347900390625e-05,
      "model_forward_time": 3.155839443206787,
      "step": 7753
    },
    {
      "epoch": 2.9575347900390625e-05,
      "step": 7753,
      "training_step_time": 9.930619716644287
    },
    {
      "epoch": 2.9575347900390625e-05,
      "model_forward_time": 3.1407630443573,
      "step": 7753
    },
    {
      "epoch": 2.9575347900390625e-05,
      "step": 7753,
      "training_step_time": 9.911850690841675
    },
    {
      "epoch": 2.9575347900390625e-05,
      "model_forward_time": 3.143688440322876,
      "step": 7753
    },
    {
      "epoch": 2.9575347900390625e-05,
      "step": 7753,
      "training_step_time": 9.917245864868164
    },
    {
      "epoch": 2.9575347900390625e-05,
      "model_forward_time": 3.142120838165283,
      "step": 7753
    },
    {
      "epoch": 2.9575347900390625e-05,
      "step": 7753,
      "training_step_time": 10.289143800735474
    },
    {
      "epoch": 2.957916259765625e-05,
      "model_forward_time": 3.1482226848602295,
      "step": 7754
    },
    {
      "epoch": 2.957916259765625e-05,
      "step": 7754,
      "training_step_time": 10.197059392929077
    },
    {
      "epoch": 2.957916259765625e-05,
      "model_forward_time": 3.138920783996582,
      "step": 7754
    },
    {
      "epoch": 2.957916259765625e-05,
      "step": 7754,
      "training_step_time": 9.916281461715698
    },
    {
      "epoch": 2.957916259765625e-05,
      "model_forward_time": 3.149780750274658,
      "step": 7754
    },
    {
      "epoch": 2.957916259765625e-05,
      "step": 7754,
      "training_step_time": 9.92978811264038
    },
    {
      "epoch": 2.957916259765625e-05,
      "model_forward_time": 2.44134521484375,
      "step": 7754
    },
    {
      "epoch": 2.957916259765625e-05,
      "step": 7754,
      "training_step_time": 10.252516508102417
    },
    {
      "epoch": 2.9582977294921875e-05,
      "model_forward_time": 3.1609365940093994,
      "step": 7755
    },
    {
      "dynamics_loss_avg": 0.06846152096986771,
      "epoch": 2.9582977294921875e-05,
      "step": 7755
    },
    {
      "action_loss_avg": 0.023479581787250937,
      "epoch": 2.9582977294921875e-05,
      "step": 7755
    },
    {
      "epoch": 2.9582977294921875e-05,
      "step": 7755,
      "training_step_time": 9.933260202407837
    },
    {
      "epoch": 2.9582977294921875e-05,
      "model_forward_time": 3.4048514366149902,
      "step": 7755
    },
    {
      "epoch": 2.9582977294921875e-05,
      "step": 7755,
      "training_step_time": 10.170008897781372
    },
    {
      "epoch": 2.9582977294921875e-05,
      "model_forward_time": 3.1426877975463867,
      "step": 7755
    },
    {
      "epoch": 2.9582977294921875e-05,
      "step": 7755,
      "training_step_time": 9.907063722610474
    },
    {
      "epoch": 2.9582977294921875e-05,
      "model_forward_time": 3.1390562057495117,
      "step": 7755
    },
    {
      "epoch": 2.9582977294921875e-05,
      "step": 7755,
      "training_step_time": 10.325231552124023
    },
    {
      "epoch": 2.95867919921875e-05,
      "model_forward_time": 3.145714521408081,
      "step": 7756
    },
    {
      "epoch": 2.95867919921875e-05,
      "step": 7756,
      "training_step_time": 9.91755723953247
    },
    {
      "epoch": 2.95867919921875e-05,
      "model_forward_time": 3.140580177307129,
      "step": 7756
    },
    {
      "epoch": 2.95867919921875e-05,
      "step": 7756,
      "training_step_time": 9.910389423370361
    },
    {
      "epoch": 2.95867919921875e-05,
      "model_forward_time": 3.138348340988159,
      "step": 7756
    },
    {
      "epoch": 2.95867919921875e-05,
      "step": 7756,
      "training_step_time": 10.170498609542847
    },
    {
      "epoch": 2.95867919921875e-05,
      "model_forward_time": 3.138073444366455,
      "step": 7756
    },
    {
      "epoch": 2.95867919921875e-05,
      "step": 7756,
      "training_step_time": 10.310137271881104
    },
    {
      "epoch": 2.9590606689453124e-05,
      "model_forward_time": 2.447892665863037,
      "step": 7757
    },
    {
      "epoch": 2.9590606689453124e-05,
      "step": 7757,
      "training_step_time": 7.477977275848389
    },
    {
      "epoch": 2.9590606689453124e-05,
      "model_forward_time": 3.133925199508667,
      "step": 7757
    },
    {
      "epoch": 2.9590606689453124e-05,
      "step": 7757,
      "training_step_time": 9.901206731796265
    },
    {
      "epoch": 2.9590606689453124e-05,
      "model_forward_time": 3.131532907485962,
      "step": 7757
    },
    {
      "dynamics_loss_avg": 0.04817546084523201,
      "epoch": 2.9590606689453124e-05,
      "step": 7757
    },
    {
      "action_loss_avg": 0.016791147319599985,
      "epoch": 2.9590606689453124e-05,
      "step": 7757
    },
    {
      "epoch": 2.9590606689453124e-05,
      "step": 7757,
      "training_step_time": 9.899795532226562
    },
    {
      "epoch": 2.9590606689453124e-05,
      "model_forward_time": 3.390812397003174,
      "step": 7757
    },
    {
      "epoch": 2.9590606689453124e-05,
      "step": 7757,
      "training_step_time": 10.304591178894043
    },
    {
      "epoch": 2.959442138671875e-05,
      "model_forward_time": 3.1503372192382812,
      "step": 7758
    },
    {
      "epoch": 2.959442138671875e-05,
      "step": 7758,
      "training_step_time": 9.921285152435303
    },
    {
      "epoch": 2.959442138671875e-05,
      "model_forward_time": 3.1337881088256836,
      "step": 7758
    },
    {
      "epoch": 2.959442138671875e-05,
      "step": 7758,
      "training_step_time": 9.903832912445068
    },
    {
      "epoch": 2.959442138671875e-05,
      "model_forward_time": 3.1329503059387207,
      "step": 7758
    },
    {
      "epoch": 2.959442138671875e-05,
      "step": 7758,
      "training_step_time": 9.908541679382324
    },
    {
      "epoch": 2.959442138671875e-05,
      "model_forward_time": 3.135432243347168,
      "step": 7758
    },
    {
      "epoch": 2.959442138671875e-05,
      "step": 7758,
      "training_step_time": 10.346315383911133
    },
    {
      "epoch": 2.9598236083984376e-05,
      "model_forward_time": 3.398273468017578,
      "step": 7759
    },
    {
      "epoch": 2.9598236083984376e-05,
      "step": 7759,
      "training_step_time": 10.169926881790161
    },
    {
      "epoch": 2.9598236083984376e-05,
      "model_forward_time": 3.1316637992858887,
      "step": 7759
    },
    {
      "epoch": 2.9598236083984376e-05,
      "step": 7759,
      "training_step_time": 9.899648666381836
    },
    {
      "epoch": 2.9598236083984376e-05,
      "model_forward_time": 3.13171648979187,
      "step": 7759
    },
    {
      "epoch": 2.9598236083984376e-05,
      "step": 7759,
      "training_step_time": 9.899056434631348
    },
    {
      "epoch": 2.9598236083984376e-05,
      "model_forward_time": 3.1359143257141113,
      "step": 7759
    },
    {
      "epoch": 2.9598236083984376e-05,
      "step": 7759,
      "training_step_time": 10.27727723121643
    },
    {
      "epoch": 2.960205078125e-05,
      "grad_norm": 0.3430546820163727,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 7760
    },
    {
      "epoch": 2.960205078125e-05,
      "model_forward_time": 3.152249574661255,
      "step": 7760
    },
    {
      "dynamics_loss_avg": 0.056196306832134724,
      "epoch": 2.960205078125e-05,
      "step": 7760
    },
    {
      "action_loss_avg": 0.017920656129717827,
      "epoch": 2.960205078125e-05,
      "step": 7760
    },
    {
      "epoch": 2.960205078125e-05,
      "step": 7760,
      "training_step_time": 9.929192543029785
    },
    {
      "epoch": 2.960205078125e-05,
      "model_forward_time": 3.1290128231048584,
      "step": 7760
    },
    {
      "epoch": 2.960205078125e-05,
      "step": 7760,
      "training_step_time": 10.18221664428711
    },
    {
      "epoch": 2.960205078125e-05,
      "model_forward_time": 3.136110544204712,
      "step": 7760
    },
    {
      "epoch": 2.960205078125e-05,
      "step": 7760,
      "training_step_time": 9.9112548828125
    },
    {
      "epoch": 2.960205078125e-05,
      "model_forward_time": 3.1402111053466797,
      "step": 7760
    },
    {
      "epoch": 2.960205078125e-05,
      "step": 7760,
      "training_step_time": 10.028923988342285
    },
    {
      "epoch": 2.9605865478515625e-05,
      "model_forward_time": 3.162731409072876,
      "step": 7761
    },
    {
      "epoch": 2.9605865478515625e-05,
      "step": 7761,
      "training_step_time": 9.942813396453857
    },
    {
      "epoch": 2.9605865478515625e-05,
      "model_forward_time": 3.1424319744110107,
      "step": 7761
    },
    {
      "epoch": 2.9605865478515625e-05,
      "step": 7761,
      "training_step_time": 9.921125411987305
    },
    {
      "epoch": 2.9605865478515625e-05,
      "model_forward_time": 3.1496365070343018,
      "step": 7761
    },
    {
      "epoch": 2.9605865478515625e-05,
      "step": 7761,
      "training_step_time": 10.228753805160522
    },
    {
      "epoch": 2.9605865478515625e-05,
      "model_forward_time": 3.150981903076172,
      "step": 7761
    },
    {
      "epoch": 2.9605865478515625e-05,
      "step": 7761,
      "training_step_time": 10.046433448791504
    },
    {
      "epoch": 2.960968017578125e-05,
      "model_forward_time": 3.163595676422119,
      "step": 7762
    },
    {
      "epoch": 2.960968017578125e-05,
      "step": 7762,
      "training_step_time": 9.94978642463684
    },
    {
      "epoch": 2.960968017578125e-05,
      "model_forward_time": 3.1448614597320557,
      "step": 7762
    },
    {
      "epoch": 2.960968017578125e-05,
      "step": 7762,
      "training_step_time": 9.927348136901855
    },
    {
      "epoch": 2.960968017578125e-05,
      "model_forward_time": 3.150073528289795,
      "step": 7762
    },
    {
      "dynamics_loss_avg": 0.04795456137508154,
      "epoch": 2.960968017578125e-05,
      "step": 7762
    },
    {
      "action_loss_avg": 0.015514146955683827,
      "epoch": 2.960968017578125e-05,
      "step": 7762
    },
    {
      "epoch": 2.960968017578125e-05,
      "step": 7762,
      "training_step_time": 9.934026002883911
    },
    {
      "epoch": 2.960968017578125e-05,
      "model_forward_time": 3.1472909450531006,
      "step": 7762
    },
    {
      "epoch": 2.960968017578125e-05,
      "step": 7762,
      "training_step_time": 10.302307367324829
    },
    {
      "epoch": 2.9613494873046875e-05,
      "model_forward_time": 3.1640636920928955,
      "step": 7763
    },
    {
      "epoch": 2.9613494873046875e-05,
      "step": 7763,
      "training_step_time": 9.946458578109741
    },
    {
      "epoch": 2.9613494873046875e-05,
      "model_forward_time": 3.1479156017303467,
      "step": 7763
    },
    {
      "epoch": 2.9613494873046875e-05,
      "step": 7763,
      "training_step_time": 9.927536487579346
    },
    {
      "epoch": 2.9613494873046875e-05,
      "model_forward_time": 3.147848129272461,
      "step": 7763
    },
    {
      "epoch": 2.9613494873046875e-05,
      "step": 7763,
      "training_step_time": 9.921804428100586
    },
    {
      "epoch": 2.9613494873046875e-05,
      "model_forward_time": 3.140393018722534,
      "step": 7763
    },
    {
      "epoch": 2.9613494873046875e-05,
      "step": 7763,
      "training_step_time": 10.321142435073853
    },
    {
      "epoch": 2.96173095703125e-05,
      "model_forward_time": 3.446629524230957,
      "step": 7764
    },
    {
      "epoch": 2.96173095703125e-05,
      "step": 7764,
      "training_step_time": 10.221080303192139
    },
    {
      "epoch": 2.96173095703125e-05,
      "model_forward_time": 3.1385843753814697,
      "step": 7764
    },
    {
      "epoch": 2.96173095703125e-05,
      "step": 7764,
      "training_step_time": 9.912214756011963
    },
    {
      "epoch": 2.96173095703125e-05,
      "model_forward_time": 3.1432623863220215,
      "step": 7764
    },
    {
      "epoch": 2.96173095703125e-05,
      "step": 7764,
      "training_step_time": 9.917093992233276
    },
    {
      "epoch": 2.96173095703125e-05,
      "model_forward_time": 2.436004400253296,
      "step": 7764
    },
    {
      "epoch": 2.96173095703125e-05,
      "step": 7764,
      "training_step_time": 10.295831680297852
    },
    {
      "epoch": 2.9621124267578124e-05,
      "model_forward_time": 3.1429617404937744,
      "step": 7765
    },
    {
      "dynamics_loss_avg": 0.05382230430841446,
      "epoch": 2.9621124267578124e-05,
      "step": 7765
    },
    {
      "action_loss_avg": 0.01642629215493798,
      "epoch": 2.9621124267578124e-05,
      "step": 7765
    },
    {
      "epoch": 2.9621124267578124e-05,
      "step": 7765,
      "training_step_time": 9.913902759552002
    },
    {
      "epoch": 2.9621124267578124e-05,
      "model_forward_time": 3.140080213546753,
      "step": 7765
    },
    {
      "epoch": 2.9621124267578124e-05,
      "step": 7765,
      "training_step_time": 9.913400173187256
    },
    {
      "epoch": 2.9621124267578124e-05,
      "model_forward_time": 3.411799669265747,
      "step": 7765
    },
    {
      "epoch": 2.9621124267578124e-05,
      "step": 7765,
      "training_step_time": 10.183846712112427
    },
    {
      "epoch": 2.9621124267578124e-05,
      "model_forward_time": 3.1423447132110596,
      "step": 7765
    },
    {
      "epoch": 2.9621124267578124e-05,
      "step": 7765,
      "training_step_time": 10.3150475025177
    },
    {
      "epoch": 2.962493896484375e-05,
      "model_forward_time": 3.1517155170440674,
      "step": 7766
    },
    {
      "epoch": 2.962493896484375e-05,
      "step": 7766,
      "training_step_time": 9.92623257637024
    },
    {
      "epoch": 2.962493896484375e-05,
      "model_forward_time": 3.1400158405303955,
      "step": 7766
    },
    {
      "epoch": 2.962493896484375e-05,
      "step": 7766,
      "training_step_time": 9.917040824890137
    },
    {
      "epoch": 2.962493896484375e-05,
      "model_forward_time": 3.4310030937194824,
      "step": 7766
    },
    {
      "epoch": 2.962493896484375e-05,
      "step": 7766,
      "training_step_time": 10.204756498336792
    },
    {
      "epoch": 2.962493896484375e-05,
      "model_forward_time": 3.1438965797424316,
      "step": 7766
    },
    {
      "epoch": 2.962493896484375e-05,
      "step": 7766,
      "training_step_time": 10.044010162353516
    },
    {
      "epoch": 2.9628753662109376e-05,
      "model_forward_time": 3.152069330215454,
      "step": 7767
    },
    {
      "epoch": 2.9628753662109376e-05,
      "step": 7767,
      "training_step_time": 9.935942888259888
    },
    {
      "epoch": 2.9628753662109376e-05,
      "model_forward_time": 3.1516270637512207,
      "step": 7767
    },
    {
      "epoch": 2.9628753662109376e-05,
      "step": 7767,
      "training_step_time": 9.930095911026001
    },
    {
      "epoch": 2.9628753662109376e-05,
      "model_forward_time": 3.1467742919921875,
      "step": 7767
    },
    {
      "dynamics_loss_avg": 0.049372858460992576,
      "epoch": 2.9628753662109376e-05,
      "step": 7767
    },
    {
      "action_loss_avg": 0.016876195650547744,
      "epoch": 2.9628753662109376e-05,
      "step": 7767
    },
    {
      "epoch": 2.9628753662109376e-05,
      "step": 7767,
      "training_step_time": 9.928558111190796
    },
    {
      "epoch": 2.9628753662109376e-05,
      "model_forward_time": 3.1496737003326416,
      "step": 7767
    },
    {
      "epoch": 2.9628753662109376e-05,
      "step": 7767,
      "training_step_time": 10.276100158691406
    },
    {
      "epoch": 2.9632568359375e-05,
      "model_forward_time": 3.4351468086242676,
      "step": 7768
    },
    {
      "epoch": 2.9632568359375e-05,
      "step": 7768,
      "training_step_time": 10.215410470962524
    },
    {
      "epoch": 2.9632568359375e-05,
      "model_forward_time": 3.1472887992858887,
      "step": 7768
    },
    {
      "epoch": 2.9632568359375e-05,
      "step": 7768,
      "training_step_time": 9.925075769424438
    },
    {
      "epoch": 2.9632568359375e-05,
      "model_forward_time": 3.1441638469696045,
      "step": 7768
    },
    {
      "epoch": 2.9632568359375e-05,
      "step": 7768,
      "training_step_time": 9.914580821990967
    },
    {
      "epoch": 2.9632568359375e-05,
      "model_forward_time": 3.1437690258026123,
      "step": 7768
    },
    {
      "epoch": 2.9632568359375e-05,
      "step": 7768,
      "training_step_time": 10.269352197647095
    },
    {
      "epoch": 2.9636383056640625e-05,
      "model_forward_time": 3.4506027698516846,
      "step": 7769
    },
    {
      "epoch": 2.9636383056640625e-05,
      "step": 7769,
      "training_step_time": 10.227198362350464
    },
    {
      "epoch": 2.9636383056640625e-05,
      "model_forward_time": 2.4350790977478027,
      "step": 7769
    },
    {
      "epoch": 2.9636383056640625e-05,
      "step": 7769,
      "training_step_time": 7.459392786026001
    },
    {
      "epoch": 2.9636383056640625e-05,
      "model_forward_time": 3.1339523792266846,
      "step": 7769
    },
    {
      "epoch": 2.9636383056640625e-05,
      "step": 7769,
      "training_step_time": 9.903249263763428
    },
    {
      "epoch": 2.9636383056640625e-05,
      "model_forward_time": 3.139737367630005,
      "step": 7769
    },
    {
      "epoch": 2.9636383056640625e-05,
      "step": 7769,
      "training_step_time": 10.324046850204468
    },
    {
      "epoch": 2.964019775390625e-05,
      "grad_norm": 0.39814603328704834,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 7770
    },
    {
      "epoch": 2.964019775390625e-05,
      "model_forward_time": 3.1569697856903076,
      "step": 7770
    },
    {
      "dynamics_loss_avg": 0.0617991279810667,
      "epoch": 2.964019775390625e-05,
      "step": 7770
    },
    {
      "action_loss_avg": 0.021991757676005363,
      "epoch": 2.964019775390625e-05,
      "step": 7770
    },
    {
      "epoch": 2.964019775390625e-05,
      "step": 7770,
      "training_step_time": 9.930068254470825
    },
    {
      "epoch": 2.964019775390625e-05,
      "model_forward_time": 3.1349565982818604,
      "step": 7770
    },
    {
      "epoch": 2.964019775390625e-05,
      "step": 7770,
      "training_step_time": 9.900991201400757
    },
    {
      "epoch": 2.964019775390625e-05,
      "model_forward_time": 3.388702154159546,
      "step": 7770
    },
    {
      "epoch": 2.964019775390625e-05,
      "step": 7770,
      "training_step_time": 10.155715703964233
    },
    {
      "epoch": 2.964019775390625e-05,
      "model_forward_time": 3.1443300247192383,
      "step": 7770
    },
    {
      "epoch": 2.964019775390625e-05,
      "step": 7770,
      "training_step_time": 10.269687175750732
    },
    {
      "epoch": 2.9644012451171875e-05,
      "model_forward_time": 3.1618242263793945,
      "step": 7771
    },
    {
      "epoch": 2.9644012451171875e-05,
      "step": 7771,
      "training_step_time": 9.931338787078857
    },
    {
      "epoch": 2.9644012451171875e-05,
      "model_forward_time": 3.137174606323242,
      "step": 7771
    },
    {
      "epoch": 2.9644012451171875e-05,
      "step": 7771,
      "training_step_time": 9.905157327651978
    },
    {
      "epoch": 2.9644012451171875e-05,
      "model_forward_time": 3.1399083137512207,
      "step": 7771
    },
    {
      "epoch": 2.9644012451171875e-05,
      "step": 7771,
      "training_step_time": 10.191717147827148
    },
    {
      "epoch": 2.9644012451171875e-05,
      "model_forward_time": 3.1398913860321045,
      "step": 7771
    },
    {
      "epoch": 2.9644012451171875e-05,
      "step": 7771,
      "training_step_time": 10.014890909194946
    },
    {
      "epoch": 2.96478271484375e-05,
      "model_forward_time": 3.1573102474212646,
      "step": 7772
    },
    {
      "epoch": 2.96478271484375e-05,
      "step": 7772,
      "training_step_time": 9.929779291152954
    },
    {
      "epoch": 2.96478271484375e-05,
      "model_forward_time": 3.1386845111846924,
      "step": 7772
    },
    {
      "epoch": 2.96478271484375e-05,
      "step": 7772,
      "training_step_time": 9.913947820663452
    },
    {
      "epoch": 2.96478271484375e-05,
      "model_forward_time": 3.1416923999786377,
      "step": 7772
    },
    {
      "dynamics_loss_avg": 0.05140738412737846,
      "epoch": 2.96478271484375e-05,
      "step": 7772
    },
    {
      "action_loss_avg": 0.014982938300818205,
      "epoch": 2.96478271484375e-05,
      "step": 7772
    },
    {
      "epoch": 2.96478271484375e-05,
      "step": 7772,
      "training_step_time": 9.919276714324951
    },
    {
      "epoch": 2.96478271484375e-05,
      "model_forward_time": 3.432377338409424,
      "step": 7772
    },
    {
      "epoch": 2.96478271484375e-05,
      "step": 7772,
      "training_step_time": 10.299684762954712
    },
    {
      "epoch": 2.9651641845703124e-05,
      "model_forward_time": 3.156331777572632,
      "step": 7773
    },
    {
      "epoch": 2.9651641845703124e-05,
      "step": 7773,
      "training_step_time": 9.92677640914917
    },
    {
      "epoch": 2.9651641845703124e-05,
      "model_forward_time": 3.1314568519592285,
      "step": 7773
    },
    {
      "epoch": 2.9651641845703124e-05,
      "step": 7773,
      "training_step_time": 9.902349472045898
    },
    {
      "epoch": 2.9651641845703124e-05,
      "model_forward_time": 2.4299099445343018,
      "step": 7773
    },
    {
      "epoch": 2.9651641845703124e-05,
      "step": 7773,
      "training_step_time": 7.454939842224121
    },
    {
      "epoch": 2.9651641845703124e-05,
      "model_forward_time": 3.1280770301818848,
      "step": 7773
    },
    {
      "epoch": 2.9651641845703124e-05,
      "step": 7773,
      "training_step_time": 10.247442245483398
    },
    {
      "epoch": 2.965545654296875e-05,
      "model_forward_time": 3.14143967628479,
      "step": 7774
    },
    {
      "epoch": 2.965545654296875e-05,
      "step": 7774,
      "training_step_time": 9.918579816818237
    },
    {
      "epoch": 2.965545654296875e-05,
      "model_forward_time": 3.3752024173736572,
      "step": 7774
    },
    {
      "epoch": 2.965545654296875e-05,
      "step": 7774,
      "training_step_time": 10.145416021347046
    },
    {
      "epoch": 2.965545654296875e-05,
      "model_forward_time": 3.138145923614502,
      "step": 7774
    },
    {
      "epoch": 2.965545654296875e-05,
      "step": 7774,
      "training_step_time": 9.912134647369385
    },
    {
      "epoch": 2.965545654296875e-05,
      "model_forward_time": 2.4325523376464844,
      "step": 7774
    },
    {
      "epoch": 2.965545654296875e-05,
      "step": 7774,
      "training_step_time": 10.302508354187012
    },
    {
      "epoch": 2.9659271240234376e-05,
      "model_forward_time": 3.1459686756134033,
      "step": 7775
    },
    {
      "dynamics_loss_avg": 0.06168174408376217,
      "epoch": 2.9659271240234376e-05,
      "step": 7775
    },
    {
      "action_loss_avg": 0.01916032861918211,
      "epoch": 2.9659271240234376e-05,
      "step": 7775
    },
    {
      "epoch": 2.9659271240234376e-05,
      "step": 7775,
      "training_step_time": 9.924292802810669
    },
    {
      "epoch": 2.9659271240234376e-05,
      "model_forward_time": 3.388665199279785,
      "step": 7775
    },
    {
      "epoch": 2.9659271240234376e-05,
      "step": 7775,
      "training_step_time": 10.164567708969116
    },
    {
      "epoch": 2.9659271240234376e-05,
      "model_forward_time": 3.1326091289520264,
      "step": 7775
    },
    {
      "epoch": 2.9659271240234376e-05,
      "step": 7775,
      "training_step_time": 9.90690541267395
    },
    {
      "epoch": 2.9659271240234376e-05,
      "model_forward_time": 2.4337215423583984,
      "step": 7775
    },
    {
      "epoch": 2.9659271240234376e-05,
      "step": 7775,
      "training_step_time": 10.271071910858154
    },
    {
      "epoch": 2.96630859375e-05,
      "model_forward_time": 3.139803171157837,
      "step": 7776
    },
    {
      "epoch": 2.96630859375e-05,
      "step": 7776,
      "training_step_time": 9.913531064987183
    },
    {
      "epoch": 2.96630859375e-05,
      "model_forward_time": 3.133375883102417,
      "step": 7776
    },
    {
      "epoch": 2.96630859375e-05,
      "step": 7776,
      "training_step_time": 9.90696382522583
    },
    {
      "epoch": 2.96630859375e-05,
      "model_forward_time": 3.1356167793273926,
      "step": 7776
    },
    {
      "epoch": 2.96630859375e-05,
      "step": 7776,
      "training_step_time": 9.909261226654053
    },
    {
      "epoch": 2.96630859375e-05,
      "model_forward_time": 3.380586862564087,
      "step": 7776
    },
    {
      "epoch": 2.96630859375e-05,
      "step": 7776,
      "training_step_time": 10.278221845626831
    },
    {
      "epoch": 2.9666900634765626e-05,
      "model_forward_time": 3.1528873443603516,
      "step": 7777
    },
    {
      "epoch": 2.9666900634765626e-05,
      "step": 7777,
      "training_step_time": 9.93008804321289
    },
    {
      "epoch": 2.9666900634765626e-05,
      "model_forward_time": 3.132272481918335,
      "step": 7777
    },
    {
      "epoch": 2.9666900634765626e-05,
      "step": 7777,
      "training_step_time": 9.901564836502075
    },
    {
      "epoch": 2.9666900634765626e-05,
      "model_forward_time": 3.135176181793213,
      "step": 7777
    },
    {
      "dynamics_loss_avg": 0.0636667188256979,
      "epoch": 2.9666900634765626e-05,
      "step": 7777
    },
    {
      "action_loss_avg": 0.02365031950175762,
      "epoch": 2.9666900634765626e-05,
      "step": 7777
    },
    {
      "epoch": 2.9666900634765626e-05,
      "step": 7777,
      "training_step_time": 9.905322074890137
    },
    {
      "epoch": 2.9666900634765626e-05,
      "model_forward_time": 3.4009225368499756,
      "step": 7777
    },
    {
      "epoch": 2.9666900634765626e-05,
      "step": 7777,
      "training_step_time": 10.264408826828003
    },
    {
      "epoch": 2.967071533203125e-05,
      "model_forward_time": 3.1486077308654785,
      "step": 7778
    },
    {
      "epoch": 2.967071533203125e-05,
      "step": 7778,
      "training_step_time": 9.923827886581421
    },
    {
      "epoch": 2.967071533203125e-05,
      "model_forward_time": 3.141716480255127,
      "step": 7778
    },
    {
      "epoch": 2.967071533203125e-05,
      "step": 7778,
      "training_step_time": 9.912165880203247
    },
    {
      "epoch": 2.967071533203125e-05,
      "model_forward_time": 3.137204647064209,
      "step": 7778
    },
    {
      "epoch": 2.967071533203125e-05,
      "step": 7778,
      "training_step_time": 9.909709453582764
    },
    {
      "epoch": 2.967071533203125e-05,
      "model_forward_time": 3.1406333446502686,
      "step": 7778
    },
    {
      "epoch": 2.967071533203125e-05,
      "step": 7778,
      "training_step_time": 10.264859437942505
    },
    {
      "epoch": 2.9674530029296875e-05,
      "model_forward_time": 3.1444520950317383,
      "step": 7779
    },
    {
      "epoch": 2.9674530029296875e-05,
      "step": 7779,
      "training_step_time": 10.18447995185852
    },
    {
      "epoch": 2.9674530029296875e-05,
      "model_forward_time": 3.137246608734131,
      "step": 7779
    },
    {
      "epoch": 2.9674530029296875e-05,
      "step": 7779,
      "training_step_time": 9.912516355514526
    },
    {
      "epoch": 2.9674530029296875e-05,
      "model_forward_time": 3.133373975753784,
      "step": 7779
    },
    {
      "epoch": 2.9674530029296875e-05,
      "step": 7779,
      "training_step_time": 9.912882089614868
    },
    {
      "epoch": 2.9674530029296875e-05,
      "model_forward_time": 3.1397206783294678,
      "step": 7779
    },
    {
      "epoch": 2.9674530029296875e-05,
      "step": 7779,
      "training_step_time": 10.308968305587769
    },
    {
      "epoch": 2.96783447265625e-05,
      "grad_norm": 0.30618658661842346,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 7780
    },
    {
      "epoch": 2.96783447265625e-05,
      "model_forward_time": 3.1526074409484863,
      "step": 7780
    },
    {
      "dynamics_loss_avg": 0.06231446582823992,
      "epoch": 2.96783447265625e-05,
      "step": 7780
    },
    {
      "action_loss_avg": 0.017720518074929713,
      "epoch": 2.96783447265625e-05,
      "step": 7780
    },
    {
      "epoch": 2.96783447265625e-05,
      "step": 7780,
      "training_step_time": 9.93281626701355
    },
    {
      "epoch": 2.96783447265625e-05,
      "model_forward_time": 2.4324872493743896,
      "step": 7780
    },
    {
      "epoch": 2.96783447265625e-05,
      "step": 7780,
      "training_step_time": 7.750709533691406
    },
    {
      "epoch": 2.96783447265625e-05,
      "model_forward_time": 3.134046792984009,
      "step": 7780
    },
    {
      "epoch": 2.96783447265625e-05,
      "step": 7780,
      "training_step_time": 9.907581806182861
    },
    {
      "epoch": 2.96783447265625e-05,
      "model_forward_time": 3.138273000717163,
      "step": 7780
    },
    {
      "epoch": 2.96783447265625e-05,
      "step": 7780,
      "training_step_time": 10.328906774520874
    },
    {
      "epoch": 2.9682159423828124e-05,
      "model_forward_time": 3.1503398418426514,
      "step": 7781
    },
    {
      "epoch": 2.9682159423828124e-05,
      "step": 7781,
      "training_step_time": 9.925728797912598
    },
    {
      "epoch": 2.9682159423828124e-05,
      "model_forward_time": 3.1343328952789307,
      "step": 7781
    },
    {
      "epoch": 2.9682159423828124e-05,
      "step": 7781,
      "training_step_time": 9.908913373947144
    },
    {
      "epoch": 2.9682159423828124e-05,
      "model_forward_time": 3.4146969318389893,
      "step": 7781
    },
    {
      "epoch": 2.9682159423828124e-05,
      "step": 7781,
      "training_step_time": 10.18898344039917
    },
    {
      "epoch": 2.9682159423828124e-05,
      "model_forward_time": 2.4341750144958496,
      "step": 7781
    },
    {
      "epoch": 2.9682159423828124e-05,
      "step": 7781,
      "training_step_time": 10.344940185546875
    },
    {
      "epoch": 2.968597412109375e-05,
      "model_forward_time": 3.152184009552002,
      "step": 7782
    },
    {
      "epoch": 2.968597412109375e-05,
      "step": 7782,
      "training_step_time": 9.927385807037354
    },
    {
      "epoch": 2.968597412109375e-05,
      "model_forward_time": 3.133833885192871,
      "step": 7782
    },
    {
      "epoch": 2.968597412109375e-05,
      "step": 7782,
      "training_step_time": 9.907875299453735
    },
    {
      "epoch": 2.968597412109375e-05,
      "model_forward_time": 3.1402459144592285,
      "step": 7782
    },
    {
      "dynamics_loss_avg": 0.05801473231986165,
      "epoch": 2.968597412109375e-05,
      "step": 7782
    },
    {
      "action_loss_avg": 0.019471197703387587,
      "epoch": 2.968597412109375e-05,
      "step": 7782
    },
    {
      "epoch": 2.968597412109375e-05,
      "step": 7782,
      "training_step_time": 9.917864322662354
    },
    {
      "epoch": 2.968597412109375e-05,
      "model_forward_time": 3.1392385959625244,
      "step": 7782
    },
    {
      "epoch": 2.968597412109375e-05,
      "step": 7782,
      "training_step_time": 10.273695468902588
    },
    {
      "epoch": 2.9689788818359376e-05,
      "model_forward_time": 3.1577606201171875,
      "step": 7783
    },
    {
      "epoch": 2.9689788818359376e-05,
      "step": 7783,
      "training_step_time": 9.935212850570679
    },
    {
      "epoch": 2.9689788818359376e-05,
      "model_forward_time": 3.137928009033203,
      "step": 7783
    },
    {
      "epoch": 2.9689788818359376e-05,
      "step": 7783,
      "training_step_time": 9.915040254592896
    },
    {
      "epoch": 2.9689788818359376e-05,
      "model_forward_time": 3.138385534286499,
      "step": 7783
    },
    {
      "epoch": 2.9689788818359376e-05,
      "step": 7783,
      "training_step_time": 9.917057514190674
    },
    {
      "epoch": 2.9689788818359376e-05,
      "model_forward_time": 3.136671543121338,
      "step": 7783
    },
    {
      "epoch": 2.9689788818359376e-05,
      "step": 7783,
      "training_step_time": 10.019998788833618
    },
    {
      "epoch": 2.9693603515625e-05,
      "model_forward_time": 3.4189138412475586,
      "step": 7784
    },
    {
      "epoch": 2.9693603515625e-05,
      "step": 7784,
      "training_step_time": 10.193142175674438
    },
    {
      "epoch": 2.9693603515625e-05,
      "model_forward_time": 3.1358163356781006,
      "step": 7784
    },
    {
      "epoch": 2.9693603515625e-05,
      "step": 7784,
      "training_step_time": 9.906529903411865
    },
    {
      "epoch": 2.9693603515625e-05,
      "model_forward_time": 3.1341629028320312,
      "step": 7784
    },
    {
      "epoch": 2.9693603515625e-05,
      "step": 7784,
      "training_step_time": 9.90754246711731
    },
    {
      "epoch": 2.9693603515625e-05,
      "model_forward_time": 2.4333789348602295,
      "step": 7784
    },
    {
      "epoch": 2.9693603515625e-05,
      "step": 7784,
      "training_step_time": 10.285449504852295
    },
    {
      "epoch": 2.9697418212890626e-05,
      "model_forward_time": 2.439528703689575,
      "step": 7785
    },
    {
      "dynamics_loss_avg": 0.050767576042562726,
      "epoch": 2.9697418212890626e-05,
      "step": 7785
    },
    {
      "action_loss_avg": 0.017458157567307352,
      "epoch": 2.9697418212890626e-05,
      "step": 7785
    },
    {
      "epoch": 2.9697418212890626e-05,
      "step": 7785,
      "training_step_time": 7.477728605270386
    },
    {
      "epoch": 2.9697418212890626e-05,
      "model_forward_time": 3.134854316711426,
      "step": 7785
    },
    {
      "epoch": 2.9697418212890626e-05,
      "step": 7785,
      "training_step_time": 9.905177116394043
    },
    {
      "epoch": 2.9697418212890626e-05,
      "model_forward_time": 3.415379047393799,
      "step": 7785
    },
    {
      "epoch": 2.9697418212890626e-05,
      "step": 7785,
      "training_step_time": 10.1853187084198
    },
    {
      "epoch": 2.9697418212890626e-05,
      "model_forward_time": 3.1375010013580322,
      "step": 7785
    },
    {
      "epoch": 2.9697418212890626e-05,
      "step": 7785,
      "training_step_time": 10.32203722000122
    },
    {
      "epoch": 2.970123291015625e-05,
      "model_forward_time": 3.155639886856079,
      "step": 7786
    },
    {
      "epoch": 2.970123291015625e-05,
      "step": 7786,
      "training_step_time": 9.930034160614014
    },
    {
      "epoch": 2.970123291015625e-05,
      "model_forward_time": 3.13641095161438,
      "step": 7786
    },
    {
      "epoch": 2.970123291015625e-05,
      "step": 7786,
      "training_step_time": 9.90566897392273
    },
    {
      "epoch": 2.970123291015625e-05,
      "model_forward_time": 3.4144489765167236,
      "step": 7786
    },
    {
      "epoch": 2.970123291015625e-05,
      "step": 7786,
      "training_step_time": 10.190885543823242
    },
    {
      "epoch": 2.970123291015625e-05,
      "model_forward_time": 3.1423492431640625,
      "step": 7786
    },
    {
      "epoch": 2.970123291015625e-05,
      "step": 7786,
      "training_step_time": 10.321880578994751
    },
    {
      "epoch": 2.9705047607421875e-05,
      "model_forward_time": 3.1594669818878174,
      "step": 7787
    },
    {
      "epoch": 2.9705047607421875e-05,
      "step": 7787,
      "training_step_time": 9.935909986495972
    },
    {
      "epoch": 2.9705047607421875e-05,
      "model_forward_time": 3.1420586109161377,
      "step": 7787
    },
    {
      "epoch": 2.9705047607421875e-05,
      "step": 7787,
      "training_step_time": 9.920855283737183
    },
    {
      "epoch": 2.9705047607421875e-05,
      "model_forward_time": 3.1377501487731934,
      "step": 7787
    },
    {
      "dynamics_loss_avg": 0.051012674905359746,
      "epoch": 2.9705047607421875e-05,
      "step": 7787
    },
    {
      "action_loss_avg": 0.016993461921811105,
      "epoch": 2.9705047607421875e-05,
      "step": 7787
    },
    {
      "epoch": 2.9705047607421875e-05,
      "step": 7787,
      "training_step_time": 9.915403842926025
    },
    {
      "epoch": 2.9705047607421875e-05,
      "model_forward_time": 3.1423566341400146,
      "step": 7787
    },
    {
      "epoch": 2.9705047607421875e-05,
      "step": 7787,
      "training_step_time": 10.031873226165771
    },
    {
      "epoch": 2.97088623046875e-05,
      "model_forward_time": 3.4529995918273926,
      "step": 7788
    },
    {
      "epoch": 2.97088623046875e-05,
      "step": 7788,
      "training_step_time": 10.22916293144226
    },
    {
      "epoch": 2.97088623046875e-05,
      "model_forward_time": 3.137709856033325,
      "step": 7788
    },
    {
      "epoch": 2.97088623046875e-05,
      "step": 7788,
      "training_step_time": 9.908905029296875
    },
    {
      "epoch": 2.97088623046875e-05,
      "model_forward_time": 3.13276743888855,
      "step": 7788
    },
    {
      "epoch": 2.97088623046875e-05,
      "step": 7788,
      "training_step_time": 9.900052547454834
    },
    {
      "epoch": 2.97088623046875e-05,
      "model_forward_time": 3.133796453475952,
      "step": 7788
    },
    {
      "epoch": 2.97088623046875e-05,
      "step": 7788,
      "training_step_time": 10.017335653305054
    },
    {
      "epoch": 2.9712677001953124e-05,
      "model_forward_time": 3.405461311340332,
      "step": 7789
    },
    {
      "epoch": 2.9712677001953124e-05,
      "step": 7789,
      "training_step_time": 10.171603441238403
    },
    {
      "epoch": 2.9712677001953124e-05,
      "model_forward_time": 3.1371700763702393,
      "step": 7789
    },
    {
      "epoch": 2.9712677001953124e-05,
      "step": 7789,
      "training_step_time": 9.908722877502441
    },
    {
      "epoch": 2.9712677001953124e-05,
      "model_forward_time": 3.1362924575805664,
      "step": 7789
    },
    {
      "epoch": 2.9712677001953124e-05,
      "step": 7789,
      "training_step_time": 9.90758466720581
    },
    {
      "epoch": 2.9712677001953124e-05,
      "model_forward_time": 3.137763261795044,
      "step": 7789
    },
    {
      "epoch": 2.9712677001953124e-05,
      "step": 7789,
      "training_step_time": 10.286141872406006
    },
    {
      "epoch": 2.971649169921875e-05,
      "grad_norm": 0.2449178546667099,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 7790
    },
    {
      "epoch": 2.971649169921875e-05,
      "model_forward_time": 2.448096513748169,
      "step": 7790
    },
    {
      "dynamics_loss_avg": 0.06786197982728481,
      "epoch": 2.971649169921875e-05,
      "step": 7790
    },
    {
      "action_loss_avg": 0.02148897238075733,
      "epoch": 2.971649169921875e-05,
      "step": 7790
    },
    {
      "epoch": 2.971649169921875e-05,
      "step": 7790,
      "training_step_time": 7.486642599105835
    },
    {
      "epoch": 2.971649169921875e-05,
      "model_forward_time": 3.1315863132476807,
      "step": 7790
    },
    {
      "epoch": 2.971649169921875e-05,
      "step": 7790,
      "training_step_time": 9.90145754814148
    },
    {
      "epoch": 2.971649169921875e-05,
      "model_forward_time": 3.382354736328125,
      "step": 7790
    },
    {
      "epoch": 2.971649169921875e-05,
      "step": 7790,
      "training_step_time": 10.153047800064087
    },
    {
      "epoch": 2.971649169921875e-05,
      "model_forward_time": 3.136752128601074,
      "step": 7790
    },
    {
      "epoch": 2.971649169921875e-05,
      "step": 7790,
      "training_step_time": 10.28840446472168
    },
    {
      "epoch": 2.9720306396484376e-05,
      "model_forward_time": 3.15101957321167,
      "step": 7791
    },
    {
      "epoch": 2.9720306396484376e-05,
      "step": 7791,
      "training_step_time": 9.925519704818726
    },
    {
      "epoch": 2.9720306396484376e-05,
      "model_forward_time": 3.132309913635254,
      "step": 7791
    },
    {
      "epoch": 2.9720306396484376e-05,
      "step": 7791,
      "training_step_time": 9.90361213684082
    },
    {
      "epoch": 2.9720306396484376e-05,
      "model_forward_time": 3.138451099395752,
      "step": 7791
    },
    {
      "epoch": 2.9720306396484376e-05,
      "step": 7791,
      "training_step_time": 10.226791143417358
    },
    {
      "epoch": 2.9720306396484376e-05,
      "model_forward_time": 3.136627435684204,
      "step": 7791
    },
    {
      "epoch": 2.9720306396484376e-05,
      "step": 7791,
      "training_step_time": 10.32711410522461
    },
    {
      "epoch": 2.972412109375e-05,
      "model_forward_time": 3.1446316242218018,
      "step": 7792
    },
    {
      "epoch": 2.972412109375e-05,
      "step": 7792,
      "training_step_time": 9.9197359085083
    },
    {
      "epoch": 2.972412109375e-05,
      "model_forward_time": 3.13328218460083,
      "step": 7792
    },
    {
      "epoch": 2.972412109375e-05,
      "step": 7792,
      "training_step_time": 9.905959367752075
    },
    {
      "epoch": 2.972412109375e-05,
      "model_forward_time": 3.1333110332489014,
      "step": 7792
    },
    {
      "dynamics_loss_avg": 0.05344853959977627,
      "epoch": 2.972412109375e-05,
      "step": 7792
    },
    {
      "action_loss_avg": 0.015539551945403218,
      "epoch": 2.972412109375e-05,
      "step": 7792
    },
    {
      "epoch": 2.972412109375e-05,
      "step": 7792,
      "training_step_time": 9.90758466720581
    },
    {
      "epoch": 2.972412109375e-05,
      "model_forward_time": 3.4631049633026123,
      "step": 7792
    },
    {
      "epoch": 2.972412109375e-05,
      "step": 7792,
      "training_step_time": 10.346943616867065
    },
    {
      "epoch": 2.9727935791015626e-05,
      "model_forward_time": 3.1517741680145264,
      "step": 7793
    },
    {
      "epoch": 2.9727935791015626e-05,
      "step": 7793,
      "training_step_time": 9.921380281448364
    },
    {
      "epoch": 2.9727935791015626e-05,
      "model_forward_time": 3.130364179611206,
      "step": 7793
    },
    {
      "epoch": 2.9727935791015626e-05,
      "step": 7793,
      "training_step_time": 9.89554500579834
    },
    {
      "epoch": 2.9727935791015626e-05,
      "model_forward_time": 3.1323907375335693,
      "step": 7793
    },
    {
      "epoch": 2.9727935791015626e-05,
      "step": 7793,
      "training_step_time": 9.902962923049927
    },
    {
      "epoch": 2.9727935791015626e-05,
      "model_forward_time": 3.130553722381592,
      "step": 7793
    },
    {
      "epoch": 2.9727935791015626e-05,
      "step": 7793,
      "training_step_time": 10.027379512786865
    },
    {
      "epoch": 2.973175048828125e-05,
      "model_forward_time": 3.142728090286255,
      "step": 7794
    },
    {
      "epoch": 2.973175048828125e-05,
      "step": 7794,
      "training_step_time": 10.199084758758545
    },
    {
      "epoch": 2.973175048828125e-05,
      "model_forward_time": 3.1335363388061523,
      "step": 7794
    },
    {
      "epoch": 2.973175048828125e-05,
      "step": 7794,
      "training_step_time": 9.903799772262573
    },
    {
      "epoch": 2.973175048828125e-05,
      "model_forward_time": 3.1366636753082275,
      "step": 7794
    },
    {
      "epoch": 2.973175048828125e-05,
      "step": 7794,
      "training_step_time": 9.912529230117798
    },
    {
      "epoch": 2.973175048828125e-05,
      "model_forward_time": 3.1377320289611816,
      "step": 7794
    },
    {
      "epoch": 2.973175048828125e-05,
      "step": 7794,
      "training_step_time": 10.286304950714111
    },
    {
      "epoch": 2.9735565185546875e-05,
      "model_forward_time": 3.1447575092315674,
      "step": 7795
    },
    {
      "dynamics_loss_avg": 0.0643728144466877,
      "epoch": 2.9735565185546875e-05,
      "step": 7795
    },
    {
      "action_loss_avg": 0.02133756661787629,
      "epoch": 2.9735565185546875e-05,
      "step": 7795
    },
    {
      "epoch": 2.9735565185546875e-05,
      "step": 7795,
      "training_step_time": 9.927811622619629
    },
    {
      "epoch": 2.9735565185546875e-05,
      "model_forward_time": 2.7147293090820312,
      "step": 7795
    },
    {
      "epoch": 2.9735565185546875e-05,
      "step": 7795,
      "training_step_time": 7.743201494216919
    },
    {
      "epoch": 2.9735565185546875e-05,
      "model_forward_time": 3.1322848796844482,
      "step": 7795
    },
    {
      "epoch": 2.9735565185546875e-05,
      "step": 7795,
      "training_step_time": 9.903690338134766
    },
    {
      "epoch": 2.9735565185546875e-05,
      "model_forward_time": 3.132507562637329,
      "step": 7795
    },
    {
      "epoch": 2.9735565185546875e-05,
      "step": 7795,
      "training_step_time": 10.351029396057129
    },
    {
      "epoch": 2.97393798828125e-05,
      "model_forward_time": 3.1401636600494385,
      "step": 7796
    },
    {
      "epoch": 2.97393798828125e-05,
      "step": 7796,
      "training_step_time": 9.912675142288208
    },
    {
      "epoch": 2.97393798828125e-05,
      "model_forward_time": 3.1338138580322266,
      "step": 7796
    },
    {
      "epoch": 2.97393798828125e-05,
      "step": 7796,
      "training_step_time": 9.899408340454102
    },
    {
      "epoch": 2.97393798828125e-05,
      "model_forward_time": 3.133867025375366,
      "step": 7796
    },
    {
      "epoch": 2.97393798828125e-05,
      "step": 7796,
      "training_step_time": 10.209487915039062
    },
    {
      "epoch": 2.97393798828125e-05,
      "model_forward_time": 3.1350502967834473,
      "step": 7796
    },
    {
      "epoch": 2.97393798828125e-05,
      "step": 7796,
      "training_step_time": 10.303735494613647
    },
    {
      "epoch": 2.9743194580078124e-05,
      "model_forward_time": 2.4438912868499756,
      "step": 7797
    },
    {
      "epoch": 2.9743194580078124e-05,
      "step": 7797,
      "training_step_time": 7.47158145904541
    },
    {
      "epoch": 2.9743194580078124e-05,
      "model_forward_time": 3.132049560546875,
      "step": 7797
    },
    {
      "epoch": 2.9743194580078124e-05,
      "step": 7797,
      "training_step_time": 9.901105642318726
    },
    {
      "epoch": 2.9743194580078124e-05,
      "model_forward_time": 3.1348259449005127,
      "step": 7797
    },
    {
      "dynamics_loss_avg": 0.06258273757994175,
      "epoch": 2.9743194580078124e-05,
      "step": 7797
    },
    {
      "action_loss_avg": 0.019895030395127834,
      "epoch": 2.9743194580078124e-05,
      "step": 7797
    },
    {
      "epoch": 2.9743194580078124e-05,
      "step": 7797,
      "training_step_time": 9.910382270812988
    },
    {
      "epoch": 2.9743194580078124e-05,
      "model_forward_time": 3.4011523723602295,
      "step": 7797
    },
    {
      "epoch": 2.9743194580078124e-05,
      "step": 7797,
      "training_step_time": 10.273474216461182
    },
    {
      "epoch": 2.974700927734375e-05,
      "model_forward_time": 3.1549251079559326,
      "step": 7798
    },
    {
      "epoch": 2.974700927734375e-05,
      "step": 7798,
      "training_step_time": 9.933035612106323
    },
    {
      "epoch": 2.974700927734375e-05,
      "model_forward_time": 3.1403684616088867,
      "step": 7798
    },
    {
      "epoch": 2.974700927734375e-05,
      "step": 7798,
      "training_step_time": 9.917557716369629
    },
    {
      "epoch": 2.974700927734375e-05,
      "model_forward_time": 3.1353015899658203,
      "step": 7798
    },
    {
      "epoch": 2.974700927734375e-05,
      "step": 7798,
      "training_step_time": 9.906316995620728
    },
    {
      "epoch": 2.974700927734375e-05,
      "model_forward_time": 3.13912034034729,
      "step": 7798
    },
    {
      "epoch": 2.974700927734375e-05,
      "step": 7798,
      "training_step_time": 10.028114795684814
    },
    {
      "epoch": 2.9750823974609376e-05,
      "model_forward_time": 3.151805877685547,
      "step": 7799
    },
    {
      "epoch": 2.9750823974609376e-05,
      "step": 7799,
      "training_step_time": 10.193934440612793
    },
    {
      "epoch": 2.9750823974609376e-05,
      "model_forward_time": 3.134596347808838,
      "step": 7799
    },
    {
      "epoch": 2.9750823974609376e-05,
      "step": 7799,
      "training_step_time": 9.909288167953491
    },
    {
      "epoch": 2.9750823974609376e-05,
      "model_forward_time": 2.4292712211608887,
      "step": 7799
    },
    {
      "epoch": 2.9750823974609376e-05,
      "step": 7799,
      "training_step_time": 7.457743167877197
    },
    {
      "epoch": 2.9750823974609376e-05,
      "model_forward_time": 3.136080265045166,
      "step": 7799
    },
    {
      "epoch": 2.9750823974609376e-05,
      "step": 7799,
      "training_step_time": 10.267419338226318
    },
    {
      "epoch": 2.9754638671875e-05,
      "grad_norm": 0.2980998754501343,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 7800
    },
    {
      "epoch": 2.9754638671875e-05,
      "model_forward_time": 3.1531195640563965,
      "step": 7800
    },
    {
      "dynamics_loss_avg": 0.06675461884588003,
      "epoch": 2.9754638671875e-05,
      "step": 7800
    },
    {
      "action_loss_avg": 0.01939341672696173,
      "epoch": 2.9754638671875e-05,
      "step": 7800
    },
    {
      "epoch": 2.9754638671875e-05,
      "step": 7800,
      "training_step_time": 9.926347255706787
    },
    {
      "epoch": 2.9754638671875e-05,
      "model_forward_time": 3.134035110473633,
      "step": 7800
    },
    {
      "epoch": 2.9754638671875e-05,
      "step": 7800,
      "training_step_time": 10.221120357513428
    },
    {
      "epoch": 2.9754638671875e-05,
      "model_forward_time": 3.1346819400787354,
      "step": 7800
    },
    {
      "epoch": 2.9754638671875e-05,
      "step": 7800,
      "training_step_time": 9.911687135696411
    },
    {
      "epoch": 2.9754638671875e-05,
      "model_forward_time": 3.1397430896759033,
      "step": 7800
    },
    {
      "epoch": 2.9754638671875e-05,
      "step": 7800,
      "training_step_time": 10.263339757919312
    },
    {
      "epoch": 2.9758453369140626e-05,
      "model_forward_time": 3.159407377243042,
      "step": 7801
    },
    {
      "epoch": 2.9758453369140626e-05,
      "step": 7801,
      "training_step_time": 9.933384895324707
    },
    {
      "epoch": 2.9758453369140626e-05,
      "model_forward_time": 3.134594440460205,
      "step": 7801
    },
    {
      "epoch": 2.9758453369140626e-05,
      "step": 7801,
      "training_step_time": 9.907660245895386
    },
    {
      "epoch": 2.9758453369140626e-05,
      "model_forward_time": 2.4364922046661377,
      "step": 7801
    },
    {
      "epoch": 2.9758453369140626e-05,
      "step": 7801,
      "training_step_time": 7.466236591339111
    },
    {
      "epoch": 2.9758453369140626e-05,
      "model_forward_time": 3.4137117862701416,
      "step": 7801
    },
    {
      "epoch": 2.9758453369140626e-05,
      "step": 7801,
      "training_step_time": 10.28925347328186
    },
    {
      "epoch": 2.976226806640625e-05,
      "model_forward_time": 3.151718854904175,
      "step": 7802
    },
    {
      "epoch": 2.976226806640625e-05,
      "step": 7802,
      "training_step_time": 9.930088520050049
    },
    {
      "epoch": 2.976226806640625e-05,
      "model_forward_time": 3.1357057094573975,
      "step": 7802
    },
    {
      "epoch": 2.976226806640625e-05,
      "step": 7802,
      "training_step_time": 9.914788246154785
    },
    {
      "epoch": 2.976226806640625e-05,
      "model_forward_time": 3.1346418857574463,
      "step": 7802
    },
    {
      "dynamics_loss_avg": 0.0680418049916625,
      "epoch": 2.976226806640625e-05,
      "step": 7802
    },
    {
      "action_loss_avg": 0.021088229026645422,
      "epoch": 2.976226806640625e-05,
      "step": 7802
    },
    {
      "epoch": 2.976226806640625e-05,
      "step": 7802,
      "training_step_time": 9.909439086914062
    },
    {
      "epoch": 2.976226806640625e-05,
      "model_forward_time": 3.1395211219787598,
      "step": 7802
    },
    {
      "epoch": 2.976226806640625e-05,
      "step": 7802,
      "training_step_time": 10.291556358337402
    },
    {
      "epoch": 2.9766082763671875e-05,
      "model_forward_time": 3.152520179748535,
      "step": 7803
    },
    {
      "epoch": 2.9766082763671875e-05,
      "step": 7803,
      "training_step_time": 9.925575017929077
    },
    {
      "epoch": 2.9766082763671875e-05,
      "model_forward_time": 3.1322877407073975,
      "step": 7803
    },
    {
      "epoch": 2.9766082763671875e-05,
      "step": 7803,
      "training_step_time": 9.9039888381958
    },
    {
      "epoch": 2.9766082763671875e-05,
      "model_forward_time": 3.1364519596099854,
      "step": 7803
    },
    {
      "epoch": 2.9766082763671875e-05,
      "step": 7803,
      "training_step_time": 9.905943632125854
    },
    {
      "epoch": 2.9766082763671875e-05,
      "model_forward_time": 3.1333417892456055,
      "step": 7803
    },
    {
      "epoch": 2.9766082763671875e-05,
      "step": 7803,
      "training_step_time": 10.263103723526001
    },
    {
      "epoch": 2.97698974609375e-05,
      "model_forward_time": 3.4233760833740234,
      "step": 7804
    },
    {
      "epoch": 2.97698974609375e-05,
      "step": 7804,
      "training_step_time": 10.193737745285034
    },
    {
      "epoch": 2.97698974609375e-05,
      "model_forward_time": 3.1318976879119873,
      "step": 7804
    },
    {
      "epoch": 2.97698974609375e-05,
      "step": 7804,
      "training_step_time": 9.899660587310791
    },
    {
      "epoch": 2.97698974609375e-05,
      "model_forward_time": 3.1362674236297607,
      "step": 7804
    },
    {
      "epoch": 2.97698974609375e-05,
      "step": 7804,
      "training_step_time": 9.908814668655396
    },
    {
      "epoch": 2.97698974609375e-05,
      "model_forward_time": 3.133453130722046,
      "step": 7804
    },
    {
      "epoch": 2.97698974609375e-05,
      "step": 7804,
      "training_step_time": 10.023056268692017
    },
    {
      "epoch": 2.9773712158203124e-05,
      "model_forward_time": 2.445542097091675,
      "step": 7805
    },
    {
      "dynamics_loss_avg": 0.0745239496231079,
      "epoch": 2.9773712158203124e-05,
      "step": 7805
    },
    {
      "action_loss_avg": 0.024443382490426303,
      "epoch": 2.9773712158203124e-05,
      "step": 7805
    },
    {
      "epoch": 2.9773712158203124e-05,
      "step": 7805,
      "training_step_time": 7.4783055782318115
    },
    {
      "epoch": 2.9773712158203124e-05,
      "model_forward_time": 3.1323587894439697,
      "step": 7805
    },
    {
      "epoch": 2.9773712158203124e-05,
      "step": 7805,
      "training_step_time": 9.899381637573242
    },
    {
      "epoch": 2.9773712158203124e-05,
      "model_forward_time": 3.4038360118865967,
      "step": 7805
    },
    {
      "epoch": 2.9773712158203124e-05,
      "step": 7805,
      "training_step_time": 10.172264099121094
    },
    {
      "epoch": 2.9773712158203124e-05,
      "model_forward_time": 3.1347098350524902,
      "step": 7805
    },
    {
      "epoch": 2.9773712158203124e-05,
      "step": 7805,
      "training_step_time": 10.288554906845093
    },
    {
      "epoch": 2.977752685546875e-05,
      "model_forward_time": 3.146760940551758,
      "step": 7806
    },
    {
      "epoch": 2.977752685546875e-05,
      "step": 7806,
      "training_step_time": 9.918745756149292
    },
    {
      "epoch": 2.977752685546875e-05,
      "model_forward_time": 3.1381685733795166,
      "step": 7806
    },
    {
      "epoch": 2.977752685546875e-05,
      "step": 7806,
      "training_step_time": 9.908684968948364
    },
    {
      "epoch": 2.977752685546875e-05,
      "model_forward_time": 3.4143612384796143,
      "step": 7806
    },
    {
      "epoch": 2.977752685546875e-05,
      "step": 7806,
      "training_step_time": 10.184613704681396
    },
    {
      "epoch": 2.977752685546875e-05,
      "model_forward_time": 3.1349120140075684,
      "step": 7806
    },
    {
      "epoch": 2.977752685546875e-05,
      "step": 7806,
      "training_step_time": 10.334815502166748
    },
    {
      "epoch": 2.9781341552734376e-05,
      "model_forward_time": 3.1582274436950684,
      "step": 7807
    },
    {
      "epoch": 2.9781341552734376e-05,
      "step": 7807,
      "training_step_time": 9.93151044845581
    },
    {
      "epoch": 2.9781341552734376e-05,
      "model_forward_time": 3.1531946659088135,
      "step": 7807
    },
    {
      "epoch": 2.9781341552734376e-05,
      "step": 7807,
      "training_step_time": 9.925238847732544
    },
    {
      "epoch": 2.9781341552734376e-05,
      "model_forward_time": 3.1421220302581787,
      "step": 7807
    },
    {
      "dynamics_loss_avg": 0.06339550837874412,
      "epoch": 2.9781341552734376e-05,
      "step": 7807
    },
    {
      "action_loss_avg": 0.02247189832851291,
      "epoch": 2.9781341552734376e-05,
      "step": 7807
    },
    {
      "epoch": 2.9781341552734376e-05,
      "step": 7807,
      "training_step_time": 9.919140577316284
    },
    {
      "epoch": 2.9781341552734376e-05,
      "model_forward_time": 3.147822380065918,
      "step": 7807
    },
    {
      "epoch": 2.9781341552734376e-05,
      "step": 7807,
      "training_step_time": 10.255338430404663
    },
    {
      "epoch": 2.978515625e-05,
      "model_forward_time": 3.5070250034332275,
      "step": 7808
    },
    {
      "epoch": 2.978515625e-05,
      "step": 7808,
      "training_step_time": 10.289472103118896
    },
    {
      "epoch": 2.978515625e-05,
      "model_forward_time": 3.1542000770568848,
      "step": 7808
    },
    {
      "epoch": 2.978515625e-05,
      "step": 7808,
      "training_step_time": 9.945565938949585
    },
    {
      "epoch": 2.978515625e-05,
      "model_forward_time": 3.14316987991333,
      "step": 7808
    },
    {
      "epoch": 2.978515625e-05,
      "step": 7808,
      "training_step_time": 9.920162200927734
    },
    {
      "epoch": 2.978515625e-05,
      "model_forward_time": 3.150350332260132,
      "step": 7808
    },
    {
      "epoch": 2.978515625e-05,
      "step": 7808,
      "training_step_time": 10.269015073776245
    },
    {
      "epoch": 2.9788970947265626e-05,
      "model_forward_time": 3.464418649673462,
      "step": 7809
    },
    {
      "epoch": 2.9788970947265626e-05,
      "step": 7809,
      "training_step_time": 10.243252038955688
    },
    {
      "epoch": 2.9788970947265626e-05,
      "model_forward_time": 3.1468894481658936,
      "step": 7809
    },
    {
      "epoch": 2.9788970947265626e-05,
      "step": 7809,
      "training_step_time": 9.931403160095215
    },
    {
      "epoch": 2.9788970947265626e-05,
      "model_forward_time": 3.1501948833465576,
      "step": 7809
    },
    {
      "epoch": 2.9788970947265626e-05,
      "step": 7809,
      "training_step_time": 9.938436269760132
    },
    {
      "epoch": 2.9788970947265626e-05,
      "model_forward_time": 3.1478683948516846,
      "step": 7809
    },
    {
      "epoch": 2.9788970947265626e-05,
      "step": 7809,
      "training_step_time": 10.263389348983765
    },
    {
      "epoch": 2.979278564453125e-05,
      "grad_norm": 0.45151346921920776,
      "learning_rate": 1e-05,
      "loss": 0.0827,
      "step": 7810
    },
    {
      "epoch": 2.979278564453125e-05,
      "model_forward_time": 3.160385847091675,
      "step": 7810
    },
    {
      "dynamics_loss_avg": 0.06586992647498846,
      "epoch": 2.979278564453125e-05,
      "step": 7810
    },
    {
      "action_loss_avg": 0.02188930567353964,
      "epoch": 2.979278564453125e-05,
      "step": 7810
    },
    {
      "epoch": 2.979278564453125e-05,
      "step": 7810,
      "training_step_time": 9.944033861160278
    },
    {
      "epoch": 2.979278564453125e-05,
      "model_forward_time": 3.149900436401367,
      "step": 7810
    },
    {
      "epoch": 2.979278564453125e-05,
      "step": 7810,
      "training_step_time": 10.251559972763062
    },
    {
      "epoch": 2.979278564453125e-05,
      "model_forward_time": 3.1443729400634766,
      "step": 7810
    },
    {
      "epoch": 2.979278564453125e-05,
      "step": 7810,
      "training_step_time": 9.924715995788574
    },
    {
      "epoch": 2.979278564453125e-05,
      "model_forward_time": 3.14890718460083,
      "step": 7810
    },
    {
      "epoch": 2.979278564453125e-05,
      "step": 7810,
      "training_step_time": 10.313490152359009
    },
    {
      "epoch": 2.9796600341796875e-05,
      "model_forward_time": 3.1639699935913086,
      "step": 7811
    },
    {
      "epoch": 2.9796600341796875e-05,
      "step": 7811,
      "training_step_time": 9.943750619888306
    },
    {
      "epoch": 2.9796600341796875e-05,
      "model_forward_time": 3.1422247886657715,
      "step": 7811
    },
    {
      "epoch": 2.9796600341796875e-05,
      "step": 7811,
      "training_step_time": 9.921134948730469
    },
    {
      "epoch": 2.9796600341796875e-05,
      "model_forward_time": 3.1501753330230713,
      "step": 7811
    },
    {
      "epoch": 2.9796600341796875e-05,
      "step": 7811,
      "training_step_time": 10.25827693939209
    },
    {
      "epoch": 2.9796600341796875e-05,
      "model_forward_time": 3.1442043781280518,
      "step": 7811
    },
    {
      "epoch": 2.9796600341796875e-05,
      "step": 7811,
      "training_step_time": 10.307347536087036
    },
    {
      "epoch": 2.98004150390625e-05,
      "model_forward_time": 3.16894793510437,
      "step": 7812
    },
    {
      "epoch": 2.98004150390625e-05,
      "step": 7812,
      "training_step_time": 9.950973749160767
    },
    {
      "epoch": 2.98004150390625e-05,
      "model_forward_time": 3.144522190093994,
      "step": 7812
    },
    {
      "epoch": 2.98004150390625e-05,
      "step": 7812,
      "training_step_time": 9.920888662338257
    },
    {
      "epoch": 2.98004150390625e-05,
      "model_forward_time": 3.147996664047241,
      "step": 7812
    },
    {
      "dynamics_loss_avg": 0.05105213541537523,
      "epoch": 2.98004150390625e-05,
      "step": 7812
    },
    {
      "action_loss_avg": 0.01658065724186599,
      "epoch": 2.98004150390625e-05,
      "step": 7812
    },
    {
      "epoch": 2.98004150390625e-05,
      "step": 7812,
      "training_step_time": 9.92771577835083
    },
    {
      "epoch": 2.98004150390625e-05,
      "model_forward_time": 3.429551124572754,
      "step": 7812
    },
    {
      "epoch": 2.98004150390625e-05,
      "step": 7812,
      "training_step_time": 10.307851076126099
    },
    {
      "epoch": 2.9804229736328124e-05,
      "model_forward_time": 3.1588938236236572,
      "step": 7813
    },
    {
      "epoch": 2.9804229736328124e-05,
      "step": 7813,
      "training_step_time": 9.947499513626099
    },
    {
      "epoch": 2.9804229736328124e-05,
      "model_forward_time": 3.1467843055725098,
      "step": 7813
    },
    {
      "epoch": 2.9804229736328124e-05,
      "step": 7813,
      "training_step_time": 9.932673454284668
    },
    {
      "epoch": 2.9804229736328124e-05,
      "model_forward_time": 3.1518938541412354,
      "step": 7813
    },
    {
      "epoch": 2.9804229736328124e-05,
      "step": 7813,
      "training_step_time": 9.936424493789673
    },
    {
      "epoch": 2.9804229736328124e-05,
      "model_forward_time": 3.1432149410247803,
      "step": 7813
    },
    {
      "epoch": 2.9804229736328124e-05,
      "step": 7813,
      "training_step_time": 10.328363180160522
    },
    {
      "epoch": 2.980804443359375e-05,
      "model_forward_time": 3.1524155139923096,
      "step": 7814
    },
    {
      "epoch": 2.980804443359375e-05,
      "step": 7814,
      "training_step_time": 10.243000268936157
    },
    {
      "epoch": 2.980804443359375e-05,
      "model_forward_time": 3.140167236328125,
      "step": 7814
    },
    {
      "epoch": 2.980804443359375e-05,
      "step": 7814,
      "training_step_time": 9.917997360229492
    },
    {
      "epoch": 2.980804443359375e-05,
      "model_forward_time": 3.1455342769622803,
      "step": 7814
    },
    {
      "epoch": 2.980804443359375e-05,
      "step": 7814,
      "training_step_time": 9.931276559829712
    },
    {
      "epoch": 2.980804443359375e-05,
      "model_forward_time": 3.1451377868652344,
      "step": 7814
    },
    {
      "epoch": 2.980804443359375e-05,
      "step": 7814,
      "training_step_time": 10.254264116287231
    },
    {
      "epoch": 2.9811859130859376e-05,
      "model_forward_time": 3.1570847034454346,
      "step": 7815
    },
    {
      "dynamics_loss_avg": 0.051542540080845356,
      "epoch": 2.9811859130859376e-05,
      "step": 7815
    },
    {
      "action_loss_avg": 0.01922518266364932,
      "epoch": 2.9811859130859376e-05,
      "step": 7815
    },
    {
      "epoch": 2.9811859130859376e-05,
      "step": 7815,
      "training_step_time": 9.943851470947266
    },
    {
      "epoch": 2.9811859130859376e-05,
      "model_forward_time": 3.4008986949920654,
      "step": 7815
    },
    {
      "epoch": 2.9811859130859376e-05,
      "step": 7815,
      "training_step_time": 10.178264617919922
    },
    {
      "epoch": 2.9811859130859376e-05,
      "model_forward_time": 3.1402997970581055,
      "step": 7815
    },
    {
      "epoch": 2.9811859130859376e-05,
      "step": 7815,
      "training_step_time": 9.91776728630066
    },
    {
      "epoch": 2.9811859130859376e-05,
      "model_forward_time": 3.1356983184814453,
      "step": 7815
    },
    {
      "epoch": 2.9811859130859376e-05,
      "step": 7815,
      "training_step_time": 10.026138067245483
    },
    {
      "epoch": 2.9815673828125e-05,
      "model_forward_time": 3.159088134765625,
      "step": 7816
    },
    {
      "epoch": 2.9815673828125e-05,
      "step": 7816,
      "training_step_time": 9.937457084655762
    },
    {
      "epoch": 2.9815673828125e-05,
      "model_forward_time": 3.139622926712036,
      "step": 7816
    },
    {
      "epoch": 2.9815673828125e-05,
      "step": 7816,
      "training_step_time": 9.921592473983765
    },
    {
      "epoch": 2.9815673828125e-05,
      "model_forward_time": 3.418670177459717,
      "step": 7816
    },
    {
      "epoch": 2.9815673828125e-05,
      "step": 7816,
      "training_step_time": 10.198593616485596
    },
    {
      "epoch": 2.9815673828125e-05,
      "model_forward_time": 3.138807535171509,
      "step": 7816
    },
    {
      "epoch": 2.9815673828125e-05,
      "step": 7816,
      "training_step_time": 10.305656671524048
    },
    {
      "epoch": 2.9819488525390626e-05,
      "model_forward_time": 3.151244878768921,
      "step": 7817
    },
    {
      "epoch": 2.9819488525390626e-05,
      "step": 7817,
      "training_step_time": 9.930121183395386
    },
    {
      "epoch": 2.9819488525390626e-05,
      "model_forward_time": 3.138542890548706,
      "step": 7817
    },
    {
      "epoch": 2.9819488525390626e-05,
      "step": 7817,
      "training_step_time": 9.921212673187256
    },
    {
      "epoch": 2.9819488525390626e-05,
      "model_forward_time": 3.13956618309021,
      "step": 7817
    },
    {
      "dynamics_loss_avg": 0.059736050106585024,
      "epoch": 2.9819488525390626e-05,
      "step": 7817
    },
    {
      "action_loss_avg": 0.020871867472305895,
      "epoch": 2.9819488525390626e-05,
      "step": 7817
    },
    {
      "epoch": 2.9819488525390626e-05,
      "step": 7817,
      "training_step_time": 9.922356367111206
    },
    {
      "epoch": 2.9819488525390626e-05,
      "model_forward_time": 3.3916401863098145,
      "step": 7817
    },
    {
      "epoch": 2.9819488525390626e-05,
      "step": 7817,
      "training_step_time": 10.303682804107666
    },
    {
      "epoch": 2.982330322265625e-05,
      "model_forward_time": 3.1460914611816406,
      "step": 7818
    },
    {
      "epoch": 2.982330322265625e-05,
      "step": 7818,
      "training_step_time": 9.929343700408936
    },
    {
      "epoch": 2.982330322265625e-05,
      "model_forward_time": 3.1403095722198486,
      "step": 7818
    },
    {
      "epoch": 2.982330322265625e-05,
      "step": 7818,
      "training_step_time": 9.918855428695679
    },
    {
      "epoch": 2.982330322265625e-05,
      "model_forward_time": 3.133897304534912,
      "step": 7818
    },
    {
      "epoch": 2.982330322265625e-05,
      "step": 7818,
      "training_step_time": 9.908986568450928
    },
    {
      "epoch": 2.982330322265625e-05,
      "model_forward_time": 3.146380662918091,
      "step": 7818
    },
    {
      "epoch": 2.982330322265625e-05,
      "step": 7818,
      "training_step_time": 10.292096853256226
    },
    {
      "epoch": 2.9827117919921875e-05,
      "model_forward_time": 3.4022371768951416,
      "step": 7819
    },
    {
      "epoch": 2.9827117919921875e-05,
      "step": 7819,
      "training_step_time": 10.181658506393433
    },
    {
      "epoch": 2.9827117919921875e-05,
      "model_forward_time": 3.1388468742370605,
      "step": 7819
    },
    {
      "epoch": 2.9827117919921875e-05,
      "step": 7819,
      "training_step_time": 9.91635274887085
    },
    {
      "epoch": 2.9827117919921875e-05,
      "model_forward_time": 3.1355719566345215,
      "step": 7819
    },
    {
      "epoch": 2.9827117919921875e-05,
      "step": 7819,
      "training_step_time": 9.915704488754272
    },
    {
      "epoch": 2.9827117919921875e-05,
      "model_forward_time": 3.1388626098632812,
      "step": 7819
    },
    {
      "epoch": 2.9827117919921875e-05,
      "step": 7819,
      "training_step_time": 10.265003681182861
    },
    {
      "epoch": 2.98309326171875e-05,
      "grad_norm": 0.4289419949054718,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 7820
    },
    {
      "epoch": 2.98309326171875e-05,
      "model_forward_time": 3.1542816162109375,
      "step": 7820
    },
    {
      "dynamics_loss_avg": 0.0658166928216815,
      "epoch": 2.98309326171875e-05,
      "step": 7820
    },
    {
      "action_loss_avg": 0.02153342179954052,
      "epoch": 2.98309326171875e-05,
      "step": 7820
    },
    {
      "epoch": 2.98309326171875e-05,
      "step": 7820,
      "training_step_time": 9.94104790687561
    },
    {
      "epoch": 2.98309326171875e-05,
      "model_forward_time": 3.137773275375366,
      "step": 7820
    },
    {
      "epoch": 2.98309326171875e-05,
      "step": 7820,
      "training_step_time": 10.239122152328491
    },
    {
      "epoch": 2.98309326171875e-05,
      "model_forward_time": 3.1410062313079834,
      "step": 7820
    },
    {
      "epoch": 2.98309326171875e-05,
      "step": 7820,
      "training_step_time": 9.920958042144775
    },
    {
      "epoch": 2.98309326171875e-05,
      "model_forward_time": 3.1434969902038574,
      "step": 7820
    },
    {
      "epoch": 2.98309326171875e-05,
      "step": 7820,
      "training_step_time": 10.27359652519226
    },
    {
      "epoch": 2.9834747314453124e-05,
      "model_forward_time": 3.15982723236084,
      "step": 7821
    },
    {
      "epoch": 2.9834747314453124e-05,
      "step": 7821,
      "training_step_time": 9.947954893112183
    },
    {
      "epoch": 2.9834747314453124e-05,
      "model_forward_time": 3.140636444091797,
      "step": 7821
    },
    {
      "epoch": 2.9834747314453124e-05,
      "step": 7821,
      "training_step_time": 9.922478675842285
    },
    {
      "epoch": 2.9834747314453124e-05,
      "model_forward_time": 3.4202592372894287,
      "step": 7821
    },
    {
      "epoch": 2.9834747314453124e-05,
      "step": 7821,
      "training_step_time": 10.205048561096191
    },
    {
      "epoch": 2.9834747314453124e-05,
      "model_forward_time": 3.1451351642608643,
      "step": 7821
    },
    {
      "epoch": 2.9834747314453124e-05,
      "step": 7821,
      "training_step_time": 10.3175687789917
    },
    {
      "epoch": 2.983856201171875e-05,
      "model_forward_time": 3.1533849239349365,
      "step": 7822
    },
    {
      "epoch": 2.983856201171875e-05,
      "step": 7822,
      "training_step_time": 9.93496584892273
    },
    {
      "epoch": 2.983856201171875e-05,
      "model_forward_time": 3.1384353637695312,
      "step": 7822
    },
    {
      "epoch": 2.983856201171875e-05,
      "step": 7822,
      "training_step_time": 9.918687582015991
    },
    {
      "epoch": 2.983856201171875e-05,
      "model_forward_time": 3.1418490409851074,
      "step": 7822
    },
    {
      "dynamics_loss_avg": 0.06294334270060062,
      "epoch": 2.983856201171875e-05,
      "step": 7822
    },
    {
      "action_loss_avg": 0.01987993512302637,
      "epoch": 2.983856201171875e-05,
      "step": 7822
    },
    {
      "epoch": 2.983856201171875e-05,
      "step": 7822,
      "training_step_time": 9.924461603164673
    },
    {
      "epoch": 2.983856201171875e-05,
      "model_forward_time": 3.1364874839782715,
      "step": 7822
    },
    {
      "epoch": 2.983856201171875e-05,
      "step": 7822,
      "training_step_time": 10.289737939834595
    },
    {
      "epoch": 2.9842376708984377e-05,
      "model_forward_time": 3.1577303409576416,
      "step": 7823
    },
    {
      "epoch": 2.9842376708984377e-05,
      "step": 7823,
      "training_step_time": 9.942498922348022
    },
    {
      "epoch": 2.9842376708984377e-05,
      "model_forward_time": 3.150412082672119,
      "step": 7823
    },
    {
      "epoch": 2.9842376708984377e-05,
      "step": 7823,
      "training_step_time": 9.929977416992188
    },
    {
      "epoch": 2.9842376708984377e-05,
      "model_forward_time": 3.1514365673065186,
      "step": 7823
    },
    {
      "epoch": 2.9842376708984377e-05,
      "step": 7823,
      "training_step_time": 9.926618576049805
    },
    {
      "epoch": 2.9842376708984377e-05,
      "model_forward_time": 3.144056558609009,
      "step": 7823
    },
    {
      "epoch": 2.9842376708984377e-05,
      "step": 7823,
      "training_step_time": 10.3144211769104
    },
    {
      "epoch": 2.984619140625e-05,
      "model_forward_time": 3.495955228805542,
      "step": 7824
    },
    {
      "epoch": 2.984619140625e-05,
      "step": 7824,
      "training_step_time": 10.268677473068237
    },
    {
      "epoch": 2.984619140625e-05,
      "model_forward_time": 3.151296377182007,
      "step": 7824
    },
    {
      "epoch": 2.984619140625e-05,
      "step": 7824,
      "training_step_time": 9.928546905517578
    },
    {
      "epoch": 2.984619140625e-05,
      "model_forward_time": 3.149111032485962,
      "step": 7824
    },
    {
      "epoch": 2.984619140625e-05,
      "step": 7824,
      "training_step_time": 9.918455123901367
    },
    {
      "epoch": 2.984619140625e-05,
      "model_forward_time": 3.1398088932037354,
      "step": 7824
    },
    {
      "epoch": 2.984619140625e-05,
      "step": 7824,
      "training_step_time": 10.313054323196411
    },
    {
      "epoch": 2.9850006103515626e-05,
      "model_forward_time": 3.1472442150115967,
      "step": 7825
    },
    {
      "dynamics_loss_avg": 0.06988648101687431,
      "epoch": 2.9850006103515626e-05,
      "step": 7825
    },
    {
      "action_loss_avg": 0.02337776308413595,
      "epoch": 2.9850006103515626e-05,
      "step": 7825
    },
    {
      "epoch": 2.9850006103515626e-05,
      "step": 7825,
      "training_step_time": 9.92169451713562
    },
    {
      "epoch": 2.9850006103515626e-05,
      "model_forward_time": 3.1545464992523193,
      "step": 7825
    },
    {
      "epoch": 2.9850006103515626e-05,
      "step": 7825,
      "training_step_time": 10.212622880935669
    },
    {
      "epoch": 2.9850006103515626e-05,
      "model_forward_time": 3.1440255641937256,
      "step": 7825
    },
    {
      "epoch": 2.9850006103515626e-05,
      "step": 7825,
      "training_step_time": 9.917006492614746
    },
    {
      "epoch": 2.9850006103515626e-05,
      "model_forward_time": 3.144366979598999,
      "step": 7825
    },
    {
      "epoch": 2.9850006103515626e-05,
      "step": 7825,
      "training_step_time": 10.322952032089233
    },
    {
      "epoch": 2.985382080078125e-05,
      "model_forward_time": 3.173407793045044,
      "step": 7826
    },
    {
      "epoch": 2.985382080078125e-05,
      "step": 7826,
      "training_step_time": 9.95141053199768
    },
    {
      "epoch": 2.985382080078125e-05,
      "model_forward_time": 3.145197629928589,
      "step": 7826
    },
    {
      "epoch": 2.985382080078125e-05,
      "step": 7826,
      "training_step_time": 9.9136643409729
    },
    {
      "epoch": 2.985382080078125e-05,
      "model_forward_time": 3.458275556564331,
      "step": 7826
    },
    {
      "epoch": 2.985382080078125e-05,
      "step": 7826,
      "training_step_time": 10.228148698806763
    },
    {
      "epoch": 2.985382080078125e-05,
      "model_forward_time": 3.1391427516937256,
      "step": 7826
    },
    {
      "epoch": 2.985382080078125e-05,
      "step": 7826,
      "training_step_time": 10.28488540649414
    },
    {
      "epoch": 2.9857635498046875e-05,
      "model_forward_time": 3.1551895141601562,
      "step": 7827
    },
    {
      "epoch": 2.9857635498046875e-05,
      "step": 7827,
      "training_step_time": 9.930192947387695
    },
    {
      "epoch": 2.9857635498046875e-05,
      "model_forward_time": 3.152594566345215,
      "step": 7827
    },
    {
      "epoch": 2.9857635498046875e-05,
      "step": 7827,
      "training_step_time": 9.92607069015503
    },
    {
      "epoch": 2.9857635498046875e-05,
      "model_forward_time": 3.1445417404174805,
      "step": 7827
    },
    {
      "dynamics_loss_avg": 0.06449871025979519,
      "epoch": 2.9857635498046875e-05,
      "step": 7827
    },
    {
      "action_loss_avg": 0.02518772454932332,
      "epoch": 2.9857635498046875e-05,
      "step": 7827
    },
    {
      "epoch": 2.9857635498046875e-05,
      "step": 7827,
      "training_step_time": 9.921462774276733
    },
    {
      "epoch": 2.9857635498046875e-05,
      "model_forward_time": 3.4183623790740967,
      "step": 7827
    },
    {
      "epoch": 2.9857635498046875e-05,
      "step": 7827,
      "training_step_time": 10.310073614120483
    },
    {
      "epoch": 2.98614501953125e-05,
      "model_forward_time": 3.154327154159546,
      "step": 7828
    },
    {
      "epoch": 2.98614501953125e-05,
      "step": 7828,
      "training_step_time": 9.929413318634033
    },
    {
      "epoch": 2.98614501953125e-05,
      "model_forward_time": 3.1436939239501953,
      "step": 7828
    },
    {
      "epoch": 2.98614501953125e-05,
      "step": 7828,
      "training_step_time": 9.916866779327393
    },
    {
      "epoch": 2.98614501953125e-05,
      "model_forward_time": 3.1482059955596924,
      "step": 7828
    },
    {
      "epoch": 2.98614501953125e-05,
      "step": 7828,
      "training_step_time": 9.926659345626831
    },
    {
      "epoch": 2.98614501953125e-05,
      "model_forward_time": 3.1480886936187744,
      "step": 7828
    },
    {
      "epoch": 2.98614501953125e-05,
      "step": 7828,
      "training_step_time": 10.019315719604492
    },
    {
      "epoch": 2.9865264892578124e-05,
      "model_forward_time": 3.414621591567993,
      "step": 7829
    },
    {
      "epoch": 2.9865264892578124e-05,
      "step": 7829,
      "training_step_time": 10.188224792480469
    },
    {
      "epoch": 2.9865264892578124e-05,
      "model_forward_time": 3.1462197303771973,
      "step": 7829
    },
    {
      "epoch": 2.9865264892578124e-05,
      "step": 7829,
      "training_step_time": 9.918590545654297
    },
    {
      "epoch": 2.9865264892578124e-05,
      "model_forward_time": 3.144284725189209,
      "step": 7829
    },
    {
      "epoch": 2.9865264892578124e-05,
      "step": 7829,
      "training_step_time": 9.9181969165802
    },
    {
      "epoch": 2.9865264892578124e-05,
      "model_forward_time": 3.1493613719940186,
      "step": 7829
    },
    {
      "epoch": 2.9865264892578124e-05,
      "step": 7829,
      "training_step_time": 10.306384325027466
    },
    {
      "epoch": 2.986907958984375e-05,
      "grad_norm": 0.43117862939834595,
      "learning_rate": 1e-05,
      "loss": 0.0839,
      "step": 7830
    },
    {
      "epoch": 2.986907958984375e-05,
      "model_forward_time": 3.15731143951416,
      "step": 7830
    },
    {
      "dynamics_loss_avg": 0.06646592114120722,
      "epoch": 2.986907958984375e-05,
      "step": 7830
    },
    {
      "action_loss_avg": 0.022552044317126273,
      "epoch": 2.986907958984375e-05,
      "step": 7830
    },
    {
      "epoch": 2.986907958984375e-05,
      "step": 7830,
      "training_step_time": 9.93591022491455
    },
    {
      "epoch": 2.986907958984375e-05,
      "model_forward_time": 3.426100492477417,
      "step": 7830
    },
    {
      "epoch": 2.986907958984375e-05,
      "step": 7830,
      "training_step_time": 10.204034090042114
    },
    {
      "epoch": 2.986907958984375e-05,
      "model_forward_time": 3.145127058029175,
      "step": 7830
    },
    {
      "epoch": 2.986907958984375e-05,
      "step": 7830,
      "training_step_time": 9.921504974365234
    },
    {
      "epoch": 2.986907958984375e-05,
      "model_forward_time": 3.1475226879119873,
      "step": 7830
    },
    {
      "epoch": 2.986907958984375e-05,
      "step": 7830,
      "training_step_time": 10.033056497573853
    },
    {
      "epoch": 2.9872894287109377e-05,
      "model_forward_time": 3.1587777137756348,
      "step": 7831
    },
    {
      "epoch": 2.9872894287109377e-05,
      "step": 7831,
      "training_step_time": 9.933518886566162
    },
    {
      "epoch": 2.9872894287109377e-05,
      "model_forward_time": 3.1413843631744385,
      "step": 7831
    },
    {
      "epoch": 2.9872894287109377e-05,
      "step": 7831,
      "training_step_time": 9.918731451034546
    },
    {
      "epoch": 2.9872894287109377e-05,
      "model_forward_time": 3.1508102416992188,
      "step": 7831
    },
    {
      "epoch": 2.9872894287109377e-05,
      "step": 7831,
      "training_step_time": 10.249225616455078
    },
    {
      "epoch": 2.9872894287109377e-05,
      "model_forward_time": 3.1515183448791504,
      "step": 7831
    },
    {
      "epoch": 2.9872894287109377e-05,
      "step": 7831,
      "training_step_time": 10.296223640441895
    },
    {
      "epoch": 2.9876708984375e-05,
      "model_forward_time": 3.1634039878845215,
      "step": 7832
    },
    {
      "epoch": 2.9876708984375e-05,
      "step": 7832,
      "training_step_time": 9.945762395858765
    },
    {
      "epoch": 2.9876708984375e-05,
      "model_forward_time": 3.1460366249084473,
      "step": 7832
    },
    {
      "epoch": 2.9876708984375e-05,
      "step": 7832,
      "training_step_time": 9.924156665802002
    },
    {
      "epoch": 2.9876708984375e-05,
      "model_forward_time": 3.1466622352600098,
      "step": 7832
    },
    {
      "dynamics_loss_avg": 0.05569753795862198,
      "epoch": 2.9876708984375e-05,
      "step": 7832
    },
    {
      "action_loss_avg": 0.01951066069304943,
      "epoch": 2.9876708984375e-05,
      "step": 7832
    },
    {
      "epoch": 2.9876708984375e-05,
      "step": 7832,
      "training_step_time": 9.923279523849487
    },
    {
      "epoch": 2.9876708984375e-05,
      "model_forward_time": 3.4351861476898193,
      "step": 7832
    },
    {
      "epoch": 2.9876708984375e-05,
      "step": 7832,
      "training_step_time": 10.319743156433105
    },
    {
      "epoch": 2.9880523681640626e-05,
      "model_forward_time": 3.1535465717315674,
      "step": 7833
    },
    {
      "epoch": 2.9880523681640626e-05,
      "step": 7833,
      "training_step_time": 9.924262285232544
    },
    {
      "epoch": 2.9880523681640626e-05,
      "model_forward_time": 3.1382369995117188,
      "step": 7833
    },
    {
      "epoch": 2.9880523681640626e-05,
      "step": 7833,
      "training_step_time": 9.910354375839233
    },
    {
      "epoch": 2.9880523681640626e-05,
      "model_forward_time": 3.1443750858306885,
      "step": 7833
    },
    {
      "epoch": 2.9880523681640626e-05,
      "step": 7833,
      "training_step_time": 9.913467407226562
    },
    {
      "epoch": 2.9880523681640626e-05,
      "model_forward_time": 3.139263868331909,
      "step": 7833
    },
    {
      "epoch": 2.9880523681640626e-05,
      "step": 7833,
      "training_step_time": 10.293529987335205
    },
    {
      "epoch": 2.988433837890625e-05,
      "model_forward_time": 3.14731764793396,
      "step": 7834
    },
    {
      "epoch": 2.988433837890625e-05,
      "step": 7834,
      "training_step_time": 10.18995451927185
    },
    {
      "epoch": 2.988433837890625e-05,
      "model_forward_time": 3.1381618976593018,
      "step": 7834
    },
    {
      "epoch": 2.988433837890625e-05,
      "step": 7834,
      "training_step_time": 9.90552282333374
    },
    {
      "epoch": 2.988433837890625e-05,
      "model_forward_time": 3.1414051055908203,
      "step": 7834
    },
    {
      "epoch": 2.988433837890625e-05,
      "step": 7834,
      "training_step_time": 9.90593957901001
    },
    {
      "epoch": 2.988433837890625e-05,
      "model_forward_time": 3.1382882595062256,
      "step": 7834
    },
    {
      "epoch": 2.988433837890625e-05,
      "step": 7834,
      "training_step_time": 10.323775053024292
    },
    {
      "epoch": 2.9888153076171875e-05,
      "model_forward_time": 3.158949613571167,
      "step": 7835
    },
    {
      "dynamics_loss_avg": 0.06826389618217946,
      "epoch": 2.9888153076171875e-05,
      "step": 7835
    },
    {
      "action_loss_avg": 0.020960842445492744,
      "epoch": 2.9888153076171875e-05,
      "step": 7835
    },
    {
      "epoch": 2.9888153076171875e-05,
      "step": 7835,
      "training_step_time": 9.934102296829224
    },
    {
      "epoch": 2.9888153076171875e-05,
      "model_forward_time": 3.399376630783081,
      "step": 7835
    },
    {
      "epoch": 2.9888153076171875e-05,
      "step": 7835,
      "training_step_time": 10.162917375564575
    },
    {
      "epoch": 2.9888153076171875e-05,
      "model_forward_time": 3.151470899581909,
      "step": 7835
    },
    {
      "epoch": 2.9888153076171875e-05,
      "step": 7835,
      "training_step_time": 9.917127132415771
    },
    {
      "epoch": 2.9888153076171875e-05,
      "model_forward_time": 3.134596824645996,
      "step": 7835
    },
    {
      "epoch": 2.9888153076171875e-05,
      "step": 7835,
      "training_step_time": 10.282751321792603
    },
    {
      "epoch": 2.98919677734375e-05,
      "model_forward_time": 3.1544761657714844,
      "step": 7836
    },
    {
      "epoch": 2.98919677734375e-05,
      "step": 7836,
      "training_step_time": 9.919174671173096
    },
    {
      "epoch": 2.98919677734375e-05,
      "model_forward_time": 3.131726026535034,
      "step": 7836
    },
    {
      "epoch": 2.98919677734375e-05,
      "step": 7836,
      "training_step_time": 9.898786067962646
    },
    {
      "epoch": 2.98919677734375e-05,
      "model_forward_time": 3.4407458305358887,
      "step": 7836
    },
    {
      "epoch": 2.98919677734375e-05,
      "step": 7836,
      "training_step_time": 10.20626187324524
    },
    {
      "epoch": 2.98919677734375e-05,
      "model_forward_time": 3.1342837810516357,
      "step": 7836
    },
    {
      "epoch": 2.98919677734375e-05,
      "step": 7836,
      "training_step_time": 10.264617681503296
    },
    {
      "epoch": 2.9895782470703124e-05,
      "model_forward_time": 3.152344226837158,
      "step": 7837
    },
    {
      "epoch": 2.9895782470703124e-05,
      "step": 7837,
      "training_step_time": 9.918708801269531
    },
    {
      "epoch": 2.9895782470703124e-05,
      "model_forward_time": 3.1375086307525635,
      "step": 7837
    },
    {
      "epoch": 2.9895782470703124e-05,
      "step": 7837,
      "training_step_time": 9.907857418060303
    },
    {
      "epoch": 2.9895782470703124e-05,
      "model_forward_time": 2.4356184005737305,
      "step": 7837
    },
    {
      "dynamics_loss_avg": 0.0721978560090065,
      "epoch": 2.9895782470703124e-05,
      "step": 7837
    },
    {
      "action_loss_avg": 0.021073264721781015,
      "epoch": 2.9895782470703124e-05,
      "step": 7837
    },
    {
      "epoch": 2.9895782470703124e-05,
      "step": 7837,
      "training_step_time": 7.467797517776489
    },
    {
      "epoch": 2.9895782470703124e-05,
      "model_forward_time": 3.4262406826019287,
      "step": 7837
    },
    {
      "epoch": 2.9895782470703124e-05,
      "step": 7837,
      "training_step_time": 10.290618419647217
    },
    {
      "epoch": 2.989959716796875e-05,
      "model_forward_time": 3.1602816581726074,
      "step": 7838
    },
    {
      "epoch": 2.989959716796875e-05,
      "step": 7838,
      "training_step_time": 9.932426929473877
    },
    {
      "epoch": 2.989959716796875e-05,
      "model_forward_time": 3.139660120010376,
      "step": 7838
    },
    {
      "epoch": 2.989959716796875e-05,
      "step": 7838,
      "training_step_time": 9.914990186691284
    },
    {
      "epoch": 2.989959716796875e-05,
      "model_forward_time": 3.1419947147369385,
      "step": 7838
    },
    {
      "epoch": 2.989959716796875e-05,
      "step": 7838,
      "training_step_time": 9.911698579788208
    },
    {
      "epoch": 2.989959716796875e-05,
      "model_forward_time": 3.1398849487304688,
      "step": 7838
    },
    {
      "epoch": 2.989959716796875e-05,
      "step": 7838,
      "training_step_time": 10.297783374786377
    },
    {
      "epoch": 2.9903411865234377e-05,
      "model_forward_time": 3.1484501361846924,
      "step": 7839
    },
    {
      "epoch": 2.9903411865234377e-05,
      "step": 7839,
      "training_step_time": 10.186120748519897
    },
    {
      "epoch": 2.9903411865234377e-05,
      "model_forward_time": 3.1413049697875977,
      "step": 7839
    },
    {
      "epoch": 2.9903411865234377e-05,
      "step": 7839,
      "training_step_time": 9.908458709716797
    },
    {
      "epoch": 2.9903411865234377e-05,
      "model_forward_time": 3.1394145488739014,
      "step": 7839
    },
    {
      "epoch": 2.9903411865234377e-05,
      "step": 7839,
      "training_step_time": 9.909881114959717
    },
    {
      "epoch": 2.9903411865234377e-05,
      "model_forward_time": 3.1442770957946777,
      "step": 7839
    },
    {
      "epoch": 2.9903411865234377e-05,
      "step": 7839,
      "training_step_time": 10.2569899559021
    },
    {
      "epoch": 2.99072265625e-05,
      "grad_norm": 0.409226655960083,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 7840
    },
    {
      "epoch": 2.99072265625e-05,
      "model_forward_time": 3.1594789028167725,
      "step": 7840
    },
    {
      "dynamics_loss_avg": 0.05887682773172855,
      "epoch": 2.99072265625e-05,
      "step": 7840
    },
    {
      "action_loss_avg": 0.021331236977130175,
      "epoch": 2.99072265625e-05,
      "step": 7840
    },
    {
      "epoch": 2.99072265625e-05,
      "step": 7840,
      "training_step_time": 9.939256429672241
    },
    {
      "epoch": 2.99072265625e-05,
      "model_forward_time": 3.4400429725646973,
      "step": 7840
    },
    {
      "epoch": 2.99072265625e-05,
      "step": 7840,
      "training_step_time": 10.215749502182007
    },
    {
      "epoch": 2.99072265625e-05,
      "model_forward_time": 3.1472465991973877,
      "step": 7840
    },
    {
      "epoch": 2.99072265625e-05,
      "step": 7840,
      "training_step_time": 9.920381546020508
    },
    {
      "epoch": 2.99072265625e-05,
      "model_forward_time": 3.152688503265381,
      "step": 7840
    },
    {
      "epoch": 2.99072265625e-05,
      "step": 7840,
      "training_step_time": 10.29189682006836
    },
    {
      "epoch": 2.9911041259765626e-05,
      "model_forward_time": 3.168445348739624,
      "step": 7841
    },
    {
      "epoch": 2.9911041259765626e-05,
      "step": 7841,
      "training_step_time": 9.947507381439209
    },
    {
      "epoch": 2.9911041259765626e-05,
      "model_forward_time": 3.1476423740386963,
      "step": 7841
    },
    {
      "epoch": 2.9911041259765626e-05,
      "step": 7841,
      "training_step_time": 9.920963287353516
    },
    {
      "epoch": 2.9911041259765626e-05,
      "model_forward_time": 3.403101921081543,
      "step": 7841
    },
    {
      "epoch": 2.9911041259765626e-05,
      "step": 7841,
      "training_step_time": 10.173839569091797
    },
    {
      "epoch": 2.9911041259765626e-05,
      "model_forward_time": 3.140824556350708,
      "step": 7841
    },
    {
      "epoch": 2.9911041259765626e-05,
      "step": 7841,
      "training_step_time": 10.324597597122192
    },
    {
      "epoch": 2.991485595703125e-05,
      "model_forward_time": 3.1492156982421875,
      "step": 7842
    },
    {
      "epoch": 2.991485595703125e-05,
      "step": 7842,
      "training_step_time": 9.923580646514893
    },
    {
      "epoch": 2.991485595703125e-05,
      "model_forward_time": 3.141059160232544,
      "step": 7842
    },
    {
      "epoch": 2.991485595703125e-05,
      "step": 7842,
      "training_step_time": 9.912652015686035
    },
    {
      "epoch": 2.991485595703125e-05,
      "model_forward_time": 3.1445488929748535,
      "step": 7842
    },
    {
      "dynamics_loss_avg": 0.05704279039055109,
      "epoch": 2.991485595703125e-05,
      "step": 7842
    },
    {
      "action_loss_avg": 0.023837937274947762,
      "epoch": 2.991485595703125e-05,
      "step": 7842
    },
    {
      "epoch": 2.991485595703125e-05,
      "step": 7842,
      "training_step_time": 9.919948816299438
    },
    {
      "epoch": 2.991485595703125e-05,
      "model_forward_time": 3.415462017059326,
      "step": 7842
    },
    {
      "epoch": 2.991485595703125e-05,
      "step": 7842,
      "training_step_time": 10.287699460983276
    },
    {
      "epoch": 2.9918670654296875e-05,
      "model_forward_time": 3.149320125579834,
      "step": 7843
    },
    {
      "epoch": 2.9918670654296875e-05,
      "step": 7843,
      "training_step_time": 9.925273180007935
    },
    {
      "epoch": 2.9918670654296875e-05,
      "model_forward_time": 3.1369175910949707,
      "step": 7843
    },
    {
      "epoch": 2.9918670654296875e-05,
      "step": 7843,
      "training_step_time": 9.906925201416016
    },
    {
      "epoch": 2.9918670654296875e-05,
      "model_forward_time": 3.1385414600372314,
      "step": 7843
    },
    {
      "epoch": 2.9918670654296875e-05,
      "step": 7843,
      "training_step_time": 9.90704870223999
    },
    {
      "epoch": 2.9918670654296875e-05,
      "model_forward_time": 3.1348273754119873,
      "step": 7843
    },
    {
      "epoch": 2.9918670654296875e-05,
      "step": 7843,
      "training_step_time": 10.298986911773682
    },
    {
      "epoch": 2.99224853515625e-05,
      "model_forward_time": 3.4238665103912354,
      "step": 7844
    },
    {
      "epoch": 2.99224853515625e-05,
      "step": 7844,
      "training_step_time": 10.194661855697632
    },
    {
      "epoch": 2.99224853515625e-05,
      "model_forward_time": 3.1323161125183105,
      "step": 7844
    },
    {
      "epoch": 2.99224853515625e-05,
      "step": 7844,
      "training_step_time": 9.90238070487976
    },
    {
      "epoch": 2.99224853515625e-05,
      "model_forward_time": 3.141122341156006,
      "step": 7844
    },
    {
      "epoch": 2.99224853515625e-05,
      "step": 7844,
      "training_step_time": 9.916829824447632
    },
    {
      "epoch": 2.99224853515625e-05,
      "model_forward_time": 3.133971929550171,
      "step": 7844
    },
    {
      "epoch": 2.99224853515625e-05,
      "step": 7844,
      "training_step_time": 10.276965379714966
    },
    {
      "epoch": 2.9926300048828124e-05,
      "model_forward_time": 3.155891180038452,
      "step": 7845
    },
    {
      "dynamics_loss_avg": 0.06528383325785399,
      "epoch": 2.9926300048828124e-05,
      "step": 7845
    },
    {
      "action_loss_avg": 0.019363529141992332,
      "epoch": 2.9926300048828124e-05,
      "step": 7845
    },
    {
      "epoch": 2.9926300048828124e-05,
      "step": 7845,
      "training_step_time": 9.934046745300293
    },
    {
      "epoch": 2.9926300048828124e-05,
      "model_forward_time": 3.3935141563415527,
      "step": 7845
    },
    {
      "epoch": 2.9926300048828124e-05,
      "step": 7845,
      "training_step_time": 10.165253639221191
    },
    {
      "epoch": 2.9926300048828124e-05,
      "model_forward_time": 3.139204263687134,
      "step": 7845
    },
    {
      "epoch": 2.9926300048828124e-05,
      "step": 7845,
      "training_step_time": 9.909127235412598
    },
    {
      "epoch": 2.9926300048828124e-05,
      "model_forward_time": 3.1325266361236572,
      "step": 7845
    },
    {
      "epoch": 2.9926300048828124e-05,
      "step": 7845,
      "training_step_time": 10.325304508209229
    },
    {
      "epoch": 2.993011474609375e-05,
      "model_forward_time": 3.1495401859283447,
      "step": 7846
    },
    {
      "epoch": 2.993011474609375e-05,
      "step": 7846,
      "training_step_time": 9.91711163520813
    },
    {
      "epoch": 2.993011474609375e-05,
      "model_forward_time": 3.1338541507720947,
      "step": 7846
    },
    {
      "epoch": 2.993011474609375e-05,
      "step": 7846,
      "training_step_time": 9.899227142333984
    },
    {
      "epoch": 2.993011474609375e-05,
      "model_forward_time": 3.4431872367858887,
      "step": 7846
    },
    {
      "epoch": 2.993011474609375e-05,
      "step": 7846,
      "training_step_time": 10.21216344833374
    },
    {
      "epoch": 2.993011474609375e-05,
      "model_forward_time": 3.135215997695923,
      "step": 7846
    },
    {
      "epoch": 2.993011474609375e-05,
      "step": 7846,
      "training_step_time": 10.298437118530273
    },
    {
      "epoch": 2.9933929443359377e-05,
      "model_forward_time": 3.1496310234069824,
      "step": 7847
    },
    {
      "epoch": 2.9933929443359377e-05,
      "step": 7847,
      "training_step_time": 9.922598361968994
    },
    {
      "epoch": 2.9933929443359377e-05,
      "model_forward_time": 3.1348798274993896,
      "step": 7847
    },
    {
      "epoch": 2.9933929443359377e-05,
      "step": 7847,
      "training_step_time": 9.903356790542603
    },
    {
      "epoch": 2.9933929443359377e-05,
      "model_forward_time": 2.4294326305389404,
      "step": 7847
    },
    {
      "dynamics_loss_avg": 0.057375804521143436,
      "epoch": 2.9933929443359377e-05,
      "step": 7847
    },
    {
      "action_loss_avg": 0.021605828870087863,
      "epoch": 2.9933929443359377e-05,
      "step": 7847
    },
    {
      "epoch": 2.9933929443359377e-05,
      "step": 7847,
      "training_step_time": 7.453532457351685
    },
    {
      "epoch": 2.9933929443359377e-05,
      "model_forward_time": 3.1345715522766113,
      "step": 7847
    },
    {
      "epoch": 2.9933929443359377e-05,
      "step": 7847,
      "training_step_time": 10.274487495422363
    },
    {
      "epoch": 2.9937744140625e-05,
      "model_forward_time": 3.1464309692382812,
      "step": 7848
    },
    {
      "epoch": 2.9937744140625e-05,
      "step": 7848,
      "training_step_time": 9.915201663970947
    },
    {
      "epoch": 2.9937744140625e-05,
      "model_forward_time": 3.132568120956421,
      "step": 7848
    },
    {
      "epoch": 2.9937744140625e-05,
      "step": 7848,
      "training_step_time": 9.898468255996704
    },
    {
      "epoch": 2.9937744140625e-05,
      "model_forward_time": 3.132664203643799,
      "step": 7848
    },
    {
      "epoch": 2.9937744140625e-05,
      "step": 7848,
      "training_step_time": 9.897761821746826
    },
    {
      "epoch": 2.9937744140625e-05,
      "model_forward_time": 3.140239715576172,
      "step": 7848
    },
    {
      "epoch": 2.9937744140625e-05,
      "step": 7848,
      "training_step_time": 10.292061805725098
    },
    {
      "epoch": 2.9941558837890626e-05,
      "model_forward_time": 3.4069762229919434,
      "step": 7849
    },
    {
      "epoch": 2.9941558837890626e-05,
      "step": 7849,
      "training_step_time": 10.175113439559937
    },
    {
      "epoch": 2.9941558837890626e-05,
      "model_forward_time": 2.427713394165039,
      "step": 7849
    },
    {
      "epoch": 2.9941558837890626e-05,
      "step": 7849,
      "training_step_time": 7.45743727684021
    },
    {
      "epoch": 2.9941558837890626e-05,
      "model_forward_time": 3.13372540473938,
      "step": 7849
    },
    {
      "epoch": 2.9941558837890626e-05,
      "step": 7849,
      "training_step_time": 9.908776998519897
    },
    {
      "epoch": 2.9941558837890626e-05,
      "model_forward_time": 3.139831304550171,
      "step": 7849
    },
    {
      "epoch": 2.9941558837890626e-05,
      "step": 7849,
      "training_step_time": 10.321054220199585
    },
    {
      "epoch": 2.994537353515625e-05,
      "grad_norm": 0.48677071928977966,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 7850
    },
    {
      "epoch": 2.994537353515625e-05,
      "model_forward_time": 3.145561456680298,
      "step": 7850
    },
    {
      "dynamics_loss_avg": 0.05684382896870375,
      "epoch": 2.994537353515625e-05,
      "step": 7850
    },
    {
      "action_loss_avg": 0.018690065806731583,
      "epoch": 2.994537353515625e-05,
      "step": 7850
    },
    {
      "epoch": 2.994537353515625e-05,
      "step": 7850,
      "training_step_time": 9.92482590675354
    },
    {
      "epoch": 2.994537353515625e-05,
      "model_forward_time": 3.422318935394287,
      "step": 7850
    },
    {
      "epoch": 2.994537353515625e-05,
      "step": 7850,
      "training_step_time": 10.195497512817383
    },
    {
      "epoch": 2.994537353515625e-05,
      "model_forward_time": 3.1420300006866455,
      "step": 7850
    },
    {
      "epoch": 2.994537353515625e-05,
      "step": 7850,
      "training_step_time": 9.920305728912354
    },
    {
      "epoch": 2.994537353515625e-05,
      "model_forward_time": 3.1421172618865967,
      "step": 7850
    },
    {
      "epoch": 2.994537353515625e-05,
      "step": 7850,
      "training_step_time": 10.282822608947754
    },
    {
      "epoch": 2.9949188232421875e-05,
      "model_forward_time": 3.14994478225708,
      "step": 7851
    },
    {
      "epoch": 2.9949188232421875e-05,
      "step": 7851,
      "training_step_time": 9.931675672531128
    },
    {
      "epoch": 2.9949188232421875e-05,
      "model_forward_time": 3.140193462371826,
      "step": 7851
    },
    {
      "epoch": 2.9949188232421875e-05,
      "step": 7851,
      "training_step_time": 9.91649866104126
    },
    {
      "epoch": 2.9949188232421875e-05,
      "model_forward_time": 3.1359927654266357,
      "step": 7851
    },
    {
      "epoch": 2.9949188232421875e-05,
      "step": 7851,
      "training_step_time": 10.193241119384766
    },
    {
      "epoch": 2.9949188232421875e-05,
      "model_forward_time": 3.142211437225342,
      "step": 7851
    },
    {
      "epoch": 2.9949188232421875e-05,
      "step": 7851,
      "training_step_time": 10.30766487121582
    },
    {
      "epoch": 2.99530029296875e-05,
      "model_forward_time": 3.1651182174682617,
      "step": 7852
    },
    {
      "epoch": 2.99530029296875e-05,
      "step": 7852,
      "training_step_time": 9.941388368606567
    },
    {
      "epoch": 2.99530029296875e-05,
      "model_forward_time": 3.147451162338257,
      "step": 7852
    },
    {
      "epoch": 2.99530029296875e-05,
      "step": 7852,
      "training_step_time": 9.923745155334473
    },
    {
      "epoch": 2.99530029296875e-05,
      "model_forward_time": 3.1509926319122314,
      "step": 7852
    },
    {
      "dynamics_loss_avg": 0.06871092747896909,
      "epoch": 2.99530029296875e-05,
      "step": 7852
    },
    {
      "action_loss_avg": 0.02353336103260517,
      "epoch": 2.99530029296875e-05,
      "step": 7852
    },
    {
      "epoch": 2.99530029296875e-05,
      "step": 7852,
      "training_step_time": 9.925889492034912
    },
    {
      "epoch": 2.99530029296875e-05,
      "model_forward_time": 3.4343690872192383,
      "step": 7852
    },
    {
      "epoch": 2.99530029296875e-05,
      "step": 7852,
      "training_step_time": 10.303433418273926
    },
    {
      "epoch": 2.9956817626953124e-05,
      "model_forward_time": 3.1716442108154297,
      "step": 7853
    },
    {
      "epoch": 2.9956817626953124e-05,
      "step": 7853,
      "training_step_time": 9.955891132354736
    },
    {
      "epoch": 2.9956817626953124e-05,
      "model_forward_time": 3.1424055099487305,
      "step": 7853
    },
    {
      "epoch": 2.9956817626953124e-05,
      "step": 7853,
      "training_step_time": 9.916684865951538
    },
    {
      "epoch": 2.9956817626953124e-05,
      "model_forward_time": 3.147479295730591,
      "step": 7853
    },
    {
      "epoch": 2.9956817626953124e-05,
      "step": 7853,
      "training_step_time": 9.920511960983276
    },
    {
      "epoch": 2.9956817626953124e-05,
      "model_forward_time": 3.1455485820770264,
      "step": 7853
    },
    {
      "epoch": 2.9956817626953124e-05,
      "step": 7853,
      "training_step_time": 10.303712844848633
    },
    {
      "epoch": 2.996063232421875e-05,
      "model_forward_time": 3.152878522872925,
      "step": 7854
    },
    {
      "epoch": 2.996063232421875e-05,
      "step": 7854,
      "training_step_time": 10.20150113105774
    },
    {
      "epoch": 2.996063232421875e-05,
      "model_forward_time": 3.14870548248291,
      "step": 7854
    },
    {
      "epoch": 2.996063232421875e-05,
      "step": 7854,
      "training_step_time": 9.926942586898804
    },
    {
      "epoch": 2.996063232421875e-05,
      "model_forward_time": 3.1584696769714355,
      "step": 7854
    },
    {
      "epoch": 2.996063232421875e-05,
      "step": 7854,
      "training_step_time": 9.937796592712402
    },
    {
      "epoch": 2.996063232421875e-05,
      "model_forward_time": 3.1477367877960205,
      "step": 7854
    },
    {
      "epoch": 2.996063232421875e-05,
      "step": 7854,
      "training_step_time": 10.28405213356018
    },
    {
      "epoch": 2.9964447021484377e-05,
      "model_forward_time": 3.1737210750579834,
      "step": 7855
    },
    {
      "dynamics_loss_avg": 0.057619822956621644,
      "epoch": 2.9964447021484377e-05,
      "step": 7855
    },
    {
      "action_loss_avg": 0.01861877115443349,
      "epoch": 2.9964447021484377e-05,
      "step": 7855
    },
    {
      "epoch": 2.9964447021484377e-05,
      "step": 7855,
      "training_step_time": 9.969122886657715
    },
    {
      "epoch": 2.9964447021484377e-05,
      "model_forward_time": 3.444911241531372,
      "step": 7855
    },
    {
      "epoch": 2.9964447021484377e-05,
      "step": 7855,
      "training_step_time": 10.230067253112793
    },
    {
      "epoch": 2.9964447021484377e-05,
      "model_forward_time": 3.144253969192505,
      "step": 7855
    },
    {
      "epoch": 2.9964447021484377e-05,
      "step": 7855,
      "training_step_time": 9.92555284500122
    },
    {
      "epoch": 2.9964447021484377e-05,
      "model_forward_time": 3.139206886291504,
      "step": 7855
    },
    {
      "epoch": 2.9964447021484377e-05,
      "step": 7855,
      "training_step_time": 10.32115912437439
    },
    {
      "epoch": 2.996826171875e-05,
      "model_forward_time": 3.1473350524902344,
      "step": 7856
    },
    {
      "epoch": 2.996826171875e-05,
      "step": 7856,
      "training_step_time": 9.929591655731201
    },
    {
      "epoch": 2.996826171875e-05,
      "model_forward_time": 3.139087677001953,
      "step": 7856
    },
    {
      "epoch": 2.996826171875e-05,
      "step": 7856,
      "training_step_time": 9.921368598937988
    },
    {
      "epoch": 2.996826171875e-05,
      "model_forward_time": 3.4248692989349365,
      "step": 7856
    },
    {
      "epoch": 2.996826171875e-05,
      "step": 7856,
      "training_step_time": 10.206616163253784
    },
    {
      "epoch": 2.996826171875e-05,
      "model_forward_time": 3.134474992752075,
      "step": 7856
    },
    {
      "epoch": 2.996826171875e-05,
      "step": 7856,
      "training_step_time": 10.325102090835571
    },
    {
      "epoch": 2.9972076416015626e-05,
      "model_forward_time": 3.1535990238189697,
      "step": 7857
    },
    {
      "epoch": 2.9972076416015626e-05,
      "step": 7857,
      "training_step_time": 9.932687997817993
    },
    {
      "epoch": 2.9972076416015626e-05,
      "model_forward_time": 3.1344237327575684,
      "step": 7857
    },
    {
      "epoch": 2.9972076416015626e-05,
      "step": 7857,
      "training_step_time": 9.905930042266846
    },
    {
      "epoch": 2.9972076416015626e-05,
      "model_forward_time": 3.134634256362915,
      "step": 7857
    },
    {
      "dynamics_loss_avg": 0.07293610051274299,
      "epoch": 2.9972076416015626e-05,
      "step": 7857
    },
    {
      "action_loss_avg": 0.020642316062003375,
      "epoch": 2.9972076416015626e-05,
      "step": 7857
    },
    {
      "epoch": 2.9972076416015626e-05,
      "step": 7857,
      "training_step_time": 9.908831119537354
    },
    {
      "epoch": 2.9972076416015626e-05,
      "model_forward_time": 3.429776906967163,
      "step": 7857
    },
    {
      "epoch": 2.9972076416015626e-05,
      "step": 7857,
      "training_step_time": 10.303034782409668
    },
    {
      "epoch": 2.997589111328125e-05,
      "model_forward_time": 3.1369216442108154,
      "step": 7858
    },
    {
      "epoch": 2.997589111328125e-05,
      "step": 7858,
      "training_step_time": 9.910667896270752
    },
    {
      "epoch": 2.997589111328125e-05,
      "model_forward_time": 3.1323964595794678,
      "step": 7858
    },
    {
      "epoch": 2.997589111328125e-05,
      "step": 7858,
      "training_step_time": 9.904239654541016
    },
    {
      "epoch": 2.997589111328125e-05,
      "model_forward_time": 3.1317615509033203,
      "step": 7858
    },
    {
      "epoch": 2.997589111328125e-05,
      "step": 7858,
      "training_step_time": 9.904017925262451
    },
    {
      "epoch": 2.997589111328125e-05,
      "model_forward_time": 3.1347992420196533,
      "step": 7858
    },
    {
      "epoch": 2.997589111328125e-05,
      "step": 7858,
      "training_step_time": 10.256329536437988
    },
    {
      "epoch": 2.9979705810546875e-05,
      "model_forward_time": 3.4170491695404053,
      "step": 7859
    },
    {
      "epoch": 2.9979705810546875e-05,
      "step": 7859,
      "training_step_time": 10.191746950149536
    },
    {
      "epoch": 2.9979705810546875e-05,
      "model_forward_time": 2.433854103088379,
      "step": 7859
    },
    {
      "epoch": 2.9979705810546875e-05,
      "step": 7859,
      "training_step_time": 7.469270706176758
    },
    {
      "epoch": 2.9979705810546875e-05,
      "model_forward_time": 3.134192943572998,
      "step": 7859
    },
    {
      "epoch": 2.9979705810546875e-05,
      "step": 7859,
      "training_step_time": 9.90853500366211
    },
    {
      "epoch": 2.9979705810546875e-05,
      "model_forward_time": 3.137341022491455,
      "step": 7859
    },
    {
      "epoch": 2.9979705810546875e-05,
      "step": 7859,
      "training_step_time": 10.30220890045166
    },
    {
      "epoch": 2.99835205078125e-05,
      "grad_norm": 0.4472799301147461,
      "learning_rate": 1e-05,
      "loss": 0.0805,
      "step": 7860
    },
    {
      "epoch": 2.99835205078125e-05,
      "model_forward_time": 3.155799627304077,
      "step": 7860
    },
    {
      "dynamics_loss_avg": 0.04665014185011387,
      "epoch": 2.99835205078125e-05,
      "step": 7860
    },
    {
      "action_loss_avg": 0.017338177748024465,
      "epoch": 2.99835205078125e-05,
      "step": 7860
    },
    {
      "epoch": 2.99835205078125e-05,
      "step": 7860,
      "training_step_time": 9.943536520004272
    },
    {
      "epoch": 2.99835205078125e-05,
      "model_forward_time": 3.13525390625,
      "step": 7860
    },
    {
      "epoch": 2.99835205078125e-05,
      "step": 7860,
      "training_step_time": 10.206360578536987
    },
    {
      "epoch": 2.99835205078125e-05,
      "model_forward_time": 3.1377062797546387,
      "step": 7860
    },
    {
      "epoch": 2.99835205078125e-05,
      "step": 7860,
      "training_step_time": 9.917607069015503
    },
    {
      "epoch": 2.99835205078125e-05,
      "model_forward_time": 3.1412322521209717,
      "step": 7860
    },
    {
      "epoch": 2.99835205078125e-05,
      "step": 7860,
      "training_step_time": 10.348296165466309
    },
    {
      "epoch": 2.9987335205078124e-05,
      "model_forward_time": 3.186284303665161,
      "step": 7861
    },
    {
      "epoch": 2.9987335205078124e-05,
      "step": 7861,
      "training_step_time": 9.964908361434937
    },
    {
      "epoch": 2.9987335205078124e-05,
      "model_forward_time": 3.1319286823272705,
      "step": 7861
    },
    {
      "epoch": 2.9987335205078124e-05,
      "step": 7861,
      "training_step_time": 9.909058094024658
    },
    {
      "epoch": 2.9987335205078124e-05,
      "model_forward_time": 3.4333531856536865,
      "step": 7861
    },
    {
      "epoch": 2.9987335205078124e-05,
      "step": 7861,
      "training_step_time": 10.216232061386108
    },
    {
      "epoch": 2.9987335205078124e-05,
      "model_forward_time": 2.441192150115967,
      "step": 7861
    },
    {
      "epoch": 2.9987335205078124e-05,
      "step": 7861,
      "training_step_time": 10.3403902053833
    },
    {
      "epoch": 2.999114990234375e-05,
      "model_forward_time": 3.1570053100585938,
      "step": 7862
    },
    {
      "epoch": 2.999114990234375e-05,
      "step": 7862,
      "training_step_time": 9.942258596420288
    },
    {
      "epoch": 2.999114990234375e-05,
      "model_forward_time": 3.1406846046447754,
      "step": 7862
    },
    {
      "epoch": 2.999114990234375e-05,
      "step": 7862,
      "training_step_time": 9.924675464630127
    },
    {
      "epoch": 2.999114990234375e-05,
      "model_forward_time": 3.144402503967285,
      "step": 7862
    },
    {
      "dynamics_loss_avg": 0.048185511492192745,
      "epoch": 2.999114990234375e-05,
      "step": 7862
    },
    {
      "action_loss_avg": 0.01670973962172866,
      "epoch": 2.999114990234375e-05,
      "step": 7862
    },
    {
      "epoch": 2.999114990234375e-05,
      "step": 7862,
      "training_step_time": 9.927178621292114
    },
    {
      "epoch": 2.999114990234375e-05,
      "model_forward_time": 2.436138391494751,
      "step": 7862
    },
    {
      "epoch": 2.999114990234375e-05,
      "step": 7862,
      "training_step_time": 10.341017484664917
    },
    {
      "epoch": 2.9994964599609377e-05,
      "model_forward_time": 3.4941914081573486,
      "step": 7863
    },
    {
      "epoch": 2.9994964599609377e-05,
      "step": 7863,
      "training_step_time": 10.274412155151367
    },
    {
      "epoch": 2.9994964599609377e-05,
      "model_forward_time": 3.13641619682312,
      "step": 7863
    },
    {
      "epoch": 2.9994964599609377e-05,
      "step": 7863,
      "training_step_time": 9.914469242095947
    },
    {
      "epoch": 2.9994964599609377e-05,
      "model_forward_time": 3.1417956352233887,
      "step": 7863
    },
    {
      "epoch": 2.9994964599609377e-05,
      "step": 7863,
      "training_step_time": 9.920346975326538
    },
    {
      "epoch": 2.9994964599609377e-05,
      "model_forward_time": 3.137098789215088,
      "step": 7863
    },
    {
      "epoch": 2.9994964599609377e-05,
      "step": 7863,
      "training_step_time": 10.315577030181885
    },
    {
      "epoch": 2.9998779296875e-05,
      "model_forward_time": 3.1473779678344727,
      "step": 7864
    },
    {
      "epoch": 2.9998779296875e-05,
      "step": 7864,
      "training_step_time": 10.198429584503174
    },
    {
      "epoch": 2.9998779296875e-05,
      "model_forward_time": 3.132957696914673,
      "step": 7864
    },
    {
      "epoch": 2.9998779296875e-05,
      "step": 7864,
      "training_step_time": 9.905662298202515
    },
    {
      "epoch": 2.9998779296875e-05,
      "model_forward_time": 3.1400375366210938,
      "step": 7864
    },
    {
      "epoch": 2.9998779296875e-05,
      "step": 7864,
      "training_step_time": 9.916204690933228
    },
    {
      "epoch": 2.9998779296875e-05,
      "model_forward_time": 3.1367874145507812,
      "step": 7864
    },
    {
      "epoch": 2.9998779296875e-05,
      "step": 7864,
      "training_step_time": 10.246683597564697
    },
    {
      "epoch": 3.0002593994140626e-05,
      "model_forward_time": 3.145174503326416,
      "step": 7865
    },
    {
      "dynamics_loss_avg": 0.0388224933296442,
      "epoch": 3.0002593994140626e-05,
      "step": 7865
    },
    {
      "action_loss_avg": 0.014228481939062477,
      "epoch": 3.0002593994140626e-05,
      "step": 7865
    },
    {
      "epoch": 3.0002593994140626e-05,
      "step": 7865,
      "training_step_time": 9.927904844284058
    },
    {
      "epoch": 3.0002593994140626e-05,
      "model_forward_time": 3.137408971786499,
      "step": 7865
    },
    {
      "epoch": 3.0002593994140626e-05,
      "step": 7865,
      "training_step_time": 10.188599824905396
    },
    {
      "epoch": 3.0002593994140626e-05,
      "model_forward_time": 3.1361770629882812,
      "step": 7865
    },
    {
      "epoch": 3.0002593994140626e-05,
      "step": 7865,
      "training_step_time": 9.910364151000977
    },
    {
      "epoch": 3.0002593994140626e-05,
      "model_forward_time": 3.1324303150177,
      "step": 7865
    },
    {
      "epoch": 3.0002593994140626e-05,
      "step": 7865,
      "training_step_time": 10.298717498779297
    },
    {
      "epoch": 3.000640869140625e-05,
      "model_forward_time": 3.156768321990967,
      "step": 7866
    },
    {
      "epoch": 3.000640869140625e-05,
      "step": 7866,
      "training_step_time": 9.927270889282227
    },
    {
      "epoch": 3.000640869140625e-05,
      "model_forward_time": 3.1339049339294434,
      "step": 7866
    },
    {
      "epoch": 3.000640869140625e-05,
      "step": 7866,
      "training_step_time": 9.905593395233154
    },
    {
      "epoch": 3.000640869140625e-05,
      "model_forward_time": 3.4196019172668457,
      "step": 7866
    },
    {
      "epoch": 3.000640869140625e-05,
      "step": 7866,
      "training_step_time": 10.188806533813477
    },
    {
      "epoch": 3.000640869140625e-05,
      "model_forward_time": 2.4371612071990967,
      "step": 7866
    },
    {
      "epoch": 3.000640869140625e-05,
      "step": 7866,
      "training_step_time": 10.276894569396973
    },
    {
      "epoch": 3.0010223388671875e-05,
      "model_forward_time": 3.150749921798706,
      "step": 7867
    },
    {
      "epoch": 3.0010223388671875e-05,
      "step": 7867,
      "training_step_time": 9.925206184387207
    },
    {
      "epoch": 3.0010223388671875e-05,
      "model_forward_time": 3.1371448040008545,
      "step": 7867
    },
    {
      "epoch": 3.0010223388671875e-05,
      "step": 7867,
      "training_step_time": 9.91062045097351
    },
    {
      "epoch": 3.0010223388671875e-05,
      "model_forward_time": 3.1383824348449707,
      "step": 7867
    },
    {
      "dynamics_loss_avg": 0.05765625927597284,
      "epoch": 3.0010223388671875e-05,
      "step": 7867
    },
    {
      "action_loss_avg": 0.017261400585994124,
      "epoch": 3.0010223388671875e-05,
      "step": 7867
    },
    {
      "epoch": 3.0010223388671875e-05,
      "step": 7867,
      "training_step_time": 9.928698539733887
    },
    {
      "epoch": 3.0010223388671875e-05,
      "model_forward_time": 3.1469204425811768,
      "step": 7867
    },
    {
      "epoch": 3.0010223388671875e-05,
      "step": 7867,
      "training_step_time": 10.338863372802734
    },
    {
      "epoch": 3.00140380859375e-05,
      "model_forward_time": 3.1638946533203125,
      "step": 7868
    },
    {
      "epoch": 3.00140380859375e-05,
      "step": 7868,
      "training_step_time": 9.943423748016357
    },
    {
      "epoch": 3.00140380859375e-05,
      "model_forward_time": 3.147587299346924,
      "step": 7868
    },
    {
      "epoch": 3.00140380859375e-05,
      "step": 7868,
      "training_step_time": 9.927016735076904
    },
    {
      "epoch": 3.00140380859375e-05,
      "model_forward_time": 3.1425039768218994,
      "step": 7868
    },
    {
      "epoch": 3.00140380859375e-05,
      "step": 7868,
      "training_step_time": 9.922208786010742
    },
    {
      "epoch": 3.00140380859375e-05,
      "model_forward_time": 3.1498687267303467,
      "step": 7868
    },
    {
      "epoch": 3.00140380859375e-05,
      "step": 7868,
      "training_step_time": 10.267142295837402
    },
    {
      "epoch": 3.0017852783203124e-05,
      "model_forward_time": 3.41973614692688,
      "step": 7869
    },
    {
      "epoch": 3.0017852783203124e-05,
      "step": 7869,
      "training_step_time": 10.202759265899658
    },
    {
      "epoch": 3.0017852783203124e-05,
      "model_forward_time": 3.144252300262451,
      "step": 7869
    },
    {
      "epoch": 3.0017852783203124e-05,
      "step": 7869,
      "training_step_time": 9.919275999069214
    },
    {
      "epoch": 3.0017852783203124e-05,
      "model_forward_time": 3.1434082984924316,
      "step": 7869
    },
    {
      "epoch": 3.0017852783203124e-05,
      "step": 7869,
      "training_step_time": 9.921949863433838
    },
    {
      "epoch": 3.0017852783203124e-05,
      "model_forward_time": 3.146341323852539,
      "step": 7869
    },
    {
      "epoch": 3.0017852783203124e-05,
      "step": 7869,
      "training_step_time": 10.021663427352905
    },
    {
      "epoch": 3.002166748046875e-05,
      "grad_norm": 0.49915245175361633,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 7870
    },
    {
      "epoch": 3.002166748046875e-05,
      "model_forward_time": 3.1618337631225586,
      "step": 7870
    },
    {
      "dynamics_loss_avg": 0.06881293058395385,
      "epoch": 3.002166748046875e-05,
      "step": 7870
    },
    {
      "action_loss_avg": 0.02612710166722536,
      "epoch": 3.002166748046875e-05,
      "step": 7870
    },
    {
      "epoch": 3.002166748046875e-05,
      "step": 7870,
      "training_step_time": 9.946556329727173
    },
    {
      "epoch": 3.002166748046875e-05,
      "model_forward_time": 3.4341022968292236,
      "step": 7870
    },
    {
      "epoch": 3.002166748046875e-05,
      "step": 7870,
      "training_step_time": 10.214415550231934
    },
    {
      "epoch": 3.002166748046875e-05,
      "model_forward_time": 3.1482455730438232,
      "step": 7870
    },
    {
      "epoch": 3.002166748046875e-05,
      "step": 7870,
      "training_step_time": 9.932868480682373
    },
    {
      "epoch": 3.002166748046875e-05,
      "model_forward_time": 3.146270513534546,
      "step": 7870
    },
    {
      "epoch": 3.002166748046875e-05,
      "step": 7870,
      "training_step_time": 10.303038120269775
    },
    {
      "epoch": 3.0025482177734377e-05,
      "model_forward_time": 3.1594114303588867,
      "step": 7871
    },
    {
      "epoch": 3.0025482177734377e-05,
      "step": 7871,
      "training_step_time": 9.943773984909058
    },
    {
      "epoch": 3.0025482177734377e-05,
      "model_forward_time": 2.4375858306884766,
      "step": 7871
    },
    {
      "epoch": 3.0025482177734377e-05,
      "step": 7871,
      "training_step_time": 7.46702766418457
    },
    {
      "epoch": 3.0025482177734377e-05,
      "model_forward_time": 3.149562120437622,
      "step": 7871
    },
    {
      "epoch": 3.0025482177734377e-05,
      "step": 7871,
      "training_step_time": 10.207804679870605
    },
    {
      "epoch": 3.0025482177734377e-05,
      "model_forward_time": 3.147289514541626,
      "step": 7871
    },
    {
      "epoch": 3.0025482177734377e-05,
      "step": 7871,
      "training_step_time": 10.265934705734253
    },
    {
      "epoch": 3.0029296875e-05,
      "model_forward_time": 3.157224178314209,
      "step": 7872
    },
    {
      "epoch": 3.0029296875e-05,
      "step": 7872,
      "training_step_time": 9.938419103622437
    },
    {
      "epoch": 3.0029296875e-05,
      "model_forward_time": 3.1404144763946533,
      "step": 7872
    },
    {
      "epoch": 3.0029296875e-05,
      "step": 7872,
      "training_step_time": 9.920350790023804
    },
    {
      "epoch": 3.0029296875e-05,
      "model_forward_time": 3.143709897994995,
      "step": 7872
    },
    {
      "dynamics_loss_avg": 0.06714977454394103,
      "epoch": 3.0029296875e-05,
      "step": 7872
    },
    {
      "action_loss_avg": 0.024388849409297107,
      "epoch": 3.0029296875e-05,
      "step": 7872
    },
    {
      "epoch": 3.0029296875e-05,
      "step": 7872,
      "training_step_time": 9.930214166641235
    },
    {
      "epoch": 3.0029296875e-05,
      "model_forward_time": 3.457737445831299,
      "step": 7872
    },
    {
      "epoch": 3.0029296875e-05,
      "step": 7872,
      "training_step_time": 10.332283020019531
    },
    {
      "epoch": 3.0033111572265626e-05,
      "model_forward_time": 3.161024808883667,
      "step": 7873
    },
    {
      "epoch": 3.0033111572265626e-05,
      "step": 7873,
      "training_step_time": 9.94478988647461
    },
    {
      "epoch": 3.0033111572265626e-05,
      "model_forward_time": 3.1420929431915283,
      "step": 7873
    },
    {
      "epoch": 3.0033111572265626e-05,
      "step": 7873,
      "training_step_time": 9.92079210281372
    },
    {
      "epoch": 3.0033111572265626e-05,
      "model_forward_time": 3.1510119438171387,
      "step": 7873
    },
    {
      "epoch": 3.0033111572265626e-05,
      "step": 7873,
      "training_step_time": 9.928627252578735
    },
    {
      "epoch": 3.0033111572265626e-05,
      "model_forward_time": 3.1442341804504395,
      "step": 7873
    },
    {
      "epoch": 3.0033111572265626e-05,
      "step": 7873,
      "training_step_time": 10.29195499420166
    },
    {
      "epoch": 3.003692626953125e-05,
      "model_forward_time": 3.1534323692321777,
      "step": 7874
    },
    {
      "epoch": 3.003692626953125e-05,
      "step": 7874,
      "training_step_time": 10.216018915176392
    },
    {
      "epoch": 3.003692626953125e-05,
      "model_forward_time": 3.137075424194336,
      "step": 7874
    },
    {
      "epoch": 3.003692626953125e-05,
      "step": 7874,
      "training_step_time": 9.915973901748657
    },
    {
      "epoch": 3.003692626953125e-05,
      "model_forward_time": 3.1415326595306396,
      "step": 7874
    },
    {
      "epoch": 3.003692626953125e-05,
      "step": 7874,
      "training_step_time": 9.92018985748291
    },
    {
      "epoch": 3.003692626953125e-05,
      "model_forward_time": 3.141604423522949,
      "step": 7874
    },
    {
      "epoch": 3.003692626953125e-05,
      "step": 7874,
      "training_step_time": 10.246983528137207
    },
    {
      "epoch": 3.0040740966796875e-05,
      "model_forward_time": 3.1612257957458496,
      "step": 7875
    },
    {
      "dynamics_loss_avg": 0.06845263056457043,
      "epoch": 3.0040740966796875e-05,
      "step": 7875
    },
    {
      "action_loss_avg": 0.027312128338962793,
      "epoch": 3.0040740966796875e-05,
      "step": 7875
    },
    {
      "epoch": 3.0040740966796875e-05,
      "step": 7875,
      "training_step_time": 9.938380241394043
    },
    {
      "epoch": 3.0040740966796875e-05,
      "model_forward_time": 3.400542974472046,
      "step": 7875
    },
    {
      "epoch": 3.0040740966796875e-05,
      "step": 7875,
      "training_step_time": 10.176911115646362
    },
    {
      "epoch": 3.0040740966796875e-05,
      "model_forward_time": 3.143547773361206,
      "step": 7875
    },
    {
      "epoch": 3.0040740966796875e-05,
      "step": 7875,
      "training_step_time": 9.917616844177246
    },
    {
      "epoch": 3.0040740966796875e-05,
      "model_forward_time": 3.1425845623016357,
      "step": 7875
    },
    {
      "epoch": 3.0040740966796875e-05,
      "step": 7875,
      "training_step_time": 10.265100717544556
    },
    {
      "epoch": 3.00445556640625e-05,
      "model_forward_time": 3.1491005420684814,
      "step": 7876
    },
    {
      "epoch": 3.00445556640625e-05,
      "step": 7876,
      "training_step_time": 9.927491664886475
    },
    {
      "epoch": 3.00445556640625e-05,
      "model_forward_time": 3.142402410507202,
      "step": 7876
    },
    {
      "epoch": 3.00445556640625e-05,
      "step": 7876,
      "training_step_time": 9.917712450027466
    },
    {
      "epoch": 3.00445556640625e-05,
      "model_forward_time": 3.425806760787964,
      "step": 7876
    },
    {
      "epoch": 3.00445556640625e-05,
      "step": 7876,
      "training_step_time": 10.199110746383667
    },
    {
      "epoch": 3.00445556640625e-05,
      "model_forward_time": 3.140444755554199,
      "step": 7876
    },
    {
      "epoch": 3.00445556640625e-05,
      "step": 7876,
      "training_step_time": 10.016405820846558
    },
    {
      "epoch": 3.0048370361328124e-05,
      "model_forward_time": 3.15859317779541,
      "step": 7877
    },
    {
      "epoch": 3.0048370361328124e-05,
      "step": 7877,
      "training_step_time": 9.93319821357727
    },
    {
      "epoch": 3.0048370361328124e-05,
      "model_forward_time": 3.139204740524292,
      "step": 7877
    },
    {
      "epoch": 3.0048370361328124e-05,
      "step": 7877,
      "training_step_time": 9.913464546203613
    },
    {
      "epoch": 3.0048370361328124e-05,
      "model_forward_time": 3.138103723526001,
      "step": 7877
    },
    {
      "dynamics_loss_avg": 0.07027407363057137,
      "epoch": 3.0048370361328124e-05,
      "step": 7877
    },
    {
      "action_loss_avg": 0.023058401234447957,
      "epoch": 3.0048370361328124e-05,
      "step": 7877
    },
    {
      "epoch": 3.0048370361328124e-05,
      "step": 7877,
      "training_step_time": 9.918864011764526
    },
    {
      "epoch": 3.0048370361328124e-05,
      "model_forward_time": 3.405900001525879,
      "step": 7877
    },
    {
      "epoch": 3.0048370361328124e-05,
      "step": 7877,
      "training_step_time": 10.278797149658203
    },
    {
      "epoch": 3.005218505859375e-05,
      "model_forward_time": 3.1528759002685547,
      "step": 7878
    },
    {
      "epoch": 3.005218505859375e-05,
      "step": 7878,
      "training_step_time": 9.931310653686523
    },
    {
      "epoch": 3.005218505859375e-05,
      "model_forward_time": 3.145176887512207,
      "step": 7878
    },
    {
      "epoch": 3.005218505859375e-05,
      "step": 7878,
      "training_step_time": 9.920622110366821
    },
    {
      "epoch": 3.005218505859375e-05,
      "model_forward_time": 3.1392948627471924,
      "step": 7878
    },
    {
      "epoch": 3.005218505859375e-05,
      "step": 7878,
      "training_step_time": 9.917308807373047
    },
    {
      "epoch": 3.005218505859375e-05,
      "model_forward_time": 3.1476612091064453,
      "step": 7878
    },
    {
      "epoch": 3.005218505859375e-05,
      "step": 7878,
      "training_step_time": 10.305108308792114
    },
    {
      "epoch": 3.0055999755859373e-05,
      "model_forward_time": 3.4203040599823,
      "step": 7879
    },
    {
      "epoch": 3.0055999755859373e-05,
      "step": 7879,
      "training_step_time": 10.199498176574707
    },
    {
      "epoch": 3.0055999755859373e-05,
      "model_forward_time": 3.150214433670044,
      "step": 7879
    },
    {
      "epoch": 3.0055999755859373e-05,
      "step": 7879,
      "training_step_time": 9.929666996002197
    },
    {
      "epoch": 3.0055999755859373e-05,
      "model_forward_time": 2.4394631385803223,
      "step": 7879
    },
    {
      "epoch": 3.0055999755859373e-05,
      "step": 7879,
      "training_step_time": 7.473649740219116
    },
    {
      "epoch": 3.0055999755859373e-05,
      "model_forward_time": 3.1411452293395996,
      "step": 7879
    },
    {
      "epoch": 3.0055999755859373e-05,
      "step": 7879,
      "training_step_time": 10.281715154647827
    },
    {
      "epoch": 3.0059814453125e-05,
      "grad_norm": 0.43285053968429565,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 7880
    },
    {
      "epoch": 3.0059814453125e-05,
      "model_forward_time": 3.1645069122314453,
      "step": 7880
    },
    {
      "dynamics_loss_avg": 0.0680248061195016,
      "epoch": 3.0059814453125e-05,
      "step": 7880
    },
    {
      "action_loss_avg": 0.024575406778603794,
      "epoch": 3.0059814453125e-05,
      "step": 7880
    },
    {
      "epoch": 3.0059814453125e-05,
      "step": 7880,
      "training_step_time": 9.952709674835205
    },
    {
      "epoch": 3.0059814453125e-05,
      "model_forward_time": 3.1455891132354736,
      "step": 7880
    },
    {
      "epoch": 3.0059814453125e-05,
      "step": 7880,
      "training_step_time": 10.202674865722656
    },
    {
      "epoch": 3.0059814453125e-05,
      "model_forward_time": 3.145296812057495,
      "step": 7880
    },
    {
      "epoch": 3.0059814453125e-05,
      "step": 7880,
      "training_step_time": 9.929723262786865
    },
    {
      "epoch": 3.0059814453125e-05,
      "model_forward_time": 3.14892840385437,
      "step": 7880
    },
    {
      "epoch": 3.0059814453125e-05,
      "step": 7880,
      "training_step_time": 10.307316064834595
    },
    {
      "epoch": 3.0063629150390626e-05,
      "model_forward_time": 2.4513912200927734,
      "step": 7881
    },
    {
      "epoch": 3.0063629150390626e-05,
      "step": 7881,
      "training_step_time": 7.484388589859009
    },
    {
      "epoch": 3.0063629150390626e-05,
      "model_forward_time": 3.138861894607544,
      "step": 7881
    },
    {
      "epoch": 3.0063629150390626e-05,
      "step": 7881,
      "training_step_time": 9.916560888290405
    },
    {
      "epoch": 3.0063629150390626e-05,
      "model_forward_time": 3.147702693939209,
      "step": 7881
    },
    {
      "epoch": 3.0063629150390626e-05,
      "step": 7881,
      "training_step_time": 9.929969549179077
    },
    {
      "epoch": 3.0063629150390626e-05,
      "model_forward_time": 3.4503469467163086,
      "step": 7881
    },
    {
      "epoch": 3.0063629150390626e-05,
      "step": 7881,
      "training_step_time": 10.326704502105713
    },
    {
      "epoch": 3.006744384765625e-05,
      "model_forward_time": 3.165203332901001,
      "step": 7882
    },
    {
      "epoch": 3.006744384765625e-05,
      "step": 7882,
      "training_step_time": 9.947420120239258
    },
    {
      "epoch": 3.006744384765625e-05,
      "model_forward_time": 3.144714832305908,
      "step": 7882
    },
    {
      "epoch": 3.006744384765625e-05,
      "step": 7882,
      "training_step_time": 9.924656629562378
    },
    {
      "epoch": 3.006744384765625e-05,
      "model_forward_time": 3.1483147144317627,
      "step": 7882
    },
    {
      "dynamics_loss_avg": 0.05491674523800612,
      "epoch": 3.006744384765625e-05,
      "step": 7882
    },
    {
      "action_loss_avg": 0.01847775634378195,
      "epoch": 3.006744384765625e-05,
      "step": 7882
    },
    {
      "epoch": 3.006744384765625e-05,
      "step": 7882,
      "training_step_time": 9.92885684967041
    },
    {
      "epoch": 3.006744384765625e-05,
      "model_forward_time": 3.427757501602173,
      "step": 7882
    },
    {
      "epoch": 3.006744384765625e-05,
      "step": 7882,
      "training_step_time": 10.30005431175232
    },
    {
      "epoch": 3.0071258544921875e-05,
      "model_forward_time": 3.1591482162475586,
      "step": 7883
    },
    {
      "epoch": 3.0071258544921875e-05,
      "step": 7883,
      "training_step_time": 9.943136930465698
    },
    {
      "epoch": 3.0071258544921875e-05,
      "model_forward_time": 3.1395680904388428,
      "step": 7883
    },
    {
      "epoch": 3.0071258544921875e-05,
      "step": 7883,
      "training_step_time": 9.917030811309814
    },
    {
      "epoch": 3.0071258544921875e-05,
      "model_forward_time": 3.145012617111206,
      "step": 7883
    },
    {
      "epoch": 3.0071258544921875e-05,
      "step": 7883,
      "training_step_time": 9.921085834503174
    },
    {
      "epoch": 3.0071258544921875e-05,
      "model_forward_time": 3.1413328647613525,
      "step": 7883
    },
    {
      "epoch": 3.0071258544921875e-05,
      "step": 7883,
      "training_step_time": 10.33116626739502
    },
    {
      "epoch": 3.00750732421875e-05,
      "model_forward_time": 3.43947696685791,
      "step": 7884
    },
    {
      "epoch": 3.00750732421875e-05,
      "step": 7884,
      "training_step_time": 10.21593976020813
    },
    {
      "epoch": 3.00750732421875e-05,
      "model_forward_time": 3.1405673027038574,
      "step": 7884
    },
    {
      "epoch": 3.00750732421875e-05,
      "step": 7884,
      "training_step_time": 9.919735193252563
    },
    {
      "epoch": 3.00750732421875e-05,
      "model_forward_time": 3.1457903385162354,
      "step": 7884
    },
    {
      "epoch": 3.00750732421875e-05,
      "step": 7884,
      "training_step_time": 9.925078392028809
    },
    {
      "epoch": 3.00750732421875e-05,
      "model_forward_time": 3.1416587829589844,
      "step": 7884
    },
    {
      "epoch": 3.00750732421875e-05,
      "step": 7884,
      "training_step_time": 10.250006437301636
    },
    {
      "epoch": 3.0078887939453124e-05,
      "model_forward_time": 3.1501388549804688,
      "step": 7885
    },
    {
      "dynamics_loss_avg": 0.057518668659031394,
      "epoch": 3.0078887939453124e-05,
      "step": 7885
    },
    {
      "action_loss_avg": 0.018855396751314403,
      "epoch": 3.0078887939453124e-05,
      "step": 7885
    },
    {
      "epoch": 3.0078887939453124e-05,
      "step": 7885,
      "training_step_time": 9.92963433265686
    },
    {
      "epoch": 3.0078887939453124e-05,
      "model_forward_time": 3.3981828689575195,
      "step": 7885
    },
    {
      "epoch": 3.0078887939453124e-05,
      "step": 7885,
      "training_step_time": 10.167330026626587
    },
    {
      "epoch": 3.0078887939453124e-05,
      "model_forward_time": 3.1418025493621826,
      "step": 7885
    },
    {
      "epoch": 3.0078887939453124e-05,
      "step": 7885,
      "training_step_time": 9.909995555877686
    },
    {
      "epoch": 3.0078887939453124e-05,
      "model_forward_time": 3.135361671447754,
      "step": 7885
    },
    {
      "epoch": 3.0078887939453124e-05,
      "step": 7885,
      "training_step_time": 10.25749945640564
    },
    {
      "epoch": 3.008270263671875e-05,
      "model_forward_time": 3.158900499343872,
      "step": 7886
    },
    {
      "epoch": 3.008270263671875e-05,
      "step": 7886,
      "training_step_time": 9.931625604629517
    },
    {
      "epoch": 3.008270263671875e-05,
      "model_forward_time": 3.1379129886627197,
      "step": 7886
    },
    {
      "epoch": 3.008270263671875e-05,
      "step": 7886,
      "training_step_time": 9.903904914855957
    },
    {
      "epoch": 3.008270263671875e-05,
      "model_forward_time": 3.416691541671753,
      "step": 7886
    },
    {
      "epoch": 3.008270263671875e-05,
      "step": 7886,
      "training_step_time": 10.18543004989624
    },
    {
      "epoch": 3.008270263671875e-05,
      "model_forward_time": 3.139655828475952,
      "step": 7886
    },
    {
      "epoch": 3.008270263671875e-05,
      "step": 7886,
      "training_step_time": 10.292649269104004
    },
    {
      "epoch": 3.0086517333984373e-05,
      "model_forward_time": 3.156660795211792,
      "step": 7887
    },
    {
      "epoch": 3.0086517333984373e-05,
      "step": 7887,
      "training_step_time": 9.941208362579346
    },
    {
      "epoch": 3.0086517333984373e-05,
      "model_forward_time": 3.139941692352295,
      "step": 7887
    },
    {
      "epoch": 3.0086517333984373e-05,
      "step": 7887,
      "training_step_time": 9.910561800003052
    },
    {
      "epoch": 3.0086517333984373e-05,
      "model_forward_time": 3.131984233856201,
      "step": 7887
    },
    {
      "dynamics_loss_avg": 0.05075706345960498,
      "epoch": 3.0086517333984373e-05,
      "step": 7887
    },
    {
      "action_loss_avg": 0.01701600437518209,
      "epoch": 3.0086517333984373e-05,
      "step": 7887
    },
    {
      "epoch": 3.0086517333984373e-05,
      "step": 7887,
      "training_step_time": 9.90691351890564
    },
    {
      "epoch": 3.0086517333984373e-05,
      "model_forward_time": 3.3923444747924805,
      "step": 7887
    },
    {
      "epoch": 3.0086517333984373e-05,
      "step": 7887,
      "training_step_time": 10.260413646697998
    },
    {
      "epoch": 3.009033203125e-05,
      "model_forward_time": 3.1477267742156982,
      "step": 7888
    },
    {
      "epoch": 3.009033203125e-05,
      "step": 7888,
      "training_step_time": 9.92340874671936
    },
    {
      "epoch": 3.009033203125e-05,
      "model_forward_time": 3.1374611854553223,
      "step": 7888
    },
    {
      "epoch": 3.009033203125e-05,
      "step": 7888,
      "training_step_time": 9.910548448562622
    },
    {
      "epoch": 3.009033203125e-05,
      "model_forward_time": 3.1336045265197754,
      "step": 7888
    },
    {
      "epoch": 3.009033203125e-05,
      "step": 7888,
      "training_step_time": 9.906556844711304
    },
    {
      "epoch": 3.009033203125e-05,
      "model_forward_time": 3.1345460414886475,
      "step": 7888
    },
    {
      "epoch": 3.009033203125e-05,
      "step": 7888,
      "training_step_time": 10.274353265762329
    },
    {
      "epoch": 3.0094146728515626e-05,
      "model_forward_time": 3.3987793922424316,
      "step": 7889
    },
    {
      "epoch": 3.0094146728515626e-05,
      "step": 7889,
      "training_step_time": 10.175390481948853
    },
    {
      "epoch": 3.0094146728515626e-05,
      "model_forward_time": 3.1396729946136475,
      "step": 7889
    },
    {
      "epoch": 3.0094146728515626e-05,
      "step": 7889,
      "training_step_time": 9.914528369903564
    },
    {
      "epoch": 3.0094146728515626e-05,
      "model_forward_time": 2.433122158050537,
      "step": 7889
    },
    {
      "epoch": 3.0094146728515626e-05,
      "step": 7889,
      "training_step_time": 7.463496208190918
    },
    {
      "epoch": 3.0094146728515626e-05,
      "model_forward_time": 3.1368935108184814,
      "step": 7889
    },
    {
      "epoch": 3.0094146728515626e-05,
      "step": 7889,
      "training_step_time": 10.309985637664795
    },
    {
      "epoch": 3.009796142578125e-05,
      "grad_norm": 0.3386217951774597,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 7890
    },
    {
      "epoch": 3.009796142578125e-05,
      "model_forward_time": 3.1552021503448486,
      "step": 7890
    },
    {
      "dynamics_loss_avg": 0.06282064057886601,
      "epoch": 3.009796142578125e-05,
      "step": 7890
    },
    {
      "action_loss_avg": 0.02139469878748059,
      "epoch": 3.009796142578125e-05,
      "step": 7890
    },
    {
      "epoch": 3.009796142578125e-05,
      "step": 7890,
      "training_step_time": 9.93471384048462
    },
    {
      "epoch": 3.009796142578125e-05,
      "model_forward_time": 3.4143097400665283,
      "step": 7890
    },
    {
      "epoch": 3.009796142578125e-05,
      "step": 7890,
      "training_step_time": 10.188944101333618
    },
    {
      "epoch": 3.009796142578125e-05,
      "model_forward_time": 3.133150815963745,
      "step": 7890
    },
    {
      "epoch": 3.009796142578125e-05,
      "step": 7890,
      "training_step_time": 9.904558181762695
    },
    {
      "epoch": 3.009796142578125e-05,
      "model_forward_time": 3.13909649848938,
      "step": 7890
    },
    {
      "epoch": 3.009796142578125e-05,
      "step": 7890,
      "training_step_time": 10.250218152999878
    },
    {
      "epoch": 3.0101776123046875e-05,
      "model_forward_time": 3.142836809158325,
      "step": 7891
    },
    {
      "epoch": 3.0101776123046875e-05,
      "step": 7891,
      "training_step_time": 9.91483998298645
    },
    {
      "epoch": 3.0101776123046875e-05,
      "model_forward_time": 3.131354331970215,
      "step": 7891
    },
    {
      "epoch": 3.0101776123046875e-05,
      "step": 7891,
      "training_step_time": 9.90986704826355
    },
    {
      "epoch": 3.0101776123046875e-05,
      "model_forward_time": 3.1336605548858643,
      "step": 7891
    },
    {
      "epoch": 3.0101776123046875e-05,
      "step": 7891,
      "training_step_time": 10.187921047210693
    },
    {
      "epoch": 3.0101776123046875e-05,
      "model_forward_time": 3.1413090229034424,
      "step": 7891
    },
    {
      "epoch": 3.0101776123046875e-05,
      "step": 7891,
      "training_step_time": 10.321795225143433
    },
    {
      "epoch": 3.01055908203125e-05,
      "model_forward_time": 3.148214340209961,
      "step": 7892
    },
    {
      "epoch": 3.01055908203125e-05,
      "step": 7892,
      "training_step_time": 9.926036834716797
    },
    {
      "epoch": 3.01055908203125e-05,
      "model_forward_time": 3.135650157928467,
      "step": 7892
    },
    {
      "epoch": 3.01055908203125e-05,
      "step": 7892,
      "training_step_time": 9.911898136138916
    },
    {
      "epoch": 3.01055908203125e-05,
      "model_forward_time": 3.1397056579589844,
      "step": 7892
    },
    {
      "dynamics_loss_avg": 0.04679475715383887,
      "epoch": 3.01055908203125e-05,
      "step": 7892
    },
    {
      "action_loss_avg": 0.019477189588360488,
      "epoch": 3.01055908203125e-05,
      "step": 7892
    },
    {
      "epoch": 3.01055908203125e-05,
      "step": 7892,
      "training_step_time": 9.915242195129395
    },
    {
      "epoch": 3.01055908203125e-05,
      "model_forward_time": 3.4725704193115234,
      "step": 7892
    },
    {
      "epoch": 3.01055908203125e-05,
      "step": 7892,
      "training_step_time": 10.342577934265137
    },
    {
      "epoch": 3.0109405517578124e-05,
      "model_forward_time": 3.1582860946655273,
      "step": 7893
    },
    {
      "epoch": 3.0109405517578124e-05,
      "step": 7893,
      "training_step_time": 9.932120561599731
    },
    {
      "epoch": 3.0109405517578124e-05,
      "model_forward_time": 3.136672258377075,
      "step": 7893
    },
    {
      "epoch": 3.0109405517578124e-05,
      "step": 7893,
      "training_step_time": 9.905714511871338
    },
    {
      "epoch": 3.0109405517578124e-05,
      "model_forward_time": 3.140266180038452,
      "step": 7893
    },
    {
      "epoch": 3.0109405517578124e-05,
      "step": 7893,
      "training_step_time": 9.905440092086792
    },
    {
      "epoch": 3.0109405517578124e-05,
      "model_forward_time": 3.1325578689575195,
      "step": 7893
    },
    {
      "epoch": 3.0109405517578124e-05,
      "step": 7893,
      "training_step_time": 10.30732536315918
    },
    {
      "epoch": 3.011322021484375e-05,
      "model_forward_time": 3.144040107727051,
      "step": 7894
    },
    {
      "epoch": 3.011322021484375e-05,
      "step": 7894,
      "training_step_time": 10.18565845489502
    },
    {
      "epoch": 3.011322021484375e-05,
      "model_forward_time": 3.1347899436950684,
      "step": 7894
    },
    {
      "epoch": 3.011322021484375e-05,
      "step": 7894,
      "training_step_time": 9.905306339263916
    },
    {
      "epoch": 3.011322021484375e-05,
      "model_forward_time": 3.1406702995300293,
      "step": 7894
    },
    {
      "epoch": 3.011322021484375e-05,
      "step": 7894,
      "training_step_time": 9.909137964248657
    },
    {
      "epoch": 3.011322021484375e-05,
      "model_forward_time": 3.1344194412231445,
      "step": 7894
    },
    {
      "epoch": 3.011322021484375e-05,
      "step": 7894,
      "training_step_time": 10.26797866821289
    },
    {
      "epoch": 3.0117034912109373e-05,
      "model_forward_time": 3.146740436553955,
      "step": 7895
    },
    {
      "dynamics_loss_avg": 0.0673151295632124,
      "epoch": 3.0117034912109373e-05,
      "step": 7895
    },
    {
      "action_loss_avg": 0.023287216341122983,
      "epoch": 3.0117034912109373e-05,
      "step": 7895
    },
    {
      "epoch": 3.0117034912109373e-05,
      "step": 7895,
      "training_step_time": 9.918546438217163
    },
    {
      "epoch": 3.0117034912109373e-05,
      "model_forward_time": 3.410959482192993,
      "step": 7895
    },
    {
      "epoch": 3.0117034912109373e-05,
      "step": 7895,
      "training_step_time": 10.177309274673462
    },
    {
      "epoch": 3.0117034912109373e-05,
      "model_forward_time": 3.1397690773010254,
      "step": 7895
    },
    {
      "epoch": 3.0117034912109373e-05,
      "step": 7895,
      "training_step_time": 9.909781694412231
    },
    {
      "epoch": 3.0117034912109373e-05,
      "model_forward_time": 3.137795925140381,
      "step": 7895
    },
    {
      "epoch": 3.0117034912109373e-05,
      "step": 7895,
      "training_step_time": 10.016347885131836
    },
    {
      "epoch": 3.0120849609375e-05,
      "model_forward_time": 3.1549794673919678,
      "step": 7896
    },
    {
      "epoch": 3.0120849609375e-05,
      "step": 7896,
      "training_step_time": 9.93028473854065
    },
    {
      "epoch": 3.0120849609375e-05,
      "model_forward_time": 3.138730049133301,
      "step": 7896
    },
    {
      "epoch": 3.0120849609375e-05,
      "step": 7896,
      "training_step_time": 9.911670684814453
    },
    {
      "epoch": 3.0120849609375e-05,
      "model_forward_time": 3.4258124828338623,
      "step": 7896
    },
    {
      "epoch": 3.0120849609375e-05,
      "step": 7896,
      "training_step_time": 10.200486183166504
    },
    {
      "epoch": 3.0120849609375e-05,
      "model_forward_time": 3.1416444778442383,
      "step": 7896
    },
    {
      "epoch": 3.0120849609375e-05,
      "step": 7896,
      "training_step_time": 10.272161722183228
    },
    {
      "epoch": 3.0124664306640626e-05,
      "model_forward_time": 3.166534900665283,
      "step": 7897
    },
    {
      "epoch": 3.0124664306640626e-05,
      "step": 7897,
      "training_step_time": 9.948108911514282
    },
    {
      "epoch": 3.0124664306640626e-05,
      "model_forward_time": 3.1457600593566895,
      "step": 7897
    },
    {
      "epoch": 3.0124664306640626e-05,
      "step": 7897,
      "training_step_time": 9.9252769947052
    },
    {
      "epoch": 3.0124664306640626e-05,
      "model_forward_time": 3.144272565841675,
      "step": 7897
    },
    {
      "dynamics_loss_avg": 0.04333528485149145,
      "epoch": 3.0124664306640626e-05,
      "step": 7897
    },
    {
      "action_loss_avg": 0.016131171071901918,
      "epoch": 3.0124664306640626e-05,
      "step": 7897
    },
    {
      "epoch": 3.0124664306640626e-05,
      "step": 7897,
      "training_step_time": 9.927875518798828
    },
    {
      "epoch": 3.0124664306640626e-05,
      "model_forward_time": 3.427901268005371,
      "step": 7897
    },
    {
      "epoch": 3.0124664306640626e-05,
      "step": 7897,
      "training_step_time": 10.288258075714111
    },
    {
      "epoch": 3.012847900390625e-05,
      "model_forward_time": 3.1561970710754395,
      "step": 7898
    },
    {
      "epoch": 3.012847900390625e-05,
      "step": 7898,
      "training_step_time": 9.937167406082153
    },
    {
      "epoch": 3.012847900390625e-05,
      "model_forward_time": 3.144871950149536,
      "step": 7898
    },
    {
      "epoch": 3.012847900390625e-05,
      "step": 7898,
      "training_step_time": 9.927753686904907
    },
    {
      "epoch": 3.012847900390625e-05,
      "model_forward_time": 3.1428470611572266,
      "step": 7898
    },
    {
      "epoch": 3.012847900390625e-05,
      "step": 7898,
      "training_step_time": 9.917133808135986
    },
    {
      "epoch": 3.012847900390625e-05,
      "model_forward_time": 3.144352436065674,
      "step": 7898
    },
    {
      "epoch": 3.012847900390625e-05,
      "step": 7898,
      "training_step_time": 10.261468410491943
    },
    {
      "epoch": 3.0132293701171875e-05,
      "model_forward_time": 3.467672348022461,
      "step": 7899
    },
    {
      "epoch": 3.0132293701171875e-05,
      "step": 7899,
      "training_step_time": 10.246487140655518
    },
    {
      "epoch": 3.0132293701171875e-05,
      "model_forward_time": 3.148289203643799,
      "step": 7899
    },
    {
      "epoch": 3.0132293701171875e-05,
      "step": 7899,
      "training_step_time": 9.924497604370117
    },
    {
      "epoch": 3.0132293701171875e-05,
      "model_forward_time": 3.1468870639801025,
      "step": 7899
    },
    {
      "epoch": 3.0132293701171875e-05,
      "step": 7899,
      "training_step_time": 9.918868064880371
    },
    {
      "epoch": 3.0132293701171875e-05,
      "model_forward_time": 3.1564278602600098,
      "step": 7899
    },
    {
      "epoch": 3.0132293701171875e-05,
      "step": 7899,
      "training_step_time": 10.299128532409668
    },
    {
      "epoch": 3.01361083984375e-05,
      "grad_norm": 0.2807524502277374,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7900
    },
    {
      "epoch": 3.01361083984375e-05,
      "model_forward_time": 3.161726474761963,
      "step": 7900
    },
    {
      "dynamics_loss_avg": 0.05806278623640537,
      "epoch": 3.01361083984375e-05,
      "step": 7900
    },
    {
      "action_loss_avg": 0.017662509670481087,
      "epoch": 3.01361083984375e-05,
      "step": 7900
    },
    {
      "epoch": 3.01361083984375e-05,
      "step": 7900,
      "training_step_time": 9.948549747467041
    },
    {
      "epoch": 3.01361083984375e-05,
      "model_forward_time": 3.1485724449157715,
      "step": 7900
    },
    {
      "epoch": 3.01361083984375e-05,
      "step": 7900,
      "training_step_time": 10.223471879959106
    },
    {
      "epoch": 3.01361083984375e-05,
      "model_forward_time": 3.1521921157836914,
      "step": 7900
    },
    {
      "epoch": 3.01361083984375e-05,
      "step": 7900,
      "training_step_time": 9.929136753082275
    },
    {
      "epoch": 3.01361083984375e-05,
      "model_forward_time": 3.15415358543396,
      "step": 7900
    },
    {
      "epoch": 3.01361083984375e-05,
      "step": 7900,
      "training_step_time": 10.024492502212524
    },
    {
      "epoch": 3.0139923095703124e-05,
      "model_forward_time": 3.156562328338623,
      "step": 7901
    },
    {
      "epoch": 3.0139923095703124e-05,
      "step": 7901,
      "training_step_time": 9.937196493148804
    },
    {
      "epoch": 3.0139923095703124e-05,
      "model_forward_time": 3.1471335887908936,
      "step": 7901
    },
    {
      "epoch": 3.0139923095703124e-05,
      "step": 7901,
      "training_step_time": 9.922900199890137
    },
    {
      "epoch": 3.0139923095703124e-05,
      "model_forward_time": 3.415322780609131,
      "step": 7901
    },
    {
      "epoch": 3.0139923095703124e-05,
      "step": 7901,
      "training_step_time": 10.185156106948853
    },
    {
      "epoch": 3.0139923095703124e-05,
      "model_forward_time": 3.147373914718628,
      "step": 7901
    },
    {
      "epoch": 3.0139923095703124e-05,
      "step": 7901,
      "training_step_time": 10.248241186141968
    },
    {
      "epoch": 3.014373779296875e-05,
      "model_forward_time": 3.1604740619659424,
      "step": 7902
    },
    {
      "epoch": 3.014373779296875e-05,
      "step": 7902,
      "training_step_time": 9.931304931640625
    },
    {
      "epoch": 3.014373779296875e-05,
      "model_forward_time": 2.4348390102386475,
      "step": 7902
    },
    {
      "epoch": 3.014373779296875e-05,
      "step": 7902,
      "training_step_time": 7.46219801902771
    },
    {
      "epoch": 3.014373779296875e-05,
      "model_forward_time": 3.1396517753601074,
      "step": 7902
    },
    {
      "dynamics_loss_avg": 0.048469052836298945,
      "epoch": 3.014373779296875e-05,
      "step": 7902
    },
    {
      "action_loss_avg": 0.01925262496806681,
      "epoch": 3.014373779296875e-05,
      "step": 7902
    },
    {
      "epoch": 3.014373779296875e-05,
      "step": 7902,
      "training_step_time": 9.914815425872803
    },
    {
      "epoch": 3.014373779296875e-05,
      "model_forward_time": 3.1433279514312744,
      "step": 7902
    },
    {
      "epoch": 3.014373779296875e-05,
      "step": 7902,
      "training_step_time": 10.309680938720703
    },
    {
      "epoch": 3.0147552490234373e-05,
      "model_forward_time": 3.152890682220459,
      "step": 7903
    },
    {
      "epoch": 3.0147552490234373e-05,
      "step": 7903,
      "training_step_time": 9.922661781311035
    },
    {
      "epoch": 3.0147552490234373e-05,
      "model_forward_time": 3.1328048706054688,
      "step": 7903
    },
    {
      "epoch": 3.0147552490234373e-05,
      "step": 7903,
      "training_step_time": 9.901681661605835
    },
    {
      "epoch": 3.0147552490234373e-05,
      "model_forward_time": 3.140794515609741,
      "step": 7903
    },
    {
      "epoch": 3.0147552490234373e-05,
      "step": 7903,
      "training_step_time": 9.910073280334473
    },
    {
      "epoch": 3.0147552490234373e-05,
      "model_forward_time": 3.137336015701294,
      "step": 7903
    },
    {
      "epoch": 3.0147552490234373e-05,
      "step": 7903,
      "training_step_time": 10.266032934188843
    },
    {
      "epoch": 3.01513671875e-05,
      "model_forward_time": 3.439221143722534,
      "step": 7904
    },
    {
      "epoch": 3.01513671875e-05,
      "step": 7904,
      "training_step_time": 10.211351871490479
    },
    {
      "epoch": 3.01513671875e-05,
      "model_forward_time": 2.4340157508850098,
      "step": 7904
    },
    {
      "epoch": 3.01513671875e-05,
      "step": 7904,
      "training_step_time": 7.459593057632446
    },
    {
      "epoch": 3.01513671875e-05,
      "model_forward_time": 3.136155128479004,
      "step": 7904
    },
    {
      "epoch": 3.01513671875e-05,
      "step": 7904,
      "training_step_time": 9.898320436477661
    },
    {
      "epoch": 3.01513671875e-05,
      "model_forward_time": 3.1403841972351074,
      "step": 7904
    },
    {
      "epoch": 3.01513671875e-05,
      "step": 7904,
      "training_step_time": 10.329889059066772
    },
    {
      "epoch": 3.0155181884765626e-05,
      "model_forward_time": 3.1463894844055176,
      "step": 7905
    },
    {
      "dynamics_loss_avg": 0.06773067004978657,
      "epoch": 3.0155181884765626e-05,
      "step": 7905
    },
    {
      "action_loss_avg": 0.020325743267312645,
      "epoch": 3.0155181884765626e-05,
      "step": 7905
    },
    {
      "epoch": 3.0155181884765626e-05,
      "step": 7905,
      "training_step_time": 9.928982973098755
    },
    {
      "epoch": 3.0155181884765626e-05,
      "model_forward_time": 3.135035276412964,
      "step": 7905
    },
    {
      "epoch": 3.0155181884765626e-05,
      "step": 7905,
      "training_step_time": 10.176074981689453
    },
    {
      "epoch": 3.0155181884765626e-05,
      "model_forward_time": 3.1381747722625732,
      "step": 7905
    },
    {
      "epoch": 3.0155181884765626e-05,
      "step": 7905,
      "training_step_time": 9.90188479423523
    },
    {
      "epoch": 3.0155181884765626e-05,
      "model_forward_time": 3.1334307193756104,
      "step": 7905
    },
    {
      "epoch": 3.0155181884765626e-05,
      "step": 7905,
      "training_step_time": 10.27223014831543
    },
    {
      "epoch": 3.015899658203125e-05,
      "model_forward_time": 3.154609441757202,
      "step": 7906
    },
    {
      "epoch": 3.015899658203125e-05,
      "step": 7906,
      "training_step_time": 9.927453517913818
    },
    {
      "epoch": 3.015899658203125e-05,
      "model_forward_time": 3.1391913890838623,
      "step": 7906
    },
    {
      "epoch": 3.015899658203125e-05,
      "step": 7906,
      "training_step_time": 9.912709474563599
    },
    {
      "epoch": 3.015899658203125e-05,
      "model_forward_time": 3.421691417694092,
      "step": 7906
    },
    {
      "epoch": 3.015899658203125e-05,
      "step": 7906,
      "training_step_time": 10.192353963851929
    },
    {
      "epoch": 3.015899658203125e-05,
      "model_forward_time": 3.1386499404907227,
      "step": 7906
    },
    {
      "epoch": 3.015899658203125e-05,
      "step": 7906,
      "training_step_time": 10.030312538146973
    },
    {
      "epoch": 3.0162811279296875e-05,
      "model_forward_time": 3.148052453994751,
      "step": 7907
    },
    {
      "epoch": 3.0162811279296875e-05,
      "step": 7907,
      "training_step_time": 9.921631813049316
    },
    {
      "epoch": 3.0162811279296875e-05,
      "model_forward_time": 3.1407785415649414,
      "step": 7907
    },
    {
      "epoch": 3.0162811279296875e-05,
      "step": 7907,
      "training_step_time": 9.916867017745972
    },
    {
      "epoch": 3.0162811279296875e-05,
      "model_forward_time": 3.140056848526001,
      "step": 7907
    },
    {
      "dynamics_loss_avg": 0.053038527630269526,
      "epoch": 3.0162811279296875e-05,
      "step": 7907
    },
    {
      "action_loss_avg": 0.018806374724954368,
      "epoch": 3.0162811279296875e-05,
      "step": 7907
    },
    {
      "epoch": 3.0162811279296875e-05,
      "step": 7907,
      "training_step_time": 9.913626909255981
    },
    {
      "epoch": 3.0162811279296875e-05,
      "model_forward_time": 3.401477098464966,
      "step": 7907
    },
    {
      "epoch": 3.0162811279296875e-05,
      "step": 7907,
      "training_step_time": 10.304988384246826
    },
    {
      "epoch": 3.01666259765625e-05,
      "model_forward_time": 3.1564345359802246,
      "step": 7908
    },
    {
      "epoch": 3.01666259765625e-05,
      "step": 7908,
      "training_step_time": 9.931924104690552
    },
    {
      "epoch": 3.01666259765625e-05,
      "model_forward_time": 3.1399953365325928,
      "step": 7908
    },
    {
      "epoch": 3.01666259765625e-05,
      "step": 7908,
      "training_step_time": 9.91582202911377
    },
    {
      "epoch": 3.01666259765625e-05,
      "model_forward_time": 3.139941692352295,
      "step": 7908
    },
    {
      "epoch": 3.01666259765625e-05,
      "step": 7908,
      "training_step_time": 9.9153892993927
    },
    {
      "epoch": 3.01666259765625e-05,
      "model_forward_time": 3.1425771713256836,
      "step": 7908
    },
    {
      "epoch": 3.01666259765625e-05,
      "step": 7908,
      "training_step_time": 10.308482646942139
    },
    {
      "epoch": 3.0170440673828124e-05,
      "model_forward_time": 3.409662961959839,
      "step": 7909
    },
    {
      "epoch": 3.0170440673828124e-05,
      "step": 7909,
      "training_step_time": 10.182862520217896
    },
    {
      "epoch": 3.0170440673828124e-05,
      "model_forward_time": 3.1415345668792725,
      "step": 7909
    },
    {
      "epoch": 3.0170440673828124e-05,
      "step": 7909,
      "training_step_time": 9.920748710632324
    },
    {
      "epoch": 3.0170440673828124e-05,
      "model_forward_time": 3.1427042484283447,
      "step": 7909
    },
    {
      "epoch": 3.0170440673828124e-05,
      "step": 7909,
      "training_step_time": 9.922547578811646
    },
    {
      "epoch": 3.0170440673828124e-05,
      "model_forward_time": 3.1415233612060547,
      "step": 7909
    },
    {
      "epoch": 3.0170440673828124e-05,
      "step": 7909,
      "training_step_time": 10.349963188171387
    },
    {
      "epoch": 3.017425537109375e-05,
      "grad_norm": 0.35840198397636414,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 7910
    },
    {
      "epoch": 3.017425537109375e-05,
      "model_forward_time": 3.1617791652679443,
      "step": 7910
    },
    {
      "dynamics_loss_avg": 0.05255282437428832,
      "epoch": 3.017425537109375e-05,
      "step": 7910
    },
    {
      "action_loss_avg": 0.02151231127791107,
      "epoch": 3.017425537109375e-05,
      "step": 7910
    },
    {
      "epoch": 3.017425537109375e-05,
      "step": 7910,
      "training_step_time": 9.94251537322998
    },
    {
      "epoch": 3.017425537109375e-05,
      "model_forward_time": 3.424494504928589,
      "step": 7910
    },
    {
      "epoch": 3.017425537109375e-05,
      "step": 7910,
      "training_step_time": 10.199598789215088
    },
    {
      "epoch": 3.017425537109375e-05,
      "model_forward_time": 3.136488914489746,
      "step": 7910
    },
    {
      "epoch": 3.017425537109375e-05,
      "step": 7910,
      "training_step_time": 9.912047386169434
    },
    {
      "epoch": 3.017425537109375e-05,
      "model_forward_time": 2.4356601238250732,
      "step": 7910
    },
    {
      "epoch": 3.017425537109375e-05,
      "step": 7910,
      "training_step_time": 10.323861837387085
    },
    {
      "epoch": 3.0178070068359373e-05,
      "model_forward_time": 3.148089647293091,
      "step": 7911
    },
    {
      "epoch": 3.0178070068359373e-05,
      "step": 7911,
      "training_step_time": 9.918414831161499
    },
    {
      "epoch": 3.0178070068359373e-05,
      "model_forward_time": 3.131948471069336,
      "step": 7911
    },
    {
      "epoch": 3.0178070068359373e-05,
      "step": 7911,
      "training_step_time": 9.900063276290894
    },
    {
      "epoch": 3.0178070068359373e-05,
      "model_forward_time": 3.13191556930542,
      "step": 7911
    },
    {
      "epoch": 3.0178070068359373e-05,
      "step": 7911,
      "training_step_time": 10.171499013900757
    },
    {
      "epoch": 3.0178070068359373e-05,
      "model_forward_time": 3.1320271492004395,
      "step": 7911
    },
    {
      "epoch": 3.0178070068359373e-05,
      "step": 7911,
      "training_step_time": 10.00844430923462
    },
    {
      "epoch": 3.0181884765625e-05,
      "model_forward_time": 3.1521270275115967,
      "step": 7912
    },
    {
      "epoch": 3.0181884765625e-05,
      "step": 7912,
      "training_step_time": 9.922050952911377
    },
    {
      "epoch": 3.0181884765625e-05,
      "model_forward_time": 3.1324963569641113,
      "step": 7912
    },
    {
      "epoch": 3.0181884765625e-05,
      "step": 7912,
      "training_step_time": 9.903425931930542
    },
    {
      "epoch": 3.0181884765625e-05,
      "model_forward_time": 3.1425981521606445,
      "step": 7912
    },
    {
      "dynamics_loss_avg": 0.058829582203179596,
      "epoch": 3.0181884765625e-05,
      "step": 7912
    },
    {
      "action_loss_avg": 0.02150640352629125,
      "epoch": 3.0181884765625e-05,
      "step": 7912
    },
    {
      "epoch": 3.0181884765625e-05,
      "step": 7912,
      "training_step_time": 9.907992124557495
    },
    {
      "epoch": 3.0181884765625e-05,
      "model_forward_time": 3.430295944213867,
      "step": 7912
    },
    {
      "epoch": 3.0181884765625e-05,
      "step": 7912,
      "training_step_time": 10.28816843032837
    },
    {
      "epoch": 3.0185699462890626e-05,
      "model_forward_time": 3.149526596069336,
      "step": 7913
    },
    {
      "epoch": 3.0185699462890626e-05,
      "step": 7913,
      "training_step_time": 9.913580179214478
    },
    {
      "epoch": 3.0185699462890626e-05,
      "model_forward_time": 3.137202501296997,
      "step": 7913
    },
    {
      "epoch": 3.0185699462890626e-05,
      "step": 7913,
      "training_step_time": 9.901084423065186
    },
    {
      "epoch": 3.0185699462890626e-05,
      "model_forward_time": 3.139587640762329,
      "step": 7913
    },
    {
      "epoch": 3.0185699462890626e-05,
      "step": 7913,
      "training_step_time": 9.904359340667725
    },
    {
      "epoch": 3.0185699462890626e-05,
      "model_forward_time": 3.132112741470337,
      "step": 7913
    },
    {
      "epoch": 3.0185699462890626e-05,
      "step": 7913,
      "training_step_time": 10.29541015625
    },
    {
      "epoch": 3.018951416015625e-05,
      "model_forward_time": 3.1473255157470703,
      "step": 7914
    },
    {
      "epoch": 3.018951416015625e-05,
      "step": 7914,
      "training_step_time": 10.206761598587036
    },
    {
      "epoch": 3.018951416015625e-05,
      "model_forward_time": 3.1358139514923096,
      "step": 7914
    },
    {
      "epoch": 3.018951416015625e-05,
      "step": 7914,
      "training_step_time": 9.897794485092163
    },
    {
      "epoch": 3.018951416015625e-05,
      "model_forward_time": 3.1390299797058105,
      "step": 7914
    },
    {
      "epoch": 3.018951416015625e-05,
      "step": 7914,
      "training_step_time": 9.90241527557373
    },
    {
      "epoch": 3.018951416015625e-05,
      "model_forward_time": 3.1330416202545166,
      "step": 7914
    },
    {
      "epoch": 3.018951416015625e-05,
      "step": 7914,
      "training_step_time": 10.304446458816528
    },
    {
      "epoch": 3.0193328857421875e-05,
      "model_forward_time": 3.144287109375,
      "step": 7915
    },
    {
      "dynamics_loss_avg": 0.06588606052100658,
      "epoch": 3.0193328857421875e-05,
      "step": 7915
    },
    {
      "action_loss_avg": 0.017914065811783075,
      "epoch": 3.0193328857421875e-05,
      "step": 7915
    },
    {
      "epoch": 3.0193328857421875e-05,
      "step": 7915,
      "training_step_time": 9.921567916870117
    },
    {
      "epoch": 3.0193328857421875e-05,
      "model_forward_time": 3.4188268184661865,
      "step": 7915
    },
    {
      "epoch": 3.0193328857421875e-05,
      "step": 7915,
      "training_step_time": 10.186207294464111
    },
    {
      "epoch": 3.0193328857421875e-05,
      "model_forward_time": 3.140928268432617,
      "step": 7915
    },
    {
      "epoch": 3.0193328857421875e-05,
      "step": 7915,
      "training_step_time": 9.910025119781494
    },
    {
      "epoch": 3.0193328857421875e-05,
      "model_forward_time": 3.1426968574523926,
      "step": 7915
    },
    {
      "epoch": 3.0193328857421875e-05,
      "step": 7915,
      "training_step_time": 10.02546215057373
    },
    {
      "epoch": 3.01971435546875e-05,
      "model_forward_time": 3.1512696743011475,
      "step": 7916
    },
    {
      "epoch": 3.01971435546875e-05,
      "step": 7916,
      "training_step_time": 9.929800748825073
    },
    {
      "epoch": 3.01971435546875e-05,
      "model_forward_time": 3.139235258102417,
      "step": 7916
    },
    {
      "epoch": 3.01971435546875e-05,
      "step": 7916,
      "training_step_time": 9.9076988697052
    },
    {
      "epoch": 3.01971435546875e-05,
      "model_forward_time": 2.4316649436950684,
      "step": 7916
    },
    {
      "epoch": 3.01971435546875e-05,
      "step": 7916,
      "training_step_time": 7.72606897354126
    },
    {
      "epoch": 3.01971435546875e-05,
      "model_forward_time": 3.133869171142578,
      "step": 7916
    },
    {
      "epoch": 3.01971435546875e-05,
      "step": 7916,
      "training_step_time": 10.27334451675415
    },
    {
      "epoch": 3.0200958251953124e-05,
      "model_forward_time": 3.1571788787841797,
      "step": 7917
    },
    {
      "epoch": 3.0200958251953124e-05,
      "step": 7917,
      "training_step_time": 9.92361855506897
    },
    {
      "epoch": 3.0200958251953124e-05,
      "model_forward_time": 3.1368825435638428,
      "step": 7917
    },
    {
      "epoch": 3.0200958251953124e-05,
      "step": 7917,
      "training_step_time": 9.906562328338623
    },
    {
      "epoch": 3.0200958251953124e-05,
      "model_forward_time": 3.132146120071411,
      "step": 7917
    },
    {
      "dynamics_loss_avg": 0.056413629278540614,
      "epoch": 3.0200958251953124e-05,
      "step": 7917
    },
    {
      "action_loss_avg": 0.01885271822102368,
      "epoch": 3.0200958251953124e-05,
      "step": 7917
    },
    {
      "epoch": 3.0200958251953124e-05,
      "step": 7917,
      "training_step_time": 9.904218912124634
    },
    {
      "epoch": 3.0200958251953124e-05,
      "model_forward_time": 3.3902535438537598,
      "step": 7917
    },
    {
      "epoch": 3.0200958251953124e-05,
      "step": 7917,
      "training_step_time": 10.316253185272217
    },
    {
      "epoch": 3.020477294921875e-05,
      "model_forward_time": 3.1532931327819824,
      "step": 7918
    },
    {
      "epoch": 3.020477294921875e-05,
      "step": 7918,
      "training_step_time": 9.92405104637146
    },
    {
      "epoch": 3.020477294921875e-05,
      "model_forward_time": 3.1360559463500977,
      "step": 7918
    },
    {
      "epoch": 3.020477294921875e-05,
      "step": 7918,
      "training_step_time": 9.905470609664917
    },
    {
      "epoch": 3.020477294921875e-05,
      "model_forward_time": 3.1372182369232178,
      "step": 7918
    },
    {
      "epoch": 3.020477294921875e-05,
      "step": 7918,
      "training_step_time": 9.906328678131104
    },
    {
      "epoch": 3.020477294921875e-05,
      "model_forward_time": 3.1408965587615967,
      "step": 7918
    },
    {
      "epoch": 3.020477294921875e-05,
      "step": 7918,
      "training_step_time": 10.280571222305298
    },
    {
      "epoch": 3.0208587646484373e-05,
      "model_forward_time": 3.4099578857421875,
      "step": 7919
    },
    {
      "epoch": 3.0208587646484373e-05,
      "step": 7919,
      "training_step_time": 10.18050217628479
    },
    {
      "epoch": 3.0208587646484373e-05,
      "model_forward_time": 3.1420905590057373,
      "step": 7919
    },
    {
      "epoch": 3.0208587646484373e-05,
      "step": 7919,
      "training_step_time": 9.916309833526611
    },
    {
      "epoch": 3.0208587646484373e-05,
      "model_forward_time": 3.13510799407959,
      "step": 7919
    },
    {
      "epoch": 3.0208587646484373e-05,
      "step": 7919,
      "training_step_time": 9.906281471252441
    },
    {
      "epoch": 3.0208587646484373e-05,
      "model_forward_time": 3.146794080734253,
      "step": 7919
    },
    {
      "epoch": 3.0208587646484373e-05,
      "step": 7919,
      "training_step_time": 10.311776399612427
    },
    {
      "epoch": 3.021240234375e-05,
      "grad_norm": 0.3132888078689575,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 7920
    },
    {
      "epoch": 3.021240234375e-05,
      "model_forward_time": 3.160281181335449,
      "step": 7920
    },
    {
      "dynamics_loss_avg": 0.05509168356657028,
      "epoch": 3.021240234375e-05,
      "step": 7920
    },
    {
      "action_loss_avg": 0.02193992827087641,
      "epoch": 3.021240234375e-05,
      "step": 7920
    },
    {
      "epoch": 3.021240234375e-05,
      "step": 7920,
      "training_step_time": 9.942006587982178
    },
    {
      "epoch": 3.021240234375e-05,
      "model_forward_time": 3.1395952701568604,
      "step": 7920
    },
    {
      "epoch": 3.021240234375e-05,
      "step": 7920,
      "training_step_time": 10.197213411331177
    },
    {
      "epoch": 3.021240234375e-05,
      "model_forward_time": 3.139158248901367,
      "step": 7920
    },
    {
      "epoch": 3.021240234375e-05,
      "step": 7920,
      "training_step_time": 9.91275691986084
    },
    {
      "epoch": 3.021240234375e-05,
      "model_forward_time": 3.139134407043457,
      "step": 7920
    },
    {
      "epoch": 3.021240234375e-05,
      "step": 7920,
      "training_step_time": 10.023782014846802
    },
    {
      "epoch": 3.0216217041015626e-05,
      "model_forward_time": 3.147507667541504,
      "step": 7921
    },
    {
      "epoch": 3.0216217041015626e-05,
      "step": 7921,
      "training_step_time": 9.917018175125122
    },
    {
      "epoch": 3.0216217041015626e-05,
      "model_forward_time": 3.1342966556549072,
      "step": 7921
    },
    {
      "epoch": 3.0216217041015626e-05,
      "step": 7921,
      "training_step_time": 9.905961275100708
    },
    {
      "epoch": 3.0216217041015626e-05,
      "model_forward_time": 3.393315076828003,
      "step": 7921
    },
    {
      "epoch": 3.0216217041015626e-05,
      "step": 7921,
      "training_step_time": 10.166627645492554
    },
    {
      "epoch": 3.0216217041015626e-05,
      "model_forward_time": 3.140437364578247,
      "step": 7921
    },
    {
      "epoch": 3.0216217041015626e-05,
      "step": 7921,
      "training_step_time": 10.300490617752075
    },
    {
      "epoch": 3.022003173828125e-05,
      "model_forward_time": 3.152149200439453,
      "step": 7922
    },
    {
      "epoch": 3.022003173828125e-05,
      "step": 7922,
      "training_step_time": 9.929111957550049
    },
    {
      "epoch": 3.022003173828125e-05,
      "model_forward_time": 3.136924982070923,
      "step": 7922
    },
    {
      "epoch": 3.022003173828125e-05,
      "step": 7922,
      "training_step_time": 9.906044244766235
    },
    {
      "epoch": 3.022003173828125e-05,
      "model_forward_time": 3.140625238418579,
      "step": 7922
    },
    {
      "dynamics_loss_avg": 0.07026119958609342,
      "epoch": 3.022003173828125e-05,
      "step": 7922
    },
    {
      "action_loss_avg": 0.020079000340774655,
      "epoch": 3.022003173828125e-05,
      "step": 7922
    },
    {
      "epoch": 3.022003173828125e-05,
      "step": 7922,
      "training_step_time": 9.916375160217285
    },
    {
      "epoch": 3.022003173828125e-05,
      "model_forward_time": 3.1351964473724365,
      "step": 7922
    },
    {
      "epoch": 3.022003173828125e-05,
      "step": 7922,
      "training_step_time": 10.274062633514404
    },
    {
      "epoch": 3.0223846435546875e-05,
      "model_forward_time": 3.1573219299316406,
      "step": 7923
    },
    {
      "epoch": 3.0223846435546875e-05,
      "step": 7923,
      "training_step_time": 9.932977199554443
    },
    {
      "epoch": 3.0223846435546875e-05,
      "model_forward_time": 3.139145612716675,
      "step": 7923
    },
    {
      "epoch": 3.0223846435546875e-05,
      "step": 7923,
      "training_step_time": 9.9088876247406
    },
    {
      "epoch": 3.0223846435546875e-05,
      "model_forward_time": 3.1424145698547363,
      "step": 7923
    },
    {
      "epoch": 3.0223846435546875e-05,
      "step": 7923,
      "training_step_time": 9.917312622070312
    },
    {
      "epoch": 3.0223846435546875e-05,
      "model_forward_time": 3.139866828918457,
      "step": 7923
    },
    {
      "epoch": 3.0223846435546875e-05,
      "step": 7923,
      "training_step_time": 10.305437803268433
    },
    {
      "epoch": 3.02276611328125e-05,
      "model_forward_time": 3.426989793777466,
      "step": 7924
    },
    {
      "epoch": 3.02276611328125e-05,
      "step": 7924,
      "training_step_time": 10.20181941986084
    },
    {
      "epoch": 3.02276611328125e-05,
      "model_forward_time": 2.4372804164886475,
      "step": 7924
    },
    {
      "epoch": 3.02276611328125e-05,
      "step": 7924,
      "training_step_time": 7.460569143295288
    },
    {
      "epoch": 3.02276611328125e-05,
      "model_forward_time": 3.1350162029266357,
      "step": 7924
    },
    {
      "epoch": 3.02276611328125e-05,
      "step": 7924,
      "training_step_time": 9.901136636734009
    },
    {
      "epoch": 3.02276611328125e-05,
      "model_forward_time": 3.131711721420288,
      "step": 7924
    },
    {
      "epoch": 3.02276611328125e-05,
      "step": 7924,
      "training_step_time": 10.284096240997314
    },
    {
      "epoch": 3.0231475830078124e-05,
      "model_forward_time": 3.1453611850738525,
      "step": 7925
    },
    {
      "dynamics_loss_avg": 0.06940720919519663,
      "epoch": 3.0231475830078124e-05,
      "step": 7925
    },
    {
      "action_loss_avg": 0.020787010528147222,
      "epoch": 3.0231475830078124e-05,
      "step": 7925
    },
    {
      "epoch": 3.0231475830078124e-05,
      "step": 7925,
      "training_step_time": 9.917585611343384
    },
    {
      "epoch": 3.0231475830078124e-05,
      "model_forward_time": 3.132570743560791,
      "step": 7925
    },
    {
      "epoch": 3.0231475830078124e-05,
      "step": 7925,
      "training_step_time": 9.899596691131592
    },
    {
      "epoch": 3.0231475830078124e-05,
      "model_forward_time": 3.4031264781951904,
      "step": 7925
    },
    {
      "epoch": 3.0231475830078124e-05,
      "step": 7925,
      "training_step_time": 10.17338752746582
    },
    {
      "epoch": 3.0231475830078124e-05,
      "model_forward_time": 3.1348414421081543,
      "step": 7925
    },
    {
      "epoch": 3.0231475830078124e-05,
      "step": 7925,
      "training_step_time": 10.281880378723145
    },
    {
      "epoch": 3.023529052734375e-05,
      "model_forward_time": 3.155425548553467,
      "step": 7926
    },
    {
      "epoch": 3.023529052734375e-05,
      "step": 7926,
      "training_step_time": 9.925925493240356
    },
    {
      "epoch": 3.023529052734375e-05,
      "model_forward_time": 3.1378326416015625,
      "step": 7926
    },
    {
      "epoch": 3.023529052734375e-05,
      "step": 7926,
      "training_step_time": 9.915146350860596
    },
    {
      "epoch": 3.023529052734375e-05,
      "model_forward_time": 3.414804220199585,
      "step": 7926
    },
    {
      "epoch": 3.023529052734375e-05,
      "step": 7926,
      "training_step_time": 10.185261011123657
    },
    {
      "epoch": 3.023529052734375e-05,
      "model_forward_time": 2.4373738765716553,
      "step": 7926
    },
    {
      "epoch": 3.023529052734375e-05,
      "step": 7926,
      "training_step_time": 10.305481433868408
    },
    {
      "epoch": 3.0239105224609373e-05,
      "model_forward_time": 3.144650459289551,
      "step": 7927
    },
    {
      "epoch": 3.0239105224609373e-05,
      "step": 7927,
      "training_step_time": 9.91605520248413
    },
    {
      "epoch": 3.0239105224609373e-05,
      "model_forward_time": 3.1338372230529785,
      "step": 7927
    },
    {
      "epoch": 3.0239105224609373e-05,
      "step": 7927,
      "training_step_time": 9.902292490005493
    },
    {
      "epoch": 3.0239105224609373e-05,
      "model_forward_time": 3.1320414543151855,
      "step": 7927
    },
    {
      "dynamics_loss_avg": 0.0434956856071949,
      "epoch": 3.0239105224609373e-05,
      "step": 7927
    },
    {
      "action_loss_avg": 0.016835530381649734,
      "epoch": 3.0239105224609373e-05,
      "step": 7927
    },
    {
      "epoch": 3.0239105224609373e-05,
      "step": 7927,
      "training_step_time": 9.904874563217163
    },
    {
      "epoch": 3.0239105224609373e-05,
      "model_forward_time": 3.1346192359924316,
      "step": 7927
    },
    {
      "epoch": 3.0239105224609373e-05,
      "step": 7927,
      "training_step_time": 10.295862674713135
    },
    {
      "epoch": 3.0242919921875e-05,
      "model_forward_time": 2.760669469833374,
      "step": 7928
    },
    {
      "epoch": 3.0242919921875e-05,
      "step": 7928,
      "training_step_time": 7.780612945556641
    },
    {
      "epoch": 3.0242919921875e-05,
      "model_forward_time": 3.1380796432495117,
      "step": 7928
    },
    {
      "epoch": 3.0242919921875e-05,
      "step": 7928,
      "training_step_time": 9.903717279434204
    },
    {
      "epoch": 3.0242919921875e-05,
      "model_forward_time": 3.1334214210510254,
      "step": 7928
    },
    {
      "epoch": 3.0242919921875e-05,
      "step": 7928,
      "training_step_time": 9.901427507400513
    },
    {
      "epoch": 3.0242919921875e-05,
      "model_forward_time": 3.134993314743042,
      "step": 7928
    },
    {
      "epoch": 3.0242919921875e-05,
      "step": 7928,
      "training_step_time": 10.322970867156982
    },
    {
      "epoch": 3.0246734619140626e-05,
      "model_forward_time": 3.4300649166107178,
      "step": 7929
    },
    {
      "epoch": 3.0246734619140626e-05,
      "step": 7929,
      "training_step_time": 10.204817295074463
    },
    {
      "epoch": 3.0246734619140626e-05,
      "model_forward_time": 3.14174222946167,
      "step": 7929
    },
    {
      "epoch": 3.0246734619140626e-05,
      "step": 7929,
      "training_step_time": 9.91417908668518
    },
    {
      "epoch": 3.0246734619140626e-05,
      "model_forward_time": 3.1366567611694336,
      "step": 7929
    },
    {
      "epoch": 3.0246734619140626e-05,
      "step": 7929,
      "training_step_time": 9.908080577850342
    },
    {
      "epoch": 3.0246734619140626e-05,
      "model_forward_time": 3.136488199234009,
      "step": 7929
    },
    {
      "epoch": 3.0246734619140626e-05,
      "step": 7929,
      "training_step_time": 10.297873735427856
    },
    {
      "epoch": 3.025054931640625e-05,
      "grad_norm": 0.5569909811019897,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 7930
    },
    {
      "epoch": 3.025054931640625e-05,
      "model_forward_time": 3.1486313343048096,
      "step": 7930
    },
    {
      "dynamics_loss_avg": 0.07092927992343903,
      "epoch": 3.025054931640625e-05,
      "step": 7930
    },
    {
      "action_loss_avg": 0.023358801286667584,
      "epoch": 3.025054931640625e-05,
      "step": 7930
    },
    {
      "epoch": 3.025054931640625e-05,
      "step": 7930,
      "training_step_time": 9.922533750534058
    },
    {
      "epoch": 3.025054931640625e-05,
      "model_forward_time": 2.4316823482513428,
      "step": 7930
    },
    {
      "epoch": 3.025054931640625e-05,
      "step": 7930,
      "training_step_time": 7.458315372467041
    },
    {
      "epoch": 3.025054931640625e-05,
      "model_forward_time": 3.430246353149414,
      "step": 7930
    },
    {
      "epoch": 3.025054931640625e-05,
      "step": 7930,
      "training_step_time": 10.19530701637268
    },
    {
      "epoch": 3.025054931640625e-05,
      "model_forward_time": 3.141331911087036,
      "step": 7930
    },
    {
      "epoch": 3.025054931640625e-05,
      "step": 7930,
      "training_step_time": 10.242985725402832
    },
    {
      "epoch": 3.0254364013671875e-05,
      "model_forward_time": 3.15108060836792,
      "step": 7931
    },
    {
      "epoch": 3.0254364013671875e-05,
      "step": 7931,
      "training_step_time": 9.915341854095459
    },
    {
      "epoch": 3.0254364013671875e-05,
      "model_forward_time": 3.1329753398895264,
      "step": 7931
    },
    {
      "epoch": 3.0254364013671875e-05,
      "step": 7931,
      "training_step_time": 9.895975112915039
    },
    {
      "epoch": 3.0254364013671875e-05,
      "model_forward_time": 3.140643358230591,
      "step": 7931
    },
    {
      "epoch": 3.0254364013671875e-05,
      "step": 7931,
      "training_step_time": 10.185437440872192
    },
    {
      "epoch": 3.0254364013671875e-05,
      "model_forward_time": 2.435607433319092,
      "step": 7931
    },
    {
      "epoch": 3.0254364013671875e-05,
      "step": 7931,
      "training_step_time": 10.261057138442993
    },
    {
      "epoch": 3.02581787109375e-05,
      "model_forward_time": 3.153475761413574,
      "step": 7932
    },
    {
      "epoch": 3.02581787109375e-05,
      "step": 7932,
      "training_step_time": 9.917104959487915
    },
    {
      "epoch": 3.02581787109375e-05,
      "model_forward_time": 3.1400516033172607,
      "step": 7932
    },
    {
      "epoch": 3.02581787109375e-05,
      "step": 7932,
      "training_step_time": 9.90434741973877
    },
    {
      "epoch": 3.02581787109375e-05,
      "model_forward_time": 3.135958671569824,
      "step": 7932
    },
    {
      "dynamics_loss_avg": 0.06281367652118205,
      "epoch": 3.02581787109375e-05,
      "step": 7932
    },
    {
      "action_loss_avg": 0.01963977050036192,
      "epoch": 3.02581787109375e-05,
      "step": 7932
    },
    {
      "epoch": 3.02581787109375e-05,
      "step": 7932,
      "training_step_time": 9.907857656478882
    },
    {
      "epoch": 3.02581787109375e-05,
      "model_forward_time": 3.138967275619507,
      "step": 7932
    },
    {
      "epoch": 3.02581787109375e-05,
      "step": 7932,
      "training_step_time": 10.288881063461304
    },
    {
      "epoch": 3.0261993408203124e-05,
      "model_forward_time": 3.1546857357025146,
      "step": 7933
    },
    {
      "epoch": 3.0261993408203124e-05,
      "step": 7933,
      "training_step_time": 9.923165798187256
    },
    {
      "epoch": 3.0261993408203124e-05,
      "model_forward_time": 3.13918137550354,
      "step": 7933
    },
    {
      "epoch": 3.0261993408203124e-05,
      "step": 7933,
      "training_step_time": 9.907279968261719
    },
    {
      "epoch": 3.0261993408203124e-05,
      "model_forward_time": 3.144669532775879,
      "step": 7933
    },
    {
      "epoch": 3.0261993408203124e-05,
      "step": 7933,
      "training_step_time": 9.916421175003052
    },
    {
      "epoch": 3.0261993408203124e-05,
      "model_forward_time": 3.1432247161865234,
      "step": 7933
    },
    {
      "epoch": 3.0261993408203124e-05,
      "step": 7933,
      "training_step_time": 10.036997318267822
    },
    {
      "epoch": 3.026580810546875e-05,
      "model_forward_time": 3.160813093185425,
      "step": 7934
    },
    {
      "epoch": 3.026580810546875e-05,
      "step": 7934,
      "training_step_time": 10.224933624267578
    },
    {
      "epoch": 3.026580810546875e-05,
      "model_forward_time": 3.144315242767334,
      "step": 7934
    },
    {
      "epoch": 3.026580810546875e-05,
      "step": 7934,
      "training_step_time": 9.918352127075195
    },
    {
      "epoch": 3.026580810546875e-05,
      "model_forward_time": 3.1437997817993164,
      "step": 7934
    },
    {
      "epoch": 3.026580810546875e-05,
      "step": 7934,
      "training_step_time": 9.918733835220337
    },
    {
      "epoch": 3.026580810546875e-05,
      "model_forward_time": 3.1455302238464355,
      "step": 7934
    },
    {
      "epoch": 3.026580810546875e-05,
      "step": 7934,
      "training_step_time": 10.33779501914978
    },
    {
      "epoch": 3.0269622802734374e-05,
      "model_forward_time": 2.4543280601501465,
      "step": 7935
    },
    {
      "dynamics_loss_avg": 0.0571110837161541,
      "epoch": 3.0269622802734374e-05,
      "step": 7935
    },
    {
      "action_loss_avg": 0.016731853876262904,
      "epoch": 3.0269622802734374e-05,
      "step": 7935
    },
    {
      "epoch": 3.0269622802734374e-05,
      "step": 7935,
      "training_step_time": 7.482351541519165
    },
    {
      "epoch": 3.0269622802734374e-05,
      "model_forward_time": 3.3973562717437744,
      "step": 7935
    },
    {
      "epoch": 3.0269622802734374e-05,
      "step": 7935,
      "training_step_time": 10.157809734344482
    },
    {
      "epoch": 3.0269622802734374e-05,
      "model_forward_time": 3.1407101154327393,
      "step": 7935
    },
    {
      "epoch": 3.0269622802734374e-05,
      "step": 7935,
      "training_step_time": 9.902783393859863
    },
    {
      "epoch": 3.0269622802734374e-05,
      "model_forward_time": 3.1345531940460205,
      "step": 7935
    },
    {
      "epoch": 3.0269622802734374e-05,
      "step": 7935,
      "training_step_time": 10.302854299545288
    },
    {
      "epoch": 3.02734375e-05,
      "model_forward_time": 3.151275396347046,
      "step": 7936
    },
    {
      "epoch": 3.02734375e-05,
      "step": 7936,
      "training_step_time": 9.931896924972534
    },
    {
      "epoch": 3.02734375e-05,
      "model_forward_time": 3.1454379558563232,
      "step": 7936
    },
    {
      "epoch": 3.02734375e-05,
      "step": 7936,
      "training_step_time": 9.910730838775635
    },
    {
      "epoch": 3.02734375e-05,
      "model_forward_time": 3.1423587799072266,
      "step": 7936
    },
    {
      "epoch": 3.02734375e-05,
      "step": 7936,
      "training_step_time": 10.167561531066895
    },
    {
      "epoch": 3.02734375e-05,
      "model_forward_time": 3.144493579864502,
      "step": 7936
    },
    {
      "epoch": 3.02734375e-05,
      "step": 7936,
      "training_step_time": 10.283402681350708
    },
    {
      "epoch": 3.0277252197265626e-05,
      "model_forward_time": 3.1532013416290283,
      "step": 7937
    },
    {
      "epoch": 3.0277252197265626e-05,
      "step": 7937,
      "training_step_time": 9.925446510314941
    },
    {
      "epoch": 3.0277252197265626e-05,
      "model_forward_time": 3.1460611820220947,
      "step": 7937
    },
    {
      "epoch": 3.0277252197265626e-05,
      "step": 7937,
      "training_step_time": 9.916483163833618
    },
    {
      "epoch": 3.0277252197265626e-05,
      "model_forward_time": 3.1372768878936768,
      "step": 7937
    },
    {
      "dynamics_loss_avg": 0.05949512552469969,
      "epoch": 3.0277252197265626e-05,
      "step": 7937
    },
    {
      "action_loss_avg": 0.01918912474066019,
      "epoch": 3.0277252197265626e-05,
      "step": 7937
    },
    {
      "epoch": 3.0277252197265626e-05,
      "step": 7937,
      "training_step_time": 9.917575120925903
    },
    {
      "epoch": 3.0277252197265626e-05,
      "model_forward_time": 3.4032578468322754,
      "step": 7937
    },
    {
      "epoch": 3.0277252197265626e-05,
      "step": 7937,
      "training_step_time": 10.274559497833252
    },
    {
      "epoch": 3.028106689453125e-05,
      "model_forward_time": 3.160329818725586,
      "step": 7938
    },
    {
      "epoch": 3.028106689453125e-05,
      "step": 7938,
      "training_step_time": 9.935138463973999
    },
    {
      "epoch": 3.028106689453125e-05,
      "model_forward_time": 3.1402945518493652,
      "step": 7938
    },
    {
      "epoch": 3.028106689453125e-05,
      "step": 7938,
      "training_step_time": 9.90993356704712
    },
    {
      "epoch": 3.028106689453125e-05,
      "model_forward_time": 3.138772964477539,
      "step": 7938
    },
    {
      "epoch": 3.028106689453125e-05,
      "step": 7938,
      "training_step_time": 9.910067796707153
    },
    {
      "epoch": 3.028106689453125e-05,
      "model_forward_time": 3.145843267440796,
      "step": 7938
    },
    {
      "epoch": 3.028106689453125e-05,
      "step": 7938,
      "training_step_time": 10.283626317977905
    },
    {
      "epoch": 3.0284881591796875e-05,
      "model_forward_time": 3.406794786453247,
      "step": 7939
    },
    {
      "epoch": 3.0284881591796875e-05,
      "step": 7939,
      "training_step_time": 10.180234432220459
    },
    {
      "epoch": 3.0284881591796875e-05,
      "model_forward_time": 3.142042875289917,
      "step": 7939
    },
    {
      "epoch": 3.0284881591796875e-05,
      "step": 7939,
      "training_step_time": 9.91390323638916
    },
    {
      "epoch": 3.0284881591796875e-05,
      "model_forward_time": 3.1388933658599854,
      "step": 7939
    },
    {
      "epoch": 3.0284881591796875e-05,
      "step": 7939,
      "training_step_time": 9.911980867385864
    },
    {
      "epoch": 3.0284881591796875e-05,
      "model_forward_time": 3.148325204849243,
      "step": 7939
    },
    {
      "epoch": 3.0284881591796875e-05,
      "step": 7939,
      "training_step_time": 10.278648853302002
    },
    {
      "epoch": 3.02886962890625e-05,
      "grad_norm": 0.4146586060523987,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 7940
    },
    {
      "epoch": 3.02886962890625e-05,
      "model_forward_time": 3.15986967086792,
      "step": 7940
    },
    {
      "dynamics_loss_avg": 0.05879592448472977,
      "epoch": 3.02886962890625e-05,
      "step": 7940
    },
    {
      "action_loss_avg": 0.019966253824532033,
      "epoch": 3.02886962890625e-05,
      "step": 7940
    },
    {
      "epoch": 3.02886962890625e-05,
      "step": 7940,
      "training_step_time": 9.939155101776123
    },
    {
      "epoch": 3.02886962890625e-05,
      "model_forward_time": 3.4259846210479736,
      "step": 7940
    },
    {
      "epoch": 3.02886962890625e-05,
      "step": 7940,
      "training_step_time": 10.202868461608887
    },
    {
      "epoch": 3.02886962890625e-05,
      "model_forward_time": 3.140385866165161,
      "step": 7940
    },
    {
      "epoch": 3.02886962890625e-05,
      "step": 7940,
      "training_step_time": 9.913612127304077
    },
    {
      "epoch": 3.02886962890625e-05,
      "model_forward_time": 3.141528844833374,
      "step": 7940
    },
    {
      "epoch": 3.02886962890625e-05,
      "step": 7940,
      "training_step_time": 10.027616500854492
    },
    {
      "epoch": 3.0292510986328124e-05,
      "model_forward_time": 3.1535704135894775,
      "step": 7941
    },
    {
      "epoch": 3.0292510986328124e-05,
      "step": 7941,
      "training_step_time": 9.925188302993774
    },
    {
      "epoch": 3.0292510986328124e-05,
      "model_forward_time": 3.135118007659912,
      "step": 7941
    },
    {
      "epoch": 3.0292510986328124e-05,
      "step": 7941,
      "training_step_time": 9.906967878341675
    },
    {
      "epoch": 3.0292510986328124e-05,
      "model_forward_time": 3.3946902751922607,
      "step": 7941
    },
    {
      "epoch": 3.0292510986328124e-05,
      "step": 7941,
      "training_step_time": 10.166513919830322
    },
    {
      "epoch": 3.0292510986328124e-05,
      "model_forward_time": 3.1416149139404297,
      "step": 7941
    },
    {
      "epoch": 3.0292510986328124e-05,
      "step": 7941,
      "training_step_time": 10.292675256729126
    },
    {
      "epoch": 3.029632568359375e-05,
      "model_forward_time": 3.1538286209106445,
      "step": 7942
    },
    {
      "epoch": 3.029632568359375e-05,
      "step": 7942,
      "training_step_time": 9.927757501602173
    },
    {
      "epoch": 3.029632568359375e-05,
      "model_forward_time": 3.1367299556732178,
      "step": 7942
    },
    {
      "epoch": 3.029632568359375e-05,
      "step": 7942,
      "training_step_time": 9.907821416854858
    },
    {
      "epoch": 3.029632568359375e-05,
      "model_forward_time": 3.1429293155670166,
      "step": 7942
    },
    {
      "dynamics_loss_avg": 0.04566281186416745,
      "epoch": 3.029632568359375e-05,
      "step": 7942
    },
    {
      "action_loss_avg": 0.015383180230855942,
      "epoch": 3.029632568359375e-05,
      "step": 7942
    },
    {
      "epoch": 3.029632568359375e-05,
      "step": 7942,
      "training_step_time": 9.913770914077759
    },
    {
      "epoch": 3.029632568359375e-05,
      "model_forward_time": 3.418034791946411,
      "step": 7942
    },
    {
      "epoch": 3.029632568359375e-05,
      "step": 7942,
      "training_step_time": 10.285340309143066
    },
    {
      "epoch": 3.0300140380859374e-05,
      "model_forward_time": 3.1516473293304443,
      "step": 7943
    },
    {
      "epoch": 3.0300140380859374e-05,
      "step": 7943,
      "training_step_time": 9.924468994140625
    },
    {
      "epoch": 3.0300140380859374e-05,
      "model_forward_time": 3.1418261528015137,
      "step": 7943
    },
    {
      "epoch": 3.0300140380859374e-05,
      "step": 7943,
      "training_step_time": 9.915287494659424
    },
    {
      "epoch": 3.0300140380859374e-05,
      "model_forward_time": 3.1452081203460693,
      "step": 7943
    },
    {
      "epoch": 3.0300140380859374e-05,
      "step": 7943,
      "training_step_time": 9.922996520996094
    },
    {
      "epoch": 3.0300140380859374e-05,
      "model_forward_time": 3.1413633823394775,
      "step": 7943
    },
    {
      "epoch": 3.0300140380859374e-05,
      "step": 7943,
      "training_step_time": 10.313310861587524
    },
    {
      "epoch": 3.0303955078125e-05,
      "model_forward_time": 3.460951566696167,
      "step": 7944
    },
    {
      "epoch": 3.0303955078125e-05,
      "step": 7944,
      "training_step_time": 10.236568450927734
    },
    {
      "epoch": 3.0303955078125e-05,
      "model_forward_time": 3.1440634727478027,
      "step": 7944
    },
    {
      "epoch": 3.0303955078125e-05,
      "step": 7944,
      "training_step_time": 9.91140604019165
    },
    {
      "epoch": 3.0303955078125e-05,
      "model_forward_time": 2.4380104541778564,
      "step": 7944
    },
    {
      "epoch": 3.0303955078125e-05,
      "step": 7944,
      "training_step_time": 7.466102600097656
    },
    {
      "epoch": 3.0303955078125e-05,
      "model_forward_time": 3.1364028453826904,
      "step": 7944
    },
    {
      "epoch": 3.0303955078125e-05,
      "step": 7944,
      "training_step_time": 10.290507078170776
    },
    {
      "epoch": 3.0307769775390626e-05,
      "model_forward_time": 3.1695475578308105,
      "step": 7945
    },
    {
      "dynamics_loss_avg": 0.06853927224874497,
      "epoch": 3.0307769775390626e-05,
      "step": 7945
    },
    {
      "action_loss_avg": 0.019224681705236436,
      "epoch": 3.0307769775390626e-05,
      "step": 7945
    },
    {
      "epoch": 3.0307769775390626e-05,
      "step": 7945,
      "training_step_time": 9.944293022155762
    },
    {
      "epoch": 3.0307769775390626e-05,
      "model_forward_time": 3.438214063644409,
      "step": 7945
    },
    {
      "epoch": 3.0307769775390626e-05,
      "step": 7945,
      "training_step_time": 10.205121278762817
    },
    {
      "epoch": 3.0307769775390626e-05,
      "model_forward_time": 2.4416470527648926,
      "step": 7945
    },
    {
      "epoch": 3.0307769775390626e-05,
      "step": 7945,
      "training_step_time": 7.469495058059692
    },
    {
      "epoch": 3.0307769775390626e-05,
      "model_forward_time": 3.1403391361236572,
      "step": 7945
    },
    {
      "epoch": 3.0307769775390626e-05,
      "step": 7945,
      "training_step_time": 10.30922269821167
    },
    {
      "epoch": 3.031158447265625e-05,
      "model_forward_time": 3.1476500034332275,
      "step": 7946
    },
    {
      "epoch": 3.031158447265625e-05,
      "step": 7946,
      "training_step_time": 9.914663791656494
    },
    {
      "epoch": 3.031158447265625e-05,
      "model_forward_time": 3.142488956451416,
      "step": 7946
    },
    {
      "epoch": 3.031158447265625e-05,
      "step": 7946,
      "training_step_time": 9.910577535629272
    },
    {
      "epoch": 3.031158447265625e-05,
      "model_forward_time": 3.42564058303833,
      "step": 7946
    },
    {
      "epoch": 3.031158447265625e-05,
      "step": 7946,
      "training_step_time": 10.201308965682983
    },
    {
      "epoch": 3.031158447265625e-05,
      "model_forward_time": 3.141023635864258,
      "step": 7946
    },
    {
      "epoch": 3.031158447265625e-05,
      "step": 7946,
      "training_step_time": 10.284624814987183
    },
    {
      "epoch": 3.0315399169921875e-05,
      "model_forward_time": 3.1710636615753174,
      "step": 7947
    },
    {
      "epoch": 3.0315399169921875e-05,
      "step": 7947,
      "training_step_time": 9.944657325744629
    },
    {
      "epoch": 3.0315399169921875e-05,
      "model_forward_time": 3.1468541622161865,
      "step": 7947
    },
    {
      "epoch": 3.0315399169921875e-05,
      "step": 7947,
      "training_step_time": 9.915073156356812
    },
    {
      "epoch": 3.0315399169921875e-05,
      "model_forward_time": 2.4404759407043457,
      "step": 7947
    },
    {
      "dynamics_loss_avg": 0.06365996971726418,
      "epoch": 3.0315399169921875e-05,
      "step": 7947
    },
    {
      "action_loss_avg": 0.021027784328907728,
      "epoch": 3.0315399169921875e-05,
      "step": 7947
    },
    {
      "epoch": 3.0315399169921875e-05,
      "step": 7947,
      "training_step_time": 7.469367504119873
    },
    {
      "epoch": 3.0315399169921875e-05,
      "model_forward_time": 2.4349536895751953,
      "step": 7947
    },
    {
      "epoch": 3.0315399169921875e-05,
      "step": 7947,
      "training_step_time": 10.263542413711548
    },
    {
      "epoch": 3.03192138671875e-05,
      "model_forward_time": 3.4633429050445557,
      "step": 7948
    },
    {
      "epoch": 3.03192138671875e-05,
      "step": 7948,
      "training_step_time": 10.235180139541626
    },
    {
      "epoch": 3.03192138671875e-05,
      "model_forward_time": 3.1434061527252197,
      "step": 7948
    },
    {
      "epoch": 3.03192138671875e-05,
      "step": 7948,
      "training_step_time": 9.913261890411377
    },
    {
      "epoch": 3.03192138671875e-05,
      "model_forward_time": 3.1425914764404297,
      "step": 7948
    },
    {
      "epoch": 3.03192138671875e-05,
      "step": 7948,
      "training_step_time": 9.913878440856934
    },
    {
      "epoch": 3.03192138671875e-05,
      "model_forward_time": 3.1459646224975586,
      "step": 7948
    },
    {
      "epoch": 3.03192138671875e-05,
      "step": 7948,
      "training_step_time": 10.276750326156616
    },
    {
      "epoch": 3.0323028564453124e-05,
      "model_forward_time": 3.1598801612854004,
      "step": 7949
    },
    {
      "epoch": 3.0323028564453124e-05,
      "step": 7949,
      "training_step_time": 10.203226327896118
    },
    {
      "epoch": 3.0323028564453124e-05,
      "model_forward_time": 3.145638942718506,
      "step": 7949
    },
    {
      "epoch": 3.0323028564453124e-05,
      "step": 7949,
      "training_step_time": 9.918611526489258
    },
    {
      "epoch": 3.0323028564453124e-05,
      "model_forward_time": 2.4384217262268066,
      "step": 7949
    },
    {
      "epoch": 3.0323028564453124e-05,
      "step": 7949,
      "training_step_time": 7.467209815979004
    },
    {
      "epoch": 3.0323028564453124e-05,
      "model_forward_time": 3.136845350265503,
      "step": 7949
    },
    {
      "epoch": 3.0323028564453124e-05,
      "step": 7949,
      "training_step_time": 10.31325387954712
    },
    {
      "epoch": 3.032684326171875e-05,
      "grad_norm": 0.40018174052238464,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 7950
    },
    {
      "epoch": 3.032684326171875e-05,
      "model_forward_time": 3.1452956199645996,
      "step": 7950
    },
    {
      "dynamics_loss_avg": 0.058393890224397185,
      "epoch": 3.032684326171875e-05,
      "step": 7950
    },
    {
      "action_loss_avg": 0.01677907877601683,
      "epoch": 3.032684326171875e-05,
      "step": 7950
    },
    {
      "epoch": 3.032684326171875e-05,
      "step": 7950,
      "training_step_time": 9.923000812530518
    },
    {
      "epoch": 3.032684326171875e-05,
      "model_forward_time": 3.137192964553833,
      "step": 7950
    },
    {
      "epoch": 3.032684326171875e-05,
      "step": 7950,
      "training_step_time": 9.905795812606812
    },
    {
      "epoch": 3.032684326171875e-05,
      "model_forward_time": 3.4022037982940674,
      "step": 7950
    },
    {
      "epoch": 3.032684326171875e-05,
      "step": 7950,
      "training_step_time": 10.169156312942505
    },
    {
      "epoch": 3.032684326171875e-05,
      "model_forward_time": 3.139125108718872,
      "step": 7950
    },
    {
      "epoch": 3.032684326171875e-05,
      "step": 7950,
      "training_step_time": 10.313641786575317
    },
    {
      "epoch": 3.0330657958984374e-05,
      "model_forward_time": 3.162550449371338,
      "step": 7951
    },
    {
      "epoch": 3.0330657958984374e-05,
      "step": 7951,
      "training_step_time": 9.932809352874756
    },
    {
      "epoch": 3.0330657958984374e-05,
      "model_forward_time": 3.143094539642334,
      "step": 7951
    },
    {
      "epoch": 3.0330657958984374e-05,
      "step": 7951,
      "training_step_time": 9.918637037277222
    },
    {
      "epoch": 3.0330657958984374e-05,
      "model_forward_time": 3.4179534912109375,
      "step": 7951
    },
    {
      "epoch": 3.0330657958984374e-05,
      "step": 7951,
      "training_step_time": 10.19183874130249
    },
    {
      "epoch": 3.0330657958984374e-05,
      "model_forward_time": 3.144378423690796,
      "step": 7951
    },
    {
      "epoch": 3.0330657958984374e-05,
      "step": 7951,
      "training_step_time": 10.021623134613037
    },
    {
      "epoch": 3.033447265625e-05,
      "model_forward_time": 3.16102933883667,
      "step": 7952
    },
    {
      "epoch": 3.033447265625e-05,
      "step": 7952,
      "training_step_time": 9.932657480239868
    },
    {
      "epoch": 3.033447265625e-05,
      "model_forward_time": 3.1445178985595703,
      "step": 7952
    },
    {
      "epoch": 3.033447265625e-05,
      "step": 7952,
      "training_step_time": 9.921286344528198
    },
    {
      "epoch": 3.033447265625e-05,
      "model_forward_time": 3.148878574371338,
      "step": 7952
    },
    {
      "dynamics_loss_avg": 0.0578005313873291,
      "epoch": 3.033447265625e-05,
      "step": 7952
    },
    {
      "action_loss_avg": 0.02082571517676115,
      "epoch": 3.033447265625e-05,
      "step": 7952
    },
    {
      "epoch": 3.033447265625e-05,
      "step": 7952,
      "training_step_time": 9.928388357162476
    },
    {
      "epoch": 3.033447265625e-05,
      "model_forward_time": 3.4614381790161133,
      "step": 7952
    },
    {
      "epoch": 3.033447265625e-05,
      "step": 7952,
      "training_step_time": 10.332175493240356
    },
    {
      "epoch": 3.0338287353515626e-05,
      "model_forward_time": 3.156543016433716,
      "step": 7953
    },
    {
      "epoch": 3.0338287353515626e-05,
      "step": 7953,
      "training_step_time": 9.933856964111328
    },
    {
      "epoch": 3.0338287353515626e-05,
      "model_forward_time": 3.1380794048309326,
      "step": 7953
    },
    {
      "epoch": 3.0338287353515626e-05,
      "step": 7953,
      "training_step_time": 9.911479234695435
    },
    {
      "epoch": 3.0338287353515626e-05,
      "model_forward_time": 3.1400229930877686,
      "step": 7953
    },
    {
      "epoch": 3.0338287353515626e-05,
      "step": 7953,
      "training_step_time": 9.911094903945923
    },
    {
      "epoch": 3.0338287353515626e-05,
      "model_forward_time": 3.132124423980713,
      "step": 7953
    },
    {
      "epoch": 3.0338287353515626e-05,
      "step": 7953,
      "training_step_time": 10.271791458129883
    },
    {
      "epoch": 3.034210205078125e-05,
      "model_forward_time": 3.43019437789917,
      "step": 7954
    },
    {
      "epoch": 3.034210205078125e-05,
      "step": 7954,
      "training_step_time": 10.20437240600586
    },
    {
      "epoch": 3.034210205078125e-05,
      "model_forward_time": 3.1351070404052734,
      "step": 7954
    },
    {
      "epoch": 3.034210205078125e-05,
      "step": 7954,
      "training_step_time": 9.90764570236206
    },
    {
      "epoch": 3.034210205078125e-05,
      "model_forward_time": 3.1371560096740723,
      "step": 7954
    },
    {
      "epoch": 3.034210205078125e-05,
      "step": 7954,
      "training_step_time": 9.909011125564575
    },
    {
      "epoch": 3.034210205078125e-05,
      "model_forward_time": 3.133014678955078,
      "step": 7954
    },
    {
      "epoch": 3.034210205078125e-05,
      "step": 7954,
      "training_step_time": 10.30857229232788
    },
    {
      "epoch": 3.0345916748046875e-05,
      "model_forward_time": 3.154815435409546,
      "step": 7955
    },
    {
      "dynamics_loss_avg": 0.07401942666620016,
      "epoch": 3.0345916748046875e-05,
      "step": 7955
    },
    {
      "action_loss_avg": 0.02722259722650051,
      "epoch": 3.0345916748046875e-05,
      "step": 7955
    },
    {
      "epoch": 3.0345916748046875e-05,
      "step": 7955,
      "training_step_time": 9.92958116531372
    },
    {
      "epoch": 3.0345916748046875e-05,
      "model_forward_time": 3.416386842727661,
      "step": 7955
    },
    {
      "epoch": 3.0345916748046875e-05,
      "step": 7955,
      "training_step_time": 10.186478614807129
    },
    {
      "epoch": 3.0345916748046875e-05,
      "model_forward_time": 2.4359521865844727,
      "step": 7955
    },
    {
      "epoch": 3.0345916748046875e-05,
      "step": 7955,
      "training_step_time": 7.4611945152282715
    },
    {
      "epoch": 3.0345916748046875e-05,
      "model_forward_time": 3.1526882648468018,
      "step": 7955
    },
    {
      "epoch": 3.0345916748046875e-05,
      "step": 7955,
      "training_step_time": 10.322123289108276
    },
    {
      "epoch": 3.03497314453125e-05,
      "model_forward_time": 3.17975115776062,
      "step": 7956
    },
    {
      "epoch": 3.03497314453125e-05,
      "step": 7956,
      "training_step_time": 9.947128534317017
    },
    {
      "epoch": 3.03497314453125e-05,
      "model_forward_time": 3.138664960861206,
      "step": 7956
    },
    {
      "epoch": 3.03497314453125e-05,
      "step": 7956,
      "training_step_time": 9.903188228607178
    },
    {
      "epoch": 3.03497314453125e-05,
      "model_forward_time": 3.1416923999786377,
      "step": 7956
    },
    {
      "epoch": 3.03497314453125e-05,
      "step": 7956,
      "training_step_time": 10.221159219741821
    },
    {
      "epoch": 3.03497314453125e-05,
      "model_forward_time": 2.436077833175659,
      "step": 7956
    },
    {
      "epoch": 3.03497314453125e-05,
      "step": 7956,
      "training_step_time": 10.015062808990479
    },
    {
      "epoch": 3.0353546142578124e-05,
      "model_forward_time": 3.156388998031616,
      "step": 7957
    },
    {
      "epoch": 3.0353546142578124e-05,
      "step": 7957,
      "training_step_time": 9.94449782371521
    },
    {
      "epoch": 3.0353546142578124e-05,
      "model_forward_time": 3.140869617462158,
      "step": 7957
    },
    {
      "epoch": 3.0353546142578124e-05,
      "step": 7957,
      "training_step_time": 9.910869359970093
    },
    {
      "epoch": 3.0353546142578124e-05,
      "model_forward_time": 3.140228033065796,
      "step": 7957
    },
    {
      "dynamics_loss_avg": 0.05215054266154766,
      "epoch": 3.0353546142578124e-05,
      "step": 7957
    },
    {
      "action_loss_avg": 0.018700793152675033,
      "epoch": 3.0353546142578124e-05,
      "step": 7957
    },
    {
      "epoch": 3.0353546142578124e-05,
      "step": 7957,
      "training_step_time": 9.916188716888428
    },
    {
      "epoch": 3.0353546142578124e-05,
      "model_forward_time": 2.43778395652771,
      "step": 7957
    },
    {
      "epoch": 3.0353546142578124e-05,
      "step": 7957,
      "training_step_time": 10.277047395706177
    },
    {
      "epoch": 3.035736083984375e-05,
      "model_forward_time": 3.1484432220458984,
      "step": 7958
    },
    {
      "epoch": 3.035736083984375e-05,
      "step": 7958,
      "training_step_time": 9.917020559310913
    },
    {
      "epoch": 3.035736083984375e-05,
      "model_forward_time": 3.141234874725342,
      "step": 7958
    },
    {
      "epoch": 3.035736083984375e-05,
      "step": 7958,
      "training_step_time": 9.911232948303223
    },
    {
      "epoch": 3.035736083984375e-05,
      "model_forward_time": 2.439875841140747,
      "step": 7958
    },
    {
      "epoch": 3.035736083984375e-05,
      "step": 7958,
      "training_step_time": 7.472830295562744
    },
    {
      "epoch": 3.035736083984375e-05,
      "model_forward_time": 3.1457767486572266,
      "step": 7958
    },
    {
      "epoch": 3.035736083984375e-05,
      "step": 7958,
      "training_step_time": 10.269457578659058
    },
    {
      "epoch": 3.0361175537109374e-05,
      "model_forward_time": 3.151064157485962,
      "step": 7959
    },
    {
      "epoch": 3.0361175537109374e-05,
      "step": 7959,
      "training_step_time": 9.933091163635254
    },
    {
      "epoch": 3.0361175537109374e-05,
      "model_forward_time": 3.4521327018737793,
      "step": 7959
    },
    {
      "epoch": 3.0361175537109374e-05,
      "step": 7959,
      "training_step_time": 10.226959705352783
    },
    {
      "epoch": 3.0361175537109374e-05,
      "model_forward_time": 3.137085199356079,
      "step": 7959
    },
    {
      "epoch": 3.0361175537109374e-05,
      "step": 7959,
      "training_step_time": 9.907981634140015
    },
    {
      "epoch": 3.0361175537109374e-05,
      "model_forward_time": 2.435431480407715,
      "step": 7959
    },
    {
      "epoch": 3.0361175537109374e-05,
      "step": 7959,
      "training_step_time": 10.295738935470581
    },
    {
      "epoch": 3.0364990234375e-05,
      "grad_norm": 0.5499204397201538,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 7960
    },
    {
      "epoch": 3.0364990234375e-05,
      "model_forward_time": 3.1461727619171143,
      "step": 7960
    },
    {
      "dynamics_loss_avg": 0.06374213937669992,
      "epoch": 3.0364990234375e-05,
      "step": 7960
    },
    {
      "action_loss_avg": 0.020479001710191368,
      "epoch": 3.0364990234375e-05,
      "step": 7960
    },
    {
      "epoch": 3.0364990234375e-05,
      "step": 7960,
      "training_step_time": 9.923189401626587
    },
    {
      "epoch": 3.0364990234375e-05,
      "model_forward_time": 3.1325504779815674,
      "step": 7960
    },
    {
      "epoch": 3.0364990234375e-05,
      "step": 7960,
      "training_step_time": 10.18143630027771
    },
    {
      "epoch": 3.0364990234375e-05,
      "model_forward_time": 3.132561445236206,
      "step": 7960
    },
    {
      "epoch": 3.0364990234375e-05,
      "step": 7960,
      "training_step_time": 9.901753902435303
    },
    {
      "epoch": 3.0364990234375e-05,
      "model_forward_time": 3.1375346183776855,
      "step": 7960
    },
    {
      "epoch": 3.0364990234375e-05,
      "step": 7960,
      "training_step_time": 10.245224714279175
    },
    {
      "epoch": 3.0368804931640626e-05,
      "model_forward_time": 3.15307879447937,
      "step": 7961
    },
    {
      "epoch": 3.0368804931640626e-05,
      "step": 7961,
      "training_step_time": 9.93233609199524
    },
    {
      "epoch": 3.0368804931640626e-05,
      "model_forward_time": 3.13704514503479,
      "step": 7961
    },
    {
      "epoch": 3.0368804931640626e-05,
      "step": 7961,
      "training_step_time": 9.925339460372925
    },
    {
      "epoch": 3.0368804931640626e-05,
      "model_forward_time": 3.431218147277832,
      "step": 7961
    },
    {
      "epoch": 3.0368804931640626e-05,
      "step": 7961,
      "training_step_time": 10.208475112915039
    },
    {
      "epoch": 3.0368804931640626e-05,
      "model_forward_time": 3.1389589309692383,
      "step": 7961
    },
    {
      "epoch": 3.0368804931640626e-05,
      "step": 7961,
      "training_step_time": 10.259180307388306
    },
    {
      "epoch": 3.037261962890625e-05,
      "model_forward_time": 2.4469759464263916,
      "step": 7962
    },
    {
      "epoch": 3.037261962890625e-05,
      "step": 7962,
      "training_step_time": 7.479719638824463
    },
    {
      "epoch": 3.037261962890625e-05,
      "model_forward_time": 3.1328516006469727,
      "step": 7962
    },
    {
      "epoch": 3.037261962890625e-05,
      "step": 7962,
      "training_step_time": 9.908732175827026
    },
    {
      "epoch": 3.037261962890625e-05,
      "model_forward_time": 3.1384928226470947,
      "step": 7962
    },
    {
      "dynamics_loss_avg": 0.08229073435068131,
      "epoch": 3.037261962890625e-05,
      "step": 7962
    },
    {
      "action_loss_avg": 0.025816951505839826,
      "epoch": 3.037261962890625e-05,
      "step": 7962
    },
    {
      "epoch": 3.037261962890625e-05,
      "step": 7962,
      "training_step_time": 9.925997495651245
    },
    {
      "epoch": 3.037261962890625e-05,
      "model_forward_time": 3.139613628387451,
      "step": 7962
    },
    {
      "epoch": 3.037261962890625e-05,
      "step": 7962,
      "training_step_time": 10.304645776748657
    },
    {
      "epoch": 3.0376434326171875e-05,
      "model_forward_time": 3.150930166244507,
      "step": 7963
    },
    {
      "epoch": 3.0376434326171875e-05,
      "step": 7963,
      "training_step_time": 9.929242372512817
    },
    {
      "epoch": 3.0376434326171875e-05,
      "model_forward_time": 3.135483980178833,
      "step": 7963
    },
    {
      "epoch": 3.0376434326171875e-05,
      "step": 7963,
      "training_step_time": 9.916194915771484
    },
    {
      "epoch": 3.0376434326171875e-05,
      "model_forward_time": 3.1430540084838867,
      "step": 7963
    },
    {
      "epoch": 3.0376434326171875e-05,
      "step": 7963,
      "training_step_time": 9.922300338745117
    },
    {
      "epoch": 3.0376434326171875e-05,
      "model_forward_time": 3.1396565437316895,
      "step": 7963
    },
    {
      "epoch": 3.0376434326171875e-05,
      "step": 7963,
      "training_step_time": 10.294912338256836
    },
    {
      "epoch": 3.03802490234375e-05,
      "model_forward_time": 3.4260103702545166,
      "step": 7964
    },
    {
      "epoch": 3.03802490234375e-05,
      "step": 7964,
      "training_step_time": 10.203765153884888
    },
    {
      "epoch": 3.03802490234375e-05,
      "model_forward_time": 3.136983871459961,
      "step": 7964
    },
    {
      "epoch": 3.03802490234375e-05,
      "step": 7964,
      "training_step_time": 9.913570880889893
    },
    {
      "epoch": 3.03802490234375e-05,
      "model_forward_time": 3.141695737838745,
      "step": 7964
    },
    {
      "epoch": 3.03802490234375e-05,
      "step": 7964,
      "training_step_time": 9.920671224594116
    },
    {
      "epoch": 3.03802490234375e-05,
      "model_forward_time": 3.1399893760681152,
      "step": 7964
    },
    {
      "epoch": 3.03802490234375e-05,
      "step": 7964,
      "training_step_time": 10.297793626785278
    },
    {
      "epoch": 3.0384063720703124e-05,
      "model_forward_time": 3.1433088779449463,
      "step": 7965
    },
    {
      "dynamics_loss_avg": 0.05293168947100639,
      "epoch": 3.0384063720703124e-05,
      "step": 7965
    },
    {
      "action_loss_avg": 0.017811361514031888,
      "epoch": 3.0384063720703124e-05,
      "step": 7965
    },
    {
      "epoch": 3.0384063720703124e-05,
      "step": 7965,
      "training_step_time": 9.92598819732666
    },
    {
      "epoch": 3.0384063720703124e-05,
      "model_forward_time": 3.4301769733428955,
      "step": 7965
    },
    {
      "epoch": 3.0384063720703124e-05,
      "step": 7965,
      "training_step_time": 10.210291862487793
    },
    {
      "epoch": 3.0384063720703124e-05,
      "model_forward_time": 3.143359422683716,
      "step": 7965
    },
    {
      "epoch": 3.0384063720703124e-05,
      "step": 7965,
      "training_step_time": 9.930553197860718
    },
    {
      "epoch": 3.0384063720703124e-05,
      "model_forward_time": 3.139617443084717,
      "step": 7965
    },
    {
      "epoch": 3.0384063720703124e-05,
      "step": 7965,
      "training_step_time": 10.281932592391968
    },
    {
      "epoch": 3.038787841796875e-05,
      "model_forward_time": 3.1541244983673096,
      "step": 7966
    },
    {
      "epoch": 3.038787841796875e-05,
      "step": 7966,
      "training_step_time": 9.932304382324219
    },
    {
      "epoch": 3.038787841796875e-05,
      "model_forward_time": 3.143232822418213,
      "step": 7966
    },
    {
      "epoch": 3.038787841796875e-05,
      "step": 7966,
      "training_step_time": 9.91948390007019
    },
    {
      "epoch": 3.038787841796875e-05,
      "model_forward_time": 3.4280190467834473,
      "step": 7966
    },
    {
      "epoch": 3.038787841796875e-05,
      "step": 7966,
      "training_step_time": 10.203107357025146
    },
    {
      "epoch": 3.038787841796875e-05,
      "model_forward_time": 3.1388559341430664,
      "step": 7966
    },
    {
      "epoch": 3.038787841796875e-05,
      "step": 7966,
      "training_step_time": 10.302085161209106
    },
    {
      "epoch": 3.0391693115234374e-05,
      "model_forward_time": 3.1598284244537354,
      "step": 7967
    },
    {
      "epoch": 3.0391693115234374e-05,
      "step": 7967,
      "training_step_time": 9.93922472000122
    },
    {
      "epoch": 3.0391693115234374e-05,
      "model_forward_time": 3.137791872024536,
      "step": 7967
    },
    {
      "epoch": 3.0391693115234374e-05,
      "step": 7967,
      "training_step_time": 9.910383939743042
    },
    {
      "epoch": 3.0391693115234374e-05,
      "model_forward_time": 3.1366732120513916,
      "step": 7967
    },
    {
      "dynamics_loss_avg": 0.053153205290436746,
      "epoch": 3.0391693115234374e-05,
      "step": 7967
    },
    {
      "action_loss_avg": 0.018188449274748563,
      "epoch": 3.0391693115234374e-05,
      "step": 7967
    },
    {
      "epoch": 3.0391693115234374e-05,
      "step": 7967,
      "training_step_time": 9.913065195083618
    },
    {
      "epoch": 3.0391693115234374e-05,
      "model_forward_time": 3.399869441986084,
      "step": 7967
    },
    {
      "epoch": 3.0391693115234374e-05,
      "step": 7967,
      "training_step_time": 10.268125057220459
    },
    {
      "epoch": 3.03955078125e-05,
      "model_forward_time": 3.151597261428833,
      "step": 7968
    },
    {
      "epoch": 3.03955078125e-05,
      "step": 7968,
      "training_step_time": 9.93158769607544
    },
    {
      "epoch": 3.03955078125e-05,
      "model_forward_time": 3.1400671005249023,
      "step": 7968
    },
    {
      "epoch": 3.03955078125e-05,
      "step": 7968,
      "training_step_time": 9.916494846343994
    },
    {
      "epoch": 3.03955078125e-05,
      "model_forward_time": 3.1361992359161377,
      "step": 7968
    },
    {
      "epoch": 3.03955078125e-05,
      "step": 7968,
      "training_step_time": 9.91189455986023
    },
    {
      "epoch": 3.03955078125e-05,
      "model_forward_time": 3.140855312347412,
      "step": 7968
    },
    {
      "epoch": 3.03955078125e-05,
      "step": 7968,
      "training_step_time": 10.25939679145813
    },
    {
      "epoch": 3.0399322509765626e-05,
      "model_forward_time": 3.453913450241089,
      "step": 7969
    },
    {
      "epoch": 3.0399322509765626e-05,
      "step": 7969,
      "training_step_time": 10.232254028320312
    },
    {
      "epoch": 3.0399322509765626e-05,
      "model_forward_time": 3.1401712894439697,
      "step": 7969
    },
    {
      "epoch": 3.0399322509765626e-05,
      "step": 7969,
      "training_step_time": 9.912856578826904
    },
    {
      "epoch": 3.0399322509765626e-05,
      "model_forward_time": 2.4380698204040527,
      "step": 7969
    },
    {
      "epoch": 3.0399322509765626e-05,
      "step": 7969,
      "training_step_time": 7.473292350769043
    },
    {
      "epoch": 3.0399322509765626e-05,
      "model_forward_time": 3.1421303749084473,
      "step": 7969
    },
    {
      "epoch": 3.0399322509765626e-05,
      "step": 7969,
      "training_step_time": 10.308474063873291
    },
    {
      "epoch": 3.040313720703125e-05,
      "grad_norm": 0.34781864285469055,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 7970
    },
    {
      "epoch": 3.040313720703125e-05,
      "model_forward_time": 3.1550710201263428,
      "step": 7970
    },
    {
      "dynamics_loss_avg": 0.06472952030599118,
      "epoch": 3.040313720703125e-05,
      "step": 7970
    },
    {
      "action_loss_avg": 0.021380497235804798,
      "epoch": 3.040313720703125e-05,
      "step": 7970
    },
    {
      "epoch": 3.040313720703125e-05,
      "step": 7970,
      "training_step_time": 9.956074714660645
    },
    {
      "epoch": 3.040313720703125e-05,
      "model_forward_time": 3.459717273712158,
      "step": 7970
    },
    {
      "epoch": 3.040313720703125e-05,
      "step": 7970,
      "training_step_time": 10.241567850112915
    },
    {
      "epoch": 3.040313720703125e-05,
      "model_forward_time": 3.1410131454467773,
      "step": 7970
    },
    {
      "epoch": 3.040313720703125e-05,
      "step": 7970,
      "training_step_time": 9.918557167053223
    },
    {
      "epoch": 3.040313720703125e-05,
      "model_forward_time": 3.145956516265869,
      "step": 7970
    },
    {
      "epoch": 3.040313720703125e-05,
      "step": 7970,
      "training_step_time": 10.281641244888306
    },
    {
      "epoch": 3.0406951904296875e-05,
      "model_forward_time": 3.1539058685302734,
      "step": 7971
    },
    {
      "epoch": 3.0406951904296875e-05,
      "step": 7971,
      "training_step_time": 9.938715696334839
    },
    {
      "epoch": 3.0406951904296875e-05,
      "model_forward_time": 3.1374011039733887,
      "step": 7971
    },
    {
      "epoch": 3.0406951904296875e-05,
      "step": 7971,
      "training_step_time": 9.912352323532104
    },
    {
      "epoch": 3.0406951904296875e-05,
      "model_forward_time": 3.136631965637207,
      "step": 7971
    },
    {
      "epoch": 3.0406951904296875e-05,
      "step": 7971,
      "training_step_time": 10.191242933273315
    },
    {
      "epoch": 3.0406951904296875e-05,
      "model_forward_time": 3.137446880340576,
      "step": 7971
    },
    {
      "epoch": 3.0406951904296875e-05,
      "step": 7971,
      "training_step_time": 10.012102842330933
    },
    {
      "epoch": 3.04107666015625e-05,
      "model_forward_time": 3.1522281169891357,
      "step": 7972
    },
    {
      "epoch": 3.04107666015625e-05,
      "step": 7972,
      "training_step_time": 9.926889181137085
    },
    {
      "epoch": 3.04107666015625e-05,
      "model_forward_time": 3.1354360580444336,
      "step": 7972
    },
    {
      "epoch": 3.04107666015625e-05,
      "step": 7972,
      "training_step_time": 9.911076068878174
    },
    {
      "epoch": 3.04107666015625e-05,
      "model_forward_time": 3.1390795707702637,
      "step": 7972
    },
    {
      "dynamics_loss_avg": 0.06402569785714149,
      "epoch": 3.04107666015625e-05,
      "step": 7972
    },
    {
      "action_loss_avg": 0.022829485964030026,
      "epoch": 3.04107666015625e-05,
      "step": 7972
    },
    {
      "epoch": 3.04107666015625e-05,
      "step": 7972,
      "training_step_time": 9.914912939071655
    },
    {
      "epoch": 3.04107666015625e-05,
      "model_forward_time": 3.423473358154297,
      "step": 7972
    },
    {
      "epoch": 3.04107666015625e-05,
      "step": 7972,
      "training_step_time": 10.319022417068481
    },
    {
      "epoch": 3.0414581298828124e-05,
      "model_forward_time": 3.1568307876586914,
      "step": 7973
    },
    {
      "epoch": 3.0414581298828124e-05,
      "step": 7973,
      "training_step_time": 9.934524536132812
    },
    {
      "epoch": 3.0414581298828124e-05,
      "model_forward_time": 3.13875412940979,
      "step": 7973
    },
    {
      "epoch": 3.0414581298828124e-05,
      "step": 7973,
      "training_step_time": 9.91650128364563
    },
    {
      "epoch": 3.0414581298828124e-05,
      "model_forward_time": 3.140385150909424,
      "step": 7973
    },
    {
      "epoch": 3.0414581298828124e-05,
      "step": 7973,
      "training_step_time": 9.91788625717163
    },
    {
      "epoch": 3.0414581298828124e-05,
      "model_forward_time": 3.1371309757232666,
      "step": 7973
    },
    {
      "epoch": 3.0414581298828124e-05,
      "step": 7973,
      "training_step_time": 10.227701663970947
    },
    {
      "epoch": 3.041839599609375e-05,
      "model_forward_time": 3.1451258659362793,
      "step": 7974
    },
    {
      "epoch": 3.041839599609375e-05,
      "step": 7974,
      "training_step_time": 10.1969473361969
    },
    {
      "epoch": 3.041839599609375e-05,
      "model_forward_time": 3.135392665863037,
      "step": 7974
    },
    {
      "epoch": 3.041839599609375e-05,
      "step": 7974,
      "training_step_time": 9.906962156295776
    },
    {
      "epoch": 3.041839599609375e-05,
      "model_forward_time": 3.137030839920044,
      "step": 7974
    },
    {
      "epoch": 3.041839599609375e-05,
      "step": 7974,
      "training_step_time": 9.910563945770264
    },
    {
      "epoch": 3.041839599609375e-05,
      "model_forward_time": 3.131429672241211,
      "step": 7974
    },
    {
      "epoch": 3.041839599609375e-05,
      "step": 7974,
      "training_step_time": 10.241262435913086
    },
    {
      "epoch": 3.0422210693359374e-05,
      "model_forward_time": 3.1513757705688477,
      "step": 7975
    },
    {
      "dynamics_loss_avg": 0.049522423464804885,
      "epoch": 3.0422210693359374e-05,
      "step": 7975
    },
    {
      "action_loss_avg": 0.014608868793584407,
      "epoch": 3.0422210693359374e-05,
      "step": 7975
    },
    {
      "epoch": 3.0422210693359374e-05,
      "step": 7975,
      "training_step_time": 9.935969829559326
    },
    {
      "epoch": 3.0422210693359374e-05,
      "model_forward_time": 3.4077651500701904,
      "step": 7975
    },
    {
      "epoch": 3.0422210693359374e-05,
      "step": 7975,
      "training_step_time": 10.180766344070435
    },
    {
      "epoch": 3.0422210693359374e-05,
      "model_forward_time": 3.149177312850952,
      "step": 7975
    },
    {
      "epoch": 3.0422210693359374e-05,
      "step": 7975,
      "training_step_time": 9.926023244857788
    },
    {
      "epoch": 3.0422210693359374e-05,
      "model_forward_time": 3.1418886184692383,
      "step": 7975
    },
    {
      "epoch": 3.0422210693359374e-05,
      "step": 7975,
      "training_step_time": 10.311790466308594
    },
    {
      "epoch": 3.0426025390625e-05,
      "model_forward_time": 3.160494565963745,
      "step": 7976
    },
    {
      "epoch": 3.0426025390625e-05,
      "step": 7976,
      "training_step_time": 9.939582347869873
    },
    {
      "epoch": 3.0426025390625e-05,
      "model_forward_time": 3.1488590240478516,
      "step": 7976
    },
    {
      "epoch": 3.0426025390625e-05,
      "step": 7976,
      "training_step_time": 9.929860830307007
    },
    {
      "epoch": 3.0426025390625e-05,
      "model_forward_time": 2.4303901195526123,
      "step": 7976
    },
    {
      "epoch": 3.0426025390625e-05,
      "step": 7976,
      "training_step_time": 7.7783520221710205
    },
    {
      "epoch": 3.0426025390625e-05,
      "model_forward_time": 3.1391427516937256,
      "step": 7976
    },
    {
      "epoch": 3.0426025390625e-05,
      "step": 7976,
      "training_step_time": 10.288772106170654
    },
    {
      "epoch": 3.0429840087890626e-05,
      "model_forward_time": 3.156729221343994,
      "step": 7977
    },
    {
      "epoch": 3.0429840087890626e-05,
      "step": 7977,
      "training_step_time": 9.935318231582642
    },
    {
      "epoch": 3.0429840087890626e-05,
      "model_forward_time": 3.139779567718506,
      "step": 7977
    },
    {
      "epoch": 3.0429840087890626e-05,
      "step": 7977,
      "training_step_time": 9.913519859313965
    },
    {
      "epoch": 3.0429840087890626e-05,
      "model_forward_time": 3.1429474353790283,
      "step": 7977
    },
    {
      "dynamics_loss_avg": 0.05781356487423182,
      "epoch": 3.0429840087890626e-05,
      "step": 7977
    },
    {
      "action_loss_avg": 0.020931521104648708,
      "epoch": 3.0429840087890626e-05,
      "step": 7977
    },
    {
      "epoch": 3.0429840087890626e-05,
      "step": 7977,
      "training_step_time": 9.925140857696533
    },
    {
      "epoch": 3.0429840087890626e-05,
      "model_forward_time": 2.7091846466064453,
      "step": 7977
    },
    {
      "epoch": 3.0429840087890626e-05,
      "step": 7977,
      "training_step_time": 10.32298755645752
    },
    {
      "epoch": 3.043365478515625e-05,
      "model_forward_time": 3.1599299907684326,
      "step": 7978
    },
    {
      "epoch": 3.043365478515625e-05,
      "step": 7978,
      "training_step_time": 9.940689086914062
    },
    {
      "epoch": 3.043365478515625e-05,
      "model_forward_time": 3.138847589492798,
      "step": 7978
    },
    {
      "epoch": 3.043365478515625e-05,
      "step": 7978,
      "training_step_time": 9.912610530853271
    },
    {
      "epoch": 3.043365478515625e-05,
      "model_forward_time": 3.1338350772857666,
      "step": 7978
    },
    {
      "epoch": 3.043365478515625e-05,
      "step": 7978,
      "training_step_time": 9.906593084335327
    },
    {
      "epoch": 3.043365478515625e-05,
      "model_forward_time": 3.1370556354522705,
      "step": 7978
    },
    {
      "epoch": 3.043365478515625e-05,
      "step": 7978,
      "training_step_time": 10.013797998428345
    },
    {
      "epoch": 3.0437469482421875e-05,
      "model_forward_time": 3.1542906761169434,
      "step": 7979
    },
    {
      "epoch": 3.0437469482421875e-05,
      "step": 7979,
      "training_step_time": 10.24765658378601
    },
    {
      "epoch": 3.0437469482421875e-05,
      "model_forward_time": 3.1446938514709473,
      "step": 7979
    },
    {
      "epoch": 3.0437469482421875e-05,
      "step": 7979,
      "training_step_time": 9.93010950088501
    },
    {
      "epoch": 3.0437469482421875e-05,
      "model_forward_time": 3.1393630504608154,
      "step": 7979
    },
    {
      "epoch": 3.0437469482421875e-05,
      "step": 7979,
      "training_step_time": 9.91332745552063
    },
    {
      "epoch": 3.0437469482421875e-05,
      "model_forward_time": 3.1425280570983887,
      "step": 7979
    },
    {
      "epoch": 3.0437469482421875e-05,
      "step": 7979,
      "training_step_time": 10.261357307434082
    },
    {
      "epoch": 3.04412841796875e-05,
      "grad_norm": 0.3494287133216858,
      "learning_rate": 1e-05,
      "loss": 0.0822,
      "step": 7980
    },
    {
      "epoch": 3.04412841796875e-05,
      "model_forward_time": 3.152099847793579,
      "step": 7980
    },
    {
      "dynamics_loss_avg": 0.05641402620822191,
      "epoch": 3.04412841796875e-05,
      "step": 7980
    },
    {
      "action_loss_avg": 0.017829704005271195,
      "epoch": 3.04412841796875e-05,
      "step": 7980
    },
    {
      "epoch": 3.04412841796875e-05,
      "step": 7980,
      "training_step_time": 9.933565616607666
    },
    {
      "epoch": 3.04412841796875e-05,
      "model_forward_time": 3.131812572479248,
      "step": 7980
    },
    {
      "epoch": 3.04412841796875e-05,
      "step": 7980,
      "training_step_time": 10.188075542449951
    },
    {
      "epoch": 3.04412841796875e-05,
      "model_forward_time": 3.1340537071228027,
      "step": 7980
    },
    {
      "epoch": 3.04412841796875e-05,
      "step": 7980,
      "training_step_time": 9.908135175704956
    },
    {
      "epoch": 3.04412841796875e-05,
      "model_forward_time": 3.1405980587005615,
      "step": 7980
    },
    {
      "epoch": 3.04412841796875e-05,
      "step": 7980,
      "training_step_time": 10.306396007537842
    },
    {
      "epoch": 3.0445098876953125e-05,
      "model_forward_time": 3.1506385803222656,
      "step": 7981
    },
    {
      "epoch": 3.0445098876953125e-05,
      "step": 7981,
      "training_step_time": 9.926302671432495
    },
    {
      "epoch": 3.0445098876953125e-05,
      "model_forward_time": 3.134999990463257,
      "step": 7981
    },
    {
      "epoch": 3.0445098876953125e-05,
      "step": 7981,
      "training_step_time": 9.907416820526123
    },
    {
      "epoch": 3.0445098876953125e-05,
      "model_forward_time": 3.4490065574645996,
      "step": 7981
    },
    {
      "epoch": 3.0445098876953125e-05,
      "step": 7981,
      "training_step_time": 10.223298072814941
    },
    {
      "epoch": 3.0445098876953125e-05,
      "model_forward_time": 3.1401636600494385,
      "step": 7981
    },
    {
      "epoch": 3.0445098876953125e-05,
      "step": 7981,
      "training_step_time": 10.280573606491089
    },
    {
      "epoch": 3.044891357421875e-05,
      "model_forward_time": 2.4492483139038086,
      "step": 7982
    },
    {
      "epoch": 3.044891357421875e-05,
      "step": 7982,
      "training_step_time": 7.482156038284302
    },
    {
      "epoch": 3.044891357421875e-05,
      "model_forward_time": 3.136575222015381,
      "step": 7982
    },
    {
      "epoch": 3.044891357421875e-05,
      "step": 7982,
      "training_step_time": 9.911333322525024
    },
    {
      "epoch": 3.044891357421875e-05,
      "model_forward_time": 3.134896755218506,
      "step": 7982
    },
    {
      "dynamics_loss_avg": 0.07402013801038265,
      "epoch": 3.044891357421875e-05,
      "step": 7982
    },
    {
      "action_loss_avg": 0.025419728923588993,
      "epoch": 3.044891357421875e-05,
      "step": 7982
    },
    {
      "epoch": 3.044891357421875e-05,
      "step": 7982,
      "training_step_time": 9.910197257995605
    },
    {
      "epoch": 3.044891357421875e-05,
      "model_forward_time": 3.132990837097168,
      "step": 7982
    },
    {
      "epoch": 3.044891357421875e-05,
      "step": 7982,
      "training_step_time": 10.265672445297241
    },
    {
      "epoch": 3.0452728271484374e-05,
      "model_forward_time": 3.1591455936431885,
      "step": 7983
    },
    {
      "epoch": 3.0452728271484374e-05,
      "step": 7983,
      "training_step_time": 9.949782133102417
    },
    {
      "epoch": 3.0452728271484374e-05,
      "model_forward_time": 3.150521993637085,
      "step": 7983
    },
    {
      "epoch": 3.0452728271484374e-05,
      "step": 7983,
      "training_step_time": 9.934441328048706
    },
    {
      "epoch": 3.0452728271484374e-05,
      "model_forward_time": 3.1531004905700684,
      "step": 7983
    },
    {
      "epoch": 3.0452728271484374e-05,
      "step": 7983,
      "training_step_time": 9.940702676773071
    },
    {
      "epoch": 3.0452728271484374e-05,
      "model_forward_time": 3.1528570652008057,
      "step": 7983
    },
    {
      "epoch": 3.0452728271484374e-05,
      "step": 7983,
      "training_step_time": 10.309579372406006
    },
    {
      "epoch": 3.045654296875e-05,
      "model_forward_time": 3.465538740158081,
      "step": 7984
    },
    {
      "epoch": 3.045654296875e-05,
      "step": 7984,
      "training_step_time": 10.244896650314331
    },
    {
      "epoch": 3.045654296875e-05,
      "model_forward_time": 3.1355838775634766,
      "step": 7984
    },
    {
      "epoch": 3.045654296875e-05,
      "step": 7984,
      "training_step_time": 9.920254468917847
    },
    {
      "epoch": 3.045654296875e-05,
      "model_forward_time": 3.1423330307006836,
      "step": 7984
    },
    {
      "epoch": 3.045654296875e-05,
      "step": 7984,
      "training_step_time": 9.918385982513428
    },
    {
      "epoch": 3.045654296875e-05,
      "model_forward_time": 3.132340431213379,
      "step": 7984
    },
    {
      "epoch": 3.045654296875e-05,
      "step": 7984,
      "training_step_time": 10.278825759887695
    },
    {
      "epoch": 3.0460357666015626e-05,
      "model_forward_time": 3.152575731277466,
      "step": 7985
    },
    {
      "dynamics_loss_avg": 0.06106156092137098,
      "epoch": 3.0460357666015626e-05,
      "step": 7985
    },
    {
      "action_loss_avg": 0.018906350107863547,
      "epoch": 3.0460357666015626e-05,
      "step": 7985
    },
    {
      "epoch": 3.0460357666015626e-05,
      "step": 7985,
      "training_step_time": 9.931087017059326
    },
    {
      "epoch": 3.0460357666015626e-05,
      "model_forward_time": 3.435804605484009,
      "step": 7985
    },
    {
      "epoch": 3.0460357666015626e-05,
      "step": 7985,
      "training_step_time": 10.209993124008179
    },
    {
      "epoch": 3.0460357666015626e-05,
      "model_forward_time": 3.135563611984253,
      "step": 7985
    },
    {
      "epoch": 3.0460357666015626e-05,
      "step": 7985,
      "training_step_time": 9.91306471824646
    },
    {
      "epoch": 3.0460357666015626e-05,
      "model_forward_time": 3.1329121589660645,
      "step": 7985
    },
    {
      "epoch": 3.0460357666015626e-05,
      "step": 7985,
      "training_step_time": 10.238759279251099
    },
    {
      "epoch": 3.046417236328125e-05,
      "model_forward_time": 2.444211959838867,
      "step": 7986
    },
    {
      "epoch": 3.046417236328125e-05,
      "step": 7986,
      "training_step_time": 7.4771888256073
    },
    {
      "epoch": 3.046417236328125e-05,
      "model_forward_time": 3.1350343227386475,
      "step": 7986
    },
    {
      "epoch": 3.046417236328125e-05,
      "step": 7986,
      "training_step_time": 9.915390491485596
    },
    {
      "epoch": 3.046417236328125e-05,
      "model_forward_time": 3.4488706588745117,
      "step": 7986
    },
    {
      "epoch": 3.046417236328125e-05,
      "step": 7986,
      "training_step_time": 10.222624063491821
    },
    {
      "epoch": 3.046417236328125e-05,
      "model_forward_time": 3.1335818767547607,
      "step": 7986
    },
    {
      "epoch": 3.046417236328125e-05,
      "step": 7986,
      "training_step_time": 10.327747821807861
    },
    {
      "epoch": 3.0467987060546875e-05,
      "model_forward_time": 3.1393420696258545,
      "step": 7987
    },
    {
      "epoch": 3.0467987060546875e-05,
      "step": 7987,
      "training_step_time": 9.91984224319458
    },
    {
      "epoch": 3.0467987060546875e-05,
      "model_forward_time": 3.1394917964935303,
      "step": 7987
    },
    {
      "epoch": 3.0467987060546875e-05,
      "step": 7987,
      "training_step_time": 9.914128541946411
    },
    {
      "epoch": 3.0467987060546875e-05,
      "model_forward_time": 3.1360225677490234,
      "step": 7987
    },
    {
      "dynamics_loss_avg": 0.04749784767627716,
      "epoch": 3.0467987060546875e-05,
      "step": 7987
    },
    {
      "action_loss_avg": 0.01646025669761002,
      "epoch": 3.0467987060546875e-05,
      "step": 7987
    },
    {
      "epoch": 3.0467987060546875e-05,
      "step": 7987,
      "training_step_time": 9.924340963363647
    },
    {
      "epoch": 3.0467987060546875e-05,
      "model_forward_time": 3.1371147632598877,
      "step": 7987
    },
    {
      "epoch": 3.0467987060546875e-05,
      "step": 7987,
      "training_step_time": 10.016644716262817
    },
    {
      "epoch": 3.04718017578125e-05,
      "model_forward_time": 3.439117670059204,
      "step": 7988
    },
    {
      "epoch": 3.04718017578125e-05,
      "step": 7988,
      "training_step_time": 10.218812942504883
    },
    {
      "epoch": 3.04718017578125e-05,
      "model_forward_time": 3.1400375366210938,
      "step": 7988
    },
    {
      "epoch": 3.04718017578125e-05,
      "step": 7988,
      "training_step_time": 9.924778461456299
    },
    {
      "epoch": 3.04718017578125e-05,
      "model_forward_time": 3.1390111446380615,
      "step": 7988
    },
    {
      "epoch": 3.04718017578125e-05,
      "step": 7988,
      "training_step_time": 9.924034357070923
    },
    {
      "epoch": 3.04718017578125e-05,
      "model_forward_time": 3.140528440475464,
      "step": 7988
    },
    {
      "epoch": 3.04718017578125e-05,
      "step": 7988,
      "training_step_time": 10.306358337402344
    },
    {
      "epoch": 3.0475616455078125e-05,
      "model_forward_time": 3.440670967102051,
      "step": 7989
    },
    {
      "epoch": 3.0475616455078125e-05,
      "step": 7989,
      "training_step_time": 10.224502325057983
    },
    {
      "epoch": 3.0475616455078125e-05,
      "model_forward_time": 3.1564810276031494,
      "step": 7989
    },
    {
      "epoch": 3.0475616455078125e-05,
      "step": 7989,
      "training_step_time": 9.940234899520874
    },
    {
      "epoch": 3.0475616455078125e-05,
      "model_forward_time": 3.1435413360595703,
      "step": 7989
    },
    {
      "epoch": 3.0475616455078125e-05,
      "step": 7989,
      "training_step_time": 9.917939186096191
    },
    {
      "epoch": 3.0475616455078125e-05,
      "model_forward_time": 3.1432411670684814,
      "step": 7989
    },
    {
      "epoch": 3.0475616455078125e-05,
      "step": 7989,
      "training_step_time": 10.018194198608398
    },
    {
      "epoch": 3.047943115234375e-05,
      "grad_norm": 0.6279850006103516,
      "learning_rate": 1e-05,
      "loss": 0.0814,
      "step": 7990
    },
    {
      "epoch": 3.047943115234375e-05,
      "model_forward_time": 3.164118528366089,
      "step": 7990
    },
    {
      "dynamics_loss_avg": 0.0563420532271266,
      "epoch": 3.047943115234375e-05,
      "step": 7990
    },
    {
      "action_loss_avg": 0.016142072807997464,
      "epoch": 3.047943115234375e-05,
      "step": 7990
    },
    {
      "epoch": 3.047943115234375e-05,
      "step": 7990,
      "training_step_time": 9.943576574325562
    },
    {
      "epoch": 3.047943115234375e-05,
      "model_forward_time": 3.133939027786255,
      "step": 7990
    },
    {
      "epoch": 3.047943115234375e-05,
      "step": 7990,
      "training_step_time": 10.220329999923706
    },
    {
      "epoch": 3.047943115234375e-05,
      "model_forward_time": 3.1364293098449707,
      "step": 7990
    },
    {
      "epoch": 3.047943115234375e-05,
      "step": 7990,
      "training_step_time": 9.911316871643066
    },
    {
      "epoch": 3.047943115234375e-05,
      "model_forward_time": 3.139387607574463,
      "step": 7990
    },
    {
      "epoch": 3.047943115234375e-05,
      "step": 7990,
      "training_step_time": 10.262317895889282
    },
    {
      "epoch": 3.0483245849609374e-05,
      "model_forward_time": 3.1503713130950928,
      "step": 7991
    },
    {
      "epoch": 3.0483245849609374e-05,
      "step": 7991,
      "training_step_time": 9.931578397750854
    },
    {
      "epoch": 3.0483245849609374e-05,
      "model_forward_time": 3.1346442699432373,
      "step": 7991
    },
    {
      "epoch": 3.0483245849609374e-05,
      "step": 7991,
      "training_step_time": 9.907843828201294
    },
    {
      "epoch": 3.0483245849609374e-05,
      "model_forward_time": 3.1399776935577393,
      "step": 7991
    },
    {
      "epoch": 3.0483245849609374e-05,
      "step": 7991,
      "training_step_time": 10.208393096923828
    },
    {
      "epoch": 3.0483245849609374e-05,
      "model_forward_time": 3.143646478652954,
      "step": 7991
    },
    {
      "epoch": 3.0483245849609374e-05,
      "step": 7991,
      "training_step_time": 10.283206701278687
    },
    {
      "epoch": 3.0487060546875e-05,
      "model_forward_time": 3.156818151473999,
      "step": 7992
    },
    {
      "epoch": 3.0487060546875e-05,
      "step": 7992,
      "training_step_time": 9.9349946975708
    },
    {
      "epoch": 3.0487060546875e-05,
      "model_forward_time": 3.134836435317993,
      "step": 7992
    },
    {
      "epoch": 3.0487060546875e-05,
      "step": 7992,
      "training_step_time": 9.910948038101196
    },
    {
      "epoch": 3.0487060546875e-05,
      "model_forward_time": 3.1399312019348145,
      "step": 7992
    },
    {
      "dynamics_loss_avg": 0.05137131754308939,
      "epoch": 3.0487060546875e-05,
      "step": 7992
    },
    {
      "action_loss_avg": 0.015560019901022315,
      "epoch": 3.0487060546875e-05,
      "step": 7992
    },
    {
      "epoch": 3.0487060546875e-05,
      "step": 7992,
      "training_step_time": 9.923253536224365
    },
    {
      "epoch": 3.0487060546875e-05,
      "model_forward_time": 3.4607179164886475,
      "step": 7992
    },
    {
      "epoch": 3.0487060546875e-05,
      "step": 7992,
      "training_step_time": 10.334809303283691
    },
    {
      "epoch": 3.0490875244140626e-05,
      "model_forward_time": 3.1554770469665527,
      "step": 7993
    },
    {
      "epoch": 3.0490875244140626e-05,
      "step": 7993,
      "training_step_time": 9.934320449829102
    },
    {
      "epoch": 3.0490875244140626e-05,
      "model_forward_time": 3.1383979320526123,
      "step": 7993
    },
    {
      "epoch": 3.0490875244140626e-05,
      "step": 7993,
      "training_step_time": 9.915767192840576
    },
    {
      "epoch": 3.0490875244140626e-05,
      "model_forward_time": 3.1446621417999268,
      "step": 7993
    },
    {
      "epoch": 3.0490875244140626e-05,
      "step": 7993,
      "training_step_time": 9.92556619644165
    },
    {
      "epoch": 3.0490875244140626e-05,
      "model_forward_time": 3.137068510055542,
      "step": 7993
    },
    {
      "epoch": 3.0490875244140626e-05,
      "step": 7993,
      "training_step_time": 10.02497386932373
    },
    {
      "epoch": 3.049468994140625e-05,
      "model_forward_time": 3.1465659141540527,
      "step": 7994
    },
    {
      "epoch": 3.049468994140625e-05,
      "step": 7994,
      "training_step_time": 10.201648473739624
    },
    {
      "epoch": 3.049468994140625e-05,
      "model_forward_time": 3.1392269134521484,
      "step": 7994
    },
    {
      "epoch": 3.049468994140625e-05,
      "step": 7994,
      "training_step_time": 9.916520357131958
    },
    {
      "epoch": 3.049468994140625e-05,
      "model_forward_time": 3.14375376701355,
      "step": 7994
    },
    {
      "epoch": 3.049468994140625e-05,
      "step": 7994,
      "training_step_time": 9.923400402069092
    },
    {
      "epoch": 3.049468994140625e-05,
      "model_forward_time": 3.137563467025757,
      "step": 7994
    },
    {
      "epoch": 3.049468994140625e-05,
      "step": 7994,
      "training_step_time": 10.294769048690796
    },
    {
      "epoch": 3.0498504638671875e-05,
      "model_forward_time": 3.1486215591430664,
      "step": 7995
    },
    {
      "dynamics_loss_avg": 0.06631245352327823,
      "epoch": 3.0498504638671875e-05,
      "step": 7995
    },
    {
      "action_loss_avg": 0.02144027571193874,
      "epoch": 3.0498504638671875e-05,
      "step": 7995
    },
    {
      "epoch": 3.0498504638671875e-05,
      "step": 7995,
      "training_step_time": 9.931050539016724
    },
    {
      "epoch": 3.0498504638671875e-05,
      "model_forward_time": 3.431973695755005,
      "step": 7995
    },
    {
      "epoch": 3.0498504638671875e-05,
      "step": 7995,
      "training_step_time": 10.205095291137695
    },
    {
      "epoch": 3.0498504638671875e-05,
      "model_forward_time": 3.145209550857544,
      "step": 7995
    },
    {
      "epoch": 3.0498504638671875e-05,
      "step": 7995,
      "training_step_time": 9.91843843460083
    },
    {
      "epoch": 3.0498504638671875e-05,
      "model_forward_time": 2.441337823867798,
      "step": 7995
    },
    {
      "epoch": 3.0498504638671875e-05,
      "step": 7995,
      "training_step_time": 10.277307033538818
    },
    {
      "epoch": 3.05023193359375e-05,
      "model_forward_time": 3.148231267929077,
      "step": 7996
    },
    {
      "epoch": 3.05023193359375e-05,
      "step": 7996,
      "training_step_time": 9.922999858856201
    },
    {
      "epoch": 3.05023193359375e-05,
      "model_forward_time": 3.1473183631896973,
      "step": 7996
    },
    {
      "epoch": 3.05023193359375e-05,
      "step": 7996,
      "training_step_time": 9.919315099716187
    },
    {
      "epoch": 3.05023193359375e-05,
      "model_forward_time": 3.140806198120117,
      "step": 7996
    },
    {
      "epoch": 3.05023193359375e-05,
      "step": 7996,
      "training_step_time": 10.192166090011597
    },
    {
      "epoch": 3.05023193359375e-05,
      "model_forward_time": 3.1431679725646973,
      "step": 7996
    },
    {
      "epoch": 3.05023193359375e-05,
      "step": 7996,
      "training_step_time": 10.316555976867676
    },
    {
      "epoch": 3.0506134033203125e-05,
      "model_forward_time": 3.1541523933410645,
      "step": 7997
    },
    {
      "epoch": 3.0506134033203125e-05,
      "step": 7997,
      "training_step_time": 9.926326036453247
    },
    {
      "epoch": 3.0506134033203125e-05,
      "model_forward_time": 3.145670175552368,
      "step": 7997
    },
    {
      "epoch": 3.0506134033203125e-05,
      "step": 7997,
      "training_step_time": 9.918021440505981
    },
    {
      "epoch": 3.0506134033203125e-05,
      "model_forward_time": 3.1434037685394287,
      "step": 7997
    },
    {
      "dynamics_loss_avg": 0.06704708021134138,
      "epoch": 3.0506134033203125e-05,
      "step": 7997
    },
    {
      "action_loss_avg": 0.022172865737229586,
      "epoch": 3.0506134033203125e-05,
      "step": 7997
    },
    {
      "epoch": 3.0506134033203125e-05,
      "step": 7997,
      "training_step_time": 9.923577785491943
    },
    {
      "epoch": 3.0506134033203125e-05,
      "model_forward_time": 3.411980152130127,
      "step": 7997
    },
    {
      "epoch": 3.0506134033203125e-05,
      "step": 7997,
      "training_step_time": 10.299838781356812
    },
    {
      "epoch": 3.050994873046875e-05,
      "model_forward_time": 3.165106773376465,
      "step": 7998
    },
    {
      "epoch": 3.050994873046875e-05,
      "step": 7998,
      "training_step_time": 9.942520141601562
    },
    {
      "epoch": 3.050994873046875e-05,
      "model_forward_time": 3.145174503326416,
      "step": 7998
    },
    {
      "epoch": 3.050994873046875e-05,
      "step": 7998,
      "training_step_time": 9.930343389511108
    },
    {
      "epoch": 3.050994873046875e-05,
      "model_forward_time": 3.145789623260498,
      "step": 7998
    },
    {
      "epoch": 3.050994873046875e-05,
      "step": 7998,
      "training_step_time": 9.925288438796997
    },
    {
      "epoch": 3.050994873046875e-05,
      "model_forward_time": 3.1494414806365967,
      "step": 7998
    },
    {
      "epoch": 3.050994873046875e-05,
      "step": 7998,
      "training_step_time": 10.031867504119873
    },
    {
      "epoch": 3.0513763427734374e-05,
      "model_forward_time": 3.4740562438964844,
      "step": 7999
    },
    {
      "epoch": 3.0513763427734374e-05,
      "step": 7999,
      "training_step_time": 10.258599519729614
    },
    {
      "epoch": 3.0513763427734374e-05,
      "model_forward_time": 3.146207571029663,
      "step": 7999
    },
    {
      "epoch": 3.0513763427734374e-05,
      "step": 7999,
      "training_step_time": 9.923883199691772
    },
    {
      "epoch": 3.0513763427734374e-05,
      "model_forward_time": 3.139092445373535,
      "step": 7999
    },
    {
      "epoch": 3.0513763427734374e-05,
      "step": 7999,
      "training_step_time": 9.915970087051392
    },
    {
      "epoch": 3.0513763427734374e-05,
      "model_forward_time": 3.1416513919830322,
      "step": 7999
    },
    {
      "epoch": 3.0513763427734374e-05,
      "step": 7999,
      "training_step_time": 10.309554100036621
    },
    {
      "epoch": 3.0517578125e-05,
      "grad_norm": 0.6606543660163879,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 8000
    },
    {
      "epoch": 3.0517578125e-05,
      "model_forward_time": 3.1634650230407715,
      "step": 8000
    },
    {
      "dynamics_loss_avg": 0.04539692094549537,
      "epoch": 3.0517578125e-05,
      "step": 8000
    },
    {
      "action_loss_avg": 0.016109728999435902,
      "epoch": 3.0517578125e-05,
      "step": 8000
    },
    {
      "epoch": 3.0517578125e-05,
      "step": 8000,
      "training_step_time": 9.924184083938599
    },
    {
      "epoch": 3.0517578125e-05,
      "model_forward_time": 5.010578632354736,
      "step": 8000
    },
    {
      "epoch": 3.0517578125e-05,
      "step": 8000,
      "training_step_time": 11.787510871887207
    },
    {
      "epoch": 3.0517578125e-05,
      "model_forward_time": 3.1313486099243164,
      "step": 8000
    },
    {
      "epoch": 3.0517578125e-05,
      "step": 8000,
      "training_step_time": 9.892995595932007
    },
    {
      "epoch": 3.0517578125e-05,
      "model_forward_time": 3.131441831588745,
      "step": 8000
    },
    {
      "epoch": 3.0517578125e-05,
      "step": 8000,
      "training_step_time": 10.04039978981018
    },
    {
      "epoch": 3.052139282226562e-05,
      "model_forward_time": 2.445047616958618,
      "step": 8001
    },
    {
      "epoch": 3.052139282226562e-05,
      "step": 8001,
      "training_step_time": 7.476161479949951
    },
    {
      "epoch": 3.052139282226562e-05,
      "model_forward_time": 3.4275290966033936,
      "step": 8001
    },
    {
      "epoch": 3.052139282226562e-05,
      "step": 8001,
      "training_step_time": 10.201014995574951
    },
    {
      "epoch": 3.052139282226562e-05,
      "model_forward_time": 2.430217981338501,
      "step": 8001
    },
    {
      "epoch": 3.052139282226562e-05,
      "step": 8001,
      "training_step_time": 7.4629127979278564
    },
    {
      "epoch": 3.052139282226562e-05,
      "model_forward_time": 3.135179042816162,
      "step": 8001
    },
    {
      "epoch": 3.052139282226562e-05,
      "step": 8001,
      "training_step_time": 10.278499603271484
    },
    {
      "epoch": 3.052520751953125e-05,
      "model_forward_time": 2.4477505683898926,
      "step": 8002
    },
    {
      "epoch": 3.052520751953125e-05,
      "step": 8002,
      "training_step_time": 7.48354697227478
    },
    {
      "epoch": 3.052520751953125e-05,
      "model_forward_time": 2.430206060409546,
      "step": 8002
    },
    {
      "epoch": 3.052520751953125e-05,
      "step": 8002,
      "training_step_time": 7.461947441101074
    },
    {
      "epoch": 3.052520751953125e-05,
      "model_forward_time": 3.130845785140991,
      "step": 8002
    },
    {
      "dynamics_loss_avg": 0.06281681433320045,
      "epoch": 3.052520751953125e-05,
      "step": 8002
    },
    {
      "action_loss_avg": 0.02265960453078151,
      "epoch": 3.052520751953125e-05,
      "step": 8002
    },
    {
      "epoch": 3.052520751953125e-05,
      "step": 8002,
      "training_step_time": 9.91234278678894
    },
    {
      "epoch": 3.052520751953125e-05,
      "model_forward_time": 3.4193074703216553,
      "step": 8002
    },
    {
      "epoch": 3.052520751953125e-05,
      "step": 8002,
      "training_step_time": 10.315746307373047
    },
    {
      "epoch": 3.0529022216796875e-05,
      "model_forward_time": 2.455843210220337,
      "step": 8003
    },
    {
      "epoch": 3.0529022216796875e-05,
      "step": 8003,
      "training_step_time": 7.494038820266724
    },
    {
      "epoch": 3.0529022216796875e-05,
      "model_forward_time": 3.1329290866851807,
      "step": 8003
    },
    {
      "epoch": 3.0529022216796875e-05,
      "step": 8003,
      "training_step_time": 9.918958902359009
    },
    {
      "epoch": 3.0529022216796875e-05,
      "model_forward_time": 2.4312362670898438,
      "step": 8003
    },
    {
      "epoch": 3.0529022216796875e-05,
      "step": 8003,
      "training_step_time": 7.464112281799316
    },
    {
      "epoch": 3.0529022216796875e-05,
      "model_forward_time": 3.1337788105010986,
      "step": 8003
    },
    {
      "epoch": 3.0529022216796875e-05,
      "step": 8003,
      "training_step_time": 10.307703018188477
    },
    {
      "epoch": 3.05328369140625e-05,
      "model_forward_time": 3.4236459732055664,
      "step": 8004
    },
    {
      "epoch": 3.05328369140625e-05,
      "step": 8004,
      "training_step_time": 10.207245588302612
    },
    {
      "epoch": 3.05328369140625e-05,
      "model_forward_time": 3.146120548248291,
      "step": 8004
    },
    {
      "epoch": 3.05328369140625e-05,
      "step": 8004,
      "training_step_time": 9.930315971374512
    },
    {
      "epoch": 3.05328369140625e-05,
      "model_forward_time": 3.1438472270965576,
      "step": 8004
    },
    {
      "epoch": 3.05328369140625e-05,
      "step": 8004,
      "training_step_time": 9.927528142929077
    },
    {
      "epoch": 3.05328369140625e-05,
      "model_forward_time": 3.147083044052124,
      "step": 8004
    },
    {
      "epoch": 3.05328369140625e-05,
      "step": 8004,
      "training_step_time": 10.028420209884644
    },
    {
      "epoch": 3.053665161132813e-05,
      "model_forward_time": 3.162588119506836,
      "step": 8005
    },
    {
      "dynamics_loss_avg": 0.04838749310001731,
      "epoch": 3.053665161132813e-05,
      "step": 8005
    },
    {
      "action_loss_avg": 0.022748211165890098,
      "epoch": 3.053665161132813e-05,
      "step": 8005
    },
    {
      "epoch": 3.053665161132813e-05,
      "step": 8005,
      "training_step_time": 9.942640542984009
    },
    {
      "epoch": 3.053665161132813e-05,
      "model_forward_time": 3.436326265335083,
      "step": 8005
    },
    {
      "epoch": 3.053665161132813e-05,
      "step": 8005,
      "training_step_time": 10.214223623275757
    },
    {
      "epoch": 3.053665161132813e-05,
      "model_forward_time": 3.1387455463409424,
      "step": 8005
    },
    {
      "epoch": 3.053665161132813e-05,
      "step": 8005,
      "training_step_time": 9.91765308380127
    },
    {
      "epoch": 3.053665161132813e-05,
      "model_forward_time": 3.147925853729248,
      "step": 8005
    },
    {
      "epoch": 3.053665161132813e-05,
      "step": 8005,
      "training_step_time": 10.017526149749756
    },
    {
      "epoch": 3.054046630859375e-05,
      "model_forward_time": 3.1501498222351074,
      "step": 8006
    },
    {
      "epoch": 3.054046630859375e-05,
      "step": 8006,
      "training_step_time": 9.927515745162964
    },
    {
      "epoch": 3.054046630859375e-05,
      "model_forward_time": 3.1385931968688965,
      "step": 8006
    },
    {
      "epoch": 3.054046630859375e-05,
      "step": 8006,
      "training_step_time": 9.913919687271118
    },
    {
      "epoch": 3.054046630859375e-05,
      "model_forward_time": 3.4224023818969727,
      "step": 8006
    },
    {
      "epoch": 3.054046630859375e-05,
      "step": 8006,
      "training_step_time": 10.198144435882568
    },
    {
      "epoch": 3.054046630859375e-05,
      "model_forward_time": 3.1434783935546875,
      "step": 8006
    },
    {
      "epoch": 3.054046630859375e-05,
      "step": 8006,
      "training_step_time": 10.291430711746216
    },
    {
      "epoch": 3.0544281005859374e-05,
      "model_forward_time": 2.447232246398926,
      "step": 8007
    },
    {
      "epoch": 3.0544281005859374e-05,
      "step": 8007,
      "training_step_time": 7.480830669403076
    },
    {
      "epoch": 3.0544281005859374e-05,
      "model_forward_time": 3.1370832920074463,
      "step": 8007
    },
    {
      "epoch": 3.0544281005859374e-05,
      "step": 8007,
      "training_step_time": 9.912947416305542
    },
    {
      "epoch": 3.0544281005859374e-05,
      "model_forward_time": 2.4335153102874756,
      "step": 8007
    },
    {
      "dynamics_loss_avg": 0.05424673482775688,
      "epoch": 3.0544281005859374e-05,
      "step": 8007
    },
    {
      "action_loss_avg": 0.028812430519610643,
      "epoch": 3.0544281005859374e-05,
      "step": 8007
    },
    {
      "epoch": 3.0544281005859374e-05,
      "step": 8007,
      "training_step_time": 7.4705376625061035
    },
    {
      "epoch": 3.0544281005859374e-05,
      "model_forward_time": 3.4343302249908447,
      "step": 8007
    },
    {
      "epoch": 3.0544281005859374e-05,
      "step": 8007,
      "training_step_time": 10.310818195343018
    },
    {
      "epoch": 3.0548095703125e-05,
      "model_forward_time": 3.1561269760131836,
      "step": 8008
    },
    {
      "epoch": 3.0548095703125e-05,
      "step": 8008,
      "training_step_time": 9.936740636825562
    },
    {
      "epoch": 3.0548095703125e-05,
      "model_forward_time": 3.1453983783721924,
      "step": 8008
    },
    {
      "epoch": 3.0548095703125e-05,
      "step": 8008,
      "training_step_time": 9.926681756973267
    },
    {
      "epoch": 3.0548095703125e-05,
      "model_forward_time": 3.146206855773926,
      "step": 8008
    },
    {
      "epoch": 3.0548095703125e-05,
      "step": 8008,
      "training_step_time": 9.931516885757446
    },
    {
      "epoch": 3.0548095703125e-05,
      "model_forward_time": 3.1456260681152344,
      "step": 8008
    },
    {
      "epoch": 3.0548095703125e-05,
      "step": 8008,
      "training_step_time": 10.028184413909912
    },
    {
      "epoch": 3.0551910400390626e-05,
      "model_forward_time": 2.745671272277832,
      "step": 8009
    },
    {
      "epoch": 3.0551910400390626e-05,
      "step": 8009,
      "training_step_time": 7.7872395515441895
    },
    {
      "epoch": 3.0551910400390626e-05,
      "model_forward_time": 3.1392087936401367,
      "step": 8009
    },
    {
      "epoch": 3.0551910400390626e-05,
      "step": 8009,
      "training_step_time": 9.918349742889404
    },
    {
      "epoch": 3.0551910400390626e-05,
      "model_forward_time": 2.4367427825927734,
      "step": 8009
    },
    {
      "epoch": 3.0551910400390626e-05,
      "step": 8009,
      "training_step_time": 7.474647045135498
    },
    {
      "epoch": 3.0551910400390626e-05,
      "model_forward_time": 3.1413378715515137,
      "step": 8009
    },
    {
      "epoch": 3.0551910400390626e-05,
      "step": 8009,
      "training_step_time": 10.016714334487915
    },
    {
      "epoch": 3.055572509765625e-05,
      "grad_norm": 0.5210621356964111,
      "learning_rate": 1e-05,
      "loss": 0.0789,
      "step": 8010
    },
    {
      "epoch": 3.055572509765625e-05,
      "model_forward_time": 2.451047658920288,
      "step": 8010
    },
    {
      "dynamics_loss_avg": 0.05478549189865589,
      "epoch": 3.055572509765625e-05,
      "step": 8010
    },
    {
      "action_loss_avg": 0.022039891593158244,
      "epoch": 3.055572509765625e-05,
      "step": 8010
    },
    {
      "epoch": 3.055572509765625e-05,
      "step": 8010,
      "training_step_time": 7.487201929092407
    },
    {
      "epoch": 3.055572509765625e-05,
      "model_forward_time": 3.4389941692352295,
      "step": 8010
    },
    {
      "epoch": 3.055572509765625e-05,
      "step": 8010,
      "training_step_time": 10.215357065200806
    },
    {
      "epoch": 3.055572509765625e-05,
      "model_forward_time": 3.141768217086792,
      "step": 8010
    },
    {
      "epoch": 3.055572509765625e-05,
      "step": 8010,
      "training_step_time": 9.916417837142944
    },
    {
      "epoch": 3.055572509765625e-05,
      "model_forward_time": 3.136195659637451,
      "step": 8010
    },
    {
      "epoch": 3.055572509765625e-05,
      "step": 8010,
      "training_step_time": 10.268521308898926
    },
    {
      "epoch": 3.055953979492187e-05,
      "model_forward_time": 3.1507654190063477,
      "step": 8011
    },
    {
      "epoch": 3.055953979492187e-05,
      "step": 8011,
      "training_step_time": 9.92748761177063
    },
    {
      "epoch": 3.055953979492187e-05,
      "model_forward_time": 3.1407642364501953,
      "step": 8011
    },
    {
      "epoch": 3.055953979492187e-05,
      "step": 8011,
      "training_step_time": 9.913103342056274
    },
    {
      "epoch": 3.055953979492187e-05,
      "model_forward_time": 2.434731960296631,
      "step": 8011
    },
    {
      "epoch": 3.055953979492187e-05,
      "step": 8011,
      "training_step_time": 7.740202188491821
    },
    {
      "epoch": 3.055953979492187e-05,
      "model_forward_time": 2.431711435317993,
      "step": 8011
    },
    {
      "epoch": 3.055953979492187e-05,
      "step": 8011,
      "training_step_time": 10.272199630737305
    },
    {
      "epoch": 3.05633544921875e-05,
      "model_forward_time": 3.1571786403656006,
      "step": 8012
    },
    {
      "epoch": 3.05633544921875e-05,
      "step": 8012,
      "training_step_time": 9.935993194580078
    },
    {
      "epoch": 3.05633544921875e-05,
      "model_forward_time": 3.1480157375335693,
      "step": 8012
    },
    {
      "epoch": 3.05633544921875e-05,
      "step": 8012,
      "training_step_time": 9.927725553512573
    },
    {
      "epoch": 3.05633544921875e-05,
      "model_forward_time": 2.43990159034729,
      "step": 8012
    },
    {
      "dynamics_loss_avg": 0.04755137450993061,
      "epoch": 3.05633544921875e-05,
      "step": 8012
    },
    {
      "action_loss_avg": 0.015722960885614156,
      "epoch": 3.05633544921875e-05,
      "step": 8012
    },
    {
      "epoch": 3.05633544921875e-05,
      "step": 8012,
      "training_step_time": 7.478292226791382
    },
    {
      "epoch": 3.05633544921875e-05,
      "model_forward_time": 3.143087863922119,
      "step": 8012
    },
    {
      "epoch": 3.05633544921875e-05,
      "step": 8012,
      "training_step_time": 10.300598621368408
    },
    {
      "epoch": 3.0567169189453125e-05,
      "model_forward_time": 3.1528515815734863,
      "step": 8013
    },
    {
      "epoch": 3.0567169189453125e-05,
      "step": 8013,
      "training_step_time": 9.938995838165283
    },
    {
      "epoch": 3.0567169189453125e-05,
      "model_forward_time": 3.147353410720825,
      "step": 8013
    },
    {
      "epoch": 3.0567169189453125e-05,
      "step": 8013,
      "training_step_time": 9.930660486221313
    },
    {
      "epoch": 3.0567169189453125e-05,
      "model_forward_time": 3.1430044174194336,
      "step": 8013
    },
    {
      "epoch": 3.0567169189453125e-05,
      "step": 8013,
      "training_step_time": 9.922127962112427
    },
    {
      "epoch": 3.0567169189453125e-05,
      "model_forward_time": 3.1459124088287354,
      "step": 8013
    },
    {
      "epoch": 3.0567169189453125e-05,
      "step": 8013,
      "training_step_time": 10.029157876968384
    },
    {
      "epoch": 3.057098388671875e-05,
      "model_forward_time": 3.1502575874328613,
      "step": 8014
    },
    {
      "epoch": 3.057098388671875e-05,
      "step": 8014,
      "training_step_time": 10.243311882019043
    },
    {
      "epoch": 3.057098388671875e-05,
      "model_forward_time": 3.14263653755188,
      "step": 8014
    },
    {
      "epoch": 3.057098388671875e-05,
      "step": 8014,
      "training_step_time": 9.920820474624634
    },
    {
      "epoch": 3.057098388671875e-05,
      "model_forward_time": 3.1407225131988525,
      "step": 8014
    },
    {
      "epoch": 3.057098388671875e-05,
      "step": 8014,
      "training_step_time": 9.923318147659302
    },
    {
      "epoch": 3.057098388671875e-05,
      "model_forward_time": 2.440636157989502,
      "step": 8014
    },
    {
      "epoch": 3.057098388671875e-05,
      "step": 8014,
      "training_step_time": 10.021297216415405
    },
    {
      "epoch": 3.057479858398438e-05,
      "model_forward_time": 3.1534674167633057,
      "step": 8015
    },
    {
      "dynamics_loss_avg": 0.05774828176945448,
      "epoch": 3.057479858398438e-05,
      "step": 8015
    },
    {
      "action_loss_avg": 0.02194392867386341,
      "epoch": 3.057479858398438e-05,
      "step": 8015
    },
    {
      "epoch": 3.057479858398438e-05,
      "step": 8015,
      "training_step_time": 9.94016432762146
    },
    {
      "epoch": 3.057479858398438e-05,
      "model_forward_time": 3.1416046619415283,
      "step": 8015
    },
    {
      "epoch": 3.057479858398438e-05,
      "step": 8015,
      "training_step_time": 10.229679584503174
    },
    {
      "epoch": 3.057479858398438e-05,
      "model_forward_time": 3.146258592605591,
      "step": 8015
    },
    {
      "epoch": 3.057479858398438e-05,
      "step": 8015,
      "training_step_time": 9.929584741592407
    },
    {
      "epoch": 3.057479858398438e-05,
      "model_forward_time": 3.148221254348755,
      "step": 8015
    },
    {
      "epoch": 3.057479858398438e-05,
      "step": 8015,
      "training_step_time": 10.278433799743652
    },
    {
      "epoch": 3.057861328125e-05,
      "model_forward_time": 3.161505699157715,
      "step": 8016
    },
    {
      "epoch": 3.057861328125e-05,
      "step": 8016,
      "training_step_time": 9.94680404663086
    },
    {
      "epoch": 3.057861328125e-05,
      "model_forward_time": 3.1478335857391357,
      "step": 8016
    },
    {
      "epoch": 3.057861328125e-05,
      "step": 8016,
      "training_step_time": 9.930711507797241
    },
    {
      "epoch": 3.057861328125e-05,
      "model_forward_time": 2.4380016326904297,
      "step": 8016
    },
    {
      "epoch": 3.057861328125e-05,
      "step": 8016,
      "training_step_time": 7.476608991622925
    },
    {
      "epoch": 3.057861328125e-05,
      "model_forward_time": 3.412806272506714,
      "step": 8016
    },
    {
      "epoch": 3.057861328125e-05,
      "step": 8016,
      "training_step_time": 10.289256811141968
    },
    {
      "epoch": 3.058242797851562e-05,
      "model_forward_time": 2.455436944961548,
      "step": 8017
    },
    {
      "epoch": 3.058242797851562e-05,
      "step": 8017,
      "training_step_time": 7.494875907897949
    },
    {
      "epoch": 3.058242797851562e-05,
      "model_forward_time": 3.1400935649871826,
      "step": 8017
    },
    {
      "epoch": 3.058242797851562e-05,
      "step": 8017,
      "training_step_time": 9.923752307891846
    },
    {
      "epoch": 3.058242797851562e-05,
      "model_forward_time": 3.1445460319519043,
      "step": 8017
    },
    {
      "dynamics_loss_avg": 0.07288857828825712,
      "epoch": 3.058242797851562e-05,
      "step": 8017
    },
    {
      "action_loss_avg": 0.026938728895038365,
      "epoch": 3.058242797851562e-05,
      "step": 8017
    },
    {
      "epoch": 3.058242797851562e-05,
      "step": 8017,
      "training_step_time": 9.930583238601685
    },
    {
      "epoch": 3.058242797851562e-05,
      "model_forward_time": 3.145548105239868,
      "step": 8017
    },
    {
      "epoch": 3.058242797851562e-05,
      "step": 8017,
      "training_step_time": 10.296459197998047
    },
    {
      "epoch": 3.058624267578125e-05,
      "model_forward_time": 2.758183717727661,
      "step": 8018
    },
    {
      "epoch": 3.058624267578125e-05,
      "step": 8018,
      "training_step_time": 7.792506456375122
    },
    {
      "epoch": 3.058624267578125e-05,
      "model_forward_time": 2.43695068359375,
      "step": 8018
    },
    {
      "epoch": 3.058624267578125e-05,
      "step": 8018,
      "training_step_time": 7.469721555709839
    },
    {
      "epoch": 3.058624267578125e-05,
      "model_forward_time": 3.141536235809326,
      "step": 8018
    },
    {
      "epoch": 3.058624267578125e-05,
      "step": 8018,
      "training_step_time": 9.920090436935425
    },
    {
      "epoch": 3.058624267578125e-05,
      "model_forward_time": 2.439990520477295,
      "step": 8018
    },
    {
      "epoch": 3.058624267578125e-05,
      "step": 8018,
      "training_step_time": 10.01743221282959
    },
    {
      "epoch": 3.0590057373046875e-05,
      "model_forward_time": 2.4505510330200195,
      "step": 8019
    },
    {
      "epoch": 3.0590057373046875e-05,
      "step": 8019,
      "training_step_time": 7.483762264251709
    },
    {
      "epoch": 3.0590057373046875e-05,
      "model_forward_time": 2.708188533782959,
      "step": 8019
    },
    {
      "epoch": 3.0590057373046875e-05,
      "step": 8019,
      "training_step_time": 7.733828783035278
    },
    {
      "epoch": 3.0590057373046875e-05,
      "model_forward_time": 3.1386780738830566,
      "step": 8019
    },
    {
      "epoch": 3.0590057373046875e-05,
      "step": 8019,
      "training_step_time": 9.9154794216156
    },
    {
      "epoch": 3.0590057373046875e-05,
      "model_forward_time": 2.432830333709717,
      "step": 8019
    },
    {
      "epoch": 3.0590057373046875e-05,
      "step": 8019,
      "training_step_time": 10.01309871673584
    },
    {
      "epoch": 3.05938720703125e-05,
      "grad_norm": 0.5442033410072327,
      "learning_rate": 1e-05,
      "loss": 0.0825,
      "step": 8020
    },
    {
      "epoch": 3.05938720703125e-05,
      "model_forward_time": 2.445668935775757,
      "step": 8020
    },
    {
      "dynamics_loss_avg": 0.05492973104119301,
      "epoch": 3.05938720703125e-05,
      "step": 8020
    },
    {
      "action_loss_avg": 0.020640996098518372,
      "epoch": 3.05938720703125e-05,
      "step": 8020
    },
    {
      "epoch": 3.05938720703125e-05,
      "step": 8020,
      "training_step_time": 7.475555658340454
    },
    {
      "epoch": 3.05938720703125e-05,
      "model_forward_time": 3.1429054737091064,
      "step": 8020
    },
    {
      "epoch": 3.05938720703125e-05,
      "step": 8020,
      "training_step_time": 9.910603284835815
    },
    {
      "epoch": 3.05938720703125e-05,
      "model_forward_time": 3.4429538249969482,
      "step": 8020
    },
    {
      "epoch": 3.05938720703125e-05,
      "step": 8020,
      "training_step_time": 10.206622838973999
    },
    {
      "epoch": 3.05938720703125e-05,
      "model_forward_time": 2.4305453300476074,
      "step": 8020
    },
    {
      "epoch": 3.05938720703125e-05,
      "step": 8020,
      "training_step_time": 10.30978512763977
    },
    {
      "epoch": 3.059768676757813e-05,
      "model_forward_time": 2.449385404586792,
      "step": 8021
    },
    {
      "epoch": 3.059768676757813e-05,
      "step": 8021,
      "training_step_time": 7.474466323852539
    },
    {
      "epoch": 3.059768676757813e-05,
      "model_forward_time": 2.4317338466644287,
      "step": 8021
    },
    {
      "epoch": 3.059768676757813e-05,
      "step": 8021,
      "training_step_time": 7.455021142959595
    },
    {
      "epoch": 3.059768676757813e-05,
      "model_forward_time": 3.13759183883667,
      "step": 8021
    },
    {
      "epoch": 3.059768676757813e-05,
      "step": 8021,
      "training_step_time": 9.910796642303467
    },
    {
      "epoch": 3.059768676757813e-05,
      "model_forward_time": 2.433401346206665,
      "step": 8021
    },
    {
      "epoch": 3.059768676757813e-05,
      "step": 8021,
      "training_step_time": 10.022101879119873
    },
    {
      "epoch": 3.060150146484375e-05,
      "model_forward_time": 3.4774582386016846,
      "step": 8022
    },
    {
      "epoch": 3.060150146484375e-05,
      "step": 8022,
      "training_step_time": 10.25607943534851
    },
    {
      "epoch": 3.060150146484375e-05,
      "model_forward_time": 2.4343907833099365,
      "step": 8022
    },
    {
      "epoch": 3.060150146484375e-05,
      "step": 8022,
      "training_step_time": 7.469091176986694
    },
    {
      "epoch": 3.060150146484375e-05,
      "model_forward_time": 2.4350814819335938,
      "step": 8022
    },
    {
      "dynamics_loss_avg": 0.05530915968120098,
      "epoch": 3.060150146484375e-05,
      "step": 8022
    },
    {
      "action_loss_avg": 0.020798274921253322,
      "epoch": 3.060150146484375e-05,
      "step": 8022
    },
    {
      "epoch": 3.060150146484375e-05,
      "step": 8022,
      "training_step_time": 7.469668626785278
    },
    {
      "epoch": 3.060150146484375e-05,
      "model_forward_time": 3.148085117340088,
      "step": 8022
    },
    {
      "epoch": 3.060150146484375e-05,
      "step": 8022,
      "training_step_time": 10.343924045562744
    },
    {
      "epoch": 3.0605316162109374e-05,
      "model_forward_time": 3.1597952842712402,
      "step": 8023
    },
    {
      "epoch": 3.0605316162109374e-05,
      "step": 8023,
      "training_step_time": 9.941842794418335
    },
    {
      "epoch": 3.0605316162109374e-05,
      "model_forward_time": 2.713984251022339,
      "step": 8023
    },
    {
      "epoch": 3.0605316162109374e-05,
      "step": 8023,
      "training_step_time": 7.749335765838623
    },
    {
      "epoch": 3.0605316162109374e-05,
      "model_forward_time": 2.43500018119812,
      "step": 8023
    },
    {
      "epoch": 3.0605316162109374e-05,
      "step": 8023,
      "training_step_time": 7.465792894363403
    },
    {
      "epoch": 3.0605316162109374e-05,
      "model_forward_time": 3.1381049156188965,
      "step": 8023
    },
    {
      "epoch": 3.0605316162109374e-05,
      "step": 8023,
      "training_step_time": 10.030981302261353
    },
    {
      "epoch": 3.0609130859375e-05,
      "model_forward_time": 3.159479856491089,
      "step": 8024
    },
    {
      "epoch": 3.0609130859375e-05,
      "step": 8024,
      "training_step_time": 9.939911365509033
    },
    {
      "epoch": 3.0609130859375e-05,
      "model_forward_time": 3.138702154159546,
      "step": 8024
    },
    {
      "epoch": 3.0609130859375e-05,
      "step": 8024,
      "training_step_time": 9.914389848709106
    },
    {
      "epoch": 3.0609130859375e-05,
      "model_forward_time": 3.3890604972839355,
      "step": 8024
    },
    {
      "epoch": 3.0609130859375e-05,
      "step": 8024,
      "training_step_time": 10.166602849960327
    },
    {
      "epoch": 3.0609130859375e-05,
      "model_forward_time": 2.4373347759246826,
      "step": 8024
    },
    {
      "epoch": 3.0609130859375e-05,
      "step": 8024,
      "training_step_time": 10.22756552696228
    },
    {
      "epoch": 3.0612945556640626e-05,
      "model_forward_time": 3.155818462371826,
      "step": 8025
    },
    {
      "dynamics_loss_avg": 0.047636266052722934,
      "epoch": 3.0612945556640626e-05,
      "step": 8025
    },
    {
      "action_loss_avg": 0.016828974103555085,
      "epoch": 3.0612945556640626e-05,
      "step": 8025
    },
    {
      "epoch": 3.0612945556640626e-05,
      "step": 8025,
      "training_step_time": 9.93674087524414
    },
    {
      "epoch": 3.0612945556640626e-05,
      "model_forward_time": 3.142812490463257,
      "step": 8025
    },
    {
      "epoch": 3.0612945556640626e-05,
      "step": 8025,
      "training_step_time": 9.9158616065979
    },
    {
      "epoch": 3.0612945556640626e-05,
      "model_forward_time": 3.1392576694488525,
      "step": 8025
    },
    {
      "epoch": 3.0612945556640626e-05,
      "step": 8025,
      "training_step_time": 9.911474227905273
    },
    {
      "epoch": 3.0612945556640626e-05,
      "model_forward_time": 3.415156126022339,
      "step": 8025
    },
    {
      "epoch": 3.0612945556640626e-05,
      "step": 8025,
      "training_step_time": 10.291982650756836
    },
    {
      "epoch": 3.061676025390625e-05,
      "model_forward_time": 3.150540351867676,
      "step": 8026
    },
    {
      "epoch": 3.061676025390625e-05,
      "step": 8026,
      "training_step_time": 9.93248963356018
    },
    {
      "epoch": 3.061676025390625e-05,
      "model_forward_time": 1.7460260391235352,
      "step": 8026
    },
    {
      "epoch": 3.061676025390625e-05,
      "step": 8026,
      "training_step_time": 5.205547094345093
    },
    {
      "epoch": 3.061676025390625e-05,
      "model_forward_time": 2.4368176460266113,
      "step": 8026
    },
    {
      "epoch": 3.061676025390625e-05,
      "step": 8026,
      "training_step_time": 7.467049837112427
    },
    {
      "epoch": 3.061676025390625e-05,
      "model_forward_time": 3.137815475463867,
      "step": 8026
    },
    {
      "epoch": 3.061676025390625e-05,
      "step": 8026,
      "training_step_time": 10.265676498413086
    },
    {
      "epoch": 3.062057495117187e-05,
      "model_forward_time": 3.430314302444458,
      "step": 8027
    },
    {
      "epoch": 3.062057495117187e-05,
      "step": 8027,
      "training_step_time": 10.21333646774292
    },
    {
      "epoch": 3.062057495117187e-05,
      "model_forward_time": 3.1415414810180664,
      "step": 8027
    },
    {
      "epoch": 3.062057495117187e-05,
      "step": 8027,
      "training_step_time": 9.921143054962158
    },
    {
      "epoch": 3.062057495117187e-05,
      "model_forward_time": 2.439761161804199,
      "step": 8027
    },
    {
      "dynamics_loss_avg": 0.056841404736042024,
      "epoch": 3.062057495117187e-05,
      "step": 8027
    },
    {
      "action_loss_avg": 0.020480999490246177,
      "epoch": 3.062057495117187e-05,
      "step": 8027
    },
    {
      "epoch": 3.062057495117187e-05,
      "step": 8027,
      "training_step_time": 7.479644060134888
    },
    {
      "epoch": 3.062057495117187e-05,
      "model_forward_time": 2.4398233890533447,
      "step": 8027
    },
    {
      "epoch": 3.062057495117187e-05,
      "step": 8027,
      "training_step_time": 10.294779539108276
    },
    {
      "epoch": 3.06243896484375e-05,
      "model_forward_time": 3.1542975902557373,
      "step": 8028
    },
    {
      "epoch": 3.06243896484375e-05,
      "step": 8028,
      "training_step_time": 9.932862997055054
    },
    {
      "epoch": 3.06243896484375e-05,
      "model_forward_time": 2.7278552055358887,
      "step": 8028
    },
    {
      "epoch": 3.06243896484375e-05,
      "step": 8028,
      "training_step_time": 7.760723829269409
    },
    {
      "epoch": 3.06243896484375e-05,
      "model_forward_time": 3.137565851211548,
      "step": 8028
    },
    {
      "epoch": 3.06243896484375e-05,
      "step": 8028,
      "training_step_time": 9.910181999206543
    },
    {
      "epoch": 3.06243896484375e-05,
      "model_forward_time": 3.1374614238739014,
      "step": 8028
    },
    {
      "epoch": 3.06243896484375e-05,
      "step": 8028,
      "training_step_time": 10.263987302780151
    },
    {
      "epoch": 3.0628204345703125e-05,
      "model_forward_time": 3.1509041786193848,
      "step": 8029
    },
    {
      "epoch": 3.0628204345703125e-05,
      "step": 8029,
      "training_step_time": 9.919994354248047
    },
    {
      "epoch": 3.0628204345703125e-05,
      "model_forward_time": 2.4347028732299805,
      "step": 8029
    },
    {
      "epoch": 3.0628204345703125e-05,
      "step": 8029,
      "training_step_time": 7.460732936859131
    },
    {
      "epoch": 3.0628204345703125e-05,
      "model_forward_time": 2.429615020751953,
      "step": 8029
    },
    {
      "epoch": 3.0628204345703125e-05,
      "step": 8029,
      "training_step_time": 7.74392032623291
    },
    {
      "epoch": 3.0628204345703125e-05,
      "model_forward_time": 2.429067611694336,
      "step": 8029
    },
    {
      "epoch": 3.0628204345703125e-05,
      "step": 8029,
      "training_step_time": 10.267468929290771
    },
    {
      "epoch": 3.063201904296875e-05,
      "grad_norm": 0.48658424615859985,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 8030
    },
    {
      "epoch": 3.063201904296875e-05,
      "model_forward_time": 3.1556801795959473,
      "step": 8030
    },
    {
      "dynamics_loss_avg": 0.05900987144559622,
      "epoch": 3.063201904296875e-05,
      "step": 8030
    },
    {
      "action_loss_avg": 0.020689969835802912,
      "epoch": 3.063201904296875e-05,
      "step": 8030
    },
    {
      "epoch": 3.063201904296875e-05,
      "step": 8030,
      "training_step_time": 9.92516303062439
    },
    {
      "epoch": 3.063201904296875e-05,
      "model_forward_time": 3.142256498336792,
      "step": 8030
    },
    {
      "epoch": 3.063201904296875e-05,
      "step": 8030,
      "training_step_time": 9.914133787155151
    },
    {
      "epoch": 3.063201904296875e-05,
      "model_forward_time": 2.4457755088806152,
      "step": 8030
    },
    {
      "epoch": 3.063201904296875e-05,
      "step": 8030,
      "training_step_time": 7.474184036254883
    },
    {
      "epoch": 3.063201904296875e-05,
      "model_forward_time": 3.1444497108459473,
      "step": 8030
    },
    {
      "epoch": 3.063201904296875e-05,
      "step": 8030,
      "training_step_time": 10.325798988342285
    },
    {
      "epoch": 3.063583374023438e-05,
      "model_forward_time": 2.7655253410339355,
      "step": 8031
    },
    {
      "epoch": 3.063583374023438e-05,
      "step": 8031,
      "training_step_time": 7.793744087219238
    },
    {
      "epoch": 3.063583374023438e-05,
      "model_forward_time": 3.1458539962768555,
      "step": 8031
    },
    {
      "epoch": 3.063583374023438e-05,
      "step": 8031,
      "training_step_time": 9.912795305252075
    },
    {
      "epoch": 3.063583374023438e-05,
      "model_forward_time": 2.4404802322387695,
      "step": 8031
    },
    {
      "epoch": 3.063583374023438e-05,
      "step": 8031,
      "training_step_time": 7.4698498249053955
    },
    {
      "epoch": 3.063583374023438e-05,
      "model_forward_time": 2.4427762031555176,
      "step": 8031
    },
    {
      "epoch": 3.063583374023438e-05,
      "step": 8031,
      "training_step_time": 10.2508864402771
    },
    {
      "epoch": 3.06396484375e-05,
      "model_forward_time": 3.1588428020477295,
      "step": 8032
    },
    {
      "epoch": 3.06396484375e-05,
      "step": 8032,
      "training_step_time": 9.932362794876099
    },
    {
      "epoch": 3.06396484375e-05,
      "model_forward_time": 2.71026349067688,
      "step": 8032
    },
    {
      "epoch": 3.06396484375e-05,
      "step": 8032,
      "training_step_time": 7.734001398086548
    },
    {
      "epoch": 3.06396484375e-05,
      "model_forward_time": 3.136354923248291,
      "step": 8032
    },
    {
      "dynamics_loss_avg": 0.061748244985938075,
      "epoch": 3.06396484375e-05,
      "step": 8032
    },
    {
      "action_loss_avg": 0.02759496672078967,
      "epoch": 3.06396484375e-05,
      "step": 8032
    },
    {
      "epoch": 3.06396484375e-05,
      "step": 8032,
      "training_step_time": 9.907477855682373
    },
    {
      "epoch": 3.06396484375e-05,
      "model_forward_time": 3.1357812881469727,
      "step": 8032
    },
    {
      "epoch": 3.06396484375e-05,
      "step": 8032,
      "training_step_time": 9.997965574264526
    },
    {
      "epoch": 3.064346313476562e-05,
      "model_forward_time": 3.1508660316467285,
      "step": 8033
    },
    {
      "epoch": 3.064346313476562e-05,
      "step": 8033,
      "training_step_time": 9.918606758117676
    },
    {
      "epoch": 3.064346313476562e-05,
      "model_forward_time": 2.4339394569396973,
      "step": 8033
    },
    {
      "epoch": 3.064346313476562e-05,
      "step": 8033,
      "training_step_time": 7.453736066818237
    },
    {
      "epoch": 3.064346313476562e-05,
      "model_forward_time": 3.385694742202759,
      "step": 8033
    },
    {
      "epoch": 3.064346313476562e-05,
      "step": 8033,
      "training_step_time": 10.145217895507812
    },
    {
      "epoch": 3.064346313476562e-05,
      "model_forward_time": 3.1392698287963867,
      "step": 8033
    },
    {
      "epoch": 3.064346313476562e-05,
      "step": 8033,
      "training_step_time": 10.265215635299683
    },
    {
      "epoch": 3.064727783203125e-05,
      "model_forward_time": 3.1459240913391113,
      "step": 8034
    },
    {
      "epoch": 3.064727783203125e-05,
      "step": 8034,
      "training_step_time": 9.908981084823608
    },
    {
      "epoch": 3.064727783203125e-05,
      "model_forward_time": 3.1308045387268066,
      "step": 8034
    },
    {
      "epoch": 3.064727783203125e-05,
      "step": 8034,
      "training_step_time": 9.890671014785767
    },
    {
      "epoch": 3.064727783203125e-05,
      "model_forward_time": 3.132845878601074,
      "step": 8034
    },
    {
      "epoch": 3.064727783203125e-05,
      "step": 8034,
      "training_step_time": 9.899143934249878
    },
    {
      "epoch": 3.064727783203125e-05,
      "model_forward_time": 3.411973714828491,
      "step": 8034
    },
    {
      "epoch": 3.064727783203125e-05,
      "step": 8034,
      "training_step_time": 10.276116132736206
    },
    {
      "epoch": 3.0651092529296876e-05,
      "model_forward_time": 2.4472768306732178,
      "step": 8035
    },
    {
      "dynamics_loss_avg": 0.05642961002886295,
      "epoch": 3.0651092529296876e-05,
      "step": 8035
    },
    {
      "action_loss_avg": 0.024924755655229093,
      "epoch": 3.0651092529296876e-05,
      "step": 8035
    },
    {
      "epoch": 3.0651092529296876e-05,
      "step": 8035,
      "training_step_time": 7.478998422622681
    },
    {
      "epoch": 3.0651092529296876e-05,
      "model_forward_time": 2.430213212966919,
      "step": 8035
    },
    {
      "epoch": 3.0651092529296876e-05,
      "step": 8035,
      "training_step_time": 7.452624797821045
    },
    {
      "epoch": 3.0651092529296876e-05,
      "model_forward_time": 3.134495973587036,
      "step": 8035
    },
    {
      "epoch": 3.0651092529296876e-05,
      "step": 8035,
      "training_step_time": 9.901750564575195
    },
    {
      "epoch": 3.0651092529296876e-05,
      "model_forward_time": 2.4357168674468994,
      "step": 8035
    },
    {
      "epoch": 3.0651092529296876e-05,
      "step": 8035,
      "training_step_time": 10.273216247558594
    },
    {
      "epoch": 3.06549072265625e-05,
      "model_forward_time": 3.429203987121582,
      "step": 8036
    },
    {
      "epoch": 3.06549072265625e-05,
      "step": 8036,
      "training_step_time": 10.199206590652466
    },
    {
      "epoch": 3.06549072265625e-05,
      "model_forward_time": 2.4352309703826904,
      "step": 8036
    },
    {
      "epoch": 3.06549072265625e-05,
      "step": 8036,
      "training_step_time": 7.464169979095459
    },
    {
      "epoch": 3.06549072265625e-05,
      "model_forward_time": 3.1371405124664307,
      "step": 8036
    },
    {
      "epoch": 3.06549072265625e-05,
      "step": 8036,
      "training_step_time": 9.90766167640686
    },
    {
      "epoch": 3.06549072265625e-05,
      "model_forward_time": 3.137537717819214,
      "step": 8036
    },
    {
      "epoch": 3.06549072265625e-05,
      "step": 8036,
      "training_step_time": 10.017462968826294
    },
    {
      "epoch": 3.065872192382813e-05,
      "model_forward_time": 3.157681703567505,
      "step": 8037
    },
    {
      "epoch": 3.065872192382813e-05,
      "step": 8037,
      "training_step_time": 9.929805755615234
    },
    {
      "epoch": 3.065872192382813e-05,
      "model_forward_time": 3.3882222175598145,
      "step": 8037
    },
    {
      "epoch": 3.065872192382813e-05,
      "step": 8037,
      "training_step_time": 10.161809921264648
    },
    {
      "epoch": 3.065872192382813e-05,
      "model_forward_time": 3.1479995250701904,
      "step": 8037
    },
    {
      "dynamics_loss_avg": 0.0702846772968769,
      "epoch": 3.065872192382813e-05,
      "step": 8037
    },
    {
      "action_loss_avg": 0.030813123239204286,
      "epoch": 3.065872192382813e-05,
      "step": 8037
    },
    {
      "epoch": 3.065872192382813e-05,
      "step": 8037,
      "training_step_time": 9.92340087890625
    },
    {
      "epoch": 3.065872192382813e-05,
      "model_forward_time": 2.4329326152801514,
      "step": 8037
    },
    {
      "epoch": 3.065872192382813e-05,
      "step": 8037,
      "training_step_time": 10.275817632675171
    },
    {
      "epoch": 3.066253662109375e-05,
      "model_forward_time": 3.1463162899017334,
      "step": 8038
    },
    {
      "epoch": 3.066253662109375e-05,
      "step": 8038,
      "training_step_time": 9.919199705123901
    },
    {
      "epoch": 3.066253662109375e-05,
      "model_forward_time": 2.4347176551818848,
      "step": 8038
    },
    {
      "epoch": 3.066253662109375e-05,
      "step": 8038,
      "training_step_time": 7.462991237640381
    },
    {
      "epoch": 3.066253662109375e-05,
      "model_forward_time": 2.7129831314086914,
      "step": 8038
    },
    {
      "epoch": 3.066253662109375e-05,
      "step": 8038,
      "training_step_time": 7.73958158493042
    },
    {
      "epoch": 3.066253662109375e-05,
      "model_forward_time": 3.137270450592041,
      "step": 8038
    },
    {
      "epoch": 3.066253662109375e-05,
      "step": 8038,
      "training_step_time": 10.286014556884766
    },
    {
      "epoch": 3.0666351318359374e-05,
      "model_forward_time": 2.4560768604278564,
      "step": 8039
    },
    {
      "epoch": 3.0666351318359374e-05,
      "step": 8039,
      "training_step_time": 7.486830234527588
    },
    {
      "epoch": 3.0666351318359374e-05,
      "model_forward_time": 2.432037353515625,
      "step": 8039
    },
    {
      "epoch": 3.0666351318359374e-05,
      "step": 8039,
      "training_step_time": 7.460892200469971
    },
    {
      "epoch": 3.0666351318359374e-05,
      "model_forward_time": 2.434483528137207,
      "step": 8039
    },
    {
      "epoch": 3.0666351318359374e-05,
      "step": 8039,
      "training_step_time": 7.459563255310059
    },
    {
      "epoch": 3.0666351318359374e-05,
      "model_forward_time": 2.7148489952087402,
      "step": 8039
    },
    {
      "epoch": 3.0666351318359374e-05,
      "step": 8039,
      "training_step_time": 10.007871627807617
    },
    {
      "epoch": 3.0670166015625e-05,
      "grad_norm": 0.45583653450012207,
      "learning_rate": 1e-05,
      "loss": 0.081,
      "step": 8040
    },
    {
      "epoch": 3.0670166015625e-05,
      "model_forward_time": 3.156888246536255,
      "step": 8040
    },
    {
      "dynamics_loss_avg": 0.0515455661341548,
      "epoch": 3.0670166015625e-05,
      "step": 8040
    },
    {
      "action_loss_avg": 0.021091273706406354,
      "epoch": 3.0670166015625e-05,
      "step": 8040
    },
    {
      "epoch": 3.0670166015625e-05,
      "step": 8040,
      "training_step_time": 9.926701068878174
    },
    {
      "epoch": 3.0670166015625e-05,
      "model_forward_time": 2.435994863510132,
      "step": 8040
    },
    {
      "epoch": 3.0670166015625e-05,
      "step": 8040,
      "training_step_time": 7.466279029846191
    },
    {
      "epoch": 3.0670166015625e-05,
      "model_forward_time": 2.4296441078186035,
      "step": 8040
    },
    {
      "epoch": 3.0670166015625e-05,
      "step": 8040,
      "training_step_time": 7.455238580703735
    },
    {
      "epoch": 3.0670166015625e-05,
      "model_forward_time": 2.434711217880249,
      "step": 8040
    },
    {
      "epoch": 3.0670166015625e-05,
      "step": 8040,
      "training_step_time": 10.293396711349487
    },
    {
      "epoch": 3.0673980712890626e-05,
      "model_forward_time": 2.4484007358551025,
      "step": 8041
    },
    {
      "epoch": 3.0673980712890626e-05,
      "step": 8041,
      "training_step_time": 7.479350328445435
    },
    {
      "epoch": 3.0673980712890626e-05,
      "model_forward_time": 2.714174270629883,
      "step": 8041
    },
    {
      "epoch": 3.0673980712890626e-05,
      "step": 8041,
      "training_step_time": 7.736027479171753
    },
    {
      "epoch": 3.0673980712890626e-05,
      "model_forward_time": 2.4317972660064697,
      "step": 8041
    },
    {
      "epoch": 3.0673980712890626e-05,
      "step": 8041,
      "training_step_time": 7.453139066696167
    },
    {
      "epoch": 3.0673980712890626e-05,
      "model_forward_time": 2.4298856258392334,
      "step": 8041
    },
    {
      "epoch": 3.0673980712890626e-05,
      "step": 8041,
      "training_step_time": 10.298540830612183
    },
    {
      "epoch": 3.067779541015625e-05,
      "model_forward_time": 2.445305585861206,
      "step": 8042
    },
    {
      "epoch": 3.067779541015625e-05,
      "step": 8042,
      "training_step_time": 7.466552972793579
    },
    {
      "epoch": 3.067779541015625e-05,
      "model_forward_time": 3.130258560180664,
      "step": 8042
    },
    {
      "epoch": 3.067779541015625e-05,
      "step": 8042,
      "training_step_time": 9.894442081451416
    },
    {
      "epoch": 3.067779541015625e-05,
      "model_forward_time": 2.7206599712371826,
      "step": 8042
    },
    {
      "dynamics_loss_avg": 0.06273351721465588,
      "epoch": 3.067779541015625e-05,
      "step": 8042
    },
    {
      "action_loss_avg": 0.019343131640926003,
      "epoch": 3.067779541015625e-05,
      "step": 8042
    },
    {
      "epoch": 3.067779541015625e-05,
      "step": 8042,
      "training_step_time": 7.746837615966797
    },
    {
      "epoch": 3.067779541015625e-05,
      "model_forward_time": 2.432310104370117,
      "step": 8042
    },
    {
      "epoch": 3.067779541015625e-05,
      "step": 8042,
      "training_step_time": 10.260304927825928
    },
    {
      "epoch": 3.068161010742187e-05,
      "model_forward_time": 2.44176983833313,
      "step": 8043
    },
    {
      "epoch": 3.068161010742187e-05,
      "step": 8043,
      "training_step_time": 7.4646923542022705
    },
    {
      "epoch": 3.068161010742187e-05,
      "model_forward_time": 2.429124355316162,
      "step": 8043
    },
    {
      "epoch": 3.068161010742187e-05,
      "step": 8043,
      "training_step_time": 7.450559854507446
    },
    {
      "epoch": 3.068161010742187e-05,
      "model_forward_time": 3.134909152984619,
      "step": 8043
    },
    {
      "epoch": 3.068161010742187e-05,
      "step": 8043,
      "training_step_time": 9.902658224105835
    },
    {
      "epoch": 3.068161010742187e-05,
      "model_forward_time": 3.1366968154907227,
      "step": 8043
    },
    {
      "epoch": 3.068161010742187e-05,
      "step": 8043,
      "training_step_time": 10.301559925079346
    },
    {
      "epoch": 3.06854248046875e-05,
      "model_forward_time": 3.4486591815948486,
      "step": 8044
    },
    {
      "epoch": 3.06854248046875e-05,
      "step": 8044,
      "training_step_time": 10.223601579666138
    },
    {
      "epoch": 3.06854248046875e-05,
      "model_forward_time": 3.134911298751831,
      "step": 8044
    },
    {
      "epoch": 3.06854248046875e-05,
      "step": 8044,
      "training_step_time": 9.910743474960327
    },
    {
      "epoch": 3.06854248046875e-05,
      "model_forward_time": 3.1418662071228027,
      "step": 8044
    },
    {
      "epoch": 3.06854248046875e-05,
      "step": 8044,
      "training_step_time": 9.917575359344482
    },
    {
      "epoch": 3.06854248046875e-05,
      "model_forward_time": 2.4395132064819336,
      "step": 8044
    },
    {
      "epoch": 3.06854248046875e-05,
      "step": 8044,
      "training_step_time": 10.012296915054321
    },
    {
      "epoch": 3.0689239501953125e-05,
      "model_forward_time": 3.1704747676849365,
      "step": 8045
    },
    {
      "dynamics_loss_avg": 0.06368729565292597,
      "epoch": 3.0689239501953125e-05,
      "step": 8045
    },
    {
      "action_loss_avg": 0.02391323079355061,
      "epoch": 3.0689239501953125e-05,
      "step": 8045
    },
    {
      "epoch": 3.0689239501953125e-05,
      "step": 8045,
      "training_step_time": 9.954653024673462
    },
    {
      "epoch": 3.0689239501953125e-05,
      "model_forward_time": 3.420853853225708,
      "step": 8045
    },
    {
      "epoch": 3.0689239501953125e-05,
      "step": 8045,
      "training_step_time": 10.186724424362183
    },
    {
      "epoch": 3.0689239501953125e-05,
      "model_forward_time": 3.141688108444214,
      "step": 8045
    },
    {
      "epoch": 3.0689239501953125e-05,
      "step": 8045,
      "training_step_time": 9.910534381866455
    },
    {
      "epoch": 3.0689239501953125e-05,
      "model_forward_time": 3.137707233428955,
      "step": 8045
    },
    {
      "epoch": 3.0689239501953125e-05,
      "step": 8045,
      "training_step_time": 10.012693405151367
    },
    {
      "epoch": 3.069305419921875e-05,
      "model_forward_time": 3.1543493270874023,
      "step": 8046
    },
    {
      "epoch": 3.069305419921875e-05,
      "step": 8046,
      "training_step_time": 9.937313795089722
    },
    {
      "epoch": 3.069305419921875e-05,
      "model_forward_time": 2.440732717514038,
      "step": 8046
    },
    {
      "epoch": 3.069305419921875e-05,
      "step": 8046,
      "training_step_time": 7.467231273651123
    },
    {
      "epoch": 3.069305419921875e-05,
      "model_forward_time": 3.417724370956421,
      "step": 8046
    },
    {
      "epoch": 3.069305419921875e-05,
      "step": 8046,
      "training_step_time": 10.183067798614502
    },
    {
      "epoch": 3.069305419921875e-05,
      "model_forward_time": 3.136168956756592,
      "step": 8046
    },
    {
      "epoch": 3.069305419921875e-05,
      "step": 8046,
      "training_step_time": 10.309110879898071
    },
    {
      "epoch": 3.069686889648438e-05,
      "model_forward_time": 2.449267864227295,
      "step": 8047
    },
    {
      "epoch": 3.069686889648438e-05,
      "step": 8047,
      "training_step_time": 7.472400665283203
    },
    {
      "epoch": 3.069686889648438e-05,
      "model_forward_time": 2.4288170337677,
      "step": 8047
    },
    {
      "epoch": 3.069686889648438e-05,
      "step": 8047,
      "training_step_time": 7.458724021911621
    },
    {
      "epoch": 3.069686889648438e-05,
      "model_forward_time": 3.1285579204559326,
      "step": 8047
    },
    {
      "dynamics_loss_avg": 0.057370337471365926,
      "epoch": 3.069686889648438e-05,
      "step": 8047
    },
    {
      "action_loss_avg": 0.022438811743631958,
      "epoch": 3.069686889648438e-05,
      "step": 8047
    },
    {
      "epoch": 3.069686889648438e-05,
      "step": 8047,
      "training_step_time": 9.899735450744629
    },
    {
      "epoch": 3.069686889648438e-05,
      "model_forward_time": 2.704432964324951,
      "step": 8047
    },
    {
      "epoch": 3.069686889648438e-05,
      "step": 8047,
      "training_step_time": 10.00908637046814
    },
    {
      "epoch": 3.070068359375e-05,
      "model_forward_time": 2.4651787281036377,
      "step": 8048
    },
    {
      "epoch": 3.070068359375e-05,
      "step": 8048,
      "training_step_time": 7.491335391998291
    },
    {
      "epoch": 3.070068359375e-05,
      "model_forward_time": 2.4314229488372803,
      "step": 8048
    },
    {
      "epoch": 3.070068359375e-05,
      "step": 8048,
      "training_step_time": 7.45235276222229
    },
    {
      "epoch": 3.070068359375e-05,
      "model_forward_time": 3.1336567401885986,
      "step": 8048
    },
    {
      "epoch": 3.070068359375e-05,
      "step": 8048,
      "training_step_time": 9.895957469940186
    },
    {
      "epoch": 3.070068359375e-05,
      "model_forward_time": 2.4413459300994873,
      "step": 8048
    },
    {
      "epoch": 3.070068359375e-05,
      "step": 8048,
      "training_step_time": 10.009284973144531
    },
    {
      "epoch": 3.070449829101562e-05,
      "model_forward_time": 3.411710262298584,
      "step": 8049
    },
    {
      "epoch": 3.070449829101562e-05,
      "step": 8049,
      "training_step_time": 10.177901268005371
    },
    {
      "epoch": 3.070449829101562e-05,
      "model_forward_time": 2.434640884399414,
      "step": 8049
    },
    {
      "epoch": 3.070449829101562e-05,
      "step": 8049,
      "training_step_time": 7.459068775177002
    },
    {
      "epoch": 3.070449829101562e-05,
      "model_forward_time": 2.4329793453216553,
      "step": 8049
    },
    {
      "epoch": 3.070449829101562e-05,
      "step": 8049,
      "training_step_time": 7.456304311752319
    },
    {
      "epoch": 3.070449829101562e-05,
      "model_forward_time": 2.4327571392059326,
      "step": 8049
    },
    {
      "epoch": 3.070449829101562e-05,
      "step": 8049,
      "training_step_time": 10.292746305465698
    },
    {
      "epoch": 3.070831298828125e-05,
      "grad_norm": 0.36429363489151,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 8050
    },
    {
      "epoch": 3.070831298828125e-05,
      "model_forward_time": 3.1498661041259766,
      "step": 8050
    },
    {
      "dynamics_loss_avg": 0.04943641796708107,
      "epoch": 3.070831298828125e-05,
      "step": 8050
    },
    {
      "action_loss_avg": 0.021059086686000227,
      "epoch": 3.070831298828125e-05,
      "step": 8050
    },
    {
      "epoch": 3.070831298828125e-05,
      "step": 8050,
      "training_step_time": 9.92383861541748
    },
    {
      "epoch": 3.070831298828125e-05,
      "model_forward_time": 3.142423152923584,
      "step": 8050
    },
    {
      "epoch": 3.070831298828125e-05,
      "step": 8050,
      "training_step_time": 9.916224479675293
    },
    {
      "epoch": 3.070831298828125e-05,
      "model_forward_time": 3.3965766429901123,
      "step": 8050
    },
    {
      "epoch": 3.070831298828125e-05,
      "step": 8050,
      "training_step_time": 10.16902208328247
    },
    {
      "epoch": 3.070831298828125e-05,
      "model_forward_time": 3.1489970684051514,
      "step": 8050
    },
    {
      "epoch": 3.070831298828125e-05,
      "step": 8050,
      "training_step_time": 10.022719383239746
    },
    {
      "epoch": 3.0712127685546876e-05,
      "model_forward_time": 3.17333722114563,
      "step": 8051
    },
    {
      "epoch": 3.0712127685546876e-05,
      "step": 8051,
      "training_step_time": 9.966355562210083
    },
    {
      "epoch": 3.0712127685546876e-05,
      "model_forward_time": 2.441093921661377,
      "step": 8051
    },
    {
      "epoch": 3.0712127685546876e-05,
      "step": 8051,
      "training_step_time": 7.478024959564209
    },
    {
      "epoch": 3.0712127685546876e-05,
      "model_forward_time": 2.4416677951812744,
      "step": 8051
    },
    {
      "epoch": 3.0712127685546876e-05,
      "step": 8051,
      "training_step_time": 7.471980094909668
    },
    {
      "epoch": 3.0712127685546876e-05,
      "model_forward_time": 3.4175868034362793,
      "step": 8051
    },
    {
      "epoch": 3.0712127685546876e-05,
      "step": 8051,
      "training_step_time": 10.292896509170532
    },
    {
      "epoch": 3.07159423828125e-05,
      "model_forward_time": 3.155758857727051,
      "step": 8052
    },
    {
      "epoch": 3.07159423828125e-05,
      "step": 8052,
      "training_step_time": 9.93038535118103
    },
    {
      "epoch": 3.07159423828125e-05,
      "model_forward_time": 2.4308383464813232,
      "step": 8052
    },
    {
      "epoch": 3.07159423828125e-05,
      "step": 8052,
      "training_step_time": 7.46074104309082
    },
    {
      "epoch": 3.07159423828125e-05,
      "model_forward_time": 3.132861614227295,
      "step": 8052
    },
    {
      "dynamics_loss_avg": 0.05046054236590862,
      "epoch": 3.07159423828125e-05,
      "step": 8052
    },
    {
      "action_loss_avg": 0.015341949742287397,
      "epoch": 3.07159423828125e-05,
      "step": 8052
    },
    {
      "epoch": 3.07159423828125e-05,
      "step": 8052,
      "training_step_time": 9.909559965133667
    },
    {
      "epoch": 3.07159423828125e-05,
      "model_forward_time": 3.131598472595215,
      "step": 8052
    },
    {
      "epoch": 3.07159423828125e-05,
      "step": 8052,
      "training_step_time": 10.012771368026733
    },
    {
      "epoch": 3.071975708007813e-05,
      "model_forward_time": 3.4585092067718506,
      "step": 8053
    },
    {
      "epoch": 3.071975708007813e-05,
      "step": 8053,
      "training_step_time": 10.232711791992188
    },
    {
      "epoch": 3.071975708007813e-05,
      "model_forward_time": 3.132211923599243,
      "step": 8053
    },
    {
      "epoch": 3.071975708007813e-05,
      "step": 8053,
      "training_step_time": 9.900503873825073
    },
    {
      "epoch": 3.071975708007813e-05,
      "model_forward_time": 2.4302115440368652,
      "step": 8053
    },
    {
      "epoch": 3.071975708007813e-05,
      "step": 8053,
      "training_step_time": 7.457481145858765
    },
    {
      "epoch": 3.071975708007813e-05,
      "model_forward_time": 2.4309189319610596,
      "step": 8053
    },
    {
      "epoch": 3.071975708007813e-05,
      "step": 8053,
      "training_step_time": 10.302242279052734
    },
    {
      "epoch": 3.072357177734375e-05,
      "model_forward_time": 3.1415069103240967,
      "step": 8054
    },
    {
      "epoch": 3.072357177734375e-05,
      "step": 8054,
      "training_step_time": 9.910896062850952
    },
    {
      "epoch": 3.072357177734375e-05,
      "model_forward_time": 2.674812078475952,
      "step": 8054
    },
    {
      "epoch": 3.072357177734375e-05,
      "step": 8054,
      "training_step_time": 7.702132225036621
    },
    {
      "epoch": 3.072357177734375e-05,
      "model_forward_time": 3.136810541152954,
      "step": 8054
    },
    {
      "epoch": 3.072357177734375e-05,
      "step": 8054,
      "training_step_time": 9.907360315322876
    },
    {
      "epoch": 3.072357177734375e-05,
      "model_forward_time": 3.133469343185425,
      "step": 8054
    },
    {
      "epoch": 3.072357177734375e-05,
      "step": 8054,
      "training_step_time": 10.30255651473999
    },
    {
      "epoch": 3.0727386474609374e-05,
      "model_forward_time": 2.4468226432800293,
      "step": 8055
    },
    {
      "dynamics_loss_avg": 0.05042199455201626,
      "epoch": 3.0727386474609374e-05,
      "step": 8055
    },
    {
      "action_loss_avg": 0.020593546330928802,
      "epoch": 3.0727386474609374e-05,
      "step": 8055
    },
    {
      "epoch": 3.0727386474609374e-05,
      "step": 8055,
      "training_step_time": 7.481034278869629
    },
    {
      "epoch": 3.0727386474609374e-05,
      "model_forward_time": 2.4277307987213135,
      "step": 8055
    },
    {
      "epoch": 3.0727386474609374e-05,
      "step": 8055,
      "training_step_time": 7.452653884887695
    },
    {
      "epoch": 3.0727386474609374e-05,
      "model_forward_time": 3.4040493965148926,
      "step": 8055
    },
    {
      "epoch": 3.0727386474609374e-05,
      "step": 8055,
      "training_step_time": 10.16602349281311
    },
    {
      "epoch": 3.0727386474609374e-05,
      "model_forward_time": 2.430629014968872,
      "step": 8055
    },
    {
      "epoch": 3.0727386474609374e-05,
      "step": 8055,
      "training_step_time": 10.005138158798218
    },
    {
      "epoch": 3.0731201171875e-05,
      "model_forward_time": 3.1370081901550293,
      "step": 8056
    },
    {
      "epoch": 3.0731201171875e-05,
      "step": 8056,
      "training_step_time": 9.905493974685669
    },
    {
      "epoch": 3.0731201171875e-05,
      "model_forward_time": 3.132434606552124,
      "step": 8056
    },
    {
      "epoch": 3.0731201171875e-05,
      "step": 8056,
      "training_step_time": 9.899516582489014
    },
    {
      "epoch": 3.0731201171875e-05,
      "model_forward_time": 2.430906295776367,
      "step": 8056
    },
    {
      "epoch": 3.0731201171875e-05,
      "step": 8056,
      "training_step_time": 7.4512059688568115
    },
    {
      "epoch": 3.0731201171875e-05,
      "model_forward_time": 3.389246940612793,
      "step": 8056
    },
    {
      "epoch": 3.0731201171875e-05,
      "step": 8056,
      "training_step_time": 10.253624677658081
    },
    {
      "epoch": 3.0735015869140626e-05,
      "model_forward_time": 3.1456117630004883,
      "step": 8057
    },
    {
      "epoch": 3.0735015869140626e-05,
      "step": 8057,
      "training_step_time": 9.915553092956543
    },
    {
      "epoch": 3.0735015869140626e-05,
      "model_forward_time": 3.1350557804107666,
      "step": 8057
    },
    {
      "epoch": 3.0735015869140626e-05,
      "step": 8057,
      "training_step_time": 9.906338691711426
    },
    {
      "epoch": 3.0735015869140626e-05,
      "model_forward_time": 3.130950450897217,
      "step": 8057
    },
    {
      "dynamics_loss_avg": 0.059427811205387114,
      "epoch": 3.0735015869140626e-05,
      "step": 8057
    },
    {
      "action_loss_avg": 0.026042426843196152,
      "epoch": 3.0735015869140626e-05,
      "step": 8057
    },
    {
      "epoch": 3.0735015869140626e-05,
      "step": 8057,
      "training_step_time": 9.898617029190063
    },
    {
      "epoch": 3.0735015869140626e-05,
      "model_forward_time": 3.135260581970215,
      "step": 8057
    },
    {
      "epoch": 3.0735015869140626e-05,
      "step": 8057,
      "training_step_time": 10.235779047012329
    },
    {
      "epoch": 3.073883056640625e-05,
      "model_forward_time": 3.143446445465088,
      "step": 8058
    },
    {
      "epoch": 3.073883056640625e-05,
      "step": 8058,
      "training_step_time": 10.190447807312012
    },
    {
      "epoch": 3.073883056640625e-05,
      "model_forward_time": 3.133000612258911,
      "step": 8058
    },
    {
      "epoch": 3.073883056640625e-05,
      "step": 8058,
      "training_step_time": 9.899467706680298
    },
    {
      "epoch": 3.073883056640625e-05,
      "model_forward_time": 2.429880380630493,
      "step": 8058
    },
    {
      "epoch": 3.073883056640625e-05,
      "step": 8058,
      "training_step_time": 7.452098369598389
    },
    {
      "epoch": 3.073883056640625e-05,
      "model_forward_time": 2.427760601043701,
      "step": 8058
    },
    {
      "epoch": 3.073883056640625e-05,
      "step": 8058,
      "training_step_time": 10.252763986587524
    },
    {
      "epoch": 3.074264526367187e-05,
      "model_forward_time": 3.1516706943511963,
      "step": 8059
    },
    {
      "epoch": 3.074264526367187e-05,
      "step": 8059,
      "training_step_time": 9.913245439529419
    },
    {
      "epoch": 3.074264526367187e-05,
      "model_forward_time": 2.432128429412842,
      "step": 8059
    },
    {
      "epoch": 3.074264526367187e-05,
      "step": 8059,
      "training_step_time": 7.449256181716919
    },
    {
      "epoch": 3.074264526367187e-05,
      "model_forward_time": 3.382711172103882,
      "step": 8059
    },
    {
      "epoch": 3.074264526367187e-05,
      "step": 8059,
      "training_step_time": 10.146007776260376
    },
    {
      "epoch": 3.074264526367187e-05,
      "model_forward_time": 2.433809757232666,
      "step": 8059
    },
    {
      "epoch": 3.074264526367187e-05,
      "step": 8059,
      "training_step_time": 10.282781839370728
    },
    {
      "epoch": 3.07464599609375e-05,
      "grad_norm": 0.34726816415786743,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 8060
    },
    {
      "epoch": 3.07464599609375e-05,
      "model_forward_time": 3.1542859077453613,
      "step": 8060
    },
    {
      "dynamics_loss_avg": 0.05351205132901669,
      "epoch": 3.07464599609375e-05,
      "step": 8060
    },
    {
      "action_loss_avg": 0.025560809392482044,
      "epoch": 3.07464599609375e-05,
      "step": 8060
    },
    {
      "epoch": 3.07464599609375e-05,
      "step": 8060,
      "training_step_time": 9.932950258255005
    },
    {
      "epoch": 3.07464599609375e-05,
      "model_forward_time": 2.4299817085266113,
      "step": 8060
    },
    {
      "epoch": 3.07464599609375e-05,
      "step": 8060,
      "training_step_time": 7.457090616226196
    },
    {
      "epoch": 3.07464599609375e-05,
      "model_forward_time": 3.1323587894439697,
      "step": 8060
    },
    {
      "epoch": 3.07464599609375e-05,
      "step": 8060,
      "training_step_time": 9.900015115737915
    },
    {
      "epoch": 3.07464599609375e-05,
      "model_forward_time": 2.7052788734436035,
      "step": 8060
    },
    {
      "epoch": 3.07464599609375e-05,
      "step": 8060,
      "training_step_time": 10.009562730789185
    },
    {
      "epoch": 3.0750274658203125e-05,
      "model_forward_time": 3.1468358039855957,
      "step": 8061
    },
    {
      "epoch": 3.0750274658203125e-05,
      "step": 8061,
      "training_step_time": 9.916844606399536
    },
    {
      "epoch": 3.0750274658203125e-05,
      "model_forward_time": 2.433093547821045,
      "step": 8061
    },
    {
      "epoch": 3.0750274658203125e-05,
      "step": 8061,
      "training_step_time": 7.459030389785767
    },
    {
      "epoch": 3.0750274658203125e-05,
      "model_forward_time": 3.1342294216156006,
      "step": 8061
    },
    {
      "epoch": 3.0750274658203125e-05,
      "step": 8061,
      "training_step_time": 9.90008282661438
    },
    {
      "epoch": 3.0750274658203125e-05,
      "model_forward_time": 2.433544635772705,
      "step": 8061
    },
    {
      "epoch": 3.0750274658203125e-05,
      "step": 8061,
      "training_step_time": 10.271180629730225
    },
    {
      "epoch": 3.075408935546875e-05,
      "model_forward_time": 3.4191415309906006,
      "step": 8062
    },
    {
      "epoch": 3.075408935546875e-05,
      "step": 8062,
      "training_step_time": 10.189470291137695
    },
    {
      "epoch": 3.075408935546875e-05,
      "model_forward_time": 2.434142827987671,
      "step": 8062
    },
    {
      "epoch": 3.075408935546875e-05,
      "step": 8062,
      "training_step_time": 7.462739944458008
    },
    {
      "epoch": 3.075408935546875e-05,
      "model_forward_time": 3.135166883468628,
      "step": 8062
    },
    {
      "dynamics_loss_avg": 0.05647373860701919,
      "epoch": 3.075408935546875e-05,
      "step": 8062
    },
    {
      "action_loss_avg": 0.02088819316122681,
      "epoch": 3.075408935546875e-05,
      "step": 8062
    },
    {
      "epoch": 3.075408935546875e-05,
      "step": 8062,
      "training_step_time": 9.911973476409912
    },
    {
      "epoch": 3.075408935546875e-05,
      "model_forward_time": 2.4362471103668213,
      "step": 8062
    },
    {
      "epoch": 3.075408935546875e-05,
      "step": 8062,
      "training_step_time": 10.025436162948608
    },
    {
      "epoch": 3.075790405273438e-05,
      "model_forward_time": 3.1394765377044678,
      "step": 8063
    },
    {
      "epoch": 3.075790405273438e-05,
      "step": 8063,
      "training_step_time": 9.904441833496094
    },
    {
      "epoch": 3.075790405273438e-05,
      "model_forward_time": 3.137350082397461,
      "step": 8063
    },
    {
      "epoch": 3.075790405273438e-05,
      "step": 8063,
      "training_step_time": 9.901633501052856
    },
    {
      "epoch": 3.075790405273438e-05,
      "model_forward_time": 3.40230131149292,
      "step": 8063
    },
    {
      "epoch": 3.075790405273438e-05,
      "step": 8063,
      "training_step_time": 10.165338277816772
    },
    {
      "epoch": 3.075790405273438e-05,
      "model_forward_time": 3.132326364517212,
      "step": 8063
    },
    {
      "epoch": 3.075790405273438e-05,
      "step": 8063,
      "training_step_time": 10.305591344833374
    },
    {
      "epoch": 3.076171875e-05,
      "model_forward_time": 3.137807846069336,
      "step": 8064
    },
    {
      "epoch": 3.076171875e-05,
      "step": 8064,
      "training_step_time": 9.904205322265625
    },
    {
      "epoch": 3.076171875e-05,
      "model_forward_time": 2.431518316268921,
      "step": 8064
    },
    {
      "epoch": 3.076171875e-05,
      "step": 8064,
      "training_step_time": 7.455132961273193
    },
    {
      "epoch": 3.076171875e-05,
      "model_forward_time": 3.1334574222564697,
      "step": 8064
    },
    {
      "epoch": 3.076171875e-05,
      "step": 8064,
      "training_step_time": 10.169148206710815
    },
    {
      "epoch": 3.076171875e-05,
      "model_forward_time": 3.143646240234375,
      "step": 8064
    },
    {
      "epoch": 3.076171875e-05,
      "step": 8064,
      "training_step_time": 10.295538902282715
    },
    {
      "epoch": 3.076553344726562e-05,
      "model_forward_time": 3.1459646224975586,
      "step": 8065
    },
    {
      "dynamics_loss_avg": 0.06665646973997355,
      "epoch": 3.076553344726562e-05,
      "step": 8065
    },
    {
      "action_loss_avg": 0.023740858677774666,
      "epoch": 3.076553344726562e-05,
      "step": 8065
    },
    {
      "epoch": 3.076553344726562e-05,
      "step": 8065,
      "training_step_time": 9.919471502304077
    },
    {
      "epoch": 3.076553344726562e-05,
      "model_forward_time": 3.140003204345703,
      "step": 8065
    },
    {
      "epoch": 3.076553344726562e-05,
      "step": 8065,
      "training_step_time": 9.912583351135254
    },
    {
      "epoch": 3.076553344726562e-05,
      "model_forward_time": 2.4378867149353027,
      "step": 8065
    },
    {
      "epoch": 3.076553344726562e-05,
      "step": 8065,
      "training_step_time": 7.466062784194946
    },
    {
      "epoch": 3.076553344726562e-05,
      "model_forward_time": 2.4341375827789307,
      "step": 8065
    },
    {
      "epoch": 3.076553344726562e-05,
      "step": 8065,
      "training_step_time": 10.016454935073853
    },
    {
      "epoch": 3.076934814453125e-05,
      "model_forward_time": 2.7695231437683105,
      "step": 8066
    },
    {
      "epoch": 3.076934814453125e-05,
      "step": 8066,
      "training_step_time": 7.79432225227356
    },
    {
      "epoch": 3.076934814453125e-05,
      "model_forward_time": 3.135408639907837,
      "step": 8066
    },
    {
      "epoch": 3.076934814453125e-05,
      "step": 8066,
      "training_step_time": 9.897357702255249
    },
    {
      "epoch": 3.076934814453125e-05,
      "model_forward_time": 3.1325294971466064,
      "step": 8066
    },
    {
      "epoch": 3.076934814453125e-05,
      "step": 8066,
      "training_step_time": 9.898932456970215
    },
    {
      "epoch": 3.076934814453125e-05,
      "model_forward_time": 3.145536184310913,
      "step": 8066
    },
    {
      "epoch": 3.076934814453125e-05,
      "step": 8066,
      "training_step_time": 10.246780395507812
    },
    {
      "epoch": 3.0773162841796876e-05,
      "model_forward_time": 3.164879322052002,
      "step": 8067
    },
    {
      "epoch": 3.0773162841796876e-05,
      "step": 8067,
      "training_step_time": 9.938344478607178
    },
    {
      "epoch": 3.0773162841796876e-05,
      "model_forward_time": 2.6987626552581787,
      "step": 8067
    },
    {
      "epoch": 3.0773162841796876e-05,
      "step": 8067,
      "training_step_time": 7.7249274253845215
    },
    {
      "epoch": 3.0773162841796876e-05,
      "model_forward_time": 2.4366698265075684,
      "step": 8067
    },
    {
      "dynamics_loss_avg": 0.06810286417603492,
      "epoch": 3.0773162841796876e-05,
      "step": 8067
    },
    {
      "action_loss_avg": 0.02391140628606081,
      "epoch": 3.0773162841796876e-05,
      "step": 8067
    },
    {
      "epoch": 3.0773162841796876e-05,
      "step": 8067,
      "training_step_time": 7.4642298221588135
    },
    {
      "epoch": 3.0773162841796876e-05,
      "model_forward_time": 2.439326524734497,
      "step": 8067
    },
    {
      "epoch": 3.0773162841796876e-05,
      "step": 8067,
      "training_step_time": 10.016830921173096
    },
    {
      "epoch": 3.07769775390625e-05,
      "model_forward_time": 3.162415027618408,
      "step": 8068
    },
    {
      "epoch": 3.07769775390625e-05,
      "step": 8068,
      "training_step_time": 9.932661056518555
    },
    {
      "epoch": 3.07769775390625e-05,
      "model_forward_time": 2.4389700889587402,
      "step": 8068
    },
    {
      "epoch": 3.07769775390625e-05,
      "step": 8068,
      "training_step_time": 7.4689247608184814
    },
    {
      "epoch": 3.07769775390625e-05,
      "model_forward_time": 2.754270315170288,
      "step": 8068
    },
    {
      "epoch": 3.07769775390625e-05,
      "step": 8068,
      "training_step_time": 7.78796911239624
    },
    {
      "epoch": 3.07769775390625e-05,
      "model_forward_time": 2.4416213035583496,
      "step": 8068
    },
    {
      "epoch": 3.07769775390625e-05,
      "step": 8068,
      "training_step_time": 10.297830581665039
    },
    {
      "epoch": 3.078079223632813e-05,
      "model_forward_time": 2.455456256866455,
      "step": 8069
    },
    {
      "epoch": 3.078079223632813e-05,
      "step": 8069,
      "training_step_time": 7.4823102951049805
    },
    {
      "epoch": 3.078079223632813e-05,
      "model_forward_time": 2.4380900859832764,
      "step": 8069
    },
    {
      "epoch": 3.078079223632813e-05,
      "step": 8069,
      "training_step_time": 7.464061975479126
    },
    {
      "epoch": 3.078079223632813e-05,
      "model_forward_time": 2.437204122543335,
      "step": 8069
    },
    {
      "epoch": 3.078079223632813e-05,
      "step": 8069,
      "training_step_time": 7.463332176208496
    },
    {
      "epoch": 3.078079223632813e-05,
      "model_forward_time": 2.4350221157073975,
      "step": 8069
    },
    {
      "epoch": 3.078079223632813e-05,
      "step": 8069,
      "training_step_time": 10.015091180801392
    },
    {
      "epoch": 3.078460693359375e-05,
      "grad_norm": 0.39223313331604004,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 8070
    },
    {
      "epoch": 3.078460693359375e-05,
      "model_forward_time": 3.435966968536377,
      "step": 8070
    },
    {
      "dynamics_loss_avg": 0.06378544140607119,
      "epoch": 3.078460693359375e-05,
      "step": 8070
    },
    {
      "action_loss_avg": 0.021310376469045877,
      "epoch": 3.078460693359375e-05,
      "step": 8070
    },
    {
      "epoch": 3.078460693359375e-05,
      "step": 8070,
      "training_step_time": 10.209305763244629
    },
    {
      "epoch": 3.078460693359375e-05,
      "model_forward_time": 3.1425511837005615,
      "step": 8070
    },
    {
      "epoch": 3.078460693359375e-05,
      "step": 8070,
      "training_step_time": 9.906355381011963
    },
    {
      "epoch": 3.078460693359375e-05,
      "model_forward_time": 3.1451425552368164,
      "step": 8070
    },
    {
      "epoch": 3.078460693359375e-05,
      "step": 8070,
      "training_step_time": 9.910914659500122
    },
    {
      "epoch": 3.078460693359375e-05,
      "model_forward_time": 3.1415722370147705,
      "step": 8070
    },
    {
      "epoch": 3.078460693359375e-05,
      "step": 8070,
      "training_step_time": 10.253223657608032
    },
    {
      "epoch": 3.0788421630859374e-05,
      "model_forward_time": 2.4517297744750977,
      "step": 8071
    },
    {
      "epoch": 3.0788421630859374e-05,
      "step": 8071,
      "training_step_time": 7.476255655288696
    },
    {
      "epoch": 3.0788421630859374e-05,
      "model_forward_time": 3.3980777263641357,
      "step": 8071
    },
    {
      "epoch": 3.0788421630859374e-05,
      "step": 8071,
      "training_step_time": 10.16278624534607
    },
    {
      "epoch": 3.0788421630859374e-05,
      "model_forward_time": 2.442140579223633,
      "step": 8071
    },
    {
      "epoch": 3.0788421630859374e-05,
      "step": 8071,
      "training_step_time": 7.46549916267395
    },
    {
      "epoch": 3.0788421630859374e-05,
      "model_forward_time": 2.433927059173584,
      "step": 8071
    },
    {
      "epoch": 3.0788421630859374e-05,
      "step": 8071,
      "training_step_time": 10.28743314743042
    },
    {
      "epoch": 3.0792236328125e-05,
      "model_forward_time": 3.148130178451538,
      "step": 8072
    },
    {
      "epoch": 3.0792236328125e-05,
      "step": 8072,
      "training_step_time": 9.916970014572144
    },
    {
      "epoch": 3.0792236328125e-05,
      "model_forward_time": 3.145970344543457,
      "step": 8072
    },
    {
      "epoch": 3.0792236328125e-05,
      "step": 8072,
      "training_step_time": 9.914031982421875
    },
    {
      "epoch": 3.0792236328125e-05,
      "model_forward_time": 3.425994873046875,
      "step": 8072
    },
    {
      "dynamics_loss_avg": 0.07414800673723221,
      "epoch": 3.0792236328125e-05,
      "step": 8072
    },
    {
      "action_loss_avg": 0.030066553223878145,
      "epoch": 3.0792236328125e-05,
      "step": 8072
    },
    {
      "epoch": 3.0792236328125e-05,
      "step": 8072,
      "training_step_time": 10.197455406188965
    },
    {
      "epoch": 3.0792236328125e-05,
      "model_forward_time": 3.1398468017578125,
      "step": 8072
    },
    {
      "epoch": 3.0792236328125e-05,
      "step": 8072,
      "training_step_time": 10.275642156600952
    },
    {
      "epoch": 3.0796051025390627e-05,
      "model_forward_time": 3.1630077362060547,
      "step": 8073
    },
    {
      "epoch": 3.0796051025390627e-05,
      "step": 8073,
      "training_step_time": 9.934627532958984
    },
    {
      "epoch": 3.0796051025390627e-05,
      "model_forward_time": 3.147412061691284,
      "step": 8073
    },
    {
      "epoch": 3.0796051025390627e-05,
      "step": 8073,
      "training_step_time": 9.916055917739868
    },
    {
      "epoch": 3.0796051025390627e-05,
      "model_forward_time": 3.145540714263916,
      "step": 8073
    },
    {
      "epoch": 3.0796051025390627e-05,
      "step": 8073,
      "training_step_time": 9.913760423660278
    },
    {
      "epoch": 3.0796051025390627e-05,
      "model_forward_time": 2.719193458557129,
      "step": 8073
    },
    {
      "epoch": 3.0796051025390627e-05,
      "step": 8073,
      "training_step_time": 10.265058279037476
    },
    {
      "epoch": 3.079986572265625e-05,
      "model_forward_time": 3.1460955142974854,
      "step": 8074
    },
    {
      "epoch": 3.079986572265625e-05,
      "step": 8074,
      "training_step_time": 9.918739080429077
    },
    {
      "epoch": 3.079986572265625e-05,
      "model_forward_time": 3.1489129066467285,
      "step": 8074
    },
    {
      "epoch": 3.079986572265625e-05,
      "step": 8074,
      "training_step_time": 9.918107986450195
    },
    {
      "epoch": 3.079986572265625e-05,
      "model_forward_time": 3.148848056793213,
      "step": 8074
    },
    {
      "epoch": 3.079986572265625e-05,
      "step": 8074,
      "training_step_time": 9.920172452926636
    },
    {
      "epoch": 3.079986572265625e-05,
      "model_forward_time": 2.4435250759124756,
      "step": 8074
    },
    {
      "epoch": 3.079986572265625e-05,
      "step": 8074,
      "training_step_time": 10.290456533432007
    },
    {
      "epoch": 3.080368041992187e-05,
      "model_forward_time": 3.453173875808716,
      "step": 8075
    },
    {
      "dynamics_loss_avg": 0.06102541647851467,
      "epoch": 3.080368041992187e-05,
      "step": 8075
    },
    {
      "action_loss_avg": 0.022358656115829944,
      "epoch": 3.080368041992187e-05,
      "step": 8075
    },
    {
      "epoch": 3.080368041992187e-05,
      "step": 8075,
      "training_step_time": 10.232014894485474
    },
    {
      "epoch": 3.080368041992187e-05,
      "model_forward_time": 2.4428796768188477,
      "step": 8075
    },
    {
      "epoch": 3.080368041992187e-05,
      "step": 8075,
      "training_step_time": 7.471506595611572
    },
    {
      "epoch": 3.080368041992187e-05,
      "model_forward_time": 3.1386730670928955,
      "step": 8075
    },
    {
      "epoch": 3.080368041992187e-05,
      "step": 8075,
      "training_step_time": 9.908550024032593
    },
    {
      "epoch": 3.080368041992187e-05,
      "model_forward_time": 2.4395318031311035,
      "step": 8075
    },
    {
      "epoch": 3.080368041992187e-05,
      "step": 8075,
      "training_step_time": 10.272152662277222
    },
    {
      "epoch": 3.08074951171875e-05,
      "model_forward_time": 2.4516642093658447,
      "step": 8076
    },
    {
      "epoch": 3.08074951171875e-05,
      "step": 8076,
      "training_step_time": 7.475169658660889
    },
    {
      "epoch": 3.08074951171875e-05,
      "model_forward_time": 3.4282727241516113,
      "step": 8076
    },
    {
      "epoch": 3.08074951171875e-05,
      "step": 8076,
      "training_step_time": 10.194436073303223
    },
    {
      "epoch": 3.08074951171875e-05,
      "model_forward_time": 2.441293478012085,
      "step": 8076
    },
    {
      "epoch": 3.08074951171875e-05,
      "step": 8076,
      "training_step_time": 7.466309547424316
    },
    {
      "epoch": 3.08074951171875e-05,
      "model_forward_time": 3.1453146934509277,
      "step": 8076
    },
    {
      "epoch": 3.08074951171875e-05,
      "step": 8076,
      "training_step_time": 10.284634828567505
    },
    {
      "epoch": 3.0811309814453125e-05,
      "model_forward_time": 3.161447525024414,
      "step": 8077
    },
    {
      "epoch": 3.0811309814453125e-05,
      "step": 8077,
      "training_step_time": 9.937221050262451
    },
    {
      "epoch": 3.0811309814453125e-05,
      "model_forward_time": 3.143015146255493,
      "step": 8077
    },
    {
      "epoch": 3.0811309814453125e-05,
      "step": 8077,
      "training_step_time": 9.914239883422852
    },
    {
      "epoch": 3.0811309814453125e-05,
      "model_forward_time": 2.4441089630126953,
      "step": 8077
    },
    {
      "dynamics_loss_avg": 0.05929868761450052,
      "epoch": 3.0811309814453125e-05,
      "step": 8077
    },
    {
      "action_loss_avg": 0.022823983617126942,
      "epoch": 3.0811309814453125e-05,
      "step": 8077
    },
    {
      "epoch": 3.0811309814453125e-05,
      "step": 8077,
      "training_step_time": 7.472366094589233
    },
    {
      "epoch": 3.0811309814453125e-05,
      "model_forward_time": 2.716437578201294,
      "step": 8077
    },
    {
      "epoch": 3.0811309814453125e-05,
      "step": 8077,
      "training_step_time": 10.312379360198975
    },
    {
      "epoch": 3.081512451171875e-05,
      "model_forward_time": 2.4506947994232178,
      "step": 8078
    },
    {
      "epoch": 3.081512451171875e-05,
      "step": 8078,
      "training_step_time": 7.477378606796265
    },
    {
      "epoch": 3.081512451171875e-05,
      "model_forward_time": 3.140009880065918,
      "step": 8078
    },
    {
      "epoch": 3.081512451171875e-05,
      "step": 8078,
      "training_step_time": 9.907163858413696
    },
    {
      "epoch": 3.081512451171875e-05,
      "model_forward_time": 3.1423139572143555,
      "step": 8078
    },
    {
      "epoch": 3.081512451171875e-05,
      "step": 8078,
      "training_step_time": 9.910757780075073
    },
    {
      "epoch": 3.081512451171875e-05,
      "model_forward_time": 3.143744945526123,
      "step": 8078
    },
    {
      "epoch": 3.081512451171875e-05,
      "step": 8078,
      "training_step_time": 10.004901885986328
    },
    {
      "epoch": 3.081893920898438e-05,
      "model_forward_time": 3.447652578353882,
      "step": 8079
    },
    {
      "epoch": 3.081893920898438e-05,
      "step": 8079,
      "training_step_time": 10.216007471084595
    },
    {
      "epoch": 3.081893920898438e-05,
      "model_forward_time": 2.4404044151306152,
      "step": 8079
    },
    {
      "epoch": 3.081893920898438e-05,
      "step": 8079,
      "training_step_time": 7.466052055358887
    },
    {
      "epoch": 3.081893920898438e-05,
      "model_forward_time": 2.4427244663238525,
      "step": 8079
    },
    {
      "epoch": 3.081893920898438e-05,
      "step": 8079,
      "training_step_time": 7.470342397689819
    },
    {
      "epoch": 3.081893920898438e-05,
      "model_forward_time": 2.4384889602661133,
      "step": 8079
    },
    {
      "epoch": 3.081893920898438e-05,
      "step": 8079,
      "training_step_time": 10.274753093719482
    },
    {
      "epoch": 3.082275390625e-05,
      "grad_norm": 0.4390036165714264,
      "learning_rate": 1e-05,
      "loss": 0.0798,
      "step": 8080
    },
    {
      "epoch": 3.082275390625e-05,
      "model_forward_time": 2.4407641887664795,
      "step": 8080
    },
    {
      "dynamics_loss_avg": 0.05113236168399453,
      "epoch": 3.082275390625e-05,
      "step": 8080
    },
    {
      "action_loss_avg": 0.01758885639719665,
      "epoch": 3.082275390625e-05,
      "step": 8080
    },
    {
      "epoch": 3.082275390625e-05,
      "step": 8080,
      "training_step_time": 7.46593165397644
    },
    {
      "epoch": 3.082275390625e-05,
      "model_forward_time": 2.432716131210327,
      "step": 8080
    },
    {
      "epoch": 3.082275390625e-05,
      "step": 8080,
      "training_step_time": 7.730827331542969
    },
    {
      "epoch": 3.082275390625e-05,
      "model_forward_time": 3.133824586868286,
      "step": 8080
    },
    {
      "epoch": 3.082275390625e-05,
      "step": 8080,
      "training_step_time": 9.903186082839966
    },
    {
      "epoch": 3.082275390625e-05,
      "model_forward_time": 3.1357767581939697,
      "step": 8080
    },
    {
      "epoch": 3.082275390625e-05,
      "step": 8080,
      "training_step_time": 10.248983383178711
    },
    {
      "epoch": 3.082656860351562e-05,
      "model_forward_time": 3.1517722606658936,
      "step": 8081
    },
    {
      "epoch": 3.082656860351562e-05,
      "step": 8081,
      "training_step_time": 9.92205286026001
    },
    {
      "epoch": 3.082656860351562e-05,
      "model_forward_time": 3.139658212661743,
      "step": 8081
    },
    {
      "epoch": 3.082656860351562e-05,
      "step": 8081,
      "training_step_time": 9.911941289901733
    },
    {
      "epoch": 3.082656860351562e-05,
      "model_forward_time": 3.419764518737793,
      "step": 8081
    },
    {
      "epoch": 3.082656860351562e-05,
      "step": 8081,
      "training_step_time": 10.189680814743042
    },
    {
      "epoch": 3.082656860351562e-05,
      "model_forward_time": 2.438148260116577,
      "step": 8081
    },
    {
      "epoch": 3.082656860351562e-05,
      "step": 8081,
      "training_step_time": 10.297805547714233
    },
    {
      "epoch": 3.083038330078125e-05,
      "model_forward_time": 3.1596083641052246,
      "step": 8082
    },
    {
      "epoch": 3.083038330078125e-05,
      "step": 8082,
      "training_step_time": 9.92764163017273
    },
    {
      "epoch": 3.083038330078125e-05,
      "model_forward_time": 3.137507200241089,
      "step": 8082
    },
    {
      "epoch": 3.083038330078125e-05,
      "step": 8082,
      "training_step_time": 9.897409200668335
    },
    {
      "epoch": 3.083038330078125e-05,
      "model_forward_time": 2.4296185970306396,
      "step": 8082
    },
    {
      "dynamics_loss_avg": 0.05594316236674786,
      "epoch": 3.083038330078125e-05,
      "step": 8082
    },
    {
      "action_loss_avg": 0.019540521875023843,
      "epoch": 3.083038330078125e-05,
      "step": 8082
    },
    {
      "epoch": 3.083038330078125e-05,
      "step": 8082,
      "training_step_time": 7.454827070236206
    },
    {
      "epoch": 3.083038330078125e-05,
      "model_forward_time": 1.7405147552490234,
      "step": 8082
    },
    {
      "epoch": 3.083038330078125e-05,
      "step": 8082,
      "training_step_time": 10.30902886390686
    },
    {
      "epoch": 3.0834197998046876e-05,
      "model_forward_time": 2.697035789489746,
      "step": 8083
    },
    {
      "epoch": 3.0834197998046876e-05,
      "step": 8083,
      "training_step_time": 7.715006589889526
    },
    {
      "epoch": 3.0834197998046876e-05,
      "model_forward_time": 2.430659294128418,
      "step": 8083
    },
    {
      "epoch": 3.0834197998046876e-05,
      "step": 8083,
      "training_step_time": 7.446011066436768
    },
    {
      "epoch": 3.0834197998046876e-05,
      "model_forward_time": 3.1303133964538574,
      "step": 8083
    },
    {
      "epoch": 3.0834197998046876e-05,
      "step": 8083,
      "training_step_time": 9.894016981124878
    },
    {
      "epoch": 3.0834197998046876e-05,
      "model_forward_time": 3.1351513862609863,
      "step": 8083
    },
    {
      "epoch": 3.0834197998046876e-05,
      "step": 8083,
      "training_step_time": 10.276007413864136
    },
    {
      "epoch": 3.08380126953125e-05,
      "model_forward_time": 2.4441113471984863,
      "step": 8084
    },
    {
      "epoch": 3.08380126953125e-05,
      "step": 8084,
      "training_step_time": 7.467421531677246
    },
    {
      "epoch": 3.08380126953125e-05,
      "model_forward_time": 3.4097604751586914,
      "step": 8084
    },
    {
      "epoch": 3.08380126953125e-05,
      "step": 8084,
      "training_step_time": 10.171937942504883
    },
    {
      "epoch": 3.08380126953125e-05,
      "model_forward_time": 3.133216142654419,
      "step": 8084
    },
    {
      "epoch": 3.08380126953125e-05,
      "step": 8084,
      "training_step_time": 9.898680210113525
    },
    {
      "epoch": 3.08380126953125e-05,
      "model_forward_time": 2.4341506958007812,
      "step": 8084
    },
    {
      "epoch": 3.08380126953125e-05,
      "step": 8084,
      "training_step_time": 10.27769136428833
    },
    {
      "epoch": 3.084182739257813e-05,
      "model_forward_time": 3.1530306339263916,
      "step": 8085
    },
    {
      "dynamics_loss_avg": 0.035952338855713604,
      "epoch": 3.084182739257813e-05,
      "step": 8085
    },
    {
      "action_loss_avg": 0.01242677764967084,
      "epoch": 3.084182739257813e-05,
      "step": 8085
    },
    {
      "epoch": 3.084182739257813e-05,
      "step": 8085,
      "training_step_time": 9.92132043838501
    },
    {
      "epoch": 3.084182739257813e-05,
      "model_forward_time": 2.4353818893432617,
      "step": 8085
    },
    {
      "epoch": 3.084182739257813e-05,
      "step": 8085,
      "training_step_time": 7.457701683044434
    },
    {
      "epoch": 3.084182739257813e-05,
      "model_forward_time": 3.133448362350464,
      "step": 8085
    },
    {
      "epoch": 3.084182739257813e-05,
      "step": 8085,
      "training_step_time": 10.161768674850464
    },
    {
      "epoch": 3.084182739257813e-05,
      "model_forward_time": 3.13730788230896,
      "step": 8085
    },
    {
      "epoch": 3.084182739257813e-05,
      "step": 8085,
      "training_step_time": 10.283109903335571
    },
    {
      "epoch": 3.084564208984375e-05,
      "model_forward_time": 2.4501450061798096,
      "step": 8086
    },
    {
      "epoch": 3.084564208984375e-05,
      "step": 8086,
      "training_step_time": 7.4731245040893555
    },
    {
      "epoch": 3.084564208984375e-05,
      "model_forward_time": 3.133553981781006,
      "step": 8086
    },
    {
      "epoch": 3.084564208984375e-05,
      "step": 8086,
      "training_step_time": 9.90047550201416
    },
    {
      "epoch": 3.084564208984375e-05,
      "model_forward_time": 2.4340248107910156,
      "step": 8086
    },
    {
      "epoch": 3.084564208984375e-05,
      "step": 8086,
      "training_step_time": 7.461486577987671
    },
    {
      "epoch": 3.084564208984375e-05,
      "model_forward_time": 2.4284698963165283,
      "step": 8086
    },
    {
      "epoch": 3.084564208984375e-05,
      "step": 8086,
      "training_step_time": 10.281080722808838
    },
    {
      "epoch": 3.0849456787109374e-05,
      "model_forward_time": 3.4186744689941406,
      "step": 8087
    },
    {
      "epoch": 3.0849456787109374e-05,
      "step": 8087,
      "training_step_time": 10.18995451927185
    },
    {
      "epoch": 3.0849456787109374e-05,
      "model_forward_time": 2.43593168258667,
      "step": 8087
    },
    {
      "epoch": 3.0849456787109374e-05,
      "step": 8087,
      "training_step_time": 7.4638166427612305
    },
    {
      "epoch": 3.0849456787109374e-05,
      "model_forward_time": 3.1429569721221924,
      "step": 8087
    },
    {
      "dynamics_loss_avg": 0.060898447223007676,
      "epoch": 3.0849456787109374e-05,
      "step": 8087
    },
    {
      "action_loss_avg": 0.02157313362695277,
      "epoch": 3.0849456787109374e-05,
      "step": 8087
    },
    {
      "epoch": 3.0849456787109374e-05,
      "step": 8087,
      "training_step_time": 9.921573638916016
    },
    {
      "epoch": 3.0849456787109374e-05,
      "model_forward_time": 3.1444597244262695,
      "step": 8087
    },
    {
      "epoch": 3.0849456787109374e-05,
      "step": 8087,
      "training_step_time": 10.028432846069336
    },
    {
      "epoch": 3.0853271484375e-05,
      "model_forward_time": 3.157499313354492,
      "step": 8088
    },
    {
      "epoch": 3.0853271484375e-05,
      "step": 8088,
      "training_step_time": 9.934061288833618
    },
    {
      "epoch": 3.0853271484375e-05,
      "model_forward_time": 2.7413973808288574,
      "step": 8088
    },
    {
      "epoch": 3.0853271484375e-05,
      "step": 8088,
      "training_step_time": 7.769649267196655
    },
    {
      "epoch": 3.0853271484375e-05,
      "model_forward_time": 3.14278244972229,
      "step": 8088
    },
    {
      "epoch": 3.0853271484375e-05,
      "step": 8088,
      "training_step_time": 9.91023588180542
    },
    {
      "epoch": 3.0853271484375e-05,
      "model_forward_time": 2.4391651153564453,
      "step": 8088
    },
    {
      "epoch": 3.0853271484375e-05,
      "step": 8088,
      "training_step_time": 10.279796361923218
    },
    {
      "epoch": 3.0857086181640627e-05,
      "model_forward_time": 3.1470096111297607,
      "step": 8089
    },
    {
      "epoch": 3.0857086181640627e-05,
      "step": 8089,
      "training_step_time": 9.91510796546936
    },
    {
      "epoch": 3.0857086181640627e-05,
      "model_forward_time": 3.143132448196411,
      "step": 8089
    },
    {
      "epoch": 3.0857086181640627e-05,
      "step": 8089,
      "training_step_time": 9.911309480667114
    },
    {
      "epoch": 3.0857086181640627e-05,
      "model_forward_time": 2.7159430980682373,
      "step": 8089
    },
    {
      "epoch": 3.0857086181640627e-05,
      "step": 8089,
      "training_step_time": 7.742618083953857
    },
    {
      "epoch": 3.0857086181640627e-05,
      "model_forward_time": 3.1392855644226074,
      "step": 8089
    },
    {
      "epoch": 3.0857086181640627e-05,
      "step": 8089,
      "training_step_time": 10.016887187957764
    },
    {
      "epoch": 3.086090087890625e-05,
      "grad_norm": 0.4440235495567322,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 8090
    },
    {
      "epoch": 3.086090087890625e-05,
      "model_forward_time": 3.150420665740967,
      "step": 8090
    },
    {
      "dynamics_loss_avg": 0.06514189084991813,
      "epoch": 3.086090087890625e-05,
      "step": 8090
    },
    {
      "action_loss_avg": 0.02693300738465041,
      "epoch": 3.086090087890625e-05,
      "step": 8090
    },
    {
      "epoch": 3.086090087890625e-05,
      "step": 8090,
      "training_step_time": 9.920772314071655
    },
    {
      "epoch": 3.086090087890625e-05,
      "model_forward_time": 3.142951250076294,
      "step": 8090
    },
    {
      "epoch": 3.086090087890625e-05,
      "step": 8090,
      "training_step_time": 9.910980939865112
    },
    {
      "epoch": 3.086090087890625e-05,
      "model_forward_time": 2.4390814304351807,
      "step": 8090
    },
    {
      "epoch": 3.086090087890625e-05,
      "step": 8090,
      "training_step_time": 7.466064691543579
    },
    {
      "epoch": 3.086090087890625e-05,
      "model_forward_time": 2.72160005569458,
      "step": 8090
    },
    {
      "epoch": 3.086090087890625e-05,
      "step": 8090,
      "training_step_time": 10.267348766326904
    },
    {
      "epoch": 3.086471557617187e-05,
      "model_forward_time": 3.157961130142212,
      "step": 8091
    },
    {
      "epoch": 3.086471557617187e-05,
      "step": 8091,
      "training_step_time": 9.92967677116394
    },
    {
      "epoch": 3.086471557617187e-05,
      "model_forward_time": 3.1438493728637695,
      "step": 8091
    },
    {
      "epoch": 3.086471557617187e-05,
      "step": 8091,
      "training_step_time": 9.913969993591309
    },
    {
      "epoch": 3.086471557617187e-05,
      "model_forward_time": 2.436816930770874,
      "step": 8091
    },
    {
      "epoch": 3.086471557617187e-05,
      "step": 8091,
      "training_step_time": 7.462177753448486
    },
    {
      "epoch": 3.086471557617187e-05,
      "model_forward_time": 2.438432455062866,
      "step": 8091
    },
    {
      "epoch": 3.086471557617187e-05,
      "step": 8091,
      "training_step_time": 10.238304853439331
    },
    {
      "epoch": 3.08685302734375e-05,
      "model_forward_time": 2.4504244327545166,
      "step": 8092
    },
    {
      "epoch": 3.08685302734375e-05,
      "step": 8092,
      "training_step_time": 7.747021675109863
    },
    {
      "epoch": 3.08685302734375e-05,
      "model_forward_time": 2.4355525970458984,
      "step": 8092
    },
    {
      "epoch": 3.08685302734375e-05,
      "step": 8092,
      "training_step_time": 7.459470272064209
    },
    {
      "epoch": 3.08685302734375e-05,
      "model_forward_time": 3.1389806270599365,
      "step": 8092
    },
    {
      "dynamics_loss_avg": 0.05365351364016533,
      "epoch": 3.08685302734375e-05,
      "step": 8092
    },
    {
      "action_loss_avg": 0.01992441890761256,
      "epoch": 3.08685302734375e-05,
      "step": 8092
    },
    {
      "epoch": 3.08685302734375e-05,
      "step": 8092,
      "training_step_time": 9.917507648468018
    },
    {
      "epoch": 3.08685302734375e-05,
      "model_forward_time": 2.4365439414978027,
      "step": 8092
    },
    {
      "epoch": 3.08685302734375e-05,
      "step": 8092,
      "training_step_time": 10.277928829193115
    },
    {
      "epoch": 3.0872344970703125e-05,
      "model_forward_time": 2.443336009979248,
      "step": 8093
    },
    {
      "epoch": 3.0872344970703125e-05,
      "step": 8093,
      "training_step_time": 7.467291593551636
    },
    {
      "epoch": 3.0872344970703125e-05,
      "model_forward_time": 3.1379706859588623,
      "step": 8093
    },
    {
      "epoch": 3.0872344970703125e-05,
      "step": 8093,
      "training_step_time": 9.90489935874939
    },
    {
      "epoch": 3.0872344970703125e-05,
      "model_forward_time": 3.387096881866455,
      "step": 8093
    },
    {
      "epoch": 3.0872344970703125e-05,
      "step": 8093,
      "training_step_time": 10.150696754455566
    },
    {
      "epoch": 3.0872344970703125e-05,
      "model_forward_time": 3.1407485008239746,
      "step": 8093
    },
    {
      "epoch": 3.0872344970703125e-05,
      "step": 8093,
      "training_step_time": 10.259872198104858
    },
    {
      "epoch": 3.087615966796875e-05,
      "model_forward_time": 2.4476494789123535,
      "step": 8094
    },
    {
      "epoch": 3.087615966796875e-05,
      "step": 8094,
      "training_step_time": 7.472078084945679
    },
    {
      "epoch": 3.087615966796875e-05,
      "model_forward_time": 2.4312822818756104,
      "step": 8094
    },
    {
      "epoch": 3.087615966796875e-05,
      "step": 8094,
      "training_step_time": 7.45631742477417
    },
    {
      "epoch": 3.087615966796875e-05,
      "model_forward_time": 3.140359878540039,
      "step": 8094
    },
    {
      "epoch": 3.087615966796875e-05,
      "step": 8094,
      "training_step_time": 9.9089195728302
    },
    {
      "epoch": 3.087615966796875e-05,
      "model_forward_time": 3.4202616214752197,
      "step": 8094
    },
    {
      "epoch": 3.087615966796875e-05,
      "step": 8094,
      "training_step_time": 10.284106969833374
    },
    {
      "epoch": 3.087997436523438e-05,
      "model_forward_time": 2.4549102783203125,
      "step": 8095
    },
    {
      "dynamics_loss_avg": 0.04100094866007566,
      "epoch": 3.087997436523438e-05,
      "step": 8095
    },
    {
      "action_loss_avg": 0.01989760415162891,
      "epoch": 3.087997436523438e-05,
      "step": 8095
    },
    {
      "epoch": 3.087997436523438e-05,
      "step": 8095,
      "training_step_time": 7.486926555633545
    },
    {
      "epoch": 3.087997436523438e-05,
      "model_forward_time": 3.1434457302093506,
      "step": 8095
    },
    {
      "epoch": 3.087997436523438e-05,
      "step": 8095,
      "training_step_time": 9.917983293533325
    },
    {
      "epoch": 3.087997436523438e-05,
      "model_forward_time": 3.1492342948913574,
      "step": 8095
    },
    {
      "epoch": 3.087997436523438e-05,
      "step": 8095,
      "training_step_time": 9.92412543296814
    },
    {
      "epoch": 3.087997436523438e-05,
      "model_forward_time": 2.443129539489746,
      "step": 8095
    },
    {
      "epoch": 3.087997436523438e-05,
      "step": 8095,
      "training_step_time": 10.022549152374268
    },
    {
      "epoch": 3.08837890625e-05,
      "model_forward_time": 2.709890842437744,
      "step": 8096
    },
    {
      "epoch": 3.08837890625e-05,
      "step": 8096,
      "training_step_time": 7.737692594528198
    },
    {
      "epoch": 3.08837890625e-05,
      "model_forward_time": 2.437523365020752,
      "step": 8096
    },
    {
      "epoch": 3.08837890625e-05,
      "step": 8096,
      "training_step_time": 7.464986324310303
    },
    {
      "epoch": 3.08837890625e-05,
      "model_forward_time": 3.1430134773254395,
      "step": 8096
    },
    {
      "epoch": 3.08837890625e-05,
      "step": 8096,
      "training_step_time": 9.911372900009155
    },
    {
      "epoch": 3.08837890625e-05,
      "model_forward_time": 2.439390182495117,
      "step": 8096
    },
    {
      "epoch": 3.08837890625e-05,
      "step": 8096,
      "training_step_time": 10.029590129852295
    },
    {
      "epoch": 3.088760375976562e-05,
      "model_forward_time": 3.1607820987701416,
      "step": 8097
    },
    {
      "epoch": 3.088760375976562e-05,
      "step": 8097,
      "training_step_time": 9.934612035751343
    },
    {
      "epoch": 3.088760375976562e-05,
      "model_forward_time": 2.4376308917999268,
      "step": 8097
    },
    {
      "epoch": 3.088760375976562e-05,
      "step": 8097,
      "training_step_time": 7.727365493774414
    },
    {
      "epoch": 3.088760375976562e-05,
      "model_forward_time": 2.4335672855377197,
      "step": 8097
    },
    {
      "dynamics_loss_avg": 0.053703613579273224,
      "epoch": 3.088760375976562e-05,
      "step": 8097
    },
    {
      "action_loss_avg": 0.021357599180191755,
      "epoch": 3.088760375976562e-05,
      "step": 8097
    },
    {
      "epoch": 3.088760375976562e-05,
      "step": 8097,
      "training_step_time": 7.454348802566528
    },
    {
      "epoch": 3.088760375976562e-05,
      "model_forward_time": 3.133249521255493,
      "step": 8097
    },
    {
      "epoch": 3.088760375976562e-05,
      "step": 8097,
      "training_step_time": 10.002856492996216
    },
    {
      "epoch": 3.089141845703125e-05,
      "model_forward_time": 2.453098773956299,
      "step": 8098
    },
    {
      "epoch": 3.089141845703125e-05,
      "step": 8098,
      "training_step_time": 7.473681211471558
    },
    {
      "epoch": 3.089141845703125e-05,
      "model_forward_time": 2.4331741333007812,
      "step": 8098
    },
    {
      "epoch": 3.089141845703125e-05,
      "step": 8098,
      "training_step_time": 7.455617904663086
    },
    {
      "epoch": 3.089141845703125e-05,
      "model_forward_time": 2.43369197845459,
      "step": 8098
    },
    {
      "epoch": 3.089141845703125e-05,
      "step": 8098,
      "training_step_time": 7.456418752670288
    },
    {
      "epoch": 3.089141845703125e-05,
      "model_forward_time": 2.018580436706543,
      "step": 8098
    },
    {
      "epoch": 3.089141845703125e-05,
      "step": 8098,
      "training_step_time": 10.296927213668823
    },
    {
      "epoch": 3.0895233154296876e-05,
      "model_forward_time": 3.14298677444458,
      "step": 8099
    },
    {
      "epoch": 3.0895233154296876e-05,
      "step": 8099,
      "training_step_time": 9.92014765739441
    },
    {
      "epoch": 3.0895233154296876e-05,
      "model_forward_time": 3.1349174976348877,
      "step": 8099
    },
    {
      "epoch": 3.0895233154296876e-05,
      "step": 8099,
      "training_step_time": 9.897828578948975
    },
    {
      "epoch": 3.0895233154296876e-05,
      "model_forward_time": 3.1323249340057373,
      "step": 8099
    },
    {
      "epoch": 3.0895233154296876e-05,
      "step": 8099,
      "training_step_time": 9.8981032371521
    },
    {
      "epoch": 3.0895233154296876e-05,
      "model_forward_time": 2.428731679916382,
      "step": 8099
    },
    {
      "epoch": 3.0895233154296876e-05,
      "step": 8099,
      "training_step_time": 10.013160943984985
    },
    {
      "epoch": 3.08990478515625e-05,
      "grad_norm": 0.5260604023933411,
      "learning_rate": 1e-05,
      "loss": 0.0797,
      "step": 8100
    },
    {
      "epoch": 3.08990478515625e-05,
      "model_forward_time": 2.7746565341949463,
      "step": 8100
    },
    {
      "dynamics_loss_avg": 0.04144721403717995,
      "epoch": 3.08990478515625e-05,
      "step": 8100
    },
    {
      "action_loss_avg": 0.01673783954465762,
      "epoch": 3.08990478515625e-05,
      "step": 8100
    },
    {
      "epoch": 3.08990478515625e-05,
      "step": 8100,
      "training_step_time": 7.795947074890137
    },
    {
      "epoch": 3.08990478515625e-05,
      "model_forward_time": 3.1288092136383057,
      "step": 8100
    },
    {
      "epoch": 3.08990478515625e-05,
      "step": 8100,
      "training_step_time": 9.892507553100586
    },
    {
      "epoch": 3.08990478515625e-05,
      "model_forward_time": 2.4302892684936523,
      "step": 8100
    },
    {
      "epoch": 3.08990478515625e-05,
      "step": 8100,
      "training_step_time": 7.453559637069702
    },
    {
      "epoch": 3.08990478515625e-05,
      "model_forward_time": 1.7414193153381348,
      "step": 8100
    },
    {
      "epoch": 3.08990478515625e-05,
      "step": 8100,
      "training_step_time": 10.26321530342102
    },
    {
      "epoch": 3.090286254882813e-05,
      "model_forward_time": 2.4577689170837402,
      "step": 8101
    },
    {
      "epoch": 3.090286254882813e-05,
      "step": 8101,
      "training_step_time": 7.481608629226685
    },
    {
      "epoch": 3.090286254882813e-05,
      "model_forward_time": 3.137622594833374,
      "step": 8101
    },
    {
      "epoch": 3.090286254882813e-05,
      "step": 8101,
      "training_step_time": 9.90143346786499
    },
    {
      "epoch": 3.090286254882813e-05,
      "model_forward_time": 2.688292980194092,
      "step": 8101
    },
    {
      "epoch": 3.090286254882813e-05,
      "step": 8101,
      "training_step_time": 7.708275079727173
    },
    {
      "epoch": 3.090286254882813e-05,
      "model_forward_time": 1.7459378242492676,
      "step": 8101
    },
    {
      "epoch": 3.090286254882813e-05,
      "step": 8101,
      "training_step_time": 10.233604907989502
    },
    {
      "epoch": 3.090667724609375e-05,
      "model_forward_time": 3.1485307216644287,
      "step": 8102
    },
    {
      "epoch": 3.090667724609375e-05,
      "step": 8102,
      "training_step_time": 9.909468412399292
    },
    {
      "epoch": 3.090667724609375e-05,
      "model_forward_time": 3.131763458251953,
      "step": 8102
    },
    {
      "epoch": 3.090667724609375e-05,
      "step": 8102,
      "training_step_time": 9.889853715896606
    },
    {
      "epoch": 3.090667724609375e-05,
      "model_forward_time": 2.435486078262329,
      "step": 8102
    },
    {
      "dynamics_loss_avg": 0.05612381547689438,
      "epoch": 3.090667724609375e-05,
      "step": 8102
    },
    {
      "action_loss_avg": 0.01957050459459424,
      "epoch": 3.090667724609375e-05,
      "step": 8102
    },
    {
      "epoch": 3.090667724609375e-05,
      "step": 8102,
      "training_step_time": 7.457722187042236
    },
    {
      "epoch": 3.090667724609375e-05,
      "model_forward_time": 2.712524175643921,
      "step": 8102
    },
    {
      "epoch": 3.090667724609375e-05,
      "step": 8102,
      "training_step_time": 10.0046865940094
    },
    {
      "epoch": 3.0910491943359374e-05,
      "model_forward_time": 3.1538164615631104,
      "step": 8103
    },
    {
      "epoch": 3.0910491943359374e-05,
      "step": 8103,
      "training_step_time": 9.91493272781372
    },
    {
      "epoch": 3.0910491943359374e-05,
      "model_forward_time": 2.4372048377990723,
      "step": 8103
    },
    {
      "epoch": 3.0910491943359374e-05,
      "step": 8103,
      "training_step_time": 7.452763557434082
    },
    {
      "epoch": 3.0910491943359374e-05,
      "model_forward_time": 3.137361526489258,
      "step": 8103
    },
    {
      "epoch": 3.0910491943359374e-05,
      "step": 8103,
      "training_step_time": 9.895473003387451
    },
    {
      "epoch": 3.0910491943359374e-05,
      "model_forward_time": 2.4363913536071777,
      "step": 8103
    },
    {
      "epoch": 3.0910491943359374e-05,
      "step": 8103,
      "training_step_time": 10.297634363174438
    },
    {
      "epoch": 3.0914306640625e-05,
      "model_forward_time": 3.449195146560669,
      "step": 8104
    },
    {
      "epoch": 3.0914306640625e-05,
      "step": 8104,
      "training_step_time": 10.227306604385376
    },
    {
      "epoch": 3.0914306640625e-05,
      "model_forward_time": 2.4358701705932617,
      "step": 8104
    },
    {
      "epoch": 3.0914306640625e-05,
      "step": 8104,
      "training_step_time": 7.461177110671997
    },
    {
      "epoch": 3.0914306640625e-05,
      "model_forward_time": 3.1437668800354004,
      "step": 8104
    },
    {
      "epoch": 3.0914306640625e-05,
      "step": 8104,
      "training_step_time": 9.907492399215698
    },
    {
      "epoch": 3.0914306640625e-05,
      "model_forward_time": 3.142646074295044,
      "step": 8104
    },
    {
      "epoch": 3.0914306640625e-05,
      "step": 8104,
      "training_step_time": 10.012932538986206
    },
    {
      "epoch": 3.091812133789063e-05,
      "model_forward_time": 3.1518306732177734,
      "step": 8105
    },
    {
      "dynamics_loss_avg": 0.07071052128449082,
      "epoch": 3.091812133789063e-05,
      "step": 8105
    },
    {
      "action_loss_avg": 0.025370058231055738,
      "epoch": 3.091812133789063e-05,
      "step": 8105
    },
    {
      "epoch": 3.091812133789063e-05,
      "step": 8105,
      "training_step_time": 9.920023679733276
    },
    {
      "epoch": 3.091812133789063e-05,
      "model_forward_time": 3.145780086517334,
      "step": 8105
    },
    {
      "epoch": 3.091812133789063e-05,
      "step": 8105,
      "training_step_time": 10.182878017425537
    },
    {
      "epoch": 3.091812133789063e-05,
      "model_forward_time": 3.1506831645965576,
      "step": 8105
    },
    {
      "epoch": 3.091812133789063e-05,
      "step": 8105,
      "training_step_time": 9.9199857711792
    },
    {
      "epoch": 3.091812133789063e-05,
      "model_forward_time": 2.4435510635375977,
      "step": 8105
    },
    {
      "epoch": 3.091812133789063e-05,
      "step": 8105,
      "training_step_time": 10.276624917984009
    },
    {
      "epoch": 3.092193603515625e-05,
      "model_forward_time": 2.4532878398895264,
      "step": 8106
    },
    {
      "epoch": 3.092193603515625e-05,
      "step": 8106,
      "training_step_time": 7.478258371353149
    },
    {
      "epoch": 3.092193603515625e-05,
      "model_forward_time": 2.4344475269317627,
      "step": 8106
    },
    {
      "epoch": 3.092193603515625e-05,
      "step": 8106,
      "training_step_time": 7.458152532577515
    },
    {
      "epoch": 3.092193603515625e-05,
      "model_forward_time": 3.1390271186828613,
      "step": 8106
    },
    {
      "epoch": 3.092193603515625e-05,
      "step": 8106,
      "training_step_time": 9.905867338180542
    },
    {
      "epoch": 3.092193603515625e-05,
      "model_forward_time": 2.7194926738739014,
      "step": 8106
    },
    {
      "epoch": 3.092193603515625e-05,
      "step": 8106,
      "training_step_time": 10.271533489227295
    },
    {
      "epoch": 3.092575073242187e-05,
      "model_forward_time": 3.1632096767425537,
      "step": 8107
    },
    {
      "epoch": 3.092575073242187e-05,
      "step": 8107,
      "training_step_time": 9.927411079406738
    },
    {
      "epoch": 3.092575073242187e-05,
      "model_forward_time": 3.1412203311920166,
      "step": 8107
    },
    {
      "epoch": 3.092575073242187e-05,
      "step": 8107,
      "training_step_time": 9.900792121887207
    },
    {
      "epoch": 3.092575073242187e-05,
      "model_forward_time": 3.142383098602295,
      "step": 8107
    },
    {
      "dynamics_loss_avg": 0.05122783905826509,
      "epoch": 3.092575073242187e-05,
      "step": 8107
    },
    {
      "action_loss_avg": 0.01978609279030934,
      "epoch": 3.092575073242187e-05,
      "step": 8107
    },
    {
      "epoch": 3.092575073242187e-05,
      "step": 8107,
      "training_step_time": 9.906453847885132
    },
    {
      "epoch": 3.092575073242187e-05,
      "model_forward_time": 2.4404196739196777,
      "step": 8107
    },
    {
      "epoch": 3.092575073242187e-05,
      "step": 8107,
      "training_step_time": 9.999816417694092
    },
    {
      "epoch": 3.09295654296875e-05,
      "model_forward_time": 3.438656806945801,
      "step": 8108
    },
    {
      "epoch": 3.09295654296875e-05,
      "step": 8108,
      "training_step_time": 10.206680536270142
    },
    {
      "epoch": 3.09295654296875e-05,
      "model_forward_time": 3.146130323410034,
      "step": 8108
    },
    {
      "epoch": 3.09295654296875e-05,
      "step": 8108,
      "training_step_time": 9.912157535552979
    },
    {
      "epoch": 3.09295654296875e-05,
      "model_forward_time": 3.142587423324585,
      "step": 8108
    },
    {
      "epoch": 3.09295654296875e-05,
      "step": 8108,
      "training_step_time": 9.909757137298584
    },
    {
      "epoch": 3.09295654296875e-05,
      "model_forward_time": 3.144700765609741,
      "step": 8108
    },
    {
      "epoch": 3.09295654296875e-05,
      "step": 8108,
      "training_step_time": 10.286884069442749
    },
    {
      "epoch": 3.0933380126953125e-05,
      "model_forward_time": 2.4585025310516357,
      "step": 8109
    },
    {
      "epoch": 3.0933380126953125e-05,
      "step": 8109,
      "training_step_time": 7.484724760055542
    },
    {
      "epoch": 3.0933380126953125e-05,
      "model_forward_time": 3.423323154449463,
      "step": 8109
    },
    {
      "epoch": 3.0933380126953125e-05,
      "step": 8109,
      "training_step_time": 10.195143699645996
    },
    {
      "epoch": 3.0933380126953125e-05,
      "model_forward_time": 1.7511744499206543,
      "step": 8109
    },
    {
      "epoch": 3.0933380126953125e-05,
      "step": 8109,
      "training_step_time": 5.210882902145386
    },
    {
      "epoch": 3.0933380126953125e-05,
      "model_forward_time": 2.4373912811279297,
      "step": 8109
    },
    {
      "epoch": 3.0933380126953125e-05,
      "step": 8109,
      "training_step_time": 10.017738580703735
    },
    {
      "epoch": 3.093719482421875e-05,
      "grad_norm": 0.2732536196708679,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 8110
    },
    {
      "epoch": 3.093719482421875e-05,
      "model_forward_time": 3.1675150394439697,
      "step": 8110
    },
    {
      "dynamics_loss_avg": 0.049147780798375604,
      "epoch": 3.093719482421875e-05,
      "step": 8110
    },
    {
      "action_loss_avg": 0.017810512473806737,
      "epoch": 3.093719482421875e-05,
      "step": 8110
    },
    {
      "epoch": 3.093719482421875e-05,
      "step": 8110,
      "training_step_time": 9.937921524047852
    },
    {
      "epoch": 3.093719482421875e-05,
      "model_forward_time": 3.1436679363250732,
      "step": 8110
    },
    {
      "epoch": 3.093719482421875e-05,
      "step": 8110,
      "training_step_time": 9.912275791168213
    },
    {
      "epoch": 3.093719482421875e-05,
      "model_forward_time": 2.444429636001587,
      "step": 8110
    },
    {
      "epoch": 3.093719482421875e-05,
      "step": 8110,
      "training_step_time": 7.762990236282349
    },
    {
      "epoch": 3.093719482421875e-05,
      "model_forward_time": 2.4395523071289062,
      "step": 8110
    },
    {
      "epoch": 3.093719482421875e-05,
      "step": 8110,
      "training_step_time": 10.017853260040283
    },
    {
      "epoch": 3.094100952148438e-05,
      "model_forward_time": 3.1640830039978027,
      "step": 8111
    },
    {
      "epoch": 3.094100952148438e-05,
      "step": 8111,
      "training_step_time": 9.947707653045654
    },
    {
      "epoch": 3.094100952148438e-05,
      "model_forward_time": 2.4403319358825684,
      "step": 8111
    },
    {
      "epoch": 3.094100952148438e-05,
      "step": 8111,
      "training_step_time": 7.464315176010132
    },
    {
      "epoch": 3.094100952148438e-05,
      "model_forward_time": 3.1403861045837402,
      "step": 8111
    },
    {
      "epoch": 3.094100952148438e-05,
      "step": 8111,
      "training_step_time": 9.9089035987854
    },
    {
      "epoch": 3.094100952148438e-05,
      "model_forward_time": 2.435883045196533,
      "step": 8111
    },
    {
      "epoch": 3.094100952148438e-05,
      "step": 8111,
      "training_step_time": 10.285165071487427
    },
    {
      "epoch": 3.094482421875e-05,
      "model_forward_time": 3.4433629512786865,
      "step": 8112
    },
    {
      "epoch": 3.094482421875e-05,
      "step": 8112,
      "training_step_time": 10.207130670547485
    },
    {
      "epoch": 3.094482421875e-05,
      "model_forward_time": 3.130535125732422,
      "step": 8112
    },
    {
      "epoch": 3.094482421875e-05,
      "step": 8112,
      "training_step_time": 9.893790006637573
    },
    {
      "epoch": 3.094482421875e-05,
      "model_forward_time": 3.1344354152679443,
      "step": 8112
    },
    {
      "dynamics_loss_avg": 0.05723336730152369,
      "epoch": 3.094482421875e-05,
      "step": 8112
    },
    {
      "action_loss_avg": 0.02074209274724126,
      "epoch": 3.094482421875e-05,
      "step": 8112
    },
    {
      "epoch": 3.094482421875e-05,
      "step": 8112,
      "training_step_time": 9.901359558105469
    },
    {
      "epoch": 3.094482421875e-05,
      "model_forward_time": 3.131162405014038,
      "step": 8112
    },
    {
      "epoch": 3.094482421875e-05,
      "step": 8112,
      "training_step_time": 10.282073974609375
    },
    {
      "epoch": 3.094863891601562e-05,
      "model_forward_time": 2.4452216625213623,
      "step": 8113
    },
    {
      "epoch": 3.094863891601562e-05,
      "step": 8113,
      "training_step_time": 7.482789039611816
    },
    {
      "epoch": 3.094863891601562e-05,
      "model_forward_time": 2.6844654083251953,
      "step": 8113
    },
    {
      "epoch": 3.094863891601562e-05,
      "step": 8113,
      "training_step_time": 7.713506460189819
    },
    {
      "epoch": 3.094863891601562e-05,
      "model_forward_time": 3.136643409729004,
      "step": 8113
    },
    {
      "epoch": 3.094863891601562e-05,
      "step": 8113,
      "training_step_time": 9.906556844711304
    },
    {
      "epoch": 3.094863891601562e-05,
      "model_forward_time": 3.1321094036102295,
      "step": 8113
    },
    {
      "epoch": 3.094863891601562e-05,
      "step": 8113,
      "training_step_time": 10.264085292816162
    },
    {
      "epoch": 3.095245361328125e-05,
      "model_forward_time": 3.158801794052124,
      "step": 8114
    },
    {
      "epoch": 3.095245361328125e-05,
      "step": 8114,
      "training_step_time": 9.935474157333374
    },
    {
      "epoch": 3.095245361328125e-05,
      "model_forward_time": 2.4361989498138428,
      "step": 8114
    },
    {
      "epoch": 3.095245361328125e-05,
      "step": 8114,
      "training_step_time": 7.4687628746032715
    },
    {
      "epoch": 3.095245361328125e-05,
      "model_forward_time": 2.7128453254699707,
      "step": 8114
    },
    {
      "epoch": 3.095245361328125e-05,
      "step": 8114,
      "training_step_time": 7.745240688323975
    },
    {
      "epoch": 3.095245361328125e-05,
      "model_forward_time": 2.4364871978759766,
      "step": 8114
    },
    {
      "epoch": 3.095245361328125e-05,
      "step": 8114,
      "training_step_time": 10.304742097854614
    },
    {
      "epoch": 3.0956268310546876e-05,
      "model_forward_time": 3.146591901779175,
      "step": 8115
    },
    {
      "dynamics_loss_avg": 0.055554609000682834,
      "epoch": 3.0956268310546876e-05,
      "step": 8115
    },
    {
      "action_loss_avg": 0.02147110477089882,
      "epoch": 3.0956268310546876e-05,
      "step": 8115
    },
    {
      "epoch": 3.0956268310546876e-05,
      "step": 8115,
      "training_step_time": 9.920925855636597
    },
    {
      "epoch": 3.0956268310546876e-05,
      "model_forward_time": 2.4363009929656982,
      "step": 8115
    },
    {
      "epoch": 3.0956268310546876e-05,
      "step": 8115,
      "training_step_time": 7.473228693008423
    },
    {
      "epoch": 3.0956268310546876e-05,
      "model_forward_time": 3.1381850242614746,
      "step": 8115
    },
    {
      "epoch": 3.0956268310546876e-05,
      "step": 8115,
      "training_step_time": 9.921257257461548
    },
    {
      "epoch": 3.0956268310546876e-05,
      "model_forward_time": 3.398833990097046,
      "step": 8115
    },
    {
      "epoch": 3.0956268310546876e-05,
      "step": 8115,
      "training_step_time": 10.276610851287842
    },
    {
      "epoch": 3.09600830078125e-05,
      "model_forward_time": 2.4485106468200684,
      "step": 8116
    },
    {
      "epoch": 3.09600830078125e-05,
      "step": 8116,
      "training_step_time": 7.489369630813599
    },
    {
      "epoch": 3.09600830078125e-05,
      "model_forward_time": 2.4381613731384277,
      "step": 8116
    },
    {
      "epoch": 3.09600830078125e-05,
      "step": 8116,
      "training_step_time": 7.4758758544921875
    },
    {
      "epoch": 3.09600830078125e-05,
      "model_forward_time": 3.1383321285247803,
      "step": 8116
    },
    {
      "epoch": 3.09600830078125e-05,
      "step": 8116,
      "training_step_time": 9.923177003860474
    },
    {
      "epoch": 3.09600830078125e-05,
      "model_forward_time": 3.153038263320923,
      "step": 8116
    },
    {
      "epoch": 3.09600830078125e-05,
      "step": 8116,
      "training_step_time": 10.313932657241821
    },
    {
      "epoch": 3.096389770507813e-05,
      "model_forward_time": 3.417951822280884,
      "step": 8117
    },
    {
      "epoch": 3.096389770507813e-05,
      "step": 8117,
      "training_step_time": 10.206128358840942
    },
    {
      "epoch": 3.096389770507813e-05,
      "model_forward_time": 3.1514499187469482,
      "step": 8117
    },
    {
      "epoch": 3.096389770507813e-05,
      "step": 8117,
      "training_step_time": 9.941749334335327
    },
    {
      "epoch": 3.096389770507813e-05,
      "model_forward_time": 3.151143789291382,
      "step": 8117
    },
    {
      "dynamics_loss_avg": 0.04899838995188475,
      "epoch": 3.096389770507813e-05,
      "step": 8117
    },
    {
      "action_loss_avg": 0.017762859258800744,
      "epoch": 3.096389770507813e-05,
      "step": 8117
    },
    {
      "epoch": 3.096389770507813e-05,
      "step": 8117,
      "training_step_time": 9.949784517288208
    },
    {
      "epoch": 3.096389770507813e-05,
      "model_forward_time": 3.158250093460083,
      "step": 8117
    },
    {
      "epoch": 3.096389770507813e-05,
      "step": 8117,
      "training_step_time": 10.292237997055054
    },
    {
      "epoch": 3.096771240234375e-05,
      "model_forward_time": 3.175530195236206,
      "step": 8118
    },
    {
      "epoch": 3.096771240234375e-05,
      "step": 8118,
      "training_step_time": 9.966553211212158
    },
    {
      "epoch": 3.096771240234375e-05,
      "model_forward_time": 2.7094480991363525,
      "step": 8118
    },
    {
      "epoch": 3.096771240234375e-05,
      "step": 8118,
      "training_step_time": 7.749849319458008
    },
    {
      "epoch": 3.096771240234375e-05,
      "model_forward_time": 2.43757963180542,
      "step": 8118
    },
    {
      "epoch": 3.096771240234375e-05,
      "step": 8118,
      "training_step_time": 7.467128038406372
    },
    {
      "epoch": 3.096771240234375e-05,
      "model_forward_time": 2.435441732406616,
      "step": 8118
    },
    {
      "epoch": 3.096771240234375e-05,
      "step": 8118,
      "training_step_time": 10.277096271514893
    },
    {
      "epoch": 3.0971527099609374e-05,
      "model_forward_time": 2.455317735671997,
      "step": 8119
    },
    {
      "epoch": 3.0971527099609374e-05,
      "step": 8119,
      "training_step_time": 7.486706972122192
    },
    {
      "epoch": 3.0971527099609374e-05,
      "model_forward_time": 2.430396795272827,
      "step": 8119
    },
    {
      "epoch": 3.0971527099609374e-05,
      "step": 8119,
      "training_step_time": 7.462291479110718
    },
    {
      "epoch": 3.0971527099609374e-05,
      "model_forward_time": 2.4340460300445557,
      "step": 8119
    },
    {
      "epoch": 3.0971527099609374e-05,
      "step": 8119,
      "training_step_time": 7.4643425941467285
    },
    {
      "epoch": 3.0971527099609374e-05,
      "model_forward_time": 2.714390277862549,
      "step": 8119
    },
    {
      "epoch": 3.0971527099609374e-05,
      "step": 8119,
      "training_step_time": 10.28183650970459
    },
    {
      "epoch": 3.0975341796875e-05,
      "grad_norm": 0.339733749628067,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 8120
    },
    {
      "epoch": 3.0975341796875e-05,
      "model_forward_time": 2.4495949745178223,
      "step": 8120
    },
    {
      "dynamics_loss_avg": 0.05114830490201712,
      "epoch": 3.0975341796875e-05,
      "step": 8120
    },
    {
      "action_loss_avg": 0.019586264342069625,
      "epoch": 3.0975341796875e-05,
      "step": 8120
    },
    {
      "epoch": 3.0975341796875e-05,
      "step": 8120,
      "training_step_time": 7.483386516571045
    },
    {
      "epoch": 3.0975341796875e-05,
      "model_forward_time": 3.1337759494781494,
      "step": 8120
    },
    {
      "epoch": 3.0975341796875e-05,
      "step": 8120,
      "training_step_time": 9.909249782562256
    },
    {
      "epoch": 3.0975341796875e-05,
      "model_forward_time": 2.4360220432281494,
      "step": 8120
    },
    {
      "epoch": 3.0975341796875e-05,
      "step": 8120,
      "training_step_time": 7.466443061828613
    },
    {
      "epoch": 3.0975341796875e-05,
      "model_forward_time": 2.436366319656372,
      "step": 8120
    },
    {
      "epoch": 3.0975341796875e-05,
      "step": 8120,
      "training_step_time": 10.012319803237915
    },
    {
      "epoch": 3.097915649414063e-05,
      "model_forward_time": 3.439473867416382,
      "step": 8121
    },
    {
      "epoch": 3.097915649414063e-05,
      "step": 8121,
      "training_step_time": 10.214128732681274
    },
    {
      "epoch": 3.097915649414063e-05,
      "model_forward_time": 2.438237428665161,
      "step": 8121
    },
    {
      "epoch": 3.097915649414063e-05,
      "step": 8121,
      "training_step_time": 7.467689514160156
    },
    {
      "epoch": 3.097915649414063e-05,
      "model_forward_time": 2.433119773864746,
      "step": 8121
    },
    {
      "epoch": 3.097915649414063e-05,
      "step": 8121,
      "training_step_time": 7.465174674987793
    },
    {
      "epoch": 3.097915649414063e-05,
      "model_forward_time": 3.135671854019165,
      "step": 8121
    },
    {
      "epoch": 3.097915649414063e-05,
      "step": 8121,
      "training_step_time": 10.28873348236084
    },
    {
      "epoch": 3.098297119140625e-05,
      "model_forward_time": 3.1598522663116455,
      "step": 8122
    },
    {
      "epoch": 3.098297119140625e-05,
      "step": 8122,
      "training_step_time": 9.936004400253296
    },
    {
      "epoch": 3.098297119140625e-05,
      "model_forward_time": 2.438157081604004,
      "step": 8122
    },
    {
      "epoch": 3.098297119140625e-05,
      "step": 8122,
      "training_step_time": 7.469686269760132
    },
    {
      "epoch": 3.098297119140625e-05,
      "model_forward_time": 3.4156124591827393,
      "step": 8122
    },
    {
      "dynamics_loss_avg": 0.05176578965038061,
      "epoch": 3.098297119140625e-05,
      "step": 8122
    },
    {
      "action_loss_avg": 0.02060925466939807,
      "epoch": 3.098297119140625e-05,
      "step": 8122
    },
    {
      "epoch": 3.098297119140625e-05,
      "step": 8122,
      "training_step_time": 10.191213369369507
    },
    {
      "epoch": 3.098297119140625e-05,
      "model_forward_time": 2.4309656620025635,
      "step": 8122
    },
    {
      "epoch": 3.098297119140625e-05,
      "step": 8122,
      "training_step_time": 10.24403977394104
    },
    {
      "epoch": 3.098678588867187e-05,
      "model_forward_time": 3.1474738121032715,
      "step": 8123
    },
    {
      "epoch": 3.098678588867187e-05,
      "step": 8123,
      "training_step_time": 9.926139831542969
    },
    {
      "epoch": 3.098678588867187e-05,
      "model_forward_time": 3.1379880905151367,
      "step": 8123
    },
    {
      "epoch": 3.098678588867187e-05,
      "step": 8123,
      "training_step_time": 9.917064428329468
    },
    {
      "epoch": 3.098678588867187e-05,
      "model_forward_time": 2.4368765354156494,
      "step": 8123
    },
    {
      "epoch": 3.098678588867187e-05,
      "step": 8123,
      "training_step_time": 7.4700939655303955
    },
    {
      "epoch": 3.098678588867187e-05,
      "model_forward_time": 3.396764039993286,
      "step": 8123
    },
    {
      "epoch": 3.098678588867187e-05,
      "step": 8123,
      "training_step_time": 10.270851135253906
    },
    {
      "epoch": 3.09906005859375e-05,
      "model_forward_time": 2.444805860519409,
      "step": 8124
    },
    {
      "epoch": 3.09906005859375e-05,
      "step": 8124,
      "training_step_time": 7.482255220413208
    },
    {
      "epoch": 3.09906005859375e-05,
      "model_forward_time": 2.43446946144104,
      "step": 8124
    },
    {
      "epoch": 3.09906005859375e-05,
      "step": 8124,
      "training_step_time": 7.461531639099121
    },
    {
      "epoch": 3.09906005859375e-05,
      "model_forward_time": 3.1336097717285156,
      "step": 8124
    },
    {
      "epoch": 3.09906005859375e-05,
      "step": 8124,
      "training_step_time": 9.909398317337036
    },
    {
      "epoch": 3.09906005859375e-05,
      "model_forward_time": 3.1401212215423584,
      "step": 8124
    },
    {
      "epoch": 3.09906005859375e-05,
      "step": 8124,
      "training_step_time": 10.298032522201538
    },
    {
      "epoch": 3.0994415283203125e-05,
      "model_forward_time": 2.447446346282959,
      "step": 8125
    },
    {
      "dynamics_loss_avg": 0.04828798305243254,
      "epoch": 3.0994415283203125e-05,
      "step": 8125
    },
    {
      "action_loss_avg": 0.016822712495923042,
      "epoch": 3.0994415283203125e-05,
      "step": 8125
    },
    {
      "epoch": 3.0994415283203125e-05,
      "step": 8125,
      "training_step_time": 7.765887022018433
    },
    {
      "epoch": 3.0994415283203125e-05,
      "model_forward_time": 3.1397902965545654,
      "step": 8125
    },
    {
      "epoch": 3.0994415283203125e-05,
      "step": 8125,
      "training_step_time": 9.914198875427246
    },
    {
      "epoch": 3.0994415283203125e-05,
      "model_forward_time": 3.140526294708252,
      "step": 8125
    },
    {
      "epoch": 3.0994415283203125e-05,
      "step": 8125,
      "training_step_time": 9.915461540222168
    },
    {
      "epoch": 3.0994415283203125e-05,
      "model_forward_time": 3.143301010131836,
      "step": 8125
    },
    {
      "epoch": 3.0994415283203125e-05,
      "step": 8125,
      "training_step_time": 10.267872095108032
    },
    {
      "epoch": 3.099822998046875e-05,
      "model_forward_time": 2.446669578552246,
      "step": 8126
    },
    {
      "epoch": 3.099822998046875e-05,
      "step": 8126,
      "training_step_time": 7.482813119888306
    },
    {
      "epoch": 3.099822998046875e-05,
      "model_forward_time": 3.456300735473633,
      "step": 8126
    },
    {
      "epoch": 3.099822998046875e-05,
      "step": 8126,
      "training_step_time": 10.230603694915771
    },
    {
      "epoch": 3.099822998046875e-05,
      "model_forward_time": 3.1464743614196777,
      "step": 8126
    },
    {
      "epoch": 3.099822998046875e-05,
      "step": 8126,
      "training_step_time": 9.926997423171997
    },
    {
      "epoch": 3.099822998046875e-05,
      "model_forward_time": 3.1405742168426514,
      "step": 8126
    },
    {
      "epoch": 3.099822998046875e-05,
      "step": 8126,
      "training_step_time": 10.281245231628418
    },
    {
      "epoch": 3.100204467773438e-05,
      "model_forward_time": 2.4541125297546387,
      "step": 8127
    },
    {
      "epoch": 3.100204467773438e-05,
      "step": 8127,
      "training_step_time": 7.490216970443726
    },
    {
      "epoch": 3.100204467773438e-05,
      "model_forward_time": 2.4288721084594727,
      "step": 8127
    },
    {
      "epoch": 3.100204467773438e-05,
      "step": 8127,
      "training_step_time": 7.457774639129639
    },
    {
      "epoch": 3.100204467773438e-05,
      "model_forward_time": 2.4323463439941406,
      "step": 8127
    },
    {
      "dynamics_loss_avg": 0.04982094094157219,
      "epoch": 3.100204467773438e-05,
      "step": 8127
    },
    {
      "action_loss_avg": 0.02150947260670364,
      "epoch": 3.100204467773438e-05,
      "step": 8127
    },
    {
      "epoch": 3.100204467773438e-05,
      "step": 8127,
      "training_step_time": 7.465577840805054
    },
    {
      "epoch": 3.100204467773438e-05,
      "model_forward_time": 3.4253997802734375,
      "step": 8127
    },
    {
      "epoch": 3.100204467773438e-05,
      "step": 8127,
      "training_step_time": 10.299089193344116
    },
    {
      "epoch": 3.1005859375e-05,
      "model_forward_time": 1.7686035633087158,
      "step": 8128
    },
    {
      "epoch": 3.1005859375e-05,
      "step": 8128,
      "training_step_time": 5.234625816345215
    },
    {
      "epoch": 3.1005859375e-05,
      "model_forward_time": 3.1355514526367188,
      "step": 8128
    },
    {
      "epoch": 3.1005859375e-05,
      "step": 8128,
      "training_step_time": 9.909145832061768
    },
    {
      "epoch": 3.1005859375e-05,
      "model_forward_time": 2.4361507892608643,
      "step": 8128
    },
    {
      "epoch": 3.1005859375e-05,
      "step": 8128,
      "training_step_time": 7.4675188064575195
    },
    {
      "epoch": 3.1005859375e-05,
      "model_forward_time": 3.136120319366455,
      "step": 8128
    },
    {
      "epoch": 3.1005859375e-05,
      "step": 8128,
      "training_step_time": 10.01870059967041
    },
    {
      "epoch": 3.100967407226562e-05,
      "model_forward_time": 2.753371238708496,
      "step": 8129
    },
    {
      "epoch": 3.100967407226562e-05,
      "step": 8129,
      "training_step_time": 7.789690732955933
    },
    {
      "epoch": 3.100967407226562e-05,
      "model_forward_time": 3.1398425102233887,
      "step": 8129
    },
    {
      "epoch": 3.100967407226562e-05,
      "step": 8129,
      "training_step_time": 9.923174142837524
    },
    {
      "epoch": 3.100967407226562e-05,
      "model_forward_time": 2.4357004165649414,
      "step": 8129
    },
    {
      "epoch": 3.100967407226562e-05,
      "step": 8129,
      "training_step_time": 7.469798803329468
    },
    {
      "epoch": 3.100967407226562e-05,
      "model_forward_time": 3.139486074447632,
      "step": 8129
    },
    {
      "epoch": 3.100967407226562e-05,
      "step": 8129,
      "training_step_time": 10.293225765228271
    },
    {
      "epoch": 3.101348876953125e-05,
      "grad_norm": 0.3312898874282837,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 8130
    },
    {
      "epoch": 3.101348876953125e-05,
      "model_forward_time": 2.449936628341675,
      "step": 8130
    },
    {
      "dynamics_loss_avg": 0.04784111548215151,
      "epoch": 3.101348876953125e-05,
      "step": 8130
    },
    {
      "action_loss_avg": 0.01899756621569395,
      "epoch": 3.101348876953125e-05,
      "step": 8130
    },
    {
      "epoch": 3.101348876953125e-05,
      "step": 8130,
      "training_step_time": 7.4872965812683105
    },
    {
      "epoch": 3.101348876953125e-05,
      "model_forward_time": 3.144749879837036,
      "step": 8130
    },
    {
      "epoch": 3.101348876953125e-05,
      "step": 8130,
      "training_step_time": 10.198655128479004
    },
    {
      "epoch": 3.101348876953125e-05,
      "model_forward_time": 3.1458070278167725,
      "step": 8130
    },
    {
      "epoch": 3.101348876953125e-05,
      "step": 8130,
      "training_step_time": 9.928027868270874
    },
    {
      "epoch": 3.101348876953125e-05,
      "model_forward_time": 3.1461944580078125,
      "step": 8130
    },
    {
      "epoch": 3.101348876953125e-05,
      "step": 8130,
      "training_step_time": 10.27480435371399
    },
    {
      "epoch": 3.1017303466796876e-05,
      "model_forward_time": 2.451869010925293,
      "step": 8131
    },
    {
      "epoch": 3.1017303466796876e-05,
      "step": 8131,
      "training_step_time": 7.491593837738037
    },
    {
      "epoch": 3.1017303466796876e-05,
      "model_forward_time": 2.4390013217926025,
      "step": 8131
    },
    {
      "epoch": 3.1017303466796876e-05,
      "step": 8131,
      "training_step_time": 7.476705551147461
    },
    {
      "epoch": 3.1017303466796876e-05,
      "model_forward_time": 2.4361629486083984,
      "step": 8131
    },
    {
      "epoch": 3.1017303466796876e-05,
      "step": 8131,
      "training_step_time": 7.468372583389282
    },
    {
      "epoch": 3.1017303466796876e-05,
      "model_forward_time": 3.3918662071228027,
      "step": 8131
    },
    {
      "epoch": 3.1017303466796876e-05,
      "step": 8131,
      "training_step_time": 10.2594575881958
    },
    {
      "epoch": 3.10211181640625e-05,
      "model_forward_time": 3.1514508724212646,
      "step": 8132
    },
    {
      "epoch": 3.10211181640625e-05,
      "step": 8132,
      "training_step_time": 9.931397676467896
    },
    {
      "epoch": 3.10211181640625e-05,
      "model_forward_time": 2.4379358291625977,
      "step": 8132
    },
    {
      "epoch": 3.10211181640625e-05,
      "step": 8132,
      "training_step_time": 7.472262859344482
    },
    {
      "epoch": 3.10211181640625e-05,
      "model_forward_time": 3.1359031200408936,
      "step": 8132
    },
    {
      "dynamics_loss_avg": 0.05796122923493385,
      "epoch": 3.10211181640625e-05,
      "step": 8132
    },
    {
      "action_loss_avg": 0.024790375120937826,
      "epoch": 3.10211181640625e-05,
      "step": 8132
    },
    {
      "epoch": 3.10211181640625e-05,
      "step": 8132,
      "training_step_time": 9.913564682006836
    },
    {
      "epoch": 3.10211181640625e-05,
      "model_forward_time": 2.4367856979370117,
      "step": 8132
    },
    {
      "epoch": 3.10211181640625e-05,
      "step": 8132,
      "training_step_time": 10.241748094558716
    },
    {
      "epoch": 3.102493286132812e-05,
      "model_forward_time": 2.761777400970459,
      "step": 8133
    },
    {
      "epoch": 3.102493286132812e-05,
      "step": 8133,
      "training_step_time": 7.793607950210571
    },
    {
      "epoch": 3.102493286132812e-05,
      "model_forward_time": 3.137439727783203,
      "step": 8133
    },
    {
      "epoch": 3.102493286132812e-05,
      "step": 8133,
      "training_step_time": 9.912525415420532
    },
    {
      "epoch": 3.102493286132812e-05,
      "model_forward_time": 2.435663938522339,
      "step": 8133
    },
    {
      "epoch": 3.102493286132812e-05,
      "step": 8133,
      "training_step_time": 7.475063800811768
    },
    {
      "epoch": 3.102493286132812e-05,
      "model_forward_time": 3.14011549949646,
      "step": 8133
    },
    {
      "epoch": 3.102493286132812e-05,
      "step": 8133,
      "training_step_time": 10.290277481079102
    },
    {
      "epoch": 3.102874755859375e-05,
      "model_forward_time": 2.4564590454101562,
      "step": 8134
    },
    {
      "epoch": 3.102874755859375e-05,
      "step": 8134,
      "training_step_time": 7.4886698722839355
    },
    {
      "epoch": 3.102874755859375e-05,
      "model_forward_time": 3.414583921432495,
      "step": 8134
    },
    {
      "epoch": 3.102874755859375e-05,
      "step": 8134,
      "training_step_time": 10.198333501815796
    },
    {
      "epoch": 3.102874755859375e-05,
      "model_forward_time": 2.4341447353363037,
      "step": 8134
    },
    {
      "epoch": 3.102874755859375e-05,
      "step": 8134,
      "training_step_time": 7.463743209838867
    },
    {
      "epoch": 3.102874755859375e-05,
      "model_forward_time": 2.434154748916626,
      "step": 8134
    },
    {
      "epoch": 3.102874755859375e-05,
      "step": 8134,
      "training_step_time": 10.009215354919434
    },
    {
      "epoch": 3.1032562255859374e-05,
      "model_forward_time": 3.1536166667938232,
      "step": 8135
    },
    {
      "dynamics_loss_avg": 0.06185902180150151,
      "epoch": 3.1032562255859374e-05,
      "step": 8135
    },
    {
      "action_loss_avg": 0.020009302371181547,
      "epoch": 3.1032562255859374e-05,
      "step": 8135
    },
    {
      "epoch": 3.1032562255859374e-05,
      "step": 8135,
      "training_step_time": 9.933959007263184
    },
    {
      "epoch": 3.1032562255859374e-05,
      "model_forward_time": 2.43701171875,
      "step": 8135
    },
    {
      "epoch": 3.1032562255859374e-05,
      "step": 8135,
      "training_step_time": 7.473280191421509
    },
    {
      "epoch": 3.1032562255859374e-05,
      "model_forward_time": 3.4018683433532715,
      "step": 8135
    },
    {
      "epoch": 3.1032562255859374e-05,
      "step": 8135,
      "training_step_time": 10.174052238464355
    },
    {
      "epoch": 3.1032562255859374e-05,
      "model_forward_time": 3.1411001682281494,
      "step": 8135
    },
    {
      "epoch": 3.1032562255859374e-05,
      "step": 8135,
      "training_step_time": 10.267947912216187
    },
    {
      "epoch": 3.1036376953125e-05,
      "model_forward_time": 3.149380922317505,
      "step": 8136
    },
    {
      "epoch": 3.1036376953125e-05,
      "step": 8136,
      "training_step_time": 9.935699939727783
    },
    {
      "epoch": 3.1036376953125e-05,
      "model_forward_time": 3.131040334701538,
      "step": 8136
    },
    {
      "epoch": 3.1036376953125e-05,
      "step": 8136,
      "training_step_time": 9.906557559967041
    },
    {
      "epoch": 3.1036376953125e-05,
      "model_forward_time": 3.135256052017212,
      "step": 8136
    },
    {
      "epoch": 3.1036376953125e-05,
      "step": 8136,
      "training_step_time": 9.910719633102417
    },
    {
      "epoch": 3.1036376953125e-05,
      "model_forward_time": 2.4289236068725586,
      "step": 8136
    },
    {
      "epoch": 3.1036376953125e-05,
      "step": 8136,
      "training_step_time": 10.013318538665771
    },
    {
      "epoch": 3.104019165039063e-05,
      "model_forward_time": 2.7596611976623535,
      "step": 8137
    },
    {
      "epoch": 3.104019165039063e-05,
      "step": 8137,
      "training_step_time": 7.792847394943237
    },
    {
      "epoch": 3.104019165039063e-05,
      "model_forward_time": 3.1304640769958496,
      "step": 8137
    },
    {
      "epoch": 3.104019165039063e-05,
      "step": 8137,
      "training_step_time": 9.907135725021362
    },
    {
      "epoch": 3.104019165039063e-05,
      "model_forward_time": 2.431178331375122,
      "step": 8137
    },
    {
      "dynamics_loss_avg": 0.062044719234108925,
      "epoch": 3.104019165039063e-05,
      "step": 8137
    },
    {
      "action_loss_avg": 0.026657255366444588,
      "epoch": 3.104019165039063e-05,
      "step": 8137
    },
    {
      "epoch": 3.104019165039063e-05,
      "step": 8137,
      "training_step_time": 7.466854095458984
    },
    {
      "epoch": 3.104019165039063e-05,
      "model_forward_time": 3.130986452102661,
      "step": 8137
    },
    {
      "epoch": 3.104019165039063e-05,
      "step": 8137,
      "training_step_time": 10.299545526504517
    },
    {
      "epoch": 3.104400634765625e-05,
      "model_forward_time": 1.780043601989746,
      "step": 8138
    },
    {
      "epoch": 3.104400634765625e-05,
      "step": 8138,
      "training_step_time": 5.243073463439941
    },
    {
      "epoch": 3.104400634765625e-05,
      "model_forward_time": 3.3768043518066406,
      "step": 8138
    },
    {
      "epoch": 3.104400634765625e-05,
      "step": 8138,
      "training_step_time": 10.151669263839722
    },
    {
      "epoch": 3.104400634765625e-05,
      "model_forward_time": 2.4317431449890137,
      "step": 8138
    },
    {
      "epoch": 3.104400634765625e-05,
      "step": 8138,
      "training_step_time": 7.467360019683838
    },
    {
      "epoch": 3.104400634765625e-05,
      "model_forward_time": 3.133882522583008,
      "step": 8138
    },
    {
      "epoch": 3.104400634765625e-05,
      "step": 8138,
      "training_step_time": 10.257305383682251
    },
    {
      "epoch": 3.104782104492187e-05,
      "model_forward_time": 2.445006847381592,
      "step": 8139
    },
    {
      "epoch": 3.104782104492187e-05,
      "step": 8139,
      "training_step_time": 7.4829161167144775
    },
    {
      "epoch": 3.104782104492187e-05,
      "model_forward_time": 3.1350390911102295,
      "step": 8139
    },
    {
      "epoch": 3.104782104492187e-05,
      "step": 8139,
      "training_step_time": 9.912217378616333
    },
    {
      "epoch": 3.104782104492187e-05,
      "model_forward_time": 2.73038649559021,
      "step": 8139
    },
    {
      "epoch": 3.104782104492187e-05,
      "step": 8139,
      "training_step_time": 7.764130592346191
    },
    {
      "epoch": 3.104782104492187e-05,
      "model_forward_time": 2.429441213607788,
      "step": 8139
    },
    {
      "epoch": 3.104782104492187e-05,
      "step": 8139,
      "training_step_time": 10.277231216430664
    },
    {
      "epoch": 3.10516357421875e-05,
      "grad_norm": 0.3892100155353546,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 8140
    },
    {
      "epoch": 3.10516357421875e-05,
      "model_forward_time": 2.4391191005706787,
      "step": 8140
    },
    {
      "dynamics_loss_avg": 0.05568242985755205,
      "epoch": 3.10516357421875e-05,
      "step": 8140
    },
    {
      "action_loss_avg": 0.021828700695186852,
      "epoch": 3.10516357421875e-05,
      "step": 8140
    },
    {
      "epoch": 3.10516357421875e-05,
      "step": 8140,
      "training_step_time": 7.475605249404907
    },
    {
      "epoch": 3.10516357421875e-05,
      "model_forward_time": 2.4342479705810547,
      "step": 8140
    },
    {
      "epoch": 3.10516357421875e-05,
      "step": 8140,
      "training_step_time": 7.468410015106201
    },
    {
      "epoch": 3.10516357421875e-05,
      "model_forward_time": 3.131558656692505,
      "step": 8140
    },
    {
      "epoch": 3.10516357421875e-05,
      "step": 8140,
      "training_step_time": 9.912996053695679
    },
    {
      "epoch": 3.10516357421875e-05,
      "model_forward_time": 3.1388931274414062,
      "step": 8140
    },
    {
      "epoch": 3.10516357421875e-05,
      "step": 8140,
      "training_step_time": 10.28930377960205
    },
    {
      "epoch": 3.1055450439453125e-05,
      "model_forward_time": 2.454380989074707,
      "step": 8141
    },
    {
      "epoch": 3.1055450439453125e-05,
      "step": 8141,
      "training_step_time": 7.48910117149353
    },
    {
      "epoch": 3.1055450439453125e-05,
      "model_forward_time": 2.4323294162750244,
      "step": 8141
    },
    {
      "epoch": 3.1055450439453125e-05,
      "step": 8141,
      "training_step_time": 7.467694282531738
    },
    {
      "epoch": 3.1055450439453125e-05,
      "model_forward_time": 2.434084177017212,
      "step": 8141
    },
    {
      "epoch": 3.1055450439453125e-05,
      "step": 8141,
      "training_step_time": 7.466345548629761
    },
    {
      "epoch": 3.1055450439453125e-05,
      "model_forward_time": 2.4251298904418945,
      "step": 8141
    },
    {
      "epoch": 3.1055450439453125e-05,
      "step": 8141,
      "training_step_time": 10.002910852432251
    },
    {
      "epoch": 3.105926513671875e-05,
      "model_forward_time": 3.1506450176239014,
      "step": 8142
    },
    {
      "epoch": 3.105926513671875e-05,
      "step": 8142,
      "training_step_time": 9.927118301391602
    },
    {
      "epoch": 3.105926513671875e-05,
      "model_forward_time": 3.408860683441162,
      "step": 8142
    },
    {
      "epoch": 3.105926513671875e-05,
      "step": 8142,
      "training_step_time": 10.180588006973267
    },
    {
      "epoch": 3.105926513671875e-05,
      "model_forward_time": 3.130854845046997,
      "step": 8142
    },
    {
      "dynamics_loss_avg": 0.05697720050811768,
      "epoch": 3.105926513671875e-05,
      "step": 8142
    },
    {
      "action_loss_avg": 0.02630990706384182,
      "epoch": 3.105926513671875e-05,
      "step": 8142
    },
    {
      "epoch": 3.105926513671875e-05,
      "step": 8142,
      "training_step_time": 9.900617361068726
    },
    {
      "epoch": 3.105926513671875e-05,
      "model_forward_time": 3.13470721244812,
      "step": 8142
    },
    {
      "epoch": 3.105926513671875e-05,
      "step": 8142,
      "training_step_time": 10.280688285827637
    },
    {
      "epoch": 3.106307983398438e-05,
      "model_forward_time": 3.146138906478882,
      "step": 8143
    },
    {
      "epoch": 3.106307983398438e-05,
      "step": 8143,
      "training_step_time": 9.921014785766602
    },
    {
      "epoch": 3.106307983398438e-05,
      "model_forward_time": 2.42867374420166,
      "step": 8143
    },
    {
      "epoch": 3.106307983398438e-05,
      "step": 8143,
      "training_step_time": 7.454714775085449
    },
    {
      "epoch": 3.106307983398438e-05,
      "model_forward_time": 2.6932499408721924,
      "step": 8143
    },
    {
      "epoch": 3.106307983398438e-05,
      "step": 8143,
      "training_step_time": 7.720280885696411
    },
    {
      "epoch": 3.106307983398438e-05,
      "model_forward_time": 2.426362991333008,
      "step": 8143
    },
    {
      "epoch": 3.106307983398438e-05,
      "step": 8143,
      "training_step_time": 10.252334594726562
    },
    {
      "epoch": 3.106689453125e-05,
      "model_forward_time": 2.438563585281372,
      "step": 8144
    },
    {
      "epoch": 3.106689453125e-05,
      "step": 8144,
      "training_step_time": 7.464598894119263
    },
    {
      "epoch": 3.106689453125e-05,
      "model_forward_time": 2.4262378215789795,
      "step": 8144
    },
    {
      "epoch": 3.106689453125e-05,
      "step": 8144,
      "training_step_time": 7.449118614196777
    },
    {
      "epoch": 3.106689453125e-05,
      "model_forward_time": 2.4238882064819336,
      "step": 8144
    },
    {
      "epoch": 3.106689453125e-05,
      "step": 8144,
      "training_step_time": 7.449315547943115
    },
    {
      "epoch": 3.106689453125e-05,
      "model_forward_time": 2.6843764781951904,
      "step": 8144
    },
    {
      "epoch": 3.106689453125e-05,
      "step": 8144,
      "training_step_time": 10.27020263671875
    },
    {
      "epoch": 3.107070922851562e-05,
      "model_forward_time": 2.4419844150543213,
      "step": 8145
    },
    {
      "dynamics_loss_avg": 0.057869347464293244,
      "epoch": 3.107070922851562e-05,
      "step": 8145
    },
    {
      "action_loss_avg": 0.02522990305442363,
      "epoch": 3.107070922851562e-05,
      "step": 8145
    },
    {
      "epoch": 3.107070922851562e-05,
      "step": 8145,
      "training_step_time": 7.473291635513306
    },
    {
      "epoch": 3.107070922851562e-05,
      "model_forward_time": 2.427716016769409,
      "step": 8145
    },
    {
      "epoch": 3.107070922851562e-05,
      "step": 8145,
      "training_step_time": 7.458200693130493
    },
    {
      "epoch": 3.107070922851562e-05,
      "model_forward_time": 2.4412949085235596,
      "step": 8145
    },
    {
      "epoch": 3.107070922851562e-05,
      "step": 8145,
      "training_step_time": 7.477071285247803
    },
    {
      "epoch": 3.107070922851562e-05,
      "model_forward_time": 3.139097213745117,
      "step": 8145
    },
    {
      "epoch": 3.107070922851562e-05,
      "step": 8145,
      "training_step_time": 10.270851612091064
    },
    {
      "epoch": 3.107452392578125e-05,
      "model_forward_time": 3.1550686359405518,
      "step": 8146
    },
    {
      "epoch": 3.107452392578125e-05,
      "step": 8146,
      "training_step_time": 9.937641143798828
    },
    {
      "epoch": 3.107452392578125e-05,
      "model_forward_time": 3.4062116146087646,
      "step": 8146
    },
    {
      "epoch": 3.107452392578125e-05,
      "step": 8146,
      "training_step_time": 10.186542987823486
    },
    {
      "epoch": 3.107452392578125e-05,
      "model_forward_time": 3.1478281021118164,
      "step": 8146
    },
    {
      "epoch": 3.107452392578125e-05,
      "step": 8146,
      "training_step_time": 9.928714513778687
    },
    {
      "epoch": 3.107452392578125e-05,
      "model_forward_time": 1.7503933906555176,
      "step": 8146
    },
    {
      "epoch": 3.107452392578125e-05,
      "step": 8146,
      "training_step_time": 10.282688617706299
    },
    {
      "epoch": 3.1078338623046876e-05,
      "model_forward_time": 3.154794931411743,
      "step": 8147
    },
    {
      "epoch": 3.1078338623046876e-05,
      "step": 8147,
      "training_step_time": 9.92919135093689
    },
    {
      "epoch": 3.1078338623046876e-05,
      "model_forward_time": 3.1389498710632324,
      "step": 8147
    },
    {
      "epoch": 3.1078338623046876e-05,
      "step": 8147,
      "training_step_time": 9.91663932800293
    },
    {
      "epoch": 3.1078338623046876e-05,
      "model_forward_time": 3.424247980117798,
      "step": 8147
    },
    {
      "dynamics_loss_avg": 0.06207660436630249,
      "epoch": 3.1078338623046876e-05,
      "step": 8147
    },
    {
      "action_loss_avg": 0.02500256048515439,
      "epoch": 3.1078338623046876e-05,
      "step": 8147
    },
    {
      "epoch": 3.1078338623046876e-05,
      "step": 8147,
      "training_step_time": 10.200528144836426
    },
    {
      "epoch": 3.1078338623046876e-05,
      "model_forward_time": 3.1444811820983887,
      "step": 8147
    },
    {
      "epoch": 3.1078338623046876e-05,
      "step": 8147,
      "training_step_time": 10.251659154891968
    },
    {
      "epoch": 3.10821533203125e-05,
      "model_forward_time": 3.150960922241211,
      "step": 8148
    },
    {
      "epoch": 3.10821533203125e-05,
      "step": 8148,
      "training_step_time": 9.932862997055054
    },
    {
      "epoch": 3.10821533203125e-05,
      "model_forward_time": 2.442401885986328,
      "step": 8148
    },
    {
      "epoch": 3.10821533203125e-05,
      "step": 8148,
      "training_step_time": 7.480693817138672
    },
    {
      "epoch": 3.10821533203125e-05,
      "model_forward_time": 3.1440930366516113,
      "step": 8148
    },
    {
      "epoch": 3.10821533203125e-05,
      "step": 8148,
      "training_step_time": 9.924197673797607
    },
    {
      "epoch": 3.10821533203125e-05,
      "model_forward_time": 3.4171149730682373,
      "step": 8148
    },
    {
      "epoch": 3.10821533203125e-05,
      "step": 8148,
      "training_step_time": 10.327696561813354
    },
    {
      "epoch": 3.108596801757812e-05,
      "model_forward_time": 3.1653995513916016,
      "step": 8149
    },
    {
      "epoch": 3.108596801757812e-05,
      "step": 8149,
      "training_step_time": 9.952295541763306
    },
    {
      "epoch": 3.108596801757812e-05,
      "model_forward_time": 2.435638189315796,
      "step": 8149
    },
    {
      "epoch": 3.108596801757812e-05,
      "step": 8149,
      "training_step_time": 7.472610712051392
    },
    {
      "epoch": 3.108596801757812e-05,
      "model_forward_time": 3.134046792984009,
      "step": 8149
    },
    {
      "epoch": 3.108596801757812e-05,
      "step": 8149,
      "training_step_time": 9.918615341186523
    },
    {
      "epoch": 3.108596801757812e-05,
      "model_forward_time": 3.1410934925079346,
      "step": 8149
    },
    {
      "epoch": 3.108596801757812e-05,
      "step": 8149,
      "training_step_time": 10.293752193450928
    },
    {
      "epoch": 3.108978271484375e-05,
      "grad_norm": 0.3942389488220215,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 8150
    },
    {
      "epoch": 3.108978271484375e-05,
      "model_forward_time": 2.7215301990509033,
      "step": 8150
    },
    {
      "dynamics_loss_avg": 0.08037116043269635,
      "epoch": 3.108978271484375e-05,
      "step": 8150
    },
    {
      "action_loss_avg": 0.02825078899040818,
      "epoch": 3.108978271484375e-05,
      "step": 8150
    },
    {
      "epoch": 3.108978271484375e-05,
      "step": 8150,
      "training_step_time": 7.757401943206787
    },
    {
      "epoch": 3.108978271484375e-05,
      "model_forward_time": 2.427534341812134,
      "step": 8150
    },
    {
      "epoch": 3.108978271484375e-05,
      "step": 8150,
      "training_step_time": 7.454615831375122
    },
    {
      "epoch": 3.108978271484375e-05,
      "model_forward_time": 2.434793710708618,
      "step": 8150
    },
    {
      "epoch": 3.108978271484375e-05,
      "step": 8150,
      "training_step_time": 7.4677510261535645
    },
    {
      "epoch": 3.108978271484375e-05,
      "model_forward_time": 2.421288013458252,
      "step": 8150
    },
    {
      "epoch": 3.108978271484375e-05,
      "step": 8150,
      "training_step_time": 10.005361080169678
    },
    {
      "epoch": 3.1093597412109374e-05,
      "model_forward_time": 2.450141668319702,
      "step": 8151
    },
    {
      "epoch": 3.1093597412109374e-05,
      "step": 8151,
      "training_step_time": 7.481381177902222
    },
    {
      "epoch": 3.1093597412109374e-05,
      "model_forward_time": 2.4300882816314697,
      "step": 8151
    },
    {
      "epoch": 3.1093597412109374e-05,
      "step": 8151,
      "training_step_time": 7.4573609828948975
    },
    {
      "epoch": 3.1093597412109374e-05,
      "model_forward_time": 3.391122817993164,
      "step": 8151
    },
    {
      "epoch": 3.1093597412109374e-05,
      "step": 8151,
      "training_step_time": 10.162143468856812
    },
    {
      "epoch": 3.1093597412109374e-05,
      "model_forward_time": 2.435753583908081,
      "step": 8151
    },
    {
      "epoch": 3.1093597412109374e-05,
      "step": 8151,
      "training_step_time": 9.996407985687256
    },
    {
      "epoch": 3.1097412109375e-05,
      "model_forward_time": 3.156372547149658,
      "step": 8152
    },
    {
      "epoch": 3.1097412109375e-05,
      "step": 8152,
      "training_step_time": 9.926303386688232
    },
    {
      "epoch": 3.1097412109375e-05,
      "model_forward_time": 3.1303210258483887,
      "step": 8152
    },
    {
      "epoch": 3.1097412109375e-05,
      "step": 8152,
      "training_step_time": 9.899035215377808
    },
    {
      "epoch": 3.1097412109375e-05,
      "model_forward_time": 3.1333584785461426,
      "step": 8152
    },
    {
      "dynamics_loss_avg": 0.06165169812738895,
      "epoch": 3.1097412109375e-05,
      "step": 8152
    },
    {
      "action_loss_avg": 0.0226724935695529,
      "epoch": 3.1097412109375e-05,
      "step": 8152
    },
    {
      "epoch": 3.1097412109375e-05,
      "step": 8152,
      "training_step_time": 9.915594339370728
    },
    {
      "epoch": 3.1097412109375e-05,
      "model_forward_time": 3.4064934253692627,
      "step": 8152
    },
    {
      "epoch": 3.1097412109375e-05,
      "step": 8152,
      "training_step_time": 10.281071901321411
    },
    {
      "epoch": 3.110122680664063e-05,
      "model_forward_time": 2.4448485374450684,
      "step": 8153
    },
    {
      "epoch": 3.110122680664063e-05,
      "step": 8153,
      "training_step_time": 7.481072902679443
    },
    {
      "epoch": 3.110122680664063e-05,
      "model_forward_time": 3.1353187561035156,
      "step": 8153
    },
    {
      "epoch": 3.110122680664063e-05,
      "step": 8153,
      "training_step_time": 9.918564796447754
    },
    {
      "epoch": 3.110122680664063e-05,
      "model_forward_time": 3.1461520195007324,
      "step": 8153
    },
    {
      "epoch": 3.110122680664063e-05,
      "step": 8153,
      "training_step_time": 9.928897619247437
    },
    {
      "epoch": 3.110122680664063e-05,
      "model_forward_time": 2.444096326828003,
      "step": 8153
    },
    {
      "epoch": 3.110122680664063e-05,
      "step": 8153,
      "training_step_time": 10.321328163146973
    },
    {
      "epoch": 3.110504150390625e-05,
      "model_forward_time": 2.4467647075653076,
      "step": 8154
    },
    {
      "epoch": 3.110504150390625e-05,
      "step": 8154,
      "training_step_time": 7.76393985748291
    },
    {
      "epoch": 3.110504150390625e-05,
      "model_forward_time": 2.433126449584961,
      "step": 8154
    },
    {
      "epoch": 3.110504150390625e-05,
      "step": 8154,
      "training_step_time": 7.471086502075195
    },
    {
      "epoch": 3.110504150390625e-05,
      "model_forward_time": 3.143646240234375,
      "step": 8154
    },
    {
      "epoch": 3.110504150390625e-05,
      "step": 8154,
      "training_step_time": 9.927046775817871
    },
    {
      "epoch": 3.110504150390625e-05,
      "model_forward_time": 3.1452271938323975,
      "step": 8154
    },
    {
      "epoch": 3.110504150390625e-05,
      "step": 8154,
      "training_step_time": 10.305877447128296
    },
    {
      "epoch": 3.110885620117187e-05,
      "model_forward_time": 3.1640994548797607,
      "step": 8155
    },
    {
      "dynamics_loss_avg": 0.05171454027295112,
      "epoch": 3.110885620117187e-05,
      "step": 8155
    },
    {
      "action_loss_avg": 0.022466426668688654,
      "epoch": 3.110885620117187e-05,
      "step": 8155
    },
    {
      "epoch": 3.110885620117187e-05,
      "step": 8155,
      "training_step_time": 9.95273208618164
    },
    {
      "epoch": 3.110885620117187e-05,
      "model_forward_time": 2.4353883266448975,
      "step": 8155
    },
    {
      "epoch": 3.110885620117187e-05,
      "step": 8155,
      "training_step_time": 7.478201389312744
    },
    {
      "epoch": 3.110885620117187e-05,
      "model_forward_time": 2.675429105758667,
      "step": 8155
    },
    {
      "epoch": 3.110885620117187e-05,
      "step": 8155,
      "training_step_time": 7.711330413818359
    },
    {
      "epoch": 3.110885620117187e-05,
      "model_forward_time": 2.432231903076172,
      "step": 8155
    },
    {
      "epoch": 3.110885620117187e-05,
      "step": 8155,
      "training_step_time": 10.027935028076172
    },
    {
      "epoch": 3.11126708984375e-05,
      "model_forward_time": 2.4531402587890625,
      "step": 8156
    },
    {
      "epoch": 3.11126708984375e-05,
      "step": 8156,
      "training_step_time": 7.4910969734191895
    },
    {
      "epoch": 3.11126708984375e-05,
      "model_forward_time": 3.1455564498901367,
      "step": 8156
    },
    {
      "epoch": 3.11126708984375e-05,
      "step": 8156,
      "training_step_time": 9.930498123168945
    },
    {
      "epoch": 3.11126708984375e-05,
      "model_forward_time": 3.1431949138641357,
      "step": 8156
    },
    {
      "epoch": 3.11126708984375e-05,
      "step": 8156,
      "training_step_time": 9.925314903259277
    },
    {
      "epoch": 3.11126708984375e-05,
      "model_forward_time": 2.7453291416168213,
      "step": 8156
    },
    {
      "epoch": 3.11126708984375e-05,
      "step": 8156,
      "training_step_time": 10.26774549484253
    },
    {
      "epoch": 3.1116485595703125e-05,
      "model_forward_time": 2.455240488052368,
      "step": 8157
    },
    {
      "epoch": 3.1116485595703125e-05,
      "step": 8157,
      "training_step_time": 7.493098020553589
    },
    {
      "epoch": 3.1116485595703125e-05,
      "model_forward_time": 2.4403769969940186,
      "step": 8157
    },
    {
      "epoch": 3.1116485595703125e-05,
      "step": 8157,
      "training_step_time": 7.475764989852905
    },
    {
      "epoch": 3.1116485595703125e-05,
      "model_forward_time": 2.4339816570281982,
      "step": 8157
    },
    {
      "dynamics_loss_avg": 0.05850564856082201,
      "epoch": 3.1116485595703125e-05,
      "step": 8157
    },
    {
      "action_loss_avg": 0.018465811386704445,
      "epoch": 3.1116485595703125e-05,
      "step": 8157
    },
    {
      "epoch": 3.1116485595703125e-05,
      "step": 8157,
      "training_step_time": 7.467815399169922
    },
    {
      "epoch": 3.1116485595703125e-05,
      "model_forward_time": 2.43544340133667,
      "step": 8157
    },
    {
      "epoch": 3.1116485595703125e-05,
      "step": 8157,
      "training_step_time": 10.006989002227783
    },
    {
      "epoch": 3.112030029296875e-05,
      "model_forward_time": 3.4100444316864014,
      "step": 8158
    },
    {
      "epoch": 3.112030029296875e-05,
      "step": 8158,
      "training_step_time": 10.186880588531494
    },
    {
      "epoch": 3.112030029296875e-05,
      "model_forward_time": 3.1427223682403564,
      "step": 8158
    },
    {
      "epoch": 3.112030029296875e-05,
      "step": 8158,
      "training_step_time": 9.921616792678833
    },
    {
      "epoch": 3.112030029296875e-05,
      "model_forward_time": 3.1376287937164307,
      "step": 8158
    },
    {
      "epoch": 3.112030029296875e-05,
      "step": 8158,
      "training_step_time": 9.910652875900269
    },
    {
      "epoch": 3.112030029296875e-05,
      "model_forward_time": 3.140268564224243,
      "step": 8158
    },
    {
      "epoch": 3.112030029296875e-05,
      "step": 8158,
      "training_step_time": 10.265668630599976
    },
    {
      "epoch": 3.112411499023438e-05,
      "model_forward_time": 2.448927640914917,
      "step": 8159
    },
    {
      "epoch": 3.112411499023438e-05,
      "step": 8159,
      "training_step_time": 7.479723930358887
    },
    {
      "epoch": 3.112411499023438e-05,
      "model_forward_time": 3.4150025844573975,
      "step": 8159
    },
    {
      "epoch": 3.112411499023438e-05,
      "step": 8159,
      "training_step_time": 10.187775135040283
    },
    {
      "epoch": 3.112411499023438e-05,
      "model_forward_time": 2.4309744834899902,
      "step": 8159
    },
    {
      "epoch": 3.112411499023438e-05,
      "step": 8159,
      "training_step_time": 7.465024948120117
    },
    {
      "epoch": 3.112411499023438e-05,
      "model_forward_time": 3.1333518028259277,
      "step": 8159
    },
    {
      "epoch": 3.112411499023438e-05,
      "step": 8159,
      "training_step_time": 10.280254602432251
    },
    {
      "epoch": 3.11279296875e-05,
      "grad_norm": 0.5035433769226074,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 8160
    },
    {
      "epoch": 3.11279296875e-05,
      "model_forward_time": 2.452828884124756,
      "step": 8160
    },
    {
      "dynamics_loss_avg": 0.0572571687400341,
      "epoch": 3.11279296875e-05,
      "step": 8160
    },
    {
      "action_loss_avg": 0.024076608009636402,
      "epoch": 3.11279296875e-05,
      "step": 8160
    },
    {
      "epoch": 3.11279296875e-05,
      "step": 8160,
      "training_step_time": 7.490888357162476
    },
    {
      "epoch": 3.11279296875e-05,
      "model_forward_time": 2.430553674697876,
      "step": 8160
    },
    {
      "epoch": 3.11279296875e-05,
      "step": 8160,
      "training_step_time": 7.46109938621521
    },
    {
      "epoch": 3.11279296875e-05,
      "model_forward_time": 2.431574821472168,
      "step": 8160
    },
    {
      "epoch": 3.11279296875e-05,
      "step": 8160,
      "training_step_time": 7.465550899505615
    },
    {
      "epoch": 3.11279296875e-05,
      "model_forward_time": 3.4106101989746094,
      "step": 8160
    },
    {
      "epoch": 3.11279296875e-05,
      "step": 8160,
      "training_step_time": 10.291480541229248
    },
    {
      "epoch": 3.1131744384765623e-05,
      "model_forward_time": 3.1522138118743896,
      "step": 8161
    },
    {
      "epoch": 3.1131744384765623e-05,
      "step": 8161,
      "training_step_time": 9.941184520721436
    },
    {
      "epoch": 3.1131744384765623e-05,
      "model_forward_time": 2.437825918197632,
      "step": 8161
    },
    {
      "epoch": 3.1131744384765623e-05,
      "step": 8161,
      "training_step_time": 7.4749555587768555
    },
    {
      "epoch": 3.1131744384765623e-05,
      "model_forward_time": 3.1404850482940674,
      "step": 8161
    },
    {
      "epoch": 3.1131744384765623e-05,
      "step": 8161,
      "training_step_time": 9.923208713531494
    },
    {
      "epoch": 3.1131744384765623e-05,
      "model_forward_time": 3.143848180770874,
      "step": 8161
    },
    {
      "epoch": 3.1131744384765623e-05,
      "step": 8161,
      "training_step_time": 10.249366521835327
    },
    {
      "epoch": 3.113555908203125e-05,
      "model_forward_time": 2.721731662750244,
      "step": 8162
    },
    {
      "epoch": 3.113555908203125e-05,
      "step": 8162,
      "training_step_time": 7.764088153839111
    },
    {
      "epoch": 3.113555908203125e-05,
      "model_forward_time": 3.1397247314453125,
      "step": 8162
    },
    {
      "epoch": 3.113555908203125e-05,
      "step": 8162,
      "training_step_time": 9.923620700836182
    },
    {
      "epoch": 3.113555908203125e-05,
      "model_forward_time": 3.1445600986480713,
      "step": 8162
    },
    {
      "dynamics_loss_avg": 0.07575750686228275,
      "epoch": 3.113555908203125e-05,
      "step": 8162
    },
    {
      "action_loss_avg": 0.028894787561148404,
      "epoch": 3.113555908203125e-05,
      "step": 8162
    },
    {
      "epoch": 3.113555908203125e-05,
      "step": 8162,
      "training_step_time": 9.932774543762207
    },
    {
      "epoch": 3.113555908203125e-05,
      "model_forward_time": 3.14323091506958,
      "step": 8162
    },
    {
      "epoch": 3.113555908203125e-05,
      "step": 8162,
      "training_step_time": 10.027368545532227
    },
    {
      "epoch": 3.1139373779296876e-05,
      "model_forward_time": 3.147844076156616,
      "step": 8163
    },
    {
      "epoch": 3.1139373779296876e-05,
      "step": 8163,
      "training_step_time": 9.935872316360474
    },
    {
      "epoch": 3.1139373779296876e-05,
      "model_forward_time": 3.4198713302612305,
      "step": 8163
    },
    {
      "epoch": 3.1139373779296876e-05,
      "step": 8163,
      "training_step_time": 10.212605953216553
    },
    {
      "epoch": 3.1139373779296876e-05,
      "model_forward_time": 3.1491355895996094,
      "step": 8163
    },
    {
      "epoch": 3.1139373779296876e-05,
      "step": 8163,
      "training_step_time": 9.935748100280762
    },
    {
      "epoch": 3.1139373779296876e-05,
      "model_forward_time": 2.4362356662750244,
      "step": 8163
    },
    {
      "epoch": 3.1139373779296876e-05,
      "step": 8163,
      "training_step_time": 10.250602722167969
    },
    {
      "epoch": 3.11431884765625e-05,
      "model_forward_time": 3.160215377807617,
      "step": 8164
    },
    {
      "epoch": 3.11431884765625e-05,
      "step": 8164,
      "training_step_time": 9.94950532913208
    },
    {
      "epoch": 3.11431884765625e-05,
      "model_forward_time": 2.4331107139587402,
      "step": 8164
    },
    {
      "epoch": 3.11431884765625e-05,
      "step": 8164,
      "training_step_time": 7.470352649688721
    },
    {
      "epoch": 3.11431884765625e-05,
      "model_forward_time": 3.4169273376464844,
      "step": 8164
    },
    {
      "epoch": 3.11431884765625e-05,
      "step": 8164,
      "training_step_time": 10.197256326675415
    },
    {
      "epoch": 3.11431884765625e-05,
      "model_forward_time": 3.1413040161132812,
      "step": 8164
    },
    {
      "epoch": 3.11431884765625e-05,
      "step": 8164,
      "training_step_time": 10.288336277008057
    },
    {
      "epoch": 3.114700317382812e-05,
      "model_forward_time": 2.4520905017852783,
      "step": 8165
    },
    {
      "dynamics_loss_avg": 0.05919662602245808,
      "epoch": 3.114700317382812e-05,
      "step": 8165
    },
    {
      "action_loss_avg": 0.024753373162820936,
      "epoch": 3.114700317382812e-05,
      "step": 8165
    },
    {
      "epoch": 3.114700317382812e-05,
      "step": 8165,
      "training_step_time": 7.488025188446045
    },
    {
      "epoch": 3.114700317382812e-05,
      "model_forward_time": 3.136470317840576,
      "step": 8165
    },
    {
      "epoch": 3.114700317382812e-05,
      "step": 8165,
      "training_step_time": 9.91349196434021
    },
    {
      "epoch": 3.114700317382812e-05,
      "model_forward_time": 3.1374447345733643,
      "step": 8165
    },
    {
      "epoch": 3.114700317382812e-05,
      "step": 8165,
      "training_step_time": 9.908422708511353
    },
    {
      "epoch": 3.114700317382812e-05,
      "model_forward_time": 2.719710111618042,
      "step": 8165
    },
    {
      "epoch": 3.114700317382812e-05,
      "step": 8165,
      "training_step_time": 10.273484945297241
    },
    {
      "epoch": 3.115081787109375e-05,
      "model_forward_time": 3.1463611125946045,
      "step": 8166
    },
    {
      "epoch": 3.115081787109375e-05,
      "step": 8166,
      "training_step_time": 9.926926612854004
    },
    {
      "epoch": 3.115081787109375e-05,
      "model_forward_time": 3.134777784347534,
      "step": 8166
    },
    {
      "epoch": 3.115081787109375e-05,
      "step": 8166,
      "training_step_time": 9.910863876342773
    },
    {
      "epoch": 3.115081787109375e-05,
      "model_forward_time": 2.433880090713501,
      "step": 8166
    },
    {
      "epoch": 3.115081787109375e-05,
      "step": 8166,
      "training_step_time": 7.465953826904297
    },
    {
      "epoch": 3.115081787109375e-05,
      "model_forward_time": 2.4331767559051514,
      "step": 8166
    },
    {
      "epoch": 3.115081787109375e-05,
      "step": 8166,
      "training_step_time": 10.027115821838379
    },
    {
      "epoch": 3.1154632568359374e-05,
      "model_forward_time": 2.7177388668060303,
      "step": 8167
    },
    {
      "epoch": 3.1154632568359374e-05,
      "step": 8167,
      "training_step_time": 7.7556493282318115
    },
    {
      "epoch": 3.1154632568359374e-05,
      "model_forward_time": 2.4339799880981445,
      "step": 8167
    },
    {
      "epoch": 3.1154632568359374e-05,
      "step": 8167,
      "training_step_time": 7.463164329528809
    },
    {
      "epoch": 3.1154632568359374e-05,
      "model_forward_time": 2.440217971801758,
      "step": 8167
    },
    {
      "dynamics_loss_avg": 0.046325675770640375,
      "epoch": 3.1154632568359374e-05,
      "step": 8167
    },
    {
      "action_loss_avg": 0.01742244476918131,
      "epoch": 3.1154632568359374e-05,
      "step": 8167
    },
    {
      "epoch": 3.1154632568359374e-05,
      "step": 8167,
      "training_step_time": 7.476340055465698
    },
    {
      "epoch": 3.1154632568359374e-05,
      "model_forward_time": 3.13818359375,
      "step": 8167
    },
    {
      "epoch": 3.1154632568359374e-05,
      "step": 8167,
      "training_step_time": 10.257318019866943
    },
    {
      "epoch": 3.1158447265625e-05,
      "model_forward_time": 3.153059244155884,
      "step": 8168
    },
    {
      "epoch": 3.1158447265625e-05,
      "step": 8168,
      "training_step_time": 9.927776098251343
    },
    {
      "epoch": 3.1158447265625e-05,
      "model_forward_time": 3.1370744705200195,
      "step": 8168
    },
    {
      "epoch": 3.1158447265625e-05,
      "step": 8168,
      "training_step_time": 9.911252737045288
    },
    {
      "epoch": 3.1158447265625e-05,
      "model_forward_time": 3.4020228385925293,
      "step": 8168
    },
    {
      "epoch": 3.1158447265625e-05,
      "step": 8168,
      "training_step_time": 10.17542815208435
    },
    {
      "epoch": 3.1158447265625e-05,
      "model_forward_time": 3.1361727714538574,
      "step": 8168
    },
    {
      "epoch": 3.1158447265625e-05,
      "step": 8168,
      "training_step_time": 10.009444952011108
    },
    {
      "epoch": 3.116226196289063e-05,
      "model_forward_time": 3.153007745742798,
      "step": 8169
    },
    {
      "epoch": 3.116226196289063e-05,
      "step": 8169,
      "training_step_time": 9.93666672706604
    },
    {
      "epoch": 3.116226196289063e-05,
      "model_forward_time": 2.433673858642578,
      "step": 8169
    },
    {
      "epoch": 3.116226196289063e-05,
      "step": 8169,
      "training_step_time": 7.465999126434326
    },
    {
      "epoch": 3.116226196289063e-05,
      "model_forward_time": 2.434083938598633,
      "step": 8169
    },
    {
      "epoch": 3.116226196289063e-05,
      "step": 8169,
      "training_step_time": 7.4649035930633545
    },
    {
      "epoch": 3.116226196289063e-05,
      "model_forward_time": 2.7357029914855957,
      "step": 8169
    },
    {
      "epoch": 3.116226196289063e-05,
      "step": 8169,
      "training_step_time": 10.3187415599823
    },
    {
      "epoch": 3.116607666015625e-05,
      "grad_norm": 0.7223096489906311,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 8170
    },
    {
      "epoch": 3.116607666015625e-05,
      "model_forward_time": 2.4504292011260986,
      "step": 8170
    },
    {
      "dynamics_loss_avg": 0.05779455937445164,
      "epoch": 3.116607666015625e-05,
      "step": 8170
    },
    {
      "action_loss_avg": 0.0201425036881119,
      "epoch": 3.116607666015625e-05,
      "step": 8170
    },
    {
      "epoch": 3.116607666015625e-05,
      "step": 8170,
      "training_step_time": 7.485302209854126
    },
    {
      "epoch": 3.116607666015625e-05,
      "model_forward_time": 3.136498212814331,
      "step": 8170
    },
    {
      "epoch": 3.116607666015625e-05,
      "step": 8170,
      "training_step_time": 9.915968894958496
    },
    {
      "epoch": 3.116607666015625e-05,
      "model_forward_time": 2.4349255561828613,
      "step": 8170
    },
    {
      "epoch": 3.116607666015625e-05,
      "step": 8170,
      "training_step_time": 7.472480297088623
    },
    {
      "epoch": 3.116607666015625e-05,
      "model_forward_time": 2.4395573139190674,
      "step": 8170
    },
    {
      "epoch": 3.116607666015625e-05,
      "step": 8170,
      "training_step_time": 10.275893688201904
    },
    {
      "epoch": 3.116989135742187e-05,
      "model_forward_time": 3.4341371059417725,
      "step": 8171
    },
    {
      "epoch": 3.116989135742187e-05,
      "step": 8171,
      "training_step_time": 10.210624694824219
    },
    {
      "epoch": 3.116989135742187e-05,
      "model_forward_time": 2.433582067489624,
      "step": 8171
    },
    {
      "epoch": 3.116989135742187e-05,
      "step": 8171,
      "training_step_time": 7.463318347930908
    },
    {
      "epoch": 3.116989135742187e-05,
      "model_forward_time": 3.1399588584899902,
      "step": 8171
    },
    {
      "epoch": 3.116989135742187e-05,
      "step": 8171,
      "training_step_time": 9.907582998275757
    },
    {
      "epoch": 3.116989135742187e-05,
      "model_forward_time": 3.1371116638183594,
      "step": 8171
    },
    {
      "epoch": 3.116989135742187e-05,
      "step": 8171,
      "training_step_time": 10.008958339691162
    },
    {
      "epoch": 3.11737060546875e-05,
      "model_forward_time": 3.1490538120269775,
      "step": 8172
    },
    {
      "epoch": 3.11737060546875e-05,
      "step": 8172,
      "training_step_time": 9.927176475524902
    },
    {
      "epoch": 3.11737060546875e-05,
      "model_forward_time": 2.4299960136413574,
      "step": 8172
    },
    {
      "epoch": 3.11737060546875e-05,
      "step": 8172,
      "training_step_time": 7.74014687538147
    },
    {
      "epoch": 3.11737060546875e-05,
      "model_forward_time": 2.434713125228882,
      "step": 8172
    },
    {
      "dynamics_loss_avg": 0.04926853058859706,
      "epoch": 3.11737060546875e-05,
      "step": 8172
    },
    {
      "action_loss_avg": 0.017102906480431557,
      "epoch": 3.11737060546875e-05,
      "step": 8172
    },
    {
      "epoch": 3.11737060546875e-05,
      "step": 8172,
      "training_step_time": 7.470116853713989
    },
    {
      "epoch": 3.11737060546875e-05,
      "model_forward_time": 3.1368956565856934,
      "step": 8172
    },
    {
      "epoch": 3.11737060546875e-05,
      "step": 8172,
      "training_step_time": 10.020341157913208
    },
    {
      "epoch": 3.1177520751953125e-05,
      "model_forward_time": 3.149400472640991,
      "step": 8173
    },
    {
      "epoch": 3.1177520751953125e-05,
      "step": 8173,
      "training_step_time": 9.932977437973022
    },
    {
      "epoch": 3.1177520751953125e-05,
      "model_forward_time": 2.4361867904663086,
      "step": 8173
    },
    {
      "epoch": 3.1177520751953125e-05,
      "step": 8173,
      "training_step_time": 7.474453687667847
    },
    {
      "epoch": 3.1177520751953125e-05,
      "model_forward_time": 3.1376919746398926,
      "step": 8173
    },
    {
      "epoch": 3.1177520751953125e-05,
      "step": 8173,
      "training_step_time": 9.91304636001587
    },
    {
      "epoch": 3.1177520751953125e-05,
      "model_forward_time": 3.3918819427490234,
      "step": 8173
    },
    {
      "epoch": 3.1177520751953125e-05,
      "step": 8173,
      "training_step_time": 10.266390562057495
    },
    {
      "epoch": 3.118133544921875e-05,
      "model_forward_time": 2.4449195861816406,
      "step": 8174
    },
    {
      "epoch": 3.118133544921875e-05,
      "step": 8174,
      "training_step_time": 7.480517625808716
    },
    {
      "epoch": 3.118133544921875e-05,
      "model_forward_time": 2.435307264328003,
      "step": 8174
    },
    {
      "epoch": 3.118133544921875e-05,
      "step": 8174,
      "training_step_time": 7.466752767562866
    },
    {
      "epoch": 3.118133544921875e-05,
      "model_forward_time": 2.428194284439087,
      "step": 8174
    },
    {
      "epoch": 3.118133544921875e-05,
      "step": 8174,
      "training_step_time": 7.462593078613281
    },
    {
      "epoch": 3.118133544921875e-05,
      "model_forward_time": 3.134782075881958,
      "step": 8174
    },
    {
      "epoch": 3.118133544921875e-05,
      "step": 8174,
      "training_step_time": 10.263116598129272
    },
    {
      "epoch": 3.118515014648438e-05,
      "model_forward_time": 3.4082767963409424,
      "step": 8175
    },
    {
      "dynamics_loss_avg": 0.04978569950908422,
      "epoch": 3.118515014648438e-05,
      "step": 8175
    },
    {
      "action_loss_avg": 0.0178950734436512,
      "epoch": 3.118515014648438e-05,
      "step": 8175
    },
    {
      "epoch": 3.118515014648438e-05,
      "step": 8175,
      "training_step_time": 10.194936752319336
    },
    {
      "epoch": 3.118515014648438e-05,
      "model_forward_time": 3.1469309329986572,
      "step": 8175
    },
    {
      "epoch": 3.118515014648438e-05,
      "step": 8175,
      "training_step_time": 9.936182975769043
    },
    {
      "epoch": 3.118515014648438e-05,
      "model_forward_time": 3.1482021808624268,
      "step": 8175
    },
    {
      "epoch": 3.118515014648438e-05,
      "step": 8175,
      "training_step_time": 9.936849117279053
    },
    {
      "epoch": 3.118515014648438e-05,
      "model_forward_time": 3.1525611877441406,
      "step": 8175
    },
    {
      "epoch": 3.118515014648438e-05,
      "step": 8175,
      "training_step_time": 10.282069683074951
    },
    {
      "epoch": 3.118896484375e-05,
      "model_forward_time": 2.459266185760498,
      "step": 8176
    },
    {
      "epoch": 3.118896484375e-05,
      "step": 8176,
      "training_step_time": 7.501999139785767
    },
    {
      "epoch": 3.118896484375e-05,
      "model_forward_time": 3.4237165451049805,
      "step": 8176
    },
    {
      "epoch": 3.118896484375e-05,
      "step": 8176,
      "training_step_time": 10.206470489501953
    },
    {
      "epoch": 3.118896484375e-05,
      "model_forward_time": 3.1424710750579834,
      "step": 8176
    },
    {
      "epoch": 3.118896484375e-05,
      "step": 8176,
      "training_step_time": 9.922865867614746
    },
    {
      "epoch": 3.118896484375e-05,
      "model_forward_time": 2.436760187149048,
      "step": 8176
    },
    {
      "epoch": 3.118896484375e-05,
      "step": 8176,
      "training_step_time": 10.28108286857605
    },
    {
      "epoch": 3.1192779541015623e-05,
      "model_forward_time": 3.15480637550354,
      "step": 8177
    },
    {
      "epoch": 3.1192779541015623e-05,
      "step": 8177,
      "training_step_time": 9.938090324401855
    },
    {
      "epoch": 3.1192779541015623e-05,
      "model_forward_time": 3.138838052749634,
      "step": 8177
    },
    {
      "epoch": 3.1192779541015623e-05,
      "step": 8177,
      "training_step_time": 9.9207284450531
    },
    {
      "epoch": 3.1192779541015623e-05,
      "model_forward_time": 3.410645008087158,
      "step": 8177
    },
    {
      "dynamics_loss_avg": 0.04545202031731606,
      "epoch": 3.1192779541015623e-05,
      "step": 8177
    },
    {
      "action_loss_avg": 0.023562729032710193,
      "epoch": 3.1192779541015623e-05,
      "step": 8177
    },
    {
      "epoch": 3.1192779541015623e-05,
      "step": 8177,
      "training_step_time": 10.197532176971436
    },
    {
      "epoch": 3.1192779541015623e-05,
      "model_forward_time": 2.444289207458496,
      "step": 8177
    },
    {
      "epoch": 3.1192779541015623e-05,
      "step": 8177,
      "training_step_time": 10.011163711547852
    },
    {
      "epoch": 3.119659423828125e-05,
      "model_forward_time": 2.448162078857422,
      "step": 8178
    },
    {
      "epoch": 3.119659423828125e-05,
      "step": 8178,
      "training_step_time": 7.480889081954956
    },
    {
      "epoch": 3.119659423828125e-05,
      "model_forward_time": 1.744795560836792,
      "step": 8178
    },
    {
      "epoch": 3.119659423828125e-05,
      "step": 8178,
      "training_step_time": 5.209970235824585
    },
    {
      "epoch": 3.119659423828125e-05,
      "model_forward_time": 3.1327829360961914,
      "step": 8178
    },
    {
      "epoch": 3.119659423828125e-05,
      "step": 8178,
      "training_step_time": 9.900210618972778
    },
    {
      "epoch": 3.119659423828125e-05,
      "model_forward_time": 3.4438750743865967,
      "step": 8178
    },
    {
      "epoch": 3.119659423828125e-05,
      "step": 8178,
      "training_step_time": 10.318529605865479
    },
    {
      "epoch": 3.1200408935546876e-05,
      "model_forward_time": 3.1528513431549072,
      "step": 8179
    },
    {
      "epoch": 3.1200408935546876e-05,
      "step": 8179,
      "training_step_time": 9.934032201766968
    },
    {
      "epoch": 3.1200408935546876e-05,
      "model_forward_time": 2.436147451400757,
      "step": 8179
    },
    {
      "epoch": 3.1200408935546876e-05,
      "step": 8179,
      "training_step_time": 7.473764657974243
    },
    {
      "epoch": 3.1200408935546876e-05,
      "model_forward_time": 3.1405601501464844,
      "step": 8179
    },
    {
      "epoch": 3.1200408935546876e-05,
      "step": 8179,
      "training_step_time": 9.922683954238892
    },
    {
      "epoch": 3.1200408935546876e-05,
      "model_forward_time": 2.436422109603882,
      "step": 8179
    },
    {
      "epoch": 3.1200408935546876e-05,
      "step": 8179,
      "training_step_time": 10.261462450027466
    },
    {
      "epoch": 3.12042236328125e-05,
      "grad_norm": 0.4005160629749298,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 8180
    },
    {
      "epoch": 3.12042236328125e-05,
      "model_forward_time": 3.1469085216522217,
      "step": 8180
    },
    {
      "dynamics_loss_avg": 0.04449825696647167,
      "epoch": 3.12042236328125e-05,
      "step": 8180
    },
    {
      "action_loss_avg": 0.017256633820943536,
      "epoch": 3.12042236328125e-05,
      "step": 8180
    },
    {
      "epoch": 3.12042236328125e-05,
      "step": 8180,
      "training_step_time": 9.934852361679077
    },
    {
      "epoch": 3.12042236328125e-05,
      "model_forward_time": 3.4129440784454346,
      "step": 8180
    },
    {
      "epoch": 3.12042236328125e-05,
      "step": 8180,
      "training_step_time": 10.196446180343628
    },
    {
      "epoch": 3.12042236328125e-05,
      "model_forward_time": 3.141402006149292,
      "step": 8180
    },
    {
      "epoch": 3.12042236328125e-05,
      "step": 8180,
      "training_step_time": 9.923736810684204
    },
    {
      "epoch": 3.12042236328125e-05,
      "model_forward_time": 2.4342334270477295,
      "step": 8180
    },
    {
      "epoch": 3.12042236328125e-05,
      "step": 8180,
      "training_step_time": 10.291687250137329
    },
    {
      "epoch": 3.120803833007812e-05,
      "model_forward_time": 2.445591926574707,
      "step": 8181
    },
    {
      "epoch": 3.120803833007812e-05,
      "step": 8181,
      "training_step_time": 7.4836249351501465
    },
    {
      "epoch": 3.120803833007812e-05,
      "model_forward_time": 3.136362075805664,
      "step": 8181
    },
    {
      "epoch": 3.120803833007812e-05,
      "step": 8181,
      "training_step_time": 10.18385362625122
    },
    {
      "epoch": 3.120803833007812e-05,
      "model_forward_time": 3.1386301517486572,
      "step": 8181
    },
    {
      "epoch": 3.120803833007812e-05,
      "step": 8181,
      "training_step_time": 9.923122882843018
    },
    {
      "epoch": 3.120803833007812e-05,
      "model_forward_time": 2.4375317096710205,
      "step": 8181
    },
    {
      "epoch": 3.120803833007812e-05,
      "step": 8181,
      "training_step_time": 10.01648211479187
    },
    {
      "epoch": 3.121185302734375e-05,
      "model_forward_time": 3.158261299133301,
      "step": 8182
    },
    {
      "epoch": 3.121185302734375e-05,
      "step": 8182,
      "training_step_time": 9.933892726898193
    },
    {
      "epoch": 3.121185302734375e-05,
      "model_forward_time": 2.4247684478759766,
      "step": 8182
    },
    {
      "epoch": 3.121185302734375e-05,
      "step": 8182,
      "training_step_time": 7.456110000610352
    },
    {
      "epoch": 3.121185302734375e-05,
      "model_forward_time": 2.4249374866485596,
      "step": 8182
    },
    {
      "dynamics_loss_avg": 0.05929566156119108,
      "epoch": 3.121185302734375e-05,
      "step": 8182
    },
    {
      "action_loss_avg": 0.023746745940297843,
      "epoch": 3.121185302734375e-05,
      "step": 8182
    },
    {
      "epoch": 3.121185302734375e-05,
      "step": 8182,
      "training_step_time": 7.456629514694214
    },
    {
      "epoch": 3.121185302734375e-05,
      "model_forward_time": 3.369619846343994,
      "step": 8182
    },
    {
      "epoch": 3.121185302734375e-05,
      "step": 8182,
      "training_step_time": 10.234832286834717
    },
    {
      "epoch": 3.1215667724609374e-05,
      "model_forward_time": 2.4471418857574463,
      "step": 8183
    },
    {
      "epoch": 3.1215667724609374e-05,
      "step": 8183,
      "training_step_time": 7.480875015258789
    },
    {
      "epoch": 3.1215667724609374e-05,
      "model_forward_time": 2.434171438217163,
      "step": 8183
    },
    {
      "epoch": 3.1215667724609374e-05,
      "step": 8183,
      "training_step_time": 7.462383985519409
    },
    {
      "epoch": 3.1215667724609374e-05,
      "model_forward_time": 2.4276578426361084,
      "step": 8183
    },
    {
      "epoch": 3.1215667724609374e-05,
      "step": 8183,
      "training_step_time": 7.45612096786499
    },
    {
      "epoch": 3.1215667724609374e-05,
      "model_forward_time": 3.1287403106689453,
      "step": 8183
    },
    {
      "epoch": 3.1215667724609374e-05,
      "step": 8183,
      "training_step_time": 10.25854206085205
    },
    {
      "epoch": 3.1219482421875e-05,
      "model_forward_time": 2.7528975009918213,
      "step": 8184
    },
    {
      "epoch": 3.1219482421875e-05,
      "step": 8184,
      "training_step_time": 7.7828192710876465
    },
    {
      "epoch": 3.1219482421875e-05,
      "model_forward_time": 2.4309885501861572,
      "step": 8184
    },
    {
      "epoch": 3.1219482421875e-05,
      "step": 8184,
      "training_step_time": 7.465877056121826
    },
    {
      "epoch": 3.1219482421875e-05,
      "model_forward_time": 2.4343628883361816,
      "step": 8184
    },
    {
      "epoch": 3.1219482421875e-05,
      "step": 8184,
      "training_step_time": 7.468181848526001
    },
    {
      "epoch": 3.1219482421875e-05,
      "model_forward_time": 2.4289045333862305,
      "step": 8184
    },
    {
      "epoch": 3.1219482421875e-05,
      "step": 8184,
      "training_step_time": 10.31125521659851
    },
    {
      "epoch": 3.122329711914063e-05,
      "model_forward_time": 2.4472200870513916,
      "step": 8185
    },
    {
      "dynamics_loss_avg": 0.057022713124752045,
      "epoch": 3.122329711914063e-05,
      "step": 8185
    },
    {
      "action_loss_avg": 0.020425737649202347,
      "epoch": 3.122329711914063e-05,
      "step": 8185
    },
    {
      "epoch": 3.122329711914063e-05,
      "step": 8185,
      "training_step_time": 7.48608136177063
    },
    {
      "epoch": 3.122329711914063e-05,
      "model_forward_time": 2.6872873306274414,
      "step": 8185
    },
    {
      "epoch": 3.122329711914063e-05,
      "step": 8185,
      "training_step_time": 7.721029758453369
    },
    {
      "epoch": 3.122329711914063e-05,
      "model_forward_time": 2.4307053089141846,
      "step": 8185
    },
    {
      "epoch": 3.122329711914063e-05,
      "step": 8185,
      "training_step_time": 7.458438873291016
    },
    {
      "epoch": 3.122329711914063e-05,
      "model_forward_time": 2.430255651473999,
      "step": 8185
    },
    {
      "epoch": 3.122329711914063e-05,
      "step": 8185,
      "training_step_time": 10.269054651260376
    },
    {
      "epoch": 3.122711181640625e-05,
      "model_forward_time": 2.445488452911377,
      "step": 8186
    },
    {
      "epoch": 3.122711181640625e-05,
      "step": 8186,
      "training_step_time": 7.473551273345947
    },
    {
      "epoch": 3.122711181640625e-05,
      "model_forward_time": 2.4248173236846924,
      "step": 8186
    },
    {
      "epoch": 3.122711181640625e-05,
      "step": 8186,
      "training_step_time": 7.451321125030518
    },
    {
      "epoch": 3.122711181640625e-05,
      "model_forward_time": 3.136685848236084,
      "step": 8186
    },
    {
      "epoch": 3.122711181640625e-05,
      "step": 8186,
      "training_step_time": 9.913621425628662
    },
    {
      "epoch": 3.122711181640625e-05,
      "model_forward_time": 2.7068724632263184,
      "step": 8186
    },
    {
      "epoch": 3.122711181640625e-05,
      "step": 8186,
      "training_step_time": 10.269879817962646
    },
    {
      "epoch": 3.123092651367187e-05,
      "model_forward_time": 3.1473617553710938,
      "step": 8187
    },
    {
      "epoch": 3.123092651367187e-05,
      "step": 8187,
      "training_step_time": 9.929426193237305
    },
    {
      "epoch": 3.123092651367187e-05,
      "model_forward_time": 3.1325762271881104,
      "step": 8187
    },
    {
      "epoch": 3.123092651367187e-05,
      "step": 8187,
      "training_step_time": 9.913859844207764
    },
    {
      "epoch": 3.123092651367187e-05,
      "model_forward_time": 2.4307217597961426,
      "step": 8187
    },
    {
      "dynamics_loss_avg": 0.050118494965136054,
      "epoch": 3.123092651367187e-05,
      "step": 8187
    },
    {
      "action_loss_avg": 0.02228522431105375,
      "epoch": 3.123092651367187e-05,
      "step": 8187
    },
    {
      "epoch": 3.123092651367187e-05,
      "step": 8187,
      "training_step_time": 7.460722208023071
    },
    {
      "epoch": 3.123092651367187e-05,
      "model_forward_time": 3.1278305053710938,
      "step": 8187
    },
    {
      "epoch": 3.123092651367187e-05,
      "step": 8187,
      "training_step_time": 10.007925033569336
    },
    {
      "epoch": 3.12347412109375e-05,
      "model_forward_time": 2.740861654281616,
      "step": 8188
    },
    {
      "epoch": 3.12347412109375e-05,
      "step": 8188,
      "training_step_time": 7.766514301300049
    },
    {
      "epoch": 3.12347412109375e-05,
      "model_forward_time": 3.1290283203125,
      "step": 8188
    },
    {
      "epoch": 3.12347412109375e-05,
      "step": 8188,
      "training_step_time": 9.895219802856445
    },
    {
      "epoch": 3.12347412109375e-05,
      "model_forward_time": 3.1310055255889893,
      "step": 8188
    },
    {
      "epoch": 3.12347412109375e-05,
      "step": 8188,
      "training_step_time": 9.897578239440918
    },
    {
      "epoch": 3.12347412109375e-05,
      "model_forward_time": 3.1334521770477295,
      "step": 8188
    },
    {
      "epoch": 3.12347412109375e-05,
      "step": 8188,
      "training_step_time": 10.269582271575928
    },
    {
      "epoch": 3.1238555908203125e-05,
      "model_forward_time": 2.440131187438965,
      "step": 8189
    },
    {
      "epoch": 3.1238555908203125e-05,
      "step": 8189,
      "training_step_time": 7.4653637409210205
    },
    {
      "epoch": 3.1238555908203125e-05,
      "model_forward_time": 3.39371919631958,
      "step": 8189
    },
    {
      "epoch": 3.1238555908203125e-05,
      "step": 8189,
      "training_step_time": 10.158636093139648
    },
    {
      "epoch": 3.1238555908203125e-05,
      "model_forward_time": 3.137308120727539,
      "step": 8189
    },
    {
      "epoch": 3.1238555908203125e-05,
      "step": 8189,
      "training_step_time": 9.901943922042847
    },
    {
      "epoch": 3.1238555908203125e-05,
      "model_forward_time": 3.127756357192993,
      "step": 8189
    },
    {
      "epoch": 3.1238555908203125e-05,
      "step": 8189,
      "training_step_time": 10.23488712310791
    },
    {
      "epoch": 3.124237060546875e-05,
      "grad_norm": 0.37703248858451843,
      "learning_rate": 1e-05,
      "loss": 0.0846,
      "step": 8190
    },
    {
      "epoch": 3.124237060546875e-05,
      "model_forward_time": 3.14784836769104,
      "step": 8190
    },
    {
      "dynamics_loss_avg": 0.06659327652305365,
      "epoch": 3.124237060546875e-05,
      "step": 8190
    },
    {
      "action_loss_avg": 0.026583975972607733,
      "epoch": 3.124237060546875e-05,
      "step": 8190
    },
    {
      "epoch": 3.124237060546875e-05,
      "step": 8190,
      "training_step_time": 9.921899557113647
    },
    {
      "epoch": 3.124237060546875e-05,
      "model_forward_time": 2.431443452835083,
      "step": 8190
    },
    {
      "epoch": 3.124237060546875e-05,
      "step": 8190,
      "training_step_time": 7.457968473434448
    },
    {
      "epoch": 3.124237060546875e-05,
      "model_forward_time": 3.408506155014038,
      "step": 8190
    },
    {
      "epoch": 3.124237060546875e-05,
      "step": 8190,
      "training_step_time": 10.176496028900146
    },
    {
      "epoch": 3.124237060546875e-05,
      "model_forward_time": 3.1303906440734863,
      "step": 8190
    },
    {
      "epoch": 3.124237060546875e-05,
      "step": 8190,
      "training_step_time": 10.263453483581543
    },
    {
      "epoch": 3.124618530273438e-05,
      "model_forward_time": 2.462644338607788,
      "step": 8191
    },
    {
      "epoch": 3.124618530273438e-05,
      "step": 8191,
      "training_step_time": 7.4904584884643555
    },
    {
      "epoch": 3.124618530273438e-05,
      "model_forward_time": 2.4370908737182617,
      "step": 8191
    },
    {
      "epoch": 3.124618530273438e-05,
      "step": 8191,
      "training_step_time": 7.4616100788116455
    },
    {
      "epoch": 3.124618530273438e-05,
      "model_forward_time": 3.1389522552490234,
      "step": 8191
    },
    {
      "epoch": 3.124618530273438e-05,
      "step": 8191,
      "training_step_time": 9.911468982696533
    },
    {
      "epoch": 3.124618530273438e-05,
      "model_forward_time": 2.4435834884643555,
      "step": 8191
    },
    {
      "epoch": 3.124618530273438e-05,
      "step": 8191,
      "training_step_time": 10.30577039718628
    },
    {
      "epoch": 3.125e-05,
      "model_forward_time": 3.404946804046631,
      "step": 8192
    },
    {
      "epoch": 3.125e-05,
      "step": 8192,
      "training_step_time": 10.181336164474487
    },
    {
      "epoch": 3.125e-05,
      "model_forward_time": 3.1462671756744385,
      "step": 8192
    },
    {
      "epoch": 3.125e-05,
      "step": 8192,
      "training_step_time": 9.923929929733276
    },
    {
      "epoch": 3.125e-05,
      "model_forward_time": 3.1475069522857666,
      "step": 8192
    },
    {
      "dynamics_loss_avg": 0.04962912704795599,
      "epoch": 3.125e-05,
      "step": 8192
    },
    {
      "action_loss_avg": 0.022052776627242567,
      "epoch": 3.125e-05,
      "step": 8192
    },
    {
      "epoch": 3.125e-05,
      "step": 8192,
      "training_step_time": 9.929367542266846
    },
    {
      "epoch": 3.125e-05,
      "model_forward_time": 3.147672653198242,
      "step": 8192
    },
    {
      "epoch": 3.125e-05,
      "step": 8192,
      "training_step_time": 10.02073073387146
    },
    {
      "epoch": 3.1253814697265624e-05,
      "model_forward_time": 3.167649745941162,
      "step": 8193
    },
    {
      "epoch": 3.1253814697265624e-05,
      "step": 8193,
      "training_step_time": 10.216870069503784
    },
    {
      "epoch": 3.1253814697265624e-05,
      "model_forward_time": 2.43803071975708,
      "step": 8193
    },
    {
      "epoch": 3.1253814697265624e-05,
      "step": 8193,
      "training_step_time": 7.468006372451782
    },
    {
      "epoch": 3.1253814697265624e-05,
      "model_forward_time": 2.439802885055542,
      "step": 8193
    },
    {
      "epoch": 3.1253814697265624e-05,
      "step": 8193,
      "training_step_time": 7.467425107955933
    },
    {
      "epoch": 3.1253814697265624e-05,
      "model_forward_time": 3.139098882675171,
      "step": 8193
    },
    {
      "epoch": 3.1253814697265624e-05,
      "step": 8193,
      "training_step_time": 10.000263929367065
    },
    {
      "epoch": 3.125762939453125e-05,
      "model_forward_time": 2.4559576511383057,
      "step": 8194
    },
    {
      "epoch": 3.125762939453125e-05,
      "step": 8194,
      "training_step_time": 7.478576898574829
    },
    {
      "epoch": 3.125762939453125e-05,
      "model_forward_time": 2.432888984680176,
      "step": 8194
    },
    {
      "epoch": 3.125762939453125e-05,
      "step": 8194,
      "training_step_time": 7.453674793243408
    },
    {
      "epoch": 3.125762939453125e-05,
      "model_forward_time": 2.685434341430664,
      "step": 8194
    },
    {
      "epoch": 3.125762939453125e-05,
      "step": 8194,
      "training_step_time": 7.705586910247803
    },
    {
      "epoch": 3.125762939453125e-05,
      "model_forward_time": 1.7433476448059082,
      "step": 8194
    },
    {
      "epoch": 3.125762939453125e-05,
      "step": 8194,
      "training_step_time": 10.258914947509766
    },
    {
      "epoch": 3.1261444091796876e-05,
      "model_forward_time": 3.1477088928222656,
      "step": 8195
    },
    {
      "dynamics_loss_avg": 0.050598450005054474,
      "epoch": 3.1261444091796876e-05,
      "step": 8195
    },
    {
      "action_loss_avg": 0.02280821409076452,
      "epoch": 3.1261444091796876e-05,
      "step": 8195
    },
    {
      "epoch": 3.1261444091796876e-05,
      "step": 8195,
      "training_step_time": 9.911495447158813
    },
    {
      "epoch": 3.1261444091796876e-05,
      "model_forward_time": 2.436115264892578,
      "step": 8195
    },
    {
      "epoch": 3.1261444091796876e-05,
      "step": 8195,
      "training_step_time": 7.4538280963897705
    },
    {
      "epoch": 3.1261444091796876e-05,
      "model_forward_time": 3.1351096630096436,
      "step": 8195
    },
    {
      "epoch": 3.1261444091796876e-05,
      "step": 8195,
      "training_step_time": 9.90101408958435
    },
    {
      "epoch": 3.1261444091796876e-05,
      "model_forward_time": 2.432589530944824,
      "step": 8195
    },
    {
      "epoch": 3.1261444091796876e-05,
      "step": 8195,
      "training_step_time": 9.99666166305542
    },
    {
      "epoch": 3.12652587890625e-05,
      "model_forward_time": 2.769597291946411,
      "step": 8196
    },
    {
      "epoch": 3.12652587890625e-05,
      "step": 8196,
      "training_step_time": 7.792181968688965
    },
    {
      "epoch": 3.12652587890625e-05,
      "model_forward_time": 3.135084867477417,
      "step": 8196
    },
    {
      "epoch": 3.12652587890625e-05,
      "step": 8196,
      "training_step_time": 9.912838697433472
    },
    {
      "epoch": 3.12652587890625e-05,
      "model_forward_time": 3.1391375064849854,
      "step": 8196
    },
    {
      "epoch": 3.12652587890625e-05,
      "step": 8196,
      "training_step_time": 9.911281824111938
    },
    {
      "epoch": 3.12652587890625e-05,
      "model_forward_time": 2.439429521560669,
      "step": 8196
    },
    {
      "epoch": 3.12652587890625e-05,
      "step": 8196,
      "training_step_time": 10.262574911117554
    },
    {
      "epoch": 3.126907348632812e-05,
      "model_forward_time": 3.158390760421753,
      "step": 8197
    },
    {
      "epoch": 3.126907348632812e-05,
      "step": 8197,
      "training_step_time": 9.924040079116821
    },
    {
      "epoch": 3.126907348632812e-05,
      "model_forward_time": 3.4098458290100098,
      "step": 8197
    },
    {
      "epoch": 3.126907348632812e-05,
      "step": 8197,
      "training_step_time": 10.179633140563965
    },
    {
      "epoch": 3.126907348632812e-05,
      "model_forward_time": 3.1469430923461914,
      "step": 8197
    },
    {
      "dynamics_loss_avg": 0.0663997920230031,
      "epoch": 3.126907348632812e-05,
      "step": 8197
    },
    {
      "action_loss_avg": 0.021693374821916224,
      "epoch": 3.126907348632812e-05,
      "step": 8197
    },
    {
      "epoch": 3.126907348632812e-05,
      "step": 8197,
      "training_step_time": 9.917431354522705
    },
    {
      "epoch": 3.126907348632812e-05,
      "model_forward_time": 2.4383037090301514,
      "step": 8197
    },
    {
      "epoch": 3.126907348632812e-05,
      "step": 8197,
      "training_step_time": 10.277348756790161
    },
    {
      "epoch": 3.127288818359375e-05,
      "model_forward_time": 2.4453024864196777,
      "step": 8198
    },
    {
      "epoch": 3.127288818359375e-05,
      "step": 8198,
      "training_step_time": 7.472512722015381
    },
    {
      "epoch": 3.127288818359375e-05,
      "model_forward_time": 3.139686107635498,
      "step": 8198
    },
    {
      "epoch": 3.127288818359375e-05,
      "step": 8198,
      "training_step_time": 9.902690172195435
    },
    {
      "epoch": 3.127288818359375e-05,
      "model_forward_time": 3.4217207431793213,
      "step": 8198
    },
    {
      "epoch": 3.127288818359375e-05,
      "step": 8198,
      "training_step_time": 10.184433221817017
    },
    {
      "epoch": 3.127288818359375e-05,
      "model_forward_time": 3.141411542892456,
      "step": 8198
    },
    {
      "epoch": 3.127288818359375e-05,
      "step": 8198,
      "training_step_time": 10.00913381576538
    },
    {
      "epoch": 3.1276702880859374e-05,
      "model_forward_time": 3.144005298614502,
      "step": 8199
    },
    {
      "epoch": 3.1276702880859374e-05,
      "step": 8199,
      "training_step_time": 9.904791593551636
    },
    {
      "epoch": 3.1276702880859374e-05,
      "model_forward_time": 1.747377634048462,
      "step": 8199
    },
    {
      "epoch": 3.1276702880859374e-05,
      "step": 8199,
      "training_step_time": 5.198465824127197
    },
    {
      "epoch": 3.1276702880859374e-05,
      "model_forward_time": 3.1333863735198975,
      "step": 8199
    },
    {
      "epoch": 3.1276702880859374e-05,
      "step": 8199,
      "training_step_time": 9.893929719924927
    },
    {
      "epoch": 3.1276702880859374e-05,
      "model_forward_time": 3.390475273132324,
      "step": 8199
    },
    {
      "epoch": 3.1276702880859374e-05,
      "step": 8199,
      "training_step_time": 10.246479988098145
    },
    {
      "epoch": 3.1280517578125e-05,
      "grad_norm": 0.3606869578361511,
      "learning_rate": 1e-05,
      "loss": 0.0806,
      "step": 8200
    },
    {
      "epoch": 3.1280517578125e-05,
      "model_forward_time": 2.4525883197784424,
      "step": 8200
    },
    {
      "dynamics_loss_avg": 0.06831643152981996,
      "epoch": 3.1280517578125e-05,
      "step": 8200
    },
    {
      "action_loss_avg": 0.028415249101817606,
      "epoch": 3.1280517578125e-05,
      "step": 8200
    },
    {
      "epoch": 3.1280517578125e-05,
      "step": 8200,
      "training_step_time": 7.480215787887573
    },
    {
      "epoch": 3.1280517578125e-05,
      "model_forward_time": 3.1396024227142334,
      "step": 8200
    },
    {
      "epoch": 3.1280517578125e-05,
      "step": 8200,
      "training_step_time": 9.911384344100952
    },
    {
      "epoch": 3.1280517578125e-05,
      "model_forward_time": 2.436307430267334,
      "step": 8200
    },
    {
      "epoch": 3.1280517578125e-05,
      "step": 8200,
      "training_step_time": 7.464622974395752
    },
    {
      "epoch": 3.1280517578125e-05,
      "model_forward_time": 3.1389992237091064,
      "step": 8200
    },
    {
      "epoch": 3.1280517578125e-05,
      "step": 8200,
      "training_step_time": 10.260266780853271
    },
    {
      "epoch": 3.128433227539063e-05,
      "model_forward_time": 2.7397117614746094,
      "step": 8201
    },
    {
      "epoch": 3.128433227539063e-05,
      "step": 8201,
      "training_step_time": 7.7703094482421875
    },
    {
      "epoch": 3.128433227539063e-05,
      "model_forward_time": 3.1427841186523438,
      "step": 8201
    },
    {
      "epoch": 3.128433227539063e-05,
      "step": 8201,
      "training_step_time": 9.91582202911377
    },
    {
      "epoch": 3.128433227539063e-05,
      "model_forward_time": 3.1428732872009277,
      "step": 8201
    },
    {
      "epoch": 3.128433227539063e-05,
      "step": 8201,
      "training_step_time": 9.914622783660889
    },
    {
      "epoch": 3.128433227539063e-05,
      "model_forward_time": 3.1446847915649414,
      "step": 8201
    },
    {
      "epoch": 3.128433227539063e-05,
      "step": 8201,
      "training_step_time": 10.278675317764282
    },
    {
      "epoch": 3.128814697265625e-05,
      "model_forward_time": 3.150672435760498,
      "step": 8202
    },
    {
      "epoch": 3.128814697265625e-05,
      "step": 8202,
      "training_step_time": 9.925255537033081
    },
    {
      "epoch": 3.128814697265625e-05,
      "model_forward_time": 3.4257874488830566,
      "step": 8202
    },
    {
      "epoch": 3.128814697265625e-05,
      "step": 8202,
      "training_step_time": 10.200441360473633
    },
    {
      "epoch": 3.128814697265625e-05,
      "model_forward_time": 3.142038583755493,
      "step": 8202
    },
    {
      "dynamics_loss_avg": 0.07488506715744733,
      "epoch": 3.128814697265625e-05,
      "step": 8202
    },
    {
      "action_loss_avg": 0.028637384064495564,
      "epoch": 3.128814697265625e-05,
      "step": 8202
    },
    {
      "epoch": 3.128814697265625e-05,
      "step": 8202,
      "training_step_time": 9.918029069900513
    },
    {
      "epoch": 3.128814697265625e-05,
      "model_forward_time": 3.147966146469116,
      "step": 8202
    },
    {
      "epoch": 3.128814697265625e-05,
      "step": 8202,
      "training_step_time": 10.289356708526611
    },
    {
      "epoch": 3.129196166992187e-05,
      "model_forward_time": 1.7730944156646729,
      "step": 8203
    },
    {
      "epoch": 3.129196166992187e-05,
      "step": 8203,
      "training_step_time": 5.2367236614227295
    },
    {
      "epoch": 3.129196166992187e-05,
      "model_forward_time": 3.138005256652832,
      "step": 8203
    },
    {
      "epoch": 3.129196166992187e-05,
      "step": 8203,
      "training_step_time": 9.906882762908936
    },
    {
      "epoch": 3.129196166992187e-05,
      "model_forward_time": 2.43648099899292,
      "step": 8203
    },
    {
      "epoch": 3.129196166992187e-05,
      "step": 8203,
      "training_step_time": 7.461432695388794
    },
    {
      "epoch": 3.129196166992187e-05,
      "model_forward_time": 2.708357810974121,
      "step": 8203
    },
    {
      "epoch": 3.129196166992187e-05,
      "step": 8203,
      "training_step_time": 10.015448331832886
    },
    {
      "epoch": 3.12957763671875e-05,
      "model_forward_time": 2.4487717151641846,
      "step": 8204
    },
    {
      "epoch": 3.12957763671875e-05,
      "step": 8204,
      "training_step_time": 7.471330404281616
    },
    {
      "epoch": 3.12957763671875e-05,
      "model_forward_time": 3.137439727783203,
      "step": 8204
    },
    {
      "epoch": 3.12957763671875e-05,
      "step": 8204,
      "training_step_time": 9.906660318374634
    },
    {
      "epoch": 3.12957763671875e-05,
      "model_forward_time": 3.1398322582244873,
      "step": 8204
    },
    {
      "epoch": 3.12957763671875e-05,
      "step": 8204,
      "training_step_time": 9.90272045135498
    },
    {
      "epoch": 3.12957763671875e-05,
      "model_forward_time": 3.1371634006500244,
      "step": 8204
    },
    {
      "epoch": 3.12957763671875e-05,
      "step": 8204,
      "training_step_time": 10.006208896636963
    },
    {
      "epoch": 3.1299591064453125e-05,
      "model_forward_time": 3.454202175140381,
      "step": 8205
    },
    {
      "dynamics_loss_avg": 0.04995736479759216,
      "epoch": 3.1299591064453125e-05,
      "step": 8205
    },
    {
      "action_loss_avg": 0.017448511673137546,
      "epoch": 3.1299591064453125e-05,
      "step": 8205
    },
    {
      "epoch": 3.1299591064453125e-05,
      "step": 8205,
      "training_step_time": 10.226263284683228
    },
    {
      "epoch": 3.1299591064453125e-05,
      "model_forward_time": 3.1379282474517822,
      "step": 8205
    },
    {
      "epoch": 3.1299591064453125e-05,
      "step": 8205,
      "training_step_time": 9.908151626586914
    },
    {
      "epoch": 3.1299591064453125e-05,
      "model_forward_time": 2.4385032653808594,
      "step": 8205
    },
    {
      "epoch": 3.1299591064453125e-05,
      "step": 8205,
      "training_step_time": 7.466490983963013
    },
    {
      "epoch": 3.1299591064453125e-05,
      "model_forward_time": 3.1392762660980225,
      "step": 8205
    },
    {
      "epoch": 3.1299591064453125e-05,
      "step": 8205,
      "training_step_time": 10.279367446899414
    },
    {
      "epoch": 3.130340576171875e-05,
      "model_forward_time": 2.461846113204956,
      "step": 8206
    },
    {
      "epoch": 3.130340576171875e-05,
      "step": 8206,
      "training_step_time": 7.492717981338501
    },
    {
      "epoch": 3.130340576171875e-05,
      "model_forward_time": 2.7296030521392822,
      "step": 8206
    },
    {
      "epoch": 3.130340576171875e-05,
      "step": 8206,
      "training_step_time": 7.757906913757324
    },
    {
      "epoch": 3.130340576171875e-05,
      "model_forward_time": 1.7528681755065918,
      "step": 8206
    },
    {
      "epoch": 3.130340576171875e-05,
      "step": 8206,
      "training_step_time": 5.21200156211853
    },
    {
      "epoch": 3.130340576171875e-05,
      "model_forward_time": 2.4424076080322266,
      "step": 8206
    },
    {
      "epoch": 3.130340576171875e-05,
      "step": 8206,
      "training_step_time": 10.305906772613525
    },
    {
      "epoch": 3.130722045898438e-05,
      "model_forward_time": 2.4510421752929688,
      "step": 8207
    },
    {
      "epoch": 3.130722045898438e-05,
      "step": 8207,
      "training_step_time": 7.476070880889893
    },
    {
      "epoch": 3.130722045898438e-05,
      "model_forward_time": 2.4346537590026855,
      "step": 8207
    },
    {
      "epoch": 3.130722045898438e-05,
      "step": 8207,
      "training_step_time": 7.455427408218384
    },
    {
      "epoch": 3.130722045898438e-05,
      "model_forward_time": 2.4334774017333984,
      "step": 8207
    },
    {
      "dynamics_loss_avg": 0.04755855090916157,
      "epoch": 3.130722045898438e-05,
      "step": 8207
    },
    {
      "action_loss_avg": 0.01274131638929248,
      "epoch": 3.130722045898438e-05,
      "step": 8207
    },
    {
      "epoch": 3.130722045898438e-05,
      "step": 8207,
      "training_step_time": 7.46003532409668
    },
    {
      "epoch": 3.130722045898438e-05,
      "model_forward_time": 2.688530683517456,
      "step": 8207
    },
    {
      "epoch": 3.130722045898438e-05,
      "step": 8207,
      "training_step_time": 10.017047643661499
    },
    {
      "epoch": 3.131103515625e-05,
      "model_forward_time": 3.152434825897217,
      "step": 8208
    },
    {
      "epoch": 3.131103515625e-05,
      "step": 8208,
      "training_step_time": 9.919653415679932
    },
    {
      "epoch": 3.131103515625e-05,
      "model_forward_time": 2.434600353240967,
      "step": 8208
    },
    {
      "epoch": 3.131103515625e-05,
      "step": 8208,
      "training_step_time": 7.45875883102417
    },
    {
      "epoch": 3.131103515625e-05,
      "model_forward_time": 2.4315218925476074,
      "step": 8208
    },
    {
      "epoch": 3.131103515625e-05,
      "step": 8208,
      "training_step_time": 7.454888343811035
    },
    {
      "epoch": 3.131103515625e-05,
      "model_forward_time": 2.4310948848724365,
      "step": 8208
    },
    {
      "epoch": 3.131103515625e-05,
      "step": 8208,
      "training_step_time": 10.00431776046753
    },
    {
      "epoch": 3.1314849853515624e-05,
      "model_forward_time": 3.41282320022583,
      "step": 8209
    },
    {
      "epoch": 3.1314849853515624e-05,
      "step": 8209,
      "training_step_time": 10.17685317993164
    },
    {
      "epoch": 3.1314849853515624e-05,
      "model_forward_time": 3.13969349861145,
      "step": 8209
    },
    {
      "epoch": 3.1314849853515624e-05,
      "step": 8209,
      "training_step_time": 9.902313232421875
    },
    {
      "epoch": 3.1314849853515624e-05,
      "model_forward_time": 2.435762405395508,
      "step": 8209
    },
    {
      "epoch": 3.1314849853515624e-05,
      "step": 8209,
      "training_step_time": 7.463006973266602
    },
    {
      "epoch": 3.1314849853515624e-05,
      "model_forward_time": 3.1408028602600098,
      "step": 8209
    },
    {
      "epoch": 3.1314849853515624e-05,
      "step": 8209,
      "training_step_time": 10.00467586517334
    },
    {
      "epoch": 3.131866455078125e-05,
      "grad_norm": 0.31131380796432495,
      "learning_rate": 1e-05,
      "loss": 0.0813,
      "step": 8210
    },
    {
      "epoch": 3.131866455078125e-05,
      "model_forward_time": 2.457677125930786,
      "step": 8210
    },
    {
      "dynamics_loss_avg": 0.04130065441131592,
      "epoch": 3.131866455078125e-05,
      "step": 8210
    },
    {
      "action_loss_avg": 0.01511262208223343,
      "epoch": 3.131866455078125e-05,
      "step": 8210
    },
    {
      "epoch": 3.131866455078125e-05,
      "step": 8210,
      "training_step_time": 7.508359670639038
    },
    {
      "epoch": 3.131866455078125e-05,
      "model_forward_time": 3.1457226276397705,
      "step": 8210
    },
    {
      "epoch": 3.131866455078125e-05,
      "step": 8210,
      "training_step_time": 9.927700519561768
    },
    {
      "epoch": 3.131866455078125e-05,
      "model_forward_time": 2.019446611404419,
      "step": 8210
    },
    {
      "epoch": 3.131866455078125e-05,
      "step": 8210,
      "training_step_time": 5.484490156173706
    },
    {
      "epoch": 3.131866455078125e-05,
      "model_forward_time": 2.436511754989624,
      "step": 8210
    },
    {
      "epoch": 3.131866455078125e-05,
      "step": 8210,
      "training_step_time": 10.255366563796997
    },
    {
      "epoch": 3.1322479248046876e-05,
      "model_forward_time": 3.159099817276001,
      "step": 8211
    },
    {
      "epoch": 3.1322479248046876e-05,
      "step": 8211,
      "training_step_time": 9.942140817642212
    },
    {
      "epoch": 3.1322479248046876e-05,
      "model_forward_time": 2.433734178543091,
      "step": 8211
    },
    {
      "epoch": 3.1322479248046876e-05,
      "step": 8211,
      "training_step_time": 7.46947979927063
    },
    {
      "epoch": 3.1322479248046876e-05,
      "model_forward_time": 2.4392037391662598,
      "step": 8211
    },
    {
      "epoch": 3.1322479248046876e-05,
      "step": 8211,
      "training_step_time": 7.478673696517944
    },
    {
      "epoch": 3.1322479248046876e-05,
      "model_forward_time": 2.4369447231292725,
      "step": 8211
    },
    {
      "epoch": 3.1322479248046876e-05,
      "step": 8211,
      "training_step_time": 10.317666292190552
    },
    {
      "epoch": 3.13262939453125e-05,
      "model_forward_time": 2.440065622329712,
      "step": 8212
    },
    {
      "epoch": 3.13262939453125e-05,
      "step": 8212,
      "training_step_time": 7.474107265472412
    },
    {
      "epoch": 3.13262939453125e-05,
      "model_forward_time": 3.132429838180542,
      "step": 8212
    },
    {
      "epoch": 3.13262939453125e-05,
      "step": 8212,
      "training_step_time": 9.907797574996948
    },
    {
      "epoch": 3.13262939453125e-05,
      "model_forward_time": 3.1397173404693604,
      "step": 8212
    },
    {
      "dynamics_loss_avg": 0.06354135591536761,
      "epoch": 3.13262939453125e-05,
      "step": 8212
    },
    {
      "action_loss_avg": 0.02261929688975215,
      "epoch": 3.13262939453125e-05,
      "step": 8212
    },
    {
      "epoch": 3.13262939453125e-05,
      "step": 8212,
      "training_step_time": 9.917134046554565
    },
    {
      "epoch": 3.13262939453125e-05,
      "model_forward_time": 2.4332830905914307,
      "step": 8212
    },
    {
      "epoch": 3.13262939453125e-05,
      "step": 8212,
      "training_step_time": 10.265542030334473
    },
    {
      "epoch": 3.133010864257812e-05,
      "model_forward_time": 2.4437828063964844,
      "step": 8213
    },
    {
      "epoch": 3.133010864257812e-05,
      "step": 8213,
      "training_step_time": 7.478715896606445
    },
    {
      "epoch": 3.133010864257812e-05,
      "model_forward_time": 3.3769075870513916,
      "step": 8213
    },
    {
      "epoch": 3.133010864257812e-05,
      "step": 8213,
      "training_step_time": 10.150525093078613
    },
    {
      "epoch": 3.133010864257812e-05,
      "model_forward_time": 3.1372973918914795,
      "step": 8213
    },
    {
      "epoch": 3.133010864257812e-05,
      "step": 8213,
      "training_step_time": 9.914238452911377
    },
    {
      "epoch": 3.133010864257812e-05,
      "model_forward_time": 3.1372103691101074,
      "step": 8213
    },
    {
      "epoch": 3.133010864257812e-05,
      "step": 8213,
      "training_step_time": 10.287358522415161
    },
    {
      "epoch": 3.133392333984375e-05,
      "model_forward_time": 3.150033712387085,
      "step": 8214
    },
    {
      "epoch": 3.133392333984375e-05,
      "step": 8214,
      "training_step_time": 9.93104076385498
    },
    {
      "epoch": 3.133392333984375e-05,
      "model_forward_time": 3.149665355682373,
      "step": 8214
    },
    {
      "epoch": 3.133392333984375e-05,
      "step": 8214,
      "training_step_time": 9.95275330543518
    },
    {
      "epoch": 3.133392333984375e-05,
      "model_forward_time": 2.7097907066345215,
      "step": 8214
    },
    {
      "epoch": 3.133392333984375e-05,
      "step": 8214,
      "training_step_time": 7.75214409828186
    },
    {
      "epoch": 3.133392333984375e-05,
      "model_forward_time": 2.4311883449554443,
      "step": 8214
    },
    {
      "epoch": 3.133392333984375e-05,
      "step": 8214,
      "training_step_time": 10.017719745635986
    },
    {
      "epoch": 3.1337738037109374e-05,
      "model_forward_time": 3.1566336154937744,
      "step": 8215
    },
    {
      "dynamics_loss_avg": 0.061254081688821314,
      "epoch": 3.1337738037109374e-05,
      "step": 8215
    },
    {
      "action_loss_avg": 0.02414110323879868,
      "epoch": 3.1337738037109374e-05,
      "step": 8215
    },
    {
      "epoch": 3.1337738037109374e-05,
      "step": 8215,
      "training_step_time": 9.941800117492676
    },
    {
      "epoch": 3.1337738037109374e-05,
      "model_forward_time": 3.1396992206573486,
      "step": 8215
    },
    {
      "epoch": 3.1337738037109374e-05,
      "step": 8215,
      "training_step_time": 9.924095630645752
    },
    {
      "epoch": 3.1337738037109374e-05,
      "model_forward_time": 2.434359312057495,
      "step": 8215
    },
    {
      "epoch": 3.1337738037109374e-05,
      "step": 8215,
      "training_step_time": 7.47136378288269
    },
    {
      "epoch": 3.1337738037109374e-05,
      "model_forward_time": 2.7438509464263916,
      "step": 8215
    },
    {
      "epoch": 3.1337738037109374e-05,
      "step": 8215,
      "training_step_time": 10.28601598739624
    },
    {
      "epoch": 3.1341552734375e-05,
      "model_forward_time": 2.447366714477539,
      "step": 8216
    },
    {
      "epoch": 3.1341552734375e-05,
      "step": 8216,
      "training_step_time": 7.484591960906982
    },
    {
      "epoch": 3.1341552734375e-05,
      "model_forward_time": 2.428217649459839,
      "step": 8216
    },
    {
      "epoch": 3.1341552734375e-05,
      "step": 8216,
      "training_step_time": 7.46242880821228
    },
    {
      "epoch": 3.1341552734375e-05,
      "model_forward_time": 2.4360413551330566,
      "step": 8216
    },
    {
      "epoch": 3.1341552734375e-05,
      "step": 8216,
      "training_step_time": 7.475353717803955
    },
    {
      "epoch": 3.1341552734375e-05,
      "model_forward_time": 2.4347360134124756,
      "step": 8216
    },
    {
      "epoch": 3.1341552734375e-05,
      "step": 8216,
      "training_step_time": 10.275692462921143
    },
    {
      "epoch": 3.134536743164063e-05,
      "model_forward_time": 3.4236762523651123,
      "step": 8217
    },
    {
      "epoch": 3.134536743164063e-05,
      "step": 8217,
      "training_step_time": 10.211256265640259
    },
    {
      "epoch": 3.134536743164063e-05,
      "model_forward_time": 2.43804931640625,
      "step": 8217
    },
    {
      "epoch": 3.134536743164063e-05,
      "step": 8217,
      "training_step_time": 7.472059488296509
    },
    {
      "epoch": 3.134536743164063e-05,
      "model_forward_time": 2.42997145652771,
      "step": 8217
    },
    {
      "dynamics_loss_avg": 0.049634326435625556,
      "epoch": 3.134536743164063e-05,
      "step": 8217
    },
    {
      "action_loss_avg": 0.016702309530228377,
      "epoch": 3.134536743164063e-05,
      "step": 8217
    },
    {
      "epoch": 3.134536743164063e-05,
      "step": 8217,
      "training_step_time": 7.468358278274536
    },
    {
      "epoch": 3.134536743164063e-05,
      "model_forward_time": 3.1413888931274414,
      "step": 8217
    },
    {
      "epoch": 3.134536743164063e-05,
      "step": 8217,
      "training_step_time": 10.268699407577515
    },
    {
      "epoch": 3.134918212890625e-05,
      "model_forward_time": 2.4443719387054443,
      "step": 8218
    },
    {
      "epoch": 3.134918212890625e-05,
      "step": 8218,
      "training_step_time": 7.482672929763794
    },
    {
      "epoch": 3.134918212890625e-05,
      "model_forward_time": 3.1337497234344482,
      "step": 8218
    },
    {
      "epoch": 3.134918212890625e-05,
      "step": 8218,
      "training_step_time": 9.91911244392395
    },
    {
      "epoch": 3.134918212890625e-05,
      "model_forward_time": 2.700383186340332,
      "step": 8218
    },
    {
      "epoch": 3.134918212890625e-05,
      "step": 8218,
      "training_step_time": 7.742297887802124
    },
    {
      "epoch": 3.134918212890625e-05,
      "model_forward_time": 2.429544448852539,
      "step": 8218
    },
    {
      "epoch": 3.134918212890625e-05,
      "step": 8218,
      "training_step_time": 10.255706548690796
    },
    {
      "epoch": 3.135299682617187e-05,
      "model_forward_time": 2.4465579986572266,
      "step": 8219
    },
    {
      "epoch": 3.135299682617187e-05,
      "step": 8219,
      "training_step_time": 7.482043027877808
    },
    {
      "epoch": 3.135299682617187e-05,
      "model_forward_time": 2.4309308528900146,
      "step": 8219
    },
    {
      "epoch": 3.135299682617187e-05,
      "step": 8219,
      "training_step_time": 7.470629930496216
    },
    {
      "epoch": 3.135299682617187e-05,
      "model_forward_time": 3.136667013168335,
      "step": 8219
    },
    {
      "epoch": 3.135299682617187e-05,
      "step": 8219,
      "training_step_time": 9.919476985931396
    },
    {
      "epoch": 3.135299682617187e-05,
      "model_forward_time": 2.7059555053710938,
      "step": 8219
    },
    {
      "epoch": 3.135299682617187e-05,
      "step": 8219,
      "training_step_time": 10.029610633850098
    },
    {
      "epoch": 3.13568115234375e-05,
      "grad_norm": 0.4198561906814575,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 8220
    },
    {
      "epoch": 3.13568115234375e-05,
      "model_forward_time": 3.1569089889526367,
      "step": 8220
    },
    {
      "dynamics_loss_avg": 0.05123129235580563,
      "epoch": 3.13568115234375e-05,
      "step": 8220
    },
    {
      "action_loss_avg": 0.016103411093354225,
      "epoch": 3.13568115234375e-05,
      "step": 8220
    },
    {
      "epoch": 3.13568115234375e-05,
      "step": 8220,
      "training_step_time": 9.941588401794434
    },
    {
      "epoch": 3.13568115234375e-05,
      "model_forward_time": 2.4304065704345703,
      "step": 8220
    },
    {
      "epoch": 3.13568115234375e-05,
      "step": 8220,
      "training_step_time": 7.464634895324707
    },
    {
      "epoch": 3.13568115234375e-05,
      "model_forward_time": 2.4265027046203613,
      "step": 8220
    },
    {
      "epoch": 3.13568115234375e-05,
      "step": 8220,
      "training_step_time": 7.460295915603638
    },
    {
      "epoch": 3.13568115234375e-05,
      "model_forward_time": 3.133692741394043,
      "step": 8220
    },
    {
      "epoch": 3.13568115234375e-05,
      "step": 8220,
      "training_step_time": 10.022434949874878
    },
    {
      "epoch": 3.1360626220703125e-05,
      "model_forward_time": 2.702667474746704,
      "step": 8221
    },
    {
      "epoch": 3.1360626220703125e-05,
      "step": 8221,
      "training_step_time": 7.740027904510498
    },
    {
      "epoch": 3.1360626220703125e-05,
      "model_forward_time": 3.134671688079834,
      "step": 8221
    },
    {
      "epoch": 3.1360626220703125e-05,
      "step": 8221,
      "training_step_time": 9.918635845184326
    },
    {
      "epoch": 3.1360626220703125e-05,
      "model_forward_time": 3.1401798725128174,
      "step": 8221
    },
    {
      "epoch": 3.1360626220703125e-05,
      "step": 8221,
      "training_step_time": 9.92138409614563
    },
    {
      "epoch": 3.1360626220703125e-05,
      "model_forward_time": 3.133878231048584,
      "step": 8221
    },
    {
      "epoch": 3.1360626220703125e-05,
      "step": 8221,
      "training_step_time": 10.278592348098755
    },
    {
      "epoch": 3.136444091796875e-05,
      "model_forward_time": 2.4460794925689697,
      "step": 8222
    },
    {
      "epoch": 3.136444091796875e-05,
      "step": 8222,
      "training_step_time": 7.4808855056762695
    },
    {
      "epoch": 3.136444091796875e-05,
      "model_forward_time": 3.1333072185516357,
      "step": 8222
    },
    {
      "epoch": 3.136444091796875e-05,
      "step": 8222,
      "training_step_time": 10.194542646408081
    },
    {
      "epoch": 3.136444091796875e-05,
      "model_forward_time": 2.43709397315979,
      "step": 8222
    },
    {
      "dynamics_loss_avg": 0.05748388469219208,
      "epoch": 3.136444091796875e-05,
      "step": 8222
    },
    {
      "action_loss_avg": 0.01808414561673999,
      "epoch": 3.136444091796875e-05,
      "step": 8222
    },
    {
      "epoch": 3.136444091796875e-05,
      "step": 8222,
      "training_step_time": 7.473963737487793
    },
    {
      "epoch": 3.136444091796875e-05,
      "model_forward_time": 3.1308016777038574,
      "step": 8222
    },
    {
      "epoch": 3.136444091796875e-05,
      "step": 8222,
      "training_step_time": 10.011558294296265
    },
    {
      "epoch": 3.136825561523438e-05,
      "model_forward_time": 3.141810894012451,
      "step": 8223
    },
    {
      "epoch": 3.136825561523438e-05,
      "step": 8223,
      "training_step_time": 9.921885967254639
    },
    {
      "epoch": 3.136825561523438e-05,
      "model_forward_time": 2.4332542419433594,
      "step": 8223
    },
    {
      "epoch": 3.136825561523438e-05,
      "step": 8223,
      "training_step_time": 7.467134714126587
    },
    {
      "epoch": 3.136825561523438e-05,
      "model_forward_time": 3.1321685314178467,
      "step": 8223
    },
    {
      "epoch": 3.136825561523438e-05,
      "step": 8223,
      "training_step_time": 9.91401195526123
    },
    {
      "epoch": 3.136825561523438e-05,
      "model_forward_time": 3.380220651626587,
      "step": 8223
    },
    {
      "epoch": 3.136825561523438e-05,
      "step": 8223,
      "training_step_time": 10.317498445510864
    },
    {
      "epoch": 3.13720703125e-05,
      "model_forward_time": 2.446655511856079,
      "step": 8224
    },
    {
      "epoch": 3.13720703125e-05,
      "step": 8224,
      "training_step_time": 7.47788405418396
    },
    {
      "epoch": 3.13720703125e-05,
      "model_forward_time": 3.138895034790039,
      "step": 8224
    },
    {
      "epoch": 3.13720703125e-05,
      "step": 8224,
      "training_step_time": 9.918375730514526
    },
    {
      "epoch": 3.13720703125e-05,
      "model_forward_time": 3.134050130844116,
      "step": 8224
    },
    {
      "epoch": 3.13720703125e-05,
      "step": 8224,
      "training_step_time": 9.914650201797485
    },
    {
      "epoch": 3.13720703125e-05,
      "model_forward_time": 2.432208299636841,
      "step": 8224
    },
    {
      "epoch": 3.13720703125e-05,
      "step": 8224,
      "training_step_time": 10.282747745513916
    },
    {
      "epoch": 3.1375885009765624e-05,
      "model_forward_time": 2.095940113067627,
      "step": 8225
    },
    {
      "dynamics_loss_avg": 0.04438158511184156,
      "epoch": 3.1375885009765624e-05,
      "step": 8225
    },
    {
      "action_loss_avg": 0.01670974721200764,
      "epoch": 3.1375885009765624e-05,
      "step": 8225
    },
    {
      "epoch": 3.1375885009765624e-05,
      "step": 8225,
      "training_step_time": 5.560906171798706
    },
    {
      "epoch": 3.1375885009765624e-05,
      "model_forward_time": 2.426478385925293,
      "step": 8225
    },
    {
      "epoch": 3.1375885009765624e-05,
      "step": 8225,
      "training_step_time": 7.455226898193359
    },
    {
      "epoch": 3.1375885009765624e-05,
      "model_forward_time": 2.4242324829101562,
      "step": 8225
    },
    {
      "epoch": 3.1375885009765624e-05,
      "step": 8225,
      "training_step_time": 7.4527106285095215
    },
    {
      "epoch": 3.1375885009765624e-05,
      "model_forward_time": 2.4248363971710205,
      "step": 8225
    },
    {
      "epoch": 3.1375885009765624e-05,
      "step": 8225,
      "training_step_time": 9.999701738357544
    },
    {
      "epoch": 3.137969970703125e-05,
      "model_forward_time": 2.4429209232330322,
      "step": 8226
    },
    {
      "epoch": 3.137969970703125e-05,
      "step": 8226,
      "training_step_time": 7.470637321472168
    },
    {
      "epoch": 3.137969970703125e-05,
      "model_forward_time": 3.1251509189605713,
      "step": 8226
    },
    {
      "epoch": 3.137969970703125e-05,
      "step": 8226,
      "training_step_time": 10.205506086349487
    },
    {
      "epoch": 3.137969970703125e-05,
      "model_forward_time": 2.427543878555298,
      "step": 8226
    },
    {
      "epoch": 3.137969970703125e-05,
      "step": 8226,
      "training_step_time": 7.455747127532959
    },
    {
      "epoch": 3.137969970703125e-05,
      "model_forward_time": 3.13116455078125,
      "step": 8226
    },
    {
      "epoch": 3.137969970703125e-05,
      "step": 8226,
      "training_step_time": 10.000702619552612
    },
    {
      "epoch": 3.1383514404296876e-05,
      "model_forward_time": 2.4424099922180176,
      "step": 8227
    },
    {
      "epoch": 3.1383514404296876e-05,
      "step": 8227,
      "training_step_time": 7.4763805866241455
    },
    {
      "epoch": 3.1383514404296876e-05,
      "model_forward_time": 3.128591299057007,
      "step": 8227
    },
    {
      "epoch": 3.1383514404296876e-05,
      "step": 8227,
      "training_step_time": 9.904903650283813
    },
    {
      "epoch": 3.1383514404296876e-05,
      "model_forward_time": 2.4266233444213867,
      "step": 8227
    },
    {
      "dynamics_loss_avg": 0.04067458705976605,
      "epoch": 3.1383514404296876e-05,
      "step": 8227
    },
    {
      "action_loss_avg": 0.01688485114136711,
      "epoch": 3.1383514404296876e-05,
      "step": 8227
    },
    {
      "epoch": 3.1383514404296876e-05,
      "step": 8227,
      "training_step_time": 7.462906837463379
    },
    {
      "epoch": 3.1383514404296876e-05,
      "model_forward_time": 2.699760675430298,
      "step": 8227
    },
    {
      "epoch": 3.1383514404296876e-05,
      "step": 8227,
      "training_step_time": 10.01772665977478
    },
    {
      "epoch": 3.13873291015625e-05,
      "model_forward_time": 2.4480180740356445,
      "step": 8228
    },
    {
      "epoch": 3.13873291015625e-05,
      "step": 8228,
      "training_step_time": 7.486966371536255
    },
    {
      "epoch": 3.13873291015625e-05,
      "model_forward_time": 2.4324727058410645,
      "step": 8228
    },
    {
      "epoch": 3.13873291015625e-05,
      "step": 8228,
      "training_step_time": 7.466332197189331
    },
    {
      "epoch": 3.13873291015625e-05,
      "model_forward_time": 3.1379451751708984,
      "step": 8228
    },
    {
      "epoch": 3.13873291015625e-05,
      "step": 8228,
      "training_step_time": 9.92712116241455
    },
    {
      "epoch": 3.13873291015625e-05,
      "model_forward_time": 2.432783603668213,
      "step": 8228
    },
    {
      "epoch": 3.13873291015625e-05,
      "step": 8228,
      "training_step_time": 10.307671070098877
    },
    {
      "epoch": 3.139114379882812e-05,
      "model_forward_time": 3.4282610416412354,
      "step": 8229
    },
    {
      "epoch": 3.139114379882812e-05,
      "step": 8229,
      "training_step_time": 10.214750289916992
    },
    {
      "epoch": 3.139114379882812e-05,
      "model_forward_time": 3.138678550720215,
      "step": 8229
    },
    {
      "epoch": 3.139114379882812e-05,
      "step": 8229,
      "training_step_time": 9.921093463897705
    },
    {
      "epoch": 3.139114379882812e-05,
      "model_forward_time": 2.435029983520508,
      "step": 8229
    },
    {
      "epoch": 3.139114379882812e-05,
      "step": 8229,
      "training_step_time": 7.469516038894653
    },
    {
      "epoch": 3.139114379882812e-05,
      "model_forward_time": 2.433328628540039,
      "step": 8229
    },
    {
      "epoch": 3.139114379882812e-05,
      "step": 8229,
      "training_step_time": 10.291033744812012
    },
    {
      "epoch": 3.139495849609375e-05,
      "grad_norm": 0.6278226375579834,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 8230
    },
    {
      "epoch": 3.139495849609375e-05,
      "model_forward_time": 2.451660633087158,
      "step": 8230
    },
    {
      "dynamics_loss_avg": 0.07051143236458302,
      "epoch": 3.139495849609375e-05,
      "step": 8230
    },
    {
      "action_loss_avg": 0.02290835790336132,
      "epoch": 3.139495849609375e-05,
      "step": 8230
    },
    {
      "epoch": 3.139495849609375e-05,
      "step": 8230,
      "training_step_time": 7.488479137420654
    },
    {
      "epoch": 3.139495849609375e-05,
      "model_forward_time": 3.3919637203216553,
      "step": 8230
    },
    {
      "epoch": 3.139495849609375e-05,
      "step": 8230,
      "training_step_time": 10.16810131072998
    },
    {
      "epoch": 3.139495849609375e-05,
      "model_forward_time": 2.4338808059692383,
      "step": 8230
    },
    {
      "epoch": 3.139495849609375e-05,
      "step": 8230,
      "training_step_time": 7.466800689697266
    },
    {
      "epoch": 3.139495849609375e-05,
      "model_forward_time": 2.431633472442627,
      "step": 8230
    },
    {
      "epoch": 3.139495849609375e-05,
      "step": 8230,
      "training_step_time": 10.01907753944397
    },
    {
      "epoch": 3.1398773193359375e-05,
      "model_forward_time": 1.7642574310302734,
      "step": 8231
    },
    {
      "epoch": 3.1398773193359375e-05,
      "step": 8231,
      "training_step_time": 5.224334001541138
    },
    {
      "epoch": 3.1398773193359375e-05,
      "model_forward_time": 3.1359453201293945,
      "step": 8231
    },
    {
      "epoch": 3.1398773193359375e-05,
      "step": 8231,
      "training_step_time": 9.908469438552856
    },
    {
      "epoch": 3.1398773193359375e-05,
      "model_forward_time": 1.7425079345703125,
      "step": 8231
    },
    {
      "epoch": 3.1398773193359375e-05,
      "step": 8231,
      "training_step_time": 5.219922065734863
    },
    {
      "epoch": 3.1398773193359375e-05,
      "model_forward_time": 3.1312522888183594,
      "step": 8231
    },
    {
      "epoch": 3.1398773193359375e-05,
      "step": 8231,
      "training_step_time": 10.29568862915039
    },
    {
      "epoch": 3.1402587890625e-05,
      "model_forward_time": 2.0604567527770996,
      "step": 8232
    },
    {
      "epoch": 3.1402587890625e-05,
      "step": 8232,
      "training_step_time": 5.515203237533569
    },
    {
      "epoch": 3.1402587890625e-05,
      "model_forward_time": 2.428877353668213,
      "step": 8232
    },
    {
      "epoch": 3.1402587890625e-05,
      "step": 8232,
      "training_step_time": 7.4532630443573
    },
    {
      "epoch": 3.1402587890625e-05,
      "model_forward_time": 3.140035629272461,
      "step": 8232
    },
    {
      "dynamics_loss_avg": 0.05433428250253201,
      "epoch": 3.1402587890625e-05,
      "step": 8232
    },
    {
      "action_loss_avg": 0.023752414155751467,
      "epoch": 3.1402587890625e-05,
      "step": 8232
    },
    {
      "epoch": 3.1402587890625e-05,
      "step": 8232,
      "training_step_time": 9.912590265274048
    },
    {
      "epoch": 3.1402587890625e-05,
      "model_forward_time": 3.145185947418213,
      "step": 8232
    },
    {
      "epoch": 3.1402587890625e-05,
      "step": 8232,
      "training_step_time": 10.022762537002563
    },
    {
      "epoch": 3.140640258789063e-05,
      "model_forward_time": 2.4570257663726807,
      "step": 8233
    },
    {
      "epoch": 3.140640258789063e-05,
      "step": 8233,
      "training_step_time": 7.491388320922852
    },
    {
      "epoch": 3.140640258789063e-05,
      "model_forward_time": 3.424337863922119,
      "step": 8233
    },
    {
      "epoch": 3.140640258789063e-05,
      "step": 8233,
      "training_step_time": 10.198475122451782
    },
    {
      "epoch": 3.140640258789063e-05,
      "model_forward_time": 2.4382622241973877,
      "step": 8233
    },
    {
      "epoch": 3.140640258789063e-05,
      "step": 8233,
      "training_step_time": 7.470330238342285
    },
    {
      "epoch": 3.140640258789063e-05,
      "model_forward_time": 3.145362138748169,
      "step": 8233
    },
    {
      "epoch": 3.140640258789063e-05,
      "step": 8233,
      "training_step_time": 10.287396907806396
    },
    {
      "epoch": 3.141021728515625e-05,
      "model_forward_time": 3.1464784145355225,
      "step": 8234
    },
    {
      "epoch": 3.141021728515625e-05,
      "step": 8234,
      "training_step_time": 9.923815965652466
    },
    {
      "epoch": 3.141021728515625e-05,
      "model_forward_time": 3.1373541355133057,
      "step": 8234
    },
    {
      "epoch": 3.141021728515625e-05,
      "step": 8234,
      "training_step_time": 9.915034055709839
    },
    {
      "epoch": 3.141021728515625e-05,
      "model_forward_time": 3.3938064575195312,
      "step": 8234
    },
    {
      "epoch": 3.141021728515625e-05,
      "step": 8234,
      "training_step_time": 10.164388656616211
    },
    {
      "epoch": 3.141021728515625e-05,
      "model_forward_time": 3.1379916667938232,
      "step": 8234
    },
    {
      "epoch": 3.141021728515625e-05,
      "step": 8234,
      "training_step_time": 10.299161195755005
    },
    {
      "epoch": 3.141403198242187e-05,
      "model_forward_time": 3.1598072052001953,
      "step": 8235
    },
    {
      "dynamics_loss_avg": 0.04490471575409174,
      "epoch": 3.141403198242187e-05,
      "step": 8235
    },
    {
      "action_loss_avg": 0.019804982421919703,
      "epoch": 3.141403198242187e-05,
      "step": 8235
    },
    {
      "epoch": 3.141403198242187e-05,
      "step": 8235,
      "training_step_time": 9.943167686462402
    },
    {
      "epoch": 3.141403198242187e-05,
      "model_forward_time": 2.4314584732055664,
      "step": 8235
    },
    {
      "epoch": 3.141403198242187e-05,
      "step": 8235,
      "training_step_time": 7.467154264450073
    },
    {
      "epoch": 3.141403198242187e-05,
      "model_forward_time": 2.434494972229004,
      "step": 8235
    },
    {
      "epoch": 3.141403198242187e-05,
      "step": 8235,
      "training_step_time": 7.472208261489868
    },
    {
      "epoch": 3.141403198242187e-05,
      "model_forward_time": 1.7424860000610352,
      "step": 8235
    },
    {
      "epoch": 3.141403198242187e-05,
      "step": 8235,
      "training_step_time": 10.296928405761719
    },
    {
      "epoch": 3.14178466796875e-05,
      "model_forward_time": 3.4891233444213867,
      "step": 8236
    },
    {
      "epoch": 3.14178466796875e-05,
      "step": 8236,
      "training_step_time": 10.265813827514648
    },
    {
      "epoch": 3.14178466796875e-05,
      "model_forward_time": 3.1371779441833496,
      "step": 8236
    },
    {
      "epoch": 3.14178466796875e-05,
      "step": 8236,
      "training_step_time": 9.909072875976562
    },
    {
      "epoch": 3.14178466796875e-05,
      "model_forward_time": 3.140512228012085,
      "step": 8236
    },
    {
      "epoch": 3.14178466796875e-05,
      "step": 8236,
      "training_step_time": 9.91404104232788
    },
    {
      "epoch": 3.14178466796875e-05,
      "model_forward_time": 2.4355130195617676,
      "step": 8236
    },
    {
      "epoch": 3.14178466796875e-05,
      "step": 8236,
      "training_step_time": 10.302014589309692
    },
    {
      "epoch": 3.1421661376953125e-05,
      "model_forward_time": 2.45031476020813,
      "step": 8237
    },
    {
      "epoch": 3.1421661376953125e-05,
      "step": 8237,
      "training_step_time": 7.484620571136475
    },
    {
      "epoch": 3.1421661376953125e-05,
      "model_forward_time": 2.711460828781128,
      "step": 8237
    },
    {
      "epoch": 3.1421661376953125e-05,
      "step": 8237,
      "training_step_time": 7.737256765365601
    },
    {
      "epoch": 3.1421661376953125e-05,
      "model_forward_time": 3.131558418273926,
      "step": 8237
    },
    {
      "dynamics_loss_avg": 0.05823520328849554,
      "epoch": 3.1421661376953125e-05,
      "step": 8237
    },
    {
      "action_loss_avg": 0.020901856385171412,
      "epoch": 3.1421661376953125e-05,
      "step": 8237
    },
    {
      "epoch": 3.1421661376953125e-05,
      "step": 8237,
      "training_step_time": 9.903875827789307
    },
    {
      "epoch": 3.1421661376953125e-05,
      "model_forward_time": 2.433095693588257,
      "step": 8237
    },
    {
      "epoch": 3.1421661376953125e-05,
      "step": 8237,
      "training_step_time": 10.306898832321167
    },
    {
      "epoch": 3.142547607421875e-05,
      "model_forward_time": 2.4400320053100586,
      "step": 8238
    },
    {
      "epoch": 3.142547607421875e-05,
      "step": 8238,
      "training_step_time": 7.468115329742432
    },
    {
      "epoch": 3.142547607421875e-05,
      "model_forward_time": 3.1279685497283936,
      "step": 8238
    },
    {
      "epoch": 3.142547607421875e-05,
      "step": 8238,
      "training_step_time": 9.89579463005066
    },
    {
      "epoch": 3.142547607421875e-05,
      "model_forward_time": 3.415752410888672,
      "step": 8238
    },
    {
      "epoch": 3.142547607421875e-05,
      "step": 8238,
      "training_step_time": 10.183162212371826
    },
    {
      "epoch": 3.142547607421875e-05,
      "model_forward_time": 2.428866386413574,
      "step": 8238
    },
    {
      "epoch": 3.142547607421875e-05,
      "step": 8238,
      "training_step_time": 10.002055406570435
    },
    {
      "epoch": 3.142929077148438e-05,
      "model_forward_time": 3.1400837898254395,
      "step": 8239
    },
    {
      "epoch": 3.142929077148438e-05,
      "step": 8239,
      "training_step_time": 9.909709215164185
    },
    {
      "epoch": 3.142929077148438e-05,
      "model_forward_time": 2.429213762283325,
      "step": 8239
    },
    {
      "epoch": 3.142929077148438e-05,
      "step": 8239,
      "training_step_time": 7.461446523666382
    },
    {
      "epoch": 3.142929077148438e-05,
      "model_forward_time": 2.4278125762939453,
      "step": 8239
    },
    {
      "epoch": 3.142929077148438e-05,
      "step": 8239,
      "training_step_time": 7.45914888381958
    },
    {
      "epoch": 3.142929077148438e-05,
      "model_forward_time": 2.6994569301605225,
      "step": 8239
    },
    {
      "epoch": 3.142929077148438e-05,
      "step": 8239,
      "training_step_time": 10.327423095703125
    },
    {
      "epoch": 3.143310546875e-05,
      "grad_norm": 0.44438210129737854,
      "learning_rate": 1e-05,
      "loss": 0.0803,
      "step": 8240
    },
    {
      "epoch": 3.143310546875e-05,
      "model_forward_time": 2.4412782192230225,
      "step": 8240
    },
    {
      "dynamics_loss_avg": 0.06395979877561331,
      "epoch": 3.143310546875e-05,
      "step": 8240
    },
    {
      "action_loss_avg": 0.0228053861297667,
      "epoch": 3.143310546875e-05,
      "step": 8240
    },
    {
      "epoch": 3.143310546875e-05,
      "step": 8240,
      "training_step_time": 7.4756388664245605
    },
    {
      "epoch": 3.143310546875e-05,
      "model_forward_time": 2.425246238708496,
      "step": 8240
    },
    {
      "epoch": 3.143310546875e-05,
      "step": 8240,
      "training_step_time": 7.456672430038452
    },
    {
      "epoch": 3.143310546875e-05,
      "model_forward_time": 3.130810499191284,
      "step": 8240
    },
    {
      "epoch": 3.143310546875e-05,
      "step": 8240,
      "training_step_time": 9.902282238006592
    },
    {
      "epoch": 3.143310546875e-05,
      "model_forward_time": 3.1367459297180176,
      "step": 8240
    },
    {
      "epoch": 3.143310546875e-05,
      "step": 8240,
      "training_step_time": 10.01329517364502
    },
    {
      "epoch": 3.1436920166015624e-05,
      "model_forward_time": 3.147024393081665,
      "step": 8241
    },
    {
      "epoch": 3.1436920166015624e-05,
      "step": 8241,
      "training_step_time": 10.203564643859863
    },
    {
      "epoch": 3.1436920166015624e-05,
      "model_forward_time": 2.429215669631958,
      "step": 8241
    },
    {
      "epoch": 3.1436920166015624e-05,
      "step": 8241,
      "training_step_time": 7.461421728134155
    },
    {
      "epoch": 3.1436920166015624e-05,
      "model_forward_time": 2.4279799461364746,
      "step": 8241
    },
    {
      "epoch": 3.1436920166015624e-05,
      "step": 8241,
      "training_step_time": 7.454784154891968
    },
    {
      "epoch": 3.1436920166015624e-05,
      "model_forward_time": 3.1252126693725586,
      "step": 8241
    },
    {
      "epoch": 3.1436920166015624e-05,
      "step": 8241,
      "training_step_time": 10.279520273208618
    },
    {
      "epoch": 3.144073486328125e-05,
      "model_forward_time": 2.447457790374756,
      "step": 8242
    },
    {
      "epoch": 3.144073486328125e-05,
      "step": 8242,
      "training_step_time": 7.4753642082214355
    },
    {
      "epoch": 3.144073486328125e-05,
      "model_forward_time": 2.4286279678344727,
      "step": 8242
    },
    {
      "epoch": 3.144073486328125e-05,
      "step": 8242,
      "training_step_time": 7.455801725387573
    },
    {
      "epoch": 3.144073486328125e-05,
      "model_forward_time": 2.695763349533081,
      "step": 8242
    },
    {
      "dynamics_loss_avg": 0.049555523321032524,
      "epoch": 3.144073486328125e-05,
      "step": 8242
    },
    {
      "action_loss_avg": 0.017411885596811772,
      "epoch": 3.144073486328125e-05,
      "step": 8242
    },
    {
      "epoch": 3.144073486328125e-05,
      "step": 8242,
      "training_step_time": 7.73035454750061
    },
    {
      "epoch": 3.144073486328125e-05,
      "model_forward_time": 3.1340603828430176,
      "step": 8242
    },
    {
      "epoch": 3.144073486328125e-05,
      "step": 8242,
      "training_step_time": 10.319695472717285
    },
    {
      "epoch": 3.1444549560546876e-05,
      "model_forward_time": 3.1618990898132324,
      "step": 8243
    },
    {
      "epoch": 3.1444549560546876e-05,
      "step": 8243,
      "training_step_time": 9.938997030258179
    },
    {
      "epoch": 3.1444549560546876e-05,
      "model_forward_time": 3.138073444366455,
      "step": 8243
    },
    {
      "epoch": 3.1444549560546876e-05,
      "step": 8243,
      "training_step_time": 9.91064190864563
    },
    {
      "epoch": 3.1444549560546876e-05,
      "model_forward_time": 2.4364237785339355,
      "step": 8243
    },
    {
      "epoch": 3.1444549560546876e-05,
      "step": 8243,
      "training_step_time": 7.471786975860596
    },
    {
      "epoch": 3.1444549560546876e-05,
      "model_forward_time": 2.695219039916992,
      "step": 8243
    },
    {
      "epoch": 3.1444549560546876e-05,
      "step": 8243,
      "training_step_time": 10.025275468826294
    },
    {
      "epoch": 3.14483642578125e-05,
      "model_forward_time": 3.159877300262451,
      "step": 8244
    },
    {
      "epoch": 3.14483642578125e-05,
      "step": 8244,
      "training_step_time": 9.947156429290771
    },
    {
      "epoch": 3.14483642578125e-05,
      "model_forward_time": 2.438340187072754,
      "step": 8244
    },
    {
      "epoch": 3.14483642578125e-05,
      "step": 8244,
      "training_step_time": 7.474804878234863
    },
    {
      "epoch": 3.14483642578125e-05,
      "model_forward_time": 3.138737201690674,
      "step": 8244
    },
    {
      "epoch": 3.14483642578125e-05,
      "step": 8244,
      "training_step_time": 9.917582035064697
    },
    {
      "epoch": 3.14483642578125e-05,
      "model_forward_time": 2.435946226119995,
      "step": 8244
    },
    {
      "epoch": 3.14483642578125e-05,
      "step": 8244,
      "training_step_time": 10.314967393875122
    },
    {
      "epoch": 3.145217895507812e-05,
      "model_forward_time": 3.143083333969116,
      "step": 8245
    },
    {
      "dynamics_loss_avg": 0.06661250619217754,
      "epoch": 3.145217895507812e-05,
      "step": 8245
    },
    {
      "action_loss_avg": 0.02403140738606453,
      "epoch": 3.145217895507812e-05,
      "step": 8245
    },
    {
      "epoch": 3.145217895507812e-05,
      "step": 8245,
      "training_step_time": 10.188220739364624
    },
    {
      "epoch": 3.145217895507812e-05,
      "model_forward_time": 2.4337103366851807,
      "step": 8245
    },
    {
      "epoch": 3.145217895507812e-05,
      "step": 8245,
      "training_step_time": 7.46816086769104
    },
    {
      "epoch": 3.145217895507812e-05,
      "model_forward_time": 2.4352433681488037,
      "step": 8245
    },
    {
      "epoch": 3.145217895507812e-05,
      "step": 8245,
      "training_step_time": 7.470283269882202
    },
    {
      "epoch": 3.145217895507812e-05,
      "model_forward_time": 3.1333634853363037,
      "step": 8245
    },
    {
      "epoch": 3.145217895507812e-05,
      "step": 8245,
      "training_step_time": 10.008928537368774
    },
    {
      "epoch": 3.145599365234375e-05,
      "model_forward_time": 2.451369047164917,
      "step": 8246
    },
    {
      "epoch": 3.145599365234375e-05,
      "step": 8246,
      "training_step_time": 7.4836695194244385
    },
    {
      "epoch": 3.145599365234375e-05,
      "model_forward_time": 3.1394667625427246,
      "step": 8246
    },
    {
      "epoch": 3.145599365234375e-05,
      "step": 8246,
      "training_step_time": 9.912193059921265
    },
    {
      "epoch": 3.145599365234375e-05,
      "model_forward_time": 2.6973838806152344,
      "step": 8246
    },
    {
      "epoch": 3.145599365234375e-05,
      "step": 8246,
      "training_step_time": 7.727386474609375
    },
    {
      "epoch": 3.145599365234375e-05,
      "model_forward_time": 3.1350016593933105,
      "step": 8246
    },
    {
      "epoch": 3.145599365234375e-05,
      "step": 8246,
      "training_step_time": 10.300026178359985
    },
    {
      "epoch": 3.1459808349609375e-05,
      "model_forward_time": 3.157198667526245,
      "step": 8247
    },
    {
      "epoch": 3.1459808349609375e-05,
      "step": 8247,
      "training_step_time": 9.93621277809143
    },
    {
      "epoch": 3.1459808349609375e-05,
      "model_forward_time": 3.143364429473877,
      "step": 8247
    },
    {
      "epoch": 3.1459808349609375e-05,
      "step": 8247,
      "training_step_time": 9.916744232177734
    },
    {
      "epoch": 3.1459808349609375e-05,
      "model_forward_time": 3.135861873626709,
      "step": 8247
    },
    {
      "dynamics_loss_avg": 0.05290517695248127,
      "epoch": 3.1459808349609375e-05,
      "step": 8247
    },
    {
      "action_loss_avg": 0.0217740741558373,
      "epoch": 3.1459808349609375e-05,
      "step": 8247
    },
    {
      "epoch": 3.1459808349609375e-05,
      "step": 8247,
      "training_step_time": 9.913551568984985
    },
    {
      "epoch": 3.1459808349609375e-05,
      "model_forward_time": 2.6876871585845947,
      "step": 8247
    },
    {
      "epoch": 3.1459808349609375e-05,
      "step": 8247,
      "training_step_time": 10.279743194580078
    },
    {
      "epoch": 3.1463623046875e-05,
      "model_forward_time": 3.142920970916748,
      "step": 8248
    },
    {
      "epoch": 3.1463623046875e-05,
      "step": 8248,
      "training_step_time": 9.91886830329895
    },
    {
      "epoch": 3.1463623046875e-05,
      "model_forward_time": 2.4354135990142822,
      "step": 8248
    },
    {
      "epoch": 3.1463623046875e-05,
      "step": 8248,
      "training_step_time": 7.469498157501221
    },
    {
      "epoch": 3.1463623046875e-05,
      "model_forward_time": 3.133380174636841,
      "step": 8248
    },
    {
      "epoch": 3.1463623046875e-05,
      "step": 8248,
      "training_step_time": 9.905049562454224
    },
    {
      "epoch": 3.1463623046875e-05,
      "model_forward_time": 3.13789963722229,
      "step": 8248
    },
    {
      "epoch": 3.1463623046875e-05,
      "step": 8248,
      "training_step_time": 10.006673336029053
    },
    {
      "epoch": 3.146743774414063e-05,
      "model_forward_time": 3.4869446754455566,
      "step": 8249
    },
    {
      "epoch": 3.146743774414063e-05,
      "step": 8249,
      "training_step_time": 10.262750387191772
    },
    {
      "epoch": 3.146743774414063e-05,
      "model_forward_time": 3.144683837890625,
      "step": 8249
    },
    {
      "epoch": 3.146743774414063e-05,
      "step": 8249,
      "training_step_time": 9.919167280197144
    },
    {
      "epoch": 3.146743774414063e-05,
      "model_forward_time": 3.143320322036743,
      "step": 8249
    },
    {
      "epoch": 3.146743774414063e-05,
      "step": 8249,
      "training_step_time": 9.923348665237427
    },
    {
      "epoch": 3.146743774414063e-05,
      "model_forward_time": 3.152308702468872,
      "step": 8249
    },
    {
      "epoch": 3.146743774414063e-05,
      "step": 8249,
      "training_step_time": 10.02841830253601
    },
    {
      "epoch": 3.147125244140625e-05,
      "grad_norm": 0.5852088332176208,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 8250
    },
    {
      "epoch": 3.147125244140625e-05,
      "model_forward_time": 3.160078287124634,
      "step": 8250
    },
    {
      "dynamics_loss_avg": 0.053263604454696176,
      "epoch": 3.147125244140625e-05,
      "step": 8250
    },
    {
      "action_loss_avg": 0.022928203362971543,
      "epoch": 3.147125244140625e-05,
      "step": 8250
    },
    {
      "epoch": 3.147125244140625e-05,
      "step": 8250,
      "training_step_time": 9.946104288101196
    },
    {
      "epoch": 3.147125244140625e-05,
      "model_forward_time": 3.4497909545898438,
      "step": 8250
    },
    {
      "epoch": 3.147125244140625e-05,
      "step": 8250,
      "training_step_time": 10.228145837783813
    },
    {
      "epoch": 3.147125244140625e-05,
      "model_forward_time": 3.147268295288086,
      "step": 8250
    },
    {
      "epoch": 3.147125244140625e-05,
      "step": 8250,
      "training_step_time": 9.928452968597412
    },
    {
      "epoch": 3.147125244140625e-05,
      "model_forward_time": 3.148249387741089,
      "step": 8250
    },
    {
      "epoch": 3.147125244140625e-05,
      "step": 8250,
      "training_step_time": 10.303262948989868
    },
    {
      "epoch": 3.147506713867187e-05,
      "model_forward_time": 3.155250072479248,
      "step": 8251
    },
    {
      "epoch": 3.147506713867187e-05,
      "step": 8251,
      "training_step_time": 9.940852880477905
    },
    {
      "epoch": 3.147506713867187e-05,
      "model_forward_time": 3.1498944759368896,
      "step": 8251
    },
    {
      "epoch": 3.147506713867187e-05,
      "step": 8251,
      "training_step_time": 9.937315225601196
    },
    {
      "epoch": 3.147506713867187e-05,
      "model_forward_time": 3.4007909297943115,
      "step": 8251
    },
    {
      "epoch": 3.147506713867187e-05,
      "step": 8251,
      "training_step_time": 10.189229249954224
    },
    {
      "epoch": 3.147506713867187e-05,
      "model_forward_time": 3.157317876815796,
      "step": 8251
    },
    {
      "epoch": 3.147506713867187e-05,
      "step": 8251,
      "training_step_time": 10.29313039779663
    },
    {
      "epoch": 3.14788818359375e-05,
      "model_forward_time": 3.1642277240753174,
      "step": 8252
    },
    {
      "epoch": 3.14788818359375e-05,
      "step": 8252,
      "training_step_time": 9.948310375213623
    },
    {
      "epoch": 3.14788818359375e-05,
      "model_forward_time": 3.13450026512146,
      "step": 8252
    },
    {
      "epoch": 3.14788818359375e-05,
      "step": 8252,
      "training_step_time": 9.908133029937744
    },
    {
      "epoch": 3.14788818359375e-05,
      "model_forward_time": 3.4129645824432373,
      "step": 8252
    },
    {
      "dynamics_loss_avg": 0.04470635131001473,
      "epoch": 3.14788818359375e-05,
      "step": 8252
    },
    {
      "action_loss_avg": 0.012992535484954715,
      "epoch": 3.14788818359375e-05,
      "step": 8252
    },
    {
      "epoch": 3.14788818359375e-05,
      "step": 8252,
      "training_step_time": 10.19241189956665
    },
    {
      "epoch": 3.14788818359375e-05,
      "model_forward_time": 3.135795831680298,
      "step": 8252
    },
    {
      "epoch": 3.14788818359375e-05,
      "step": 8252,
      "training_step_time": 10.031384468078613
    },
    {
      "epoch": 3.1482696533203125e-05,
      "model_forward_time": 3.144634962081909,
      "step": 8253
    },
    {
      "epoch": 3.1482696533203125e-05,
      "step": 8253,
      "training_step_time": 9.913640022277832
    },
    {
      "epoch": 3.1482696533203125e-05,
      "model_forward_time": 3.1286368370056152,
      "step": 8253
    },
    {
      "epoch": 3.1482696533203125e-05,
      "step": 8253,
      "training_step_time": 9.901376485824585
    },
    {
      "epoch": 3.1482696533203125e-05,
      "model_forward_time": 3.1320104598999023,
      "step": 8253
    },
    {
      "epoch": 3.1482696533203125e-05,
      "step": 8253,
      "training_step_time": 9.899062633514404
    },
    {
      "epoch": 3.1482696533203125e-05,
      "model_forward_time": 3.1286306381225586,
      "step": 8253
    },
    {
      "epoch": 3.1482696533203125e-05,
      "step": 8253,
      "training_step_time": 10.26945185661316
    },
    {
      "epoch": 3.148651123046875e-05,
      "model_forward_time": 3.1453611850738525,
      "step": 8254
    },
    {
      "epoch": 3.148651123046875e-05,
      "step": 8254,
      "training_step_time": 9.919093132019043
    },
    {
      "epoch": 3.148651123046875e-05,
      "model_forward_time": 3.1323983669281006,
      "step": 8254
    },
    {
      "epoch": 3.148651123046875e-05,
      "step": 8254,
      "training_step_time": 9.902195453643799
    },
    {
      "epoch": 3.148651123046875e-05,
      "model_forward_time": 3.1395325660705566,
      "step": 8254
    },
    {
      "epoch": 3.148651123046875e-05,
      "step": 8254,
      "training_step_time": 9.910340785980225
    },
    {
      "epoch": 3.148651123046875e-05,
      "model_forward_time": 3.1365795135498047,
      "step": 8254
    },
    {
      "epoch": 3.148651123046875e-05,
      "step": 8254,
      "training_step_time": 10.274148225784302
    },
    {
      "epoch": 3.149032592773438e-05,
      "model_forward_time": 3.426309823989868,
      "step": 8255
    },
    {
      "dynamics_loss_avg": 0.07270286250859499,
      "epoch": 3.149032592773438e-05,
      "step": 8255
    },
    {
      "action_loss_avg": 0.026625801529735326,
      "epoch": 3.149032592773438e-05,
      "step": 8255
    },
    {
      "epoch": 3.149032592773438e-05,
      "step": 8255,
      "training_step_time": 10.204343557357788
    },
    {
      "epoch": 3.149032592773438e-05,
      "model_forward_time": 3.138474225997925,
      "step": 8255
    },
    {
      "epoch": 3.149032592773438e-05,
      "step": 8255,
      "training_step_time": 9.917391300201416
    },
    {
      "epoch": 3.149032592773438e-05,
      "model_forward_time": 3.1399567127227783,
      "step": 8255
    },
    {
      "epoch": 3.149032592773438e-05,
      "step": 8255,
      "training_step_time": 9.9121732711792
    },
    {
      "epoch": 3.149032592773438e-05,
      "model_forward_time": 3.137953281402588,
      "step": 8255
    },
    {
      "epoch": 3.149032592773438e-05,
      "step": 8255,
      "training_step_time": 10.308550357818604
    },
    {
      "epoch": 3.1494140625e-05,
      "model_forward_time": 3.154001235961914,
      "step": 8256
    },
    {
      "epoch": 3.1494140625e-05,
      "step": 8256,
      "training_step_time": 9.92785930633545
    },
    {
      "epoch": 3.1494140625e-05,
      "model_forward_time": 3.1354339122772217,
      "step": 8256
    },
    {
      "epoch": 3.1494140625e-05,
      "step": 8256,
      "training_step_time": 10.178478479385376
    },
    {
      "epoch": 3.1494140625e-05,
      "model_forward_time": 3.13503098487854,
      "step": 8256
    },
    {
      "epoch": 3.1494140625e-05,
      "step": 8256,
      "training_step_time": 9.908108949661255
    },
    {
      "epoch": 3.1494140625e-05,
      "model_forward_time": 3.1379024982452393,
      "step": 8256
    },
    {
      "epoch": 3.1494140625e-05,
      "step": 8256,
      "training_step_time": 10.252945899963379
    },
    {
      "epoch": 3.1497955322265624e-05,
      "model_forward_time": 3.1402170658111572,
      "step": 8257
    },
    {
      "epoch": 3.1497955322265624e-05,
      "step": 8257,
      "training_step_time": 9.912378549575806
    },
    {
      "epoch": 3.1497955322265624e-05,
      "model_forward_time": 3.1385726928710938,
      "step": 8257
    },
    {
      "epoch": 3.1497955322265624e-05,
      "step": 8257,
      "training_step_time": 9.911409616470337
    },
    {
      "epoch": 3.1497955322265624e-05,
      "model_forward_time": 3.4136359691619873,
      "step": 8257
    },
    {
      "dynamics_loss_avg": 0.0691016461700201,
      "epoch": 3.1497955322265624e-05,
      "step": 8257
    },
    {
      "action_loss_avg": 0.02310127029195428,
      "epoch": 3.1497955322265624e-05,
      "step": 8257
    },
    {
      "epoch": 3.1497955322265624e-05,
      "step": 8257,
      "training_step_time": 10.192369222640991
    },
    {
      "epoch": 3.1497955322265624e-05,
      "model_forward_time": 3.1403143405914307,
      "step": 8257
    },
    {
      "epoch": 3.1497955322265624e-05,
      "step": 8257,
      "training_step_time": 10.308782577514648
    },
    {
      "epoch": 3.150177001953125e-05,
      "model_forward_time": 3.145385503768921,
      "step": 8258
    },
    {
      "epoch": 3.150177001953125e-05,
      "step": 8258,
      "training_step_time": 9.92548942565918
    },
    {
      "epoch": 3.150177001953125e-05,
      "model_forward_time": 3.140566825866699,
      "step": 8258
    },
    {
      "epoch": 3.150177001953125e-05,
      "step": 8258,
      "training_step_time": 9.913562774658203
    },
    {
      "epoch": 3.150177001953125e-05,
      "model_forward_time": 3.1358819007873535,
      "step": 8258
    },
    {
      "epoch": 3.150177001953125e-05,
      "step": 8258,
      "training_step_time": 9.910534381866455
    },
    {
      "epoch": 3.150177001953125e-05,
      "model_forward_time": 3.3980655670166016,
      "step": 8258
    },
    {
      "epoch": 3.150177001953125e-05,
      "step": 8258,
      "training_step_time": 10.267817735671997
    },
    {
      "epoch": 3.1505584716796876e-05,
      "model_forward_time": 3.1517176628112793,
      "step": 8259
    },
    {
      "epoch": 3.1505584716796876e-05,
      "step": 8259,
      "training_step_time": 9.922192335128784
    },
    {
      "epoch": 3.1505584716796876e-05,
      "model_forward_time": 3.139164924621582,
      "step": 8259
    },
    {
      "epoch": 3.1505584716796876e-05,
      "step": 8259,
      "training_step_time": 9.910848140716553
    },
    {
      "epoch": 3.1505584716796876e-05,
      "model_forward_time": 3.1306116580963135,
      "step": 8259
    },
    {
      "epoch": 3.1505584716796876e-05,
      "step": 8259,
      "training_step_time": 9.90573787689209
    },
    {
      "epoch": 3.1505584716796876e-05,
      "model_forward_time": 3.135699987411499,
      "step": 8259
    },
    {
      "epoch": 3.1505584716796876e-05,
      "step": 8259,
      "training_step_time": 10.24289870262146
    },
    {
      "epoch": 3.15093994140625e-05,
      "grad_norm": 0.3250868618488312,
      "learning_rate": 1e-05,
      "loss": 0.0785,
      "step": 8260
    },
    {
      "epoch": 3.15093994140625e-05,
      "model_forward_time": 3.398291826248169,
      "step": 8260
    },
    {
      "dynamics_loss_avg": 0.05744605353102088,
      "epoch": 3.15093994140625e-05,
      "step": 8260
    },
    {
      "action_loss_avg": 0.01741276360116899,
      "epoch": 3.15093994140625e-05,
      "step": 8260
    },
    {
      "epoch": 3.15093994140625e-05,
      "step": 8260,
      "training_step_time": 10.173599243164062
    },
    {
      "epoch": 3.15093994140625e-05,
      "model_forward_time": 3.1370766162872314,
      "step": 8260
    },
    {
      "epoch": 3.15093994140625e-05,
      "step": 8260,
      "training_step_time": 9.907973766326904
    },
    {
      "epoch": 3.15093994140625e-05,
      "model_forward_time": 3.1335015296936035,
      "step": 8260
    },
    {
      "epoch": 3.15093994140625e-05,
      "step": 8260,
      "training_step_time": 9.907031536102295
    },
    {
      "epoch": 3.15093994140625e-05,
      "model_forward_time": 3.139108419418335,
      "step": 8260
    },
    {
      "epoch": 3.15093994140625e-05,
      "step": 8260,
      "training_step_time": 10.273524522781372
    },
    {
      "epoch": 3.151321411132812e-05,
      "model_forward_time": 3.1485111713409424,
      "step": 8261
    },
    {
      "epoch": 3.151321411132812e-05,
      "step": 8261,
      "training_step_time": 9.926308870315552
    },
    {
      "epoch": 3.151321411132812e-05,
      "model_forward_time": 3.420201539993286,
      "step": 8261
    },
    {
      "epoch": 3.151321411132812e-05,
      "step": 8261,
      "training_step_time": 10.193668127059937
    },
    {
      "epoch": 3.151321411132812e-05,
      "model_forward_time": 3.132836103439331,
      "step": 8261
    },
    {
      "epoch": 3.151321411132812e-05,
      "step": 8261,
      "training_step_time": 9.904538631439209
    },
    {
      "epoch": 3.151321411132812e-05,
      "model_forward_time": 3.1387078762054443,
      "step": 8261
    },
    {
      "epoch": 3.151321411132812e-05,
      "step": 8261,
      "training_step_time": 10.285065412521362
    },
    {
      "epoch": 3.151702880859375e-05,
      "model_forward_time": 3.149901866912842,
      "step": 8262
    },
    {
      "epoch": 3.151702880859375e-05,
      "step": 8262,
      "training_step_time": 9.92362642288208
    },
    {
      "epoch": 3.151702880859375e-05,
      "model_forward_time": 3.1357553005218506,
      "step": 8262
    },
    {
      "epoch": 3.151702880859375e-05,
      "step": 8262,
      "training_step_time": 9.914438486099243
    },
    {
      "epoch": 3.151702880859375e-05,
      "model_forward_time": 3.1376962661743164,
      "step": 8262
    },
    {
      "dynamics_loss_avg": 0.052556459605693814,
      "epoch": 3.151702880859375e-05,
      "step": 8262
    },
    {
      "action_loss_avg": 0.018821738194674253,
      "epoch": 3.151702880859375e-05,
      "step": 8262
    },
    {
      "epoch": 3.151702880859375e-05,
      "step": 8262,
      "training_step_time": 10.196066617965698
    },
    {
      "epoch": 3.151702880859375e-05,
      "model_forward_time": 3.136735677719116,
      "step": 8262
    },
    {
      "epoch": 3.151702880859375e-05,
      "step": 8262,
      "training_step_time": 10.266361951828003
    },
    {
      "epoch": 3.1520843505859375e-05,
      "model_forward_time": 3.146271228790283,
      "step": 8263
    },
    {
      "epoch": 3.1520843505859375e-05,
      "step": 8263,
      "training_step_time": 9.923811674118042
    },
    {
      "epoch": 3.1520843505859375e-05,
      "model_forward_time": 3.1334781646728516,
      "step": 8263
    },
    {
      "epoch": 3.1520843505859375e-05,
      "step": 8263,
      "training_step_time": 9.90536379814148
    },
    {
      "epoch": 3.1520843505859375e-05,
      "model_forward_time": 3.1371023654937744,
      "step": 8263
    },
    {
      "epoch": 3.1520843505859375e-05,
      "step": 8263,
      "training_step_time": 9.909718990325928
    },
    {
      "epoch": 3.1520843505859375e-05,
      "model_forward_time": 3.4239048957824707,
      "step": 8263
    },
    {
      "epoch": 3.1520843505859375e-05,
      "step": 8263,
      "training_step_time": 10.290889024734497
    },
    {
      "epoch": 3.1524658203125e-05,
      "model_forward_time": 3.1584362983703613,
      "step": 8264
    },
    {
      "epoch": 3.1524658203125e-05,
      "step": 8264,
      "training_step_time": 9.933774471282959
    },
    {
      "epoch": 3.1524658203125e-05,
      "model_forward_time": 3.135836124420166,
      "step": 8264
    },
    {
      "epoch": 3.1524658203125e-05,
      "step": 8264,
      "training_step_time": 9.90769362449646
    },
    {
      "epoch": 3.1524658203125e-05,
      "model_forward_time": 3.140734910964966,
      "step": 8264
    },
    {
      "epoch": 3.1524658203125e-05,
      "step": 8264,
      "training_step_time": 9.915883302688599
    },
    {
      "epoch": 3.1524658203125e-05,
      "model_forward_time": 3.1360890865325928,
      "step": 8264
    },
    {
      "epoch": 3.1524658203125e-05,
      "step": 8264,
      "training_step_time": 10.01098346710205
    },
    {
      "epoch": 3.152847290039063e-05,
      "model_forward_time": 3.1451919078826904,
      "step": 8265
    },
    {
      "dynamics_loss_avg": 0.0650742881000042,
      "epoch": 3.152847290039063e-05,
      "step": 8265
    },
    {
      "action_loss_avg": 0.025533493608236313,
      "epoch": 3.152847290039063e-05,
      "step": 8265
    },
    {
      "epoch": 3.152847290039063e-05,
      "step": 8265,
      "training_step_time": 10.195659160614014
    },
    {
      "epoch": 3.152847290039063e-05,
      "model_forward_time": 3.139650344848633,
      "step": 8265
    },
    {
      "epoch": 3.152847290039063e-05,
      "step": 8265,
      "training_step_time": 9.910583972930908
    },
    {
      "epoch": 3.152847290039063e-05,
      "model_forward_time": 3.139272928237915,
      "step": 8265
    },
    {
      "epoch": 3.152847290039063e-05,
      "step": 8265,
      "training_step_time": 9.910402536392212
    },
    {
      "epoch": 3.152847290039063e-05,
      "model_forward_time": 3.1376123428344727,
      "step": 8265
    },
    {
      "epoch": 3.152847290039063e-05,
      "step": 8265,
      "training_step_time": 10.298964023590088
    },
    {
      "epoch": 3.153228759765625e-05,
      "model_forward_time": 2.448194742202759,
      "step": 8266
    },
    {
      "epoch": 3.153228759765625e-05,
      "step": 8266,
      "training_step_time": 7.4751060009002686
    },
    {
      "epoch": 3.153228759765625e-05,
      "model_forward_time": 3.393571376800537,
      "step": 8266
    },
    {
      "epoch": 3.153228759765625e-05,
      "step": 8266,
      "training_step_time": 10.161731243133545
    },
    {
      "epoch": 3.153228759765625e-05,
      "model_forward_time": 3.145655393600464,
      "step": 8266
    },
    {
      "epoch": 3.153228759765625e-05,
      "step": 8266,
      "training_step_time": 9.923195362091064
    },
    {
      "epoch": 3.153228759765625e-05,
      "model_forward_time": 3.1401405334472656,
      "step": 8266
    },
    {
      "epoch": 3.153228759765625e-05,
      "step": 8266,
      "training_step_time": 10.313317775726318
    },
    {
      "epoch": 3.153610229492187e-05,
      "model_forward_time": 3.1614623069763184,
      "step": 8267
    },
    {
      "epoch": 3.153610229492187e-05,
      "step": 8267,
      "training_step_time": 9.938700437545776
    },
    {
      "epoch": 3.153610229492187e-05,
      "model_forward_time": 3.1452200412750244,
      "step": 8267
    },
    {
      "epoch": 3.153610229492187e-05,
      "step": 8267,
      "training_step_time": 9.918598175048828
    },
    {
      "epoch": 3.153610229492187e-05,
      "model_forward_time": 3.1442275047302246,
      "step": 8267
    },
    {
      "dynamics_loss_avg": 0.05519727822393179,
      "epoch": 3.153610229492187e-05,
      "step": 8267
    },
    {
      "action_loss_avg": 0.01632747775875032,
      "epoch": 3.153610229492187e-05,
      "step": 8267
    },
    {
      "epoch": 3.153610229492187e-05,
      "step": 8267,
      "training_step_time": 10.179400444030762
    },
    {
      "epoch": 3.153610229492187e-05,
      "model_forward_time": 3.140676975250244,
      "step": 8267
    },
    {
      "epoch": 3.153610229492187e-05,
      "step": 8267,
      "training_step_time": 10.019489288330078
    },
    {
      "epoch": 3.15399169921875e-05,
      "model_forward_time": 3.1490581035614014,
      "step": 8268
    },
    {
      "epoch": 3.15399169921875e-05,
      "step": 8268,
      "training_step_time": 9.924880504608154
    },
    {
      "epoch": 3.15399169921875e-05,
      "model_forward_time": 3.1395974159240723,
      "step": 8268
    },
    {
      "epoch": 3.15399169921875e-05,
      "step": 8268,
      "training_step_time": 9.909626722335815
    },
    {
      "epoch": 3.15399169921875e-05,
      "model_forward_time": 3.1359012126922607,
      "step": 8268
    },
    {
      "epoch": 3.15399169921875e-05,
      "step": 8268,
      "training_step_time": 9.905362844467163
    },
    {
      "epoch": 3.15399169921875e-05,
      "model_forward_time": 3.4019315242767334,
      "step": 8268
    },
    {
      "epoch": 3.15399169921875e-05,
      "step": 8268,
      "training_step_time": 10.295615434646606
    },
    {
      "epoch": 3.1543731689453125e-05,
      "model_forward_time": 3.145622491836548,
      "step": 8269
    },
    {
      "epoch": 3.1543731689453125e-05,
      "step": 8269,
      "training_step_time": 9.914862632751465
    },
    {
      "epoch": 3.1543731689453125e-05,
      "model_forward_time": 3.1392171382904053,
      "step": 8269
    },
    {
      "epoch": 3.1543731689453125e-05,
      "step": 8269,
      "training_step_time": 9.910185098648071
    },
    {
      "epoch": 3.1543731689453125e-05,
      "model_forward_time": 3.1350865364074707,
      "step": 8269
    },
    {
      "epoch": 3.1543731689453125e-05,
      "step": 8269,
      "training_step_time": 9.901259899139404
    },
    {
      "epoch": 3.1543731689453125e-05,
      "model_forward_time": 3.13822078704834,
      "step": 8269
    },
    {
      "epoch": 3.1543731689453125e-05,
      "step": 8269,
      "training_step_time": 10.00223994255066
    },
    {
      "epoch": 3.154754638671875e-05,
      "grad_norm": 0.3086850047111511,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 8270
    },
    {
      "epoch": 3.154754638671875e-05,
      "model_forward_time": 3.4290547370910645,
      "step": 8270
    },
    {
      "dynamics_loss_avg": 0.05737053062766791,
      "epoch": 3.154754638671875e-05,
      "step": 8270
    },
    {
      "action_loss_avg": 0.019967066403478385,
      "epoch": 3.154754638671875e-05,
      "step": 8270
    },
    {
      "epoch": 3.154754638671875e-05,
      "step": 8270,
      "training_step_time": 10.199831008911133
    },
    {
      "epoch": 3.154754638671875e-05,
      "model_forward_time": 3.136195182800293,
      "step": 8270
    },
    {
      "epoch": 3.154754638671875e-05,
      "step": 8270,
      "training_step_time": 9.905070066452026
    },
    {
      "epoch": 3.154754638671875e-05,
      "model_forward_time": 3.1358699798583984,
      "step": 8270
    },
    {
      "epoch": 3.154754638671875e-05,
      "step": 8270,
      "training_step_time": 9.905250310897827
    },
    {
      "epoch": 3.154754638671875e-05,
      "model_forward_time": 3.141407012939453,
      "step": 8270
    },
    {
      "epoch": 3.154754638671875e-05,
      "step": 8270,
      "training_step_time": 10.286765336990356
    },
    {
      "epoch": 3.155136108398438e-05,
      "model_forward_time": 3.1524956226348877,
      "step": 8271
    },
    {
      "epoch": 3.155136108398438e-05,
      "step": 8271,
      "training_step_time": 9.92791748046875
    },
    {
      "epoch": 3.155136108398438e-05,
      "model_forward_time": 3.426361083984375,
      "step": 8271
    },
    {
      "epoch": 3.155136108398438e-05,
      "step": 8271,
      "training_step_time": 10.201941013336182
    },
    {
      "epoch": 3.155136108398438e-05,
      "model_forward_time": 3.1395647525787354,
      "step": 8271
    },
    {
      "epoch": 3.155136108398438e-05,
      "step": 8271,
      "training_step_time": 9.917065143585205
    },
    {
      "epoch": 3.155136108398438e-05,
      "model_forward_time": 3.1428062915802,
      "step": 8271
    },
    {
      "epoch": 3.155136108398438e-05,
      "step": 8271,
      "training_step_time": 10.293673276901245
    },
    {
      "epoch": 3.155517578125e-05,
      "model_forward_time": 3.1510910987854004,
      "step": 8272
    },
    {
      "epoch": 3.155517578125e-05,
      "step": 8272,
      "training_step_time": 9.932146549224854
    },
    {
      "epoch": 3.155517578125e-05,
      "model_forward_time": 3.1422789096832275,
      "step": 8272
    },
    {
      "epoch": 3.155517578125e-05,
      "step": 8272,
      "training_step_time": 9.921967029571533
    },
    {
      "epoch": 3.155517578125e-05,
      "model_forward_time": 3.4114458560943604,
      "step": 8272
    },
    {
      "dynamics_loss_avg": 0.06015328951179981,
      "epoch": 3.155517578125e-05,
      "step": 8272
    },
    {
      "action_loss_avg": 0.0202771688811481,
      "epoch": 3.155517578125e-05,
      "step": 8272
    },
    {
      "epoch": 3.155517578125e-05,
      "step": 8272,
      "training_step_time": 10.191621780395508
    },
    {
      "epoch": 3.155517578125e-05,
      "model_forward_time": 3.1413214206695557,
      "step": 8272
    },
    {
      "epoch": 3.155517578125e-05,
      "step": 8272,
      "training_step_time": 10.02272629737854
    },
    {
      "epoch": 3.1558990478515624e-05,
      "model_forward_time": 3.1592659950256348,
      "step": 8273
    },
    {
      "epoch": 3.1558990478515624e-05,
      "step": 8273,
      "training_step_time": 9.934897184371948
    },
    {
      "epoch": 3.1558990478515624e-05,
      "model_forward_time": 3.1358370780944824,
      "step": 8273
    },
    {
      "epoch": 3.1558990478515624e-05,
      "step": 8273,
      "training_step_time": 9.905186653137207
    },
    {
      "epoch": 3.1558990478515624e-05,
      "model_forward_time": 3.1376538276672363,
      "step": 8273
    },
    {
      "epoch": 3.1558990478515624e-05,
      "step": 8273,
      "training_step_time": 9.908114671707153
    },
    {
      "epoch": 3.1558990478515624e-05,
      "model_forward_time": 3.4183928966522217,
      "step": 8273
    },
    {
      "epoch": 3.1558990478515624e-05,
      "step": 8273,
      "training_step_time": 10.313626766204834
    },
    {
      "epoch": 3.156280517578125e-05,
      "model_forward_time": 3.1547436714172363,
      "step": 8274
    },
    {
      "epoch": 3.156280517578125e-05,
      "step": 8274,
      "training_step_time": 9.92887806892395
    },
    {
      "epoch": 3.156280517578125e-05,
      "model_forward_time": 3.1342966556549072,
      "step": 8274
    },
    {
      "epoch": 3.156280517578125e-05,
      "step": 8274,
      "training_step_time": 9.902040958404541
    },
    {
      "epoch": 3.156280517578125e-05,
      "model_forward_time": 3.138077735900879,
      "step": 8274
    },
    {
      "epoch": 3.156280517578125e-05,
      "step": 8274,
      "training_step_time": 9.909714221954346
    },
    {
      "epoch": 3.156280517578125e-05,
      "model_forward_time": 3.132786273956299,
      "step": 8274
    },
    {
      "epoch": 3.156280517578125e-05,
      "step": 8274,
      "training_step_time": 10.274107694625854
    },
    {
      "epoch": 3.1566619873046876e-05,
      "model_forward_time": 3.427117347717285,
      "step": 8275
    },
    {
      "dynamics_loss_avg": 0.0555116007104516,
      "epoch": 3.1566619873046876e-05,
      "step": 8275
    },
    {
      "action_loss_avg": 0.02201068205758929,
      "epoch": 3.1566619873046876e-05,
      "step": 8275
    },
    {
      "epoch": 3.1566619873046876e-05,
      "step": 8275,
      "training_step_time": 10.20050573348999
    },
    {
      "epoch": 3.1566619873046876e-05,
      "model_forward_time": 3.130764961242676,
      "step": 8275
    },
    {
      "epoch": 3.1566619873046876e-05,
      "step": 8275,
      "training_step_time": 9.898681402206421
    },
    {
      "epoch": 3.1566619873046876e-05,
      "model_forward_time": 3.1378610134124756,
      "step": 8275
    },
    {
      "epoch": 3.1566619873046876e-05,
      "step": 8275,
      "training_step_time": 9.904738426208496
    },
    {
      "epoch": 3.1566619873046876e-05,
      "model_forward_time": 3.134373664855957,
      "step": 8275
    },
    {
      "epoch": 3.1566619873046876e-05,
      "step": 8275,
      "training_step_time": 10.289769172668457
    },
    {
      "epoch": 3.15704345703125e-05,
      "model_forward_time": 3.156123161315918,
      "step": 8276
    },
    {
      "epoch": 3.15704345703125e-05,
      "step": 8276,
      "training_step_time": 9.930912733078003
    },
    {
      "epoch": 3.15704345703125e-05,
      "model_forward_time": 3.3911702632904053,
      "step": 8276
    },
    {
      "epoch": 3.15704345703125e-05,
      "step": 8276,
      "training_step_time": 10.16792345046997
    },
    {
      "epoch": 3.15704345703125e-05,
      "model_forward_time": 3.144275188446045,
      "step": 8276
    },
    {
      "epoch": 3.15704345703125e-05,
      "step": 8276,
      "training_step_time": 9.92313838005066
    },
    {
      "epoch": 3.15704345703125e-05,
      "model_forward_time": 3.1421334743499756,
      "step": 8276
    },
    {
      "epoch": 3.15704345703125e-05,
      "step": 8276,
      "training_step_time": 10.32595443725586
    },
    {
      "epoch": 3.157424926757812e-05,
      "model_forward_time": 3.157560110092163,
      "step": 8277
    },
    {
      "epoch": 3.157424926757812e-05,
      "step": 8277,
      "training_step_time": 9.939726829528809
    },
    {
      "epoch": 3.157424926757812e-05,
      "model_forward_time": 3.140934705734253,
      "step": 8277
    },
    {
      "epoch": 3.157424926757812e-05,
      "step": 8277,
      "training_step_time": 9.914461851119995
    },
    {
      "epoch": 3.157424926757812e-05,
      "model_forward_time": 3.4169533252716064,
      "step": 8277
    },
    {
      "dynamics_loss_avg": 0.06740055754780769,
      "epoch": 3.157424926757812e-05,
      "step": 8277
    },
    {
      "action_loss_avg": 0.02285967655479908,
      "epoch": 3.157424926757812e-05,
      "step": 8277
    },
    {
      "epoch": 3.157424926757812e-05,
      "step": 8277,
      "training_step_time": 10.19452714920044
    },
    {
      "epoch": 3.157424926757812e-05,
      "model_forward_time": 3.13866925239563,
      "step": 8277
    },
    {
      "epoch": 3.157424926757812e-05,
      "step": 8277,
      "training_step_time": 10.066139459609985
    },
    {
      "epoch": 3.157806396484375e-05,
      "model_forward_time": 3.154289722442627,
      "step": 8278
    },
    {
      "epoch": 3.157806396484375e-05,
      "step": 8278,
      "training_step_time": 9.924579381942749
    },
    {
      "epoch": 3.157806396484375e-05,
      "model_forward_time": 3.143969774246216,
      "step": 8278
    },
    {
      "epoch": 3.157806396484375e-05,
      "step": 8278,
      "training_step_time": 9.93211054801941
    },
    {
      "epoch": 3.157806396484375e-05,
      "model_forward_time": 3.1445841789245605,
      "step": 8278
    },
    {
      "epoch": 3.157806396484375e-05,
      "step": 8278,
      "training_step_time": 9.91940975189209
    },
    {
      "epoch": 3.157806396484375e-05,
      "model_forward_time": 3.439450979232788,
      "step": 8278
    },
    {
      "epoch": 3.157806396484375e-05,
      "step": 8278,
      "training_step_time": 10.298314809799194
    },
    {
      "epoch": 3.1581878662109375e-05,
      "model_forward_time": 3.144352912902832,
      "step": 8279
    },
    {
      "epoch": 3.1581878662109375e-05,
      "step": 8279,
      "training_step_time": 9.913525104522705
    },
    {
      "epoch": 3.1581878662109375e-05,
      "model_forward_time": 3.145700693130493,
      "step": 8279
    },
    {
      "epoch": 3.1581878662109375e-05,
      "step": 8279,
      "training_step_time": 9.916712045669556
    },
    {
      "epoch": 3.1581878662109375e-05,
      "model_forward_time": 3.1379642486572266,
      "step": 8279
    },
    {
      "epoch": 3.1581878662109375e-05,
      "step": 8279,
      "training_step_time": 9.908153772354126
    },
    {
      "epoch": 3.1581878662109375e-05,
      "model_forward_time": 3.141242027282715,
      "step": 8279
    },
    {
      "epoch": 3.1581878662109375e-05,
      "step": 8279,
      "training_step_time": 10.287140369415283
    },
    {
      "epoch": 3.1585693359375e-05,
      "grad_norm": 0.48000553250312805,
      "learning_rate": 1e-05,
      "loss": 0.0821,
      "step": 8280
    },
    {
      "epoch": 3.1585693359375e-05,
      "model_forward_time": 3.419586181640625,
      "step": 8280
    },
    {
      "dynamics_loss_avg": 0.07070453390479088,
      "epoch": 3.1585693359375e-05,
      "step": 8280
    },
    {
      "action_loss_avg": 0.023301988560706376,
      "epoch": 3.1585693359375e-05,
      "step": 8280
    },
    {
      "epoch": 3.1585693359375e-05,
      "step": 8280,
      "training_step_time": 10.200466394424438
    },
    {
      "epoch": 3.1585693359375e-05,
      "model_forward_time": 3.145395040512085,
      "step": 8280
    },
    {
      "epoch": 3.1585693359375e-05,
      "step": 8280,
      "training_step_time": 9.918852090835571
    },
    {
      "epoch": 3.1585693359375e-05,
      "model_forward_time": 3.140453577041626,
      "step": 8280
    },
    {
      "epoch": 3.1585693359375e-05,
      "step": 8280,
      "training_step_time": 9.918736219406128
    },
    {
      "epoch": 3.1585693359375e-05,
      "model_forward_time": 3.1495535373687744,
      "step": 8280
    },
    {
      "epoch": 3.1585693359375e-05,
      "step": 8280,
      "training_step_time": 10.29319953918457
    },
    {
      "epoch": 3.158950805664063e-05,
      "model_forward_time": 3.1637322902679443,
      "step": 8281
    },
    {
      "epoch": 3.158950805664063e-05,
      "step": 8281,
      "training_step_time": 9.941054105758667
    },
    {
      "epoch": 3.158950805664063e-05,
      "model_forward_time": 3.4486382007598877,
      "step": 8281
    },
    {
      "epoch": 3.158950805664063e-05,
      "step": 8281,
      "training_step_time": 10.225494384765625
    },
    {
      "epoch": 3.158950805664063e-05,
      "model_forward_time": 3.147550106048584,
      "step": 8281
    },
    {
      "epoch": 3.158950805664063e-05,
      "step": 8281,
      "training_step_time": 9.928142309188843
    },
    {
      "epoch": 3.158950805664063e-05,
      "model_forward_time": 2.4390687942504883,
      "step": 8281
    },
    {
      "epoch": 3.158950805664063e-05,
      "step": 8281,
      "training_step_time": 10.336188554763794
    },
    {
      "epoch": 3.159332275390625e-05,
      "model_forward_time": 3.1603591442108154,
      "step": 8282
    },
    {
      "epoch": 3.159332275390625e-05,
      "step": 8282,
      "training_step_time": 9.938162088394165
    },
    {
      "epoch": 3.159332275390625e-05,
      "model_forward_time": 3.142624855041504,
      "step": 8282
    },
    {
      "epoch": 3.159332275390625e-05,
      "step": 8282,
      "training_step_time": 9.919330596923828
    },
    {
      "epoch": 3.159332275390625e-05,
      "model_forward_time": 3.4024851322174072,
      "step": 8282
    },
    {
      "dynamics_loss_avg": 0.05282021258026361,
      "epoch": 3.159332275390625e-05,
      "step": 8282
    },
    {
      "action_loss_avg": 0.017517177807167172,
      "epoch": 3.159332275390625e-05,
      "step": 8282
    },
    {
      "epoch": 3.159332275390625e-05,
      "step": 8282,
      "training_step_time": 10.181143283843994
    },
    {
      "epoch": 3.159332275390625e-05,
      "model_forward_time": 3.1415021419525146,
      "step": 8282
    },
    {
      "epoch": 3.159332275390625e-05,
      "step": 8282,
      "training_step_time": 10.247333526611328
    },
    {
      "epoch": 3.159713745117187e-05,
      "model_forward_time": 3.1591858863830566,
      "step": 8283
    },
    {
      "epoch": 3.159713745117187e-05,
      "step": 8283,
      "training_step_time": 9.936139106750488
    },
    {
      "epoch": 3.159713745117187e-05,
      "model_forward_time": 3.137073516845703,
      "step": 8283
    },
    {
      "epoch": 3.159713745117187e-05,
      "step": 8283,
      "training_step_time": 9.914363145828247
    },
    {
      "epoch": 3.159713745117187e-05,
      "model_forward_time": 3.145683765411377,
      "step": 8283
    },
    {
      "epoch": 3.159713745117187e-05,
      "step": 8283,
      "training_step_time": 9.920877933502197
    },
    {
      "epoch": 3.159713745117187e-05,
      "model_forward_time": 3.4126811027526855,
      "step": 8283
    },
    {
      "epoch": 3.159713745117187e-05,
      "step": 8283,
      "training_step_time": 10.275477409362793
    },
    {
      "epoch": 3.16009521484375e-05,
      "model_forward_time": 3.145033836364746,
      "step": 8284
    },
    {
      "epoch": 3.16009521484375e-05,
      "step": 8284,
      "training_step_time": 9.919641733169556
    },
    {
      "epoch": 3.16009521484375e-05,
      "model_forward_time": 3.1365439891815186,
      "step": 8284
    },
    {
      "epoch": 3.16009521484375e-05,
      "step": 8284,
      "training_step_time": 9.899877309799194
    },
    {
      "epoch": 3.16009521484375e-05,
      "model_forward_time": 3.1372299194335938,
      "step": 8284
    },
    {
      "epoch": 3.16009521484375e-05,
      "step": 8284,
      "training_step_time": 9.904052495956421
    },
    {
      "epoch": 3.16009521484375e-05,
      "model_forward_time": 3.1341300010681152,
      "step": 8284
    },
    {
      "epoch": 3.16009521484375e-05,
      "step": 8284,
      "training_step_time": 10.315354108810425
    },
    {
      "epoch": 3.1604766845703126e-05,
      "model_forward_time": 3.4798743724823,
      "step": 8285
    },
    {
      "dynamics_loss_avg": 0.07807974983006716,
      "epoch": 3.1604766845703126e-05,
      "step": 8285
    },
    {
      "action_loss_avg": 0.03444499708712101,
      "epoch": 3.1604766845703126e-05,
      "step": 8285
    },
    {
      "epoch": 3.1604766845703126e-05,
      "step": 8285,
      "training_step_time": 10.258502960205078
    },
    {
      "epoch": 3.1604766845703126e-05,
      "model_forward_time": 3.1398885250091553,
      "step": 8285
    },
    {
      "epoch": 3.1604766845703126e-05,
      "step": 8285,
      "training_step_time": 9.907999038696289
    },
    {
      "epoch": 3.1604766845703126e-05,
      "model_forward_time": 3.144101858139038,
      "step": 8285
    },
    {
      "epoch": 3.1604766845703126e-05,
      "step": 8285,
      "training_step_time": 9.917814254760742
    },
    {
      "epoch": 3.1604766845703126e-05,
      "model_forward_time": 3.1395204067230225,
      "step": 8285
    },
    {
      "epoch": 3.1604766845703126e-05,
      "step": 8285,
      "training_step_time": 10.013564586639404
    },
    {
      "epoch": 3.160858154296875e-05,
      "model_forward_time": 3.1526763439178467,
      "step": 8286
    },
    {
      "epoch": 3.160858154296875e-05,
      "step": 8286,
      "training_step_time": 9.9251708984375
    },
    {
      "epoch": 3.160858154296875e-05,
      "model_forward_time": 3.4040286540985107,
      "step": 8286
    },
    {
      "epoch": 3.160858154296875e-05,
      "step": 8286,
      "training_step_time": 10.177331924438477
    },
    {
      "epoch": 3.160858154296875e-05,
      "model_forward_time": 3.1459920406341553,
      "step": 8286
    },
    {
      "epoch": 3.160858154296875e-05,
      "step": 8286,
      "training_step_time": 9.919299602508545
    },
    {
      "epoch": 3.160858154296875e-05,
      "model_forward_time": 3.141347646713257,
      "step": 8286
    },
    {
      "epoch": 3.160858154296875e-05,
      "step": 8286,
      "training_step_time": 10.269964694976807
    },
    {
      "epoch": 3.161239624023438e-05,
      "model_forward_time": 3.1693274974823,
      "step": 8287
    },
    {
      "epoch": 3.161239624023438e-05,
      "step": 8287,
      "training_step_time": 9.943798542022705
    },
    {
      "epoch": 3.161239624023438e-05,
      "model_forward_time": 3.1430623531341553,
      "step": 8287
    },
    {
      "epoch": 3.161239624023438e-05,
      "step": 8287,
      "training_step_time": 9.910384893417358
    },
    {
      "epoch": 3.161239624023438e-05,
      "model_forward_time": 3.4263923168182373,
      "step": 8287
    },
    {
      "dynamics_loss_avg": 0.07178214639425277,
      "epoch": 3.161239624023438e-05,
      "step": 8287
    },
    {
      "action_loss_avg": 0.02199704209342599,
      "epoch": 3.161239624023438e-05,
      "step": 8287
    },
    {
      "epoch": 3.161239624023438e-05,
      "step": 8287,
      "training_step_time": 10.20113468170166
    },
    {
      "epoch": 3.161239624023438e-05,
      "model_forward_time": 3.138468027114868,
      "step": 8287
    },
    {
      "epoch": 3.161239624023438e-05,
      "step": 8287,
      "training_step_time": 10.23693060874939
    },
    {
      "epoch": 3.16162109375e-05,
      "model_forward_time": 3.160536050796509,
      "step": 8288
    },
    {
      "epoch": 3.16162109375e-05,
      "step": 8288,
      "training_step_time": 9.932234525680542
    },
    {
      "epoch": 3.16162109375e-05,
      "model_forward_time": 3.1454532146453857,
      "step": 8288
    },
    {
      "epoch": 3.16162109375e-05,
      "step": 8288,
      "training_step_time": 9.920417785644531
    },
    {
      "epoch": 3.16162109375e-05,
      "model_forward_time": 3.1395933628082275,
      "step": 8288
    },
    {
      "epoch": 3.16162109375e-05,
      "step": 8288,
      "training_step_time": 9.903061866760254
    },
    {
      "epoch": 3.16162109375e-05,
      "model_forward_time": 3.4324803352355957,
      "step": 8288
    },
    {
      "epoch": 3.16162109375e-05,
      "step": 8288,
      "training_step_time": 10.30101203918457
    },
    {
      "epoch": 3.1620025634765624e-05,
      "model_forward_time": 3.169509172439575,
      "step": 8289
    },
    {
      "epoch": 3.1620025634765624e-05,
      "step": 8289,
      "training_step_time": 9.951884984970093
    },
    {
      "epoch": 3.1620025634765624e-05,
      "model_forward_time": 3.14516544342041,
      "step": 8289
    },
    {
      "epoch": 3.1620025634765624e-05,
      "step": 8289,
      "training_step_time": 9.922404289245605
    },
    {
      "epoch": 3.1620025634765624e-05,
      "model_forward_time": 3.149867534637451,
      "step": 8289
    },
    {
      "epoch": 3.1620025634765624e-05,
      "step": 8289,
      "training_step_time": 9.930685043334961
    },
    {
      "epoch": 3.1620025634765624e-05,
      "model_forward_time": 3.147953510284424,
      "step": 8289
    },
    {
      "epoch": 3.1620025634765624e-05,
      "step": 8289,
      "training_step_time": 10.295962810516357
    },
    {
      "epoch": 3.162384033203125e-05,
      "grad_norm": 0.5399959683418274,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 8290
    },
    {
      "epoch": 3.162384033203125e-05,
      "model_forward_time": 3.413968801498413,
      "step": 8290
    },
    {
      "dynamics_loss_avg": 0.06168823484331369,
      "epoch": 3.162384033203125e-05,
      "step": 8290
    },
    {
      "action_loss_avg": 0.026347340596839786,
      "epoch": 3.162384033203125e-05,
      "step": 8290
    },
    {
      "epoch": 3.162384033203125e-05,
      "step": 8290,
      "training_step_time": 10.193655729293823
    },
    {
      "epoch": 3.162384033203125e-05,
      "model_forward_time": 3.147050380706787,
      "step": 8290
    },
    {
      "epoch": 3.162384033203125e-05,
      "step": 8290,
      "training_step_time": 9.929043769836426
    },
    {
      "epoch": 3.162384033203125e-05,
      "model_forward_time": 3.144852638244629,
      "step": 8290
    },
    {
      "epoch": 3.162384033203125e-05,
      "step": 8290,
      "training_step_time": 9.923233032226562
    },
    {
      "epoch": 3.162384033203125e-05,
      "model_forward_time": 3.1476988792419434,
      "step": 8290
    },
    {
      "epoch": 3.162384033203125e-05,
      "step": 8290,
      "training_step_time": 10.291939496994019
    },
    {
      "epoch": 3.1627655029296876e-05,
      "model_forward_time": 3.1594927310943604,
      "step": 8291
    },
    {
      "epoch": 3.1627655029296876e-05,
      "step": 8291,
      "training_step_time": 9.945690155029297
    },
    {
      "epoch": 3.1627655029296876e-05,
      "model_forward_time": 3.423332929611206,
      "step": 8291
    },
    {
      "epoch": 3.1627655029296876e-05,
      "step": 8291,
      "training_step_time": 10.207434892654419
    },
    {
      "epoch": 3.1627655029296876e-05,
      "model_forward_time": 3.147221803665161,
      "step": 8291
    },
    {
      "epoch": 3.1627655029296876e-05,
      "step": 8291,
      "training_step_time": 9.926567554473877
    },
    {
      "epoch": 3.1627655029296876e-05,
      "model_forward_time": 3.151858329772949,
      "step": 8291
    },
    {
      "epoch": 3.1627655029296876e-05,
      "step": 8291,
      "training_step_time": 10.246243238449097
    },
    {
      "epoch": 3.16314697265625e-05,
      "model_forward_time": 3.1665680408477783,
      "step": 8292
    },
    {
      "epoch": 3.16314697265625e-05,
      "step": 8292,
      "training_step_time": 9.941184520721436
    },
    {
      "epoch": 3.16314697265625e-05,
      "model_forward_time": 3.140486240386963,
      "step": 8292
    },
    {
      "epoch": 3.16314697265625e-05,
      "step": 8292,
      "training_step_time": 9.915609359741211
    },
    {
      "epoch": 3.16314697265625e-05,
      "model_forward_time": 3.3963897228240967,
      "step": 8292
    },
    {
      "dynamics_loss_avg": 0.060661381483078,
      "epoch": 3.16314697265625e-05,
      "step": 8292
    },
    {
      "action_loss_avg": 0.024265051074326038,
      "epoch": 3.16314697265625e-05,
      "step": 8292
    },
    {
      "epoch": 3.16314697265625e-05,
      "step": 8292,
      "training_step_time": 10.163767099380493
    },
    {
      "epoch": 3.16314697265625e-05,
      "model_forward_time": 3.147794485092163,
      "step": 8292
    },
    {
      "epoch": 3.16314697265625e-05,
      "step": 8292,
      "training_step_time": 10.24994444847107
    },
    {
      "epoch": 3.163528442382812e-05,
      "model_forward_time": 3.1619975566864014,
      "step": 8293
    },
    {
      "epoch": 3.163528442382812e-05,
      "step": 8293,
      "training_step_time": 9.92812466621399
    },
    {
      "epoch": 3.163528442382812e-05,
      "model_forward_time": 3.1399903297424316,
      "step": 8293
    },
    {
      "epoch": 3.163528442382812e-05,
      "step": 8293,
      "training_step_time": 9.912216186523438
    },
    {
      "epoch": 3.163528442382812e-05,
      "model_forward_time": 3.142259359359741,
      "step": 8293
    },
    {
      "epoch": 3.163528442382812e-05,
      "step": 8293,
      "training_step_time": 9.910377740859985
    },
    {
      "epoch": 3.163528442382812e-05,
      "model_forward_time": 3.4128453731536865,
      "step": 8293
    },
    {
      "epoch": 3.163528442382812e-05,
      "step": 8293,
      "training_step_time": 10.281256675720215
    },
    {
      "epoch": 3.163909912109375e-05,
      "model_forward_time": 3.1613388061523438,
      "step": 8294
    },
    {
      "epoch": 3.163909912109375e-05,
      "step": 8294,
      "training_step_time": 9.938040733337402
    },
    {
      "epoch": 3.163909912109375e-05,
      "model_forward_time": 3.1419785022735596,
      "step": 8294
    },
    {
      "epoch": 3.163909912109375e-05,
      "step": 8294,
      "training_step_time": 9.914364099502563
    },
    {
      "epoch": 3.163909912109375e-05,
      "model_forward_time": 3.1469178199768066,
      "step": 8294
    },
    {
      "epoch": 3.163909912109375e-05,
      "step": 8294,
      "training_step_time": 9.922353506088257
    },
    {
      "epoch": 3.163909912109375e-05,
      "model_forward_time": 3.144789218902588,
      "step": 8294
    },
    {
      "epoch": 3.163909912109375e-05,
      "step": 8294,
      "training_step_time": 10.305605411529541
    },
    {
      "epoch": 3.1642913818359375e-05,
      "model_forward_time": 3.1593658924102783,
      "step": 8295
    },
    {
      "dynamics_loss_avg": 0.06431246846914292,
      "epoch": 3.1642913818359375e-05,
      "step": 8295
    },
    {
      "action_loss_avg": 0.02316054394468665,
      "epoch": 3.1642913818359375e-05,
      "step": 8295
    },
    {
      "epoch": 3.1642913818359375e-05,
      "step": 8295,
      "training_step_time": 9.934648752212524
    },
    {
      "epoch": 3.1642913818359375e-05,
      "model_forward_time": 3.1403942108154297,
      "step": 8295
    },
    {
      "epoch": 3.1642913818359375e-05,
      "step": 8295,
      "training_step_time": 9.910644769668579
    },
    {
      "epoch": 3.1642913818359375e-05,
      "model_forward_time": 3.144131898880005,
      "step": 8295
    },
    {
      "epoch": 3.1642913818359375e-05,
      "step": 8295,
      "training_step_time": 9.91288685798645
    },
    {
      "epoch": 3.1642913818359375e-05,
      "model_forward_time": 3.1393580436706543,
      "step": 8295
    },
    {
      "epoch": 3.1642913818359375e-05,
      "step": 8295,
      "training_step_time": 10.312717914581299
    },
    {
      "epoch": 3.1646728515625e-05,
      "model_forward_time": 3.1580424308776855,
      "step": 8296
    },
    {
      "epoch": 3.1646728515625e-05,
      "step": 8296,
      "training_step_time": 10.221992254257202
    },
    {
      "epoch": 3.1646728515625e-05,
      "model_forward_time": 3.1368839740753174,
      "step": 8296
    },
    {
      "epoch": 3.1646728515625e-05,
      "step": 8296,
      "training_step_time": 9.904457569122314
    },
    {
      "epoch": 3.1646728515625e-05,
      "model_forward_time": 3.142885208129883,
      "step": 8296
    },
    {
      "epoch": 3.1646728515625e-05,
      "step": 8296,
      "training_step_time": 9.911555767059326
    },
    {
      "epoch": 3.1646728515625e-05,
      "model_forward_time": 3.1415867805480957,
      "step": 8296
    },
    {
      "epoch": 3.1646728515625e-05,
      "step": 8296,
      "training_step_time": 10.345951318740845
    },
    {
      "epoch": 3.165054321289063e-05,
      "model_forward_time": 3.1588361263275146,
      "step": 8297
    },
    {
      "epoch": 3.165054321289063e-05,
      "step": 8297,
      "training_step_time": 9.935403823852539
    },
    {
      "epoch": 3.165054321289063e-05,
      "model_forward_time": 3.390885353088379,
      "step": 8297
    },
    {
      "epoch": 3.165054321289063e-05,
      "step": 8297,
      "training_step_time": 10.166046619415283
    },
    {
      "epoch": 3.165054321289063e-05,
      "model_forward_time": 3.138291120529175,
      "step": 8297
    },
    {
      "dynamics_loss_avg": 0.05750335603952408,
      "epoch": 3.165054321289063e-05,
      "step": 8297
    },
    {
      "action_loss_avg": 0.0206556532997638,
      "epoch": 3.165054321289063e-05,
      "step": 8297
    },
    {
      "epoch": 3.165054321289063e-05,
      "step": 8297,
      "training_step_time": 9.913570165634155
    },
    {
      "epoch": 3.165054321289063e-05,
      "model_forward_time": 3.1356029510498047,
      "step": 8297
    },
    {
      "epoch": 3.165054321289063e-05,
      "step": 8297,
      "training_step_time": 10.25158405303955
    },
    {
      "epoch": 3.165435791015625e-05,
      "model_forward_time": 3.143193006515503,
      "step": 8298
    },
    {
      "epoch": 3.165435791015625e-05,
      "step": 8298,
      "training_step_time": 9.916574001312256
    },
    {
      "epoch": 3.165435791015625e-05,
      "model_forward_time": 3.140363931655884,
      "step": 8298
    },
    {
      "epoch": 3.165435791015625e-05,
      "step": 8298,
      "training_step_time": 9.911136865615845
    },
    {
      "epoch": 3.165435791015625e-05,
      "model_forward_time": 3.4256136417388916,
      "step": 8298
    },
    {
      "epoch": 3.165435791015625e-05,
      "step": 8298,
      "training_step_time": 10.1997652053833
    },
    {
      "epoch": 3.165435791015625e-05,
      "model_forward_time": 2.4335243701934814,
      "step": 8298
    },
    {
      "epoch": 3.165435791015625e-05,
      "step": 8298,
      "training_step_time": 10.27095627784729
    },
    {
      "epoch": 3.165817260742187e-05,
      "model_forward_time": 3.154977798461914,
      "step": 8299
    },
    {
      "epoch": 3.165817260742187e-05,
      "step": 8299,
      "training_step_time": 9.931882619857788
    },
    {
      "epoch": 3.165817260742187e-05,
      "model_forward_time": 3.137737512588501,
      "step": 8299
    },
    {
      "epoch": 3.165817260742187e-05,
      "step": 8299,
      "training_step_time": 9.908751726150513
    },
    {
      "epoch": 3.165817260742187e-05,
      "model_forward_time": 3.1408090591430664,
      "step": 8299
    },
    {
      "epoch": 3.165817260742187e-05,
      "step": 8299,
      "training_step_time": 9.91047215461731
    },
    {
      "epoch": 3.165817260742187e-05,
      "model_forward_time": 3.1449997425079346,
      "step": 8299
    },
    {
      "epoch": 3.165817260742187e-05,
      "step": 8299,
      "training_step_time": 10.287861108779907
    },
    {
      "epoch": 3.16619873046875e-05,
      "grad_norm": 0.48156532645225525,
      "learning_rate": 1e-05,
      "loss": 0.08,
      "step": 8300
    },
    {
      "epoch": 3.16619873046875e-05,
      "model_forward_time": 3.1621084213256836,
      "step": 8300
    },
    {
      "dynamics_loss_avg": 0.0659817574545741,
      "epoch": 3.16619873046875e-05,
      "step": 8300
    },
    {
      "action_loss_avg": 0.024892371986061333,
      "epoch": 3.16619873046875e-05,
      "step": 8300
    },
    {
      "epoch": 3.16619873046875e-05,
      "step": 8300,
      "training_step_time": 9.942204713821411
    },
    {
      "epoch": 3.16619873046875e-05,
      "model_forward_time": 3.1489741802215576,
      "step": 8300
    },
    {
      "epoch": 3.16619873046875e-05,
      "step": 8300,
      "training_step_time": 9.92332124710083
    },
    {
      "epoch": 3.16619873046875e-05,
      "model_forward_time": 3.1467676162719727,
      "step": 8300
    },
    {
      "epoch": 3.16619873046875e-05,
      "step": 8300,
      "training_step_time": 9.92228651046753
    },
    {
      "epoch": 3.16619873046875e-05,
      "model_forward_time": 3.1487274169921875,
      "step": 8300
    },
    {
      "epoch": 3.16619873046875e-05,
      "step": 8300,
      "training_step_time": 10.306610584259033
    },
    {
      "epoch": 3.1665802001953126e-05,
      "model_forward_time": 3.445540189743042,
      "step": 8301
    },
    {
      "epoch": 3.1665802001953126e-05,
      "step": 8301,
      "training_step_time": 10.237428188323975
    },
    {
      "epoch": 3.1665802001953126e-05,
      "model_forward_time": 3.1404519081115723,
      "step": 8301
    },
    {
      "epoch": 3.1665802001953126e-05,
      "step": 8301,
      "training_step_time": 9.91830563545227
    },
    {
      "epoch": 3.1665802001953126e-05,
      "model_forward_time": 3.1353187561035156,
      "step": 8301
    },
    {
      "epoch": 3.1665802001953126e-05,
      "step": 8301,
      "training_step_time": 9.914821147918701
    },
    {
      "epoch": 3.1665802001953126e-05,
      "model_forward_time": 3.1369876861572266,
      "step": 8301
    },
    {
      "epoch": 3.1665802001953126e-05,
      "step": 8301,
      "training_step_time": 10.290493965148926
    },
    {
      "epoch": 3.166961669921875e-05,
      "model_forward_time": 3.158236026763916,
      "step": 8302
    },
    {
      "epoch": 3.166961669921875e-05,
      "step": 8302,
      "training_step_time": 9.942614316940308
    },
    {
      "epoch": 3.166961669921875e-05,
      "model_forward_time": 3.136298656463623,
      "step": 8302
    },
    {
      "epoch": 3.166961669921875e-05,
      "step": 8302,
      "training_step_time": 10.230129480361938
    },
    {
      "epoch": 3.166961669921875e-05,
      "model_forward_time": 3.134979486465454,
      "step": 8302
    },
    {
      "dynamics_loss_avg": 0.06424418464303017,
      "epoch": 3.166961669921875e-05,
      "step": 8302
    },
    {
      "action_loss_avg": 0.02368770968168974,
      "epoch": 3.166961669921875e-05,
      "step": 8302
    },
    {
      "epoch": 3.166961669921875e-05,
      "step": 8302,
      "training_step_time": 9.909096717834473
    },
    {
      "epoch": 3.166961669921875e-05,
      "model_forward_time": 3.1407904624938965,
      "step": 8302
    },
    {
      "epoch": 3.166961669921875e-05,
      "step": 8302,
      "training_step_time": 10.238458395004272
    },
    {
      "epoch": 3.167343139648438e-05,
      "model_forward_time": 3.163933038711548,
      "step": 8303
    },
    {
      "epoch": 3.167343139648438e-05,
      "step": 8303,
      "training_step_time": 9.940767526626587
    },
    {
      "epoch": 3.167343139648438e-05,
      "model_forward_time": 3.1379940509796143,
      "step": 8303
    },
    {
      "epoch": 3.167343139648438e-05,
      "step": 8303,
      "training_step_time": 9.914656162261963
    },
    {
      "epoch": 3.167343139648438e-05,
      "model_forward_time": 3.4422903060913086,
      "step": 8303
    },
    {
      "epoch": 3.167343139648438e-05,
      "step": 8303,
      "training_step_time": 10.211717128753662
    },
    {
      "epoch": 3.167343139648438e-05,
      "model_forward_time": 3.1392650604248047,
      "step": 8303
    },
    {
      "epoch": 3.167343139648438e-05,
      "step": 8303,
      "training_step_time": 10.004865407943726
    },
    {
      "epoch": 3.167724609375e-05,
      "model_forward_time": 3.149352788925171,
      "step": 8304
    },
    {
      "epoch": 3.167724609375e-05,
      "step": 8304,
      "training_step_time": 9.925643682479858
    },
    {
      "epoch": 3.167724609375e-05,
      "model_forward_time": 3.1366145610809326,
      "step": 8304
    },
    {
      "epoch": 3.167724609375e-05,
      "step": 8304,
      "training_step_time": 9.91630506515503
    },
    {
      "epoch": 3.167724609375e-05,
      "model_forward_time": 3.1499853134155273,
      "step": 8304
    },
    {
      "epoch": 3.167724609375e-05,
      "step": 8304,
      "training_step_time": 9.92551565170288
    },
    {
      "epoch": 3.167724609375e-05,
      "model_forward_time": 3.1373417377471924,
      "step": 8304
    },
    {
      "epoch": 3.167724609375e-05,
      "step": 8304,
      "training_step_time": 10.29695987701416
    },
    {
      "epoch": 3.1681060791015624e-05,
      "model_forward_time": 3.1579720973968506,
      "step": 8305
    },
    {
      "dynamics_loss_avg": 0.07548449486494065,
      "epoch": 3.1681060791015624e-05,
      "step": 8305
    },
    {
      "action_loss_avg": 0.026586029306054117,
      "epoch": 3.1681060791015624e-05,
      "step": 8305
    },
    {
      "epoch": 3.1681060791015624e-05,
      "step": 8305,
      "training_step_time": 9.94637393951416
    },
    {
      "epoch": 3.1681060791015624e-05,
      "model_forward_time": 3.139376640319824,
      "step": 8305
    },
    {
      "epoch": 3.1681060791015624e-05,
      "step": 8305,
      "training_step_time": 9.921627759933472
    },
    {
      "epoch": 3.1681060791015624e-05,
      "model_forward_time": 3.1447370052337646,
      "step": 8305
    },
    {
      "epoch": 3.1681060791015624e-05,
      "step": 8305,
      "training_step_time": 9.926521062850952
    },
    {
      "epoch": 3.1681060791015624e-05,
      "model_forward_time": 3.1409146785736084,
      "step": 8305
    },
    {
      "epoch": 3.1681060791015624e-05,
      "step": 8305,
      "training_step_time": 10.24904465675354
    },
    {
      "epoch": 3.168487548828125e-05,
      "model_forward_time": 3.466179370880127,
      "step": 8306
    },
    {
      "epoch": 3.168487548828125e-05,
      "step": 8306,
      "training_step_time": 10.238330602645874
    },
    {
      "epoch": 3.168487548828125e-05,
      "model_forward_time": 3.132319927215576,
      "step": 8306
    },
    {
      "epoch": 3.168487548828125e-05,
      "step": 8306,
      "training_step_time": 9.902415990829468
    },
    {
      "epoch": 3.168487548828125e-05,
      "model_forward_time": 3.1386168003082275,
      "step": 8306
    },
    {
      "epoch": 3.168487548828125e-05,
      "step": 8306,
      "training_step_time": 9.9090256690979
    },
    {
      "epoch": 3.168487548828125e-05,
      "model_forward_time": 3.137122392654419,
      "step": 8306
    },
    {
      "epoch": 3.168487548828125e-05,
      "step": 8306,
      "training_step_time": 10.302531480789185
    },
    {
      "epoch": 3.1688690185546876e-05,
      "model_forward_time": 3.1430459022521973,
      "step": 8307
    },
    {
      "epoch": 3.1688690185546876e-05,
      "step": 8307,
      "training_step_time": 9.924312829971313
    },
    {
      "epoch": 3.1688690185546876e-05,
      "model_forward_time": 3.4113526344299316,
      "step": 8307
    },
    {
      "epoch": 3.1688690185546876e-05,
      "step": 8307,
      "training_step_time": 10.193201303482056
    },
    {
      "epoch": 3.1688690185546876e-05,
      "model_forward_time": 3.141934633255005,
      "step": 8307
    },
    {
      "dynamics_loss_avg": 0.044708441756665704,
      "epoch": 3.1688690185546876e-05,
      "step": 8307
    },
    {
      "action_loss_avg": 0.014471908751875161,
      "epoch": 3.1688690185546876e-05,
      "step": 8307
    },
    {
      "epoch": 3.1688690185546876e-05,
      "step": 8307,
      "training_step_time": 9.927972316741943
    },
    {
      "epoch": 3.1688690185546876e-05,
      "model_forward_time": 3.1414809226989746,
      "step": 8307
    },
    {
      "epoch": 3.1688690185546876e-05,
      "step": 8307,
      "training_step_time": 10.342307090759277
    },
    {
      "epoch": 3.16925048828125e-05,
      "model_forward_time": 3.1628217697143555,
      "step": 8308
    },
    {
      "epoch": 3.16925048828125e-05,
      "step": 8308,
      "training_step_time": 9.947682619094849
    },
    {
      "epoch": 3.16925048828125e-05,
      "model_forward_time": 3.135300397872925,
      "step": 8308
    },
    {
      "epoch": 3.16925048828125e-05,
      "step": 8308,
      "training_step_time": 9.918675899505615
    },
    {
      "epoch": 3.16925048828125e-05,
      "model_forward_time": 3.422307014465332,
      "step": 8308
    },
    {
      "epoch": 3.16925048828125e-05,
      "step": 8308,
      "training_step_time": 10.20134973526001
    },
    {
      "epoch": 3.16925048828125e-05,
      "model_forward_time": 3.1369855403900146,
      "step": 8308
    },
    {
      "epoch": 3.16925048828125e-05,
      "step": 8308,
      "training_step_time": 10.298527717590332
    },
    {
      "epoch": 3.169631958007812e-05,
      "model_forward_time": 3.1380434036254883,
      "step": 8309
    },
    {
      "epoch": 3.169631958007812e-05,
      "step": 8309,
      "training_step_time": 9.91444730758667
    },
    {
      "epoch": 3.169631958007812e-05,
      "model_forward_time": 3.135606527328491,
      "step": 8309
    },
    {
      "epoch": 3.169631958007812e-05,
      "step": 8309,
      "training_step_time": 9.912119150161743
    },
    {
      "epoch": 3.169631958007812e-05,
      "model_forward_time": 3.134037733078003,
      "step": 8309
    },
    {
      "epoch": 3.169631958007812e-05,
      "step": 8309,
      "training_step_time": 9.91559386253357
    },
    {
      "epoch": 3.169631958007812e-05,
      "model_forward_time": 3.391824960708618,
      "step": 8309
    },
    {
      "epoch": 3.169631958007812e-05,
      "step": 8309,
      "training_step_time": 10.319602966308594
    },
    {
      "epoch": 3.170013427734375e-05,
      "grad_norm": 0.5345423221588135,
      "learning_rate": 1e-05,
      "loss": 0.0833,
      "step": 8310
    },
    {
      "epoch": 3.170013427734375e-05,
      "model_forward_time": 3.159856081008911,
      "step": 8310
    },
    {
      "dynamics_loss_avg": 0.07283168323338032,
      "epoch": 3.170013427734375e-05,
      "step": 8310
    },
    {
      "action_loss_avg": 0.027798955515027046,
      "epoch": 3.170013427734375e-05,
      "step": 8310
    },
    {
      "epoch": 3.170013427734375e-05,
      "step": 8310,
      "training_step_time": 9.945832252502441
    },
    {
      "epoch": 3.170013427734375e-05,
      "model_forward_time": 3.1366052627563477,
      "step": 8310
    },
    {
      "epoch": 3.170013427734375e-05,
      "step": 8310,
      "training_step_time": 9.912635803222656
    },
    {
      "epoch": 3.170013427734375e-05,
      "model_forward_time": 3.135420560836792,
      "step": 8310
    },
    {
      "epoch": 3.170013427734375e-05,
      "step": 8310,
      "training_step_time": 9.914136171340942
    },
    {
      "epoch": 3.170013427734375e-05,
      "model_forward_time": 3.1400983333587646,
      "step": 8310
    },
    {
      "epoch": 3.170013427734375e-05,
      "step": 8310,
      "training_step_time": 10.305803060531616
    },
    {
      "epoch": 3.1703948974609375e-05,
      "model_forward_time": 3.3991544246673584,
      "step": 8311
    },
    {
      "epoch": 3.1703948974609375e-05,
      "step": 8311,
      "training_step_time": 10.187471628189087
    },
    {
      "epoch": 3.1703948974609375e-05,
      "model_forward_time": 3.1360673904418945,
      "step": 8311
    },
    {
      "epoch": 3.1703948974609375e-05,
      "step": 8311,
      "training_step_time": 9.9201021194458
    },
    {
      "epoch": 3.1703948974609375e-05,
      "model_forward_time": 3.136401414871216,
      "step": 8311
    },
    {
      "epoch": 3.1703948974609375e-05,
      "step": 8311,
      "training_step_time": 9.916035175323486
    },
    {
      "epoch": 3.1703948974609375e-05,
      "model_forward_time": 3.139069080352783,
      "step": 8311
    },
    {
      "epoch": 3.1703948974609375e-05,
      "step": 8311,
      "training_step_time": 10.260719776153564
    },
    {
      "epoch": 3.1707763671875e-05,
      "model_forward_time": 3.1548545360565186,
      "step": 8312
    },
    {
      "epoch": 3.1707763671875e-05,
      "step": 8312,
      "training_step_time": 9.932094097137451
    },
    {
      "epoch": 3.1707763671875e-05,
      "model_forward_time": 3.4366261959075928,
      "step": 8312
    },
    {
      "epoch": 3.1707763671875e-05,
      "step": 8312,
      "training_step_time": 10.212517499923706
    },
    {
      "epoch": 3.1707763671875e-05,
      "model_forward_time": 3.130352020263672,
      "step": 8312
    },
    {
      "dynamics_loss_avg": 0.0726580161601305,
      "epoch": 3.1707763671875e-05,
      "step": 8312
    },
    {
      "action_loss_avg": 0.021410261373966934,
      "epoch": 3.1707763671875e-05,
      "step": 8312
    },
    {
      "epoch": 3.1707763671875e-05,
      "step": 8312,
      "training_step_time": 9.913283586502075
    },
    {
      "epoch": 3.1707763671875e-05,
      "model_forward_time": 3.142822265625,
      "step": 8312
    },
    {
      "epoch": 3.1707763671875e-05,
      "step": 8312,
      "training_step_time": 10.27751612663269
    },
    {
      "epoch": 3.171157836914063e-05,
      "model_forward_time": 3.154874324798584,
      "step": 8313
    },
    {
      "epoch": 3.171157836914063e-05,
      "step": 8313,
      "training_step_time": 9.93082594871521
    },
    {
      "epoch": 3.171157836914063e-05,
      "model_forward_time": 3.1309218406677246,
      "step": 8313
    },
    {
      "epoch": 3.171157836914063e-05,
      "step": 8313,
      "training_step_time": 9.908014059066772
    },
    {
      "epoch": 3.171157836914063e-05,
      "model_forward_time": 3.4005026817321777,
      "step": 8313
    },
    {
      "epoch": 3.171157836914063e-05,
      "step": 8313,
      "training_step_time": 10.173947811126709
    },
    {
      "epoch": 3.171157836914063e-05,
      "model_forward_time": 3.137617826461792,
      "step": 8313
    },
    {
      "epoch": 3.171157836914063e-05,
      "step": 8313,
      "training_step_time": 10.279134035110474
    },
    {
      "epoch": 3.171539306640625e-05,
      "model_forward_time": 3.1506900787353516,
      "step": 8314
    },
    {
      "epoch": 3.171539306640625e-05,
      "step": 8314,
      "training_step_time": 9.931489944458008
    },
    {
      "epoch": 3.171539306640625e-05,
      "model_forward_time": 3.1338119506835938,
      "step": 8314
    },
    {
      "epoch": 3.171539306640625e-05,
      "step": 8314,
      "training_step_time": 9.90670919418335
    },
    {
      "epoch": 3.171539306640625e-05,
      "model_forward_time": 3.1334710121154785,
      "step": 8314
    },
    {
      "epoch": 3.171539306640625e-05,
      "step": 8314,
      "training_step_time": 9.904708623886108
    },
    {
      "epoch": 3.171539306640625e-05,
      "model_forward_time": 3.408454179763794,
      "step": 8314
    },
    {
      "epoch": 3.171539306640625e-05,
      "step": 8314,
      "training_step_time": 10.273175716400146
    },
    {
      "epoch": 3.171920776367187e-05,
      "model_forward_time": 3.147599220275879,
      "step": 8315
    },
    {
      "dynamics_loss_avg": 0.05523599721491337,
      "epoch": 3.171920776367187e-05,
      "step": 8315
    },
    {
      "action_loss_avg": 0.018633404979482293,
      "epoch": 3.171920776367187e-05,
      "step": 8315
    },
    {
      "epoch": 3.171920776367187e-05,
      "step": 8315,
      "training_step_time": 9.926688194274902
    },
    {
      "epoch": 3.171920776367187e-05,
      "model_forward_time": 3.1308999061584473,
      "step": 8315
    },
    {
      "epoch": 3.171920776367187e-05,
      "step": 8315,
      "training_step_time": 9.908266305923462
    },
    {
      "epoch": 3.171920776367187e-05,
      "model_forward_time": 3.1383464336395264,
      "step": 8315
    },
    {
      "epoch": 3.171920776367187e-05,
      "step": 8315,
      "training_step_time": 9.920910835266113
    },
    {
      "epoch": 3.171920776367187e-05,
      "model_forward_time": 3.1400482654571533,
      "step": 8315
    },
    {
      "epoch": 3.171920776367187e-05,
      "step": 8315,
      "training_step_time": 10.321051359176636
    },
    {
      "epoch": 3.17230224609375e-05,
      "model_forward_time": 3.4249393939971924,
      "step": 8316
    },
    {
      "epoch": 3.17230224609375e-05,
      "step": 8316,
      "training_step_time": 10.206477165222168
    },
    {
      "epoch": 3.17230224609375e-05,
      "model_forward_time": 3.1378867626190186,
      "step": 8316
    },
    {
      "epoch": 3.17230224609375e-05,
      "step": 8316,
      "training_step_time": 9.923346996307373
    },
    {
      "epoch": 3.17230224609375e-05,
      "model_forward_time": 3.141521453857422,
      "step": 8316
    },
    {
      "epoch": 3.17230224609375e-05,
      "step": 8316,
      "training_step_time": 9.927118301391602
    },
    {
      "epoch": 3.17230224609375e-05,
      "model_forward_time": 3.139578104019165,
      "step": 8316
    },
    {
      "epoch": 3.17230224609375e-05,
      "step": 8316,
      "training_step_time": 10.254303216934204
    },
    {
      "epoch": 3.1726837158203126e-05,
      "model_forward_time": 3.1782045364379883,
      "step": 8317
    },
    {
      "epoch": 3.1726837158203126e-05,
      "step": 8317,
      "training_step_time": 9.960996389389038
    },
    {
      "epoch": 3.1726837158203126e-05,
      "model_forward_time": 3.3902206420898438,
      "step": 8317
    },
    {
      "epoch": 3.1726837158203126e-05,
      "step": 8317,
      "training_step_time": 10.169087409973145
    },
    {
      "epoch": 3.1726837158203126e-05,
      "model_forward_time": 3.1385726928710938,
      "step": 8317
    },
    {
      "dynamics_loss_avg": 0.05282283239066601,
      "epoch": 3.1726837158203126e-05,
      "step": 8317
    },
    {
      "action_loss_avg": 0.01983138232026249,
      "epoch": 3.1726837158203126e-05,
      "step": 8317
    },
    {
      "epoch": 3.1726837158203126e-05,
      "step": 8317,
      "training_step_time": 9.920290231704712
    },
    {
      "epoch": 3.1726837158203126e-05,
      "model_forward_time": 3.12857985496521,
      "step": 8317
    },
    {
      "epoch": 3.1726837158203126e-05,
      "step": 8317,
      "training_step_time": 10.332903623580933
    },
    {
      "epoch": 3.173065185546875e-05,
      "model_forward_time": 3.142338752746582,
      "step": 8318
    },
    {
      "epoch": 3.173065185546875e-05,
      "step": 8318,
      "training_step_time": 9.920196294784546
    },
    {
      "epoch": 3.173065185546875e-05,
      "model_forward_time": 3.1329402923583984,
      "step": 8318
    },
    {
      "epoch": 3.173065185546875e-05,
      "step": 8318,
      "training_step_time": 9.905418157577515
    },
    {
      "epoch": 3.173065185546875e-05,
      "model_forward_time": 3.4108288288116455,
      "step": 8318
    },
    {
      "epoch": 3.173065185546875e-05,
      "step": 8318,
      "training_step_time": 10.190950155258179
    },
    {
      "epoch": 3.173065185546875e-05,
      "model_forward_time": 3.1348962783813477,
      "step": 8318
    },
    {
      "epoch": 3.173065185546875e-05,
      "step": 8318,
      "training_step_time": 10.278767824172974
    },
    {
      "epoch": 3.173446655273438e-05,
      "model_forward_time": 3.1434504985809326,
      "step": 8319
    },
    {
      "epoch": 3.173446655273438e-05,
      "step": 8319,
      "training_step_time": 9.922475099563599
    },
    {
      "epoch": 3.173446655273438e-05,
      "model_forward_time": 3.136200189590454,
      "step": 8319
    },
    {
      "epoch": 3.173446655273438e-05,
      "step": 8319,
      "training_step_time": 9.914303302764893
    },
    {
      "epoch": 3.173446655273438e-05,
      "model_forward_time": 3.133721113204956,
      "step": 8319
    },
    {
      "epoch": 3.173446655273438e-05,
      "step": 8319,
      "training_step_time": 9.90758204460144
    },
    {
      "epoch": 3.173446655273438e-05,
      "model_forward_time": 3.415616035461426,
      "step": 8319
    },
    {
      "epoch": 3.173446655273438e-05,
      "step": 8319,
      "training_step_time": 10.285791635513306
    },
    {
      "epoch": 3.173828125e-05,
      "grad_norm": 0.4586537480354309,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 8320
    },
    {
      "epoch": 3.173828125e-05,
      "model_forward_time": 3.1546437740325928,
      "step": 8320
    },
    {
      "dynamics_loss_avg": 0.053955495916306974,
      "epoch": 3.173828125e-05,
      "step": 8320
    },
    {
      "action_loss_avg": 0.019149825954809786,
      "epoch": 3.173828125e-05,
      "step": 8320
    },
    {
      "epoch": 3.173828125e-05,
      "step": 8320,
      "training_step_time": 9.934415102005005
    },
    {
      "epoch": 3.173828125e-05,
      "model_forward_time": 3.1354920864105225,
      "step": 8320
    },
    {
      "epoch": 3.173828125e-05,
      "step": 8320,
      "training_step_time": 9.905220985412598
    },
    {
      "epoch": 3.173828125e-05,
      "model_forward_time": 3.1300971508026123,
      "step": 8320
    },
    {
      "epoch": 3.173828125e-05,
      "step": 8320,
      "training_step_time": 9.901997089385986
    },
    {
      "epoch": 3.173828125e-05,
      "model_forward_time": 3.14160418510437,
      "step": 8320
    },
    {
      "epoch": 3.173828125e-05,
      "step": 8320,
      "training_step_time": 10.332334518432617
    },
    {
      "epoch": 3.1742095947265624e-05,
      "model_forward_time": 3.429372549057007,
      "step": 8321
    },
    {
      "epoch": 3.1742095947265624e-05,
      "step": 8321,
      "training_step_time": 10.226014614105225
    },
    {
      "epoch": 3.1742095947265624e-05,
      "model_forward_time": 3.1413280963897705,
      "step": 8321
    },
    {
      "epoch": 3.1742095947265624e-05,
      "step": 8321,
      "training_step_time": 9.919028043746948
    },
    {
      "epoch": 3.1742095947265624e-05,
      "model_forward_time": 2.4325404167175293,
      "step": 8321
    },
    {
      "epoch": 3.1742095947265624e-05,
      "step": 8321,
      "training_step_time": 7.464319944381714
    },
    {
      "epoch": 3.1742095947265624e-05,
      "model_forward_time": 3.1338651180267334,
      "step": 8321
    },
    {
      "epoch": 3.1742095947265624e-05,
      "step": 8321,
      "training_step_time": 10.267624855041504
    },
    {
      "epoch": 3.174591064453125e-05,
      "model_forward_time": 3.1580464839935303,
      "step": 8322
    },
    {
      "epoch": 3.174591064453125e-05,
      "step": 8322,
      "training_step_time": 9.939253568649292
    },
    {
      "epoch": 3.174591064453125e-05,
      "model_forward_time": 3.4100615978240967,
      "step": 8322
    },
    {
      "epoch": 3.174591064453125e-05,
      "step": 8322,
      "training_step_time": 10.185839176177979
    },
    {
      "epoch": 3.174591064453125e-05,
      "model_forward_time": 3.1329023838043213,
      "step": 8322
    },
    {
      "dynamics_loss_avg": 0.0621937358751893,
      "epoch": 3.174591064453125e-05,
      "step": 8322
    },
    {
      "action_loss_avg": 0.02253214018419385,
      "epoch": 3.174591064453125e-05,
      "step": 8322
    },
    {
      "epoch": 3.174591064453125e-05,
      "step": 8322,
      "training_step_time": 9.914762258529663
    },
    {
      "epoch": 3.174591064453125e-05,
      "model_forward_time": 2.4331159591674805,
      "step": 8322
    },
    {
      "epoch": 3.174591064453125e-05,
      "step": 8322,
      "training_step_time": 10.299824714660645
    },
    {
      "epoch": 3.1749725341796877e-05,
      "model_forward_time": 3.1519553661346436,
      "step": 8323
    },
    {
      "epoch": 3.1749725341796877e-05,
      "step": 8323,
      "training_step_time": 9.92929220199585
    },
    {
      "epoch": 3.1749725341796877e-05,
      "model_forward_time": 3.12943696975708,
      "step": 8323
    },
    {
      "epoch": 3.1749725341796877e-05,
      "step": 8323,
      "training_step_time": 9.89694857597351
    },
    {
      "epoch": 3.1749725341796877e-05,
      "model_forward_time": 3.3864264488220215,
      "step": 8323
    },
    {
      "epoch": 3.1749725341796877e-05,
      "step": 8323,
      "training_step_time": 10.155600786209106
    },
    {
      "epoch": 3.1749725341796877e-05,
      "model_forward_time": 3.134798049926758,
      "step": 8323
    },
    {
      "epoch": 3.1749725341796877e-05,
      "step": 8323,
      "training_step_time": 10.253719806671143
    },
    {
      "epoch": 3.17535400390625e-05,
      "model_forward_time": 3.1540284156799316,
      "step": 8324
    },
    {
      "epoch": 3.17535400390625e-05,
      "step": 8324,
      "training_step_time": 9.92918586730957
    },
    {
      "epoch": 3.17535400390625e-05,
      "model_forward_time": 3.130767583847046,
      "step": 8324
    },
    {
      "epoch": 3.17535400390625e-05,
      "step": 8324,
      "training_step_time": 9.904022693634033
    },
    {
      "epoch": 3.17535400390625e-05,
      "model_forward_time": 3.1339991092681885,
      "step": 8324
    },
    {
      "epoch": 3.17535400390625e-05,
      "step": 8324,
      "training_step_time": 9.903581619262695
    },
    {
      "epoch": 3.17535400390625e-05,
      "model_forward_time": 3.4203686714172363,
      "step": 8324
    },
    {
      "epoch": 3.17535400390625e-05,
      "step": 8324,
      "training_step_time": 10.295986652374268
    },
    {
      "epoch": 3.175735473632812e-05,
      "model_forward_time": 3.152719259262085,
      "step": 8325
    },
    {
      "dynamics_loss_avg": 0.06838551722466946,
      "epoch": 3.175735473632812e-05,
      "step": 8325
    },
    {
      "action_loss_avg": 0.02123686606064439,
      "epoch": 3.175735473632812e-05,
      "step": 8325
    },
    {
      "epoch": 3.175735473632812e-05,
      "step": 8325,
      "training_step_time": 9.93167233467102
    },
    {
      "epoch": 3.175735473632812e-05,
      "model_forward_time": 3.1360788345336914,
      "step": 8325
    },
    {
      "epoch": 3.175735473632812e-05,
      "step": 8325,
      "training_step_time": 9.914675235748291
    },
    {
      "epoch": 3.175735473632812e-05,
      "model_forward_time": 3.1369166374206543,
      "step": 8325
    },
    {
      "epoch": 3.175735473632812e-05,
      "step": 8325,
      "training_step_time": 9.916152715682983
    },
    {
      "epoch": 3.175735473632812e-05,
      "model_forward_time": 3.136410713195801,
      "step": 8325
    },
    {
      "epoch": 3.175735473632812e-05,
      "step": 8325,
      "training_step_time": 10.328925848007202
    },
    {
      "epoch": 3.176116943359375e-05,
      "model_forward_time": 3.4291977882385254,
      "step": 8326
    },
    {
      "epoch": 3.176116943359375e-05,
      "step": 8326,
      "training_step_time": 10.205423831939697
    },
    {
      "epoch": 3.176116943359375e-05,
      "model_forward_time": 3.1346237659454346,
      "step": 8326
    },
    {
      "epoch": 3.176116943359375e-05,
      "step": 8326,
      "training_step_time": 9.906173467636108
    },
    {
      "epoch": 3.176116943359375e-05,
      "model_forward_time": 3.1351444721221924,
      "step": 8326
    },
    {
      "epoch": 3.176116943359375e-05,
      "step": 8326,
      "training_step_time": 9.910296201705933
    },
    {
      "epoch": 3.176116943359375e-05,
      "model_forward_time": 3.1343600749969482,
      "step": 8326
    },
    {
      "epoch": 3.176116943359375e-05,
      "step": 8326,
      "training_step_time": 10.244681358337402
    },
    {
      "epoch": 3.1764984130859375e-05,
      "model_forward_time": 3.156480073928833,
      "step": 8327
    },
    {
      "epoch": 3.1764984130859375e-05,
      "step": 8327,
      "training_step_time": 9.937379121780396
    },
    {
      "epoch": 3.1764984130859375e-05,
      "model_forward_time": 3.406486988067627,
      "step": 8327
    },
    {
      "epoch": 3.1764984130859375e-05,
      "step": 8327,
      "training_step_time": 10.188999891281128
    },
    {
      "epoch": 3.1764984130859375e-05,
      "model_forward_time": 3.1566073894500732,
      "step": 8327
    },
    {
      "dynamics_loss_avg": 0.06710760099813343,
      "epoch": 3.1764984130859375e-05,
      "step": 8327
    },
    {
      "action_loss_avg": 0.024234455032274126,
      "epoch": 3.1764984130859375e-05,
      "step": 8327
    },
    {
      "epoch": 3.1764984130859375e-05,
      "step": 8327,
      "training_step_time": 9.936594247817993
    },
    {
      "epoch": 3.1764984130859375e-05,
      "model_forward_time": 3.1425745487213135,
      "step": 8327
    },
    {
      "epoch": 3.1764984130859375e-05,
      "step": 8327,
      "training_step_time": 10.282667398452759
    },
    {
      "epoch": 3.1768798828125e-05,
      "model_forward_time": 3.1556997299194336,
      "step": 8328
    },
    {
      "epoch": 3.1768798828125e-05,
      "step": 8328,
      "training_step_time": 9.93474292755127
    },
    {
      "epoch": 3.1768798828125e-05,
      "model_forward_time": 3.143958330154419,
      "step": 8328
    },
    {
      "epoch": 3.1768798828125e-05,
      "step": 8328,
      "training_step_time": 9.922773838043213
    },
    {
      "epoch": 3.1768798828125e-05,
      "model_forward_time": 3.4647204875946045,
      "step": 8328
    },
    {
      "epoch": 3.1768798828125e-05,
      "step": 8328,
      "training_step_time": 10.246362209320068
    },
    {
      "epoch": 3.1768798828125e-05,
      "model_forward_time": 3.1394591331481934,
      "step": 8328
    },
    {
      "epoch": 3.1768798828125e-05,
      "step": 8328,
      "training_step_time": 10.289603233337402
    },
    {
      "epoch": 3.177261352539063e-05,
      "model_forward_time": 3.143697500228882,
      "step": 8329
    },
    {
      "epoch": 3.177261352539063e-05,
      "step": 8329,
      "training_step_time": 9.918869972229004
    },
    {
      "epoch": 3.177261352539063e-05,
      "model_forward_time": 3.1374051570892334,
      "step": 8329
    },
    {
      "epoch": 3.177261352539063e-05,
      "step": 8329,
      "training_step_time": 9.91323447227478
    },
    {
      "epoch": 3.177261352539063e-05,
      "model_forward_time": 3.1370699405670166,
      "step": 8329
    },
    {
      "epoch": 3.177261352539063e-05,
      "step": 8329,
      "training_step_time": 9.915635824203491
    },
    {
      "epoch": 3.177261352539063e-05,
      "model_forward_time": 3.435210943222046,
      "step": 8329
    },
    {
      "epoch": 3.177261352539063e-05,
      "step": 8329,
      "training_step_time": 10.307575941085815
    },
    {
      "epoch": 3.177642822265625e-05,
      "grad_norm": 0.4665328860282898,
      "learning_rate": 1e-05,
      "loss": 0.0815,
      "step": 8330
    },
    {
      "epoch": 3.177642822265625e-05,
      "model_forward_time": 3.1577024459838867,
      "step": 8330
    },
    {
      "dynamics_loss_avg": 0.05163654983043671,
      "epoch": 3.177642822265625e-05,
      "step": 8330
    },
    {
      "action_loss_avg": 0.023921186942607163,
      "epoch": 3.177642822265625e-05,
      "step": 8330
    },
    {
      "epoch": 3.177642822265625e-05,
      "step": 8330,
      "training_step_time": 9.938937425613403
    },
    {
      "epoch": 3.177642822265625e-05,
      "model_forward_time": 3.1447246074676514,
      "step": 8330
    },
    {
      "epoch": 3.177642822265625e-05,
      "step": 8330,
      "training_step_time": 9.926977396011353
    },
    {
      "epoch": 3.177642822265625e-05,
      "model_forward_time": 3.135068893432617,
      "step": 8330
    },
    {
      "epoch": 3.177642822265625e-05,
      "step": 8330,
      "training_step_time": 9.910505294799805
    },
    {
      "epoch": 3.177642822265625e-05,
      "model_forward_time": 2.4334912300109863,
      "step": 8330
    },
    {
      "epoch": 3.177642822265625e-05,
      "step": 8330,
      "training_step_time": 10.016996622085571
    },
    {
      "epoch": 3.178024291992187e-05,
      "model_forward_time": 3.4518520832061768,
      "step": 8331
    },
    {
      "epoch": 3.178024291992187e-05,
      "step": 8331,
      "training_step_time": 10.231628894805908
    },
    {
      "epoch": 3.178024291992187e-05,
      "model_forward_time": 3.133613109588623,
      "step": 8331
    },
    {
      "epoch": 3.178024291992187e-05,
      "step": 8331,
      "training_step_time": 9.913618326187134
    },
    {
      "epoch": 3.178024291992187e-05,
      "model_forward_time": 3.143059492111206,
      "step": 8331
    },
    {
      "epoch": 3.178024291992187e-05,
      "step": 8331,
      "training_step_time": 9.919474363327026
    },
    {
      "epoch": 3.178024291992187e-05,
      "model_forward_time": 3.138082981109619,
      "step": 8331
    },
    {
      "epoch": 3.178024291992187e-05,
      "step": 8331,
      "training_step_time": 10.27774691581726
    },
    {
      "epoch": 3.17840576171875e-05,
      "model_forward_time": 3.1479384899139404,
      "step": 8332
    },
    {
      "epoch": 3.17840576171875e-05,
      "step": 8332,
      "training_step_time": 9.927849054336548
    },
    {
      "epoch": 3.17840576171875e-05,
      "model_forward_time": 3.4719343185424805,
      "step": 8332
    },
    {
      "epoch": 3.17840576171875e-05,
      "step": 8332,
      "training_step_time": 10.2411048412323
    },
    {
      "epoch": 3.17840576171875e-05,
      "model_forward_time": 3.1365888118743896,
      "step": 8332
    },
    {
      "dynamics_loss_avg": 0.05539685245603323,
      "epoch": 3.17840576171875e-05,
      "step": 8332
    },
    {
      "action_loss_avg": 0.016262675588950514,
      "epoch": 3.17840576171875e-05,
      "step": 8332
    },
    {
      "epoch": 3.17840576171875e-05,
      "step": 8332,
      "training_step_time": 9.907410144805908
    },
    {
      "epoch": 3.17840576171875e-05,
      "model_forward_time": 3.1417160034179688,
      "step": 8332
    },
    {
      "epoch": 3.17840576171875e-05,
      "step": 8332,
      "training_step_time": 10.243168354034424
    },
    {
      "epoch": 3.1787872314453126e-05,
      "model_forward_time": 3.1394028663635254,
      "step": 8333
    },
    {
      "epoch": 3.1787872314453126e-05,
      "step": 8333,
      "training_step_time": 9.912033796310425
    },
    {
      "epoch": 3.1787872314453126e-05,
      "model_forward_time": 3.130337715148926,
      "step": 8333
    },
    {
      "epoch": 3.1787872314453126e-05,
      "step": 8333,
      "training_step_time": 9.9037446975708
    },
    {
      "epoch": 3.1787872314453126e-05,
      "model_forward_time": 3.4114341735839844,
      "step": 8333
    },
    {
      "epoch": 3.1787872314453126e-05,
      "step": 8333,
      "training_step_time": 10.185086011886597
    },
    {
      "epoch": 3.1787872314453126e-05,
      "model_forward_time": 3.1387739181518555,
      "step": 8333
    },
    {
      "epoch": 3.1787872314453126e-05,
      "step": 8333,
      "training_step_time": 10.283562421798706
    },
    {
      "epoch": 3.179168701171875e-05,
      "model_forward_time": 3.1614632606506348,
      "step": 8334
    },
    {
      "epoch": 3.179168701171875e-05,
      "step": 8334,
      "training_step_time": 9.943385124206543
    },
    {
      "epoch": 3.179168701171875e-05,
      "model_forward_time": 3.141190767288208,
      "step": 8334
    },
    {
      "epoch": 3.179168701171875e-05,
      "step": 8334,
      "training_step_time": 9.923681497573853
    },
    {
      "epoch": 3.179168701171875e-05,
      "model_forward_time": 3.144770860671997,
      "step": 8334
    },
    {
      "epoch": 3.179168701171875e-05,
      "step": 8334,
      "training_step_time": 9.919353008270264
    },
    {
      "epoch": 3.179168701171875e-05,
      "model_forward_time": 3.451874256134033,
      "step": 8334
    },
    {
      "epoch": 3.179168701171875e-05,
      "step": 8334,
      "training_step_time": 10.323555946350098
    },
    {
      "epoch": 3.179550170898438e-05,
      "model_forward_time": 3.1616122722625732,
      "step": 8335
    },
    {
      "dynamics_loss_avg": 0.052636850718408824,
      "epoch": 3.179550170898438e-05,
      "step": 8335
    },
    {
      "action_loss_avg": 0.019893307518213988,
      "epoch": 3.179550170898438e-05,
      "step": 8335
    },
    {
      "epoch": 3.179550170898438e-05,
      "step": 8335,
      "training_step_time": 9.941428899765015
    },
    {
      "epoch": 3.179550170898438e-05,
      "model_forward_time": 3.1371448040008545,
      "step": 8335
    },
    {
      "epoch": 3.179550170898438e-05,
      "step": 8335,
      "training_step_time": 9.914337873458862
    },
    {
      "epoch": 3.179550170898438e-05,
      "model_forward_time": 3.1442959308624268,
      "step": 8335
    },
    {
      "epoch": 3.179550170898438e-05,
      "step": 8335,
      "training_step_time": 9.924991130828857
    },
    {
      "epoch": 3.179550170898438e-05,
      "model_forward_time": 3.1448216438293457,
      "step": 8335
    },
    {
      "epoch": 3.179550170898438e-05,
      "step": 8335,
      "training_step_time": 10.281256675720215
    },
    {
      "epoch": 3.179931640625e-05,
      "model_forward_time": 3.4634807109832764,
      "step": 8336
    },
    {
      "epoch": 3.179931640625e-05,
      "step": 8336,
      "training_step_time": 10.241472005844116
    },
    {
      "epoch": 3.179931640625e-05,
      "model_forward_time": 3.1327412128448486,
      "step": 8336
    },
    {
      "epoch": 3.179931640625e-05,
      "step": 8336,
      "training_step_time": 9.906981945037842
    },
    {
      "epoch": 3.179931640625e-05,
      "model_forward_time": 3.137575149536133,
      "step": 8336
    },
    {
      "epoch": 3.179931640625e-05,
      "step": 8336,
      "training_step_time": 9.91750192642212
    },
    {
      "epoch": 3.179931640625e-05,
      "model_forward_time": 3.1379098892211914,
      "step": 8336
    },
    {
      "epoch": 3.179931640625e-05,
      "step": 8336,
      "training_step_time": 10.245716333389282
    },
    {
      "epoch": 3.1803131103515624e-05,
      "model_forward_time": 3.141000747680664,
      "step": 8337
    },
    {
      "epoch": 3.1803131103515624e-05,
      "step": 8337,
      "training_step_time": 9.920330286026001
    },
    {
      "epoch": 3.1803131103515624e-05,
      "model_forward_time": 3.3972384929656982,
      "step": 8337
    },
    {
      "epoch": 3.1803131103515624e-05,
      "step": 8337,
      "training_step_time": 10.177613496780396
    },
    {
      "epoch": 3.1803131103515624e-05,
      "model_forward_time": 3.1447935104370117,
      "step": 8337
    },
    {
      "dynamics_loss_avg": 0.06218303609639406,
      "epoch": 3.1803131103515624e-05,
      "step": 8337
    },
    {
      "action_loss_avg": 0.021279680682346223,
      "epoch": 3.1803131103515624e-05,
      "step": 8337
    },
    {
      "epoch": 3.1803131103515624e-05,
      "step": 8337,
      "training_step_time": 9.932267427444458
    },
    {
      "epoch": 3.1803131103515624e-05,
      "model_forward_time": 3.138364315032959,
      "step": 8337
    },
    {
      "epoch": 3.1803131103515624e-05,
      "step": 8337,
      "training_step_time": 10.26430869102478
    },
    {
      "epoch": 3.180694580078125e-05,
      "model_forward_time": 2.4515223503112793,
      "step": 8338
    },
    {
      "epoch": 3.180694580078125e-05,
      "step": 8338,
      "training_step_time": 7.492484092712402
    },
    {
      "epoch": 3.180694580078125e-05,
      "model_forward_time": 3.1394667625427246,
      "step": 8338
    },
    {
      "epoch": 3.180694580078125e-05,
      "step": 8338,
      "training_step_time": 9.926723957061768
    },
    {
      "epoch": 3.180694580078125e-05,
      "model_forward_time": 3.139066696166992,
      "step": 8338
    },
    {
      "epoch": 3.180694580078125e-05,
      "step": 8338,
      "training_step_time": 10.203995943069458
    },
    {
      "epoch": 3.180694580078125e-05,
      "model_forward_time": 3.1404001712799072,
      "step": 8338
    },
    {
      "epoch": 3.180694580078125e-05,
      "step": 8338,
      "training_step_time": 10.320528268814087
    },
    {
      "epoch": 3.1810760498046877e-05,
      "model_forward_time": 3.158337354660034,
      "step": 8339
    },
    {
      "epoch": 3.1810760498046877e-05,
      "step": 8339,
      "training_step_time": 9.942099809646606
    },
    {
      "epoch": 3.1810760498046877e-05,
      "model_forward_time": 3.1393954753875732,
      "step": 8339
    },
    {
      "epoch": 3.1810760498046877e-05,
      "step": 8339,
      "training_step_time": 9.924619674682617
    },
    {
      "epoch": 3.1810760498046877e-05,
      "model_forward_time": 3.1380598545074463,
      "step": 8339
    },
    {
      "epoch": 3.1810760498046877e-05,
      "step": 8339,
      "training_step_time": 9.920124769210815
    },
    {
      "epoch": 3.1810760498046877e-05,
      "model_forward_time": 3.1353163719177246,
      "step": 8339
    },
    {
      "epoch": 3.1810760498046877e-05,
      "step": 8339,
      "training_step_time": 10.331592559814453
    },
    {
      "epoch": 3.18145751953125e-05,
      "grad_norm": 0.4436710774898529,
      "learning_rate": 1e-05,
      "loss": 0.0766,
      "step": 8340
    },
    {
      "epoch": 3.18145751953125e-05,
      "model_forward_time": 3.155865430831909,
      "step": 8340
    },
    {
      "dynamics_loss_avg": 0.04858962893486023,
      "epoch": 3.18145751953125e-05,
      "step": 8340
    },
    {
      "action_loss_avg": 0.016875315783545376,
      "epoch": 3.18145751953125e-05,
      "step": 8340
    },
    {
      "epoch": 3.18145751953125e-05,
      "step": 8340,
      "training_step_time": 9.935074806213379
    },
    {
      "epoch": 3.18145751953125e-05,
      "model_forward_time": 3.1371963024139404,
      "step": 8340
    },
    {
      "epoch": 3.18145751953125e-05,
      "step": 8340,
      "training_step_time": 9.913983821868896
    },
    {
      "epoch": 3.18145751953125e-05,
      "model_forward_time": 3.135704278945923,
      "step": 8340
    },
    {
      "epoch": 3.18145751953125e-05,
      "step": 8340,
      "training_step_time": 9.915435791015625
    },
    {
      "epoch": 3.18145751953125e-05,
      "model_forward_time": 3.1344730854034424,
      "step": 8340
    },
    {
      "epoch": 3.18145751953125e-05,
      "step": 8340,
      "training_step_time": 10.332020998001099
    },
    {
      "epoch": 3.181838989257812e-05,
      "model_forward_time": 2.4735870361328125,
      "step": 8341
    },
    {
      "epoch": 3.181838989257812e-05,
      "step": 8341,
      "training_step_time": 7.772829294204712
    },
    {
      "epoch": 3.181838989257812e-05,
      "model_forward_time": 3.133007049560547,
      "step": 8341
    },
    {
      "epoch": 3.181838989257812e-05,
      "step": 8341,
      "training_step_time": 9.909956693649292
    },
    {
      "epoch": 3.181838989257812e-05,
      "model_forward_time": 3.1438772678375244,
      "step": 8341
    },
    {
      "epoch": 3.181838989257812e-05,
      "step": 8341,
      "training_step_time": 9.923097372055054
    },
    {
      "epoch": 3.181838989257812e-05,
      "model_forward_time": 3.1383345127105713,
      "step": 8341
    },
    {
      "epoch": 3.181838989257812e-05,
      "step": 8341,
      "training_step_time": 10.271049976348877
    },
    {
      "epoch": 3.182220458984375e-05,
      "model_forward_time": 3.1540186405181885,
      "step": 8342
    },
    {
      "epoch": 3.182220458984375e-05,
      "step": 8342,
      "training_step_time": 9.938009977340698
    },
    {
      "epoch": 3.182220458984375e-05,
      "model_forward_time": 3.136897325515747,
      "step": 8342
    },
    {
      "epoch": 3.182220458984375e-05,
      "step": 8342,
      "training_step_time": 10.20199465751648
    },
    {
      "epoch": 3.182220458984375e-05,
      "model_forward_time": 3.1403276920318604,
      "step": 8342
    },
    {
      "dynamics_loss_avg": 0.06073576919734478,
      "epoch": 3.182220458984375e-05,
      "step": 8342
    },
    {
      "action_loss_avg": 0.024113473109900953,
      "epoch": 3.182220458984375e-05,
      "step": 8342
    },
    {
      "epoch": 3.182220458984375e-05,
      "step": 8342,
      "training_step_time": 9.92702341079712
    },
    {
      "epoch": 3.182220458984375e-05,
      "model_forward_time": 3.1374332904815674,
      "step": 8342
    },
    {
      "epoch": 3.182220458984375e-05,
      "step": 8342,
      "training_step_time": 10.31991958618164
    },
    {
      "epoch": 3.1826019287109375e-05,
      "model_forward_time": 3.1513781547546387,
      "step": 8343
    },
    {
      "epoch": 3.1826019287109375e-05,
      "step": 8343,
      "training_step_time": 9.931244134902954
    },
    {
      "epoch": 3.1826019287109375e-05,
      "model_forward_time": 3.1431286334991455,
      "step": 8343
    },
    {
      "epoch": 3.1826019287109375e-05,
      "step": 8343,
      "training_step_time": 9.919931888580322
    },
    {
      "epoch": 3.1826019287109375e-05,
      "model_forward_time": 3.41497540473938,
      "step": 8343
    },
    {
      "epoch": 3.1826019287109375e-05,
      "step": 8343,
      "training_step_time": 10.194596529006958
    },
    {
      "epoch": 3.1826019287109375e-05,
      "model_forward_time": 3.141103982925415,
      "step": 8343
    },
    {
      "epoch": 3.1826019287109375e-05,
      "step": 8343,
      "training_step_time": 10.275065898895264
    },
    {
      "epoch": 3.1829833984375e-05,
      "model_forward_time": 3.1546359062194824,
      "step": 8344
    },
    {
      "epoch": 3.1829833984375e-05,
      "step": 8344,
      "training_step_time": 9.932923316955566
    },
    {
      "epoch": 3.1829833984375e-05,
      "model_forward_time": 3.135129690170288,
      "step": 8344
    },
    {
      "epoch": 3.1829833984375e-05,
      "step": 8344,
      "training_step_time": 9.914766311645508
    },
    {
      "epoch": 3.1829833984375e-05,
      "model_forward_time": 3.1340548992156982,
      "step": 8344
    },
    {
      "epoch": 3.1829833984375e-05,
      "step": 8344,
      "training_step_time": 9.916778087615967
    },
    {
      "epoch": 3.1829833984375e-05,
      "model_forward_time": 2.4367315769195557,
      "step": 8344
    },
    {
      "epoch": 3.1829833984375e-05,
      "step": 8344,
      "training_step_time": 10.245264291763306
    },
    {
      "epoch": 3.183364868164063e-05,
      "model_forward_time": 3.478137254714966,
      "step": 8345
    },
    {
      "dynamics_loss_avg": 0.03499097023159266,
      "epoch": 3.183364868164063e-05,
      "step": 8345
    },
    {
      "action_loss_avg": 0.013270110404118896,
      "epoch": 3.183364868164063e-05,
      "step": 8345
    },
    {
      "epoch": 3.183364868164063e-05,
      "step": 8345,
      "training_step_time": 10.260990619659424
    },
    {
      "epoch": 3.183364868164063e-05,
      "model_forward_time": 3.135031223297119,
      "step": 8345
    },
    {
      "epoch": 3.183364868164063e-05,
      "step": 8345,
      "training_step_time": 9.914191722869873
    },
    {
      "epoch": 3.183364868164063e-05,
      "model_forward_time": 3.134967803955078,
      "step": 8345
    },
    {
      "epoch": 3.183364868164063e-05,
      "step": 8345,
      "training_step_time": 9.913524866104126
    },
    {
      "epoch": 3.183364868164063e-05,
      "model_forward_time": 3.1340508460998535,
      "step": 8345
    },
    {
      "epoch": 3.183364868164063e-05,
      "step": 8345,
      "training_step_time": 10.256486177444458
    },
    {
      "epoch": 3.183746337890625e-05,
      "model_forward_time": 3.4427828788757324,
      "step": 8346
    },
    {
      "epoch": 3.183746337890625e-05,
      "step": 8346,
      "training_step_time": 10.22591519355774
    },
    {
      "epoch": 3.183746337890625e-05,
      "model_forward_time": 3.140153169631958,
      "step": 8346
    },
    {
      "epoch": 3.183746337890625e-05,
      "step": 8346,
      "training_step_time": 9.920173406600952
    },
    {
      "epoch": 3.183746337890625e-05,
      "model_forward_time": 3.1353206634521484,
      "step": 8346
    },
    {
      "epoch": 3.183746337890625e-05,
      "step": 8346,
      "training_step_time": 9.912655115127563
    },
    {
      "epoch": 3.183746337890625e-05,
      "model_forward_time": 3.1356842517852783,
      "step": 8346
    },
    {
      "epoch": 3.183746337890625e-05,
      "step": 8346,
      "training_step_time": 10.25937271118164
    },
    {
      "epoch": 3.184127807617187e-05,
      "model_forward_time": 3.1540417671203613,
      "step": 8347
    },
    {
      "epoch": 3.184127807617187e-05,
      "step": 8347,
      "training_step_time": 9.931890487670898
    },
    {
      "epoch": 3.184127807617187e-05,
      "model_forward_time": 3.132802963256836,
      "step": 8347
    },
    {
      "epoch": 3.184127807617187e-05,
      "step": 8347,
      "training_step_time": 10.179399490356445
    },
    {
      "epoch": 3.184127807617187e-05,
      "model_forward_time": 3.1372501850128174,
      "step": 8347
    },
    {
      "dynamics_loss_avg": 0.054115744214504956,
      "epoch": 3.184127807617187e-05,
      "step": 8347
    },
    {
      "action_loss_avg": 0.01940153636969626,
      "epoch": 3.184127807617187e-05,
      "step": 8347
    },
    {
      "epoch": 3.184127807617187e-05,
      "step": 8347,
      "training_step_time": 9.91761064529419
    },
    {
      "epoch": 3.184127807617187e-05,
      "model_forward_time": 3.1413657665252686,
      "step": 8347
    },
    {
      "epoch": 3.184127807617187e-05,
      "step": 8347,
      "training_step_time": 10.013750314712524
    },
    {
      "epoch": 3.18450927734375e-05,
      "model_forward_time": 3.156752109527588,
      "step": 8348
    },
    {
      "epoch": 3.18450927734375e-05,
      "step": 8348,
      "training_step_time": 9.933954000473022
    },
    {
      "epoch": 3.18450927734375e-05,
      "model_forward_time": 3.1324172019958496,
      "step": 8348
    },
    {
      "epoch": 3.18450927734375e-05,
      "step": 8348,
      "training_step_time": 9.905827283859253
    },
    {
      "epoch": 3.18450927734375e-05,
      "model_forward_time": 3.136723756790161,
      "step": 8348
    },
    {
      "epoch": 3.18450927734375e-05,
      "step": 8348,
      "training_step_time": 10.212699174880981
    },
    {
      "epoch": 3.18450927734375e-05,
      "model_forward_time": 3.1368353366851807,
      "step": 8348
    },
    {
      "epoch": 3.18450927734375e-05,
      "step": 8348,
      "training_step_time": 10.281251192092896
    },
    {
      "epoch": 3.1848907470703126e-05,
      "model_forward_time": 3.1525633335113525,
      "step": 8349
    },
    {
      "epoch": 3.1848907470703126e-05,
      "step": 8349,
      "training_step_time": 9.928652286529541
    },
    {
      "epoch": 3.1848907470703126e-05,
      "model_forward_time": 3.13446307182312,
      "step": 8349
    },
    {
      "epoch": 3.1848907470703126e-05,
      "step": 8349,
      "training_step_time": 9.90964126586914
    },
    {
      "epoch": 3.1848907470703126e-05,
      "model_forward_time": 3.1360065937042236,
      "step": 8349
    },
    {
      "epoch": 3.1848907470703126e-05,
      "step": 8349,
      "training_step_time": 9.90964651107788
    },
    {
      "epoch": 3.1848907470703126e-05,
      "model_forward_time": 3.419205665588379,
      "step": 8349
    },
    {
      "epoch": 3.1848907470703126e-05,
      "step": 8349,
      "training_step_time": 10.288256883621216
    },
    {
      "epoch": 3.185272216796875e-05,
      "grad_norm": 0.5722823143005371,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 8350
    },
    {
      "epoch": 3.185272216796875e-05,
      "model_forward_time": 3.1509501934051514,
      "step": 8350
    },
    {
      "dynamics_loss_avg": 0.05344503838568926,
      "epoch": 3.185272216796875e-05,
      "step": 8350
    },
    {
      "action_loss_avg": 0.019589120091404767,
      "epoch": 3.185272216796875e-05,
      "step": 8350
    },
    {
      "epoch": 3.185272216796875e-05,
      "step": 8350,
      "training_step_time": 9.932049751281738
    },
    {
      "epoch": 3.185272216796875e-05,
      "model_forward_time": 3.1382429599761963,
      "step": 8350
    },
    {
      "epoch": 3.185272216796875e-05,
      "step": 8350,
      "training_step_time": 9.922757863998413
    },
    {
      "epoch": 3.185272216796875e-05,
      "model_forward_time": 3.139967441558838,
      "step": 8350
    },
    {
      "epoch": 3.185272216796875e-05,
      "step": 8350,
      "training_step_time": 9.926856517791748
    },
    {
      "epoch": 3.185272216796875e-05,
      "model_forward_time": 3.1392664909362793,
      "step": 8350
    },
    {
      "epoch": 3.185272216796875e-05,
      "step": 8350,
      "training_step_time": 10.335436582565308
    },
    {
      "epoch": 3.185653686523438e-05,
      "model_forward_time": 3.15580677986145,
      "step": 8351
    },
    {
      "epoch": 3.185653686523438e-05,
      "step": 8351,
      "training_step_time": 10.211212873458862
    },
    {
      "epoch": 3.185653686523438e-05,
      "model_forward_time": 3.137481689453125,
      "step": 8351
    },
    {
      "epoch": 3.185653686523438e-05,
      "step": 8351,
      "training_step_time": 9.913905143737793
    },
    {
      "epoch": 3.185653686523438e-05,
      "model_forward_time": 3.136279582977295,
      "step": 8351
    },
    {
      "epoch": 3.185653686523438e-05,
      "step": 8351,
      "training_step_time": 9.914207696914673
    },
    {
      "epoch": 3.185653686523438e-05,
      "model_forward_time": 3.1342480182647705,
      "step": 8351
    },
    {
      "epoch": 3.185653686523438e-05,
      "step": 8351,
      "training_step_time": 10.343592882156372
    },
    {
      "epoch": 3.18603515625e-05,
      "model_forward_time": 3.1579980850219727,
      "step": 8352
    },
    {
      "epoch": 3.18603515625e-05,
      "step": 8352,
      "training_step_time": 9.937328815460205
    },
    {
      "epoch": 3.18603515625e-05,
      "model_forward_time": 3.392717123031616,
      "step": 8352
    },
    {
      "epoch": 3.18603515625e-05,
      "step": 8352,
      "training_step_time": 10.164072036743164
    },
    {
      "epoch": 3.18603515625e-05,
      "model_forward_time": 3.140366792678833,
      "step": 8352
    },
    {
      "dynamics_loss_avg": 0.07950766868889332,
      "epoch": 3.18603515625e-05,
      "step": 8352
    },
    {
      "action_loss_avg": 0.022943992353975772,
      "epoch": 3.18603515625e-05,
      "step": 8352
    },
    {
      "epoch": 3.18603515625e-05,
      "step": 8352,
      "training_step_time": 9.92330002784729
    },
    {
      "epoch": 3.18603515625e-05,
      "model_forward_time": 3.1357011795043945,
      "step": 8352
    },
    {
      "epoch": 3.18603515625e-05,
      "step": 8352,
      "training_step_time": 10.020611047744751
    },
    {
      "epoch": 3.1864166259765624e-05,
      "model_forward_time": 3.148808479309082,
      "step": 8353
    },
    {
      "epoch": 3.1864166259765624e-05,
      "step": 8353,
      "training_step_time": 9.929732322692871
    },
    {
      "epoch": 3.1864166259765624e-05,
      "model_forward_time": 3.141057252883911,
      "step": 8353
    },
    {
      "epoch": 3.1864166259765624e-05,
      "step": 8353,
      "training_step_time": 9.922341585159302
    },
    {
      "epoch": 3.1864166259765624e-05,
      "model_forward_time": 3.450239658355713,
      "step": 8353
    },
    {
      "epoch": 3.1864166259765624e-05,
      "step": 8353,
      "training_step_time": 10.227882862091064
    },
    {
      "epoch": 3.1864166259765624e-05,
      "model_forward_time": 3.135828971862793,
      "step": 8353
    },
    {
      "epoch": 3.1864166259765624e-05,
      "step": 8353,
      "training_step_time": 10.281222820281982
    },
    {
      "epoch": 3.186798095703125e-05,
      "model_forward_time": 3.147456169128418,
      "step": 8354
    },
    {
      "epoch": 3.186798095703125e-05,
      "step": 8354,
      "training_step_time": 9.924952745437622
    },
    {
      "epoch": 3.186798095703125e-05,
      "model_forward_time": 3.13765025138855,
      "step": 8354
    },
    {
      "epoch": 3.186798095703125e-05,
      "step": 8354,
      "training_step_time": 9.911527395248413
    },
    {
      "epoch": 3.186798095703125e-05,
      "model_forward_time": 3.133509874343872,
      "step": 8354
    },
    {
      "epoch": 3.186798095703125e-05,
      "step": 8354,
      "training_step_time": 9.91013216972351
    },
    {
      "epoch": 3.186798095703125e-05,
      "model_forward_time": 3.4104716777801514,
      "step": 8354
    },
    {
      "epoch": 3.186798095703125e-05,
      "step": 8354,
      "training_step_time": 10.283295631408691
    },
    {
      "epoch": 3.187179565429688e-05,
      "model_forward_time": 3.151200294494629,
      "step": 8355
    },
    {
      "dynamics_loss_avg": 0.06833019889891148,
      "epoch": 3.187179565429688e-05,
      "step": 8355
    },
    {
      "action_loss_avg": 0.0217828712426126,
      "epoch": 3.187179565429688e-05,
      "step": 8355
    },
    {
      "epoch": 3.187179565429688e-05,
      "step": 8355,
      "training_step_time": 9.930491924285889
    },
    {
      "epoch": 3.187179565429688e-05,
      "model_forward_time": 3.1354520320892334,
      "step": 8355
    },
    {
      "epoch": 3.187179565429688e-05,
      "step": 8355,
      "training_step_time": 9.91335153579712
    },
    {
      "epoch": 3.187179565429688e-05,
      "model_forward_time": 3.135315418243408,
      "step": 8355
    },
    {
      "epoch": 3.187179565429688e-05,
      "step": 8355,
      "training_step_time": 9.914657831192017
    },
    {
      "epoch": 3.187179565429688e-05,
      "model_forward_time": 3.1421117782592773,
      "step": 8355
    },
    {
      "epoch": 3.187179565429688e-05,
      "step": 8355,
      "training_step_time": 10.03747034072876
    },
    {
      "epoch": 3.18756103515625e-05,
      "model_forward_time": 3.414337158203125,
      "step": 8356
    },
    {
      "epoch": 3.18756103515625e-05,
      "step": 8356,
      "training_step_time": 10.197446346282959
    },
    {
      "epoch": 3.18756103515625e-05,
      "model_forward_time": 3.140371322631836,
      "step": 8356
    },
    {
      "epoch": 3.18756103515625e-05,
      "step": 8356,
      "training_step_time": 9.917623281478882
    },
    {
      "epoch": 3.18756103515625e-05,
      "model_forward_time": 3.137460947036743,
      "step": 8356
    },
    {
      "epoch": 3.18756103515625e-05,
      "step": 8356,
      "training_step_time": 9.919753789901733
    },
    {
      "epoch": 3.18756103515625e-05,
      "model_forward_time": 3.1362392902374268,
      "step": 8356
    },
    {
      "epoch": 3.18756103515625e-05,
      "step": 8356,
      "training_step_time": 10.30260443687439
    },
    {
      "epoch": 3.187942504882812e-05,
      "model_forward_time": 3.1451194286346436,
      "step": 8357
    },
    {
      "epoch": 3.187942504882812e-05,
      "step": 8357,
      "training_step_time": 9.91979193687439
    },
    {
      "epoch": 3.187942504882812e-05,
      "model_forward_time": 3.1351318359375,
      "step": 8357
    },
    {
      "epoch": 3.187942504882812e-05,
      "step": 8357,
      "training_step_time": 10.198434352874756
    },
    {
      "epoch": 3.187942504882812e-05,
      "model_forward_time": 3.134868621826172,
      "step": 8357
    },
    {
      "dynamics_loss_avg": 0.049824921600520614,
      "epoch": 3.187942504882812e-05,
      "step": 8357
    },
    {
      "action_loss_avg": 0.017165245674550533,
      "epoch": 3.187942504882812e-05,
      "step": 8357
    },
    {
      "epoch": 3.187942504882812e-05,
      "step": 8357,
      "training_step_time": 9.91531229019165
    },
    {
      "epoch": 3.187942504882812e-05,
      "model_forward_time": 3.137260913848877,
      "step": 8357
    },
    {
      "epoch": 3.187942504882812e-05,
      "step": 8357,
      "training_step_time": 10.288935899734497
    },
    {
      "epoch": 3.188323974609375e-05,
      "model_forward_time": 3.1501219272613525,
      "step": 8358
    },
    {
      "epoch": 3.188323974609375e-05,
      "step": 8358,
      "training_step_time": 9.927675008773804
    },
    {
      "epoch": 3.188323974609375e-05,
      "model_forward_time": 3.1350574493408203,
      "step": 8358
    },
    {
      "epoch": 3.188323974609375e-05,
      "step": 8358,
      "training_step_time": 9.912108421325684
    },
    {
      "epoch": 3.188323974609375e-05,
      "model_forward_time": 3.4113190174102783,
      "step": 8358
    },
    {
      "epoch": 3.188323974609375e-05,
      "step": 8358,
      "training_step_time": 10.186255931854248
    },
    {
      "epoch": 3.188323974609375e-05,
      "model_forward_time": 3.1381723880767822,
      "step": 8358
    },
    {
      "epoch": 3.188323974609375e-05,
      "step": 8358,
      "training_step_time": 10.307929992675781
    },
    {
      "epoch": 3.1887054443359375e-05,
      "model_forward_time": 3.1543285846710205,
      "step": 8359
    },
    {
      "epoch": 3.1887054443359375e-05,
      "step": 8359,
      "training_step_time": 9.932914733886719
    },
    {
      "epoch": 3.1887054443359375e-05,
      "model_forward_time": 3.136021137237549,
      "step": 8359
    },
    {
      "epoch": 3.1887054443359375e-05,
      "step": 8359,
      "training_step_time": 9.909081220626831
    },
    {
      "epoch": 3.1887054443359375e-05,
      "model_forward_time": 3.140354871749878,
      "step": 8359
    },
    {
      "epoch": 3.1887054443359375e-05,
      "step": 8359,
      "training_step_time": 9.920526266098022
    },
    {
      "epoch": 3.1887054443359375e-05,
      "model_forward_time": 3.140810489654541,
      "step": 8359
    },
    {
      "epoch": 3.1887054443359375e-05,
      "step": 8359,
      "training_step_time": 10.298534154891968
    },
    {
      "epoch": 3.1890869140625e-05,
      "grad_norm": 0.3485201895236969,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 8360
    },
    {
      "epoch": 3.1890869140625e-05,
      "model_forward_time": 3.1611101627349854,
      "step": 8360
    },
    {
      "dynamics_loss_avg": 0.043970807641744616,
      "epoch": 3.1890869140625e-05,
      "step": 8360
    },
    {
      "action_loss_avg": 0.019621937349438668,
      "epoch": 3.1890869140625e-05,
      "step": 8360
    },
    {
      "epoch": 3.1890869140625e-05,
      "step": 8360,
      "training_step_time": 9.94662094116211
    },
    {
      "epoch": 3.1890869140625e-05,
      "model_forward_time": 3.137932062149048,
      "step": 8360
    },
    {
      "epoch": 3.1890869140625e-05,
      "step": 8360,
      "training_step_time": 9.917757511138916
    },
    {
      "epoch": 3.1890869140625e-05,
      "model_forward_time": 3.156862735748291,
      "step": 8360
    },
    {
      "epoch": 3.1890869140625e-05,
      "step": 8360,
      "training_step_time": 9.939350605010986
    },
    {
      "epoch": 3.1890869140625e-05,
      "model_forward_time": 3.143541097640991,
      "step": 8360
    },
    {
      "epoch": 3.1890869140625e-05,
      "step": 8360,
      "training_step_time": 10.0298171043396
    },
    {
      "epoch": 3.189468383789063e-05,
      "model_forward_time": 3.4432525634765625,
      "step": 8361
    },
    {
      "epoch": 3.189468383789063e-05,
      "step": 8361,
      "training_step_time": 10.223445892333984
    },
    {
      "epoch": 3.189468383789063e-05,
      "model_forward_time": 3.145195245742798,
      "step": 8361
    },
    {
      "epoch": 3.189468383789063e-05,
      "step": 8361,
      "training_step_time": 9.922045946121216
    },
    {
      "epoch": 3.189468383789063e-05,
      "model_forward_time": 3.151118516921997,
      "step": 8361
    },
    {
      "epoch": 3.189468383789063e-05,
      "step": 8361,
      "training_step_time": 9.929577350616455
    },
    {
      "epoch": 3.189468383789063e-05,
      "model_forward_time": 3.1496241092681885,
      "step": 8361
    },
    {
      "epoch": 3.189468383789063e-05,
      "step": 8361,
      "training_step_time": 10.30261516571045
    },
    {
      "epoch": 3.189849853515625e-05,
      "model_forward_time": 3.1649110317230225,
      "step": 8362
    },
    {
      "epoch": 3.189849853515625e-05,
      "step": 8362,
      "training_step_time": 9.948612689971924
    },
    {
      "epoch": 3.189849853515625e-05,
      "model_forward_time": 3.148536205291748,
      "step": 8362
    },
    {
      "epoch": 3.189849853515625e-05,
      "step": 8362,
      "training_step_time": 10.201814889907837
    },
    {
      "epoch": 3.189849853515625e-05,
      "model_forward_time": 3.1488864421844482,
      "step": 8362
    },
    {
      "dynamics_loss_avg": 0.05466045029461384,
      "epoch": 3.189849853515625e-05,
      "step": 8362
    },
    {
      "action_loss_avg": 0.021142883831635116,
      "epoch": 3.189849853515625e-05,
      "step": 8362
    },
    {
      "epoch": 3.189849853515625e-05,
      "step": 8362,
      "training_step_time": 9.927149057388306
    },
    {
      "epoch": 3.189849853515625e-05,
      "model_forward_time": 3.1382453441619873,
      "step": 8362
    },
    {
      "epoch": 3.189849853515625e-05,
      "step": 8362,
      "training_step_time": 10.27546238899231
    },
    {
      "epoch": 3.190231323242187e-05,
      "model_forward_time": 3.160271167755127,
      "step": 8363
    },
    {
      "epoch": 3.190231323242187e-05,
      "step": 8363,
      "training_step_time": 9.942455768585205
    },
    {
      "epoch": 3.190231323242187e-05,
      "model_forward_time": 3.146028757095337,
      "step": 8363
    },
    {
      "epoch": 3.190231323242187e-05,
      "step": 8363,
      "training_step_time": 9.919777870178223
    },
    {
      "epoch": 3.190231323242187e-05,
      "model_forward_time": 3.429762363433838,
      "step": 8363
    },
    {
      "epoch": 3.190231323242187e-05,
      "step": 8363,
      "training_step_time": 10.199162244796753
    },
    {
      "epoch": 3.190231323242187e-05,
      "model_forward_time": 3.1406970024108887,
      "step": 8363
    },
    {
      "epoch": 3.190231323242187e-05,
      "step": 8363,
      "training_step_time": 10.261508703231812
    },
    {
      "epoch": 3.19061279296875e-05,
      "model_forward_time": 3.1476006507873535,
      "step": 8364
    },
    {
      "epoch": 3.19061279296875e-05,
      "step": 8364,
      "training_step_time": 9.919174909591675
    },
    {
      "epoch": 3.19061279296875e-05,
      "model_forward_time": 3.1447439193725586,
      "step": 8364
    },
    {
      "epoch": 3.19061279296875e-05,
      "step": 8364,
      "training_step_time": 9.91394305229187
    },
    {
      "epoch": 3.19061279296875e-05,
      "model_forward_time": 3.1433987617492676,
      "step": 8364
    },
    {
      "epoch": 3.19061279296875e-05,
      "step": 8364,
      "training_step_time": 9.922047138214111
    },
    {
      "epoch": 3.19061279296875e-05,
      "model_forward_time": 3.4122509956359863,
      "step": 8364
    },
    {
      "epoch": 3.19061279296875e-05,
      "step": 8364,
      "training_step_time": 10.28335690498352
    },
    {
      "epoch": 3.1909942626953126e-05,
      "model_forward_time": 3.1548190116882324,
      "step": 8365
    },
    {
      "dynamics_loss_avg": 0.05335512664169073,
      "epoch": 3.1909942626953126e-05,
      "step": 8365
    },
    {
      "action_loss_avg": 0.021307878848165273,
      "epoch": 3.1909942626953126e-05,
      "step": 8365
    },
    {
      "epoch": 3.1909942626953126e-05,
      "step": 8365,
      "training_step_time": 9.938119411468506
    },
    {
      "epoch": 3.1909942626953126e-05,
      "model_forward_time": 3.148482084274292,
      "step": 8365
    },
    {
      "epoch": 3.1909942626953126e-05,
      "step": 8365,
      "training_step_time": 9.926717042922974
    },
    {
      "epoch": 3.1909942626953126e-05,
      "model_forward_time": 3.148733377456665,
      "step": 8365
    },
    {
      "epoch": 3.1909942626953126e-05,
      "step": 8365,
      "training_step_time": 9.924968481063843
    },
    {
      "epoch": 3.1909942626953126e-05,
      "model_forward_time": 3.1517281532287598,
      "step": 8365
    },
    {
      "epoch": 3.1909942626953126e-05,
      "step": 8365,
      "training_step_time": 10.03354287147522
    },
    {
      "epoch": 3.191375732421875e-05,
      "model_forward_time": 3.4217870235443115,
      "step": 8366
    },
    {
      "epoch": 3.191375732421875e-05,
      "step": 8366,
      "training_step_time": 10.206192255020142
    },
    {
      "epoch": 3.191375732421875e-05,
      "model_forward_time": 3.14192271232605,
      "step": 8366
    },
    {
      "epoch": 3.191375732421875e-05,
      "step": 8366,
      "training_step_time": 9.919573068618774
    },
    {
      "epoch": 3.191375732421875e-05,
      "model_forward_time": 3.142664909362793,
      "step": 8366
    },
    {
      "epoch": 3.191375732421875e-05,
      "step": 8366,
      "training_step_time": 9.922576904296875
    },
    {
      "epoch": 3.191375732421875e-05,
      "model_forward_time": 3.1466050148010254,
      "step": 8366
    },
    {
      "epoch": 3.191375732421875e-05,
      "step": 8366,
      "training_step_time": 10.250320196151733
    },
    {
      "epoch": 3.191757202148438e-05,
      "model_forward_time": 3.160909414291382,
      "step": 8367
    },
    {
      "epoch": 3.191757202148438e-05,
      "step": 8367,
      "training_step_time": 9.938881635665894
    },
    {
      "epoch": 3.191757202148438e-05,
      "model_forward_time": 3.4366793632507324,
      "step": 8367
    },
    {
      "epoch": 3.191757202148438e-05,
      "step": 8367,
      "training_step_time": 10.210447072982788
    },
    {
      "epoch": 3.191757202148438e-05,
      "model_forward_time": 3.1453349590301514,
      "step": 8367
    },
    {
      "dynamics_loss_avg": 0.04412258882075548,
      "epoch": 3.191757202148438e-05,
      "step": 8367
    },
    {
      "action_loss_avg": 0.014592986926436425,
      "epoch": 3.191757202148438e-05,
      "step": 8367
    },
    {
      "epoch": 3.191757202148438e-05,
      "step": 8367,
      "training_step_time": 9.919634103775024
    },
    {
      "epoch": 3.191757202148438e-05,
      "model_forward_time": 2.436591386795044,
      "step": 8367
    },
    {
      "epoch": 3.191757202148438e-05,
      "step": 8367,
      "training_step_time": 10.330199480056763
    },
    {
      "epoch": 3.192138671875e-05,
      "model_forward_time": 3.151200532913208,
      "step": 8368
    },
    {
      "epoch": 3.192138671875e-05,
      "step": 8368,
      "training_step_time": 9.924870014190674
    },
    {
      "epoch": 3.192138671875e-05,
      "model_forward_time": 3.133082151412964,
      "step": 8368
    },
    {
      "epoch": 3.192138671875e-05,
      "step": 8368,
      "training_step_time": 9.902913570404053
    },
    {
      "epoch": 3.192138671875e-05,
      "model_forward_time": 3.135662794113159,
      "step": 8368
    },
    {
      "epoch": 3.192138671875e-05,
      "step": 8368,
      "training_step_time": 10.201698303222656
    },
    {
      "epoch": 3.192138671875e-05,
      "model_forward_time": 3.142186164855957,
      "step": 8368
    },
    {
      "epoch": 3.192138671875e-05,
      "step": 8368,
      "training_step_time": 10.27420711517334
    },
    {
      "epoch": 3.1925201416015624e-05,
      "model_forward_time": 3.1514203548431396,
      "step": 8369
    },
    {
      "epoch": 3.1925201416015624e-05,
      "step": 8369,
      "training_step_time": 9.927266597747803
    },
    {
      "epoch": 3.1925201416015624e-05,
      "model_forward_time": 3.134899854660034,
      "step": 8369
    },
    {
      "epoch": 3.1925201416015624e-05,
      "step": 8369,
      "training_step_time": 9.909743309020996
    },
    {
      "epoch": 3.1925201416015624e-05,
      "model_forward_time": 3.1397862434387207,
      "step": 8369
    },
    {
      "epoch": 3.1925201416015624e-05,
      "step": 8369,
      "training_step_time": 9.913851499557495
    },
    {
      "epoch": 3.1925201416015624e-05,
      "model_forward_time": 3.4212000370025635,
      "step": 8369
    },
    {
      "epoch": 3.1925201416015624e-05,
      "step": 8369,
      "training_step_time": 10.34716796875
    },
    {
      "epoch": 3.192901611328125e-05,
      "grad_norm": 0.4187029302120209,
      "learning_rate": 1e-05,
      "loss": 0.0812,
      "step": 8370
    },
    {
      "epoch": 3.192901611328125e-05,
      "model_forward_time": 3.149613618850708,
      "step": 8370
    },
    {
      "dynamics_loss_avg": 0.07443179432302713,
      "epoch": 3.192901611328125e-05,
      "step": 8370
    },
    {
      "action_loss_avg": 0.028946486581116915,
      "epoch": 3.192901611328125e-05,
      "step": 8370
    },
    {
      "epoch": 3.192901611328125e-05,
      "step": 8370,
      "training_step_time": 9.929136514663696
    },
    {
      "epoch": 3.192901611328125e-05,
      "model_forward_time": 3.1354658603668213,
      "step": 8370
    },
    {
      "epoch": 3.192901611328125e-05,
      "step": 8370,
      "training_step_time": 9.923930883407593
    },
    {
      "epoch": 3.192901611328125e-05,
      "model_forward_time": 3.1424999237060547,
      "step": 8370
    },
    {
      "epoch": 3.192901611328125e-05,
      "step": 8370,
      "training_step_time": 9.917255878448486
    },
    {
      "epoch": 3.192901611328125e-05,
      "model_forward_time": 3.1384127140045166,
      "step": 8370
    },
    {
      "epoch": 3.192901611328125e-05,
      "step": 8370,
      "training_step_time": 10.297688722610474
    },
    {
      "epoch": 3.193283081054688e-05,
      "model_forward_time": 3.148174285888672,
      "step": 8371
    },
    {
      "epoch": 3.193283081054688e-05,
      "step": 8371,
      "training_step_time": 10.205466508865356
    },
    {
      "epoch": 3.193283081054688e-05,
      "model_forward_time": 3.135066509246826,
      "step": 8371
    },
    {
      "epoch": 3.193283081054688e-05,
      "step": 8371,
      "training_step_time": 9.906944036483765
    },
    {
      "epoch": 3.193283081054688e-05,
      "model_forward_time": 3.140578031539917,
      "step": 8371
    },
    {
      "epoch": 3.193283081054688e-05,
      "step": 8371,
      "training_step_time": 9.918295860290527
    },
    {
      "epoch": 3.193283081054688e-05,
      "model_forward_time": 3.1390702724456787,
      "step": 8371
    },
    {
      "epoch": 3.193283081054688e-05,
      "step": 8371,
      "training_step_time": 10.327949285507202
    },
    {
      "epoch": 3.19366455078125e-05,
      "model_forward_time": 3.1528711318969727,
      "step": 8372
    },
    {
      "epoch": 3.19366455078125e-05,
      "step": 8372,
      "training_step_time": 9.928590297698975
    },
    {
      "epoch": 3.19366455078125e-05,
      "model_forward_time": 3.3885252475738525,
      "step": 8372
    },
    {
      "epoch": 3.19366455078125e-05,
      "step": 8372,
      "training_step_time": 10.15824031829834
    },
    {
      "epoch": 3.19366455078125e-05,
      "model_forward_time": 3.1362826824188232,
      "step": 8372
    },
    {
      "dynamics_loss_avg": 0.08057674653828144,
      "epoch": 3.19366455078125e-05,
      "step": 8372
    },
    {
      "action_loss_avg": 0.02903449721634388,
      "epoch": 3.19366455078125e-05,
      "step": 8372
    },
    {
      "epoch": 3.19366455078125e-05,
      "step": 8372,
      "training_step_time": 9.90908432006836
    },
    {
      "epoch": 3.19366455078125e-05,
      "model_forward_time": 3.130065441131592,
      "step": 8372
    },
    {
      "epoch": 3.19366455078125e-05,
      "step": 8372,
      "training_step_time": 10.287400722503662
    },
    {
      "epoch": 3.194046020507812e-05,
      "model_forward_time": 3.151136875152588,
      "step": 8373
    },
    {
      "epoch": 3.194046020507812e-05,
      "step": 8373,
      "training_step_time": 9.92349362373352
    },
    {
      "epoch": 3.194046020507812e-05,
      "model_forward_time": 3.1354598999023438,
      "step": 8373
    },
    {
      "epoch": 3.194046020507812e-05,
      "step": 8373,
      "training_step_time": 9.902830600738525
    },
    {
      "epoch": 3.194046020507812e-05,
      "model_forward_time": 3.413163423538208,
      "step": 8373
    },
    {
      "epoch": 3.194046020507812e-05,
      "step": 8373,
      "training_step_time": 10.179894208908081
    },
    {
      "epoch": 3.194046020507812e-05,
      "model_forward_time": 3.1307122707366943,
      "step": 8373
    },
    {
      "epoch": 3.194046020507812e-05,
      "step": 8373,
      "training_step_time": 10.311921119689941
    },
    {
      "epoch": 3.194427490234375e-05,
      "model_forward_time": 3.141420602798462,
      "step": 8374
    },
    {
      "epoch": 3.194427490234375e-05,
      "step": 8374,
      "training_step_time": 9.908060073852539
    },
    {
      "epoch": 3.194427490234375e-05,
      "model_forward_time": 3.131721019744873,
      "step": 8374
    },
    {
      "epoch": 3.194427490234375e-05,
      "step": 8374,
      "training_step_time": 9.904731750488281
    },
    {
      "epoch": 3.194427490234375e-05,
      "model_forward_time": 3.1349828243255615,
      "step": 8374
    },
    {
      "epoch": 3.194427490234375e-05,
      "step": 8374,
      "training_step_time": 9.908466100692749
    },
    {
      "epoch": 3.194427490234375e-05,
      "model_forward_time": 3.393458843231201,
      "step": 8374
    },
    {
      "epoch": 3.194427490234375e-05,
      "step": 8374,
      "training_step_time": 10.335612535476685
    },
    {
      "epoch": 3.1948089599609375e-05,
      "model_forward_time": 3.1534242630004883,
      "step": 8375
    },
    {
      "dynamics_loss_avg": 0.07802957408130169,
      "epoch": 3.1948089599609375e-05,
      "step": 8375
    },
    {
      "action_loss_avg": 0.029473119182512164,
      "epoch": 3.1948089599609375e-05,
      "step": 8375
    },
    {
      "epoch": 3.1948089599609375e-05,
      "step": 8375,
      "training_step_time": 9.932328462600708
    },
    {
      "epoch": 3.1948089599609375e-05,
      "model_forward_time": 3.1404306888580322,
      "step": 8375
    },
    {
      "epoch": 3.1948089599609375e-05,
      "step": 8375,
      "training_step_time": 9.909193277359009
    },
    {
      "epoch": 3.1948089599609375e-05,
      "model_forward_time": 3.1348085403442383,
      "step": 8375
    },
    {
      "epoch": 3.1948089599609375e-05,
      "step": 8375,
      "training_step_time": 9.907672882080078
    },
    {
      "epoch": 3.1948089599609375e-05,
      "model_forward_time": 3.1479127407073975,
      "step": 8375
    },
    {
      "epoch": 3.1948089599609375e-05,
      "step": 8375,
      "training_step_time": 10.294407606124878
    },
    {
      "epoch": 3.1951904296875e-05,
      "model_forward_time": 3.412320613861084,
      "step": 8376
    },
    {
      "epoch": 3.1951904296875e-05,
      "step": 8376,
      "training_step_time": 10.189953804016113
    },
    {
      "epoch": 3.1951904296875e-05,
      "model_forward_time": 3.1392948627471924,
      "step": 8376
    },
    {
      "epoch": 3.1951904296875e-05,
      "step": 8376,
      "training_step_time": 9.910375356674194
    },
    {
      "epoch": 3.1951904296875e-05,
      "model_forward_time": 3.1303248405456543,
      "step": 8376
    },
    {
      "epoch": 3.1951904296875e-05,
      "step": 8376,
      "training_step_time": 9.898520469665527
    },
    {
      "epoch": 3.1951904296875e-05,
      "model_forward_time": 3.1312739849090576,
      "step": 8376
    },
    {
      "epoch": 3.1951904296875e-05,
      "step": 8376,
      "training_step_time": 10.278019189834595
    },
    {
      "epoch": 3.195571899414063e-05,
      "model_forward_time": 3.1512067317962646,
      "step": 8377
    },
    {
      "epoch": 3.195571899414063e-05,
      "step": 8377,
      "training_step_time": 9.933094263076782
    },
    {
      "epoch": 3.195571899414063e-05,
      "model_forward_time": 3.142115831375122,
      "step": 8377
    },
    {
      "epoch": 3.195571899414063e-05,
      "step": 8377,
      "training_step_time": 10.197630882263184
    },
    {
      "epoch": 3.195571899414063e-05,
      "model_forward_time": 3.1397311687469482,
      "step": 8377
    },
    {
      "dynamics_loss_avg": 0.05146589260548353,
      "epoch": 3.195571899414063e-05,
      "step": 8377
    },
    {
      "action_loss_avg": 0.016021483298391103,
      "epoch": 3.195571899414063e-05,
      "step": 8377
    },
    {
      "epoch": 3.195571899414063e-05,
      "step": 8377,
      "training_step_time": 9.909717559814453
    },
    {
      "epoch": 3.195571899414063e-05,
      "model_forward_time": 3.1404690742492676,
      "step": 8377
    },
    {
      "epoch": 3.195571899414063e-05,
      "step": 8377,
      "training_step_time": 10.313995361328125
    },
    {
      "epoch": 3.195953369140625e-05,
      "model_forward_time": 3.1531717777252197,
      "step": 8378
    },
    {
      "epoch": 3.195953369140625e-05,
      "step": 8378,
      "training_step_time": 9.927458763122559
    },
    {
      "epoch": 3.195953369140625e-05,
      "model_forward_time": 3.140315055847168,
      "step": 8378
    },
    {
      "epoch": 3.195953369140625e-05,
      "step": 8378,
      "training_step_time": 9.911973237991333
    },
    {
      "epoch": 3.195953369140625e-05,
      "model_forward_time": 3.4529590606689453,
      "step": 8378
    },
    {
      "epoch": 3.195953369140625e-05,
      "step": 8378,
      "training_step_time": 10.228161334991455
    },
    {
      "epoch": 3.195953369140625e-05,
      "model_forward_time": 3.144360065460205,
      "step": 8378
    },
    {
      "epoch": 3.195953369140625e-05,
      "step": 8378,
      "training_step_time": 10.0244140625
    },
    {
      "epoch": 3.196334838867187e-05,
      "model_forward_time": 3.1566288471221924,
      "step": 8379
    },
    {
      "epoch": 3.196334838867187e-05,
      "step": 8379,
      "training_step_time": 9.93807339668274
    },
    {
      "epoch": 3.196334838867187e-05,
      "model_forward_time": 3.1458542346954346,
      "step": 8379
    },
    {
      "epoch": 3.196334838867187e-05,
      "step": 8379,
      "training_step_time": 9.921408414840698
    },
    {
      "epoch": 3.196334838867187e-05,
      "model_forward_time": 3.1524736881256104,
      "step": 8379
    },
    {
      "epoch": 3.196334838867187e-05,
      "step": 8379,
      "training_step_time": 9.927819013595581
    },
    {
      "epoch": 3.196334838867187e-05,
      "model_forward_time": 3.1479673385620117,
      "step": 8379
    },
    {
      "epoch": 3.196334838867187e-05,
      "step": 8379,
      "training_step_time": 10.34054183959961
    },
    {
      "epoch": 3.19671630859375e-05,
      "grad_norm": 0.42436671257019043,
      "learning_rate": 1e-05,
      "loss": 0.0809,
      "step": 8380
    },
    {
      "epoch": 3.19671630859375e-05,
      "model_forward_time": 3.163051128387451,
      "step": 8380
    },
    {
      "dynamics_loss_avg": 0.05834438428282738,
      "epoch": 3.19671630859375e-05,
      "step": 8380
    },
    {
      "action_loss_avg": 0.019973226822912693,
      "epoch": 3.19671630859375e-05,
      "step": 8380
    },
    {
      "epoch": 3.19671630859375e-05,
      "step": 8380,
      "training_step_time": 9.952900886535645
    },
    {
      "epoch": 3.19671630859375e-05,
      "model_forward_time": 3.141376256942749,
      "step": 8380
    },
    {
      "epoch": 3.19671630859375e-05,
      "step": 8380,
      "training_step_time": 9.916790962219238
    },
    {
      "epoch": 3.19671630859375e-05,
      "model_forward_time": 3.1436989307403564,
      "step": 8380
    },
    {
      "epoch": 3.19671630859375e-05,
      "step": 8380,
      "training_step_time": 9.917743921279907
    },
    {
      "epoch": 3.19671630859375e-05,
      "model_forward_time": 3.1455862522125244,
      "step": 8380
    },
    {
      "epoch": 3.19671630859375e-05,
      "step": 8380,
      "training_step_time": 10.272345542907715
    },
    {
      "epoch": 3.1970977783203126e-05,
      "model_forward_time": 3.4724466800689697,
      "step": 8381
    },
    {
      "epoch": 3.1970977783203126e-05,
      "step": 8381,
      "training_step_time": 10.24323058128357
    },
    {
      "epoch": 3.1970977783203126e-05,
      "model_forward_time": 3.140981674194336,
      "step": 8381
    },
    {
      "epoch": 3.1970977783203126e-05,
      "step": 8381,
      "training_step_time": 9.924851179122925
    },
    {
      "epoch": 3.1970977783203126e-05,
      "model_forward_time": 3.150646209716797,
      "step": 8381
    },
    {
      "epoch": 3.1970977783203126e-05,
      "step": 8381,
      "training_step_time": 9.926144361495972
    },
    {
      "epoch": 3.1970977783203126e-05,
      "model_forward_time": 3.145595073699951,
      "step": 8381
    },
    {
      "epoch": 3.1970977783203126e-05,
      "step": 8381,
      "training_step_time": 10.022154331207275
    },
    {
      "epoch": 3.197479248046875e-05,
      "model_forward_time": 3.1522369384765625,
      "step": 8382
    },
    {
      "epoch": 3.197479248046875e-05,
      "step": 8382,
      "training_step_time": 9.939525604248047
    },
    {
      "epoch": 3.197479248046875e-05,
      "model_forward_time": 3.148568868637085,
      "step": 8382
    },
    {
      "epoch": 3.197479248046875e-05,
      "step": 8382,
      "training_step_time": 10.205838680267334
    },
    {
      "epoch": 3.197479248046875e-05,
      "model_forward_time": 3.1466190814971924,
      "step": 8382
    },
    {
      "dynamics_loss_avg": 0.056258767284452914,
      "epoch": 3.197479248046875e-05,
      "step": 8382
    },
    {
      "action_loss_avg": 0.021710605826228857,
      "epoch": 3.197479248046875e-05,
      "step": 8382
    },
    {
      "epoch": 3.197479248046875e-05,
      "step": 8382,
      "training_step_time": 9.92239236831665
    },
    {
      "epoch": 3.197479248046875e-05,
      "model_forward_time": 3.148538589477539,
      "step": 8382
    },
    {
      "epoch": 3.197479248046875e-05,
      "step": 8382,
      "training_step_time": 10.325105905532837
    },
    {
      "epoch": 3.197860717773437e-05,
      "model_forward_time": 3.151740789413452,
      "step": 8383
    },
    {
      "epoch": 3.197860717773437e-05,
      "step": 8383,
      "training_step_time": 9.92831540107727
    },
    {
      "epoch": 3.197860717773437e-05,
      "model_forward_time": 3.1438231468200684,
      "step": 8383
    },
    {
      "epoch": 3.197860717773437e-05,
      "step": 8383,
      "training_step_time": 9.920212507247925
    },
    {
      "epoch": 3.197860717773437e-05,
      "model_forward_time": 3.4668285846710205,
      "step": 8383
    },
    {
      "epoch": 3.197860717773437e-05,
      "step": 8383,
      "training_step_time": 10.242302656173706
    },
    {
      "epoch": 3.197860717773437e-05,
      "model_forward_time": 3.144227981567383,
      "step": 8383
    },
    {
      "epoch": 3.197860717773437e-05,
      "step": 8383,
      "training_step_time": 10.28835654258728
    },
    {
      "epoch": 3.1982421875e-05,
      "model_forward_time": 3.1483263969421387,
      "step": 8384
    },
    {
      "epoch": 3.1982421875e-05,
      "step": 8384,
      "training_step_time": 9.923356056213379
    },
    {
      "epoch": 3.1982421875e-05,
      "model_forward_time": 3.1453680992126465,
      "step": 8384
    },
    {
      "epoch": 3.1982421875e-05,
      "step": 8384,
      "training_step_time": 9.91920280456543
    },
    {
      "epoch": 3.1982421875e-05,
      "model_forward_time": 3.15392804145813,
      "step": 8384
    },
    {
      "epoch": 3.1982421875e-05,
      "step": 8384,
      "training_step_time": 9.93232011795044
    },
    {
      "epoch": 3.1982421875e-05,
      "model_forward_time": 3.4233224391937256,
      "step": 8384
    },
    {
      "epoch": 3.1982421875e-05,
      "step": 8384,
      "training_step_time": 10.297832727432251
    },
    {
      "epoch": 3.1986236572265624e-05,
      "model_forward_time": 3.1667795181274414,
      "step": 8385
    },
    {
      "dynamics_loss_avg": 0.0514886548742652,
      "epoch": 3.1986236572265624e-05,
      "step": 8385
    },
    {
      "action_loss_avg": 0.02257453864440322,
      "epoch": 3.1986236572265624e-05,
      "step": 8385
    },
    {
      "epoch": 3.1986236572265624e-05,
      "step": 8385,
      "training_step_time": 9.948874473571777
    },
    {
      "epoch": 3.1986236572265624e-05,
      "model_forward_time": 3.147677183151245,
      "step": 8385
    },
    {
      "epoch": 3.1986236572265624e-05,
      "step": 8385,
      "training_step_time": 9.925002813339233
    },
    {
      "epoch": 3.1986236572265624e-05,
      "model_forward_time": 3.1484720706939697,
      "step": 8385
    },
    {
      "epoch": 3.1986236572265624e-05,
      "step": 8385,
      "training_step_time": 9.930476427078247
    },
    {
      "epoch": 3.1986236572265624e-05,
      "model_forward_time": 3.1510133743286133,
      "step": 8385
    },
    {
      "epoch": 3.1986236572265624e-05,
      "step": 8385,
      "training_step_time": 10.326074600219727
    },
    {
      "epoch": 3.199005126953125e-05,
      "model_forward_time": 3.454493761062622,
      "step": 8386
    },
    {
      "epoch": 3.199005126953125e-05,
      "step": 8386,
      "training_step_time": 10.231814861297607
    },
    {
      "epoch": 3.199005126953125e-05,
      "model_forward_time": 3.152838945388794,
      "step": 8386
    },
    {
      "epoch": 3.199005126953125e-05,
      "step": 8386,
      "training_step_time": 9.929302453994751
    },
    {
      "epoch": 3.199005126953125e-05,
      "model_forward_time": 3.1502304077148438,
      "step": 8386
    },
    {
      "epoch": 3.199005126953125e-05,
      "step": 8386,
      "training_step_time": 9.929033517837524
    },
    {
      "epoch": 3.199005126953125e-05,
      "model_forward_time": 3.161778211593628,
      "step": 8386
    },
    {
      "epoch": 3.199005126953125e-05,
      "step": 8386,
      "training_step_time": 10.033440589904785
    },
    {
      "epoch": 3.199386596679688e-05,
      "model_forward_time": 3.1736228466033936,
      "step": 8387
    },
    {
      "epoch": 3.199386596679688e-05,
      "step": 8387,
      "training_step_time": 9.961772918701172
    },
    {
      "epoch": 3.199386596679688e-05,
      "model_forward_time": 3.451724052429199,
      "step": 8387
    },
    {
      "epoch": 3.199386596679688e-05,
      "step": 8387,
      "training_step_time": 10.23173975944519
    },
    {
      "epoch": 3.199386596679688e-05,
      "model_forward_time": 3.1560397148132324,
      "step": 8387
    },
    {
      "dynamics_loss_avg": 0.05542350187897682,
      "epoch": 3.199386596679688e-05,
      "step": 8387
    },
    {
      "action_loss_avg": 0.02372596606146544,
      "epoch": 3.199386596679688e-05,
      "step": 8387
    },
    {
      "epoch": 3.199386596679688e-05,
      "step": 8387,
      "training_step_time": 9.939128160476685
    },
    {
      "epoch": 3.199386596679688e-05,
      "model_forward_time": 3.1561739444732666,
      "step": 8387
    },
    {
      "epoch": 3.199386596679688e-05,
      "step": 8387,
      "training_step_time": 10.318601846694946
    },
    {
      "epoch": 3.19976806640625e-05,
      "model_forward_time": 3.1460108757019043,
      "step": 8388
    },
    {
      "epoch": 3.19976806640625e-05,
      "step": 8388,
      "training_step_time": 9.920960664749146
    },
    {
      "epoch": 3.19976806640625e-05,
      "model_forward_time": 3.145953893661499,
      "step": 8388
    },
    {
      "epoch": 3.19976806640625e-05,
      "step": 8388,
      "training_step_time": 9.928810834884644
    },
    {
      "epoch": 3.19976806640625e-05,
      "model_forward_time": 3.1483154296875,
      "step": 8388
    },
    {
      "epoch": 3.19976806640625e-05,
      "step": 8388,
      "training_step_time": 10.252909898757935
    },
    {
      "epoch": 3.19976806640625e-05,
      "model_forward_time": 3.1393260955810547,
      "step": 8388
    },
    {
      "epoch": 3.19976806640625e-05,
      "step": 8388,
      "training_step_time": 10.31643295288086
    },
    {
      "epoch": 3.200149536132812e-05,
      "model_forward_time": 3.1538851261138916,
      "step": 8389
    },
    {
      "epoch": 3.200149536132812e-05,
      "step": 8389,
      "training_step_time": 9.929903507232666
    },
    {
      "epoch": 3.200149536132812e-05,
      "model_forward_time": 2.4341866970062256,
      "step": 8389
    },
    {
      "epoch": 3.200149536132812e-05,
      "step": 8389,
      "training_step_time": 7.46198844909668
    },
    {
      "epoch": 3.200149536132812e-05,
      "model_forward_time": 3.1278860569000244,
      "step": 8389
    },
    {
      "epoch": 3.200149536132812e-05,
      "step": 8389,
      "training_step_time": 9.894615888595581
    },
    {
      "epoch": 3.200149536132812e-05,
      "model_forward_time": 3.1408979892730713,
      "step": 8389
    },
    {
      "epoch": 3.200149536132812e-05,
      "step": 8389,
      "training_step_time": 10.276064157485962
    },
    {
      "epoch": 3.200531005859375e-05,
      "grad_norm": 0.5877845287322998,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 8390
    },
    {
      "epoch": 3.200531005859375e-05,
      "model_forward_time": 3.150080680847168,
      "step": 8390
    },
    {
      "dynamics_loss_avg": 0.04973548706620932,
      "epoch": 3.200531005859375e-05,
      "step": 8390
    },
    {
      "action_loss_avg": 0.02161705745384097,
      "epoch": 3.200531005859375e-05,
      "step": 8390
    },
    {
      "epoch": 3.200531005859375e-05,
      "step": 8390,
      "training_step_time": 9.922277450561523
    },
    {
      "epoch": 3.200531005859375e-05,
      "model_forward_time": 3.1360158920288086,
      "step": 8390
    },
    {
      "epoch": 3.200531005859375e-05,
      "step": 8390,
      "training_step_time": 9.90568208694458
    },
    {
      "epoch": 3.200531005859375e-05,
      "model_forward_time": 3.136331558227539,
      "step": 8390
    },
    {
      "epoch": 3.200531005859375e-05,
      "step": 8390,
      "training_step_time": 9.907564878463745
    },
    {
      "epoch": 3.200531005859375e-05,
      "model_forward_time": 3.1355302333831787,
      "step": 8390
    },
    {
      "epoch": 3.200531005859375e-05,
      "step": 8390,
      "training_step_time": 10.304382562637329
    },
    {
      "epoch": 3.2009124755859375e-05,
      "model_forward_time": 3.1500535011291504,
      "step": 8391
    },
    {
      "epoch": 3.2009124755859375e-05,
      "step": 8391,
      "training_step_time": 10.243398427963257
    },
    {
      "epoch": 3.2009124755859375e-05,
      "model_forward_time": 3.1335184574127197,
      "step": 8391
    },
    {
      "epoch": 3.2009124755859375e-05,
      "step": 8391,
      "training_step_time": 9.899605989456177
    },
    {
      "epoch": 3.2009124755859375e-05,
      "model_forward_time": 3.1321704387664795,
      "step": 8391
    },
    {
      "epoch": 3.2009124755859375e-05,
      "step": 8391,
      "training_step_time": 9.895837545394897
    },
    {
      "epoch": 3.2009124755859375e-05,
      "model_forward_time": 3.135572671890259,
      "step": 8391
    },
    {
      "epoch": 3.2009124755859375e-05,
      "step": 8391,
      "training_step_time": 10.012235164642334
    },
    {
      "epoch": 3.2012939453125e-05,
      "model_forward_time": 3.196075677871704,
      "step": 8392
    },
    {
      "epoch": 3.2012939453125e-05,
      "step": 8392,
      "training_step_time": 9.97414493560791
    },
    {
      "epoch": 3.2012939453125e-05,
      "model_forward_time": 3.414440870285034,
      "step": 8392
    },
    {
      "epoch": 3.2012939453125e-05,
      "step": 8392,
      "training_step_time": 10.182414770126343
    },
    {
      "epoch": 3.2012939453125e-05,
      "model_forward_time": 3.1480493545532227,
      "step": 8392
    },
    {
      "dynamics_loss_avg": 0.060810323990881446,
      "epoch": 3.2012939453125e-05,
      "step": 8392
    },
    {
      "action_loss_avg": 0.016118097957223652,
      "epoch": 3.2012939453125e-05,
      "step": 8392
    },
    {
      "epoch": 3.2012939453125e-05,
      "step": 8392,
      "training_step_time": 9.924724817276001
    },
    {
      "epoch": 3.2012939453125e-05,
      "model_forward_time": 3.141376256942749,
      "step": 8392
    },
    {
      "epoch": 3.2012939453125e-05,
      "step": 8392,
      "training_step_time": 10.284228086471558
    },
    {
      "epoch": 3.201675415039063e-05,
      "model_forward_time": 2.4493863582611084,
      "step": 8393
    },
    {
      "epoch": 3.201675415039063e-05,
      "step": 8393,
      "training_step_time": 7.476743936538696
    },
    {
      "epoch": 3.201675415039063e-05,
      "model_forward_time": 3.1425113677978516,
      "step": 8393
    },
    {
      "epoch": 3.201675415039063e-05,
      "step": 8393,
      "training_step_time": 9.914555549621582
    },
    {
      "epoch": 3.201675415039063e-05,
      "model_forward_time": 3.143197536468506,
      "step": 8393
    },
    {
      "epoch": 3.201675415039063e-05,
      "step": 8393,
      "training_step_time": 9.924876689910889
    },
    {
      "epoch": 3.201675415039063e-05,
      "model_forward_time": 3.398937225341797,
      "step": 8393
    },
    {
      "epoch": 3.201675415039063e-05,
      "step": 8393,
      "training_step_time": 10.30246353149414
    },
    {
      "epoch": 3.202056884765625e-05,
      "model_forward_time": 3.1497573852539062,
      "step": 8394
    },
    {
      "epoch": 3.202056884765625e-05,
      "step": 8394,
      "training_step_time": 9.920756101608276
    },
    {
      "epoch": 3.202056884765625e-05,
      "model_forward_time": 3.143162250518799,
      "step": 8394
    },
    {
      "epoch": 3.202056884765625e-05,
      "step": 8394,
      "training_step_time": 9.91315507888794
    },
    {
      "epoch": 3.202056884765625e-05,
      "model_forward_time": 3.1408801078796387,
      "step": 8394
    },
    {
      "epoch": 3.202056884765625e-05,
      "step": 8394,
      "training_step_time": 9.91240906715393
    },
    {
      "epoch": 3.202056884765625e-05,
      "model_forward_time": 3.411667585372925,
      "step": 8394
    },
    {
      "epoch": 3.202056884765625e-05,
      "step": 8394,
      "training_step_time": 10.278561115264893
    },
    {
      "epoch": 3.202438354492187e-05,
      "model_forward_time": 3.1621150970458984,
      "step": 8395
    },
    {
      "dynamics_loss_avg": 0.053322330303490165,
      "epoch": 3.202438354492187e-05,
      "step": 8395
    },
    {
      "action_loss_avg": 0.019628668343648313,
      "epoch": 3.202438354492187e-05,
      "step": 8395
    },
    {
      "epoch": 3.202438354492187e-05,
      "step": 8395,
      "training_step_time": 9.936858415603638
    },
    {
      "epoch": 3.202438354492187e-05,
      "model_forward_time": 3.144176959991455,
      "step": 8395
    },
    {
      "epoch": 3.202438354492187e-05,
      "step": 8395,
      "training_step_time": 9.916003942489624
    },
    {
      "epoch": 3.202438354492187e-05,
      "model_forward_time": 3.139897108078003,
      "step": 8395
    },
    {
      "epoch": 3.202438354492187e-05,
      "step": 8395,
      "training_step_time": 9.918073415756226
    },
    {
      "epoch": 3.202438354492187e-05,
      "model_forward_time": 3.142275810241699,
      "step": 8395
    },
    {
      "epoch": 3.202438354492187e-05,
      "step": 8395,
      "training_step_time": 10.277892351150513
    },
    {
      "epoch": 3.20281982421875e-05,
      "model_forward_time": 3.149240255355835,
      "step": 8396
    },
    {
      "epoch": 3.20281982421875e-05,
      "step": 8396,
      "training_step_time": 10.214160442352295
    },
    {
      "epoch": 3.20281982421875e-05,
      "model_forward_time": 3.1444385051727295,
      "step": 8396
    },
    {
      "epoch": 3.20281982421875e-05,
      "step": 8396,
      "training_step_time": 9.915218114852905
    },
    {
      "epoch": 3.20281982421875e-05,
      "model_forward_time": 3.140748977661133,
      "step": 8396
    },
    {
      "epoch": 3.20281982421875e-05,
      "step": 8396,
      "training_step_time": 9.910808563232422
    },
    {
      "epoch": 3.20281982421875e-05,
      "model_forward_time": 3.1526811122894287,
      "step": 8396
    },
    {
      "epoch": 3.20281982421875e-05,
      "step": 8396,
      "training_step_time": 10.318557262420654
    },
    {
      "epoch": 3.2032012939453126e-05,
      "model_forward_time": 3.1608498096466064,
      "step": 8397
    },
    {
      "epoch": 3.2032012939453126e-05,
      "step": 8397,
      "training_step_time": 9.933170318603516
    },
    {
      "epoch": 3.2032012939453126e-05,
      "model_forward_time": 3.438554525375366,
      "step": 8397
    },
    {
      "epoch": 3.2032012939453126e-05,
      "step": 8397,
      "training_step_time": 10.208340167999268
    },
    {
      "epoch": 3.2032012939453126e-05,
      "model_forward_time": 3.1443827152252197,
      "step": 8397
    },
    {
      "dynamics_loss_avg": 0.050222021713852885,
      "epoch": 3.2032012939453126e-05,
      "step": 8397
    },
    {
      "action_loss_avg": 0.016743680601939558,
      "epoch": 3.2032012939453126e-05,
      "step": 8397
    },
    {
      "epoch": 3.2032012939453126e-05,
      "step": 8397,
      "training_step_time": 9.921260118484497
    },
    {
      "epoch": 3.2032012939453126e-05,
      "model_forward_time": 3.1468358039855957,
      "step": 8397
    },
    {
      "epoch": 3.2032012939453126e-05,
      "step": 8397,
      "training_step_time": 10.32965898513794
    },
    {
      "epoch": 3.203582763671875e-05,
      "model_forward_time": 3.1630358695983887,
      "step": 8398
    },
    {
      "epoch": 3.203582763671875e-05,
      "step": 8398,
      "training_step_time": 9.93912410736084
    },
    {
      "epoch": 3.203582763671875e-05,
      "model_forward_time": 3.1452219486236572,
      "step": 8398
    },
    {
      "epoch": 3.203582763671875e-05,
      "step": 8398,
      "training_step_time": 9.916182279586792
    },
    {
      "epoch": 3.203582763671875e-05,
      "model_forward_time": 3.403596878051758,
      "step": 8398
    },
    {
      "epoch": 3.203582763671875e-05,
      "step": 8398,
      "training_step_time": 10.170706033706665
    },
    {
      "epoch": 3.203582763671875e-05,
      "model_forward_time": 3.1382110118865967,
      "step": 8398
    },
    {
      "epoch": 3.203582763671875e-05,
      "step": 8398,
      "training_step_time": 10.262182474136353
    },
    {
      "epoch": 3.203964233398437e-05,
      "model_forward_time": 3.1599349975585938,
      "step": 8399
    },
    {
      "epoch": 3.203964233398437e-05,
      "step": 8399,
      "training_step_time": 9.929490327835083
    },
    {
      "epoch": 3.203964233398437e-05,
      "model_forward_time": 3.139981508255005,
      "step": 8399
    },
    {
      "epoch": 3.203964233398437e-05,
      "step": 8399,
      "training_step_time": 9.921052694320679
    },
    {
      "epoch": 3.203964233398437e-05,
      "model_forward_time": 3.135472059249878,
      "step": 8399
    },
    {
      "epoch": 3.203964233398437e-05,
      "step": 8399,
      "training_step_time": 9.912369012832642
    },
    {
      "epoch": 3.203964233398437e-05,
      "model_forward_time": 3.419825792312622,
      "step": 8399
    },
    {
      "epoch": 3.203964233398437e-05,
      "step": 8399,
      "training_step_time": 10.284857749938965
    },
    {
      "epoch": 3.204345703125e-05,
      "grad_norm": 0.5899885296821594,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 8400
    },
    {
      "epoch": 3.204345703125e-05,
      "model_forward_time": 3.1498613357543945,
      "step": 8400
    },
    {
      "dynamics_loss_avg": 0.059587613493204114,
      "epoch": 3.204345703125e-05,
      "step": 8400
    },
    {
      "action_loss_avg": 0.027978188544511794,
      "epoch": 3.204345703125e-05,
      "step": 8400
    },
    {
      "epoch": 3.204345703125e-05,
      "step": 8400,
      "training_step_time": 9.925399541854858
    },
    {
      "epoch": 3.204345703125e-05,
      "model_forward_time": 3.1376287937164307,
      "step": 8400
    },
    {
      "epoch": 3.204345703125e-05,
      "step": 8400,
      "training_step_time": 9.909847259521484
    },
    {
      "epoch": 3.204345703125e-05,
      "model_forward_time": 3.1441469192504883,
      "step": 8400
    },
    {
      "epoch": 3.204345703125e-05,
      "step": 8400,
      "training_step_time": 9.916860818862915
    },
    {
      "epoch": 3.204345703125e-05,
      "model_forward_time": 3.135411262512207,
      "step": 8400
    },
    {
      "epoch": 3.204345703125e-05,
      "step": 8400,
      "training_step_time": 10.267985343933105
    },
    {
      "epoch": 3.2047271728515624e-05,
      "model_forward_time": 3.4293837547302246,
      "step": 8401
    },
    {
      "epoch": 3.2047271728515624e-05,
      "step": 8401,
      "training_step_time": 10.20613694190979
    },
    {
      "epoch": 3.2047271728515624e-05,
      "model_forward_time": 3.1398961544036865,
      "step": 8401
    },
    {
      "epoch": 3.2047271728515624e-05,
      "step": 8401,
      "training_step_time": 9.91446828842163
    },
    {
      "epoch": 3.2047271728515624e-05,
      "model_forward_time": 3.144221782684326,
      "step": 8401
    },
    {
      "epoch": 3.2047271728515624e-05,
      "step": 8401,
      "training_step_time": 9.924158334732056
    },
    {
      "epoch": 3.2047271728515624e-05,
      "model_forward_time": 3.146206855773926,
      "step": 8401
    },
    {
      "epoch": 3.2047271728515624e-05,
      "step": 8401,
      "training_step_time": 10.298313617706299
    },
    {
      "epoch": 3.205108642578125e-05,
      "model_forward_time": 3.1547653675079346,
      "step": 8402
    },
    {
      "epoch": 3.205108642578125e-05,
      "step": 8402,
      "training_step_time": 9.933536767959595
    },
    {
      "epoch": 3.205108642578125e-05,
      "model_forward_time": 3.4048378467559814,
      "step": 8402
    },
    {
      "epoch": 3.205108642578125e-05,
      "step": 8402,
      "training_step_time": 10.181601285934448
    },
    {
      "epoch": 3.205108642578125e-05,
      "model_forward_time": 3.1495718955993652,
      "step": 8402
    },
    {
      "dynamics_loss_avg": 0.05929176770150661,
      "epoch": 3.205108642578125e-05,
      "step": 8402
    },
    {
      "action_loss_avg": 0.017241705488413573,
      "epoch": 3.205108642578125e-05,
      "step": 8402
    },
    {
      "epoch": 3.205108642578125e-05,
      "step": 8402,
      "training_step_time": 9.930684089660645
    },
    {
      "epoch": 3.205108642578125e-05,
      "model_forward_time": 3.1473374366760254,
      "step": 8402
    },
    {
      "epoch": 3.205108642578125e-05,
      "step": 8402,
      "training_step_time": 10.033701181411743
    },
    {
      "epoch": 3.205490112304688e-05,
      "model_forward_time": 3.1553099155426025,
      "step": 8403
    },
    {
      "epoch": 3.205490112304688e-05,
      "step": 8403,
      "training_step_time": 9.932550191879272
    },
    {
      "epoch": 3.205490112304688e-05,
      "model_forward_time": 3.1463546752929688,
      "step": 8403
    },
    {
      "epoch": 3.205490112304688e-05,
      "step": 8403,
      "training_step_time": 9.922987699508667
    },
    {
      "epoch": 3.205490112304688e-05,
      "model_forward_time": 3.4251725673675537,
      "step": 8403
    },
    {
      "epoch": 3.205490112304688e-05,
      "step": 8403,
      "training_step_time": 10.203514575958252
    },
    {
      "epoch": 3.205490112304688e-05,
      "model_forward_time": 3.147031784057617,
      "step": 8403
    },
    {
      "epoch": 3.205490112304688e-05,
      "step": 8403,
      "training_step_time": 10.321200847625732
    },
    {
      "epoch": 3.20587158203125e-05,
      "model_forward_time": 3.1638972759246826,
      "step": 8404
    },
    {
      "epoch": 3.20587158203125e-05,
      "step": 8404,
      "training_step_time": 9.94498586654663
    },
    {
      "epoch": 3.20587158203125e-05,
      "model_forward_time": 3.150303363800049,
      "step": 8404
    },
    {
      "epoch": 3.20587158203125e-05,
      "step": 8404,
      "training_step_time": 9.927614688873291
    },
    {
      "epoch": 3.20587158203125e-05,
      "model_forward_time": 3.153475761413574,
      "step": 8404
    },
    {
      "epoch": 3.20587158203125e-05,
      "step": 8404,
      "training_step_time": 9.931759119033813
    },
    {
      "epoch": 3.20587158203125e-05,
      "model_forward_time": 3.412801504135132,
      "step": 8404
    },
    {
      "epoch": 3.20587158203125e-05,
      "step": 8404,
      "training_step_time": 10.491806507110596
    },
    {
      "epoch": 3.206253051757812e-05,
      "model_forward_time": 3.1569037437438965,
      "step": 8405
    },
    {
      "dynamics_loss_avg": 0.052927068434655666,
      "epoch": 3.206253051757812e-05,
      "step": 8405
    },
    {
      "action_loss_avg": 0.01586278425529599,
      "epoch": 3.206253051757812e-05,
      "step": 8405
    },
    {
      "epoch": 3.206253051757812e-05,
      "step": 8405,
      "training_step_time": 9.940454959869385
    },
    {
      "epoch": 3.206253051757812e-05,
      "model_forward_time": 3.1475231647491455,
      "step": 8405
    },
    {
      "epoch": 3.206253051757812e-05,
      "step": 8405,
      "training_step_time": 9.92401385307312
    },
    {
      "epoch": 3.206253051757812e-05,
      "model_forward_time": 3.1477468013763428,
      "step": 8405
    },
    {
      "epoch": 3.206253051757812e-05,
      "step": 8405,
      "training_step_time": 9.92442011833191
    },
    {
      "epoch": 3.206253051757812e-05,
      "model_forward_time": 3.1472177505493164,
      "step": 8405
    },
    {
      "epoch": 3.206253051757812e-05,
      "step": 8405,
      "training_step_time": 10.29265809059143
    },
    {
      "epoch": 3.206634521484375e-05,
      "model_forward_time": 3.4527642726898193,
      "step": 8406
    },
    {
      "epoch": 3.206634521484375e-05,
      "step": 8406,
      "training_step_time": 10.228850841522217
    },
    {
      "epoch": 3.206634521484375e-05,
      "model_forward_time": 3.145972490310669,
      "step": 8406
    },
    {
      "epoch": 3.206634521484375e-05,
      "step": 8406,
      "training_step_time": 9.921741724014282
    },
    {
      "epoch": 3.206634521484375e-05,
      "model_forward_time": 3.1433537006378174,
      "step": 8406
    },
    {
      "epoch": 3.206634521484375e-05,
      "step": 8406,
      "training_step_time": 9.920238971710205
    },
    {
      "epoch": 3.206634521484375e-05,
      "model_forward_time": 3.147991418838501,
      "step": 8406
    },
    {
      "epoch": 3.206634521484375e-05,
      "step": 8406,
      "training_step_time": 10.236761093139648
    },
    {
      "epoch": 3.2070159912109375e-05,
      "model_forward_time": 3.1619982719421387,
      "step": 8407
    },
    {
      "epoch": 3.2070159912109375e-05,
      "step": 8407,
      "training_step_time": 9.942062377929688
    },
    {
      "epoch": 3.2070159912109375e-05,
      "model_forward_time": 3.4549312591552734,
      "step": 8407
    },
    {
      "epoch": 3.2070159912109375e-05,
      "step": 8407,
      "training_step_time": 10.232285976409912
    },
    {
      "epoch": 3.2070159912109375e-05,
      "model_forward_time": 3.145810842514038,
      "step": 8407
    },
    {
      "dynamics_loss_avg": 0.03863688334822655,
      "epoch": 3.2070159912109375e-05,
      "step": 8407
    },
    {
      "action_loss_avg": 0.014147900696843862,
      "epoch": 3.2070159912109375e-05,
      "step": 8407
    },
    {
      "epoch": 3.2070159912109375e-05,
      "step": 8407,
      "training_step_time": 9.923783540725708
    },
    {
      "epoch": 3.2070159912109375e-05,
      "model_forward_time": 3.1493873596191406,
      "step": 8407
    },
    {
      "epoch": 3.2070159912109375e-05,
      "step": 8407,
      "training_step_time": 10.299424171447754
    },
    {
      "epoch": 3.2073974609375e-05,
      "model_forward_time": 3.1675260066986084,
      "step": 8408
    },
    {
      "epoch": 3.2073974609375e-05,
      "step": 8408,
      "training_step_time": 9.943942070007324
    },
    {
      "epoch": 3.2073974609375e-05,
      "model_forward_time": 3.1487059593200684,
      "step": 8408
    },
    {
      "epoch": 3.2073974609375e-05,
      "step": 8408,
      "training_step_time": 9.925544261932373
    },
    {
      "epoch": 3.2073974609375e-05,
      "model_forward_time": 3.1466903686523438,
      "step": 8408
    },
    {
      "epoch": 3.2073974609375e-05,
      "step": 8408,
      "training_step_time": 10.207454204559326
    },
    {
      "epoch": 3.2073974609375e-05,
      "model_forward_time": 3.1475741863250732,
      "step": 8408
    },
    {
      "epoch": 3.2073974609375e-05,
      "step": 8408,
      "training_step_time": 10.33629846572876
    },
    {
      "epoch": 3.207778930664063e-05,
      "model_forward_time": 3.1600470542907715,
      "step": 8409
    },
    {
      "epoch": 3.207778930664063e-05,
      "step": 8409,
      "training_step_time": 9.935334205627441
    },
    {
      "epoch": 3.207778930664063e-05,
      "model_forward_time": 3.141094446182251,
      "step": 8409
    },
    {
      "epoch": 3.207778930664063e-05,
      "step": 8409,
      "training_step_time": 9.922502517700195
    },
    {
      "epoch": 3.207778930664063e-05,
      "model_forward_time": 3.145719528198242,
      "step": 8409
    },
    {
      "epoch": 3.207778930664063e-05,
      "step": 8409,
      "training_step_time": 9.924511671066284
    },
    {
      "epoch": 3.207778930664063e-05,
      "model_forward_time": 3.4312820434570312,
      "step": 8409
    },
    {
      "epoch": 3.207778930664063e-05,
      "step": 8409,
      "training_step_time": 10.33033037185669
    },
    {
      "epoch": 3.208160400390625e-05,
      "grad_norm": 0.3725653886795044,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 8410
    },
    {
      "epoch": 3.208160400390625e-05,
      "model_forward_time": 3.1586270332336426,
      "step": 8410
    },
    {
      "dynamics_loss_avg": 0.06543295346200466,
      "epoch": 3.208160400390625e-05,
      "step": 8410
    },
    {
      "action_loss_avg": 0.02152804294601083,
      "epoch": 3.208160400390625e-05,
      "step": 8410
    },
    {
      "epoch": 3.208160400390625e-05,
      "step": 8410,
      "training_step_time": 9.94247317314148
    },
    {
      "epoch": 3.208160400390625e-05,
      "model_forward_time": 3.1456944942474365,
      "step": 8410
    },
    {
      "epoch": 3.208160400390625e-05,
      "step": 8410,
      "training_step_time": 9.924033403396606
    },
    {
      "epoch": 3.208160400390625e-05,
      "model_forward_time": 3.1490318775177,
      "step": 8410
    },
    {
      "epoch": 3.208160400390625e-05,
      "step": 8410,
      "training_step_time": 9.928240299224854
    },
    {
      "epoch": 3.208160400390625e-05,
      "model_forward_time": 3.1527814865112305,
      "step": 8410
    },
    {
      "epoch": 3.208160400390625e-05,
      "step": 8410,
      "training_step_time": 10.305155754089355
    },
    {
      "epoch": 3.2085418701171873e-05,
      "model_forward_time": 3.15787935256958,
      "step": 8411
    },
    {
      "epoch": 3.2085418701171873e-05,
      "step": 8411,
      "training_step_time": 10.21998929977417
    },
    {
      "epoch": 3.2085418701171873e-05,
      "model_forward_time": 3.1346118450164795,
      "step": 8411
    },
    {
      "epoch": 3.2085418701171873e-05,
      "step": 8411,
      "training_step_time": 9.909642934799194
    },
    {
      "epoch": 3.2085418701171873e-05,
      "model_forward_time": 3.139159917831421,
      "step": 8411
    },
    {
      "epoch": 3.2085418701171873e-05,
      "step": 8411,
      "training_step_time": 9.92223310470581
    },
    {
      "epoch": 3.2085418701171873e-05,
      "model_forward_time": 3.138261079788208,
      "step": 8411
    },
    {
      "epoch": 3.2085418701171873e-05,
      "step": 8411,
      "training_step_time": 10.033639192581177
    },
    {
      "epoch": 3.20892333984375e-05,
      "model_forward_time": 3.154231071472168,
      "step": 8412
    },
    {
      "epoch": 3.20892333984375e-05,
      "step": 8412,
      "training_step_time": 9.934837579727173
    },
    {
      "epoch": 3.20892333984375e-05,
      "model_forward_time": 3.4042601585388184,
      "step": 8412
    },
    {
      "epoch": 3.20892333984375e-05,
      "step": 8412,
      "training_step_time": 10.17926287651062
    },
    {
      "epoch": 3.20892333984375e-05,
      "model_forward_time": 3.146664619445801,
      "step": 8412
    },
    {
      "dynamics_loss_avg": 0.05583957005292177,
      "epoch": 3.20892333984375e-05,
      "step": 8412
    },
    {
      "action_loss_avg": 0.01830203542485833,
      "epoch": 3.20892333984375e-05,
      "step": 8412
    },
    {
      "epoch": 3.20892333984375e-05,
      "step": 8412,
      "training_step_time": 9.92955207824707
    },
    {
      "epoch": 3.20892333984375e-05,
      "model_forward_time": 3.1424448490142822,
      "step": 8412
    },
    {
      "epoch": 3.20892333984375e-05,
      "step": 8412,
      "training_step_time": 10.260915994644165
    },
    {
      "epoch": 3.2093048095703126e-05,
      "model_forward_time": 3.161500930786133,
      "step": 8413
    },
    {
      "epoch": 3.2093048095703126e-05,
      "step": 8413,
      "training_step_time": 9.935438632965088
    },
    {
      "epoch": 3.2093048095703126e-05,
      "model_forward_time": 3.144395351409912,
      "step": 8413
    },
    {
      "epoch": 3.2093048095703126e-05,
      "step": 8413,
      "training_step_time": 9.916193008422852
    },
    {
      "epoch": 3.2093048095703126e-05,
      "model_forward_time": 3.4308485984802246,
      "step": 8413
    },
    {
      "epoch": 3.2093048095703126e-05,
      "step": 8413,
      "training_step_time": 10.21207571029663
    },
    {
      "epoch": 3.2093048095703126e-05,
      "model_forward_time": 2.4370832443237305,
      "step": 8413
    },
    {
      "epoch": 3.2093048095703126e-05,
      "step": 8413,
      "training_step_time": 10.266928672790527
    },
    {
      "epoch": 3.209686279296875e-05,
      "model_forward_time": 3.1525468826293945,
      "step": 8414
    },
    {
      "epoch": 3.209686279296875e-05,
      "step": 8414,
      "training_step_time": 9.933954238891602
    },
    {
      "epoch": 3.209686279296875e-05,
      "model_forward_time": 3.1381778717041016,
      "step": 8414
    },
    {
      "epoch": 3.209686279296875e-05,
      "step": 8414,
      "training_step_time": 9.910707473754883
    },
    {
      "epoch": 3.209686279296875e-05,
      "model_forward_time": 3.1364474296569824,
      "step": 8414
    },
    {
      "epoch": 3.209686279296875e-05,
      "step": 8414,
      "training_step_time": 9.91139817237854
    },
    {
      "epoch": 3.209686279296875e-05,
      "model_forward_time": 3.395500421524048,
      "step": 8414
    },
    {
      "epoch": 3.209686279296875e-05,
      "step": 8414,
      "training_step_time": 10.27642297744751
    },
    {
      "epoch": 3.210067749023437e-05,
      "model_forward_time": 3.157912254333496,
      "step": 8415
    },
    {
      "dynamics_loss_avg": 0.0608082078397274,
      "epoch": 3.210067749023437e-05,
      "step": 8415
    },
    {
      "action_loss_avg": 0.020962748769670726,
      "epoch": 3.210067749023437e-05,
      "step": 8415
    },
    {
      "epoch": 3.210067749023437e-05,
      "step": 8415,
      "training_step_time": 9.930947542190552
    },
    {
      "epoch": 3.210067749023437e-05,
      "model_forward_time": 3.136265516281128,
      "step": 8415
    },
    {
      "epoch": 3.210067749023437e-05,
      "step": 8415,
      "training_step_time": 9.904624938964844
    },
    {
      "epoch": 3.210067749023437e-05,
      "model_forward_time": 3.13191819190979,
      "step": 8415
    },
    {
      "epoch": 3.210067749023437e-05,
      "step": 8415,
      "training_step_time": 9.898450374603271
    },
    {
      "epoch": 3.210067749023437e-05,
      "model_forward_time": 3.136140823364258,
      "step": 8415
    },
    {
      "epoch": 3.210067749023437e-05,
      "step": 8415,
      "training_step_time": 10.270408868789673
    },
    {
      "epoch": 3.21044921875e-05,
      "model_forward_time": 3.143723726272583,
      "step": 8416
    },
    {
      "epoch": 3.21044921875e-05,
      "step": 8416,
      "training_step_time": 10.185969114303589
    },
    {
      "epoch": 3.21044921875e-05,
      "model_forward_time": 3.1395318508148193,
      "step": 8416
    },
    {
      "epoch": 3.21044921875e-05,
      "step": 8416,
      "training_step_time": 9.915965557098389
    },
    {
      "epoch": 3.21044921875e-05,
      "model_forward_time": 3.1439013481140137,
      "step": 8416
    },
    {
      "epoch": 3.21044921875e-05,
      "step": 8416,
      "training_step_time": 9.920399188995361
    },
    {
      "epoch": 3.21044921875e-05,
      "model_forward_time": 3.142455577850342,
      "step": 8416
    },
    {
      "epoch": 3.21044921875e-05,
      "step": 8416,
      "training_step_time": 10.261110305786133
    },
    {
      "epoch": 3.2108306884765624e-05,
      "model_forward_time": 3.1552717685699463,
      "step": 8417
    },
    {
      "epoch": 3.2108306884765624e-05,
      "step": 8417,
      "training_step_time": 9.936763286590576
    },
    {
      "epoch": 3.2108306884765624e-05,
      "model_forward_time": 3.4191901683807373,
      "step": 8417
    },
    {
      "epoch": 3.2108306884765624e-05,
      "step": 8417,
      "training_step_time": 10.190654039382935
    },
    {
      "epoch": 3.2108306884765624e-05,
      "model_forward_time": 3.14070987701416,
      "step": 8417
    },
    {
      "dynamics_loss_avg": 0.06516428645700216,
      "epoch": 3.2108306884765624e-05,
      "step": 8417
    },
    {
      "action_loss_avg": 0.01926016677170992,
      "epoch": 3.2108306884765624e-05,
      "step": 8417
    },
    {
      "epoch": 3.2108306884765624e-05,
      "step": 8417,
      "training_step_time": 9.910477876663208
    },
    {
      "epoch": 3.2108306884765624e-05,
      "model_forward_time": 3.141723155975342,
      "step": 8417
    },
    {
      "epoch": 3.2108306884765624e-05,
      "step": 8417,
      "training_step_time": 10.244430303573608
    },
    {
      "epoch": 3.211212158203125e-05,
      "model_forward_time": 3.1575534343719482,
      "step": 8418
    },
    {
      "epoch": 3.211212158203125e-05,
      "step": 8418,
      "training_step_time": 9.935190439224243
    },
    {
      "epoch": 3.211212158203125e-05,
      "model_forward_time": 3.1339426040649414,
      "step": 8418
    },
    {
      "epoch": 3.211212158203125e-05,
      "step": 8418,
      "training_step_time": 9.908148288726807
    },
    {
      "epoch": 3.211212158203125e-05,
      "model_forward_time": 3.401318311691284,
      "step": 8418
    },
    {
      "epoch": 3.211212158203125e-05,
      "step": 8418,
      "training_step_time": 10.170616388320923
    },
    {
      "epoch": 3.211212158203125e-05,
      "model_forward_time": 3.1382832527160645,
      "step": 8418
    },
    {
      "epoch": 3.211212158203125e-05,
      "step": 8418,
      "training_step_time": 10.307533264160156
    },
    {
      "epoch": 3.211593627929688e-05,
      "model_forward_time": 3.1596455574035645,
      "step": 8419
    },
    {
      "epoch": 3.211593627929688e-05,
      "step": 8419,
      "training_step_time": 9.945164918899536
    },
    {
      "epoch": 3.211593627929688e-05,
      "model_forward_time": 3.1382510662078857,
      "step": 8419
    },
    {
      "epoch": 3.211593627929688e-05,
      "step": 8419,
      "training_step_time": 9.91430950164795
    },
    {
      "epoch": 3.211593627929688e-05,
      "model_forward_time": 3.14335560798645,
      "step": 8419
    },
    {
      "epoch": 3.211593627929688e-05,
      "step": 8419,
      "training_step_time": 9.919947862625122
    },
    {
      "epoch": 3.211593627929688e-05,
      "model_forward_time": 3.455695152282715,
      "step": 8419
    },
    {
      "epoch": 3.211593627929688e-05,
      "step": 8419,
      "training_step_time": 10.331782102584839
    },
    {
      "epoch": 3.21197509765625e-05,
      "grad_norm": 0.4108879566192627,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 8420
    },
    {
      "epoch": 3.21197509765625e-05,
      "model_forward_time": 3.160149097442627,
      "step": 8420
    },
    {
      "dynamics_loss_avg": 0.05910341646522284,
      "epoch": 3.21197509765625e-05,
      "step": 8420
    },
    {
      "action_loss_avg": 0.02158716581761837,
      "epoch": 3.21197509765625e-05,
      "step": 8420
    },
    {
      "epoch": 3.21197509765625e-05,
      "step": 8420,
      "training_step_time": 9.94700026512146
    },
    {
      "epoch": 3.21197509765625e-05,
      "model_forward_time": 3.1412885189056396,
      "step": 8420
    },
    {
      "epoch": 3.21197509765625e-05,
      "step": 8420,
      "training_step_time": 9.920030355453491
    },
    {
      "epoch": 3.21197509765625e-05,
      "model_forward_time": 3.1462934017181396,
      "step": 8420
    },
    {
      "epoch": 3.21197509765625e-05,
      "step": 8420,
      "training_step_time": 9.92505168914795
    },
    {
      "epoch": 3.21197509765625e-05,
      "model_forward_time": 3.1363883018493652,
      "step": 8420
    },
    {
      "epoch": 3.21197509765625e-05,
      "step": 8420,
      "training_step_time": 10.24826455116272
    },
    {
      "epoch": 3.212356567382812e-05,
      "model_forward_time": 3.447772264480591,
      "step": 8421
    },
    {
      "epoch": 3.212356567382812e-05,
      "step": 8421,
      "training_step_time": 10.227387189865112
    },
    {
      "epoch": 3.212356567382812e-05,
      "model_forward_time": 3.1369636058807373,
      "step": 8421
    },
    {
      "epoch": 3.212356567382812e-05,
      "step": 8421,
      "training_step_time": 9.911860704421997
    },
    {
      "epoch": 3.212356567382812e-05,
      "model_forward_time": 3.1450037956237793,
      "step": 8421
    },
    {
      "epoch": 3.212356567382812e-05,
      "step": 8421,
      "training_step_time": 9.922529458999634
    },
    {
      "epoch": 3.212356567382812e-05,
      "model_forward_time": 3.1404197216033936,
      "step": 8421
    },
    {
      "epoch": 3.212356567382812e-05,
      "step": 8421,
      "training_step_time": 10.32499098777771
    },
    {
      "epoch": 3.212738037109375e-05,
      "model_forward_time": 3.1539719104766846,
      "step": 8422
    },
    {
      "epoch": 3.212738037109375e-05,
      "step": 8422,
      "training_step_time": 9.931426048278809
    },
    {
      "epoch": 3.212738037109375e-05,
      "model_forward_time": 3.4191513061523438,
      "step": 8422
    },
    {
      "epoch": 3.212738037109375e-05,
      "step": 8422,
      "training_step_time": 10.193369388580322
    },
    {
      "epoch": 3.212738037109375e-05,
      "model_forward_time": 3.1399309635162354,
      "step": 8422
    },
    {
      "dynamics_loss_avg": 0.0672353370115161,
      "epoch": 3.212738037109375e-05,
      "step": 8422
    },
    {
      "action_loss_avg": 0.02435255702584982,
      "epoch": 3.212738037109375e-05,
      "step": 8422
    },
    {
      "epoch": 3.212738037109375e-05,
      "step": 8422,
      "training_step_time": 9.917365550994873
    },
    {
      "epoch": 3.212738037109375e-05,
      "model_forward_time": 3.1325571537017822,
      "step": 8422
    },
    {
      "epoch": 3.212738037109375e-05,
      "step": 8422,
      "training_step_time": 10.007388591766357
    },
    {
      "epoch": 3.2131195068359375e-05,
      "model_forward_time": 3.145205497741699,
      "step": 8423
    },
    {
      "epoch": 3.2131195068359375e-05,
      "step": 8423,
      "training_step_time": 9.920753240585327
    },
    {
      "epoch": 3.2131195068359375e-05,
      "model_forward_time": 3.137087106704712,
      "step": 8423
    },
    {
      "epoch": 3.2131195068359375e-05,
      "step": 8423,
      "training_step_time": 9.909816265106201
    },
    {
      "epoch": 3.2131195068359375e-05,
      "model_forward_time": 3.412986993789673,
      "step": 8423
    },
    {
      "epoch": 3.2131195068359375e-05,
      "step": 8423,
      "training_step_time": 10.17991328239441
    },
    {
      "epoch": 3.2131195068359375e-05,
      "model_forward_time": 3.134598731994629,
      "step": 8423
    },
    {
      "epoch": 3.2131195068359375e-05,
      "step": 8423,
      "training_step_time": 10.25653862953186
    },
    {
      "epoch": 3.2135009765625e-05,
      "model_forward_time": 3.1460206508636475,
      "step": 8424
    },
    {
      "epoch": 3.2135009765625e-05,
      "step": 8424,
      "training_step_time": 9.924576044082642
    },
    {
      "epoch": 3.2135009765625e-05,
      "model_forward_time": 3.133878707885742,
      "step": 8424
    },
    {
      "epoch": 3.2135009765625e-05,
      "step": 8424,
      "training_step_time": 9.903497457504272
    },
    {
      "epoch": 3.2135009765625e-05,
      "model_forward_time": 3.132796049118042,
      "step": 8424
    },
    {
      "epoch": 3.2135009765625e-05,
      "step": 8424,
      "training_step_time": 9.915052890777588
    },
    {
      "epoch": 3.2135009765625e-05,
      "model_forward_time": 3.407148838043213,
      "step": 8424
    },
    {
      "epoch": 3.2135009765625e-05,
      "step": 8424,
      "training_step_time": 10.284363508224487
    },
    {
      "epoch": 3.213882446289063e-05,
      "model_forward_time": 3.1660706996917725,
      "step": 8425
    },
    {
      "dynamics_loss_avg": 0.04784854128956795,
      "epoch": 3.213882446289063e-05,
      "step": 8425
    },
    {
      "action_loss_avg": 0.020607734378427268,
      "epoch": 3.213882446289063e-05,
      "step": 8425
    },
    {
      "epoch": 3.213882446289063e-05,
      "step": 8425,
      "training_step_time": 9.953644514083862
    },
    {
      "epoch": 3.213882446289063e-05,
      "model_forward_time": 3.1441564559936523,
      "step": 8425
    },
    {
      "epoch": 3.213882446289063e-05,
      "step": 8425,
      "training_step_time": 9.919018030166626
    },
    {
      "epoch": 3.213882446289063e-05,
      "model_forward_time": 3.1414785385131836,
      "step": 8425
    },
    {
      "epoch": 3.213882446289063e-05,
      "step": 8425,
      "training_step_time": 9.912714004516602
    },
    {
      "epoch": 3.213882446289063e-05,
      "model_forward_time": 3.143477201461792,
      "step": 8425
    },
    {
      "epoch": 3.213882446289063e-05,
      "step": 8425,
      "training_step_time": 10.27553677558899
    },
    {
      "epoch": 3.214263916015625e-05,
      "model_forward_time": 3.418281078338623,
      "step": 8426
    },
    {
      "epoch": 3.214263916015625e-05,
      "step": 8426,
      "training_step_time": 10.196352243423462
    },
    {
      "epoch": 3.214263916015625e-05,
      "model_forward_time": 2.439610242843628,
      "step": 8426
    },
    {
      "epoch": 3.214263916015625e-05,
      "step": 8426,
      "training_step_time": 7.47128701210022
    },
    {
      "epoch": 3.214263916015625e-05,
      "model_forward_time": 3.139982223510742,
      "step": 8426
    },
    {
      "epoch": 3.214263916015625e-05,
      "step": 8426,
      "training_step_time": 9.914214134216309
    },
    {
      "epoch": 3.214263916015625e-05,
      "model_forward_time": 3.148054361343384,
      "step": 8426
    },
    {
      "epoch": 3.214263916015625e-05,
      "step": 8426,
      "training_step_time": 10.265441417694092
    },
    {
      "epoch": 3.2146453857421873e-05,
      "model_forward_time": 3.1567397117614746,
      "step": 8427
    },
    {
      "epoch": 3.2146453857421873e-05,
      "step": 8427,
      "training_step_time": 9.938327550888062
    },
    {
      "epoch": 3.2146453857421873e-05,
      "model_forward_time": 3.438973903656006,
      "step": 8427
    },
    {
      "epoch": 3.2146453857421873e-05,
      "step": 8427,
      "training_step_time": 10.21784782409668
    },
    {
      "epoch": 3.2146453857421873e-05,
      "model_forward_time": 3.1430256366729736,
      "step": 8427
    },
    {
      "dynamics_loss_avg": 0.06934124864637851,
      "epoch": 3.2146453857421873e-05,
      "step": 8427
    },
    {
      "action_loss_avg": 0.02393217645585537,
      "epoch": 3.2146453857421873e-05,
      "step": 8427
    },
    {
      "epoch": 3.2146453857421873e-05,
      "step": 8427,
      "training_step_time": 9.928190469741821
    },
    {
      "epoch": 3.2146453857421873e-05,
      "model_forward_time": 3.1500442028045654,
      "step": 8427
    },
    {
      "epoch": 3.2146453857421873e-05,
      "step": 8427,
      "training_step_time": 10.290653705596924
    },
    {
      "epoch": 3.21502685546875e-05,
      "model_forward_time": 3.156532049179077,
      "step": 8428
    },
    {
      "epoch": 3.21502685546875e-05,
      "step": 8428,
      "training_step_time": 9.937724113464355
    },
    {
      "epoch": 3.21502685546875e-05,
      "model_forward_time": 3.1574606895446777,
      "step": 8428
    },
    {
      "epoch": 3.21502685546875e-05,
      "step": 8428,
      "training_step_time": 9.93861436843872
    },
    {
      "epoch": 3.21502685546875e-05,
      "model_forward_time": 3.1534433364868164,
      "step": 8428
    },
    {
      "epoch": 3.21502685546875e-05,
      "step": 8428,
      "training_step_time": 10.261317014694214
    },
    {
      "epoch": 3.21502685546875e-05,
      "model_forward_time": 3.146310329437256,
      "step": 8428
    },
    {
      "epoch": 3.21502685546875e-05,
      "step": 8428,
      "training_step_time": 10.297572135925293
    },
    {
      "epoch": 3.2154083251953126e-05,
      "model_forward_time": 3.159931182861328,
      "step": 8429
    },
    {
      "epoch": 3.2154083251953126e-05,
      "step": 8429,
      "training_step_time": 9.93777322769165
    },
    {
      "epoch": 3.2154083251953126e-05,
      "model_forward_time": 3.147287368774414,
      "step": 8429
    },
    {
      "epoch": 3.2154083251953126e-05,
      "step": 8429,
      "training_step_time": 9.928380012512207
    },
    {
      "epoch": 3.2154083251953126e-05,
      "model_forward_time": 3.1477408409118652,
      "step": 8429
    },
    {
      "epoch": 3.2154083251953126e-05,
      "step": 8429,
      "training_step_time": 9.929943323135376
    },
    {
      "epoch": 3.2154083251953126e-05,
      "model_forward_time": 3.435425281524658,
      "step": 8429
    },
    {
      "epoch": 3.2154083251953126e-05,
      "step": 8429,
      "training_step_time": 10.309613227844238
    },
    {
      "epoch": 3.215789794921875e-05,
      "grad_norm": 0.4044230878353119,
      "learning_rate": 1e-05,
      "loss": 0.0808,
      "step": 8430
    },
    {
      "epoch": 3.215789794921875e-05,
      "model_forward_time": 3.1659576892852783,
      "step": 8430
    },
    {
      "dynamics_loss_avg": 0.056716890819370744,
      "epoch": 3.215789794921875e-05,
      "step": 8430
    },
    {
      "action_loss_avg": 0.01253080260939896,
      "epoch": 3.215789794921875e-05,
      "step": 8430
    },
    {
      "epoch": 3.215789794921875e-05,
      "step": 8430,
      "training_step_time": 9.95043420791626
    },
    {
      "epoch": 3.215789794921875e-05,
      "model_forward_time": 2.4500930309295654,
      "step": 8430
    },
    {
      "epoch": 3.215789794921875e-05,
      "step": 8430,
      "training_step_time": 7.485490560531616
    },
    {
      "epoch": 3.215789794921875e-05,
      "model_forward_time": 3.145242214202881,
      "step": 8430
    },
    {
      "epoch": 3.215789794921875e-05,
      "step": 8430,
      "training_step_time": 9.92503046989441
    },
    {
      "epoch": 3.215789794921875e-05,
      "model_forward_time": 3.1450862884521484,
      "step": 8430
    },
    {
      "epoch": 3.215789794921875e-05,
      "step": 8430,
      "training_step_time": 10.300585746765137
    },
    {
      "epoch": 3.216171264648437e-05,
      "model_forward_time": 3.1568243503570557,
      "step": 8431
    },
    {
      "epoch": 3.216171264648437e-05,
      "step": 8431,
      "training_step_time": 9.941230773925781
    },
    {
      "epoch": 3.216171264648437e-05,
      "model_forward_time": 3.3915538787841797,
      "step": 8431
    },
    {
      "epoch": 3.216171264648437e-05,
      "step": 8431,
      "training_step_time": 10.16850757598877
    },
    {
      "epoch": 3.216171264648437e-05,
      "model_forward_time": 3.1497271060943604,
      "step": 8431
    },
    {
      "epoch": 3.216171264648437e-05,
      "step": 8431,
      "training_step_time": 9.930726289749146
    },
    {
      "epoch": 3.216171264648437e-05,
      "model_forward_time": 2.4438233375549316,
      "step": 8431
    },
    {
      "epoch": 3.216171264648437e-05,
      "step": 8431,
      "training_step_time": 10.302873134613037
    },
    {
      "epoch": 3.216552734375e-05,
      "model_forward_time": 3.1619837284088135,
      "step": 8432
    },
    {
      "epoch": 3.216552734375e-05,
      "step": 8432,
      "training_step_time": 9.941313743591309
    },
    {
      "epoch": 3.216552734375e-05,
      "model_forward_time": 3.456224203109741,
      "step": 8432
    },
    {
      "epoch": 3.216552734375e-05,
      "step": 8432,
      "training_step_time": 10.23840045928955
    },
    {
      "epoch": 3.216552734375e-05,
      "model_forward_time": 3.1467065811157227,
      "step": 8432
    },
    {
      "dynamics_loss_avg": 0.0635410537943244,
      "epoch": 3.216552734375e-05,
      "step": 8432
    },
    {
      "action_loss_avg": 0.020795615296810864,
      "epoch": 3.216552734375e-05,
      "step": 8432
    },
    {
      "epoch": 3.216552734375e-05,
      "step": 8432,
      "training_step_time": 9.920097827911377
    },
    {
      "epoch": 3.216552734375e-05,
      "model_forward_time": 3.144589424133301,
      "step": 8432
    },
    {
      "epoch": 3.216552734375e-05,
      "step": 8432,
      "training_step_time": 10.310508966445923
    },
    {
      "epoch": 3.2169342041015624e-05,
      "model_forward_time": 3.1608219146728516,
      "step": 8433
    },
    {
      "epoch": 3.2169342041015624e-05,
      "step": 8433,
      "training_step_time": 9.944088459014893
    },
    {
      "epoch": 3.2169342041015624e-05,
      "model_forward_time": 3.1405913829803467,
      "step": 8433
    },
    {
      "epoch": 3.2169342041015624e-05,
      "step": 8433,
      "training_step_time": 9.914491891860962
    },
    {
      "epoch": 3.2169342041015624e-05,
      "model_forward_time": 3.137688636779785,
      "step": 8433
    },
    {
      "epoch": 3.2169342041015624e-05,
      "step": 8433,
      "training_step_time": 10.167004346847534
    },
    {
      "epoch": 3.2169342041015624e-05,
      "model_forward_time": 3.1359620094299316,
      "step": 8433
    },
    {
      "epoch": 3.2169342041015624e-05,
      "step": 8433,
      "training_step_time": 10.241731882095337
    },
    {
      "epoch": 3.217315673828125e-05,
      "model_forward_time": 3.155459403991699,
      "step": 8434
    },
    {
      "epoch": 3.217315673828125e-05,
      "step": 8434,
      "training_step_time": 9.929635286331177
    },
    {
      "epoch": 3.217315673828125e-05,
      "model_forward_time": 3.143947124481201,
      "step": 8434
    },
    {
      "epoch": 3.217315673828125e-05,
      "step": 8434,
      "training_step_time": 9.926121950149536
    },
    {
      "epoch": 3.217315673828125e-05,
      "model_forward_time": 3.142606735229492,
      "step": 8434
    },
    {
      "epoch": 3.217315673828125e-05,
      "step": 8434,
      "training_step_time": 9.923880100250244
    },
    {
      "epoch": 3.217315673828125e-05,
      "model_forward_time": 3.3969247341156006,
      "step": 8434
    },
    {
      "epoch": 3.217315673828125e-05,
      "step": 8434,
      "training_step_time": 10.301298141479492
    },
    {
      "epoch": 3.217697143554688e-05,
      "model_forward_time": 3.1489005088806152,
      "step": 8435
    },
    {
      "dynamics_loss_avg": 0.06698398739099502,
      "epoch": 3.217697143554688e-05,
      "step": 8435
    },
    {
      "action_loss_avg": 0.02619970953091979,
      "epoch": 3.217697143554688e-05,
      "step": 8435
    },
    {
      "epoch": 3.217697143554688e-05,
      "step": 8435,
      "training_step_time": 9.926851511001587
    },
    {
      "epoch": 3.217697143554688e-05,
      "model_forward_time": 3.1452691555023193,
      "step": 8435
    },
    {
      "epoch": 3.217697143554688e-05,
      "step": 8435,
      "training_step_time": 9.91906476020813
    },
    {
      "epoch": 3.217697143554688e-05,
      "model_forward_time": 3.1435537338256836,
      "step": 8435
    },
    {
      "epoch": 3.217697143554688e-05,
      "step": 8435,
      "training_step_time": 9.921832799911499
    },
    {
      "epoch": 3.217697143554688e-05,
      "model_forward_time": 3.1481990814208984,
      "step": 8435
    },
    {
      "epoch": 3.217697143554688e-05,
      "step": 8435,
      "training_step_time": 10.323315382003784
    },
    {
      "epoch": 3.21807861328125e-05,
      "model_forward_time": 3.413740396499634,
      "step": 8436
    },
    {
      "epoch": 3.21807861328125e-05,
      "step": 8436,
      "training_step_time": 10.191691637039185
    },
    {
      "epoch": 3.21807861328125e-05,
      "model_forward_time": 3.1505677700042725,
      "step": 8436
    },
    {
      "epoch": 3.21807861328125e-05,
      "step": 8436,
      "training_step_time": 9.927714824676514
    },
    {
      "epoch": 3.21807861328125e-05,
      "model_forward_time": 3.143982410430908,
      "step": 8436
    },
    {
      "epoch": 3.21807861328125e-05,
      "step": 8436,
      "training_step_time": 9.91929030418396
    },
    {
      "epoch": 3.21807861328125e-05,
      "model_forward_time": 3.1465656757354736,
      "step": 8436
    },
    {
      "epoch": 3.21807861328125e-05,
      "step": 8436,
      "training_step_time": 10.298851728439331
    },
    {
      "epoch": 3.218460083007812e-05,
      "model_forward_time": 3.160853147506714,
      "step": 8437
    },
    {
      "epoch": 3.218460083007812e-05,
      "step": 8437,
      "training_step_time": 9.93819785118103
    },
    {
      "epoch": 3.218460083007812e-05,
      "model_forward_time": 3.474198341369629,
      "step": 8437
    },
    {
      "epoch": 3.218460083007812e-05,
      "step": 8437,
      "training_step_time": 10.256499528884888
    },
    {
      "epoch": 3.218460083007812e-05,
      "model_forward_time": 3.1531059741973877,
      "step": 8437
    },
    {
      "dynamics_loss_avg": 0.05110359424725175,
      "epoch": 3.218460083007812e-05,
      "step": 8437
    },
    {
      "action_loss_avg": 0.016509108780883252,
      "epoch": 3.218460083007812e-05,
      "step": 8437
    },
    {
      "epoch": 3.218460083007812e-05,
      "step": 8437,
      "training_step_time": 9.936712265014648
    },
    {
      "epoch": 3.218460083007812e-05,
      "model_forward_time": 3.153303623199463,
      "step": 8437
    },
    {
      "epoch": 3.218460083007812e-05,
      "step": 8437,
      "training_step_time": 10.033568143844604
    },
    {
      "epoch": 3.218841552734375e-05,
      "model_forward_time": 3.1774280071258545,
      "step": 8438
    },
    {
      "epoch": 3.218841552734375e-05,
      "step": 8438,
      "training_step_time": 9.963112831115723
    },
    {
      "epoch": 3.218841552734375e-05,
      "model_forward_time": 3.1428449153900146,
      "step": 8438
    },
    {
      "epoch": 3.218841552734375e-05,
      "step": 8438,
      "training_step_time": 9.923973083496094
    },
    {
      "epoch": 3.218841552734375e-05,
      "model_forward_time": 3.452547311782837,
      "step": 8438
    },
    {
      "epoch": 3.218841552734375e-05,
      "step": 8438,
      "training_step_time": 10.237019538879395
    },
    {
      "epoch": 3.218841552734375e-05,
      "model_forward_time": 3.153102397918701,
      "step": 8438
    },
    {
      "epoch": 3.218841552734375e-05,
      "step": 8438,
      "training_step_time": 10.246525287628174
    },
    {
      "epoch": 3.2192230224609375e-05,
      "model_forward_time": 3.1802144050598145,
      "step": 8439
    },
    {
      "epoch": 3.2192230224609375e-05,
      "step": 8439,
      "training_step_time": 9.956713676452637
    },
    {
      "epoch": 3.2192230224609375e-05,
      "model_forward_time": 3.1452741622924805,
      "step": 8439
    },
    {
      "epoch": 3.2192230224609375e-05,
      "step": 8439,
      "training_step_time": 9.924330234527588
    },
    {
      "epoch": 3.2192230224609375e-05,
      "model_forward_time": 3.1484761238098145,
      "step": 8439
    },
    {
      "epoch": 3.2192230224609375e-05,
      "step": 8439,
      "training_step_time": 9.9276123046875
    },
    {
      "epoch": 3.2192230224609375e-05,
      "model_forward_time": 3.446270227432251,
      "step": 8439
    },
    {
      "epoch": 3.2192230224609375e-05,
      "step": 8439,
      "training_step_time": 10.318624258041382
    },
    {
      "epoch": 3.2196044921875e-05,
      "grad_norm": 0.3653946816921234,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 8440
    },
    {
      "epoch": 3.2196044921875e-05,
      "model_forward_time": 3.167818784713745,
      "step": 8440
    },
    {
      "dynamics_loss_avg": 0.07615211606025696,
      "epoch": 3.2196044921875e-05,
      "step": 8440
    },
    {
      "action_loss_avg": 0.02764995153993368,
      "epoch": 3.2196044921875e-05,
      "step": 8440
    },
    {
      "epoch": 3.2196044921875e-05,
      "step": 8440,
      "training_step_time": 9.96044135093689
    },
    {
      "epoch": 3.2196044921875e-05,
      "model_forward_time": 3.1437828540802,
      "step": 8440
    },
    {
      "epoch": 3.2196044921875e-05,
      "step": 8440,
      "training_step_time": 9.920720100402832
    },
    {
      "epoch": 3.2196044921875e-05,
      "model_forward_time": 3.1480813026428223,
      "step": 8440
    },
    {
      "epoch": 3.2196044921875e-05,
      "step": 8440,
      "training_step_time": 9.92874002456665
    },
    {
      "epoch": 3.2196044921875e-05,
      "model_forward_time": 3.136540174484253,
      "step": 8440
    },
    {
      "epoch": 3.2196044921875e-05,
      "step": 8440,
      "training_step_time": 10.30340838432312
    },
    {
      "epoch": 3.219985961914063e-05,
      "model_forward_time": 3.4607784748077393,
      "step": 8441
    },
    {
      "epoch": 3.219985961914063e-05,
      "step": 8441,
      "training_step_time": 10.24654507637024
    },
    {
      "epoch": 3.219985961914063e-05,
      "model_forward_time": 3.1401634216308594,
      "step": 8441
    },
    {
      "epoch": 3.219985961914063e-05,
      "step": 8441,
      "training_step_time": 9.918259382247925
    },
    {
      "epoch": 3.219985961914063e-05,
      "model_forward_time": 3.1395909786224365,
      "step": 8441
    },
    {
      "epoch": 3.219985961914063e-05,
      "step": 8441,
      "training_step_time": 9.917827844619751
    },
    {
      "epoch": 3.219985961914063e-05,
      "model_forward_time": 3.1366074085235596,
      "step": 8441
    },
    {
      "epoch": 3.219985961914063e-05,
      "step": 8441,
      "training_step_time": 10.307340621948242
    },
    {
      "epoch": 3.220367431640625e-05,
      "model_forward_time": 3.1496665477752686,
      "step": 8442
    },
    {
      "epoch": 3.220367431640625e-05,
      "step": 8442,
      "training_step_time": 9.930001258850098
    },
    {
      "epoch": 3.220367431640625e-05,
      "model_forward_time": 3.3897409439086914,
      "step": 8442
    },
    {
      "epoch": 3.220367431640625e-05,
      "step": 8442,
      "training_step_time": 10.166388273239136
    },
    {
      "epoch": 3.220367431640625e-05,
      "model_forward_time": 3.1385116577148438,
      "step": 8442
    },
    {
      "dynamics_loss_avg": 0.060884840227663516,
      "epoch": 3.220367431640625e-05,
      "step": 8442
    },
    {
      "action_loss_avg": 0.020563617791049184,
      "epoch": 3.220367431640625e-05,
      "step": 8442
    },
    {
      "epoch": 3.220367431640625e-05,
      "step": 8442,
      "training_step_time": 9.917234659194946
    },
    {
      "epoch": 3.220367431640625e-05,
      "model_forward_time": 3.133411169052124,
      "step": 8442
    },
    {
      "epoch": 3.220367431640625e-05,
      "step": 8442,
      "training_step_time": 10.014459371566772
    },
    {
      "epoch": 3.2207489013671874e-05,
      "model_forward_time": 3.146421432495117,
      "step": 8443
    },
    {
      "epoch": 3.2207489013671874e-05,
      "step": 8443,
      "training_step_time": 9.920468807220459
    },
    {
      "epoch": 3.2207489013671874e-05,
      "model_forward_time": 2.4294893741607666,
      "step": 8443
    },
    {
      "epoch": 3.2207489013671874e-05,
      "step": 8443,
      "training_step_time": 7.460841655731201
    },
    {
      "epoch": 3.2207489013671874e-05,
      "model_forward_time": 3.4320592880249023,
      "step": 8443
    },
    {
      "epoch": 3.2207489013671874e-05,
      "step": 8443,
      "training_step_time": 10.212164163589478
    },
    {
      "epoch": 3.2207489013671874e-05,
      "model_forward_time": 3.1394307613372803,
      "step": 8443
    },
    {
      "epoch": 3.2207489013671874e-05,
      "step": 8443,
      "training_step_time": 10.277238130569458
    },
    {
      "epoch": 3.22113037109375e-05,
      "model_forward_time": 3.159162998199463,
      "step": 8444
    },
    {
      "epoch": 3.22113037109375e-05,
      "step": 8444,
      "training_step_time": 9.937272787094116
    },
    {
      "epoch": 3.22113037109375e-05,
      "model_forward_time": 3.1426186561584473,
      "step": 8444
    },
    {
      "epoch": 3.22113037109375e-05,
      "step": 8444,
      "training_step_time": 9.920468091964722
    },
    {
      "epoch": 3.22113037109375e-05,
      "model_forward_time": 2.4337711334228516,
      "step": 8444
    },
    {
      "epoch": 3.22113037109375e-05,
      "step": 8444,
      "training_step_time": 7.467534780502319
    },
    {
      "epoch": 3.22113037109375e-05,
      "model_forward_time": 3.1369271278381348,
      "step": 8444
    },
    {
      "epoch": 3.22113037109375e-05,
      "step": 8444,
      "training_step_time": 10.30575966835022
    },
    {
      "epoch": 3.2215118408203126e-05,
      "model_forward_time": 3.1586437225341797,
      "step": 8445
    },
    {
      "dynamics_loss_avg": 0.07272461634129286,
      "epoch": 3.2215118408203126e-05,
      "step": 8445
    },
    {
      "action_loss_avg": 0.020080596255138517,
      "epoch": 3.2215118408203126e-05,
      "step": 8445
    },
    {
      "epoch": 3.2215118408203126e-05,
      "step": 8445,
      "training_step_time": 9.934911012649536
    },
    {
      "epoch": 3.2215118408203126e-05,
      "model_forward_time": 3.1339635848999023,
      "step": 8445
    },
    {
      "epoch": 3.2215118408203126e-05,
      "step": 8445,
      "training_step_time": 9.908633947372437
    },
    {
      "epoch": 3.2215118408203126e-05,
      "model_forward_time": 3.1362876892089844,
      "step": 8445
    },
    {
      "epoch": 3.2215118408203126e-05,
      "step": 8445,
      "training_step_time": 9.909983158111572
    },
    {
      "epoch": 3.2215118408203126e-05,
      "model_forward_time": 3.1387181282043457,
      "step": 8445
    },
    {
      "epoch": 3.2215118408203126e-05,
      "step": 8445,
      "training_step_time": 10.274337768554688
    },
    {
      "epoch": 3.221893310546875e-05,
      "model_forward_time": 3.4021503925323486,
      "step": 8446
    },
    {
      "epoch": 3.221893310546875e-05,
      "step": 8446,
      "training_step_time": 10.177440404891968
    },
    {
      "epoch": 3.221893310546875e-05,
      "model_forward_time": 3.1416046619415283,
      "step": 8446
    },
    {
      "epoch": 3.221893310546875e-05,
      "step": 8446,
      "training_step_time": 9.916706800460815
    },
    {
      "epoch": 3.221893310546875e-05,
      "model_forward_time": 3.138864040374756,
      "step": 8446
    },
    {
      "epoch": 3.221893310546875e-05,
      "step": 8446,
      "training_step_time": 9.915672302246094
    },
    {
      "epoch": 3.221893310546875e-05,
      "model_forward_time": 3.1428897380828857,
      "step": 8446
    },
    {
      "epoch": 3.221893310546875e-05,
      "step": 8446,
      "training_step_time": 10.350502252578735
    },
    {
      "epoch": 3.222274780273437e-05,
      "model_forward_time": 3.162285327911377,
      "step": 8447
    },
    {
      "epoch": 3.222274780273437e-05,
      "step": 8447,
      "training_step_time": 9.943382740020752
    },
    {
      "epoch": 3.222274780273437e-05,
      "model_forward_time": 3.426722526550293,
      "step": 8447
    },
    {
      "epoch": 3.222274780273437e-05,
      "step": 8447,
      "training_step_time": 10.206920623779297
    },
    {
      "epoch": 3.222274780273437e-05,
      "model_forward_time": 3.1404693126678467,
      "step": 8447
    },
    {
      "dynamics_loss_avg": 0.04774245265871287,
      "epoch": 3.222274780273437e-05,
      "step": 8447
    },
    {
      "action_loss_avg": 0.015807536244392396,
      "epoch": 3.222274780273437e-05,
      "step": 8447
    },
    {
      "epoch": 3.222274780273437e-05,
      "step": 8447,
      "training_step_time": 9.92112398147583
    },
    {
      "epoch": 3.222274780273437e-05,
      "model_forward_time": 3.1408166885375977,
      "step": 8447
    },
    {
      "epoch": 3.222274780273437e-05,
      "step": 8447,
      "training_step_time": 10.262471199035645
    },
    {
      "epoch": 3.22265625e-05,
      "model_forward_time": 2.4541234970092773,
      "step": 8448
    },
    {
      "epoch": 3.22265625e-05,
      "step": 8448,
      "training_step_time": 7.482634782791138
    },
    {
      "epoch": 3.22265625e-05,
      "model_forward_time": 3.138324737548828,
      "step": 8448
    },
    {
      "epoch": 3.22265625e-05,
      "step": 8448,
      "training_step_time": 9.916358709335327
    },
    {
      "epoch": 3.22265625e-05,
      "model_forward_time": 3.1437301635742188,
      "step": 8448
    },
    {
      "epoch": 3.22265625e-05,
      "step": 8448,
      "training_step_time": 10.186300039291382
    },
    {
      "epoch": 3.22265625e-05,
      "model_forward_time": 3.1442155838012695,
      "step": 8448
    },
    {
      "epoch": 3.22265625e-05,
      "step": 8448,
      "training_step_time": 10.232683181762695
    },
    {
      "epoch": 3.2230377197265624e-05,
      "model_forward_time": 3.151473045349121,
      "step": 8449
    },
    {
      "epoch": 3.2230377197265624e-05,
      "step": 8449,
      "training_step_time": 9.924214363098145
    },
    {
      "epoch": 3.2230377197265624e-05,
      "model_forward_time": 3.136507987976074,
      "step": 8449
    },
    {
      "epoch": 3.2230377197265624e-05,
      "step": 8449,
      "training_step_time": 9.907150506973267
    },
    {
      "epoch": 3.2230377197265624e-05,
      "model_forward_time": 3.146005630493164,
      "step": 8449
    },
    {
      "epoch": 3.2230377197265624e-05,
      "step": 8449,
      "training_step_time": 9.917159080505371
    },
    {
      "epoch": 3.2230377197265624e-05,
      "model_forward_time": 3.469595193862915,
      "step": 8449
    },
    {
      "epoch": 3.2230377197265624e-05,
      "step": 8449,
      "training_step_time": 10.3358314037323
    },
    {
      "epoch": 3.223419189453125e-05,
      "grad_norm": 0.39090749621391296,
      "learning_rate": 1e-05,
      "loss": 0.0822,
      "step": 8450
    },
    {
      "epoch": 3.223419189453125e-05,
      "model_forward_time": 3.158357858657837,
      "step": 8450
    },
    {
      "dynamics_loss_avg": 0.07116393335163593,
      "epoch": 3.223419189453125e-05,
      "step": 8450
    },
    {
      "action_loss_avg": 0.02518674936145544,
      "epoch": 3.223419189453125e-05,
      "step": 8450
    },
    {
      "epoch": 3.223419189453125e-05,
      "step": 8450,
      "training_step_time": 9.933486938476562
    },
    {
      "epoch": 3.223419189453125e-05,
      "model_forward_time": 3.1409616470336914,
      "step": 8450
    },
    {
      "epoch": 3.223419189453125e-05,
      "step": 8450,
      "training_step_time": 9.913325309753418
    },
    {
      "epoch": 3.223419189453125e-05,
      "model_forward_time": 3.1404950618743896,
      "step": 8450
    },
    {
      "epoch": 3.223419189453125e-05,
      "step": 8450,
      "training_step_time": 9.912596702575684
    },
    {
      "epoch": 3.223419189453125e-05,
      "model_forward_time": 3.136695384979248,
      "step": 8450
    },
    {
      "epoch": 3.223419189453125e-05,
      "step": 8450,
      "training_step_time": 10.016530990600586
    },
    {
      "epoch": 3.223800659179688e-05,
      "model_forward_time": 3.150562047958374,
      "step": 8451
    },
    {
      "epoch": 3.223800659179688e-05,
      "step": 8451,
      "training_step_time": 10.209198474884033
    },
    {
      "epoch": 3.223800659179688e-05,
      "model_forward_time": 3.1432745456695557,
      "step": 8451
    },
    {
      "epoch": 3.223800659179688e-05,
      "step": 8451,
      "training_step_time": 9.925736427307129
    },
    {
      "epoch": 3.223800659179688e-05,
      "model_forward_time": 3.144026041030884,
      "step": 8451
    },
    {
      "epoch": 3.223800659179688e-05,
      "step": 8451,
      "training_step_time": 9.920127868652344
    },
    {
      "epoch": 3.223800659179688e-05,
      "model_forward_time": 3.1838254928588867,
      "step": 8451
    },
    {
      "epoch": 3.223800659179688e-05,
      "step": 8451,
      "training_step_time": 10.280729055404663
    },
    {
      "epoch": 3.22418212890625e-05,
      "model_forward_time": 3.1620941162109375,
      "step": 8452
    },
    {
      "epoch": 3.22418212890625e-05,
      "step": 8452,
      "training_step_time": 9.935043334960938
    },
    {
      "epoch": 3.22418212890625e-05,
      "model_forward_time": 3.4137165546417236,
      "step": 8452
    },
    {
      "epoch": 3.22418212890625e-05,
      "step": 8452,
      "training_step_time": 10.184235572814941
    },
    {
      "epoch": 3.22418212890625e-05,
      "model_forward_time": 3.1441824436187744,
      "step": 8452
    },
    {
      "dynamics_loss_avg": 0.05453560594469309,
      "epoch": 3.22418212890625e-05,
      "step": 8452
    },
    {
      "action_loss_avg": 0.021601835591718556,
      "epoch": 3.22418212890625e-05,
      "step": 8452
    },
    {
      "epoch": 3.22418212890625e-05,
      "step": 8452,
      "training_step_time": 9.92084527015686
    },
    {
      "epoch": 3.22418212890625e-05,
      "model_forward_time": 3.14724063873291,
      "step": 8452
    },
    {
      "epoch": 3.22418212890625e-05,
      "step": 8452,
      "training_step_time": 10.328843355178833
    },
    {
      "epoch": 3.224563598632812e-05,
      "model_forward_time": 3.1536123752593994,
      "step": 8453
    },
    {
      "epoch": 3.224563598632812e-05,
      "step": 8453,
      "training_step_time": 9.929563522338867
    },
    {
      "epoch": 3.224563598632812e-05,
      "model_forward_time": 3.147594928741455,
      "step": 8453
    },
    {
      "epoch": 3.224563598632812e-05,
      "step": 8453,
      "training_step_time": 9.926403522491455
    },
    {
      "epoch": 3.224563598632812e-05,
      "model_forward_time": 3.443402051925659,
      "step": 8453
    },
    {
      "epoch": 3.224563598632812e-05,
      "step": 8453,
      "training_step_time": 10.220240354537964
    },
    {
      "epoch": 3.224563598632812e-05,
      "model_forward_time": 3.1456105709075928,
      "step": 8453
    },
    {
      "epoch": 3.224563598632812e-05,
      "step": 8453,
      "training_step_time": 10.033962965011597
    },
    {
      "epoch": 3.224945068359375e-05,
      "model_forward_time": 2.498006582260132,
      "step": 8454
    },
    {
      "epoch": 3.224945068359375e-05,
      "step": 8454,
      "training_step_time": 7.534006595611572
    },
    {
      "epoch": 3.224945068359375e-05,
      "model_forward_time": 3.1450467109680176,
      "step": 8454
    },
    {
      "epoch": 3.224945068359375e-05,
      "step": 8454,
      "training_step_time": 9.917669296264648
    },
    {
      "epoch": 3.224945068359375e-05,
      "model_forward_time": 3.1399879455566406,
      "step": 8454
    },
    {
      "epoch": 3.224945068359375e-05,
      "step": 8454,
      "training_step_time": 9.913260459899902
    },
    {
      "epoch": 3.224945068359375e-05,
      "model_forward_time": 3.41375470161438,
      "step": 8454
    },
    {
      "epoch": 3.224945068359375e-05,
      "step": 8454,
      "training_step_time": 10.2791268825531
    },
    {
      "epoch": 3.2253265380859375e-05,
      "model_forward_time": 3.1582727432250977,
      "step": 8455
    },
    {
      "dynamics_loss_avg": 0.04344389233738184,
      "epoch": 3.2253265380859375e-05,
      "step": 8455
    },
    {
      "action_loss_avg": 0.014531342126429082,
      "epoch": 3.2253265380859375e-05,
      "step": 8455
    },
    {
      "epoch": 3.2253265380859375e-05,
      "step": 8455,
      "training_step_time": 9.936903715133667
    },
    {
      "epoch": 3.2253265380859375e-05,
      "model_forward_time": 3.14269757270813,
      "step": 8455
    },
    {
      "epoch": 3.2253265380859375e-05,
      "step": 8455,
      "training_step_time": 9.911888837814331
    },
    {
      "epoch": 3.2253265380859375e-05,
      "model_forward_time": 3.1419270038604736,
      "step": 8455
    },
    {
      "epoch": 3.2253265380859375e-05,
      "step": 8455,
      "training_step_time": 9.911999464035034
    },
    {
      "epoch": 3.2253265380859375e-05,
      "model_forward_time": 3.138141393661499,
      "step": 8455
    },
    {
      "epoch": 3.2253265380859375e-05,
      "step": 8455,
      "training_step_time": 10.014811038970947
    },
    {
      "epoch": 3.2257080078125e-05,
      "model_forward_time": 3.150705337524414,
      "step": 8456
    },
    {
      "epoch": 3.2257080078125e-05,
      "step": 8456,
      "training_step_time": 10.217880249023438
    },
    {
      "epoch": 3.2257080078125e-05,
      "model_forward_time": 3.139697551727295,
      "step": 8456
    },
    {
      "epoch": 3.2257080078125e-05,
      "step": 8456,
      "training_step_time": 9.914061546325684
    },
    {
      "epoch": 3.2257080078125e-05,
      "model_forward_time": 3.140946626663208,
      "step": 8456
    },
    {
      "epoch": 3.2257080078125e-05,
      "step": 8456,
      "training_step_time": 9.912008285522461
    },
    {
      "epoch": 3.2257080078125e-05,
      "model_forward_time": 3.1493825912475586,
      "step": 8456
    },
    {
      "epoch": 3.2257080078125e-05,
      "step": 8456,
      "training_step_time": 10.314223289489746
    },
    {
      "epoch": 3.226089477539063e-05,
      "model_forward_time": 3.154599905014038,
      "step": 8457
    },
    {
      "epoch": 3.226089477539063e-05,
      "step": 8457,
      "training_step_time": 9.92884349822998
    },
    {
      "epoch": 3.226089477539063e-05,
      "model_forward_time": 3.4321365356445312,
      "step": 8457
    },
    {
      "epoch": 3.226089477539063e-05,
      "step": 8457,
      "training_step_time": 10.200359582901001
    },
    {
      "epoch": 3.226089477539063e-05,
      "model_forward_time": 3.1411631107330322,
      "step": 8457
    },
    {
      "dynamics_loss_avg": 0.05814762506633997,
      "epoch": 3.226089477539063e-05,
      "step": 8457
    },
    {
      "action_loss_avg": 0.022290880884975196,
      "epoch": 3.226089477539063e-05,
      "step": 8457
    },
    {
      "epoch": 3.226089477539063e-05,
      "step": 8457,
      "training_step_time": 9.916637897491455
    },
    {
      "epoch": 3.226089477539063e-05,
      "model_forward_time": 3.1480085849761963,
      "step": 8457
    },
    {
      "epoch": 3.226089477539063e-05,
      "step": 8457,
      "training_step_time": 10.298578262329102
    },
    {
      "epoch": 3.226470947265625e-05,
      "model_forward_time": 3.161330223083496,
      "step": 8458
    },
    {
      "epoch": 3.226470947265625e-05,
      "step": 8458,
      "training_step_time": 9.937962055206299
    },
    {
      "epoch": 3.226470947265625e-05,
      "model_forward_time": 3.1512534618377686,
      "step": 8458
    },
    {
      "epoch": 3.226470947265625e-05,
      "step": 8458,
      "training_step_time": 9.930593252182007
    },
    {
      "epoch": 3.226470947265625e-05,
      "model_forward_time": 3.405932664871216,
      "step": 8458
    },
    {
      "epoch": 3.226470947265625e-05,
      "step": 8458,
      "training_step_time": 10.181066513061523
    },
    {
      "epoch": 3.226470947265625e-05,
      "model_forward_time": 3.1473286151885986,
      "step": 8458
    },
    {
      "epoch": 3.226470947265625e-05,
      "step": 8458,
      "training_step_time": 10.290643215179443
    },
    {
      "epoch": 3.2268524169921874e-05,
      "model_forward_time": 3.1576731204986572,
      "step": 8459
    },
    {
      "epoch": 3.2268524169921874e-05,
      "step": 8459,
      "training_step_time": 9.930020332336426
    },
    {
      "epoch": 3.2268524169921874e-05,
      "model_forward_time": 3.138556957244873,
      "step": 8459
    },
    {
      "epoch": 3.2268524169921874e-05,
      "step": 8459,
      "training_step_time": 9.909358024597168
    },
    {
      "epoch": 3.2268524169921874e-05,
      "model_forward_time": 3.1436712741851807,
      "step": 8459
    },
    {
      "epoch": 3.2268524169921874e-05,
      "step": 8459,
      "training_step_time": 9.91287088394165
    },
    {
      "epoch": 3.2268524169921874e-05,
      "model_forward_time": 3.421785593032837,
      "step": 8459
    },
    {
      "epoch": 3.2268524169921874e-05,
      "step": 8459,
      "training_step_time": 10.30714201927185
    },
    {
      "epoch": 3.22723388671875e-05,
      "grad_norm": 0.35793259739875793,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 8460
    },
    {
      "epoch": 3.22723388671875e-05,
      "model_forward_time": 3.1516687870025635,
      "step": 8460
    },
    {
      "dynamics_loss_avg": 0.06501461528241634,
      "epoch": 3.22723388671875e-05,
      "step": 8460
    },
    {
      "action_loss_avg": 0.02590581951662898,
      "epoch": 3.22723388671875e-05,
      "step": 8460
    },
    {
      "epoch": 3.22723388671875e-05,
      "step": 8460,
      "training_step_time": 9.921707391738892
    },
    {
      "epoch": 3.22723388671875e-05,
      "model_forward_time": 3.1348094940185547,
      "step": 8460
    },
    {
      "epoch": 3.22723388671875e-05,
      "step": 8460,
      "training_step_time": 9.8999605178833
    },
    {
      "epoch": 3.22723388671875e-05,
      "model_forward_time": 3.1381945610046387,
      "step": 8460
    },
    {
      "epoch": 3.22723388671875e-05,
      "step": 8460,
      "training_step_time": 9.901592016220093
    },
    {
      "epoch": 3.22723388671875e-05,
      "model_forward_time": 3.1368987560272217,
      "step": 8460
    },
    {
      "epoch": 3.22723388671875e-05,
      "step": 8460,
      "training_step_time": 10.010424375534058
    },
    {
      "epoch": 3.2276153564453126e-05,
      "model_forward_time": 3.426934242248535,
      "step": 8461
    },
    {
      "epoch": 3.2276153564453126e-05,
      "step": 8461,
      "training_step_time": 10.2023446559906
    },
    {
      "epoch": 3.2276153564453126e-05,
      "model_forward_time": 3.137979507446289,
      "step": 8461
    },
    {
      "epoch": 3.2276153564453126e-05,
      "step": 8461,
      "training_step_time": 9.907638788223267
    },
    {
      "epoch": 3.2276153564453126e-05,
      "model_forward_time": 3.138329267501831,
      "step": 8461
    },
    {
      "epoch": 3.2276153564453126e-05,
      "step": 8461,
      "training_step_time": 9.907499074935913
    },
    {
      "epoch": 3.2276153564453126e-05,
      "model_forward_time": 3.135753870010376,
      "step": 8461
    },
    {
      "epoch": 3.2276153564453126e-05,
      "step": 8461,
      "training_step_time": 10.305672883987427
    },
    {
      "epoch": 3.227996826171875e-05,
      "model_forward_time": 3.143641948699951,
      "step": 8462
    },
    {
      "epoch": 3.227996826171875e-05,
      "step": 8462,
      "training_step_time": 9.908276557922363
    },
    {
      "epoch": 3.227996826171875e-05,
      "model_forward_time": 3.3918914794921875,
      "step": 8462
    },
    {
      "epoch": 3.227996826171875e-05,
      "step": 8462,
      "training_step_time": 10.158125638961792
    },
    {
      "epoch": 3.227996826171875e-05,
      "model_forward_time": 3.142263650894165,
      "step": 8462
    },
    {
      "dynamics_loss_avg": 0.06639867462217808,
      "epoch": 3.227996826171875e-05,
      "step": 8462
    },
    {
      "action_loss_avg": 0.01814785082824528,
      "epoch": 3.227996826171875e-05,
      "step": 8462
    },
    {
      "epoch": 3.227996826171875e-05,
      "step": 8462,
      "training_step_time": 9.916187286376953
    },
    {
      "epoch": 3.227996826171875e-05,
      "model_forward_time": 3.1349568367004395,
      "step": 8462
    },
    {
      "epoch": 3.227996826171875e-05,
      "step": 8462,
      "training_step_time": 10.270950317382812
    },
    {
      "epoch": 3.228378295898437e-05,
      "model_forward_time": 3.1496496200561523,
      "step": 8463
    },
    {
      "epoch": 3.228378295898437e-05,
      "step": 8463,
      "training_step_time": 9.927918434143066
    },
    {
      "epoch": 3.228378295898437e-05,
      "model_forward_time": 3.1415088176727295,
      "step": 8463
    },
    {
      "epoch": 3.228378295898437e-05,
      "step": 8463,
      "training_step_time": 9.913696527481079
    },
    {
      "epoch": 3.228378295898437e-05,
      "model_forward_time": 3.41477632522583,
      "step": 8463
    },
    {
      "epoch": 3.228378295898437e-05,
      "step": 8463,
      "training_step_time": 10.189236164093018
    },
    {
      "epoch": 3.228378295898437e-05,
      "model_forward_time": 3.1388654708862305,
      "step": 8463
    },
    {
      "epoch": 3.228378295898437e-05,
      "step": 8463,
      "training_step_time": 10.349217891693115
    },
    {
      "epoch": 3.228759765625e-05,
      "model_forward_time": 3.1476895809173584,
      "step": 8464
    },
    {
      "epoch": 3.228759765625e-05,
      "step": 8464,
      "training_step_time": 9.920960664749146
    },
    {
      "epoch": 3.228759765625e-05,
      "model_forward_time": 3.1352877616882324,
      "step": 8464
    },
    {
      "epoch": 3.228759765625e-05,
      "step": 8464,
      "training_step_time": 9.902742862701416
    },
    {
      "epoch": 3.228759765625e-05,
      "model_forward_time": 3.1334891319274902,
      "step": 8464
    },
    {
      "epoch": 3.228759765625e-05,
      "step": 8464,
      "training_step_time": 9.899219274520874
    },
    {
      "epoch": 3.228759765625e-05,
      "model_forward_time": 3.3989570140838623,
      "step": 8464
    },
    {
      "epoch": 3.228759765625e-05,
      "step": 8464,
      "training_step_time": 10.266635417938232
    },
    {
      "epoch": 3.2291412353515624e-05,
      "model_forward_time": 3.1538150310516357,
      "step": 8465
    },
    {
      "dynamics_loss_avg": 0.06295780539512634,
      "epoch": 3.2291412353515624e-05,
      "step": 8465
    },
    {
      "action_loss_avg": 0.024126232694834472,
      "epoch": 3.2291412353515624e-05,
      "step": 8465
    },
    {
      "epoch": 3.2291412353515624e-05,
      "step": 8465,
      "training_step_time": 9.922593116760254
    },
    {
      "epoch": 3.2291412353515624e-05,
      "model_forward_time": 3.1343677043914795,
      "step": 8465
    },
    {
      "epoch": 3.2291412353515624e-05,
      "step": 8465,
      "training_step_time": 9.903436422348022
    },
    {
      "epoch": 3.2291412353515624e-05,
      "model_forward_time": 3.1335880756378174,
      "step": 8465
    },
    {
      "epoch": 3.2291412353515624e-05,
      "step": 8465,
      "training_step_time": 9.900911331176758
    },
    {
      "epoch": 3.2291412353515624e-05,
      "model_forward_time": 3.136612892150879,
      "step": 8465
    },
    {
      "epoch": 3.2291412353515624e-05,
      "step": 8465,
      "training_step_time": 10.0141122341156
    },
    {
      "epoch": 3.229522705078125e-05,
      "model_forward_time": 3.4025254249572754,
      "step": 8466
    },
    {
      "epoch": 3.229522705078125e-05,
      "step": 8466,
      "training_step_time": 10.17261028289795
    },
    {
      "epoch": 3.229522705078125e-05,
      "model_forward_time": 3.1375598907470703,
      "step": 8466
    },
    {
      "epoch": 3.229522705078125e-05,
      "step": 8466,
      "training_step_time": 9.908448934555054
    },
    {
      "epoch": 3.229522705078125e-05,
      "model_forward_time": 3.1293630599975586,
      "step": 8466
    },
    {
      "epoch": 3.229522705078125e-05,
      "step": 8466,
      "training_step_time": 9.906940937042236
    },
    {
      "epoch": 3.229522705078125e-05,
      "model_forward_time": 3.1614367961883545,
      "step": 8466
    },
    {
      "epoch": 3.229522705078125e-05,
      "step": 8466,
      "training_step_time": 10.286991834640503
    },
    {
      "epoch": 3.229904174804688e-05,
      "model_forward_time": 3.1507904529571533,
      "step": 8467
    },
    {
      "epoch": 3.229904174804688e-05,
      "step": 8467,
      "training_step_time": 9.924832105636597
    },
    {
      "epoch": 3.229904174804688e-05,
      "model_forward_time": 3.4168593883514404,
      "step": 8467
    },
    {
      "epoch": 3.229904174804688e-05,
      "step": 8467,
      "training_step_time": 10.193536520004272
    },
    {
      "epoch": 3.229904174804688e-05,
      "model_forward_time": 3.133396625518799,
      "step": 8467
    },
    {
      "dynamics_loss_avg": 0.054086392279714346,
      "epoch": 3.229904174804688e-05,
      "step": 8467
    },
    {
      "action_loss_avg": 0.017520672315731643,
      "epoch": 3.229904174804688e-05,
      "step": 8467
    },
    {
      "epoch": 3.229904174804688e-05,
      "step": 8467,
      "training_step_time": 9.913792371749878
    },
    {
      "epoch": 3.229904174804688e-05,
      "model_forward_time": 3.137345552444458,
      "step": 8467
    },
    {
      "epoch": 3.229904174804688e-05,
      "step": 8467,
      "training_step_time": 10.306277513504028
    },
    {
      "epoch": 3.23028564453125e-05,
      "model_forward_time": 3.1473548412323,
      "step": 8468
    },
    {
      "epoch": 3.23028564453125e-05,
      "step": 8468,
      "training_step_time": 9.928754329681396
    },
    {
      "epoch": 3.23028564453125e-05,
      "model_forward_time": 3.1367828845977783,
      "step": 8468
    },
    {
      "epoch": 3.23028564453125e-05,
      "step": 8468,
      "training_step_time": 9.910829782485962
    },
    {
      "epoch": 3.23028564453125e-05,
      "model_forward_time": 3.1393790245056152,
      "step": 8468
    },
    {
      "epoch": 3.23028564453125e-05,
      "step": 8468,
      "training_step_time": 10.228534698486328
    },
    {
      "epoch": 3.23028564453125e-05,
      "model_forward_time": 3.13346791267395,
      "step": 8468
    },
    {
      "epoch": 3.23028564453125e-05,
      "step": 8468,
      "training_step_time": 10.250871658325195
    },
    {
      "epoch": 3.230667114257812e-05,
      "model_forward_time": 3.1915409564971924,
      "step": 8469
    },
    {
      "epoch": 3.230667114257812e-05,
      "step": 8469,
      "training_step_time": 9.965488910675049
    },
    {
      "epoch": 3.230667114257812e-05,
      "model_forward_time": 3.1355316638946533,
      "step": 8469
    },
    {
      "epoch": 3.230667114257812e-05,
      "step": 8469,
      "training_step_time": 9.908334493637085
    },
    {
      "epoch": 3.230667114257812e-05,
      "model_forward_time": 3.1431984901428223,
      "step": 8469
    },
    {
      "epoch": 3.230667114257812e-05,
      "step": 8469,
      "training_step_time": 9.919754028320312
    },
    {
      "epoch": 3.230667114257812e-05,
      "model_forward_time": 3.4427120685577393,
      "step": 8469
    },
    {
      "epoch": 3.230667114257812e-05,
      "step": 8469,
      "training_step_time": 10.3100004196167
    },
    {
      "epoch": 3.231048583984375e-05,
      "grad_norm": 0.37045523524284363,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 8470
    },
    {
      "epoch": 3.231048583984375e-05,
      "model_forward_time": 3.155745506286621,
      "step": 8470
    },
    {
      "dynamics_loss_avg": 0.0418758824467659,
      "epoch": 3.231048583984375e-05,
      "step": 8470
    },
    {
      "action_loss_avg": 0.014130413392558694,
      "epoch": 3.231048583984375e-05,
      "step": 8470
    },
    {
      "epoch": 3.231048583984375e-05,
      "step": 8470,
      "training_step_time": 9.933817625045776
    },
    {
      "epoch": 3.231048583984375e-05,
      "model_forward_time": 3.136121988296509,
      "step": 8470
    },
    {
      "epoch": 3.231048583984375e-05,
      "step": 8470,
      "training_step_time": 9.914174795150757
    },
    {
      "epoch": 3.231048583984375e-05,
      "model_forward_time": 3.1365559101104736,
      "step": 8470
    },
    {
      "epoch": 3.231048583984375e-05,
      "step": 8470,
      "training_step_time": 9.912871599197388
    },
    {
      "epoch": 3.231048583984375e-05,
      "model_forward_time": 3.1431655883789062,
      "step": 8470
    },
    {
      "epoch": 3.231048583984375e-05,
      "step": 8470,
      "training_step_time": 10.027719736099243
    },
    {
      "epoch": 3.2314300537109375e-05,
      "model_forward_time": 3.164677619934082,
      "step": 8471
    },
    {
      "epoch": 3.2314300537109375e-05,
      "step": 8471,
      "training_step_time": 10.26073169708252
    },
    {
      "epoch": 3.2314300537109375e-05,
      "model_forward_time": 3.1452479362487793,
      "step": 8471
    },
    {
      "epoch": 3.2314300537109375e-05,
      "step": 8471,
      "training_step_time": 9.922101020812988
    },
    {
      "epoch": 3.2314300537109375e-05,
      "model_forward_time": 3.143827199935913,
      "step": 8471
    },
    {
      "epoch": 3.2314300537109375e-05,
      "step": 8471,
      "training_step_time": 9.921092510223389
    },
    {
      "epoch": 3.2314300537109375e-05,
      "model_forward_time": 3.1399166584014893,
      "step": 8471
    },
    {
      "epoch": 3.2314300537109375e-05,
      "step": 8471,
      "training_step_time": 10.014846086502075
    },
    {
      "epoch": 3.2318115234375e-05,
      "model_forward_time": 3.1462178230285645,
      "step": 8472
    },
    {
      "epoch": 3.2318115234375e-05,
      "step": 8472,
      "training_step_time": 9.922365188598633
    },
    {
      "epoch": 3.2318115234375e-05,
      "model_forward_time": 3.411900281906128,
      "step": 8472
    },
    {
      "epoch": 3.2318115234375e-05,
      "step": 8472,
      "training_step_time": 10.18790578842163
    },
    {
      "epoch": 3.2318115234375e-05,
      "model_forward_time": 3.140460968017578,
      "step": 8472
    },
    {
      "dynamics_loss_avg": 0.05782559402287006,
      "epoch": 3.2318115234375e-05,
      "step": 8472
    },
    {
      "action_loss_avg": 0.020002685859799384,
      "epoch": 3.2318115234375e-05,
      "step": 8472
    },
    {
      "epoch": 3.2318115234375e-05,
      "step": 8472,
      "training_step_time": 9.918036222457886
    },
    {
      "epoch": 3.2318115234375e-05,
      "model_forward_time": 3.1426842212677,
      "step": 8472
    },
    {
      "epoch": 3.2318115234375e-05,
      "step": 8472,
      "training_step_time": 10.28225064277649
    },
    {
      "epoch": 3.232192993164063e-05,
      "model_forward_time": 3.1573429107666016,
      "step": 8473
    },
    {
      "epoch": 3.232192993164063e-05,
      "step": 8473,
      "training_step_time": 9.93306589126587
    },
    {
      "epoch": 3.232192993164063e-05,
      "model_forward_time": 3.1435701847076416,
      "step": 8473
    },
    {
      "epoch": 3.232192993164063e-05,
      "step": 8473,
      "training_step_time": 9.924101114273071
    },
    {
      "epoch": 3.232192993164063e-05,
      "model_forward_time": 3.4306106567382812,
      "step": 8473
    },
    {
      "epoch": 3.232192993164063e-05,
      "step": 8473,
      "training_step_time": 10.206978559494019
    },
    {
      "epoch": 3.232192993164063e-05,
      "model_forward_time": 3.1385180950164795,
      "step": 8473
    },
    {
      "epoch": 3.232192993164063e-05,
      "step": 8473,
      "training_step_time": 10.014148473739624
    },
    {
      "epoch": 3.232574462890625e-05,
      "model_forward_time": 3.144024133682251,
      "step": 8474
    },
    {
      "epoch": 3.232574462890625e-05,
      "step": 8474,
      "training_step_time": 9.920928716659546
    },
    {
      "epoch": 3.232574462890625e-05,
      "model_forward_time": 3.13871431350708,
      "step": 8474
    },
    {
      "epoch": 3.232574462890625e-05,
      "step": 8474,
      "training_step_time": 9.90663194656372
    },
    {
      "epoch": 3.232574462890625e-05,
      "model_forward_time": 3.1411499977111816,
      "step": 8474
    },
    {
      "epoch": 3.232574462890625e-05,
      "step": 8474,
      "training_step_time": 9.912298440933228
    },
    {
      "epoch": 3.232574462890625e-05,
      "model_forward_time": 3.441000461578369,
      "step": 8474
    },
    {
      "epoch": 3.232574462890625e-05,
      "step": 8474,
      "training_step_time": 10.316009759902954
    },
    {
      "epoch": 3.2329559326171874e-05,
      "model_forward_time": 3.1506009101867676,
      "step": 8475
    },
    {
      "dynamics_loss_avg": 0.06208160407841205,
      "epoch": 3.2329559326171874e-05,
      "step": 8475
    },
    {
      "action_loss_avg": 0.025335893500596285,
      "epoch": 3.2329559326171874e-05,
      "step": 8475
    },
    {
      "epoch": 3.2329559326171874e-05,
      "step": 8475,
      "training_step_time": 9.933731079101562
    },
    {
      "epoch": 3.2329559326171874e-05,
      "model_forward_time": 3.1499271392822266,
      "step": 8475
    },
    {
      "epoch": 3.2329559326171874e-05,
      "step": 8475,
      "training_step_time": 9.927549123764038
    },
    {
      "epoch": 3.2329559326171874e-05,
      "model_forward_time": 3.140138626098633,
      "step": 8475
    },
    {
      "epoch": 3.2329559326171874e-05,
      "step": 8475,
      "training_step_time": 9.911625385284424
    },
    {
      "epoch": 3.2329559326171874e-05,
      "model_forward_time": 3.1439974308013916,
      "step": 8475
    },
    {
      "epoch": 3.2329559326171874e-05,
      "step": 8475,
      "training_step_time": 10.278886318206787
    },
    {
      "epoch": 3.23333740234375e-05,
      "model_forward_time": 3.4226126670837402,
      "step": 8476
    },
    {
      "epoch": 3.23333740234375e-05,
      "step": 8476,
      "training_step_time": 10.194056749343872
    },
    {
      "epoch": 3.23333740234375e-05,
      "model_forward_time": 3.1395158767700195,
      "step": 8476
    },
    {
      "epoch": 3.23333740234375e-05,
      "step": 8476,
      "training_step_time": 9.909404277801514
    },
    {
      "epoch": 3.23333740234375e-05,
      "model_forward_time": 3.140162706375122,
      "step": 8476
    },
    {
      "epoch": 3.23333740234375e-05,
      "step": 8476,
      "training_step_time": 9.913731336593628
    },
    {
      "epoch": 3.23333740234375e-05,
      "model_forward_time": 3.1392509937286377,
      "step": 8476
    },
    {
      "epoch": 3.23333740234375e-05,
      "step": 8476,
      "training_step_time": 10.02098536491394
    },
    {
      "epoch": 3.2337188720703126e-05,
      "model_forward_time": 3.1595826148986816,
      "step": 8477
    },
    {
      "epoch": 3.2337188720703126e-05,
      "step": 8477,
      "training_step_time": 9.933856964111328
    },
    {
      "epoch": 3.2337188720703126e-05,
      "model_forward_time": 3.1421170234680176,
      "step": 8477
    },
    {
      "epoch": 3.2337188720703126e-05,
      "step": 8477,
      "training_step_time": 10.196410417556763
    },
    {
      "epoch": 3.2337188720703126e-05,
      "model_forward_time": 3.1414248943328857,
      "step": 8477
    },
    {
      "dynamics_loss_avg": 0.06105822455137968,
      "epoch": 3.2337188720703126e-05,
      "step": 8477
    },
    {
      "action_loss_avg": 0.0215242525562644,
      "epoch": 3.2337188720703126e-05,
      "step": 8477
    },
    {
      "epoch": 3.2337188720703126e-05,
      "step": 8477,
      "training_step_time": 9.916695833206177
    },
    {
      "epoch": 3.2337188720703126e-05,
      "model_forward_time": 3.142826557159424,
      "step": 8477
    },
    {
      "epoch": 3.2337188720703126e-05,
      "step": 8477,
      "training_step_time": 10.332192420959473
    },
    {
      "epoch": 3.234100341796875e-05,
      "model_forward_time": 3.1601932048797607,
      "step": 8478
    },
    {
      "epoch": 3.234100341796875e-05,
      "step": 8478,
      "training_step_time": 9.945966958999634
    },
    {
      "epoch": 3.234100341796875e-05,
      "model_forward_time": 3.1372432708740234,
      "step": 8478
    },
    {
      "epoch": 3.234100341796875e-05,
      "step": 8478,
      "training_step_time": 9.90776252746582
    },
    {
      "epoch": 3.234100341796875e-05,
      "model_forward_time": 3.4060370922088623,
      "step": 8478
    },
    {
      "epoch": 3.234100341796875e-05,
      "step": 8478,
      "training_step_time": 10.171546697616577
    },
    {
      "epoch": 3.234100341796875e-05,
      "model_forward_time": 3.1460466384887695,
      "step": 8478
    },
    {
      "epoch": 3.234100341796875e-05,
      "step": 8478,
      "training_step_time": 10.019020795822144
    },
    {
      "epoch": 3.234481811523437e-05,
      "model_forward_time": 3.149606466293335,
      "step": 8479
    },
    {
      "epoch": 3.234481811523437e-05,
      "step": 8479,
      "training_step_time": 9.919316291809082
    },
    {
      "epoch": 3.234481811523437e-05,
      "model_forward_time": 3.135788679122925,
      "step": 8479
    },
    {
      "epoch": 3.234481811523437e-05,
      "step": 8479,
      "training_step_time": 9.907297849655151
    },
    {
      "epoch": 3.234481811523437e-05,
      "model_forward_time": 3.139749526977539,
      "step": 8479
    },
    {
      "epoch": 3.234481811523437e-05,
      "step": 8479,
      "training_step_time": 9.90701174736023
    },
    {
      "epoch": 3.234481811523437e-05,
      "model_forward_time": 3.1359121799468994,
      "step": 8479
    },
    {
      "epoch": 3.234481811523437e-05,
      "step": 8479,
      "training_step_time": 10.303231954574585
    },
    {
      "epoch": 3.23486328125e-05,
      "grad_norm": 0.2713688313961029,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 8480
    },
    {
      "epoch": 3.23486328125e-05,
      "model_forward_time": 3.152642011642456,
      "step": 8480
    },
    {
      "dynamics_loss_avg": 0.06737993471324444,
      "epoch": 3.23486328125e-05,
      "step": 8480
    },
    {
      "action_loss_avg": 0.026503045577555895,
      "epoch": 3.23486328125e-05,
      "step": 8480
    },
    {
      "epoch": 3.23486328125e-05,
      "step": 8480,
      "training_step_time": 9.923943042755127
    },
    {
      "epoch": 3.23486328125e-05,
      "model_forward_time": 3.132512092590332,
      "step": 8480
    },
    {
      "epoch": 3.23486328125e-05,
      "step": 8480,
      "training_step_time": 9.89889669418335
    },
    {
      "epoch": 3.23486328125e-05,
      "model_forward_time": 3.1407036781311035,
      "step": 8480
    },
    {
      "epoch": 3.23486328125e-05,
      "step": 8480,
      "training_step_time": 9.912266969680786
    },
    {
      "epoch": 3.23486328125e-05,
      "model_forward_time": 3.145418882369995,
      "step": 8480
    },
    {
      "epoch": 3.23486328125e-05,
      "step": 8480,
      "training_step_time": 10.291531085968018
    },
    {
      "epoch": 3.2352447509765625e-05,
      "model_forward_time": 3.436030149459839,
      "step": 8481
    },
    {
      "epoch": 3.2352447509765625e-05,
      "step": 8481,
      "training_step_time": 10.213730335235596
    },
    {
      "epoch": 3.2352447509765625e-05,
      "model_forward_time": 3.1369450092315674,
      "step": 8481
    },
    {
      "epoch": 3.2352447509765625e-05,
      "step": 8481,
      "training_step_time": 9.915519952774048
    },
    {
      "epoch": 3.2352447509765625e-05,
      "model_forward_time": 3.1415393352508545,
      "step": 8481
    },
    {
      "epoch": 3.2352447509765625e-05,
      "step": 8481,
      "training_step_time": 9.918196439743042
    },
    {
      "epoch": 3.2352447509765625e-05,
      "model_forward_time": 3.139993906021118,
      "step": 8481
    },
    {
      "epoch": 3.2352447509765625e-05,
      "step": 8481,
      "training_step_time": 10.262720584869385
    },
    {
      "epoch": 3.235626220703125e-05,
      "model_forward_time": 2.4499549865722656,
      "step": 8482
    },
    {
      "epoch": 3.235626220703125e-05,
      "step": 8482,
      "training_step_time": 7.4872870445251465
    },
    {
      "epoch": 3.235626220703125e-05,
      "model_forward_time": 3.143843650817871,
      "step": 8482
    },
    {
      "epoch": 3.235626220703125e-05,
      "step": 8482,
      "training_step_time": 9.922673463821411
    },
    {
      "epoch": 3.235626220703125e-05,
      "model_forward_time": 3.4072329998016357,
      "step": 8482
    },
    {
      "dynamics_loss_avg": 0.06230633929371834,
      "epoch": 3.235626220703125e-05,
      "step": 8482
    },
    {
      "action_loss_avg": 0.023707742057740688,
      "epoch": 3.235626220703125e-05,
      "step": 8482
    },
    {
      "epoch": 3.235626220703125e-05,
      "step": 8482,
      "training_step_time": 10.192890882492065
    },
    {
      "epoch": 3.235626220703125e-05,
      "model_forward_time": 3.1399617195129395,
      "step": 8482
    },
    {
      "epoch": 3.235626220703125e-05,
      "step": 8482,
      "training_step_time": 10.280692100524902
    },
    {
      "epoch": 3.236007690429688e-05,
      "model_forward_time": 3.1567916870117188,
      "step": 8483
    },
    {
      "epoch": 3.236007690429688e-05,
      "step": 8483,
      "training_step_time": 9.94355583190918
    },
    {
      "epoch": 3.236007690429688e-05,
      "model_forward_time": 3.1422839164733887,
      "step": 8483
    },
    {
      "epoch": 3.236007690429688e-05,
      "step": 8483,
      "training_step_time": 9.924556732177734
    },
    {
      "epoch": 3.236007690429688e-05,
      "model_forward_time": 3.4260001182556152,
      "step": 8483
    },
    {
      "epoch": 3.236007690429688e-05,
      "step": 8483,
      "training_step_time": 10.212686538696289
    },
    {
      "epoch": 3.236007690429688e-05,
      "model_forward_time": 3.1454989910125732,
      "step": 8483
    },
    {
      "epoch": 3.236007690429688e-05,
      "step": 8483,
      "training_step_time": 10.281172037124634
    },
    {
      "epoch": 3.23638916015625e-05,
      "model_forward_time": 3.153580665588379,
      "step": 8484
    },
    {
      "epoch": 3.23638916015625e-05,
      "step": 8484,
      "training_step_time": 9.941778659820557
    },
    {
      "epoch": 3.23638916015625e-05,
      "model_forward_time": 3.1402671337127686,
      "step": 8484
    },
    {
      "epoch": 3.23638916015625e-05,
      "step": 8484,
      "training_step_time": 9.923747062683105
    },
    {
      "epoch": 3.23638916015625e-05,
      "model_forward_time": 3.1391847133636475,
      "step": 8484
    },
    {
      "epoch": 3.23638916015625e-05,
      "step": 8484,
      "training_step_time": 9.921374320983887
    },
    {
      "epoch": 3.23638916015625e-05,
      "model_forward_time": 3.144915819168091,
      "step": 8484
    },
    {
      "epoch": 3.23638916015625e-05,
      "step": 8484,
      "training_step_time": 10.310729742050171
    },
    {
      "epoch": 3.236770629882812e-05,
      "model_forward_time": 3.439059019088745,
      "step": 8485
    },
    {
      "dynamics_loss_avg": 0.062351055815815924,
      "epoch": 3.236770629882812e-05,
      "step": 8485
    },
    {
      "action_loss_avg": 0.021274163341149687,
      "epoch": 3.236770629882812e-05,
      "step": 8485
    },
    {
      "epoch": 3.236770629882812e-05,
      "step": 8485,
      "training_step_time": 10.220349788665771
    },
    {
      "epoch": 3.236770629882812e-05,
      "model_forward_time": 3.1336669921875,
      "step": 8485
    },
    {
      "epoch": 3.236770629882812e-05,
      "step": 8485,
      "training_step_time": 9.909533500671387
    },
    {
      "epoch": 3.236770629882812e-05,
      "model_forward_time": 3.131639242172241,
      "step": 8485
    },
    {
      "epoch": 3.236770629882812e-05,
      "step": 8485,
      "training_step_time": 9.908874034881592
    },
    {
      "epoch": 3.236770629882812e-05,
      "model_forward_time": 3.13924503326416,
      "step": 8485
    },
    {
      "epoch": 3.236770629882812e-05,
      "step": 8485,
      "training_step_time": 10.290585994720459
    },
    {
      "epoch": 3.237152099609375e-05,
      "model_forward_time": 3.4620096683502197,
      "step": 8486
    },
    {
      "epoch": 3.237152099609375e-05,
      "step": 8486,
      "training_step_time": 10.24167513847351
    },
    {
      "epoch": 3.237152099609375e-05,
      "model_forward_time": 3.135769844055176,
      "step": 8486
    },
    {
      "epoch": 3.237152099609375e-05,
      "step": 8486,
      "training_step_time": 9.919476747512817
    },
    {
      "epoch": 3.237152099609375e-05,
      "model_forward_time": 3.1331679821014404,
      "step": 8486
    },
    {
      "epoch": 3.237152099609375e-05,
      "step": 8486,
      "training_step_time": 9.909191608428955
    },
    {
      "epoch": 3.237152099609375e-05,
      "model_forward_time": 3.1451539993286133,
      "step": 8486
    },
    {
      "epoch": 3.237152099609375e-05,
      "step": 8486,
      "training_step_time": 10.03933596611023
    },
    {
      "epoch": 3.2375335693359375e-05,
      "model_forward_time": 3.1498990058898926,
      "step": 8487
    },
    {
      "epoch": 3.2375335693359375e-05,
      "step": 8487,
      "training_step_time": 9.928883790969849
    },
    {
      "epoch": 3.2375335693359375e-05,
      "model_forward_time": 3.134284257888794,
      "step": 8487
    },
    {
      "epoch": 3.2375335693359375e-05,
      "step": 8487,
      "training_step_time": 10.185423612594604
    },
    {
      "epoch": 3.2375335693359375e-05,
      "model_forward_time": 3.135728120803833,
      "step": 8487
    },
    {
      "dynamics_loss_avg": 0.06023130044341087,
      "epoch": 3.2375335693359375e-05,
      "step": 8487
    },
    {
      "action_loss_avg": 0.021682071033865213,
      "epoch": 3.2375335693359375e-05,
      "step": 8487
    },
    {
      "epoch": 3.2375335693359375e-05,
      "step": 8487,
      "training_step_time": 9.908702850341797
    },
    {
      "epoch": 3.2375335693359375e-05,
      "model_forward_time": 3.1386783123016357,
      "step": 8487
    },
    {
      "epoch": 3.2375335693359375e-05,
      "step": 8487,
      "training_step_time": 10.304692268371582
    },
    {
      "epoch": 3.2379150390625e-05,
      "model_forward_time": 3.1540563106536865,
      "step": 8488
    },
    {
      "epoch": 3.2379150390625e-05,
      "step": 8488,
      "training_step_time": 9.926875829696655
    },
    {
      "epoch": 3.2379150390625e-05,
      "model_forward_time": 3.1387178897857666,
      "step": 8488
    },
    {
      "epoch": 3.2379150390625e-05,
      "step": 8488,
      "training_step_time": 9.906997203826904
    },
    {
      "epoch": 3.2379150390625e-05,
      "model_forward_time": 3.1386094093322754,
      "step": 8488
    },
    {
      "epoch": 3.2379150390625e-05,
      "step": 8488,
      "training_step_time": 10.186151266098022
    },
    {
      "epoch": 3.2379150390625e-05,
      "model_forward_time": 3.1402032375335693,
      "step": 8488
    },
    {
      "epoch": 3.2379150390625e-05,
      "step": 8488,
      "training_step_time": 10.317001819610596
    },
    {
      "epoch": 3.238296508789063e-05,
      "model_forward_time": 3.1502315998077393,
      "step": 8489
    },
    {
      "epoch": 3.238296508789063e-05,
      "step": 8489,
      "training_step_time": 9.921934604644775
    },
    {
      "epoch": 3.238296508789063e-05,
      "model_forward_time": 3.1379215717315674,
      "step": 8489
    },
    {
      "epoch": 3.238296508789063e-05,
      "step": 8489,
      "training_step_time": 9.906774759292603
    },
    {
      "epoch": 3.238296508789063e-05,
      "model_forward_time": 3.145639657974243,
      "step": 8489
    },
    {
      "epoch": 3.238296508789063e-05,
      "step": 8489,
      "training_step_time": 9.913579225540161
    },
    {
      "epoch": 3.238296508789063e-05,
      "model_forward_time": 3.4247283935546875,
      "step": 8489
    },
    {
      "epoch": 3.238296508789063e-05,
      "step": 8489,
      "training_step_time": 10.288094520568848
    },
    {
      "epoch": 3.238677978515625e-05,
      "grad_norm": 0.35403329133987427,
      "learning_rate": 1e-05,
      "loss": 0.0762,
      "step": 8490
    },
    {
      "epoch": 3.238677978515625e-05,
      "model_forward_time": 3.162759304046631,
      "step": 8490
    },
    {
      "dynamics_loss_avg": 0.05921016372740269,
      "epoch": 3.238677978515625e-05,
      "step": 8490
    },
    {
      "action_loss_avg": 0.018805702961981297,
      "epoch": 3.238677978515625e-05,
      "step": 8490
    },
    {
      "epoch": 3.238677978515625e-05,
      "step": 8490,
      "training_step_time": 9.936506509780884
    },
    {
      "epoch": 3.238677978515625e-05,
      "model_forward_time": 3.1430768966674805,
      "step": 8490
    },
    {
      "epoch": 3.238677978515625e-05,
      "step": 8490,
      "training_step_time": 9.917226791381836
    },
    {
      "epoch": 3.238677978515625e-05,
      "model_forward_time": 3.149470329284668,
      "step": 8490
    },
    {
      "epoch": 3.238677978515625e-05,
      "step": 8490,
      "training_step_time": 9.9313383102417
    },
    {
      "epoch": 3.238677978515625e-05,
      "model_forward_time": 3.150144338607788,
      "step": 8490
    },
    {
      "epoch": 3.238677978515625e-05,
      "step": 8490,
      "training_step_time": 10.310036897659302
    },
    {
      "epoch": 3.2390594482421874e-05,
      "model_forward_time": 3.1516523361206055,
      "step": 8491
    },
    {
      "epoch": 3.2390594482421874e-05,
      "step": 8491,
      "training_step_time": 10.203744411468506
    },
    {
      "epoch": 3.2390594482421874e-05,
      "model_forward_time": 3.1413495540618896,
      "step": 8491
    },
    {
      "epoch": 3.2390594482421874e-05,
      "step": 8491,
      "training_step_time": 9.91857647895813
    },
    {
      "epoch": 3.2390594482421874e-05,
      "model_forward_time": 3.146743059158325,
      "step": 8491
    },
    {
      "epoch": 3.2390594482421874e-05,
      "step": 8491,
      "training_step_time": 9.927866697311401
    },
    {
      "epoch": 3.2390594482421874e-05,
      "model_forward_time": 3.1392176151275635,
      "step": 8491
    },
    {
      "epoch": 3.2390594482421874e-05,
      "step": 8491,
      "training_step_time": 10.265052080154419
    },
    {
      "epoch": 3.23944091796875e-05,
      "model_forward_time": 3.1567306518554688,
      "step": 8492
    },
    {
      "epoch": 3.23944091796875e-05,
      "step": 8492,
      "training_step_time": 9.931878089904785
    },
    {
      "epoch": 3.23944091796875e-05,
      "model_forward_time": 3.3967058658599854,
      "step": 8492
    },
    {
      "epoch": 3.23944091796875e-05,
      "step": 8492,
      "training_step_time": 10.176111221313477
    },
    {
      "epoch": 3.23944091796875e-05,
      "model_forward_time": 3.139155626296997,
      "step": 8492
    },
    {
      "dynamics_loss_avg": 0.05033665429800749,
      "epoch": 3.23944091796875e-05,
      "step": 8492
    },
    {
      "action_loss_avg": 0.018445313163101672,
      "epoch": 3.23944091796875e-05,
      "step": 8492
    },
    {
      "epoch": 3.23944091796875e-05,
      "step": 8492,
      "training_step_time": 9.913580417633057
    },
    {
      "epoch": 3.23944091796875e-05,
      "model_forward_time": 3.139411687850952,
      "step": 8492
    },
    {
      "epoch": 3.23944091796875e-05,
      "step": 8492,
      "training_step_time": 10.309826135635376
    },
    {
      "epoch": 3.2398223876953126e-05,
      "model_forward_time": 3.154940605163574,
      "step": 8493
    },
    {
      "epoch": 3.2398223876953126e-05,
      "step": 8493,
      "training_step_time": 9.931477546691895
    },
    {
      "epoch": 3.2398223876953126e-05,
      "model_forward_time": 3.145780563354492,
      "step": 8493
    },
    {
      "epoch": 3.2398223876953126e-05,
      "step": 8493,
      "training_step_time": 9.924670457839966
    },
    {
      "epoch": 3.2398223876953126e-05,
      "model_forward_time": 3.428624153137207,
      "step": 8493
    },
    {
      "epoch": 3.2398223876953126e-05,
      "step": 8493,
      "training_step_time": 10.205000877380371
    },
    {
      "epoch": 3.2398223876953126e-05,
      "model_forward_time": 3.144097328186035,
      "step": 8493
    },
    {
      "epoch": 3.2398223876953126e-05,
      "step": 8493,
      "training_step_time": 10.297152996063232
    },
    {
      "epoch": 3.240203857421875e-05,
      "model_forward_time": 3.1515135765075684,
      "step": 8494
    },
    {
      "epoch": 3.240203857421875e-05,
      "step": 8494,
      "training_step_time": 9.926672458648682
    },
    {
      "epoch": 3.240203857421875e-05,
      "model_forward_time": 3.1406192779541016,
      "step": 8494
    },
    {
      "epoch": 3.240203857421875e-05,
      "step": 8494,
      "training_step_time": 9.91543960571289
    },
    {
      "epoch": 3.240203857421875e-05,
      "model_forward_time": 3.1411044597625732,
      "step": 8494
    },
    {
      "epoch": 3.240203857421875e-05,
      "step": 8494,
      "training_step_time": 9.920779943466187
    },
    {
      "epoch": 3.240203857421875e-05,
      "model_forward_time": 3.406548261642456,
      "step": 8494
    },
    {
      "epoch": 3.240203857421875e-05,
      "step": 8494,
      "training_step_time": 10.275340795516968
    },
    {
      "epoch": 3.240585327148437e-05,
      "model_forward_time": 3.1519482135772705,
      "step": 8495
    },
    {
      "dynamics_loss_avg": 0.04748817123472691,
      "epoch": 3.240585327148437e-05,
      "step": 8495
    },
    {
      "action_loss_avg": 0.014532695803791285,
      "epoch": 3.240585327148437e-05,
      "step": 8495
    },
    {
      "epoch": 3.240585327148437e-05,
      "step": 8495,
      "training_step_time": 9.93669080734253
    },
    {
      "epoch": 3.240585327148437e-05,
      "model_forward_time": 3.1409847736358643,
      "step": 8495
    },
    {
      "epoch": 3.240585327148437e-05,
      "step": 8495,
      "training_step_time": 9.916444778442383
    },
    {
      "epoch": 3.240585327148437e-05,
      "model_forward_time": 3.135338068008423,
      "step": 8495
    },
    {
      "epoch": 3.240585327148437e-05,
      "step": 8495,
      "training_step_time": 9.90359902381897
    },
    {
      "epoch": 3.240585327148437e-05,
      "model_forward_time": 3.1431546211242676,
      "step": 8495
    },
    {
      "epoch": 3.240585327148437e-05,
      "step": 8495,
      "training_step_time": 10.303858995437622
    },
    {
      "epoch": 3.240966796875e-05,
      "model_forward_time": 3.4149045944213867,
      "step": 8496
    },
    {
      "epoch": 3.240966796875e-05,
      "step": 8496,
      "training_step_time": 10.190138578414917
    },
    {
      "epoch": 3.240966796875e-05,
      "model_forward_time": 3.137209177017212,
      "step": 8496
    },
    {
      "epoch": 3.240966796875e-05,
      "step": 8496,
      "training_step_time": 9.898684740066528
    },
    {
      "epoch": 3.240966796875e-05,
      "model_forward_time": 3.131375789642334,
      "step": 8496
    },
    {
      "epoch": 3.240966796875e-05,
      "step": 8496,
      "training_step_time": 9.893988609313965
    },
    {
      "epoch": 3.240966796875e-05,
      "model_forward_time": 3.1393630504608154,
      "step": 8496
    },
    {
      "epoch": 3.240966796875e-05,
      "step": 8496,
      "training_step_time": 10.266542911529541
    },
    {
      "epoch": 3.2413482666015625e-05,
      "model_forward_time": 3.1523871421813965,
      "step": 8497
    },
    {
      "epoch": 3.2413482666015625e-05,
      "step": 8497,
      "training_step_time": 9.916857242584229
    },
    {
      "epoch": 3.2413482666015625e-05,
      "model_forward_time": 3.1323118209838867,
      "step": 8497
    },
    {
      "epoch": 3.2413482666015625e-05,
      "step": 8497,
      "training_step_time": 10.210506677627563
    },
    {
      "epoch": 3.2413482666015625e-05,
      "model_forward_time": 3.137446880340576,
      "step": 8497
    },
    {
      "dynamics_loss_avg": 0.06929453909397125,
      "epoch": 3.2413482666015625e-05,
      "step": 8497
    },
    {
      "action_loss_avg": 0.029602790530771018,
      "epoch": 3.2413482666015625e-05,
      "step": 8497
    },
    {
      "epoch": 3.2413482666015625e-05,
      "step": 8497,
      "training_step_time": 9.912227630615234
    },
    {
      "epoch": 3.2413482666015625e-05,
      "model_forward_time": 3.1411678791046143,
      "step": 8497
    },
    {
      "epoch": 3.2413482666015625e-05,
      "step": 8497,
      "training_step_time": 10.286177396774292
    },
    {
      "epoch": 3.241729736328125e-05,
      "model_forward_time": 3.161144971847534,
      "step": 8498
    },
    {
      "epoch": 3.241729736328125e-05,
      "step": 8498,
      "training_step_time": 9.933789730072021
    },
    {
      "epoch": 3.241729736328125e-05,
      "model_forward_time": 3.1378164291381836,
      "step": 8498
    },
    {
      "epoch": 3.241729736328125e-05,
      "step": 8498,
      "training_step_time": 9.908888101577759
    },
    {
      "epoch": 3.241729736328125e-05,
      "model_forward_time": 3.4099948406219482,
      "step": 8498
    },
    {
      "epoch": 3.241729736328125e-05,
      "step": 8498,
      "training_step_time": 10.176113843917847
    },
    {
      "epoch": 3.241729736328125e-05,
      "model_forward_time": 3.139404058456421,
      "step": 8498
    },
    {
      "epoch": 3.241729736328125e-05,
      "step": 8498,
      "training_step_time": 10.32595419883728
    },
    {
      "epoch": 3.242111206054688e-05,
      "model_forward_time": 3.1901330947875977,
      "step": 8499
    },
    {
      "epoch": 3.242111206054688e-05,
      "step": 8499,
      "training_step_time": 9.96168065071106
    },
    {
      "epoch": 3.242111206054688e-05,
      "model_forward_time": 3.13086199760437,
      "step": 8499
    },
    {
      "epoch": 3.242111206054688e-05,
      "step": 8499,
      "training_step_time": 9.89783787727356
    },
    {
      "epoch": 3.242111206054688e-05,
      "model_forward_time": 3.1387617588043213,
      "step": 8499
    },
    {
      "epoch": 3.242111206054688e-05,
      "step": 8499,
      "training_step_time": 9.906755208969116
    },
    {
      "epoch": 3.242111206054688e-05,
      "model_forward_time": 3.135767698287964,
      "step": 8499
    },
    {
      "epoch": 3.242111206054688e-05,
      "step": 8499,
      "training_step_time": 10.275231122970581
    },
    {
      "epoch": 3.24249267578125e-05,
      "grad_norm": 0.5530783534049988,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 8500
    },
    {
      "epoch": 3.24249267578125e-05,
      "model_forward_time": 2.466244697570801,
      "step": 8500
    },
    {
      "dynamics_loss_avg": 0.06454244591295719,
      "epoch": 3.24249267578125e-05,
      "step": 8500
    },
    {
      "action_loss_avg": 0.022465296369045974,
      "epoch": 3.24249267578125e-05,
      "step": 8500
    },
    {
      "epoch": 3.24249267578125e-05,
      "step": 8500,
      "training_step_time": 7.4800896644592285
    },
    {
      "epoch": 3.24249267578125e-05,
      "model_forward_time": 2.4198946952819824,
      "step": 8500
    },
    {
      "epoch": 3.24249267578125e-05,
      "step": 8500,
      "training_step_time": 7.430458307266235
    },
    {
      "epoch": 3.24249267578125e-05,
      "model_forward_time": 2.4239046573638916,
      "step": 8500
    },
    {
      "epoch": 3.24249267578125e-05,
      "step": 8500,
      "training_step_time": 7.431772708892822
    },
    {
      "epoch": 3.24249267578125e-05,
      "model_forward_time": 2.4268712997436523,
      "step": 8500
    },
    {
      "epoch": 3.24249267578125e-05,
      "step": 8500,
      "training_step_time": 10.296321868896484
    },
    {
      "epoch": 3.242874145507812e-05,
      "model_forward_time": 5.24606728553772,
      "step": 8501
    },
    {
      "epoch": 3.242874145507812e-05,
      "step": 8501,
      "training_step_time": 12.01433539390564
    },
    {
      "epoch": 3.242874145507812e-05,
      "model_forward_time": 3.1542723178863525,
      "step": 8501
    },
    {
      "epoch": 3.242874145507812e-05,
      "step": 8501,
      "training_step_time": 9.928927421569824
    },
    {
      "epoch": 3.242874145507812e-05,
      "model_forward_time": 2.4290406703948975,
      "step": 8501
    },
    {
      "epoch": 3.242874145507812e-05,
      "step": 8501,
      "training_step_time": 7.451992988586426
    },
    {
      "epoch": 3.242874145507812e-05,
      "model_forward_time": 2.4333555698394775,
      "step": 8501
    },
    {
      "epoch": 3.242874145507812e-05,
      "step": 8501,
      "training_step_time": 10.337612390518188
    },
    {
      "epoch": 3.243255615234375e-05,
      "model_forward_time": 3.1525728702545166,
      "step": 8502
    },
    {
      "epoch": 3.243255615234375e-05,
      "step": 8502,
      "training_step_time": 9.917550086975098
    },
    {
      "epoch": 3.243255615234375e-05,
      "model_forward_time": 3.13273286819458,
      "step": 8502
    },
    {
      "epoch": 3.243255615234375e-05,
      "step": 8502,
      "training_step_time": 10.175455331802368
    },
    {
      "epoch": 3.243255615234375e-05,
      "model_forward_time": 3.137401819229126,
      "step": 8502
    },
    {
      "dynamics_loss_avg": 0.04438473079353571,
      "epoch": 3.243255615234375e-05,
      "step": 8502
    },
    {
      "action_loss_avg": 0.017444993928074838,
      "epoch": 3.243255615234375e-05,
      "step": 8502
    },
    {
      "epoch": 3.243255615234375e-05,
      "step": 8502,
      "training_step_time": 9.907981395721436
    },
    {
      "epoch": 3.243255615234375e-05,
      "model_forward_time": 3.1401119232177734,
      "step": 8502
    },
    {
      "epoch": 3.243255615234375e-05,
      "step": 8502,
      "training_step_time": 10.319519996643066
    },
    {
      "epoch": 3.2436370849609375e-05,
      "model_forward_time": 3.158733367919922,
      "step": 8503
    },
    {
      "epoch": 3.2436370849609375e-05,
      "step": 8503,
      "training_step_time": 9.934122562408447
    },
    {
      "epoch": 3.2436370849609375e-05,
      "model_forward_time": 2.4439430236816406,
      "step": 8503
    },
    {
      "epoch": 3.2436370849609375e-05,
      "step": 8503,
      "training_step_time": 7.471636772155762
    },
    {
      "epoch": 3.2436370849609375e-05,
      "model_forward_time": 2.4368574619293213,
      "step": 8503
    },
    {
      "epoch": 3.2436370849609375e-05,
      "step": 8503,
      "training_step_time": 7.461853981018066
    },
    {
      "epoch": 3.2436370849609375e-05,
      "model_forward_time": 3.412180185317993,
      "step": 8503
    },
    {
      "epoch": 3.2436370849609375e-05,
      "step": 8503,
      "training_step_time": 10.304028987884521
    },
    {
      "epoch": 3.2440185546875e-05,
      "model_forward_time": 3.1707956790924072,
      "step": 8504
    },
    {
      "epoch": 3.2440185546875e-05,
      "step": 8504,
      "training_step_time": 9.94111680984497
    },
    {
      "epoch": 3.2440185546875e-05,
      "model_forward_time": 2.4360458850860596,
      "step": 8504
    },
    {
      "epoch": 3.2440185546875e-05,
      "step": 8504,
      "training_step_time": 7.465389251708984
    },
    {
      "epoch": 3.2440185546875e-05,
      "model_forward_time": 2.4401164054870605,
      "step": 8504
    },
    {
      "epoch": 3.2440185546875e-05,
      "step": 8504,
      "training_step_time": 7.471555709838867
    },
    {
      "epoch": 3.2440185546875e-05,
      "model_forward_time": 2.4351587295532227,
      "step": 8504
    },
    {
      "epoch": 3.2440185546875e-05,
      "step": 8504,
      "training_step_time": 10.314216136932373
    },
    {
      "epoch": 3.244400024414063e-05,
      "model_forward_time": 2.7477633953094482,
      "step": 8505
    },
    {
      "dynamics_loss_avg": 0.046647810749709605,
      "epoch": 3.244400024414063e-05,
      "step": 8505
    },
    {
      "action_loss_avg": 0.014082572562620044,
      "epoch": 3.244400024414063e-05,
      "step": 8505
    },
    {
      "epoch": 3.244400024414063e-05,
      "step": 8505,
      "training_step_time": 7.776954650878906
    },
    {
      "epoch": 3.244400024414063e-05,
      "model_forward_time": 3.134523391723633,
      "step": 8505
    },
    {
      "epoch": 3.244400024414063e-05,
      "step": 8505,
      "training_step_time": 9.899038076400757
    },
    {
      "epoch": 3.244400024414063e-05,
      "model_forward_time": 2.430917263031006,
      "step": 8505
    },
    {
      "epoch": 3.244400024414063e-05,
      "step": 8505,
      "training_step_time": 7.450712442398071
    },
    {
      "epoch": 3.244400024414063e-05,
      "model_forward_time": 2.433168649673462,
      "step": 8505
    },
    {
      "epoch": 3.244400024414063e-05,
      "step": 8505,
      "training_step_time": 10.016326189041138
    },
    {
      "epoch": 3.244781494140625e-05,
      "model_forward_time": 3.1443092823028564,
      "step": 8506
    },
    {
      "epoch": 3.244781494140625e-05,
      "step": 8506,
      "training_step_time": 9.908187866210938
    },
    {
      "epoch": 3.244781494140625e-05,
      "model_forward_time": 3.404872179031372,
      "step": 8506
    },
    {
      "epoch": 3.244781494140625e-05,
      "step": 8506,
      "training_step_time": 10.165861129760742
    },
    {
      "epoch": 3.244781494140625e-05,
      "model_forward_time": 3.137521505355835,
      "step": 8506
    },
    {
      "epoch": 3.244781494140625e-05,
      "step": 8506,
      "training_step_time": 9.906181573867798
    },
    {
      "epoch": 3.244781494140625e-05,
      "model_forward_time": 3.1320319175720215,
      "step": 8506
    },
    {
      "epoch": 3.244781494140625e-05,
      "step": 8506,
      "training_step_time": 10.31404185295105
    },
    {
      "epoch": 3.2451629638671874e-05,
      "model_forward_time": 2.4477481842041016,
      "step": 8507
    },
    {
      "epoch": 3.2451629638671874e-05,
      "step": 8507,
      "training_step_time": 7.473848819732666
    },
    {
      "epoch": 3.2451629638671874e-05,
      "model_forward_time": 2.4345879554748535,
      "step": 8507
    },
    {
      "epoch": 3.2451629638671874e-05,
      "step": 8507,
      "training_step_time": 7.464386224746704
    },
    {
      "epoch": 3.2451629638671874e-05,
      "model_forward_time": 3.4301156997680664,
      "step": 8507
    },
    {
      "dynamics_loss_avg": 0.03964799945242703,
      "epoch": 3.2451629638671874e-05,
      "step": 8507
    },
    {
      "action_loss_avg": 0.013286848238203674,
      "epoch": 3.2451629638671874e-05,
      "step": 8507
    },
    {
      "epoch": 3.2451629638671874e-05,
      "step": 8507,
      "training_step_time": 10.196064472198486
    },
    {
      "epoch": 3.2451629638671874e-05,
      "model_forward_time": 2.4343655109405518,
      "step": 8507
    },
    {
      "epoch": 3.2451629638671874e-05,
      "step": 8507,
      "training_step_time": 10.30940556526184
    },
    {
      "epoch": 3.24554443359375e-05,
      "model_forward_time": 3.1566901206970215,
      "step": 8508
    },
    {
      "epoch": 3.24554443359375e-05,
      "step": 8508,
      "training_step_time": 9.922965288162231
    },
    {
      "epoch": 3.24554443359375e-05,
      "model_forward_time": 3.1342108249664307,
      "step": 8508
    },
    {
      "epoch": 3.24554443359375e-05,
      "step": 8508,
      "training_step_time": 9.899631023406982
    },
    {
      "epoch": 3.24554443359375e-05,
      "model_forward_time": 3.134186267852783,
      "step": 8508
    },
    {
      "epoch": 3.24554443359375e-05,
      "step": 8508,
      "training_step_time": 9.901159763336182
    },
    {
      "epoch": 3.24554443359375e-05,
      "model_forward_time": 2.432457208633423,
      "step": 8508
    },
    {
      "epoch": 3.24554443359375e-05,
      "step": 8508,
      "training_step_time": 10.00564169883728
    },
    {
      "epoch": 3.2459259033203126e-05,
      "model_forward_time": 2.450654983520508,
      "step": 8509
    },
    {
      "epoch": 3.2459259033203126e-05,
      "step": 8509,
      "training_step_time": 7.47576117515564
    },
    {
      "epoch": 3.2459259033203126e-05,
      "model_forward_time": 3.1348605155944824,
      "step": 8509
    },
    {
      "epoch": 3.2459259033203126e-05,
      "step": 8509,
      "training_step_time": 9.899225234985352
    },
    {
      "epoch": 3.2459259033203126e-05,
      "model_forward_time": 3.1321496963500977,
      "step": 8509
    },
    {
      "epoch": 3.2459259033203126e-05,
      "step": 8509,
      "training_step_time": 9.895723342895508
    },
    {
      "epoch": 3.2459259033203126e-05,
      "model_forward_time": 2.438750982284546,
      "step": 8509
    },
    {
      "epoch": 3.2459259033203126e-05,
      "step": 8509,
      "training_step_time": 10.269962072372437
    },
    {
      "epoch": 3.246307373046875e-05,
      "grad_norm": 0.5368279218673706,
      "learning_rate": 1e-05,
      "loss": 0.0775,
      "step": 8510
    },
    {
      "epoch": 3.246307373046875e-05,
      "model_forward_time": 3.149120569229126,
      "step": 8510
    },
    {
      "dynamics_loss_avg": 0.060351238772273066,
      "epoch": 3.246307373046875e-05,
      "step": 8510
    },
    {
      "action_loss_avg": 0.01984290862455964,
      "epoch": 3.246307373046875e-05,
      "step": 8510
    },
    {
      "epoch": 3.246307373046875e-05,
      "step": 8510,
      "training_step_time": 9.921151161193848
    },
    {
      "epoch": 3.246307373046875e-05,
      "model_forward_time": 2.7054145336151123,
      "step": 8510
    },
    {
      "epoch": 3.246307373046875e-05,
      "step": 8510,
      "training_step_time": 7.73127293586731
    },
    {
      "epoch": 3.246307373046875e-05,
      "model_forward_time": 2.431053876876831,
      "step": 8510
    },
    {
      "epoch": 3.246307373046875e-05,
      "step": 8510,
      "training_step_time": 7.455275774002075
    },
    {
      "epoch": 3.246307373046875e-05,
      "model_forward_time": 3.13560152053833,
      "step": 8510
    },
    {
      "epoch": 3.246307373046875e-05,
      "step": 8510,
      "training_step_time": 10.037094116210938
    },
    {
      "epoch": 3.246688842773437e-05,
      "model_forward_time": 3.152362108230591,
      "step": 8511
    },
    {
      "epoch": 3.246688842773437e-05,
      "step": 8511,
      "training_step_time": 9.92549180984497
    },
    {
      "epoch": 3.246688842773437e-05,
      "model_forward_time": 3.1375279426574707,
      "step": 8511
    },
    {
      "epoch": 3.246688842773437e-05,
      "step": 8511,
      "training_step_time": 9.907636642456055
    },
    {
      "epoch": 3.246688842773437e-05,
      "model_forward_time": 3.389901876449585,
      "step": 8511
    },
    {
      "epoch": 3.246688842773437e-05,
      "step": 8511,
      "training_step_time": 10.160947322845459
    },
    {
      "epoch": 3.246688842773437e-05,
      "model_forward_time": 3.1444895267486572,
      "step": 8511
    },
    {
      "epoch": 3.246688842773437e-05,
      "step": 8511,
      "training_step_time": 10.323489665985107
    },
    {
      "epoch": 3.2470703125e-05,
      "model_forward_time": 2.4509599208831787,
      "step": 8512
    },
    {
      "epoch": 3.2470703125e-05,
      "step": 8512,
      "training_step_time": 7.478435516357422
    },
    {
      "epoch": 3.2470703125e-05,
      "model_forward_time": 3.129633665084839,
      "step": 8512
    },
    {
      "epoch": 3.2470703125e-05,
      "step": 8512,
      "training_step_time": 9.90008807182312
    },
    {
      "epoch": 3.2470703125e-05,
      "model_forward_time": 3.1341793537139893,
      "step": 8512
    },
    {
      "dynamics_loss_avg": 0.06021851003170013,
      "epoch": 3.2470703125e-05,
      "step": 8512
    },
    {
      "action_loss_avg": 0.019948842562735082,
      "epoch": 3.2470703125e-05,
      "step": 8512
    },
    {
      "epoch": 3.2470703125e-05,
      "step": 8512,
      "training_step_time": 9.911294221878052
    },
    {
      "epoch": 3.2470703125e-05,
      "model_forward_time": 2.7041916847229004,
      "step": 8512
    },
    {
      "epoch": 3.2470703125e-05,
      "step": 8512,
      "training_step_time": 10.01590633392334
    },
    {
      "epoch": 3.2474517822265625e-05,
      "model_forward_time": 3.168806552886963,
      "step": 8513
    },
    {
      "epoch": 3.2474517822265625e-05,
      "step": 8513,
      "training_step_time": 9.94662094116211
    },
    {
      "epoch": 3.2474517822265625e-05,
      "model_forward_time": 2.429331064224243,
      "step": 8513
    },
    {
      "epoch": 3.2474517822265625e-05,
      "step": 8513,
      "training_step_time": 7.455316543579102
    },
    {
      "epoch": 3.2474517822265625e-05,
      "model_forward_time": 3.130890369415283,
      "step": 8513
    },
    {
      "epoch": 3.2474517822265625e-05,
      "step": 8513,
      "training_step_time": 9.901272296905518
    },
    {
      "epoch": 3.2474517822265625e-05,
      "model_forward_time": 1.7451393604278564,
      "step": 8513
    },
    {
      "epoch": 3.2474517822265625e-05,
      "step": 8513,
      "training_step_time": 10.023300409317017
    },
    {
      "epoch": 3.247833251953125e-05,
      "model_forward_time": 2.720757246017456,
      "step": 8514
    },
    {
      "epoch": 3.247833251953125e-05,
      "step": 8514,
      "training_step_time": 7.747625827789307
    },
    {
      "epoch": 3.247833251953125e-05,
      "model_forward_time": 2.4326558113098145,
      "step": 8514
    },
    {
      "epoch": 3.247833251953125e-05,
      "step": 8514,
      "training_step_time": 7.461361646652222
    },
    {
      "epoch": 3.247833251953125e-05,
      "model_forward_time": 2.4275214672088623,
      "step": 8514
    },
    {
      "epoch": 3.247833251953125e-05,
      "step": 8514,
      "training_step_time": 7.452908039093018
    },
    {
      "epoch": 3.247833251953125e-05,
      "model_forward_time": 3.133269786834717,
      "step": 8514
    },
    {
      "epoch": 3.247833251953125e-05,
      "step": 8514,
      "training_step_time": 10.02421259880066
    },
    {
      "epoch": 3.248214721679688e-05,
      "model_forward_time": 2.4532721042633057,
      "step": 8515
    },
    {
      "dynamics_loss_avg": 0.06024703476577997,
      "epoch": 3.248214721679688e-05,
      "step": 8515
    },
    {
      "action_loss_avg": 0.02080767115112394,
      "epoch": 3.248214721679688e-05,
      "step": 8515
    },
    {
      "epoch": 3.248214721679688e-05,
      "step": 8515,
      "training_step_time": 7.499954700469971
    },
    {
      "epoch": 3.248214721679688e-05,
      "model_forward_time": 2.4362173080444336,
      "step": 8515
    },
    {
      "epoch": 3.248214721679688e-05,
      "step": 8515,
      "training_step_time": 7.460848093032837
    },
    {
      "epoch": 3.248214721679688e-05,
      "model_forward_time": 3.4194819927215576,
      "step": 8515
    },
    {
      "epoch": 3.248214721679688e-05,
      "step": 8515,
      "training_step_time": 10.186866044998169
    },
    {
      "epoch": 3.248214721679688e-05,
      "model_forward_time": 2.437767744064331,
      "step": 8515
    },
    {
      "epoch": 3.248214721679688e-05,
      "step": 8515,
      "training_step_time": 10.310460567474365
    },
    {
      "epoch": 3.24859619140625e-05,
      "model_forward_time": 2.4528963565826416,
      "step": 8516
    },
    {
      "epoch": 3.24859619140625e-05,
      "step": 8516,
      "training_step_time": 7.488530397415161
    },
    {
      "epoch": 3.24859619140625e-05,
      "model_forward_time": 3.1342761516571045,
      "step": 8516
    },
    {
      "epoch": 3.24859619140625e-05,
      "step": 8516,
      "training_step_time": 9.899168491363525
    },
    {
      "epoch": 3.24859619140625e-05,
      "model_forward_time": 2.4323370456695557,
      "step": 8516
    },
    {
      "epoch": 3.24859619140625e-05,
      "step": 8516,
      "training_step_time": 7.451448440551758
    },
    {
      "epoch": 3.24859619140625e-05,
      "model_forward_time": 2.7082347869873047,
      "step": 8516
    },
    {
      "epoch": 3.24859619140625e-05,
      "step": 8516,
      "training_step_time": 10.277750253677368
    },
    {
      "epoch": 3.248977661132812e-05,
      "model_forward_time": 2.4456377029418945,
      "step": 8517
    },
    {
      "epoch": 3.248977661132812e-05,
      "step": 8517,
      "training_step_time": 7.472510099411011
    },
    {
      "epoch": 3.248977661132812e-05,
      "model_forward_time": 3.1370761394500732,
      "step": 8517
    },
    {
      "epoch": 3.248977661132812e-05,
      "step": 8517,
      "training_step_time": 9.900897026062012
    },
    {
      "epoch": 3.248977661132812e-05,
      "model_forward_time": 3.1404125690460205,
      "step": 8517
    },
    {
      "dynamics_loss_avg": 0.056846832670271394,
      "epoch": 3.248977661132812e-05,
      "step": 8517
    },
    {
      "action_loss_avg": 0.020245547546073793,
      "epoch": 3.248977661132812e-05,
      "step": 8517
    },
    {
      "epoch": 3.248977661132812e-05,
      "step": 8517,
      "training_step_time": 9.910251140594482
    },
    {
      "epoch": 3.248977661132812e-05,
      "model_forward_time": 3.135586738586426,
      "step": 8517
    },
    {
      "epoch": 3.248977661132812e-05,
      "step": 8517,
      "training_step_time": 10.331807613372803
    },
    {
      "epoch": 3.249359130859375e-05,
      "model_forward_time": 2.729363203048706,
      "step": 8518
    },
    {
      "epoch": 3.249359130859375e-05,
      "step": 8518,
      "training_step_time": 7.766822099685669
    },
    {
      "epoch": 3.249359130859375e-05,
      "model_forward_time": 3.1343941688537598,
      "step": 8518
    },
    {
      "epoch": 3.249359130859375e-05,
      "step": 8518,
      "training_step_time": 9.897149801254272
    },
    {
      "epoch": 3.249359130859375e-05,
      "model_forward_time": 3.1322567462921143,
      "step": 8518
    },
    {
      "epoch": 3.249359130859375e-05,
      "step": 8518,
      "training_step_time": 9.901665210723877
    },
    {
      "epoch": 3.249359130859375e-05,
      "model_forward_time": 3.1366169452667236,
      "step": 8518
    },
    {
      "epoch": 3.249359130859375e-05,
      "step": 8518,
      "training_step_time": 10.035265445709229
    },
    {
      "epoch": 3.2497406005859375e-05,
      "model_forward_time": 3.162334442138672,
      "step": 8519
    },
    {
      "epoch": 3.2497406005859375e-05,
      "step": 8519,
      "training_step_time": 9.947235584259033
    },
    {
      "epoch": 3.2497406005859375e-05,
      "model_forward_time": 3.41379451751709,
      "step": 8519
    },
    {
      "epoch": 3.2497406005859375e-05,
      "step": 8519,
      "training_step_time": 10.185333251953125
    },
    {
      "epoch": 3.2497406005859375e-05,
      "model_forward_time": 3.1377124786376953,
      "step": 8519
    },
    {
      "epoch": 3.2497406005859375e-05,
      "step": 8519,
      "training_step_time": 9.91361141204834
    },
    {
      "epoch": 3.2497406005859375e-05,
      "model_forward_time": 3.1403422355651855,
      "step": 8519
    },
    {
      "epoch": 3.2497406005859375e-05,
      "step": 8519,
      "training_step_time": 10.297151327133179
    },
    {
      "epoch": 3.2501220703125e-05,
      "grad_norm": 0.3631563186645508,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 8520
    },
    {
      "epoch": 3.2501220703125e-05,
      "model_forward_time": 3.1630163192749023,
      "step": 8520
    },
    {
      "dynamics_loss_avg": 0.049637246318161486,
      "epoch": 3.2501220703125e-05,
      "step": 8520
    },
    {
      "action_loss_avg": 0.02053874754346907,
      "epoch": 3.2501220703125e-05,
      "step": 8520
    },
    {
      "epoch": 3.2501220703125e-05,
      "step": 8520,
      "training_step_time": 9.938523769378662
    },
    {
      "epoch": 3.2501220703125e-05,
      "model_forward_time": 3.138387441635132,
      "step": 8520
    },
    {
      "epoch": 3.2501220703125e-05,
      "step": 8520,
      "training_step_time": 9.906428098678589
    },
    {
      "epoch": 3.2501220703125e-05,
      "model_forward_time": 2.712038516998291,
      "step": 8520
    },
    {
      "epoch": 3.2501220703125e-05,
      "step": 8520,
      "training_step_time": 7.737705230712891
    },
    {
      "epoch": 3.2501220703125e-05,
      "model_forward_time": 2.430469512939453,
      "step": 8520
    },
    {
      "epoch": 3.2501220703125e-05,
      "step": 8520,
      "training_step_time": 10.228368520736694
    },
    {
      "epoch": 3.250503540039063e-05,
      "model_forward_time": 3.1493520736694336,
      "step": 8521
    },
    {
      "epoch": 3.250503540039063e-05,
      "step": 8521,
      "training_step_time": 9.917762756347656
    },
    {
      "epoch": 3.250503540039063e-05,
      "model_forward_time": 2.4316041469573975,
      "step": 8521
    },
    {
      "epoch": 3.250503540039063e-05,
      "step": 8521,
      "training_step_time": 7.460395812988281
    },
    {
      "epoch": 3.250503540039063e-05,
      "model_forward_time": 3.137108325958252,
      "step": 8521
    },
    {
      "epoch": 3.250503540039063e-05,
      "step": 8521,
      "training_step_time": 9.905343532562256
    },
    {
      "epoch": 3.250503540039063e-05,
      "model_forward_time": 2.4300904273986816,
      "step": 8521
    },
    {
      "epoch": 3.250503540039063e-05,
      "step": 8521,
      "training_step_time": 10.291353940963745
    },
    {
      "epoch": 3.250885009765625e-05,
      "model_forward_time": 3.4420368671417236,
      "step": 8522
    },
    {
      "epoch": 3.250885009765625e-05,
      "step": 8522,
      "training_step_time": 10.20994782447815
    },
    {
      "epoch": 3.250885009765625e-05,
      "model_forward_time": 3.1288368701934814,
      "step": 8522
    },
    {
      "epoch": 3.250885009765625e-05,
      "step": 8522,
      "training_step_time": 9.897433042526245
    },
    {
      "epoch": 3.250885009765625e-05,
      "model_forward_time": 2.4296092987060547,
      "step": 8522
    },
    {
      "dynamics_loss_avg": 0.04740804918110371,
      "epoch": 3.250885009765625e-05,
      "step": 8522
    },
    {
      "action_loss_avg": 0.017067881440743803,
      "epoch": 3.250885009765625e-05,
      "step": 8522
    },
    {
      "epoch": 3.250885009765625e-05,
      "step": 8522,
      "training_step_time": 7.454530954360962
    },
    {
      "epoch": 3.250885009765625e-05,
      "model_forward_time": 2.428697347640991,
      "step": 8522
    },
    {
      "epoch": 3.250885009765625e-05,
      "step": 8522,
      "training_step_time": 10.306861400604248
    },
    {
      "epoch": 3.2512664794921874e-05,
      "model_forward_time": 3.151071071624756,
      "step": 8523
    },
    {
      "epoch": 3.2512664794921874e-05,
      "step": 8523,
      "training_step_time": 9.91771125793457
    },
    {
      "epoch": 3.2512664794921874e-05,
      "model_forward_time": 2.7047922611236572,
      "step": 8523
    },
    {
      "epoch": 3.2512664794921874e-05,
      "step": 8523,
      "training_step_time": 7.728292226791382
    },
    {
      "epoch": 3.2512664794921874e-05,
      "model_forward_time": 2.428945541381836,
      "step": 8523
    },
    {
      "epoch": 3.2512664794921874e-05,
      "step": 8523,
      "training_step_time": 7.453277349472046
    },
    {
      "epoch": 3.2512664794921874e-05,
      "model_forward_time": 3.131610870361328,
      "step": 8523
    },
    {
      "epoch": 3.2512664794921874e-05,
      "step": 8523,
      "training_step_time": 10.02439546585083
    },
    {
      "epoch": 3.25164794921875e-05,
      "model_forward_time": 2.445590019226074,
      "step": 8524
    },
    {
      "epoch": 3.25164794921875e-05,
      "step": 8524,
      "training_step_time": 7.482238531112671
    },
    {
      "epoch": 3.25164794921875e-05,
      "model_forward_time": 2.4302008152008057,
      "step": 8524
    },
    {
      "epoch": 3.25164794921875e-05,
      "step": 8524,
      "training_step_time": 7.455859661102295
    },
    {
      "epoch": 3.25164794921875e-05,
      "model_forward_time": 3.124481201171875,
      "step": 8524
    },
    {
      "epoch": 3.25164794921875e-05,
      "step": 8524,
      "training_step_time": 10.159672975540161
    },
    {
      "epoch": 3.25164794921875e-05,
      "model_forward_time": 3.1280224323272705,
      "step": 8524
    },
    {
      "epoch": 3.25164794921875e-05,
      "step": 8524,
      "training_step_time": 10.301066875457764
    },
    {
      "epoch": 3.2520294189453126e-05,
      "model_forward_time": 3.1430716514587402,
      "step": 8525
    },
    {
      "dynamics_loss_avg": 0.04694144306704402,
      "epoch": 3.2520294189453126e-05,
      "step": 8525
    },
    {
      "action_loss_avg": 0.017440586956217884,
      "epoch": 3.2520294189453126e-05,
      "step": 8525
    },
    {
      "epoch": 3.2520294189453126e-05,
      "step": 8525,
      "training_step_time": 9.926014423370361
    },
    {
      "epoch": 3.2520294189453126e-05,
      "model_forward_time": 2.4331295490264893,
      "step": 8525
    },
    {
      "epoch": 3.2520294189453126e-05,
      "step": 8525,
      "training_step_time": 7.465872526168823
    },
    {
      "epoch": 3.2520294189453126e-05,
      "model_forward_time": 2.4338479042053223,
      "step": 8525
    },
    {
      "epoch": 3.2520294189453126e-05,
      "step": 8525,
      "training_step_time": 7.467127561569214
    },
    {
      "epoch": 3.2520294189453126e-05,
      "model_forward_time": 2.427561044692993,
      "step": 8525
    },
    {
      "epoch": 3.2520294189453126e-05,
      "step": 8525,
      "training_step_time": 10.039007425308228
    },
    {
      "epoch": 3.252410888671875e-05,
      "model_forward_time": 3.522451877593994,
      "step": 8526
    },
    {
      "epoch": 3.252410888671875e-05,
      "step": 8526,
      "training_step_time": 10.306076049804688
    },
    {
      "epoch": 3.252410888671875e-05,
      "model_forward_time": 3.1494460105895996,
      "step": 8526
    },
    {
      "epoch": 3.252410888671875e-05,
      "step": 8526,
      "training_step_time": 9.934937477111816
    },
    {
      "epoch": 3.252410888671875e-05,
      "model_forward_time": 2.4434633255004883,
      "step": 8526
    },
    {
      "epoch": 3.252410888671875e-05,
      "step": 8526,
      "training_step_time": 7.484532356262207
    },
    {
      "epoch": 3.252410888671875e-05,
      "model_forward_time": 2.4386212825775146,
      "step": 8526
    },
    {
      "epoch": 3.252410888671875e-05,
      "step": 8526,
      "training_step_time": 10.304336071014404
    },
    {
      "epoch": 3.252792358398437e-05,
      "model_forward_time": 3.1538190841674805,
      "step": 8527
    },
    {
      "epoch": 3.252792358398437e-05,
      "step": 8527,
      "training_step_time": 9.938326835632324
    },
    {
      "epoch": 3.252792358398437e-05,
      "model_forward_time": 2.761425733566284,
      "step": 8527
    },
    {
      "epoch": 3.252792358398437e-05,
      "step": 8527,
      "training_step_time": 7.796695232391357
    },
    {
      "epoch": 3.252792358398437e-05,
      "model_forward_time": 2.434419631958008,
      "step": 8527
    },
    {
      "dynamics_loss_avg": 0.05545008787885308,
      "epoch": 3.252792358398437e-05,
      "step": 8527
    },
    {
      "action_loss_avg": 0.01975397828500718,
      "epoch": 3.252792358398437e-05,
      "step": 8527
    },
    {
      "epoch": 3.252792358398437e-05,
      "step": 8527,
      "training_step_time": 7.472656488418579
    },
    {
      "epoch": 3.252792358398437e-05,
      "model_forward_time": 2.435551643371582,
      "step": 8527
    },
    {
      "epoch": 3.252792358398437e-05,
      "step": 8527,
      "training_step_time": 10.300430297851562
    },
    {
      "epoch": 3.253173828125e-05,
      "model_forward_time": 2.452026844024658,
      "step": 8528
    },
    {
      "epoch": 3.253173828125e-05,
      "step": 8528,
      "training_step_time": 7.489776849746704
    },
    {
      "epoch": 3.253173828125e-05,
      "model_forward_time": 2.435882568359375,
      "step": 8528
    },
    {
      "epoch": 3.253173828125e-05,
      "step": 8528,
      "training_step_time": 7.473673105239868
    },
    {
      "epoch": 3.253173828125e-05,
      "model_forward_time": 2.716566324234009,
      "step": 8528
    },
    {
      "epoch": 3.253173828125e-05,
      "step": 8528,
      "training_step_time": 7.747630596160889
    },
    {
      "epoch": 3.253173828125e-05,
      "model_forward_time": 2.4329333305358887,
      "step": 8528
    },
    {
      "epoch": 3.253173828125e-05,
      "step": 8528,
      "training_step_time": 10.291269063949585
    },
    {
      "epoch": 3.2535552978515625e-05,
      "model_forward_time": 3.1518893241882324,
      "step": 8529
    },
    {
      "epoch": 3.2535552978515625e-05,
      "step": 8529,
      "training_step_time": 9.928890228271484
    },
    {
      "epoch": 3.2535552978515625e-05,
      "model_forward_time": 2.436138868331909,
      "step": 8529
    },
    {
      "epoch": 3.2535552978515625e-05,
      "step": 8529,
      "training_step_time": 7.469066858291626
    },
    {
      "epoch": 3.2535552978515625e-05,
      "model_forward_time": 3.140679121017456,
      "step": 8529
    },
    {
      "epoch": 3.2535552978515625e-05,
      "step": 8529,
      "training_step_time": 9.910109519958496
    },
    {
      "epoch": 3.2535552978515625e-05,
      "model_forward_time": 3.139967679977417,
      "step": 8529
    },
    {
      "epoch": 3.2535552978515625e-05,
      "step": 8529,
      "training_step_time": 10.285449743270874
    },
    {
      "epoch": 3.253936767578125e-05,
      "grad_norm": 0.43243756890296936,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 8530
    },
    {
      "epoch": 3.253936767578125e-05,
      "model_forward_time": 3.4796793460845947,
      "step": 8530
    },
    {
      "dynamics_loss_avg": 0.06091589350253344,
      "epoch": 3.253936767578125e-05,
      "step": 8530
    },
    {
      "action_loss_avg": 0.025258039589971304,
      "epoch": 3.253936767578125e-05,
      "step": 8530
    },
    {
      "epoch": 3.253936767578125e-05,
      "step": 8530,
      "training_step_time": 10.26437520980835
    },
    {
      "epoch": 3.253936767578125e-05,
      "model_forward_time": 2.4373457431793213,
      "step": 8530
    },
    {
      "epoch": 3.253936767578125e-05,
      "step": 8530,
      "training_step_time": 7.473928689956665
    },
    {
      "epoch": 3.253936767578125e-05,
      "model_forward_time": 3.139064311981201,
      "step": 8530
    },
    {
      "epoch": 3.253936767578125e-05,
      "step": 8530,
      "training_step_time": 9.917499542236328
    },
    {
      "epoch": 3.253936767578125e-05,
      "model_forward_time": 3.139814615249634,
      "step": 8530
    },
    {
      "epoch": 3.253936767578125e-05,
      "step": 8530,
      "training_step_time": 10.306782960891724
    },
    {
      "epoch": 3.254318237304688e-05,
      "model_forward_time": 2.453582286834717,
      "step": 8531
    },
    {
      "epoch": 3.254318237304688e-05,
      "step": 8531,
      "training_step_time": 7.492089033126831
    },
    {
      "epoch": 3.254318237304688e-05,
      "model_forward_time": 2.748365640640259,
      "step": 8531
    },
    {
      "epoch": 3.254318237304688e-05,
      "step": 8531,
      "training_step_time": 7.777840614318848
    },
    {
      "epoch": 3.254318237304688e-05,
      "model_forward_time": 3.1408984661102295,
      "step": 8531
    },
    {
      "epoch": 3.254318237304688e-05,
      "step": 8531,
      "training_step_time": 9.918437957763672
    },
    {
      "epoch": 3.254318237304688e-05,
      "model_forward_time": 2.438825845718384,
      "step": 8531
    },
    {
      "epoch": 3.254318237304688e-05,
      "step": 8531,
      "training_step_time": 10.041034460067749
    },
    {
      "epoch": 3.25469970703125e-05,
      "model_forward_time": 3.1518027782440186,
      "step": 8532
    },
    {
      "epoch": 3.25469970703125e-05,
      "step": 8532,
      "training_step_time": 9.928585290908813
    },
    {
      "epoch": 3.25469970703125e-05,
      "model_forward_time": 2.4409446716308594,
      "step": 8532
    },
    {
      "epoch": 3.25469970703125e-05,
      "step": 8532,
      "training_step_time": 7.4713239669799805
    },
    {
      "epoch": 3.25469970703125e-05,
      "model_forward_time": 3.4693408012390137,
      "step": 8532
    },
    {
      "dynamics_loss_avg": 0.05811587274074555,
      "epoch": 3.25469970703125e-05,
      "step": 8532
    },
    {
      "action_loss_avg": 0.019184206100180744,
      "epoch": 3.25469970703125e-05,
      "step": 8532
    },
    {
      "epoch": 3.25469970703125e-05,
      "step": 8532,
      "training_step_time": 10.250383853912354
    },
    {
      "epoch": 3.25469970703125e-05,
      "model_forward_time": 3.1451897621154785,
      "step": 8532
    },
    {
      "epoch": 3.25469970703125e-05,
      "step": 8532,
      "training_step_time": 10.340790271759033
    },
    {
      "epoch": 3.255081176757812e-05,
      "model_forward_time": 3.1492605209350586,
      "step": 8533
    },
    {
      "epoch": 3.255081176757812e-05,
      "step": 8533,
      "training_step_time": 9.927946329116821
    },
    {
      "epoch": 3.255081176757812e-05,
      "model_forward_time": 3.1456382274627686,
      "step": 8533
    },
    {
      "epoch": 3.255081176757812e-05,
      "step": 8533,
      "training_step_time": 9.92659330368042
    },
    {
      "epoch": 3.255081176757812e-05,
      "model_forward_time": 2.4407284259796143,
      "step": 8533
    },
    {
      "epoch": 3.255081176757812e-05,
      "step": 8533,
      "training_step_time": 7.470001220703125
    },
    {
      "epoch": 3.255081176757812e-05,
      "model_forward_time": 2.7461130619049072,
      "step": 8533
    },
    {
      "epoch": 3.255081176757812e-05,
      "step": 8533,
      "training_step_time": 10.335244178771973
    },
    {
      "epoch": 3.255462646484375e-05,
      "model_forward_time": 2.4632763862609863,
      "step": 8534
    },
    {
      "epoch": 3.255462646484375e-05,
      "step": 8534,
      "training_step_time": 7.497810363769531
    },
    {
      "epoch": 3.255462646484375e-05,
      "model_forward_time": 2.4331846237182617,
      "step": 8534
    },
    {
      "epoch": 3.255462646484375e-05,
      "step": 8534,
      "training_step_time": 7.4581685066223145
    },
    {
      "epoch": 3.255462646484375e-05,
      "model_forward_time": 2.4368107318878174,
      "step": 8534
    },
    {
      "epoch": 3.255462646484375e-05,
      "step": 8534,
      "training_step_time": 7.459157943725586
    },
    {
      "epoch": 3.255462646484375e-05,
      "model_forward_time": 3.139059066772461,
      "step": 8534
    },
    {
      "epoch": 3.255462646484375e-05,
      "step": 8534,
      "training_step_time": 10.257406234741211
    },
    {
      "epoch": 3.2558441162109376e-05,
      "model_forward_time": 2.44869327545166,
      "step": 8535
    },
    {
      "dynamics_loss_avg": 0.06533634960651398,
      "epoch": 3.2558441162109376e-05,
      "step": 8535
    },
    {
      "action_loss_avg": 0.028470317553728818,
      "epoch": 3.2558441162109376e-05,
      "step": 8535
    },
    {
      "epoch": 3.2558441162109376e-05,
      "step": 8535,
      "training_step_time": 7.750334024429321
    },
    {
      "epoch": 3.2558441162109376e-05,
      "model_forward_time": 3.1438229084014893,
      "step": 8535
    },
    {
      "epoch": 3.2558441162109376e-05,
      "step": 8535,
      "training_step_time": 9.912710428237915
    },
    {
      "epoch": 3.2558441162109376e-05,
      "model_forward_time": 3.141035318374634,
      "step": 8535
    },
    {
      "epoch": 3.2558441162109376e-05,
      "step": 8535,
      "training_step_time": 9.919946670532227
    },
    {
      "epoch": 3.2558441162109376e-05,
      "model_forward_time": 3.1412177085876465,
      "step": 8535
    },
    {
      "epoch": 3.2558441162109376e-05,
      "step": 8535,
      "training_step_time": 10.30257534980774
    },
    {
      "epoch": 3.2562255859375e-05,
      "model_forward_time": 3.158505916595459,
      "step": 8536
    },
    {
      "epoch": 3.2562255859375e-05,
      "step": 8536,
      "training_step_time": 9.929715633392334
    },
    {
      "epoch": 3.2562255859375e-05,
      "model_forward_time": 2.434113025665283,
      "step": 8536
    },
    {
      "epoch": 3.2562255859375e-05,
      "step": 8536,
      "training_step_time": 7.734098434448242
    },
    {
      "epoch": 3.2562255859375e-05,
      "model_forward_time": 3.135446786880493,
      "step": 8536
    },
    {
      "epoch": 3.2562255859375e-05,
      "step": 8536,
      "training_step_time": 9.903260231018066
    },
    {
      "epoch": 3.2562255859375e-05,
      "model_forward_time": 3.141430616378784,
      "step": 8536
    },
    {
      "epoch": 3.2562255859375e-05,
      "step": 8536,
      "training_step_time": 10.013261318206787
    },
    {
      "epoch": 3.256607055664063e-05,
      "model_forward_time": 3.1606407165527344,
      "step": 8537
    },
    {
      "epoch": 3.256607055664063e-05,
      "step": 8537,
      "training_step_time": 9.93983769416809
    },
    {
      "epoch": 3.256607055664063e-05,
      "model_forward_time": 3.1394054889678955,
      "step": 8537
    },
    {
      "epoch": 3.256607055664063e-05,
      "step": 8537,
      "training_step_time": 9.910816431045532
    },
    {
      "epoch": 3.256607055664063e-05,
      "model_forward_time": 3.4372024536132812,
      "step": 8537
    },
    {
      "dynamics_loss_avg": 0.05969129540026188,
      "epoch": 3.256607055664063e-05,
      "step": 8537
    },
    {
      "action_loss_avg": 0.027548361150547863,
      "epoch": 3.256607055664063e-05,
      "step": 8537
    },
    {
      "epoch": 3.256607055664063e-05,
      "step": 8537,
      "training_step_time": 10.211984634399414
    },
    {
      "epoch": 3.256607055664063e-05,
      "model_forward_time": 2.437591314315796,
      "step": 8537
    },
    {
      "epoch": 3.256607055664063e-05,
      "step": 8537,
      "training_step_time": 10.2865309715271
    },
    {
      "epoch": 3.256988525390625e-05,
      "model_forward_time": 2.449057102203369,
      "step": 8538
    },
    {
      "epoch": 3.256988525390625e-05,
      "step": 8538,
      "training_step_time": 7.478487014770508
    },
    {
      "epoch": 3.256988525390625e-05,
      "model_forward_time": 2.4358294010162354,
      "step": 8538
    },
    {
      "epoch": 3.256988525390625e-05,
      "step": 8538,
      "training_step_time": 7.4650304317474365
    },
    {
      "epoch": 3.256988525390625e-05,
      "model_forward_time": 3.1406431198120117,
      "step": 8538
    },
    {
      "epoch": 3.256988525390625e-05,
      "step": 8538,
      "training_step_time": 9.914632558822632
    },
    {
      "epoch": 3.256988525390625e-05,
      "model_forward_time": 3.144414186477661,
      "step": 8538
    },
    {
      "epoch": 3.256988525390625e-05,
      "step": 8538,
      "training_step_time": 10.335083723068237
    },
    {
      "epoch": 3.2573699951171874e-05,
      "model_forward_time": 2.779677391052246,
      "step": 8539
    },
    {
      "epoch": 3.2573699951171874e-05,
      "step": 8539,
      "training_step_time": 7.82813835144043
    },
    {
      "epoch": 3.2573699951171874e-05,
      "model_forward_time": 2.4388787746429443,
      "step": 8539
    },
    {
      "epoch": 3.2573699951171874e-05,
      "step": 8539,
      "training_step_time": 7.471058368682861
    },
    {
      "epoch": 3.2573699951171874e-05,
      "model_forward_time": 2.436526298522949,
      "step": 8539
    },
    {
      "epoch": 3.2573699951171874e-05,
      "step": 8539,
      "training_step_time": 7.474881410598755
    },
    {
      "epoch": 3.2573699951171874e-05,
      "model_forward_time": 2.4368770122528076,
      "step": 8539
    },
    {
      "epoch": 3.2573699951171874e-05,
      "step": 8539,
      "training_step_time": 10.03139615058899
    },
    {
      "epoch": 3.25775146484375e-05,
      "grad_norm": 0.41012659668922424,
      "learning_rate": 1e-05,
      "loss": 0.0771,
      "step": 8540
    },
    {
      "epoch": 3.25775146484375e-05,
      "model_forward_time": 2.453972816467285,
      "step": 8540
    },
    {
      "dynamics_loss_avg": 0.03611866952851415,
      "epoch": 3.25775146484375e-05,
      "step": 8540
    },
    {
      "action_loss_avg": 0.015068251569755376,
      "epoch": 3.25775146484375e-05,
      "step": 8540
    },
    {
      "epoch": 3.25775146484375e-05,
      "step": 8540,
      "training_step_time": 7.493469476699829
    },
    {
      "epoch": 3.25775146484375e-05,
      "model_forward_time": 2.771364450454712,
      "step": 8540
    },
    {
      "epoch": 3.25775146484375e-05,
      "step": 8540,
      "training_step_time": 7.8054163455963135
    },
    {
      "epoch": 3.25775146484375e-05,
      "model_forward_time": 3.1409175395965576,
      "step": 8540
    },
    {
      "epoch": 3.25775146484375e-05,
      "step": 8540,
      "training_step_time": 9.920574188232422
    },
    {
      "epoch": 3.25775146484375e-05,
      "model_forward_time": 3.141122341156006,
      "step": 8540
    },
    {
      "epoch": 3.25775146484375e-05,
      "step": 8540,
      "training_step_time": 10.030567169189453
    },
    {
      "epoch": 3.2581329345703126e-05,
      "model_forward_time": 3.1713716983795166,
      "step": 8541
    },
    {
      "epoch": 3.2581329345703126e-05,
      "step": 8541,
      "training_step_time": 9.962356567382812
    },
    {
      "epoch": 3.2581329345703126e-05,
      "model_forward_time": 3.1495933532714844,
      "step": 8541
    },
    {
      "epoch": 3.2581329345703126e-05,
      "step": 8541,
      "training_step_time": 9.934787034988403
    },
    {
      "epoch": 3.2581329345703126e-05,
      "model_forward_time": 2.4376866817474365,
      "step": 8541
    },
    {
      "epoch": 3.2581329345703126e-05,
      "step": 8541,
      "training_step_time": 7.780900239944458
    },
    {
      "epoch": 3.2581329345703126e-05,
      "model_forward_time": 2.4351518154144287,
      "step": 8541
    },
    {
      "epoch": 3.2581329345703126e-05,
      "step": 8541,
      "training_step_time": 10.024028062820435
    },
    {
      "epoch": 3.258514404296875e-05,
      "model_forward_time": 3.147599935531616,
      "step": 8542
    },
    {
      "epoch": 3.258514404296875e-05,
      "step": 8542,
      "training_step_time": 9.933343410491943
    },
    {
      "epoch": 3.258514404296875e-05,
      "model_forward_time": 3.1420958042144775,
      "step": 8542
    },
    {
      "epoch": 3.258514404296875e-05,
      "step": 8542,
      "training_step_time": 9.919779062271118
    },
    {
      "epoch": 3.258514404296875e-05,
      "model_forward_time": 3.137197256088257,
      "step": 8542
    },
    {
      "dynamics_loss_avg": 0.060824428871274,
      "epoch": 3.258514404296875e-05,
      "step": 8542
    },
    {
      "action_loss_avg": 0.018613330740481614,
      "epoch": 3.258514404296875e-05,
      "step": 8542
    },
    {
      "epoch": 3.258514404296875e-05,
      "step": 8542,
      "training_step_time": 9.909952878952026
    },
    {
      "epoch": 3.258514404296875e-05,
      "model_forward_time": 2.433337688446045,
      "step": 8542
    },
    {
      "epoch": 3.258514404296875e-05,
      "step": 8542,
      "training_step_time": 10.228160381317139
    },
    {
      "epoch": 3.258895874023437e-05,
      "model_forward_time": 2.449591636657715,
      "step": 8543
    },
    {
      "epoch": 3.258895874023437e-05,
      "step": 8543,
      "training_step_time": 7.47648024559021
    },
    {
      "epoch": 3.258895874023437e-05,
      "model_forward_time": 3.1389520168304443,
      "step": 8543
    },
    {
      "epoch": 3.258895874023437e-05,
      "step": 8543,
      "training_step_time": 9.906123161315918
    },
    {
      "epoch": 3.258895874023437e-05,
      "model_forward_time": 2.435838222503662,
      "step": 8543
    },
    {
      "epoch": 3.258895874023437e-05,
      "step": 8543,
      "training_step_time": 7.462180137634277
    },
    {
      "epoch": 3.258895874023437e-05,
      "model_forward_time": 2.43583345413208,
      "step": 8543
    },
    {
      "epoch": 3.258895874023437e-05,
      "step": 8543,
      "training_step_time": 10.32839846611023
    },
    {
      "epoch": 3.25927734375e-05,
      "model_forward_time": 3.154541254043579,
      "step": 8544
    },
    {
      "epoch": 3.25927734375e-05,
      "step": 8544,
      "training_step_time": 9.927462816238403
    },
    {
      "epoch": 3.25927734375e-05,
      "model_forward_time": 2.713181972503662,
      "step": 8544
    },
    {
      "epoch": 3.25927734375e-05,
      "step": 8544,
      "training_step_time": 7.743471384048462
    },
    {
      "epoch": 3.25927734375e-05,
      "model_forward_time": 2.4348089694976807,
      "step": 8544
    },
    {
      "epoch": 3.25927734375e-05,
      "step": 8544,
      "training_step_time": 7.462754726409912
    },
    {
      "epoch": 3.25927734375e-05,
      "model_forward_time": 2.4362905025482178,
      "step": 8544
    },
    {
      "epoch": 3.25927734375e-05,
      "step": 8544,
      "training_step_time": 10.309074878692627
    },
    {
      "epoch": 3.2596588134765625e-05,
      "model_forward_time": 2.451573371887207,
      "step": 8545
    },
    {
      "dynamics_loss_avg": 0.057820603251457214,
      "epoch": 3.2596588134765625e-05,
      "step": 8545
    },
    {
      "action_loss_avg": 0.022747602593153716,
      "epoch": 3.2596588134765625e-05,
      "step": 8545
    },
    {
      "epoch": 3.2596588134765625e-05,
      "step": 8545,
      "training_step_time": 7.483147859573364
    },
    {
      "epoch": 3.2596588134765625e-05,
      "model_forward_time": 3.13572096824646,
      "step": 8545
    },
    {
      "epoch": 3.2596588134765625e-05,
      "step": 8545,
      "training_step_time": 9.905877113342285
    },
    {
      "epoch": 3.2596588134765625e-05,
      "model_forward_time": 3.4145290851593018,
      "step": 8545
    },
    {
      "epoch": 3.2596588134765625e-05,
      "step": 8545,
      "training_step_time": 10.184672117233276
    },
    {
      "epoch": 3.2596588134765625e-05,
      "model_forward_time": 2.4360175132751465,
      "step": 8545
    },
    {
      "epoch": 3.2596588134765625e-05,
      "step": 8545,
      "training_step_time": 10.284723043441772
    },
    {
      "epoch": 3.260040283203125e-05,
      "model_forward_time": 3.1528971195220947,
      "step": 8546
    },
    {
      "epoch": 3.260040283203125e-05,
      "step": 8546,
      "training_step_time": 9.91781497001648
    },
    {
      "epoch": 3.260040283203125e-05,
      "model_forward_time": 3.135650873184204,
      "step": 8546
    },
    {
      "epoch": 3.260040283203125e-05,
      "step": 8546,
      "training_step_time": 9.905862808227539
    },
    {
      "epoch": 3.260040283203125e-05,
      "model_forward_time": 3.1441712379455566,
      "step": 8546
    },
    {
      "epoch": 3.260040283203125e-05,
      "step": 8546,
      "training_step_time": 9.919474124908447
    },
    {
      "epoch": 3.260040283203125e-05,
      "model_forward_time": 3.4562416076660156,
      "step": 8546
    },
    {
      "epoch": 3.260040283203125e-05,
      "step": 8546,
      "training_step_time": 10.322976350784302
    },
    {
      "epoch": 3.260421752929688e-05,
      "model_forward_time": 2.459561586380005,
      "step": 8547
    },
    {
      "epoch": 3.260421752929688e-05,
      "step": 8547,
      "training_step_time": 7.491036891937256
    },
    {
      "epoch": 3.260421752929688e-05,
      "model_forward_time": 2.4339792728424072,
      "step": 8547
    },
    {
      "epoch": 3.260421752929688e-05,
      "step": 8547,
      "training_step_time": 7.461405992507935
    },
    {
      "epoch": 3.260421752929688e-05,
      "model_forward_time": 2.4347705841064453,
      "step": 8547
    },
    {
      "dynamics_loss_avg": 0.07108360268175602,
      "epoch": 3.260421752929688e-05,
      "step": 8547
    },
    {
      "action_loss_avg": 0.024466697219759227,
      "epoch": 3.260421752929688e-05,
      "step": 8547
    },
    {
      "epoch": 3.260421752929688e-05,
      "step": 8547,
      "training_step_time": 7.460520029067993
    },
    {
      "epoch": 3.260421752929688e-05,
      "model_forward_time": 2.43202805519104,
      "step": 8547
    },
    {
      "epoch": 3.260421752929688e-05,
      "step": 8547,
      "training_step_time": 10.016205787658691
    },
    {
      "epoch": 3.26080322265625e-05,
      "model_forward_time": 3.1513144969940186,
      "step": 8548
    },
    {
      "epoch": 3.26080322265625e-05,
      "step": 8548,
      "training_step_time": 9.936543226242065
    },
    {
      "epoch": 3.26080322265625e-05,
      "model_forward_time": 2.841938018798828,
      "step": 8548
    },
    {
      "epoch": 3.26080322265625e-05,
      "step": 8548,
      "training_step_time": 7.879395961761475
    },
    {
      "epoch": 3.26080322265625e-05,
      "model_forward_time": 3.1599159240722656,
      "step": 8548
    },
    {
      "epoch": 3.26080322265625e-05,
      "step": 8548,
      "training_step_time": 9.949648141860962
    },
    {
      "epoch": 3.26080322265625e-05,
      "model_forward_time": 2.4549026489257812,
      "step": 8548
    },
    {
      "epoch": 3.26080322265625e-05,
      "step": 8548,
      "training_step_time": 10.321573495864868
    },
    {
      "epoch": 3.261184692382812e-05,
      "model_forward_time": 3.1510603427886963,
      "step": 8549
    },
    {
      "epoch": 3.261184692382812e-05,
      "step": 8549,
      "training_step_time": 9.924503803253174
    },
    {
      "epoch": 3.261184692382812e-05,
      "model_forward_time": 3.136672258377075,
      "step": 8549
    },
    {
      "epoch": 3.261184692382812e-05,
      "step": 8549,
      "training_step_time": 9.911216259002686
    },
    {
      "epoch": 3.261184692382812e-05,
      "model_forward_time": 2.7687089443206787,
      "step": 8549
    },
    {
      "epoch": 3.261184692382812e-05,
      "step": 8549,
      "training_step_time": 7.801058769226074
    },
    {
      "epoch": 3.261184692382812e-05,
      "model_forward_time": 3.1377439498901367,
      "step": 8549
    },
    {
      "epoch": 3.261184692382812e-05,
      "step": 8549,
      "training_step_time": 10.322680473327637
    },
    {
      "epoch": 3.261566162109375e-05,
      "grad_norm": 0.2973555624485016,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 8550
    },
    {
      "epoch": 3.261566162109375e-05,
      "model_forward_time": 3.151364803314209,
      "step": 8550
    },
    {
      "dynamics_loss_avg": 0.05512622762471438,
      "epoch": 3.261566162109375e-05,
      "step": 8550
    },
    {
      "action_loss_avg": 0.018405925063416363,
      "epoch": 3.261566162109375e-05,
      "step": 8550
    },
    {
      "epoch": 3.261566162109375e-05,
      "step": 8550,
      "training_step_time": 9.931932926177979
    },
    {
      "epoch": 3.261566162109375e-05,
      "model_forward_time": 3.1399660110473633,
      "step": 8550
    },
    {
      "epoch": 3.261566162109375e-05,
      "step": 8550,
      "training_step_time": 9.914289951324463
    },
    {
      "epoch": 3.261566162109375e-05,
      "model_forward_time": 2.436643600463867,
      "step": 8550
    },
    {
      "epoch": 3.261566162109375e-05,
      "step": 8550,
      "training_step_time": 7.4663004875183105
    },
    {
      "epoch": 3.261566162109375e-05,
      "model_forward_time": 2.7057533264160156,
      "step": 8550
    },
    {
      "epoch": 3.261566162109375e-05,
      "step": 8550,
      "training_step_time": 10.28005599975586
    },
    {
      "epoch": 3.2619476318359376e-05,
      "model_forward_time": 3.167085647583008,
      "step": 8551
    },
    {
      "epoch": 3.2619476318359376e-05,
      "step": 8551,
      "training_step_time": 9.941271781921387
    },
    {
      "epoch": 3.2619476318359376e-05,
      "model_forward_time": 2.4318811893463135,
      "step": 8551
    },
    {
      "epoch": 3.2619476318359376e-05,
      "step": 8551,
      "training_step_time": 7.458524942398071
    },
    {
      "epoch": 3.2619476318359376e-05,
      "model_forward_time": 3.129143238067627,
      "step": 8551
    },
    {
      "epoch": 3.2619476318359376e-05,
      "step": 8551,
      "training_step_time": 9.895543575286865
    },
    {
      "epoch": 3.2619476318359376e-05,
      "model_forward_time": 2.4306483268737793,
      "step": 8551
    },
    {
      "epoch": 3.2619476318359376e-05,
      "step": 8551,
      "training_step_time": 10.00707721710205
    },
    {
      "epoch": 3.2623291015625e-05,
      "model_forward_time": 3.439525604248047,
      "step": 8552
    },
    {
      "epoch": 3.2623291015625e-05,
      "step": 8552,
      "training_step_time": 10.208890914916992
    },
    {
      "epoch": 3.2623291015625e-05,
      "model_forward_time": 2.430833101272583,
      "step": 8552
    },
    {
      "epoch": 3.2623291015625e-05,
      "step": 8552,
      "training_step_time": 7.45464015007019
    },
    {
      "epoch": 3.2623291015625e-05,
      "model_forward_time": 3.1271190643310547,
      "step": 8552
    },
    {
      "dynamics_loss_avg": 0.06051911283284426,
      "epoch": 3.2623291015625e-05,
      "step": 8552
    },
    {
      "action_loss_avg": 0.021652685990557075,
      "epoch": 3.2623291015625e-05,
      "step": 8552
    },
    {
      "epoch": 3.2623291015625e-05,
      "step": 8552,
      "training_step_time": 9.896905183792114
    },
    {
      "epoch": 3.2623291015625e-05,
      "model_forward_time": 3.138758897781372,
      "step": 8552
    },
    {
      "epoch": 3.2623291015625e-05,
      "step": 8552,
      "training_step_time": 10.023083448410034
    },
    {
      "epoch": 3.262710571289063e-05,
      "model_forward_time": 3.1568686962127686,
      "step": 8553
    },
    {
      "epoch": 3.262710571289063e-05,
      "step": 8553,
      "training_step_time": 9.929324865341187
    },
    {
      "epoch": 3.262710571289063e-05,
      "model_forward_time": 2.431920289993286,
      "step": 8553
    },
    {
      "epoch": 3.262710571289063e-05,
      "step": 8553,
      "training_step_time": 7.745454788208008
    },
    {
      "epoch": 3.262710571289063e-05,
      "model_forward_time": 3.1305480003356934,
      "step": 8553
    },
    {
      "epoch": 3.262710571289063e-05,
      "step": 8553,
      "training_step_time": 9.907848358154297
    },
    {
      "epoch": 3.262710571289063e-05,
      "model_forward_time": 2.4359312057495117,
      "step": 8553
    },
    {
      "epoch": 3.262710571289063e-05,
      "step": 8553,
      "training_step_time": 10.287777185440063
    },
    {
      "epoch": 3.263092041015625e-05,
      "model_forward_time": 3.1460249423980713,
      "step": 8554
    },
    {
      "epoch": 3.263092041015625e-05,
      "step": 8554,
      "training_step_time": 9.918298959732056
    },
    {
      "epoch": 3.263092041015625e-05,
      "model_forward_time": 2.435593366622925,
      "step": 8554
    },
    {
      "epoch": 3.263092041015625e-05,
      "step": 8554,
      "training_step_time": 7.46662163734436
    },
    {
      "epoch": 3.263092041015625e-05,
      "model_forward_time": 3.1352157592773438,
      "step": 8554
    },
    {
      "epoch": 3.263092041015625e-05,
      "step": 8554,
      "training_step_time": 9.906129360198975
    },
    {
      "epoch": 3.263092041015625e-05,
      "model_forward_time": 3.399486541748047,
      "step": 8554
    },
    {
      "epoch": 3.263092041015625e-05,
      "step": 8554,
      "training_step_time": 10.323901176452637
    },
    {
      "epoch": 3.2634735107421874e-05,
      "model_forward_time": 2.45100736618042,
      "step": 8555
    },
    {
      "dynamics_loss_avg": 0.05171463210135698,
      "epoch": 3.2634735107421874e-05,
      "step": 8555
    },
    {
      "action_loss_avg": 0.01607174929231405,
      "epoch": 3.2634735107421874e-05,
      "step": 8555
    },
    {
      "epoch": 3.2634735107421874e-05,
      "step": 8555,
      "training_step_time": 7.481144905090332
    },
    {
      "epoch": 3.2634735107421874e-05,
      "model_forward_time": 3.1374199390411377,
      "step": 8555
    },
    {
      "epoch": 3.2634735107421874e-05,
      "step": 8555,
      "training_step_time": 9.90472960472107
    },
    {
      "epoch": 3.2634735107421874e-05,
      "model_forward_time": 2.430647134780884,
      "step": 8555
    },
    {
      "epoch": 3.2634735107421874e-05,
      "step": 8555,
      "training_step_time": 7.45003080368042
    },
    {
      "epoch": 3.2634735107421874e-05,
      "model_forward_time": 2.4260144233703613,
      "step": 8555
    },
    {
      "epoch": 3.2634735107421874e-05,
      "step": 8555,
      "training_step_time": 10.297587394714355
    },
    {
      "epoch": 3.26385498046875e-05,
      "model_forward_time": 3.4650511741638184,
      "step": 8556
    },
    {
      "epoch": 3.26385498046875e-05,
      "step": 8556,
      "training_step_time": 10.230782747268677
    },
    {
      "epoch": 3.26385498046875e-05,
      "model_forward_time": 2.4331045150756836,
      "step": 8556
    },
    {
      "epoch": 3.26385498046875e-05,
      "step": 8556,
      "training_step_time": 7.458495855331421
    },
    {
      "epoch": 3.26385498046875e-05,
      "model_forward_time": 3.1353795528411865,
      "step": 8556
    },
    {
      "epoch": 3.26385498046875e-05,
      "step": 8556,
      "training_step_time": 9.90346646308899
    },
    {
      "epoch": 3.26385498046875e-05,
      "model_forward_time": 3.1322147846221924,
      "step": 8556
    },
    {
      "epoch": 3.26385498046875e-05,
      "step": 8556,
      "training_step_time": 10.013052225112915
    },
    {
      "epoch": 3.2642364501953126e-05,
      "model_forward_time": 2.4548933506011963,
      "step": 8557
    },
    {
      "epoch": 3.2642364501953126e-05,
      "step": 8557,
      "training_step_time": 7.481130123138428
    },
    {
      "epoch": 3.2642364501953126e-05,
      "model_forward_time": 2.726435899734497,
      "step": 8557
    },
    {
      "epoch": 3.2642364501953126e-05,
      "step": 8557,
      "training_step_time": 7.747527122497559
    },
    {
      "epoch": 3.2642364501953126e-05,
      "model_forward_time": 3.140749931335449,
      "step": 8557
    },
    {
      "dynamics_loss_avg": 0.06487548006698489,
      "epoch": 3.2642364501953126e-05,
      "step": 8557
    },
    {
      "action_loss_avg": 0.026364520750939847,
      "epoch": 3.2642364501953126e-05,
      "step": 8557
    },
    {
      "epoch": 3.2642364501953126e-05,
      "step": 8557,
      "training_step_time": 9.904171466827393
    },
    {
      "epoch": 3.2642364501953126e-05,
      "model_forward_time": 3.137854814529419,
      "step": 8557
    },
    {
      "epoch": 3.2642364501953126e-05,
      "step": 8557,
      "training_step_time": 10.003813028335571
    },
    {
      "epoch": 3.264617919921875e-05,
      "model_forward_time": 3.1479690074920654,
      "step": 8558
    },
    {
      "epoch": 3.264617919921875e-05,
      "step": 8558,
      "training_step_time": 9.912886381149292
    },
    {
      "epoch": 3.264617919921875e-05,
      "model_forward_time": 3.1406285762786865,
      "step": 8558
    },
    {
      "epoch": 3.264617919921875e-05,
      "step": 8558,
      "training_step_time": 9.907941579818726
    },
    {
      "epoch": 3.264617919921875e-05,
      "model_forward_time": 3.4216361045837402,
      "step": 8558
    },
    {
      "epoch": 3.264617919921875e-05,
      "step": 8558,
      "training_step_time": 10.191811561584473
    },
    {
      "epoch": 3.264617919921875e-05,
      "model_forward_time": 3.139761447906494,
      "step": 8558
    },
    {
      "epoch": 3.264617919921875e-05,
      "step": 8558,
      "training_step_time": 10.268294095993042
    },
    {
      "epoch": 3.264999389648437e-05,
      "model_forward_time": 3.155156373977661,
      "step": 8559
    },
    {
      "epoch": 3.264999389648437e-05,
      "step": 8559,
      "training_step_time": 9.930832386016846
    },
    {
      "epoch": 3.264999389648437e-05,
      "model_forward_time": 3.1510860919952393,
      "step": 8559
    },
    {
      "epoch": 3.264999389648437e-05,
      "step": 8559,
      "training_step_time": 9.929394245147705
    },
    {
      "epoch": 3.264999389648437e-05,
      "model_forward_time": 3.1493327617645264,
      "step": 8559
    },
    {
      "epoch": 3.264999389648437e-05,
      "step": 8559,
      "training_step_time": 9.932085275650024
    },
    {
      "epoch": 3.264999389648437e-05,
      "model_forward_time": 2.7176108360290527,
      "step": 8559
    },
    {
      "epoch": 3.264999389648437e-05,
      "step": 8559,
      "training_step_time": 10.305734872817993
    },
    {
      "epoch": 3.265380859375e-05,
      "grad_norm": 0.45270562171936035,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 8560
    },
    {
      "epoch": 3.265380859375e-05,
      "model_forward_time": 3.1700563430786133,
      "step": 8560
    },
    {
      "dynamics_loss_avg": 0.05506820026785135,
      "epoch": 3.265380859375e-05,
      "step": 8560
    },
    {
      "action_loss_avg": 0.02232725014910102,
      "epoch": 3.265380859375e-05,
      "step": 8560
    },
    {
      "epoch": 3.265380859375e-05,
      "step": 8560,
      "training_step_time": 9.950183153152466
    },
    {
      "epoch": 3.265380859375e-05,
      "model_forward_time": 3.1488001346588135,
      "step": 8560
    },
    {
      "epoch": 3.265380859375e-05,
      "step": 8560,
      "training_step_time": 9.921799659729004
    },
    {
      "epoch": 3.265380859375e-05,
      "model_forward_time": 3.1451785564422607,
      "step": 8560
    },
    {
      "epoch": 3.265380859375e-05,
      "step": 8560,
      "training_step_time": 9.919315099716187
    },
    {
      "epoch": 3.265380859375e-05,
      "model_forward_time": 3.1460344791412354,
      "step": 8560
    },
    {
      "epoch": 3.265380859375e-05,
      "step": 8560,
      "training_step_time": 10.306765794754028
    },
    {
      "epoch": 3.2657623291015625e-05,
      "model_forward_time": 3.409552812576294,
      "step": 8561
    },
    {
      "epoch": 3.2657623291015625e-05,
      "step": 8561,
      "training_step_time": 10.178863286972046
    },
    {
      "epoch": 3.2657623291015625e-05,
      "model_forward_time": 3.1418650150299072,
      "step": 8561
    },
    {
      "epoch": 3.2657623291015625e-05,
      "step": 8561,
      "training_step_time": 9.912307500839233
    },
    {
      "epoch": 3.2657623291015625e-05,
      "model_forward_time": 2.430978298187256,
      "step": 8561
    },
    {
      "epoch": 3.2657623291015625e-05,
      "step": 8561,
      "training_step_time": 7.45622992515564
    },
    {
      "epoch": 3.2657623291015625e-05,
      "model_forward_time": 2.433680534362793,
      "step": 8561
    },
    {
      "epoch": 3.2657623291015625e-05,
      "step": 8561,
      "training_step_time": 10.317028760910034
    },
    {
      "epoch": 3.266143798828125e-05,
      "model_forward_time": 2.445577621459961,
      "step": 8562
    },
    {
      "epoch": 3.266143798828125e-05,
      "step": 8562,
      "training_step_time": 7.47456693649292
    },
    {
      "epoch": 3.266143798828125e-05,
      "model_forward_time": 2.4330039024353027,
      "step": 8562
    },
    {
      "epoch": 3.266143798828125e-05,
      "step": 8562,
      "training_step_time": 7.461443901062012
    },
    {
      "epoch": 3.266143798828125e-05,
      "model_forward_time": 3.4215540885925293,
      "step": 8562
    },
    {
      "dynamics_loss_avg": 0.05283960029482841,
      "epoch": 3.266143798828125e-05,
      "step": 8562
    },
    {
      "action_loss_avg": 0.02012638719752431,
      "epoch": 3.266143798828125e-05,
      "step": 8562
    },
    {
      "epoch": 3.266143798828125e-05,
      "step": 8562,
      "training_step_time": 10.197800874710083
    },
    {
      "epoch": 3.266143798828125e-05,
      "model_forward_time": 3.1406760215759277,
      "step": 8562
    },
    {
      "epoch": 3.266143798828125e-05,
      "step": 8562,
      "training_step_time": 10.026498317718506
    },
    {
      "epoch": 3.266525268554688e-05,
      "model_forward_time": 2.4496731758117676,
      "step": 8563
    },
    {
      "epoch": 3.266525268554688e-05,
      "step": 8563,
      "training_step_time": 7.4806907176971436
    },
    {
      "epoch": 3.266525268554688e-05,
      "model_forward_time": 2.435452461242676,
      "step": 8563
    },
    {
      "epoch": 3.266525268554688e-05,
      "step": 8563,
      "training_step_time": 7.462588548660278
    },
    {
      "epoch": 3.266525268554688e-05,
      "model_forward_time": 3.136791706085205,
      "step": 8563
    },
    {
      "epoch": 3.266525268554688e-05,
      "step": 8563,
      "training_step_time": 9.905081748962402
    },
    {
      "epoch": 3.266525268554688e-05,
      "model_forward_time": 2.732719659805298,
      "step": 8563
    },
    {
      "epoch": 3.266525268554688e-05,
      "step": 8563,
      "training_step_time": 10.302578926086426
    },
    {
      "epoch": 3.26690673828125e-05,
      "model_forward_time": 3.1524851322174072,
      "step": 8564
    },
    {
      "epoch": 3.26690673828125e-05,
      "step": 8564,
      "training_step_time": 9.922040224075317
    },
    {
      "epoch": 3.26690673828125e-05,
      "model_forward_time": 2.431952953338623,
      "step": 8564
    },
    {
      "epoch": 3.26690673828125e-05,
      "step": 8564,
      "training_step_time": 7.452943325042725
    },
    {
      "epoch": 3.26690673828125e-05,
      "model_forward_time": 3.135057210922241,
      "step": 8564
    },
    {
      "epoch": 3.26690673828125e-05,
      "step": 8564,
      "training_step_time": 9.90060806274414
    },
    {
      "epoch": 3.26690673828125e-05,
      "model_forward_time": 2.431334972381592,
      "step": 8564
    },
    {
      "epoch": 3.26690673828125e-05,
      "step": 8564,
      "training_step_time": 10.22589635848999
    },
    {
      "epoch": 3.267288208007812e-05,
      "model_forward_time": 3.1441140174865723,
      "step": 8565
    },
    {
      "dynamics_loss_avg": 0.0676220901310444,
      "epoch": 3.267288208007812e-05,
      "step": 8565
    },
    {
      "action_loss_avg": 0.030220097955316307,
      "epoch": 3.267288208007812e-05,
      "step": 8565
    },
    {
      "epoch": 3.267288208007812e-05,
      "step": 8565,
      "training_step_time": 10.196674346923828
    },
    {
      "epoch": 3.267288208007812e-05,
      "model_forward_time": 2.4336373805999756,
      "step": 8565
    },
    {
      "epoch": 3.267288208007812e-05,
      "step": 8565,
      "training_step_time": 7.455640077590942
    },
    {
      "epoch": 3.267288208007812e-05,
      "model_forward_time": 2.4312217235565186,
      "step": 8565
    },
    {
      "epoch": 3.267288208007812e-05,
      "step": 8565,
      "training_step_time": 7.455075740814209
    },
    {
      "epoch": 3.267288208007812e-05,
      "model_forward_time": 3.136631965637207,
      "step": 8565
    },
    {
      "epoch": 3.267288208007812e-05,
      "step": 8565,
      "training_step_time": 10.306051015853882
    },
    {
      "epoch": 3.267669677734375e-05,
      "model_forward_time": 2.448704957962036,
      "step": 8566
    },
    {
      "epoch": 3.267669677734375e-05,
      "step": 8566,
      "training_step_time": 7.47739577293396
    },
    {
      "epoch": 3.267669677734375e-05,
      "model_forward_time": 2.4342710971832275,
      "step": 8566
    },
    {
      "epoch": 3.267669677734375e-05,
      "step": 8566,
      "training_step_time": 7.473572492599487
    },
    {
      "epoch": 3.267669677734375e-05,
      "model_forward_time": 3.415137529373169,
      "step": 8566
    },
    {
      "epoch": 3.267669677734375e-05,
      "step": 8566,
      "training_step_time": 10.18697738647461
    },
    {
      "epoch": 3.267669677734375e-05,
      "model_forward_time": 2.4364209175109863,
      "step": 8566
    },
    {
      "epoch": 3.267669677734375e-05,
      "step": 8566,
      "training_step_time": 10.296887397766113
    },
    {
      "epoch": 3.2680511474609376e-05,
      "model_forward_time": 3.1429853439331055,
      "step": 8567
    },
    {
      "epoch": 3.2680511474609376e-05,
      "step": 8567,
      "training_step_time": 9.919083595275879
    },
    {
      "epoch": 3.2680511474609376e-05,
      "model_forward_time": 3.136486053466797,
      "step": 8567
    },
    {
      "epoch": 3.2680511474609376e-05,
      "step": 8567,
      "training_step_time": 9.910704612731934
    },
    {
      "epoch": 3.2680511474609376e-05,
      "model_forward_time": 2.433544874191284,
      "step": 8567
    },
    {
      "dynamics_loss_avg": 0.05899505484849214,
      "epoch": 3.2680511474609376e-05,
      "step": 8567
    },
    {
      "action_loss_avg": 0.024100185092538595,
      "epoch": 3.2680511474609376e-05,
      "step": 8567
    },
    {
      "epoch": 3.2680511474609376e-05,
      "step": 8567,
      "training_step_time": 7.46652626991272
    },
    {
      "epoch": 3.2680511474609376e-05,
      "model_forward_time": 3.3964502811431885,
      "step": 8567
    },
    {
      "epoch": 3.2680511474609376e-05,
      "step": 8567,
      "training_step_time": 10.267547845840454
    },
    {
      "epoch": 3.2684326171875e-05,
      "model_forward_time": 2.4500885009765625,
      "step": 8568
    },
    {
      "epoch": 3.2684326171875e-05,
      "step": 8568,
      "training_step_time": 7.47910475730896
    },
    {
      "epoch": 3.2684326171875e-05,
      "model_forward_time": 3.1365554332733154,
      "step": 8568
    },
    {
      "epoch": 3.2684326171875e-05,
      "step": 8568,
      "training_step_time": 9.905985832214355
    },
    {
      "epoch": 3.2684326171875e-05,
      "model_forward_time": 3.134951591491699,
      "step": 8568
    },
    {
      "epoch": 3.2684326171875e-05,
      "step": 8568,
      "training_step_time": 9.903546810150146
    },
    {
      "epoch": 3.2684326171875e-05,
      "model_forward_time": 2.430379867553711,
      "step": 8568
    },
    {
      "epoch": 3.2684326171875e-05,
      "step": 8568,
      "training_step_time": 10.320676803588867
    },
    {
      "epoch": 3.268814086914063e-05,
      "model_forward_time": 3.41074538230896,
      "step": 8569
    },
    {
      "epoch": 3.268814086914063e-05,
      "step": 8569,
      "training_step_time": 10.189102411270142
    },
    {
      "epoch": 3.268814086914063e-05,
      "model_forward_time": 2.4323575496673584,
      "step": 8569
    },
    {
      "epoch": 3.268814086914063e-05,
      "step": 8569,
      "training_step_time": 7.457180738449097
    },
    {
      "epoch": 3.268814086914063e-05,
      "model_forward_time": 2.4318392276763916,
      "step": 8569
    },
    {
      "epoch": 3.268814086914063e-05,
      "step": 8569,
      "training_step_time": 7.4540863037109375
    },
    {
      "epoch": 3.268814086914063e-05,
      "model_forward_time": 2.4332640171051025,
      "step": 8569
    },
    {
      "epoch": 3.268814086914063e-05,
      "step": 8569,
      "training_step_time": 10.264795303344727
    },
    {
      "epoch": 3.269195556640625e-05,
      "grad_norm": 0.46424955129623413,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 8570
    },
    {
      "epoch": 3.269195556640625e-05,
      "model_forward_time": 3.15261173248291,
      "step": 8570
    },
    {
      "dynamics_loss_avg": 0.041811653040349485,
      "epoch": 3.269195556640625e-05,
      "step": 8570
    },
    {
      "action_loss_avg": 0.01874101855792105,
      "epoch": 3.269195556640625e-05,
      "step": 8570
    },
    {
      "epoch": 3.269195556640625e-05,
      "step": 8570,
      "training_step_time": 9.925129175186157
    },
    {
      "epoch": 3.269195556640625e-05,
      "model_forward_time": 2.430631637573242,
      "step": 8570
    },
    {
      "epoch": 3.269195556640625e-05,
      "step": 8570,
      "training_step_time": 7.452868938446045
    },
    {
      "epoch": 3.269195556640625e-05,
      "model_forward_time": 2.735095977783203,
      "step": 8570
    },
    {
      "epoch": 3.269195556640625e-05,
      "step": 8570,
      "training_step_time": 7.756576776504517
    },
    {
      "epoch": 3.269195556640625e-05,
      "model_forward_time": 2.4267141819000244,
      "step": 8570
    },
    {
      "epoch": 3.269195556640625e-05,
      "step": 8570,
      "training_step_time": 10.286554098129272
    },
    {
      "epoch": 3.2695770263671874e-05,
      "model_forward_time": 2.444237232208252,
      "step": 8571
    },
    {
      "epoch": 3.2695770263671874e-05,
      "step": 8571,
      "training_step_time": 7.46638298034668
    },
    {
      "epoch": 3.2695770263671874e-05,
      "model_forward_time": 3.1363096237182617,
      "step": 8571
    },
    {
      "epoch": 3.2695770263671874e-05,
      "step": 8571,
      "training_step_time": 9.902632474899292
    },
    {
      "epoch": 3.2695770263671874e-05,
      "model_forward_time": 3.1387784481048584,
      "step": 8571
    },
    {
      "epoch": 3.2695770263671874e-05,
      "step": 8571,
      "training_step_time": 9.906703472137451
    },
    {
      "epoch": 3.2695770263671874e-05,
      "model_forward_time": 3.426072835922241,
      "step": 8571
    },
    {
      "epoch": 3.2695770263671874e-05,
      "step": 8571,
      "training_step_time": 10.326070070266724
    },
    {
      "epoch": 3.26995849609375e-05,
      "model_forward_time": 3.155747175216675,
      "step": 8572
    },
    {
      "epoch": 3.26995849609375e-05,
      "step": 8572,
      "training_step_time": 9.93227767944336
    },
    {
      "epoch": 3.26995849609375e-05,
      "model_forward_time": 3.136620283126831,
      "step": 8572
    },
    {
      "epoch": 3.26995849609375e-05,
      "step": 8572,
      "training_step_time": 9.908386707305908
    },
    {
      "epoch": 3.26995849609375e-05,
      "model_forward_time": 3.1383655071258545,
      "step": 8572
    },
    {
      "dynamics_loss_avg": 0.06231448221951723,
      "epoch": 3.26995849609375e-05,
      "step": 8572
    },
    {
      "action_loss_avg": 0.023834061902016403,
      "epoch": 3.26995849609375e-05,
      "step": 8572
    },
    {
      "epoch": 3.26995849609375e-05,
      "step": 8572,
      "training_step_time": 9.914817810058594
    },
    {
      "epoch": 3.26995849609375e-05,
      "model_forward_time": 3.1401381492614746,
      "step": 8572
    },
    {
      "epoch": 3.26995849609375e-05,
      "step": 8572,
      "training_step_time": 10.286678075790405
    },
    {
      "epoch": 3.2703399658203127e-05,
      "model_forward_time": 2.7355270385742188,
      "step": 8573
    },
    {
      "epoch": 3.2703399658203127e-05,
      "step": 8573,
      "training_step_time": 7.773947477340698
    },
    {
      "epoch": 3.2703399658203127e-05,
      "model_forward_time": 2.4314682483673096,
      "step": 8573
    },
    {
      "epoch": 3.2703399658203127e-05,
      "step": 8573,
      "training_step_time": 7.480523109436035
    },
    {
      "epoch": 3.2703399658203127e-05,
      "model_forward_time": 2.4362025260925293,
      "step": 8573
    },
    {
      "epoch": 3.2703399658203127e-05,
      "step": 8573,
      "training_step_time": 7.468492031097412
    },
    {
      "epoch": 3.2703399658203127e-05,
      "model_forward_time": 3.136073350906372,
      "step": 8573
    },
    {
      "epoch": 3.2703399658203127e-05,
      "step": 8573,
      "training_step_time": 10.26224684715271
    },
    {
      "epoch": 3.270721435546875e-05,
      "model_forward_time": 3.149663209915161,
      "step": 8574
    },
    {
      "epoch": 3.270721435546875e-05,
      "step": 8574,
      "training_step_time": 9.940381526947021
    },
    {
      "epoch": 3.270721435546875e-05,
      "model_forward_time": 2.4331088066101074,
      "step": 8574
    },
    {
      "epoch": 3.270721435546875e-05,
      "step": 8574,
      "training_step_time": 7.781307220458984
    },
    {
      "epoch": 3.270721435546875e-05,
      "model_forward_time": 2.431023597717285,
      "step": 8574
    },
    {
      "epoch": 3.270721435546875e-05,
      "step": 8574,
      "training_step_time": 7.461724281311035
    },
    {
      "epoch": 3.270721435546875e-05,
      "model_forward_time": 3.1372625827789307,
      "step": 8574
    },
    {
      "epoch": 3.270721435546875e-05,
      "step": 8574,
      "training_step_time": 10.30053162574768
    },
    {
      "epoch": 3.271102905273437e-05,
      "model_forward_time": 2.453634262084961,
      "step": 8575
    },
    {
      "dynamics_loss_avg": 0.044872136879712345,
      "epoch": 3.271102905273437e-05,
      "step": 8575
    },
    {
      "action_loss_avg": 0.01433262825012207,
      "epoch": 3.271102905273437e-05,
      "step": 8575
    },
    {
      "epoch": 3.271102905273437e-05,
      "step": 8575,
      "training_step_time": 7.494640350341797
    },
    {
      "epoch": 3.271102905273437e-05,
      "model_forward_time": 2.4310638904571533,
      "step": 8575
    },
    {
      "epoch": 3.271102905273437e-05,
      "step": 8575,
      "training_step_time": 7.45747447013855
    },
    {
      "epoch": 3.271102905273437e-05,
      "model_forward_time": 2.4289064407348633,
      "step": 8575
    },
    {
      "epoch": 3.271102905273437e-05,
      "step": 8575,
      "training_step_time": 7.454651594161987
    },
    {
      "epoch": 3.271102905273437e-05,
      "model_forward_time": 2.71793270111084,
      "step": 8575
    },
    {
      "epoch": 3.271102905273437e-05,
      "step": 8575,
      "training_step_time": 10.29589033126831
    },
    {
      "epoch": 3.271484375e-05,
      "model_forward_time": 3.1533660888671875,
      "step": 8576
    },
    {
      "epoch": 3.271484375e-05,
      "step": 8576,
      "training_step_time": 9.934269905090332
    },
    {
      "epoch": 3.271484375e-05,
      "model_forward_time": 3.136258125305176,
      "step": 8576
    },
    {
      "epoch": 3.271484375e-05,
      "step": 8576,
      "training_step_time": 9.910921573638916
    },
    {
      "epoch": 3.271484375e-05,
      "model_forward_time": 3.1377131938934326,
      "step": 8576
    },
    {
      "epoch": 3.271484375e-05,
      "step": 8576,
      "training_step_time": 9.919514656066895
    },
    {
      "epoch": 3.271484375e-05,
      "model_forward_time": 2.43804931640625,
      "step": 8576
    },
    {
      "epoch": 3.271484375e-05,
      "step": 8576,
      "training_step_time": 10.255719184875488
    },
    {
      "epoch": 3.2718658447265625e-05,
      "model_forward_time": 2.7951695919036865,
      "step": 8577
    },
    {
      "epoch": 3.2718658447265625e-05,
      "step": 8577,
      "training_step_time": 7.829495906829834
    },
    {
      "epoch": 3.2718658447265625e-05,
      "model_forward_time": 3.1409752368927,
      "step": 8577
    },
    {
      "epoch": 3.2718658447265625e-05,
      "step": 8577,
      "training_step_time": 9.915563583374023
    },
    {
      "epoch": 3.2718658447265625e-05,
      "model_forward_time": 3.143239974975586,
      "step": 8577
    },
    {
      "dynamics_loss_avg": 0.061436860077083114,
      "epoch": 3.2718658447265625e-05,
      "step": 8577
    },
    {
      "action_loss_avg": 0.025472506415098906,
      "epoch": 3.2718658447265625e-05,
      "step": 8577
    },
    {
      "epoch": 3.2718658447265625e-05,
      "step": 8577,
      "training_step_time": 9.922627449035645
    },
    {
      "epoch": 3.2718658447265625e-05,
      "model_forward_time": 2.440580368041992,
      "step": 8577
    },
    {
      "epoch": 3.2718658447265625e-05,
      "step": 8577,
      "training_step_time": 10.294861793518066
    },
    {
      "epoch": 3.272247314453125e-05,
      "model_forward_time": 2.4536898136138916,
      "step": 8578
    },
    {
      "epoch": 3.272247314453125e-05,
      "step": 8578,
      "training_step_time": 7.492324590682983
    },
    {
      "epoch": 3.272247314453125e-05,
      "model_forward_time": 3.4463298320770264,
      "step": 8578
    },
    {
      "epoch": 3.272247314453125e-05,
      "step": 8578,
      "training_step_time": 10.223361730575562
    },
    {
      "epoch": 3.272247314453125e-05,
      "model_forward_time": 2.4418246746063232,
      "step": 8578
    },
    {
      "epoch": 3.272247314453125e-05,
      "step": 8578,
      "training_step_time": 7.4731764793396
    },
    {
      "epoch": 3.272247314453125e-05,
      "model_forward_time": 3.1416454315185547,
      "step": 8578
    },
    {
      "epoch": 3.272247314453125e-05,
      "step": 8578,
      "training_step_time": 10.021166801452637
    },
    {
      "epoch": 3.272628784179688e-05,
      "model_forward_time": 2.444581985473633,
      "step": 8579
    },
    {
      "epoch": 3.272628784179688e-05,
      "step": 8579,
      "training_step_time": 7.487088203430176
    },
    {
      "epoch": 3.272628784179688e-05,
      "model_forward_time": 3.1286840438842773,
      "step": 8579
    },
    {
      "epoch": 3.272628784179688e-05,
      "step": 8579,
      "training_step_time": 9.905052900314331
    },
    {
      "epoch": 3.272628784179688e-05,
      "model_forward_time": 3.1370606422424316,
      "step": 8579
    },
    {
      "epoch": 3.272628784179688e-05,
      "step": 8579,
      "training_step_time": 10.234727621078491
    },
    {
      "epoch": 3.272628784179688e-05,
      "model_forward_time": 3.1358354091644287,
      "step": 8579
    },
    {
      "epoch": 3.272628784179688e-05,
      "step": 8579,
      "training_step_time": 10.01169490814209
    },
    {
      "epoch": 3.27301025390625e-05,
      "grad_norm": 0.33237266540527344,
      "learning_rate": 1e-05,
      "loss": 0.0815,
      "step": 8580
    },
    {
      "epoch": 3.27301025390625e-05,
      "model_forward_time": 2.4480302333831787,
      "step": 8580
    },
    {
      "dynamics_loss_avg": 0.06210833117365837,
      "epoch": 3.27301025390625e-05,
      "step": 8580
    },
    {
      "action_loss_avg": 0.023658674908801913,
      "epoch": 3.27301025390625e-05,
      "step": 8580
    },
    {
      "epoch": 3.27301025390625e-05,
      "step": 8580,
      "training_step_time": 7.490058660507202
    },
    {
      "epoch": 3.27301025390625e-05,
      "model_forward_time": 3.1327664852142334,
      "step": 8580
    },
    {
      "epoch": 3.27301025390625e-05,
      "step": 8580,
      "training_step_time": 9.912048816680908
    },
    {
      "epoch": 3.27301025390625e-05,
      "model_forward_time": 2.4276528358459473,
      "step": 8580
    },
    {
      "epoch": 3.27301025390625e-05,
      "step": 8580,
      "training_step_time": 7.459532260894775
    },
    {
      "epoch": 3.27301025390625e-05,
      "model_forward_time": 2.4293079376220703,
      "step": 8580
    },
    {
      "epoch": 3.27301025390625e-05,
      "step": 8580,
      "training_step_time": 10.26059079170227
    },
    {
      "epoch": 3.273391723632812e-05,
      "model_forward_time": 3.469783067703247,
      "step": 8581
    },
    {
      "epoch": 3.273391723632812e-05,
      "step": 8581,
      "training_step_time": 10.257650375366211
    },
    {
      "epoch": 3.273391723632812e-05,
      "model_forward_time": 3.1325864791870117,
      "step": 8581
    },
    {
      "epoch": 3.273391723632812e-05,
      "step": 8581,
      "training_step_time": 9.916586637496948
    },
    {
      "epoch": 3.273391723632812e-05,
      "model_forward_time": 2.4389445781707764,
      "step": 8581
    },
    {
      "epoch": 3.273391723632812e-05,
      "step": 8581,
      "training_step_time": 7.477934837341309
    },
    {
      "epoch": 3.273391723632812e-05,
      "model_forward_time": 2.4310595989227295,
      "step": 8581
    },
    {
      "epoch": 3.273391723632812e-05,
      "step": 8581,
      "training_step_time": 10.028957843780518
    },
    {
      "epoch": 3.273773193359375e-05,
      "model_forward_time": 2.449537754058838,
      "step": 8582
    },
    {
      "epoch": 3.273773193359375e-05,
      "step": 8582,
      "training_step_time": 7.481460332870483
    },
    {
      "epoch": 3.273773193359375e-05,
      "model_forward_time": 3.396605968475342,
      "step": 8582
    },
    {
      "epoch": 3.273773193359375e-05,
      "step": 8582,
      "training_step_time": 10.17374300956726
    },
    {
      "epoch": 3.273773193359375e-05,
      "model_forward_time": 3.133519172668457,
      "step": 8582
    },
    {
      "dynamics_loss_avg": 0.05833643153309822,
      "epoch": 3.273773193359375e-05,
      "step": 8582
    },
    {
      "action_loss_avg": 0.019465768802911045,
      "epoch": 3.273773193359375e-05,
      "step": 8582
    },
    {
      "epoch": 3.273773193359375e-05,
      "step": 8582,
      "training_step_time": 9.907841444015503
    },
    {
      "epoch": 3.273773193359375e-05,
      "model_forward_time": 3.125859022140503,
      "step": 8582
    },
    {
      "epoch": 3.273773193359375e-05,
      "step": 8582,
      "training_step_time": 10.30033564567566
    },
    {
      "epoch": 3.2741546630859376e-05,
      "model_forward_time": 2.4422521591186523,
      "step": 8583
    },
    {
      "epoch": 3.2741546630859376e-05,
      "step": 8583,
      "training_step_time": 7.480092287063599
    },
    {
      "epoch": 3.2741546630859376e-05,
      "model_forward_time": 3.1328811645507812,
      "step": 8583
    },
    {
      "epoch": 3.2741546630859376e-05,
      "step": 8583,
      "training_step_time": 9.908854484558105
    },
    {
      "epoch": 3.2741546630859376e-05,
      "model_forward_time": 2.7335867881774902,
      "step": 8583
    },
    {
      "epoch": 3.2741546630859376e-05,
      "step": 8583,
      "training_step_time": 7.7656450271606445
    },
    {
      "epoch": 3.2741546630859376e-05,
      "model_forward_time": 2.42964243888855,
      "step": 8583
    },
    {
      "epoch": 3.2741546630859376e-05,
      "step": 8583,
      "training_step_time": 10.302443981170654
    },
    {
      "epoch": 3.2745361328125e-05,
      "model_forward_time": 2.441035509109497,
      "step": 8584
    },
    {
      "epoch": 3.2745361328125e-05,
      "step": 8584,
      "training_step_time": 7.474687099456787
    },
    {
      "epoch": 3.2745361328125e-05,
      "model_forward_time": 2.4316420555114746,
      "step": 8584
    },
    {
      "epoch": 3.2745361328125e-05,
      "step": 8584,
      "training_step_time": 7.4631922245025635
    },
    {
      "epoch": 3.2745361328125e-05,
      "model_forward_time": 3.1278600692749023,
      "step": 8584
    },
    {
      "epoch": 3.2745361328125e-05,
      "step": 8584,
      "training_step_time": 9.904587984085083
    },
    {
      "epoch": 3.2745361328125e-05,
      "model_forward_time": 3.1353259086608887,
      "step": 8584
    },
    {
      "epoch": 3.2745361328125e-05,
      "step": 8584,
      "training_step_time": 10.290063619613647
    },
    {
      "epoch": 3.274917602539063e-05,
      "model_forward_time": 2.447838068008423,
      "step": 8585
    },
    {
      "dynamics_loss_avg": 0.062307149171829224,
      "epoch": 3.274917602539063e-05,
      "step": 8585
    },
    {
      "action_loss_avg": 0.01936897253617644,
      "epoch": 3.274917602539063e-05,
      "step": 8585
    },
    {
      "epoch": 3.274917602539063e-05,
      "step": 8585,
      "training_step_time": 7.486648797988892
    },
    {
      "epoch": 3.274917602539063e-05,
      "model_forward_time": 3.134934663772583,
      "step": 8585
    },
    {
      "epoch": 3.274917602539063e-05,
      "step": 8585,
      "training_step_time": 9.912991762161255
    },
    {
      "epoch": 3.274917602539063e-05,
      "model_forward_time": 3.1310338973999023,
      "step": 8585
    },
    {
      "epoch": 3.274917602539063e-05,
      "step": 8585,
      "training_step_time": 9.91015076637268
    },
    {
      "epoch": 3.274917602539063e-05,
      "model_forward_time": 2.429273843765259,
      "step": 8585
    },
    {
      "epoch": 3.274917602539063e-05,
      "step": 8585,
      "training_step_time": 10.29787015914917
    },
    {
      "epoch": 3.275299072265625e-05,
      "model_forward_time": 3.1482009887695312,
      "step": 8586
    },
    {
      "epoch": 3.275299072265625e-05,
      "step": 8586,
      "training_step_time": 9.935234785079956
    },
    {
      "epoch": 3.275299072265625e-05,
      "model_forward_time": 2.7045483589172363,
      "step": 8586
    },
    {
      "epoch": 3.275299072265625e-05,
      "step": 8586,
      "training_step_time": 7.7332141399383545
    },
    {
      "epoch": 3.275299072265625e-05,
      "model_forward_time": 3.127840995788574,
      "step": 8586
    },
    {
      "epoch": 3.275299072265625e-05,
      "step": 8586,
      "training_step_time": 9.900313377380371
    },
    {
      "epoch": 3.275299072265625e-05,
      "model_forward_time": 3.132448434829712,
      "step": 8586
    },
    {
      "epoch": 3.275299072265625e-05,
      "step": 8586,
      "training_step_time": 10.29701018333435
    },
    {
      "epoch": 3.2756805419921874e-05,
      "model_forward_time": 3.1464738845825195,
      "step": 8587
    },
    {
      "epoch": 3.2756805419921874e-05,
      "step": 8587,
      "training_step_time": 9.926427125930786
    },
    {
      "epoch": 3.2756805419921874e-05,
      "model_forward_time": 2.4346799850463867,
      "step": 8587
    },
    {
      "epoch": 3.2756805419921874e-05,
      "step": 8587,
      "training_step_time": 7.470364332199097
    },
    {
      "epoch": 3.2756805419921874e-05,
      "model_forward_time": 2.702688694000244,
      "step": 8587
    },
    {
      "dynamics_loss_avg": 0.049633072875440123,
      "epoch": 3.2756805419921874e-05,
      "step": 8587
    },
    {
      "action_loss_avg": 0.020280035585165022,
      "epoch": 3.2756805419921874e-05,
      "step": 8587
    },
    {
      "epoch": 3.2756805419921874e-05,
      "step": 8587,
      "training_step_time": 7.738037347793579
    },
    {
      "epoch": 3.2756805419921874e-05,
      "model_forward_time": 3.1331865787506104,
      "step": 8587
    },
    {
      "epoch": 3.2756805419921874e-05,
      "step": 8587,
      "training_step_time": 10.280652284622192
    },
    {
      "epoch": 3.27606201171875e-05,
      "model_forward_time": 2.4411373138427734,
      "step": 8588
    },
    {
      "epoch": 3.27606201171875e-05,
      "step": 8588,
      "training_step_time": 7.4726173877716064
    },
    {
      "epoch": 3.27606201171875e-05,
      "model_forward_time": 2.4332664012908936,
      "step": 8588
    },
    {
      "epoch": 3.27606201171875e-05,
      "step": 8588,
      "training_step_time": 7.462059497833252
    },
    {
      "epoch": 3.27606201171875e-05,
      "model_forward_time": 3.1350769996643066,
      "step": 8588
    },
    {
      "epoch": 3.27606201171875e-05,
      "step": 8588,
      "training_step_time": 9.909285306930542
    },
    {
      "epoch": 3.27606201171875e-05,
      "model_forward_time": 3.4217443466186523,
      "step": 8588
    },
    {
      "epoch": 3.27606201171875e-05,
      "step": 8588,
      "training_step_time": 10.2957124710083
    },
    {
      "epoch": 3.2764434814453127e-05,
      "model_forward_time": 3.150139570236206,
      "step": 8589
    },
    {
      "epoch": 3.2764434814453127e-05,
      "step": 8589,
      "training_step_time": 9.929230213165283
    },
    {
      "epoch": 3.2764434814453127e-05,
      "model_forward_time": 3.1330068111419678,
      "step": 8589
    },
    {
      "epoch": 3.2764434814453127e-05,
      "step": 8589,
      "training_step_time": 9.908219337463379
    },
    {
      "epoch": 3.2764434814453127e-05,
      "model_forward_time": 3.138747453689575,
      "step": 8589
    },
    {
      "epoch": 3.2764434814453127e-05,
      "step": 8589,
      "training_step_time": 9.918940544128418
    },
    {
      "epoch": 3.2764434814453127e-05,
      "model_forward_time": 2.4290051460266113,
      "step": 8589
    },
    {
      "epoch": 3.2764434814453127e-05,
      "step": 8589,
      "training_step_time": 10.33966588973999
    },
    {
      "epoch": 3.276824951171875e-05,
      "grad_norm": 0.46153298020362854,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 8590
    },
    {
      "epoch": 3.276824951171875e-05,
      "model_forward_time": 3.4667470455169678,
      "step": 8590
    },
    {
      "dynamics_loss_avg": 0.04759326782077551,
      "epoch": 3.276824951171875e-05,
      "step": 8590
    },
    {
      "action_loss_avg": 0.019928963971324265,
      "epoch": 3.276824951171875e-05,
      "step": 8590
    },
    {
      "epoch": 3.276824951171875e-05,
      "step": 8590,
      "training_step_time": 10.256908893585205
    },
    {
      "epoch": 3.276824951171875e-05,
      "model_forward_time": 3.1362311840057373,
      "step": 8590
    },
    {
      "epoch": 3.276824951171875e-05,
      "step": 8590,
      "training_step_time": 9.915951251983643
    },
    {
      "epoch": 3.276824951171875e-05,
      "model_forward_time": 2.4359171390533447,
      "step": 8590
    },
    {
      "epoch": 3.276824951171875e-05,
      "step": 8590,
      "training_step_time": 7.471243619918823
    },
    {
      "epoch": 3.276824951171875e-05,
      "model_forward_time": 2.4334187507629395,
      "step": 8590
    },
    {
      "epoch": 3.276824951171875e-05,
      "step": 8590,
      "training_step_time": 10.262885332107544
    },
    {
      "epoch": 3.277206420898437e-05,
      "model_forward_time": 3.147068977355957,
      "step": 8591
    },
    {
      "epoch": 3.277206420898437e-05,
      "step": 8591,
      "training_step_time": 9.93180227279663
    },
    {
      "epoch": 3.277206420898437e-05,
      "model_forward_time": 2.7245914936065674,
      "step": 8591
    },
    {
      "epoch": 3.277206420898437e-05,
      "step": 8591,
      "training_step_time": 7.763019561767578
    },
    {
      "epoch": 3.277206420898437e-05,
      "model_forward_time": 3.1421735286712646,
      "step": 8591
    },
    {
      "epoch": 3.277206420898437e-05,
      "step": 8591,
      "training_step_time": 9.929054260253906
    },
    {
      "epoch": 3.277206420898437e-05,
      "model_forward_time": 2.438380241394043,
      "step": 8591
    },
    {
      "epoch": 3.277206420898437e-05,
      "step": 8591,
      "training_step_time": 10.299213647842407
    },
    {
      "epoch": 3.277587890625e-05,
      "model_forward_time": 2.4477951526641846,
      "step": 8592
    },
    {
      "epoch": 3.277587890625e-05,
      "step": 8592,
      "training_step_time": 7.48820424079895
    },
    {
      "epoch": 3.277587890625e-05,
      "model_forward_time": 3.1425459384918213,
      "step": 8592
    },
    {
      "epoch": 3.277587890625e-05,
      "step": 8592,
      "training_step_time": 9.934495449066162
    },
    {
      "epoch": 3.277587890625e-05,
      "model_forward_time": 2.4347078800201416,
      "step": 8592
    },
    {
      "dynamics_loss_avg": 0.057119563408195975,
      "epoch": 3.277587890625e-05,
      "step": 8592
    },
    {
      "action_loss_avg": 0.01911474554799497,
      "epoch": 3.277587890625e-05,
      "step": 8592
    },
    {
      "epoch": 3.277587890625e-05,
      "step": 8592,
      "training_step_time": 7.4690916538238525
    },
    {
      "epoch": 3.277587890625e-05,
      "model_forward_time": 3.4373674392700195,
      "step": 8592
    },
    {
      "epoch": 3.277587890625e-05,
      "step": 8592,
      "training_step_time": 10.310524463653564
    },
    {
      "epoch": 3.2779693603515625e-05,
      "model_forward_time": 1.764233112335205,
      "step": 8593
    },
    {
      "epoch": 3.2779693603515625e-05,
      "step": 8593,
      "training_step_time": 5.224797964096069
    },
    {
      "epoch": 3.2779693603515625e-05,
      "model_forward_time": 2.433715581893921,
      "step": 8593
    },
    {
      "epoch": 3.2779693603515625e-05,
      "step": 8593,
      "training_step_time": 7.4691078662872314
    },
    {
      "epoch": 3.2779693603515625e-05,
      "model_forward_time": 3.138559579849243,
      "step": 8593
    },
    {
      "epoch": 3.2779693603515625e-05,
      "step": 8593,
      "training_step_time": 9.915194988250732
    },
    {
      "epoch": 3.2779693603515625e-05,
      "model_forward_time": 3.137180805206299,
      "step": 8593
    },
    {
      "epoch": 3.2779693603515625e-05,
      "step": 8593,
      "training_step_time": 10.024261951446533
    },
    {
      "epoch": 3.278350830078125e-05,
      "model_forward_time": 3.430361270904541,
      "step": 8594
    },
    {
      "epoch": 3.278350830078125e-05,
      "step": 8594,
      "training_step_time": 10.218489170074463
    },
    {
      "epoch": 3.278350830078125e-05,
      "model_forward_time": 3.1408207416534424,
      "step": 8594
    },
    {
      "epoch": 3.278350830078125e-05,
      "step": 8594,
      "training_step_time": 9.916736364364624
    },
    {
      "epoch": 3.278350830078125e-05,
      "model_forward_time": 3.1358401775360107,
      "step": 8594
    },
    {
      "epoch": 3.278350830078125e-05,
      "step": 8594,
      "training_step_time": 9.91790509223938
    },
    {
      "epoch": 3.278350830078125e-05,
      "model_forward_time": 2.433335304260254,
      "step": 8594
    },
    {
      "epoch": 3.278350830078125e-05,
      "step": 8594,
      "training_step_time": 10.334525108337402
    },
    {
      "epoch": 3.278732299804688e-05,
      "model_forward_time": 1.7614805698394775,
      "step": 8595
    },
    {
      "dynamics_loss_avg": 0.048726920317858455,
      "epoch": 3.278732299804688e-05,
      "step": 8595
    },
    {
      "action_loss_avg": 0.01563249509781599,
      "epoch": 3.278732299804688e-05,
      "step": 8595
    },
    {
      "epoch": 3.278732299804688e-05,
      "step": 8595,
      "training_step_time": 5.225608825683594
    },
    {
      "epoch": 3.278732299804688e-05,
      "model_forward_time": 3.12831449508667,
      "step": 8595
    },
    {
      "epoch": 3.278732299804688e-05,
      "step": 8595,
      "training_step_time": 9.900195121765137
    },
    {
      "epoch": 3.278732299804688e-05,
      "model_forward_time": 3.4109628200531006,
      "step": 8595
    },
    {
      "epoch": 3.278732299804688e-05,
      "step": 8595,
      "training_step_time": 10.183641910552979
    },
    {
      "epoch": 3.278732299804688e-05,
      "model_forward_time": 2.4324352741241455,
      "step": 8595
    },
    {
      "epoch": 3.278732299804688e-05,
      "step": 8595,
      "training_step_time": 10.262389421463013
    },
    {
      "epoch": 3.27911376953125e-05,
      "model_forward_time": 3.1593503952026367,
      "step": 8596
    },
    {
      "epoch": 3.27911376953125e-05,
      "step": 8596,
      "training_step_time": 9.930910348892212
    },
    {
      "epoch": 3.27911376953125e-05,
      "model_forward_time": 2.425807237625122,
      "step": 8596
    },
    {
      "epoch": 3.27911376953125e-05,
      "step": 8596,
      "training_step_time": 7.452139139175415
    },
    {
      "epoch": 3.27911376953125e-05,
      "model_forward_time": 2.4307477474212646,
      "step": 8596
    },
    {
      "epoch": 3.27911376953125e-05,
      "step": 8596,
      "training_step_time": 7.4574878215789795
    },
    {
      "epoch": 3.27911376953125e-05,
      "model_forward_time": 3.450659990310669,
      "step": 8596
    },
    {
      "epoch": 3.27911376953125e-05,
      "step": 8596,
      "training_step_time": 10.320604085922241
    },
    {
      "epoch": 3.279495239257812e-05,
      "model_forward_time": 3.1455390453338623,
      "step": 8597
    },
    {
      "epoch": 3.279495239257812e-05,
      "step": 8597,
      "training_step_time": 9.921411514282227
    },
    {
      "epoch": 3.279495239257812e-05,
      "model_forward_time": 2.4340338706970215,
      "step": 8597
    },
    {
      "epoch": 3.279495239257812e-05,
      "step": 8597,
      "training_step_time": 7.464471340179443
    },
    {
      "epoch": 3.279495239257812e-05,
      "model_forward_time": 3.1354658603668213,
      "step": 8597
    },
    {
      "dynamics_loss_avg": 0.07431278415024281,
      "epoch": 3.279495239257812e-05,
      "step": 8597
    },
    {
      "action_loss_avg": 0.022780985850840808,
      "epoch": 3.279495239257812e-05,
      "step": 8597
    },
    {
      "epoch": 3.279495239257812e-05,
      "step": 8597,
      "training_step_time": 9.909089803695679
    },
    {
      "epoch": 3.279495239257812e-05,
      "model_forward_time": 3.1358418464660645,
      "step": 8597
    },
    {
      "epoch": 3.279495239257812e-05,
      "step": 8597,
      "training_step_time": 10.329635858535767
    },
    {
      "epoch": 3.279876708984375e-05,
      "model_forward_time": 2.085437297821045,
      "step": 8598
    },
    {
      "epoch": 3.279876708984375e-05,
      "step": 8598,
      "training_step_time": 5.54674506187439
    },
    {
      "epoch": 3.279876708984375e-05,
      "model_forward_time": 3.131525993347168,
      "step": 8598
    },
    {
      "epoch": 3.279876708984375e-05,
      "step": 8598,
      "training_step_time": 9.910156011581421
    },
    {
      "epoch": 3.279876708984375e-05,
      "model_forward_time": 3.14076566696167,
      "step": 8598
    },
    {
      "epoch": 3.279876708984375e-05,
      "step": 8598,
      "training_step_time": 9.919190406799316
    },
    {
      "epoch": 3.279876708984375e-05,
      "model_forward_time": 2.435300588607788,
      "step": 8598
    },
    {
      "epoch": 3.279876708984375e-05,
      "step": 8598,
      "training_step_time": 10.26561164855957
    },
    {
      "epoch": 3.2802581787109376e-05,
      "model_forward_time": 2.4487481117248535,
      "step": 8599
    },
    {
      "epoch": 3.2802581787109376e-05,
      "step": 8599,
      "training_step_time": 7.483299493789673
    },
    {
      "epoch": 3.2802581787109376e-05,
      "model_forward_time": 3.4329833984375,
      "step": 8599
    },
    {
      "epoch": 3.2802581787109376e-05,
      "step": 8599,
      "training_step_time": 10.199720859527588
    },
    {
      "epoch": 3.2802581787109376e-05,
      "model_forward_time": 3.1373894214630127,
      "step": 8599
    },
    {
      "epoch": 3.2802581787109376e-05,
      "step": 8599,
      "training_step_time": 9.908240556716919
    },
    {
      "epoch": 3.2802581787109376e-05,
      "model_forward_time": 1.742969036102295,
      "step": 8599
    },
    {
      "epoch": 3.2802581787109376e-05,
      "step": 8599,
      "training_step_time": 10.017930030822754
    },
    {
      "epoch": 3.2806396484375e-05,
      "grad_norm": 0.36261218786239624,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 8600
    },
    {
      "epoch": 3.2806396484375e-05,
      "model_forward_time": 3.148285388946533,
      "step": 8600
    },
    {
      "dynamics_loss_avg": 0.07036654576659203,
      "epoch": 3.2806396484375e-05,
      "step": 8600
    },
    {
      "action_loss_avg": 0.026617830712348224,
      "epoch": 3.2806396484375e-05,
      "step": 8600
    },
    {
      "epoch": 3.2806396484375e-05,
      "step": 8600,
      "training_step_time": 9.92160439491272
    },
    {
      "epoch": 3.2806396484375e-05,
      "model_forward_time": 3.1346073150634766,
      "step": 8600
    },
    {
      "epoch": 3.2806396484375e-05,
      "step": 8600,
      "training_step_time": 9.90426754951477
    },
    {
      "epoch": 3.2806396484375e-05,
      "model_forward_time": 2.4285428524017334,
      "step": 8600
    },
    {
      "epoch": 3.2806396484375e-05,
      "step": 8600,
      "training_step_time": 7.452292203903198
    },
    {
      "epoch": 3.2806396484375e-05,
      "model_forward_time": 3.3932394981384277,
      "step": 8600
    },
    {
      "epoch": 3.2806396484375e-05,
      "step": 8600,
      "training_step_time": 10.321722984313965
    },
    {
      "epoch": 3.281021118164063e-05,
      "model_forward_time": 3.1397626399993896,
      "step": 8601
    },
    {
      "epoch": 3.281021118164063e-05,
      "step": 8601,
      "training_step_time": 9.912374496459961
    },
    {
      "epoch": 3.281021118164063e-05,
      "model_forward_time": 3.1387503147125244,
      "step": 8601
    },
    {
      "epoch": 3.281021118164063e-05,
      "step": 8601,
      "training_step_time": 9.909064292907715
    },
    {
      "epoch": 3.281021118164063e-05,
      "model_forward_time": 3.1350347995758057,
      "step": 8601
    },
    {
      "epoch": 3.281021118164063e-05,
      "step": 8601,
      "training_step_time": 9.905189514160156
    },
    {
      "epoch": 3.281021118164063e-05,
      "model_forward_time": 3.138185977935791,
      "step": 8601
    },
    {
      "epoch": 3.281021118164063e-05,
      "step": 8601,
      "training_step_time": 10.282570600509644
    },
    {
      "epoch": 3.281402587890625e-05,
      "model_forward_time": 2.7353568077087402,
      "step": 8602
    },
    {
      "epoch": 3.281402587890625e-05,
      "step": 8602,
      "training_step_time": 7.782882213592529
    },
    {
      "epoch": 3.281402587890625e-05,
      "model_forward_time": 3.136289358139038,
      "step": 8602
    },
    {
      "epoch": 3.281402587890625e-05,
      "step": 8602,
      "training_step_time": 9.916663646697998
    },
    {
      "epoch": 3.281402587890625e-05,
      "model_forward_time": 3.138383388519287,
      "step": 8602
    },
    {
      "dynamics_loss_avg": 0.05149069065228105,
      "epoch": 3.281402587890625e-05,
      "step": 8602
    },
    {
      "action_loss_avg": 0.01892871451564133,
      "epoch": 3.281402587890625e-05,
      "step": 8602
    },
    {
      "epoch": 3.281402587890625e-05,
      "step": 8602,
      "training_step_time": 9.924381732940674
    },
    {
      "epoch": 3.281402587890625e-05,
      "model_forward_time": 3.143552541732788,
      "step": 8602
    },
    {
      "epoch": 3.281402587890625e-05,
      "step": 8602,
      "training_step_time": 10.302566289901733
    },
    {
      "epoch": 3.2817840576171874e-05,
      "model_forward_time": 2.4577817916870117,
      "step": 8603
    },
    {
      "epoch": 3.2817840576171874e-05,
      "step": 8603,
      "training_step_time": 7.491206169128418
    },
    {
      "epoch": 3.2817840576171874e-05,
      "model_forward_time": 2.7093427181243896,
      "step": 8603
    },
    {
      "epoch": 3.2817840576171874e-05,
      "step": 8603,
      "training_step_time": 7.7412779331207275
    },
    {
      "epoch": 3.2817840576171874e-05,
      "model_forward_time": 3.1322433948516846,
      "step": 8603
    },
    {
      "epoch": 3.2817840576171874e-05,
      "step": 8603,
      "training_step_time": 9.906929969787598
    },
    {
      "epoch": 3.2817840576171874e-05,
      "model_forward_time": 2.4362683296203613,
      "step": 8603
    },
    {
      "epoch": 3.2817840576171874e-05,
      "step": 8603,
      "training_step_time": 10.273240327835083
    },
    {
      "epoch": 3.28216552734375e-05,
      "model_forward_time": 2.4475786685943604,
      "step": 8604
    },
    {
      "epoch": 3.28216552734375e-05,
      "step": 8604,
      "training_step_time": 7.480132818222046
    },
    {
      "epoch": 3.28216552734375e-05,
      "model_forward_time": 3.1305110454559326,
      "step": 8604
    },
    {
      "epoch": 3.28216552734375e-05,
      "step": 8604,
      "training_step_time": 9.896575450897217
    },
    {
      "epoch": 3.28216552734375e-05,
      "model_forward_time": 2.754046678543091,
      "step": 8604
    },
    {
      "epoch": 3.28216552734375e-05,
      "step": 8604,
      "training_step_time": 7.780927658081055
    },
    {
      "epoch": 3.28216552734375e-05,
      "model_forward_time": 2.4276487827301025,
      "step": 8604
    },
    {
      "epoch": 3.28216552734375e-05,
      "step": 8604,
      "training_step_time": 10.28820013999939
    },
    {
      "epoch": 3.282546997070313e-05,
      "model_forward_time": 3.1459763050079346,
      "step": 8605
    },
    {
      "dynamics_loss_avg": 0.06058263313025236,
      "epoch": 3.282546997070313e-05,
      "step": 8605
    },
    {
      "action_loss_avg": 0.025908278208225964,
      "epoch": 3.282546997070313e-05,
      "step": 8605
    },
    {
      "epoch": 3.282546997070313e-05,
      "step": 8605,
      "training_step_time": 9.917112350463867
    },
    {
      "epoch": 3.282546997070313e-05,
      "model_forward_time": 2.442593812942505,
      "step": 8605
    },
    {
      "epoch": 3.282546997070313e-05,
      "step": 8605,
      "training_step_time": 7.474226474761963
    },
    {
      "epoch": 3.282546997070313e-05,
      "model_forward_time": 3.141526460647583,
      "step": 8605
    },
    {
      "epoch": 3.282546997070313e-05,
      "step": 8605,
      "training_step_time": 9.910682201385498
    },
    {
      "epoch": 3.282546997070313e-05,
      "model_forward_time": 3.145831823348999,
      "step": 8605
    },
    {
      "epoch": 3.282546997070313e-05,
      "step": 8605,
      "training_step_time": 10.291519403457642
    },
    {
      "epoch": 3.282928466796875e-05,
      "model_forward_time": 3.5241949558258057,
      "step": 8606
    },
    {
      "epoch": 3.282928466796875e-05,
      "step": 8606,
      "training_step_time": 10.296136856079102
    },
    {
      "epoch": 3.282928466796875e-05,
      "model_forward_time": 2.4364116191864014,
      "step": 8606
    },
    {
      "epoch": 3.282928466796875e-05,
      "step": 8606,
      "training_step_time": 7.46709132194519
    },
    {
      "epoch": 3.282928466796875e-05,
      "model_forward_time": 2.4440696239471436,
      "step": 8606
    },
    {
      "epoch": 3.282928466796875e-05,
      "step": 8606,
      "training_step_time": 7.476383209228516
    },
    {
      "epoch": 3.282928466796875e-05,
      "model_forward_time": 2.4514496326446533,
      "step": 8606
    },
    {
      "epoch": 3.282928466796875e-05,
      "step": 8606,
      "training_step_time": 10.012486934661865
    },
    {
      "epoch": 3.283309936523437e-05,
      "model_forward_time": 2.4637060165405273,
      "step": 8607
    },
    {
      "epoch": 3.283309936523437e-05,
      "step": 8607,
      "training_step_time": 7.490724325180054
    },
    {
      "epoch": 3.283309936523437e-05,
      "model_forward_time": 3.4499120712280273,
      "step": 8607
    },
    {
      "epoch": 3.283309936523437e-05,
      "step": 8607,
      "training_step_time": 10.219957113265991
    },
    {
      "epoch": 3.283309936523437e-05,
      "model_forward_time": 1.7499642372131348,
      "step": 8607
    },
    {
      "dynamics_loss_avg": 0.04916304554790259,
      "epoch": 3.283309936523437e-05,
      "step": 8607
    },
    {
      "action_loss_avg": 0.02090051709674299,
      "epoch": 3.283309936523437e-05,
      "step": 8607
    },
    {
      "epoch": 3.283309936523437e-05,
      "step": 8607,
      "training_step_time": 5.2056708335876465
    },
    {
      "epoch": 3.283309936523437e-05,
      "model_forward_time": 2.4413886070251465,
      "step": 8607
    },
    {
      "epoch": 3.283309936523437e-05,
      "step": 8607,
      "training_step_time": 10.33413577079773
    },
    {
      "epoch": 3.28369140625e-05,
      "model_forward_time": 2.452875852584839,
      "step": 8608
    },
    {
      "epoch": 3.28369140625e-05,
      "step": 8608,
      "training_step_time": 7.47638201713562
    },
    {
      "epoch": 3.28369140625e-05,
      "model_forward_time": 2.4419546127319336,
      "step": 8608
    },
    {
      "epoch": 3.28369140625e-05,
      "step": 8608,
      "training_step_time": 7.46855616569519
    },
    {
      "epoch": 3.28369140625e-05,
      "model_forward_time": 2.435276746749878,
      "step": 8608
    },
    {
      "epoch": 3.28369140625e-05,
      "step": 8608,
      "training_step_time": 7.4596662521362305
    },
    {
      "epoch": 3.28369140625e-05,
      "model_forward_time": 2.732816696166992,
      "step": 8608
    },
    {
      "epoch": 3.28369140625e-05,
      "step": 8608,
      "training_step_time": 10.030781984329224
    },
    {
      "epoch": 3.2840728759765625e-05,
      "model_forward_time": 2.454329252243042,
      "step": 8609
    },
    {
      "epoch": 3.2840728759765625e-05,
      "step": 8609,
      "training_step_time": 7.4824059009552
    },
    {
      "epoch": 3.2840728759765625e-05,
      "model_forward_time": 3.1452178955078125,
      "step": 8609
    },
    {
      "epoch": 3.2840728759765625e-05,
      "step": 8609,
      "training_step_time": 9.916566610336304
    },
    {
      "epoch": 3.2840728759765625e-05,
      "model_forward_time": 2.4372551441192627,
      "step": 8609
    },
    {
      "epoch": 3.2840728759765625e-05,
      "step": 8609,
      "training_step_time": 7.467227220535278
    },
    {
      "epoch": 3.2840728759765625e-05,
      "model_forward_time": 2.4367663860321045,
      "step": 8609
    },
    {
      "epoch": 3.2840728759765625e-05,
      "step": 8609,
      "training_step_time": 10.29672646522522
    },
    {
      "epoch": 3.284454345703125e-05,
      "grad_norm": 0.42646512389183044,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 8610
    },
    {
      "epoch": 3.284454345703125e-05,
      "model_forward_time": 3.4191880226135254,
      "step": 8610
    },
    {
      "dynamics_loss_avg": 0.07747046127915383,
      "epoch": 3.284454345703125e-05,
      "step": 8610
    },
    {
      "action_loss_avg": 0.027703987248241902,
      "epoch": 3.284454345703125e-05,
      "step": 8610
    },
    {
      "epoch": 3.284454345703125e-05,
      "step": 8610,
      "training_step_time": 10.195576190948486
    },
    {
      "epoch": 3.284454345703125e-05,
      "model_forward_time": 2.4392552375793457,
      "step": 8610
    },
    {
      "epoch": 3.284454345703125e-05,
      "step": 8610,
      "training_step_time": 7.467651128768921
    },
    {
      "epoch": 3.284454345703125e-05,
      "model_forward_time": 3.1406309604644775,
      "step": 8610
    },
    {
      "epoch": 3.284454345703125e-05,
      "step": 8610,
      "training_step_time": 9.914970636367798
    },
    {
      "epoch": 3.284454345703125e-05,
      "model_forward_time": 2.438886880874634,
      "step": 8610
    },
    {
      "epoch": 3.284454345703125e-05,
      "step": 8610,
      "training_step_time": 10.268990755081177
    },
    {
      "epoch": 3.284835815429688e-05,
      "model_forward_time": 3.1562273502349854,
      "step": 8611
    },
    {
      "epoch": 3.284835815429688e-05,
      "step": 8611,
      "training_step_time": 9.925764083862305
    },
    {
      "epoch": 3.284835815429688e-05,
      "model_forward_time": 3.4574286937713623,
      "step": 8611
    },
    {
      "epoch": 3.284835815429688e-05,
      "step": 8611,
      "training_step_time": 10.222318172454834
    },
    {
      "epoch": 3.284835815429688e-05,
      "model_forward_time": 3.142568349838257,
      "step": 8611
    },
    {
      "epoch": 3.284835815429688e-05,
      "step": 8611,
      "training_step_time": 9.910412311553955
    },
    {
      "epoch": 3.284835815429688e-05,
      "model_forward_time": 3.14219069480896,
      "step": 8611
    },
    {
      "epoch": 3.284835815429688e-05,
      "step": 8611,
      "training_step_time": 10.262213945388794
    },
    {
      "epoch": 3.28521728515625e-05,
      "model_forward_time": 2.454951286315918,
      "step": 8612
    },
    {
      "epoch": 3.28521728515625e-05,
      "step": 8612,
      "training_step_time": 7.4795002937316895
    },
    {
      "epoch": 3.28521728515625e-05,
      "model_forward_time": 3.14105224609375,
      "step": 8612
    },
    {
      "epoch": 3.28521728515625e-05,
      "step": 8612,
      "training_step_time": 9.917722702026367
    },
    {
      "epoch": 3.28521728515625e-05,
      "model_forward_time": 2.7474379539489746,
      "step": 8612
    },
    {
      "dynamics_loss_avg": 0.06274226568639278,
      "epoch": 3.28521728515625e-05,
      "step": 8612
    },
    {
      "action_loss_avg": 0.02617897167801857,
      "epoch": 3.28521728515625e-05,
      "step": 8612
    },
    {
      "epoch": 3.28521728515625e-05,
      "step": 8612,
      "training_step_time": 7.774940729141235
    },
    {
      "epoch": 3.28521728515625e-05,
      "model_forward_time": 2.4368762969970703,
      "step": 8612
    },
    {
      "epoch": 3.28521728515625e-05,
      "step": 8612,
      "training_step_time": 10.316468477249146
    },
    {
      "epoch": 3.285598754882812e-05,
      "model_forward_time": 2.450619697570801,
      "step": 8613
    },
    {
      "epoch": 3.285598754882812e-05,
      "step": 8613,
      "training_step_time": 7.484845876693726
    },
    {
      "epoch": 3.285598754882812e-05,
      "model_forward_time": 3.143181324005127,
      "step": 8613
    },
    {
      "epoch": 3.285598754882812e-05,
      "step": 8613,
      "training_step_time": 9.915252447128296
    },
    {
      "epoch": 3.285598754882812e-05,
      "model_forward_time": 3.1462087631225586,
      "step": 8613
    },
    {
      "epoch": 3.285598754882812e-05,
      "step": 8613,
      "training_step_time": 9.923000574111938
    },
    {
      "epoch": 3.285598754882812e-05,
      "model_forward_time": 3.4350569248199463,
      "step": 8613
    },
    {
      "epoch": 3.285598754882812e-05,
      "step": 8613,
      "training_step_time": 10.308591365814209
    },
    {
      "epoch": 3.285980224609375e-05,
      "model_forward_time": 3.1628329753875732,
      "step": 8614
    },
    {
      "epoch": 3.285980224609375e-05,
      "step": 8614,
      "training_step_time": 9.94107961654663
    },
    {
      "epoch": 3.285980224609375e-05,
      "model_forward_time": 2.440364360809326,
      "step": 8614
    },
    {
      "epoch": 3.285980224609375e-05,
      "step": 8614,
      "training_step_time": 7.468644618988037
    },
    {
      "epoch": 3.285980224609375e-05,
      "model_forward_time": 3.145411968231201,
      "step": 8614
    },
    {
      "epoch": 3.285980224609375e-05,
      "step": 8614,
      "training_step_time": 9.924196481704712
    },
    {
      "epoch": 3.285980224609375e-05,
      "model_forward_time": 3.1440086364746094,
      "step": 8614
    },
    {
      "epoch": 3.285980224609375e-05,
      "step": 8614,
      "training_step_time": 10.29032588005066
    },
    {
      "epoch": 3.2863616943359376e-05,
      "model_forward_time": 3.1611154079437256,
      "step": 8615
    },
    {
      "dynamics_loss_avg": 0.051198501326143744,
      "epoch": 3.2863616943359376e-05,
      "step": 8615
    },
    {
      "action_loss_avg": 0.02577919801697135,
      "epoch": 3.2863616943359376e-05,
      "step": 8615
    },
    {
      "epoch": 3.2863616943359376e-05,
      "step": 8615,
      "training_step_time": 9.946512699127197
    },
    {
      "epoch": 3.2863616943359376e-05,
      "model_forward_time": 2.761338949203491,
      "step": 8615
    },
    {
      "epoch": 3.2863616943359376e-05,
      "step": 8615,
      "training_step_time": 7.7966837882995605
    },
    {
      "epoch": 3.2863616943359376e-05,
      "model_forward_time": 2.4396891593933105,
      "step": 8615
    },
    {
      "epoch": 3.2863616943359376e-05,
      "step": 8615,
      "training_step_time": 7.472499370574951
    },
    {
      "epoch": 3.2863616943359376e-05,
      "model_forward_time": 2.439401626586914,
      "step": 8615
    },
    {
      "epoch": 3.2863616943359376e-05,
      "step": 8615,
      "training_step_time": 10.274149179458618
    },
    {
      "epoch": 3.2867431640625e-05,
      "model_forward_time": 2.4605884552001953,
      "step": 8616
    },
    {
      "epoch": 3.2867431640625e-05,
      "step": 8616,
      "training_step_time": 7.4935081005096436
    },
    {
      "epoch": 3.2867431640625e-05,
      "model_forward_time": 2.4383509159088135,
      "step": 8616
    },
    {
      "epoch": 3.2867431640625e-05,
      "step": 8616,
      "training_step_time": 7.465938568115234
    },
    {
      "epoch": 3.2867431640625e-05,
      "model_forward_time": 3.427830219268799,
      "step": 8616
    },
    {
      "epoch": 3.2867431640625e-05,
      "step": 8616,
      "training_step_time": 10.214109420776367
    },
    {
      "epoch": 3.2867431640625e-05,
      "model_forward_time": 2.4404690265655518,
      "step": 8616
    },
    {
      "epoch": 3.2867431640625e-05,
      "step": 8616,
      "training_step_time": 10.32593560218811
    },
    {
      "epoch": 3.287124633789063e-05,
      "model_forward_time": 3.157724380493164,
      "step": 8617
    },
    {
      "epoch": 3.287124633789063e-05,
      "step": 8617,
      "training_step_time": 9.938446760177612
    },
    {
      "epoch": 3.287124633789063e-05,
      "model_forward_time": 2.4358770847320557,
      "step": 8617
    },
    {
      "epoch": 3.287124633789063e-05,
      "step": 8617,
      "training_step_time": 7.462393283843994
    },
    {
      "epoch": 3.287124633789063e-05,
      "model_forward_time": 3.1330819129943848,
      "step": 8617
    },
    {
      "dynamics_loss_avg": 0.057514659874141214,
      "epoch": 3.287124633789063e-05,
      "step": 8617
    },
    {
      "action_loss_avg": 0.019321938836947085,
      "epoch": 3.287124633789063e-05,
      "step": 8617
    },
    {
      "epoch": 3.287124633789063e-05,
      "step": 8617,
      "training_step_time": 9.910712242126465
    },
    {
      "epoch": 3.287124633789063e-05,
      "model_forward_time": 3.394253969192505,
      "step": 8617
    },
    {
      "epoch": 3.287124633789063e-05,
      "step": 8617,
      "training_step_time": 10.320520401000977
    },
    {
      "epoch": 3.287506103515625e-05,
      "model_forward_time": 2.4449737071990967,
      "step": 8618
    },
    {
      "epoch": 3.287506103515625e-05,
      "step": 8618,
      "training_step_time": 7.478226661682129
    },
    {
      "epoch": 3.287506103515625e-05,
      "model_forward_time": 3.138368844985962,
      "step": 8618
    },
    {
      "epoch": 3.287506103515625e-05,
      "step": 8618,
      "training_step_time": 9.913722038269043
    },
    {
      "epoch": 3.287506103515625e-05,
      "model_forward_time": 3.137378692626953,
      "step": 8618
    },
    {
      "epoch": 3.287506103515625e-05,
      "step": 8618,
      "training_step_time": 9.914076805114746
    },
    {
      "epoch": 3.287506103515625e-05,
      "model_forward_time": 3.140108108520508,
      "step": 8618
    },
    {
      "epoch": 3.287506103515625e-05,
      "step": 8618,
      "training_step_time": 10.014450550079346
    },
    {
      "epoch": 3.2878875732421874e-05,
      "model_forward_time": 3.4109578132629395,
      "step": 8619
    },
    {
      "epoch": 3.2878875732421874e-05,
      "step": 8619,
      "training_step_time": 10.19890809059143
    },
    {
      "epoch": 3.2878875732421874e-05,
      "model_forward_time": 2.4373223781585693,
      "step": 8619
    },
    {
      "epoch": 3.2878875732421874e-05,
      "step": 8619,
      "training_step_time": 7.474712371826172
    },
    {
      "epoch": 3.2878875732421874e-05,
      "model_forward_time": 3.1381895542144775,
      "step": 8619
    },
    {
      "epoch": 3.2878875732421874e-05,
      "step": 8619,
      "training_step_time": 9.920506477355957
    },
    {
      "epoch": 3.2878875732421874e-05,
      "model_forward_time": 2.4355902671813965,
      "step": 8619
    },
    {
      "epoch": 3.2878875732421874e-05,
      "step": 8619,
      "training_step_time": 10.022798538208008
    },
    {
      "epoch": 3.28826904296875e-05,
      "grad_norm": 0.6740884780883789,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 8620
    },
    {
      "epoch": 3.28826904296875e-05,
      "model_forward_time": 3.1540446281433105,
      "step": 8620
    },
    {
      "dynamics_loss_avg": 0.05503826010972261,
      "epoch": 3.28826904296875e-05,
      "step": 8620
    },
    {
      "action_loss_avg": 0.020782363787293433,
      "epoch": 3.28826904296875e-05,
      "step": 8620
    },
    {
      "epoch": 3.28826904296875e-05,
      "step": 8620,
      "training_step_time": 9.939727067947388
    },
    {
      "epoch": 3.28826904296875e-05,
      "model_forward_time": 2.4349358081817627,
      "step": 8620
    },
    {
      "epoch": 3.28826904296875e-05,
      "step": 8620,
      "training_step_time": 7.7447638511657715
    },
    {
      "epoch": 3.28826904296875e-05,
      "model_forward_time": 3.134294271469116,
      "step": 8620
    },
    {
      "epoch": 3.28826904296875e-05,
      "step": 8620,
      "training_step_time": 9.907337665557861
    },
    {
      "epoch": 3.28826904296875e-05,
      "model_forward_time": 2.4357993602752686,
      "step": 8620
    },
    {
      "epoch": 3.28826904296875e-05,
      "step": 8620,
      "training_step_time": 10.275104999542236
    },
    {
      "epoch": 3.288650512695313e-05,
      "model_forward_time": 2.4535186290740967,
      "step": 8621
    },
    {
      "epoch": 3.288650512695313e-05,
      "step": 8621,
      "training_step_time": 7.4940619468688965
    },
    {
      "epoch": 3.288650512695313e-05,
      "model_forward_time": 3.132150650024414,
      "step": 8621
    },
    {
      "epoch": 3.288650512695313e-05,
      "step": 8621,
      "training_step_time": 9.90365743637085
    },
    {
      "epoch": 3.288650512695313e-05,
      "model_forward_time": 3.4116976261138916,
      "step": 8621
    },
    {
      "epoch": 3.288650512695313e-05,
      "step": 8621,
      "training_step_time": 10.180290699005127
    },
    {
      "epoch": 3.288650512695313e-05,
      "model_forward_time": 2.4302327632904053,
      "step": 8621
    },
    {
      "epoch": 3.288650512695313e-05,
      "step": 8621,
      "training_step_time": 10.299883604049683
    },
    {
      "epoch": 3.289031982421875e-05,
      "model_forward_time": 2.445613384246826,
      "step": 8622
    },
    {
      "epoch": 3.289031982421875e-05,
      "step": 8622,
      "training_step_time": 7.4717888832092285
    },
    {
      "epoch": 3.289031982421875e-05,
      "model_forward_time": 2.433225631713867,
      "step": 8622
    },
    {
      "epoch": 3.289031982421875e-05,
      "step": 8622,
      "training_step_time": 7.463463544845581
    },
    {
      "epoch": 3.289031982421875e-05,
      "model_forward_time": 2.4307193756103516,
      "step": 8622
    },
    {
      "dynamics_loss_avg": 0.04897860810160637,
      "epoch": 3.289031982421875e-05,
      "step": 8622
    },
    {
      "action_loss_avg": 0.016189449327066542,
      "epoch": 3.289031982421875e-05,
      "step": 8622
    },
    {
      "epoch": 3.289031982421875e-05,
      "step": 8622,
      "training_step_time": 7.463472127914429
    },
    {
      "epoch": 3.289031982421875e-05,
      "model_forward_time": 2.434720039367676,
      "step": 8622
    },
    {
      "epoch": 3.289031982421875e-05,
      "step": 8622,
      "training_step_time": 10.314189434051514
    },
    {
      "epoch": 3.289413452148437e-05,
      "model_forward_time": 2.761314868927002,
      "step": 8623
    },
    {
      "epoch": 3.289413452148437e-05,
      "step": 8623,
      "training_step_time": 7.788244724273682
    },
    {
      "epoch": 3.289413452148437e-05,
      "model_forward_time": 3.134009838104248,
      "step": 8623
    },
    {
      "epoch": 3.289413452148437e-05,
      "step": 8623,
      "training_step_time": 9.909898519515991
    },
    {
      "epoch": 3.289413452148437e-05,
      "model_forward_time": 2.4344358444213867,
      "step": 8623
    },
    {
      "epoch": 3.289413452148437e-05,
      "step": 8623,
      "training_step_time": 7.468598127365112
    },
    {
      "epoch": 3.289413452148437e-05,
      "model_forward_time": 3.1358981132507324,
      "step": 8623
    },
    {
      "epoch": 3.289413452148437e-05,
      "step": 8623,
      "training_step_time": 10.398699760437012
    },
    {
      "epoch": 3.289794921875e-05,
      "model_forward_time": 3.1426377296447754,
      "step": 8624
    },
    {
      "epoch": 3.289794921875e-05,
      "step": 8624,
      "training_step_time": 9.923033237457275
    },
    {
      "epoch": 3.289794921875e-05,
      "model_forward_time": 2.435236930847168,
      "step": 8624
    },
    {
      "epoch": 3.289794921875e-05,
      "step": 8624,
      "training_step_time": 7.747446060180664
    },
    {
      "epoch": 3.289794921875e-05,
      "model_forward_time": 3.143129587173462,
      "step": 8624
    },
    {
      "epoch": 3.289794921875e-05,
      "step": 8624,
      "training_step_time": 9.919776916503906
    },
    {
      "epoch": 3.289794921875e-05,
      "model_forward_time": 3.1414408683776855,
      "step": 8624
    },
    {
      "epoch": 3.289794921875e-05,
      "step": 8624,
      "training_step_time": 10.036296844482422
    },
    {
      "epoch": 3.2901763916015625e-05,
      "model_forward_time": 2.449310779571533,
      "step": 8625
    },
    {
      "dynamics_loss_avg": 0.064675254188478,
      "epoch": 3.2901763916015625e-05,
      "step": 8625
    },
    {
      "action_loss_avg": 0.020950486604124307,
      "epoch": 3.2901763916015625e-05,
      "step": 8625
    },
    {
      "epoch": 3.2901763916015625e-05,
      "step": 8625,
      "training_step_time": 7.488936424255371
    },
    {
      "epoch": 3.2901763916015625e-05,
      "model_forward_time": 3.1378982067108154,
      "step": 8625
    },
    {
      "epoch": 3.2901763916015625e-05,
      "step": 8625,
      "training_step_time": 9.915725469589233
    },
    {
      "epoch": 3.2901763916015625e-05,
      "model_forward_time": 3.140498161315918,
      "step": 8625
    },
    {
      "epoch": 3.2901763916015625e-05,
      "step": 8625,
      "training_step_time": 9.923223733901978
    },
    {
      "epoch": 3.2901763916015625e-05,
      "model_forward_time": 3.389659881591797,
      "step": 8625
    },
    {
      "epoch": 3.2901763916015625e-05,
      "step": 8625,
      "training_step_time": 10.282796144485474
    },
    {
      "epoch": 3.290557861328125e-05,
      "model_forward_time": 3.162182569503784,
      "step": 8626
    },
    {
      "epoch": 3.290557861328125e-05,
      "step": 8626,
      "training_step_time": 9.943489074707031
    },
    {
      "epoch": 3.290557861328125e-05,
      "model_forward_time": 3.1346163749694824,
      "step": 8626
    },
    {
      "epoch": 3.290557861328125e-05,
      "step": 8626,
      "training_step_time": 9.910411357879639
    },
    {
      "epoch": 3.290557861328125e-05,
      "model_forward_time": 2.430401086807251,
      "step": 8626
    },
    {
      "epoch": 3.290557861328125e-05,
      "step": 8626,
      "training_step_time": 7.459835529327393
    },
    {
      "epoch": 3.290557861328125e-05,
      "model_forward_time": 2.4261982440948486,
      "step": 8626
    },
    {
      "epoch": 3.290557861328125e-05,
      "step": 8626,
      "training_step_time": 10.30676555633545
    },
    {
      "epoch": 3.290939331054688e-05,
      "model_forward_time": 2.08359432220459,
      "step": 8627
    },
    {
      "epoch": 3.290939331054688e-05,
      "step": 8627,
      "training_step_time": 5.53666353225708
    },
    {
      "epoch": 3.290939331054688e-05,
      "model_forward_time": 3.1275839805603027,
      "step": 8627
    },
    {
      "epoch": 3.290939331054688e-05,
      "step": 8627,
      "training_step_time": 9.901057004928589
    },
    {
      "epoch": 3.290939331054688e-05,
      "model_forward_time": 3.1287097930908203,
      "step": 8627
    },
    {
      "dynamics_loss_avg": 0.05565260164439678,
      "epoch": 3.290939331054688e-05,
      "step": 8627
    },
    {
      "action_loss_avg": 0.019010375253856182,
      "epoch": 3.290939331054688e-05,
      "step": 8627
    },
    {
      "epoch": 3.290939331054688e-05,
      "step": 8627,
      "training_step_time": 9.901181936264038
    },
    {
      "epoch": 3.290939331054688e-05,
      "model_forward_time": 3.136618137359619,
      "step": 8627
    },
    {
      "epoch": 3.290939331054688e-05,
      "step": 8627,
      "training_step_time": 10.283149719238281
    },
    {
      "epoch": 3.29132080078125e-05,
      "model_forward_time": 2.444432497024536,
      "step": 8628
    },
    {
      "epoch": 3.29132080078125e-05,
      "step": 8628,
      "training_step_time": 7.4713053703308105
    },
    {
      "epoch": 3.29132080078125e-05,
      "model_forward_time": 3.420215368270874,
      "step": 8628
    },
    {
      "epoch": 3.29132080078125e-05,
      "step": 8628,
      "training_step_time": 10.194738626480103
    },
    {
      "epoch": 3.29132080078125e-05,
      "model_forward_time": 2.4346606731414795,
      "step": 8628
    },
    {
      "epoch": 3.29132080078125e-05,
      "step": 8628,
      "training_step_time": 7.468853950500488
    },
    {
      "epoch": 3.29132080078125e-05,
      "model_forward_time": 3.1338605880737305,
      "step": 8628
    },
    {
      "epoch": 3.29132080078125e-05,
      "step": 8628,
      "training_step_time": 10.345294713973999
    },
    {
      "epoch": 3.291702270507812e-05,
      "model_forward_time": 2.447394609451294,
      "step": 8629
    },
    {
      "epoch": 3.291702270507812e-05,
      "step": 8629,
      "training_step_time": 7.481889963150024
    },
    {
      "epoch": 3.291702270507812e-05,
      "model_forward_time": 2.433835983276367,
      "step": 8629
    },
    {
      "epoch": 3.291702270507812e-05,
      "step": 8629,
      "training_step_time": 7.467345476150513
    },
    {
      "epoch": 3.291702270507812e-05,
      "model_forward_time": 3.4325296878814697,
      "step": 8629
    },
    {
      "epoch": 3.291702270507812e-05,
      "step": 8629,
      "training_step_time": 10.209310531616211
    },
    {
      "epoch": 3.291702270507812e-05,
      "model_forward_time": 2.4347498416900635,
      "step": 8629
    },
    {
      "epoch": 3.291702270507812e-05,
      "step": 8629,
      "training_step_time": 10.027327299118042
    },
    {
      "epoch": 3.292083740234375e-05,
      "grad_norm": 0.3505253195762634,
      "learning_rate": 1e-05,
      "loss": 0.0751,
      "step": 8630
    },
    {
      "epoch": 3.292083740234375e-05,
      "model_forward_time": 2.4424643516540527,
      "step": 8630
    },
    {
      "dynamics_loss_avg": 0.05884705297648907,
      "epoch": 3.292083740234375e-05,
      "step": 8630
    },
    {
      "action_loss_avg": 0.021973782684653997,
      "epoch": 3.292083740234375e-05,
      "step": 8630
    },
    {
      "epoch": 3.292083740234375e-05,
      "step": 8630,
      "training_step_time": 7.476653337478638
    },
    {
      "epoch": 3.292083740234375e-05,
      "model_forward_time": 2.42730712890625,
      "step": 8630
    },
    {
      "epoch": 3.292083740234375e-05,
      "step": 8630,
      "training_step_time": 7.452131509780884
    },
    {
      "epoch": 3.292083740234375e-05,
      "model_forward_time": 3.1304986476898193,
      "step": 8630
    },
    {
      "epoch": 3.292083740234375e-05,
      "step": 8630,
      "training_step_time": 9.901250123977661
    },
    {
      "epoch": 3.292083740234375e-05,
      "model_forward_time": 3.1302196979522705,
      "step": 8630
    },
    {
      "epoch": 3.292083740234375e-05,
      "step": 8630,
      "training_step_time": 10.305073022842407
    },
    {
      "epoch": 3.2924652099609376e-05,
      "model_forward_time": 3.445539951324463,
      "step": 8631
    },
    {
      "epoch": 3.2924652099609376e-05,
      "step": 8631,
      "training_step_time": 10.220563650131226
    },
    {
      "epoch": 3.2924652099609376e-05,
      "model_forward_time": 3.1278460025787354,
      "step": 8631
    },
    {
      "epoch": 3.2924652099609376e-05,
      "step": 8631,
      "training_step_time": 9.898338556289673
    },
    {
      "epoch": 3.2924652099609376e-05,
      "model_forward_time": 2.427581310272217,
      "step": 8631
    },
    {
      "epoch": 3.2924652099609376e-05,
      "step": 8631,
      "training_step_time": 7.453930854797363
    },
    {
      "epoch": 3.2924652099609376e-05,
      "model_forward_time": 2.4294488430023193,
      "step": 8631
    },
    {
      "epoch": 3.2924652099609376e-05,
      "step": 8631,
      "training_step_time": 10.247766256332397
    },
    {
      "epoch": 3.2928466796875e-05,
      "model_forward_time": 3.151296615600586,
      "step": 8632
    },
    {
      "epoch": 3.2928466796875e-05,
      "step": 8632,
      "training_step_time": 9.930461406707764
    },
    {
      "epoch": 3.2928466796875e-05,
      "model_forward_time": 3.3812315464019775,
      "step": 8632
    },
    {
      "epoch": 3.2928466796875e-05,
      "step": 8632,
      "training_step_time": 10.155424356460571
    },
    {
      "epoch": 3.2928466796875e-05,
      "model_forward_time": 2.4313809871673584,
      "step": 8632
    },
    {
      "dynamics_loss_avg": 0.057808800414204595,
      "epoch": 3.2928466796875e-05,
      "step": 8632
    },
    {
      "action_loss_avg": 0.023657256597653033,
      "epoch": 3.2928466796875e-05,
      "step": 8632
    },
    {
      "epoch": 3.2928466796875e-05,
      "step": 8632,
      "training_step_time": 7.462256669998169
    },
    {
      "epoch": 3.2928466796875e-05,
      "model_forward_time": 3.1280481815338135,
      "step": 8632
    },
    {
      "epoch": 3.2928466796875e-05,
      "step": 8632,
      "training_step_time": 10.016940116882324
    },
    {
      "epoch": 3.293228149414062e-05,
      "model_forward_time": 3.1411519050598145,
      "step": 8633
    },
    {
      "epoch": 3.293228149414062e-05,
      "step": 8633,
      "training_step_time": 9.91508436203003
    },
    {
      "epoch": 3.293228149414062e-05,
      "model_forward_time": 3.133711814880371,
      "step": 8633
    },
    {
      "epoch": 3.293228149414062e-05,
      "step": 8633,
      "training_step_time": 9.905813455581665
    },
    {
      "epoch": 3.293228149414062e-05,
      "model_forward_time": 3.4146430492401123,
      "step": 8633
    },
    {
      "epoch": 3.293228149414062e-05,
      "step": 8633,
      "training_step_time": 10.193313837051392
    },
    {
      "epoch": 3.293228149414062e-05,
      "model_forward_time": 3.134287118911743,
      "step": 8633
    },
    {
      "epoch": 3.293228149414062e-05,
      "step": 8633,
      "training_step_time": 10.326348066329956
    },
    {
      "epoch": 3.293609619140625e-05,
      "model_forward_time": 3.144566297531128,
      "step": 8634
    },
    {
      "epoch": 3.293609619140625e-05,
      "step": 8634,
      "training_step_time": 9.926129579544067
    },
    {
      "epoch": 3.293609619140625e-05,
      "model_forward_time": 3.155815601348877,
      "step": 8634
    },
    {
      "epoch": 3.293609619140625e-05,
      "step": 8634,
      "training_step_time": 9.93821382522583
    },
    {
      "epoch": 3.293609619140625e-05,
      "model_forward_time": 3.1432900428771973,
      "step": 8634
    },
    {
      "epoch": 3.293609619140625e-05,
      "step": 8634,
      "training_step_time": 9.923275232315063
    },
    {
      "epoch": 3.293609619140625e-05,
      "model_forward_time": 2.740983486175537,
      "step": 8634
    },
    {
      "epoch": 3.293609619140625e-05,
      "step": 8634,
      "training_step_time": 10.295512437820435
    },
    {
      "epoch": 3.2939910888671874e-05,
      "model_forward_time": 1.7765719890594482,
      "step": 8635
    },
    {
      "dynamics_loss_avg": 0.04901841562241316,
      "epoch": 3.2939910888671874e-05,
      "step": 8635
    },
    {
      "action_loss_avg": 0.01694939946755767,
      "epoch": 3.2939910888671874e-05,
      "step": 8635
    },
    {
      "epoch": 3.2939910888671874e-05,
      "step": 8635,
      "training_step_time": 5.252399444580078
    },
    {
      "epoch": 3.2939910888671874e-05,
      "model_forward_time": 2.443645477294922,
      "step": 8635
    },
    {
      "epoch": 3.2939910888671874e-05,
      "step": 8635,
      "training_step_time": 7.4854512214660645
    },
    {
      "epoch": 3.2939910888671874e-05,
      "model_forward_time": 2.432217597961426,
      "step": 8635
    },
    {
      "epoch": 3.2939910888671874e-05,
      "step": 8635,
      "training_step_time": 7.466715335845947
    },
    {
      "epoch": 3.2939910888671874e-05,
      "model_forward_time": 3.1345956325531006,
      "step": 8635
    },
    {
      "epoch": 3.2939910888671874e-05,
      "step": 8635,
      "training_step_time": 10.024287939071655
    },
    {
      "epoch": 3.29437255859375e-05,
      "model_forward_time": 3.1634578704833984,
      "step": 8636
    },
    {
      "epoch": 3.29437255859375e-05,
      "step": 8636,
      "training_step_time": 10.278495073318481
    },
    {
      "epoch": 3.29437255859375e-05,
      "model_forward_time": 3.144437789916992,
      "step": 8636
    },
    {
      "epoch": 3.29437255859375e-05,
      "step": 8636,
      "training_step_time": 9.923295974731445
    },
    {
      "epoch": 3.29437255859375e-05,
      "model_forward_time": 2.4334805011749268,
      "step": 8636
    },
    {
      "epoch": 3.29437255859375e-05,
      "step": 8636,
      "training_step_time": 7.464796781539917
    },
    {
      "epoch": 3.29437255859375e-05,
      "model_forward_time": 2.431764841079712,
      "step": 8636
    },
    {
      "epoch": 3.29437255859375e-05,
      "step": 8636,
      "training_step_time": 10.015759229660034
    },
    {
      "epoch": 3.294754028320313e-05,
      "model_forward_time": 3.160316228866577,
      "step": 8637
    },
    {
      "epoch": 3.294754028320313e-05,
      "step": 8637,
      "training_step_time": 9.941340446472168
    },
    {
      "epoch": 3.294754028320313e-05,
      "model_forward_time": 3.141029119491577,
      "step": 8637
    },
    {
      "epoch": 3.294754028320313e-05,
      "step": 8637,
      "training_step_time": 9.922428131103516
    },
    {
      "epoch": 3.294754028320313e-05,
      "model_forward_time": 2.7264645099639893,
      "step": 8637
    },
    {
      "dynamics_loss_avg": 0.057211273536086085,
      "epoch": 3.294754028320313e-05,
      "step": 8637
    },
    {
      "action_loss_avg": 0.021807950083166362,
      "epoch": 3.294754028320313e-05,
      "step": 8637
    },
    {
      "epoch": 3.294754028320313e-05,
      "step": 8637,
      "training_step_time": 7.7643914222717285
    },
    {
      "epoch": 3.294754028320313e-05,
      "model_forward_time": 3.1421267986297607,
      "step": 8637
    },
    {
      "epoch": 3.294754028320313e-05,
      "step": 8637,
      "training_step_time": 10.025641441345215
    },
    {
      "epoch": 3.295135498046875e-05,
      "model_forward_time": 2.45695161819458,
      "step": 8638
    },
    {
      "epoch": 3.295135498046875e-05,
      "step": 8638,
      "training_step_time": 7.49344277381897
    },
    {
      "epoch": 3.295135498046875e-05,
      "model_forward_time": 3.140408515930176,
      "step": 8638
    },
    {
      "epoch": 3.295135498046875e-05,
      "step": 8638,
      "training_step_time": 9.92347526550293
    },
    {
      "epoch": 3.295135498046875e-05,
      "model_forward_time": 3.1454379558563232,
      "step": 8638
    },
    {
      "epoch": 3.295135498046875e-05,
      "step": 8638,
      "training_step_time": 9.926294088363647
    },
    {
      "epoch": 3.295135498046875e-05,
      "model_forward_time": 2.754287004470825,
      "step": 8638
    },
    {
      "epoch": 3.295135498046875e-05,
      "step": 8638,
      "training_step_time": 10.332634449005127
    },
    {
      "epoch": 3.295516967773437e-05,
      "model_forward_time": 3.1545119285583496,
      "step": 8639
    },
    {
      "epoch": 3.295516967773437e-05,
      "step": 8639,
      "training_step_time": 9.938277244567871
    },
    {
      "epoch": 3.295516967773437e-05,
      "model_forward_time": 3.140913963317871,
      "step": 8639
    },
    {
      "epoch": 3.295516967773437e-05,
      "step": 8639,
      "training_step_time": 9.919847011566162
    },
    {
      "epoch": 3.295516967773437e-05,
      "model_forward_time": 3.1515488624572754,
      "step": 8639
    },
    {
      "epoch": 3.295516967773437e-05,
      "step": 8639,
      "training_step_time": 9.934293746948242
    },
    {
      "epoch": 3.295516967773437e-05,
      "model_forward_time": 3.1419970989227295,
      "step": 8639
    },
    {
      "epoch": 3.295516967773437e-05,
      "step": 8639,
      "training_step_time": 10.322230577468872
    },
    {
      "epoch": 3.2958984375e-05,
      "grad_norm": 0.3337949812412262,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 8640
    },
    {
      "epoch": 3.2958984375e-05,
      "model_forward_time": 2.777205228805542,
      "step": 8640
    },
    {
      "dynamics_loss_avg": 0.05394797082990408,
      "epoch": 3.2958984375e-05,
      "step": 8640
    },
    {
      "action_loss_avg": 0.022425444237887858,
      "epoch": 3.2958984375e-05,
      "step": 8640
    },
    {
      "epoch": 3.2958984375e-05,
      "step": 8640,
      "training_step_time": 7.818699359893799
    },
    {
      "epoch": 3.2958984375e-05,
      "model_forward_time": 2.4363343715667725,
      "step": 8640
    },
    {
      "epoch": 3.2958984375e-05,
      "step": 8640,
      "training_step_time": 7.470458984375
    },
    {
      "epoch": 3.2958984375e-05,
      "model_forward_time": 2.4419658184051514,
      "step": 8640
    },
    {
      "epoch": 3.2958984375e-05,
      "step": 8640,
      "training_step_time": 7.476419925689697
    },
    {
      "epoch": 3.2958984375e-05,
      "model_forward_time": 3.13828444480896,
      "step": 8640
    },
    {
      "epoch": 3.2958984375e-05,
      "step": 8640,
      "training_step_time": 10.311380386352539
    },
    {
      "epoch": 3.2962799072265625e-05,
      "model_forward_time": 2.446453094482422,
      "step": 8641
    },
    {
      "epoch": 3.2962799072265625e-05,
      "step": 8641,
      "training_step_time": 7.478934288024902
    },
    {
      "epoch": 3.2962799072265625e-05,
      "model_forward_time": 3.431871175765991,
      "step": 8641
    },
    {
      "epoch": 3.2962799072265625e-05,
      "step": 8641,
      "training_step_time": 10.205282926559448
    },
    {
      "epoch": 3.2962799072265625e-05,
      "model_forward_time": 3.1374242305755615,
      "step": 8641
    },
    {
      "epoch": 3.2962799072265625e-05,
      "step": 8641,
      "training_step_time": 9.921874046325684
    },
    {
      "epoch": 3.2962799072265625e-05,
      "model_forward_time": 3.1391618251800537,
      "step": 8641
    },
    {
      "epoch": 3.2962799072265625e-05,
      "step": 8641,
      "training_step_time": 10.031293392181396
    },
    {
      "epoch": 3.296661376953125e-05,
      "model_forward_time": 2.4536654949188232,
      "step": 8642
    },
    {
      "epoch": 3.296661376953125e-05,
      "step": 8642,
      "training_step_time": 7.488365411758423
    },
    {
      "epoch": 3.296661376953125e-05,
      "model_forward_time": 3.144071578979492,
      "step": 8642
    },
    {
      "epoch": 3.296661376953125e-05,
      "step": 8642,
      "training_step_time": 9.91335654258728
    },
    {
      "epoch": 3.296661376953125e-05,
      "model_forward_time": 2.70064377784729,
      "step": 8642
    },
    {
      "dynamics_loss_avg": 0.04906138489022851,
      "epoch": 3.296661376953125e-05,
      "step": 8642
    },
    {
      "action_loss_avg": 0.015575489518232644,
      "epoch": 3.296661376953125e-05,
      "step": 8642
    },
    {
      "epoch": 3.296661376953125e-05,
      "step": 8642,
      "training_step_time": 7.73662805557251
    },
    {
      "epoch": 3.296661376953125e-05,
      "model_forward_time": 2.439984083175659,
      "step": 8642
    },
    {
      "epoch": 3.296661376953125e-05,
      "step": 8642,
      "training_step_time": 10.252059698104858
    },
    {
      "epoch": 3.297042846679688e-05,
      "model_forward_time": 2.448345184326172,
      "step": 8643
    },
    {
      "epoch": 3.297042846679688e-05,
      "step": 8643,
      "training_step_time": 7.478618144989014
    },
    {
      "epoch": 3.297042846679688e-05,
      "model_forward_time": 3.1431398391723633,
      "step": 8643
    },
    {
      "epoch": 3.297042846679688e-05,
      "step": 8643,
      "training_step_time": 9.921029329299927
    },
    {
      "epoch": 3.297042846679688e-05,
      "model_forward_time": 2.4377200603485107,
      "step": 8643
    },
    {
      "epoch": 3.297042846679688e-05,
      "step": 8643,
      "training_step_time": 7.470299482345581
    },
    {
      "epoch": 3.297042846679688e-05,
      "model_forward_time": 3.1356043815612793,
      "step": 8643
    },
    {
      "epoch": 3.297042846679688e-05,
      "step": 8643,
      "training_step_time": 10.309361696243286
    },
    {
      "epoch": 3.29742431640625e-05,
      "model_forward_time": 3.450242757797241,
      "step": 8644
    },
    {
      "epoch": 3.29742431640625e-05,
      "step": 8644,
      "training_step_time": 10.22470736503601
    },
    {
      "epoch": 3.29742431640625e-05,
      "model_forward_time": 3.1419615745544434,
      "step": 8644
    },
    {
      "epoch": 3.29742431640625e-05,
      "step": 8644,
      "training_step_time": 9.916969299316406
    },
    {
      "epoch": 3.29742431640625e-05,
      "model_forward_time": 2.4386470317840576,
      "step": 8644
    },
    {
      "epoch": 3.29742431640625e-05,
      "step": 8644,
      "training_step_time": 7.471637725830078
    },
    {
      "epoch": 3.29742431640625e-05,
      "model_forward_time": 3.1438162326812744,
      "step": 8644
    },
    {
      "epoch": 3.29742431640625e-05,
      "step": 8644,
      "training_step_time": 10.306051969528198
    },
    {
      "epoch": 3.297805786132812e-05,
      "model_forward_time": 3.17492413520813,
      "step": 8645
    },
    {
      "dynamics_loss_avg": 0.054010228952392936,
      "epoch": 3.297805786132812e-05,
      "step": 8645
    },
    {
      "action_loss_avg": 0.02115623663412407,
      "epoch": 3.297805786132812e-05,
      "step": 8645
    },
    {
      "epoch": 3.297805786132812e-05,
      "step": 8645,
      "training_step_time": 9.95884084701538
    },
    {
      "epoch": 3.297805786132812e-05,
      "model_forward_time": 2.721933126449585,
      "step": 8645
    },
    {
      "epoch": 3.297805786132812e-05,
      "step": 8645,
      "training_step_time": 7.751166343688965
    },
    {
      "epoch": 3.297805786132812e-05,
      "model_forward_time": 2.434865713119507,
      "step": 8645
    },
    {
      "epoch": 3.297805786132812e-05,
      "step": 8645,
      "training_step_time": 7.467819929122925
    },
    {
      "epoch": 3.297805786132812e-05,
      "model_forward_time": 3.1410889625549316,
      "step": 8645
    },
    {
      "epoch": 3.297805786132812e-05,
      "step": 8645,
      "training_step_time": 10.319868803024292
    },
    {
      "epoch": 3.298187255859375e-05,
      "model_forward_time": 3.167173147201538,
      "step": 8646
    },
    {
      "epoch": 3.298187255859375e-05,
      "step": 8646,
      "training_step_time": 9.950178623199463
    },
    {
      "epoch": 3.298187255859375e-05,
      "model_forward_time": 3.140422821044922,
      "step": 8646
    },
    {
      "epoch": 3.298187255859375e-05,
      "step": 8646,
      "training_step_time": 9.925355911254883
    },
    {
      "epoch": 3.298187255859375e-05,
      "model_forward_time": 2.6940605640411377,
      "step": 8646
    },
    {
      "epoch": 3.298187255859375e-05,
      "step": 8646,
      "training_step_time": 7.73309063911438
    },
    {
      "epoch": 3.298187255859375e-05,
      "model_forward_time": 2.4368014335632324,
      "step": 8646
    },
    {
      "epoch": 3.298187255859375e-05,
      "step": 8646,
      "training_step_time": 10.342832326889038
    },
    {
      "epoch": 3.2985687255859376e-05,
      "model_forward_time": 2.4555232524871826,
      "step": 8647
    },
    {
      "epoch": 3.2985687255859376e-05,
      "step": 8647,
      "training_step_time": 7.492759704589844
    },
    {
      "epoch": 3.2985687255859376e-05,
      "model_forward_time": 2.4337644577026367,
      "step": 8647
    },
    {
      "epoch": 3.2985687255859376e-05,
      "step": 8647,
      "training_step_time": 7.466077089309692
    },
    {
      "epoch": 3.2985687255859376e-05,
      "model_forward_time": 2.4305546283721924,
      "step": 8647
    },
    {
      "dynamics_loss_avg": 0.06553903855383396,
      "epoch": 3.2985687255859376e-05,
      "step": 8647
    },
    {
      "action_loss_avg": 0.022596035292372106,
      "epoch": 3.2985687255859376e-05,
      "step": 8647
    },
    {
      "epoch": 3.2985687255859376e-05,
      "step": 8647,
      "training_step_time": 7.467658758163452
    },
    {
      "epoch": 3.2985687255859376e-05,
      "model_forward_time": 2.4309041500091553,
      "step": 8647
    },
    {
      "epoch": 3.2985687255859376e-05,
      "step": 8647,
      "training_step_time": 10.022923469543457
    },
    {
      "epoch": 3.2989501953125e-05,
      "model_forward_time": 2.443189859390259,
      "step": 8648
    },
    {
      "epoch": 3.2989501953125e-05,
      "step": 8648,
      "training_step_time": 7.473736763000488
    },
    {
      "epoch": 3.2989501953125e-05,
      "model_forward_time": 2.4336414337158203,
      "step": 8648
    },
    {
      "epoch": 3.2989501953125e-05,
      "step": 8648,
      "training_step_time": 7.460221529006958
    },
    {
      "epoch": 3.2989501953125e-05,
      "model_forward_time": 2.4295716285705566,
      "step": 8648
    },
    {
      "epoch": 3.2989501953125e-05,
      "step": 8648,
      "training_step_time": 7.458791017532349
    },
    {
      "epoch": 3.2989501953125e-05,
      "model_forward_time": 3.1315908432006836,
      "step": 8648
    },
    {
      "epoch": 3.2989501953125e-05,
      "step": 8648,
      "training_step_time": 10.30393671989441
    },
    {
      "epoch": 3.299331665039062e-05,
      "model_forward_time": 2.4488110542297363,
      "step": 8649
    },
    {
      "epoch": 3.299331665039062e-05,
      "step": 8649,
      "training_step_time": 7.476800203323364
    },
    {
      "epoch": 3.299331665039062e-05,
      "model_forward_time": 3.385916233062744,
      "step": 8649
    },
    {
      "epoch": 3.299331665039062e-05,
      "step": 8649,
      "training_step_time": 10.155789375305176
    },
    {
      "epoch": 3.299331665039062e-05,
      "model_forward_time": 2.4340381622314453,
      "step": 8649
    },
    {
      "epoch": 3.299331665039062e-05,
      "step": 8649,
      "training_step_time": 7.4633259773254395
    },
    {
      "epoch": 3.299331665039062e-05,
      "model_forward_time": 3.1323585510253906,
      "step": 8649
    },
    {
      "epoch": 3.299331665039062e-05,
      "step": 8649,
      "training_step_time": 10.344350099563599
    },
    {
      "epoch": 3.299713134765625e-05,
      "grad_norm": 0.5260953307151794,
      "learning_rate": 1e-05,
      "loss": 0.0729,
      "step": 8650
    },
    {
      "epoch": 3.299713134765625e-05,
      "model_forward_time": 3.1503260135650635,
      "step": 8650
    },
    {
      "dynamics_loss_avg": 0.05939795784652233,
      "epoch": 3.299713134765625e-05,
      "step": 8650
    },
    {
      "action_loss_avg": 0.02349727097898722,
      "epoch": 3.299713134765625e-05,
      "step": 8650
    },
    {
      "epoch": 3.299713134765625e-05,
      "step": 8650,
      "training_step_time": 9.924660444259644
    },
    {
      "epoch": 3.299713134765625e-05,
      "model_forward_time": 2.4390342235565186,
      "step": 8650
    },
    {
      "epoch": 3.299713134765625e-05,
      "step": 8650,
      "training_step_time": 7.476996421813965
    },
    {
      "epoch": 3.299713134765625e-05,
      "model_forward_time": 2.709411382675171,
      "step": 8650
    },
    {
      "epoch": 3.299713134765625e-05,
      "step": 8650,
      "training_step_time": 7.738025188446045
    },
    {
      "epoch": 3.299713134765625e-05,
      "model_forward_time": 3.134658098220825,
      "step": 8650
    },
    {
      "epoch": 3.299713134765625e-05,
      "step": 8650,
      "training_step_time": 10.279210567474365
    },
    {
      "epoch": 3.3000946044921874e-05,
      "model_forward_time": 3.142164945602417,
      "step": 8651
    },
    {
      "epoch": 3.3000946044921874e-05,
      "step": 8651,
      "training_step_time": 9.912837266921997
    },
    {
      "epoch": 3.3000946044921874e-05,
      "model_forward_time": 3.129549741744995,
      "step": 8651
    },
    {
      "epoch": 3.3000946044921874e-05,
      "step": 8651,
      "training_step_time": 9.89956259727478
    },
    {
      "epoch": 3.3000946044921874e-05,
      "model_forward_time": 3.1305198669433594,
      "step": 8651
    },
    {
      "epoch": 3.3000946044921874e-05,
      "step": 8651,
      "training_step_time": 9.903459548950195
    },
    {
      "epoch": 3.3000946044921874e-05,
      "model_forward_time": 3.4429078102111816,
      "step": 8651
    },
    {
      "epoch": 3.3000946044921874e-05,
      "step": 8651,
      "training_step_time": 10.309210538864136
    },
    {
      "epoch": 3.30047607421875e-05,
      "model_forward_time": 2.4424967765808105,
      "step": 8652
    },
    {
      "epoch": 3.30047607421875e-05,
      "step": 8652,
      "training_step_time": 7.476702928543091
    },
    {
      "epoch": 3.30047607421875e-05,
      "model_forward_time": 3.145354986190796,
      "step": 8652
    },
    {
      "epoch": 3.30047607421875e-05,
      "step": 8652,
      "training_step_time": 9.92275881767273
    },
    {
      "epoch": 3.30047607421875e-05,
      "model_forward_time": 2.43876051902771,
      "step": 8652
    },
    {
      "dynamics_loss_avg": 0.06362605225294829,
      "epoch": 3.30047607421875e-05,
      "step": 8652
    },
    {
      "action_loss_avg": 0.02221549996174872,
      "epoch": 3.30047607421875e-05,
      "step": 8652
    },
    {
      "epoch": 3.30047607421875e-05,
      "step": 8652,
      "training_step_time": 7.4829254150390625
    },
    {
      "epoch": 3.30047607421875e-05,
      "model_forward_time": 3.156569480895996,
      "step": 8652
    },
    {
      "epoch": 3.30047607421875e-05,
      "step": 8652,
      "training_step_time": 10.255425453186035
    },
    {
      "epoch": 3.300857543945313e-05,
      "model_forward_time": 3.4569690227508545,
      "step": 8653
    },
    {
      "epoch": 3.300857543945313e-05,
      "step": 8653,
      "training_step_time": 10.244749784469604
    },
    {
      "epoch": 3.300857543945313e-05,
      "model_forward_time": 3.1529123783111572,
      "step": 8653
    },
    {
      "epoch": 3.300857543945313e-05,
      "step": 8653,
      "training_step_time": 9.926434993743896
    },
    {
      "epoch": 3.300857543945313e-05,
      "model_forward_time": 2.441465377807617,
      "step": 8653
    },
    {
      "epoch": 3.300857543945313e-05,
      "step": 8653,
      "training_step_time": 7.474946975708008
    },
    {
      "epoch": 3.300857543945313e-05,
      "model_forward_time": 3.1394238471984863,
      "step": 8653
    },
    {
      "epoch": 3.300857543945313e-05,
      "step": 8653,
      "training_step_time": 10.274156093597412
    },
    {
      "epoch": 3.301239013671875e-05,
      "model_forward_time": 3.16141939163208,
      "step": 8654
    },
    {
      "epoch": 3.301239013671875e-05,
      "step": 8654,
      "training_step_time": 9.931209325790405
    },
    {
      "epoch": 3.301239013671875e-05,
      "model_forward_time": 3.4700069427490234,
      "step": 8654
    },
    {
      "epoch": 3.301239013671875e-05,
      "step": 8654,
      "training_step_time": 10.238316059112549
    },
    {
      "epoch": 3.301239013671875e-05,
      "model_forward_time": 2.445129871368408,
      "step": 8654
    },
    {
      "epoch": 3.301239013671875e-05,
      "step": 8654,
      "training_step_time": 7.474634170532227
    },
    {
      "epoch": 3.301239013671875e-05,
      "model_forward_time": 2.4410860538482666,
      "step": 8654
    },
    {
      "epoch": 3.301239013671875e-05,
      "step": 8654,
      "training_step_time": 10.026808500289917
    },
    {
      "epoch": 3.301620483398437e-05,
      "model_forward_time": 3.1644439697265625,
      "step": 8655
    },
    {
      "dynamics_loss_avg": 0.05945933498442173,
      "epoch": 3.301620483398437e-05,
      "step": 8655
    },
    {
      "action_loss_avg": 0.026221491489559413,
      "epoch": 3.301620483398437e-05,
      "step": 8655
    },
    {
      "epoch": 3.301620483398437e-05,
      "step": 8655,
      "training_step_time": 9.946327924728394
    },
    {
      "epoch": 3.301620483398437e-05,
      "model_forward_time": 3.146387815475464,
      "step": 8655
    },
    {
      "epoch": 3.301620483398437e-05,
      "step": 8655,
      "training_step_time": 9.921530723571777
    },
    {
      "epoch": 3.301620483398437e-05,
      "model_forward_time": 3.1455442905426025,
      "step": 8655
    },
    {
      "epoch": 3.301620483398437e-05,
      "step": 8655,
      "training_step_time": 9.915165424346924
    },
    {
      "epoch": 3.301620483398437e-05,
      "model_forward_time": 2.758200168609619,
      "step": 8655
    },
    {
      "epoch": 3.301620483398437e-05,
      "step": 8655,
      "training_step_time": 10.26533555984497
    },
    {
      "epoch": 3.302001953125e-05,
      "model_forward_time": 2.4859907627105713,
      "step": 8656
    },
    {
      "epoch": 3.302001953125e-05,
      "step": 8656,
      "training_step_time": 7.5282793045043945
    },
    {
      "epoch": 3.302001953125e-05,
      "model_forward_time": 3.137789487838745,
      "step": 8656
    },
    {
      "epoch": 3.302001953125e-05,
      "step": 8656,
      "training_step_time": 9.90490984916687
    },
    {
      "epoch": 3.302001953125e-05,
      "model_forward_time": 3.146101951599121,
      "step": 8656
    },
    {
      "epoch": 3.302001953125e-05,
      "step": 8656,
      "training_step_time": 9.915799617767334
    },
    {
      "epoch": 3.302001953125e-05,
      "model_forward_time": 2.4457905292510986,
      "step": 8656
    },
    {
      "epoch": 3.302001953125e-05,
      "step": 8656,
      "training_step_time": 10.272208213806152
    },
    {
      "epoch": 3.3023834228515625e-05,
      "model_forward_time": 3.45723557472229,
      "step": 8657
    },
    {
      "epoch": 3.3023834228515625e-05,
      "step": 8657,
      "training_step_time": 10.238288164138794
    },
    {
      "epoch": 3.3023834228515625e-05,
      "model_forward_time": 2.440652370452881,
      "step": 8657
    },
    {
      "epoch": 3.3023834228515625e-05,
      "step": 8657,
      "training_step_time": 7.474962472915649
    },
    {
      "epoch": 3.3023834228515625e-05,
      "model_forward_time": 3.1448190212249756,
      "step": 8657
    },
    {
      "dynamics_loss_avg": 0.05494370125234127,
      "epoch": 3.3023834228515625e-05,
      "step": 8657
    },
    {
      "action_loss_avg": 0.018292689742520452,
      "epoch": 3.3023834228515625e-05,
      "step": 8657
    },
    {
      "epoch": 3.3023834228515625e-05,
      "step": 8657,
      "training_step_time": 9.924912691116333
    },
    {
      "epoch": 3.3023834228515625e-05,
      "model_forward_time": 3.146744728088379,
      "step": 8657
    },
    {
      "epoch": 3.3023834228515625e-05,
      "step": 8657,
      "training_step_time": 10.028770923614502
    },
    {
      "epoch": 3.302764892578125e-05,
      "model_forward_time": 3.1818618774414062,
      "step": 8658
    },
    {
      "epoch": 3.302764892578125e-05,
      "step": 8658,
      "training_step_time": 9.963578224182129
    },
    {
      "epoch": 3.302764892578125e-05,
      "model_forward_time": 3.441525936126709,
      "step": 8658
    },
    {
      "epoch": 3.302764892578125e-05,
      "step": 8658,
      "training_step_time": 10.216107368469238
    },
    {
      "epoch": 3.302764892578125e-05,
      "model_forward_time": 2.440741777420044,
      "step": 8658
    },
    {
      "epoch": 3.302764892578125e-05,
      "step": 8658,
      "training_step_time": 7.471968412399292
    },
    {
      "epoch": 3.302764892578125e-05,
      "model_forward_time": 3.140641927719116,
      "step": 8658
    },
    {
      "epoch": 3.302764892578125e-05,
      "step": 8658,
      "training_step_time": 10.313003778457642
    },
    {
      "epoch": 3.303146362304688e-05,
      "model_forward_time": 2.4695208072662354,
      "step": 8659
    },
    {
      "epoch": 3.303146362304688e-05,
      "step": 8659,
      "training_step_time": 7.4964635372161865
    },
    {
      "epoch": 3.303146362304688e-05,
      "model_forward_time": 2.431368827819824,
      "step": 8659
    },
    {
      "epoch": 3.303146362304688e-05,
      "step": 8659,
      "training_step_time": 7.456310033798218
    },
    {
      "epoch": 3.303146362304688e-05,
      "model_forward_time": 3.4195268154144287,
      "step": 8659
    },
    {
      "epoch": 3.303146362304688e-05,
      "step": 8659,
      "training_step_time": 10.197383403778076
    },
    {
      "epoch": 3.303146362304688e-05,
      "model_forward_time": 3.1396918296813965,
      "step": 8659
    },
    {
      "epoch": 3.303146362304688e-05,
      "step": 8659,
      "training_step_time": 10.273418664932251
    },
    {
      "epoch": 3.30352783203125e-05,
      "grad_norm": 0.29420363903045654,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 8660
    },
    {
      "epoch": 3.30352783203125e-05,
      "model_forward_time": 2.4519238471984863,
      "step": 8660
    },
    {
      "dynamics_loss_avg": 0.07597046848386527,
      "epoch": 3.30352783203125e-05,
      "step": 8660
    },
    {
      "action_loss_avg": 0.027676021307706834,
      "epoch": 3.30352783203125e-05,
      "step": 8660
    },
    {
      "epoch": 3.30352783203125e-05,
      "step": 8660,
      "training_step_time": 7.489043951034546
    },
    {
      "epoch": 3.30352783203125e-05,
      "model_forward_time": 3.1556503772735596,
      "step": 8660
    },
    {
      "epoch": 3.30352783203125e-05,
      "step": 8660,
      "training_step_time": 9.928515672683716
    },
    {
      "epoch": 3.30352783203125e-05,
      "model_forward_time": 3.1413486003875732,
      "step": 8660
    },
    {
      "epoch": 3.30352783203125e-05,
      "step": 8660,
      "training_step_time": 9.916711807250977
    },
    {
      "epoch": 3.30352783203125e-05,
      "model_forward_time": 2.4367640018463135,
      "step": 8660
    },
    {
      "epoch": 3.30352783203125e-05,
      "step": 8660,
      "training_step_time": 10.031188726425171
    },
    {
      "epoch": 3.3039093017578123e-05,
      "model_forward_time": 3.4901137351989746,
      "step": 8661
    },
    {
      "epoch": 3.3039093017578123e-05,
      "step": 8661,
      "training_step_time": 10.286221981048584
    },
    {
      "epoch": 3.3039093017578123e-05,
      "model_forward_time": 1.7474815845489502,
      "step": 8661
    },
    {
      "epoch": 3.3039093017578123e-05,
      "step": 8661,
      "training_step_time": 5.208916187286377
    },
    {
      "epoch": 3.3039093017578123e-05,
      "model_forward_time": 2.4305570125579834,
      "step": 8661
    },
    {
      "epoch": 3.3039093017578123e-05,
      "step": 8661,
      "training_step_time": 7.462998151779175
    },
    {
      "epoch": 3.3039093017578123e-05,
      "model_forward_time": 3.1375370025634766,
      "step": 8661
    },
    {
      "epoch": 3.3039093017578123e-05,
      "step": 8661,
      "training_step_time": 10.486803531646729
    },
    {
      "epoch": 3.304290771484375e-05,
      "model_forward_time": 3.150651693344116,
      "step": 8662
    },
    {
      "epoch": 3.304290771484375e-05,
      "step": 8662,
      "training_step_time": 9.931677341461182
    },
    {
      "epoch": 3.304290771484375e-05,
      "model_forward_time": 3.457273006439209,
      "step": 8662
    },
    {
      "epoch": 3.304290771484375e-05,
      "step": 8662,
      "training_step_time": 10.237987995147705
    },
    {
      "epoch": 3.304290771484375e-05,
      "model_forward_time": 2.4326584339141846,
      "step": 8662
    },
    {
      "dynamics_loss_avg": 0.042049098294228315,
      "epoch": 3.304290771484375e-05,
      "step": 8662
    },
    {
      "action_loss_avg": 0.014816803438588978,
      "epoch": 3.304290771484375e-05,
      "step": 8662
    },
    {
      "epoch": 3.304290771484375e-05,
      "step": 8662,
      "training_step_time": 7.4758689403533936
    },
    {
      "epoch": 3.304290771484375e-05,
      "model_forward_time": 2.431520938873291,
      "step": 8662
    },
    {
      "epoch": 3.304290771484375e-05,
      "step": 8662,
      "training_step_time": 10.02208161354065
    },
    {
      "epoch": 3.3046722412109376e-05,
      "model_forward_time": 3.1624228954315186,
      "step": 8663
    },
    {
      "epoch": 3.3046722412109376e-05,
      "step": 8663,
      "training_step_time": 9.940741777420044
    },
    {
      "epoch": 3.3046722412109376e-05,
      "model_forward_time": 2.4350574016571045,
      "step": 8663
    },
    {
      "epoch": 3.3046722412109376e-05,
      "step": 8663,
      "training_step_time": 7.471799850463867
    },
    {
      "epoch": 3.3046722412109376e-05,
      "model_forward_time": 3.4244120121002197,
      "step": 8663
    },
    {
      "epoch": 3.3046722412109376e-05,
      "step": 8663,
      "training_step_time": 10.201478719711304
    },
    {
      "epoch": 3.3046722412109376e-05,
      "model_forward_time": 2.4719905853271484,
      "step": 8663
    },
    {
      "epoch": 3.3046722412109376e-05,
      "step": 8663,
      "training_step_time": 10.024502277374268
    },
    {
      "epoch": 3.3050537109375e-05,
      "model_forward_time": 3.156893730163574,
      "step": 8664
    },
    {
      "epoch": 3.3050537109375e-05,
      "step": 8664,
      "training_step_time": 9.934911727905273
    },
    {
      "epoch": 3.3050537109375e-05,
      "model_forward_time": 2.43387508392334,
      "step": 8664
    },
    {
      "epoch": 3.3050537109375e-05,
      "step": 8664,
      "training_step_time": 7.471334934234619
    },
    {
      "epoch": 3.3050537109375e-05,
      "model_forward_time": 3.136101722717285,
      "step": 8664
    },
    {
      "epoch": 3.3050537109375e-05,
      "step": 8664,
      "training_step_time": 9.91850996017456
    },
    {
      "epoch": 3.3050537109375e-05,
      "model_forward_time": 2.763038396835327,
      "step": 8664
    },
    {
      "epoch": 3.3050537109375e-05,
      "step": 8664,
      "training_step_time": 10.247088432312012
    },
    {
      "epoch": 3.305435180664062e-05,
      "model_forward_time": 2.452782392501831,
      "step": 8665
    },
    {
      "dynamics_loss_avg": 0.05939690489321947,
      "epoch": 3.305435180664062e-05,
      "step": 8665
    },
    {
      "action_loss_avg": 0.022137654339894654,
      "epoch": 3.305435180664062e-05,
      "step": 8665
    },
    {
      "epoch": 3.305435180664062e-05,
      "step": 8665,
      "training_step_time": 7.493407249450684
    },
    {
      "epoch": 3.305435180664062e-05,
      "model_forward_time": 3.1384220123291016,
      "step": 8665
    },
    {
      "epoch": 3.305435180664062e-05,
      "step": 8665,
      "training_step_time": 9.915363311767578
    },
    {
      "epoch": 3.305435180664062e-05,
      "model_forward_time": 2.4330477714538574,
      "step": 8665
    },
    {
      "epoch": 3.305435180664062e-05,
      "step": 8665,
      "training_step_time": 7.465762615203857
    },
    {
      "epoch": 3.305435180664062e-05,
      "model_forward_time": 3.1354451179504395,
      "step": 8665
    },
    {
      "epoch": 3.305435180664062e-05,
      "step": 8665,
      "training_step_time": 10.042043447494507
    },
    {
      "epoch": 3.305816650390625e-05,
      "model_forward_time": 3.5056562423706055,
      "step": 8666
    },
    {
      "epoch": 3.305816650390625e-05,
      "step": 8666,
      "training_step_time": 10.2926185131073
    },
    {
      "epoch": 3.305816650390625e-05,
      "model_forward_time": 3.1394355297088623,
      "step": 8666
    },
    {
      "epoch": 3.305816650390625e-05,
      "step": 8666,
      "training_step_time": 9.92187213897705
    },
    {
      "epoch": 3.305816650390625e-05,
      "model_forward_time": 3.145143985748291,
      "step": 8666
    },
    {
      "epoch": 3.305816650390625e-05,
      "step": 8666,
      "training_step_time": 9.925888061523438
    },
    {
      "epoch": 3.305816650390625e-05,
      "model_forward_time": 3.1502134799957275,
      "step": 8666
    },
    {
      "epoch": 3.305816650390625e-05,
      "step": 8666,
      "training_step_time": 10.308752298355103
    },
    {
      "epoch": 3.3061981201171874e-05,
      "model_forward_time": 3.1547069549560547,
      "step": 8667
    },
    {
      "epoch": 3.3061981201171874e-05,
      "step": 8667,
      "training_step_time": 9.941935539245605
    },
    {
      "epoch": 3.3061981201171874e-05,
      "model_forward_time": 2.4380598068237305,
      "step": 8667
    },
    {
      "epoch": 3.3061981201171874e-05,
      "step": 8667,
      "training_step_time": 7.479635715484619
    },
    {
      "epoch": 3.3061981201171874e-05,
      "model_forward_time": 2.7393758296966553,
      "step": 8667
    },
    {
      "dynamics_loss_avg": 0.07350767701864243,
      "epoch": 3.3061981201171874e-05,
      "step": 8667
    },
    {
      "action_loss_avg": 0.03038159664720297,
      "epoch": 3.3061981201171874e-05,
      "step": 8667
    },
    {
      "epoch": 3.3061981201171874e-05,
      "step": 8667,
      "training_step_time": 7.772274732589722
    },
    {
      "epoch": 3.3061981201171874e-05,
      "model_forward_time": 2.4355525970458984,
      "step": 8667
    },
    {
      "epoch": 3.3061981201171874e-05,
      "step": 8667,
      "training_step_time": 10.309219598770142
    },
    {
      "epoch": 3.30657958984375e-05,
      "model_forward_time": 2.45660662651062,
      "step": 8668
    },
    {
      "epoch": 3.30657958984375e-05,
      "step": 8668,
      "training_step_time": 7.4984283447265625
    },
    {
      "epoch": 3.30657958984375e-05,
      "model_forward_time": 3.139531135559082,
      "step": 8668
    },
    {
      "epoch": 3.30657958984375e-05,
      "step": 8668,
      "training_step_time": 9.918156385421753
    },
    {
      "epoch": 3.30657958984375e-05,
      "model_forward_time": 2.4392614364624023,
      "step": 8668
    },
    {
      "epoch": 3.30657958984375e-05,
      "step": 8668,
      "training_step_time": 7.477029323577881
    },
    {
      "epoch": 3.30657958984375e-05,
      "model_forward_time": 2.7573280334472656,
      "step": 8668
    },
    {
      "epoch": 3.30657958984375e-05,
      "step": 8668,
      "training_step_time": 10.266900062561035
    },
    {
      "epoch": 3.306961059570313e-05,
      "model_forward_time": 3.143665075302124,
      "step": 8669
    },
    {
      "epoch": 3.306961059570313e-05,
      "step": 8669,
      "training_step_time": 9.927700519561768
    },
    {
      "epoch": 3.306961059570313e-05,
      "model_forward_time": 3.1410887241363525,
      "step": 8669
    },
    {
      "epoch": 3.306961059570313e-05,
      "step": 8669,
      "training_step_time": 9.914095878601074
    },
    {
      "epoch": 3.306961059570313e-05,
      "model_forward_time": 3.136049747467041,
      "step": 8669
    },
    {
      "epoch": 3.306961059570313e-05,
      "step": 8669,
      "training_step_time": 9.910920143127441
    },
    {
      "epoch": 3.306961059570313e-05,
      "model_forward_time": 3.1424176692962646,
      "step": 8669
    },
    {
      "epoch": 3.306961059570313e-05,
      "step": 8669,
      "training_step_time": 10.01563549041748
    },
    {
      "epoch": 3.307342529296875e-05,
      "grad_norm": 0.3653770685195923,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 8670
    },
    {
      "epoch": 3.307342529296875e-05,
      "model_forward_time": 2.7334671020507812,
      "step": 8670
    },
    {
      "dynamics_loss_avg": 0.04917521271854639,
      "epoch": 3.307342529296875e-05,
      "step": 8670
    },
    {
      "action_loss_avg": 0.016246188804507254,
      "epoch": 3.307342529296875e-05,
      "step": 8670
    },
    {
      "epoch": 3.307342529296875e-05,
      "step": 8670,
      "training_step_time": 7.773561239242554
    },
    {
      "epoch": 3.307342529296875e-05,
      "model_forward_time": 3.137852430343628,
      "step": 8670
    },
    {
      "epoch": 3.307342529296875e-05,
      "step": 8670,
      "training_step_time": 9.918951034545898
    },
    {
      "epoch": 3.307342529296875e-05,
      "model_forward_time": 3.1337385177612305,
      "step": 8670
    },
    {
      "epoch": 3.307342529296875e-05,
      "step": 8670,
      "training_step_time": 9.91737675666809
    },
    {
      "epoch": 3.307342529296875e-05,
      "model_forward_time": 3.1409852504730225,
      "step": 8670
    },
    {
      "epoch": 3.307342529296875e-05,
      "step": 8670,
      "training_step_time": 10.29625940322876
    },
    {
      "epoch": 3.307723999023437e-05,
      "model_forward_time": 3.1524932384490967,
      "step": 8671
    },
    {
      "epoch": 3.307723999023437e-05,
      "step": 8671,
      "training_step_time": 9.9335036277771
    },
    {
      "epoch": 3.307723999023437e-05,
      "model_forward_time": 3.452416181564331,
      "step": 8671
    },
    {
      "epoch": 3.307723999023437e-05,
      "step": 8671,
      "training_step_time": 10.234241724014282
    },
    {
      "epoch": 3.307723999023437e-05,
      "model_forward_time": 3.1400814056396484,
      "step": 8671
    },
    {
      "epoch": 3.307723999023437e-05,
      "step": 8671,
      "training_step_time": 9.922444343566895
    },
    {
      "epoch": 3.307723999023437e-05,
      "model_forward_time": 3.139827251434326,
      "step": 8671
    },
    {
      "epoch": 3.307723999023437e-05,
      "step": 8671,
      "training_step_time": 10.301107168197632
    },
    {
      "epoch": 3.30810546875e-05,
      "model_forward_time": 3.1693832874298096,
      "step": 8672
    },
    {
      "epoch": 3.30810546875e-05,
      "step": 8672,
      "training_step_time": 9.95576786994934
    },
    {
      "epoch": 3.30810546875e-05,
      "model_forward_time": 2.4352593421936035,
      "step": 8672
    },
    {
      "epoch": 3.30810546875e-05,
      "step": 8672,
      "training_step_time": 7.465421915054321
    },
    {
      "epoch": 3.30810546875e-05,
      "model_forward_time": 3.4322264194488525,
      "step": 8672
    },
    {
      "dynamics_loss_avg": 0.04880406409502029,
      "epoch": 3.30810546875e-05,
      "step": 8672
    },
    {
      "action_loss_avg": 0.023440363071858882,
      "epoch": 3.30810546875e-05,
      "step": 8672
    },
    {
      "epoch": 3.30810546875e-05,
      "step": 8672,
      "training_step_time": 10.2140052318573
    },
    {
      "epoch": 3.30810546875e-05,
      "model_forward_time": 2.4432525634765625,
      "step": 8672
    },
    {
      "epoch": 3.30810546875e-05,
      "step": 8672,
      "training_step_time": 10.289682388305664
    },
    {
      "epoch": 3.3084869384765625e-05,
      "model_forward_time": 3.1455535888671875,
      "step": 8673
    },
    {
      "epoch": 3.3084869384765625e-05,
      "step": 8673,
      "training_step_time": 9.921108722686768
    },
    {
      "epoch": 3.3084869384765625e-05,
      "model_forward_time": 3.1282174587249756,
      "step": 8673
    },
    {
      "epoch": 3.3084869384765625e-05,
      "step": 8673,
      "training_step_time": 9.901578187942505
    },
    {
      "epoch": 3.3084869384765625e-05,
      "model_forward_time": 2.4268929958343506,
      "step": 8673
    },
    {
      "epoch": 3.3084869384765625e-05,
      "step": 8673,
      "training_step_time": 7.454767465591431
    },
    {
      "epoch": 3.3084869384765625e-05,
      "model_forward_time": 3.4171624183654785,
      "step": 8673
    },
    {
      "epoch": 3.3084869384765625e-05,
      "step": 8673,
      "training_step_time": 10.283613681793213
    },
    {
      "epoch": 3.308868408203125e-05,
      "model_forward_time": 3.160888671875,
      "step": 8674
    },
    {
      "epoch": 3.308868408203125e-05,
      "step": 8674,
      "training_step_time": 9.939750671386719
    },
    {
      "epoch": 3.308868408203125e-05,
      "model_forward_time": 2.4382126331329346,
      "step": 8674
    },
    {
      "epoch": 3.308868408203125e-05,
      "step": 8674,
      "training_step_time": 7.467126846313477
    },
    {
      "epoch": 3.308868408203125e-05,
      "model_forward_time": 2.4305129051208496,
      "step": 8674
    },
    {
      "epoch": 3.308868408203125e-05,
      "step": 8674,
      "training_step_time": 7.456685543060303
    },
    {
      "epoch": 3.308868408203125e-05,
      "model_forward_time": 3.135819911956787,
      "step": 8674
    },
    {
      "epoch": 3.308868408203125e-05,
      "step": 8674,
      "training_step_time": 10.018064022064209
    },
    {
      "epoch": 3.309249877929688e-05,
      "model_forward_time": 2.4522430896759033,
      "step": 8675
    },
    {
      "dynamics_loss_avg": 0.06300459336489439,
      "epoch": 3.309249877929688e-05,
      "step": 8675
    },
    {
      "action_loss_avg": 0.02054919060319662,
      "epoch": 3.309249877929688e-05,
      "step": 8675
    },
    {
      "epoch": 3.309249877929688e-05,
      "step": 8675,
      "training_step_time": 7.489875555038452
    },
    {
      "epoch": 3.309249877929688e-05,
      "model_forward_time": 3.4190263748168945,
      "step": 8675
    },
    {
      "epoch": 3.309249877929688e-05,
      "step": 8675,
      "training_step_time": 10.193352222442627
    },
    {
      "epoch": 3.309249877929688e-05,
      "model_forward_time": 2.4412841796875,
      "step": 8675
    },
    {
      "epoch": 3.309249877929688e-05,
      "step": 8675,
      "training_step_time": 7.471449851989746
    },
    {
      "epoch": 3.309249877929688e-05,
      "model_forward_time": 3.133683204650879,
      "step": 8675
    },
    {
      "epoch": 3.309249877929688e-05,
      "step": 8675,
      "training_step_time": 10.24642276763916
    },
    {
      "epoch": 3.30963134765625e-05,
      "model_forward_time": 3.158073902130127,
      "step": 8676
    },
    {
      "epoch": 3.30963134765625e-05,
      "step": 8676,
      "training_step_time": 9.927651643753052
    },
    {
      "epoch": 3.30963134765625e-05,
      "model_forward_time": 3.139103889465332,
      "step": 8676
    },
    {
      "epoch": 3.30963134765625e-05,
      "step": 8676,
      "training_step_time": 9.910183668136597
    },
    {
      "epoch": 3.30963134765625e-05,
      "model_forward_time": 3.4179306030273438,
      "step": 8676
    },
    {
      "epoch": 3.30963134765625e-05,
      "step": 8676,
      "training_step_time": 10.194267511367798
    },
    {
      "epoch": 3.30963134765625e-05,
      "model_forward_time": 3.140245199203491,
      "step": 8676
    },
    {
      "epoch": 3.30963134765625e-05,
      "step": 8676,
      "training_step_time": 10.24893569946289
    },
    {
      "epoch": 3.3100128173828123e-05,
      "model_forward_time": 2.450286626815796,
      "step": 8677
    },
    {
      "epoch": 3.3100128173828123e-05,
      "step": 8677,
      "training_step_time": 7.481999635696411
    },
    {
      "epoch": 3.3100128173828123e-05,
      "model_forward_time": 3.136516571044922,
      "step": 8677
    },
    {
      "epoch": 3.3100128173828123e-05,
      "step": 8677,
      "training_step_time": 9.927793979644775
    },
    {
      "epoch": 3.3100128173828123e-05,
      "model_forward_time": 3.1397697925567627,
      "step": 8677
    },
    {
      "dynamics_loss_avg": 0.06648709774017333,
      "epoch": 3.3100128173828123e-05,
      "step": 8677
    },
    {
      "action_loss_avg": 0.01857788781635463,
      "epoch": 3.3100128173828123e-05,
      "step": 8677
    },
    {
      "epoch": 3.3100128173828123e-05,
      "step": 8677,
      "training_step_time": 10.208905458450317
    },
    {
      "epoch": 3.3100128173828123e-05,
      "model_forward_time": 2.44124436378479,
      "step": 8677
    },
    {
      "epoch": 3.3100128173828123e-05,
      "step": 8677,
      "training_step_time": 10.34306025505066
    },
    {
      "epoch": 3.310394287109375e-05,
      "model_forward_time": 2.4709889888763428,
      "step": 8678
    },
    {
      "epoch": 3.310394287109375e-05,
      "step": 8678,
      "training_step_time": 7.5264623165130615
    },
    {
      "epoch": 3.310394287109375e-05,
      "model_forward_time": 3.1409151554107666,
      "step": 8678
    },
    {
      "epoch": 3.310394287109375e-05,
      "step": 8678,
      "training_step_time": 9.925080299377441
    },
    {
      "epoch": 3.310394287109375e-05,
      "model_forward_time": 2.438563585281372,
      "step": 8678
    },
    {
      "epoch": 3.310394287109375e-05,
      "step": 8678,
      "training_step_time": 7.478337526321411
    },
    {
      "epoch": 3.310394287109375e-05,
      "model_forward_time": 3.142674684524536,
      "step": 8678
    },
    {
      "epoch": 3.310394287109375e-05,
      "step": 8678,
      "training_step_time": 10.311386346817017
    },
    {
      "epoch": 3.3107757568359376e-05,
      "model_forward_time": 2.7792205810546875,
      "step": 8679
    },
    {
      "epoch": 3.3107757568359376e-05,
      "step": 8679,
      "training_step_time": 7.826779365539551
    },
    {
      "epoch": 3.3107757568359376e-05,
      "model_forward_time": 3.147998094558716,
      "step": 8679
    },
    {
      "epoch": 3.3107757568359376e-05,
      "step": 8679,
      "training_step_time": 9.93002700805664
    },
    {
      "epoch": 3.3107757568359376e-05,
      "model_forward_time": 2.4495019912719727,
      "step": 8679
    },
    {
      "epoch": 3.3107757568359376e-05,
      "step": 8679,
      "training_step_time": 7.486573696136475
    },
    {
      "epoch": 3.3107757568359376e-05,
      "model_forward_time": 3.142212152481079,
      "step": 8679
    },
    {
      "epoch": 3.3107757568359376e-05,
      "step": 8679,
      "training_step_time": 10.306955337524414
    },
    {
      "epoch": 3.3111572265625e-05,
      "grad_norm": 0.5879446268081665,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 8680
    },
    {
      "epoch": 3.3111572265625e-05,
      "model_forward_time": 3.156346082687378,
      "step": 8680
    },
    {
      "dynamics_loss_avg": 0.0567069185897708,
      "epoch": 3.3111572265625e-05,
      "step": 8680
    },
    {
      "action_loss_avg": 0.020130656938999892,
      "epoch": 3.3111572265625e-05,
      "step": 8680
    },
    {
      "epoch": 3.3111572265625e-05,
      "step": 8680,
      "training_step_time": 9.938523292541504
    },
    {
      "epoch": 3.3111572265625e-05,
      "model_forward_time": 3.4588382244110107,
      "step": 8680
    },
    {
      "epoch": 3.3111572265625e-05,
      "step": 8680,
      "training_step_time": 10.229288339614868
    },
    {
      "epoch": 3.3111572265625e-05,
      "model_forward_time": 2.43646240234375,
      "step": 8680
    },
    {
      "epoch": 3.3111572265625e-05,
      "step": 8680,
      "training_step_time": 7.464323043823242
    },
    {
      "epoch": 3.3111572265625e-05,
      "model_forward_time": 3.1346092224121094,
      "step": 8680
    },
    {
      "epoch": 3.3111572265625e-05,
      "step": 8680,
      "training_step_time": 10.013229846954346
    },
    {
      "epoch": 3.311538696289062e-05,
      "model_forward_time": 3.1491737365722656,
      "step": 8681
    },
    {
      "epoch": 3.311538696289062e-05,
      "step": 8681,
      "training_step_time": 9.921554327011108
    },
    {
      "epoch": 3.311538696289062e-05,
      "model_forward_time": 2.43391752243042,
      "step": 8681
    },
    {
      "epoch": 3.311538696289062e-05,
      "step": 8681,
      "training_step_time": 7.461110830307007
    },
    {
      "epoch": 3.311538696289062e-05,
      "model_forward_time": 2.7209725379943848,
      "step": 8681
    },
    {
      "epoch": 3.311538696289062e-05,
      "step": 8681,
      "training_step_time": 7.746283769607544
    },
    {
      "epoch": 3.311538696289062e-05,
      "model_forward_time": 3.134852647781372,
      "step": 8681
    },
    {
      "epoch": 3.311538696289062e-05,
      "step": 8681,
      "training_step_time": 10.273975133895874
    },
    {
      "epoch": 3.311920166015625e-05,
      "model_forward_time": 3.156791925430298,
      "step": 8682
    },
    {
      "epoch": 3.311920166015625e-05,
      "step": 8682,
      "training_step_time": 9.937256336212158
    },
    {
      "epoch": 3.311920166015625e-05,
      "model_forward_time": 2.433535099029541,
      "step": 8682
    },
    {
      "epoch": 3.311920166015625e-05,
      "step": 8682,
      "training_step_time": 7.460558652877808
    },
    {
      "epoch": 3.311920166015625e-05,
      "model_forward_time": 2.437997341156006,
      "step": 8682
    },
    {
      "dynamics_loss_avg": 0.05897158831357956,
      "epoch": 3.311920166015625e-05,
      "step": 8682
    },
    {
      "action_loss_avg": 0.0209413162432611,
      "epoch": 3.311920166015625e-05,
      "step": 8682
    },
    {
      "epoch": 3.311920166015625e-05,
      "step": 8682,
      "training_step_time": 7.462658405303955
    },
    {
      "epoch": 3.311920166015625e-05,
      "model_forward_time": 3.453803777694702,
      "step": 8682
    },
    {
      "epoch": 3.311920166015625e-05,
      "step": 8682,
      "training_step_time": 10.31505799293518
    },
    {
      "epoch": 3.3123016357421874e-05,
      "model_forward_time": 3.1462271213531494,
      "step": 8683
    },
    {
      "epoch": 3.3123016357421874e-05,
      "step": 8683,
      "training_step_time": 9.91703748703003
    },
    {
      "epoch": 3.3123016357421874e-05,
      "model_forward_time": 3.1286239624023438,
      "step": 8683
    },
    {
      "epoch": 3.3123016357421874e-05,
      "step": 8683,
      "training_step_time": 9.89641261100769
    },
    {
      "epoch": 3.3123016357421874e-05,
      "model_forward_time": 3.1401124000549316,
      "step": 8683
    },
    {
      "epoch": 3.3123016357421874e-05,
      "step": 8683,
      "training_step_time": 9.912696123123169
    },
    {
      "epoch": 3.3123016357421874e-05,
      "model_forward_time": 3.13600492477417,
      "step": 8683
    },
    {
      "epoch": 3.3123016357421874e-05,
      "step": 8683,
      "training_step_time": 10.28470778465271
    },
    {
      "epoch": 3.31268310546875e-05,
      "model_forward_time": 2.4573299884796143,
      "step": 8684
    },
    {
      "epoch": 3.31268310546875e-05,
      "step": 8684,
      "training_step_time": 7.489357948303223
    },
    {
      "epoch": 3.31268310546875e-05,
      "model_forward_time": 2.7454123497009277,
      "step": 8684
    },
    {
      "epoch": 3.31268310546875e-05,
      "step": 8684,
      "training_step_time": 7.772265911102295
    },
    {
      "epoch": 3.31268310546875e-05,
      "model_forward_time": 1.745795726776123,
      "step": 8684
    },
    {
      "epoch": 3.31268310546875e-05,
      "step": 8684,
      "training_step_time": 5.201867580413818
    },
    {
      "epoch": 3.31268310546875e-05,
      "model_forward_time": 2.4307057857513428,
      "step": 8684
    },
    {
      "epoch": 3.31268310546875e-05,
      "step": 8684,
      "training_step_time": 10.315349578857422
    },
    {
      "epoch": 3.313064575195313e-05,
      "model_forward_time": 3.1544063091278076,
      "step": 8685
    },
    {
      "dynamics_loss_avg": 0.06207293476909399,
      "epoch": 3.313064575195313e-05,
      "step": 8685
    },
    {
      "action_loss_avg": 0.027975863963365554,
      "epoch": 3.313064575195313e-05,
      "step": 8685
    },
    {
      "epoch": 3.313064575195313e-05,
      "step": 8685,
      "training_step_time": 9.926580667495728
    },
    {
      "epoch": 3.313064575195313e-05,
      "model_forward_time": 3.13966703414917,
      "step": 8685
    },
    {
      "epoch": 3.313064575195313e-05,
      "step": 8685,
      "training_step_time": 9.91468095779419
    },
    {
      "epoch": 3.313064575195313e-05,
      "model_forward_time": 3.4294490814208984,
      "step": 8685
    },
    {
      "epoch": 3.313064575195313e-05,
      "step": 8685,
      "training_step_time": 10.203788995742798
    },
    {
      "epoch": 3.313064575195313e-05,
      "model_forward_time": 2.439979076385498,
      "step": 8685
    },
    {
      "epoch": 3.313064575195313e-05,
      "step": 8685,
      "training_step_time": 10.040722131729126
    },
    {
      "epoch": 3.313446044921875e-05,
      "model_forward_time": 3.1605687141418457,
      "step": 8686
    },
    {
      "epoch": 3.313446044921875e-05,
      "step": 8686,
      "training_step_time": 9.933629512786865
    },
    {
      "epoch": 3.313446044921875e-05,
      "model_forward_time": 2.4315173625946045,
      "step": 8686
    },
    {
      "epoch": 3.313446044921875e-05,
      "step": 8686,
      "training_step_time": 7.457471609115601
    },
    {
      "epoch": 3.313446044921875e-05,
      "model_forward_time": 2.438425064086914,
      "step": 8686
    },
    {
      "epoch": 3.313446044921875e-05,
      "step": 8686,
      "training_step_time": 7.463054180145264
    },
    {
      "epoch": 3.313446044921875e-05,
      "model_forward_time": 2.7161338329315186,
      "step": 8686
    },
    {
      "epoch": 3.313446044921875e-05,
      "step": 8686,
      "training_step_time": 10.258155822753906
    },
    {
      "epoch": 3.313827514648437e-05,
      "model_forward_time": 2.4449431896209717,
      "step": 8687
    },
    {
      "epoch": 3.313827514648437e-05,
      "step": 8687,
      "training_step_time": 7.466914892196655
    },
    {
      "epoch": 3.313827514648437e-05,
      "model_forward_time": 3.1379663944244385,
      "step": 8687
    },
    {
      "epoch": 3.313827514648437e-05,
      "step": 8687,
      "training_step_time": 9.900592803955078
    },
    {
      "epoch": 3.313827514648437e-05,
      "model_forward_time": 3.1364505290985107,
      "step": 8687
    },
    {
      "dynamics_loss_avg": 0.0532753630541265,
      "epoch": 3.313827514648437e-05,
      "step": 8687
    },
    {
      "action_loss_avg": 0.01940359752625227,
      "epoch": 3.313827514648437e-05,
      "step": 8687
    },
    {
      "epoch": 3.313827514648437e-05,
      "step": 8687,
      "training_step_time": 9.907087326049805
    },
    {
      "epoch": 3.313827514648437e-05,
      "model_forward_time": 2.431945323944092,
      "step": 8687
    },
    {
      "epoch": 3.313827514648437e-05,
      "step": 8687,
      "training_step_time": 10.28116250038147
    },
    {
      "epoch": 3.314208984375e-05,
      "model_forward_time": 2.445955276489258,
      "step": 8688
    },
    {
      "epoch": 3.314208984375e-05,
      "step": 8688,
      "training_step_time": 7.472413778305054
    },
    {
      "epoch": 3.314208984375e-05,
      "model_forward_time": 3.420830726623535,
      "step": 8688
    },
    {
      "epoch": 3.314208984375e-05,
      "step": 8688,
      "training_step_time": 10.190858602523804
    },
    {
      "epoch": 3.314208984375e-05,
      "model_forward_time": 3.135781764984131,
      "step": 8688
    },
    {
      "epoch": 3.314208984375e-05,
      "step": 8688,
      "training_step_time": 9.908156156539917
    },
    {
      "epoch": 3.314208984375e-05,
      "model_forward_time": 2.4334075450897217,
      "step": 8688
    },
    {
      "epoch": 3.314208984375e-05,
      "step": 8688,
      "training_step_time": 10.261332988739014
    },
    {
      "epoch": 3.3145904541015625e-05,
      "model_forward_time": 2.446521043777466,
      "step": 8689
    },
    {
      "epoch": 3.3145904541015625e-05,
      "step": 8689,
      "training_step_time": 7.474806547164917
    },
    {
      "epoch": 3.3145904541015625e-05,
      "model_forward_time": 3.128572940826416,
      "step": 8689
    },
    {
      "epoch": 3.3145904541015625e-05,
      "step": 8689,
      "training_step_time": 9.898743867874146
    },
    {
      "epoch": 3.3145904541015625e-05,
      "model_forward_time": 3.4105379581451416,
      "step": 8689
    },
    {
      "epoch": 3.3145904541015625e-05,
      "step": 8689,
      "training_step_time": 10.180745363235474
    },
    {
      "epoch": 3.3145904541015625e-05,
      "model_forward_time": 2.4297940731048584,
      "step": 8689
    },
    {
      "epoch": 3.3145904541015625e-05,
      "step": 8689,
      "training_step_time": 10.295607805252075
    },
    {
      "epoch": 3.314971923828125e-05,
      "grad_norm": 0.7803246378898621,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 8690
    },
    {
      "epoch": 3.314971923828125e-05,
      "model_forward_time": 2.450908660888672,
      "step": 8690
    },
    {
      "dynamics_loss_avg": 0.06999237500131131,
      "epoch": 3.314971923828125e-05,
      "step": 8690
    },
    {
      "action_loss_avg": 0.02643125122413039,
      "epoch": 3.314971923828125e-05,
      "step": 8690
    },
    {
      "epoch": 3.314971923828125e-05,
      "step": 8690,
      "training_step_time": 7.478912830352783
    },
    {
      "epoch": 3.314971923828125e-05,
      "model_forward_time": 2.4275968074798584,
      "step": 8690
    },
    {
      "epoch": 3.314971923828125e-05,
      "step": 8690,
      "training_step_time": 7.449197053909302
    },
    {
      "epoch": 3.314971923828125e-05,
      "model_forward_time": 2.4324398040771484,
      "step": 8690
    },
    {
      "epoch": 3.314971923828125e-05,
      "step": 8690,
      "training_step_time": 7.455173015594482
    },
    {
      "epoch": 3.314971923828125e-05,
      "model_forward_time": 2.702064037322998,
      "step": 8690
    },
    {
      "epoch": 3.314971923828125e-05,
      "step": 8690,
      "training_step_time": 10.033303022384644
    },
    {
      "epoch": 3.315353393554688e-05,
      "model_forward_time": 2.4632411003112793,
      "step": 8691
    },
    {
      "epoch": 3.315353393554688e-05,
      "step": 8691,
      "training_step_time": 7.489509582519531
    },
    {
      "epoch": 3.315353393554688e-05,
      "model_forward_time": 1.7476110458374023,
      "step": 8691
    },
    {
      "epoch": 3.315353393554688e-05,
      "step": 8691,
      "training_step_time": 5.200169324874878
    },
    {
      "epoch": 3.315353393554688e-05,
      "model_forward_time": 2.436383008956909,
      "step": 8691
    },
    {
      "epoch": 3.315353393554688e-05,
      "step": 8691,
      "training_step_time": 7.458251953125
    },
    {
      "epoch": 3.315353393554688e-05,
      "model_forward_time": 2.4340591430664062,
      "step": 8691
    },
    {
      "epoch": 3.315353393554688e-05,
      "step": 8691,
      "training_step_time": 10.266246795654297
    },
    {
      "epoch": 3.31573486328125e-05,
      "model_forward_time": 3.150794506072998,
      "step": 8692
    },
    {
      "epoch": 3.31573486328125e-05,
      "step": 8692,
      "training_step_time": 9.912702798843384
    },
    {
      "epoch": 3.31573486328125e-05,
      "model_forward_time": 3.3958656787872314,
      "step": 8692
    },
    {
      "epoch": 3.31573486328125e-05,
      "step": 8692,
      "training_step_time": 10.162662982940674
    },
    {
      "epoch": 3.31573486328125e-05,
      "model_forward_time": 3.1429319381713867,
      "step": 8692
    },
    {
      "dynamics_loss_avg": 0.05126238083466887,
      "epoch": 3.31573486328125e-05,
      "step": 8692
    },
    {
      "action_loss_avg": 0.020759571716189383,
      "epoch": 3.31573486328125e-05,
      "step": 8692
    },
    {
      "epoch": 3.31573486328125e-05,
      "step": 8692,
      "training_step_time": 9.909227848052979
    },
    {
      "epoch": 3.31573486328125e-05,
      "model_forward_time": 2.4389400482177734,
      "step": 8692
    },
    {
      "epoch": 3.31573486328125e-05,
      "step": 8692,
      "training_step_time": 10.311131715774536
    },
    {
      "epoch": 3.3161163330078124e-05,
      "model_forward_time": 3.1519010066986084,
      "step": 8693
    },
    {
      "epoch": 3.3161163330078124e-05,
      "step": 8693,
      "training_step_time": 9.91947627067566
    },
    {
      "epoch": 3.3161163330078124e-05,
      "model_forward_time": 2.4397740364074707,
      "step": 8693
    },
    {
      "epoch": 3.3161163330078124e-05,
      "step": 8693,
      "training_step_time": 7.464071750640869
    },
    {
      "epoch": 3.3161163330078124e-05,
      "model_forward_time": 2.7167489528656006,
      "step": 8693
    },
    {
      "epoch": 3.3161163330078124e-05,
      "step": 8693,
      "training_step_time": 7.741464138031006
    },
    {
      "epoch": 3.3161163330078124e-05,
      "model_forward_time": 2.435492515563965,
      "step": 8693
    },
    {
      "epoch": 3.3161163330078124e-05,
      "step": 8693,
      "training_step_time": 10.282565116882324
    },
    {
      "epoch": 3.316497802734375e-05,
      "model_forward_time": 2.4531307220458984,
      "step": 8694
    },
    {
      "epoch": 3.316497802734375e-05,
      "step": 8694,
      "training_step_time": 7.474351406097412
    },
    {
      "epoch": 3.316497802734375e-05,
      "model_forward_time": 2.439699172973633,
      "step": 8694
    },
    {
      "epoch": 3.316497802734375e-05,
      "step": 8694,
      "training_step_time": 7.46101975440979
    },
    {
      "epoch": 3.316497802734375e-05,
      "model_forward_time": 3.1383743286132812,
      "step": 8694
    },
    {
      "epoch": 3.316497802734375e-05,
      "step": 8694,
      "training_step_time": 9.899993896484375
    },
    {
      "epoch": 3.316497802734375e-05,
      "model_forward_time": 2.722182512283325,
      "step": 8694
    },
    {
      "epoch": 3.316497802734375e-05,
      "step": 8694,
      "training_step_time": 10.310692310333252
    },
    {
      "epoch": 3.3168792724609376e-05,
      "model_forward_time": 3.154411792755127,
      "step": 8695
    },
    {
      "dynamics_loss_avg": 0.053149789199233057,
      "epoch": 3.3168792724609376e-05,
      "step": 8695
    },
    {
      "action_loss_avg": 0.018061305675655604,
      "epoch": 3.3168792724609376e-05,
      "step": 8695
    },
    {
      "epoch": 3.3168792724609376e-05,
      "step": 8695,
      "training_step_time": 9.922391176223755
    },
    {
      "epoch": 3.3168792724609376e-05,
      "model_forward_time": 3.144101619720459,
      "step": 8695
    },
    {
      "epoch": 3.3168792724609376e-05,
      "step": 8695,
      "training_step_time": 9.909833192825317
    },
    {
      "epoch": 3.3168792724609376e-05,
      "model_forward_time": 2.4385273456573486,
      "step": 8695
    },
    {
      "epoch": 3.3168792724609376e-05,
      "step": 8695,
      "training_step_time": 7.458838701248169
    },
    {
      "epoch": 3.3168792724609376e-05,
      "model_forward_time": 3.1428658962249756,
      "step": 8695
    },
    {
      "epoch": 3.3168792724609376e-05,
      "step": 8695,
      "training_step_time": 10.01910924911499
    },
    {
      "epoch": 3.3172607421875e-05,
      "model_forward_time": 3.161407232284546,
      "step": 8696
    },
    {
      "epoch": 3.3172607421875e-05,
      "step": 8696,
      "training_step_time": 9.93565559387207
    },
    {
      "epoch": 3.3172607421875e-05,
      "model_forward_time": 3.438674211502075,
      "step": 8696
    },
    {
      "epoch": 3.3172607421875e-05,
      "step": 8696,
      "training_step_time": 10.210897207260132
    },
    {
      "epoch": 3.3172607421875e-05,
      "model_forward_time": 2.4424076080322266,
      "step": 8696
    },
    {
      "epoch": 3.3172607421875e-05,
      "step": 8696,
      "training_step_time": 7.475823163986206
    },
    {
      "epoch": 3.3172607421875e-05,
      "model_forward_time": 3.1458096504211426,
      "step": 8696
    },
    {
      "epoch": 3.3172607421875e-05,
      "step": 8696,
      "training_step_time": 10.281229734420776
    },
    {
      "epoch": 3.317642211914062e-05,
      "model_forward_time": 3.1667935848236084,
      "step": 8697
    },
    {
      "epoch": 3.317642211914062e-05,
      "step": 8697,
      "training_step_time": 9.951166868209839
    },
    {
      "epoch": 3.317642211914062e-05,
      "model_forward_time": 3.141676425933838,
      "step": 8697
    },
    {
      "epoch": 3.317642211914062e-05,
      "step": 8697,
      "training_step_time": 10.22761583328247
    },
    {
      "epoch": 3.317642211914062e-05,
      "model_forward_time": 3.153043270111084,
      "step": 8697
    },
    {
      "dynamics_loss_avg": 0.06283250506967306,
      "epoch": 3.317642211914062e-05,
      "step": 8697
    },
    {
      "action_loss_avg": 0.0214033009018749,
      "epoch": 3.317642211914062e-05,
      "step": 8697
    },
    {
      "epoch": 3.317642211914062e-05,
      "step": 8697,
      "training_step_time": 9.944019317626953
    },
    {
      "epoch": 3.317642211914062e-05,
      "model_forward_time": 3.1487014293670654,
      "step": 8697
    },
    {
      "epoch": 3.317642211914062e-05,
      "step": 8697,
      "training_step_time": 10.303079605102539
    },
    {
      "epoch": 3.318023681640625e-05,
      "model_forward_time": 3.1593661308288574,
      "step": 8698
    },
    {
      "epoch": 3.318023681640625e-05,
      "step": 8698,
      "training_step_time": 9.939383745193481
    },
    {
      "epoch": 3.318023681640625e-05,
      "model_forward_time": 3.1362924575805664,
      "step": 8698
    },
    {
      "epoch": 3.318023681640625e-05,
      "step": 8698,
      "training_step_time": 9.907456398010254
    },
    {
      "epoch": 3.318023681640625e-05,
      "model_forward_time": 2.4323863983154297,
      "step": 8698
    },
    {
      "epoch": 3.318023681640625e-05,
      "step": 8698,
      "training_step_time": 7.771379470825195
    },
    {
      "epoch": 3.318023681640625e-05,
      "model_forward_time": 2.433319091796875,
      "step": 8698
    },
    {
      "epoch": 3.318023681640625e-05,
      "step": 8698,
      "training_step_time": 10.27589225769043
    },
    {
      "epoch": 3.3184051513671874e-05,
      "model_forward_time": 2.4495060443878174,
      "step": 8699
    },
    {
      "epoch": 3.3184051513671874e-05,
      "step": 8699,
      "training_step_time": 7.481051683425903
    },
    {
      "epoch": 3.3184051513671874e-05,
      "model_forward_time": 3.136028528213501,
      "step": 8699
    },
    {
      "epoch": 3.3184051513671874e-05,
      "step": 8699,
      "training_step_time": 9.908869981765747
    },
    {
      "epoch": 3.3184051513671874e-05,
      "model_forward_time": 3.1474404335021973,
      "step": 8699
    },
    {
      "epoch": 3.3184051513671874e-05,
      "step": 8699,
      "training_step_time": 9.920454502105713
    },
    {
      "epoch": 3.3184051513671874e-05,
      "model_forward_time": 2.4345788955688477,
      "step": 8699
    },
    {
      "epoch": 3.3184051513671874e-05,
      "step": 8699,
      "training_step_time": 10.303870916366577
    },
    {
      "epoch": 3.31878662109375e-05,
      "grad_norm": 0.5137676000595093,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 8700
    },
    {
      "epoch": 3.31878662109375e-05,
      "model_forward_time": 2.7698726654052734,
      "step": 8700
    },
    {
      "dynamics_loss_avg": 0.041458627209067345,
      "epoch": 3.31878662109375e-05,
      "step": 8700
    },
    {
      "action_loss_avg": 0.015106655680574477,
      "epoch": 3.31878662109375e-05,
      "step": 8700
    },
    {
      "epoch": 3.31878662109375e-05,
      "step": 8700,
      "training_step_time": 7.806005477905273
    },
    {
      "epoch": 3.31878662109375e-05,
      "model_forward_time": 3.132427215576172,
      "step": 8700
    },
    {
      "epoch": 3.31878662109375e-05,
      "step": 8700,
      "training_step_time": 9.901090383529663
    },
    {
      "epoch": 3.31878662109375e-05,
      "model_forward_time": 2.4303812980651855,
      "step": 8700
    },
    {
      "epoch": 3.31878662109375e-05,
      "step": 8700,
      "training_step_time": 7.455300331115723
    },
    {
      "epoch": 3.31878662109375e-05,
      "model_forward_time": 2.4384946823120117,
      "step": 8700
    },
    {
      "epoch": 3.31878662109375e-05,
      "step": 8700,
      "training_step_time": 10.023077249526978
    },
    {
      "epoch": 3.319168090820313e-05,
      "model_forward_time": 3.154822826385498,
      "step": 8701
    },
    {
      "epoch": 3.319168090820313e-05,
      "step": 8701,
      "training_step_time": 9.93050765991211
    },
    {
      "epoch": 3.319168090820313e-05,
      "model_forward_time": 3.428194046020508,
      "step": 8701
    },
    {
      "epoch": 3.319168090820313e-05,
      "step": 8701,
      "training_step_time": 10.19761037826538
    },
    {
      "epoch": 3.319168090820313e-05,
      "model_forward_time": 2.4373297691345215,
      "step": 8701
    },
    {
      "epoch": 3.319168090820313e-05,
      "step": 8701,
      "training_step_time": 7.470308065414429
    },
    {
      "epoch": 3.319168090820313e-05,
      "model_forward_time": 3.136889934539795,
      "step": 8701
    },
    {
      "epoch": 3.319168090820313e-05,
      "step": 8701,
      "training_step_time": 10.041422367095947
    },
    {
      "epoch": 3.319549560546875e-05,
      "model_forward_time": 3.166656017303467,
      "step": 8702
    },
    {
      "epoch": 3.319549560546875e-05,
      "step": 8702,
      "training_step_time": 9.950082540512085
    },
    {
      "epoch": 3.319549560546875e-05,
      "model_forward_time": 2.438011407852173,
      "step": 8702
    },
    {
      "epoch": 3.319549560546875e-05,
      "step": 8702,
      "training_step_time": 7.474099397659302
    },
    {
      "epoch": 3.319549560546875e-05,
      "model_forward_time": 3.41268253326416,
      "step": 8702
    },
    {
      "dynamics_loss_avg": 0.05993529576808214,
      "epoch": 3.319549560546875e-05,
      "step": 8702
    },
    {
      "action_loss_avg": 0.023111498262733222,
      "epoch": 3.319549560546875e-05,
      "step": 8702
    },
    {
      "epoch": 3.319549560546875e-05,
      "step": 8702,
      "training_step_time": 10.191690921783447
    },
    {
      "epoch": 3.319549560546875e-05,
      "model_forward_time": 3.1400258541107178,
      "step": 8702
    },
    {
      "epoch": 3.319549560546875e-05,
      "step": 8702,
      "training_step_time": 10.333621978759766
    },
    {
      "epoch": 3.319931030273437e-05,
      "model_forward_time": 2.450571060180664,
      "step": 8703
    },
    {
      "epoch": 3.319931030273437e-05,
      "step": 8703,
      "training_step_time": 7.4866249561309814
    },
    {
      "epoch": 3.319931030273437e-05,
      "model_forward_time": 3.140124559402466,
      "step": 8703
    },
    {
      "epoch": 3.319931030273437e-05,
      "step": 8703,
      "training_step_time": 9.919603109359741
    },
    {
      "epoch": 3.319931030273437e-05,
      "model_forward_time": 3.1445064544677734,
      "step": 8703
    },
    {
      "epoch": 3.319931030273437e-05,
      "step": 8703,
      "training_step_time": 9.942549228668213
    },
    {
      "epoch": 3.319931030273437e-05,
      "model_forward_time": 3.3947646617889404,
      "step": 8703
    },
    {
      "epoch": 3.319931030273437e-05,
      "step": 8703,
      "training_step_time": 10.309560775756836
    },
    {
      "epoch": 3.3203125e-05,
      "model_forward_time": 3.1623361110687256,
      "step": 8704
    },
    {
      "epoch": 3.3203125e-05,
      "step": 8704,
      "training_step_time": 9.942842245101929
    },
    {
      "epoch": 3.3203125e-05,
      "model_forward_time": 3.1380271911621094,
      "step": 8704
    },
    {
      "epoch": 3.3203125e-05,
      "step": 8704,
      "training_step_time": 9.910429954528809
    },
    {
      "epoch": 3.3203125e-05,
      "model_forward_time": 2.4314279556274414,
      "step": 8704
    },
    {
      "epoch": 3.3203125e-05,
      "step": 8704,
      "training_step_time": 7.461705446243286
    },
    {
      "epoch": 3.3203125e-05,
      "model_forward_time": 3.140821933746338,
      "step": 8704
    },
    {
      "epoch": 3.3203125e-05,
      "step": 8704,
      "training_step_time": 10.314283609390259
    },
    {
      "epoch": 3.3206939697265625e-05,
      "model_forward_time": 2.4498891830444336,
      "step": 8705
    },
    {
      "dynamics_loss_avg": 0.046043310407549144,
      "epoch": 3.3206939697265625e-05,
      "step": 8705
    },
    {
      "action_loss_avg": 0.01764974615070969,
      "epoch": 3.3206939697265625e-05,
      "step": 8705
    },
    {
      "epoch": 3.3206939697265625e-05,
      "step": 8705,
      "training_step_time": 7.839769124984741
    },
    {
      "epoch": 3.3206939697265625e-05,
      "model_forward_time": 2.4334607124328613,
      "step": 8705
    },
    {
      "epoch": 3.3206939697265625e-05,
      "step": 8705,
      "training_step_time": 7.4633848667144775
    },
    {
      "epoch": 3.3206939697265625e-05,
      "model_forward_time": 3.138441324234009,
      "step": 8705
    },
    {
      "epoch": 3.3206939697265625e-05,
      "step": 8705,
      "training_step_time": 9.917328119277954
    },
    {
      "epoch": 3.3206939697265625e-05,
      "model_forward_time": 2.4354898929595947,
      "step": 8705
    },
    {
      "epoch": 3.3206939697265625e-05,
      "step": 8705,
      "training_step_time": 10.03792142868042
    },
    {
      "epoch": 3.321075439453125e-05,
      "model_forward_time": 3.1583566665649414,
      "step": 8706
    },
    {
      "epoch": 3.321075439453125e-05,
      "step": 8706,
      "training_step_time": 9.93704605102539
    },
    {
      "epoch": 3.321075439453125e-05,
      "model_forward_time": 3.138888120651245,
      "step": 8706
    },
    {
      "epoch": 3.321075439453125e-05,
      "step": 8706,
      "training_step_time": 10.213024139404297
    },
    {
      "epoch": 3.321075439453125e-05,
      "model_forward_time": 2.4412992000579834,
      "step": 8706
    },
    {
      "epoch": 3.321075439453125e-05,
      "step": 8706,
      "training_step_time": 7.476944446563721
    },
    {
      "epoch": 3.321075439453125e-05,
      "model_forward_time": 2.436793565750122,
      "step": 8706
    },
    {
      "epoch": 3.321075439453125e-05,
      "step": 8706,
      "training_step_time": 10.024896144866943
    },
    {
      "epoch": 3.321456909179688e-05,
      "model_forward_time": 3.155963182449341,
      "step": 8707
    },
    {
      "epoch": 3.321456909179688e-05,
      "step": 8707,
      "training_step_time": 9.946489334106445
    },
    {
      "epoch": 3.321456909179688e-05,
      "model_forward_time": 2.435513496398926,
      "step": 8707
    },
    {
      "epoch": 3.321456909179688e-05,
      "step": 8707,
      "training_step_time": 7.466754674911499
    },
    {
      "epoch": 3.321456909179688e-05,
      "model_forward_time": 3.1365444660186768,
      "step": 8707
    },
    {
      "dynamics_loss_avg": 0.06235729269683361,
      "epoch": 3.321456909179688e-05,
      "step": 8707
    },
    {
      "action_loss_avg": 0.019317282177507878,
      "epoch": 3.321456909179688e-05,
      "step": 8707
    },
    {
      "epoch": 3.321456909179688e-05,
      "step": 8707,
      "training_step_time": 9.922898054122925
    },
    {
      "epoch": 3.321456909179688e-05,
      "model_forward_time": 2.754384994506836,
      "step": 8707
    },
    {
      "epoch": 3.321456909179688e-05,
      "step": 8707,
      "training_step_time": 10.227442979812622
    },
    {
      "epoch": 3.32183837890625e-05,
      "model_forward_time": 3.1598398685455322,
      "step": 8708
    },
    {
      "epoch": 3.32183837890625e-05,
      "step": 8708,
      "training_step_time": 9.935900449752808
    },
    {
      "epoch": 3.32183837890625e-05,
      "model_forward_time": 2.430635929107666,
      "step": 8708
    },
    {
      "epoch": 3.32183837890625e-05,
      "step": 8708,
      "training_step_time": 7.462719440460205
    },
    {
      "epoch": 3.32183837890625e-05,
      "model_forward_time": 2.4295411109924316,
      "step": 8708
    },
    {
      "epoch": 3.32183837890625e-05,
      "step": 8708,
      "training_step_time": 7.4583258628845215
    },
    {
      "epoch": 3.32183837890625e-05,
      "model_forward_time": 2.436007022857666,
      "step": 8708
    },
    {
      "epoch": 3.32183837890625e-05,
      "step": 8708,
      "training_step_time": 10.008163928985596
    },
    {
      "epoch": 3.3222198486328124e-05,
      "model_forward_time": 3.4933974742889404,
      "step": 8709
    },
    {
      "epoch": 3.3222198486328124e-05,
      "step": 8709,
      "training_step_time": 10.279351711273193
    },
    {
      "epoch": 3.3222198486328124e-05,
      "model_forward_time": 3.135021924972534,
      "step": 8709
    },
    {
      "epoch": 3.3222198486328124e-05,
      "step": 8709,
      "training_step_time": 9.904987335205078
    },
    {
      "epoch": 3.3222198486328124e-05,
      "model_forward_time": 2.430454969406128,
      "step": 8709
    },
    {
      "epoch": 3.3222198486328124e-05,
      "step": 8709,
      "training_step_time": 7.462901830673218
    },
    {
      "epoch": 3.3222198486328124e-05,
      "model_forward_time": 2.435523509979248,
      "step": 8709
    },
    {
      "epoch": 3.3222198486328124e-05,
      "step": 8709,
      "training_step_time": 10.249780178070068
    },
    {
      "epoch": 3.322601318359375e-05,
      "grad_norm": 0.47176486253738403,
      "learning_rate": 1e-05,
      "loss": 0.0833,
      "step": 8710
    },
    {
      "epoch": 3.322601318359375e-05,
      "model_forward_time": 2.479057788848877,
      "step": 8710
    },
    {
      "dynamics_loss_avg": 0.054612872563302516,
      "epoch": 3.322601318359375e-05,
      "step": 8710
    },
    {
      "action_loss_avg": 0.018266021320596338,
      "epoch": 3.322601318359375e-05,
      "step": 8710
    },
    {
      "epoch": 3.322601318359375e-05,
      "step": 8710,
      "training_step_time": 7.523879528045654
    },
    {
      "epoch": 3.322601318359375e-05,
      "model_forward_time": 3.1468842029571533,
      "step": 8710
    },
    {
      "epoch": 3.322601318359375e-05,
      "step": 8710,
      "training_step_time": 9.924267768859863
    },
    {
      "epoch": 3.322601318359375e-05,
      "model_forward_time": 2.7654149532318115,
      "step": 8710
    },
    {
      "epoch": 3.322601318359375e-05,
      "step": 8710,
      "training_step_time": 7.800699472427368
    },
    {
      "epoch": 3.322601318359375e-05,
      "model_forward_time": 3.139505386352539,
      "step": 8710
    },
    {
      "epoch": 3.322601318359375e-05,
      "step": 8710,
      "training_step_time": 10.285414695739746
    },
    {
      "epoch": 3.3229827880859376e-05,
      "model_forward_time": 2.4512696266174316,
      "step": 8711
    },
    {
      "epoch": 3.3229827880859376e-05,
      "step": 8711,
      "training_step_time": 7.484160900115967
    },
    {
      "epoch": 3.3229827880859376e-05,
      "model_forward_time": 2.4376256465911865,
      "step": 8711
    },
    {
      "epoch": 3.3229827880859376e-05,
      "step": 8711,
      "training_step_time": 7.4667205810546875
    },
    {
      "epoch": 3.3229827880859376e-05,
      "model_forward_time": 3.1474192142486572,
      "step": 8711
    },
    {
      "epoch": 3.3229827880859376e-05,
      "step": 8711,
      "training_step_time": 9.92093539237976
    },
    {
      "epoch": 3.3229827880859376e-05,
      "model_forward_time": 2.760100841522217,
      "step": 8711
    },
    {
      "epoch": 3.3229827880859376e-05,
      "step": 8711,
      "training_step_time": 10.021748304367065
    },
    {
      "epoch": 3.3233642578125e-05,
      "model_forward_time": 2.4582109451293945,
      "step": 8712
    },
    {
      "epoch": 3.3233642578125e-05,
      "step": 8712,
      "training_step_time": 7.495579957962036
    },
    {
      "epoch": 3.3233642578125e-05,
      "model_forward_time": 2.438659191131592,
      "step": 8712
    },
    {
      "epoch": 3.3233642578125e-05,
      "step": 8712,
      "training_step_time": 7.468701362609863
    },
    {
      "epoch": 3.3233642578125e-05,
      "model_forward_time": 2.4407153129577637,
      "step": 8712
    },
    {
      "dynamics_loss_avg": 0.05077543314546347,
      "epoch": 3.3233642578125e-05,
      "step": 8712
    },
    {
      "action_loss_avg": 0.017377007659524678,
      "epoch": 3.3233642578125e-05,
      "step": 8712
    },
    {
      "epoch": 3.3233642578125e-05,
      "step": 8712,
      "training_step_time": 7.475609064102173
    },
    {
      "epoch": 3.3233642578125e-05,
      "model_forward_time": 3.1461048126220703,
      "step": 8712
    },
    {
      "epoch": 3.3233642578125e-05,
      "step": 8712,
      "training_step_time": 10.301069259643555
    },
    {
      "epoch": 3.323745727539062e-05,
      "model_forward_time": 3.1583690643310547,
      "step": 8713
    },
    {
      "epoch": 3.323745727539062e-05,
      "step": 8713,
      "training_step_time": 10.248775959014893
    },
    {
      "epoch": 3.323745727539062e-05,
      "model_forward_time": 3.146413803100586,
      "step": 8713
    },
    {
      "epoch": 3.323745727539062e-05,
      "step": 8713,
      "training_step_time": 9.921900987625122
    },
    {
      "epoch": 3.323745727539062e-05,
      "model_forward_time": 3.143366575241089,
      "step": 8713
    },
    {
      "epoch": 3.323745727539062e-05,
      "step": 8713,
      "training_step_time": 9.914645433425903
    },
    {
      "epoch": 3.323745727539062e-05,
      "model_forward_time": 2.4414355754852295,
      "step": 8713
    },
    {
      "epoch": 3.323745727539062e-05,
      "step": 8713,
      "training_step_time": 10.32738184928894
    },
    {
      "epoch": 3.324127197265625e-05,
      "model_forward_time": 2.4718244075775146,
      "step": 8714
    },
    {
      "epoch": 3.324127197265625e-05,
      "step": 8714,
      "training_step_time": 7.506532192230225
    },
    {
      "epoch": 3.324127197265625e-05,
      "model_forward_time": 2.4399261474609375,
      "step": 8714
    },
    {
      "epoch": 3.324127197265625e-05,
      "step": 8714,
      "training_step_time": 7.476556777954102
    },
    {
      "epoch": 3.324127197265625e-05,
      "model_forward_time": 2.756598711013794,
      "step": 8714
    },
    {
      "epoch": 3.324127197265625e-05,
      "step": 8714,
      "training_step_time": 7.783085584640503
    },
    {
      "epoch": 3.324127197265625e-05,
      "model_forward_time": 2.4307308197021484,
      "step": 8714
    },
    {
      "epoch": 3.324127197265625e-05,
      "step": 8714,
      "training_step_time": 10.28560757637024
    },
    {
      "epoch": 3.3245086669921874e-05,
      "model_forward_time": 2.4485175609588623,
      "step": 8715
    },
    {
      "dynamics_loss_avg": 0.0740142084658146,
      "epoch": 3.3245086669921874e-05,
      "step": 8715
    },
    {
      "action_loss_avg": 0.029352338518947363,
      "epoch": 3.3245086669921874e-05,
      "step": 8715
    },
    {
      "epoch": 3.3245086669921874e-05,
      "step": 8715,
      "training_step_time": 7.48198676109314
    },
    {
      "epoch": 3.3245086669921874e-05,
      "model_forward_time": 3.1425769329071045,
      "step": 8715
    },
    {
      "epoch": 3.3245086669921874e-05,
      "step": 8715,
      "training_step_time": 9.911770582199097
    },
    {
      "epoch": 3.3245086669921874e-05,
      "model_forward_time": 2.434849500656128,
      "step": 8715
    },
    {
      "epoch": 3.3245086669921874e-05,
      "step": 8715,
      "training_step_time": 7.461313009262085
    },
    {
      "epoch": 3.3245086669921874e-05,
      "model_forward_time": 2.719210147857666,
      "step": 8715
    },
    {
      "epoch": 3.3245086669921874e-05,
      "step": 8715,
      "training_step_time": 10.352550745010376
    },
    {
      "epoch": 3.32489013671875e-05,
      "model_forward_time": 3.1653730869293213,
      "step": 8716
    },
    {
      "epoch": 3.32489013671875e-05,
      "step": 8716,
      "training_step_time": 9.94241714477539
    },
    {
      "epoch": 3.32489013671875e-05,
      "model_forward_time": 3.1358425617218018,
      "step": 8716
    },
    {
      "epoch": 3.32489013671875e-05,
      "step": 8716,
      "training_step_time": 9.908961772918701
    },
    {
      "epoch": 3.32489013671875e-05,
      "model_forward_time": 2.4382922649383545,
      "step": 8716
    },
    {
      "epoch": 3.32489013671875e-05,
      "step": 8716,
      "training_step_time": 7.468428611755371
    },
    {
      "epoch": 3.32489013671875e-05,
      "model_forward_time": 2.43361496925354,
      "step": 8716
    },
    {
      "epoch": 3.32489013671875e-05,
      "step": 8716,
      "training_step_time": 10.034605503082275
    },
    {
      "epoch": 3.325271606445313e-05,
      "model_forward_time": 2.506831645965576,
      "step": 8717
    },
    {
      "epoch": 3.325271606445313e-05,
      "step": 8717,
      "training_step_time": 7.85988187789917
    },
    {
      "epoch": 3.325271606445313e-05,
      "model_forward_time": 2.4380531311035156,
      "step": 8717
    },
    {
      "epoch": 3.325271606445313e-05,
      "step": 8717,
      "training_step_time": 7.466273546218872
    },
    {
      "epoch": 3.325271606445313e-05,
      "model_forward_time": 2.4346466064453125,
      "step": 8717
    },
    {
      "dynamics_loss_avg": 0.05790312439203262,
      "epoch": 3.325271606445313e-05,
      "step": 8717
    },
    {
      "action_loss_avg": 0.021597642917186022,
      "epoch": 3.325271606445313e-05,
      "step": 8717
    },
    {
      "epoch": 3.325271606445313e-05,
      "step": 8717,
      "training_step_time": 7.465338706970215
    },
    {
      "epoch": 3.325271606445313e-05,
      "model_forward_time": 2.4335336685180664,
      "step": 8717
    },
    {
      "epoch": 3.325271606445313e-05,
      "step": 8717,
      "training_step_time": 10.01767349243164
    },
    {
      "epoch": 3.325653076171875e-05,
      "model_forward_time": 3.1435635089874268,
      "step": 8718
    },
    {
      "epoch": 3.325653076171875e-05,
      "step": 8718,
      "training_step_time": 9.914037942886353
    },
    {
      "epoch": 3.325653076171875e-05,
      "model_forward_time": 3.132948398590088,
      "step": 8718
    },
    {
      "epoch": 3.325653076171875e-05,
      "step": 8718,
      "training_step_time": 9.904697895050049
    },
    {
      "epoch": 3.325653076171875e-05,
      "model_forward_time": 2.0612878799438477,
      "step": 8718
    },
    {
      "epoch": 3.325653076171875e-05,
      "step": 8718,
      "training_step_time": 5.513230085372925
    },
    {
      "epoch": 3.325653076171875e-05,
      "model_forward_time": 3.1322638988494873,
      "step": 8718
    },
    {
      "epoch": 3.325653076171875e-05,
      "step": 8718,
      "training_step_time": 10.285208225250244
    },
    {
      "epoch": 3.326034545898437e-05,
      "model_forward_time": 3.1400997638702393,
      "step": 8719
    },
    {
      "epoch": 3.326034545898437e-05,
      "step": 8719,
      "training_step_time": 9.908950328826904
    },
    {
      "epoch": 3.326034545898437e-05,
      "model_forward_time": 3.1319785118103027,
      "step": 8719
    },
    {
      "epoch": 3.326034545898437e-05,
      "step": 8719,
      "training_step_time": 9.9071786403656
    },
    {
      "epoch": 3.326034545898437e-05,
      "model_forward_time": 2.4374616146087646,
      "step": 8719
    },
    {
      "epoch": 3.326034545898437e-05,
      "step": 8719,
      "training_step_time": 7.46984338760376
    },
    {
      "epoch": 3.326034545898437e-05,
      "model_forward_time": 2.435537338256836,
      "step": 8719
    },
    {
      "epoch": 3.326034545898437e-05,
      "step": 8719,
      "training_step_time": 10.324065208435059
    },
    {
      "epoch": 3.326416015625e-05,
      "grad_norm": 0.5896924734115601,
      "learning_rate": 1e-05,
      "loss": 0.0749,
      "step": 8720
    },
    {
      "epoch": 3.326416015625e-05,
      "model_forward_time": 2.4506032466888428,
      "step": 8720
    },
    {
      "dynamics_loss_avg": 0.048618565406650306,
      "epoch": 3.326416015625e-05,
      "step": 8720
    },
    {
      "action_loss_avg": 0.015011458424851298,
      "epoch": 3.326416015625e-05,
      "step": 8720
    },
    {
      "epoch": 3.326416015625e-05,
      "step": 8720,
      "training_step_time": 7.489901304244995
    },
    {
      "epoch": 3.326416015625e-05,
      "model_forward_time": 2.4377522468566895,
      "step": 8720
    },
    {
      "epoch": 3.326416015625e-05,
      "step": 8720,
      "training_step_time": 7.468628168106079
    },
    {
      "epoch": 3.326416015625e-05,
      "model_forward_time": 2.44108247756958,
      "step": 8720
    },
    {
      "epoch": 3.326416015625e-05,
      "step": 8720,
      "training_step_time": 7.4701056480407715
    },
    {
      "epoch": 3.326416015625e-05,
      "model_forward_time": 2.4391028881073,
      "step": 8720
    },
    {
      "epoch": 3.326416015625e-05,
      "step": 8720,
      "training_step_time": 10.298171758651733
    },
    {
      "epoch": 3.3267974853515625e-05,
      "model_forward_time": 2.4746415615081787,
      "step": 8721
    },
    {
      "epoch": 3.3267974853515625e-05,
      "step": 8721,
      "training_step_time": 7.502692699432373
    },
    {
      "epoch": 3.3267974853515625e-05,
      "model_forward_time": 2.7309069633483887,
      "step": 8721
    },
    {
      "epoch": 3.3267974853515625e-05,
      "step": 8721,
      "training_step_time": 7.752418279647827
    },
    {
      "epoch": 3.3267974853515625e-05,
      "model_forward_time": 2.4370782375335693,
      "step": 8721
    },
    {
      "epoch": 3.3267974853515625e-05,
      "step": 8721,
      "training_step_time": 7.464823246002197
    },
    {
      "epoch": 3.3267974853515625e-05,
      "model_forward_time": 2.4364240169525146,
      "step": 8721
    },
    {
      "epoch": 3.3267974853515625e-05,
      "step": 8721,
      "training_step_time": 10.027784824371338
    },
    {
      "epoch": 3.327178955078125e-05,
      "model_forward_time": 3.1739695072174072,
      "step": 8722
    },
    {
      "epoch": 3.327178955078125e-05,
      "step": 8722,
      "training_step_time": 9.958383321762085
    },
    {
      "epoch": 3.327178955078125e-05,
      "model_forward_time": 3.136025905609131,
      "step": 8722
    },
    {
      "epoch": 3.327178955078125e-05,
      "step": 8722,
      "training_step_time": 9.919428586959839
    },
    {
      "epoch": 3.327178955078125e-05,
      "model_forward_time": 3.456033706665039,
      "step": 8722
    },
    {
      "dynamics_loss_avg": 0.04379820954054594,
      "epoch": 3.327178955078125e-05,
      "step": 8722
    },
    {
      "action_loss_avg": 0.01241433450486511,
      "epoch": 3.327178955078125e-05,
      "step": 8722
    },
    {
      "epoch": 3.327178955078125e-05,
      "step": 8722,
      "training_step_time": 10.241535186767578
    },
    {
      "epoch": 3.327178955078125e-05,
      "model_forward_time": 2.4393908977508545,
      "step": 8722
    },
    {
      "epoch": 3.327178955078125e-05,
      "step": 8722,
      "training_step_time": 10.288055658340454
    },
    {
      "epoch": 3.327560424804688e-05,
      "model_forward_time": 3.1528210639953613,
      "step": 8723
    },
    {
      "epoch": 3.327560424804688e-05,
      "step": 8723,
      "training_step_time": 9.93489933013916
    },
    {
      "epoch": 3.327560424804688e-05,
      "model_forward_time": 3.1423938274383545,
      "step": 8723
    },
    {
      "epoch": 3.327560424804688e-05,
      "step": 8723,
      "training_step_time": 9.919761419296265
    },
    {
      "epoch": 3.327560424804688e-05,
      "model_forward_time": 2.445251941680908,
      "step": 8723
    },
    {
      "epoch": 3.327560424804688e-05,
      "step": 8723,
      "training_step_time": 7.48001503944397
    },
    {
      "epoch": 3.327560424804688e-05,
      "model_forward_time": 3.147979736328125,
      "step": 8723
    },
    {
      "epoch": 3.327560424804688e-05,
      "step": 8723,
      "training_step_time": 10.30203628540039
    },
    {
      "epoch": 3.32794189453125e-05,
      "model_forward_time": 3.1609156131744385,
      "step": 8724
    },
    {
      "epoch": 3.32794189453125e-05,
      "step": 8724,
      "training_step_time": 9.943034887313843
    },
    {
      "epoch": 3.32794189453125e-05,
      "model_forward_time": 3.1425278186798096,
      "step": 8724
    },
    {
      "epoch": 3.32794189453125e-05,
      "step": 8724,
      "training_step_time": 9.921810388565063
    },
    {
      "epoch": 3.32794189453125e-05,
      "model_forward_time": 2.4487946033477783,
      "step": 8724
    },
    {
      "epoch": 3.32794189453125e-05,
      "step": 8724,
      "training_step_time": 7.483209848403931
    },
    {
      "epoch": 3.32794189453125e-05,
      "model_forward_time": 3.14121150970459,
      "step": 8724
    },
    {
      "epoch": 3.32794189453125e-05,
      "step": 8724,
      "training_step_time": 10.257993698120117
    },
    {
      "epoch": 3.3283233642578124e-05,
      "model_forward_time": 3.1506433486938477,
      "step": 8725
    },
    {
      "dynamics_loss_avg": 0.05564378267154098,
      "epoch": 3.3283233642578124e-05,
      "step": 8725
    },
    {
      "action_loss_avg": 0.01792440437711775,
      "epoch": 3.3283233642578124e-05,
      "step": 8725
    },
    {
      "epoch": 3.3283233642578124e-05,
      "step": 8725,
      "training_step_time": 9.929836511611938
    },
    {
      "epoch": 3.3283233642578124e-05,
      "model_forward_time": 3.4361305236816406,
      "step": 8725
    },
    {
      "epoch": 3.3283233642578124e-05,
      "step": 8725,
      "training_step_time": 10.21118950843811
    },
    {
      "epoch": 3.3283233642578124e-05,
      "model_forward_time": 3.146656036376953,
      "step": 8725
    },
    {
      "epoch": 3.3283233642578124e-05,
      "step": 8725,
      "training_step_time": 9.92097544670105
    },
    {
      "epoch": 3.3283233642578124e-05,
      "model_forward_time": 3.143232583999634,
      "step": 8725
    },
    {
      "epoch": 3.3283233642578124e-05,
      "step": 8725,
      "training_step_time": 10.329391717910767
    },
    {
      "epoch": 3.328704833984375e-05,
      "model_forward_time": 3.154221773147583,
      "step": 8726
    },
    {
      "epoch": 3.328704833984375e-05,
      "step": 8726,
      "training_step_time": 9.933595418930054
    },
    {
      "epoch": 3.328704833984375e-05,
      "model_forward_time": 3.405829429626465,
      "step": 8726
    },
    {
      "epoch": 3.328704833984375e-05,
      "step": 8726,
      "training_step_time": 10.177749872207642
    },
    {
      "epoch": 3.328704833984375e-05,
      "model_forward_time": 3.1453027725219727,
      "step": 8726
    },
    {
      "epoch": 3.328704833984375e-05,
      "step": 8726,
      "training_step_time": 9.919035911560059
    },
    {
      "epoch": 3.328704833984375e-05,
      "model_forward_time": 1.749143123626709,
      "step": 8726
    },
    {
      "epoch": 3.328704833984375e-05,
      "step": 8726,
      "training_step_time": 10.329367637634277
    },
    {
      "epoch": 3.3290863037109376e-05,
      "model_forward_time": 3.1446738243103027,
      "step": 8727
    },
    {
      "epoch": 3.3290863037109376e-05,
      "step": 8727,
      "training_step_time": 9.91363263130188
    },
    {
      "epoch": 3.3290863037109376e-05,
      "model_forward_time": 3.1378426551818848,
      "step": 8727
    },
    {
      "epoch": 3.3290863037109376e-05,
      "step": 8727,
      "training_step_time": 9.906041860580444
    },
    {
      "epoch": 3.3290863037109376e-05,
      "model_forward_time": 3.1362762451171875,
      "step": 8727
    },
    {
      "dynamics_loss_avg": 0.061098809354007244,
      "epoch": 3.3290863037109376e-05,
      "step": 8727
    },
    {
      "action_loss_avg": 0.027415607031434773,
      "epoch": 3.3290863037109376e-05,
      "step": 8727
    },
    {
      "epoch": 3.3290863037109376e-05,
      "step": 8727,
      "training_step_time": 9.903775930404663
    },
    {
      "epoch": 3.3290863037109376e-05,
      "model_forward_time": 2.7102038860321045,
      "step": 8727
    },
    {
      "epoch": 3.3290863037109376e-05,
      "step": 8727,
      "training_step_time": 10.01958966255188
    },
    {
      "epoch": 3.3294677734375e-05,
      "model_forward_time": 2.451364755630493,
      "step": 8728
    },
    {
      "epoch": 3.3294677734375e-05,
      "step": 8728,
      "training_step_time": 7.4780871868133545
    },
    {
      "epoch": 3.3294677734375e-05,
      "model_forward_time": 2.432373046875,
      "step": 8728
    },
    {
      "epoch": 3.3294677734375e-05,
      "step": 8728,
      "training_step_time": 7.452199459075928
    },
    {
      "epoch": 3.3294677734375e-05,
      "model_forward_time": 1.7461302280426025,
      "step": 8728
    },
    {
      "epoch": 3.3294677734375e-05,
      "step": 8728,
      "training_step_time": 5.197335720062256
    },
    {
      "epoch": 3.3294677734375e-05,
      "model_forward_time": 2.4450700283050537,
      "step": 8728
    },
    {
      "epoch": 3.3294677734375e-05,
      "step": 8728,
      "training_step_time": 10.247704029083252
    },
    {
      "epoch": 3.329849243164062e-05,
      "model_forward_time": 2.4515233039855957,
      "step": 8729
    },
    {
      "epoch": 3.329849243164062e-05,
      "step": 8729,
      "training_step_time": 7.476153612136841
    },
    {
      "epoch": 3.329849243164062e-05,
      "model_forward_time": 2.742034435272217,
      "step": 8729
    },
    {
      "epoch": 3.329849243164062e-05,
      "step": 8729,
      "training_step_time": 7.760654449462891
    },
    {
      "epoch": 3.329849243164062e-05,
      "model_forward_time": 2.432460308074951,
      "step": 8729
    },
    {
      "epoch": 3.329849243164062e-05,
      "step": 8729,
      "training_step_time": 7.451372861862183
    },
    {
      "epoch": 3.329849243164062e-05,
      "model_forward_time": 3.137305498123169,
      "step": 8729
    },
    {
      "epoch": 3.329849243164062e-05,
      "step": 8729,
      "training_step_time": 10.274460315704346
    },
    {
      "epoch": 3.330230712890625e-05,
      "grad_norm": 0.5062488317489624,
      "learning_rate": 1e-05,
      "loss": 0.0822,
      "step": 8730
    },
    {
      "epoch": 3.330230712890625e-05,
      "model_forward_time": 3.1552069187164307,
      "step": 8730
    },
    {
      "dynamics_loss_avg": 0.054718551225960255,
      "epoch": 3.330230712890625e-05,
      "step": 8730
    },
    {
      "action_loss_avg": 0.019084419962018727,
      "epoch": 3.330230712890625e-05,
      "step": 8730
    },
    {
      "epoch": 3.330230712890625e-05,
      "step": 8730,
      "training_step_time": 9.939105033874512
    },
    {
      "epoch": 3.330230712890625e-05,
      "model_forward_time": 2.4352428913116455,
      "step": 8730
    },
    {
      "epoch": 3.330230712890625e-05,
      "step": 8730,
      "training_step_time": 7.459724426269531
    },
    {
      "epoch": 3.330230712890625e-05,
      "model_forward_time": 3.3750112056732178,
      "step": 8730
    },
    {
      "epoch": 3.330230712890625e-05,
      "step": 8730,
      "training_step_time": 10.141602516174316
    },
    {
      "epoch": 3.330230712890625e-05,
      "model_forward_time": 3.135570526123047,
      "step": 8730
    },
    {
      "epoch": 3.330230712890625e-05,
      "step": 8730,
      "training_step_time": 10.025567293167114
    },
    {
      "epoch": 3.3306121826171875e-05,
      "model_forward_time": 2.496854305267334,
      "step": 8731
    },
    {
      "epoch": 3.3306121826171875e-05,
      "step": 8731,
      "training_step_time": 7.536337852478027
    },
    {
      "epoch": 3.3306121826171875e-05,
      "model_forward_time": 3.1355886459350586,
      "step": 8731
    },
    {
      "epoch": 3.3306121826171875e-05,
      "step": 8731,
      "training_step_time": 9.90939736366272
    },
    {
      "epoch": 3.3306121826171875e-05,
      "model_forward_time": 3.1359851360321045,
      "step": 8731
    },
    {
      "epoch": 3.3306121826171875e-05,
      "step": 8731,
      "training_step_time": 9.9066641330719
    },
    {
      "epoch": 3.3306121826171875e-05,
      "model_forward_time": 3.4114632606506348,
      "step": 8731
    },
    {
      "epoch": 3.3306121826171875e-05,
      "step": 8731,
      "training_step_time": 10.330397605895996
    },
    {
      "epoch": 3.33099365234375e-05,
      "model_forward_time": 2.4526679515838623,
      "step": 8732
    },
    {
      "epoch": 3.33099365234375e-05,
      "step": 8732,
      "training_step_time": 7.481209754943848
    },
    {
      "epoch": 3.33099365234375e-05,
      "model_forward_time": 2.4306890964508057,
      "step": 8732
    },
    {
      "epoch": 3.33099365234375e-05,
      "step": 8732,
      "training_step_time": 7.459419250488281
    },
    {
      "epoch": 3.33099365234375e-05,
      "model_forward_time": 2.4334609508514404,
      "step": 8732
    },
    {
      "dynamics_loss_avg": 0.04430058561265469,
      "epoch": 3.33099365234375e-05,
      "step": 8732
    },
    {
      "action_loss_avg": 0.017747393017634748,
      "epoch": 3.33099365234375e-05,
      "step": 8732
    },
    {
      "epoch": 3.33099365234375e-05,
      "step": 8732,
      "training_step_time": 7.465136289596558
    },
    {
      "epoch": 3.33099365234375e-05,
      "model_forward_time": 3.136431932449341,
      "step": 8732
    },
    {
      "epoch": 3.33099365234375e-05,
      "step": 8732,
      "training_step_time": 10.307866096496582
    },
    {
      "epoch": 3.331375122070313e-05,
      "model_forward_time": 2.708780288696289,
      "step": 8733
    },
    {
      "epoch": 3.331375122070313e-05,
      "step": 8733,
      "training_step_time": 7.739371061325073
    },
    {
      "epoch": 3.331375122070313e-05,
      "model_forward_time": 3.1365628242492676,
      "step": 8733
    },
    {
      "epoch": 3.331375122070313e-05,
      "step": 8733,
      "training_step_time": 9.913758039474487
    },
    {
      "epoch": 3.331375122070313e-05,
      "model_forward_time": 2.434800386428833,
      "step": 8733
    },
    {
      "epoch": 3.331375122070313e-05,
      "step": 8733,
      "training_step_time": 7.4698357582092285
    },
    {
      "epoch": 3.331375122070313e-05,
      "model_forward_time": 3.14044451713562,
      "step": 8733
    },
    {
      "epoch": 3.331375122070313e-05,
      "step": 8733,
      "training_step_time": 10.041826248168945
    },
    {
      "epoch": 3.331756591796875e-05,
      "model_forward_time": 2.4565327167510986,
      "step": 8734
    },
    {
      "epoch": 3.331756591796875e-05,
      "step": 8734,
      "training_step_time": 7.492180824279785
    },
    {
      "epoch": 3.331756591796875e-05,
      "model_forward_time": 2.4370501041412354,
      "step": 8734
    },
    {
      "epoch": 3.331756591796875e-05,
      "step": 8734,
      "training_step_time": 7.472061395645142
    },
    {
      "epoch": 3.331756591796875e-05,
      "model_forward_time": 2.005685806274414,
      "step": 8734
    },
    {
      "epoch": 3.331756591796875e-05,
      "step": 8734,
      "training_step_time": 5.465011119842529
    },
    {
      "epoch": 3.331756591796875e-05,
      "model_forward_time": 2.430675983428955,
      "step": 8734
    },
    {
      "epoch": 3.331756591796875e-05,
      "step": 8734,
      "training_step_time": 10.292957305908203
    },
    {
      "epoch": 3.332138061523437e-05,
      "model_forward_time": 3.157994270324707,
      "step": 8735
    },
    {
      "dynamics_loss_avg": 0.044494778104126456,
      "epoch": 3.332138061523437e-05,
      "step": 8735
    },
    {
      "action_loss_avg": 0.015416515502147376,
      "epoch": 3.332138061523437e-05,
      "step": 8735
    },
    {
      "epoch": 3.332138061523437e-05,
      "step": 8735,
      "training_step_time": 9.935585260391235
    },
    {
      "epoch": 3.332138061523437e-05,
      "model_forward_time": 3.1381192207336426,
      "step": 8735
    },
    {
      "epoch": 3.332138061523437e-05,
      "step": 8735,
      "training_step_time": 9.910265684127808
    },
    {
      "epoch": 3.332138061523437e-05,
      "model_forward_time": 2.4346423149108887,
      "step": 8735
    },
    {
      "epoch": 3.332138061523437e-05,
      "step": 8735,
      "training_step_time": 7.465280771255493
    },
    {
      "epoch": 3.332138061523437e-05,
      "model_forward_time": 3.396005630493164,
      "step": 8735
    },
    {
      "epoch": 3.332138061523437e-05,
      "step": 8735,
      "training_step_time": 10.279920101165771
    },
    {
      "epoch": 3.33251953125e-05,
      "model_forward_time": 3.1491310596466064,
      "step": 8736
    },
    {
      "epoch": 3.33251953125e-05,
      "step": 8736,
      "training_step_time": 9.927528142929077
    },
    {
      "epoch": 3.33251953125e-05,
      "model_forward_time": 3.139214515686035,
      "step": 8736
    },
    {
      "epoch": 3.33251953125e-05,
      "step": 8736,
      "training_step_time": 9.914771318435669
    },
    {
      "epoch": 3.33251953125e-05,
      "model_forward_time": 3.1386845111846924,
      "step": 8736
    },
    {
      "epoch": 3.33251953125e-05,
      "step": 8736,
      "training_step_time": 9.910066604614258
    },
    {
      "epoch": 3.33251953125e-05,
      "model_forward_time": 2.431072950363159,
      "step": 8736
    },
    {
      "epoch": 3.33251953125e-05,
      "step": 8736,
      "training_step_time": 10.335194110870361
    },
    {
      "epoch": 3.3329010009765625e-05,
      "model_forward_time": 3.420156240463257,
      "step": 8737
    },
    {
      "epoch": 3.3329010009765625e-05,
      "step": 8737,
      "training_step_time": 10.192500352859497
    },
    {
      "epoch": 3.3329010009765625e-05,
      "model_forward_time": 2.4339754581451416,
      "step": 8737
    },
    {
      "epoch": 3.3329010009765625e-05,
      "step": 8737,
      "training_step_time": 7.45891547203064
    },
    {
      "epoch": 3.3329010009765625e-05,
      "model_forward_time": 3.1320512294769287,
      "step": 8737
    },
    {
      "dynamics_loss_avg": 0.06075407918542623,
      "epoch": 3.3329010009765625e-05,
      "step": 8737
    },
    {
      "action_loss_avg": 0.02099303901195526,
      "epoch": 3.3329010009765625e-05,
      "step": 8737
    },
    {
      "epoch": 3.3329010009765625e-05,
      "step": 8737,
      "training_step_time": 9.90081000328064
    },
    {
      "epoch": 3.3329010009765625e-05,
      "model_forward_time": 2.4334614276885986,
      "step": 8737
    },
    {
      "epoch": 3.3329010009765625e-05,
      "step": 8737,
      "training_step_time": 10.33014464378357
    },
    {
      "epoch": 3.333282470703125e-05,
      "model_forward_time": 2.446377754211426,
      "step": 8738
    },
    {
      "epoch": 3.333282470703125e-05,
      "step": 8738,
      "training_step_time": 7.470072031021118
    },
    {
      "epoch": 3.333282470703125e-05,
      "model_forward_time": 3.1320242881774902,
      "step": 8738
    },
    {
      "epoch": 3.333282470703125e-05,
      "step": 8738,
      "training_step_time": 9.90142822265625
    },
    {
      "epoch": 3.333282470703125e-05,
      "model_forward_time": 3.423588991165161,
      "step": 8738
    },
    {
      "epoch": 3.333282470703125e-05,
      "step": 8738,
      "training_step_time": 10.19493579864502
    },
    {
      "epoch": 3.333282470703125e-05,
      "model_forward_time": 2.4344024658203125,
      "step": 8738
    },
    {
      "epoch": 3.333282470703125e-05,
      "step": 8738,
      "training_step_time": 10.284738779067993
    },
    {
      "epoch": 3.333663940429688e-05,
      "model_forward_time": 2.4483118057250977,
      "step": 8739
    },
    {
      "epoch": 3.333663940429688e-05,
      "step": 8739,
      "training_step_time": 7.490624189376831
    },
    {
      "epoch": 3.333663940429688e-05,
      "model_forward_time": 2.4326391220092773,
      "step": 8739
    },
    {
      "epoch": 3.333663940429688e-05,
      "step": 8739,
      "training_step_time": 7.457091331481934
    },
    {
      "epoch": 3.333663940429688e-05,
      "model_forward_time": 3.1383213996887207,
      "step": 8739
    },
    {
      "epoch": 3.333663940429688e-05,
      "step": 8739,
      "training_step_time": 9.90933609008789
    },
    {
      "epoch": 3.333663940429688e-05,
      "model_forward_time": 2.702725648880005,
      "step": 8739
    },
    {
      "epoch": 3.333663940429688e-05,
      "step": 8739,
      "training_step_time": 10.2993745803833
    },
    {
      "epoch": 3.33404541015625e-05,
      "grad_norm": 0.3943876326084137,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 8740
    },
    {
      "epoch": 3.33404541015625e-05,
      "model_forward_time": 2.4477014541625977,
      "step": 8740
    },
    {
      "dynamics_loss_avg": 0.05410916013643145,
      "epoch": 3.33404541015625e-05,
      "step": 8740
    },
    {
      "action_loss_avg": 0.018277575029060245,
      "epoch": 3.33404541015625e-05,
      "step": 8740
    },
    {
      "epoch": 3.33404541015625e-05,
      "step": 8740,
      "training_step_time": 7.481500864028931
    },
    {
      "epoch": 3.33404541015625e-05,
      "model_forward_time": 2.428931713104248,
      "step": 8740
    },
    {
      "epoch": 3.33404541015625e-05,
      "step": 8740,
      "training_step_time": 7.46074104309082
    },
    {
      "epoch": 3.33404541015625e-05,
      "model_forward_time": 2.4317703247070312,
      "step": 8740
    },
    {
      "epoch": 3.33404541015625e-05,
      "step": 8740,
      "training_step_time": 7.455590009689331
    },
    {
      "epoch": 3.33404541015625e-05,
      "model_forward_time": 3.1299397945404053,
      "step": 8740
    },
    {
      "epoch": 3.33404541015625e-05,
      "step": 8740,
      "training_step_time": 10.302907466888428
    },
    {
      "epoch": 3.3344268798828124e-05,
      "model_forward_time": 2.4480841159820557,
      "step": 8741
    },
    {
      "epoch": 3.3344268798828124e-05,
      "step": 8741,
      "training_step_time": 7.48544454574585
    },
    {
      "epoch": 3.3344268798828124e-05,
      "model_forward_time": 3.4296698570251465,
      "step": 8741
    },
    {
      "epoch": 3.3344268798828124e-05,
      "step": 8741,
      "training_step_time": 10.20441484451294
    },
    {
      "epoch": 3.3344268798828124e-05,
      "model_forward_time": 2.4340667724609375,
      "step": 8741
    },
    {
      "epoch": 3.3344268798828124e-05,
      "step": 8741,
      "training_step_time": 7.461871147155762
    },
    {
      "epoch": 3.3344268798828124e-05,
      "model_forward_time": 1.74538254737854,
      "step": 8741
    },
    {
      "epoch": 3.3344268798828124e-05,
      "step": 8741,
      "training_step_time": 10.300344467163086
    },
    {
      "epoch": 3.334808349609375e-05,
      "model_forward_time": 3.1415343284606934,
      "step": 8742
    },
    {
      "epoch": 3.334808349609375e-05,
      "step": 8742,
      "training_step_time": 9.914540529251099
    },
    {
      "epoch": 3.334808349609375e-05,
      "model_forward_time": 3.138270616531372,
      "step": 8742
    },
    {
      "epoch": 3.334808349609375e-05,
      "step": 8742,
      "training_step_time": 9.909916639328003
    },
    {
      "epoch": 3.334808349609375e-05,
      "model_forward_time": 3.4491560459136963,
      "step": 8742
    },
    {
      "dynamics_loss_avg": 0.05309197269380093,
      "epoch": 3.334808349609375e-05,
      "step": 8742
    },
    {
      "action_loss_avg": 0.018063316028565167,
      "epoch": 3.334808349609375e-05,
      "step": 8742
    },
    {
      "epoch": 3.334808349609375e-05,
      "step": 8742,
      "training_step_time": 10.236063957214355
    },
    {
      "epoch": 3.334808349609375e-05,
      "model_forward_time": 3.1446948051452637,
      "step": 8742
    },
    {
      "epoch": 3.334808349609375e-05,
      "step": 8742,
      "training_step_time": 10.320305109024048
    },
    {
      "epoch": 3.3351898193359376e-05,
      "model_forward_time": 3.1517856121063232,
      "step": 8743
    },
    {
      "epoch": 3.3351898193359376e-05,
      "step": 8743,
      "training_step_time": 9.937279462814331
    },
    {
      "epoch": 3.3351898193359376e-05,
      "model_forward_time": 3.1481173038482666,
      "step": 8743
    },
    {
      "epoch": 3.3351898193359376e-05,
      "step": 8743,
      "training_step_time": 9.929282188415527
    },
    {
      "epoch": 3.3351898193359376e-05,
      "model_forward_time": 3.1468665599823,
      "step": 8743
    },
    {
      "epoch": 3.3351898193359376e-05,
      "step": 8743,
      "training_step_time": 9.927249908447266
    },
    {
      "epoch": 3.3351898193359376e-05,
      "model_forward_time": 2.733391284942627,
      "step": 8743
    },
    {
      "epoch": 3.3351898193359376e-05,
      "step": 8743,
      "training_step_time": 10.319162130355835
    },
    {
      "epoch": 3.3355712890625e-05,
      "model_forward_time": 3.159142255783081,
      "step": 8744
    },
    {
      "epoch": 3.3355712890625e-05,
      "step": 8744,
      "training_step_time": 9.942704200744629
    },
    {
      "epoch": 3.3355712890625e-05,
      "model_forward_time": 3.1433329582214355,
      "step": 8744
    },
    {
      "epoch": 3.3355712890625e-05,
      "step": 8744,
      "training_step_time": 9.924027681350708
    },
    {
      "epoch": 3.3355712890625e-05,
      "model_forward_time": 2.4400148391723633,
      "step": 8744
    },
    {
      "epoch": 3.3355712890625e-05,
      "step": 8744,
      "training_step_time": 7.476502418518066
    },
    {
      "epoch": 3.3355712890625e-05,
      "model_forward_time": 2.436457395553589,
      "step": 8744
    },
    {
      "epoch": 3.3355712890625e-05,
      "step": 8744,
      "training_step_time": 10.311543464660645
    },
    {
      "epoch": 3.335952758789062e-05,
      "model_forward_time": 2.7715156078338623,
      "step": 8745
    },
    {
      "dynamics_loss_avg": 0.06236121617257595,
      "epoch": 3.335952758789062e-05,
      "step": 8745
    },
    {
      "action_loss_avg": 0.019021338410675527,
      "epoch": 3.335952758789062e-05,
      "step": 8745
    },
    {
      "epoch": 3.335952758789062e-05,
      "step": 8745,
      "training_step_time": 7.808810472488403
    },
    {
      "epoch": 3.335952758789062e-05,
      "model_forward_time": 3.1345324516296387,
      "step": 8745
    },
    {
      "epoch": 3.335952758789062e-05,
      "step": 8745,
      "training_step_time": 9.909104108810425
    },
    {
      "epoch": 3.335952758789062e-05,
      "model_forward_time": 2.438053607940674,
      "step": 8745
    },
    {
      "epoch": 3.335952758789062e-05,
      "step": 8745,
      "training_step_time": 7.473495006561279
    },
    {
      "epoch": 3.335952758789062e-05,
      "model_forward_time": 2.4303066730499268,
      "step": 8745
    },
    {
      "epoch": 3.335952758789062e-05,
      "step": 8745,
      "training_step_time": 10.00935173034668
    },
    {
      "epoch": 3.336334228515625e-05,
      "model_forward_time": 3.157588481903076,
      "step": 8746
    },
    {
      "epoch": 3.336334228515625e-05,
      "step": 8746,
      "training_step_time": 9.952937602996826
    },
    {
      "epoch": 3.336334228515625e-05,
      "model_forward_time": 3.141768455505371,
      "step": 8746
    },
    {
      "epoch": 3.336334228515625e-05,
      "step": 8746,
      "training_step_time": 9.925477743148804
    },
    {
      "epoch": 3.336334228515625e-05,
      "model_forward_time": 2.741380214691162,
      "step": 8746
    },
    {
      "epoch": 3.336334228515625e-05,
      "step": 8746,
      "training_step_time": 7.781814813613892
    },
    {
      "epoch": 3.336334228515625e-05,
      "model_forward_time": 3.135470151901245,
      "step": 8746
    },
    {
      "epoch": 3.336334228515625e-05,
      "step": 8746,
      "training_step_time": 10.018504858016968
    },
    {
      "epoch": 3.3367156982421875e-05,
      "model_forward_time": 3.1536142826080322,
      "step": 8747
    },
    {
      "epoch": 3.3367156982421875e-05,
      "step": 8747,
      "training_step_time": 9.929909944534302
    },
    {
      "epoch": 3.3367156982421875e-05,
      "model_forward_time": 2.43760347366333,
      "step": 8747
    },
    {
      "epoch": 3.3367156982421875e-05,
      "step": 8747,
      "training_step_time": 7.47064995765686
    },
    {
      "epoch": 3.3367156982421875e-05,
      "model_forward_time": 2.4469730854034424,
      "step": 8747
    },
    {
      "dynamics_loss_avg": 0.0525376059114933,
      "epoch": 3.3367156982421875e-05,
      "step": 8747
    },
    {
      "action_loss_avg": 0.017695375671610235,
      "epoch": 3.3367156982421875e-05,
      "step": 8747
    },
    {
      "epoch": 3.3367156982421875e-05,
      "step": 8747,
      "training_step_time": 7.483151197433472
    },
    {
      "epoch": 3.3367156982421875e-05,
      "model_forward_time": 3.4689180850982666,
      "step": 8747
    },
    {
      "epoch": 3.3367156982421875e-05,
      "step": 8747,
      "training_step_time": 10.335946798324585
    },
    {
      "epoch": 3.33709716796875e-05,
      "model_forward_time": 2.452906370162964,
      "step": 8748
    },
    {
      "epoch": 3.33709716796875e-05,
      "step": 8748,
      "training_step_time": 7.485235691070557
    },
    {
      "epoch": 3.33709716796875e-05,
      "model_forward_time": 2.4344468116760254,
      "step": 8748
    },
    {
      "epoch": 3.33709716796875e-05,
      "step": 8748,
      "training_step_time": 7.463021516799927
    },
    {
      "epoch": 3.33709716796875e-05,
      "model_forward_time": 2.4363255500793457,
      "step": 8748
    },
    {
      "epoch": 3.33709716796875e-05,
      "step": 8748,
      "training_step_time": 7.463199138641357
    },
    {
      "epoch": 3.33709716796875e-05,
      "model_forward_time": 2.4384307861328125,
      "step": 8748
    },
    {
      "epoch": 3.33709716796875e-05,
      "step": 8748,
      "training_step_time": 10.288106203079224
    },
    {
      "epoch": 3.337478637695313e-05,
      "model_forward_time": 2.492750644683838,
      "step": 8749
    },
    {
      "epoch": 3.337478637695313e-05,
      "step": 8749,
      "training_step_time": 7.524938106536865
    },
    {
      "epoch": 3.337478637695313e-05,
      "model_forward_time": 2.7090556621551514,
      "step": 8749
    },
    {
      "epoch": 3.337478637695313e-05,
      "step": 8749,
      "training_step_time": 7.72835373878479
    },
    {
      "epoch": 3.337478637695313e-05,
      "model_forward_time": 3.1345608234405518,
      "step": 8749
    },
    {
      "epoch": 3.337478637695313e-05,
      "step": 8749,
      "training_step_time": 9.906379461288452
    },
    {
      "epoch": 3.337478637695313e-05,
      "model_forward_time": 3.1329526901245117,
      "step": 8749
    },
    {
      "epoch": 3.337478637695313e-05,
      "step": 8749,
      "training_step_time": 10.27058458328247
    },
    {
      "epoch": 3.337860107421875e-05,
      "grad_norm": 0.3245183825492859,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 8750
    },
    {
      "epoch": 3.337860107421875e-05,
      "model_forward_time": 3.1434576511383057,
      "step": 8750
    },
    {
      "dynamics_loss_avg": 0.06554641425609589,
      "epoch": 3.337860107421875e-05,
      "step": 8750
    },
    {
      "action_loss_avg": 0.023395398678258062,
      "epoch": 3.337860107421875e-05,
      "step": 8750
    },
    {
      "epoch": 3.337860107421875e-05,
      "step": 8750,
      "training_step_time": 9.942386865615845
    },
    {
      "epoch": 3.337860107421875e-05,
      "model_forward_time": 3.1461849212646484,
      "step": 8750
    },
    {
      "epoch": 3.337860107421875e-05,
      "step": 8750,
      "training_step_time": 9.9275803565979
    },
    {
      "epoch": 3.337860107421875e-05,
      "model_forward_time": 2.722273588180542,
      "step": 8750
    },
    {
      "epoch": 3.337860107421875e-05,
      "step": 8750,
      "training_step_time": 7.750356197357178
    },
    {
      "epoch": 3.337860107421875e-05,
      "model_forward_time": 2.4274098873138428,
      "step": 8750
    },
    {
      "epoch": 3.337860107421875e-05,
      "step": 8750,
      "training_step_time": 10.062071561813354
    },
    {
      "epoch": 3.338241577148437e-05,
      "model_forward_time": 3.145874261856079,
      "step": 8751
    },
    {
      "epoch": 3.338241577148437e-05,
      "step": 8751,
      "training_step_time": 9.928552389144897
    },
    {
      "epoch": 3.338241577148437e-05,
      "model_forward_time": 3.143463373184204,
      "step": 8751
    },
    {
      "epoch": 3.338241577148437e-05,
      "step": 8751,
      "training_step_time": 9.920209884643555
    },
    {
      "epoch": 3.338241577148437e-05,
      "model_forward_time": 3.133122205734253,
      "step": 8751
    },
    {
      "epoch": 3.338241577148437e-05,
      "step": 8751,
      "training_step_time": 9.911632299423218
    },
    {
      "epoch": 3.338241577148437e-05,
      "model_forward_time": 3.428368330001831,
      "step": 8751
    },
    {
      "epoch": 3.338241577148437e-05,
      "step": 8751,
      "training_step_time": 10.312224864959717
    },
    {
      "epoch": 3.338623046875e-05,
      "model_forward_time": 3.1562788486480713,
      "step": 8752
    },
    {
      "epoch": 3.338623046875e-05,
      "step": 8752,
      "training_step_time": 9.93772578239441
    },
    {
      "epoch": 3.338623046875e-05,
      "model_forward_time": 3.13905930519104,
      "step": 8752
    },
    {
      "epoch": 3.338623046875e-05,
      "step": 8752,
      "training_step_time": 9.918026685714722
    },
    {
      "epoch": 3.338623046875e-05,
      "model_forward_time": 2.4341087341308594,
      "step": 8752
    },
    {
      "dynamics_loss_avg": 0.05503249056637287,
      "epoch": 3.338623046875e-05,
      "step": 8752
    },
    {
      "action_loss_avg": 0.015483906911686062,
      "epoch": 3.338623046875e-05,
      "step": 8752
    },
    {
      "epoch": 3.338623046875e-05,
      "step": 8752,
      "training_step_time": 7.477830410003662
    },
    {
      "epoch": 3.338623046875e-05,
      "model_forward_time": 3.1381378173828125,
      "step": 8752
    },
    {
      "epoch": 3.338623046875e-05,
      "step": 8752,
      "training_step_time": 10.019088983535767
    },
    {
      "epoch": 3.3390045166015625e-05,
      "model_forward_time": 3.4398858547210693,
      "step": 8753
    },
    {
      "epoch": 3.3390045166015625e-05,
      "step": 8753,
      "training_step_time": 10.221293687820435
    },
    {
      "epoch": 3.3390045166015625e-05,
      "model_forward_time": 3.136517286300659,
      "step": 8753
    },
    {
      "epoch": 3.3390045166015625e-05,
      "step": 8753,
      "training_step_time": 9.920188665390015
    },
    {
      "epoch": 3.3390045166015625e-05,
      "model_forward_time": 3.1835362911224365,
      "step": 8753
    },
    {
      "epoch": 3.3390045166015625e-05,
      "step": 8753,
      "training_step_time": 9.97056245803833
    },
    {
      "epoch": 3.3390045166015625e-05,
      "model_forward_time": 3.145448684692383,
      "step": 8753
    },
    {
      "epoch": 3.3390045166015625e-05,
      "step": 8753,
      "training_step_time": 10.287913799285889
    },
    {
      "epoch": 3.339385986328125e-05,
      "model_forward_time": 3.150904655456543,
      "step": 8754
    },
    {
      "epoch": 3.339385986328125e-05,
      "step": 8754,
      "training_step_time": 10.214214324951172
    },
    {
      "epoch": 3.339385986328125e-05,
      "model_forward_time": 3.1380224227905273,
      "step": 8754
    },
    {
      "epoch": 3.339385986328125e-05,
      "step": 8754,
      "training_step_time": 9.918121337890625
    },
    {
      "epoch": 3.339385986328125e-05,
      "model_forward_time": 2.43564772605896,
      "step": 8754
    },
    {
      "epoch": 3.339385986328125e-05,
      "step": 8754,
      "training_step_time": 7.470588207244873
    },
    {
      "epoch": 3.339385986328125e-05,
      "model_forward_time": 3.138512134552002,
      "step": 8754
    },
    {
      "epoch": 3.339385986328125e-05,
      "step": 8754,
      "training_step_time": 10.311726093292236
    },
    {
      "epoch": 3.339767456054688e-05,
      "model_forward_time": 3.1503074169158936,
      "step": 8755
    },
    {
      "dynamics_loss_avg": 0.0625510971993208,
      "epoch": 3.339767456054688e-05,
      "step": 8755
    },
    {
      "action_loss_avg": 0.02057223143056035,
      "epoch": 3.339767456054688e-05,
      "step": 8755
    },
    {
      "epoch": 3.339767456054688e-05,
      "step": 8755,
      "training_step_time": 9.932567119598389
    },
    {
      "epoch": 3.339767456054688e-05,
      "model_forward_time": 3.1283161640167236,
      "step": 8755
    },
    {
      "epoch": 3.339767456054688e-05,
      "step": 8755,
      "training_step_time": 9.913192749023438
    },
    {
      "epoch": 3.339767456054688e-05,
      "model_forward_time": 3.4212892055511475,
      "step": 8755
    },
    {
      "epoch": 3.339767456054688e-05,
      "step": 8755,
      "training_step_time": 10.198177576065063
    },
    {
      "epoch": 3.339767456054688e-05,
      "model_forward_time": 3.1394941806793213,
      "step": 8755
    },
    {
      "epoch": 3.339767456054688e-05,
      "step": 8755,
      "training_step_time": 10.01780891418457
    },
    {
      "epoch": 3.34014892578125e-05,
      "model_forward_time": 3.1534423828125,
      "step": 8756
    },
    {
      "epoch": 3.34014892578125e-05,
      "step": 8756,
      "training_step_time": 9.93697714805603
    },
    {
      "epoch": 3.34014892578125e-05,
      "model_forward_time": 3.1304996013641357,
      "step": 8756
    },
    {
      "epoch": 3.34014892578125e-05,
      "step": 8756,
      "training_step_time": 9.901610136032104
    },
    {
      "epoch": 3.34014892578125e-05,
      "model_forward_time": 3.433422088623047,
      "step": 8756
    },
    {
      "epoch": 3.34014892578125e-05,
      "step": 8756,
      "training_step_time": 10.20432186126709
    },
    {
      "epoch": 3.34014892578125e-05,
      "model_forward_time": 3.1348464488983154,
      "step": 8756
    },
    {
      "epoch": 3.34014892578125e-05,
      "step": 8756,
      "training_step_time": 10.342936992645264
    },
    {
      "epoch": 3.3405303955078124e-05,
      "model_forward_time": 3.1448755264282227,
      "step": 8757
    },
    {
      "epoch": 3.3405303955078124e-05,
      "step": 8757,
      "training_step_time": 9.931127071380615
    },
    {
      "epoch": 3.3405303955078124e-05,
      "model_forward_time": 3.139718770980835,
      "step": 8757
    },
    {
      "epoch": 3.3405303955078124e-05,
      "step": 8757,
      "training_step_time": 9.918246507644653
    },
    {
      "epoch": 3.3405303955078124e-05,
      "model_forward_time": 3.1382126808166504,
      "step": 8757
    },
    {
      "dynamics_loss_avg": 0.06234435886144638,
      "epoch": 3.3405303955078124e-05,
      "step": 8757
    },
    {
      "action_loss_avg": 0.023279009060934185,
      "epoch": 3.3405303955078124e-05,
      "step": 8757
    },
    {
      "epoch": 3.3405303955078124e-05,
      "step": 8757,
      "training_step_time": 9.922109842300415
    },
    {
      "epoch": 3.3405303955078124e-05,
      "model_forward_time": 3.137594223022461,
      "step": 8757
    },
    {
      "epoch": 3.3405303955078124e-05,
      "step": 8757,
      "training_step_time": 10.329901933670044
    },
    {
      "epoch": 3.340911865234375e-05,
      "model_forward_time": 3.1591367721557617,
      "step": 8758
    },
    {
      "epoch": 3.340911865234375e-05,
      "step": 8758,
      "training_step_time": 9.951803207397461
    },
    {
      "epoch": 3.340911865234375e-05,
      "model_forward_time": 3.1411800384521484,
      "step": 8758
    },
    {
      "epoch": 3.340911865234375e-05,
      "step": 8758,
      "training_step_time": 9.931015968322754
    },
    {
      "epoch": 3.340911865234375e-05,
      "model_forward_time": 3.145127296447754,
      "step": 8758
    },
    {
      "epoch": 3.340911865234375e-05,
      "step": 8758,
      "training_step_time": 9.927797794342041
    },
    {
      "epoch": 3.340911865234375e-05,
      "model_forward_time": 3.1380441188812256,
      "step": 8758
    },
    {
      "epoch": 3.340911865234375e-05,
      "step": 8758,
      "training_step_time": 10.323798894882202
    },
    {
      "epoch": 3.3412933349609376e-05,
      "model_forward_time": 2.4535655975341797,
      "step": 8759
    },
    {
      "epoch": 3.3412933349609376e-05,
      "step": 8759,
      "training_step_time": 7.8230226039886475
    },
    {
      "epoch": 3.3412933349609376e-05,
      "model_forward_time": 3.131614923477173,
      "step": 8759
    },
    {
      "epoch": 3.3412933349609376e-05,
      "step": 8759,
      "training_step_time": 9.913632154464722
    },
    {
      "epoch": 3.3412933349609376e-05,
      "model_forward_time": 3.1396548748016357,
      "step": 8759
    },
    {
      "epoch": 3.3412933349609376e-05,
      "step": 8759,
      "training_step_time": 9.918445348739624
    },
    {
      "epoch": 3.3412933349609376e-05,
      "model_forward_time": 2.4308063983917236,
      "step": 8759
    },
    {
      "epoch": 3.3412933349609376e-05,
      "step": 8759,
      "training_step_time": 10.325236797332764
    },
    {
      "epoch": 3.3416748046875e-05,
      "grad_norm": 0.4524073302745819,
      "learning_rate": 1e-05,
      "loss": 0.0836,
      "step": 8760
    },
    {
      "epoch": 3.3416748046875e-05,
      "model_forward_time": 3.139234781265259,
      "step": 8760
    },
    {
      "dynamics_loss_avg": 0.03547952473163605,
      "epoch": 3.3416748046875e-05,
      "step": 8760
    },
    {
      "action_loss_avg": 0.012267103744670748,
      "epoch": 3.3416748046875e-05,
      "step": 8760
    },
    {
      "epoch": 3.3416748046875e-05,
      "step": 8760,
      "training_step_time": 9.918510437011719
    },
    {
      "epoch": 3.3416748046875e-05,
      "model_forward_time": 3.1407783031463623,
      "step": 8760
    },
    {
      "epoch": 3.3416748046875e-05,
      "step": 8760,
      "training_step_time": 9.913343906402588
    },
    {
      "epoch": 3.3416748046875e-05,
      "model_forward_time": 3.4115993976593018,
      "step": 8760
    },
    {
      "epoch": 3.3416748046875e-05,
      "step": 8760,
      "training_step_time": 10.179923057556152
    },
    {
      "epoch": 3.3416748046875e-05,
      "model_forward_time": 3.1287381649017334,
      "step": 8760
    },
    {
      "epoch": 3.3416748046875e-05,
      "step": 8760,
      "training_step_time": 10.288598775863647
    },
    {
      "epoch": 3.342056274414062e-05,
      "model_forward_time": 3.1489737033843994,
      "step": 8761
    },
    {
      "epoch": 3.342056274414062e-05,
      "step": 8761,
      "training_step_time": 9.948475122451782
    },
    {
      "epoch": 3.342056274414062e-05,
      "model_forward_time": 3.1360385417938232,
      "step": 8761
    },
    {
      "epoch": 3.342056274414062e-05,
      "step": 8761,
      "training_step_time": 9.905239820480347
    },
    {
      "epoch": 3.342056274414062e-05,
      "model_forward_time": 3.435572624206543,
      "step": 8761
    },
    {
      "epoch": 3.342056274414062e-05,
      "step": 8761,
      "training_step_time": 10.21815013885498
    },
    {
      "epoch": 3.342056274414062e-05,
      "model_forward_time": 3.136169672012329,
      "step": 8761
    },
    {
      "epoch": 3.342056274414062e-05,
      "step": 8761,
      "training_step_time": 10.260475158691406
    },
    {
      "epoch": 3.342437744140625e-05,
      "model_forward_time": 3.155012607574463,
      "step": 8762
    },
    {
      "epoch": 3.342437744140625e-05,
      "step": 8762,
      "training_step_time": 9.933348417282104
    },
    {
      "epoch": 3.342437744140625e-05,
      "model_forward_time": 3.1386988162994385,
      "step": 8762
    },
    {
      "epoch": 3.342437744140625e-05,
      "step": 8762,
      "training_step_time": 9.921093225479126
    },
    {
      "epoch": 3.342437744140625e-05,
      "model_forward_time": 3.1369121074676514,
      "step": 8762
    },
    {
      "dynamics_loss_avg": 0.05598984677344561,
      "epoch": 3.342437744140625e-05,
      "step": 8762
    },
    {
      "action_loss_avg": 0.017765698255971074,
      "epoch": 3.342437744140625e-05,
      "step": 8762
    },
    {
      "epoch": 3.342437744140625e-05,
      "step": 8762,
      "training_step_time": 9.923318147659302
    },
    {
      "epoch": 3.342437744140625e-05,
      "model_forward_time": 2.7605514526367188,
      "step": 8762
    },
    {
      "epoch": 3.342437744140625e-05,
      "step": 8762,
      "training_step_time": 10.322052240371704
    },
    {
      "epoch": 3.3428192138671875e-05,
      "model_forward_time": 3.1562416553497314,
      "step": 8763
    },
    {
      "epoch": 3.3428192138671875e-05,
      "step": 8763,
      "training_step_time": 9.934255838394165
    },
    {
      "epoch": 3.3428192138671875e-05,
      "model_forward_time": 3.139511823654175,
      "step": 8763
    },
    {
      "epoch": 3.3428192138671875e-05,
      "step": 8763,
      "training_step_time": 9.918092489242554
    },
    {
      "epoch": 3.3428192138671875e-05,
      "model_forward_time": 3.137491464614868,
      "step": 8763
    },
    {
      "epoch": 3.3428192138671875e-05,
      "step": 8763,
      "training_step_time": 9.914150953292847
    },
    {
      "epoch": 3.3428192138671875e-05,
      "model_forward_time": 3.1456918716430664,
      "step": 8763
    },
    {
      "epoch": 3.3428192138671875e-05,
      "step": 8763,
      "training_step_time": 10.016601800918579
    },
    {
      "epoch": 3.34320068359375e-05,
      "model_forward_time": 3.168135404586792,
      "step": 8764
    },
    {
      "epoch": 3.34320068359375e-05,
      "step": 8764,
      "training_step_time": 10.255024909973145
    },
    {
      "epoch": 3.34320068359375e-05,
      "model_forward_time": 3.1464946269989014,
      "step": 8764
    },
    {
      "epoch": 3.34320068359375e-05,
      "step": 8764,
      "training_step_time": 9.925512552261353
    },
    {
      "epoch": 3.34320068359375e-05,
      "model_forward_time": 3.145214319229126,
      "step": 8764
    },
    {
      "epoch": 3.34320068359375e-05,
      "step": 8764,
      "training_step_time": 9.924007654190063
    },
    {
      "epoch": 3.34320068359375e-05,
      "model_forward_time": 2.4434542655944824,
      "step": 8764
    },
    {
      "epoch": 3.34320068359375e-05,
      "step": 8764,
      "training_step_time": 10.282134294509888
    },
    {
      "epoch": 3.343582153320313e-05,
      "model_forward_time": 3.1576526165008545,
      "step": 8765
    },
    {
      "dynamics_loss_avg": 0.0723835613578558,
      "epoch": 3.343582153320313e-05,
      "step": 8765
    },
    {
      "action_loss_avg": 0.023771705571562052,
      "epoch": 3.343582153320313e-05,
      "step": 8765
    },
    {
      "epoch": 3.343582153320313e-05,
      "step": 8765,
      "training_step_time": 9.942239046096802
    },
    {
      "epoch": 3.343582153320313e-05,
      "model_forward_time": 3.4566187858581543,
      "step": 8765
    },
    {
      "epoch": 3.343582153320313e-05,
      "step": 8765,
      "training_step_time": 10.232423305511475
    },
    {
      "epoch": 3.343582153320313e-05,
      "model_forward_time": 3.1425044536590576,
      "step": 8765
    },
    {
      "epoch": 3.343582153320313e-05,
      "step": 8765,
      "training_step_time": 9.92381477355957
    },
    {
      "epoch": 3.343582153320313e-05,
      "model_forward_time": 3.149421215057373,
      "step": 8765
    },
    {
      "epoch": 3.343582153320313e-05,
      "step": 8765,
      "training_step_time": 10.288137435913086
    },
    {
      "epoch": 3.343963623046875e-05,
      "model_forward_time": 3.1562399864196777,
      "step": 8766
    },
    {
      "epoch": 3.343963623046875e-05,
      "step": 8766,
      "training_step_time": 9.932285070419312
    },
    {
      "epoch": 3.343963623046875e-05,
      "model_forward_time": 3.1427626609802246,
      "step": 8766
    },
    {
      "epoch": 3.343963623046875e-05,
      "step": 8766,
      "training_step_time": 9.918455839157104
    },
    {
      "epoch": 3.343963623046875e-05,
      "model_forward_time": 3.1489574909210205,
      "step": 8766
    },
    {
      "epoch": 3.343963623046875e-05,
      "step": 8766,
      "training_step_time": 9.929572582244873
    },
    {
      "epoch": 3.343963623046875e-05,
      "model_forward_time": 3.461076021194458,
      "step": 8766
    },
    {
      "epoch": 3.343963623046875e-05,
      "step": 8766,
      "training_step_time": 10.336004972457886
    },
    {
      "epoch": 3.344345092773437e-05,
      "model_forward_time": 3.164525032043457,
      "step": 8767
    },
    {
      "epoch": 3.344345092773437e-05,
      "step": 8767,
      "training_step_time": 9.941975355148315
    },
    {
      "epoch": 3.344345092773437e-05,
      "model_forward_time": 3.140218496322632,
      "step": 8767
    },
    {
      "epoch": 3.344345092773437e-05,
      "step": 8767,
      "training_step_time": 9.914582252502441
    },
    {
      "epoch": 3.344345092773437e-05,
      "model_forward_time": 3.143003463745117,
      "step": 8767
    },
    {
      "dynamics_loss_avg": 0.05006309580057859,
      "epoch": 3.344345092773437e-05,
      "step": 8767
    },
    {
      "action_loss_avg": 0.0179222269449383,
      "epoch": 3.344345092773437e-05,
      "step": 8767
    },
    {
      "epoch": 3.344345092773437e-05,
      "step": 8767,
      "training_step_time": 9.919223308563232
    },
    {
      "epoch": 3.344345092773437e-05,
      "model_forward_time": 3.4174656867980957,
      "step": 8767
    },
    {
      "epoch": 3.344345092773437e-05,
      "step": 8767,
      "training_step_time": 10.284637689590454
    },
    {
      "epoch": 3.3447265625e-05,
      "model_forward_time": 3.158031940460205,
      "step": 8768
    },
    {
      "epoch": 3.3447265625e-05,
      "step": 8768,
      "training_step_time": 9.934990644454956
    },
    {
      "epoch": 3.3447265625e-05,
      "model_forward_time": 3.140108346939087,
      "step": 8768
    },
    {
      "epoch": 3.3447265625e-05,
      "step": 8768,
      "training_step_time": 9.914483785629272
    },
    {
      "epoch": 3.3447265625e-05,
      "model_forward_time": 3.1445064544677734,
      "step": 8768
    },
    {
      "epoch": 3.3447265625e-05,
      "step": 8768,
      "training_step_time": 9.916640520095825
    },
    {
      "epoch": 3.3447265625e-05,
      "model_forward_time": 3.1424014568328857,
      "step": 8768
    },
    {
      "epoch": 3.3447265625e-05,
      "step": 8768,
      "training_step_time": 10.033029317855835
    },
    {
      "epoch": 3.3451080322265625e-05,
      "model_forward_time": 3.436249017715454,
      "step": 8769
    },
    {
      "epoch": 3.3451080322265625e-05,
      "step": 8769,
      "training_step_time": 10.20887303352356
    },
    {
      "epoch": 3.3451080322265625e-05,
      "model_forward_time": 3.1390554904937744,
      "step": 8769
    },
    {
      "epoch": 3.3451080322265625e-05,
      "step": 8769,
      "training_step_time": 9.916557550430298
    },
    {
      "epoch": 3.3451080322265625e-05,
      "model_forward_time": 2.4376919269561768,
      "step": 8769
    },
    {
      "epoch": 3.3451080322265625e-05,
      "step": 8769,
      "training_step_time": 7.463027477264404
    },
    {
      "epoch": 3.3451080322265625e-05,
      "model_forward_time": 3.1346840858459473,
      "step": 8769
    },
    {
      "epoch": 3.3451080322265625e-05,
      "step": 8769,
      "training_step_time": 10.287568807601929
    },
    {
      "epoch": 3.345489501953125e-05,
      "grad_norm": 0.30689892172813416,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 8770
    },
    {
      "epoch": 3.345489501953125e-05,
      "model_forward_time": 3.1469619274139404,
      "step": 8770
    },
    {
      "dynamics_loss_avg": 0.05088107399642468,
      "epoch": 3.345489501953125e-05,
      "step": 8770
    },
    {
      "action_loss_avg": 0.021804772783070803,
      "epoch": 3.345489501953125e-05,
      "step": 8770
    },
    {
      "epoch": 3.345489501953125e-05,
      "step": 8770,
      "training_step_time": 9.919142961502075
    },
    {
      "epoch": 3.345489501953125e-05,
      "model_forward_time": 3.431097984313965,
      "step": 8770
    },
    {
      "epoch": 3.345489501953125e-05,
      "step": 8770,
      "training_step_time": 10.198130369186401
    },
    {
      "epoch": 3.345489501953125e-05,
      "model_forward_time": 3.1391005516052246,
      "step": 8770
    },
    {
      "epoch": 3.345489501953125e-05,
      "step": 8770,
      "training_step_time": 9.906574249267578
    },
    {
      "epoch": 3.345489501953125e-05,
      "model_forward_time": 2.4400711059570312,
      "step": 8770
    },
    {
      "epoch": 3.345489501953125e-05,
      "step": 8770,
      "training_step_time": 10.306495189666748
    },
    {
      "epoch": 3.345870971679688e-05,
      "model_forward_time": 3.1495676040649414,
      "step": 8771
    },
    {
      "epoch": 3.345870971679688e-05,
      "step": 8771,
      "training_step_time": 9.920959949493408
    },
    {
      "epoch": 3.345870971679688e-05,
      "model_forward_time": 2.4368937015533447,
      "step": 8771
    },
    {
      "epoch": 3.345870971679688e-05,
      "step": 8771,
      "training_step_time": 7.457409143447876
    },
    {
      "epoch": 3.345870971679688e-05,
      "model_forward_time": 3.4390804767608643,
      "step": 8771
    },
    {
      "epoch": 3.345870971679688e-05,
      "step": 8771,
      "training_step_time": 10.214120149612427
    },
    {
      "epoch": 3.345870971679688e-05,
      "model_forward_time": 3.142737627029419,
      "step": 8771
    },
    {
      "epoch": 3.345870971679688e-05,
      "step": 8771,
      "training_step_time": 10.29878568649292
    },
    {
      "epoch": 3.34625244140625e-05,
      "model_forward_time": 3.163017511367798,
      "step": 8772
    },
    {
      "epoch": 3.34625244140625e-05,
      "step": 8772,
      "training_step_time": 9.936479091644287
    },
    {
      "epoch": 3.34625244140625e-05,
      "model_forward_time": 3.152172803878784,
      "step": 8772
    },
    {
      "epoch": 3.34625244140625e-05,
      "step": 8772,
      "training_step_time": 9.924912214279175
    },
    {
      "epoch": 3.34625244140625e-05,
      "model_forward_time": 3.1408064365386963,
      "step": 8772
    },
    {
      "dynamics_loss_avg": 0.07006773706525564,
      "epoch": 3.34625244140625e-05,
      "step": 8772
    },
    {
      "action_loss_avg": 0.020220883749425413,
      "epoch": 3.34625244140625e-05,
      "step": 8772
    },
    {
      "epoch": 3.34625244140625e-05,
      "step": 8772,
      "training_step_time": 9.918375730514526
    },
    {
      "epoch": 3.34625244140625e-05,
      "model_forward_time": 3.456827402114868,
      "step": 8772
    },
    {
      "epoch": 3.34625244140625e-05,
      "step": 8772,
      "training_step_time": 10.325747013092041
    },
    {
      "epoch": 3.3466339111328124e-05,
      "model_forward_time": 3.165656328201294,
      "step": 8773
    },
    {
      "epoch": 3.3466339111328124e-05,
      "step": 8773,
      "training_step_time": 9.94210410118103
    },
    {
      "epoch": 3.3466339111328124e-05,
      "model_forward_time": 3.145963191986084,
      "step": 8773
    },
    {
      "epoch": 3.3466339111328124e-05,
      "step": 8773,
      "training_step_time": 9.92105746269226
    },
    {
      "epoch": 3.3466339111328124e-05,
      "model_forward_time": 3.147542715072632,
      "step": 8773
    },
    {
      "epoch": 3.3466339111328124e-05,
      "step": 8773,
      "training_step_time": 9.928306102752686
    },
    {
      "epoch": 3.3466339111328124e-05,
      "model_forward_time": 3.1542038917541504,
      "step": 8773
    },
    {
      "epoch": 3.3466339111328124e-05,
      "step": 8773,
      "training_step_time": 10.287046194076538
    },
    {
      "epoch": 3.347015380859375e-05,
      "model_forward_time": 2.7482504844665527,
      "step": 8774
    },
    {
      "epoch": 3.347015380859375e-05,
      "step": 8774,
      "training_step_time": 7.795277833938599
    },
    {
      "epoch": 3.347015380859375e-05,
      "model_forward_time": 3.156158924102783,
      "step": 8774
    },
    {
      "epoch": 3.347015380859375e-05,
      "step": 8774,
      "training_step_time": 9.942949056625366
    },
    {
      "epoch": 3.347015380859375e-05,
      "model_forward_time": 3.143526792526245,
      "step": 8774
    },
    {
      "epoch": 3.347015380859375e-05,
      "step": 8774,
      "training_step_time": 9.921594619750977
    },
    {
      "epoch": 3.347015380859375e-05,
      "model_forward_time": 3.142153024673462,
      "step": 8774
    },
    {
      "epoch": 3.347015380859375e-05,
      "step": 8774,
      "training_step_time": 10.015030860900879
    },
    {
      "epoch": 3.3473968505859376e-05,
      "model_forward_time": 3.1613824367523193,
      "step": 8775
    },
    {
      "dynamics_loss_avg": 0.05226033199578524,
      "epoch": 3.3473968505859376e-05,
      "step": 8775
    },
    {
      "action_loss_avg": 0.019491824507713317,
      "epoch": 3.3473968505859376e-05,
      "step": 8775
    },
    {
      "epoch": 3.3473968505859376e-05,
      "step": 8775,
      "training_step_time": 9.948741674423218
    },
    {
      "epoch": 3.3473968505859376e-05,
      "model_forward_time": 3.1408603191375732,
      "step": 8775
    },
    {
      "epoch": 3.3473968505859376e-05,
      "step": 8775,
      "training_step_time": 10.235694646835327
    },
    {
      "epoch": 3.3473968505859376e-05,
      "model_forward_time": 3.1452059745788574,
      "step": 8775
    },
    {
      "epoch": 3.3473968505859376e-05,
      "step": 8775,
      "training_step_time": 9.922788143157959
    },
    {
      "epoch": 3.3473968505859376e-05,
      "model_forward_time": 3.1506967544555664,
      "step": 8775
    },
    {
      "epoch": 3.3473968505859376e-05,
      "step": 8775,
      "training_step_time": 10.26777982711792
    },
    {
      "epoch": 3.3477783203125e-05,
      "model_forward_time": 3.1824114322662354,
      "step": 8776
    },
    {
      "epoch": 3.3477783203125e-05,
      "step": 8776,
      "training_step_time": 9.97365427017212
    },
    {
      "epoch": 3.3477783203125e-05,
      "model_forward_time": 3.14518666267395,
      "step": 8776
    },
    {
      "epoch": 3.3477783203125e-05,
      "step": 8776,
      "training_step_time": 9.927515983581543
    },
    {
      "epoch": 3.3477783203125e-05,
      "model_forward_time": 3.4511001110076904,
      "step": 8776
    },
    {
      "epoch": 3.3477783203125e-05,
      "step": 8776,
      "training_step_time": 10.227477312088013
    },
    {
      "epoch": 3.3477783203125e-05,
      "model_forward_time": 3.1545662879943848,
      "step": 8776
    },
    {
      "epoch": 3.3477783203125e-05,
      "step": 8776,
      "training_step_time": 10.338594675064087
    },
    {
      "epoch": 3.348159790039062e-05,
      "model_forward_time": 3.178521156311035,
      "step": 8777
    },
    {
      "epoch": 3.348159790039062e-05,
      "step": 8777,
      "training_step_time": 9.954318284988403
    },
    {
      "epoch": 3.348159790039062e-05,
      "model_forward_time": 2.442645788192749,
      "step": 8777
    },
    {
      "epoch": 3.348159790039062e-05,
      "step": 8777,
      "training_step_time": 7.470429420471191
    },
    {
      "epoch": 3.348159790039062e-05,
      "model_forward_time": 3.1464574337005615,
      "step": 8777
    },
    {
      "dynamics_loss_avg": 0.05331302648410201,
      "epoch": 3.348159790039062e-05,
      "step": 8777
    },
    {
      "action_loss_avg": 0.017689059698022903,
      "epoch": 3.348159790039062e-05,
      "step": 8777
    },
    {
      "epoch": 3.348159790039062e-05,
      "step": 8777,
      "training_step_time": 9.929641246795654
    },
    {
      "epoch": 3.348159790039062e-05,
      "model_forward_time": 3.145659923553467,
      "step": 8777
    },
    {
      "epoch": 3.348159790039062e-05,
      "step": 8777,
      "training_step_time": 10.30199408531189
    },
    {
      "epoch": 3.348541259765625e-05,
      "model_forward_time": 3.1559531688690186,
      "step": 8778
    },
    {
      "epoch": 3.348541259765625e-05,
      "step": 8778,
      "training_step_time": 9.932881593704224
    },
    {
      "epoch": 3.348541259765625e-05,
      "model_forward_time": 3.134321928024292,
      "step": 8778
    },
    {
      "epoch": 3.348541259765625e-05,
      "step": 8778,
      "training_step_time": 9.908348083496094
    },
    {
      "epoch": 3.348541259765625e-05,
      "model_forward_time": 3.1399970054626465,
      "step": 8778
    },
    {
      "epoch": 3.348541259765625e-05,
      "step": 8778,
      "training_step_time": 9.915193319320679
    },
    {
      "epoch": 3.348541259765625e-05,
      "model_forward_time": 3.137953996658325,
      "step": 8778
    },
    {
      "epoch": 3.348541259765625e-05,
      "step": 8778,
      "training_step_time": 10.329663753509521
    },
    {
      "epoch": 3.3489227294921875e-05,
      "model_forward_time": 3.1463496685028076,
      "step": 8779
    },
    {
      "epoch": 3.3489227294921875e-05,
      "step": 8779,
      "training_step_time": 10.238733053207397
    },
    {
      "epoch": 3.3489227294921875e-05,
      "model_forward_time": 3.153066396713257,
      "step": 8779
    },
    {
      "epoch": 3.3489227294921875e-05,
      "step": 8779,
      "training_step_time": 9.928503513336182
    },
    {
      "epoch": 3.3489227294921875e-05,
      "model_forward_time": 3.1439566612243652,
      "step": 8779
    },
    {
      "epoch": 3.3489227294921875e-05,
      "step": 8779,
      "training_step_time": 9.914553880691528
    },
    {
      "epoch": 3.3489227294921875e-05,
      "model_forward_time": 3.140925168991089,
      "step": 8779
    },
    {
      "epoch": 3.3489227294921875e-05,
      "step": 8779,
      "training_step_time": 10.026732206344604
    },
    {
      "epoch": 3.34930419921875e-05,
      "grad_norm": 0.2541750967502594,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 8780
    },
    {
      "epoch": 3.34930419921875e-05,
      "model_forward_time": 2.448998212814331,
      "step": 8780
    },
    {
      "dynamics_loss_avg": 0.06644765753298998,
      "epoch": 3.34930419921875e-05,
      "step": 8780
    },
    {
      "action_loss_avg": 0.022420202614739537,
      "epoch": 3.34930419921875e-05,
      "step": 8780
    },
    {
      "epoch": 3.34930419921875e-05,
      "step": 8780,
      "training_step_time": 7.488579034805298
    },
    {
      "epoch": 3.34930419921875e-05,
      "model_forward_time": 3.1389834880828857,
      "step": 8780
    },
    {
      "epoch": 3.34930419921875e-05,
      "step": 8780,
      "training_step_time": 9.915756702423096
    },
    {
      "epoch": 3.34930419921875e-05,
      "model_forward_time": 3.4622654914855957,
      "step": 8780
    },
    {
      "epoch": 3.34930419921875e-05,
      "step": 8780,
      "training_step_time": 10.237303972244263
    },
    {
      "epoch": 3.34930419921875e-05,
      "model_forward_time": 3.1436655521392822,
      "step": 8780
    },
    {
      "epoch": 3.34930419921875e-05,
      "step": 8780,
      "training_step_time": 10.303250312805176
    },
    {
      "epoch": 3.349685668945313e-05,
      "model_forward_time": 3.1603169441223145,
      "step": 8781
    },
    {
      "epoch": 3.349685668945313e-05,
      "step": 8781,
      "training_step_time": 9.936019659042358
    },
    {
      "epoch": 3.349685668945313e-05,
      "model_forward_time": 3.1452090740203857,
      "step": 8781
    },
    {
      "epoch": 3.349685668945313e-05,
      "step": 8781,
      "training_step_time": 9.920207023620605
    },
    {
      "epoch": 3.349685668945313e-05,
      "model_forward_time": 3.1395256519317627,
      "step": 8781
    },
    {
      "epoch": 3.349685668945313e-05,
      "step": 8781,
      "training_step_time": 10.194287061691284
    },
    {
      "epoch": 3.349685668945313e-05,
      "model_forward_time": 3.139009475708008,
      "step": 8781
    },
    {
      "epoch": 3.349685668945313e-05,
      "step": 8781,
      "training_step_time": 10.322978734970093
    },
    {
      "epoch": 3.350067138671875e-05,
      "model_forward_time": 3.1644325256347656,
      "step": 8782
    },
    {
      "epoch": 3.350067138671875e-05,
      "step": 8782,
      "training_step_time": 9.942933082580566
    },
    {
      "epoch": 3.350067138671875e-05,
      "model_forward_time": 3.1451661586761475,
      "step": 8782
    },
    {
      "epoch": 3.350067138671875e-05,
      "step": 8782,
      "training_step_time": 9.922446489334106
    },
    {
      "epoch": 3.350067138671875e-05,
      "model_forward_time": 3.142193078994751,
      "step": 8782
    },
    {
      "dynamics_loss_avg": 0.05756182000041008,
      "epoch": 3.350067138671875e-05,
      "step": 8782
    },
    {
      "action_loss_avg": 0.0160100718960166,
      "epoch": 3.350067138671875e-05,
      "step": 8782
    },
    {
      "epoch": 3.350067138671875e-05,
      "step": 8782,
      "training_step_time": 9.921711921691895
    },
    {
      "epoch": 3.350067138671875e-05,
      "model_forward_time": 3.450356960296631,
      "step": 8782
    },
    {
      "epoch": 3.350067138671875e-05,
      "step": 8782,
      "training_step_time": 10.320304155349731
    },
    {
      "epoch": 3.350448608398437e-05,
      "model_forward_time": 2.4646947383880615,
      "step": 8783
    },
    {
      "epoch": 3.350448608398437e-05,
      "step": 8783,
      "training_step_time": 7.507349252700806
    },
    {
      "epoch": 3.350448608398437e-05,
      "model_forward_time": 3.157949924468994,
      "step": 8783
    },
    {
      "epoch": 3.350448608398437e-05,
      "step": 8783,
      "training_step_time": 9.94871735572815
    },
    {
      "epoch": 3.350448608398437e-05,
      "model_forward_time": 3.153093099594116,
      "step": 8783
    },
    {
      "epoch": 3.350448608398437e-05,
      "step": 8783,
      "training_step_time": 9.936812400817871
    },
    {
      "epoch": 3.350448608398437e-05,
      "model_forward_time": 3.1495893001556396,
      "step": 8783
    },
    {
      "epoch": 3.350448608398437e-05,
      "step": 8783,
      "training_step_time": 10.02829098701477
    },
    {
      "epoch": 3.350830078125e-05,
      "model_forward_time": 3.1728415489196777,
      "step": 8784
    },
    {
      "epoch": 3.350830078125e-05,
      "step": 8784,
      "training_step_time": 10.270444631576538
    },
    {
      "epoch": 3.350830078125e-05,
      "model_forward_time": 3.1537718772888184,
      "step": 8784
    },
    {
      "epoch": 3.350830078125e-05,
      "step": 8784,
      "training_step_time": 9.93473768234253
    },
    {
      "epoch": 3.350830078125e-05,
      "model_forward_time": 2.4428350925445557,
      "step": 8784
    },
    {
      "epoch": 3.350830078125e-05,
      "step": 8784,
      "training_step_time": 7.477001190185547
    },
    {
      "epoch": 3.350830078125e-05,
      "model_forward_time": 3.145219326019287,
      "step": 8784
    },
    {
      "epoch": 3.350830078125e-05,
      "step": 8784,
      "training_step_time": 10.02005386352539
    },
    {
      "epoch": 3.3512115478515626e-05,
      "model_forward_time": 3.165984630584717,
      "step": 8785
    },
    {
      "dynamics_loss_avg": 0.06240201480686665,
      "epoch": 3.3512115478515626e-05,
      "step": 8785
    },
    {
      "action_loss_avg": 0.02125036921352148,
      "epoch": 3.3512115478515626e-05,
      "step": 8785
    },
    {
      "epoch": 3.3512115478515626e-05,
      "step": 8785,
      "training_step_time": 9.943590641021729
    },
    {
      "epoch": 3.3512115478515626e-05,
      "model_forward_time": 2.4363455772399902,
      "step": 8785
    },
    {
      "epoch": 3.3512115478515626e-05,
      "step": 8785,
      "training_step_time": 7.724666595458984
    },
    {
      "epoch": 3.3512115478515626e-05,
      "model_forward_time": 3.1347415447235107,
      "step": 8785
    },
    {
      "epoch": 3.3512115478515626e-05,
      "step": 8785,
      "training_step_time": 9.906722068786621
    },
    {
      "epoch": 3.3512115478515626e-05,
      "model_forward_time": 3.137326717376709,
      "step": 8785
    },
    {
      "epoch": 3.3512115478515626e-05,
      "step": 8785,
      "training_step_time": 10.241243839263916
    },
    {
      "epoch": 3.351593017578125e-05,
      "model_forward_time": 3.141545057296753,
      "step": 8786
    },
    {
      "epoch": 3.351593017578125e-05,
      "step": 8786,
      "training_step_time": 9.910118818283081
    },
    {
      "epoch": 3.351593017578125e-05,
      "model_forward_time": 3.1324963569641113,
      "step": 8786
    },
    {
      "epoch": 3.351593017578125e-05,
      "step": 8786,
      "training_step_time": 9.90007495880127
    },
    {
      "epoch": 3.351593017578125e-05,
      "model_forward_time": 3.429576873779297,
      "step": 8786
    },
    {
      "epoch": 3.351593017578125e-05,
      "step": 8786,
      "training_step_time": 10.197482824325562
    },
    {
      "epoch": 3.351593017578125e-05,
      "model_forward_time": 3.1433470249176025,
      "step": 8786
    },
    {
      "epoch": 3.351593017578125e-05,
      "step": 8786,
      "training_step_time": 10.250472068786621
    },
    {
      "epoch": 3.351974487304688e-05,
      "model_forward_time": 3.145170211791992,
      "step": 8787
    },
    {
      "epoch": 3.351974487304688e-05,
      "step": 8787,
      "training_step_time": 9.914022445678711
    },
    {
      "epoch": 3.351974487304688e-05,
      "model_forward_time": 3.13454008102417,
      "step": 8787
    },
    {
      "epoch": 3.351974487304688e-05,
      "step": 8787,
      "training_step_time": 9.906839370727539
    },
    {
      "epoch": 3.351974487304688e-05,
      "model_forward_time": 3.1408395767211914,
      "step": 8787
    },
    {
      "dynamics_loss_avg": 0.06642743330448866,
      "epoch": 3.351974487304688e-05,
      "step": 8787
    },
    {
      "action_loss_avg": 0.021722256578505038,
      "epoch": 3.351974487304688e-05,
      "step": 8787
    },
    {
      "epoch": 3.351974487304688e-05,
      "step": 8787,
      "training_step_time": 9.920011043548584
    },
    {
      "epoch": 3.351974487304688e-05,
      "model_forward_time": 3.4359869956970215,
      "step": 8787
    },
    {
      "epoch": 3.351974487304688e-05,
      "step": 8787,
      "training_step_time": 10.31454586982727
    },
    {
      "epoch": 3.35235595703125e-05,
      "model_forward_time": 3.152251958847046,
      "step": 8788
    },
    {
      "epoch": 3.35235595703125e-05,
      "step": 8788,
      "training_step_time": 9.932812213897705
    },
    {
      "epoch": 3.35235595703125e-05,
      "model_forward_time": 3.1384358406066895,
      "step": 8788
    },
    {
      "epoch": 3.35235595703125e-05,
      "step": 8788,
      "training_step_time": 9.915727615356445
    },
    {
      "epoch": 3.35235595703125e-05,
      "model_forward_time": 3.147691488265991,
      "step": 8788
    },
    {
      "epoch": 3.35235595703125e-05,
      "step": 8788,
      "training_step_time": 9.92457127571106
    },
    {
      "epoch": 3.35235595703125e-05,
      "model_forward_time": 3.140577793121338,
      "step": 8788
    },
    {
      "epoch": 3.35235595703125e-05,
      "step": 8788,
      "training_step_time": 10.278613805770874
    },
    {
      "epoch": 3.3527374267578124e-05,
      "model_forward_time": 3.4694297313690186,
      "step": 8789
    },
    {
      "epoch": 3.3527374267578124e-05,
      "step": 8789,
      "training_step_time": 10.245122909545898
    },
    {
      "epoch": 3.3527374267578124e-05,
      "model_forward_time": 3.1419544219970703,
      "step": 8789
    },
    {
      "epoch": 3.3527374267578124e-05,
      "step": 8789,
      "training_step_time": 9.914538383483887
    },
    {
      "epoch": 3.3527374267578124e-05,
      "model_forward_time": 3.1442081928253174,
      "step": 8789
    },
    {
      "epoch": 3.3527374267578124e-05,
      "step": 8789,
      "training_step_time": 9.919219493865967
    },
    {
      "epoch": 3.3527374267578124e-05,
      "model_forward_time": 3.1458306312561035,
      "step": 8789
    },
    {
      "epoch": 3.3527374267578124e-05,
      "step": 8789,
      "training_step_time": 10.03208613395691
    },
    {
      "epoch": 3.353118896484375e-05,
      "grad_norm": 0.3161143362522125,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 8790
    },
    {
      "epoch": 3.353118896484375e-05,
      "model_forward_time": 3.1662187576293945,
      "step": 8790
    },
    {
      "dynamics_loss_avg": 0.06409241594374179,
      "epoch": 3.353118896484375e-05,
      "step": 8790
    },
    {
      "action_loss_avg": 0.026169203128665684,
      "epoch": 3.353118896484375e-05,
      "step": 8790
    },
    {
      "epoch": 3.353118896484375e-05,
      "step": 8790,
      "training_step_time": 9.947265148162842
    },
    {
      "epoch": 3.353118896484375e-05,
      "model_forward_time": 3.4074625968933105,
      "step": 8790
    },
    {
      "epoch": 3.353118896484375e-05,
      "step": 8790,
      "training_step_time": 10.181546449661255
    },
    {
      "epoch": 3.353118896484375e-05,
      "model_forward_time": 3.140734910964966,
      "step": 8790
    },
    {
      "epoch": 3.353118896484375e-05,
      "step": 8790,
      "training_step_time": 9.91598653793335
    },
    {
      "epoch": 3.353118896484375e-05,
      "model_forward_time": 3.136772394180298,
      "step": 8790
    },
    {
      "epoch": 3.353118896484375e-05,
      "step": 8790,
      "training_step_time": 10.317407608032227
    },
    {
      "epoch": 3.3535003662109376e-05,
      "model_forward_time": 3.15903377532959,
      "step": 8791
    },
    {
      "epoch": 3.3535003662109376e-05,
      "step": 8791,
      "training_step_time": 9.936513662338257
    },
    {
      "epoch": 3.3535003662109376e-05,
      "model_forward_time": 3.1405084133148193,
      "step": 8791
    },
    {
      "epoch": 3.3535003662109376e-05,
      "step": 8791,
      "training_step_time": 9.912063598632812
    },
    {
      "epoch": 3.3535003662109376e-05,
      "model_forward_time": 3.4150819778442383,
      "step": 8791
    },
    {
      "epoch": 3.3535003662109376e-05,
      "step": 8791,
      "training_step_time": 10.186869859695435
    },
    {
      "epoch": 3.3535003662109376e-05,
      "model_forward_time": 3.137491464614868,
      "step": 8791
    },
    {
      "epoch": 3.3535003662109376e-05,
      "step": 8791,
      "training_step_time": 10.347290992736816
    },
    {
      "epoch": 3.3538818359375e-05,
      "model_forward_time": 3.153388738632202,
      "step": 8792
    },
    {
      "epoch": 3.3538818359375e-05,
      "step": 8792,
      "training_step_time": 9.937301397323608
    },
    {
      "epoch": 3.3538818359375e-05,
      "model_forward_time": 3.148791551589966,
      "step": 8792
    },
    {
      "epoch": 3.3538818359375e-05,
      "step": 8792,
      "training_step_time": 9.92120885848999
    },
    {
      "epoch": 3.3538818359375e-05,
      "model_forward_time": 3.147878408432007,
      "step": 8792
    },
    {
      "dynamics_loss_avg": 0.06520500145852566,
      "epoch": 3.3538818359375e-05,
      "step": 8792
    },
    {
      "action_loss_avg": 0.022525019897148012,
      "epoch": 3.3538818359375e-05,
      "step": 8792
    },
    {
      "epoch": 3.3538818359375e-05,
      "step": 8792,
      "training_step_time": 9.930115938186646
    },
    {
      "epoch": 3.3538818359375e-05,
      "model_forward_time": 3.423976421356201,
      "step": 8792
    },
    {
      "epoch": 3.3538818359375e-05,
      "step": 8792,
      "training_step_time": 10.298596620559692
    },
    {
      "epoch": 3.354263305664062e-05,
      "model_forward_time": 3.1574184894561768,
      "step": 8793
    },
    {
      "epoch": 3.354263305664062e-05,
      "step": 8793,
      "training_step_time": 9.93680453300476
    },
    {
      "epoch": 3.354263305664062e-05,
      "model_forward_time": 3.1428773403167725,
      "step": 8793
    },
    {
      "epoch": 3.354263305664062e-05,
      "step": 8793,
      "training_step_time": 9.9234299659729
    },
    {
      "epoch": 3.354263305664062e-05,
      "model_forward_time": 3.1391279697418213,
      "step": 8793
    },
    {
      "epoch": 3.354263305664062e-05,
      "step": 8793,
      "training_step_time": 9.913773775100708
    },
    {
      "epoch": 3.354263305664062e-05,
      "model_forward_time": 2.436035633087158,
      "step": 8793
    },
    {
      "epoch": 3.354263305664062e-05,
      "step": 8793,
      "training_step_time": 10.273098230361938
    },
    {
      "epoch": 3.354644775390625e-05,
      "model_forward_time": 3.4304416179656982,
      "step": 8794
    },
    {
      "epoch": 3.354644775390625e-05,
      "step": 8794,
      "training_step_time": 10.200846672058105
    },
    {
      "epoch": 3.354644775390625e-05,
      "model_forward_time": 3.134972333908081,
      "step": 8794
    },
    {
      "epoch": 3.354644775390625e-05,
      "step": 8794,
      "training_step_time": 9.904093980789185
    },
    {
      "epoch": 3.354644775390625e-05,
      "model_forward_time": 3.1350796222686768,
      "step": 8794
    },
    {
      "epoch": 3.354644775390625e-05,
      "step": 8794,
      "training_step_time": 9.907027244567871
    },
    {
      "epoch": 3.354644775390625e-05,
      "model_forward_time": 3.1420888900756836,
      "step": 8794
    },
    {
      "epoch": 3.354644775390625e-05,
      "step": 8794,
      "training_step_time": 10.298184394836426
    },
    {
      "epoch": 3.3550262451171875e-05,
      "model_forward_time": 3.157710075378418,
      "step": 8795
    },
    {
      "dynamics_loss_avg": 0.058815565519034864,
      "epoch": 3.3550262451171875e-05,
      "step": 8795
    },
    {
      "action_loss_avg": 0.021118160523474216,
      "epoch": 3.3550262451171875e-05,
      "step": 8795
    },
    {
      "epoch": 3.3550262451171875e-05,
      "step": 8795,
      "training_step_time": 9.934342861175537
    },
    {
      "epoch": 3.3550262451171875e-05,
      "model_forward_time": 3.1312263011932373,
      "step": 8795
    },
    {
      "epoch": 3.3550262451171875e-05,
      "step": 8795,
      "training_step_time": 10.16891074180603
    },
    {
      "epoch": 3.3550262451171875e-05,
      "model_forward_time": 3.1372501850128174,
      "step": 8795
    },
    {
      "epoch": 3.3550262451171875e-05,
      "step": 8795,
      "training_step_time": 9.90839695930481
    },
    {
      "epoch": 3.3550262451171875e-05,
      "model_forward_time": 3.1409730911254883,
      "step": 8795
    },
    {
      "epoch": 3.3550262451171875e-05,
      "step": 8795,
      "training_step_time": 10.042089462280273
    },
    {
      "epoch": 3.35540771484375e-05,
      "model_forward_time": 3.1508755683898926,
      "step": 8796
    },
    {
      "epoch": 3.35540771484375e-05,
      "step": 8796,
      "training_step_time": 9.928144454956055
    },
    {
      "epoch": 3.35540771484375e-05,
      "model_forward_time": 3.137063503265381,
      "step": 8796
    },
    {
      "epoch": 3.35540771484375e-05,
      "step": 8796,
      "training_step_time": 9.914627075195312
    },
    {
      "epoch": 3.35540771484375e-05,
      "model_forward_time": 3.4034152030944824,
      "step": 8796
    },
    {
      "epoch": 3.35540771484375e-05,
      "step": 8796,
      "training_step_time": 10.176196098327637
    },
    {
      "epoch": 3.35540771484375e-05,
      "model_forward_time": 3.1394455432891846,
      "step": 8796
    },
    {
      "epoch": 3.35540771484375e-05,
      "step": 8796,
      "training_step_time": 10.3091139793396
    },
    {
      "epoch": 3.355789184570313e-05,
      "model_forward_time": 2.447965383529663,
      "step": 8797
    },
    {
      "epoch": 3.355789184570313e-05,
      "step": 8797,
      "training_step_time": 7.476361036300659
    },
    {
      "epoch": 3.355789184570313e-05,
      "model_forward_time": 3.1289432048797607,
      "step": 8797
    },
    {
      "epoch": 3.355789184570313e-05,
      "step": 8797,
      "training_step_time": 9.899426221847534
    },
    {
      "epoch": 3.355789184570313e-05,
      "model_forward_time": 3.1414554119110107,
      "step": 8797
    },
    {
      "dynamics_loss_avg": 0.06700990367680788,
      "epoch": 3.355789184570313e-05,
      "step": 8797
    },
    {
      "action_loss_avg": 0.025447111949324606,
      "epoch": 3.355789184570313e-05,
      "step": 8797
    },
    {
      "epoch": 3.355789184570313e-05,
      "step": 8797,
      "training_step_time": 9.914422035217285
    },
    {
      "epoch": 3.355789184570313e-05,
      "model_forward_time": 3.1371774673461914,
      "step": 8797
    },
    {
      "epoch": 3.355789184570313e-05,
      "step": 8797,
      "training_step_time": 10.275079488754272
    },
    {
      "epoch": 3.356170654296875e-05,
      "model_forward_time": 3.1595561504364014,
      "step": 8798
    },
    {
      "epoch": 3.356170654296875e-05,
      "step": 8798,
      "training_step_time": 9.933717489242554
    },
    {
      "epoch": 3.356170654296875e-05,
      "model_forward_time": 3.1346278190612793,
      "step": 8798
    },
    {
      "epoch": 3.356170654296875e-05,
      "step": 8798,
      "training_step_time": 9.90078616142273
    },
    {
      "epoch": 3.356170654296875e-05,
      "model_forward_time": 3.138728380203247,
      "step": 8798
    },
    {
      "epoch": 3.356170654296875e-05,
      "step": 8798,
      "training_step_time": 9.905102729797363
    },
    {
      "epoch": 3.356170654296875e-05,
      "model_forward_time": 3.132803201675415,
      "step": 8798
    },
    {
      "epoch": 3.356170654296875e-05,
      "step": 8798,
      "training_step_time": 10.23497200012207
    },
    {
      "epoch": 3.356552124023437e-05,
      "model_forward_time": 3.143216609954834,
      "step": 8799
    },
    {
      "epoch": 3.356552124023437e-05,
      "step": 8799,
      "training_step_time": 10.242966413497925
    },
    {
      "epoch": 3.356552124023437e-05,
      "model_forward_time": 3.1470043659210205,
      "step": 8799
    },
    {
      "epoch": 3.356552124023437e-05,
      "step": 8799,
      "training_step_time": 9.923952579498291
    },
    {
      "epoch": 3.356552124023437e-05,
      "model_forward_time": 3.1469972133636475,
      "step": 8799
    },
    {
      "epoch": 3.356552124023437e-05,
      "step": 8799,
      "training_step_time": 9.918358087539673
    },
    {
      "epoch": 3.356552124023437e-05,
      "model_forward_time": 2.440300464630127,
      "step": 8799
    },
    {
      "epoch": 3.356552124023437e-05,
      "step": 8799,
      "training_step_time": 10.284067869186401
    },
    {
      "epoch": 3.35693359375e-05,
      "grad_norm": 0.2932729423046112,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 8800
    },
    {
      "epoch": 3.35693359375e-05,
      "model_forward_time": 3.15936279296875,
      "step": 8800
    },
    {
      "dynamics_loss_avg": 0.04953831266611815,
      "epoch": 3.35693359375e-05,
      "step": 8800
    },
    {
      "action_loss_avg": 0.016657402040436864,
      "epoch": 3.35693359375e-05,
      "step": 8800
    },
    {
      "epoch": 3.35693359375e-05,
      "step": 8800,
      "training_step_time": 9.932713031768799
    },
    {
      "epoch": 3.35693359375e-05,
      "model_forward_time": 3.1475346088409424,
      "step": 8800
    },
    {
      "epoch": 3.35693359375e-05,
      "step": 8800,
      "training_step_time": 9.924954175949097
    },
    {
      "epoch": 3.35693359375e-05,
      "model_forward_time": 3.43843412399292,
      "step": 8800
    },
    {
      "epoch": 3.35693359375e-05,
      "step": 8800,
      "training_step_time": 10.214666366577148
    },
    {
      "epoch": 3.35693359375e-05,
      "model_forward_time": 3.151499032974243,
      "step": 8800
    },
    {
      "epoch": 3.35693359375e-05,
      "step": 8800,
      "training_step_time": 10.030369997024536
    },
    {
      "epoch": 3.3573150634765626e-05,
      "model_forward_time": 2.4602513313293457,
      "step": 8801
    },
    {
      "epoch": 3.3573150634765626e-05,
      "step": 8801,
      "training_step_time": 7.496988773345947
    },
    {
      "epoch": 3.3573150634765626e-05,
      "model_forward_time": 3.1449172496795654,
      "step": 8801
    },
    {
      "epoch": 3.3573150634765626e-05,
      "step": 8801,
      "training_step_time": 9.92072582244873
    },
    {
      "epoch": 3.3573150634765626e-05,
      "model_forward_time": 3.143311023712158,
      "step": 8801
    },
    {
      "epoch": 3.3573150634765626e-05,
      "step": 8801,
      "training_step_time": 9.91688346862793
    },
    {
      "epoch": 3.3573150634765626e-05,
      "model_forward_time": 2.715060234069824,
      "step": 8801
    },
    {
      "epoch": 3.3573150634765626e-05,
      "step": 8801,
      "training_step_time": 10.333115339279175
    },
    {
      "epoch": 3.357696533203125e-05,
      "model_forward_time": 2.459254264831543,
      "step": 8802
    },
    {
      "epoch": 3.357696533203125e-05,
      "step": 8802,
      "training_step_time": 7.498513460159302
    },
    {
      "epoch": 3.357696533203125e-05,
      "model_forward_time": 2.4362566471099854,
      "step": 8802
    },
    {
      "epoch": 3.357696533203125e-05,
      "step": 8802,
      "training_step_time": 7.46725869178772
    },
    {
      "epoch": 3.357696533203125e-05,
      "model_forward_time": 3.1398165225982666,
      "step": 8802
    },
    {
      "dynamics_loss_avg": 0.06841301023960114,
      "epoch": 3.357696533203125e-05,
      "step": 8802
    },
    {
      "action_loss_avg": 0.018773125484585762,
      "epoch": 3.357696533203125e-05,
      "step": 8802
    },
    {
      "epoch": 3.357696533203125e-05,
      "step": 8802,
      "training_step_time": 9.917182683944702
    },
    {
      "epoch": 3.357696533203125e-05,
      "model_forward_time": 2.4410369396209717,
      "step": 8802
    },
    {
      "epoch": 3.357696533203125e-05,
      "step": 8802,
      "training_step_time": 10.315847873687744
    },
    {
      "epoch": 3.358078002929688e-05,
      "model_forward_time": 3.485846757888794,
      "step": 8803
    },
    {
      "epoch": 3.358078002929688e-05,
      "step": 8803,
      "training_step_time": 10.271171569824219
    },
    {
      "epoch": 3.358078002929688e-05,
      "model_forward_time": 3.137769937515259,
      "step": 8803
    },
    {
      "epoch": 3.358078002929688e-05,
      "step": 8803,
      "training_step_time": 9.922545671463013
    },
    {
      "epoch": 3.358078002929688e-05,
      "model_forward_time": 3.1437721252441406,
      "step": 8803
    },
    {
      "epoch": 3.358078002929688e-05,
      "step": 8803,
      "training_step_time": 9.92460322380066
    },
    {
      "epoch": 3.358078002929688e-05,
      "model_forward_time": 3.1452083587646484,
      "step": 8803
    },
    {
      "epoch": 3.358078002929688e-05,
      "step": 8803,
      "training_step_time": 10.272644996643066
    },
    {
      "epoch": 3.35845947265625e-05,
      "model_forward_time": 3.1546735763549805,
      "step": 8804
    },
    {
      "epoch": 3.35845947265625e-05,
      "step": 8804,
      "training_step_time": 9.935588598251343
    },
    {
      "epoch": 3.35845947265625e-05,
      "model_forward_time": 3.449368953704834,
      "step": 8804
    },
    {
      "epoch": 3.35845947265625e-05,
      "step": 8804,
      "training_step_time": 10.231494903564453
    },
    {
      "epoch": 3.35845947265625e-05,
      "model_forward_time": 3.1405320167541504,
      "step": 8804
    },
    {
      "epoch": 3.35845947265625e-05,
      "step": 8804,
      "training_step_time": 9.924052476882935
    },
    {
      "epoch": 3.35845947265625e-05,
      "model_forward_time": 3.1446585655212402,
      "step": 8804
    },
    {
      "epoch": 3.35845947265625e-05,
      "step": 8804,
      "training_step_time": 10.23914909362793
    },
    {
      "epoch": 3.3588409423828124e-05,
      "model_forward_time": 2.446638822555542,
      "step": 8805
    },
    {
      "dynamics_loss_avg": 0.06282130442559719,
      "epoch": 3.3588409423828124e-05,
      "step": 8805
    },
    {
      "action_loss_avg": 0.019989883992820978,
      "epoch": 3.3588409423828124e-05,
      "step": 8805
    },
    {
      "epoch": 3.3588409423828124e-05,
      "step": 8805,
      "training_step_time": 7.493901968002319
    },
    {
      "epoch": 3.3588409423828124e-05,
      "model_forward_time": 3.132748603820801,
      "step": 8805
    },
    {
      "epoch": 3.3588409423828124e-05,
      "step": 8805,
      "training_step_time": 9.904280662536621
    },
    {
      "epoch": 3.3588409423828124e-05,
      "model_forward_time": 3.4273488521575928,
      "step": 8805
    },
    {
      "epoch": 3.3588409423828124e-05,
      "step": 8805,
      "training_step_time": 10.201431512832642
    },
    {
      "epoch": 3.3588409423828124e-05,
      "model_forward_time": 3.1434104442596436,
      "step": 8805
    },
    {
      "epoch": 3.3588409423828124e-05,
      "step": 8805,
      "training_step_time": 10.024359703063965
    },
    {
      "epoch": 3.359222412109375e-05,
      "model_forward_time": 3.1528921127319336,
      "step": 8806
    },
    {
      "epoch": 3.359222412109375e-05,
      "step": 8806,
      "training_step_time": 9.929571151733398
    },
    {
      "epoch": 3.359222412109375e-05,
      "model_forward_time": 3.1365673542022705,
      "step": 8806
    },
    {
      "epoch": 3.359222412109375e-05,
      "step": 8806,
      "training_step_time": 9.914428234100342
    },
    {
      "epoch": 3.359222412109375e-05,
      "model_forward_time": 3.142657518386841,
      "step": 8806
    },
    {
      "epoch": 3.359222412109375e-05,
      "step": 8806,
      "training_step_time": 9.917022466659546
    },
    {
      "epoch": 3.359222412109375e-05,
      "model_forward_time": 3.4575612545013428,
      "step": 8806
    },
    {
      "epoch": 3.359222412109375e-05,
      "step": 8806,
      "training_step_time": 10.325232028961182
    },
    {
      "epoch": 3.3596038818359376e-05,
      "model_forward_time": 3.1495471000671387,
      "step": 8807
    },
    {
      "epoch": 3.3596038818359376e-05,
      "step": 8807,
      "training_step_time": 9.93807578086853
    },
    {
      "epoch": 3.3596038818359376e-05,
      "model_forward_time": 3.1414496898651123,
      "step": 8807
    },
    {
      "epoch": 3.3596038818359376e-05,
      "step": 8807,
      "training_step_time": 9.925366401672363
    },
    {
      "epoch": 3.3596038818359376e-05,
      "model_forward_time": 3.1453025341033936,
      "step": 8807
    },
    {
      "dynamics_loss_avg": 0.06051970608532429,
      "epoch": 3.3596038818359376e-05,
      "step": 8807
    },
    {
      "action_loss_avg": 0.021526909014210106,
      "epoch": 3.3596038818359376e-05,
      "step": 8807
    },
    {
      "epoch": 3.3596038818359376e-05,
      "step": 8807,
      "training_step_time": 9.928648471832275
    },
    {
      "epoch": 3.3596038818359376e-05,
      "model_forward_time": 3.14286470413208,
      "step": 8807
    },
    {
      "epoch": 3.3596038818359376e-05,
      "step": 8807,
      "training_step_time": 10.347128629684448
    },
    {
      "epoch": 3.3599853515625e-05,
      "model_forward_time": 2.450944662094116,
      "step": 8808
    },
    {
      "epoch": 3.3599853515625e-05,
      "step": 8808,
      "training_step_time": 7.492898941040039
    },
    {
      "epoch": 3.3599853515625e-05,
      "model_forward_time": 3.142782211303711,
      "step": 8808
    },
    {
      "epoch": 3.3599853515625e-05,
      "step": 8808,
      "training_step_time": 9.928877592086792
    },
    {
      "epoch": 3.3599853515625e-05,
      "model_forward_time": 3.1428258419036865,
      "step": 8808
    },
    {
      "epoch": 3.3599853515625e-05,
      "step": 8808,
      "training_step_time": 9.932522058486938
    },
    {
      "epoch": 3.3599853515625e-05,
      "model_forward_time": 3.146815776824951,
      "step": 8808
    },
    {
      "epoch": 3.3599853515625e-05,
      "step": 8808,
      "training_step_time": 10.273611545562744
    },
    {
      "epoch": 3.360366821289062e-05,
      "model_forward_time": 3.16166615486145,
      "step": 8809
    },
    {
      "epoch": 3.360366821289062e-05,
      "step": 8809,
      "training_step_time": 9.949668169021606
    },
    {
      "epoch": 3.360366821289062e-05,
      "model_forward_time": 3.435516595840454,
      "step": 8809
    },
    {
      "epoch": 3.360366821289062e-05,
      "step": 8809,
      "training_step_time": 10.229294776916504
    },
    {
      "epoch": 3.360366821289062e-05,
      "model_forward_time": 3.1515040397644043,
      "step": 8809
    },
    {
      "epoch": 3.360366821289062e-05,
      "step": 8809,
      "training_step_time": 9.936340093612671
    },
    {
      "epoch": 3.360366821289062e-05,
      "model_forward_time": 3.1413705348968506,
      "step": 8809
    },
    {
      "epoch": 3.360366821289062e-05,
      "step": 8809,
      "training_step_time": 10.316761493682861
    },
    {
      "epoch": 3.360748291015625e-05,
      "grad_norm": 0.33041101694107056,
      "learning_rate": 1e-05,
      "loss": 0.0799,
      "step": 8810
    },
    {
      "epoch": 3.360748291015625e-05,
      "model_forward_time": 3.172236919403076,
      "step": 8810
    },
    {
      "dynamics_loss_avg": 0.053157460130751136,
      "epoch": 3.360748291015625e-05,
      "step": 8810
    },
    {
      "action_loss_avg": 0.0187063992023468,
      "epoch": 3.360748291015625e-05,
      "step": 8810
    },
    {
      "epoch": 3.360748291015625e-05,
      "step": 8810,
      "training_step_time": 9.971109390258789
    },
    {
      "epoch": 3.360748291015625e-05,
      "model_forward_time": 2.4395620822906494,
      "step": 8810
    },
    {
      "epoch": 3.360748291015625e-05,
      "step": 8810,
      "training_step_time": 7.4745800495147705
    },
    {
      "epoch": 3.360748291015625e-05,
      "model_forward_time": 3.4547476768493652,
      "step": 8810
    },
    {
      "epoch": 3.360748291015625e-05,
      "step": 8810,
      "training_step_time": 10.240998268127441
    },
    {
      "epoch": 3.360748291015625e-05,
      "model_forward_time": 3.1399855613708496,
      "step": 8810
    },
    {
      "epoch": 3.360748291015625e-05,
      "step": 8810,
      "training_step_time": 10.287186861038208
    },
    {
      "epoch": 3.3611297607421875e-05,
      "model_forward_time": 3.153719663619995,
      "step": 8811
    },
    {
      "epoch": 3.3611297607421875e-05,
      "step": 8811,
      "training_step_time": 9.938949346542358
    },
    {
      "epoch": 3.3611297607421875e-05,
      "model_forward_time": 3.138627767562866,
      "step": 8811
    },
    {
      "epoch": 3.3611297607421875e-05,
      "step": 8811,
      "training_step_time": 9.917733430862427
    },
    {
      "epoch": 3.3611297607421875e-05,
      "model_forward_time": 3.4509437084198,
      "step": 8811
    },
    {
      "epoch": 3.3611297607421875e-05,
      "step": 8811,
      "training_step_time": 10.23460578918457
    },
    {
      "epoch": 3.3611297607421875e-05,
      "model_forward_time": 3.135709524154663,
      "step": 8811
    },
    {
      "epoch": 3.3611297607421875e-05,
      "step": 8811,
      "training_step_time": 10.291884422302246
    },
    {
      "epoch": 3.36151123046875e-05,
      "model_forward_time": 3.158489465713501,
      "step": 8812
    },
    {
      "epoch": 3.36151123046875e-05,
      "step": 8812,
      "training_step_time": 9.942426204681396
    },
    {
      "epoch": 3.36151123046875e-05,
      "model_forward_time": 3.1350808143615723,
      "step": 8812
    },
    {
      "epoch": 3.36151123046875e-05,
      "step": 8812,
      "training_step_time": 9.91428542137146
    },
    {
      "epoch": 3.36151123046875e-05,
      "model_forward_time": 3.132654905319214,
      "step": 8812
    },
    {
      "dynamics_loss_avg": 0.06899640373885632,
      "epoch": 3.36151123046875e-05,
      "step": 8812
    },
    {
      "action_loss_avg": 0.028687713388353586,
      "epoch": 3.36151123046875e-05,
      "step": 8812
    },
    {
      "epoch": 3.36151123046875e-05,
      "step": 8812,
      "training_step_time": 9.913776636123657
    },
    {
      "epoch": 3.36151123046875e-05,
      "model_forward_time": 3.137112617492676,
      "step": 8812
    },
    {
      "epoch": 3.36151123046875e-05,
      "step": 8812,
      "training_step_time": 10.319670915603638
    },
    {
      "epoch": 3.361892700195313e-05,
      "model_forward_time": 3.1549642086029053,
      "step": 8813
    },
    {
      "epoch": 3.361892700195313e-05,
      "step": 8813,
      "training_step_time": 9.935923337936401
    },
    {
      "epoch": 3.361892700195313e-05,
      "model_forward_time": 3.136735200881958,
      "step": 8813
    },
    {
      "epoch": 3.361892700195313e-05,
      "step": 8813,
      "training_step_time": 9.912699222564697
    },
    {
      "epoch": 3.361892700195313e-05,
      "model_forward_time": 3.134479522705078,
      "step": 8813
    },
    {
      "epoch": 3.361892700195313e-05,
      "step": 8813,
      "training_step_time": 9.91471266746521
    },
    {
      "epoch": 3.361892700195313e-05,
      "model_forward_time": 3.141723155975342,
      "step": 8813
    },
    {
      "epoch": 3.361892700195313e-05,
      "step": 8813,
      "training_step_time": 10.313315153121948
    },
    {
      "epoch": 3.362274169921875e-05,
      "model_forward_time": 2.7452197074890137,
      "step": 8814
    },
    {
      "epoch": 3.362274169921875e-05,
      "step": 8814,
      "training_step_time": 7.779338359832764
    },
    {
      "epoch": 3.362274169921875e-05,
      "model_forward_time": 3.1376636028289795,
      "step": 8814
    },
    {
      "epoch": 3.362274169921875e-05,
      "step": 8814,
      "training_step_time": 9.917633533477783
    },
    {
      "epoch": 3.362274169921875e-05,
      "model_forward_time": 3.1366989612579346,
      "step": 8814
    },
    {
      "epoch": 3.362274169921875e-05,
      "step": 8814,
      "training_step_time": 9.922207117080688
    },
    {
      "epoch": 3.362274169921875e-05,
      "model_forward_time": 3.142622470855713,
      "step": 8814
    },
    {
      "epoch": 3.362274169921875e-05,
      "step": 8814,
      "training_step_time": 10.260604619979858
    },
    {
      "epoch": 3.362655639648437e-05,
      "model_forward_time": 3.1568329334259033,
      "step": 8815
    },
    {
      "dynamics_loss_avg": 0.06221569217741489,
      "epoch": 3.362655639648437e-05,
      "step": 8815
    },
    {
      "action_loss_avg": 0.021828777715563775,
      "epoch": 3.362655639648437e-05,
      "step": 8815
    },
    {
      "epoch": 3.362655639648437e-05,
      "step": 8815,
      "training_step_time": 9.94132375717163
    },
    {
      "epoch": 3.362655639648437e-05,
      "model_forward_time": 2.432565212249756,
      "step": 8815
    },
    {
      "epoch": 3.362655639648437e-05,
      "step": 8815,
      "training_step_time": 7.4707114696502686
    },
    {
      "epoch": 3.362655639648437e-05,
      "model_forward_time": 3.394101858139038,
      "step": 8815
    },
    {
      "epoch": 3.362655639648437e-05,
      "step": 8815,
      "training_step_time": 10.171736478805542
    },
    {
      "epoch": 3.362655639648437e-05,
      "model_forward_time": 3.139500141143799,
      "step": 8815
    },
    {
      "epoch": 3.362655639648437e-05,
      "step": 8815,
      "training_step_time": 10.236934423446655
    },
    {
      "epoch": 3.363037109375e-05,
      "model_forward_time": 3.1533210277557373,
      "step": 8816
    },
    {
      "epoch": 3.363037109375e-05,
      "step": 8816,
      "training_step_time": 9.929320335388184
    },
    {
      "epoch": 3.363037109375e-05,
      "model_forward_time": 3.1323788166046143,
      "step": 8816
    },
    {
      "epoch": 3.363037109375e-05,
      "step": 8816,
      "training_step_time": 9.915282011032104
    },
    {
      "epoch": 3.363037109375e-05,
      "model_forward_time": 3.430753469467163,
      "step": 8816
    },
    {
      "epoch": 3.363037109375e-05,
      "step": 8816,
      "training_step_time": 10.20842719078064
    },
    {
      "epoch": 3.363037109375e-05,
      "model_forward_time": 3.1375300884246826,
      "step": 8816
    },
    {
      "epoch": 3.363037109375e-05,
      "step": 8816,
      "training_step_time": 10.275899887084961
    },
    {
      "epoch": 3.3634185791015626e-05,
      "model_forward_time": 3.157383441925049,
      "step": 8817
    },
    {
      "epoch": 3.3634185791015626e-05,
      "step": 8817,
      "training_step_time": 9.939401865005493
    },
    {
      "epoch": 3.3634185791015626e-05,
      "model_forward_time": 2.4387102127075195,
      "step": 8817
    },
    {
      "epoch": 3.3634185791015626e-05,
      "step": 8817,
      "training_step_time": 7.473740100860596
    },
    {
      "epoch": 3.3634185791015626e-05,
      "model_forward_time": 3.139225721359253,
      "step": 8817
    },
    {
      "dynamics_loss_avg": 0.048913934454321864,
      "epoch": 3.3634185791015626e-05,
      "step": 8817
    },
    {
      "action_loss_avg": 0.01769488546997309,
      "epoch": 3.3634185791015626e-05,
      "step": 8817
    },
    {
      "epoch": 3.3634185791015626e-05,
      "step": 8817,
      "training_step_time": 9.917406797409058
    },
    {
      "epoch": 3.3634185791015626e-05,
      "model_forward_time": 3.1352102756500244,
      "step": 8817
    },
    {
      "epoch": 3.3634185791015626e-05,
      "step": 8817,
      "training_step_time": 10.288342475891113
    },
    {
      "epoch": 3.363800048828125e-05,
      "model_forward_time": 3.4756643772125244,
      "step": 8818
    },
    {
      "epoch": 3.363800048828125e-05,
      "step": 8818,
      "training_step_time": 10.252439498901367
    },
    {
      "epoch": 3.363800048828125e-05,
      "model_forward_time": 3.1361327171325684,
      "step": 8818
    },
    {
      "epoch": 3.363800048828125e-05,
      "step": 8818,
      "training_step_time": 9.914138793945312
    },
    {
      "epoch": 3.363800048828125e-05,
      "model_forward_time": 2.4350039958953857,
      "step": 8818
    },
    {
      "epoch": 3.363800048828125e-05,
      "step": 8818,
      "training_step_time": 7.465457439422607
    },
    {
      "epoch": 3.363800048828125e-05,
      "model_forward_time": 3.128948450088501,
      "step": 8818
    },
    {
      "epoch": 3.363800048828125e-05,
      "step": 8818,
      "training_step_time": 9.998368978500366
    },
    {
      "epoch": 3.364181518554688e-05,
      "model_forward_time": 2.448599100112915,
      "step": 8819
    },
    {
      "epoch": 3.364181518554688e-05,
      "step": 8819,
      "training_step_time": 7.4773783683776855
    },
    {
      "epoch": 3.364181518554688e-05,
      "model_forward_time": 3.3877689838409424,
      "step": 8819
    },
    {
      "epoch": 3.364181518554688e-05,
      "step": 8819,
      "training_step_time": 10.161916732788086
    },
    {
      "epoch": 3.364181518554688e-05,
      "model_forward_time": 3.1396965980529785,
      "step": 8819
    },
    {
      "epoch": 3.364181518554688e-05,
      "step": 8819,
      "training_step_time": 9.920347690582275
    },
    {
      "epoch": 3.364181518554688e-05,
      "model_forward_time": 3.1392643451690674,
      "step": 8819
    },
    {
      "epoch": 3.364181518554688e-05,
      "step": 8819,
      "training_step_time": 10.27592396736145
    },
    {
      "epoch": 3.36456298828125e-05,
      "grad_norm": 0.31909608840942383,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 8820
    },
    {
      "epoch": 3.36456298828125e-05,
      "model_forward_time": 2.451396942138672,
      "step": 8820
    },
    {
      "dynamics_loss_avg": 0.04447497203946114,
      "epoch": 3.36456298828125e-05,
      "step": 8820
    },
    {
      "action_loss_avg": 0.012732032965868712,
      "epoch": 3.36456298828125e-05,
      "step": 8820
    },
    {
      "epoch": 3.36456298828125e-05,
      "step": 8820,
      "training_step_time": 7.489238739013672
    },
    {
      "epoch": 3.36456298828125e-05,
      "model_forward_time": 2.433440923690796,
      "step": 8820
    },
    {
      "epoch": 3.36456298828125e-05,
      "step": 8820,
      "training_step_time": 7.461935520172119
    },
    {
      "epoch": 3.36456298828125e-05,
      "model_forward_time": 3.4341177940368652,
      "step": 8820
    },
    {
      "epoch": 3.36456298828125e-05,
      "step": 8820,
      "training_step_time": 10.208168983459473
    },
    {
      "epoch": 3.36456298828125e-05,
      "model_forward_time": 3.1351797580718994,
      "step": 8820
    },
    {
      "epoch": 3.36456298828125e-05,
      "step": 8820,
      "training_step_time": 10.29240369796753
    },
    {
      "epoch": 3.3649444580078124e-05,
      "model_forward_time": 3.143549680709839,
      "step": 8821
    },
    {
      "epoch": 3.3649444580078124e-05,
      "step": 8821,
      "training_step_time": 9.921394348144531
    },
    {
      "epoch": 3.3649444580078124e-05,
      "model_forward_time": 2.4448165893554688,
      "step": 8821
    },
    {
      "epoch": 3.3649444580078124e-05,
      "step": 8821,
      "training_step_time": 7.471746206283569
    },
    {
      "epoch": 3.3649444580078124e-05,
      "model_forward_time": 2.433659791946411,
      "step": 8821
    },
    {
      "epoch": 3.3649444580078124e-05,
      "step": 8821,
      "training_step_time": 7.456193208694458
    },
    {
      "epoch": 3.3649444580078124e-05,
      "model_forward_time": 3.4136133193969727,
      "step": 8821
    },
    {
      "epoch": 3.3649444580078124e-05,
      "step": 8821,
      "training_step_time": 10.292888641357422
    },
    {
      "epoch": 3.365325927734375e-05,
      "model_forward_time": 3.1709518432617188,
      "step": 8822
    },
    {
      "epoch": 3.365325927734375e-05,
      "step": 8822,
      "training_step_time": 9.95885944366455
    },
    {
      "epoch": 3.365325927734375e-05,
      "model_forward_time": 3.1581857204437256,
      "step": 8822
    },
    {
      "epoch": 3.365325927734375e-05,
      "step": 8822,
      "training_step_time": 9.9441659450531
    },
    {
      "epoch": 3.365325927734375e-05,
      "model_forward_time": 3.1509976387023926,
      "step": 8822
    },
    {
      "dynamics_loss_avg": 0.049198761908337475,
      "epoch": 3.365325927734375e-05,
      "step": 8822
    },
    {
      "action_loss_avg": 0.016474423371255396,
      "epoch": 3.365325927734375e-05,
      "step": 8822
    },
    {
      "epoch": 3.365325927734375e-05,
      "step": 8822,
      "training_step_time": 9.938946723937988
    },
    {
      "epoch": 3.365325927734375e-05,
      "model_forward_time": 3.1669113636016846,
      "step": 8822
    },
    {
      "epoch": 3.365325927734375e-05,
      "step": 8822,
      "training_step_time": 10.3652663230896
    },
    {
      "epoch": 3.3657073974609377e-05,
      "model_forward_time": 2.7590699195861816,
      "step": 8823
    },
    {
      "epoch": 3.3657073974609377e-05,
      "step": 8823,
      "training_step_time": 7.78408408164978
    },
    {
      "epoch": 3.3657073974609377e-05,
      "model_forward_time": 3.145610809326172,
      "step": 8823
    },
    {
      "epoch": 3.3657073974609377e-05,
      "step": 8823,
      "training_step_time": 9.918430805206299
    },
    {
      "epoch": 3.3657073974609377e-05,
      "model_forward_time": 3.137751817703247,
      "step": 8823
    },
    {
      "epoch": 3.3657073974609377e-05,
      "step": 8823,
      "training_step_time": 9.907636880874634
    },
    {
      "epoch": 3.3657073974609377e-05,
      "model_forward_time": 3.140263557434082,
      "step": 8823
    },
    {
      "epoch": 3.3657073974609377e-05,
      "step": 8823,
      "training_step_time": 10.026695728302002
    },
    {
      "epoch": 3.3660888671875e-05,
      "model_forward_time": 3.1588573455810547,
      "step": 8824
    },
    {
      "epoch": 3.3660888671875e-05,
      "step": 8824,
      "training_step_time": 9.935431480407715
    },
    {
      "epoch": 3.3660888671875e-05,
      "model_forward_time": 2.7651829719543457,
      "step": 8824
    },
    {
      "epoch": 3.3660888671875e-05,
      "step": 8824,
      "training_step_time": 7.795741558074951
    },
    {
      "epoch": 3.3660888671875e-05,
      "model_forward_time": 3.139209032058716,
      "step": 8824
    },
    {
      "epoch": 3.3660888671875e-05,
      "step": 8824,
      "training_step_time": 9.908410549163818
    },
    {
      "epoch": 3.3660888671875e-05,
      "model_forward_time": 3.1450414657592773,
      "step": 8824
    },
    {
      "epoch": 3.3660888671875e-05,
      "step": 8824,
      "training_step_time": 10.289541244506836
    },
    {
      "epoch": 3.366470336914062e-05,
      "model_forward_time": 3.15604829788208,
      "step": 8825
    },
    {
      "dynamics_loss_avg": 0.060249831154942514,
      "epoch": 3.366470336914062e-05,
      "step": 8825
    },
    {
      "action_loss_avg": 0.01840503406710923,
      "epoch": 3.366470336914062e-05,
      "step": 8825
    },
    {
      "epoch": 3.366470336914062e-05,
      "step": 8825,
      "training_step_time": 9.936205387115479
    },
    {
      "epoch": 3.366470336914062e-05,
      "model_forward_time": 3.144552230834961,
      "step": 8825
    },
    {
      "epoch": 3.366470336914062e-05,
      "step": 8825,
      "training_step_time": 9.921251058578491
    },
    {
      "epoch": 3.366470336914062e-05,
      "model_forward_time": 3.3958446979522705,
      "step": 8825
    },
    {
      "epoch": 3.366470336914062e-05,
      "step": 8825,
      "training_step_time": 10.17085599899292
    },
    {
      "epoch": 3.366470336914062e-05,
      "model_forward_time": 3.1386218070983887,
      "step": 8825
    },
    {
      "epoch": 3.366470336914062e-05,
      "step": 8825,
      "training_step_time": 10.298827171325684
    },
    {
      "epoch": 3.366851806640625e-05,
      "model_forward_time": 2.4578354358673096,
      "step": 8826
    },
    {
      "epoch": 3.366851806640625e-05,
      "step": 8826,
      "training_step_time": 7.497913122177124
    },
    {
      "epoch": 3.366851806640625e-05,
      "model_forward_time": 3.1324820518493652,
      "step": 8826
    },
    {
      "epoch": 3.366851806640625e-05,
      "step": 8826,
      "training_step_time": 9.909825563430786
    },
    {
      "epoch": 3.366851806640625e-05,
      "model_forward_time": 3.1360716819763184,
      "step": 8826
    },
    {
      "epoch": 3.366851806640625e-05,
      "step": 8826,
      "training_step_time": 9.910409927368164
    },
    {
      "epoch": 3.366851806640625e-05,
      "model_forward_time": 3.4595160484313965,
      "step": 8826
    },
    {
      "epoch": 3.366851806640625e-05,
      "step": 8826,
      "training_step_time": 10.341669082641602
    },
    {
      "epoch": 3.3672332763671875e-05,
      "model_forward_time": 3.1500892639160156,
      "step": 8827
    },
    {
      "epoch": 3.3672332763671875e-05,
      "step": 8827,
      "training_step_time": 9.926048040390015
    },
    {
      "epoch": 3.3672332763671875e-05,
      "model_forward_time": 3.133667469024658,
      "step": 8827
    },
    {
      "epoch": 3.3672332763671875e-05,
      "step": 8827,
      "training_step_time": 9.910657167434692
    },
    {
      "epoch": 3.3672332763671875e-05,
      "model_forward_time": 3.136913299560547,
      "step": 8827
    },
    {
      "dynamics_loss_avg": 0.065925077162683,
      "epoch": 3.3672332763671875e-05,
      "step": 8827
    },
    {
      "action_loss_avg": 0.016679284768179058,
      "epoch": 3.3672332763671875e-05,
      "step": 8827
    },
    {
      "epoch": 3.3672332763671875e-05,
      "step": 8827,
      "training_step_time": 9.917181730270386
    },
    {
      "epoch": 3.3672332763671875e-05,
      "model_forward_time": 3.136277198791504,
      "step": 8827
    },
    {
      "epoch": 3.3672332763671875e-05,
      "step": 8827,
      "training_step_time": 10.287655591964722
    },
    {
      "epoch": 3.36761474609375e-05,
      "model_forward_time": 3.1551859378814697,
      "step": 8828
    },
    {
      "epoch": 3.36761474609375e-05,
      "step": 8828,
      "training_step_time": 9.935425758361816
    },
    {
      "epoch": 3.36761474609375e-05,
      "model_forward_time": 3.1456708908081055,
      "step": 8828
    },
    {
      "epoch": 3.36761474609375e-05,
      "step": 8828,
      "training_step_time": 9.925438404083252
    },
    {
      "epoch": 3.36761474609375e-05,
      "model_forward_time": 3.140320062637329,
      "step": 8828
    },
    {
      "epoch": 3.36761474609375e-05,
      "step": 8828,
      "training_step_time": 9.915411472320557
    },
    {
      "epoch": 3.36761474609375e-05,
      "model_forward_time": 3.1363556385040283,
      "step": 8828
    },
    {
      "epoch": 3.36761474609375e-05,
      "step": 8828,
      "training_step_time": 10.01507019996643
    },
    {
      "epoch": 3.367996215820313e-05,
      "model_forward_time": 3.4620659351348877,
      "step": 8829
    },
    {
      "epoch": 3.367996215820313e-05,
      "step": 8829,
      "training_step_time": 10.245664119720459
    },
    {
      "epoch": 3.367996215820313e-05,
      "model_forward_time": 3.1367151737213135,
      "step": 8829
    },
    {
      "epoch": 3.367996215820313e-05,
      "step": 8829,
      "training_step_time": 9.91567325592041
    },
    {
      "epoch": 3.367996215820313e-05,
      "model_forward_time": 3.1365509033203125,
      "step": 8829
    },
    {
      "epoch": 3.367996215820313e-05,
      "step": 8829,
      "training_step_time": 9.914430856704712
    },
    {
      "epoch": 3.367996215820313e-05,
      "model_forward_time": 3.136146068572998,
      "step": 8829
    },
    {
      "epoch": 3.367996215820313e-05,
      "step": 8829,
      "training_step_time": 10.2519371509552
    },
    {
      "epoch": 3.368377685546875e-05,
      "grad_norm": 0.46296417713165283,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 8830
    },
    {
      "epoch": 3.368377685546875e-05,
      "model_forward_time": 3.1536850929260254,
      "step": 8830
    },
    {
      "dynamics_loss_avg": 0.06142371706664562,
      "epoch": 3.368377685546875e-05,
      "step": 8830
    },
    {
      "action_loss_avg": 0.022162861190736294,
      "epoch": 3.368377685546875e-05,
      "step": 8830
    },
    {
      "epoch": 3.368377685546875e-05,
      "step": 8830,
      "training_step_time": 9.929438352584839
    },
    {
      "epoch": 3.368377685546875e-05,
      "model_forward_time": 2.4280755519866943,
      "step": 8830
    },
    {
      "epoch": 3.368377685546875e-05,
      "step": 8830,
      "training_step_time": 7.731829881668091
    },
    {
      "epoch": 3.368377685546875e-05,
      "model_forward_time": 3.1310811042785645,
      "step": 8830
    },
    {
      "epoch": 3.368377685546875e-05,
      "step": 8830,
      "training_step_time": 9.901306629180908
    },
    {
      "epoch": 3.368377685546875e-05,
      "model_forward_time": 3.129849433898926,
      "step": 8830
    },
    {
      "epoch": 3.368377685546875e-05,
      "step": 8830,
      "training_step_time": 10.233064413070679
    },
    {
      "epoch": 3.368759155273437e-05,
      "model_forward_time": 3.141559362411499,
      "step": 8831
    },
    {
      "epoch": 3.368759155273437e-05,
      "step": 8831,
      "training_step_time": 9.927196025848389
    },
    {
      "epoch": 3.368759155273437e-05,
      "model_forward_time": 3.1364383697509766,
      "step": 8831
    },
    {
      "epoch": 3.368759155273437e-05,
      "step": 8831,
      "training_step_time": 9.908490657806396
    },
    {
      "epoch": 3.368759155273437e-05,
      "model_forward_time": 2.695655345916748,
      "step": 8831
    },
    {
      "epoch": 3.368759155273437e-05,
      "step": 8831,
      "training_step_time": 7.7189342975616455
    },
    {
      "epoch": 3.368759155273437e-05,
      "model_forward_time": 3.1317737102508545,
      "step": 8831
    },
    {
      "epoch": 3.368759155273437e-05,
      "step": 8831,
      "training_step_time": 10.01521921157837
    },
    {
      "epoch": 3.369140625e-05,
      "model_forward_time": 3.152245283126831,
      "step": 8832
    },
    {
      "epoch": 3.369140625e-05,
      "step": 8832,
      "training_step_time": 9.925826072692871
    },
    {
      "epoch": 3.369140625e-05,
      "model_forward_time": 3.135800361633301,
      "step": 8832
    },
    {
      "epoch": 3.369140625e-05,
      "step": 8832,
      "training_step_time": 9.910297155380249
    },
    {
      "epoch": 3.369140625e-05,
      "model_forward_time": 3.1358180046081543,
      "step": 8832
    },
    {
      "dynamics_loss_avg": 0.04735180158168077,
      "epoch": 3.369140625e-05,
      "step": 8832
    },
    {
      "action_loss_avg": 0.013857268379069865,
      "epoch": 3.369140625e-05,
      "step": 8832
    },
    {
      "epoch": 3.369140625e-05,
      "step": 8832,
      "training_step_time": 9.913200378417969
    },
    {
      "epoch": 3.369140625e-05,
      "model_forward_time": 3.1578369140625,
      "step": 8832
    },
    {
      "epoch": 3.369140625e-05,
      "step": 8832,
      "training_step_time": 10.344475507736206
    },
    {
      "epoch": 3.3695220947265626e-05,
      "model_forward_time": 3.1528003215789795,
      "step": 8833
    },
    {
      "epoch": 3.3695220947265626e-05,
      "step": 8833,
      "training_step_time": 9.934005737304688
    },
    {
      "epoch": 3.3695220947265626e-05,
      "model_forward_time": 3.1458561420440674,
      "step": 8833
    },
    {
      "epoch": 3.3695220947265626e-05,
      "step": 8833,
      "training_step_time": 9.928251504898071
    },
    {
      "epoch": 3.3695220947265626e-05,
      "model_forward_time": 3.138397216796875,
      "step": 8833
    },
    {
      "epoch": 3.3695220947265626e-05,
      "step": 8833,
      "training_step_time": 9.91193962097168
    },
    {
      "epoch": 3.3695220947265626e-05,
      "model_forward_time": 3.143746852874756,
      "step": 8833
    },
    {
      "epoch": 3.3695220947265626e-05,
      "step": 8833,
      "training_step_time": 10.310795068740845
    },
    {
      "epoch": 3.369903564453125e-05,
      "model_forward_time": 3.4273369312286377,
      "step": 8834
    },
    {
      "epoch": 3.369903564453125e-05,
      "step": 8834,
      "training_step_time": 10.21655535697937
    },
    {
      "epoch": 3.369903564453125e-05,
      "model_forward_time": 2.4375226497650146,
      "step": 8834
    },
    {
      "epoch": 3.369903564453125e-05,
      "step": 8834,
      "training_step_time": 7.468151569366455
    },
    {
      "epoch": 3.369903564453125e-05,
      "model_forward_time": 3.138136148452759,
      "step": 8834
    },
    {
      "epoch": 3.369903564453125e-05,
      "step": 8834,
      "training_step_time": 9.913882732391357
    },
    {
      "epoch": 3.369903564453125e-05,
      "model_forward_time": 3.1450083255767822,
      "step": 8834
    },
    {
      "epoch": 3.369903564453125e-05,
      "step": 8834,
      "training_step_time": 10.028554677963257
    },
    {
      "epoch": 3.370285034179688e-05,
      "model_forward_time": 3.152845859527588,
      "step": 8835
    },
    {
      "dynamics_loss_avg": 0.0458346570841968,
      "epoch": 3.370285034179688e-05,
      "step": 8835
    },
    {
      "action_loss_avg": 0.014999922085553407,
      "epoch": 3.370285034179688e-05,
      "step": 8835
    },
    {
      "epoch": 3.370285034179688e-05,
      "step": 8835,
      "training_step_time": 9.936914205551147
    },
    {
      "epoch": 3.370285034179688e-05,
      "model_forward_time": 3.1462249755859375,
      "step": 8835
    },
    {
      "epoch": 3.370285034179688e-05,
      "step": 8835,
      "training_step_time": 10.203273296356201
    },
    {
      "epoch": 3.370285034179688e-05,
      "model_forward_time": 3.1453826427459717,
      "step": 8835
    },
    {
      "epoch": 3.370285034179688e-05,
      "step": 8835,
      "training_step_time": 9.928226709365845
    },
    {
      "epoch": 3.370285034179688e-05,
      "model_forward_time": 3.1513803005218506,
      "step": 8835
    },
    {
      "epoch": 3.370285034179688e-05,
      "step": 8835,
      "training_step_time": 10.288302421569824
    },
    {
      "epoch": 3.37066650390625e-05,
      "model_forward_time": 3.158808946609497,
      "step": 8836
    },
    {
      "epoch": 3.37066650390625e-05,
      "step": 8836,
      "training_step_time": 9.940926313400269
    },
    {
      "epoch": 3.37066650390625e-05,
      "model_forward_time": 3.136349678039551,
      "step": 8836
    },
    {
      "epoch": 3.37066650390625e-05,
      "step": 8836,
      "training_step_time": 9.91919755935669
    },
    {
      "epoch": 3.37066650390625e-05,
      "model_forward_time": 3.4016525745391846,
      "step": 8836
    },
    {
      "epoch": 3.37066650390625e-05,
      "step": 8836,
      "training_step_time": 10.187779664993286
    },
    {
      "epoch": 3.37066650390625e-05,
      "model_forward_time": 3.1454122066497803,
      "step": 8836
    },
    {
      "epoch": 3.37066650390625e-05,
      "step": 8836,
      "training_step_time": 10.326388835906982
    },
    {
      "epoch": 3.3710479736328124e-05,
      "model_forward_time": 3.15478777885437,
      "step": 8837
    },
    {
      "epoch": 3.3710479736328124e-05,
      "step": 8837,
      "training_step_time": 9.941750288009644
    },
    {
      "epoch": 3.3710479736328124e-05,
      "model_forward_time": 3.141284465789795,
      "step": 8837
    },
    {
      "epoch": 3.3710479736328124e-05,
      "step": 8837,
      "training_step_time": 9.933314085006714
    },
    {
      "epoch": 3.3710479736328124e-05,
      "model_forward_time": 3.1481211185455322,
      "step": 8837
    },
    {
      "dynamics_loss_avg": 0.055050524324178694,
      "epoch": 3.3710479736328124e-05,
      "step": 8837
    },
    {
      "action_loss_avg": 0.01942146704532206,
      "epoch": 3.3710479736328124e-05,
      "step": 8837
    },
    {
      "epoch": 3.3710479736328124e-05,
      "step": 8837,
      "training_step_time": 9.928314924240112
    },
    {
      "epoch": 3.3710479736328124e-05,
      "model_forward_time": 3.468594551086426,
      "step": 8837
    },
    {
      "epoch": 3.3710479736328124e-05,
      "step": 8837,
      "training_step_time": 10.334791660308838
    },
    {
      "epoch": 3.371429443359375e-05,
      "model_forward_time": 3.1514992713928223,
      "step": 8838
    },
    {
      "epoch": 3.371429443359375e-05,
      "step": 8838,
      "training_step_time": 9.929657459259033
    },
    {
      "epoch": 3.371429443359375e-05,
      "model_forward_time": 3.1394503116607666,
      "step": 8838
    },
    {
      "epoch": 3.371429443359375e-05,
      "step": 8838,
      "training_step_time": 9.917535305023193
    },
    {
      "epoch": 3.371429443359375e-05,
      "model_forward_time": 3.144359827041626,
      "step": 8838
    },
    {
      "epoch": 3.371429443359375e-05,
      "step": 8838,
      "training_step_time": 9.926853895187378
    },
    {
      "epoch": 3.371429443359375e-05,
      "model_forward_time": 3.144073724746704,
      "step": 8838
    },
    {
      "epoch": 3.371429443359375e-05,
      "step": 8838,
      "training_step_time": 10.337592840194702
    },
    {
      "epoch": 3.3718109130859377e-05,
      "model_forward_time": 3.1471478939056396,
      "step": 8839
    },
    {
      "epoch": 3.3718109130859377e-05,
      "step": 8839,
      "training_step_time": 10.250743389129639
    },
    {
      "epoch": 3.3718109130859377e-05,
      "model_forward_time": 3.139636278152466,
      "step": 8839
    },
    {
      "epoch": 3.3718109130859377e-05,
      "step": 8839,
      "training_step_time": 9.913546800613403
    },
    {
      "epoch": 3.3718109130859377e-05,
      "model_forward_time": 3.141152858734131,
      "step": 8839
    },
    {
      "epoch": 3.3718109130859377e-05,
      "step": 8839,
      "training_step_time": 9.922261238098145
    },
    {
      "epoch": 3.3718109130859377e-05,
      "model_forward_time": 2.4356496334075928,
      "step": 8839
    },
    {
      "epoch": 3.3718109130859377e-05,
      "step": 8839,
      "training_step_time": 10.267706394195557
    },
    {
      "epoch": 3.3721923828125e-05,
      "grad_norm": 0.3916585445404053,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 8840
    },
    {
      "epoch": 3.3721923828125e-05,
      "model_forward_time": 3.1450037956237793,
      "step": 8840
    },
    {
      "dynamics_loss_avg": 0.06027985662221909,
      "epoch": 3.3721923828125e-05,
      "step": 8840
    },
    {
      "action_loss_avg": 0.020070266677066684,
      "epoch": 3.3721923828125e-05,
      "step": 8840
    },
    {
      "epoch": 3.3721923828125e-05,
      "step": 8840,
      "training_step_time": 9.924737930297852
    },
    {
      "epoch": 3.3721923828125e-05,
      "model_forward_time": 3.135281562805176,
      "step": 8840
    },
    {
      "epoch": 3.3721923828125e-05,
      "step": 8840,
      "training_step_time": 9.911878824234009
    },
    {
      "epoch": 3.3721923828125e-05,
      "model_forward_time": 3.4048709869384766,
      "step": 8840
    },
    {
      "epoch": 3.3721923828125e-05,
      "step": 8840,
      "training_step_time": 10.181931734085083
    },
    {
      "epoch": 3.3721923828125e-05,
      "model_forward_time": 3.1390581130981445,
      "step": 8840
    },
    {
      "epoch": 3.3721923828125e-05,
      "step": 8840,
      "training_step_time": 10.309463500976562
    },
    {
      "epoch": 3.372573852539062e-05,
      "model_forward_time": 2.442518472671509,
      "step": 8841
    },
    {
      "epoch": 3.372573852539062e-05,
      "step": 8841,
      "training_step_time": 7.470880508422852
    },
    {
      "epoch": 3.372573852539062e-05,
      "model_forward_time": 3.1370558738708496,
      "step": 8841
    },
    {
      "epoch": 3.372573852539062e-05,
      "step": 8841,
      "training_step_time": 9.905184745788574
    },
    {
      "epoch": 3.372573852539062e-05,
      "model_forward_time": 3.132535457611084,
      "step": 8841
    },
    {
      "epoch": 3.372573852539062e-05,
      "step": 8841,
      "training_step_time": 9.904239416122437
    },
    {
      "epoch": 3.372573852539062e-05,
      "model_forward_time": 3.4161131381988525,
      "step": 8841
    },
    {
      "epoch": 3.372573852539062e-05,
      "step": 8841,
      "training_step_time": 10.297378301620483
    },
    {
      "epoch": 3.372955322265625e-05,
      "model_forward_time": 3.157268762588501,
      "step": 8842
    },
    {
      "epoch": 3.372955322265625e-05,
      "step": 8842,
      "training_step_time": 9.934494018554688
    },
    {
      "epoch": 3.372955322265625e-05,
      "model_forward_time": 3.1491663455963135,
      "step": 8842
    },
    {
      "epoch": 3.372955322265625e-05,
      "step": 8842,
      "training_step_time": 9.926196813583374
    },
    {
      "epoch": 3.372955322265625e-05,
      "model_forward_time": 3.1341962814331055,
      "step": 8842
    },
    {
      "dynamics_loss_avg": 0.05415998045355082,
      "epoch": 3.372955322265625e-05,
      "step": 8842
    },
    {
      "action_loss_avg": 0.018443648191168904,
      "epoch": 3.372955322265625e-05,
      "step": 8842
    },
    {
      "epoch": 3.372955322265625e-05,
      "step": 8842,
      "training_step_time": 9.909831523895264
    },
    {
      "epoch": 3.372955322265625e-05,
      "model_forward_time": 3.403550386428833,
      "step": 8842
    },
    {
      "epoch": 3.372955322265625e-05,
      "step": 8842,
      "training_step_time": 10.270297288894653
    },
    {
      "epoch": 3.3733367919921875e-05,
      "model_forward_time": 3.1475841999053955,
      "step": 8843
    },
    {
      "epoch": 3.3733367919921875e-05,
      "step": 8843,
      "training_step_time": 9.927098989486694
    },
    {
      "epoch": 3.3733367919921875e-05,
      "model_forward_time": 3.1410293579101562,
      "step": 8843
    },
    {
      "epoch": 3.3733367919921875e-05,
      "step": 8843,
      "training_step_time": 9.917412519454956
    },
    {
      "epoch": 3.3733367919921875e-05,
      "model_forward_time": 3.134185314178467,
      "step": 8843
    },
    {
      "epoch": 3.3733367919921875e-05,
      "step": 8843,
      "training_step_time": 9.913578987121582
    },
    {
      "epoch": 3.3733367919921875e-05,
      "model_forward_time": 3.142906427383423,
      "step": 8843
    },
    {
      "epoch": 3.3733367919921875e-05,
      "step": 8843,
      "training_step_time": 10.31679368019104
    },
    {
      "epoch": 3.37371826171875e-05,
      "model_forward_time": 3.158996105194092,
      "step": 8844
    },
    {
      "epoch": 3.37371826171875e-05,
      "step": 8844,
      "training_step_time": 10.23638105392456
    },
    {
      "epoch": 3.37371826171875e-05,
      "model_forward_time": 3.143327474594116,
      "step": 8844
    },
    {
      "epoch": 3.37371826171875e-05,
      "step": 8844,
      "training_step_time": 9.914738416671753
    },
    {
      "epoch": 3.37371826171875e-05,
      "model_forward_time": 3.1398041248321533,
      "step": 8844
    },
    {
      "epoch": 3.37371826171875e-05,
      "step": 8844,
      "training_step_time": 9.911320686340332
    },
    {
      "epoch": 3.37371826171875e-05,
      "model_forward_time": 3.139660358428955,
      "step": 8844
    },
    {
      "epoch": 3.37371826171875e-05,
      "step": 8844,
      "training_step_time": 10.248301029205322
    },
    {
      "epoch": 3.374099731445313e-05,
      "model_forward_time": 3.1484286785125732,
      "step": 8845
    },
    {
      "dynamics_loss_avg": 0.05716768316924572,
      "epoch": 3.374099731445313e-05,
      "step": 8845
    },
    {
      "action_loss_avg": 0.023039055336266757,
      "epoch": 3.374099731445313e-05,
      "step": 8845
    },
    {
      "epoch": 3.374099731445313e-05,
      "step": 8845,
      "training_step_time": 9.92792558670044
    },
    {
      "epoch": 3.374099731445313e-05,
      "model_forward_time": 3.4210989475250244,
      "step": 8845
    },
    {
      "epoch": 3.374099731445313e-05,
      "step": 8845,
      "training_step_time": 10.196362018585205
    },
    {
      "epoch": 3.374099731445313e-05,
      "model_forward_time": 3.1415092945098877,
      "step": 8845
    },
    {
      "epoch": 3.374099731445313e-05,
      "step": 8845,
      "training_step_time": 9.918928623199463
    },
    {
      "epoch": 3.374099731445313e-05,
      "model_forward_time": 3.144122838973999,
      "step": 8845
    },
    {
      "epoch": 3.374099731445313e-05,
      "step": 8845,
      "training_step_time": 10.01507019996643
    },
    {
      "epoch": 3.374481201171875e-05,
      "model_forward_time": 3.1643869876861572,
      "step": 8846
    },
    {
      "epoch": 3.374481201171875e-05,
      "step": 8846,
      "training_step_time": 9.941567182540894
    },
    {
      "epoch": 3.374481201171875e-05,
      "model_forward_time": 3.1409575939178467,
      "step": 8846
    },
    {
      "epoch": 3.374481201171875e-05,
      "step": 8846,
      "training_step_time": 9.91431975364685
    },
    {
      "epoch": 3.374481201171875e-05,
      "model_forward_time": 3.4093260765075684,
      "step": 8846
    },
    {
      "epoch": 3.374481201171875e-05,
      "step": 8846,
      "training_step_time": 10.18640422821045
    },
    {
      "epoch": 3.374481201171875e-05,
      "model_forward_time": 3.1449239253997803,
      "step": 8846
    },
    {
      "epoch": 3.374481201171875e-05,
      "step": 8846,
      "training_step_time": 10.293588638305664
    },
    {
      "epoch": 3.374862670898437e-05,
      "model_forward_time": 3.1553311347961426,
      "step": 8847
    },
    {
      "epoch": 3.374862670898437e-05,
      "step": 8847,
      "training_step_time": 9.933324813842773
    },
    {
      "epoch": 3.374862670898437e-05,
      "model_forward_time": 3.1396467685699463,
      "step": 8847
    },
    {
      "epoch": 3.374862670898437e-05,
      "step": 8847,
      "training_step_time": 9.91969633102417
    },
    {
      "epoch": 3.374862670898437e-05,
      "model_forward_time": 3.146970748901367,
      "step": 8847
    },
    {
      "dynamics_loss_avg": 0.05337412357330322,
      "epoch": 3.374862670898437e-05,
      "step": 8847
    },
    {
      "action_loss_avg": 0.01920587127096951,
      "epoch": 3.374862670898437e-05,
      "step": 8847
    },
    {
      "epoch": 3.374862670898437e-05,
      "step": 8847,
      "training_step_time": 9.926049947738647
    },
    {
      "epoch": 3.374862670898437e-05,
      "model_forward_time": 2.7241756916046143,
      "step": 8847
    },
    {
      "epoch": 3.374862670898437e-05,
      "step": 8847,
      "training_step_time": 10.282742977142334
    },
    {
      "epoch": 3.375244140625e-05,
      "model_forward_time": 2.4493484497070312,
      "step": 8848
    },
    {
      "epoch": 3.375244140625e-05,
      "step": 8848,
      "training_step_time": 7.482576370239258
    },
    {
      "epoch": 3.375244140625e-05,
      "model_forward_time": 2.43688702583313,
      "step": 8848
    },
    {
      "epoch": 3.375244140625e-05,
      "step": 8848,
      "training_step_time": 7.462491035461426
    },
    {
      "epoch": 3.375244140625e-05,
      "model_forward_time": 3.141366958618164,
      "step": 8848
    },
    {
      "epoch": 3.375244140625e-05,
      "step": 8848,
      "training_step_time": 9.909687995910645
    },
    {
      "epoch": 3.375244140625e-05,
      "model_forward_time": 3.133056163787842,
      "step": 8848
    },
    {
      "epoch": 3.375244140625e-05,
      "step": 8848,
      "training_step_time": 10.278261423110962
    },
    {
      "epoch": 3.3756256103515626e-05,
      "model_forward_time": 3.1475963592529297,
      "step": 8849
    },
    {
      "epoch": 3.3756256103515626e-05,
      "step": 8849,
      "training_step_time": 10.185662031173706
    },
    {
      "epoch": 3.3756256103515626e-05,
      "model_forward_time": 3.1396689414978027,
      "step": 8849
    },
    {
      "epoch": 3.3756256103515626e-05,
      "step": 8849,
      "training_step_time": 9.911149501800537
    },
    {
      "epoch": 3.3756256103515626e-05,
      "model_forward_time": 3.1415350437164307,
      "step": 8849
    },
    {
      "epoch": 3.3756256103515626e-05,
      "step": 8849,
      "training_step_time": 9.90907597541809
    },
    {
      "epoch": 3.3756256103515626e-05,
      "model_forward_time": 3.139247417449951,
      "step": 8849
    },
    {
      "epoch": 3.3756256103515626e-05,
      "step": 8849,
      "training_step_time": 10.222285509109497
    },
    {
      "epoch": 3.376007080078125e-05,
      "grad_norm": 0.31609490513801575,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 8850
    },
    {
      "epoch": 3.376007080078125e-05,
      "model_forward_time": 2.4484457969665527,
      "step": 8850
    },
    {
      "dynamics_loss_avg": 0.06807798836380244,
      "epoch": 3.376007080078125e-05,
      "step": 8850
    },
    {
      "action_loss_avg": 0.023939118534326554,
      "epoch": 3.376007080078125e-05,
      "step": 8850
    },
    {
      "epoch": 3.376007080078125e-05,
      "step": 8850,
      "training_step_time": 7.476894855499268
    },
    {
      "epoch": 3.376007080078125e-05,
      "model_forward_time": 3.1358513832092285,
      "step": 8850
    },
    {
      "epoch": 3.376007080078125e-05,
      "step": 8850,
      "training_step_time": 10.216298818588257
    },
    {
      "epoch": 3.376007080078125e-05,
      "model_forward_time": 3.1372809410095215,
      "step": 8850
    },
    {
      "epoch": 3.376007080078125e-05,
      "step": 8850,
      "training_step_time": 9.910826683044434
    },
    {
      "epoch": 3.376007080078125e-05,
      "model_forward_time": 3.14021897315979,
      "step": 8850
    },
    {
      "epoch": 3.376007080078125e-05,
      "step": 8850,
      "training_step_time": 10.248486518859863
    },
    {
      "epoch": 3.376388549804688e-05,
      "model_forward_time": 3.1455531120300293,
      "step": 8851
    },
    {
      "epoch": 3.376388549804688e-05,
      "step": 8851,
      "training_step_time": 9.917624950408936
    },
    {
      "epoch": 3.376388549804688e-05,
      "model_forward_time": 3.1394660472869873,
      "step": 8851
    },
    {
      "epoch": 3.376388549804688e-05,
      "step": 8851,
      "training_step_time": 9.915385007858276
    },
    {
      "epoch": 3.376388549804688e-05,
      "model_forward_time": 3.4337852001190186,
      "step": 8851
    },
    {
      "epoch": 3.376388549804688e-05,
      "step": 8851,
      "training_step_time": 10.201040983200073
    },
    {
      "epoch": 3.376388549804688e-05,
      "model_forward_time": 3.137826919555664,
      "step": 8851
    },
    {
      "epoch": 3.376388549804688e-05,
      "step": 8851,
      "training_step_time": 10.009987354278564
    },
    {
      "epoch": 3.37677001953125e-05,
      "model_forward_time": 3.159188747406006,
      "step": 8852
    },
    {
      "epoch": 3.37677001953125e-05,
      "step": 8852,
      "training_step_time": 9.937735795974731
    },
    {
      "epoch": 3.37677001953125e-05,
      "model_forward_time": 3.141209363937378,
      "step": 8852
    },
    {
      "epoch": 3.37677001953125e-05,
      "step": 8852,
      "training_step_time": 9.916142702102661
    },
    {
      "epoch": 3.37677001953125e-05,
      "model_forward_time": 3.14176869392395,
      "step": 8852
    },
    {
      "dynamics_loss_avg": 0.07312612421810627,
      "epoch": 3.37677001953125e-05,
      "step": 8852
    },
    {
      "action_loss_avg": 0.02635545041412115,
      "epoch": 3.37677001953125e-05,
      "step": 8852
    },
    {
      "epoch": 3.37677001953125e-05,
      "step": 8852,
      "training_step_time": 9.920788764953613
    },
    {
      "epoch": 3.37677001953125e-05,
      "model_forward_time": 2.442448616027832,
      "step": 8852
    },
    {
      "epoch": 3.37677001953125e-05,
      "step": 8852,
      "training_step_time": 10.316033124923706
    },
    {
      "epoch": 3.3771514892578124e-05,
      "model_forward_time": 3.4799649715423584,
      "step": 8853
    },
    {
      "epoch": 3.3771514892578124e-05,
      "step": 8853,
      "training_step_time": 10.253558158874512
    },
    {
      "epoch": 3.3771514892578124e-05,
      "model_forward_time": 3.1480202674865723,
      "step": 8853
    },
    {
      "epoch": 3.3771514892578124e-05,
      "step": 8853,
      "training_step_time": 9.926924228668213
    },
    {
      "epoch": 3.3771514892578124e-05,
      "model_forward_time": 2.4424076080322266,
      "step": 8853
    },
    {
      "epoch": 3.3771514892578124e-05,
      "step": 8853,
      "training_step_time": 7.475637197494507
    },
    {
      "epoch": 3.3771514892578124e-05,
      "model_forward_time": 3.146644115447998,
      "step": 8853
    },
    {
      "epoch": 3.3771514892578124e-05,
      "step": 8853,
      "training_step_time": 10.306258916854858
    },
    {
      "epoch": 3.377532958984375e-05,
      "model_forward_time": 3.4188382625579834,
      "step": 8854
    },
    {
      "epoch": 3.377532958984375e-05,
      "step": 8854,
      "training_step_time": 10.193519830703735
    },
    {
      "epoch": 3.377532958984375e-05,
      "model_forward_time": 3.1462602615356445,
      "step": 8854
    },
    {
      "epoch": 3.377532958984375e-05,
      "step": 8854,
      "training_step_time": 9.922105073928833
    },
    {
      "epoch": 3.377532958984375e-05,
      "model_forward_time": 3.1452651023864746,
      "step": 8854
    },
    {
      "epoch": 3.377532958984375e-05,
      "step": 8854,
      "training_step_time": 9.922977209091187
    },
    {
      "epoch": 3.377532958984375e-05,
      "model_forward_time": 3.149824380874634,
      "step": 8854
    },
    {
      "epoch": 3.377532958984375e-05,
      "step": 8854,
      "training_step_time": 10.023902416229248
    },
    {
      "epoch": 3.377914428710938e-05,
      "model_forward_time": 3.163783073425293,
      "step": 8855
    },
    {
      "dynamics_loss_avg": 0.05526059158146381,
      "epoch": 3.377914428710938e-05,
      "step": 8855
    },
    {
      "action_loss_avg": 0.019324333406984806,
      "epoch": 3.377914428710938e-05,
      "step": 8855
    },
    {
      "epoch": 3.377914428710938e-05,
      "step": 8855,
      "training_step_time": 9.944568634033203
    },
    {
      "epoch": 3.377914428710938e-05,
      "model_forward_time": 3.148695707321167,
      "step": 8855
    },
    {
      "epoch": 3.377914428710938e-05,
      "step": 8855,
      "training_step_time": 10.1952543258667
    },
    {
      "epoch": 3.377914428710938e-05,
      "model_forward_time": 2.4410858154296875,
      "step": 8855
    },
    {
      "epoch": 3.377914428710938e-05,
      "step": 8855,
      "training_step_time": 7.472200632095337
    },
    {
      "epoch": 3.377914428710938e-05,
      "model_forward_time": 3.145977020263672,
      "step": 8855
    },
    {
      "epoch": 3.377914428710938e-05,
      "step": 8855,
      "training_step_time": 10.261459350585938
    },
    {
      "epoch": 3.3782958984375e-05,
      "model_forward_time": 3.162142753601074,
      "step": 8856
    },
    {
      "epoch": 3.3782958984375e-05,
      "step": 8856,
      "training_step_time": 9.93000841140747
    },
    {
      "epoch": 3.3782958984375e-05,
      "model_forward_time": 3.1497802734375,
      "step": 8856
    },
    {
      "epoch": 3.3782958984375e-05,
      "step": 8856,
      "training_step_time": 9.929333209991455
    },
    {
      "epoch": 3.3782958984375e-05,
      "model_forward_time": 3.4143292903900146,
      "step": 8856
    },
    {
      "epoch": 3.3782958984375e-05,
      "step": 8856,
      "training_step_time": 10.193593263626099
    },
    {
      "epoch": 3.3782958984375e-05,
      "model_forward_time": 3.149949312210083,
      "step": 8856
    },
    {
      "epoch": 3.3782958984375e-05,
      "step": 8856,
      "training_step_time": 10.026572465896606
    },
    {
      "epoch": 3.378677368164062e-05,
      "model_forward_time": 3.1601147651672363,
      "step": 8857
    },
    {
      "epoch": 3.378677368164062e-05,
      "step": 8857,
      "training_step_time": 9.939144849777222
    },
    {
      "epoch": 3.378677368164062e-05,
      "model_forward_time": 3.139254093170166,
      "step": 8857
    },
    {
      "epoch": 3.378677368164062e-05,
      "step": 8857,
      "training_step_time": 9.910328388214111
    },
    {
      "epoch": 3.378677368164062e-05,
      "model_forward_time": 3.1418988704681396,
      "step": 8857
    },
    {
      "dynamics_loss_avg": 0.05764119103550911,
      "epoch": 3.378677368164062e-05,
      "step": 8857
    },
    {
      "action_loss_avg": 0.02271627513691783,
      "epoch": 3.378677368164062e-05,
      "step": 8857
    },
    {
      "epoch": 3.378677368164062e-05,
      "step": 8857,
      "training_step_time": 9.916423797607422
    },
    {
      "epoch": 3.378677368164062e-05,
      "model_forward_time": 2.4364852905273438,
      "step": 8857
    },
    {
      "epoch": 3.378677368164062e-05,
      "step": 8857,
      "training_step_time": 10.304991483688354
    },
    {
      "epoch": 3.379058837890625e-05,
      "model_forward_time": 3.4969544410705566,
      "step": 8858
    },
    {
      "epoch": 3.379058837890625e-05,
      "step": 8858,
      "training_step_time": 10.272647142410278
    },
    {
      "epoch": 3.379058837890625e-05,
      "model_forward_time": 3.1379554271698,
      "step": 8858
    },
    {
      "epoch": 3.379058837890625e-05,
      "step": 8858,
      "training_step_time": 9.908880949020386
    },
    {
      "epoch": 3.379058837890625e-05,
      "model_forward_time": 2.4375874996185303,
      "step": 8858
    },
    {
      "epoch": 3.379058837890625e-05,
      "step": 8858,
      "training_step_time": 7.461509466171265
    },
    {
      "epoch": 3.379058837890625e-05,
      "model_forward_time": 3.136687994003296,
      "step": 8858
    },
    {
      "epoch": 3.379058837890625e-05,
      "step": 8858,
      "training_step_time": 10.296642780303955
    },
    {
      "epoch": 3.3794403076171875e-05,
      "model_forward_time": 3.1514229774475098,
      "step": 8859
    },
    {
      "epoch": 3.3794403076171875e-05,
      "step": 8859,
      "training_step_time": 10.21019196510315
    },
    {
      "epoch": 3.3794403076171875e-05,
      "model_forward_time": 3.1401069164276123,
      "step": 8859
    },
    {
      "epoch": 3.3794403076171875e-05,
      "step": 8859,
      "training_step_time": 9.914896249771118
    },
    {
      "epoch": 3.3794403076171875e-05,
      "model_forward_time": 3.143768310546875,
      "step": 8859
    },
    {
      "epoch": 3.3794403076171875e-05,
      "step": 8859,
      "training_step_time": 9.91924786567688
    },
    {
      "epoch": 3.3794403076171875e-05,
      "model_forward_time": 3.141723155975342,
      "step": 8859
    },
    {
      "epoch": 3.3794403076171875e-05,
      "step": 8859,
      "training_step_time": 10.022320032119751
    },
    {
      "epoch": 3.37982177734375e-05,
      "grad_norm": 0.47001415491104126,
      "learning_rate": 1e-05,
      "loss": 0.0811,
      "step": 8860
    },
    {
      "epoch": 3.37982177734375e-05,
      "model_forward_time": 3.1502039432525635,
      "step": 8860
    },
    {
      "dynamics_loss_avg": 0.050700183399021624,
      "epoch": 3.37982177734375e-05,
      "step": 8860
    },
    {
      "action_loss_avg": 0.023521446879021822,
      "epoch": 3.37982177734375e-05,
      "step": 8860
    },
    {
      "epoch": 3.37982177734375e-05,
      "step": 8860,
      "training_step_time": 9.928763389587402
    },
    {
      "epoch": 3.37982177734375e-05,
      "model_forward_time": 3.4080257415771484,
      "step": 8860
    },
    {
      "epoch": 3.37982177734375e-05,
      "step": 8860,
      "training_step_time": 10.184260129928589
    },
    {
      "epoch": 3.37982177734375e-05,
      "model_forward_time": 3.1542937755584717,
      "step": 8860
    },
    {
      "epoch": 3.37982177734375e-05,
      "step": 8860,
      "training_step_time": 9.932848453521729
    },
    {
      "epoch": 3.37982177734375e-05,
      "model_forward_time": 3.1439967155456543,
      "step": 8860
    },
    {
      "epoch": 3.37982177734375e-05,
      "step": 8860,
      "training_step_time": 10.020774841308594
    },
    {
      "epoch": 3.380203247070313e-05,
      "model_forward_time": 3.163007974624634,
      "step": 8861
    },
    {
      "epoch": 3.380203247070313e-05,
      "step": 8861,
      "training_step_time": 9.942611694335938
    },
    {
      "epoch": 3.380203247070313e-05,
      "model_forward_time": 3.1414546966552734,
      "step": 8861
    },
    {
      "epoch": 3.380203247070313e-05,
      "step": 8861,
      "training_step_time": 9.918604135513306
    },
    {
      "epoch": 3.380203247070313e-05,
      "model_forward_time": 3.1406893730163574,
      "step": 8861
    },
    {
      "epoch": 3.380203247070313e-05,
      "step": 8861,
      "training_step_time": 10.185761451721191
    },
    {
      "epoch": 3.380203247070313e-05,
      "model_forward_time": 3.1419947147369385,
      "step": 8861
    },
    {
      "epoch": 3.380203247070313e-05,
      "step": 8861,
      "training_step_time": 10.344481229782104
    },
    {
      "epoch": 3.380584716796875e-05,
      "model_forward_time": 3.1639516353607178,
      "step": 8862
    },
    {
      "epoch": 3.380584716796875e-05,
      "step": 8862,
      "training_step_time": 9.935837030410767
    },
    {
      "epoch": 3.380584716796875e-05,
      "model_forward_time": 3.1350414752960205,
      "step": 8862
    },
    {
      "epoch": 3.380584716796875e-05,
      "step": 8862,
      "training_step_time": 9.908226251602173
    },
    {
      "epoch": 3.380584716796875e-05,
      "model_forward_time": 3.1315371990203857,
      "step": 8862
    },
    {
      "dynamics_loss_avg": 0.062050309777259824,
      "epoch": 3.380584716796875e-05,
      "step": 8862
    },
    {
      "action_loss_avg": 0.020720021100714803,
      "epoch": 3.380584716796875e-05,
      "step": 8862
    },
    {
      "epoch": 3.380584716796875e-05,
      "step": 8862,
      "training_step_time": 9.910658836364746
    },
    {
      "epoch": 3.380584716796875e-05,
      "model_forward_time": 3.3934428691864014,
      "step": 8862
    },
    {
      "epoch": 3.380584716796875e-05,
      "step": 8862,
      "training_step_time": 10.317172050476074
    },
    {
      "epoch": 3.380966186523437e-05,
      "model_forward_time": 3.1574690341949463,
      "step": 8863
    },
    {
      "epoch": 3.380966186523437e-05,
      "step": 8863,
      "training_step_time": 9.935646772384644
    },
    {
      "epoch": 3.380966186523437e-05,
      "model_forward_time": 3.1388542652130127,
      "step": 8863
    },
    {
      "epoch": 3.380966186523437e-05,
      "step": 8863,
      "training_step_time": 9.918259382247925
    },
    {
      "epoch": 3.380966186523437e-05,
      "model_forward_time": 3.136890411376953,
      "step": 8863
    },
    {
      "epoch": 3.380966186523437e-05,
      "step": 8863,
      "training_step_time": 9.915152549743652
    },
    {
      "epoch": 3.380966186523437e-05,
      "model_forward_time": 3.137240409851074,
      "step": 8863
    },
    {
      "epoch": 3.380966186523437e-05,
      "step": 8863,
      "training_step_time": 10.334171295166016
    },
    {
      "epoch": 3.38134765625e-05,
      "model_forward_time": 3.1473400592803955,
      "step": 8864
    },
    {
      "epoch": 3.38134765625e-05,
      "step": 8864,
      "training_step_time": 10.181626081466675
    },
    {
      "epoch": 3.38134765625e-05,
      "model_forward_time": 3.1387383937835693,
      "step": 8864
    },
    {
      "epoch": 3.38134765625e-05,
      "step": 8864,
      "training_step_time": 9.914440155029297
    },
    {
      "epoch": 3.38134765625e-05,
      "model_forward_time": 3.1331443786621094,
      "step": 8864
    },
    {
      "epoch": 3.38134765625e-05,
      "step": 8864,
      "training_step_time": 9.906474351882935
    },
    {
      "epoch": 3.38134765625e-05,
      "model_forward_time": 3.144379138946533,
      "step": 8864
    },
    {
      "epoch": 3.38134765625e-05,
      "step": 8864,
      "training_step_time": 10.037842750549316
    },
    {
      "epoch": 3.3817291259765626e-05,
      "model_forward_time": 3.15039324760437,
      "step": 8865
    },
    {
      "dynamics_loss_avg": 0.046633414924144745,
      "epoch": 3.3817291259765626e-05,
      "step": 8865
    },
    {
      "action_loss_avg": 0.020344611816108228,
      "epoch": 3.3817291259765626e-05,
      "step": 8865
    },
    {
      "epoch": 3.3817291259765626e-05,
      "step": 8865,
      "training_step_time": 9.937445878982544
    },
    {
      "epoch": 3.3817291259765626e-05,
      "model_forward_time": 3.4112067222595215,
      "step": 8865
    },
    {
      "epoch": 3.3817291259765626e-05,
      "step": 8865,
      "training_step_time": 10.193108081817627
    },
    {
      "epoch": 3.3817291259765626e-05,
      "model_forward_time": 3.1359548568725586,
      "step": 8865
    },
    {
      "epoch": 3.3817291259765626e-05,
      "step": 8865,
      "training_step_time": 9.91367220878601
    },
    {
      "epoch": 3.3817291259765626e-05,
      "model_forward_time": 3.1344692707061768,
      "step": 8865
    },
    {
      "epoch": 3.3817291259765626e-05,
      "step": 8865,
      "training_step_time": 10.275665998458862
    },
    {
      "epoch": 3.382110595703125e-05,
      "model_forward_time": 3.1494009494781494,
      "step": 8866
    },
    {
      "epoch": 3.382110595703125e-05,
      "step": 8866,
      "training_step_time": 9.926419496536255
    },
    {
      "epoch": 3.382110595703125e-05,
      "model_forward_time": 3.13228178024292,
      "step": 8866
    },
    {
      "epoch": 3.382110595703125e-05,
      "step": 8866,
      "training_step_time": 9.908142805099487
    },
    {
      "epoch": 3.382110595703125e-05,
      "model_forward_time": 2.747758388519287,
      "step": 8866
    },
    {
      "epoch": 3.382110595703125e-05,
      "step": 8866,
      "training_step_time": 7.7770349979400635
    },
    {
      "epoch": 3.382110595703125e-05,
      "model_forward_time": 3.132927179336548,
      "step": 8866
    },
    {
      "epoch": 3.382110595703125e-05,
      "step": 8866,
      "training_step_time": 10.297113180160522
    },
    {
      "epoch": 3.382492065429688e-05,
      "model_forward_time": 3.1493587493896484,
      "step": 8867
    },
    {
      "epoch": 3.382492065429688e-05,
      "step": 8867,
      "training_step_time": 9.92205548286438
    },
    {
      "epoch": 3.382492065429688e-05,
      "model_forward_time": 2.426307439804077,
      "step": 8867
    },
    {
      "epoch": 3.382492065429688e-05,
      "step": 8867,
      "training_step_time": 7.448756694793701
    },
    {
      "epoch": 3.382492065429688e-05,
      "model_forward_time": 2.4282186031341553,
      "step": 8867
    },
    {
      "dynamics_loss_avg": 0.05577159393578768,
      "epoch": 3.382492065429688e-05,
      "step": 8867
    },
    {
      "action_loss_avg": 0.0157312395516783,
      "epoch": 3.382492065429688e-05,
      "step": 8867
    },
    {
      "epoch": 3.382492065429688e-05,
      "step": 8867,
      "training_step_time": 7.459513902664185
    },
    {
      "epoch": 3.382492065429688e-05,
      "model_forward_time": 3.1297450065612793,
      "step": 8867
    },
    {
      "epoch": 3.382492065429688e-05,
      "step": 8867,
      "training_step_time": 10.296522378921509
    },
    {
      "epoch": 3.38287353515625e-05,
      "model_forward_time": 2.443401575088501,
      "step": 8868
    },
    {
      "epoch": 3.38287353515625e-05,
      "step": 8868,
      "training_step_time": 7.468166828155518
    },
    {
      "epoch": 3.38287353515625e-05,
      "model_forward_time": 3.1319186687469482,
      "step": 8868
    },
    {
      "epoch": 3.38287353515625e-05,
      "step": 8868,
      "training_step_time": 9.914189577102661
    },
    {
      "epoch": 3.38287353515625e-05,
      "model_forward_time": 3.1391215324401855,
      "step": 8868
    },
    {
      "epoch": 3.38287353515625e-05,
      "step": 8868,
      "training_step_time": 9.923738479614258
    },
    {
      "epoch": 3.38287353515625e-05,
      "model_forward_time": 3.1364662647247314,
      "step": 8868
    },
    {
      "epoch": 3.38287353515625e-05,
      "step": 8868,
      "training_step_time": 10.337289333343506
    },
    {
      "epoch": 3.3832550048828124e-05,
      "model_forward_time": 3.1478371620178223,
      "step": 8869
    },
    {
      "epoch": 3.3832550048828124e-05,
      "step": 8869,
      "training_step_time": 9.927438974380493
    },
    {
      "epoch": 3.3832550048828124e-05,
      "model_forward_time": 3.408250570297241,
      "step": 8869
    },
    {
      "epoch": 3.3832550048828124e-05,
      "step": 8869,
      "training_step_time": 10.182843446731567
    },
    {
      "epoch": 3.3832550048828124e-05,
      "model_forward_time": 3.1363308429718018,
      "step": 8869
    },
    {
      "epoch": 3.3832550048828124e-05,
      "step": 8869,
      "training_step_time": 9.913648843765259
    },
    {
      "epoch": 3.3832550048828124e-05,
      "model_forward_time": 3.135418653488159,
      "step": 8869
    },
    {
      "epoch": 3.3832550048828124e-05,
      "step": 8869,
      "training_step_time": 10.331353425979614
    },
    {
      "epoch": 3.383636474609375e-05,
      "grad_norm": 0.4779341518878937,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 8870
    },
    {
      "epoch": 3.383636474609375e-05,
      "model_forward_time": 3.1476213932037354,
      "step": 8870
    },
    {
      "dynamics_loss_avg": 0.04859975483268499,
      "epoch": 3.383636474609375e-05,
      "step": 8870
    },
    {
      "action_loss_avg": 0.015419928880874068,
      "epoch": 3.383636474609375e-05,
      "step": 8870
    },
    {
      "epoch": 3.383636474609375e-05,
      "step": 8870,
      "training_step_time": 9.919668197631836
    },
    {
      "epoch": 3.383636474609375e-05,
      "model_forward_time": 3.428680419921875,
      "step": 8870
    },
    {
      "epoch": 3.383636474609375e-05,
      "step": 8870,
      "training_step_time": 10.200369834899902
    },
    {
      "epoch": 3.383636474609375e-05,
      "model_forward_time": 2.4328837394714355,
      "step": 8870
    },
    {
      "epoch": 3.383636474609375e-05,
      "step": 8870,
      "training_step_time": 7.458903551101685
    },
    {
      "epoch": 3.383636474609375e-05,
      "model_forward_time": 3.131348133087158,
      "step": 8870
    },
    {
      "epoch": 3.383636474609375e-05,
      "step": 8870,
      "training_step_time": 10.257946968078613
    },
    {
      "epoch": 3.384017944335938e-05,
      "model_forward_time": 3.205287218093872,
      "step": 8871
    },
    {
      "epoch": 3.384017944335938e-05,
      "step": 8871,
      "training_step_time": 9.98488473892212
    },
    {
      "epoch": 3.384017944335938e-05,
      "model_forward_time": 2.430589199066162,
      "step": 8871
    },
    {
      "epoch": 3.384017944335938e-05,
      "step": 8871,
      "training_step_time": 7.4565839767456055
    },
    {
      "epoch": 3.384017944335938e-05,
      "model_forward_time": 3.137450933456421,
      "step": 8871
    },
    {
      "epoch": 3.384017944335938e-05,
      "step": 8871,
      "training_step_time": 10.192063093185425
    },
    {
      "epoch": 3.384017944335938e-05,
      "model_forward_time": 2.4518439769744873,
      "step": 8871
    },
    {
      "epoch": 3.384017944335938e-05,
      "step": 8871,
      "training_step_time": 10.29972505569458
    },
    {
      "epoch": 3.3843994140625e-05,
      "model_forward_time": 3.1569268703460693,
      "step": 8872
    },
    {
      "epoch": 3.3843994140625e-05,
      "step": 8872,
      "training_step_time": 9.934999942779541
    },
    {
      "epoch": 3.3843994140625e-05,
      "model_forward_time": 3.1410984992980957,
      "step": 8872
    },
    {
      "epoch": 3.3843994140625e-05,
      "step": 8872,
      "training_step_time": 9.916765213012695
    },
    {
      "epoch": 3.3843994140625e-05,
      "model_forward_time": 3.143861770629883,
      "step": 8872
    },
    {
      "dynamics_loss_avg": 0.058344853576272726,
      "epoch": 3.3843994140625e-05,
      "step": 8872
    },
    {
      "action_loss_avg": 0.016688469774089755,
      "epoch": 3.3843994140625e-05,
      "step": 8872
    },
    {
      "epoch": 3.3843994140625e-05,
      "step": 8872,
      "training_step_time": 9.92328429222107
    },
    {
      "epoch": 3.3843994140625e-05,
      "model_forward_time": 3.1455860137939453,
      "step": 8872
    },
    {
      "epoch": 3.3843994140625e-05,
      "step": 8872,
      "training_step_time": 10.23615837097168
    },
    {
      "epoch": 3.384780883789062e-05,
      "model_forward_time": 3.4546003341674805,
      "step": 8873
    },
    {
      "epoch": 3.384780883789062e-05,
      "step": 8873,
      "training_step_time": 10.23236632347107
    },
    {
      "epoch": 3.384780883789062e-05,
      "model_forward_time": 3.145827531814575,
      "step": 8873
    },
    {
      "epoch": 3.384780883789062e-05,
      "step": 8873,
      "training_step_time": 9.919376850128174
    },
    {
      "epoch": 3.384780883789062e-05,
      "model_forward_time": 3.138505220413208,
      "step": 8873
    },
    {
      "epoch": 3.384780883789062e-05,
      "step": 8873,
      "training_step_time": 9.91520357131958
    },
    {
      "epoch": 3.384780883789062e-05,
      "model_forward_time": 3.1431634426116943,
      "step": 8873
    },
    {
      "epoch": 3.384780883789062e-05,
      "step": 8873,
      "training_step_time": 10.322901487350464
    },
    {
      "epoch": 3.385162353515625e-05,
      "model_forward_time": 2.773982048034668,
      "step": 8874
    },
    {
      "epoch": 3.385162353515625e-05,
      "step": 8874,
      "training_step_time": 7.801082611083984
    },
    {
      "epoch": 3.385162353515625e-05,
      "model_forward_time": 3.1392388343811035,
      "step": 8874
    },
    {
      "epoch": 3.385162353515625e-05,
      "step": 8874,
      "training_step_time": 9.91229772567749
    },
    {
      "epoch": 3.385162353515625e-05,
      "model_forward_time": 2.4403162002563477,
      "step": 8874
    },
    {
      "epoch": 3.385162353515625e-05,
      "step": 8874,
      "training_step_time": 7.474336624145508
    },
    {
      "epoch": 3.385162353515625e-05,
      "model_forward_time": 3.138397455215454,
      "step": 8874
    },
    {
      "epoch": 3.385162353515625e-05,
      "step": 8874,
      "training_step_time": 10.309612274169922
    },
    {
      "epoch": 3.3855438232421875e-05,
      "model_forward_time": 3.1554763317108154,
      "step": 8875
    },
    {
      "dynamics_loss_avg": 0.0544509382918477,
      "epoch": 3.3855438232421875e-05,
      "step": 8875
    },
    {
      "action_loss_avg": 0.02688112435862422,
      "epoch": 3.3855438232421875e-05,
      "step": 8875
    },
    {
      "epoch": 3.3855438232421875e-05,
      "step": 8875,
      "training_step_time": 9.941448450088501
    },
    {
      "epoch": 3.3855438232421875e-05,
      "model_forward_time": 3.1433756351470947,
      "step": 8875
    },
    {
      "epoch": 3.3855438232421875e-05,
      "step": 8875,
      "training_step_time": 9.924849033355713
    },
    {
      "epoch": 3.3855438232421875e-05,
      "model_forward_time": 3.4116640090942383,
      "step": 8875
    },
    {
      "epoch": 3.3855438232421875e-05,
      "step": 8875,
      "training_step_time": 10.190685749053955
    },
    {
      "epoch": 3.3855438232421875e-05,
      "model_forward_time": 3.1456892490386963,
      "step": 8875
    },
    {
      "epoch": 3.3855438232421875e-05,
      "step": 8875,
      "training_step_time": 10.271354675292969
    },
    {
      "epoch": 3.38592529296875e-05,
      "model_forward_time": 3.177266836166382,
      "step": 8876
    },
    {
      "epoch": 3.38592529296875e-05,
      "step": 8876,
      "training_step_time": 9.958417654037476
    },
    {
      "epoch": 3.38592529296875e-05,
      "model_forward_time": 3.1411967277526855,
      "step": 8876
    },
    {
      "epoch": 3.38592529296875e-05,
      "step": 8876,
      "training_step_time": 9.916695594787598
    },
    {
      "epoch": 3.38592529296875e-05,
      "model_forward_time": 3.4090635776519775,
      "step": 8876
    },
    {
      "epoch": 3.38592529296875e-05,
      "step": 8876,
      "training_step_time": 10.189505815505981
    },
    {
      "epoch": 3.38592529296875e-05,
      "model_forward_time": 2.4403188228607178,
      "step": 8876
    },
    {
      "epoch": 3.38592529296875e-05,
      "step": 8876,
      "training_step_time": 10.030069589614868
    },
    {
      "epoch": 3.386306762695313e-05,
      "model_forward_time": 3.1805808544158936,
      "step": 8877
    },
    {
      "epoch": 3.386306762695313e-05,
      "step": 8877,
      "training_step_time": 9.96731948852539
    },
    {
      "epoch": 3.386306762695313e-05,
      "model_forward_time": 3.1477816104888916,
      "step": 8877
    },
    {
      "epoch": 3.386306762695313e-05,
      "step": 8877,
      "training_step_time": 9.927562236785889
    },
    {
      "epoch": 3.386306762695313e-05,
      "model_forward_time": 3.149270534515381,
      "step": 8877
    },
    {
      "dynamics_loss_avg": 0.04497255040332675,
      "epoch": 3.386306762695313e-05,
      "step": 8877
    },
    {
      "action_loss_avg": 0.019424069114029406,
      "epoch": 3.386306762695313e-05,
      "step": 8877
    },
    {
      "epoch": 3.386306762695313e-05,
      "step": 8877,
      "training_step_time": 9.928293943405151
    },
    {
      "epoch": 3.386306762695313e-05,
      "model_forward_time": 3.144087553024292,
      "step": 8877
    },
    {
      "epoch": 3.386306762695313e-05,
      "step": 8877,
      "training_step_time": 10.014700174331665
    },
    {
      "epoch": 3.386688232421875e-05,
      "model_forward_time": 3.477121353149414,
      "step": 8878
    },
    {
      "epoch": 3.386688232421875e-05,
      "step": 8878,
      "training_step_time": 10.248875856399536
    },
    {
      "epoch": 3.386688232421875e-05,
      "model_forward_time": 2.435131549835205,
      "step": 8878
    },
    {
      "epoch": 3.386688232421875e-05,
      "step": 8878,
      "training_step_time": 7.462221145629883
    },
    {
      "epoch": 3.386688232421875e-05,
      "model_forward_time": 3.139692544937134,
      "step": 8878
    },
    {
      "epoch": 3.386688232421875e-05,
      "step": 8878,
      "training_step_time": 9.905949592590332
    },
    {
      "epoch": 3.386688232421875e-05,
      "model_forward_time": 3.1398098468780518,
      "step": 8878
    },
    {
      "epoch": 3.386688232421875e-05,
      "step": 8878,
      "training_step_time": 10.338500499725342
    },
    {
      "epoch": 3.387069702148437e-05,
      "model_forward_time": 3.149103879928589,
      "step": 8879
    },
    {
      "epoch": 3.387069702148437e-05,
      "step": 8879,
      "training_step_time": 10.178696393966675
    },
    {
      "epoch": 3.387069702148437e-05,
      "model_forward_time": 2.439077138900757,
      "step": 8879
    },
    {
      "epoch": 3.387069702148437e-05,
      "step": 8879,
      "training_step_time": 7.461106538772583
    },
    {
      "epoch": 3.387069702148437e-05,
      "model_forward_time": 3.1398909091949463,
      "step": 8879
    },
    {
      "epoch": 3.387069702148437e-05,
      "step": 8879,
      "training_step_time": 9.904786825180054
    },
    {
      "epoch": 3.387069702148437e-05,
      "model_forward_time": 3.1389102935791016,
      "step": 8879
    },
    {
      "epoch": 3.387069702148437e-05,
      "step": 8879,
      "training_step_time": 10.308639764785767
    },
    {
      "epoch": 3.387451171875e-05,
      "grad_norm": 0.4355250895023346,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 8880
    },
    {
      "epoch": 3.387451171875e-05,
      "model_forward_time": 3.151142120361328,
      "step": 8880
    },
    {
      "dynamics_loss_avg": 0.04881035163998604,
      "epoch": 3.387451171875e-05,
      "step": 8880
    },
    {
      "action_loss_avg": 0.013470290112309158,
      "epoch": 3.387451171875e-05,
      "step": 8880
    },
    {
      "epoch": 3.387451171875e-05,
      "step": 8880,
      "training_step_time": 9.926593780517578
    },
    {
      "epoch": 3.387451171875e-05,
      "model_forward_time": 3.416304349899292,
      "step": 8880
    },
    {
      "epoch": 3.387451171875e-05,
      "step": 8880,
      "training_step_time": 10.185055017471313
    },
    {
      "epoch": 3.387451171875e-05,
      "model_forward_time": 3.1431567668914795,
      "step": 8880
    },
    {
      "epoch": 3.387451171875e-05,
      "step": 8880,
      "training_step_time": 9.911499261856079
    },
    {
      "epoch": 3.387451171875e-05,
      "model_forward_time": 3.139969825744629,
      "step": 8880
    },
    {
      "epoch": 3.387451171875e-05,
      "step": 8880,
      "training_step_time": 10.026753902435303
    },
    {
      "epoch": 3.3878326416015626e-05,
      "model_forward_time": 3.152869939804077,
      "step": 8881
    },
    {
      "epoch": 3.3878326416015626e-05,
      "step": 8881,
      "training_step_time": 9.925054788589478
    },
    {
      "epoch": 3.3878326416015626e-05,
      "model_forward_time": 3.144136905670166,
      "step": 8881
    },
    {
      "epoch": 3.3878326416015626e-05,
      "step": 8881,
      "training_step_time": 9.913566827774048
    },
    {
      "epoch": 3.3878326416015626e-05,
      "model_forward_time": 3.1398651599884033,
      "step": 8881
    },
    {
      "epoch": 3.3878326416015626e-05,
      "step": 8881,
      "training_step_time": 10.216161489486694
    },
    {
      "epoch": 3.3878326416015626e-05,
      "model_forward_time": 3.1436233520507812,
      "step": 8881
    },
    {
      "epoch": 3.3878326416015626e-05,
      "step": 8881,
      "training_step_time": 10.020658254623413
    },
    {
      "epoch": 3.388214111328125e-05,
      "model_forward_time": 3.1662769317626953,
      "step": 8882
    },
    {
      "epoch": 3.388214111328125e-05,
      "step": 8882,
      "training_step_time": 9.944933414459229
    },
    {
      "epoch": 3.388214111328125e-05,
      "model_forward_time": 3.147078275680542,
      "step": 8882
    },
    {
      "epoch": 3.388214111328125e-05,
      "step": 8882,
      "training_step_time": 9.924555540084839
    },
    {
      "epoch": 3.388214111328125e-05,
      "model_forward_time": 3.1447269916534424,
      "step": 8882
    },
    {
      "dynamics_loss_avg": 0.0536149650812149,
      "epoch": 3.388214111328125e-05,
      "step": 8882
    },
    {
      "action_loss_avg": 0.01563028562813997,
      "epoch": 3.388214111328125e-05,
      "step": 8882
    },
    {
      "epoch": 3.388214111328125e-05,
      "step": 8882,
      "training_step_time": 9.92432713508606
    },
    {
      "epoch": 3.388214111328125e-05,
      "model_forward_time": 2.4372832775115967,
      "step": 8882
    },
    {
      "epoch": 3.388214111328125e-05,
      "step": 8882,
      "training_step_time": 10.02634859085083
    },
    {
      "epoch": 3.388595581054687e-05,
      "model_forward_time": 2.452432632446289,
      "step": 8883
    },
    {
      "epoch": 3.388595581054687e-05,
      "step": 8883,
      "training_step_time": 7.486020088195801
    },
    {
      "epoch": 3.388595581054687e-05,
      "model_forward_time": 3.14210844039917,
      "step": 8883
    },
    {
      "epoch": 3.388595581054687e-05,
      "step": 8883,
      "training_step_time": 9.915878772735596
    },
    {
      "epoch": 3.388595581054687e-05,
      "model_forward_time": 3.1421236991882324,
      "step": 8883
    },
    {
      "epoch": 3.388595581054687e-05,
      "step": 8883,
      "training_step_time": 9.913374900817871
    },
    {
      "epoch": 3.388595581054687e-05,
      "model_forward_time": 3.1436493396759033,
      "step": 8883
    },
    {
      "epoch": 3.388595581054687e-05,
      "step": 8883,
      "training_step_time": 10.027170658111572
    },
    {
      "epoch": 3.38897705078125e-05,
      "model_forward_time": 3.154085874557495,
      "step": 8884
    },
    {
      "epoch": 3.38897705078125e-05,
      "step": 8884,
      "training_step_time": 9.925795793533325
    },
    {
      "epoch": 3.38897705078125e-05,
      "model_forward_time": 3.414321184158325,
      "step": 8884
    },
    {
      "epoch": 3.38897705078125e-05,
      "step": 8884,
      "training_step_time": 10.188440799713135
    },
    {
      "epoch": 3.38897705078125e-05,
      "model_forward_time": 3.1420915126800537,
      "step": 8884
    },
    {
      "epoch": 3.38897705078125e-05,
      "step": 8884,
      "training_step_time": 9.91306185722351
    },
    {
      "epoch": 3.38897705078125e-05,
      "model_forward_time": 3.1498475074768066,
      "step": 8884
    },
    {
      "epoch": 3.38897705078125e-05,
      "step": 8884,
      "training_step_time": 10.27045750617981
    },
    {
      "epoch": 3.3893585205078124e-05,
      "model_forward_time": 3.1612436771392822,
      "step": 8885
    },
    {
      "dynamics_loss_avg": 0.05901688300073147,
      "epoch": 3.3893585205078124e-05,
      "step": 8885
    },
    {
      "action_loss_avg": 0.019896175898611545,
      "epoch": 3.3893585205078124e-05,
      "step": 8885
    },
    {
      "epoch": 3.3893585205078124e-05,
      "step": 8885,
      "training_step_time": 9.936514616012573
    },
    {
      "epoch": 3.3893585205078124e-05,
      "model_forward_time": 3.4292871952056885,
      "step": 8885
    },
    {
      "epoch": 3.3893585205078124e-05,
      "step": 8885,
      "training_step_time": 10.200355529785156
    },
    {
      "epoch": 3.3893585205078124e-05,
      "model_forward_time": 3.1443254947662354,
      "step": 8885
    },
    {
      "epoch": 3.3893585205078124e-05,
      "step": 8885,
      "training_step_time": 9.918832302093506
    },
    {
      "epoch": 3.3893585205078124e-05,
      "model_forward_time": 3.146235466003418,
      "step": 8885
    },
    {
      "epoch": 3.3893585205078124e-05,
      "step": 8885,
      "training_step_time": 10.320067167282104
    },
    {
      "epoch": 3.389739990234375e-05,
      "model_forward_time": 3.1616744995117188,
      "step": 8886
    },
    {
      "epoch": 3.389739990234375e-05,
      "step": 8886,
      "training_step_time": 9.928032875061035
    },
    {
      "epoch": 3.389739990234375e-05,
      "model_forward_time": 3.1355621814727783,
      "step": 8886
    },
    {
      "epoch": 3.389739990234375e-05,
      "step": 8886,
      "training_step_time": 9.899711847305298
    },
    {
      "epoch": 3.389739990234375e-05,
      "model_forward_time": 3.1388068199157715,
      "step": 8886
    },
    {
      "epoch": 3.389739990234375e-05,
      "step": 8886,
      "training_step_time": 9.907196044921875
    },
    {
      "epoch": 3.389739990234375e-05,
      "model_forward_time": 3.4098868370056152,
      "step": 8886
    },
    {
      "epoch": 3.389739990234375e-05,
      "step": 8886,
      "training_step_time": 10.268491744995117
    },
    {
      "epoch": 3.390121459960938e-05,
      "model_forward_time": 3.1562485694885254,
      "step": 8887
    },
    {
      "epoch": 3.390121459960938e-05,
      "step": 8887,
      "training_step_time": 9.923408031463623
    },
    {
      "epoch": 3.390121459960938e-05,
      "model_forward_time": 3.1350111961364746,
      "step": 8887
    },
    {
      "epoch": 3.390121459960938e-05,
      "step": 8887,
      "training_step_time": 9.903377532958984
    },
    {
      "epoch": 3.390121459960938e-05,
      "model_forward_time": 3.139814853668213,
      "step": 8887
    },
    {
      "dynamics_loss_avg": 0.0614232674241066,
      "epoch": 3.390121459960938e-05,
      "step": 8887
    },
    {
      "action_loss_avg": 0.018029230739921333,
      "epoch": 3.390121459960938e-05,
      "step": 8887
    },
    {
      "epoch": 3.390121459960938e-05,
      "step": 8887,
      "training_step_time": 9.910546779632568
    },
    {
      "epoch": 3.390121459960938e-05,
      "model_forward_time": 3.4209554195404053,
      "step": 8887
    },
    {
      "epoch": 3.390121459960938e-05,
      "step": 8887,
      "training_step_time": 10.28513240814209
    },
    {
      "epoch": 3.3905029296875e-05,
      "model_forward_time": 3.155545473098755,
      "step": 8888
    },
    {
      "epoch": 3.3905029296875e-05,
      "step": 8888,
      "training_step_time": 9.928434371948242
    },
    {
      "epoch": 3.3905029296875e-05,
      "model_forward_time": 3.1331303119659424,
      "step": 8888
    },
    {
      "epoch": 3.3905029296875e-05,
      "step": 8888,
      "training_step_time": 9.903986692428589
    },
    {
      "epoch": 3.3905029296875e-05,
      "model_forward_time": 3.140064001083374,
      "step": 8888
    },
    {
      "epoch": 3.3905029296875e-05,
      "step": 8888,
      "training_step_time": 9.91438627243042
    },
    {
      "epoch": 3.3905029296875e-05,
      "model_forward_time": 3.1388001441955566,
      "step": 8888
    },
    {
      "epoch": 3.3905029296875e-05,
      "step": 8888,
      "training_step_time": 10.012887001037598
    },
    {
      "epoch": 3.390884399414062e-05,
      "model_forward_time": 3.4420440196990967,
      "step": 8889
    },
    {
      "epoch": 3.390884399414062e-05,
      "step": 8889,
      "training_step_time": 10.213757753372192
    },
    {
      "epoch": 3.390884399414062e-05,
      "model_forward_time": 3.1406946182250977,
      "step": 8889
    },
    {
      "epoch": 3.390884399414062e-05,
      "step": 8889,
      "training_step_time": 9.912676095962524
    },
    {
      "epoch": 3.390884399414062e-05,
      "model_forward_time": 3.1830556392669678,
      "step": 8889
    },
    {
      "epoch": 3.390884399414062e-05,
      "step": 8889,
      "training_step_time": 9.959495782852173
    },
    {
      "epoch": 3.390884399414062e-05,
      "model_forward_time": 2.433020830154419,
      "step": 8889
    },
    {
      "epoch": 3.390884399414062e-05,
      "step": 8889,
      "training_step_time": 10.295597076416016
    },
    {
      "epoch": 3.391265869140625e-05,
      "grad_norm": 0.676627516746521,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 8890
    },
    {
      "epoch": 3.391265869140625e-05,
      "model_forward_time": 3.1408803462982178,
      "step": 8890
    },
    {
      "dynamics_loss_avg": 0.08221543654799461,
      "epoch": 3.391265869140625e-05,
      "step": 8890
    },
    {
      "action_loss_avg": 0.02733728103339672,
      "epoch": 3.391265869140625e-05,
      "step": 8890
    },
    {
      "epoch": 3.391265869140625e-05,
      "step": 8890,
      "training_step_time": 9.922259092330933
    },
    {
      "epoch": 3.391265869140625e-05,
      "model_forward_time": 3.3906075954437256,
      "step": 8890
    },
    {
      "epoch": 3.391265869140625e-05,
      "step": 8890,
      "training_step_time": 10.159956932067871
    },
    {
      "epoch": 3.391265869140625e-05,
      "model_forward_time": 3.1386187076568604,
      "step": 8890
    },
    {
      "epoch": 3.391265869140625e-05,
      "step": 8890,
      "training_step_time": 9.911389589309692
    },
    {
      "epoch": 3.391265869140625e-05,
      "model_forward_time": 3.1356616020202637,
      "step": 8890
    },
    {
      "epoch": 3.391265869140625e-05,
      "step": 8890,
      "training_step_time": 10.282225370407104
    },
    {
      "epoch": 3.3916473388671875e-05,
      "model_forward_time": 2.456045389175415,
      "step": 8891
    },
    {
      "epoch": 3.3916473388671875e-05,
      "step": 8891,
      "training_step_time": 7.488963603973389
    },
    {
      "epoch": 3.3916473388671875e-05,
      "model_forward_time": 3.1395986080169678,
      "step": 8891
    },
    {
      "epoch": 3.3916473388671875e-05,
      "step": 8891,
      "training_step_time": 9.912048101425171
    },
    {
      "epoch": 3.3916473388671875e-05,
      "model_forward_time": 3.421044111251831,
      "step": 8891
    },
    {
      "epoch": 3.3916473388671875e-05,
      "step": 8891,
      "training_step_time": 10.193623065948486
    },
    {
      "epoch": 3.3916473388671875e-05,
      "model_forward_time": 2.4392359256744385,
      "step": 8891
    },
    {
      "epoch": 3.3916473388671875e-05,
      "step": 8891,
      "training_step_time": 10.03554391860962
    },
    {
      "epoch": 3.39202880859375e-05,
      "model_forward_time": 3.155038595199585,
      "step": 8892
    },
    {
      "epoch": 3.39202880859375e-05,
      "step": 8892,
      "training_step_time": 9.925138235092163
    },
    {
      "epoch": 3.39202880859375e-05,
      "model_forward_time": 3.1380372047424316,
      "step": 8892
    },
    {
      "epoch": 3.39202880859375e-05,
      "step": 8892,
      "training_step_time": 9.91235065460205
    },
    {
      "epoch": 3.39202880859375e-05,
      "model_forward_time": 3.1379714012145996,
      "step": 8892
    },
    {
      "dynamics_loss_avg": 0.049299226608127356,
      "epoch": 3.39202880859375e-05,
      "step": 8892
    },
    {
      "action_loss_avg": 0.015050578210502864,
      "epoch": 3.39202880859375e-05,
      "step": 8892
    },
    {
      "epoch": 3.39202880859375e-05,
      "step": 8892,
      "training_step_time": 9.910141944885254
    },
    {
      "epoch": 3.39202880859375e-05,
      "model_forward_time": 3.432901382446289,
      "step": 8892
    },
    {
      "epoch": 3.39202880859375e-05,
      "step": 8892,
      "training_step_time": 10.302223920822144
    },
    {
      "epoch": 3.392410278320313e-05,
      "model_forward_time": 3.1489264965057373,
      "step": 8893
    },
    {
      "epoch": 3.392410278320313e-05,
      "step": 8893,
      "training_step_time": 9.915360927581787
    },
    {
      "epoch": 3.392410278320313e-05,
      "model_forward_time": 3.1390957832336426,
      "step": 8893
    },
    {
      "epoch": 3.392410278320313e-05,
      "step": 8893,
      "training_step_time": 9.904854536056519
    },
    {
      "epoch": 3.392410278320313e-05,
      "model_forward_time": 3.134413957595825,
      "step": 8893
    },
    {
      "epoch": 3.392410278320313e-05,
      "step": 8893,
      "training_step_time": 9.89806580543518
    },
    {
      "epoch": 3.392410278320313e-05,
      "model_forward_time": 3.140768527984619,
      "step": 8893
    },
    {
      "epoch": 3.392410278320313e-05,
      "step": 8893,
      "training_step_time": 10.02312707901001
    },
    {
      "epoch": 3.392791748046875e-05,
      "model_forward_time": 2.7386996746063232,
      "step": 8894
    },
    {
      "epoch": 3.392791748046875e-05,
      "step": 8894,
      "training_step_time": 7.764899015426636
    },
    {
      "epoch": 3.392791748046875e-05,
      "model_forward_time": 3.141930341720581,
      "step": 8894
    },
    {
      "epoch": 3.392791748046875e-05,
      "step": 8894,
      "training_step_time": 9.911144256591797
    },
    {
      "epoch": 3.392791748046875e-05,
      "model_forward_time": 3.1382622718811035,
      "step": 8894
    },
    {
      "epoch": 3.392791748046875e-05,
      "step": 8894,
      "training_step_time": 9.908964157104492
    },
    {
      "epoch": 3.392791748046875e-05,
      "model_forward_time": 3.141375780105591,
      "step": 8894
    },
    {
      "epoch": 3.392791748046875e-05,
      "step": 8894,
      "training_step_time": 10.292545557022095
    },
    {
      "epoch": 3.393173217773437e-05,
      "model_forward_time": 2.4563794136047363,
      "step": 8895
    },
    {
      "dynamics_loss_avg": 0.06146624106913805,
      "epoch": 3.393173217773437e-05,
      "step": 8895
    },
    {
      "action_loss_avg": 0.020553499413654207,
      "epoch": 3.393173217773437e-05,
      "step": 8895
    },
    {
      "epoch": 3.393173217773437e-05,
      "step": 8895,
      "training_step_time": 7.4838175773620605
    },
    {
      "epoch": 3.393173217773437e-05,
      "model_forward_time": 3.1346683502197266,
      "step": 8895
    },
    {
      "epoch": 3.393173217773437e-05,
      "step": 8895,
      "training_step_time": 9.901647329330444
    },
    {
      "epoch": 3.393173217773437e-05,
      "model_forward_time": 3.385702610015869,
      "step": 8895
    },
    {
      "epoch": 3.393173217773437e-05,
      "step": 8895,
      "training_step_time": 10.156496286392212
    },
    {
      "epoch": 3.393173217773437e-05,
      "model_forward_time": 3.1473522186279297,
      "step": 8895
    },
    {
      "epoch": 3.393173217773437e-05,
      "step": 8895,
      "training_step_time": 10.295031547546387
    },
    {
      "epoch": 3.3935546875e-05,
      "model_forward_time": 3.151628255844116,
      "step": 8896
    },
    {
      "epoch": 3.3935546875e-05,
      "step": 8896,
      "training_step_time": 9.923289060592651
    },
    {
      "epoch": 3.3935546875e-05,
      "model_forward_time": 3.1400251388549805,
      "step": 8896
    },
    {
      "epoch": 3.3935546875e-05,
      "step": 8896,
      "training_step_time": 9.910173416137695
    },
    {
      "epoch": 3.3935546875e-05,
      "model_forward_time": 3.403470516204834,
      "step": 8896
    },
    {
      "epoch": 3.3935546875e-05,
      "step": 8896,
      "training_step_time": 10.176545143127441
    },
    {
      "epoch": 3.3935546875e-05,
      "model_forward_time": 3.14420223236084,
      "step": 8896
    },
    {
      "epoch": 3.3935546875e-05,
      "step": 8896,
      "training_step_time": 10.041374444961548
    },
    {
      "epoch": 3.3939361572265626e-05,
      "model_forward_time": 3.159860610961914,
      "step": 8897
    },
    {
      "epoch": 3.3939361572265626e-05,
      "step": 8897,
      "training_step_time": 9.941911458969116
    },
    {
      "epoch": 3.3939361572265626e-05,
      "model_forward_time": 3.1378347873687744,
      "step": 8897
    },
    {
      "epoch": 3.3939361572265626e-05,
      "step": 8897,
      "training_step_time": 9.912738800048828
    },
    {
      "epoch": 3.3939361572265626e-05,
      "model_forward_time": 3.1402909755706787,
      "step": 8897
    },
    {
      "dynamics_loss_avg": 0.07278515957295895,
      "epoch": 3.3939361572265626e-05,
      "step": 8897
    },
    {
      "action_loss_avg": 0.02399287335574627,
      "epoch": 3.3939361572265626e-05,
      "step": 8897
    },
    {
      "epoch": 3.3939361572265626e-05,
      "step": 8897,
      "training_step_time": 9.913610935211182
    },
    {
      "epoch": 3.3939361572265626e-05,
      "model_forward_time": 3.138171434402466,
      "step": 8897
    },
    {
      "epoch": 3.3939361572265626e-05,
      "step": 8897,
      "training_step_time": 10.313197135925293
    },
    {
      "epoch": 3.394317626953125e-05,
      "model_forward_time": 3.1555960178375244,
      "step": 8898
    },
    {
      "epoch": 3.394317626953125e-05,
      "step": 8898,
      "training_step_time": 9.927867650985718
    },
    {
      "epoch": 3.394317626953125e-05,
      "model_forward_time": 2.4338459968566895,
      "step": 8898
    },
    {
      "epoch": 3.394317626953125e-05,
      "step": 8898,
      "training_step_time": 7.466521501541138
    },
    {
      "epoch": 3.394317626953125e-05,
      "model_forward_time": 3.1328039169311523,
      "step": 8898
    },
    {
      "epoch": 3.394317626953125e-05,
      "step": 8898,
      "training_step_time": 9.902433156967163
    },
    {
      "epoch": 3.394317626953125e-05,
      "model_forward_time": 3.1359305381774902,
      "step": 8898
    },
    {
      "epoch": 3.394317626953125e-05,
      "step": 8898,
      "training_step_time": 10.027690649032593
    },
    {
      "epoch": 3.394699096679687e-05,
      "model_forward_time": 3.1427931785583496,
      "step": 8899
    },
    {
      "epoch": 3.394699096679687e-05,
      "step": 8899,
      "training_step_time": 10.194043397903442
    },
    {
      "epoch": 3.394699096679687e-05,
      "model_forward_time": 3.1362953186035156,
      "step": 8899
    },
    {
      "epoch": 3.394699096679687e-05,
      "step": 8899,
      "training_step_time": 9.903449296951294
    },
    {
      "epoch": 3.394699096679687e-05,
      "model_forward_time": 3.1403112411499023,
      "step": 8899
    },
    {
      "epoch": 3.394699096679687e-05,
      "step": 8899,
      "training_step_time": 9.903784990310669
    },
    {
      "epoch": 3.394699096679687e-05,
      "model_forward_time": 3.1367950439453125,
      "step": 8899
    },
    {
      "epoch": 3.394699096679687e-05,
      "step": 8899,
      "training_step_time": 10.310656785964966
    },
    {
      "epoch": 3.39508056640625e-05,
      "grad_norm": 0.4412417411804199,
      "learning_rate": 1e-05,
      "loss": 0.0819,
      "step": 8900
    },
    {
      "epoch": 3.39508056640625e-05,
      "model_forward_time": 3.1457462310791016,
      "step": 8900
    },
    {
      "dynamics_loss_avg": 0.07420644909143448,
      "epoch": 3.39508056640625e-05,
      "step": 8900
    },
    {
      "action_loss_avg": 0.028108219616115093,
      "epoch": 3.39508056640625e-05,
      "step": 8900
    },
    {
      "epoch": 3.39508056640625e-05,
      "step": 8900,
      "training_step_time": 9.921778440475464
    },
    {
      "epoch": 3.39508056640625e-05,
      "model_forward_time": 3.398045301437378,
      "step": 8900
    },
    {
      "epoch": 3.39508056640625e-05,
      "step": 8900,
      "training_step_time": 10.162802934646606
    },
    {
      "epoch": 3.39508056640625e-05,
      "model_forward_time": 3.140408992767334,
      "step": 8900
    },
    {
      "epoch": 3.39508056640625e-05,
      "step": 8900,
      "training_step_time": 9.906195402145386
    },
    {
      "epoch": 3.39508056640625e-05,
      "model_forward_time": 3.1356983184814453,
      "step": 8900
    },
    {
      "epoch": 3.39508056640625e-05,
      "step": 8900,
      "training_step_time": 10.299710750579834
    },
    {
      "epoch": 3.3954620361328124e-05,
      "model_forward_time": 3.154646396636963,
      "step": 8901
    },
    {
      "epoch": 3.3954620361328124e-05,
      "step": 8901,
      "training_step_time": 9.93595576286316
    },
    {
      "epoch": 3.3954620361328124e-05,
      "model_forward_time": 3.137281656265259,
      "step": 8901
    },
    {
      "epoch": 3.3954620361328124e-05,
      "step": 8901,
      "training_step_time": 9.91295576095581
    },
    {
      "epoch": 3.3954620361328124e-05,
      "model_forward_time": 3.1424953937530518,
      "step": 8901
    },
    {
      "epoch": 3.3954620361328124e-05,
      "step": 8901,
      "training_step_time": 10.186833381652832
    },
    {
      "epoch": 3.3954620361328124e-05,
      "model_forward_time": 3.139909267425537,
      "step": 8901
    },
    {
      "epoch": 3.3954620361328124e-05,
      "step": 8901,
      "training_step_time": 10.333012104034424
    },
    {
      "epoch": 3.395843505859375e-05,
      "model_forward_time": 3.145306348800659,
      "step": 8902
    },
    {
      "epoch": 3.395843505859375e-05,
      "step": 8902,
      "training_step_time": 9.919224739074707
    },
    {
      "epoch": 3.395843505859375e-05,
      "model_forward_time": 3.138427495956421,
      "step": 8902
    },
    {
      "epoch": 3.395843505859375e-05,
      "step": 8902,
      "training_step_time": 9.912007331848145
    },
    {
      "epoch": 3.395843505859375e-05,
      "model_forward_time": 3.1342999935150146,
      "step": 8902
    },
    {
      "dynamics_loss_avg": 0.06030202051624656,
      "epoch": 3.395843505859375e-05,
      "step": 8902
    },
    {
      "action_loss_avg": 0.0260654681827873,
      "epoch": 3.395843505859375e-05,
      "step": 8902
    },
    {
      "epoch": 3.395843505859375e-05,
      "step": 8902,
      "training_step_time": 9.913896322250366
    },
    {
      "epoch": 3.395843505859375e-05,
      "model_forward_time": 3.393167495727539,
      "step": 8902
    },
    {
      "epoch": 3.395843505859375e-05,
      "step": 8902,
      "training_step_time": 10.307994365692139
    },
    {
      "epoch": 3.396224975585938e-05,
      "model_forward_time": 3.1478521823883057,
      "step": 8903
    },
    {
      "epoch": 3.396224975585938e-05,
      "step": 8903,
      "training_step_time": 9.922200679779053
    },
    {
      "epoch": 3.396224975585938e-05,
      "model_forward_time": 3.1340878009796143,
      "step": 8903
    },
    {
      "epoch": 3.396224975585938e-05,
      "step": 8903,
      "training_step_time": 9.902024269104004
    },
    {
      "epoch": 3.396224975585938e-05,
      "model_forward_time": 3.128413677215576,
      "step": 8903
    },
    {
      "epoch": 3.396224975585938e-05,
      "step": 8903,
      "training_step_time": 9.908395767211914
    },
    {
      "epoch": 3.396224975585938e-05,
      "model_forward_time": 3.1395318508148193,
      "step": 8903
    },
    {
      "epoch": 3.396224975585938e-05,
      "step": 8903,
      "training_step_time": 10.011878252029419
    },
    {
      "epoch": 3.3966064453125e-05,
      "model_forward_time": 3.1525065898895264,
      "step": 8904
    },
    {
      "epoch": 3.3966064453125e-05,
      "step": 8904,
      "training_step_time": 10.192698955535889
    },
    {
      "epoch": 3.3966064453125e-05,
      "model_forward_time": 2.4382169246673584,
      "step": 8904
    },
    {
      "epoch": 3.3966064453125e-05,
      "step": 8904,
      "training_step_time": 7.469736814498901
    },
    {
      "epoch": 3.3966064453125e-05,
      "model_forward_time": 3.1374194622039795,
      "step": 8904
    },
    {
      "epoch": 3.3966064453125e-05,
      "step": 8904,
      "training_step_time": 9.909496307373047
    },
    {
      "epoch": 3.3966064453125e-05,
      "model_forward_time": 3.1438236236572266,
      "step": 8904
    },
    {
      "epoch": 3.3966064453125e-05,
      "step": 8904,
      "training_step_time": 10.031753063201904
    },
    {
      "epoch": 3.396987915039062e-05,
      "model_forward_time": 3.156728744506836,
      "step": 8905
    },
    {
      "dynamics_loss_avg": 0.0527552729472518,
      "epoch": 3.396987915039062e-05,
      "step": 8905
    },
    {
      "action_loss_avg": 0.01790065662935376,
      "epoch": 3.396987915039062e-05,
      "step": 8905
    },
    {
      "epoch": 3.396987915039062e-05,
      "step": 8905,
      "training_step_time": 9.935625314712524
    },
    {
      "epoch": 3.396987915039062e-05,
      "model_forward_time": 3.4279842376708984,
      "step": 8905
    },
    {
      "epoch": 3.396987915039062e-05,
      "step": 8905,
      "training_step_time": 10.205358982086182
    },
    {
      "epoch": 3.396987915039062e-05,
      "model_forward_time": 3.152984857559204,
      "step": 8905
    },
    {
      "epoch": 3.396987915039062e-05,
      "step": 8905,
      "training_step_time": 9.934772968292236
    },
    {
      "epoch": 3.396987915039062e-05,
      "model_forward_time": 3.152381420135498,
      "step": 8905
    },
    {
      "epoch": 3.396987915039062e-05,
      "step": 8905,
      "training_step_time": 10.341290473937988
    },
    {
      "epoch": 3.397369384765625e-05,
      "model_forward_time": 3.1598005294799805,
      "step": 8906
    },
    {
      "epoch": 3.397369384765625e-05,
      "step": 8906,
      "training_step_time": 9.940810680389404
    },
    {
      "epoch": 3.397369384765625e-05,
      "model_forward_time": 2.4357059001922607,
      "step": 8906
    },
    {
      "epoch": 3.397369384765625e-05,
      "step": 8906,
      "training_step_time": 7.466914892196655
    },
    {
      "epoch": 3.397369384765625e-05,
      "model_forward_time": 3.139314651489258,
      "step": 8906
    },
    {
      "epoch": 3.397369384765625e-05,
      "step": 8906,
      "training_step_time": 10.180725336074829
    },
    {
      "epoch": 3.397369384765625e-05,
      "model_forward_time": 3.1460347175598145,
      "step": 8906
    },
    {
      "epoch": 3.397369384765625e-05,
      "step": 8906,
      "training_step_time": 10.273429870605469
    },
    {
      "epoch": 3.3977508544921875e-05,
      "model_forward_time": 3.1543171405792236,
      "step": 8907
    },
    {
      "epoch": 3.3977508544921875e-05,
      "step": 8907,
      "training_step_time": 9.93827199935913
    },
    {
      "epoch": 3.3977508544921875e-05,
      "model_forward_time": 2.4328103065490723,
      "step": 8907
    },
    {
      "epoch": 3.3977508544921875e-05,
      "step": 8907,
      "training_step_time": 7.467942237854004
    },
    {
      "epoch": 3.3977508544921875e-05,
      "model_forward_time": 2.437293291091919,
      "step": 8907
    },
    {
      "dynamics_loss_avg": 0.057862674817442894,
      "epoch": 3.3977508544921875e-05,
      "step": 8907
    },
    {
      "action_loss_avg": 0.019581330800428988,
      "epoch": 3.3977508544921875e-05,
      "step": 8907
    },
    {
      "epoch": 3.3977508544921875e-05,
      "step": 8907,
      "training_step_time": 7.469747304916382
    },
    {
      "epoch": 3.3977508544921875e-05,
      "model_forward_time": 3.1315255165100098,
      "step": 8907
    },
    {
      "epoch": 3.3977508544921875e-05,
      "step": 8907,
      "training_step_time": 10.33455777168274
    },
    {
      "epoch": 3.39813232421875e-05,
      "model_forward_time": 3.1581263542175293,
      "step": 8908
    },
    {
      "epoch": 3.39813232421875e-05,
      "step": 8908,
      "training_step_time": 9.938663959503174
    },
    {
      "epoch": 3.39813232421875e-05,
      "model_forward_time": 3.1444013118743896,
      "step": 8908
    },
    {
      "epoch": 3.39813232421875e-05,
      "step": 8908,
      "training_step_time": 9.918861627578735
    },
    {
      "epoch": 3.39813232421875e-05,
      "model_forward_time": 3.133716583251953,
      "step": 8908
    },
    {
      "epoch": 3.39813232421875e-05,
      "step": 8908,
      "training_step_time": 9.911147117614746
    },
    {
      "epoch": 3.39813232421875e-05,
      "model_forward_time": 3.1360487937927246,
      "step": 8908
    },
    {
      "epoch": 3.39813232421875e-05,
      "step": 8908,
      "training_step_time": 10.014468431472778
    },
    {
      "epoch": 3.398513793945313e-05,
      "model_forward_time": 3.459268569946289,
      "step": 8909
    },
    {
      "epoch": 3.398513793945313e-05,
      "step": 8909,
      "training_step_time": 10.233861207962036
    },
    {
      "epoch": 3.398513793945313e-05,
      "model_forward_time": 2.428302526473999,
      "step": 8909
    },
    {
      "epoch": 3.398513793945313e-05,
      "step": 8909,
      "training_step_time": 7.462896823883057
    },
    {
      "epoch": 3.398513793945313e-05,
      "model_forward_time": 3.130169153213501,
      "step": 8909
    },
    {
      "epoch": 3.398513793945313e-05,
      "step": 8909,
      "training_step_time": 9.901540517807007
    },
    {
      "epoch": 3.398513793945313e-05,
      "model_forward_time": 3.1329355239868164,
      "step": 8909
    },
    {
      "epoch": 3.398513793945313e-05,
      "step": 8909,
      "training_step_time": 10.016812562942505
    },
    {
      "epoch": 3.398895263671875e-05,
      "grad_norm": 0.4146392047405243,
      "learning_rate": 1e-05,
      "loss": 0.0773,
      "step": 8910
    },
    {
      "epoch": 3.398895263671875e-05,
      "model_forward_time": 3.1447222232818604,
      "step": 8910
    },
    {
      "dynamics_loss_avg": 0.05821361280977726,
      "epoch": 3.398895263671875e-05,
      "step": 8910
    },
    {
      "action_loss_avg": 0.019129766384139657,
      "epoch": 3.398895263671875e-05,
      "step": 8910
    },
    {
      "epoch": 3.398895263671875e-05,
      "step": 8910,
      "training_step_time": 9.921638488769531
    },
    {
      "epoch": 3.398895263671875e-05,
      "model_forward_time": 2.43100905418396,
      "step": 8910
    },
    {
      "epoch": 3.398895263671875e-05,
      "step": 8910,
      "training_step_time": 7.464729070663452
    },
    {
      "epoch": 3.398895263671875e-05,
      "model_forward_time": 3.4124579429626465,
      "step": 8910
    },
    {
      "epoch": 3.398895263671875e-05,
      "step": 8910,
      "training_step_time": 10.1915602684021
    },
    {
      "epoch": 3.398895263671875e-05,
      "model_forward_time": 3.144371509552002,
      "step": 8910
    },
    {
      "epoch": 3.398895263671875e-05,
      "step": 8910,
      "training_step_time": 10.322672843933105
    },
    {
      "epoch": 3.3992767333984373e-05,
      "model_forward_time": 3.1638102531433105,
      "step": 8911
    },
    {
      "epoch": 3.3992767333984373e-05,
      "step": 8911,
      "training_step_time": 9.948895931243896
    },
    {
      "epoch": 3.3992767333984373e-05,
      "model_forward_time": 3.141336441040039,
      "step": 8911
    },
    {
      "epoch": 3.3992767333984373e-05,
      "step": 8911,
      "training_step_time": 9.920037984848022
    },
    {
      "epoch": 3.3992767333984373e-05,
      "model_forward_time": 3.1386680603027344,
      "step": 8911
    },
    {
      "epoch": 3.3992767333984373e-05,
      "step": 8911,
      "training_step_time": 9.916789054870605
    },
    {
      "epoch": 3.3992767333984373e-05,
      "model_forward_time": 3.4085769653320312,
      "step": 8911
    },
    {
      "epoch": 3.3992767333984373e-05,
      "step": 8911,
      "training_step_time": 10.285106897354126
    },
    {
      "epoch": 3.399658203125e-05,
      "model_forward_time": 3.1599762439727783,
      "step": 8912
    },
    {
      "epoch": 3.399658203125e-05,
      "step": 8912,
      "training_step_time": 9.943990230560303
    },
    {
      "epoch": 3.399658203125e-05,
      "model_forward_time": 3.1433558464050293,
      "step": 8912
    },
    {
      "epoch": 3.399658203125e-05,
      "step": 8912,
      "training_step_time": 9.927133798599243
    },
    {
      "epoch": 3.399658203125e-05,
      "model_forward_time": 3.1545827388763428,
      "step": 8912
    },
    {
      "dynamics_loss_avg": 0.06811212468892336,
      "epoch": 3.399658203125e-05,
      "step": 8912
    },
    {
      "action_loss_avg": 0.02215414955280721,
      "epoch": 3.399658203125e-05,
      "step": 8912
    },
    {
      "epoch": 3.399658203125e-05,
      "step": 8912,
      "training_step_time": 9.939882516860962
    },
    {
      "epoch": 3.399658203125e-05,
      "model_forward_time": 3.1473987102508545,
      "step": 8912
    },
    {
      "epoch": 3.399658203125e-05,
      "step": 8912,
      "training_step_time": 10.297127723693848
    },
    {
      "epoch": 3.4000396728515626e-05,
      "model_forward_time": 2.454307794570923,
      "step": 8913
    },
    {
      "epoch": 3.4000396728515626e-05,
      "step": 8913,
      "training_step_time": 7.487514972686768
    },
    {
      "epoch": 3.4000396728515626e-05,
      "model_forward_time": 3.1409056186676025,
      "step": 8913
    },
    {
      "epoch": 3.4000396728515626e-05,
      "step": 8913,
      "training_step_time": 9.912119388580322
    },
    {
      "epoch": 3.4000396728515626e-05,
      "model_forward_time": 3.140690803527832,
      "step": 8913
    },
    {
      "epoch": 3.4000396728515626e-05,
      "step": 8913,
      "training_step_time": 9.912769079208374
    },
    {
      "epoch": 3.4000396728515626e-05,
      "model_forward_time": 3.141141176223755,
      "step": 8913
    },
    {
      "epoch": 3.4000396728515626e-05,
      "step": 8913,
      "training_step_time": 10.015607357025146
    },
    {
      "epoch": 3.400421142578125e-05,
      "model_forward_time": 2.455984592437744,
      "step": 8914
    },
    {
      "epoch": 3.400421142578125e-05,
      "step": 8914,
      "training_step_time": 7.489318132400513
    },
    {
      "epoch": 3.400421142578125e-05,
      "model_forward_time": 3.4596025943756104,
      "step": 8914
    },
    {
      "epoch": 3.400421142578125e-05,
      "step": 8914,
      "training_step_time": 10.234002590179443
    },
    {
      "epoch": 3.400421142578125e-05,
      "model_forward_time": 3.137763023376465,
      "step": 8914
    },
    {
      "epoch": 3.400421142578125e-05,
      "step": 8914,
      "training_step_time": 9.915659666061401
    },
    {
      "epoch": 3.400421142578125e-05,
      "model_forward_time": 3.1412320137023926,
      "step": 8914
    },
    {
      "epoch": 3.400421142578125e-05,
      "step": 8914,
      "training_step_time": 10.024397134780884
    },
    {
      "epoch": 3.400802612304687e-05,
      "model_forward_time": 3.171658992767334,
      "step": 8915
    },
    {
      "dynamics_loss_avg": 0.0515576989389956,
      "epoch": 3.400802612304687e-05,
      "step": 8915
    },
    {
      "action_loss_avg": 0.015387575211934745,
      "epoch": 3.400802612304687e-05,
      "step": 8915
    },
    {
      "epoch": 3.400802612304687e-05,
      "step": 8915,
      "training_step_time": 9.964084148406982
    },
    {
      "epoch": 3.400802612304687e-05,
      "model_forward_time": 3.1380646228790283,
      "step": 8915
    },
    {
      "epoch": 3.400802612304687e-05,
      "step": 8915,
      "training_step_time": 10.190434455871582
    },
    {
      "epoch": 3.400802612304687e-05,
      "model_forward_time": 2.4308502674102783,
      "step": 8915
    },
    {
      "epoch": 3.400802612304687e-05,
      "step": 8915,
      "training_step_time": 7.463442325592041
    },
    {
      "epoch": 3.400802612304687e-05,
      "model_forward_time": 3.1307342052459717,
      "step": 8915
    },
    {
      "epoch": 3.400802612304687e-05,
      "step": 8915,
      "training_step_time": 10.322304487228394
    },
    {
      "epoch": 3.40118408203125e-05,
      "model_forward_time": 3.142679214477539,
      "step": 8916
    },
    {
      "epoch": 3.40118408203125e-05,
      "step": 8916,
      "training_step_time": 9.92335319519043
    },
    {
      "epoch": 3.40118408203125e-05,
      "model_forward_time": 2.4318766593933105,
      "step": 8916
    },
    {
      "epoch": 3.40118408203125e-05,
      "step": 8916,
      "training_step_time": 7.465173721313477
    },
    {
      "epoch": 3.40118408203125e-05,
      "model_forward_time": 3.131166458129883,
      "step": 8916
    },
    {
      "epoch": 3.40118408203125e-05,
      "step": 8916,
      "training_step_time": 9.909315586090088
    },
    {
      "epoch": 3.40118408203125e-05,
      "model_forward_time": 3.410111665725708,
      "step": 8916
    },
    {
      "epoch": 3.40118408203125e-05,
      "step": 8916,
      "training_step_time": 10.280500888824463
    },
    {
      "epoch": 3.4015655517578124e-05,
      "model_forward_time": 3.151221513748169,
      "step": 8917
    },
    {
      "epoch": 3.4015655517578124e-05,
      "step": 8917,
      "training_step_time": 9.929438352584839
    },
    {
      "epoch": 3.4015655517578124e-05,
      "model_forward_time": 3.135101318359375,
      "step": 8917
    },
    {
      "epoch": 3.4015655517578124e-05,
      "step": 8917,
      "training_step_time": 9.906965970993042
    },
    {
      "epoch": 3.4015655517578124e-05,
      "model_forward_time": 3.141108989715576,
      "step": 8917
    },
    {
      "dynamics_loss_avg": 0.05998460426926613,
      "epoch": 3.4015655517578124e-05,
      "step": 8917
    },
    {
      "action_loss_avg": 0.019597051478922368,
      "epoch": 3.4015655517578124e-05,
      "step": 8917
    },
    {
      "epoch": 3.4015655517578124e-05,
      "step": 8917,
      "training_step_time": 9.917448997497559
    },
    {
      "epoch": 3.4015655517578124e-05,
      "model_forward_time": 3.1352694034576416,
      "step": 8917
    },
    {
      "epoch": 3.4015655517578124e-05,
      "step": 8917,
      "training_step_time": 10.280974388122559
    },
    {
      "epoch": 3.401947021484375e-05,
      "model_forward_time": 3.148184061050415,
      "step": 8918
    },
    {
      "epoch": 3.401947021484375e-05,
      "step": 8918,
      "training_step_time": 9.932727336883545
    },
    {
      "epoch": 3.401947021484375e-05,
      "model_forward_time": 3.130615711212158,
      "step": 8918
    },
    {
      "epoch": 3.401947021484375e-05,
      "step": 8918,
      "training_step_time": 9.90600323677063
    },
    {
      "epoch": 3.401947021484375e-05,
      "model_forward_time": 3.1327426433563232,
      "step": 8918
    },
    {
      "epoch": 3.401947021484375e-05,
      "step": 8918,
      "training_step_time": 9.909077644348145
    },
    {
      "epoch": 3.401947021484375e-05,
      "model_forward_time": 3.1434428691864014,
      "step": 8918
    },
    {
      "epoch": 3.401947021484375e-05,
      "step": 8918,
      "training_step_time": 10.276128768920898
    },
    {
      "epoch": 3.402328491210938e-05,
      "model_forward_time": 2.4449970722198486,
      "step": 8919
    },
    {
      "epoch": 3.402328491210938e-05,
      "step": 8919,
      "training_step_time": 7.780290126800537
    },
    {
      "epoch": 3.402328491210938e-05,
      "model_forward_time": 3.126394271850586,
      "step": 8919
    },
    {
      "epoch": 3.402328491210938e-05,
      "step": 8919,
      "training_step_time": 9.896425485610962
    },
    {
      "epoch": 3.402328491210938e-05,
      "model_forward_time": 3.129504680633545,
      "step": 8919
    },
    {
      "epoch": 3.402328491210938e-05,
      "step": 8919,
      "training_step_time": 9.908711910247803
    },
    {
      "epoch": 3.402328491210938e-05,
      "model_forward_time": 3.13305926322937,
      "step": 8919
    },
    {
      "epoch": 3.402328491210938e-05,
      "step": 8919,
      "training_step_time": 10.020859718322754
    },
    {
      "epoch": 3.4027099609375e-05,
      "grad_norm": 0.5104550719261169,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 8920
    },
    {
      "epoch": 3.4027099609375e-05,
      "model_forward_time": 3.1555585861206055,
      "step": 8920
    },
    {
      "dynamics_loss_avg": 0.06791609115898609,
      "epoch": 3.4027099609375e-05,
      "step": 8920
    },
    {
      "action_loss_avg": 0.026280301716178657,
      "epoch": 3.4027099609375e-05,
      "step": 8920
    },
    {
      "epoch": 3.4027099609375e-05,
      "step": 8920,
      "training_step_time": 9.930051803588867
    },
    {
      "epoch": 3.4027099609375e-05,
      "model_forward_time": 3.405945301055908,
      "step": 8920
    },
    {
      "epoch": 3.4027099609375e-05,
      "step": 8920,
      "training_step_time": 10.182551145553589
    },
    {
      "epoch": 3.4027099609375e-05,
      "model_forward_time": 2.437788724899292,
      "step": 8920
    },
    {
      "epoch": 3.4027099609375e-05,
      "step": 8920,
      "training_step_time": 7.470848560333252
    },
    {
      "epoch": 3.4027099609375e-05,
      "model_forward_time": 3.1319968700408936,
      "step": 8920
    },
    {
      "epoch": 3.4027099609375e-05,
      "step": 8920,
      "training_step_time": 10.2847740650177
    },
    {
      "epoch": 3.403091430664062e-05,
      "model_forward_time": 3.161482334136963,
      "step": 8921
    },
    {
      "epoch": 3.403091430664062e-05,
      "step": 8921,
      "training_step_time": 9.94257664680481
    },
    {
      "epoch": 3.403091430664062e-05,
      "model_forward_time": 3.145296096801758,
      "step": 8921
    },
    {
      "epoch": 3.403091430664062e-05,
      "step": 8921,
      "training_step_time": 9.921098232269287
    },
    {
      "epoch": 3.403091430664062e-05,
      "model_forward_time": 3.144258975982666,
      "step": 8921
    },
    {
      "epoch": 3.403091430664062e-05,
      "step": 8921,
      "training_step_time": 9.916600942611694
    },
    {
      "epoch": 3.403091430664062e-05,
      "model_forward_time": 3.391598701477051,
      "step": 8921
    },
    {
      "epoch": 3.403091430664062e-05,
      "step": 8921,
      "training_step_time": 10.321064710617065
    },
    {
      "epoch": 3.403472900390625e-05,
      "model_forward_time": 3.1591854095458984,
      "step": 8922
    },
    {
      "epoch": 3.403472900390625e-05,
      "step": 8922,
      "training_step_time": 9.934545516967773
    },
    {
      "epoch": 3.403472900390625e-05,
      "model_forward_time": 3.142317533493042,
      "step": 8922
    },
    {
      "epoch": 3.403472900390625e-05,
      "step": 8922,
      "training_step_time": 9.914826154708862
    },
    {
      "epoch": 3.403472900390625e-05,
      "model_forward_time": 3.14365553855896,
      "step": 8922
    },
    {
      "dynamics_loss_avg": 0.06145025659352541,
      "epoch": 3.403472900390625e-05,
      "step": 8922
    },
    {
      "action_loss_avg": 0.023053720500320195,
      "epoch": 3.403472900390625e-05,
      "step": 8922
    },
    {
      "epoch": 3.403472900390625e-05,
      "step": 8922,
      "training_step_time": 9.919216394424438
    },
    {
      "epoch": 3.403472900390625e-05,
      "model_forward_time": 3.409905433654785,
      "step": 8922
    },
    {
      "epoch": 3.403472900390625e-05,
      "step": 8922,
      "training_step_time": 10.277501106262207
    },
    {
      "epoch": 3.4038543701171875e-05,
      "model_forward_time": 3.1604137420654297,
      "step": 8923
    },
    {
      "epoch": 3.4038543701171875e-05,
      "step": 8923,
      "training_step_time": 9.93696641921997
    },
    {
      "epoch": 3.4038543701171875e-05,
      "model_forward_time": 3.147641897201538,
      "step": 8923
    },
    {
      "epoch": 3.4038543701171875e-05,
      "step": 8923,
      "training_step_time": 9.922153234481812
    },
    {
      "epoch": 3.4038543701171875e-05,
      "model_forward_time": 2.4381327629089355,
      "step": 8923
    },
    {
      "epoch": 3.4038543701171875e-05,
      "step": 8923,
      "training_step_time": 7.4693763256073
    },
    {
      "epoch": 3.4038543701171875e-05,
      "model_forward_time": 3.14841890335083,
      "step": 8923
    },
    {
      "epoch": 3.4038543701171875e-05,
      "step": 8923,
      "training_step_time": 10.306641578674316
    },
    {
      "epoch": 3.40423583984375e-05,
      "model_forward_time": 3.154221534729004,
      "step": 8924
    },
    {
      "epoch": 3.40423583984375e-05,
      "step": 8924,
      "training_step_time": 9.932319402694702
    },
    {
      "epoch": 3.40423583984375e-05,
      "model_forward_time": 3.410064220428467,
      "step": 8924
    },
    {
      "epoch": 3.40423583984375e-05,
      "step": 8924,
      "training_step_time": 10.184974908828735
    },
    {
      "epoch": 3.40423583984375e-05,
      "model_forward_time": 3.140073537826538,
      "step": 8924
    },
    {
      "epoch": 3.40423583984375e-05,
      "step": 8924,
      "training_step_time": 9.915013790130615
    },
    {
      "epoch": 3.40423583984375e-05,
      "model_forward_time": 3.1411705017089844,
      "step": 8924
    },
    {
      "epoch": 3.40423583984375e-05,
      "step": 8924,
      "training_step_time": 10.037172794342041
    },
    {
      "epoch": 3.404617309570313e-05,
      "model_forward_time": 3.1543688774108887,
      "step": 8925
    },
    {
      "dynamics_loss_avg": 0.06732429694384337,
      "epoch": 3.404617309570313e-05,
      "step": 8925
    },
    {
      "action_loss_avg": 0.022244738414883614,
      "epoch": 3.404617309570313e-05,
      "step": 8925
    },
    {
      "epoch": 3.404617309570313e-05,
      "step": 8925,
      "training_step_time": 9.94326400756836
    },
    {
      "epoch": 3.404617309570313e-05,
      "model_forward_time": 3.424466371536255,
      "step": 8925
    },
    {
      "epoch": 3.404617309570313e-05,
      "step": 8925,
      "training_step_time": 10.202176570892334
    },
    {
      "epoch": 3.404617309570313e-05,
      "model_forward_time": 3.143829584121704,
      "step": 8925
    },
    {
      "epoch": 3.404617309570313e-05,
      "step": 8925,
      "training_step_time": 9.925641536712646
    },
    {
      "epoch": 3.404617309570313e-05,
      "model_forward_time": 3.145242214202881,
      "step": 8925
    },
    {
      "epoch": 3.404617309570313e-05,
      "step": 8925,
      "training_step_time": 10.333868741989136
    },
    {
      "epoch": 3.404998779296875e-05,
      "model_forward_time": 3.153233289718628,
      "step": 8926
    },
    {
      "epoch": 3.404998779296875e-05,
      "step": 8926,
      "training_step_time": 9.939036846160889
    },
    {
      "epoch": 3.404998779296875e-05,
      "model_forward_time": 3.138805866241455,
      "step": 8926
    },
    {
      "epoch": 3.404998779296875e-05,
      "step": 8926,
      "training_step_time": 9.921094179153442
    },
    {
      "epoch": 3.404998779296875e-05,
      "model_forward_time": 3.446028470993042,
      "step": 8926
    },
    {
      "epoch": 3.404998779296875e-05,
      "step": 8926,
      "training_step_time": 10.223320484161377
    },
    {
      "epoch": 3.404998779296875e-05,
      "model_forward_time": 3.1419577598571777,
      "step": 8926
    },
    {
      "epoch": 3.404998779296875e-05,
      "step": 8926,
      "training_step_time": 10.336857557296753
    },
    {
      "epoch": 3.4053802490234373e-05,
      "model_forward_time": 3.157217502593994,
      "step": 8927
    },
    {
      "epoch": 3.4053802490234373e-05,
      "step": 8927,
      "training_step_time": 9.932091236114502
    },
    {
      "epoch": 3.4053802490234373e-05,
      "model_forward_time": 3.134147882461548,
      "step": 8927
    },
    {
      "epoch": 3.4053802490234373e-05,
      "step": 8927,
      "training_step_time": 9.9117591381073
    },
    {
      "epoch": 3.4053802490234373e-05,
      "model_forward_time": 3.1419875621795654,
      "step": 8927
    },
    {
      "dynamics_loss_avg": 0.06109337527304888,
      "epoch": 3.4053802490234373e-05,
      "step": 8927
    },
    {
      "action_loss_avg": 0.02146962145343423,
      "epoch": 3.4053802490234373e-05,
      "step": 8927
    },
    {
      "epoch": 3.4053802490234373e-05,
      "step": 8927,
      "training_step_time": 9.922854900360107
    },
    {
      "epoch": 3.4053802490234373e-05,
      "model_forward_time": 2.438412666320801,
      "step": 8927
    },
    {
      "epoch": 3.4053802490234373e-05,
      "step": 8927,
      "training_step_time": 10.256235599517822
    },
    {
      "epoch": 3.40576171875e-05,
      "model_forward_time": 3.166585683822632,
      "step": 8928
    },
    {
      "epoch": 3.40576171875e-05,
      "step": 8928,
      "training_step_time": 9.943182229995728
    },
    {
      "epoch": 3.40576171875e-05,
      "model_forward_time": 3.1372060775756836,
      "step": 8928
    },
    {
      "epoch": 3.40576171875e-05,
      "step": 8928,
      "training_step_time": 9.912363290786743
    },
    {
      "epoch": 3.40576171875e-05,
      "model_forward_time": 3.144479751586914,
      "step": 8928
    },
    {
      "epoch": 3.40576171875e-05,
      "step": 8928,
      "training_step_time": 9.922260999679565
    },
    {
      "epoch": 3.40576171875e-05,
      "model_forward_time": 3.1435649394989014,
      "step": 8928
    },
    {
      "epoch": 3.40576171875e-05,
      "step": 8928,
      "training_step_time": 10.277082920074463
    },
    {
      "epoch": 3.4061431884765626e-05,
      "model_forward_time": 3.1493966579437256,
      "step": 8929
    },
    {
      "epoch": 3.4061431884765626e-05,
      "step": 8929,
      "training_step_time": 10.198131322860718
    },
    {
      "epoch": 3.4061431884765626e-05,
      "model_forward_time": 3.1385505199432373,
      "step": 8929
    },
    {
      "epoch": 3.4061431884765626e-05,
      "step": 8929,
      "training_step_time": 9.914076089859009
    },
    {
      "epoch": 3.4061431884765626e-05,
      "model_forward_time": 3.1390304565429688,
      "step": 8929
    },
    {
      "epoch": 3.4061431884765626e-05,
      "step": 8929,
      "training_step_time": 9.906973600387573
    },
    {
      "epoch": 3.4061431884765626e-05,
      "model_forward_time": 3.1373424530029297,
      "step": 8929
    },
    {
      "epoch": 3.4061431884765626e-05,
      "step": 8929,
      "training_step_time": 10.246929168701172
    },
    {
      "epoch": 3.406524658203125e-05,
      "grad_norm": 0.392143189907074,
      "learning_rate": 1e-05,
      "loss": 0.082,
      "step": 8930
    },
    {
      "epoch": 3.406524658203125e-05,
      "model_forward_time": 3.1472253799438477,
      "step": 8930
    },
    {
      "dynamics_loss_avg": 0.06410506293177605,
      "epoch": 3.406524658203125e-05,
      "step": 8930
    },
    {
      "action_loss_avg": 0.022020407393574713,
      "epoch": 3.406524658203125e-05,
      "step": 8930
    },
    {
      "epoch": 3.406524658203125e-05,
      "step": 8930,
      "training_step_time": 9.923240661621094
    },
    {
      "epoch": 3.406524658203125e-05,
      "model_forward_time": 3.1378231048583984,
      "step": 8930
    },
    {
      "epoch": 3.406524658203125e-05,
      "step": 8930,
      "training_step_time": 10.198131084442139
    },
    {
      "epoch": 3.406524658203125e-05,
      "model_forward_time": 3.140596866607666,
      "step": 8930
    },
    {
      "epoch": 3.406524658203125e-05,
      "step": 8930,
      "training_step_time": 9.915526628494263
    },
    {
      "epoch": 3.406524658203125e-05,
      "model_forward_time": 3.1366920471191406,
      "step": 8930
    },
    {
      "epoch": 3.406524658203125e-05,
      "step": 8930,
      "training_step_time": 10.025261878967285
    },
    {
      "epoch": 3.406906127929687e-05,
      "model_forward_time": 3.1601603031158447,
      "step": 8931
    },
    {
      "epoch": 3.406906127929687e-05,
      "step": 8931,
      "training_step_time": 9.935598611831665
    },
    {
      "epoch": 3.406906127929687e-05,
      "model_forward_time": 3.1443891525268555,
      "step": 8931
    },
    {
      "epoch": 3.406906127929687e-05,
      "step": 8931,
      "training_step_time": 9.916090965270996
    },
    {
      "epoch": 3.406906127929687e-05,
      "model_forward_time": 3.41715931892395,
      "step": 8931
    },
    {
      "epoch": 3.406906127929687e-05,
      "step": 8931,
      "training_step_time": 10.188937187194824
    },
    {
      "epoch": 3.406906127929687e-05,
      "model_forward_time": 3.139219284057617,
      "step": 8931
    },
    {
      "epoch": 3.406906127929687e-05,
      "step": 8931,
      "training_step_time": 10.308310270309448
    },
    {
      "epoch": 3.40728759765625e-05,
      "model_forward_time": 3.155726194381714,
      "step": 8932
    },
    {
      "epoch": 3.40728759765625e-05,
      "step": 8932,
      "training_step_time": 9.937629222869873
    },
    {
      "epoch": 3.40728759765625e-05,
      "model_forward_time": 3.1446950435638428,
      "step": 8932
    },
    {
      "epoch": 3.40728759765625e-05,
      "step": 8932,
      "training_step_time": 9.921627044677734
    },
    {
      "epoch": 3.40728759765625e-05,
      "model_forward_time": 3.146397352218628,
      "step": 8932
    },
    {
      "dynamics_loss_avg": 0.056303145736455916,
      "epoch": 3.40728759765625e-05,
      "step": 8932
    },
    {
      "action_loss_avg": 0.014979958347976208,
      "epoch": 3.40728759765625e-05,
      "step": 8932
    },
    {
      "epoch": 3.40728759765625e-05,
      "step": 8932,
      "training_step_time": 9.927208662033081
    },
    {
      "epoch": 3.40728759765625e-05,
      "model_forward_time": 3.1478312015533447,
      "step": 8932
    },
    {
      "epoch": 3.40728759765625e-05,
      "step": 8932,
      "training_step_time": 10.294988632202148
    },
    {
      "epoch": 3.4076690673828124e-05,
      "model_forward_time": 3.1678519248962402,
      "step": 8933
    },
    {
      "epoch": 3.4076690673828124e-05,
      "step": 8933,
      "training_step_time": 9.95087480545044
    },
    {
      "epoch": 3.4076690673828124e-05,
      "model_forward_time": 3.140477180480957,
      "step": 8933
    },
    {
      "epoch": 3.4076690673828124e-05,
      "step": 8933,
      "training_step_time": 9.919279098510742
    },
    {
      "epoch": 3.4076690673828124e-05,
      "model_forward_time": 3.1429975032806396,
      "step": 8933
    },
    {
      "epoch": 3.4076690673828124e-05,
      "step": 8933,
      "training_step_time": 9.927668571472168
    },
    {
      "epoch": 3.4076690673828124e-05,
      "model_forward_time": 2.435885429382324,
      "step": 8933
    },
    {
      "epoch": 3.4076690673828124e-05,
      "step": 8933,
      "training_step_time": 10.002703666687012
    },
    {
      "epoch": 3.408050537109375e-05,
      "model_forward_time": 3.414608955383301,
      "step": 8934
    },
    {
      "epoch": 3.408050537109375e-05,
      "step": 8934,
      "training_step_time": 10.18965768814087
    },
    {
      "epoch": 3.408050537109375e-05,
      "model_forward_time": 3.1382956504821777,
      "step": 8934
    },
    {
      "epoch": 3.408050537109375e-05,
      "step": 8934,
      "training_step_time": 9.910694599151611
    },
    {
      "epoch": 3.408050537109375e-05,
      "model_forward_time": 3.133451223373413,
      "step": 8934
    },
    {
      "epoch": 3.408050537109375e-05,
      "step": 8934,
      "training_step_time": 9.90213418006897
    },
    {
      "epoch": 3.408050537109375e-05,
      "model_forward_time": 2.4326157569885254,
      "step": 8934
    },
    {
      "epoch": 3.408050537109375e-05,
      "step": 8934,
      "training_step_time": 10.281850576400757
    },
    {
      "epoch": 3.408432006835938e-05,
      "model_forward_time": 3.15187406539917,
      "step": 8935
    },
    {
      "dynamics_loss_avg": 0.05892867697402835,
      "epoch": 3.408432006835938e-05,
      "step": 8935
    },
    {
      "action_loss_avg": 0.01799876014702022,
      "epoch": 3.408432006835938e-05,
      "step": 8935
    },
    {
      "epoch": 3.408432006835938e-05,
      "step": 8935,
      "training_step_time": 9.92697024345398
    },
    {
      "epoch": 3.408432006835938e-05,
      "model_forward_time": 3.1312944889068604,
      "step": 8935
    },
    {
      "epoch": 3.408432006835938e-05,
      "step": 8935,
      "training_step_time": 9.901064157485962
    },
    {
      "epoch": 3.408432006835938e-05,
      "model_forward_time": 2.6961965560913086,
      "step": 8935
    },
    {
      "epoch": 3.408432006835938e-05,
      "step": 8935,
      "training_step_time": 7.724052429199219
    },
    {
      "epoch": 3.408432006835938e-05,
      "model_forward_time": 3.131971836090088,
      "step": 8935
    },
    {
      "epoch": 3.408432006835938e-05,
      "step": 8935,
      "training_step_time": 10.024885416030884
    },
    {
      "epoch": 3.4088134765625e-05,
      "model_forward_time": 3.1545846462249756,
      "step": 8936
    },
    {
      "epoch": 3.4088134765625e-05,
      "step": 8936,
      "training_step_time": 9.925282955169678
    },
    {
      "epoch": 3.4088134765625e-05,
      "model_forward_time": 3.134678363800049,
      "step": 8936
    },
    {
      "epoch": 3.4088134765625e-05,
      "step": 8936,
      "training_step_time": 9.906489133834839
    },
    {
      "epoch": 3.4088134765625e-05,
      "model_forward_time": 3.1398777961730957,
      "step": 8936
    },
    {
      "epoch": 3.4088134765625e-05,
      "step": 8936,
      "training_step_time": 10.187394142150879
    },
    {
      "epoch": 3.4088134765625e-05,
      "model_forward_time": 3.141482353210449,
      "step": 8936
    },
    {
      "epoch": 3.4088134765625e-05,
      "step": 8936,
      "training_step_time": 10.30826210975647
    },
    {
      "epoch": 3.409194946289062e-05,
      "model_forward_time": 3.1514647006988525,
      "step": 8937
    },
    {
      "epoch": 3.409194946289062e-05,
      "step": 8937,
      "training_step_time": 9.925508499145508
    },
    {
      "epoch": 3.409194946289062e-05,
      "model_forward_time": 3.1374576091766357,
      "step": 8937
    },
    {
      "epoch": 3.409194946289062e-05,
      "step": 8937,
      "training_step_time": 9.912473201751709
    },
    {
      "epoch": 3.409194946289062e-05,
      "model_forward_time": 3.1393768787384033,
      "step": 8937
    },
    {
      "dynamics_loss_avg": 0.05122757349163294,
      "epoch": 3.409194946289062e-05,
      "step": 8937
    },
    {
      "action_loss_avg": 0.013820454897359013,
      "epoch": 3.409194946289062e-05,
      "step": 8937
    },
    {
      "epoch": 3.409194946289062e-05,
      "step": 8937,
      "training_step_time": 9.919060230255127
    },
    {
      "epoch": 3.409194946289062e-05,
      "model_forward_time": 3.136284112930298,
      "step": 8937
    },
    {
      "epoch": 3.409194946289062e-05,
      "step": 8937,
      "training_step_time": 10.28066873550415
    },
    {
      "epoch": 3.409576416015625e-05,
      "model_forward_time": 3.1489949226379395,
      "step": 8938
    },
    {
      "epoch": 3.409576416015625e-05,
      "step": 8938,
      "training_step_time": 9.921718835830688
    },
    {
      "epoch": 3.409576416015625e-05,
      "model_forward_time": 3.1359660625457764,
      "step": 8938
    },
    {
      "epoch": 3.409576416015625e-05,
      "step": 8938,
      "training_step_time": 9.905396699905396
    },
    {
      "epoch": 3.409576416015625e-05,
      "model_forward_time": 3.1378164291381836,
      "step": 8938
    },
    {
      "epoch": 3.409576416015625e-05,
      "step": 8938,
      "training_step_time": 9.903869390487671
    },
    {
      "epoch": 3.409576416015625e-05,
      "model_forward_time": 2.4299628734588623,
      "step": 8938
    },
    {
      "epoch": 3.409576416015625e-05,
      "step": 8938,
      "training_step_time": 10.259308099746704
    },
    {
      "epoch": 3.4099578857421875e-05,
      "model_forward_time": 3.1428275108337402,
      "step": 8939
    },
    {
      "epoch": 3.4099578857421875e-05,
      "step": 8939,
      "training_step_time": 10.223285675048828
    },
    {
      "epoch": 3.4099578857421875e-05,
      "model_forward_time": 3.1343300342559814,
      "step": 8939
    },
    {
      "epoch": 3.4099578857421875e-05,
      "step": 8939,
      "training_step_time": 9.898700952529907
    },
    {
      "epoch": 3.4099578857421875e-05,
      "model_forward_time": 3.138186454772949,
      "step": 8939
    },
    {
      "epoch": 3.4099578857421875e-05,
      "step": 8939,
      "training_step_time": 9.906697988510132
    },
    {
      "epoch": 3.4099578857421875e-05,
      "model_forward_time": 2.4375967979431152,
      "step": 8939
    },
    {
      "epoch": 3.4099578857421875e-05,
      "step": 8939,
      "training_step_time": 10.235706329345703
    },
    {
      "epoch": 3.41033935546875e-05,
      "grad_norm": 0.3139307200908661,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 8940
    },
    {
      "epoch": 3.41033935546875e-05,
      "model_forward_time": 3.1554558277130127,
      "step": 8940
    },
    {
      "dynamics_loss_avg": 0.06111018192023039,
      "epoch": 3.41033935546875e-05,
      "step": 8940
    },
    {
      "action_loss_avg": 0.019708540150895716,
      "epoch": 3.41033935546875e-05,
      "step": 8940
    },
    {
      "epoch": 3.41033935546875e-05,
      "step": 8940,
      "training_step_time": 9.927506685256958
    },
    {
      "epoch": 3.41033935546875e-05,
      "model_forward_time": 3.1383469104766846,
      "step": 8940
    },
    {
      "epoch": 3.41033935546875e-05,
      "step": 8940,
      "training_step_time": 9.904913187026978
    },
    {
      "epoch": 3.41033935546875e-05,
      "model_forward_time": 2.7069311141967773,
      "step": 8940
    },
    {
      "epoch": 3.41033935546875e-05,
      "step": 8940,
      "training_step_time": 7.729436635971069
    },
    {
      "epoch": 3.41033935546875e-05,
      "model_forward_time": 3.136009454727173,
      "step": 8940
    },
    {
      "epoch": 3.41033935546875e-05,
      "step": 8940,
      "training_step_time": 10.269243955612183
    },
    {
      "epoch": 3.410720825195313e-05,
      "model_forward_time": 2.4457156658172607,
      "step": 8941
    },
    {
      "epoch": 3.410720825195313e-05,
      "step": 8941,
      "training_step_time": 7.476595640182495
    },
    {
      "epoch": 3.410720825195313e-05,
      "model_forward_time": 3.136967420578003,
      "step": 8941
    },
    {
      "epoch": 3.410720825195313e-05,
      "step": 8941,
      "training_step_time": 9.911611795425415
    },
    {
      "epoch": 3.410720825195313e-05,
      "model_forward_time": 3.1424953937530518,
      "step": 8941
    },
    {
      "epoch": 3.410720825195313e-05,
      "step": 8941,
      "training_step_time": 9.916269779205322
    },
    {
      "epoch": 3.410720825195313e-05,
      "model_forward_time": 3.4003801345825195,
      "step": 8941
    },
    {
      "epoch": 3.410720825195313e-05,
      "step": 8941,
      "training_step_time": 10.272810697555542
    },
    {
      "epoch": 3.411102294921875e-05,
      "model_forward_time": 3.1673028469085693,
      "step": 8942
    },
    {
      "epoch": 3.411102294921875e-05,
      "step": 8942,
      "training_step_time": 9.947128534317017
    },
    {
      "epoch": 3.411102294921875e-05,
      "model_forward_time": 3.142768144607544,
      "step": 8942
    },
    {
      "epoch": 3.411102294921875e-05,
      "step": 8942,
      "training_step_time": 9.917776584625244
    },
    {
      "epoch": 3.411102294921875e-05,
      "model_forward_time": 3.141087532043457,
      "step": 8942
    },
    {
      "dynamics_loss_avg": 0.05414221696555614,
      "epoch": 3.411102294921875e-05,
      "step": 8942
    },
    {
      "action_loss_avg": 0.016041721403598785,
      "epoch": 3.411102294921875e-05,
      "step": 8942
    },
    {
      "epoch": 3.411102294921875e-05,
      "step": 8942,
      "training_step_time": 9.917726516723633
    },
    {
      "epoch": 3.411102294921875e-05,
      "model_forward_time": 2.4344918727874756,
      "step": 8942
    },
    {
      "epoch": 3.411102294921875e-05,
      "step": 8942,
      "training_step_time": 10.314829587936401
    },
    {
      "epoch": 3.4114837646484374e-05,
      "model_forward_time": 3.4338271617889404,
      "step": 8943
    },
    {
      "epoch": 3.4114837646484374e-05,
      "step": 8943,
      "training_step_time": 10.21064805984497
    },
    {
      "epoch": 3.4114837646484374e-05,
      "model_forward_time": 3.1466825008392334,
      "step": 8943
    },
    {
      "epoch": 3.4114837646484374e-05,
      "step": 8943,
      "training_step_time": 9.920320272445679
    },
    {
      "epoch": 3.4114837646484374e-05,
      "model_forward_time": 3.1381614208221436,
      "step": 8943
    },
    {
      "epoch": 3.4114837646484374e-05,
      "step": 8943,
      "training_step_time": 9.910506010055542
    },
    {
      "epoch": 3.4114837646484374e-05,
      "model_forward_time": 3.142793655395508,
      "step": 8943
    },
    {
      "epoch": 3.4114837646484374e-05,
      "step": 8943,
      "training_step_time": 10.30736255645752
    },
    {
      "epoch": 3.411865234375e-05,
      "model_forward_time": 3.179990291595459,
      "step": 8944
    },
    {
      "epoch": 3.411865234375e-05,
      "step": 8944,
      "training_step_time": 9.951206922531128
    },
    {
      "epoch": 3.411865234375e-05,
      "model_forward_time": 3.416121244430542,
      "step": 8944
    },
    {
      "epoch": 3.411865234375e-05,
      "step": 8944,
      "training_step_time": 10.184141874313354
    },
    {
      "epoch": 3.411865234375e-05,
      "model_forward_time": 2.439272403717041,
      "step": 8944
    },
    {
      "epoch": 3.411865234375e-05,
      "step": 8944,
      "training_step_time": 7.463366270065308
    },
    {
      "epoch": 3.411865234375e-05,
      "model_forward_time": 3.165088176727295,
      "step": 8944
    },
    {
      "epoch": 3.411865234375e-05,
      "step": 8944,
      "training_step_time": 10.028722524642944
    },
    {
      "epoch": 3.4122467041015626e-05,
      "model_forward_time": 3.1511740684509277,
      "step": 8945
    },
    {
      "dynamics_loss_avg": 0.06263708155602217,
      "epoch": 3.4122467041015626e-05,
      "step": 8945
    },
    {
      "action_loss_avg": 0.02073814608156681,
      "epoch": 3.4122467041015626e-05,
      "step": 8945
    },
    {
      "epoch": 3.4122467041015626e-05,
      "step": 8945,
      "training_step_time": 9.926338911056519
    },
    {
      "epoch": 3.4122467041015626e-05,
      "model_forward_time": 3.4131178855895996,
      "step": 8945
    },
    {
      "epoch": 3.4122467041015626e-05,
      "step": 8945,
      "training_step_time": 10.181593418121338
    },
    {
      "epoch": 3.4122467041015626e-05,
      "model_forward_time": 3.1365158557891846,
      "step": 8945
    },
    {
      "epoch": 3.4122467041015626e-05,
      "step": 8945,
      "training_step_time": 9.908766746520996
    },
    {
      "epoch": 3.4122467041015626e-05,
      "model_forward_time": 3.140927791595459,
      "step": 8945
    },
    {
      "epoch": 3.4122467041015626e-05,
      "step": 8945,
      "training_step_time": 10.279995203018188
    },
    {
      "epoch": 3.412628173828125e-05,
      "model_forward_time": 3.156989574432373,
      "step": 8946
    },
    {
      "epoch": 3.412628173828125e-05,
      "step": 8946,
      "training_step_time": 9.933341979980469
    },
    {
      "epoch": 3.412628173828125e-05,
      "model_forward_time": 3.1395938396453857,
      "step": 8946
    },
    {
      "epoch": 3.412628173828125e-05,
      "step": 8946,
      "training_step_time": 9.911916017532349
    },
    {
      "epoch": 3.412628173828125e-05,
      "model_forward_time": 3.1409666538238525,
      "step": 8946
    },
    {
      "epoch": 3.412628173828125e-05,
      "step": 8946,
      "training_step_time": 9.91492247581482
    },
    {
      "epoch": 3.412628173828125e-05,
      "model_forward_time": 3.4122867584228516,
      "step": 8946
    },
    {
      "epoch": 3.412628173828125e-05,
      "step": 8946,
      "training_step_time": 10.2888925075531
    },
    {
      "epoch": 3.413009643554687e-05,
      "model_forward_time": 3.150209665298462,
      "step": 8947
    },
    {
      "epoch": 3.413009643554687e-05,
      "step": 8947,
      "training_step_time": 9.928049087524414
    },
    {
      "epoch": 3.413009643554687e-05,
      "model_forward_time": 3.1386563777923584,
      "step": 8947
    },
    {
      "epoch": 3.413009643554687e-05,
      "step": 8947,
      "training_step_time": 9.91078233718872
    },
    {
      "epoch": 3.413009643554687e-05,
      "model_forward_time": 3.1385302543640137,
      "step": 8947
    },
    {
      "dynamics_loss_avg": 0.062002077139914034,
      "epoch": 3.413009643554687e-05,
      "step": 8947
    },
    {
      "action_loss_avg": 0.021702640037983657,
      "epoch": 3.413009643554687e-05,
      "step": 8947
    },
    {
      "epoch": 3.413009643554687e-05,
      "step": 8947,
      "training_step_time": 9.920589923858643
    },
    {
      "epoch": 3.413009643554687e-05,
      "model_forward_time": 3.4193668365478516,
      "step": 8947
    },
    {
      "epoch": 3.413009643554687e-05,
      "step": 8947,
      "training_step_time": 10.300586462020874
    },
    {
      "epoch": 3.41339111328125e-05,
      "model_forward_time": 3.153290271759033,
      "step": 8948
    },
    {
      "epoch": 3.41339111328125e-05,
      "step": 8948,
      "training_step_time": 9.93265438079834
    },
    {
      "epoch": 3.41339111328125e-05,
      "model_forward_time": 3.13649845123291,
      "step": 8948
    },
    {
      "epoch": 3.41339111328125e-05,
      "step": 8948,
      "training_step_time": 9.911640405654907
    },
    {
      "epoch": 3.41339111328125e-05,
      "model_forward_time": 3.1406726837158203,
      "step": 8948
    },
    {
      "epoch": 3.41339111328125e-05,
      "step": 8948,
      "training_step_time": 9.917052268981934
    },
    {
      "epoch": 3.41339111328125e-05,
      "model_forward_time": 3.1377158164978027,
      "step": 8948
    },
    {
      "epoch": 3.41339111328125e-05,
      "step": 8948,
      "training_step_time": 10.341084003448486
    },
    {
      "epoch": 3.4137725830078124e-05,
      "model_forward_time": 3.4226629734039307,
      "step": 8949
    },
    {
      "epoch": 3.4137725830078124e-05,
      "step": 8949,
      "training_step_time": 10.197387218475342
    },
    {
      "epoch": 3.4137725830078124e-05,
      "model_forward_time": 3.148458480834961,
      "step": 8949
    },
    {
      "epoch": 3.4137725830078124e-05,
      "step": 8949,
      "training_step_time": 9.929917335510254
    },
    {
      "epoch": 3.4137725830078124e-05,
      "model_forward_time": 3.1517045497894287,
      "step": 8949
    },
    {
      "epoch": 3.4137725830078124e-05,
      "step": 8949,
      "training_step_time": 9.933324337005615
    },
    {
      "epoch": 3.4137725830078124e-05,
      "model_forward_time": 2.437526226043701,
      "step": 8949
    },
    {
      "epoch": 3.4137725830078124e-05,
      "step": 8949,
      "training_step_time": 10.289687395095825
    },
    {
      "epoch": 3.414154052734375e-05,
      "grad_norm": 0.5402819514274597,
      "learning_rate": 1e-05,
      "loss": 0.0773,
      "step": 8950
    },
    {
      "epoch": 3.414154052734375e-05,
      "model_forward_time": 3.1633336544036865,
      "step": 8950
    },
    {
      "dynamics_loss_avg": 0.05859027877449989,
      "epoch": 3.414154052734375e-05,
      "step": 8950
    },
    {
      "action_loss_avg": 0.019311645161360504,
      "epoch": 3.414154052734375e-05,
      "step": 8950
    },
    {
      "epoch": 3.414154052734375e-05,
      "step": 8950,
      "training_step_time": 9.939449548721313
    },
    {
      "epoch": 3.414154052734375e-05,
      "model_forward_time": 3.4346554279327393,
      "step": 8950
    },
    {
      "epoch": 3.414154052734375e-05,
      "step": 8950,
      "training_step_time": 10.206724166870117
    },
    {
      "epoch": 3.414154052734375e-05,
      "model_forward_time": 3.1496036052703857,
      "step": 8950
    },
    {
      "epoch": 3.414154052734375e-05,
      "step": 8950,
      "training_step_time": 9.927309036254883
    },
    {
      "epoch": 3.414154052734375e-05,
      "model_forward_time": 3.145764112472534,
      "step": 8950
    },
    {
      "epoch": 3.414154052734375e-05,
      "step": 8950,
      "training_step_time": 10.259049892425537
    },
    {
      "epoch": 3.414535522460938e-05,
      "model_forward_time": 3.158597707748413,
      "step": 8951
    },
    {
      "epoch": 3.414535522460938e-05,
      "step": 8951,
      "training_step_time": 9.936041116714478
    },
    {
      "epoch": 3.414535522460938e-05,
      "model_forward_time": 3.1424973011016846,
      "step": 8951
    },
    {
      "epoch": 3.414535522460938e-05,
      "step": 8951,
      "training_step_time": 9.920247793197632
    },
    {
      "epoch": 3.414535522460938e-05,
      "model_forward_time": 3.440927028656006,
      "step": 8951
    },
    {
      "epoch": 3.414535522460938e-05,
      "step": 8951,
      "training_step_time": 10.217371463775635
    },
    {
      "epoch": 3.414535522460938e-05,
      "model_forward_time": 3.1478419303894043,
      "step": 8951
    },
    {
      "epoch": 3.414535522460938e-05,
      "step": 8951,
      "training_step_time": 10.322527408599854
    },
    {
      "epoch": 3.4149169921875e-05,
      "model_forward_time": 3.1581811904907227,
      "step": 8952
    },
    {
      "epoch": 3.4149169921875e-05,
      "step": 8952,
      "training_step_time": 9.938976526260376
    },
    {
      "epoch": 3.4149169921875e-05,
      "model_forward_time": 3.1447136402130127,
      "step": 8952
    },
    {
      "epoch": 3.4149169921875e-05,
      "step": 8952,
      "training_step_time": 9.924955368041992
    },
    {
      "epoch": 3.4149169921875e-05,
      "model_forward_time": 3.1453311443328857,
      "step": 8952
    },
    {
      "dynamics_loss_avg": 0.0636328773573041,
      "epoch": 3.4149169921875e-05,
      "step": 8952
    },
    {
      "action_loss_avg": 0.017357954010367393,
      "epoch": 3.4149169921875e-05,
      "step": 8952
    },
    {
      "epoch": 3.4149169921875e-05,
      "step": 8952,
      "training_step_time": 9.93090295791626
    },
    {
      "epoch": 3.4149169921875e-05,
      "model_forward_time": 3.402655839920044,
      "step": 8952
    },
    {
      "epoch": 3.4149169921875e-05,
      "step": 8952,
      "training_step_time": 10.327313661575317
    },
    {
      "epoch": 3.415298461914062e-05,
      "model_forward_time": 3.1514158248901367,
      "step": 8953
    },
    {
      "epoch": 3.415298461914062e-05,
      "step": 8953,
      "training_step_time": 9.933067083358765
    },
    {
      "epoch": 3.415298461914062e-05,
      "model_forward_time": 3.143678665161133,
      "step": 8953
    },
    {
      "epoch": 3.415298461914062e-05,
      "step": 8953,
      "training_step_time": 9.921216011047363
    },
    {
      "epoch": 3.415298461914062e-05,
      "model_forward_time": 3.143742561340332,
      "step": 8953
    },
    {
      "epoch": 3.415298461914062e-05,
      "step": 8953,
      "training_step_time": 9.920793771743774
    },
    {
      "epoch": 3.415298461914062e-05,
      "model_forward_time": 3.148817539215088,
      "step": 8953
    },
    {
      "epoch": 3.415298461914062e-05,
      "step": 8953,
      "training_step_time": 10.353125095367432
    },
    {
      "epoch": 3.415679931640625e-05,
      "model_forward_time": 3.4468307495117188,
      "step": 8954
    },
    {
      "epoch": 3.415679931640625e-05,
      "step": 8954,
      "training_step_time": 10.226587057113647
    },
    {
      "epoch": 3.415679931640625e-05,
      "model_forward_time": 3.148601531982422,
      "step": 8954
    },
    {
      "epoch": 3.415679931640625e-05,
      "step": 8954,
      "training_step_time": 9.932817220687866
    },
    {
      "epoch": 3.415679931640625e-05,
      "model_forward_time": 3.1379008293151855,
      "step": 8954
    },
    {
      "epoch": 3.415679931640625e-05,
      "step": 8954,
      "training_step_time": 9.915613412857056
    },
    {
      "epoch": 3.415679931640625e-05,
      "model_forward_time": 3.148009777069092,
      "step": 8954
    },
    {
      "epoch": 3.415679931640625e-05,
      "step": 8954,
      "training_step_time": 10.318159818649292
    },
    {
      "epoch": 3.4160614013671875e-05,
      "model_forward_time": 3.164801836013794,
      "step": 8955
    },
    {
      "dynamics_loss_avg": 0.06606265027076005,
      "epoch": 3.4160614013671875e-05,
      "step": 8955
    },
    {
      "action_loss_avg": 0.024560080654919148,
      "epoch": 3.4160614013671875e-05,
      "step": 8955
    },
    {
      "epoch": 3.4160614013671875e-05,
      "step": 8955,
      "training_step_time": 9.950178384780884
    },
    {
      "epoch": 3.4160614013671875e-05,
      "model_forward_time": 3.431016445159912,
      "step": 8955
    },
    {
      "epoch": 3.4160614013671875e-05,
      "step": 8955,
      "training_step_time": 10.205336570739746
    },
    {
      "epoch": 3.4160614013671875e-05,
      "model_forward_time": 3.144758701324463,
      "step": 8955
    },
    {
      "epoch": 3.4160614013671875e-05,
      "step": 8955,
      "training_step_time": 9.916749715805054
    },
    {
      "epoch": 3.4160614013671875e-05,
      "model_forward_time": 2.436613082885742,
      "step": 8955
    },
    {
      "epoch": 3.4160614013671875e-05,
      "step": 8955,
      "training_step_time": 10.01956582069397
    },
    {
      "epoch": 3.41644287109375e-05,
      "model_forward_time": 3.1575725078582764,
      "step": 8956
    },
    {
      "epoch": 3.41644287109375e-05,
      "step": 8956,
      "training_step_time": 9.940065383911133
    },
    {
      "epoch": 3.41644287109375e-05,
      "model_forward_time": 2.4344587326049805,
      "step": 8956
    },
    {
      "epoch": 3.41644287109375e-05,
      "step": 8956,
      "training_step_time": 7.460389137268066
    },
    {
      "epoch": 3.41644287109375e-05,
      "model_forward_time": 3.3917131423950195,
      "step": 8956
    },
    {
      "epoch": 3.41644287109375e-05,
      "step": 8956,
      "training_step_time": 10.155020952224731
    },
    {
      "epoch": 3.41644287109375e-05,
      "model_forward_time": 3.155113458633423,
      "step": 8956
    },
    {
      "epoch": 3.41644287109375e-05,
      "step": 8956,
      "training_step_time": 10.303744554519653
    },
    {
      "epoch": 3.416824340820313e-05,
      "model_forward_time": 3.1544511318206787,
      "step": 8957
    },
    {
      "epoch": 3.416824340820313e-05,
      "step": 8957,
      "training_step_time": 9.918707132339478
    },
    {
      "epoch": 3.416824340820313e-05,
      "model_forward_time": 3.1303021907806396,
      "step": 8957
    },
    {
      "epoch": 3.416824340820313e-05,
      "step": 8957,
      "training_step_time": 9.898378610610962
    },
    {
      "epoch": 3.416824340820313e-05,
      "model_forward_time": 3.142698049545288,
      "step": 8957
    },
    {
      "dynamics_loss_avg": 0.057367723807692526,
      "epoch": 3.416824340820313e-05,
      "step": 8957
    },
    {
      "action_loss_avg": 0.01862782393582165,
      "epoch": 3.416824340820313e-05,
      "step": 8957
    },
    {
      "epoch": 3.416824340820313e-05,
      "step": 8957,
      "training_step_time": 9.915342330932617
    },
    {
      "epoch": 3.416824340820313e-05,
      "model_forward_time": 3.136667490005493,
      "step": 8957
    },
    {
      "epoch": 3.416824340820313e-05,
      "step": 8957,
      "training_step_time": 10.313889026641846
    },
    {
      "epoch": 3.417205810546875e-05,
      "model_forward_time": 3.157236099243164,
      "step": 8958
    },
    {
      "epoch": 3.417205810546875e-05,
      "step": 8958,
      "training_step_time": 9.936454057693481
    },
    {
      "epoch": 3.417205810546875e-05,
      "model_forward_time": 3.1396429538726807,
      "step": 8958
    },
    {
      "epoch": 3.417205810546875e-05,
      "step": 8958,
      "training_step_time": 9.91402292251587
    },
    {
      "epoch": 3.417205810546875e-05,
      "model_forward_time": 3.143505096435547,
      "step": 8958
    },
    {
      "epoch": 3.417205810546875e-05,
      "step": 8958,
      "training_step_time": 9.921558141708374
    },
    {
      "epoch": 3.417205810546875e-05,
      "model_forward_time": 3.137979507446289,
      "step": 8958
    },
    {
      "epoch": 3.417205810546875e-05,
      "step": 8958,
      "training_step_time": 10.008570194244385
    },
    {
      "epoch": 3.4175872802734374e-05,
      "model_forward_time": 3.4264986515045166,
      "step": 8959
    },
    {
      "epoch": 3.4175872802734374e-05,
      "step": 8959,
      "training_step_time": 10.198007583618164
    },
    {
      "epoch": 3.4175872802734374e-05,
      "model_forward_time": 3.133225440979004,
      "step": 8959
    },
    {
      "epoch": 3.4175872802734374e-05,
      "step": 8959,
      "training_step_time": 9.89735198020935
    },
    {
      "epoch": 3.4175872802734374e-05,
      "model_forward_time": 3.140704870223999,
      "step": 8959
    },
    {
      "epoch": 3.4175872802734374e-05,
      "step": 8959,
      "training_step_time": 9.909384727478027
    },
    {
      "epoch": 3.4175872802734374e-05,
      "model_forward_time": 3.137246608734131,
      "step": 8959
    },
    {
      "epoch": 3.4175872802734374e-05,
      "step": 8959,
      "training_step_time": 10.239376068115234
    },
    {
      "epoch": 3.41796875e-05,
      "grad_norm": 0.3895495533943176,
      "learning_rate": 1e-05,
      "loss": 0.0804,
      "step": 8960
    },
    {
      "epoch": 3.41796875e-05,
      "model_forward_time": 3.1399784088134766,
      "step": 8960
    },
    {
      "dynamics_loss_avg": 0.05820121224969625,
      "epoch": 3.41796875e-05,
      "step": 8960
    },
    {
      "action_loss_avg": 0.019918595533818008,
      "epoch": 3.41796875e-05,
      "step": 8960
    },
    {
      "epoch": 3.41796875e-05,
      "step": 8960,
      "training_step_time": 9.913717985153198
    },
    {
      "epoch": 3.41796875e-05,
      "model_forward_time": 3.425750732421875,
      "step": 8960
    },
    {
      "epoch": 3.41796875e-05,
      "step": 8960,
      "training_step_time": 10.191049575805664
    },
    {
      "epoch": 3.41796875e-05,
      "model_forward_time": 3.13371205329895,
      "step": 8960
    },
    {
      "epoch": 3.41796875e-05,
      "step": 8960,
      "training_step_time": 9.904629707336426
    },
    {
      "epoch": 3.41796875e-05,
      "model_forward_time": 3.1582655906677246,
      "step": 8960
    },
    {
      "epoch": 3.41796875e-05,
      "step": 8960,
      "training_step_time": 10.024465560913086
    },
    {
      "epoch": 3.4183502197265626e-05,
      "model_forward_time": 3.1445000171661377,
      "step": 8961
    },
    {
      "epoch": 3.4183502197265626e-05,
      "step": 8961,
      "training_step_time": 9.921393156051636
    },
    {
      "epoch": 3.4183502197265626e-05,
      "model_forward_time": 3.1351847648620605,
      "step": 8961
    },
    {
      "epoch": 3.4183502197265626e-05,
      "step": 8961,
      "training_step_time": 9.91103720664978
    },
    {
      "epoch": 3.4183502197265626e-05,
      "model_forward_time": 3.4300696849823,
      "step": 8961
    },
    {
      "epoch": 3.4183502197265626e-05,
      "step": 8961,
      "training_step_time": 10.210285902023315
    },
    {
      "epoch": 3.4183502197265626e-05,
      "model_forward_time": 3.1382346153259277,
      "step": 8961
    },
    {
      "epoch": 3.4183502197265626e-05,
      "step": 8961,
      "training_step_time": 10.016950607299805
    },
    {
      "epoch": 3.418731689453125e-05,
      "model_forward_time": 3.151853561401367,
      "step": 8962
    },
    {
      "epoch": 3.418731689453125e-05,
      "step": 8962,
      "training_step_time": 9.92752456665039
    },
    {
      "epoch": 3.418731689453125e-05,
      "model_forward_time": 3.1364901065826416,
      "step": 8962
    },
    {
      "epoch": 3.418731689453125e-05,
      "step": 8962,
      "training_step_time": 9.913499593734741
    },
    {
      "epoch": 3.418731689453125e-05,
      "model_forward_time": 3.131406784057617,
      "step": 8962
    },
    {
      "dynamics_loss_avg": 0.05568865612149239,
      "epoch": 3.418731689453125e-05,
      "step": 8962
    },
    {
      "action_loss_avg": 0.016631679516285656,
      "epoch": 3.418731689453125e-05,
      "step": 8962
    },
    {
      "epoch": 3.418731689453125e-05,
      "step": 8962,
      "training_step_time": 9.907705307006836
    },
    {
      "epoch": 3.418731689453125e-05,
      "model_forward_time": 3.3935678005218506,
      "step": 8962
    },
    {
      "epoch": 3.418731689453125e-05,
      "step": 8962,
      "training_step_time": 10.264342069625854
    },
    {
      "epoch": 3.419113159179687e-05,
      "model_forward_time": 3.155285120010376,
      "step": 8963
    },
    {
      "epoch": 3.419113159179687e-05,
      "step": 8963,
      "training_step_time": 9.934146642684937
    },
    {
      "epoch": 3.419113159179687e-05,
      "model_forward_time": 3.137181520462036,
      "step": 8963
    },
    {
      "epoch": 3.419113159179687e-05,
      "step": 8963,
      "training_step_time": 9.915695428848267
    },
    {
      "epoch": 3.419113159179687e-05,
      "model_forward_time": 3.1323435306549072,
      "step": 8963
    },
    {
      "epoch": 3.419113159179687e-05,
      "step": 8963,
      "training_step_time": 9.909910440444946
    },
    {
      "epoch": 3.419113159179687e-05,
      "model_forward_time": 3.1370532512664795,
      "step": 8963
    },
    {
      "epoch": 3.419113159179687e-05,
      "step": 8963,
      "training_step_time": 10.265825748443604
    },
    {
      "epoch": 3.41949462890625e-05,
      "model_forward_time": 3.4194982051849365,
      "step": 8964
    },
    {
      "epoch": 3.41949462890625e-05,
      "step": 8964,
      "training_step_time": 10.194676399230957
    },
    {
      "epoch": 3.41949462890625e-05,
      "model_forward_time": 3.1386852264404297,
      "step": 8964
    },
    {
      "epoch": 3.41949462890625e-05,
      "step": 8964,
      "training_step_time": 9.91625714302063
    },
    {
      "epoch": 3.41949462890625e-05,
      "model_forward_time": 2.4332668781280518,
      "step": 8964
    },
    {
      "epoch": 3.41949462890625e-05,
      "step": 8964,
      "training_step_time": 7.46550726890564
    },
    {
      "epoch": 3.41949462890625e-05,
      "model_forward_time": 3.1338188648223877,
      "step": 8964
    },
    {
      "epoch": 3.41949462890625e-05,
      "step": 8964,
      "training_step_time": 10.005781173706055
    },
    {
      "epoch": 3.4198760986328124e-05,
      "model_forward_time": 3.1460843086242676,
      "step": 8965
    },
    {
      "dynamics_loss_avg": 0.048453297838568686,
      "epoch": 3.4198760986328124e-05,
      "step": 8965
    },
    {
      "action_loss_avg": 0.016160996351391076,
      "epoch": 3.4198760986328124e-05,
      "step": 8965
    },
    {
      "epoch": 3.4198760986328124e-05,
      "step": 8965,
      "training_step_time": 9.921429634094238
    },
    {
      "epoch": 3.4198760986328124e-05,
      "model_forward_time": 3.422269344329834,
      "step": 8965
    },
    {
      "epoch": 3.4198760986328124e-05,
      "step": 8965,
      "training_step_time": 10.196259021759033
    },
    {
      "epoch": 3.4198760986328124e-05,
      "model_forward_time": 3.1296615600585938,
      "step": 8965
    },
    {
      "epoch": 3.4198760986328124e-05,
      "step": 8965,
      "training_step_time": 9.906586408615112
    },
    {
      "epoch": 3.4198760986328124e-05,
      "model_forward_time": 3.13409161567688,
      "step": 8965
    },
    {
      "epoch": 3.4198760986328124e-05,
      "step": 8965,
      "training_step_time": 10.01203179359436
    },
    {
      "epoch": 3.420257568359375e-05,
      "model_forward_time": 3.149280309677124,
      "step": 8966
    },
    {
      "epoch": 3.420257568359375e-05,
      "step": 8966,
      "training_step_time": 9.92500901222229
    },
    {
      "epoch": 3.420257568359375e-05,
      "model_forward_time": 2.4311327934265137,
      "step": 8966
    },
    {
      "epoch": 3.420257568359375e-05,
      "step": 8966,
      "training_step_time": 7.463129043579102
    },
    {
      "epoch": 3.420257568359375e-05,
      "model_forward_time": 3.1338274478912354,
      "step": 8966
    },
    {
      "epoch": 3.420257568359375e-05,
      "step": 8966,
      "training_step_time": 9.914007902145386
    },
    {
      "epoch": 3.420257568359375e-05,
      "model_forward_time": 3.4083170890808105,
      "step": 8966
    },
    {
      "epoch": 3.420257568359375e-05,
      "step": 8966,
      "training_step_time": 10.285400152206421
    },
    {
      "epoch": 3.420639038085938e-05,
      "model_forward_time": 3.1467623710632324,
      "step": 8967
    },
    {
      "epoch": 3.420639038085938e-05,
      "step": 8967,
      "training_step_time": 9.944331645965576
    },
    {
      "epoch": 3.420639038085938e-05,
      "model_forward_time": 3.14713454246521,
      "step": 8967
    },
    {
      "epoch": 3.420639038085938e-05,
      "step": 8967,
      "training_step_time": 9.936919927597046
    },
    {
      "epoch": 3.420639038085938e-05,
      "model_forward_time": 3.1468212604522705,
      "step": 8967
    },
    {
      "dynamics_loss_avg": 0.05817187847569585,
      "epoch": 3.420639038085938e-05,
      "step": 8967
    },
    {
      "action_loss_avg": 0.01873470894061029,
      "epoch": 3.420639038085938e-05,
      "step": 8967
    },
    {
      "epoch": 3.420639038085938e-05,
      "step": 8967,
      "training_step_time": 9.940513134002686
    },
    {
      "epoch": 3.420639038085938e-05,
      "model_forward_time": 3.140554666519165,
      "step": 8967
    },
    {
      "epoch": 3.420639038085938e-05,
      "step": 8967,
      "training_step_time": 10.321489810943604
    },
    {
      "epoch": 3.4210205078125e-05,
      "model_forward_time": 3.1528306007385254,
      "step": 8968
    },
    {
      "epoch": 3.4210205078125e-05,
      "step": 8968,
      "training_step_time": 9.948963403701782
    },
    {
      "epoch": 3.4210205078125e-05,
      "model_forward_time": 3.1332836151123047,
      "step": 8968
    },
    {
      "epoch": 3.4210205078125e-05,
      "step": 8968,
      "training_step_time": 9.908870935440063
    },
    {
      "epoch": 3.4210205078125e-05,
      "model_forward_time": 3.132197141647339,
      "step": 8968
    },
    {
      "epoch": 3.4210205078125e-05,
      "step": 8968,
      "training_step_time": 9.907137632369995
    },
    {
      "epoch": 3.4210205078125e-05,
      "model_forward_time": 3.1316893100738525,
      "step": 8968
    },
    {
      "epoch": 3.4210205078125e-05,
      "step": 8968,
      "training_step_time": 10.32853651046753
    },
    {
      "epoch": 3.421401977539062e-05,
      "model_forward_time": 3.1396689414978027,
      "step": 8969
    },
    {
      "epoch": 3.421401977539062e-05,
      "step": 8969,
      "training_step_time": 10.225539922714233
    },
    {
      "epoch": 3.421401977539062e-05,
      "model_forward_time": 3.1293110847473145,
      "step": 8969
    },
    {
      "epoch": 3.421401977539062e-05,
      "step": 8969,
      "training_step_time": 9.898337364196777
    },
    {
      "epoch": 3.421401977539062e-05,
      "model_forward_time": 3.132791519165039,
      "step": 8969
    },
    {
      "epoch": 3.421401977539062e-05,
      "step": 8969,
      "training_step_time": 9.907397747039795
    },
    {
      "epoch": 3.421401977539062e-05,
      "model_forward_time": 2.4260356426239014,
      "step": 8969
    },
    {
      "epoch": 3.421401977539062e-05,
      "step": 8969,
      "training_step_time": 10.023332118988037
    },
    {
      "epoch": 3.421783447265625e-05,
      "grad_norm": 0.3637670576572418,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 8970
    },
    {
      "epoch": 3.421783447265625e-05,
      "model_forward_time": 3.1488654613494873,
      "step": 8970
    },
    {
      "dynamics_loss_avg": 0.06547902878373861,
      "epoch": 3.421783447265625e-05,
      "step": 8970
    },
    {
      "action_loss_avg": 0.020447480818256737,
      "epoch": 3.421783447265625e-05,
      "step": 8970
    },
    {
      "epoch": 3.421783447265625e-05,
      "step": 8970,
      "training_step_time": 9.932392358779907
    },
    {
      "epoch": 3.421783447265625e-05,
      "model_forward_time": 3.13374924659729,
      "step": 8970
    },
    {
      "epoch": 3.421783447265625e-05,
      "step": 8970,
      "training_step_time": 10.192764520645142
    },
    {
      "epoch": 3.421783447265625e-05,
      "model_forward_time": 3.136976957321167,
      "step": 8970
    },
    {
      "epoch": 3.421783447265625e-05,
      "step": 8970,
      "training_step_time": 9.914834260940552
    },
    {
      "epoch": 3.421783447265625e-05,
      "model_forward_time": 3.1337404251098633,
      "step": 8970
    },
    {
      "epoch": 3.421783447265625e-05,
      "step": 8970,
      "training_step_time": 10.281453371047974
    },
    {
      "epoch": 3.4221649169921875e-05,
      "model_forward_time": 3.1410512924194336,
      "step": 8971
    },
    {
      "epoch": 3.4221649169921875e-05,
      "step": 8971,
      "training_step_time": 9.918546199798584
    },
    {
      "epoch": 3.4221649169921875e-05,
      "model_forward_time": 3.1365771293640137,
      "step": 8971
    },
    {
      "epoch": 3.4221649169921875e-05,
      "step": 8971,
      "training_step_time": 9.913919925689697
    },
    {
      "epoch": 3.4221649169921875e-05,
      "model_forward_time": 3.4159185886383057,
      "step": 8971
    },
    {
      "epoch": 3.4221649169921875e-05,
      "step": 8971,
      "training_step_time": 10.193084716796875
    },
    {
      "epoch": 3.4221649169921875e-05,
      "model_forward_time": 3.138227701187134,
      "step": 8971
    },
    {
      "epoch": 3.4221649169921875e-05,
      "step": 8971,
      "training_step_time": 10.264657258987427
    },
    {
      "epoch": 3.42254638671875e-05,
      "model_forward_time": 3.153377056121826,
      "step": 8972
    },
    {
      "epoch": 3.42254638671875e-05,
      "step": 8972,
      "training_step_time": 9.934955358505249
    },
    {
      "epoch": 3.42254638671875e-05,
      "model_forward_time": 2.4370713233947754,
      "step": 8972
    },
    {
      "epoch": 3.42254638671875e-05,
      "step": 8972,
      "training_step_time": 7.472073078155518
    },
    {
      "epoch": 3.42254638671875e-05,
      "model_forward_time": 3.138733148574829,
      "step": 8972
    },
    {
      "dynamics_loss_avg": 0.05174008700996637,
      "epoch": 3.42254638671875e-05,
      "step": 8972
    },
    {
      "action_loss_avg": 0.013484134059399366,
      "epoch": 3.42254638671875e-05,
      "step": 8972
    },
    {
      "epoch": 3.42254638671875e-05,
      "step": 8972,
      "training_step_time": 9.919874668121338
    },
    {
      "epoch": 3.42254638671875e-05,
      "model_forward_time": 3.140944004058838,
      "step": 8972
    },
    {
      "epoch": 3.42254638671875e-05,
      "step": 8972,
      "training_step_time": 10.023415088653564
    },
    {
      "epoch": 3.422927856445313e-05,
      "model_forward_time": 3.4517669677734375,
      "step": 8973
    },
    {
      "epoch": 3.422927856445313e-05,
      "step": 8973,
      "training_step_time": 10.231759309768677
    },
    {
      "epoch": 3.422927856445313e-05,
      "model_forward_time": 3.138491153717041,
      "step": 8973
    },
    {
      "epoch": 3.422927856445313e-05,
      "step": 8973,
      "training_step_time": 9.916539669036865
    },
    {
      "epoch": 3.422927856445313e-05,
      "model_forward_time": 3.133951187133789,
      "step": 8973
    },
    {
      "epoch": 3.422927856445313e-05,
      "step": 8973,
      "training_step_time": 9.909441947937012
    },
    {
      "epoch": 3.422927856445313e-05,
      "model_forward_time": 3.1408517360687256,
      "step": 8973
    },
    {
      "epoch": 3.422927856445313e-05,
      "step": 8973,
      "training_step_time": 10.35169792175293
    },
    {
      "epoch": 3.423309326171875e-05,
      "model_forward_time": 3.400787591934204,
      "step": 8974
    },
    {
      "epoch": 3.423309326171875e-05,
      "step": 8974,
      "training_step_time": 10.178056240081787
    },
    {
      "epoch": 3.423309326171875e-05,
      "model_forward_time": 3.134124279022217,
      "step": 8974
    },
    {
      "epoch": 3.423309326171875e-05,
      "step": 8974,
      "training_step_time": 9.903252601623535
    },
    {
      "epoch": 3.423309326171875e-05,
      "model_forward_time": 3.1295671463012695,
      "step": 8974
    },
    {
      "epoch": 3.423309326171875e-05,
      "step": 8974,
      "training_step_time": 9.89511752128601
    },
    {
      "epoch": 3.423309326171875e-05,
      "model_forward_time": 3.1356122493743896,
      "step": 8974
    },
    {
      "epoch": 3.423309326171875e-05,
      "step": 8974,
      "training_step_time": 10.016448259353638
    },
    {
      "epoch": 3.4236907958984374e-05,
      "model_forward_time": 3.148444890975952,
      "step": 8975
    },
    {
      "dynamics_loss_avg": 0.07652043253183365,
      "epoch": 3.4236907958984374e-05,
      "step": 8975
    },
    {
      "action_loss_avg": 0.02635811921209097,
      "epoch": 3.4236907958984374e-05,
      "step": 8975
    },
    {
      "epoch": 3.4236907958984374e-05,
      "step": 8975,
      "training_step_time": 9.920976638793945
    },
    {
      "epoch": 3.4236907958984374e-05,
      "model_forward_time": 3.406419277191162,
      "step": 8975
    },
    {
      "epoch": 3.4236907958984374e-05,
      "step": 8975,
      "training_step_time": 10.179366111755371
    },
    {
      "epoch": 3.4236907958984374e-05,
      "model_forward_time": 3.129647731781006,
      "step": 8975
    },
    {
      "epoch": 3.4236907958984374e-05,
      "step": 8975,
      "training_step_time": 9.908101320266724
    },
    {
      "epoch": 3.4236907958984374e-05,
      "model_forward_time": 3.1393115520477295,
      "step": 8975
    },
    {
      "epoch": 3.4236907958984374e-05,
      "step": 8975,
      "training_step_time": 10.029454469680786
    },
    {
      "epoch": 3.424072265625e-05,
      "model_forward_time": 3.157132863998413,
      "step": 8976
    },
    {
      "epoch": 3.424072265625e-05,
      "step": 8976,
      "training_step_time": 9.942005395889282
    },
    {
      "epoch": 3.424072265625e-05,
      "model_forward_time": 3.147916555404663,
      "step": 8976
    },
    {
      "epoch": 3.424072265625e-05,
      "step": 8976,
      "training_step_time": 9.92738151550293
    },
    {
      "epoch": 3.424072265625e-05,
      "model_forward_time": 3.404458522796631,
      "step": 8976
    },
    {
      "epoch": 3.424072265625e-05,
      "step": 8976,
      "training_step_time": 10.182448863983154
    },
    {
      "epoch": 3.424072265625e-05,
      "model_forward_time": 3.1491639614105225,
      "step": 8976
    },
    {
      "epoch": 3.424072265625e-05,
      "step": 8976,
      "training_step_time": 10.281357765197754
    },
    {
      "epoch": 3.4244537353515626e-05,
      "model_forward_time": 3.1646366119384766,
      "step": 8977
    },
    {
      "epoch": 3.4244537353515626e-05,
      "step": 8977,
      "training_step_time": 9.95252013206482
    },
    {
      "epoch": 3.4244537353515626e-05,
      "model_forward_time": 3.144352912902832,
      "step": 8977
    },
    {
      "epoch": 3.4244537353515626e-05,
      "step": 8977,
      "training_step_time": 9.925742626190186
    },
    {
      "epoch": 3.4244537353515626e-05,
      "model_forward_time": 2.4419288635253906,
      "step": 8977
    },
    {
      "dynamics_loss_avg": 0.056228610686957836,
      "epoch": 3.4244537353515626e-05,
      "step": 8977
    },
    {
      "action_loss_avg": 0.021179066598415376,
      "epoch": 3.4244537353515626e-05,
      "step": 8977
    },
    {
      "epoch": 3.4244537353515626e-05,
      "step": 8977,
      "training_step_time": 7.479917287826538
    },
    {
      "epoch": 3.4244537353515626e-05,
      "model_forward_time": 3.1360130310058594,
      "step": 8977
    },
    {
      "epoch": 3.4244537353515626e-05,
      "step": 8977,
      "training_step_time": 10.29168701171875
    },
    {
      "epoch": 3.424835205078125e-05,
      "model_forward_time": 3.156529664993286,
      "step": 8978
    },
    {
      "epoch": 3.424835205078125e-05,
      "step": 8978,
      "training_step_time": 9.935240030288696
    },
    {
      "epoch": 3.424835205078125e-05,
      "model_forward_time": 3.137486696243286,
      "step": 8978
    },
    {
      "epoch": 3.424835205078125e-05,
      "step": 8978,
      "training_step_time": 9.913341760635376
    },
    {
      "epoch": 3.424835205078125e-05,
      "model_forward_time": 3.1383321285247803,
      "step": 8978
    },
    {
      "epoch": 3.424835205078125e-05,
      "step": 8978,
      "training_step_time": 9.919119834899902
    },
    {
      "epoch": 3.424835205078125e-05,
      "model_forward_time": 3.1405153274536133,
      "step": 8978
    },
    {
      "epoch": 3.424835205078125e-05,
      "step": 8978,
      "training_step_time": 10.321691513061523
    },
    {
      "epoch": 3.425216674804687e-05,
      "model_forward_time": 3.151109457015991,
      "step": 8979
    },
    {
      "epoch": 3.425216674804687e-05,
      "step": 8979,
      "training_step_time": 10.257680654525757
    },
    {
      "epoch": 3.425216674804687e-05,
      "model_forward_time": 3.1297080516815186,
      "step": 8979
    },
    {
      "epoch": 3.425216674804687e-05,
      "step": 8979,
      "training_step_time": 9.903159618377686
    },
    {
      "epoch": 3.425216674804687e-05,
      "model_forward_time": 3.132293939590454,
      "step": 8979
    },
    {
      "epoch": 3.425216674804687e-05,
      "step": 8979,
      "training_step_time": 9.909852743148804
    },
    {
      "epoch": 3.425216674804687e-05,
      "model_forward_time": 3.1446120738983154,
      "step": 8979
    },
    {
      "epoch": 3.425216674804687e-05,
      "step": 8979,
      "training_step_time": 10.33739423751831
    },
    {
      "epoch": 3.42559814453125e-05,
      "grad_norm": 0.5664283037185669,
      "learning_rate": 1e-05,
      "loss": 0.0836,
      "step": 8980
    },
    {
      "epoch": 3.42559814453125e-05,
      "model_forward_time": 2.449814796447754,
      "step": 8980
    },
    {
      "dynamics_loss_avg": 0.06594786792993546,
      "epoch": 3.42559814453125e-05,
      "step": 8980
    },
    {
      "action_loss_avg": 0.016584715712815522,
      "epoch": 3.42559814453125e-05,
      "step": 8980
    },
    {
      "epoch": 3.42559814453125e-05,
      "step": 8980,
      "training_step_time": 7.48519492149353
    },
    {
      "epoch": 3.42559814453125e-05,
      "model_forward_time": 3.1301369667053223,
      "step": 8980
    },
    {
      "epoch": 3.42559814453125e-05,
      "step": 8980,
      "training_step_time": 9.904544353485107
    },
    {
      "epoch": 3.42559814453125e-05,
      "model_forward_time": 3.409853219985962,
      "step": 8980
    },
    {
      "epoch": 3.42559814453125e-05,
      "step": 8980,
      "training_step_time": 10.18462872505188
    },
    {
      "epoch": 3.42559814453125e-05,
      "model_forward_time": 2.433263063430786,
      "step": 8980
    },
    {
      "epoch": 3.42559814453125e-05,
      "step": 8980,
      "training_step_time": 10.017122507095337
    },
    {
      "epoch": 3.4259796142578125e-05,
      "model_forward_time": 2.4429478645324707,
      "step": 8981
    },
    {
      "epoch": 3.4259796142578125e-05,
      "step": 8981,
      "training_step_time": 7.482542276382446
    },
    {
      "epoch": 3.4259796142578125e-05,
      "model_forward_time": 3.1345458030700684,
      "step": 8981
    },
    {
      "epoch": 3.4259796142578125e-05,
      "step": 8981,
      "training_step_time": 9.92015027999878
    },
    {
      "epoch": 3.4259796142578125e-05,
      "model_forward_time": 3.1368134021759033,
      "step": 8981
    },
    {
      "epoch": 3.4259796142578125e-05,
      "step": 8981,
      "training_step_time": 9.913170576095581
    },
    {
      "epoch": 3.4259796142578125e-05,
      "model_forward_time": 2.744433641433716,
      "step": 8981
    },
    {
      "epoch": 3.4259796142578125e-05,
      "step": 8981,
      "training_step_time": 10.347888708114624
    },
    {
      "epoch": 3.426361083984375e-05,
      "model_forward_time": 3.1400644779205322,
      "step": 8982
    },
    {
      "epoch": 3.426361083984375e-05,
      "step": 8982,
      "training_step_time": 9.913535356521606
    },
    {
      "epoch": 3.426361083984375e-05,
      "model_forward_time": 3.1381189823150635,
      "step": 8982
    },
    {
      "epoch": 3.426361083984375e-05,
      "step": 8982,
      "training_step_time": 9.921935081481934
    },
    {
      "epoch": 3.426361083984375e-05,
      "model_forward_time": 3.140584945678711,
      "step": 8982
    },
    {
      "dynamics_loss_avg": 0.04714491534978151,
      "epoch": 3.426361083984375e-05,
      "step": 8982
    },
    {
      "action_loss_avg": 0.015430467575788498,
      "epoch": 3.426361083984375e-05,
      "step": 8982
    },
    {
      "epoch": 3.426361083984375e-05,
      "step": 8982,
      "training_step_time": 9.928258419036865
    },
    {
      "epoch": 3.426361083984375e-05,
      "model_forward_time": 3.143237352371216,
      "step": 8982
    },
    {
      "epoch": 3.426361083984375e-05,
      "step": 8982,
      "training_step_time": 10.022615194320679
    },
    {
      "epoch": 3.426742553710938e-05,
      "model_forward_time": 3.452284812927246,
      "step": 8983
    },
    {
      "epoch": 3.426742553710938e-05,
      "step": 8983,
      "training_step_time": 10.232661247253418
    },
    {
      "epoch": 3.426742553710938e-05,
      "model_forward_time": 3.150301456451416,
      "step": 8983
    },
    {
      "epoch": 3.426742553710938e-05,
      "step": 8983,
      "training_step_time": 9.934317588806152
    },
    {
      "epoch": 3.426742553710938e-05,
      "model_forward_time": 3.149761438369751,
      "step": 8983
    },
    {
      "epoch": 3.426742553710938e-05,
      "step": 8983,
      "training_step_time": 9.932722568511963
    },
    {
      "epoch": 3.426742553710938e-05,
      "model_forward_time": 3.148885726928711,
      "step": 8983
    },
    {
      "epoch": 3.426742553710938e-05,
      "step": 8983,
      "training_step_time": 10.29566240310669
    },
    {
      "epoch": 3.4271240234375e-05,
      "model_forward_time": 3.156761407852173,
      "step": 8984
    },
    {
      "epoch": 3.4271240234375e-05,
      "step": 8984,
      "training_step_time": 9.950988531112671
    },
    {
      "epoch": 3.4271240234375e-05,
      "model_forward_time": 3.457697629928589,
      "step": 8984
    },
    {
      "epoch": 3.4271240234375e-05,
      "step": 8984,
      "training_step_time": 10.234870910644531
    },
    {
      "epoch": 3.4271240234375e-05,
      "model_forward_time": 3.1480698585510254,
      "step": 8984
    },
    {
      "epoch": 3.4271240234375e-05,
      "step": 8984,
      "training_step_time": 9.922801733016968
    },
    {
      "epoch": 3.4271240234375e-05,
      "model_forward_time": 3.1524157524108887,
      "step": 8984
    },
    {
      "epoch": 3.4271240234375e-05,
      "step": 8984,
      "training_step_time": 10.30063772201538
    },
    {
      "epoch": 3.427505493164062e-05,
      "model_forward_time": 3.158773422241211,
      "step": 8985
    },
    {
      "dynamics_loss_avg": 0.05564645789563656,
      "epoch": 3.427505493164062e-05,
      "step": 8985
    },
    {
      "action_loss_avg": 0.018598280800506472,
      "epoch": 3.427505493164062e-05,
      "step": 8985
    },
    {
      "epoch": 3.427505493164062e-05,
      "step": 8985,
      "training_step_time": 9.940284252166748
    },
    {
      "epoch": 3.427505493164062e-05,
      "model_forward_time": 3.4345176219940186,
      "step": 8985
    },
    {
      "epoch": 3.427505493164062e-05,
      "step": 8985,
      "training_step_time": 10.21408486366272
    },
    {
      "epoch": 3.427505493164062e-05,
      "model_forward_time": 3.1458182334899902,
      "step": 8985
    },
    {
      "epoch": 3.427505493164062e-05,
      "step": 8985,
      "training_step_time": 9.9261314868927
    },
    {
      "epoch": 3.427505493164062e-05,
      "model_forward_time": 2.4409308433532715,
      "step": 8985
    },
    {
      "epoch": 3.427505493164062e-05,
      "step": 8985,
      "training_step_time": 10.022017240524292
    },
    {
      "epoch": 3.427886962890625e-05,
      "model_forward_time": 3.15761661529541,
      "step": 8986
    },
    {
      "epoch": 3.427886962890625e-05,
      "step": 8986,
      "training_step_time": 9.935862302780151
    },
    {
      "epoch": 3.427886962890625e-05,
      "model_forward_time": 2.4385197162628174,
      "step": 8986
    },
    {
      "epoch": 3.427886962890625e-05,
      "step": 8986,
      "training_step_time": 7.470988988876343
    },
    {
      "epoch": 3.427886962890625e-05,
      "model_forward_time": 3.144256114959717,
      "step": 8986
    },
    {
      "epoch": 3.427886962890625e-05,
      "step": 8986,
      "training_step_time": 9.920340061187744
    },
    {
      "epoch": 3.427886962890625e-05,
      "model_forward_time": 3.4595425128936768,
      "step": 8986
    },
    {
      "epoch": 3.427886962890625e-05,
      "step": 8986,
      "training_step_time": 10.340085983276367
    },
    {
      "epoch": 3.4282684326171875e-05,
      "model_forward_time": 3.171886444091797,
      "step": 8987
    },
    {
      "epoch": 3.4282684326171875e-05,
      "step": 8987,
      "training_step_time": 9.964316368103027
    },
    {
      "epoch": 3.4282684326171875e-05,
      "model_forward_time": 3.156416654586792,
      "step": 8987
    },
    {
      "epoch": 3.4282684326171875e-05,
      "step": 8987,
      "training_step_time": 9.94825530052185
    },
    {
      "epoch": 3.4282684326171875e-05,
      "model_forward_time": 3.1530401706695557,
      "step": 8987
    },
    {
      "dynamics_loss_avg": 0.054413306340575215,
      "epoch": 3.4282684326171875e-05,
      "step": 8987
    },
    {
      "action_loss_avg": 0.017046417435631157,
      "epoch": 3.4282684326171875e-05,
      "step": 8987
    },
    {
      "epoch": 3.4282684326171875e-05,
      "step": 8987,
      "training_step_time": 9.94443655014038
    },
    {
      "epoch": 3.4282684326171875e-05,
      "model_forward_time": 3.143617868423462,
      "step": 8987
    },
    {
      "epoch": 3.4282684326171875e-05,
      "step": 8987,
      "training_step_time": 10.305742263793945
    },
    {
      "epoch": 3.42864990234375e-05,
      "model_forward_time": 3.172377347946167,
      "step": 8988
    },
    {
      "epoch": 3.42864990234375e-05,
      "step": 8988,
      "training_step_time": 9.951873540878296
    },
    {
      "epoch": 3.42864990234375e-05,
      "model_forward_time": 3.1436033248901367,
      "step": 8988
    },
    {
      "epoch": 3.42864990234375e-05,
      "step": 8988,
      "training_step_time": 9.922590732574463
    },
    {
      "epoch": 3.42864990234375e-05,
      "model_forward_time": 3.151789903640747,
      "step": 8988
    },
    {
      "epoch": 3.42864990234375e-05,
      "step": 8988,
      "training_step_time": 9.937764167785645
    },
    {
      "epoch": 3.42864990234375e-05,
      "model_forward_time": 3.154129981994629,
      "step": 8988
    },
    {
      "epoch": 3.42864990234375e-05,
      "step": 8988,
      "training_step_time": 10.335676193237305
    },
    {
      "epoch": 3.429031372070313e-05,
      "model_forward_time": 3.1582038402557373,
      "step": 8989
    },
    {
      "epoch": 3.429031372070313e-05,
      "step": 8989,
      "training_step_time": 10.245536804199219
    },
    {
      "epoch": 3.429031372070313e-05,
      "model_forward_time": 3.1453471183776855,
      "step": 8989
    },
    {
      "epoch": 3.429031372070313e-05,
      "step": 8989,
      "training_step_time": 9.928504943847656
    },
    {
      "epoch": 3.429031372070313e-05,
      "model_forward_time": 2.45092511177063,
      "step": 8989
    },
    {
      "epoch": 3.429031372070313e-05,
      "step": 8989,
      "training_step_time": 7.492680072784424
    },
    {
      "epoch": 3.429031372070313e-05,
      "model_forward_time": 3.1502597332000732,
      "step": 8989
    },
    {
      "epoch": 3.429031372070313e-05,
      "step": 8989,
      "training_step_time": 10.307472944259644
    },
    {
      "epoch": 3.429412841796875e-05,
      "grad_norm": 0.4529324769973755,
      "learning_rate": 1e-05,
      "loss": 0.0733,
      "step": 8990
    },
    {
      "epoch": 3.429412841796875e-05,
      "model_forward_time": 3.1700072288513184,
      "step": 8990
    },
    {
      "dynamics_loss_avg": 0.056862011831253764,
      "epoch": 3.429412841796875e-05,
      "step": 8990
    },
    {
      "action_loss_avg": 0.021460226667113603,
      "epoch": 3.429412841796875e-05,
      "step": 8990
    },
    {
      "epoch": 3.429412841796875e-05,
      "step": 8990,
      "training_step_time": 9.958479642868042
    },
    {
      "epoch": 3.429412841796875e-05,
      "model_forward_time": 3.1404428482055664,
      "step": 8990
    },
    {
      "epoch": 3.429412841796875e-05,
      "step": 8990,
      "training_step_time": 10.204328298568726
    },
    {
      "epoch": 3.429412841796875e-05,
      "model_forward_time": 3.142606258392334,
      "step": 8990
    },
    {
      "epoch": 3.429412841796875e-05,
      "step": 8990,
      "training_step_time": 9.920640230178833
    },
    {
      "epoch": 3.429412841796875e-05,
      "model_forward_time": 3.140568256378174,
      "step": 8990
    },
    {
      "epoch": 3.429412841796875e-05,
      "step": 8990,
      "training_step_time": 10.329074621200562
    },
    {
      "epoch": 3.4297943115234374e-05,
      "model_forward_time": 3.1474156379699707,
      "step": 8991
    },
    {
      "epoch": 3.4297943115234374e-05,
      "step": 8991,
      "training_step_time": 9.924503326416016
    },
    {
      "epoch": 3.4297943115234374e-05,
      "model_forward_time": 3.13981032371521,
      "step": 8991
    },
    {
      "epoch": 3.4297943115234374e-05,
      "step": 8991,
      "training_step_time": 9.91286039352417
    },
    {
      "epoch": 3.4297943115234374e-05,
      "model_forward_time": 3.4642984867095947,
      "step": 8991
    },
    {
      "epoch": 3.4297943115234374e-05,
      "step": 8991,
      "training_step_time": 10.243799448013306
    },
    {
      "epoch": 3.4297943115234374e-05,
      "model_forward_time": 2.4377849102020264,
      "step": 8991
    },
    {
      "epoch": 3.4297943115234374e-05,
      "step": 8991,
      "training_step_time": 10.015481233596802
    },
    {
      "epoch": 3.43017578125e-05,
      "model_forward_time": 3.1404645442962646,
      "step": 8992
    },
    {
      "epoch": 3.43017578125e-05,
      "step": 8992,
      "training_step_time": 9.916479587554932
    },
    {
      "epoch": 3.43017578125e-05,
      "model_forward_time": 3.1380670070648193,
      "step": 8992
    },
    {
      "epoch": 3.43017578125e-05,
      "step": 8992,
      "training_step_time": 9.915782690048218
    },
    {
      "epoch": 3.43017578125e-05,
      "model_forward_time": 3.1348445415496826,
      "step": 8992
    },
    {
      "dynamics_loss_avg": 0.05591437611728907,
      "epoch": 3.43017578125e-05,
      "step": 8992
    },
    {
      "action_loss_avg": 0.01842293981462717,
      "epoch": 3.43017578125e-05,
      "step": 8992
    },
    {
      "epoch": 3.43017578125e-05,
      "step": 8992,
      "training_step_time": 9.907659769058228
    },
    {
      "epoch": 3.43017578125e-05,
      "model_forward_time": 3.1382927894592285,
      "step": 8992
    },
    {
      "epoch": 3.43017578125e-05,
      "step": 8992,
      "training_step_time": 10.295460939407349
    },
    {
      "epoch": 3.4305572509765626e-05,
      "model_forward_time": 3.432413101196289,
      "step": 8993
    },
    {
      "epoch": 3.4305572509765626e-05,
      "step": 8993,
      "training_step_time": 10.207476615905762
    },
    {
      "epoch": 3.4305572509765626e-05,
      "model_forward_time": 3.1393074989318848,
      "step": 8993
    },
    {
      "epoch": 3.4305572509765626e-05,
      "step": 8993,
      "training_step_time": 9.916849851608276
    },
    {
      "epoch": 3.4305572509765626e-05,
      "model_forward_time": 3.1377017498016357,
      "step": 8993
    },
    {
      "epoch": 3.4305572509765626e-05,
      "step": 8993,
      "training_step_time": 9.913403987884521
    },
    {
      "epoch": 3.4305572509765626e-05,
      "model_forward_time": 3.138282299041748,
      "step": 8993
    },
    {
      "epoch": 3.4305572509765626e-05,
      "step": 8993,
      "training_step_time": 10.011255741119385
    },
    {
      "epoch": 3.430938720703125e-05,
      "model_forward_time": 3.4329257011413574,
      "step": 8994
    },
    {
      "epoch": 3.430938720703125e-05,
      "step": 8994,
      "training_step_time": 10.211791038513184
    },
    {
      "epoch": 3.430938720703125e-05,
      "model_forward_time": 3.1399667263031006,
      "step": 8994
    },
    {
      "epoch": 3.430938720703125e-05,
      "step": 8994,
      "training_step_time": 9.915706872940063
    },
    {
      "epoch": 3.430938720703125e-05,
      "model_forward_time": 3.138474702835083,
      "step": 8994
    },
    {
      "epoch": 3.430938720703125e-05,
      "step": 8994,
      "training_step_time": 9.91678261756897
    },
    {
      "epoch": 3.430938720703125e-05,
      "model_forward_time": 3.141979932785034,
      "step": 8994
    },
    {
      "epoch": 3.430938720703125e-05,
      "step": 8994,
      "training_step_time": 10.388386487960815
    },
    {
      "epoch": 3.431320190429687e-05,
      "model_forward_time": 3.1477537155151367,
      "step": 8995
    },
    {
      "dynamics_loss_avg": 0.0721886308863759,
      "epoch": 3.431320190429687e-05,
      "step": 8995
    },
    {
      "action_loss_avg": 0.023651415761560202,
      "epoch": 3.431320190429687e-05,
      "step": 8995
    },
    {
      "epoch": 3.431320190429687e-05,
      "step": 8995,
      "training_step_time": 9.92793869972229
    },
    {
      "epoch": 3.431320190429687e-05,
      "model_forward_time": 3.4583182334899902,
      "step": 8995
    },
    {
      "epoch": 3.431320190429687e-05,
      "step": 8995,
      "training_step_time": 10.231420040130615
    },
    {
      "epoch": 3.431320190429687e-05,
      "model_forward_time": 3.1340012550354004,
      "step": 8995
    },
    {
      "epoch": 3.431320190429687e-05,
      "step": 8995,
      "training_step_time": 9.906015157699585
    },
    {
      "epoch": 3.431320190429687e-05,
      "model_forward_time": 3.1385748386383057,
      "step": 8995
    },
    {
      "epoch": 3.431320190429687e-05,
      "step": 8995,
      "training_step_time": 10.28146505355835
    },
    {
      "epoch": 3.43170166015625e-05,
      "model_forward_time": 3.155933141708374,
      "step": 8996
    },
    {
      "epoch": 3.43170166015625e-05,
      "step": 8996,
      "training_step_time": 9.935099124908447
    },
    {
      "epoch": 3.43170166015625e-05,
      "model_forward_time": 3.142011880874634,
      "step": 8996
    },
    {
      "epoch": 3.43170166015625e-05,
      "step": 8996,
      "training_step_time": 9.920122861862183
    },
    {
      "epoch": 3.43170166015625e-05,
      "model_forward_time": 2.731879472732544,
      "step": 8996
    },
    {
      "epoch": 3.43170166015625e-05,
      "step": 8996,
      "training_step_time": 7.766392230987549
    },
    {
      "epoch": 3.43170166015625e-05,
      "model_forward_time": 3.1397502422332764,
      "step": 8996
    },
    {
      "epoch": 3.43170166015625e-05,
      "step": 8996,
      "training_step_time": 10.03028917312622
    },
    {
      "epoch": 3.4320831298828125e-05,
      "model_forward_time": 3.1501312255859375,
      "step": 8997
    },
    {
      "epoch": 3.4320831298828125e-05,
      "step": 8997,
      "training_step_time": 9.928142309188843
    },
    {
      "epoch": 3.4320831298828125e-05,
      "model_forward_time": 2.4374184608459473,
      "step": 8997
    },
    {
      "epoch": 3.4320831298828125e-05,
      "step": 8997,
      "training_step_time": 7.467161178588867
    },
    {
      "epoch": 3.4320831298828125e-05,
      "model_forward_time": 3.1351687908172607,
      "step": 8997
    },
    {
      "dynamics_loss_avg": 0.06335480976849794,
      "epoch": 3.4320831298828125e-05,
      "step": 8997
    },
    {
      "action_loss_avg": 0.023474736884236336,
      "epoch": 3.4320831298828125e-05,
      "step": 8997
    },
    {
      "epoch": 3.4320831298828125e-05,
      "step": 8997,
      "training_step_time": 9.911206483840942
    },
    {
      "epoch": 3.4320831298828125e-05,
      "model_forward_time": 3.1355950832366943,
      "step": 8997
    },
    {
      "epoch": 3.4320831298828125e-05,
      "step": 8997,
      "training_step_time": 10.339221477508545
    },
    {
      "epoch": 3.432464599609375e-05,
      "model_forward_time": 3.460810661315918,
      "step": 8998
    },
    {
      "epoch": 3.432464599609375e-05,
      "step": 8998,
      "training_step_time": 10.240018129348755
    },
    {
      "epoch": 3.432464599609375e-05,
      "model_forward_time": 3.1405465602874756,
      "step": 8998
    },
    {
      "epoch": 3.432464599609375e-05,
      "step": 8998,
      "training_step_time": 9.920038223266602
    },
    {
      "epoch": 3.432464599609375e-05,
      "model_forward_time": 3.140786647796631,
      "step": 8998
    },
    {
      "epoch": 3.432464599609375e-05,
      "step": 8998,
      "training_step_time": 9.916521787643433
    },
    {
      "epoch": 3.432464599609375e-05,
      "model_forward_time": 3.1333134174346924,
      "step": 8998
    },
    {
      "epoch": 3.432464599609375e-05,
      "step": 8998,
      "training_step_time": 10.298096656799316
    },
    {
      "epoch": 3.432846069335938e-05,
      "model_forward_time": 3.1886675357818604,
      "step": 8999
    },
    {
      "epoch": 3.432846069335938e-05,
      "step": 8999,
      "training_step_time": 10.24827265739441
    },
    {
      "epoch": 3.432846069335938e-05,
      "model_forward_time": 3.1372318267822266,
      "step": 8999
    },
    {
      "epoch": 3.432846069335938e-05,
      "step": 8999,
      "training_step_time": 9.919641494750977
    },
    {
      "epoch": 3.432846069335938e-05,
      "model_forward_time": 3.145045042037964,
      "step": 8999
    },
    {
      "epoch": 3.432846069335938e-05,
      "step": 8999,
      "training_step_time": 9.920355081558228
    },
    {
      "epoch": 3.432846069335938e-05,
      "model_forward_time": 3.1359641551971436,
      "step": 8999
    },
    {
      "epoch": 3.432846069335938e-05,
      "step": 8999,
      "training_step_time": 10.424257278442383
    },
    {
      "epoch": 3.4332275390625e-05,
      "grad_norm": 0.6304811239242554,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 9000
    },
    {
      "epoch": 3.4332275390625e-05,
      "model_forward_time": 3.1784183979034424,
      "step": 9000
    },
    {
      "dynamics_loss_avg": 0.07369113862514495,
      "epoch": 3.4332275390625e-05,
      "step": 9000
    },
    {
      "action_loss_avg": 0.029308376600965858,
      "epoch": 3.4332275390625e-05,
      "step": 9000
    },
    {
      "epoch": 3.4332275390625e-05,
      "step": 9000,
      "training_step_time": 9.935446500778198
    },
    {
      "epoch": 3.4332275390625e-05,
      "model_forward_time": 5.349946975708008,
      "step": 9000
    },
    {
      "epoch": 3.4332275390625e-05,
      "step": 9000,
      "training_step_time": 12.117239475250244
    },
    {
      "epoch": 3.4332275390625e-05,
      "model_forward_time": 3.1322827339172363,
      "step": 9000
    },
    {
      "epoch": 3.4332275390625e-05,
      "step": 9000,
      "training_step_time": 9.890358924865723
    },
    {
      "epoch": 3.4332275390625e-05,
      "model_forward_time": 3.1282472610473633,
      "step": 9000
    },
    {
      "epoch": 3.4332275390625e-05,
      "step": 9000,
      "training_step_time": 10.329814195632935
    },
    {
      "epoch": 3.433609008789062e-05,
      "model_forward_time": 3.1389882564544678,
      "step": 9001
    },
    {
      "epoch": 3.433609008789062e-05,
      "step": 9001,
      "training_step_time": 9.91066575050354
    },
    {
      "epoch": 3.433609008789062e-05,
      "model_forward_time": 3.133056402206421,
      "step": 9001
    },
    {
      "epoch": 3.433609008789062e-05,
      "step": 9001,
      "training_step_time": 10.218900918960571
    },
    {
      "epoch": 3.433609008789062e-05,
      "model_forward_time": 3.131969451904297,
      "step": 9001
    },
    {
      "epoch": 3.433609008789062e-05,
      "step": 9001,
      "training_step_time": 9.902709722518921
    },
    {
      "epoch": 3.433609008789062e-05,
      "model_forward_time": 3.1315784454345703,
      "step": 9001
    },
    {
      "epoch": 3.433609008789062e-05,
      "step": 9001,
      "training_step_time": 10.268279790878296
    },
    {
      "epoch": 3.433990478515625e-05,
      "model_forward_time": 3.1376771926879883,
      "step": 9002
    },
    {
      "epoch": 3.433990478515625e-05,
      "step": 9002,
      "training_step_time": 9.915491342544556
    },
    {
      "epoch": 3.433990478515625e-05,
      "model_forward_time": 3.134997844696045,
      "step": 9002
    },
    {
      "epoch": 3.433990478515625e-05,
      "step": 9002,
      "training_step_time": 9.913999080657959
    },
    {
      "epoch": 3.433990478515625e-05,
      "model_forward_time": 3.1363182067871094,
      "step": 9002
    },
    {
      "dynamics_loss_avg": 0.05856617596000433,
      "epoch": 3.433990478515625e-05,
      "step": 9002
    },
    {
      "action_loss_avg": 0.02075489712879062,
      "epoch": 3.433990478515625e-05,
      "step": 9002
    },
    {
      "epoch": 3.433990478515625e-05,
      "step": 9002,
      "training_step_time": 10.207587242126465
    },
    {
      "epoch": 3.433990478515625e-05,
      "model_forward_time": 3.1368839740753174,
      "step": 9002
    },
    {
      "epoch": 3.433990478515625e-05,
      "step": 9002,
      "training_step_time": 10.024299144744873
    },
    {
      "epoch": 3.4343719482421875e-05,
      "model_forward_time": 3.151498317718506,
      "step": 9003
    },
    {
      "epoch": 3.4343719482421875e-05,
      "step": 9003,
      "training_step_time": 9.932857751846313
    },
    {
      "epoch": 3.4343719482421875e-05,
      "model_forward_time": 3.1450958251953125,
      "step": 9003
    },
    {
      "epoch": 3.4343719482421875e-05,
      "step": 9003,
      "training_step_time": 9.926789283752441
    },
    {
      "epoch": 3.4343719482421875e-05,
      "model_forward_time": 3.145615577697754,
      "step": 9003
    },
    {
      "epoch": 3.4343719482421875e-05,
      "step": 9003,
      "training_step_time": 9.932112216949463
    },
    {
      "epoch": 3.4343719482421875e-05,
      "model_forward_time": 3.139219045639038,
      "step": 9003
    },
    {
      "epoch": 3.4343719482421875e-05,
      "step": 9003,
      "training_step_time": 10.328343152999878
    },
    {
      "epoch": 3.43475341796875e-05,
      "model_forward_time": 2.4467623233795166,
      "step": 9004
    },
    {
      "epoch": 3.43475341796875e-05,
      "step": 9004,
      "training_step_time": 7.47764778137207
    },
    {
      "epoch": 3.43475341796875e-05,
      "model_forward_time": 3.140652894973755,
      "step": 9004
    },
    {
      "epoch": 3.43475341796875e-05,
      "step": 9004,
      "training_step_time": 9.909794807434082
    },
    {
      "epoch": 3.43475341796875e-05,
      "model_forward_time": 3.137253761291504,
      "step": 9004
    },
    {
      "epoch": 3.43475341796875e-05,
      "step": 9004,
      "training_step_time": 9.913154602050781
    },
    {
      "epoch": 3.43475341796875e-05,
      "model_forward_time": 3.143822193145752,
      "step": 9004
    },
    {
      "epoch": 3.43475341796875e-05,
      "step": 9004,
      "training_step_time": 10.01695442199707
    },
    {
      "epoch": 3.435134887695313e-05,
      "model_forward_time": 3.1498751640319824,
      "step": 9005
    },
    {
      "dynamics_loss_avg": 0.06422882936894894,
      "epoch": 3.435134887695313e-05,
      "step": 9005
    },
    {
      "action_loss_avg": 0.021939313411712645,
      "epoch": 3.435134887695313e-05,
      "step": 9005
    },
    {
      "epoch": 3.435134887695313e-05,
      "step": 9005,
      "training_step_time": 9.927680015563965
    },
    {
      "epoch": 3.435134887695313e-05,
      "model_forward_time": 3.4126734733581543,
      "step": 9005
    },
    {
      "epoch": 3.435134887695313e-05,
      "step": 9005,
      "training_step_time": 10.18126130104065
    },
    {
      "epoch": 3.435134887695313e-05,
      "model_forward_time": 3.143692970275879,
      "step": 9005
    },
    {
      "epoch": 3.435134887695313e-05,
      "step": 9005,
      "training_step_time": 9.921428680419922
    },
    {
      "epoch": 3.435134887695313e-05,
      "model_forward_time": 3.144103527069092,
      "step": 9005
    },
    {
      "epoch": 3.435134887695313e-05,
      "step": 9005,
      "training_step_time": 10.013057470321655
    },
    {
      "epoch": 3.435516357421875e-05,
      "model_forward_time": 3.1679799556732178,
      "step": 9006
    },
    {
      "epoch": 3.435516357421875e-05,
      "step": 9006,
      "training_step_time": 9.950967788696289
    },
    {
      "epoch": 3.435516357421875e-05,
      "model_forward_time": 3.485186815261841,
      "step": 9006
    },
    {
      "epoch": 3.435516357421875e-05,
      "step": 9006,
      "training_step_time": 10.26263952255249
    },
    {
      "epoch": 3.435516357421875e-05,
      "model_forward_time": 3.1458566188812256,
      "step": 9006
    },
    {
      "epoch": 3.435516357421875e-05,
      "step": 9006,
      "training_step_time": 9.924339294433594
    },
    {
      "epoch": 3.435516357421875e-05,
      "model_forward_time": 3.147569179534912,
      "step": 9006
    },
    {
      "epoch": 3.435516357421875e-05,
      "step": 9006,
      "training_step_time": 10.018415451049805
    },
    {
      "epoch": 3.4358978271484374e-05,
      "model_forward_time": 3.1674177646636963,
      "step": 9007
    },
    {
      "epoch": 3.4358978271484374e-05,
      "step": 9007,
      "training_step_time": 9.950927019119263
    },
    {
      "epoch": 3.4358978271484374e-05,
      "model_forward_time": 3.1419155597686768,
      "step": 9007
    },
    {
      "epoch": 3.4358978271484374e-05,
      "step": 9007,
      "training_step_time": 9.921128511428833
    },
    {
      "epoch": 3.4358978271484374e-05,
      "model_forward_time": 2.4430959224700928,
      "step": 9007
    },
    {
      "dynamics_loss_avg": 0.06921802517026662,
      "epoch": 3.4358978271484374e-05,
      "step": 9007
    },
    {
      "action_loss_avg": 0.027692848164588214,
      "epoch": 3.4358978271484374e-05,
      "step": 9007
    },
    {
      "epoch": 3.4358978271484374e-05,
      "step": 9007,
      "training_step_time": 7.482132196426392
    },
    {
      "epoch": 3.4358978271484374e-05,
      "model_forward_time": 3.412172317504883,
      "step": 9007
    },
    {
      "epoch": 3.4358978271484374e-05,
      "step": 9007,
      "training_step_time": 10.2820885181427
    },
    {
      "epoch": 3.436279296875e-05,
      "model_forward_time": 3.175353527069092,
      "step": 9008
    },
    {
      "epoch": 3.436279296875e-05,
      "step": 9008,
      "training_step_time": 9.953451871871948
    },
    {
      "epoch": 3.436279296875e-05,
      "model_forward_time": 3.145535707473755,
      "step": 9008
    },
    {
      "epoch": 3.436279296875e-05,
      "step": 9008,
      "training_step_time": 9.922865867614746
    },
    {
      "epoch": 3.436279296875e-05,
      "model_forward_time": 3.14737606048584,
      "step": 9008
    },
    {
      "epoch": 3.436279296875e-05,
      "step": 9008,
      "training_step_time": 9.924149751663208
    },
    {
      "epoch": 3.436279296875e-05,
      "model_forward_time": 3.428208112716675,
      "step": 9008
    },
    {
      "epoch": 3.436279296875e-05,
      "step": 9008,
      "training_step_time": 10.31438946723938
    },
    {
      "epoch": 3.4366607666015626e-05,
      "model_forward_time": 3.168363332748413,
      "step": 9009
    },
    {
      "epoch": 3.4366607666015626e-05,
      "step": 9009,
      "training_step_time": 9.950793504714966
    },
    {
      "epoch": 3.4366607666015626e-05,
      "model_forward_time": 3.1426100730895996,
      "step": 9009
    },
    {
      "epoch": 3.4366607666015626e-05,
      "step": 9009,
      "training_step_time": 9.921407699584961
    },
    {
      "epoch": 3.4366607666015626e-05,
      "model_forward_time": 2.439586639404297,
      "step": 9009
    },
    {
      "epoch": 3.4366607666015626e-05,
      "step": 9009,
      "training_step_time": 7.473520755767822
    },
    {
      "epoch": 3.4366607666015626e-05,
      "model_forward_time": 3.1461808681488037,
      "step": 9009
    },
    {
      "epoch": 3.4366607666015626e-05,
      "step": 9009,
      "training_step_time": 10.299527168273926
    },
    {
      "epoch": 3.437042236328125e-05,
      "grad_norm": 0.436233788728714,
      "learning_rate": 1e-05,
      "loss": 0.0826,
      "step": 9010
    },
    {
      "epoch": 3.437042236328125e-05,
      "model_forward_time": 3.1503868103027344,
      "step": 9010
    },
    {
      "dynamics_loss_avg": 0.07108514700084925,
      "epoch": 3.437042236328125e-05,
      "step": 9010
    },
    {
      "action_loss_avg": 0.023476110119372607,
      "epoch": 3.437042236328125e-05,
      "step": 9010
    },
    {
      "epoch": 3.437042236328125e-05,
      "step": 9010,
      "training_step_time": 9.935157299041748
    },
    {
      "epoch": 3.437042236328125e-05,
      "model_forward_time": 3.421055555343628,
      "step": 9010
    },
    {
      "epoch": 3.437042236328125e-05,
      "step": 9010,
      "training_step_time": 10.199397325515747
    },
    {
      "epoch": 3.437042236328125e-05,
      "model_forward_time": 3.143001079559326,
      "step": 9010
    },
    {
      "epoch": 3.437042236328125e-05,
      "step": 9010,
      "training_step_time": 9.919766426086426
    },
    {
      "epoch": 3.437042236328125e-05,
      "model_forward_time": 3.1476969718933105,
      "step": 9010
    },
    {
      "epoch": 3.437042236328125e-05,
      "step": 9010,
      "training_step_time": 10.018141269683838
    },
    {
      "epoch": 3.437423706054687e-05,
      "model_forward_time": 3.1666250228881836,
      "step": 9011
    },
    {
      "epoch": 3.437423706054687e-05,
      "step": 9011,
      "training_step_time": 9.960933685302734
    },
    {
      "epoch": 3.437423706054687e-05,
      "model_forward_time": 3.408860206604004,
      "step": 9011
    },
    {
      "epoch": 3.437423706054687e-05,
      "step": 9011,
      "training_step_time": 10.17864179611206
    },
    {
      "epoch": 3.437423706054687e-05,
      "model_forward_time": 2.4345805644989014,
      "step": 9011
    },
    {
      "epoch": 3.437423706054687e-05,
      "step": 9011,
      "training_step_time": 7.467173099517822
    },
    {
      "epoch": 3.437423706054687e-05,
      "model_forward_time": 3.1321518421173096,
      "step": 9011
    },
    {
      "epoch": 3.437423706054687e-05,
      "step": 9011,
      "training_step_time": 10.0181884765625
    },
    {
      "epoch": 3.43780517578125e-05,
      "model_forward_time": 3.161404848098755,
      "step": 9012
    },
    {
      "epoch": 3.43780517578125e-05,
      "step": 9012,
      "training_step_time": 9.934269428253174
    },
    {
      "epoch": 3.43780517578125e-05,
      "model_forward_time": 3.1354546546936035,
      "step": 9012
    },
    {
      "epoch": 3.43780517578125e-05,
      "step": 9012,
      "training_step_time": 9.906986951828003
    },
    {
      "epoch": 3.43780517578125e-05,
      "model_forward_time": 3.1331751346588135,
      "step": 9012
    },
    {
      "dynamics_loss_avg": 0.04620840642601252,
      "epoch": 3.43780517578125e-05,
      "step": 9012
    },
    {
      "action_loss_avg": 0.017044936725869774,
      "epoch": 3.43780517578125e-05,
      "step": 9012
    },
    {
      "epoch": 3.43780517578125e-05,
      "step": 9012,
      "training_step_time": 9.911948204040527
    },
    {
      "epoch": 3.43780517578125e-05,
      "model_forward_time": 3.433316707611084,
      "step": 9012
    },
    {
      "epoch": 3.43780517578125e-05,
      "step": 9012,
      "training_step_time": 10.30323576927185
    },
    {
      "epoch": 3.4381866455078125e-05,
      "model_forward_time": 3.1947481632232666,
      "step": 9013
    },
    {
      "epoch": 3.4381866455078125e-05,
      "step": 9013,
      "training_step_time": 9.971604585647583
    },
    {
      "epoch": 3.4381866455078125e-05,
      "model_forward_time": 3.1393520832061768,
      "step": 9013
    },
    {
      "epoch": 3.4381866455078125e-05,
      "step": 9013,
      "training_step_time": 9.916377305984497
    },
    {
      "epoch": 3.4381866455078125e-05,
      "model_forward_time": 3.1340067386627197,
      "step": 9013
    },
    {
      "epoch": 3.4381866455078125e-05,
      "step": 9013,
      "training_step_time": 9.911566972732544
    },
    {
      "epoch": 3.4381866455078125e-05,
      "model_forward_time": 3.1394426822662354,
      "step": 9013
    },
    {
      "epoch": 3.4381866455078125e-05,
      "step": 9013,
      "training_step_time": 10.290268898010254
    },
    {
      "epoch": 3.438568115234375e-05,
      "model_forward_time": 3.1457223892211914,
      "step": 9014
    },
    {
      "epoch": 3.438568115234375e-05,
      "step": 9014,
      "training_step_time": 9.926884651184082
    },
    {
      "epoch": 3.438568115234375e-05,
      "model_forward_time": 3.1362760066986084,
      "step": 9014
    },
    {
      "epoch": 3.438568115234375e-05,
      "step": 9014,
      "training_step_time": 9.914708614349365
    },
    {
      "epoch": 3.438568115234375e-05,
      "model_forward_time": 3.1360738277435303,
      "step": 9014
    },
    {
      "epoch": 3.438568115234375e-05,
      "step": 9014,
      "training_step_time": 9.923872470855713
    },
    {
      "epoch": 3.438568115234375e-05,
      "model_forward_time": 3.140533924102783,
      "step": 9014
    },
    {
      "epoch": 3.438568115234375e-05,
      "step": 9014,
      "training_step_time": 10.271626472473145
    },
    {
      "epoch": 3.438949584960938e-05,
      "model_forward_time": 2.451477527618408,
      "step": 9015
    },
    {
      "dynamics_loss_avg": 0.04721311889588833,
      "epoch": 3.438949584960938e-05,
      "step": 9015
    },
    {
      "action_loss_avg": 0.015135139273479582,
      "epoch": 3.438949584960938e-05,
      "step": 9015
    },
    {
      "epoch": 3.438949584960938e-05,
      "step": 9015,
      "training_step_time": 7.4872846603393555
    },
    {
      "epoch": 3.438949584960938e-05,
      "model_forward_time": 3.4139797687530518,
      "step": 9015
    },
    {
      "epoch": 3.438949584960938e-05,
      "step": 9015,
      "training_step_time": 10.185734033584595
    },
    {
      "epoch": 3.438949584960938e-05,
      "model_forward_time": 3.142638921737671,
      "step": 9015
    },
    {
      "epoch": 3.438949584960938e-05,
      "step": 9015,
      "training_step_time": 9.91435194015503
    },
    {
      "epoch": 3.438949584960938e-05,
      "model_forward_time": 3.1448752880096436,
      "step": 9015
    },
    {
      "epoch": 3.438949584960938e-05,
      "step": 9015,
      "training_step_time": 10.280327558517456
    },
    {
      "epoch": 3.4393310546875e-05,
      "model_forward_time": 3.15735125541687,
      "step": 9016
    },
    {
      "epoch": 3.4393310546875e-05,
      "step": 9016,
      "training_step_time": 9.93822193145752
    },
    {
      "epoch": 3.4393310546875e-05,
      "model_forward_time": 3.1431877613067627,
      "step": 9016
    },
    {
      "epoch": 3.4393310546875e-05,
      "step": 9016,
      "training_step_time": 10.207070112228394
    },
    {
      "epoch": 3.4393310546875e-05,
      "model_forward_time": 3.1448962688446045,
      "step": 9016
    },
    {
      "epoch": 3.4393310546875e-05,
      "step": 9016,
      "training_step_time": 9.914275884628296
    },
    {
      "epoch": 3.4393310546875e-05,
      "model_forward_time": 3.1434755325317383,
      "step": 9016
    },
    {
      "epoch": 3.4393310546875e-05,
      "step": 9016,
      "training_step_time": 10.020536661148071
    },
    {
      "epoch": 3.439712524414062e-05,
      "model_forward_time": 3.153170347213745,
      "step": 9017
    },
    {
      "epoch": 3.439712524414062e-05,
      "step": 9017,
      "training_step_time": 9.931723356246948
    },
    {
      "epoch": 3.439712524414062e-05,
      "model_forward_time": 3.1413826942443848,
      "step": 9017
    },
    {
      "epoch": 3.439712524414062e-05,
      "step": 9017,
      "training_step_time": 9.918694496154785
    },
    {
      "epoch": 3.439712524414062e-05,
      "model_forward_time": 3.411633253097534,
      "step": 9017
    },
    {
      "dynamics_loss_avg": 0.05851534232497215,
      "epoch": 3.439712524414062e-05,
      "step": 9017
    },
    {
      "action_loss_avg": 0.017879657563753426,
      "epoch": 3.439712524414062e-05,
      "step": 9017
    },
    {
      "epoch": 3.439712524414062e-05,
      "step": 9017,
      "training_step_time": 10.190643072128296
    },
    {
      "epoch": 3.439712524414062e-05,
      "model_forward_time": 3.1515228748321533,
      "step": 9017
    },
    {
      "epoch": 3.439712524414062e-05,
      "step": 9017,
      "training_step_time": 10.301469564437866
    },
    {
      "epoch": 3.440093994140625e-05,
      "model_forward_time": 3.150790214538574,
      "step": 9018
    },
    {
      "epoch": 3.440093994140625e-05,
      "step": 9018,
      "training_step_time": 9.92668104171753
    },
    {
      "epoch": 3.440093994140625e-05,
      "model_forward_time": 3.1431055068969727,
      "step": 9018
    },
    {
      "epoch": 3.440093994140625e-05,
      "step": 9018,
      "training_step_time": 9.917572736740112
    },
    {
      "epoch": 3.440093994140625e-05,
      "model_forward_time": 3.148533821105957,
      "step": 9018
    },
    {
      "epoch": 3.440093994140625e-05,
      "step": 9018,
      "training_step_time": 9.926759243011475
    },
    {
      "epoch": 3.440093994140625e-05,
      "model_forward_time": 3.144444227218628,
      "step": 9018
    },
    {
      "epoch": 3.440093994140625e-05,
      "step": 9018,
      "training_step_time": 10.317630767822266
    },
    {
      "epoch": 3.4404754638671875e-05,
      "model_forward_time": 2.451718807220459,
      "step": 9019
    },
    {
      "epoch": 3.4404754638671875e-05,
      "step": 9019,
      "training_step_time": 7.485600709915161
    },
    {
      "epoch": 3.4404754638671875e-05,
      "model_forward_time": 3.1385698318481445,
      "step": 9019
    },
    {
      "epoch": 3.4404754638671875e-05,
      "step": 9019,
      "training_step_time": 9.912172555923462
    },
    {
      "epoch": 3.4404754638671875e-05,
      "model_forward_time": 3.1438865661621094,
      "step": 9019
    },
    {
      "epoch": 3.4404754638671875e-05,
      "step": 9019,
      "training_step_time": 9.919495344161987
    },
    {
      "epoch": 3.4404754638671875e-05,
      "model_forward_time": 3.144932270050049,
      "step": 9019
    },
    {
      "epoch": 3.4404754638671875e-05,
      "step": 9019,
      "training_step_time": 10.23133111000061
    },
    {
      "epoch": 3.44085693359375e-05,
      "grad_norm": 0.4454194903373718,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9020
    },
    {
      "epoch": 3.44085693359375e-05,
      "model_forward_time": 3.431140661239624,
      "step": 9020
    },
    {
      "dynamics_loss_avg": 0.06944362316280603,
      "epoch": 3.44085693359375e-05,
      "step": 9020
    },
    {
      "action_loss_avg": 0.023194733448326586,
      "epoch": 3.44085693359375e-05,
      "step": 9020
    },
    {
      "epoch": 3.44085693359375e-05,
      "step": 9020,
      "training_step_time": 10.205803632736206
    },
    {
      "epoch": 3.44085693359375e-05,
      "model_forward_time": 3.142914295196533,
      "step": 9020
    },
    {
      "epoch": 3.44085693359375e-05,
      "step": 9020,
      "training_step_time": 9.920664072036743
    },
    {
      "epoch": 3.44085693359375e-05,
      "model_forward_time": 3.151068687438965,
      "step": 9020
    },
    {
      "epoch": 3.44085693359375e-05,
      "step": 9020,
      "training_step_time": 9.930078983306885
    },
    {
      "epoch": 3.44085693359375e-05,
      "model_forward_time": 3.14534854888916,
      "step": 9020
    },
    {
      "epoch": 3.44085693359375e-05,
      "step": 9020,
      "training_step_time": 10.021748304367065
    },
    {
      "epoch": 3.441238403320313e-05,
      "model_forward_time": 3.1657931804656982,
      "step": 9021
    },
    {
      "epoch": 3.441238403320313e-05,
      "step": 9021,
      "training_step_time": 9.94123649597168
    },
    {
      "epoch": 3.441238403320313e-05,
      "model_forward_time": 2.43880033493042,
      "step": 9021
    },
    {
      "epoch": 3.441238403320313e-05,
      "step": 9021,
      "training_step_time": 7.785396099090576
    },
    {
      "epoch": 3.441238403320313e-05,
      "model_forward_time": 3.1430318355560303,
      "step": 9021
    },
    {
      "epoch": 3.441238403320313e-05,
      "step": 9021,
      "training_step_time": 9.915018558502197
    },
    {
      "epoch": 3.441238403320313e-05,
      "model_forward_time": 3.1442718505859375,
      "step": 9021
    },
    {
      "epoch": 3.441238403320313e-05,
      "step": 9021,
      "training_step_time": 10.269431829452515
    },
    {
      "epoch": 3.441619873046875e-05,
      "model_forward_time": 3.1703150272369385,
      "step": 9022
    },
    {
      "epoch": 3.441619873046875e-05,
      "step": 9022,
      "training_step_time": 9.952869892120361
    },
    {
      "epoch": 3.441619873046875e-05,
      "model_forward_time": 3.150388240814209,
      "step": 9022
    },
    {
      "epoch": 3.441619873046875e-05,
      "step": 9022,
      "training_step_time": 9.927828073501587
    },
    {
      "epoch": 3.441619873046875e-05,
      "model_forward_time": 3.4361913204193115,
      "step": 9022
    },
    {
      "dynamics_loss_avg": 0.048569814674556254,
      "epoch": 3.441619873046875e-05,
      "step": 9022
    },
    {
      "action_loss_avg": 0.013709906814619898,
      "epoch": 3.441619873046875e-05,
      "step": 9022
    },
    {
      "epoch": 3.441619873046875e-05,
      "step": 9022,
      "training_step_time": 10.216086626052856
    },
    {
      "epoch": 3.441619873046875e-05,
      "model_forward_time": 3.1449356079101562,
      "step": 9022
    },
    {
      "epoch": 3.441619873046875e-05,
      "step": 9022,
      "training_step_time": 10.01659607887268
    },
    {
      "epoch": 3.4420013427734374e-05,
      "model_forward_time": 3.175071954727173,
      "step": 9023
    },
    {
      "epoch": 3.4420013427734374e-05,
      "step": 9023,
      "training_step_time": 9.957730770111084
    },
    {
      "epoch": 3.4420013427734374e-05,
      "model_forward_time": 2.4344780445098877,
      "step": 9023
    },
    {
      "epoch": 3.4420013427734374e-05,
      "step": 9023,
      "training_step_time": 7.468569993972778
    },
    {
      "epoch": 3.4420013427734374e-05,
      "model_forward_time": 3.1401209831237793,
      "step": 9023
    },
    {
      "epoch": 3.4420013427734374e-05,
      "step": 9023,
      "training_step_time": 9.916496992111206
    },
    {
      "epoch": 3.4420013427734374e-05,
      "model_forward_time": 3.4039700031280518,
      "step": 9023
    },
    {
      "epoch": 3.4420013427734374e-05,
      "step": 9023,
      "training_step_time": 10.271319150924683
    },
    {
      "epoch": 3.4423828125e-05,
      "model_forward_time": 3.1533379554748535,
      "step": 9024
    },
    {
      "epoch": 3.4423828125e-05,
      "step": 9024,
      "training_step_time": 9.93464732170105
    },
    {
      "epoch": 3.4423828125e-05,
      "model_forward_time": 3.14516544342041,
      "step": 9024
    },
    {
      "epoch": 3.4423828125e-05,
      "step": 9024,
      "training_step_time": 9.944470167160034
    },
    {
      "epoch": 3.4423828125e-05,
      "model_forward_time": 3.1342973709106445,
      "step": 9024
    },
    {
      "epoch": 3.4423828125e-05,
      "step": 9024,
      "training_step_time": 9.908068418502808
    },
    {
      "epoch": 3.4423828125e-05,
      "model_forward_time": 2.436504364013672,
      "step": 9024
    },
    {
      "epoch": 3.4423828125e-05,
      "step": 9024,
      "training_step_time": 10.281009674072266
    },
    {
      "epoch": 3.4427642822265626e-05,
      "model_forward_time": 3.144021511077881,
      "step": 9025
    },
    {
      "dynamics_loss_avg": 0.06662099808454514,
      "epoch": 3.4427642822265626e-05,
      "step": 9025
    },
    {
      "action_loss_avg": 0.023369285510852933,
      "epoch": 3.4427642822265626e-05,
      "step": 9025
    },
    {
      "epoch": 3.4427642822265626e-05,
      "step": 9025,
      "training_step_time": 10.22045636177063
    },
    {
      "epoch": 3.4427642822265626e-05,
      "model_forward_time": 3.147390127182007,
      "step": 9025
    },
    {
      "epoch": 3.4427642822265626e-05,
      "step": 9025,
      "training_step_time": 9.928886651992798
    },
    {
      "epoch": 3.4427642822265626e-05,
      "model_forward_time": 3.1394035816192627,
      "step": 9025
    },
    {
      "epoch": 3.4427642822265626e-05,
      "step": 9025,
      "training_step_time": 9.912120819091797
    },
    {
      "epoch": 3.4427642822265626e-05,
      "model_forward_time": 3.145163059234619,
      "step": 9025
    },
    {
      "epoch": 3.4427642822265626e-05,
      "step": 9025,
      "training_step_time": 10.34603476524353
    },
    {
      "epoch": 3.443145751953125e-05,
      "model_forward_time": 3.150895118713379,
      "step": 9026
    },
    {
      "epoch": 3.443145751953125e-05,
      "step": 9026,
      "training_step_time": 9.927310705184937
    },
    {
      "epoch": 3.443145751953125e-05,
      "model_forward_time": 3.4244794845581055,
      "step": 9026
    },
    {
      "epoch": 3.443145751953125e-05,
      "step": 9026,
      "training_step_time": 10.203745365142822
    },
    {
      "epoch": 3.443145751953125e-05,
      "model_forward_time": 3.1397817134857178,
      "step": 9026
    },
    {
      "epoch": 3.443145751953125e-05,
      "step": 9026,
      "training_step_time": 9.91526985168457
    },
    {
      "epoch": 3.443145751953125e-05,
      "model_forward_time": 3.140573263168335,
      "step": 9026
    },
    {
      "epoch": 3.443145751953125e-05,
      "step": 9026,
      "training_step_time": 10.012005805969238
    },
    {
      "epoch": 3.443527221679687e-05,
      "model_forward_time": 3.1525866985321045,
      "step": 9027
    },
    {
      "epoch": 3.443527221679687e-05,
      "step": 9027,
      "training_step_time": 9.930469036102295
    },
    {
      "epoch": 3.443527221679687e-05,
      "model_forward_time": 3.1353678703308105,
      "step": 9027
    },
    {
      "epoch": 3.443527221679687e-05,
      "step": 9027,
      "training_step_time": 9.910598039627075
    },
    {
      "epoch": 3.443527221679687e-05,
      "model_forward_time": 3.4050426483154297,
      "step": 9027
    },
    {
      "dynamics_loss_avg": 0.05641684755682945,
      "epoch": 3.443527221679687e-05,
      "step": 9027
    },
    {
      "action_loss_avg": 0.02153294449672103,
      "epoch": 3.443527221679687e-05,
      "step": 9027
    },
    {
      "epoch": 3.443527221679687e-05,
      "step": 9027,
      "training_step_time": 10.188412189483643
    },
    {
      "epoch": 3.443527221679687e-05,
      "model_forward_time": 3.1379518508911133,
      "step": 9027
    },
    {
      "epoch": 3.443527221679687e-05,
      "step": 9027,
      "training_step_time": 10.2758469581604
    },
    {
      "epoch": 3.44390869140625e-05,
      "model_forward_time": 3.1457176208496094,
      "step": 9028
    },
    {
      "epoch": 3.44390869140625e-05,
      "step": 9028,
      "training_step_time": 9.920842170715332
    },
    {
      "epoch": 3.44390869140625e-05,
      "model_forward_time": 3.1442980766296387,
      "step": 9028
    },
    {
      "epoch": 3.44390869140625e-05,
      "step": 9028,
      "training_step_time": 9.920514583587646
    },
    {
      "epoch": 3.44390869140625e-05,
      "model_forward_time": 2.4453744888305664,
      "step": 9028
    },
    {
      "epoch": 3.44390869140625e-05,
      "step": 9028,
      "training_step_time": 7.483034610748291
    },
    {
      "epoch": 3.44390869140625e-05,
      "model_forward_time": 3.470741033554077,
      "step": 9028
    },
    {
      "epoch": 3.44390869140625e-05,
      "step": 9028,
      "training_step_time": 10.345369815826416
    },
    {
      "epoch": 3.4442901611328125e-05,
      "model_forward_time": 3.1638474464416504,
      "step": 9029
    },
    {
      "epoch": 3.4442901611328125e-05,
      "step": 9029,
      "training_step_time": 9.944695711135864
    },
    {
      "epoch": 3.4442901611328125e-05,
      "model_forward_time": 2.4469103813171387,
      "step": 9029
    },
    {
      "epoch": 3.4442901611328125e-05,
      "step": 9029,
      "training_step_time": 7.484285354614258
    },
    {
      "epoch": 3.4442901611328125e-05,
      "model_forward_time": 3.1536056995391846,
      "step": 9029
    },
    {
      "epoch": 3.4442901611328125e-05,
      "step": 9029,
      "training_step_time": 9.934858322143555
    },
    {
      "epoch": 3.4442901611328125e-05,
      "model_forward_time": 2.4399538040161133,
      "step": 9029
    },
    {
      "epoch": 3.4442901611328125e-05,
      "step": 9029,
      "training_step_time": 10.323269844055176
    },
    {
      "epoch": 3.444671630859375e-05,
      "grad_norm": 0.3851524889469147,
      "learning_rate": 1e-05,
      "loss": 0.0778,
      "step": 9030
    },
    {
      "epoch": 3.444671630859375e-05,
      "model_forward_time": 3.1584513187408447,
      "step": 9030
    },
    {
      "dynamics_loss_avg": 0.047721843048930165,
      "epoch": 3.444671630859375e-05,
      "step": 9030
    },
    {
      "action_loss_avg": 0.014632371487095952,
      "epoch": 3.444671630859375e-05,
      "step": 9030
    },
    {
      "epoch": 3.444671630859375e-05,
      "step": 9030,
      "training_step_time": 9.949158906936646
    },
    {
      "epoch": 3.444671630859375e-05,
      "model_forward_time": 3.520782232284546,
      "step": 9030
    },
    {
      "epoch": 3.444671630859375e-05,
      "step": 9030,
      "training_step_time": 10.299048900604248
    },
    {
      "epoch": 3.444671630859375e-05,
      "model_forward_time": 3.145582675933838,
      "step": 9030
    },
    {
      "epoch": 3.444671630859375e-05,
      "step": 9030,
      "training_step_time": 9.92437744140625
    },
    {
      "epoch": 3.444671630859375e-05,
      "model_forward_time": 3.1410458087921143,
      "step": 9030
    },
    {
      "epoch": 3.444671630859375e-05,
      "step": 9030,
      "training_step_time": 10.294784545898438
    },
    {
      "epoch": 3.445053100585938e-05,
      "model_forward_time": 3.1706504821777344,
      "step": 9031
    },
    {
      "epoch": 3.445053100585938e-05,
      "step": 9031,
      "training_step_time": 9.95442509651184
    },
    {
      "epoch": 3.445053100585938e-05,
      "model_forward_time": 3.1448588371276855,
      "step": 9031
    },
    {
      "epoch": 3.445053100585938e-05,
      "step": 9031,
      "training_step_time": 10.208407640457153
    },
    {
      "epoch": 3.445053100585938e-05,
      "model_forward_time": 2.4359664916992188,
      "step": 9031
    },
    {
      "epoch": 3.445053100585938e-05,
      "step": 9031,
      "training_step_time": 7.473303556442261
    },
    {
      "epoch": 3.445053100585938e-05,
      "model_forward_time": 3.1366448402404785,
      "step": 9031
    },
    {
      "epoch": 3.445053100585938e-05,
      "step": 9031,
      "training_step_time": 10.22459363937378
    },
    {
      "epoch": 3.4454345703125e-05,
      "model_forward_time": 3.1630072593688965,
      "step": 9032
    },
    {
      "epoch": 3.4454345703125e-05,
      "step": 9032,
      "training_step_time": 9.946977376937866
    },
    {
      "epoch": 3.4454345703125e-05,
      "model_forward_time": 2.4375412464141846,
      "step": 9032
    },
    {
      "epoch": 3.4454345703125e-05,
      "step": 9032,
      "training_step_time": 7.46778130531311
    },
    {
      "epoch": 3.4454345703125e-05,
      "model_forward_time": 2.4344866275787354,
      "step": 9032
    },
    {
      "dynamics_loss_avg": 0.07240967079997063,
      "epoch": 3.4454345703125e-05,
      "step": 9032
    },
    {
      "action_loss_avg": 0.028554277680814266,
      "epoch": 3.4454345703125e-05,
      "step": 9032
    },
    {
      "epoch": 3.4454345703125e-05,
      "step": 9032,
      "training_step_time": 7.466267824172974
    },
    {
      "epoch": 3.4454345703125e-05,
      "model_forward_time": 3.426137924194336,
      "step": 9032
    },
    {
      "epoch": 3.4454345703125e-05,
      "step": 9032,
      "training_step_time": 10.294575929641724
    },
    {
      "epoch": 3.445816040039062e-05,
      "model_forward_time": 3.161367177963257,
      "step": 9033
    },
    {
      "epoch": 3.445816040039062e-05,
      "step": 9033,
      "training_step_time": 9.940603256225586
    },
    {
      "epoch": 3.445816040039062e-05,
      "model_forward_time": 3.1389479637145996,
      "step": 9033
    },
    {
      "epoch": 3.445816040039062e-05,
      "step": 9033,
      "training_step_time": 9.914967060089111
    },
    {
      "epoch": 3.445816040039062e-05,
      "model_forward_time": 3.1419811248779297,
      "step": 9033
    },
    {
      "epoch": 3.445816040039062e-05,
      "step": 9033,
      "training_step_time": 9.921741724014282
    },
    {
      "epoch": 3.445816040039062e-05,
      "model_forward_time": 2.4389727115631104,
      "step": 9033
    },
    {
      "epoch": 3.445816040039062e-05,
      "step": 9033,
      "training_step_time": 10.285825252532959
    },
    {
      "epoch": 3.446197509765625e-05,
      "model_forward_time": 3.4434304237365723,
      "step": 9034
    },
    {
      "epoch": 3.446197509765625e-05,
      "step": 9034,
      "training_step_time": 10.223642349243164
    },
    {
      "epoch": 3.446197509765625e-05,
      "model_forward_time": 2.437664747238159,
      "step": 9034
    },
    {
      "epoch": 3.446197509765625e-05,
      "step": 9034,
      "training_step_time": 7.472321033477783
    },
    {
      "epoch": 3.446197509765625e-05,
      "model_forward_time": 3.140854597091675,
      "step": 9034
    },
    {
      "epoch": 3.446197509765625e-05,
      "step": 9034,
      "training_step_time": 9.917174100875854
    },
    {
      "epoch": 3.446197509765625e-05,
      "model_forward_time": 2.4348647594451904,
      "step": 9034
    },
    {
      "epoch": 3.446197509765625e-05,
      "step": 9034,
      "training_step_time": 10.015707969665527
    },
    {
      "epoch": 3.4465789794921876e-05,
      "model_forward_time": 3.1533875465393066,
      "step": 9035
    },
    {
      "dynamics_loss_avg": 0.046386593393981455,
      "epoch": 3.4465789794921876e-05,
      "step": 9035
    },
    {
      "action_loss_avg": 0.01692160894162953,
      "epoch": 3.4465789794921876e-05,
      "step": 9035
    },
    {
      "epoch": 3.4465789794921876e-05,
      "step": 9035,
      "training_step_time": 9.934087753295898
    },
    {
      "epoch": 3.4465789794921876e-05,
      "model_forward_time": 3.4119505882263184,
      "step": 9035
    },
    {
      "epoch": 3.4465789794921876e-05,
      "step": 9035,
      "training_step_time": 10.187869548797607
    },
    {
      "epoch": 3.4465789794921876e-05,
      "model_forward_time": 3.13507080078125,
      "step": 9035
    },
    {
      "epoch": 3.4465789794921876e-05,
      "step": 9035,
      "training_step_time": 9.904088497161865
    },
    {
      "epoch": 3.4465789794921876e-05,
      "model_forward_time": 3.1433143615722656,
      "step": 9035
    },
    {
      "epoch": 3.4465789794921876e-05,
      "step": 9035,
      "training_step_time": 10.271878242492676
    },
    {
      "epoch": 3.44696044921875e-05,
      "model_forward_time": 3.153796911239624,
      "step": 9036
    },
    {
      "epoch": 3.44696044921875e-05,
      "step": 9036,
      "training_step_time": 9.932643413543701
    },
    {
      "epoch": 3.44696044921875e-05,
      "model_forward_time": 3.1401376724243164,
      "step": 9036
    },
    {
      "epoch": 3.44696044921875e-05,
      "step": 9036,
      "training_step_time": 9.914899826049805
    },
    {
      "epoch": 3.44696044921875e-05,
      "model_forward_time": 3.411858320236206,
      "step": 9036
    },
    {
      "epoch": 3.44696044921875e-05,
      "step": 9036,
      "training_step_time": 10.185664176940918
    },
    {
      "epoch": 3.44696044921875e-05,
      "model_forward_time": 3.1471168994903564,
      "step": 9036
    },
    {
      "epoch": 3.44696044921875e-05,
      "step": 9036,
      "training_step_time": 10.251877546310425
    },
    {
      "epoch": 3.447341918945313e-05,
      "model_forward_time": 3.160771369934082,
      "step": 9037
    },
    {
      "epoch": 3.447341918945313e-05,
      "step": 9037,
      "training_step_time": 9.941967010498047
    },
    {
      "epoch": 3.447341918945313e-05,
      "model_forward_time": 3.146153450012207,
      "step": 9037
    },
    {
      "epoch": 3.447341918945313e-05,
      "step": 9037,
      "training_step_time": 9.928766012191772
    },
    {
      "epoch": 3.447341918945313e-05,
      "model_forward_time": 3.154304265975952,
      "step": 9037
    },
    {
      "dynamics_loss_avg": 0.05812468435615301,
      "epoch": 3.447341918945313e-05,
      "step": 9037
    },
    {
      "action_loss_avg": 0.019546891539357604,
      "epoch": 3.447341918945313e-05,
      "step": 9037
    },
    {
      "epoch": 3.447341918945313e-05,
      "step": 9037,
      "training_step_time": 10.243988990783691
    },
    {
      "epoch": 3.447341918945313e-05,
      "model_forward_time": 3.1586496829986572,
      "step": 9037
    },
    {
      "epoch": 3.447341918945313e-05,
      "step": 9037,
      "training_step_time": 10.312355995178223
    },
    {
      "epoch": 3.447723388671875e-05,
      "model_forward_time": 3.164555549621582,
      "step": 9038
    },
    {
      "epoch": 3.447723388671875e-05,
      "step": 9038,
      "training_step_time": 9.949932098388672
    },
    {
      "epoch": 3.447723388671875e-05,
      "model_forward_time": 3.145625352859497,
      "step": 9038
    },
    {
      "epoch": 3.447723388671875e-05,
      "step": 9038,
      "training_step_time": 9.92333459854126
    },
    {
      "epoch": 3.447723388671875e-05,
      "model_forward_time": 3.148864984512329,
      "step": 9038
    },
    {
      "epoch": 3.447723388671875e-05,
      "step": 9038,
      "training_step_time": 9.928485870361328
    },
    {
      "epoch": 3.447723388671875e-05,
      "model_forward_time": 2.4393818378448486,
      "step": 9038
    },
    {
      "epoch": 3.447723388671875e-05,
      "step": 9038,
      "training_step_time": 10.324520826339722
    },
    {
      "epoch": 3.4481048583984374e-05,
      "model_forward_time": 3.457249402999878,
      "step": 9039
    },
    {
      "epoch": 3.4481048583984374e-05,
      "step": 9039,
      "training_step_time": 10.237072706222534
    },
    {
      "epoch": 3.4481048583984374e-05,
      "model_forward_time": 2.436849355697632,
      "step": 9039
    },
    {
      "epoch": 3.4481048583984374e-05,
      "step": 9039,
      "training_step_time": 7.4678053855896
    },
    {
      "epoch": 3.4481048583984374e-05,
      "model_forward_time": 3.1496071815490723,
      "step": 9039
    },
    {
      "epoch": 3.4481048583984374e-05,
      "step": 9039,
      "training_step_time": 9.928220510482788
    },
    {
      "epoch": 3.4481048583984374e-05,
      "model_forward_time": 3.1485211849212646,
      "step": 9039
    },
    {
      "epoch": 3.4481048583984374e-05,
      "step": 9039,
      "training_step_time": 10.023158073425293
    },
    {
      "epoch": 3.448486328125e-05,
      "grad_norm": 0.576591432094574,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 9040
    },
    {
      "epoch": 3.448486328125e-05,
      "model_forward_time": 3.4476921558380127,
      "step": 9040
    },
    {
      "dynamics_loss_avg": 0.06338572259992362,
      "epoch": 3.448486328125e-05,
      "step": 9040
    },
    {
      "action_loss_avg": 0.023276973608881237,
      "epoch": 3.448486328125e-05,
      "step": 9040
    },
    {
      "epoch": 3.448486328125e-05,
      "step": 9040,
      "training_step_time": 10.239991188049316
    },
    {
      "epoch": 3.448486328125e-05,
      "model_forward_time": 3.144747018814087,
      "step": 9040
    },
    {
      "epoch": 3.448486328125e-05,
      "step": 9040,
      "training_step_time": 9.919632911682129
    },
    {
      "epoch": 3.448486328125e-05,
      "model_forward_time": 3.1450061798095703,
      "step": 9040
    },
    {
      "epoch": 3.448486328125e-05,
      "step": 9040,
      "training_step_time": 9.923527717590332
    },
    {
      "epoch": 3.448486328125e-05,
      "model_forward_time": 3.145150661468506,
      "step": 9040
    },
    {
      "epoch": 3.448486328125e-05,
      "step": 9040,
      "training_step_time": 10.233259677886963
    },
    {
      "epoch": 3.4488677978515626e-05,
      "model_forward_time": 3.1640238761901855,
      "step": 9041
    },
    {
      "epoch": 3.4488677978515626e-05,
      "step": 9041,
      "training_step_time": 9.941734552383423
    },
    {
      "epoch": 3.4488677978515626e-05,
      "model_forward_time": 3.1468899250030518,
      "step": 9041
    },
    {
      "epoch": 3.4488677978515626e-05,
      "step": 9041,
      "training_step_time": 9.920226335525513
    },
    {
      "epoch": 3.4488677978515626e-05,
      "model_forward_time": 3.431419610977173,
      "step": 9041
    },
    {
      "epoch": 3.4488677978515626e-05,
      "step": 9041,
      "training_step_time": 10.207537412643433
    },
    {
      "epoch": 3.4488677978515626e-05,
      "model_forward_time": 3.146383047103882,
      "step": 9041
    },
    {
      "epoch": 3.4488677978515626e-05,
      "step": 9041,
      "training_step_time": 10.236318826675415
    },
    {
      "epoch": 3.449249267578125e-05,
      "model_forward_time": 3.1694095134735107,
      "step": 9042
    },
    {
      "epoch": 3.449249267578125e-05,
      "step": 9042,
      "training_step_time": 9.947770833969116
    },
    {
      "epoch": 3.449249267578125e-05,
      "model_forward_time": 3.1541450023651123,
      "step": 9042
    },
    {
      "epoch": 3.449249267578125e-05,
      "step": 9042,
      "training_step_time": 9.93622612953186
    },
    {
      "epoch": 3.449249267578125e-05,
      "model_forward_time": 3.471280813217163,
      "step": 9042
    },
    {
      "dynamics_loss_avg": 0.051082057133316996,
      "epoch": 3.449249267578125e-05,
      "step": 9042
    },
    {
      "action_loss_avg": 0.020318445563316346,
      "epoch": 3.449249267578125e-05,
      "step": 9042
    },
    {
      "epoch": 3.449249267578125e-05,
      "step": 9042,
      "training_step_time": 10.252171993255615
    },
    {
      "epoch": 3.449249267578125e-05,
      "model_forward_time": 3.146348476409912,
      "step": 9042
    },
    {
      "epoch": 3.449249267578125e-05,
      "step": 9042,
      "training_step_time": 10.328216552734375
    },
    {
      "epoch": 3.449630737304687e-05,
      "model_forward_time": 3.1577906608581543,
      "step": 9043
    },
    {
      "epoch": 3.449630737304687e-05,
      "step": 9043,
      "training_step_time": 9.93923807144165
    },
    {
      "epoch": 3.449630737304687e-05,
      "model_forward_time": 3.151512384414673,
      "step": 9043
    },
    {
      "epoch": 3.449630737304687e-05,
      "step": 9043,
      "training_step_time": 9.92828893661499
    },
    {
      "epoch": 3.449630737304687e-05,
      "model_forward_time": 3.1502578258514404,
      "step": 9043
    },
    {
      "epoch": 3.449630737304687e-05,
      "step": 9043,
      "training_step_time": 9.932222127914429
    },
    {
      "epoch": 3.449630737304687e-05,
      "model_forward_time": 3.4134440422058105,
      "step": 9043
    },
    {
      "epoch": 3.449630737304687e-05,
      "step": 9043,
      "training_step_time": 10.32203221321106
    },
    {
      "epoch": 3.45001220703125e-05,
      "model_forward_time": 3.162541151046753,
      "step": 9044
    },
    {
      "epoch": 3.45001220703125e-05,
      "step": 9044,
      "training_step_time": 9.947826862335205
    },
    {
      "epoch": 3.45001220703125e-05,
      "model_forward_time": 2.439854145050049,
      "step": 9044
    },
    {
      "epoch": 3.45001220703125e-05,
      "step": 9044,
      "training_step_time": 7.470581531524658
    },
    {
      "epoch": 3.45001220703125e-05,
      "model_forward_time": 3.1395132541656494,
      "step": 9044
    },
    {
      "epoch": 3.45001220703125e-05,
      "step": 9044,
      "training_step_time": 9.918081760406494
    },
    {
      "epoch": 3.45001220703125e-05,
      "model_forward_time": 3.145439386367798,
      "step": 9044
    },
    {
      "epoch": 3.45001220703125e-05,
      "step": 9044,
      "training_step_time": 10.310763359069824
    },
    {
      "epoch": 3.4503936767578125e-05,
      "model_forward_time": 3.1492137908935547,
      "step": 9045
    },
    {
      "dynamics_loss_avg": 0.05835157446563244,
      "epoch": 3.4503936767578125e-05,
      "step": 9045
    },
    {
      "action_loss_avg": 0.021939523704349994,
      "epoch": 3.4503936767578125e-05,
      "step": 9045
    },
    {
      "epoch": 3.4503936767578125e-05,
      "step": 9045,
      "training_step_time": 10.21701955795288
    },
    {
      "epoch": 3.4503936767578125e-05,
      "model_forward_time": 2.4381632804870605,
      "step": 9045
    },
    {
      "epoch": 3.4503936767578125e-05,
      "step": 9045,
      "training_step_time": 7.464498043060303
    },
    {
      "epoch": 3.4503936767578125e-05,
      "model_forward_time": 3.1376726627349854,
      "step": 9045
    },
    {
      "epoch": 3.4503936767578125e-05,
      "step": 9045,
      "training_step_time": 9.91744327545166
    },
    {
      "epoch": 3.4503936767578125e-05,
      "model_forward_time": 3.13942289352417,
      "step": 9045
    },
    {
      "epoch": 3.4503936767578125e-05,
      "step": 9045,
      "training_step_time": 10.015065908432007
    },
    {
      "epoch": 3.450775146484375e-05,
      "model_forward_time": 3.151163101196289,
      "step": 9046
    },
    {
      "epoch": 3.450775146484375e-05,
      "step": 9046,
      "training_step_time": 9.927521228790283
    },
    {
      "epoch": 3.450775146484375e-05,
      "model_forward_time": 3.1312878131866455,
      "step": 9046
    },
    {
      "epoch": 3.450775146484375e-05,
      "step": 9046,
      "training_step_time": 10.176963329315186
    },
    {
      "epoch": 3.450775146484375e-05,
      "model_forward_time": 3.1325461864471436,
      "step": 9046
    },
    {
      "epoch": 3.450775146484375e-05,
      "step": 9046,
      "training_step_time": 9.902551651000977
    },
    {
      "epoch": 3.450775146484375e-05,
      "model_forward_time": 3.138794183731079,
      "step": 9046
    },
    {
      "epoch": 3.450775146484375e-05,
      "step": 9046,
      "training_step_time": 10.01758599281311
    },
    {
      "epoch": 3.451156616210938e-05,
      "model_forward_time": 3.156651496887207,
      "step": 9047
    },
    {
      "epoch": 3.451156616210938e-05,
      "step": 9047,
      "training_step_time": 9.935630083084106
    },
    {
      "epoch": 3.451156616210938e-05,
      "model_forward_time": 3.1417200565338135,
      "step": 9047
    },
    {
      "epoch": 3.451156616210938e-05,
      "step": 9047,
      "training_step_time": 9.914552927017212
    },
    {
      "epoch": 3.451156616210938e-05,
      "model_forward_time": 3.141512632369995,
      "step": 9047
    },
    {
      "dynamics_loss_avg": 0.05475715324282646,
      "epoch": 3.451156616210938e-05,
      "step": 9047
    },
    {
      "action_loss_avg": 0.01801108461804688,
      "epoch": 3.451156616210938e-05,
      "step": 9047
    },
    {
      "epoch": 3.451156616210938e-05,
      "step": 9047,
      "training_step_time": 10.19465446472168
    },
    {
      "epoch": 3.451156616210938e-05,
      "model_forward_time": 3.141608715057373,
      "step": 9047
    },
    {
      "epoch": 3.451156616210938e-05,
      "step": 9047,
      "training_step_time": 10.264371156692505
    },
    {
      "epoch": 3.4515380859375e-05,
      "model_forward_time": 3.1547329425811768,
      "step": 9048
    },
    {
      "epoch": 3.4515380859375e-05,
      "step": 9048,
      "training_step_time": 9.934414625167847
    },
    {
      "epoch": 3.4515380859375e-05,
      "model_forward_time": 3.136902332305908,
      "step": 9048
    },
    {
      "epoch": 3.4515380859375e-05,
      "step": 9048,
      "training_step_time": 9.918222188949585
    },
    {
      "epoch": 3.4515380859375e-05,
      "model_forward_time": 3.1451475620269775,
      "step": 9048
    },
    {
      "epoch": 3.4515380859375e-05,
      "step": 9048,
      "training_step_time": 9.924792766571045
    },
    {
      "epoch": 3.4515380859375e-05,
      "model_forward_time": 2.722849130630493,
      "step": 9048
    },
    {
      "epoch": 3.4515380859375e-05,
      "step": 9048,
      "training_step_time": 10.011339902877808
    },
    {
      "epoch": 3.451919555664062e-05,
      "model_forward_time": 2.4447174072265625,
      "step": 9049
    },
    {
      "epoch": 3.451919555664062e-05,
      "step": 9049,
      "training_step_time": 7.47890305519104
    },
    {
      "epoch": 3.451919555664062e-05,
      "model_forward_time": 3.1372182369232178,
      "step": 9049
    },
    {
      "epoch": 3.451919555664062e-05,
      "step": 9049,
      "training_step_time": 9.918514728546143
    },
    {
      "epoch": 3.451919555664062e-05,
      "model_forward_time": 3.1469666957855225,
      "step": 9049
    },
    {
      "epoch": 3.451919555664062e-05,
      "step": 9049,
      "training_step_time": 9.924275159835815
    },
    {
      "epoch": 3.451919555664062e-05,
      "model_forward_time": 2.4378862380981445,
      "step": 9049
    },
    {
      "epoch": 3.451919555664062e-05,
      "step": 9049,
      "training_step_time": 10.024853706359863
    },
    {
      "epoch": 3.452301025390625e-05,
      "grad_norm": 0.5602498054504395,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 9050
    },
    {
      "epoch": 3.452301025390625e-05,
      "model_forward_time": 3.144683361053467,
      "step": 9050
    },
    {
      "dynamics_loss_avg": 0.07096138782799244,
      "epoch": 3.452301025390625e-05,
      "step": 9050
    },
    {
      "action_loss_avg": 0.02679763762280345,
      "epoch": 3.452301025390625e-05,
      "step": 9050
    },
    {
      "epoch": 3.452301025390625e-05,
      "step": 9050,
      "training_step_time": 9.926978588104248
    },
    {
      "epoch": 3.452301025390625e-05,
      "model_forward_time": 3.4249656200408936,
      "step": 9050
    },
    {
      "epoch": 3.452301025390625e-05,
      "step": 9050,
      "training_step_time": 10.202173948287964
    },
    {
      "epoch": 3.452301025390625e-05,
      "model_forward_time": 3.1466598510742188,
      "step": 9050
    },
    {
      "epoch": 3.452301025390625e-05,
      "step": 9050,
      "training_step_time": 9.929481029510498
    },
    {
      "epoch": 3.452301025390625e-05,
      "model_forward_time": 3.1404786109924316,
      "step": 9050
    },
    {
      "epoch": 3.452301025390625e-05,
      "step": 9050,
      "training_step_time": 10.025196075439453
    },
    {
      "epoch": 3.4526824951171876e-05,
      "model_forward_time": 1.7644853591918945,
      "step": 9051
    },
    {
      "epoch": 3.4526824951171876e-05,
      "step": 9051,
      "training_step_time": 5.247265100479126
    },
    {
      "epoch": 3.4526824951171876e-05,
      "model_forward_time": 3.1447415351867676,
      "step": 9051
    },
    {
      "epoch": 3.4526824951171876e-05,
      "step": 9051,
      "training_step_time": 9.926270961761475
    },
    {
      "epoch": 3.4526824951171876e-05,
      "model_forward_time": 3.4247615337371826,
      "step": 9051
    },
    {
      "epoch": 3.4526824951171876e-05,
      "step": 9051,
      "training_step_time": 10.199773788452148
    },
    {
      "epoch": 3.4526824951171876e-05,
      "model_forward_time": 3.138467311859131,
      "step": 9051
    },
    {
      "epoch": 3.4526824951171876e-05,
      "step": 9051,
      "training_step_time": 10.265939950942993
    },
    {
      "epoch": 3.45306396484375e-05,
      "model_forward_time": 3.1548776626586914,
      "step": 9052
    },
    {
      "epoch": 3.45306396484375e-05,
      "step": 9052,
      "training_step_time": 9.932414054870605
    },
    {
      "epoch": 3.45306396484375e-05,
      "model_forward_time": 3.1424670219421387,
      "step": 9052
    },
    {
      "epoch": 3.45306396484375e-05,
      "step": 9052,
      "training_step_time": 9.913819313049316
    },
    {
      "epoch": 3.45306396484375e-05,
      "model_forward_time": 3.137441873550415,
      "step": 9052
    },
    {
      "dynamics_loss_avg": 0.07114548739045859,
      "epoch": 3.45306396484375e-05,
      "step": 9052
    },
    {
      "action_loss_avg": 0.029169400921091437,
      "epoch": 3.45306396484375e-05,
      "step": 9052
    },
    {
      "epoch": 3.45306396484375e-05,
      "step": 9052,
      "training_step_time": 9.910288095474243
    },
    {
      "epoch": 3.45306396484375e-05,
      "model_forward_time": 3.4284534454345703,
      "step": 9052
    },
    {
      "epoch": 3.45306396484375e-05,
      "step": 9052,
      "training_step_time": 10.286761045455933
    },
    {
      "epoch": 3.453445434570313e-05,
      "model_forward_time": 3.143580913543701,
      "step": 9053
    },
    {
      "epoch": 3.453445434570313e-05,
      "step": 9053,
      "training_step_time": 9.908203125
    },
    {
      "epoch": 3.453445434570313e-05,
      "model_forward_time": 2.437321901321411,
      "step": 9053
    },
    {
      "epoch": 3.453445434570313e-05,
      "step": 9053,
      "training_step_time": 7.461045265197754
    },
    {
      "epoch": 3.453445434570313e-05,
      "model_forward_time": 3.138002395629883,
      "step": 9053
    },
    {
      "epoch": 3.453445434570313e-05,
      "step": 9053,
      "training_step_time": 9.912134408950806
    },
    {
      "epoch": 3.453445434570313e-05,
      "model_forward_time": 3.144428253173828,
      "step": 9053
    },
    {
      "epoch": 3.453445434570313e-05,
      "step": 9053,
      "training_step_time": 10.32065725326538
    },
    {
      "epoch": 3.453826904296875e-05,
      "model_forward_time": 3.4215750694274902,
      "step": 9054
    },
    {
      "epoch": 3.453826904296875e-05,
      "step": 9054,
      "training_step_time": 10.201945304870605
    },
    {
      "epoch": 3.453826904296875e-05,
      "model_forward_time": 3.1465609073638916,
      "step": 9054
    },
    {
      "epoch": 3.453826904296875e-05,
      "step": 9054,
      "training_step_time": 9.926519870758057
    },
    {
      "epoch": 3.453826904296875e-05,
      "model_forward_time": 3.149905204772949,
      "step": 9054
    },
    {
      "epoch": 3.453826904296875e-05,
      "step": 9054,
      "training_step_time": 9.923134326934814
    },
    {
      "epoch": 3.453826904296875e-05,
      "model_forward_time": 3.151792287826538,
      "step": 9054
    },
    {
      "epoch": 3.453826904296875e-05,
      "step": 9054,
      "training_step_time": 10.283017635345459
    },
    {
      "epoch": 3.4542083740234374e-05,
      "model_forward_time": 3.15334415435791,
      "step": 9055
    },
    {
      "dynamics_loss_avg": 0.062195473536849025,
      "epoch": 3.4542083740234374e-05,
      "step": 9055
    },
    {
      "action_loss_avg": 0.018959164340049028,
      "epoch": 3.4542083740234374e-05,
      "step": 9055
    },
    {
      "epoch": 3.4542083740234374e-05,
      "step": 9055,
      "training_step_time": 9.932135105133057
    },
    {
      "epoch": 3.4542083740234374e-05,
      "model_forward_time": 3.456295967102051,
      "step": 9055
    },
    {
      "epoch": 3.4542083740234374e-05,
      "step": 9055,
      "training_step_time": 10.235178232192993
    },
    {
      "epoch": 3.4542083740234374e-05,
      "model_forward_time": 3.1429691314697266,
      "step": 9055
    },
    {
      "epoch": 3.4542083740234374e-05,
      "step": 9055,
      "training_step_time": 9.925047397613525
    },
    {
      "epoch": 3.4542083740234374e-05,
      "model_forward_time": 3.143463373184204,
      "step": 9055
    },
    {
      "epoch": 3.4542083740234374e-05,
      "step": 9055,
      "training_step_time": 10.27381944656372
    },
    {
      "epoch": 3.45458984375e-05,
      "model_forward_time": 3.1529080867767334,
      "step": 9056
    },
    {
      "epoch": 3.45458984375e-05,
      "step": 9056,
      "training_step_time": 9.934496402740479
    },
    {
      "epoch": 3.45458984375e-05,
      "model_forward_time": 3.13781476020813,
      "step": 9056
    },
    {
      "epoch": 3.45458984375e-05,
      "step": 9056,
      "training_step_time": 9.91258716583252
    },
    {
      "epoch": 3.45458984375e-05,
      "model_forward_time": 3.3968260288238525,
      "step": 9056
    },
    {
      "epoch": 3.45458984375e-05,
      "step": 9056,
      "training_step_time": 10.172879219055176
    },
    {
      "epoch": 3.45458984375e-05,
      "model_forward_time": 3.1481597423553467,
      "step": 9056
    },
    {
      "epoch": 3.45458984375e-05,
      "step": 9056,
      "training_step_time": 10.037145376205444
    },
    {
      "epoch": 3.4549713134765626e-05,
      "model_forward_time": 3.1528592109680176,
      "step": 9057
    },
    {
      "epoch": 3.4549713134765626e-05,
      "step": 9057,
      "training_step_time": 9.927418231964111
    },
    {
      "epoch": 3.4549713134765626e-05,
      "model_forward_time": 3.1346476078033447,
      "step": 9057
    },
    {
      "epoch": 3.4549713134765626e-05,
      "step": 9057,
      "training_step_time": 9.902327060699463
    },
    {
      "epoch": 3.4549713134765626e-05,
      "model_forward_time": 3.430661201477051,
      "step": 9057
    },
    {
      "dynamics_loss_avg": 0.06898046676069498,
      "epoch": 3.4549713134765626e-05,
      "step": 9057
    },
    {
      "action_loss_avg": 0.024029491003602743,
      "epoch": 3.4549713134765626e-05,
      "step": 9057
    },
    {
      "epoch": 3.4549713134765626e-05,
      "step": 9057,
      "training_step_time": 10.206577777862549
    },
    {
      "epoch": 3.4549713134765626e-05,
      "model_forward_time": 3.1378631591796875,
      "step": 9057
    },
    {
      "epoch": 3.4549713134765626e-05,
      "step": 9057,
      "training_step_time": 10.292967796325684
    },
    {
      "epoch": 3.455352783203125e-05,
      "model_forward_time": 3.143190383911133,
      "step": 9058
    },
    {
      "epoch": 3.455352783203125e-05,
      "step": 9058,
      "training_step_time": 9.911454200744629
    },
    {
      "epoch": 3.455352783203125e-05,
      "model_forward_time": 3.132218837738037,
      "step": 9058
    },
    {
      "epoch": 3.455352783203125e-05,
      "step": 9058,
      "training_step_time": 9.9030442237854
    },
    {
      "epoch": 3.455352783203125e-05,
      "model_forward_time": 3.136873483657837,
      "step": 9058
    },
    {
      "epoch": 3.455352783203125e-05,
      "step": 9058,
      "training_step_time": 9.902831315994263
    },
    {
      "epoch": 3.455352783203125e-05,
      "model_forward_time": 3.1344401836395264,
      "step": 9058
    },
    {
      "epoch": 3.455352783203125e-05,
      "step": 9058,
      "training_step_time": 10.27041745185852
    },
    {
      "epoch": 3.455734252929687e-05,
      "model_forward_time": 3.1462249755859375,
      "step": 9059
    },
    {
      "epoch": 3.455734252929687e-05,
      "step": 9059,
      "training_step_time": 9.915159702301025
    },
    {
      "epoch": 3.455734252929687e-05,
      "model_forward_time": 3.138363838195801,
      "step": 9059
    },
    {
      "epoch": 3.455734252929687e-05,
      "step": 9059,
      "training_step_time": 9.910625696182251
    },
    {
      "epoch": 3.455734252929687e-05,
      "model_forward_time": 3.143686056137085,
      "step": 9059
    },
    {
      "epoch": 3.455734252929687e-05,
      "step": 9059,
      "training_step_time": 9.921175479888916
    },
    {
      "epoch": 3.455734252929687e-05,
      "model_forward_time": 3.143359661102295,
      "step": 9059
    },
    {
      "epoch": 3.455734252929687e-05,
      "step": 9059,
      "training_step_time": 10.29086446762085
    },
    {
      "epoch": 3.45611572265625e-05,
      "grad_norm": 0.33481040596961975,
      "learning_rate": 1e-05,
      "loss": 0.0782,
      "step": 9060
    },
    {
      "epoch": 3.45611572265625e-05,
      "model_forward_time": 2.4549546241760254,
      "step": 9060
    },
    {
      "dynamics_loss_avg": 0.06793668195605278,
      "epoch": 3.45611572265625e-05,
      "step": 9060
    },
    {
      "action_loss_avg": 0.024967469833791257,
      "epoch": 3.45611572265625e-05,
      "step": 9060
    },
    {
      "epoch": 3.45611572265625e-05,
      "step": 9060,
      "training_step_time": 7.768864154815674
    },
    {
      "epoch": 3.45611572265625e-05,
      "model_forward_time": 3.1325697898864746,
      "step": 9060
    },
    {
      "epoch": 3.45611572265625e-05,
      "step": 9060,
      "training_step_time": 9.904829263687134
    },
    {
      "epoch": 3.45611572265625e-05,
      "model_forward_time": 3.1460862159729004,
      "step": 9060
    },
    {
      "epoch": 3.45611572265625e-05,
      "step": 9060,
      "training_step_time": 9.919666290283203
    },
    {
      "epoch": 3.45611572265625e-05,
      "model_forward_time": 2.442323923110962,
      "step": 9060
    },
    {
      "epoch": 3.45611572265625e-05,
      "step": 9060,
      "training_step_time": 10.03922176361084
    },
    {
      "epoch": 3.4564971923828125e-05,
      "model_forward_time": 3.1562657356262207,
      "step": 9061
    },
    {
      "epoch": 3.4564971923828125e-05,
      "step": 9061,
      "training_step_time": 9.936895847320557
    },
    {
      "epoch": 3.4564971923828125e-05,
      "model_forward_time": 2.4409446716308594,
      "step": 9061
    },
    {
      "epoch": 3.4564971923828125e-05,
      "step": 9061,
      "training_step_time": 7.471007585525513
    },
    {
      "epoch": 3.4564971923828125e-05,
      "model_forward_time": 3.4532322883605957,
      "step": 9061
    },
    {
      "epoch": 3.4564971923828125e-05,
      "step": 9061,
      "training_step_time": 10.22295594215393
    },
    {
      "epoch": 3.4564971923828125e-05,
      "model_forward_time": 3.1449625492095947,
      "step": 9061
    },
    {
      "epoch": 3.4564971923828125e-05,
      "step": 9061,
      "training_step_time": 10.039345026016235
    },
    {
      "epoch": 3.456878662109375e-05,
      "model_forward_time": 3.176706552505493,
      "step": 9062
    },
    {
      "epoch": 3.456878662109375e-05,
      "step": 9062,
      "training_step_time": 9.971805810928345
    },
    {
      "epoch": 3.456878662109375e-05,
      "model_forward_time": 2.4480173587799072,
      "step": 9062
    },
    {
      "epoch": 3.456878662109375e-05,
      "step": 9062,
      "training_step_time": 7.487164497375488
    },
    {
      "epoch": 3.456878662109375e-05,
      "model_forward_time": 3.134706974029541,
      "step": 9062
    },
    {
      "dynamics_loss_avg": 0.05077347513288259,
      "epoch": 3.456878662109375e-05,
      "step": 9062
    },
    {
      "action_loss_avg": 0.018046999420039355,
      "epoch": 3.456878662109375e-05,
      "step": 9062
    },
    {
      "epoch": 3.456878662109375e-05,
      "step": 9062,
      "training_step_time": 9.91563868522644
    },
    {
      "epoch": 3.456878662109375e-05,
      "model_forward_time": 3.4031448364257812,
      "step": 9062
    },
    {
      "epoch": 3.456878662109375e-05,
      "step": 9062,
      "training_step_time": 10.275559902191162
    },
    {
      "epoch": 3.457260131835938e-05,
      "model_forward_time": 3.167684316635132,
      "step": 9063
    },
    {
      "epoch": 3.457260131835938e-05,
      "step": 9063,
      "training_step_time": 9.945197582244873
    },
    {
      "epoch": 3.457260131835938e-05,
      "model_forward_time": 3.1391525268554688,
      "step": 9063
    },
    {
      "epoch": 3.457260131835938e-05,
      "step": 9063,
      "training_step_time": 9.912975549697876
    },
    {
      "epoch": 3.457260131835938e-05,
      "model_forward_time": 3.1376843452453613,
      "step": 9063
    },
    {
      "epoch": 3.457260131835938e-05,
      "step": 9063,
      "training_step_time": 9.9098219871521
    },
    {
      "epoch": 3.457260131835938e-05,
      "model_forward_time": 2.4372692108154297,
      "step": 9063
    },
    {
      "epoch": 3.457260131835938e-05,
      "step": 9063,
      "training_step_time": 10.01756739616394
    },
    {
      "epoch": 3.4576416015625e-05,
      "model_forward_time": 3.146315813064575,
      "step": 9064
    },
    {
      "epoch": 3.4576416015625e-05,
      "step": 9064,
      "training_step_time": 9.917412042617798
    },
    {
      "epoch": 3.4576416015625e-05,
      "model_forward_time": 3.13710880279541,
      "step": 9064
    },
    {
      "epoch": 3.4576416015625e-05,
      "step": 9064,
      "training_step_time": 9.905316829681396
    },
    {
      "epoch": 3.4576416015625e-05,
      "model_forward_time": 3.13381290435791,
      "step": 9064
    },
    {
      "epoch": 3.4576416015625e-05,
      "step": 9064,
      "training_step_time": 9.89624810218811
    },
    {
      "epoch": 3.4576416015625e-05,
      "model_forward_time": 3.136486291885376,
      "step": 9064
    },
    {
      "epoch": 3.4576416015625e-05,
      "step": 9064,
      "training_step_time": 10.253849744796753
    },
    {
      "epoch": 3.458023071289062e-05,
      "model_forward_time": 3.1522576808929443,
      "step": 9065
    },
    {
      "dynamics_loss_avg": 0.05529917869716883,
      "epoch": 3.458023071289062e-05,
      "step": 9065
    },
    {
      "action_loss_avg": 0.018042347161099315,
      "epoch": 3.458023071289062e-05,
      "step": 9065
    },
    {
      "epoch": 3.458023071289062e-05,
      "step": 9065,
      "training_step_time": 10.223483562469482
    },
    {
      "epoch": 3.458023071289062e-05,
      "model_forward_time": 3.1353204250335693,
      "step": 9065
    },
    {
      "epoch": 3.458023071289062e-05,
      "step": 9065,
      "training_step_time": 9.899096250534058
    },
    {
      "epoch": 3.458023071289062e-05,
      "model_forward_time": 3.1318812370300293,
      "step": 9065
    },
    {
      "epoch": 3.458023071289062e-05,
      "step": 9065,
      "training_step_time": 9.89166784286499
    },
    {
      "epoch": 3.458023071289062e-05,
      "model_forward_time": 3.132185220718384,
      "step": 9065
    },
    {
      "epoch": 3.458023071289062e-05,
      "step": 9065,
      "training_step_time": 10.244157314300537
    },
    {
      "epoch": 3.458404541015625e-05,
      "model_forward_time": 3.1520373821258545,
      "step": 9066
    },
    {
      "epoch": 3.458404541015625e-05,
      "step": 9066,
      "training_step_time": 9.92499041557312
    },
    {
      "epoch": 3.458404541015625e-05,
      "model_forward_time": 3.4312613010406494,
      "step": 9066
    },
    {
      "epoch": 3.458404541015625e-05,
      "step": 9066,
      "training_step_time": 10.201220273971558
    },
    {
      "epoch": 3.458404541015625e-05,
      "model_forward_time": 3.1385297775268555,
      "step": 9066
    },
    {
      "epoch": 3.458404541015625e-05,
      "step": 9066,
      "training_step_time": 9.91490125656128
    },
    {
      "epoch": 3.458404541015625e-05,
      "model_forward_time": 2.438344717025757,
      "step": 9066
    },
    {
      "epoch": 3.458404541015625e-05,
      "step": 9066,
      "training_step_time": 10.299665451049805
    },
    {
      "epoch": 3.4587860107421876e-05,
      "model_forward_time": 3.1607773303985596,
      "step": 9067
    },
    {
      "epoch": 3.4587860107421876e-05,
      "step": 9067,
      "training_step_time": 9.93628215789795
    },
    {
      "epoch": 3.4587860107421876e-05,
      "model_forward_time": 3.137146472930908,
      "step": 9067
    },
    {
      "epoch": 3.4587860107421876e-05,
      "step": 9067,
      "training_step_time": 9.908568143844604
    },
    {
      "epoch": 3.4587860107421876e-05,
      "model_forward_time": 3.136357307434082,
      "step": 9067
    },
    {
      "dynamics_loss_avg": 0.07275008941069246,
      "epoch": 3.4587860107421876e-05,
      "step": 9067
    },
    {
      "action_loss_avg": 0.021845380205195396,
      "epoch": 3.4587860107421876e-05,
      "step": 9067
    },
    {
      "epoch": 3.4587860107421876e-05,
      "step": 9067,
      "training_step_time": 10.218799114227295
    },
    {
      "epoch": 3.4587860107421876e-05,
      "model_forward_time": 3.1378793716430664,
      "step": 9067
    },
    {
      "epoch": 3.4587860107421876e-05,
      "step": 9067,
      "training_step_time": 10.280903339385986
    },
    {
      "epoch": 3.45916748046875e-05,
      "model_forward_time": 3.143423318862915,
      "step": 9068
    },
    {
      "epoch": 3.45916748046875e-05,
      "step": 9068,
      "training_step_time": 9.91449785232544
    },
    {
      "epoch": 3.45916748046875e-05,
      "model_forward_time": 3.13978910446167,
      "step": 9068
    },
    {
      "epoch": 3.45916748046875e-05,
      "step": 9068,
      "training_step_time": 9.903726577758789
    },
    {
      "epoch": 3.45916748046875e-05,
      "model_forward_time": 3.136525869369507,
      "step": 9068
    },
    {
      "epoch": 3.45916748046875e-05,
      "step": 9068,
      "training_step_time": 9.900514364242554
    },
    {
      "epoch": 3.45916748046875e-05,
      "model_forward_time": 3.410684108734131,
      "step": 9068
    },
    {
      "epoch": 3.45916748046875e-05,
      "step": 9068,
      "training_step_time": 10.266838788986206
    },
    {
      "epoch": 3.459548950195313e-05,
      "model_forward_time": 3.154567241668701,
      "step": 9069
    },
    {
      "epoch": 3.459548950195313e-05,
      "step": 9069,
      "training_step_time": 9.9281644821167
    },
    {
      "epoch": 3.459548950195313e-05,
      "model_forward_time": 2.434969186782837,
      "step": 9069
    },
    {
      "epoch": 3.459548950195313e-05,
      "step": 9069,
      "training_step_time": 7.452489852905273
    },
    {
      "epoch": 3.459548950195313e-05,
      "model_forward_time": 3.1453559398651123,
      "step": 9069
    },
    {
      "epoch": 3.459548950195313e-05,
      "step": 9069,
      "training_step_time": 9.909046173095703
    },
    {
      "epoch": 3.459548950195313e-05,
      "model_forward_time": 3.138322591781616,
      "step": 9069
    },
    {
      "epoch": 3.459548950195313e-05,
      "step": 9069,
      "training_step_time": 10.271426916122437
    },
    {
      "epoch": 3.459930419921875e-05,
      "grad_norm": 0.42907944321632385,
      "learning_rate": 1e-05,
      "loss": 0.0793,
      "step": 9070
    },
    {
      "epoch": 3.459930419921875e-05,
      "model_forward_time": 3.4302008152008057,
      "step": 9070
    },
    {
      "dynamics_loss_avg": 0.05068326238542795,
      "epoch": 3.459930419921875e-05,
      "step": 9070
    },
    {
      "action_loss_avg": 0.017822724836878478,
      "epoch": 3.459930419921875e-05,
      "step": 9070
    },
    {
      "epoch": 3.459930419921875e-05,
      "step": 9070,
      "training_step_time": 10.194430589675903
    },
    {
      "epoch": 3.459930419921875e-05,
      "model_forward_time": 3.1351287364959717,
      "step": 9070
    },
    {
      "epoch": 3.459930419921875e-05,
      "step": 9070,
      "training_step_time": 9.893224000930786
    },
    {
      "epoch": 3.459930419921875e-05,
      "model_forward_time": 3.1388919353485107,
      "step": 9070
    },
    {
      "epoch": 3.459930419921875e-05,
      "step": 9070,
      "training_step_time": 9.89378833770752
    },
    {
      "epoch": 3.459930419921875e-05,
      "model_forward_time": 2.4327309131622314,
      "step": 9070
    },
    {
      "epoch": 3.459930419921875e-05,
      "step": 9070,
      "training_step_time": 10.26949691772461
    },
    {
      "epoch": 3.4603118896484374e-05,
      "model_forward_time": 3.1427297592163086,
      "step": 9071
    },
    {
      "epoch": 3.4603118896484374e-05,
      "step": 9071,
      "training_step_time": 9.906081914901733
    },
    {
      "epoch": 3.4603118896484374e-05,
      "model_forward_time": 3.1345906257629395,
      "step": 9071
    },
    {
      "epoch": 3.4603118896484374e-05,
      "step": 9071,
      "training_step_time": 10.18673825263977
    },
    {
      "epoch": 3.4603118896484374e-05,
      "model_forward_time": 3.1344387531280518,
      "step": 9071
    },
    {
      "epoch": 3.4603118896484374e-05,
      "step": 9071,
      "training_step_time": 9.896968364715576
    },
    {
      "epoch": 3.4603118896484374e-05,
      "model_forward_time": 3.1330227851867676,
      "step": 9071
    },
    {
      "epoch": 3.4603118896484374e-05,
      "step": 9071,
      "training_step_time": 10.014097452163696
    },
    {
      "epoch": 3.460693359375e-05,
      "model_forward_time": 3.1432583332061768,
      "step": 9072
    },
    {
      "epoch": 3.460693359375e-05,
      "step": 9072,
      "training_step_time": 9.912641048431396
    },
    {
      "epoch": 3.460693359375e-05,
      "model_forward_time": 3.1449246406555176,
      "step": 9072
    },
    {
      "epoch": 3.460693359375e-05,
      "step": 9072,
      "training_step_time": 9.91928744316101
    },
    {
      "epoch": 3.460693359375e-05,
      "model_forward_time": 3.4273393154144287,
      "step": 9072
    },
    {
      "dynamics_loss_avg": 0.07266751732677221,
      "epoch": 3.460693359375e-05,
      "step": 9072
    },
    {
      "action_loss_avg": 0.02837354624643922,
      "epoch": 3.460693359375e-05,
      "step": 9072
    },
    {
      "epoch": 3.460693359375e-05,
      "step": 9072,
      "training_step_time": 10.199278593063354
    },
    {
      "epoch": 3.460693359375e-05,
      "model_forward_time": 3.1432764530181885,
      "step": 9072
    },
    {
      "epoch": 3.460693359375e-05,
      "step": 9072,
      "training_step_time": 10.01723861694336
    },
    {
      "epoch": 3.4610748291015627e-05,
      "model_forward_time": 3.1610071659088135,
      "step": 9073
    },
    {
      "epoch": 3.4610748291015627e-05,
      "step": 9073,
      "training_step_time": 9.94074296951294
    },
    {
      "epoch": 3.4610748291015627e-05,
      "model_forward_time": 2.4401206970214844,
      "step": 9073
    },
    {
      "epoch": 3.4610748291015627e-05,
      "step": 9073,
      "training_step_time": 7.469483375549316
    },
    {
      "epoch": 3.4610748291015627e-05,
      "model_forward_time": 2.43630313873291,
      "step": 9073
    },
    {
      "epoch": 3.4610748291015627e-05,
      "step": 9073,
      "training_step_time": 7.4700915813446045
    },
    {
      "epoch": 3.4610748291015627e-05,
      "model_forward_time": 3.1487233638763428,
      "step": 9073
    },
    {
      "epoch": 3.4610748291015627e-05,
      "step": 9073,
      "training_step_time": 10.292216539382935
    },
    {
      "epoch": 3.461456298828125e-05,
      "model_forward_time": 3.4397706985473633,
      "step": 9074
    },
    {
      "epoch": 3.461456298828125e-05,
      "step": 9074,
      "training_step_time": 10.2139253616333
    },
    {
      "epoch": 3.461456298828125e-05,
      "model_forward_time": 3.138014793395996,
      "step": 9074
    },
    {
      "epoch": 3.461456298828125e-05,
      "step": 9074,
      "training_step_time": 9.905507802963257
    },
    {
      "epoch": 3.461456298828125e-05,
      "model_forward_time": 3.135715961456299,
      "step": 9074
    },
    {
      "epoch": 3.461456298828125e-05,
      "step": 9074,
      "training_step_time": 9.901941776275635
    },
    {
      "epoch": 3.461456298828125e-05,
      "model_forward_time": 3.140925884246826,
      "step": 9074
    },
    {
      "epoch": 3.461456298828125e-05,
      "step": 9074,
      "training_step_time": 10.296680450439453
    },
    {
      "epoch": 3.461837768554687e-05,
      "model_forward_time": 3.4078524112701416,
      "step": 9075
    },
    {
      "dynamics_loss_avg": 0.04452857505530119,
      "epoch": 3.461837768554687e-05,
      "step": 9075
    },
    {
      "action_loss_avg": 0.018070805910974742,
      "epoch": 3.461837768554687e-05,
      "step": 9075
    },
    {
      "epoch": 3.461837768554687e-05,
      "step": 9075,
      "training_step_time": 10.176440477371216
    },
    {
      "epoch": 3.461837768554687e-05,
      "model_forward_time": 2.4345169067382812,
      "step": 9075
    },
    {
      "epoch": 3.461837768554687e-05,
      "step": 9075,
      "training_step_time": 7.459068775177002
    },
    {
      "epoch": 3.461837768554687e-05,
      "model_forward_time": 2.4353623390197754,
      "step": 9075
    },
    {
      "epoch": 3.461837768554687e-05,
      "step": 9075,
      "training_step_time": 7.456378936767578
    },
    {
      "epoch": 3.461837768554687e-05,
      "model_forward_time": 3.1399190425872803,
      "step": 9075
    },
    {
      "epoch": 3.461837768554687e-05,
      "step": 9075,
      "training_step_time": 10.264838457107544
    },
    {
      "epoch": 3.46221923828125e-05,
      "model_forward_time": 3.156179428100586,
      "step": 9076
    },
    {
      "epoch": 3.46221923828125e-05,
      "step": 9076,
      "training_step_time": 9.93154239654541
    },
    {
      "epoch": 3.46221923828125e-05,
      "model_forward_time": 3.139875888824463,
      "step": 9076
    },
    {
      "epoch": 3.46221923828125e-05,
      "step": 9076,
      "training_step_time": 9.909892559051514
    },
    {
      "epoch": 3.46221923828125e-05,
      "model_forward_time": 3.3961641788482666,
      "step": 9076
    },
    {
      "epoch": 3.46221923828125e-05,
      "step": 9076,
      "training_step_time": 10.171919822692871
    },
    {
      "epoch": 3.46221923828125e-05,
      "model_forward_time": 3.1453709602355957,
      "step": 9076
    },
    {
      "epoch": 3.46221923828125e-05,
      "step": 9076,
      "training_step_time": 10.033119678497314
    },
    {
      "epoch": 3.4626007080078125e-05,
      "model_forward_time": 2.450110673904419,
      "step": 9077
    },
    {
      "epoch": 3.4626007080078125e-05,
      "step": 9077,
      "training_step_time": 7.476898193359375
    },
    {
      "epoch": 3.4626007080078125e-05,
      "model_forward_time": 3.1353325843811035,
      "step": 9077
    },
    {
      "epoch": 3.4626007080078125e-05,
      "step": 9077,
      "training_step_time": 9.905855178833008
    },
    {
      "epoch": 3.4626007080078125e-05,
      "model_forward_time": 3.141406297683716,
      "step": 9077
    },
    {
      "dynamics_loss_avg": 0.054975477419793604,
      "epoch": 3.4626007080078125e-05,
      "step": 9077
    },
    {
      "action_loss_avg": 0.01886349283158779,
      "epoch": 3.4626007080078125e-05,
      "step": 9077
    },
    {
      "epoch": 3.4626007080078125e-05,
      "step": 9077,
      "training_step_time": 10.189771890640259
    },
    {
      "epoch": 3.4626007080078125e-05,
      "model_forward_time": 3.1474592685699463,
      "step": 9077
    },
    {
      "epoch": 3.4626007080078125e-05,
      "step": 9077,
      "training_step_time": 10.248413562774658
    },
    {
      "epoch": 3.462982177734375e-05,
      "model_forward_time": 3.1423068046569824,
      "step": 9078
    },
    {
      "epoch": 3.462982177734375e-05,
      "step": 9078,
      "training_step_time": 9.913331270217896
    },
    {
      "epoch": 3.462982177734375e-05,
      "model_forward_time": 3.1324965953826904,
      "step": 9078
    },
    {
      "epoch": 3.462982177734375e-05,
      "step": 9078,
      "training_step_time": 9.896650314331055
    },
    {
      "epoch": 3.462982177734375e-05,
      "model_forward_time": 3.1359453201293945,
      "step": 9078
    },
    {
      "epoch": 3.462982177734375e-05,
      "step": 9078,
      "training_step_time": 9.899389743804932
    },
    {
      "epoch": 3.462982177734375e-05,
      "model_forward_time": 3.134160041809082,
      "step": 9078
    },
    {
      "epoch": 3.462982177734375e-05,
      "step": 9078,
      "training_step_time": 10.293949604034424
    },
    {
      "epoch": 3.463363647460938e-05,
      "model_forward_time": 3.1480915546417236,
      "step": 9079
    },
    {
      "epoch": 3.463363647460938e-05,
      "step": 9079,
      "training_step_time": 9.910841703414917
    },
    {
      "epoch": 3.463363647460938e-05,
      "model_forward_time": 3.130791187286377,
      "step": 9079
    },
    {
      "epoch": 3.463363647460938e-05,
      "step": 9079,
      "training_step_time": 9.895847797393799
    },
    {
      "epoch": 3.463363647460938e-05,
      "model_forward_time": 3.1374664306640625,
      "step": 9079
    },
    {
      "epoch": 3.463363647460938e-05,
      "step": 9079,
      "training_step_time": 9.903095483779907
    },
    {
      "epoch": 3.463363647460938e-05,
      "model_forward_time": 3.132281541824341,
      "step": 9079
    },
    {
      "epoch": 3.463363647460938e-05,
      "step": 9079,
      "training_step_time": 10.286624431610107
    },
    {
      "epoch": 3.4637451171875e-05,
      "grad_norm": 0.5080252885818481,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 9080
    },
    {
      "epoch": 3.4637451171875e-05,
      "model_forward_time": 3.4339747428894043,
      "step": 9080
    },
    {
      "dynamics_loss_avg": 0.06805392410606145,
      "epoch": 3.4637451171875e-05,
      "step": 9080
    },
    {
      "action_loss_avg": 0.022541831247508526,
      "epoch": 3.4637451171875e-05,
      "step": 9080
    },
    {
      "epoch": 3.4637451171875e-05,
      "step": 9080,
      "training_step_time": 10.206611394882202
    },
    {
      "epoch": 3.4637451171875e-05,
      "model_forward_time": 3.136693239212036,
      "step": 9080
    },
    {
      "epoch": 3.4637451171875e-05,
      "step": 9080,
      "training_step_time": 9.911649703979492
    },
    {
      "epoch": 3.4637451171875e-05,
      "model_forward_time": 3.1413302421569824,
      "step": 9080
    },
    {
      "epoch": 3.4637451171875e-05,
      "step": 9080,
      "training_step_time": 9.919432163238525
    },
    {
      "epoch": 3.4637451171875e-05,
      "model_forward_time": 3.1537890434265137,
      "step": 9080
    },
    {
      "epoch": 3.4637451171875e-05,
      "step": 9080,
      "training_step_time": 10.027529001235962
    },
    {
      "epoch": 3.464126586914062e-05,
      "model_forward_time": 3.154341220855713,
      "step": 9081
    },
    {
      "epoch": 3.464126586914062e-05,
      "step": 9081,
      "training_step_time": 9.932274341583252
    },
    {
      "epoch": 3.464126586914062e-05,
      "model_forward_time": 3.4058542251586914,
      "step": 9081
    },
    {
      "epoch": 3.464126586914062e-05,
      "step": 9081,
      "training_step_time": 10.181881189346313
    },
    {
      "epoch": 3.464126586914062e-05,
      "model_forward_time": 3.1500155925750732,
      "step": 9081
    },
    {
      "epoch": 3.464126586914062e-05,
      "step": 9081,
      "training_step_time": 9.924995422363281
    },
    {
      "epoch": 3.464126586914062e-05,
      "model_forward_time": 3.1497819423675537,
      "step": 9081
    },
    {
      "epoch": 3.464126586914062e-05,
      "step": 9081,
      "training_step_time": 10.315543413162231
    },
    {
      "epoch": 3.464508056640625e-05,
      "model_forward_time": 3.1546502113342285,
      "step": 9082
    },
    {
      "epoch": 3.464508056640625e-05,
      "step": 9082,
      "training_step_time": 9.934438228607178
    },
    {
      "epoch": 3.464508056640625e-05,
      "model_forward_time": 3.1446709632873535,
      "step": 9082
    },
    {
      "epoch": 3.464508056640625e-05,
      "step": 9082,
      "training_step_time": 9.91726303100586
    },
    {
      "epoch": 3.464508056640625e-05,
      "model_forward_time": 3.439164400100708,
      "step": 9082
    },
    {
      "dynamics_loss_avg": 0.050363650918006896,
      "epoch": 3.464508056640625e-05,
      "step": 9082
    },
    {
      "action_loss_avg": 0.017203647689893842,
      "epoch": 3.464508056640625e-05,
      "step": 9082
    },
    {
      "epoch": 3.464508056640625e-05,
      "step": 9082,
      "training_step_time": 10.208740711212158
    },
    {
      "epoch": 3.464508056640625e-05,
      "model_forward_time": 3.142662286758423,
      "step": 9082
    },
    {
      "epoch": 3.464508056640625e-05,
      "step": 9082,
      "training_step_time": 10.01153826713562
    },
    {
      "epoch": 3.4648895263671876e-05,
      "model_forward_time": 3.1470859050750732,
      "step": 9083
    },
    {
      "epoch": 3.4648895263671876e-05,
      "step": 9083,
      "training_step_time": 9.915639877319336
    },
    {
      "epoch": 3.4648895263671876e-05,
      "model_forward_time": 3.137479543685913,
      "step": 9083
    },
    {
      "epoch": 3.4648895263671876e-05,
      "step": 9083,
      "training_step_time": 9.898898601531982
    },
    {
      "epoch": 3.4648895263671876e-05,
      "model_forward_time": 3.136258840560913,
      "step": 9083
    },
    {
      "epoch": 3.4648895263671876e-05,
      "step": 9083,
      "training_step_time": 9.90390419960022
    },
    {
      "epoch": 3.4648895263671876e-05,
      "model_forward_time": 3.400840997695923,
      "step": 9083
    },
    {
      "epoch": 3.4648895263671876e-05,
      "step": 9083,
      "training_step_time": 10.263509035110474
    },
    {
      "epoch": 3.46527099609375e-05,
      "model_forward_time": 3.1582725048065186,
      "step": 9084
    },
    {
      "epoch": 3.46527099609375e-05,
      "step": 9084,
      "training_step_time": 9.92859148979187
    },
    {
      "epoch": 3.46527099609375e-05,
      "model_forward_time": 3.140388011932373,
      "step": 9084
    },
    {
      "epoch": 3.46527099609375e-05,
      "step": 9084,
      "training_step_time": 9.910812854766846
    },
    {
      "epoch": 3.46527099609375e-05,
      "model_forward_time": 3.1402947902679443,
      "step": 9084
    },
    {
      "epoch": 3.46527099609375e-05,
      "step": 9084,
      "training_step_time": 9.909602880477905
    },
    {
      "epoch": 3.46527099609375e-05,
      "model_forward_time": 3.1462793350219727,
      "step": 9084
    },
    {
      "epoch": 3.46527099609375e-05,
      "step": 9084,
      "training_step_time": 10.025973796844482
    },
    {
      "epoch": 3.465652465820313e-05,
      "model_forward_time": 3.402523994445801,
      "step": 9085
    },
    {
      "dynamics_loss_avg": 0.05340428501367569,
      "epoch": 3.465652465820313e-05,
      "step": 9085
    },
    {
      "action_loss_avg": 0.019556807726621626,
      "epoch": 3.465652465820313e-05,
      "step": 9085
    },
    {
      "epoch": 3.465652465820313e-05,
      "step": 9085,
      "training_step_time": 10.179306507110596
    },
    {
      "epoch": 3.465652465820313e-05,
      "model_forward_time": 3.138908863067627,
      "step": 9085
    },
    {
      "epoch": 3.465652465820313e-05,
      "step": 9085,
      "training_step_time": 9.90559720993042
    },
    {
      "epoch": 3.465652465820313e-05,
      "model_forward_time": 3.1405081748962402,
      "step": 9085
    },
    {
      "epoch": 3.465652465820313e-05,
      "step": 9085,
      "training_step_time": 9.909587621688843
    },
    {
      "epoch": 3.465652465820313e-05,
      "model_forward_time": 3.136117696762085,
      "step": 9085
    },
    {
      "epoch": 3.465652465820313e-05,
      "step": 9085,
      "training_step_time": 10.27161431312561
    },
    {
      "epoch": 3.466033935546875e-05,
      "model_forward_time": 3.146756649017334,
      "step": 9086
    },
    {
      "epoch": 3.466033935546875e-05,
      "step": 9086,
      "training_step_time": 9.913490772247314
    },
    {
      "epoch": 3.466033935546875e-05,
      "model_forward_time": 3.4683802127838135,
      "step": 9086
    },
    {
      "epoch": 3.466033935546875e-05,
      "step": 9086,
      "training_step_time": 10.242308139801025
    },
    {
      "epoch": 3.466033935546875e-05,
      "model_forward_time": 3.139038324356079,
      "step": 9086
    },
    {
      "epoch": 3.466033935546875e-05,
      "step": 9086,
      "training_step_time": 9.906224250793457
    },
    {
      "epoch": 3.466033935546875e-05,
      "model_forward_time": 3.138603687286377,
      "step": 9086
    },
    {
      "epoch": 3.466033935546875e-05,
      "step": 9086,
      "training_step_time": 10.012574672698975
    },
    {
      "epoch": 3.4664154052734374e-05,
      "model_forward_time": 3.1466758251190186,
      "step": 9087
    },
    {
      "epoch": 3.4664154052734374e-05,
      "step": 9087,
      "training_step_time": 9.915468215942383
    },
    {
      "epoch": 3.4664154052734374e-05,
      "model_forward_time": 3.13693904876709,
      "step": 9087
    },
    {
      "epoch": 3.4664154052734374e-05,
      "step": 9087,
      "training_step_time": 9.908940553665161
    },
    {
      "epoch": 3.4664154052734374e-05,
      "model_forward_time": 3.406898021697998,
      "step": 9087
    },
    {
      "dynamics_loss_avg": 0.06294547524303198,
      "epoch": 3.4664154052734374e-05,
      "step": 9087
    },
    {
      "action_loss_avg": 0.019763848558068276,
      "epoch": 3.4664154052734374e-05,
      "step": 9087
    },
    {
      "epoch": 3.4664154052734374e-05,
      "step": 9087,
      "training_step_time": 10.181970834732056
    },
    {
      "epoch": 3.4664154052734374e-05,
      "model_forward_time": 3.141969680786133,
      "step": 9087
    },
    {
      "epoch": 3.4664154052734374e-05,
      "step": 9087,
      "training_step_time": 10.351078748703003
    },
    {
      "epoch": 3.466796875e-05,
      "model_forward_time": 3.1627166271209717,
      "step": 9088
    },
    {
      "epoch": 3.466796875e-05,
      "step": 9088,
      "training_step_time": 9.934741258621216
    },
    {
      "epoch": 3.466796875e-05,
      "model_forward_time": 3.1399803161621094,
      "step": 9088
    },
    {
      "epoch": 3.466796875e-05,
      "step": 9088,
      "training_step_time": 9.90704894065857
    },
    {
      "epoch": 3.466796875e-05,
      "model_forward_time": 3.141416311264038,
      "step": 9088
    },
    {
      "epoch": 3.466796875e-05,
      "step": 9088,
      "training_step_time": 9.91290283203125
    },
    {
      "epoch": 3.466796875e-05,
      "model_forward_time": 3.4532532691955566,
      "step": 9088
    },
    {
      "epoch": 3.466796875e-05,
      "step": 9088,
      "training_step_time": 10.31679368019104
    },
    {
      "epoch": 3.4671783447265627e-05,
      "model_forward_time": 3.147502899169922,
      "step": 9089
    },
    {
      "epoch": 3.4671783447265627e-05,
      "step": 9089,
      "training_step_time": 9.921691656112671
    },
    {
      "epoch": 3.4671783447265627e-05,
      "model_forward_time": 3.148399829864502,
      "step": 9089
    },
    {
      "epoch": 3.4671783447265627e-05,
      "step": 9089,
      "training_step_time": 9.924056768417358
    },
    {
      "epoch": 3.4671783447265627e-05,
      "model_forward_time": 2.4382779598236084,
      "step": 9089
    },
    {
      "epoch": 3.4671783447265627e-05,
      "step": 9089,
      "training_step_time": 7.468485593795776
    },
    {
      "epoch": 3.4671783447265627e-05,
      "model_forward_time": 3.1352083683013916,
      "step": 9089
    },
    {
      "epoch": 3.4671783447265627e-05,
      "step": 9089,
      "training_step_time": 10.272159099578857
    },
    {
      "epoch": 3.467559814453125e-05,
      "grad_norm": 0.5903734564781189,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 9090
    },
    {
      "epoch": 3.467559814453125e-05,
      "model_forward_time": 3.447340726852417,
      "step": 9090
    },
    {
      "dynamics_loss_avg": 0.06367497071623802,
      "epoch": 3.467559814453125e-05,
      "step": 9090
    },
    {
      "action_loss_avg": 0.0236516990698874,
      "epoch": 3.467559814453125e-05,
      "step": 9090
    },
    {
      "epoch": 3.467559814453125e-05,
      "step": 9090,
      "training_step_time": 10.224812269210815
    },
    {
      "epoch": 3.467559814453125e-05,
      "model_forward_time": 3.1382787227630615,
      "step": 9090
    },
    {
      "epoch": 3.467559814453125e-05,
      "step": 9090,
      "training_step_time": 9.907866716384888
    },
    {
      "epoch": 3.467559814453125e-05,
      "model_forward_time": 3.142163038253784,
      "step": 9090
    },
    {
      "epoch": 3.467559814453125e-05,
      "step": 9090,
      "training_step_time": 9.91780710220337
    },
    {
      "epoch": 3.467559814453125e-05,
      "model_forward_time": 2.4333255290985107,
      "step": 9090
    },
    {
      "epoch": 3.467559814453125e-05,
      "step": 9090,
      "training_step_time": 10.300059795379639
    },
    {
      "epoch": 3.467941284179687e-05,
      "model_forward_time": 2.442780017852783,
      "step": 9091
    },
    {
      "epoch": 3.467941284179687e-05,
      "step": 9091,
      "training_step_time": 7.468437671661377
    },
    {
      "epoch": 3.467941284179687e-05,
      "model_forward_time": 3.1332449913024902,
      "step": 9091
    },
    {
      "epoch": 3.467941284179687e-05,
      "step": 9091,
      "training_step_time": 9.8991060256958
    },
    {
      "epoch": 3.467941284179687e-05,
      "model_forward_time": 3.4140288829803467,
      "step": 9091
    },
    {
      "epoch": 3.467941284179687e-05,
      "step": 9091,
      "training_step_time": 10.180426359176636
    },
    {
      "epoch": 3.467941284179687e-05,
      "model_forward_time": 3.1393065452575684,
      "step": 9091
    },
    {
      "epoch": 3.467941284179687e-05,
      "step": 9091,
      "training_step_time": 10.248911619186401
    },
    {
      "epoch": 3.46832275390625e-05,
      "model_forward_time": 3.144263744354248,
      "step": 9092
    },
    {
      "epoch": 3.46832275390625e-05,
      "step": 9092,
      "training_step_time": 9.919458389282227
    },
    {
      "epoch": 3.46832275390625e-05,
      "model_forward_time": 3.1398470401763916,
      "step": 9092
    },
    {
      "epoch": 3.46832275390625e-05,
      "step": 9092,
      "training_step_time": 9.90466046333313
    },
    {
      "epoch": 3.46832275390625e-05,
      "model_forward_time": 3.418571949005127,
      "step": 9092
    },
    {
      "dynamics_loss_avg": 0.06560640912503005,
      "epoch": 3.46832275390625e-05,
      "step": 9092
    },
    {
      "action_loss_avg": 0.022339130099862815,
      "epoch": 3.46832275390625e-05,
      "step": 9092
    },
    {
      "epoch": 3.46832275390625e-05,
      "step": 9092,
      "training_step_time": 10.186348915100098
    },
    {
      "epoch": 3.46832275390625e-05,
      "model_forward_time": 3.1380672454833984,
      "step": 9092
    },
    {
      "epoch": 3.46832275390625e-05,
      "step": 9092,
      "training_step_time": 10.303406000137329
    },
    {
      "epoch": 3.4687042236328125e-05,
      "model_forward_time": 3.1556999683380127,
      "step": 9093
    },
    {
      "epoch": 3.4687042236328125e-05,
      "step": 9093,
      "training_step_time": 9.926252126693726
    },
    {
      "epoch": 3.4687042236328125e-05,
      "model_forward_time": 2.4379029273986816,
      "step": 9093
    },
    {
      "epoch": 3.4687042236328125e-05,
      "step": 9093,
      "training_step_time": 7.468411207199097
    },
    {
      "epoch": 3.4687042236328125e-05,
      "model_forward_time": 3.134168863296509,
      "step": 9093
    },
    {
      "epoch": 3.4687042236328125e-05,
      "step": 9093,
      "training_step_time": 9.90153455734253
    },
    {
      "epoch": 3.4687042236328125e-05,
      "model_forward_time": 3.1420748233795166,
      "step": 9093
    },
    {
      "epoch": 3.4687042236328125e-05,
      "step": 9093,
      "training_step_time": 10.22365117073059
    },
    {
      "epoch": 3.469085693359375e-05,
      "model_forward_time": 2.8041300773620605,
      "step": 9094
    },
    {
      "epoch": 3.469085693359375e-05,
      "step": 9094,
      "training_step_time": 7.828134775161743
    },
    {
      "epoch": 3.469085693359375e-05,
      "model_forward_time": 3.1355834007263184,
      "step": 9094
    },
    {
      "epoch": 3.469085693359375e-05,
      "step": 9094,
      "training_step_time": 9.900323152542114
    },
    {
      "epoch": 3.469085693359375e-05,
      "model_forward_time": 3.1345043182373047,
      "step": 9094
    },
    {
      "epoch": 3.469085693359375e-05,
      "step": 9094,
      "training_step_time": 9.902968406677246
    },
    {
      "epoch": 3.469085693359375e-05,
      "model_forward_time": 3.134446859359741,
      "step": 9094
    },
    {
      "epoch": 3.469085693359375e-05,
      "step": 9094,
      "training_step_time": 10.241707801818848
    },
    {
      "epoch": 3.469467163085938e-05,
      "model_forward_time": 3.418447494506836,
      "step": 9095
    },
    {
      "dynamics_loss_avg": 0.04577849037013948,
      "epoch": 3.469467163085938e-05,
      "step": 9095
    },
    {
      "action_loss_avg": 0.017409819061867893,
      "epoch": 3.469467163085938e-05,
      "step": 9095
    },
    {
      "epoch": 3.469467163085938e-05,
      "step": 9095,
      "training_step_time": 10.187859535217285
    },
    {
      "epoch": 3.469467163085938e-05,
      "model_forward_time": 3.146219491958618,
      "step": 9095
    },
    {
      "epoch": 3.469467163085938e-05,
      "step": 9095,
      "training_step_time": 9.917331218719482
    },
    {
      "epoch": 3.469467163085938e-05,
      "model_forward_time": 3.13970947265625,
      "step": 9095
    },
    {
      "epoch": 3.469467163085938e-05,
      "step": 9095,
      "training_step_time": 9.912254571914673
    },
    {
      "epoch": 3.469467163085938e-05,
      "model_forward_time": 3.150029182434082,
      "step": 9095
    },
    {
      "epoch": 3.469467163085938e-05,
      "step": 9095,
      "training_step_time": 10.317772150039673
    },
    {
      "epoch": 3.4698486328125e-05,
      "model_forward_time": 3.1607422828674316,
      "step": 9096
    },
    {
      "epoch": 3.4698486328125e-05,
      "step": 9096,
      "training_step_time": 9.94204044342041
    },
    {
      "epoch": 3.4698486328125e-05,
      "model_forward_time": 3.143970012664795,
      "step": 9096
    },
    {
      "epoch": 3.4698486328125e-05,
      "step": 9096,
      "training_step_time": 10.177268981933594
    },
    {
      "epoch": 3.4698486328125e-05,
      "model_forward_time": 3.1423633098602295,
      "step": 9096
    },
    {
      "epoch": 3.4698486328125e-05,
      "step": 9096,
      "training_step_time": 9.912259578704834
    },
    {
      "epoch": 3.4698486328125e-05,
      "model_forward_time": 3.1467883586883545,
      "step": 9096
    },
    {
      "epoch": 3.4698486328125e-05,
      "step": 9096,
      "training_step_time": 10.024578332901001
    },
    {
      "epoch": 3.470230102539062e-05,
      "model_forward_time": 3.1569015979766846,
      "step": 9097
    },
    {
      "epoch": 3.470230102539062e-05,
      "step": 9097,
      "training_step_time": 9.926898717880249
    },
    {
      "epoch": 3.470230102539062e-05,
      "model_forward_time": 3.1390106678009033,
      "step": 9097
    },
    {
      "epoch": 3.470230102539062e-05,
      "step": 9097,
      "training_step_time": 9.904378414154053
    },
    {
      "epoch": 3.470230102539062e-05,
      "model_forward_time": 3.4018161296844482,
      "step": 9097
    },
    {
      "dynamics_loss_avg": 0.06068790731951594,
      "epoch": 3.470230102539062e-05,
      "step": 9097
    },
    {
      "action_loss_avg": 0.022276828298345208,
      "epoch": 3.470230102539062e-05,
      "step": 9097
    },
    {
      "epoch": 3.470230102539062e-05,
      "step": 9097,
      "training_step_time": 10.173306226730347
    },
    {
      "epoch": 3.470230102539062e-05,
      "model_forward_time": 3.1420822143554688,
      "step": 9097
    },
    {
      "epoch": 3.470230102539062e-05,
      "step": 9097,
      "training_step_time": 10.020697355270386
    },
    {
      "epoch": 3.470611572265625e-05,
      "model_forward_time": 3.155557632446289,
      "step": 9098
    },
    {
      "epoch": 3.470611572265625e-05,
      "step": 9098,
      "training_step_time": 9.937057256698608
    },
    {
      "epoch": 3.470611572265625e-05,
      "model_forward_time": 3.133991003036499,
      "step": 9098
    },
    {
      "epoch": 3.470611572265625e-05,
      "step": 9098,
      "training_step_time": 9.908432006835938
    },
    {
      "epoch": 3.470611572265625e-05,
      "model_forward_time": 3.1441781520843506,
      "step": 9098
    },
    {
      "epoch": 3.470611572265625e-05,
      "step": 9098,
      "training_step_time": 9.919325590133667
    },
    {
      "epoch": 3.470611572265625e-05,
      "model_forward_time": 3.4316017627716064,
      "step": 9098
    },
    {
      "epoch": 3.470611572265625e-05,
      "step": 9098,
      "training_step_time": 10.296870708465576
    },
    {
      "epoch": 3.4709930419921876e-05,
      "model_forward_time": 3.164322853088379,
      "step": 9099
    },
    {
      "epoch": 3.4709930419921876e-05,
      "step": 9099,
      "training_step_time": 9.953548192977905
    },
    {
      "epoch": 3.4709930419921876e-05,
      "model_forward_time": 3.1365833282470703,
      "step": 9099
    },
    {
      "epoch": 3.4709930419921876e-05,
      "step": 9099,
      "training_step_time": 9.914782524108887
    },
    {
      "epoch": 3.4709930419921876e-05,
      "model_forward_time": 3.1439456939697266,
      "step": 9099
    },
    {
      "epoch": 3.4709930419921876e-05,
      "step": 9099,
      "training_step_time": 9.913184404373169
    },
    {
      "epoch": 3.4709930419921876e-05,
      "model_forward_time": 3.1387646198272705,
      "step": 9099
    },
    {
      "epoch": 3.4709930419921876e-05,
      "step": 9099,
      "training_step_time": 10.313488245010376
    },
    {
      "epoch": 3.47137451171875e-05,
      "grad_norm": 0.44959062337875366,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9100
    },
    {
      "epoch": 3.47137451171875e-05,
      "model_forward_time": 3.167874574661255,
      "step": 9100
    },
    {
      "dynamics_loss_avg": 0.06070797294378281,
      "epoch": 3.47137451171875e-05,
      "step": 9100
    },
    {
      "action_loss_avg": 0.022955460613593458,
      "epoch": 3.47137451171875e-05,
      "step": 9100
    },
    {
      "epoch": 3.47137451171875e-05,
      "step": 9100,
      "training_step_time": 10.244659185409546
    },
    {
      "epoch": 3.47137451171875e-05,
      "model_forward_time": 3.1409263610839844,
      "step": 9100
    },
    {
      "epoch": 3.47137451171875e-05,
      "step": 9100,
      "training_step_time": 9.924047470092773
    },
    {
      "epoch": 3.47137451171875e-05,
      "model_forward_time": 3.142897129058838,
      "step": 9100
    },
    {
      "epoch": 3.47137451171875e-05,
      "step": 9100,
      "training_step_time": 9.920235633850098
    },
    {
      "epoch": 3.47137451171875e-05,
      "model_forward_time": 3.146982431411743,
      "step": 9100
    },
    {
      "epoch": 3.47137451171875e-05,
      "step": 9100,
      "training_step_time": 10.306510210037231
    },
    {
      "epoch": 3.471755981445313e-05,
      "model_forward_time": 3.146461009979248,
      "step": 9101
    },
    {
      "epoch": 3.471755981445313e-05,
      "step": 9101,
      "training_step_time": 9.925257682800293
    },
    {
      "epoch": 3.471755981445313e-05,
      "model_forward_time": 3.4379045963287354,
      "step": 9101
    },
    {
      "epoch": 3.471755981445313e-05,
      "step": 9101,
      "training_step_time": 10.21588134765625
    },
    {
      "epoch": 3.471755981445313e-05,
      "model_forward_time": 2.4378867149353027,
      "step": 9101
    },
    {
      "epoch": 3.471755981445313e-05,
      "step": 9101,
      "training_step_time": 7.46800971031189
    },
    {
      "epoch": 3.471755981445313e-05,
      "model_forward_time": 3.1407060623168945,
      "step": 9101
    },
    {
      "epoch": 3.471755981445313e-05,
      "step": 9101,
      "training_step_time": 10.01400113105774
    },
    {
      "epoch": 3.472137451171875e-05,
      "model_forward_time": 3.1588456630706787,
      "step": 9102
    },
    {
      "epoch": 3.472137451171875e-05,
      "step": 9102,
      "training_step_time": 9.944932222366333
    },
    {
      "epoch": 3.472137451171875e-05,
      "model_forward_time": 2.4352028369903564,
      "step": 9102
    },
    {
      "epoch": 3.472137451171875e-05,
      "step": 9102,
      "training_step_time": 7.470694065093994
    },
    {
      "epoch": 3.472137451171875e-05,
      "model_forward_time": 2.4304986000061035,
      "step": 9102
    },
    {
      "dynamics_loss_avg": 0.06251604706048966,
      "epoch": 3.472137451171875e-05,
      "step": 9102
    },
    {
      "action_loss_avg": 0.01933175651356578,
      "epoch": 3.472137451171875e-05,
      "step": 9102
    },
    {
      "epoch": 3.472137451171875e-05,
      "step": 9102,
      "training_step_time": 7.464752197265625
    },
    {
      "epoch": 3.472137451171875e-05,
      "model_forward_time": 3.425769805908203,
      "step": 9102
    },
    {
      "epoch": 3.472137451171875e-05,
      "step": 9102,
      "training_step_time": 10.299700498580933
    },
    {
      "epoch": 3.4725189208984374e-05,
      "model_forward_time": 2.4467973709106445,
      "step": 9103
    },
    {
      "epoch": 3.4725189208984374e-05,
      "step": 9103,
      "training_step_time": 7.478796720504761
    },
    {
      "epoch": 3.4725189208984374e-05,
      "model_forward_time": 3.1437735557556152,
      "step": 9103
    },
    {
      "epoch": 3.4725189208984374e-05,
      "step": 9103,
      "training_step_time": 9.927026271820068
    },
    {
      "epoch": 3.4725189208984374e-05,
      "model_forward_time": 3.133859872817993,
      "step": 9103
    },
    {
      "epoch": 3.4725189208984374e-05,
      "step": 9103,
      "training_step_time": 9.915235757827759
    },
    {
      "epoch": 3.4725189208984374e-05,
      "model_forward_time": 3.134761333465576,
      "step": 9103
    },
    {
      "epoch": 3.4725189208984374e-05,
      "step": 9103,
      "training_step_time": 10.346093893051147
    },
    {
      "epoch": 3.472900390625e-05,
      "model_forward_time": 3.4242680072784424,
      "step": 9104
    },
    {
      "epoch": 3.472900390625e-05,
      "step": 9104,
      "training_step_time": 10.203158855438232
    },
    {
      "epoch": 3.472900390625e-05,
      "model_forward_time": 3.134748935699463,
      "step": 9104
    },
    {
      "epoch": 3.472900390625e-05,
      "step": 9104,
      "training_step_time": 9.913705348968506
    },
    {
      "epoch": 3.472900390625e-05,
      "model_forward_time": 3.1355786323547363,
      "step": 9104
    },
    {
      "epoch": 3.472900390625e-05,
      "step": 9104,
      "training_step_time": 9.918174982070923
    },
    {
      "epoch": 3.472900390625e-05,
      "model_forward_time": 3.137521982192993,
      "step": 9104
    },
    {
      "epoch": 3.472900390625e-05,
      "step": 9104,
      "training_step_time": 10.24393081665039
    },
    {
      "epoch": 3.473281860351563e-05,
      "model_forward_time": 3.157827615737915,
      "step": 9105
    },
    {
      "dynamics_loss_avg": 0.05399625161662698,
      "epoch": 3.473281860351563e-05,
      "step": 9105
    },
    {
      "action_loss_avg": 0.019781392067670822,
      "epoch": 3.473281860351563e-05,
      "step": 9105
    },
    {
      "epoch": 3.473281860351563e-05,
      "step": 9105,
      "training_step_time": 9.935468912124634
    },
    {
      "epoch": 3.473281860351563e-05,
      "model_forward_time": 3.411006450653076,
      "step": 9105
    },
    {
      "epoch": 3.473281860351563e-05,
      "step": 9105,
      "training_step_time": 10.18309497833252
    },
    {
      "epoch": 3.473281860351563e-05,
      "model_forward_time": 3.134247064590454,
      "step": 9105
    },
    {
      "epoch": 3.473281860351563e-05,
      "step": 9105,
      "training_step_time": 9.912702083587646
    },
    {
      "epoch": 3.473281860351563e-05,
      "model_forward_time": 3.1395554542541504,
      "step": 9105
    },
    {
      "epoch": 3.473281860351563e-05,
      "step": 9105,
      "training_step_time": 10.011112451553345
    },
    {
      "epoch": 3.473663330078125e-05,
      "model_forward_time": 3.14996600151062,
      "step": 9106
    },
    {
      "epoch": 3.473663330078125e-05,
      "step": 9106,
      "training_step_time": 9.947007894515991
    },
    {
      "epoch": 3.473663330078125e-05,
      "model_forward_time": 3.455293655395508,
      "step": 9106
    },
    {
      "epoch": 3.473663330078125e-05,
      "step": 9106,
      "training_step_time": 10.23586893081665
    },
    {
      "epoch": 3.473663330078125e-05,
      "model_forward_time": 3.140368700027466,
      "step": 9106
    },
    {
      "epoch": 3.473663330078125e-05,
      "step": 9106,
      "training_step_time": 9.92043924331665
    },
    {
      "epoch": 3.473663330078125e-05,
      "model_forward_time": 3.141209602355957,
      "step": 9106
    },
    {
      "epoch": 3.473663330078125e-05,
      "step": 9106,
      "training_step_time": 10.30427622795105
    },
    {
      "epoch": 3.474044799804687e-05,
      "model_forward_time": 3.149481773376465,
      "step": 9107
    },
    {
      "epoch": 3.474044799804687e-05,
      "step": 9107,
      "training_step_time": 9.936417579650879
    },
    {
      "epoch": 3.474044799804687e-05,
      "model_forward_time": 3.1325879096984863,
      "step": 9107
    },
    {
      "epoch": 3.474044799804687e-05,
      "step": 9107,
      "training_step_time": 9.916250705718994
    },
    {
      "epoch": 3.474044799804687e-05,
      "model_forward_time": 2.437363386154175,
      "step": 9107
    },
    {
      "dynamics_loss_avg": 0.06995340436697006,
      "epoch": 3.474044799804687e-05,
      "step": 9107
    },
    {
      "action_loss_avg": 0.02299806857481599,
      "epoch": 3.474044799804687e-05,
      "step": 9107
    },
    {
      "epoch": 3.474044799804687e-05,
      "step": 9107,
      "training_step_time": 7.7546916007995605
    },
    {
      "epoch": 3.474044799804687e-05,
      "model_forward_time": 3.1367509365081787,
      "step": 9107
    },
    {
      "epoch": 3.474044799804687e-05,
      "step": 9107,
      "training_step_time": 10.011497020721436
    },
    {
      "epoch": 3.47442626953125e-05,
      "model_forward_time": 3.1449103355407715,
      "step": 9108
    },
    {
      "epoch": 3.47442626953125e-05,
      "step": 9108,
      "training_step_time": 9.929649591445923
    },
    {
      "epoch": 3.47442626953125e-05,
      "model_forward_time": 3.135640859603882,
      "step": 9108
    },
    {
      "epoch": 3.47442626953125e-05,
      "step": 9108,
      "training_step_time": 9.919895648956299
    },
    {
      "epoch": 3.47442626953125e-05,
      "model_forward_time": 3.139893054962158,
      "step": 9108
    },
    {
      "epoch": 3.47442626953125e-05,
      "step": 9108,
      "training_step_time": 9.921218156814575
    },
    {
      "epoch": 3.47442626953125e-05,
      "model_forward_time": 3.4196414947509766,
      "step": 9108
    },
    {
      "epoch": 3.47442626953125e-05,
      "step": 9108,
      "training_step_time": 10.330897569656372
    },
    {
      "epoch": 3.4748077392578125e-05,
      "model_forward_time": 3.1549041271209717,
      "step": 9109
    },
    {
      "epoch": 3.4748077392578125e-05,
      "step": 9109,
      "training_step_time": 9.942114353179932
    },
    {
      "epoch": 3.4748077392578125e-05,
      "model_forward_time": 3.132537841796875,
      "step": 9109
    },
    {
      "epoch": 3.4748077392578125e-05,
      "step": 9109,
      "training_step_time": 9.909518957138062
    },
    {
      "epoch": 3.4748077392578125e-05,
      "model_forward_time": 3.1405746936798096,
      "step": 9109
    },
    {
      "epoch": 3.4748077392578125e-05,
      "step": 9109,
      "training_step_time": 9.918824911117554
    },
    {
      "epoch": 3.4748077392578125e-05,
      "model_forward_time": 3.1368119716644287,
      "step": 9109
    },
    {
      "epoch": 3.4748077392578125e-05,
      "step": 9109,
      "training_step_time": 10.025654554367065
    },
    {
      "epoch": 3.475189208984375e-05,
      "grad_norm": 0.408702552318573,
      "learning_rate": 1e-05,
      "loss": 0.0754,
      "step": 9110
    },
    {
      "epoch": 3.475189208984375e-05,
      "model_forward_time": 3.459017038345337,
      "step": 9110
    },
    {
      "dynamics_loss_avg": 0.04750232771039009,
      "epoch": 3.475189208984375e-05,
      "step": 9110
    },
    {
      "action_loss_avg": 0.015068551246076822,
      "epoch": 3.475189208984375e-05,
      "step": 9110
    },
    {
      "epoch": 3.475189208984375e-05,
      "step": 9110,
      "training_step_time": 10.245529413223267
    },
    {
      "epoch": 3.475189208984375e-05,
      "model_forward_time": 3.1471691131591797,
      "step": 9110
    },
    {
      "epoch": 3.475189208984375e-05,
      "step": 9110,
      "training_step_time": 9.939868927001953
    },
    {
      "epoch": 3.475189208984375e-05,
      "model_forward_time": 3.1492674350738525,
      "step": 9110
    },
    {
      "epoch": 3.475189208984375e-05,
      "step": 9110,
      "training_step_time": 9.93254542350769
    },
    {
      "epoch": 3.475189208984375e-05,
      "model_forward_time": 3.1442019939422607,
      "step": 9110
    },
    {
      "epoch": 3.475189208984375e-05,
      "step": 9110,
      "training_step_time": 10.314933061599731
    },
    {
      "epoch": 3.475570678710938e-05,
      "model_forward_time": 3.16483473777771,
      "step": 9111
    },
    {
      "epoch": 3.475570678710938e-05,
      "step": 9111,
      "training_step_time": 9.946143865585327
    },
    {
      "epoch": 3.475570678710938e-05,
      "model_forward_time": 3.4156577587127686,
      "step": 9111
    },
    {
      "epoch": 3.475570678710938e-05,
      "step": 9111,
      "training_step_time": 10.195512294769287
    },
    {
      "epoch": 3.475570678710938e-05,
      "model_forward_time": 3.1443564891815186,
      "step": 9111
    },
    {
      "epoch": 3.475570678710938e-05,
      "step": 9111,
      "training_step_time": 9.93031907081604
    },
    {
      "epoch": 3.475570678710938e-05,
      "model_forward_time": 3.1398916244506836,
      "step": 9111
    },
    {
      "epoch": 3.475570678710938e-05,
      "step": 9111,
      "training_step_time": 10.301230669021606
    },
    {
      "epoch": 3.4759521484375e-05,
      "model_forward_time": 3.1446211338043213,
      "step": 9112
    },
    {
      "epoch": 3.4759521484375e-05,
      "step": 9112,
      "training_step_time": 9.921711206436157
    },
    {
      "epoch": 3.4759521484375e-05,
      "model_forward_time": 3.1443207263946533,
      "step": 9112
    },
    {
      "epoch": 3.4759521484375e-05,
      "step": 9112,
      "training_step_time": 9.922370910644531
    },
    {
      "epoch": 3.4759521484375e-05,
      "model_forward_time": 3.4222068786621094,
      "step": 9112
    },
    {
      "dynamics_loss_avg": 0.05046812789514661,
      "epoch": 3.4759521484375e-05,
      "step": 9112
    },
    {
      "action_loss_avg": 0.013568834355100989,
      "epoch": 3.4759521484375e-05,
      "step": 9112
    },
    {
      "epoch": 3.4759521484375e-05,
      "step": 9112,
      "training_step_time": 10.201117038726807
    },
    {
      "epoch": 3.4759521484375e-05,
      "model_forward_time": 3.1411666870117188,
      "step": 9112
    },
    {
      "epoch": 3.4759521484375e-05,
      "step": 9112,
      "training_step_time": 10.283270120620728
    },
    {
      "epoch": 3.476333618164062e-05,
      "model_forward_time": 3.144399881362915,
      "step": 9113
    },
    {
      "epoch": 3.476333618164062e-05,
      "step": 9113,
      "training_step_time": 9.924869775772095
    },
    {
      "epoch": 3.476333618164062e-05,
      "model_forward_time": 3.1392900943756104,
      "step": 9113
    },
    {
      "epoch": 3.476333618164062e-05,
      "step": 9113,
      "training_step_time": 9.917393207550049
    },
    {
      "epoch": 3.476333618164062e-05,
      "model_forward_time": 3.142115354537964,
      "step": 9113
    },
    {
      "epoch": 3.476333618164062e-05,
      "step": 9113,
      "training_step_time": 9.922324657440186
    },
    {
      "epoch": 3.476333618164062e-05,
      "model_forward_time": 3.4035205841064453,
      "step": 9113
    },
    {
      "epoch": 3.476333618164062e-05,
      "step": 9113,
      "training_step_time": 10.281425476074219
    },
    {
      "epoch": 3.476715087890625e-05,
      "model_forward_time": 2.451597213745117,
      "step": 9114
    },
    {
      "epoch": 3.476715087890625e-05,
      "step": 9114,
      "training_step_time": 7.492662668228149
    },
    {
      "epoch": 3.476715087890625e-05,
      "model_forward_time": 3.1393585205078125,
      "step": 9114
    },
    {
      "epoch": 3.476715087890625e-05,
      "step": 9114,
      "training_step_time": 9.914747714996338
    },
    {
      "epoch": 3.476715087890625e-05,
      "model_forward_time": 3.138045310974121,
      "step": 9114
    },
    {
      "epoch": 3.476715087890625e-05,
      "step": 9114,
      "training_step_time": 9.914119720458984
    },
    {
      "epoch": 3.476715087890625e-05,
      "model_forward_time": 3.1429929733276367,
      "step": 9114
    },
    {
      "epoch": 3.476715087890625e-05,
      "step": 9114,
      "training_step_time": 10.27244520187378
    },
    {
      "epoch": 3.4770965576171876e-05,
      "model_forward_time": 3.4057412147521973,
      "step": 9115
    },
    {
      "dynamics_loss_avg": 0.059812357649207115,
      "epoch": 3.4770965576171876e-05,
      "step": 9115
    },
    {
      "action_loss_avg": 0.01970740910619497,
      "epoch": 3.4770965576171876e-05,
      "step": 9115
    },
    {
      "epoch": 3.4770965576171876e-05,
      "step": 9115,
      "training_step_time": 10.185076713562012
    },
    {
      "epoch": 3.4770965576171876e-05,
      "model_forward_time": 3.1379592418670654,
      "step": 9115
    },
    {
      "epoch": 3.4770965576171876e-05,
      "step": 9115,
      "training_step_time": 9.910947561264038
    },
    {
      "epoch": 3.4770965576171876e-05,
      "model_forward_time": 3.1333460807800293,
      "step": 9115
    },
    {
      "epoch": 3.4770965576171876e-05,
      "step": 9115,
      "training_step_time": 9.909511804580688
    },
    {
      "epoch": 3.4770965576171876e-05,
      "model_forward_time": 3.132777452468872,
      "step": 9115
    },
    {
      "epoch": 3.4770965576171876e-05,
      "step": 9115,
      "training_step_time": 10.29695987701416
    },
    {
      "epoch": 3.47747802734375e-05,
      "model_forward_time": 3.1519901752471924,
      "step": 9116
    },
    {
      "epoch": 3.47747802734375e-05,
      "step": 9116,
      "training_step_time": 9.925938844680786
    },
    {
      "epoch": 3.47747802734375e-05,
      "model_forward_time": 3.131577253341675,
      "step": 9116
    },
    {
      "epoch": 3.47747802734375e-05,
      "step": 9116,
      "training_step_time": 10.182472705841064
    },
    {
      "epoch": 3.47747802734375e-05,
      "model_forward_time": 3.1324474811553955,
      "step": 9116
    },
    {
      "epoch": 3.47747802734375e-05,
      "step": 9116,
      "training_step_time": 9.90000319480896
    },
    {
      "epoch": 3.47747802734375e-05,
      "model_forward_time": 3.134220600128174,
      "step": 9116
    },
    {
      "epoch": 3.47747802734375e-05,
      "step": 9116,
      "training_step_time": 10.282974243164062
    },
    {
      "epoch": 3.477859497070313e-05,
      "model_forward_time": 3.144263505935669,
      "step": 9117
    },
    {
      "epoch": 3.477859497070313e-05,
      "step": 9117,
      "training_step_time": 9.9202880859375
    },
    {
      "epoch": 3.477859497070313e-05,
      "model_forward_time": 3.128613233566284,
      "step": 9117
    },
    {
      "epoch": 3.477859497070313e-05,
      "step": 9117,
      "training_step_time": 9.898873567581177
    },
    {
      "epoch": 3.477859497070313e-05,
      "model_forward_time": 3.4130876064300537,
      "step": 9117
    },
    {
      "dynamics_loss_avg": 0.08172408342361451,
      "epoch": 3.477859497070313e-05,
      "step": 9117
    },
    {
      "action_loss_avg": 0.02478034673258662,
      "epoch": 3.477859497070313e-05,
      "step": 9117
    },
    {
      "epoch": 3.477859497070313e-05,
      "step": 9117,
      "training_step_time": 10.184739351272583
    },
    {
      "epoch": 3.477859497070313e-05,
      "model_forward_time": 2.431230068206787,
      "step": 9117
    },
    {
      "epoch": 3.477859497070313e-05,
      "step": 9117,
      "training_step_time": 10.011981725692749
    },
    {
      "epoch": 3.478240966796875e-05,
      "model_forward_time": 3.1480629444122314,
      "step": 9118
    },
    {
      "epoch": 3.478240966796875e-05,
      "step": 9118,
      "training_step_time": 9.92278790473938
    },
    {
      "epoch": 3.478240966796875e-05,
      "model_forward_time": 2.427530288696289,
      "step": 9118
    },
    {
      "epoch": 3.478240966796875e-05,
      "step": 9118,
      "training_step_time": 7.460642576217651
    },
    {
      "epoch": 3.478240966796875e-05,
      "model_forward_time": 2.4274308681488037,
      "step": 9118
    },
    {
      "epoch": 3.478240966796875e-05,
      "step": 9118,
      "training_step_time": 7.458913803100586
    },
    {
      "epoch": 3.478240966796875e-05,
      "model_forward_time": 3.130620241165161,
      "step": 9118
    },
    {
      "epoch": 3.478240966796875e-05,
      "step": 9118,
      "training_step_time": 10.019879817962646
    },
    {
      "epoch": 3.4786224365234374e-05,
      "model_forward_time": 3.450608253479004,
      "step": 9119
    },
    {
      "epoch": 3.4786224365234374e-05,
      "step": 9119,
      "training_step_time": 10.232535362243652
    },
    {
      "epoch": 3.4786224365234374e-05,
      "model_forward_time": 3.134944438934326,
      "step": 9119
    },
    {
      "epoch": 3.4786224365234374e-05,
      "step": 9119,
      "training_step_time": 9.918946743011475
    },
    {
      "epoch": 3.4786224365234374e-05,
      "model_forward_time": 2.4331870079040527,
      "step": 9119
    },
    {
      "epoch": 3.4786224365234374e-05,
      "step": 9119,
      "training_step_time": 7.467355728149414
    },
    {
      "epoch": 3.4786224365234374e-05,
      "model_forward_time": 3.1328773498535156,
      "step": 9119
    },
    {
      "epoch": 3.4786224365234374e-05,
      "step": 9119,
      "training_step_time": 10.291773557662964
    },
    {
      "epoch": 3.47900390625e-05,
      "grad_norm": 0.49559932947158813,
      "learning_rate": 1e-05,
      "loss": 0.0824,
      "step": 9120
    },
    {
      "epoch": 3.47900390625e-05,
      "model_forward_time": 3.14587664604187,
      "step": 9120
    },
    {
      "dynamics_loss_avg": 0.05511340405791998,
      "epoch": 3.47900390625e-05,
      "step": 9120
    },
    {
      "action_loss_avg": 0.023430301714688538,
      "epoch": 3.47900390625e-05,
      "step": 9120
    },
    {
      "epoch": 3.47900390625e-05,
      "step": 9120,
      "training_step_time": 9.928965091705322
    },
    {
      "epoch": 3.47900390625e-05,
      "model_forward_time": 3.3857791423797607,
      "step": 9120
    },
    {
      "epoch": 3.47900390625e-05,
      "step": 9120,
      "training_step_time": 10.167168140411377
    },
    {
      "epoch": 3.47900390625e-05,
      "model_forward_time": 3.143735647201538,
      "step": 9120
    },
    {
      "epoch": 3.47900390625e-05,
      "step": 9120,
      "training_step_time": 9.923647165298462
    },
    {
      "epoch": 3.47900390625e-05,
      "model_forward_time": 2.4413509368896484,
      "step": 9120
    },
    {
      "epoch": 3.47900390625e-05,
      "step": 9120,
      "training_step_time": 10.322242975234985
    },
    {
      "epoch": 3.479385375976563e-05,
      "model_forward_time": 3.157017469406128,
      "step": 9121
    },
    {
      "epoch": 3.479385375976563e-05,
      "step": 9121,
      "training_step_time": 9.938376188278198
    },
    {
      "epoch": 3.479385375976563e-05,
      "model_forward_time": 2.4344000816345215,
      "step": 9121
    },
    {
      "epoch": 3.479385375976563e-05,
      "step": 9121,
      "training_step_time": 7.465468406677246
    },
    {
      "epoch": 3.479385375976563e-05,
      "model_forward_time": 3.4251773357391357,
      "step": 9121
    },
    {
      "epoch": 3.479385375976563e-05,
      "step": 9121,
      "training_step_time": 10.201762914657593
    },
    {
      "epoch": 3.479385375976563e-05,
      "model_forward_time": 3.135780096054077,
      "step": 9121
    },
    {
      "epoch": 3.479385375976563e-05,
      "step": 9121,
      "training_step_time": 10.281690835952759
    },
    {
      "epoch": 3.479766845703125e-05,
      "model_forward_time": 3.1582398414611816,
      "step": 9122
    },
    {
      "epoch": 3.479766845703125e-05,
      "step": 9122,
      "training_step_time": 9.93358826637268
    },
    {
      "epoch": 3.479766845703125e-05,
      "model_forward_time": 3.1359593868255615,
      "step": 9122
    },
    {
      "epoch": 3.479766845703125e-05,
      "step": 9122,
      "training_step_time": 9.906559228897095
    },
    {
      "epoch": 3.479766845703125e-05,
      "model_forward_time": 3.130051612854004,
      "step": 9122
    },
    {
      "dynamics_loss_avg": 0.06056766007095575,
      "epoch": 3.479766845703125e-05,
      "step": 9122
    },
    {
      "action_loss_avg": 0.018285329453647137,
      "epoch": 3.479766845703125e-05,
      "step": 9122
    },
    {
      "epoch": 3.479766845703125e-05,
      "step": 9122,
      "training_step_time": 9.90083909034729
    },
    {
      "epoch": 3.479766845703125e-05,
      "model_forward_time": 3.3848583698272705,
      "step": 9122
    },
    {
      "epoch": 3.479766845703125e-05,
      "step": 9122,
      "training_step_time": 10.289685726165771
    },
    {
      "epoch": 3.480148315429687e-05,
      "model_forward_time": 3.1493842601776123,
      "step": 9123
    },
    {
      "epoch": 3.480148315429687e-05,
      "step": 9123,
      "training_step_time": 9.924168348312378
    },
    {
      "epoch": 3.480148315429687e-05,
      "model_forward_time": 3.133071184158325,
      "step": 9123
    },
    {
      "epoch": 3.480148315429687e-05,
      "step": 9123,
      "training_step_time": 9.906931400299072
    },
    {
      "epoch": 3.480148315429687e-05,
      "model_forward_time": 3.134582281112671,
      "step": 9123
    },
    {
      "epoch": 3.480148315429687e-05,
      "step": 9123,
      "training_step_time": 9.908813238143921
    },
    {
      "epoch": 3.480148315429687e-05,
      "model_forward_time": 3.136918783187866,
      "step": 9123
    },
    {
      "epoch": 3.480148315429687e-05,
      "step": 9123,
      "training_step_time": 10.273141860961914
    },
    {
      "epoch": 3.48052978515625e-05,
      "model_forward_time": 3.153198719024658,
      "step": 9124
    },
    {
      "epoch": 3.48052978515625e-05,
      "step": 9124,
      "training_step_time": 9.930700778961182
    },
    {
      "epoch": 3.48052978515625e-05,
      "model_forward_time": 3.1370961666107178,
      "step": 9124
    },
    {
      "epoch": 3.48052978515625e-05,
      "step": 9124,
      "training_step_time": 9.91711163520813
    },
    {
      "epoch": 3.48052978515625e-05,
      "model_forward_time": 3.137423038482666,
      "step": 9124
    },
    {
      "epoch": 3.48052978515625e-05,
      "step": 9124,
      "training_step_time": 9.914238691329956
    },
    {
      "epoch": 3.48052978515625e-05,
      "model_forward_time": 3.1413464546203613,
      "step": 9124
    },
    {
      "epoch": 3.48052978515625e-05,
      "step": 9124,
      "training_step_time": 10.293172121047974
    },
    {
      "epoch": 3.4809112548828125e-05,
      "model_forward_time": 3.1531283855438232,
      "step": 9125
    },
    {
      "dynamics_loss_avg": 0.04277013298124075,
      "epoch": 3.4809112548828125e-05,
      "step": 9125
    },
    {
      "action_loss_avg": 0.013456177804619074,
      "epoch": 3.4809112548828125e-05,
      "step": 9125
    },
    {
      "epoch": 3.4809112548828125e-05,
      "step": 9125,
      "training_step_time": 10.219647884368896
    },
    {
      "epoch": 3.4809112548828125e-05,
      "model_forward_time": 3.144758939743042,
      "step": 9125
    },
    {
      "epoch": 3.4809112548828125e-05,
      "step": 9125,
      "training_step_time": 9.928313970565796
    },
    {
      "epoch": 3.4809112548828125e-05,
      "model_forward_time": 3.1394565105438232,
      "step": 9125
    },
    {
      "epoch": 3.4809112548828125e-05,
      "step": 9125,
      "training_step_time": 9.920684337615967
    },
    {
      "epoch": 3.4809112548828125e-05,
      "model_forward_time": 3.146308183670044,
      "step": 9125
    },
    {
      "epoch": 3.4809112548828125e-05,
      "step": 9125,
      "training_step_time": 10.01885199546814
    },
    {
      "epoch": 3.481292724609375e-05,
      "model_forward_time": 3.1676864624023438,
      "step": 9126
    },
    {
      "epoch": 3.481292724609375e-05,
      "step": 9126,
      "training_step_time": 9.947051763534546
    },
    {
      "epoch": 3.481292724609375e-05,
      "model_forward_time": 3.436572790145874,
      "step": 9126
    },
    {
      "epoch": 3.481292724609375e-05,
      "step": 9126,
      "training_step_time": 10.215903759002686
    },
    {
      "epoch": 3.481292724609375e-05,
      "model_forward_time": 3.1439263820648193,
      "step": 9126
    },
    {
      "epoch": 3.481292724609375e-05,
      "step": 9126,
      "training_step_time": 9.923316717147827
    },
    {
      "epoch": 3.481292724609375e-05,
      "model_forward_time": 2.4375908374786377,
      "step": 9126
    },
    {
      "epoch": 3.481292724609375e-05,
      "step": 9126,
      "training_step_time": 10.301111459732056
    },
    {
      "epoch": 3.481674194335938e-05,
      "model_forward_time": 3.1618518829345703,
      "step": 9127
    },
    {
      "epoch": 3.481674194335938e-05,
      "step": 9127,
      "training_step_time": 9.943408966064453
    },
    {
      "epoch": 3.481674194335938e-05,
      "model_forward_time": 3.136918544769287,
      "step": 9127
    },
    {
      "epoch": 3.481674194335938e-05,
      "step": 9127,
      "training_step_time": 9.911886930465698
    },
    {
      "epoch": 3.481674194335938e-05,
      "model_forward_time": 2.4371471405029297,
      "step": 9127
    },
    {
      "dynamics_loss_avg": 0.05455042887479067,
      "epoch": 3.481674194335938e-05,
      "step": 9127
    },
    {
      "action_loss_avg": 0.016274462966248394,
      "epoch": 3.481674194335938e-05,
      "step": 9127
    },
    {
      "epoch": 3.481674194335938e-05,
      "step": 9127,
      "training_step_time": 7.470215559005737
    },
    {
      "epoch": 3.481674194335938e-05,
      "model_forward_time": 3.44844913482666,
      "step": 9127
    },
    {
      "epoch": 3.481674194335938e-05,
      "step": 9127,
      "training_step_time": 10.317505836486816
    },
    {
      "epoch": 3.4820556640625e-05,
      "model_forward_time": 3.1582460403442383,
      "step": 9128
    },
    {
      "epoch": 3.4820556640625e-05,
      "step": 9128,
      "training_step_time": 9.955954313278198
    },
    {
      "epoch": 3.4820556640625e-05,
      "model_forward_time": 3.149569272994995,
      "step": 9128
    },
    {
      "epoch": 3.4820556640625e-05,
      "step": 9128,
      "training_step_time": 9.935235977172852
    },
    {
      "epoch": 3.4820556640625e-05,
      "model_forward_time": 3.160797357559204,
      "step": 9128
    },
    {
      "epoch": 3.4820556640625e-05,
      "step": 9128,
      "training_step_time": 9.941389560699463
    },
    {
      "epoch": 3.4820556640625e-05,
      "model_forward_time": 3.136666774749756,
      "step": 9128
    },
    {
      "epoch": 3.4820556640625e-05,
      "step": 9128,
      "training_step_time": 10.29789686203003
    },
    {
      "epoch": 3.482437133789062e-05,
      "model_forward_time": 3.1478374004364014,
      "step": 9129
    },
    {
      "epoch": 3.482437133789062e-05,
      "step": 9129,
      "training_step_time": 9.93642783164978
    },
    {
      "epoch": 3.482437133789062e-05,
      "model_forward_time": 3.159893751144409,
      "step": 9129
    },
    {
      "epoch": 3.482437133789062e-05,
      "step": 9129,
      "training_step_time": 9.94113826751709
    },
    {
      "epoch": 3.482437133789062e-05,
      "model_forward_time": 3.153306245803833,
      "step": 9129
    },
    {
      "epoch": 3.482437133789062e-05,
      "step": 9129,
      "training_step_time": 9.93279218673706
    },
    {
      "epoch": 3.482437133789062e-05,
      "model_forward_time": 3.144237756729126,
      "step": 9129
    },
    {
      "epoch": 3.482437133789062e-05,
      "step": 9129,
      "training_step_time": 10.016628503799438
    },
    {
      "epoch": 3.482818603515625e-05,
      "grad_norm": 0.3307078182697296,
      "learning_rate": 1e-05,
      "loss": 0.0764,
      "step": 9130
    },
    {
      "epoch": 3.482818603515625e-05,
      "model_forward_time": 3.144817352294922,
      "step": 9130
    },
    {
      "dynamics_loss_avg": 0.0713908176869154,
      "epoch": 3.482818603515625e-05,
      "step": 9130
    },
    {
      "action_loss_avg": 0.020696615148335696,
      "epoch": 3.482818603515625e-05,
      "step": 9130
    },
    {
      "epoch": 3.482818603515625e-05,
      "step": 9130,
      "training_step_time": 10.252649307250977
    },
    {
      "epoch": 3.482818603515625e-05,
      "model_forward_time": 2.4395413398742676,
      "step": 9130
    },
    {
      "epoch": 3.482818603515625e-05,
      "step": 9130,
      "training_step_time": 7.473585367202759
    },
    {
      "epoch": 3.482818603515625e-05,
      "model_forward_time": 3.1437296867370605,
      "step": 9130
    },
    {
      "epoch": 3.482818603515625e-05,
      "step": 9130,
      "training_step_time": 9.922561407089233
    },
    {
      "epoch": 3.482818603515625e-05,
      "model_forward_time": 3.136295795440674,
      "step": 9130
    },
    {
      "epoch": 3.482818603515625e-05,
      "step": 9130,
      "training_step_time": 10.314257860183716
    },
    {
      "epoch": 3.4832000732421876e-05,
      "model_forward_time": 3.147326707839966,
      "step": 9131
    },
    {
      "epoch": 3.4832000732421876e-05,
      "step": 9131,
      "training_step_time": 9.925727367401123
    },
    {
      "epoch": 3.4832000732421876e-05,
      "model_forward_time": 3.1425466537475586,
      "step": 9131
    },
    {
      "epoch": 3.4832000732421876e-05,
      "step": 9131,
      "training_step_time": 10.237584590911865
    },
    {
      "epoch": 3.4832000732421876e-05,
      "model_forward_time": 3.1453773975372314,
      "step": 9131
    },
    {
      "epoch": 3.4832000732421876e-05,
      "step": 9131,
      "training_step_time": 9.927083015441895
    },
    {
      "epoch": 3.4832000732421876e-05,
      "model_forward_time": 3.1424477100372314,
      "step": 9131
    },
    {
      "epoch": 3.4832000732421876e-05,
      "step": 9131,
      "training_step_time": 10.277114629745483
    },
    {
      "epoch": 3.48358154296875e-05,
      "model_forward_time": 3.1465280055999756,
      "step": 9132
    },
    {
      "epoch": 3.48358154296875e-05,
      "step": 9132,
      "training_step_time": 9.927836894989014
    },
    {
      "epoch": 3.48358154296875e-05,
      "model_forward_time": 2.4389560222625732,
      "step": 9132
    },
    {
      "epoch": 3.48358154296875e-05,
      "step": 9132,
      "training_step_time": 7.476700067520142
    },
    {
      "epoch": 3.48358154296875e-05,
      "model_forward_time": 3.432034492492676,
      "step": 9132
    },
    {
      "dynamics_loss_avg": 0.05520546492189169,
      "epoch": 3.48358154296875e-05,
      "step": 9132
    },
    {
      "action_loss_avg": 0.018189293378964067,
      "epoch": 3.48358154296875e-05,
      "step": 9132
    },
    {
      "epoch": 3.48358154296875e-05,
      "step": 9132,
      "training_step_time": 10.215104103088379
    },
    {
      "epoch": 3.48358154296875e-05,
      "model_forward_time": 3.1417195796966553,
      "step": 9132
    },
    {
      "epoch": 3.48358154296875e-05,
      "step": 9132,
      "training_step_time": 10.312195777893066
    },
    {
      "epoch": 3.483963012695312e-05,
      "model_forward_time": 3.1563050746917725,
      "step": 9133
    },
    {
      "epoch": 3.483963012695312e-05,
      "step": 9133,
      "training_step_time": 9.940498113632202
    },
    {
      "epoch": 3.483963012695312e-05,
      "model_forward_time": 3.142672300338745,
      "step": 9133
    },
    {
      "epoch": 3.483963012695312e-05,
      "step": 9133,
      "training_step_time": 9.918353080749512
    },
    {
      "epoch": 3.483963012695312e-05,
      "model_forward_time": 3.139601230621338,
      "step": 9133
    },
    {
      "epoch": 3.483963012695312e-05,
      "step": 9133,
      "training_step_time": 9.916141033172607
    },
    {
      "epoch": 3.483963012695312e-05,
      "model_forward_time": 3.1415390968322754,
      "step": 9133
    },
    {
      "epoch": 3.483963012695312e-05,
      "step": 9133,
      "training_step_time": 10.29460334777832
    },
    {
      "epoch": 3.484344482421875e-05,
      "model_forward_time": 3.1620304584503174,
      "step": 9134
    },
    {
      "epoch": 3.484344482421875e-05,
      "step": 9134,
      "training_step_time": 9.94776701927185
    },
    {
      "epoch": 3.484344482421875e-05,
      "model_forward_time": 3.147226333618164,
      "step": 9134
    },
    {
      "epoch": 3.484344482421875e-05,
      "step": 9134,
      "training_step_time": 9.931017398834229
    },
    {
      "epoch": 3.484344482421875e-05,
      "model_forward_time": 3.1499757766723633,
      "step": 9134
    },
    {
      "epoch": 3.484344482421875e-05,
      "step": 9134,
      "training_step_time": 9.929511070251465
    },
    {
      "epoch": 3.484344482421875e-05,
      "model_forward_time": 3.147003412246704,
      "step": 9134
    },
    {
      "epoch": 3.484344482421875e-05,
      "step": 9134,
      "training_step_time": 10.02159571647644
    },
    {
      "epoch": 3.4847259521484374e-05,
      "model_forward_time": 3.447110652923584,
      "step": 9135
    },
    {
      "dynamics_loss_avg": 0.051795424334704875,
      "epoch": 3.4847259521484374e-05,
      "step": 9135
    },
    {
      "action_loss_avg": 0.01644981554709375,
      "epoch": 3.4847259521484374e-05,
      "step": 9135
    },
    {
      "epoch": 3.4847259521484374e-05,
      "step": 9135,
      "training_step_time": 10.22890329360962
    },
    {
      "epoch": 3.4847259521484374e-05,
      "model_forward_time": 3.1426942348480225,
      "step": 9135
    },
    {
      "epoch": 3.4847259521484374e-05,
      "step": 9135,
      "training_step_time": 9.920768737792969
    },
    {
      "epoch": 3.4847259521484374e-05,
      "model_forward_time": 2.4423904418945312,
      "step": 9135
    },
    {
      "epoch": 3.4847259521484374e-05,
      "step": 9135,
      "training_step_time": 7.478392124176025
    },
    {
      "epoch": 3.4847259521484374e-05,
      "model_forward_time": 2.436922311782837,
      "step": 9135
    },
    {
      "epoch": 3.4847259521484374e-05,
      "step": 9135,
      "training_step_time": 10.291537046432495
    },
    {
      "epoch": 3.485107421875e-05,
      "model_forward_time": 3.152946949005127,
      "step": 9136
    },
    {
      "epoch": 3.485107421875e-05,
      "step": 9136,
      "training_step_time": 9.924484729766846
    },
    {
      "epoch": 3.485107421875e-05,
      "model_forward_time": 2.43404221534729,
      "step": 9136
    },
    {
      "epoch": 3.485107421875e-05,
      "step": 9136,
      "training_step_time": 7.462112665176392
    },
    {
      "epoch": 3.485107421875e-05,
      "model_forward_time": 3.4359476566314697,
      "step": 9136
    },
    {
      "epoch": 3.485107421875e-05,
      "step": 9136,
      "training_step_time": 10.216376543045044
    },
    {
      "epoch": 3.485107421875e-05,
      "model_forward_time": 3.148026704788208,
      "step": 9136
    },
    {
      "epoch": 3.485107421875e-05,
      "step": 9136,
      "training_step_time": 10.274569272994995
    },
    {
      "epoch": 3.485488891601563e-05,
      "model_forward_time": 3.1660189628601074,
      "step": 9137
    },
    {
      "epoch": 3.485488891601563e-05,
      "step": 9137,
      "training_step_time": 9.95521330833435
    },
    {
      "epoch": 3.485488891601563e-05,
      "model_forward_time": 3.144378662109375,
      "step": 9137
    },
    {
      "epoch": 3.485488891601563e-05,
      "step": 9137,
      "training_step_time": 9.925534725189209
    },
    {
      "epoch": 3.485488891601563e-05,
      "model_forward_time": 3.1477930545806885,
      "step": 9137
    },
    {
      "dynamics_loss_avg": 0.0536962291225791,
      "epoch": 3.485488891601563e-05,
      "step": 9137
    },
    {
      "action_loss_avg": 0.016750614950433375,
      "epoch": 3.485488891601563e-05,
      "step": 9137
    },
    {
      "epoch": 3.485488891601563e-05,
      "step": 9137,
      "training_step_time": 10.236821413040161
    },
    {
      "epoch": 3.485488891601563e-05,
      "model_forward_time": 3.148594856262207,
      "step": 9137
    },
    {
      "epoch": 3.485488891601563e-05,
      "step": 9137,
      "training_step_time": 10.32709550857544
    },
    {
      "epoch": 3.485870361328125e-05,
      "model_forward_time": 3.1574082374572754,
      "step": 9138
    },
    {
      "epoch": 3.485870361328125e-05,
      "step": 9138,
      "training_step_time": 9.939146995544434
    },
    {
      "epoch": 3.485870361328125e-05,
      "model_forward_time": 3.1410491466522217,
      "step": 9138
    },
    {
      "epoch": 3.485870361328125e-05,
      "step": 9138,
      "training_step_time": 9.916738033294678
    },
    {
      "epoch": 3.485870361328125e-05,
      "model_forward_time": 3.148165464401245,
      "step": 9138
    },
    {
      "epoch": 3.485870361328125e-05,
      "step": 9138,
      "training_step_time": 9.923969745635986
    },
    {
      "epoch": 3.485870361328125e-05,
      "model_forward_time": 3.426896333694458,
      "step": 9138
    },
    {
      "epoch": 3.485870361328125e-05,
      "step": 9138,
      "training_step_time": 10.300551652908325
    },
    {
      "epoch": 3.486251831054687e-05,
      "model_forward_time": 3.1560583114624023,
      "step": 9139
    },
    {
      "epoch": 3.486251831054687e-05,
      "step": 9139,
      "training_step_time": 9.936436653137207
    },
    {
      "epoch": 3.486251831054687e-05,
      "model_forward_time": 3.138551712036133,
      "step": 9139
    },
    {
      "epoch": 3.486251831054687e-05,
      "step": 9139,
      "training_step_time": 9.905235528945923
    },
    {
      "epoch": 3.486251831054687e-05,
      "model_forward_time": 3.139376640319824,
      "step": 9139
    },
    {
      "epoch": 3.486251831054687e-05,
      "step": 9139,
      "training_step_time": 9.903955221176147
    },
    {
      "epoch": 3.486251831054687e-05,
      "model_forward_time": 3.1422011852264404,
      "step": 9139
    },
    {
      "epoch": 3.486251831054687e-05,
      "step": 9139,
      "training_step_time": 10.329009294509888
    },
    {
      "epoch": 3.48663330078125e-05,
      "grad_norm": 0.43855926394462585,
      "learning_rate": 1e-05,
      "loss": 0.0767,
      "step": 9140
    },
    {
      "epoch": 3.48663330078125e-05,
      "model_forward_time": 2.448014974594116,
      "step": 9140
    },
    {
      "dynamics_loss_avg": 0.05305119310505688,
      "epoch": 3.48663330078125e-05,
      "step": 9140
    },
    {
      "action_loss_avg": 0.01653342517092824,
      "epoch": 3.48663330078125e-05,
      "step": 9140
    },
    {
      "epoch": 3.48663330078125e-05,
      "step": 9140,
      "training_step_time": 7.796673536300659
    },
    {
      "epoch": 3.48663330078125e-05,
      "model_forward_time": 3.140622854232788,
      "step": 9140
    },
    {
      "epoch": 3.48663330078125e-05,
      "step": 9140,
      "training_step_time": 9.912725448608398
    },
    {
      "epoch": 3.48663330078125e-05,
      "model_forward_time": 2.4385793209075928,
      "step": 9140
    },
    {
      "epoch": 3.48663330078125e-05,
      "step": 9140,
      "training_step_time": 7.466840982437134
    },
    {
      "epoch": 3.48663330078125e-05,
      "model_forward_time": 3.1344549655914307,
      "step": 9140
    },
    {
      "epoch": 3.48663330078125e-05,
      "step": 9140,
      "training_step_time": 10.01608681678772
    },
    {
      "epoch": 3.4870147705078125e-05,
      "model_forward_time": 3.1554126739501953,
      "step": 9141
    },
    {
      "epoch": 3.4870147705078125e-05,
      "step": 9141,
      "training_step_time": 9.92947006225586
    },
    {
      "epoch": 3.4870147705078125e-05,
      "model_forward_time": 3.1462628841400146,
      "step": 9141
    },
    {
      "epoch": 3.4870147705078125e-05,
      "step": 9141,
      "training_step_time": 9.918363809585571
    },
    {
      "epoch": 3.4870147705078125e-05,
      "model_forward_time": 3.4292802810668945,
      "step": 9141
    },
    {
      "epoch": 3.4870147705078125e-05,
      "step": 9141,
      "training_step_time": 10.204264640808105
    },
    {
      "epoch": 3.4870147705078125e-05,
      "model_forward_time": 3.143050193786621,
      "step": 9141
    },
    {
      "epoch": 3.4870147705078125e-05,
      "step": 9141,
      "training_step_time": 10.281666040420532
    },
    {
      "epoch": 3.487396240234375e-05,
      "model_forward_time": 3.147843599319458,
      "step": 9142
    },
    {
      "epoch": 3.487396240234375e-05,
      "step": 9142,
      "training_step_time": 9.927109241485596
    },
    {
      "epoch": 3.487396240234375e-05,
      "model_forward_time": 3.151013135910034,
      "step": 9142
    },
    {
      "epoch": 3.487396240234375e-05,
      "step": 9142,
      "training_step_time": 9.926801204681396
    },
    {
      "epoch": 3.487396240234375e-05,
      "model_forward_time": 3.430541753768921,
      "step": 9142
    },
    {
      "dynamics_loss_avg": 0.05689334496855736,
      "epoch": 3.487396240234375e-05,
      "step": 9142
    },
    {
      "action_loss_avg": 0.019209937611594795,
      "epoch": 3.487396240234375e-05,
      "step": 9142
    },
    {
      "epoch": 3.487396240234375e-05,
      "step": 9142,
      "training_step_time": 10.213507890701294
    },
    {
      "epoch": 3.487396240234375e-05,
      "model_forward_time": 3.146085262298584,
      "step": 9142
    },
    {
      "epoch": 3.487396240234375e-05,
      "step": 9142,
      "training_step_time": 10.29940152168274
    },
    {
      "epoch": 3.487777709960938e-05,
      "model_forward_time": 3.1654820442199707,
      "step": 9143
    },
    {
      "epoch": 3.487777709960938e-05,
      "step": 9143,
      "training_step_time": 9.93900442123413
    },
    {
      "epoch": 3.487777709960938e-05,
      "model_forward_time": 3.14194393157959,
      "step": 9143
    },
    {
      "epoch": 3.487777709960938e-05,
      "step": 9143,
      "training_step_time": 9.918253421783447
    },
    {
      "epoch": 3.487777709960938e-05,
      "model_forward_time": 3.1414568424224854,
      "step": 9143
    },
    {
      "epoch": 3.487777709960938e-05,
      "step": 9143,
      "training_step_time": 9.92487621307373
    },
    {
      "epoch": 3.487777709960938e-05,
      "model_forward_time": 3.434819221496582,
      "step": 9143
    },
    {
      "epoch": 3.487777709960938e-05,
      "step": 9143,
      "training_step_time": 10.312760829925537
    },
    {
      "epoch": 3.4881591796875e-05,
      "model_forward_time": 3.159038782119751,
      "step": 9144
    },
    {
      "epoch": 3.4881591796875e-05,
      "step": 9144,
      "training_step_time": 9.943833112716675
    },
    {
      "epoch": 3.4881591796875e-05,
      "model_forward_time": 3.142319917678833,
      "step": 9144
    },
    {
      "epoch": 3.4881591796875e-05,
      "step": 9144,
      "training_step_time": 9.923089742660522
    },
    {
      "epoch": 3.4881591796875e-05,
      "model_forward_time": 3.138697385787964,
      "step": 9144
    },
    {
      "epoch": 3.4881591796875e-05,
      "step": 9144,
      "training_step_time": 9.916475534439087
    },
    {
      "epoch": 3.4881591796875e-05,
      "model_forward_time": 3.142515182495117,
      "step": 9144
    },
    {
      "epoch": 3.4881591796875e-05,
      "step": 9144,
      "training_step_time": 10.273211240768433
    },
    {
      "epoch": 3.488540649414062e-05,
      "model_forward_time": 3.40436053276062,
      "step": 9145
    },
    {
      "dynamics_loss_avg": 0.06264945808798075,
      "epoch": 3.488540649414062e-05,
      "step": 9145
    },
    {
      "action_loss_avg": 0.019805799424648284,
      "epoch": 3.488540649414062e-05,
      "step": 9145
    },
    {
      "epoch": 3.488540649414062e-05,
      "step": 9145,
      "training_step_time": 10.182850122451782
    },
    {
      "epoch": 3.488540649414062e-05,
      "model_forward_time": 3.1407954692840576,
      "step": 9145
    },
    {
      "epoch": 3.488540649414062e-05,
      "step": 9145,
      "training_step_time": 9.917389869689941
    },
    {
      "epoch": 3.488540649414062e-05,
      "model_forward_time": 3.1394379138946533,
      "step": 9145
    },
    {
      "epoch": 3.488540649414062e-05,
      "step": 9145,
      "training_step_time": 9.918087720870972
    },
    {
      "epoch": 3.488540649414062e-05,
      "model_forward_time": 3.1417064666748047,
      "step": 9145
    },
    {
      "epoch": 3.488540649414062e-05,
      "step": 9145,
      "training_step_time": 10.028066635131836
    },
    {
      "epoch": 3.488922119140625e-05,
      "model_forward_time": 3.155456066131592,
      "step": 9146
    },
    {
      "epoch": 3.488922119140625e-05,
      "step": 9146,
      "training_step_time": 9.934017658233643
    },
    {
      "epoch": 3.488922119140625e-05,
      "model_forward_time": 2.7149887084960938,
      "step": 9146
    },
    {
      "epoch": 3.488922119140625e-05,
      "step": 9146,
      "training_step_time": 7.741135358810425
    },
    {
      "epoch": 3.488922119140625e-05,
      "model_forward_time": 3.1337764263153076,
      "step": 9146
    },
    {
      "epoch": 3.488922119140625e-05,
      "step": 9146,
      "training_step_time": 9.909276247024536
    },
    {
      "epoch": 3.488922119140625e-05,
      "model_forward_time": 3.140491008758545,
      "step": 9146
    },
    {
      "epoch": 3.488922119140625e-05,
      "step": 9146,
      "training_step_time": 10.301210880279541
    },
    {
      "epoch": 3.4893035888671876e-05,
      "model_forward_time": 3.1498610973358154,
      "step": 9147
    },
    {
      "epoch": 3.4893035888671876e-05,
      "step": 9147,
      "training_step_time": 9.927258253097534
    },
    {
      "epoch": 3.4893035888671876e-05,
      "model_forward_time": 3.1353073120117188,
      "step": 9147
    },
    {
      "epoch": 3.4893035888671876e-05,
      "step": 9147,
      "training_step_time": 9.91361927986145
    },
    {
      "epoch": 3.4893035888671876e-05,
      "model_forward_time": 3.1406941413879395,
      "step": 9147
    },
    {
      "dynamics_loss_avg": 0.05961187221109867,
      "epoch": 3.4893035888671876e-05,
      "step": 9147
    },
    {
      "action_loss_avg": 0.022038233652710914,
      "epoch": 3.4893035888671876e-05,
      "step": 9147
    },
    {
      "epoch": 3.4893035888671876e-05,
      "step": 9147,
      "training_step_time": 10.191259622573853
    },
    {
      "epoch": 3.4893035888671876e-05,
      "model_forward_time": 3.13710880279541,
      "step": 9147
    },
    {
      "epoch": 3.4893035888671876e-05,
      "step": 9147,
      "training_step_time": 10.299952745437622
    },
    {
      "epoch": 3.48968505859375e-05,
      "model_forward_time": 2.4500415325164795,
      "step": 9148
    },
    {
      "epoch": 3.48968505859375e-05,
      "step": 9148,
      "training_step_time": 7.484224319458008
    },
    {
      "epoch": 3.48968505859375e-05,
      "model_forward_time": 3.134610176086426,
      "step": 9148
    },
    {
      "epoch": 3.48968505859375e-05,
      "step": 9148,
      "training_step_time": 9.91230320930481
    },
    {
      "epoch": 3.48968505859375e-05,
      "model_forward_time": 3.1397416591644287,
      "step": 9148
    },
    {
      "epoch": 3.48968505859375e-05,
      "step": 9148,
      "training_step_time": 9.91579008102417
    },
    {
      "epoch": 3.48968505859375e-05,
      "model_forward_time": 3.136899948120117,
      "step": 9148
    },
    {
      "epoch": 3.48968505859375e-05,
      "step": 9148,
      "training_step_time": 10.29384994506836
    },
    {
      "epoch": 3.490066528320312e-05,
      "model_forward_time": 3.1474902629852295,
      "step": 9149
    },
    {
      "epoch": 3.490066528320312e-05,
      "step": 9149,
      "training_step_time": 9.929977416992188
    },
    {
      "epoch": 3.490066528320312e-05,
      "model_forward_time": 3.137413501739502,
      "step": 9149
    },
    {
      "epoch": 3.490066528320312e-05,
      "step": 9149,
      "training_step_time": 9.911985635757446
    },
    {
      "epoch": 3.490066528320312e-05,
      "model_forward_time": 3.140695095062256,
      "step": 9149
    },
    {
      "epoch": 3.490066528320312e-05,
      "step": 9149,
      "training_step_time": 9.921074867248535
    },
    {
      "epoch": 3.490066528320312e-05,
      "model_forward_time": 3.1407973766326904,
      "step": 9149
    },
    {
      "epoch": 3.490066528320312e-05,
      "step": 9149,
      "training_step_time": 10.294023513793945
    },
    {
      "epoch": 3.490447998046875e-05,
      "grad_norm": 0.4754078984260559,
      "learning_rate": 1e-05,
      "loss": 0.0779,
      "step": 9150
    },
    {
      "epoch": 3.490447998046875e-05,
      "model_forward_time": 2.443342685699463,
      "step": 9150
    },
    {
      "dynamics_loss_avg": 0.06203106790781021,
      "epoch": 3.490447998046875e-05,
      "step": 9150
    },
    {
      "action_loss_avg": 0.02700124797411263,
      "epoch": 3.490447998046875e-05,
      "step": 9150
    },
    {
      "epoch": 3.490447998046875e-05,
      "step": 9150,
      "training_step_time": 7.492949485778809
    },
    {
      "epoch": 3.490447998046875e-05,
      "model_forward_time": 3.41298508644104,
      "step": 9150
    },
    {
      "epoch": 3.490447998046875e-05,
      "step": 9150,
      "training_step_time": 10.18669605255127
    },
    {
      "epoch": 3.490447998046875e-05,
      "model_forward_time": 3.1383676528930664,
      "step": 9150
    },
    {
      "epoch": 3.490447998046875e-05,
      "step": 9150,
      "training_step_time": 9.909178972244263
    },
    {
      "epoch": 3.490447998046875e-05,
      "model_forward_time": 3.1340529918670654,
      "step": 9150
    },
    {
      "epoch": 3.490447998046875e-05,
      "step": 9150,
      "training_step_time": 10.02516770362854
    },
    {
      "epoch": 3.4908294677734374e-05,
      "model_forward_time": 3.151951551437378,
      "step": 9151
    },
    {
      "epoch": 3.4908294677734374e-05,
      "step": 9151,
      "training_step_time": 9.938108205795288
    },
    {
      "epoch": 3.4908294677734374e-05,
      "model_forward_time": 2.4344966411590576,
      "step": 9151
    },
    {
      "epoch": 3.4908294677734374e-05,
      "step": 9151,
      "training_step_time": 7.469247579574585
    },
    {
      "epoch": 3.4908294677734374e-05,
      "model_forward_time": 3.418261766433716,
      "step": 9151
    },
    {
      "epoch": 3.4908294677734374e-05,
      "step": 9151,
      "training_step_time": 10.193695068359375
    },
    {
      "epoch": 3.4908294677734374e-05,
      "model_forward_time": 3.1363608837127686,
      "step": 9151
    },
    {
      "epoch": 3.4908294677734374e-05,
      "step": 9151,
      "training_step_time": 10.032375812530518
    },
    {
      "epoch": 3.4912109375e-05,
      "model_forward_time": 3.161600351333618,
      "step": 9152
    },
    {
      "epoch": 3.4912109375e-05,
      "step": 9152,
      "training_step_time": 9.9503755569458
    },
    {
      "epoch": 3.4912109375e-05,
      "model_forward_time": 3.145089864730835,
      "step": 9152
    },
    {
      "epoch": 3.4912109375e-05,
      "step": 9152,
      "training_step_time": 9.918156385421753
    },
    {
      "epoch": 3.4912109375e-05,
      "model_forward_time": 3.436239242553711,
      "step": 9152
    },
    {
      "dynamics_loss_avg": 0.06117832809686661,
      "epoch": 3.4912109375e-05,
      "step": 9152
    },
    {
      "action_loss_avg": 0.023186704609543085,
      "epoch": 3.4912109375e-05,
      "step": 9152
    },
    {
      "epoch": 3.4912109375e-05,
      "step": 9152,
      "training_step_time": 10.214522123336792
    },
    {
      "epoch": 3.4912109375e-05,
      "model_forward_time": 2.4414968490600586,
      "step": 9152
    },
    {
      "epoch": 3.4912109375e-05,
      "step": 9152,
      "training_step_time": 10.296915531158447
    },
    {
      "epoch": 3.491592407226563e-05,
      "model_forward_time": 2.4446587562561035,
      "step": 9153
    },
    {
      "epoch": 3.491592407226563e-05,
      "step": 9153,
      "training_step_time": 7.47565221786499
    },
    {
      "epoch": 3.491592407226563e-05,
      "model_forward_time": 3.144386053085327,
      "step": 9153
    },
    {
      "epoch": 3.491592407226563e-05,
      "step": 9153,
      "training_step_time": 9.919865369796753
    },
    {
      "epoch": 3.491592407226563e-05,
      "model_forward_time": 2.4357213973999023,
      "step": 9153
    },
    {
      "epoch": 3.491592407226563e-05,
      "step": 9153,
      "training_step_time": 7.4687275886535645
    },
    {
      "epoch": 3.491592407226563e-05,
      "model_forward_time": 3.142561912536621,
      "step": 9153
    },
    {
      "epoch": 3.491592407226563e-05,
      "step": 9153,
      "training_step_time": 10.339584350585938
    },
    {
      "epoch": 3.491973876953125e-05,
      "model_forward_time": 3.4669106006622314,
      "step": 9154
    },
    {
      "epoch": 3.491973876953125e-05,
      "step": 9154,
      "training_step_time": 10.25514554977417
    },
    {
      "epoch": 3.491973876953125e-05,
      "model_forward_time": 3.1481845378875732,
      "step": 9154
    },
    {
      "epoch": 3.491973876953125e-05,
      "step": 9154,
      "training_step_time": 9.926844358444214
    },
    {
      "epoch": 3.491973876953125e-05,
      "model_forward_time": 3.1443874835968018,
      "step": 9154
    },
    {
      "epoch": 3.491973876953125e-05,
      "step": 9154,
      "training_step_time": 9.925662517547607
    },
    {
      "epoch": 3.491973876953125e-05,
      "model_forward_time": 3.1506564617156982,
      "step": 9154
    },
    {
      "epoch": 3.491973876953125e-05,
      "step": 9154,
      "training_step_time": 10.030073642730713
    },
    {
      "epoch": 3.492355346679687e-05,
      "model_forward_time": 3.1586992740631104,
      "step": 9155
    },
    {
      "dynamics_loss_avg": 0.06012876681052148,
      "epoch": 3.492355346679687e-05,
      "step": 9155
    },
    {
      "action_loss_avg": 0.024704523070249708,
      "epoch": 3.492355346679687e-05,
      "step": 9155
    },
    {
      "epoch": 3.492355346679687e-05,
      "step": 9155,
      "training_step_time": 9.941665410995483
    },
    {
      "epoch": 3.492355346679687e-05,
      "model_forward_time": 3.4422640800476074,
      "step": 9155
    },
    {
      "epoch": 3.492355346679687e-05,
      "step": 9155,
      "training_step_time": 10.214481353759766
    },
    {
      "epoch": 3.492355346679687e-05,
      "model_forward_time": 3.1357076168060303,
      "step": 9155
    },
    {
      "epoch": 3.492355346679687e-05,
      "step": 9155,
      "training_step_time": 9.90650725364685
    },
    {
      "epoch": 3.492355346679687e-05,
      "model_forward_time": 3.1390371322631836,
      "step": 9155
    },
    {
      "epoch": 3.492355346679687e-05,
      "step": 9155,
      "training_step_time": 10.009354829788208
    },
    {
      "epoch": 3.49273681640625e-05,
      "model_forward_time": 3.1480295658111572,
      "step": 9156
    },
    {
      "epoch": 3.49273681640625e-05,
      "step": 9156,
      "training_step_time": 9.925832271575928
    },
    {
      "epoch": 3.49273681640625e-05,
      "model_forward_time": 3.1410136222839355,
      "step": 9156
    },
    {
      "epoch": 3.49273681640625e-05,
      "step": 9156,
      "training_step_time": 9.913455486297607
    },
    {
      "epoch": 3.49273681640625e-05,
      "model_forward_time": 3.4277853965759277,
      "step": 9156
    },
    {
      "epoch": 3.49273681640625e-05,
      "step": 9156,
      "training_step_time": 10.20152473449707
    },
    {
      "epoch": 3.49273681640625e-05,
      "model_forward_time": 3.139944553375244,
      "step": 9156
    },
    {
      "epoch": 3.49273681640625e-05,
      "step": 9156,
      "training_step_time": 10.012382745742798
    },
    {
      "epoch": 3.4931182861328125e-05,
      "model_forward_time": 3.1406426429748535,
      "step": 9157
    },
    {
      "epoch": 3.4931182861328125e-05,
      "step": 9157,
      "training_step_time": 9.909883737564087
    },
    {
      "epoch": 3.4931182861328125e-05,
      "model_forward_time": 3.129983425140381,
      "step": 9157
    },
    {
      "epoch": 3.4931182861328125e-05,
      "step": 9157,
      "training_step_time": 9.898536205291748
    },
    {
      "epoch": 3.4931182861328125e-05,
      "model_forward_time": 3.396913766860962,
      "step": 9157
    },
    {
      "dynamics_loss_avg": 0.0510133434087038,
      "epoch": 3.4931182861328125e-05,
      "step": 9157
    },
    {
      "action_loss_avg": 0.015246582310646772,
      "epoch": 3.4931182861328125e-05,
      "step": 9157
    },
    {
      "epoch": 3.4931182861328125e-05,
      "step": 9157,
      "training_step_time": 10.168825149536133
    },
    {
      "epoch": 3.4931182861328125e-05,
      "model_forward_time": 3.1379525661468506,
      "step": 9157
    },
    {
      "epoch": 3.4931182861328125e-05,
      "step": 9157,
      "training_step_time": 10.2928946018219
    },
    {
      "epoch": 3.493499755859375e-05,
      "model_forward_time": 3.1414449214935303,
      "step": 9158
    },
    {
      "epoch": 3.493499755859375e-05,
      "step": 9158,
      "training_step_time": 9.915897130966187
    },
    {
      "epoch": 3.493499755859375e-05,
      "model_forward_time": 3.1328980922698975,
      "step": 9158
    },
    {
      "epoch": 3.493499755859375e-05,
      "step": 9158,
      "training_step_time": 9.90697193145752
    },
    {
      "epoch": 3.493499755859375e-05,
      "model_forward_time": 3.138291597366333,
      "step": 9158
    },
    {
      "epoch": 3.493499755859375e-05,
      "step": 9158,
      "training_step_time": 9.912997245788574
    },
    {
      "epoch": 3.493499755859375e-05,
      "model_forward_time": 2.4353902339935303,
      "step": 9158
    },
    {
      "epoch": 3.493499755859375e-05,
      "step": 9158,
      "training_step_time": 10.30198335647583
    },
    {
      "epoch": 3.493881225585938e-05,
      "model_forward_time": 3.4578261375427246,
      "step": 9159
    },
    {
      "epoch": 3.493881225585938e-05,
      "step": 9159,
      "training_step_time": 10.234392642974854
    },
    {
      "epoch": 3.493881225585938e-05,
      "model_forward_time": 2.4308271408081055,
      "step": 9159
    },
    {
      "epoch": 3.493881225585938e-05,
      "step": 9159,
      "training_step_time": 7.462622165679932
    },
    {
      "epoch": 3.493881225585938e-05,
      "model_forward_time": 3.147692918777466,
      "step": 9159
    },
    {
      "epoch": 3.493881225585938e-05,
      "step": 9159,
      "training_step_time": 9.923080682754517
    },
    {
      "epoch": 3.493881225585938e-05,
      "model_forward_time": 3.130476713180542,
      "step": 9159
    },
    {
      "epoch": 3.493881225585938e-05,
      "step": 9159,
      "training_step_time": 10.294475317001343
    },
    {
      "epoch": 3.4942626953125e-05,
      "grad_norm": 0.4124305844306946,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 9160
    },
    {
      "epoch": 3.4942626953125e-05,
      "model_forward_time": 3.138801097869873,
      "step": 9160
    },
    {
      "dynamics_loss_avg": 0.05851899385452271,
      "epoch": 3.4942626953125e-05,
      "step": 9160
    },
    {
      "action_loss_avg": 0.01897413320839405,
      "epoch": 3.4942626953125e-05,
      "step": 9160
    },
    {
      "epoch": 3.4942626953125e-05,
      "step": 9160,
      "training_step_time": 10.19179105758667
    },
    {
      "epoch": 3.4942626953125e-05,
      "model_forward_time": 2.4329729080200195,
      "step": 9160
    },
    {
      "epoch": 3.4942626953125e-05,
      "step": 9160,
      "training_step_time": 7.465656042098999
    },
    {
      "epoch": 3.4942626953125e-05,
      "model_forward_time": 3.1358819007873535,
      "step": 9160
    },
    {
      "epoch": 3.4942626953125e-05,
      "step": 9160,
      "training_step_time": 9.910570621490479
    },
    {
      "epoch": 3.4942626953125e-05,
      "model_forward_time": 3.1371469497680664,
      "step": 9160
    },
    {
      "epoch": 3.4942626953125e-05,
      "step": 9160,
      "training_step_time": 10.038290023803711
    },
    {
      "epoch": 3.4946441650390623e-05,
      "model_forward_time": 3.148566484451294,
      "step": 9161
    },
    {
      "epoch": 3.4946441650390623e-05,
      "step": 9161,
      "training_step_time": 9.928408861160278
    },
    {
      "epoch": 3.4946441650390623e-05,
      "model_forward_time": 2.4335572719573975,
      "step": 9161
    },
    {
      "epoch": 3.4946441650390623e-05,
      "step": 9161,
      "training_step_time": 7.465413570404053
    },
    {
      "epoch": 3.4946441650390623e-05,
      "model_forward_time": 3.40340518951416,
      "step": 9161
    },
    {
      "epoch": 3.4946441650390623e-05,
      "step": 9161,
      "training_step_time": 10.173654794692993
    },
    {
      "epoch": 3.4946441650390623e-05,
      "model_forward_time": 3.134211778640747,
      "step": 9161
    },
    {
      "epoch": 3.4946441650390623e-05,
      "step": 9161,
      "training_step_time": 10.028898477554321
    },
    {
      "epoch": 3.495025634765625e-05,
      "model_forward_time": 3.149984121322632,
      "step": 9162
    },
    {
      "epoch": 3.495025634765625e-05,
      "step": 9162,
      "training_step_time": 9.921766996383667
    },
    {
      "epoch": 3.495025634765625e-05,
      "model_forward_time": 3.1348767280578613,
      "step": 9162
    },
    {
      "epoch": 3.495025634765625e-05,
      "step": 9162,
      "training_step_time": 9.91441011428833
    },
    {
      "epoch": 3.495025634765625e-05,
      "model_forward_time": 3.1363513469696045,
      "step": 9162
    },
    {
      "dynamics_loss_avg": 0.055903203785419464,
      "epoch": 3.495025634765625e-05,
      "step": 9162
    },
    {
      "action_loss_avg": 0.020789298973977566,
      "epoch": 3.495025634765625e-05,
      "step": 9162
    },
    {
      "epoch": 3.495025634765625e-05,
      "step": 9162,
      "training_step_time": 10.181363821029663
    },
    {
      "epoch": 3.495025634765625e-05,
      "model_forward_time": 3.1397314071655273,
      "step": 9162
    },
    {
      "epoch": 3.495025634765625e-05,
      "step": 9162,
      "training_step_time": 10.321185827255249
    },
    {
      "epoch": 3.4954071044921876e-05,
      "model_forward_time": 3.1616053581237793,
      "step": 9163
    },
    {
      "epoch": 3.4954071044921876e-05,
      "step": 9163,
      "training_step_time": 9.956843137741089
    },
    {
      "epoch": 3.4954071044921876e-05,
      "model_forward_time": 3.139057159423828,
      "step": 9163
    },
    {
      "epoch": 3.4954071044921876e-05,
      "step": 9163,
      "training_step_time": 9.91961669921875
    },
    {
      "epoch": 3.4954071044921876e-05,
      "model_forward_time": 3.1321499347686768,
      "step": 9163
    },
    {
      "epoch": 3.4954071044921876e-05,
      "step": 9163,
      "training_step_time": 9.91199254989624
    },
    {
      "epoch": 3.4954071044921876e-05,
      "model_forward_time": 1.7485792636871338,
      "step": 9163
    },
    {
      "epoch": 3.4954071044921876e-05,
      "step": 9163,
      "training_step_time": 10.319896459579468
    },
    {
      "epoch": 3.49578857421875e-05,
      "model_forward_time": 3.4405593872070312,
      "step": 9164
    },
    {
      "epoch": 3.49578857421875e-05,
      "step": 9164,
      "training_step_time": 10.223180532455444
    },
    {
      "epoch": 3.49578857421875e-05,
      "model_forward_time": 2.4372496604919434,
      "step": 9164
    },
    {
      "epoch": 3.49578857421875e-05,
      "step": 9164,
      "training_step_time": 7.47283935546875
    },
    {
      "epoch": 3.49578857421875e-05,
      "model_forward_time": 2.4322991371154785,
      "step": 9164
    },
    {
      "epoch": 3.49578857421875e-05,
      "step": 9164,
      "training_step_time": 7.470046758651733
    },
    {
      "epoch": 3.49578857421875e-05,
      "model_forward_time": 3.1352880001068115,
      "step": 9164
    },
    {
      "epoch": 3.49578857421875e-05,
      "step": 9164,
      "training_step_time": 10.005149364471436
    },
    {
      "epoch": 3.496170043945312e-05,
      "model_forward_time": 3.1546406745910645,
      "step": 9165
    },
    {
      "dynamics_loss_avg": 0.04989791754633188,
      "epoch": 3.496170043945312e-05,
      "step": 9165
    },
    {
      "action_loss_avg": 0.016518139070831238,
      "epoch": 3.496170043945312e-05,
      "step": 9165
    },
    {
      "epoch": 3.496170043945312e-05,
      "step": 9165,
      "training_step_time": 9.936376571655273
    },
    {
      "epoch": 3.496170043945312e-05,
      "model_forward_time": 2.758758783340454,
      "step": 9165
    },
    {
      "epoch": 3.496170043945312e-05,
      "step": 9165,
      "training_step_time": 7.787720203399658
    },
    {
      "epoch": 3.496170043945312e-05,
      "model_forward_time": 3.139134407043457,
      "step": 9165
    },
    {
      "epoch": 3.496170043945312e-05,
      "step": 9165,
      "training_step_time": 9.912655115127563
    },
    {
      "epoch": 3.496170043945312e-05,
      "model_forward_time": 3.141359329223633,
      "step": 9165
    },
    {
      "epoch": 3.496170043945312e-05,
      "step": 9165,
      "training_step_time": 10.291449308395386
    },
    {
      "epoch": 3.496551513671875e-05,
      "model_forward_time": 3.1582677364349365,
      "step": 9166
    },
    {
      "epoch": 3.496551513671875e-05,
      "step": 9166,
      "training_step_time": 9.942957878112793
    },
    {
      "epoch": 3.496551513671875e-05,
      "model_forward_time": 3.140895128250122,
      "step": 9166
    },
    {
      "epoch": 3.496551513671875e-05,
      "step": 9166,
      "training_step_time": 9.918278932571411
    },
    {
      "epoch": 3.496551513671875e-05,
      "model_forward_time": 3.3965418338775635,
      "step": 9166
    },
    {
      "epoch": 3.496551513671875e-05,
      "step": 9166,
      "training_step_time": 10.183830738067627
    },
    {
      "epoch": 3.496551513671875e-05,
      "model_forward_time": 3.1512396335601807,
      "step": 9166
    },
    {
      "epoch": 3.496551513671875e-05,
      "step": 9166,
      "training_step_time": 10.31675386428833
    },
    {
      "epoch": 3.4969329833984374e-05,
      "model_forward_time": 3.147210121154785,
      "step": 9167
    },
    {
      "epoch": 3.4969329833984374e-05,
      "step": 9167,
      "training_step_time": 9.932580709457397
    },
    {
      "epoch": 3.4969329833984374e-05,
      "model_forward_time": 3.1375279426574707,
      "step": 9167
    },
    {
      "epoch": 3.4969329833984374e-05,
      "step": 9167,
      "training_step_time": 9.92046856880188
    },
    {
      "epoch": 3.4969329833984374e-05,
      "model_forward_time": 3.1407432556152344,
      "step": 9167
    },
    {
      "dynamics_loss_avg": 0.052821809984743597,
      "epoch": 3.4969329833984374e-05,
      "step": 9167
    },
    {
      "action_loss_avg": 0.016083013825118543,
      "epoch": 3.4969329833984374e-05,
      "step": 9167
    },
    {
      "epoch": 3.4969329833984374e-05,
      "step": 9167,
      "training_step_time": 9.921379804611206
    },
    {
      "epoch": 3.4969329833984374e-05,
      "model_forward_time": 3.4504573345184326,
      "step": 9167
    },
    {
      "epoch": 3.4969329833984374e-05,
      "step": 9167,
      "training_step_time": 10.320967435836792
    },
    {
      "epoch": 3.497314453125e-05,
      "model_forward_time": 3.1502444744110107,
      "step": 9168
    },
    {
      "epoch": 3.497314453125e-05,
      "step": 9168,
      "training_step_time": 9.93363642692566
    },
    {
      "epoch": 3.497314453125e-05,
      "model_forward_time": 3.136115312576294,
      "step": 9168
    },
    {
      "epoch": 3.497314453125e-05,
      "step": 9168,
      "training_step_time": 9.913106918334961
    },
    {
      "epoch": 3.497314453125e-05,
      "model_forward_time": 3.1404452323913574,
      "step": 9168
    },
    {
      "epoch": 3.497314453125e-05,
      "step": 9168,
      "training_step_time": 9.917679071426392
    },
    {
      "epoch": 3.497314453125e-05,
      "model_forward_time": 3.1329476833343506,
      "step": 9168
    },
    {
      "epoch": 3.497314453125e-05,
      "step": 9168,
      "training_step_time": 10.2725191116333
    },
    {
      "epoch": 3.497695922851563e-05,
      "model_forward_time": 3.4152135848999023,
      "step": 9169
    },
    {
      "epoch": 3.497695922851563e-05,
      "step": 9169,
      "training_step_time": 10.196691751480103
    },
    {
      "epoch": 3.497695922851563e-05,
      "model_forward_time": 3.1377458572387695,
      "step": 9169
    },
    {
      "epoch": 3.497695922851563e-05,
      "step": 9169,
      "training_step_time": 9.914734125137329
    },
    {
      "epoch": 3.497695922851563e-05,
      "model_forward_time": 3.140010118484497,
      "step": 9169
    },
    {
      "epoch": 3.497695922851563e-05,
      "step": 9169,
      "training_step_time": 9.91708493232727
    },
    {
      "epoch": 3.497695922851563e-05,
      "model_forward_time": 3.1371243000030518,
      "step": 9169
    },
    {
      "epoch": 3.497695922851563e-05,
      "step": 9169,
      "training_step_time": 10.251492738723755
    },
    {
      "epoch": 3.498077392578125e-05,
      "grad_norm": 0.5114365220069885,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 9170
    },
    {
      "epoch": 3.498077392578125e-05,
      "model_forward_time": 2.4468634128570557,
      "step": 9170
    },
    {
      "dynamics_loss_avg": 0.044800692796707155,
      "epoch": 3.498077392578125e-05,
      "step": 9170
    },
    {
      "action_loss_avg": 0.015196334477514029,
      "epoch": 3.498077392578125e-05,
      "step": 9170
    },
    {
      "epoch": 3.498077392578125e-05,
      "step": 9170,
      "training_step_time": 7.485084295272827
    },
    {
      "epoch": 3.498077392578125e-05,
      "model_forward_time": 2.750307321548462,
      "step": 9170
    },
    {
      "epoch": 3.498077392578125e-05,
      "step": 9170,
      "training_step_time": 7.780743837356567
    },
    {
      "epoch": 3.498077392578125e-05,
      "model_forward_time": 3.1405255794525146,
      "step": 9170
    },
    {
      "epoch": 3.498077392578125e-05,
      "step": 9170,
      "training_step_time": 9.916692972183228
    },
    {
      "epoch": 3.498077392578125e-05,
      "model_forward_time": 3.143644094467163,
      "step": 9170
    },
    {
      "epoch": 3.498077392578125e-05,
      "step": 9170,
      "training_step_time": 10.278831720352173
    },
    {
      "epoch": 3.498458862304687e-05,
      "model_forward_time": 3.1452674865722656,
      "step": 9171
    },
    {
      "epoch": 3.498458862304687e-05,
      "step": 9171,
      "training_step_time": 9.918915748596191
    },
    {
      "epoch": 3.498458862304687e-05,
      "model_forward_time": 3.142404317855835,
      "step": 9171
    },
    {
      "epoch": 3.498458862304687e-05,
      "step": 9171,
      "training_step_time": 9.914286613464355
    },
    {
      "epoch": 3.498458862304687e-05,
      "model_forward_time": 3.4137191772460938,
      "step": 9171
    },
    {
      "epoch": 3.498458862304687e-05,
      "step": 9171,
      "training_step_time": 10.18516731262207
    },
    {
      "epoch": 3.498458862304687e-05,
      "model_forward_time": 3.140181303024292,
      "step": 9171
    },
    {
      "epoch": 3.498458862304687e-05,
      "step": 9171,
      "training_step_time": 10.257999420166016
    },
    {
      "epoch": 3.49884033203125e-05,
      "model_forward_time": 3.16532301902771,
      "step": 9172
    },
    {
      "epoch": 3.49884033203125e-05,
      "step": 9172,
      "training_step_time": 9.941673278808594
    },
    {
      "epoch": 3.49884033203125e-05,
      "model_forward_time": 3.1450393199920654,
      "step": 9172
    },
    {
      "epoch": 3.49884033203125e-05,
      "step": 9172,
      "training_step_time": 9.921669006347656
    },
    {
      "epoch": 3.49884033203125e-05,
      "model_forward_time": 2.440380811691284,
      "step": 9172
    },
    {
      "dynamics_loss_avg": 0.05613689962774515,
      "epoch": 3.49884033203125e-05,
      "step": 9172
    },
    {
      "action_loss_avg": 0.026484437054023147,
      "epoch": 3.49884033203125e-05,
      "step": 9172
    },
    {
      "epoch": 3.49884033203125e-05,
      "step": 9172,
      "training_step_time": 7.476308345794678
    },
    {
      "epoch": 3.49884033203125e-05,
      "model_forward_time": 3.4213478565216064,
      "step": 9172
    },
    {
      "epoch": 3.49884033203125e-05,
      "step": 9172,
      "training_step_time": 10.351588249206543
    },
    {
      "epoch": 3.4992218017578125e-05,
      "model_forward_time": 3.1462318897247314,
      "step": 9173
    },
    {
      "epoch": 3.4992218017578125e-05,
      "step": 9173,
      "training_step_time": 9.92162561416626
    },
    {
      "epoch": 3.4992218017578125e-05,
      "model_forward_time": 3.1456665992736816,
      "step": 9173
    },
    {
      "epoch": 3.4992218017578125e-05,
      "step": 9173,
      "training_step_time": 9.924415588378906
    },
    {
      "epoch": 3.4992218017578125e-05,
      "model_forward_time": 3.1416499614715576,
      "step": 9173
    },
    {
      "epoch": 3.4992218017578125e-05,
      "step": 9173,
      "training_step_time": 9.923404932022095
    },
    {
      "epoch": 3.4992218017578125e-05,
      "model_forward_time": 3.14532208442688,
      "step": 9173
    },
    {
      "epoch": 3.4992218017578125e-05,
      "step": 9173,
      "training_step_time": 10.32823395729065
    },
    {
      "epoch": 3.499603271484375e-05,
      "model_forward_time": 3.435915231704712,
      "step": 9174
    },
    {
      "epoch": 3.499603271484375e-05,
      "step": 9174,
      "training_step_time": 10.211685419082642
    },
    {
      "epoch": 3.499603271484375e-05,
      "model_forward_time": 3.142829179763794,
      "step": 9174
    },
    {
      "epoch": 3.499603271484375e-05,
      "step": 9174,
      "training_step_time": 9.914491415023804
    },
    {
      "epoch": 3.499603271484375e-05,
      "model_forward_time": 3.1387827396392822,
      "step": 9174
    },
    {
      "epoch": 3.499603271484375e-05,
      "step": 9174,
      "training_step_time": 9.908353567123413
    },
    {
      "epoch": 3.499603271484375e-05,
      "model_forward_time": 3.1408281326293945,
      "step": 9174
    },
    {
      "epoch": 3.499603271484375e-05,
      "step": 9174,
      "training_step_time": 10.309537172317505
    },
    {
      "epoch": 3.499984741210938e-05,
      "model_forward_time": 3.440761089324951,
      "step": 9175
    },
    {
      "dynamics_loss_avg": 0.05942098516970873,
      "epoch": 3.499984741210938e-05,
      "step": 9175
    },
    {
      "action_loss_avg": 0.025408858247101308,
      "epoch": 3.499984741210938e-05,
      "step": 9175
    },
    {
      "epoch": 3.499984741210938e-05,
      "step": 9175,
      "training_step_time": 10.21221137046814
    },
    {
      "epoch": 3.499984741210938e-05,
      "model_forward_time": 3.13991379737854,
      "step": 9175
    },
    {
      "epoch": 3.499984741210938e-05,
      "step": 9175,
      "training_step_time": 9.90755295753479
    },
    {
      "epoch": 3.499984741210938e-05,
      "model_forward_time": 3.137908697128296,
      "step": 9175
    },
    {
      "epoch": 3.499984741210938e-05,
      "step": 9175,
      "training_step_time": 9.906018495559692
    },
    {
      "epoch": 3.499984741210938e-05,
      "model_forward_time": 2.436466932296753,
      "step": 9175
    },
    {
      "epoch": 3.499984741210938e-05,
      "step": 9175,
      "training_step_time": 10.33189606666565
    },
    {
      "epoch": 3.5003662109375e-05,
      "model_forward_time": 3.1464555263519287,
      "step": 9176
    },
    {
      "epoch": 3.5003662109375e-05,
      "step": 9176,
      "training_step_time": 9.9179208278656
    },
    {
      "epoch": 3.5003662109375e-05,
      "model_forward_time": 3.135084390640259,
      "step": 9176
    },
    {
      "epoch": 3.5003662109375e-05,
      "step": 9176,
      "training_step_time": 9.908732175827026
    },
    {
      "epoch": 3.5003662109375e-05,
      "model_forward_time": 3.3869752883911133,
      "step": 9176
    },
    {
      "epoch": 3.5003662109375e-05,
      "step": 9176,
      "training_step_time": 10.153572082519531
    },
    {
      "epoch": 3.5003662109375e-05,
      "model_forward_time": 3.1375105381011963,
      "step": 9176
    },
    {
      "epoch": 3.5003662109375e-05,
      "step": 9176,
      "training_step_time": 10.275479078292847
    },
    {
      "epoch": 3.5007476806640623e-05,
      "model_forward_time": 3.1489744186401367,
      "step": 9177
    },
    {
      "epoch": 3.5007476806640623e-05,
      "step": 9177,
      "training_step_time": 9.919034481048584
    },
    {
      "epoch": 3.5007476806640623e-05,
      "model_forward_time": 2.426969528198242,
      "step": 9177
    },
    {
      "epoch": 3.5007476806640623e-05,
      "step": 9177,
      "training_step_time": 7.447908878326416
    },
    {
      "epoch": 3.5007476806640623e-05,
      "model_forward_time": 3.131136655807495,
      "step": 9177
    },
    {
      "dynamics_loss_avg": 0.05186720266938209,
      "epoch": 3.5007476806640623e-05,
      "step": 9177
    },
    {
      "action_loss_avg": 0.017739767860621213,
      "epoch": 3.5007476806640623e-05,
      "step": 9177
    },
    {
      "epoch": 3.5007476806640623e-05,
      "step": 9177,
      "training_step_time": 9.894641876220703
    },
    {
      "epoch": 3.5007476806640623e-05,
      "model_forward_time": 3.402926445007324,
      "step": 9177
    },
    {
      "epoch": 3.5007476806640623e-05,
      "step": 9177,
      "training_step_time": 10.26344084739685
    },
    {
      "epoch": 3.501129150390625e-05,
      "model_forward_time": 3.1797616481781006,
      "step": 9178
    },
    {
      "epoch": 3.501129150390625e-05,
      "step": 9178,
      "training_step_time": 9.949742555618286
    },
    {
      "epoch": 3.501129150390625e-05,
      "model_forward_time": 3.1425201892852783,
      "step": 9178
    },
    {
      "epoch": 3.501129150390625e-05,
      "step": 9178,
      "training_step_time": 9.90920090675354
    },
    {
      "epoch": 3.501129150390625e-05,
      "model_forward_time": 3.1489861011505127,
      "step": 9178
    },
    {
      "epoch": 3.501129150390625e-05,
      "step": 9178,
      "training_step_time": 9.917745351791382
    },
    {
      "epoch": 3.501129150390625e-05,
      "model_forward_time": 3.48726487159729,
      "step": 9178
    },
    {
      "epoch": 3.501129150390625e-05,
      "step": 9178,
      "training_step_time": 10.357367753982544
    },
    {
      "epoch": 3.5015106201171876e-05,
      "model_forward_time": 3.1644182205200195,
      "step": 9179
    },
    {
      "epoch": 3.5015106201171876e-05,
      "step": 9179,
      "training_step_time": 9.939367532730103
    },
    {
      "epoch": 3.5015106201171876e-05,
      "model_forward_time": 3.147958278656006,
      "step": 9179
    },
    {
      "epoch": 3.5015106201171876e-05,
      "step": 9179,
      "training_step_time": 9.925633430480957
    },
    {
      "epoch": 3.5015106201171876e-05,
      "model_forward_time": 3.1530909538269043,
      "step": 9179
    },
    {
      "epoch": 3.5015106201171876e-05,
      "step": 9179,
      "training_step_time": 9.927644968032837
    },
    {
      "epoch": 3.5015106201171876e-05,
      "model_forward_time": 3.1472976207733154,
      "step": 9179
    },
    {
      "epoch": 3.5015106201171876e-05,
      "step": 9179,
      "training_step_time": 10.278406381607056
    },
    {
      "epoch": 3.50189208984375e-05,
      "grad_norm": 0.33742496371269226,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 9180
    },
    {
      "epoch": 3.50189208984375e-05,
      "model_forward_time": 3.1712870597839355,
      "step": 9180
    },
    {
      "dynamics_loss_avg": 0.07573963478207588,
      "epoch": 3.50189208984375e-05,
      "step": 9180
    },
    {
      "action_loss_avg": 0.029131639562547208,
      "epoch": 3.50189208984375e-05,
      "step": 9180
    },
    {
      "epoch": 3.50189208984375e-05,
      "step": 9180,
      "training_step_time": 10.264478921890259
    },
    {
      "epoch": 3.50189208984375e-05,
      "model_forward_time": 3.1372714042663574,
      "step": 9180
    },
    {
      "epoch": 3.50189208984375e-05,
      "step": 9180,
      "training_step_time": 9.911155462265015
    },
    {
      "epoch": 3.50189208984375e-05,
      "model_forward_time": 3.1464226245880127,
      "step": 9180
    },
    {
      "epoch": 3.50189208984375e-05,
      "step": 9180,
      "training_step_time": 9.923690795898438
    },
    {
      "epoch": 3.50189208984375e-05,
      "model_forward_time": 3.1437253952026367,
      "step": 9180
    },
    {
      "epoch": 3.50189208984375e-05,
      "step": 9180,
      "training_step_time": 10.272077560424805
    },
    {
      "epoch": 3.502273559570312e-05,
      "model_forward_time": 3.1447253227233887,
      "step": 9181
    },
    {
      "epoch": 3.502273559570312e-05,
      "step": 9181,
      "training_step_time": 9.923051834106445
    },
    {
      "epoch": 3.502273559570312e-05,
      "model_forward_time": 3.4040441513061523,
      "step": 9181
    },
    {
      "epoch": 3.502273559570312e-05,
      "step": 9181,
      "training_step_time": 10.171437978744507
    },
    {
      "epoch": 3.502273559570312e-05,
      "model_forward_time": 3.138339042663574,
      "step": 9181
    },
    {
      "epoch": 3.502273559570312e-05,
      "step": 9181,
      "training_step_time": 9.909531831741333
    },
    {
      "epoch": 3.502273559570312e-05,
      "model_forward_time": 3.1333255767822266,
      "step": 9181
    },
    {
      "epoch": 3.502273559570312e-05,
      "step": 9181,
      "training_step_time": 10.018017053604126
    },
    {
      "epoch": 3.502655029296875e-05,
      "model_forward_time": 3.1501598358154297,
      "step": 9182
    },
    {
      "epoch": 3.502655029296875e-05,
      "step": 9182,
      "training_step_time": 9.923290252685547
    },
    {
      "epoch": 3.502655029296875e-05,
      "model_forward_time": 3.1379590034484863,
      "step": 9182
    },
    {
      "epoch": 3.502655029296875e-05,
      "step": 9182,
      "training_step_time": 9.903677225112915
    },
    {
      "epoch": 3.502655029296875e-05,
      "model_forward_time": 2.4303436279296875,
      "step": 9182
    },
    {
      "dynamics_loss_avg": 0.05773348081856966,
      "epoch": 3.502655029296875e-05,
      "step": 9182
    },
    {
      "action_loss_avg": 0.01989605212584138,
      "epoch": 3.502655029296875e-05,
      "step": 9182
    },
    {
      "epoch": 3.502655029296875e-05,
      "step": 9182,
      "training_step_time": 7.753175735473633
    },
    {
      "epoch": 3.502655029296875e-05,
      "model_forward_time": 3.1311776638031006,
      "step": 9182
    },
    {
      "epoch": 3.502655029296875e-05,
      "step": 9182,
      "training_step_time": 10.014307737350464
    },
    {
      "epoch": 3.5030364990234374e-05,
      "model_forward_time": 3.152808427810669,
      "step": 9183
    },
    {
      "epoch": 3.5030364990234374e-05,
      "step": 9183,
      "training_step_time": 9.929120540618896
    },
    {
      "epoch": 3.5030364990234374e-05,
      "model_forward_time": 3.1354775428771973,
      "step": 9183
    },
    {
      "epoch": 3.5030364990234374e-05,
      "step": 9183,
      "training_step_time": 9.911768436431885
    },
    {
      "epoch": 3.5030364990234374e-05,
      "model_forward_time": 3.1392199993133545,
      "step": 9183
    },
    {
      "epoch": 3.5030364990234374e-05,
      "step": 9183,
      "training_step_time": 9.910059690475464
    },
    {
      "epoch": 3.5030364990234374e-05,
      "model_forward_time": 3.4079041481018066,
      "step": 9183
    },
    {
      "epoch": 3.5030364990234374e-05,
      "step": 9183,
      "training_step_time": 10.318682670593262
    },
    {
      "epoch": 3.50341796875e-05,
      "model_forward_time": 3.15664005279541,
      "step": 9184
    },
    {
      "epoch": 3.50341796875e-05,
      "step": 9184,
      "training_step_time": 9.932266473770142
    },
    {
      "epoch": 3.50341796875e-05,
      "model_forward_time": 3.1368606090545654,
      "step": 9184
    },
    {
      "epoch": 3.50341796875e-05,
      "step": 9184,
      "training_step_time": 9.905367851257324
    },
    {
      "epoch": 3.50341796875e-05,
      "model_forward_time": 3.142705202102661,
      "step": 9184
    },
    {
      "epoch": 3.50341796875e-05,
      "step": 9184,
      "training_step_time": 9.919174194335938
    },
    {
      "epoch": 3.50341796875e-05,
      "model_forward_time": 2.4396700859069824,
      "step": 9184
    },
    {
      "epoch": 3.50341796875e-05,
      "step": 9184,
      "training_step_time": 10.28457498550415
    },
    {
      "epoch": 3.503799438476563e-05,
      "model_forward_time": 3.1472084522247314,
      "step": 9185
    },
    {
      "dynamics_loss_avg": 0.07399695925414562,
      "epoch": 3.503799438476563e-05,
      "step": 9185
    },
    {
      "action_loss_avg": 0.02278004717081785,
      "epoch": 3.503799438476563e-05,
      "step": 9185
    },
    {
      "epoch": 3.503799438476563e-05,
      "step": 9185,
      "training_step_time": 10.197385311126709
    },
    {
      "epoch": 3.503799438476563e-05,
      "model_forward_time": 3.142263650894165,
      "step": 9185
    },
    {
      "epoch": 3.503799438476563e-05,
      "step": 9185,
      "training_step_time": 9.915762662887573
    },
    {
      "epoch": 3.503799438476563e-05,
      "model_forward_time": 3.1444387435913086,
      "step": 9185
    },
    {
      "epoch": 3.503799438476563e-05,
      "step": 9185,
      "training_step_time": 9.925584554672241
    },
    {
      "epoch": 3.503799438476563e-05,
      "model_forward_time": 3.1453592777252197,
      "step": 9185
    },
    {
      "epoch": 3.503799438476563e-05,
      "step": 9185,
      "training_step_time": 10.306861400604248
    },
    {
      "epoch": 3.504180908203125e-05,
      "model_forward_time": 3.160830497741699,
      "step": 9186
    },
    {
      "epoch": 3.504180908203125e-05,
      "step": 9186,
      "training_step_time": 9.938080072402954
    },
    {
      "epoch": 3.504180908203125e-05,
      "model_forward_time": 2.4365553855895996,
      "step": 9186
    },
    {
      "epoch": 3.504180908203125e-05,
      "step": 9186,
      "training_step_time": 7.765407562255859
    },
    {
      "epoch": 3.504180908203125e-05,
      "model_forward_time": 3.131110906600952,
      "step": 9186
    },
    {
      "epoch": 3.504180908203125e-05,
      "step": 9186,
      "training_step_time": 9.90341067314148
    },
    {
      "epoch": 3.504180908203125e-05,
      "model_forward_time": 3.1368298530578613,
      "step": 9186
    },
    {
      "epoch": 3.504180908203125e-05,
      "step": 9186,
      "training_step_time": 10.025663375854492
    },
    {
      "epoch": 3.504562377929687e-05,
      "model_forward_time": 3.147968053817749,
      "step": 9187
    },
    {
      "epoch": 3.504562377929687e-05,
      "step": 9187,
      "training_step_time": 9.92149806022644
    },
    {
      "epoch": 3.504562377929687e-05,
      "model_forward_time": 2.437156915664673,
      "step": 9187
    },
    {
      "epoch": 3.504562377929687e-05,
      "step": 9187,
      "training_step_time": 7.465214014053345
    },
    {
      "epoch": 3.504562377929687e-05,
      "model_forward_time": 3.13545298576355,
      "step": 9187
    },
    {
      "dynamics_loss_avg": 0.05101587269455195,
      "epoch": 3.504562377929687e-05,
      "step": 9187
    },
    {
      "action_loss_avg": 0.019392646430060268,
      "epoch": 3.504562377929687e-05,
      "step": 9187
    },
    {
      "epoch": 3.504562377929687e-05,
      "step": 9187,
      "training_step_time": 10.187071084976196
    },
    {
      "epoch": 3.504562377929687e-05,
      "model_forward_time": 3.1350607872009277,
      "step": 9187
    },
    {
      "epoch": 3.504562377929687e-05,
      "step": 9187,
      "training_step_time": 10.011890172958374
    },
    {
      "epoch": 3.50494384765625e-05,
      "model_forward_time": 2.4480576515197754,
      "step": 9188
    },
    {
      "epoch": 3.50494384765625e-05,
      "step": 9188,
      "training_step_time": 7.474423170089722
    },
    {
      "epoch": 3.50494384765625e-05,
      "model_forward_time": 3.12969708442688,
      "step": 9188
    },
    {
      "epoch": 3.50494384765625e-05,
      "step": 9188,
      "training_step_time": 9.893803358078003
    },
    {
      "epoch": 3.50494384765625e-05,
      "model_forward_time": 3.132298469543457,
      "step": 9188
    },
    {
      "epoch": 3.50494384765625e-05,
      "step": 9188,
      "training_step_time": 9.898344278335571
    },
    {
      "epoch": 3.50494384765625e-05,
      "model_forward_time": 2.4290618896484375,
      "step": 9188
    },
    {
      "epoch": 3.50494384765625e-05,
      "step": 9188,
      "training_step_time": 10.293562650680542
    },
    {
      "epoch": 3.5053253173828125e-05,
      "model_forward_time": 3.4375698566436768,
      "step": 9189
    },
    {
      "epoch": 3.5053253173828125e-05,
      "step": 9189,
      "training_step_time": 10.207924127578735
    },
    {
      "epoch": 3.5053253173828125e-05,
      "model_forward_time": 3.1347477436065674,
      "step": 9189
    },
    {
      "epoch": 3.5053253173828125e-05,
      "step": 9189,
      "training_step_time": 9.902243852615356
    },
    {
      "epoch": 3.5053253173828125e-05,
      "model_forward_time": 3.139876127243042,
      "step": 9189
    },
    {
      "epoch": 3.5053253173828125e-05,
      "step": 9189,
      "training_step_time": 9.914526224136353
    },
    {
      "epoch": 3.5053253173828125e-05,
      "model_forward_time": 3.1346516609191895,
      "step": 9189
    },
    {
      "epoch": 3.5053253173828125e-05,
      "step": 9189,
      "training_step_time": 10.26928997039795
    },
    {
      "epoch": 3.505706787109375e-05,
      "grad_norm": 0.3665870726108551,
      "learning_rate": 1e-05,
      "loss": 0.0795,
      "step": 9190
    },
    {
      "epoch": 3.505706787109375e-05,
      "model_forward_time": 3.4244606494903564,
      "step": 9190
    },
    {
      "dynamics_loss_avg": 0.04811970312148332,
      "epoch": 3.505706787109375e-05,
      "step": 9190
    },
    {
      "action_loss_avg": 0.015993493748828767,
      "epoch": 3.505706787109375e-05,
      "step": 9190
    },
    {
      "epoch": 3.505706787109375e-05,
      "step": 9190,
      "training_step_time": 10.200413942337036
    },
    {
      "epoch": 3.505706787109375e-05,
      "model_forward_time": 3.136117935180664,
      "step": 9190
    },
    {
      "epoch": 3.505706787109375e-05,
      "step": 9190,
      "training_step_time": 9.908012866973877
    },
    {
      "epoch": 3.505706787109375e-05,
      "model_forward_time": 3.137517213821411,
      "step": 9190
    },
    {
      "epoch": 3.505706787109375e-05,
      "step": 9190,
      "training_step_time": 9.909157991409302
    },
    {
      "epoch": 3.505706787109375e-05,
      "model_forward_time": 3.1338319778442383,
      "step": 9190
    },
    {
      "epoch": 3.505706787109375e-05,
      "step": 9190,
      "training_step_time": 10.251183032989502
    },
    {
      "epoch": 3.506088256835938e-05,
      "model_forward_time": 3.146160125732422,
      "step": 9191
    },
    {
      "epoch": 3.506088256835938e-05,
      "step": 9191,
      "training_step_time": 9.91922664642334
    },
    {
      "epoch": 3.506088256835938e-05,
      "model_forward_time": 3.148735284805298,
      "step": 9191
    },
    {
      "epoch": 3.506088256835938e-05,
      "step": 9191,
      "training_step_time": 10.195961713790894
    },
    {
      "epoch": 3.506088256835938e-05,
      "model_forward_time": 3.1449716091156006,
      "step": 9191
    },
    {
      "epoch": 3.506088256835938e-05,
      "step": 9191,
      "training_step_time": 9.917046546936035
    },
    {
      "epoch": 3.506088256835938e-05,
      "model_forward_time": 3.1362733840942383,
      "step": 9191
    },
    {
      "epoch": 3.506088256835938e-05,
      "step": 9191,
      "training_step_time": 10.247024536132812
    },
    {
      "epoch": 3.5064697265625e-05,
      "model_forward_time": 3.1399505138397217,
      "step": 9192
    },
    {
      "epoch": 3.5064697265625e-05,
      "step": 9192,
      "training_step_time": 9.914541244506836
    },
    {
      "epoch": 3.5064697265625e-05,
      "model_forward_time": 3.1348607540130615,
      "step": 9192
    },
    {
      "epoch": 3.5064697265625e-05,
      "step": 9192,
      "training_step_time": 9.904821634292603
    },
    {
      "epoch": 3.5064697265625e-05,
      "model_forward_time": 3.445491075515747,
      "step": 9192
    },
    {
      "dynamics_loss_avg": 0.054966994747519496,
      "epoch": 3.5064697265625e-05,
      "step": 9192
    },
    {
      "action_loss_avg": 0.0186893739271909,
      "epoch": 3.5064697265625e-05,
      "step": 9192
    },
    {
      "epoch": 3.5064697265625e-05,
      "step": 9192,
      "training_step_time": 10.219861268997192
    },
    {
      "epoch": 3.5064697265625e-05,
      "model_forward_time": 3.1326587200164795,
      "step": 9192
    },
    {
      "epoch": 3.5064697265625e-05,
      "step": 9192,
      "training_step_time": 10.016530513763428
    },
    {
      "epoch": 3.5068511962890624e-05,
      "model_forward_time": 3.1402053833007812,
      "step": 9193
    },
    {
      "epoch": 3.5068511962890624e-05,
      "step": 9193,
      "training_step_time": 9.910653591156006
    },
    {
      "epoch": 3.5068511962890624e-05,
      "model_forward_time": 3.1338794231414795,
      "step": 9193
    },
    {
      "epoch": 3.5068511962890624e-05,
      "step": 9193,
      "training_step_time": 9.901226282119751
    },
    {
      "epoch": 3.5068511962890624e-05,
      "model_forward_time": 3.1330299377441406,
      "step": 9193
    },
    {
      "epoch": 3.5068511962890624e-05,
      "step": 9193,
      "training_step_time": 9.901388883590698
    },
    {
      "epoch": 3.5068511962890624e-05,
      "model_forward_time": 3.4061903953552246,
      "step": 9193
    },
    {
      "epoch": 3.5068511962890624e-05,
      "step": 9193,
      "training_step_time": 10.329236030578613
    },
    {
      "epoch": 3.507232666015625e-05,
      "model_forward_time": 3.1408896446228027,
      "step": 9194
    },
    {
      "epoch": 3.507232666015625e-05,
      "step": 9194,
      "training_step_time": 9.90814757347107
    },
    {
      "epoch": 3.507232666015625e-05,
      "model_forward_time": 3.1381540298461914,
      "step": 9194
    },
    {
      "epoch": 3.507232666015625e-05,
      "step": 9194,
      "training_step_time": 9.903380393981934
    },
    {
      "epoch": 3.507232666015625e-05,
      "model_forward_time": 3.1335434913635254,
      "step": 9194
    },
    {
      "epoch": 3.507232666015625e-05,
      "step": 9194,
      "training_step_time": 9.904676914215088
    },
    {
      "epoch": 3.507232666015625e-05,
      "model_forward_time": 3.1388871669769287,
      "step": 9194
    },
    {
      "epoch": 3.507232666015625e-05,
      "step": 9194,
      "training_step_time": 10.308583498001099
    },
    {
      "epoch": 3.5076141357421876e-05,
      "model_forward_time": 2.715585231781006,
      "step": 9195
    },
    {
      "dynamics_loss_avg": 0.05601199008524418,
      "epoch": 3.5076141357421876e-05,
      "step": 9195
    },
    {
      "action_loss_avg": 0.019646959006786348,
      "epoch": 3.5076141357421876e-05,
      "step": 9195
    },
    {
      "epoch": 3.5076141357421876e-05,
      "step": 9195,
      "training_step_time": 7.746983051300049
    },
    {
      "epoch": 3.5076141357421876e-05,
      "model_forward_time": 3.137054681777954,
      "step": 9195
    },
    {
      "epoch": 3.5076141357421876e-05,
      "step": 9195,
      "training_step_time": 9.908464670181274
    },
    {
      "epoch": 3.5076141357421876e-05,
      "model_forward_time": 3.1499319076538086,
      "step": 9195
    },
    {
      "epoch": 3.5076141357421876e-05,
      "step": 9195,
      "training_step_time": 9.922786235809326
    },
    {
      "epoch": 3.5076141357421876e-05,
      "model_forward_time": 3.156275749206543,
      "step": 9195
    },
    {
      "epoch": 3.5076141357421876e-05,
      "step": 9195,
      "training_step_time": 10.283735275268555
    },
    {
      "epoch": 3.50799560546875e-05,
      "model_forward_time": 3.1592085361480713,
      "step": 9196
    },
    {
      "epoch": 3.50799560546875e-05,
      "step": 9196,
      "training_step_time": 9.938639879226685
    },
    {
      "epoch": 3.50799560546875e-05,
      "model_forward_time": 3.146533250808716,
      "step": 9196
    },
    {
      "epoch": 3.50799560546875e-05,
      "step": 9196,
      "training_step_time": 10.18738603591919
    },
    {
      "epoch": 3.50799560546875e-05,
      "model_forward_time": 3.1429736614227295,
      "step": 9196
    },
    {
      "epoch": 3.50799560546875e-05,
      "step": 9196,
      "training_step_time": 9.912816286087036
    },
    {
      "epoch": 3.50799560546875e-05,
      "model_forward_time": 3.145751476287842,
      "step": 9196
    },
    {
      "epoch": 3.50799560546875e-05,
      "step": 9196,
      "training_step_time": 10.300376892089844
    },
    {
      "epoch": 3.508377075195312e-05,
      "model_forward_time": 3.1428112983703613,
      "step": 9197
    },
    {
      "epoch": 3.508377075195312e-05,
      "step": 9197,
      "training_step_time": 9.91850471496582
    },
    {
      "epoch": 3.508377075195312e-05,
      "model_forward_time": 3.130505323410034,
      "step": 9197
    },
    {
      "epoch": 3.508377075195312e-05,
      "step": 9197,
      "training_step_time": 9.912354946136475
    },
    {
      "epoch": 3.508377075195312e-05,
      "model_forward_time": 3.1383843421936035,
      "step": 9197
    },
    {
      "dynamics_loss_avg": 0.054163958132267,
      "epoch": 3.508377075195312e-05,
      "step": 9197
    },
    {
      "action_loss_avg": 0.018557240441441535,
      "epoch": 3.508377075195312e-05,
      "step": 9197
    },
    {
      "epoch": 3.508377075195312e-05,
      "step": 9197,
      "training_step_time": 10.193795204162598
    },
    {
      "epoch": 3.508377075195312e-05,
      "model_forward_time": 3.139796257019043,
      "step": 9197
    },
    {
      "epoch": 3.508377075195312e-05,
      "step": 9197,
      "training_step_time": 10.306382894515991
    },
    {
      "epoch": 3.508758544921875e-05,
      "model_forward_time": 2.442800998687744,
      "step": 9198
    },
    {
      "epoch": 3.508758544921875e-05,
      "step": 9198,
      "training_step_time": 7.469743251800537
    },
    {
      "epoch": 3.508758544921875e-05,
      "model_forward_time": 3.1320157051086426,
      "step": 9198
    },
    {
      "epoch": 3.508758544921875e-05,
      "step": 9198,
      "training_step_time": 9.895988464355469
    },
    {
      "epoch": 3.508758544921875e-05,
      "model_forward_time": 3.135700225830078,
      "step": 9198
    },
    {
      "epoch": 3.508758544921875e-05,
      "step": 9198,
      "training_step_time": 9.906689167022705
    },
    {
      "epoch": 3.508758544921875e-05,
      "model_forward_time": 3.1349034309387207,
      "step": 9198
    },
    {
      "epoch": 3.508758544921875e-05,
      "step": 9198,
      "training_step_time": 10.322507619857788
    },
    {
      "epoch": 3.5091400146484374e-05,
      "model_forward_time": 3.4564549922943115,
      "step": 9199
    },
    {
      "epoch": 3.5091400146484374e-05,
      "step": 9199,
      "training_step_time": 10.231320142745972
    },
    {
      "epoch": 3.5091400146484374e-05,
      "model_forward_time": 3.130542278289795,
      "step": 9199
    },
    {
      "epoch": 3.5091400146484374e-05,
      "step": 9199,
      "training_step_time": 9.903613805770874
    },
    {
      "epoch": 3.5091400146484374e-05,
      "model_forward_time": 3.139854669570923,
      "step": 9199
    },
    {
      "epoch": 3.5091400146484374e-05,
      "step": 9199,
      "training_step_time": 9.913583755493164
    },
    {
      "epoch": 3.5091400146484374e-05,
      "model_forward_time": 2.431354522705078,
      "step": 9199
    },
    {
      "epoch": 3.5091400146484374e-05,
      "step": 9199,
      "training_step_time": 10.018279314041138
    },
    {
      "epoch": 3.509521484375e-05,
      "grad_norm": 0.6323558688163757,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 9200
    },
    {
      "epoch": 3.509521484375e-05,
      "model_forward_time": 3.1639466285705566,
      "step": 9200
    },
    {
      "dynamics_loss_avg": 0.05837721265852451,
      "epoch": 3.509521484375e-05,
      "step": 9200
    },
    {
      "action_loss_avg": 0.022170756105333568,
      "epoch": 3.509521484375e-05,
      "step": 9200
    },
    {
      "epoch": 3.509521484375e-05,
      "step": 9200,
      "training_step_time": 10.21643853187561
    },
    {
      "epoch": 3.509521484375e-05,
      "model_forward_time": 3.1363890171051025,
      "step": 9200
    },
    {
      "epoch": 3.509521484375e-05,
      "step": 9200,
      "training_step_time": 9.908592939376831
    },
    {
      "epoch": 3.509521484375e-05,
      "model_forward_time": 3.1416594982147217,
      "step": 9200
    },
    {
      "epoch": 3.509521484375e-05,
      "step": 9200,
      "training_step_time": 9.913060188293457
    },
    {
      "epoch": 3.509521484375e-05,
      "model_forward_time": 3.141486644744873,
      "step": 9200
    },
    {
      "epoch": 3.509521484375e-05,
      "step": 9200,
      "training_step_time": 10.017313241958618
    },
    {
      "epoch": 3.509902954101563e-05,
      "model_forward_time": 3.152156352996826,
      "step": 9201
    },
    {
      "epoch": 3.509902954101563e-05,
      "step": 9201,
      "training_step_time": 9.927816152572632
    },
    {
      "epoch": 3.509902954101563e-05,
      "model_forward_time": 3.4294984340667725,
      "step": 9201
    },
    {
      "epoch": 3.509902954101563e-05,
      "step": 9201,
      "training_step_time": 10.197472333908081
    },
    {
      "epoch": 3.509902954101563e-05,
      "model_forward_time": 3.140420913696289,
      "step": 9201
    },
    {
      "epoch": 3.509902954101563e-05,
      "step": 9201,
      "training_step_time": 9.909234523773193
    },
    {
      "epoch": 3.509902954101563e-05,
      "model_forward_time": 3.1402652263641357,
      "step": 9201
    },
    {
      "epoch": 3.509902954101563e-05,
      "step": 9201,
      "training_step_time": 10.257126331329346
    },
    {
      "epoch": 3.510284423828125e-05,
      "model_forward_time": 3.158094882965088,
      "step": 9202
    },
    {
      "epoch": 3.510284423828125e-05,
      "step": 9202,
      "training_step_time": 9.930156469345093
    },
    {
      "epoch": 3.510284423828125e-05,
      "model_forward_time": 3.1397716999053955,
      "step": 9202
    },
    {
      "epoch": 3.510284423828125e-05,
      "step": 9202,
      "training_step_time": 9.909244298934937
    },
    {
      "epoch": 3.510284423828125e-05,
      "model_forward_time": 3.439110517501831,
      "step": 9202
    },
    {
      "dynamics_loss_avg": 0.057408223114907744,
      "epoch": 3.510284423828125e-05,
      "step": 9202
    },
    {
      "action_loss_avg": 0.021521993400529026,
      "epoch": 3.510284423828125e-05,
      "step": 9202
    },
    {
      "epoch": 3.510284423828125e-05,
      "step": 9202,
      "training_step_time": 10.209433794021606
    },
    {
      "epoch": 3.510284423828125e-05,
      "model_forward_time": 3.14302396774292,
      "step": 9202
    },
    {
      "epoch": 3.510284423828125e-05,
      "step": 9202,
      "training_step_time": 10.280686616897583
    },
    {
      "epoch": 3.510665893554687e-05,
      "model_forward_time": 2.449409246444702,
      "step": 9203
    },
    {
      "epoch": 3.510665893554687e-05,
      "step": 9203,
      "training_step_time": 7.4782233238220215
    },
    {
      "epoch": 3.510665893554687e-05,
      "model_forward_time": 3.143199920654297,
      "step": 9203
    },
    {
      "epoch": 3.510665893554687e-05,
      "step": 9203,
      "training_step_time": 9.910398244857788
    },
    {
      "epoch": 3.510665893554687e-05,
      "model_forward_time": 3.1369807720184326,
      "step": 9203
    },
    {
      "epoch": 3.510665893554687e-05,
      "step": 9203,
      "training_step_time": 9.904237031936646
    },
    {
      "epoch": 3.510665893554687e-05,
      "model_forward_time": 2.4347453117370605,
      "step": 9203
    },
    {
      "epoch": 3.510665893554687e-05,
      "step": 9203,
      "training_step_time": 10.279200077056885
    },
    {
      "epoch": 3.51104736328125e-05,
      "model_forward_time": 3.1439008712768555,
      "step": 9204
    },
    {
      "epoch": 3.51104736328125e-05,
      "step": 9204,
      "training_step_time": 9.91116213798523
    },
    {
      "epoch": 3.51104736328125e-05,
      "model_forward_time": 3.137707233428955,
      "step": 9204
    },
    {
      "epoch": 3.51104736328125e-05,
      "step": 9204,
      "training_step_time": 9.902536153793335
    },
    {
      "epoch": 3.51104736328125e-05,
      "model_forward_time": 3.132162094116211,
      "step": 9204
    },
    {
      "epoch": 3.51104736328125e-05,
      "step": 9204,
      "training_step_time": 9.902999639511108
    },
    {
      "epoch": 3.51104736328125e-05,
      "model_forward_time": 3.1422226428985596,
      "step": 9204
    },
    {
      "epoch": 3.51104736328125e-05,
      "step": 9204,
      "training_step_time": 10.301058530807495
    },
    {
      "epoch": 3.5114288330078125e-05,
      "model_forward_time": 2.4445087909698486,
      "step": 9205
    },
    {
      "dynamics_loss_avg": 0.04238102398812771,
      "epoch": 3.5114288330078125e-05,
      "step": 9205
    },
    {
      "action_loss_avg": 0.01282971492037177,
      "epoch": 3.5114288330078125e-05,
      "step": 9205
    },
    {
      "epoch": 3.5114288330078125e-05,
      "step": 9205,
      "training_step_time": 7.472480535507202
    },
    {
      "epoch": 3.5114288330078125e-05,
      "model_forward_time": 3.4024312496185303,
      "step": 9205
    },
    {
      "epoch": 3.5114288330078125e-05,
      "step": 9205,
      "training_step_time": 10.167521476745605
    },
    {
      "epoch": 3.5114288330078125e-05,
      "model_forward_time": 3.135296583175659,
      "step": 9205
    },
    {
      "epoch": 3.5114288330078125e-05,
      "step": 9205,
      "training_step_time": 9.899805784225464
    },
    {
      "epoch": 3.5114288330078125e-05,
      "model_forward_time": 3.1352431774139404,
      "step": 9205
    },
    {
      "epoch": 3.5114288330078125e-05,
      "step": 9205,
      "training_step_time": 10.002761840820312
    },
    {
      "epoch": 3.511810302734375e-05,
      "model_forward_time": 2.4452221393585205,
      "step": 9206
    },
    {
      "epoch": 3.511810302734375e-05,
      "step": 9206,
      "training_step_time": 7.4684154987335205
    },
    {
      "epoch": 3.511810302734375e-05,
      "model_forward_time": 3.133350133895874,
      "step": 9206
    },
    {
      "epoch": 3.511810302734375e-05,
      "step": 9206,
      "training_step_time": 10.180911540985107
    },
    {
      "epoch": 3.511810302734375e-05,
      "model_forward_time": 3.1381664276123047,
      "step": 9206
    },
    {
      "epoch": 3.511810302734375e-05,
      "step": 9206,
      "training_step_time": 9.908767223358154
    },
    {
      "epoch": 3.511810302734375e-05,
      "model_forward_time": 3.145195960998535,
      "step": 9206
    },
    {
      "epoch": 3.511810302734375e-05,
      "step": 9206,
      "training_step_time": 10.308953285217285
    },
    {
      "epoch": 3.512191772460938e-05,
      "model_forward_time": 3.158332586288452,
      "step": 9207
    },
    {
      "epoch": 3.512191772460938e-05,
      "step": 9207,
      "training_step_time": 9.931238651275635
    },
    {
      "epoch": 3.512191772460938e-05,
      "model_forward_time": 3.143244981765747,
      "step": 9207
    },
    {
      "epoch": 3.512191772460938e-05,
      "step": 9207,
      "training_step_time": 9.91571831703186
    },
    {
      "epoch": 3.512191772460938e-05,
      "model_forward_time": 3.4066178798675537,
      "step": 9207
    },
    {
      "dynamics_loss_avg": 0.06393910869956017,
      "epoch": 3.512191772460938e-05,
      "step": 9207
    },
    {
      "action_loss_avg": 0.027229491248726846,
      "epoch": 3.512191772460938e-05,
      "step": 9207
    },
    {
      "epoch": 3.512191772460938e-05,
      "step": 9207,
      "training_step_time": 10.17966365814209
    },
    {
      "epoch": 3.512191772460938e-05,
      "model_forward_time": 3.144620180130005,
      "step": 9207
    },
    {
      "epoch": 3.512191772460938e-05,
      "step": 9207,
      "training_step_time": 10.298463106155396
    },
    {
      "epoch": 3.5125732421875e-05,
      "model_forward_time": 3.157792806625366,
      "step": 9208
    },
    {
      "epoch": 3.5125732421875e-05,
      "step": 9208,
      "training_step_time": 9.929963111877441
    },
    {
      "epoch": 3.5125732421875e-05,
      "model_forward_time": 3.1388726234436035,
      "step": 9208
    },
    {
      "epoch": 3.5125732421875e-05,
      "step": 9208,
      "training_step_time": 9.904248476028442
    },
    {
      "epoch": 3.5125732421875e-05,
      "model_forward_time": 3.142015218734741,
      "step": 9208
    },
    {
      "epoch": 3.5125732421875e-05,
      "step": 9208,
      "training_step_time": 9.90756344795227
    },
    {
      "epoch": 3.5125732421875e-05,
      "model_forward_time": 3.1385724544525146,
      "step": 9208
    },
    {
      "epoch": 3.5125732421875e-05,
      "step": 9208,
      "training_step_time": 10.273854970932007
    },
    {
      "epoch": 3.5129547119140624e-05,
      "model_forward_time": 2.442640781402588,
      "step": 9209
    },
    {
      "epoch": 3.5129547119140624e-05,
      "step": 9209,
      "training_step_time": 7.467362403869629
    },
    {
      "epoch": 3.5129547119140624e-05,
      "model_forward_time": 3.137117862701416,
      "step": 9209
    },
    {
      "epoch": 3.5129547119140624e-05,
      "step": 9209,
      "training_step_time": 9.903597831726074
    },
    {
      "epoch": 3.5129547119140624e-05,
      "model_forward_time": 3.143275022506714,
      "step": 9209
    },
    {
      "epoch": 3.5129547119140624e-05,
      "step": 9209,
      "training_step_time": 9.910011053085327
    },
    {
      "epoch": 3.5129547119140624e-05,
      "model_forward_time": 3.1388607025146484,
      "step": 9209
    },
    {
      "epoch": 3.5129547119140624e-05,
      "step": 9209,
      "training_step_time": 10.016916990280151
    },
    {
      "epoch": 3.513336181640625e-05,
      "grad_norm": 0.476877897977829,
      "learning_rate": 1e-05,
      "loss": 0.0802,
      "step": 9210
    },
    {
      "epoch": 3.513336181640625e-05,
      "model_forward_time": 3.146376371383667,
      "step": 9210
    },
    {
      "dynamics_loss_avg": 0.04901727344840765,
      "epoch": 3.513336181640625e-05,
      "step": 9210
    },
    {
      "action_loss_avg": 0.015503999311476946,
      "epoch": 3.513336181640625e-05,
      "step": 9210
    },
    {
      "epoch": 3.513336181640625e-05,
      "step": 9210,
      "training_step_time": 10.191158056259155
    },
    {
      "epoch": 3.513336181640625e-05,
      "model_forward_time": 3.139786720275879,
      "step": 9210
    },
    {
      "epoch": 3.513336181640625e-05,
      "step": 9210,
      "training_step_time": 9.91103196144104
    },
    {
      "epoch": 3.513336181640625e-05,
      "model_forward_time": 3.14265513420105,
      "step": 9210
    },
    {
      "epoch": 3.513336181640625e-05,
      "step": 9210,
      "training_step_time": 9.910549402236938
    },
    {
      "epoch": 3.513336181640625e-05,
      "model_forward_time": 2.437080144882202,
      "step": 9210
    },
    {
      "epoch": 3.513336181640625e-05,
      "step": 9210,
      "training_step_time": 10.024880647659302
    },
    {
      "epoch": 3.5137176513671876e-05,
      "model_forward_time": 3.152926206588745,
      "step": 9211
    },
    {
      "epoch": 3.5137176513671876e-05,
      "step": 9211,
      "training_step_time": 9.913535118103027
    },
    {
      "epoch": 3.5137176513671876e-05,
      "model_forward_time": 3.1390726566314697,
      "step": 9211
    },
    {
      "epoch": 3.5137176513671876e-05,
      "step": 9211,
      "training_step_time": 9.89751935005188
    },
    {
      "epoch": 3.5137176513671876e-05,
      "model_forward_time": 2.7052366733551025,
      "step": 9211
    },
    {
      "epoch": 3.5137176513671876e-05,
      "step": 9211,
      "training_step_time": 7.722599506378174
    },
    {
      "epoch": 3.5137176513671876e-05,
      "model_forward_time": 3.126232624053955,
      "step": 9211
    },
    {
      "epoch": 3.5137176513671876e-05,
      "step": 9211,
      "training_step_time": 10.280522108078003
    },
    {
      "epoch": 3.51409912109375e-05,
      "model_forward_time": 3.173276662826538,
      "step": 9212
    },
    {
      "epoch": 3.51409912109375e-05,
      "step": 9212,
      "training_step_time": 9.933459043502808
    },
    {
      "epoch": 3.51409912109375e-05,
      "model_forward_time": 3.129199266433716,
      "step": 9212
    },
    {
      "epoch": 3.51409912109375e-05,
      "step": 9212,
      "training_step_time": 9.889966011047363
    },
    {
      "epoch": 3.51409912109375e-05,
      "model_forward_time": 3.409903049468994,
      "step": 9212
    },
    {
      "dynamics_loss_avg": 0.05019040890038014,
      "epoch": 3.51409912109375e-05,
      "step": 9212
    },
    {
      "action_loss_avg": 0.017562490608543157,
      "epoch": 3.51409912109375e-05,
      "step": 9212
    },
    {
      "epoch": 3.51409912109375e-05,
      "step": 9212,
      "training_step_time": 10.17669939994812
    },
    {
      "epoch": 3.51409912109375e-05,
      "model_forward_time": 3.131682872772217,
      "step": 9212
    },
    {
      "epoch": 3.51409912109375e-05,
      "step": 9212,
      "training_step_time": 10.264712810516357
    },
    {
      "epoch": 3.514480590820312e-05,
      "model_forward_time": 2.4495973587036133,
      "step": 9213
    },
    {
      "epoch": 3.514480590820312e-05,
      "step": 9213,
      "training_step_time": 7.477321147918701
    },
    {
      "epoch": 3.514480590820312e-05,
      "model_forward_time": 3.134423017501831,
      "step": 9213
    },
    {
      "epoch": 3.514480590820312e-05,
      "step": 9213,
      "training_step_time": 9.900617599487305
    },
    {
      "epoch": 3.514480590820312e-05,
      "model_forward_time": 2.4291703701019287,
      "step": 9213
    },
    {
      "epoch": 3.514480590820312e-05,
      "step": 9213,
      "training_step_time": 7.448947191238403
    },
    {
      "epoch": 3.514480590820312e-05,
      "model_forward_time": 3.1348307132720947,
      "step": 9213
    },
    {
      "epoch": 3.514480590820312e-05,
      "step": 9213,
      "training_step_time": 10.315162181854248
    },
    {
      "epoch": 3.514862060546875e-05,
      "model_forward_time": 3.4188179969787598,
      "step": 9214
    },
    {
      "epoch": 3.514862060546875e-05,
      "step": 9214,
      "training_step_time": 10.186706781387329
    },
    {
      "epoch": 3.514862060546875e-05,
      "model_forward_time": 3.1382100582122803,
      "step": 9214
    },
    {
      "epoch": 3.514862060546875e-05,
      "step": 9214,
      "training_step_time": 9.911556482315063
    },
    {
      "epoch": 3.514862060546875e-05,
      "model_forward_time": 3.142002820968628,
      "step": 9214
    },
    {
      "epoch": 3.514862060546875e-05,
      "step": 9214,
      "training_step_time": 9.917048692703247
    },
    {
      "epoch": 3.514862060546875e-05,
      "model_forward_time": 3.137556314468384,
      "step": 9214
    },
    {
      "epoch": 3.514862060546875e-05,
      "step": 9214,
      "training_step_time": 10.021244287490845
    },
    {
      "epoch": 3.5152435302734374e-05,
      "model_forward_time": 3.156059503555298,
      "step": 9215
    },
    {
      "dynamics_loss_avg": 0.06254386459477246,
      "epoch": 3.5152435302734374e-05,
      "step": 9215
    },
    {
      "action_loss_avg": 0.025131545157637446,
      "epoch": 3.5152435302734374e-05,
      "step": 9215
    },
    {
      "epoch": 3.5152435302734374e-05,
      "step": 9215,
      "training_step_time": 10.200282573699951
    },
    {
      "epoch": 3.5152435302734374e-05,
      "model_forward_time": 3.139338970184326,
      "step": 9215
    },
    {
      "epoch": 3.5152435302734374e-05,
      "step": 9215,
      "training_step_time": 9.911108493804932
    },
    {
      "epoch": 3.5152435302734374e-05,
      "model_forward_time": 2.4344704151153564,
      "step": 9215
    },
    {
      "epoch": 3.5152435302734374e-05,
      "step": 9215,
      "training_step_time": 7.462907314300537
    },
    {
      "epoch": 3.5152435302734374e-05,
      "model_forward_time": 3.1357250213623047,
      "step": 9215
    },
    {
      "epoch": 3.5152435302734374e-05,
      "step": 9215,
      "training_step_time": 10.253930568695068
    },
    {
      "epoch": 3.515625e-05,
      "model_forward_time": 3.1596829891204834,
      "step": 9216
    },
    {
      "epoch": 3.515625e-05,
      "step": 9216,
      "training_step_time": 9.934769868850708
    },
    {
      "epoch": 3.515625e-05,
      "model_forward_time": 3.1387412548065186,
      "step": 9216
    },
    {
      "epoch": 3.515625e-05,
      "step": 9216,
      "training_step_time": 10.17430067062378
    },
    {
      "epoch": 3.515625e-05,
      "model_forward_time": 3.143360137939453,
      "step": 9216
    },
    {
      "epoch": 3.515625e-05,
      "step": 9216,
      "training_step_time": 9.914222717285156
    },
    {
      "epoch": 3.515625e-05,
      "model_forward_time": 2.4376657009124756,
      "step": 9216
    },
    {
      "epoch": 3.515625e-05,
      "step": 9216,
      "training_step_time": 10.019607543945312
    },
    {
      "epoch": 3.516006469726563e-05,
      "model_forward_time": 2.4478020668029785,
      "step": 9217
    },
    {
      "epoch": 3.516006469726563e-05,
      "step": 9217,
      "training_step_time": 7.478271484375
    },
    {
      "epoch": 3.516006469726563e-05,
      "model_forward_time": 3.138228416442871,
      "step": 9217
    },
    {
      "epoch": 3.516006469726563e-05,
      "step": 9217,
      "training_step_time": 9.913004636764526
    },
    {
      "epoch": 3.516006469726563e-05,
      "model_forward_time": 3.136867046356201,
      "step": 9217
    },
    {
      "dynamics_loss_avg": 0.06094734463840723,
      "epoch": 3.516006469726563e-05,
      "step": 9217
    },
    {
      "action_loss_avg": 0.021853712014853956,
      "epoch": 3.516006469726563e-05,
      "step": 9217
    },
    {
      "epoch": 3.516006469726563e-05,
      "step": 9217,
      "training_step_time": 10.172805786132812
    },
    {
      "epoch": 3.516006469726563e-05,
      "model_forward_time": 3.142435073852539,
      "step": 9217
    },
    {
      "epoch": 3.516006469726563e-05,
      "step": 9217,
      "training_step_time": 10.304588556289673
    },
    {
      "epoch": 3.516387939453125e-05,
      "model_forward_time": 3.1528749465942383,
      "step": 9218
    },
    {
      "epoch": 3.516387939453125e-05,
      "step": 9218,
      "training_step_time": 9.924712419509888
    },
    {
      "epoch": 3.516387939453125e-05,
      "model_forward_time": 1.7452313899993896,
      "step": 9218
    },
    {
      "epoch": 3.516387939453125e-05,
      "step": 9218,
      "training_step_time": 5.199650049209595
    },
    {
      "epoch": 3.516387939453125e-05,
      "model_forward_time": 3.132105827331543,
      "step": 9218
    },
    {
      "epoch": 3.516387939453125e-05,
      "step": 9218,
      "training_step_time": 9.89666199684143
    },
    {
      "epoch": 3.516387939453125e-05,
      "model_forward_time": 3.13500714302063,
      "step": 9218
    },
    {
      "epoch": 3.516387939453125e-05,
      "step": 9218,
      "training_step_time": 10.284414768218994
    },
    {
      "epoch": 3.516769409179687e-05,
      "model_forward_time": 3.4704999923706055,
      "step": 9219
    },
    {
      "epoch": 3.516769409179687e-05,
      "step": 9219,
      "training_step_time": 10.234425067901611
    },
    {
      "epoch": 3.516769409179687e-05,
      "model_forward_time": 2.435465097427368,
      "step": 9219
    },
    {
      "epoch": 3.516769409179687e-05,
      "step": 9219,
      "training_step_time": 7.458938121795654
    },
    {
      "epoch": 3.516769409179687e-05,
      "model_forward_time": 3.141014337539673,
      "step": 9219
    },
    {
      "epoch": 3.516769409179687e-05,
      "step": 9219,
      "training_step_time": 9.906555414199829
    },
    {
      "epoch": 3.516769409179687e-05,
      "model_forward_time": 3.1424596309661865,
      "step": 9219
    },
    {
      "epoch": 3.516769409179687e-05,
      "step": 9219,
      "training_step_time": 10.26581072807312
    },
    {
      "epoch": 3.51715087890625e-05,
      "grad_norm": 0.4857575297355652,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 9220
    },
    {
      "epoch": 3.51715087890625e-05,
      "model_forward_time": 3.1587371826171875,
      "step": 9220
    },
    {
      "dynamics_loss_avg": 0.03539532627910376,
      "epoch": 3.51715087890625e-05,
      "step": 9220
    },
    {
      "action_loss_avg": 0.015293292049318553,
      "epoch": 3.51715087890625e-05,
      "step": 9220
    },
    {
      "epoch": 3.51715087890625e-05,
      "step": 9220,
      "training_step_time": 9.93738842010498
    },
    {
      "epoch": 3.51715087890625e-05,
      "model_forward_time": 3.388695240020752,
      "step": 9220
    },
    {
      "epoch": 3.51715087890625e-05,
      "step": 9220,
      "training_step_time": 10.158965587615967
    },
    {
      "epoch": 3.51715087890625e-05,
      "model_forward_time": 3.1441357135772705,
      "step": 9220
    },
    {
      "epoch": 3.51715087890625e-05,
      "step": 9220,
      "training_step_time": 9.916528463363647
    },
    {
      "epoch": 3.51715087890625e-05,
      "model_forward_time": 2.4362502098083496,
      "step": 9220
    },
    {
      "epoch": 3.51715087890625e-05,
      "step": 9220,
      "training_step_time": 10.328843832015991
    },
    {
      "epoch": 3.5175323486328125e-05,
      "model_forward_time": 3.145017385482788,
      "step": 9221
    },
    {
      "epoch": 3.5175323486328125e-05,
      "step": 9221,
      "training_step_time": 9.916267395019531
    },
    {
      "epoch": 3.5175323486328125e-05,
      "model_forward_time": 3.138845443725586,
      "step": 9221
    },
    {
      "epoch": 3.5175323486328125e-05,
      "step": 9221,
      "training_step_time": 10.17999267578125
    },
    {
      "epoch": 3.5175323486328125e-05,
      "model_forward_time": 3.1469898223876953,
      "step": 9221
    },
    {
      "epoch": 3.5175323486328125e-05,
      "step": 9221,
      "training_step_time": 9.91651177406311
    },
    {
      "epoch": 3.5175323486328125e-05,
      "model_forward_time": 3.142902135848999,
      "step": 9221
    },
    {
      "epoch": 3.5175323486328125e-05,
      "step": 9221,
      "training_step_time": 10.022546291351318
    },
    {
      "epoch": 3.517913818359375e-05,
      "model_forward_time": 3.149286985397339,
      "step": 9222
    },
    {
      "epoch": 3.517913818359375e-05,
      "step": 9222,
      "training_step_time": 9.920618534088135
    },
    {
      "epoch": 3.517913818359375e-05,
      "model_forward_time": 3.1493046283721924,
      "step": 9222
    },
    {
      "epoch": 3.517913818359375e-05,
      "step": 9222,
      "training_step_time": 9.921248435974121
    },
    {
      "epoch": 3.517913818359375e-05,
      "model_forward_time": 3.146282196044922,
      "step": 9222
    },
    {
      "dynamics_loss_avg": 0.07226459160447121,
      "epoch": 3.517913818359375e-05,
      "step": 9222
    },
    {
      "action_loss_avg": 0.02171115018427372,
      "epoch": 3.517913818359375e-05,
      "step": 9222
    },
    {
      "epoch": 3.517913818359375e-05,
      "step": 9222,
      "training_step_time": 10.192021131515503
    },
    {
      "epoch": 3.517913818359375e-05,
      "model_forward_time": 3.143925905227661,
      "step": 9222
    },
    {
      "epoch": 3.517913818359375e-05,
      "step": 9222,
      "training_step_time": 10.020225524902344
    },
    {
      "epoch": 3.518295288085938e-05,
      "model_forward_time": 3.166010618209839,
      "step": 9223
    },
    {
      "epoch": 3.518295288085938e-05,
      "step": 9223,
      "training_step_time": 9.946017503738403
    },
    {
      "epoch": 3.518295288085938e-05,
      "model_forward_time": 3.146819591522217,
      "step": 9223
    },
    {
      "epoch": 3.518295288085938e-05,
      "step": 9223,
      "training_step_time": 9.919426202774048
    },
    {
      "epoch": 3.518295288085938e-05,
      "model_forward_time": 2.441715717315674,
      "step": 9223
    },
    {
      "epoch": 3.518295288085938e-05,
      "step": 9223,
      "training_step_time": 7.472701549530029
    },
    {
      "epoch": 3.518295288085938e-05,
      "model_forward_time": 3.1446266174316406,
      "step": 9223
    },
    {
      "epoch": 3.518295288085938e-05,
      "step": 9223,
      "training_step_time": 10.342089414596558
    },
    {
      "epoch": 3.5186767578125e-05,
      "model_forward_time": 3.14766526222229,
      "step": 9224
    },
    {
      "epoch": 3.5186767578125e-05,
      "step": 9224,
      "training_step_time": 9.927650213241577
    },
    {
      "epoch": 3.5186767578125e-05,
      "model_forward_time": 2.4423434734344482,
      "step": 9224
    },
    {
      "epoch": 3.5186767578125e-05,
      "step": 9224,
      "training_step_time": 7.4734883308410645
    },
    {
      "epoch": 3.5186767578125e-05,
      "model_forward_time": 3.137986898422241,
      "step": 9224
    },
    {
      "epoch": 3.5186767578125e-05,
      "step": 9224,
      "training_step_time": 9.91146445274353
    },
    {
      "epoch": 3.5186767578125e-05,
      "model_forward_time": 3.144115447998047,
      "step": 9224
    },
    {
      "epoch": 3.5186767578125e-05,
      "step": 9224,
      "training_step_time": 10.025209665298462
    },
    {
      "epoch": 3.5190582275390624e-05,
      "model_forward_time": 3.1601483821868896,
      "step": 9225
    },
    {
      "dynamics_loss_avg": 0.05588063932955265,
      "epoch": 3.5190582275390624e-05,
      "step": 9225
    },
    {
      "action_loss_avg": 0.020790855586528777,
      "epoch": 3.5190582275390624e-05,
      "step": 9225
    },
    {
      "epoch": 3.5190582275390624e-05,
      "step": 9225,
      "training_step_time": 9.951361894607544
    },
    {
      "epoch": 3.5190582275390624e-05,
      "model_forward_time": 3.461099624633789,
      "step": 9225
    },
    {
      "epoch": 3.5190582275390624e-05,
      "step": 9225,
      "training_step_time": 10.244530439376831
    },
    {
      "epoch": 3.5190582275390624e-05,
      "model_forward_time": 3.1387784481048584,
      "step": 9225
    },
    {
      "epoch": 3.5190582275390624e-05,
      "step": 9225,
      "training_step_time": 9.913174867630005
    },
    {
      "epoch": 3.5190582275390624e-05,
      "model_forward_time": 2.4401438236236572,
      "step": 9225
    },
    {
      "epoch": 3.5190582275390624e-05,
      "step": 9225,
      "training_step_time": 10.255769491195679
    },
    {
      "epoch": 3.519439697265625e-05,
      "model_forward_time": 2.444477081298828,
      "step": 9226
    },
    {
      "epoch": 3.519439697265625e-05,
      "step": 9226,
      "training_step_time": 7.476646184921265
    },
    {
      "epoch": 3.519439697265625e-05,
      "model_forward_time": 3.136863946914673,
      "step": 9226
    },
    {
      "epoch": 3.519439697265625e-05,
      "step": 9226,
      "training_step_time": 9.913787126541138
    },
    {
      "epoch": 3.519439697265625e-05,
      "model_forward_time": 3.433332681655884,
      "step": 9226
    },
    {
      "epoch": 3.519439697265625e-05,
      "step": 9226,
      "training_step_time": 10.20090651512146
    },
    {
      "epoch": 3.519439697265625e-05,
      "model_forward_time": 3.1381945610046387,
      "step": 9226
    },
    {
      "epoch": 3.519439697265625e-05,
      "step": 9226,
      "training_step_time": 10.00084376335144
    },
    {
      "epoch": 3.5198211669921876e-05,
      "model_forward_time": 3.1461331844329834,
      "step": 9227
    },
    {
      "epoch": 3.5198211669921876e-05,
      "step": 9227,
      "training_step_time": 9.91929578781128
    },
    {
      "epoch": 3.5198211669921876e-05,
      "model_forward_time": 2.434382677078247,
      "step": 9227
    },
    {
      "epoch": 3.5198211669921876e-05,
      "step": 9227,
      "training_step_time": 7.458617448806763
    },
    {
      "epoch": 3.5198211669921876e-05,
      "model_forward_time": 3.1363348960876465,
      "step": 9227
    },
    {
      "dynamics_loss_avg": 0.05500519368797541,
      "epoch": 3.5198211669921876e-05,
      "step": 9227
    },
    {
      "action_loss_avg": 0.01648396267555654,
      "epoch": 3.5198211669921876e-05,
      "step": 9227
    },
    {
      "epoch": 3.5198211669921876e-05,
      "step": 9227,
      "training_step_time": 9.905009031295776
    },
    {
      "epoch": 3.5198211669921876e-05,
      "model_forward_time": 3.4139790534973145,
      "step": 9227
    },
    {
      "epoch": 3.5198211669921876e-05,
      "step": 9227,
      "training_step_time": 10.280436277389526
    },
    {
      "epoch": 3.52020263671875e-05,
      "model_forward_time": 3.1518592834472656,
      "step": 9228
    },
    {
      "epoch": 3.52020263671875e-05,
      "step": 9228,
      "training_step_time": 9.92716908454895
    },
    {
      "epoch": 3.52020263671875e-05,
      "model_forward_time": 3.136415958404541,
      "step": 9228
    },
    {
      "epoch": 3.52020263671875e-05,
      "step": 9228,
      "training_step_time": 9.910595178604126
    },
    {
      "epoch": 3.52020263671875e-05,
      "model_forward_time": 3.1423697471618652,
      "step": 9228
    },
    {
      "epoch": 3.52020263671875e-05,
      "step": 9228,
      "training_step_time": 9.929402589797974
    },
    {
      "epoch": 3.52020263671875e-05,
      "model_forward_time": 3.1456844806671143,
      "step": 9228
    },
    {
      "epoch": 3.52020263671875e-05,
      "step": 9228,
      "training_step_time": 10.249336242675781
    },
    {
      "epoch": 3.520584106445312e-05,
      "model_forward_time": 3.4930219650268555,
      "step": 9229
    },
    {
      "epoch": 3.520584106445312e-05,
      "step": 9229,
      "training_step_time": 10.272798538208008
    },
    {
      "epoch": 3.520584106445312e-05,
      "model_forward_time": 3.1437935829162598,
      "step": 9229
    },
    {
      "epoch": 3.520584106445312e-05,
      "step": 9229,
      "training_step_time": 9.923845052719116
    },
    {
      "epoch": 3.520584106445312e-05,
      "model_forward_time": 3.1458241939544678,
      "step": 9229
    },
    {
      "epoch": 3.520584106445312e-05,
      "step": 9229,
      "training_step_time": 9.924253463745117
    },
    {
      "epoch": 3.520584106445312e-05,
      "model_forward_time": 3.145479917526245,
      "step": 9229
    },
    {
      "epoch": 3.520584106445312e-05,
      "step": 9229,
      "training_step_time": 10.322640180587769
    },
    {
      "epoch": 3.520965576171875e-05,
      "grad_norm": 0.36103832721710205,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 9230
    },
    {
      "epoch": 3.520965576171875e-05,
      "model_forward_time": 3.4743854999542236,
      "step": 9230
    },
    {
      "dynamics_loss_avg": 0.04798458181321621,
      "epoch": 3.520965576171875e-05,
      "step": 9230
    },
    {
      "action_loss_avg": 0.01817132388241589,
      "epoch": 3.520965576171875e-05,
      "step": 9230
    },
    {
      "epoch": 3.520965576171875e-05,
      "step": 9230,
      "training_step_time": 10.249334812164307
    },
    {
      "epoch": 3.520965576171875e-05,
      "model_forward_time": 3.1384363174438477,
      "step": 9230
    },
    {
      "epoch": 3.520965576171875e-05,
      "step": 9230,
      "training_step_time": 9.91530156135559
    },
    {
      "epoch": 3.520965576171875e-05,
      "model_forward_time": 3.140122652053833,
      "step": 9230
    },
    {
      "epoch": 3.520965576171875e-05,
      "step": 9230,
      "training_step_time": 9.90753436088562
    },
    {
      "epoch": 3.520965576171875e-05,
      "model_forward_time": 3.1392292976379395,
      "step": 9230
    },
    {
      "epoch": 3.520965576171875e-05,
      "step": 9230,
      "training_step_time": 10.320957899093628
    },
    {
      "epoch": 3.5213470458984375e-05,
      "model_forward_time": 3.1453499794006348,
      "step": 9231
    },
    {
      "epoch": 3.5213470458984375e-05,
      "step": 9231,
      "training_step_time": 9.914575338363647
    },
    {
      "epoch": 3.5213470458984375e-05,
      "model_forward_time": 3.4159669876098633,
      "step": 9231
    },
    {
      "epoch": 3.5213470458984375e-05,
      "step": 9231,
      "training_step_time": 10.181756496429443
    },
    {
      "epoch": 3.5213470458984375e-05,
      "model_forward_time": 2.4360034465789795,
      "step": 9231
    },
    {
      "epoch": 3.5213470458984375e-05,
      "step": 9231,
      "training_step_time": 7.462226629257202
    },
    {
      "epoch": 3.5213470458984375e-05,
      "model_forward_time": 3.130577802658081,
      "step": 9231
    },
    {
      "epoch": 3.5213470458984375e-05,
      "step": 9231,
      "training_step_time": 10.282672643661499
    },
    {
      "epoch": 3.521728515625e-05,
      "model_forward_time": 2.448493719100952,
      "step": 9232
    },
    {
      "epoch": 3.521728515625e-05,
      "step": 9232,
      "training_step_time": 7.474234104156494
    },
    {
      "epoch": 3.521728515625e-05,
      "model_forward_time": 3.137791156768799,
      "step": 9232
    },
    {
      "epoch": 3.521728515625e-05,
      "step": 9232,
      "training_step_time": 9.903684377670288
    },
    {
      "epoch": 3.521728515625e-05,
      "model_forward_time": 3.136019706726074,
      "step": 9232
    },
    {
      "dynamics_loss_avg": 0.06421975493431091,
      "epoch": 3.521728515625e-05,
      "step": 9232
    },
    {
      "action_loss_avg": 0.025849657226353885,
      "epoch": 3.521728515625e-05,
      "step": 9232
    },
    {
      "epoch": 3.521728515625e-05,
      "step": 9232,
      "training_step_time": 9.906067609786987
    },
    {
      "epoch": 3.521728515625e-05,
      "model_forward_time": 3.3928885459899902,
      "step": 9232
    },
    {
      "epoch": 3.521728515625e-05,
      "step": 9232,
      "training_step_time": 10.256746053695679
    },
    {
      "epoch": 3.522109985351563e-05,
      "model_forward_time": 3.1476314067840576,
      "step": 9233
    },
    {
      "epoch": 3.522109985351563e-05,
      "step": 9233,
      "training_step_time": 9.915912628173828
    },
    {
      "epoch": 3.522109985351563e-05,
      "model_forward_time": 3.133395195007324,
      "step": 9233
    },
    {
      "epoch": 3.522109985351563e-05,
      "step": 9233,
      "training_step_time": 9.903616666793823
    },
    {
      "epoch": 3.522109985351563e-05,
      "model_forward_time": 3.134000778198242,
      "step": 9233
    },
    {
      "epoch": 3.522109985351563e-05,
      "step": 9233,
      "training_step_time": 9.903666257858276
    },
    {
      "epoch": 3.522109985351563e-05,
      "model_forward_time": 3.440822124481201,
      "step": 9233
    },
    {
      "epoch": 3.522109985351563e-05,
      "step": 9233,
      "training_step_time": 10.308018207550049
    },
    {
      "epoch": 3.522491455078125e-05,
      "model_forward_time": 3.156059980392456,
      "step": 9234
    },
    {
      "epoch": 3.522491455078125e-05,
      "step": 9234,
      "training_step_time": 9.938637018203735
    },
    {
      "epoch": 3.522491455078125e-05,
      "model_forward_time": 3.1403331756591797,
      "step": 9234
    },
    {
      "epoch": 3.522491455078125e-05,
      "step": 9234,
      "training_step_time": 9.915996551513672
    },
    {
      "epoch": 3.522491455078125e-05,
      "model_forward_time": 3.133908748626709,
      "step": 9234
    },
    {
      "epoch": 3.522491455078125e-05,
      "step": 9234,
      "training_step_time": 9.910756349563599
    },
    {
      "epoch": 3.522491455078125e-05,
      "model_forward_time": 3.140949010848999,
      "step": 9234
    },
    {
      "epoch": 3.522491455078125e-05,
      "step": 9234,
      "training_step_time": 10.31814956665039
    },
    {
      "epoch": 3.522872924804687e-05,
      "model_forward_time": 3.1576766967773438,
      "step": 9235
    },
    {
      "dynamics_loss_avg": 0.060244889557361604,
      "epoch": 3.522872924804687e-05,
      "step": 9235
    },
    {
      "action_loss_avg": 0.020588214695453643,
      "epoch": 3.522872924804687e-05,
      "step": 9235
    },
    {
      "epoch": 3.522872924804687e-05,
      "step": 9235,
      "training_step_time": 10.255141258239746
    },
    {
      "epoch": 3.522872924804687e-05,
      "model_forward_time": 3.1331114768981934,
      "step": 9235
    },
    {
      "epoch": 3.522872924804687e-05,
      "step": 9235,
      "training_step_time": 9.902016878128052
    },
    {
      "epoch": 3.522872924804687e-05,
      "model_forward_time": 3.1340720653533936,
      "step": 9235
    },
    {
      "epoch": 3.522872924804687e-05,
      "step": 9235,
      "training_step_time": 9.906327962875366
    },
    {
      "epoch": 3.522872924804687e-05,
      "model_forward_time": 3.1377830505371094,
      "step": 9235
    },
    {
      "epoch": 3.522872924804687e-05,
      "step": 9235,
      "training_step_time": 10.007854461669922
    },
    {
      "epoch": 3.52325439453125e-05,
      "model_forward_time": 3.1493465900421143,
      "step": 9236
    },
    {
      "epoch": 3.52325439453125e-05,
      "step": 9236,
      "training_step_time": 9.91818618774414
    },
    {
      "epoch": 3.52325439453125e-05,
      "model_forward_time": 3.4143996238708496,
      "step": 9236
    },
    {
      "epoch": 3.52325439453125e-05,
      "step": 9236,
      "training_step_time": 10.186212062835693
    },
    {
      "epoch": 3.52325439453125e-05,
      "model_forward_time": 3.1344902515411377,
      "step": 9236
    },
    {
      "epoch": 3.52325439453125e-05,
      "step": 9236,
      "training_step_time": 9.906738996505737
    },
    {
      "epoch": 3.52325439453125e-05,
      "model_forward_time": 3.1416175365448,
      "step": 9236
    },
    {
      "epoch": 3.52325439453125e-05,
      "step": 9236,
      "training_step_time": 10.010210990905762
    },
    {
      "epoch": 3.5236358642578125e-05,
      "model_forward_time": 3.1506166458129883,
      "step": 9237
    },
    {
      "epoch": 3.5236358642578125e-05,
      "step": 9237,
      "training_step_time": 9.924063682556152
    },
    {
      "epoch": 3.5236358642578125e-05,
      "model_forward_time": 3.1358962059020996,
      "step": 9237
    },
    {
      "epoch": 3.5236358642578125e-05,
      "step": 9237,
      "training_step_time": 9.907033205032349
    },
    {
      "epoch": 3.5236358642578125e-05,
      "model_forward_time": 3.1363296508789062,
      "step": 9237
    },
    {
      "dynamics_loss_avg": 0.06402848977595568,
      "epoch": 3.5236358642578125e-05,
      "step": 9237
    },
    {
      "action_loss_avg": 0.01858648136258125,
      "epoch": 3.5236358642578125e-05,
      "step": 9237
    },
    {
      "epoch": 3.5236358642578125e-05,
      "step": 9237,
      "training_step_time": 10.185025930404663
    },
    {
      "epoch": 3.5236358642578125e-05,
      "model_forward_time": 3.136303663253784,
      "step": 9237
    },
    {
      "epoch": 3.5236358642578125e-05,
      "step": 9237,
      "training_step_time": 10.020321130752563
    },
    {
      "epoch": 3.524017333984375e-05,
      "model_forward_time": 3.1552929878234863,
      "step": 9238
    },
    {
      "epoch": 3.524017333984375e-05,
      "step": 9238,
      "training_step_time": 9.934454202651978
    },
    {
      "epoch": 3.524017333984375e-05,
      "model_forward_time": 3.1374497413635254,
      "step": 9238
    },
    {
      "epoch": 3.524017333984375e-05,
      "step": 9238,
      "training_step_time": 9.913718223571777
    },
    {
      "epoch": 3.524017333984375e-05,
      "model_forward_time": 3.139664649963379,
      "step": 9238
    },
    {
      "epoch": 3.524017333984375e-05,
      "step": 9238,
      "training_step_time": 9.913759708404541
    },
    {
      "epoch": 3.524017333984375e-05,
      "model_forward_time": 3.45233416557312,
      "step": 9238
    },
    {
      "epoch": 3.524017333984375e-05,
      "step": 9238,
      "training_step_time": 10.321869850158691
    },
    {
      "epoch": 3.524398803710938e-05,
      "model_forward_time": 3.1492083072662354,
      "step": 9239
    },
    {
      "epoch": 3.524398803710938e-05,
      "step": 9239,
      "training_step_time": 9.926514387130737
    },
    {
      "epoch": 3.524398803710938e-05,
      "model_forward_time": 3.1363699436187744,
      "step": 9239
    },
    {
      "epoch": 3.524398803710938e-05,
      "step": 9239,
      "training_step_time": 9.914840459823608
    },
    {
      "epoch": 3.524398803710938e-05,
      "model_forward_time": 3.1455576419830322,
      "step": 9239
    },
    {
      "epoch": 3.524398803710938e-05,
      "step": 9239,
      "training_step_time": 9.921483039855957
    },
    {
      "epoch": 3.524398803710938e-05,
      "model_forward_time": 3.138284206390381,
      "step": 9239
    },
    {
      "epoch": 3.524398803710938e-05,
      "step": 9239,
      "training_step_time": 10.289708375930786
    },
    {
      "epoch": 3.5247802734375e-05,
      "grad_norm": 0.4634937644004822,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9240
    },
    {
      "epoch": 3.5247802734375e-05,
      "model_forward_time": 3.4394280910491943,
      "step": 9240
    },
    {
      "dynamics_loss_avg": 0.05552647262811661,
      "epoch": 3.5247802734375e-05,
      "step": 9240
    },
    {
      "action_loss_avg": 0.018259362038224935,
      "epoch": 3.5247802734375e-05,
      "step": 9240
    },
    {
      "epoch": 3.5247802734375e-05,
      "step": 9240,
      "training_step_time": 10.220131874084473
    },
    {
      "epoch": 3.5247802734375e-05,
      "model_forward_time": 3.1430864334106445,
      "step": 9240
    },
    {
      "epoch": 3.5247802734375e-05,
      "step": 9240,
      "training_step_time": 9.920032739639282
    },
    {
      "epoch": 3.5247802734375e-05,
      "model_forward_time": 3.146289825439453,
      "step": 9240
    },
    {
      "epoch": 3.5247802734375e-05,
      "step": 9240,
      "training_step_time": 9.924773216247559
    },
    {
      "epoch": 3.5247802734375e-05,
      "model_forward_time": 2.437382221221924,
      "step": 9240
    },
    {
      "epoch": 3.5247802734375e-05,
      "step": 9240,
      "training_step_time": 10.034785985946655
    },
    {
      "epoch": 3.5251617431640624e-05,
      "model_forward_time": 2.451301097869873,
      "step": 9241
    },
    {
      "epoch": 3.5251617431640624e-05,
      "step": 9241,
      "training_step_time": 7.488666534423828
    },
    {
      "epoch": 3.5251617431640624e-05,
      "model_forward_time": 3.1413440704345703,
      "step": 9241
    },
    {
      "epoch": 3.5251617431640624e-05,
      "step": 9241,
      "training_step_time": 10.185714960098267
    },
    {
      "epoch": 3.5251617431640624e-05,
      "model_forward_time": 3.1434695720672607,
      "step": 9241
    },
    {
      "epoch": 3.5251617431640624e-05,
      "step": 9241,
      "training_step_time": 9.921901226043701
    },
    {
      "epoch": 3.5251617431640624e-05,
      "model_forward_time": 3.140071392059326,
      "step": 9241
    },
    {
      "epoch": 3.5251617431640624e-05,
      "step": 9241,
      "training_step_time": 10.026457071304321
    },
    {
      "epoch": 3.525543212890625e-05,
      "model_forward_time": 3.1507396697998047,
      "step": 9242
    },
    {
      "epoch": 3.525543212890625e-05,
      "step": 9242,
      "training_step_time": 9.930713176727295
    },
    {
      "epoch": 3.525543212890625e-05,
      "model_forward_time": 3.1394596099853516,
      "step": 9242
    },
    {
      "epoch": 3.525543212890625e-05,
      "step": 9242,
      "training_step_time": 9.917836666107178
    },
    {
      "epoch": 3.525543212890625e-05,
      "model_forward_time": 3.4371769428253174,
      "step": 9242
    },
    {
      "dynamics_loss_avg": 0.06062191836535931,
      "epoch": 3.525543212890625e-05,
      "step": 9242
    },
    {
      "action_loss_avg": 0.021107257483527063,
      "epoch": 3.525543212890625e-05,
      "step": 9242
    },
    {
      "epoch": 3.525543212890625e-05,
      "step": 9242,
      "training_step_time": 10.212292909622192
    },
    {
      "epoch": 3.525543212890625e-05,
      "model_forward_time": 3.133540391921997,
      "step": 9242
    },
    {
      "epoch": 3.525543212890625e-05,
      "step": 9242,
      "training_step_time": 10.303616046905518
    },
    {
      "epoch": 3.5259246826171876e-05,
      "model_forward_time": 3.151242971420288,
      "step": 9243
    },
    {
      "epoch": 3.5259246826171876e-05,
      "step": 9243,
      "training_step_time": 9.926785707473755
    },
    {
      "epoch": 3.5259246826171876e-05,
      "model_forward_time": 2.432147264480591,
      "step": 9243
    },
    {
      "epoch": 3.5259246826171876e-05,
      "step": 9243,
      "training_step_time": 7.459213733673096
    },
    {
      "epoch": 3.5259246826171876e-05,
      "model_forward_time": 2.426018714904785,
      "step": 9243
    },
    {
      "epoch": 3.5259246826171876e-05,
      "step": 9243,
      "training_step_time": 7.448364734649658
    },
    {
      "epoch": 3.5259246826171876e-05,
      "model_forward_time": 3.131537914276123,
      "step": 9243
    },
    {
      "epoch": 3.5259246826171876e-05,
      "step": 9243,
      "training_step_time": 10.01126503944397
    },
    {
      "epoch": 3.52630615234375e-05,
      "model_forward_time": 3.4728987216949463,
      "step": 9244
    },
    {
      "epoch": 3.52630615234375e-05,
      "step": 9244,
      "training_step_time": 10.242925882339478
    },
    {
      "epoch": 3.52630615234375e-05,
      "model_forward_time": 2.433025598526001,
      "step": 9244
    },
    {
      "epoch": 3.52630615234375e-05,
      "step": 9244,
      "training_step_time": 7.457306385040283
    },
    {
      "epoch": 3.52630615234375e-05,
      "model_forward_time": 3.134056329727173,
      "step": 9244
    },
    {
      "epoch": 3.52630615234375e-05,
      "step": 9244,
      "training_step_time": 9.904133081436157
    },
    {
      "epoch": 3.52630615234375e-05,
      "model_forward_time": 3.140899658203125,
      "step": 9244
    },
    {
      "epoch": 3.52630615234375e-05,
      "step": 9244,
      "training_step_time": 10.246647596359253
    },
    {
      "epoch": 3.526687622070312e-05,
      "model_forward_time": 2.4485971927642822,
      "step": 9245
    },
    {
      "dynamics_loss_avg": 0.06250276528298855,
      "epoch": 3.526687622070312e-05,
      "step": 9245
    },
    {
      "action_loss_avg": 0.01998268482275307,
      "epoch": 3.526687622070312e-05,
      "step": 9245
    },
    {
      "epoch": 3.526687622070312e-05,
      "step": 9245,
      "training_step_time": 7.483370065689087
    },
    {
      "epoch": 3.526687622070312e-05,
      "model_forward_time": 3.4448370933532715,
      "step": 9245
    },
    {
      "epoch": 3.526687622070312e-05,
      "step": 9245,
      "training_step_time": 10.228191137313843
    },
    {
      "epoch": 3.526687622070312e-05,
      "model_forward_time": 3.142906904220581,
      "step": 9245
    },
    {
      "epoch": 3.526687622070312e-05,
      "step": 9245,
      "training_step_time": 9.920928478240967
    },
    {
      "epoch": 3.526687622070312e-05,
      "model_forward_time": 2.4389073848724365,
      "step": 9245
    },
    {
      "epoch": 3.526687622070312e-05,
      "step": 9245,
      "training_step_time": 10.031202554702759
    },
    {
      "epoch": 3.527069091796875e-05,
      "model_forward_time": 3.1512811183929443,
      "step": 9246
    },
    {
      "epoch": 3.527069091796875e-05,
      "step": 9246,
      "training_step_time": 9.932909727096558
    },
    {
      "epoch": 3.527069091796875e-05,
      "model_forward_time": 3.1387040615081787,
      "step": 9246
    },
    {
      "epoch": 3.527069091796875e-05,
      "step": 9246,
      "training_step_time": 9.918356895446777
    },
    {
      "epoch": 3.527069091796875e-05,
      "model_forward_time": 3.3935470581054688,
      "step": 9246
    },
    {
      "epoch": 3.527069091796875e-05,
      "step": 9246,
      "training_step_time": 10.171677589416504
    },
    {
      "epoch": 3.527069091796875e-05,
      "model_forward_time": 3.1446242332458496,
      "step": 9246
    },
    {
      "epoch": 3.527069091796875e-05,
      "step": 9246,
      "training_step_time": 10.03382134437561
    },
    {
      "epoch": 3.5274505615234375e-05,
      "model_forward_time": 3.1631619930267334,
      "step": 9247
    },
    {
      "epoch": 3.5274505615234375e-05,
      "step": 9247,
      "training_step_time": 9.939191341400146
    },
    {
      "epoch": 3.5274505615234375e-05,
      "model_forward_time": 3.1379501819610596,
      "step": 9247
    },
    {
      "epoch": 3.5274505615234375e-05,
      "step": 9247,
      "training_step_time": 9.913819789886475
    },
    {
      "epoch": 3.5274505615234375e-05,
      "model_forward_time": 3.405585527420044,
      "step": 9247
    },
    {
      "dynamics_loss_avg": 0.06659065913408994,
      "epoch": 3.5274505615234375e-05,
      "step": 9247
    },
    {
      "action_loss_avg": 0.020583706628531217,
      "epoch": 3.5274505615234375e-05,
      "step": 9247
    },
    {
      "epoch": 3.5274505615234375e-05,
      "step": 9247,
      "training_step_time": 10.17931079864502
    },
    {
      "epoch": 3.5274505615234375e-05,
      "model_forward_time": 3.1401407718658447,
      "step": 9247
    },
    {
      "epoch": 3.5274505615234375e-05,
      "step": 9247,
      "training_step_time": 10.020018100738525
    },
    {
      "epoch": 3.52783203125e-05,
      "model_forward_time": 3.1467103958129883,
      "step": 9248
    },
    {
      "epoch": 3.52783203125e-05,
      "step": 9248,
      "training_step_time": 9.933905601501465
    },
    {
      "epoch": 3.52783203125e-05,
      "model_forward_time": 3.139582395553589,
      "step": 9248
    },
    {
      "epoch": 3.52783203125e-05,
      "step": 9248,
      "training_step_time": 9.91940426826477
    },
    {
      "epoch": 3.52783203125e-05,
      "model_forward_time": 3.137789249420166,
      "step": 9248
    },
    {
      "epoch": 3.52783203125e-05,
      "step": 9248,
      "training_step_time": 9.911457300186157
    },
    {
      "epoch": 3.52783203125e-05,
      "model_forward_time": 3.134547710418701,
      "step": 9248
    },
    {
      "epoch": 3.52783203125e-05,
      "step": 9248,
      "training_step_time": 10.312592029571533
    },
    {
      "epoch": 3.528213500976563e-05,
      "model_forward_time": 3.199125289916992,
      "step": 9249
    },
    {
      "epoch": 3.528213500976563e-05,
      "step": 9249,
      "training_step_time": 9.982267141342163
    },
    {
      "epoch": 3.528213500976563e-05,
      "model_forward_time": 3.1406829357147217,
      "step": 9249
    },
    {
      "epoch": 3.528213500976563e-05,
      "step": 9249,
      "training_step_time": 9.924266815185547
    },
    {
      "epoch": 3.528213500976563e-05,
      "model_forward_time": 3.143324613571167,
      "step": 9249
    },
    {
      "epoch": 3.528213500976563e-05,
      "step": 9249,
      "training_step_time": 9.924708127975464
    },
    {
      "epoch": 3.528213500976563e-05,
      "model_forward_time": 3.1425418853759766,
      "step": 9249
    },
    {
      "epoch": 3.528213500976563e-05,
      "step": 9249,
      "training_step_time": 10.31962251663208
    },
    {
      "epoch": 3.528594970703125e-05,
      "grad_norm": 0.48934507369995117,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 9250
    },
    {
      "epoch": 3.528594970703125e-05,
      "model_forward_time": 2.7627105712890625,
      "step": 9250
    },
    {
      "dynamics_loss_avg": 0.06439825259149075,
      "epoch": 3.528594970703125e-05,
      "step": 9250
    },
    {
      "action_loss_avg": 0.023383509647101165,
      "epoch": 3.528594970703125e-05,
      "step": 9250
    },
    {
      "epoch": 3.528594970703125e-05,
      "step": 9250,
      "training_step_time": 7.811487674713135
    },
    {
      "epoch": 3.528594970703125e-05,
      "model_forward_time": 3.1392457485198975,
      "step": 9250
    },
    {
      "epoch": 3.528594970703125e-05,
      "step": 9250,
      "training_step_time": 9.912288904190063
    },
    {
      "epoch": 3.528594970703125e-05,
      "model_forward_time": 3.145590305328369,
      "step": 9250
    },
    {
      "epoch": 3.528594970703125e-05,
      "step": 9250,
      "training_step_time": 9.927541732788086
    },
    {
      "epoch": 3.528594970703125e-05,
      "model_forward_time": 2.4439196586608887,
      "step": 9250
    },
    {
      "epoch": 3.528594970703125e-05,
      "step": 9250,
      "training_step_time": 10.248108386993408
    },
    {
      "epoch": 3.528976440429687e-05,
      "model_forward_time": 3.1488678455352783,
      "step": 9251
    },
    {
      "epoch": 3.528976440429687e-05,
      "step": 9251,
      "training_step_time": 9.927596092224121
    },
    {
      "epoch": 3.528976440429687e-05,
      "model_forward_time": 3.142421245574951,
      "step": 9251
    },
    {
      "epoch": 3.528976440429687e-05,
      "step": 9251,
      "training_step_time": 10.229552507400513
    },
    {
      "epoch": 3.528976440429687e-05,
      "model_forward_time": 3.142076253890991,
      "step": 9251
    },
    {
      "epoch": 3.528976440429687e-05,
      "step": 9251,
      "training_step_time": 9.90703821182251
    },
    {
      "epoch": 3.528976440429687e-05,
      "model_forward_time": 3.1393353939056396,
      "step": 9251
    },
    {
      "epoch": 3.528976440429687e-05,
      "step": 9251,
      "training_step_time": 10.27217411994934
    },
    {
      "epoch": 3.52935791015625e-05,
      "model_forward_time": 3.1472814083099365,
      "step": 9252
    },
    {
      "epoch": 3.52935791015625e-05,
      "step": 9252,
      "training_step_time": 9.916597604751587
    },
    {
      "epoch": 3.52935791015625e-05,
      "model_forward_time": 3.1367027759552,
      "step": 9252
    },
    {
      "epoch": 3.52935791015625e-05,
      "step": 9252,
      "training_step_time": 9.91303563117981
    },
    {
      "epoch": 3.52935791015625e-05,
      "model_forward_time": 3.137373924255371,
      "step": 9252
    },
    {
      "dynamics_loss_avg": 0.04764657784253359,
      "epoch": 3.52935791015625e-05,
      "step": 9252
    },
    {
      "action_loss_avg": 0.018382694758474826,
      "epoch": 3.52935791015625e-05,
      "step": 9252
    },
    {
      "epoch": 3.52935791015625e-05,
      "step": 9252,
      "training_step_time": 10.208548545837402
    },
    {
      "epoch": 3.52935791015625e-05,
      "model_forward_time": 3.139660358428955,
      "step": 9252
    },
    {
      "epoch": 3.52935791015625e-05,
      "step": 9252,
      "training_step_time": 10.009894609451294
    },
    {
      "epoch": 3.5297393798828125e-05,
      "model_forward_time": 3.152843952178955,
      "step": 9253
    },
    {
      "epoch": 3.5297393798828125e-05,
      "step": 9253,
      "training_step_time": 9.929391622543335
    },
    {
      "epoch": 3.5297393798828125e-05,
      "model_forward_time": 3.1426355838775635,
      "step": 9253
    },
    {
      "epoch": 3.5297393798828125e-05,
      "step": 9253,
      "training_step_time": 9.917551517486572
    },
    {
      "epoch": 3.5297393798828125e-05,
      "model_forward_time": 3.144463539123535,
      "step": 9253
    },
    {
      "epoch": 3.5297393798828125e-05,
      "step": 9253,
      "training_step_time": 9.926517486572266
    },
    {
      "epoch": 3.5297393798828125e-05,
      "model_forward_time": 3.1491854190826416,
      "step": 9253
    },
    {
      "epoch": 3.5297393798828125e-05,
      "step": 9253,
      "training_step_time": 10.335721969604492
    },
    {
      "epoch": 3.530120849609375e-05,
      "model_forward_time": 3.160088062286377,
      "step": 9254
    },
    {
      "epoch": 3.530120849609375e-05,
      "step": 9254,
      "training_step_time": 9.940594673156738
    },
    {
      "epoch": 3.530120849609375e-05,
      "model_forward_time": 3.1466712951660156,
      "step": 9254
    },
    {
      "epoch": 3.530120849609375e-05,
      "step": 9254,
      "training_step_time": 9.921200513839722
    },
    {
      "epoch": 3.530120849609375e-05,
      "model_forward_time": 3.14263916015625,
      "step": 9254
    },
    {
      "epoch": 3.530120849609375e-05,
      "step": 9254,
      "training_step_time": 9.918490886688232
    },
    {
      "epoch": 3.530120849609375e-05,
      "model_forward_time": 3.1473946571350098,
      "step": 9254
    },
    {
      "epoch": 3.530120849609375e-05,
      "step": 9254,
      "training_step_time": 10.311723232269287
    },
    {
      "epoch": 3.530502319335938e-05,
      "model_forward_time": 3.1602566242218018,
      "step": 9255
    },
    {
      "dynamics_loss_avg": 0.037764635775238274,
      "epoch": 3.530502319335938e-05,
      "step": 9255
    },
    {
      "action_loss_avg": 0.014052727119997143,
      "epoch": 3.530502319335938e-05,
      "step": 9255
    },
    {
      "epoch": 3.530502319335938e-05,
      "step": 9255,
      "training_step_time": 10.249789237976074
    },
    {
      "epoch": 3.530502319335938e-05,
      "model_forward_time": 2.4382002353668213,
      "step": 9255
    },
    {
      "epoch": 3.530502319335938e-05,
      "step": 9255,
      "training_step_time": 7.469844341278076
    },
    {
      "epoch": 3.530502319335938e-05,
      "model_forward_time": 2.4368488788604736,
      "step": 9255
    },
    {
      "epoch": 3.530502319335938e-05,
      "step": 9255,
      "training_step_time": 7.463325023651123
    },
    {
      "epoch": 3.530502319335938e-05,
      "model_forward_time": 2.4338090419769287,
      "step": 9255
    },
    {
      "epoch": 3.530502319335938e-05,
      "step": 9255,
      "training_step_time": 10.296215534210205
    },
    {
      "epoch": 3.5308837890625e-05,
      "model_forward_time": 3.159336566925049,
      "step": 9256
    },
    {
      "epoch": 3.5308837890625e-05,
      "step": 9256,
      "training_step_time": 9.933529615402222
    },
    {
      "epoch": 3.5308837890625e-05,
      "model_forward_time": 3.138969659805298,
      "step": 9256
    },
    {
      "epoch": 3.5308837890625e-05,
      "step": 9256,
      "training_step_time": 9.907889604568481
    },
    {
      "epoch": 3.5308837890625e-05,
      "model_forward_time": 3.3987977504730225,
      "step": 9256
    },
    {
      "epoch": 3.5308837890625e-05,
      "step": 9256,
      "training_step_time": 10.165065288543701
    },
    {
      "epoch": 3.5308837890625e-05,
      "model_forward_time": 3.1397736072540283,
      "step": 9256
    },
    {
      "epoch": 3.5308837890625e-05,
      "step": 9256,
      "training_step_time": 10.010631561279297
    },
    {
      "epoch": 3.5312652587890624e-05,
      "model_forward_time": 3.1518466472625732,
      "step": 9257
    },
    {
      "epoch": 3.5312652587890624e-05,
      "step": 9257,
      "training_step_time": 9.919389724731445
    },
    {
      "epoch": 3.5312652587890624e-05,
      "model_forward_time": 3.136251211166382,
      "step": 9257
    },
    {
      "epoch": 3.5312652587890624e-05,
      "step": 9257,
      "training_step_time": 9.902734518051147
    },
    {
      "epoch": 3.5312652587890624e-05,
      "model_forward_time": 3.138015031814575,
      "step": 9257
    },
    {
      "dynamics_loss_avg": 0.057095821108669044,
      "epoch": 3.5312652587890624e-05,
      "step": 9257
    },
    {
      "action_loss_avg": 0.02228779518045485,
      "epoch": 3.5312652587890624e-05,
      "step": 9257
    },
    {
      "epoch": 3.5312652587890624e-05,
      "step": 9257,
      "training_step_time": 9.90608525276184
    },
    {
      "epoch": 3.5312652587890624e-05,
      "model_forward_time": 3.4132018089294434,
      "step": 9257
    },
    {
      "epoch": 3.5312652587890624e-05,
      "step": 9257,
      "training_step_time": 10.276115655899048
    },
    {
      "epoch": 3.531646728515625e-05,
      "model_forward_time": 3.1574759483337402,
      "step": 9258
    },
    {
      "epoch": 3.531646728515625e-05,
      "step": 9258,
      "training_step_time": 9.928338289260864
    },
    {
      "epoch": 3.531646728515625e-05,
      "model_forward_time": 3.138023853302002,
      "step": 9258
    },
    {
      "epoch": 3.531646728515625e-05,
      "step": 9258,
      "training_step_time": 9.904882192611694
    },
    {
      "epoch": 3.531646728515625e-05,
      "model_forward_time": 3.140819549560547,
      "step": 9258
    },
    {
      "epoch": 3.531646728515625e-05,
      "step": 9258,
      "training_step_time": 9.913395643234253
    },
    {
      "epoch": 3.531646728515625e-05,
      "model_forward_time": 3.138397693634033,
      "step": 9258
    },
    {
      "epoch": 3.531646728515625e-05,
      "step": 9258,
      "training_step_time": 10.327057123184204
    },
    {
      "epoch": 3.5320281982421876e-05,
      "model_forward_time": 3.160787582397461,
      "step": 9259
    },
    {
      "epoch": 3.5320281982421876e-05,
      "step": 9259,
      "training_step_time": 9.936064004898071
    },
    {
      "epoch": 3.5320281982421876e-05,
      "model_forward_time": 3.140002489089966,
      "step": 9259
    },
    {
      "epoch": 3.5320281982421876e-05,
      "step": 9259,
      "training_step_time": 9.912882089614868
    },
    {
      "epoch": 3.5320281982421876e-05,
      "model_forward_time": 3.1500275135040283,
      "step": 9259
    },
    {
      "epoch": 3.5320281982421876e-05,
      "step": 9259,
      "training_step_time": 9.927430152893066
    },
    {
      "epoch": 3.5320281982421876e-05,
      "model_forward_time": 3.1490390300750732,
      "step": 9259
    },
    {
      "epoch": 3.5320281982421876e-05,
      "step": 9259,
      "training_step_time": 10.327719926834106
    },
    {
      "epoch": 3.53240966796875e-05,
      "grad_norm": 0.372417688369751,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 9260
    },
    {
      "epoch": 3.53240966796875e-05,
      "model_forward_time": 3.163996458053589,
      "step": 9260
    },
    {
      "dynamics_loss_avg": 0.044564954936504364,
      "epoch": 3.53240966796875e-05,
      "step": 9260
    },
    {
      "action_loss_avg": 0.01700329682789743,
      "epoch": 3.53240966796875e-05,
      "step": 9260
    },
    {
      "epoch": 3.53240966796875e-05,
      "step": 9260,
      "training_step_time": 10.228603601455688
    },
    {
      "epoch": 3.53240966796875e-05,
      "model_forward_time": 3.1449201107025146,
      "step": 9260
    },
    {
      "epoch": 3.53240966796875e-05,
      "step": 9260,
      "training_step_time": 9.916685342788696
    },
    {
      "epoch": 3.53240966796875e-05,
      "model_forward_time": 3.1508727073669434,
      "step": 9260
    },
    {
      "epoch": 3.53240966796875e-05,
      "step": 9260,
      "training_step_time": 9.925280570983887
    },
    {
      "epoch": 3.53240966796875e-05,
      "model_forward_time": 3.14154314994812,
      "step": 9260
    },
    {
      "epoch": 3.53240966796875e-05,
      "step": 9260,
      "training_step_time": 10.264696836471558
    },
    {
      "epoch": 3.532791137695312e-05,
      "model_forward_time": 3.154325485229492,
      "step": 9261
    },
    {
      "epoch": 3.532791137695312e-05,
      "step": 9261,
      "training_step_time": 9.933227062225342
    },
    {
      "epoch": 3.532791137695312e-05,
      "model_forward_time": 3.4056994915008545,
      "step": 9261
    },
    {
      "epoch": 3.532791137695312e-05,
      "step": 9261,
      "training_step_time": 10.182605266571045
    },
    {
      "epoch": 3.532791137695312e-05,
      "model_forward_time": 3.146963119506836,
      "step": 9261
    },
    {
      "epoch": 3.532791137695312e-05,
      "step": 9261,
      "training_step_time": 9.926066637039185
    },
    {
      "epoch": 3.532791137695312e-05,
      "model_forward_time": 3.1449930667877197,
      "step": 9261
    },
    {
      "epoch": 3.532791137695312e-05,
      "step": 9261,
      "training_step_time": 10.038257360458374
    },
    {
      "epoch": 3.533172607421875e-05,
      "model_forward_time": 3.1533803939819336,
      "step": 9262
    },
    {
      "epoch": 3.533172607421875e-05,
      "step": 9262,
      "training_step_time": 9.936365604400635
    },
    {
      "epoch": 3.533172607421875e-05,
      "model_forward_time": 2.441260814666748,
      "step": 9262
    },
    {
      "epoch": 3.533172607421875e-05,
      "step": 9262,
      "training_step_time": 7.478730916976929
    },
    {
      "epoch": 3.533172607421875e-05,
      "model_forward_time": 3.1415867805480957,
      "step": 9262
    },
    {
      "dynamics_loss_avg": 0.06870624907314778,
      "epoch": 3.533172607421875e-05,
      "step": 9262
    },
    {
      "action_loss_avg": 0.025087555311620234,
      "epoch": 3.533172607421875e-05,
      "step": 9262
    },
    {
      "epoch": 3.533172607421875e-05,
      "step": 9262,
      "training_step_time": 10.205513954162598
    },
    {
      "epoch": 3.533172607421875e-05,
      "model_forward_time": 3.146467924118042,
      "step": 9262
    },
    {
      "epoch": 3.533172607421875e-05,
      "step": 9262,
      "training_step_time": 10.285367488861084
    },
    {
      "epoch": 3.5335540771484375e-05,
      "model_forward_time": 3.1576993465423584,
      "step": 9263
    },
    {
      "epoch": 3.5335540771484375e-05,
      "step": 9263,
      "training_step_time": 9.940420389175415
    },
    {
      "epoch": 3.5335540771484375e-05,
      "model_forward_time": 3.14634370803833,
      "step": 9263
    },
    {
      "epoch": 3.5335540771484375e-05,
      "step": 9263,
      "training_step_time": 9.91839075088501
    },
    {
      "epoch": 3.5335540771484375e-05,
      "model_forward_time": 3.143765926361084,
      "step": 9263
    },
    {
      "epoch": 3.5335540771484375e-05,
      "step": 9263,
      "training_step_time": 9.920918941497803
    },
    {
      "epoch": 3.5335540771484375e-05,
      "model_forward_time": 3.1460838317871094,
      "step": 9263
    },
    {
      "epoch": 3.5335540771484375e-05,
      "step": 9263,
      "training_step_time": 10.295607328414917
    },
    {
      "epoch": 3.533935546875e-05,
      "model_forward_time": 2.4508209228515625,
      "step": 9264
    },
    {
      "epoch": 3.533935546875e-05,
      "step": 9264,
      "training_step_time": 7.4848082065582275
    },
    {
      "epoch": 3.533935546875e-05,
      "model_forward_time": 3.14322566986084,
      "step": 9264
    },
    {
      "epoch": 3.533935546875e-05,
      "step": 9264,
      "training_step_time": 9.920537233352661
    },
    {
      "epoch": 3.533935546875e-05,
      "model_forward_time": 3.1433615684509277,
      "step": 9264
    },
    {
      "epoch": 3.533935546875e-05,
      "step": 9264,
      "training_step_time": 9.917935609817505
    },
    {
      "epoch": 3.533935546875e-05,
      "model_forward_time": 3.1511690616607666,
      "step": 9264
    },
    {
      "epoch": 3.533935546875e-05,
      "step": 9264,
      "training_step_time": 10.341250658035278
    },
    {
      "epoch": 3.534317016601563e-05,
      "model_forward_time": 3.1535236835479736,
      "step": 9265
    },
    {
      "dynamics_loss_avg": 0.07563069490715861,
      "epoch": 3.534317016601563e-05,
      "step": 9265
    },
    {
      "action_loss_avg": 0.025893862685188652,
      "epoch": 3.534317016601563e-05,
      "step": 9265
    },
    {
      "epoch": 3.534317016601563e-05,
      "step": 9265,
      "training_step_time": 9.932522058486938
    },
    {
      "epoch": 3.534317016601563e-05,
      "model_forward_time": 3.4200873374938965,
      "step": 9265
    },
    {
      "epoch": 3.534317016601563e-05,
      "step": 9265,
      "training_step_time": 10.193598508834839
    },
    {
      "epoch": 3.534317016601563e-05,
      "model_forward_time": 3.1446197032928467,
      "step": 9265
    },
    {
      "epoch": 3.534317016601563e-05,
      "step": 9265,
      "training_step_time": 9.91694188117981
    },
    {
      "epoch": 3.534317016601563e-05,
      "model_forward_time": 3.148411273956299,
      "step": 9265
    },
    {
      "epoch": 3.534317016601563e-05,
      "step": 9265,
      "training_step_time": 10.02587080001831
    },
    {
      "epoch": 3.534698486328125e-05,
      "model_forward_time": 3.165581703186035,
      "step": 9266
    },
    {
      "epoch": 3.534698486328125e-05,
      "step": 9266,
      "training_step_time": 9.946752786636353
    },
    {
      "epoch": 3.534698486328125e-05,
      "model_forward_time": 3.1459968090057373,
      "step": 9266
    },
    {
      "epoch": 3.534698486328125e-05,
      "step": 9266,
      "training_step_time": 10.23727035522461
    },
    {
      "epoch": 3.534698486328125e-05,
      "model_forward_time": 2.4473044872283936,
      "step": 9266
    },
    {
      "epoch": 3.534698486328125e-05,
      "step": 9266,
      "training_step_time": 7.487175226211548
    },
    {
      "epoch": 3.534698486328125e-05,
      "model_forward_time": 2.4411158561706543,
      "step": 9266
    },
    {
      "epoch": 3.534698486328125e-05,
      "step": 9266,
      "training_step_time": 10.029544830322266
    },
    {
      "epoch": 3.535079956054687e-05,
      "model_forward_time": 3.1550464630126953,
      "step": 9267
    },
    {
      "epoch": 3.535079956054687e-05,
      "step": 9267,
      "training_step_time": 9.932405471801758
    },
    {
      "epoch": 3.535079956054687e-05,
      "model_forward_time": 3.1421895027160645,
      "step": 9267
    },
    {
      "epoch": 3.535079956054687e-05,
      "step": 9267,
      "training_step_time": 9.919553279876709
    },
    {
      "epoch": 3.535079956054687e-05,
      "model_forward_time": 3.1445727348327637,
      "step": 9267
    },
    {
      "dynamics_loss_avg": 0.06103233369067311,
      "epoch": 3.535079956054687e-05,
      "step": 9267
    },
    {
      "action_loss_avg": 0.018936555041000246,
      "epoch": 3.535079956054687e-05,
      "step": 9267
    },
    {
      "epoch": 3.535079956054687e-05,
      "step": 9267,
      "training_step_time": 9.919930219650269
    },
    {
      "epoch": 3.535079956054687e-05,
      "model_forward_time": 3.435420036315918,
      "step": 9267
    },
    {
      "epoch": 3.535079956054687e-05,
      "step": 9267,
      "training_step_time": 10.308351993560791
    },
    {
      "epoch": 3.53546142578125e-05,
      "model_forward_time": 2.4569785594940186,
      "step": 9268
    },
    {
      "epoch": 3.53546142578125e-05,
      "step": 9268,
      "training_step_time": 7.487035512924194
    },
    {
      "epoch": 3.53546142578125e-05,
      "model_forward_time": 2.436488628387451,
      "step": 9268
    },
    {
      "epoch": 3.53546142578125e-05,
      "step": 9268,
      "training_step_time": 7.46790075302124
    },
    {
      "epoch": 3.53546142578125e-05,
      "model_forward_time": 3.145998477935791,
      "step": 9268
    },
    {
      "epoch": 3.53546142578125e-05,
      "step": 9268,
      "training_step_time": 9.923054695129395
    },
    {
      "epoch": 3.53546142578125e-05,
      "model_forward_time": 2.43778657913208,
      "step": 9268
    },
    {
      "epoch": 3.53546142578125e-05,
      "step": 9268,
      "training_step_time": 10.282649278640747
    },
    {
      "epoch": 3.5358428955078125e-05,
      "model_forward_time": 3.4577443599700928,
      "step": 9269
    },
    {
      "epoch": 3.5358428955078125e-05,
      "step": 9269,
      "training_step_time": 10.240532398223877
    },
    {
      "epoch": 3.5358428955078125e-05,
      "model_forward_time": 3.1460249423980713,
      "step": 9269
    },
    {
      "epoch": 3.5358428955078125e-05,
      "step": 9269,
      "training_step_time": 9.930397272109985
    },
    {
      "epoch": 3.5358428955078125e-05,
      "model_forward_time": 3.150175094604492,
      "step": 9269
    },
    {
      "epoch": 3.5358428955078125e-05,
      "step": 9269,
      "training_step_time": 9.935964107513428
    },
    {
      "epoch": 3.5358428955078125e-05,
      "model_forward_time": 3.1503264904022217,
      "step": 9269
    },
    {
      "epoch": 3.5358428955078125e-05,
      "step": 9269,
      "training_step_time": 10.300676107406616
    },
    {
      "epoch": 3.536224365234375e-05,
      "grad_norm": 0.5300769805908203,
      "learning_rate": 1e-05,
      "loss": 0.0788,
      "step": 9270
    },
    {
      "epoch": 3.536224365234375e-05,
      "model_forward_time": 3.1585421562194824,
      "step": 9270
    },
    {
      "dynamics_loss_avg": 0.04681230057030916,
      "epoch": 3.536224365234375e-05,
      "step": 9270
    },
    {
      "action_loss_avg": 0.014225353812798857,
      "epoch": 3.536224365234375e-05,
      "step": 9270
    },
    {
      "epoch": 3.536224365234375e-05,
      "step": 9270,
      "training_step_time": 9.939907312393188
    },
    {
      "epoch": 3.536224365234375e-05,
      "model_forward_time": 3.401561737060547,
      "step": 9270
    },
    {
      "epoch": 3.536224365234375e-05,
      "step": 9270,
      "training_step_time": 10.178929090499878
    },
    {
      "epoch": 3.536224365234375e-05,
      "model_forward_time": 3.146322727203369,
      "step": 9270
    },
    {
      "epoch": 3.536224365234375e-05,
      "step": 9270,
      "training_step_time": 9.922208547592163
    },
    {
      "epoch": 3.536224365234375e-05,
      "model_forward_time": 2.445405960083008,
      "step": 9270
    },
    {
      "epoch": 3.536224365234375e-05,
      "step": 9270,
      "training_step_time": 10.22590947151184
    },
    {
      "epoch": 3.536605834960938e-05,
      "model_forward_time": 3.1629281044006348,
      "step": 9271
    },
    {
      "epoch": 3.536605834960938e-05,
      "step": 9271,
      "training_step_time": 9.938195705413818
    },
    {
      "epoch": 3.536605834960938e-05,
      "model_forward_time": 3.145068883895874,
      "step": 9271
    },
    {
      "epoch": 3.536605834960938e-05,
      "step": 9271,
      "training_step_time": 9.916776418685913
    },
    {
      "epoch": 3.536605834960938e-05,
      "model_forward_time": 3.42117977142334,
      "step": 9271
    },
    {
      "epoch": 3.536605834960938e-05,
      "step": 9271,
      "training_step_time": 10.19392466545105
    },
    {
      "epoch": 3.536605834960938e-05,
      "model_forward_time": 3.140650987625122,
      "step": 9271
    },
    {
      "epoch": 3.536605834960938e-05,
      "step": 9271,
      "training_step_time": 10.246146440505981
    },
    {
      "epoch": 3.5369873046875e-05,
      "model_forward_time": 2.4536795616149902,
      "step": 9272
    },
    {
      "epoch": 3.5369873046875e-05,
      "step": 9272,
      "training_step_time": 7.488409042358398
    },
    {
      "epoch": 3.5369873046875e-05,
      "model_forward_time": 3.140153169631958,
      "step": 9272
    },
    {
      "epoch": 3.5369873046875e-05,
      "step": 9272,
      "training_step_time": 9.90946626663208
    },
    {
      "epoch": 3.5369873046875e-05,
      "model_forward_time": 3.13364315032959,
      "step": 9272
    },
    {
      "dynamics_loss_avg": 0.06663151755928994,
      "epoch": 3.5369873046875e-05,
      "step": 9272
    },
    {
      "action_loss_avg": 0.029310445301234722,
      "epoch": 3.5369873046875e-05,
      "step": 9272
    },
    {
      "epoch": 3.5369873046875e-05,
      "step": 9272,
      "training_step_time": 9.905423641204834
    },
    {
      "epoch": 3.5369873046875e-05,
      "model_forward_time": 3.3945376873016357,
      "step": 9272
    },
    {
      "epoch": 3.5369873046875e-05,
      "step": 9272,
      "training_step_time": 10.2643723487854
    },
    {
      "epoch": 3.5373687744140624e-05,
      "model_forward_time": 3.1482391357421875,
      "step": 9273
    },
    {
      "epoch": 3.5373687744140624e-05,
      "step": 9273,
      "training_step_time": 9.92102861404419
    },
    {
      "epoch": 3.5373687744140624e-05,
      "model_forward_time": 2.437091112136841,
      "step": 9273
    },
    {
      "epoch": 3.5373687744140624e-05,
      "step": 9273,
      "training_step_time": 7.469714879989624
    },
    {
      "epoch": 3.5373687744140624e-05,
      "model_forward_time": 3.1412861347198486,
      "step": 9273
    },
    {
      "epoch": 3.5373687744140624e-05,
      "step": 9273,
      "training_step_time": 9.912158966064453
    },
    {
      "epoch": 3.5373687744140624e-05,
      "model_forward_time": 3.145371675491333,
      "step": 9273
    },
    {
      "epoch": 3.5373687744140624e-05,
      "step": 9273,
      "training_step_time": 10.275052309036255
    },
    {
      "epoch": 3.537750244140625e-05,
      "model_forward_time": 3.4112865924835205,
      "step": 9274
    },
    {
      "epoch": 3.537750244140625e-05,
      "step": 9274,
      "training_step_time": 10.18764853477478
    },
    {
      "epoch": 3.537750244140625e-05,
      "model_forward_time": 3.146251916885376,
      "step": 9274
    },
    {
      "epoch": 3.537750244140625e-05,
      "step": 9274,
      "training_step_time": 9.9241042137146
    },
    {
      "epoch": 3.537750244140625e-05,
      "model_forward_time": 3.1469192504882812,
      "step": 9274
    },
    {
      "epoch": 3.537750244140625e-05,
      "step": 9274,
      "training_step_time": 9.924489259719849
    },
    {
      "epoch": 3.537750244140625e-05,
      "model_forward_time": 3.146116256713867,
      "step": 9274
    },
    {
      "epoch": 3.537750244140625e-05,
      "step": 9274,
      "training_step_time": 10.325035810470581
    },
    {
      "epoch": 3.5381317138671876e-05,
      "model_forward_time": 3.168334722518921,
      "step": 9275
    },
    {
      "dynamics_loss_avg": 0.058939211815595624,
      "epoch": 3.5381317138671876e-05,
      "step": 9275
    },
    {
      "action_loss_avg": 0.022612866666167974,
      "epoch": 3.5381317138671876e-05,
      "step": 9275
    },
    {
      "epoch": 3.5381317138671876e-05,
      "step": 9275,
      "training_step_time": 9.956288576126099
    },
    {
      "epoch": 3.5381317138671876e-05,
      "model_forward_time": 3.4271628856658936,
      "step": 9275
    },
    {
      "epoch": 3.5381317138671876e-05,
      "step": 9275,
      "training_step_time": 10.202715158462524
    },
    {
      "epoch": 3.5381317138671876e-05,
      "model_forward_time": 3.14506459236145,
      "step": 9275
    },
    {
      "epoch": 3.5381317138671876e-05,
      "step": 9275,
      "training_step_time": 9.916746139526367
    },
    {
      "epoch": 3.5381317138671876e-05,
      "model_forward_time": 3.149613618850708,
      "step": 9275
    },
    {
      "epoch": 3.5381317138671876e-05,
      "step": 9275,
      "training_step_time": 10.270157814025879
    },
    {
      "epoch": 3.53851318359375e-05,
      "model_forward_time": 2.4543893337249756,
      "step": 9276
    },
    {
      "epoch": 3.53851318359375e-05,
      "step": 9276,
      "training_step_time": 7.487335920333862
    },
    {
      "epoch": 3.53851318359375e-05,
      "model_forward_time": 3.141909599304199,
      "step": 9276
    },
    {
      "epoch": 3.53851318359375e-05,
      "step": 9276,
      "training_step_time": 9.916863918304443
    },
    {
      "epoch": 3.53851318359375e-05,
      "model_forward_time": 3.399869680404663,
      "step": 9276
    },
    {
      "epoch": 3.53851318359375e-05,
      "step": 9276,
      "training_step_time": 10.174427270889282
    },
    {
      "epoch": 3.53851318359375e-05,
      "model_forward_time": 3.1458961963653564,
      "step": 9276
    },
    {
      "epoch": 3.53851318359375e-05,
      "step": 9276,
      "training_step_time": 10.016377210617065
    },
    {
      "epoch": 3.538894653320312e-05,
      "model_forward_time": 3.1581084728240967,
      "step": 9277
    },
    {
      "epoch": 3.538894653320312e-05,
      "step": 9277,
      "training_step_time": 9.93409013748169
    },
    {
      "epoch": 3.538894653320312e-05,
      "model_forward_time": 3.1425697803497314,
      "step": 9277
    },
    {
      "epoch": 3.538894653320312e-05,
      "step": 9277,
      "training_step_time": 9.91771411895752
    },
    {
      "epoch": 3.538894653320312e-05,
      "model_forward_time": 3.148932456970215,
      "step": 9277
    },
    {
      "dynamics_loss_avg": 0.04925363352522254,
      "epoch": 3.538894653320312e-05,
      "step": 9277
    },
    {
      "action_loss_avg": 0.01777772419154644,
      "epoch": 3.538894653320312e-05,
      "step": 9277
    },
    {
      "epoch": 3.538894653320312e-05,
      "step": 9277,
      "training_step_time": 9.929259061813354
    },
    {
      "epoch": 3.538894653320312e-05,
      "model_forward_time": 3.4565176963806152,
      "step": 9277
    },
    {
      "epoch": 3.538894653320312e-05,
      "step": 9277,
      "training_step_time": 10.331223487854004
    },
    {
      "epoch": 3.539276123046875e-05,
      "model_forward_time": 3.1682417392730713,
      "step": 9278
    },
    {
      "epoch": 3.539276123046875e-05,
      "step": 9278,
      "training_step_time": 9.948073148727417
    },
    {
      "epoch": 3.539276123046875e-05,
      "model_forward_time": 3.1470844745635986,
      "step": 9278
    },
    {
      "epoch": 3.539276123046875e-05,
      "step": 9278,
      "training_step_time": 9.925196886062622
    },
    {
      "epoch": 3.539276123046875e-05,
      "model_forward_time": 3.148977518081665,
      "step": 9278
    },
    {
      "epoch": 3.539276123046875e-05,
      "step": 9278,
      "training_step_time": 9.927454710006714
    },
    {
      "epoch": 3.539276123046875e-05,
      "model_forward_time": 2.7402029037475586,
      "step": 9278
    },
    {
      "epoch": 3.539276123046875e-05,
      "step": 9278,
      "training_step_time": 10.316603422164917
    },
    {
      "epoch": 3.5396575927734375e-05,
      "model_forward_time": 2.4514694213867188,
      "step": 9279
    },
    {
      "epoch": 3.5396575927734375e-05,
      "step": 9279,
      "training_step_time": 7.495176792144775
    },
    {
      "epoch": 3.5396575927734375e-05,
      "model_forward_time": 3.141043186187744,
      "step": 9279
    },
    {
      "epoch": 3.5396575927734375e-05,
      "step": 9279,
      "training_step_time": 9.934577941894531
    },
    {
      "epoch": 3.5396575927734375e-05,
      "model_forward_time": 3.144545316696167,
      "step": 9279
    },
    {
      "epoch": 3.5396575927734375e-05,
      "step": 9279,
      "training_step_time": 9.922392129898071
    },
    {
      "epoch": 3.5396575927734375e-05,
      "model_forward_time": 3.1344475746154785,
      "step": 9279
    },
    {
      "epoch": 3.5396575927734375e-05,
      "step": 9279,
      "training_step_time": 10.305267810821533
    },
    {
      "epoch": 3.5400390625e-05,
      "grad_norm": 0.31702110171318054,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 9280
    },
    {
      "epoch": 3.5400390625e-05,
      "model_forward_time": 3.1467537879943848,
      "step": 9280
    },
    {
      "dynamics_loss_avg": 0.049510125257074836,
      "epoch": 3.5400390625e-05,
      "step": 9280
    },
    {
      "action_loss_avg": 0.02352017816156149,
      "epoch": 3.5400390625e-05,
      "step": 9280
    },
    {
      "epoch": 3.5400390625e-05,
      "step": 9280,
      "training_step_time": 9.9155752658844
    },
    {
      "epoch": 3.5400390625e-05,
      "model_forward_time": 3.522883892059326,
      "step": 9280
    },
    {
      "epoch": 3.5400390625e-05,
      "step": 9280,
      "training_step_time": 10.315370798110962
    },
    {
      "epoch": 3.5400390625e-05,
      "model_forward_time": 3.1541099548339844,
      "step": 9280
    },
    {
      "epoch": 3.5400390625e-05,
      "step": 9280,
      "training_step_time": 9.930684804916382
    },
    {
      "epoch": 3.5400390625e-05,
      "model_forward_time": 3.1398348808288574,
      "step": 9280
    },
    {
      "epoch": 3.5400390625e-05,
      "step": 9280,
      "training_step_time": 10.32567834854126
    },
    {
      "epoch": 3.540420532226563e-05,
      "model_forward_time": 3.160611629486084,
      "step": 9281
    },
    {
      "epoch": 3.540420532226563e-05,
      "step": 9281,
      "training_step_time": 9.939636707305908
    },
    {
      "epoch": 3.540420532226563e-05,
      "model_forward_time": 3.4524412155151367,
      "step": 9281
    },
    {
      "epoch": 3.540420532226563e-05,
      "step": 9281,
      "training_step_time": 10.22290325164795
    },
    {
      "epoch": 3.540420532226563e-05,
      "model_forward_time": 3.136219024658203,
      "step": 9281
    },
    {
      "epoch": 3.540420532226563e-05,
      "step": 9281,
      "training_step_time": 9.903964281082153
    },
    {
      "epoch": 3.540420532226563e-05,
      "model_forward_time": 3.132493734359741,
      "step": 9281
    },
    {
      "epoch": 3.540420532226563e-05,
      "step": 9281,
      "training_step_time": 10.317885875701904
    },
    {
      "epoch": 3.540802001953125e-05,
      "model_forward_time": 3.142207384109497,
      "step": 9282
    },
    {
      "epoch": 3.540802001953125e-05,
      "step": 9282,
      "training_step_time": 9.913547277450562
    },
    {
      "epoch": 3.540802001953125e-05,
      "model_forward_time": 3.1334383487701416,
      "step": 9282
    },
    {
      "epoch": 3.540802001953125e-05,
      "step": 9282,
      "training_step_time": 9.904389142990112
    },
    {
      "epoch": 3.540802001953125e-05,
      "model_forward_time": 2.4290378093719482,
      "step": 9282
    },
    {
      "dynamics_loss_avg": 0.046364357881248,
      "epoch": 3.540802001953125e-05,
      "step": 9282
    },
    {
      "action_loss_avg": 0.021883268700912596,
      "epoch": 3.540802001953125e-05,
      "step": 9282
    },
    {
      "epoch": 3.540802001953125e-05,
      "step": 9282,
      "training_step_time": 7.45601224899292
    },
    {
      "epoch": 3.540802001953125e-05,
      "model_forward_time": 3.383857250213623,
      "step": 9282
    },
    {
      "epoch": 3.540802001953125e-05,
      "step": 9282,
      "training_step_time": 10.302777767181396
    },
    {
      "epoch": 3.541183471679687e-05,
      "model_forward_time": 3.140841007232666,
      "step": 9283
    },
    {
      "epoch": 3.541183471679687e-05,
      "step": 9283,
      "training_step_time": 9.917057275772095
    },
    {
      "epoch": 3.541183471679687e-05,
      "model_forward_time": 3.1378512382507324,
      "step": 9283
    },
    {
      "epoch": 3.541183471679687e-05,
      "step": 9283,
      "training_step_time": 9.904911279678345
    },
    {
      "epoch": 3.541183471679687e-05,
      "model_forward_time": 3.1298274993896484,
      "step": 9283
    },
    {
      "epoch": 3.541183471679687e-05,
      "step": 9283,
      "training_step_time": 9.902076244354248
    },
    {
      "epoch": 3.541183471679687e-05,
      "model_forward_time": 3.1319477558135986,
      "step": 9283
    },
    {
      "epoch": 3.541183471679687e-05,
      "step": 9283,
      "training_step_time": 10.32055115699768
    },
    {
      "epoch": 3.54156494140625e-05,
      "model_forward_time": 3.151958465576172,
      "step": 9284
    },
    {
      "epoch": 3.54156494140625e-05,
      "step": 9284,
      "training_step_time": 9.925205707550049
    },
    {
      "epoch": 3.54156494140625e-05,
      "model_forward_time": 2.4382259845733643,
      "step": 9284
    },
    {
      "epoch": 3.54156494140625e-05,
      "step": 9284,
      "training_step_time": 7.469955205917358
    },
    {
      "epoch": 3.54156494140625e-05,
      "model_forward_time": 3.132405996322632,
      "step": 9284
    },
    {
      "epoch": 3.54156494140625e-05,
      "step": 9284,
      "training_step_time": 9.91196060180664
    },
    {
      "epoch": 3.54156494140625e-05,
      "model_forward_time": 3.1449217796325684,
      "step": 9284
    },
    {
      "epoch": 3.54156494140625e-05,
      "step": 9284,
      "training_step_time": 10.350499629974365
    },
    {
      "epoch": 3.5419464111328126e-05,
      "model_forward_time": 2.451915979385376,
      "step": 9285
    },
    {
      "dynamics_loss_avg": 0.05240807691588998,
      "epoch": 3.5419464111328126e-05,
      "step": 9285
    },
    {
      "action_loss_avg": 0.01769459233619273,
      "epoch": 3.5419464111328126e-05,
      "step": 9285
    },
    {
      "epoch": 3.5419464111328126e-05,
      "step": 9285,
      "training_step_time": 7.501739740371704
    },
    {
      "epoch": 3.5419464111328126e-05,
      "model_forward_time": 3.421035051345825,
      "step": 9285
    },
    {
      "epoch": 3.5419464111328126e-05,
      "step": 9285,
      "training_step_time": 10.210913181304932
    },
    {
      "epoch": 3.5419464111328126e-05,
      "model_forward_time": 3.1427652835845947,
      "step": 9285
    },
    {
      "epoch": 3.5419464111328126e-05,
      "step": 9285,
      "training_step_time": 9.92782735824585
    },
    {
      "epoch": 3.5419464111328126e-05,
      "model_forward_time": 3.141395092010498,
      "step": 9285
    },
    {
      "epoch": 3.5419464111328126e-05,
      "step": 9285,
      "training_step_time": 10.297990560531616
    },
    {
      "epoch": 3.542327880859375e-05,
      "model_forward_time": 2.4487924575805664,
      "step": 9286
    },
    {
      "epoch": 3.542327880859375e-05,
      "step": 9286,
      "training_step_time": 7.485940217971802
    },
    {
      "epoch": 3.542327880859375e-05,
      "model_forward_time": 3.1340184211730957,
      "step": 9286
    },
    {
      "epoch": 3.542327880859375e-05,
      "step": 9286,
      "training_step_time": 9.906527519226074
    },
    {
      "epoch": 3.542327880859375e-05,
      "model_forward_time": 3.4241890907287598,
      "step": 9286
    },
    {
      "epoch": 3.542327880859375e-05,
      "step": 9286,
      "training_step_time": 10.200453996658325
    },
    {
      "epoch": 3.542327880859375e-05,
      "model_forward_time": 3.1404850482940674,
      "step": 9286
    },
    {
      "epoch": 3.542327880859375e-05,
      "step": 9286,
      "training_step_time": 10.308295011520386
    },
    {
      "epoch": 3.542709350585938e-05,
      "model_forward_time": 3.1571977138519287,
      "step": 9287
    },
    {
      "epoch": 3.542709350585938e-05,
      "step": 9287,
      "training_step_time": 9.945630550384521
    },
    {
      "epoch": 3.542709350585938e-05,
      "model_forward_time": 3.1385412216186523,
      "step": 9287
    },
    {
      "epoch": 3.542709350585938e-05,
      "step": 9287,
      "training_step_time": 9.921497106552124
    },
    {
      "epoch": 3.542709350585938e-05,
      "model_forward_time": 3.1409964561462402,
      "step": 9287
    },
    {
      "dynamics_loss_avg": 0.051910156942904,
      "epoch": 3.542709350585938e-05,
      "step": 9287
    },
    {
      "action_loss_avg": 0.020985221210867166,
      "epoch": 3.542709350585938e-05,
      "step": 9287
    },
    {
      "epoch": 3.542709350585938e-05,
      "step": 9287,
      "training_step_time": 10.22141981124878
    },
    {
      "epoch": 3.542709350585938e-05,
      "model_forward_time": 3.138453722000122,
      "step": 9287
    },
    {
      "epoch": 3.542709350585938e-05,
      "step": 9287,
      "training_step_time": 10.275472164154053
    },
    {
      "epoch": 3.5430908203125e-05,
      "model_forward_time": 3.155311107635498,
      "step": 9288
    },
    {
      "epoch": 3.5430908203125e-05,
      "step": 9288,
      "training_step_time": 9.936274528503418
    },
    {
      "epoch": 3.5430908203125e-05,
      "model_forward_time": 3.133836030960083,
      "step": 9288
    },
    {
      "epoch": 3.5430908203125e-05,
      "step": 9288,
      "training_step_time": 9.907073020935059
    },
    {
      "epoch": 3.5430908203125e-05,
      "model_forward_time": 3.1353957653045654,
      "step": 9288
    },
    {
      "epoch": 3.5430908203125e-05,
      "step": 9288,
      "training_step_time": 9.91688871383667
    },
    {
      "epoch": 3.5430908203125e-05,
      "model_forward_time": 3.133927583694458,
      "step": 9288
    },
    {
      "epoch": 3.5430908203125e-05,
      "step": 9288,
      "training_step_time": 10.31813359260559
    },
    {
      "epoch": 3.5434722900390624e-05,
      "model_forward_time": 3.145785331726074,
      "step": 9289
    },
    {
      "epoch": 3.5434722900390624e-05,
      "step": 9289,
      "training_step_time": 9.92188048362732
    },
    {
      "epoch": 3.5434722900390624e-05,
      "model_forward_time": 3.1342105865478516,
      "step": 9289
    },
    {
      "epoch": 3.5434722900390624e-05,
      "step": 9289,
      "training_step_time": 9.916034936904907
    },
    {
      "epoch": 3.5434722900390624e-05,
      "model_forward_time": 3.1429035663604736,
      "step": 9289
    },
    {
      "epoch": 3.5434722900390624e-05,
      "step": 9289,
      "training_step_time": 9.922287225723267
    },
    {
      "epoch": 3.5434722900390624e-05,
      "model_forward_time": 3.1383559703826904,
      "step": 9289
    },
    {
      "epoch": 3.5434722900390624e-05,
      "step": 9289,
      "training_step_time": 10.321234703063965
    },
    {
      "epoch": 3.543853759765625e-05,
      "grad_norm": 0.3318486213684082,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9290
    },
    {
      "epoch": 3.543853759765625e-05,
      "model_forward_time": 3.4259791374206543,
      "step": 9290
    },
    {
      "dynamics_loss_avg": 0.049974322132766244,
      "epoch": 3.543853759765625e-05,
      "step": 9290
    },
    {
      "action_loss_avg": 0.017697873688302935,
      "epoch": 3.543853759765625e-05,
      "step": 9290
    },
    {
      "epoch": 3.543853759765625e-05,
      "step": 9290,
      "training_step_time": 10.208534240722656
    },
    {
      "epoch": 3.543853759765625e-05,
      "model_forward_time": 3.1383378505706787,
      "step": 9290
    },
    {
      "epoch": 3.543853759765625e-05,
      "step": 9290,
      "training_step_time": 9.916881322860718
    },
    {
      "epoch": 3.543853759765625e-05,
      "model_forward_time": 3.14400053024292,
      "step": 9290
    },
    {
      "epoch": 3.543853759765625e-05,
      "step": 9290,
      "training_step_time": 9.91945195198059
    },
    {
      "epoch": 3.543853759765625e-05,
      "model_forward_time": 3.1374924182891846,
      "step": 9290
    },
    {
      "epoch": 3.543853759765625e-05,
      "step": 9290,
      "training_step_time": 10.313732147216797
    },
    {
      "epoch": 3.5442352294921876e-05,
      "model_forward_time": 3.1472060680389404,
      "step": 9291
    },
    {
      "epoch": 3.5442352294921876e-05,
      "step": 9291,
      "training_step_time": 9.926165342330933
    },
    {
      "epoch": 3.5442352294921876e-05,
      "model_forward_time": 3.447075605392456,
      "step": 9291
    },
    {
      "epoch": 3.5442352294921876e-05,
      "step": 9291,
      "training_step_time": 10.21983289718628
    },
    {
      "epoch": 3.5442352294921876e-05,
      "model_forward_time": 3.1439270973205566,
      "step": 9291
    },
    {
      "epoch": 3.5442352294921876e-05,
      "step": 9291,
      "training_step_time": 9.926169872283936
    },
    {
      "epoch": 3.5442352294921876e-05,
      "model_forward_time": 3.13795804977417,
      "step": 9291
    },
    {
      "epoch": 3.5442352294921876e-05,
      "step": 9291,
      "training_step_time": 10.036665678024292
    },
    {
      "epoch": 3.54461669921875e-05,
      "model_forward_time": 3.1525163650512695,
      "step": 9292
    },
    {
      "epoch": 3.54461669921875e-05,
      "step": 9292,
      "training_step_time": 9.938414335250854
    },
    {
      "epoch": 3.54461669921875e-05,
      "model_forward_time": 2.436448574066162,
      "step": 9292
    },
    {
      "epoch": 3.54461669921875e-05,
      "step": 9292,
      "training_step_time": 7.472784519195557
    },
    {
      "epoch": 3.54461669921875e-05,
      "model_forward_time": 3.1359593868255615,
      "step": 9292
    },
    {
      "dynamics_loss_avg": 0.056597793288528916,
      "epoch": 3.54461669921875e-05,
      "step": 9292
    },
    {
      "action_loss_avg": 0.02335872743278742,
      "epoch": 3.54461669921875e-05,
      "step": 9292
    },
    {
      "epoch": 3.54461669921875e-05,
      "step": 9292,
      "training_step_time": 10.214547634124756
    },
    {
      "epoch": 3.54461669921875e-05,
      "model_forward_time": 3.1397945880889893,
      "step": 9292
    },
    {
      "epoch": 3.54461669921875e-05,
      "step": 9292,
      "training_step_time": 10.332026481628418
    },
    {
      "epoch": 3.544998168945312e-05,
      "model_forward_time": 3.144117832183838,
      "step": 9293
    },
    {
      "epoch": 3.544998168945312e-05,
      "step": 9293,
      "training_step_time": 9.92771601676941
    },
    {
      "epoch": 3.544998168945312e-05,
      "model_forward_time": 3.1360602378845215,
      "step": 9293
    },
    {
      "epoch": 3.544998168945312e-05,
      "step": 9293,
      "training_step_time": 9.911327838897705
    },
    {
      "epoch": 3.544998168945312e-05,
      "model_forward_time": 3.1330244541168213,
      "step": 9293
    },
    {
      "epoch": 3.544998168945312e-05,
      "step": 9293,
      "training_step_time": 9.907017469406128
    },
    {
      "epoch": 3.544998168945312e-05,
      "model_forward_time": 3.1380534172058105,
      "step": 9293
    },
    {
      "epoch": 3.544998168945312e-05,
      "step": 9293,
      "training_step_time": 10.287739276885986
    },
    {
      "epoch": 3.545379638671875e-05,
      "model_forward_time": 3.1472666263580322,
      "step": 9294
    },
    {
      "epoch": 3.545379638671875e-05,
      "step": 9294,
      "training_step_time": 9.927166223526001
    },
    {
      "epoch": 3.545379638671875e-05,
      "model_forward_time": 3.1353094577789307,
      "step": 9294
    },
    {
      "epoch": 3.545379638671875e-05,
      "step": 9294,
      "training_step_time": 9.919594764709473
    },
    {
      "epoch": 3.545379638671875e-05,
      "model_forward_time": 3.1389927864074707,
      "step": 9294
    },
    {
      "epoch": 3.545379638671875e-05,
      "step": 9294,
      "training_step_time": 9.919724464416504
    },
    {
      "epoch": 3.545379638671875e-05,
      "model_forward_time": 3.1404175758361816,
      "step": 9294
    },
    {
      "epoch": 3.545379638671875e-05,
      "step": 9294,
      "training_step_time": 10.343548774719238
    },
    {
      "epoch": 3.5457611083984375e-05,
      "model_forward_time": 3.1477980613708496,
      "step": 9295
    },
    {
      "dynamics_loss_avg": 0.0570307545363903,
      "epoch": 3.5457611083984375e-05,
      "step": 9295
    },
    {
      "action_loss_avg": 0.026160612236708403,
      "epoch": 3.5457611083984375e-05,
      "step": 9295
    },
    {
      "epoch": 3.5457611083984375e-05,
      "step": 9295,
      "training_step_time": 10.22585415840149
    },
    {
      "epoch": 3.5457611083984375e-05,
      "model_forward_time": 3.1418979167938232,
      "step": 9295
    },
    {
      "epoch": 3.5457611083984375e-05,
      "step": 9295,
      "training_step_time": 9.927158117294312
    },
    {
      "epoch": 3.5457611083984375e-05,
      "model_forward_time": 3.1372172832489014,
      "step": 9295
    },
    {
      "epoch": 3.5457611083984375e-05,
      "step": 9295,
      "training_step_time": 9.919211626052856
    },
    {
      "epoch": 3.5457611083984375e-05,
      "model_forward_time": 3.1440446376800537,
      "step": 9295
    },
    {
      "epoch": 3.5457611083984375e-05,
      "step": 9295,
      "training_step_time": 10.023168087005615
    },
    {
      "epoch": 3.546142578125e-05,
      "model_forward_time": 3.151207447052002,
      "step": 9296
    },
    {
      "epoch": 3.546142578125e-05,
      "step": 9296,
      "training_step_time": 9.935975074768066
    },
    {
      "epoch": 3.546142578125e-05,
      "model_forward_time": 3.4711997509002686,
      "step": 9296
    },
    {
      "epoch": 3.546142578125e-05,
      "step": 9296,
      "training_step_time": 10.248917818069458
    },
    {
      "epoch": 3.546142578125e-05,
      "model_forward_time": 3.13132643699646,
      "step": 9296
    },
    {
      "epoch": 3.546142578125e-05,
      "step": 9296,
      "training_step_time": 9.906761407852173
    },
    {
      "epoch": 3.546142578125e-05,
      "model_forward_time": 3.136136054992676,
      "step": 9296
    },
    {
      "epoch": 3.546142578125e-05,
      "step": 9296,
      "training_step_time": 10.015779972076416
    },
    {
      "epoch": 3.546524047851563e-05,
      "model_forward_time": 3.1492927074432373,
      "step": 9297
    },
    {
      "epoch": 3.546524047851563e-05,
      "step": 9297,
      "training_step_time": 9.930955648422241
    },
    {
      "epoch": 3.546524047851563e-05,
      "model_forward_time": 3.1320512294769287,
      "step": 9297
    },
    {
      "epoch": 3.546524047851563e-05,
      "step": 9297,
      "training_step_time": 9.91229796409607
    },
    {
      "epoch": 3.546524047851563e-05,
      "model_forward_time": 3.139899730682373,
      "step": 9297
    },
    {
      "dynamics_loss_avg": 0.05952637102454901,
      "epoch": 3.546524047851563e-05,
      "step": 9297
    },
    {
      "action_loss_avg": 0.019440429657697676,
      "epoch": 3.546524047851563e-05,
      "step": 9297
    },
    {
      "epoch": 3.546524047851563e-05,
      "step": 9297,
      "training_step_time": 10.19912338256836
    },
    {
      "epoch": 3.546524047851563e-05,
      "model_forward_time": 3.1409318447113037,
      "step": 9297
    },
    {
      "epoch": 3.546524047851563e-05,
      "step": 9297,
      "training_step_time": 10.019875764846802
    },
    {
      "epoch": 3.546905517578125e-05,
      "model_forward_time": 3.166646718978882,
      "step": 9298
    },
    {
      "epoch": 3.546905517578125e-05,
      "step": 9298,
      "training_step_time": 9.951358079910278
    },
    {
      "epoch": 3.546905517578125e-05,
      "model_forward_time": 3.1369667053222656,
      "step": 9298
    },
    {
      "epoch": 3.546905517578125e-05,
      "step": 9298,
      "training_step_time": 9.914807558059692
    },
    {
      "epoch": 3.546905517578125e-05,
      "model_forward_time": 3.14548921585083,
      "step": 9298
    },
    {
      "epoch": 3.546905517578125e-05,
      "step": 9298,
      "training_step_time": 9.928813219070435
    },
    {
      "epoch": 3.546905517578125e-05,
      "model_forward_time": 3.4306399822235107,
      "step": 9298
    },
    {
      "epoch": 3.546905517578125e-05,
      "step": 9298,
      "training_step_time": 10.298993587493896
    },
    {
      "epoch": 3.547286987304687e-05,
      "model_forward_time": 3.1614990234375,
      "step": 9299
    },
    {
      "epoch": 3.547286987304687e-05,
      "step": 9299,
      "training_step_time": 9.929365873336792
    },
    {
      "epoch": 3.547286987304687e-05,
      "model_forward_time": 3.1400535106658936,
      "step": 9299
    },
    {
      "epoch": 3.547286987304687e-05,
      "step": 9299,
      "training_step_time": 9.909071683883667
    },
    {
      "epoch": 3.547286987304687e-05,
      "model_forward_time": 3.1462931632995605,
      "step": 9299
    },
    {
      "epoch": 3.547286987304687e-05,
      "step": 9299,
      "training_step_time": 9.91370153427124
    },
    {
      "epoch": 3.547286987304687e-05,
      "model_forward_time": 3.142110824584961,
      "step": 9299
    },
    {
      "epoch": 3.547286987304687e-05,
      "step": 9299,
      "training_step_time": 10.323235511779785
    },
    {
      "epoch": 3.54766845703125e-05,
      "grad_norm": 0.33765339851379395,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9300
    },
    {
      "epoch": 3.54766845703125e-05,
      "model_forward_time": 3.443260908126831,
      "step": 9300
    },
    {
      "dynamics_loss_avg": 0.05807896796613932,
      "epoch": 3.54766845703125e-05,
      "step": 9300
    },
    {
      "action_loss_avg": 0.02275834186002612,
      "epoch": 3.54766845703125e-05,
      "step": 9300
    },
    {
      "epoch": 3.54766845703125e-05,
      "step": 9300,
      "training_step_time": 10.218394994735718
    },
    {
      "epoch": 3.54766845703125e-05,
      "model_forward_time": 3.1450603008270264,
      "step": 9300
    },
    {
      "epoch": 3.54766845703125e-05,
      "step": 9300,
      "training_step_time": 9.920884609222412
    },
    {
      "epoch": 3.54766845703125e-05,
      "model_forward_time": 3.153027057647705,
      "step": 9300
    },
    {
      "epoch": 3.54766845703125e-05,
      "step": 9300,
      "training_step_time": 9.928727149963379
    },
    {
      "epoch": 3.54766845703125e-05,
      "model_forward_time": 3.148214340209961,
      "step": 9300
    },
    {
      "epoch": 3.54766845703125e-05,
      "step": 9300,
      "training_step_time": 10.299702882766724
    },
    {
      "epoch": 3.5480499267578126e-05,
      "model_forward_time": 3.154301166534424,
      "step": 9301
    },
    {
      "epoch": 3.5480499267578126e-05,
      "step": 9301,
      "training_step_time": 9.930678606033325
    },
    {
      "epoch": 3.5480499267578126e-05,
      "model_forward_time": 3.4396491050720215,
      "step": 9301
    },
    {
      "epoch": 3.5480499267578126e-05,
      "step": 9301,
      "training_step_time": 10.209344387054443
    },
    {
      "epoch": 3.5480499267578126e-05,
      "model_forward_time": 3.148233413696289,
      "step": 9301
    },
    {
      "epoch": 3.5480499267578126e-05,
      "step": 9301,
      "training_step_time": 9.915602922439575
    },
    {
      "epoch": 3.5480499267578126e-05,
      "model_forward_time": 3.1409215927124023,
      "step": 9301
    },
    {
      "epoch": 3.5480499267578126e-05,
      "step": 9301,
      "training_step_time": 10.26341724395752
    },
    {
      "epoch": 3.548431396484375e-05,
      "model_forward_time": 3.151620626449585,
      "step": 9302
    },
    {
      "epoch": 3.548431396484375e-05,
      "step": 9302,
      "training_step_time": 9.91599416732788
    },
    {
      "epoch": 3.548431396484375e-05,
      "model_forward_time": 3.140246868133545,
      "step": 9302
    },
    {
      "epoch": 3.548431396484375e-05,
      "step": 9302,
      "training_step_time": 9.905599355697632
    },
    {
      "epoch": 3.548431396484375e-05,
      "model_forward_time": 3.455249071121216,
      "step": 9302
    },
    {
      "dynamics_loss_avg": 0.0614288080483675,
      "epoch": 3.548431396484375e-05,
      "step": 9302
    },
    {
      "action_loss_avg": 0.023151111137121917,
      "epoch": 3.548431396484375e-05,
      "step": 9302
    },
    {
      "epoch": 3.548431396484375e-05,
      "step": 9302,
      "training_step_time": 10.22015380859375
    },
    {
      "epoch": 3.548431396484375e-05,
      "model_forward_time": 3.138455390930176,
      "step": 9302
    },
    {
      "epoch": 3.548431396484375e-05,
      "step": 9302,
      "training_step_time": 10.023144721984863
    },
    {
      "epoch": 3.548812866210938e-05,
      "model_forward_time": 2.4499337673187256,
      "step": 9303
    },
    {
      "epoch": 3.548812866210938e-05,
      "step": 9303,
      "training_step_time": 7.475194692611694
    },
    {
      "epoch": 3.548812866210938e-05,
      "model_forward_time": 2.4382760524749756,
      "step": 9303
    },
    {
      "epoch": 3.548812866210938e-05,
      "step": 9303,
      "training_step_time": 7.4627416133880615
    },
    {
      "epoch": 3.548812866210938e-05,
      "model_forward_time": 3.140977621078491,
      "step": 9303
    },
    {
      "epoch": 3.548812866210938e-05,
      "step": 9303,
      "training_step_time": 9.912504434585571
    },
    {
      "epoch": 3.548812866210938e-05,
      "model_forward_time": 3.1454732418060303,
      "step": 9303
    },
    {
      "epoch": 3.548812866210938e-05,
      "step": 9303,
      "training_step_time": 10.26064920425415
    },
    {
      "epoch": 3.5491943359375e-05,
      "model_forward_time": 3.484969139099121,
      "step": 9304
    },
    {
      "epoch": 3.5491943359375e-05,
      "step": 9304,
      "training_step_time": 10.256829261779785
    },
    {
      "epoch": 3.5491943359375e-05,
      "model_forward_time": 3.1491806507110596,
      "step": 9304
    },
    {
      "epoch": 3.5491943359375e-05,
      "step": 9304,
      "training_step_time": 9.924070835113525
    },
    {
      "epoch": 3.5491943359375e-05,
      "model_forward_time": 3.1465022563934326,
      "step": 9304
    },
    {
      "epoch": 3.5491943359375e-05,
      "step": 9304,
      "training_step_time": 9.9232337474823
    },
    {
      "epoch": 3.5491943359375e-05,
      "model_forward_time": 3.146700382232666,
      "step": 9304
    },
    {
      "epoch": 3.5491943359375e-05,
      "step": 9304,
      "training_step_time": 10.344411373138428
    },
    {
      "epoch": 3.5495758056640624e-05,
      "model_forward_time": 3.433422803878784,
      "step": 9305
    },
    {
      "dynamics_loss_avg": 0.05316530875861645,
      "epoch": 3.5495758056640624e-05,
      "step": 9305
    },
    {
      "action_loss_avg": 0.01818477865308523,
      "epoch": 3.5495758056640624e-05,
      "step": 9305
    },
    {
      "epoch": 3.5495758056640624e-05,
      "step": 9305,
      "training_step_time": 10.214382648468018
    },
    {
      "epoch": 3.5495758056640624e-05,
      "model_forward_time": 3.147460460662842,
      "step": 9305
    },
    {
      "epoch": 3.5495758056640624e-05,
      "step": 9305,
      "training_step_time": 9.920850992202759
    },
    {
      "epoch": 3.5495758056640624e-05,
      "model_forward_time": 2.4405906200408936,
      "step": 9305
    },
    {
      "epoch": 3.5495758056640624e-05,
      "step": 9305,
      "training_step_time": 7.472975730895996
    },
    {
      "epoch": 3.5495758056640624e-05,
      "model_forward_time": 3.1418073177337646,
      "step": 9305
    },
    {
      "epoch": 3.5495758056640624e-05,
      "step": 9305,
      "training_step_time": 10.296769380569458
    },
    {
      "epoch": 3.549957275390625e-05,
      "model_forward_time": 3.1537814140319824,
      "step": 9306
    },
    {
      "epoch": 3.549957275390625e-05,
      "step": 9306,
      "training_step_time": 9.920405864715576
    },
    {
      "epoch": 3.549957275390625e-05,
      "model_forward_time": 3.140582323074341,
      "step": 9306
    },
    {
      "epoch": 3.549957275390625e-05,
      "step": 9306,
      "training_step_time": 9.908821105957031
    },
    {
      "epoch": 3.549957275390625e-05,
      "model_forward_time": 3.399027109146118,
      "step": 9306
    },
    {
      "epoch": 3.549957275390625e-05,
      "step": 9306,
      "training_step_time": 10.166100978851318
    },
    {
      "epoch": 3.549957275390625e-05,
      "model_forward_time": 3.148850917816162,
      "step": 9306
    },
    {
      "epoch": 3.549957275390625e-05,
      "step": 9306,
      "training_step_time": 10.288404941558838
    },
    {
      "epoch": 3.5503387451171876e-05,
      "model_forward_time": 3.163759231567383,
      "step": 9307
    },
    {
      "epoch": 3.5503387451171876e-05,
      "step": 9307,
      "training_step_time": 9.949717044830322
    },
    {
      "epoch": 3.5503387451171876e-05,
      "model_forward_time": 3.1466240882873535,
      "step": 9307
    },
    {
      "epoch": 3.5503387451171876e-05,
      "step": 9307,
      "training_step_time": 9.923824071884155
    },
    {
      "epoch": 3.5503387451171876e-05,
      "model_forward_time": 3.4141597747802734,
      "step": 9307
    },
    {
      "dynamics_loss_avg": 0.06237121149897575,
      "epoch": 3.5503387451171876e-05,
      "step": 9307
    },
    {
      "action_loss_avg": 0.020688051264733075,
      "epoch": 3.5503387451171876e-05,
      "step": 9307
    },
    {
      "epoch": 3.5503387451171876e-05,
      "step": 9307,
      "training_step_time": 10.194447040557861
    },
    {
      "epoch": 3.5503387451171876e-05,
      "model_forward_time": 3.149653196334839,
      "step": 9307
    },
    {
      "epoch": 3.5503387451171876e-05,
      "step": 9307,
      "training_step_time": 10.036524295806885
    },
    {
      "epoch": 3.55072021484375e-05,
      "model_forward_time": 2.4593489170074463,
      "step": 9308
    },
    {
      "epoch": 3.55072021484375e-05,
      "step": 9308,
      "training_step_time": 7.495812654495239
    },
    {
      "epoch": 3.55072021484375e-05,
      "model_forward_time": 2.438847541809082,
      "step": 9308
    },
    {
      "epoch": 3.55072021484375e-05,
      "step": 9308,
      "training_step_time": 7.470564842224121
    },
    {
      "epoch": 3.55072021484375e-05,
      "model_forward_time": 3.145730495452881,
      "step": 9308
    },
    {
      "epoch": 3.55072021484375e-05,
      "step": 9308,
      "training_step_time": 9.914264678955078
    },
    {
      "epoch": 3.55072021484375e-05,
      "model_forward_time": 3.141037940979004,
      "step": 9308
    },
    {
      "epoch": 3.55072021484375e-05,
      "step": 9308,
      "training_step_time": 10.273452520370483
    },
    {
      "epoch": 3.551101684570312e-05,
      "model_forward_time": 3.4429001808166504,
      "step": 9309
    },
    {
      "epoch": 3.551101684570312e-05,
      "step": 9309,
      "training_step_time": 10.215969800949097
    },
    {
      "epoch": 3.551101684570312e-05,
      "model_forward_time": 3.1447629928588867,
      "step": 9309
    },
    {
      "epoch": 3.551101684570312e-05,
      "step": 9309,
      "training_step_time": 9.91575574874878
    },
    {
      "epoch": 3.551101684570312e-05,
      "model_forward_time": 3.1443724632263184,
      "step": 9309
    },
    {
      "epoch": 3.551101684570312e-05,
      "step": 9309,
      "training_step_time": 9.923564672470093
    },
    {
      "epoch": 3.551101684570312e-05,
      "model_forward_time": 3.1472747325897217,
      "step": 9309
    },
    {
      "epoch": 3.551101684570312e-05,
      "step": 9309,
      "training_step_time": 10.326131820678711
    },
    {
      "epoch": 3.551483154296875e-05,
      "grad_norm": 0.3646998405456543,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9310
    },
    {
      "epoch": 3.551483154296875e-05,
      "model_forward_time": 2.457331657409668,
      "step": 9310
    },
    {
      "dynamics_loss_avg": 0.06794756539165973,
      "epoch": 3.551483154296875e-05,
      "step": 9310
    },
    {
      "action_loss_avg": 0.02035491680726409,
      "epoch": 3.551483154296875e-05,
      "step": 9310
    },
    {
      "epoch": 3.551483154296875e-05,
      "step": 9310,
      "training_step_time": 7.495109558105469
    },
    {
      "epoch": 3.551483154296875e-05,
      "model_forward_time": 3.3946592807769775,
      "step": 9310
    },
    {
      "epoch": 3.551483154296875e-05,
      "step": 9310,
      "training_step_time": 10.173791885375977
    },
    {
      "epoch": 3.551483154296875e-05,
      "model_forward_time": 3.1503608226776123,
      "step": 9310
    },
    {
      "epoch": 3.551483154296875e-05,
      "step": 9310,
      "training_step_time": 9.929977416992188
    },
    {
      "epoch": 3.551483154296875e-05,
      "model_forward_time": 3.1460134983062744,
      "step": 9310
    },
    {
      "epoch": 3.551483154296875e-05,
      "step": 9310,
      "training_step_time": 10.031283855438232
    },
    {
      "epoch": 3.5518646240234375e-05,
      "model_forward_time": 3.1590418815612793,
      "step": 9311
    },
    {
      "epoch": 3.5518646240234375e-05,
      "step": 9311,
      "training_step_time": 9.938257694244385
    },
    {
      "epoch": 3.5518646240234375e-05,
      "model_forward_time": 3.147057294845581,
      "step": 9311
    },
    {
      "epoch": 3.5518646240234375e-05,
      "step": 9311,
      "training_step_time": 9.921192646026611
    },
    {
      "epoch": 3.5518646240234375e-05,
      "model_forward_time": 3.4173953533172607,
      "step": 9311
    },
    {
      "epoch": 3.5518646240234375e-05,
      "step": 9311,
      "training_step_time": 10.192209959030151
    },
    {
      "epoch": 3.5518646240234375e-05,
      "model_forward_time": 3.1482465267181396,
      "step": 9311
    },
    {
      "epoch": 3.5518646240234375e-05,
      "step": 9311,
      "training_step_time": 10.312162160873413
    },
    {
      "epoch": 3.55224609375e-05,
      "model_forward_time": 3.163332223892212,
      "step": 9312
    },
    {
      "epoch": 3.55224609375e-05,
      "step": 9312,
      "training_step_time": 9.93883752822876
    },
    {
      "epoch": 3.55224609375e-05,
      "model_forward_time": 3.147252321243286,
      "step": 9312
    },
    {
      "epoch": 3.55224609375e-05,
      "step": 9312,
      "training_step_time": 9.919745922088623
    },
    {
      "epoch": 3.55224609375e-05,
      "model_forward_time": 3.144702434539795,
      "step": 9312
    },
    {
      "dynamics_loss_avg": 0.05223001893609762,
      "epoch": 3.55224609375e-05,
      "step": 9312
    },
    {
      "action_loss_avg": 0.01887009567581117,
      "epoch": 3.55224609375e-05,
      "step": 9312
    },
    {
      "epoch": 3.55224609375e-05,
      "step": 9312,
      "training_step_time": 10.239659547805786
    },
    {
      "epoch": 3.55224609375e-05,
      "model_forward_time": 3.14408802986145,
      "step": 9312
    },
    {
      "epoch": 3.55224609375e-05,
      "step": 9312,
      "training_step_time": 10.252507209777832
    },
    {
      "epoch": 3.552627563476563e-05,
      "model_forward_time": 3.164496898651123,
      "step": 9313
    },
    {
      "epoch": 3.552627563476563e-05,
      "step": 9313,
      "training_step_time": 9.937774896621704
    },
    {
      "epoch": 3.552627563476563e-05,
      "model_forward_time": 2.4367363452911377,
      "step": 9313
    },
    {
      "epoch": 3.552627563476563e-05,
      "step": 9313,
      "training_step_time": 7.468719959259033
    },
    {
      "epoch": 3.552627563476563e-05,
      "model_forward_time": 3.1423017978668213,
      "step": 9313
    },
    {
      "epoch": 3.552627563476563e-05,
      "step": 9313,
      "training_step_time": 9.91707730293274
    },
    {
      "epoch": 3.552627563476563e-05,
      "model_forward_time": 3.148242950439453,
      "step": 9313
    },
    {
      "epoch": 3.552627563476563e-05,
      "step": 9313,
      "training_step_time": 10.260359287261963
    },
    {
      "epoch": 3.553009033203125e-05,
      "model_forward_time": 3.4673149585723877,
      "step": 9314
    },
    {
      "epoch": 3.553009033203125e-05,
      "step": 9314,
      "training_step_time": 10.24487042427063
    },
    {
      "epoch": 3.553009033203125e-05,
      "model_forward_time": 3.1471827030181885,
      "step": 9314
    },
    {
      "epoch": 3.553009033203125e-05,
      "step": 9314,
      "training_step_time": 9.91821551322937
    },
    {
      "epoch": 3.553009033203125e-05,
      "model_forward_time": 3.1427576541900635,
      "step": 9314
    },
    {
      "epoch": 3.553009033203125e-05,
      "step": 9314,
      "training_step_time": 9.915335893630981
    },
    {
      "epoch": 3.553009033203125e-05,
      "model_forward_time": 3.144320487976074,
      "step": 9314
    },
    {
      "epoch": 3.553009033203125e-05,
      "step": 9314,
      "training_step_time": 10.020737171173096
    },
    {
      "epoch": 3.553390502929687e-05,
      "model_forward_time": 3.1589014530181885,
      "step": 9315
    },
    {
      "dynamics_loss_avg": 0.042831188347190616,
      "epoch": 3.553390502929687e-05,
      "step": 9315
    },
    {
      "action_loss_avg": 0.01338541298173368,
      "epoch": 3.553390502929687e-05,
      "step": 9315
    },
    {
      "epoch": 3.553390502929687e-05,
      "step": 9315,
      "training_step_time": 10.208329916000366
    },
    {
      "epoch": 3.553390502929687e-05,
      "model_forward_time": 3.1441328525543213,
      "step": 9315
    },
    {
      "epoch": 3.553390502929687e-05,
      "step": 9315,
      "training_step_time": 9.917264938354492
    },
    {
      "epoch": 3.553390502929687e-05,
      "model_forward_time": 3.1444666385650635,
      "step": 9315
    },
    {
      "epoch": 3.553390502929687e-05,
      "step": 9315,
      "training_step_time": 9.917625427246094
    },
    {
      "epoch": 3.553390502929687e-05,
      "model_forward_time": 3.1462130546569824,
      "step": 9315
    },
    {
      "epoch": 3.553390502929687e-05,
      "step": 9315,
      "training_step_time": 10.306095600128174
    },
    {
      "epoch": 3.55377197265625e-05,
      "model_forward_time": 3.1631343364715576,
      "step": 9316
    },
    {
      "epoch": 3.55377197265625e-05,
      "step": 9316,
      "training_step_time": 9.934803247451782
    },
    {
      "epoch": 3.55377197265625e-05,
      "model_forward_time": 3.423008918762207,
      "step": 9316
    },
    {
      "epoch": 3.55377197265625e-05,
      "step": 9316,
      "training_step_time": 10.192087411880493
    },
    {
      "epoch": 3.55377197265625e-05,
      "model_forward_time": 3.1366798877716064,
      "step": 9316
    },
    {
      "epoch": 3.55377197265625e-05,
      "step": 9316,
      "training_step_time": 9.91911768913269
    },
    {
      "epoch": 3.55377197265625e-05,
      "model_forward_time": 3.1439268589019775,
      "step": 9316
    },
    {
      "epoch": 3.55377197265625e-05,
      "step": 9316,
      "training_step_time": 10.317667484283447
    },
    {
      "epoch": 3.5541534423828126e-05,
      "model_forward_time": 3.1517443656921387,
      "step": 9317
    },
    {
      "epoch": 3.5541534423828126e-05,
      "step": 9317,
      "training_step_time": 9.928249597549438
    },
    {
      "epoch": 3.5541534423828126e-05,
      "model_forward_time": 3.1411261558532715,
      "step": 9317
    },
    {
      "epoch": 3.5541534423828126e-05,
      "step": 9317,
      "training_step_time": 9.916096448898315
    },
    {
      "epoch": 3.5541534423828126e-05,
      "model_forward_time": 3.1352744102478027,
      "step": 9317
    },
    {
      "dynamics_loss_avg": 0.05735910907387733,
      "epoch": 3.5541534423828126e-05,
      "step": 9317
    },
    {
      "action_loss_avg": 0.022708504274487496,
      "epoch": 3.5541534423828126e-05,
      "step": 9317
    },
    {
      "epoch": 3.5541534423828126e-05,
      "step": 9317,
      "training_step_time": 10.218257665634155
    },
    {
      "epoch": 3.5541534423828126e-05,
      "model_forward_time": 3.137178659439087,
      "step": 9317
    },
    {
      "epoch": 3.5541534423828126e-05,
      "step": 9317,
      "training_step_time": 10.25433874130249
    },
    {
      "epoch": 3.554534912109375e-05,
      "model_forward_time": 3.1496386528015137,
      "step": 9318
    },
    {
      "epoch": 3.554534912109375e-05,
      "step": 9318,
      "training_step_time": 9.92050051689148
    },
    {
      "epoch": 3.554534912109375e-05,
      "model_forward_time": 3.1411538124084473,
      "step": 9318
    },
    {
      "epoch": 3.554534912109375e-05,
      "step": 9318,
      "training_step_time": 9.914824724197388
    },
    {
      "epoch": 3.554534912109375e-05,
      "model_forward_time": 3.140888214111328,
      "step": 9318
    },
    {
      "epoch": 3.554534912109375e-05,
      "step": 9318,
      "training_step_time": 9.913069725036621
    },
    {
      "epoch": 3.554534912109375e-05,
      "model_forward_time": 3.4309420585632324,
      "step": 9318
    },
    {
      "epoch": 3.554534912109375e-05,
      "step": 9318,
      "training_step_time": 10.301709175109863
    },
    {
      "epoch": 3.554916381835938e-05,
      "model_forward_time": 3.1508383750915527,
      "step": 9319
    },
    {
      "epoch": 3.554916381835938e-05,
      "step": 9319,
      "training_step_time": 9.928373098373413
    },
    {
      "epoch": 3.554916381835938e-05,
      "model_forward_time": 3.1342415809631348,
      "step": 9319
    },
    {
      "epoch": 3.554916381835938e-05,
      "step": 9319,
      "training_step_time": 9.908195972442627
    },
    {
      "epoch": 3.554916381835938e-05,
      "model_forward_time": 3.143065929412842,
      "step": 9319
    },
    {
      "epoch": 3.554916381835938e-05,
      "step": 9319,
      "training_step_time": 9.922574281692505
    },
    {
      "epoch": 3.554916381835938e-05,
      "model_forward_time": 3.143501043319702,
      "step": 9319
    },
    {
      "epoch": 3.554916381835938e-05,
      "step": 9319,
      "training_step_time": 10.289767980575562
    },
    {
      "epoch": 3.5552978515625e-05,
      "grad_norm": 0.5199810862541199,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 9320
    },
    {
      "epoch": 3.5552978515625e-05,
      "model_forward_time": 3.604349374771118,
      "step": 9320
    },
    {
      "dynamics_loss_avg": 0.04499358590692282,
      "epoch": 3.5552978515625e-05,
      "step": 9320
    },
    {
      "action_loss_avg": 0.01564875771291554,
      "epoch": 3.5552978515625e-05,
      "step": 9320
    },
    {
      "epoch": 3.5552978515625e-05,
      "step": 9320,
      "training_step_time": 10.401885271072388
    },
    {
      "epoch": 3.5552978515625e-05,
      "model_forward_time": 3.1550397872924805,
      "step": 9320
    },
    {
      "epoch": 3.5552978515625e-05,
      "step": 9320,
      "training_step_time": 9.946108102798462
    },
    {
      "epoch": 3.5552978515625e-05,
      "model_forward_time": 3.144989252090454,
      "step": 9320
    },
    {
      "epoch": 3.5552978515625e-05,
      "step": 9320,
      "training_step_time": 9.923242330551147
    },
    {
      "epoch": 3.5552978515625e-05,
      "model_forward_time": 3.1500978469848633,
      "step": 9320
    },
    {
      "epoch": 3.5552978515625e-05,
      "step": 9320,
      "training_step_time": 10.234267473220825
    },
    {
      "epoch": 3.5556793212890624e-05,
      "model_forward_time": 3.1692733764648438,
      "step": 9321
    },
    {
      "epoch": 3.5556793212890624e-05,
      "step": 9321,
      "training_step_time": 9.943071365356445
    },
    {
      "epoch": 3.5556793212890624e-05,
      "model_forward_time": 3.420382499694824,
      "step": 9321
    },
    {
      "epoch": 3.5556793212890624e-05,
      "step": 9321,
      "training_step_time": 10.191433906555176
    },
    {
      "epoch": 3.5556793212890624e-05,
      "model_forward_time": 3.148993492126465,
      "step": 9321
    },
    {
      "epoch": 3.5556793212890624e-05,
      "step": 9321,
      "training_step_time": 9.931460857391357
    },
    {
      "epoch": 3.5556793212890624e-05,
      "model_forward_time": 3.14113450050354,
      "step": 9321
    },
    {
      "epoch": 3.5556793212890624e-05,
      "step": 9321,
      "training_step_time": 10.28915786743164
    },
    {
      "epoch": 3.556060791015625e-05,
      "model_forward_time": 3.158876419067383,
      "step": 9322
    },
    {
      "epoch": 3.556060791015625e-05,
      "step": 9322,
      "training_step_time": 9.93770170211792
    },
    {
      "epoch": 3.556060791015625e-05,
      "model_forward_time": 3.150195598602295,
      "step": 9322
    },
    {
      "epoch": 3.556060791015625e-05,
      "step": 9322,
      "training_step_time": 9.929828405380249
    },
    {
      "epoch": 3.556060791015625e-05,
      "model_forward_time": 3.471855640411377,
      "step": 9322
    },
    {
      "dynamics_loss_avg": 0.038284255750477315,
      "epoch": 3.556060791015625e-05,
      "step": 9322
    },
    {
      "action_loss_avg": 0.013060328736901283,
      "epoch": 3.556060791015625e-05,
      "step": 9322
    },
    {
      "epoch": 3.556060791015625e-05,
      "step": 9322,
      "training_step_time": 10.256084680557251
    },
    {
      "epoch": 3.556060791015625e-05,
      "model_forward_time": 3.1451590061187744,
      "step": 9322
    },
    {
      "epoch": 3.556060791015625e-05,
      "step": 9322,
      "training_step_time": 10.333497524261475
    },
    {
      "epoch": 3.5564422607421877e-05,
      "model_forward_time": 3.1551949977874756,
      "step": 9323
    },
    {
      "epoch": 3.5564422607421877e-05,
      "step": 9323,
      "training_step_time": 9.937066316604614
    },
    {
      "epoch": 3.5564422607421877e-05,
      "model_forward_time": 3.1411187648773193,
      "step": 9323
    },
    {
      "epoch": 3.5564422607421877e-05,
      "step": 9323,
      "training_step_time": 9.92024827003479
    },
    {
      "epoch": 3.5564422607421877e-05,
      "model_forward_time": 3.1426923274993896,
      "step": 9323
    },
    {
      "epoch": 3.5564422607421877e-05,
      "step": 9323,
      "training_step_time": 9.924317359924316
    },
    {
      "epoch": 3.5564422607421877e-05,
      "model_forward_time": 2.445175886154175,
      "step": 9323
    },
    {
      "epoch": 3.5564422607421877e-05,
      "step": 9323,
      "training_step_time": 10.27043628692627
    },
    {
      "epoch": 3.55682373046875e-05,
      "model_forward_time": 2.458543300628662,
      "step": 9324
    },
    {
      "epoch": 3.55682373046875e-05,
      "step": 9324,
      "training_step_time": 7.490415573120117
    },
    {
      "epoch": 3.55682373046875e-05,
      "model_forward_time": 3.1402454376220703,
      "step": 9324
    },
    {
      "epoch": 3.55682373046875e-05,
      "step": 9324,
      "training_step_time": 9.912564277648926
    },
    {
      "epoch": 3.55682373046875e-05,
      "model_forward_time": 3.142512083053589,
      "step": 9324
    },
    {
      "epoch": 3.55682373046875e-05,
      "step": 9324,
      "training_step_time": 9.915738582611084
    },
    {
      "epoch": 3.55682373046875e-05,
      "model_forward_time": 2.435544490814209,
      "step": 9324
    },
    {
      "epoch": 3.55682373046875e-05,
      "step": 9324,
      "training_step_time": 10.269484281539917
    },
    {
      "epoch": 3.557205200195312e-05,
      "model_forward_time": 3.154818296432495,
      "step": 9325
    },
    {
      "dynamics_loss_avg": 0.048554909974336626,
      "epoch": 3.557205200195312e-05,
      "step": 9325
    },
    {
      "action_loss_avg": 0.01572325681336224,
      "epoch": 3.557205200195312e-05,
      "step": 9325
    },
    {
      "epoch": 3.557205200195312e-05,
      "step": 9325,
      "training_step_time": 9.931641340255737
    },
    {
      "epoch": 3.557205200195312e-05,
      "model_forward_time": 2.7545998096466064,
      "step": 9325
    },
    {
      "epoch": 3.557205200195312e-05,
      "step": 9325,
      "training_step_time": 7.78314995765686
    },
    {
      "epoch": 3.557205200195312e-05,
      "model_forward_time": 3.1367850303649902,
      "step": 9325
    },
    {
      "epoch": 3.557205200195312e-05,
      "step": 9325,
      "training_step_time": 9.910215377807617
    },
    {
      "epoch": 3.557205200195312e-05,
      "model_forward_time": 3.1377944946289062,
      "step": 9325
    },
    {
      "epoch": 3.557205200195312e-05,
      "step": 9325,
      "training_step_time": 10.012707233428955
    },
    {
      "epoch": 3.557586669921875e-05,
      "model_forward_time": 2.4893641471862793,
      "step": 9326
    },
    {
      "epoch": 3.557586669921875e-05,
      "step": 9326,
      "training_step_time": 7.5138585567474365
    },
    {
      "epoch": 3.557586669921875e-05,
      "model_forward_time": 3.131507635116577,
      "step": 9326
    },
    {
      "epoch": 3.557586669921875e-05,
      "step": 9326,
      "training_step_time": 9.897140502929688
    },
    {
      "epoch": 3.557586669921875e-05,
      "model_forward_time": 2.7330586910247803,
      "step": 9326
    },
    {
      "epoch": 3.557586669921875e-05,
      "step": 9326,
      "training_step_time": 7.75537109375
    },
    {
      "epoch": 3.557586669921875e-05,
      "model_forward_time": 2.430945634841919,
      "step": 9326
    },
    {
      "epoch": 3.557586669921875e-05,
      "step": 9326,
      "training_step_time": 10.309630155563354
    },
    {
      "epoch": 3.5579681396484375e-05,
      "model_forward_time": 3.1508312225341797,
      "step": 9327
    },
    {
      "epoch": 3.5579681396484375e-05,
      "step": 9327,
      "training_step_time": 9.920676469802856
    },
    {
      "epoch": 3.5579681396484375e-05,
      "model_forward_time": 3.1293158531188965,
      "step": 9327
    },
    {
      "epoch": 3.5579681396484375e-05,
      "step": 9327,
      "training_step_time": 9.897304773330688
    },
    {
      "epoch": 3.5579681396484375e-05,
      "model_forward_time": 3.130133628845215,
      "step": 9327
    },
    {
      "dynamics_loss_avg": 0.049777434952557084,
      "epoch": 3.5579681396484375e-05,
      "step": 9327
    },
    {
      "action_loss_avg": 0.015259986463934182,
      "epoch": 3.5579681396484375e-05,
      "step": 9327
    },
    {
      "epoch": 3.5579681396484375e-05,
      "step": 9327,
      "training_step_time": 9.89440393447876
    },
    {
      "epoch": 3.5579681396484375e-05,
      "model_forward_time": 2.754237174987793,
      "step": 9327
    },
    {
      "epoch": 3.5579681396484375e-05,
      "step": 9327,
      "training_step_time": 10.272573709487915
    },
    {
      "epoch": 3.558349609375e-05,
      "model_forward_time": 2.4476964473724365,
      "step": 9328
    },
    {
      "epoch": 3.558349609375e-05,
      "step": 9328,
      "training_step_time": 7.472357273101807
    },
    {
      "epoch": 3.558349609375e-05,
      "model_forward_time": 3.128535509109497,
      "step": 9328
    },
    {
      "epoch": 3.558349609375e-05,
      "step": 9328,
      "training_step_time": 9.896241903305054
    },
    {
      "epoch": 3.558349609375e-05,
      "model_forward_time": 3.1322665214538574,
      "step": 9328
    },
    {
      "epoch": 3.558349609375e-05,
      "step": 9328,
      "training_step_time": 9.904274225234985
    },
    {
      "epoch": 3.558349609375e-05,
      "model_forward_time": 2.429743766784668,
      "step": 9328
    },
    {
      "epoch": 3.558349609375e-05,
      "step": 9328,
      "training_step_time": 10.252417802810669
    },
    {
      "epoch": 3.558731079101563e-05,
      "model_forward_time": 3.435109853744507,
      "step": 9329
    },
    {
      "epoch": 3.558731079101563e-05,
      "step": 9329,
      "training_step_time": 10.213783502578735
    },
    {
      "epoch": 3.558731079101563e-05,
      "model_forward_time": 3.1353213787078857,
      "step": 9329
    },
    {
      "epoch": 3.558731079101563e-05,
      "step": 9329,
      "training_step_time": 9.904886722564697
    },
    {
      "epoch": 3.558731079101563e-05,
      "model_forward_time": 2.4337732791900635,
      "step": 9329
    },
    {
      "epoch": 3.558731079101563e-05,
      "step": 9329,
      "training_step_time": 7.464230537414551
    },
    {
      "epoch": 3.558731079101563e-05,
      "model_forward_time": 3.1330182552337646,
      "step": 9329
    },
    {
      "epoch": 3.558731079101563e-05,
      "step": 9329,
      "training_step_time": 10.319581747055054
    },
    {
      "epoch": 3.559112548828125e-05,
      "grad_norm": 0.5638971328735352,
      "learning_rate": 1e-05,
      "loss": 0.077,
      "step": 9330
    },
    {
      "epoch": 3.559112548828125e-05,
      "model_forward_time": 3.157630205154419,
      "step": 9330
    },
    {
      "dynamics_loss_avg": 0.05991159938275814,
      "epoch": 3.559112548828125e-05,
      "step": 9330
    },
    {
      "action_loss_avg": 0.02680779965594411,
      "epoch": 3.559112548828125e-05,
      "step": 9330
    },
    {
      "epoch": 3.559112548828125e-05,
      "step": 9330,
      "training_step_time": 9.934616804122925
    },
    {
      "epoch": 3.559112548828125e-05,
      "model_forward_time": 3.432860851287842,
      "step": 9330
    },
    {
      "epoch": 3.559112548828125e-05,
      "step": 9330,
      "training_step_time": 10.203388452529907
    },
    {
      "epoch": 3.559112548828125e-05,
      "model_forward_time": 3.1389288902282715,
      "step": 9330
    },
    {
      "epoch": 3.559112548828125e-05,
      "step": 9330,
      "training_step_time": 9.909482717514038
    },
    {
      "epoch": 3.559112548828125e-05,
      "model_forward_time": 3.136096477508545,
      "step": 9330
    },
    {
      "epoch": 3.559112548828125e-05,
      "step": 9330,
      "training_step_time": 10.250548124313354
    },
    {
      "epoch": 3.559494018554687e-05,
      "model_forward_time": 2.4484992027282715,
      "step": 9331
    },
    {
      "epoch": 3.559494018554687e-05,
      "step": 9331,
      "training_step_time": 7.480087757110596
    },
    {
      "epoch": 3.559494018554687e-05,
      "model_forward_time": 3.130852699279785,
      "step": 9331
    },
    {
      "epoch": 3.559494018554687e-05,
      "step": 9331,
      "training_step_time": 9.902960062026978
    },
    {
      "epoch": 3.559494018554687e-05,
      "model_forward_time": 2.431244373321533,
      "step": 9331
    },
    {
      "epoch": 3.559494018554687e-05,
      "step": 9331,
      "training_step_time": 7.770994424819946
    },
    {
      "epoch": 3.559494018554687e-05,
      "model_forward_time": 3.1352007389068604,
      "step": 9331
    },
    {
      "epoch": 3.559494018554687e-05,
      "step": 9331,
      "training_step_time": 10.286879301071167
    },
    {
      "epoch": 3.55987548828125e-05,
      "model_forward_time": 3.1385421752929688,
      "step": 9332
    },
    {
      "epoch": 3.55987548828125e-05,
      "step": 9332,
      "training_step_time": 9.917904615402222
    },
    {
      "epoch": 3.55987548828125e-05,
      "model_forward_time": 3.135523557662964,
      "step": 9332
    },
    {
      "epoch": 3.55987548828125e-05,
      "step": 9332,
      "training_step_time": 9.906107664108276
    },
    {
      "epoch": 3.55987548828125e-05,
      "model_forward_time": 3.131756067276001,
      "step": 9332
    },
    {
      "dynamics_loss_avg": 0.059791759587824345,
      "epoch": 3.55987548828125e-05,
      "step": 9332
    },
    {
      "action_loss_avg": 0.025644179526716472,
      "epoch": 3.55987548828125e-05,
      "step": 9332
    },
    {
      "epoch": 3.55987548828125e-05,
      "step": 9332,
      "training_step_time": 9.906104326248169
    },
    {
      "epoch": 3.55987548828125e-05,
      "model_forward_time": 3.401104688644409,
      "step": 9332
    },
    {
      "epoch": 3.55987548828125e-05,
      "step": 9332,
      "training_step_time": 10.296430826187134
    },
    {
      "epoch": 3.5602569580078126e-05,
      "model_forward_time": 3.1448700428009033,
      "step": 9333
    },
    {
      "epoch": 3.5602569580078126e-05,
      "step": 9333,
      "training_step_time": 9.92143988609314
    },
    {
      "epoch": 3.5602569580078126e-05,
      "model_forward_time": 3.1319680213928223,
      "step": 9333
    },
    {
      "epoch": 3.5602569580078126e-05,
      "step": 9333,
      "training_step_time": 9.901643514633179
    },
    {
      "epoch": 3.5602569580078126e-05,
      "model_forward_time": 3.1318323612213135,
      "step": 9333
    },
    {
      "epoch": 3.5602569580078126e-05,
      "step": 9333,
      "training_step_time": 9.900486707687378
    },
    {
      "epoch": 3.5602569580078126e-05,
      "model_forward_time": 2.4291296005249023,
      "step": 9333
    },
    {
      "epoch": 3.5602569580078126e-05,
      "step": 9333,
      "training_step_time": 10.281295537948608
    },
    {
      "epoch": 3.560638427734375e-05,
      "model_forward_time": 3.1577768325805664,
      "step": 9334
    },
    {
      "epoch": 3.560638427734375e-05,
      "step": 9334,
      "training_step_time": 10.251346826553345
    },
    {
      "epoch": 3.560638427734375e-05,
      "model_forward_time": 3.1466643810272217,
      "step": 9334
    },
    {
      "epoch": 3.560638427734375e-05,
      "step": 9334,
      "training_step_time": 9.923725605010986
    },
    {
      "epoch": 3.560638427734375e-05,
      "model_forward_time": 3.143446922302246,
      "step": 9334
    },
    {
      "epoch": 3.560638427734375e-05,
      "step": 9334,
      "training_step_time": 9.913417339324951
    },
    {
      "epoch": 3.560638427734375e-05,
      "model_forward_time": 3.1498661041259766,
      "step": 9334
    },
    {
      "epoch": 3.560638427734375e-05,
      "step": 9334,
      "training_step_time": 10.268885850906372
    },
    {
      "epoch": 3.561019897460938e-05,
      "model_forward_time": 3.157926082611084,
      "step": 9335
    },
    {
      "dynamics_loss_avg": 0.04665262270718813,
      "epoch": 3.561019897460938e-05,
      "step": 9335
    },
    {
      "action_loss_avg": 0.016722759045660496,
      "epoch": 3.561019897460938e-05,
      "step": 9335
    },
    {
      "epoch": 3.561019897460938e-05,
      "step": 9335,
      "training_step_time": 9.939166069030762
    },
    {
      "epoch": 3.561019897460938e-05,
      "model_forward_time": 3.143857479095459,
      "step": 9335
    },
    {
      "epoch": 3.561019897460938e-05,
      "step": 9335,
      "training_step_time": 10.194980144500732
    },
    {
      "epoch": 3.561019897460938e-05,
      "model_forward_time": 3.1478404998779297,
      "step": 9335
    },
    {
      "epoch": 3.561019897460938e-05,
      "step": 9335,
      "training_step_time": 9.920045614242554
    },
    {
      "epoch": 3.561019897460938e-05,
      "model_forward_time": 3.1505167484283447,
      "step": 9335
    },
    {
      "epoch": 3.561019897460938e-05,
      "step": 9335,
      "training_step_time": 10.024643421173096
    },
    {
      "epoch": 3.5614013671875e-05,
      "model_forward_time": 3.1674206256866455,
      "step": 9336
    },
    {
      "epoch": 3.5614013671875e-05,
      "step": 9336,
      "training_step_time": 9.948876142501831
    },
    {
      "epoch": 3.5614013671875e-05,
      "model_forward_time": 3.1473212242126465,
      "step": 9336
    },
    {
      "epoch": 3.5614013671875e-05,
      "step": 9336,
      "training_step_time": 9.93014907836914
    },
    {
      "epoch": 3.5614013671875e-05,
      "model_forward_time": 3.4537699222564697,
      "step": 9336
    },
    {
      "epoch": 3.5614013671875e-05,
      "step": 9336,
      "training_step_time": 10.22913384437561
    },
    {
      "epoch": 3.5614013671875e-05,
      "model_forward_time": 2.444427967071533,
      "step": 9336
    },
    {
      "epoch": 3.5614013671875e-05,
      "step": 9336,
      "training_step_time": 10.305354356765747
    },
    {
      "epoch": 3.5617828369140624e-05,
      "model_forward_time": 3.1671230792999268,
      "step": 9337
    },
    {
      "epoch": 3.5617828369140624e-05,
      "step": 9337,
      "training_step_time": 9.939278841018677
    },
    {
      "epoch": 3.5617828369140624e-05,
      "model_forward_time": 3.1445999145507812,
      "step": 9337
    },
    {
      "epoch": 3.5617828369140624e-05,
      "step": 9337,
      "training_step_time": 9.91462230682373
    },
    {
      "epoch": 3.5617828369140624e-05,
      "model_forward_time": 2.439427614212036,
      "step": 9337
    },
    {
      "dynamics_loss_avg": 0.06706124469637871,
      "epoch": 3.5617828369140624e-05,
      "step": 9337
    },
    {
      "action_loss_avg": 0.030422660894691943,
      "epoch": 3.5617828369140624e-05,
      "step": 9337
    },
    {
      "epoch": 3.5617828369140624e-05,
      "step": 9337,
      "training_step_time": 7.4719719886779785
    },
    {
      "epoch": 3.5617828369140624e-05,
      "model_forward_time": 3.1417975425720215,
      "step": 9337
    },
    {
      "epoch": 3.5617828369140624e-05,
      "step": 9337,
      "training_step_time": 10.011892080307007
    },
    {
      "epoch": 3.562164306640625e-05,
      "model_forward_time": 3.467535972595215,
      "step": 9338
    },
    {
      "epoch": 3.562164306640625e-05,
      "step": 9338,
      "training_step_time": 10.236464738845825
    },
    {
      "epoch": 3.562164306640625e-05,
      "model_forward_time": 3.1381189823150635,
      "step": 9338
    },
    {
      "epoch": 3.562164306640625e-05,
      "step": 9338,
      "training_step_time": 9.90323257446289
    },
    {
      "epoch": 3.562164306640625e-05,
      "model_forward_time": 3.143958330154419,
      "step": 9338
    },
    {
      "epoch": 3.562164306640625e-05,
      "step": 9338,
      "training_step_time": 9.916821718215942
    },
    {
      "epoch": 3.562164306640625e-05,
      "model_forward_time": 2.441735029220581,
      "step": 9338
    },
    {
      "epoch": 3.562164306640625e-05,
      "step": 9338,
      "training_step_time": 10.269815683364868
    },
    {
      "epoch": 3.5625457763671877e-05,
      "model_forward_time": 3.151625871658325,
      "step": 9339
    },
    {
      "epoch": 3.5625457763671877e-05,
      "step": 9339,
      "training_step_time": 9.930720090866089
    },
    {
      "epoch": 3.5625457763671877e-05,
      "model_forward_time": 3.3970775604248047,
      "step": 9339
    },
    {
      "epoch": 3.5625457763671877e-05,
      "step": 9339,
      "training_step_time": 10.167276620864868
    },
    {
      "epoch": 3.5625457763671877e-05,
      "model_forward_time": 3.1486763954162598,
      "step": 9339
    },
    {
      "epoch": 3.5625457763671877e-05,
      "step": 9339,
      "training_step_time": 9.918160915374756
    },
    {
      "epoch": 3.5625457763671877e-05,
      "model_forward_time": 3.1477675437927246,
      "step": 9339
    },
    {
      "epoch": 3.5625457763671877e-05,
      "step": 9339,
      "training_step_time": 10.321523666381836
    },
    {
      "epoch": 3.56292724609375e-05,
      "grad_norm": 0.3348400294780731,
      "learning_rate": 1e-05,
      "loss": 0.0791,
      "step": 9340
    },
    {
      "epoch": 3.56292724609375e-05,
      "model_forward_time": 3.161736249923706,
      "step": 9340
    },
    {
      "dynamics_loss_avg": 0.07157087177038193,
      "epoch": 3.56292724609375e-05,
      "step": 9340
    },
    {
      "action_loss_avg": 0.02908522021025419,
      "epoch": 3.56292724609375e-05,
      "step": 9340
    },
    {
      "epoch": 3.56292724609375e-05,
      "step": 9340,
      "training_step_time": 9.938894748687744
    },
    {
      "epoch": 3.56292724609375e-05,
      "model_forward_time": 3.1484060287475586,
      "step": 9340
    },
    {
      "epoch": 3.56292724609375e-05,
      "step": 9340,
      "training_step_time": 9.92253589630127
    },
    {
      "epoch": 3.56292724609375e-05,
      "model_forward_time": 3.4242658615112305,
      "step": 9340
    },
    {
      "epoch": 3.56292724609375e-05,
      "step": 9340,
      "training_step_time": 10.199559450149536
    },
    {
      "epoch": 3.56292724609375e-05,
      "model_forward_time": 3.147548198699951,
      "step": 9340
    },
    {
      "epoch": 3.56292724609375e-05,
      "step": 9340,
      "training_step_time": 10.022969961166382
    },
    {
      "epoch": 3.563308715820312e-05,
      "model_forward_time": 2.4591622352600098,
      "step": 9341
    },
    {
      "epoch": 3.563308715820312e-05,
      "step": 9341,
      "training_step_time": 7.493035793304443
    },
    {
      "epoch": 3.563308715820312e-05,
      "model_forward_time": 3.142094373703003,
      "step": 9341
    },
    {
      "epoch": 3.563308715820312e-05,
      "step": 9341,
      "training_step_time": 9.91148829460144
    },
    {
      "epoch": 3.563308715820312e-05,
      "model_forward_time": 3.4366142749786377,
      "step": 9341
    },
    {
      "epoch": 3.563308715820312e-05,
      "step": 9341,
      "training_step_time": 10.204918622970581
    },
    {
      "epoch": 3.563308715820312e-05,
      "model_forward_time": 3.1481850147247314,
      "step": 9341
    },
    {
      "epoch": 3.563308715820312e-05,
      "step": 9341,
      "training_step_time": 10.306585788726807
    },
    {
      "epoch": 3.563690185546875e-05,
      "model_forward_time": 3.1609113216400146,
      "step": 9342
    },
    {
      "epoch": 3.563690185546875e-05,
      "step": 9342,
      "training_step_time": 9.940166473388672
    },
    {
      "epoch": 3.563690185546875e-05,
      "model_forward_time": 3.1547658443450928,
      "step": 9342
    },
    {
      "epoch": 3.563690185546875e-05,
      "step": 9342,
      "training_step_time": 9.930752754211426
    },
    {
      "epoch": 3.563690185546875e-05,
      "model_forward_time": 3.1450629234313965,
      "step": 9342
    },
    {
      "dynamics_loss_avg": 0.05170746855437756,
      "epoch": 3.563690185546875e-05,
      "step": 9342
    },
    {
      "action_loss_avg": 0.017750630807131528,
      "epoch": 3.563690185546875e-05,
      "step": 9342
    },
    {
      "epoch": 3.563690185546875e-05,
      "step": 9342,
      "training_step_time": 9.92553162574768
    },
    {
      "epoch": 3.563690185546875e-05,
      "model_forward_time": 3.1461472511291504,
      "step": 9342
    },
    {
      "epoch": 3.563690185546875e-05,
      "step": 9342,
      "training_step_time": 10.023635864257812
    },
    {
      "epoch": 3.5640716552734375e-05,
      "model_forward_time": 3.4723618030548096,
      "step": 9343
    },
    {
      "epoch": 3.5640716552734375e-05,
      "step": 9343,
      "training_step_time": 10.247848987579346
    },
    {
      "epoch": 3.5640716552734375e-05,
      "model_forward_time": 3.1491105556488037,
      "step": 9343
    },
    {
      "epoch": 3.5640716552734375e-05,
      "step": 9343,
      "training_step_time": 9.924782037734985
    },
    {
      "epoch": 3.5640716552734375e-05,
      "model_forward_time": 3.1447038650512695,
      "step": 9343
    },
    {
      "epoch": 3.5640716552734375e-05,
      "step": 9343,
      "training_step_time": 9.918149471282959
    },
    {
      "epoch": 3.5640716552734375e-05,
      "model_forward_time": 3.145312786102295,
      "step": 9343
    },
    {
      "epoch": 3.5640716552734375e-05,
      "step": 9343,
      "training_step_time": 10.30854320526123
    },
    {
      "epoch": 3.564453125e-05,
      "model_forward_time": 3.4247851371765137,
      "step": 9344
    },
    {
      "epoch": 3.564453125e-05,
      "step": 9344,
      "training_step_time": 10.198370695114136
    },
    {
      "epoch": 3.564453125e-05,
      "model_forward_time": 3.149167060852051,
      "step": 9344
    },
    {
      "epoch": 3.564453125e-05,
      "step": 9344,
      "training_step_time": 9.92148208618164
    },
    {
      "epoch": 3.564453125e-05,
      "model_forward_time": 3.1442787647247314,
      "step": 9344
    },
    {
      "epoch": 3.564453125e-05,
      "step": 9344,
      "training_step_time": 9.916051626205444
    },
    {
      "epoch": 3.564453125e-05,
      "model_forward_time": 3.1495020389556885,
      "step": 9344
    },
    {
      "epoch": 3.564453125e-05,
      "step": 9344,
      "training_step_time": 10.021025657653809
    },
    {
      "epoch": 3.564834594726563e-05,
      "model_forward_time": 3.1601643562316895,
      "step": 9345
    },
    {
      "dynamics_loss_avg": 0.03681304585188627,
      "epoch": 3.564834594726563e-05,
      "step": 9345
    },
    {
      "action_loss_avg": 0.019233420351520182,
      "epoch": 3.564834594726563e-05,
      "step": 9345
    },
    {
      "epoch": 3.564834594726563e-05,
      "step": 9345,
      "training_step_time": 9.934519052505493
    },
    {
      "epoch": 3.564834594726563e-05,
      "model_forward_time": 3.1418447494506836,
      "step": 9345
    },
    {
      "epoch": 3.564834594726563e-05,
      "step": 9345,
      "training_step_time": 10.184070825576782
    },
    {
      "epoch": 3.564834594726563e-05,
      "model_forward_time": 3.1434073448181152,
      "step": 9345
    },
    {
      "epoch": 3.564834594726563e-05,
      "step": 9345,
      "training_step_time": 9.910505771636963
    },
    {
      "epoch": 3.564834594726563e-05,
      "model_forward_time": 3.144514560699463,
      "step": 9345
    },
    {
      "epoch": 3.564834594726563e-05,
      "step": 9345,
      "training_step_time": 10.2518789768219
    },
    {
      "epoch": 3.565216064453125e-05,
      "model_forward_time": 3.158186435699463,
      "step": 9346
    },
    {
      "epoch": 3.565216064453125e-05,
      "step": 9346,
      "training_step_time": 9.93141222000122
    },
    {
      "epoch": 3.565216064453125e-05,
      "model_forward_time": 3.142897605895996,
      "step": 9346
    },
    {
      "epoch": 3.565216064453125e-05,
      "step": 9346,
      "training_step_time": 9.90580677986145
    },
    {
      "epoch": 3.565216064453125e-05,
      "model_forward_time": 2.4310812950134277,
      "step": 9346
    },
    {
      "epoch": 3.565216064453125e-05,
      "step": 9346,
      "training_step_time": 7.723746299743652
    },
    {
      "epoch": 3.565216064453125e-05,
      "model_forward_time": 3.1332013607025146,
      "step": 9346
    },
    {
      "epoch": 3.565216064453125e-05,
      "step": 9346,
      "training_step_time": 10.331295251846313
    },
    {
      "epoch": 3.565597534179687e-05,
      "model_forward_time": 3.1571309566497803,
      "step": 9347
    },
    {
      "epoch": 3.565597534179687e-05,
      "step": 9347,
      "training_step_time": 9.925421237945557
    },
    {
      "epoch": 3.565597534179687e-05,
      "model_forward_time": 3.1339111328125,
      "step": 9347
    },
    {
      "epoch": 3.565597534179687e-05,
      "step": 9347,
      "training_step_time": 9.902806758880615
    },
    {
      "epoch": 3.565597534179687e-05,
      "model_forward_time": 3.136791229248047,
      "step": 9347
    },
    {
      "dynamics_loss_avg": 0.04705222696065903,
      "epoch": 3.565597534179687e-05,
      "step": 9347
    },
    {
      "action_loss_avg": 0.017344062542542817,
      "epoch": 3.565597534179687e-05,
      "step": 9347
    },
    {
      "epoch": 3.565597534179687e-05,
      "step": 9347,
      "training_step_time": 9.90589952468872
    },
    {
      "epoch": 3.565597534179687e-05,
      "model_forward_time": 2.434486150741577,
      "step": 9347
    },
    {
      "epoch": 3.565597534179687e-05,
      "step": 9347,
      "training_step_time": 10.309830904006958
    },
    {
      "epoch": 3.56597900390625e-05,
      "model_forward_time": 3.1520440578460693,
      "step": 9348
    },
    {
      "epoch": 3.56597900390625e-05,
      "step": 9348,
      "training_step_time": 9.920232057571411
    },
    {
      "epoch": 3.56597900390625e-05,
      "model_forward_time": 3.129141092300415,
      "step": 9348
    },
    {
      "epoch": 3.56597900390625e-05,
      "step": 9348,
      "training_step_time": 9.89788007736206
    },
    {
      "epoch": 3.56597900390625e-05,
      "model_forward_time": 3.134338140487671,
      "step": 9348
    },
    {
      "epoch": 3.56597900390625e-05,
      "step": 9348,
      "training_step_time": 9.903505802154541
    },
    {
      "epoch": 3.56597900390625e-05,
      "model_forward_time": 3.132413148880005,
      "step": 9348
    },
    {
      "epoch": 3.56597900390625e-05,
      "step": 9348,
      "training_step_time": 10.313164234161377
    },
    {
      "epoch": 3.5663604736328126e-05,
      "model_forward_time": 3.141392469406128,
      "step": 9349
    },
    {
      "epoch": 3.5663604736328126e-05,
      "step": 9349,
      "training_step_time": 10.193583011627197
    },
    {
      "epoch": 3.5663604736328126e-05,
      "model_forward_time": 3.1316306591033936,
      "step": 9349
    },
    {
      "epoch": 3.5663604736328126e-05,
      "step": 9349,
      "training_step_time": 9.89755129814148
    },
    {
      "epoch": 3.5663604736328126e-05,
      "model_forward_time": 3.1329402923583984,
      "step": 9349
    },
    {
      "epoch": 3.5663604736328126e-05,
      "step": 9349,
      "training_step_time": 9.899881362915039
    },
    {
      "epoch": 3.5663604736328126e-05,
      "model_forward_time": 3.1336467266082764,
      "step": 9349
    },
    {
      "epoch": 3.5663604736328126e-05,
      "step": 9349,
      "training_step_time": 10.271906614303589
    },
    {
      "epoch": 3.566741943359375e-05,
      "grad_norm": 0.3087809085845947,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 9350
    },
    {
      "epoch": 3.566741943359375e-05,
      "model_forward_time": 3.1485068798065186,
      "step": 9350
    },
    {
      "dynamics_loss_avg": 0.07018826678395271,
      "epoch": 3.566741943359375e-05,
      "step": 9350
    },
    {
      "action_loss_avg": 0.021829572785645724,
      "epoch": 3.566741943359375e-05,
      "step": 9350
    },
    {
      "epoch": 3.566741943359375e-05,
      "step": 9350,
      "training_step_time": 9.924275875091553
    },
    {
      "epoch": 3.566741943359375e-05,
      "model_forward_time": 2.430485963821411,
      "step": 9350
    },
    {
      "epoch": 3.566741943359375e-05,
      "step": 9350,
      "training_step_time": 7.732407093048096
    },
    {
      "epoch": 3.566741943359375e-05,
      "model_forward_time": 3.1348254680633545,
      "step": 9350
    },
    {
      "epoch": 3.566741943359375e-05,
      "step": 9350,
      "training_step_time": 9.907989025115967
    },
    {
      "epoch": 3.566741943359375e-05,
      "model_forward_time": 3.134199380874634,
      "step": 9350
    },
    {
      "epoch": 3.566741943359375e-05,
      "step": 9350,
      "training_step_time": 10.271687269210815
    },
    {
      "epoch": 3.567123413085938e-05,
      "model_forward_time": 2.446178674697876,
      "step": 9351
    },
    {
      "epoch": 3.567123413085938e-05,
      "step": 9351,
      "training_step_time": 7.4739251136779785
    },
    {
      "epoch": 3.567123413085938e-05,
      "model_forward_time": 3.136345386505127,
      "step": 9351
    },
    {
      "epoch": 3.567123413085938e-05,
      "step": 9351,
      "training_step_time": 9.914148569107056
    },
    {
      "epoch": 3.567123413085938e-05,
      "model_forward_time": 3.1374340057373047,
      "step": 9351
    },
    {
      "epoch": 3.567123413085938e-05,
      "step": 9351,
      "training_step_time": 9.910109996795654
    },
    {
      "epoch": 3.567123413085938e-05,
      "model_forward_time": 3.3839683532714844,
      "step": 9351
    },
    {
      "epoch": 3.567123413085938e-05,
      "step": 9351,
      "training_step_time": 10.320096254348755
    },
    {
      "epoch": 3.5675048828125e-05,
      "model_forward_time": 3.145538330078125,
      "step": 9352
    },
    {
      "epoch": 3.5675048828125e-05,
      "step": 9352,
      "training_step_time": 9.936949968338013
    },
    {
      "epoch": 3.5675048828125e-05,
      "model_forward_time": 3.1368753910064697,
      "step": 9352
    },
    {
      "epoch": 3.5675048828125e-05,
      "step": 9352,
      "training_step_time": 9.913380861282349
    },
    {
      "epoch": 3.5675048828125e-05,
      "model_forward_time": 3.139300584793091,
      "step": 9352
    },
    {
      "dynamics_loss_avg": 0.049766024388372895,
      "epoch": 3.5675048828125e-05,
      "step": 9352
    },
    {
      "action_loss_avg": 0.023006861610338093,
      "epoch": 3.5675048828125e-05,
      "step": 9352
    },
    {
      "epoch": 3.5675048828125e-05,
      "step": 9352,
      "training_step_time": 9.919275999069214
    },
    {
      "epoch": 3.5675048828125e-05,
      "model_forward_time": 2.434725046157837,
      "step": 9352
    },
    {
      "epoch": 3.5675048828125e-05,
      "step": 9352,
      "training_step_time": 10.277912616729736
    },
    {
      "epoch": 3.5678863525390624e-05,
      "model_forward_time": 3.153488874435425,
      "step": 9353
    },
    {
      "epoch": 3.5678863525390624e-05,
      "step": 9353,
      "training_step_time": 9.927762508392334
    },
    {
      "epoch": 3.5678863525390624e-05,
      "model_forward_time": 3.140895128250122,
      "step": 9353
    },
    {
      "epoch": 3.5678863525390624e-05,
      "step": 9353,
      "training_step_time": 9.916788816452026
    },
    {
      "epoch": 3.5678863525390624e-05,
      "model_forward_time": 2.435512065887451,
      "step": 9353
    },
    {
      "epoch": 3.5678863525390624e-05,
      "step": 9353,
      "training_step_time": 7.467860221862793
    },
    {
      "epoch": 3.5678863525390624e-05,
      "model_forward_time": 2.4329445362091064,
      "step": 9353
    },
    {
      "epoch": 3.5678863525390624e-05,
      "step": 9353,
      "training_step_time": 10.286056518554688
    },
    {
      "epoch": 3.568267822265625e-05,
      "model_forward_time": 3.1447830200195312,
      "step": 9354
    },
    {
      "epoch": 3.568267822265625e-05,
      "step": 9354,
      "training_step_time": 9.918945550918579
    },
    {
      "epoch": 3.568267822265625e-05,
      "model_forward_time": 3.4396917819976807,
      "step": 9354
    },
    {
      "epoch": 3.568267822265625e-05,
      "step": 9354,
      "training_step_time": 10.206971406936646
    },
    {
      "epoch": 3.568267822265625e-05,
      "model_forward_time": 3.1326911449432373,
      "step": 9354
    },
    {
      "epoch": 3.568267822265625e-05,
      "step": 9354,
      "training_step_time": 9.903340101242065
    },
    {
      "epoch": 3.568267822265625e-05,
      "model_forward_time": 3.1363255977630615,
      "step": 9354
    },
    {
      "epoch": 3.568267822265625e-05,
      "step": 9354,
      "training_step_time": 10.01780390739441
    },
    {
      "epoch": 3.568649291992188e-05,
      "model_forward_time": 3.14285945892334,
      "step": 9355
    },
    {
      "dynamics_loss_avg": 0.053557426668703556,
      "epoch": 3.568649291992188e-05,
      "step": 9355
    },
    {
      "action_loss_avg": 0.017430614260956646,
      "epoch": 3.568649291992188e-05,
      "step": 9355
    },
    {
      "epoch": 3.568649291992188e-05,
      "step": 9355,
      "training_step_time": 9.912100315093994
    },
    {
      "epoch": 3.568649291992188e-05,
      "model_forward_time": 3.126953363418579,
      "step": 9355
    },
    {
      "epoch": 3.568649291992188e-05,
      "step": 9355,
      "training_step_time": 9.890652418136597
    },
    {
      "epoch": 3.568649291992188e-05,
      "model_forward_time": 3.374479293823242,
      "step": 9355
    },
    {
      "epoch": 3.568649291992188e-05,
      "step": 9355,
      "training_step_time": 10.136042356491089
    },
    {
      "epoch": 3.568649291992188e-05,
      "model_forward_time": 3.133044481277466,
      "step": 9355
    },
    {
      "epoch": 3.568649291992188e-05,
      "step": 9355,
      "training_step_time": 10.280967235565186
    },
    {
      "epoch": 3.56903076171875e-05,
      "model_forward_time": 3.1434824466705322,
      "step": 9356
    },
    {
      "epoch": 3.56903076171875e-05,
      "step": 9356,
      "training_step_time": 9.923884630203247
    },
    {
      "epoch": 3.56903076171875e-05,
      "model_forward_time": 3.131990432739258,
      "step": 9356
    },
    {
      "epoch": 3.56903076171875e-05,
      "step": 9356,
      "training_step_time": 9.909030675888062
    },
    {
      "epoch": 3.56903076171875e-05,
      "model_forward_time": 3.404651641845703,
      "step": 9356
    },
    {
      "epoch": 3.56903076171875e-05,
      "step": 9356,
      "training_step_time": 10.18155026435852
    },
    {
      "epoch": 3.56903076171875e-05,
      "model_forward_time": 3.1358323097229004,
      "step": 9356
    },
    {
      "epoch": 3.56903076171875e-05,
      "step": 9356,
      "training_step_time": 10.295560121536255
    },
    {
      "epoch": 3.569412231445312e-05,
      "model_forward_time": 3.1499581336975098,
      "step": 9357
    },
    {
      "epoch": 3.569412231445312e-05,
      "step": 9357,
      "training_step_time": 9.925678014755249
    },
    {
      "epoch": 3.569412231445312e-05,
      "model_forward_time": 3.132286548614502,
      "step": 9357
    },
    {
      "epoch": 3.569412231445312e-05,
      "step": 9357,
      "training_step_time": 9.910789966583252
    },
    {
      "epoch": 3.569412231445312e-05,
      "model_forward_time": 3.1419148445129395,
      "step": 9357
    },
    {
      "dynamics_loss_avg": 0.05140699278563261,
      "epoch": 3.569412231445312e-05,
      "step": 9357
    },
    {
      "action_loss_avg": 0.01773903479333967,
      "epoch": 3.569412231445312e-05,
      "step": 9357
    },
    {
      "epoch": 3.569412231445312e-05,
      "step": 9357,
      "training_step_time": 9.92151927947998
    },
    {
      "epoch": 3.569412231445312e-05,
      "model_forward_time": 3.137223243713379,
      "step": 9357
    },
    {
      "epoch": 3.569412231445312e-05,
      "step": 9357,
      "training_step_time": 10.283488750457764
    },
    {
      "epoch": 3.569793701171875e-05,
      "model_forward_time": 3.148042917251587,
      "step": 9358
    },
    {
      "epoch": 3.569793701171875e-05,
      "step": 9358,
      "training_step_time": 9.923555612564087
    },
    {
      "epoch": 3.569793701171875e-05,
      "model_forward_time": 3.1397321224212646,
      "step": 9358
    },
    {
      "epoch": 3.569793701171875e-05,
      "step": 9358,
      "training_step_time": 9.920536041259766
    },
    {
      "epoch": 3.569793701171875e-05,
      "model_forward_time": 3.1384010314941406,
      "step": 9358
    },
    {
      "epoch": 3.569793701171875e-05,
      "step": 9358,
      "training_step_time": 9.908695697784424
    },
    {
      "epoch": 3.569793701171875e-05,
      "model_forward_time": 3.137188196182251,
      "step": 9358
    },
    {
      "epoch": 3.569793701171875e-05,
      "step": 9358,
      "training_step_time": 10.343543529510498
    },
    {
      "epoch": 3.5701751708984375e-05,
      "model_forward_time": 3.4268341064453125,
      "step": 9359
    },
    {
      "epoch": 3.5701751708984375e-05,
      "step": 9359,
      "training_step_time": 10.20182728767395
    },
    {
      "epoch": 3.5701751708984375e-05,
      "model_forward_time": 3.131779193878174,
      "step": 9359
    },
    {
      "epoch": 3.5701751708984375e-05,
      "step": 9359,
      "training_step_time": 9.900732517242432
    },
    {
      "epoch": 3.5701751708984375e-05,
      "model_forward_time": 3.1349987983703613,
      "step": 9359
    },
    {
      "epoch": 3.5701751708984375e-05,
      "step": 9359,
      "training_step_time": 9.901892185211182
    },
    {
      "epoch": 3.5701751708984375e-05,
      "model_forward_time": 2.4314000606536865,
      "step": 9359
    },
    {
      "epoch": 3.5701751708984375e-05,
      "step": 9359,
      "training_step_time": 10.282465696334839
    },
    {
      "epoch": 3.570556640625e-05,
      "grad_norm": 0.27537697553634644,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 9360
    },
    {
      "epoch": 3.570556640625e-05,
      "model_forward_time": 3.1395957469940186,
      "step": 9360
    },
    {
      "dynamics_loss_avg": 0.04753018859773874,
      "epoch": 3.570556640625e-05,
      "step": 9360
    },
    {
      "action_loss_avg": 0.016654872242361306,
      "epoch": 3.570556640625e-05,
      "step": 9360
    },
    {
      "epoch": 3.570556640625e-05,
      "step": 9360,
      "training_step_time": 9.909553527832031
    },
    {
      "epoch": 3.570556640625e-05,
      "model_forward_time": 3.1355228424072266,
      "step": 9360
    },
    {
      "epoch": 3.570556640625e-05,
      "step": 9360,
      "training_step_time": 9.903409957885742
    },
    {
      "epoch": 3.570556640625e-05,
      "model_forward_time": 3.3990933895111084,
      "step": 9360
    },
    {
      "epoch": 3.570556640625e-05,
      "step": 9360,
      "training_step_time": 10.166876316070557
    },
    {
      "epoch": 3.570556640625e-05,
      "model_forward_time": 3.1339058876037598,
      "step": 9360
    },
    {
      "epoch": 3.570556640625e-05,
      "step": 9360,
      "training_step_time": 10.238446235656738
    },
    {
      "epoch": 3.570938110351563e-05,
      "model_forward_time": 3.1396217346191406,
      "step": 9361
    },
    {
      "epoch": 3.570938110351563e-05,
      "step": 9361,
      "training_step_time": 9.911336660385132
    },
    {
      "epoch": 3.570938110351563e-05,
      "model_forward_time": 3.135495662689209,
      "step": 9361
    },
    {
      "epoch": 3.570938110351563e-05,
      "step": 9361,
      "training_step_time": 9.905231952667236
    },
    {
      "epoch": 3.570938110351563e-05,
      "model_forward_time": 3.412069320678711,
      "step": 9361
    },
    {
      "epoch": 3.570938110351563e-05,
      "step": 9361,
      "training_step_time": 10.18147587776184
    },
    {
      "epoch": 3.570938110351563e-05,
      "model_forward_time": 3.1288962364196777,
      "step": 9361
    },
    {
      "epoch": 3.570938110351563e-05,
      "step": 9361,
      "training_step_time": 10.290852785110474
    },
    {
      "epoch": 3.571319580078125e-05,
      "model_forward_time": 3.1408426761627197,
      "step": 9362
    },
    {
      "epoch": 3.571319580078125e-05,
      "step": 9362,
      "training_step_time": 9.909542798995972
    },
    {
      "epoch": 3.571319580078125e-05,
      "model_forward_time": 3.1305181980133057,
      "step": 9362
    },
    {
      "epoch": 3.571319580078125e-05,
      "step": 9362,
      "training_step_time": 9.898427248001099
    },
    {
      "epoch": 3.571319580078125e-05,
      "model_forward_time": 3.1311087608337402,
      "step": 9362
    },
    {
      "dynamics_loss_avg": 0.05125534376129508,
      "epoch": 3.571319580078125e-05,
      "step": 9362
    },
    {
      "action_loss_avg": 0.018329187156632544,
      "epoch": 3.571319580078125e-05,
      "step": 9362
    },
    {
      "epoch": 3.571319580078125e-05,
      "step": 9362,
      "training_step_time": 9.906966924667358
    },
    {
      "epoch": 3.571319580078125e-05,
      "model_forward_time": 3.1342227458953857,
      "step": 9362
    },
    {
      "epoch": 3.571319580078125e-05,
      "step": 9362,
      "training_step_time": 10.248335361480713
    },
    {
      "epoch": 3.571701049804687e-05,
      "model_forward_time": 2.7689294815063477,
      "step": 9363
    },
    {
      "epoch": 3.571701049804687e-05,
      "step": 9363,
      "training_step_time": 7.802356243133545
    },
    {
      "epoch": 3.571701049804687e-05,
      "model_forward_time": 2.4301469326019287,
      "step": 9363
    },
    {
      "epoch": 3.571701049804687e-05,
      "step": 9363,
      "training_step_time": 7.454559087753296
    },
    {
      "epoch": 3.571701049804687e-05,
      "model_forward_time": 3.1309502124786377,
      "step": 9363
    },
    {
      "epoch": 3.571701049804687e-05,
      "step": 9363,
      "training_step_time": 9.89792275428772
    },
    {
      "epoch": 3.571701049804687e-05,
      "model_forward_time": 3.1342692375183105,
      "step": 9363
    },
    {
      "epoch": 3.571701049804687e-05,
      "step": 9363,
      "training_step_time": 10.262362241744995
    },
    {
      "epoch": 3.57208251953125e-05,
      "model_forward_time": 3.1554737091064453,
      "step": 9364
    },
    {
      "epoch": 3.57208251953125e-05,
      "step": 9364,
      "training_step_time": 9.926729917526245
    },
    {
      "epoch": 3.57208251953125e-05,
      "model_forward_time": 3.4069581031799316,
      "step": 9364
    },
    {
      "epoch": 3.57208251953125e-05,
      "step": 9364,
      "training_step_time": 10.184152841567993
    },
    {
      "epoch": 3.57208251953125e-05,
      "model_forward_time": 3.14121413230896,
      "step": 9364
    },
    {
      "epoch": 3.57208251953125e-05,
      "step": 9364,
      "training_step_time": 9.92068862915039
    },
    {
      "epoch": 3.57208251953125e-05,
      "model_forward_time": 3.144367218017578,
      "step": 9364
    },
    {
      "epoch": 3.57208251953125e-05,
      "step": 9364,
      "training_step_time": 10.330057859420776
    },
    {
      "epoch": 3.5724639892578126e-05,
      "model_forward_time": 2.4513778686523438,
      "step": 9365
    },
    {
      "dynamics_loss_avg": 0.05676620118319988,
      "epoch": 3.5724639892578126e-05,
      "step": 9365
    },
    {
      "action_loss_avg": 0.01860041720792651,
      "epoch": 3.5724639892578126e-05,
      "step": 9365
    },
    {
      "epoch": 3.5724639892578126e-05,
      "step": 9365,
      "training_step_time": 7.490571975708008
    },
    {
      "epoch": 3.5724639892578126e-05,
      "model_forward_time": 3.1378977298736572,
      "step": 9365
    },
    {
      "epoch": 3.5724639892578126e-05,
      "step": 9365,
      "training_step_time": 9.912611722946167
    },
    {
      "epoch": 3.5724639892578126e-05,
      "model_forward_time": 3.382786989212036,
      "step": 9365
    },
    {
      "epoch": 3.5724639892578126e-05,
      "step": 9365,
      "training_step_time": 10.163153886795044
    },
    {
      "epoch": 3.5724639892578126e-05,
      "model_forward_time": 3.1392617225646973,
      "step": 9365
    },
    {
      "epoch": 3.5724639892578126e-05,
      "step": 9365,
      "training_step_time": 10.282221555709839
    },
    {
      "epoch": 3.572845458984375e-05,
      "model_forward_time": 3.155090093612671,
      "step": 9366
    },
    {
      "epoch": 3.572845458984375e-05,
      "step": 9366,
      "training_step_time": 9.928000926971436
    },
    {
      "epoch": 3.572845458984375e-05,
      "model_forward_time": 3.1371827125549316,
      "step": 9366
    },
    {
      "epoch": 3.572845458984375e-05,
      "step": 9366,
      "training_step_time": 9.912535429000854
    },
    {
      "epoch": 3.572845458984375e-05,
      "model_forward_time": 3.1356804370880127,
      "step": 9366
    },
    {
      "epoch": 3.572845458984375e-05,
      "step": 9366,
      "training_step_time": 10.203800439834595
    },
    {
      "epoch": 3.572845458984375e-05,
      "model_forward_time": 2.4358057975769043,
      "step": 9366
    },
    {
      "epoch": 3.572845458984375e-05,
      "step": 9366,
      "training_step_time": 10.348381996154785
    },
    {
      "epoch": 3.573226928710938e-05,
      "model_forward_time": 2.4467737674713135,
      "step": 9367
    },
    {
      "epoch": 3.573226928710938e-05,
      "step": 9367,
      "training_step_time": 7.476632595062256
    },
    {
      "epoch": 3.573226928710938e-05,
      "model_forward_time": 3.1339075565338135,
      "step": 9367
    },
    {
      "epoch": 3.573226928710938e-05,
      "step": 9367,
      "training_step_time": 9.906442642211914
    },
    {
      "epoch": 3.573226928710938e-05,
      "model_forward_time": 3.1404213905334473,
      "step": 9367
    },
    {
      "dynamics_loss_avg": 0.05769931934773922,
      "epoch": 3.573226928710938e-05,
      "step": 9367
    },
    {
      "action_loss_avg": 0.023095931950956584,
      "epoch": 3.573226928710938e-05,
      "step": 9367
    },
    {
      "epoch": 3.573226928710938e-05,
      "step": 9367,
      "training_step_time": 9.914905548095703
    },
    {
      "epoch": 3.573226928710938e-05,
      "model_forward_time": 2.4346580505371094,
      "step": 9367
    },
    {
      "epoch": 3.573226928710938e-05,
      "step": 9367,
      "training_step_time": 10.308971881866455
    },
    {
      "epoch": 3.5736083984375e-05,
      "model_forward_time": 3.4314208030700684,
      "step": 9368
    },
    {
      "epoch": 3.5736083984375e-05,
      "step": 9368,
      "training_step_time": 10.214263439178467
    },
    {
      "epoch": 3.5736083984375e-05,
      "model_forward_time": 3.1475448608398438,
      "step": 9368
    },
    {
      "epoch": 3.5736083984375e-05,
      "step": 9368,
      "training_step_time": 9.921137571334839
    },
    {
      "epoch": 3.5736083984375e-05,
      "model_forward_time": 2.437974214553833,
      "step": 9368
    },
    {
      "epoch": 3.5736083984375e-05,
      "step": 9368,
      "training_step_time": 7.468969821929932
    },
    {
      "epoch": 3.5736083984375e-05,
      "model_forward_time": 3.137497663497925,
      "step": 9368
    },
    {
      "epoch": 3.5736083984375e-05,
      "step": 9368,
      "training_step_time": 10.268368482589722
    },
    {
      "epoch": 3.5739898681640624e-05,
      "model_forward_time": 3.1708781719207764,
      "step": 9369
    },
    {
      "epoch": 3.5739898681640624e-05,
      "step": 9369,
      "training_step_time": 9.94957947731018
    },
    {
      "epoch": 3.5739898681640624e-05,
      "model_forward_time": 2.752194881439209,
      "step": 9369
    },
    {
      "epoch": 3.5739898681640624e-05,
      "step": 9369,
      "training_step_time": 7.783974885940552
    },
    {
      "epoch": 3.5739898681640624e-05,
      "model_forward_time": 3.139167308807373,
      "step": 9369
    },
    {
      "epoch": 3.5739898681640624e-05,
      "step": 9369,
      "training_step_time": 9.910276651382446
    },
    {
      "epoch": 3.5739898681640624e-05,
      "model_forward_time": 3.137026309967041,
      "step": 9369
    },
    {
      "epoch": 3.5739898681640624e-05,
      "step": 9369,
      "training_step_time": 10.313090085983276
    },
    {
      "epoch": 3.574371337890625e-05,
      "grad_norm": 0.31542810797691345,
      "learning_rate": 1e-05,
      "loss": 0.0752,
      "step": 9370
    },
    {
      "epoch": 3.574371337890625e-05,
      "model_forward_time": 3.147230386734009,
      "step": 9370
    },
    {
      "dynamics_loss_avg": 0.07173856161534786,
      "epoch": 3.574371337890625e-05,
      "step": 9370
    },
    {
      "action_loss_avg": 0.02387380739673972,
      "epoch": 3.574371337890625e-05,
      "step": 9370
    },
    {
      "epoch": 3.574371337890625e-05,
      "step": 9370,
      "training_step_time": 9.92338252067566
    },
    {
      "epoch": 3.574371337890625e-05,
      "model_forward_time": 2.4339544773101807,
      "step": 9370
    },
    {
      "epoch": 3.574371337890625e-05,
      "step": 9370,
      "training_step_time": 7.465068101882935
    },
    {
      "epoch": 3.574371337890625e-05,
      "model_forward_time": 3.4187674522399902,
      "step": 9370
    },
    {
      "epoch": 3.574371337890625e-05,
      "step": 9370,
      "training_step_time": 10.195168256759644
    },
    {
      "epoch": 3.574371337890625e-05,
      "model_forward_time": 3.1390087604522705,
      "step": 9370
    },
    {
      "epoch": 3.574371337890625e-05,
      "step": 9370,
      "training_step_time": 10.276503086090088
    },
    {
      "epoch": 3.574752807617188e-05,
      "model_forward_time": 3.1557066440582275,
      "step": 9371
    },
    {
      "epoch": 3.574752807617188e-05,
      "step": 9371,
      "training_step_time": 9.942206621170044
    },
    {
      "epoch": 3.574752807617188e-05,
      "model_forward_time": 3.1445252895355225,
      "step": 9371
    },
    {
      "epoch": 3.574752807617188e-05,
      "step": 9371,
      "training_step_time": 9.93370246887207
    },
    {
      "epoch": 3.574752807617188e-05,
      "model_forward_time": 3.143832206726074,
      "step": 9371
    },
    {
      "epoch": 3.574752807617188e-05,
      "step": 9371,
      "training_step_time": 9.913796663284302
    },
    {
      "epoch": 3.574752807617188e-05,
      "model_forward_time": 3.4066390991210938,
      "step": 9371
    },
    {
      "epoch": 3.574752807617188e-05,
      "step": 9371,
      "training_step_time": 10.281028747558594
    },
    {
      "epoch": 3.57513427734375e-05,
      "model_forward_time": 3.164783477783203,
      "step": 9372
    },
    {
      "epoch": 3.57513427734375e-05,
      "step": 9372,
      "training_step_time": 9.94617509841919
    },
    {
      "epoch": 3.57513427734375e-05,
      "model_forward_time": 3.1442296504974365,
      "step": 9372
    },
    {
      "epoch": 3.57513427734375e-05,
      "step": 9372,
      "training_step_time": 9.918011665344238
    },
    {
      "epoch": 3.57513427734375e-05,
      "model_forward_time": 2.439274311065674,
      "step": 9372
    },
    {
      "dynamics_loss_avg": 0.05302540771663189,
      "epoch": 3.57513427734375e-05,
      "step": 9372
    },
    {
      "action_loss_avg": 0.01861308729276061,
      "epoch": 3.57513427734375e-05,
      "step": 9372
    },
    {
      "epoch": 3.57513427734375e-05,
      "step": 9372,
      "training_step_time": 7.4689414501190186
    },
    {
      "epoch": 3.57513427734375e-05,
      "model_forward_time": 3.153195858001709,
      "step": 9372
    },
    {
      "epoch": 3.57513427734375e-05,
      "step": 9372,
      "training_step_time": 10.294888973236084
    },
    {
      "epoch": 3.575515747070312e-05,
      "model_forward_time": 3.4613828659057617,
      "step": 9373
    },
    {
      "epoch": 3.575515747070312e-05,
      "step": 9373,
      "training_step_time": 10.236754179000854
    },
    {
      "epoch": 3.575515747070312e-05,
      "model_forward_time": 3.149817705154419,
      "step": 9373
    },
    {
      "epoch": 3.575515747070312e-05,
      "step": 9373,
      "training_step_time": 9.918858766555786
    },
    {
      "epoch": 3.575515747070312e-05,
      "model_forward_time": 2.4463789463043213,
      "step": 9373
    },
    {
      "epoch": 3.575515747070312e-05,
      "step": 9373,
      "training_step_time": 7.475779056549072
    },
    {
      "epoch": 3.575515747070312e-05,
      "model_forward_time": 3.1454923152923584,
      "step": 9373
    },
    {
      "epoch": 3.575515747070312e-05,
      "step": 9373,
      "training_step_time": 10.025643348693848
    },
    {
      "epoch": 3.575897216796875e-05,
      "model_forward_time": 3.1620261669158936,
      "step": 9374
    },
    {
      "epoch": 3.575897216796875e-05,
      "step": 9374,
      "training_step_time": 9.94847583770752
    },
    {
      "epoch": 3.575897216796875e-05,
      "model_forward_time": 3.436312675476074,
      "step": 9374
    },
    {
      "epoch": 3.575897216796875e-05,
      "step": 9374,
      "training_step_time": 10.206664323806763
    },
    {
      "epoch": 3.575897216796875e-05,
      "model_forward_time": 2.444216012954712,
      "step": 9374
    },
    {
      "epoch": 3.575897216796875e-05,
      "step": 9374,
      "training_step_time": 7.468848466873169
    },
    {
      "epoch": 3.575897216796875e-05,
      "model_forward_time": 3.1477880477905273,
      "step": 9374
    },
    {
      "epoch": 3.575897216796875e-05,
      "step": 9374,
      "training_step_time": 10.327971935272217
    },
    {
      "epoch": 3.5762786865234375e-05,
      "model_forward_time": 3.1646790504455566,
      "step": 9375
    },
    {
      "dynamics_loss_avg": 0.054986769519746306,
      "epoch": 3.5762786865234375e-05,
      "step": 9375
    },
    {
      "action_loss_avg": 0.017729462403804064,
      "epoch": 3.5762786865234375e-05,
      "step": 9375
    },
    {
      "epoch": 3.5762786865234375e-05,
      "step": 9375,
      "training_step_time": 9.941084861755371
    },
    {
      "epoch": 3.5762786865234375e-05,
      "model_forward_time": 3.150968551635742,
      "step": 9375
    },
    {
      "epoch": 3.5762786865234375e-05,
      "step": 9375,
      "training_step_time": 9.926022291183472
    },
    {
      "epoch": 3.5762786865234375e-05,
      "model_forward_time": 3.413876533508301,
      "step": 9375
    },
    {
      "epoch": 3.5762786865234375e-05,
      "step": 9375,
      "training_step_time": 10.199371099472046
    },
    {
      "epoch": 3.5762786865234375e-05,
      "model_forward_time": 3.1485347747802734,
      "step": 9375
    },
    {
      "epoch": 3.5762786865234375e-05,
      "step": 9375,
      "training_step_time": 10.324483394622803
    },
    {
      "epoch": 3.57666015625e-05,
      "model_forward_time": 3.159201145172119,
      "step": 9376
    },
    {
      "epoch": 3.57666015625e-05,
      "step": 9376,
      "training_step_time": 9.936109066009521
    },
    {
      "epoch": 3.57666015625e-05,
      "model_forward_time": 3.152056932449341,
      "step": 9376
    },
    {
      "epoch": 3.57666015625e-05,
      "step": 9376,
      "training_step_time": 9.928466796875
    },
    {
      "epoch": 3.57666015625e-05,
      "model_forward_time": 3.1480870246887207,
      "step": 9376
    },
    {
      "epoch": 3.57666015625e-05,
      "step": 9376,
      "training_step_time": 9.918492794036865
    },
    {
      "epoch": 3.57666015625e-05,
      "model_forward_time": 3.462111234664917,
      "step": 9376
    },
    {
      "epoch": 3.57666015625e-05,
      "step": 9376,
      "training_step_time": 10.327436447143555
    },
    {
      "epoch": 3.577041625976563e-05,
      "model_forward_time": 3.165635347366333,
      "step": 9377
    },
    {
      "epoch": 3.577041625976563e-05,
      "step": 9377,
      "training_step_time": 9.94239854812622
    },
    {
      "epoch": 3.577041625976563e-05,
      "model_forward_time": 3.149229049682617,
      "step": 9377
    },
    {
      "epoch": 3.577041625976563e-05,
      "step": 9377,
      "training_step_time": 9.919187068939209
    },
    {
      "epoch": 3.577041625976563e-05,
      "model_forward_time": 3.1522445678710938,
      "step": 9377
    },
    {
      "dynamics_loss_avg": 0.04660696294158697,
      "epoch": 3.577041625976563e-05,
      "step": 9377
    },
    {
      "action_loss_avg": 0.022023786138743163,
      "epoch": 3.577041625976563e-05,
      "step": 9377
    },
    {
      "epoch": 3.577041625976563e-05,
      "step": 9377,
      "training_step_time": 9.927500247955322
    },
    {
      "epoch": 3.577041625976563e-05,
      "model_forward_time": 2.449894666671753,
      "step": 9377
    },
    {
      "epoch": 3.577041625976563e-05,
      "step": 9377,
      "training_step_time": 10.024869203567505
    },
    {
      "epoch": 3.577423095703125e-05,
      "model_forward_time": 3.4557807445526123,
      "step": 9378
    },
    {
      "epoch": 3.577423095703125e-05,
      "step": 9378,
      "training_step_time": 10.230360746383667
    },
    {
      "epoch": 3.577423095703125e-05,
      "model_forward_time": 3.146204948425293,
      "step": 9378
    },
    {
      "epoch": 3.577423095703125e-05,
      "step": 9378,
      "training_step_time": 9.921931028366089
    },
    {
      "epoch": 3.577423095703125e-05,
      "model_forward_time": 3.152162551879883,
      "step": 9378
    },
    {
      "epoch": 3.577423095703125e-05,
      "step": 9378,
      "training_step_time": 9.920311689376831
    },
    {
      "epoch": 3.577423095703125e-05,
      "model_forward_time": 2.445284843444824,
      "step": 9378
    },
    {
      "epoch": 3.577423095703125e-05,
      "step": 9378,
      "training_step_time": 10.007319688796997
    },
    {
      "epoch": 3.577804565429687e-05,
      "model_forward_time": 3.162365674972534,
      "step": 9379
    },
    {
      "epoch": 3.577804565429687e-05,
      "step": 9379,
      "training_step_time": 9.934412956237793
    },
    {
      "epoch": 3.577804565429687e-05,
      "model_forward_time": 3.403132915496826,
      "step": 9379
    },
    {
      "epoch": 3.577804565429687e-05,
      "step": 9379,
      "training_step_time": 10.172942161560059
    },
    {
      "epoch": 3.577804565429687e-05,
      "model_forward_time": 3.1475908756256104,
      "step": 9379
    },
    {
      "epoch": 3.577804565429687e-05,
      "step": 9379,
      "training_step_time": 9.918492555618286
    },
    {
      "epoch": 3.577804565429687e-05,
      "model_forward_time": 3.1500799655914307,
      "step": 9379
    },
    {
      "epoch": 3.577804565429687e-05,
      "step": 9379,
      "training_step_time": 10.310975551605225
    },
    {
      "epoch": 3.57818603515625e-05,
      "grad_norm": 0.4719890356063843,
      "learning_rate": 1e-05,
      "loss": 0.0726,
      "step": 9380
    },
    {
      "epoch": 3.57818603515625e-05,
      "model_forward_time": 2.452124834060669,
      "step": 9380
    },
    {
      "dynamics_loss_avg": 0.052425187081098557,
      "epoch": 3.57818603515625e-05,
      "step": 9380
    },
    {
      "action_loss_avg": 0.020450177742168306,
      "epoch": 3.57818603515625e-05,
      "step": 9380
    },
    {
      "epoch": 3.57818603515625e-05,
      "step": 9380,
      "training_step_time": 7.485666751861572
    },
    {
      "epoch": 3.57818603515625e-05,
      "model_forward_time": 2.440333366394043,
      "step": 9380
    },
    {
      "epoch": 3.57818603515625e-05,
      "step": 9380,
      "training_step_time": 7.467844486236572
    },
    {
      "epoch": 3.57818603515625e-05,
      "model_forward_time": 3.434109926223755,
      "step": 9380
    },
    {
      "epoch": 3.57818603515625e-05,
      "step": 9380,
      "training_step_time": 10.20050859451294
    },
    {
      "epoch": 3.57818603515625e-05,
      "model_forward_time": 3.1433937549591064,
      "step": 9380
    },
    {
      "epoch": 3.57818603515625e-05,
      "step": 9380,
      "training_step_time": 10.005654096603394
    },
    {
      "epoch": 3.5785675048828126e-05,
      "model_forward_time": 3.1604526042938232,
      "step": 9381
    },
    {
      "epoch": 3.5785675048828126e-05,
      "step": 9381,
      "training_step_time": 9.930498600006104
    },
    {
      "epoch": 3.5785675048828126e-05,
      "model_forward_time": 2.4416556358337402,
      "step": 9381
    },
    {
      "epoch": 3.5785675048828126e-05,
      "step": 9381,
      "training_step_time": 7.471011161804199
    },
    {
      "epoch": 3.5785675048828126e-05,
      "model_forward_time": 3.1431314945220947,
      "step": 9381
    },
    {
      "epoch": 3.5785675048828126e-05,
      "step": 9381,
      "training_step_time": 9.908498287200928
    },
    {
      "epoch": 3.5785675048828126e-05,
      "model_forward_time": 3.4423465728759766,
      "step": 9381
    },
    {
      "epoch": 3.5785675048828126e-05,
      "step": 9381,
      "training_step_time": 10.307855129241943
    },
    {
      "epoch": 3.578948974609375e-05,
      "model_forward_time": 3.1557064056396484,
      "step": 9382
    },
    {
      "epoch": 3.578948974609375e-05,
      "step": 9382,
      "training_step_time": 9.926761150360107
    },
    {
      "epoch": 3.578948974609375e-05,
      "model_forward_time": 2.4400742053985596,
      "step": 9382
    },
    {
      "epoch": 3.578948974609375e-05,
      "step": 9382,
      "training_step_time": 7.4630126953125
    },
    {
      "epoch": 3.578948974609375e-05,
      "model_forward_time": 3.141855001449585,
      "step": 9382
    },
    {
      "dynamics_loss_avg": 0.05315748360008001,
      "epoch": 3.578948974609375e-05,
      "step": 9382
    },
    {
      "action_loss_avg": 0.015620027668774128,
      "epoch": 3.578948974609375e-05,
      "step": 9382
    },
    {
      "epoch": 3.578948974609375e-05,
      "step": 9382,
      "training_step_time": 9.923694610595703
    },
    {
      "epoch": 3.578948974609375e-05,
      "model_forward_time": 3.15132999420166,
      "step": 9382
    },
    {
      "epoch": 3.578948974609375e-05,
      "step": 9382,
      "training_step_time": 10.269617557525635
    },
    {
      "epoch": 3.579330444335937e-05,
      "model_forward_time": 3.1571319103240967,
      "step": 9383
    },
    {
      "epoch": 3.579330444335937e-05,
      "step": 9383,
      "training_step_time": 10.215578556060791
    },
    {
      "epoch": 3.579330444335937e-05,
      "model_forward_time": 3.1546871662139893,
      "step": 9383
    },
    {
      "epoch": 3.579330444335937e-05,
      "step": 9383,
      "training_step_time": 9.932905912399292
    },
    {
      "epoch": 3.579330444335937e-05,
      "model_forward_time": 3.1498260498046875,
      "step": 9383
    },
    {
      "epoch": 3.579330444335937e-05,
      "step": 9383,
      "training_step_time": 9.925496816635132
    },
    {
      "epoch": 3.579330444335937e-05,
      "model_forward_time": 3.155489206314087,
      "step": 9383
    },
    {
      "epoch": 3.579330444335937e-05,
      "step": 9383,
      "training_step_time": 10.025093793869019
    },
    {
      "epoch": 3.5797119140625e-05,
      "model_forward_time": 3.1612818241119385,
      "step": 9384
    },
    {
      "epoch": 3.5797119140625e-05,
      "step": 9384,
      "training_step_time": 9.935972452163696
    },
    {
      "epoch": 3.5797119140625e-05,
      "model_forward_time": 3.4200947284698486,
      "step": 9384
    },
    {
      "epoch": 3.5797119140625e-05,
      "step": 9384,
      "training_step_time": 10.198556661605835
    },
    {
      "epoch": 3.5797119140625e-05,
      "model_forward_time": 2.439767599105835,
      "step": 9384
    },
    {
      "epoch": 3.5797119140625e-05,
      "step": 9384,
      "training_step_time": 7.472583532333374
    },
    {
      "epoch": 3.5797119140625e-05,
      "model_forward_time": 3.1405088901519775,
      "step": 9384
    },
    {
      "epoch": 3.5797119140625e-05,
      "step": 9384,
      "training_step_time": 10.320822715759277
    },
    {
      "epoch": 3.5800933837890624e-05,
      "model_forward_time": 3.148477077484131,
      "step": 9385
    },
    {
      "dynamics_loss_avg": 0.044835522398352624,
      "epoch": 3.5800933837890624e-05,
      "step": 9385
    },
    {
      "action_loss_avg": 0.018043951224535705,
      "epoch": 3.5800933837890624e-05,
      "step": 9385
    },
    {
      "epoch": 3.5800933837890624e-05,
      "step": 9385,
      "training_step_time": 9.926259756088257
    },
    {
      "epoch": 3.5800933837890624e-05,
      "model_forward_time": 3.1357572078704834,
      "step": 9385
    },
    {
      "epoch": 3.5800933837890624e-05,
      "step": 9385,
      "training_step_time": 9.91132664680481
    },
    {
      "epoch": 3.5800933837890624e-05,
      "model_forward_time": 3.139650583267212,
      "step": 9385
    },
    {
      "epoch": 3.5800933837890624e-05,
      "step": 9385,
      "training_step_time": 10.179165840148926
    },
    {
      "epoch": 3.5800933837890624e-05,
      "model_forward_time": 3.140476703643799,
      "step": 9385
    },
    {
      "epoch": 3.5800933837890624e-05,
      "step": 9385,
      "training_step_time": 10.305432319641113
    },
    {
      "epoch": 3.580474853515625e-05,
      "model_forward_time": 2.448340654373169,
      "step": 9386
    },
    {
      "epoch": 3.580474853515625e-05,
      "step": 9386,
      "training_step_time": 7.478579998016357
    },
    {
      "epoch": 3.580474853515625e-05,
      "model_forward_time": 3.1313226222991943,
      "step": 9386
    },
    {
      "epoch": 3.580474853515625e-05,
      "step": 9386,
      "training_step_time": 9.899570941925049
    },
    {
      "epoch": 3.580474853515625e-05,
      "model_forward_time": 3.1364078521728516,
      "step": 9386
    },
    {
      "epoch": 3.580474853515625e-05,
      "step": 9386,
      "training_step_time": 9.904651403427124
    },
    {
      "epoch": 3.580474853515625e-05,
      "model_forward_time": 2.432737350463867,
      "step": 9386
    },
    {
      "epoch": 3.580474853515625e-05,
      "step": 9386,
      "training_step_time": 10.311052083969116
    },
    {
      "epoch": 3.580856323242188e-05,
      "model_forward_time": 3.1506001949310303,
      "step": 9387
    },
    {
      "epoch": 3.580856323242188e-05,
      "step": 9387,
      "training_step_time": 9.919737100601196
    },
    {
      "epoch": 3.580856323242188e-05,
      "model_forward_time": 3.132713794708252,
      "step": 9387
    },
    {
      "epoch": 3.580856323242188e-05,
      "step": 9387,
      "training_step_time": 9.898380279541016
    },
    {
      "epoch": 3.580856323242188e-05,
      "model_forward_time": 3.1362361907958984,
      "step": 9387
    },
    {
      "dynamics_loss_avg": 0.05703881531953812,
      "epoch": 3.580856323242188e-05,
      "step": 9387
    },
    {
      "action_loss_avg": 0.01718573537655175,
      "epoch": 3.580856323242188e-05,
      "step": 9387
    },
    {
      "epoch": 3.580856323242188e-05,
      "step": 9387,
      "training_step_time": 9.90721583366394
    },
    {
      "epoch": 3.580856323242188e-05,
      "model_forward_time": 3.1355373859405518,
      "step": 9387
    },
    {
      "epoch": 3.580856323242188e-05,
      "step": 9387,
      "training_step_time": 10.338634014129639
    },
    {
      "epoch": 3.58123779296875e-05,
      "model_forward_time": 3.428452491760254,
      "step": 9388
    },
    {
      "epoch": 3.58123779296875e-05,
      "step": 9388,
      "training_step_time": 10.202255487442017
    },
    {
      "epoch": 3.58123779296875e-05,
      "model_forward_time": 2.4314751625061035,
      "step": 9388
    },
    {
      "epoch": 3.58123779296875e-05,
      "step": 9388,
      "training_step_time": 7.459292888641357
    },
    {
      "epoch": 3.58123779296875e-05,
      "model_forward_time": 3.134439468383789,
      "step": 9388
    },
    {
      "epoch": 3.58123779296875e-05,
      "step": 9388,
      "training_step_time": 9.904899597167969
    },
    {
      "epoch": 3.58123779296875e-05,
      "model_forward_time": 3.134507656097412,
      "step": 9388
    },
    {
      "epoch": 3.58123779296875e-05,
      "step": 9388,
      "training_step_time": 10.007126092910767
    },
    {
      "epoch": 3.581619262695312e-05,
      "model_forward_time": 3.149894952774048,
      "step": 9389
    },
    {
      "epoch": 3.581619262695312e-05,
      "step": 9389,
      "training_step_time": 9.915291786193848
    },
    {
      "epoch": 3.581619262695312e-05,
      "model_forward_time": 3.1378064155578613,
      "step": 9389
    },
    {
      "epoch": 3.581619262695312e-05,
      "step": 9389,
      "training_step_time": 10.170961618423462
    },
    {
      "epoch": 3.581619262695312e-05,
      "model_forward_time": 3.1362853050231934,
      "step": 9389
    },
    {
      "epoch": 3.581619262695312e-05,
      "step": 9389,
      "training_step_time": 9.90527868270874
    },
    {
      "epoch": 3.581619262695312e-05,
      "model_forward_time": 3.133605718612671,
      "step": 9389
    },
    {
      "epoch": 3.581619262695312e-05,
      "step": 9389,
      "training_step_time": 10.005261182785034
    },
    {
      "epoch": 3.582000732421875e-05,
      "grad_norm": 0.5037087798118591,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 9390
    },
    {
      "epoch": 3.582000732421875e-05,
      "model_forward_time": 3.146007537841797,
      "step": 9390
    },
    {
      "dynamics_loss_avg": 0.05976319145411253,
      "epoch": 3.582000732421875e-05,
      "step": 9390
    },
    {
      "action_loss_avg": 0.026084326766431333,
      "epoch": 3.582000732421875e-05,
      "step": 9390
    },
    {
      "epoch": 3.582000732421875e-05,
      "step": 9390,
      "training_step_time": 9.920171022415161
    },
    {
      "epoch": 3.582000732421875e-05,
      "model_forward_time": 2.427743673324585,
      "step": 9390
    },
    {
      "epoch": 3.582000732421875e-05,
      "step": 9390,
      "training_step_time": 7.454193115234375
    },
    {
      "epoch": 3.582000732421875e-05,
      "model_forward_time": 3.449707508087158,
      "step": 9390
    },
    {
      "epoch": 3.582000732421875e-05,
      "step": 9390,
      "training_step_time": 10.215689420700073
    },
    {
      "epoch": 3.582000732421875e-05,
      "model_forward_time": 3.137129306793213,
      "step": 9390
    },
    {
      "epoch": 3.582000732421875e-05,
      "step": 9390,
      "training_step_time": 10.020941734313965
    },
    {
      "epoch": 3.5823822021484375e-05,
      "model_forward_time": 3.1619081497192383,
      "step": 9391
    },
    {
      "epoch": 3.5823822021484375e-05,
      "step": 9391,
      "training_step_time": 9.938175916671753
    },
    {
      "epoch": 3.5823822021484375e-05,
      "model_forward_time": 3.142941951751709,
      "step": 9391
    },
    {
      "epoch": 3.5823822021484375e-05,
      "step": 9391,
      "training_step_time": 9.920151948928833
    },
    {
      "epoch": 3.5823822021484375e-05,
      "model_forward_time": 3.1432483196258545,
      "step": 9391
    },
    {
      "epoch": 3.5823822021484375e-05,
      "step": 9391,
      "training_step_time": 9.917680740356445
    },
    {
      "epoch": 3.5823822021484375e-05,
      "model_forward_time": 3.1412513256073,
      "step": 9391
    },
    {
      "epoch": 3.5823822021484375e-05,
      "step": 9391,
      "training_step_time": 10.311907052993774
    },
    {
      "epoch": 3.582763671875e-05,
      "model_forward_time": 2.456822395324707,
      "step": 9392
    },
    {
      "epoch": 3.582763671875e-05,
      "step": 9392,
      "training_step_time": 7.49224328994751
    },
    {
      "epoch": 3.582763671875e-05,
      "model_forward_time": 3.1415042877197266,
      "step": 9392
    },
    {
      "epoch": 3.582763671875e-05,
      "step": 9392,
      "training_step_time": 9.924412250518799
    },
    {
      "epoch": 3.582763671875e-05,
      "model_forward_time": 3.1415982246398926,
      "step": 9392
    },
    {
      "dynamics_loss_avg": 0.05400853641331196,
      "epoch": 3.582763671875e-05,
      "step": 9392
    },
    {
      "action_loss_avg": 0.019787500891834498,
      "epoch": 3.582763671875e-05,
      "step": 9392
    },
    {
      "epoch": 3.582763671875e-05,
      "step": 9392,
      "training_step_time": 9.91996717453003
    },
    {
      "epoch": 3.582763671875e-05,
      "model_forward_time": 2.432070255279541,
      "step": 9392
    },
    {
      "epoch": 3.582763671875e-05,
      "step": 9392,
      "training_step_time": 10.251269817352295
    },
    {
      "epoch": 3.583145141601563e-05,
      "model_forward_time": 3.145667314529419,
      "step": 9393
    },
    {
      "epoch": 3.583145141601563e-05,
      "step": 9393,
      "training_step_time": 10.199836730957031
    },
    {
      "epoch": 3.583145141601563e-05,
      "model_forward_time": 3.140456438064575,
      "step": 9393
    },
    {
      "epoch": 3.583145141601563e-05,
      "step": 9393,
      "training_step_time": 9.917185068130493
    },
    {
      "epoch": 3.583145141601563e-05,
      "model_forward_time": 3.1364340782165527,
      "step": 9393
    },
    {
      "epoch": 3.583145141601563e-05,
      "step": 9393,
      "training_step_time": 9.911134719848633
    },
    {
      "epoch": 3.583145141601563e-05,
      "model_forward_time": 3.1425228118896484,
      "step": 9393
    },
    {
      "epoch": 3.583145141601563e-05,
      "step": 9393,
      "training_step_time": 10.326255083084106
    },
    {
      "epoch": 3.583526611328125e-05,
      "model_forward_time": 3.1516833305358887,
      "step": 9394
    },
    {
      "epoch": 3.583526611328125e-05,
      "step": 9394,
      "training_step_time": 9.932246208190918
    },
    {
      "epoch": 3.583526611328125e-05,
      "model_forward_time": 3.1387271881103516,
      "step": 9394
    },
    {
      "epoch": 3.583526611328125e-05,
      "step": 9394,
      "training_step_time": 10.198516607284546
    },
    {
      "epoch": 3.583526611328125e-05,
      "model_forward_time": 3.140920400619507,
      "step": 9394
    },
    {
      "epoch": 3.583526611328125e-05,
      "step": 9394,
      "training_step_time": 9.925005674362183
    },
    {
      "epoch": 3.583526611328125e-05,
      "model_forward_time": 3.1448721885681152,
      "step": 9394
    },
    {
      "epoch": 3.583526611328125e-05,
      "step": 9394,
      "training_step_time": 10.030880451202393
    },
    {
      "epoch": 3.583908081054687e-05,
      "model_forward_time": 3.1571247577667236,
      "step": 9395
    },
    {
      "dynamics_loss_avg": 0.04416099060326815,
      "epoch": 3.583908081054687e-05,
      "step": 9395
    },
    {
      "action_loss_avg": 0.01835641972720623,
      "epoch": 3.583908081054687e-05,
      "step": 9395
    },
    {
      "epoch": 3.583908081054687e-05,
      "step": 9395,
      "training_step_time": 9.94479489326477
    },
    {
      "epoch": 3.583908081054687e-05,
      "model_forward_time": 3.1416194438934326,
      "step": 9395
    },
    {
      "epoch": 3.583908081054687e-05,
      "step": 9395,
      "training_step_time": 9.926457643508911
    },
    {
      "epoch": 3.583908081054687e-05,
      "model_forward_time": 3.40364933013916,
      "step": 9395
    },
    {
      "epoch": 3.583908081054687e-05,
      "step": 9395,
      "training_step_time": 10.181309461593628
    },
    {
      "epoch": 3.583908081054687e-05,
      "model_forward_time": 3.143939733505249,
      "step": 9395
    },
    {
      "epoch": 3.583908081054687e-05,
      "step": 9395,
      "training_step_time": 10.298215866088867
    },
    {
      "epoch": 3.58428955078125e-05,
      "model_forward_time": 2.4512617588043213,
      "step": 9396
    },
    {
      "epoch": 3.58428955078125e-05,
      "step": 9396,
      "training_step_time": 7.484736680984497
    },
    {
      "epoch": 3.58428955078125e-05,
      "model_forward_time": 3.133668899536133,
      "step": 9396
    },
    {
      "epoch": 3.58428955078125e-05,
      "step": 9396,
      "training_step_time": 9.911502361297607
    },
    {
      "epoch": 3.58428955078125e-05,
      "model_forward_time": 3.137221097946167,
      "step": 9396
    },
    {
      "epoch": 3.58428955078125e-05,
      "step": 9396,
      "training_step_time": 9.917198181152344
    },
    {
      "epoch": 3.58428955078125e-05,
      "model_forward_time": 3.136828660964966,
      "step": 9396
    },
    {
      "epoch": 3.58428955078125e-05,
      "step": 9396,
      "training_step_time": 10.284852743148804
    },
    {
      "epoch": 3.5846710205078126e-05,
      "model_forward_time": 3.1658735275268555,
      "step": 9397
    },
    {
      "epoch": 3.5846710205078126e-05,
      "step": 9397,
      "training_step_time": 9.94043755531311
    },
    {
      "epoch": 3.5846710205078126e-05,
      "model_forward_time": 3.14390230178833,
      "step": 9397
    },
    {
      "epoch": 3.5846710205078126e-05,
      "step": 9397,
      "training_step_time": 9.91627025604248
    },
    {
      "epoch": 3.5846710205078126e-05,
      "model_forward_time": 3.148624897003174,
      "step": 9397
    },
    {
      "dynamics_loss_avg": 0.05645406749099493,
      "epoch": 3.5846710205078126e-05,
      "step": 9397
    },
    {
      "action_loss_avg": 0.02308504590764642,
      "epoch": 3.5846710205078126e-05,
      "step": 9397
    },
    {
      "epoch": 3.5846710205078126e-05,
      "step": 9397,
      "training_step_time": 9.925883531570435
    },
    {
      "epoch": 3.5846710205078126e-05,
      "model_forward_time": 2.4398410320281982,
      "step": 9397
    },
    {
      "epoch": 3.5846710205078126e-05,
      "step": 9397,
      "training_step_time": 10.292861938476562
    },
    {
      "epoch": 3.585052490234375e-05,
      "model_forward_time": 3.1667354106903076,
      "step": 9398
    },
    {
      "epoch": 3.585052490234375e-05,
      "step": 9398,
      "training_step_time": 10.223971843719482
    },
    {
      "epoch": 3.585052490234375e-05,
      "model_forward_time": 3.13521671295166,
      "step": 9398
    },
    {
      "epoch": 3.585052490234375e-05,
      "step": 9398,
      "training_step_time": 9.9079110622406
    },
    {
      "epoch": 3.585052490234375e-05,
      "model_forward_time": 3.1406519412994385,
      "step": 9398
    },
    {
      "epoch": 3.585052490234375e-05,
      "step": 9398,
      "training_step_time": 9.914127588272095
    },
    {
      "epoch": 3.585052490234375e-05,
      "model_forward_time": 3.1409335136413574,
      "step": 9398
    },
    {
      "epoch": 3.585052490234375e-05,
      "step": 9398,
      "training_step_time": 10.239225625991821
    },
    {
      "epoch": 3.585433959960937e-05,
      "model_forward_time": 3.1461446285247803,
      "step": 9399
    },
    {
      "epoch": 3.585433959960937e-05,
      "step": 9399,
      "training_step_time": 9.921936273574829
    },
    {
      "epoch": 3.585433959960937e-05,
      "model_forward_time": 2.435516357421875,
      "step": 9399
    },
    {
      "epoch": 3.585433959960937e-05,
      "step": 9399,
      "training_step_time": 7.469813823699951
    },
    {
      "epoch": 3.585433959960937e-05,
      "model_forward_time": 3.4071764945983887,
      "step": 9399
    },
    {
      "epoch": 3.585433959960937e-05,
      "step": 9399,
      "training_step_time": 10.181001901626587
    },
    {
      "epoch": 3.585433959960937e-05,
      "model_forward_time": 3.136322259902954,
      "step": 9399
    },
    {
      "epoch": 3.585433959960937e-05,
      "step": 9399,
      "training_step_time": 10.008768081665039
    },
    {
      "epoch": 3.5858154296875e-05,
      "grad_norm": 0.46147581934928894,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9400
    },
    {
      "epoch": 3.5858154296875e-05,
      "model_forward_time": 3.166837215423584,
      "step": 9400
    },
    {
      "dynamics_loss_avg": 0.0584857914596796,
      "epoch": 3.5858154296875e-05,
      "step": 9400
    },
    {
      "action_loss_avg": 0.01961528053507209,
      "epoch": 3.5858154296875e-05,
      "step": 9400
    },
    {
      "epoch": 3.5858154296875e-05,
      "step": 9400,
      "training_step_time": 9.947762727737427
    },
    {
      "epoch": 3.5858154296875e-05,
      "model_forward_time": 3.1498377323150635,
      "step": 9400
    },
    {
      "epoch": 3.5858154296875e-05,
      "step": 9400,
      "training_step_time": 9.931061029434204
    },
    {
      "epoch": 3.5858154296875e-05,
      "model_forward_time": 3.462141990661621,
      "step": 9400
    },
    {
      "epoch": 3.5858154296875e-05,
      "step": 9400,
      "training_step_time": 10.24464750289917
    },
    {
      "epoch": 3.5858154296875e-05,
      "model_forward_time": 3.148949384689331,
      "step": 9400
    },
    {
      "epoch": 3.5858154296875e-05,
      "step": 9400,
      "training_step_time": 10.027623891830444
    },
    {
      "epoch": 3.5861968994140624e-05,
      "model_forward_time": 3.16764760017395,
      "step": 9401
    },
    {
      "epoch": 3.5861968994140624e-05,
      "step": 9401,
      "training_step_time": 9.951770067214966
    },
    {
      "epoch": 3.5861968994140624e-05,
      "model_forward_time": 3.145005941390991,
      "step": 9401
    },
    {
      "epoch": 3.5861968994140624e-05,
      "step": 9401,
      "training_step_time": 9.92156434059143
    },
    {
      "epoch": 3.5861968994140624e-05,
      "model_forward_time": 2.444613218307495,
      "step": 9401
    },
    {
      "epoch": 3.5861968994140624e-05,
      "step": 9401,
      "training_step_time": 7.473344326019287
    },
    {
      "epoch": 3.5861968994140624e-05,
      "model_forward_time": 3.411776065826416,
      "step": 9401
    },
    {
      "epoch": 3.5861968994140624e-05,
      "step": 9401,
      "training_step_time": 10.27494502067566
    },
    {
      "epoch": 3.586578369140625e-05,
      "model_forward_time": 2.4485669136047363,
      "step": 9402
    },
    {
      "epoch": 3.586578369140625e-05,
      "step": 9402,
      "training_step_time": 7.490450859069824
    },
    {
      "epoch": 3.586578369140625e-05,
      "model_forward_time": 3.1526939868927,
      "step": 9402
    },
    {
      "epoch": 3.586578369140625e-05,
      "step": 9402,
      "training_step_time": 9.9315345287323
    },
    {
      "epoch": 3.586578369140625e-05,
      "model_forward_time": 3.1431849002838135,
      "step": 9402
    },
    {
      "dynamics_loss_avg": 0.05417542466893792,
      "epoch": 3.586578369140625e-05,
      "step": 9402
    },
    {
      "action_loss_avg": 0.021340804593637585,
      "epoch": 3.586578369140625e-05,
      "step": 9402
    },
    {
      "epoch": 3.586578369140625e-05,
      "step": 9402,
      "training_step_time": 9.926505088806152
    },
    {
      "epoch": 3.586578369140625e-05,
      "model_forward_time": 2.4464173316955566,
      "step": 9402
    },
    {
      "epoch": 3.586578369140625e-05,
      "step": 9402,
      "training_step_time": 10.32780909538269
    },
    {
      "epoch": 3.586959838867188e-05,
      "model_forward_time": 3.1578967571258545,
      "step": 9403
    },
    {
      "epoch": 3.586959838867188e-05,
      "step": 9403,
      "training_step_time": 9.940035820007324
    },
    {
      "epoch": 3.586959838867188e-05,
      "model_forward_time": 3.4651923179626465,
      "step": 9403
    },
    {
      "epoch": 3.586959838867188e-05,
      "step": 9403,
      "training_step_time": 10.251219511032104
    },
    {
      "epoch": 3.586959838867188e-05,
      "model_forward_time": 3.1442325115203857,
      "step": 9403
    },
    {
      "epoch": 3.586959838867188e-05,
      "step": 9403,
      "training_step_time": 9.93476414680481
    },
    {
      "epoch": 3.586959838867188e-05,
      "model_forward_time": 3.1552231311798096,
      "step": 9403
    },
    {
      "epoch": 3.586959838867188e-05,
      "step": 9403,
      "training_step_time": 10.317879915237427
    },
    {
      "epoch": 3.58734130859375e-05,
      "model_forward_time": 2.475809335708618,
      "step": 9404
    },
    {
      "epoch": 3.58734130859375e-05,
      "step": 9404,
      "training_step_time": 7.509223222732544
    },
    {
      "epoch": 3.58734130859375e-05,
      "model_forward_time": 3.1316018104553223,
      "step": 9404
    },
    {
      "epoch": 3.58734130859375e-05,
      "step": 9404,
      "training_step_time": 9.904541969299316
    },
    {
      "epoch": 3.58734130859375e-05,
      "model_forward_time": 3.4335596561431885,
      "step": 9404
    },
    {
      "epoch": 3.58734130859375e-05,
      "step": 9404,
      "training_step_time": 10.206399202346802
    },
    {
      "epoch": 3.58734130859375e-05,
      "model_forward_time": 3.133903980255127,
      "step": 9404
    },
    {
      "epoch": 3.58734130859375e-05,
      "step": 9404,
      "training_step_time": 10.262526035308838
    },
    {
      "epoch": 3.587722778320312e-05,
      "model_forward_time": 3.147254228591919,
      "step": 9405
    },
    {
      "dynamics_loss_avg": 0.05645597539842129,
      "epoch": 3.587722778320312e-05,
      "step": 9405
    },
    {
      "action_loss_avg": 0.021279322821646927,
      "epoch": 3.587722778320312e-05,
      "step": 9405
    },
    {
      "epoch": 3.587722778320312e-05,
      "step": 9405,
      "training_step_time": 9.931377172470093
    },
    {
      "epoch": 3.587722778320312e-05,
      "model_forward_time": 3.132786989212036,
      "step": 9405
    },
    {
      "epoch": 3.587722778320312e-05,
      "step": 9405,
      "training_step_time": 9.902315139770508
    },
    {
      "epoch": 3.587722778320312e-05,
      "model_forward_time": 3.1300923824310303,
      "step": 9405
    },
    {
      "epoch": 3.587722778320312e-05,
      "step": 9405,
      "training_step_time": 9.90575623512268
    },
    {
      "epoch": 3.587722778320312e-05,
      "model_forward_time": 3.4474411010742188,
      "step": 9405
    },
    {
      "epoch": 3.587722778320312e-05,
      "step": 9405,
      "training_step_time": 10.313780546188354
    },
    {
      "epoch": 3.588104248046875e-05,
      "model_forward_time": 3.1531591415405273,
      "step": 9406
    },
    {
      "epoch": 3.588104248046875e-05,
      "step": 9406,
      "training_step_time": 9.928104877471924
    },
    {
      "epoch": 3.588104248046875e-05,
      "model_forward_time": 3.133260726928711,
      "step": 9406
    },
    {
      "epoch": 3.588104248046875e-05,
      "step": 9406,
      "training_step_time": 9.913747310638428
    },
    {
      "epoch": 3.588104248046875e-05,
      "model_forward_time": 2.434431552886963,
      "step": 9406
    },
    {
      "epoch": 3.588104248046875e-05,
      "step": 9406,
      "training_step_time": 7.469501972198486
    },
    {
      "epoch": 3.588104248046875e-05,
      "model_forward_time": 3.132685661315918,
      "step": 9406
    },
    {
      "epoch": 3.588104248046875e-05,
      "step": 9406,
      "training_step_time": 10.294821977615356
    },
    {
      "epoch": 3.5884857177734375e-05,
      "model_forward_time": 2.4456586837768555,
      "step": 9407
    },
    {
      "epoch": 3.5884857177734375e-05,
      "step": 9407,
      "training_step_time": 7.492693185806274
    },
    {
      "epoch": 3.5884857177734375e-05,
      "model_forward_time": 3.142970561981201,
      "step": 9407
    },
    {
      "epoch": 3.5884857177734375e-05,
      "step": 9407,
      "training_step_time": 9.916527271270752
    },
    {
      "epoch": 3.5884857177734375e-05,
      "model_forward_time": 3.157949686050415,
      "step": 9407
    },
    {
      "dynamics_loss_avg": 0.0522403160110116,
      "epoch": 3.5884857177734375e-05,
      "step": 9407
    },
    {
      "action_loss_avg": 0.0197685569524765,
      "epoch": 3.5884857177734375e-05,
      "step": 9407
    },
    {
      "epoch": 3.5884857177734375e-05,
      "step": 9407,
      "training_step_time": 9.956636905670166
    },
    {
      "epoch": 3.5884857177734375e-05,
      "model_forward_time": 3.146929979324341,
      "step": 9407
    },
    {
      "epoch": 3.5884857177734375e-05,
      "step": 9407,
      "training_step_time": 10.327683210372925
    },
    {
      "epoch": 3.5888671875e-05,
      "model_forward_time": 2.4435908794403076,
      "step": 9408
    },
    {
      "epoch": 3.5888671875e-05,
      "step": 9408,
      "training_step_time": 7.473690748214722
    },
    {
      "epoch": 3.5888671875e-05,
      "model_forward_time": 3.421583890914917,
      "step": 9408
    },
    {
      "epoch": 3.5888671875e-05,
      "step": 9408,
      "training_step_time": 10.201395273208618
    },
    {
      "epoch": 3.5888671875e-05,
      "model_forward_time": 3.143359899520874,
      "step": 9408
    },
    {
      "epoch": 3.5888671875e-05,
      "step": 9408,
      "training_step_time": 9.92810606956482
    },
    {
      "epoch": 3.5888671875e-05,
      "model_forward_time": 3.1380186080932617,
      "step": 9408
    },
    {
      "epoch": 3.5888671875e-05,
      "step": 9408,
      "training_step_time": 10.305005311965942
    },
    {
      "epoch": 3.589248657226563e-05,
      "model_forward_time": 3.150484323501587,
      "step": 9409
    },
    {
      "epoch": 3.589248657226563e-05,
      "step": 9409,
      "training_step_time": 9.929026126861572
    },
    {
      "epoch": 3.589248657226563e-05,
      "model_forward_time": 3.1349856853485107,
      "step": 9409
    },
    {
      "epoch": 3.589248657226563e-05,
      "step": 9409,
      "training_step_time": 9.914986371994019
    },
    {
      "epoch": 3.589248657226563e-05,
      "model_forward_time": 3.4466774463653564,
      "step": 9409
    },
    {
      "epoch": 3.589248657226563e-05,
      "step": 9409,
      "training_step_time": 10.216623783111572
    },
    {
      "epoch": 3.589248657226563e-05,
      "model_forward_time": 3.1333985328674316,
      "step": 9409
    },
    {
      "epoch": 3.589248657226563e-05,
      "step": 9409,
      "training_step_time": 10.263664484024048
    },
    {
      "epoch": 3.589630126953125e-05,
      "grad_norm": 0.2954578101634979,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 9410
    },
    {
      "epoch": 3.589630126953125e-05,
      "model_forward_time": 2.44620418548584,
      "step": 9410
    },
    {
      "dynamics_loss_avg": 0.06464981706812978,
      "epoch": 3.589630126953125e-05,
      "step": 9410
    },
    {
      "action_loss_avg": 0.01923958449624479,
      "epoch": 3.589630126953125e-05,
      "step": 9410
    },
    {
      "epoch": 3.589630126953125e-05,
      "step": 9410,
      "training_step_time": 7.477715730667114
    },
    {
      "epoch": 3.589630126953125e-05,
      "model_forward_time": 3.1312053203582764,
      "step": 9410
    },
    {
      "epoch": 3.589630126953125e-05,
      "step": 9410,
      "training_step_time": 9.904384851455688
    },
    {
      "epoch": 3.589630126953125e-05,
      "model_forward_time": 3.1317803859710693,
      "step": 9410
    },
    {
      "epoch": 3.589630126953125e-05,
      "step": 9410,
      "training_step_time": 9.91364073753357
    },
    {
      "epoch": 3.589630126953125e-05,
      "model_forward_time": 3.4422991275787354,
      "step": 9410
    },
    {
      "epoch": 3.589630126953125e-05,
      "step": 9410,
      "training_step_time": 10.310473680496216
    },
    {
      "epoch": 3.5900115966796873e-05,
      "model_forward_time": 3.1495771408081055,
      "step": 9411
    },
    {
      "epoch": 3.5900115966796873e-05,
      "step": 9411,
      "training_step_time": 9.93347716331482
    },
    {
      "epoch": 3.5900115966796873e-05,
      "model_forward_time": 3.1492812633514404,
      "step": 9411
    },
    {
      "epoch": 3.5900115966796873e-05,
      "step": 9411,
      "training_step_time": 9.91843557357788
    },
    {
      "epoch": 3.5900115966796873e-05,
      "model_forward_time": 3.1440272331237793,
      "step": 9411
    },
    {
      "epoch": 3.5900115966796873e-05,
      "step": 9411,
      "training_step_time": 9.915006637573242
    },
    {
      "epoch": 3.5900115966796873e-05,
      "model_forward_time": 3.1502256393432617,
      "step": 9411
    },
    {
      "epoch": 3.5900115966796873e-05,
      "step": 9411,
      "training_step_time": 10.296704292297363
    },
    {
      "epoch": 3.59039306640625e-05,
      "model_forward_time": 2.838064193725586,
      "step": 9412
    },
    {
      "epoch": 3.59039306640625e-05,
      "step": 9412,
      "training_step_time": 7.8827948570251465
    },
    {
      "epoch": 3.59039306640625e-05,
      "model_forward_time": 3.1468899250030518,
      "step": 9412
    },
    {
      "epoch": 3.59039306640625e-05,
      "step": 9412,
      "training_step_time": 9.922544002532959
    },
    {
      "epoch": 3.59039306640625e-05,
      "model_forward_time": 3.145285129547119,
      "step": 9412
    },
    {
      "dynamics_loss_avg": 0.039955836907029155,
      "epoch": 3.59039306640625e-05,
      "step": 9412
    },
    {
      "action_loss_avg": 0.016562627721577884,
      "epoch": 3.59039306640625e-05,
      "step": 9412
    },
    {
      "epoch": 3.59039306640625e-05,
      "step": 9412,
      "training_step_time": 9.925454378128052
    },
    {
      "epoch": 3.59039306640625e-05,
      "model_forward_time": 2.4390063285827637,
      "step": 9412
    },
    {
      "epoch": 3.59039306640625e-05,
      "step": 9412,
      "training_step_time": 10.290323257446289
    },
    {
      "epoch": 3.5907745361328126e-05,
      "model_forward_time": 3.1461474895477295,
      "step": 9413
    },
    {
      "epoch": 3.5907745361328126e-05,
      "step": 9413,
      "training_step_time": 9.916725873947144
    },
    {
      "epoch": 3.5907745361328126e-05,
      "model_forward_time": 3.4160003662109375,
      "step": 9413
    },
    {
      "epoch": 3.5907745361328126e-05,
      "step": 9413,
      "training_step_time": 10.191404819488525
    },
    {
      "epoch": 3.5907745361328126e-05,
      "model_forward_time": 3.1411502361297607,
      "step": 9413
    },
    {
      "epoch": 3.5907745361328126e-05,
      "step": 9413,
      "training_step_time": 9.912931680679321
    },
    {
      "epoch": 3.5907745361328126e-05,
      "model_forward_time": 3.1411261558532715,
      "step": 9413
    },
    {
      "epoch": 3.5907745361328126e-05,
      "step": 9413,
      "training_step_time": 10.320524454116821
    },
    {
      "epoch": 3.591156005859375e-05,
      "model_forward_time": 3.151313304901123,
      "step": 9414
    },
    {
      "epoch": 3.591156005859375e-05,
      "step": 9414,
      "training_step_time": 9.927943229675293
    },
    {
      "epoch": 3.591156005859375e-05,
      "model_forward_time": 3.1376137733459473,
      "step": 9414
    },
    {
      "epoch": 3.591156005859375e-05,
      "step": 9414,
      "training_step_time": 10.222314596176147
    },
    {
      "epoch": 3.591156005859375e-05,
      "model_forward_time": 2.4391250610351562,
      "step": 9414
    },
    {
      "epoch": 3.591156005859375e-05,
      "step": 9414,
      "training_step_time": 7.463693618774414
    },
    {
      "epoch": 3.591156005859375e-05,
      "model_forward_time": 3.1427180767059326,
      "step": 9414
    },
    {
      "epoch": 3.591156005859375e-05,
      "step": 9414,
      "training_step_time": 10.28065276145935
    },
    {
      "epoch": 3.591537475585937e-05,
      "model_forward_time": 3.1551570892333984,
      "step": 9415
    },
    {
      "dynamics_loss_avg": 0.053458320908248425,
      "epoch": 3.591537475585937e-05,
      "step": 9415
    },
    {
      "action_loss_avg": 0.019033264927566053,
      "epoch": 3.591537475585937e-05,
      "step": 9415
    },
    {
      "epoch": 3.591537475585937e-05,
      "step": 9415,
      "training_step_time": 9.929218053817749
    },
    {
      "epoch": 3.591537475585937e-05,
      "model_forward_time": 2.430708408355713,
      "step": 9415
    },
    {
      "epoch": 3.591537475585937e-05,
      "step": 9415,
      "training_step_time": 7.455019474029541
    },
    {
      "epoch": 3.591537475585937e-05,
      "model_forward_time": 3.1336352825164795,
      "step": 9415
    },
    {
      "epoch": 3.591537475585937e-05,
      "step": 9415,
      "training_step_time": 9.898898601531982
    },
    {
      "epoch": 3.591537475585937e-05,
      "model_forward_time": 3.4021871089935303,
      "step": 9415
    },
    {
      "epoch": 3.591537475585937e-05,
      "step": 9415,
      "training_step_time": 10.262862205505371
    },
    {
      "epoch": 3.5919189453125e-05,
      "model_forward_time": 3.1563022136688232,
      "step": 9416
    },
    {
      "epoch": 3.5919189453125e-05,
      "step": 9416,
      "training_step_time": 9.923441886901855
    },
    {
      "epoch": 3.5919189453125e-05,
      "model_forward_time": 3.1373753547668457,
      "step": 9416
    },
    {
      "epoch": 3.5919189453125e-05,
      "step": 9416,
      "training_step_time": 9.90225601196289
    },
    {
      "epoch": 3.5919189453125e-05,
      "model_forward_time": 3.131767511367798,
      "step": 9416
    },
    {
      "epoch": 3.5919189453125e-05,
      "step": 9416,
      "training_step_time": 9.893529891967773
    },
    {
      "epoch": 3.5919189453125e-05,
      "model_forward_time": 3.129971742630005,
      "step": 9416
    },
    {
      "epoch": 3.5919189453125e-05,
      "step": 9416,
      "training_step_time": 10.294448375701904
    },
    {
      "epoch": 3.5923004150390624e-05,
      "model_forward_time": 3.1538808345794678,
      "step": 9417
    },
    {
      "epoch": 3.5923004150390624e-05,
      "step": 9417,
      "training_step_time": 9.923235654830933
    },
    {
      "epoch": 3.5923004150390624e-05,
      "model_forward_time": 3.1380724906921387,
      "step": 9417
    },
    {
      "epoch": 3.5923004150390624e-05,
      "step": 9417,
      "training_step_time": 9.904367923736572
    },
    {
      "epoch": 3.5923004150390624e-05,
      "model_forward_time": 3.1387217044830322,
      "step": 9417
    },
    {
      "dynamics_loss_avg": 0.057769822888076305,
      "epoch": 3.5923004150390624e-05,
      "step": 9417
    },
    {
      "action_loss_avg": 0.022009872691705824,
      "epoch": 3.5923004150390624e-05,
      "step": 9417
    },
    {
      "epoch": 3.5923004150390624e-05,
      "step": 9417,
      "training_step_time": 9.907972812652588
    },
    {
      "epoch": 3.5923004150390624e-05,
      "model_forward_time": 3.137986421585083,
      "step": 9417
    },
    {
      "epoch": 3.5923004150390624e-05,
      "step": 9417,
      "training_step_time": 10.010385274887085
    },
    {
      "epoch": 3.592681884765625e-05,
      "model_forward_time": 3.149852991104126,
      "step": 9418
    },
    {
      "epoch": 3.592681884765625e-05,
      "step": 9418,
      "training_step_time": 10.212173700332642
    },
    {
      "epoch": 3.592681884765625e-05,
      "model_forward_time": 3.1398799419403076,
      "step": 9418
    },
    {
      "epoch": 3.592681884765625e-05,
      "step": 9418,
      "training_step_time": 9.905494689941406
    },
    {
      "epoch": 3.592681884765625e-05,
      "model_forward_time": 3.14029598236084,
      "step": 9418
    },
    {
      "epoch": 3.592681884765625e-05,
      "step": 9418,
      "training_step_time": 9.90817666053772
    },
    {
      "epoch": 3.592681884765625e-05,
      "model_forward_time": 3.1364378929138184,
      "step": 9418
    },
    {
      "epoch": 3.592681884765625e-05,
      "step": 9418,
      "training_step_time": 10.332576513290405
    },
    {
      "epoch": 3.593063354492188e-05,
      "model_forward_time": 3.160372495651245,
      "step": 9419
    },
    {
      "epoch": 3.593063354492188e-05,
      "step": 9419,
      "training_step_time": 9.939271211624146
    },
    {
      "epoch": 3.593063354492188e-05,
      "model_forward_time": 3.404506206512451,
      "step": 9419
    },
    {
      "epoch": 3.593063354492188e-05,
      "step": 9419,
      "training_step_time": 10.173956632614136
    },
    {
      "epoch": 3.593063354492188e-05,
      "model_forward_time": 2.433277130126953,
      "step": 9419
    },
    {
      "epoch": 3.593063354492188e-05,
      "step": 9419,
      "training_step_time": 7.456686973571777
    },
    {
      "epoch": 3.593063354492188e-05,
      "model_forward_time": 3.1355087757110596,
      "step": 9419
    },
    {
      "epoch": 3.593063354492188e-05,
      "step": 9419,
      "training_step_time": 10.280635595321655
    },
    {
      "epoch": 3.59344482421875e-05,
      "grad_norm": 0.4184696078300476,
      "learning_rate": 1e-05,
      "loss": 0.0708,
      "step": 9420
    },
    {
      "epoch": 3.59344482421875e-05,
      "model_forward_time": 3.1430306434631348,
      "step": 9420
    },
    {
      "dynamics_loss_avg": 0.049291244614869355,
      "epoch": 3.59344482421875e-05,
      "step": 9420
    },
    {
      "action_loss_avg": 0.023642372945323585,
      "epoch": 3.59344482421875e-05,
      "step": 9420
    },
    {
      "epoch": 3.59344482421875e-05,
      "step": 9420,
      "training_step_time": 9.914023876190186
    },
    {
      "epoch": 3.59344482421875e-05,
      "model_forward_time": 3.1398262977600098,
      "step": 9420
    },
    {
      "epoch": 3.59344482421875e-05,
      "step": 9420,
      "training_step_time": 9.906399250030518
    },
    {
      "epoch": 3.59344482421875e-05,
      "model_forward_time": 3.137397527694702,
      "step": 9420
    },
    {
      "epoch": 3.59344482421875e-05,
      "step": 9420,
      "training_step_time": 9.913089036941528
    },
    {
      "epoch": 3.59344482421875e-05,
      "model_forward_time": 3.3864972591400146,
      "step": 9420
    },
    {
      "epoch": 3.59344482421875e-05,
      "step": 9420,
      "training_step_time": 10.248127222061157
    },
    {
      "epoch": 3.593826293945312e-05,
      "model_forward_time": 2.4517149925231934,
      "step": 9421
    },
    {
      "epoch": 3.593826293945312e-05,
      "step": 9421,
      "training_step_time": 7.478880882263184
    },
    {
      "epoch": 3.593826293945312e-05,
      "model_forward_time": 3.139691114425659,
      "step": 9421
    },
    {
      "epoch": 3.593826293945312e-05,
      "step": 9421,
      "training_step_time": 9.90533995628357
    },
    {
      "epoch": 3.593826293945312e-05,
      "model_forward_time": 3.135554790496826,
      "step": 9421
    },
    {
      "epoch": 3.593826293945312e-05,
      "step": 9421,
      "training_step_time": 9.899144411087036
    },
    {
      "epoch": 3.593826293945312e-05,
      "model_forward_time": 3.14001727104187,
      "step": 9421
    },
    {
      "epoch": 3.593826293945312e-05,
      "step": 9421,
      "training_step_time": 10.305323362350464
    },
    {
      "epoch": 3.594207763671875e-05,
      "model_forward_time": 2.4514846801757812,
      "step": 9422
    },
    {
      "epoch": 3.594207763671875e-05,
      "step": 9422,
      "training_step_time": 7.474382400512695
    },
    {
      "epoch": 3.594207763671875e-05,
      "model_forward_time": 3.1365227699279785,
      "step": 9422
    },
    {
      "epoch": 3.594207763671875e-05,
      "step": 9422,
      "training_step_time": 9.906608819961548
    },
    {
      "epoch": 3.594207763671875e-05,
      "model_forward_time": 3.1371891498565674,
      "step": 9422
    },
    {
      "dynamics_loss_avg": 0.05672554764896631,
      "epoch": 3.594207763671875e-05,
      "step": 9422
    },
    {
      "action_loss_avg": 0.027404855331406,
      "epoch": 3.594207763671875e-05,
      "step": 9422
    },
    {
      "epoch": 3.594207763671875e-05,
      "step": 9422,
      "training_step_time": 9.9074547290802
    },
    {
      "epoch": 3.594207763671875e-05,
      "model_forward_time": 3.1421008110046387,
      "step": 9422
    },
    {
      "epoch": 3.594207763671875e-05,
      "step": 9422,
      "training_step_time": 10.308627605438232
    },
    {
      "epoch": 3.5945892333984375e-05,
      "model_forward_time": 2.450045347213745,
      "step": 9423
    },
    {
      "epoch": 3.5945892333984375e-05,
      "step": 9423,
      "training_step_time": 7.485773801803589
    },
    {
      "epoch": 3.5945892333984375e-05,
      "model_forward_time": 3.440626382827759,
      "step": 9423
    },
    {
      "epoch": 3.5945892333984375e-05,
      "step": 9423,
      "training_step_time": 10.212398767471313
    },
    {
      "epoch": 3.5945892333984375e-05,
      "model_forward_time": 3.145421028137207,
      "step": 9423
    },
    {
      "epoch": 3.5945892333984375e-05,
      "step": 9423,
      "training_step_time": 9.923466682434082
    },
    {
      "epoch": 3.5945892333984375e-05,
      "model_forward_time": 3.1450083255767822,
      "step": 9423
    },
    {
      "epoch": 3.5945892333984375e-05,
      "step": 9423,
      "training_step_time": 10.258748054504395
    },
    {
      "epoch": 3.594970703125e-05,
      "model_forward_time": 3.1523969173431396,
      "step": 9424
    },
    {
      "epoch": 3.594970703125e-05,
      "step": 9424,
      "training_step_time": 9.936656951904297
    },
    {
      "epoch": 3.594970703125e-05,
      "model_forward_time": 3.142232656478882,
      "step": 9424
    },
    {
      "epoch": 3.594970703125e-05,
      "step": 9424,
      "training_step_time": 10.247998476028442
    },
    {
      "epoch": 3.594970703125e-05,
      "model_forward_time": 3.14554500579834,
      "step": 9424
    },
    {
      "epoch": 3.594970703125e-05,
      "step": 9424,
      "training_step_time": 9.928574323654175
    },
    {
      "epoch": 3.594970703125e-05,
      "model_forward_time": 3.149458646774292,
      "step": 9424
    },
    {
      "epoch": 3.594970703125e-05,
      "step": 9424,
      "training_step_time": 10.31888461112976
    },
    {
      "epoch": 3.595352172851563e-05,
      "model_forward_time": 3.1663131713867188,
      "step": 9425
    },
    {
      "dynamics_loss_avg": 0.060040372610092166,
      "epoch": 3.595352172851563e-05,
      "step": 9425
    },
    {
      "action_loss_avg": 0.0225572865922004,
      "epoch": 3.595352172851563e-05,
      "step": 9425
    },
    {
      "epoch": 3.595352172851563e-05,
      "step": 9425,
      "training_step_time": 9.959211587905884
    },
    {
      "epoch": 3.595352172851563e-05,
      "model_forward_time": 3.1448569297790527,
      "step": 9425
    },
    {
      "epoch": 3.595352172851563e-05,
      "step": 9425,
      "training_step_time": 9.920318126678467
    },
    {
      "epoch": 3.595352172851563e-05,
      "model_forward_time": 3.1481845378875732,
      "step": 9425
    },
    {
      "epoch": 3.595352172851563e-05,
      "step": 9425,
      "training_step_time": 10.231083393096924
    },
    {
      "epoch": 3.595352172851563e-05,
      "model_forward_time": 3.1504082679748535,
      "step": 9425
    },
    {
      "epoch": 3.595352172851563e-05,
      "step": 9425,
      "training_step_time": 10.037303686141968
    },
    {
      "epoch": 3.595733642578125e-05,
      "model_forward_time": 3.1711885929107666,
      "step": 9426
    },
    {
      "epoch": 3.595733642578125e-05,
      "step": 9426,
      "training_step_time": 9.95496392250061
    },
    {
      "epoch": 3.595733642578125e-05,
      "model_forward_time": 3.1427783966064453,
      "step": 9426
    },
    {
      "epoch": 3.595733642578125e-05,
      "step": 9426,
      "training_step_time": 9.914143562316895
    },
    {
      "epoch": 3.595733642578125e-05,
      "model_forward_time": 3.1474251747131348,
      "step": 9426
    },
    {
      "epoch": 3.595733642578125e-05,
      "step": 9426,
      "training_step_time": 9.916025400161743
    },
    {
      "epoch": 3.595733642578125e-05,
      "model_forward_time": 3.4695546627044678,
      "step": 9426
    },
    {
      "epoch": 3.595733642578125e-05,
      "step": 9426,
      "training_step_time": 10.339954376220703
    },
    {
      "epoch": 3.5961151123046873e-05,
      "model_forward_time": 3.155287981033325,
      "step": 9427
    },
    {
      "epoch": 3.5961151123046873e-05,
      "step": 9427,
      "training_step_time": 9.924322605133057
    },
    {
      "epoch": 3.5961151123046873e-05,
      "model_forward_time": 2.4390249252319336,
      "step": 9427
    },
    {
      "epoch": 3.5961151123046873e-05,
      "step": 9427,
      "training_step_time": 7.466095209121704
    },
    {
      "epoch": 3.5961151123046873e-05,
      "model_forward_time": 2.432344675064087,
      "step": 9427
    },
    {
      "dynamics_loss_avg": 0.06238704510033131,
      "epoch": 3.5961151123046873e-05,
      "step": 9427
    },
    {
      "action_loss_avg": 0.023188345646485687,
      "epoch": 3.5961151123046873e-05,
      "step": 9427
    },
    {
      "epoch": 3.5961151123046873e-05,
      "step": 9427,
      "training_step_time": 7.462259531021118
    },
    {
      "epoch": 3.5961151123046873e-05,
      "model_forward_time": 3.134795904159546,
      "step": 9427
    },
    {
      "epoch": 3.5961151123046873e-05,
      "step": 9427,
      "training_step_time": 10.33584213256836
    },
    {
      "epoch": 3.59649658203125e-05,
      "model_forward_time": 3.1516036987304688,
      "step": 9428
    },
    {
      "epoch": 3.59649658203125e-05,
      "step": 9428,
      "training_step_time": 9.918723821640015
    },
    {
      "epoch": 3.59649658203125e-05,
      "model_forward_time": 3.3932383060455322,
      "step": 9428
    },
    {
      "epoch": 3.59649658203125e-05,
      "step": 9428,
      "training_step_time": 10.162054538726807
    },
    {
      "epoch": 3.59649658203125e-05,
      "model_forward_time": 3.144975423812866,
      "step": 9428
    },
    {
      "epoch": 3.59649658203125e-05,
      "step": 9428,
      "training_step_time": 9.911765336990356
    },
    {
      "epoch": 3.59649658203125e-05,
      "model_forward_time": 2.4353740215301514,
      "step": 9428
    },
    {
      "epoch": 3.59649658203125e-05,
      "step": 9428,
      "training_step_time": 10.360315322875977
    },
    {
      "epoch": 3.5968780517578126e-05,
      "model_forward_time": 3.1568498611450195,
      "step": 9429
    },
    {
      "epoch": 3.5968780517578126e-05,
      "step": 9429,
      "training_step_time": 9.924119710922241
    },
    {
      "epoch": 3.5968780517578126e-05,
      "model_forward_time": 3.142300605773926,
      "step": 9429
    },
    {
      "epoch": 3.5968780517578126e-05,
      "step": 9429,
      "training_step_time": 9.908638715744019
    },
    {
      "epoch": 3.5968780517578126e-05,
      "model_forward_time": 3.463480234146118,
      "step": 9429
    },
    {
      "epoch": 3.5968780517578126e-05,
      "step": 9429,
      "training_step_time": 10.232615947723389
    },
    {
      "epoch": 3.5968780517578126e-05,
      "model_forward_time": 3.1399059295654297,
      "step": 9429
    },
    {
      "epoch": 3.5968780517578126e-05,
      "step": 9429,
      "training_step_time": 10.277553796768188
    },
    {
      "epoch": 3.597259521484375e-05,
      "grad_norm": 0.32084307074546814,
      "learning_rate": 1e-05,
      "loss": 0.0796,
      "step": 9430
    },
    {
      "epoch": 3.597259521484375e-05,
      "model_forward_time": 3.1480584144592285,
      "step": 9430
    },
    {
      "dynamics_loss_avg": 0.07282953076064587,
      "epoch": 3.597259521484375e-05,
      "step": 9430
    },
    {
      "action_loss_avg": 0.020905601140111685,
      "epoch": 3.597259521484375e-05,
      "step": 9430
    },
    {
      "epoch": 3.597259521484375e-05,
      "step": 9430,
      "training_step_time": 9.924054861068726
    },
    {
      "epoch": 3.597259521484375e-05,
      "model_forward_time": 3.144439220428467,
      "step": 9430
    },
    {
      "epoch": 3.597259521484375e-05,
      "step": 9430,
      "training_step_time": 9.919948816299438
    },
    {
      "epoch": 3.597259521484375e-05,
      "model_forward_time": 3.4319772720336914,
      "step": 9430
    },
    {
      "epoch": 3.597259521484375e-05,
      "step": 9430,
      "training_step_time": 10.203226804733276
    },
    {
      "epoch": 3.597259521484375e-05,
      "model_forward_time": 3.1428396701812744,
      "step": 9430
    },
    {
      "epoch": 3.597259521484375e-05,
      "step": 9430,
      "training_step_time": 10.306206464767456
    },
    {
      "epoch": 3.597640991210937e-05,
      "model_forward_time": 3.1526646614074707,
      "step": 9431
    },
    {
      "epoch": 3.597640991210937e-05,
      "step": 9431,
      "training_step_time": 9.941091299057007
    },
    {
      "epoch": 3.597640991210937e-05,
      "model_forward_time": 3.142523765563965,
      "step": 9431
    },
    {
      "epoch": 3.597640991210937e-05,
      "step": 9431,
      "training_step_time": 9.913483142852783
    },
    {
      "epoch": 3.597640991210937e-05,
      "model_forward_time": 2.4342570304870605,
      "step": 9431
    },
    {
      "epoch": 3.597640991210937e-05,
      "step": 9431,
      "training_step_time": 7.466077089309692
    },
    {
      "epoch": 3.597640991210937e-05,
      "model_forward_time": 3.138676881790161,
      "step": 9431
    },
    {
      "epoch": 3.597640991210937e-05,
      "step": 9431,
      "training_step_time": 10.038341522216797
    },
    {
      "epoch": 3.5980224609375e-05,
      "model_forward_time": 3.4558701515197754,
      "step": 9432
    },
    {
      "epoch": 3.5980224609375e-05,
      "step": 9432,
      "training_step_time": 10.23121976852417
    },
    {
      "epoch": 3.5980224609375e-05,
      "model_forward_time": 3.1462793350219727,
      "step": 9432
    },
    {
      "epoch": 3.5980224609375e-05,
      "step": 9432,
      "training_step_time": 9.922916650772095
    },
    {
      "epoch": 3.5980224609375e-05,
      "model_forward_time": 3.142834186553955,
      "step": 9432
    },
    {
      "dynamics_loss_avg": 0.03939152965322137,
      "epoch": 3.5980224609375e-05,
      "step": 9432
    },
    {
      "action_loss_avg": 0.013827394414693116,
      "epoch": 3.5980224609375e-05,
      "step": 9432
    },
    {
      "epoch": 3.5980224609375e-05,
      "step": 9432,
      "training_step_time": 9.920759677886963
    },
    {
      "epoch": 3.5980224609375e-05,
      "model_forward_time": 3.1470787525177,
      "step": 9432
    },
    {
      "epoch": 3.5980224609375e-05,
      "step": 9432,
      "training_step_time": 10.222470998764038
    },
    {
      "epoch": 3.5984039306640624e-05,
      "model_forward_time": 3.490490436553955,
      "step": 9433
    },
    {
      "epoch": 3.5984039306640624e-05,
      "step": 9433,
      "training_step_time": 10.265082836151123
    },
    {
      "epoch": 3.5984039306640624e-05,
      "model_forward_time": 3.1818480491638184,
      "step": 9433
    },
    {
      "epoch": 3.5984039306640624e-05,
      "step": 9433,
      "training_step_time": 9.962408065795898
    },
    {
      "epoch": 3.5984039306640624e-05,
      "model_forward_time": 2.470978021621704,
      "step": 9433
    },
    {
      "epoch": 3.5984039306640624e-05,
      "step": 9433,
      "training_step_time": 7.505671501159668
    },
    {
      "epoch": 3.5984039306640624e-05,
      "model_forward_time": 3.1486546993255615,
      "step": 9433
    },
    {
      "epoch": 3.5984039306640624e-05,
      "step": 9433,
      "training_step_time": 10.29067325592041
    },
    {
      "epoch": 3.598785400390625e-05,
      "model_forward_time": 3.159569263458252,
      "step": 9434
    },
    {
      "epoch": 3.598785400390625e-05,
      "step": 9434,
      "training_step_time": 9.936182498931885
    },
    {
      "epoch": 3.598785400390625e-05,
      "model_forward_time": 3.144286870956421,
      "step": 9434
    },
    {
      "epoch": 3.598785400390625e-05,
      "step": 9434,
      "training_step_time": 10.208617210388184
    },
    {
      "epoch": 3.598785400390625e-05,
      "model_forward_time": 3.1479039192199707,
      "step": 9434
    },
    {
      "epoch": 3.598785400390625e-05,
      "step": 9434,
      "training_step_time": 9.922832489013672
    },
    {
      "epoch": 3.598785400390625e-05,
      "model_forward_time": 3.147023916244507,
      "step": 9434
    },
    {
      "epoch": 3.598785400390625e-05,
      "step": 9434,
      "training_step_time": 10.311713933944702
    },
    {
      "epoch": 3.599166870117188e-05,
      "model_forward_time": 3.1613504886627197,
      "step": 9435
    },
    {
      "dynamics_loss_avg": 0.06661175964400172,
      "epoch": 3.599166870117188e-05,
      "step": 9435
    },
    {
      "action_loss_avg": 0.021083188802003862,
      "epoch": 3.599166870117188e-05,
      "step": 9435
    },
    {
      "epoch": 3.599166870117188e-05,
      "step": 9435,
      "training_step_time": 9.94190001487732
    },
    {
      "epoch": 3.599166870117188e-05,
      "model_forward_time": 3.14072585105896,
      "step": 9435
    },
    {
      "epoch": 3.599166870117188e-05,
      "step": 9435,
      "training_step_time": 9.915756940841675
    },
    {
      "epoch": 3.599166870117188e-05,
      "model_forward_time": 3.4078361988067627,
      "step": 9435
    },
    {
      "epoch": 3.599166870117188e-05,
      "step": 9435,
      "training_step_time": 10.184308767318726
    },
    {
      "epoch": 3.599166870117188e-05,
      "model_forward_time": 3.148810625076294,
      "step": 9435
    },
    {
      "epoch": 3.599166870117188e-05,
      "step": 9435,
      "training_step_time": 10.291016578674316
    },
    {
      "epoch": 3.59954833984375e-05,
      "model_forward_time": 3.1559722423553467,
      "step": 9436
    },
    {
      "epoch": 3.59954833984375e-05,
      "step": 9436,
      "training_step_time": 9.935922622680664
    },
    {
      "epoch": 3.59954833984375e-05,
      "model_forward_time": 3.1439311504364014,
      "step": 9436
    },
    {
      "epoch": 3.59954833984375e-05,
      "step": 9436,
      "training_step_time": 9.921655893325806
    },
    {
      "epoch": 3.59954833984375e-05,
      "model_forward_time": 3.143425941467285,
      "step": 9436
    },
    {
      "epoch": 3.59954833984375e-05,
      "step": 9436,
      "training_step_time": 9.920414447784424
    },
    {
      "epoch": 3.59954833984375e-05,
      "model_forward_time": 3.4325690269470215,
      "step": 9436
    },
    {
      "epoch": 3.59954833984375e-05,
      "step": 9436,
      "training_step_time": 10.30689001083374
    },
    {
      "epoch": 3.599929809570312e-05,
      "model_forward_time": 3.1623408794403076,
      "step": 9437
    },
    {
      "epoch": 3.599929809570312e-05,
      "step": 9437,
      "training_step_time": 9.942391872406006
    },
    {
      "epoch": 3.599929809570312e-05,
      "model_forward_time": 3.140883684158325,
      "step": 9437
    },
    {
      "epoch": 3.599929809570312e-05,
      "step": 9437,
      "training_step_time": 9.92158818244934
    },
    {
      "epoch": 3.599929809570312e-05,
      "model_forward_time": 3.1466872692108154,
      "step": 9437
    },
    {
      "dynamics_loss_avg": 0.06099763866513967,
      "epoch": 3.599929809570312e-05,
      "step": 9437
    },
    {
      "action_loss_avg": 0.023056007083505393,
      "epoch": 3.599929809570312e-05,
      "step": 9437
    },
    {
      "epoch": 3.599929809570312e-05,
      "step": 9437,
      "training_step_time": 9.933964252471924
    },
    {
      "epoch": 3.599929809570312e-05,
      "model_forward_time": 2.435302495956421,
      "step": 9437
    },
    {
      "epoch": 3.599929809570312e-05,
      "step": 9437,
      "training_step_time": 10.313541173934937
    },
    {
      "epoch": 3.600311279296875e-05,
      "model_forward_time": 3.1497435569763184,
      "step": 9438
    },
    {
      "epoch": 3.600311279296875e-05,
      "step": 9438,
      "training_step_time": 10.212819814682007
    },
    {
      "epoch": 3.600311279296875e-05,
      "model_forward_time": 3.1377298831939697,
      "step": 9438
    },
    {
      "epoch": 3.600311279296875e-05,
      "step": 9438,
      "training_step_time": 9.909401893615723
    },
    {
      "epoch": 3.600311279296875e-05,
      "model_forward_time": 3.138087272644043,
      "step": 9438
    },
    {
      "epoch": 3.600311279296875e-05,
      "step": 9438,
      "training_step_time": 9.915132522583008
    },
    {
      "epoch": 3.600311279296875e-05,
      "model_forward_time": 3.1327407360076904,
      "step": 9438
    },
    {
      "epoch": 3.600311279296875e-05,
      "step": 9438,
      "training_step_time": 10.322556972503662
    },
    {
      "epoch": 3.6006927490234375e-05,
      "model_forward_time": 3.1510274410247803,
      "step": 9439
    },
    {
      "epoch": 3.6006927490234375e-05,
      "step": 9439,
      "training_step_time": 9.923871040344238
    },
    {
      "epoch": 3.6006927490234375e-05,
      "model_forward_time": 2.4299588203430176,
      "step": 9439
    },
    {
      "epoch": 3.6006927490234375e-05,
      "step": 9439,
      "training_step_time": 7.45613169670105
    },
    {
      "epoch": 3.6006927490234375e-05,
      "model_forward_time": 3.4018044471740723,
      "step": 9439
    },
    {
      "epoch": 3.6006927490234375e-05,
      "step": 9439,
      "training_step_time": 10.174780368804932
    },
    {
      "epoch": 3.6006927490234375e-05,
      "model_forward_time": 3.136474132537842,
      "step": 9439
    },
    {
      "epoch": 3.6006927490234375e-05,
      "step": 9439,
      "training_step_time": 10.267598628997803
    },
    {
      "epoch": 3.60107421875e-05,
      "grad_norm": 0.41108864545822144,
      "learning_rate": 1e-05,
      "loss": 0.0748,
      "step": 9440
    },
    {
      "epoch": 3.60107421875e-05,
      "model_forward_time": 3.1561033725738525,
      "step": 9440
    },
    {
      "dynamics_loss_avg": 0.03791881417855621,
      "epoch": 3.60107421875e-05,
      "step": 9440
    },
    {
      "action_loss_avg": 0.01166588980704546,
      "epoch": 3.60107421875e-05,
      "step": 9440
    },
    {
      "epoch": 3.60107421875e-05,
      "step": 9440,
      "training_step_time": 9.938305139541626
    },
    {
      "epoch": 3.60107421875e-05,
      "model_forward_time": 3.1390202045440674,
      "step": 9440
    },
    {
      "epoch": 3.60107421875e-05,
      "step": 9440,
      "training_step_time": 9.910971403121948
    },
    {
      "epoch": 3.60107421875e-05,
      "model_forward_time": 3.1363842487335205,
      "step": 9440
    },
    {
      "epoch": 3.60107421875e-05,
      "step": 9440,
      "training_step_time": 10.206903457641602
    },
    {
      "epoch": 3.60107421875e-05,
      "model_forward_time": 3.137615203857422,
      "step": 9440
    },
    {
      "epoch": 3.60107421875e-05,
      "step": 9440,
      "training_step_time": 10.247390031814575
    },
    {
      "epoch": 3.601455688476563e-05,
      "model_forward_time": 3.1555233001708984,
      "step": 9441
    },
    {
      "epoch": 3.601455688476563e-05,
      "step": 9441,
      "training_step_time": 9.93151330947876
    },
    {
      "epoch": 3.601455688476563e-05,
      "model_forward_time": 3.1366872787475586,
      "step": 9441
    },
    {
      "epoch": 3.601455688476563e-05,
      "step": 9441,
      "training_step_time": 9.918372631072998
    },
    {
      "epoch": 3.601455688476563e-05,
      "model_forward_time": 3.1402480602264404,
      "step": 9441
    },
    {
      "epoch": 3.601455688476563e-05,
      "step": 9441,
      "training_step_time": 9.920414924621582
    },
    {
      "epoch": 3.601455688476563e-05,
      "model_forward_time": 3.4075870513916016,
      "step": 9441
    },
    {
      "epoch": 3.601455688476563e-05,
      "step": 9441,
      "training_step_time": 10.280932664871216
    },
    {
      "epoch": 3.601837158203125e-05,
      "model_forward_time": 3.143186330795288,
      "step": 9442
    },
    {
      "epoch": 3.601837158203125e-05,
      "step": 9442,
      "training_step_time": 9.922345399856567
    },
    {
      "epoch": 3.601837158203125e-05,
      "model_forward_time": 3.139497995376587,
      "step": 9442
    },
    {
      "epoch": 3.601837158203125e-05,
      "step": 9442,
      "training_step_time": 9.91730546951294
    },
    {
      "epoch": 3.601837158203125e-05,
      "model_forward_time": 3.1401360034942627,
      "step": 9442
    },
    {
      "dynamics_loss_avg": 0.07163034547120332,
      "epoch": 3.601837158203125e-05,
      "step": 9442
    },
    {
      "action_loss_avg": 0.022679893160238863,
      "epoch": 3.601837158203125e-05,
      "step": 9442
    },
    {
      "epoch": 3.601837158203125e-05,
      "step": 9442,
      "training_step_time": 9.919192790985107
    },
    {
      "epoch": 3.601837158203125e-05,
      "model_forward_time": 3.1451938152313232,
      "step": 9442
    },
    {
      "epoch": 3.601837158203125e-05,
      "step": 9442,
      "training_step_time": 10.30753231048584
    },
    {
      "epoch": 3.6022186279296874e-05,
      "model_forward_time": 3.4060137271881104,
      "step": 9443
    },
    {
      "epoch": 3.6022186279296874e-05,
      "step": 9443,
      "training_step_time": 10.18605351448059
    },
    {
      "epoch": 3.6022186279296874e-05,
      "model_forward_time": 2.4356935024261475,
      "step": 9443
    },
    {
      "epoch": 3.6022186279296874e-05,
      "step": 9443,
      "training_step_time": 7.469904184341431
    },
    {
      "epoch": 3.6022186279296874e-05,
      "model_forward_time": 3.1351606845855713,
      "step": 9443
    },
    {
      "epoch": 3.6022186279296874e-05,
      "step": 9443,
      "training_step_time": 9.90922737121582
    },
    {
      "epoch": 3.6022186279296874e-05,
      "model_forward_time": 3.139286994934082,
      "step": 9443
    },
    {
      "epoch": 3.6022186279296874e-05,
      "step": 9443,
      "training_step_time": 10.305115222930908
    },
    {
      "epoch": 3.60260009765625e-05,
      "model_forward_time": 3.1468045711517334,
      "step": 9444
    },
    {
      "epoch": 3.60260009765625e-05,
      "step": 9444,
      "training_step_time": 9.92693281173706
    },
    {
      "epoch": 3.60260009765625e-05,
      "model_forward_time": 3.430622100830078,
      "step": 9444
    },
    {
      "epoch": 3.60260009765625e-05,
      "step": 9444,
      "training_step_time": 10.208604574203491
    },
    {
      "epoch": 3.60260009765625e-05,
      "model_forward_time": 3.138960361480713,
      "step": 9444
    },
    {
      "epoch": 3.60260009765625e-05,
      "step": 9444,
      "training_step_time": 9.91640329360962
    },
    {
      "epoch": 3.60260009765625e-05,
      "model_forward_time": 3.1407930850982666,
      "step": 9444
    },
    {
      "epoch": 3.60260009765625e-05,
      "step": 9444,
      "training_step_time": 10.29629397392273
    },
    {
      "epoch": 3.6029815673828126e-05,
      "model_forward_time": 3.1551263332366943,
      "step": 9445
    },
    {
      "dynamics_loss_avg": 0.06030396055430174,
      "epoch": 3.6029815673828126e-05,
      "step": 9445
    },
    {
      "action_loss_avg": 0.0186893779784441,
      "epoch": 3.6029815673828126e-05,
      "step": 9445
    },
    {
      "epoch": 3.6029815673828126e-05,
      "step": 9445,
      "training_step_time": 9.93011736869812
    },
    {
      "epoch": 3.6029815673828126e-05,
      "model_forward_time": 3.13494873046875,
      "step": 9445
    },
    {
      "epoch": 3.6029815673828126e-05,
      "step": 9445,
      "training_step_time": 9.903031826019287
    },
    {
      "epoch": 3.6029815673828126e-05,
      "model_forward_time": 3.1330955028533936,
      "step": 9445
    },
    {
      "epoch": 3.6029815673828126e-05,
      "step": 9445,
      "training_step_time": 9.902717113494873
    },
    {
      "epoch": 3.6029815673828126e-05,
      "model_forward_time": 3.398641347885132,
      "step": 9445
    },
    {
      "epoch": 3.6029815673828126e-05,
      "step": 9445,
      "training_step_time": 10.25990605354309
    },
    {
      "epoch": 3.603363037109375e-05,
      "model_forward_time": 3.1476831436157227,
      "step": 9446
    },
    {
      "epoch": 3.603363037109375e-05,
      "step": 9446,
      "training_step_time": 9.917163133621216
    },
    {
      "epoch": 3.603363037109375e-05,
      "model_forward_time": 3.128019094467163,
      "step": 9446
    },
    {
      "epoch": 3.603363037109375e-05,
      "step": 9446,
      "training_step_time": 9.897857189178467
    },
    {
      "epoch": 3.603363037109375e-05,
      "model_forward_time": 3.1347432136535645,
      "step": 9446
    },
    {
      "epoch": 3.603363037109375e-05,
      "step": 9446,
      "training_step_time": 9.90680742263794
    },
    {
      "epoch": 3.603363037109375e-05,
      "model_forward_time": 3.4108922481536865,
      "step": 9446
    },
    {
      "epoch": 3.603363037109375e-05,
      "step": 9446,
      "training_step_time": 10.317132949829102
    },
    {
      "epoch": 3.603744506835937e-05,
      "model_forward_time": 3.1565310955047607,
      "step": 9447
    },
    {
      "epoch": 3.603744506835937e-05,
      "step": 9447,
      "training_step_time": 9.93128514289856
    },
    {
      "epoch": 3.603744506835937e-05,
      "model_forward_time": 3.131450891494751,
      "step": 9447
    },
    {
      "epoch": 3.603744506835937e-05,
      "step": 9447,
      "training_step_time": 9.904674768447876
    },
    {
      "epoch": 3.603744506835937e-05,
      "model_forward_time": 3.134636878967285,
      "step": 9447
    },
    {
      "dynamics_loss_avg": 0.06035525556653738,
      "epoch": 3.603744506835937e-05,
      "step": 9447
    },
    {
      "action_loss_avg": 0.019124345434829593,
      "epoch": 3.603744506835937e-05,
      "step": 9447
    },
    {
      "epoch": 3.603744506835937e-05,
      "step": 9447,
      "training_step_time": 9.913827419281006
    },
    {
      "epoch": 3.603744506835937e-05,
      "model_forward_time": 3.1348657608032227,
      "step": 9447
    },
    {
      "epoch": 3.603744506835937e-05,
      "step": 9447,
      "training_step_time": 10.281150102615356
    },
    {
      "epoch": 3.6041259765625e-05,
      "model_forward_time": 3.4438271522521973,
      "step": 9448
    },
    {
      "epoch": 3.6041259765625e-05,
      "step": 9448,
      "training_step_time": 10.22491979598999
    },
    {
      "epoch": 3.6041259765625e-05,
      "model_forward_time": 3.1587517261505127,
      "step": 9448
    },
    {
      "epoch": 3.6041259765625e-05,
      "step": 9448,
      "training_step_time": 9.943410873413086
    },
    {
      "epoch": 3.6041259765625e-05,
      "model_forward_time": 3.1506237983703613,
      "step": 9448
    },
    {
      "epoch": 3.6041259765625e-05,
      "step": 9448,
      "training_step_time": 9.927701950073242
    },
    {
      "epoch": 3.6041259765625e-05,
      "model_forward_time": 3.1463427543640137,
      "step": 9448
    },
    {
      "epoch": 3.6041259765625e-05,
      "step": 9448,
      "training_step_time": 10.283398389816284
    },
    {
      "epoch": 3.6045074462890624e-05,
      "model_forward_time": 3.187167167663574,
      "step": 9449
    },
    {
      "epoch": 3.6045074462890624e-05,
      "step": 9449,
      "training_step_time": 9.97024130821228
    },
    {
      "epoch": 3.6045074462890624e-05,
      "model_forward_time": 3.436701774597168,
      "step": 9449
    },
    {
      "epoch": 3.6045074462890624e-05,
      "step": 9449,
      "training_step_time": 10.211627006530762
    },
    {
      "epoch": 3.6045074462890624e-05,
      "model_forward_time": 3.1431350708007812,
      "step": 9449
    },
    {
      "epoch": 3.6045074462890624e-05,
      "step": 9449,
      "training_step_time": 9.923011779785156
    },
    {
      "epoch": 3.6045074462890624e-05,
      "model_forward_time": 3.1424801349639893,
      "step": 9449
    },
    {
      "epoch": 3.6045074462890624e-05,
      "step": 9449,
      "training_step_time": 10.326555013656616
    },
    {
      "epoch": 3.604888916015625e-05,
      "grad_norm": 0.36561647057533264,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 9450
    },
    {
      "epoch": 3.604888916015625e-05,
      "model_forward_time": 3.1511595249176025,
      "step": 9450
    },
    {
      "dynamics_loss_avg": 0.04308776725083589,
      "epoch": 3.604888916015625e-05,
      "step": 9450
    },
    {
      "action_loss_avg": 0.012677801679819823,
      "epoch": 3.604888916015625e-05,
      "step": 9450
    },
    {
      "epoch": 3.604888916015625e-05,
      "step": 9450,
      "training_step_time": 9.924490213394165
    },
    {
      "epoch": 3.604888916015625e-05,
      "model_forward_time": 3.1473093032836914,
      "step": 9450
    },
    {
      "epoch": 3.604888916015625e-05,
      "step": 9450,
      "training_step_time": 9.92242956161499
    },
    {
      "epoch": 3.604888916015625e-05,
      "model_forward_time": 2.74241042137146,
      "step": 9450
    },
    {
      "epoch": 3.604888916015625e-05,
      "step": 9450,
      "training_step_time": 7.774823904037476
    },
    {
      "epoch": 3.604888916015625e-05,
      "model_forward_time": 3.1397502422332764,
      "step": 9450
    },
    {
      "epoch": 3.604888916015625e-05,
      "step": 9450,
      "training_step_time": 10.271597623825073
    },
    {
      "epoch": 3.605270385742188e-05,
      "model_forward_time": 3.1534719467163086,
      "step": 9451
    },
    {
      "epoch": 3.605270385742188e-05,
      "step": 9451,
      "training_step_time": 9.934178590774536
    },
    {
      "epoch": 3.605270385742188e-05,
      "model_forward_time": 3.1457529067993164,
      "step": 9451
    },
    {
      "epoch": 3.605270385742188e-05,
      "step": 9451,
      "training_step_time": 9.91928482055664
    },
    {
      "epoch": 3.605270385742188e-05,
      "model_forward_time": 3.1427547931671143,
      "step": 9451
    },
    {
      "epoch": 3.605270385742188e-05,
      "step": 9451,
      "training_step_time": 9.915042638778687
    },
    {
      "epoch": 3.605270385742188e-05,
      "model_forward_time": 3.4263575077056885,
      "step": 9451
    },
    {
      "epoch": 3.605270385742188e-05,
      "step": 9451,
      "training_step_time": 10.305899858474731
    },
    {
      "epoch": 3.60565185546875e-05,
      "model_forward_time": 3.1560678482055664,
      "step": 9452
    },
    {
      "epoch": 3.60565185546875e-05,
      "step": 9452,
      "training_step_time": 9.929784059524536
    },
    {
      "epoch": 3.60565185546875e-05,
      "model_forward_time": 2.4399616718292236,
      "step": 9452
    },
    {
      "epoch": 3.60565185546875e-05,
      "step": 9452,
      "training_step_time": 7.47323203086853
    },
    {
      "epoch": 3.60565185546875e-05,
      "model_forward_time": 3.137430429458618,
      "step": 9452
    },
    {
      "dynamics_loss_avg": 0.04261013045907021,
      "epoch": 3.60565185546875e-05,
      "step": 9452
    },
    {
      "action_loss_avg": 0.015974812395870685,
      "epoch": 3.60565185546875e-05,
      "step": 9452
    },
    {
      "epoch": 3.60565185546875e-05,
      "step": 9452,
      "training_step_time": 9.917089939117432
    },
    {
      "epoch": 3.60565185546875e-05,
      "model_forward_time": 2.4381651878356934,
      "step": 9452
    },
    {
      "epoch": 3.60565185546875e-05,
      "step": 9452,
      "training_step_time": 10.328328371047974
    },
    {
      "epoch": 3.606033325195312e-05,
      "model_forward_time": 3.1514298915863037,
      "step": 9453
    },
    {
      "epoch": 3.606033325195312e-05,
      "step": 9453,
      "training_step_time": 10.20529842376709
    },
    {
      "epoch": 3.606033325195312e-05,
      "model_forward_time": 3.144578456878662,
      "step": 9453
    },
    {
      "epoch": 3.606033325195312e-05,
      "step": 9453,
      "training_step_time": 9.914543628692627
    },
    {
      "epoch": 3.606033325195312e-05,
      "model_forward_time": 3.1405270099639893,
      "step": 9453
    },
    {
      "epoch": 3.606033325195312e-05,
      "step": 9453,
      "training_step_time": 9.907629013061523
    },
    {
      "epoch": 3.606033325195312e-05,
      "model_forward_time": 3.1437432765960693,
      "step": 9453
    },
    {
      "epoch": 3.606033325195312e-05,
      "step": 9453,
      "training_step_time": 10.021602869033813
    },
    {
      "epoch": 3.606414794921875e-05,
      "model_forward_time": 3.158081531524658,
      "step": 9454
    },
    {
      "epoch": 3.606414794921875e-05,
      "step": 9454,
      "training_step_time": 9.9270761013031
    },
    {
      "epoch": 3.606414794921875e-05,
      "model_forward_time": 3.1326684951782227,
      "step": 9454
    },
    {
      "epoch": 3.606414794921875e-05,
      "step": 9454,
      "training_step_time": 10.178288221359253
    },
    {
      "epoch": 3.606414794921875e-05,
      "model_forward_time": 3.132108688354492,
      "step": 9454
    },
    {
      "epoch": 3.606414794921875e-05,
      "step": 9454,
      "training_step_time": 9.90333867073059
    },
    {
      "epoch": 3.606414794921875e-05,
      "model_forward_time": 3.13104248046875,
      "step": 9454
    },
    {
      "epoch": 3.606414794921875e-05,
      "step": 9454,
      "training_step_time": 10.292715311050415
    },
    {
      "epoch": 3.6067962646484375e-05,
      "model_forward_time": 3.1506199836730957,
      "step": 9455
    },
    {
      "dynamics_loss_avg": 0.057839663699269295,
      "epoch": 3.6067962646484375e-05,
      "step": 9455
    },
    {
      "action_loss_avg": 0.0218128418084234,
      "epoch": 3.6067962646484375e-05,
      "step": 9455
    },
    {
      "epoch": 3.6067962646484375e-05,
      "step": 9455,
      "training_step_time": 9.928256750106812
    },
    {
      "epoch": 3.6067962646484375e-05,
      "model_forward_time": 3.128964900970459,
      "step": 9455
    },
    {
      "epoch": 3.6067962646484375e-05,
      "step": 9455,
      "training_step_time": 9.90720510482788
    },
    {
      "epoch": 3.6067962646484375e-05,
      "model_forward_time": 3.388262987136841,
      "step": 9455
    },
    {
      "epoch": 3.6067962646484375e-05,
      "step": 9455,
      "training_step_time": 10.15720248222351
    },
    {
      "epoch": 3.6067962646484375e-05,
      "model_forward_time": 3.1352810859680176,
      "step": 9455
    },
    {
      "epoch": 3.6067962646484375e-05,
      "step": 9455,
      "training_step_time": 10.03564739227295
    },
    {
      "epoch": 3.607177734375e-05,
      "model_forward_time": 3.157227039337158,
      "step": 9456
    },
    {
      "epoch": 3.607177734375e-05,
      "step": 9456,
      "training_step_time": 9.931368827819824
    },
    {
      "epoch": 3.607177734375e-05,
      "model_forward_time": 2.435321092605591,
      "step": 9456
    },
    {
      "epoch": 3.607177734375e-05,
      "step": 9456,
      "training_step_time": 7.4668803215026855
    },
    {
      "epoch": 3.607177734375e-05,
      "model_forward_time": 3.137472152709961,
      "step": 9456
    },
    {
      "epoch": 3.607177734375e-05,
      "step": 9456,
      "training_step_time": 9.911693572998047
    },
    {
      "epoch": 3.607177734375e-05,
      "model_forward_time": 3.1391544342041016,
      "step": 9456
    },
    {
      "epoch": 3.607177734375e-05,
      "step": 9456,
      "training_step_time": 10.33109712600708
    },
    {
      "epoch": 3.607559204101563e-05,
      "model_forward_time": 3.1694278717041016,
      "step": 9457
    },
    {
      "epoch": 3.607559204101563e-05,
      "step": 9457,
      "training_step_time": 9.957285165786743
    },
    {
      "epoch": 3.607559204101563e-05,
      "model_forward_time": 3.1509556770324707,
      "step": 9457
    },
    {
      "epoch": 3.607559204101563e-05,
      "step": 9457,
      "training_step_time": 9.928226232528687
    },
    {
      "epoch": 3.607559204101563e-05,
      "model_forward_time": 3.1512980461120605,
      "step": 9457
    },
    {
      "dynamics_loss_avg": 0.04610855598002672,
      "epoch": 3.607559204101563e-05,
      "step": 9457
    },
    {
      "action_loss_avg": 0.020416390080936254,
      "epoch": 3.607559204101563e-05,
      "step": 9457
    },
    {
      "epoch": 3.607559204101563e-05,
      "step": 9457,
      "training_step_time": 9.939253807067871
    },
    {
      "epoch": 3.607559204101563e-05,
      "model_forward_time": 2.4449007511138916,
      "step": 9457
    },
    {
      "epoch": 3.607559204101563e-05,
      "step": 9457,
      "training_step_time": 10.333417654037476
    },
    {
      "epoch": 3.607940673828125e-05,
      "model_forward_time": 2.4547805786132812,
      "step": 9458
    },
    {
      "epoch": 3.607940673828125e-05,
      "step": 9458,
      "training_step_time": 7.488600015640259
    },
    {
      "epoch": 3.607940673828125e-05,
      "model_forward_time": 3.4366791248321533,
      "step": 9458
    },
    {
      "epoch": 3.607940673828125e-05,
      "step": 9458,
      "training_step_time": 10.213928699493408
    },
    {
      "epoch": 3.607940673828125e-05,
      "model_forward_time": 3.1441164016723633,
      "step": 9458
    },
    {
      "epoch": 3.607940673828125e-05,
      "step": 9458,
      "training_step_time": 9.920435190200806
    },
    {
      "epoch": 3.607940673828125e-05,
      "model_forward_time": 3.1448519229888916,
      "step": 9458
    },
    {
      "epoch": 3.607940673828125e-05,
      "step": 9458,
      "training_step_time": 10.013507604598999
    },
    {
      "epoch": 3.6083221435546874e-05,
      "model_forward_time": 3.156271457672119,
      "step": 9459
    },
    {
      "epoch": 3.6083221435546874e-05,
      "step": 9459,
      "training_step_time": 9.928004264831543
    },
    {
      "epoch": 3.6083221435546874e-05,
      "model_forward_time": 3.4458518028259277,
      "step": 9459
    },
    {
      "epoch": 3.6083221435546874e-05,
      "step": 9459,
      "training_step_time": 10.215055227279663
    },
    {
      "epoch": 3.6083221435546874e-05,
      "model_forward_time": 3.1448071002960205,
      "step": 9459
    },
    {
      "epoch": 3.6083221435546874e-05,
      "step": 9459,
      "training_step_time": 9.917043924331665
    },
    {
      "epoch": 3.6083221435546874e-05,
      "model_forward_time": 3.1456706523895264,
      "step": 9459
    },
    {
      "epoch": 3.6083221435546874e-05,
      "step": 9459,
      "training_step_time": 10.2850821018219
    },
    {
      "epoch": 3.60870361328125e-05,
      "grad_norm": 0.3816385269165039,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 9460
    },
    {
      "epoch": 3.60870361328125e-05,
      "model_forward_time": 3.161533832550049,
      "step": 9460
    },
    {
      "dynamics_loss_avg": 0.06523959264159203,
      "epoch": 3.60870361328125e-05,
      "step": 9460
    },
    {
      "action_loss_avg": 0.02614220939576626,
      "epoch": 3.60870361328125e-05,
      "step": 9460
    },
    {
      "epoch": 3.60870361328125e-05,
      "step": 9460,
      "training_step_time": 9.939872741699219
    },
    {
      "epoch": 3.60870361328125e-05,
      "model_forward_time": 3.1462326049804688,
      "step": 9460
    },
    {
      "epoch": 3.60870361328125e-05,
      "step": 9460,
      "training_step_time": 9.919102668762207
    },
    {
      "epoch": 3.60870361328125e-05,
      "model_forward_time": 3.13740611076355,
      "step": 9460
    },
    {
      "epoch": 3.60870361328125e-05,
      "step": 9460,
      "training_step_time": 10.195910930633545
    },
    {
      "epoch": 3.60870361328125e-05,
      "model_forward_time": 3.139338731765747,
      "step": 9460
    },
    {
      "epoch": 3.60870361328125e-05,
      "step": 9460,
      "training_step_time": 10.289946794509888
    },
    {
      "epoch": 3.6090850830078126e-05,
      "model_forward_time": 3.1721653938293457,
      "step": 9461
    },
    {
      "epoch": 3.6090850830078126e-05,
      "step": 9461,
      "training_step_time": 9.949732065200806
    },
    {
      "epoch": 3.6090850830078126e-05,
      "model_forward_time": 3.143272876739502,
      "step": 9461
    },
    {
      "epoch": 3.6090850830078126e-05,
      "step": 9461,
      "training_step_time": 9.910893678665161
    },
    {
      "epoch": 3.6090850830078126e-05,
      "model_forward_time": 3.139629364013672,
      "step": 9461
    },
    {
      "epoch": 3.6090850830078126e-05,
      "step": 9461,
      "training_step_time": 9.910454750061035
    },
    {
      "epoch": 3.6090850830078126e-05,
      "model_forward_time": 3.4087202548980713,
      "step": 9461
    },
    {
      "epoch": 3.6090850830078126e-05,
      "step": 9461,
      "training_step_time": 10.278775691986084
    },
    {
      "epoch": 3.609466552734375e-05,
      "model_forward_time": 3.151402473449707,
      "step": 9462
    },
    {
      "epoch": 3.609466552734375e-05,
      "step": 9462,
      "training_step_time": 9.924118995666504
    },
    {
      "epoch": 3.609466552734375e-05,
      "model_forward_time": 3.148287296295166,
      "step": 9462
    },
    {
      "epoch": 3.609466552734375e-05,
      "step": 9462,
      "training_step_time": 9.925658226013184
    },
    {
      "epoch": 3.609466552734375e-05,
      "model_forward_time": 3.149648666381836,
      "step": 9462
    },
    {
      "dynamics_loss_avg": 0.07313535176217556,
      "epoch": 3.609466552734375e-05,
      "step": 9462
    },
    {
      "action_loss_avg": 0.031146917119622232,
      "epoch": 3.609466552734375e-05,
      "step": 9462
    },
    {
      "epoch": 3.609466552734375e-05,
      "step": 9462,
      "training_step_time": 9.929506301879883
    },
    {
      "epoch": 3.609466552734375e-05,
      "model_forward_time": 3.1542465686798096,
      "step": 9462
    },
    {
      "epoch": 3.609466552734375e-05,
      "step": 9462,
      "training_step_time": 10.243902206420898
    },
    {
      "epoch": 3.609848022460937e-05,
      "model_forward_time": 3.162858009338379,
      "step": 9463
    },
    {
      "epoch": 3.609848022460937e-05,
      "step": 9463,
      "training_step_time": 10.255673885345459
    },
    {
      "epoch": 3.609848022460937e-05,
      "model_forward_time": 3.1495587825775146,
      "step": 9463
    },
    {
      "epoch": 3.609848022460937e-05,
      "step": 9463,
      "training_step_time": 9.921868085861206
    },
    {
      "epoch": 3.609848022460937e-05,
      "model_forward_time": 3.1450448036193848,
      "step": 9463
    },
    {
      "epoch": 3.609848022460937e-05,
      "step": 9463,
      "training_step_time": 9.919806241989136
    },
    {
      "epoch": 3.609848022460937e-05,
      "model_forward_time": 3.1522064208984375,
      "step": 9463
    },
    {
      "epoch": 3.609848022460937e-05,
      "step": 9463,
      "training_step_time": 10.2708740234375
    },
    {
      "epoch": 3.6102294921875e-05,
      "model_forward_time": 3.1566667556762695,
      "step": 9464
    },
    {
      "epoch": 3.6102294921875e-05,
      "step": 9464,
      "training_step_time": 9.937524557113647
    },
    {
      "epoch": 3.6102294921875e-05,
      "model_forward_time": 3.4227700233459473,
      "step": 9464
    },
    {
      "epoch": 3.6102294921875e-05,
      "step": 9464,
      "training_step_time": 10.20451831817627
    },
    {
      "epoch": 3.6102294921875e-05,
      "model_forward_time": 3.143723964691162,
      "step": 9464
    },
    {
      "epoch": 3.6102294921875e-05,
      "step": 9464,
      "training_step_time": 9.928527116775513
    },
    {
      "epoch": 3.6102294921875e-05,
      "model_forward_time": 3.148615837097168,
      "step": 9464
    },
    {
      "epoch": 3.6102294921875e-05,
      "step": 9464,
      "training_step_time": 10.02887773513794
    },
    {
      "epoch": 3.6106109619140624e-05,
      "model_forward_time": 3.1673777103424072,
      "step": 9465
    },
    {
      "dynamics_loss_avg": 0.053763705492019656,
      "epoch": 3.6106109619140624e-05,
      "step": 9465
    },
    {
      "action_loss_avg": 0.017574530374258757,
      "epoch": 3.6106109619140624e-05,
      "step": 9465
    },
    {
      "epoch": 3.6106109619140624e-05,
      "step": 9465,
      "training_step_time": 9.956009864807129
    },
    {
      "epoch": 3.6106109619140624e-05,
      "model_forward_time": 3.147463798522949,
      "step": 9465
    },
    {
      "epoch": 3.6106109619140624e-05,
      "step": 9465,
      "training_step_time": 9.943327903747559
    },
    {
      "epoch": 3.6106109619140624e-05,
      "model_forward_time": 3.401693105697632,
      "step": 9465
    },
    {
      "epoch": 3.6106109619140624e-05,
      "step": 9465,
      "training_step_time": 10.180876970291138
    },
    {
      "epoch": 3.6106109619140624e-05,
      "model_forward_time": 3.1405930519104004,
      "step": 9465
    },
    {
      "epoch": 3.6106109619140624e-05,
      "step": 9465,
      "training_step_time": 10.310036420822144
    },
    {
      "epoch": 3.610992431640625e-05,
      "model_forward_time": 2.453200101852417,
      "step": 9466
    },
    {
      "epoch": 3.610992431640625e-05,
      "step": 9466,
      "training_step_time": 7.487787246704102
    },
    {
      "epoch": 3.610992431640625e-05,
      "model_forward_time": 3.13634991645813,
      "step": 9466
    },
    {
      "epoch": 3.610992431640625e-05,
      "step": 9466,
      "training_step_time": 9.908020734786987
    },
    {
      "epoch": 3.610992431640625e-05,
      "model_forward_time": 3.138561248779297,
      "step": 9466
    },
    {
      "epoch": 3.610992431640625e-05,
      "step": 9466,
      "training_step_time": 9.92620038986206
    },
    {
      "epoch": 3.610992431640625e-05,
      "model_forward_time": 3.4227397441864014,
      "step": 9466
    },
    {
      "epoch": 3.610992431640625e-05,
      "step": 9466,
      "training_step_time": 10.291807889938354
    },
    {
      "epoch": 3.611373901367188e-05,
      "model_forward_time": 3.159789800643921,
      "step": 9467
    },
    {
      "epoch": 3.611373901367188e-05,
      "step": 9467,
      "training_step_time": 9.940088033676147
    },
    {
      "epoch": 3.611373901367188e-05,
      "model_forward_time": 3.138970375061035,
      "step": 9467
    },
    {
      "epoch": 3.611373901367188e-05,
      "step": 9467,
      "training_step_time": 9.917509078979492
    },
    {
      "epoch": 3.611373901367188e-05,
      "model_forward_time": 3.142613172531128,
      "step": 9467
    },
    {
      "dynamics_loss_avg": 0.058952069655060765,
      "epoch": 3.611373901367188e-05,
      "step": 9467
    },
    {
      "action_loss_avg": 0.020337328873574735,
      "epoch": 3.611373901367188e-05,
      "step": 9467
    },
    {
      "epoch": 3.611373901367188e-05,
      "step": 9467,
      "training_step_time": 9.919750213623047
    },
    {
      "epoch": 3.611373901367188e-05,
      "model_forward_time": 3.1350042819976807,
      "step": 9467
    },
    {
      "epoch": 3.611373901367188e-05,
      "step": 9467,
      "training_step_time": 10.253485918045044
    },
    {
      "epoch": 3.61175537109375e-05,
      "model_forward_time": 2.7734572887420654,
      "step": 9468
    },
    {
      "epoch": 3.61175537109375e-05,
      "step": 9468,
      "training_step_time": 7.8059468269348145
    },
    {
      "epoch": 3.61175537109375e-05,
      "model_forward_time": 3.137885332107544,
      "step": 9468
    },
    {
      "epoch": 3.61175537109375e-05,
      "step": 9468,
      "training_step_time": 9.914643049240112
    },
    {
      "epoch": 3.61175537109375e-05,
      "model_forward_time": 3.1403894424438477,
      "step": 9468
    },
    {
      "epoch": 3.61175537109375e-05,
      "step": 9468,
      "training_step_time": 9.919402837753296
    },
    {
      "epoch": 3.61175537109375e-05,
      "model_forward_time": 3.1462552547454834,
      "step": 9468
    },
    {
      "epoch": 3.61175537109375e-05,
      "step": 9468,
      "training_step_time": 10.31252145767212
    },
    {
      "epoch": 3.612136840820312e-05,
      "model_forward_time": 3.1500778198242188,
      "step": 9469
    },
    {
      "epoch": 3.612136840820312e-05,
      "step": 9469,
      "training_step_time": 9.939624309539795
    },
    {
      "epoch": 3.612136840820312e-05,
      "model_forward_time": 3.1500837802886963,
      "step": 9469
    },
    {
      "epoch": 3.612136840820312e-05,
      "step": 9469,
      "training_step_time": 10.240803956985474
    },
    {
      "epoch": 3.612136840820312e-05,
      "model_forward_time": 3.1487932205200195,
      "step": 9469
    },
    {
      "epoch": 3.612136840820312e-05,
      "step": 9469,
      "training_step_time": 9.935126781463623
    },
    {
      "epoch": 3.612136840820312e-05,
      "model_forward_time": 3.1464574337005615,
      "step": 9469
    },
    {
      "epoch": 3.612136840820312e-05,
      "step": 9469,
      "training_step_time": 10.323765754699707
    },
    {
      "epoch": 3.612518310546875e-05,
      "grad_norm": 0.42619723081588745,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9470
    },
    {
      "epoch": 3.612518310546875e-05,
      "model_forward_time": 3.154554605484009,
      "step": 9470
    },
    {
      "dynamics_loss_avg": 0.058331430330872534,
      "epoch": 3.612518310546875e-05,
      "step": 9470
    },
    {
      "action_loss_avg": 0.022606442077085377,
      "epoch": 3.612518310546875e-05,
      "step": 9470
    },
    {
      "epoch": 3.612518310546875e-05,
      "step": 9470,
      "training_step_time": 9.94868779182434
    },
    {
      "epoch": 3.612518310546875e-05,
      "model_forward_time": 3.1480581760406494,
      "step": 9470
    },
    {
      "epoch": 3.612518310546875e-05,
      "step": 9470,
      "training_step_time": 9.928888082504272
    },
    {
      "epoch": 3.612518310546875e-05,
      "model_forward_time": 3.4288489818573,
      "step": 9470
    },
    {
      "epoch": 3.612518310546875e-05,
      "step": 9470,
      "training_step_time": 10.204900979995728
    },
    {
      "epoch": 3.612518310546875e-05,
      "model_forward_time": 3.1390879154205322,
      "step": 9470
    },
    {
      "epoch": 3.612518310546875e-05,
      "step": 9470,
      "training_step_time": 10.295013427734375
    },
    {
      "epoch": 3.6128997802734375e-05,
      "model_forward_time": 3.1742544174194336,
      "step": 9471
    },
    {
      "epoch": 3.6128997802734375e-05,
      "step": 9471,
      "training_step_time": 9.948631525039673
    },
    {
      "epoch": 3.6128997802734375e-05,
      "model_forward_time": 3.1433281898498535,
      "step": 9471
    },
    {
      "epoch": 3.6128997802734375e-05,
      "step": 9471,
      "training_step_time": 9.912076950073242
    },
    {
      "epoch": 3.6128997802734375e-05,
      "model_forward_time": 3.140080451965332,
      "step": 9471
    },
    {
      "epoch": 3.6128997802734375e-05,
      "step": 9471,
      "training_step_time": 9.914745569229126
    },
    {
      "epoch": 3.6128997802734375e-05,
      "model_forward_time": 3.146054267883301,
      "step": 9471
    },
    {
      "epoch": 3.6128997802734375e-05,
      "step": 9471,
      "training_step_time": 10.311156034469604
    },
    {
      "epoch": 3.61328125e-05,
      "model_forward_time": 3.168508291244507,
      "step": 9472
    },
    {
      "epoch": 3.61328125e-05,
      "step": 9472,
      "training_step_time": 9.937379598617554
    },
    {
      "epoch": 3.61328125e-05,
      "model_forward_time": 3.1446304321289062,
      "step": 9472
    },
    {
      "epoch": 3.61328125e-05,
      "step": 9472,
      "training_step_time": 9.912891387939453
    },
    {
      "epoch": 3.61328125e-05,
      "model_forward_time": 3.1382522583007812,
      "step": 9472
    },
    {
      "dynamics_loss_avg": 0.05181284192949533,
      "epoch": 3.61328125e-05,
      "step": 9472
    },
    {
      "action_loss_avg": 0.01961204344406724,
      "epoch": 3.61328125e-05,
      "step": 9472
    },
    {
      "epoch": 3.61328125e-05,
      "step": 9472,
      "training_step_time": 9.91552186012268
    },
    {
      "epoch": 3.61328125e-05,
      "model_forward_time": 2.436302900314331,
      "step": 9472
    },
    {
      "epoch": 3.61328125e-05,
      "step": 9472,
      "training_step_time": 10.229536533355713
    },
    {
      "epoch": 3.613662719726563e-05,
      "model_forward_time": 2.467100143432617,
      "step": 9473
    },
    {
      "epoch": 3.613662719726563e-05,
      "step": 9473,
      "training_step_time": 7.7795116901397705
    },
    {
      "epoch": 3.613662719726563e-05,
      "model_forward_time": 3.135268449783325,
      "step": 9473
    },
    {
      "epoch": 3.613662719726563e-05,
      "step": 9473,
      "training_step_time": 9.9075288772583
    },
    {
      "epoch": 3.613662719726563e-05,
      "model_forward_time": 3.1369223594665527,
      "step": 9473
    },
    {
      "epoch": 3.613662719726563e-05,
      "step": 9473,
      "training_step_time": 9.914388656616211
    },
    {
      "epoch": 3.613662719726563e-05,
      "model_forward_time": 3.139827013015747,
      "step": 9473
    },
    {
      "epoch": 3.613662719726563e-05,
      "step": 9473,
      "training_step_time": 10.264807939529419
    },
    {
      "epoch": 3.614044189453125e-05,
      "model_forward_time": 3.152198314666748,
      "step": 9474
    },
    {
      "epoch": 3.614044189453125e-05,
      "step": 9474,
      "training_step_time": 9.924562215805054
    },
    {
      "epoch": 3.614044189453125e-05,
      "model_forward_time": 3.1365699768066406,
      "step": 9474
    },
    {
      "epoch": 3.614044189453125e-05,
      "step": 9474,
      "training_step_time": 10.207745552062988
    },
    {
      "epoch": 3.614044189453125e-05,
      "model_forward_time": 3.1353445053100586,
      "step": 9474
    },
    {
      "epoch": 3.614044189453125e-05,
      "step": 9474,
      "training_step_time": 9.915389776229858
    },
    {
      "epoch": 3.614044189453125e-05,
      "model_forward_time": 3.139953374862671,
      "step": 9474
    },
    {
      "epoch": 3.614044189453125e-05,
      "step": 9474,
      "training_step_time": 10.281912088394165
    },
    {
      "epoch": 3.6144256591796874e-05,
      "model_forward_time": 3.156073808670044,
      "step": 9475
    },
    {
      "dynamics_loss_avg": 0.06123232711106539,
      "epoch": 3.6144256591796874e-05,
      "step": 9475
    },
    {
      "action_loss_avg": 0.02321378546766937,
      "epoch": 3.6144256591796874e-05,
      "step": 9475
    },
    {
      "epoch": 3.6144256591796874e-05,
      "step": 9475,
      "training_step_time": 9.928616046905518
    },
    {
      "epoch": 3.6144256591796874e-05,
      "model_forward_time": 3.1303277015686035,
      "step": 9475
    },
    {
      "epoch": 3.6144256591796874e-05,
      "step": 9475,
      "training_step_time": 9.896786212921143
    },
    {
      "epoch": 3.6144256591796874e-05,
      "model_forward_time": 3.3995118141174316,
      "step": 9475
    },
    {
      "epoch": 3.6144256591796874e-05,
      "step": 9475,
      "training_step_time": 10.162227153778076
    },
    {
      "epoch": 3.6144256591796874e-05,
      "model_forward_time": 3.1315741539001465,
      "step": 9475
    },
    {
      "epoch": 3.6144256591796874e-05,
      "step": 9475,
      "training_step_time": 10.24475646018982
    },
    {
      "epoch": 3.61480712890625e-05,
      "model_forward_time": 3.142153263092041,
      "step": 9476
    },
    {
      "epoch": 3.61480712890625e-05,
      "step": 9476,
      "training_step_time": 9.907392740249634
    },
    {
      "epoch": 3.61480712890625e-05,
      "model_forward_time": 3.132699728012085,
      "step": 9476
    },
    {
      "epoch": 3.61480712890625e-05,
      "step": 9476,
      "training_step_time": 9.90098237991333
    },
    {
      "epoch": 3.61480712890625e-05,
      "model_forward_time": 3.13887357711792,
      "step": 9476
    },
    {
      "epoch": 3.61480712890625e-05,
      "step": 9476,
      "training_step_time": 9.908150672912598
    },
    {
      "epoch": 3.61480712890625e-05,
      "model_forward_time": 3.456105947494507,
      "step": 9476
    },
    {
      "epoch": 3.61480712890625e-05,
      "step": 9476,
      "training_step_time": 10.317214012145996
    },
    {
      "epoch": 3.6151885986328126e-05,
      "model_forward_time": 3.153921127319336,
      "step": 9477
    },
    {
      "epoch": 3.6151885986328126e-05,
      "step": 9477,
      "training_step_time": 9.920909643173218
    },
    {
      "epoch": 3.6151885986328126e-05,
      "model_forward_time": 2.4292612075805664,
      "step": 9477
    },
    {
      "epoch": 3.6151885986328126e-05,
      "step": 9477,
      "training_step_time": 7.4516236782073975
    },
    {
      "epoch": 3.6151885986328126e-05,
      "model_forward_time": 2.4315855503082275,
      "step": 9477
    },
    {
      "dynamics_loss_avg": 0.06163441073149443,
      "epoch": 3.6151885986328126e-05,
      "step": 9477
    },
    {
      "action_loss_avg": 0.027331541664898394,
      "epoch": 3.6151885986328126e-05,
      "step": 9477
    },
    {
      "epoch": 3.6151885986328126e-05,
      "step": 9477,
      "training_step_time": 7.453840494155884
    },
    {
      "epoch": 3.6151885986328126e-05,
      "model_forward_time": 3.1335668563842773,
      "step": 9477
    },
    {
      "epoch": 3.6151885986328126e-05,
      "step": 9477,
      "training_step_time": 10.261436462402344
    },
    {
      "epoch": 3.615570068359375e-05,
      "model_forward_time": 3.142155170440674,
      "step": 9478
    },
    {
      "epoch": 3.615570068359375e-05,
      "step": 9478,
      "training_step_time": 9.909289360046387
    },
    {
      "epoch": 3.615570068359375e-05,
      "model_forward_time": 3.38643479347229,
      "step": 9478
    },
    {
      "epoch": 3.615570068359375e-05,
      "step": 9478,
      "training_step_time": 10.14891505241394
    },
    {
      "epoch": 3.615570068359375e-05,
      "model_forward_time": 3.133504867553711,
      "step": 9478
    },
    {
      "epoch": 3.615570068359375e-05,
      "step": 9478,
      "training_step_time": 9.897053241729736
    },
    {
      "epoch": 3.615570068359375e-05,
      "model_forward_time": 2.4338276386260986,
      "step": 9478
    },
    {
      "epoch": 3.615570068359375e-05,
      "step": 9478,
      "training_step_time": 10.294652223587036
    },
    {
      "epoch": 3.615951538085937e-05,
      "model_forward_time": 3.149165391921997,
      "step": 9479
    },
    {
      "epoch": 3.615951538085937e-05,
      "step": 9479,
      "training_step_time": 9.916882991790771
    },
    {
      "epoch": 3.615951538085937e-05,
      "model_forward_time": 3.1346311569213867,
      "step": 9479
    },
    {
      "epoch": 3.615951538085937e-05,
      "step": 9479,
      "training_step_time": 10.16318941116333
    },
    {
      "epoch": 3.615951538085937e-05,
      "model_forward_time": 3.136171579360962,
      "step": 9479
    },
    {
      "epoch": 3.615951538085937e-05,
      "step": 9479,
      "training_step_time": 9.898268461227417
    },
    {
      "epoch": 3.615951538085937e-05,
      "model_forward_time": 3.1294689178466797,
      "step": 9479
    },
    {
      "epoch": 3.615951538085937e-05,
      "step": 9479,
      "training_step_time": 10.2385413646698
    },
    {
      "epoch": 3.6163330078125e-05,
      "grad_norm": 0.5805912017822266,
      "learning_rate": 1e-05,
      "loss": 0.0731,
      "step": 9480
    },
    {
      "epoch": 3.6163330078125e-05,
      "model_forward_time": 2.4371232986450195,
      "step": 9480
    },
    {
      "dynamics_loss_avg": 0.06792703829705715,
      "epoch": 3.6163330078125e-05,
      "step": 9480
    },
    {
      "action_loss_avg": 0.029362602904438973,
      "epoch": 3.6163330078125e-05,
      "step": 9480
    },
    {
      "epoch": 3.6163330078125e-05,
      "step": 9480,
      "training_step_time": 7.46045994758606
    },
    {
      "epoch": 3.6163330078125e-05,
      "model_forward_time": 2.4297890663146973,
      "step": 9480
    },
    {
      "epoch": 3.6163330078125e-05,
      "step": 9480,
      "training_step_time": 7.446261405944824
    },
    {
      "epoch": 3.6163330078125e-05,
      "model_forward_time": 3.126919984817505,
      "step": 9480
    },
    {
      "epoch": 3.6163330078125e-05,
      "step": 9480,
      "training_step_time": 9.890548706054688
    },
    {
      "epoch": 3.6163330078125e-05,
      "model_forward_time": 3.3757576942443848,
      "step": 9480
    },
    {
      "epoch": 3.6163330078125e-05,
      "step": 9480,
      "training_step_time": 10.314099788665771
    },
    {
      "epoch": 3.6167144775390625e-05,
      "model_forward_time": 3.154934883117676,
      "step": 9481
    },
    {
      "epoch": 3.6167144775390625e-05,
      "step": 9481,
      "training_step_time": 9.91737985610962
    },
    {
      "epoch": 3.6167144775390625e-05,
      "model_forward_time": 3.1322829723358154,
      "step": 9481
    },
    {
      "epoch": 3.6167144775390625e-05,
      "step": 9481,
      "training_step_time": 9.894490242004395
    },
    {
      "epoch": 3.6167144775390625e-05,
      "model_forward_time": 3.131744623184204,
      "step": 9481
    },
    {
      "epoch": 3.6167144775390625e-05,
      "step": 9481,
      "training_step_time": 9.89200472831726
    },
    {
      "epoch": 3.6167144775390625e-05,
      "model_forward_time": 3.4208290576934814,
      "step": 9481
    },
    {
      "epoch": 3.6167144775390625e-05,
      "step": 9481,
      "training_step_time": 10.280932664871216
    },
    {
      "epoch": 3.617095947265625e-05,
      "model_forward_time": 3.152050018310547,
      "step": 9482
    },
    {
      "epoch": 3.617095947265625e-05,
      "step": 9482,
      "training_step_time": 9.922227382659912
    },
    {
      "epoch": 3.617095947265625e-05,
      "model_forward_time": 3.1351616382598877,
      "step": 9482
    },
    {
      "epoch": 3.617095947265625e-05,
      "step": 9482,
      "training_step_time": 9.900324583053589
    },
    {
      "epoch": 3.617095947265625e-05,
      "model_forward_time": 3.1394262313842773,
      "step": 9482
    },
    {
      "dynamics_loss_avg": 0.04904271848499775,
      "epoch": 3.617095947265625e-05,
      "step": 9482
    },
    {
      "action_loss_avg": 0.01654395752120763,
      "epoch": 3.617095947265625e-05,
      "step": 9482
    },
    {
      "epoch": 3.617095947265625e-05,
      "step": 9482,
      "training_step_time": 9.912935018539429
    },
    {
      "epoch": 3.617095947265625e-05,
      "model_forward_time": 3.141343116760254,
      "step": 9482
    },
    {
      "epoch": 3.617095947265625e-05,
      "step": 9482,
      "training_step_time": 10.347929239273071
    },
    {
      "epoch": 3.617477416992188e-05,
      "model_forward_time": 2.4503843784332275,
      "step": 9483
    },
    {
      "epoch": 3.617477416992188e-05,
      "step": 9483,
      "training_step_time": 7.480403184890747
    },
    {
      "epoch": 3.617477416992188e-05,
      "model_forward_time": 3.4235918521881104,
      "step": 9483
    },
    {
      "epoch": 3.617477416992188e-05,
      "step": 9483,
      "training_step_time": 10.196813106536865
    },
    {
      "epoch": 3.617477416992188e-05,
      "model_forward_time": 2.437265157699585,
      "step": 9483
    },
    {
      "epoch": 3.617477416992188e-05,
      "step": 9483,
      "training_step_time": 7.4671242237091064
    },
    {
      "epoch": 3.617477416992188e-05,
      "model_forward_time": 3.138296365737915,
      "step": 9483
    },
    {
      "epoch": 3.617477416992188e-05,
      "step": 9483,
      "training_step_time": 10.304983854293823
    },
    {
      "epoch": 3.61785888671875e-05,
      "model_forward_time": 3.153294324874878,
      "step": 9484
    },
    {
      "epoch": 3.61785888671875e-05,
      "step": 9484,
      "training_step_time": 9.930559396743774
    },
    {
      "epoch": 3.61785888671875e-05,
      "model_forward_time": 3.4221389293670654,
      "step": 9484
    },
    {
      "epoch": 3.61785888671875e-05,
      "step": 9484,
      "training_step_time": 10.191969633102417
    },
    {
      "epoch": 3.61785888671875e-05,
      "model_forward_time": 3.1385374069213867,
      "step": 9484
    },
    {
      "epoch": 3.61785888671875e-05,
      "step": 9484,
      "training_step_time": 9.909130334854126
    },
    {
      "epoch": 3.61785888671875e-05,
      "model_forward_time": 2.4314067363739014,
      "step": 9484
    },
    {
      "epoch": 3.61785888671875e-05,
      "step": 9484,
      "training_step_time": 10.273862361907959
    },
    {
      "epoch": 3.618240356445312e-05,
      "model_forward_time": 3.1482644081115723,
      "step": 9485
    },
    {
      "dynamics_loss_avg": 0.04461404252797365,
      "epoch": 3.618240356445312e-05,
      "step": 9485
    },
    {
      "action_loss_avg": 0.01490043317899108,
      "epoch": 3.618240356445312e-05,
      "step": 9485
    },
    {
      "epoch": 3.618240356445312e-05,
      "step": 9485,
      "training_step_time": 9.920982837677002
    },
    {
      "epoch": 3.618240356445312e-05,
      "model_forward_time": 2.431389093399048,
      "step": 9485
    },
    {
      "epoch": 3.618240356445312e-05,
      "step": 9485,
      "training_step_time": 7.4573750495910645
    },
    {
      "epoch": 3.618240356445312e-05,
      "model_forward_time": 3.135000467300415,
      "step": 9485
    },
    {
      "epoch": 3.618240356445312e-05,
      "step": 9485,
      "training_step_time": 9.895833015441895
    },
    {
      "epoch": 3.618240356445312e-05,
      "model_forward_time": 3.4041848182678223,
      "step": 9485
    },
    {
      "epoch": 3.618240356445312e-05,
      "step": 9485,
      "training_step_time": 10.263960838317871
    },
    {
      "epoch": 3.618621826171875e-05,
      "model_forward_time": 3.142143964767456,
      "step": 9486
    },
    {
      "epoch": 3.618621826171875e-05,
      "step": 9486,
      "training_step_time": 9.909878969192505
    },
    {
      "epoch": 3.618621826171875e-05,
      "model_forward_time": 3.134920358657837,
      "step": 9486
    },
    {
      "epoch": 3.618621826171875e-05,
      "step": 9486,
      "training_step_time": 9.901508569717407
    },
    {
      "epoch": 3.618621826171875e-05,
      "model_forward_time": 3.139253854751587,
      "step": 9486
    },
    {
      "epoch": 3.618621826171875e-05,
      "step": 9486,
      "training_step_time": 9.913615226745605
    },
    {
      "epoch": 3.618621826171875e-05,
      "model_forward_time": 3.139378786087036,
      "step": 9486
    },
    {
      "epoch": 3.618621826171875e-05,
      "step": 9486,
      "training_step_time": 10.318260192871094
    },
    {
      "epoch": 3.6190032958984375e-05,
      "model_forward_time": 3.1584386825561523,
      "step": 9487
    },
    {
      "epoch": 3.6190032958984375e-05,
      "step": 9487,
      "training_step_time": 9.930680513381958
    },
    {
      "epoch": 3.6190032958984375e-05,
      "model_forward_time": 3.136584997177124,
      "step": 9487
    },
    {
      "epoch": 3.6190032958984375e-05,
      "step": 9487,
      "training_step_time": 9.906569957733154
    },
    {
      "epoch": 3.6190032958984375e-05,
      "model_forward_time": 3.135859489440918,
      "step": 9487
    },
    {
      "dynamics_loss_avg": 0.05306922886520624,
      "epoch": 3.6190032958984375e-05,
      "step": 9487
    },
    {
      "action_loss_avg": 0.019849635567516087,
      "epoch": 3.6190032958984375e-05,
      "step": 9487
    },
    {
      "epoch": 3.6190032958984375e-05,
      "step": 9487,
      "training_step_time": 9.906230688095093
    },
    {
      "epoch": 3.6190032958984375e-05,
      "model_forward_time": 3.132526159286499,
      "step": 9487
    },
    {
      "epoch": 3.6190032958984375e-05,
      "step": 9487,
      "training_step_time": 10.31308627128601
    },
    {
      "epoch": 3.619384765625e-05,
      "model_forward_time": 3.425386905670166,
      "step": 9488
    },
    {
      "epoch": 3.619384765625e-05,
      "step": 9488,
      "training_step_time": 10.192480564117432
    },
    {
      "epoch": 3.619384765625e-05,
      "model_forward_time": 3.131852388381958,
      "step": 9488
    },
    {
      "epoch": 3.619384765625e-05,
      "step": 9488,
      "training_step_time": 9.899539232254028
    },
    {
      "epoch": 3.619384765625e-05,
      "model_forward_time": 3.1387834548950195,
      "step": 9488
    },
    {
      "epoch": 3.619384765625e-05,
      "step": 9488,
      "training_step_time": 9.907037496566772
    },
    {
      "epoch": 3.619384765625e-05,
      "model_forward_time": 3.1355199813842773,
      "step": 9488
    },
    {
      "epoch": 3.619384765625e-05,
      "step": 9488,
      "training_step_time": 10.279574394226074
    },
    {
      "epoch": 3.619766235351563e-05,
      "model_forward_time": 3.1435320377349854,
      "step": 9489
    },
    {
      "epoch": 3.619766235351563e-05,
      "step": 9489,
      "training_step_time": 9.915559768676758
    },
    {
      "epoch": 3.619766235351563e-05,
      "model_forward_time": 3.1384689807891846,
      "step": 9489
    },
    {
      "epoch": 3.619766235351563e-05,
      "step": 9489,
      "training_step_time": 10.187514305114746
    },
    {
      "epoch": 3.619766235351563e-05,
      "model_forward_time": 3.136953592300415,
      "step": 9489
    },
    {
      "epoch": 3.619766235351563e-05,
      "step": 9489,
      "training_step_time": 9.908766269683838
    },
    {
      "epoch": 3.619766235351563e-05,
      "model_forward_time": 3.136378049850464,
      "step": 9489
    },
    {
      "epoch": 3.619766235351563e-05,
      "step": 9489,
      "training_step_time": 10.257315874099731
    },
    {
      "epoch": 3.620147705078125e-05,
      "grad_norm": 0.6055600047111511,
      "learning_rate": 1e-05,
      "loss": 0.0701,
      "step": 9490
    },
    {
      "epoch": 3.620147705078125e-05,
      "model_forward_time": 3.1516058444976807,
      "step": 9490
    },
    {
      "dynamics_loss_avg": 0.059381506964564325,
      "epoch": 3.620147705078125e-05,
      "step": 9490
    },
    {
      "action_loss_avg": 0.022067328728735448,
      "epoch": 3.620147705078125e-05,
      "step": 9490
    },
    {
      "epoch": 3.620147705078125e-05,
      "step": 9490,
      "training_step_time": 9.927161931991577
    },
    {
      "epoch": 3.620147705078125e-05,
      "model_forward_time": 2.428867816925049,
      "step": 9490
    },
    {
      "epoch": 3.620147705078125e-05,
      "step": 9490,
      "training_step_time": 7.456122159957886
    },
    {
      "epoch": 3.620147705078125e-05,
      "model_forward_time": 3.1298813819885254,
      "step": 9490
    },
    {
      "epoch": 3.620147705078125e-05,
      "step": 9490,
      "training_step_time": 10.188610792160034
    },
    {
      "epoch": 3.620147705078125e-05,
      "model_forward_time": 3.1329376697540283,
      "step": 9490
    },
    {
      "epoch": 3.620147705078125e-05,
      "step": 9490,
      "training_step_time": 10.009114027023315
    },
    {
      "epoch": 3.6205291748046874e-05,
      "model_forward_time": 3.14142107963562,
      "step": 9491
    },
    {
      "epoch": 3.6205291748046874e-05,
      "step": 9491,
      "training_step_time": 9.909546375274658
    },
    {
      "epoch": 3.6205291748046874e-05,
      "model_forward_time": 3.131826400756836,
      "step": 9491
    },
    {
      "epoch": 3.6205291748046874e-05,
      "step": 9491,
      "training_step_time": 9.897006273269653
    },
    {
      "epoch": 3.6205291748046874e-05,
      "model_forward_time": 3.133061647415161,
      "step": 9491
    },
    {
      "epoch": 3.6205291748046874e-05,
      "step": 9491,
      "training_step_time": 9.897836923599243
    },
    {
      "epoch": 3.6205291748046874e-05,
      "model_forward_time": 3.134477376937866,
      "step": 9491
    },
    {
      "epoch": 3.6205291748046874e-05,
      "step": 9491,
      "training_step_time": 10.329603433609009
    },
    {
      "epoch": 3.62091064453125e-05,
      "model_forward_time": 3.1395158767700195,
      "step": 9492
    },
    {
      "epoch": 3.62091064453125e-05,
      "step": 9492,
      "training_step_time": 9.903777837753296
    },
    {
      "epoch": 3.62091064453125e-05,
      "model_forward_time": 3.128922939300537,
      "step": 9492
    },
    {
      "epoch": 3.62091064453125e-05,
      "step": 9492,
      "training_step_time": 9.892014741897583
    },
    {
      "epoch": 3.62091064453125e-05,
      "model_forward_time": 3.1297528743743896,
      "step": 9492
    },
    {
      "dynamics_loss_avg": 0.04613106958568096,
      "epoch": 3.62091064453125e-05,
      "step": 9492
    },
    {
      "action_loss_avg": 0.01572222784161568,
      "epoch": 3.62091064453125e-05,
      "step": 9492
    },
    {
      "epoch": 3.62091064453125e-05,
      "step": 9492,
      "training_step_time": 9.896843194961548
    },
    {
      "epoch": 3.62091064453125e-05,
      "model_forward_time": 2.4257290363311768,
      "step": 9492
    },
    {
      "epoch": 3.62091064453125e-05,
      "step": 9492,
      "training_step_time": 10.332098007202148
    },
    {
      "epoch": 3.6212921142578126e-05,
      "model_forward_time": 3.406881809234619,
      "step": 9493
    },
    {
      "epoch": 3.6212921142578126e-05,
      "step": 9493,
      "training_step_time": 10.180975437164307
    },
    {
      "epoch": 3.6212921142578126e-05,
      "model_forward_time": 3.134472370147705,
      "step": 9493
    },
    {
      "epoch": 3.6212921142578126e-05,
      "step": 9493,
      "training_step_time": 9.904058694839478
    },
    {
      "epoch": 3.6212921142578126e-05,
      "model_forward_time": 3.1312143802642822,
      "step": 9493
    },
    {
      "epoch": 3.6212921142578126e-05,
      "step": 9493,
      "training_step_time": 9.905229330062866
    },
    {
      "epoch": 3.6212921142578126e-05,
      "model_forward_time": 3.135645866394043,
      "step": 9493
    },
    {
      "epoch": 3.6212921142578126e-05,
      "step": 9493,
      "training_step_time": 10.288973808288574
    },
    {
      "epoch": 3.621673583984375e-05,
      "model_forward_time": 3.156118392944336,
      "step": 9494
    },
    {
      "epoch": 3.621673583984375e-05,
      "step": 9494,
      "training_step_time": 9.931537866592407
    },
    {
      "epoch": 3.621673583984375e-05,
      "model_forward_time": 3.1299641132354736,
      "step": 9494
    },
    {
      "epoch": 3.621673583984375e-05,
      "step": 9494,
      "training_step_time": 10.201298713684082
    },
    {
      "epoch": 3.621673583984375e-05,
      "model_forward_time": 3.1288084983825684,
      "step": 9494
    },
    {
      "epoch": 3.621673583984375e-05,
      "step": 9494,
      "training_step_time": 9.900871515274048
    },
    {
      "epoch": 3.621673583984375e-05,
      "model_forward_time": 3.1346142292022705,
      "step": 9494
    },
    {
      "epoch": 3.621673583984375e-05,
      "step": 9494,
      "training_step_time": 10.30322003364563
    },
    {
      "epoch": 3.622055053710937e-05,
      "model_forward_time": 3.171929359436035,
      "step": 9495
    },
    {
      "dynamics_loss_avg": 0.05153792100027203,
      "epoch": 3.622055053710937e-05,
      "step": 9495
    },
    {
      "action_loss_avg": 0.01987289178650826,
      "epoch": 3.622055053710937e-05,
      "step": 9495
    },
    {
      "epoch": 3.622055053710937e-05,
      "step": 9495,
      "training_step_time": 9.955568790435791
    },
    {
      "epoch": 3.622055053710937e-05,
      "model_forward_time": 3.1321895122528076,
      "step": 9495
    },
    {
      "epoch": 3.622055053710937e-05,
      "step": 9495,
      "training_step_time": 9.911499977111816
    },
    {
      "epoch": 3.622055053710937e-05,
      "model_forward_time": 3.434532403945923,
      "step": 9495
    },
    {
      "epoch": 3.622055053710937e-05,
      "step": 9495,
      "training_step_time": 10.212703943252563
    },
    {
      "epoch": 3.622055053710937e-05,
      "model_forward_time": 3.1392407417297363,
      "step": 9495
    },
    {
      "epoch": 3.622055053710937e-05,
      "step": 9495,
      "training_step_time": 10.029701232910156
    },
    {
      "epoch": 3.6224365234375e-05,
      "model_forward_time": 3.151491165161133,
      "step": 9496
    },
    {
      "epoch": 3.6224365234375e-05,
      "step": 9496,
      "training_step_time": 9.940375566482544
    },
    {
      "epoch": 3.6224365234375e-05,
      "model_forward_time": 3.1392881870269775,
      "step": 9496
    },
    {
      "epoch": 3.6224365234375e-05,
      "step": 9496,
      "training_step_time": 9.921226263046265
    },
    {
      "epoch": 3.6224365234375e-05,
      "model_forward_time": 3.1392061710357666,
      "step": 9496
    },
    {
      "epoch": 3.6224365234375e-05,
      "step": 9496,
      "training_step_time": 9.92679500579834
    },
    {
      "epoch": 3.6224365234375e-05,
      "model_forward_time": 3.430089235305786,
      "step": 9496
    },
    {
      "epoch": 3.6224365234375e-05,
      "step": 9496,
      "training_step_time": 10.309675216674805
    },
    {
      "epoch": 3.6228179931640625e-05,
      "model_forward_time": 3.154906988143921,
      "step": 9497
    },
    {
      "epoch": 3.6228179931640625e-05,
      "step": 9497,
      "training_step_time": 9.941523790359497
    },
    {
      "epoch": 3.6228179931640625e-05,
      "model_forward_time": 3.1402981281280518,
      "step": 9497
    },
    {
      "epoch": 3.6228179931640625e-05,
      "step": 9497,
      "training_step_time": 9.922775983810425
    },
    {
      "epoch": 3.6228179931640625e-05,
      "model_forward_time": 3.1420772075653076,
      "step": 9497
    },
    {
      "dynamics_loss_avg": 0.04844231233000755,
      "epoch": 3.6228179931640625e-05,
      "step": 9497
    },
    {
      "action_loss_avg": 0.015499859722331167,
      "epoch": 3.6228179931640625e-05,
      "step": 9497
    },
    {
      "epoch": 3.6228179931640625e-05,
      "step": 9497,
      "training_step_time": 9.921839237213135
    },
    {
      "epoch": 3.6228179931640625e-05,
      "model_forward_time": 3.1345436573028564,
      "step": 9497
    },
    {
      "epoch": 3.6228179931640625e-05,
      "step": 9497,
      "training_step_time": 10.280516147613525
    },
    {
      "epoch": 3.623199462890625e-05,
      "model_forward_time": 3.1479837894439697,
      "step": 9498
    },
    {
      "epoch": 3.623199462890625e-05,
      "step": 9498,
      "training_step_time": 10.279553413391113
    },
    {
      "epoch": 3.623199462890625e-05,
      "model_forward_time": 3.1410155296325684,
      "step": 9498
    },
    {
      "epoch": 3.623199462890625e-05,
      "step": 9498,
      "training_step_time": 9.916626691818237
    },
    {
      "epoch": 3.623199462890625e-05,
      "model_forward_time": 3.1433448791503906,
      "step": 9498
    },
    {
      "epoch": 3.623199462890625e-05,
      "step": 9498,
      "training_step_time": 9.9180588722229
    },
    {
      "epoch": 3.623199462890625e-05,
      "model_forward_time": 3.141247510910034,
      "step": 9498
    },
    {
      "epoch": 3.623199462890625e-05,
      "step": 9498,
      "training_step_time": 10.296834945678711
    },
    {
      "epoch": 3.623580932617188e-05,
      "model_forward_time": 3.1908228397369385,
      "step": 9499
    },
    {
      "epoch": 3.623580932617188e-05,
      "step": 9499,
      "training_step_time": 9.968997955322266
    },
    {
      "epoch": 3.623580932617188e-05,
      "model_forward_time": 3.4109179973602295,
      "step": 9499
    },
    {
      "epoch": 3.623580932617188e-05,
      "step": 9499,
      "training_step_time": 10.187115907669067
    },
    {
      "epoch": 3.623580932617188e-05,
      "model_forward_time": 3.1455228328704834,
      "step": 9499
    },
    {
      "epoch": 3.623580932617188e-05,
      "step": 9499,
      "training_step_time": 9.926973104476929
    },
    {
      "epoch": 3.623580932617188e-05,
      "model_forward_time": 3.1409189701080322,
      "step": 9499
    },
    {
      "epoch": 3.623580932617188e-05,
      "step": 9499,
      "training_step_time": 10.292960166931152
    },
    {
      "epoch": 3.62396240234375e-05,
      "grad_norm": 0.4705740511417389,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 9500
    },
    {
      "epoch": 3.62396240234375e-05,
      "model_forward_time": 3.173027276992798,
      "step": 9500
    },
    {
      "dynamics_loss_avg": 0.05264526456594467,
      "epoch": 3.62396240234375e-05,
      "step": 9500
    },
    {
      "action_loss_avg": 0.01607219045981765,
      "epoch": 3.62396240234375e-05,
      "step": 9500
    },
    {
      "epoch": 3.62396240234375e-05,
      "step": 9500,
      "training_step_time": 9.923751592636108
    },
    {
      "epoch": 3.62396240234375e-05,
      "model_forward_time": 3.1340575218200684,
      "step": 9500
    },
    {
      "epoch": 3.62396240234375e-05,
      "step": 9500,
      "training_step_time": 9.889321088790894
    },
    {
      "epoch": 3.62396240234375e-05,
      "model_forward_time": 5.402768611907959,
      "step": 9500
    },
    {
      "epoch": 3.62396240234375e-05,
      "step": 9500,
      "training_step_time": 12.169967651367188
    },
    {
      "epoch": 3.62396240234375e-05,
      "model_forward_time": 3.134704351425171,
      "step": 9500
    },
    {
      "epoch": 3.62396240234375e-05,
      "step": 9500,
      "training_step_time": 10.315441608428955
    },
    {
      "epoch": 3.624343872070312e-05,
      "model_forward_time": 3.1495132446289062,
      "step": 9501
    },
    {
      "epoch": 3.624343872070312e-05,
      "step": 9501,
      "training_step_time": 9.923941373825073
    },
    {
      "epoch": 3.624343872070312e-05,
      "model_forward_time": 2.4374847412109375,
      "step": 9501
    },
    {
      "epoch": 3.624343872070312e-05,
      "step": 9501,
      "training_step_time": 7.4658989906311035
    },
    {
      "epoch": 3.624343872070312e-05,
      "model_forward_time": 3.1401312351226807,
      "step": 9501
    },
    {
      "epoch": 3.624343872070312e-05,
      "step": 9501,
      "training_step_time": 9.912297248840332
    },
    {
      "epoch": 3.624343872070312e-05,
      "model_forward_time": 3.435575008392334,
      "step": 9501
    },
    {
      "epoch": 3.624343872070312e-05,
      "step": 9501,
      "training_step_time": 10.326120376586914
    },
    {
      "epoch": 3.624725341796875e-05,
      "model_forward_time": 3.1510746479034424,
      "step": 9502
    },
    {
      "epoch": 3.624725341796875e-05,
      "step": 9502,
      "training_step_time": 9.94585633277893
    },
    {
      "epoch": 3.624725341796875e-05,
      "model_forward_time": 3.1520872116088867,
      "step": 9502
    },
    {
      "epoch": 3.624725341796875e-05,
      "step": 9502,
      "training_step_time": 9.932224988937378
    },
    {
      "epoch": 3.624725341796875e-05,
      "model_forward_time": 3.1453301906585693,
      "step": 9502
    },
    {
      "dynamics_loss_avg": 0.05042965319007635,
      "epoch": 3.624725341796875e-05,
      "step": 9502
    },
    {
      "action_loss_avg": 0.015850907703861595,
      "epoch": 3.624725341796875e-05,
      "step": 9502
    },
    {
      "epoch": 3.624725341796875e-05,
      "step": 9502,
      "training_step_time": 9.92926573753357
    },
    {
      "epoch": 3.624725341796875e-05,
      "model_forward_time": 3.1505041122436523,
      "step": 9502
    },
    {
      "epoch": 3.624725341796875e-05,
      "step": 9502,
      "training_step_time": 10.38633394241333
    },
    {
      "epoch": 3.6251068115234375e-05,
      "model_forward_time": 3.1581904888153076,
      "step": 9503
    },
    {
      "epoch": 3.6251068115234375e-05,
      "step": 9503,
      "training_step_time": 9.944757461547852
    },
    {
      "epoch": 3.6251068115234375e-05,
      "model_forward_time": 3.1471383571624756,
      "step": 9503
    },
    {
      "epoch": 3.6251068115234375e-05,
      "step": 9503,
      "training_step_time": 9.930850505828857
    },
    {
      "epoch": 3.6251068115234375e-05,
      "model_forward_time": 3.1481165885925293,
      "step": 9503
    },
    {
      "epoch": 3.6251068115234375e-05,
      "step": 9503,
      "training_step_time": 9.931683778762817
    },
    {
      "epoch": 3.6251068115234375e-05,
      "model_forward_time": 3.1467695236206055,
      "step": 9503
    },
    {
      "epoch": 3.6251068115234375e-05,
      "step": 9503,
      "training_step_time": 10.022591829299927
    },
    {
      "epoch": 3.62548828125e-05,
      "model_forward_time": 3.4750208854675293,
      "step": 9504
    },
    {
      "epoch": 3.62548828125e-05,
      "step": 9504,
      "training_step_time": 10.259977340698242
    },
    {
      "epoch": 3.62548828125e-05,
      "model_forward_time": 3.1501290798187256,
      "step": 9504
    },
    {
      "epoch": 3.62548828125e-05,
      "step": 9504,
      "training_step_time": 9.933125019073486
    },
    {
      "epoch": 3.62548828125e-05,
      "model_forward_time": 3.144151449203491,
      "step": 9504
    },
    {
      "epoch": 3.62548828125e-05,
      "step": 9504,
      "training_step_time": 9.927805185317993
    },
    {
      "epoch": 3.62548828125e-05,
      "model_forward_time": 3.1508357524871826,
      "step": 9504
    },
    {
      "epoch": 3.62548828125e-05,
      "step": 9504,
      "training_step_time": 10.035457134246826
    },
    {
      "epoch": 3.625869750976563e-05,
      "model_forward_time": 3.1643998622894287,
      "step": 9505
    },
    {
      "dynamics_loss_avg": 0.06763413473963738,
      "epoch": 3.625869750976563e-05,
      "step": 9505
    },
    {
      "action_loss_avg": 0.027556756045669317,
      "epoch": 3.625869750976563e-05,
      "step": 9505
    },
    {
      "epoch": 3.625869750976563e-05,
      "step": 9505,
      "training_step_time": 9.951418399810791
    },
    {
      "epoch": 3.625869750976563e-05,
      "model_forward_time": 3.1463122367858887,
      "step": 9505
    },
    {
      "epoch": 3.625869750976563e-05,
      "step": 9505,
      "training_step_time": 10.245401620864868
    },
    {
      "epoch": 3.625869750976563e-05,
      "model_forward_time": 3.149991989135742,
      "step": 9505
    },
    {
      "epoch": 3.625869750976563e-05,
      "step": 9505,
      "training_step_time": 9.938774347305298
    },
    {
      "epoch": 3.625869750976563e-05,
      "model_forward_time": 3.149601459503174,
      "step": 9505
    },
    {
      "epoch": 3.625869750976563e-05,
      "step": 9505,
      "training_step_time": 10.259363412857056
    },
    {
      "epoch": 3.626251220703125e-05,
      "model_forward_time": 3.1631174087524414,
      "step": 9506
    },
    {
      "epoch": 3.626251220703125e-05,
      "step": 9506,
      "training_step_time": 9.950406312942505
    },
    {
      "epoch": 3.626251220703125e-05,
      "model_forward_time": 3.144249439239502,
      "step": 9506
    },
    {
      "epoch": 3.626251220703125e-05,
      "step": 9506,
      "training_step_time": 9.922295093536377
    },
    {
      "epoch": 3.626251220703125e-05,
      "model_forward_time": 3.412118673324585,
      "step": 9506
    },
    {
      "epoch": 3.626251220703125e-05,
      "step": 9506,
      "training_step_time": 10.187245845794678
    },
    {
      "epoch": 3.626251220703125e-05,
      "model_forward_time": 3.146463632583618,
      "step": 9506
    },
    {
      "epoch": 3.626251220703125e-05,
      "step": 9506,
      "training_step_time": 10.321916103363037
    },
    {
      "epoch": 3.6266326904296874e-05,
      "model_forward_time": 3.1532397270202637,
      "step": 9507
    },
    {
      "epoch": 3.6266326904296874e-05,
      "step": 9507,
      "training_step_time": 9.934804916381836
    },
    {
      "epoch": 3.6266326904296874e-05,
      "model_forward_time": 3.132178544998169,
      "step": 9507
    },
    {
      "epoch": 3.6266326904296874e-05,
      "step": 9507,
      "training_step_time": 9.90531063079834
    },
    {
      "epoch": 3.6266326904296874e-05,
      "model_forward_time": 3.1350924968719482,
      "step": 9507
    },
    {
      "dynamics_loss_avg": 0.04164732843637466,
      "epoch": 3.6266326904296874e-05,
      "step": 9507
    },
    {
      "action_loss_avg": 0.012988586165010928,
      "epoch": 3.6266326904296874e-05,
      "step": 9507
    },
    {
      "epoch": 3.6266326904296874e-05,
      "step": 9507,
      "training_step_time": 9.917836904525757
    },
    {
      "epoch": 3.6266326904296874e-05,
      "model_forward_time": 3.136268138885498,
      "step": 9507
    },
    {
      "epoch": 3.6266326904296874e-05,
      "step": 9507,
      "training_step_time": 10.322142124176025
    },
    {
      "epoch": 3.62701416015625e-05,
      "model_forward_time": 3.1585540771484375,
      "step": 9508
    },
    {
      "epoch": 3.62701416015625e-05,
      "step": 9508,
      "training_step_time": 9.942174434661865
    },
    {
      "epoch": 3.62701416015625e-05,
      "model_forward_time": 3.143282175064087,
      "step": 9508
    },
    {
      "epoch": 3.62701416015625e-05,
      "step": 9508,
      "training_step_time": 9.931490182876587
    },
    {
      "epoch": 3.62701416015625e-05,
      "model_forward_time": 3.1421403884887695,
      "step": 9508
    },
    {
      "epoch": 3.62701416015625e-05,
      "step": 9508,
      "training_step_time": 9.925765991210938
    },
    {
      "epoch": 3.62701416015625e-05,
      "model_forward_time": 3.138169527053833,
      "step": 9508
    },
    {
      "epoch": 3.62701416015625e-05,
      "step": 9508,
      "training_step_time": 10.275887966156006
    },
    {
      "epoch": 3.6273956298828126e-05,
      "model_forward_time": 3.4300966262817383,
      "step": 9509
    },
    {
      "epoch": 3.6273956298828126e-05,
      "step": 9509,
      "training_step_time": 10.215818643569946
    },
    {
      "epoch": 3.6273956298828126e-05,
      "model_forward_time": 3.1657705307006836,
      "step": 9509
    },
    {
      "epoch": 3.6273956298828126e-05,
      "step": 9509,
      "training_step_time": 9.944892644882202
    },
    {
      "epoch": 3.6273956298828126e-05,
      "model_forward_time": 3.1441240310668945,
      "step": 9509
    },
    {
      "epoch": 3.6273956298828126e-05,
      "step": 9509,
      "training_step_time": 9.926353931427002
    },
    {
      "epoch": 3.6273956298828126e-05,
      "model_forward_time": 3.1367197036743164,
      "step": 9509
    },
    {
      "epoch": 3.6273956298828126e-05,
      "step": 9509,
      "training_step_time": 10.343909502029419
    },
    {
      "epoch": 3.627777099609375e-05,
      "grad_norm": 0.42205145955085754,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 9510
    },
    {
      "epoch": 3.627777099609375e-05,
      "model_forward_time": 3.158803939819336,
      "step": 9510
    },
    {
      "dynamics_loss_avg": 0.06774542275816202,
      "epoch": 3.627777099609375e-05,
      "step": 9510
    },
    {
      "action_loss_avg": 0.021454214490950106,
      "epoch": 3.627777099609375e-05,
      "step": 9510
    },
    {
      "epoch": 3.627777099609375e-05,
      "step": 9510,
      "training_step_time": 9.937021493911743
    },
    {
      "epoch": 3.627777099609375e-05,
      "model_forward_time": 3.4302046298980713,
      "step": 9510
    },
    {
      "epoch": 3.627777099609375e-05,
      "step": 9510,
      "training_step_time": 10.204313278198242
    },
    {
      "epoch": 3.627777099609375e-05,
      "model_forward_time": 3.143867015838623,
      "step": 9510
    },
    {
      "epoch": 3.627777099609375e-05,
      "step": 9510,
      "training_step_time": 9.921472311019897
    },
    {
      "epoch": 3.627777099609375e-05,
      "model_forward_time": 3.1396255493164062,
      "step": 9510
    },
    {
      "epoch": 3.627777099609375e-05,
      "step": 9510,
      "training_step_time": 10.285592317581177
    },
    {
      "epoch": 3.628158569335937e-05,
      "model_forward_time": 3.148499011993408,
      "step": 9511
    },
    {
      "epoch": 3.628158569335937e-05,
      "step": 9511,
      "training_step_time": 9.93093466758728
    },
    {
      "epoch": 3.628158569335937e-05,
      "model_forward_time": 3.140042543411255,
      "step": 9511
    },
    {
      "epoch": 3.628158569335937e-05,
      "step": 9511,
      "training_step_time": 9.918996810913086
    },
    {
      "epoch": 3.628158569335937e-05,
      "model_forward_time": 3.412670612335205,
      "step": 9511
    },
    {
      "epoch": 3.628158569335937e-05,
      "step": 9511,
      "training_step_time": 10.194429636001587
    },
    {
      "epoch": 3.628158569335937e-05,
      "model_forward_time": 3.138807773590088,
      "step": 9511
    },
    {
      "epoch": 3.628158569335937e-05,
      "step": 9511,
      "training_step_time": 10.304265260696411
    },
    {
      "epoch": 3.6285400390625e-05,
      "model_forward_time": 3.1564314365386963,
      "step": 9512
    },
    {
      "epoch": 3.6285400390625e-05,
      "step": 9512,
      "training_step_time": 9.94071626663208
    },
    {
      "epoch": 3.6285400390625e-05,
      "model_forward_time": 3.141092300415039,
      "step": 9512
    },
    {
      "epoch": 3.6285400390625e-05,
      "step": 9512,
      "training_step_time": 9.923419952392578
    },
    {
      "epoch": 3.6285400390625e-05,
      "model_forward_time": 3.147796869277954,
      "step": 9512
    },
    {
      "dynamics_loss_avg": 0.0442048342898488,
      "epoch": 3.6285400390625e-05,
      "step": 9512
    },
    {
      "action_loss_avg": 0.014276574598625302,
      "epoch": 3.6285400390625e-05,
      "step": 9512
    },
    {
      "epoch": 3.6285400390625e-05,
      "step": 9512,
      "training_step_time": 9.935535907745361
    },
    {
      "epoch": 3.6285400390625e-05,
      "model_forward_time": 3.39551043510437,
      "step": 9512
    },
    {
      "epoch": 3.6285400390625e-05,
      "step": 9512,
      "training_step_time": 10.26797103881836
    },
    {
      "epoch": 3.6289215087890625e-05,
      "model_forward_time": 3.1557300090789795,
      "step": 9513
    },
    {
      "epoch": 3.6289215087890625e-05,
      "step": 9513,
      "training_step_time": 9.932172536849976
    },
    {
      "epoch": 3.6289215087890625e-05,
      "model_forward_time": 3.139443874359131,
      "step": 9513
    },
    {
      "epoch": 3.6289215087890625e-05,
      "step": 9513,
      "training_step_time": 9.917640924453735
    },
    {
      "epoch": 3.6289215087890625e-05,
      "model_forward_time": 3.134772777557373,
      "step": 9513
    },
    {
      "epoch": 3.6289215087890625e-05,
      "step": 9513,
      "training_step_time": 9.917702436447144
    },
    {
      "epoch": 3.6289215087890625e-05,
      "model_forward_time": 3.142874240875244,
      "step": 9513
    },
    {
      "epoch": 3.6289215087890625e-05,
      "step": 9513,
      "training_step_time": 10.289250612258911
    },
    {
      "epoch": 3.629302978515625e-05,
      "model_forward_time": 3.435791015625,
      "step": 9514
    },
    {
      "epoch": 3.629302978515625e-05,
      "step": 9514,
      "training_step_time": 10.213738203048706
    },
    {
      "epoch": 3.629302978515625e-05,
      "model_forward_time": 3.139723300933838,
      "step": 9514
    },
    {
      "epoch": 3.629302978515625e-05,
      "step": 9514,
      "training_step_time": 9.913684368133545
    },
    {
      "epoch": 3.629302978515625e-05,
      "model_forward_time": 3.139110803604126,
      "step": 9514
    },
    {
      "epoch": 3.629302978515625e-05,
      "step": 9514,
      "training_step_time": 9.916703224182129
    },
    {
      "epoch": 3.629302978515625e-05,
      "model_forward_time": 3.1403517723083496,
      "step": 9514
    },
    {
      "epoch": 3.629302978515625e-05,
      "step": 9514,
      "training_step_time": 10.288667440414429
    },
    {
      "epoch": 3.629684448242188e-05,
      "model_forward_time": 3.1511635780334473,
      "step": 9515
    },
    {
      "dynamics_loss_avg": 0.06494416035711766,
      "epoch": 3.629684448242188e-05,
      "step": 9515
    },
    {
      "action_loss_avg": 0.02070668563246727,
      "epoch": 3.629684448242188e-05,
      "step": 9515
    },
    {
      "epoch": 3.629684448242188e-05,
      "step": 9515,
      "training_step_time": 9.926324129104614
    },
    {
      "epoch": 3.629684448242188e-05,
      "model_forward_time": 3.421400308609009,
      "step": 9515
    },
    {
      "epoch": 3.629684448242188e-05,
      "step": 9515,
      "training_step_time": 10.206496953964233
    },
    {
      "epoch": 3.629684448242188e-05,
      "model_forward_time": 3.132690906524658,
      "step": 9515
    },
    {
      "epoch": 3.629684448242188e-05,
      "step": 9515,
      "training_step_time": 9.901018142700195
    },
    {
      "epoch": 3.629684448242188e-05,
      "model_forward_time": 3.1308257579803467,
      "step": 9515
    },
    {
      "epoch": 3.629684448242188e-05,
      "step": 9515,
      "training_step_time": 10.281348943710327
    },
    {
      "epoch": 3.63006591796875e-05,
      "model_forward_time": 3.144502878189087,
      "step": 9516
    },
    {
      "epoch": 3.63006591796875e-05,
      "step": 9516,
      "training_step_time": 9.921463966369629
    },
    {
      "epoch": 3.63006591796875e-05,
      "model_forward_time": 3.133286952972412,
      "step": 9516
    },
    {
      "epoch": 3.63006591796875e-05,
      "step": 9516,
      "training_step_time": 9.90749454498291
    },
    {
      "epoch": 3.63006591796875e-05,
      "model_forward_time": 3.386148452758789,
      "step": 9516
    },
    {
      "epoch": 3.63006591796875e-05,
      "step": 9516,
      "training_step_time": 10.157562494277954
    },
    {
      "epoch": 3.63006591796875e-05,
      "model_forward_time": 3.1365091800689697,
      "step": 9516
    },
    {
      "epoch": 3.63006591796875e-05,
      "step": 9516,
      "training_step_time": 10.24025011062622
    },
    {
      "epoch": 3.630447387695312e-05,
      "model_forward_time": 3.1522114276885986,
      "step": 9517
    },
    {
      "epoch": 3.630447387695312e-05,
      "step": 9517,
      "training_step_time": 9.93011999130249
    },
    {
      "epoch": 3.630447387695312e-05,
      "model_forward_time": 3.1307337284088135,
      "step": 9517
    },
    {
      "epoch": 3.630447387695312e-05,
      "step": 9517,
      "training_step_time": 9.908408641815186
    },
    {
      "epoch": 3.630447387695312e-05,
      "model_forward_time": 3.1339852809906006,
      "step": 9517
    },
    {
      "dynamics_loss_avg": 0.06954420693218707,
      "epoch": 3.630447387695312e-05,
      "step": 9517
    },
    {
      "action_loss_avg": 0.022021716553717853,
      "epoch": 3.630447387695312e-05,
      "step": 9517
    },
    {
      "epoch": 3.630447387695312e-05,
      "step": 9517,
      "training_step_time": 9.910915613174438
    },
    {
      "epoch": 3.630447387695312e-05,
      "model_forward_time": 3.4079041481018066,
      "step": 9517
    },
    {
      "epoch": 3.630447387695312e-05,
      "step": 9517,
      "training_step_time": 10.322217226028442
    },
    {
      "epoch": 3.630828857421875e-05,
      "model_forward_time": 3.1496734619140625,
      "step": 9518
    },
    {
      "epoch": 3.630828857421875e-05,
      "step": 9518,
      "training_step_time": 9.931555271148682
    },
    {
      "epoch": 3.630828857421875e-05,
      "model_forward_time": 3.135359764099121,
      "step": 9518
    },
    {
      "epoch": 3.630828857421875e-05,
      "step": 9518,
      "training_step_time": 9.919846057891846
    },
    {
      "epoch": 3.630828857421875e-05,
      "model_forward_time": 3.139021158218384,
      "step": 9518
    },
    {
      "epoch": 3.630828857421875e-05,
      "step": 9518,
      "training_step_time": 9.918123483657837
    },
    {
      "epoch": 3.630828857421875e-05,
      "model_forward_time": 3.1349668502807617,
      "step": 9518
    },
    {
      "epoch": 3.630828857421875e-05,
      "step": 9518,
      "training_step_time": 10.019781112670898
    },
    {
      "epoch": 3.6312103271484375e-05,
      "model_forward_time": 3.4271535873413086,
      "step": 9519
    },
    {
      "epoch": 3.6312103271484375e-05,
      "step": 9519,
      "training_step_time": 10.210936784744263
    },
    {
      "epoch": 3.6312103271484375e-05,
      "model_forward_time": 3.137662172317505,
      "step": 9519
    },
    {
      "epoch": 3.6312103271484375e-05,
      "step": 9519,
      "training_step_time": 9.916985273361206
    },
    {
      "epoch": 3.6312103271484375e-05,
      "model_forward_time": 2.438708543777466,
      "step": 9519
    },
    {
      "epoch": 3.6312103271484375e-05,
      "step": 9519,
      "training_step_time": 7.475461483001709
    },
    {
      "epoch": 3.6312103271484375e-05,
      "model_forward_time": 3.133915662765503,
      "step": 9519
    },
    {
      "epoch": 3.6312103271484375e-05,
      "step": 9519,
      "training_step_time": 10.304945468902588
    },
    {
      "epoch": 3.631591796875e-05,
      "grad_norm": 0.36084985733032227,
      "learning_rate": 1e-05,
      "loss": 0.0756,
      "step": 9520
    },
    {
      "epoch": 3.631591796875e-05,
      "model_forward_time": 3.145066499710083,
      "step": 9520
    },
    {
      "dynamics_loss_avg": 0.05912196505814791,
      "epoch": 3.631591796875e-05,
      "step": 9520
    },
    {
      "action_loss_avg": 0.01990220295265317,
      "epoch": 3.631591796875e-05,
      "step": 9520
    },
    {
      "epoch": 3.631591796875e-05,
      "step": 9520,
      "training_step_time": 9.93249225616455
    },
    {
      "epoch": 3.631591796875e-05,
      "model_forward_time": 3.393662691116333,
      "step": 9520
    },
    {
      "epoch": 3.631591796875e-05,
      "step": 9520,
      "training_step_time": 10.171633005142212
    },
    {
      "epoch": 3.631591796875e-05,
      "model_forward_time": 3.1392736434936523,
      "step": 9520
    },
    {
      "epoch": 3.631591796875e-05,
      "step": 9520,
      "training_step_time": 9.916190385818481
    },
    {
      "epoch": 3.631591796875e-05,
      "model_forward_time": 3.135201930999756,
      "step": 9520
    },
    {
      "epoch": 3.631591796875e-05,
      "step": 9520,
      "training_step_time": 10.02249813079834
    },
    {
      "epoch": 3.631973266601563e-05,
      "model_forward_time": 3.1457810401916504,
      "step": 9521
    },
    {
      "epoch": 3.631973266601563e-05,
      "step": 9521,
      "training_step_time": 9.922300338745117
    },
    {
      "epoch": 3.631973266601563e-05,
      "model_forward_time": 3.1351654529571533,
      "step": 9521
    },
    {
      "epoch": 3.631973266601563e-05,
      "step": 9521,
      "training_step_time": 9.906565427780151
    },
    {
      "epoch": 3.631973266601563e-05,
      "model_forward_time": 3.418760299682617,
      "step": 9521
    },
    {
      "epoch": 3.631973266601563e-05,
      "step": 9521,
      "training_step_time": 10.193639755249023
    },
    {
      "epoch": 3.631973266601563e-05,
      "model_forward_time": 2.432860851287842,
      "step": 9521
    },
    {
      "epoch": 3.631973266601563e-05,
      "step": 9521,
      "training_step_time": 10.296700716018677
    },
    {
      "epoch": 3.632354736328125e-05,
      "model_forward_time": 3.1566789150238037,
      "step": 9522
    },
    {
      "epoch": 3.632354736328125e-05,
      "step": 9522,
      "training_step_time": 9.934164047241211
    },
    {
      "epoch": 3.632354736328125e-05,
      "model_forward_time": 3.135854482650757,
      "step": 9522
    },
    {
      "epoch": 3.632354736328125e-05,
      "step": 9522,
      "training_step_time": 9.91513204574585
    },
    {
      "epoch": 3.632354736328125e-05,
      "model_forward_time": 3.137835741043091,
      "step": 9522
    },
    {
      "dynamics_loss_avg": 0.06188838519155979,
      "epoch": 3.632354736328125e-05,
      "step": 9522
    },
    {
      "action_loss_avg": 0.018850906658917664,
      "epoch": 3.632354736328125e-05,
      "step": 9522
    },
    {
      "epoch": 3.632354736328125e-05,
      "step": 9522,
      "training_step_time": 9.91706919670105
    },
    {
      "epoch": 3.632354736328125e-05,
      "model_forward_time": 3.141108512878418,
      "step": 9522
    },
    {
      "epoch": 3.632354736328125e-05,
      "step": 9522,
      "training_step_time": 10.322495222091675
    },
    {
      "epoch": 3.6327362060546874e-05,
      "model_forward_time": 3.154848098754883,
      "step": 9523
    },
    {
      "epoch": 3.6327362060546874e-05,
      "step": 9523,
      "training_step_time": 9.936721086502075
    },
    {
      "epoch": 3.6327362060546874e-05,
      "model_forward_time": 3.1374127864837646,
      "step": 9523
    },
    {
      "epoch": 3.6327362060546874e-05,
      "step": 9523,
      "training_step_time": 9.91358232498169
    },
    {
      "epoch": 3.6327362060546874e-05,
      "model_forward_time": 3.1447958946228027,
      "step": 9523
    },
    {
      "epoch": 3.6327362060546874e-05,
      "step": 9523,
      "training_step_time": 9.927784204483032
    },
    {
      "epoch": 3.6327362060546874e-05,
      "model_forward_time": 3.1405112743377686,
      "step": 9523
    },
    {
      "epoch": 3.6327362060546874e-05,
      "step": 9523,
      "training_step_time": 10.017662763595581
    },
    {
      "epoch": 3.63311767578125e-05,
      "model_forward_time": 3.4126029014587402,
      "step": 9524
    },
    {
      "epoch": 3.63311767578125e-05,
      "step": 9524,
      "training_step_time": 10.192594289779663
    },
    {
      "epoch": 3.63311767578125e-05,
      "model_forward_time": 3.13936448097229,
      "step": 9524
    },
    {
      "epoch": 3.63311767578125e-05,
      "step": 9524,
      "training_step_time": 9.918787240982056
    },
    {
      "epoch": 3.63311767578125e-05,
      "model_forward_time": 3.1352453231811523,
      "step": 9524
    },
    {
      "epoch": 3.63311767578125e-05,
      "step": 9524,
      "training_step_time": 9.915385246276855
    },
    {
      "epoch": 3.63311767578125e-05,
      "model_forward_time": 3.1381938457489014,
      "step": 9524
    },
    {
      "epoch": 3.63311767578125e-05,
      "step": 9524,
      "training_step_time": 10.28581976890564
    },
    {
      "epoch": 3.6334991455078126e-05,
      "model_forward_time": 3.1480867862701416,
      "step": 9525
    },
    {
      "dynamics_loss_avg": 0.06682817600667476,
      "epoch": 3.6334991455078126e-05,
      "step": 9525
    },
    {
      "action_loss_avg": 0.020101570058614016,
      "epoch": 3.6334991455078126e-05,
      "step": 9525
    },
    {
      "epoch": 3.6334991455078126e-05,
      "step": 9525,
      "training_step_time": 9.928723812103271
    },
    {
      "epoch": 3.6334991455078126e-05,
      "model_forward_time": 3.1312713623046875,
      "step": 9525
    },
    {
      "epoch": 3.6334991455078126e-05,
      "step": 9525,
      "training_step_time": 10.179827213287354
    },
    {
      "epoch": 3.6334991455078126e-05,
      "model_forward_time": 3.1321070194244385,
      "step": 9525
    },
    {
      "epoch": 3.6334991455078126e-05,
      "step": 9525,
      "training_step_time": 9.902990579605103
    },
    {
      "epoch": 3.6334991455078126e-05,
      "model_forward_time": 3.1342856884002686,
      "step": 9525
    },
    {
      "epoch": 3.6334991455078126e-05,
      "step": 9525,
      "training_step_time": 9.998113870620728
    },
    {
      "epoch": 3.633880615234375e-05,
      "model_forward_time": 3.1532042026519775,
      "step": 9526
    },
    {
      "epoch": 3.633880615234375e-05,
      "step": 9526,
      "training_step_time": 9.930512189865112
    },
    {
      "epoch": 3.633880615234375e-05,
      "model_forward_time": 3.1286821365356445,
      "step": 9526
    },
    {
      "epoch": 3.633880615234375e-05,
      "step": 9526,
      "training_step_time": 9.904964447021484
    },
    {
      "epoch": 3.633880615234375e-05,
      "model_forward_time": 3.39361834526062,
      "step": 9526
    },
    {
      "epoch": 3.633880615234375e-05,
      "step": 9526,
      "training_step_time": 10.173040866851807
    },
    {
      "epoch": 3.633880615234375e-05,
      "model_forward_time": 3.1367249488830566,
      "step": 9526
    },
    {
      "epoch": 3.633880615234375e-05,
      "step": 9526,
      "training_step_time": 10.246163845062256
    },
    {
      "epoch": 3.634262084960937e-05,
      "model_forward_time": 3.1445202827453613,
      "step": 9527
    },
    {
      "epoch": 3.634262084960937e-05,
      "step": 9527,
      "training_step_time": 9.923600196838379
    },
    {
      "epoch": 3.634262084960937e-05,
      "model_forward_time": 3.1362247467041016,
      "step": 9527
    },
    {
      "epoch": 3.634262084960937e-05,
      "step": 9527,
      "training_step_time": 9.90839147567749
    },
    {
      "epoch": 3.634262084960937e-05,
      "model_forward_time": 3.1359376907348633,
      "step": 9527
    },
    {
      "dynamics_loss_avg": 0.0715036092326045,
      "epoch": 3.634262084960937e-05,
      "step": 9527
    },
    {
      "action_loss_avg": 0.020966812735423446,
      "epoch": 3.634262084960937e-05,
      "step": 9527
    },
    {
      "epoch": 3.634262084960937e-05,
      "step": 9527,
      "training_step_time": 9.908057451248169
    },
    {
      "epoch": 3.634262084960937e-05,
      "model_forward_time": 3.134357213973999,
      "step": 9527
    },
    {
      "epoch": 3.634262084960937e-05,
      "step": 9527,
      "training_step_time": 10.2833411693573
    },
    {
      "epoch": 3.6346435546875e-05,
      "model_forward_time": 3.15968656539917,
      "step": 9528
    },
    {
      "epoch": 3.6346435546875e-05,
      "step": 9528,
      "training_step_time": 9.95283055305481
    },
    {
      "epoch": 3.6346435546875e-05,
      "model_forward_time": 3.13722562789917,
      "step": 9528
    },
    {
      "epoch": 3.6346435546875e-05,
      "step": 9528,
      "training_step_time": 9.912514209747314
    },
    {
      "epoch": 3.6346435546875e-05,
      "model_forward_time": 3.1420507431030273,
      "step": 9528
    },
    {
      "epoch": 3.6346435546875e-05,
      "step": 9528,
      "training_step_time": 9.914982557296753
    },
    {
      "epoch": 3.6346435546875e-05,
      "model_forward_time": 3.139842987060547,
      "step": 9528
    },
    {
      "epoch": 3.6346435546875e-05,
      "step": 9528,
      "training_step_time": 10.27660059928894
    },
    {
      "epoch": 3.6350250244140625e-05,
      "model_forward_time": 3.431103229522705,
      "step": 9529
    },
    {
      "epoch": 3.6350250244140625e-05,
      "step": 9529,
      "training_step_time": 10.208720445632935
    },
    {
      "epoch": 3.6350250244140625e-05,
      "model_forward_time": 2.4392004013061523,
      "step": 9529
    },
    {
      "epoch": 3.6350250244140625e-05,
      "step": 9529,
      "training_step_time": 7.473929405212402
    },
    {
      "epoch": 3.6350250244140625e-05,
      "model_forward_time": 3.139353036880493,
      "step": 9529
    },
    {
      "epoch": 3.6350250244140625e-05,
      "step": 9529,
      "training_step_time": 9.92339015007019
    },
    {
      "epoch": 3.6350250244140625e-05,
      "model_forward_time": 3.1422977447509766,
      "step": 9529
    },
    {
      "epoch": 3.6350250244140625e-05,
      "step": 9529,
      "training_step_time": 10.288665771484375
    },
    {
      "epoch": 3.635406494140625e-05,
      "grad_norm": 0.46017685532569885,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 9530
    },
    {
      "epoch": 3.635406494140625e-05,
      "model_forward_time": 3.144117832183838,
      "step": 9530
    },
    {
      "dynamics_loss_avg": 0.06827184930443764,
      "epoch": 3.635406494140625e-05,
      "step": 9530
    },
    {
      "action_loss_avg": 0.0230473130941391,
      "epoch": 3.635406494140625e-05,
      "step": 9530
    },
    {
      "epoch": 3.635406494140625e-05,
      "step": 9530,
      "training_step_time": 9.929744005203247
    },
    {
      "epoch": 3.635406494140625e-05,
      "model_forward_time": 3.4002902507781982,
      "step": 9530
    },
    {
      "epoch": 3.635406494140625e-05,
      "step": 9530,
      "training_step_time": 10.175408124923706
    },
    {
      "epoch": 3.635406494140625e-05,
      "model_forward_time": 3.1379647254943848,
      "step": 9530
    },
    {
      "epoch": 3.635406494140625e-05,
      "step": 9530,
      "training_step_time": 9.913818120956421
    },
    {
      "epoch": 3.635406494140625e-05,
      "model_forward_time": 3.136030912399292,
      "step": 9530
    },
    {
      "epoch": 3.635406494140625e-05,
      "step": 9530,
      "training_step_time": 10.296210050582886
    },
    {
      "epoch": 3.635787963867188e-05,
      "model_forward_time": 3.145220994949341,
      "step": 9531
    },
    {
      "epoch": 3.635787963867188e-05,
      "step": 9531,
      "training_step_time": 9.931379795074463
    },
    {
      "epoch": 3.635787963867188e-05,
      "model_forward_time": 3.136495590209961,
      "step": 9531
    },
    {
      "epoch": 3.635787963867188e-05,
      "step": 9531,
      "training_step_time": 9.91287112236023
    },
    {
      "epoch": 3.635787963867188e-05,
      "model_forward_time": 3.1355221271514893,
      "step": 9531
    },
    {
      "epoch": 3.635787963867188e-05,
      "step": 9531,
      "training_step_time": 10.201707363128662
    },
    {
      "epoch": 3.635787963867188e-05,
      "model_forward_time": 3.132739543914795,
      "step": 9531
    },
    {
      "epoch": 3.635787963867188e-05,
      "step": 9531,
      "training_step_time": 10.012800216674805
    },
    {
      "epoch": 3.63616943359375e-05,
      "model_forward_time": 3.147657871246338,
      "step": 9532
    },
    {
      "epoch": 3.63616943359375e-05,
      "step": 9532,
      "training_step_time": 9.926957845687866
    },
    {
      "epoch": 3.63616943359375e-05,
      "model_forward_time": 3.1406149864196777,
      "step": 9532
    },
    {
      "epoch": 3.63616943359375e-05,
      "step": 9532,
      "training_step_time": 9.918393850326538
    },
    {
      "epoch": 3.63616943359375e-05,
      "model_forward_time": 3.137238025665283,
      "step": 9532
    },
    {
      "dynamics_loss_avg": 0.0670405562967062,
      "epoch": 3.63616943359375e-05,
      "step": 9532
    },
    {
      "action_loss_avg": 0.02495404863730073,
      "epoch": 3.63616943359375e-05,
      "step": 9532
    },
    {
      "epoch": 3.63616943359375e-05,
      "step": 9532,
      "training_step_time": 9.919461011886597
    },
    {
      "epoch": 3.63616943359375e-05,
      "model_forward_time": 3.1416537761688232,
      "step": 9532
    },
    {
      "epoch": 3.63616943359375e-05,
      "step": 9532,
      "training_step_time": 10.293989181518555
    },
    {
      "epoch": 3.636550903320312e-05,
      "model_forward_time": 3.155052661895752,
      "step": 9533
    },
    {
      "epoch": 3.636550903320312e-05,
      "step": 9533,
      "training_step_time": 9.935409545898438
    },
    {
      "epoch": 3.636550903320312e-05,
      "model_forward_time": 3.1414854526519775,
      "step": 9533
    },
    {
      "epoch": 3.636550903320312e-05,
      "step": 9533,
      "training_step_time": 9.918114185333252
    },
    {
      "epoch": 3.636550903320312e-05,
      "model_forward_time": 3.142683267593384,
      "step": 9533
    },
    {
      "epoch": 3.636550903320312e-05,
      "step": 9533,
      "training_step_time": 9.928634881973267
    },
    {
      "epoch": 3.636550903320312e-05,
      "model_forward_time": 3.1453969478607178,
      "step": 9533
    },
    {
      "epoch": 3.636550903320312e-05,
      "step": 9533,
      "training_step_time": 10.26389217376709
    },
    {
      "epoch": 3.636932373046875e-05,
      "model_forward_time": 3.154597043991089,
      "step": 9534
    },
    {
      "epoch": 3.636932373046875e-05,
      "step": 9534,
      "training_step_time": 10.228436470031738
    },
    {
      "epoch": 3.636932373046875e-05,
      "model_forward_time": 3.1456637382507324,
      "step": 9534
    },
    {
      "epoch": 3.636932373046875e-05,
      "step": 9534,
      "training_step_time": 9.93115758895874
    },
    {
      "epoch": 3.636932373046875e-05,
      "model_forward_time": 3.14308762550354,
      "step": 9534
    },
    {
      "epoch": 3.636932373046875e-05,
      "step": 9534,
      "training_step_time": 9.926543474197388
    },
    {
      "epoch": 3.636932373046875e-05,
      "model_forward_time": 3.142043113708496,
      "step": 9534
    },
    {
      "epoch": 3.636932373046875e-05,
      "step": 9534,
      "training_step_time": 10.01588249206543
    },
    {
      "epoch": 3.6373138427734376e-05,
      "model_forward_time": 3.160780668258667,
      "step": 9535
    },
    {
      "dynamics_loss_avg": 0.06521724611520767,
      "epoch": 3.6373138427734376e-05,
      "step": 9535
    },
    {
      "action_loss_avg": 0.027826330112293362,
      "epoch": 3.6373138427734376e-05,
      "step": 9535
    },
    {
      "epoch": 3.6373138427734376e-05,
      "step": 9535,
      "training_step_time": 9.955074548721313
    },
    {
      "epoch": 3.6373138427734376e-05,
      "model_forward_time": 3.4543142318725586,
      "step": 9535
    },
    {
      "epoch": 3.6373138427734376e-05,
      "step": 9535,
      "training_step_time": 10.235802173614502
    },
    {
      "epoch": 3.6373138427734376e-05,
      "model_forward_time": 3.1393868923187256,
      "step": 9535
    },
    {
      "epoch": 3.6373138427734376e-05,
      "step": 9535,
      "training_step_time": 9.921264886856079
    },
    {
      "epoch": 3.6373138427734376e-05,
      "model_forward_time": 3.1433892250061035,
      "step": 9535
    },
    {
      "epoch": 3.6373138427734376e-05,
      "step": 9535,
      "training_step_time": 10.277756929397583
    },
    {
      "epoch": 3.6376953125e-05,
      "model_forward_time": 3.1522653102874756,
      "step": 9536
    },
    {
      "epoch": 3.6376953125e-05,
      "step": 9536,
      "training_step_time": 9.935738563537598
    },
    {
      "epoch": 3.6376953125e-05,
      "model_forward_time": 3.1388657093048096,
      "step": 9536
    },
    {
      "epoch": 3.6376953125e-05,
      "step": 9536,
      "training_step_time": 9.92807126045227
    },
    {
      "epoch": 3.6376953125e-05,
      "model_forward_time": 3.146756172180176,
      "step": 9536
    },
    {
      "epoch": 3.6376953125e-05,
      "step": 9536,
      "training_step_time": 10.215754985809326
    },
    {
      "epoch": 3.6376953125e-05,
      "model_forward_time": 3.138922691345215,
      "step": 9536
    },
    {
      "epoch": 3.6376953125e-05,
      "step": 9536,
      "training_step_time": 10.012329339981079
    },
    {
      "epoch": 3.638076782226563e-05,
      "model_forward_time": 3.155973196029663,
      "step": 9537
    },
    {
      "epoch": 3.638076782226563e-05,
      "step": 9537,
      "training_step_time": 9.938275575637817
    },
    {
      "epoch": 3.638076782226563e-05,
      "model_forward_time": 3.1371004581451416,
      "step": 9537
    },
    {
      "epoch": 3.638076782226563e-05,
      "step": 9537,
      "training_step_time": 9.922852516174316
    },
    {
      "epoch": 3.638076782226563e-05,
      "model_forward_time": 3.1432111263275146,
      "step": 9537
    },
    {
      "dynamics_loss_avg": 0.04286164958029985,
      "epoch": 3.638076782226563e-05,
      "step": 9537
    },
    {
      "action_loss_avg": 0.013211696222424508,
      "epoch": 3.638076782226563e-05,
      "step": 9537
    },
    {
      "epoch": 3.638076782226563e-05,
      "step": 9537,
      "training_step_time": 9.925218343734741
    },
    {
      "epoch": 3.638076782226563e-05,
      "model_forward_time": 3.4262516498565674,
      "step": 9537
    },
    {
      "epoch": 3.638076782226563e-05,
      "step": 9537,
      "training_step_time": 10.313892364501953
    },
    {
      "epoch": 3.638458251953125e-05,
      "model_forward_time": 3.1557414531707764,
      "step": 9538
    },
    {
      "epoch": 3.638458251953125e-05,
      "step": 9538,
      "training_step_time": 9.94182276725769
    },
    {
      "epoch": 3.638458251953125e-05,
      "model_forward_time": 3.140106678009033,
      "step": 9538
    },
    {
      "epoch": 3.638458251953125e-05,
      "step": 9538,
      "training_step_time": 9.918199300765991
    },
    {
      "epoch": 3.638458251953125e-05,
      "model_forward_time": 3.141092538833618,
      "step": 9538
    },
    {
      "epoch": 3.638458251953125e-05,
      "step": 9538,
      "training_step_time": 9.920535326004028
    },
    {
      "epoch": 3.638458251953125e-05,
      "model_forward_time": 3.1376960277557373,
      "step": 9538
    },
    {
      "epoch": 3.638458251953125e-05,
      "step": 9538,
      "training_step_time": 10.289989948272705
    },
    {
      "epoch": 3.6388397216796874e-05,
      "model_forward_time": 2.4457459449768066,
      "step": 9539
    },
    {
      "epoch": 3.6388397216796874e-05,
      "step": 9539,
      "training_step_time": 7.760558843612671
    },
    {
      "epoch": 3.6388397216796874e-05,
      "model_forward_time": 3.133923053741455,
      "step": 9539
    },
    {
      "epoch": 3.6388397216796874e-05,
      "step": 9539,
      "training_step_time": 9.916620969772339
    },
    {
      "epoch": 3.6388397216796874e-05,
      "model_forward_time": 3.138500452041626,
      "step": 9539
    },
    {
      "epoch": 3.6388397216796874e-05,
      "step": 9539,
      "training_step_time": 9.91471242904663
    },
    {
      "epoch": 3.6388397216796874e-05,
      "model_forward_time": 3.1393637657165527,
      "step": 9539
    },
    {
      "epoch": 3.6388397216796874e-05,
      "step": 9539,
      "training_step_time": 10.292409181594849
    },
    {
      "epoch": 3.63922119140625e-05,
      "grad_norm": 0.38240012526512146,
      "learning_rate": 1e-05,
      "loss": 0.0787,
      "step": 9540
    },
    {
      "epoch": 3.63922119140625e-05,
      "model_forward_time": 3.1486706733703613,
      "step": 9540
    },
    {
      "dynamics_loss_avg": 0.06670399364084005,
      "epoch": 3.63922119140625e-05,
      "step": 9540
    },
    {
      "action_loss_avg": 0.022762833070009945,
      "epoch": 3.63922119140625e-05,
      "step": 9540
    },
    {
      "epoch": 3.63922119140625e-05,
      "step": 9540,
      "training_step_time": 9.933863878250122
    },
    {
      "epoch": 3.63922119140625e-05,
      "model_forward_time": 3.4013969898223877,
      "step": 9540
    },
    {
      "epoch": 3.63922119140625e-05,
      "step": 9540,
      "training_step_time": 10.181159019470215
    },
    {
      "epoch": 3.63922119140625e-05,
      "model_forward_time": 3.1455495357513428,
      "step": 9540
    },
    {
      "epoch": 3.63922119140625e-05,
      "step": 9540,
      "training_step_time": 9.927619218826294
    },
    {
      "epoch": 3.63922119140625e-05,
      "model_forward_time": 3.142709255218506,
      "step": 9540
    },
    {
      "epoch": 3.63922119140625e-05,
      "step": 9540,
      "training_step_time": 10.031739234924316
    },
    {
      "epoch": 3.6396026611328126e-05,
      "model_forward_time": 3.1541907787323,
      "step": 9541
    },
    {
      "epoch": 3.6396026611328126e-05,
      "step": 9541,
      "training_step_time": 9.933321475982666
    },
    {
      "epoch": 3.6396026611328126e-05,
      "model_forward_time": 3.138648748397827,
      "step": 9541
    },
    {
      "epoch": 3.6396026611328126e-05,
      "step": 9541,
      "training_step_time": 9.914271354675293
    },
    {
      "epoch": 3.6396026611328126e-05,
      "model_forward_time": 3.4237489700317383,
      "step": 9541
    },
    {
      "epoch": 3.6396026611328126e-05,
      "step": 9541,
      "training_step_time": 10.195300340652466
    },
    {
      "epoch": 3.6396026611328126e-05,
      "model_forward_time": 3.137192487716675,
      "step": 9541
    },
    {
      "epoch": 3.6396026611328126e-05,
      "step": 9541,
      "training_step_time": 10.291008234024048
    },
    {
      "epoch": 3.639984130859375e-05,
      "model_forward_time": 3.14951229095459,
      "step": 9542
    },
    {
      "epoch": 3.639984130859375e-05,
      "step": 9542,
      "training_step_time": 9.927181720733643
    },
    {
      "epoch": 3.639984130859375e-05,
      "model_forward_time": 3.143623113632202,
      "step": 9542
    },
    {
      "epoch": 3.639984130859375e-05,
      "step": 9542,
      "training_step_time": 9.918405294418335
    },
    {
      "epoch": 3.639984130859375e-05,
      "model_forward_time": 3.1403708457946777,
      "step": 9542
    },
    {
      "dynamics_loss_avg": 0.050055477768182755,
      "epoch": 3.639984130859375e-05,
      "step": 9542
    },
    {
      "action_loss_avg": 0.01701334211975336,
      "epoch": 3.639984130859375e-05,
      "step": 9542
    },
    {
      "epoch": 3.639984130859375e-05,
      "step": 9542,
      "training_step_time": 9.91337513923645
    },
    {
      "epoch": 3.639984130859375e-05,
      "model_forward_time": 3.1414732933044434,
      "step": 9542
    },
    {
      "epoch": 3.639984130859375e-05,
      "step": 9542,
      "training_step_time": 10.28590703010559
    },
    {
      "epoch": 3.640365600585937e-05,
      "model_forward_time": 3.164395809173584,
      "step": 9543
    },
    {
      "epoch": 3.640365600585937e-05,
      "step": 9543,
      "training_step_time": 9.937418222427368
    },
    {
      "epoch": 3.640365600585937e-05,
      "model_forward_time": 3.1419286727905273,
      "step": 9543
    },
    {
      "epoch": 3.640365600585937e-05,
      "step": 9543,
      "training_step_time": 9.91187858581543
    },
    {
      "epoch": 3.640365600585937e-05,
      "model_forward_time": 3.141279458999634,
      "step": 9543
    },
    {
      "epoch": 3.640365600585937e-05,
      "step": 9543,
      "training_step_time": 9.913272142410278
    },
    {
      "epoch": 3.640365600585937e-05,
      "model_forward_time": 3.141883611679077,
      "step": 9543
    },
    {
      "epoch": 3.640365600585937e-05,
      "step": 9543,
      "training_step_time": 10.278456211090088
    },
    {
      "epoch": 3.6407470703125e-05,
      "model_forward_time": 3.428591728210449,
      "step": 9544
    },
    {
      "epoch": 3.6407470703125e-05,
      "step": 9544,
      "training_step_time": 10.20301628112793
    },
    {
      "epoch": 3.6407470703125e-05,
      "model_forward_time": 3.139226198196411,
      "step": 9544
    },
    {
      "epoch": 3.6407470703125e-05,
      "step": 9544,
      "training_step_time": 9.907089471817017
    },
    {
      "epoch": 3.6407470703125e-05,
      "model_forward_time": 3.136070966720581,
      "step": 9544
    },
    {
      "epoch": 3.6407470703125e-05,
      "step": 9544,
      "training_step_time": 9.907385110855103
    },
    {
      "epoch": 3.6407470703125e-05,
      "model_forward_time": 3.1431782245635986,
      "step": 9544
    },
    {
      "epoch": 3.6407470703125e-05,
      "step": 9544,
      "training_step_time": 10.284866094589233
    },
    {
      "epoch": 3.6411285400390625e-05,
      "model_forward_time": 3.181307315826416,
      "step": 9545
    },
    {
      "dynamics_loss_avg": 0.04800447523593902,
      "epoch": 3.6411285400390625e-05,
      "step": 9545
    },
    {
      "action_loss_avg": 0.014676016569137574,
      "epoch": 3.6411285400390625e-05,
      "step": 9545
    },
    {
      "epoch": 3.6411285400390625e-05,
      "step": 9545,
      "training_step_time": 9.95876169204712
    },
    {
      "epoch": 3.6411285400390625e-05,
      "model_forward_time": 3.1379799842834473,
      "step": 9545
    },
    {
      "epoch": 3.6411285400390625e-05,
      "step": 9545,
      "training_step_time": 10.18999695777893
    },
    {
      "epoch": 3.6411285400390625e-05,
      "model_forward_time": 3.137427568435669,
      "step": 9545
    },
    {
      "epoch": 3.6411285400390625e-05,
      "step": 9545,
      "training_step_time": 9.912542819976807
    },
    {
      "epoch": 3.6411285400390625e-05,
      "model_forward_time": 3.138505220413208,
      "step": 9545
    },
    {
      "epoch": 3.6411285400390625e-05,
      "step": 9545,
      "training_step_time": 10.0138418674469
    },
    {
      "epoch": 3.641510009765625e-05,
      "model_forward_time": 3.1506550312042236,
      "step": 9546
    },
    {
      "epoch": 3.641510009765625e-05,
      "step": 9546,
      "training_step_time": 9.928872346878052
    },
    {
      "epoch": 3.641510009765625e-05,
      "model_forward_time": 3.136345386505127,
      "step": 9546
    },
    {
      "epoch": 3.641510009765625e-05,
      "step": 9546,
      "training_step_time": 9.911016941070557
    },
    {
      "epoch": 3.641510009765625e-05,
      "model_forward_time": 3.463696241378784,
      "step": 9546
    },
    {
      "epoch": 3.641510009765625e-05,
      "step": 9546,
      "training_step_time": 10.232664585113525
    },
    {
      "epoch": 3.641510009765625e-05,
      "model_forward_time": 3.148444890975952,
      "step": 9546
    },
    {
      "epoch": 3.641510009765625e-05,
      "step": 9546,
      "training_step_time": 10.30598497390747
    },
    {
      "epoch": 3.641891479492188e-05,
      "model_forward_time": 3.150552749633789,
      "step": 9547
    },
    {
      "epoch": 3.641891479492188e-05,
      "step": 9547,
      "training_step_time": 9.921120166778564
    },
    {
      "epoch": 3.641891479492188e-05,
      "model_forward_time": 3.1320672035217285,
      "step": 9547
    },
    {
      "epoch": 3.641891479492188e-05,
      "step": 9547,
      "training_step_time": 9.895796298980713
    },
    {
      "epoch": 3.641891479492188e-05,
      "model_forward_time": 3.1346404552459717,
      "step": 9547
    },
    {
      "dynamics_loss_avg": 0.06455518081784248,
      "epoch": 3.641891479492188e-05,
      "step": 9547
    },
    {
      "action_loss_avg": 0.021550987800583243,
      "epoch": 3.641891479492188e-05,
      "step": 9547
    },
    {
      "epoch": 3.641891479492188e-05,
      "step": 9547,
      "training_step_time": 9.903318166732788
    },
    {
      "epoch": 3.641891479492188e-05,
      "model_forward_time": 3.135735511779785,
      "step": 9547
    },
    {
      "epoch": 3.641891479492188e-05,
      "step": 9547,
      "training_step_time": 10.274687051773071
    },
    {
      "epoch": 3.64227294921875e-05,
      "model_forward_time": 3.151479482650757,
      "step": 9548
    },
    {
      "epoch": 3.64227294921875e-05,
      "step": 9548,
      "training_step_time": 9.922089338302612
    },
    {
      "epoch": 3.64227294921875e-05,
      "model_forward_time": 3.13932466506958,
      "step": 9548
    },
    {
      "epoch": 3.64227294921875e-05,
      "step": 9548,
      "training_step_time": 9.912654161453247
    },
    {
      "epoch": 3.64227294921875e-05,
      "model_forward_time": 3.140798807144165,
      "step": 9548
    },
    {
      "epoch": 3.64227294921875e-05,
      "step": 9548,
      "training_step_time": 9.911842823028564
    },
    {
      "epoch": 3.64227294921875e-05,
      "model_forward_time": 3.13942551612854,
      "step": 9548
    },
    {
      "epoch": 3.64227294921875e-05,
      "step": 9548,
      "training_step_time": 10.26627779006958
    },
    {
      "epoch": 3.642654418945312e-05,
      "model_forward_time": 3.43196702003479,
      "step": 9549
    },
    {
      "epoch": 3.642654418945312e-05,
      "step": 9549,
      "training_step_time": 10.208166360855103
    },
    {
      "epoch": 3.642654418945312e-05,
      "model_forward_time": 3.1393051147460938,
      "step": 9549
    },
    {
      "epoch": 3.642654418945312e-05,
      "step": 9549,
      "training_step_time": 9.915623903274536
    },
    {
      "epoch": 3.642654418945312e-05,
      "model_forward_time": 3.146014451980591,
      "step": 9549
    },
    {
      "epoch": 3.642654418945312e-05,
      "step": 9549,
      "training_step_time": 9.926613330841064
    },
    {
      "epoch": 3.642654418945312e-05,
      "model_forward_time": 3.1421310901641846,
      "step": 9549
    },
    {
      "epoch": 3.642654418945312e-05,
      "step": 9549,
      "training_step_time": 10.340774297714233
    },
    {
      "epoch": 3.643035888671875e-05,
      "grad_norm": 0.374082088470459,
      "learning_rate": 1e-05,
      "loss": 0.074,
      "step": 9550
    },
    {
      "epoch": 3.643035888671875e-05,
      "model_forward_time": 3.185079336166382,
      "step": 9550
    },
    {
      "dynamics_loss_avg": 0.06593709392473102,
      "epoch": 3.643035888671875e-05,
      "step": 9550
    },
    {
      "action_loss_avg": 0.019599765725433826,
      "epoch": 3.643035888671875e-05,
      "step": 9550
    },
    {
      "epoch": 3.643035888671875e-05,
      "step": 9550,
      "training_step_time": 9.964686393737793
    },
    {
      "epoch": 3.643035888671875e-05,
      "model_forward_time": 3.4146599769592285,
      "step": 9550
    },
    {
      "epoch": 3.643035888671875e-05,
      "step": 9550,
      "training_step_time": 10.19205617904663
    },
    {
      "epoch": 3.643035888671875e-05,
      "model_forward_time": 3.1447818279266357,
      "step": 9550
    },
    {
      "epoch": 3.643035888671875e-05,
      "step": 9550,
      "training_step_time": 9.922362327575684
    },
    {
      "epoch": 3.643035888671875e-05,
      "model_forward_time": 3.1420836448669434,
      "step": 9550
    },
    {
      "epoch": 3.643035888671875e-05,
      "step": 9550,
      "training_step_time": 10.277411699295044
    },
    {
      "epoch": 3.6434173583984376e-05,
      "model_forward_time": 3.155827283859253,
      "step": 9551
    },
    {
      "epoch": 3.6434173583984376e-05,
      "step": 9551,
      "training_step_time": 9.932219505310059
    },
    {
      "epoch": 3.6434173583984376e-05,
      "model_forward_time": 3.1382176876068115,
      "step": 9551
    },
    {
      "epoch": 3.6434173583984376e-05,
      "step": 9551,
      "training_step_time": 9.913312196731567
    },
    {
      "epoch": 3.6434173583984376e-05,
      "model_forward_time": 3.4318647384643555,
      "step": 9551
    },
    {
      "epoch": 3.6434173583984376e-05,
      "step": 9551,
      "training_step_time": 10.204503536224365
    },
    {
      "epoch": 3.6434173583984376e-05,
      "model_forward_time": 3.138216257095337,
      "step": 9551
    },
    {
      "epoch": 3.6434173583984376e-05,
      "step": 9551,
      "training_step_time": 10.011034488677979
    },
    {
      "epoch": 3.643798828125e-05,
      "model_forward_time": 3.147765636444092,
      "step": 9552
    },
    {
      "epoch": 3.643798828125e-05,
      "step": 9552,
      "training_step_time": 9.921195030212402
    },
    {
      "epoch": 3.643798828125e-05,
      "model_forward_time": 3.1375787258148193,
      "step": 9552
    },
    {
      "epoch": 3.643798828125e-05,
      "step": 9552,
      "training_step_time": 9.910383462905884
    },
    {
      "epoch": 3.643798828125e-05,
      "model_forward_time": 3.136826276779175,
      "step": 9552
    },
    {
      "dynamics_loss_avg": 0.045951164327561855,
      "epoch": 3.643798828125e-05,
      "step": 9552
    },
    {
      "action_loss_avg": 0.014803709788247943,
      "epoch": 3.643798828125e-05,
      "step": 9552
    },
    {
      "epoch": 3.643798828125e-05,
      "step": 9552,
      "training_step_time": 9.914952754974365
    },
    {
      "epoch": 3.643798828125e-05,
      "model_forward_time": 2.74778413772583,
      "step": 9552
    },
    {
      "epoch": 3.643798828125e-05,
      "step": 9552,
      "training_step_time": 10.284830331802368
    },
    {
      "epoch": 3.644180297851563e-05,
      "model_forward_time": 3.1503920555114746,
      "step": 9553
    },
    {
      "epoch": 3.644180297851563e-05,
      "step": 9553,
      "training_step_time": 9.922937631607056
    },
    {
      "epoch": 3.644180297851563e-05,
      "model_forward_time": 3.1434381008148193,
      "step": 9553
    },
    {
      "epoch": 3.644180297851563e-05,
      "step": 9553,
      "training_step_time": 9.92018723487854
    },
    {
      "epoch": 3.644180297851563e-05,
      "model_forward_time": 3.140394449234009,
      "step": 9553
    },
    {
      "epoch": 3.644180297851563e-05,
      "step": 9553,
      "training_step_time": 9.915351390838623
    },
    {
      "epoch": 3.644180297851563e-05,
      "model_forward_time": 3.1469810009002686,
      "step": 9553
    },
    {
      "epoch": 3.644180297851563e-05,
      "step": 9553,
      "training_step_time": 10.306285381317139
    },
    {
      "epoch": 3.644561767578125e-05,
      "model_forward_time": 3.421373128890991,
      "step": 9554
    },
    {
      "epoch": 3.644561767578125e-05,
      "step": 9554,
      "training_step_time": 10.203893661499023
    },
    {
      "epoch": 3.644561767578125e-05,
      "model_forward_time": 3.1484570503234863,
      "step": 9554
    },
    {
      "epoch": 3.644561767578125e-05,
      "step": 9554,
      "training_step_time": 9.929152250289917
    },
    {
      "epoch": 3.644561767578125e-05,
      "model_forward_time": 3.141123056411743,
      "step": 9554
    },
    {
      "epoch": 3.644561767578125e-05,
      "step": 9554,
      "training_step_time": 9.922919034957886
    },
    {
      "epoch": 3.644561767578125e-05,
      "model_forward_time": 3.150641441345215,
      "step": 9554
    },
    {
      "epoch": 3.644561767578125e-05,
      "step": 9554,
      "training_step_time": 10.28849744796753
    },
    {
      "epoch": 3.6449432373046874e-05,
      "model_forward_time": 3.161484479904175,
      "step": 9555
    },
    {
      "dynamics_loss_avg": 0.05400778278708458,
      "epoch": 3.6449432373046874e-05,
      "step": 9555
    },
    {
      "action_loss_avg": 0.018148271553218364,
      "epoch": 3.6449432373046874e-05,
      "step": 9555
    },
    {
      "epoch": 3.6449432373046874e-05,
      "step": 9555,
      "training_step_time": 9.947552919387817
    },
    {
      "epoch": 3.6449432373046874e-05,
      "model_forward_time": 3.4381139278411865,
      "step": 9555
    },
    {
      "epoch": 3.6449432373046874e-05,
      "step": 9555,
      "training_step_time": 10.224949598312378
    },
    {
      "epoch": 3.6449432373046874e-05,
      "model_forward_time": 3.1476030349731445,
      "step": 9555
    },
    {
      "epoch": 3.6449432373046874e-05,
      "step": 9555,
      "training_step_time": 9.931236982345581
    },
    {
      "epoch": 3.6449432373046874e-05,
      "model_forward_time": 3.1552231311798096,
      "step": 9555
    },
    {
      "epoch": 3.6449432373046874e-05,
      "step": 9555,
      "training_step_time": 10.290922403335571
    },
    {
      "epoch": 3.64532470703125e-05,
      "model_forward_time": 3.1631217002868652,
      "step": 9556
    },
    {
      "epoch": 3.64532470703125e-05,
      "step": 9556,
      "training_step_time": 9.95366621017456
    },
    {
      "epoch": 3.64532470703125e-05,
      "model_forward_time": 3.148311138153076,
      "step": 9556
    },
    {
      "epoch": 3.64532470703125e-05,
      "step": 9556,
      "training_step_time": 9.93275237083435
    },
    {
      "epoch": 3.64532470703125e-05,
      "model_forward_time": 3.4042701721191406,
      "step": 9556
    },
    {
      "epoch": 3.64532470703125e-05,
      "step": 9556,
      "training_step_time": 10.189421892166138
    },
    {
      "epoch": 3.64532470703125e-05,
      "model_forward_time": 3.1490910053253174,
      "step": 9556
    },
    {
      "epoch": 3.64532470703125e-05,
      "step": 9556,
      "training_step_time": 10.029385089874268
    },
    {
      "epoch": 3.6457061767578126e-05,
      "model_forward_time": 3.1583590507507324,
      "step": 9557
    },
    {
      "epoch": 3.6457061767578126e-05,
      "step": 9557,
      "training_step_time": 9.943344354629517
    },
    {
      "epoch": 3.6457061767578126e-05,
      "model_forward_time": 3.1410908699035645,
      "step": 9557
    },
    {
      "epoch": 3.6457061767578126e-05,
      "step": 9557,
      "training_step_time": 9.9247465133667
    },
    {
      "epoch": 3.6457061767578126e-05,
      "model_forward_time": 3.1486637592315674,
      "step": 9557
    },
    {
      "dynamics_loss_avg": 0.05955170467495918,
      "epoch": 3.6457061767578126e-05,
      "step": 9557
    },
    {
      "action_loss_avg": 0.019266239600256085,
      "epoch": 3.6457061767578126e-05,
      "step": 9557
    },
    {
      "epoch": 3.6457061767578126e-05,
      "step": 9557,
      "training_step_time": 9.93594741821289
    },
    {
      "epoch": 3.6457061767578126e-05,
      "model_forward_time": 3.423931121826172,
      "step": 9557
    },
    {
      "epoch": 3.6457061767578126e-05,
      "step": 9557,
      "training_step_time": 10.30235743522644
    },
    {
      "epoch": 3.646087646484375e-05,
      "model_forward_time": 3.162717819213867,
      "step": 9558
    },
    {
      "epoch": 3.646087646484375e-05,
      "step": 9558,
      "training_step_time": 9.94045352935791
    },
    {
      "epoch": 3.646087646484375e-05,
      "model_forward_time": 3.141287088394165,
      "step": 9558
    },
    {
      "epoch": 3.646087646484375e-05,
      "step": 9558,
      "training_step_time": 9.916643381118774
    },
    {
      "epoch": 3.646087646484375e-05,
      "model_forward_time": 3.138479471206665,
      "step": 9558
    },
    {
      "epoch": 3.646087646484375e-05,
      "step": 9558,
      "training_step_time": 9.915745735168457
    },
    {
      "epoch": 3.646087646484375e-05,
      "model_forward_time": 3.1437971591949463,
      "step": 9558
    },
    {
      "epoch": 3.646087646484375e-05,
      "step": 9558,
      "training_step_time": 10.276290893554688
    },
    {
      "epoch": 3.646469116210937e-05,
      "model_forward_time": 3.4749794006347656,
      "step": 9559
    },
    {
      "epoch": 3.646469116210937e-05,
      "step": 9559,
      "training_step_time": 10.253190279006958
    },
    {
      "epoch": 3.646469116210937e-05,
      "model_forward_time": 3.1415722370147705,
      "step": 9559
    },
    {
      "epoch": 3.646469116210937e-05,
      "step": 9559,
      "training_step_time": 9.923784732818604
    },
    {
      "epoch": 3.646469116210937e-05,
      "model_forward_time": 3.148676633834839,
      "step": 9559
    },
    {
      "epoch": 3.646469116210937e-05,
      "step": 9559,
      "training_step_time": 9.930694103240967
    },
    {
      "epoch": 3.646469116210937e-05,
      "model_forward_time": 3.1465930938720703,
      "step": 9559
    },
    {
      "epoch": 3.646469116210937e-05,
      "step": 9559,
      "training_step_time": 10.24918246269226
    },
    {
      "epoch": 3.6468505859375e-05,
      "grad_norm": 0.4406702220439911,
      "learning_rate": 1e-05,
      "loss": 0.0757,
      "step": 9560
    },
    {
      "epoch": 3.6468505859375e-05,
      "model_forward_time": 3.1520543098449707,
      "step": 9560
    },
    {
      "dynamics_loss_avg": 0.056487787328660485,
      "epoch": 3.6468505859375e-05,
      "step": 9560
    },
    {
      "action_loss_avg": 0.01446903683245182,
      "epoch": 3.6468505859375e-05,
      "step": 9560
    },
    {
      "epoch": 3.6468505859375e-05,
      "step": 9560,
      "training_step_time": 9.938928127288818
    },
    {
      "epoch": 3.6468505859375e-05,
      "model_forward_time": 3.453951835632324,
      "step": 9560
    },
    {
      "epoch": 3.6468505859375e-05,
      "step": 9560,
      "training_step_time": 10.237578392028809
    },
    {
      "epoch": 3.6468505859375e-05,
      "model_forward_time": 3.1454827785491943,
      "step": 9560
    },
    {
      "epoch": 3.6468505859375e-05,
      "step": 9560,
      "training_step_time": 9.92314887046814
    },
    {
      "epoch": 3.6468505859375e-05,
      "model_forward_time": 3.1381049156188965,
      "step": 9560
    },
    {
      "epoch": 3.6468505859375e-05,
      "step": 9560,
      "training_step_time": 10.328740358352661
    },
    {
      "epoch": 3.6472320556640625e-05,
      "model_forward_time": 3.1508500576019287,
      "step": 9561
    },
    {
      "epoch": 3.6472320556640625e-05,
      "step": 9561,
      "training_step_time": 9.929269075393677
    },
    {
      "epoch": 3.6472320556640625e-05,
      "model_forward_time": 3.1451385021209717,
      "step": 9561
    },
    {
      "epoch": 3.6472320556640625e-05,
      "step": 9561,
      "training_step_time": 9.922315835952759
    },
    {
      "epoch": 3.6472320556640625e-05,
      "model_forward_time": 3.4318368434906006,
      "step": 9561
    },
    {
      "epoch": 3.6472320556640625e-05,
      "step": 9561,
      "training_step_time": 10.213732957839966
    },
    {
      "epoch": 3.6472320556640625e-05,
      "model_forward_time": 3.1398448944091797,
      "step": 9561
    },
    {
      "epoch": 3.6472320556640625e-05,
      "step": 9561,
      "training_step_time": 10.021242380142212
    },
    {
      "epoch": 3.647613525390625e-05,
      "model_forward_time": 3.150446653366089,
      "step": 9562
    },
    {
      "epoch": 3.647613525390625e-05,
      "step": 9562,
      "training_step_time": 9.92573881149292
    },
    {
      "epoch": 3.647613525390625e-05,
      "model_forward_time": 3.1458933353424072,
      "step": 9562
    },
    {
      "epoch": 3.647613525390625e-05,
      "step": 9562,
      "training_step_time": 9.924204349517822
    },
    {
      "epoch": 3.647613525390625e-05,
      "model_forward_time": 3.1451895236968994,
      "step": 9562
    },
    {
      "dynamics_loss_avg": 0.046361423656344415,
      "epoch": 3.647613525390625e-05,
      "step": 9562
    },
    {
      "action_loss_avg": 0.019203988322988152,
      "epoch": 3.647613525390625e-05,
      "step": 9562
    },
    {
      "epoch": 3.647613525390625e-05,
      "step": 9562,
      "training_step_time": 9.921604871749878
    },
    {
      "epoch": 3.647613525390625e-05,
      "model_forward_time": 3.1448893547058105,
      "step": 9562
    },
    {
      "epoch": 3.647613525390625e-05,
      "step": 9562,
      "training_step_time": 10.33956789970398
    },
    {
      "epoch": 3.647994995117188e-05,
      "model_forward_time": 3.1629860401153564,
      "step": 9563
    },
    {
      "epoch": 3.647994995117188e-05,
      "step": 9563,
      "training_step_time": 9.952629327774048
    },
    {
      "epoch": 3.647994995117188e-05,
      "model_forward_time": 3.146700143814087,
      "step": 9563
    },
    {
      "epoch": 3.647994995117188e-05,
      "step": 9563,
      "training_step_time": 9.92238712310791
    },
    {
      "epoch": 3.647994995117188e-05,
      "model_forward_time": 3.142857551574707,
      "step": 9563
    },
    {
      "epoch": 3.647994995117188e-05,
      "step": 9563,
      "training_step_time": 9.919592380523682
    },
    {
      "epoch": 3.647994995117188e-05,
      "model_forward_time": 3.145005941390991,
      "step": 9563
    },
    {
      "epoch": 3.647994995117188e-05,
      "step": 9563,
      "training_step_time": 10.354494333267212
    },
    {
      "epoch": 3.64837646484375e-05,
      "model_forward_time": 3.4536845684051514,
      "step": 9564
    },
    {
      "epoch": 3.64837646484375e-05,
      "step": 9564,
      "training_step_time": 10.231451988220215
    },
    {
      "epoch": 3.64837646484375e-05,
      "model_forward_time": 3.1502296924591064,
      "step": 9564
    },
    {
      "epoch": 3.64837646484375e-05,
      "step": 9564,
      "training_step_time": 9.932661294937134
    },
    {
      "epoch": 3.64837646484375e-05,
      "model_forward_time": 3.147435188293457,
      "step": 9564
    },
    {
      "epoch": 3.64837646484375e-05,
      "step": 9564,
      "training_step_time": 9.930144309997559
    },
    {
      "epoch": 3.64837646484375e-05,
      "model_forward_time": 3.1516172885894775,
      "step": 9564
    },
    {
      "epoch": 3.64837646484375e-05,
      "step": 9564,
      "training_step_time": 10.299944400787354
    },
    {
      "epoch": 3.648757934570312e-05,
      "model_forward_time": 3.156846046447754,
      "step": 9565
    },
    {
      "dynamics_loss_avg": 0.06526689697057009,
      "epoch": 3.648757934570312e-05,
      "step": 9565
    },
    {
      "action_loss_avg": 0.020723516400903464,
      "epoch": 3.648757934570312e-05,
      "step": 9565
    },
    {
      "epoch": 3.648757934570312e-05,
      "step": 9565,
      "training_step_time": 9.938950538635254
    },
    {
      "epoch": 3.648757934570312e-05,
      "model_forward_time": 3.14253830909729,
      "step": 9565
    },
    {
      "epoch": 3.648757934570312e-05,
      "step": 9565,
      "training_step_time": 10.193522691726685
    },
    {
      "epoch": 3.648757934570312e-05,
      "model_forward_time": 3.1397898197174072,
      "step": 9565
    },
    {
      "epoch": 3.648757934570312e-05,
      "step": 9565,
      "training_step_time": 9.912568092346191
    },
    {
      "epoch": 3.648757934570312e-05,
      "model_forward_time": 3.1439523696899414,
      "step": 9565
    },
    {
      "epoch": 3.648757934570312e-05,
      "step": 9565,
      "training_step_time": 10.317305088043213
    },
    {
      "epoch": 3.649139404296875e-05,
      "model_forward_time": 3.1595466136932373,
      "step": 9566
    },
    {
      "epoch": 3.649139404296875e-05,
      "step": 9566,
      "training_step_time": 9.932769775390625
    },
    {
      "epoch": 3.649139404296875e-05,
      "model_forward_time": 3.1350605487823486,
      "step": 9566
    },
    {
      "epoch": 3.649139404296875e-05,
      "step": 9566,
      "training_step_time": 9.905203819274902
    },
    {
      "epoch": 3.649139404296875e-05,
      "model_forward_time": 3.433959484100342,
      "step": 9566
    },
    {
      "epoch": 3.649139404296875e-05,
      "step": 9566,
      "training_step_time": 10.202201128005981
    },
    {
      "epoch": 3.649139404296875e-05,
      "model_forward_time": 3.1418731212615967,
      "step": 9566
    },
    {
      "epoch": 3.649139404296875e-05,
      "step": 9566,
      "training_step_time": 10.012987613677979
    },
    {
      "epoch": 3.6495208740234376e-05,
      "model_forward_time": 3.1582884788513184,
      "step": 9567
    },
    {
      "epoch": 3.6495208740234376e-05,
      "step": 9567,
      "training_step_time": 9.9334876537323
    },
    {
      "epoch": 3.6495208740234376e-05,
      "model_forward_time": 3.1406188011169434,
      "step": 9567
    },
    {
      "epoch": 3.6495208740234376e-05,
      "step": 9567,
      "training_step_time": 9.91048550605774
    },
    {
      "epoch": 3.6495208740234376e-05,
      "model_forward_time": 3.1416614055633545,
      "step": 9567
    },
    {
      "dynamics_loss_avg": 0.04979475047439337,
      "epoch": 3.6495208740234376e-05,
      "step": 9567
    },
    {
      "action_loss_avg": 0.015220481809228659,
      "epoch": 3.6495208740234376e-05,
      "step": 9567
    },
    {
      "epoch": 3.6495208740234376e-05,
      "step": 9567,
      "training_step_time": 9.908831357955933
    },
    {
      "epoch": 3.6495208740234376e-05,
      "model_forward_time": 3.1317553520202637,
      "step": 9567
    },
    {
      "epoch": 3.6495208740234376e-05,
      "step": 9567,
      "training_step_time": 10.268506526947021
    },
    {
      "epoch": 3.64990234375e-05,
      "model_forward_time": 3.1521928310394287,
      "step": 9568
    },
    {
      "epoch": 3.64990234375e-05,
      "step": 9568,
      "training_step_time": 9.923346042633057
    },
    {
      "epoch": 3.64990234375e-05,
      "model_forward_time": 3.137460470199585,
      "step": 9568
    },
    {
      "epoch": 3.64990234375e-05,
      "step": 9568,
      "training_step_time": 9.904999256134033
    },
    {
      "epoch": 3.64990234375e-05,
      "model_forward_time": 3.1376798152923584,
      "step": 9568
    },
    {
      "epoch": 3.64990234375e-05,
      "step": 9568,
      "training_step_time": 9.90978217124939
    },
    {
      "epoch": 3.64990234375e-05,
      "model_forward_time": 3.134681224822998,
      "step": 9568
    },
    {
      "epoch": 3.64990234375e-05,
      "step": 9568,
      "training_step_time": 10.012441873550415
    },
    {
      "epoch": 3.650283813476563e-05,
      "model_forward_time": 3.4299967288970947,
      "step": 9569
    },
    {
      "epoch": 3.650283813476563e-05,
      "step": 9569,
      "training_step_time": 10.200476169586182
    },
    {
      "epoch": 3.650283813476563e-05,
      "model_forward_time": 3.1348392963409424,
      "step": 9569
    },
    {
      "epoch": 3.650283813476563e-05,
      "step": 9569,
      "training_step_time": 9.90236210823059
    },
    {
      "epoch": 3.650283813476563e-05,
      "model_forward_time": 3.1369824409484863,
      "step": 9569
    },
    {
      "epoch": 3.650283813476563e-05,
      "step": 9569,
      "training_step_time": 9.906587362289429
    },
    {
      "epoch": 3.650283813476563e-05,
      "model_forward_time": 3.131765604019165,
      "step": 9569
    },
    {
      "epoch": 3.650283813476563e-05,
      "step": 9569,
      "training_step_time": 10.302288293838501
    },
    {
      "epoch": 3.650665283203125e-05,
      "grad_norm": 0.4584478735923767,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 9570
    },
    {
      "epoch": 3.650665283203125e-05,
      "model_forward_time": 3.159130334854126,
      "step": 9570
    },
    {
      "dynamics_loss_avg": 0.054753882065415385,
      "epoch": 3.650665283203125e-05,
      "step": 9570
    },
    {
      "action_loss_avg": 0.015769408736377954,
      "epoch": 3.650665283203125e-05,
      "step": 9570
    },
    {
      "epoch": 3.650665283203125e-05,
      "step": 9570,
      "training_step_time": 9.936153888702393
    },
    {
      "epoch": 3.650665283203125e-05,
      "model_forward_time": 3.4136195182800293,
      "step": 9570
    },
    {
      "epoch": 3.650665283203125e-05,
      "step": 9570,
      "training_step_time": 10.1873140335083
    },
    {
      "epoch": 3.650665283203125e-05,
      "model_forward_time": 3.144221305847168,
      "step": 9570
    },
    {
      "epoch": 3.650665283203125e-05,
      "step": 9570,
      "training_step_time": 9.9212327003479
    },
    {
      "epoch": 3.650665283203125e-05,
      "model_forward_time": 3.1461315155029297,
      "step": 9570
    },
    {
      "epoch": 3.650665283203125e-05,
      "step": 9570,
      "training_step_time": 10.275673866271973
    },
    {
      "epoch": 3.6510467529296874e-05,
      "model_forward_time": 3.1527061462402344,
      "step": 9571
    },
    {
      "epoch": 3.6510467529296874e-05,
      "step": 9571,
      "training_step_time": 9.927720785140991
    },
    {
      "epoch": 3.6510467529296874e-05,
      "model_forward_time": 3.144249677658081,
      "step": 9571
    },
    {
      "epoch": 3.6510467529296874e-05,
      "step": 9571,
      "training_step_time": 9.920948505401611
    },
    {
      "epoch": 3.6510467529296874e-05,
      "model_forward_time": 3.4297735691070557,
      "step": 9571
    },
    {
      "epoch": 3.6510467529296874e-05,
      "step": 9571,
      "training_step_time": 10.210859537124634
    },
    {
      "epoch": 3.6510467529296874e-05,
      "model_forward_time": 3.144619941711426,
      "step": 9571
    },
    {
      "epoch": 3.6510467529296874e-05,
      "step": 9571,
      "training_step_time": 10.270684003829956
    },
    {
      "epoch": 3.65142822265625e-05,
      "model_forward_time": 3.154086112976074,
      "step": 9572
    },
    {
      "epoch": 3.65142822265625e-05,
      "step": 9572,
      "training_step_time": 9.930155277252197
    },
    {
      "epoch": 3.65142822265625e-05,
      "model_forward_time": 3.144753932952881,
      "step": 9572
    },
    {
      "epoch": 3.65142822265625e-05,
      "step": 9572,
      "training_step_time": 9.917929410934448
    },
    {
      "epoch": 3.65142822265625e-05,
      "model_forward_time": 3.137395143508911,
      "step": 9572
    },
    {
      "dynamics_loss_avg": 0.05092080570757389,
      "epoch": 3.65142822265625e-05,
      "step": 9572
    },
    {
      "action_loss_avg": 0.018772989651188253,
      "epoch": 3.65142822265625e-05,
      "step": 9572
    },
    {
      "epoch": 3.65142822265625e-05,
      "step": 9572,
      "training_step_time": 9.912449836730957
    },
    {
      "epoch": 3.65142822265625e-05,
      "model_forward_time": 3.3998262882232666,
      "step": 9572
    },
    {
      "epoch": 3.65142822265625e-05,
      "step": 9572,
      "training_step_time": 10.271607637405396
    },
    {
      "epoch": 3.6518096923828127e-05,
      "model_forward_time": 3.142373561859131,
      "step": 9573
    },
    {
      "epoch": 3.6518096923828127e-05,
      "step": 9573,
      "training_step_time": 9.92234754562378
    },
    {
      "epoch": 3.6518096923828127e-05,
      "model_forward_time": 3.1360855102539062,
      "step": 9573
    },
    {
      "epoch": 3.6518096923828127e-05,
      "step": 9573,
      "training_step_time": 9.906405687332153
    },
    {
      "epoch": 3.6518096923828127e-05,
      "model_forward_time": 3.136392831802368,
      "step": 9573
    },
    {
      "epoch": 3.6518096923828127e-05,
      "step": 9573,
      "training_step_time": 9.907481670379639
    },
    {
      "epoch": 3.6518096923828127e-05,
      "model_forward_time": 3.140652656555176,
      "step": 9573
    },
    {
      "epoch": 3.6518096923828127e-05,
      "step": 9573,
      "training_step_time": 10.014893054962158
    },
    {
      "epoch": 3.652191162109375e-05,
      "model_forward_time": 3.4049527645111084,
      "step": 9574
    },
    {
      "epoch": 3.652191162109375e-05,
      "step": 9574,
      "training_step_time": 10.182734966278076
    },
    {
      "epoch": 3.652191162109375e-05,
      "model_forward_time": 3.146221876144409,
      "step": 9574
    },
    {
      "epoch": 3.652191162109375e-05,
      "step": 9574,
      "training_step_time": 9.948759317398071
    },
    {
      "epoch": 3.652191162109375e-05,
      "model_forward_time": 3.139828681945801,
      "step": 9574
    },
    {
      "epoch": 3.652191162109375e-05,
      "step": 9574,
      "training_step_time": 9.914628982543945
    },
    {
      "epoch": 3.652191162109375e-05,
      "model_forward_time": 3.143566608428955,
      "step": 9574
    },
    {
      "epoch": 3.652191162109375e-05,
      "step": 9574,
      "training_step_time": 10.258230447769165
    },
    {
      "epoch": 3.652572631835937e-05,
      "model_forward_time": 3.158647298812866,
      "step": 9575
    },
    {
      "dynamics_loss_avg": 0.06269595082849264,
      "epoch": 3.652572631835937e-05,
      "step": 9575
    },
    {
      "action_loss_avg": 0.018371160514652728,
      "epoch": 3.652572631835937e-05,
      "step": 9575
    },
    {
      "epoch": 3.652572631835937e-05,
      "step": 9575,
      "training_step_time": 9.939971208572388
    },
    {
      "epoch": 3.652572631835937e-05,
      "model_forward_time": 3.445897102355957,
      "step": 9575
    },
    {
      "epoch": 3.652572631835937e-05,
      "step": 9575,
      "training_step_time": 10.218169212341309
    },
    {
      "epoch": 3.652572631835937e-05,
      "model_forward_time": 3.136037588119507,
      "step": 9575
    },
    {
      "epoch": 3.652572631835937e-05,
      "step": 9575,
      "training_step_time": 9.911939144134521
    },
    {
      "epoch": 3.652572631835937e-05,
      "model_forward_time": 3.144519329071045,
      "step": 9575
    },
    {
      "epoch": 3.652572631835937e-05,
      "step": 9575,
      "training_step_time": 10.323301076889038
    },
    {
      "epoch": 3.6529541015625e-05,
      "model_forward_time": 3.164067506790161,
      "step": 9576
    },
    {
      "epoch": 3.6529541015625e-05,
      "step": 9576,
      "training_step_time": 9.94394302368164
    },
    {
      "epoch": 3.6529541015625e-05,
      "model_forward_time": 3.1383543014526367,
      "step": 9576
    },
    {
      "epoch": 3.6529541015625e-05,
      "step": 9576,
      "training_step_time": 9.916686773300171
    },
    {
      "epoch": 3.6529541015625e-05,
      "model_forward_time": 3.4072296619415283,
      "step": 9576
    },
    {
      "epoch": 3.6529541015625e-05,
      "step": 9576,
      "training_step_time": 10.188660860061646
    },
    {
      "epoch": 3.6529541015625e-05,
      "model_forward_time": 3.1455299854278564,
      "step": 9576
    },
    {
      "epoch": 3.6529541015625e-05,
      "step": 9576,
      "training_step_time": 10.24728775024414
    },
    {
      "epoch": 3.6533355712890625e-05,
      "model_forward_time": 3.1541690826416016,
      "step": 9577
    },
    {
      "epoch": 3.6533355712890625e-05,
      "step": 9577,
      "training_step_time": 9.938294410705566
    },
    {
      "epoch": 3.6533355712890625e-05,
      "model_forward_time": 3.142348051071167,
      "step": 9577
    },
    {
      "epoch": 3.6533355712890625e-05,
      "step": 9577,
      "training_step_time": 9.920941352844238
    },
    {
      "epoch": 3.6533355712890625e-05,
      "model_forward_time": 3.135030746459961,
      "step": 9577
    },
    {
      "dynamics_loss_avg": 0.06950217187404632,
      "epoch": 3.6533355712890625e-05,
      "step": 9577
    },
    {
      "action_loss_avg": 0.026915114000439645,
      "epoch": 3.6533355712890625e-05,
      "step": 9577
    },
    {
      "epoch": 3.6533355712890625e-05,
      "step": 9577,
      "training_step_time": 9.910737037658691
    },
    {
      "epoch": 3.6533355712890625e-05,
      "model_forward_time": 3.413853883743286,
      "step": 9577
    },
    {
      "epoch": 3.6533355712890625e-05,
      "step": 9577,
      "training_step_time": 10.280646800994873
    },
    {
      "epoch": 3.653717041015625e-05,
      "model_forward_time": 2.4470443725585938,
      "step": 9578
    },
    {
      "epoch": 3.653717041015625e-05,
      "step": 9578,
      "training_step_time": 7.480298042297363
    },
    {
      "epoch": 3.653717041015625e-05,
      "model_forward_time": 3.131054401397705,
      "step": 9578
    },
    {
      "epoch": 3.653717041015625e-05,
      "step": 9578,
      "training_step_time": 9.902308464050293
    },
    {
      "epoch": 3.653717041015625e-05,
      "model_forward_time": 3.1374242305755615,
      "step": 9578
    },
    {
      "epoch": 3.653717041015625e-05,
      "step": 9578,
      "training_step_time": 9.915291547775269
    },
    {
      "epoch": 3.653717041015625e-05,
      "model_forward_time": 3.1340763568878174,
      "step": 9578
    },
    {
      "epoch": 3.653717041015625e-05,
      "step": 9578,
      "training_step_time": 10.01484227180481
    },
    {
      "epoch": 3.654098510742188e-05,
      "model_forward_time": 3.426677942276001,
      "step": 9579
    },
    {
      "epoch": 3.654098510742188e-05,
      "step": 9579,
      "training_step_time": 10.201650381088257
    },
    {
      "epoch": 3.654098510742188e-05,
      "model_forward_time": 3.1378300189971924,
      "step": 9579
    },
    {
      "epoch": 3.654098510742188e-05,
      "step": 9579,
      "training_step_time": 9.918331861495972
    },
    {
      "epoch": 3.654098510742188e-05,
      "model_forward_time": 3.1404993534088135,
      "step": 9579
    },
    {
      "epoch": 3.654098510742188e-05,
      "step": 9579,
      "training_step_time": 9.92099404335022
    },
    {
      "epoch": 3.654098510742188e-05,
      "model_forward_time": 3.1396584510803223,
      "step": 9579
    },
    {
      "epoch": 3.654098510742188e-05,
      "step": 9579,
      "training_step_time": 10.278544425964355
    },
    {
      "epoch": 3.65447998046875e-05,
      "grad_norm": 0.4968046247959137,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 9580
    },
    {
      "epoch": 3.65447998046875e-05,
      "model_forward_time": 3.1556153297424316,
      "step": 9580
    },
    {
      "dynamics_loss_avg": 0.06254760269075632,
      "epoch": 3.65447998046875e-05,
      "step": 9580
    },
    {
      "action_loss_avg": 0.01992750633507967,
      "epoch": 3.65447998046875e-05,
      "step": 9580
    },
    {
      "epoch": 3.65447998046875e-05,
      "step": 9580,
      "training_step_time": 9.935191869735718
    },
    {
      "epoch": 3.65447998046875e-05,
      "model_forward_time": 3.3908751010894775,
      "step": 9580
    },
    {
      "epoch": 3.65447998046875e-05,
      "step": 9580,
      "training_step_time": 10.164035320281982
    },
    {
      "epoch": 3.65447998046875e-05,
      "model_forward_time": 3.139366865158081,
      "step": 9580
    },
    {
      "epoch": 3.65447998046875e-05,
      "step": 9580,
      "training_step_time": 9.911306142807007
    },
    {
      "epoch": 3.65447998046875e-05,
      "model_forward_time": 3.1344194412231445,
      "step": 9580
    },
    {
      "epoch": 3.65447998046875e-05,
      "step": 9580,
      "training_step_time": 10.239713430404663
    },
    {
      "epoch": 3.654861450195312e-05,
      "model_forward_time": 3.149449348449707,
      "step": 9581
    },
    {
      "epoch": 3.654861450195312e-05,
      "step": 9581,
      "training_step_time": 9.921391725540161
    },
    {
      "epoch": 3.654861450195312e-05,
      "model_forward_time": 3.1362814903259277,
      "step": 9581
    },
    {
      "epoch": 3.654861450195312e-05,
      "step": 9581,
      "training_step_time": 9.91152310371399
    },
    {
      "epoch": 3.654861450195312e-05,
      "model_forward_time": 3.418001174926758,
      "step": 9581
    },
    {
      "epoch": 3.654861450195312e-05,
      "step": 9581,
      "training_step_time": 10.194729328155518
    },
    {
      "epoch": 3.654861450195312e-05,
      "model_forward_time": 3.135915994644165,
      "step": 9581
    },
    {
      "epoch": 3.654861450195312e-05,
      "step": 9581,
      "training_step_time": 10.462781190872192
    },
    {
      "epoch": 3.655242919921875e-05,
      "model_forward_time": 3.1564090251922607,
      "step": 9582
    },
    {
      "epoch": 3.655242919921875e-05,
      "step": 9582,
      "training_step_time": 9.938449621200562
    },
    {
      "epoch": 3.655242919921875e-05,
      "model_forward_time": 3.1385655403137207,
      "step": 9582
    },
    {
      "epoch": 3.655242919921875e-05,
      "step": 9582,
      "training_step_time": 9.913216352462769
    },
    {
      "epoch": 3.655242919921875e-05,
      "model_forward_time": 3.1347241401672363,
      "step": 9582
    },
    {
      "dynamics_loss_avg": 0.055794842168688775,
      "epoch": 3.655242919921875e-05,
      "step": 9582
    },
    {
      "action_loss_avg": 0.020031414832919835,
      "epoch": 3.655242919921875e-05,
      "step": 9582
    },
    {
      "epoch": 3.655242919921875e-05,
      "step": 9582,
      "training_step_time": 9.918859004974365
    },
    {
      "epoch": 3.655242919921875e-05,
      "model_forward_time": 3.1373395919799805,
      "step": 9582
    },
    {
      "epoch": 3.655242919921875e-05,
      "step": 9582,
      "training_step_time": 10.292863607406616
    },
    {
      "epoch": 3.6556243896484376e-05,
      "model_forward_time": 3.1637556552886963,
      "step": 9583
    },
    {
      "epoch": 3.6556243896484376e-05,
      "step": 9583,
      "training_step_time": 9.947619676589966
    },
    {
      "epoch": 3.6556243896484376e-05,
      "model_forward_time": 3.141801595687866,
      "step": 9583
    },
    {
      "epoch": 3.6556243896484376e-05,
      "step": 9583,
      "training_step_time": 9.912672281265259
    },
    {
      "epoch": 3.6556243896484376e-05,
      "model_forward_time": 3.136915683746338,
      "step": 9583
    },
    {
      "epoch": 3.6556243896484376e-05,
      "step": 9583,
      "training_step_time": 9.905815601348877
    },
    {
      "epoch": 3.6556243896484376e-05,
      "model_forward_time": 3.140779972076416,
      "step": 9583
    },
    {
      "epoch": 3.6556243896484376e-05,
      "step": 9583,
      "training_step_time": 10.329702615737915
    },
    {
      "epoch": 3.656005859375e-05,
      "model_forward_time": 3.4161393642425537,
      "step": 9584
    },
    {
      "epoch": 3.656005859375e-05,
      "step": 9584,
      "training_step_time": 10.199422597885132
    },
    {
      "epoch": 3.656005859375e-05,
      "model_forward_time": 3.145277261734009,
      "step": 9584
    },
    {
      "epoch": 3.656005859375e-05,
      "step": 9584,
      "training_step_time": 9.92461371421814
    },
    {
      "epoch": 3.656005859375e-05,
      "model_forward_time": 3.140756130218506,
      "step": 9584
    },
    {
      "epoch": 3.656005859375e-05,
      "step": 9584,
      "training_step_time": 9.92057466506958
    },
    {
      "epoch": 3.656005859375e-05,
      "model_forward_time": 3.145923614501953,
      "step": 9584
    },
    {
      "epoch": 3.656005859375e-05,
      "step": 9584,
      "training_step_time": 10.322043657302856
    },
    {
      "epoch": 3.656387329101563e-05,
      "model_forward_time": 3.1572537422180176,
      "step": 9585
    },
    {
      "dynamics_loss_avg": 0.04264033325016499,
      "epoch": 3.656387329101563e-05,
      "step": 9585
    },
    {
      "action_loss_avg": 0.013688064087182284,
      "epoch": 3.656387329101563e-05,
      "step": 9585
    },
    {
      "epoch": 3.656387329101563e-05,
      "step": 9585,
      "training_step_time": 9.942856073379517
    },
    {
      "epoch": 3.656387329101563e-05,
      "model_forward_time": 3.1451117992401123,
      "step": 9585
    },
    {
      "epoch": 3.656387329101563e-05,
      "step": 9585,
      "training_step_time": 10.207465171813965
    },
    {
      "epoch": 3.656387329101563e-05,
      "model_forward_time": 3.1509006023406982,
      "step": 9585
    },
    {
      "epoch": 3.656387329101563e-05,
      "step": 9585,
      "training_step_time": 9.936368703842163
    },
    {
      "epoch": 3.656387329101563e-05,
      "model_forward_time": 3.1495463848114014,
      "step": 9585
    },
    {
      "epoch": 3.656387329101563e-05,
      "step": 9585,
      "training_step_time": 10.323039531707764
    },
    {
      "epoch": 3.656768798828125e-05,
      "model_forward_time": 3.1640162467956543,
      "step": 9586
    },
    {
      "epoch": 3.656768798828125e-05,
      "step": 9586,
      "training_step_time": 9.949206829071045
    },
    {
      "epoch": 3.656768798828125e-05,
      "model_forward_time": 3.1450185775756836,
      "step": 9586
    },
    {
      "epoch": 3.656768798828125e-05,
      "step": 9586,
      "training_step_time": 9.927903652191162
    },
    {
      "epoch": 3.656768798828125e-05,
      "model_forward_time": 3.407958984375,
      "step": 9586
    },
    {
      "epoch": 3.656768798828125e-05,
      "step": 9586,
      "training_step_time": 10.18760633468628
    },
    {
      "epoch": 3.656768798828125e-05,
      "model_forward_time": 3.1468915939331055,
      "step": 9586
    },
    {
      "epoch": 3.656768798828125e-05,
      "step": 9586,
      "training_step_time": 10.313546180725098
    },
    {
      "epoch": 3.6571502685546874e-05,
      "model_forward_time": 3.1557865142822266,
      "step": 9587
    },
    {
      "epoch": 3.6571502685546874e-05,
      "step": 9587,
      "training_step_time": 9.93898892402649
    },
    {
      "epoch": 3.6571502685546874e-05,
      "model_forward_time": 3.1277339458465576,
      "step": 9587
    },
    {
      "epoch": 3.6571502685546874e-05,
      "step": 9587,
      "training_step_time": 9.89717411994934
    },
    {
      "epoch": 3.6571502685546874e-05,
      "model_forward_time": 3.1336917877197266,
      "step": 9587
    },
    {
      "dynamics_loss_avg": 0.0648321308195591,
      "epoch": 3.6571502685546874e-05,
      "step": 9587
    },
    {
      "action_loss_avg": 0.024107427056878804,
      "epoch": 3.6571502685546874e-05,
      "step": 9587
    },
    {
      "epoch": 3.6571502685546874e-05,
      "step": 9587,
      "training_step_time": 9.906792879104614
    },
    {
      "epoch": 3.6571502685546874e-05,
      "model_forward_time": 3.1302530765533447,
      "step": 9587
    },
    {
      "epoch": 3.6571502685546874e-05,
      "step": 9587,
      "training_step_time": 10.290650367736816
    },
    {
      "epoch": 3.65753173828125e-05,
      "model_forward_time": 3.1499736309051514,
      "step": 9588
    },
    {
      "epoch": 3.65753173828125e-05,
      "step": 9588,
      "training_step_time": 9.922999620437622
    },
    {
      "epoch": 3.65753173828125e-05,
      "model_forward_time": 3.1327857971191406,
      "step": 9588
    },
    {
      "epoch": 3.65753173828125e-05,
      "step": 9588,
      "training_step_time": 9.90293288230896
    },
    {
      "epoch": 3.65753173828125e-05,
      "model_forward_time": 2.4307708740234375,
      "step": 9588
    },
    {
      "epoch": 3.65753173828125e-05,
      "step": 9588,
      "training_step_time": 7.45691442489624
    },
    {
      "epoch": 3.65753173828125e-05,
      "model_forward_time": 3.127044916152954,
      "step": 9588
    },
    {
      "epoch": 3.65753173828125e-05,
      "step": 9588,
      "training_step_time": 10.007527112960815
    },
    {
      "epoch": 3.6579132080078127e-05,
      "model_forward_time": 3.422755718231201,
      "step": 9589
    },
    {
      "epoch": 3.6579132080078127e-05,
      "step": 9589,
      "training_step_time": 10.19399094581604
    },
    {
      "epoch": 3.6579132080078127e-05,
      "model_forward_time": 3.1312966346740723,
      "step": 9589
    },
    {
      "epoch": 3.6579132080078127e-05,
      "step": 9589,
      "training_step_time": 9.902987480163574
    },
    {
      "epoch": 3.6579132080078127e-05,
      "model_forward_time": 3.1364519596099854,
      "step": 9589
    },
    {
      "epoch": 3.6579132080078127e-05,
      "step": 9589,
      "training_step_time": 9.913590431213379
    },
    {
      "epoch": 3.6579132080078127e-05,
      "model_forward_time": 3.133437156677246,
      "step": 9589
    },
    {
      "epoch": 3.6579132080078127e-05,
      "step": 9589,
      "training_step_time": 10.317277193069458
    },
    {
      "epoch": 3.658294677734375e-05,
      "grad_norm": 0.42017892003059387,
      "learning_rate": 1e-05,
      "loss": 0.0783,
      "step": 9590
    },
    {
      "epoch": 3.658294677734375e-05,
      "model_forward_time": 3.1461539268493652,
      "step": 9590
    },
    {
      "dynamics_loss_avg": 0.05893856696784496,
      "epoch": 3.658294677734375e-05,
      "step": 9590
    },
    {
      "action_loss_avg": 0.01991810938343406,
      "epoch": 3.658294677734375e-05,
      "step": 9590
    },
    {
      "epoch": 3.658294677734375e-05,
      "step": 9590,
      "training_step_time": 9.924143552780151
    },
    {
      "epoch": 3.658294677734375e-05,
      "model_forward_time": 3.4087915420532227,
      "step": 9590
    },
    {
      "epoch": 3.658294677734375e-05,
      "step": 9590,
      "training_step_time": 10.178986310958862
    },
    {
      "epoch": 3.658294677734375e-05,
      "model_forward_time": 3.1385791301727295,
      "step": 9590
    },
    {
      "epoch": 3.658294677734375e-05,
      "step": 9590,
      "training_step_time": 9.91524863243103
    },
    {
      "epoch": 3.658294677734375e-05,
      "model_forward_time": 3.131270170211792,
      "step": 9590
    },
    {
      "epoch": 3.658294677734375e-05,
      "step": 9590,
      "training_step_time": 10.275944232940674
    },
    {
      "epoch": 3.658676147460937e-05,
      "model_forward_time": 3.1431937217712402,
      "step": 9591
    },
    {
      "epoch": 3.658676147460937e-05,
      "step": 9591,
      "training_step_time": 9.927682876586914
    },
    {
      "epoch": 3.658676147460937e-05,
      "model_forward_time": 3.1383659839630127,
      "step": 9591
    },
    {
      "epoch": 3.658676147460937e-05,
      "step": 9591,
      "training_step_time": 9.917632102966309
    },
    {
      "epoch": 3.658676147460937e-05,
      "model_forward_time": 3.136942148208618,
      "step": 9591
    },
    {
      "epoch": 3.658676147460937e-05,
      "step": 9591,
      "training_step_time": 10.203001737594604
    },
    {
      "epoch": 3.658676147460937e-05,
      "model_forward_time": 3.13588809967041,
      "step": 9591
    },
    {
      "epoch": 3.658676147460937e-05,
      "step": 9591,
      "training_step_time": 10.337852001190186
    },
    {
      "epoch": 3.6590576171875e-05,
      "model_forward_time": 3.1430482864379883,
      "step": 9592
    },
    {
      "epoch": 3.6590576171875e-05,
      "step": 9592,
      "training_step_time": 9.922303676605225
    },
    {
      "epoch": 3.6590576171875e-05,
      "model_forward_time": 3.1408073902130127,
      "step": 9592
    },
    {
      "epoch": 3.6590576171875e-05,
      "step": 9592,
      "training_step_time": 9.916795492172241
    },
    {
      "epoch": 3.6590576171875e-05,
      "model_forward_time": 3.1422641277313232,
      "step": 9592
    },
    {
      "dynamics_loss_avg": 0.0642837880179286,
      "epoch": 3.6590576171875e-05,
      "step": 9592
    },
    {
      "action_loss_avg": 0.019127300567924978,
      "epoch": 3.6590576171875e-05,
      "step": 9592
    },
    {
      "epoch": 3.6590576171875e-05,
      "step": 9592,
      "training_step_time": 9.9256010055542
    },
    {
      "epoch": 3.6590576171875e-05,
      "model_forward_time": 3.1432135105133057,
      "step": 9592
    },
    {
      "epoch": 3.6590576171875e-05,
      "step": 9592,
      "training_step_time": 10.293888092041016
    },
    {
      "epoch": 3.6594390869140625e-05,
      "model_forward_time": 3.1421546936035156,
      "step": 9593
    },
    {
      "epoch": 3.6594390869140625e-05,
      "step": 9593,
      "training_step_time": 9.91921091079712
    },
    {
      "epoch": 3.6594390869140625e-05,
      "model_forward_time": 3.138617753982544,
      "step": 9593
    },
    {
      "epoch": 3.6594390869140625e-05,
      "step": 9593,
      "training_step_time": 9.915241003036499
    },
    {
      "epoch": 3.6594390869140625e-05,
      "model_forward_time": 3.135138750076294,
      "step": 9593
    },
    {
      "epoch": 3.6594390869140625e-05,
      "step": 9593,
      "training_step_time": 9.912429332733154
    },
    {
      "epoch": 3.6594390869140625e-05,
      "model_forward_time": 3.138705015182495,
      "step": 9593
    },
    {
      "epoch": 3.6594390869140625e-05,
      "step": 9593,
      "training_step_time": 10.019311904907227
    },
    {
      "epoch": 3.659820556640625e-05,
      "model_forward_time": 3.1620852947235107,
      "step": 9594
    },
    {
      "epoch": 3.659820556640625e-05,
      "step": 9594,
      "training_step_time": 10.24709939956665
    },
    {
      "epoch": 3.659820556640625e-05,
      "model_forward_time": 3.160616159439087,
      "step": 9594
    },
    {
      "epoch": 3.659820556640625e-05,
      "step": 9594,
      "training_step_time": 9.943379878997803
    },
    {
      "epoch": 3.659820556640625e-05,
      "model_forward_time": 3.1457996368408203,
      "step": 9594
    },
    {
      "epoch": 3.659820556640625e-05,
      "step": 9594,
      "training_step_time": 9.924302101135254
    },
    {
      "epoch": 3.659820556640625e-05,
      "model_forward_time": 3.146986722946167,
      "step": 9594
    },
    {
      "epoch": 3.659820556640625e-05,
      "step": 9594,
      "training_step_time": 10.324877738952637
    },
    {
      "epoch": 3.660202026367188e-05,
      "model_forward_time": 3.1701855659484863,
      "step": 9595
    },
    {
      "dynamics_loss_avg": 0.05026281028985977,
      "epoch": 3.660202026367188e-05,
      "step": 9595
    },
    {
      "action_loss_avg": 0.015897554624825717,
      "epoch": 3.660202026367188e-05,
      "step": 9595
    },
    {
      "epoch": 3.660202026367188e-05,
      "step": 9595,
      "training_step_time": 9.953691005706787
    },
    {
      "epoch": 3.660202026367188e-05,
      "model_forward_time": 3.4192044734954834,
      "step": 9595
    },
    {
      "epoch": 3.660202026367188e-05,
      "step": 9595,
      "training_step_time": 10.194152116775513
    },
    {
      "epoch": 3.660202026367188e-05,
      "model_forward_time": 3.1354007720947266,
      "step": 9595
    },
    {
      "epoch": 3.660202026367188e-05,
      "step": 9595,
      "training_step_time": 9.909976482391357
    },
    {
      "epoch": 3.660202026367188e-05,
      "model_forward_time": 3.1391074657440186,
      "step": 9595
    },
    {
      "epoch": 3.660202026367188e-05,
      "step": 9595,
      "training_step_time": 10.230448484420776
    },
    {
      "epoch": 3.66058349609375e-05,
      "model_forward_time": 3.1458239555358887,
      "step": 9596
    },
    {
      "epoch": 3.66058349609375e-05,
      "step": 9596,
      "training_step_time": 9.921181917190552
    },
    {
      "epoch": 3.66058349609375e-05,
      "model_forward_time": 3.127488374710083,
      "step": 9596
    },
    {
      "epoch": 3.66058349609375e-05,
      "step": 9596,
      "training_step_time": 9.899779558181763
    },
    {
      "epoch": 3.66058349609375e-05,
      "model_forward_time": 3.1309010982513428,
      "step": 9596
    },
    {
      "epoch": 3.66058349609375e-05,
      "step": 9596,
      "training_step_time": 10.210607528686523
    },
    {
      "epoch": 3.66058349609375e-05,
      "model_forward_time": 3.132563829421997,
      "step": 9596
    },
    {
      "epoch": 3.66058349609375e-05,
      "step": 9596,
      "training_step_time": 10.278308629989624
    },
    {
      "epoch": 3.660964965820312e-05,
      "model_forward_time": 3.1532862186431885,
      "step": 9597
    },
    {
      "epoch": 3.660964965820312e-05,
      "step": 9597,
      "training_step_time": 9.92800498008728
    },
    {
      "epoch": 3.660964965820312e-05,
      "model_forward_time": 3.134009599685669,
      "step": 9597
    },
    {
      "epoch": 3.660964965820312e-05,
      "step": 9597,
      "training_step_time": 9.907341957092285
    },
    {
      "epoch": 3.660964965820312e-05,
      "model_forward_time": 3.138059377670288,
      "step": 9597
    },
    {
      "dynamics_loss_avg": 0.050083721056580544,
      "epoch": 3.660964965820312e-05,
      "step": 9597
    },
    {
      "action_loss_avg": 0.013313964288681746,
      "epoch": 3.660964965820312e-05,
      "step": 9597
    },
    {
      "epoch": 3.660964965820312e-05,
      "step": 9597,
      "training_step_time": 9.91381025314331
    },
    {
      "epoch": 3.660964965820312e-05,
      "model_forward_time": 3.4384028911590576,
      "step": 9597
    },
    {
      "epoch": 3.660964965820312e-05,
      "step": 9597,
      "training_step_time": 10.34927773475647
    },
    {
      "epoch": 3.661346435546875e-05,
      "model_forward_time": 3.1518261432647705,
      "step": 9598
    },
    {
      "epoch": 3.661346435546875e-05,
      "step": 9598,
      "training_step_time": 9.930450677871704
    },
    {
      "epoch": 3.661346435546875e-05,
      "model_forward_time": 3.1364333629608154,
      "step": 9598
    },
    {
      "epoch": 3.661346435546875e-05,
      "step": 9598,
      "training_step_time": 9.911418437957764
    },
    {
      "epoch": 3.661346435546875e-05,
      "model_forward_time": 3.139228582382202,
      "step": 9598
    },
    {
      "epoch": 3.661346435546875e-05,
      "step": 9598,
      "training_step_time": 9.914109945297241
    },
    {
      "epoch": 3.661346435546875e-05,
      "model_forward_time": 3.136104106903076,
      "step": 9598
    },
    {
      "epoch": 3.661346435546875e-05,
      "step": 9598,
      "training_step_time": 10.320585250854492
    },
    {
      "epoch": 3.6617279052734376e-05,
      "model_forward_time": 3.4739060401916504,
      "step": 9599
    },
    {
      "epoch": 3.6617279052734376e-05,
      "step": 9599,
      "training_step_time": 10.257124900817871
    },
    {
      "epoch": 3.6617279052734376e-05,
      "model_forward_time": 3.1407859325408936,
      "step": 9599
    },
    {
      "epoch": 3.6617279052734376e-05,
      "step": 9599,
      "training_step_time": 9.919713497161865
    },
    {
      "epoch": 3.6617279052734376e-05,
      "model_forward_time": 3.1436524391174316,
      "step": 9599
    },
    {
      "epoch": 3.6617279052734376e-05,
      "step": 9599,
      "training_step_time": 9.922090768814087
    },
    {
      "epoch": 3.6617279052734376e-05,
      "model_forward_time": 3.1350014209747314,
      "step": 9599
    },
    {
      "epoch": 3.6617279052734376e-05,
      "step": 9599,
      "training_step_time": 10.307000637054443
    },
    {
      "epoch": 3.662109375e-05,
      "grad_norm": 0.4884834885597229,
      "learning_rate": 1e-05,
      "loss": 0.0728,
      "step": 9600
    },
    {
      "epoch": 3.662109375e-05,
      "model_forward_time": 3.14646577835083,
      "step": 9600
    },
    {
      "dynamics_loss_avg": 0.0645367693156004,
      "epoch": 3.662109375e-05,
      "step": 9600
    },
    {
      "action_loss_avg": 0.022967802081257104,
      "epoch": 3.662109375e-05,
      "step": 9600
    },
    {
      "epoch": 3.662109375e-05,
      "step": 9600,
      "training_step_time": 9.925345182418823
    },
    {
      "epoch": 3.662109375e-05,
      "model_forward_time": 3.400444984436035,
      "step": 9600
    },
    {
      "epoch": 3.662109375e-05,
      "step": 9600,
      "training_step_time": 10.173918962478638
    },
    {
      "epoch": 3.662109375e-05,
      "model_forward_time": 3.138737916946411,
      "step": 9600
    },
    {
      "epoch": 3.662109375e-05,
      "step": 9600,
      "training_step_time": 9.91275429725647
    },
    {
      "epoch": 3.662109375e-05,
      "model_forward_time": 3.135646104812622,
      "step": 9600
    },
    {
      "epoch": 3.662109375e-05,
      "step": 9600,
      "training_step_time": 10.009732961654663
    },
    {
      "epoch": 3.662490844726563e-05,
      "model_forward_time": 3.1641623973846436,
      "step": 9601
    },
    {
      "epoch": 3.662490844726563e-05,
      "step": 9601,
      "training_step_time": 9.941669940948486
    },
    {
      "epoch": 3.662490844726563e-05,
      "model_forward_time": 3.1474173069000244,
      "step": 9601
    },
    {
      "epoch": 3.662490844726563e-05,
      "step": 9601,
      "training_step_time": 9.92358660697937
    },
    {
      "epoch": 3.662490844726563e-05,
      "model_forward_time": 3.434676170349121,
      "step": 9601
    },
    {
      "epoch": 3.662490844726563e-05,
      "step": 9601,
      "training_step_time": 10.204071998596191
    },
    {
      "epoch": 3.662490844726563e-05,
      "model_forward_time": 3.1470582485198975,
      "step": 9601
    },
    {
      "epoch": 3.662490844726563e-05,
      "step": 9601,
      "training_step_time": 10.315074443817139
    },
    {
      "epoch": 3.662872314453125e-05,
      "model_forward_time": 3.147449254989624,
      "step": 9602
    },
    {
      "epoch": 3.662872314453125e-05,
      "step": 9602,
      "training_step_time": 9.918347835540771
    },
    {
      "epoch": 3.662872314453125e-05,
      "model_forward_time": 3.145787239074707,
      "step": 9602
    },
    {
      "epoch": 3.662872314453125e-05,
      "step": 9602,
      "training_step_time": 9.919269800186157
    },
    {
      "epoch": 3.662872314453125e-05,
      "model_forward_time": 3.1443898677825928,
      "step": 9602
    },
    {
      "dynamics_loss_avg": 0.06792313754558563,
      "epoch": 3.662872314453125e-05,
      "step": 9602
    },
    {
      "action_loss_avg": 0.024613550677895547,
      "epoch": 3.662872314453125e-05,
      "step": 9602
    },
    {
      "epoch": 3.662872314453125e-05,
      "step": 9602,
      "training_step_time": 9.926925420761108
    },
    {
      "epoch": 3.662872314453125e-05,
      "model_forward_time": 3.1478962898254395,
      "step": 9602
    },
    {
      "epoch": 3.662872314453125e-05,
      "step": 9602,
      "training_step_time": 10.308908700942993
    },
    {
      "epoch": 3.6632537841796874e-05,
      "model_forward_time": 3.166856527328491,
      "step": 9603
    },
    {
      "epoch": 3.6632537841796874e-05,
      "step": 9603,
      "training_step_time": 9.952745914459229
    },
    {
      "epoch": 3.6632537841796874e-05,
      "model_forward_time": 3.149744987487793,
      "step": 9603
    },
    {
      "epoch": 3.6632537841796874e-05,
      "step": 9603,
      "training_step_time": 9.92908239364624
    },
    {
      "epoch": 3.6632537841796874e-05,
      "model_forward_time": 3.1517910957336426,
      "step": 9603
    },
    {
      "epoch": 3.6632537841796874e-05,
      "step": 9603,
      "training_step_time": 9.934355974197388
    },
    {
      "epoch": 3.6632537841796874e-05,
      "model_forward_time": 3.160813808441162,
      "step": 9603
    },
    {
      "epoch": 3.6632537841796874e-05,
      "step": 9603,
      "training_step_time": 10.282638549804688
    },
    {
      "epoch": 3.66363525390625e-05,
      "model_forward_time": 3.4602479934692383,
      "step": 9604
    },
    {
      "epoch": 3.66363525390625e-05,
      "step": 9604,
      "training_step_time": 10.245065689086914
    },
    {
      "epoch": 3.66363525390625e-05,
      "model_forward_time": 3.153144598007202,
      "step": 9604
    },
    {
      "epoch": 3.66363525390625e-05,
      "step": 9604,
      "training_step_time": 9.932750701904297
    },
    {
      "epoch": 3.66363525390625e-05,
      "model_forward_time": 3.1463663578033447,
      "step": 9604
    },
    {
      "epoch": 3.66363525390625e-05,
      "step": 9604,
      "training_step_time": 9.921440362930298
    },
    {
      "epoch": 3.66363525390625e-05,
      "model_forward_time": 3.1523358821868896,
      "step": 9604
    },
    {
      "epoch": 3.66363525390625e-05,
      "step": 9604,
      "training_step_time": 10.33063817024231
    },
    {
      "epoch": 3.664016723632813e-05,
      "model_forward_time": 3.1435065269470215,
      "step": 9605
    },
    {
      "dynamics_loss_avg": 0.0709509577602148,
      "epoch": 3.664016723632813e-05,
      "step": 9605
    },
    {
      "action_loss_avg": 0.027157011535018684,
      "epoch": 3.664016723632813e-05,
      "step": 9605
    },
    {
      "epoch": 3.664016723632813e-05,
      "step": 9605,
      "training_step_time": 9.922626733779907
    },
    {
      "epoch": 3.664016723632813e-05,
      "model_forward_time": 3.128925323486328,
      "step": 9605
    },
    {
      "epoch": 3.664016723632813e-05,
      "step": 9605,
      "training_step_time": 10.193572759628296
    },
    {
      "epoch": 3.664016723632813e-05,
      "model_forward_time": 3.1289825439453125,
      "step": 9605
    },
    {
      "epoch": 3.664016723632813e-05,
      "step": 9605,
      "training_step_time": 9.901208639144897
    },
    {
      "epoch": 3.664016723632813e-05,
      "model_forward_time": 3.1374154090881348,
      "step": 9605
    },
    {
      "epoch": 3.664016723632813e-05,
      "step": 9605,
      "training_step_time": 10.022809743881226
    },
    {
      "epoch": 3.664398193359375e-05,
      "model_forward_time": 3.1545164585113525,
      "step": 9606
    },
    {
      "epoch": 3.664398193359375e-05,
      "step": 9606,
      "training_step_time": 9.926225662231445
    },
    {
      "epoch": 3.664398193359375e-05,
      "model_forward_time": 3.1393752098083496,
      "step": 9606
    },
    {
      "epoch": 3.664398193359375e-05,
      "step": 9606,
      "training_step_time": 9.913169384002686
    },
    {
      "epoch": 3.664398193359375e-05,
      "model_forward_time": 3.3980307579040527,
      "step": 9606
    },
    {
      "epoch": 3.664398193359375e-05,
      "step": 9606,
      "training_step_time": 10.172811031341553
    },
    {
      "epoch": 3.664398193359375e-05,
      "model_forward_time": 3.145575523376465,
      "step": 9606
    },
    {
      "epoch": 3.664398193359375e-05,
      "step": 9606,
      "training_step_time": 10.25454330444336
    },
    {
      "epoch": 3.664779663085937e-05,
      "model_forward_time": 3.1481354236602783,
      "step": 9607
    },
    {
      "epoch": 3.664779663085937e-05,
      "step": 9607,
      "training_step_time": 9.92328953742981
    },
    {
      "epoch": 3.664779663085937e-05,
      "model_forward_time": 3.1364173889160156,
      "step": 9607
    },
    {
      "epoch": 3.664779663085937e-05,
      "step": 9607,
      "training_step_time": 9.908419847488403
    },
    {
      "epoch": 3.664779663085937e-05,
      "model_forward_time": 3.1402742862701416,
      "step": 9607
    },
    {
      "dynamics_loss_avg": 0.057674144580960277,
      "epoch": 3.664779663085937e-05,
      "step": 9607
    },
    {
      "action_loss_avg": 0.018672868970315903,
      "epoch": 3.664779663085937e-05,
      "step": 9607
    },
    {
      "epoch": 3.664779663085937e-05,
      "step": 9607,
      "training_step_time": 9.912293195724487
    },
    {
      "epoch": 3.664779663085937e-05,
      "model_forward_time": 3.1370482444763184,
      "step": 9607
    },
    {
      "epoch": 3.664779663085937e-05,
      "step": 9607,
      "training_step_time": 10.27347469329834
    },
    {
      "epoch": 3.6651611328125e-05,
      "model_forward_time": 3.1601645946502686,
      "step": 9608
    },
    {
      "epoch": 3.6651611328125e-05,
      "step": 9608,
      "training_step_time": 9.938387632369995
    },
    {
      "epoch": 3.6651611328125e-05,
      "model_forward_time": 3.1365225315093994,
      "step": 9608
    },
    {
      "epoch": 3.6651611328125e-05,
      "step": 9608,
      "training_step_time": 9.910162925720215
    },
    {
      "epoch": 3.6651611328125e-05,
      "model_forward_time": 3.139984607696533,
      "step": 9608
    },
    {
      "epoch": 3.6651611328125e-05,
      "step": 9608,
      "training_step_time": 9.913860321044922
    },
    {
      "epoch": 3.6651611328125e-05,
      "model_forward_time": 3.1384522914886475,
      "step": 9608
    },
    {
      "epoch": 3.6651611328125e-05,
      "step": 9608,
      "training_step_time": 10.026425838470459
    },
    {
      "epoch": 3.6655426025390625e-05,
      "model_forward_time": 3.4413247108459473,
      "step": 9609
    },
    {
      "epoch": 3.6655426025390625e-05,
      "step": 9609,
      "training_step_time": 10.213335037231445
    },
    {
      "epoch": 3.6655426025390625e-05,
      "model_forward_time": 3.1366512775421143,
      "step": 9609
    },
    {
      "epoch": 3.6655426025390625e-05,
      "step": 9609,
      "training_step_time": 9.909521341323853
    },
    {
      "epoch": 3.6655426025390625e-05,
      "model_forward_time": 3.140096664428711,
      "step": 9609
    },
    {
      "epoch": 3.6655426025390625e-05,
      "step": 9609,
      "training_step_time": 9.911319971084595
    },
    {
      "epoch": 3.6655426025390625e-05,
      "model_forward_time": 3.1354284286499023,
      "step": 9609
    },
    {
      "epoch": 3.6655426025390625e-05,
      "step": 9609,
      "training_step_time": 10.286875486373901
    },
    {
      "epoch": 3.665924072265625e-05,
      "grad_norm": 0.3380354642868042,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 9610
    },
    {
      "epoch": 3.665924072265625e-05,
      "model_forward_time": 3.1427903175354004,
      "step": 9610
    },
    {
      "dynamics_loss_avg": 0.07140912525355816,
      "epoch": 3.665924072265625e-05,
      "step": 9610
    },
    {
      "action_loss_avg": 0.023132942989468573,
      "epoch": 3.665924072265625e-05,
      "step": 9610
    },
    {
      "epoch": 3.665924072265625e-05,
      "step": 9610,
      "training_step_time": 9.914809703826904
    },
    {
      "epoch": 3.665924072265625e-05,
      "model_forward_time": 3.391317844390869,
      "step": 9610
    },
    {
      "epoch": 3.665924072265625e-05,
      "step": 9610,
      "training_step_time": 10.157809734344482
    },
    {
      "epoch": 3.665924072265625e-05,
      "model_forward_time": 3.1351206302642822,
      "step": 9610
    },
    {
      "epoch": 3.665924072265625e-05,
      "step": 9610,
      "training_step_time": 9.900225162506104
    },
    {
      "epoch": 3.665924072265625e-05,
      "model_forward_time": 3.1362743377685547,
      "step": 9610
    },
    {
      "epoch": 3.665924072265625e-05,
      "step": 9610,
      "training_step_time": 10.287595748901367
    },
    {
      "epoch": 3.666305541992188e-05,
      "model_forward_time": 3.1392810344696045,
      "step": 9611
    },
    {
      "epoch": 3.666305541992188e-05,
      "step": 9611,
      "training_step_time": 9.904788255691528
    },
    {
      "epoch": 3.666305541992188e-05,
      "model_forward_time": 3.1366071701049805,
      "step": 9611
    },
    {
      "epoch": 3.666305541992188e-05,
      "step": 9611,
      "training_step_time": 9.901134967803955
    },
    {
      "epoch": 3.666305541992188e-05,
      "model_forward_time": 3.45092511177063,
      "step": 9611
    },
    {
      "epoch": 3.666305541992188e-05,
      "step": 9611,
      "training_step_time": 10.21911907196045
    },
    {
      "epoch": 3.666305541992188e-05,
      "model_forward_time": 3.131640911102295,
      "step": 9611
    },
    {
      "epoch": 3.666305541992188e-05,
      "step": 9611,
      "training_step_time": 10.01858401298523
    },
    {
      "epoch": 3.66668701171875e-05,
      "model_forward_time": 3.153637409210205,
      "step": 9612
    },
    {
      "epoch": 3.66668701171875e-05,
      "step": 9612,
      "training_step_time": 9.9337899684906
    },
    {
      "epoch": 3.66668701171875e-05,
      "model_forward_time": 3.1448097229003906,
      "step": 9612
    },
    {
      "epoch": 3.66668701171875e-05,
      "step": 9612,
      "training_step_time": 9.91156792640686
    },
    {
      "epoch": 3.66668701171875e-05,
      "model_forward_time": 3.142904043197632,
      "step": 9612
    },
    {
      "dynamics_loss_avg": 0.05548812747001648,
      "epoch": 3.66668701171875e-05,
      "step": 9612
    },
    {
      "action_loss_avg": 0.017458105832338332,
      "epoch": 3.66668701171875e-05,
      "step": 9612
    },
    {
      "epoch": 3.66668701171875e-05,
      "step": 9612,
      "training_step_time": 9.915590047836304
    },
    {
      "epoch": 3.66668701171875e-05,
      "model_forward_time": 3.410238027572632,
      "step": 9612
    },
    {
      "epoch": 3.66668701171875e-05,
      "step": 9612,
      "training_step_time": 10.333376407623291
    },
    {
      "epoch": 3.667068481445312e-05,
      "model_forward_time": 3.160824775695801,
      "step": 9613
    },
    {
      "epoch": 3.667068481445312e-05,
      "step": 9613,
      "training_step_time": 9.936511039733887
    },
    {
      "epoch": 3.667068481445312e-05,
      "model_forward_time": 3.144576072692871,
      "step": 9613
    },
    {
      "epoch": 3.667068481445312e-05,
      "step": 9613,
      "training_step_time": 9.917317390441895
    },
    {
      "epoch": 3.667068481445312e-05,
      "model_forward_time": 3.1433494091033936,
      "step": 9613
    },
    {
      "epoch": 3.667068481445312e-05,
      "step": 9613,
      "training_step_time": 9.921602010726929
    },
    {
      "epoch": 3.667068481445312e-05,
      "model_forward_time": 3.14168381690979,
      "step": 9613
    },
    {
      "epoch": 3.667068481445312e-05,
      "step": 9613,
      "training_step_time": 10.315789937973022
    },
    {
      "epoch": 3.667449951171875e-05,
      "model_forward_time": 3.410931348800659,
      "step": 9614
    },
    {
      "epoch": 3.667449951171875e-05,
      "step": 9614,
      "training_step_time": 10.184104919433594
    },
    {
      "epoch": 3.667449951171875e-05,
      "model_forward_time": 3.134150505065918,
      "step": 9614
    },
    {
      "epoch": 3.667449951171875e-05,
      "step": 9614,
      "training_step_time": 9.906721353530884
    },
    {
      "epoch": 3.667449951171875e-05,
      "model_forward_time": 3.1391162872314453,
      "step": 9614
    },
    {
      "epoch": 3.667449951171875e-05,
      "step": 9614,
      "training_step_time": 9.905852317810059
    },
    {
      "epoch": 3.667449951171875e-05,
      "model_forward_time": 3.140507459640503,
      "step": 9614
    },
    {
      "epoch": 3.667449951171875e-05,
      "step": 9614,
      "training_step_time": 10.264547348022461
    },
    {
      "epoch": 3.6678314208984376e-05,
      "model_forward_time": 3.1530370712280273,
      "step": 9615
    },
    {
      "dynamics_loss_avg": 0.053150747902691364,
      "epoch": 3.6678314208984376e-05,
      "step": 9615
    },
    {
      "action_loss_avg": 0.016009164694696664,
      "epoch": 3.6678314208984376e-05,
      "step": 9615
    },
    {
      "epoch": 3.6678314208984376e-05,
      "step": 9615,
      "training_step_time": 9.92538595199585
    },
    {
      "epoch": 3.6678314208984376e-05,
      "model_forward_time": 3.417314052581787,
      "step": 9615
    },
    {
      "epoch": 3.6678314208984376e-05,
      "step": 9615,
      "training_step_time": 10.184540271759033
    },
    {
      "epoch": 3.6678314208984376e-05,
      "model_forward_time": 3.140934705734253,
      "step": 9615
    },
    {
      "epoch": 3.6678314208984376e-05,
      "step": 9615,
      "training_step_time": 9.914925813674927
    },
    {
      "epoch": 3.6678314208984376e-05,
      "model_forward_time": 3.1432838439941406,
      "step": 9615
    },
    {
      "epoch": 3.6678314208984376e-05,
      "step": 9615,
      "training_step_time": 10.289225101470947
    },
    {
      "epoch": 3.668212890625e-05,
      "model_forward_time": 3.1554179191589355,
      "step": 9616
    },
    {
      "epoch": 3.668212890625e-05,
      "step": 9616,
      "training_step_time": 9.931291580200195
    },
    {
      "epoch": 3.668212890625e-05,
      "model_forward_time": 3.1443893909454346,
      "step": 9616
    },
    {
      "epoch": 3.668212890625e-05,
      "step": 9616,
      "training_step_time": 9.9276864528656
    },
    {
      "epoch": 3.668212890625e-05,
      "model_forward_time": 3.407169818878174,
      "step": 9616
    },
    {
      "epoch": 3.668212890625e-05,
      "step": 9616,
      "training_step_time": 10.1858069896698
    },
    {
      "epoch": 3.668212890625e-05,
      "model_forward_time": 3.1463406085968018,
      "step": 9616
    },
    {
      "epoch": 3.668212890625e-05,
      "step": 9616,
      "training_step_time": 10.032159566879272
    },
    {
      "epoch": 3.668594360351563e-05,
      "model_forward_time": 3.1541714668273926,
      "step": 9617
    },
    {
      "epoch": 3.668594360351563e-05,
      "step": 9617,
      "training_step_time": 9.932814598083496
    },
    {
      "epoch": 3.668594360351563e-05,
      "model_forward_time": 3.138122081756592,
      "step": 9617
    },
    {
      "epoch": 3.668594360351563e-05,
      "step": 9617,
      "training_step_time": 9.909770250320435
    },
    {
      "epoch": 3.668594360351563e-05,
      "model_forward_time": 3.143038511276245,
      "step": 9617
    },
    {
      "dynamics_loss_avg": 0.05561670120805502,
      "epoch": 3.668594360351563e-05,
      "step": 9617
    },
    {
      "action_loss_avg": 0.020758112892508507,
      "epoch": 3.668594360351563e-05,
      "step": 9617
    },
    {
      "epoch": 3.668594360351563e-05,
      "step": 9617,
      "training_step_time": 9.916642904281616
    },
    {
      "epoch": 3.668594360351563e-05,
      "model_forward_time": 3.4204490184783936,
      "step": 9617
    },
    {
      "epoch": 3.668594360351563e-05,
      "step": 9617,
      "training_step_time": 10.290008306503296
    },
    {
      "epoch": 3.668975830078125e-05,
      "model_forward_time": 3.1462595462799072,
      "step": 9618
    },
    {
      "epoch": 3.668975830078125e-05,
      "step": 9618,
      "training_step_time": 9.917347192764282
    },
    {
      "epoch": 3.668975830078125e-05,
      "model_forward_time": 3.1315579414367676,
      "step": 9618
    },
    {
      "epoch": 3.668975830078125e-05,
      "step": 9618,
      "training_step_time": 9.897834777832031
    },
    {
      "epoch": 3.668975830078125e-05,
      "model_forward_time": 3.1387228965759277,
      "step": 9618
    },
    {
      "epoch": 3.668975830078125e-05,
      "step": 9618,
      "training_step_time": 9.902079582214355
    },
    {
      "epoch": 3.668975830078125e-05,
      "model_forward_time": 3.136279344558716,
      "step": 9618
    },
    {
      "epoch": 3.668975830078125e-05,
      "step": 9618,
      "training_step_time": 10.328294515609741
    },
    {
      "epoch": 3.6693572998046874e-05,
      "model_forward_time": 3.4262118339538574,
      "step": 9619
    },
    {
      "epoch": 3.6693572998046874e-05,
      "step": 9619,
      "training_step_time": 10.194494247436523
    },
    {
      "epoch": 3.6693572998046874e-05,
      "model_forward_time": 3.1337504386901855,
      "step": 9619
    },
    {
      "epoch": 3.6693572998046874e-05,
      "step": 9619,
      "training_step_time": 9.900717735290527
    },
    {
      "epoch": 3.6693572998046874e-05,
      "model_forward_time": 3.1376636028289795,
      "step": 9619
    },
    {
      "epoch": 3.6693572998046874e-05,
      "step": 9619,
      "training_step_time": 9.906496524810791
    },
    {
      "epoch": 3.6693572998046874e-05,
      "model_forward_time": 3.1376852989196777,
      "step": 9619
    },
    {
      "epoch": 3.6693572998046874e-05,
      "step": 9619,
      "training_step_time": 10.010576486587524
    },
    {
      "epoch": 3.66973876953125e-05,
      "grad_norm": 0.4361012578010559,
      "learning_rate": 1e-05,
      "loss": 0.0744,
      "step": 9620
    },
    {
      "epoch": 3.66973876953125e-05,
      "model_forward_time": 3.1456501483917236,
      "step": 9620
    },
    {
      "dynamics_loss_avg": 0.05549437925219536,
      "epoch": 3.66973876953125e-05,
      "step": 9620
    },
    {
      "action_loss_avg": 0.020337127917446195,
      "epoch": 3.66973876953125e-05,
      "step": 9620
    },
    {
      "epoch": 3.66973876953125e-05,
      "step": 9620,
      "training_step_time": 9.916790962219238
    },
    {
      "epoch": 3.66973876953125e-05,
      "model_forward_time": 3.3921592235565186,
      "step": 9620
    },
    {
      "epoch": 3.66973876953125e-05,
      "step": 9620,
      "training_step_time": 10.156485319137573
    },
    {
      "epoch": 3.66973876953125e-05,
      "model_forward_time": 3.135829448699951,
      "step": 9620
    },
    {
      "epoch": 3.66973876953125e-05,
      "step": 9620,
      "training_step_time": 9.903998851776123
    },
    {
      "epoch": 3.66973876953125e-05,
      "model_forward_time": 3.136836051940918,
      "step": 9620
    },
    {
      "epoch": 3.66973876953125e-05,
      "step": 9620,
      "training_step_time": 10.294493198394775
    },
    {
      "epoch": 3.670120239257813e-05,
      "model_forward_time": 3.141263008117676,
      "step": 9621
    },
    {
      "epoch": 3.670120239257813e-05,
      "step": 9621,
      "training_step_time": 9.911608934402466
    },
    {
      "epoch": 3.670120239257813e-05,
      "model_forward_time": 3.149399995803833,
      "step": 9621
    },
    {
      "epoch": 3.670120239257813e-05,
      "step": 9621,
      "training_step_time": 9.92436957359314
    },
    {
      "epoch": 3.670120239257813e-05,
      "model_forward_time": 3.4610488414764404,
      "step": 9621
    },
    {
      "epoch": 3.670120239257813e-05,
      "step": 9621,
      "training_step_time": 10.234217405319214
    },
    {
      "epoch": 3.670120239257813e-05,
      "model_forward_time": 3.147956609725952,
      "step": 9621
    },
    {
      "epoch": 3.670120239257813e-05,
      "step": 9621,
      "training_step_time": 10.032243728637695
    },
    {
      "epoch": 3.670501708984375e-05,
      "model_forward_time": 3.1566479206085205,
      "step": 9622
    },
    {
      "epoch": 3.670501708984375e-05,
      "step": 9622,
      "training_step_time": 9.935704231262207
    },
    {
      "epoch": 3.670501708984375e-05,
      "model_forward_time": 3.1389079093933105,
      "step": 9622
    },
    {
      "epoch": 3.670501708984375e-05,
      "step": 9622,
      "training_step_time": 9.916154861450195
    },
    {
      "epoch": 3.670501708984375e-05,
      "model_forward_time": 3.140672206878662,
      "step": 9622
    },
    {
      "dynamics_loss_avg": 0.07011717967689038,
      "epoch": 3.670501708984375e-05,
      "step": 9622
    },
    {
      "action_loss_avg": 0.023732725903391837,
      "epoch": 3.670501708984375e-05,
      "step": 9622
    },
    {
      "epoch": 3.670501708984375e-05,
      "step": 9622,
      "training_step_time": 9.917009353637695
    },
    {
      "epoch": 3.670501708984375e-05,
      "model_forward_time": 3.395881175994873,
      "step": 9622
    },
    {
      "epoch": 3.670501708984375e-05,
      "step": 9622,
      "training_step_time": 10.310092449188232
    },
    {
      "epoch": 3.670883178710937e-05,
      "model_forward_time": 3.1460464000701904,
      "step": 9623
    },
    {
      "epoch": 3.670883178710937e-05,
      "step": 9623,
      "training_step_time": 9.921541929244995
    },
    {
      "epoch": 3.670883178710937e-05,
      "model_forward_time": 3.1409881114959717,
      "step": 9623
    },
    {
      "epoch": 3.670883178710937e-05,
      "step": 9623,
      "training_step_time": 9.916164875030518
    },
    {
      "epoch": 3.670883178710937e-05,
      "model_forward_time": 3.137559413909912,
      "step": 9623
    },
    {
      "epoch": 3.670883178710937e-05,
      "step": 9623,
      "training_step_time": 9.908037424087524
    },
    {
      "epoch": 3.670883178710937e-05,
      "model_forward_time": 2.4355790615081787,
      "step": 9623
    },
    {
      "epoch": 3.670883178710937e-05,
      "step": 9623,
      "training_step_time": 10.31225299835205
    },
    {
      "epoch": 3.6712646484375e-05,
      "model_forward_time": 3.4140403270721436,
      "step": 9624
    },
    {
      "epoch": 3.6712646484375e-05,
      "step": 9624,
      "training_step_time": 10.181203126907349
    },
    {
      "epoch": 3.6712646484375e-05,
      "model_forward_time": 3.138829469680786,
      "step": 9624
    },
    {
      "epoch": 3.6712646484375e-05,
      "step": 9624,
      "training_step_time": 9.906012058258057
    },
    {
      "epoch": 3.6712646484375e-05,
      "model_forward_time": 3.137711524963379,
      "step": 9624
    },
    {
      "epoch": 3.6712646484375e-05,
      "step": 9624,
      "training_step_time": 9.905410528182983
    },
    {
      "epoch": 3.6712646484375e-05,
      "model_forward_time": 3.1426541805267334,
      "step": 9624
    },
    {
      "epoch": 3.6712646484375e-05,
      "step": 9624,
      "training_step_time": 10.320308685302734
    },
    {
      "epoch": 3.6716461181640625e-05,
      "model_forward_time": 3.155651807785034,
      "step": 9625
    },
    {
      "dynamics_loss_avg": 0.05139680746942758,
      "epoch": 3.6716461181640625e-05,
      "step": 9625
    },
    {
      "action_loss_avg": 0.01902054399251938,
      "epoch": 3.6716461181640625e-05,
      "step": 9625
    },
    {
      "epoch": 3.6716461181640625e-05,
      "step": 9625,
      "training_step_time": 9.926559448242188
    },
    {
      "epoch": 3.6716461181640625e-05,
      "model_forward_time": 3.417656660079956,
      "step": 9625
    },
    {
      "epoch": 3.6716461181640625e-05,
      "step": 9625,
      "training_step_time": 10.186894178390503
    },
    {
      "epoch": 3.6716461181640625e-05,
      "model_forward_time": 3.141974449157715,
      "step": 9625
    },
    {
      "epoch": 3.6716461181640625e-05,
      "step": 9625,
      "training_step_time": 9.914592742919922
    },
    {
      "epoch": 3.6716461181640625e-05,
      "model_forward_time": 3.148700714111328,
      "step": 9625
    },
    {
      "epoch": 3.6716461181640625e-05,
      "step": 9625,
      "training_step_time": 10.32004976272583
    },
    {
      "epoch": 3.672027587890625e-05,
      "model_forward_time": 3.1590070724487305,
      "step": 9626
    },
    {
      "epoch": 3.672027587890625e-05,
      "step": 9626,
      "training_step_time": 9.94580864906311
    },
    {
      "epoch": 3.672027587890625e-05,
      "model_forward_time": 3.1355936527252197,
      "step": 9626
    },
    {
      "epoch": 3.672027587890625e-05,
      "step": 9626,
      "training_step_time": 9.908376932144165
    },
    {
      "epoch": 3.672027587890625e-05,
      "model_forward_time": 2.706955909729004,
      "step": 9626
    },
    {
      "epoch": 3.672027587890625e-05,
      "step": 9626,
      "training_step_time": 7.731149673461914
    },
    {
      "epoch": 3.672027587890625e-05,
      "model_forward_time": 3.133107900619507,
      "step": 9626
    },
    {
      "epoch": 3.672027587890625e-05,
      "step": 9626,
      "training_step_time": 10.005367755889893
    },
    {
      "epoch": 3.672409057617188e-05,
      "model_forward_time": 3.1511895656585693,
      "step": 9627
    },
    {
      "epoch": 3.672409057617188e-05,
      "step": 9627,
      "training_step_time": 9.92029595375061
    },
    {
      "epoch": 3.672409057617188e-05,
      "model_forward_time": 3.1352410316467285,
      "step": 9627
    },
    {
      "epoch": 3.672409057617188e-05,
      "step": 9627,
      "training_step_time": 9.907389163970947
    },
    {
      "epoch": 3.672409057617188e-05,
      "model_forward_time": 3.1393229961395264,
      "step": 9627
    },
    {
      "dynamics_loss_avg": 0.05278539732098579,
      "epoch": 3.672409057617188e-05,
      "step": 9627
    },
    {
      "action_loss_avg": 0.018110769242048262,
      "epoch": 3.672409057617188e-05,
      "step": 9627
    },
    {
      "epoch": 3.672409057617188e-05,
      "step": 9627,
      "training_step_time": 9.9113028049469
    },
    {
      "epoch": 3.672409057617188e-05,
      "model_forward_time": 3.412247896194458,
      "step": 9627
    },
    {
      "epoch": 3.672409057617188e-05,
      "step": 9627,
      "training_step_time": 10.299883127212524
    },
    {
      "epoch": 3.67279052734375e-05,
      "model_forward_time": 3.1479604244232178,
      "step": 9628
    },
    {
      "epoch": 3.67279052734375e-05,
      "step": 9628,
      "training_step_time": 9.918625593185425
    },
    {
      "epoch": 3.67279052734375e-05,
      "model_forward_time": 3.1411774158477783,
      "step": 9628
    },
    {
      "epoch": 3.67279052734375e-05,
      "step": 9628,
      "training_step_time": 9.907622337341309
    },
    {
      "epoch": 3.67279052734375e-05,
      "model_forward_time": 3.145665168762207,
      "step": 9628
    },
    {
      "epoch": 3.67279052734375e-05,
      "step": 9628,
      "training_step_time": 9.912722826004028
    },
    {
      "epoch": 3.67279052734375e-05,
      "model_forward_time": 3.1416056156158447,
      "step": 9628
    },
    {
      "epoch": 3.67279052734375e-05,
      "step": 9628,
      "training_step_time": 10.258385419845581
    },
    {
      "epoch": 3.673171997070312e-05,
      "model_forward_time": 3.4378304481506348,
      "step": 9629
    },
    {
      "epoch": 3.673171997070312e-05,
      "step": 9629,
      "training_step_time": 10.206858396530151
    },
    {
      "epoch": 3.673171997070312e-05,
      "model_forward_time": 3.1436893939971924,
      "step": 9629
    },
    {
      "epoch": 3.673171997070312e-05,
      "step": 9629,
      "training_step_time": 9.915435314178467
    },
    {
      "epoch": 3.673171997070312e-05,
      "model_forward_time": 3.149108648300171,
      "step": 9629
    },
    {
      "epoch": 3.673171997070312e-05,
      "step": 9629,
      "training_step_time": 9.92194938659668
    },
    {
      "epoch": 3.673171997070312e-05,
      "model_forward_time": 3.142514944076538,
      "step": 9629
    },
    {
      "epoch": 3.673171997070312e-05,
      "step": 9629,
      "training_step_time": 10.310192108154297
    },
    {
      "epoch": 3.673553466796875e-05,
      "grad_norm": 0.5417753458023071,
      "learning_rate": 1e-05,
      "loss": 0.0785,
      "step": 9630
    },
    {
      "epoch": 3.673553466796875e-05,
      "model_forward_time": 3.167813301086426,
      "step": 9630
    },
    {
      "dynamics_loss_avg": 0.047580557875335217,
      "epoch": 3.673553466796875e-05,
      "step": 9630
    },
    {
      "action_loss_avg": 0.0150915662990883,
      "epoch": 3.673553466796875e-05,
      "step": 9630
    },
    {
      "epoch": 3.673553466796875e-05,
      "step": 9630,
      "training_step_time": 9.949844121932983
    },
    {
      "epoch": 3.673553466796875e-05,
      "model_forward_time": 3.4151980876922607,
      "step": 9630
    },
    {
      "epoch": 3.673553466796875e-05,
      "step": 9630,
      "training_step_time": 10.187551259994507
    },
    {
      "epoch": 3.673553466796875e-05,
      "model_forward_time": 3.1527750492095947,
      "step": 9630
    },
    {
      "epoch": 3.673553466796875e-05,
      "step": 9630,
      "training_step_time": 9.926485776901245
    },
    {
      "epoch": 3.673553466796875e-05,
      "model_forward_time": 3.145143508911133,
      "step": 9630
    },
    {
      "epoch": 3.673553466796875e-05,
      "step": 9630,
      "training_step_time": 10.301926612854004
    },
    {
      "epoch": 3.6739349365234376e-05,
      "model_forward_time": 3.1630516052246094,
      "step": 9631
    },
    {
      "epoch": 3.6739349365234376e-05,
      "step": 9631,
      "training_step_time": 9.933668375015259
    },
    {
      "epoch": 3.6739349365234376e-05,
      "model_forward_time": 2.440075635910034,
      "step": 9631
    },
    {
      "epoch": 3.6739349365234376e-05,
      "step": 9631,
      "training_step_time": 7.469530820846558
    },
    {
      "epoch": 3.6739349365234376e-05,
      "model_forward_time": 3.4279510974884033,
      "step": 9631
    },
    {
      "epoch": 3.6739349365234376e-05,
      "step": 9631,
      "training_step_time": 10.200692892074585
    },
    {
      "epoch": 3.6739349365234376e-05,
      "model_forward_time": 3.153062105178833,
      "step": 9631
    },
    {
      "epoch": 3.6739349365234376e-05,
      "step": 9631,
      "training_step_time": 10.327175617218018
    },
    {
      "epoch": 3.67431640625e-05,
      "model_forward_time": 3.1550092697143555,
      "step": 9632
    },
    {
      "epoch": 3.67431640625e-05,
      "step": 9632,
      "training_step_time": 9.930768728256226
    },
    {
      "epoch": 3.67431640625e-05,
      "model_forward_time": 3.153536081314087,
      "step": 9632
    },
    {
      "epoch": 3.67431640625e-05,
      "step": 9632,
      "training_step_time": 9.9286789894104
    },
    {
      "epoch": 3.67431640625e-05,
      "model_forward_time": 3.1464736461639404,
      "step": 9632
    },
    {
      "dynamics_loss_avg": 0.0733243977651,
      "epoch": 3.67431640625e-05,
      "step": 9632
    },
    {
      "action_loss_avg": 0.02458222219720483,
      "epoch": 3.67431640625e-05,
      "step": 9632
    },
    {
      "epoch": 3.67431640625e-05,
      "step": 9632,
      "training_step_time": 9.927034139633179
    },
    {
      "epoch": 3.67431640625e-05,
      "model_forward_time": 3.4142439365386963,
      "step": 9632
    },
    {
      "epoch": 3.67431640625e-05,
      "step": 9632,
      "training_step_time": 10.309087991714478
    },
    {
      "epoch": 3.674697875976562e-05,
      "model_forward_time": 3.157046318054199,
      "step": 9633
    },
    {
      "epoch": 3.674697875976562e-05,
      "step": 9633,
      "training_step_time": 9.938487529754639
    },
    {
      "epoch": 3.674697875976562e-05,
      "model_forward_time": 3.150524377822876,
      "step": 9633
    },
    {
      "epoch": 3.674697875976562e-05,
      "step": 9633,
      "training_step_time": 9.928179264068604
    },
    {
      "epoch": 3.674697875976562e-05,
      "model_forward_time": 3.148547649383545,
      "step": 9633
    },
    {
      "epoch": 3.674697875976562e-05,
      "step": 9633,
      "training_step_time": 9.927613735198975
    },
    {
      "epoch": 3.674697875976562e-05,
      "model_forward_time": 3.1530778408050537,
      "step": 9633
    },
    {
      "epoch": 3.674697875976562e-05,
      "step": 9633,
      "training_step_time": 10.30951714515686
    },
    {
      "epoch": 3.675079345703125e-05,
      "model_forward_time": 3.426717758178711,
      "step": 9634
    },
    {
      "epoch": 3.675079345703125e-05,
      "step": 9634,
      "training_step_time": 10.205010175704956
    },
    {
      "epoch": 3.675079345703125e-05,
      "model_forward_time": 3.151252269744873,
      "step": 9634
    },
    {
      "epoch": 3.675079345703125e-05,
      "step": 9634,
      "training_step_time": 9.928599834442139
    },
    {
      "epoch": 3.675079345703125e-05,
      "model_forward_time": 3.148529052734375,
      "step": 9634
    },
    {
      "epoch": 3.675079345703125e-05,
      "step": 9634,
      "training_step_time": 9.921867370605469
    },
    {
      "epoch": 3.675079345703125e-05,
      "model_forward_time": 3.153257369995117,
      "step": 9634
    },
    {
      "epoch": 3.675079345703125e-05,
      "step": 9634,
      "training_step_time": 10.303043127059937
    },
    {
      "epoch": 3.6754608154296874e-05,
      "model_forward_time": 3.1565358638763428,
      "step": 9635
    },
    {
      "dynamics_loss_avg": 0.05452475938946009,
      "epoch": 3.6754608154296874e-05,
      "step": 9635
    },
    {
      "action_loss_avg": 0.015268555190414191,
      "epoch": 3.6754608154296874e-05,
      "step": 9635
    },
    {
      "epoch": 3.6754608154296874e-05,
      "step": 9635,
      "training_step_time": 9.933276414871216
    },
    {
      "epoch": 3.6754608154296874e-05,
      "model_forward_time": 3.431264877319336,
      "step": 9635
    },
    {
      "epoch": 3.6754608154296874e-05,
      "step": 9635,
      "training_step_time": 10.202174425125122
    },
    {
      "epoch": 3.6754608154296874e-05,
      "model_forward_time": 3.144951105117798,
      "step": 9635
    },
    {
      "epoch": 3.6754608154296874e-05,
      "step": 9635,
      "training_step_time": 9.914525032043457
    },
    {
      "epoch": 3.6754608154296874e-05,
      "model_forward_time": 3.1442108154296875,
      "step": 9635
    },
    {
      "epoch": 3.6754608154296874e-05,
      "step": 9635,
      "training_step_time": 10.311926126480103
    },
    {
      "epoch": 3.67584228515625e-05,
      "model_forward_time": 2.4532971382141113,
      "step": 9636
    },
    {
      "epoch": 3.67584228515625e-05,
      "step": 9636,
      "training_step_time": 7.477786064147949
    },
    {
      "epoch": 3.67584228515625e-05,
      "model_forward_time": 3.13547945022583,
      "step": 9636
    },
    {
      "epoch": 3.67584228515625e-05,
      "step": 9636,
      "training_step_time": 9.899329900741577
    },
    {
      "epoch": 3.67584228515625e-05,
      "model_forward_time": 3.1418721675872803,
      "step": 9636
    },
    {
      "epoch": 3.67584228515625e-05,
      "step": 9636,
      "training_step_time": 10.208741188049316
    },
    {
      "epoch": 3.67584228515625e-05,
      "model_forward_time": 3.147900104522705,
      "step": 9636
    },
    {
      "epoch": 3.67584228515625e-05,
      "step": 9636,
      "training_step_time": 10.267872333526611
    },
    {
      "epoch": 3.676223754882813e-05,
      "model_forward_time": 2.4495725631713867,
      "step": 9637
    },
    {
      "epoch": 3.676223754882813e-05,
      "step": 9637,
      "training_step_time": 7.474064350128174
    },
    {
      "epoch": 3.676223754882813e-05,
      "model_forward_time": 3.1389219760894775,
      "step": 9637
    },
    {
      "epoch": 3.676223754882813e-05,
      "step": 9637,
      "training_step_time": 9.90065050125122
    },
    {
      "epoch": 3.676223754882813e-05,
      "model_forward_time": 3.1403250694274902,
      "step": 9637
    },
    {
      "dynamics_loss_avg": 0.056076855026185514,
      "epoch": 3.676223754882813e-05,
      "step": 9637
    },
    {
      "action_loss_avg": 0.017082814848981796,
      "epoch": 3.676223754882813e-05,
      "step": 9637
    },
    {
      "epoch": 3.676223754882813e-05,
      "step": 9637,
      "training_step_time": 9.908172607421875
    },
    {
      "epoch": 3.676223754882813e-05,
      "model_forward_time": 3.436145782470703,
      "step": 9637
    },
    {
      "epoch": 3.676223754882813e-05,
      "step": 9637,
      "training_step_time": 10.295993328094482
    },
    {
      "epoch": 3.676605224609375e-05,
      "model_forward_time": 3.1622610092163086,
      "step": 9638
    },
    {
      "epoch": 3.676605224609375e-05,
      "step": 9638,
      "training_step_time": 9.934997081756592
    },
    {
      "epoch": 3.676605224609375e-05,
      "model_forward_time": 3.1456258296966553,
      "step": 9638
    },
    {
      "epoch": 3.676605224609375e-05,
      "step": 9638,
      "training_step_time": 9.914997816085815
    },
    {
      "epoch": 3.676605224609375e-05,
      "model_forward_time": 3.150695562362671,
      "step": 9638
    },
    {
      "epoch": 3.676605224609375e-05,
      "step": 9638,
      "training_step_time": 9.921305656433105
    },
    {
      "epoch": 3.676605224609375e-05,
      "model_forward_time": 3.1459474563598633,
      "step": 9638
    },
    {
      "epoch": 3.676605224609375e-05,
      "step": 9638,
      "training_step_time": 10.323013067245483
    },
    {
      "epoch": 3.676986694335937e-05,
      "model_forward_time": 3.154438018798828,
      "step": 9639
    },
    {
      "epoch": 3.676986694335937e-05,
      "step": 9639,
      "training_step_time": 10.185807466506958
    },
    {
      "epoch": 3.676986694335937e-05,
      "model_forward_time": 3.1394758224487305,
      "step": 9639
    },
    {
      "epoch": 3.676986694335937e-05,
      "step": 9639,
      "training_step_time": 9.906282186508179
    },
    {
      "epoch": 3.676986694335937e-05,
      "model_forward_time": 3.144702911376953,
      "step": 9639
    },
    {
      "epoch": 3.676986694335937e-05,
      "step": 9639,
      "training_step_time": 9.914178609848022
    },
    {
      "epoch": 3.676986694335937e-05,
      "model_forward_time": 3.1383845806121826,
      "step": 9639
    },
    {
      "epoch": 3.676986694335937e-05,
      "step": 9639,
      "training_step_time": 10.264184713363647
    },
    {
      "epoch": 3.6773681640625e-05,
      "grad_norm": 0.4734163284301758,
      "learning_rate": 1e-05,
      "loss": 0.078,
      "step": 9640
    },
    {
      "epoch": 3.6773681640625e-05,
      "model_forward_time": 3.158762216567993,
      "step": 9640
    },
    {
      "dynamics_loss_avg": 0.05868047010153532,
      "epoch": 3.6773681640625e-05,
      "step": 9640
    },
    {
      "action_loss_avg": 0.020954543631523846,
      "epoch": 3.6773681640625e-05,
      "step": 9640
    },
    {
      "epoch": 3.6773681640625e-05,
      "step": 9640,
      "training_step_time": 9.932428121566772
    },
    {
      "epoch": 3.6773681640625e-05,
      "model_forward_time": 3.4090001583099365,
      "step": 9640
    },
    {
      "epoch": 3.6773681640625e-05,
      "step": 9640,
      "training_step_time": 10.174458742141724
    },
    {
      "epoch": 3.6773681640625e-05,
      "model_forward_time": 3.14263653755188,
      "step": 9640
    },
    {
      "epoch": 3.6773681640625e-05,
      "step": 9640,
      "training_step_time": 9.906446695327759
    },
    {
      "epoch": 3.6773681640625e-05,
      "model_forward_time": 3.1395914554595947,
      "step": 9640
    },
    {
      "epoch": 3.6773681640625e-05,
      "step": 9640,
      "training_step_time": 10.28465461730957
    },
    {
      "epoch": 3.6777496337890625e-05,
      "model_forward_time": 3.1469624042510986,
      "step": 9641
    },
    {
      "epoch": 3.6777496337890625e-05,
      "step": 9641,
      "training_step_time": 9.913261890411377
    },
    {
      "epoch": 3.6777496337890625e-05,
      "model_forward_time": 3.1378118991851807,
      "step": 9641
    },
    {
      "epoch": 3.6777496337890625e-05,
      "step": 9641,
      "training_step_time": 9.901633501052856
    },
    {
      "epoch": 3.6777496337890625e-05,
      "model_forward_time": 3.425464630126953,
      "step": 9641
    },
    {
      "epoch": 3.6777496337890625e-05,
      "step": 9641,
      "training_step_time": 10.190514087677002
    },
    {
      "epoch": 3.6777496337890625e-05,
      "model_forward_time": 3.1381723880767822,
      "step": 9641
    },
    {
      "epoch": 3.6777496337890625e-05,
      "step": 9641,
      "training_step_time": 10.012390375137329
    },
    {
      "epoch": 3.678131103515625e-05,
      "model_forward_time": 3.1586802005767822,
      "step": 9642
    },
    {
      "epoch": 3.678131103515625e-05,
      "step": 9642,
      "training_step_time": 9.926406145095825
    },
    {
      "epoch": 3.678131103515625e-05,
      "model_forward_time": 3.1427907943725586,
      "step": 9642
    },
    {
      "epoch": 3.678131103515625e-05,
      "step": 9642,
      "training_step_time": 9.908670663833618
    },
    {
      "epoch": 3.678131103515625e-05,
      "model_forward_time": 3.1381564140319824,
      "step": 9642
    },
    {
      "dynamics_loss_avg": 0.05439172233454883,
      "epoch": 3.678131103515625e-05,
      "step": 9642
    },
    {
      "action_loss_avg": 0.01567859265487641,
      "epoch": 3.678131103515625e-05,
      "step": 9642
    },
    {
      "epoch": 3.678131103515625e-05,
      "step": 9642,
      "training_step_time": 9.910234689712524
    },
    {
      "epoch": 3.678131103515625e-05,
      "model_forward_time": 3.1450467109680176,
      "step": 9642
    },
    {
      "epoch": 3.678131103515625e-05,
      "step": 9642,
      "training_step_time": 10.289875268936157
    },
    {
      "epoch": 3.678512573242188e-05,
      "model_forward_time": 3.1577670574188232,
      "step": 9643
    },
    {
      "epoch": 3.678512573242188e-05,
      "step": 9643,
      "training_step_time": 9.931268692016602
    },
    {
      "epoch": 3.678512573242188e-05,
      "model_forward_time": 3.173727035522461,
      "step": 9643
    },
    {
      "epoch": 3.678512573242188e-05,
      "step": 9643,
      "training_step_time": 9.948933601379395
    },
    {
      "epoch": 3.678512573242188e-05,
      "model_forward_time": 3.1439168453216553,
      "step": 9643
    },
    {
      "epoch": 3.678512573242188e-05,
      "step": 9643,
      "training_step_time": 9.921268701553345
    },
    {
      "epoch": 3.678512573242188e-05,
      "model_forward_time": 3.1490049362182617,
      "step": 9643
    },
    {
      "epoch": 3.678512573242188e-05,
      "step": 9643,
      "training_step_time": 10.314523696899414
    },
    {
      "epoch": 3.67889404296875e-05,
      "model_forward_time": 3.420555591583252,
      "step": 9644
    },
    {
      "epoch": 3.67889404296875e-05,
      "step": 9644,
      "training_step_time": 10.191436529159546
    },
    {
      "epoch": 3.67889404296875e-05,
      "model_forward_time": 3.1460418701171875,
      "step": 9644
    },
    {
      "epoch": 3.67889404296875e-05,
      "step": 9644,
      "training_step_time": 9.920008897781372
    },
    {
      "epoch": 3.67889404296875e-05,
      "model_forward_time": 3.1442954540252686,
      "step": 9644
    },
    {
      "epoch": 3.67889404296875e-05,
      "step": 9644,
      "training_step_time": 9.917879343032837
    },
    {
      "epoch": 3.67889404296875e-05,
      "model_forward_time": 3.1466457843780518,
      "step": 9644
    },
    {
      "epoch": 3.67889404296875e-05,
      "step": 9644,
      "training_step_time": 10.312438249588013
    },
    {
      "epoch": 3.679275512695312e-05,
      "model_forward_time": 3.15556263923645,
      "step": 9645
    },
    {
      "dynamics_loss_avg": 0.06500148586928844,
      "epoch": 3.679275512695312e-05,
      "step": 9645
    },
    {
      "action_loss_avg": 0.024621064588427544,
      "epoch": 3.679275512695312e-05,
      "step": 9645
    },
    {
      "epoch": 3.679275512695312e-05,
      "step": 9645,
      "training_step_time": 9.928744792938232
    },
    {
      "epoch": 3.679275512695312e-05,
      "model_forward_time": 3.1459884643554688,
      "step": 9645
    },
    {
      "epoch": 3.679275512695312e-05,
      "step": 9645,
      "training_step_time": 10.188156366348267
    },
    {
      "epoch": 3.679275512695312e-05,
      "model_forward_time": 3.142141819000244,
      "step": 9645
    },
    {
      "epoch": 3.679275512695312e-05,
      "step": 9645,
      "training_step_time": 9.912663698196411
    },
    {
      "epoch": 3.679275512695312e-05,
      "model_forward_time": 3.1476619243621826,
      "step": 9645
    },
    {
      "epoch": 3.679275512695312e-05,
      "step": 9645,
      "training_step_time": 10.305170059204102
    },
    {
      "epoch": 3.679656982421875e-05,
      "model_forward_time": 3.1652109622955322,
      "step": 9646
    },
    {
      "epoch": 3.679656982421875e-05,
      "step": 9646,
      "training_step_time": 9.936553955078125
    },
    {
      "epoch": 3.679656982421875e-05,
      "model_forward_time": 3.1414647102355957,
      "step": 9646
    },
    {
      "epoch": 3.679656982421875e-05,
      "step": 9646,
      "training_step_time": 9.911845445632935
    },
    {
      "epoch": 3.679656982421875e-05,
      "model_forward_time": 3.4066412448883057,
      "step": 9646
    },
    {
      "epoch": 3.679656982421875e-05,
      "step": 9646,
      "training_step_time": 10.179793357849121
    },
    {
      "epoch": 3.679656982421875e-05,
      "model_forward_time": 3.1462717056274414,
      "step": 9646
    },
    {
      "epoch": 3.679656982421875e-05,
      "step": 9646,
      "training_step_time": 10.018905401229858
    },
    {
      "epoch": 3.6800384521484376e-05,
      "model_forward_time": 3.1504147052764893,
      "step": 9647
    },
    {
      "epoch": 3.6800384521484376e-05,
      "step": 9647,
      "training_step_time": 9.930215835571289
    },
    {
      "epoch": 3.6800384521484376e-05,
      "model_forward_time": 3.141711711883545,
      "step": 9647
    },
    {
      "epoch": 3.6800384521484376e-05,
      "step": 9647,
      "training_step_time": 9.909531354904175
    },
    {
      "epoch": 3.6800384521484376e-05,
      "model_forward_time": 3.1438050270080566,
      "step": 9647
    },
    {
      "dynamics_loss_avg": 0.08209716826677323,
      "epoch": 3.6800384521484376e-05,
      "step": 9647
    },
    {
      "action_loss_avg": 0.02797974515706301,
      "epoch": 3.6800384521484376e-05,
      "step": 9647
    },
    {
      "epoch": 3.6800384521484376e-05,
      "step": 9647,
      "training_step_time": 9.92017912864685
    },
    {
      "epoch": 3.6800384521484376e-05,
      "model_forward_time": 3.1409971714019775,
      "step": 9647
    },
    {
      "epoch": 3.6800384521484376e-05,
      "step": 9647,
      "training_step_time": 10.26848292350769
    },
    {
      "epoch": 3.680419921875e-05,
      "model_forward_time": 3.165531635284424,
      "step": 9648
    },
    {
      "epoch": 3.680419921875e-05,
      "step": 9648,
      "training_step_time": 9.939612627029419
    },
    {
      "epoch": 3.680419921875e-05,
      "model_forward_time": 3.1408541202545166,
      "step": 9648
    },
    {
      "epoch": 3.680419921875e-05,
      "step": 9648,
      "training_step_time": 9.91454792022705
    },
    {
      "epoch": 3.680419921875e-05,
      "model_forward_time": 3.1450016498565674,
      "step": 9648
    },
    {
      "epoch": 3.680419921875e-05,
      "step": 9648,
      "training_step_time": 9.921361684799194
    },
    {
      "epoch": 3.680419921875e-05,
      "model_forward_time": 2.439007043838501,
      "step": 9648
    },
    {
      "epoch": 3.680419921875e-05,
      "step": 9648,
      "training_step_time": 10.310829639434814
    },
    {
      "epoch": 3.680801391601562e-05,
      "model_forward_time": 3.4628000259399414,
      "step": 9649
    },
    {
      "epoch": 3.680801391601562e-05,
      "step": 9649,
      "training_step_time": 10.23216700553894
    },
    {
      "epoch": 3.680801391601562e-05,
      "model_forward_time": 3.142219066619873,
      "step": 9649
    },
    {
      "epoch": 3.680801391601562e-05,
      "step": 9649,
      "training_step_time": 9.91294527053833
    },
    {
      "epoch": 3.680801391601562e-05,
      "model_forward_time": 3.1467254161834717,
      "step": 9649
    },
    {
      "epoch": 3.680801391601562e-05,
      "step": 9649,
      "training_step_time": 9.919792413711548
    },
    {
      "epoch": 3.680801391601562e-05,
      "model_forward_time": 3.1403844356536865,
      "step": 9649
    },
    {
      "epoch": 3.680801391601562e-05,
      "step": 9649,
      "training_step_time": 10.310133218765259
    },
    {
      "epoch": 3.681182861328125e-05,
      "grad_norm": 0.6655360460281372,
      "learning_rate": 1e-05,
      "loss": 0.083,
      "step": 9650
    },
    {
      "epoch": 3.681182861328125e-05,
      "model_forward_time": 3.1599934101104736,
      "step": 9650
    },
    {
      "dynamics_loss_avg": 0.06246403455734253,
      "epoch": 3.681182861328125e-05,
      "step": 9650
    },
    {
      "action_loss_avg": 0.022737820725888015,
      "epoch": 3.681182861328125e-05,
      "step": 9650
    },
    {
      "epoch": 3.681182861328125e-05,
      "step": 9650,
      "training_step_time": 9.941845178604126
    },
    {
      "epoch": 3.681182861328125e-05,
      "model_forward_time": 3.4214670658111572,
      "step": 9650
    },
    {
      "epoch": 3.681182861328125e-05,
      "step": 9650,
      "training_step_time": 10.192063093185425
    },
    {
      "epoch": 3.681182861328125e-05,
      "model_forward_time": 3.1455740928649902,
      "step": 9650
    },
    {
      "epoch": 3.681182861328125e-05,
      "step": 9650,
      "training_step_time": 9.91688323020935
    },
    {
      "epoch": 3.681182861328125e-05,
      "model_forward_time": 3.1450302600860596,
      "step": 9650
    },
    {
      "epoch": 3.681182861328125e-05,
      "step": 9650,
      "training_step_time": 10.032817125320435
    },
    {
      "epoch": 3.6815643310546874e-05,
      "model_forward_time": 3.1526598930358887,
      "step": 9651
    },
    {
      "epoch": 3.6815643310546874e-05,
      "step": 9651,
      "training_step_time": 9.924912452697754
    },
    {
      "epoch": 3.6815643310546874e-05,
      "model_forward_time": 3.1424942016601562,
      "step": 9651
    },
    {
      "epoch": 3.6815643310546874e-05,
      "step": 9651,
      "training_step_time": 9.914132356643677
    },
    {
      "epoch": 3.6815643310546874e-05,
      "model_forward_time": 3.143552303314209,
      "step": 9651
    },
    {
      "epoch": 3.6815643310546874e-05,
      "step": 9651,
      "training_step_time": 10.191696405410767
    },
    {
      "epoch": 3.6815643310546874e-05,
      "model_forward_time": 3.1417510509490967,
      "step": 9651
    },
    {
      "epoch": 3.6815643310546874e-05,
      "step": 9651,
      "training_step_time": 10.261933088302612
    },
    {
      "epoch": 3.68194580078125e-05,
      "model_forward_time": 3.150221824645996,
      "step": 9652
    },
    {
      "epoch": 3.68194580078125e-05,
      "step": 9652,
      "training_step_time": 9.917978525161743
    },
    {
      "epoch": 3.68194580078125e-05,
      "model_forward_time": 3.14166259765625,
      "step": 9652
    },
    {
      "epoch": 3.68194580078125e-05,
      "step": 9652,
      "training_step_time": 9.907186508178711
    },
    {
      "epoch": 3.68194580078125e-05,
      "model_forward_time": 3.138084650039673,
      "step": 9652
    },
    {
      "dynamics_loss_avg": 0.07093339823186398,
      "epoch": 3.68194580078125e-05,
      "step": 9652
    },
    {
      "action_loss_avg": 0.028242228738963603,
      "epoch": 3.68194580078125e-05,
      "step": 9652
    },
    {
      "epoch": 3.68194580078125e-05,
      "step": 9652,
      "training_step_time": 9.904406547546387
    },
    {
      "epoch": 3.68194580078125e-05,
      "model_forward_time": 3.1422555446624756,
      "step": 9652
    },
    {
      "epoch": 3.68194580078125e-05,
      "step": 9652,
      "training_step_time": 10.273266553878784
    },
    {
      "epoch": 3.682327270507813e-05,
      "model_forward_time": 3.158010959625244,
      "step": 9653
    },
    {
      "epoch": 3.682327270507813e-05,
      "step": 9653,
      "training_step_time": 9.926834106445312
    },
    {
      "epoch": 3.682327270507813e-05,
      "model_forward_time": 3.1408867835998535,
      "step": 9653
    },
    {
      "epoch": 3.682327270507813e-05,
      "step": 9653,
      "training_step_time": 9.909783363342285
    },
    {
      "epoch": 3.682327270507813e-05,
      "model_forward_time": 3.1398558616638184,
      "step": 9653
    },
    {
      "epoch": 3.682327270507813e-05,
      "step": 9653,
      "training_step_time": 9.912106275558472
    },
    {
      "epoch": 3.682327270507813e-05,
      "model_forward_time": 3.1455254554748535,
      "step": 9653
    },
    {
      "epoch": 3.682327270507813e-05,
      "step": 9653,
      "training_step_time": 10.299283981323242
    },
    {
      "epoch": 3.682708740234375e-05,
      "model_forward_time": 3.1569712162017822,
      "step": 9654
    },
    {
      "epoch": 3.682708740234375e-05,
      "step": 9654,
      "training_step_time": 10.212438583374023
    },
    {
      "epoch": 3.682708740234375e-05,
      "model_forward_time": 3.1468286514282227,
      "step": 9654
    },
    {
      "epoch": 3.682708740234375e-05,
      "step": 9654,
      "training_step_time": 9.919045209884644
    },
    {
      "epoch": 3.682708740234375e-05,
      "model_forward_time": 3.141045570373535,
      "step": 9654
    },
    {
      "epoch": 3.682708740234375e-05,
      "step": 9654,
      "training_step_time": 9.913422346115112
    },
    {
      "epoch": 3.682708740234375e-05,
      "model_forward_time": 3.146951913833618,
      "step": 9654
    },
    {
      "epoch": 3.682708740234375e-05,
      "step": 9654,
      "training_step_time": 10.30752182006836
    },
    {
      "epoch": 3.683090209960937e-05,
      "model_forward_time": 3.1550495624542236,
      "step": 9655
    },
    {
      "dynamics_loss_avg": 0.054046927578747274,
      "epoch": 3.683090209960937e-05,
      "step": 9655
    },
    {
      "action_loss_avg": 0.020199557347223163,
      "epoch": 3.683090209960937e-05,
      "step": 9655
    },
    {
      "epoch": 3.683090209960937e-05,
      "step": 9655,
      "training_step_time": 9.931387662887573
    },
    {
      "epoch": 3.683090209960937e-05,
      "model_forward_time": 3.4237141609191895,
      "step": 9655
    },
    {
      "epoch": 3.683090209960937e-05,
      "step": 9655,
      "training_step_time": 10.192399263381958
    },
    {
      "epoch": 3.683090209960937e-05,
      "model_forward_time": 3.138331413269043,
      "step": 9655
    },
    {
      "epoch": 3.683090209960937e-05,
      "step": 9655,
      "training_step_time": 9.904645442962646
    },
    {
      "epoch": 3.683090209960937e-05,
      "model_forward_time": 3.140066385269165,
      "step": 9655
    },
    {
      "epoch": 3.683090209960937e-05,
      "step": 9655,
      "training_step_time": 10.013007879257202
    },
    {
      "epoch": 3.6834716796875e-05,
      "model_forward_time": 3.153728723526001,
      "step": 9656
    },
    {
      "epoch": 3.6834716796875e-05,
      "step": 9656,
      "training_step_time": 9.917399883270264
    },
    {
      "epoch": 3.6834716796875e-05,
      "model_forward_time": 3.1352250576019287,
      "step": 9656
    },
    {
      "epoch": 3.6834716796875e-05,
      "step": 9656,
      "training_step_time": 9.900201320648193
    },
    {
      "epoch": 3.6834716796875e-05,
      "model_forward_time": 3.139085054397583,
      "step": 9656
    },
    {
      "epoch": 3.6834716796875e-05,
      "step": 9656,
      "training_step_time": 10.168884038925171
    },
    {
      "epoch": 3.6834716796875e-05,
      "model_forward_time": 3.1376378536224365,
      "step": 9656
    },
    {
      "epoch": 3.6834716796875e-05,
      "step": 9656,
      "training_step_time": 10.269957780838013
    },
    {
      "epoch": 3.6838531494140625e-05,
      "model_forward_time": 3.1516034603118896,
      "step": 9657
    },
    {
      "epoch": 3.6838531494140625e-05,
      "step": 9657,
      "training_step_time": 9.92429804801941
    },
    {
      "epoch": 3.6838531494140625e-05,
      "model_forward_time": 3.1315858364105225,
      "step": 9657
    },
    {
      "epoch": 3.6838531494140625e-05,
      "step": 9657,
      "training_step_time": 9.894931316375732
    },
    {
      "epoch": 3.6838531494140625e-05,
      "model_forward_time": 3.1397225856781006,
      "step": 9657
    },
    {
      "dynamics_loss_avg": 0.05349636301398277,
      "epoch": 3.6838531494140625e-05,
      "step": 9657
    },
    {
      "action_loss_avg": 0.018122834898531438,
      "epoch": 3.6838531494140625e-05,
      "step": 9657
    },
    {
      "epoch": 3.6838531494140625e-05,
      "step": 9657,
      "training_step_time": 9.906770944595337
    },
    {
      "epoch": 3.6838531494140625e-05,
      "model_forward_time": 3.4231390953063965,
      "step": 9657
    },
    {
      "epoch": 3.6838531494140625e-05,
      "step": 9657,
      "training_step_time": 10.282723665237427
    },
    {
      "epoch": 3.684234619140625e-05,
      "model_forward_time": 3.159885883331299,
      "step": 9658
    },
    {
      "epoch": 3.684234619140625e-05,
      "step": 9658,
      "training_step_time": 9.943004369735718
    },
    {
      "epoch": 3.684234619140625e-05,
      "model_forward_time": 3.13619327545166,
      "step": 9658
    },
    {
      "epoch": 3.684234619140625e-05,
      "step": 9658,
      "training_step_time": 9.907837152481079
    },
    {
      "epoch": 3.684234619140625e-05,
      "model_forward_time": 3.142507553100586,
      "step": 9658
    },
    {
      "epoch": 3.684234619140625e-05,
      "step": 9658,
      "training_step_time": 9.916498184204102
    },
    {
      "epoch": 3.684234619140625e-05,
      "model_forward_time": 3.141674518585205,
      "step": 9658
    },
    {
      "epoch": 3.684234619140625e-05,
      "step": 9658,
      "training_step_time": 10.318272352218628
    },
    {
      "epoch": 3.684616088867188e-05,
      "model_forward_time": 3.433159351348877,
      "step": 9659
    },
    {
      "epoch": 3.684616088867188e-05,
      "step": 9659,
      "training_step_time": 10.207317113876343
    },
    {
      "epoch": 3.684616088867188e-05,
      "model_forward_time": 3.137235641479492,
      "step": 9659
    },
    {
      "epoch": 3.684616088867188e-05,
      "step": 9659,
      "training_step_time": 9.907076835632324
    },
    {
      "epoch": 3.684616088867188e-05,
      "model_forward_time": 3.1391990184783936,
      "step": 9659
    },
    {
      "epoch": 3.684616088867188e-05,
      "step": 9659,
      "training_step_time": 9.909384489059448
    },
    {
      "epoch": 3.684616088867188e-05,
      "model_forward_time": 3.138904571533203,
      "step": 9659
    },
    {
      "epoch": 3.684616088867188e-05,
      "step": 9659,
      "training_step_time": 10.31896448135376
    },
    {
      "epoch": 3.68499755859375e-05,
      "grad_norm": 0.41468527913093567,
      "learning_rate": 1e-05,
      "loss": 0.0776,
      "step": 9660
    },
    {
      "epoch": 3.68499755859375e-05,
      "model_forward_time": 3.155064105987549,
      "step": 9660
    },
    {
      "dynamics_loss_avg": 0.05035811886191368,
      "epoch": 3.68499755859375e-05,
      "step": 9660
    },
    {
      "action_loss_avg": 0.017986524384468794,
      "epoch": 3.68499755859375e-05,
      "step": 9660
    },
    {
      "epoch": 3.68499755859375e-05,
      "step": 9660,
      "training_step_time": 9.92693281173706
    },
    {
      "epoch": 3.68499755859375e-05,
      "model_forward_time": 3.4049532413482666,
      "step": 9660
    },
    {
      "epoch": 3.68499755859375e-05,
      "step": 9660,
      "training_step_time": 10.174109935760498
    },
    {
      "epoch": 3.68499755859375e-05,
      "model_forward_time": 3.144765853881836,
      "step": 9660
    },
    {
      "epoch": 3.68499755859375e-05,
      "step": 9660,
      "training_step_time": 9.914927959442139
    },
    {
      "epoch": 3.68499755859375e-05,
      "model_forward_time": 3.1413819789886475,
      "step": 9660
    },
    {
      "epoch": 3.68499755859375e-05,
      "step": 9660,
      "training_step_time": 10.26621389389038
    },
    {
      "epoch": 3.6853790283203123e-05,
      "model_forward_time": 3.159228563308716,
      "step": 9661
    },
    {
      "epoch": 3.6853790283203123e-05,
      "step": 9661,
      "training_step_time": 9.932185411453247
    },
    {
      "epoch": 3.6853790283203123e-05,
      "model_forward_time": 3.1413462162017822,
      "step": 9661
    },
    {
      "epoch": 3.6853790283203123e-05,
      "step": 9661,
      "training_step_time": 9.912427425384521
    },
    {
      "epoch": 3.6853790283203123e-05,
      "model_forward_time": 3.433664083480835,
      "step": 9661
    },
    {
      "epoch": 3.6853790283203123e-05,
      "step": 9661,
      "training_step_time": 10.207678079605103
    },
    {
      "epoch": 3.6853790283203123e-05,
      "model_forward_time": 3.1450459957122803,
      "step": 9661
    },
    {
      "epoch": 3.6853790283203123e-05,
      "step": 9661,
      "training_step_time": 10.03255009651184
    },
    {
      "epoch": 3.685760498046875e-05,
      "model_forward_time": 3.1573410034179688,
      "step": 9662
    },
    {
      "epoch": 3.685760498046875e-05,
      "step": 9662,
      "training_step_time": 9.928751468658447
    },
    {
      "epoch": 3.685760498046875e-05,
      "model_forward_time": 3.142704963684082,
      "step": 9662
    },
    {
      "epoch": 3.685760498046875e-05,
      "step": 9662,
      "training_step_time": 9.915226221084595
    },
    {
      "epoch": 3.685760498046875e-05,
      "model_forward_time": 3.1349236965179443,
      "step": 9662
    },
    {
      "dynamics_loss_avg": 0.05515785235911608,
      "epoch": 3.685760498046875e-05,
      "step": 9662
    },
    {
      "action_loss_avg": 0.02027933797799051,
      "epoch": 3.685760498046875e-05,
      "step": 9662
    },
    {
      "epoch": 3.685760498046875e-05,
      "step": 9662,
      "training_step_time": 9.906519889831543
    },
    {
      "epoch": 3.685760498046875e-05,
      "model_forward_time": 3.138918399810791,
      "step": 9662
    },
    {
      "epoch": 3.685760498046875e-05,
      "step": 9662,
      "training_step_time": 10.281937837600708
    },
    {
      "epoch": 3.6861419677734376e-05,
      "model_forward_time": 3.1419572830200195,
      "step": 9663
    },
    {
      "epoch": 3.6861419677734376e-05,
      "step": 9663,
      "training_step_time": 9.905988693237305
    },
    {
      "epoch": 3.6861419677734376e-05,
      "model_forward_time": 3.1344985961914062,
      "step": 9663
    },
    {
      "epoch": 3.6861419677734376e-05,
      "step": 9663,
      "training_step_time": 9.897138118743896
    },
    {
      "epoch": 3.6861419677734376e-05,
      "model_forward_time": 3.135089874267578,
      "step": 9663
    },
    {
      "epoch": 3.6861419677734376e-05,
      "step": 9663,
      "training_step_time": 9.8991858959198
    },
    {
      "epoch": 3.6861419677734376e-05,
      "model_forward_time": 3.138463020324707,
      "step": 9663
    },
    {
      "epoch": 3.6861419677734376e-05,
      "step": 9663,
      "training_step_time": 10.274372816085815
    },
    {
      "epoch": 3.6865234375e-05,
      "model_forward_time": 3.440779447555542,
      "step": 9664
    },
    {
      "epoch": 3.6865234375e-05,
      "step": 9664,
      "training_step_time": 10.20679783821106
    },
    {
      "epoch": 3.6865234375e-05,
      "model_forward_time": 3.142991781234741,
      "step": 9664
    },
    {
      "epoch": 3.6865234375e-05,
      "step": 9664,
      "training_step_time": 9.91334319114685
    },
    {
      "epoch": 3.6865234375e-05,
      "model_forward_time": 3.137887477874756,
      "step": 9664
    },
    {
      "epoch": 3.6865234375e-05,
      "step": 9664,
      "training_step_time": 9.905429601669312
    },
    {
      "epoch": 3.6865234375e-05,
      "model_forward_time": 3.149080276489258,
      "step": 9664
    },
    {
      "epoch": 3.6865234375e-05,
      "step": 9664,
      "training_step_time": 10.348053693771362
    },
    {
      "epoch": 3.686904907226562e-05,
      "model_forward_time": 3.156569242477417,
      "step": 9665
    },
    {
      "dynamics_loss_avg": 0.05714659187942743,
      "epoch": 3.686904907226562e-05,
      "step": 9665
    },
    {
      "action_loss_avg": 0.016652858443558217,
      "epoch": 3.686904907226562e-05,
      "step": 9665
    },
    {
      "epoch": 3.686904907226562e-05,
      "step": 9665,
      "training_step_time": 9.92919921875
    },
    {
      "epoch": 3.686904907226562e-05,
      "model_forward_time": 3.1479132175445557,
      "step": 9665
    },
    {
      "epoch": 3.686904907226562e-05,
      "step": 9665,
      "training_step_time": 10.205930948257446
    },
    {
      "epoch": 3.686904907226562e-05,
      "model_forward_time": 3.1439249515533447,
      "step": 9665
    },
    {
      "epoch": 3.686904907226562e-05,
      "step": 9665,
      "training_step_time": 9.915110111236572
    },
    {
      "epoch": 3.686904907226562e-05,
      "model_forward_time": 3.146669387817383,
      "step": 9665
    },
    {
      "epoch": 3.686904907226562e-05,
      "step": 9665,
      "training_step_time": 10.290849208831787
    },
    {
      "epoch": 3.687286376953125e-05,
      "model_forward_time": 3.158804416656494,
      "step": 9666
    },
    {
      "epoch": 3.687286376953125e-05,
      "step": 9666,
      "training_step_time": 9.932229042053223
    },
    {
      "epoch": 3.687286376953125e-05,
      "model_forward_time": 3.1415584087371826,
      "step": 9666
    },
    {
      "epoch": 3.687286376953125e-05,
      "step": 9666,
      "training_step_time": 9.914682626724243
    },
    {
      "epoch": 3.687286376953125e-05,
      "model_forward_time": 3.4053657054901123,
      "step": 9666
    },
    {
      "epoch": 3.687286376953125e-05,
      "step": 9666,
      "training_step_time": 10.169736385345459
    },
    {
      "epoch": 3.687286376953125e-05,
      "model_forward_time": 3.1460163593292236,
      "step": 9666
    },
    {
      "epoch": 3.687286376953125e-05,
      "step": 9666,
      "training_step_time": 10.266193389892578
    },
    {
      "epoch": 3.6876678466796874e-05,
      "model_forward_time": 3.160139560699463,
      "step": 9667
    },
    {
      "epoch": 3.6876678466796874e-05,
      "step": 9667,
      "training_step_time": 9.939916372299194
    },
    {
      "epoch": 3.6876678466796874e-05,
      "model_forward_time": 3.1388330459594727,
      "step": 9667
    },
    {
      "epoch": 3.6876678466796874e-05,
      "step": 9667,
      "training_step_time": 9.906604766845703
    },
    {
      "epoch": 3.6876678466796874e-05,
      "model_forward_time": 3.139747142791748,
      "step": 9667
    },
    {
      "dynamics_loss_avg": 0.05929920338094234,
      "epoch": 3.6876678466796874e-05,
      "step": 9667
    },
    {
      "action_loss_avg": 0.023182712867856024,
      "epoch": 3.6876678466796874e-05,
      "step": 9667
    },
    {
      "epoch": 3.6876678466796874e-05,
      "step": 9667,
      "training_step_time": 9.90961742401123
    },
    {
      "epoch": 3.6876678466796874e-05,
      "model_forward_time": 3.1331143379211426,
      "step": 9667
    },
    {
      "epoch": 3.6876678466796874e-05,
      "step": 9667,
      "training_step_time": 10.287368774414062
    },
    {
      "epoch": 3.68804931640625e-05,
      "model_forward_time": 3.1518359184265137,
      "step": 9668
    },
    {
      "epoch": 3.68804931640625e-05,
      "step": 9668,
      "training_step_time": 9.92340087890625
    },
    {
      "epoch": 3.68804931640625e-05,
      "model_forward_time": 3.1366770267486572,
      "step": 9668
    },
    {
      "epoch": 3.68804931640625e-05,
      "step": 9668,
      "training_step_time": 9.903869390487671
    },
    {
      "epoch": 3.68804931640625e-05,
      "model_forward_time": 3.1362369060516357,
      "step": 9668
    },
    {
      "epoch": 3.68804931640625e-05,
      "step": 9668,
      "training_step_time": 9.905104637145996
    },
    {
      "epoch": 3.68804931640625e-05,
      "model_forward_time": 3.1337192058563232,
      "step": 9668
    },
    {
      "epoch": 3.68804931640625e-05,
      "step": 9668,
      "training_step_time": 10.33603572845459
    },
    {
      "epoch": 3.688430786132813e-05,
      "model_forward_time": 3.445202589035034,
      "step": 9669
    },
    {
      "epoch": 3.688430786132813e-05,
      "step": 9669,
      "training_step_time": 10.21334719657898
    },
    {
      "epoch": 3.688430786132813e-05,
      "model_forward_time": 3.1384177207946777,
      "step": 9669
    },
    {
      "epoch": 3.688430786132813e-05,
      "step": 9669,
      "training_step_time": 9.90413761138916
    },
    {
      "epoch": 3.688430786132813e-05,
      "model_forward_time": 3.14133620262146,
      "step": 9669
    },
    {
      "epoch": 3.688430786132813e-05,
      "step": 9669,
      "training_step_time": 9.90765118598938
    },
    {
      "epoch": 3.688430786132813e-05,
      "model_forward_time": 2.440459966659546,
      "step": 9669
    },
    {
      "epoch": 3.688430786132813e-05,
      "step": 9669,
      "training_step_time": 10.249089241027832
    },
    {
      "epoch": 3.688812255859375e-05,
      "grad_norm": 0.35437914729118347,
      "learning_rate": 1e-05,
      "loss": 0.0772,
      "step": 9670
    },
    {
      "epoch": 3.688812255859375e-05,
      "model_forward_time": 3.1410741806030273,
      "step": 9670
    },
    {
      "dynamics_loss_avg": 0.049559290148317815,
      "epoch": 3.688812255859375e-05,
      "step": 9670
    },
    {
      "action_loss_avg": 0.017289271764457225,
      "epoch": 3.688812255859375e-05,
      "step": 9670
    },
    {
      "epoch": 3.688812255859375e-05,
      "step": 9670,
      "training_step_time": 9.911583185195923
    },
    {
      "epoch": 3.688812255859375e-05,
      "model_forward_time": 3.4058725833892822,
      "step": 9670
    },
    {
      "epoch": 3.688812255859375e-05,
      "step": 9670,
      "training_step_time": 10.176247119903564
    },
    {
      "epoch": 3.688812255859375e-05,
      "model_forward_time": 3.1426877975463867,
      "step": 9670
    },
    {
      "epoch": 3.688812255859375e-05,
      "step": 9670,
      "training_step_time": 9.91336703300476
    },
    {
      "epoch": 3.688812255859375e-05,
      "model_forward_time": 3.1426494121551514,
      "step": 9670
    },
    {
      "epoch": 3.688812255859375e-05,
      "step": 9670,
      "training_step_time": 10.324041366577148
    },
    {
      "epoch": 3.689193725585937e-05,
      "model_forward_time": 3.1545205116271973,
      "step": 9671
    },
    {
      "epoch": 3.689193725585937e-05,
      "step": 9671,
      "training_step_time": 9.931583642959595
    },
    {
      "epoch": 3.689193725585937e-05,
      "model_forward_time": 3.1370110511779785,
      "step": 9671
    },
    {
      "epoch": 3.689193725585937e-05,
      "step": 9671,
      "training_step_time": 9.906323909759521
    },
    {
      "epoch": 3.689193725585937e-05,
      "model_forward_time": 3.1360065937042236,
      "step": 9671
    },
    {
      "epoch": 3.689193725585937e-05,
      "step": 9671,
      "training_step_time": 10.185357093811035
    },
    {
      "epoch": 3.689193725585937e-05,
      "model_forward_time": 3.133331775665283,
      "step": 9671
    },
    {
      "epoch": 3.689193725585937e-05,
      "step": 9671,
      "training_step_time": 10.253599166870117
    },
    {
      "epoch": 3.6895751953125e-05,
      "model_forward_time": 3.155738115310669,
      "step": 9672
    },
    {
      "epoch": 3.6895751953125e-05,
      "step": 9672,
      "training_step_time": 9.928455591201782
    },
    {
      "epoch": 3.6895751953125e-05,
      "model_forward_time": 3.1339757442474365,
      "step": 9672
    },
    {
      "epoch": 3.6895751953125e-05,
      "step": 9672,
      "training_step_time": 9.903370141983032
    },
    {
      "epoch": 3.6895751953125e-05,
      "model_forward_time": 3.1338508129119873,
      "step": 9672
    },
    {
      "dynamics_loss_avg": 0.05262108612805605,
      "epoch": 3.6895751953125e-05,
      "step": 9672
    },
    {
      "action_loss_avg": 0.01620291518047452,
      "epoch": 3.6895751953125e-05,
      "step": 9672
    },
    {
      "epoch": 3.6895751953125e-05,
      "step": 9672,
      "training_step_time": 9.904138088226318
    },
    {
      "epoch": 3.6895751953125e-05,
      "model_forward_time": 3.138829469680786,
      "step": 9672
    },
    {
      "epoch": 3.6895751953125e-05,
      "step": 9672,
      "training_step_time": 10.325875282287598
    },
    {
      "epoch": 3.6899566650390625e-05,
      "model_forward_time": 3.144768714904785,
      "step": 9673
    },
    {
      "epoch": 3.6899566650390625e-05,
      "step": 9673,
      "training_step_time": 9.917359590530396
    },
    {
      "epoch": 3.6899566650390625e-05,
      "model_forward_time": 3.137162446975708,
      "step": 9673
    },
    {
      "epoch": 3.6899566650390625e-05,
      "step": 9673,
      "training_step_time": 9.911073207855225
    },
    {
      "epoch": 3.6899566650390625e-05,
      "model_forward_time": 3.138542890548706,
      "step": 9673
    },
    {
      "epoch": 3.6899566650390625e-05,
      "step": 9673,
      "training_step_time": 9.92081618309021
    },
    {
      "epoch": 3.6899566650390625e-05,
      "model_forward_time": 3.140787363052368,
      "step": 9673
    },
    {
      "epoch": 3.6899566650390625e-05,
      "step": 9673,
      "training_step_time": 10.044995546340942
    },
    {
      "epoch": 3.690338134765625e-05,
      "model_forward_time": 3.156511068344116,
      "step": 9674
    },
    {
      "epoch": 3.690338134765625e-05,
      "step": 9674,
      "training_step_time": 10.205871820449829
    },
    {
      "epoch": 3.690338134765625e-05,
      "model_forward_time": 3.1421642303466797,
      "step": 9674
    },
    {
      "epoch": 3.690338134765625e-05,
      "step": 9674,
      "training_step_time": 9.910224437713623
    },
    {
      "epoch": 3.690338134765625e-05,
      "model_forward_time": 3.1413285732269287,
      "step": 9674
    },
    {
      "epoch": 3.690338134765625e-05,
      "step": 9674,
      "training_step_time": 9.90834927558899
    },
    {
      "epoch": 3.690338134765625e-05,
      "model_forward_time": 3.141477584838867,
      "step": 9674
    },
    {
      "epoch": 3.690338134765625e-05,
      "step": 9674,
      "training_step_time": 10.013071060180664
    },
    {
      "epoch": 3.690719604492188e-05,
      "model_forward_time": 3.1532328128814697,
      "step": 9675
    },
    {
      "dynamics_loss_avg": 0.041418157517910004,
      "epoch": 3.690719604492188e-05,
      "step": 9675
    },
    {
      "action_loss_avg": 0.015342934150248766,
      "epoch": 3.690719604492188e-05,
      "step": 9675
    },
    {
      "epoch": 3.690719604492188e-05,
      "step": 9675,
      "training_step_time": 9.926097631454468
    },
    {
      "epoch": 3.690719604492188e-05,
      "model_forward_time": 3.4244213104248047,
      "step": 9675
    },
    {
      "epoch": 3.690719604492188e-05,
      "step": 9675,
      "training_step_time": 10.188545227050781
    },
    {
      "epoch": 3.690719604492188e-05,
      "model_forward_time": 3.1386053562164307,
      "step": 9675
    },
    {
      "epoch": 3.690719604492188e-05,
      "step": 9675,
      "training_step_time": 9.909264326095581
    },
    {
      "epoch": 3.690719604492188e-05,
      "model_forward_time": 3.1443278789520264,
      "step": 9675
    },
    {
      "epoch": 3.690719604492188e-05,
      "step": 9675,
      "training_step_time": 10.296401977539062
    },
    {
      "epoch": 3.69110107421875e-05,
      "model_forward_time": 3.154801607131958,
      "step": 9676
    },
    {
      "epoch": 3.69110107421875e-05,
      "step": 9676,
      "training_step_time": 9.927385091781616
    },
    {
      "epoch": 3.69110107421875e-05,
      "model_forward_time": 3.1390411853790283,
      "step": 9676
    },
    {
      "epoch": 3.69110107421875e-05,
      "step": 9676,
      "training_step_time": 9.909562349319458
    },
    {
      "epoch": 3.69110107421875e-05,
      "model_forward_time": 3.146014451980591,
      "step": 9676
    },
    {
      "epoch": 3.69110107421875e-05,
      "step": 9676,
      "training_step_time": 10.186740636825562
    },
    {
      "epoch": 3.69110107421875e-05,
      "model_forward_time": 3.140195369720459,
      "step": 9676
    },
    {
      "epoch": 3.69110107421875e-05,
      "step": 9676,
      "training_step_time": 10.272441387176514
    },
    {
      "epoch": 3.6914825439453123e-05,
      "model_forward_time": 3.1566925048828125,
      "step": 9677
    },
    {
      "epoch": 3.6914825439453123e-05,
      "step": 9677,
      "training_step_time": 9.923635721206665
    },
    {
      "epoch": 3.6914825439453123e-05,
      "model_forward_time": 3.1365842819213867,
      "step": 9677
    },
    {
      "epoch": 3.6914825439453123e-05,
      "step": 9677,
      "training_step_time": 9.898892641067505
    },
    {
      "epoch": 3.6914825439453123e-05,
      "model_forward_time": 3.13783597946167,
      "step": 9677
    },
    {
      "dynamics_loss_avg": 0.049523367919027804,
      "epoch": 3.6914825439453123e-05,
      "step": 9677
    },
    {
      "action_loss_avg": 0.01489200689829886,
      "epoch": 3.6914825439453123e-05,
      "step": 9677
    },
    {
      "epoch": 3.6914825439453123e-05,
      "step": 9677,
      "training_step_time": 9.906407117843628
    },
    {
      "epoch": 3.6914825439453123e-05,
      "model_forward_time": 3.42549991607666,
      "step": 9677
    },
    {
      "epoch": 3.6914825439453123e-05,
      "step": 9677,
      "training_step_time": 10.325759649276733
    },
    {
      "epoch": 3.691864013671875e-05,
      "model_forward_time": 3.1498029232025146,
      "step": 9678
    },
    {
      "epoch": 3.691864013671875e-05,
      "step": 9678,
      "training_step_time": 9.920393943786621
    },
    {
      "epoch": 3.691864013671875e-05,
      "model_forward_time": 3.1376101970672607,
      "step": 9678
    },
    {
      "epoch": 3.691864013671875e-05,
      "step": 9678,
      "training_step_time": 9.904829263687134
    },
    {
      "epoch": 3.691864013671875e-05,
      "model_forward_time": 3.1432433128356934,
      "step": 9678
    },
    {
      "epoch": 3.691864013671875e-05,
      "step": 9678,
      "training_step_time": 9.90649938583374
    },
    {
      "epoch": 3.691864013671875e-05,
      "model_forward_time": 3.1385600566864014,
      "step": 9678
    },
    {
      "epoch": 3.691864013671875e-05,
      "step": 9678,
      "training_step_time": 10.005090713500977
    },
    {
      "epoch": 3.6922454833984376e-05,
      "model_forward_time": 3.4265878200531006,
      "step": 9679
    },
    {
      "epoch": 3.6922454833984376e-05,
      "step": 9679,
      "training_step_time": 10.196695804595947
    },
    {
      "epoch": 3.6922454833984376e-05,
      "model_forward_time": 3.135559558868408,
      "step": 9679
    },
    {
      "epoch": 3.6922454833984376e-05,
      "step": 9679,
      "training_step_time": 9.909376382827759
    },
    {
      "epoch": 3.6922454833984376e-05,
      "model_forward_time": 3.1462759971618652,
      "step": 9679
    },
    {
      "epoch": 3.6922454833984376e-05,
      "step": 9679,
      "training_step_time": 9.922251224517822
    },
    {
      "epoch": 3.6922454833984376e-05,
      "model_forward_time": 3.14261531829834,
      "step": 9679
    },
    {
      "epoch": 3.6922454833984376e-05,
      "step": 9679,
      "training_step_time": 10.03918719291687
    },
    {
      "epoch": 3.692626953125e-05,
      "grad_norm": 0.482757568359375,
      "learning_rate": 1e-05,
      "loss": 0.075,
      "step": 9680
    },
    {
      "epoch": 3.692626953125e-05,
      "model_forward_time": 3.153095245361328,
      "step": 9680
    },
    {
      "dynamics_loss_avg": 0.0649995643645525,
      "epoch": 3.692626953125e-05,
      "step": 9680
    },
    {
      "action_loss_avg": 0.021848801989108324,
      "epoch": 3.692626953125e-05,
      "step": 9680
    },
    {
      "epoch": 3.692626953125e-05,
      "step": 9680,
      "training_step_time": 9.931331396102905
    },
    {
      "epoch": 3.692626953125e-05,
      "model_forward_time": 3.396017551422119,
      "step": 9680
    },
    {
      "epoch": 3.692626953125e-05,
      "step": 9680,
      "training_step_time": 10.160768985748291
    },
    {
      "epoch": 3.692626953125e-05,
      "model_forward_time": 3.1455726623535156,
      "step": 9680
    },
    {
      "epoch": 3.692626953125e-05,
      "step": 9680,
      "training_step_time": 9.913320064544678
    },
    {
      "epoch": 3.692626953125e-05,
      "model_forward_time": 3.1348836421966553,
      "step": 9680
    },
    {
      "epoch": 3.692626953125e-05,
      "step": 9680,
      "training_step_time": 10.263753890991211
    },
    {
      "epoch": 3.693008422851562e-05,
      "model_forward_time": 3.152015209197998,
      "step": 9681
    },
    {
      "epoch": 3.693008422851562e-05,
      "step": 9681,
      "training_step_time": 9.91896653175354
    },
    {
      "epoch": 3.693008422851562e-05,
      "model_forward_time": 3.1384077072143555,
      "step": 9681
    },
    {
      "epoch": 3.693008422851562e-05,
      "step": 9681,
      "training_step_time": 9.906372547149658
    },
    {
      "epoch": 3.693008422851562e-05,
      "model_forward_time": 3.422800064086914,
      "step": 9681
    },
    {
      "epoch": 3.693008422851562e-05,
      "step": 9681,
      "training_step_time": 10.18787431716919
    },
    {
      "epoch": 3.693008422851562e-05,
      "model_forward_time": 3.135972261428833,
      "step": 9681
    },
    {
      "epoch": 3.693008422851562e-05,
      "step": 9681,
      "training_step_time": 10.308026313781738
    },
    {
      "epoch": 3.693389892578125e-05,
      "model_forward_time": 3.1554932594299316,
      "step": 9682
    },
    {
      "epoch": 3.693389892578125e-05,
      "step": 9682,
      "training_step_time": 9.926931381225586
    },
    {
      "epoch": 3.693389892578125e-05,
      "model_forward_time": 3.1382384300231934,
      "step": 9682
    },
    {
      "epoch": 3.693389892578125e-05,
      "step": 9682,
      "training_step_time": 9.906112909317017
    },
    {
      "epoch": 3.693389892578125e-05,
      "model_forward_time": 3.1353867053985596,
      "step": 9682
    },
    {
      "dynamics_loss_avg": 0.07254745103418828,
      "epoch": 3.693389892578125e-05,
      "step": 9682
    },
    {
      "action_loss_avg": 0.025679819658398627,
      "epoch": 3.693389892578125e-05,
      "step": 9682
    },
    {
      "epoch": 3.693389892578125e-05,
      "step": 9682,
      "training_step_time": 9.904865741729736
    },
    {
      "epoch": 3.693389892578125e-05,
      "model_forward_time": 3.1406991481781006,
      "step": 9682
    },
    {
      "epoch": 3.693389892578125e-05,
      "step": 9682,
      "training_step_time": 10.286384105682373
    },
    {
      "epoch": 3.6937713623046874e-05,
      "model_forward_time": 3.145303249359131,
      "step": 9683
    },
    {
      "epoch": 3.6937713623046874e-05,
      "step": 9683,
      "training_step_time": 9.914013147354126
    },
    {
      "epoch": 3.6937713623046874e-05,
      "model_forward_time": 3.1372883319854736,
      "step": 9683
    },
    {
      "epoch": 3.6937713623046874e-05,
      "step": 9683,
      "training_step_time": 9.909606695175171
    },
    {
      "epoch": 3.6937713623046874e-05,
      "model_forward_time": 3.142686128616333,
      "step": 9683
    },
    {
      "epoch": 3.6937713623046874e-05,
      "step": 9683,
      "training_step_time": 9.923424005508423
    },
    {
      "epoch": 3.6937713623046874e-05,
      "model_forward_time": 3.1476235389709473,
      "step": 9683
    },
    {
      "epoch": 3.6937713623046874e-05,
      "step": 9683,
      "training_step_time": 10.038206100463867
    },
    {
      "epoch": 3.69415283203125e-05,
      "model_forward_time": 3.4533801078796387,
      "step": 9684
    },
    {
      "epoch": 3.69415283203125e-05,
      "step": 9684,
      "training_step_time": 10.228392601013184
    },
    {
      "epoch": 3.69415283203125e-05,
      "model_forward_time": 3.1418981552124023,
      "step": 9684
    },
    {
      "epoch": 3.69415283203125e-05,
      "step": 9684,
      "training_step_time": 9.918533563613892
    },
    {
      "epoch": 3.69415283203125e-05,
      "model_forward_time": 3.1436784267425537,
      "step": 9684
    },
    {
      "epoch": 3.69415283203125e-05,
      "step": 9684,
      "training_step_time": 9.918946266174316
    },
    {
      "epoch": 3.69415283203125e-05,
      "model_forward_time": 3.149256706237793,
      "step": 9684
    },
    {
      "epoch": 3.69415283203125e-05,
      "step": 9684,
      "training_step_time": 10.032439231872559
    },
    {
      "epoch": 3.694534301757813e-05,
      "model_forward_time": 3.1636784076690674,
      "step": 9685
    },
    {
      "dynamics_loss_avg": 0.06635162858292461,
      "epoch": 3.694534301757813e-05,
      "step": 9685
    },
    {
      "action_loss_avg": 0.02218977946322411,
      "epoch": 3.694534301757813e-05,
      "step": 9685
    },
    {
      "epoch": 3.694534301757813e-05,
      "step": 9685,
      "training_step_time": 9.941627502441406
    },
    {
      "epoch": 3.694534301757813e-05,
      "model_forward_time": 3.1426594257354736,
      "step": 9685
    },
    {
      "epoch": 3.694534301757813e-05,
      "step": 9685,
      "training_step_time": 10.18739652633667
    },
    {
      "epoch": 3.694534301757813e-05,
      "model_forward_time": 3.1408209800720215,
      "step": 9685
    },
    {
      "epoch": 3.694534301757813e-05,
      "step": 9685,
      "training_step_time": 9.920820951461792
    },
    {
      "epoch": 3.694534301757813e-05,
      "model_forward_time": 3.1455283164978027,
      "step": 9685
    },
    {
      "epoch": 3.694534301757813e-05,
      "step": 9685,
      "training_step_time": 10.013060808181763
    },
    {
      "epoch": 3.694915771484375e-05,
      "model_forward_time": 3.1613500118255615,
      "step": 9686
    },
    {
      "epoch": 3.694915771484375e-05,
      "step": 9686,
      "training_step_time": 9.933954000473022
    },
    {
      "epoch": 3.694915771484375e-05,
      "model_forward_time": 3.144165277481079,
      "step": 9686
    },
    {
      "epoch": 3.694915771484375e-05,
      "step": 9686,
      "training_step_time": 9.911132574081421
    },
    {
      "epoch": 3.694915771484375e-05,
      "model_forward_time": 3.415529727935791,
      "step": 9686
    },
    {
      "epoch": 3.694915771484375e-05,
      "step": 9686,
      "training_step_time": 10.178122282028198
    },
    {
      "epoch": 3.694915771484375e-05,
      "model_forward_time": 3.13860821723938,
      "step": 9686
    },
    {
      "epoch": 3.694915771484375e-05,
      "step": 9686,
      "training_step_time": 10.294127464294434
    },
    {
      "epoch": 3.695297241210937e-05,
      "model_forward_time": 3.146599292755127,
      "step": 9687
    },
    {
      "epoch": 3.695297241210937e-05,
      "step": 9687,
      "training_step_time": 9.909323692321777
    },
    {
      "epoch": 3.695297241210937e-05,
      "model_forward_time": 3.137653112411499,
      "step": 9687
    },
    {
      "epoch": 3.695297241210937e-05,
      "step": 9687,
      "training_step_time": 9.898367881774902
    },
    {
      "epoch": 3.695297241210937e-05,
      "model_forward_time": 3.1393775939941406,
      "step": 9687
    },
    {
      "dynamics_loss_avg": 0.0583014365285635,
      "epoch": 3.695297241210937e-05,
      "step": 9687
    },
    {
      "action_loss_avg": 0.01805384671315551,
      "epoch": 3.695297241210937e-05,
      "step": 9687
    },
    {
      "epoch": 3.695297241210937e-05,
      "step": 9687,
      "training_step_time": 9.90824842453003
    },
    {
      "epoch": 3.695297241210937e-05,
      "model_forward_time": 3.1376147270202637,
      "step": 9687
    },
    {
      "epoch": 3.695297241210937e-05,
      "step": 9687,
      "training_step_time": 10.285686731338501
    },
    {
      "epoch": 3.6956787109375e-05,
      "model_forward_time": 3.156818151473999,
      "step": 9688
    },
    {
      "epoch": 3.6956787109375e-05,
      "step": 9688,
      "training_step_time": 9.926094055175781
    },
    {
      "epoch": 3.6956787109375e-05,
      "model_forward_time": 3.1375176906585693,
      "step": 9688
    },
    {
      "epoch": 3.6956787109375e-05,
      "step": 9688,
      "training_step_time": 9.903634548187256
    },
    {
      "epoch": 3.6956787109375e-05,
      "model_forward_time": 3.14275860786438,
      "step": 9688
    },
    {
      "epoch": 3.6956787109375e-05,
      "step": 9688,
      "training_step_time": 9.913533687591553
    },
    {
      "epoch": 3.6956787109375e-05,
      "model_forward_time": 3.145460367202759,
      "step": 9688
    },
    {
      "epoch": 3.6956787109375e-05,
      "step": 9688,
      "training_step_time": 10.347771883010864
    },
    {
      "epoch": 3.6960601806640625e-05,
      "model_forward_time": 3.4281904697418213,
      "step": 9689
    },
    {
      "epoch": 3.6960601806640625e-05,
      "step": 9689,
      "training_step_time": 10.201870918273926
    },
    {
      "epoch": 3.6960601806640625e-05,
      "model_forward_time": 3.1429920196533203,
      "step": 9689
    },
    {
      "epoch": 3.6960601806640625e-05,
      "step": 9689,
      "training_step_time": 9.917343378067017
    },
    {
      "epoch": 3.6960601806640625e-05,
      "model_forward_time": 3.1427183151245117,
      "step": 9689
    },
    {
      "epoch": 3.6960601806640625e-05,
      "step": 9689,
      "training_step_time": 9.915047645568848
    },
    {
      "epoch": 3.6960601806640625e-05,
      "model_forward_time": 3.169567823410034,
      "step": 9689
    },
    {
      "epoch": 3.6960601806640625e-05,
      "step": 9689,
      "training_step_time": 10.037727117538452
    },
    {
      "epoch": 3.696441650390625e-05,
      "grad_norm": 0.5225673317909241,
      "learning_rate": 1e-05,
      "loss": 0.0816,
      "step": 9690
    },
    {
      "epoch": 3.696441650390625e-05,
      "model_forward_time": 3.161731719970703,
      "step": 9690
    },
    {
      "dynamics_loss_avg": 0.05865914672613144,
      "epoch": 3.696441650390625e-05,
      "step": 9690
    },
    {
      "action_loss_avg": 0.017507866211235523,
      "epoch": 3.696441650390625e-05,
      "step": 9690
    },
    {
      "epoch": 3.696441650390625e-05,
      "step": 9690,
      "training_step_time": 9.936952590942383
    },
    {
      "epoch": 3.696441650390625e-05,
      "model_forward_time": 3.406233549118042,
      "step": 9690
    },
    {
      "epoch": 3.696441650390625e-05,
      "step": 9690,
      "training_step_time": 10.175753355026245
    },
    {
      "epoch": 3.696441650390625e-05,
      "model_forward_time": 3.1479408740997314,
      "step": 9690
    },
    {
      "epoch": 3.696441650390625e-05,
      "step": 9690,
      "training_step_time": 9.926948070526123
    },
    {
      "epoch": 3.696441650390625e-05,
      "model_forward_time": 3.1418778896331787,
      "step": 9690
    },
    {
      "epoch": 3.696441650390625e-05,
      "step": 9690,
      "training_step_time": 10.26423454284668
    },
    {
      "epoch": 3.696823120117188e-05,
      "model_forward_time": 3.163457155227661,
      "step": 9691
    },
    {
      "epoch": 3.696823120117188e-05,
      "step": 9691,
      "training_step_time": 9.934361219406128
    },
    {
      "epoch": 3.696823120117188e-05,
      "model_forward_time": 3.1468935012817383,
      "step": 9691
    },
    {
      "epoch": 3.696823120117188e-05,
      "step": 9691,
      "training_step_time": 9.918304681777954
    },
    {
      "epoch": 3.696823120117188e-05,
      "model_forward_time": 2.7201101779937744,
      "step": 9691
    },
    {
      "epoch": 3.696823120117188e-05,
      "step": 9691,
      "training_step_time": 7.7499613761901855
    },
    {
      "epoch": 3.696823120117188e-05,
      "model_forward_time": 3.141488790512085,
      "step": 9691
    },
    {
      "epoch": 3.696823120117188e-05,
      "step": 9691,
      "training_step_time": 10.31031608581543
    },
    {
      "epoch": 3.69720458984375e-05,
      "model_forward_time": 3.150965452194214,
      "step": 9692
    },
    {
      "epoch": 3.69720458984375e-05,
      "step": 9692,
      "training_step_time": 9.928098917007446
    },
    {
      "epoch": 3.69720458984375e-05,
      "model_forward_time": 3.146164894104004,
      "step": 9692
    },
    {
      "epoch": 3.69720458984375e-05,
      "step": 9692,
      "training_step_time": 9.91854476928711
    },
    {
      "epoch": 3.69720458984375e-05,
      "model_forward_time": 3.1528666019439697,
      "step": 9692
    },
    {
      "dynamics_loss_avg": 0.06204508282244205,
      "epoch": 3.69720458984375e-05,
      "step": 9692
    },
    {
      "action_loss_avg": 0.02029249807819724,
      "epoch": 3.69720458984375e-05,
      "step": 9692
    },
    {
      "epoch": 3.69720458984375e-05,
      "step": 9692,
      "training_step_time": 9.93724536895752
    },
    {
      "epoch": 3.69720458984375e-05,
      "model_forward_time": 3.4090940952301025,
      "step": 9692
    },
    {
      "epoch": 3.69720458984375e-05,
      "step": 9692,
      "training_step_time": 10.277805089950562
    },
    {
      "epoch": 3.6975860595703124e-05,
      "model_forward_time": 3.1529712677001953,
      "step": 9693
    },
    {
      "epoch": 3.6975860595703124e-05,
      "step": 9693,
      "training_step_time": 9.92637324333191
    },
    {
      "epoch": 3.6975860595703124e-05,
      "model_forward_time": 3.1455800533294678,
      "step": 9693
    },
    {
      "epoch": 3.6975860595703124e-05,
      "step": 9693,
      "training_step_time": 9.919481039047241
    },
    {
      "epoch": 3.6975860595703124e-05,
      "model_forward_time": 3.1412265300750732,
      "step": 9693
    },
    {
      "epoch": 3.6975860595703124e-05,
      "step": 9693,
      "training_step_time": 9.916818380355835
    },
    {
      "epoch": 3.6975860595703124e-05,
      "model_forward_time": 3.147355318069458,
      "step": 9693
    },
    {
      "epoch": 3.6975860595703124e-05,
      "step": 9693,
      "training_step_time": 10.329229354858398
    },
    {
      "epoch": 3.697967529296875e-05,
      "model_forward_time": 3.4178290367126465,
      "step": 9694
    },
    {
      "epoch": 3.697967529296875e-05,
      "step": 9694,
      "training_step_time": 10.193586349487305
    },
    {
      "epoch": 3.697967529296875e-05,
      "model_forward_time": 3.1482021808624268,
      "step": 9694
    },
    {
      "epoch": 3.697967529296875e-05,
      "step": 9694,
      "training_step_time": 9.920306921005249
    },
    {
      "epoch": 3.697967529296875e-05,
      "model_forward_time": 3.142141342163086,
      "step": 9694
    },
    {
      "epoch": 3.697967529296875e-05,
      "step": 9694,
      "training_step_time": 9.911585569381714
    },
    {
      "epoch": 3.697967529296875e-05,
      "model_forward_time": 3.1479110717773438,
      "step": 9694
    },
    {
      "epoch": 3.697967529296875e-05,
      "step": 9694,
      "training_step_time": 10.28403615951538
    },
    {
      "epoch": 3.6983489990234376e-05,
      "model_forward_time": 3.1649279594421387,
      "step": 9695
    },
    {
      "dynamics_loss_avg": 0.0504789937287569,
      "epoch": 3.6983489990234376e-05,
      "step": 9695
    },
    {
      "action_loss_avg": 0.014980474393814803,
      "epoch": 3.6983489990234376e-05,
      "step": 9695
    },
    {
      "epoch": 3.6983489990234376e-05,
      "step": 9695,
      "training_step_time": 9.94110631942749
    },
    {
      "epoch": 3.6983489990234376e-05,
      "model_forward_time": 3.423487424850464,
      "step": 9695
    },
    {
      "epoch": 3.6983489990234376e-05,
      "step": 9695,
      "training_step_time": 10.184418201446533
    },
    {
      "epoch": 3.6983489990234376e-05,
      "model_forward_time": 3.1401195526123047,
      "step": 9695
    },
    {
      "epoch": 3.6983489990234376e-05,
      "step": 9695,
      "training_step_time": 9.90405535697937
    },
    {
      "epoch": 3.6983489990234376e-05,
      "model_forward_time": 3.1396400928497314,
      "step": 9695
    },
    {
      "epoch": 3.6983489990234376e-05,
      "step": 9695,
      "training_step_time": 10.014917135238647
    },
    {
      "epoch": 3.69873046875e-05,
      "model_forward_time": 3.1535568237304688,
      "step": 9696
    },
    {
      "epoch": 3.69873046875e-05,
      "step": 9696,
      "training_step_time": 9.918084621429443
    },
    {
      "epoch": 3.69873046875e-05,
      "model_forward_time": 3.138746976852417,
      "step": 9696
    },
    {
      "epoch": 3.69873046875e-05,
      "step": 9696,
      "training_step_time": 9.904483318328857
    },
    {
      "epoch": 3.69873046875e-05,
      "model_forward_time": 3.4143056869506836,
      "step": 9696
    },
    {
      "epoch": 3.69873046875e-05,
      "step": 9696,
      "training_step_time": 10.180311441421509
    },
    {
      "epoch": 3.69873046875e-05,
      "model_forward_time": 3.1421172618865967,
      "step": 9696
    },
    {
      "epoch": 3.69873046875e-05,
      "step": 9696,
      "training_step_time": 10.28803539276123
    },
    {
      "epoch": 3.699111938476562e-05,
      "model_forward_time": 3.1568925380706787,
      "step": 9697
    },
    {
      "epoch": 3.699111938476562e-05,
      "step": 9697,
      "training_step_time": 9.928727149963379
    },
    {
      "epoch": 3.699111938476562e-05,
      "model_forward_time": 3.1385176181793213,
      "step": 9697
    },
    {
      "epoch": 3.699111938476562e-05,
      "step": 9697,
      "training_step_time": 9.910473346710205
    },
    {
      "epoch": 3.699111938476562e-05,
      "model_forward_time": 3.1434099674224854,
      "step": 9697
    },
    {
      "dynamics_loss_avg": 0.04906335868872702,
      "epoch": 3.699111938476562e-05,
      "step": 9697
    },
    {
      "action_loss_avg": 0.011797584779560565,
      "epoch": 3.699111938476562e-05,
      "step": 9697
    },
    {
      "epoch": 3.699111938476562e-05,
      "step": 9697,
      "training_step_time": 9.917268753051758
    },
    {
      "epoch": 3.699111938476562e-05,
      "model_forward_time": 3.4223439693450928,
      "step": 9697
    },
    {
      "epoch": 3.699111938476562e-05,
      "step": 9697,
      "training_step_time": 10.285359144210815
    },
    {
      "epoch": 3.699493408203125e-05,
      "model_forward_time": 3.1542739868164062,
      "step": 9698
    },
    {
      "epoch": 3.699493408203125e-05,
      "step": 9698,
      "training_step_time": 9.937803983688354
    },
    {
      "epoch": 3.699493408203125e-05,
      "model_forward_time": 3.1449549198150635,
      "step": 9698
    },
    {
      "epoch": 3.699493408203125e-05,
      "step": 9698,
      "training_step_time": 9.911026000976562
    },
    {
      "epoch": 3.699493408203125e-05,
      "model_forward_time": 3.1431121826171875,
      "step": 9698
    },
    {
      "epoch": 3.699493408203125e-05,
      "step": 9698,
      "training_step_time": 9.906050443649292
    },
    {
      "epoch": 3.699493408203125e-05,
      "model_forward_time": 3.136718273162842,
      "step": 9698
    },
    {
      "epoch": 3.699493408203125e-05,
      "step": 9698,
      "training_step_time": 10.355367183685303
    },
    {
      "epoch": 3.6998748779296874e-05,
      "model_forward_time": 3.490804433822632,
      "step": 9699
    },
    {
      "epoch": 3.6998748779296874e-05,
      "step": 9699,
      "training_step_time": 10.25991439819336
    },
    {
      "epoch": 3.6998748779296874e-05,
      "model_forward_time": 3.139862060546875,
      "step": 9699
    },
    {
      "epoch": 3.6998748779296874e-05,
      "step": 9699,
      "training_step_time": 9.909239292144775
    },
    {
      "epoch": 3.6998748779296874e-05,
      "model_forward_time": 3.146101236343384,
      "step": 9699
    },
    {
      "epoch": 3.6998748779296874e-05,
      "step": 9699,
      "training_step_time": 9.91402292251587
    },
    {
      "epoch": 3.6998748779296874e-05,
      "model_forward_time": 3.143216609954834,
      "step": 9699
    },
    {
      "epoch": 3.6998748779296874e-05,
      "step": 9699,
      "training_step_time": 10.268808126449585
    },
    {
      "epoch": 3.70025634765625e-05,
      "grad_norm": 0.2428249716758728,
      "learning_rate": 1e-05,
      "loss": 0.0774,
      "step": 9700
    },
    {
      "epoch": 3.70025634765625e-05,
      "model_forward_time": 3.14851975440979,
      "step": 9700
    },
    {
      "dynamics_loss_avg": 0.05768970903009176,
      "epoch": 3.70025634765625e-05,
      "step": 9700
    },
    {
      "action_loss_avg": 0.01806102329865098,
      "epoch": 3.70025634765625e-05,
      "step": 9700
    },
    {
      "epoch": 3.70025634765625e-05,
      "step": 9700,
      "training_step_time": 9.92223048210144
    },
    {
      "epoch": 3.70025634765625e-05,
      "model_forward_time": 2.7236976623535156,
      "step": 9700
    },
    {
      "epoch": 3.70025634765625e-05,
      "step": 9700,
      "training_step_time": 7.752074718475342
    },
    {
      "epoch": 3.70025634765625e-05,
      "model_forward_time": 3.150317907333374,
      "step": 9700
    },
    {
      "epoch": 3.70025634765625e-05,
      "step": 9700,
      "training_step_time": 9.926324844360352
    },
    {
      "epoch": 3.70025634765625e-05,
      "model_forward_time": 3.1432993412017822,
      "step": 9700
    },
    {
      "epoch": 3.70025634765625e-05,
      "step": 9700,
      "training_step_time": 10.025163173675537
    },
    {
      "epoch": 3.700637817382813e-05,
      "model_forward_time": 3.1578235626220703,
      "step": 9701
    },
    {
      "epoch": 3.700637817382813e-05,
      "step": 9701,
      "training_step_time": 9.933618068695068
    },
    {
      "epoch": 3.700637817382813e-05,
      "model_forward_time": 3.1487746238708496,
      "step": 9701
    },
    {
      "epoch": 3.700637817382813e-05,
      "step": 9701,
      "training_step_time": 9.924519538879395
    },
    {
      "epoch": 3.700637817382813e-05,
      "model_forward_time": 3.145047903060913,
      "step": 9701
    },
    {
      "epoch": 3.700637817382813e-05,
      "step": 9701,
      "training_step_time": 10.22476577758789
    },
    {
      "epoch": 3.700637817382813e-05,
      "model_forward_time": 3.1430699825286865,
      "step": 9701
    },
    {
      "epoch": 3.700637817382813e-05,
      "step": 9701,
      "training_step_time": 10.27394700050354
    },
    {
      "epoch": 3.701019287109375e-05,
      "model_forward_time": 3.1517014503479004,
      "step": 9702
    },
    {
      "epoch": 3.701019287109375e-05,
      "step": 9702,
      "training_step_time": 9.923921823501587
    },
    {
      "epoch": 3.701019287109375e-05,
      "model_forward_time": 3.1426706314086914,
      "step": 9702
    },
    {
      "epoch": 3.701019287109375e-05,
      "step": 9702,
      "training_step_time": 9.913626432418823
    },
    {
      "epoch": 3.701019287109375e-05,
      "model_forward_time": 3.141669988632202,
      "step": 9702
    },
    {
      "dynamics_loss_avg": 0.07184238787740468,
      "epoch": 3.701019287109375e-05,
      "step": 9702
    },
    {
      "action_loss_avg": 0.02636267514899373,
      "epoch": 3.701019287109375e-05,
      "step": 9702
    },
    {
      "epoch": 3.701019287109375e-05,
      "step": 9702,
      "training_step_time": 9.912896156311035
    },
    {
      "epoch": 3.701019287109375e-05,
      "model_forward_time": 3.149308204650879,
      "step": 9702
    },
    {
      "epoch": 3.701019287109375e-05,
      "step": 9702,
      "training_step_time": 10.288540840148926
    },
    {
      "epoch": 3.701400756835937e-05,
      "model_forward_time": 3.1605746746063232,
      "step": 9703
    },
    {
      "epoch": 3.701400756835937e-05,
      "step": 9703,
      "training_step_time": 9.9486403465271
    },
    {
      "epoch": 3.701400756835937e-05,
      "model_forward_time": 3.1452651023864746,
      "step": 9703
    },
    {
      "epoch": 3.701400756835937e-05,
      "step": 9703,
      "training_step_time": 9.91428256034851
    },
    {
      "epoch": 3.701400756835937e-05,
      "model_forward_time": 3.1352620124816895,
      "step": 9703
    },
    {
      "epoch": 3.701400756835937e-05,
      "step": 9703,
      "training_step_time": 9.89853024482727
    },
    {
      "epoch": 3.701400756835937e-05,
      "model_forward_time": 3.1359333992004395,
      "step": 9703
    },
    {
      "epoch": 3.701400756835937e-05,
      "step": 9703,
      "training_step_time": 10.310009002685547
    },
    {
      "epoch": 3.7017822265625e-05,
      "model_forward_time": 3.4106922149658203,
      "step": 9704
    },
    {
      "epoch": 3.7017822265625e-05,
      "step": 9704,
      "training_step_time": 10.177894353866577
    },
    {
      "epoch": 3.7017822265625e-05,
      "model_forward_time": 3.1311001777648926,
      "step": 9704
    },
    {
      "epoch": 3.7017822265625e-05,
      "step": 9704,
      "training_step_time": 9.895243644714355
    },
    {
      "epoch": 3.7017822265625e-05,
      "model_forward_time": 3.133028984069824,
      "step": 9704
    },
    {
      "epoch": 3.7017822265625e-05,
      "step": 9704,
      "training_step_time": 9.898892879486084
    },
    {
      "epoch": 3.7017822265625e-05,
      "model_forward_time": 3.143167734146118,
      "step": 9704
    },
    {
      "epoch": 3.7017822265625e-05,
      "step": 9704,
      "training_step_time": 10.261990785598755
    },
    {
      "epoch": 3.7021636962890625e-05,
      "model_forward_time": 3.152862787246704,
      "step": 9705
    },
    {
      "dynamics_loss_avg": 0.06532254591584205,
      "epoch": 3.7021636962890625e-05,
      "step": 9705
    },
    {
      "action_loss_avg": 0.018808716721832753,
      "epoch": 3.7021636962890625e-05,
      "step": 9705
    },
    {
      "epoch": 3.7021636962890625e-05,
      "step": 9705,
      "training_step_time": 9.92518663406372
    },
    {
      "epoch": 3.7021636962890625e-05,
      "model_forward_time": 3.128479480743408,
      "step": 9705
    },
    {
      "epoch": 3.7021636962890625e-05,
      "step": 9705,
      "training_step_time": 10.203611135482788
    },
    {
      "epoch": 3.7021636962890625e-05,
      "model_forward_time": 3.1332364082336426,
      "step": 9705
    },
    {
      "epoch": 3.7021636962890625e-05,
      "step": 9705,
      "training_step_time": 9.903388261795044
    },
    {
      "epoch": 3.7021636962890625e-05,
      "model_forward_time": 3.137531280517578,
      "step": 9705
    },
    {
      "epoch": 3.7021636962890625e-05,
      "step": 9705,
      "training_step_time": 10.024271965026855
    },
    {
      "epoch": 3.702545166015625e-05,
      "model_forward_time": 3.1559903621673584,
      "step": 9706
    },
    {
      "epoch": 3.702545166015625e-05,
      "step": 9706,
      "training_step_time": 9.937739849090576
    },
    {
      "epoch": 3.702545166015625e-05,
      "model_forward_time": 3.1395328044891357,
      "step": 9706
    },
    {
      "epoch": 3.702545166015625e-05,
      "step": 9706,
      "training_step_time": 9.90853214263916
    },
    {
      "epoch": 3.702545166015625e-05,
      "model_forward_time": 3.4151155948638916,
      "step": 9706
    },
    {
      "epoch": 3.702545166015625e-05,
      "step": 9706,
      "training_step_time": 10.1832275390625
    },
    {
      "epoch": 3.702545166015625e-05,
      "model_forward_time": 3.140427589416504,
      "step": 9706
    },
    {
      "epoch": 3.702545166015625e-05,
      "step": 9706,
      "training_step_time": 10.278791666030884
    },
    {
      "epoch": 3.702926635742188e-05,
      "model_forward_time": 3.153458833694458,
      "step": 9707
    },
    {
      "epoch": 3.702926635742188e-05,
      "step": 9707,
      "training_step_time": 9.921823263168335
    },
    {
      "epoch": 3.702926635742188e-05,
      "model_forward_time": 3.137575149536133,
      "step": 9707
    },
    {
      "epoch": 3.702926635742188e-05,
      "step": 9707,
      "training_step_time": 9.903106927871704
    },
    {
      "epoch": 3.702926635742188e-05,
      "model_forward_time": 3.141730546951294,
      "step": 9707
    },
    {
      "dynamics_loss_avg": 0.046688392944633963,
      "epoch": 3.702926635742188e-05,
      "step": 9707
    },
    {
      "action_loss_avg": 0.01141475564800203,
      "epoch": 3.702926635742188e-05,
      "step": 9707
    },
    {
      "epoch": 3.702926635742188e-05,
      "step": 9707,
      "training_step_time": 9.91212272644043
    },
    {
      "epoch": 3.702926635742188e-05,
      "model_forward_time": 3.1366798877716064,
      "step": 9707
    },
    {
      "epoch": 3.702926635742188e-05,
      "step": 9707,
      "training_step_time": 10.27724814414978
    },
    {
      "epoch": 3.70330810546875e-05,
      "model_forward_time": 3.1547839641571045,
      "step": 9708
    },
    {
      "epoch": 3.70330810546875e-05,
      "step": 9708,
      "training_step_time": 9.923550367355347
    },
    {
      "epoch": 3.70330810546875e-05,
      "model_forward_time": 3.13670015335083,
      "step": 9708
    },
    {
      "epoch": 3.70330810546875e-05,
      "step": 9708,
      "training_step_time": 9.904387474060059
    },
    {
      "epoch": 3.70330810546875e-05,
      "model_forward_time": 3.141873836517334,
      "step": 9708
    },
    {
      "epoch": 3.70330810546875e-05,
      "step": 9708,
      "training_step_time": 9.909169912338257
    },
    {
      "epoch": 3.70330810546875e-05,
      "model_forward_time": 3.141782760620117,
      "step": 9708
    },
    {
      "epoch": 3.70330810546875e-05,
      "step": 9708,
      "training_step_time": 10.265902042388916
    },
    {
      "epoch": 3.7036895751953124e-05,
      "model_forward_time": 3.4308011531829834,
      "step": 9709
    },
    {
      "epoch": 3.7036895751953124e-05,
      "step": 9709,
      "training_step_time": 10.20366382598877
    },
    {
      "epoch": 3.7036895751953124e-05,
      "model_forward_time": 3.1401493549346924,
      "step": 9709
    },
    {
      "epoch": 3.7036895751953124e-05,
      "step": 9709,
      "training_step_time": 9.919041633605957
    },
    {
      "epoch": 3.7036895751953124e-05,
      "model_forward_time": 3.147327184677124,
      "step": 9709
    },
    {
      "epoch": 3.7036895751953124e-05,
      "step": 9709,
      "training_step_time": 9.93096375465393
    },
    {
      "epoch": 3.7036895751953124e-05,
      "model_forward_time": 3.148014545440674,
      "step": 9709
    },
    {
      "epoch": 3.7036895751953124e-05,
      "step": 9709,
      "training_step_time": 10.316696405410767
    },
    {
      "epoch": 3.704071044921875e-05,
      "grad_norm": 0.3118247091770172,
      "learning_rate": 1e-05,
      "loss": 0.0743,
      "step": 9710
    },
    {
      "epoch": 3.704071044921875e-05,
      "model_forward_time": 3.155895471572876,
      "step": 9710
    },
    {
      "dynamics_loss_avg": 0.06989851165562869,
      "epoch": 3.704071044921875e-05,
      "step": 9710
    },
    {
      "action_loss_avg": 0.027112357318401337,
      "epoch": 3.704071044921875e-05,
      "step": 9710
    },
    {
      "epoch": 3.704071044921875e-05,
      "step": 9710,
      "training_step_time": 9.939956903457642
    },
    {
      "epoch": 3.704071044921875e-05,
      "model_forward_time": 3.4122016429901123,
      "step": 9710
    },
    {
      "epoch": 3.704071044921875e-05,
      "step": 9710,
      "training_step_time": 10.19579291343689
    },
    {
      "epoch": 3.704071044921875e-05,
      "model_forward_time": 3.1419966220855713,
      "step": 9710
    },
    {
      "epoch": 3.704071044921875e-05,
      "step": 9710,
      "training_step_time": 9.9187912940979
    },
    {
      "epoch": 3.704071044921875e-05,
      "model_forward_time": 3.136664867401123,
      "step": 9710
    },
    {
      "epoch": 3.704071044921875e-05,
      "step": 9710,
      "training_step_time": 10.337231636047363
    },
    {
      "epoch": 3.7044525146484376e-05,
      "model_forward_time": 3.16176700592041,
      "step": 9711
    },
    {
      "epoch": 3.7044525146484376e-05,
      "step": 9711,
      "training_step_time": 9.938167810440063
    },
    {
      "epoch": 3.7044525146484376e-05,
      "model_forward_time": 3.1452138423919678,
      "step": 9711
    },
    {
      "epoch": 3.7044525146484376e-05,
      "step": 9711,
      "training_step_time": 9.91912317276001
    },
    {
      "epoch": 3.7044525146484376e-05,
      "model_forward_time": 3.4240312576293945,
      "step": 9711
    },
    {
      "epoch": 3.7044525146484376e-05,
      "step": 9711,
      "training_step_time": 10.200258731842041
    },
    {
      "epoch": 3.7044525146484376e-05,
      "model_forward_time": 3.1465303897857666,
      "step": 9711
    },
    {
      "epoch": 3.7044525146484376e-05,
      "step": 9711,
      "training_step_time": 10.03941297531128
    },
    {
      "epoch": 3.704833984375e-05,
      "model_forward_time": 3.1522927284240723,
      "step": 9712
    },
    {
      "epoch": 3.704833984375e-05,
      "step": 9712,
      "training_step_time": 9.925531387329102
    },
    {
      "epoch": 3.704833984375e-05,
      "model_forward_time": 3.1440064907073975,
      "step": 9712
    },
    {
      "epoch": 3.704833984375e-05,
      "step": 9712,
      "training_step_time": 9.916086912155151
    },
    {
      "epoch": 3.704833984375e-05,
      "model_forward_time": 3.139960289001465,
      "step": 9712
    },
    {
      "dynamics_loss_avg": 0.06608248911798001,
      "epoch": 3.704833984375e-05,
      "step": 9712
    },
    {
      "action_loss_avg": 0.024338666535913944,
      "epoch": 3.704833984375e-05,
      "step": 9712
    },
    {
      "epoch": 3.704833984375e-05,
      "step": 9712,
      "training_step_time": 9.910910367965698
    },
    {
      "epoch": 3.704833984375e-05,
      "model_forward_time": 3.4318900108337402,
      "step": 9712
    },
    {
      "epoch": 3.704833984375e-05,
      "step": 9712,
      "training_step_time": 10.294677257537842
    },
    {
      "epoch": 3.705215454101562e-05,
      "model_forward_time": 3.1453142166137695,
      "step": 9713
    },
    {
      "epoch": 3.705215454101562e-05,
      "step": 9713,
      "training_step_time": 9.916225910186768
    },
    {
      "epoch": 3.705215454101562e-05,
      "model_forward_time": 3.135319232940674,
      "step": 9713
    },
    {
      "epoch": 3.705215454101562e-05,
      "step": 9713,
      "training_step_time": 9.901846170425415
    },
    {
      "epoch": 3.705215454101562e-05,
      "model_forward_time": 3.135352849960327,
      "step": 9713
    },
    {
      "epoch": 3.705215454101562e-05,
      "step": 9713,
      "training_step_time": 9.906378269195557
    },
    {
      "epoch": 3.705215454101562e-05,
      "model_forward_time": 3.1327483654022217,
      "step": 9713
    },
    {
      "epoch": 3.705215454101562e-05,
      "step": 9713,
      "training_step_time": 10.278393268585205
    },
    {
      "epoch": 3.705596923828125e-05,
      "model_forward_time": 3.419654130935669,
      "step": 9714
    },
    {
      "epoch": 3.705596923828125e-05,
      "step": 9714,
      "training_step_time": 10.192159652709961
    },
    {
      "epoch": 3.705596923828125e-05,
      "model_forward_time": 3.140806198120117,
      "step": 9714
    },
    {
      "epoch": 3.705596923828125e-05,
      "step": 9714,
      "training_step_time": 9.907264232635498
    },
    {
      "epoch": 3.705596923828125e-05,
      "model_forward_time": 2.43703293800354,
      "step": 9714
    },
    {
      "epoch": 3.705596923828125e-05,
      "step": 9714,
      "training_step_time": 7.464138031005859
    },
    {
      "epoch": 3.705596923828125e-05,
      "model_forward_time": 3.1437602043151855,
      "step": 9714
    },
    {
      "epoch": 3.705596923828125e-05,
      "step": 9714,
      "training_step_time": 10.305084466934204
    },
    {
      "epoch": 3.7059783935546874e-05,
      "model_forward_time": 3.1753036975860596,
      "step": 9715
    },
    {
      "dynamics_loss_avg": 0.060452924482524394,
      "epoch": 3.7059783935546874e-05,
      "step": 9715
    },
    {
      "action_loss_avg": 0.01645706556737423,
      "epoch": 3.7059783935546874e-05,
      "step": 9715
    },
    {
      "epoch": 3.7059783935546874e-05,
      "step": 9715,
      "training_step_time": 9.952953100204468
    },
    {
      "epoch": 3.7059783935546874e-05,
      "model_forward_time": 3.4261019229888916,
      "step": 9715
    },
    {
      "epoch": 3.7059783935546874e-05,
      "step": 9715,
      "training_step_time": 10.196114540100098
    },
    {
      "epoch": 3.7059783935546874e-05,
      "model_forward_time": 3.144566774368286,
      "step": 9715
    },
    {
      "epoch": 3.7059783935546874e-05,
      "step": 9715,
      "training_step_time": 9.919806480407715
    },
    {
      "epoch": 3.7059783935546874e-05,
      "model_forward_time": 3.1548659801483154,
      "step": 9715
    },
    {
      "epoch": 3.7059783935546874e-05,
      "step": 9715,
      "training_step_time": 10.30354928970337
    },
    {
      "epoch": 3.70635986328125e-05,
      "model_forward_time": 3.155384063720703,
      "step": 9716
    },
    {
      "epoch": 3.70635986328125e-05,
      "step": 9716,
      "training_step_time": 9.933515787124634
    },
    {
      "epoch": 3.70635986328125e-05,
      "model_forward_time": 3.1384525299072266,
      "step": 9716
    },
    {
      "epoch": 3.70635986328125e-05,
      "step": 9716,
      "training_step_time": 9.910308837890625
    },
    {
      "epoch": 3.70635986328125e-05,
      "model_forward_time": 3.391573905944824,
      "step": 9716
    },
    {
      "epoch": 3.70635986328125e-05,
      "step": 9716,
      "training_step_time": 10.162031888961792
    },
    {
      "epoch": 3.70635986328125e-05,
      "model_forward_time": 3.1374332904815674,
      "step": 9716
    },
    {
      "epoch": 3.70635986328125e-05,
      "step": 9716,
      "training_step_time": 10.360490798950195
    },
    {
      "epoch": 3.706741333007813e-05,
      "model_forward_time": 3.148627758026123,
      "step": 9717
    },
    {
      "epoch": 3.706741333007813e-05,
      "step": 9717,
      "training_step_time": 9.923293113708496
    },
    {
      "epoch": 3.706741333007813e-05,
      "model_forward_time": 3.1333580017089844,
      "step": 9717
    },
    {
      "epoch": 3.706741333007813e-05,
      "step": 9717,
      "training_step_time": 9.905732870101929
    },
    {
      "epoch": 3.706741333007813e-05,
      "model_forward_time": 3.1371679306030273,
      "step": 9717
    },
    {
      "dynamics_loss_avg": 0.06572736389935016,
      "epoch": 3.706741333007813e-05,
      "step": 9717
    },
    {
      "action_loss_avg": 0.022914672642946242,
      "epoch": 3.706741333007813e-05,
      "step": 9717
    },
    {
      "epoch": 3.706741333007813e-05,
      "step": 9717,
      "training_step_time": 9.909790992736816
    },
    {
      "epoch": 3.706741333007813e-05,
      "model_forward_time": 3.409580707550049,
      "step": 9717
    },
    {
      "epoch": 3.706741333007813e-05,
      "step": 9717,
      "training_step_time": 10.340516328811646
    },
    {
      "epoch": 3.707122802734375e-05,
      "model_forward_time": 3.1444621086120605,
      "step": 9718
    },
    {
      "epoch": 3.707122802734375e-05,
      "step": 9718,
      "training_step_time": 9.914214134216309
    },
    {
      "epoch": 3.707122802734375e-05,
      "model_forward_time": 3.1350924968719482,
      "step": 9718
    },
    {
      "epoch": 3.707122802734375e-05,
      "step": 9718,
      "training_step_time": 9.903924226760864
    },
    {
      "epoch": 3.707122802734375e-05,
      "model_forward_time": 3.141758918762207,
      "step": 9718
    },
    {
      "epoch": 3.707122802734375e-05,
      "step": 9718,
      "training_step_time": 9.916950702667236
    },
    {
      "epoch": 3.707122802734375e-05,
      "model_forward_time": 3.140587568283081,
      "step": 9718
    },
    {
      "epoch": 3.707122802734375e-05,
      "step": 9718,
      "training_step_time": 10.265120029449463
    },
    {
      "epoch": 3.707504272460937e-05,
      "model_forward_time": 3.4253814220428467,
      "step": 9719
    },
    {
      "epoch": 3.707504272460937e-05,
      "step": 9719,
      "training_step_time": 10.197103500366211
    },
    {
      "epoch": 3.707504272460937e-05,
      "model_forward_time": 3.1345412731170654,
      "step": 9719
    },
    {
      "epoch": 3.707504272460937e-05,
      "step": 9719,
      "training_step_time": 9.904371976852417
    },
    {
      "epoch": 3.707504272460937e-05,
      "model_forward_time": 3.1435112953186035,
      "step": 9719
    },
    {
      "epoch": 3.707504272460937e-05,
      "step": 9719,
      "training_step_time": 9.916635990142822
    },
    {
      "epoch": 3.707504272460937e-05,
      "model_forward_time": 3.13850736618042,
      "step": 9719
    },
    {
      "epoch": 3.707504272460937e-05,
      "step": 9719,
      "training_step_time": 10.269267082214355
    },
    {
      "epoch": 3.7078857421875e-05,
      "grad_norm": 0.3346847891807556,
      "learning_rate": 1e-05,
      "loss": 0.0773,
      "step": 9720
    },
    {
      "epoch": 3.7078857421875e-05,
      "model_forward_time": 3.153740406036377,
      "step": 9720
    },
    {
      "dynamics_loss_avg": 0.045567705668509005,
      "epoch": 3.7078857421875e-05,
      "step": 9720
    },
    {
      "action_loss_avg": 0.014084505138453097,
      "epoch": 3.7078857421875e-05,
      "step": 9720
    },
    {
      "epoch": 3.7078857421875e-05,
      "step": 9720,
      "training_step_time": 9.938063383102417
    },
    {
      "epoch": 3.7078857421875e-05,
      "model_forward_time": 3.390866279602051,
      "step": 9720
    },
    {
      "epoch": 3.7078857421875e-05,
      "step": 9720,
      "training_step_time": 10.159179925918579
    },
    {
      "epoch": 3.7078857421875e-05,
      "model_forward_time": 3.1424343585968018,
      "step": 9720
    },
    {
      "epoch": 3.7078857421875e-05,
      "step": 9720,
      "training_step_time": 9.909529685974121
    },
    {
      "epoch": 3.7078857421875e-05,
      "model_forward_time": 3.138258457183838,
      "step": 9720
    },
    {
      "epoch": 3.7078857421875e-05,
      "step": 9720,
      "training_step_time": 10.332857131958008
    },
    {
      "epoch": 3.7082672119140625e-05,
      "model_forward_time": 3.1549530029296875,
      "step": 9721
    },
    {
      "epoch": 3.7082672119140625e-05,
      "step": 9721,
      "training_step_time": 9.92805552482605
    },
    {
      "epoch": 3.7082672119140625e-05,
      "model_forward_time": 3.1457910537719727,
      "step": 9721
    },
    {
      "epoch": 3.7082672119140625e-05,
      "step": 9721,
      "training_step_time": 9.918574094772339
    },
    {
      "epoch": 3.7082672119140625e-05,
      "model_forward_time": 3.423304319381714,
      "step": 9721
    },
    {
      "epoch": 3.7082672119140625e-05,
      "step": 9721,
      "training_step_time": 10.200685262680054
    },
    {
      "epoch": 3.7082672119140625e-05,
      "model_forward_time": 3.14080548286438,
      "step": 9721
    },
    {
      "epoch": 3.7082672119140625e-05,
      "step": 9721,
      "training_step_time": 10.02937912940979
    },
    {
      "epoch": 3.708648681640625e-05,
      "model_forward_time": 3.1487069129943848,
      "step": 9722
    },
    {
      "epoch": 3.708648681640625e-05,
      "step": 9722,
      "training_step_time": 9.92482852935791
    },
    {
      "epoch": 3.708648681640625e-05,
      "model_forward_time": 3.1427533626556396,
      "step": 9722
    },
    {
      "epoch": 3.708648681640625e-05,
      "step": 9722,
      "training_step_time": 9.916995286941528
    },
    {
      "epoch": 3.708648681640625e-05,
      "model_forward_time": 3.1389763355255127,
      "step": 9722
    },
    {
      "dynamics_loss_avg": 0.06250850521028042,
      "epoch": 3.708648681640625e-05,
      "step": 9722
    },
    {
      "action_loss_avg": 0.021667202934622765,
      "epoch": 3.708648681640625e-05,
      "step": 9722
    },
    {
      "epoch": 3.708648681640625e-05,
      "step": 9722,
      "training_step_time": 9.914050817489624
    },
    {
      "epoch": 3.708648681640625e-05,
      "model_forward_time": 3.425302267074585,
      "step": 9722
    },
    {
      "epoch": 3.708648681640625e-05,
      "step": 9722,
      "training_step_time": 10.337388515472412
    },
    {
      "epoch": 3.709030151367188e-05,
      "model_forward_time": 3.1428186893463135,
      "step": 9723
    },
    {
      "epoch": 3.709030151367188e-05,
      "step": 9723,
      "training_step_time": 9.912086963653564
    },
    {
      "epoch": 3.709030151367188e-05,
      "model_forward_time": 3.138015031814575,
      "step": 9723
    },
    {
      "epoch": 3.709030151367188e-05,
      "step": 9723,
      "training_step_time": 9.908006191253662
    },
    {
      "epoch": 3.709030151367188e-05,
      "model_forward_time": 3.1344096660614014,
      "step": 9723
    },
    {
      "epoch": 3.709030151367188e-05,
      "step": 9723,
      "training_step_time": 9.899422407150269
    },
    {
      "epoch": 3.709030151367188e-05,
      "model_forward_time": 3.133575439453125,
      "step": 9723
    },
    {
      "epoch": 3.709030151367188e-05,
      "step": 9723,
      "training_step_time": 10.260663986206055
    },
    {
      "epoch": 3.70941162109375e-05,
      "model_forward_time": 3.4014713764190674,
      "step": 9724
    },
    {
      "epoch": 3.70941162109375e-05,
      "step": 9724,
      "training_step_time": 10.168032169342041
    },
    {
      "epoch": 3.70941162109375e-05,
      "model_forward_time": 3.149630069732666,
      "step": 9724
    },
    {
      "epoch": 3.70941162109375e-05,
      "step": 9724,
      "training_step_time": 9.918526649475098
    },
    {
      "epoch": 3.70941162109375e-05,
      "model_forward_time": 3.1367347240448,
      "step": 9724
    },
    {
      "epoch": 3.70941162109375e-05,
      "step": 9724,
      "training_step_time": 9.903311491012573
    },
    {
      "epoch": 3.70941162109375e-05,
      "model_forward_time": 3.140302896499634,
      "step": 9724
    },
    {
      "epoch": 3.70941162109375e-05,
      "step": 9724,
      "training_step_time": 10.009795904159546
    },
    {
      "epoch": 3.7097930908203124e-05,
      "model_forward_time": 3.1579763889312744,
      "step": 9725
    },
    {
      "dynamics_loss_avg": 0.06681746542453766,
      "epoch": 3.7097930908203124e-05,
      "step": 9725
    },
    {
      "action_loss_avg": 0.02360862968489528,
      "epoch": 3.7097930908203124e-05,
      "step": 9725
    },
    {
      "epoch": 3.7097930908203124e-05,
      "step": 9725,
      "training_step_time": 9.934112071990967
    },
    {
      "epoch": 3.7097930908203124e-05,
      "model_forward_time": 3.4249258041381836,
      "step": 9725
    },
    {
      "epoch": 3.7097930908203124e-05,
      "step": 9725,
      "training_step_time": 10.198066234588623
    },
    {
      "epoch": 3.7097930908203124e-05,
      "model_forward_time": 3.1434967517852783,
      "step": 9725
    },
    {
      "epoch": 3.7097930908203124e-05,
      "step": 9725,
      "training_step_time": 9.919721126556396
    },
    {
      "epoch": 3.7097930908203124e-05,
      "model_forward_time": 3.1418440341949463,
      "step": 9725
    },
    {
      "epoch": 3.7097930908203124e-05,
      "step": 9725,
      "training_step_time": 10.322919607162476
    },
    {
      "epoch": 3.710174560546875e-05,
      "model_forward_time": 3.152440309524536,
      "step": 9726
    },
    {
      "epoch": 3.710174560546875e-05,
      "step": 9726,
      "training_step_time": 9.934506893157959
    },
    {
      "epoch": 3.710174560546875e-05,
      "model_forward_time": 3.1361796855926514,
      "step": 9726
    },
    {
      "epoch": 3.710174560546875e-05,
      "step": 9726,
      "training_step_time": 9.90652585029602
    },
    {
      "epoch": 3.710174560546875e-05,
      "model_forward_time": 3.3817098140716553,
      "step": 9726
    },
    {
      "epoch": 3.710174560546875e-05,
      "step": 9726,
      "training_step_time": 10.153245687484741
    },
    {
      "epoch": 3.710174560546875e-05,
      "model_forward_time": 3.1385695934295654,
      "step": 9726
    },
    {
      "epoch": 3.710174560546875e-05,
      "step": 9726,
      "training_step_time": 10.30810284614563
    },
    {
      "epoch": 3.7105560302734376e-05,
      "model_forward_time": 3.150193929672241,
      "step": 9727
    },
    {
      "epoch": 3.7105560302734376e-05,
      "step": 9727,
      "training_step_time": 9.929574012756348
    },
    {
      "epoch": 3.7105560302734376e-05,
      "model_forward_time": 3.1331119537353516,
      "step": 9727
    },
    {
      "epoch": 3.7105560302734376e-05,
      "step": 9727,
      "training_step_time": 9.898755550384521
    },
    {
      "epoch": 3.7105560302734376e-05,
      "model_forward_time": 3.137147903442383,
      "step": 9727
    },
    {
      "dynamics_loss_avg": 0.06524568349123001,
      "epoch": 3.7105560302734376e-05,
      "step": 9727
    },
    {
      "action_loss_avg": 0.02055648765526712,
      "epoch": 3.7105560302734376e-05,
      "step": 9727
    },
    {
      "epoch": 3.7105560302734376e-05,
      "step": 9727,
      "training_step_time": 9.907977104187012
    },
    {
      "epoch": 3.7105560302734376e-05,
      "model_forward_time": 3.405208110809326,
      "step": 9727
    },
    {
      "epoch": 3.7105560302734376e-05,
      "step": 9727,
      "training_step_time": 10.3120596408844
    },
    {
      "epoch": 3.7109375e-05,
      "model_forward_time": 3.1538853645324707,
      "step": 9728
    },
    {
      "epoch": 3.7109375e-05,
      "step": 9728,
      "training_step_time": 9.921921253204346
    },
    {
      "epoch": 3.7109375e-05,
      "model_forward_time": 3.1336612701416016,
      "step": 9728
    },
    {
      "epoch": 3.7109375e-05,
      "step": 9728,
      "training_step_time": 9.902560234069824
    },
    {
      "epoch": 3.7109375e-05,
      "model_forward_time": 3.139394521713257,
      "step": 9728
    },
    {
      "epoch": 3.7109375e-05,
      "step": 9728,
      "training_step_time": 9.910633087158203
    },
    {
      "epoch": 3.7109375e-05,
      "model_forward_time": 3.140279531478882,
      "step": 9728
    },
    {
      "epoch": 3.7109375e-05,
      "step": 9728,
      "training_step_time": 10.271775960922241
    },
    {
      "epoch": 3.711318969726562e-05,
      "model_forward_time": 2.4520444869995117,
      "step": 9729
    },
    {
      "epoch": 3.711318969726562e-05,
      "step": 9729,
      "training_step_time": 7.479557037353516
    },
    {
      "epoch": 3.711318969726562e-05,
      "model_forward_time": 3.139723777770996,
      "step": 9729
    },
    {
      "epoch": 3.711318969726562e-05,
      "step": 9729,
      "training_step_time": 9.90746808052063
    },
    {
      "epoch": 3.711318969726562e-05,
      "model_forward_time": 3.1440858840942383,
      "step": 9729
    },
    {
      "epoch": 3.711318969726562e-05,
      "step": 9729,
      "training_step_time": 9.91400694847107
    },
    {
      "epoch": 3.711318969726562e-05,
      "model_forward_time": 3.137328624725342,
      "step": 9729
    },
    {
      "epoch": 3.711318969726562e-05,
      "step": 9729,
      "training_step_time": 10.025572776794434
    },
    {
      "epoch": 3.711700439453125e-05,
      "grad_norm": 0.564863920211792,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 9730
    },
    {
      "epoch": 3.711700439453125e-05,
      "model_forward_time": 3.1508593559265137,
      "step": 9730
    },
    {
      "dynamics_loss_avg": 0.07609231434762478,
      "epoch": 3.711700439453125e-05,
      "step": 9730
    },
    {
      "action_loss_avg": 0.022557141818106176,
      "epoch": 3.711700439453125e-05,
      "step": 9730
    },
    {
      "epoch": 3.711700439453125e-05,
      "step": 9730,
      "training_step_time": 10.192522525787354
    },
    {
      "epoch": 3.711700439453125e-05,
      "model_forward_time": 3.140084981918335,
      "step": 9730
    },
    {
      "epoch": 3.711700439453125e-05,
      "step": 9730,
      "training_step_time": 9.907943487167358
    },
    {
      "epoch": 3.711700439453125e-05,
      "model_forward_time": 3.142932415008545,
      "step": 9730
    },
    {
      "epoch": 3.711700439453125e-05,
      "step": 9730,
      "training_step_time": 9.908573627471924
    },
    {
      "epoch": 3.711700439453125e-05,
      "model_forward_time": 3.1411306858062744,
      "step": 9730
    },
    {
      "epoch": 3.711700439453125e-05,
      "step": 9730,
      "training_step_time": 10.251699924468994
    },
    {
      "epoch": 3.7120819091796875e-05,
      "model_forward_time": 3.156079053878784,
      "step": 9731
    },
    {
      "epoch": 3.7120819091796875e-05,
      "step": 9731,
      "training_step_time": 9.926542520523071
    },
    {
      "epoch": 3.7120819091796875e-05,
      "model_forward_time": 3.14431095123291,
      "step": 9731
    },
    {
      "epoch": 3.7120819091796875e-05,
      "step": 9731,
      "training_step_time": 10.191474676132202
    },
    {
      "epoch": 3.7120819091796875e-05,
      "model_forward_time": 3.1439461708068848,
      "step": 9731
    },
    {
      "epoch": 3.7120819091796875e-05,
      "step": 9731,
      "training_step_time": 9.910351991653442
    },
    {
      "epoch": 3.7120819091796875e-05,
      "model_forward_time": 3.1406643390655518,
      "step": 9731
    },
    {
      "epoch": 3.7120819091796875e-05,
      "step": 9731,
      "training_step_time": 10.331465005874634
    },
    {
      "epoch": 3.71246337890625e-05,
      "model_forward_time": 3.147735834121704,
      "step": 9732
    },
    {
      "epoch": 3.71246337890625e-05,
      "step": 9732,
      "training_step_time": 9.916109800338745
    },
    {
      "epoch": 3.71246337890625e-05,
      "model_forward_time": 3.1435930728912354,
      "step": 9732
    },
    {
      "epoch": 3.71246337890625e-05,
      "step": 9732,
      "training_step_time": 9.909051418304443
    },
    {
      "epoch": 3.71246337890625e-05,
      "model_forward_time": 3.4302706718444824,
      "step": 9732
    },
    {
      "dynamics_loss_avg": 0.05205075610429048,
      "epoch": 3.71246337890625e-05,
      "step": 9732
    },
    {
      "action_loss_avg": 0.02038013022392988,
      "epoch": 3.71246337890625e-05,
      "step": 9732
    },
    {
      "epoch": 3.71246337890625e-05,
      "step": 9732,
      "training_step_time": 10.199573755264282
    },
    {
      "epoch": 3.71246337890625e-05,
      "model_forward_time": 3.143962860107422,
      "step": 9732
    },
    {
      "epoch": 3.71246337890625e-05,
      "step": 9732,
      "training_step_time": 10.252417087554932
    },
    {
      "epoch": 3.712844848632813e-05,
      "model_forward_time": 3.1496715545654297,
      "step": 9733
    },
    {
      "epoch": 3.712844848632813e-05,
      "step": 9733,
      "training_step_time": 9.918039798736572
    },
    {
      "epoch": 3.712844848632813e-05,
      "model_forward_time": 3.1426074504852295,
      "step": 9733
    },
    {
      "epoch": 3.712844848632813e-05,
      "step": 9733,
      "training_step_time": 9.908689022064209
    },
    {
      "epoch": 3.712844848632813e-05,
      "model_forward_time": 3.142540693283081,
      "step": 9733
    },
    {
      "epoch": 3.712844848632813e-05,
      "step": 9733,
      "training_step_time": 9.912128210067749
    },
    {
      "epoch": 3.712844848632813e-05,
      "model_forward_time": 3.1467974185943604,
      "step": 9733
    },
    {
      "epoch": 3.712844848632813e-05,
      "step": 9733,
      "training_step_time": 10.270454406738281
    },
    {
      "epoch": 3.713226318359375e-05,
      "model_forward_time": 3.1490509510040283,
      "step": 9734
    },
    {
      "epoch": 3.713226318359375e-05,
      "step": 9734,
      "training_step_time": 9.921266794204712
    },
    {
      "epoch": 3.713226318359375e-05,
      "model_forward_time": 3.14890456199646,
      "step": 9734
    },
    {
      "epoch": 3.713226318359375e-05,
      "step": 9734,
      "training_step_time": 9.924862623214722
    },
    {
      "epoch": 3.713226318359375e-05,
      "model_forward_time": 3.1465678215026855,
      "step": 9734
    },
    {
      "epoch": 3.713226318359375e-05,
      "step": 9734,
      "training_step_time": 9.924908638000488
    },
    {
      "epoch": 3.713226318359375e-05,
      "model_forward_time": 3.1521806716918945,
      "step": 9734
    },
    {
      "epoch": 3.713226318359375e-05,
      "step": 9734,
      "training_step_time": 10.041129350662231
    },
    {
      "epoch": 3.713607788085937e-05,
      "model_forward_time": 3.421250820159912,
      "step": 9735
    },
    {
      "dynamics_loss_avg": 0.057391686365008354,
      "epoch": 3.713607788085937e-05,
      "step": 9735
    },
    {
      "action_loss_avg": 0.02119133984670043,
      "epoch": 3.713607788085937e-05,
      "step": 9735
    },
    {
      "epoch": 3.713607788085937e-05,
      "step": 9735,
      "training_step_time": 10.198373794555664
    },
    {
      "epoch": 3.713607788085937e-05,
      "model_forward_time": 3.14456844329834,
      "step": 9735
    },
    {
      "epoch": 3.713607788085937e-05,
      "step": 9735,
      "training_step_time": 9.91307806968689
    },
    {
      "epoch": 3.713607788085937e-05,
      "model_forward_time": 3.148266315460205,
      "step": 9735
    },
    {
      "epoch": 3.713607788085937e-05,
      "step": 9735,
      "training_step_time": 9.919824600219727
    },
    {
      "epoch": 3.713607788085937e-05,
      "model_forward_time": 3.146811008453369,
      "step": 9735
    },
    {
      "epoch": 3.713607788085937e-05,
      "step": 9735,
      "training_step_time": 10.019210815429688
    },
    {
      "epoch": 3.7139892578125e-05,
      "model_forward_time": 3.160209894180298,
      "step": 9736
    },
    {
      "epoch": 3.7139892578125e-05,
      "step": 9736,
      "training_step_time": 9.932865619659424
    },
    {
      "epoch": 3.7139892578125e-05,
      "model_forward_time": 3.431020498275757,
      "step": 9736
    },
    {
      "epoch": 3.7139892578125e-05,
      "step": 9736,
      "training_step_time": 10.201444864273071
    },
    {
      "epoch": 3.7139892578125e-05,
      "model_forward_time": 3.142313241958618,
      "step": 9736
    },
    {
      "epoch": 3.7139892578125e-05,
      "step": 9736,
      "training_step_time": 9.9067702293396
    },
    {
      "epoch": 3.7139892578125e-05,
      "model_forward_time": 3.1452841758728027,
      "step": 9736
    },
    {
      "epoch": 3.7139892578125e-05,
      "step": 9736,
      "training_step_time": 10.320047855377197
    },
    {
      "epoch": 3.7143707275390625e-05,
      "model_forward_time": 3.1565425395965576,
      "step": 9737
    },
    {
      "epoch": 3.7143707275390625e-05,
      "step": 9737,
      "training_step_time": 9.929231882095337
    },
    {
      "epoch": 3.7143707275390625e-05,
      "model_forward_time": 3.139127016067505,
      "step": 9737
    },
    {
      "epoch": 3.7143707275390625e-05,
      "step": 9737,
      "training_step_time": 9.903292417526245
    },
    {
      "epoch": 3.7143707275390625e-05,
      "model_forward_time": 2.7439894676208496,
      "step": 9737
    },
    {
      "dynamics_loss_avg": 0.04821818638592958,
      "epoch": 3.7143707275390625e-05,
      "step": 9737
    },
    {
      "action_loss_avg": 0.014213076001033187,
      "epoch": 3.7143707275390625e-05,
      "step": 9737
    },
    {
      "epoch": 3.7143707275390625e-05,
      "step": 9737,
      "training_step_time": 7.765824556350708
    },
    {
      "epoch": 3.7143707275390625e-05,
      "model_forward_time": 3.1372053623199463,
      "step": 9737
    },
    {
      "epoch": 3.7143707275390625e-05,
      "step": 9737,
      "training_step_time": 10.007743835449219
    },
    {
      "epoch": 3.714752197265625e-05,
      "model_forward_time": 3.16925311088562,
      "step": 9738
    },
    {
      "epoch": 3.714752197265625e-05,
      "step": 9738,
      "training_step_time": 9.943715333938599
    },
    {
      "epoch": 3.714752197265625e-05,
      "model_forward_time": 3.1429355144500732,
      "step": 9738
    },
    {
      "epoch": 3.714752197265625e-05,
      "step": 9738,
      "training_step_time": 9.91456413269043
    },
    {
      "epoch": 3.714752197265625e-05,
      "model_forward_time": 3.143476963043213,
      "step": 9738
    },
    {
      "epoch": 3.714752197265625e-05,
      "step": 9738,
      "training_step_time": 9.91602873802185
    },
    {
      "epoch": 3.714752197265625e-05,
      "model_forward_time": 3.146484136581421,
      "step": 9738
    },
    {
      "epoch": 3.714752197265625e-05,
      "step": 9738,
      "training_step_time": 10.299662828445435
    },
    {
      "epoch": 3.715133666992188e-05,
      "model_forward_time": 3.1771838665008545,
      "step": 9739
    },
    {
      "epoch": 3.715133666992188e-05,
      "step": 9739,
      "training_step_time": 9.958712339401245
    },
    {
      "epoch": 3.715133666992188e-05,
      "model_forward_time": 3.145726442337036,
      "step": 9739
    },
    {
      "epoch": 3.715133666992188e-05,
      "step": 9739,
      "training_step_time": 9.922906398773193
    },
    {
      "epoch": 3.715133666992188e-05,
      "model_forward_time": 3.1438815593719482,
      "step": 9739
    },
    {
      "epoch": 3.715133666992188e-05,
      "step": 9739,
      "training_step_time": 9.921356201171875
    },
    {
      "epoch": 3.715133666992188e-05,
      "model_forward_time": 3.142693519592285,
      "step": 9739
    },
    {
      "epoch": 3.715133666992188e-05,
      "step": 9739,
      "training_step_time": 10.273636817932129
    },
    {
      "epoch": 3.71551513671875e-05,
      "grad_norm": 0.4211280047893524,
      "learning_rate": 1e-05,
      "loss": 0.0807,
      "step": 9740
    },
    {
      "epoch": 3.71551513671875e-05,
      "model_forward_time": 3.4710943698883057,
      "step": 9740
    },
    {
      "dynamics_loss_avg": 0.05611554989591241,
      "epoch": 3.71551513671875e-05,
      "step": 9740
    },
    {
      "action_loss_avg": 0.017734393244609237,
      "epoch": 3.71551513671875e-05,
      "step": 9740
    },
    {
      "epoch": 3.71551513671875e-05,
      "step": 9740,
      "training_step_time": 10.247778415679932
    },
    {
      "epoch": 3.71551513671875e-05,
      "model_forward_time": 3.135981798171997,
      "step": 9740
    },
    {
      "epoch": 3.71551513671875e-05,
      "step": 9740,
      "training_step_time": 9.91286563873291
    },
    {
      "epoch": 3.71551513671875e-05,
      "model_forward_time": 3.1444802284240723,
      "step": 9740
    },
    {
      "epoch": 3.71551513671875e-05,
      "step": 9740,
      "training_step_time": 9.916773080825806
    },
    {
      "epoch": 3.71551513671875e-05,
      "model_forward_time": 3.1362900733947754,
      "step": 9740
    },
    {
      "epoch": 3.71551513671875e-05,
      "step": 9740,
      "training_step_time": 10.327924966812134
    },
    {
      "epoch": 3.7158966064453124e-05,
      "model_forward_time": 3.142164468765259,
      "step": 9741
    },
    {
      "epoch": 3.7158966064453124e-05,
      "step": 9741,
      "training_step_time": 9.919320583343506
    },
    {
      "epoch": 3.7158966064453124e-05,
      "model_forward_time": 3.405282974243164,
      "step": 9741
    },
    {
      "epoch": 3.7158966064453124e-05,
      "step": 9741,
      "training_step_time": 10.18104362487793
    },
    {
      "epoch": 3.7158966064453124e-05,
      "model_forward_time": 3.143143653869629,
      "step": 9741
    },
    {
      "epoch": 3.7158966064453124e-05,
      "step": 9741,
      "training_step_time": 9.914303302764893
    },
    {
      "epoch": 3.7158966064453124e-05,
      "model_forward_time": 3.136282444000244,
      "step": 9741
    },
    {
      "epoch": 3.7158966064453124e-05,
      "step": 9741,
      "training_step_time": 10.312584161758423
    },
    {
      "epoch": 3.716278076171875e-05,
      "model_forward_time": 3.14831280708313,
      "step": 9742
    },
    {
      "epoch": 3.716278076171875e-05,
      "step": 9742,
      "training_step_time": 9.92124319076538
    },
    {
      "epoch": 3.716278076171875e-05,
      "model_forward_time": 3.1404776573181152,
      "step": 9742
    },
    {
      "epoch": 3.716278076171875e-05,
      "step": 9742,
      "training_step_time": 9.91581392288208
    },
    {
      "epoch": 3.716278076171875e-05,
      "model_forward_time": 3.4655120372772217,
      "step": 9742
    },
    {
      "dynamics_loss_avg": 0.05238491743803024,
      "epoch": 3.716278076171875e-05,
      "step": 9742
    },
    {
      "action_loss_avg": 0.0163219244685024,
      "epoch": 3.716278076171875e-05,
      "step": 9742
    },
    {
      "epoch": 3.716278076171875e-05,
      "step": 9742,
      "training_step_time": 10.24121904373169
    },
    {
      "epoch": 3.716278076171875e-05,
      "model_forward_time": 3.1392226219177246,
      "step": 9742
    },
    {
      "epoch": 3.716278076171875e-05,
      "step": 9742,
      "training_step_time": 10.307382345199585
    },
    {
      "epoch": 3.7166595458984376e-05,
      "model_forward_time": 3.1488900184631348,
      "step": 9743
    },
    {
      "epoch": 3.7166595458984376e-05,
      "step": 9743,
      "training_step_time": 9.919938087463379
    },
    {
      "epoch": 3.7166595458984376e-05,
      "model_forward_time": 3.1418418884277344,
      "step": 9743
    },
    {
      "epoch": 3.7166595458984376e-05,
      "step": 9743,
      "training_step_time": 9.909226179122925
    },
    {
      "epoch": 3.7166595458984376e-05,
      "model_forward_time": 3.1326093673706055,
      "step": 9743
    },
    {
      "epoch": 3.7166595458984376e-05,
      "step": 9743,
      "training_step_time": 9.898816585540771
    },
    {
      "epoch": 3.7166595458984376e-05,
      "model_forward_time": 3.388143301010132,
      "step": 9743
    },
    {
      "epoch": 3.7166595458984376e-05,
      "step": 9743,
      "training_step_time": 10.288220882415771
    },
    {
      "epoch": 3.717041015625e-05,
      "model_forward_time": 3.149256706237793,
      "step": 9744
    },
    {
      "epoch": 3.717041015625e-05,
      "step": 9744,
      "training_step_time": 9.917404890060425
    },
    {
      "epoch": 3.717041015625e-05,
      "model_forward_time": 3.1378841400146484,
      "step": 9744
    },
    {
      "epoch": 3.717041015625e-05,
      "step": 9744,
      "training_step_time": 9.903997659683228
    },
    {
      "epoch": 3.717041015625e-05,
      "model_forward_time": 3.1352415084838867,
      "step": 9744
    },
    {
      "epoch": 3.717041015625e-05,
      "step": 9744,
      "training_step_time": 9.898745775222778
    },
    {
      "epoch": 3.717041015625e-05,
      "model_forward_time": 3.1437573432922363,
      "step": 9744
    },
    {
      "epoch": 3.717041015625e-05,
      "step": 9744,
      "training_step_time": 10.011260986328125
    },
    {
      "epoch": 3.717422485351562e-05,
      "model_forward_time": 3.411285400390625,
      "step": 9745
    },
    {
      "dynamics_loss_avg": 0.054790341295301914,
      "epoch": 3.717422485351562e-05,
      "step": 9745
    },
    {
      "action_loss_avg": 0.016546725551597773,
      "epoch": 3.717422485351562e-05,
      "step": 9745
    },
    {
      "epoch": 3.717422485351562e-05,
      "step": 9745,
      "training_step_time": 10.182496070861816
    },
    {
      "epoch": 3.717422485351562e-05,
      "model_forward_time": 3.147538900375366,
      "step": 9745
    },
    {
      "epoch": 3.717422485351562e-05,
      "step": 9745,
      "training_step_time": 9.918704986572266
    },
    {
      "epoch": 3.717422485351562e-05,
      "model_forward_time": 3.1390020847320557,
      "step": 9745
    },
    {
      "epoch": 3.717422485351562e-05,
      "step": 9745,
      "training_step_time": 9.905312538146973
    },
    {
      "epoch": 3.717422485351562e-05,
      "model_forward_time": 3.141122579574585,
      "step": 9745
    },
    {
      "epoch": 3.717422485351562e-05,
      "step": 9745,
      "training_step_time": 10.223034620285034
    },
    {
      "epoch": 3.717803955078125e-05,
      "model_forward_time": 3.159147262573242,
      "step": 9746
    },
    {
      "epoch": 3.717803955078125e-05,
      "step": 9746,
      "training_step_time": 9.934832096099854
    },
    {
      "epoch": 3.717803955078125e-05,
      "model_forward_time": 3.4629180431365967,
      "step": 9746
    },
    {
      "epoch": 3.717803955078125e-05,
      "step": 9746,
      "training_step_time": 10.22752332687378
    },
    {
      "epoch": 3.717803955078125e-05,
      "model_forward_time": 3.1388823986053467,
      "step": 9746
    },
    {
      "epoch": 3.717803955078125e-05,
      "step": 9746,
      "training_step_time": 9.907642841339111
    },
    {
      "epoch": 3.717803955078125e-05,
      "model_forward_time": 3.138766050338745,
      "step": 9746
    },
    {
      "epoch": 3.717803955078125e-05,
      "step": 9746,
      "training_step_time": 10.275170087814331
    },
    {
      "epoch": 3.7181854248046875e-05,
      "model_forward_time": 3.154958486557007,
      "step": 9747
    },
    {
      "epoch": 3.7181854248046875e-05,
      "step": 9747,
      "training_step_time": 9.923258543014526
    },
    {
      "epoch": 3.7181854248046875e-05,
      "model_forward_time": 3.1382837295532227,
      "step": 9747
    },
    {
      "epoch": 3.7181854248046875e-05,
      "step": 9747,
      "training_step_time": 9.906982183456421
    },
    {
      "epoch": 3.7181854248046875e-05,
      "model_forward_time": 3.403834104537964,
      "step": 9747
    },
    {
      "dynamics_loss_avg": 0.05721190813928843,
      "epoch": 3.7181854248046875e-05,
      "step": 9747
    },
    {
      "action_loss_avg": 0.016276484075933694,
      "epoch": 3.7181854248046875e-05,
      "step": 9747
    },
    {
      "epoch": 3.7181854248046875e-05,
      "step": 9747,
      "training_step_time": 10.17259168624878
    },
    {
      "epoch": 3.7181854248046875e-05,
      "model_forward_time": 3.1459720134735107,
      "step": 9747
    },
    {
      "epoch": 3.7181854248046875e-05,
      "step": 9747,
      "training_step_time": 10.291297912597656
    },
    {
      "epoch": 3.71856689453125e-05,
      "model_forward_time": 3.1521971225738525,
      "step": 9748
    },
    {
      "epoch": 3.71856689453125e-05,
      "step": 9748,
      "training_step_time": 9.926096200942993
    },
    {
      "epoch": 3.71856689453125e-05,
      "model_forward_time": 3.13702654838562,
      "step": 9748
    },
    {
      "epoch": 3.71856689453125e-05,
      "step": 9748,
      "training_step_time": 9.906221866607666
    },
    {
      "epoch": 3.71856689453125e-05,
      "model_forward_time": 3.1401681900024414,
      "step": 9748
    },
    {
      "epoch": 3.71856689453125e-05,
      "step": 9748,
      "training_step_time": 9.910268306732178
    },
    {
      "epoch": 3.71856689453125e-05,
      "model_forward_time": 3.4240994453430176,
      "step": 9748
    },
    {
      "epoch": 3.71856689453125e-05,
      "step": 9748,
      "training_step_time": 10.286365985870361
    },
    {
      "epoch": 3.718948364257813e-05,
      "model_forward_time": 3.1932547092437744,
      "step": 9749
    },
    {
      "epoch": 3.718948364257813e-05,
      "step": 9749,
      "training_step_time": 9.958008766174316
    },
    {
      "epoch": 3.718948364257813e-05,
      "model_forward_time": 3.1424577236175537,
      "step": 9749
    },
    {
      "epoch": 3.718948364257813e-05,
      "step": 9749,
      "training_step_time": 9.91529631614685
    },
    {
      "epoch": 3.718948364257813e-05,
      "model_forward_time": 3.1467719078063965,
      "step": 9749
    },
    {
      "epoch": 3.718948364257813e-05,
      "step": 9749,
      "training_step_time": 9.92195725440979
    },
    {
      "epoch": 3.718948364257813e-05,
      "model_forward_time": 3.1419425010681152,
      "step": 9749
    },
    {
      "epoch": 3.718948364257813e-05,
      "step": 9749,
      "training_step_time": 10.338375329971313
    },
    {
      "epoch": 3.719329833984375e-05,
      "grad_norm": 0.31127408146858215,
      "learning_rate": 1e-05,
      "loss": 0.079,
      "step": 9750
    },
    {
      "epoch": 3.719329833984375e-05,
      "model_forward_time": 3.469292640686035,
      "step": 9750
    },
    {
      "dynamics_loss_avg": 0.051896781846880916,
      "epoch": 3.719329833984375e-05,
      "step": 9750
    },
    {
      "action_loss_avg": 0.018679490499198436,
      "epoch": 3.719329833984375e-05,
      "step": 9750
    },
    {
      "epoch": 3.719329833984375e-05,
      "step": 9750,
      "training_step_time": 10.247904777526855
    },
    {
      "epoch": 3.719329833984375e-05,
      "model_forward_time": 2.439250946044922,
      "step": 9750
    },
    {
      "epoch": 3.719329833984375e-05,
      "step": 9750,
      "training_step_time": 7.470944881439209
    },
    {
      "epoch": 3.719329833984375e-05,
      "model_forward_time": 3.1438727378845215,
      "step": 9750
    },
    {
      "epoch": 3.719329833984375e-05,
      "step": 9750,
      "training_step_time": 9.918155193328857
    },
    {
      "epoch": 3.719329833984375e-05,
      "model_forward_time": 3.1430206298828125,
      "step": 9750
    },
    {
      "epoch": 3.719329833984375e-05,
      "step": 9750,
      "training_step_time": 10.022351026535034
    },
    {
      "epoch": 3.719711303710937e-05,
      "model_forward_time": 3.166081428527832,
      "step": 9751
    },
    {
      "epoch": 3.719711303710937e-05,
      "step": 9751,
      "training_step_time": 9.940388679504395
    },
    {
      "epoch": 3.719711303710937e-05,
      "model_forward_time": 3.1467125415802,
      "step": 9751
    },
    {
      "epoch": 3.719711303710937e-05,
      "step": 9751,
      "training_step_time": 10.204280138015747
    },
    {
      "epoch": 3.719711303710937e-05,
      "model_forward_time": 2.4370639324188232,
      "step": 9751
    },
    {
      "epoch": 3.719711303710937e-05,
      "step": 9751,
      "training_step_time": 7.469660043716431
    },
    {
      "epoch": 3.719711303710937e-05,
      "model_forward_time": 3.135881185531616,
      "step": 9751
    },
    {
      "epoch": 3.719711303710937e-05,
      "step": 9751,
      "training_step_time": 10.310393333435059
    },
    {
      "epoch": 3.7200927734375e-05,
      "model_forward_time": 3.149233102798462,
      "step": 9752
    },
    {
      "epoch": 3.7200927734375e-05,
      "step": 9752,
      "training_step_time": 9.920570850372314
    },
    {
      "epoch": 3.7200927734375e-05,
      "model_forward_time": 2.4370639324188232,
      "step": 9752
    },
    {
      "epoch": 3.7200927734375e-05,
      "step": 9752,
      "training_step_time": 7.470766544342041
    },
    {
      "epoch": 3.7200927734375e-05,
      "model_forward_time": 3.1376447677612305,
      "step": 9752
    },
    {
      "dynamics_loss_avg": 0.06410821508616209,
      "epoch": 3.7200927734375e-05,
      "step": 9752
    },
    {
      "action_loss_avg": 0.02718318169936538,
      "epoch": 3.7200927734375e-05,
      "step": 9752
    },
    {
      "epoch": 3.7200927734375e-05,
      "step": 9752,
      "training_step_time": 9.91229510307312
    },
    {
      "epoch": 3.7200927734375e-05,
      "model_forward_time": 2.716522455215454,
      "step": 9752
    },
    {
      "epoch": 3.7200927734375e-05,
      "step": 9752,
      "training_step_time": 10.291173219680786
    },
    {
      "epoch": 3.7204742431640625e-05,
      "model_forward_time": 3.150123357772827,
      "step": 9753
    },
    {
      "epoch": 3.7204742431640625e-05,
      "step": 9753,
      "training_step_time": 9.925923585891724
    },
    {
      "epoch": 3.7204742431640625e-05,
      "model_forward_time": 3.138824939727783,
      "step": 9753
    },
    {
      "epoch": 3.7204742431640625e-05,
      "step": 9753,
      "training_step_time": 9.91138219833374
    },
    {
      "epoch": 3.7204742431640625e-05,
      "model_forward_time": 3.1510989665985107,
      "step": 9753
    },
    {
      "epoch": 3.7204742431640625e-05,
      "step": 9753,
      "training_step_time": 9.937000274658203
    },
    {
      "epoch": 3.7204742431640625e-05,
      "model_forward_time": 2.437793493270874,
      "step": 9753
    },
    {
      "epoch": 3.7204742431640625e-05,
      "step": 9753,
      "training_step_time": 10.291137218475342
    },
    {
      "epoch": 3.720855712890625e-05,
      "model_forward_time": 3.447640895843506,
      "step": 9754
    },
    {
      "epoch": 3.720855712890625e-05,
      "step": 9754,
      "training_step_time": 10.240139961242676
    },
    {
      "epoch": 3.720855712890625e-05,
      "model_forward_time": 3.167912721633911,
      "step": 9754
    },
    {
      "epoch": 3.720855712890625e-05,
      "step": 9754,
      "training_step_time": 9.96464228630066
    },
    {
      "epoch": 3.720855712890625e-05,
      "model_forward_time": 3.1512322425842285,
      "step": 9754
    },
    {
      "epoch": 3.720855712890625e-05,
      "step": 9754,
      "training_step_time": 9.927555084228516
    },
    {
      "epoch": 3.720855712890625e-05,
      "model_forward_time": 2.4408702850341797,
      "step": 9754
    },
    {
      "epoch": 3.720855712890625e-05,
      "step": 9754,
      "training_step_time": 10.306153535842896
    },
    {
      "epoch": 3.721237182617188e-05,
      "model_forward_time": 3.1830687522888184,
      "step": 9755
    },
    {
      "dynamics_loss_avg": 0.04450778486207128,
      "epoch": 3.721237182617188e-05,
      "step": 9755
    },
    {
      "action_loss_avg": 0.015342323947697878,
      "epoch": 3.721237182617188e-05,
      "step": 9755
    },
    {
      "epoch": 3.721237182617188e-05,
      "step": 9755,
      "training_step_time": 9.96168065071106
    },
    {
      "epoch": 3.721237182617188e-05,
      "model_forward_time": 2.071112871170044,
      "step": 9755
    },
    {
      "epoch": 3.721237182617188e-05,
      "step": 9755,
      "training_step_time": 5.528963327407837
    },
    {
      "epoch": 3.721237182617188e-05,
      "model_forward_time": 3.139075756072998,
      "step": 9755
    },
    {
      "epoch": 3.721237182617188e-05,
      "step": 9755,
      "training_step_time": 9.91482138633728
    },
    {
      "epoch": 3.721237182617188e-05,
      "model_forward_time": 3.1629624366760254,
      "step": 9755
    },
    {
      "epoch": 3.721237182617188e-05,
      "step": 9755,
      "training_step_time": 10.035046100616455
    },
    {
      "epoch": 3.72161865234375e-05,
      "model_forward_time": 3.176116943359375,
      "step": 9756
    },
    {
      "epoch": 3.72161865234375e-05,
      "step": 9756,
      "training_step_time": 9.953070640563965
    },
    {
      "epoch": 3.72161865234375e-05,
      "model_forward_time": 2.4440488815307617,
      "step": 9756
    },
    {
      "epoch": 3.72161865234375e-05,
      "step": 9756,
      "training_step_time": 7.475881099700928
    },
    {
      "epoch": 3.72161865234375e-05,
      "model_forward_time": 2.4491658210754395,
      "step": 9756
    },
    {
      "epoch": 3.72161865234375e-05,
      "step": 9756,
      "training_step_time": 7.7876670360565186
    },
    {
      "epoch": 3.72161865234375e-05,
      "model_forward_time": 3.153134346008301,
      "step": 9756
    },
    {
      "epoch": 3.72161865234375e-05,
      "step": 9756,
      "training_step_time": 10.395783424377441
    },
    {
      "epoch": 3.7220001220703124e-05,
      "model_forward_time": 3.1590402126312256,
      "step": 9757
    },
    {
      "epoch": 3.7220001220703124e-05,
      "step": 9757,
      "training_step_time": 9.935342073440552
    },
    {
      "epoch": 3.7220001220703124e-05,
      "model_forward_time": 3.144782543182373,
      "step": 9757
    },
    {
      "epoch": 3.7220001220703124e-05,
      "step": 9757,
      "training_step_time": 9.91799259185791
    },
    {
      "epoch": 3.7220001220703124e-05,
      "model_forward_time": 2.4422357082366943,
      "step": 9757
    },
    {
      "dynamics_loss_avg": 0.06060667028650642,
      "epoch": 3.7220001220703124e-05,
      "step": 9757
    },
    {
      "action_loss_avg": 0.021698371367529036,
      "epoch": 3.7220001220703124e-05,
      "step": 9757
    },
    {
      "epoch": 3.7220001220703124e-05,
      "step": 9757,
      "training_step_time": 7.473000764846802
    },
    {
      "epoch": 3.7220001220703124e-05,
      "model_forward_time": 3.1480274200439453,
      "step": 9757
    },
    {
      "epoch": 3.7220001220703124e-05,
      "step": 9757,
      "training_step_time": 10.314521312713623
    },
    {
      "epoch": 3.722381591796875e-05,
      "model_forward_time": 2.451462984085083,
      "step": 9758
    },
    {
      "epoch": 3.722381591796875e-05,
      "step": 9758,
      "training_step_time": 7.481375694274902
    },
    {
      "epoch": 3.722381591796875e-05,
      "model_forward_time": 3.144937515258789,
      "step": 9758
    },
    {
      "epoch": 3.722381591796875e-05,
      "step": 9758,
      "training_step_time": 9.918539047241211
    },
    {
      "epoch": 3.722381591796875e-05,
      "model_forward_time": 3.1529829502105713,
      "step": 9758
    },
    {
      "epoch": 3.722381591796875e-05,
      "step": 9758,
      "training_step_time": 9.928677797317505
    },
    {
      "epoch": 3.722381591796875e-05,
      "model_forward_time": 2.442206382751465,
      "step": 9758
    },
    {
      "epoch": 3.722381591796875e-05,
      "step": 9758,
      "training_step_time": 10.23482346534729
    },
    {
      "epoch": 3.7227630615234376e-05,
      "model_forward_time": 2.4600203037261963,
      "step": 9759
    },
    {
      "epoch": 3.7227630615234376e-05,
      "step": 9759,
      "training_step_time": 7.492877006530762
    },
    {
      "epoch": 3.7227630615234376e-05,
      "model_forward_time": 3.4440126419067383,
      "step": 9759
    },
    {
      "epoch": 3.7227630615234376e-05,
      "step": 9759,
      "training_step_time": 10.213963031768799
    },
    {
      "epoch": 3.7227630615234376e-05,
      "model_forward_time": 2.4403374195098877,
      "step": 9759
    },
    {
      "epoch": 3.7227630615234376e-05,
      "step": 9759,
      "training_step_time": 7.467543840408325
    },
    {
      "epoch": 3.7227630615234376e-05,
      "model_forward_time": 3.140988826751709,
      "step": 9759
    },
    {
      "epoch": 3.7227630615234376e-05,
      "step": 9759,
      "training_step_time": 10.268619775772095
    },
    {
      "epoch": 3.72314453125e-05,
      "grad_norm": 0.5164713859558105,
      "learning_rate": 1e-05,
      "loss": 0.0777,
      "step": 9760
    },
    {
      "epoch": 3.72314453125e-05,
      "model_forward_time": 2.4512360095977783,
      "step": 9760
    },
    {
      "dynamics_loss_avg": 0.042343768291175365,
      "epoch": 3.72314453125e-05,
      "step": 9760
    },
    {
      "action_loss_avg": 0.012845250777900219,
      "epoch": 3.72314453125e-05,
      "step": 9760
    },
    {
      "epoch": 3.72314453125e-05,
      "step": 9760,
      "training_step_time": 7.484497308731079
    },
    {
      "epoch": 3.72314453125e-05,
      "model_forward_time": 2.436311960220337,
      "step": 9760
    },
    {
      "epoch": 3.72314453125e-05,
      "step": 9760,
      "training_step_time": 7.465224266052246
    },
    {
      "epoch": 3.72314453125e-05,
      "model_forward_time": 3.4310336112976074,
      "step": 9760
    },
    {
      "epoch": 3.72314453125e-05,
      "step": 9760,
      "training_step_time": 10.2088942527771
    },
    {
      "epoch": 3.72314453125e-05,
      "model_forward_time": 3.1497278213500977,
      "step": 9760
    },
    {
      "epoch": 3.72314453125e-05,
      "step": 9760,
      "training_step_time": 10.02226185798645
    },
    {
      "epoch": 3.723526000976562e-05,
      "model_forward_time": 3.1676955223083496,
      "step": 9761
    },
    {
      "epoch": 3.723526000976562e-05,
      "step": 9761,
      "training_step_time": 9.942487001419067
    },
    {
      "epoch": 3.723526000976562e-05,
      "model_forward_time": 2.4389123916625977,
      "step": 9761
    },
    {
      "epoch": 3.723526000976562e-05,
      "step": 9761,
      "training_step_time": 7.467993259429932
    },
    {
      "epoch": 3.723526000976562e-05,
      "model_forward_time": 3.1473889350891113,
      "step": 9761
    },
    {
      "epoch": 3.723526000976562e-05,
      "step": 9761,
      "training_step_time": 9.922268152236938
    },
    {
      "epoch": 3.723526000976562e-05,
      "model_forward_time": 2.7614500522613525,
      "step": 9761
    },
    {
      "epoch": 3.723526000976562e-05,
      "step": 9761,
      "training_step_time": 10.028371334075928
    },
    {
      "epoch": 3.723907470703125e-05,
      "model_forward_time": 2.452014923095703,
      "step": 9762
    },
    {
      "epoch": 3.723907470703125e-05,
      "step": 9762,
      "training_step_time": 7.4842987060546875
    },
    {
      "epoch": 3.723907470703125e-05,
      "model_forward_time": 3.1424198150634766,
      "step": 9762
    },
    {
      "epoch": 3.723907470703125e-05,
      "step": 9762,
      "training_step_time": 9.913752555847168
    },
    {
      "epoch": 3.723907470703125e-05,
      "model_forward_time": 2.439661979675293,
      "step": 9762
    },
    {
      "dynamics_loss_avg": 0.055929421447217464,
      "epoch": 3.723907470703125e-05,
      "step": 9762
    },
    {
      "action_loss_avg": 0.01946005839854479,
      "epoch": 3.723907470703125e-05,
      "step": 9762
    },
    {
      "epoch": 3.723907470703125e-05,
      "step": 9762,
      "training_step_time": 7.473256587982178
    },
    {
      "epoch": 3.723907470703125e-05,
      "model_forward_time": 3.1453962326049805,
      "step": 9762
    },
    {
      "epoch": 3.723907470703125e-05,
      "step": 9762,
      "training_step_time": 10.304081439971924
    },
    {
      "epoch": 3.7242889404296875e-05,
      "model_forward_time": 3.164236068725586,
      "step": 9763
    },
    {
      "epoch": 3.7242889404296875e-05,
      "step": 9763,
      "training_step_time": 9.935102701187134
    },
    {
      "epoch": 3.7242889404296875e-05,
      "model_forward_time": 2.7057876586914062,
      "step": 9763
    },
    {
      "epoch": 3.7242889404296875e-05,
      "step": 9763,
      "training_step_time": 7.735986232757568
    },
    {
      "epoch": 3.7242889404296875e-05,
      "model_forward_time": 3.146902561187744,
      "step": 9763
    },
    {
      "epoch": 3.7242889404296875e-05,
      "step": 9763,
      "training_step_time": 9.917165994644165
    },
    {
      "epoch": 3.7242889404296875e-05,
      "model_forward_time": 3.1461803913116455,
      "step": 9763
    },
    {
      "epoch": 3.7242889404296875e-05,
      "step": 9763,
      "training_step_time": 10.012874841690063
    },
    {
      "epoch": 3.72467041015625e-05,
      "model_forward_time": 3.161726236343384,
      "step": 9764
    },
    {
      "epoch": 3.72467041015625e-05,
      "step": 9764,
      "training_step_time": 9.929666996002197
    },
    {
      "epoch": 3.72467041015625e-05,
      "model_forward_time": 2.4393107891082764,
      "step": 9764
    },
    {
      "epoch": 3.72467041015625e-05,
      "step": 9764,
      "training_step_time": 7.463759422302246
    },
    {
      "epoch": 3.72467041015625e-05,
      "model_forward_time": 3.3988561630249023,
      "step": 9764
    },
    {
      "epoch": 3.72467041015625e-05,
      "step": 9764,
      "training_step_time": 10.168139696121216
    },
    {
      "epoch": 3.72467041015625e-05,
      "model_forward_time": 2.437713146209717,
      "step": 9764
    },
    {
      "epoch": 3.72467041015625e-05,
      "step": 9764,
      "training_step_time": 10.286316871643066
    },
    {
      "epoch": 3.725051879882813e-05,
      "model_forward_time": 2.460909605026245,
      "step": 9765
    },
    {
      "dynamics_loss_avg": 0.04835313819348812,
      "epoch": 3.725051879882813e-05,
      "step": 9765
    },
    {
      "action_loss_avg": 0.01823517270386219,
      "epoch": 3.725051879882813e-05,
      "step": 9765
    },
    {
      "epoch": 3.725051879882813e-05,
      "step": 9765,
      "training_step_time": 7.4952921867370605
    },
    {
      "epoch": 3.725051879882813e-05,
      "model_forward_time": 3.1422536373138428,
      "step": 9765
    },
    {
      "epoch": 3.725051879882813e-05,
      "step": 9765,
      "training_step_time": 9.911138534545898
    },
    {
      "epoch": 3.725051879882813e-05,
      "model_forward_time": 2.4398138523101807,
      "step": 9765
    },
    {
      "epoch": 3.725051879882813e-05,
      "step": 9765,
      "training_step_time": 7.464627265930176
    },
    {
      "epoch": 3.725051879882813e-05,
      "model_forward_time": 2.7188942432403564,
      "step": 9765
    },
    {
      "epoch": 3.725051879882813e-05,
      "step": 9765,
      "training_step_time": 10.0082106590271
    },
    {
      "epoch": 3.725433349609375e-05,
      "model_forward_time": 3.1645543575286865,
      "step": 9766
    },
    {
      "epoch": 3.725433349609375e-05,
      "step": 9766,
      "training_step_time": 9.933069944381714
    },
    {
      "epoch": 3.725433349609375e-05,
      "model_forward_time": 3.139955759048462,
      "step": 9766
    },
    {
      "epoch": 3.725433349609375e-05,
      "step": 9766,
      "training_step_time": 9.909315824508667
    },
    {
      "epoch": 3.725433349609375e-05,
      "model_forward_time": 3.1450955867767334,
      "step": 9766
    },
    {
      "epoch": 3.725433349609375e-05,
      "step": 9766,
      "training_step_time": 9.910920858383179
    },
    {
      "epoch": 3.725433349609375e-05,
      "model_forward_time": 2.43979549407959,
      "step": 9766
    },
    {
      "epoch": 3.725433349609375e-05,
      "step": 9766,
      "training_step_time": 10.020587682723999
    },
    {
      "epoch": 3.725814819335937e-05,
      "model_forward_time": 3.431595802307129,
      "step": 9767
    },
    {
      "epoch": 3.725814819335937e-05,
      "step": 9767,
      "training_step_time": 10.195045471191406
    },
    {
      "epoch": 3.725814819335937e-05,
      "model_forward_time": 2.435767889022827,
      "step": 9767
    },
    {
      "epoch": 3.725814819335937e-05,
      "step": 9767,
      "training_step_time": 7.457196474075317
    },
    {
      "epoch": 3.725814819335937e-05,
      "model_forward_time": 2.435595989227295,
      "step": 9767
    },
    {
      "dynamics_loss_avg": 0.05978717505931854,
      "epoch": 3.725814819335937e-05,
      "step": 9767
    },
    {
      "action_loss_avg": 0.019495782628655433,
      "epoch": 3.725814819335937e-05,
      "step": 9767
    },
    {
      "epoch": 3.725814819335937e-05,
      "step": 9767,
      "training_step_time": 7.460334062576294
    },
    {
      "epoch": 3.725814819335937e-05,
      "model_forward_time": 2.4430928230285645,
      "step": 9767
    },
    {
      "epoch": 3.725814819335937e-05,
      "step": 9767,
      "training_step_time": 10.286166429519653
    },
    {
      "epoch": 3.7261962890625e-05,
      "model_forward_time": 2.4498701095581055,
      "step": 9768
    },
    {
      "epoch": 3.7261962890625e-05,
      "step": 9768,
      "training_step_time": 7.47319769859314
    },
    {
      "epoch": 3.7261962890625e-05,
      "model_forward_time": 2.4357080459594727,
      "step": 9768
    },
    {
      "epoch": 3.7261962890625e-05,
      "step": 9768,
      "training_step_time": 7.458817005157471
    },
    {
      "epoch": 3.7261962890625e-05,
      "model_forward_time": 2.7445805072784424,
      "step": 9768
    },
    {
      "epoch": 3.7261962890625e-05,
      "step": 9768,
      "training_step_time": 7.768665790557861
    },
    {
      "epoch": 3.7261962890625e-05,
      "model_forward_time": 2.434913158416748,
      "step": 9768
    },
    {
      "epoch": 3.7261962890625e-05,
      "step": 9768,
      "training_step_time": 10.288501024246216
    },
    {
      "epoch": 3.7265777587890625e-05,
      "model_forward_time": 3.1636006832122803,
      "step": 9769
    },
    {
      "epoch": 3.7265777587890625e-05,
      "step": 9769,
      "training_step_time": 9.939280986785889
    },
    {
      "epoch": 3.7265777587890625e-05,
      "model_forward_time": 2.4378042221069336,
      "step": 9769
    },
    {
      "epoch": 3.7265777587890625e-05,
      "step": 9769,
      "training_step_time": 7.461819410324097
    },
    {
      "epoch": 3.7265777587890625e-05,
      "model_forward_time": 2.4385273456573486,
      "step": 9769
    },
    {
      "epoch": 3.7265777587890625e-05,
      "step": 9769,
      "training_step_time": 7.459447145462036
    },
    {
      "epoch": 3.7265777587890625e-05,
      "model_forward_time": 2.7452476024627686,
      "step": 9769
    },
    {
      "epoch": 3.7265777587890625e-05,
      "step": 9769,
      "training_step_time": 10.300430059432983
    },
    {
      "epoch": 3.726959228515625e-05,
      "grad_norm": 0.42041438817977905,
      "learning_rate": 1e-05,
      "loss": 0.0765,
      "step": 9770
    },
    {
      "epoch": 3.726959228515625e-05,
      "model_forward_time": 3.1618454456329346,
      "step": 9770
    },
    {
      "dynamics_loss_avg": 0.045807562116533515,
      "epoch": 3.726959228515625e-05,
      "step": 9770
    },
    {
      "action_loss_avg": 0.014794777240604162,
      "epoch": 3.726959228515625e-05,
      "step": 9770
    },
    {
      "epoch": 3.726959228515625e-05,
      "step": 9770,
      "training_step_time": 9.933584451675415
    },
    {
      "epoch": 3.726959228515625e-05,
      "model_forward_time": 3.138423204421997,
      "step": 9770
    },
    {
      "epoch": 3.726959228515625e-05,
      "step": 9770,
      "training_step_time": 9.901989221572876
    },
    {
      "epoch": 3.726959228515625e-05,
      "model_forward_time": 2.436069965362549,
      "step": 9770
    },
    {
      "epoch": 3.726959228515625e-05,
      "step": 9770,
      "training_step_time": 7.458566904067993
    },
    {
      "epoch": 3.726959228515625e-05,
      "model_forward_time": 2.4378747940063477,
      "step": 9770
    },
    {
      "epoch": 3.726959228515625e-05,
      "step": 9770,
      "training_step_time": 10.007221460342407
    },
    {
      "epoch": 3.727340698242188e-05,
      "model_forward_time": 2.719092607498169,
      "step": 9771
    },
    {
      "epoch": 3.727340698242188e-05,
      "step": 9771,
      "training_step_time": 7.744625091552734
    },
    {
      "epoch": 3.727340698242188e-05,
      "model_forward_time": 3.138711929321289,
      "step": 9771
    },
    {
      "epoch": 3.727340698242188e-05,
      "step": 9771,
      "training_step_time": 9.901914834976196
    },
    {
      "epoch": 3.727340698242188e-05,
      "model_forward_time": 2.434349536895752,
      "step": 9771
    },
    {
      "epoch": 3.727340698242188e-05,
      "step": 9771,
      "training_step_time": 7.454995155334473
    },
    {
      "epoch": 3.727340698242188e-05,
      "model_forward_time": 3.1381444931030273,
      "step": 9771
    },
    {
      "epoch": 3.727340698242188e-05,
      "step": 9771,
      "training_step_time": 10.005993843078613
    },
    {
      "epoch": 3.72772216796875e-05,
      "model_forward_time": 2.4952216148376465,
      "step": 9772
    },
    {
      "epoch": 3.72772216796875e-05,
      "step": 9772,
      "training_step_time": 7.522023439407349
    },
    {
      "epoch": 3.72772216796875e-05,
      "model_forward_time": 2.432441234588623,
      "step": 9772
    },
    {
      "epoch": 3.72772216796875e-05,
      "step": 9772,
      "training_step_time": 7.712061166763306
    },
    {
      "epoch": 3.72772216796875e-05,
      "model_forward_time": 2.441699266433716,
      "step": 9772
    },
    {
      "dynamics_loss_avg": 0.053978667967021464,
      "epoch": 3.72772216796875e-05,
      "step": 9772
    },
    {
      "action_loss_avg": 0.021944285603240132,
      "epoch": 3.72772216796875e-05,
      "step": 9772
    },
    {
      "epoch": 3.72772216796875e-05,
      "step": 9772,
      "training_step_time": 7.467446804046631
    },
    {
      "epoch": 3.72772216796875e-05,
      "model_forward_time": 3.1458427906036377,
      "step": 9772
    },
    {
      "epoch": 3.72772216796875e-05,
      "step": 9772,
      "training_step_time": 10.261136054992676
    },
    {
      "epoch": 3.7281036376953124e-05,
      "model_forward_time": 2.4475436210632324,
      "step": 9773
    },
    {
      "epoch": 3.7281036376953124e-05,
      "step": 9773,
      "training_step_time": 7.469320058822632
    },
    {
      "epoch": 3.7281036376953124e-05,
      "model_forward_time": 3.1444618701934814,
      "step": 9773
    },
    {
      "epoch": 3.7281036376953124e-05,
      "step": 9773,
      "training_step_time": 9.91292953491211
    },
    {
      "epoch": 3.7281036376953124e-05,
      "model_forward_time": 3.147395610809326,
      "step": 9773
    },
    {
      "epoch": 3.7281036376953124e-05,
      "step": 9773,
      "training_step_time": 9.926530599594116
    },
    {
      "epoch": 3.7281036376953124e-05,
      "model_forward_time": 3.467562198638916,
      "step": 9773
    },
    {
      "epoch": 3.7281036376953124e-05,
      "step": 9773,
      "training_step_time": 10.329740285873413
    },
    {
      "epoch": 3.728485107421875e-05,
      "model_forward_time": 2.4552385807037354,
      "step": 9774
    },
    {
      "epoch": 3.728485107421875e-05,
      "step": 9774,
      "training_step_time": 7.480260610580444
    },
    {
      "epoch": 3.728485107421875e-05,
      "model_forward_time": 2.4401350021362305,
      "step": 9774
    },
    {
      "epoch": 3.728485107421875e-05,
      "step": 9774,
      "training_step_time": 7.466979265213013
    },
    {
      "epoch": 3.728485107421875e-05,
      "model_forward_time": 2.4391610622406006,
      "step": 9774
    },
    {
      "epoch": 3.728485107421875e-05,
      "step": 9774,
      "training_step_time": 7.464877367019653
    },
    {
      "epoch": 3.728485107421875e-05,
      "model_forward_time": 2.4348158836364746,
      "step": 9774
    },
    {
      "epoch": 3.728485107421875e-05,
      "step": 9774,
      "training_step_time": 10.324320793151855
    },
    {
      "epoch": 3.7288665771484376e-05,
      "model_forward_time": 3.4780678749084473,
      "step": 9775
    },
    {
      "dynamics_loss_avg": 0.05632412899285555,
      "epoch": 3.7288665771484376e-05,
      "step": 9775
    },
    {
      "action_loss_avg": 0.019953582854941487,
      "epoch": 3.7288665771484376e-05,
      "step": 9775
    },
    {
      "epoch": 3.7288665771484376e-05,
      "step": 9775,
      "training_step_time": 10.246304988861084
    },
    {
      "epoch": 3.7288665771484376e-05,
      "model_forward_time": 2.4292852878570557,
      "step": 9775
    },
    {
      "epoch": 3.7288665771484376e-05,
      "step": 9775,
      "training_step_time": 7.4488067626953125
    },
    {
      "epoch": 3.7288665771484376e-05,
      "model_forward_time": 2.425654411315918,
      "step": 9775
    },
    {
      "epoch": 3.7288665771484376e-05,
      "step": 9775,
      "training_step_time": 7.445305347442627
    },
    {
      "epoch": 3.7288665771484376e-05,
      "model_forward_time": 2.429518461227417,
      "step": 9775
    },
    {
      "epoch": 3.7288665771484376e-05,
      "step": 9775,
      "training_step_time": 10.285126209259033
    },
    {
      "epoch": 3.729248046875e-05,
      "model_forward_time": 2.436549425125122,
      "step": 9776
    },
    {
      "epoch": 3.729248046875e-05,
      "step": 9776,
      "training_step_time": 7.453373193740845
    },
    {
      "epoch": 3.729248046875e-05,
      "model_forward_time": 3.3966376781463623,
      "step": 9776
    },
    {
      "epoch": 3.729248046875e-05,
      "step": 9776,
      "training_step_time": 10.152205228805542
    },
    {
      "epoch": 3.729248046875e-05,
      "model_forward_time": 3.1270906925201416,
      "step": 9776
    },
    {
      "epoch": 3.729248046875e-05,
      "step": 9776,
      "training_step_time": 9.886053323745728
    },
    {
      "epoch": 3.729248046875e-05,
      "model_forward_time": 3.127774238586426,
      "step": 9776
    },
    {
      "epoch": 3.729248046875e-05,
      "step": 9776,
      "training_step_time": 9.998638391494751
    },
    {
      "epoch": 3.729629516601562e-05,
      "model_forward_time": 3.148797035217285,
      "step": 9777
    },
    {
      "epoch": 3.729629516601562e-05,
      "step": 9777,
      "training_step_time": 9.906357049942017
    },
    {
      "epoch": 3.729629516601562e-05,
      "model_forward_time": 3.129487991333008,
      "step": 9777
    },
    {
      "epoch": 3.729629516601562e-05,
      "step": 9777,
      "training_step_time": 9.890634775161743
    },
    {
      "epoch": 3.729629516601562e-05,
      "model_forward_time": 3.133720874786377,
      "step": 9777
    },
    {
      "dynamics_loss_avg": 0.04205071795731783,
      "epoch": 3.729629516601562e-05,
      "step": 9777
    },
    {
      "action_loss_avg": 0.014957919460721314,
      "epoch": 3.729629516601562e-05,
      "step": 9777
    },
    {
      "epoch": 3.729629516601562e-05,
      "step": 9777,
      "training_step_time": 10.22983694076538
    },
    {
      "epoch": 3.729629516601562e-05,
      "model_forward_time": 3.133052110671997,
      "step": 9777
    },
    {
      "epoch": 3.729629516601562e-05,
      "step": 9777,
      "training_step_time": 10.003435850143433
    },
    {
      "epoch": 3.730010986328125e-05,
      "model_forward_time": 2.446901321411133,
      "step": 9778
    },
    {
      "epoch": 3.730010986328125e-05,
      "step": 9778,
      "training_step_time": 7.471125364303589
    },
    {
      "epoch": 3.730010986328125e-05,
      "model_forward_time": 2.428492784500122,
      "step": 9778
    },
    {
      "epoch": 3.730010986328125e-05,
      "step": 9778,
      "training_step_time": 7.451937675476074
    },
    {
      "epoch": 3.730010986328125e-05,
      "model_forward_time": 2.4314937591552734,
      "step": 9778
    },
    {
      "epoch": 3.730010986328125e-05,
      "step": 9778,
      "training_step_time": 7.45361590385437
    },
    {
      "epoch": 3.730010986328125e-05,
      "model_forward_time": 3.1342856884002686,
      "step": 9778
    },
    {
      "epoch": 3.730010986328125e-05,
      "step": 9778,
      "training_step_time": 10.28381872177124
    },
    {
      "epoch": 3.7303924560546875e-05,
      "model_forward_time": 3.390939712524414,
      "step": 9779
    },
    {
      "epoch": 3.7303924560546875e-05,
      "step": 9779,
      "training_step_time": 10.155170679092407
    },
    {
      "epoch": 3.7303924560546875e-05,
      "model_forward_time": 3.1370182037353516,
      "step": 9779
    },
    {
      "epoch": 3.7303924560546875e-05,
      "step": 9779,
      "training_step_time": 9.909816026687622
    },
    {
      "epoch": 3.7303924560546875e-05,
      "model_forward_time": 3.133418321609497,
      "step": 9779
    },
    {
      "epoch": 3.7303924560546875e-05,
      "step": 9779,
      "training_step_time": 9.907040357589722
    },
    {
      "epoch": 3.7303924560546875e-05,
      "model_forward_time": 3.140169143676758,
      "step": 9779
    },
    {
      "epoch": 3.7303924560546875e-05,
      "step": 9779,
      "training_step_time": 10.264869689941406
    },
    {
      "epoch": 3.73077392578125e-05,
      "grad_norm": 0.36279889941215515,
      "learning_rate": 1e-05,
      "loss": 0.076,
      "step": 9780
    },
    {
      "epoch": 3.73077392578125e-05,
      "model_forward_time": 3.152297019958496,
      "step": 9780
    },
    {
      "dynamics_loss_avg": 0.06373766567558051,
      "epoch": 3.73077392578125e-05,
      "step": 9780
    },
    {
      "action_loss_avg": 0.02494661109521985,
      "epoch": 3.73077392578125e-05,
      "step": 9780
    },
    {
      "epoch": 3.73077392578125e-05,
      "step": 9780,
      "training_step_time": 9.932316541671753
    },
    {
      "epoch": 3.73077392578125e-05,
      "model_forward_time": 3.418766498565674,
      "step": 9780
    },
    {
      "epoch": 3.73077392578125e-05,
      "step": 9780,
      "training_step_time": 10.196693897247314
    },
    {
      "epoch": 3.73077392578125e-05,
      "model_forward_time": 3.1401684284210205,
      "step": 9780
    },
    {
      "epoch": 3.73077392578125e-05,
      "step": 9780,
      "training_step_time": 9.909447431564331
    },
    {
      "epoch": 3.73077392578125e-05,
      "model_forward_time": 2.4361655712127686,
      "step": 9780
    },
    {
      "epoch": 3.73077392578125e-05,
      "step": 9780,
      "training_step_time": 10.313190221786499
    },
    {
      "epoch": 3.731155395507813e-05,
      "model_forward_time": 2.4541141986846924,
      "step": 9781
    },
    {
      "epoch": 3.731155395507813e-05,
      "step": 9781,
      "training_step_time": 7.483267068862915
    },
    {
      "epoch": 3.731155395507813e-05,
      "model_forward_time": 2.438063859939575,
      "step": 9781
    },
    {
      "epoch": 3.731155395507813e-05,
      "step": 9781,
      "training_step_time": 7.465994596481323
    },
    {
      "epoch": 3.731155395507813e-05,
      "model_forward_time": 2.7439653873443604,
      "step": 9781
    },
    {
      "epoch": 3.731155395507813e-05,
      "step": 9781,
      "training_step_time": 7.7702319622039795
    },
    {
      "epoch": 3.731155395507813e-05,
      "model_forward_time": 2.4325263500213623,
      "step": 9781
    },
    {
      "epoch": 3.731155395507813e-05,
      "step": 9781,
      "training_step_time": 10.247747421264648
    },
    {
      "epoch": 3.731536865234375e-05,
      "model_forward_time": 3.148371458053589,
      "step": 9782
    },
    {
      "epoch": 3.731536865234375e-05,
      "step": 9782,
      "training_step_time": 9.919154644012451
    },
    {
      "epoch": 3.731536865234375e-05,
      "model_forward_time": 3.131103754043579,
      "step": 9782
    },
    {
      "epoch": 3.731536865234375e-05,
      "step": 9782,
      "training_step_time": 9.894019842147827
    },
    {
      "epoch": 3.731536865234375e-05,
      "model_forward_time": 3.136566638946533,
      "step": 9782
    },
    {
      "dynamics_loss_avg": 0.05210344642400742,
      "epoch": 3.731536865234375e-05,
      "step": 9782
    },
    {
      "action_loss_avg": 0.01691994951106608,
      "epoch": 3.731536865234375e-05,
      "step": 9782
    },
    {
      "epoch": 3.731536865234375e-05,
      "step": 9782,
      "training_step_time": 9.905720949172974
    },
    {
      "epoch": 3.731536865234375e-05,
      "model_forward_time": 3.4150421619415283,
      "step": 9782
    },
    {
      "epoch": 3.731536865234375e-05,
      "step": 9782,
      "training_step_time": 10.270913124084473
    },
    {
      "epoch": 3.731918334960937e-05,
      "model_forward_time": 3.148314952850342,
      "step": 9783
    },
    {
      "epoch": 3.731918334960937e-05,
      "step": 9783,
      "training_step_time": 9.915721893310547
    },
    {
      "epoch": 3.731918334960937e-05,
      "model_forward_time": 3.130619764328003,
      "step": 9783
    },
    {
      "epoch": 3.731918334960937e-05,
      "step": 9783,
      "training_step_time": 9.896419763565063
    },
    {
      "epoch": 3.731918334960937e-05,
      "model_forward_time": 2.4279658794403076,
      "step": 9783
    },
    {
      "epoch": 3.731918334960937e-05,
      "step": 9783,
      "training_step_time": 7.448826789855957
    },
    {
      "epoch": 3.731918334960937e-05,
      "model_forward_time": 3.1257357597351074,
      "step": 9783
    },
    {
      "epoch": 3.731918334960937e-05,
      "step": 9783,
      "training_step_time": 10.274465084075928
    },
    {
      "epoch": 3.7322998046875e-05,
      "model_forward_time": 3.411224365234375,
      "step": 9784
    },
    {
      "epoch": 3.7322998046875e-05,
      "step": 9784,
      "training_step_time": 10.172251462936401
    },
    {
      "epoch": 3.7322998046875e-05,
      "model_forward_time": 2.4311375617980957,
      "step": 9784
    },
    {
      "epoch": 3.7322998046875e-05,
      "step": 9784,
      "training_step_time": 7.450848817825317
    },
    {
      "epoch": 3.7322998046875e-05,
      "model_forward_time": 3.1335928440093994,
      "step": 9784
    },
    {
      "epoch": 3.7322998046875e-05,
      "step": 9784,
      "training_step_time": 9.899246454238892
    },
    {
      "epoch": 3.7322998046875e-05,
      "model_forward_time": 3.1286439895629883,
      "step": 9784
    },
    {
      "epoch": 3.7322998046875e-05,
      "step": 9784,
      "training_step_time": 10.32452392578125
    },
    {
      "epoch": 3.7326812744140626e-05,
      "model_forward_time": 3.146045446395874,
      "step": 9785
    },
    {
      "dynamics_loss_avg": 0.0649645134806633,
      "epoch": 3.7326812744140626e-05,
      "step": 9785
    },
    {
      "action_loss_avg": 0.02215242562815547,
      "epoch": 3.7326812744140626e-05,
      "step": 9785
    },
    {
      "epoch": 3.7326812744140626e-05,
      "step": 9785,
      "training_step_time": 9.91723918914795
    },
    {
      "epoch": 3.7326812744140626e-05,
      "model_forward_time": 3.133939504623413,
      "step": 9785
    },
    {
      "epoch": 3.7326812744140626e-05,
      "step": 9785,
      "training_step_time": 10.17625117301941
    },
    {
      "epoch": 3.7326812744140626e-05,
      "model_forward_time": 2.435863733291626,
      "step": 9785
    },
    {
      "epoch": 3.7326812744140626e-05,
      "step": 9785,
      "training_step_time": 7.464479684829712
    },
    {
      "epoch": 3.7326812744140626e-05,
      "model_forward_time": 2.4290244579315186,
      "step": 9785
    },
    {
      "epoch": 3.7326812744140626e-05,
      "step": 9785,
      "training_step_time": 10.036709070205688
    },
    {
      "epoch": 3.733062744140625e-05,
      "model_forward_time": 2.448978900909424,
      "step": 9786
    },
    {
      "epoch": 3.733062744140625e-05,
      "step": 9786,
      "training_step_time": 7.484313726425171
    },
    {
      "epoch": 3.733062744140625e-05,
      "model_forward_time": 2.4290502071380615,
      "step": 9786
    },
    {
      "epoch": 3.733062744140625e-05,
      "step": 9786,
      "training_step_time": 7.455126047134399
    },
    {
      "epoch": 3.733062744140625e-05,
      "model_forward_time": 3.128373384475708,
      "step": 9786
    },
    {
      "epoch": 3.733062744140625e-05,
      "step": 9786,
      "training_step_time": 9.901505708694458
    },
    {
      "epoch": 3.733062744140625e-05,
      "model_forward_time": 2.7169249057769775,
      "step": 9786
    },
    {
      "epoch": 3.733062744140625e-05,
      "step": 9786,
      "training_step_time": 10.28377079963684
    },
    {
      "epoch": 3.733444213867188e-05,
      "model_forward_time": 3.147268533706665,
      "step": 9787
    },
    {
      "epoch": 3.733444213867188e-05,
      "step": 9787,
      "training_step_time": 9.921733379364014
    },
    {
      "epoch": 3.733444213867188e-05,
      "model_forward_time": 3.1362862586975098,
      "step": 9787
    },
    {
      "epoch": 3.733444213867188e-05,
      "step": 9787,
      "training_step_time": 9.905054807662964
    },
    {
      "epoch": 3.733444213867188e-05,
      "model_forward_time": 3.13261079788208,
      "step": 9787
    },
    {
      "dynamics_loss_avg": 0.050801753904670476,
      "epoch": 3.733444213867188e-05,
      "step": 9787
    },
    {
      "action_loss_avg": 0.01583947162143886,
      "epoch": 3.733444213867188e-05,
      "step": 9787
    },
    {
      "epoch": 3.733444213867188e-05,
      "step": 9787,
      "training_step_time": 9.900529623031616
    },
    {
      "epoch": 3.733444213867188e-05,
      "model_forward_time": 2.4337680339813232,
      "step": 9787
    },
    {
      "epoch": 3.733444213867188e-05,
      "step": 9787,
      "training_step_time": 10.015571355819702
    },
    {
      "epoch": 3.73382568359375e-05,
      "model_forward_time": 2.7326340675354004,
      "step": 9788
    },
    {
      "epoch": 3.73382568359375e-05,
      "step": 9788,
      "training_step_time": 7.763669013977051
    },
    {
      "epoch": 3.73382568359375e-05,
      "model_forward_time": 3.1284592151641846,
      "step": 9788
    },
    {
      "epoch": 3.73382568359375e-05,
      "step": 9788,
      "training_step_time": 9.895618200302124
    },
    {
      "epoch": 3.73382568359375e-05,
      "model_forward_time": 3.1310956478118896,
      "step": 9788
    },
    {
      "epoch": 3.73382568359375e-05,
      "step": 9788,
      "training_step_time": 9.905026197433472
    },
    {
      "epoch": 3.73382568359375e-05,
      "model_forward_time": 2.4333677291870117,
      "step": 9788
    },
    {
      "epoch": 3.73382568359375e-05,
      "step": 9788,
      "training_step_time": 10.303028106689453
    },
    {
      "epoch": 3.7342071533203124e-05,
      "model_forward_time": 3.162721872329712,
      "step": 9789
    },
    {
      "epoch": 3.7342071533203124e-05,
      "step": 9789,
      "training_step_time": 9.936276197433472
    },
    {
      "epoch": 3.7342071533203124e-05,
      "model_forward_time": 3.4252753257751465,
      "step": 9789
    },
    {
      "epoch": 3.7342071533203124e-05,
      "step": 9789,
      "training_step_time": 10.197229385375977
    },
    {
      "epoch": 3.7342071533203124e-05,
      "model_forward_time": 3.140874147415161,
      "step": 9789
    },
    {
      "epoch": 3.7342071533203124e-05,
      "step": 9789,
      "training_step_time": 9.921007871627808
    },
    {
      "epoch": 3.7342071533203124e-05,
      "model_forward_time": 1.7488303184509277,
      "step": 9789
    },
    {
      "epoch": 3.7342071533203124e-05,
      "step": 9789,
      "training_step_time": 10.295238256454468
    },
    {
      "epoch": 3.734588623046875e-05,
      "grad_norm": 0.4208700656890869,
      "learning_rate": 1e-05,
      "loss": 0.0784,
      "step": 9790
    },
    {
      "epoch": 3.734588623046875e-05,
      "model_forward_time": 3.1594104766845703,
      "step": 9790
    },
    {
      "dynamics_loss_avg": 0.04576461771503091,
      "epoch": 3.734588623046875e-05,
      "step": 9790
    },
    {
      "action_loss_avg": 0.014515640982426702,
      "epoch": 3.734588623046875e-05,
      "step": 9790
    },
    {
      "epoch": 3.734588623046875e-05,
      "step": 9790,
      "training_step_time": 9.9383544921875
    },
    {
      "epoch": 3.734588623046875e-05,
      "model_forward_time": 2.4384069442749023,
      "step": 9790
    },
    {
      "epoch": 3.734588623046875e-05,
      "step": 9790,
      "training_step_time": 7.472478628158569
    },
    {
      "epoch": 3.734588623046875e-05,
      "model_forward_time": 3.1398439407348633,
      "step": 9790
    },
    {
      "epoch": 3.734588623046875e-05,
      "step": 9790,
      "training_step_time": 9.907526969909668
    },
    {
      "epoch": 3.734588623046875e-05,
      "model_forward_time": 3.4577229022979736,
      "step": 9790
    },
    {
      "epoch": 3.734588623046875e-05,
      "step": 9790,
      "training_step_time": 10.324616193771362
    },
    {
      "epoch": 3.7349700927734376e-05,
      "model_forward_time": 2.450565814971924,
      "step": 9791
    },
    {
      "epoch": 3.7349700927734376e-05,
      "step": 9791,
      "training_step_time": 7.488591909408569
    },
    {
      "epoch": 3.7349700927734376e-05,
      "model_forward_time": 3.138162851333618,
      "step": 9791
    },
    {
      "epoch": 3.7349700927734376e-05,
      "step": 9791,
      "training_step_time": 9.918522834777832
    },
    {
      "epoch": 3.7349700927734376e-05,
      "model_forward_time": 3.1474411487579346,
      "step": 9791
    },
    {
      "epoch": 3.7349700927734376e-05,
      "step": 9791,
      "training_step_time": 9.927147388458252
    },
    {
      "epoch": 3.7349700927734376e-05,
      "model_forward_time": 2.4387998580932617,
      "step": 9791
    },
    {
      "epoch": 3.7349700927734376e-05,
      "step": 9791,
      "training_step_time": 10.27826189994812
    },
    {
      "epoch": 3.7353515625e-05,
      "model_forward_time": 2.7834794521331787,
      "step": 9792
    },
    {
      "epoch": 3.7353515625e-05,
      "step": 9792,
      "training_step_time": 7.81969952583313
    },
    {
      "epoch": 3.7353515625e-05,
      "model_forward_time": 3.1404519081115723,
      "step": 9792
    },
    {
      "epoch": 3.7353515625e-05,
      "step": 9792,
      "training_step_time": 9.916188716888428
    },
    {
      "epoch": 3.7353515625e-05,
      "model_forward_time": 2.436216354370117,
      "step": 9792
    },
    {
      "dynamics_loss_avg": 0.05784154124557972,
      "epoch": 3.7353515625e-05,
      "step": 9792
    },
    {
      "action_loss_avg": 0.02058523055166006,
      "epoch": 3.7353515625e-05,
      "step": 9792
    },
    {
      "epoch": 3.7353515625e-05,
      "step": 9792,
      "training_step_time": 7.465129375457764
    },
    {
      "epoch": 3.7353515625e-05,
      "model_forward_time": 3.135896682739258,
      "step": 9792
    },
    {
      "epoch": 3.7353515625e-05,
      "step": 9792,
      "training_step_time": 10.324043273925781
    },
    {
      "epoch": 3.735733032226562e-05,
      "model_forward_time": 3.147693157196045,
      "step": 9793
    },
    {
      "epoch": 3.735733032226562e-05,
      "step": 9793,
      "training_step_time": 9.928483963012695
    },
    {
      "epoch": 3.735733032226562e-05,
      "model_forward_time": 3.434340476989746,
      "step": 9793
    },
    {
      "epoch": 3.735733032226562e-05,
      "step": 9793,
      "training_step_time": 10.206283569335938
    },
    {
      "epoch": 3.735733032226562e-05,
      "model_forward_time": 3.1414902210235596,
      "step": 9793
    },
    {
      "epoch": 3.735733032226562e-05,
      "step": 9793,
      "training_step_time": 9.91776418685913
    },
    {
      "epoch": 3.735733032226562e-05,
      "model_forward_time": 2.438101053237915,
      "step": 9793
    },
    {
      "epoch": 3.735733032226562e-05,
      "step": 9793,
      "training_step_time": 10.23915958404541
    },
    {
      "epoch": 3.736114501953125e-05,
      "model_forward_time": 3.1434335708618164,
      "step": 9794
    },
    {
      "epoch": 3.736114501953125e-05,
      "step": 9794,
      "training_step_time": 9.916748762130737
    },
    {
      "epoch": 3.736114501953125e-05,
      "model_forward_time": 3.1411120891571045,
      "step": 9794
    },
    {
      "epoch": 3.736114501953125e-05,
      "step": 9794,
      "training_step_time": 9.921740293502808
    },
    {
      "epoch": 3.736114501953125e-05,
      "model_forward_time": 3.429058313369751,
      "step": 9794
    },
    {
      "epoch": 3.736114501953125e-05,
      "step": 9794,
      "training_step_time": 10.206970453262329
    },
    {
      "epoch": 3.736114501953125e-05,
      "model_forward_time": 3.142866373062134,
      "step": 9794
    },
    {
      "epoch": 3.736114501953125e-05,
      "step": 9794,
      "training_step_time": 10.290231704711914
    },
    {
      "epoch": 3.7364959716796875e-05,
      "model_forward_time": 3.160926103591919,
      "step": 9795
    },
    {
      "dynamics_loss_avg": 0.0545557577162981,
      "epoch": 3.7364959716796875e-05,
      "step": 9795
    },
    {
      "action_loss_avg": 0.02653964392375201,
      "epoch": 3.7364959716796875e-05,
      "step": 9795
    },
    {
      "epoch": 3.7364959716796875e-05,
      "step": 9795,
      "training_step_time": 9.947676420211792
    },
    {
      "epoch": 3.7364959716796875e-05,
      "model_forward_time": 3.1481502056121826,
      "step": 9795
    },
    {
      "epoch": 3.7364959716796875e-05,
      "step": 9795,
      "training_step_time": 9.91822338104248
    },
    {
      "epoch": 3.7364959716796875e-05,
      "model_forward_time": 3.140211343765259,
      "step": 9795
    },
    {
      "epoch": 3.7364959716796875e-05,
      "step": 9795,
      "training_step_time": 9.909823417663574
    },
    {
      "epoch": 3.7364959716796875e-05,
      "model_forward_time": 2.750915765762329,
      "step": 9795
    },
    {
      "epoch": 3.7364959716796875e-05,
      "step": 9795,
      "training_step_time": 10.273979663848877
    },
    {
      "epoch": 3.73687744140625e-05,
      "model_forward_time": 3.1657357215881348,
      "step": 9796
    },
    {
      "epoch": 3.73687744140625e-05,
      "step": 9796,
      "training_step_time": 9.940099954605103
    },
    {
      "epoch": 3.73687744140625e-05,
      "model_forward_time": 2.4353511333465576,
      "step": 9796
    },
    {
      "epoch": 3.73687744140625e-05,
      "step": 9796,
      "training_step_time": 7.46294641494751
    },
    {
      "epoch": 3.73687744140625e-05,
      "model_forward_time": 1.745222568511963,
      "step": 9796
    },
    {
      "epoch": 3.73687744140625e-05,
      "step": 9796,
      "training_step_time": 5.204402208328247
    },
    {
      "epoch": 3.73687744140625e-05,
      "model_forward_time": 3.1377363204956055,
      "step": 9796
    },
    {
      "epoch": 3.73687744140625e-05,
      "step": 9796,
      "training_step_time": 10.288962602615356
    },
    {
      "epoch": 3.737258911132813e-05,
      "model_forward_time": 2.7570133209228516,
      "step": 9797
    },
    {
      "epoch": 3.737258911132813e-05,
      "step": 9797,
      "training_step_time": 7.800917625427246
    },
    {
      "epoch": 3.737258911132813e-05,
      "model_forward_time": 3.1352365016937256,
      "step": 9797
    },
    {
      "epoch": 3.737258911132813e-05,
      "step": 9797,
      "training_step_time": 9.909105777740479
    },
    {
      "epoch": 3.737258911132813e-05,
      "model_forward_time": 3.1388282775878906,
      "step": 9797
    },
    {
      "dynamics_loss_avg": 0.03668547570705414,
      "epoch": 3.737258911132813e-05,
      "step": 9797
    },
    {
      "action_loss_avg": 0.011585205909796059,
      "epoch": 3.737258911132813e-05,
      "step": 9797
    },
    {
      "epoch": 3.737258911132813e-05,
      "step": 9797,
      "training_step_time": 9.916255235671997
    },
    {
      "epoch": 3.737258911132813e-05,
      "model_forward_time": 2.438732862472534,
      "step": 9797
    },
    {
      "epoch": 3.737258911132813e-05,
      "step": 9797,
      "training_step_time": 10.307605743408203
    },
    {
      "epoch": 3.737640380859375e-05,
      "model_forward_time": 2.450040578842163,
      "step": 9798
    },
    {
      "epoch": 3.737640380859375e-05,
      "step": 9798,
      "training_step_time": 7.4825358390808105
    },
    {
      "epoch": 3.737640380859375e-05,
      "model_forward_time": 3.133518934249878,
      "step": 9798
    },
    {
      "epoch": 3.737640380859375e-05,
      "step": 9798,
      "training_step_time": 9.907371520996094
    },
    {
      "epoch": 3.737640380859375e-05,
      "model_forward_time": 2.7179527282714844,
      "step": 9798
    },
    {
      "epoch": 3.737640380859375e-05,
      "step": 9798,
      "training_step_time": 7.745617628097534
    },
    {
      "epoch": 3.737640380859375e-05,
      "model_forward_time": 2.431755542755127,
      "step": 9798
    },
    {
      "epoch": 3.737640380859375e-05,
      "step": 9798,
      "training_step_time": 10.302473545074463
    },
    {
      "epoch": 3.738021850585937e-05,
      "model_forward_time": 3.149362325668335,
      "step": 9799
    },
    {
      "epoch": 3.738021850585937e-05,
      "step": 9799,
      "training_step_time": 9.926072835922241
    },
    {
      "epoch": 3.738021850585937e-05,
      "model_forward_time": 3.1369729042053223,
      "step": 9799
    },
    {
      "epoch": 3.738021850585937e-05,
      "step": 9799,
      "training_step_time": 9.913059949874878
    },
    {
      "epoch": 3.738021850585937e-05,
      "model_forward_time": 3.1426002979278564,
      "step": 9799
    },
    {
      "epoch": 3.738021850585937e-05,
      "step": 9799,
      "training_step_time": 9.923898220062256
    },
    {
      "epoch": 3.738021850585937e-05,
      "model_forward_time": 3.4571595191955566,
      "step": 9799
    },
    {
      "epoch": 3.738021850585937e-05,
      "step": 9799,
      "training_step_time": 10.329294204711914
    },
    {
      "epoch": 3.7384033203125e-05,
      "grad_norm": 0.2509705424308777,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9800
    },
    {
      "epoch": 3.7384033203125e-05,
      "model_forward_time": 2.4524922370910645,
      "step": 9800
    },
    {
      "dynamics_loss_avg": 0.04332505241036415,
      "epoch": 3.7384033203125e-05,
      "step": 9800
    },
    {
      "action_loss_avg": 0.01170029486529529,
      "epoch": 3.7384033203125e-05,
      "step": 9800
    },
    {
      "epoch": 3.7384033203125e-05,
      "step": 9800,
      "training_step_time": 7.490327835083008
    },
    {
      "epoch": 3.7384033203125e-05,
      "model_forward_time": 3.1374523639678955,
      "step": 9800
    },
    {
      "epoch": 3.7384033203125e-05,
      "step": 9800,
      "training_step_time": 9.91342306137085
    },
    {
      "epoch": 3.7384033203125e-05,
      "model_forward_time": 3.1431705951690674,
      "step": 9800
    },
    {
      "epoch": 3.7384033203125e-05,
      "step": 9800,
      "training_step_time": 9.917622327804565
    },
    {
      "epoch": 3.7384033203125e-05,
      "model_forward_time": 3.14172101020813,
      "step": 9800
    },
    {
      "epoch": 3.7384033203125e-05,
      "step": 9800,
      "training_step_time": 10.273524522781372
    },
    {
      "epoch": 3.7387847900390626e-05,
      "model_forward_time": 2.7505130767822266,
      "step": 9801
    },
    {
      "epoch": 3.7387847900390626e-05,
      "step": 9801,
      "training_step_time": 7.79529881477356
    },
    {
      "epoch": 3.7387847900390626e-05,
      "model_forward_time": 2.435964584350586,
      "step": 9801
    },
    {
      "epoch": 3.7387847900390626e-05,
      "step": 9801,
      "training_step_time": 7.468212604522705
    },
    {
      "epoch": 3.7387847900390626e-05,
      "model_forward_time": 2.435884475708008,
      "step": 9801
    },
    {
      "epoch": 3.7387847900390626e-05,
      "step": 9801,
      "training_step_time": 7.466224431991577
    },
    {
      "epoch": 3.7387847900390626e-05,
      "model_forward_time": 2.438804864883423,
      "step": 9801
    },
    {
      "epoch": 3.7387847900390626e-05,
      "step": 9801,
      "training_step_time": 10.318856239318848
    },
    {
      "epoch": 3.739166259765625e-05,
      "model_forward_time": 3.157165050506592,
      "step": 9802
    },
    {
      "epoch": 3.739166259765625e-05,
      "step": 9802,
      "training_step_time": 9.933645248413086
    },
    {
      "epoch": 3.739166259765625e-05,
      "model_forward_time": 2.7609894275665283,
      "step": 9802
    },
    {
      "epoch": 3.739166259765625e-05,
      "step": 9802,
      "training_step_time": 7.785450458526611
    },
    {
      "epoch": 3.739166259765625e-05,
      "model_forward_time": 3.146411657333374,
      "step": 9802
    },
    {
      "dynamics_loss_avg": 0.07056104950606823,
      "epoch": 3.739166259765625e-05,
      "step": 9802
    },
    {
      "action_loss_avg": 0.021900286618620157,
      "epoch": 3.739166259765625e-05,
      "step": 9802
    },
    {
      "epoch": 3.739166259765625e-05,
      "step": 9802,
      "training_step_time": 9.929142951965332
    },
    {
      "epoch": 3.739166259765625e-05,
      "model_forward_time": 2.4358606338500977,
      "step": 9802
    },
    {
      "epoch": 3.739166259765625e-05,
      "step": 9802,
      "training_step_time": 10.01768708229065
    },
    {
      "epoch": 3.739547729492188e-05,
      "model_forward_time": 2.453700542449951,
      "step": 9803
    },
    {
      "epoch": 3.739547729492188e-05,
      "step": 9803,
      "training_step_time": 7.481867551803589
    },
    {
      "epoch": 3.739547729492188e-05,
      "model_forward_time": 3.142191171646118,
      "step": 9803
    },
    {
      "epoch": 3.739547729492188e-05,
      "step": 9803,
      "training_step_time": 9.91371488571167
    },
    {
      "epoch": 3.739547729492188e-05,
      "model_forward_time": 3.4691646099090576,
      "step": 9803
    },
    {
      "epoch": 3.739547729492188e-05,
      "step": 9803,
      "training_step_time": 10.241371870040894
    },
    {
      "epoch": 3.739547729492188e-05,
      "model_forward_time": 2.436450242996216,
      "step": 9803
    },
    {
      "epoch": 3.739547729492188e-05,
      "step": 9803,
      "training_step_time": 10.293887853622437
    },
    {
      "epoch": 3.73992919921875e-05,
      "model_forward_time": 3.155823230743408,
      "step": 9804
    },
    {
      "epoch": 3.73992919921875e-05,
      "step": 9804,
      "training_step_time": 9.931433200836182
    },
    {
      "epoch": 3.73992919921875e-05,
      "model_forward_time": 3.1448235511779785,
      "step": 9804
    },
    {
      "epoch": 3.73992919921875e-05,
      "step": 9804,
      "training_step_time": 9.94750452041626
    },
    {
      "epoch": 3.73992919921875e-05,
      "model_forward_time": 3.1426684856414795,
      "step": 9804
    },
    {
      "epoch": 3.73992919921875e-05,
      "step": 9804,
      "training_step_time": 9.922621250152588
    },
    {
      "epoch": 3.73992919921875e-05,
      "model_forward_time": 3.455148696899414,
      "step": 9804
    },
    {
      "epoch": 3.73992919921875e-05,
      "step": 9804,
      "training_step_time": 10.330462455749512
    },
    {
      "epoch": 3.7403106689453124e-05,
      "model_forward_time": 2.453348398208618,
      "step": 9805
    },
    {
      "dynamics_loss_avg": 0.05892626252025366,
      "epoch": 3.7403106689453124e-05,
      "step": 9805
    },
    {
      "action_loss_avg": 0.02175838123075664,
      "epoch": 3.7403106689453124e-05,
      "step": 9805
    },
    {
      "epoch": 3.7403106689453124e-05,
      "step": 9805,
      "training_step_time": 7.4992570877075195
    },
    {
      "epoch": 3.7403106689453124e-05,
      "model_forward_time": 2.4382166862487793,
      "step": 9805
    },
    {
      "epoch": 3.7403106689453124e-05,
      "step": 9805,
      "training_step_time": 7.479955196380615
    },
    {
      "epoch": 3.7403106689453124e-05,
      "model_forward_time": 3.1405699253082275,
      "step": 9805
    },
    {
      "epoch": 3.7403106689453124e-05,
      "step": 9805,
      "training_step_time": 9.917956113815308
    },
    {
      "epoch": 3.7403106689453124e-05,
      "model_forward_time": 3.141141653060913,
      "step": 9805
    },
    {
      "epoch": 3.7403106689453124e-05,
      "step": 9805,
      "training_step_time": 10.310483455657959
    },
    {
      "epoch": 3.740692138671875e-05,
      "model_forward_time": 2.457111120223999,
      "step": 9806
    },
    {
      "epoch": 3.740692138671875e-05,
      "step": 9806,
      "training_step_time": 7.826023101806641
    },
    {
      "epoch": 3.740692138671875e-05,
      "model_forward_time": 3.1428616046905518,
      "step": 9806
    },
    {
      "epoch": 3.740692138671875e-05,
      "step": 9806,
      "training_step_time": 9.917446613311768
    },
    {
      "epoch": 3.740692138671875e-05,
      "model_forward_time": 2.4334404468536377,
      "step": 9806
    },
    {
      "epoch": 3.740692138671875e-05,
      "step": 9806,
      "training_step_time": 7.45794153213501
    },
    {
      "epoch": 3.740692138671875e-05,
      "model_forward_time": 2.4356515407562256,
      "step": 9806
    },
    {
      "epoch": 3.740692138671875e-05,
      "step": 9806,
      "training_step_time": 10.312562704086304
    },
    {
      "epoch": 3.7410736083984376e-05,
      "model_forward_time": 3.1586787700653076,
      "step": 9807
    },
    {
      "epoch": 3.7410736083984376e-05,
      "step": 9807,
      "training_step_time": 9.92894172668457
    },
    {
      "epoch": 3.7410736083984376e-05,
      "model_forward_time": 3.4340603351593018,
      "step": 9807
    },
    {
      "epoch": 3.7410736083984376e-05,
      "step": 9807,
      "training_step_time": 10.20402216911316
    },
    {
      "epoch": 3.7410736083984376e-05,
      "model_forward_time": 3.1395206451416016,
      "step": 9807
    },
    {
      "dynamics_loss_avg": 0.05161949703469872,
      "epoch": 3.7410736083984376e-05,
      "step": 9807
    },
    {
      "action_loss_avg": 0.023746166052296756,
      "epoch": 3.7410736083984376e-05,
      "step": 9807
    },
    {
      "epoch": 3.7410736083984376e-05,
      "step": 9807,
      "training_step_time": 9.91571593284607
    },
    {
      "epoch": 3.7410736083984376e-05,
      "model_forward_time": 3.1409759521484375,
      "step": 9807
    },
    {
      "epoch": 3.7410736083984376e-05,
      "step": 9807,
      "training_step_time": 10.010809183120728
    },
    {
      "epoch": 3.741455078125e-05,
      "model_forward_time": 3.153688669204712,
      "step": 9808
    },
    {
      "epoch": 3.741455078125e-05,
      "step": 9808,
      "training_step_time": 9.935899257659912
    },
    {
      "epoch": 3.741455078125e-05,
      "model_forward_time": 3.1318182945251465,
      "step": 9808
    },
    {
      "epoch": 3.741455078125e-05,
      "step": 9808,
      "training_step_time": 9.913835048675537
    },
    {
      "epoch": 3.741455078125e-05,
      "model_forward_time": 3.1415576934814453,
      "step": 9808
    },
    {
      "epoch": 3.741455078125e-05,
      "step": 9808,
      "training_step_time": 9.917906045913696
    },
    {
      "epoch": 3.741455078125e-05,
      "model_forward_time": 2.7555112838745117,
      "step": 9808
    },
    {
      "epoch": 3.741455078125e-05,
      "step": 9808,
      "training_step_time": 10.317337036132812
    },
    {
      "epoch": 3.741836547851562e-05,
      "model_forward_time": 3.148493528366089,
      "step": 9809
    },
    {
      "epoch": 3.741836547851562e-05,
      "step": 9809,
      "training_step_time": 9.924713850021362
    },
    {
      "epoch": 3.741836547851562e-05,
      "model_forward_time": 2.43231463432312,
      "step": 9809
    },
    {
      "epoch": 3.741836547851562e-05,
      "step": 9809,
      "training_step_time": 7.459322690963745
    },
    {
      "epoch": 3.741836547851562e-05,
      "model_forward_time": 3.1366493701934814,
      "step": 9809
    },
    {
      "epoch": 3.741836547851562e-05,
      "step": 9809,
      "training_step_time": 9.90831446647644
    },
    {
      "epoch": 3.741836547851562e-05,
      "model_forward_time": 3.1341700553894043,
      "step": 9809
    },
    {
      "epoch": 3.741836547851562e-05,
      "step": 9809,
      "training_step_time": 10.306808948516846
    },
    {
      "epoch": 3.742218017578125e-05,
      "grad_norm": 0.22765780985355377,
      "learning_rate": 1e-05,
      "loss": 0.0781,
      "step": 9810
    },
    {
      "epoch": 3.742218017578125e-05,
      "model_forward_time": 3.152341604232788,
      "step": 9810
    },
    {
      "dynamics_loss_avg": 0.04982788544148207,
      "epoch": 3.742218017578125e-05,
      "step": 9810
    },
    {
      "action_loss_avg": 0.02114889845252037,
      "epoch": 3.742218017578125e-05,
      "step": 9810
    },
    {
      "epoch": 3.742218017578125e-05,
      "step": 9810,
      "training_step_time": 9.930074691772461
    },
    {
      "epoch": 3.742218017578125e-05,
      "model_forward_time": 3.133284568786621,
      "step": 9810
    },
    {
      "epoch": 3.742218017578125e-05,
      "step": 9810,
      "training_step_time": 9.909410953521729
    },
    {
      "epoch": 3.742218017578125e-05,
      "model_forward_time": 3.139164686203003,
      "step": 9810
    },
    {
      "epoch": 3.742218017578125e-05,
      "step": 9810,
      "training_step_time": 9.9207022190094
    },
    {
      "epoch": 3.742218017578125e-05,
      "model_forward_time": 2.4375340938568115,
      "step": 9810
    },
    {
      "epoch": 3.742218017578125e-05,
      "step": 9810,
      "training_step_time": 10.315934896469116
    },
    {
      "epoch": 3.7425994873046875e-05,
      "model_forward_time": 2.4446723461151123,
      "step": 9811
    },
    {
      "epoch": 3.7425994873046875e-05,
      "step": 9811,
      "training_step_time": 7.481175661087036
    },
    {
      "epoch": 3.7425994873046875e-05,
      "model_forward_time": 3.441762924194336,
      "step": 9811
    },
    {
      "epoch": 3.7425994873046875e-05,
      "step": 9811,
      "training_step_time": 10.219603061676025
    },
    {
      "epoch": 3.7425994873046875e-05,
      "model_forward_time": 3.1445932388305664,
      "step": 9811
    },
    {
      "epoch": 3.7425994873046875e-05,
      "step": 9811,
      "training_step_time": 9.936367750167847
    },
    {
      "epoch": 3.7425994873046875e-05,
      "model_forward_time": 3.1413962841033936,
      "step": 9811
    },
    {
      "epoch": 3.7425994873046875e-05,
      "step": 9811,
      "training_step_time": 10.306615114212036
    },
    {
      "epoch": 3.74298095703125e-05,
      "model_forward_time": 3.167065382003784,
      "step": 9812
    },
    {
      "epoch": 3.74298095703125e-05,
      "step": 9812,
      "training_step_time": 9.953932762145996
    },
    {
      "epoch": 3.74298095703125e-05,
      "model_forward_time": 2.441953659057617,
      "step": 9812
    },
    {
      "epoch": 3.74298095703125e-05,
      "step": 9812,
      "training_step_time": 7.480358362197876
    },
    {
      "epoch": 3.74298095703125e-05,
      "model_forward_time": 2.6989238262176514,
      "step": 9812
    },
    {
      "dynamics_loss_avg": 0.05636274851858616,
      "epoch": 3.74298095703125e-05,
      "step": 9812
    },
    {
      "action_loss_avg": 0.021669870149344204,
      "epoch": 3.74298095703125e-05,
      "step": 9812
    },
    {
      "epoch": 3.74298095703125e-05,
      "step": 9812,
      "training_step_time": 7.7360734939575195
    },
    {
      "epoch": 3.74298095703125e-05,
      "model_forward_time": 3.1436142921447754,
      "step": 9812
    },
    {
      "epoch": 3.74298095703125e-05,
      "step": 9812,
      "training_step_time": 10.03547215461731
    },
    {
      "epoch": 3.743362426757813e-05,
      "model_forward_time": 3.1616837978363037,
      "step": 9813
    },
    {
      "epoch": 3.743362426757813e-05,
      "step": 9813,
      "training_step_time": 9.948160171508789
    },
    {
      "epoch": 3.743362426757813e-05,
      "model_forward_time": 3.1448135375976562,
      "step": 9813
    },
    {
      "epoch": 3.743362426757813e-05,
      "step": 9813,
      "training_step_time": 9.92773151397705
    },
    {
      "epoch": 3.743362426757813e-05,
      "model_forward_time": 3.14670467376709,
      "step": 9813
    },
    {
      "epoch": 3.743362426757813e-05,
      "step": 9813,
      "training_step_time": 9.936992645263672
    },
    {
      "epoch": 3.743362426757813e-05,
      "model_forward_time": 2.0205366611480713,
      "step": 9813
    },
    {
      "epoch": 3.743362426757813e-05,
      "step": 9813,
      "training_step_time": 10.030563831329346
    },
    {
      "epoch": 3.743743896484375e-05,
      "model_forward_time": 3.152195930480957,
      "step": 9814
    },
    {
      "epoch": 3.743743896484375e-05,
      "step": 9814,
      "training_step_time": 9.938628673553467
    },
    {
      "epoch": 3.743743896484375e-05,
      "model_forward_time": 3.148691177368164,
      "step": 9814
    },
    {
      "epoch": 3.743743896484375e-05,
      "step": 9814,
      "training_step_time": 9.933706045150757
    },
    {
      "epoch": 3.743743896484375e-05,
      "model_forward_time": 2.442128896713257,
      "step": 9814
    },
    {
      "epoch": 3.743743896484375e-05,
      "step": 9814,
      "training_step_time": 7.47918176651001
    },
    {
      "epoch": 3.743743896484375e-05,
      "model_forward_time": 2.442894220352173,
      "step": 9814
    },
    {
      "epoch": 3.743743896484375e-05,
      "step": 9814,
      "training_step_time": 10.293144226074219
    },
    {
      "epoch": 3.744125366210937e-05,
      "model_forward_time": 2.7161900997161865,
      "step": 9815
    },
    {
      "dynamics_loss_avg": 0.04107961636036635,
      "epoch": 3.744125366210937e-05,
      "step": 9815
    },
    {
      "action_loss_avg": 0.014581468189135194,
      "epoch": 3.744125366210937e-05,
      "step": 9815
    },
    {
      "epoch": 3.744125366210937e-05,
      "step": 9815,
      "training_step_time": 7.756169557571411
    },
    {
      "epoch": 3.744125366210937e-05,
      "model_forward_time": 3.1462855339050293,
      "step": 9815
    },
    {
      "epoch": 3.744125366210937e-05,
      "step": 9815,
      "training_step_time": 9.929812669754028
    },
    {
      "epoch": 3.744125366210937e-05,
      "model_forward_time": 3.143890380859375,
      "step": 9815
    },
    {
      "epoch": 3.744125366210937e-05,
      "step": 9815,
      "training_step_time": 9.92963457107544
    },
    {
      "epoch": 3.744125366210937e-05,
      "model_forward_time": 3.151231527328491,
      "step": 9815
    },
    {
      "epoch": 3.744125366210937e-05,
      "step": 9815,
      "training_step_time": 10.293696403503418
    },
    {
      "epoch": 3.7445068359375e-05,
      "model_forward_time": 2.4536478519439697,
      "step": 9816
    },
    {
      "epoch": 3.7445068359375e-05,
      "step": 9816,
      "training_step_time": 7.493591785430908
    },
    {
      "epoch": 3.7445068359375e-05,
      "model_forward_time": 3.4604671001434326,
      "step": 9816
    },
    {
      "epoch": 3.7445068359375e-05,
      "step": 9816,
      "training_step_time": 10.241623640060425
    },
    {
      "epoch": 3.7445068359375e-05,
      "model_forward_time": 3.1450917720794678,
      "step": 9816
    },
    {
      "epoch": 3.7445068359375e-05,
      "step": 9816,
      "training_step_time": 9.925880432128906
    },
    {
      "epoch": 3.7445068359375e-05,
      "model_forward_time": 2.4405109882354736,
      "step": 9816
    },
    {
      "epoch": 3.7445068359375e-05,
      "step": 9816,
      "training_step_time": 10.258690357208252
    },
    {
      "epoch": 3.7448883056640626e-05,
      "model_forward_time": 3.1597394943237305,
      "step": 9817
    },
    {
      "epoch": 3.7448883056640626e-05,
      "step": 9817,
      "training_step_time": 9.945640563964844
    },
    {
      "epoch": 3.7448883056640626e-05,
      "model_forward_time": 2.438591241836548,
      "step": 9817
    },
    {
      "epoch": 3.7448883056640626e-05,
      "step": 9817,
      "training_step_time": 7.473723411560059
    },
    {
      "epoch": 3.7448883056640626e-05,
      "model_forward_time": 2.4418883323669434,
      "step": 9817
    },
    {
      "dynamics_loss_avg": 0.048529403004795316,
      "epoch": 3.7448883056640626e-05,
      "step": 9817
    },
    {
      "action_loss_avg": 0.01374480437953025,
      "epoch": 3.7448883056640626e-05,
      "step": 9817
    },
    {
      "epoch": 3.7448883056640626e-05,
      "step": 9817,
      "training_step_time": 7.480891466140747
    },
    {
      "epoch": 3.7448883056640626e-05,
      "model_forward_time": 3.430347442626953,
      "step": 9817
    },
    {
      "epoch": 3.7448883056640626e-05,
      "step": 9817,
      "training_step_time": 10.335732460021973
    },
    {
      "epoch": 3.745269775390625e-05,
      "model_forward_time": 2.457975149154663,
      "step": 9818
    },
    {
      "epoch": 3.745269775390625e-05,
      "step": 9818,
      "training_step_time": 7.498860836029053
    },
    {
      "epoch": 3.745269775390625e-05,
      "model_forward_time": 2.4367833137512207,
      "step": 9818
    },
    {
      "epoch": 3.745269775390625e-05,
      "step": 9818,
      "training_step_time": 7.473735094070435
    },
    {
      "epoch": 3.745269775390625e-05,
      "model_forward_time": 2.4385077953338623,
      "step": 9818
    },
    {
      "epoch": 3.745269775390625e-05,
      "step": 9818,
      "training_step_time": 7.4784464836120605
    },
    {
      "epoch": 3.745269775390625e-05,
      "model_forward_time": 2.4380404949188232,
      "step": 9818
    },
    {
      "epoch": 3.745269775390625e-05,
      "step": 9818,
      "training_step_time": 10.035969734191895
    },
    {
      "epoch": 3.745651245117188e-05,
      "model_forward_time": 2.7148900032043457,
      "step": 9819
    },
    {
      "epoch": 3.745651245117188e-05,
      "step": 9819,
      "training_step_time": 7.7493062019348145
    },
    {
      "epoch": 3.745651245117188e-05,
      "model_forward_time": 3.13960862159729,
      "step": 9819
    },
    {
      "epoch": 3.745651245117188e-05,
      "step": 9819,
      "training_step_time": 9.914376020431519
    },
    {
      "epoch": 3.745651245117188e-05,
      "model_forward_time": 2.4357612133026123,
      "step": 9819
    },
    {
      "epoch": 3.745651245117188e-05,
      "step": 9819,
      "training_step_time": 7.464646100997925
    },
    {
      "epoch": 3.745651245117188e-05,
      "model_forward_time": 2.436370611190796,
      "step": 9819
    },
    {
      "epoch": 3.745651245117188e-05,
      "step": 9819,
      "training_step_time": 10.284330606460571
    },
    {
      "epoch": 3.74603271484375e-05,
      "grad_norm": 0.3303239941596985,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 9820
    },
    {
      "epoch": 3.74603271484375e-05,
      "model_forward_time": 3.156959056854248,
      "step": 9820
    },
    {
      "dynamics_loss_avg": 0.05492766965180636,
      "epoch": 3.74603271484375e-05,
      "step": 9820
    },
    {
      "action_loss_avg": 0.01656426810659468,
      "epoch": 3.74603271484375e-05,
      "step": 9820
    },
    {
      "epoch": 3.74603271484375e-05,
      "step": 9820,
      "training_step_time": 9.93136477470398
    },
    {
      "epoch": 3.74603271484375e-05,
      "model_forward_time": 3.400080680847168,
      "step": 9820
    },
    {
      "epoch": 3.74603271484375e-05,
      "step": 9820,
      "training_step_time": 10.168226957321167
    },
    {
      "epoch": 3.74603271484375e-05,
      "model_forward_time": 2.4354636669158936,
      "step": 9820
    },
    {
      "epoch": 3.74603271484375e-05,
      "step": 9820,
      "training_step_time": 7.45920205116272
    },
    {
      "epoch": 3.74603271484375e-05,
      "model_forward_time": 3.1310958862304688,
      "step": 9820
    },
    {
      "epoch": 3.74603271484375e-05,
      "step": 9820,
      "training_step_time": 10.005380392074585
    },
    {
      "epoch": 3.7464141845703124e-05,
      "model_forward_time": 3.14994215965271,
      "step": 9821
    },
    {
      "epoch": 3.7464141845703124e-05,
      "step": 9821,
      "training_step_time": 9.92287278175354
    },
    {
      "epoch": 3.7464141845703124e-05,
      "model_forward_time": 2.4353113174438477,
      "step": 9821
    },
    {
      "epoch": 3.7464141845703124e-05,
      "step": 9821,
      "training_step_time": 7.461359262466431
    },
    {
      "epoch": 3.7464141845703124e-05,
      "model_forward_time": 1.7454910278320312,
      "step": 9821
    },
    {
      "epoch": 3.7464141845703124e-05,
      "step": 9821,
      "training_step_time": 5.1975836753845215
    },
    {
      "epoch": 3.7464141845703124e-05,
      "model_forward_time": 3.3932249546051025,
      "step": 9821
    },
    {
      "epoch": 3.7464141845703124e-05,
      "step": 9821,
      "training_step_time": 10.295886993408203
    },
    {
      "epoch": 3.746795654296875e-05,
      "model_forward_time": 2.4528865814208984,
      "step": 9822
    },
    {
      "epoch": 3.746795654296875e-05,
      "step": 9822,
      "training_step_time": 7.480403423309326
    },
    {
      "epoch": 3.746795654296875e-05,
      "model_forward_time": 2.436347246170044,
      "step": 9822
    },
    {
      "epoch": 3.746795654296875e-05,
      "step": 9822,
      "training_step_time": 7.471623659133911
    },
    {
      "epoch": 3.746795654296875e-05,
      "model_forward_time": 3.140719175338745,
      "step": 9822
    },
    {
      "dynamics_loss_avg": 0.05686618117615581,
      "epoch": 3.746795654296875e-05,
      "step": 9822
    },
    {
      "action_loss_avg": 0.020845092582749202,
      "epoch": 3.746795654296875e-05,
      "step": 9822
    },
    {
      "epoch": 3.746795654296875e-05,
      "step": 9822,
      "training_step_time": 9.920072078704834
    },
    {
      "epoch": 3.746795654296875e-05,
      "model_forward_time": 3.147691011428833,
      "step": 9822
    },
    {
      "epoch": 3.746795654296875e-05,
      "step": 9822,
      "training_step_time": 10.297914505004883
    },
    {
      "epoch": 3.7471771240234377e-05,
      "model_forward_time": 3.414022207260132,
      "step": 9823
    },
    {
      "epoch": 3.7471771240234377e-05,
      "step": 9823,
      "training_step_time": 10.194403886795044
    },
    {
      "epoch": 3.7471771240234377e-05,
      "model_forward_time": 2.4387283325195312,
      "step": 9823
    },
    {
      "epoch": 3.7471771240234377e-05,
      "step": 9823,
      "training_step_time": 7.471441745758057
    },
    {
      "epoch": 3.7471771240234377e-05,
      "model_forward_time": 3.1361448764801025,
      "step": 9823
    },
    {
      "epoch": 3.7471771240234377e-05,
      "step": 9823,
      "training_step_time": 9.913676023483276
    },
    {
      "epoch": 3.7471771240234377e-05,
      "model_forward_time": 3.141062021255493,
      "step": 9823
    },
    {
      "epoch": 3.7471771240234377e-05,
      "step": 9823,
      "training_step_time": 10.010841846466064
    },
    {
      "epoch": 3.74755859375e-05,
      "model_forward_time": 3.1548116207122803,
      "step": 9824
    },
    {
      "epoch": 3.74755859375e-05,
      "step": 9824,
      "training_step_time": 9.939305305480957
    },
    {
      "epoch": 3.74755859375e-05,
      "model_forward_time": 3.423736572265625,
      "step": 9824
    },
    {
      "epoch": 3.74755859375e-05,
      "step": 9824,
      "training_step_time": 10.205150365829468
    },
    {
      "epoch": 3.74755859375e-05,
      "model_forward_time": 3.1446962356567383,
      "step": 9824
    },
    {
      "epoch": 3.74755859375e-05,
      "step": 9824,
      "training_step_time": 9.92656660079956
    },
    {
      "epoch": 3.74755859375e-05,
      "model_forward_time": 2.4390580654144287,
      "step": 9824
    },
    {
      "epoch": 3.74755859375e-05,
      "step": 9824,
      "training_step_time": 10.327216625213623
    },
    {
      "epoch": 3.747940063476562e-05,
      "model_forward_time": 2.4505982398986816,
      "step": 9825
    },
    {
      "dynamics_loss_avg": 0.0415944954380393,
      "epoch": 3.747940063476562e-05,
      "step": 9825
    },
    {
      "action_loss_avg": 0.012091896287165583,
      "epoch": 3.747940063476562e-05,
      "step": 9825
    },
    {
      "epoch": 3.747940063476562e-05,
      "step": 9825,
      "training_step_time": 7.489546298980713
    },
    {
      "epoch": 3.747940063476562e-05,
      "model_forward_time": 2.4380884170532227,
      "step": 9825
    },
    {
      "epoch": 3.747940063476562e-05,
      "step": 9825,
      "training_step_time": 7.472497463226318
    },
    {
      "epoch": 3.747940063476562e-05,
      "model_forward_time": 3.435357093811035,
      "step": 9825
    },
    {
      "epoch": 3.747940063476562e-05,
      "step": 9825,
      "training_step_time": 10.214802026748657
    },
    {
      "epoch": 3.747940063476562e-05,
      "model_forward_time": 2.436707019805908,
      "step": 9825
    },
    {
      "epoch": 3.747940063476562e-05,
      "step": 9825,
      "training_step_time": 10.30483102798462
    },
    {
      "epoch": 3.748321533203125e-05,
      "model_forward_time": 3.1530537605285645,
      "step": 9826
    },
    {
      "epoch": 3.748321533203125e-05,
      "step": 9826,
      "training_step_time": 9.944366693496704
    },
    {
      "epoch": 3.748321533203125e-05,
      "model_forward_time": 2.438690662384033,
      "step": 9826
    },
    {
      "epoch": 3.748321533203125e-05,
      "step": 9826,
      "training_step_time": 7.470409870147705
    },
    {
      "epoch": 3.748321533203125e-05,
      "model_forward_time": 2.4363040924072266,
      "step": 9826
    },
    {
      "epoch": 3.748321533203125e-05,
      "step": 9826,
      "training_step_time": 7.474833965301514
    },
    {
      "epoch": 3.748321533203125e-05,
      "model_forward_time": 2.4342405796051025,
      "step": 9826
    },
    {
      "epoch": 3.748321533203125e-05,
      "step": 9826,
      "training_step_time": 10.314764499664307
    },
    {
      "epoch": 3.7487030029296875e-05,
      "model_forward_time": 3.472990036010742,
      "step": 9827
    },
    {
      "epoch": 3.7487030029296875e-05,
      "step": 9827,
      "training_step_time": 10.261950969696045
    },
    {
      "epoch": 3.7487030029296875e-05,
      "model_forward_time": 3.143439769744873,
      "step": 9827
    },
    {
      "epoch": 3.7487030029296875e-05,
      "step": 9827,
      "training_step_time": 9.935303926467896
    },
    {
      "epoch": 3.7487030029296875e-05,
      "model_forward_time": 3.147857904434204,
      "step": 9827
    },
    {
      "dynamics_loss_avg": 0.04557694438844919,
      "epoch": 3.7487030029296875e-05,
      "step": 9827
    },
    {
      "action_loss_avg": 0.019420280726626514,
      "epoch": 3.7487030029296875e-05,
      "step": 9827
    },
    {
      "epoch": 3.7487030029296875e-05,
      "step": 9827,
      "training_step_time": 9.931499481201172
    },
    {
      "epoch": 3.7487030029296875e-05,
      "model_forward_time": 3.137470006942749,
      "step": 9827
    },
    {
      "epoch": 3.7487030029296875e-05,
      "step": 9827,
      "training_step_time": 10.024380445480347
    },
    {
      "epoch": 3.74908447265625e-05,
      "model_forward_time": 3.1633658409118652,
      "step": 9828
    },
    {
      "epoch": 3.74908447265625e-05,
      "step": 9828,
      "training_step_time": 9.948490619659424
    },
    {
      "epoch": 3.74908447265625e-05,
      "model_forward_time": 2.762007474899292,
      "step": 9828
    },
    {
      "epoch": 3.74908447265625e-05,
      "step": 9828,
      "training_step_time": 7.789353132247925
    },
    {
      "epoch": 3.74908447265625e-05,
      "model_forward_time": 3.141183853149414,
      "step": 9828
    },
    {
      "epoch": 3.74908447265625e-05,
      "step": 9828,
      "training_step_time": 9.91663932800293
    },
    {
      "epoch": 3.74908447265625e-05,
      "model_forward_time": 2.439268112182617,
      "step": 9828
    },
    {
      "epoch": 3.74908447265625e-05,
      "step": 9828,
      "training_step_time": 10.343629598617554
    },
    {
      "epoch": 3.749465942382813e-05,
      "model_forward_time": 2.443222999572754,
      "step": 9829
    },
    {
      "epoch": 3.749465942382813e-05,
      "step": 9829,
      "training_step_time": 7.475680828094482
    },
    {
      "epoch": 3.749465942382813e-05,
      "model_forward_time": 3.144909620285034,
      "step": 9829
    },
    {
      "epoch": 3.749465942382813e-05,
      "step": 9829,
      "training_step_time": 9.926827192306519
    },
    {
      "epoch": 3.749465942382813e-05,
      "model_forward_time": 2.7387936115264893,
      "step": 9829
    },
    {
      "epoch": 3.749465942382813e-05,
      "step": 9829,
      "training_step_time": 7.76903772354126
    },
    {
      "epoch": 3.749465942382813e-05,
      "model_forward_time": 3.135493278503418,
      "step": 9829
    },
    {
      "epoch": 3.749465942382813e-05,
      "step": 9829,
      "training_step_time": 10.27452301979065
    },
    {
      "epoch": 3.749847412109375e-05,
      "grad_norm": 0.37621593475341797,
      "learning_rate": 1e-05,
      "loss": 0.0763,
      "step": 9830
    },
    {
      "epoch": 3.749847412109375e-05,
      "model_forward_time": 3.155900239944458,
      "step": 9830
    },
    {
      "dynamics_loss_avg": 0.05646335817873478,
      "epoch": 3.749847412109375e-05,
      "step": 9830
    },
    {
      "action_loss_avg": 0.019004746433347465,
      "epoch": 3.749847412109375e-05,
      "step": 9830
    },
    {
      "epoch": 3.749847412109375e-05,
      "step": 9830,
      "training_step_time": 9.937772989273071
    },
    {
      "epoch": 3.749847412109375e-05,
      "model_forward_time": 2.436279296875,
      "step": 9830
    },
    {
      "epoch": 3.749847412109375e-05,
      "step": 9830,
      "training_step_time": 7.4597015380859375
    },
    {
      "epoch": 3.749847412109375e-05,
      "model_forward_time": 2.43251371383667,
      "step": 9830
    },
    {
      "epoch": 3.749847412109375e-05,
      "step": 9830,
      "training_step_time": 7.456063747406006
    },
    {
      "epoch": 3.749847412109375e-05,
      "model_forward_time": 3.448596954345703,
      "step": 9830
    },
    {
      "epoch": 3.749847412109375e-05,
      "step": 9830,
      "training_step_time": 10.306005001068115
    },
    {
      "epoch": 3.750228881835937e-05,
      "model_forward_time": 3.1426639556884766,
      "step": 9831
    },
    {
      "epoch": 3.750228881835937e-05,
      "step": 9831,
      "training_step_time": 9.917809963226318
    },
    {
      "epoch": 3.750228881835937e-05,
      "model_forward_time": 2.4329488277435303,
      "step": 9831
    },
    {
      "epoch": 3.750228881835937e-05,
      "step": 9831,
      "training_step_time": 7.458536863327026
    },
    {
      "epoch": 3.750228881835937e-05,
      "model_forward_time": 2.434401750564575,
      "step": 9831
    },
    {
      "epoch": 3.750228881835937e-05,
      "step": 9831,
      "training_step_time": 7.464826583862305
    },
    {
      "epoch": 3.750228881835937e-05,
      "model_forward_time": 3.1384265422821045,
      "step": 9831
    },
    {
      "epoch": 3.750228881835937e-05,
      "step": 9831,
      "training_step_time": 10.325323581695557
    },
    {
      "epoch": 3.7506103515625e-05,
      "model_forward_time": 3.1496315002441406,
      "step": 9832
    },
    {
      "epoch": 3.7506103515625e-05,
      "step": 9832,
      "training_step_time": 9.932056427001953
    },
    {
      "epoch": 3.7506103515625e-05,
      "model_forward_time": 3.4633948802948,
      "step": 9832
    },
    {
      "epoch": 3.7506103515625e-05,
      "step": 9832,
      "training_step_time": 10.240090608596802
    },
    {
      "epoch": 3.7506103515625e-05,
      "model_forward_time": 3.141200542449951,
      "step": 9832
    },
    {
      "dynamics_loss_avg": 0.06505006179213524,
      "epoch": 3.7506103515625e-05,
      "step": 9832
    },
    {
      "action_loss_avg": 0.022066659247502684,
      "epoch": 3.7506103515625e-05,
      "step": 9832
    },
    {
      "epoch": 3.7506103515625e-05,
      "step": 9832,
      "training_step_time": 9.923406600952148
    },
    {
      "epoch": 3.7506103515625e-05,
      "model_forward_time": 2.439472198486328,
      "step": 9832
    },
    {
      "epoch": 3.7506103515625e-05,
      "step": 9832,
      "training_step_time": 10.264366865158081
    },
    {
      "epoch": 3.7509918212890626e-05,
      "model_forward_time": 2.4480271339416504,
      "step": 9833
    },
    {
      "epoch": 3.7509918212890626e-05,
      "step": 9833,
      "training_step_time": 7.478834390640259
    },
    {
      "epoch": 3.7509918212890626e-05,
      "model_forward_time": 3.1380019187927246,
      "step": 9833
    },
    {
      "epoch": 3.7509918212890626e-05,
      "step": 9833,
      "training_step_time": 9.902280807495117
    },
    {
      "epoch": 3.7509918212890626e-05,
      "model_forward_time": 3.442023992538452,
      "step": 9833
    },
    {
      "epoch": 3.7509918212890626e-05,
      "step": 9833,
      "training_step_time": 10.20989203453064
    },
    {
      "epoch": 3.7509918212890626e-05,
      "model_forward_time": 3.136634588241577,
      "step": 9833
    },
    {
      "epoch": 3.7509918212890626e-05,
      "step": 9833,
      "training_step_time": 10.26058292388916
    },
    {
      "epoch": 3.751373291015625e-05,
      "model_forward_time": 3.1482462882995605,
      "step": 9834
    },
    {
      "epoch": 3.751373291015625e-05,
      "step": 9834,
      "training_step_time": 9.911763191223145
    },
    {
      "epoch": 3.751373291015625e-05,
      "model_forward_time": 2.434335708618164,
      "step": 9834
    },
    {
      "epoch": 3.751373291015625e-05,
      "step": 9834,
      "training_step_time": 7.4595887660980225
    },
    {
      "epoch": 3.751373291015625e-05,
      "model_forward_time": 3.1381938457489014,
      "step": 9834
    },
    {
      "epoch": 3.751373291015625e-05,
      "step": 9834,
      "training_step_time": 9.90452241897583
    },
    {
      "epoch": 3.751373291015625e-05,
      "model_forward_time": 3.418792724609375,
      "step": 9834
    },
    {
      "epoch": 3.751373291015625e-05,
      "step": 9834,
      "training_step_time": 10.282534837722778
    },
    {
      "epoch": 3.751754760742188e-05,
      "model_forward_time": 3.1629796028137207,
      "step": 9835
    },
    {
      "dynamics_loss_avg": 0.050941914040595296,
      "epoch": 3.751754760742188e-05,
      "step": 9835
    },
    {
      "action_loss_avg": 0.016402813885360957,
      "epoch": 3.751754760742188e-05,
      "step": 9835
    },
    {
      "epoch": 3.751754760742188e-05,
      "step": 9835,
      "training_step_time": 9.942917108535767
    },
    {
      "epoch": 3.751754760742188e-05,
      "model_forward_time": 2.4355428218841553,
      "step": 9835
    },
    {
      "epoch": 3.751754760742188e-05,
      "step": 9835,
      "training_step_time": 7.466131210327148
    },
    {
      "epoch": 3.751754760742188e-05,
      "model_forward_time": 3.1438376903533936,
      "step": 9835
    },
    {
      "epoch": 3.751754760742188e-05,
      "step": 9835,
      "training_step_time": 9.91737151145935
    },
    {
      "epoch": 3.751754760742188e-05,
      "model_forward_time": 2.4392337799072266,
      "step": 9835
    },
    {
      "epoch": 3.751754760742188e-05,
      "step": 9835,
      "training_step_time": 10.019788265228271
    },
    {
      "epoch": 3.75213623046875e-05,
      "model_forward_time": 2.7749643325805664,
      "step": 9836
    },
    {
      "epoch": 3.75213623046875e-05,
      "step": 9836,
      "training_step_time": 7.8083178997039795
    },
    {
      "epoch": 3.75213623046875e-05,
      "model_forward_time": 3.136728048324585,
      "step": 9836
    },
    {
      "epoch": 3.75213623046875e-05,
      "step": 9836,
      "training_step_time": 9.905702114105225
    },
    {
      "epoch": 3.75213623046875e-05,
      "model_forward_time": 3.142819881439209,
      "step": 9836
    },
    {
      "epoch": 3.75213623046875e-05,
      "step": 9836,
      "training_step_time": 9.913421154022217
    },
    {
      "epoch": 3.75213623046875e-05,
      "model_forward_time": 2.4398865699768066,
      "step": 9836
    },
    {
      "epoch": 3.75213623046875e-05,
      "step": 9836,
      "training_step_time": 10.286862134933472
    },
    {
      "epoch": 3.7525177001953124e-05,
      "model_forward_time": 2.4480457305908203,
      "step": 9837
    },
    {
      "epoch": 3.7525177001953124e-05,
      "step": 9837,
      "training_step_time": 7.471103668212891
    },
    {
      "epoch": 3.7525177001953124e-05,
      "model_forward_time": 3.3946890830993652,
      "step": 9837
    },
    {
      "epoch": 3.7525177001953124e-05,
      "step": 9837,
      "training_step_time": 10.158920288085938
    },
    {
      "epoch": 3.7525177001953124e-05,
      "model_forward_time": 3.141889810562134,
      "step": 9837
    },
    {
      "dynamics_loss_avg": 0.048304231651127336,
      "epoch": 3.7525177001953124e-05,
      "step": 9837
    },
    {
      "action_loss_avg": 0.02354794358834624,
      "epoch": 3.7525177001953124e-05,
      "step": 9837
    },
    {
      "epoch": 3.7525177001953124e-05,
      "step": 9837,
      "training_step_time": 9.914056301116943
    },
    {
      "epoch": 3.7525177001953124e-05,
      "model_forward_time": 2.436981678009033,
      "step": 9837
    },
    {
      "epoch": 3.7525177001953124e-05,
      "step": 9837,
      "training_step_time": 10.019540786743164
    },
    {
      "epoch": 3.752899169921875e-05,
      "model_forward_time": 3.1668918132781982,
      "step": 9838
    },
    {
      "epoch": 3.752899169921875e-05,
      "step": 9838,
      "training_step_time": 9.939167022705078
    },
    {
      "epoch": 3.752899169921875e-05,
      "model_forward_time": 2.4394068717956543,
      "step": 9838
    },
    {
      "epoch": 3.752899169921875e-05,
      "step": 9838,
      "training_step_time": 7.472341299057007
    },
    {
      "epoch": 3.752899169921875e-05,
      "model_forward_time": 3.1387975215911865,
      "step": 9838
    },
    {
      "epoch": 3.752899169921875e-05,
      "step": 9838,
      "training_step_time": 10.19059133529663
    },
    {
      "epoch": 3.752899169921875e-05,
      "model_forward_time": 3.136619806289673,
      "step": 9838
    },
    {
      "epoch": 3.752899169921875e-05,
      "step": 9838,
      "training_step_time": 10.299089431762695
    },
    {
      "epoch": 3.7532806396484377e-05,
      "model_forward_time": 2.449274778366089,
      "step": 9839
    },
    {
      "epoch": 3.7532806396484377e-05,
      "step": 9839,
      "training_step_time": 7.471445560455322
    },
    {
      "epoch": 3.7532806396484377e-05,
      "model_forward_time": 3.1356780529022217,
      "step": 9839
    },
    {
      "epoch": 3.7532806396484377e-05,
      "step": 9839,
      "training_step_time": 9.899601936340332
    },
    {
      "epoch": 3.7532806396484377e-05,
      "model_forward_time": 3.1302521228790283,
      "step": 9839
    },
    {
      "epoch": 3.7532806396484377e-05,
      "step": 9839,
      "training_step_time": 9.898555040359497
    },
    {
      "epoch": 3.7532806396484377e-05,
      "model_forward_time": 2.4370534420013428,
      "step": 9839
    },
    {
      "epoch": 3.7532806396484377e-05,
      "step": 9839,
      "training_step_time": 10.29824185371399
    },
    {
      "epoch": 3.753662109375e-05,
      "grad_norm": 0.47323891520500183,
      "learning_rate": 1e-05,
      "loss": 0.0758,
      "step": 9840
    },
    {
      "epoch": 3.753662109375e-05,
      "model_forward_time": 2.7941575050354004,
      "step": 9840
    },
    {
      "dynamics_loss_avg": 0.060899038892239334,
      "epoch": 3.753662109375e-05,
      "step": 9840
    },
    {
      "action_loss_avg": 0.0227471363148652,
      "epoch": 3.753662109375e-05,
      "step": 9840
    },
    {
      "epoch": 3.753662109375e-05,
      "step": 9840,
      "training_step_time": 7.819719314575195
    },
    {
      "epoch": 3.753662109375e-05,
      "model_forward_time": 2.433584451675415,
      "step": 9840
    },
    {
      "epoch": 3.753662109375e-05,
      "step": 9840,
      "training_step_time": 7.457038879394531
    },
    {
      "epoch": 3.753662109375e-05,
      "model_forward_time": 3.1336584091186523,
      "step": 9840
    },
    {
      "epoch": 3.753662109375e-05,
      "step": 9840,
      "training_step_time": 9.903600215911865
    },
    {
      "epoch": 3.753662109375e-05,
      "model_forward_time": 2.4371118545532227,
      "step": 9840
    },
    {
      "epoch": 3.753662109375e-05,
      "step": 9840,
      "training_step_time": 10.019291162490845
    },
    {
      "epoch": 3.754043579101562e-05,
      "model_forward_time": 3.146681070327759,
      "step": 9841
    },
    {
      "epoch": 3.754043579101562e-05,
      "step": 9841,
      "training_step_time": 9.91757082939148
    },
    {
      "epoch": 3.754043579101562e-05,
      "model_forward_time": 2.7087724208831787,
      "step": 9841
    },
    {
      "epoch": 3.754043579101562e-05,
      "step": 9841,
      "training_step_time": 7.738960027694702
    },
    {
      "epoch": 3.754043579101562e-05,
      "model_forward_time": 3.1364965438842773,
      "step": 9841
    },
    {
      "epoch": 3.754043579101562e-05,
      "step": 9841,
      "training_step_time": 9.906745910644531
    },
    {
      "epoch": 3.754043579101562e-05,
      "model_forward_time": 3.1425600051879883,
      "step": 9841
    },
    {
      "epoch": 3.754043579101562e-05,
      "step": 9841,
      "training_step_time": 10.350409507751465
    },
    {
      "epoch": 3.754425048828125e-05,
      "model_forward_time": 3.1662354469299316,
      "step": 9842
    },
    {
      "epoch": 3.754425048828125e-05,
      "step": 9842,
      "training_step_time": 9.944824934005737
    },
    {
      "epoch": 3.754425048828125e-05,
      "model_forward_time": 2.434569835662842,
      "step": 9842
    },
    {
      "epoch": 3.754425048828125e-05,
      "step": 9842,
      "training_step_time": 7.462467193603516
    },
    {
      "epoch": 3.754425048828125e-05,
      "model_forward_time": 3.3794643878936768,
      "step": 9842
    },
    {
      "dynamics_loss_avg": 0.050319099240005015,
      "epoch": 3.754425048828125e-05,
      "step": 9842
    },
    {
      "action_loss_avg": 0.019113160995766522,
      "epoch": 3.754425048828125e-05,
      "step": 9842
    },
    {
      "epoch": 3.754425048828125e-05,
      "step": 9842,
      "training_step_time": 10.15205192565918
    },
    {
      "epoch": 3.754425048828125e-05,
      "model_forward_time": 2.4324498176574707,
      "step": 9842
    },
    {
      "epoch": 3.754425048828125e-05,
      "step": 9842,
      "training_step_time": 10.281075477600098
    },
    {
      "epoch": 3.7548065185546875e-05,
      "model_forward_time": 3.161832094192505,
      "step": 9843
    },
    {
      "epoch": 3.7548065185546875e-05,
      "step": 9843,
      "training_step_time": 9.928436994552612
    },
    {
      "epoch": 3.7548065185546875e-05,
      "model_forward_time": 2.4325919151306152,
      "step": 9843
    },
    {
      "epoch": 3.7548065185546875e-05,
      "step": 9843,
      "training_step_time": 7.4603376388549805
    },
    {
      "epoch": 3.7548065185546875e-05,
      "model_forward_time": 2.435018301010132,
      "step": 9843
    },
    {
      "epoch": 3.7548065185546875e-05,
      "step": 9843,
      "training_step_time": 7.462570905685425
    },
    {
      "epoch": 3.7548065185546875e-05,
      "model_forward_time": 2.710601568222046,
      "step": 9843
    },
    {
      "epoch": 3.7548065185546875e-05,
      "step": 9843,
      "training_step_time": 10.021188497543335
    },
    {
      "epoch": 3.75518798828125e-05,
      "model_forward_time": 2.4505670070648193,
      "step": 9844
    },
    {
      "epoch": 3.75518798828125e-05,
      "step": 9844,
      "training_step_time": 7.478353977203369
    },
    {
      "epoch": 3.75518798828125e-05,
      "model_forward_time": 3.131181240081787,
      "step": 9844
    },
    {
      "epoch": 3.75518798828125e-05,
      "step": 9844,
      "training_step_time": 9.897375345230103
    },
    {
      "epoch": 3.75518798828125e-05,
      "model_forward_time": 3.1343729496002197,
      "step": 9844
    },
    {
      "epoch": 3.75518798828125e-05,
      "step": 9844,
      "training_step_time": 9.901408195495605
    },
    {
      "epoch": 3.75518798828125e-05,
      "model_forward_time": 1.744903802871704,
      "step": 9844
    },
    {
      "epoch": 3.75518798828125e-05,
      "step": 9844,
      "training_step_time": 10.312994241714478
    },
    {
      "epoch": 3.755569458007813e-05,
      "model_forward_time": 3.136336088180542,
      "step": 9845
    },
    {
      "dynamics_loss_avg": 0.05085001345723868,
      "epoch": 3.755569458007813e-05,
      "step": 9845
    },
    {
      "action_loss_avg": 0.016976968524977563,
      "epoch": 3.755569458007813e-05,
      "step": 9845
    },
    {
      "epoch": 3.755569458007813e-05,
      "step": 9845,
      "training_step_time": 10.210553407669067
    },
    {
      "epoch": 3.755569458007813e-05,
      "model_forward_time": 3.132232904434204,
      "step": 9845
    },
    {
      "epoch": 3.755569458007813e-05,
      "step": 9845,
      "training_step_time": 9.89421534538269
    },
    {
      "epoch": 3.755569458007813e-05,
      "model_forward_time": 3.1384670734405518,
      "step": 9845
    },
    {
      "epoch": 3.755569458007813e-05,
      "step": 9845,
      "training_step_time": 9.905377626419067
    },
    {
      "epoch": 3.755569458007813e-05,
      "model_forward_time": 3.1335930824279785,
      "step": 9845
    },
    {
      "epoch": 3.755569458007813e-05,
      "step": 9845,
      "training_step_time": 10.259305238723755
    },
    {
      "epoch": 3.755950927734375e-05,
      "model_forward_time": 3.1437463760375977,
      "step": 9846
    },
    {
      "epoch": 3.755950927734375e-05,
      "step": 9846,
      "training_step_time": 9.919812202453613
    },
    {
      "epoch": 3.755950927734375e-05,
      "model_forward_time": 2.436603546142578,
      "step": 9846
    },
    {
      "epoch": 3.755950927734375e-05,
      "step": 9846,
      "training_step_time": 7.465606451034546
    },
    {
      "epoch": 3.755950927734375e-05,
      "model_forward_time": 3.4178266525268555,
      "step": 9846
    },
    {
      "epoch": 3.755950927734375e-05,
      "step": 9846,
      "training_step_time": 10.184746980667114
    },
    {
      "epoch": 3.755950927734375e-05,
      "model_forward_time": 3.1320018768310547,
      "step": 9846
    },
    {
      "epoch": 3.755950927734375e-05,
      "step": 9846,
      "training_step_time": 10.263030290603638
    },
    {
      "epoch": 3.756332397460937e-05,
      "model_forward_time": 3.14607834815979,
      "step": 9847
    },
    {
      "epoch": 3.756332397460937e-05,
      "step": 9847,
      "training_step_time": 9.919505834579468
    },
    {
      "epoch": 3.756332397460937e-05,
      "model_forward_time": 3.140723466873169,
      "step": 9847
    },
    {
      "epoch": 3.756332397460937e-05,
      "step": 9847,
      "training_step_time": 9.911030769348145
    },
    {
      "epoch": 3.756332397460937e-05,
      "model_forward_time": 2.4339911937713623,
      "step": 9847
    },
    {
      "dynamics_loss_avg": 0.045613431558012965,
      "epoch": 3.756332397460937e-05,
      "step": 9847
    },
    {
      "action_loss_avg": 0.021984740812331437,
      "epoch": 3.756332397460937e-05,
      "step": 9847
    },
    {
      "epoch": 3.756332397460937e-05,
      "step": 9847,
      "training_step_time": 7.470958709716797
    },
    {
      "epoch": 3.756332397460937e-05,
      "model_forward_time": 2.7050204277038574,
      "step": 9847
    },
    {
      "epoch": 3.756332397460937e-05,
      "step": 9847,
      "training_step_time": 10.282799005508423
    },
    {
      "epoch": 3.7567138671875e-05,
      "model_forward_time": 2.4534647464752197,
      "step": 9848
    },
    {
      "epoch": 3.7567138671875e-05,
      "step": 9848,
      "training_step_time": 7.486912488937378
    },
    {
      "epoch": 3.7567138671875e-05,
      "model_forward_time": 2.4323408603668213,
      "step": 9848
    },
    {
      "epoch": 3.7567138671875e-05,
      "step": 9848,
      "training_step_time": 7.458774566650391
    },
    {
      "epoch": 3.7567138671875e-05,
      "model_forward_time": 2.4355556964874268,
      "step": 9848
    },
    {
      "epoch": 3.7567138671875e-05,
      "step": 9848,
      "training_step_time": 7.461282730102539
    },
    {
      "epoch": 3.7567138671875e-05,
      "model_forward_time": 2.4305009841918945,
      "step": 9848
    },
    {
      "epoch": 3.7567138671875e-05,
      "step": 9848,
      "training_step_time": 10.262798309326172
    },
    {
      "epoch": 3.7570953369140626e-05,
      "model_forward_time": 3.4075441360473633,
      "step": 9849
    },
    {
      "epoch": 3.7570953369140626e-05,
      "step": 9849,
      "training_step_time": 10.174493789672852
    },
    {
      "epoch": 3.7570953369140626e-05,
      "model_forward_time": 2.435079336166382,
      "step": 9849
    },
    {
      "epoch": 3.7570953369140626e-05,
      "step": 9849,
      "training_step_time": 7.461550712585449
    },
    {
      "epoch": 3.7570953369140626e-05,
      "model_forward_time": 2.4351823329925537,
      "step": 9849
    },
    {
      "epoch": 3.7570953369140626e-05,
      "step": 9849,
      "training_step_time": 7.459205627441406
    },
    {
      "epoch": 3.7570953369140626e-05,
      "model_forward_time": 2.433298349380493,
      "step": 9849
    },
    {
      "epoch": 3.7570953369140626e-05,
      "step": 9849,
      "training_step_time": 10.277087211608887
    },
    {
      "epoch": 3.757476806640625e-05,
      "grad_norm": 0.24816815555095673,
      "learning_rate": 1e-05,
      "loss": 0.0768,
      "step": 9850
    },
    {
      "epoch": 3.757476806640625e-05,
      "model_forward_time": 2.447735548019409,
      "step": 9850
    },
    {
      "dynamics_loss_avg": 0.05001607369631529,
      "epoch": 3.757476806640625e-05,
      "step": 9850
    },
    {
      "action_loss_avg": 0.017237214744091033,
      "epoch": 3.757476806640625e-05,
      "step": 9850
    },
    {
      "epoch": 3.757476806640625e-05,
      "step": 9850,
      "training_step_time": 7.47480320930481
    },
    {
      "epoch": 3.757476806640625e-05,
      "model_forward_time": 3.1294169425964355,
      "step": 9850
    },
    {
      "epoch": 3.757476806640625e-05,
      "step": 9850,
      "training_step_time": 10.170799255371094
    },
    {
      "epoch": 3.757476806640625e-05,
      "model_forward_time": 3.1320908069610596,
      "step": 9850
    },
    {
      "epoch": 3.757476806640625e-05,
      "step": 9850,
      "training_step_time": 9.896870613098145
    },
    {
      "epoch": 3.757476806640625e-05,
      "model_forward_time": 3.1285459995269775,
      "step": 9850
    },
    {
      "epoch": 3.757476806640625e-05,
      "step": 9850,
      "training_step_time": 10.318964719772339
    },
    {
      "epoch": 3.757858276367188e-05,
      "model_forward_time": 2.446115493774414,
      "step": 9851
    },
    {
      "epoch": 3.757858276367188e-05,
      "step": 9851,
      "training_step_time": 7.467557430267334
    },
    {
      "epoch": 3.757858276367188e-05,
      "model_forward_time": 2.431741714477539,
      "step": 9851
    },
    {
      "epoch": 3.757858276367188e-05,
      "step": 9851,
      "training_step_time": 7.455732345581055
    },
    {
      "epoch": 3.757858276367188e-05,
      "model_forward_time": 2.4281797409057617,
      "step": 9851
    },
    {
      "epoch": 3.757858276367188e-05,
      "step": 9851,
      "training_step_time": 7.449614524841309
    },
    {
      "epoch": 3.757858276367188e-05,
      "model_forward_time": 3.4071311950683594,
      "step": 9851
    },
    {
      "epoch": 3.757858276367188e-05,
      "step": 9851,
      "training_step_time": 10.299336194992065
    },
    {
      "epoch": 3.75823974609375e-05,
      "model_forward_time": 3.1524829864501953,
      "step": 9852
    },
    {
      "epoch": 3.75823974609375e-05,
      "step": 9852,
      "training_step_time": 9.924674034118652
    },
    {
      "epoch": 3.75823974609375e-05,
      "model_forward_time": 2.434774398803711,
      "step": 9852
    },
    {
      "epoch": 3.75823974609375e-05,
      "step": 9852,
      "training_step_time": 7.463522434234619
    },
    {
      "epoch": 3.75823974609375e-05,
      "model_forward_time": 3.1405997276306152,
      "step": 9852
    },
    {
      "dynamics_loss_avg": 0.05280132181942463,
      "epoch": 3.75823974609375e-05,
      "step": 9852
    },
    {
      "action_loss_avg": 0.016540990862995385,
      "epoch": 3.75823974609375e-05,
      "step": 9852
    },
    {
      "epoch": 3.75823974609375e-05,
      "step": 9852,
      "training_step_time": 9.9169442653656
    },
    {
      "epoch": 3.75823974609375e-05,
      "model_forward_time": 2.432908058166504,
      "step": 9852
    },
    {
      "epoch": 3.75823974609375e-05,
      "step": 9852,
      "training_step_time": 10.320698738098145
    },
    {
      "epoch": 3.7586212158203124e-05,
      "model_forward_time": 2.746812582015991,
      "step": 9853
    },
    {
      "epoch": 3.7586212158203124e-05,
      "step": 9853,
      "training_step_time": 7.776449203491211
    },
    {
      "epoch": 3.7586212158203124e-05,
      "model_forward_time": 2.4431371688842773,
      "step": 9853
    },
    {
      "epoch": 3.7586212158203124e-05,
      "step": 9853,
      "training_step_time": 7.4751527309417725
    },
    {
      "epoch": 3.7586212158203124e-05,
      "model_forward_time": 2.4404048919677734,
      "step": 9853
    },
    {
      "epoch": 3.7586212158203124e-05,
      "step": 9853,
      "training_step_time": 7.47139048576355
    },
    {
      "epoch": 3.7586212158203124e-05,
      "model_forward_time": 2.4379963874816895,
      "step": 9853
    },
    {
      "epoch": 3.7586212158203124e-05,
      "step": 9853,
      "training_step_time": 10.282642602920532
    },
    {
      "epoch": 3.759002685546875e-05,
      "model_forward_time": 2.4550158977508545,
      "step": 9854
    },
    {
      "epoch": 3.759002685546875e-05,
      "step": 9854,
      "training_step_time": 7.483100652694702
    },
    {
      "epoch": 3.759002685546875e-05,
      "model_forward_time": 3.407883882522583,
      "step": 9854
    },
    {
      "epoch": 3.759002685546875e-05,
      "step": 9854,
      "training_step_time": 10.171220064163208
    },
    {
      "epoch": 3.759002685546875e-05,
      "model_forward_time": 3.1411936283111572,
      "step": 9854
    },
    {
      "epoch": 3.759002685546875e-05,
      "step": 9854,
      "training_step_time": 9.900314807891846
    },
    {
      "epoch": 3.759002685546875e-05,
      "model_forward_time": 1.7468321323394775,
      "step": 9854
    },
    {
      "epoch": 3.759002685546875e-05,
      "step": 9854,
      "training_step_time": 10.318428754806519
    },
    {
      "epoch": 3.759384155273438e-05,
      "model_forward_time": 3.1452584266662598,
      "step": 9855
    },
    {
      "dynamics_loss_avg": 0.05070169121026993,
      "epoch": 3.759384155273438e-05,
      "step": 9855
    },
    {
      "action_loss_avg": 0.014951181691139936,
      "epoch": 3.759384155273438e-05,
      "step": 9855
    },
    {
      "epoch": 3.759384155273438e-05,
      "step": 9855,
      "training_step_time": 9.926783084869385
    },
    {
      "epoch": 3.759384155273438e-05,
      "model_forward_time": 2.434311866760254,
      "step": 9855
    },
    {
      "epoch": 3.759384155273438e-05,
      "step": 9855,
      "training_step_time": 7.459951877593994
    },
    {
      "epoch": 3.759384155273438e-05,
      "model_forward_time": 3.1302905082702637,
      "step": 9855
    },
    {
      "epoch": 3.759384155273438e-05,
      "step": 9855,
      "training_step_time": 9.89749813079834
    },
    {
      "epoch": 3.759384155273438e-05,
      "model_forward_time": 2.724041223526001,
      "step": 9855
    },
    {
      "epoch": 3.759384155273438e-05,
      "step": 9855,
      "training_step_time": 10.312394857406616
    },
    {
      "epoch": 3.759765625e-05,
      "model_forward_time": 3.161862850189209,
      "step": 9856
    },
    {
      "epoch": 3.759765625e-05,
      "step": 9856,
      "training_step_time": 9.936243057250977
    },
    {
      "epoch": 3.759765625e-05,
      "model_forward_time": 3.1398606300354004,
      "step": 9856
    },
    {
      "epoch": 3.759765625e-05,
      "step": 9856,
      "training_step_time": 9.910364151000977
    },
    {
      "epoch": 3.759765625e-05,
      "model_forward_time": 3.135099172592163,
      "step": 9856
    },
    {
      "epoch": 3.759765625e-05,
      "step": 9856,
      "training_step_time": 9.908804655075073
    },
    {
      "epoch": 3.759765625e-05,
      "model_forward_time": 3.1386680603027344,
      "step": 9856
    },
    {
      "epoch": 3.759765625e-05,
      "step": 9856,
      "training_step_time": 10.313456296920776
    },
    {
      "epoch": 3.760147094726562e-05,
      "model_forward_time": 3.4236459732055664,
      "step": 9857
    },
    {
      "epoch": 3.760147094726562e-05,
      "step": 9857,
      "training_step_time": 10.198843240737915
    },
    {
      "epoch": 3.760147094726562e-05,
      "model_forward_time": 2.4331159591674805,
      "step": 9857
    },
    {
      "epoch": 3.760147094726562e-05,
      "step": 9857,
      "training_step_time": 7.459179162979126
    },
    {
      "epoch": 3.760147094726562e-05,
      "model_forward_time": 2.4329655170440674,
      "step": 9857
    },
    {
      "dynamics_loss_avg": 0.058803205750882624,
      "epoch": 3.760147094726562e-05,
      "step": 9857
    },
    {
      "action_loss_avg": 0.02087886184453964,
      "epoch": 3.760147094726562e-05,
      "step": 9857
    },
    {
      "epoch": 3.760147094726562e-05,
      "step": 9857,
      "training_step_time": 7.461299419403076
    },
    {
      "epoch": 3.760147094726562e-05,
      "model_forward_time": 3.1314594745635986,
      "step": 9857
    },
    {
      "epoch": 3.760147094726562e-05,
      "step": 9857,
      "training_step_time": 10.278156280517578
    },
    {
      "epoch": 3.760528564453125e-05,
      "model_forward_time": 3.147489547729492,
      "step": 9858
    },
    {
      "epoch": 3.760528564453125e-05,
      "step": 9858,
      "training_step_time": 9.923001289367676
    },
    {
      "epoch": 3.760528564453125e-05,
      "model_forward_time": 3.4477150440216064,
      "step": 9858
    },
    {
      "epoch": 3.760528564453125e-05,
      "step": 9858,
      "training_step_time": 10.223689794540405
    },
    {
      "epoch": 3.760528564453125e-05,
      "model_forward_time": 3.1385533809661865,
      "step": 9858
    },
    {
      "epoch": 3.760528564453125e-05,
      "step": 9858,
      "training_step_time": 9.915151357650757
    },
    {
      "epoch": 3.760528564453125e-05,
      "model_forward_time": 3.140662431716919,
      "step": 9858
    },
    {
      "epoch": 3.760528564453125e-05,
      "step": 9858,
      "training_step_time": 10.264386653900146
    },
    {
      "epoch": 3.7609100341796875e-05,
      "model_forward_time": 3.1519010066986084,
      "step": 9859
    },
    {
      "epoch": 3.7609100341796875e-05,
      "step": 9859,
      "training_step_time": 9.933063983917236
    },
    {
      "epoch": 3.7609100341796875e-05,
      "model_forward_time": 3.1387290954589844,
      "step": 9859
    },
    {
      "epoch": 3.7609100341796875e-05,
      "step": 9859,
      "training_step_time": 9.917280912399292
    },
    {
      "epoch": 3.7609100341796875e-05,
      "model_forward_time": 3.424687147140503,
      "step": 9859
    },
    {
      "epoch": 3.7609100341796875e-05,
      "step": 9859,
      "training_step_time": 10.205172061920166
    },
    {
      "epoch": 3.7609100341796875e-05,
      "model_forward_time": 2.437532901763916,
      "step": 9859
    },
    {
      "epoch": 3.7609100341796875e-05,
      "step": 9859,
      "training_step_time": 10.319074869155884
    },
    {
      "epoch": 3.76129150390625e-05,
      "grad_norm": 0.3005591034889221,
      "learning_rate": 1e-05,
      "loss": 0.0761,
      "step": 9860
    },
    {
      "epoch": 3.76129150390625e-05,
      "model_forward_time": 3.1526544094085693,
      "step": 9860
    },
    {
      "dynamics_loss_avg": 0.05173585619777441,
      "epoch": 3.76129150390625e-05,
      "step": 9860
    },
    {
      "action_loss_avg": 0.017742783087305723,
      "epoch": 3.76129150390625e-05,
      "step": 9860
    },
    {
      "epoch": 3.76129150390625e-05,
      "step": 9860,
      "training_step_time": 9.93250846862793
    },
    {
      "epoch": 3.76129150390625e-05,
      "model_forward_time": 2.4344100952148438,
      "step": 9860
    },
    {
      "epoch": 3.76129150390625e-05,
      "step": 9860,
      "training_step_time": 7.467694044113159
    },
    {
      "epoch": 3.76129150390625e-05,
      "model_forward_time": 2.4271080493927,
      "step": 9860
    },
    {
      "epoch": 3.76129150390625e-05,
      "step": 9860,
      "training_step_time": 7.454423427581787
    },
    {
      "epoch": 3.76129150390625e-05,
      "model_forward_time": 2.7003114223480225,
      "step": 9860
    },
    {
      "epoch": 3.76129150390625e-05,
      "step": 9860,
      "training_step_time": 10.00887393951416
    },
    {
      "epoch": 3.761672973632813e-05,
      "model_forward_time": 3.1563992500305176,
      "step": 9861
    },
    {
      "epoch": 3.761672973632813e-05,
      "step": 9861,
      "training_step_time": 9.949931144714355
    },
    {
      "epoch": 3.761672973632813e-05,
      "model_forward_time": 1.7400903701782227,
      "step": 9861
    },
    {
      "epoch": 3.761672973632813e-05,
      "step": 9861,
      "training_step_time": 5.195845603942871
    },
    {
      "epoch": 3.761672973632813e-05,
      "model_forward_time": 1.7353706359863281,
      "step": 9861
    },
    {
      "epoch": 3.761672973632813e-05,
      "step": 9861,
      "training_step_time": 5.1847922801971436
    },
    {
      "epoch": 3.761672973632813e-05,
      "model_forward_time": 2.428588390350342,
      "step": 9861
    },
    {
      "epoch": 3.761672973632813e-05,
      "step": 9861,
      "training_step_time": 10.28930926322937
    },
    {
      "epoch": 3.762054443359375e-05,
      "model_forward_time": 2.4354794025421143,
      "step": 9862
    },
    {
      "epoch": 3.762054443359375e-05,
      "step": 9862,
      "training_step_time": 7.460411787033081
    },
    {
      "epoch": 3.762054443359375e-05,
      "model_forward_time": 2.430161714553833,
      "step": 9862
    },
    {
      "epoch": 3.762054443359375e-05,
      "step": 9862,
      "training_step_time": 7.453024625778198
    },
    {
      "epoch": 3.762054443359375e-05,
      "model_forward_time": 3.432518482208252,
      "step": 9862
    },
    {
      "dynamics_loss_avg": 0.05730335395783186,
      "epoch": 3.762054443359375e-05,
      "step": 9862
    },
    {
      "action_loss_avg": 0.02155271009542048,
      "epoch": 3.762054443359375e-05,
      "step": 9862
    },
    {
      "epoch": 3.762054443359375e-05,
      "step": 9862,
      "training_step_time": 10.205307245254517
    },
    {
      "epoch": 3.762054443359375e-05,
      "model_forward_time": 3.1285548210144043,
      "step": 9862
    },
    {
      "epoch": 3.762054443359375e-05,
      "step": 9862,
      "training_step_time": 10.28145146369934
    },
    {
      "epoch": 3.762435913085937e-05,
      "model_forward_time": 2.446505546569824,
      "step": 9863
    },
    {
      "epoch": 3.762435913085937e-05,
      "step": 9863,
      "training_step_time": 7.47648549079895
    },
    {
      "epoch": 3.762435913085937e-05,
      "model_forward_time": 3.1307108402252197,
      "step": 9863
    },
    {
      "epoch": 3.762435913085937e-05,
      "step": 9863,
      "training_step_time": 9.90336298942566
    },
    {
      "epoch": 3.762435913085937e-05,
      "model_forward_time": 2.433173179626465,
      "step": 9863
    },
    {
      "epoch": 3.762435913085937e-05,
      "step": 9863,
      "training_step_time": 7.4601476192474365
    },
    {
      "epoch": 3.762435913085937e-05,
      "model_forward_time": 3.4169223308563232,
      "step": 9863
    },
    {
      "epoch": 3.762435913085937e-05,
      "step": 9863,
      "training_step_time": 10.28064751625061
    },
    {
      "epoch": 3.7628173828125e-05,
      "model_forward_time": 2.448091506958008,
      "step": 9864
    },
    {
      "epoch": 3.7628173828125e-05,
      "step": 9864,
      "training_step_time": 7.4761061668396
    },
    {
      "epoch": 3.7628173828125e-05,
      "model_forward_time": 2.4257378578186035,
      "step": 9864
    },
    {
      "epoch": 3.7628173828125e-05,
      "step": 9864,
      "training_step_time": 7.454196453094482
    },
    {
      "epoch": 3.7628173828125e-05,
      "model_forward_time": 2.430447816848755,
      "step": 9864
    },
    {
      "epoch": 3.7628173828125e-05,
      "step": 9864,
      "training_step_time": 7.458876609802246
    },
    {
      "epoch": 3.7628173828125e-05,
      "model_forward_time": 2.4312288761138916,
      "step": 9864
    },
    {
      "epoch": 3.7628173828125e-05,
      "step": 9864,
      "training_step_time": 10.294611692428589
    },
    {
      "epoch": 3.7631988525390626e-05,
      "model_forward_time": 3.4007863998413086,
      "step": 9865
    },
    {
      "dynamics_loss_avg": 0.053774548694491386,
      "epoch": 3.7631988525390626e-05,
      "step": 9865
    },
    {
      "action_loss_avg": 0.015910643967799842,
      "epoch": 3.7631988525390626e-05,
      "step": 9865
    },
    {
      "epoch": 3.7631988525390626e-05,
      "step": 9865,
      "training_step_time": 10.181962490081787
    },
    {
      "epoch": 3.7631988525390626e-05,
      "model_forward_time": 3.1379306316375732,
      "step": 9865
    },
    {
      "epoch": 3.7631988525390626e-05,
      "step": 9865,
      "training_step_time": 9.911925077438354
    },
    {
      "epoch": 3.7631988525390626e-05,
      "model_forward_time": 2.4279448986053467,
      "step": 9865
    },
    {
      "epoch": 3.7631988525390626e-05,
      "step": 9865,
      "training_step_time": 7.45930814743042
    },
    {
      "epoch": 3.7631988525390626e-05,
      "model_forward_time": 3.134734869003296,
      "step": 9865
    },
    {
      "epoch": 3.7631988525390626e-05,
      "step": 9865,
      "training_step_time": 10.278563499450684
    },
    {
      "epoch": 3.763580322265625e-05,
      "model_forward_time": 2.4447438716888428,
      "step": 9866
    },
    {
      "epoch": 3.763580322265625e-05,
      "step": 9866,
      "training_step_time": 7.480421781539917
    },
    {
      "epoch": 3.763580322265625e-05,
      "model_forward_time": 3.4491591453552246,
      "step": 9866
    },
    {
      "epoch": 3.763580322265625e-05,
      "step": 9866,
      "training_step_time": 10.224185228347778
    },
    {
      "epoch": 3.763580322265625e-05,
      "model_forward_time": 3.1371946334838867,
      "step": 9866
    },
    {
      "epoch": 3.763580322265625e-05,
      "step": 9866,
      "training_step_time": 9.918090105056763
    },
    {
      "epoch": 3.763580322265625e-05,
      "model_forward_time": 2.432696580886841,
      "step": 9866
    },
    {
      "epoch": 3.763580322265625e-05,
      "step": 9866,
      "training_step_time": 10.290201187133789
    },
    {
      "epoch": 3.763961791992188e-05,
      "model_forward_time": 3.1474907398223877,
      "step": 9867
    },
    {
      "epoch": 3.763961791992188e-05,
      "step": 9867,
      "training_step_time": 9.93026351928711
    },
    {
      "epoch": 3.763961791992188e-05,
      "model_forward_time": 3.1411092281341553,
      "step": 9867
    },
    {
      "epoch": 3.763961791992188e-05,
      "step": 9867,
      "training_step_time": 9.923277139663696
    },
    {
      "epoch": 3.763961791992188e-05,
      "model_forward_time": 2.716827154159546,
      "step": 9867
    },
    {
      "dynamics_loss_avg": 0.05186120234429836,
      "epoch": 3.763961791992188e-05,
      "step": 9867
    },
    {
      "action_loss_avg": 0.01715577128343284,
      "epoch": 3.763961791992188e-05,
      "step": 9867
    },
    {
      "epoch": 3.763961791992188e-05,
      "step": 9867,
      "training_step_time": 7.753420352935791
    },
    {
      "epoch": 3.763961791992188e-05,
      "model_forward_time": 3.139346122741699,
      "step": 9867
    },
    {
      "epoch": 3.763961791992188e-05,
      "step": 9867,
      "training_step_time": 10.292507410049438
    },
    {
      "epoch": 3.76434326171875e-05,
      "model_forward_time": 3.158748149871826,
      "step": 9868
    },
    {
      "epoch": 3.76434326171875e-05,
      "step": 9868,
      "training_step_time": 9.944823741912842
    },
    {
      "epoch": 3.76434326171875e-05,
      "model_forward_time": 3.1381947994232178,
      "step": 9868
    },
    {
      "epoch": 3.76434326171875e-05,
      "step": 9868,
      "training_step_time": 9.919551610946655
    },
    {
      "epoch": 3.76434326171875e-05,
      "model_forward_time": 3.1452999114990234,
      "step": 9868
    },
    {
      "epoch": 3.76434326171875e-05,
      "step": 9868,
      "training_step_time": 9.930976867675781
    },
    {
      "epoch": 3.76434326171875e-05,
      "model_forward_time": 3.142531156539917,
      "step": 9868
    },
    {
      "epoch": 3.76434326171875e-05,
      "step": 9868,
      "training_step_time": 10.34034538269043
    },
    {
      "epoch": 3.7647247314453124e-05,
      "model_forward_time": 2.4470713138580322,
      "step": 9869
    },
    {
      "epoch": 3.7647247314453124e-05,
      "step": 9869,
      "training_step_time": 7.478885650634766
    },
    {
      "epoch": 3.7647247314453124e-05,
      "model_forward_time": 2.4474241733551025,
      "step": 9869
    },
    {
      "epoch": 3.7647247314453124e-05,
      "step": 9869,
      "training_step_time": 7.488423585891724
    },
    {
      "epoch": 3.7647247314453124e-05,
      "model_forward_time": 3.162243604660034,
      "step": 9869
    },
    {
      "epoch": 3.7647247314453124e-05,
      "step": 9869,
      "training_step_time": 9.940317153930664
    },
    {
      "epoch": 3.7647247314453124e-05,
      "model_forward_time": 2.444474697113037,
      "step": 9869
    },
    {
      "epoch": 3.7647247314453124e-05,
      "step": 9869,
      "training_step_time": 10.300068140029907
    },
    {
      "epoch": 3.765106201171875e-05,
      "grad_norm": 0.31080445647239685,
      "learning_rate": 1e-05,
      "loss": 0.0783,
      "step": 9870
    },
    {
      "epoch": 3.765106201171875e-05,
      "model_forward_time": 3.1529879570007324,
      "step": 9870
    },
    {
      "dynamics_loss_avg": 0.05745410267263651,
      "epoch": 3.765106201171875e-05,
      "step": 9870
    },
    {
      "action_loss_avg": 0.019085253309458493,
      "epoch": 3.765106201171875e-05,
      "step": 9870
    },
    {
      "epoch": 3.765106201171875e-05,
      "step": 9870,
      "training_step_time": 9.933573961257935
    },
    {
      "epoch": 3.765106201171875e-05,
      "model_forward_time": 2.77248477935791,
      "step": 9870
    },
    {
      "epoch": 3.765106201171875e-05,
      "step": 9870,
      "training_step_time": 7.808877229690552
    },
    {
      "epoch": 3.765106201171875e-05,
      "model_forward_time": 3.1478111743927,
      "step": 9870
    },
    {
      "epoch": 3.765106201171875e-05,
      "step": 9870,
      "training_step_time": 9.921984434127808
    },
    {
      "epoch": 3.765106201171875e-05,
      "model_forward_time": 2.4485671520233154,
      "step": 9870
    },
    {
      "epoch": 3.765106201171875e-05,
      "step": 9870,
      "training_step_time": 10.030452728271484
    },
    {
      "epoch": 3.765487670898438e-05,
      "model_forward_time": 2.4545040130615234,
      "step": 9871
    },
    {
      "epoch": 3.765487670898438e-05,
      "step": 9871,
      "training_step_time": 7.4856040477752686
    },
    {
      "epoch": 3.765487670898438e-05,
      "model_forward_time": 2.4401607513427734,
      "step": 9871
    },
    {
      "epoch": 3.765487670898438e-05,
      "step": 9871,
      "training_step_time": 7.472200393676758
    },
    {
      "epoch": 3.765487670898438e-05,
      "model_forward_time": 3.475808620452881,
      "step": 9871
    },
    {
      "epoch": 3.765487670898438e-05,
      "step": 9871,
      "training_step_time": 10.251404285430908
    },
    {
      "epoch": 3.765487670898438e-05,
      "model_forward_time": 2.4420621395111084,
      "step": 9871
    },
    {
      "epoch": 3.765487670898438e-05,
      "step": 9871,
      "training_step_time": 10.289483547210693
    },
    {
      "epoch": 3.765869140625e-05,
      "model_forward_time": 3.1674346923828125,
      "step": 9872
    },
    {
      "epoch": 3.765869140625e-05,
      "step": 9872,
      "training_step_time": 9.950632810592651
    },
    {
      "epoch": 3.765869140625e-05,
      "model_forward_time": 2.4441676139831543,
      "step": 9872
    },
    {
      "epoch": 3.765869140625e-05,
      "step": 9872,
      "training_step_time": 7.475483655929565
    },
    {
      "epoch": 3.765869140625e-05,
      "model_forward_time": 3.147630453109741,
      "step": 9872
    },
    {
      "dynamics_loss_avg": 0.038916324730962515,
      "epoch": 3.765869140625e-05,
      "step": 9872
    },
    {
      "action_loss_avg": 0.013314370019361376,
      "epoch": 3.765869140625e-05,
      "step": 9872
    },
    {
      "epoch": 3.765869140625e-05,
      "step": 9872,
      "training_step_time": 9.923603296279907
    },
    {
      "epoch": 3.765869140625e-05,
      "model_forward_time": 2.4405617713928223,
      "step": 9872
    },
    {
      "epoch": 3.765869140625e-05,
      "step": 9872,
      "training_step_time": 10.275165557861328
    },
    {
      "epoch": 3.766250610351562e-05,
      "model_forward_time": 3.479015827178955,
      "step": 9873
    },
    {
      "epoch": 3.766250610351562e-05,
      "step": 9873,
      "training_step_time": 10.247160196304321
    },
    {
      "epoch": 3.766250610351562e-05,
      "model_forward_time": 3.145930290222168,
      "step": 9873
    },
    {
      "epoch": 3.766250610351562e-05,
      "step": 9873,
      "training_step_time": 9.914312601089478
    },
    {
      "epoch": 3.766250610351562e-05,
      "model_forward_time": 2.436952590942383,
      "step": 9873
    },
    {
      "epoch": 3.766250610351562e-05,
      "step": 9873,
      "training_step_time": 7.4610755443573
    },
    {
      "epoch": 3.766250610351562e-05,
      "model_forward_time": 3.1387791633605957,
      "step": 9873
    },
    {
      "epoch": 3.766250610351562e-05,
      "step": 9873,
      "training_step_time": 10.277719020843506
    },
    {
      "epoch": 3.766632080078125e-05,
      "model_forward_time": 2.457730770111084,
      "step": 9874
    },
    {
      "epoch": 3.766632080078125e-05,
      "step": 9874,
      "training_step_time": 7.487653970718384
    },
    {
      "epoch": 3.766632080078125e-05,
      "model_forward_time": 2.727588653564453,
      "step": 9874
    },
    {
      "epoch": 3.766632080078125e-05,
      "step": 9874,
      "training_step_time": 7.749700307846069
    },
    {
      "epoch": 3.766632080078125e-05,
      "model_forward_time": 3.1381735801696777,
      "step": 9874
    },
    {
      "epoch": 3.766632080078125e-05,
      "step": 9874,
      "training_step_time": 9.910379886627197
    },
    {
      "epoch": 3.766632080078125e-05,
      "model_forward_time": 2.434173107147217,
      "step": 9874
    },
    {
      "epoch": 3.766632080078125e-05,
      "step": 9874,
      "training_step_time": 10.00085163116455
    },
    {
      "epoch": 3.7670135498046875e-05,
      "model_forward_time": 1.761519193649292,
      "step": 9875
    },
    {
      "dynamics_loss_avg": 0.05960047002881765,
      "epoch": 3.7670135498046875e-05,
      "step": 9875
    },
    {
      "action_loss_avg": 0.022159140650182964,
      "epoch": 3.7670135498046875e-05,
      "step": 9875
    },
    {
      "epoch": 3.7670135498046875e-05,
      "step": 9875,
      "training_step_time": 5.213782072067261
    },
    {
      "epoch": 3.7670135498046875e-05,
      "model_forward_time": 2.4337563514709473,
      "step": 9875
    },
    {
      "epoch": 3.7670135498046875e-05,
      "step": 9875,
      "training_step_time": 7.455217361450195
    },
    {
      "epoch": 3.7670135498046875e-05,
      "model_forward_time": 3.4353718757629395,
      "step": 9875
    },
    {
      "epoch": 3.7670135498046875e-05,
      "step": 9875,
      "training_step_time": 10.200089931488037
    },
    {
      "epoch": 3.7670135498046875e-05,
      "model_forward_time": 2.433608055114746,
      "step": 9875
    },
    {
      "epoch": 3.7670135498046875e-05,
      "step": 9875,
      "training_step_time": 10.281676054000854
    },
    {
      "epoch": 3.76739501953125e-05,
      "model_forward_time": 3.1605801582336426,
      "step": 9876
    },
    {
      "epoch": 3.76739501953125e-05,
      "step": 9876,
      "training_step_time": 9.922683715820312
    },
    {
      "epoch": 3.76739501953125e-05,
      "model_forward_time": 2.4364013671875,
      "step": 9876
    },
    {
      "epoch": 3.76739501953125e-05,
      "step": 9876,
      "training_step_time": 7.4539220333099365
    },
    {
      "epoch": 3.76739501953125e-05,
      "model_forward_time": 2.434394359588623,
      "step": 9876
    },
    {
      "epoch": 3.76739501953125e-05,
      "step": 9876,
      "training_step_time": 7.454878091812134
    },
    {
      "epoch": 3.76739501953125e-05,
      "model_forward_time": 2.4315152168273926,
      "step": 9876
    },
    {
      "epoch": 3.76739501953125e-05,
      "step": 9876,
      "training_step_time": 10.232820987701416
    },
    {
      "epoch": 3.767776489257813e-05,
      "model_forward_time": 2.753605365753174,
      "step": 9877
    },
    {
      "epoch": 3.767776489257813e-05,
      "step": 9877,
      "training_step_time": 7.776705741882324
    },
    {
      "epoch": 3.767776489257813e-05,
      "model_forward_time": 2.4338393211364746,
      "step": 9877
    },
    {
      "epoch": 3.767776489257813e-05,
      "step": 9877,
      "training_step_time": 7.454807996749878
    },
    {
      "epoch": 3.767776489257813e-05,
      "model_forward_time": 2.435533046722412,
      "step": 9877
    },
    {
      "dynamics_loss_avg": 0.06310095097869635,
      "epoch": 3.767776489257813e-05,
      "step": 9877
    },
    {
      "action_loss_avg": 0.018531749211251735,
      "epoch": 3.767776489257813e-05,
      "step": 9877
    },
    {
      "epoch": 3.767776489257813e-05,
      "step": 9877,
      "training_step_time": 7.4603376388549805
    },
    {
      "epoch": 3.767776489257813e-05,
      "model_forward_time": 3.140707492828369,
      "step": 9877
    },
    {
      "epoch": 3.767776489257813e-05,
      "step": 9877,
      "training_step_time": 10.252230167388916
    },
    {
      "epoch": 3.768157958984375e-05,
      "model_forward_time": 3.1526639461517334,
      "step": 9878
    },
    {
      "epoch": 3.768157958984375e-05,
      "step": 9878,
      "training_step_time": 9.92542576789856
    },
    {
      "epoch": 3.768157958984375e-05,
      "model_forward_time": 2.7163541316986084,
      "step": 9878
    },
    {
      "epoch": 3.768157958984375e-05,
      "step": 9878,
      "training_step_time": 7.746328830718994
    },
    {
      "epoch": 3.768157958984375e-05,
      "model_forward_time": 3.1482009887695312,
      "step": 9878
    },
    {
      "epoch": 3.768157958984375e-05,
      "step": 9878,
      "training_step_time": 9.93053388595581
    },
    {
      "epoch": 3.768157958984375e-05,
      "model_forward_time": 1.7513892650604248,
      "step": 9878
    },
    {
      "epoch": 3.768157958984375e-05,
      "step": 9878,
      "training_step_time": 10.303447484970093
    },
    {
      "epoch": 3.768539428710937e-05,
      "model_forward_time": 2.4591846466064453,
      "step": 9879
    },
    {
      "epoch": 3.768539428710937e-05,
      "step": 9879,
      "training_step_time": 7.4859724044799805
    },
    {
      "epoch": 3.768539428710937e-05,
      "model_forward_time": 3.145704984664917,
      "step": 9879
    },
    {
      "epoch": 3.768539428710937e-05,
      "step": 9879,
      "training_step_time": 9.916396617889404
    },
    {
      "epoch": 3.768539428710937e-05,
      "model_forward_time": 2.4405341148376465,
      "step": 9879
    },
    {
      "epoch": 3.768539428710937e-05,
      "step": 9879,
      "training_step_time": 7.466532945632935
    },
    {
      "epoch": 3.768539428710937e-05,
      "model_forward_time": 2.763493776321411,
      "step": 9879
    },
    {
      "epoch": 3.768539428710937e-05,
      "step": 9879,
      "training_step_time": 10.019639730453491
    },
    {
      "epoch": 3.7689208984375e-05,
      "grad_norm": 0.3450145125389099,
      "learning_rate": 1e-05,
      "loss": 0.0783,
      "step": 9880
    },
    {
      "epoch": 3.7689208984375e-05,
      "model_forward_time": 3.1597471237182617,
      "step": 9880
    },
    {
      "dynamics_loss_avg": 0.05210030451416969,
      "epoch": 3.7689208984375e-05,
      "step": 9880
    },
    {
      "action_loss_avg": 0.016379275964573024,
      "epoch": 3.7689208984375e-05,
      "step": 9880
    },
    {
      "epoch": 3.7689208984375e-05,
      "step": 9880,
      "training_step_time": 9.951233625411987
    },
    {
      "epoch": 3.7689208984375e-05,
      "model_forward_time": 2.437739133834839,
      "step": 9880
    },
    {
      "epoch": 3.7689208984375e-05,
      "step": 9880,
      "training_step_time": 7.463823318481445
    },
    {
      "epoch": 3.7689208984375e-05,
      "model_forward_time": 2.427445888519287,
      "step": 9880
    },
    {
      "epoch": 3.7689208984375e-05,
      "step": 9880,
      "training_step_time": 7.451023817062378
    },
    {
      "epoch": 3.7689208984375e-05,
      "model_forward_time": 2.4313435554504395,
      "step": 9880
    },
    {
      "epoch": 3.7689208984375e-05,
      "step": 9880,
      "training_step_time": 10.260239601135254
    },
    {
      "epoch": 3.7693023681640626e-05,
      "model_forward_time": 2.4555797576904297,
      "step": 9881
    },
    {
      "epoch": 3.7693023681640626e-05,
      "step": 9881,
      "training_step_time": 7.808818340301514
    },
    {
      "epoch": 3.7693023681640626e-05,
      "model_forward_time": 3.1364917755126953,
      "step": 9881
    },
    {
      "epoch": 3.7693023681640626e-05,
      "step": 9881,
      "training_step_time": 9.904216766357422
    },
    {
      "epoch": 3.7693023681640626e-05,
      "model_forward_time": 2.4336113929748535,
      "step": 9881
    },
    {
      "epoch": 3.7693023681640626e-05,
      "step": 9881,
      "training_step_time": 7.456703424453735
    },
    {
      "epoch": 3.7693023681640626e-05,
      "model_forward_time": 2.4294238090515137,
      "step": 9881
    },
    {
      "epoch": 3.7693023681640626e-05,
      "step": 9881,
      "training_step_time": 10.28856348991394
    },
    {
      "epoch": 3.769683837890625e-05,
      "model_forward_time": 3.1497132778167725,
      "step": 9882
    },
    {
      "epoch": 3.769683837890625e-05,
      "step": 9882,
      "training_step_time": 9.9214506149292
    },
    {
      "epoch": 3.769683837890625e-05,
      "model_forward_time": 3.1328392028808594,
      "step": 9882
    },
    {
      "epoch": 3.769683837890625e-05,
      "step": 9882,
      "training_step_time": 10.177687644958496
    },
    {
      "epoch": 3.769683837890625e-05,
      "model_forward_time": 3.139525890350342,
      "step": 9882
    },
    {
      "dynamics_loss_avg": 0.0549586390145123,
      "epoch": 3.769683837890625e-05,
      "step": 9882
    },
    {
      "action_loss_avg": 0.019037839607335627,
      "epoch": 3.769683837890625e-05,
      "step": 9882
    },
    {
      "epoch": 3.769683837890625e-05,
      "step": 9882,
      "training_step_time": 9.920719146728516
    },
    {
      "epoch": 3.769683837890625e-05,
      "model_forward_time": 3.13907790184021,
      "step": 9882
    },
    {
      "epoch": 3.769683837890625e-05,
      "step": 9882,
      "training_step_time": 10.302129030227661
    },
    {
      "epoch": 3.770065307617187e-05,
      "model_forward_time": 3.1541073322296143,
      "step": 9883
    },
    {
      "epoch": 3.770065307617187e-05,
      "step": 9883,
      "training_step_time": 9.932316064834595
    },
    {
      "epoch": 3.770065307617187e-05,
      "model_forward_time": 3.1427698135375977,
      "step": 9883
    },
    {
      "epoch": 3.770065307617187e-05,
      "step": 9883,
      "training_step_time": 9.929592370986938
    },
    {
      "epoch": 3.770065307617187e-05,
      "model_forward_time": 2.4380714893341064,
      "step": 9883
    },
    {
      "epoch": 3.770065307617187e-05,
      "step": 9883,
      "training_step_time": 7.7825751304626465
    },
    {
      "epoch": 3.770065307617187e-05,
      "model_forward_time": 3.1397342681884766,
      "step": 9883
    },
    {
      "epoch": 3.770065307617187e-05,
      "step": 9883,
      "training_step_time": 10.290733098983765
    },
    {
      "epoch": 3.77044677734375e-05,
      "model_forward_time": 3.1523308753967285,
      "step": 9884
    },
    {
      "epoch": 3.77044677734375e-05,
      "step": 9884,
      "training_step_time": 9.93205976486206
    },
    {
      "epoch": 3.77044677734375e-05,
      "model_forward_time": 2.4360854625701904,
      "step": 9884
    },
    {
      "epoch": 3.77044677734375e-05,
      "step": 9884,
      "training_step_time": 7.472720623016357
    },
    {
      "epoch": 3.77044677734375e-05,
      "model_forward_time": 3.13849139213562,
      "step": 9884
    },
    {
      "epoch": 3.77044677734375e-05,
      "step": 9884,
      "training_step_time": 9.915099859237671
    },
    {
      "epoch": 3.77044677734375e-05,
      "model_forward_time": 2.7258307933807373,
      "step": 9884
    },
    {
      "epoch": 3.77044677734375e-05,
      "step": 9884,
      "training_step_time": 10.02434778213501
    },
    {
      "epoch": 3.7708282470703124e-05,
      "model_forward_time": 3.154886484146118,
      "step": 9885
    },
    {
      "dynamics_loss_avg": 0.05379925798624754,
      "epoch": 3.7708282470703124e-05,
      "step": 9885
    },
    {
      "action_loss_avg": 0.01808970207348466,
      "epoch": 3.7708282470703124e-05,
      "step": 9885
    },
    {
      "epoch": 3.7708282470703124e-05,
      "step": 9885,
      "training_step_time": 9.939489126205444
    },
    {
      "epoch": 3.7708282470703124e-05,
      "model_forward_time": 3.1409080028533936,
      "step": 9885
    },
    {
      "epoch": 3.7708282470703124e-05,
      "step": 9885,
      "training_step_time": 9.921631574630737
    },
    {
      "epoch": 3.7708282470703124e-05,
      "model_forward_time": 3.1466832160949707,
      "step": 9885
    },
    {
      "epoch": 3.7708282470703124e-05,
      "step": 9885,
      "training_step_time": 9.932495355606079
    },
    {
      "epoch": 3.7708282470703124e-05,
      "model_forward_time": 3.1480872631073,
      "step": 9885
    },
    {
      "epoch": 3.7708282470703124e-05,
      "step": 9885,
      "training_step_time": 10.330076217651367
    },
    {
      "epoch": 3.771209716796875e-05,
      "model_forward_time": 2.4537785053253174,
      "step": 9886
    },
    {
      "epoch": 3.771209716796875e-05,
      "step": 9886,
      "training_step_time": 7.489456415176392
    },
    {
      "epoch": 3.771209716796875e-05,
      "model_forward_time": 2.745450735092163,
      "step": 9886
    },
    {
      "epoch": 3.771209716796875e-05,
      "step": 9886,
      "training_step_time": 7.775118350982666
    },
    {
      "epoch": 3.771209716796875e-05,
      "model_forward_time": 2.4371511936187744,
      "step": 9886
    },
    {
      "epoch": 3.771209716796875e-05,
      "step": 9886,
      "training_step_time": 7.474492311477661
    },
    {
      "epoch": 3.771209716796875e-05,
      "model_forward_time": 2.431603193283081,
      "step": 9886
    },
    {
      "epoch": 3.771209716796875e-05,
      "step": 9886,
      "training_step_time": 10.251909494400024
    },
    {
      "epoch": 3.771591186523438e-05,
      "model_forward_time": 3.1636528968811035,
      "step": 9887
    },
    {
      "epoch": 3.771591186523438e-05,
      "step": 9887,
      "training_step_time": 9.944941282272339
    },
    {
      "epoch": 3.771591186523438e-05,
      "model_forward_time": 3.140977621078491,
      "step": 9887
    },
    {
      "epoch": 3.771591186523438e-05,
      "step": 9887,
      "training_step_time": 9.919076919555664
    },
    {
      "epoch": 3.771591186523438e-05,
      "model_forward_time": 2.7020153999328613,
      "step": 9887
    },
    {
      "dynamics_loss_avg": 0.06594037692993879,
      "epoch": 3.771591186523438e-05,
      "step": 9887
    },
    {
      "action_loss_avg": 0.023148267576470972,
      "epoch": 3.771591186523438e-05,
      "step": 9887
    },
    {
      "epoch": 3.771591186523438e-05,
      "step": 9887,
      "training_step_time": 7.734312534332275
    },
    {
      "epoch": 3.771591186523438e-05,
      "model_forward_time": 2.437342643737793,
      "step": 9887
    },
    {
      "epoch": 3.771591186523438e-05,
      "step": 9887,
      "training_step_time": 10.279328346252441
    },
    {
      "epoch": 3.77197265625e-05,
      "model_forward_time": 3.1573076248168945,
      "step": 9888
    },
    {
      "epoch": 3.77197265625e-05,
      "step": 9888,
      "training_step_time": 9.932784080505371
    },
    {
      "epoch": 3.77197265625e-05,
      "model_forward_time": 3.140554189682007,
      "step": 9888
    },
    {
      "epoch": 3.77197265625e-05,
      "step": 9888,
      "training_step_time": 9.916903495788574
    },
    {
      "epoch": 3.77197265625e-05,
      "model_forward_time": 2.4344308376312256,
      "step": 9888
    },
    {
      "epoch": 3.77197265625e-05,
      "step": 9888,
      "training_step_time": 7.4645469188690186
    },
    {
      "epoch": 3.77197265625e-05,
      "model_forward_time": 3.4029462337493896,
      "step": 9888
    },
    {
      "epoch": 3.77197265625e-05,
      "step": 9888,
      "training_step_time": 10.298521280288696
    },
    {
      "epoch": 3.772354125976562e-05,
      "model_forward_time": 3.1477115154266357,
      "step": 9889
    },
    {
      "epoch": 3.772354125976562e-05,
      "step": 9889,
      "training_step_time": 9.928038358688354
    },
    {
      "epoch": 3.772354125976562e-05,
      "model_forward_time": 3.1377665996551514,
      "step": 9889
    },
    {
      "epoch": 3.772354125976562e-05,
      "step": 9889,
      "training_step_time": 9.922735214233398
    },
    {
      "epoch": 3.772354125976562e-05,
      "model_forward_time": 3.142003297805786,
      "step": 9889
    },
    {
      "epoch": 3.772354125976562e-05,
      "step": 9889,
      "training_step_time": 9.929549217224121
    },
    {
      "epoch": 3.772354125976562e-05,
      "model_forward_time": 3.1488020420074463,
      "step": 9889
    },
    {
      "epoch": 3.772354125976562e-05,
      "step": 9889,
      "training_step_time": 10.29170036315918
    },
    {
      "epoch": 3.772735595703125e-05,
      "grad_norm": 0.4134679436683655,
      "learning_rate": 1e-05,
      "loss": 0.0794,
      "step": 9890
    },
    {
      "epoch": 3.772735595703125e-05,
      "model_forward_time": 3.465592622756958,
      "step": 9890
    },
    {
      "dynamics_loss_avg": 0.05274524502456188,
      "epoch": 3.772735595703125e-05,
      "step": 9890
    },
    {
      "action_loss_avg": 0.024790343036875127,
      "epoch": 3.772735595703125e-05,
      "step": 9890
    },
    {
      "epoch": 3.772735595703125e-05,
      "step": 9890,
      "training_step_time": 10.252049446105957
    },
    {
      "epoch": 3.772735595703125e-05,
      "model_forward_time": 3.151876926422119,
      "step": 9890
    },
    {
      "epoch": 3.772735595703125e-05,
      "step": 9890,
      "training_step_time": 9.934535503387451
    },
    {
      "epoch": 3.772735595703125e-05,
      "model_forward_time": 3.1509206295013428,
      "step": 9890
    },
    {
      "epoch": 3.772735595703125e-05,
      "step": 9890,
      "training_step_time": 9.932207822799683
    },
    {
      "epoch": 3.772735595703125e-05,
      "model_forward_time": 2.4440038204193115,
      "step": 9890
    },
    {
      "epoch": 3.772735595703125e-05,
      "step": 9890,
      "training_step_time": 10.283297061920166
    },
    {
      "epoch": 3.7731170654296875e-05,
      "model_forward_time": 2.4563722610473633,
      "step": 9891
    },
    {
      "epoch": 3.7731170654296875e-05,
      "step": 9891,
      "training_step_time": 7.487557888031006
    },
    {
      "epoch": 3.7731170654296875e-05,
      "model_forward_time": 3.4347949028015137,
      "step": 9891
    },
    {
      "epoch": 3.7731170654296875e-05,
      "step": 9891,
      "training_step_time": 10.20482587814331
    },
    {
      "epoch": 3.7731170654296875e-05,
      "model_forward_time": 2.437086343765259,
      "step": 9891
    },
    {
      "epoch": 3.7731170654296875e-05,
      "step": 9891,
      "training_step_time": 7.463568925857544
    },
    {
      "epoch": 3.7731170654296875e-05,
      "model_forward_time": 2.4376635551452637,
      "step": 9891
    },
    {
      "epoch": 3.7731170654296875e-05,
      "step": 9891,
      "training_step_time": 10.290433406829834
    },
    {
      "epoch": 3.77349853515625e-05,
      "model_forward_time": 3.156376838684082,
      "step": 9892
    },
    {
      "epoch": 3.77349853515625e-05,
      "step": 9892,
      "training_step_time": 9.92435336112976
    },
    {
      "epoch": 3.77349853515625e-05,
      "model_forward_time": 2.433091163635254,
      "step": 9892
    },
    {
      "epoch": 3.77349853515625e-05,
      "step": 9892,
      "training_step_time": 7.460737228393555
    },
    {
      "epoch": 3.77349853515625e-05,
      "model_forward_time": 3.1406617164611816,
      "step": 9892
    },
    {
      "dynamics_loss_avg": 0.06690534222871065,
      "epoch": 3.77349853515625e-05,
      "step": 9892
    },
    {
      "action_loss_avg": 0.021074491459876298,
      "epoch": 3.77349853515625e-05,
      "step": 9892
    },
    {
      "epoch": 3.77349853515625e-05,
      "step": 9892,
      "training_step_time": 9.913932800292969
    },
    {
      "epoch": 3.77349853515625e-05,
      "model_forward_time": 2.685842752456665,
      "step": 9892
    },
    {
      "epoch": 3.77349853515625e-05,
      "step": 9892,
      "training_step_time": 10.28019666671753
    },
    {
      "epoch": 3.773880004882813e-05,
      "model_forward_time": 3.1596293449401855,
      "step": 9893
    },
    {
      "epoch": 3.773880004882813e-05,
      "step": 9893,
      "training_step_time": 9.927424669265747
    },
    {
      "epoch": 3.773880004882813e-05,
      "model_forward_time": 3.136399745941162,
      "step": 9893
    },
    {
      "epoch": 3.773880004882813e-05,
      "step": 9893,
      "training_step_time": 9.907516479492188
    },
    {
      "epoch": 3.773880004882813e-05,
      "model_forward_time": 3.139381170272827,
      "step": 9893
    },
    {
      "epoch": 3.773880004882813e-05,
      "step": 9893,
      "training_step_time": 9.904847145080566
    },
    {
      "epoch": 3.773880004882813e-05,
      "model_forward_time": 1.7466208934783936,
      "step": 9893
    },
    {
      "epoch": 3.773880004882813e-05,
      "step": 9893,
      "training_step_time": 10.334703207015991
    },
    {
      "epoch": 3.774261474609375e-05,
      "model_forward_time": 3.44217586517334,
      "step": 9894
    },
    {
      "epoch": 3.774261474609375e-05,
      "step": 9894,
      "training_step_time": 10.207048416137695
    },
    {
      "epoch": 3.774261474609375e-05,
      "model_forward_time": 2.4364209175109863,
      "step": 9894
    },
    {
      "epoch": 3.774261474609375e-05,
      "step": 9894,
      "training_step_time": 7.45894980430603
    },
    {
      "epoch": 3.774261474609375e-05,
      "model_forward_time": 2.434126138687134,
      "step": 9894
    },
    {
      "epoch": 3.774261474609375e-05,
      "step": 9894,
      "training_step_time": 7.455012559890747
    },
    {
      "epoch": 3.774261474609375e-05,
      "model_forward_time": 2.434932231903076,
      "step": 9894
    },
    {
      "epoch": 3.774261474609375e-05,
      "step": 9894,
      "training_step_time": 10.290387630462646
    },
    {
      "epoch": 3.774642944335937e-05,
      "model_forward_time": 3.147566556930542,
      "step": 9895
    },
    {
      "dynamics_loss_avg": 0.05506316749379039,
      "epoch": 3.774642944335937e-05,
      "step": 9895
    },
    {
      "action_loss_avg": 0.019912686850875615,
      "epoch": 3.774642944335937e-05,
      "step": 9895
    },
    {
      "epoch": 3.774642944335937e-05,
      "step": 9895,
      "training_step_time": 9.92157793045044
    },
    {
      "epoch": 3.774642944335937e-05,
      "model_forward_time": 2.6976265907287598,
      "step": 9895
    },
    {
      "epoch": 3.774642944335937e-05,
      "step": 9895,
      "training_step_time": 7.722174167633057
    },
    {
      "epoch": 3.774642944335937e-05,
      "model_forward_time": 2.436117649078369,
      "step": 9895
    },
    {
      "epoch": 3.774642944335937e-05,
      "step": 9895,
      "training_step_time": 7.456995487213135
    },
    {
      "epoch": 3.774642944335937e-05,
      "model_forward_time": 2.4327921867370605,
      "step": 9895
    },
    {
      "epoch": 3.774642944335937e-05,
      "step": 9895,
      "training_step_time": 10.258466005325317
    },
    {
      "epoch": 3.7750244140625e-05,
      "model_forward_time": 3.1390130519866943,
      "step": 9896
    },
    {
      "epoch": 3.7750244140625e-05,
      "step": 9896,
      "training_step_time": 9.904389381408691
    },
    {
      "epoch": 3.7750244140625e-05,
      "model_forward_time": 2.4266469478607178,
      "step": 9896
    },
    {
      "epoch": 3.7750244140625e-05,
      "step": 9896,
      "training_step_time": 7.4455602169036865
    },
    {
      "epoch": 3.7750244140625e-05,
      "model_forward_time": 3.1256368160247803,
      "step": 9896
    },
    {
      "epoch": 3.7750244140625e-05,
      "step": 9896,
      "training_step_time": 9.881979703903198
    },
    {
      "epoch": 3.7750244140625e-05,
      "model_forward_time": 2.729980945587158,
      "step": 9896
    },
    {
      "epoch": 3.7750244140625e-05,
      "step": 9896,
      "training_step_time": 10.232296228408813
    },
    {
      "epoch": 3.7754058837890626e-05,
      "model_forward_time": 2.4485955238342285,
      "step": 9897
    },
    {
      "epoch": 3.7754058837890626e-05,
      "step": 9897,
      "training_step_time": 7.464908123016357
    },
    {
      "epoch": 3.7754058837890626e-05,
      "model_forward_time": 3.1353931427001953,
      "step": 9897
    },
    {
      "epoch": 3.7754058837890626e-05,
      "step": 9897,
      "training_step_time": 9.892923831939697
    },
    {
      "epoch": 3.7754058837890626e-05,
      "model_forward_time": 2.430617094039917,
      "step": 9897
    },
    {
      "dynamics_loss_avg": 0.055941525101661685,
      "epoch": 3.7754058837890626e-05,
      "step": 9897
    },
    {
      "action_loss_avg": 0.014075076207518578,
      "epoch": 3.7754058837890626e-05,
      "step": 9897
    },
    {
      "epoch": 3.7754058837890626e-05,
      "step": 9897,
      "training_step_time": 7.451833724975586
    },
    {
      "epoch": 3.7754058837890626e-05,
      "model_forward_time": 2.4317169189453125,
      "step": 9897
    },
    {
      "epoch": 3.7754058837890626e-05,
      "step": 9897,
      "training_step_time": 10.283937692642212
    },
    {
      "epoch": 3.775787353515625e-05,
      "model_forward_time": 3.40694260597229,
      "step": 9898
    },
    {
      "epoch": 3.775787353515625e-05,
      "step": 9898,
      "training_step_time": 10.174287557601929
    },
    {
      "epoch": 3.775787353515625e-05,
      "model_forward_time": 3.1300272941589355,
      "step": 9898
    },
    {
      "epoch": 3.775787353515625e-05,
      "step": 9898,
      "training_step_time": 9.894048690795898
    },
    {
      "epoch": 3.775787353515625e-05,
      "model_forward_time": 3.1307194232940674,
      "step": 9898
    },
    {
      "epoch": 3.775787353515625e-05,
      "step": 9898,
      "training_step_time": 9.892784833908081
    },
    {
      "epoch": 3.775787353515625e-05,
      "model_forward_time": 2.4310715198516846,
      "step": 9898
    },
    {
      "epoch": 3.775787353515625e-05,
      "step": 9898,
      "training_step_time": 10.264972925186157
    },
    {
      "epoch": 3.776168823242187e-05,
      "model_forward_time": 3.1437909603118896,
      "step": 9899
    },
    {
      "epoch": 3.776168823242187e-05,
      "step": 9899,
      "training_step_time": 9.904456377029419
    },
    {
      "epoch": 3.776168823242187e-05,
      "model_forward_time": 3.4024269580841064,
      "step": 9899
    },
    {
      "epoch": 3.776168823242187e-05,
      "step": 9899,
      "training_step_time": 10.168318271636963
    },
    {
      "epoch": 3.776168823242187e-05,
      "model_forward_time": 3.1328978538513184,
      "step": 9899
    },
    {
      "epoch": 3.776168823242187e-05,
      "step": 9899,
      "training_step_time": 9.905616283416748
    },
    {
      "epoch": 3.776168823242187e-05,
      "model_forward_time": 3.1468193531036377,
      "step": 9899
    },
    {
      "epoch": 3.776168823242187e-05,
      "step": 9899,
      "training_step_time": 10.318113088607788
    },
    {
      "epoch": 3.77655029296875e-05,
      "grad_norm": 0.3799354135990143,
      "learning_rate": 1e-05,
      "loss": 0.0755,
      "step": 9900
    },
    {
      "epoch": 3.77655029296875e-05,
      "model_forward_time": 3.164945602416992,
      "step": 9900
    },
    {
      "dynamics_loss_avg": 0.05416015721857548,
      "epoch": 3.77655029296875e-05,
      "step": 9900
    },
    {
      "action_loss_avg": 0.018020513374358414,
      "epoch": 3.77655029296875e-05,
      "step": 9900
    },
    {
      "epoch": 3.77655029296875e-05,
      "step": 9900,
      "training_step_time": 9.945013761520386
    },
    {
      "epoch": 3.77655029296875e-05,
      "model_forward_time": 3.1465015411376953,
      "step": 9900
    },
    {
      "epoch": 3.77655029296875e-05,
      "step": 9900,
      "training_step_time": 9.924094200134277
    },
    {
      "epoch": 3.77655029296875e-05,
      "model_forward_time": 3.407888412475586,
      "step": 9900
    },
    {
      "epoch": 3.77655029296875e-05,
      "step": 9900,
      "training_step_time": 10.18620491027832
    },
    {
      "epoch": 3.77655029296875e-05,
      "model_forward_time": 3.146358013153076,
      "step": 9900
    },
    {
      "epoch": 3.77655029296875e-05,
      "step": 9900,
      "training_step_time": 10.326371669769287
    },
    {
      "epoch": 3.7769317626953124e-05,
      "model_forward_time": 2.459157943725586,
      "step": 9901
    },
    {
      "epoch": 3.7769317626953124e-05,
      "step": 9901,
      "training_step_time": 7.490867376327515
    },
    {
      "epoch": 3.7769317626953124e-05,
      "model_forward_time": 3.145047426223755,
      "step": 9901
    },
    {
      "epoch": 3.7769317626953124e-05,
      "step": 9901,
      "training_step_time": 9.918481349945068
    },
    {
      "epoch": 3.7769317626953124e-05,
      "model_forward_time": 3.1467769145965576,
      "step": 9901
    },
    {
      "epoch": 3.7769317626953124e-05,
      "step": 9901,
      "training_step_time": 9.920702695846558
    },
    {
      "epoch": 3.7769317626953124e-05,
      "model_forward_time": 2.729422092437744,
      "step": 9901
    },
    {
      "epoch": 3.7769317626953124e-05,
      "step": 9901,
      "training_step_time": 10.296707153320312
    },
    {
      "epoch": 3.777313232421875e-05,
      "model_forward_time": 2.457714080810547,
      "step": 9902
    },
    {
      "epoch": 3.777313232421875e-05,
      "step": 9902,
      "training_step_time": 7.485212087631226
    },
    {
      "epoch": 3.777313232421875e-05,
      "model_forward_time": 3.1428885459899902,
      "step": 9902
    },
    {
      "epoch": 3.777313232421875e-05,
      "step": 9902,
      "training_step_time": 9.90811824798584
    },
    {
      "epoch": 3.777313232421875e-05,
      "model_forward_time": 3.1431171894073486,
      "step": 9902
    },
    {
      "dynamics_loss_avg": 0.05175284445285797,
      "epoch": 3.777313232421875e-05,
      "step": 9902
    },
    {
      "action_loss_avg": 0.016310870787128805,
      "epoch": 3.777313232421875e-05,
      "step": 9902
    },
    {
      "epoch": 3.777313232421875e-05,
      "step": 9902,
      "training_step_time": 9.911487340927124
    },
    {
      "epoch": 3.777313232421875e-05,
      "model_forward_time": 2.437760829925537,
      "step": 9902
    },
    {
      "epoch": 3.777313232421875e-05,
      "step": 9902,
      "training_step_time": 10.015707731246948
    },
    {
      "epoch": 3.777694702148438e-05,
      "model_forward_time": 2.4441006183624268,
      "step": 9903
    },
    {
      "epoch": 3.777694702148438e-05,
      "step": 9903,
      "training_step_time": 7.461337089538574
    },
    {
      "epoch": 3.777694702148438e-05,
      "model_forward_time": 3.390519618988037,
      "step": 9903
    },
    {
      "epoch": 3.777694702148438e-05,
      "step": 9903,
      "training_step_time": 10.152862071990967
    },
    {
      "epoch": 3.777694702148438e-05,
      "model_forward_time": 3.1400368213653564,
      "step": 9903
    },
    {
      "epoch": 3.777694702148438e-05,
      "step": 9903,
      "training_step_time": 9.902459621429443
    },
    {
      "epoch": 3.777694702148438e-05,
      "model_forward_time": 3.1362462043762207,
      "step": 9903
    },
    {
      "epoch": 3.777694702148438e-05,
      "step": 9903,
      "training_step_time": 10.012332677841187
    },
    {
      "epoch": 3.778076171875e-05,
      "model_forward_time": 1.7695159912109375,
      "step": 9904
    },
    {
      "epoch": 3.778076171875e-05,
      "step": 9904,
      "training_step_time": 5.231017589569092
    },
    {
      "epoch": 3.778076171875e-05,
      "model_forward_time": 2.434673547744751,
      "step": 9904
    },
    {
      "epoch": 3.778076171875e-05,
      "step": 9904,
      "training_step_time": 7.457028150558472
    },
    {
      "epoch": 3.778076171875e-05,
      "model_forward_time": 2.00772762298584,
      "step": 9904
    },
    {
      "epoch": 3.778076171875e-05,
      "step": 9904,
      "training_step_time": 5.461505174636841
    },
    {
      "epoch": 3.778076171875e-05,
      "model_forward_time": 2.4333245754241943,
      "step": 9904
    },
    {
      "epoch": 3.778076171875e-05,
      "step": 9904,
      "training_step_time": 10.302860736846924
    },
    {
      "epoch": 3.778457641601562e-05,
      "model_forward_time": 2.4580466747283936,
      "step": 9905
    },
    {
      "dynamics_loss_avg": 0.06156529160216451,
      "epoch": 3.778457641601562e-05,
      "step": 9905
    },
    {
      "action_loss_avg": 0.01750202294206247,
      "epoch": 3.778457641601562e-05,
      "step": 9905
    },
    {
      "epoch": 3.778457641601562e-05,
      "step": 9905,
      "training_step_time": 7.483185768127441
    },
    {
      "epoch": 3.778457641601562e-05,
      "model_forward_time": 3.138962745666504,
      "step": 9905
    },
    {
      "epoch": 3.778457641601562e-05,
      "step": 9905,
      "training_step_time": 9.90836238861084
    },
    {
      "epoch": 3.778457641601562e-05,
      "model_forward_time": 2.4353604316711426,
      "step": 9905
    },
    {
      "epoch": 3.778457641601562e-05,
      "step": 9905,
      "training_step_time": 7.461256742477417
    },
    {
      "epoch": 3.778457641601562e-05,
      "model_forward_time": 2.711552381515503,
      "step": 9905
    },
    {
      "epoch": 3.778457641601562e-05,
      "step": 9905,
      "training_step_time": 10.286868333816528
    },
    {
      "epoch": 3.778839111328125e-05,
      "model_forward_time": 2.454210042953491,
      "step": 9906
    },
    {
      "epoch": 3.778839111328125e-05,
      "step": 9906,
      "training_step_time": 7.475674390792847
    },
    {
      "epoch": 3.778839111328125e-05,
      "model_forward_time": 3.138066053390503,
      "step": 9906
    },
    {
      "epoch": 3.778839111328125e-05,
      "step": 9906,
      "training_step_time": 9.90252685546875
    },
    {
      "epoch": 3.778839111328125e-05,
      "model_forward_time": 3.1379096508026123,
      "step": 9906
    },
    {
      "epoch": 3.778839111328125e-05,
      "step": 9906,
      "training_step_time": 9.910150051116943
    },
    {
      "epoch": 3.778839111328125e-05,
      "model_forward_time": 2.4377951622009277,
      "step": 9906
    },
    {
      "epoch": 3.778839111328125e-05,
      "step": 9906,
      "training_step_time": 10.283414363861084
    },
    {
      "epoch": 3.7792205810546875e-05,
      "model_forward_time": 3.147307872772217,
      "step": 9907
    },
    {
      "epoch": 3.7792205810546875e-05,
      "step": 9907,
      "training_step_time": 9.928911209106445
    },
    {
      "epoch": 3.7792205810546875e-05,
      "model_forward_time": 2.738847494125366,
      "step": 9907
    },
    {
      "epoch": 3.7792205810546875e-05,
      "step": 9907,
      "training_step_time": 7.772639274597168
    },
    {
      "epoch": 3.7792205810546875e-05,
      "model_forward_time": 2.43135666847229,
      "step": 9907
    },
    {
      "dynamics_loss_avg": 0.057415569014847276,
      "epoch": 3.7792205810546875e-05,
      "step": 9907
    },
    {
      "action_loss_avg": 0.018013285379856823,
      "epoch": 3.7792205810546875e-05,
      "step": 9907
    },
    {
      "epoch": 3.7792205810546875e-05,
      "step": 9907,
      "training_step_time": 7.461097478866577
    },
    {
      "epoch": 3.7792205810546875e-05,
      "model_forward_time": 2.437570095062256,
      "step": 9907
    },
    {
      "epoch": 3.7792205810546875e-05,
      "step": 9907,
      "training_step_time": 10.279596328735352
    },
    {
      "epoch": 3.77960205078125e-05,
      "model_forward_time": 3.160621166229248,
      "step": 9908
    },
    {
      "epoch": 3.77960205078125e-05,
      "step": 9908,
      "training_step_time": 9.92914366722107
    },
    {
      "epoch": 3.77960205078125e-05,
      "model_forward_time": 3.1391055583953857,
      "step": 9908
    },
    {
      "epoch": 3.77960205078125e-05,
      "step": 9908,
      "training_step_time": 9.90111517906189
    },
    {
      "epoch": 3.77960205078125e-05,
      "model_forward_time": 3.4054698944091797,
      "step": 9908
    },
    {
      "epoch": 3.77960205078125e-05,
      "step": 9908,
      "training_step_time": 10.16992974281311
    },
    {
      "epoch": 3.77960205078125e-05,
      "model_forward_time": 3.140049457550049,
      "step": 9908
    },
    {
      "epoch": 3.77960205078125e-05,
      "step": 9908,
      "training_step_time": 10.30783224105835
    },
    {
      "epoch": 3.779983520507813e-05,
      "model_forward_time": 2.449430465698242,
      "step": 9909
    },
    {
      "epoch": 3.779983520507813e-05,
      "step": 9909,
      "training_step_time": 7.477047443389893
    },
    {
      "epoch": 3.779983520507813e-05,
      "model_forward_time": 3.1423697471618652,
      "step": 9909
    },
    {
      "epoch": 3.779983520507813e-05,
      "step": 9909,
      "training_step_time": 9.911522388458252
    },
    {
      "epoch": 3.779983520507813e-05,
      "model_forward_time": 3.1353530883789062,
      "step": 9909
    },
    {
      "epoch": 3.779983520507813e-05,
      "step": 9909,
      "training_step_time": 9.903326034545898
    },
    {
      "epoch": 3.779983520507813e-05,
      "model_forward_time": 2.71244740486145,
      "step": 9909
    },
    {
      "epoch": 3.779983520507813e-05,
      "step": 9909,
      "training_step_time": 10.269200086593628
    },
    {
      "epoch": 3.780364990234375e-05,
      "grad_norm": 0.37763816118240356,
      "learning_rate": 1e-05,
      "loss": 0.0786,
      "step": 9910
    },
    {
      "epoch": 3.780364990234375e-05,
      "model_forward_time": 3.1543636322021484,
      "step": 9910
    },
    {
      "dynamics_loss_avg": 0.07295707706362009,
      "epoch": 3.780364990234375e-05,
      "step": 9910
    },
    {
      "action_loss_avg": 0.027041827188804747,
      "epoch": 3.780364990234375e-05,
      "step": 9910
    },
    {
      "epoch": 3.780364990234375e-05,
      "step": 9910,
      "training_step_time": 9.921985149383545
    },
    {
      "epoch": 3.780364990234375e-05,
      "model_forward_time": 3.1328725814819336,
      "step": 9910
    },
    {
      "epoch": 3.780364990234375e-05,
      "step": 9910,
      "training_step_time": 9.89775800704956
    },
    {
      "epoch": 3.780364990234375e-05,
      "model_forward_time": 3.136650800704956,
      "step": 9910
    },
    {
      "epoch": 3.780364990234375e-05,
      "step": 9910,
      "training_step_time": 9.904083967208862
    },
    {
      "epoch": 3.780364990234375e-05,
      "model_forward_time": 3.137380838394165,
      "step": 9910
    },
    {
      "epoch": 3.780364990234375e-05,
      "step": 9910,
      "training_step_time": 10.29970932006836
    },
    {
      "epoch": 3.7807464599609373e-05,
      "model_forward_time": 2.7419769763946533,
      "step": 9911
    },
    {
      "epoch": 3.7807464599609373e-05,
      "step": 9911,
      "training_step_time": 7.784194231033325
    },
    {
      "epoch": 3.7807464599609373e-05,
      "model_forward_time": 3.138495922088623,
      "step": 9911
    },
    {
      "epoch": 3.7807464599609373e-05,
      "step": 9911,
      "training_step_time": 9.914658069610596
    },
    {
      "epoch": 3.7807464599609373e-05,
      "model_forward_time": 3.147111177444458,
      "step": 9911
    },
    {
      "epoch": 3.7807464599609373e-05,
      "step": 9911,
      "training_step_time": 9.925379514694214
    },
    {
      "epoch": 3.7807464599609373e-05,
      "model_forward_time": 3.1409566402435303,
      "step": 9911
    },
    {
      "epoch": 3.7807464599609373e-05,
      "step": 9911,
      "training_step_time": 10.285899639129639
    },
    {
      "epoch": 3.7811279296875e-05,
      "model_forward_time": 3.1597886085510254,
      "step": 9912
    },
    {
      "epoch": 3.7811279296875e-05,
      "step": 9912,
      "training_step_time": 9.947528839111328
    },
    {
      "epoch": 3.7811279296875e-05,
      "model_forward_time": 2.7686166763305664,
      "step": 9912
    },
    {
      "epoch": 3.7811279296875e-05,
      "step": 9912,
      "training_step_time": 7.800416946411133
    },
    {
      "epoch": 3.7811279296875e-05,
      "model_forward_time": 1.7480359077453613,
      "step": 9912
    },
    {
      "dynamics_loss_avg": 0.06277160868048667,
      "epoch": 3.7811279296875e-05,
      "step": 9912
    },
    {
      "action_loss_avg": 0.019860256649553775,
      "epoch": 3.7811279296875e-05,
      "step": 9912
    },
    {
      "epoch": 3.7811279296875e-05,
      "step": 9912,
      "training_step_time": 5.213451623916626
    },
    {
      "epoch": 3.7811279296875e-05,
      "model_forward_time": 3.139655351638794,
      "step": 9912
    },
    {
      "epoch": 3.7811279296875e-05,
      "step": 9912,
      "training_step_time": 10.322780847549438
    },
    {
      "epoch": 3.7815093994140626e-05,
      "model_forward_time": 2.4569826126098633,
      "step": 9913
    },
    {
      "epoch": 3.7815093994140626e-05,
      "step": 9913,
      "training_step_time": 7.492018461227417
    },
    {
      "epoch": 3.7815093994140626e-05,
      "model_forward_time": 2.43817400932312,
      "step": 9913
    },
    {
      "epoch": 3.7815093994140626e-05,
      "step": 9913,
      "training_step_time": 7.479627847671509
    },
    {
      "epoch": 3.7815093994140626e-05,
      "model_forward_time": 2.4363491535186768,
      "step": 9913
    },
    {
      "epoch": 3.7815093994140626e-05,
      "step": 9913,
      "training_step_time": 7.466856956481934
    },
    {
      "epoch": 3.7815093994140626e-05,
      "model_forward_time": 2.7387518882751465,
      "step": 9913
    },
    {
      "epoch": 3.7815093994140626e-05,
      "step": 9913,
      "training_step_time": 10.02161955833435
    },
    {
      "epoch": 3.781890869140625e-05,
      "model_forward_time": 2.4567294120788574,
      "step": 9914
    },
    {
      "epoch": 3.781890869140625e-05,
      "step": 9914,
      "training_step_time": 7.497454643249512
    },
    {
      "epoch": 3.781890869140625e-05,
      "model_forward_time": 2.4350619316101074,
      "step": 9914
    },
    {
      "epoch": 3.781890869140625e-05,
      "step": 9914,
      "training_step_time": 7.466189861297607
    },
    {
      "epoch": 3.781890869140625e-05,
      "model_forward_time": 3.1392152309417725,
      "step": 9914
    },
    {
      "epoch": 3.781890869140625e-05,
      "step": 9914,
      "training_step_time": 9.91314172744751
    },
    {
      "epoch": 3.781890869140625e-05,
      "model_forward_time": 3.139709949493408,
      "step": 9914
    },
    {
      "epoch": 3.781890869140625e-05,
      "step": 9914,
      "training_step_time": 10.025555849075317
    },
    {
      "epoch": 3.782272338867187e-05,
      "model_forward_time": 3.468515396118164,
      "step": 9915
    },
    {
      "dynamics_loss_avg": 0.053393217734992504,
      "epoch": 3.782272338867187e-05,
      "step": 9915
    },
    {
      "action_loss_avg": 0.015900346077978612,
      "epoch": 3.782272338867187e-05,
      "step": 9915
    },
    {
      "epoch": 3.782272338867187e-05,
      "step": 9915,
      "training_step_time": 10.246324300765991
    },
    {
      "epoch": 3.782272338867187e-05,
      "model_forward_time": 2.4363832473754883,
      "step": 9915
    },
    {
      "epoch": 3.782272338867187e-05,
      "step": 9915,
      "training_step_time": 7.468410491943359
    },
    {
      "epoch": 3.782272338867187e-05,
      "model_forward_time": 2.4377501010894775,
      "step": 9915
    },
    {
      "epoch": 3.782272338867187e-05,
      "step": 9915,
      "training_step_time": 7.468275785446167
    },
    {
      "epoch": 3.782272338867187e-05,
      "model_forward_time": 2.4360501766204834,
      "step": 9915
    },
    {
      "epoch": 3.782272338867187e-05,
      "step": 9915,
      "training_step_time": 10.35057520866394
    },
    {
      "epoch": 3.78265380859375e-05,
      "model_forward_time": 3.1544928550720215,
      "step": 9916
    },
    {
      "epoch": 3.78265380859375e-05,
      "step": 9916,
      "training_step_time": 9.931648015975952
    },
    {
      "epoch": 3.78265380859375e-05,
      "model_forward_time": 2.437798500061035,
      "step": 9916
    },
    {
      "epoch": 3.78265380859375e-05,
      "step": 9916,
      "training_step_time": 7.47216534614563
    },
    {
      "epoch": 3.78265380859375e-05,
      "model_forward_time": 3.4401650428771973,
      "step": 9916
    },
    {
      "epoch": 3.78265380859375e-05,
      "step": 9916,
      "training_step_time": 10.214371919631958
    },
    {
      "epoch": 3.78265380859375e-05,
      "model_forward_time": 1.7485806941986084,
      "step": 9916
    },
    {
      "epoch": 3.78265380859375e-05,
      "step": 9916,
      "training_step_time": 10.332177639007568
    },
    {
      "epoch": 3.7830352783203124e-05,
      "model_forward_time": 2.4563658237457275,
      "step": 9917
    },
    {
      "epoch": 3.7830352783203124e-05,
      "step": 9917,
      "training_step_time": 7.489437818527222
    },
    {
      "epoch": 3.7830352783203124e-05,
      "model_forward_time": 2.4413206577301025,
      "step": 9917
    },
    {
      "epoch": 3.7830352783203124e-05,
      "step": 9917,
      "training_step_time": 7.470174789428711
    },
    {
      "epoch": 3.7830352783203124e-05,
      "model_forward_time": 3.143832206726074,
      "step": 9917
    },
    {
      "dynamics_loss_avg": 0.0715880973264575,
      "epoch": 3.7830352783203124e-05,
      "step": 9917
    },
    {
      "action_loss_avg": 0.0209622526075691,
      "epoch": 3.7830352783203124e-05,
      "step": 9917
    },
    {
      "epoch": 3.7830352783203124e-05,
      "step": 9917,
      "training_step_time": 9.924877882003784
    },
    {
      "epoch": 3.7830352783203124e-05,
      "model_forward_time": 2.4406282901763916,
      "step": 9917
    },
    {
      "epoch": 3.7830352783203124e-05,
      "step": 9917,
      "training_step_time": 10.306155443191528
    },
    {
      "epoch": 3.783416748046875e-05,
      "model_forward_time": 2.44773006439209,
      "step": 9918
    },
    {
      "epoch": 3.783416748046875e-05,
      "step": 9918,
      "training_step_time": 7.482838869094849
    },
    {
      "epoch": 3.783416748046875e-05,
      "model_forward_time": 2.4381587505340576,
      "step": 9918
    },
    {
      "epoch": 3.783416748046875e-05,
      "step": 9918,
      "training_step_time": 7.4682981967926025
    },
    {
      "epoch": 3.783416748046875e-05,
      "model_forward_time": 2.432053804397583,
      "step": 9918
    },
    {
      "epoch": 3.783416748046875e-05,
      "step": 9918,
      "training_step_time": 7.464018821716309
    },
    {
      "epoch": 3.783416748046875e-05,
      "model_forward_time": 2.435213327407837,
      "step": 9918
    },
    {
      "epoch": 3.783416748046875e-05,
      "step": 9918,
      "training_step_time": 10.025382041931152
    },
    {
      "epoch": 3.783798217773438e-05,
      "model_forward_time": 2.441969394683838,
      "step": 9919
    },
    {
      "epoch": 3.783798217773438e-05,
      "step": 9919,
      "training_step_time": 7.476783037185669
    },
    {
      "epoch": 3.783798217773438e-05,
      "model_forward_time": 3.4380617141723633,
      "step": 9919
    },
    {
      "epoch": 3.783798217773438e-05,
      "step": 9919,
      "training_step_time": 10.209527730941772
    },
    {
      "epoch": 3.783798217773438e-05,
      "model_forward_time": 3.1359047889709473,
      "step": 9919
    },
    {
      "epoch": 3.783798217773438e-05,
      "step": 9919,
      "training_step_time": 9.904078245162964
    },
    {
      "epoch": 3.783798217773438e-05,
      "model_forward_time": 3.1305577754974365,
      "step": 9919
    },
    {
      "epoch": 3.783798217773438e-05,
      "step": 9919,
      "training_step_time": 10.011308908462524
    },
    {
      "epoch": 3.7841796875e-05,
      "grad_norm": 0.44050726294517517,
      "learning_rate": 1e-05,
      "loss": 0.0792,
      "step": 9920
    },
    {
      "epoch": 3.7841796875e-05,
      "model_forward_time": 2.4465370178222656,
      "step": 9920
    },
    {
      "dynamics_loss_avg": 0.04515994694083929,
      "epoch": 3.7841796875e-05,
      "step": 9920
    },
    {
      "action_loss_avg": 0.014461924601346255,
      "epoch": 3.7841796875e-05,
      "step": 9920
    },
    {
      "epoch": 3.7841796875e-05,
      "step": 9920,
      "training_step_time": 7.4727396965026855
    },
    {
      "epoch": 3.7841796875e-05,
      "model_forward_time": 3.132021427154541,
      "step": 9920
    },
    {
      "epoch": 3.7841796875e-05,
      "step": 9920,
      "training_step_time": 9.900046825408936
    },
    {
      "epoch": 3.7841796875e-05,
      "model_forward_time": 3.4307138919830322,
      "step": 9920
    },
    {
      "epoch": 3.7841796875e-05,
      "step": 9920,
      "training_step_time": 10.20028567314148
    },
    {
      "epoch": 3.7841796875e-05,
      "model_forward_time": 2.434643507003784,
      "step": 9920
    },
    {
      "epoch": 3.7841796875e-05,
      "step": 9920,
      "training_step_time": 10.42242169380188
    },
    {
      "epoch": 3.784561157226562e-05,
      "model_forward_time": 2.448659658432007,
      "step": 9921
    },
    {
      "epoch": 3.784561157226562e-05,
      "step": 9921,
      "training_step_time": 7.474490404129028
    },
    {
      "epoch": 3.784561157226562e-05,
      "model_forward_time": 3.1343460083007812,
      "step": 9921
    },
    {
      "epoch": 3.784561157226562e-05,
      "step": 9921,
      "training_step_time": 9.902503490447998
    },
    {
      "epoch": 3.784561157226562e-05,
      "model_forward_time": 2.438128709793091,
      "step": 9921
    },
    {
      "epoch": 3.784561157226562e-05,
      "step": 9921,
      "training_step_time": 7.469380617141724
    },
    {
      "epoch": 3.784561157226562e-05,
      "model_forward_time": 3.138524055480957,
      "step": 9921
    },
    {
      "epoch": 3.784561157226562e-05,
      "step": 9921,
      "training_step_time": 10.33183765411377
    },
    {
      "epoch": 3.784942626953125e-05,
      "model_forward_time": 2.4432430267333984,
      "step": 9922
    },
    {
      "epoch": 3.784942626953125e-05,
      "step": 9922,
      "training_step_time": 7.479940176010132
    },
    {
      "epoch": 3.784942626953125e-05,
      "model_forward_time": 2.4348561763763428,
      "step": 9922
    },
    {
      "epoch": 3.784942626953125e-05,
      "step": 9922,
      "training_step_time": 7.460085868835449
    },
    {
      "epoch": 3.784942626953125e-05,
      "model_forward_time": 3.1302599906921387,
      "step": 9922
    },
    {
      "dynamics_loss_avg": 0.04892766661942005,
      "epoch": 3.784942626953125e-05,
      "step": 9922
    },
    {
      "action_loss_avg": 0.017854105867445468,
      "epoch": 3.784942626953125e-05,
      "step": 9922
    },
    {
      "epoch": 3.784942626953125e-05,
      "step": 9922,
      "training_step_time": 9.901804685592651
    },
    {
      "epoch": 3.784942626953125e-05,
      "model_forward_time": 3.1385769844055176,
      "step": 9922
    },
    {
      "epoch": 3.784942626953125e-05,
      "step": 9922,
      "training_step_time": 10.323416471481323
    },
    {
      "epoch": 3.7853240966796875e-05,
      "model_forward_time": 3.1510443687438965,
      "step": 9923
    },
    {
      "epoch": 3.7853240966796875e-05,
      "step": 9923,
      "training_step_time": 9.93781566619873
    },
    {
      "epoch": 3.7853240966796875e-05,
      "model_forward_time": 2.728562116622925,
      "step": 9923
    },
    {
      "epoch": 3.7853240966796875e-05,
      "step": 9923,
      "training_step_time": 7.755384683609009
    },
    {
      "epoch": 3.7853240966796875e-05,
      "model_forward_time": 2.432335138320923,
      "step": 9923
    },
    {
      "epoch": 3.7853240966796875e-05,
      "step": 9923,
      "training_step_time": 7.460036277770996
    },
    {
      "epoch": 3.7853240966796875e-05,
      "model_forward_time": 3.1356003284454346,
      "step": 9923
    },
    {
      "epoch": 3.7853240966796875e-05,
      "step": 9923,
      "training_step_time": 10.011921644210815
    },
    {
      "epoch": 3.78570556640625e-05,
      "model_forward_time": 3.1573386192321777,
      "step": 9924
    },
    {
      "epoch": 3.78570556640625e-05,
      "step": 9924,
      "training_step_time": 9.942802906036377
    },
    {
      "epoch": 3.78570556640625e-05,
      "model_forward_time": 3.133486270904541,
      "step": 9924
    },
    {
      "epoch": 3.78570556640625e-05,
      "step": 9924,
      "training_step_time": 9.913787126541138
    },
    {
      "epoch": 3.78570556640625e-05,
      "model_forward_time": 2.7159767150878906,
      "step": 9924
    },
    {
      "epoch": 3.78570556640625e-05,
      "step": 9924,
      "training_step_time": 7.749001741409302
    },
    {
      "epoch": 3.78570556640625e-05,
      "model_forward_time": 2.434422016143799,
      "step": 9924
    },
    {
      "epoch": 3.78570556640625e-05,
      "step": 9924,
      "training_step_time": 10.036323070526123
    },
    {
      "epoch": 3.786087036132813e-05,
      "model_forward_time": 2.4542059898376465,
      "step": 9925
    },
    {
      "dynamics_loss_avg": 0.04707305151969195,
      "epoch": 3.786087036132813e-05,
      "step": 9925
    },
    {
      "action_loss_avg": 0.016814758395776154,
      "epoch": 3.786087036132813e-05,
      "step": 9925
    },
    {
      "epoch": 3.786087036132813e-05,
      "step": 9925,
      "training_step_time": 7.489932537078857
    },
    {
      "epoch": 3.786087036132813e-05,
      "model_forward_time": 3.134216547012329,
      "step": 9925
    },
    {
      "epoch": 3.786087036132813e-05,
      "step": 9925,
      "training_step_time": 9.919215440750122
    },
    {
      "epoch": 3.786087036132813e-05,
      "model_forward_time": 3.143606424331665,
      "step": 9925
    },
    {
      "epoch": 3.786087036132813e-05,
      "step": 9925,
      "training_step_time": 9.924400329589844
    },
    {
      "epoch": 3.786087036132813e-05,
      "model_forward_time": 3.4664158821105957,
      "step": 9925
    },
    {
      "epoch": 3.786087036132813e-05,
      "step": 9925,
      "training_step_time": 10.330823183059692
    },
    {
      "epoch": 3.786468505859375e-05,
      "model_forward_time": 2.4519567489624023,
      "step": 9926
    },
    {
      "epoch": 3.786468505859375e-05,
      "step": 9926,
      "training_step_time": 7.491028785705566
    },
    {
      "epoch": 3.786468505859375e-05,
      "model_forward_time": 2.434987783432007,
      "step": 9926
    },
    {
      "epoch": 3.786468505859375e-05,
      "step": 9926,
      "training_step_time": 7.464221954345703
    },
    {
      "epoch": 3.786468505859375e-05,
      "model_forward_time": 3.1362316608428955,
      "step": 9926
    },
    {
      "epoch": 3.786468505859375e-05,
      "step": 9926,
      "training_step_time": 9.90564489364624
    },
    {
      "epoch": 3.786468505859375e-05,
      "model_forward_time": 2.4376184940338135,
      "step": 9926
    },
    {
      "epoch": 3.786468505859375e-05,
      "step": 9926,
      "training_step_time": 10.33780026435852
    },
    {
      "epoch": 3.7868499755859373e-05,
      "model_forward_time": 3.4407060146331787,
      "step": 9927
    },
    {
      "epoch": 3.7868499755859373e-05,
      "step": 9927,
      "training_step_time": 10.21614146232605
    },
    {
      "epoch": 3.7868499755859373e-05,
      "model_forward_time": 3.1387875080108643,
      "step": 9927
    },
    {
      "epoch": 3.7868499755859373e-05,
      "step": 9927,
      "training_step_time": 9.906955242156982
    },
    {
      "epoch": 3.7868499755859373e-05,
      "model_forward_time": 3.142895221710205,
      "step": 9927
    },
    {
      "dynamics_loss_avg": 0.0420862952247262,
      "epoch": 3.7868499755859373e-05,
      "step": 9927
    },
    {
      "action_loss_avg": 0.01816567131318152,
      "epoch": 3.7868499755859373e-05,
      "step": 9927
    },
    {
      "epoch": 3.7868499755859373e-05,
      "step": 9927,
      "training_step_time": 9.919244766235352
    },
    {
      "epoch": 3.7868499755859373e-05,
      "model_forward_time": 2.436884880065918,
      "step": 9927
    },
    {
      "epoch": 3.7868499755859373e-05,
      "step": 9927,
      "training_step_time": 10.319074630737305
    },
    {
      "epoch": 3.7872314453125e-05,
      "model_forward_time": 2.4431066513061523,
      "step": 9928
    },
    {
      "epoch": 3.7872314453125e-05,
      "step": 9928,
      "training_step_time": 7.473339080810547
    },
    {
      "epoch": 3.7872314453125e-05,
      "model_forward_time": 3.139791965484619,
      "step": 9928
    },
    {
      "epoch": 3.7872314453125e-05,
      "step": 9928,
      "training_step_time": 9.907164096832275
    },
    {
      "epoch": 3.7872314453125e-05,
      "model_forward_time": 3.455254554748535,
      "step": 9928
    },
    {
      "epoch": 3.7872314453125e-05,
      "step": 9928,
      "training_step_time": 10.223652839660645
    },
    {
      "epoch": 3.7872314453125e-05,
      "model_forward_time": 3.1416232585906982,
      "step": 9928
    },
    {
      "epoch": 3.7872314453125e-05,
      "step": 9928,
      "training_step_time": 10.023359537124634
    },
    {
      "epoch": 3.7876129150390626e-05,
      "model_forward_time": 2.451028347015381,
      "step": 9929
    },
    {
      "epoch": 3.7876129150390626e-05,
      "step": 9929,
      "training_step_time": 7.47961688041687
    },
    {
      "epoch": 3.7876129150390626e-05,
      "model_forward_time": 3.138597249984741,
      "step": 9929
    },
    {
      "epoch": 3.7876129150390626e-05,
      "step": 9929,
      "training_step_time": 9.906431436538696
    },
    {
      "epoch": 3.7876129150390626e-05,
      "model_forward_time": 3.13973069190979,
      "step": 9929
    },
    {
      "epoch": 3.7876129150390626e-05,
      "step": 9929,
      "training_step_time": 9.919311046600342
    },
    {
      "epoch": 3.7876129150390626e-05,
      "model_forward_time": 3.3943824768066406,
      "step": 9929
    },
    {
      "epoch": 3.7876129150390626e-05,
      "step": 9929,
      "training_step_time": 10.263591289520264
    },
    {
      "epoch": 3.787994384765625e-05,
      "grad_norm": 0.33925166726112366,
      "learning_rate": 1e-05,
      "loss": 0.0736,
      "step": 9930
    },
    {
      "epoch": 3.787994384765625e-05,
      "model_forward_time": 2.4560441970825195,
      "step": 9930
    },
    {
      "dynamics_loss_avg": 0.07361067943274975,
      "epoch": 3.787994384765625e-05,
      "step": 9930
    },
    {
      "action_loss_avg": 0.027403732389211656,
      "epoch": 3.787994384765625e-05,
      "step": 9930
    },
    {
      "epoch": 3.787994384765625e-05,
      "step": 9930,
      "training_step_time": 7.49008321762085
    },
    {
      "epoch": 3.787994384765625e-05,
      "model_forward_time": 3.143052339553833,
      "step": 9930
    },
    {
      "epoch": 3.787994384765625e-05,
      "step": 9930,
      "training_step_time": 9.917648553848267
    },
    {
      "epoch": 3.787994384765625e-05,
      "model_forward_time": 3.138673782348633,
      "step": 9930
    },
    {
      "epoch": 3.787994384765625e-05,
      "step": 9930,
      "training_step_time": 9.915129661560059
    },
    {
      "epoch": 3.787994384765625e-05,
      "model_forward_time": 2.443554162979126,
      "step": 9930
    },
    {
      "epoch": 3.787994384765625e-05,
      "step": 9930,
      "training_step_time": 10.278462886810303
    },
    {
      "epoch": 3.788375854492187e-05,
      "model_forward_time": 3.4463956356048584,
      "step": 9931
    },
    {
      "epoch": 3.788375854492187e-05,
      "step": 9931,
      "training_step_time": 10.21980333328247
    },
    {
      "epoch": 3.788375854492187e-05,
      "model_forward_time": 3.143203020095825,
      "step": 9931
    },
    {
      "epoch": 3.788375854492187e-05,
      "step": 9931,
      "training_step_time": 9.91463327407837
    },
    {
      "epoch": 3.788375854492187e-05,
      "model_forward_time": 3.135840654373169,
      "step": 9931
    },
    {
      "epoch": 3.788375854492187e-05,
      "step": 9931,
      "training_step_time": 9.906359195709229
    },
    {
      "epoch": 3.788375854492187e-05,
      "model_forward_time": 3.138681650161743,
      "step": 9931
    },
    {
      "epoch": 3.788375854492187e-05,
      "step": 9931,
      "training_step_time": 10.29609990119934
    },
    {
      "epoch": 3.78875732421875e-05,
      "model_forward_time": 3.155003070831299,
      "step": 9932
    },
    {
      "epoch": 3.78875732421875e-05,
      "step": 9932,
      "training_step_time": 9.922196626663208
    },
    {
      "epoch": 3.78875732421875e-05,
      "model_forward_time": 3.414458990097046,
      "step": 9932
    },
    {
      "epoch": 3.78875732421875e-05,
      "step": 9932,
      "training_step_time": 10.179534196853638
    },
    {
      "epoch": 3.78875732421875e-05,
      "model_forward_time": 3.137338876724243,
      "step": 9932
    },
    {
      "dynamics_loss_avg": 0.0675840362906456,
      "epoch": 3.78875732421875e-05,
      "step": 9932
    },
    {
      "action_loss_avg": 0.026998890144750477,
      "epoch": 3.78875732421875e-05,
      "step": 9932
    },
    {
      "epoch": 3.78875732421875e-05,
      "step": 9932,
      "training_step_time": 9.906238079071045
    },
    {
      "epoch": 3.78875732421875e-05,
      "model_forward_time": 3.1427204608917236,
      "step": 9932
    },
    {
      "epoch": 3.78875732421875e-05,
      "step": 9932,
      "training_step_time": 10.338616132736206
    },
    {
      "epoch": 3.7891387939453124e-05,
      "model_forward_time": 2.447035789489746,
      "step": 9933
    },
    {
      "epoch": 3.7891387939453124e-05,
      "step": 9933,
      "training_step_time": 7.472252130508423
    },
    {
      "epoch": 3.7891387939453124e-05,
      "model_forward_time": 3.129845380783081,
      "step": 9933
    },
    {
      "epoch": 3.7891387939453124e-05,
      "step": 9933,
      "training_step_time": 9.895535230636597
    },
    {
      "epoch": 3.7891387939453124e-05,
      "model_forward_time": 3.3948757648468018,
      "step": 9933
    },
    {
      "epoch": 3.7891387939453124e-05,
      "step": 9933,
      "training_step_time": 10.16520071029663
    },
    {
      "epoch": 3.7891387939453124e-05,
      "model_forward_time": 2.434131383895874,
      "step": 9933
    },
    {
      "epoch": 3.7891387939453124e-05,
      "step": 9933,
      "training_step_time": 10.022080898284912
    },
    {
      "epoch": 3.789520263671875e-05,
      "model_forward_time": 3.1538736820220947,
      "step": 9934
    },
    {
      "epoch": 3.789520263671875e-05,
      "step": 9934,
      "training_step_time": 9.939452171325684
    },
    {
      "epoch": 3.789520263671875e-05,
      "model_forward_time": 2.4350414276123047,
      "step": 9934
    },
    {
      "epoch": 3.789520263671875e-05,
      "step": 9934,
      "training_step_time": 7.463183879852295
    },
    {
      "epoch": 3.789520263671875e-05,
      "model_forward_time": 3.134265184402466,
      "step": 9934
    },
    {
      "epoch": 3.789520263671875e-05,
      "step": 9934,
      "training_step_time": 9.91597318649292
    },
    {
      "epoch": 3.789520263671875e-05,
      "model_forward_time": 2.7542331218719482,
      "step": 9934
    },
    {
      "epoch": 3.789520263671875e-05,
      "step": 9934,
      "training_step_time": 10.013779401779175
    },
    {
      "epoch": 3.789901733398438e-05,
      "model_forward_time": 3.1491684913635254,
      "step": 9935
    },
    {
      "dynamics_loss_avg": 0.04811758100986481,
      "epoch": 3.789901733398438e-05,
      "step": 9935
    },
    {
      "action_loss_avg": 0.01779167801141739,
      "epoch": 3.789901733398438e-05,
      "step": 9935
    },
    {
      "epoch": 3.789901733398438e-05,
      "step": 9935,
      "training_step_time": 9.929261207580566
    },
    {
      "epoch": 3.789901733398438e-05,
      "model_forward_time": 2.4349749088287354,
      "step": 9935
    },
    {
      "epoch": 3.789901733398438e-05,
      "step": 9935,
      "training_step_time": 7.465113639831543
    },
    {
      "epoch": 3.789901733398438e-05,
      "model_forward_time": 2.429633378982544,
      "step": 9935
    },
    {
      "epoch": 3.789901733398438e-05,
      "step": 9935,
      "training_step_time": 7.456179141998291
    },
    {
      "epoch": 3.789901733398438e-05,
      "model_forward_time": 3.1349539756774902,
      "step": 9935
    },
    {
      "epoch": 3.789901733398438e-05,
      "step": 9935,
      "training_step_time": 10.026597738265991
    },
    {
      "epoch": 3.790283203125e-05,
      "model_forward_time": 3.448383092880249,
      "step": 9936
    },
    {
      "epoch": 3.790283203125e-05,
      "step": 9936,
      "training_step_time": 10.222927808761597
    },
    {
      "epoch": 3.790283203125e-05,
      "model_forward_time": 2.4339075088500977,
      "step": 9936
    },
    {
      "epoch": 3.790283203125e-05,
      "step": 9936,
      "training_step_time": 7.460177659988403
    },
    {
      "epoch": 3.790283203125e-05,
      "model_forward_time": 1.7401678562164307,
      "step": 9936
    },
    {
      "epoch": 3.790283203125e-05,
      "step": 9936,
      "training_step_time": 5.197877883911133
    },
    {
      "epoch": 3.790283203125e-05,
      "model_forward_time": 2.4290411472320557,
      "step": 9936
    },
    {
      "epoch": 3.790283203125e-05,
      "step": 9936,
      "training_step_time": 10.241159439086914
    },
    {
      "epoch": 3.790664672851562e-05,
      "model_forward_time": 3.1353156566619873,
      "step": 9937
    },
    {
      "epoch": 3.790664672851562e-05,
      "step": 9937,
      "training_step_time": 9.897059202194214
    },
    {
      "epoch": 3.790664672851562e-05,
      "model_forward_time": 3.1299502849578857,
      "step": 9937
    },
    {
      "epoch": 3.790664672851562e-05,
      "step": 9937,
      "training_step_time": 9.898142576217651
    },
    {
      "epoch": 3.790664672851562e-05,
      "model_forward_time": 3.4557650089263916,
      "step": 9937
    },
    {
      "dynamics_loss_avg": 0.052980687469244,
      "epoch": 3.790664672851562e-05,
      "step": 9937
    },
    {
      "action_loss_avg": 0.018884498672559858,
      "epoch": 3.790664672851562e-05,
      "step": 9937
    },
    {
      "epoch": 3.790664672851562e-05,
      "step": 9937,
      "training_step_time": 10.222161769866943
    },
    {
      "epoch": 3.790664672851562e-05,
      "model_forward_time": 3.1314868927001953,
      "step": 9937
    },
    {
      "epoch": 3.790664672851562e-05,
      "step": 9937,
      "training_step_time": 10.363969326019287
    },
    {
      "epoch": 3.791046142578125e-05,
      "model_forward_time": 3.163459539413452,
      "step": 9938
    },
    {
      "epoch": 3.791046142578125e-05,
      "step": 9938,
      "training_step_time": 9.952725172042847
    },
    {
      "epoch": 3.791046142578125e-05,
      "model_forward_time": 3.1415300369262695,
      "step": 9938
    },
    {
      "epoch": 3.791046142578125e-05,
      "step": 9938,
      "training_step_time": 9.909485816955566
    },
    {
      "epoch": 3.791046142578125e-05,
      "model_forward_time": 2.4372451305389404,
      "step": 9938
    },
    {
      "epoch": 3.791046142578125e-05,
      "step": 9938,
      "training_step_time": 7.456235408782959
    },
    {
      "epoch": 3.791046142578125e-05,
      "model_forward_time": 3.428976535797119,
      "step": 9938
    },
    {
      "epoch": 3.791046142578125e-05,
      "step": 9938,
      "training_step_time": 10.317950963973999
    },
    {
      "epoch": 3.7914276123046875e-05,
      "model_forward_time": 3.1608123779296875,
      "step": 9939
    },
    {
      "epoch": 3.7914276123046875e-05,
      "step": 9939,
      "training_step_time": 9.930347681045532
    },
    {
      "epoch": 3.7914276123046875e-05,
      "model_forward_time": 3.137425661087036,
      "step": 9939
    },
    {
      "epoch": 3.7914276123046875e-05,
      "step": 9939,
      "training_step_time": 9.906864166259766
    },
    {
      "epoch": 3.7914276123046875e-05,
      "model_forward_time": 2.4370651245117188,
      "step": 9939
    },
    {
      "epoch": 3.7914276123046875e-05,
      "step": 9939,
      "training_step_time": 7.469267845153809
    },
    {
      "epoch": 3.7914276123046875e-05,
      "model_forward_time": 3.145496368408203,
      "step": 9939
    },
    {
      "epoch": 3.7914276123046875e-05,
      "step": 9939,
      "training_step_time": 10.028926849365234
    },
    {
      "epoch": 3.79180908203125e-05,
      "grad_norm": 0.31782492995262146,
      "learning_rate": 1e-05,
      "loss": 0.0801,
      "step": 9940
    },
    {
      "epoch": 3.79180908203125e-05,
      "model_forward_time": 2.7657644748687744,
      "step": 9940
    },
    {
      "dynamics_loss_avg": 0.05839108563959598,
      "epoch": 3.79180908203125e-05,
      "step": 9940
    },
    {
      "action_loss_avg": 0.019149639690294862,
      "epoch": 3.79180908203125e-05,
      "step": 9940
    },
    {
      "epoch": 3.79180908203125e-05,
      "step": 9940,
      "training_step_time": 7.797687768936157
    },
    {
      "epoch": 3.79180908203125e-05,
      "model_forward_time": 3.145221710205078,
      "step": 9940
    },
    {
      "epoch": 3.79180908203125e-05,
      "step": 9940,
      "training_step_time": 9.91920804977417
    },
    {
      "epoch": 3.79180908203125e-05,
      "model_forward_time": 2.435401201248169,
      "step": 9940
    },
    {
      "epoch": 3.79180908203125e-05,
      "step": 9940,
      "training_step_time": 7.464484691619873
    },
    {
      "epoch": 3.79180908203125e-05,
      "model_forward_time": 2.4340741634368896,
      "step": 9940
    },
    {
      "epoch": 3.79180908203125e-05,
      "step": 9940,
      "training_step_time": 10.025630474090576
    },
    {
      "epoch": 3.792190551757813e-05,
      "model_forward_time": 3.1498730182647705,
      "step": 9941
    },
    {
      "epoch": 3.792190551757813e-05,
      "step": 9941,
      "training_step_time": 9.91663146018982
    },
    {
      "epoch": 3.792190551757813e-05,
      "model_forward_time": 2.7238895893096924,
      "step": 9941
    },
    {
      "epoch": 3.792190551757813e-05,
      "step": 9941,
      "training_step_time": 7.7443318367004395
    },
    {
      "epoch": 3.792190551757813e-05,
      "model_forward_time": 2.4345109462738037,
      "step": 9941
    },
    {
      "epoch": 3.792190551757813e-05,
      "step": 9941,
      "training_step_time": 7.4612627029418945
    },
    {
      "epoch": 3.792190551757813e-05,
      "model_forward_time": 3.136326789855957,
      "step": 9941
    },
    {
      "epoch": 3.792190551757813e-05,
      "step": 9941,
      "training_step_time": 10.231327533721924
    },
    {
      "epoch": 3.792572021484375e-05,
      "model_forward_time": 3.1508758068084717,
      "step": 9942
    },
    {
      "epoch": 3.792572021484375e-05,
      "step": 9942,
      "training_step_time": 9.916974782943726
    },
    {
      "epoch": 3.792572021484375e-05,
      "model_forward_time": 3.1309654712677,
      "step": 9942
    },
    {
      "epoch": 3.792572021484375e-05,
      "step": 9942,
      "training_step_time": 9.892175674438477
    },
    {
      "epoch": 3.792572021484375e-05,
      "model_forward_time": 3.403275728225708,
      "step": 9942
    },
    {
      "dynamics_loss_avg": 0.0457686560228467,
      "epoch": 3.792572021484375e-05,
      "step": 9942
    },
    {
      "action_loss_avg": 0.012388032348826527,
      "epoch": 3.792572021484375e-05,
      "step": 9942
    },
    {
      "epoch": 3.792572021484375e-05,
      "step": 9942,
      "training_step_time": 10.167571306228638
    },
    {
      "epoch": 3.792572021484375e-05,
      "model_forward_time": 3.1354377269744873,
      "step": 9942
    },
    {
      "epoch": 3.792572021484375e-05,
      "step": 9942,
      "training_step_time": 10.341989278793335
    },
    {
      "epoch": 3.7929534912109374e-05,
      "model_forward_time": 2.4506051540374756,
      "step": 9943
    },
    {
      "epoch": 3.7929534912109374e-05,
      "step": 9943,
      "training_step_time": 7.469348907470703
    },
    {
      "epoch": 3.7929534912109374e-05,
      "model_forward_time": 2.429124593734741,
      "step": 9943
    },
    {
      "epoch": 3.7929534912109374e-05,
      "step": 9943,
      "training_step_time": 7.449612379074097
    },
    {
      "epoch": 3.7929534912109374e-05,
      "model_forward_time": 3.134119749069214,
      "step": 9943
    },
    {
      "epoch": 3.7929534912109374e-05,
      "step": 9943,
      "training_step_time": 9.90022897720337
    },
    {
      "epoch": 3.7929534912109374e-05,
      "model_forward_time": 3.4170939922332764,
      "step": 9943
    },
    {
      "epoch": 3.7929534912109374e-05,
      "step": 9943,
      "training_step_time": 10.294244527816772
    },
    {
      "epoch": 3.7933349609375e-05,
      "model_forward_time": 2.4445438385009766,
      "step": 9944
    },
    {
      "epoch": 3.7933349609375e-05,
      "step": 9944,
      "training_step_time": 7.472912788391113
    },
    {
      "epoch": 3.7933349609375e-05,
      "model_forward_time": 3.1336898803710938,
      "step": 9944
    },
    {
      "epoch": 3.7933349609375e-05,
      "step": 9944,
      "training_step_time": 9.897179126739502
    },
    {
      "epoch": 3.7933349609375e-05,
      "model_forward_time": 3.1394741535186768,
      "step": 9944
    },
    {
      "epoch": 3.7933349609375e-05,
      "step": 9944,
      "training_step_time": 9.906307935714722
    },
    {
      "epoch": 3.7933349609375e-05,
      "model_forward_time": 3.1326656341552734,
      "step": 9944
    },
    {
      "epoch": 3.7933349609375e-05,
      "step": 9944,
      "training_step_time": 10.01075530052185
    },
    {
      "epoch": 3.7937164306640626e-05,
      "model_forward_time": 3.431145429611206,
      "step": 9945
    },
    {
      "dynamics_loss_avg": 0.05374913476407528,
      "epoch": 3.7937164306640626e-05,
      "step": 9945
    },
    {
      "action_loss_avg": 0.01828329232521355,
      "epoch": 3.7937164306640626e-05,
      "step": 9945
    },
    {
      "epoch": 3.7937164306640626e-05,
      "step": 9945,
      "training_step_time": 10.203750133514404
    },
    {
      "epoch": 3.7937164306640626e-05,
      "model_forward_time": 3.14833927154541,
      "step": 9945
    },
    {
      "epoch": 3.7937164306640626e-05,
      "step": 9945,
      "training_step_time": 9.925869703292847
    },
    {
      "epoch": 3.7937164306640626e-05,
      "model_forward_time": 2.4355428218841553,
      "step": 9945
    },
    {
      "epoch": 3.7937164306640626e-05,
      "step": 9945,
      "training_step_time": 7.462414264678955
    },
    {
      "epoch": 3.7937164306640626e-05,
      "model_forward_time": 3.1378414630889893,
      "step": 9945
    },
    {
      "epoch": 3.7937164306640626e-05,
      "step": 9945,
      "training_step_time": 10.039559602737427
    },
    {
      "epoch": 3.794097900390625e-05,
      "model_forward_time": 2.447624444961548,
      "step": 9946
    },
    {
      "epoch": 3.794097900390625e-05,
      "step": 9946,
      "training_step_time": 7.4761552810668945
    },
    {
      "epoch": 3.794097900390625e-05,
      "model_forward_time": 2.4348087310791016,
      "step": 9946
    },
    {
      "epoch": 3.794097900390625e-05,
      "step": 9946,
      "training_step_time": 7.459649085998535
    },
    {
      "epoch": 3.794097900390625e-05,
      "model_forward_time": 3.446115732192993,
      "step": 9946
    },
    {
      "epoch": 3.794097900390625e-05,
      "step": 9946,
      "training_step_time": 10.217122316360474
    },
    {
      "epoch": 3.794097900390625e-05,
      "model_forward_time": 3.1334831714630127,
      "step": 9946
    },
    {
      "epoch": 3.794097900390625e-05,
      "step": 9946,
      "training_step_time": 10.29950475692749
    },
    {
      "epoch": 3.794479370117187e-05,
      "model_forward_time": 3.1682636737823486,
      "step": 9947
    },
    {
      "epoch": 3.794479370117187e-05,
      "step": 9947,
      "training_step_time": 9.943455457687378
    },
    {
      "epoch": 3.794479370117187e-05,
      "model_forward_time": 2.4357781410217285,
      "step": 9947
    },
    {
      "epoch": 3.794479370117187e-05,
      "step": 9947,
      "training_step_time": 7.466938734054565
    },
    {
      "epoch": 3.794479370117187e-05,
      "model_forward_time": 2.4278502464294434,
      "step": 9947
    },
    {
      "dynamics_loss_avg": 0.05114385075867176,
      "epoch": 3.794479370117187e-05,
      "step": 9947
    },
    {
      "action_loss_avg": 0.01710676057264209,
      "epoch": 3.794479370117187e-05,
      "step": 9947
    },
    {
      "epoch": 3.794479370117187e-05,
      "step": 9947,
      "training_step_time": 7.45305871963501
    },
    {
      "epoch": 3.794479370117187e-05,
      "model_forward_time": 2.7134032249450684,
      "step": 9947
    },
    {
      "epoch": 3.794479370117187e-05,
      "step": 9947,
      "training_step_time": 10.307864427566528
    },
    {
      "epoch": 3.79486083984375e-05,
      "model_forward_time": 3.14105486869812,
      "step": 9948
    },
    {
      "epoch": 3.79486083984375e-05,
      "step": 9948,
      "training_step_time": 9.907331228256226
    },
    {
      "epoch": 3.79486083984375e-05,
      "model_forward_time": 3.1366543769836426,
      "step": 9948
    },
    {
      "epoch": 3.79486083984375e-05,
      "step": 9948,
      "training_step_time": 9.9022216796875
    },
    {
      "epoch": 3.79486083984375e-05,
      "model_forward_time": 2.427170515060425,
      "step": 9948
    },
    {
      "epoch": 3.79486083984375e-05,
      "step": 9948,
      "training_step_time": 7.447459697723389
    },
    {
      "epoch": 3.79486083984375e-05,
      "model_forward_time": 3.1335530281066895,
      "step": 9948
    },
    {
      "epoch": 3.79486083984375e-05,
      "step": 9948,
      "training_step_time": 10.292551040649414
    },
    {
      "epoch": 3.7952423095703124e-05,
      "model_forward_time": 3.4527971744537354,
      "step": 9949
    },
    {
      "epoch": 3.7952423095703124e-05,
      "step": 9949,
      "training_step_time": 10.219846487045288
    },
    {
      "epoch": 3.7952423095703124e-05,
      "model_forward_time": 3.1324946880340576,
      "step": 9949
    },
    {
      "epoch": 3.7952423095703124e-05,
      "step": 9949,
      "training_step_time": 9.896830320358276
    },
    {
      "epoch": 3.7952423095703124e-05,
      "model_forward_time": 2.4309682846069336,
      "step": 9949
    },
    {
      "epoch": 3.7952423095703124e-05,
      "step": 9949,
      "training_step_time": 7.448489427566528
    },
    {
      "epoch": 3.7952423095703124e-05,
      "model_forward_time": 3.13110089302063,
      "step": 9949
    },
    {
      "epoch": 3.7952423095703124e-05,
      "step": 9949,
      "training_step_time": 10.209062099456787
    },
    {
      "epoch": 3.795623779296875e-05,
      "grad_norm": 0.38777586817741394,
      "learning_rate": 1e-05,
      "loss": 0.0753,
      "step": 9950
    },
    {
      "epoch": 3.795623779296875e-05,
      "model_forward_time": 3.1485507488250732,
      "step": 9950
    },
    {
      "dynamics_loss_avg": 0.05611306931823492,
      "epoch": 3.795623779296875e-05,
      "step": 9950
    },
    {
      "action_loss_avg": 0.01872732015326619,
      "epoch": 3.795623779296875e-05,
      "step": 9950
    },
    {
      "epoch": 3.795623779296875e-05,
      "step": 9950,
      "training_step_time": 9.922743082046509
    },
    {
      "epoch": 3.795623779296875e-05,
      "model_forward_time": 2.685971975326538,
      "step": 9950
    },
    {
      "epoch": 3.795623779296875e-05,
      "step": 9950,
      "training_step_time": 7.707563161849976
    },
    {
      "epoch": 3.795623779296875e-05,
      "model_forward_time": 3.1291303634643555,
      "step": 9950
    },
    {
      "epoch": 3.795623779296875e-05,
      "step": 9950,
      "training_step_time": 9.894864559173584
    },
    {
      "epoch": 3.795623779296875e-05,
      "model_forward_time": 3.130962610244751,
      "step": 9950
    },
    {
      "epoch": 3.795623779296875e-05,
      "step": 9950,
      "training_step_time": 10.2473886013031
    },
    {
      "epoch": 3.796005249023438e-05,
      "model_forward_time": 2.4520511627197266,
      "step": 9951
    },
    {
      "epoch": 3.796005249023438e-05,
      "step": 9951,
      "training_step_time": 7.477686405181885
    },
    {
      "epoch": 3.796005249023438e-05,
      "model_forward_time": 2.426561117172241,
      "step": 9951
    },
    {
      "epoch": 3.796005249023438e-05,
      "step": 9951,
      "training_step_time": 7.455334663391113
    },
    {
      "epoch": 3.796005249023438e-05,
      "model_forward_time": 2.4306581020355225,
      "step": 9951
    },
    {
      "epoch": 3.796005249023438e-05,
      "step": 9951,
      "training_step_time": 7.773208856582642
    },
    {
      "epoch": 3.796005249023438e-05,
      "model_forward_time": 2.43088436126709,
      "step": 9951
    },
    {
      "epoch": 3.796005249023438e-05,
      "step": 9951,
      "training_step_time": 10.0197114944458
    },
    {
      "epoch": 3.79638671875e-05,
      "model_forward_time": 3.14630389213562,
      "step": 9952
    },
    {
      "epoch": 3.79638671875e-05,
      "step": 9952,
      "training_step_time": 9.913535118103027
    },
    {
      "epoch": 3.79638671875e-05,
      "model_forward_time": 3.1317434310913086,
      "step": 9952
    },
    {
      "epoch": 3.79638671875e-05,
      "step": 9952,
      "training_step_time": 9.897419691085815
    },
    {
      "epoch": 3.79638671875e-05,
      "model_forward_time": 3.1338961124420166,
      "step": 9952
    },
    {
      "dynamics_loss_avg": 0.06656731460243463,
      "epoch": 3.79638671875e-05,
      "step": 9952
    },
    {
      "action_loss_avg": 0.01985721616074443,
      "epoch": 3.79638671875e-05,
      "step": 9952
    },
    {
      "epoch": 3.79638671875e-05,
      "step": 9952,
      "training_step_time": 9.904278993606567
    },
    {
      "epoch": 3.79638671875e-05,
      "model_forward_time": 3.423793315887451,
      "step": 9952
    },
    {
      "epoch": 3.79638671875e-05,
      "step": 9952,
      "training_step_time": 10.327600002288818
    },
    {
      "epoch": 3.796768188476562e-05,
      "model_forward_time": 2.4438157081604004,
      "step": 9953
    },
    {
      "epoch": 3.796768188476562e-05,
      "step": 9953,
      "training_step_time": 7.475260972976685
    },
    {
      "epoch": 3.796768188476562e-05,
      "model_forward_time": 3.130488157272339,
      "step": 9953
    },
    {
      "epoch": 3.796768188476562e-05,
      "step": 9953,
      "training_step_time": 9.900333642959595
    },
    {
      "epoch": 3.796768188476562e-05,
      "model_forward_time": 3.1386008262634277,
      "step": 9953
    },
    {
      "epoch": 3.796768188476562e-05,
      "step": 9953,
      "training_step_time": 9.909944772720337
    },
    {
      "epoch": 3.796768188476562e-05,
      "model_forward_time": 2.435199737548828,
      "step": 9953
    },
    {
      "epoch": 3.796768188476562e-05,
      "step": 9953,
      "training_step_time": 10.269422054290771
    },
    {
      "epoch": 3.797149658203125e-05,
      "model_forward_time": 2.4452173709869385,
      "step": 9954
    },
    {
      "epoch": 3.797149658203125e-05,
      "step": 9954,
      "training_step_time": 7.475485801696777
    },
    {
      "epoch": 3.797149658203125e-05,
      "model_forward_time": 2.686340093612671,
      "step": 9954
    },
    {
      "epoch": 3.797149658203125e-05,
      "step": 9954,
      "training_step_time": 7.7212419509887695
    },
    {
      "epoch": 3.797149658203125e-05,
      "model_forward_time": 3.1342861652374268,
      "step": 9954
    },
    {
      "epoch": 3.797149658203125e-05,
      "step": 9954,
      "training_step_time": 9.905318021774292
    },
    {
      "epoch": 3.797149658203125e-05,
      "model_forward_time": 3.1352596282958984,
      "step": 9954
    },
    {
      "epoch": 3.797149658203125e-05,
      "step": 9954,
      "training_step_time": 10.243590593338013
    },
    {
      "epoch": 3.7975311279296875e-05,
      "model_forward_time": 2.501258373260498,
      "step": 9955
    },
    {
      "dynamics_loss_avg": 0.06862256601452828,
      "epoch": 3.7975311279296875e-05,
      "step": 9955
    },
    {
      "action_loss_avg": 0.023462064657360317,
      "epoch": 3.7975311279296875e-05,
      "step": 9955
    },
    {
      "epoch": 3.7975311279296875e-05,
      "step": 9955,
      "training_step_time": 7.535754442214966
    },
    {
      "epoch": 3.7975311279296875e-05,
      "model_forward_time": 2.4328856468200684,
      "step": 9955
    },
    {
      "epoch": 3.7975311279296875e-05,
      "step": 9955,
      "training_step_time": 7.462730169296265
    },
    {
      "epoch": 3.7975311279296875e-05,
      "model_forward_time": 2.7161107063293457,
      "step": 9955
    },
    {
      "epoch": 3.7975311279296875e-05,
      "step": 9955,
      "training_step_time": 7.741562366485596
    },
    {
      "epoch": 3.7975311279296875e-05,
      "model_forward_time": 2.4422607421875,
      "step": 9955
    },
    {
      "epoch": 3.7975311279296875e-05,
      "step": 9955,
      "training_step_time": 10.032651662826538
    },
    {
      "epoch": 3.79791259765625e-05,
      "model_forward_time": 2.4748375415802,
      "step": 9956
    },
    {
      "epoch": 3.79791259765625e-05,
      "step": 9956,
      "training_step_time": 7.505592584609985
    },
    {
      "epoch": 3.79791259765625e-05,
      "model_forward_time": 3.1504132747650146,
      "step": 9956
    },
    {
      "epoch": 3.79791259765625e-05,
      "step": 9956,
      "training_step_time": 9.930321455001831
    },
    {
      "epoch": 3.79791259765625e-05,
      "model_forward_time": 2.4392263889312744,
      "step": 9956
    },
    {
      "epoch": 3.79791259765625e-05,
      "step": 9956,
      "training_step_time": 7.473442792892456
    },
    {
      "epoch": 3.79791259765625e-05,
      "model_forward_time": 2.437103748321533,
      "step": 9956
    },
    {
      "epoch": 3.79791259765625e-05,
      "step": 9956,
      "training_step_time": 10.307500123977661
    },
    {
      "epoch": 3.798294067382813e-05,
      "model_forward_time": 2.7936179637908936,
      "step": 9957
    },
    {
      "epoch": 3.798294067382813e-05,
      "step": 9957,
      "training_step_time": 7.831033229827881
    },
    {
      "epoch": 3.798294067382813e-05,
      "model_forward_time": 3.1429107189178467,
      "step": 9957
    },
    {
      "epoch": 3.798294067382813e-05,
      "step": 9957,
      "training_step_time": 9.920992374420166
    },
    {
      "epoch": 3.798294067382813e-05,
      "model_forward_time": 2.433831214904785,
      "step": 9957
    },
    {
      "dynamics_loss_avg": 0.045431346632540226,
      "epoch": 3.798294067382813e-05,
      "step": 9957
    },
    {
      "action_loss_avg": 0.014431120501831174,
      "epoch": 3.798294067382813e-05,
      "step": 9957
    },
    {
      "epoch": 3.798294067382813e-05,
      "step": 9957,
      "training_step_time": 7.465953826904297
    },
    {
      "epoch": 3.798294067382813e-05,
      "model_forward_time": 3.136922597885132,
      "step": 9957
    },
    {
      "epoch": 3.798294067382813e-05,
      "step": 9957,
      "training_step_time": 10.020704746246338
    },
    {
      "epoch": 3.798675537109375e-05,
      "model_forward_time": 2.4437246322631836,
      "step": 9958
    },
    {
      "epoch": 3.798675537109375e-05,
      "step": 9958,
      "training_step_time": 7.4786858558654785
    },
    {
      "epoch": 3.798675537109375e-05,
      "model_forward_time": 2.6939151287078857,
      "step": 9958
    },
    {
      "epoch": 3.798675537109375e-05,
      "step": 9958,
      "training_step_time": 7.715909957885742
    },
    {
      "epoch": 3.798675537109375e-05,
      "model_forward_time": 3.1427512168884277,
      "step": 9958
    },
    {
      "epoch": 3.798675537109375e-05,
      "step": 9958,
      "training_step_time": 9.913683652877808
    },
    {
      "epoch": 3.798675537109375e-05,
      "model_forward_time": 3.1301681995391846,
      "step": 9958
    },
    {
      "epoch": 3.798675537109375e-05,
      "step": 9958,
      "training_step_time": 10.005478858947754
    },
    {
      "epoch": 3.7990570068359374e-05,
      "model_forward_time": 2.4511899948120117,
      "step": 9959
    },
    {
      "epoch": 3.7990570068359374e-05,
      "step": 9959,
      "training_step_time": 7.47556734085083
    },
    {
      "epoch": 3.7990570068359374e-05,
      "model_forward_time": 3.129260778427124,
      "step": 9959
    },
    {
      "epoch": 3.7990570068359374e-05,
      "step": 9959,
      "training_step_time": 9.896981716156006
    },
    {
      "epoch": 3.7990570068359374e-05,
      "model_forward_time": 2.7453651428222656,
      "step": 9959
    },
    {
      "epoch": 3.7990570068359374e-05,
      "step": 9959,
      "training_step_time": 7.769148111343384
    },
    {
      "epoch": 3.7990570068359374e-05,
      "model_forward_time": 3.133969306945801,
      "step": 9959
    },
    {
      "epoch": 3.7990570068359374e-05,
      "step": 9959,
      "training_step_time": 10.335816860198975
    },
    {
      "epoch": 3.7994384765625e-05,
      "grad_norm": 0.4424822926521301,
      "learning_rate": 1e-05,
      "loss": 0.0747,
      "step": 9960
    },
    {
      "epoch": 3.7994384765625e-05,
      "model_forward_time": 2.452908992767334,
      "step": 9960
    },
    {
      "dynamics_loss_avg": 0.05538744833320379,
      "epoch": 3.7994384765625e-05,
      "step": 9960
    },
    {
      "action_loss_avg": 0.018239131104201077,
      "epoch": 3.7994384765625e-05,
      "step": 9960
    },
    {
      "epoch": 3.7994384765625e-05,
      "step": 9960,
      "training_step_time": 7.486355781555176
    },
    {
      "epoch": 3.7994384765625e-05,
      "model_forward_time": 3.128178834915161,
      "step": 9960
    },
    {
      "epoch": 3.7994384765625e-05,
      "step": 9960,
      "training_step_time": 9.889891386032104
    },
    {
      "epoch": 3.7994384765625e-05,
      "model_forward_time": 2.4330930709838867,
      "step": 9960
    },
    {
      "epoch": 3.7994384765625e-05,
      "step": 9960,
      "training_step_time": 7.455707311630249
    },
    {
      "epoch": 3.7994384765625e-05,
      "model_forward_time": 2.431290626525879,
      "step": 9960
    },
    {
      "epoch": 3.7994384765625e-05,
      "step": 9960,
      "training_step_time": 10.288043737411499
    },
    {
      "epoch": 3.7998199462890626e-05,
      "model_forward_time": 2.441481113433838,
      "step": 9961
    },
    {
      "epoch": 3.7998199462890626e-05,
      "step": 9961,
      "training_step_time": 7.465595245361328
    },
    {
      "epoch": 3.7998199462890626e-05,
      "model_forward_time": 2.429516553878784,
      "step": 9961
    },
    {
      "epoch": 3.7998199462890626e-05,
      "step": 9961,
      "training_step_time": 7.451748847961426
    },
    {
      "epoch": 3.7998199462890626e-05,
      "model_forward_time": 2.422665596008301,
      "step": 9961
    },
    {
      "epoch": 3.7998199462890626e-05,
      "step": 9961,
      "training_step_time": 7.446990013122559
    },
    {
      "epoch": 3.7998199462890626e-05,
      "model_forward_time": 3.1289045810699463,
      "step": 9961
    },
    {
      "epoch": 3.7998199462890626e-05,
      "step": 9961,
      "training_step_time": 10.274681806564331
    },
    {
      "epoch": 3.800201416015625e-05,
      "model_forward_time": 3.1391961574554443,
      "step": 9962
    },
    {
      "epoch": 3.800201416015625e-05,
      "step": 9962,
      "training_step_time": 9.909224033355713
    },
    {
      "epoch": 3.800201416015625e-05,
      "model_forward_time": 2.7187492847442627,
      "step": 9962
    },
    {
      "epoch": 3.800201416015625e-05,
      "step": 9962,
      "training_step_time": 7.737706661224365
    },
    {
      "epoch": 3.800201416015625e-05,
      "model_forward_time": 3.133714437484741,
      "step": 9962
    },
    {
      "dynamics_loss_avg": 0.04687247257679701,
      "epoch": 3.800201416015625e-05,
      "step": 9962
    },
    {
      "action_loss_avg": 0.01784136947244406,
      "epoch": 3.800201416015625e-05,
      "step": 9962
    },
    {
      "epoch": 3.800201416015625e-05,
      "step": 9962,
      "training_step_time": 9.903071641921997
    },
    {
      "epoch": 3.800201416015625e-05,
      "model_forward_time": 3.130906105041504,
      "step": 9962
    },
    {
      "epoch": 3.800201416015625e-05,
      "step": 9962,
      "training_step_time": 10.313355445861816
    },
    {
      "epoch": 3.800582885742187e-05,
      "model_forward_time": 2.441640615463257,
      "step": 9963
    },
    {
      "epoch": 3.800582885742187e-05,
      "step": 9963,
      "training_step_time": 7.465277910232544
    },
    {
      "epoch": 3.800582885742187e-05,
      "model_forward_time": 3.134765148162842,
      "step": 9963
    },
    {
      "epoch": 3.800582885742187e-05,
      "step": 9963,
      "training_step_time": 9.897953271865845
    },
    {
      "epoch": 3.800582885742187e-05,
      "model_forward_time": 3.4257514476776123,
      "step": 9963
    },
    {
      "epoch": 3.800582885742187e-05,
      "step": 9963,
      "training_step_time": 10.206766605377197
    },
    {
      "epoch": 3.800582885742187e-05,
      "model_forward_time": 3.1361992359161377,
      "step": 9963
    },
    {
      "epoch": 3.800582885742187e-05,
      "step": 9963,
      "training_step_time": 10.027737617492676
    },
    {
      "epoch": 3.80096435546875e-05,
      "model_forward_time": 2.460960865020752,
      "step": 9964
    },
    {
      "epoch": 3.80096435546875e-05,
      "step": 9964,
      "training_step_time": 7.496337413787842
    },
    {
      "epoch": 3.80096435546875e-05,
      "model_forward_time": 3.144057273864746,
      "step": 9964
    },
    {
      "epoch": 3.80096435546875e-05,
      "step": 9964,
      "training_step_time": 9.91332483291626
    },
    {
      "epoch": 3.80096435546875e-05,
      "model_forward_time": 2.4415252208709717,
      "step": 9964
    },
    {
      "epoch": 3.80096435546875e-05,
      "step": 9964,
      "training_step_time": 7.469336032867432
    },
    {
      "epoch": 3.80096435546875e-05,
      "model_forward_time": 2.436119556427002,
      "step": 9964
    },
    {
      "epoch": 3.80096435546875e-05,
      "step": 9964,
      "training_step_time": 10.275824069976807
    },
    {
      "epoch": 3.8013458251953124e-05,
      "model_forward_time": 3.1474175453186035,
      "step": 9965
    },
    {
      "dynamics_loss_avg": 0.05245348699390888,
      "epoch": 3.8013458251953124e-05,
      "step": 9965
    },
    {
      "action_loss_avg": 0.020007054368034006,
      "epoch": 3.8013458251953124e-05,
      "step": 9965
    },
    {
      "epoch": 3.8013458251953124e-05,
      "step": 9965,
      "training_step_time": 9.923032760620117
    },
    {
      "epoch": 3.8013458251953124e-05,
      "model_forward_time": 2.4381191730499268,
      "step": 9965
    },
    {
      "epoch": 3.8013458251953124e-05,
      "step": 9965,
      "training_step_time": 7.459007740020752
    },
    {
      "epoch": 3.8013458251953124e-05,
      "model_forward_time": 3.138528347015381,
      "step": 9965
    },
    {
      "epoch": 3.8013458251953124e-05,
      "step": 9965,
      "training_step_time": 9.907884120941162
    },
    {
      "epoch": 3.8013458251953124e-05,
      "model_forward_time": 3.155243158340454,
      "step": 9965
    },
    {
      "epoch": 3.8013458251953124e-05,
      "step": 9965,
      "training_step_time": 10.259801626205444
    },
    {
      "epoch": 3.801727294921875e-05,
      "model_forward_time": 3.155412197113037,
      "step": 9966
    },
    {
      "epoch": 3.801727294921875e-05,
      "step": 9966,
      "training_step_time": 10.193466424942017
    },
    {
      "epoch": 3.801727294921875e-05,
      "model_forward_time": 3.1337218284606934,
      "step": 9966
    },
    {
      "epoch": 3.801727294921875e-05,
      "step": 9966,
      "training_step_time": 9.898888349533081
    },
    {
      "epoch": 3.801727294921875e-05,
      "model_forward_time": 2.430861234664917,
      "step": 9966
    },
    {
      "epoch": 3.801727294921875e-05,
      "step": 9966,
      "training_step_time": 7.452818155288696
    },
    {
      "epoch": 3.801727294921875e-05,
      "model_forward_time": 3.129085063934326,
      "step": 9966
    },
    {
      "epoch": 3.801727294921875e-05,
      "step": 9966,
      "training_step_time": 10.006360054016113
    },
    {
      "epoch": 3.802108764648438e-05,
      "model_forward_time": 3.1521987915039062,
      "step": 9967
    },
    {
      "epoch": 3.802108764648438e-05,
      "step": 9967,
      "training_step_time": 9.91615104675293
    },
    {
      "epoch": 3.802108764648438e-05,
      "model_forward_time": 3.463292121887207,
      "step": 9967
    },
    {
      "epoch": 3.802108764648438e-05,
      "step": 9967,
      "training_step_time": 10.230624675750732
    },
    {
      "epoch": 3.802108764648438e-05,
      "model_forward_time": 3.1385955810546875,
      "step": 9967
    },
    {
      "dynamics_loss_avg": 0.05648505743592978,
      "epoch": 3.802108764648438e-05,
      "step": 9967
    },
    {
      "action_loss_avg": 0.01917470106855035,
      "epoch": 3.802108764648438e-05,
      "step": 9967
    },
    {
      "epoch": 3.802108764648438e-05,
      "step": 9967,
      "training_step_time": 9.90845513343811
    },
    {
      "epoch": 3.802108764648438e-05,
      "model_forward_time": 3.1445415019989014,
      "step": 9967
    },
    {
      "epoch": 3.802108764648438e-05,
      "step": 9967,
      "training_step_time": 10.247018575668335
    },
    {
      "epoch": 3.802490234375e-05,
      "model_forward_time": 2.4617087841033936,
      "step": 9968
    },
    {
      "epoch": 3.802490234375e-05,
      "step": 9968,
      "training_step_time": 7.485159635543823
    },
    {
      "epoch": 3.802490234375e-05,
      "model_forward_time": 3.138554811477661,
      "step": 9968
    },
    {
      "epoch": 3.802490234375e-05,
      "step": 9968,
      "training_step_time": 9.902511835098267
    },
    {
      "epoch": 3.802490234375e-05,
      "model_forward_time": 3.136990785598755,
      "step": 9968
    },
    {
      "epoch": 3.802490234375e-05,
      "step": 9968,
      "training_step_time": 9.900674819946289
    },
    {
      "epoch": 3.802490234375e-05,
      "model_forward_time": 3.406494140625,
      "step": 9968
    },
    {
      "epoch": 3.802490234375e-05,
      "step": 9968,
      "training_step_time": 10.262756824493408
    },
    {
      "epoch": 3.802871704101562e-05,
      "model_forward_time": 2.444976329803467,
      "step": 9969
    },
    {
      "epoch": 3.802871704101562e-05,
      "step": 9969,
      "training_step_time": 7.470503330230713
    },
    {
      "epoch": 3.802871704101562e-05,
      "model_forward_time": 2.43295955657959,
      "step": 9969
    },
    {
      "epoch": 3.802871704101562e-05,
      "step": 9969,
      "training_step_time": 7.453918218612671
    },
    {
      "epoch": 3.802871704101562e-05,
      "model_forward_time": 3.13468861579895,
      "step": 9969
    },
    {
      "epoch": 3.802871704101562e-05,
      "step": 9969,
      "training_step_time": 9.903286457061768
    },
    {
      "epoch": 3.802871704101562e-05,
      "model_forward_time": 3.1363065242767334,
      "step": 9969
    },
    {
      "epoch": 3.802871704101562e-05,
      "step": 9969,
      "training_step_time": 10.344201564788818
    },
    {
      "epoch": 3.803253173828125e-05,
      "grad_norm": 0.3192601501941681,
      "learning_rate": 1e-05,
      "loss": 0.0759,
      "step": 9970
    },
    {
      "epoch": 3.803253173828125e-05,
      "model_forward_time": 2.7665657997131348,
      "step": 9970
    },
    {
      "dynamics_loss_avg": 0.06670820768922567,
      "epoch": 3.803253173828125e-05,
      "step": 9970
    },
    {
      "action_loss_avg": 0.025525388261303305,
      "epoch": 3.803253173828125e-05,
      "step": 9970
    },
    {
      "epoch": 3.803253173828125e-05,
      "step": 9970,
      "training_step_time": 7.79888391494751
    },
    {
      "epoch": 3.803253173828125e-05,
      "model_forward_time": 2.441479206085205,
      "step": 9970
    },
    {
      "epoch": 3.803253173828125e-05,
      "step": 9970,
      "training_step_time": 7.4706902503967285
    },
    {
      "epoch": 3.803253173828125e-05,
      "model_forward_time": 3.156545639038086,
      "step": 9970
    },
    {
      "epoch": 3.803253173828125e-05,
      "step": 9970,
      "training_step_time": 9.937046766281128
    },
    {
      "epoch": 3.803253173828125e-05,
      "model_forward_time": 3.1435604095458984,
      "step": 9970
    },
    {
      "epoch": 3.803253173828125e-05,
      "step": 9970,
      "training_step_time": 10.296725511550903
    },
    {
      "epoch": 3.8036346435546875e-05,
      "model_forward_time": 2.4630467891693115,
      "step": 9971
    },
    {
      "epoch": 3.8036346435546875e-05,
      "step": 9971,
      "training_step_time": 7.4917542934417725
    },
    {
      "epoch": 3.8036346435546875e-05,
      "model_forward_time": 3.4285662174224854,
      "step": 9971
    },
    {
      "epoch": 3.8036346435546875e-05,
      "step": 9971,
      "training_step_time": 10.198082685470581
    },
    {
      "epoch": 3.8036346435546875e-05,
      "model_forward_time": 3.133777379989624,
      "step": 9971
    },
    {
      "epoch": 3.8036346435546875e-05,
      "step": 9971,
      "training_step_time": 9.899676084518433
    },
    {
      "epoch": 3.8036346435546875e-05,
      "model_forward_time": 2.431454658508301,
      "step": 9971
    },
    {
      "epoch": 3.8036346435546875e-05,
      "step": 9971,
      "training_step_time": 10.2810537815094
    },
    {
      "epoch": 3.80401611328125e-05,
      "model_forward_time": 2.4441490173339844,
      "step": 9972
    },
    {
      "epoch": 3.80401611328125e-05,
      "step": 9972,
      "training_step_time": 7.475099325180054
    },
    {
      "epoch": 3.80401611328125e-05,
      "model_forward_time": 2.425323724746704,
      "step": 9972
    },
    {
      "epoch": 3.80401611328125e-05,
      "step": 9972,
      "training_step_time": 7.4593236446380615
    },
    {
      "epoch": 3.80401611328125e-05,
      "model_forward_time": 3.4508771896362305,
      "step": 9972
    },
    {
      "dynamics_loss_avg": 0.05669294521212578,
      "epoch": 3.80401611328125e-05,
      "step": 9972
    },
    {
      "action_loss_avg": 0.02122381222434342,
      "epoch": 3.80401611328125e-05,
      "step": 9972
    },
    {
      "epoch": 3.80401611328125e-05,
      "step": 9972,
      "training_step_time": 10.221922159194946
    },
    {
      "epoch": 3.80401611328125e-05,
      "model_forward_time": 2.4301533699035645,
      "step": 9972
    },
    {
      "epoch": 3.80401611328125e-05,
      "step": 9972,
      "training_step_time": 10.007245779037476
    },
    {
      "epoch": 3.804397583007813e-05,
      "model_forward_time": 1.754455804824829,
      "step": 9973
    },
    {
      "epoch": 3.804397583007813e-05,
      "step": 9973,
      "training_step_time": 5.209098815917969
    },
    {
      "epoch": 3.804397583007813e-05,
      "model_forward_time": 2.4320602416992188,
      "step": 9973
    },
    {
      "epoch": 3.804397583007813e-05,
      "step": 9973,
      "training_step_time": 7.449931859970093
    },
    {
      "epoch": 3.804397583007813e-05,
      "model_forward_time": 2.43045973777771,
      "step": 9973
    },
    {
      "epoch": 3.804397583007813e-05,
      "step": 9973,
      "training_step_time": 7.446407794952393
    },
    {
      "epoch": 3.804397583007813e-05,
      "model_forward_time": 3.13240647315979,
      "step": 9973
    },
    {
      "epoch": 3.804397583007813e-05,
      "step": 9973,
      "training_step_time": 10.263342380523682
    },
    {
      "epoch": 3.804779052734375e-05,
      "model_forward_time": 2.7553176879882812,
      "step": 9974
    },
    {
      "epoch": 3.804779052734375e-05,
      "step": 9974,
      "training_step_time": 7.786473989486694
    },
    {
      "epoch": 3.804779052734375e-05,
      "model_forward_time": 2.4315786361694336,
      "step": 9974
    },
    {
      "epoch": 3.804779052734375e-05,
      "step": 9974,
      "training_step_time": 7.449429750442505
    },
    {
      "epoch": 3.804779052734375e-05,
      "model_forward_time": 2.427032232284546,
      "step": 9974
    },
    {
      "epoch": 3.804779052734375e-05,
      "step": 9974,
      "training_step_time": 7.4446117877960205
    },
    {
      "epoch": 3.804779052734375e-05,
      "model_forward_time": 2.4309840202331543,
      "step": 9974
    },
    {
      "epoch": 3.804779052734375e-05,
      "step": 9974,
      "training_step_time": 10.312369346618652
    },
    {
      "epoch": 3.8051605224609374e-05,
      "model_forward_time": 2.441112756729126,
      "step": 9975
    },
    {
      "dynamics_loss_avg": 0.04769014976918697,
      "epoch": 3.8051605224609374e-05,
      "step": 9975
    },
    {
      "action_loss_avg": 0.012659107660874724,
      "epoch": 3.8051605224609374e-05,
      "step": 9975
    },
    {
      "epoch": 3.8051605224609374e-05,
      "step": 9975,
      "training_step_time": 7.470025062561035
    },
    {
      "epoch": 3.8051605224609374e-05,
      "model_forward_time": 2.4279277324676514,
      "step": 9975
    },
    {
      "epoch": 3.8051605224609374e-05,
      "step": 9975,
      "training_step_time": 7.757583141326904
    },
    {
      "epoch": 3.8051605224609374e-05,
      "model_forward_time": 2.42777419090271,
      "step": 9975
    },
    {
      "epoch": 3.8051605224609374e-05,
      "step": 9975,
      "training_step_time": 7.459588289260864
    },
    {
      "epoch": 3.8051605224609374e-05,
      "model_forward_time": 3.1310837268829346,
      "step": 9975
    },
    {
      "epoch": 3.8051605224609374e-05,
      "step": 9975,
      "training_step_time": 10.346038341522217
    },
    {
      "epoch": 3.8055419921875e-05,
      "model_forward_time": 2.446381092071533,
      "step": 9976
    },
    {
      "epoch": 3.8055419921875e-05,
      "step": 9976,
      "training_step_time": 7.479803562164307
    },
    {
      "epoch": 3.8055419921875e-05,
      "model_forward_time": 3.1294949054718018,
      "step": 9976
    },
    {
      "epoch": 3.8055419921875e-05,
      "step": 9976,
      "training_step_time": 9.904484272003174
    },
    {
      "epoch": 3.8055419921875e-05,
      "model_forward_time": 3.131410837173462,
      "step": 9976
    },
    {
      "epoch": 3.8055419921875e-05,
      "step": 9976,
      "training_step_time": 9.90142273902893
    },
    {
      "epoch": 3.8055419921875e-05,
      "model_forward_time": 2.741706132888794,
      "step": 9976
    },
    {
      "epoch": 3.8055419921875e-05,
      "step": 9976,
      "training_step_time": 10.283499002456665
    },
    {
      "epoch": 3.8059234619140626e-05,
      "model_forward_time": 3.142272472381592,
      "step": 9977
    },
    {
      "epoch": 3.8059234619140626e-05,
      "step": 9977,
      "training_step_time": 9.906808376312256
    },
    {
      "epoch": 3.8059234619140626e-05,
      "model_forward_time": 2.426835775375366,
      "step": 9977
    },
    {
      "epoch": 3.8059234619140626e-05,
      "step": 9977,
      "training_step_time": 7.447211503982544
    },
    {
      "epoch": 3.8059234619140626e-05,
      "model_forward_time": 2.4275965690612793,
      "step": 9977
    },
    {
      "dynamics_loss_avg": 0.04271143320947886,
      "epoch": 3.8059234619140626e-05,
      "step": 9977
    },
    {
      "action_loss_avg": 0.012160590570420026,
      "epoch": 3.8059234619140626e-05,
      "step": 9977
    },
    {
      "epoch": 3.8059234619140626e-05,
      "step": 9977,
      "training_step_time": 7.44691014289856
    },
    {
      "epoch": 3.8059234619140626e-05,
      "model_forward_time": 3.1257810592651367,
      "step": 9977
    },
    {
      "epoch": 3.8059234619140626e-05,
      "step": 9977,
      "training_step_time": 10.251975059509277
    },
    {
      "epoch": 3.806304931640625e-05,
      "model_forward_time": 3.4595093727111816,
      "step": 9978
    },
    {
      "epoch": 3.806304931640625e-05,
      "step": 9978,
      "training_step_time": 10.22716212272644
    },
    {
      "epoch": 3.806304931640625e-05,
      "model_forward_time": 2.4284017086029053,
      "step": 9978
    },
    {
      "epoch": 3.806304931640625e-05,
      "step": 9978,
      "training_step_time": 7.449008226394653
    },
    {
      "epoch": 3.806304931640625e-05,
      "model_forward_time": 2.4286956787109375,
      "step": 9978
    },
    {
      "epoch": 3.806304931640625e-05,
      "step": 9978,
      "training_step_time": 7.444192171096802
    },
    {
      "epoch": 3.806304931640625e-05,
      "model_forward_time": 3.125596523284912,
      "step": 9978
    },
    {
      "epoch": 3.806304931640625e-05,
      "step": 9978,
      "training_step_time": 10.015542984008789
    },
    {
      "epoch": 3.806686401367187e-05,
      "model_forward_time": 2.4442577362060547,
      "step": 9979
    },
    {
      "epoch": 3.806686401367187e-05,
      "step": 9979,
      "training_step_time": 7.484928369522095
    },
    {
      "epoch": 3.806686401367187e-05,
      "model_forward_time": 3.4277994632720947,
      "step": 9979
    },
    {
      "epoch": 3.806686401367187e-05,
      "step": 9979,
      "training_step_time": 10.203609466552734
    },
    {
      "epoch": 3.806686401367187e-05,
      "model_forward_time": 3.1411514282226562,
      "step": 9979
    },
    {
      "epoch": 3.806686401367187e-05,
      "step": 9979,
      "training_step_time": 9.923463106155396
    },
    {
      "epoch": 3.806686401367187e-05,
      "model_forward_time": 3.13313364982605,
      "step": 9979
    },
    {
      "epoch": 3.806686401367187e-05,
      "step": 9979,
      "training_step_time": 10.30028510093689
    },
    {
      "epoch": 3.80706787109375e-05,
      "grad_norm": 0.41825294494628906,
      "learning_rate": 1e-05,
      "loss": 0.0724,
      "step": 9980
    },
    {
      "epoch": 3.80706787109375e-05,
      "model_forward_time": 3.164607524871826,
      "step": 9980
    },
    {
      "dynamics_loss_avg": 0.04337187130004168,
      "epoch": 3.80706787109375e-05,
      "step": 9980
    },
    {
      "action_loss_avg": 0.01675337259657681,
      "epoch": 3.80706787109375e-05,
      "step": 9980
    },
    {
      "epoch": 3.80706787109375e-05,
      "step": 9980,
      "training_step_time": 9.969083070755005
    },
    {
      "epoch": 3.80706787109375e-05,
      "model_forward_time": 2.451768636703491,
      "step": 9980
    },
    {
      "epoch": 3.80706787109375e-05,
      "step": 9980,
      "training_step_time": 7.496432781219482
    },
    {
      "epoch": 3.80706787109375e-05,
      "model_forward_time": 2.9280166625976562,
      "step": 9980
    },
    {
      "epoch": 3.80706787109375e-05,
      "step": 9980,
      "training_step_time": 7.975755214691162
    },
    {
      "epoch": 3.80706787109375e-05,
      "model_forward_time": 3.153879404067993,
      "step": 9980
    },
    {
      "epoch": 3.80706787109375e-05,
      "step": 9980,
      "training_step_time": 10.516501665115356
    },
    {
      "epoch": 3.8074493408203125e-05,
      "model_forward_time": 3.187088966369629,
      "step": 9981
    },
    {
      "epoch": 3.8074493408203125e-05,
      "step": 9981,
      "training_step_time": 9.96672511100769
    },
    {
      "epoch": 3.8074493408203125e-05,
      "model_forward_time": 2.435574769973755,
      "step": 9981
    },
    {
      "epoch": 3.8074493408203125e-05,
      "step": 9981,
      "training_step_time": 7.466048002243042
    },
    {
      "epoch": 3.8074493408203125e-05,
      "model_forward_time": 3.138843536376953,
      "step": 9981
    },
    {
      "epoch": 3.8074493408203125e-05,
      "step": 9981,
      "training_step_time": 9.914371967315674
    },
    {
      "epoch": 3.8074493408203125e-05,
      "model_forward_time": 1.7474501132965088,
      "step": 9981
    },
    {
      "epoch": 3.8074493408203125e-05,
      "step": 9981,
      "training_step_time": 10.31779170036316
    },
    {
      "epoch": 3.807830810546875e-05,
      "model_forward_time": 2.8173086643218994,
      "step": 9982
    },
    {
      "epoch": 3.807830810546875e-05,
      "step": 9982,
      "training_step_time": 7.843766927719116
    },
    {
      "epoch": 3.807830810546875e-05,
      "model_forward_time": 3.137742757797241,
      "step": 9982
    },
    {
      "epoch": 3.807830810546875e-05,
      "step": 9982,
      "training_step_time": 9.910003662109375
    },
    {
      "epoch": 3.807830810546875e-05,
      "model_forward_time": 3.1483280658721924,
      "step": 9982
    },
    {
      "dynamics_loss_avg": 0.0418928062543273,
      "epoch": 3.807830810546875e-05,
      "step": 9982
    },
    {
      "action_loss_avg": 0.014670123951509595,
      "epoch": 3.807830810546875e-05,
      "step": 9982
    },
    {
      "epoch": 3.807830810546875e-05,
      "step": 9982,
      "training_step_time": 9.925097942352295
    },
    {
      "epoch": 3.807830810546875e-05,
      "model_forward_time": 3.1536073684692383,
      "step": 9982
    },
    {
      "epoch": 3.807830810546875e-05,
      "step": 9982,
      "training_step_time": 10.290607452392578
    },
    {
      "epoch": 3.808212280273438e-05,
      "model_forward_time": 2.457075595855713,
      "step": 9983
    },
    {
      "epoch": 3.808212280273438e-05,
      "step": 9983,
      "training_step_time": 7.497168064117432
    },
    {
      "epoch": 3.808212280273438e-05,
      "model_forward_time": 3.439484119415283,
      "step": 9983
    },
    {
      "epoch": 3.808212280273438e-05,
      "step": 9983,
      "training_step_time": 10.221383094787598
    },
    {
      "epoch": 3.808212280273438e-05,
      "model_forward_time": 2.4416086673736572,
      "step": 9983
    },
    {
      "epoch": 3.808212280273438e-05,
      "step": 9983,
      "training_step_time": 7.481332302093506
    },
    {
      "epoch": 3.808212280273438e-05,
      "model_forward_time": 3.145885944366455,
      "step": 9983
    },
    {
      "epoch": 3.808212280273438e-05,
      "step": 9983,
      "training_step_time": 10.026363134384155
    },
    {
      "epoch": 3.80859375e-05,
      "model_forward_time": 2.4494822025299072,
      "step": 9984
    },
    {
      "epoch": 3.80859375e-05,
      "step": 9984,
      "training_step_time": 7.486154079437256
    },
    {
      "epoch": 3.80859375e-05,
      "model_forward_time": 3.149357557296753,
      "step": 9984
    },
    {
      "epoch": 3.80859375e-05,
      "step": 9984,
      "training_step_time": 9.927364826202393
    },
    {
      "epoch": 3.80859375e-05,
      "model_forward_time": 3.466087818145752,
      "step": 9984
    },
    {
      "epoch": 3.80859375e-05,
      "step": 9984,
      "training_step_time": 10.245007038116455
    },
    {
      "epoch": 3.80859375e-05,
      "model_forward_time": 2.4439735412597656,
      "step": 9984
    },
    {
      "epoch": 3.80859375e-05,
      "step": 9984,
      "training_step_time": 10.35104513168335
    },
    {
      "epoch": 3.808975219726562e-05,
      "model_forward_time": 2.4644834995269775,
      "step": 9985
    },
    {
      "dynamics_loss_avg": 0.04921348579227924,
      "epoch": 3.808975219726562e-05,
      "step": 9985
    },
    {
      "action_loss_avg": 0.017133069038391114,
      "epoch": 3.808975219726562e-05,
      "step": 9985
    },
    {
      "epoch": 3.808975219726562e-05,
      "step": 9985,
      "training_step_time": 7.502868413925171
    },
    {
      "epoch": 3.808975219726562e-05,
      "model_forward_time": 2.440028667449951,
      "step": 9985
    },
    {
      "epoch": 3.808975219726562e-05,
      "step": 9985,
      "training_step_time": 7.4781458377838135
    },
    {
      "epoch": 3.808975219726562e-05,
      "model_forward_time": 2.442840337753296,
      "step": 9985
    },
    {
      "epoch": 3.808975219726562e-05,
      "step": 9985,
      "training_step_time": 7.475170135498047
    },
    {
      "epoch": 3.808975219726562e-05,
      "model_forward_time": 2.438257932662964,
      "step": 9985
    },
    {
      "epoch": 3.808975219726562e-05,
      "step": 9985,
      "training_step_time": 10.350877285003662
    },
    {
      "epoch": 3.809356689453125e-05,
      "model_forward_time": 3.5074503421783447,
      "step": 9986
    },
    {
      "epoch": 3.809356689453125e-05,
      "step": 9986,
      "training_step_time": 10.280831336975098
    },
    {
      "epoch": 3.809356689453125e-05,
      "model_forward_time": 2.44254207611084,
      "step": 9986
    },
    {
      "epoch": 3.809356689453125e-05,
      "step": 9986,
      "training_step_time": 7.470430135726929
    },
    {
      "epoch": 3.809356689453125e-05,
      "model_forward_time": 3.142364501953125,
      "step": 9986
    },
    {
      "epoch": 3.809356689453125e-05,
      "step": 9986,
      "training_step_time": 9.918540716171265
    },
    {
      "epoch": 3.809356689453125e-05,
      "model_forward_time": 2.43943452835083,
      "step": 9986
    },
    {
      "epoch": 3.809356689453125e-05,
      "step": 9986,
      "training_step_time": 10.308389663696289
    },
    {
      "epoch": 3.8097381591796875e-05,
      "model_forward_time": 2.4505655765533447,
      "step": 9987
    },
    {
      "epoch": 3.8097381591796875e-05,
      "step": 9987,
      "training_step_time": 7.4730753898620605
    },
    {
      "epoch": 3.8097381591796875e-05,
      "model_forward_time": 2.7273001670837402,
      "step": 9987
    },
    {
      "epoch": 3.8097381591796875e-05,
      "step": 9987,
      "training_step_time": 7.747075319290161
    },
    {
      "epoch": 3.8097381591796875e-05,
      "model_forward_time": 2.4360451698303223,
      "step": 9987
    },
    {
      "dynamics_loss_avg": 0.06180628575384617,
      "epoch": 3.8097381591796875e-05,
      "step": 9987
    },
    {
      "action_loss_avg": 0.020250423299148678,
      "epoch": 3.8097381591796875e-05,
      "step": 9987
    },
    {
      "epoch": 3.8097381591796875e-05,
      "step": 9987,
      "training_step_time": 7.457106351852417
    },
    {
      "epoch": 3.8097381591796875e-05,
      "model_forward_time": 3.1372597217559814,
      "step": 9987
    },
    {
      "epoch": 3.8097381591796875e-05,
      "step": 9987,
      "training_step_time": 10.293064594268799
    },
    {
      "epoch": 3.81011962890625e-05,
      "model_forward_time": 2.449828863143921,
      "step": 9988
    },
    {
      "epoch": 3.81011962890625e-05,
      "step": 9988,
      "training_step_time": 7.472620964050293
    },
    {
      "epoch": 3.81011962890625e-05,
      "model_forward_time": 3.144641876220703,
      "step": 9988
    },
    {
      "epoch": 3.81011962890625e-05,
      "step": 9988,
      "training_step_time": 9.909377574920654
    },
    {
      "epoch": 3.81011962890625e-05,
      "model_forward_time": 3.4241294860839844,
      "step": 9988
    },
    {
      "epoch": 3.81011962890625e-05,
      "step": 9988,
      "training_step_time": 10.196572542190552
    },
    {
      "epoch": 3.81011962890625e-05,
      "model_forward_time": 2.440401077270508,
      "step": 9988
    },
    {
      "epoch": 3.81011962890625e-05,
      "step": 9988,
      "training_step_time": 10.29886269569397
    },
    {
      "epoch": 3.810501098632813e-05,
      "model_forward_time": 2.454864025115967,
      "step": 9989
    },
    {
      "epoch": 3.810501098632813e-05,
      "step": 9989,
      "training_step_time": 7.480638742446899
    },
    {
      "epoch": 3.810501098632813e-05,
      "model_forward_time": 2.450536012649536,
      "step": 9989
    },
    {
      "epoch": 3.810501098632813e-05,
      "step": 9989,
      "training_step_time": 7.481861114501953
    },
    {
      "epoch": 3.810501098632813e-05,
      "model_forward_time": 2.436478614807129,
      "step": 9989
    },
    {
      "epoch": 3.810501098632813e-05,
      "step": 9989,
      "training_step_time": 7.459019899368286
    },
    {
      "epoch": 3.810501098632813e-05,
      "model_forward_time": 2.438283920288086,
      "step": 9989
    },
    {
      "epoch": 3.810501098632813e-05,
      "step": 9989,
      "training_step_time": 10.26395320892334
    },
    {
      "epoch": 3.810882568359375e-05,
      "grad_norm": 0.42432451248168945,
      "learning_rate": 1e-05,
      "loss": 0.0746,
      "step": 9990
    },
    {
      "epoch": 3.810882568359375e-05,
      "model_forward_time": 2.4516406059265137,
      "step": 9990
    },
    {
      "dynamics_loss_avg": 0.056289704889059065,
      "epoch": 3.810882568359375e-05,
      "step": 9990
    },
    {
      "action_loss_avg": 0.021467037592083217,
      "epoch": 3.810882568359375e-05,
      "step": 9990
    },
    {
      "epoch": 3.810882568359375e-05,
      "step": 9990,
      "training_step_time": 7.479938745498657
    },
    {
      "epoch": 3.810882568359375e-05,
      "model_forward_time": 2.435152053833008,
      "step": 9990
    },
    {
      "epoch": 3.810882568359375e-05,
      "step": 9990,
      "training_step_time": 7.460819721221924
    },
    {
      "epoch": 3.810882568359375e-05,
      "model_forward_time": 3.140014171600342,
      "step": 9990
    },
    {
      "epoch": 3.810882568359375e-05,
      "step": 9990,
      "training_step_time": 9.909514427185059
    },
    {
      "epoch": 3.810882568359375e-05,
      "model_forward_time": 2.4373815059661865,
      "step": 9990
    },
    {
      "epoch": 3.810882568359375e-05,
      "step": 9990,
      "training_step_time": 10.024145364761353
    },
    {
      "epoch": 3.8112640380859374e-05,
      "model_forward_time": 3.157658338546753,
      "step": 9991
    },
    {
      "epoch": 3.8112640380859374e-05,
      "step": 9991,
      "training_step_time": 9.926306962966919
    },
    {
      "epoch": 3.8112640380859374e-05,
      "model_forward_time": 3.423396110534668,
      "step": 9991
    },
    {
      "epoch": 3.8112640380859374e-05,
      "step": 9991,
      "training_step_time": 10.197982788085938
    },
    {
      "epoch": 3.8112640380859374e-05,
      "model_forward_time": 3.144561767578125,
      "step": 9991
    },
    {
      "epoch": 3.8112640380859374e-05,
      "step": 9991,
      "training_step_time": 9.916616916656494
    },
    {
      "epoch": 3.8112640380859374e-05,
      "model_forward_time": 3.1474649906158447,
      "step": 9991
    },
    {
      "epoch": 3.8112640380859374e-05,
      "step": 9991,
      "training_step_time": 10.019925355911255
    },
    {
      "epoch": 3.8116455078125e-05,
      "model_forward_time": 2.4579415321350098,
      "step": 9992
    },
    {
      "epoch": 3.8116455078125e-05,
      "step": 9992,
      "training_step_time": 7.4909021854400635
    },
    {
      "epoch": 3.8116455078125e-05,
      "model_forward_time": 3.1446619033813477,
      "step": 9992
    },
    {
      "epoch": 3.8116455078125e-05,
      "step": 9992,
      "training_step_time": 9.933566570281982
    },
    {
      "epoch": 3.8116455078125e-05,
      "model_forward_time": 3.4087178707122803,
      "step": 9992
    },
    {
      "dynamics_loss_avg": 0.05106824245303869,
      "epoch": 3.8116455078125e-05,
      "step": 9992
    },
    {
      "action_loss_avg": 0.01516196932643652,
      "epoch": 3.8116455078125e-05,
      "step": 9992
    },
    {
      "epoch": 3.8116455078125e-05,
      "step": 9992,
      "training_step_time": 10.189202070236206
    },
    {
      "epoch": 3.8116455078125e-05,
      "model_forward_time": 2.445256233215332,
      "step": 9992
    },
    {
      "epoch": 3.8116455078125e-05,
      "step": 9992,
      "training_step_time": 10.31727147102356
    },
    {
      "epoch": 3.8120269775390626e-05,
      "model_forward_time": 3.1657934188842773,
      "step": 9993
    },
    {
      "epoch": 3.8120269775390626e-05,
      "step": 9993,
      "training_step_time": 9.958683967590332
    },
    {
      "epoch": 3.8120269775390626e-05,
      "model_forward_time": 3.1419920921325684,
      "step": 9993
    },
    {
      "epoch": 3.8120269775390626e-05,
      "step": 9993,
      "training_step_time": 9.922056913375854
    },
    {
      "epoch": 3.8120269775390626e-05,
      "model_forward_time": 3.147531747817993,
      "step": 9993
    },
    {
      "epoch": 3.8120269775390626e-05,
      "step": 9993,
      "training_step_time": 9.926530599594116
    },
    {
      "epoch": 3.8120269775390626e-05,
      "model_forward_time": 3.4658522605895996,
      "step": 9993
    },
    {
      "epoch": 3.8120269775390626e-05,
      "step": 9993,
      "training_step_time": 10.336655378341675
    },
    {
      "epoch": 3.812408447265625e-05,
      "model_forward_time": 3.166808843612671,
      "step": 9994
    },
    {
      "epoch": 3.812408447265625e-05,
      "step": 9994,
      "training_step_time": 9.942196607589722
    },
    {
      "epoch": 3.812408447265625e-05,
      "model_forward_time": 2.439772844314575,
      "step": 9994
    },
    {
      "epoch": 3.812408447265625e-05,
      "step": 9994,
      "training_step_time": 7.470384836196899
    },
    {
      "epoch": 3.812408447265625e-05,
      "model_forward_time": 2.437509536743164,
      "step": 9994
    },
    {
      "epoch": 3.812408447265625e-05,
      "step": 9994,
      "training_step_time": 7.472803115844727
    },
    {
      "epoch": 3.812408447265625e-05,
      "model_forward_time": 3.1446309089660645,
      "step": 9994
    },
    {
      "epoch": 3.812408447265625e-05,
      "step": 9994,
      "training_step_time": 10.274376630783081
    },
    {
      "epoch": 3.812789916992187e-05,
      "model_forward_time": 3.4901535511016846,
      "step": 9995
    },
    {
      "dynamics_loss_avg": 0.049019440449774264,
      "epoch": 3.812789916992187e-05,
      "step": 9995
    },
    {
      "action_loss_avg": 0.018641907535493375,
      "epoch": 3.812789916992187e-05,
      "step": 9995
    },
    {
      "epoch": 3.812789916992187e-05,
      "step": 9995,
      "training_step_time": 10.269907236099243
    },
    {
      "epoch": 3.812789916992187e-05,
      "model_forward_time": 3.145664930343628,
      "step": 9995
    },
    {
      "epoch": 3.812789916992187e-05,
      "step": 9995,
      "training_step_time": 9.921119928359985
    },
    {
      "epoch": 3.812789916992187e-05,
      "model_forward_time": 3.1496100425720215,
      "step": 9995
    },
    {
      "epoch": 3.812789916992187e-05,
      "step": 9995,
      "training_step_time": 9.932590246200562
    },
    {
      "epoch": 3.812789916992187e-05,
      "model_forward_time": 3.147688150405884,
      "step": 9995
    },
    {
      "epoch": 3.812789916992187e-05,
      "step": 9995,
      "training_step_time": 10.28963017463684
    },
    {
      "epoch": 3.81317138671875e-05,
      "model_forward_time": 3.1549160480499268,
      "step": 9996
    },
    {
      "epoch": 3.81317138671875e-05,
      "step": 9996,
      "training_step_time": 9.936693906784058
    },
    {
      "epoch": 3.81317138671875e-05,
      "model_forward_time": 3.418186902999878,
      "step": 9996
    },
    {
      "epoch": 3.81317138671875e-05,
      "step": 9996,
      "training_step_time": 10.200429916381836
    },
    {
      "epoch": 3.81317138671875e-05,
      "model_forward_time": 3.1501541137695312,
      "step": 9996
    },
    {
      "epoch": 3.81317138671875e-05,
      "step": 9996,
      "training_step_time": 9.925656795501709
    },
    {
      "epoch": 3.81317138671875e-05,
      "model_forward_time": 3.1429169178009033,
      "step": 9996
    },
    {
      "epoch": 3.81317138671875e-05,
      "step": 9996,
      "training_step_time": 10.020260334014893
    },
    {
      "epoch": 3.8135528564453125e-05,
      "model_forward_time": 3.1668121814727783,
      "step": 9997
    },
    {
      "epoch": 3.8135528564453125e-05,
      "step": 9997,
      "training_step_time": 9.955118894577026
    },
    {
      "epoch": 3.8135528564453125e-05,
      "model_forward_time": 2.4418747425079346,
      "step": 9997
    },
    {
      "epoch": 3.8135528564453125e-05,
      "step": 9997,
      "training_step_time": 7.478520154953003
    },
    {
      "epoch": 3.8135528564453125e-05,
      "model_forward_time": 3.471313238143921,
      "step": 9997
    },
    {
      "dynamics_loss_avg": 0.056034795194864276,
      "epoch": 3.8135528564453125e-05,
      "step": 9997
    },
    {
      "action_loss_avg": 0.019508385518565773,
      "epoch": 3.8135528564453125e-05,
      "step": 9997
    },
    {
      "epoch": 3.8135528564453125e-05,
      "step": 9997,
      "training_step_time": 10.266628742218018
    },
    {
      "epoch": 3.8135528564453125e-05,
      "model_forward_time": 3.157353401184082,
      "step": 9997
    },
    {
      "epoch": 3.8135528564453125e-05,
      "step": 9997,
      "training_step_time": 10.292349576950073
    },
    {
      "epoch": 3.813934326171875e-05,
      "model_forward_time": 2.463322401046753,
      "step": 9998
    },
    {
      "epoch": 3.813934326171875e-05,
      "step": 9998,
      "training_step_time": 7.501533508300781
    },
    {
      "epoch": 3.813934326171875e-05,
      "model_forward_time": 3.1517603397369385,
      "step": 9998
    },
    {
      "epoch": 3.813934326171875e-05,
      "step": 9998,
      "training_step_time": 9.936206102371216
    },
    {
      "epoch": 3.813934326171875e-05,
      "model_forward_time": 3.150763750076294,
      "step": 9998
    },
    {
      "epoch": 3.813934326171875e-05,
      "step": 9998,
      "training_step_time": 9.934284687042236
    },
    {
      "epoch": 3.813934326171875e-05,
      "model_forward_time": 2.7798383235931396,
      "step": 9998
    },
    {
      "epoch": 3.813934326171875e-05,
      "step": 9998,
      "training_step_time": 10.324358463287354
    },
    {
      "epoch": 3.814315795898438e-05,
      "model_forward_time": 2.50555157661438,
      "step": 9999
    },
    {
      "epoch": 3.814315795898438e-05,
      "step": 9999,
      "training_step_time": 7.556682109832764
    },
    {
      "epoch": 3.814315795898438e-05,
      "model_forward_time": 3.141000747680664,
      "step": 9999
    },
    {
      "epoch": 3.814315795898438e-05,
      "step": 9999,
      "training_step_time": 9.914967060089111
    },
    {
      "epoch": 3.814315795898438e-05,
      "model_forward_time": 2.4322426319122314,
      "step": 9999
    },
    {
      "epoch": 3.814315795898438e-05,
      "step": 9999,
      "training_step_time": 7.460552453994751
    },
    {
      "epoch": 3.814315795898438e-05,
      "model_forward_time": 3.132009744644165,
      "step": 9999
    },
    {
      "epoch": 3.814315795898438e-05,
      "step": 9999,
      "training_step_time": 10.269116640090942
    },
    {
      "epoch": 3.814697265625e-05,
      "grad_norm": 0.37506335973739624,
      "learning_rate": 1e-05,
      "loss": 0.0737,
      "step": 10000
    },
    {
      "epoch": 3.814697265625e-05,
      "step": 10000,
      "total_flos": 0.0,
      "train_loss": 0.10201911686062813,
      "train_runtime": 431341.903,
      "train_samples_per_second": 0.742,
      "train_steps_per_second": 0.023
    }
  ],
  "logging_steps": 10,
  "max_steps": 10000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 500,
  "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
}